about summary refs log tree commit diff
path: root/tests/programs/gnupg/default.nix
blob: 24756a3e460aadbe7df6d1a9ccf03f1db2e4f298 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
{ pkgs, ... }:

let
  mkExpect = expectScript: script: pkgs.writeScript "test-gnupg-cli" ''
    #!${pkgs.expect}/bin/expect -f
    set timeout 20
    spawn ${pkgs.writeScript "cli-testscript.sh" ''
      #!${pkgs.stdenv.shell} -e
      ${script}
    ''}
    ${expectScript}
    set ret [wait]
    exit [lindex $ret 3]
  '';

  cliTestWithPassphrase = mkExpect ''
    expect -regexp ---+.*Please.enter
    send supersecret\r
  '';

  cliTest = mkExpect "";

in {
  name = "gnupg";

  enableOCR = true;

  machine = { lib, ... }: {
    imports = map (what:
      "${import ../../../nixpkgs-path.nix}/nixos/tests/common/${what}.nix"
    ) [ "user-account" "x11" ];

    services.openssh.enable = true;
    services.xserver.displayManager.auto.user = "alice";

    vuizvui.programs.gnupg.enable = true;
    vuizvui.programs.gnupg.agent.enable = true;
    vuizvui.programs.gnupg.agent.sshSupport = true;
    vuizvui.programs.gnupg.agent.scdaemon.enable = true;

    programs.ssh.startAgent = false;
  };

  testScript = ''
    $machine->waitForUnit("sshd.service");
    $machine->succeed("ssh-keygen -t ed25519 -f /root/id_ed25519 -N '''");
    my $cmd = 'mkdir -p ~/.ssh && cat > ~/.ssh/authorized_keys';
    $machine->succeed("su -c 'umask 0077; $cmd' alice < /root/id_ed25519.pub");

    $machine->waitForX;

    sub ssh ($) {
      my $esc = $_[0] =~ s/'/'\\${"'"}'/gr;
      return "ssh -q -i /root/id_ed25519".
             " -o StrictHostKeyChecking=no".
             " alice\@127.0.0.1 -- '$esc'";
    }

    sub xsu ($) {
      my $esc = $_[0] =~ s/'/'\\${"'"}'/gr;
      return "DISPLAY=:0 su alice -c '$esc'";
    }

    $machine->nest("import snakeoil key", sub {
      $machine->succeed(ssh "${cliTestWithPassphrase ''
        gpg --import ${./snakeoil.asc}
      ''}");
      $machine->succeed(ssh "${mkExpect ''
        expect gpg>
        send trust\r
        expect decision?
        send 5\r
        expect "Do you really want"
        send y\r
        expect gpg>
        send save\r
      '' "gpg --edit-key ECC15FE1"}");
    });

    subtest "test SSH agent support", sub {
      $machine->succeed(ssh 'ssh-keygen -t ed25519 -f ~/testkey -N ""');
      $machine->succeed(ssh '${mkExpect ''
        expect -regexp ---+.*Please.enter
        send supersecret\r
        expect -regexp ---+.*Please.re-en
        send supersecret\r
      '' "ssh-add ~/testkey"}');

      $machine->succeed("umask 0077; $cmd < ~alice/testkey.pub");
      $machine->succeed(ssh 'rm ~/testkey*');

      $machine->succeed(ssh 'ssh -o StrictHostKeyChecking=no root@127.0.0.1'.
                            ' touch /i_have_thu_powarr');
      $machine->succeed("test -e /i_have_thu_powarr");

      $machine->succeed(ssh "systemctl --user reload gpg-agent");

      $machine->succeed(ssh "${cliTestWithPassphrase ''
        ssh -o StrictHostKeyChecking=no root@127.0.0.1 \
          touch /i_still_have_thu_powarr
      ''}");
      $machine->succeed("test -e /i_still_have_thu_powarr");
    };

    subtest "socket persists after restart", sub {
      $machine->succeed(ssh 'test -e "$SSH_AUTH_SOCK"');
      $machine->succeed(ssh 'systemctl --user stop gpg-agent.service');
      $machine->succeed(ssh 'test -e "$SSH_AUTH_SOCK"');
    };

    subtest "test from SSH", sub {
      $machine->execute(ssh "systemctl --user reload gpg-agent");
      $machine->succeed(ssh "${cliTestWithPassphrase ''
        echo encrypt me > to_encrypt
        gpg -sea -r ECC15FE1 to_encrypt
        rm to_encrypt
      ''}");
      $machine->succeed(ssh "${cliTest ''
        [ "$(gpg -d to_encrypt.asc)" = "encrypt me" ]
      ''}");
    };

    subtest "test from X", sub {
      $machine->execute(ssh "systemctl --user reload gpg-agent");
      my $pid = $machine->succeed(xsu
        'echo encrypt me | gpg -sea -r ECC15FE1 > encrypted_x.asc & echo $!'
      );
      chomp $pid;
      $machine->waitForText(qr/Passphrase/);
      $machine->screenshot("passphrase_dialog");
      $machine->sendChars("supersecret\n");
      $machine->waitUntilFails("kill -0 $pid");
      $machine->succeed(xsu '[ "$(gpg -d encrypted_x.asc)" = "encrypt me" ]');
    };
  '';
}