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 server2.py

Browse Source
This commit is contained in:
Namhyeon Go 2022-10-06 11:05:46 +09:00 committed by GitHub
parent 0741f26cd9
commit a3ef0fe6c9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
server2.py
View File

@ -53,9 +53,11 @@ def conn_string(conn, data, addr):
url = first_line.split()[1] url = first_line.split()[1]
http_pos = url.find(b'://') #Finding the position of :// http_pos = url.find(b'://') #Finding the position of ://
scheme = "http" # check http/https or other protocol
if http_pos == -1: if http_pos == -1:
temp = url temp = url
else: else:
scheme = url[0:http_pos]
temp = url[(http_pos+3):] temp = url[(http_pos+3):]
port_pos = temp.find(b':') port_pos = temp.find(b':')
@ -72,31 +74,31 @@ def conn_string(conn, data, addr):
port = int((temp[(port_pos+1):])[:webserver_pos-port_pos-1]) port = int((temp[(port_pos+1):])[:webserver_pos-port_pos-1])
webserver = temp[:port_pos] webserver = temp[:port_pos]
if url.find(b'https:') == 0: proxy_server(webserver, port, scheme, conn, addr, data)
port = 443
proxy_server(webserver, port, conn, addr, data)
except Exception: except Exception:
pass pass
def proxy_server(webserver, port, conn, addr, data): def proxy_server(webserver, port, scheme, conn, addr, data):
try: try:
headers = { headers = {
"User-Agent": "WebProxyTest" "User-Agent": "WebProxyTest",
} }
data = { data = {
"data": base64.b64encode(data).decode("utf-8"), "data": base64.b64encode(data).decode("utf-8"),
"client": str(addr[0]), "client": str(addr[0]),
"server": webserver.decode("utf-8"), "server": webserver.decode("utf-8"),
"port": str(port), "port": str(port),
"scheme": scheme.decode("utf-8"),
"length": str(len(data)), "length": str(len(data)),
"chunksize": str(buffer_size), "chunksize": str(buffer_size),
"datetime": datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f") "datetime": datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f")
} }
print(">>>>>", data)
relay = requests.post(proxy_url, headers=headers, json=data, stream=True) relay = requests.post(proxy_url, headers=headers, json=data, stream=True)
for chunk in relay.iter_content(chunk_size=buffer_size): for chunk in relay.iter_content(chunk_size=buffer_size):
#print(chunk) print("<<<<<", chunk)
conn.send(chunk) conn.send(chunk)
print("[*] Request Done. %s" % (str(addr[0]))) print("[*] Request Done. %s" % (str(addr[0])))