戰(zhàn)：從Socket原理到可靠傳輸實(shí)現(xiàn))
之前在做一些局域網(wǎng)內(nèi)的實(shí)時(shí)工具時(shí)想用最輕量的方式實(shí)現(xiàn)兩個(gè)節(jié)點(diǎn)互相發(fā)消息。第一反應(yīng)是 TCP后來發(fā)現(xiàn)很多場(chǎng)景其實(shí)用不到 TCP 的可靠流式傳輸反而 UDP 的“無連接 數(shù)據(jù)報(bào)”模型更簡(jiǎn)單直接。本文就來完整拆解如何用 UDP 實(shí)現(xiàn)一個(gè)可運(yùn)行的單聊程序覆蓋協(xié)議差異、Socket API、多線程收發(fā)、退出機(jī)制和常見坑點(diǎn)。本文適合學(xué)過 Python 基礎(chǔ)語法、想進(jìn)一步接觸網(wǎng)絡(luò)編程的讀者?？赐曛竽悴粌H能跑通一個(gè) UDP 單聊程序還能理解 UDP 與 TCP 在程序設(shè)計(jì)上的本質(zhì)差異知道為什么 UDP 丟包時(shí)需要自己做確認(rèn)和重傳以及在實(shí)際項(xiàng)目中如何給這種“裸 UDP 通信”加上可靠性和安全邊界。1. 背景與核心概念UDP 單聊到底在解決什么問題1.1 什么是 UDP 單聊單聊就是兩個(gè)終端之間的一對(duì)一消息通信。TCP 可以實(shí)現(xiàn)單聊UDP 同樣可以實(shí)現(xiàn)單聊。區(qū)別在于協(xié)議的傳輸模型TCP 是面向連接的字節(jié)流雙方必須先建立連接然后像讀寫文件一樣收發(fā)數(shù)據(jù)UDP 是面向無連接的數(shù)據(jù)報(bào)發(fā)送方只需要知道對(duì)方的 IP 和端口就可以把一段獨(dú)立的數(shù)據(jù)報(bào)發(fā)出去不需要提前“握手”。所以 UDP 單聊的本質(zhì)是兩個(gè) socket 端點(diǎn)之間互相發(fā)送數(shù)據(jù)報(bào)每個(gè)數(shù)據(jù)報(bào)是一個(gè)完整的消息。由于沒有連接狀態(tài)服務(wù)端和客戶端的身份不像 TCP 那樣固定兩個(gè)端點(diǎn)都可以主動(dòng)給對(duì)方發(fā)消息也可以隨時(shí)監(jiān)聽對(duì)方發(fā)來的消息。1.2 UDP 與 TCP 的核心差異要真正理解 UDP 單聊必須先分清 UDP 和 TCP 的幾個(gè)關(guān)鍵差異。對(duì)比維度TCPUDP連接狀態(tài)面向連接通信前必須三次握手無連接直接發(fā)送數(shù)據(jù)報(bào)傳輸單位字節(jié)流無消息邊界數(shù)據(jù)報(bào)每個(gè)報(bào)文獨(dú)立可靠性可靠傳輸支持確認(rèn)、重傳、排序不可靠不保證送達(dá)、不保證順序傳輸效率有握手和確認(rèn)開銷相對(duì)慢無握手和確認(rèn)開銷相對(duì)快應(yīng)用場(chǎng)景文件傳輸、網(wǎng)頁訪問、遠(yuǎn)程登錄實(shí)時(shí)音視頻、DNS 查詢、局域網(wǎng)發(fā)現(xiàn)其中最關(guān)鍵的一點(diǎn)是“消息邊界”。TCP 是流式協(xié)議發(fā)送方調(diào)用一次 send 發(fā)送的內(nèi)容可能在接收方多次 recv 才能讀完整也可能多次 send 的內(nèi)容被一次 recv 讀出來所以業(yè)務(wù)層必須自己定義消息邊界。UDP 則不同每次 sendto 發(fā)送一個(gè)數(shù)據(jù)報(bào)接收方每次 recvfrom 讀取一個(gè)完整的數(shù)據(jù)報(bào)消息邊界由操作系統(tǒng)維護(hù)。對(duì)于聊天這種天然以“一條消息”為單位的場(chǎng)景UDP 的數(shù)據(jù)報(bào)模型反而更直觀。另外UDP 的不可靠性也需要重點(diǎn)理解。UDP 不保證數(shù)據(jù)報(bào)一定能到達(dá)對(duì)端也不保證多個(gè)數(shù)據(jù)報(bào)的到達(dá)順序和發(fā)送順序一致。在局域網(wǎng)內(nèi)丟包率往往很低所以很多人第一次跑 UDP 通信會(huì)覺得“很可靠”但這是一種錯(cuò)覺。在跨公網(wǎng)、跨 Wi-Fi、網(wǎng)絡(luò)擁塞的場(chǎng)景下UDP 丟包和亂序問題會(huì)明顯暴露出來。本文先實(shí)現(xiàn)一個(gè)基礎(chǔ)可用的單聊程序第 6 章再討論如何在應(yīng)用層解決可靠性問題。1.3 UDP 單聊的典型應(yīng)用場(chǎng)景UDP 單聊看起來不如 TCP 聊天那么“標(biāo)準(zhǔn)”但它在很多場(chǎng)景下非常實(shí)用局域網(wǎng)工具兩個(gè)內(nèi)網(wǎng)節(jié)點(diǎn)之間快速互發(fā)信令不需要建連不需要維護(hù)連接狀態(tài)。實(shí)時(shí)性優(yōu)先的通信比如游戲內(nèi)的語音指令、視頻通話的信令通道實(shí)時(shí)性要求高于可靠性要求。嵌入式與 IoT 設(shè)備很多物聯(lián)網(wǎng)設(shè)備資源有限UDP 實(shí)現(xiàn)簡(jiǎn)單、占用資源少。自定義協(xié)議的原型驗(yàn)證先快速驗(yàn)證消息收發(fā)邏輯再?zèng)Q定是否要增加可靠層。如果你的場(chǎng)景需要嚴(yán)格的可靠消息、大文件傳輸、消息順序保證那么請(qǐng)選擇 TCP 或基于 TCP 的協(xié)議如 WebSocket、MQTT。如果只是輕量級(jí)消息互發(fā)且能接受消息偶爾丟失UDP 會(huì)是一個(gè)更簡(jiǎn)單的選擇。2. 環(huán)境準(zhǔn)備與前置知識(shí)2.1 運(yùn)行環(huán)境本文示例使用 Python 實(shí)現(xiàn)因?yàn)樗Z法簡(jiǎn)潔、標(biāo)準(zhǔn)庫直接支持 Socket 編程適合快速理解和驗(yàn)證 UDP 通信模型。項(xiàng)目說明操作系統(tǒng)Windows / Linux / macOS 均可Python 版本3.6 及以上依賴庫僅使用 Python 標(biāo)準(zhǔn)庫 socket、threading、sys環(huán)境要求本機(jī)或局域網(wǎng)內(nèi)兩臺(tái)機(jī)器本文不再贅述 Python 安裝步驟如果還沒有 Python 環(huán)境可以從 Python 官網(wǎng)下載并安裝安裝時(shí)記得勾選“Add Python to PATH”。2.2 需要提前掌握的 Socket 概念在寫代碼之前先梳理幾個(gè)必須掌握的 Socket 概念。IP 地址標(biāo)識(shí)網(wǎng)絡(luò)中的一臺(tái)主機(jī)。本機(jī)回環(huán)地址是 127.0.0.1局域網(wǎng)地址通常是 192.168.x.x。端口標(biāo)識(shí)主機(jī)上的一個(gè)網(wǎng)絡(luò)進(jìn)程。兩個(gè)程序不能同時(shí)綁定同一個(gè)端口。Socket操作系統(tǒng)提供的一個(gè)網(wǎng)絡(luò)通信句柄應(yīng)用層通過它讀寫網(wǎng)絡(luò)數(shù)據(jù)。AF_INET地址族表示使用 IPv4 地址。SOCK_DGRAM套接字類型表示使用數(shù)據(jù)報(bào)方式也就是 UDP。這些概念不需要背但要在看到代碼時(shí)能對(duì)上號(hào)。接下來我會(huì)逐個(gè)拆解 UDP 編程中會(huì)用到的核心 API。3. UDP Socket 核心 API 拆解3.1 創(chuàng)建套接字socket.socket(AF_INET, SOCK_DGRAM)UDP 套接字的創(chuàng)建方式和 TCP 一樣都是調(diào)用socket.socket()但類型參數(shù)不同。import socket udp_socket socket.socket(socket.AF_INET, socket.SOCK_DGRAM)這里的AF_INET表示 IPv4 地址族SOCK_DGRAM表示數(shù)據(jù)報(bào)套接字。創(chuàng)建成功后udp_socket就是一個(gè)可以用于 UDP 收發(fā)數(shù)據(jù)的對(duì)象。3.2 綁定地址bind()bind()的作用是把套接字綁定到一個(gè)本地地址和端口上。只有綁定后操作系統(tǒng)才知道把收到的 UDP 數(shù)據(jù)報(bào)交給哪個(gè)進(jìn)程。udp_socket.bind((0.0.0.0, 8001))第一個(gè)參數(shù)是本地 IP 地址0.0.0.0表示監(jiān)聽本機(jī)所有網(wǎng)卡這樣無論是從 127.0.0.1 還是局域網(wǎng) IP 發(fā)來的數(shù)據(jù)報(bào)都能收到。如果只想監(jiān)聽回環(huán)地址可以改成127.0.0.1。第二個(gè)參數(shù)是本地端口號(hào)。注意如果端口已經(jīng)被其他程序占用bind()會(huì)拋出OSError。另外UDP 不需要像 TCP 一樣調(diào)用listen()因?yàn)?UDP 沒有連接隊(duì)列的概念。3.3 發(fā)送數(shù)據(jù)sendto()UDP 發(fā)送數(shù)據(jù)用的是sendto()它需要指定目標(biāo)和端口。message 你好UDP.encode(utf-8) udp_socket.sendto(message, (127.0.0.1, 8002))sendto()的第一個(gè)參數(shù)是字節(jié)串所以字符串要用encode(utf-8)轉(zhuǎn)成字節(jié)。第二個(gè)參數(shù)是一個(gè)(ip, port)元組表示對(duì)端的地址和端口。這里有一個(gè)重要細(xì)節(jié)UDP 的sendto()即使對(duì)端不存在本地也不會(huì)立刻報(bào)錯(cuò)。因?yàn)?UDP 是無連接協(xié)議發(fā)送方只負(fù)責(zé)把數(shù)據(jù)報(bào)交給操作系統(tǒng)至于數(shù)據(jù)報(bào)能不能到達(dá)對(duì)端發(fā)送方無法馬上知道。這在調(diào)試時(shí)很容易造成“消息發(fā)出去了但對(duì)方?jīng)]收到”的假象。3.4 接收數(shù)據(jù)recvfrom()UDP 接收數(shù)據(jù)用的是recvfrom()它會(huì)阻塞等待數(shù)據(jù)報(bào)到來并在收到數(shù)據(jù)后同時(shí)返回?cái)?shù)據(jù)和發(fā)送方地址。data, addr udp_socket.recvfrom(1024)1024表示緩沖區(qū)大小單位是字節(jié)。UDP 是數(shù)據(jù)報(bào)協(xié)議一次recvfrom()讀取一個(gè)完整的數(shù)據(jù)報(bào)。如果數(shù)據(jù)報(bào)的大小超過緩沖區(qū)大小超出的部分會(huì)被截?cái)唷Ｋ栽谠O(shè)計(jì)協(xié)議時(shí)要控制單條消息大小盡量小于緩沖區(qū)值。addr是發(fā)送方的(ip, port)元組。在單聊場(chǎng)景中我們可以通過addr判斷消息是誰發(fā)來的方便后續(xù)做多端擴(kuò)展。3.5 close() 與資源釋放通信結(jié)束后需要調(diào)用close()關(guān)閉套接字釋放系統(tǒng)資源。udp_socket.close()關(guān)閉后套接字不能再用于收發(fā)數(shù)據(jù)。實(shí)際項(xiàng)目中建議使用try-finally或with語法確保資源釋放。3.6 可選 connectUDP 也有“連接”有些讀者可能聽說過 UDP 也可以調(diào)用connect()。這里特別說明一下UDP 的connect()并不是建立真正的連接它只是把對(duì)端地址保存到內(nèi)核中之后就可以用send()和recv()代替sendto()和recvfrom()寫法更簡(jiǎn)潔。udp_socket.connect((127.0.0.1, 8002)) udp_socket.send(message.encode(utf-8)) data udp_socket.recv(1024)調(diào)用connect()之后UDP 套接字只能和這個(gè)固定對(duì)端通信而且可以通過recv()收到對(duì)端返回的 ICMP 端口不可達(dá)錯(cuò)誤從而更快感知對(duì)端異常。但本文的基礎(chǔ)示例使用sendto()和recvfrom()因?yàn)檫@兩個(gè)方法更直觀地體現(xiàn)了 UDP 無連接、每次指定目標(biāo)地址的特點(diǎn)。4. 單聊程序的設(shè)計(jì)與實(shí)現(xiàn)4.1 程序職責(zé)劃分一個(gè)最簡(jiǎn)單的 UDP 單聊程序需要完成四件事創(chuàng)建 UDP 套接字并綁定本地端口。持續(xù)接收來自對(duì)端的消息并打印。持續(xù)讀取用戶輸入并發(fā)送給對(duì)端。支持退出機(jī)制關(guān)閉套接字。由于接收消息和發(fā)送消息是兩個(gè)相互獨(dú)立的動(dòng)作如果只有一個(gè)主循環(huán)就會(huì)出現(xiàn)“正在等待用戶輸入時(shí)無法接收消息”的問題。因此程序需要拆成兩個(gè)線程主線程負(fù)責(zé)讀取用戶輸入并發(fā)送消息。接收線程負(fù)責(zé)循環(huán)調(diào)用recvfrom()收到消息后打印。4.2 收發(fā)線程模型線程模型用文字描述是這樣程序啟動(dòng) | -- 創(chuàng)建 UDP Socket | -- bind 本地端口 | -- 啟動(dòng)接收線程循環(huán) recvfrom | -- 主循環(huán)循環(huán) input sendto | -- 用戶輸入 exit 或收到對(duì)方退出通知 | -- 退出程序接收線程設(shè)置為守護(hù)線程daemonTrue。守護(hù)線程的特點(diǎn)是當(dāng)主線程結(jié)束時(shí)守護(hù)線程會(huì)自動(dòng)終止。這樣即使接收線程還阻塞在recvfrom()主線程退出后程序也能正常結(jié)束。4.3 數(shù)據(jù)格式與編碼Socket 收發(fā)的是字節(jié)串而聊天內(nèi)容是字符串所以需要統(tǒng)一編碼。本文示例統(tǒng)一使用 UTF-8 編碼發(fā)送時(shí)message.encode(utf-8)接收時(shí)data.decode(utf-8)關(guān)于消息格式這里使用最簡(jiǎn)單的純文本消息。但在實(shí)際項(xiàng)目中建議設(shè)計(jì)為包含消息類型、消息序號(hào)、時(shí)間戳等字段的結(jié)構(gòu)化格式例如 JSON 字符串或自定義二進(jìn)制協(xié)議這樣后期增加 ACK、心跳、文件傳輸?shù)裙δ軙r(shí)更容易擴(kuò)展。4.4 退出機(jī)制退出機(jī)制是單聊程序最容易忽略的地方。基礎(chǔ)版本可以約定用戶輸入exit時(shí)退出程序同時(shí)向?qū)Χ税l(fā)送一條特殊消息__EXIT__通知對(duì)方自己已經(jīng)離開。由于 UDP 不可靠這條__EXIT__消息不能保證一定送達(dá)。所以更嚴(yán)謹(jǐn)?shù)脑O(shè)計(jì)是即使對(duì)方?jīng)]有收到退出通知發(fā)送方也可以直接退出接收方之后檢測(cè)到超時(shí)或心跳超時(shí)才判定對(duì)端離線。本文示例采用盡力通知的方式讓讀者理解“應(yīng)用層協(xié)議約定”的概念。5. 完整可運(yùn)行代碼5.1 項(xiàng)目結(jié)構(gòu)本文示例不需要復(fù)雜項(xiàng)目結(jié)構(gòu)一個(gè) Python 文件即可。udp_chat/ └── udp_chat.py在終端中兩個(gè)聊天方分別運(yùn)行這個(gè)腳本并傳入不同的本地端口和對(duì)方地址。5.2 完整代碼 udp_chat.py下面是完整代碼可以直接復(fù)制保存為udp_chat.py運(yùn)行。# 文件路徑udp_chat/udp_chat.py import socket import threading import sys # 全局退出事件任何一方退出時(shí)用于通知另一個(gè)線程停止 exit_event threading.Event() def receive_message(udp_socket: socket.socket): 接收線程循環(huán)接收 UDP 數(shù)據(jù)報(bào)并打印 while not exit_event.is_set(): try: data, addr udp_socket.recvfrom(1024) message data.decode(utf-8) if message __EXIT__: exit_event.set() print(\n[系統(tǒng)] 對(duì)方已退出聊天按回車可退出。) break print(f\n[對(duì)方] {message}) print( , end, flushTrue) except OSError: # 套接字被關(guān)閉時(shí)退出接收線程 break except KeyboardInterrupt: exit_event.set() break def start_chat(local_port: int, remote_ip: str, remote_port: int): 啟動(dòng) UDP 單聊主邏輯 # 1. 創(chuàng)建 UDP 套接字 udp_socket socket.socket(socket.AF_INET, socket.SOCK_DGRAM) # 2. 綁定本地端口0.0.0.0 表示監(jiān)聽本機(jī)所有網(wǎng)卡 udp_socket.bind((0.0.0.0, local_port)) # 3. 啟動(dòng)接收線程 receiver threading.Thread( targetreceive_message, args(udp_socket,), daemonTrue ) receiver.start() print(fUDP 單聊已啟動(dòng)本地端口 {local_port} - {remote_ip}:{remote_port}) print(輸入消息回車發(fā)送輸入 exit 退出。) try: # 4. 主線程循環(huán)讀取用戶輸入并發(fā)送 while not exit_event.is_set(): text input( ) if not text.strip(): continue if text.strip().lower() exit: # 盡力通知對(duì)方自己已退出 udp_socket.sendto(__EXIT__.encode(utf-8), (remote_ip, remote_port)) exit_event.set() break # 5. 發(fā)送數(shù)據(jù)報(bào) udp_socket.sendto(text.encode(utf-8), (remote_ip, remote_port)) except KeyboardInterrupt: print(\n[系統(tǒng)] 用戶主動(dòng)中斷正在退出。) exit_event.set() finally: # 6. 關(guān)閉套接字釋放系統(tǒng)資源 udp_socket.close() print([系統(tǒng)] 聊天已結(jié)束。) if __name__ __main__: if len(sys.argv) ! 4: print(用法: python udp_chat.py 本地端口 遠(yuǎn)端IP 遠(yuǎn)端端口) print(示例: ) print( A 端: python udp_chat.py 8001 127.0.0.1 8002) print( B 端: python udp_chat.py 8002 127.0.0.1 8001) sys.exit(1) start_chat( local_portint(sys.argv[1]), remote_ipsys.argv[2], remote_portint(sys.argv[3]), )5.3 運(yùn)行方式打開兩個(gè)終端窗口在第一個(gè)窗口運(yùn)行 A 端python udp_chat.py 8001 127.0.0.1 8002在第二個(gè)窗口運(yùn)行 B 端python udp_chat.py 8002 127.0.0.1 8001兩個(gè)終端都在本機(jī)所以 IP 都填寫回環(huán)地址127.0.0.1。A 端綁定本地端口8001發(fā)往127.0.0.1:8002B 端綁定本地端口8002發(fā)往127.0.0.1:8001。如果兩臺(tái)機(jī)器在同一個(gè)局域網(wǎng)內(nèi)則 IP 填寫對(duì)方機(jī)器的局域網(wǎng) IP例如python udp_chat.py 8001 192.168.1.100 8002注意局域網(wǎng)通信時(shí)需要保證兩臺(tái)機(jī)器的防火墻允許對(duì)應(yīng) UDP 端口的入站流量。5.4 預(yù)期運(yùn)行效果A 端窗口輸出UDP 單聊已啟動(dòng)本地端口 8001 - 127.0.0.1:8002 輸入消息回車發(fā)送輸入 exit 退出。 你好我是 A [對(duì)方] 你好我是 B 收到B 你好 [對(duì)方] 我們退出吧 exit [系統(tǒng)] 聊天已結(jié)束。B 端窗口輸出UDP 單聊已啟動(dòng)本地端口 8002 - 127.0.0.1:8001 輸入消息回車發(fā)送輸入 exit 退出。 [對(duì)方] 你好我是 A 你好我是 B [對(duì)方] 收到B 你好 我們退出吧 [系統(tǒng)] 對(duì)方已退出聊天按回車可退出。到這里一個(gè)基礎(chǔ)的 UDP 單聊程序就跑通了。它的核心邏輯只有五步創(chuàng)建套接字、綁定端口、接收線程、主線程輸入、關(guān)閉套接字。后面我們?cè)賮砜纯慈绾卧谶@個(gè)基礎(chǔ)上增加可靠性保障。6. 進(jìn)階為 UDP 單聊增加可靠性保障6.1 為什么需要應(yīng)用層確認(rèn)UDP 本身不保證消息可靠到達(dá)所以要想在 UDP 上實(shí)現(xiàn)更可靠的聊天必須在應(yīng)用層自己實(shí)現(xiàn)確認(rèn)和重傳機(jī)制。最簡(jiǎn)單的模型是發(fā)送方給每條消息編號(hào)。接收方收到消息后回復(fù)一條 ACK表示“某號(hào)消息已收到”。發(fā)送方如果一段時(shí)間內(nèi)沒有收到 ACK就重發(fā)該消息。這本質(zhì)上是在應(yīng)用層模仿 TCP 的確認(rèn)重傳機(jī)制但實(shí)現(xiàn)比 TCP 簡(jiǎn)單得多也足夠滿足很多輕量級(jí)通信需求。6.2 消息編號(hào)與 ACK 的簡(jiǎn)化設(shè)計(jì)這里給出一套簡(jiǎn)化的消息協(xié)議設(shè)計(jì)思路不展開完整代碼重點(diǎn)說明流程。發(fā)送方 接收方 |-------- msg:1:你好 -------- | |------- ack:1 -------------- | |-------- msg:2:在嗎 -------- | |------- ack:2 -------------- |消息格式可以采用類型:序號(hào):內(nèi)容的文本格式msg:1:你好ack:1發(fā)送方在發(fā)送消息時(shí)記錄消息序號(hào)并啟動(dòng)一個(gè)定時(shí)器。如果超時(shí)沒有收到對(duì)應(yīng)的 ACK則重發(fā)。接收方收到msg消息后立即回一條ack。6.3 核心代碼思路下面是消息格式的解析和構(gòu)造核心代碼可以作為擴(kuò)展參考。# 文件路徑udp_chat/ack_protocol.py def build_message(seq: int, content: str) - bytes: 構(gòu)造一條帶序號(hào)的聊天消息 return fmsg:{seq}:{content}.encode(utf-8) def build_ack(seq: int) - bytes: 構(gòu)造一條 ACK 確認(rèn)消息 return fack:{seq}.encode(utf-8) def parse_message(data: bytes): 解析收到的數(shù)據(jù)返回 (類型, 序號(hào), 內(nèi)容) text data.decode(utf-8) parts text.split(:, 2) if len(parts) 2: return None msg_type parts[0] seq int(parts[1]) content parts[2] if len(parts) 2 else return msg_type, seq, content在實(shí)際項(xiàng)目中還可以給消息增加時(shí)間戳、消息 ID、簽名等字段并引入滑動(dòng)窗口機(jī)制來提升吞吐量。對(duì)于單聊這種低頻場(chǎng)景每發(fā)一條消息就等待 ACK 的簡(jiǎn)單方式通常已經(jīng)夠用。7. 常見問題與排查思路7.1 高頻問題排查表問題現(xiàn)象常見原因解決思路兩端都運(yùn)行了但收不到消息地址或端口填錯(cuò)核對(duì)本地綁定端口和對(duì)端發(fā)送端口收不到消息且程序綁定報(bào)錯(cuò)端口被占用端口已被其他程序占用更換端口或用netstat查看端口占用中文亂碼編碼/解碼不一致統(tǒng)一使用 UTF-8 編碼消息被截?cái)嘞⒊^ recvfrom 緩沖區(qū)大小增大緩沖區(qū)或限制單條消息大小一方退出后另一方卡在輸入狀態(tài)輸入阻塞導(dǎo)致線程無法及時(shí)退出用退出事件 提示用戶按回車退出局域網(wǎng)內(nèi)收不到消息防火墻攔截 UDP 入站流量在防火墻中放行對(duì)應(yīng) UDP 端口7.2 幾個(gè)典型問題的詳細(xì)分析先說“收不到消息”。這是 UDP 調(diào)試中最高頻的問題。首先是檢查地址和端口A 發(fā)送給 B必須確保 A 填寫的目標(biāo)地址是 B 綁定的本地端口而不是 A 自己的端口。很多初學(xué)者會(huì)把local_port和remote_port搞混導(dǎo)致數(shù)據(jù)報(bào)發(fā)送給了自己。其次是防火墻問題。UDP 沒有連接狀態(tài)防火墻很難判斷一個(gè) UDP 數(shù)據(jù)報(bào)是否是“應(yīng)答包”所以很多系統(tǒng)的默認(rèn)策略會(huì)攔截來自外部的 UDP 入站請(qǐng)求。在局域網(wǎng)內(nèi)調(diào)試時(shí)可以先在防火墻中放行指定端口或者先用127.0.0.1回環(huán)地址測(cè)試排除網(wǎng)絡(luò)層因素。再說“消息截?cái)唷?。recvfrom(1024)表示最多讀取 1024 字節(jié)。如果發(fā)送的數(shù)據(jù)報(bào)超過 1024 字節(jié)超出的部分會(huì)被丟棄而且接收方不會(huì)收到任何“數(shù)據(jù)被截?cái)唷钡奶崾?。建議設(shè)計(jì)協(xié)議時(shí)把單條消息控制在 512 字節(jié)以內(nèi)或者在接收時(shí)使用更大的緩沖區(qū)比如recvfrom(65535)。最后是“程序無法退出”。由于input()是阻塞的當(dāng)接收線程收到對(duì)方退出通知并設(shè)置exit_event時(shí)主線程可能還卡在input()上。此時(shí)需要用戶按一次回車讓input()返回主線程才能檢查退出事件并結(jié)束。這是一個(gè)比較常見的交互問題本文代碼里已經(jīng)通過提示信息說明。8. 工程實(shí)踐建議8.1 消息大小與緩沖區(qū)UDP 數(shù)據(jù)報(bào)的長度受限于網(wǎng)絡(luò) MTU最大傳輸單元。在以太網(wǎng)環(huán)境中MTU 通常為 1500 字節(jié)扣除 IP 頭和 UDP 頭后建議單條 UDP 數(shù)據(jù)報(bào)不要超過 1472 字節(jié)。如果消息體積較大建議在應(yīng)用層拆分成多個(gè)數(shù)據(jù)報(bào)并給每個(gè)數(shù)據(jù)報(bào)編號(hào)接收方按編號(hào)重組。在實(shí)際代碼中recvfrom()的緩沖區(qū)大小可以設(shè)為 65535這是 UDP 數(shù)據(jù)報(bào)的最大長度能避免大部分截?cái)鄦栴}。8.2 安全與權(quán)限邊界UDP 無連接的特性也帶來安全風(fēng)險(xiǎn)任何人都可以向你的綁定端口發(fā)送數(shù)據(jù)報(bào)因此程序必須校驗(yàn)消息來源。如果服務(wù)暴露在公網(wǎng)容易被惡意掃描和 UDP Flood 攻擊。不要在公網(wǎng)環(huán)境中直接運(yùn)行裸 UDP 單聊程序建議增加鑒權(quán)、白名單、消息簽名等機(jī)制。本文代碼中接收線程打印了addr實(shí)際項(xiàng)目中應(yīng)加入來源 IP 白名單校驗(yàn)。需要特別說明的是本文所有示例僅用于合法授權(quán)的開發(fā)測(cè)試環(huán)境請(qǐng)不要對(duì)非授權(quán)目標(biāo)發(fā)送任何 UDP 流量。8.3 日志與可觀測(cè)性UDP 調(diào)試比 TCP 困難因?yàn)樗鼪]有連接狀態(tài)發(fā)生問題時(shí)不方便直接觀察“連接在哪一步斷了”。因此在工程實(shí)現(xiàn)中要增加日志收到消息時(shí)打印發(fā)送方地址、消息序號(hào)、消息長度。發(fā)送消息時(shí)打印目標(biāo)地址、消息序號(hào)。記錄丟包、重傳、異常等事件。有了日志就能快速定位是發(fā)送失敗、接收失敗還是網(wǎng)絡(luò)丟包。8.4 多環(huán)境驗(yàn)證建議按以下順序驗(yàn)證 UDP 單聊程序本機(jī)回環(huán)測(cè)試兩個(gè)終端都使用127.0.0.1排除網(wǎng)絡(luò)因素。局域網(wǎng)測(cè)試兩臺(tái)機(jī)器用局域網(wǎng) IP 通信驗(yàn)證防火墻和路由是否正常?？缇W(wǎng)段或 Wi-Fi 測(cè)試觀察弱網(wǎng)環(huán)境下的丟包和延遲表現(xiàn)評(píng)估是否需要增加應(yīng)用層可靠性機(jī)制。只有在多環(huán)境下驗(yàn)證過才能判斷當(dāng)前程序是否滿足需求。9. 總結(jié)與下一步學(xué)習(xí)方向本文圍繞 UDP 單聊程序從協(xié)議差異、Socket API、線程模型、完整代碼到可靠性擴(kuò)展做了一個(gè)相對(duì)完整的梳理。你現(xiàn)在可以動(dòng)手把第 5 章的代碼復(fù)制下來先在本機(jī)兩個(gè)終端中跑通回環(huán)通信再嘗試改成局域網(wǎng)內(nèi)的兩臺(tái)機(jī)器通信。跑通之后建議按以下方向繼續(xù)深入給程序加入 ACK 確認(rèn)和超時(shí)重傳理解應(yīng)用層可靠性的實(shí)現(xiàn)思路。將消息格式升級(jí)為 JSON 或自定義二進(jìn)制協(xié)議增加消息類型、時(shí)間戳、序列號(hào)等字段。把單聊擴(kuò)展為多人群聊需要在數(shù)據(jù)報(bào)中攜帶“房間號(hào)”或“目標(biāo)用戶 ID”等信息。學(xué)習(xí)如何用select、poll或異步 IO 代替多線程模型優(yōu)化高并發(fā)下的資源占用。UDP 最大的特點(diǎn)是無連接、效率高但不可靠。真正的難點(diǎn)往往不在“怎么發(fā)消息”而在“消息丟了怎么辦”“消息亂序了怎么辦”“如何區(qū)分不同來源的數(shù)據(jù)報(bào)”。這些問題值得在實(shí)際項(xiàng)目中一點(diǎn)點(diǎn)驗(yàn)證和打磨。 Python UDP單聊實(shí)戰(zhàn)：從Socket原理到可靠傳輸實(shí)現(xiàn)
之前在做一些局域網(wǎng)內(nèi)的實(shí)時(shí)工具時(shí)想用最輕量的方式實(shí)現(xiàn)兩個(gè)節(jié)點(diǎn)互相發(fā)消息。第一反應(yīng)是 TCP后來發(fā)現(xiàn)很多場(chǎng)景其實(shí)用不到 TCP 的可靠流式傳輸反而 UDP 的“無連接 數(shù)據(jù)報(bào)”模型更簡(jiǎn)單直接。本文就來完整拆解如何用 UDP 實(shí)現(xiàn)一個(gè)可運(yùn)行的單聊程序覆蓋協(xié)議差異、Socket API、多線程收發(fā)、退出機(jī)制和常見坑點(diǎn)。本文適合學(xué)過 Python 基礎(chǔ)語法、想進(jìn)一步接觸網(wǎng)絡(luò)編程的讀者?？赐曛竽悴粌H能跑通一個(gè) UDP 單聊程序還能理解 UDP 與 TCP 在程序設(shè)計(jì)上的本質(zhì)差異知道為什么 UDP 丟包時(shí)需要自己做確認(rèn)和重傳以及在實(shí)際項(xiàng)目中如何給這種“裸 UDP 通信”加上可靠性和安全邊界。1. 背景與核心概念UDP 單聊到底在解決什么問題1.1 什么是 UDP 單聊單聊就是兩個(gè)終端之間的一對(duì)一消息通信。TCP 可以實(shí)現(xiàn)單聊UDP 同樣可以實(shí)現(xiàn)單聊。區(qū)別在于協(xié)議的傳輸模型TCP 是面向連接的字節(jié)流雙方必須先建立連接然后像讀寫文件一樣收發(fā)數(shù)據(jù)UDP 是面向無連接的數(shù)據(jù)報(bào)發(fā)送方只需要知道對(duì)方的 IP 和端口就可以把一段獨(dú)立的數(shù)據(jù)報(bào)發(fā)出去不需要提前“握手”。所以 UDP 單聊的本質(zhì)是兩個(gè) socket 端點(diǎn)之間互相發(fā)送數(shù)據(jù)報(bào)每個(gè)數(shù)據(jù)報(bào)是一個(gè)完整的消息。由于沒有連接狀態(tài)服務(wù)端和客戶端的身份不像 TCP 那樣固定兩個(gè)端點(diǎn)都可以主動(dòng)給對(duì)方發(fā)消息也可以隨時(shí)監(jiān)聽對(duì)方發(fā)來的消息。1.2 UDP 與 TCP 的核心差異要真正理解 UDP 單聊必須先分清 UDP 和 TCP 的幾個(gè)關(guān)鍵差異。對(duì)比維度TCPUDP連接狀態(tài)面向連接通信前必須三次握手無連接直接發(fā)送數(shù)據(jù)報(bào)傳輸單位字節(jié)流無消息邊界數(shù)據(jù)報(bào)每個(gè)報(bào)文獨(dú)立可靠性可靠傳輸支持確認(rèn)、重傳、排序不可靠不保證送達(dá)、不保證順序傳輸效率有握手和確認(rèn)開銷相對(duì)慢無握手和確認(rèn)開銷相對(duì)快應(yīng)用場(chǎng)景文件傳輸、網(wǎng)頁訪問、遠(yuǎn)程登錄實(shí)時(shí)音視頻、DNS 查詢、局域網(wǎng)發(fā)現(xiàn)其中最關(guān)鍵的一點(diǎn)是“消息邊界”。TCP 是流式協(xié)議發(fā)送方調(diào)用一次 send 發(fā)送的內(nèi)容可能在接收方多次 recv 才能讀完整也可能多次 send 的內(nèi)容被一次 recv 讀出來所以業(yè)務(wù)層必須自己定義消息邊界。UDP 則不同每次 sendto 發(fā)送一個(gè)數(shù)據(jù)報(bào)接收方每次 recvfrom 讀取一個(gè)完整的數(shù)據(jù)報(bào)消息邊界由操作系統(tǒng)維護(hù)。對(duì)于聊天這種天然以“一條消息”為單位的場(chǎng)景UDP 的數(shù)據(jù)報(bào)模型反而更直觀。另外UDP 的不可靠性也需要重點(diǎn)理解。UDP 不保證數(shù)據(jù)報(bào)一定能到達(dá)對(duì)端也不保證多個(gè)數(shù)據(jù)報(bào)的到達(dá)順序和發(fā)送順序一致。在局域網(wǎng)內(nèi)丟包率往往很低所以很多人第一次跑 UDP 通信會(huì)覺得“很可靠”但這是一種錯(cuò)覺。在跨公網(wǎng)、跨 Wi-Fi、網(wǎng)絡(luò)擁塞的場(chǎng)景下UDP 丟包和亂序問題會(huì)明顯暴露出來。本文先實(shí)現(xiàn)一個(gè)基礎(chǔ)可用的單聊程序第 6 章再討論如何在應(yīng)用層解決可靠性問題。1.3 UDP 單聊的典型應(yīng)用場(chǎng)景UDP 單聊看起來不如 TCP 聊天那么“標(biāo)準(zhǔn)”但它在很多場(chǎng)景下非常實(shí)用局域網(wǎng)工具兩個(gè)內(nèi)網(wǎng)節(jié)點(diǎn)之間快速互發(fā)信令不需要建連不需要維護(hù)連接狀態(tài)。實(shí)時(shí)性優(yōu)先的通信比如游戲內(nèi)的語音指令、視頻通話的信令通道實(shí)時(shí)性要求高于可靠性要求。嵌入式與 IoT 設(shè)備很多物聯(lián)網(wǎng)設(shè)備資源有限UDP 實(shí)現(xiàn)簡(jiǎn)單、占用資源少。自定義協(xié)議的原型驗(yàn)證先快速驗(yàn)證消息收發(fā)邏輯再?zèng)Q定是否要增加可靠層。如果你的場(chǎng)景需要嚴(yán)格的可靠消息、大文件傳輸、消息順序保證那么請(qǐng)選擇 TCP 或基于 TCP 的協(xié)議如 WebSocket、MQTT。如果只是輕量級(jí)消息互發(fā)且能接受消息偶爾丟失UDP 會(huì)是一個(gè)更簡(jiǎn)單的選擇。2. 環(huán)境準(zhǔn)備與前置知識(shí)2.1 運(yùn)行環(huán)境本文示例使用 Python 實(shí)現(xiàn)因?yàn)樗Z法簡(jiǎn)潔、標(biāo)準(zhǔn)庫直接支持 Socket 編程適合快速理解和驗(yàn)證 UDP 通信模型。項(xiàng)目說明操作系統(tǒng)Windows / Linux / macOS 均可Python 版本3.6 及以上依賴庫僅使用 Python 標(biāo)準(zhǔn)庫 socket、threading、sys環(huán)境要求本機(jī)或局域網(wǎng)內(nèi)兩臺(tái)機(jī)器本文不再贅述 Python 安裝步驟如果還沒有 Python 環(huán)境可以從 Python 官網(wǎng)下載并安裝安裝時(shí)記得勾選“Add Python to PATH”。2.2 需要提前掌握的 Socket 概念在寫代碼之前先梳理幾個(gè)必須掌握的 Socket 概念。IP 地址標(biāo)識(shí)網(wǎng)絡(luò)中的一臺(tái)主機(jī)。本機(jī)回環(huán)地址是 127.0.0.1局域網(wǎng)地址通常是 192.168.x.x。端口標(biāo)識(shí)主機(jī)上的一個(gè)網(wǎng)絡(luò)進(jìn)程。兩個(gè)程序不能同時(shí)綁定同一個(gè)端口。Socket操作系統(tǒng)提供的一個(gè)網(wǎng)絡(luò)通信句柄應(yīng)用層通過它讀寫網(wǎng)絡(luò)數(shù)據(jù)。AF_INET地址族表示使用 IPv4 地址。SOCK_DGRAM套接字類型表示使用數(shù)據(jù)報(bào)方式也就是 UDP。這些概念不需要背但要在看到代碼時(shí)能對(duì)上號(hào)。接下來我會(huì)逐個(gè)拆解 UDP 編程中會(huì)用到的核心 API。3. UDP Socket 核心 API 拆解3.1 創(chuàng)建套接字socket.socket(AF_INET, SOCK_DGRAM)UDP 套接字的創(chuàng)建方式和 TCP 一樣都是調(diào)用socket.socket()但類型參數(shù)不同。import socket udp_socket socket.socket(socket.AF_INET, socket.SOCK_DGRAM)這里的AF_INET表示 IPv4 地址族SOCK_DGRAM表示數(shù)據(jù)報(bào)套接字。創(chuàng)建成功后udp_socket就是一個(gè)可以用于 UDP 收發(fā)數(shù)據(jù)的對(duì)象。3.2 綁定地址bind()bind()的作用是把套接字綁定到一個(gè)本地地址和端口上。只有綁定后操作系統(tǒng)才知道把收到的 UDP 數(shù)據(jù)報(bào)交給哪個(gè)進(jìn)程。udp_socket.bind((0.0.0.0, 8001))第一個(gè)參數(shù)是本地 IP 地址0.0.0.0表示監(jiān)聽本機(jī)所有網(wǎng)卡這樣無論是從 127.0.0.1 還是局域網(wǎng) IP 發(fā)來的數(shù)據(jù)報(bào)都能收到。如果只想監(jiān)聽回環(huán)地址可以改成127.0.0.1。第二個(gè)參數(shù)是本地端口號(hào)。注意如果端口已經(jīng)被其他程序占用bind()會(huì)拋出OSError。另外UDP 不需要像 TCP 一樣調(diào)用listen()因?yàn)?UDP 沒有連接隊(duì)列的概念。3.3 發(fā)送數(shù)據(jù)sendto()UDP 發(fā)送數(shù)據(jù)用的是sendto()它需要指定目標(biāo)和端口。message 你好UDP.encode(utf-8) udp_socket.sendto(message, (127.0.0.1, 8002))sendto()的第一個(gè)參數(shù)是字節(jié)串所以字符串要用encode(utf-8)轉(zhuǎn)成字節(jié)。第二個(gè)參數(shù)是一個(gè)(ip, port)元組表示對(duì)端的地址和端口。這里有一個(gè)重要細(xì)節(jié)UDP 的sendto()即使對(duì)端不存在本地也不會(huì)立刻報(bào)錯(cuò)。因?yàn)?UDP 是無連接協(xié)議發(fā)送方只負(fù)責(zé)把數(shù)據(jù)報(bào)交給操作系統(tǒng)至于數(shù)據(jù)報(bào)能不能到達(dá)對(duì)端發(fā)送方無法馬上知道。這在調(diào)試時(shí)很容易造成“消息發(fā)出去了但對(duì)方?jīng)]收到”的假象。3.4 接收數(shù)據(jù)recvfrom()UDP 接收數(shù)據(jù)用的是recvfrom()它會(huì)阻塞等待數(shù)據(jù)報(bào)到來并在收到數(shù)據(jù)后同時(shí)返回?cái)?shù)據(jù)和發(fā)送方地址。data, addr udp_socket.recvfrom(1024)1024表示緩沖區(qū)大小單位是字節(jié)。UDP 是數(shù)據(jù)報(bào)協(xié)議一次recvfrom()讀取一個(gè)完整的數(shù)據(jù)報(bào)。如果數(shù)據(jù)報(bào)的大小超過緩沖區(qū)大小超出的部分會(huì)被截?cái)唷Ｋ栽谠O(shè)計(jì)協(xié)議時(shí)要控制單條消息大小盡量小于緩沖區(qū)值。addr是發(fā)送方的(ip, port)元組。在單聊場(chǎng)景中我們可以通過addr判斷消息是誰發(fā)來的方便后續(xù)做多端擴(kuò)展。3.5 close() 與資源釋放通信結(jié)束后需要調(diào)用close()關(guān)閉套接字釋放系統(tǒng)資源。udp_socket.close()關(guān)閉后套接字不能再用于收發(fā)數(shù)據(jù)。實(shí)際項(xiàng)目中建議使用try-finally或with語法確保資源釋放。3.6 可選 connectUDP 也有“連接”有些讀者可能聽說過 UDP 也可以調(diào)用connect()。這里特別說明一下UDP 的connect()并不是建立真正的連接它只是把對(duì)端地址保存到內(nèi)核中之后就可以用send()和recv()代替sendto()和recvfrom()寫法更簡(jiǎn)潔。udp_socket.connect((127.0.0.1, 8002)) udp_socket.send(message.encode(utf-8)) data udp_socket.recv(1024)調(diào)用connect()之后UDP 套接字只能和這個(gè)固定對(duì)端通信而且可以通過recv()收到對(duì)端返回的 ICMP 端口不可達(dá)錯(cuò)誤從而更快感知對(duì)端異常。但本文的基礎(chǔ)示例使用sendto()和recvfrom()因?yàn)檫@兩個(gè)方法更直觀地體現(xiàn)了 UDP 無連接、每次指定目標(biāo)地址的特點(diǎn)。4. 單聊程序的設(shè)計(jì)與實(shí)現(xiàn)4.1 程序職責(zé)劃分一個(gè)最簡(jiǎn)單的 UDP 單聊程序需要完成四件事創(chuàng)建 UDP 套接字并綁定本地端口。持續(xù)接收來自對(duì)端的消息并打印。持續(xù)讀取用戶輸入并發(fā)送給對(duì)端。支持退出機(jī)制關(guān)閉套接字。由于接收消息和發(fā)送消息是兩個(gè)相互獨(dú)立的動(dòng)作如果只有一個(gè)主循環(huán)就會(huì)出現(xiàn)“正在等待用戶輸入時(shí)無法接收消息”的問題。因此程序需要拆成兩個(gè)線程主線程負(fù)責(zé)讀取用戶輸入并發(fā)送消息。接收線程負(fù)責(zé)循環(huán)調(diào)用recvfrom()收到消息后打印。4.2 收發(fā)線程模型線程模型用文字描述是這樣程序啟動(dòng) | -- 創(chuàng)建 UDP Socket | -- bind 本地端口 | -- 啟動(dòng)接收線程循環(huán) recvfrom | -- 主循環(huán)循環(huán) input sendto | -- 用戶輸入 exit 或收到對(duì)方退出通知 | -- 退出程序接收線程設(shè)置為守護(hù)線程daemonTrue。守護(hù)線程的特點(diǎn)是當(dāng)主線程結(jié)束時(shí)守護(hù)線程會(huì)自動(dòng)終止。這樣即使接收線程還阻塞在recvfrom()主線程退出后程序也能正常結(jié)束。4.3 數(shù)據(jù)格式與編碼Socket 收發(fā)的是字節(jié)串而聊天內(nèi)容是字符串所以需要統(tǒng)一編碼。本文示例統(tǒng)一使用 UTF-8 編碼發(fā)送時(shí)message.encode(utf-8)接收時(shí)data.decode(utf-8)關(guān)于消息格式這里使用最簡(jiǎn)單的純文本消息。但在實(shí)際項(xiàng)目中建議設(shè)計(jì)為包含消息類型、消息序號(hào)、時(shí)間戳等字段的結(jié)構(gòu)化格式例如 JSON 字符串或自定義二進(jìn)制協(xié)議這樣后期增加 ACK、心跳、文件傳輸?shù)裙δ軙r(shí)更容易擴(kuò)展。4.4 退出機(jī)制退出機(jī)制是單聊程序最容易忽略的地方。基礎(chǔ)版本可以約定用戶輸入exit時(shí)退出程序同時(shí)向?qū)Χ税l(fā)送一條特殊消息__EXIT__通知對(duì)方自己已經(jīng)離開。由于 UDP 不可靠這條__EXIT__消息不能保證一定送達(dá)。所以更嚴(yán)謹(jǐn)?shù)脑O(shè)計(jì)是即使對(duì)方?jīng)]有收到退出通知發(fā)送方也可以直接退出接收方之后檢測(cè)到超時(shí)或心跳超時(shí)才判定對(duì)端離線。本文示例采用盡力通知的方式讓讀者理解“應(yīng)用層協(xié)議約定”的概念。5. 完整可運(yùn)行代碼5.1 項(xiàng)目結(jié)構(gòu)本文示例不需要復(fù)雜項(xiàng)目結(jié)構(gòu)一個(gè) Python 文件即可。udp_chat/ └── udp_chat.py在終端中兩個(gè)聊天方分別運(yùn)行這個(gè)腳本并傳入不同的本地端口和對(duì)方地址。5.2 完整代碼 udp_chat.py下面是完整代碼可以直接復(fù)制保存為udp_chat.py運(yùn)行。# 文件路徑udp_chat/udp_chat.py import socket import threading import sys # 全局退出事件任何一方退出時(shí)用于通知另一個(gè)線程停止 exit_event threading.Event() def receive_message(udp_socket: socket.socket): 接收線程循環(huán)接收 UDP 數(shù)據(jù)報(bào)并打印 while not exit_event.is_set(): try: data, addr udp_socket.recvfrom(1024) message data.decode(utf-8) if message __EXIT__: exit_event.set() print(\n[系統(tǒng)] 對(duì)方已退出聊天按回車可退出。) break print(f\n[對(duì)方] {message}) print( , end, flushTrue) except OSError: # 套接字被關(guān)閉時(shí)退出接收線程 break except KeyboardInterrupt: exit_event.set() break def start_chat(local_port: int, remote_ip: str, remote_port: int): 啟動(dòng) UDP 單聊主邏輯 # 1. 創(chuàng)建 UDP 套接字 udp_socket socket.socket(socket.AF_INET, socket.SOCK_DGRAM) # 2. 綁定本地端口0.0.0.0 表示監(jiān)聽本機(jī)所有網(wǎng)卡 udp_socket.bind((0.0.0.0, local_port)) # 3. 啟動(dòng)接收線程 receiver threading.Thread( targetreceive_message, args(udp_socket,), daemonTrue ) receiver.start() print(fUDP 單聊已啟動(dòng)本地端口 {local_port} - {remote_ip}:{remote_port}) print(輸入消息回車發(fā)送輸入 exit 退出。) try: # 4. 主線程循環(huán)讀取用戶輸入并發(fā)送 while not exit_event.is_set(): text input( ) if not text.strip(): continue if text.strip().lower() exit: # 盡力通知對(duì)方自己已退出 udp_socket.sendto(__EXIT__.encode(utf-8), (remote_ip, remote_port)) exit_event.set() break # 5. 發(fā)送數(shù)據(jù)報(bào) udp_socket.sendto(text.encode(utf-8), (remote_ip, remote_port)) except KeyboardInterrupt: print(\n[系統(tǒng)] 用戶主動(dòng)中斷正在退出。) exit_event.set() finally: # 6. 關(guān)閉套接字釋放系統(tǒng)資源 udp_socket.close() print([系統(tǒng)] 聊天已結(jié)束。) if __name__ __main__: if len(sys.argv) ! 4: print(用法: python udp_chat.py 本地端口 遠(yuǎn)端IP 遠(yuǎn)端端口) print(示例: ) print( A 端: python udp_chat.py 8001 127.0.0.1 8002) print( B 端: python udp_chat.py 8002 127.0.0.1 8001) sys.exit(1) start_chat( local_portint(sys.argv[1]), remote_ipsys.argv[2], remote_portint(sys.argv[3]), )5.3 運(yùn)行方式打開兩個(gè)終端窗口在第一個(gè)窗口運(yùn)行 A 端python udp_chat.py 8001 127.0.0.1 8002在第二個(gè)窗口運(yùn)行 B 端python udp_chat.py 8002 127.0.0.1 8001兩個(gè)終端都在本機(jī)所以 IP 都填寫回環(huán)地址127.0.0.1。A 端綁定本地端口8001發(fā)往127.0.0.1:8002B 端綁定本地端口8002發(fā)往127.0.0.1:8001。如果兩臺(tái)機(jī)器在同一個(gè)局域網(wǎng)內(nèi)則 IP 填寫對(duì)方機(jī)器的局域網(wǎng) IP例如python udp_chat.py 8001 192.168.1.100 8002注意局域網(wǎng)通信時(shí)需要保證兩臺(tái)機(jī)器的防火墻允許對(duì)應(yīng) UDP 端口的入站流量。5.4 預(yù)期運(yùn)行效果A 端窗口輸出UDP 單聊已啟動(dòng)本地端口 8001 - 127.0.0.1:8002 輸入消息回車發(fā)送輸入 exit 退出。 你好我是 A [對(duì)方] 你好我是 B 收到B 你好 [對(duì)方] 我們退出吧 exit [系統(tǒng)] 聊天已結(jié)束。B 端窗口輸出UDP 單聊已啟動(dòng)本地端口 8002 - 127.0.0.1:8001 輸入消息回車發(fā)送輸入 exit 退出。 [對(duì)方] 你好我是 A 你好我是 B [對(duì)方] 收到B 你好 我們退出吧 [系統(tǒng)] 對(duì)方已退出聊天按回車可退出。到這里一個(gè)基礎(chǔ)的 UDP 單聊程序就跑通了。它的核心邏輯只有五步創(chuàng)建套接字、綁定端口、接收線程、主線程輸入、關(guān)閉套接字。后面我們?cè)賮砜纯慈绾卧谶@個(gè)基礎(chǔ)上增加可靠性保障。6. 進(jìn)階為 UDP 單聊增加可靠性保障6.1 為什么需要應(yīng)用層確認(rèn)UDP 本身不保證消息可靠到達(dá)所以要想在 UDP 上實(shí)現(xiàn)更可靠的聊天必須在應(yīng)用層自己實(shí)現(xiàn)確認(rèn)和重傳機(jī)制。最簡(jiǎn)單的模型是發(fā)送方給每條消息編號(hào)。接收方收到消息后回復(fù)一條 ACK表示“某號(hào)消息已收到”。發(fā)送方如果一段時(shí)間內(nèi)沒有收到 ACK就重發(fā)該消息。這本質(zhì)上是在應(yīng)用層模仿 TCP 的確認(rèn)重傳機(jī)制但實(shí)現(xiàn)比 TCP 簡(jiǎn)單得多也足夠滿足很多輕量級(jí)通信需求。6.2 消息編號(hào)與 ACK 的簡(jiǎn)化設(shè)計(jì)這里給出一套簡(jiǎn)化的消息協(xié)議設(shè)計(jì)思路不展開完整代碼重點(diǎn)說明流程。發(fā)送方 接收方 |-------- msg:1:你好 -------- | |------- ack:1 -------------- | |-------- msg:2:在嗎 -------- | |------- ack:2 -------------- |消息格式可以采用類型:序號(hào):內(nèi)容的文本格式msg:1:你好ack:1發(fā)送方在發(fā)送消息時(shí)記錄消息序號(hào)并啟動(dòng)一個(gè)定時(shí)器。如果超時(shí)沒有收到對(duì)應(yīng)的 ACK則重發(fā)。接收方收到msg消息后立即回一條ack。6.3 核心代碼思路下面是消息格式的解析和構(gòu)造核心代碼可以作為擴(kuò)展參考。# 文件路徑udp_chat/ack_protocol.py def build_message(seq: int, content: str) - bytes: 構(gòu)造一條帶序號(hào)的聊天消息 return fmsg:{seq}:{content}.encode(utf-8) def build_ack(seq: int) - bytes: 構(gòu)造一條 ACK 確認(rèn)消息 return fack:{seq}.encode(utf-8) def parse_message(data: bytes): 解析收到的數(shù)據(jù)返回 (類型, 序號(hào), 內(nèi)容) text data.decode(utf-8) parts text.split(:, 2) if len(parts) 2: return None msg_type parts[0] seq int(parts[1]) content parts[2] if len(parts) 2 else return msg_type, seq, content在實(shí)際項(xiàng)目中還可以給消息增加時(shí)間戳、消息 ID、簽名等字段并引入滑動(dòng)窗口機(jī)制來提升吞吐量。對(duì)于單聊這種低頻場(chǎng)景每發(fā)一條消息就等待 ACK 的簡(jiǎn)單方式通常已經(jīng)夠用。7. 常見問題與排查思路7.1 高頻問題排查表問題現(xiàn)象常見原因解決思路兩端都運(yùn)行了但收不到消息地址或端口填錯(cuò)核對(duì)本地綁定端口和對(duì)端發(fā)送端口收不到消息且程序綁定報(bào)錯(cuò)端口被占用端口已被其他程序占用更換端口或用netstat查看端口占用中文亂碼編碼/解碼不一致統(tǒng)一使用 UTF-8 編碼消息被截?cái)嘞⒊^ recvfrom 緩沖區(qū)大小增大緩沖區(qū)或限制單條消息大小一方退出后另一方卡在輸入狀態(tài)輸入阻塞導(dǎo)致線程無法及時(shí)退出用退出事件 提示用戶按回車退出局域網(wǎng)內(nèi)收不到消息防火墻攔截 UDP 入站流量在防火墻中放行對(duì)應(yīng) UDP 端口7.2 幾個(gè)典型問題的詳細(xì)分析先說“收不到消息”。這是 UDP 調(diào)試中最高頻的問題。首先是檢查地址和端口A 發(fā)送給 B必須確保 A 填寫的目標(biāo)地址是 B 綁定的本地端口而不是 A 自己的端口。很多初學(xué)者會(huì)把local_port和remote_port搞混導(dǎo)致數(shù)據(jù)報(bào)發(fā)送給了自己。其次是防火墻問題。UDP 沒有連接狀態(tài)防火墻很難判斷一個(gè) UDP 數(shù)據(jù)報(bào)是否是“應(yīng)答包”所以很多系統(tǒng)的默認(rèn)策略會(huì)攔截來自外部的 UDP 入站請(qǐng)求。在局域網(wǎng)內(nèi)調(diào)試時(shí)可以先在防火墻中放行指定端口或者先用127.0.0.1回環(huán)地址測(cè)試排除網(wǎng)絡(luò)層因素。再說“消息截?cái)唷?。recvfrom(1024)表示最多讀取 1024 字節(jié)。如果發(fā)送的數(shù)據(jù)報(bào)超過 1024 字節(jié)超出的部分會(huì)被丟棄而且接收方不會(huì)收到任何“數(shù)據(jù)被截?cái)唷钡奶崾?。建議設(shè)計(jì)協(xié)議時(shí)把單條消息控制在 512 字節(jié)以內(nèi)或者在接收時(shí)使用更大的緩沖區(qū)比如recvfrom(65535)。最后是“程序無法退出”。由于input()是阻塞的當(dāng)接收線程收到對(duì)方退出通知并設(shè)置exit_event時(shí)主線程可能還卡在input()上。此時(shí)需要用戶按一次回車讓input()返回主線程才能檢查退出事件并結(jié)束。這是一個(gè)比較常見的交互問題本文代碼里已經(jīng)通過提示信息說明。8. 工程實(shí)踐建議8.1 消息大小與緩沖區(qū)UDP 數(shù)據(jù)報(bào)的長度受限于網(wǎng)絡(luò) MTU最大傳輸單元。在以太網(wǎng)環(huán)境中MTU 通常為 1500 字節(jié)扣除 IP 頭和 UDP 頭后建議單條 UDP 數(shù)據(jù)報(bào)不要超過 1472 字節(jié)。如果消息體積較大建議在應(yīng)用層拆分成多個(gè)數(shù)據(jù)報(bào)并給每個(gè)數(shù)據(jù)報(bào)編號(hào)接收方按編號(hào)重組。在實(shí)際代碼中recvfrom()的緩沖區(qū)大小可以設(shè)為 65535這是 UDP 數(shù)據(jù)報(bào)的最大長度能避免大部分截?cái)鄦栴}。8.2 安全與權(quán)限邊界UDP 無連接的特性也帶來安全風(fēng)險(xiǎn)任何人都可以向你的綁定端口發(fā)送數(shù)據(jù)報(bào)因此程序必須校驗(yàn)消息來源。如果服務(wù)暴露在公網(wǎng)容易被惡意掃描和 UDP Flood 攻擊。不要在公網(wǎng)環(huán)境中直接運(yùn)行裸 UDP 單聊程序建議增加鑒權(quán)、白名單、消息簽名等機(jī)制。本文代碼中接收線程打印了addr實(shí)際項(xiàng)目中應(yīng)加入來源 IP 白名單校驗(yàn)。需要特別說明的是本文所有示例僅用于合法授權(quán)的開發(fā)測(cè)試環(huán)境請(qǐng)不要對(duì)非授權(quán)目標(biāo)發(fā)送任何 UDP 流量。8.3 日志與可觀測(cè)性UDP 調(diào)試比 TCP 困難因?yàn)樗鼪]有連接狀態(tài)發(fā)生問題時(shí)不方便直接觀察“連接在哪一步斷了”。因此在工程實(shí)現(xiàn)中要增加日志收到消息時(shí)打印發(fā)送方地址、消息序號(hào)、消息長度。發(fā)送消息時(shí)打印目標(biāo)地址、消息序號(hào)。記錄丟包、重傳、異常等事件。有了日志就能快速定位是發(fā)送失敗、接收失敗還是網(wǎng)絡(luò)丟包。8.4 多環(huán)境驗(yàn)證建議按以下順序驗(yàn)證 UDP 單聊程序本機(jī)回環(huán)測(cè)試兩個(gè)終端都使用127.0.0.1排除網(wǎng)絡(luò)因素。局域網(wǎng)測(cè)試兩臺(tái)機(jī)器用局域網(wǎng) IP 通信驗(yàn)證防火墻和路由是否正常?？缇W(wǎng)段或 Wi-Fi 測(cè)試觀察弱網(wǎng)環(huán)境下的丟包和延遲表現(xiàn)評(píng)估是否需要增加應(yīng)用層可靠性機(jī)制。只有在多環(huán)境下驗(yàn)證過才能判斷當(dāng)前程序是否滿足需求。9. 總結(jié)與下一步學(xué)習(xí)方向本文圍繞 UDP 單聊程序從協(xié)議差異、Socket API、線程模型、完整代碼到可靠性擴(kuò)展做了一個(gè)相對(duì)完整的梳理。你現(xiàn)在可以動(dòng)手把第 5 章的代碼復(fù)制下來先在本機(jī)兩個(gè)終端中跑通回環(huán)通信再嘗試改成局域網(wǎng)內(nèi)的兩臺(tái)機(jī)器通信。跑通之后建議按以下方向繼續(xù)深入給程序加入 ACK 確認(rèn)和超時(shí)重傳理解應(yīng)用層可靠性的實(shí)現(xiàn)思路。將消息格式升級(jí)為 JSON 或自定義二進(jìn)制協(xié)議增加消息類型、時(shí)間戳、序列號(hào)等字段。把單聊擴(kuò)展為多人群聊需要在數(shù)據(jù)報(bào)中攜帶“房間號(hào)”或“目標(biāo)用戶 ID”等信息。學(xué)習(xí)如何用select、poll或異步 IO 代替多線程模型優(yōu)化高并發(fā)下的資源占用。UDP 最大的特點(diǎn)是無連接、效率高但不可靠。真正的難點(diǎn)往往不在“怎么發(fā)消息”而在“消息丟了怎么辦”“消息亂序了怎么辦”“如何區(qū)分不同來源的數(shù)據(jù)報(bào)”。這些問題值得在實(shí)際項(xiàng)目中一點(diǎn)點(diǎn)驗(yàn)證和打磨。