def task_file_create():
"""Endpoint for full analysis task."""
if 'file' not in request.files:
# no file
res = ErrorAPIResponse(2020).to_dict()
return jsonify(res), 400
file = request.files['file']
if file.filename == '':
# noname file
res = ErrorAPIResponse(2021).to_dict()
return jsonify(res), 400
# get pretty print parameter
pretty = False
if 'pretty' in request.form:
pretty = request.form['pretty']
if pretty not in ('true', 'false'):
res = ErrorAPIResponse(2000).to_dict()
return jsonify(res), 400
# ger exec time parameter
exec_time = 20
if 'exec_time' in request.form:
try:
exec_time = int(request.form['exec_time'])
except ValueError:
res = ErrorAPIResponse(2022).to_dict()
return jsonify(res), 400
if (
exec_time < dynamic_config['min_exectime']
or exec_time > dynamic_config['max_exectime']
):
res = ErrorAPIResponse(2022).to_dict()
return jsonify(res), 400
task_id = uuid()
# prepare directory and save file
os.mkdir(f'{storage_path}/{task_id}')
file_path = f'{storage_path}/{task_id}/{file.filename}'
file.save(file_path)
# run analysis
args = (file_path,)
kwargs = {'pretty': pretty, 'exec_time': exec_time}
tasks.full_analysis.apply_async(args, kwargs, task_id=task_id)
res = {
'task_id': task_id
}
return jsonify(res)
def download_console_output(id):
"""Download binary's console output."""
log_file = f'{storage_path}/{id}/prog.log'
if not os.path.isfile(log_file):
res = ErrorAPIResponse(1006).to_dict()
return jsonify(res), 404
return send_file(log_file, as_attachment=True)
def download_machine_log(id):
"""Download machine log."""
log_file = f'{storage_path}/{id}/machine.log'
if not os.path.isfile(log_file):
res = ErrorAPIResponse(1005).to_dict()
return jsonify(res), 404
return send_file(log_file, as_attachment=True)
def download_json(id):
"""Download json report (serve file)."""
json_file = f'{storage_path}/{id}/report.json'
if not os.path.isfile(json_file):
res = ErrorAPIResponse(1004).to_dict()
return jsonify(res), 404
return send_file(json_file, as_attachment=True)
def download_pcap(id):
"""Get analysis pcap."""
pcaps = glob.glob(f'{storage_path}/{id}/*.pcap')
if len(pcaps) == 0:
res = ErrorAPIResponse(1003).to_dict()
return jsonify(res), 404
return send_file(pcaps[0], as_attachment=True)
def get_report(id):
"""Get task report endpoint."""
task = celery_app.AsyncResult(id)
if task.state != 'SUCCESS':
res = ErrorAPIResponse(1000).to_dict()
return jsonify(res), 404
return send_file(f'{storage_path}/{id}/report.json')
def list_pending_tasks():
"""Lists tasks with status 'PENDING'"""
limit = 100
if 'limit' in request.args:
try:
limit = int(request.args['limit'])
except ValueError:
res = ErrorAPIResponse(3000).to_dict()
return jsonify(res), 400
if limit < 1:
res = ErrorAPIResponse(3000).to_dict()
return jsonify(res), 400
i = celery_app.control.inspect()
pending = i.reserved()
return jsonify(pending)
def list_finished_tasks():
"""List tasks with status 'SUCCESS'"""
limit = 100
if 'limit' in request.args:
try:
limit = int(request.args['limit'])
except ValueError:
res = ErrorAPIResponse(3000).to_dict()
return jsonify(res), 400
if limit < 1:
res = ErrorAPIResponse(3000).to_dict()
return jsonify(res), 400
# unitialized db - no tasks
if not engine.dialect.has_table(engine, 'celery_taskmeta'):
return jsonify([])
try:
session = Session()
tasks = session.query(Task).filter(Task.status == 'SUCCESS').order_by(
Task.date_done.desc()).limit(limit)
except (exc.SQLAlchemyError, exc.OperationalError):
session.rollback()
finally:
session.close()
res = []
for task in tasks:
res.append(task.to_dict())
return jsonify(res)
def task_pcap_create():
"""Endpoint for network/pcap analysis task."""
if 'pcap' not in request.files:
# no pcap file
res = ErrorAPIResponse(2010).to_dict()
return jsonify(res), 400
pcap_file = request.files['pcap']
if pcap_file.filename == '':
# noname file
res = ErrorAPIResponse(2011).to_dict()
return jsonify(res), 400
# get pretty print parameter
pretty = False
if 'pretty' in request.form:
pretty = request.form['pretty']
if pretty not in ('true', 'false'):
res = ErrorAPIResponse(2000).to_dict()
return jsonify(res), 400
task_id = uuid()
# prepare directory and save pcap
os.mkdir(f'{storage_path}/{task_id}')
pcap_path = f'{storage_path}/{task_id}/{pcap_file.filename}'
pcap_file.save(pcap_path)
# run pcap analysis
args = (pcap_path,)
kwargs = {'pretty': pretty}
tasks.pcap_analysis.apply_async(args, kwargs, task_id=task_id)
res = {
'task_id': task_id
}
return jsonify(res)
def list_tasks(request_args, status=None):
"""General listing tasks function.
:param request_args: Arguments passed to request.
:param status: Status filter (e.g. 'SUCCESS').
"""
limit = 1000
if 'limit' in request_args:
try:
limit = int(request_args['limit'])
except ValueError:
res = ErrorAPIResponse(3000).to_dict()
return jsonify(res), 400
if limit < 1:
res = ErrorAPIResponse(3000).to_dict()
return jsonify(res), 400
# unitialized db - no tasks
if not engine.dialect.has_table(engine, 'celery_taskmeta'):
return jsonify([])
try:
session = Session()
if status:
tasks = session.query(Task).filter(Task.status == status).order_by(
Task.date_done.desc()).limit(limit)
else:
tasks = session.query(Task).order_by(
Task.date_done.desc()).limit(limit)
except (exc.SQLAlchemyError, exc.OperationalError):
session.rollback()
finally:
session.close()
res = []
for task in tasks:
res.append(task.to_dict())
return jsonify(res)
def internal_server_error(e):
"""Page 500."""
res = ErrorAPIResponse(1001).to_dict()
return jsonify(res), 500
def page_not_found(e):
"""Page 404."""
res = ErrorAPIResponse(1000).to_dict()
return jsonify(res), 404