p34709852
/
monkey
forked from p15670423/monkey
1
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

Agent: Format HTTPFingerprinter with Black

This commit is contained in:
Mike Salvatore 2022-03-04 17:05:35 -05:00
parent bf998f5021
commit efa0c5beb4
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
monkey/infection_monkey/network_scanning/http_fingerprinter.py
View File

@ -25,11 +25,11 @@ class HTTPFingerprinter(IFingerprinter):
""" """
def get_host_fingerprint( def get_host_fingerprint(
self, self,
host: str, host: str,
_: PingScanData, _: PingScanData,
port_scan_data: Dict[int, PortScanData], port_scan_data: Dict[int, PortScanData],
options: Dict, options: Dict,
) -> FingerprintData: ) -> FingerprintData:
services = {} services = {}
http_ports = set(options.get("http_ports", [])) http_ports = set(options.get("http_ports", []))
@ -85,7 +85,7 @@ def _get_http_headers(url: str) -> Optional[Dict[str, Any]]:
def _get_open_http_ports( def _get_open_http_ports(
allowed_http_ports: Set, port_scan_data: Dict[int, PortScanData] allowed_http_ports: Set, port_scan_data: Dict[int, PortScanData]
) -> Iterable[int]: ) -> Iterable[int]:
open_ports = (psd.port for psd in port_scan_data.values() if psd.status == PortStatus.OPEN) open_ports = (psd.port for psd in port_scan_data.values() if psd.status == PortStatus.OPEN)
return (port for port in open_ports if port in allowed_http_ports) return (port for port in open_ports if port in allowed_http_ports)