p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Island, Agent: fixed imports to reference credential type enum in common

This commit is contained in:
vakarisz 2022-02-17 17:29:23 +02:00
parent 73434537fe
commit c96674f834
2 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
monkey/infection_monkey/i_puppet/i_puppet.py
View File

@ -4,7 +4,8 @@ from collections import namedtuple
from enum import Enum
from typing import Dict, List, Sequence
from . import Credentials, PluginType
from . import PluginType
from .credential_collection import Credentials
class PortStatus(Enum):

16
monkey/monkey_island/cc/services/telemetry/processing/credentials/credentials_parser.py
View File

@ -1,6 +1,6 @@
import logging
from infection_monkey.i_puppet import CredentialType
from common.common_consts.credentials_type import CredentialsType
from .identities.username_processor import process_username
from .secrets.lm_hash_processor import process_lm_hash
@ -11,21 +11,21 @@ from .secrets.ssh_key_processor import process_ssh_key
logger = logging.getLogger(__name__)
SECRET_PROCESSORS = {
CredentialType.PASSWORD: process_password,
CredentialType.NT_HASH: process_nt_hash,
CredentialType.LM_HASH: process_lm_hash,
CredentialType.SSH_KEYPAIR: process_ssh_key,
CredentialsType.PASSWORD: process_password,
CredentialsType.NT_HASH: process_nt_hash,
CredentialsType.LM_HASH: process_lm_hash,
CredentialsType.SSH_KEYPAIR: process_ssh_key,
}
IDENTITY_PROCESSORS = {
CredentialType.USERNAME: process_username,
CredentialsType.USERNAME: process_username,
}
def parse_credentials(credentials: dict):
for credential in credentials["credentials"]:
if is_ssh_keypair(credentials):
IDENTITY_PROCESSORS[CredentialType.SSH_KEYPAIR](credential, credentials["monkey_guid"])
IDENTITY_PROCESSORS[CredentialsType.SSH_KEYPAIR](credential, credentials["monkey_guid"])
else:
for identity in credential["identities"]:
IDENTITY_PROCESSORS[identity["type"]](identity)
@ -35,5 +35,5 @@ def parse_credentials(credentials: dict):
def is_ssh_keypair(credentials: dict) -> bool:
return bool(
filter(credentials["secrets"], lambda secret: secret["type"] == CredentialType.SSH_KEYPAIR)
filter(credentials["secrets"], lambda secret: secret["type"] == CredentialsType.SSH_KEYPAIR)
)