def handle_result(self, request, result):
result = deepcopy(request.args[0])
if result['success']:
self.success_num += 1
logger.critical('Target: %s [Success] (%s)'
% (request.args[0]['options']['target'], result['poc_name']))
else:
logger.error('Target: %s [Failed] (%s)'
% (request.args[0]['options']['target'], result['poc_name']))
self.result_fobj.write(json.dumps(result) + '\n')
def handle_result(self, request, result):
result = deepcopy(request.args[0])
if result['success']:
self.success_num += 1
logger.critical(
'Target: %s [Success] (%s)' %
(request.args[0]['options']['target'], result['poc_name']))
else:
logger.error(
'Target: %s [Failed] (%s)' %
(request.args[0]['options']['target'], result['poc_name']))
self.result_fobj.write(json.dumps(result) + '\n')
def download_poc(poc_id, cookie):
link = download_link % poc_id
try:
poc_file = requests.get(link, headers={'Cookie': cookie} if cookie else None, timeout=10)
except Exception:
logger.error('Download "%s" [Failed] (Connection Error)' % poc_id)
return False
if _KEYWORDS not in poc_file.content:
logger.error('Download "%s" [Failed] (Permission denied or POC not exist)' % poc_id)
return False
ext = '.py'
restore_path = module_path + poc_id.replace('-', '_') + ext
open(restore_path, 'wb').write(poc_file.content)
logger.critical('Download "%s" [Success]' % poc_id)
return True
def download_poc(poc_id, cookie):
link = download_link % poc_id
try:
poc_file = requests.get(link,
headers={'Cookie': cookie} if cookie else None,
timeout=10)
except Exception:
logger.error('Download "%s" [Failed] (Connection Error)' % poc_id)
return False
if _KEYWORDS not in poc_file.content:
logger.error(
'Download "%s" [Failed] (Permission denied or POC not exist)' %
poc_id)
return False
ext = '.py'
restore_path = module_path + poc_id.replace('-', '_') + ext
open(restore_path, 'wb').write(poc_file.content)
logger.critical('Download "%s" [Success]' % poc_id)
return True
