From c526bcb6e00465bd41b96e6ab55d2a61dd97d837 Mon Sep 17 00:00:00 2001 From: "Namhyeon, Go" Date: Wed, 6 Mar 2024 11:38:53 +0900 Subject: [PATCH] Update server.py --- server.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/server.py b/server.py index f8ec6fd..b11a207 100644 --- a/server.py +++ b/server.py @@ -231,7 +231,7 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data): except: break - # do response + # localhost mode if server_url == "localhost": sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) @@ -303,7 +303,6 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data): # stateful mode elif server_connection_type == "stateful": - proxy_data = { 'headers': { "User-Agent": "php-httpproxy/0.1.5 (Client; Python " + python_version() + "; abuse@catswords.net)", @@ -387,8 +386,8 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data): print("[*] Received %s chunks. (%s bytes per chunk)" % (str(i), str(buffer_size))) - else: - # stateless mode + # stateless mode + elif server_connection_type == "stateless": proxy_data = { 'headers': { "User-Agent": "php-httpproxy/0.1.5 (Client; Python " + python_version() + "; abuse@catswords.net)", @@ -425,6 +424,10 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data): print("[*] Received %s chunks. (%s bytes per chunk)" % (str(i), str(buffer_size))) + # nothing at all + else: + raise Exception("Unsupported connection type") + print("[*] Request and received. Done. %s" % (str(addr[0]))) conn.close() except Exception as e: