TiagoRG
/
sio-2425
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add file checksum validation

This commit is contained in:
Tiago Garcia 2024-11-20 16:32:50 +00:00
parent 7fd5f1bcd8
commit 07ec667483
Signed by: TiagoRG
GPG Key ID: DFCD48E3F420DB42
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
delivery1/server/requirements.txt
View File

@ -1,4 +1,3 @@
cryptography cryptography
flask flask
flask_sqlalchemy flask_sqlalchemy
pytest

7
delivery1/server/routes/file.py
View File

@ -83,6 +83,13 @@ def file_upload_content():
if not file: if not file:
return jsonify({"error": "Invalid file data"}), 400 return jsonify({"error": "Invalid file data"}), 400
file_sum = request.headers.get("File-Checksum")
if not file_sum:
return jsonify({"error": "No file checksum provided"}), 400
if file_sum != utils.get_hash(file.stream):
return jsonify({"error": "File checksum mismatch"}), 400
file = upload_service.write_file(session_token, file.stream) file = upload_service.write_file(session_token, file.stream)
if isinstance(file, tuple): if isinstance(file, tuple):
return file return file