gnh1201/caterpillar
1
0
Fork 0
mirror of https://github.com/gnh1201/caterpillar.git synced 2025-09-06 01:41:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update server.py

Browse Source
This commit is contained in:
Namhyeon Go 2024-02-18 06:23:58 +09:00 committed by GitHub
parent 7e17fb7fe6
commit 288ae6b883
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
server.py
View File

@ -192,7 +192,19 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
except Exception as e: except Exception as e:
raise Exception("SSL negotiation failed. (%s:%s) %s" % (webserver.decode(client_encoding), str(port), str(e))) raise Exception("SSL negotiation failed. (%s:%s) %s" % (webserver.decode(client_encoding), str(port), str(e)))
# check request data # Wait to see if there is more data to transmit
if len(data) == buffer_size:
conn.settimeout(5)
while True:
try:
chunk = conn.recv(buffer_size)
if not chunk:
break
data += chunk
except:
break
# check requested data
if proxy_check_filtered(data, webserver, port, scheme, method, url): if proxy_check_filtered(data, webserver, port, scheme, method, url):
conn.sendall(b"HTTP/1.1 403 Forbidden\n\n{\"status\":403}") conn.sendall(b"HTTP/1.1 403 Forbidden\n\n{\"status\":403}")
conn.close() conn.close()