add user agent check

mux99 · Apr 17, 2024 · 1593a16 · 1593a16
1 parent 2523994
commit 1593a16
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/flask_app/app.py b/flask_app/app.py
@@ -17,6 +17,7 @@ def mptcp_status_page():
 
     addr = request.remote_addr
     port = request.environ.get('REMOTE_PORT')
+    user = request.environ.get('HTTP_USER_AGENT')
 
     try:
         conn = check_output(["ss", "-MnH", "dst", f"{addr}", "dport", f"{port}"]).decode("ascii")
@@ -30,6 +31,9 @@ def mptcp_status_page():
         state_message = '[error: ' + str(e) + ']'
         state_class = 'error'
 
+    if user.startswith("curl"):
+        return "You " + state_message + " using MPTCP.\n"
+
     return render_template('index.html', state_message=state_message, state_class=state_class)
 
 @app.route('/stream_audio')