AI 自测结束
This commit is contained in:
cheney
+116
-108
@@ -2,140 +2,146 @@
|
||||
import socket
|
||||
import threading
|
||||
import sys
|
||||
import time
|
||||
|
||||
def tunnel_thread(src, dst):
|
||||
"""Tunnel thread: continuously forward data"""
|
||||
try:
|
||||
while True:
|
||||
data = src.recv(8192)
|
||||
if not data:
|
||||
break
|
||||
dst.sendall(data)
|
||||
except Exception as e:
|
||||
pass
|
||||
TARGET_HOST = 'zentao.sunyard.com.cn'
|
||||
TARGET_PORT = 9788
|
||||
|
||||
def handle_b_device(sock):
|
||||
"""Handle communication with device B, establish tunnel to target website"""
|
||||
print("\n[System] Connected to device B")
|
||||
def log_info(source, message):
|
||||
"""Log info message"""
|
||||
timestamp = time.strftime("%Y-%m-%d %H:%M:%S")
|
||||
print("[%s] [INFO] [%s] %s" % (timestamp, source, message))
|
||||
|
||||
def log_error(source, message):
|
||||
"""Log error message"""
|
||||
timestamp = time.strftime("%Y-%m-%d %H:%M:%S")
|
||||
print("[%s] [ERROR] [%s] %s" % (timestamp, source, message))
|
||||
|
||||
def handle_request(b_socket, request_data):
|
||||
"""Handle a single request from device B"""
|
||||
# Parse HTTP request
|
||||
request_lines = request_data.split(b"\r\n")
|
||||
if request_lines:
|
||||
first_line = request_lines[0].decode('utf-8', errors='ignore')
|
||||
parts = first_line.split()
|
||||
if len(parts) >= 2:
|
||||
method = parts[0]
|
||||
url = parts[1]
|
||||
log_info("System", "Received request: %s %s" % (method, url))
|
||||
|
||||
# Connect to target
|
||||
target_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
target_sock.settimeout(30)
|
||||
|
||||
try:
|
||||
while True:
|
||||
request = b""
|
||||
target_sock.connect((TARGET_HOST, TARGET_PORT))
|
||||
log_info("System", "Connected to target: %s:%d" % (TARGET_HOST, TARGET_PORT))
|
||||
|
||||
if method == "CONNECT":
|
||||
# Send CONNECT response
|
||||
response = b"HTTP/1.1 200 Connection Established\r\n\r\n"
|
||||
b_socket.sendall(response)
|
||||
log_info("System", "Sent CONNECT response")
|
||||
|
||||
# Establish tunnel
|
||||
def forward(src, dst):
|
||||
try:
|
||||
while True:
|
||||
data = src.recv(8192)
|
||||
if not data:
|
||||
break
|
||||
dst.sendall(data)
|
||||
except:
|
||||
pass
|
||||
|
||||
t1 = threading.Thread(target=forward, args=(b_socket, target_sock))
|
||||
t2 = threading.Thread(target=forward, args=(target_sock, b_socket))
|
||||
t1.start()
|
||||
t2.start()
|
||||
t1.join()
|
||||
t2.join()
|
||||
|
||||
else:
|
||||
# Forward request and response
|
||||
target_sock.sendall(request_data)
|
||||
|
||||
while True:
|
||||
response = target_sock.recv(8192)
|
||||
if not response:
|
||||
break
|
||||
b_socket.sendall(response)
|
||||
|
||||
return True
|
||||
|
||||
except ConnectionRefusedError:
|
||||
log_error("System", "Cannot connect to target: %s:%d" % (TARGET_HOST, TARGET_PORT))
|
||||
error_msg = b"HTTP/1.1 502 Bad Gateway\r\nContent-Type: text/plain\r\n\r\nCannot connect to target\r\n"
|
||||
b_socket.sendall(error_msg)
|
||||
return False
|
||||
except socket.timeout:
|
||||
log_error("System", "Target connection timeout")
|
||||
error_msg = b"HTTP/1.1 504 Gateway Timeout\r\nContent-Type: text/plain\r\n\r\nConnection timeout\r\n"
|
||||
b_socket.sendall(error_msg)
|
||||
return False
|
||||
except Exception as e:
|
||||
log_error("System", "Error handling request: %s" % str(e))
|
||||
return False
|
||||
finally:
|
||||
target_sock.close()
|
||||
|
||||
def listen_for_requests(sock):
|
||||
"""Listen for requests from device B"""
|
||||
try:
|
||||
while True:
|
||||
# Wait for request with long timeout
|
||||
sock.settimeout(600) # 10 minutes
|
||||
|
||||
request_data = b""
|
||||
while b"\r\n\r\n" not in request_data:
|
||||
chunk = sock.recv(8192)
|
||||
if not chunk:
|
||||
print("[Device B] Disconnected")
|
||||
log_info("System", "Device B disconnected")
|
||||
return
|
||||
request += chunk
|
||||
if b"\r\n\r\n" in request:
|
||||
break
|
||||
request_data += chunk
|
||||
|
||||
print("[System] Received request:", len(request), "bytes")
|
||||
# Handle request in a separate thread
|
||||
t = threading.Thread(target=handle_request, args=(sock, request_data))
|
||||
t.start()
|
||||
|
||||
target_host = None
|
||||
target_port = 443
|
||||
|
||||
if request.startswith(b"CONNECT"):
|
||||
try:
|
||||
lines = request.decode().split("\r\n")
|
||||
connect_line = lines[0]
|
||||
target = connect_line.split()[1]
|
||||
if ':' in target:
|
||||
target_host, port_str = target.split(':')
|
||||
target_port = int(port_str)
|
||||
else:
|
||||
target_host = target
|
||||
target_port = 443
|
||||
except Exception as e:
|
||||
print("[Error] Failed to parse CONNECT request:", e)
|
||||
sock.send(b"HTTP/1.1 400 Bad Request\r\n\r\n")
|
||||
continue
|
||||
else:
|
||||
try:
|
||||
for line in request.split(b"\r\n"):
|
||||
if line.lower().startswith(b"host:"):
|
||||
host_part = line[5:].strip()
|
||||
if b':' in host_part:
|
||||
target_host, port_str = host_part.split(b':')
|
||||
target_port = int(port_str)
|
||||
else:
|
||||
target_host = host_part.decode()
|
||||
break
|
||||
except Exception as e:
|
||||
print("[Error] Failed to parse HTTP request:", e)
|
||||
sock.send(b"HTTP/1.1 400 Bad Request\r\n\r\n")
|
||||
continue
|
||||
|
||||
if not target_host:
|
||||
sock.send(b"HTTP/1.1 400 Bad Request\r\n\r\n")
|
||||
continue
|
||||
|
||||
print("[Target] Host:", target_host, "Port:", target_port)
|
||||
|
||||
target_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
target_sock.settimeout(30)
|
||||
|
||||
try:
|
||||
target_sock.connect((target_host, target_port))
|
||||
print("[Target] Connected successfully")
|
||||
|
||||
if request.startswith(b"CONNECT"):
|
||||
sock.send(b"HTTP/1.1 200 Connection Established\r\n\r\n")
|
||||
else:
|
||||
target_sock.sendall(request)
|
||||
|
||||
t1 = threading.Thread(target=tunnel_thread, args=(sock, target_sock))
|
||||
t2 = threading.Thread(target=tunnel_thread, args=(target_sock, sock))
|
||||
t1.start()
|
||||
t2.start()
|
||||
t1.join()
|
||||
t2.join()
|
||||
|
||||
except ConnectionRefusedError:
|
||||
print("[Error] Cannot connect to target:", target_host, target_port)
|
||||
error_msg = "HTTP/1.1 503 Service Unavailable\r\nContent-Type: text/plain\r\n\r\nCannot connect to " + target_host + ":" + str(target_port)
|
||||
sock.send(error_msg.encode())
|
||||
except socket.timeout:
|
||||
print("[Error] Connection timeout")
|
||||
sock.send(b"HTTP/1.1 504 Gateway Timeout\r\n\r\n")
|
||||
except Exception as e:
|
||||
print("[Error] Failed to connect to target:", e)
|
||||
error_msg = "HTTP/1.1 500 Internal Server Error\r\nContent-Type: text/plain\r\n\r\n" + str(e)
|
||||
sock.send(error_msg.encode())
|
||||
finally:
|
||||
target_sock.close()
|
||||
|
||||
except ConnectionResetError:
|
||||
print("[Device B] Connection forcibly closed")
|
||||
except socket.timeout:
|
||||
log_info("System", "Connection idle timeout")
|
||||
except Exception as e:
|
||||
print("[Error] Communication error with device B:", e)
|
||||
finally:
|
||||
sock.close()
|
||||
log_error("System", "Connection error: %s" % str(e))
|
||||
|
||||
def connect_to_b_device(b_host, b_port):
|
||||
"""Connect to device B"""
|
||||
"""Connect to device B and handle requests"""
|
||||
while True:
|
||||
try:
|
||||
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
sock.settimeout(10)
|
||||
sock.connect((b_host, b_port))
|
||||
sock.settimeout(None)
|
||||
|
||||
print("\n[System] Successfully connected to device B:", b_host, ":", b_port)
|
||||
# Wait for connection confirmation from server
|
||||
sock.settimeout(5)
|
||||
response = sock.recv(1024)
|
||||
if not response or b"PROXY_CONNECTED" not in response:
|
||||
log_error("System", "Failed to receive connection confirmation")
|
||||
sock.close()
|
||||
time.sleep(3)
|
||||
continue
|
||||
|
||||
handle_b_device(sock)
|
||||
log_info("System", "Successfully connected to device B: %s:%d" % (b_host, b_port))
|
||||
|
||||
listen_for_requests(sock)
|
||||
|
||||
except ConnectionRefusedError:
|
||||
print("[System] Device B not ready (", b_host, ":", b_port, "), retrying in 3 seconds...")
|
||||
import time
|
||||
log_info("System", "Device B not ready (%s:%d), retrying in 3 seconds..." % (b_host, b_port))
|
||||
time.sleep(3)
|
||||
except socket.timeout:
|
||||
print("[System] Connection timeout, retrying in 3 seconds...")
|
||||
import time
|
||||
log_info("System", "Connection timeout, retrying in 3 seconds...")
|
||||
time.sleep(3)
|
||||
except Exception as e:
|
||||
print("[Error] Connection error:", e)
|
||||
import time
|
||||
log_error("System", "Connection error: %s" % str(e))
|
||||
time.sleep(3)
|
||||
|
||||
def main():
|
||||
@@ -143,6 +149,7 @@ def main():
|
||||
print("=" * 60)
|
||||
print(" Proxy Client (Run on Computer A)")
|
||||
print("=" * 60)
|
||||
print("Target website:", TARGET_HOST, ":", TARGET_PORT)
|
||||
print("Usage: python proxy_client.py <Device B IP> <Device B Port>")
|
||||
print("Example: python proxy_client.py 192.168.1.100 8080")
|
||||
print("=" * 60)
|
||||
@@ -154,6 +161,7 @@ def main():
|
||||
print("=" * 60)
|
||||
print(" Proxy Client (Run on Computer A)")
|
||||
print("=" * 60)
|
||||
print("Target website:", TARGET_HOST, ":", TARGET_PORT)
|
||||
print("Device B address:", b_host, ":", b_port)
|
||||
print("=" * 60)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user