def starttask():
gmp = openvas.create_connection()
jsonReq = request.get_json(force=True)
task_id = jsonReq.get('task_id')
try:
openvas.start_task(gmp, task_id)
except (ValueError, KeyError, TypeError) as error:
return error
return "Task started"
def getreportbytask():
gmp = openvas.create_connection()
jsonReq = request.get_json(force=True)
task_id = jsonReq.get('task_id')
try:
report = openvas.getReportByTask(gmp, task_id)
except (ValueError, KeyError, TypeError) as error:
return error
return jsonify({"report": report}), 200
def deletetarget():
gmp = openvas.create_connection()
jsonReq = request.get_json(force=True)
target_id = jsonReq.get('id')
try:
openvas.delete_target(gmp, target_id)
except (ValueError, KeyError, TypeError) as error:
return error
return 'Target deleted successfully'
def gettaskstatus():
gmp = openvas.create_connection()
jsonReq = request.get_json(force=True)
task_id = jsonReq.get('task_id')
try:
status, progress = openvas.get_task_status(gmp, task_id)
except (ValueError, KeyError, TypeError) as error:
return error
return jsonify({"status": status, "progress": progress}), 200
def createtask():
gmp = openvas.create_connection()
# target_id = ov.create_target(gmp, ov.hosts, ov.target_name, ov.target_comment)
jsonReq = request.get_json(force=True)
# default scanner id
scanner_id = jsonReq.get('scanner_id', '08b69003-5fc2-4037-a479-93b440211c73') # default if scanner not specified
config_id = jsonReq.get('config_id', 'daba56c8-73ec-11df-a475-002264764cea')
task_comment = jsonReq.get('task_comment', "")
task_id = openvas.create_task(gmp, jsonReq['name'], jsonReq['id'], scanner_id, config_id,
task_comment)
return task_id
def createtarget():
gmp = openvas.create_connection()
# target_id = ov.create_target(gmp, ov.hosts, ov.target_name, ov.target_comment)
jsonReq = request.get_json(force=True)
target_name = jsonReq["name"]
hosts = jsonReq["hosts"]
print(hosts)
target_comment = jsonReq.get("comment", "")
port_list = jsonReq.get("port_list", None)
alive_tests = jsonReq.get("alive_tests", None)
target_id = openvas.create_target(gmp, hosts, target_name, target_comment, port_list, alive_tests)
# return target_id
return target_id
def getreport():
gmp = openvas.create_connection()
jsonReq = request.get_json(force=True)
report_id = jsonReq['id']
report_format_id = jsonReq.get('format_id', 'c402cc3e-b531-11e1-9163-406186ea4fc5')
try:
filename = openvas.get_report_pdf(gmp, report_id, report_format_id)
except (ValueError, KeyError, TypeError) as error:
return error
print(filename)
return send_from_directory('/home/csirt-admin/vulnerability-analysis/reports', filename, as_attachment=True)
def edittask():
gmp = openvas.create_connection()
jsonReq = request.get_json(force=True)
ctask_id = jsonReq.get('task_id')
task_name, task_target_id, task_scanner_id, task_config_id, task_comment = openvas.get_task(gmp, ctask_id)
ctask_name = jsonReq.get('task_name', task_name)
ctask_target_id = jsonReq.get('target_id', task_target_id)
ctask_scanner_id = jsonReq.get('scanner_id', task_scanner_id)
ctask_config_id = jsonReq.get('config_id', task_config_id)
ctask_comment = jsonReq.get('comment', task_comment)
try:
openvas.edit_task(gmp, ctask_id, ctask_name, ctask_target_id, ctask_scanner_id, ctask_config_id, ctask_comment)
except (ValueError, KeyError, TypeError) as error:
return error
return 'Task has been edited succesfully'
def completeflow():
gmp = openvas.create_connection()
jsonReq = request.get_json(force=True)
request_id = jsonReq['request_id']
target_name = jsonReq['target_name']
hosts = jsonReq["hosts"]
significance = jsonReq["significance"]
try:
print("starting openvas")
openvas.completeFlow(gmp, request_id, target_name, hosts, significance)
print("completed openvas")
except (ValueError, KeyError, TypeError) as error:
return error
return "Flow completed successfully"
def edittarget():
gmp = openvas.create_connection()
# target_id = ov.create_target(gmp, ov.hosts, ov.target_name, ov.target_comment)
jsonReq = request.get_json(force=True)
ctarget_id = jsonReq.get('target_id')
target_hosts, target_name, target_comment, target_port_list, target_alive_tests = openvas.get_target(gmp,
ctarget_id)
ctarget_name = jsonReq.get('target_name', target_name)
ctarget_comment = jsonReq.get('target_comment', target_comment)
ctarget_hosts = jsonReq.get('target_hosts', target_hosts)
ctarget_port_list = jsonReq.get('port_list', target_port_list)
ctarget_alive_tests = jsonReq.get('alive_tests', target_alive_tests)
try:
openvas.edit_target(gmp, ctarget_hosts, ctarget_name, ctarget_comment, ctarget_port_list, ctarget_alive_tests)
except (ValueError, KeyError, TypeError) as error:
return error
return 'Target has been edited succesfully'
def gettask():
gmp = openvas.create_connection()
jsonReq = request.get_json(force=True)
ctask_id = jsonReq.get('id')
task = {}
task_name, task_target_id, task_scanner_id, task_config_id, task_comment = openvas.get_task(gmp, ctask_id)
ctask_name = jsonReq.get('task_name', task_name)
ctask_target_id = jsonReq.get('target_id', task_target_id)
ctask_scanner_id = jsonReq.get('scanner_id', task_scanner_id)
ctask_config_id = jsonReq.get('config_id', task_config_id)
ctask_comment = jsonReq.get('comment', task_comment)
task['task_name'] = ctask_name
task['target_id'] = ctask_target_id
task['scanner_id'] = ctask_scanner_id
task['config_id'] = ctask_config_id
task['comment'] = ctask_comment
jsonRes = json.dumps(task, ensure_ascii=False)
return jsonRes
def gettarget():
gmp = openvas.create_connection()
jsonReq = request.get_json(force=True)
ctarget_id = jsonReq.get('id')
target = {}
target_hosts, target_name, target_comment, target_port_list, target_alive_tests = openvas.get_target(gmp,
ctarget_id)
ctarget_name = jsonReq.get('target_name', target_name)
ctarget_comment = jsonReq.get('target_comment', target_comment)
ctarget_hosts = jsonReq.get('target_hosts', target_hosts)
ctarget_port_list = jsonReq.get('port_list', target_port_list)
ctarget_alive_tests = jsonReq.get('alive_tests', target_alive_tests)
target['target_name'] = ctarget_name
target['target_comment'] = ctarget_comment
target['hosts'] = ctarget_hosts
target['port_list'] = ctarget_port_list
target['alive_tests'] = ctarget_alive_tests
jsonRes = json.dumps(target, ensure_ascii=False)
return jsonRes
def getalltasks():
gmp = openvas.create_connection()
all_tasks = openvas.get_all_tasks(gmp)
jsonRes = json.dumps(all_tasks, ensure_ascii=False)
return jsonRes
