# This module creates a virtual machine from the NixOS configuration.
# Building the `config.system.build.vm' attribute gives you a command
# that starts a KVM/QEMU VM running the NixOS configuration defined in
# `config'. By default, the Nix store is shared read-only with the
# host, which makes (re)building VMs very efficient.

{
  config,
  lib,
  pkgs,
  options,
  ...
}:

with lib;

let

  qemu-common = import ../../lib/qemu-common.nix { inherit lib pkgs; };

  cfg = config.virtualisation;

  qemu = cfg.qemu.package;

  hostPkgs = cfg.host.pkgs;

  consoles = lib.concatMapStringsSep " " (c: "console=${c}") cfg.qemu.consoles;

  driveOptions =
    { ... }:
    {

      options = {
        driveExtraOpts = mkOption {
          type = types.attrsOf types.str;
          default = { };
          description = "Extra options passed to drive flag.";
        };

        deviceExtraOpts = mkOption {
          type = types.attrsOf types.str;
          default = { };
          description = "Extra options passed to device flag.";
        };

        name = mkOption {
          type = types.nullOr types.str;
          default = null;
          description = "A name for the drive. Must be unique in the drives list. Not passed to qemu.";
        };

      };

    };

  selectPartitionTableLayout =
    {
      useEFIBoot,
      useDefaultFilesystems,
      useBIOSBoot,
    }:
    if useDefaultFilesystems then
      if useEFIBoot then
        "efi"
      else if useBIOSBoot then
        "legacy+boot"
      else
        "legacy"
    else
      "none";

  driveCmdline =
    idx:
    {
      file,
      driveExtraOpts,
      deviceExtraOpts,
      ...
    }:
    let
      drvId = "drive${toString idx}";
      mkKeyValue = generators.mkKeyValueDefault { } "=";
      mkOpts = opts: concatStringsSep "," (mapAttrsToList mkKeyValue opts);
      driveOpts = mkOpts (
        driveExtraOpts
        // {
          index = idx;
          id = drvId;
          "if" = "none";
          inherit file;
        }
      );
      deviceOpts = mkOpts (
        deviceExtraOpts
        // {
          drive = drvId;
        }
      );
      device =
        if cfg.qemu.diskInterface == "scsi" then
          "-device lsi53c895a -device scsi-hd,${deviceOpts}"
        else
          "-device virtio-blk-pci,${deviceOpts}";
    in
    "-drive ${driveOpts} ${device}";

  drivesCmdLine = drives: concatStringsSep "\\\n    " (imap1 driveCmdline drives);

  # Shell script to start the VM.
  startVM = ''
    #! ${hostPkgs.runtimeShell}

    export PATH=${makeBinPath [ hostPkgs.coreutils ]}''${PATH:+:}$PATH

    set -e

    # Create an empty ext4 filesystem image. A filesystem image does not
    # contain a partition table but just a filesystem.
    createEmptyFilesystemImage() {
      local name=$1
      local size=$2
      local temp=$(mktemp)
      ${qemu}/bin/qemu-img create -f raw "$temp" "$size"
      ${hostPkgs.e2fsprogs}/bin/mkfs.ext4 -L ${rootFilesystemLabel} "$temp"
      ${qemu}/bin/qemu-img convert -f raw -O qcow2 "$temp" "$name"
      rm "$temp"
    }

    NIX_DISK_IMAGE=$(readlink -f "''${NIX_DISK_IMAGE:-${toString config.virtualisation.diskImage}}") || test -z "$NIX_DISK_IMAGE"

    if test -n "$NIX_DISK_IMAGE" && ! test -e "$NIX_DISK_IMAGE"; then
        echo "Disk image does not exist, creating the virtualisation disk image..."

        ${
          if (cfg.useBootLoader && cfg.useDefaultFilesystems) then
            ''
              # Create a writable qcow2 image using the systemImage as a backing
              # image.
              BACKING_SIZE_MB=$(( $(${lib.getExe' qemu "qemu-img"} info ${systemImage}/nixos.qcow2 --output=json | ${lib.getExe hostPkgs.jq} -r '."virtual-size"') / 1024 / 1024 ))
              DISK_SIZE_MB=${toString cfg.diskSize}
              if (( DISK_SIZE_MB < BACKING_SIZE_MB )); then
                OVERLAY_SIZE_MB=$BACKING_SIZE_MB
              else
                OVERLAY_SIZE_MB=$DISK_SIZE_MB
              fi

              ${qemu}/bin/qemu-img create \
                -f qcow2 \
                -b ${systemImage}/nixos.qcow2 \
                -F qcow2 \
                "$NIX_DISK_IMAGE" \
                 ''${OVERLAY_SIZE_MB}M
            ''
          else if cfg.useDefaultFilesystems then
            ''
              createEmptyFilesystemImage "$NIX_DISK_IMAGE" "${toString cfg.diskSize}M"
            ''
          else
            ''
              # Create an empty disk image without a filesystem.
              ${qemu}/bin/qemu-img create -f qcow2 "$NIX_DISK_IMAGE" "${toString cfg.diskSize}M"
            ''
        }
        echo "Virtualisation disk image created."
    fi

    # Create a directory for storing temporary data of the running VM.
    if [ -z "$TMPDIR" ] || [ -z "$USE_TMPDIR" ]; then
        TMPDIR=$(mktemp -d nix-vm.XXXXXXXXXX --tmpdir)
    fi

    ${lib.optionalString (cfg.useNixStoreImage) ''
      echo "Creating Nix store image..."

      ${hostPkgs.gnutar}/bin/tar --create \
        --absolute-names \
        --verbatim-files-from \
        --transform 'flags=rSh;s|/nix/store/||' \
        --transform 'flags=rSh;s|~nix~case~hack~[[:digit:]]\+||g' \
        --files-from ${
          hostPkgs.closureInfo {
            rootPaths = [
              config.system.build.toplevel
              regInfo
            ];
          }
        }/store-paths \
        | ${hostPkgs.erofs-utils}/bin/mkfs.erofs \
          --quiet \
          --force-uid=0 \
          --force-gid=0 \
          -L ${nixStoreFilesystemLabel} \
          -U eb176051-bd15-49b7-9e6b-462e0b467019 \
          -T 0 \
          --hard-dereference \
          --tar=f \
          "$TMPDIR"/store.img

      echo "Created Nix store image."
    ''}

    # Create a directory for exchanging data with the VM.
    mkdir -p "$TMPDIR/xchg"

    ${lib.optionalString cfg.useHostCerts ''
      mkdir -p "$TMPDIR/certs"
      if [ -e "$NIX_SSL_CERT_FILE" ]; then
        cp -L "$NIX_SSL_CERT_FILE" "$TMPDIR"/certs/ca-certificates.crt
      else
        echo \$NIX_SSL_CERT_FILE should point to a valid file if virtualisation.useHostCerts is enabled.
      fi
    ''}

    ${lib.optionalString cfg.useEFIBoot ''
      # Expose EFI variables, it's useful even when we are not using a bootloader (!).
      # We might be interested in having EFI variable storage present even if we aren't booting via UEFI, hence
      # no guard against `useBootLoader`.  Examples:
      # - testing PXE boot or other EFI applications
      # - directbooting LinuxBoot, which `kexec()s` into a UEFI environment that can boot e.g. Windows
      NIX_EFI_VARS=$(readlink -f "''${NIX_EFI_VARS:-${config.system.name}-efi-vars.fd}")
      # VM needs writable EFI vars
      if ! test -e "$NIX_EFI_VARS"; then
      ${
        if cfg.efi.keepVariables then
          # We still need the EFI var from the make-disk-image derivation
          # because our "switch-to-configuration" process might
          # write into it and we want to keep this data.
          ''cp ${systemImage}/efi-vars.fd "$NIX_EFI_VARS"''
        else
          ''cp ${cfg.efi.variables} "$NIX_EFI_VARS"''
      }
        chmod 0644 "$NIX_EFI_VARS"
      fi
    ''}

    ${lib.optionalString cfg.tpm.enable ''
      NIX_SWTPM_DIR=$(readlink -f "''${NIX_SWTPM_DIR:-${config.system.name}-swtpm}")
      mkdir -p "$NIX_SWTPM_DIR"
      ${lib.getExe cfg.tpm.package} \
        socket \
        --tpmstate dir="$NIX_SWTPM_DIR" \
        --server type=unixio,path="$NIX_SWTPM_DIR"/socket \
        --ctrl type=unixio,path="$NIX_SWTPM_DIR"/socket.ctrl \
        --pid file="$NIX_SWTPM_DIR"/pid --daemon \
        --flags not-need-init \
        --tpm2 \
        --log file="$NIX_SWTPM_DIR"/stdout,level=6

      (
        export TPM2TOOLS_TCTI=swtpm:path="$NIX_SWTPM_DIR"/socket
        export PATH=${
          lib.makeBinPath [
            pkgs.tpm2-tools
          ]
        }:$PATH

        tpm2_startup --clear
        tpm2_startup

        ${lib.optionalString (cfg.tpm.provisioning != null)
          # Run provisioning in a subshell not to pollute vars
          ''
            (
              export TCTI=swtpm:path="$NIX_SWTPM_DIR"/socket
              ${cfg.tpm.provisioning}
            )
          ''
        }

        tpm2_shutdown
        tpm2_shutdown --clear
      )

      # Enable `fdflags` builtin in Bash
      # We will need it to perform surgical modification of the file descriptor
      # passed in the coprocess to remove `FD_CLOEXEC`, i.e. close the file descriptor
      # on exec.
      # If let alone, it will trigger the coprocess to read EOF when QEMU is `exec`
      # at the end of this script. To work around that, we will just clear
      # the `FD_CLOEXEC` bits as a first step.
      enable -f ${hostPkgs.bash}/lib/bash/fdflags fdflags
      # leave a dangling subprocess because the swtpm ctrl socket has
      # "terminate" when the last connection disconnects, it stops swtpm.
      # When qemu stops, or if the main shell process ends, the coproc will
      # get signaled by virtue of the pipe between main and coproc ending.
      # Which in turns triggers a socat connect-disconnect to swtpm which
      # will stop it.
      coproc waitingswtpm {
        read || :
        ${cfg.tpm.package}/bin/swtpm_ioctl --unix "$NIX_SWTPM_DIR"/socket.ctrl --stop 2>/dev/null
      }
      # Clear `FD_CLOEXEC` on the coprocess' file descriptor stdin.
      fdflags -s-cloexec ''${waitingswtpm[1]}
    ''}

    cd "$TMPDIR"

    ${lib.pipe cfg.emptyDiskImages [
      (lib.imap0 (
        idx:
        { size, ... }:
        ''
          test -e "empty${builtins.toString idx}.qcow2" || ${qemu}/bin/qemu-img create -f qcow2 "empty${builtins.toString idx}.qcow2" "${builtins.toString size}M"
        ''
      ))
      (builtins.concatStringsSep "")
    ]}

    # Start QEMU.
    exec ${qemu-common.qemuBinary qemu} \
        -name ${config.system.name} \
        -m ${toString config.virtualisation.memorySize} \
        -smp ${toString config.virtualisation.cores} \
        -device virtio-rng-pci \
        ${concatStringsSep " " config.virtualisation.qemu.networkingOptions} \
        ${
          concatStringsSep " \\\n    " (
            mapAttrsToList (
              tag: share:
              "-virtfs local,path=${share.source},security_model=${share.securityModel},mount_tag=${tag}"
            ) config.virtualisation.sharedDirectories
          )
        } \
        ${drivesCmdLine config.virtualisation.qemu.drives} \
        ${concatStringsSep " \\\n    " config.virtualisation.qemu.options} \
        $QEMU_OPTS \
        "$@"
  '';

  regInfo = hostPkgs.closureInfo { rootPaths = config.virtualisation.additionalPaths; };

  # Use well-defined and persistent filesystem labels to identify block devices.
  rootFilesystemLabel = "nixos";
  espFilesystemLabel = "ESP"; # Hard-coded by make-disk-image.nix
  nixStoreFilesystemLabel = "nix-store";

  # The root drive is a raw disk which does not necessarily contain a
  # filesystem or partition table. It thus cannot be identified via the typical
  # persistent naming schemes (e.g. /dev/disk/by-{label, uuid, partlabel,
  # partuuid}. Instead, supply a well-defined and persistent serial attribute
  # via QEMU. Inside the running system, the disk can then be identified via
  # the /dev/disk/by-id scheme.
  rootDriveSerialAttr = "root";

  # System image is akin to a complete NixOS install with
  # a boot partition and root partition.
  systemImage = import ../../lib/make-disk-image.nix {
    inherit pkgs config lib;
    additionalPaths = [ regInfo ];
    format = "qcow2";
    onlyNixStore = false;
    label = rootFilesystemLabel;
    partitionTableType = selectPartitionTableLayout {
      inherit (cfg) useBIOSBoot useDefaultFilesystems useEFIBoot;
    };
    installBootLoader = cfg.installBootLoader;
    touchEFIVars = cfg.useEFIBoot;
    diskSize = "auto";
    additionalSpace = "0M";
    copyChannel = false;
    OVMF = cfg.efi.OVMF;
  };
in
{
  imports = [
    ../profiles/qemu-guest.nix
    ./disk-size-option.nix
    (mkRenamedOptionModule
      [
        "virtualisation"
        "pathsInNixDB"
      ]
      [
        "virtualisation"
        "additionalPaths"
      ]
    )
    (mkRemovedOptionModule
      [
        "virtualisation"
        "bootDevice"
      ]
      "This option was renamed to `virtualisation.rootDevice`, as it was incorrectly named and misleading. Take the time to review what you want to do and look at the new options like `virtualisation.{bootLoaderDevice, bootPartition}`, open an issue in case of issues."
    )
    (mkRemovedOptionModule
      [
        "virtualisation"
        "efiVars"
      ]
      "This option was removed, it is possible to provide a template UEFI variable with `virtualisation.efi.variables` ; if this option is important to you, open an issue"
    )
    (mkRemovedOptionModule
      [
        "virtualisation"
        "persistBootDevice"
      ]
      "Boot device is always persisted if you use a bootloader through the root disk image ; if this does not work for your usecase, please examine carefully what `virtualisation.{bootDevice, rootDevice, bootPartition}` options offer you and open an issue explaining your need.`"
    )
  ];

  options = {

    virtualisation.fileSystems = options.fileSystems;

    virtualisation.memorySize = mkOption {
      type = types.ints.positive;
      default = 1024;
      description = ''
        The memory size of the virtual machine in MiB (1024×1024 bytes).
      '';
    };

    virtualisation.msize = mkOption {
      type = types.ints.positive;
      default = 16384;
      description = ''
        The msize (maximum packet size) option passed to 9p file systems, in
        bytes. Increasing this should increase performance significantly,
        at the cost of higher RAM usage.
      '';
    };

    virtualisation.diskImage = mkOption {
      type = types.nullOr types.str;
      default = "./${config.system.name}.qcow2";
      defaultText = literalExpression ''"./''${config.system.name}.qcow2"'';
      description = ''
        Path to the disk image containing the root filesystem.
        The image will be created on startup if it does not
        exist.

        If null, a tmpfs will be used as the root filesystem and
        the VM's state will not be persistent.
      '';
    };

    virtualisation.bootLoaderDevice = mkOption {
      type = types.path;
      default = "/dev/disk/by-id/virtio-${rootDriveSerialAttr}";
      defaultText = literalExpression ''/dev/disk/by-id/virtio-${rootDriveSerialAttr}'';
      example = "/dev/disk/by-id/virtio-boot-loader-device";
      description = ''
        The path (inside th VM) to the device to boot from when legacy booting.
      '';
    };

    virtualisation.bootPartition = mkOption {
      type = types.nullOr types.path;
      default =
        if cfg.useEFIBoot then
          "/dev/disk/by-label/${espFilesystemLabel}"
        else if cfg.useBIOSBoot then
          "/dev/disk/by-label/BOOT"
        else
          null;
      defaultText = literalExpression ''
        if cfg.useEFIBoot then "/dev/disk/by-label/${espFilesystemLabel}"
        else if cfg.useBIOSBoot then "/dev/disk/by-label/BOOT"
        else null'';
      example = "/dev/disk/by-label/esp";
      description = ''
        The path (inside the VM) to the device containing the EFI System Partition (ESP).

        If you are *not* booting from a UEFI firmware, this value is, by
        default, `null`. The ESP is mounted to `boot.loader.efi.efiSysMountpoint`.
      '';
    };

    virtualisation.rootDevice = mkOption {
      type = types.nullOr types.path;
      default = "/dev/disk/by-label/${rootFilesystemLabel}";
      defaultText = literalExpression ''/dev/disk/by-label/${rootFilesystemLabel}'';
      example = "/dev/disk/by-label/nixos";
      description = ''
        The path (inside the VM) to the device containing the root filesystem.
      '';
    };

    virtualisation.emptyDiskImages = mkOption {
      type = types.listOf (
        types.coercedTo types.ints.positive (size: { inherit size; }) (
          types.submodule {
            options.size = mkOption {
              type = types.ints.positive;
              description = "The size of the disk in MiB";
            };
            options.driveConfig = mkOption {
              type = lib.types.submodule driveOptions;
              default = { };
              description = "Drive configuration to pass to {option}`virtualisation.qemu.drives`";
            };
          }
        )
      );
      default = [ ];
      description = ''
        Additional disk images to provide to the VM. The value is
        a list of size in MiB (1024×1024 bytes) of each disk. These disks are
        writeable by the VM.
      '';
    };

    virtualisation.graphics = mkOption {
      type = types.bool;
      default = true;
      description = ''
        Whether to run QEMU with a graphics window, or in nographic mode.
        Serial console will be enabled on both settings, but this will
        change the preferred console.
      '';
    };

    virtualisation.resolution = mkOption {
      type = options.services.xserver.resolutions.type.nestedTypes.elemType;
      default = {
        x = 1024;
        y = 768;
      };
      description = ''
        The resolution of the virtual machine display.
      '';
    };

    virtualisation.cores = mkOption {
      type = types.ints.positive;
      default = 1;
      description = ''
        Specify the number of cores the guest is permitted to use.
        The number can be higher than the available cores on the
        host system.
      '';
    };

    virtualisation.sharedDirectories = mkOption {
      type = types.attrsOf (
        types.submodule {
          options.source = mkOption {
            type = types.str;
            description = "The path of the directory to share, can be a shell variable";
          };
          options.target = mkOption {
            type = types.path;
            description = "The mount point of the directory inside the virtual machine";
          };
          options.securityModel = mkOption {
            type = types.enum [
              "passthrough"
              "mapped-xattr"
              "mapped-file"
              "none"
            ];
            default = "mapped-xattr";
            description = ''
              The security model to use for this share:

              - `passthrough`: files are stored using the same credentials as they are created on the guest (this requires QEMU to run as root)
              - `mapped-xattr`: some of the file attributes like uid, gid, mode bits and link target are stored as file attributes
              - `mapped-file`: the attributes are stored in the hidden .virtfs_metadata directory. Directories exported by this security model cannot interact with other unix tools
              - `none`: same as "passthrough" except the sever won't report failures if it fails to set file attributes like ownership
            '';
          };
        }
      );
      default = { };
      example = {
        my-share = {
          source = "/path/to/be/shared";
          target = "/mnt/shared";
        };
      };
      description = ''
        An attributes set of directories that will be shared with the
        virtual machine using VirtFS (9P filesystem over VirtIO).
        The attribute name will be used as the 9P mount tag.
      '';
    };

    virtualisation.additionalPaths = mkOption {
      type = types.listOf types.path;
      default = [ ];
      description = ''
        A list of paths whose closure should be made available to
        the VM.

        When 9p is used, the closure is registered in the Nix
        database in the VM. All other paths in the host Nix store
        appear in the guest Nix store as well, but are considered
        garbage (because they are not registered in the Nix
        database of the guest).

        When {option}`virtualisation.useNixStoreImage` is
        set, the closure is copied to the Nix store image.
      '';
    };

    virtualisation.forwardPorts = mkOption {
      type = types.listOf (
        types.submodule {
          options.from = mkOption {
            type = types.enum [
              "host"
              "guest"
            ];
            default = "host";
            description = ''
              Controls the direction in which the ports are mapped:

              - `"host"` means traffic from the host ports
                is forwarded to the given guest port.
              - `"guest"` means traffic from the guest ports
                is forwarded to the given host port.
            '';
          };
          options.proto = mkOption {
            type = types.enum [
              "tcp"
              "udp"
            ];
            default = "tcp";
            description = "The protocol to forward.";
          };
          options.host.address = mkOption {
            type = types.str;
            default = "";
            description = "The IPv4 address of the host.";
          };
          options.host.port = mkOption {
            type = types.port;
            description = "The host port to be mapped.";
          };
          options.guest.address = mkOption {
            type = types.str;
            default = "";
            description = "The IPv4 address on the guest VLAN.";
          };
          options.guest.port = mkOption {
            type = types.port;
            description = "The guest port to be mapped.";
          };
        }
      );
      default = [ ];
      example = lib.literalExpression ''
        [ # forward local port 2222 -> 22, to ssh into the VM
          { from = "host"; host.port = 2222; guest.port = 22; }

          # forward local port 80 -> 10.0.2.10:80 in the VLAN
          { from = "guest";
            guest.address = "10.0.2.10"; guest.port = 80;
            host.address = "127.0.0.1"; host.port = 80;
          }
        ]
      '';
      description = ''
        When using the SLiRP user networking (default), this option allows to
        forward ports to/from the host/guest.

        ::: {.warning}
        If the NixOS firewall on the virtual machine is enabled, you also
        have to open the guest ports to enable the traffic between host and
        guest.
        :::

        ::: {.note}
        Currently QEMU supports only IPv4 forwarding.
        :::
      '';
    };

    virtualisation.restrictNetwork = mkOption {
      type = types.bool;
      default = false;
      example = true;
      description = ''
        If this option is enabled, the guest will be isolated, i.e. it will
        not be able to contact the host and no guest IP packets will be
        routed over the host to the outside. This option does not affect
        any explicitly set forwarding rules.
      '';
    };

    virtualisation.vlans = mkOption {
      type = types.listOf types.ints.unsigned;
      default = if config.virtualisation.interfaces == { } then [ 1 ] else [ ];
      defaultText = lib.literalExpression ''if config.virtualisation.interfaces == {} then [ 1 ] else [ ]'';
      example = [
        1
        2
      ];
      description = ''
        Virtual networks to which the VM is connected.  Each
        number «N» in this list causes
        the VM to have a virtual Ethernet interface attached to a
        separate virtual network on which it will be assigned IP
        address
        `192.168.«N».«M»`,
        where «M» is the index of this VM
        in the list of VMs.
      '';
    };

    virtualisation.interfaces = mkOption {
      default = { };
      example = {
        enp1s0.vlan = 1;
      };
      description = ''
        Network interfaces to add to the VM.
      '';
      type =
        with types;
        attrsOf (submodule {
          options = {
            vlan = mkOption {
              type = types.ints.unsigned;
              description = ''
                VLAN to which the network interface is connected.
              '';
            };

            assignIP = mkOption {
              type = types.bool;
              default = false;
              description = ''
                Automatically assign an IP address to the network interface using the same scheme as
                virtualisation.vlans.
              '';
            };
          };
        });
    };

    virtualisation.writableStore = mkOption {
      type = types.bool;
      default = cfg.mountHostNixStore;
      defaultText = literalExpression "cfg.mountHostNixStore";
      description = ''
        If enabled, the Nix store in the VM is made writable by
        layering an overlay filesystem on top of the host's Nix
        store.

        By default, this is enabled if you mount a host Nix store.
      '';
    };

    virtualisation.writableStoreUseTmpfs = mkOption {
      type = types.bool;
      default = true;
      description = ''
        Use a tmpfs for the writable store instead of writing to the VM's
        own filesystem.
      '';
    };

    networking.primaryIPAddress = mkOption {
      type = types.str;
      default = "";
      internal = true;
      description = "Primary IP address used in /etc/hosts.";
    };

    networking.primaryIPv6Address = mkOption {
      type = types.str;
      default = "";
      internal = true;
      description = "Primary IPv6 address used in /etc/hosts.";
    };

    virtualisation.host.pkgs = mkOption {
      type = options.nixpkgs.pkgs.type;
      default = pkgs;
      defaultText = literalExpression "pkgs";
      example = literalExpression ''
        import pkgs.path { system = "x86_64-darwin"; }
      '';
      description = ''
        Package set to use for the host-specific packages of the VM runner.
        Changing this to e.g. a Darwin package set allows running NixOS VMs on Darwin.
      '';
    };

    virtualisation.qemu = {
      package = mkOption {
        type = types.package;
        default =
          if hostPkgs.stdenv.hostPlatform.qemuArch == pkgs.stdenv.hostPlatform.qemuArch then
            hostPkgs.qemu_kvm
          else
            hostPkgs.qemu;
        defaultText = literalExpression "if hostPkgs.stdenv.hostPlatform.qemuArch == pkgs.stdenv.hostPlatform.qemuArch then config.virtualisation.host.pkgs.qemu_kvm else config.virtualisation.host.pkgs.qemu";
        example = literalExpression "pkgs.qemu_test";
        description = "QEMU package to use.";
      };

      options = mkOption {
        type = types.listOf types.str;
        default = [ ];
        example = [ "-vga std" ];
        description = ''
          Options passed to QEMU.
          See [QEMU User Documentation](https://www.qemu.org/docs/master/system/qemu-manpage) for a complete list.
        '';
      };

      consoles = mkOption {
        type = types.listOf types.str;
        default =
          let
            consoles = [
              "${qemu-common.qemuSerialDevice},115200n8"
              "tty0"
            ];
          in
          if cfg.graphics then consoles else reverseList consoles;
        example = [ "console=tty1" ];
        description = ''
          The output console devices to pass to the kernel command line via the
          `console` parameter, the primary console is the last
          item of this list.

          By default it enables both serial console and
          `tty0`. The preferred console (last one) is based on
          the value of {option}`virtualisation.graphics`.
        '';
      };

      networkingOptions = mkOption {
        type = types.listOf types.str;
        default = [ ];
        example = [
          "-net nic,netdev=user.0,model=virtio"
          "-netdev user,id=user.0,\${QEMU_NET_OPTS:+,$QEMU_NET_OPTS}"
        ];
        description = ''
          Networking-related command-line options that should be passed to qemu.
          The default is to use userspace networking (SLiRP).
          See the [QEMU Wiki on Networking](https://wiki.qemu.org/Documentation/Networking) for details.

          If you override this option, be advised to keep
          `''${QEMU_NET_OPTS:+,$QEMU_NET_OPTS}` (as seen in the example)
          to keep the default runtime behaviour.
        '';
      };

      drives = mkOption {
        type = types.listOf (
          types.submodule {
            imports = [ driveOptions ];

            options = {
              file = mkOption {
                type = types.str;
                description = "The file image used for this drive.";
              };
            };
          }
        );
        description = "Drives passed to qemu.";
      };

      diskInterface = mkOption {
        type = types.enum [
          "virtio"
          "scsi"
          "ide"
        ];
        default = "virtio";
        example = "scsi";
        description = "The interface used for the virtual hard disks.";
      };

      guestAgent.enable = mkOption {
        type = types.bool;
        default = true;
        description = ''
          Enable the Qemu guest agent.
        '';
      };

      virtioKeyboard = mkOption {
        type = types.bool;
        default = true;
        description = ''
          Enable the virtio-keyboard device.
        '';
      };
    };

    virtualisation.useNixStoreImage = mkOption {
      type = types.bool;
      default = false;
      description = ''
        Build and use a disk image for the Nix store, instead of
        accessing the host's one through 9p.

        For applications which do a lot of reads from the store,
        this can drastically improve performance, but at the cost of
        disk space and image build time.

        The Nix store image is built just-in-time right before the VM is
        started. Because it does not produce another derivation, the image is
        not cached between invocations and never lands in the store or binary
        cache.

        If you want a full disk image with a partition table and a root
        filesystem instead of only a store image, enable
        {option}`virtualisation.useBootLoader` instead.
      '';
    };

    virtualisation.mountHostNixStore = mkOption {
      type = types.bool;
      default = !cfg.useNixStoreImage && !cfg.useBootLoader;
      defaultText = literalExpression "!cfg.useNixStoreImage && !cfg.useBootLoader";
      description = ''
        Mount the host Nix store as a 9p mount.
      '';
    };

    virtualisation.directBoot = {
      enable = mkOption {
        type = types.bool;
        default = !cfg.useBootLoader;
        defaultText = "!cfg.useBootLoader";
        description = ''
          If enabled, the virtual machine will boot directly into the kernel instead of through a bootloader.
          Read more about this feature in the [QEMU documentation on Direct Linux Boot](https://qemu-project.gitlab.io/qemu/system/linuxboot.html)

          This is enabled by default.
          If you want to test netboot, consider disabling this option.
          Enable a bootloader with {option}`virtualisation.useBootLoader` if you need.

          Relevant parameters such as those set in `boot.initrd` and `boot.kernelParams` are also passed to QEMU.
          Additional parameters can be supplied on invocation through the environment variable `$QEMU_KERNEL_PARAMS`.
          They are added to the `-append` option, see [QEMU User Documentation](https://www.qemu.org/docs/master/system/qemu-manpage) for details
          For example, to let QEMU use the parent terminal as the serial console, set `QEMU_KERNEL_PARAMS="console=ttyS0"`.

          This will not (re-)boot correctly into a system that has switched to a different configuration on disk.
        '';
      };
      initrd = mkOption {
        type = types.str;
        default = "${config.system.build.initialRamdisk}/${config.system.boot.loader.initrdFile}";
        defaultText = "\${config.system.build.initialRamdisk}/\${config.system.boot.loader.initrdFile}";
        description = ''
          In direct boot situations, you may want to influence the initrd to load
          to use your own customized payload.

          This is useful if you want to test the netboot image without
          testing the firmware or the loading part.
        '';
      };
    };

    virtualisation.useBootLoader = mkOption {
      type = types.bool;
      default = false;
      description = ''
        Use a boot loader to boot the system.
        This allows, among other things, testing the boot loader.

        If disabled, the kernel and initrd are directly booted,
        forgoing any bootloader.

        Check the documentation on {option}`virtualisation.directBoot.enable` for details.
      '';
    };

    virtualisation.installBootLoader = mkOption {
      type = types.bool;
      default = cfg.useBootLoader && cfg.useDefaultFilesystems;
      defaultText = "cfg.useBootLoader && cfg.useDefaultFilesystems";
      description = ''
        Install boot loader to target image.

        This is best-effort and may break with unconventional partition setups.
        Use `virtualisation.useDefaultFilesystems` for a known-working configuration.
      '';
    };

    virtualisation.useBIOSBoot = mkEnableOption null // {
      description = ''
        If enabled for legacy MBR VMs, the VM image will have a separate boot
        partition mounted at /boot.
        useBIOSBoot is ignored if useEFIBoot == true.
      '';
    };

    virtualisation.useEFIBoot = mkOption {
      type = types.bool;
      default = false;
      description = ''
        If enabled, the virtual machine will provide a EFI boot
        manager.
        useEFIBoot is ignored if useBootLoader == false.
      '';
    };

    virtualisation.efi = {
      OVMF = mkOption {
        type = types.package;
        default =
          (pkgs.OVMF.override {
            secureBoot = cfg.useSecureBoot;
          }).fd;
        defaultText = ''
          (pkgs.OVMF.override {
                    secureBoot = cfg.useSecureBoot;
                  }).fd'';
        description = "OVMF firmware package, defaults to OVMF configured with secure boot if needed.";
      };

      firmware = mkOption {
        type = types.path;
        default = cfg.efi.OVMF.firmware;
        defaultText = literalExpression "cfg.efi.OVMF.firmware";
        description = ''
          Firmware binary for EFI implementation, defaults to OVMF.
        '';
      };

      variables = mkOption {
        type = types.path;
        default = cfg.efi.OVMF.variables;
        defaultText = literalExpression "cfg.efi.OVMF.variables";
        description = ''
          Platform-specific flash binary for EFI variables, implementation-dependent to the EFI firmware.
          Defaults to OVMF.
        '';
      };

      keepVariables = mkOption {
        type = types.bool;
        default = cfg.useBootLoader;
        defaultText = literalExpression "cfg.useBootLoader";
        description = "Whether to keep EFI variable values from the generated system image";
      };
    };

    virtualisation.tpm = {
      enable = mkEnableOption "a TPM device in the virtual machine with a driver, using swtpm";

      package = mkPackageOption cfg.host.pkgs "swtpm" { };

      deviceModel = mkOption {
        type = types.str;
        default = (
          {
            "i686-linux" = "tpm-tis";
            "x86_64-linux" = "tpm-tis";
            "ppc64-linux" = "tpm-spapr";
            "armv7-linux" = "tpm-tis-device";
            "aarch64-linux" = "tpm-tis-device";
          }
          .${pkgs.stdenv.hostPlatform.system} or (throw "Unsupported system for TPM2 emulation in QEMU")
        );
        defaultText = ''
          Based on the guest platform Linux system:

          - `tpm-tis` for (i686, x86_64)
          - `tpm-spapr` for ppc64
          - `tpm-tis-device` for (armv7, aarch64)
        '';
        example = "tpm-tis-device";
        description = "QEMU device model for the TPM, uses the appropriate default based on th guest platform system and the package passed.";
      };

      provisioning = mkOption {
        type = types.nullOr types.str;
        default = null;
        description = ''
          Script to provision the TPM before control is handed off to the VM.

          `TPM2TOOLS_TCTI` will be provided to configure tpm2-tools to use the
          swtpm instance transparently.
          `TCTI` is also provided as a generic value, consumer is expected to
          re-export it however it may need (`TPM2OPENSSL_TCTI`, `TPM2_PKCS11_TCTI`,
          ...).
        '';
        example = literalExpression ''
          tpm2_nvdefine 0xcafecafe \
            -C o \
            -a "ownerread|policyread|policywrite|ownerwrite|authread|authwrite"
          echo "foobar" | tpm2_nvwrite 0xcafecafe -C o
        '';
      };
    };

    virtualisation.useDefaultFilesystems = mkOption {
      type = types.bool;
      default = true;
      description = ''
        If enabled, the boot disk of the virtual machine will be
        formatted and mounted with the default filesystems for
        testing. Swap devices and LUKS will be disabled.

        If disabled, a root filesystem has to be specified and
        formatted (for example in the initial ramdisk).
      '';
    };

    virtualisation.useSecureBoot = mkOption {
      type = types.bool;
      default = false;
      description = ''
        Enable Secure Boot support in the EFI firmware.
      '';
    };

    virtualisation.bios = mkOption {
      type = types.nullOr types.package;
      default = null;
      description = ''
        An alternate BIOS (such as `qboot`) with which to start the VM.
        Should contain a file named `bios.bin`.
        If `null`, QEMU's builtin SeaBIOS will be used.
      '';
    };

    virtualisation.useHostCerts = mkOption {
      type = types.bool;
      default = false;
      description = ''
        If enabled, when `NIX_SSL_CERT_FILE` is set on the host,
        pass the CA certificates from the host to the VM.
      '';
    };

  };

  config = {

    assertions =
      lib.concatLists (
        lib.flip lib.imap cfg.forwardPorts (
          i: rule: [
            {
              assertion = rule.from == "guest" -> rule.proto == "tcp";
              message = ''
                Invalid virtualisation.forwardPorts.<entry ${toString i}>.proto:
                  Guest forwarding supports only TCP connections.
              '';
            }
            {
              assertion = rule.from == "guest" -> lib.hasPrefix "10.0.2." rule.guest.address;
              message = ''
                Invalid virtualisation.forwardPorts.<entry ${toString i}>.guest.address:
                  The address must be in the default VLAN (10.0.2.0/24).
              '';
            }
          ]
        )
      )
      ++ [
        {
          assertion = pkgs.stdenv.hostPlatform.is32bit -> cfg.memorySize < 2047;
          message = ''
            virtualisation.memorySize is above 2047, but qemu is only able to allocate 2047 MiB RAM on 32bit max.
          '';
        }
        {
          assertion =
            cfg.directBoot.enable || cfg.directBoot.initrd == options.virtualisation.directBoot.initrd.default;
          message = ''
            You changed the default of `virtualisation.directBoot.initrd` but you are not
            using QEMU direct boot. This initrd will not be used in your current
            boot configuration.

            Either do not mutate `virtualisation.directBoot.initrd` or enable direct boot.

            If you have a more advanced usecase, please open an issue or a pull request.
          '';
        }
        {
          assertion = cfg.installBootLoader -> config.system.switch.enable;
          message = ''
            `system.switch.enable` must be enabled for `virtualisation.installBootLoader` to work.
            Please enable it in your configuration.
          '';
        }
      ];

    warnings = optional (cfg.directBoot.enable && cfg.useBootLoader) ''
      You enabled direct boot and a bootloader, QEMU will not boot your bootloader, rendering
      `useBootLoader` useless. You might want to disable one of those options.
    '';

    # Install Limine on the bootloader device
    boot.loader.limine.biosDevice = cfg.bootLoaderDevice;

    # In UEFI boot, we use a EFI-only partition table layout, thus GRUB will fail when trying to install
    # legacy and UEFI. In order to avoid this, we have to put "nodev" to force UEFI-only installs.
    # Otherwise, we set the proper bootloader device for this.
    # FIXME: make a sense of this mess wrt to multiple ESP present in the system, probably use boot.efiSysMountpoint?
    boot.loader.grub.device = mkVMOverride (if cfg.useEFIBoot then "nodev" else cfg.bootLoaderDevice);
    boot.loader.grub.gfxmodeBios = with cfg.resolution; "${toString x}x${toString y}";

    boot.loader.supportsInitrdSecrets = mkIf (!cfg.useBootLoader) (mkVMOverride false);

    # After booting, register the closure of the paths in
    # `virtualisation.additionalPaths' in the Nix database in the VM.  This
    # allows Nix operations to work in the VM.  The path to the
    # registration file is passed through the kernel command line to
    # allow `system.build.toplevel' to be included.  (If we had a direct
    # reference to ${regInfo} here, then we would get a cyclic
    # dependency.)
    boot.postBootCommands = lib.mkIf config.nix.enable ''
      if [[ "$(cat /proc/cmdline)" =~ regInfo=([^ ]*) ]]; then
        ${config.nix.package.out}/bin/nix-store --load-db < ''${BASH_REMATCH[1]}
      fi
    '';

    boot.initrd.availableKernelModules =
      optional (cfg.qemu.diskInterface == "scsi") "sym53c8xx" ++ optional (cfg.tpm.enable) "tpm_tis";

    virtualisation.additionalPaths = [ config.system.build.toplevel ];

    virtualisation.sharedDirectories = {
      nix-store = mkIf cfg.mountHostNixStore {
        source = builtins.storeDir;
        # Always mount this to /nix/.ro-store because we never want to actually
        # write to the host Nix Store.
        target = "/nix/.ro-store";
        securityModel = "none";
      };
      xchg = {
        source = ''"$TMPDIR"/xchg'';
        securityModel = "none";
        target = "/tmp/xchg";
      };
      shared = {
        source = ''"''${SHARED_DIR:-$TMPDIR/xchg}"'';
        target = "/tmp/shared";
        securityModel = "none";
      };
      certs = mkIf cfg.useHostCerts {
        source = ''"$TMPDIR"/certs'';
        target = "/etc/ssl/certs";
        securityModel = "none";
      };
    };

    security.pki.installCACerts = mkIf cfg.useHostCerts false;

    virtualisation.qemu.networkingOptions =
      let
        forwardingOptions = flip concatMapStrings cfg.forwardPorts (
          {
            proto,
            from,
            host,
            guest,
          }:
          if from == "host" then
            "hostfwd=${proto}:${host.address}:${toString host.port}-"
            + "${guest.address}:${toString guest.port},"
          else
            "'guestfwd=${proto}:${guest.address}:${toString guest.port}-"
            + "cmd:${pkgs.netcat}/bin/nc ${host.address} ${toString host.port}',"
        );
        restrictNetworkOption = lib.optionalString cfg.restrictNetwork "restrict=on,";
      in
      [
        "-net nic,netdev=user.0,model=virtio"
        "-netdev user,id=user.0,${forwardingOptions}${restrictNetworkOption}\"$QEMU_NET_OPTS\""
      ];

    virtualisation.qemu.options = mkMerge [
      (mkIf cfg.qemu.virtioKeyboard [
        "-device virtio-keyboard"
      ])
      (mkIf pkgs.stdenv.hostPlatform.isx86 [
        "-usb"
        "-device usb-tablet,bus=usb-bus.0"
      ])
      (mkIf pkgs.stdenv.hostPlatform.isAarch [
        "-device virtio-gpu-pci"
        "-device usb-ehci,id=usb0"
        "-device usb-kbd"
        "-device usb-tablet"
      ])
      (
        let
          alphaNumericChars = lowerChars ++ upperChars ++ (map toString (range 0 9));
          # Replace all non-alphanumeric characters with underscores
          sanitizeShellIdent =
            s:
            concatMapStrings (c: if builtins.elem c alphaNumericChars then c else "_") (stringToCharacters s);
        in
        mkIf cfg.directBoot.enable [
          "-kernel \${NIXPKGS_QEMU_KERNEL_${sanitizeShellIdent config.system.name}:-${config.system.build.toplevel}/kernel}"
          "-initrd ${cfg.directBoot.initrd}"
          ''-append "$(cat ${config.system.build.toplevel}/kernel-params) init=${config.system.build.toplevel}/init regInfo=${regInfo}/registration ${consoles} $QEMU_KERNEL_PARAMS"''
        ]
      )
      (mkIf cfg.useEFIBoot [
        "-drive if=pflash,format=raw,unit=0,readonly=on,file=${cfg.efi.firmware}"
        "-drive if=pflash,format=raw,unit=1,readonly=off,file=$NIX_EFI_VARS"
      ])
      (mkIf (cfg.bios != null) [
        "-bios ${cfg.bios}/bios.bin"
      ])
      (mkIf (!cfg.graphics) [
        "-nographic"
      ])
      (mkIf (cfg.tpm.enable) [
        "-chardev socket,id=chrtpm,path=\"$NIX_SWTPM_DIR\"/socket.ctrl"
        "-tpmdev emulator,id=tpm_dev_0,chardev=chrtpm"
        "-device ${cfg.tpm.deviceModel},tpmdev=tpm_dev_0"
      ])
      (mkIf (pkgs.stdenv.hostPlatform.isx86 && cfg.efi.OVMF.systemManagementModeRequired) [
        "-machine"
        "q35,smm=on"
        "-global"
        "driver=cfi.pflash01,property=secure,value=on"
      ])
    ];

    virtualisation.qemu.drives = mkMerge [
      (mkIf (cfg.diskImage != null) [
        {
          name = "root";
          file = ''"$NIX_DISK_IMAGE"'';
          driveExtraOpts.cache = "writeback";
          driveExtraOpts.werror = "report";
          deviceExtraOpts.bootindex = "1";
          deviceExtraOpts.serial = rootDriveSerialAttr;
        }
      ])
      (mkIf cfg.useNixStoreImage [
        {
          name = "nix-store";
          file = ''"$TMPDIR"/store.img'';
          deviceExtraOpts.bootindex = "2";
          driveExtraOpts.format = "raw";
        }
      ])
      (imap0 (
        idx: imgCfg:
        lib.mkMerge [
          {
            file = "$(pwd)/empty${toString idx}.qcow2";
            driveExtraOpts.werror = "report";
          }
          imgCfg.driveConfig
        ]
      ) cfg.emptyDiskImages)
    ];

    # By default, use mkVMOverride to enable building test VMs (e.g. via
    # `nixos-rebuild build-vm`) of a system configuration, where the regular
    # value for the `fileSystems' attribute should be disregarded (since those
    # filesystems don't necessarily exist in the VM). You can disable this
    # override by setting `virtualisation.fileSystems = lib.mkForce { };`.
    fileSystems = lib.mkIf (cfg.fileSystems != { }) (mkVMOverride cfg.fileSystems);

    virtualisation.diskSizeAutoSupported = false;

    virtualisation.fileSystems =
      let
        mkSharedDir = tag: share: {
          name = share.target;
          value.device = tag;
          value.fsType = "9p";
          value.neededForBoot = true;
          value.options = [
            "trans=virtio"
            "version=9p2000.L"
            "msize=${toString cfg.msize}"
            "x-systemd.requires=modprobe@9pnet_virtio.service"
          ]
          ++ lib.optional (tag == "nix-store") "cache=loose";
        };
      in
      lib.mkMerge [
        (lib.mapAttrs' mkSharedDir cfg.sharedDirectories)
        {
          "/" = lib.mkIf cfg.useDefaultFilesystems (
            if cfg.diskImage == null then
              {
                device = "tmpfs";
                fsType = "tmpfs";
                options = [ "mode=755" ];
              }
            else
              {
                device = cfg.rootDevice;
                fsType = "ext4";
              }
          );
          "/tmp" = lib.mkIf config.boot.tmp.useTmpfs {
            device = "tmpfs";
            fsType = "tmpfs";
            neededForBoot = true;
            # Sync with systemd's tmp.mount;
            options = [
              "mode=1777"
              "strictatime"
              "nosuid"
              "nodev"
              "size=${toString config.boot.tmp.tmpfsSize}"
            ];
          };
          "/nix/store" = lib.mkIf (cfg.useNixStoreImage || cfg.mountHostNixStore) (
            if cfg.writableStore then
              {
                overlay = {
                  lowerdir = [ "/nix/.ro-store" ];
                  upperdir = "/nix/.rw-store/upper";
                  workdir = "/nix/.rw-store/work";
                };
              }
            else
              {
                device = "/nix/.ro-store";
                options = [ "bind" ];
              }
          );
          "/nix/.ro-store" = lib.mkIf cfg.useNixStoreImage {
            device = "/dev/disk/by-label/${nixStoreFilesystemLabel}";
            fsType = "erofs";
            neededForBoot = true;
            options = [ "ro" ];
          };
          "/nix/.rw-store" = lib.mkIf (cfg.writableStore && cfg.writableStoreUseTmpfs) {
            fsType = "tmpfs";
            options = [ "mode=0755" ];
            neededForBoot = true;
          };
          "${config.boot.loader.efi.efiSysMountPoint}" =
            lib.mkIf (cfg.useBootLoader && cfg.bootPartition != null)
              {
                device = cfg.bootPartition;
                fsType = "vfat";
              };
        }
      ];

    swapDevices = (if cfg.useDefaultFilesystems then mkVMOverride else mkDefault) [ ];
    boot.initrd.luks.devices = (if cfg.useDefaultFilesystems then mkVMOverride else mkDefault) { };

    # Don't run ntpd in the guest.  It should get the correct time from KVM.
    services.timesyncd.enable = false;

    services.qemuGuest.enable = cfg.qemu.guestAgent.enable;

    system.build.vm =
      hostPkgs.runCommand "nixos-vm"
        {
          preferLocalBuild = true;
          meta.mainProgram = "run-${config.system.name}-vm";
        }
        ''
          mkdir -p $out/bin
          ln -s ${config.system.build.toplevel} $out/system
          ln -s ${hostPkgs.writeScript "run-nixos-vm" startVM} $out/bin/run-${config.system.name}-vm
        '';

    # When building a regular system configuration, override whatever
    # video driver the host uses.
    services.xserver.videoDrivers = mkVMOverride [ "modesetting" ];
    services.xserver.defaultDepth = mkVMOverride 0;
    services.xserver.resolutions = mkVMOverride [ cfg.resolution ];
    services.xserver.monitorSection = ''
      # Set a higher refresh rate so that resolutions > 800x600 work.
      HorizSync 30-140
      VertRefresh 50-160
    '';

    # Wireless won't work in the VM.
    networking.wireless.enable = mkVMOverride false;
    services.connman.enable = mkVMOverride false;

    # Speed up booting by not waiting for ARP.
    networking.dhcpcd.extraConfig = "noarp";

    networking.usePredictableInterfaceNames = false;

    system.requiredKernelConfig =
      with config.lib.kernelConfig;
      [
        (isEnabled "VIRTIO_BLK")
        (isEnabled "VIRTIO_PCI")
        (isEnabled "VIRTIO_NET")
        (isEnabled "EXT4_FS")
        (isEnabled "NET_9P_VIRTIO")
        (isEnabled "9P_FS")
        (isYes "BLK_DEV")
        (isYes "PCI")
        (isYes "NETDEVICES")
        (isYes "NET_CORE")
        (isYes "INET")
        (isYes "NETWORK_FILESYSTEMS")
      ]
      ++ optionals (!cfg.graphics) [
        (isYes "SERIAL_8250_CONSOLE")
        (isYes "SERIAL_8250")
      ]
      ++ optionals (cfg.writableStore) [
        (isEnabled "OVERLAY_FS")
      ];

  };

  # uses types of services/x11/xserver.nix
  meta.buildDocsInSandbox = false;
}
