alex/spacedrive
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add simple webapp to upload files

Browse source
This commit is contained in:
atxr 2024-02-26 14:38:09 +01:00
parent 20021ffa49
commit 843449196d
2 changed files with 72 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

57
webapp/app.py Normal file
View file

@ -0,0 +1,57 @@
from flask import Flask, render_template, request
import socket
from struct import pack, unpack
import hashlib
import sys
PORT = 8989
files = []
app = Flask(__name__)
@app.route('/')
def index():
return render_template('index.html')
@app.route('/upload', methods=['POST'])
def upload():
if 'file' not in request.files:
return 'No file part'
file = request.files['file']
if file.filename == '':
return 'No selected file'
buf = file.read()
hash = hashlib.sha256(buf).digest()
# Scan file
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.settimeout(10)
s.connect(("localhost", PORT))
s.send(pack("I", len(buf)))
s.send(buf)
try:
recv_hash = s.recv(32)
status = unpack("B", s.recv(1))
if (hash != recv_hash):
message = 'Error: Mismatching sha256.'
elif (status == 1):
message = "Error: Potential virus found, cannot upload."
else:
message = 'File successfully uploaded'
files.append({'filename': file.filename, 'content':buf})
except:
message = 'Error: Failed to scan file'
s.close()
return message
if __name__ == '__main__':
app.run(debug=True)

15
webapp/templates/index.html Normal file
View file

@ -0,0 +1,15 @@
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>File Upload</title>
</head>
<body>
<h1>File Upload</h1>
<form action="/upload" method="post" enctype="multipart/form-data">
<input type="file" name="file">
<input type="submit" value="Upload">
</form>
</body>
</html>