fixed linting issues

This commit is contained in:
danielsagi
2022-05-28 14:51:43 -07:00
parent 2fac662c51
commit 3df7ea85bc
4 changed files with 11 additions and 6 deletions
+1 -1
View File
@@ -22,7 +22,7 @@ class Config:
- network_timeout: Timeout for network operations
- num_worker_threads: Add a flag --threads to change the default 800 thread count of the event handler
- pod: From pod scanning mode
- full_cloud_scan: disables limit for 24 cidr in cloud envs
- full_cloud_scan: disables limit for 24 cidr in cloud envs
- remote: Hosts to scan
- report: Output format
- statistics: Include hunters statistics
+5 -1
View File
@@ -18,7 +18,11 @@ def parser_add_arguments(parser):
parser.add_argument("--pod", action="store_true", help="Set hunter as an insider pod")
parser.add_argument("--full-cloud-scan", action="store_true", help="Disable hardlimit of '/24' CIDR when scraping Instance Metadata API")
parser.add_argument(
"--full-cloud-scan",
action="store_true",
help="Disable hardlimit of '/24' CIDR when scraping Instance Metadata API",
)
parser.add_argument(
"--include-patched-versions",
@@ -1,4 +1,3 @@
import enum
import logging
import requests
import ipaddress
+5 -3
View File
@@ -78,13 +78,13 @@ class AzureInstanceMetadataServiceDiscovery(Discovery):
logger.debug("Trying to access IMDS (Azure Metadata Service) from pod")
available_versions = AzureInstanceMetadataService.get_versions(network_timeout=config.network_timeout)
if not available_versions:
logger.debug("IMDS not available")
return
versions_info = dict()
for version in available_versions['apiVersions']:
for version in available_versions["apiVersions"]:
instance_data = AzureInstanceMetadataService.get_instance_data(
api_version=version, network_timeout=config.network_timeout
)
@@ -110,7 +110,9 @@ class AzureSubnetsDiscovery(Discovery):
tmp_prefix = info["network"]["interface"][0]["ipv4"]["subnet"][0]["prefix"]
if not config.full_cloud_scan:
logger.debug(f"Discovered azure subnet {tmp_prefix} but scanning {prefix} due to `full_cloud_scan=False` option ")
logger.debug(
f"Discovered azure subnet {tmp_prefix} but scanning {prefix} due to `full_cloud_scan=False` option "
)
else:
prefix = tmp_prefix