nixpkgs/nixos/tests/postgresql/postgresql-tls-client-cert.nix

{
  pkgs,
  makeTest,
}:

let
  inherit (pkgs) lib;

  runWithOpenSSL =
    file: cmd:
    pkgs.runCommand file {
      buildInputs = [ pkgs.openssl ];
    } cmd;
  caKey = runWithOpenSSL "ca.key" "openssl ecparam -name prime256v1 -genkey -noout -out $out";
  caCert = runWithOpenSSL "ca.crt" ''
    openssl req -new -x509 -sha256 -key ${caKey} -out $out -subj "/CN=test.example" -days 36500
  '';
  serverKey = runWithOpenSSL "server.key" "openssl ecparam -name prime256v1 -genkey -noout -out $out";
  serverKeyPath = "/var/lib/postgresql";
  serverCert = runWithOpenSSL "server.crt" ''
    openssl req -new -sha256 -key ${serverKey} -out server.csr -subj "/CN=db.test.example"
    openssl x509 -req -in server.csr -CA ${caCert} -CAkey ${caKey} \
      -CAcreateserial -out $out -days 36500 -sha256
  '';
  clientKey = runWithOpenSSL "client.key" "openssl ecparam -name prime256v1 -genkey -noout -out $out";
  clientCert = runWithOpenSSL "client.crt" ''
    openssl req -new -sha256 -key ${clientKey} -out client.csr -subj "/CN=test"
    openssl x509 -req -in client.csr -CA ${caCert} -CAkey ${caKey} \
      -CAcreateserial -out $out -days 36500 -sha256
  '';
  clientKeyPath = "/root";

  makeTestFor =
    package:
    makeTest {
      name = "postgresql-tls-client-cert-${package.name}";
      meta.maintainers = with lib.maintainers; [ erictapen ];

      nodes.server =
        { ... }:
        {
          system.activationScripts = {
            keyPlacement.text = ''
              mkdir -p '${serverKeyPath}'
              cp '${serverKey}' '${serverKeyPath}/server.key'
              chown postgres:postgres '${serverKeyPath}/server.key'
              chmod 600 '${serverKeyPath}/server.key'
            '';
          };
          services.postgresql = {
            inherit package;
            enable = true;
            enableJIT = lib.hasInfix "-jit-" package.name;
            enableTCPIP = true;
            ensureUsers = [
              {
                name = "test";
                ensureDBOwnership = true;
              }
            ];
            ensureDatabases = [ "test" ];
            settings = {
              ssl = "on";
              ssl_ca_file = toString caCert;
              ssl_cert_file = toString serverCert;
              ssl_key_file = "${serverKeyPath}/server.key";
            };
            authentication = ''
              hostssl test test ::/0 cert clientcert=verify-full
            '';
          };
          networking = {
            interfaces.eth1 = {
              ipv6.addresses = [
                {
                  address = "fc00::1";
                  prefixLength = 120;
                }
              ];
            };
            firewall.allowedTCPPorts = [ 5432 ];
          };
        };

      nodes.client =
        { ... }:
        {
          system.activationScripts = {
            keyPlacement.text = ''
              mkdir -p '${clientKeyPath}'
              cp '${clientKey}' '${clientKeyPath}/client.key'
              chown root:root '${clientKeyPath}/client.key'
              chmod 600 '${clientKeyPath}/client.key'
            '';
          };
          environment = {
            variables = {
              PGHOST = "db.test.example";
              PGPORT = "5432";
              PGDATABASE = "test";
              PGUSER = "test";
              PGSSLMODE = "verify-full";
              PGSSLCERT = clientCert;
              PGSSLKEY = "${clientKeyPath}/client.key";
              PGSSLROOTCERT = caCert;
            };
            systemPackages = [ package ];
          };
          networking = {
            interfaces.eth1 = {
              ipv6.addresses = [
                {
                  address = "fc00::2";
                  prefixLength = 120;
                }
              ];
            };
            hosts = {
              "fc00::1" = [ "db.test.example" ];
            };
          };
        };

      testScript = ''
        server.wait_for_unit("multi-user.target")
        client.wait_for_unit("multi-user.target")
        client.succeed("psql -c \"SELECT 1;\"")
      '';
    };
in
lib.recurseIntoAttrs (
  lib.concatMapAttrs (n: p: { ${n} = makeTestFor p; }) pkgs.postgresqlVersions
  // {
    passthru.override = p: makeTestFor p;
  }
)