From 2ae1cd1703c4207ffcb117fbc5048174028a183a Mon Sep 17 00:00:00 2001 From: Yan Date: Wed, 15 Jan 2025 18:10:55 -0700 Subject: [PATCH] moving flask template into here --- pwnshop/templates/base/flask.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100644 pwnshop/templates/base/flask.py diff --git a/pwnshop/templates/base/flask.py b/pwnshop/templates/base/flask.py new file mode 100644 index 0000000..6efab31 --- /dev/null +++ b/pwnshop/templates/base/flask.py @@ -0,0 +1,25 @@ +#!/opt/pwn.college/python + +{% block imports %} +{% endblock %} +import flask +import os + +app = flask.Flask(__name__) + +{% block handlers %} +{% endblock %} +{% block initialization %} +{% endblock %} + +{% if challenge.reset_uid %} +os.setuid(os.geteuid()) +{% endif %} +{% if challenge.reset_path %} +os.environ["PATH"] = "/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin" +{% endif %} +app.secret_key = os.urandom(8) +{% if challenge.flask_host %} +app.config['SERVER_NAME'] = f"{{challenge.flask_host}}:{{challenge.flask_port or 80}}" +{% endif %} +app.run("{{challenge.flask_host or "0.0.0.0"}}", {{challenge.flask_port}})