commit 0f46188ca10c2112e4af826233d203165ead17f4 · pyrox.dev/nixpkgs

nixos/tests/all-tests.nix

···

       81
       81
        
         gitlab = handleTest ./gitlab.nix {};

     

       82
       82
        
         gitolite = handleTest ./gitolite.nix {};

     

       83
       83
        
         gjs = handleTest ./gjs.nix {};

     

       84
       84
       +
         google-oslogin = handleTest ./google-oslogin {};

     

       84
       85
        
         gnome3 = handleTestOn ["x86_64-linux"] ./gnome3.nix {}; # libsmbios is unsupported on aarch64

     

       85
       86
        
         gnome3-gdm = handleTestOn ["x86_64-linux"] ./gnome3-gdm.nix {}; # libsmbios is unsupported on aarch64

     

       86
       87
        
         gocd-agent = handleTest ./gocd-agent.nix {};

+52

nixos/tests/google-oslogin/default.nix

···

       1
       1
       +
       import ../make-test.nix ({ pkgs, ... } :

     

       2
       2
       +
       let

     

       3
       3
       +
         inherit (import ./../ssh-keys.nix pkgs)

     

       4
       4
       +
           snakeOilPrivateKey snakeOilPublicKey;

     

       5
       5
       +
       in {

     

       6
       6
       +
         name = "google-oslogin";

     

       7
       7
       +
         meta = with pkgs.stdenv.lib.maintainers; {

     

       8
       8
       +
           maintainers = [ adisbladis flokli ];

     

       9
       9
       +
         };

     

       10
       10
       +
       

     

       11
       11
       +
         nodes = {

     

       12
       12
       +
           # the server provides both the the mocked google metadata server and the ssh server

     

       13
       13
       +
           server = (import ./server.nix pkgs);

     

       14
       14
       +
       

     

       15
       15
       +
           client = { ... }: {};

     

       16
       16
       +
         };

     

       17
       17
       +
         testScript =  ''

     

       18
       18
       +
           startAll;

     

       19
       19
       +
       

     

       20
       20
       +
           $server->waitForUnit("mock-google-metadata.service");

     

       21
       21
       +
           $server->waitForOpenPort(80);

     

       22
       22
       +
       

     

       23
       23
       +
           # mockserver should return a non-expired ssh key for both mockuser and mockadmin

     

       24
       24
       +
           $server->succeed('${pkgs.google-compute-engine-oslogin}/bin/google_authorized_keys mockuser | grep -q "${snakeOilPublicKey}"');

     

       25
       25
       +
           $server->succeed('${pkgs.google-compute-engine-oslogin}/bin/google_authorized_keys mockadmin | grep -q "${snakeOilPublicKey}"');

     

       26
       26
       +
       

     

       27
       27
       +
           # install snakeoil ssh key on the client

     

       28
       28
       +
           $client->succeed("mkdir -p ~/.ssh");

     

       29
       29
       +
           $client->succeed("cat ${snakeOilPrivateKey} > ~/.ssh/id_snakeoil");

     

       30
       30
       +
           $client->succeed("chmod 600 ~/.ssh/id_snakeoil");

     

       31
       31
       +
       

     

       32
       32
       +
           $client->waitForUnit("network.target");

     

       33
       33
       +
           $server->waitForUnit("sshd.service");

     

       34
       34
       +
       

     

       35
       35
       +
           # we should not be able to connect as non-existing user

     

       36
       36
       +
           $client->fail("ssh -o User=ghost -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no server -i ~/.ssh/id_snakeoil 'true'");

     

       37
       37
       +
       

     

       38
       38
       +
           # we should be able to connect as mockuser

     

       39
       39
       +
           $client->succeed("ssh -o User=mockuser -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no server -i ~/.ssh/id_snakeoil 'true'");

     

       40
       40
       +
           # but we shouldn't be able to sudo

     

       41
       41
       +
           $client->fail("ssh -o User=mockuser -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no server -i ~/.ssh/id_snakeoil '/run/wrappers/bin/sudo /run/current-system/sw/bin/id' | grep -q 'root'");

     

       42
       42
       +
       

     

       43
       43
       +
           # we should also be able to log in as mockadmin

     

       44
       44
       +
           $client->succeed("ssh -o User=mockadmin -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no server -i ~/.ssh/id_snakeoil 'true'");

     

       45
       45
       +
           # pam_oslogin_admin.so should now have generated a sudoers file

     

       46
       46
       +
           $server->succeed("find /run/google-sudoers.d | grep -q '/run/google-sudoers.d/mockadmin'");

     

       47
       47
       +
       

     

       48
       48
       +
           # and we should be able to sudo

     

       49
       49
       +
           $client->succeed("ssh -o User=mockadmin -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no server -i ~/.ssh/id_snakeoil '/run/wrappers/bin/sudo /run/current-system/sw/bin/id' | grep -q 'root'");

     

       50
       50
       +
         '';

     

       51
       51
       +
         })

     

       52
       52
       +

+29

nixos/tests/google-oslogin/server.nix

···

       1
       1
       +
       { pkgs, ... }:

     

       2
       2
       +
       let

     

       3
       3
       +
         inherit (import ./../ssh-keys.nix pkgs)

     

       4
       4
       +
           snakeOilPrivateKey snakeOilPublicKey;

     

       5
       5
       +
       in {

     

       6
       6
       +
         networking.firewall.allowedTCPPorts = [ 80 ];

     

       7
       7
       +
       

     

       8
       8
       +
         systemd.services.mock-google-metadata = {

     

       9
       9
       +
           description = "Mock Google metadata service";

     

       10
       10
       +
           serviceConfig.Type = "simple";

     

       11
       11
       +
           serviceConfig.ExecStart = "${pkgs.python3}/bin/python ${./server.py}";

     

       12
       12
       +
           environment = {

     

       13
       13
       +
             SNAKEOIL_PUBLIC_KEY = snakeOilPublicKey;

     

       14
       14
       +
           };

     

       15
       15
       +
           wantedBy = [ "multi-user.target" ];

     

       16
       16
       +
           after = [ "network.target" ];

     

       17
       17
       +
         };

     

       18
       18
       +
       

     

       19
       19
       +
         services.openssh.enable = true;

     

       20
       20
       +
         services.openssh.challengeResponseAuthentication = false;

     

       21
       21
       +
         services.openssh.passwordAuthentication = false;

     

       22
       22
       +
       

     

       23
       23
       +
         security.googleOsLogin.enable = true;

     

       24
       24
       +
       

     

       25
       25
       +
         # Mock google service

     

       26
       26
       +
         networking.extraHosts = ''

     

       27
       27
       +
           127.0.0.1 metadata.google.internal

     

       28
       28
       +
         '';

     

       29
       29
       +
       }

+96

nixos/tests/google-oslogin/server.py

···

       1
       1
       +
       #!/usr/bin/env python3

     

       2
       2
       +
       import json

     

       3
       3
       +
       import sys

     

       4
       4
       +
       import time

     

       5
       5
       +
       import os

     

       6
       6
       +
       import hashlib

     

       7
       7
       +
       import base64

     

       8
       8
       +
       

     

       9
       9
       +
       from http.server import BaseHTTPRequestHandler, HTTPServer

     

       10
       10
       +
       from typing import Dict

     

       11
       11
       +
       

     

       12
       12
       +
       SNAKEOIL_PUBLIC_KEY = os.environ['SNAKEOIL_PUBLIC_KEY']

     

       13
       13
       +
       

     

       14
       14
       +
       

     

       15
       15
       +
       def w(msg):

     

       16
       16
       +
           sys.stderr.write(f"{msg}\n")

     

       17
       17
       +
           sys.stderr.flush()

     

       18
       18
       +
       

     

       19
       19
       +
       

     

       20
       20
       +
       def gen_fingerprint(pubkey):

     

       21
       21
       +
           decoded_key = base64.b64decode(pubkey.encode("ascii").split()[1])

     

       22
       22
       +
           return hashlib.sha256(decoded_key).hexdigest()

     

       23
       23
       +
       

     

       24
       24
       +
       def gen_email(username):

     

       25
       25
       +
           """username seems to be a 21 characters long number string, so mimic that in a reproducible way"""

     

       26
       26
       +
           return str(int(hashlib.sha256(username.encode()).hexdigest(), 16))[0:21]

     

       27
       27
       +
       

     

       28
       28
       +
       def gen_mockuser(username: str, uid: str, gid: str, home_directory: str, snakeoil_pubkey: str) -> Dict:

     

       29
       29
       +
           snakeoil_pubkey_fingerprint = gen_fingerprint(snakeoil_pubkey)

     

       30
       30
       +
           # seems to be a 21 characters long numberstring, so mimic that in a reproducible way

     

       31
       31
       +
           email = gen_email(username)

     

       32
       32
       +
           return {

     

       33
       33
       +
               "loginProfiles": [

     

       34
       34
       +
                   {

     

       35
       35
       +
                       "name": email,

     

       36
       36
       +
                       "posixAccounts": [

     

       37
       37
       +
                           {

     

       38
       38
       +
                               "primary": True,

     

       39
       39
       +
                               "username": username,

     

       40
       40
       +
                               "uid": uid,

     

       41
       41
       +
                               "gid": gid,

     

       42
       42
       +
                               "homeDirectory": home_directory,

     

       43
       43
       +
                               "operatingSystemType": "LINUX"

     

       44
       44
       +
                           }

     

       45
       45
       +
                       ],

     

       46
       46
       +
                       "sshPublicKeys": {

     

       47
       47
       +
                           snakeoil_pubkey_fingerprint: {

     

       48
       48
       +
                               "key": snakeoil_pubkey,

     

       49
       49
       +
                               "expirationTimeUsec": str((time.time() + 600) * 1000000),  # 10 minutes in the future

     

       50
       50
       +
                               "fingerprint": snakeoil_pubkey_fingerprint

     

       51
       51
       +
                           }

     

       52
       52
       +
                       }

     

       53
       53
       +
                   }

     

       54
       54
       +
               ]

     

       55
       55
       +
           }

     

       56
       56
       +
       

     

       57
       57
       +
       

     

       58
       58
       +
       class ReqHandler(BaseHTTPRequestHandler):

     

       59
       59
       +
           def _send_json_ok(self, data):

     

       60
       60
       +
               self.send_response(200)

     

       61
       61
       +
               self.send_header('Content-type', 'application/json')

     

       62
       62
       +
               self.end_headers()

     

       63
       63
       +
               out = json.dumps(data).encode()

     

       64
       64
       +
               w(out)

     

       65
       65
       +
               self.wfile.write(out)

     

       66
       66
       +
       

     

       67
       67
       +
           def do_GET(self):

     

       68
       68
       +
               p = str(self.path)

     

       69
       69
       +
               # mockuser and mockadmin are allowed to login, both use the same snakeoil public key

     

       70
       70
       +
               if p == '/computeMetadata/v1/oslogin/users?username=mockuser' \

     

       71
       71
       +
                   or p == '/computeMetadata/v1/oslogin/users?uid=1009719690':

     

       72
       72
       +
                   self._send_json_ok(gen_mockuser(username='mockuser', uid='1009719690', gid='1009719690',

     

       73
       73
       +
                                                   home_directory='/home/mockuser', snakeoil_pubkey=SNAKEOIL_PUBLIC_KEY))

     

       74
       74
       +
               elif p == '/computeMetadata/v1/oslogin/users?username=mockadmin' \

     

       75
       75
       +
                   or p == '/computeMetadata/v1/oslogin/users?uid=1009719691':

     

       76
       76
       +
                   self._send_json_ok(gen_mockuser(username='mockadmin', uid='1009719691', gid='1009719691',

     

       77
       77
       +
                                                   home_directory='/home/mockadmin', snakeoil_pubkey=SNAKEOIL_PUBLIC_KEY))

     

       78
       78
       +
       

     

       79
       79
       +
               # mockuser is allowed to login

     

       80
       80
       +
               elif p == f"/computeMetadata/v1/oslogin/authorize?email={gen_email('mockuser')}&policy=login":

     

       81
       81
       +
                   self._send_json_ok({'success': True})

     

       82
       82
       +
       

     

       83
       83
       +
               # mockadmin may also become root

     

       84
       84
       +
               elif p == f"/computeMetadata/v1/oslogin/authorize?email={gen_email('mockadmin')}&policy=login" or p == f"/computeMetadata/v1/oslogin/authorize?email={gen_email('mockadmin')}&policy=adminLogin":

     

       85
       85
       +
                   self._send_json_ok({'success': True})

     

       86
       86
       +
               else:

     

       87
       87
       +
                   sys.stderr.write(f"Unhandled path: {p}\n")

     

       88
       88
       +
                   sys.stderr.flush()

     

       89
       89
       +
                   self.send_response(501)

     

       90
       90
       +
                   self.end_headers()

     

       91
       91
       +
                   self.wfile.write(b'')

     

       92
       92
       +
       

     

       93
       93
       +
       

     

       94
       94
       +
       if __name__ == '__main__':

     

       95
       95
       +
           s = HTTPServer(('0.0.0.0', 80), ReqHandler)

     

       96
       96
       +
           s.serve_forever()