mirror of
https://github.com/CyberMind-FR/secubox-deb.git
synced 2026-07-28 21:17:36 +00:00
Merge pull request #880 from CyberMind-FR/feat/waf-generator-dedup
Some checks are pending
License Headers / check (push) Waiting to run
Some checks are pending
License Headers / check (push) Waiting to run
feat(cve-triage): WAF generator cross-category dedup (1.1.1)
This commit is contained in:
commit
84be96ce04
|
|
@ -0,0 +1,311 @@
|
|||
# WAF generator — dédup inter-catégories — Plan d'implémentation
|
||||
|
||||
> **For agentic workers:** REQUIRED SUB-SKILL: Use superpowers:subagent-driven-development. Steps use checkbox (`- [ ]`) syntax.
|
||||
|
||||
**Goal:** The product-absent generator must not emit a probe pattern that already exists in another WAF category (e.g. the hand-written `cve_2024` blocking category), because a blocking category wins and the generated `detect` rule would never log — pure redundancy. Skip such candidates, recording an explicit rejection reason.
|
||||
|
||||
**Architecture:** A new pure `emit.existing_patterns(rules_path)` returns `{pattern_string: category_key}` for every category EXCEPT `product_absent_probes` (excluding our own key preserves idempotence). `generate()` gains an optional `existing` map and, after a candidate passes the presence barriers, rejects it if `re.escape(path)` is already covered. CLI + both panel routes build the map from the live rules file and pass it.
|
||||
|
||||
**Tech Stack:** Python 3.11 stdlib (`json`, `re`) + existing wafgen modules.
|
||||
|
||||
## Global Constraints
|
||||
|
||||
- Python 3.11 stdlib only. SPDX 4-line header preserved on every file. Copyright `Gérald Kerma <devel@cybermind.fr>`.
|
||||
- Backward compatible: `generate()`'s new param is keyword-only with a default that means "no dedup" — existing 3-arg callers and tests keep working unchanged.
|
||||
- Dedup is **best-effort**, never a safety gate: a missing/corrupt rules file → empty map → no dedup (worst case a duplicate, which is today's behaviour). This is NOT `write_category`'s fail-closed path.
|
||||
- **Exclude `emit.CATEGORY_KEY` (`product_absent_probes`) from the existing map.** Including it would make the 2nd `--apply` see our own prior patterns as "already covered" and regenerate nothing — breaking idempotence. This is the one invariant a reviewer must confirm.
|
||||
- Dedup runs LAST (kev → extract → presence barriers → dedup): only a candidate that would otherwise be kept is dedup-checked.
|
||||
- Pattern comparison uses `re.escape(candidate.path)` — the exact form `build_category` stores — matched by EXACT string equality against stored pattern strings (no regex-subset guessing, which could falsely claim coverage).
|
||||
- Detect-only / additive / atomic invariants of `write_category` are untouched. No escalate/block anywhere.
|
||||
- Tests per-directory: `.venv/bin/python -m pytest packages/secubox-cve-triage/tests -q`. Commit messages end `Co-Authored-By: Gerald KERMA <devel@cybermind.fr>`, NO Claude reference.
|
||||
|
||||
---
|
||||
|
||||
### Task 1: `emit.existing_patterns()` + `generate()` dédup (purs, testés)
|
||||
|
||||
**Files:**
|
||||
- Modify: `packages/secubox-cve-triage/api/wafgen/emit.py` (add `existing_patterns`)
|
||||
- Modify: `packages/secubox-cve-triage/api/wafgen/generate.py` (add `existing` param + dedup)
|
||||
- Test: `packages/secubox-cve-triage/tests/test_emit.py` (add existing_patterns tests)
|
||||
- Test: `packages/secubox-cve-triage/tests/test_generate.py` (add dedup tests)
|
||||
|
||||
**Interfaces:**
|
||||
- Produces: `emit.existing_patterns(rules_path: Path) -> dict[str, str]`; `generate(subset_dir, present, present_complete, *, existing: dict[str, str] | None = None) -> tuple[list[Candidate], list[tuple[str, str]]]`.
|
||||
|
||||
- [ ] **Step 1: Write failing tests for `existing_patterns`**
|
||||
|
||||
Add to `packages/secubox-cve-triage/tests/test_emit.py`:
|
||||
|
||||
```python
|
||||
def test_existing_patterns_maps_pattern_to_category_excluding_own_key(tmp_path):
|
||||
from api.wafgen.emit import existing_patterns
|
||||
rules = tmp_path / "waf-rules.json"
|
||||
rules.write_text(json.dumps({"categories": {
|
||||
"cve_2024": {"patterns": [{"pattern": "/mgmt/tm/util/bash"},
|
||||
{"pattern": "/api/v1/totp/user\\-backup"}]},
|
||||
"product_absent_probes": {"patterns": [{"pattern": "/should/not/appear"}]},
|
||||
}}))
|
||||
got = existing_patterns(rules)
|
||||
assert got["/mgmt/tm/util/bash"] == "cve_2024"
|
||||
assert got["/api/v1/totp/user\\-backup"] == "cve_2024"
|
||||
assert "/should/not/appear" not in got # our own key excluded → idempotence
|
||||
|
||||
|
||||
def test_existing_patterns_missing_or_corrupt_file_is_empty(tmp_path):
|
||||
from api.wafgen.emit import existing_patterns
|
||||
assert existing_patterns(tmp_path / "nope.json") == {}
|
||||
bad = tmp_path / "bad.json"
|
||||
bad.write_text("{ not json")
|
||||
assert existing_patterns(bad) == {}
|
||||
```
|
||||
|
||||
- [ ] **Step 2: Run — must fail** (`ImportError`/`AttributeError` for `existing_patterns`).
|
||||
|
||||
Run: `.venv/bin/python -m pytest packages/secubox-cve-triage/tests/test_emit.py -q`
|
||||
|
||||
- [ ] **Step 3: Implement `existing_patterns` in `emit.py`**
|
||||
|
||||
Add after `CATEGORY_KEY` (uses the already-imported `json`, `Path`):
|
||||
|
||||
```python
|
||||
def existing_patterns(rules_path: Path) -> dict[str, str]:
|
||||
"""Map chaque pattern-string -> sa catégorie, SAUF notre propre
|
||||
CATEGORY_KEY. Exclure product_absent_probes est vital : sinon un 2e --apply
|
||||
verrait nos propres patterns comme « déjà couverts » et régénérerait zéro
|
||||
(idempotence cassée). Best-effort : fichier absent/corrompu -> {} (aucune
|
||||
dédup, jamais une erreur — la dédup n'est pas un garde de sûreté, contrairement
|
||||
à write_category qui, lui, lève)."""
|
||||
try:
|
||||
data = json.loads(Path(rules_path).read_text(encoding="utf-8"))
|
||||
except (OSError, json.JSONDecodeError):
|
||||
return {}
|
||||
cats = data.get("categories", {})
|
||||
if not isinstance(cats, dict):
|
||||
return {}
|
||||
out: dict[str, str] = {}
|
||||
for cat_key, cat in cats.items():
|
||||
if cat_key == CATEGORY_KEY or not isinstance(cat, dict):
|
||||
continue
|
||||
for pat in cat.get("patterns", []):
|
||||
if isinstance(pat, dict) and isinstance(pat.get("pattern"), str):
|
||||
out.setdefault(pat["pattern"], cat_key)
|
||||
return out
|
||||
```
|
||||
|
||||
- [ ] **Step 4: Run — must pass.**
|
||||
|
||||
- [ ] **Step 5: Write failing tests for `generate` dedup**
|
||||
|
||||
Add to `packages/secubox-cve-triage/tests/test_generate.py` (reuses the existing `_write` helper + `generate` import):
|
||||
|
||||
```python
|
||||
import re
|
||||
|
||||
|
||||
def test_candidate_already_covered_by_another_category_is_rejected(tmp_path):
|
||||
subset = tmp_path / "nuclei-subset"
|
||||
subset.mkdir()
|
||||
_write(subset, "ivanti.yaml", """
|
||||
id: CVE-2024-21887
|
||||
info:
|
||||
severity: critical
|
||||
classification: {cve-id: CVE-2024-21887, cpe: "cpe:2.3:a:ivanti:connect_secure:*"}
|
||||
metadata: {vendor: ivanti, product: connect_secure}
|
||||
tags: cve,kev
|
||||
http:
|
||||
- method: GET
|
||||
path: ["{{BaseURL}}/api/v1/totp/user-backup"]
|
||||
""")
|
||||
# cve_2024 already covers this exact path (stored re.escape'd)
|
||||
existing = {re.escape("/api/v1/totp/user-backup"): "cve_2024"}
|
||||
kept, rejected = generate(subset, present=set(), present_complete=True,
|
||||
existing=existing)
|
||||
assert kept == []
|
||||
assert any("already covered by cve_2024" in reason for _, reason in rejected)
|
||||
|
||||
|
||||
def test_no_dedup_when_existing_absent_keeps_candidate(tmp_path):
|
||||
subset = tmp_path / "nuclei-subset"
|
||||
subset.mkdir()
|
||||
_write(subset, "ivanti.yaml", """
|
||||
id: CVE-2024-21887
|
||||
info:
|
||||
severity: critical
|
||||
classification: {cve-id: CVE-2024-21887, cpe: "cpe:2.3:a:ivanti:connect_secure:*"}
|
||||
metadata: {vendor: ivanti, product: connect_secure}
|
||||
tags: cve,kev
|
||||
http:
|
||||
- method: GET
|
||||
path: ["{{BaseURL}}/api/v1/totp/user-backup"]
|
||||
""")
|
||||
kept, _ = generate(subset, present=set(), present_complete=True) # existing default None
|
||||
assert [c.cve for c in kept] == ["CVE-2024-21887"]
|
||||
```
|
||||
|
||||
- [ ] **Step 6: Run — must fail** (dedup not implemented; first test keeps the candidate).
|
||||
|
||||
- [ ] **Step 7: Implement dedup in `generate.py`**
|
||||
|
||||
Add `import re` at the top (after `from pathlib import Path`). Change the signature and add the dedup check as the LAST step before `kept.append`:
|
||||
|
||||
```python
|
||||
def generate(subset_dir: Path, present: set[str], present_complete: bool,
|
||||
*, existing: dict[str, str] | None = None
|
||||
) -> tuple[list[Candidate], list[tuple[str, str]]]:
|
||||
"""(retenus, [(fichier, raison de rejet)]). `existing` mappe pattern-string
|
||||
-> catégorie (voir emit.existing_patterns) : un candidat dont le pattern y
|
||||
figure déjà est rejeté (redondance ; une catégorie bloquante l'emporterait
|
||||
sur notre detect). None/{} => aucune dédup."""
|
||||
existing = existing or {}
|
||||
kept: list[Candidate] = []
|
||||
rejected: list[tuple[str, str]] = []
|
||||
# ... (boucle inchangée jusqu'après should_generate) ...
|
||||
```
|
||||
|
||||
Right before `kept.append(result)`, insert:
|
||||
|
||||
```python
|
||||
pattern = re.escape(result.path)
|
||||
if pattern in existing:
|
||||
rejected.append((p.name, f"already covered by {existing[pattern]}"))
|
||||
continue
|
||||
|
||||
kept.append(result)
|
||||
```
|
||||
|
||||
- [ ] **Step 8: Run — must pass.** Then the full module suite:
|
||||
|
||||
Run: `.venv/bin/python -m pytest packages/secubox-cve-triage/tests -q`
|
||||
Expected: all prior + 4 new pass.
|
||||
|
||||
- [ ] **Step 9: Commit**
|
||||
|
||||
```bash
|
||||
git add packages/secubox-cve-triage/api/wafgen/emit.py packages/secubox-cve-triage/api/wafgen/generate.py packages/secubox-cve-triage/tests/test_emit.py packages/secubox-cve-triage/tests/test_generate.py
|
||||
git commit -m "feat(cve-triage): skip generated probes already covered by another WAF category
|
||||
|
||||
Co-Authored-By: Gerald KERMA <devel@cybermind.fr>"
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
### Task 2: câbler CLI + panel pour passer `existing`
|
||||
|
||||
**Files:**
|
||||
- Modify: `packages/secubox-cve-triage/api/wafgen/cli.py`
|
||||
- Modify: `packages/secubox-cve-triage/api/main.py` (both `/waf-rules` routes)
|
||||
- Modify: `packages/secubox-cve-triage/debian/changelog`
|
||||
- Test: `packages/secubox-cve-triage/tests/test_cli.py` (extend if present, else create)
|
||||
|
||||
**Interfaces:**
|
||||
- Consumes: `emit.existing_patterns`, `generate(..., existing=...)`.
|
||||
|
||||
- [ ] **Step 1: Write the failing CLI test**
|
||||
|
||||
Add to `packages/secubox-cve-triage/tests/test_cli.py` (create with SPDX header if absent). Drives `wafgen.cli.main` with a tmp subset + a tmp rules file whose foreign category already covers the template's path, asserting dry-run rejects it as already-covered:
|
||||
|
||||
```python
|
||||
def test_cli_dedups_against_existing_rules(tmp_path, monkeypatch, capsys):
|
||||
import re
|
||||
from api.wafgen import cli
|
||||
subset = tmp_path / "subset"
|
||||
subset.mkdir()
|
||||
(subset / "ivanti.yaml").write_text(
|
||||
'id: CVE-2024-21887\n'
|
||||
'info:\n severity: critical\n'
|
||||
' classification: {cve-id: CVE-2024-21887, cpe: "cpe:2.3:a:ivanti:connect_secure:*"}\n'
|
||||
' metadata: {vendor: ivanti, product: connect_secure}\n'
|
||||
' tags: cve,kev\n'
|
||||
'http:\n - method: GET\n path: ["{{BaseURL}}/api/v1/totp/user-backup"]\n')
|
||||
rules = tmp_path / "waf-rules.json"
|
||||
rules.write_text('{"categories": {"cve_2024": {"patterns": [{"pattern": "'
|
||||
+ re.escape("/api/v1/totp/user-backup") + '"}]}}}')
|
||||
monkeypatch.setattr(cli, "gather_present", lambda: (set(), True))
|
||||
|
||||
rc = cli.main(["waf-rules", "generate",
|
||||
"--subset", str(subset), "--rules", str(rules)]) # dry-run
|
||||
out = capsys.readouterr().out
|
||||
assert rc == 0
|
||||
assert "kept: 0 probe(s)" in out
|
||||
assert "already covered by cve_2024" in out
|
||||
```
|
||||
|
||||
- [ ] **Step 2: Run — must fail** (CLI doesn't dedup yet → kept 1).
|
||||
|
||||
Run: `.venv/bin/python -m pytest packages/secubox-cve-triage/tests/test_cli.py -q`
|
||||
|
||||
- [ ] **Step 3: Wire `wafgen/cli.py`**
|
||||
|
||||
Add the import: `from .emit import existing_patterns, write_category` (replace the existing `from .emit import write_category`). Change the generate call in `main()`:
|
||||
|
||||
```python
|
||||
present, complete = gather_present()
|
||||
kept, rejected = generate(Path(args.subset), present, complete,
|
||||
existing=existing_patterns(Path(args.rules)))
|
||||
```
|
||||
|
||||
- [ ] **Step 4: Wire both panel routes in `main.py`**
|
||||
|
||||
In `waf_rules_preview()`: import and pass existing:
|
||||
|
||||
```python
|
||||
from .wafgen.generate import generate
|
||||
from .wafgen.inventory import gather_present
|
||||
from .wafgen.emit import existing_patterns
|
||||
|
||||
present, complete = gather_present()
|
||||
kept, rejected = generate(
|
||||
Path("/usr/lib/secubox/cve-triage/nuclei-subset"), present, complete,
|
||||
existing=existing_patterns(Path("/etc/secubox/waf/waf-rules.json")),
|
||||
)
|
||||
```
|
||||
|
||||
In `waf_rules_generate()`: add `existing_patterns` to the emit import already there and pass it:
|
||||
|
||||
```python
|
||||
from .wafgen.emit import existing_patterns, write_category
|
||||
...
|
||||
kept, rejected = generate(
|
||||
Path("/usr/lib/secubox/cve-triage/nuclei-subset"), present, complete,
|
||||
existing=existing_patterns(Path("/etc/secubox/waf/waf-rules.json")),
|
||||
)
|
||||
```
|
||||
|
||||
- [ ] **Step 5: Run — must pass; then full suite**
|
||||
|
||||
Run: `.venv/bin/python -m pytest packages/secubox-cve-triage/tests -q`
|
||||
Also `python3 -c "import ast; ast.parse(open('packages/secubox-cve-triage/api/main.py').read())"` to confirm main.py parses.
|
||||
|
||||
- [ ] **Step 6: Bump changelog + commit**
|
||||
|
||||
Prepend to `packages/secubox-cve-triage/debian/changelog`:
|
||||
|
||||
```
|
||||
secubox-cve-triage (1.1.1-1~bookworm1) bookworm; urgency=medium
|
||||
|
||||
* WAF generator: skip probes already covered by another WAF category
|
||||
(e.g. the hand-written cve_2024 blocking rules) — a blocking category wins,
|
||||
so a duplicate detect rule would never log. Rejected with an explicit
|
||||
"already covered by <category>" reason, visible in dry-run. Our own
|
||||
product_absent_probes category is excluded from the check (idempotence).
|
||||
|
||||
-- Gerald KERMA <devel@cybermind.fr> Fri, 18 Jul 2026 14:00:00 +0200
|
||||
|
||||
```
|
||||
|
||||
```bash
|
||||
git add packages/secubox-cve-triage/api/wafgen/cli.py packages/secubox-cve-triage/api/main.py packages/secubox-cve-triage/debian/changelog packages/secubox-cve-triage/tests/test_cli.py
|
||||
git commit -m "feat(cve-triage): CLI + panel pass existing patterns for cross-category dedup (1.1.1)
|
||||
|
||||
Co-Authored-By: Gerald KERMA <devel@cybermind.fr>"
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
## Auto-revue du plan
|
||||
|
||||
- **Couverture** : dédup pur (Task 1: emit.existing_patterns + generate.existing) ; câblage CLI + 2 routes panel + changelog (Task 2). ✓
|
||||
- **Idempotence** : CATEGORY_KEY exclu de existing_patterns — testé (`test_existing_patterns_..._excluding_own_key`).
|
||||
- **Backward-compat** : `existing` keyword-only default None ; test `test_no_dedup_when_existing_absent_keeps_candidate` + tous les appels 3-args existants inchangés.
|
||||
- **Sûreté** : best-effort ({} sur fichier absent/corrompu), jamais un garde ; detect-only/additif/atomique de write_category intacts ; aucun escalate/block.
|
||||
- **Types** : `existing_patterns -> dict[str,str]` consommé par `generate(existing=...)` identiquement en CLI et panel.
|
||||
|
|
@ -1105,10 +1105,12 @@ def waf_rules_preview():
|
|||
"""Dry-run: what would be generated (kept + rejections). Writes nothing."""
|
||||
from .wafgen.generate import generate
|
||||
from .wafgen.inventory import gather_present
|
||||
from .wafgen.emit import existing_patterns
|
||||
|
||||
present, complete = gather_present()
|
||||
kept, rejected = generate(
|
||||
Path("/usr/lib/secubox/cve-triage/nuclei-subset"), present, complete
|
||||
Path("/usr/lib/secubox/cve-triage/nuclei-subset"), present, complete,
|
||||
existing=existing_patterns(Path("/etc/secubox/waf/waf-rules.json")),
|
||||
)
|
||||
return {
|
||||
"present_count": len(present),
|
||||
|
|
@ -1127,7 +1129,7 @@ def waf_rules_generate():
|
|||
presence inventory is incomplete (fail-safe)."""
|
||||
from .wafgen.generate import generate
|
||||
from .wafgen.inventory import gather_present
|
||||
from .wafgen.emit import write_category
|
||||
from .wafgen.emit import existing_patterns, write_category
|
||||
|
||||
present, complete = gather_present()
|
||||
if not complete:
|
||||
|
|
@ -1136,7 +1138,8 @@ def waf_rules_generate():
|
|||
detail="presence inventory incomplete — refusing (fail-safe)",
|
||||
)
|
||||
kept, rejected = generate(
|
||||
Path("/usr/lib/secubox/cve-triage/nuclei-subset"), present, complete
|
||||
Path("/usr/lib/secubox/cve-triage/nuclei-subset"), present, complete,
|
||||
existing=existing_patterns(Path("/etc/secubox/waf/waf-rules.json")),
|
||||
)
|
||||
now = datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%SZ")
|
||||
write_category(Path("/etc/secubox/waf/waf-rules.json"), kept, now=now)
|
||||
|
|
|
|||
|
|
@ -17,7 +17,7 @@ import sys
|
|||
from datetime import datetime, timezone
|
||||
from pathlib import Path
|
||||
|
||||
from .emit import write_category
|
||||
from .emit import existing_patterns, write_category
|
||||
from .generate import generate
|
||||
from .inventory import gather_present
|
||||
|
||||
|
|
@ -42,7 +42,8 @@ def main(argv: list[str] | None = None) -> int:
|
|||
args = p.parse_args(argv)
|
||||
|
||||
present, complete = gather_present()
|
||||
kept, rejected = generate(Path(args.subset), present, complete)
|
||||
kept, rejected = generate(Path(args.subset), present, complete,
|
||||
existing=existing_patterns(Path(args.rules)))
|
||||
|
||||
print(f"presence union: {len(present)} product(s), complete={complete}")
|
||||
print(f"kept: {len(kept)} probe(s), rejected: {len(rejected)}")
|
||||
|
|
|
|||
|
|
@ -26,6 +26,32 @@ from .extract import Candidate
|
|||
CATEGORY_KEY = "product_absent_probes"
|
||||
|
||||
|
||||
def existing_patterns(rules_path: Path) -> dict[str, str]:
|
||||
"""Map chaque pattern-string -> sa catégorie, SAUF notre propre
|
||||
CATEGORY_KEY. Exclure product_absent_probes est vital : sinon un 2e --apply
|
||||
verrait nos propres patterns comme « déjà couverts » et régénérerait zéro
|
||||
(idempotence cassée). Best-effort : fichier absent/corrompu -> {} (aucune
|
||||
dédup, jamais une erreur — la dédup n'est pas un garde de sûreté, contrairement
|
||||
à write_category qui, lui, lève)."""
|
||||
try:
|
||||
data = json.loads(Path(rules_path).read_text(encoding="utf-8"))
|
||||
except (OSError, json.JSONDecodeError):
|
||||
return {}
|
||||
if not isinstance(data, dict):
|
||||
return {}
|
||||
cats = data.get("categories", {})
|
||||
if not isinstance(cats, dict):
|
||||
return {}
|
||||
out: dict[str, str] = {}
|
||||
for cat_key, cat in cats.items():
|
||||
if cat_key == CATEGORY_KEY or not isinstance(cat, dict):
|
||||
continue
|
||||
for pat in cat.get("patterns", []):
|
||||
if isinstance(pat, dict) and isinstance(pat.get("pattern"), str):
|
||||
out.setdefault(pat["pattern"], cat_key)
|
||||
return out
|
||||
|
||||
|
||||
def build_category(candidates: list[Candidate], *, now: str) -> dict:
|
||||
"""Construit la catégorie (toujours mode=detect). Le pattern est le chemin
|
||||
échappé en regex (matche littéralement) — le but est de reconnaître la sonde,
|
||||
|
|
|
|||
|
|
@ -13,6 +13,7 @@ la CLI décide (dry-run par défaut).
|
|||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import re
|
||||
from pathlib import Path
|
||||
|
||||
import yaml
|
||||
|
|
@ -21,9 +22,28 @@ from .extract import Candidate, Rejection, extract, is_kev
|
|||
from .presence import is_appliance, should_generate
|
||||
|
||||
|
||||
def generate(subset_dir: Path, present: set[str],
|
||||
present_complete: bool) -> tuple[list[Candidate], list[tuple[str, str]]]:
|
||||
"""(retenus, [(fichier, raison de rejet)])."""
|
||||
def _covered_by(path: str, existing: dict[str, str]) -> str | None:
|
||||
"""Nom de la catégorie dont un pattern (traité comme regex, ce que fait le
|
||||
moteur WAF) matche `path` — donc la sonde detect y serait redondante. Un
|
||||
pattern stocké qui n'est pas une regex valide ne peut pas matcher : on le
|
||||
saute (jamais d'erreur). None si aucune couverture."""
|
||||
for pattern, category in existing.items():
|
||||
try:
|
||||
if re.search(pattern, path):
|
||||
return category
|
||||
except re.error:
|
||||
continue
|
||||
return None
|
||||
|
||||
|
||||
def generate(subset_dir: Path, present: set[str], present_complete: bool,
|
||||
*, existing: dict[str, str] | None = None
|
||||
) -> tuple[list[Candidate], list[tuple[str, str]]]:
|
||||
"""(retenus, [(fichier, raison de rejet)]). `existing` mappe pattern-string
|
||||
-> catégorie (voir emit.existing_patterns) : un candidat dont le pattern y
|
||||
figure déjà est rejeté (redondance ; une catégorie bloquante l'emporterait
|
||||
sur notre detect). None/{} => aucune dédup."""
|
||||
existing = existing or {}
|
||||
kept: list[Candidate] = []
|
||||
rejected: list[tuple[str, str]] = []
|
||||
|
||||
|
|
@ -53,6 +73,11 @@ def generate(subset_dir: Path, present: set[str],
|
|||
rejected.append((p.name, why))
|
||||
continue
|
||||
|
||||
covered_by = _covered_by(result.path, existing)
|
||||
if covered_by is not None:
|
||||
rejected.append((p.name, f"already covered by {covered_by}"))
|
||||
continue
|
||||
|
||||
kept.append(result)
|
||||
|
||||
return kept, rejected
|
||||
|
|
|
|||
|
|
@ -1,3 +1,13 @@
|
|||
secubox-cve-triage (1.1.1-1~bookworm1) bookworm; urgency=medium
|
||||
|
||||
* WAF generator: skip probes already covered by another WAF category
|
||||
(e.g. the hand-written cve_2024 blocking rules) — a blocking category wins,
|
||||
so a duplicate detect rule would never log. Rejected with an explicit
|
||||
"already covered by <category>" reason, visible in dry-run. Our own
|
||||
product_absent_probes category is excluded from the check (idempotence).
|
||||
|
||||
-- Gerald KERMA <devel@cybermind.fr> Fri, 18 Jul 2026 14:00:00 +0200
|
||||
|
||||
secubox-cve-triage (1.1.0-1~bookworm1) bookworm; urgency=medium
|
||||
|
||||
* WAF product-absent rule generator (closes #875). Turns vendored Nuclei
|
||||
|
|
|
|||
|
|
@ -87,3 +87,29 @@ def test_apply_with_incomplete_inventory_returns_3_and_leaves_rules_untouched(tm
|
|||
|
||||
assert rc == 3
|
||||
assert rules.read_text() == before
|
||||
|
||||
|
||||
def test_cli_dedups_against_existing_rules(tmp_path, monkeypatch, capsys):
|
||||
from api.wafgen import cli
|
||||
subset = tmp_path / "subset"
|
||||
subset.mkdir()
|
||||
(subset / "ivanti.yaml").write_text(
|
||||
'id: CVE-2024-21887\n'
|
||||
'info:\n severity: critical\n'
|
||||
' classification: {cve-id: CVE-2024-21887, cpe: "cpe:2.3:a:ivanti:connect_secure:*"}\n'
|
||||
' metadata: {vendor: ivanti, product: connect_secure}\n'
|
||||
' tags: cve,kev\n'
|
||||
'http:\n - method: GET\n path: ["{{BaseURL}}/api/v1/totp/user-backup"]\n')
|
||||
rules = tmp_path / "waf-rules.json"
|
||||
# Literal hyphen (unescaped) — this is the real live cve_2024 rule form,
|
||||
# a hand-written regex, NOT re.escape() output.
|
||||
rules.write_text(json.dumps({"categories": {"cve_2024": {"patterns": [
|
||||
{"pattern": "/api/v1/totp/user-backup"}]}}}))
|
||||
monkeypatch.setattr(cli, "gather_present", lambda: (set(), True))
|
||||
|
||||
rc = cli.main(["waf-rules", "generate",
|
||||
"--subset", str(subset), "--rules", str(rules)]) # dry-run
|
||||
out = capsys.readouterr().out
|
||||
assert rc == 0
|
||||
assert "kept: 0 probe(s)" in out
|
||||
assert "already covered by cve_2024" in out
|
||||
|
|
|
|||
|
|
@ -143,3 +143,37 @@ def test_write_category_raises_on_corrupt_rules_file_and_leaves_it_unchanged(tmp
|
|||
assert p.read_text() == corrupt
|
||||
leftovers = [f.name for f in tmp_path.iterdir() if f.name != p.name]
|
||||
assert leftovers == []
|
||||
|
||||
|
||||
def test_existing_patterns_maps_pattern_to_category_excluding_own_key(tmp_path):
|
||||
from api.wafgen.emit import existing_patterns
|
||||
rules = tmp_path / "waf-rules.json"
|
||||
rules.write_text(json.dumps({"categories": {
|
||||
"cve_2024": {"patterns": [{"pattern": "/mgmt/tm/util/bash"},
|
||||
{"pattern": "/api/v1/totp/user\\-backup"}]},
|
||||
"product_absent_probes": {"patterns": [{"pattern": "/should/not/appear"}]},
|
||||
}}))
|
||||
got = existing_patterns(rules)
|
||||
assert got["/mgmt/tm/util/bash"] == "cve_2024"
|
||||
assert got["/api/v1/totp/user\\-backup"] == "cve_2024"
|
||||
assert "/should/not/appear" not in got # our own key excluded → idempotence
|
||||
|
||||
|
||||
def test_existing_patterns_missing_or_corrupt_file_is_empty(tmp_path):
|
||||
from api.wafgen.emit import existing_patterns
|
||||
assert existing_patterns(tmp_path / "nope.json") == {}
|
||||
bad = tmp_path / "bad.json"
|
||||
bad.write_text("{ not json")
|
||||
assert existing_patterns(bad) == {}
|
||||
|
||||
|
||||
def test_existing_patterns_non_object_json_is_empty_not_a_raise(tmp_path):
|
||||
# Valid JSON but not a top-level object ([] / null / ...) must not raise
|
||||
# AttributeError on `.get` — best-effort contract: no dedup, never a crash.
|
||||
from api.wafgen.emit import existing_patterns
|
||||
empty_list = tmp_path / "list.json"
|
||||
empty_list.write_text("[]")
|
||||
assert existing_patterns(empty_list) == {}
|
||||
null_json = tmp_path / "null.json"
|
||||
null_json.write_text("null")
|
||||
assert existing_patterns(null_json) == {}
|
||||
|
|
|
|||
|
|
@ -90,3 +90,108 @@ def test_incomplete_inventory_generates_nothing(tmp_path):
|
|||
""")
|
||||
kept, _ = generate(subset, present=set(), present_complete=False)
|
||||
assert kept == []
|
||||
|
||||
|
||||
def test_candidate_already_covered_by_another_category_is_rejected(tmp_path):
|
||||
subset = tmp_path / "nuclei-subset"
|
||||
subset.mkdir()
|
||||
_write(subset, "ivanti.yaml", """
|
||||
id: CVE-2024-21887
|
||||
info:
|
||||
severity: critical
|
||||
classification: {cve-id: CVE-2024-21887, cpe: "cpe:2.3:a:ivanti:connect_secure:*"}
|
||||
metadata: {vendor: ivanti, product: connect_secure}
|
||||
tags: cve,kev
|
||||
http:
|
||||
- method: GET
|
||||
path: ["{{BaseURL}}/api/v1/totp/user-backup"]
|
||||
""")
|
||||
# cve_2024 already covers this path — stored as the live WAF file actually
|
||||
# would: a literal hand-written pattern (hyphen unescaped), NOT re.escape().
|
||||
existing = {"/api/v1/totp/user-backup": "cve_2024"}
|
||||
kept, rejected = generate(subset, present=set(), present_complete=True,
|
||||
existing=existing)
|
||||
assert kept == []
|
||||
assert any("already covered by cve_2024" in reason for _, reason in rejected)
|
||||
|
||||
|
||||
def test_candidate_covered_by_a_regex_variant_pattern_is_rejected(tmp_path):
|
||||
# The engine treats stored patterns as regexes: a hand-written pattern
|
||||
# using `\d` must still dedup the concrete candidate path it matches.
|
||||
subset = tmp_path / "nuclei-subset"
|
||||
subset.mkdir()
|
||||
_write(subset, "ivanti.yaml", """
|
||||
id: CVE-2024-21887
|
||||
info:
|
||||
severity: critical
|
||||
classification: {cve-id: CVE-2024-21887, cpe: "cpe:2.3:a:ivanti:connect_secure:*"}
|
||||
metadata: {vendor: ivanti, product: connect_secure}
|
||||
tags: cve,kev
|
||||
http:
|
||||
- method: GET
|
||||
path: ["{{BaseURL}}/api/v1/totp/user-backup"]
|
||||
""")
|
||||
existing = {r"/api/v\d/totp/user-backup": "cve_2024"}
|
||||
kept, rejected = generate(subset, present=set(), present_complete=True,
|
||||
existing=existing)
|
||||
assert kept == []
|
||||
assert any("already covered by cve_2024" in reason for _, reason in rejected)
|
||||
|
||||
|
||||
def test_invalid_existing_regex_is_skipped_not_raised(tmp_path):
|
||||
subset = tmp_path / "nuclei-subset"
|
||||
subset.mkdir()
|
||||
_write(subset, "ivanti.yaml", """
|
||||
id: CVE-2024-21887
|
||||
info:
|
||||
severity: critical
|
||||
classification: {cve-id: CVE-2024-21887, cpe: "cpe:2.3:a:ivanti:connect_secure:*"}
|
||||
metadata: {vendor: ivanti, product: connect_secure}
|
||||
tags: cve,kev
|
||||
http:
|
||||
- method: GET
|
||||
path: ["{{BaseURL}}/api/v1/totp/user-backup"]
|
||||
""")
|
||||
existing = {"[unclosed(": "weird"}
|
||||
kept, _ = generate(subset, present=set(), present_complete=True, existing=existing)
|
||||
assert [c.cve for c in kept] == ["CVE-2024-21887"]
|
||||
|
||||
|
||||
def test_no_dedup_when_existing_absent_keeps_candidate(tmp_path):
|
||||
subset = tmp_path / "nuclei-subset"
|
||||
subset.mkdir()
|
||||
_write(subset, "ivanti.yaml", """
|
||||
id: CVE-2024-21887
|
||||
info:
|
||||
severity: critical
|
||||
classification: {cve-id: CVE-2024-21887, cpe: "cpe:2.3:a:ivanti:connect_secure:*"}
|
||||
metadata: {vendor: ivanti, product: connect_secure}
|
||||
tags: cve,kev
|
||||
http:
|
||||
- method: GET
|
||||
path: ["{{BaseURL}}/api/v1/totp/user-backup"]
|
||||
""")
|
||||
kept, _ = generate(subset, present=set(), present_complete=True) # existing default None
|
||||
assert [c.cve for c in kept] == ["CVE-2024-21887"]
|
||||
|
||||
|
||||
def test_dedup_does_not_over_reject_unrelated_existing_patterns(tmp_path):
|
||||
# existing is non-empty but covers a DIFFERENT path → the candidate must
|
||||
# still be kept. Guards against a future refactor to substring/regex
|
||||
# matching silently over-rejecting real probes.
|
||||
subset = tmp_path / "nuclei-subset"
|
||||
subset.mkdir()
|
||||
_write(subset, "ivanti.yaml", """
|
||||
id: CVE-2024-21887
|
||||
info:
|
||||
severity: critical
|
||||
classification: {cve-id: CVE-2024-21887, cpe: "cpe:2.3:a:ivanti:connect_secure:*"}
|
||||
metadata: {vendor: ivanti, product: connect_secure}
|
||||
tags: cve,kev
|
||||
http:
|
||||
- method: GET
|
||||
path: ["{{BaseURL}}/api/v1/totp/user-backup"]
|
||||
""")
|
||||
existing = {"/some/other/unrelated/path": "cve_2024"}
|
||||
kept, _ = generate(subset, present=set(), present_complete=True, existing=existing)
|
||||
assert [c.cve for c in kept] == ["CVE-2024-21887"]
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user