Update http_server.py

qeeqbox · Oct 4, 2023 · 40b3499 · 40b3499
1 parent 286f9c1
commit 40b3499
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/honeypots/http_server.py b/honeypots/http_server.py
@@ -137,9 +137,9 @@ def check_bytes(string):
                 if 'fix_get_client_ip' in _q_s.options:
                     with suppress(Exception):
                         raw_headers = dict(request.requestHeaders.getAllRawHeaders())
-                        if b'X-Forwarded-For':
+                        if b'X-Forwarded-For' in raw_headers:
                             client_ip = check_bytes(raw_headers[b'X-Forwarded-For'][0])
-                        elif b'X-Real-IP':
+                        elif b'X-Real-IP' in raw_headers:
                             client_ip = check_bytes(raw_headers[b'X-Real-IP'][0])
 
                 if client_ip == "":