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 02:10:13 +09:00 committed by GitHub
parent 4985383556
commit a7f0ffb03d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server.py
View File

@ -218,9 +218,12 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
# Wait to see if there is more data to transmit # Wait to see if there is more data to transmit
def sendall(sock, data): def sendall(sock, data):
# send first chuck
sock.send(data)
if len(data) < buffer_size: if len(data) < buffer_size:
return return
# send following chunks
buffered = b'' buffered = b''
conn.settimeout(1) conn.settimeout(1)
while True: while True:
@ -231,6 +234,7 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
buffered += chunk buffered += chunk
if proxy_check_filtered(buffered, webserver, port, scheme, method, url): if proxy_check_filtered(buffered, webserver, port, scheme, method, url):
raise Exception("Filtered request") raise Exception("Filtered request")
sock.send(chunk)
if len(buffered) > buffer_size: if len(buffered) > buffer_size:
buffered = buffered[:-buffer_size] # reduce memory usage buffered = buffered[:-buffer_size] # reduce memory usage
except: except: