#!/bin/bash
set -e

#
# bootstrap netbsd on ext2fs
#
# XEN -- without DOS partitions
# KVM -- with DOS partitions (losetup)
#
# TODO
# - enable bare-metal (change hypervisor var name)
#

function usage {
	echo "${0##*/} <guest> <vdisk>"
	exit 1
}

[[ -z $2 ]] && usage
guest=$1
vdisk=$2

[[ ! -f network.conf ]] && echo network.conf missing && exit 1
source network.conf

[[ -z $kernel ]] && echo \$kernel not defined -- tune network.conf first && exit 1
[[ -z $sets ]] && echo \$sets not defined -- tune network.conf first && exit 1

[[ ! -f $kernel ]] && echo cannot find $kernel && exit 1
[[ ! -f $sets/base.tar.xz ]] && echo cannot find $sets/base.tar.xz not found && exit 1

if [[ -x `which xl 2>/dev/null` ]]; then
	[[ -n `xl li | grep $guest` ]] && echo $guest is already running && exti 1
fi
[[ -n `mount | grep lala` ]] && echo lala/ already mounted && exit 1
# checking against vdisk allows to have other loops alive
[[ -n `losetup --all | grep $vdisk` ]] && echo loop device for $vdisk already exists && exit 1

echo
echo VDISK SETUP
echo

if [[ $hypervisor = kvm ]]; then
	[[ ! `which sfdisk` ]] && echo install sfdisk first && exit 1
	[[ ! `which partprobe` ]] && echo install partprobe first && exit 1

	# backup vdisk filename for kvm guest xml config
	vdisk_file=$vdisk

	# e.g. /dev/loop0
	echo -n kvm -- prepare looped vdisk ...
	vdisk_loop=`losetup --find --partscan --show $vdisk` && echo $vdisk_loop

	# e.g. /dev/loop0p1
	echo -n kvm -- vdisk partition ...
	vdisk_partition=${vdisk_loop}p1 && echo $vdisk_partition

	echo -n kvm -- define overall space available ...
	total=`fdisk -l $disk | head -1 | awk '{print $7}'`
	[[ -z $total ]] && echo error: could not define total && exit 1
	echo $total

	echo -n kvm -- prepare DOS partition table ...
	cat > $guest.sfdisk <<EOF && echo done
label: dos
label-id: 0x8ae$((RANDOM % 10))$((RANDOM % 10))3cc
device: $vdisk_loop
unit: sectors

$vdisk_partition : start=2048, size=$(( total - 2048 )), type=a9, bootable
EOF

	echo -n kvm -- apply DOS partition table to $vdisk_loop ...
	sfdisk --quiet $vdisk_loop < $guest.sfdisk && echo done

	#echo DEBUG:
	#ls -lF /dev/loop*

	#echo -n kvm -- update partitions ...
	#partprobe $vdisk_loop && sleep 1 && echo done

	#echo DEBUG:
	#ls -lF /dev/loop*

	# use same variable as for XEN thereafter
	vdisk=$vdisk_partition
	[[ ! -b $vdisk ]] && echo error: cannot find block device $vdisk && exit 1
fi

mkfs.ext2 -F -O^dir_index,^ext_attr $vdisk

echo -n mounting ...
mkdir -p lala/
mount -t ext2 $vdisk lala/ && echo done

echo
echo SYSTEM SETUP
echo

for set in base comp etc man misc modules text; do
	echo -n extracting $set ...
	tar xJph --numeric-owner -f $sets/$set.tar.xz -C lala/ && echo done
	# tar xzph
done; unset set

OWD=`pwd`
echo -n making device files ...
cd lala/dev/
#grep ' -F netbsd' MAKEDEV
#grep ' -R"' MAKEDEV
#grep ' -r"' MAKEDEV
#grep ' -g ' MAKEDEV
sed -r 's/ -F netbsd//;
	s/ -R"/"/;
	s/ -r"/"/;
	s/ -g [^ ]+ -u [^ ]+//;
	' MAKEDEV > MAKEDEV.linux
#diff grep exists non-zero
#diff -bu MAKEDEV MAKEDEV.linux 
#grep ' -F netbsd' MAKEDEV.linux
#grep ' -R"' MAKEDEV.linux
#grep ' -r"' MAKEDEV.linux
#grep ' -g ' MAKEDEV.linux
chmod +x MAKEDEV.linux
./MAKEDEV.linux all 2>/dev/null
./MAKEDEV.linux opty && echo done
cd $OWD/
unset OWD

# no executable bit required on netbsd
mv lala/etc/rc.local lala/etc/rc.local.dist
cat > lala/etc/rc.local <<EOF
if [ ! -f /.makedev_done ]; then
	# CWD is /
	cd /dev/
	echo -n MAKEDEV all ...
	./MAKEDEV all && date > /.makedev_done && echo done
	echo -n MAKEDEV opty ...
	./MAKEDEV opty && echo done
	cd /
	echo OWNERSHIP FIXED E.G.
	ls -lF /dev/cgd0
	ls -lF /dev/altq/altq
fi
EOF

mv -i lala/etc/passwd lala/etc/passwd.dist
sed '1s@/bin/sh@/bin/ksh@' lala/etc/passwd.dist > lala/etc/passwd
#diff -u lala/etc/passwd.dist lala/etc/passwd

mv -i lala/etc/master.passwd lala/etc/master.passwd.dist
sed '1s@/bin/sh@/bin/ksh@' lala/etc/master.passwd.dist > lala/etc/master.passwd
#diff -u lala/etc/master.passwd.dist lala/etc/master.passwd

if [[ $hypervisor = xen ]]; then
	echo -n xen -- writing fstab ...
	echo /dev/xbd0a / ffs rw 0 1 > lala/etc/fstab && echo done
elif [[ $hypervisor = kvm ]]; then
	echo -n kvm -- writing fstab ...
	echo /dev/ld0e / ext2fs rw 0 1 > lala/etc/fstab && echo done
else
	echo warning: cannot setup fstab properly without hypervisor variable defined
fi

echo -n adding more fstab entries ...
cat >> lala/etc/fstab <<EOF && echo done
ptyfs /dev/pts ptyfs rw 0 0
tmpfs /tmp tmpfs rw 0 0
#procfs /proc procfs ro,linux 0 0
EOF

echo -n rc.conf ...
mv -i lala/etc/rc.conf lala/etc/rc.conf.dist
cat > lala/etc/rc.conf <<EOF && echo done
. /etc/defaults/rc.conf
rc_silent=no # /var/run/rc.log
rc_configured=yes
inetd=no
wscons=no

no_swap=yes
savecore=no
EOF

echo -n kvm -- enable powerd ...
[[ $hypervisor = kvm ]] && echo powerd=yes >> lala/etc/rc.conf && echo done || echo unused

echo
echo NETWORK SETUP
echo

echo -n hostname ...
echo $host > lala/etc/myname && echo done

echo -n $netif ...
echo inet $cidr up > lala/etc/ifconfig.$netif && echo done

echo -n hosts ...
cat > lala/etc/hosts <<EOF && echo done
127.0.0.1       localhost
::1             localhost

${cidr%/*} $host

EOF

if [[ -n $gw ]]; then
        echo -n gateway ...
        echo $gw > lala/etc/mygate && echo done
	echo -n hosts/gw ...
	cat >> lala/etc/hosts <<-EOF && echo done
	$gw gw
	EOF
else
        echo no gateway
fi

rm -f lala/etc/resolv.conf
for ip in $dns; do
        echo -n nameserver $ip ...
        cat >> lala/etc/resolv.conf <<EOF && echo done
nameserver $ip
EOF
done; unset ip

echo -n authorized_keys ...
if [[ -f lala/root/.ssh/authorized_keys ]]; then
        echo already:
        cat lala/root/.ssh/authorized_keys
else
        mkdir lala/root/.ssh/
        chmod 700 lala/root/.ssh/
        cat > lala/root/.ssh/authorized_keys <<EOF9
$pubkey
EOF9
        chmod 600 lala/root/.ssh/authorized_keys && echo done
fi

echo -n sshd_config ...
mv lala/etc/ssh/sshd_config lala/etc/ssh/sshd_config.dist
sed -r '/^[[:space:]]*#/d; /^[[:space:]]*$/d; /UsePam /d' lala/etc/ssh/sshd_config.dist \
	> lala/etc/ssh/sshd_config
cat >> lala/etc/ssh/sshd_config <<EOF9 && echo done

AddressFamily inet
#ListenAddress ${cidr%/*}
Protocol 2
Port 22
AllowGroups wheel
PermitRootLogin without-password
PasswordAuthentication no
PermitEmptyPasswords no
StrictModes yes
X11Forwarding no
ChallengeResponseAuthentication no
UsePAM no
UseDNS no
HostKey /etc/ssh/ssh_host_ed25519_key
PrintMotd no
EOF9

echo -n enable sshd ...
echo sshd=yes >> lala/etc/rc.conf && echo done

echo
echo GUEST SETUP
echo

if [[ $hypervisor = xen ]]; then
	echo xen -- TODO enable TMEM?
fi

if [[ $hypervisor = kvm ]]; then
	[[ -f lala/netbsd ]] && error: found lala/netbsd -- why is there a kernel in there already? && exit 1

	echo -n kvm -- deploy kernel $kernel ...
	cp $kernel lala/netbsd && echo done

	echo kvm -- write grub2 boot-code to $vdisk_file
	grub-install --version
        grub-install --target=i386-pc --boot-directory=lala/boot $vdisk_file

	echo -n kvm -- write grub2.cfg ...
	cat > lala/boot/grub/grub.cfg <<EOF && echo done
serial --unit=0 --speed=115200
terminal_input serial
terminal_output serial

set timeout=3

menuentry "netbsd" {
        insmod ext2
        set root=(hd0,msdos1)
        knetbsd /netbsd --serial=com0,115200 --root=ld0e
}
EOF
fi

echo -n un-mounting ...
umount lala/ && echo done
rmdir lala/

# now that e.g. /dev/loop0p1 is unmounted, we can also get rid of /dev/loop0
if [[ $hypervisor = kvm ]]; then
	echo -n kvm -- un-looping $vdisk_loop ...
	losetup --detach=$vdisk_loop && echo done
fi

echo -n xen guest skeleton ...
[[ $hypervisor = xen ]] && cat > $guest <<EOF && echo done || echo unused
kernel = "$kernel"
root = "xbd0a"
#extra = "-v -s"
name = "$guest"
memory = 1024
vcpus = 3
disk = [ 'phy:$vdisk,xvda,w' ]
vif = [ 'bridge=xenbr0, vifname=$guest.0',
	'bridge=guestbr0, vifname=$guest.1' ]
EOF
#disk = [ 'tap:tapdisk:aio:/data/guests/$guest/$guest.sparse,xvda,w' ]


echo -n kvm guest skeleton ...
[[ $hypervisor = kvm ]] && cat > $guest.xml <<EOF && echo done || echo unused
<domain type='kvm'>
  <name>$guest</name>
  <memory unit='GiB'>1</memory>
  <vcpu placement='static'>3</vcpu>
  <os>
    <type arch='x86_64' machine='q35'>hvm</type>
    <boot dev='hd'/>
  </os>
  <devices>
    <emulator>/usr/bin/qemu-system-x86_64</emulator>
    <disk type='file' device='disk'>
      <driver name='qemu' type='raw'/>
      <source file='/data/guests/$guest/$vdisk_file'/>
      <target dev='vda' bus='virtio'/>
    </disk>
    <interface type='bridge'>
      <source bridge='virbr0'/>
      <model type='virtio'/>
    </interface>
    <serial type='pty'>
      <target type='isa-serial' port='0'>
        <model name='isa-serial'/>
      </target>
    </serial>
    <console type='pty'>
      <target type='serial' port='0'/>
    </console>
  </devices>
</domain>
EOF

cat <<EOF

$vdisk is ready

EOF

[[ $hypervisor = xen ]] && cat <<EOF
you can now run

	xl create $guest -c

EOF

[[ $hypervisor = kvm ]] && cat <<EOF
you can now run

	virsh create $guest.xml --console

EOF