gnh1201/caterpillar
1
0
Fork 0
mirror of https://github.com/gnh1201/caterpillar.git synced 2025-02-06 15:05: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-17 18:07:22 +09:00 committed by GitHub
parent 32d0dd7e2c
commit 3be66e9b72
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
server.py
View File

@ -128,11 +128,14 @@ def proxy_connect(webserver, conn):
return (conn, data)
def proxy_check_filtered(response):
def proxy_check_filtered(response, webserver, port):
filtered = response.find(b'@misskey.io') > -1
if filtered:
print ("[*] filtered: %s" % (response.decode(client_encoding)))
print ("[*] filtered from %s:%s" % (webserver.decode(client_encoding), str(port)))
print ("[*] ====== start response data =====")
print ("%s" % (response.decode(client_encoding)))
print ("[*] ====== end response data =====")
return filtered
@ -169,7 +172,7 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
if not chunk:
break
response += chunk
if proxy_check_filtered(response):
if proxy_check_filtered(response, webserver, port):
break
conn.send(chunk)
i += 1
@ -202,7 +205,7 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
relay = requests.post(server_url, headers=proxy_data['headers'], data=raw_data, stream=True)
for chunk in relay.iter_content(chunk_size=buffer_size):
response += chuck
if not proxy_check_filtered(response):
if not proxy_check_filtered(response, webserver, port):
break
conn.send(chunk)
i += 1