增加xff兼容性

ljxi · Oct 23, 2024 · 8d83671 · 8d83671
1 parent 3fe18c2
commit 8d83671
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/docker/main.py b/docker/main.py
@@ -194,7 +194,11 @@ def query():
 @app.get("/")
 def api(request: Request, ip: str = None):
     if not ip:
-        ip = request.headers.get("x-forwarded-for") or request.headers.get("x-real-ip") or request.client.host
+        xff = request.headers.get("x-forwarded-for")
+        if xff:
+            ip = xff.split(",")[0]
+        else:
+            ip = request.headers.get("x-real-ip") or request.client.host
     return get_ip_info(ip.strip())
 
 @app.get("/{ip}")