Update server.py

server.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 import os
+import re
 
 from blacklist_layer_mobile import BlacklistLayer
 from ai_layer_mobile import AIAnalyzer
@@ -21,7 +22,6 @@ GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY", "")
 blacklist = BlacklistLayer(api_key=GOOGLE_API_KEY)
 print("[OK] Blacklist Layer Ready")
 
-
 ai = AIAnalyzer(model_path="/app")
 print("[OK] AI Analyzer Ready")
 
@@ -32,6 +32,25 @@ class URLRequest(BaseModel):
     url: str
 
 
+def validate_url(url: str) -> bool:
+    # حد أقصى للطول (مثال 255 حرف)
+    if len(url) > 255:
+        return False
+
+    # تعبير نمطي للتأكد من تنسيق URL (http أو https)
+    url_pattern = re.compile(
+        r'^(https?://)'               # بداية http أو https
+        r'([\w\-]+\.)+[\w\-]+'       # دومين
+        r'(:\d+)?'                   # منفذ اختياري
+        r'(/[\w\-.~:/?#[\$@!$&\'()*+,;=%]*)?$'  # مسار URL اختياري
+    )
+
+    if not url_pattern.match(url):
+        return False
+
+    return True
+
+
 @app.get("/")
 def root():
     return {
@@ -46,6 +65,11 @@ def scan(req: URLRequest):
     url = req.url
     print(f"[SCAN] {url}")
 
+    # التحقق من صحة الـ URL قبل المراقبة
+    if not validate_url(url):
+        print("[ERROR] Invalid URL format")
+        return {"error": "Invalid URL format", "safe": False}
+
     is_bad, msg = blacklist.check_url(url)
     print(f"[Layer 1] Blacklist: {msg}")