From 7e3f3d7fb9cee311fd95a84587c7da6b9a314397 Mon Sep 17 00:00:00 2001
From: GREENRAT-K405 <paramkanada2024@gmail.com>
Date: Wed, 11 Feb 2026 21:15:18 +0530
Subject: [PATCH] fix server crash when api missing

---
 server/Server.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/server/Server.py b/server/Server.py
index da2b36a..2e89d9f 100644
--- a/server/Server.py
+++ b/server/Server.py
@@ -34,6 +34,11 @@ def hello_world():
 @app.route('/init/<dir>', methods=['POST']) # POST has u and ym
 def init(dir=None):
     apikey = request.args.get('apikey')
+    
+    # validate apikey presence to prevent TypeError
+    if not apikey:
+        return "Missing API key", 400
+
     dirname = dir + "_" + apikey
     logging.info("Init Request received for: %s", dirname)
     # Get the file from the POST request
@@ -57,6 +62,11 @@ def init(dir=None):
 def ctl(dir=None):
     return_file = request.args.get('fetch')
     apikey = request.args.get('apikey')
+
+    # validate apikey presence to prevent TypeError
+    if not apikey:
+        return "Missing API key", 400
+
     dirname = dir + "_" + apikey
     logging.info("[CTL] Request received for: %s", dirname)
     f1 = request.files['file1']
@@ -80,6 +90,11 @@ def ctl(dir=None):
 def pm(dir=None):
     return_file = request.args.get('fetch')
     apikey = request.args.get('apikey')
+
+    # validate apikey presence to prevent TypeError
+    if not apikey:
+        return "Missing API key", 400
+
     dirname = dir + "_" + apikey
     logging.info("[PM] Request received for: %s", dirname)
     f1 = request.files['file1']
@@ -102,6 +117,11 @@ def pm(dir=None):
 @app.route('/cleanup/<dir>', methods=['DELETE'])
 def cleanup(dir=None):
     apikey = request.args.get('apikey')
+
+    # validate apikey presence to prevent TypeError
+    if not apikey:
+        return "Missing API key", 400
+
     dirname = dir + "_" + apikey
     logging.info("Cleanup Request received for: %s", dirname)
     if os.path.exists(WORKDIR + secure_filename(dirname)):