Socket 网络编程
TCP/UDP套接字编程
前置知识
- POSIX 线程:建议先完成前一篇的学习
学习目标
- 掌握「概述」的核心机制、典型用法与常见陷阱
- 掌握「基础概念」的核心机制、典型用法与常见陷阱
- 掌握「快速上手」的核心机制、典型用法与常见陷阱
- 掌握「详细用法」的核心机制、典型用法与常见陷阱
- 掌握「常见场景」的核心机制、典型用法与常见陷阱
概述
Socket(套接字)是网络通信的端点,由IP地址和端口号标识。C语言通过BSD Socket API提供网络编程接口,支持TCP(可靠传输)和UDP(快速传输)两种主要协议。Socket编程是构建网络服务器、客户端应用和分布式系统的基础。
基础概念
TCP vs UDP
| 特性 | TCP | UDP |
|---|---|---|
| 连接 | 面向连接(三次握手) | 无连接 |
| 可靠性 | 可靠传输,保证顺序 | 不保证可靠和顺序 |
| 速度 | 较慢(有确认和重传) | 较快 |
| 适用场景 | 文件传输、Web、邮件 | 视频流、DNS、游戏 |
Socket 编程基本流程
TCP 服务器:socket -> bind -> listen -> accept -> recv/send -> close
TCP 客户端:socket -> connect -> recv/send -> close
UDP 服务器:socket -> bind -> recvfrom/sendto -> close
UDP 客户端:socket -> sendto/recvfrom -> close
核心数据结构
#include <netinet/in.h>
// IPv4 地址结构
struct sockaddr_in {
sa_family_t sin_family; // AF_INET
in_port_t sin_port; // 端口号(网络字节序)
struct in_addr sin_addr; // IP地址
char sin_zero[8];// 填充
};
// IPv6 地址结构
struct sockaddr_in6 {
sa_family_t sin6_family; // AF_INET6
in_port_t sin6_port;
uint32_t sin6_flowinfo;
struct in6_addr sin6_addr;
uint32_t sin6_scope_id;
};
快速上手
TCP 服务器
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <netinet/in.h>
int main(void) {
// 创建套接字
int server_fd = socket(AF_INET, SOCK_STREAM, 0);
// 允许地址复用
int opt = 1;
setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
// 绑定地址和端口
struct sockaddr_in addr = {
.sin_family = AF_INET,
.sin_port = htons(8080),
.sin_addr.s_addr = INADDR_ANY
};
bind(server_fd, (struct sockaddr *)&addr, sizeof(addr));
// 开始监听
listen(server_fd, 5);
printf("服务器监听 8080 端口\n");
// 接受客户端连接
struct sockaddr_in client_addr;
socklen_t client_len = sizeof(client_addr);
int client_fd = accept(server_fd, (struct sockaddr *)&client_addr, &client_len);
// 读取客户端数据
char buf[1024] = {0};
read(client_fd, buf, sizeof(buf));
printf("收到: %s\n", buf);
// 发送响应
const char *response = "Hello from server";
write(client_fd, response, strlen(response));
// 关闭连接
close(client_fd);
close(server_fd);
return 0;
}
TCP 客户端
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <netinet/in.h>
#include <arpa/inet.h>
int main(void) {
// 创建套接字
int sock = socket(AF_INET, SOCK_STREAM, 0);
// 设置服务器地址
struct sockaddr_in addr = {
.sin_family = AF_INET,
.sin_port = htons(8080),
.sin_addr.s_addr = inet_addr("127.0.0.1")
};
// 连接服务器
if (connect(sock, (struct sockaddr *)&addr, sizeof(addr)) < 0) {
perror("连接失败");
return 1;
}
// 发送数据
const char *msg = "Hello from client";
write(sock, msg, strlen(msg));
// 接收响应
char buf[1024] = {0};
read(sock, buf, sizeof(buf));
printf("服务器响应: %s\n", buf);
close(sock);
return 0;
}
详细用法
UDP 服务器
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <netinet/in.h>
int main(void) {
int sock = socket(AF_INET, SOCK_DGRAM, 0);
struct sockaddr_in addr = {
.sin_family = AF_INET,
.sin_port = htons(9090),
.sin_addr.s_addr = INADDR_ANY
};
bind(sock, (struct sockaddr *)&addr, sizeof(addr));
printf("UDP 服务器监听 9090 端口\n");
char buf[1024];
struct sockaddr_in client_addr;
socklen_t client_len = sizeof(client_addr);
while (1) {
// 接收数据
ssize_t n = recvfrom(sock, buf, sizeof(buf) - 1, 0,
(struct sockaddr *)&client_addr, &client_len);
buf[n] = '\0';
printf("收到: %s\n", buf);
// 发送响应
const char *response = "UDP 响应";
sendto(sock, response, strlen(response), 0,
(struct sockaddr *)&client_addr, client_len);
}
close(sock);
return 0;
}
UDP 客户端
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <netinet/in.h>
#include <arpa/inet.h>
int main(void) {
int sock = socket(AF_INET, SOCK_DGRAM, 0);
struct sockaddr_in addr = {
.sin_family = AF_INET,
.sin_port = htons(9090),
.sin_addr.s_addr = inet_addr("127.0.0.1")
};
// 发送数据
const char *msg = "UDP 请求";
sendto(sock, msg, strlen(msg), 0, (struct sockaddr *)&addr, sizeof(addr));
// 接收响应
char buf[1024];
recvfrom(sock, buf, sizeof(buf), 0, NULL, NULL);
printf("响应: %s\n", buf);
close(sock);
return 0;
}
常用套接字选项
#include <stdio.h>
#include <unistd.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
int main(void) {
int sock = socket(AF_INET, SOCK_STREAM, 0);
// 地址复用:服务器重启时可以立即绑定同一端口
int reuse = 1;
setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse));
// 发送超时
struct timeval timeout = { .tv_sec = 5, .tv_usec = 0 };
setsockopt(sock, SOL_SOCKET, SO_SNDTIMEO, &timeout, sizeof(timeout));
// 接收超时
setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &timeout, sizeof(timeout));
// 接收缓冲区大小
int bufsize = 65536;
setsockopt(sock, SOL_SOCKET, SO_RCVBUF, &bufsize, sizeof(bufsize));
// 禁用 Nagle 算法(减少延迟)
int nodelay = 1;
setsockopt(sock, IPPROTO_TCP, TCP_NODELAY, &nodelay, sizeof(nodelay));
// 保持连接
int keepalive = 1;
setsockopt(sock, SOL_SOCKET, SO_KEEPALIVE, &keepalive, sizeof(keepalive));
close(sock);
return 0;
}
常见场景
场景一:HTTP 服务器
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <netinet/in.h>
#include <signal.h>
#define PORT 8080
int main(void) {
signal(SIGCHLD, SIG_IGN);
int server_fd = socket(AF_INET, SOCK_STREAM, 0);
int opt = 1;
setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
struct sockaddr_in addr = {
.sin_family = AF_INET,
.sin_port = htons(PORT),
.sin_addr.s_addr = INADDR_ANY
};
bind(server_fd, (struct sockaddr *)&addr, sizeof(addr));
listen(server_fd, 10);
printf("HTTP 服务器运行在 http://localhost:%d\n", PORT);
while (1) {
int client_fd = accept(server_fd, NULL, NULL);
if (client_fd < 0) continue;
char buf[4096] = {0};
read(client_fd, buf, sizeof(buf));
// 构造 HTTP 响应
const char *body = "<html><body><h1>Hello, World!</h1></body></html>";
char response[4096];
snprintf(response, sizeof(response),
"HTTP/1.1 200 OK\r\n"
"Content-Type: text/html; charset=utf-8\r\n"
"Content-Length: %zu\r\n"
"Connection: close\r\n"
"\r\n%s",
strlen(body), body);
write(client_fd, response, strlen(response));
close(client_fd);
}
return 0;
}
场景二:多线程并发服务器
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <netinet/in.h>
#include <pthread.h>
#define PORT 8080
void *handle_client(void *arg) {
int client_fd = *(int *)arg;
free(arg);
char buf[1024];
while (1) {
ssize_t n = read(client_fd, buf, sizeof(buf) - 1);
if (n <= 0) break;
buf[n] = '\0';
printf("收到: %s\n", buf);
// 回显
write(client_fd, buf, n);
}
close(client_fd);
return NULL;
}
int main(void) {
int server_fd = socket(AF_INET, SOCK_STREAM, 0);
int opt = 1;
setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
struct sockaddr_in addr = {
.sin_family = AF_INET,
.sin_port = htons(PORT),
.sin_addr.s_addr = INADDR_ANY
};
bind(server_fd, (struct sockaddr *)&addr, sizeof(addr));
listen(server_fd, 10);
printf("回显服务器运行在端口 %d\n", PORT);
while (1) {
int client_fd = accept(server_fd, NULL, NULL);
if (client_fd < 0) continue;
int *fd_ptr = malloc(sizeof(int));
*fd_ptr = client_fd;
pthread_t tid;
pthread_create(&tid, NULL, handle_client, fd_ptr);
pthread_detach(tid);
}
return 0;
}
场景三:域名解析
#include <stdio.h>
#include <string.h>
#include <netdb.h>
#include <arpa/inet.h>
int main(void) {
const char *hostname = "www.example.com";
struct addrinfo hints = {0};
hints.ai_family = AF_INET; // IPv4
hints.ai_socktype = SOCK_STREAM;
struct addrinfo *result;
int ret = getaddrinfo(hostname, "80", &hints, &result);
if (ret != 0) {
fprintf(stderr, "域名解析失败: %s\n", gai_strerror(ret));
return 1;
}
for (struct addrinfo *rp = result; rp != NULL; rp = rp->ai_next) {
struct sockaddr_in *addr = (struct sockaddr_in *)rp->ai_addr;
char ip_str[INET_ADDRSTRLEN];
inet_ntop(AF_INET, &addr->sin_addr, ip_str, sizeof(ip_str));
printf("IP 地址: %s\n", ip_str);
}
freeaddrinfo(result);
return 0;
}
注意事项
字节序转换
网络传输使用大端字节序(网络字节序),主机可能使用小端字节序。必须使用转换函数:
// 主机序转网络序
uint16_t net_port = htons(8080); // 端口
uint32_t net_addr = htonl(0x7F000001); // IP地址
// 网络序转主机序
uint16_t host_port = ntohs(net_port);
uint32_t host_addr = ntohl(net_addr);
处理部分读写
TCP 是字节流协议,read 和 write 可能只处理部分数据:
// 安全读取指定字节数
ssize_t read_full(int fd, void *buf, size_t count) {
size_t total = 0;
while (total < count) {
ssize_t n = read(fd, (char *)buf + total, count - total);
if (n <= 0) return n == 0 ? total : -1;
total += n;
}
return total;
}
// 安全写入指定字节数
ssize_t write_full(int fd, const void *buf, size_t count) {
size_t total = 0;
while (total < count) {
ssize_t n = write(fd, (const char *)buf + total, count - total);
if (n <= 0) return -1;
total += n;
}
return total;
}
TIME_WAIT 状态
TCP 连接关闭后,主动关闭方会进入 TIME_WAIT 状态,持续约2分钟。在此期间端口无法复用。使用 SO_REUSEADDR 选项可以解决:
int opt = 1;
setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
进阶用法
I/O 多路复用(select)
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <netinet/in.h>
#include <sys/select.h>
#define MAX_CLIENTS 64
int main(void) {
int server_fd = socket(AF_INET, SOCK_STREAM, 0);
int opt = 1;
setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
struct sockaddr_in addr = {
.sin_family = AF_INET,
.sin_port = htons(8080),
.sin_addr.s_addr = INADDR_ANY
};
bind(server_fd, (struct sockaddr *)&addr, sizeof(addr));
listen(server_fd, 10);
int clients[MAX_CLIENTS] = {0};
fd_set readfds;
printf("select 服务器运行在端口 8080\n");
while (1) {
FD_ZERO(&readfds);
FD_SET(server_fd, &readfds);
int max_fd = server_fd;
for (int i = 0; i < MAX_CLIENTS; i++) {
if (clients[i] > 0) {
FD_SET(clients[i], &readfds);
if (clients[i] > max_fd) max_fd = clients[i];
}
}
if (select(max_fd + 1, &readfds, NULL, NULL, NULL) < 0) continue;
// 新连接
if (FD_ISSET(server_fd, &readfds)) {
int client_fd = accept(server_fd, NULL, NULL);
for (int i = 0; i < MAX_CLIENTS; i++) {
if (clients[i] == 0) { clients[i] = client_fd; break; }
}
}
// 客户端数据
for (int i = 0; i < MAX_CLIENTS; i++) {
if (clients[i] > 0 && FD_ISSET(clients[i], &readfds)) {
char buf[1024];
ssize_t n = read(clients[i], buf, sizeof(buf));
if (n <= 0) {
close(clients[i]);
clients[i] = 0;
} else {
write(clients[i], buf, n); // 回显
}
}
}
}
return 0;
}
I/O 多路复用(epoll)
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <netinet/in.h>
#include <sys/epoll.h>
#define MAX_EVENTS 64
int main(void) {
int server_fd = socket(AF_INET, SOCK_STREAM, 0);
int opt = 1;
setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
struct sockaddr_in addr = {
.sin_family = AF_INET,
.sin_port = htons(8080),
.sin_addr.s_addr = INADDR_ANY
};
bind(server_fd, (struct sockaddr *)&addr, sizeof(addr));
listen(server_fd, 10);
// 创建 epoll 实例
int epfd = epoll_create1(0);
// 注册服务器套接字
struct epoll_event ev = { .events = EPOLLIN, .data.fd = server_fd };
epoll_ctl(epfd, EPOLL_CTL_ADD, server_fd, &ev);
struct epoll_event events[MAX_EVENTS];
printf("epoll 服务器运行在端口 8080\n");
while (1) {
int nfds = epoll_wait(epfd, events, MAX_EVENTS, -1);
for (int i = 0; i < nfds; i++) {
if (events[i].data.fd == server_fd) {
// 新连接
int client_fd = accept(server_fd, NULL, NULL);
ev.events = EPOLLIN;
ev.data.fd = client_fd;
epoll_ctl(epfd, EPOLL_CTL_ADD, client_fd, &ev);
} else {
// 客户端数据
char buf[1024];
ssize_t n = read(events[i].data.fd, buf, sizeof(buf));
if (n <= 0) {
epoll_ctl(epfd, EPOLL_CTL_DEL, events[i].data.fd, NULL);
close(events[i].data.fd);
} else {
write(events[i].data.fd, buf, n);
}
}
}
}
return 0;
}
创建 Socket
基本写法:创建 TCP socket
socket(AF_INET, SOCK_STREAM, 0);
// 创建 IPv4 TCP 套接字
int fd = socket(AF_INET, SOCK_STREAM, 0);
基本写法:创建 UDP socket
socket(AF_INET, SOCK_DGRAM, 0);
// 创建 IPv4 UDP 套接字
int fd = socket(AF_INET, SOCK_DGRAM, 0);
基本写法:创建本地 socket
socket(AF_UNIX, SOCK_STREAM, 0);
// 创建 Unix 域套接字
int fd = socket(AF_UNIX, SOCK_STREAM, 0);
地址结构
基本写法:IPv4 地址结构
struct sockaddr_in <变量>;
// 初始化服务器地址
struct sockaddr_in addr;
addr.sin_family = AF_INET;
addr.sin_port = htons(8080);
addr.sin_addr.s_addr = INADDR_ANY;
基本写法:字符串转地址
inet_pton(AF_INET, <IP串>, &<地址>);
// 将点分十进制转为二进制
inet_pton(AF_INET, "127.0.0.1", &addr.sin_addr);
基本写法:地址转字符串
inet_ntop(AF_INET, &<地址>, <缓冲>, <大小>);
// 二进制地址转可读字符串
char ip[INET_ADDRSTRLEN];
inet_ntop(AF_INET, &addr.sin_addr, ip, sizeof(ip));
服务端流程
基本写法:绑定地址
bind(<fd>, (struct sockaddr*)&<地址>, sizeof(<地址>));
// 绑定本地地址端口
bind(fd, (struct sockaddr*)&addr, sizeof(addr));
基本写法:监听连接
listen(<fd>, <队列长度>);
// 开始监听客户端连接
listen(fd, 5);
基本写法:接受连接
accept(<fd>, (struct sockaddr*)&<客户端地址>, &<长度>);
// 接受新连接返回新描述符
struct sockaddr_in cli;
socklen_t len = sizeof(cli);
int cfd = accept(fd, (struct sockaddr*)&cli, &len);
客户端流程
基本写法:连接服务器
connect(<fd>, (struct sockaddr*)&<服务器地址>, sizeof(<地址>));
// 主动连接服务器
connect(fd, (struct sockaddr*)&srv, sizeof(srv));
数据收发
基本写法:发送数据
send(<fd>, <数据>, <大小>, 0);
// TCP 发送数据
send(fd, buf, n, 0);
基本写法:接收数据
recv(<fd>, <缓冲>, <大小>, 0);
// TCP 接收数据
ssize_t n = recv(fd, buf, sizeof(buf), 0);
基本写法:UDP 发送
sendto(<fd>, <数据>, <大小>, 0, (struct sockaddr*)&<目标>, sizeof(<目标>));
// UDP 发送数据到指定地址
sendto(fd, buf, n, 0, (struct sockaddr*)&dst, sizeof(dst));
基本写法:UDP 接收
recvfrom(<fd>, <缓冲>, <大小>, 0, (struct sockaddr*)&<来源>, &<长度>);
// UDP 接收数据并获取来源
struct sockaddr_in src;
socklen_t len = sizeof(src);
recvfrom(fd, buf, sizeof(buf), 0, (struct sockaddr*)&src, &len);
Socket 选项
基本写法:设置地址复用
setsockopt(<fd>, SOL_SOCKET, SO_REUSEADDR, &<值>, sizeof(<值>));
// 避免地址占用错误
int opt = 1;
setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
基本写法:设置接收超时
setsockopt(<fd>, SOL_SOCKET, SO_RCVTIMEO, &<时长>, sizeof(<时长>));
// 设置接收超时
struct timeval tv = {5, 0};
setsockopt(fd, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));
I/O 多路复用
基本写法:select 等待
select(<最大fd+1>, &<读集>, NULL, NULL, &<超时>);
// 监视多个描述符
fd_set rfds;
FD_ZERO(&rfds);
FD_SET(fd, &rfds);
struct timeval tv = {5, 0};
select(fd + 1, &rfds, NULL, NULL, &tv);
基本写法:poll 等待
poll(<数组>, <数量>, <超时毫秒>);
// 使用 poll 监视
struct pollfd fds[1];
fds[0].fd = fd;
fds[0].events = POLLIN;
poll(fds, 1, 5000);
基本写法:epoll 创建
epoll_create1(0);
// Linux 高效多路复用
int epfd = epoll_create1(0);
基本写法:epoll 注册
epoll_ctl(<epfd>, EPOLL_CTL_ADD, <fd>, &<事件>);
// 添加描述符到 epoll
struct epoll_event ev;
ev.events = EPOLLIN;
ev.data.fd = fd;
epoll_ctl(epfd, EPOLL_CTL_ADD, fd, &ev);
基本写法:epoll 等待
epoll_wait(<epfd>, <事件数组>, <最大数>, <超时>);
// 等待事件发生
struct epoll_event events[10];
int n = epoll_wait(epfd, events, 10, -1);
关闭 Socket
基本写法:关闭描述符
close(<fd>);
// 关闭并释放资源
close(fd);
基本写法:优雅关闭
shutdown(<fd>, SHUT_WR);
// 单向关闭写端
shutdown(fd, SHUT_WR);
主机与服务查询
基本写法:获取主机信息
getaddrinfo(<主机>, <服务>, &<提示>, &<结果>);
// 现代地址查询接口
struct addrinfo hints = {0};
hints.ai_family = AF_INET;
hints.ai_socktype = SOCK_STREAM;
struct addrinfo* res;
getaddrinfo("example.com", "80", &hints, &res);
基本写法:释放结果
freeaddrinfo(<结果>);
// 释放 getaddrinfo 结果
freeaddrinfo(res);