4. Socket编程基础：从概念到实战

聊完了UDP协议的原理，咱们得动手了。写网络程序，绕不开一个东西——Socket。说白了，它就是操作系统给应用层开的一扇门，让数据能进出网络。

4.1 什么是Socket？

Socket这个词，直译是“插座”。你想想看，墙上的插座连接了电器和电网。Socket连接了应用程序和网络协议栈。它是个抽象层，把底下TCP/IP的复杂性全藏起来了。

我刚开始学的时候，总觉得Socket很神秘。后来做项目多了才明白，它就是个文件描述符。对，你没听错。在Linux里，一切皆文件。Socket也不例外。你打开一个Socket，得到一个整数句柄，然后对着它读写数据就行。

核心理解：Socket是应用层与传输层之间的编程接口。它屏蔽了网络协议的细节，让开发者像操作文件一样收发数据。

4.2 Socket的类型

Socket分好几种，咱们UDP通信用的是SOCK_DGRAM。我整理了一张表，把常见的类型列出来：

类型 常量名 传输层协议 特点
流式Socket SOCK_STREAM TCP 面向连接、可靠、有序、无边界
数据报Socket SOCK_DGRAM UDP 无连接、不可靠、无序、有边界
原始Socket SOCK_RAW IP层 直接操作IP包，需要root权限

咱们这章只关心SOCK_DGRAM。它对应UDP，每次发送都是一个独立的数据报。你发一次，对方收一次，边界清清楚楚。

个人经验：我做过一个日志采集系统，用的就是SOCK_DGRAM。因为日志丢几条无所谓，但延迟必须低。如果用SOCK_STREAM，TCP的重传机制反而会拖慢整体吞吐。选对类型，事半功倍。

4.3 Python socket模块核心函数

Python的socket模块封装了系统调用，用起来很顺手。咱们UDP编程，核心就五个函数：socket()、bind()、sendto()、recvfrom()、close()。

先看一张流程图，把整个调用关系理清楚：

UDP Socket 核心调用流程 服务端 socket(AF_INET, SOCK_DGRAM) bind(('0.0.0.0', 8888)) recvfrom(bufsize) sendto(data, addr) close() 客户端 socket(AF_INET, SOCK_DGRAM) sendto(data, ('ip', port)) recvfrom(bufsize) close() 无连接，直接发 谁发来就回给谁

4.4 逐个拆解核心函数

4.4.1 socket() — 创建端点

这是第一步。告诉操作系统：我要用UDP通信。

import socket

# 创建UDP Socket
udp_sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

AF_INET表示IPv4地址族。SOCK_DGRAM就是数据报Socket。这两个参数组合起来，就是UDP。

我建议：创建Socket时养成好习惯，用with语句或者显式close()。我曾经在线上环境遇到过文件描述符泄漏，就是因为忘了关Socket。后果嘛……服务跑了一周，突然连不上数据库了。排查了半天，发现是Socket句柄用光了。

4.4.2 bind() — 绑定地址和端口

服务端必须调用bind()，告诉内核：这个端口归我了。客户端一般不用，除非你想固定自己的端口。

# 服务端绑定到本机的8888端口
server_addr = ('0.0.0.0', 8888)
udp_sock.bind(server_addr)

0.0.0.0表示监听所有网络接口。如果你只想监听某个网卡，可以写具体的IP，比如'192.168.1.100'。

注意：端口号范围是0~65535。0~1023是知名端口，需要root权限才能绑定。开发时建议用1024以上的端口，比如8888、9999。我曾经图省事用了80端口，结果每次跑都要sudo，烦得很。

4.4.3 sendto() — 发送数据报

UDP是无连接的，所以发送时必须指定目标地址。每次sendto()都是一个独立的数据报。

# 客户端发送数据
message = b'Hello, UDP Server!'
dest_addr = ('192.168.1.10', 8888)
sent = udp_sock.sendto(message, dest_addr)

print(f'发送了 {sent} 字节')

注意，sendto()返回实际发送的字节数。在UDP里，如果数据报太大（超过MTU），IP层会分片。但Python的sendto()一般不会帮你分片，数据太大就直接报错了。

避坑指南：UDP数据报建议控制在1472字节以内（以太网MTU 1500 - IP头20 - UDP头8）。超过这个值，IP层会分片。分片后只要丢一片，整个数据报就废了。我曾经做过一个视频传输项目，没注意这个限制，结果丢包率飙升到30%。后来把每个包切成1400字节，丢包率直接降到1%以下。

4.4.4 recvfrom() — 接收数据报

接收数据时，不光拿到数据，还能知道是谁发来的。recvfrom()返回一个元组：(data, address)。

# 服务端接收数据
data, client_addr = udp_sock.recvfrom(2048)
print(f'收到来自 {client_addr} 的消息: {data.decode()}')

参数2048是缓冲区大小。如果实际数据超过这个值，多余的部分会被截断。嗯，这里要注意，UDP不会像TCP那样帮你拼包。你设多大缓冲区，最多就收多少。

个人习惯：我一般设成65535，这是UDP数据报的最大长度。虽然有点浪费内存，但省心。反正Python的字符串对象会动态管理，实际占不了那么多。

4.4.5 close() — 关闭Socket

用完记得关。释放内核资源，让端口可以被其他程序复用。

udp_sock.close()

或者用with语句自动管理：

with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
    s.sendto(b'hello', ('localhost', 8888))
    # 退出with块时自动close()

4.5 完整示例：UDP回显服务器

光说不练假把式。咱们写一个完整的UDP回显服务器和客户端。服务器收到什么，就原样返回。

服务端代码：

import socket

def udp_echo_server(host='0.0.0.0', port=8888):
    # 1. 创建Socket
    with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
        # 2. 绑定地址
        s.bind((host, port))
        print(f'UDP回显服务器启动，监听 {host}:{port}')

        while True:
            # 3. 接收数据
            data, addr = s.recvfrom(65535)
            print(f'收到 {addr} 的消息: {data.decode()}')

            # 4. 原样返回
            s.sendto(data, addr)
            print(f'已回显给 {addr}')

if __name__ == '__main__':
    udp_echo_server()

客户端代码：

import socket

def udp_echo_client(server_host='127.0.0.1', server_port=8888):
    with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
        # 客户端不需要bind，系统自动分配临时端口
        message = b'Hello, UDP!'
        s.sendto(message, (server_host, server_port))
        print(f'发送: {message.decode()}')

        # 接收回显
        data, addr = s.recvfrom(65535)
        print(f'收到回显: {data.decode()} 来自 {addr}')

if __name__ == '__main__':
    udp_echo_client()

运行步骤：

  1. 先启动服务端：python udp_server.py
  2. 再启动客户端：python udp_client.py
  3. 客户端发送"Hello, UDP!"，服务端收到后原样返回
  4. 客户端打印收到的回显内容

4.6 常见问题与避坑

  • 端口被占用：启动服务端时报OSError: [Errno 98] Address already in use。等一会儿再试，或者换个端口。我一般用setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)来复用端口。
  • 收不到数据：检查防火墙。UDP没有连接状态，防火墙很容易误杀。我曾经在云服务器上折腾了两小时，最后发现是安全组没开UDP端口。
  • 数据乱码：确保收发两端的编码一致。我习惯统一用UTF-8，encode()和decode()配对使用。
  • 缓冲区溢出：recvfrom()的缓冲区设太小，数据会被截断。安全起见，用65535。

好了，Socket编程的基础就这些。五个函数，一张流程图，一个回显示例。你动手跑一遍，比看十遍都管用。记住，UDP编程的核心就一句话：创建Socket，绑定端口，发数据报，收数据报，关Socket。就这么简单。

← 上一章 📖 返回目录 下一章 →