feat(rpki): ripestat validator + sqlite cache
RipeStatValidator makes two RIPE STAT calls per uncached IP: network-info -> announced prefix, rpki-validation -> ROA state. 2-second timeout; any network failure returns status='unknown'. SQLite cache keyed by IP, 12-hour TTL, pruned on validator init. Cache avoids per-event HTTP for the high-churn attacker pool — steady-state cost approaches zero for repeat offenders.
This commit is contained in:
64
tests/rpki/test_cache.py
Normal file
64
tests/rpki/test_cache.py
Normal file
@@ -0,0 +1,64 @@
|
||||
"""SQLite RPKI cache tests."""
|
||||
from __future__ import annotations
|
||||
|
||||
import time
|
||||
from pathlib import Path
|
||||
|
||||
import pytest
|
||||
|
||||
from decnet.rpki.cache import TTL_S, get, open_db, prune, put
|
||||
|
||||
|
||||
@pytest.fixture()
|
||||
def db(tmp_path: Path):
|
||||
return open_db(tmp_path / "rpki.db")
|
||||
|
||||
|
||||
def test_miss_returns_none(db) -> None:
|
||||
assert get(db, "8.8.8.8") is None
|
||||
|
||||
|
||||
def test_put_then_get_returns_entry(db) -> None:
|
||||
put(db, "8.8.8.8", 15169, "valid", "8.8.8.0/24")
|
||||
result = get(db, "8.8.8.8")
|
||||
assert result == ("valid", "8.8.8.0/24")
|
||||
|
||||
|
||||
def test_get_returns_none_prefix_when_stored_null(db) -> None:
|
||||
put(db, "1.2.3.4", 64496, "not-found", None)
|
||||
status, prefix = get(db, "1.2.3.4")
|
||||
assert status == "not-found"
|
||||
assert prefix is None
|
||||
|
||||
|
||||
def test_expired_entry_returns_none(db, monkeypatch: pytest.MonkeyPatch) -> None:
|
||||
put(db, "8.8.8.8", 15169, "valid", "8.8.8.0/24")
|
||||
future = time.time() + TTL_S + 1
|
||||
monkeypatch.setattr("decnet.rpki.cache.time.time", lambda: future)
|
||||
assert get(db, "8.8.8.8") is None
|
||||
|
||||
|
||||
def test_replace_updates_entry(db) -> None:
|
||||
put(db, "8.8.8.8", 15169, "valid", "8.8.8.0/24")
|
||||
put(db, "8.8.8.8", 15169, "invalid", "8.8.8.0/24")
|
||||
status, _ = get(db, "8.8.8.8")
|
||||
assert status == "invalid"
|
||||
|
||||
|
||||
def test_prune_removes_stale_rows(db, monkeypatch: pytest.MonkeyPatch) -> None:
|
||||
put(db, "1.1.1.1", 13335, "valid", "1.1.1.0/24")
|
||||
put(db, "2.2.2.2", 3215, "invalid", "2.0.0.0/8")
|
||||
future = time.time() + TTL_S + 1
|
||||
monkeypatch.setattr("decnet.rpki.cache.time.time", lambda: future)
|
||||
count = prune(db)
|
||||
assert count == 2
|
||||
# After prune, both gone
|
||||
assert get(db, "1.1.1.1") is None
|
||||
assert get(db, "2.2.2.2") is None
|
||||
|
||||
|
||||
def test_prune_keeps_fresh_rows(db) -> None:
|
||||
put(db, "8.8.8.8", 15169, "valid", "8.8.8.0/24")
|
||||
count = prune(db)
|
||||
assert count == 0
|
||||
assert get(db, "8.8.8.8") is not None
|
||||
Reference in New Issue
Block a user