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 01:50:43 +09:00 committed by GitHub
parent 5f9c5e8c35
commit 54a959f14a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
server.py
View File

@ -234,8 +234,7 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
conn.close() conn.close()
return return
# make response data # do response
response = b''
if server_url == "localhost": if server_url == "localhost":
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
@ -259,18 +258,14 @@ 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):
add_filtered_host(webserver.decode(client_encoding), '127.0.0.1')
conn.sendall(b"HTTP/1.1 403 Forbidden\n\n{\"status\":403}")
break break
conn.send(chunk) conn.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
i += 1 i += 1
if not proxy_check_filtered(response, webserver, port, scheme, method, url):
conn.sendall(response)
else:
add_filtered_host(webserver.decode(client_encoding), '127.0.0.1')
conn.sendall(b"HTTP/1.1 403 Forbidden\n\n{\"status\":403}")
print("[*] Received %s chunks. (%s bytes per chunk)" % (str(i), str(buffer_size))) print("[*] Received %s chunks. (%s bytes per chunk)" % (str(i), str(buffer_size)))
else: else:
@ -301,18 +296,14 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
for chunk in relay.iter_content(chunk_size=buffer_size): for chunk in relay.iter_content(chunk_size=buffer_size):
buffered += chunk buffered += chunk
if proxy_check_filtered(buffered, webserver, port, scheme, method, url): if proxy_check_filtered(buffered, webserver, port, scheme, method, url):
add_filtered_host(webserver.decode(client_encoding), '127.0.0.1')
conn.sendall(b"HTTP/1.1 403 Forbidden\n\n{\"status\":403}")
break break
conn.send(chunk) conn.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
i += 1 i += 1
if not proxy_check_filtered(response, webserver, port, scheme, method, url):
conn.sendall(response)
else:
add_filtered_host(webserver.decode(client_encoding), '127.0.0.1')
conn.sendall(b"HTTP/1.1 403 Forbidden\n\n{\"status\":403}")
print("[*] Received %s chunks. (%s bytes per chunk)" % (str(i), str(buffer_size))) print("[*] Received %s chunks. (%s bytes per chunk)" % (str(i), str(buffer_size)))
print("[*] Request and received. Done. %s" % (str(addr[0]))) print("[*] Request and received. Done. %s" % (str(addr[0])))