Agent: move mimikatz collector to credential collectors

This commit is contained in:
vakarisz 2022-02-14 15:55:11 +02:00 committed by Ilija Lazoroski
parent 6aa2160f31
commit 2ba793e0cf
6 changed files with 7 additions and 9 deletions

View File

@ -1,5 +1,7 @@
from typing import List from typing import List
import pypykatz_handler
from infection_monkey.credential_collectors import ( from infection_monkey.credential_collectors import (
Credentials, Credentials,
ICredentialCollector, ICredentialCollector,
@ -7,10 +9,8 @@ from infection_monkey.credential_collectors import (
Password, Password,
Username, Username,
) )
from infection_monkey.system_info.windows_cred_collector import pypykatz_handler
from infection_monkey.system_info.windows_cred_collector.windows_credentials import ( from .windows_credentials import WindowsCredentials
WindowsCredentials,
)
class MimikatzCredentialCollector(ICredentialCollector): class MimikatzCredentialCollector(ICredentialCollector):

View File

@ -3,9 +3,7 @@ from typing import Any, Dict, List, NewType
from pypykatz.pypykatz import pypykatz from pypykatz.pypykatz import pypykatz
from infection_monkey.system_info.windows_cred_collector.windows_credentials import ( from .windows_credentials import WindowsCredentials
WindowsCredentials,
)
CREDENTIAL_TYPES = [ CREDENTIAL_TYPES = [
"msv_creds", "msv_creds",

View File

@ -2,7 +2,7 @@ import logging
import sys import sys
from common.common_consts.system_info_collectors_names import MIMIKATZ_COLLECTOR from common.common_consts.system_info_collectors_names import MIMIKATZ_COLLECTOR
from infection_monkey.system_info.windows_cred_collector.mimikatz_cred_collector import ( from infection_monkey.credential_collectors.windows_cred_collector.mimikatz_cred_collector import (
MimikatzCredentialCollector, MimikatzCredentialCollector,
) )

View File

@ -1,6 +1,6 @@
from unittest import TestCase from unittest import TestCase
from infection_monkey.system_info.windows_cred_collector.pypykatz_handler import ( from infection_monkey.credential_collectors.mimikatz_collector.pypykatz_handler import (
_get_creds_from_pypykatz_session, _get_creds_from_pypykatz_session,
) )