gnh1201/caterpillar
1
0
Fork 0
mirror of https://github.com/gnh1201/caterpillar.git synced 2025-09-06 01:41: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 14:40:33 +09:00 committed by GitHub
parent dd79eb6e3d
commit cdb87f15ff
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
server.py
View File

@ -127,6 +127,10 @@ def proxy_connect(webserver, conn):
return (conn, data) return (conn, data)
def proxy_filter(response):
# todo
pass
def proxy_server(webserver, port, scheme, method, url, conn, addr, data): def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
try: try:
print("[*] Started the request. %s" % (str(addr[0]))) print("[*] Started the request. %s" % (str(addr[0])))
@ -137,13 +141,14 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
except Exception as e: except Exception as e:
raise Exception("SSL negotiation failed. %s" % (str(e))) raise Exception("SSL negotiation failed. %s" % (str(e)))
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)
context = ssl.create_default_context() context = ssl.create_default_context()
context.check_hostname = False context.check_hostname = False
context.verify_mode = ssl.CERT_NONE context.verify_mode = ssl.CERT_NONE
ssl_sock = context.wrap_socket(sock, server_hostname=webserver) ssl_sock = context.wrap_socket(sock, server_hostname=webserver)
ssl_sock.connect((webserver, port)) ssl_sock.connect((webserver, port))
ssl_sock.sendall(data) ssl_sock.sendall(data)
@ -153,8 +158,10 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
chunk = ssl_sock.recv(buffer_size) chunk = ssl_sock.recv(buffer_size)
if not chunk: if not chunk:
break break
response += chuck
proxy_filter(response)
conn.send(chunk) conn.send(chunk)
i = i + 1 i += 1
print("[*] Received %s chucks. (%s bytes per chuck)" % (str(i), str(buffer_size))) print("[*] Received %s chucks. (%s bytes per chuck)" % (str(i), str(buffer_size)))
else: else:
@ -175,15 +182,17 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
} }
} }
raw_data = json.dumps(proxy_data['data']) raw_data = json.dumps(proxy_data['data'])
print("[*] Sending %s bytes..." % (str(len(raw_data)))) print("[*] Sending %s bytes..." % (str(len(raw_data))))
i = 0 i = 0
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
proxy_filter(response)
conn.send(chunk) conn.send(chunk)
i = i + 1 i += 1
print("[*] Received %s chucks. (%s bytes per chuck)" % (str(i), str(buffer_size))) print("[*] Received %s chucks. (%s bytes per chuck)" % (str(i), str(buffer_size)))
print("[*] Request and received. Done. %s" % (str(addr[0]))) print("[*] Request and received. Done. %s" % (str(addr[0])))