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

BB: Remove usage of CredentialComponentType in Zerologon analyzer

This commit is contained in:
Shreya Malviya 2022-09-15 14:07:18 +05:30
parent cf48189a07
commit 75dddb0861
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
envs/monkey_zoo/blackbox/analyzers/zerologon_analyzer.py
View File

@ -1,7 +1,7 @@
from pprint import pformat from pprint import pformat
from typing import List from typing import List
from common.credentials import CredentialComponentType, Credentials from common.credentials import Credentials, LMHash, NTHash, Username
from envs.monkey_zoo.blackbox.analyzers.analyzer import Analyzer from envs.monkey_zoo.blackbox.analyzers.analyzer import Analyzer
from envs.monkey_zoo.blackbox.analyzers.analyzer_log import AnalyzerLog from envs.monkey_zoo.blackbox.analyzers.analyzer_log import AnalyzerLog
from envs.monkey_zoo.blackbox.island_client.monkey_island_client import MonkeyIslandClient from envs.monkey_zoo.blackbox.island_client.monkey_island_client import MonkeyIslandClient
@ -36,11 +36,11 @@ class ZerologonAnalyzer(Analyzer):
credentials_on_island = set() credentials_on_island = set()
for credentials in propagation_credentials: for credentials in propagation_credentials:
if credentials.identity.credential_type is CredentialComponentType.USERNAME: if isinstance(credentials.identity, Username):
credentials_on_island.update([credentials.identity.username]) credentials_on_island.update([credentials.identity.username])
if credentials.secret.credential_type is CredentialComponentType.NT_HASH: if isinstance(credentials.secret, NTHash):
credentials_on_island.update([credentials.secret.nt_hash]) credentials_on_island.update([credentials.secret.nt_hash])
if credentials.secret.credential_type is CredentialComponentType.LM_HASH: if isinstance(credentials.secret, LMHash):
credentials_on_island.update([credentials.secret.lm_hash]) credentials_on_island.update([credentials.secret.lm_hash])
return list(credentials_on_island) return list(credentials_on_island)