修改 proxy 方案
This commit is contained in:
cheney
+105
-67
@@ -1,86 +1,124 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import socket
|
||||
import threading
|
||||
import select
|
||||
import sys
|
||||
|
||||
TARGET_HOST = 'superstar.geelytravel.com'
|
||||
TARGET_PORT = 443
|
||||
LISTEN_PORT = 8080
|
||||
PORT_FOR_A = 8080 # Port for device A to connect
|
||||
PORT_FOR_BROWSER = 8081 # Port for browser to connect
|
||||
|
||||
b_device_socket = None
|
||||
client_socket = None
|
||||
a_device_socket = None
|
||||
a_device_lock = threading.Lock()
|
||||
|
||||
def handle_b_device(conn):
|
||||
"""处理B设备的连接"""
|
||||
global b_device_socket
|
||||
b_device_socket = conn
|
||||
print(f"B设备已连接: {conn.getpeername()}")
|
||||
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
|
||||
finally:
|
||||
try:
|
||||
src.close()
|
||||
except:
|
||||
pass
|
||||
try:
|
||||
dst.close()
|
||||
except:
|
||||
pass
|
||||
|
||||
def handle_browser(conn, addr):
|
||||
"""Handle browser connection, establish tunnel with device A"""
|
||||
global a_device_socket
|
||||
|
||||
print("\n[Browser] Connected from:", addr)
|
||||
|
||||
with a_device_lock:
|
||||
if not a_device_socket:
|
||||
conn.send(b"HTTP/1.1 503 Service Unavailable\r\nContent-Type: text/plain\r\n\r\nDevice A not connected")
|
||||
conn.close()
|
||||
print("[Browser] Disconnected (Device A not connected):", addr)
|
||||
return
|
||||
|
||||
t1 = threading.Thread(target=tunnel_thread, args=(conn, a_device_socket))
|
||||
t2 = threading.Thread(target=tunnel_thread, args=(a_device_socket, conn))
|
||||
t1.start()
|
||||
t2.start()
|
||||
|
||||
print("[System] Tunnel established between browser and device A")
|
||||
|
||||
def handle_a_device(conn):
|
||||
"""Handle device A connection"""
|
||||
global a_device_socket
|
||||
|
||||
with a_device_lock:
|
||||
if a_device_socket:
|
||||
conn.send(b"ERROR: Another A device is already connected")
|
||||
conn.close()
|
||||
return
|
||||
a_device_socket = conn
|
||||
|
||||
print("\n[Device A] Connected:", conn.getpeername())
|
||||
|
||||
try:
|
||||
while True:
|
||||
data = conn.recv(4096)
|
||||
data = conn.recv(1)
|
||||
if not data:
|
||||
print("B设备断开连接")
|
||||
b_device_socket = None
|
||||
break
|
||||
|
||||
if client_socket:
|
||||
try:
|
||||
client_socket.send(data)
|
||||
except:
|
||||
pass
|
||||
except Exception as e:
|
||||
print(f"B设备连接出错: {e}")
|
||||
b_device_socket = None
|
||||
except:
|
||||
pass
|
||||
finally:
|
||||
with a_device_lock:
|
||||
a_device_socket = None
|
||||
conn.close()
|
||||
print("[Device A] Disconnected")
|
||||
|
||||
def handle_client(conn):
|
||||
"""处理用户请求"""
|
||||
global client_socket
|
||||
client_socket = conn
|
||||
print(f"用户连接: {conn.getpeername()}")
|
||||
|
||||
try:
|
||||
while True:
|
||||
data = conn.recv(4096)
|
||||
if not data:
|
||||
print("用户断开连接")
|
||||
client_socket = None
|
||||
break
|
||||
|
||||
if b_device_socket:
|
||||
try:
|
||||
b_device_socket.send(data)
|
||||
except:
|
||||
conn.send(b"HTTP/1.1 503 Service Unavailable\r\n\r\nDevice B not connected")
|
||||
break
|
||||
else:
|
||||
conn.send(b"HTTP/1.1 503 Service Unavailable\r\n\r\nDevice B not connected")
|
||||
break
|
||||
except Exception as e:
|
||||
print(f"用户连接出错: {e}")
|
||||
client_socket = None
|
||||
|
||||
def run_server():
|
||||
def run_browser_server():
|
||||
"""Start browser listening port"""
|
||||
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||||
server.bind(('0.0.0.0', LISTEN_PORT))
|
||||
server.listen(5)
|
||||
server.bind(('0.0.0.0', PORT_FOR_BROWSER))
|
||||
server.listen(10)
|
||||
|
||||
print(f"A设备代理服务器已启动，监听端口 {LISTEN_PORT}")
|
||||
print(f"等待B设备连接和用户请求...")
|
||||
print("[System] Browser proxy port started, listening on 0.0.0.0:", PORT_FOR_BROWSER)
|
||||
|
||||
while True:
|
||||
conn, addr = server.accept()
|
||||
print(f"新连接: {addr}")
|
||||
|
||||
# 判断是B设备还是用户（简单判断：先连接的是B设备）
|
||||
if b_device_socket is None:
|
||||
# 第一个连接作为B设备
|
||||
t = threading.Thread(target=handle_b_device, args=(conn,))
|
||||
t.start()
|
||||
else:
|
||||
# 后续连接作为用户
|
||||
t = threading.Thread(target=handle_client, args=(conn,))
|
||||
t.start()
|
||||
t = threading.Thread(target=handle_browser, args=(conn, addr))
|
||||
t.start()
|
||||
|
||||
def run_a_device_server():
|
||||
"""Start device A connection port"""
|
||||
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||||
server.bind(('0.0.0.0', PORT_FOR_A))
|
||||
server.listen(1)
|
||||
|
||||
print("[System] Device A port started, listening on 0.0.0.0:", PORT_FOR_A)
|
||||
|
||||
while True:
|
||||
conn, addr = server.accept()
|
||||
t = threading.Thread(target=handle_a_device, args=(conn,))
|
||||
t.start()
|
||||
|
||||
def main():
|
||||
print("=" * 60)
|
||||
print(" Proxy Server (Run on Computer B)")
|
||||
print("=" * 60)
|
||||
print("Device A connection port:", PORT_FOR_A)
|
||||
print("Browser access port:", PORT_FOR_BROWSER)
|
||||
print("=" * 60)
|
||||
print("Waiting for connections...\n")
|
||||
|
||||
t1 = threading.Thread(target=run_a_device_server)
|
||||
t2 = threading.Thread(target=run_browser_server)
|
||||
t1.start()
|
||||
t2.start()
|
||||
|
||||
t1.join()
|
||||
t2.join()
|
||||
|
||||
if __name__ == '__main__':
|
||||
run_server()
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user