commit 59cb7a87d74813a06866f7b9a2058c7dc5c9bacc · pyrox.dev/nixpkgs

+32 -4

nixos/modules/system/activation/switch-to-configuration.pl

···

       2
        
       

     

       3
        
       use strict;

     

       4
        
       use warnings;

     

       0
        
       
     

       5
        
       use File::Basename;

     

       6
        
       use File::Slurp;

     

       7
        
       use Net::DBus;

     
···

       14
        
       my $curSystemd = abs_path("/run/current-system/sw/bin");

     

       15
        
       

     

       16
        
       # To be robust against interruption, record what units need to be started etc.

     

       17
       -
       my $startListFile = "/run/systemd/start-list";

     

       18
       -
       my $restartListFile = "/run/systemd/restart-list";

     

       19
       -
       my $reloadListFile = "/run/systemd/reload-list";

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       20
        
       

     

       21
        
       my $action = shift @ARGV;

     

       22
        
       

     
···

       150
        
           split('\n', read_file($restartListFile, err_mode => 'quiet') // "");

     

       151
        
       

     

       152
        
       $unitsToReload{$_} = 1 foreach

     

       153
       -
           split '\n', read_file($reloadListFile, err_mode => 'quiet') // "";

     

       154
        
       

     

       155
        
       my $activePrev = getActiveUnits;

     

       156
        
       while (my ($unit, $state) = each %{$activePrev}) {

     
···

       366
        
           print STDERR "would activate the configuration...\n";

     

       367
        
           system("$out/dry-activate", "$out");

     

       368
        
       

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       369
        
           print STDERR "would restart systemd\n" if $restartSystemd;

     

       370
        
           print STDERR "would restart the following units: ", join(", ", sort(keys %unitsToRestart)), "\n"

     

       371
        
               if scalar(keys %unitsToRestart) > 0;

     
···

       373
        
               if scalar @unitsToStartFiltered;

     

       374
        
           print STDERR "would reload the following units: ", join(", ", sort(keys %unitsToReload)), "\n"

     

       375
        
               if scalar(keys %unitsToReload) > 0;

     

       0
        
       
     

       0
        
       
     

       376
        
           exit 0;

     

       377
        
       }

     

       378
        
       

     
···

       395
        
       print STDERR "activating the configuration...\n";

     

       396
        
       system("$out/activate", "$out") == 0 or $res = 2;

     

       397
        
       

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       398
        
       # Restart systemd if necessary. Note that this is done using the

     

       399
        
       # current version of systemd, just in case the new one has trouble

     

       400
        
       # communicating with the running pid 1.

     
···

       434
        
           print STDERR "reloading the following units: ", join(", ", sort(keys %unitsToReload)), "\n";

     

       435
        
           system("@systemd@/bin/systemctl", "reload", "--", sort(keys %unitsToReload)) == 0 or $res = 4;

     

       436
        
           unlink($reloadListFile);

     

       0
        
       
     

       437
        
       }

     

       438
        
       

     

       439
        
       # Restart changed services (those that have to be restarted rather

     
···

       442
        
           print STDERR "restarting the following units: ", join(", ", sort(keys %unitsToRestart)), "\n";

     

       443
        
           system("@systemd@/bin/systemctl", "restart", "--", sort(keys %unitsToRestart)) == 0 or $res = 4;

     

       444
        
           unlink($restartListFile);

     

       0
        
       
     

       445
        
       }

     

       446
        
       

     

       447
        
       # Start all active targets, as well as changed units we stopped above.

···

       2
        
       

     

       3
        
       use strict;

     

       4
        
       use warnings;

     

       5
       +
       use File::Path qw(make_path);

     

       6
        
       use File::Basename;

     

       7
        
       use File::Slurp;

     

       8
        
       use Net::DBus;

     
···

       15
        
       my $curSystemd = abs_path("/run/current-system/sw/bin");

     

       16
        
       

     

       17
        
       # To be robust against interruption, record what units need to be started etc.

     

       18
       +
       my $startListFile = "/run/nixos/start-list";

     

       19
       +
       my $restartListFile = "/run/nixos/restart-list";

     

       20
       +
       my $reloadListFile = "/run/nixos/reload-list";

     

       21
       +
       

     

       22
       +
       # Parse restart/reload requests by the activation script

     

       23
       +
       my $restartByActivationFile = "/run/nixos/activation-restart-list";

     

       24
       +
       my $reloadByActivationFile = "/run/nixos/activation-reload-list";

     

       25
       +
       my $dryRestartByActivationFile = "/run/nixos/dry-activation-restart-list";

     

       26
       +
       my $dryReloadByActivationFile = "/run/nixos/dry-activation-reload-list";

     

       27
       +
       

     

       28
       +
       make_path("/run/nixos", { mode => 0755 });

     

       29
        
       

     

       30
        
       my $action = shift @ARGV;

     

       31
        
       

     
···

       159
        
           split('\n', read_file($restartListFile, err_mode => 'quiet') // "");

     

       160
        
       

     

       161
        
       $unitsToReload{$_} = 1 foreach

     

       162
       +
           split('\n', read_file($reloadListFile, err_mode => 'quiet') // "");

     

       163
        
       

     

       164
        
       my $activePrev = getActiveUnits;

     

       165
        
       while (my ($unit, $state) = each %{$activePrev}) {

     
···

       375
        
           print STDERR "would activate the configuration...\n";

     

       376
        
           system("$out/dry-activate", "$out");

     

       377
        
       

     

       378
       +
           $unitsToRestart{$_} = 1 foreach

     

       379
       +
               split('\n', read_file($dryRestartByActivationFile, err_mode => 'quiet') // "");

     

       380
       +
       

     

       381
       +
           $unitsToReload{$_} = 1 foreach

     

       382
       +
               split('\n', read_file($dryReloadByActivationFile, err_mode => 'quiet') // "");

     

       383
       +
       

     

       384
        
           print STDERR "would restart systemd\n" if $restartSystemd;

     

       385
        
           print STDERR "would restart the following units: ", join(", ", sort(keys %unitsToRestart)), "\n"

     

       386
        
               if scalar(keys %unitsToRestart) > 0;

     
···

       388
        
               if scalar @unitsToStartFiltered;

     

       389
        
           print STDERR "would reload the following units: ", join(", ", sort(keys %unitsToReload)), "\n"

     

       390
        
               if scalar(keys %unitsToReload) > 0;

     

       391
       +
           unlink($dryRestartByActivationFile);

     

       392
       +
           unlink($dryReloadByActivationFile);

     

       393
        
           exit 0;

     

       394
        
       }

     

       395
        
       

     
···

       412
        
       print STDERR "activating the configuration...\n";

     

       413
        
       system("$out/activate", "$out") == 0 or $res = 2;

     

       414
        
       

     

       415
       +
       # Handle the activation script requesting the restart or reload of a unit.

     

       416
       +
       # We can only restart and reload (not stop/start) because the units to be

     

       417
       +
       # stopped are already stopped before the activation script is run.

     

       418
       +
       $unitsToRestart{$_} = 1 foreach

     

       419
       +
           split('\n', read_file($restartByActivationFile, err_mode => 'quiet') // "");

     

       420
       +
       

     

       421
       +
       $unitsToReload{$_} = 1 foreach

     

       422
       +
           split('\n', read_file($reloadByActivationFile, err_mode => 'quiet') // "");

     

       423
       +
       

     

       424
        
       # Restart systemd if necessary. Note that this is done using the

     

       425
        
       # current version of systemd, just in case the new one has trouble

     

       426
        
       # communicating with the running pid 1.

     
···

       460
        
           print STDERR "reloading the following units: ", join(", ", sort(keys %unitsToReload)), "\n";

     

       461
        
           system("@systemd@/bin/systemctl", "reload", "--", sort(keys %unitsToReload)) == 0 or $res = 4;

     

       462
        
           unlink($reloadListFile);

     

       463
       +
           unlink($reloadByActivationFile);

     

       464
        
       }

     

       465
        
       

     

       466
        
       # Restart changed services (those that have to be restarted rather

     
···

       469
        
           print STDERR "restarting the following units: ", join(", ", sort(keys %unitsToRestart)), "\n";

     

       470
        
           system("@systemd@/bin/systemctl", "restart", "--", sort(keys %unitsToRestart)) == 0 or $res = 4;

     

       471
        
           unlink($restartListFile);

     

       472
       +
           unlink($restartByActivationFile);

     

       473
        
       }

     

       474
        
       

     

       475
        
       # Start all active targets, as well as changed units we stopped above.

nixos/tests/all-tests.nix

···

       382
        
         radicale = handleTest ./radicale.nix {};

     

       383
        
         redis = handleTest ./redis.nix {};

     

       384
        
         redmine = handleTest ./redmine.nix {};

     

       0
        
       
     

       385
        
         restic = handleTest ./restic.nix {};

     

       386
        
         robustirc-bridge = handleTest ./robustirc-bridge.nix {};

     

       387
        
         roundcube = handleTest ./roundcube.nix {};

···

       382
        
         radicale = handleTest ./radicale.nix {};

     

       383
        
         redis = handleTest ./redis.nix {};

     

       384
        
         redmine = handleTest ./redmine.nix {};

     

       385
       +
         restartByActivationScript = handleTest ./restart-by-activation-script.nix {};

     

       386
        
         restic = handleTest ./restic.nix {};

     

       387
        
         robustirc-bridge = handleTest ./robustirc-bridge.nix {};

     

       388
        
         roundcube = handleTest ./roundcube.nix {};

+73

nixos/tests/restart-by-activation-script.nix

···

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

     

       2
       +
         name = "restart-by-activation-script";

     

       3
       +
         meta = with pkgs.lib.maintainers; {

     

       4
       +
           maintainers = [ das_j ];

     

       5
       +
         };

     

       6
       +
       

     

       7
       +
         machine = { pkgs, ... }: {

     

       8
       +
           imports = [ ../modules/profiles/minimal.nix ];

     

       9
       +
       

     

       10
       +
           systemd.services.restart-me = {

     

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

     

       12
       +
             serviceConfig = {

     

       13
       +
               Type = "oneshot";

     

       14
       +
               RemainAfterExit = true;

     

       15
       +
               ExecStart = "${pkgs.coreutils}/bin/true";

     

       16
       +
             };

     

       17
       +
           };

     

       18
       +
       

     

       19
       +
           systemd.services.reload-me = {

     

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

     

       21
       +
             serviceConfig = rec {

     

       22
       +
               Type = "oneshot";

     

       23
       +
               RemainAfterExit = true;

     

       24
       +
               ExecStart = "${pkgs.coreutils}/bin/true";

     

       25
       +
               ExecReload = ExecStart;

     

       26
       +
             };

     

       27
       +
           };

     

       28
       +
       

     

       29
       +
           system.activationScripts.test = {

     

       30
       +
             supportsDryActivation = true;

     

       31
       +
             text = ''

     

       32
       +
               if [ -e /test-the-activation-script ]; then

     

       33
       +
                 if [ "$NIXOS_ACTION" != dry-activate ]; then

     

       34
       +
                   touch /activation-was-run

     

       35
       +
                   echo restart-me.service > /run/nixos/activation-restart-list

     

       36
       +
                   echo reload-me.service > /run/nixos/activation-reload-list

     

       37
       +
                 else

     

       38
       +
                   echo restart-me.service > /run/nixos/dry-activation-restart-list

     

       39
       +
                   echo reload-me.service > /run/nixos/dry-activation-reload-list

     

       40
       +
                 fi

     

       41
       +
               fi

     

       42
       +
             '';

     

       43
       +
           };

     

       44
       +
         };

     

       45
       +
       

     

       46
       +
         testScript = /* python */ ''

     

       47
       +
           machine.wait_for_unit("multi-user.target")

     

       48
       +
       

     

       49
       +
           with subtest("nothing happens when the activation script does nothing"):

     

       50
       +
               out = machine.succeed("/run/current-system/bin/switch-to-configuration dry-activate 2>&1")

     

       51
       +
               assert 'restart' not in out

     

       52
       +
               assert 'reload' not in out

     

       53
       +
               out = machine.succeed("/run/current-system/bin/switch-to-configuration test")

     

       54
       +
               assert 'restart' not in out

     

       55
       +
               assert 'reload' not in out

     

       56
       +
       

     

       57
       +
           machine.succeed("touch /test-the-activation-script")

     

       58
       +
       

     

       59
       +
           with subtest("dry activation"):

     

       60
       +
               out = machine.succeed("/run/current-system/bin/switch-to-configuration dry-activate 2>&1")

     

       61
       +
               assert 'would restart the following units: restart-me.service' in out

     

       62
       +
               assert 'would reload the following units: reload-me.service' in out

     

       63
       +
               machine.fail("test -f /run/nixos/dry-activation-restart-list")

     

       64
       +
               machine.fail("test -f /run/nixos/dry-activation-reload-list")

     

       65
       +
       

     

       66
       +
           with subtest("real activation"):

     

       67
       +
               out = machine.succeed("/run/current-system/bin/switch-to-configuration test 2>&1")

     

       68
       +
               assert 'restarting the following units: restart-me.service' in out

     

       69
       +
               assert 'reloading the following units: reload-me.service' in out

     

       70
       +
               machine.fail("test -f /run/nixos/activation-restart-list")

     

       71
       +
               machine.fail("test -f /run/nixos/activation-reload-list")

     

       72
       +
         '';

     

       73
       +
       })