forked from p34709852/monkey
Island: Reformat "finger_classes" config options before sending to Agent
This commit is contained in:
parent
d51af8a583
commit
e524718960
|
@ -488,6 +488,9 @@ class ConfigService:
|
||||||
formatted_network_scan_config["icmp"] = ConfigService._format_icmp_scan_from_flat_config(
|
formatted_network_scan_config["icmp"] = ConfigService._format_icmp_scan_from_flat_config(
|
||||||
config
|
config
|
||||||
)
|
)
|
||||||
|
formatted_network_scan_config[
|
||||||
|
"fingerprinters"
|
||||||
|
] = ConfigService._format_fingerprinters_from_flat_config(config)
|
||||||
|
|
||||||
return formatted_network_scan_config
|
return formatted_network_scan_config
|
||||||
|
|
||||||
|
@ -529,6 +532,15 @@ class ConfigService:
|
||||||
|
|
||||||
return formatted_icmp_scan_config
|
return formatted_icmp_scan_config
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def _format_fingerprinters_from_flat_config(config: Dict):
|
||||||
|
flat_fingerprinter_classes_field = "finger_classes"
|
||||||
|
|
||||||
|
formatted_fingerprinters = config[flat_fingerprinter_classes_field]
|
||||||
|
config.pop(flat_fingerprinter_classes_field)
|
||||||
|
|
||||||
|
return formatted_fingerprinters
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def _format_targets_from_flat_config(config: Dict):
|
def _format_targets_from_flat_config(config: Dict):
|
||||||
flat_blocked_ips_field = "blocked_ips"
|
flat_blocked_ips_field = "blocked_ips"
|
||||||
|
|
|
@ -66,7 +66,6 @@
|
||||||
"SMBFinger",
|
"SMBFinger",
|
||||||
"SSHFinger",
|
"SSHFinger",
|
||||||
"HTTPFinger",
|
"HTTPFinger",
|
||||||
"MySQLFinger",
|
|
||||||
"MSSQLFinger",
|
"MSSQLFinger",
|
||||||
"ElasticFinger"
|
"ElasticFinger"
|
||||||
],
|
],
|
||||||
|
|
|
@ -143,6 +143,13 @@ def test_format_config_for_agent__network_scan(flat_monkey_config):
|
||||||
"icmp": {
|
"icmp": {
|
||||||
"timeout_ms": 1000,
|
"timeout_ms": 1000,
|
||||||
},
|
},
|
||||||
|
"fingerprinters": [
|
||||||
|
"SMBFinger",
|
||||||
|
"SSHFinger",
|
||||||
|
"HTTPFinger",
|
||||||
|
"MSSQLFinger",
|
||||||
|
"ElasticFinger",
|
||||||
|
],
|
||||||
}
|
}
|
||||||
ConfigService.format_flat_config_for_agent(flat_monkey_config)
|
ConfigService.format_flat_config_for_agent(flat_monkey_config)
|
||||||
|
|
||||||
|
@ -153,3 +160,4 @@ def test_format_config_for_agent__network_scan(flat_monkey_config):
|
||||||
assert "tcp_scan_timeout" not in flat_monkey_config
|
assert "tcp_scan_timeout" not in flat_monkey_config
|
||||||
assert "tcp_target_ports" not in flat_monkey_config
|
assert "tcp_target_ports" not in flat_monkey_config
|
||||||
assert "ping_scan_timeout" not in flat_monkey_config
|
assert "ping_scan_timeout" not in flat_monkey_config
|
||||||
|
assert "finger_classes" not in flat_monkey_config
|
||||||
|
|
Loading…
Reference in New Issue