gnh1201/caterpillar
1
0
Fork 0
mirror of https://github.com/gnh1201/caterpillar.git synced 2024-11-26 23:41:44 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update server.py

Browse Source
This commit is contained in:
Namhyeon Go 2024-02-23 12:53:44 +09:00 committed by GitHub
parent a200732e84
commit 7b14da3314
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
server.py
View File

@ -90,7 +90,9 @@ def start(): #Main Program
print("\n[*] Graceful Shutdown") print("\n[*] Graceful Shutdown")
sys.exit(1) sys.exit(1)
def conn_string(conn, data, addr): def parse_first_data(data):
parsed_data = (None, None, None, None, None)
try: try:
first_line = data.split(b'\n')[0] first_line = data.split(b'\n')[0]
@ -119,9 +121,17 @@ def conn_string(conn, data, addr):
webserver = temp[:port_pos] webserver = temp[:port_pos]
if port == 443: if port == 443:
scheme = b'https' scheme = b'https'
parsed_data = (webserver, port, scheme, method, url)
except Exception as e: except Exception as e:
conn.close() conn.close()
print("[*] Exception on parsing the header of %s. Cause: %s" % (str(addr[0]), str(e))) print("[*] Exception on parsing the header of %s. Cause: %s" % (str(addr[0]), str(e)))
return parsed_data
def conn_string(conn, data, addr):
webserver, port, scheme, method, url = parse_first_data(data)
if not webserver:
return return
# if it is reverse proxy # if it is reverse proxy
@ -348,14 +358,14 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
chunk = sock.recv(buffer_size) chunk = sock.recv(buffer_size)
if not chunk: if not chunk:
break break
if i == 0 and chunk.find(b'HTTP/1.1 403') == 0:
is_http_403 = True
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(sock, is_ssl) sock_close(sock, is_ssl)
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")
if i == 0 and chunk.find(b'HTTP/1.1 403') == 0:
is_http_403 = True
break
conn.send(chunk) conn.send(chunk)
if len(buffered) > buffer_size*2: if len(buffered) > buffer_size*2:
buffered = buffered[-buffer_size*2:] buffered = buffered[-buffer_size*2:]