快捷導(dǎo)航

Python實(shí)現(xiàn)端口流量轉(zhuǎn)發(fā)的示例代碼

更新時(shí)間：2023年11月03日 10:19:49 作者：微軟技術(shù)分享

端口流量轉(zhuǎn)發(fā)（Port Forwarding）是一種網(wǎng)絡(luò)通信技術(shù),用于將特定的網(wǎng)絡(luò)流量從一個(gè)端口或網(wǎng)絡(luò)地址轉(zhuǎn)發(fā)到另一個(gè)端口或地址,它在網(wǎng)絡(luò)中扮演著一個(gè)非常重要的角色,在Python語言中實(shí)現(xiàn)端口轉(zhuǎn)發(fā)非常容易,文中有相關(guān)的代碼示例,需要的朋友可以參考下

如下這段代碼實(shí)現(xiàn)了一個(gè)基本的TCP端口映射，將本地指定端口的流量轉(zhuǎn)發(fā)到指定的遠(yuǎn)程IP和端口。實(shí)現(xiàn)方式是在接收到本地客戶端連接請(qǐng)求后，創(chuàng)建一個(gè)線程，將本地連接和遠(yuǎn)程連接之間的數(shù)據(jù)傳輸通過線程分別實(shí)現(xiàn)，這樣就實(shí)現(xiàn)了數(shù)據(jù)在本地和遠(yuǎn)程之間的單向流動(dòng)，從而實(shí)現(xiàn)了TCP端口映射。

具體實(shí)現(xiàn)方式如下：

1.定義tcp_mapping_worker()函數(shù)，接收本地客戶端連接和遠(yuǎn)程服務(wù)器連接作為參數(shù)，通過while循環(huán)監(jiān)聽本地連接，當(dāng)接收到本地連接的數(shù)據(jù)時(shí)，將數(shù)據(jù)發(fā)送到遠(yuǎn)程服務(wù)器連接，實(shí)現(xiàn)本地到遠(yuǎn)程的單向流動(dòng)。當(dāng)本地連接關(guān)閉時(shí)，結(jié)束該線程。
2.定義tcp_mapping_request()函數(shù)，接收本地客戶端連接、遠(yuǎn)程服務(wù)器IP和端口作為參數(shù)，創(chuàng)建遠(yuǎn)程連接，然后使用兩個(gè)線程分別將本地連接和遠(yuǎn)程連接之間的數(shù)據(jù)傳輸進(jìn)行處理，從而實(shí)現(xiàn)本地與遠(yuǎn)程之間的單向流動(dòng)。當(dāng)本地連接關(guān)閉時(shí)，結(jié)束該函數(shù)。
3.在主函數(shù)中，首先指定遠(yuǎn)程服務(wù)器的IP和端口，然后使用socket創(chuàng)建一個(gè)本地服務(wù)器監(jiān)聽指定端口，當(dāng)接收到本地客戶端的連接請(qǐng)求時(shí)，將其傳遞給tcp_mapping_request()函數(shù)，創(chuàng)建線程進(jìn)行數(shù)據(jù)傳輸。

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ò)誤: 無法連接到 {}:{} 遠(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()

讀者可自行運(yùn)行上述代碼片段，則當(dāng)用戶訪問本機(jī)地址127.0.0.1:10050則會(huì)將數(shù)據(jù)包自動(dòng)轉(zhuǎn)發(fā)到8.141.58.64:3389端口上，此時(shí)如果用戶使用遠(yuǎn)程鏈接等工具，則可以實(shí)現(xiàn)遠(yuǎn)程桌面鏈接功能，如下圖所示；