Gh0st
/
Nettacker
mirror of https://github.com/OWASP/Nettacker.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

bug fix in webui export name and mimetype

This commit is contained in:
Ali Razmjoo 2021-10-01 18:50:00 +02:00
parent 4115a27e25
commit 5a69fb4be1
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
api/api_core.py
View File

@ -104,6 +104,7 @@ def mime_types():
".tiff": "image/tiff", ".tiff": "image/tiff",
".ts": "application/typescript", ".ts": "application/typescript",
".ttf": "font/ttf", ".ttf": "font/ttf",
".txt": "text/plain",
".vsd": "application/vnd.visio", ".vsd": "application/vnd.visio",
".wav": "audio/x-wav", ".wav": "audio/x-wav",
".weba": "audio/webm", ".weba": "audio/webm",

9
api/engine.py
View File

@ -348,11 +348,14 @@ def get_result_content():
filename, file_content = get_scan_result(scan_id) filename, file_content = get_scan_result(scan_id)
return Response( return Response(
file_content, file_content,
mimetype=mime_types().get(filename.split('.')[-1]), mimetype=mime_types().get(
os.path.splitext(filename)[1],
"text/plain"
),
headers={ headers={
'Content-Disposition': 'attachment;filename=' + filename 'Content-Disposition': 'attachment;filename=' + filename.split('/')[-1]
} }
) if file_content not in [404, 500] else filename, content )
@app.route("/results/get_json", methods=["GET"]) @app.route("/results/get_json", methods=["GET"])