diff --git a/PROJECTS/intermediate/dlp-scanner/src/dlp_scanner/detectors/registry.py b/PROJECTS/intermediate/dlp-scanner/src/dlp_scanner/detectors/registry.py index 55e2f46c..635e8a1c 100644 --- a/PROJECTS/intermediate/dlp-scanner/src/dlp_scanner/detectors/registry.py +++ b/PROJECTS/intermediate/dlp-scanner/src/dlp_scanner/detectors/registry.py @@ -39,6 +39,7 @@ class DetectorRegistry: """ def __init__( self, + *, enable_patterns: list[str] | None = None, disable_patterns: list[str] | None = None, allowlist_values: frozenset[str] | None = None, diff --git a/PROJECTS/intermediate/dlp-scanner/src/dlp_scanner/scanners/db_scanner.py b/PROJECTS/intermediate/dlp-scanner/src/dlp_scanner/scanners/db_scanner.py index 125a632d..a3a95532 100644 --- a/PROJECTS/intermediate/dlp-scanner/src/dlp_scanner/scanners/db_scanner.py +++ b/PROJECTS/intermediate/dlp-scanner/src/dlp_scanner/scanners/db_scanner.py @@ -475,6 +475,7 @@ class DatabaseScanner: self, matches: list[DetectorMatch], text: str, + *, table_name: str, uri: str, result: ScanResult, diff --git a/PROJECTS/intermediate/dlp-scanner/src/dlp_scanner/scanners/network_scanner.py b/PROJECTS/intermediate/dlp-scanner/src/dlp_scanner/scanners/network_scanner.py index 936b9212..e7aa9f88 100644 --- a/PROJECTS/intermediate/dlp-scanner/src/dlp_scanner/scanners/network_scanner.py +++ b/PROJECTS/intermediate/dlp-scanner/src/dlp_scanner/scanners/network_scanner.py @@ -137,10 +137,10 @@ class NetworkScanner: packet.payload, packet.src_ip, packet.dst_ip, - path, - packet_count, - dns_detector, - result, + path = path, + packet_num = packet_count, + dns_detector = dns_detector, + result = result, ) if packet.payload: @@ -175,6 +175,7 @@ class NetworkScanner: payload: bytes, src_ip: str, dst_ip: str, + *, path: Path, packet_num: int, dns_detector: DnsExfilDetector, diff --git a/PROJECTS/intermediate/dlp-scanner/tests/test_network/test_flow_tracker.py b/PROJECTS/intermediate/dlp-scanner/tests/test_network/test_flow_tracker.py index 642e56bf..91ddbd42 100644 --- a/PROJECTS/intermediate/dlp-scanner/tests/test_network/test_flow_tracker.py +++ b/PROJECTS/intermediate/dlp-scanner/tests/test_network/test_flow_tracker.py @@ -12,6 +12,7 @@ from dlp_scanner.network.pcap import PacketInfo def _make_packet( + *, src_ip: str = "192.168.1.1", dst_ip: str = "10.0.0.1", src_port: int = 12345,