修改 proxy 方案
This commit is contained in:
cheney
parent
ebcec1c463
commit
cfc009f60d
@ -1,80 +1,163 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import socket
|
||||
import threading
|
||||
import sys
|
||||
|
||||
TARGET_HOST = 'superstar.geelytravel.com'
|
||||
TARGET_PORT = 443
|
||||
|
||||
def handle_request(sock):
|
||||
"""处理从A设备收到的请求，转发到目标网站"""
|
||||
def tunnel_thread(src, dst):
|
||||
"""Tunnel thread: continuously forward data"""
|
||||
try:
|
||||
# 接收来自A设备的请求
|
||||
request = b""
|
||||
while True:
|
||||
data = sock.recv(4096)
|
||||
data = src.recv(8192)
|
||||
if not data:
|
||||
break
|
||||
request += data
|
||||
if b"\r\n\r\n" in request:
|
||||
break
|
||||
|
||||
if not request:
|
||||
return
|
||||
|
||||
# 连接到目标网站
|
||||
target_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
target_sock.connect((TARGET_HOST, TARGET_PORT))
|
||||
|
||||
# 发送请求
|
||||
target_sock.send(request)
|
||||
|
||||
# 接收响应并转发回A设备
|
||||
while True:
|
||||
response = target_sock.recv(4096)
|
||||
if not response:
|
||||
break
|
||||
sock.send(response)
|
||||
|
||||
target_sock.close()
|
||||
dst.sendall(data)
|
||||
except Exception as e:
|
||||
print(f"处理请求出错: {e}")
|
||||
try:
|
||||
sock.send(b"HTTP/1.1 500 Internal Server Error\r\n\r\n")
|
||||
except:
|
||||
pass
|
||||
pass
|
||||
|
||||
def connect_to_a_device(a_host, a_port):
|
||||
"""连接到A设备"""
|
||||
def handle_b_device(sock):
|
||||
"""Handle communication with device B, establish tunnel to target website"""
|
||||
print("\n[System] Connected to device B")
|
||||
|
||||
try:
|
||||
while True:
|
||||
request = b""
|
||||
while True:
|
||||
chunk = sock.recv(8192)
|
||||
if not chunk:
|
||||
print("[Device B] Disconnected")
|
||||
return
|
||||
request += chunk
|
||||
if b"\r\n\r\n" in request:
|
||||
break
|
||||
|
||||
print("[System] Received request:", len(request), "bytes")
|
||||
|
||||
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 Exception as e:
|
||||
print("[Error] Communication error with device B:", e)
|
||||
finally:
|
||||
sock.close()
|
||||
|
||||
def connect_to_b_device(b_host, b_port):
|
||||
"""Connect to device B"""
|
||||
while True:
|
||||
try:
|
||||
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
sock.connect((a_host, a_port))
|
||||
print(f"成功连接到A设备: {a_host}:{a_port}")
|
||||
print(f"准备转发请求到: {TARGET_HOST}:{TARGET_PORT}")
|
||||
sock.settimeout(10)
|
||||
sock.connect((b_host, b_port))
|
||||
sock.settimeout(None)
|
||||
|
||||
handle_request(sock)
|
||||
print("\n[System] Successfully connected to device B:", b_host, ":", b_port)
|
||||
|
||||
handle_b_device(sock)
|
||||
|
||||
sock.close()
|
||||
print("与A设备的连接已断开")
|
||||
except ConnectionRefusedError:
|
||||
print(f"A设备未就绪，重试中...")
|
||||
print("[System] Device B not ready (", b_host, ":", b_port, "), retrying in 3 seconds...")
|
||||
import time
|
||||
time.sleep(3)
|
||||
except socket.timeout:
|
||||
print("[System] Connection timeout, retrying in 3 seconds...")
|
||||
import time
|
||||
time.sleep(3)
|
||||
except Exception as e:
|
||||
print(f"连接出错: {e}")
|
||||
print("[Error] Connection error:", e)
|
||||
import time
|
||||
time.sleep(3)
|
||||
|
||||
def main():
|
||||
if len(sys.argv) < 3:
|
||||
print("用法: python proxy_client.py <A设备IP> <A设备端口>")
|
||||
print("示例: python proxy_client.py 192.168.1.50 8080")
|
||||
print("=" * 60)
|
||||
print(" Proxy Client (Run on Computer A)")
|
||||
print("=" * 60)
|
||||
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)
|
||||
sys.exit(1)
|
||||
|
||||
a_host = sys.argv[1]
|
||||
a_port = int(sys.argv[2])
|
||||
b_host = sys.argv[1]
|
||||
b_port = int(sys.argv[2])
|
||||
|
||||
connect_to_a_device(a_host, a_port)
|
||||
print("=" * 60)
|
||||
print(" Proxy Client (Run on Computer A)")
|
||||
print("=" * 60)
|
||||
print("Device B address:", b_host, ":", b_port)
|
||||
print("=" * 60)
|
||||
|
||||
connect_to_b_device(b_host, b_port)
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
||||
@ -1,112 +0,0 @@
|
||||
import sys
|
||||
import winreg
|
||||
|
||||
def get_current_proxy():
|
||||
"""获取当前系统代理设置"""
|
||||
try:
|
||||
internet_settings = winreg.OpenKey(
|
||||
winreg.HKEY_CURRENT_USER,
|
||||
r'Software\Microsoft\Windows\CurrentVersion\Internet Settings',
|
||||
0,
|
||||
winreg.KEY_READ
|
||||
)
|
||||
proxy_server, _ = winreg.QueryValueEx(internet_settings, 'ProxyServer')
|
||||
proxy_enable, _ = winreg.QueryValueEx(internet_settings, 'ProxyEnable')
|
||||
winreg.CloseKey(internet_settings)
|
||||
return {
|
||||
'enabled': bool(proxy_enable),
|
||||
'server': proxy_server
|
||||
}
|
||||
except FileNotFoundError:
|
||||
return {'enabled': False, 'server': ''}
|
||||
except Exception as e:
|
||||
print(f"获取代理设置失败: {e}")
|
||||
return {'enabled': False, 'server': ''}
|
||||
|
||||
def set_proxy(proxy_host, proxy_port):
|
||||
"""设置系统代理"""
|
||||
try:
|
||||
internet_settings = winreg.OpenKey(
|
||||
winreg.HKEY_CURRENT_USER,
|
||||
r'Software\Microsoft\Windows\CurrentVersion\Internet Settings',
|
||||
0,
|
||||
winreg.KEY_ALL_ACCESS
|
||||
)
|
||||
|
||||
proxy_server = f'{proxy_host}:{proxy_port}'
|
||||
winreg.SetValueEx(internet_settings, 'ProxyServer', 0, winreg.REG_SZ, proxy_server)
|
||||
winreg.SetValueEx(internet_settings, 'ProxyEnable', 0, winreg.REG_DWORD, 1)
|
||||
winreg.CloseKey(internet_settings)
|
||||
|
||||
print(f"✅ 系统代理已设置为: {proxy_server}")
|
||||
return True
|
||||
except Exception as e:
|
||||
print(f"❌ 设置代理失败: {e}")
|
||||
return False
|
||||
|
||||
def unset_proxy():
|
||||
"""取消系统代理"""
|
||||
try:
|
||||
internet_settings = winreg.OpenKey(
|
||||
winreg.HKEY_CURRENT_USER,
|
||||
r'Software\Microsoft\Windows\CurrentVersion\Internet Settings',
|
||||
0,
|
||||
winreg.KEY_ALL_ACCESS
|
||||
)
|
||||
|
||||
winreg.SetValueEx(internet_settings, 'ProxyEnable', 0, winreg.REG_DWORD, 0)
|
||||
winreg.CloseKey(internet_settings)
|
||||
|
||||
print("✅ 系统代理已取消")
|
||||
return True
|
||||
except Exception as e:
|
||||
print(f"❌ 取消代理失败: {e}")
|
||||
return False
|
||||
|
||||
def show_status():
|
||||
"""显示当前代理状态"""
|
||||
proxy = get_current_proxy()
|
||||
print("\n当前代理状态:")
|
||||
print(f" 代理已启用: {'是' if proxy['enabled'] else '否'}")
|
||||
if proxy['enabled']:
|
||||
print(f" 代理服务器: {proxy['server']}")
|
||||
print()
|
||||
|
||||
def main():
|
||||
if len(sys.argv) < 2:
|
||||
print("=" * 50)
|
||||
print(" Windows系统代理管理脚本")
|
||||
print("=" * 50)
|
||||
print("用法:")
|
||||
print(f" {sys.argv[0]} status - 查看当前代理状态")
|
||||
print(f" {sys.argv[0]} set <host> <port> - 设置代理服务器")
|
||||
print(f" {sys.argv[0]} unset - 取消代理设置")
|
||||
print(f" {sys.argv[0]} localhost - 快速设置为 localhost:8080")
|
||||
print("\n示例:")
|
||||
print(f" {sys.argv[0]} set 192.168.1.50 8080")
|
||||
print(f" {sys.argv[0]} localhost")
|
||||
print(f" {sys.argv[0]} unset")
|
||||
print("=" * 50)
|
||||
return
|
||||
|
||||
command = sys.argv[1].lower()
|
||||
|
||||
if command == 'status':
|
||||
show_status()
|
||||
elif command == 'set':
|
||||
if len(sys.argv) < 4:
|
||||
print("❌ 参数不足，请提供代理服务器地址和端口")
|
||||
print(f" 示例: {sys.argv[0]} set 192.168.1.50 8080")
|
||||
return
|
||||
proxy_host = sys.argv[2]
|
||||
proxy_port = sys.argv[3]
|
||||
set_proxy(proxy_host, proxy_port)
|
||||
elif command == 'unset':
|
||||
unset_proxy()
|
||||
elif command == 'localhost':
|
||||
set_proxy('localhost', '8080')
|
||||
else:
|
||||
print(f"❌ 未知命令: {command}")
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
@ -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()
|
||||
|
||||
Loading…
Reference in New Issue
Block a user