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()