133 lines
4.0 KiB
Python
133 lines
4.0 KiB
Python
import socket
|
||
import threading
|
||
import sys
|
||
import configparser
|
||
import os
|
||
|
||
|
||
def load_config(config_path='config.ini'):
|
||
"""加载配置文件"""
|
||
config = configparser.ConfigParser()
|
||
|
||
# 如果配置文件不存在，创建默认配置
|
||
if not os.path.exists(config_path):
|
||
print(f"配置文件 {config_path} 不存在，使用默认配置")
|
||
return {
|
||
'protocol': 'https',
|
||
'target_host': 'zentao.sunyard.com.cn',
|
||
'target_port': 9788
|
||
}
|
||
|
||
config.read(config_path, encoding='utf-8')
|
||
|
||
if 'proxy' not in config:
|
||
print(f"配置文件格式错误，使用默认配置")
|
||
return {
|
||
'protocol': 'https',
|
||
'target_host': 'zentao.sunyard.com.cn',
|
||
'target_port': 9788
|
||
}
|
||
|
||
protocol = config['proxy'].get('protocol', 'https').lower()
|
||
target_host = config['proxy'].get('target_host', 'zentao.sunyard.com.cn')
|
||
target_port = config['proxy'].getint('target_port', 9788)
|
||
|
||
# 如果没有设置端口，根据协议设置默认端口
|
||
if target_port == 0:
|
||
target_port = 443 if protocol == 'https' else 80
|
||
|
||
return {
|
||
'protocol': protocol,
|
||
'target_host': target_host,
|
||
'target_port': target_port
|
||
}
|
||
|
||
def handle_request(sock, target_host, target_port):
|
||
"""处理从A设备收到的请求，转发到目标网站"""
|
||
|
||
try:
|
||
# 接收来自A设备的请求
|
||
request = b""
|
||
while True:
|
||
data = sock.recv(4096)
|
||
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()
|
||
except Exception as e:
|
||
print(f"处理请求出错: {e}")
|
||
try:
|
||
sock.send(b"HTTP/1.1 500 Internal Server Error\r\n\r\n")
|
||
except:
|
||
pass
|
||
|
||
def connect_to_a_device(a_host, a_port, target_host, target_port):
|
||
"""连接到A设备"""
|
||
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}")
|
||
|
||
handle_request(sock, target_host, target_port)
|
||
|
||
sock.close()
|
||
print("与A设备的连接已断开")
|
||
except ConnectionRefusedError:
|
||
print(f"A设备未就绪，重试中...")
|
||
import time
|
||
time.sleep(3)
|
||
except Exception as e:
|
||
print(f"连接出错: {e}")
|
||
import time
|
||
time.sleep(3)
|
||
|
||
def main():
|
||
|
||
# 加载配置
|
||
config = load_config()
|
||
|
||
|
||
# 解析命令行参数
|
||
if len(sys.argv) < 3:
|
||
print("用法: python proxy_client.py <A设备IP> <A设备端口>")
|
||
print(f"当前配置: {config['protocol']}://{config['target_host']}:{config['target_port']}")
|
||
print("示例: python proxy_client.py 192.168.1.50 8080")
|
||
sys.exit(1)
|
||
|
||
a_host = sys.argv[1]
|
||
a_port = int(sys.argv[2])
|
||
|
||
# 可选：通过命令行覆盖目标配置
|
||
if len(sys.argv) >= 4:
|
||
config['target_host'] = sys.argv[3]
|
||
if len(sys.argv) >= 5:
|
||
config['target_port'] = int(sys.argv[4])
|
||
|
||
print(f"使用配置: {config['protocol']}://{config['target_host']}:{config['target_port']}")
|
||
connect_to_a_device(a_host, a_port, config['target_host'], config['target_port'])
|
||
|
||
if __name__ == '__main__':
|
||
main()
|