def __init__(self, domain, services, name): super().__init__() self.services = services self.contact_svc = services.get('contact_svc') self.file_svc = services.get('file_svc') self.name = name self.log = BaseWorld.create_logger('contact_dns_handler') self.domain = domain self.transport = None # Stores received message chunks. self.pending_messages = {} # Stores completed messages from agents. self.completed_messages = {} # Stores instructions for agents to fetch # Key: message ID. # Value: StoredResponse obj self.pending_instructions = {} # Stores payloads for agents to fetch # Key: message ID. # Value: StoredResponse obj self.pending_payloads = {} # Stores payload names for agents to fetch # Key: message ID. # Value: StoredResponse obj self.pending_payload_names = {}
def __init__(self, services): super().__init__() self.services = services self.handles = [ Handle(tag='beacon') ] self.log = BaseWorld.create_logger('udp_handler')
def __init__(self, services, user_name, user_password): super().__init__() self.services = services self.log = BaseWorld.create_logger('ssh_server') self.valid_user_credentials = { user_name: user_password }
def __init__(self, services): self.check_config() self.name = 'ftp' self.description = 'Accept agent beacons through ftp' self.contact_svc = services.get('contact_svc') self.file_svc = services.get('file_svc') self.logger = BaseWorld.create_logger('contact_ftp') self.host = self.get_config('app.contact.ftp.host') self.port = self.get_config('app.contact.ftp.port') self.directory = self.get_config('app.contact.ftp.server.dir').lstrip( '/') self.home = os.path.join('/', self.directory) self.user = self.get_config('app.contact.ftp.user') self.pword = self.get_config('app.contact.ftp.pword') self.server = None
def __init__(self, services): self.services = services self.handles = [] self.log = BaseWorld.create_logger('websocket_handler')
def __init__(self, services): super().__init__() self.services = services self.log = BaseWorld.create_logger('udp_session')
def __init__(self, services): self.services = services self.log = BaseWorld.create_logger('websocket_session')