nixpkgs/nixos/tests/systemd-repart.nix

{
  system ? builtins.currentSystem,
  config ? { },
  pkgs ? import ../.. { inherit system config; },
}:

with import ../lib/testing-python.nix { inherit system pkgs; };
with pkgs.lib;

let
  # A testScript fragment that prepares a disk with some empty, unpartitioned
  # space. and uses it to boot the test with.
  # Takes two arguments, `machine` from which the diskImage is extracted,
  # as well an optional `sizeDiff` (defaulting to +32M), describing how should
  # be resized.
  useDiskImage =
    {
      machine,
      sizeDiff ? "+32M",
    }:
    ''
      import os
      import shutil
      import subprocess
      import tempfile

      tmp_disk_image = tempfile.NamedTemporaryFile()

      shutil.copyfile("${machine.system.build.diskImage}/nixos.img", tmp_disk_image.name)

      subprocess.run([
        "${machine.virtualisation.qemu.package}/bin/qemu-img",
        "resize",
        "-f",
        "raw",
        tmp_disk_image.name,
        "${sizeDiff}",
      ])

      # Set NIX_DISK_IMAGE so that the qemu script finds the right disk image.
      os.environ['NIX_DISK_IMAGE'] = tmp_disk_image.name
    '';

  common =
    {
      config,
      pkgs,
      lib,
      ...
    }:
    {
      virtualisation.useDefaultFilesystems = false;
      virtualisation.fileSystems = {
        "/" = {
          device = "/dev/vda2";
          fsType = "ext4";
        };
      };

      # systemd-repart operates on disks with a partition table. The qemu module,
      # however, creates separate filesystem images without a partition table, so
      # we have to create a disk image manually.
      #
      # This creates two partitions, an ESP available as /dev/vda1 and the root
      # partition available as /dev/vda2.
      system.build.diskImage = import ../lib/make-disk-image.nix {
        inherit config pkgs lib;
        # Use a raw format disk so that it can be resized before starting the
        # test VM.
        format = "raw";
        # Keep the image as small as possible but leave some room for changes.
        bootSize = "32M";
        additionalSpace = "0M";
        # GPT with an EFI System Partition is the typical use case for
        # systemd-repart because it does not support MBR.
        partitionTableType = "efi";
        # We do not actually care much about the content of the partitions, so we
        # do not need a bootloader installed.
        installBootLoader = false;
        # Improve determinism by not copying a channel.
        copyChannel = false;
      };
    };
in
{
  basic = makeTest {
    name = "systemd-repart";
    meta.maintainers = with maintainers; [ nikstur ];

    nodes.machine =
      { config, pkgs, ... }:
      {
        imports = [ common ];

        boot.initrd.systemd.enable = true;

        boot.initrd.systemd.repart.enable = true;
        systemd.repart.partitions = {
          "10-root" = {
            Type = "linux-generic";
          };
        };
      };

    testScript =
      { nodes, ... }:
      ''
        ${useDiskImage { inherit (nodes) machine; }}

        machine.start()
        machine.wait_for_unit("multi-user.target")

        machine.succeed("journalctl --boot --grep 'Growing existing partition 1.' --identifier systemd-repart")
      '';
  };

  encrypt-tpm2 = makeTest {
    name = "systemd-repart-encrypt-tpm2";
    meta.maintainers = with maintainers; [ flokli ];

    nodes.machine =
      {
        config,
        pkgs,
        lib,
        ...
      }:
      {
        imports = [ common ];

        boot.initrd.systemd.enable = true;

        boot.initrd.availableKernelModules = [ "dm_crypt" ];
        boot.initrd.luks.devices = lib.mkVMOverride {
          created-crypt = {
            device = "/dev/disk/by-partlabel/created-crypt";
            crypttabExtraOpts = [ "tpm2-device=auto" ];
          };
        };
        boot.initrd.systemd.repart.enable = true;
        boot.initrd.systemd.repart.extraArgs = [
          "--tpm2-pcrs=7"
        ];
        systemd.repart.partitions = {
          "10-root" = {
            Type = "linux-generic";
          };
          "10-crypt" = {
            Type = "var";
            Label = "created-crypt";
            Format = "ext4";
            Encrypt = "tpm2";
          };
        };
        virtualisation.tpm.enable = true;
        virtualisation.fileSystems = {
          "/var" = {
            device = "/dev/mapper/created-crypt";
            fsType = "ext4";
          };
        };
      };

    testScript =
      { nodes, ... }:
      ''
        ${useDiskImage {
          inherit (nodes) machine;
          sizeDiff = "+100M";
        }}

        machine.start()
        machine.wait_for_unit("multi-user.target")

        machine.succeed("journalctl --boot --grep 'Encrypting future partition 2' --identifier systemd-repart")

        assert "/dev/mapper/created-crypt" in machine.succeed("mount")
      '';
  };

  after-initrd = makeTest {
    name = "systemd-repart-after-initrd";
    meta.maintainers = with maintainers; [ nikstur ];

    nodes.machine =
      { config, pkgs, ... }:
      {
        imports = [ common ];

        systemd.repart.enable = true;
        systemd.repart.partitions = {
          "10-root" = {
            Type = "linux-generic";
          };
        };
      };

    testScript =
      { nodes, ... }:
      ''
        ${useDiskImage { inherit (nodes) machine; }}

        machine.start()
        machine.wait_for_unit("multi-user.target")

        machine.succeed("journalctl --grep 'Growing existing partition 1.' --identifier systemd-repart")
      '';
  };

  create-root = makeTest {
    name = "systemd-repart-create-root";
    meta.maintainers = with maintainers; [ nikstur ];

    nodes.machine =
      {
        config,
        lib,
        pkgs,
        ...
      }:
      {
        virtualisation.useDefaultFilesystems = false;
        virtualisation.mountHostNixStore = false;
        virtualisation.fileSystems = {
          "/" = {
            device = "/dev/disk/by-partlabel/created-root";
            fsType = "ext4";
          };
          "/nix/store" = {
            device = "/dev/vda2";
            fsType = "ext4";
          };
        };

        # Create an image containing only the Nix store. This enables creating
        # the root partition with systemd-repart and then successfully booting
        # into a working system.
        #
        # This creates two partitions, an ESP available as /dev/vda1 and the Nix
        # store available as /dev/vda2.
        system.build.diskImage = import ../lib/make-disk-image.nix {
          inherit config pkgs lib;
          onlyNixStore = true;
          format = "raw";
          bootSize = "32M";
          additionalSpace = "0M";
          partitionTableType = "efi";
          installBootLoader = false;
          copyChannel = false;
        };

        boot.initrd.systemd.enable = true;

        boot.initrd.systemd.repart.enable = true;
        boot.initrd.systemd.repart.device = "/dev/vda";
        systemd.repart.partitions = {
          "10-root" = {
            Type = "root";
            Label = "created-root";
            Format = "ext4";
          };
        };
      };

    testScript =
      { nodes, ... }:
      ''
        ${useDiskImage { inherit (nodes) machine; }}

        machine.start()
        machine.wait_for_unit("multi-user.target")

        machine.succeed("journalctl --boot --grep 'Adding new partition 2 to partition table.' --identifier systemd-repart")
      '';
  };

  factory-reset = makeTest {
    name = "systemd-repart-factory-reset";
    meta.maintainers = with maintainers; [ willibutz ];

    nodes.machine =
      { pkgs, lib, ... }:
      {
        imports = [ common ];

        virtualisation = {
          useEFIBoot = true;
          tpm.enable = true;
          efi.OVMF = pkgs.OVMFFull;
          useDefaultFilesystems = false;
          fileSystems = {
            "/state" = {
              device = "/dev/mapper/state";
              fsType = "ext4";
            };
          };
        };

        boot = {
          loader.systemd-boot.enable = true;

          initrd = {
            systemd = {
              enable = true;
              # avoids reaching cryptsetup.target before recreation of the
              # "state" volume completed, during the factory reset
              services.systemd-repart.before = [
                "systemd-cryptsetup@state.service"
              ];
              repart = {
                enable = true;
                extraArgs = [
                  "--tpm2-pcrs=platform-code"
                ];
              };
            };
            luks.devices = lib.mkVMOverride {
              state = {
                device = "/dev/disk/by-partlabel/state";
                crypttabExtraOpts = [ "tpm2-device=auto" ];
              };
            };
          };
        };

        systemd.repart.partitions = {
          "10-esp".Type = "esp";
          "20-root".Type = "linux-generic";
          "30-state" = {
            Type = "linux-generic";
            Label = "state";
            Format = "ext4";
            Encrypt = "tpm2";
            SizeMinBytes = "64M";
            SizeMaxBytes = "64M";
            FactoryReset = true;
          };
        };

        # doesn't actually reboot through the service because otherwise the test
        # instrumentation becomes very unreliable, instead uses machine.reboot()
        systemd.services.systemd-factory-reset-reboot.enable = false;
      };

    testScript =
      { nodes, ... }:
      # python
      ''
        ${useDiskImage {
          inherit (nodes) machine;
          sizeDiff = "+64M";
        }}

        machine.start(allow_reboot=True)
        machine.wait_for_unit("default.target")

        first_uuid = machine.succeed("blkid -s UUID -o value /dev/disk/by-label/state")

        machine.succeed("mountpoint /state")
        machine.succeed("touch /state/foo")

        with subtest("factory reset requested through target"):
          machine.systemctl("start factory-reset.target")

        # reboot manually to keep control over test vm
        machine.reboot()
        machine.wait_for_unit("default.target")

        with subtest("state partition recreated and empty after reset"):
          second_uuid = machine.succeed("blkid -s UUID -o value /dev/disk/by-label/state")
          t.assertNotEqual(first_uuid, second_uuid)

          machine.succeed("mountpoint /state")
          machine.fail("test -e /state/foo")
      '';
  };

}