def __init__(self, task_id): self.task_id = task_id self.check_result = {} # 检测结果 self.online_target = [] # 在线目标 self.result_pool = [] self.result = [] self.week_count = 0 self.auth_db = db_name_conf()['auth_db'] self.config_db = db_name_conf()['config_db'] self.weekpasswd_db = db_name_conf()['weekpasswd_db'] self.connectiondb = connectiondb2('AuthCrack') self.task_cursor = self.connectiondb[self.auth_db].find_one( {"_id": ObjectId(self.task_id)}) self.task_name = self.task_cursor['task_name'] # 191030_ssh_check self.target_list = parse_target( self.task_cursor['target']) # [u'192.168.1.141', u'192.168.1.141'] self.service_list = self.task_cursor['service'] # [u'ssh'] self.args = self.task_cursor['args'] # u'' self.config_cursor = self.connectiondb[self.config_db].find_one( {"config_name": settings.CONFIG_NAME}) self.processes = self.config_cursor['auth_tester_thread'] # 50 self.username_list = self.config_cursor['username_dict'] # ["root"] self.password_list = self.config_cursor[ 'password_dict'] # [u'123456', u'password', u'111111', u'666666', u'1qaz2wsx', u'1qaz2wsx3edc.com']
def __init__(self, instance_id): self._id = instance_id self.args_val = '-sV' self.instance_db = db_name_conf()['instance_db'] self.portinfo_db = db_name_conf()['portinfo_db'] self.connectiondb = connectiondb2('PortScanner') self.instance_info = self.connectiondb[self.instance_db].find_one( {"_id": ObjectId(self._id)}) self.target_val = self.instance_info['target'] self.ports_val = self._set_attr()
def __init__(self, task_id): self.task_id = "{}".format(task_id) self.result = [] self.plugin_db = db_name_conf()['plugin_db'] self.connectiondb = connectiondb2('PocsuiteScanner') self.tasks_cursor = self.connectiondb[tasks_db].find_one( {"_id": ObjectId(self.task_id)}) self.target_list = parse_target(self.tasks_cursor['target_list']) self.pluginid_list = self.tasks_cursor['pluginid_list'] self.processes = self.connectiondb[config_db].find_one( {"config_name": settings.CONFIG_NAME})['poc_thread'] self.tmp_app = '' # 如果是自动化探测任务、一个端口只会对应一个应用。比如它匹配的是flink poc, 不管匹配哪个都会是plugin_app=apache flink
def __init__(self, instance_id): self.instance_id = instance_id self.connectiondb = connectiondb2('AssetScanner') self.instance_cursor = self.connectiondb[instance_db].find_one( {"_id": ObjectId(instance_id)}) self.portinfo_cursor = self.connectiondb[portinfo_db].find({ "is_delete": { "$ne": 1 }, "instance_id": instance_id }) self.name = self.instance_cursor['name']
def __init__(self): self.auth_db = db_name_conf()['auth_db'] self.weekpasswd_db = db_name_conf()['weekpasswd_db'] self.connectiondb = connectiondb2('AuthTesterLoop')
def __init__(self): self.connectiondb = connectiondb2('PocScannerLoop')
def __init__(self): # 资产扫描状态更新; 确认资产下属的所有任务完成后进行更新资产扫描状态 self.connectiondb = connectiondb2('AssetDiscoveryLoop')
def __init__(self): # 更新Mongo中AWVS的任务状态 # 定时从AWVS服务端的获取target_id的状态 self.vulscan_db = db_name_conf()['vulscan_db'] self.connectiondb = connectiondb2('AwvsTaskLoop')