mirror of https://github.com/EasyCTF/librectf
54 lines
1.6 KiB
Python
Executable File
54 lines
1.6 KiB
Python
Executable File
import hashlib
|
|
import json
|
|
import logging
|
|
import os
|
|
|
|
from flask import Flask, abort, request, send_file
|
|
|
|
app = Flask(__name__)
|
|
app.config["UPLOAD_FOLDER"] = os.getenv("UPLOAD_FOLDER", "/usr/share/nginx/html")
|
|
|
|
if not os.path.exists(app.config["UPLOAD_FOLDER"]):
|
|
os.makedirs(app.config["UPLOAD_FOLDER"])
|
|
|
|
|
|
@app.route("/")
|
|
def index():
|
|
return "You shouldn't be here."
|
|
|
|
|
|
@app.route("/save", methods=["POST"])
|
|
def save():
|
|
if "file" not in request.files:
|
|
return "no file uploaded", 400
|
|
file = request.files["file"]
|
|
if file.filename == "":
|
|
return "no filename found", 400
|
|
filename = hashlib.sha256(file.read()).hexdigest()
|
|
file.seek(0)
|
|
if "filename" in request.form:
|
|
name, ext = json.loads(request.form["filename"])
|
|
filename = "%s.%s.%s" % (name, filename, ext)
|
|
else:
|
|
if "prefix" in request.form:
|
|
filename = "%s%s" % (request.form["prefix"], filename)
|
|
if "suffix" in request.form:
|
|
filename = "%s%s" % (filename, request.form["suffix"])
|
|
file.save(os.path.join(app.config["UPLOAD_FOLDER"], filename))
|
|
return filename
|
|
|
|
|
|
# This route should be used for debugging filestore locally.
|
|
@app.route("/static/<string:path>")
|
|
def serve(path):
|
|
path = os.path.join(app.config["UPLOAD_FOLDER"], path)
|
|
if not os.path.exists(path):
|
|
return abort(404)
|
|
return send_file(path)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
logging.warning("Uploading to {}".format(app.config["UPLOAD_FOLDER"]))
|
|
port = int(os.getenv("FILESTORE_PORT", "8001"))
|
|
app.run(use_debugger=True, use_reloader=True, port=port, host="0.0.0.0")
|