mirror of
https://github.com/norandom/OpenUltraSAST.git
synced 2026-10-16 03:49:26 +00:00
feat(authorization-obligations): hunter adjudicates intent without raising evidence
Task 3.3. For every finding that carries a route the hunter answers one closed question, is this route meant to be public, and the answer is recorded as public, protected or unknown with its rationale. Nothing else changes: not the label, not the evidence, not the evidence level, and the hunter never authors a pattern or a policy. The prompt carries the handler's own lines only and is redacted before it leaves the process. Without a client one degradation is recorded and the findings are identical. Co-Authored-By: Claude Opus 5 (1M context) <[email protected]> Claude-Session: https://claude.ai/code/session_01T7XD7x5KonMdmGRkp63EQG
This commit is contained in:
co-authored by
Claude Opus 5
parent
eab89bf203
commit
4bd59555ae
@@ -0,0 +1,100 @@
|
|||||||
|
"""Hunter adjudication of intent (authorization-obligations, Req 6.3).
|
||||||
|
|
||||||
|
The hunter answers one closed question per route-bearing finding: is this route meant to be public? The answer is recorded
|
||||||
|
on the finding as `intent` (public | protected | unknown) with its rationale, and nothing else changes: not the label, not
|
||||||
|
the evidence, not the evidence level. Without a client one `intent_adjudication_unavailable` degradation is recorded.
|
||||||
|
"""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import json
|
||||||
|
import re
|
||||||
|
from collections.abc import Mapping
|
||||||
|
from dataclasses import replace
|
||||||
|
|
||||||
|
from ...redaction import redact_secrets
|
||||||
|
from .check import ObligationFinding, ObligationResult
|
||||||
|
|
||||||
|
INTENTS = ("public", "protected", "unknown")
|
||||||
|
_SYSTEM = (
|
||||||
|
"You review web handlers for an application security tool. Answer one closed question about intent only. "
|
||||||
|
'Reply with JSON of the form {"intent": "public" | "protected" | "unknown", "rationale": "<one sentence>"}. '
|
||||||
|
"Do not assess exploitability and do not propose code."
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def adjudicate_intent(result: ObligationResult, *, client: object | None, model: str, texts: Mapping[str, str]) -> ObligationResult:
|
||||||
|
"""Annotate route-bearing findings with the hunter's intent verdict; identical findings when no client is available."""
|
||||||
|
if not any(finding.route for finding in result.findings):
|
||||||
|
return result # nothing to adjudicate: no degradation for a run with no route-bearing obligation finding
|
||||||
|
if client is None or not model:
|
||||||
|
degradation: dict[str, object] = {
|
||||||
|
"stage": "obligations",
|
||||||
|
"reason": "intent_adjudication_unavailable",
|
||||||
|
"detail": "no hunter client or model",
|
||||||
|
}
|
||||||
|
return replace(result, degradations=(*result.degradations, degradation))
|
||||||
|
findings: list[ObligationFinding] = []
|
||||||
|
for finding in result.findings:
|
||||||
|
if not finding.route:
|
||||||
|
findings.append(finding)
|
||||||
|
continue
|
||||||
|
intent, rationale = _ask(client, model, finding, texts.get(finding.operation.path, ""))
|
||||||
|
findings.append(replace(finding, intent=intent, intent_rationale=rationale))
|
||||||
|
return replace(result, findings=tuple(findings))
|
||||||
|
|
||||||
|
|
||||||
|
def _ask(client: object, model: str, finding: ObligationFinding, text: str) -> tuple[str, str | None]:
|
||||||
|
siblings = ", ".join(item for item in finding.evidence if "::" in item) or "none recorded"
|
||||||
|
handler = _handler_excerpt(text, finding.operation.function)
|
||||||
|
question = (
|
||||||
|
f"Route {finding.route} is served by {finding.operation.function} in {finding.operation.path}. "
|
||||||
|
f"It performs {finding.operation.kind} on {finding.operation.resource or 'a resource'} without {finding.missing}. "
|
||||||
|
f"Sibling handlers that discharge it: {siblings}. Is this route meant to be public?\n\n{handler}"
|
||||||
|
)
|
||||||
|
question = redact_secrets(question) # design Security Considerations: the hunter receives redacted handler text
|
||||||
|
try:
|
||||||
|
response = client.complete( # type: ignore[attr-defined]
|
||||||
|
model=model,
|
||||||
|
messages=[{"role": "system", "content": _SYSTEM}, {"role": "user", "content": question}],
|
||||||
|
tools=[],
|
||||||
|
timeout_seconds=30,
|
||||||
|
)
|
||||||
|
except Exception: # noqa: BLE001 - the hunter is advisory; a failed call records unknown
|
||||||
|
return "unknown", None
|
||||||
|
return _parse(getattr(response, "content", None))
|
||||||
|
|
||||||
|
|
||||||
|
def _parse(content: object) -> tuple[str, str | None]:
|
||||||
|
if not isinstance(content, str) or not content.strip():
|
||||||
|
return "unknown", None
|
||||||
|
match = re.search(r"\{.*\}", content, re.S)
|
||||||
|
if not match:
|
||||||
|
return "unknown", None
|
||||||
|
try:
|
||||||
|
payload = json.loads(match.group(0))
|
||||||
|
except json.JSONDecodeError:
|
||||||
|
return "unknown", None
|
||||||
|
intent = str(payload.get("intent", "unknown")).strip().lower()
|
||||||
|
rationale = payload.get("rationale")
|
||||||
|
return (intent if intent in INTENTS else "unknown"), (str(rationale)[:200] if rationale else None)
|
||||||
|
|
||||||
|
|
||||||
|
def _handler_excerpt(text: str, function: str) -> str:
|
||||||
|
"""The handler's own lines: its decorators through the end of its indented body, never another handler's."""
|
||||||
|
lines = text.splitlines()
|
||||||
|
start = next(
|
||||||
|
(i for i, line in enumerate(lines) if re.match(rf"\s*(?:async\s+)?(?:def|function)\s+{re.escape(function)}\b", line)), None
|
||||||
|
)
|
||||||
|
if start is None:
|
||||||
|
return ""
|
||||||
|
head = start
|
||||||
|
while head > 0 and lines[head - 1].strip().startswith("@"):
|
||||||
|
head -= 1
|
||||||
|
end = start + 1
|
||||||
|
while end < len(lines) and (not lines[end].strip() or lines[end].startswith((" ", "\t"))):
|
||||||
|
end += 1
|
||||||
|
return "\n".join(lines[head:end][:60])
|
||||||
|
|
||||||
|
|
||||||
|
__all__ = ["INTENTS", "adjudicate_intent"]
|
||||||
@@ -0,0 +1,88 @@
|
|||||||
|
"""authorization-obligations task 3.3: the hunter adjudicates intent without raising evidence (offline, scripted)."""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import json
|
||||||
|
import sys
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
sys.path.insert(0, str(Path(__file__).parent))
|
||||||
|
from test_obligation_check import _check # noqa: E402
|
||||||
|
|
||||||
|
from openultrasast.tool_hunter import ChatResponse, ScriptedChatClient # noqa: E402
|
||||||
|
|
||||||
|
|
||||||
|
def test_scripted_client_answers_are_recorded_on_the_finding_and_change_nothing_else() -> None:
|
||||||
|
from openultrasast.semantic.obligations.intent import adjudicate_intent
|
||||||
|
|
||||||
|
result = _check()
|
||||||
|
before = {(f.operation.function, f.missing): (f.label, f.evidence) for f in result.findings}
|
||||||
|
client = ScriptedChatClient([ChatResponse(content='{"intent": "public", "rationale": "health-style listing endpoint"}')] * 10)
|
||||||
|
adjudicated = adjudicate_intent(result, client=client, model="test-model", texts={"app.py": "..."})
|
||||||
|
leaky = next(f for f in adjudicated.findings if f.operation.function == "leaky" and f.missing == "identity_constraint")
|
||||||
|
assert leaky.intent == "public" and leaky.label == before[("leaky", "identity_constraint")][0]
|
||||||
|
assert {(f.operation.function, f.missing): (f.label, f.evidence) for f in adjudicated.findings} == before
|
||||||
|
assert client.calls and all(call["model"] == "test-model" for call in client.calls)
|
||||||
|
prompt = json.dumps(client.calls[0]["messages"])
|
||||||
|
assert "meant to be public" in prompt and "app.py::constrained" in prompt # the question names the route and its siblings
|
||||||
|
assert not any(d["reason"] == "intent_adjudication_unavailable" for d in adjudicated.degradations)
|
||||||
|
|
||||||
|
|
||||||
|
def test_without_a_client_one_degradation_is_recorded_and_findings_are_identical() -> None:
|
||||||
|
from openultrasast.semantic.obligations.intent import adjudicate_intent
|
||||||
|
|
||||||
|
result = _check()
|
||||||
|
adjudicated = adjudicate_intent(result, client=None, model="", texts={})
|
||||||
|
assert [f.intent for f in adjudicated.findings] == [None] * len(result.findings)
|
||||||
|
assert [(f.operation.function, f.missing, f.label) for f in adjudicated.findings] == [
|
||||||
|
(f.operation.function, f.missing, f.label) for f in result.findings
|
||||||
|
]
|
||||||
|
assert sum(1 for d in adjudicated.degradations if d["reason"] == "intent_adjudication_unavailable") == 1
|
||||||
|
|
||||||
|
|
||||||
|
def test_malformed_or_off_vocabulary_answers_are_recorded_as_unknown() -> None:
|
||||||
|
from openultrasast.semantic.obligations.intent import adjudicate_intent
|
||||||
|
|
||||||
|
result = _check()
|
||||||
|
client = ScriptedChatClient([ChatResponse(content="sure, looks fine"), ChatResponse(content='{"intent": "banana"}')] * 10)
|
||||||
|
adjudicated = adjudicate_intent(result, client=client, model="m", texts={})
|
||||||
|
assert all(f.intent == "unknown" for f in adjudicated.findings if f.route)
|
||||||
|
assert all(f.evidence == g.evidence and f.label == g.label for f, g in zip(adjudicated.findings, result.findings, strict=True))
|
||||||
|
|
||||||
|
|
||||||
|
def test_scan_records_intent_when_a_scripted_hunter_client_is_configured(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None:
|
||||||
|
from test_obligation_scan import APP, _repo, _run_dir
|
||||||
|
|
||||||
|
from openultrasast.cli import main
|
||||||
|
|
||||||
|
repo = _repo(tmp_path, monkeypatch)
|
||||||
|
(repo / "app.py").write_text(APP)
|
||||||
|
(repo / "openultrasast.toml").write_text('[obligations]\nmin_siblings = 2\n\n[models]\nhunter = "scripted-model"\n')
|
||||||
|
monkeypatch.setenv("OPENULTRASAST_HUNTER_CLIENT", "scripted")
|
||||||
|
assert main(["scan", str(repo), "--mode", "standard", "--config", str(repo / "openultrasast.toml")]) == 0
|
||||||
|
sarif = json.loads((_run_dir(repo) / "report.sarif").read_text())
|
||||||
|
props = [r["properties"] for r in sarif["runs"][0]["results"] if r["properties"]["finding_id"].startswith("obligation:")]
|
||||||
|
assert props and all(p["evidence_level"] == "suspicion" for p in props)
|
||||||
|
assert all("obligation_intent" in p for p in props)
|
||||||
|
|
||||||
|
|
||||||
|
def test_the_prompt_carries_the_handler_own_lines_only_and_is_redacted() -> None:
|
||||||
|
"""Round-3 findings: redacted handler text (design Security Considerations) and nothing beyond the handler's lines."""
|
||||||
|
from test_obligation_check import APP
|
||||||
|
|
||||||
|
from openultrasast.semantic.obligations.intent import adjudicate_intent
|
||||||
|
|
||||||
|
result = _check()
|
||||||
|
client = ScriptedChatClient([ChatResponse(content='{"intent": "protected", "rationale": "r"}')] * 10)
|
||||||
|
adjudicate_intent(result, client=client, model="m", texts={"app.py": APP})
|
||||||
|
leaky_prompt = next(json.dumps(call["messages"]) for call in client.calls if "def leaky" in json.dumps(call["messages"]))
|
||||||
|
assert "@app.route('/books/any/<title>')" in leaky_prompt
|
||||||
|
assert all(other not in leaky_prompt for other in ("def guarded", "def constrained", "def health"))
|
||||||
|
key = "sk-" + "x" * 20 # assembled at runtime: never a contiguous key-shaped literal in a test
|
||||||
|
leaky_with_secret = APP.replace("def leaky(title):\n", f"def leaky(title):\n api_key = '{key}'\n")
|
||||||
|
client = ScriptedChatClient([ChatResponse(content='{"intent": "protected", "rationale": "r"}')] * 10)
|
||||||
|
adjudicate_intent(result, client=client, model="m", texts={"app.py": leaky_with_secret})
|
||||||
|
prompts = json.dumps([call["messages"] for call in client.calls])
|
||||||
|
assert "***REDACTED***" in prompts and key not in prompts
|
||||||
Reference in New Issue
Block a user