gnh1201/caterpillar
1
0
Fork 0
mirror of https://github.com/gnh1201/caterpillar.git synced 2025-09-07 18:31:10 +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) return (conn, data)
def proxy_check_filtered(response): def proxy_check_filtered(response, webserver, port):
filtered = response.find(b'@misskey.io') > -1 filtered = response.find(b'@misskey.io') > -1
if filtered: 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 return filtered
@ -169,7 +172,7 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
if not chunk: if not chunk:
break break
response += chunk response += chunk
if proxy_check_filtered(response): if proxy_check_filtered(response, webserver, port):
break break
conn.send(chunk) conn.send(chunk)
i += 1 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) relay = requests.post(server_url, headers=proxy_data['headers'], data=raw_data, stream=True)
for chunk in relay.iter_content(chunk_size=buffer_size): for chunk in relay.iter_content(chunk_size=buffer_size):
response += chuck response += chuck
if not proxy_check_filtered(response): if not proxy_check_filtered(response, webserver, port):
break break
conn.send(chunk) conn.send(chunk)
i += 1 i += 1