nixos/vms: Implement clean shutdown
This commit is contained in:
parent
a907ae428f
commit
b3862a129a
6
flake.lock
generated
6
flake.lock
generated
@ -180,11 +180,11 @@
|
|||||||
},
|
},
|
||||||
"nixpkgs-mine": {
|
"nixpkgs-mine": {
|
||||||
"locked": {
|
"locked": {
|
||||||
"lastModified": 1651928992,
|
"lastModified": 1651945192,
|
||||||
"narHash": "sha256-U9IgIFKj1W6YBsoFy3AG7YaxBw9qQTYeGhvlLcKfUYQ=",
|
"narHash": "sha256-3pnzK9RpuWzqnqd1U7zt/z3gvn/UNY41CJuS6Ow/Vwo=",
|
||||||
"owner": "devplayer0",
|
"owner": "devplayer0",
|
||||||
"repo": "nixpkgs",
|
"repo": "nixpkgs",
|
||||||
"rev": "e0844396c59d27f26f18b8ee4bff4fdda0e39cfa",
|
"rev": "95bce541ae41d144b8edb6c47b3b7987295b006f",
|
||||||
"type": "github"
|
"type": "github"
|
||||||
},
|
},
|
||||||
"original": {
|
"original": {
|
||||||
|
@ -12,6 +12,32 @@ let
|
|||||||
qemuOpts = with lib.types; coercedTo (attrsOf unspecified) flattenQEMUOpts str;
|
qemuOpts = with lib.types; coercedTo (attrsOf unspecified) flattenQEMUOpts str;
|
||||||
extraQEMUOpts = o: optionalString (o != "") ",${o}";
|
extraQEMUOpts = o: optionalString (o != "") ",${o}";
|
||||||
|
|
||||||
|
doCleanShutdown =
|
||||||
|
let
|
||||||
|
pyEnv = pkgs.python310.withPackages (ps: with ps; [ qemu ]);
|
||||||
|
in
|
||||||
|
pkgs.writeScript "qemu-clean-shutdown" ''
|
||||||
|
#!${pyEnv}/bin/python
|
||||||
|
import sys
|
||||||
|
import os
|
||||||
|
|
||||||
|
import qemu.qmp
|
||||||
|
|
||||||
|
if len(sys.argv) != 2:
|
||||||
|
print(f'usage: {sys.argv[0]} <qmp unix socket>', file=sys.stderr)
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
if not os.path.exists(sys.argv[1]) and 'MAINPID' not in os.environ:
|
||||||
|
# Special case: systemd is calling us after QEMU exited on its own
|
||||||
|
sys.exit(0)
|
||||||
|
|
||||||
|
with qemu.qmp.QEMUMonitorProtocol(sys.argv[1]) as mon:
|
||||||
|
mon.connect()
|
||||||
|
mon.command('system_powerdown')
|
||||||
|
while mon.pull_event(wait=True)['event'] != 'SHUTDOWN':
|
||||||
|
pass
|
||||||
|
'';
|
||||||
|
|
||||||
cfg = config.my.vms;
|
cfg = config.my.vms;
|
||||||
|
|
||||||
netOpts = with lib.types; { name, ... }: {
|
netOpts = with lib.types; { name, ... }: {
|
||||||
@ -39,6 +65,10 @@ let
|
|||||||
qemuBin = mkOpt' path "${pkgs.qemu_kvm}/bin/qemu-kvm" "Path to QEMU executable.";
|
qemuBin = mkOpt' path "${pkgs.qemu_kvm}/bin/qemu-kvm" "Path to QEMU executable.";
|
||||||
qemuFlags = mkOpt' (listOf str) [ ] "Additional flags to pass to QEMU.";
|
qemuFlags = mkOpt' (listOf str) [ ] "Additional flags to pass to QEMU.";
|
||||||
autoStart = mkBoolOpt' true "Whether to start the VM automatically at boot.";
|
autoStart = mkBoolOpt' true "Whether to start the VM automatically at boot.";
|
||||||
|
cleanShutdown = {
|
||||||
|
enabled = mkBoolOpt' true "Whether to attempt to cleanly shut down the guest.";
|
||||||
|
timeout = mkOpt' ints.unsigned 30 "Clean shutdown timeout (in seconds).";
|
||||||
|
};
|
||||||
|
|
||||||
machine = mkOpt' str "q35" "QEMU machine type.";
|
machine = mkOpt' str "q35" "QEMU machine type.";
|
||||||
enableKVM = mkBoolOpt' true "Whether to enable KVM.";
|
enableKVM = mkBoolOpt' true "Whether to enable KVM.";
|
||||||
@ -68,8 +98,10 @@ let
|
|||||||
"m ${toString i.memory}"
|
"m ${toString i.memory}"
|
||||||
"nographic"
|
"nographic"
|
||||||
"vga ${i.vga}"
|
"vga ${i.vga}"
|
||||||
|
"chardev socket,id=monitor-qmp,path=/run/vms/${n}/monitor-qmp.sock,server=on,wait=off"
|
||||||
|
"mon chardev=monitor-qmp,mode=control"
|
||||||
"chardev socket,id=monitor,path=/run/vms/${n}/monitor.sock,server=on,wait=off"
|
"chardev socket,id=monitor,path=/run/vms/${n}/monitor.sock,server=on,wait=off"
|
||||||
"mon chardev=monitor"
|
"mon chardev=monitor,mode=readline"
|
||||||
"chardev socket,id=tty,path=/run/vms/${n}/tty.sock,server=on,wait=off"
|
"chardev socket,id=tty,path=/run/vms/${n}/tty.sock,server=on,wait=off"
|
||||||
"device isa-serial,chardev=tty"
|
"device isa-serial,chardev=tty"
|
||||||
] ++
|
] ++
|
||||||
@ -108,6 +140,9 @@ in
|
|||||||
description = "Virtual machine '${n}'";
|
description = "Virtual machine '${n}'";
|
||||||
serviceConfig = {
|
serviceConfig = {
|
||||||
ExecStart = mkQemuCommand n i;
|
ExecStart = mkQemuCommand n i;
|
||||||
|
ExecStop = mkIf i.cleanShutdown.enabled "${doCleanShutdown} /run/vms/${n}/monitor-qmp.sock";
|
||||||
|
TimeoutStopSec = mkIf i.cleanShutdown.enabled i.cleanShutdown.timeout;
|
||||||
|
|
||||||
RuntimeDirectory = "vms/${n}";
|
RuntimeDirectory = "vms/${n}";
|
||||||
StateDirectory = "vms/${n}";
|
StateDirectory = "vms/${n}";
|
||||||
};
|
};
|
||||||
@ -120,7 +155,7 @@ in
|
|||||||
'';
|
'';
|
||||||
postStart =
|
postStart =
|
||||||
''
|
''
|
||||||
socks=(monitor tty spice)
|
socks=(monitor-qmp monitor tty spice)
|
||||||
for s in ''${socks[@]}; do
|
for s in ''${socks[@]}; do
|
||||||
path="$RUNTIME_DIRECTORY"/''${s}.sock
|
path="$RUNTIME_DIRECTORY"/''${s}.sock
|
||||||
until [ -e "$path" ]; do sleep 0.1; done
|
until [ -e "$path" ]; do sleep 0.1; done
|
||||||
|
Loading…
Reference in New Issue
Block a user