nyx/tests/dynamic_fixtures/python_frameworks/fastapi/benign.py

21 lines
530 B
Python
Raw Normal View History

2026-06-05 10:16:30 -05:00
"""Phase 12 (Track L.10) — FastAPI CMDI benign fixture.
`GET /run?cmd=<...>` rejects anything outside an allowlist before
invoking `subprocess.run` with a fixed argv, so the sink call is
unreachable for attacker-controlled values.
"""
import subprocess
from fastapi import FastAPI
app = FastAPI()
_ALLOW = {"status", "uptime", "version"}
@app.get("/run")
def run_cmd(cmd: str = ""):
if cmd not in _ALLOW:
return {"rejected": True}
subprocess.run(["/usr/bin/echo", cmd], check=False)
return {"ok": True}