Agent: define credential collector, credentials interfaces

monkey/infection_monkey/credential_collectors/credential_components/i_credential_component.py

@@ -0,0 +1,10 @@
+from abc import ABC
+from dataclasses import dataclass
+
+from ..credential_types import CredentialTypes
+
+
+@dataclass
+class ICredentialComponent(ABC):
+    type: CredentialTypes
+    content: dict

monkey/infection_monkey/credential_collectors/credential_components/ntlm_hash.py

@@ -0,0 +1,8 @@
+from ..credential_types import CredentialTypes
+
+from .i_credential_component import ICredentialComponent
+
+
+class NtlmHash(ICredentialComponent):
+    def __init__(self, content: dict):
+        super().__init__(type=CredentialTypes.NTLM_HASH, content=content)

monkey/infection_monkey/credential_collectors/credential_components/password.py

@@ -0,0 +1,8 @@
+from ..credential_types import CredentialTypes
+
+from .i_credential_component import ICredentialComponent
+
+
+class Password(ICredentialComponent):
+    def __init__(self, content: dict):
+        super().__init__(type=CredentialTypes.PASSWORD, content=content)

monkey/infection_monkey/credential_collectors/credential_components/ssh_keypair.py

@@ -0,0 +1,8 @@
+from ..credential_types import CredentialTypes
+
+from .i_credential_component import ICredentialComponent
+
+
+class SSHKeypair(ICredentialComponent):
+    def __init__(self, content: dict):
+        super().__init__(type=CredentialTypes.KEYPAIR, content=content)

monkey/infection_monkey/credential_collectors/credential_components/username.py

@@ -0,0 +1,8 @@
+from ..credential_types import CredentialTypes
+
+from .i_credential_component import ICredentialComponent
+
+
+class Username(ICredentialComponent):
+    def __init__(self, content: dict):
+        super().__init__(type=CredentialTypes.USERNAME, content=content)

monkey/infection_monkey/credential_collectors/credential_types.py

@@ -0,0 +1,8 @@
+from enum import Enum
+
+
+class CredentialTypes(Enum):
+    KEYPAIR = 1
+    USERNAME = 2
+    PASSWORD = 3
+    NTLM_HASH = 4

monkey/infection_monkey/credential_collectors/credentials.py

@@ -0,0 +1,10 @@
+from dataclasses import dataclass
+from typing import List
+
+from .credential_components.i_credential_component import ICredentialComponent
+
+
+@dataclass
+class Credentials:
+    identities: List[ICredentialComponent]
+    secrets: List[ICredentialComponent]

monkey/infection_monkey/credential_collectors/i_credential_collector.py

@@ -0,0 +1,9 @@
+from abc import ABC, abstractmethod
+
+from .credentials import Credentials
+
+
+class ICredentialCollector(ABC):
+    @abstractmethod
+    def collect_credentials(self) -> Credentials:
+        pass