Update server.py

This commit is contained in:
Namhyeon Go 2024-02-27 16:35:03 +09:00 committed by GitHub
parent 83737a2e9f
commit 19907e2834
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -478,7 +478,7 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
# get client address
try:
_, query_data = jsonrpc2_encode('get_client_address')
query = requests.post(server_url, headers=proxy_data['headers'], data=query_data, timeout=1)
query = requests.post(server_url, headers=proxy_data['headers'], data=query_data, timeout=30)
if query.status_code == 200:
result = query.json()['result']
proxy_data['data']['client_address'] = result['client_address']
@ -489,7 +489,7 @@ def proxy_server(webserver, port, scheme, method, url, conn, addr, data):
# build a tunnel
try:
id, raw_data = jsonrpc2_encode('relay_connect', proxy_data['data'])
relay = requests.post(server_url, headers=proxy_data['headers'], data=raw_data, stream=True, timeout=1)
relay = requests.post(server_url, headers=proxy_data['headers'], data=raw_data, stream=True, timeout=2)
for chunk in relay.iter_content(chunk_size=buffer_size):
print (chunk)
except requests.exceptions.ReadTimeout as e: