Update server.py

This commit is contained in:
Namhyeon Go 2024-02-17 20:13:17 +09:00 committed by GitHub
parent 6cef3d688f
commit 24285dbff6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -143,9 +143,16 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
try:
print("[*] Started the request. %s" % (str(addr[0])))
retry = False
try:
if scheme in [b'https', b'tls', b'ssl'] and method == b'CONNECT':
conn, data = proxy_connect(webserver, conn)
except OSError as e:
if not retry:
print ("[*] Retrying SSL negotiation..." % (webserver.decode(client_encoding), str(port), str(e)))
retry = True
else:
raise Exception("SSL negotiation failed. (%s:%s) %s" % (webserver.decode(client_encoding), str(port), str(e)))
except Exception as e:
raise Exception("SSL negotiation failed. (%s:%s) %s" % (webserver.decode(client_encoding), str(port), str(e)))