ghndrx/prowler
1
0
Fork 0
mirror of https://github.com/ghndrx/prowler.git synced 2026-02-10 14:55:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(azure): subscription as parameter (#1526)

Browse Source
This commit is contained in:
Nacho Rivera
2022-11-29 13:46:38 +01:00
committed by GitHub
parent 8f802f1241
commit e5e01e51a9
4 changed files with 32 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
providers/azure/azure_provider.py
View File

@@ -7,7 +7,7 @@ from msgraph.core import GraphClient
from lib.logger import logger
from providers.azure.lib.audit_info.audit_info import azure_audit_info
from providers.azure.lib.audit_info.models import Azure_Identity_Info
from providers.azure.lib.audit_info.models import Azure_Audit_Info, Azure_Identity_Info
class Azure_Provider:
@@ -54,7 +54,7 @@ def validate_credentials(
return azure_identity
def azure_provider_set_session():
def azure_provider_set_session(subscription_ids: list) -> Azure_Audit_Info:
logger.info("Setting Azure session ...")
azure_identity = check_credential_env_vars()
azure_audit_info.credentials = Azure_Provider().get_credentials()
@@ -69,11 +69,20 @@ def azure_provider_set_session():
subscriptions_client = SubscriptionClient(
credential=azure_audit_info.credentials
)
for subscription in subscriptions_client.subscriptions.list():
if not subscription_ids:
logger.info("Scanning all the Azure subscriptions...")
for subscription in subscriptions_client.subscriptions.list():
azure_audit_info.subscriptions.update(
{subscription.display_name: subscription.subscription_id}
)
azure_audit_info.subscriptions.update(
{subscription.display_name: subscription.subscription_id}
)
else:
logger.info("Scanning the subscriptions passed as argument ...")
for id in subscription_ids:
subscription = subscriptions_client.subscriptions.get(
subscription_id=id
)
azure_audit_info.subscriptions.update({subscription.display_name: id})
except Exception as error:
logger.critical(
f"{error.__class__.__name__}[{error.__traceback__.tb_lineno}] -- {error}"

2
providers/azure/lib/audit_info/models.py
View File

@@ -14,7 +14,7 @@ class Azure_Identity_Info(BaseModel):
class Azure_Audit_Info:
credentials: DefaultAzureCredential
identity: Azure_Identity_Info
subscriptions: list[dict]
subscriptions: dict
audited_account: str
def __init__(self, credentials, identity, subscriptions):