mirror of
https://github.com/norandom/OpenUltraSAST.git
synced 2026-10-16 03:49:26 +00:00
Marius CiepluchandClaude Fable 5.1
3866e2771b
The agent passes a manifest file or project directory; pip uses -r for requirements files and the directory for pyproject/setup, npm ci/install runs in the directory, composer/maven/gradle likewise. Build and verify failures carry phase, exit code and a sanitised stderr tail. Co-Authored-By: Claude Fable 5.1 <[email protected]>
126 lines
4.0 KiB
Python
126 lines
4.0 KiB
Python
import json
|
|
|
|
import pytest
|
|
|
|
from openultrasast.search.demo import load_demo, validate_demo
|
|
|
|
|
|
def document():
|
|
return {
|
|
"oracle": "path",
|
|
"build": {"recipe": "none", "arguments": []},
|
|
"start": {"runtime": "python", "path": "app.py", "arguments": [], "mode": "cli"},
|
|
"steps": [{"type": "cli", "arguments": ["hello"]}],
|
|
}
|
|
|
|
|
|
def test_valid_and_roundtrip(tmp_path):
|
|
value = document()
|
|
(tmp_path / "demo.json").write_text(json.dumps(value))
|
|
assert load_demo(tmp_path) == value
|
|
|
|
|
|
@pytest.mark.parametrize("path", ["../app.py", "/fixture/canary", "/etc/passwd", "a/../../app.py", "-c", "a\\b"])
|
|
def test_checkout_paths_only(path):
|
|
value = document()
|
|
value["start"]["path"] = path
|
|
with pytest.raises(ValueError):
|
|
validate_demo(value)
|
|
|
|
|
|
@pytest.mark.parametrize("argument", ["$PROOF_MARKER", "${CANARY}", "/fixture/canary", "/proc/1/environ", "${later}", "x\x00y"])
|
|
def test_private_references_rejected(argument):
|
|
value = document()
|
|
value["steps"][0]["arguments"] = [argument]
|
|
with pytest.raises(ValueError):
|
|
validate_demo(value)
|
|
|
|
|
|
def test_sequential_capture():
|
|
value = document()
|
|
value["steps"][0]["capture"] = {"name": "token", "source": "json", "path": ["token"]}
|
|
value["steps"].append({"type": "cli", "arguments": ["${token}"]})
|
|
assert validate_demo(value) == value
|
|
value["steps"][0]["arguments"] = ["${token}"]
|
|
with pytest.raises(ValueError):
|
|
validate_demo(value)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"mutate",
|
|
[
|
|
lambda d: d.update(env={}),
|
|
lambda d: d["build"].update(recipe="shell"),
|
|
lambda d: d["build"].update(arguments=["--index-url=evil"]),
|
|
lambda d: d["start"].update(runtime="sh"),
|
|
lambda d: d.update(steps=d["steps"] * 33),
|
|
],
|
|
)
|
|
def test_unknown_keys_recipes_and_limits(mutate):
|
|
value = document()
|
|
mutate(value)
|
|
with pytest.raises(ValueError):
|
|
validate_demo(value)
|
|
|
|
|
|
@pytest.mark.parametrize("recipe", ["npm", "composer", "gradle"])
|
|
def test_directory_build_recipe_accepts_checkout_root(recipe):
|
|
value = document()
|
|
value["build"] = {"recipe": recipe, "arguments": ["."]}
|
|
assert validate_demo(value) == value
|
|
|
|
|
|
@pytest.mark.parametrize("recipe", ["pip", "maven"])
|
|
def test_manifest_recipe_accepts_project_directory(recipe):
|
|
value = document()
|
|
value["build"] = {"recipe": recipe, "arguments": ["."]}
|
|
assert validate_demo(value) == value
|
|
|
|
|
|
@pytest.mark.parametrize("oracle", [None, "injection", "unknown", [], 1])
|
|
def test_oracle_required_and_known(oracle):
|
|
value = document()
|
|
if oracle is None:
|
|
value.pop("oracle")
|
|
else:
|
|
value["oracle"] = oracle
|
|
with pytest.raises(ValueError):
|
|
validate_demo(value)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"family,allowed",
|
|
[
|
|
("injection", ("sql", "command")),
|
|
("path", ("path",)),
|
|
("output_encoding", ("xss",)),
|
|
("untrusted_destination", ("ssrf",)),
|
|
("deserialization", ()),
|
|
("access_control", ()),
|
|
("config_secrets", ()),
|
|
],
|
|
)
|
|
def test_family_oracle_contract(family, allowed):
|
|
from openultrasast.search.demo import FAMILY_ORACLES, validate_oracle
|
|
|
|
assert FAMILY_ORACLES[family] == allowed
|
|
for oracle in ("sql", "command", "path", "xss", "ssrf"):
|
|
if oracle in allowed:
|
|
validate_oracle(family, oracle)
|
|
else:
|
|
with pytest.raises(ValueError, match="not allowed"):
|
|
validate_oracle(family, oracle)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"recipe,manifest",
|
|
[("pip", "package.json"), ("npm", "requirements.txt"), ("composer", "pom.xml"), ("maven", "composer.json"), ("gradle", "setup.py")],
|
|
)
|
|
@pytest.mark.parametrize("directory", [False, True])
|
|
def test_recipe_rejects_other_ecosystem(tmp_path, recipe, manifest, directory):
|
|
(tmp_path / manifest).write_text("{}")
|
|
value = document()
|
|
value["build"] = {"recipe": recipe, "arguments": ["." if directory else manifest]}
|
|
with pytest.raises(ValueError, match="ecosystem"):
|
|
validate_demo(value, checkout=tmp_path)
|