feat(securityhub_check): Add check and service for SecurityHub (#1360)

Co-authored-by: Toni de la Fuente <toni@blyx.com>
Co-authored-by: sergargar <sergio@verica.io>
Co-authored-by: Pepe Fagoaga <pepe@verica.io>
This commit is contained in:
Sergio Garcia
2022-10-19 15:21:07 +02:00
committed by GitHub
parent 6e73321a95
commit 5b5b0b0405
10 changed files with 290 additions and 39 deletions

View File

@@ -0,0 +1,91 @@
import threading
from dataclasses import dataclass
from lib.logger import logger
from providers.aws.aws_provider import generate_regional_clients
################## SecurityHub
class SecurityHub:
def __init__(self, audit_info):
self.service = "securityhub"
self.session = audit_info.audit_session
self.audited_account = audit_info.audited_account
self.regional_clients = generate_regional_clients(self.service, audit_info)
self.securityhubs = []
self.__threading_call__(self.__describe_hub__)
def __get_session__(self):
return self.session
def __threading_call__(self, call):
threads = []
for regional_client in self.regional_clients.values():
threads.append(threading.Thread(target=call, args=(regional_client,)))
for t in threads:
t.start()
for t in threads:
t.join()
def __describe_hub__(self, regional_client):
logger.info("SecurityHub - Describing Hub...")
try:
get_enabled_standards_paginator = regional_client.get_paginator(
"get_enabled_standards"
)
standards = ""
for page in get_enabled_standards_paginator.paginate():
for standard in page["StandardsSubscriptions"]:
standards += f" {standard['StandardsArn'].split('/')[1]}"
# Security Hub is not enabled in region
if standards == "":
self.securityhubs.append(
SecurityHubHub(
"",
"Security Hub",
"NOT_AVAILABLE",
"",
regional_client.region,
)
)
else:
# SecurityHub is active so get HubArn
hub_arn = regional_client.describe_hub()["HubArn"]
hub_id = hub_arn.split("/")[1]
self.securityhubs.append(
SecurityHubHub(
hub_arn,
hub_id,
"ACTIVE",
standards,
regional_client.region,
)
)
except Exception as error:
logger.error(
f"{regional_client.region} -- {error.__class__.__name__}[{error.__traceback__.tb_lineno}]: {error}"
)
@dataclass
class SecurityHubHub:
arn: str
id: str
status: str
standards: str
region: str
def __init__(
self,
arn,
id,
status,
standards,
region,
):
self.arn = arn
self.id = id
self.status = status
self.standards = standards
self.region = region