mirror of
https://github.com/lucannez64/darling-lol.git
synced 2026-10-26 19:47:42 +00:00
nbandlucannez64
0d8da88894
* fix: preserve ambiguous guest processes during prefix cleanup * fix: warn only for matching runtime cleanup candidates --------- Co-authored-by: lucannez64 <[email protected]>
104 lines
3.9 KiB
Python
104 lines
3.9 KiB
Python
#!/usr/bin/env python3
|
|
"""Clear a dead Darling server's stale PID marker, and stale shared buffers once no server runs."""
|
|
import os
|
|
from pathlib import Path
|
|
import sys
|
|
import time
|
|
def recover_marker(prefix):
|
|
marker = prefix / '.init.pid'
|
|
try:
|
|
pid = int(marker.read_text().strip())
|
|
except (OSError, ValueError):
|
|
pid = 0
|
|
if pid > 0:
|
|
status = Path('/proc') / str(pid) / 'status'
|
|
try:
|
|
state = next(line.split()[1] for line in status.read_text().splitlines() if line.startswith('State:'))
|
|
dead = state == 'Z'
|
|
except FileNotFoundError:
|
|
dead = True
|
|
except (OSError, StopIteration):
|
|
dead = False
|
|
if dead:
|
|
os.replace(marker, prefix / f'.init.pid.stale-{pid}-{time.time_ns()}')
|
|
print(f'Recovered stale Darling server marker for PID {pid}', flush=True)
|
|
|
|
def any_server_running(proc_root=Path('/proc')):
|
|
"""True if any darlingserver runs, for any prefix: /dev/shm buffers are not per prefix."""
|
|
for proc in proc_root.iterdir():
|
|
if not proc.name.isdigit():
|
|
continue
|
|
try:
|
|
if (proc / 'comm').read_text().strip() == 'darlingserver':
|
|
return True
|
|
except PermissionError:
|
|
# A hidden process might be a server; shared buffers are global, so fail closed.
|
|
return True
|
|
except OSError:
|
|
pass
|
|
return False
|
|
|
|
def _matching_pids(kind, proc_root):
|
|
"""Find same-user candidates by guest executable; this does not establish ownership."""
|
|
uid = os.getuid()
|
|
for proc in proc_root.iterdir():
|
|
if not proc.name.isdigit():
|
|
continue
|
|
try:
|
|
if proc.stat().st_uid != uid:
|
|
continue
|
|
argv = (proc / 'cmdline').read_bytes()
|
|
if kind == 'launchd':
|
|
matches = argv.rstrip(b'\0') == b'/sbin/launchd'
|
|
else:
|
|
executable = argv.split(b'\0', 1)[0].decode(errors='replace')
|
|
matches = (proc / 'comm').read_text().startswith('chrome_crashpad') and any(
|
|
bundle in executable for bundle in (
|
|
'/Riot Client.app/', '/League of Legends.app/',
|
|
'/League of Legends (TM) Client.app/'))
|
|
if matches:
|
|
yield proc.name
|
|
except (OSError, ValueError):
|
|
continue
|
|
|
|
|
|
def stop_orphaned_launchd(proc_root=Path('/proc')):
|
|
"""Warn about launchd candidates without assuming prefix ownership or signalling them."""
|
|
pids = list(_matching_pids('launchd', proc_root))
|
|
if pids:
|
|
print(f'Preserving launchd candidate PIDs {", ".join(pids)}: prefix ownership '
|
|
'cannot be proved. If setup is blocked, shut down the original Darling '
|
|
'session or inspect the leftover processes manually.', file=sys.stderr, flush=True)
|
|
|
|
|
|
def stop_riot_crashpad(proc_root=Path('/proc')):
|
|
"""Warn about Riot crashpad candidates without assuming prefix ownership or signalling them."""
|
|
pids = list(_matching_pids('crashpad', proc_root))
|
|
if pids:
|
|
print(f'Preserving Riot crashpad candidate PIDs {", ".join(pids)}: prefix ownership '
|
|
'cannot be proved. Use the original session to shut down leftover helpers.',
|
|
file=sys.stderr, flush=True)
|
|
|
|
def remove_owned(paths):
|
|
uid = os.getuid()
|
|
for path in paths:
|
|
try:
|
|
if path.is_file() and path.stat().st_uid == uid:
|
|
path.unlink()
|
|
except OSError:
|
|
pass
|
|
|
|
def main():
|
|
prefix = Path(os.environ.get('DPREFIX', str(Path.home() / '.darling')))
|
|
recover_marker(prefix)
|
|
# Frame and present buffers left by a killed session confuse the next one's publishers.
|
|
if not any_server_running():
|
|
stop_orphaned_launchd()
|
|
stop_riot_crashpad()
|
|
remove_owned((prefix / 'private/tmp').glob('.com.riotgames.*'))
|
|
remove_owned(Path('/dev/shm').glob('darling-*'))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|