import ./make-test.nix ({ pkgs, ... }: with pkgs.lib; let debug = false; testVMConfig = vmName: attrs: { config, pkgs, ... }: { boot.kernelParams = let miniInit = '' #!${pkgs.stdenv.shell} -xe export PATH="${pkgs.coreutils}/bin:${pkgs.utillinux}/bin" ${pkgs.linuxPackages.virtualboxGuestAdditions}/sbin/VBoxService ${(attrs.vmScript or (const "")) pkgs} i=0 while [ ! -e /mnt-root/shutdown ]; do sleep 10 i=$(($i + 10)) [ $i -le 120 ] || fail done rm -f /mnt-root/boot-done /mnt-root/shutdown ''; in [ "console=tty0" "console=ttyS0" "ignore_loglevel" "boot.trace" "panic=1" "boot.panic_on_fail" "init=${pkgs.writeScript "mini-init.sh" miniInit}" ]; fileSystems."/" = { device = "vboxshare"; fsType = "vboxsf"; }; services.virtualboxGuest.enable = true; boot.initrd.kernelModules = [ "af_packet" "vboxsf" "virtio" "virtio_pci" "virtio_ring" "virtio_net" "vboxguest" ]; boot.initrd.extraUtilsCommands = '' cp -av -t "$out/bin/" \ "${pkgs.linuxPackages.virtualboxGuestAdditions}/sbin/mount.vboxsf" \ "${pkgs.utillinux}/bin/unshare" ${(attrs.extraUtilsCommands or (const "")) pkgs} ''; boot.initrd.postMountCommands = '' touch /mnt-root/boot-done hostname "${vmName}" mkdir -p /nix/store unshare -m "@shell@" -c ' mount -t vboxsf nixstore /nix/store exec "$stage2Init" ' poweroff -f ''; system.requiredKernelConfig = with config.lib.kernelConfig; [ (isYes "SERIAL_8250_CONSOLE") (isYes "SERIAL_8250") ]; }; mkLog = logfile: tag: let rotated = map (i: "${logfile}.${toString i}") (range 1 9); all = concatMapStringsSep " " (f: "\"${f}\"") ([logfile] ++ rotated); logcmd = "tail -F ${all} 2> /dev/null | logger -t \"${tag}\""; in optionalString debug "$machine->execute(ru '${logcmd} & disown');"; testVM = vmName: vmScript: let cfg = (import ../lib/eval-config.nix { system = "i686-linux"; modules = [ ../modules/profiles/minimal.nix (testVMConfig vmName vmScript) ]; }).config; in pkgs.vmTools.runInLinuxVM (pkgs.runCommand "virtualbox-image" { preVM = '' mkdir -p "$out" diskImage="$(pwd)/qimage" ${pkgs.vmTools.qemu}/bin/qemu-img create -f raw "$diskImage" 100M ''; postVM = '' echo "creating VirtualBox disk image..." ${pkgs.vmTools.qemu}/bin/qemu-img convert -f raw -O vdi \ "$diskImage" "$out/disk.vdi" ''; buildInputs = [ pkgs.utillinux pkgs.perl ]; } '' ${pkgs.parted}/sbin/parted /dev/vda mklabel msdos ${pkgs.parted}/sbin/parted /dev/vda -- mkpart primary ext2 1M -1s . /sys/class/block/vda1/uevent mknod /dev/vda1 b $MAJOR $MINOR ${pkgs.e2fsprogs}/sbin/mkfs.ext4 /dev/vda1 ${pkgs.e2fsprogs}/sbin/tune2fs -c 0 -i 0 /dev/vda1 mkdir /mnt mount /dev/vda1 /mnt cp "${cfg.system.build.kernel}/bzImage" /mnt/linux cp "${cfg.system.build.initialRamdisk}/initrd" /mnt/initrd ${pkgs.grub2}/bin/grub-install --boot-directory=/mnt /dev/vda cat > /mnt/grub/grub.cfg <execute(ru $cmd); return $status == 0; } sub cleanup_${name} { $machine->execute(ru "VBoxManage controlvm ${name} poweroff") if checkRunning_${name}; $machine->succeed("rm -rf ${sharePath}"); $machine->succeed("mkdir -p ${sharePath}"); $machine->succeed("chown alice.users ${sharePath}"); } sub createVM_${name} { vbm("createvm --name ${name} ${createFlags}"); vbm("modifyvm ${name} ${vmFlags}"); vbm("setextradata ${name} VBoxInternal/PDM/HaltOnReset 1"); vbm("storagectl ${name} ${controllerFlags}"); vbm("storageattach ${name} ${diskFlags}"); vbm("sharedfolder add ${name} ${sharedFlags}"); vbm("sharedfolder add ${name} ${nixstoreFlags}"); cleanup_${name}; ${mkLog "$HOME/VirtualBox VMs/${name}/Logs/VBox.log" "HOST-${name}"} } sub destroyVM_${name} { cleanup_${name}; vbm("unregistervm ${name} --delete"); } sub waitForVMBoot_${name} { $machine->execute(ru( 'set -e; i=0; '. 'while ! test -e ${sharePath}/boot-done; do '. 'sleep 10; i=$(($i + 10)); [ $i -le 3600 ]; '. 'VBoxManage list runningvms | grep -q "^\"${name}\""; '. 'done' )); } sub waitForIP_${name} ($) { my $property = "/VirtualBox/GuestInfo/Net/$_[0]/V4/IP"; my $getip = "VBoxManage guestproperty get ${name} $property | ". "sed -n -e 's/^Value: //p'"; my $ip = $machine->succeed(ru( 'for i in $(seq 1000); do '. 'if ipaddr="$('.$getip.')" && [ -n "$ipaddr" ]; then '. 'echo "$ipaddr"; exit 0; '. 'fi; '. 'sleep 1; '. 'done; '. 'echo "Could not get IPv4 address for ${name}!" >&2; '. 'exit 1' )); chomp $ip; return $ip; } sub waitForStartup_${name} { for (my $i = 0; $i <= 120; $i += 10) { $machine->sleep(10); return if checkRunning_${name}; } die "VirtualBox VM didn't start up within 2 minutes"; } sub waitForShutdown_${name} { for (my $i = 0; $i <= 120; $i += 10) { $machine->sleep(10); return unless checkRunning_${name}; } die "VirtualBox VM didn't shut down within 2 minutes"; } sub shutdownVM_${name} { $machine->succeed(ru "touch ${sharePath}/shutdown"); $machine->waitUntilSucceeds( "test ! -e ${sharePath}/shutdown ". " -a ! -e ${sharePath}/boot-done" ); waitForShutdown_${name}; } ''; }; hostonlyVMFlags = [ "--nictype1 virtio" "--nictype2 virtio" "--nic2 hostonly" "--hostonlyadapter2 vboxnet0" ]; dhcpScript = pkgs: '' ${pkgs.dhcp}/bin/dhclient \ -lf /run/dhcp.leases \ -pf /run/dhclient.pid \ -v eth0 eth1 otherIP="$(${pkgs.netcat}/bin/netcat -clp 1234 || :)" ${pkgs.iputils}/bin/ping -I eth1 -c1 "$otherIP" echo "$otherIP reachable" | ${pkgs.netcat}/bin/netcat -clp 5678 || : ''; vboxVMs = mapAttrs createVM { simple = {}; test1.vmFlags = hostonlyVMFlags; test1.vmScript = dhcpScript; test2.vmFlags = hostonlyVMFlags; test2.vmScript = dhcpScript; }; in { name = "virtualbox"; machine = { pkgs, lib, config, ... }: { imports = let mkVMConf = name: val: val.machine // { key = "${name}-config"; }; vmConfigs = mapAttrsToList mkVMConf vboxVMs; in [ ./common/user-account.nix ./common/x11.nix ] ++ vmConfigs; virtualisation.memorySize = 768; services.virtualboxHost.enable = true; users.extraUsers.alice.extraGroups = let inherit (config.services.virtualboxHost) enableHardening; in lib.mkIf enableHardening (lib.singleton "vboxusers"); }; testScript = '' sub ru ($) { my $esc = $_[0] =~ s/'/'\\${"'"}'/gr; return "su - alice -c '$esc'"; } sub vbm { $machine->succeed(ru("VBoxManage ".$_[0])); }; ${concatStrings (mapAttrsToList (_: getAttr "testSubs") vboxVMs)} $machine->waitForX; ${mkLog "$HOME/.config/VirtualBox/VBoxSVC.log" "HOST-SVC"} createVM_simple; subtest "simple-gui", sub { $machine->succeed(ru "VirtualBox &"); $machine->waitForWindow(qr/Oracle VM VirtualBox Manager/); $machine->sleep(5); $machine->screenshot("gui_manager_started"); $machine->sendKeys("ret"); $machine->screenshot("gui_manager_sent_startup"); waitForStartup_simple; $machine->screenshot("gui_started"); waitForVMBoot_simple; $machine->screenshot("gui_booted"); shutdownVM_simple; $machine->sleep(5); $machine->screenshot("gui_stopped"); $machine->sendKeys("ctrl-q"); $machine->sleep(5); $machine->screenshot("gui_manager_stopped"); }; cleanup_simple; subtest "simple-cli", sub { vbm("startvm simple"); waitForStartup_simple; $machine->screenshot("cli_started"); waitForVMBoot_simple; $machine->screenshot("cli_booted"); shutdownVM_simple; }; subtest "privilege-escalation", sub { $machine->fail("test -e '/root/VirtualBox VMs'"); $machine->fail("test -e '/root/.config/VirtualBox'"); $machine->succeed("test -e '/home/alice/VirtualBox VMs'"); }; destroyVM_simple; subtest "net-hostonlyif", sub { createVM_test1; createVM_test2; vbm("startvm test1"); waitForStartup_test1; vbm("startvm test2"); waitForStartup_test2; waitForVMBoot_test1; waitForVMBoot_test2; $machine->screenshot("net_booted"); my $test1IP = waitForIP_test1 1; my $test2IP = waitForIP_test2 1; $machine->succeed("echo '$test2IP' | netcat -c '$test1IP' 1234"); $machine->succeed("echo '$test1IP' | netcat -c '$test2IP' 1234"); $machine->waitUntilSucceeds("netcat -c '$test1IP' 5678 >&2"); $machine->waitUntilSucceeds("netcat -c '$test2IP' 5678 >&2"); shutdownVM_test1; shutdownVM_test2; destroyVM_test1; destroyVM_test2; }; ''; })