mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-10-06 23:12:22 +02:00
48 lines
2.0 KiB
Python
48 lines
2.0 KiB
Python
"""Captured spawns exercise the production runner without signalling fake PIDs."""
|
|
import asyncio
|
|
import json
|
|
import os
|
|
from types import SimpleNamespace
|
|
|
|
from src import containment
|
|
|
|
|
|
def capture_owned_spawn(monkeypatch, tmp_path):
|
|
captured = {}
|
|
monkeypatch.setattr(containment, "CONTAINMENT_MODE", containment.MODE_REPORT_ONLY)
|
|
monkeypatch.setattr(containment, "_store_path", lambda: tmp_path / "grants.json")
|
|
monkeypatch.setattr(containment, "_pgid_of", lambda pid: pid)
|
|
|
|
async def fake_exec(*argv, **kwargs):
|
|
captured.update(argv=argv, kwargs=kwargs, command=argv[-1])
|
|
if "--info-fd" in argv:
|
|
fd = int(argv[argv.index("--info-fd") + 1])
|
|
os.write(fd, json.dumps({"child-pid": 99999998}).encode())
|
|
stdout = asyncio.StreamReader()
|
|
if "ody-boundary" in argv:
|
|
stdout.feed_data((argv[argv.index("ody-boundary") + 1] + "\n").encode())
|
|
stdout.feed_data(b"ok")
|
|
stdout.feed_eof()
|
|
stderr = asyncio.StreamReader()
|
|
stderr.feed_eof()
|
|
async def wait():
|
|
return 0
|
|
async def drain():
|
|
return None
|
|
writer = SimpleNamespace(write=lambda data: None, drain=drain,
|
|
close=lambda: captured.update(stdin_closed=True))
|
|
return SimpleNamespace(pid=99999999, stdout=stdout, stderr=stderr,
|
|
stdin=writer, returncode=0, wait=wait)
|
|
|
|
async def release(*args, **kwargs):
|
|
return containment.ReleaseOutcome(dead=True, escalated=False)
|
|
|
|
monkeypatch.setattr(asyncio, "create_subprocess_exec", fake_exec)
|
|
real_capture = containment.process_ownership.capture
|
|
monkeypatch.setattr(containment.process_ownership, "capture", lambda pid:
|
|
{"pid": pid, "start_token": "captured-fake"} if pid in (99999998, 99999999) else real_capture(pid))
|
|
if hasattr(containment.os, "pidfd_open"):
|
|
monkeypatch.delattr(containment.os, "pidfd_open")
|
|
monkeypatch.setattr(containment, "_release_awaited", release)
|
|
return captured
|