#! /bin/sh

# $NetBSD: genraidconf.sh,v 1.5 2011/05/19 00:24:22 agc Exp $

# Copyright (c) 2012 Alistair Crooks <agc@NetBSD.org>
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#

setsdir=/usr/build/release/i386/binary/sets
size=128	# in MB
image=usermode.img
h=usermode1.$(uname -n)
sets="base etc modules"
usermodedirs="/var.run /var.log /etc.cow /root.cow"
specialdirs="/kern /proc"
bar="=== "

while [ $# -gt 0 ]; do
	case "$1" in
	-S)	setsdir=$2; shift ;;
	-h)	h=$2; shift ;;
	-s)	size=$2; shift ;;
	-x)	set -x ;;
	*)	break ;;
	esac
	shift
done

if [ $# -gt 0 ]; then
	# take the next argument as being the image name
	image="$1"
	shift
fi

echo "${bar} making a new ${size} MB image in ${image} ${bar}"
dd if=/dev/zero of=${image} bs=1m count=${size}

echo "${bar} mounting image via vnd ${bar}"
sudo vnconfig vnd0 ${image}
sudo newfs /dev/rvnd0a
sudo mount /dev/vnd0a /mnt

echo "${bar} installing sets ${bar}"
(cd /mnt &&
	for s in ${sets}; do
		echo ${s}
		sudo tar xpzf ${setsdir}/${s}.tgz
	done
)

echo "${bar} performing customisations ${bar}"
sudo rm -f /mnt/etc/motd

tmp=/tmp/usermode.$$
cat > ${tmp} << EOF
# NetBSD/usermode /etc/fstab
/dev/ld0a       /               ffs     ro              1 1
#/dev/ld1a      /cow            ffs     rw,log,hidden   1 1
kernfs          /kern           kernfs  rw
ptyfs           /dev/pts        ptyfs   rw
procfs          /proc           procfs  rw
# mount /root as tmpfs on top of existing dir
tmpfs           /root.cow       tmpfs   rw,-s2M         0 0
/root.cow       /root           union   rw,hidden       0 0
# mount /etc as tmpfs on top of existing dir
tmpfs           /etc.cow        tmpfs   rw,-s12M        0 0
/etc.cow        /etc            union   rw,hidden       0 0
# mount /var/run as tmpfs on top of existing dir
tmpfs           /var.run        tmpfs   rw,-s1M         0 0
/var.run        /var/run        union   rw,hidden       - -
# mount /var/log as tmpfs on top of existing dir
tmpfs           /var.log        tmpfs   rw,-s10M        0 0
/var.log        /var/log        union   rw,hidden       - -
tmpfs           /var/db         tmpfs   rw,-s8M         0 0
tmpfs           /tmp            tmpfs   rw,-s32M        0 0
/dev/cd0a       /cdrom          cd9660  ro,noauto
EOF
sudo mv ${tmp} /mnt/etc/fstab

cat > ${tmp} << EOF
#       $NetBSD: rc.conf,v 1.96 2000/10/14 17:01:29 wiz Exp $
#
# see rc.conf(5) for more information.
#
# Use program=YES to enable program, NO to disable it. program_flags are
# passed to the program on the command line.
#

# Load the defaults in from /etc/defaults/rc.conf (if it's readable).
# These can be overridden below.
#
if [ -r /etc/defaults/rc.conf ]; then
        . /etc/defaults/rc.conf
fi

# If this is not set to YES, the system will drop into single-user mode.
#
rc_configured=YES

# make sure we have the right rw filesystem at boot
critical_filesystems_local="/var/db /var.run /var/run /var.log /var/log /etc.cow /etc /root.cow /root"

# Add local overrides below
#
sshd=YES

hostname=${h}
EOF
sudo mv ${tmp} /mnt/etc/rc.conf

echo "${bar} making extra directories ${bar}"
for d in ${usermodedirs}; do
	sudo mkdir -p /mnt/${d}
done
for d in ${specialdirs}; do
	sudo mkdir -p /mnt/${d}
done

echo "${bar} customising /var/tmp ${bar}"
sudo rm -rf /mnt/var/tmp
(cd /mnt/var && sudo ln -s /tmp tmp)

df /mnt

sudo umount /mnt
sudo vnconfig -u vnd0

exit 0