fix: remove unused imports and merge comparison in dlp-scanner
Remove unused TextChunk, DetectorMatch, make_flow_key imports and merge repeated DNS_PORT equality check into set membership test.
This commit is contained in:
parent
4c5bce05c1
commit
c79d17c8ff
|
|
@ -35,10 +35,7 @@ from dlp_scanner.extractors.structured import (
|
||||||
XmlExtractor,
|
XmlExtractor,
|
||||||
YamlExtractor,
|
YamlExtractor,
|
||||||
)
|
)
|
||||||
from dlp_scanner.models import (
|
from dlp_scanner.models import ScanResult
|
||||||
ScanResult,
|
|
||||||
TextChunk,
|
|
||||||
)
|
|
||||||
from dlp_scanner.scoring import match_to_finding
|
from dlp_scanner.scoring import match_to_finding
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -10,7 +10,6 @@ from pathlib import Path
|
||||||
import structlog
|
import structlog
|
||||||
|
|
||||||
from dlp_scanner.config import ScanConfig
|
from dlp_scanner.config import ScanConfig
|
||||||
from dlp_scanner.detectors.base import DetectorMatch
|
|
||||||
from dlp_scanner.detectors.registry import DetectorRegistry
|
from dlp_scanner.detectors.registry import DetectorRegistry
|
||||||
from dlp_scanner.models import (
|
from dlp_scanner.models import (
|
||||||
Finding,
|
Finding,
|
||||||
|
|
@ -22,10 +21,7 @@ from dlp_scanner.network.exfiltration import (
|
||||||
ExfilIndicator,
|
ExfilIndicator,
|
||||||
detect_base64_payload,
|
detect_base64_payload,
|
||||||
)
|
)
|
||||||
from dlp_scanner.network.flow_tracker import (
|
from dlp_scanner.network.flow_tracker import FlowTracker
|
||||||
FlowTracker,
|
|
||||||
make_flow_key,
|
|
||||||
)
|
|
||||||
from dlp_scanner.network.pcap import read_pcap
|
from dlp_scanner.network.pcap import read_pcap
|
||||||
from dlp_scanner.network.protocols import (
|
from dlp_scanner.network.protocols import (
|
||||||
DNS_PORT,
|
DNS_PORT,
|
||||||
|
|
@ -131,8 +127,10 @@ class NetworkScanner:
|
||||||
if (
|
if (
|
||||||
packet.protocol == "udp"
|
packet.protocol == "udp"
|
||||||
and (
|
and (
|
||||||
packet.src_port == DNS_PORT
|
DNS_PORT in (
|
||||||
or packet.dst_port == DNS_PORT
|
packet.src_port,
|
||||||
|
packet.dst_port,
|
||||||
|
)
|
||||||
)
|
)
|
||||||
):
|
):
|
||||||
self._process_dns_packet(
|
self._process_dns_packet(
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue