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')
