使用套接字进行Python端口转发

import socket

def port_forwarding(local_port, remote_host, remote_port):
    local_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    local_socket.bind(('localhost', local_port))
    local_socket.listen(1)

    while True:
        client_socket, client_address = local_socket.accept()
        remote_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        remote_socket.connect((remote_host, remote_port))

        # 启动两个线程，分别进行数据的转发
        threading.Thread(target=forward_data, args=(client_socket, remote_socket)).start()
        threading.Thread(target=forward_data, args=(remote_socket, client_socket)).start()

def forward_data(src_socket, dest_socket):
    while True:
        data = src_socket.recv(1024)
        if not data:
            break
        dest_socket.sendall(data)

    src_socket.close()
    dest_socket.close()

# 示例用法
port_forwarding(8080, 'example.com', 80)