feat(ttp): E.3.8 R0001-R0030 command cohort
30 YAMLs for the shell/command rule cohort per Appendix B (rules/ttp/). Splits into engine-active (R0007-R0029, regex on command_text / raw_url / user_agent) and lifter-bound (R0001-R0006, R0030 — the v0 RuleEngine cannot count auth attempts, do identity rollups, or parse fingerprint blobs; the BehavioralLifter / IdentityLifter / CredentialLifter consume them by rule_id at E.3.9 / E.3.13). test_command_rules.py asserts: - every R000N has a YAML that compiles - lifter-bound rules NEVER fire from the v0 engine (regression guard against a YAML drifting into a regex match.spec) - engine-active rules meet their Appendix-C precision target against the seed corpus (≥0.95 high-conf, ≥0.80 medium) Conftest fixes: precision_engine moved to module-scope so module- scope precomputed dispatch fixture (fired_by_label) can request it; _RULES_DIR path bumped from parents[2] to parents[3] so the loader resolves the project root regardless of pytest cwd; make_event synthesizes attacker_uuid so TTPTag's anchor invariant is satisfied. Seed corpus broadened: positive examples for every regex rule plus 6 negative examples across innocuous shell verbs (ls, echo, cd, ps, df, free) so FPs surface in precision rather than passing vacuously.
This commit is contained in:
145
tests/ttp/rule_precision/test_command_rules.py
Normal file
145
tests/ttp/rule_precision/test_command_rules.py
Normal file
@@ -0,0 +1,145 @@
|
||||
"""Per-rule precision asserts for the command cohort (R0001-R0030).
|
||||
|
||||
Drives the labelled corpus through a real :class:`RuleEngine` populated
|
||||
from ``./rules/ttp/`` and asserts each rule meets its Appendix-C
|
||||
precision target.
|
||||
|
||||
Live vs xfail per rule:
|
||||
|
||||
* R0001-R0006 / R0030: lifter-bound (auth-attempt aggregation, identity
|
||||
rollups, fingerprint blob parsing). v0 :class:`RuleEngine` only does
|
||||
regex-on-payload-field, so these can never fire from the engine
|
||||
alone. Their precision tests are :pyfunc:`pytest.xfail` until the
|
||||
matching lifter ships (E.3.9 / E.3.13).
|
||||
* R0007-R0029: regex-driven on ``command_text`` / ``raw_url`` / ``user_agent``.
|
||||
Live precision asserts against the seed corpus (committed) and any
|
||||
operator-built ``commands.jsonl`` (gitignored, preferred).
|
||||
|
||||
Precision target per Appendix C: ≥0.95 for high-conf rules
|
||||
(base ``confidence >= 0.85``), ≥0.80 for medium (0.6-0.85). The
|
||||
fixture's :func:`precision_for` returns 1.0 vacuously when no rows
|
||||
fired the rule — :func:`pytest.skip` covers that case so a sparse
|
||||
corpus skips loudly rather than silently passing.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
from collections.abc import Callable
|
||||
|
||||
import pytest
|
||||
|
||||
from decnet.ttp.impl.rule_engine import RuleEngine
|
||||
from tests.ttp.rule_precision.conftest import (
|
||||
CorpusRow,
|
||||
make_event,
|
||||
precision_for,
|
||||
)
|
||||
|
||||
CohortLoader = Callable[[str], list[CorpusRow]]
|
||||
|
||||
# Lifter-bound rules: cannot fire from the v0 engine.
|
||||
_LIFTER_BOUND: dict[str, str] = {
|
||||
"R0001": "impl phase E.3.9 (BehavioralLifter — auth brute count)",
|
||||
"R0002": "impl phase E.3.9 (BehavioralLifter — password guessing)",
|
||||
"R0003": "impl phase E.3.13 (IdentityLifter — password spraying)",
|
||||
"R0004": "impl phase E.3.13 (CredentialLifter — credential reuse)",
|
||||
"R0005": "impl phase E.3.9 (BehavioralLifter — valid account use)",
|
||||
"R0006": "impl phase E.3.9 (BehavioralLifter — default creds)",
|
||||
"R0030": "impl phase E.3.9 (BehavioralLifter — JARM/HASSH match)",
|
||||
}
|
||||
|
||||
# Per-rule precision floor. Anything ≥0.85 base confidence in the YAML
|
||||
# is "high"; 0.6-0.85 is "medium". Sub-0.6 is not shipped in v0.
|
||||
_PRECISION_TARGET: dict[str, float] = {
|
||||
"R0007": 0.95, "R0008": 0.95, "R0009": 0.95, "R0010": 0.95,
|
||||
"R0011": 0.80, "R0012": 0.95, "R0013": 0.95, "R0014": 0.95,
|
||||
"R0015": 0.95, "R0016": 0.80, "R0017": 0.95, "R0018": 0.80,
|
||||
"R0019": 0.80, "R0020": 0.80, "R0021": 0.80, "R0022": 0.95,
|
||||
"R0023": 0.95, "R0024": 0.95, "R0025": 0.95, "R0026": 0.95,
|
||||
"R0027": 0.95, "R0028": 0.95, "R0029": 0.80,
|
||||
}
|
||||
|
||||
_ALL_RULE_IDS = [f"R{n:04d}" for n in range(1, 31)]
|
||||
|
||||
|
||||
@pytest.fixture(scope="module")
|
||||
def fired_by_label(
|
||||
precision_engine: RuleEngine,
|
||||
corpus_loader: CohortLoader,
|
||||
) -> tuple[dict[str, list[str]], list[CorpusRow]]:
|
||||
"""Pre-evaluate the corpus once per module.
|
||||
|
||||
Returns ``(label → [rule_ids that fired], rows)``. Each rule's
|
||||
test then walks the same dict — saves 30× re-evaluation.
|
||||
"""
|
||||
rows = corpus_loader("commands")
|
||||
fired: dict[str, list[str]] = {}
|
||||
import asyncio
|
||||
|
||||
async def _drive() -> None:
|
||||
for row in rows:
|
||||
tags = await precision_engine.evaluate(make_event(row, source_id=row.label))
|
||||
fired[row.label] = sorted({tag.rule_id for tag in tags})
|
||||
|
||||
asyncio.run(_drive())
|
||||
return fired, rows
|
||||
|
||||
|
||||
@pytest.mark.parametrize("rule_id", _ALL_RULE_IDS)
|
||||
def test_rule_yaml_present(rule_id: str) -> None:
|
||||
"""Every R000N rule_id has a YAML on disk that compiles.
|
||||
|
||||
Catches a missing or malformed file faster than the precision
|
||||
test would (the latter would just see zero matches).
|
||||
"""
|
||||
from pathlib import Path
|
||||
|
||||
from decnet.ttp.store.base import RuleState
|
||||
from decnet.ttp.store.impl.filesystem import _parse_and_compile
|
||||
|
||||
path = Path("rules/ttp") / f"{rule_id}.yaml"
|
||||
assert path.exists(), f"missing YAML: {path}"
|
||||
compiled = _parse_and_compile(path, RuleState())
|
||||
assert compiled.rule_id == rule_id
|
||||
|
||||
|
||||
@pytest.mark.parametrize("rule_id", list(_LIFTER_BOUND))
|
||||
def test_lifter_bound_rule_inert_in_v0(
|
||||
rule_id: str,
|
||||
fired_by_label: tuple[dict[str, list[str]], list[CorpusRow]],
|
||||
) -> None:
|
||||
"""Lifter-bound rules MUST NOT fire from the v0 engine.
|
||||
|
||||
They're carried in ``./rules/ttp/`` so the catalogue surfaces
|
||||
them and the lifter can read them by rule_id, but the regex
|
||||
engine can't interpret a ``match.kind: lifter:*`` spec — it
|
||||
falls into the ``pattern is None`` branch and silently skips.
|
||||
A regression that lit one of these up from regex would mean a
|
||||
YAML drifted into a ``pattern:`` form and we'd be emitting
|
||||
half-baked tags.
|
||||
"""
|
||||
fired, _rows = fired_by_label
|
||||
matches = [label for label, ids in fired.items() if rule_id in ids]
|
||||
assert matches == [], (
|
||||
f"{rule_id} is lifter-bound but fired on: {matches}"
|
||||
)
|
||||
|
||||
|
||||
@pytest.mark.parametrize("rule_id", list(_PRECISION_TARGET))
|
||||
def test_command_rule_precision(
|
||||
rule_id: str,
|
||||
fired_by_label: tuple[dict[str, list[str]], list[CorpusRow]],
|
||||
) -> None:
|
||||
"""Each live regex rule meets its Appendix-C precision target."""
|
||||
fired, rows = fired_by_label
|
||||
matched = sum(1 for ids in fired.values() if rule_id in ids)
|
||||
if matched == 0:
|
||||
pytest.skip(
|
||||
f"{rule_id}: no corpus rows matched — extend "
|
||||
"tests/ttp/rule_precision/corpus/seed_commands.jsonl",
|
||||
)
|
||||
target = _PRECISION_TARGET[rule_id]
|
||||
precision, tp, fp = precision_for(rule_id, rows, fired)
|
||||
assert precision >= target, (
|
||||
f"{rule_id} precision {precision:.2f} < target {target:.2f} "
|
||||
f"(tp={tp} fp={fp})"
|
||||
)
|
||||
Reference in New Issue
Block a user