mirror of
https://github.com/lucannez64/darling-lol.git
synced 2026-10-26 19:47:42 +00:00
79 lines
2.8 KiB
Python
79 lines
2.8 KiB
Python
#!/usr/bin/env python3
|
|
"""Keep namespace-local shutdown authority until the foreground and clients exit.
|
|
|
|
Never kill by name or use a marker PID as authority to signal a host process.
|
|
Uninspectable mappings retain this namespace rather than guessing it is idle.
|
|
"""
|
|
import importlib.util
|
|
import os
|
|
from pathlib import Path
|
|
import signal
|
|
import subprocess
|
|
import sys
|
|
import time
|
|
|
|
|
|
def main():
|
|
scripts = Path(__file__).resolve().parent
|
|
spec = importlib.util.spec_from_file_location('nix_idle', scripts / 'nix-idle.py')
|
|
assert spec is not None and spec.loader is not None
|
|
idle = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(idle)
|
|
prefix = Path(os.environ['DPREFIX']).resolve()
|
|
child = None
|
|
interrupted = 0
|
|
|
|
def interrupted_by(signum, _frame):
|
|
nonlocal interrupted
|
|
interrupted = 128 + signum
|
|
# Only our own foreground child, never a marker PID or host-wide search.
|
|
if child is not None and child.poll() is None:
|
|
try:
|
|
child.send_signal(signum)
|
|
except ProcessLookupError:
|
|
pass
|
|
|
|
for signum in (signal.SIGINT, signal.SIGTERM, signal.SIGHUP):
|
|
signal.signal(signum, interrupted_by)
|
|
result = 1
|
|
try:
|
|
child = subprocess.Popen(sys.argv[1:])
|
|
result = child.wait()
|
|
if result < 0:
|
|
result = 128 - result
|
|
except OSError as error:
|
|
print(f'Cannot start client: {error}', file=sys.stderr)
|
|
finally:
|
|
# A launcher can return while guest clients remain. Do not tear down
|
|
# their server, or discard the only namespace capable of shutting it down.
|
|
last_error = None
|
|
while True:
|
|
try:
|
|
if not idle.client_active(prefix):
|
|
break
|
|
last_error = None
|
|
except (OSError, RuntimeError) as error:
|
|
if str(error) != last_error:
|
|
print(f'{error}; retaining original namespace and retrying. '
|
|
'Exit clients to allow safe shutdown.', file=sys.stderr)
|
|
last_error = str(error)
|
|
time.sleep(1)
|
|
while True:
|
|
try:
|
|
shutdown = subprocess.run(['bash', str(scripts / 'darling-rootless.sh'), 'shutdown']).returncode
|
|
except OSError as error:
|
|
print(f'Cannot run namespace-local shutdown: {error}', file=sys.stderr)
|
|
shutdown = 1
|
|
if not shutdown:
|
|
break
|
|
print(f'Darling namespace-local shutdown failed (status {shutdown}); '
|
|
'retaining the original namespace and retrying in 5 seconds. '
|
|
'Keep this terminal open.', file=sys.stderr)
|
|
time.sleep(5)
|
|
result = interrupted or result
|
|
return result
|
|
|
|
|
|
if __name__ == '__main__':
|
|
raise SystemExit(main())
|