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

Update server.py

Browse Source
This commit is contained in:
Namhyeon Go 2024-02-19 03:24:22 +09:00 committed by GitHub
parent c29de5e44a
commit 9dc898f907
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
server.py
View File

@ -209,6 +209,7 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
def sendall(sock, conn, data): def sendall(sock, conn, data):
# send first chuck # send first chuck
if proxy_check_filtered(data, webserver, port, scheme, method, url): if proxy_check_filtered(data, webserver, port, scheme, method, url):
sock.close()
raise Exception("Filtered request") raise Exception("Filtered request")
sock.send(data) sock.send(data)
if len(data) < buffer_size: if len(data) < buffer_size:
@ -224,6 +225,7 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
break break
buffered += chunk buffered += chunk
if proxy_check_filtered(buffered, webserver, port, scheme, method, url): if proxy_check_filtered(buffered, webserver, port, scheme, method, url):
sock.close()
raise Exception("Filtered request") raise Exception("Filtered request")
sock.send(chunk) sock.send(chunk)
if len(buffered) > buffer_size*2: if len(buffered) > buffer_size*2:
@ -257,6 +259,7 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
break break
buffered += chunk buffered += chunk
if proxy_check_filtered(buffered, webserver, port, scheme, method, url): if proxy_check_filtered(buffered, webserver, port, scheme, method, url):
sock.close()
add_filtered_host(webserver.decode(client_encoding), '127.0.0.1') add_filtered_host(webserver.decode(client_encoding), '127.0.0.1')
raise Exception("Filtered response") raise Exception("Filtered response")
conn.send(chunk) conn.send(chunk)