import socket,threading

# 單向流數(shù)據(jù)傳遞
def tcp_mapping_worker(conn_receiver, conn_sender):
    while True:
        try:
            # 接收數(shù)據(jù)緩存大小
            data = conn_receiver.recv(2048)
        except Exception:
            print("[-] 關(guān)閉: 映射請(qǐng)求已關(guān)閉.")
            break
        if not data:
            break
        try:
            conn_sender.sendall(data)
        except Exception:
            print("[-] 錯(cuò)誤: 發(fā)送數(shù)據(jù)時(shí)出錯(cuò).")
            break
        print("[+] 映射請(qǐng)求: {} ---> 傳輸?shù)? {} ---> {} bytes"
        .format(conn_receiver.getpeername(), conn_sender.getpeername(), len(data)))
    conn_receiver.close()
    conn_sender.close()
    return

# 端口映射請(qǐng)求處理
def tcp_mapping_request(local_conn, remote_ip, remote_port):
    remote_conn = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    try:
        remote_conn.connect((remote_ip, remote_port))
    except Exception:
        local_conn.close()
        print("[x] 錯(cuò)誤: 無(wú)法連接到 {}:{} 遠(yuǎn)程服務(wù)器".format(remote_ip,remote_port))
        return
    threading.Thread(target=tcp_mapping_worker, args=(local_conn, remote_conn)).start()
    threading.Thread(target=tcp_mapping_worker, args=(remote_conn, local_conn)).start()
    return

if __name__ == "__main__":
    remote_ip = "8.141.58.64"     # 對(duì)端地址
    remote_port = 3389            # 對(duì)端端口
    local_ip = "0.0.0.0"          # 本機(jī)地址
    local_port = 10050            # 本機(jī)端口
    
    local_server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    local_server.bind((local_ip, local_port))
    local_server.listen(5)
    print("[*] 本地端口監(jiān)聽 {}:{}".format(local_ip,local_port))
    while True:
        try:
            (local_conn, local_addr) = local_server.accept()
        except Exception:
            local_server.close()
            break
        threading.Thread(target=tcp_mapping_request, args=(local_conn, remote_ip, remote_port)).start()