def unregister_all(_signal, _frame):
if 'true' == os.environ.get('WERKZEUG_RUN_MAIN'):
# noinspection PyBroadException
try:
# somehow, calling the function in BurpExtensionApi does not seem to work here
# -> re-implement
client = HTTPConnection(BurpExtensionApi.TARGET_HOST,
BurpExtensionApi.TARGET_PORT)
client.request(
'POST', '/resetall', None, {
'Content-type': 'application/json',
'Authorization': BurpExtensionApi.AUTH_TOKEN
})
data = client.getresponse().read()
json_response = json.loads(data)
if 'ok' == json_response['status']:
Registry.unregister_all()
LOGGER.info(
'Un-Registration for all extensions successful')
else:
LOGGER.warning('Error while un-registering all extensions')
except Exception:
LOGGER.exception('Error while cleaning up')
finally:
# send signal to old handler
signal.signal(signal.SIGINT, ExtensionLoader.old_handler)
# try to exit
exit(0)
def get_intruderpayloadgenerator_reset():
reg_id = request.args.get('rid')
ipg = Registry.get_by_id(reg_id)
if ipg is not None:
ipg.reset()
return ''
def get_intruderpayloadgenerator_hasmorepayloads():
reg_id = request.args.get('rid')
ipg = Registry.get_by_id(reg_id)
has_more_payloads = False
if ipg is not None:
has_more_payloads = ipg.has_more_payloads()
return jsonify(has_more_payloads)
def get_intruderpayloadgenerator_getnextpayloads():
reg_id = request.args.get('rid')
ipg = Registry.get_by_id(reg_id)
updates = []
if ipg is not None:
updates = ipg.get_next_payload_update(request.args.get('payload'))
return jsonify([u.get_data() for u in updates])
def scannercheck_passive():
reg_id = request.args.get('rid')
action_handler = Registry.get_by_id(reg_id)
scan_issues = []
if action_handler is not None:
scanner_check_request = request.json
wrapped_message = WrappedMessage(scanner_check_request)
scan_issues = action_handler.get_passive_scan_issues(wrapped_message)
return jsonify([s.get_data() for s in scan_issues])
def get_register_config():
reg_id = request.args.get('rid')
processor = Registry.get_by_id(reg_id)
data = {}
if processor is not None:
reg_data = None
if request.json:
reg_data = request.json
data = processor.get_register_config(reg_data)
return jsonify(data)
def httplistener_processmsg():
reg_id = request.args.get('rid')
http_listener = Registry.get_by_id(reg_id)
updates = []
if http_listener is not None:
# AnalyzedMessage
analyzed_message = request.json
wrapped_message = WrappedMessage(analyzed_message)
updates = http_listener.process_http_message(wrapped_message)
return jsonify([u.get_data() for u in updates])
def proxylistener_processmsg():
reg_id = request.args.get('rid')
proxy_listener = Registry.get_by_id(reg_id)
updates = []
if proxy_listener is not None:
# InterceptedMessage
intercepted_message = request.json
wrapped_message = WrappedMessage(intercepted_message)
updates = proxy_listener.process_proxy_message(wrapped_message)
return jsonify([u.get_data() for u in updates])
def get_processed_message_display():
reg_id = request.args.get('rid')
message_processor = Registry.get_by_id(reg_id)
updates = []
if message_processor is not None:
# AnalyzedMessage
analyzed_message = request.json
wrapped_message = WrappedMessage(analyzed_message)
updates = message_processor.get_readable_content_updates(
wrapped_message)
return jsonify([u.get_data() for u in updates])
def scannercheck_consolidate():
reg_id = request.args.get('rid')
action_handler = Registry.get_by_id(reg_id)
# new and old issue is default
result = 0
if action_handler is not None:
consolidate_request = request.json
result = action_handler.get_consolidated_issues_result(
consolidate_request['existingIssue'],
consolidate_request['newIssue'])
return jsonify(result)
def sessionhandlingaction_perform():
reg_id = request.args.get('rid')
action_handler = Registry.get_by_id(reg_id)
updates = []
if action_handler is not None:
# SessionHandlingActionRequest
session_handling_action_request = request.json
wrapped_message = WrappedMessage(session_handling_action_request)
macros = []
for macro in request.json.get('macroItems'):
macros.append(WrappedMessage(macro))
updates = action_handler.perform_action(wrapped_message, macros)
return jsonify([u.get_data() for u in updates])
def get_scanpoint_request():
reg_id = request.args.get('rid')
ip_processor = Registry.get_by_id(reg_id)
updates = []
if ip_processor is not None:
# InsertionPointRequest
process_payload_request = request.json
payload = process_payload_request['payload']
req = process_payload_request['request']
analyzed_request = process_payload_request['analyzedRequest']
name = process_payload_request['name']
updates = ip_processor.build_request_update(req, analyzed_request,
payload, name)
return jsonify([u.get_data() for u in updates])
def get_processed_payload():
reg_id = request.args.get('rid')
payload_processor = Registry.get_by_id(reg_id)
updates = []
if payload_processor is not None:
# ProcessPayloadRequest
process_payload_request = request.json
current_payload = base64.b64decode(
process_payload_request['currentPayload'])
original_payload = base64.b64decode(
process_payload_request['originalPayload'])
base_value = base64.b64decode(process_payload_request['baseValue'])
updates = payload_processor.get_processed_payload_update(
current_payload, original_payload, base_value)
return jsonify([u.get_data() for u in updates])
def register(self):
LOGGER.info('Registering {}'.format(self.NAME))
self.reg_id = Registry.register(self.NAME, self)
if not self._do_register_request():
Registry.unregister(self.NAME)
def register():
Registry.register()
def unregister():
Registry.unregister()
def unregister(self):
self._do_unregister_request()
Registry.unregister(self.NAME)
