Linux TCP Socket编程入门与回声服务器实现-代码聚汇网

Linux TCP Socket编程入门与回声服务器实现

用户甲

1. Linux TCP Socket 编程基础

TCP Socket 编程是 Linux 网络编程中最基础也是最重要的部分。作为一名长期从事网络开发的工程师，我经常需要处理各种网络通信问题。今天我想分享一个完整的 TCP 回声服务器实现过程，从单客户端到多客户端的演进，希望能帮助刚入门的开发者少走弯路。

1.1 TCP 协议特性

TCP（传输控制协议）是一种面向连接的、可靠的、基于字节流的传输层通信协议。它有三个核心特性：

面向连接：通信双方必须先建立连接才能传输数据
可靠传输：通过确认机制、重传机制等保证数据不丢失、不重复、按序到达
全双工通信：连接建立后，双方可以同时发送和接收数据

在实际开发中，我们最常用的就是 TCP 协议，比如 HTTP、FTP、SSH 等应用层协议都是基于 TCP 实现的。

2. TCP Socket 通信核心流程

2.1 服务端实现步骤

服务端作为被动接收连接的一方，实现流程相对复杂，主要分为 6 个步骤：

socket() - 创建通信端点
bind() - 绑定地址和端口
listen() - 开始监听连接请求
accept() - 接受客户端连接
read()/write() - 数据交互
close() - 关闭连接

让我们用一个生活中的例子来理解这个过程：想象服务端是一家餐厅，socket() 相当于租下店面，bind() 是挂上招牌并确定地址，listen() 是开门营业，accept() 是迎接客人入座，read()/write() 是为客人提供服务，close() 则是送客关门。

2.2 客户端实现步骤

客户端作为主动发起连接的一方，流程相对简单：

socket() - 创建通信端点
connect() - 连接服务端
read()/write() - 数据交互
close() - 关闭连接

继续餐厅的比喻，客户端就像顾客：socket() 是准备去吃饭，connect() 是走进餐厅，read()/write() 是点餐和用餐，close() 是结账离开。

3. 核心函数详解与实现

3.1 socket() 函数

socket() 函数用于创建一个通信端点，返回一个文件描述符。在 Linux 中，一切皆文件，套接字也不例外。

c复制int socket(int domain, int type, int protocol);

参数说明：

domain：地址族，常用 AF_INET（IPv4）
type：套接字类型，SOCK_STREAM 表示 TCP
protocol：通常设为 0，表示默认协议

示例代码：

c复制int sock_fd = socket(AF_INET, SOCK_STREAM, 0);
if (sock_fd == -1) {
    perror("socket create failed");
    exit(EXIT_FAILURE);
}

3.2 bind() 函数

bind() 函数将套接字与特定的 IP 地址和端口号绑定。

c复制int bind(int sockfd, const struct sockaddr *addr, socklen_t addrlen);

关键点：

需要使用 sockaddr_in 结构体设置地址信息
端口号需要用 htons() 转换字节序
IP 地址可以用 INADDR_ANY 表示绑定所有网卡

示例代码：

c复制struct sockaddr_in server_addr;
memset(&server_addr, 0, sizeof(server_addr));
server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(8888);
server_addr.sin_addr.s_addr = INADDR_ANY;

if (bind(sock_fd, (struct sockaddr*)&server_addr, sizeof(server_addr)) == -1) {
    perror("bind failed");
    close(sock_fd);
    exit(EXIT_FAILURE);
}

3.3 listen() 函数

listen() 函数使套接字进入被动监听状态，等待客户端连接。

c复制int listen(int sockfd, int backlog);

参数 backlog 指定了等待连接队列的最大长度。在实际应用中，这个值需要根据服务器负载能力合理设置。

3.4 accept() 函数

accept() 函数从已完成连接队列中取出一个连接，返回一个新的套接字描述符。

c复制int accept(int sockfd, struct sockaddr *addr, socklen_t *addrlen);

重要特性：

这是一个阻塞调用，如果没有连接请求，进程会一直等待
返回的新套接字专门用于与这个客户端通信
原监听套接字继续接受其他连接

3.5 connect() 函数

connect() 函数用于客户端发起连接请求。

c复制int connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen);

3.6 数据读写函数

read() 和 write() 函数用于在已建立的连接上进行数据交换。

c复制ssize_t read(int fd, void *buf, size_t count);
ssize_t write(int fd, const void *buf, size_t count);

重要注意事项：

read() 返回实际读取的字节数，0 表示对端关闭连接
TCP 是字节流协议，没有消息边界，需要应用层自己处理
网络传输的数据可能需要考虑字节序问题

4. 完整回声服务器实现

4.1 服务端代码

c复制#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#define PORT 8888
#define BUFFER_SIZE 1024

int main() {
    int server_fd, client_fd;
    struct sockaddr_in server_addr, client_addr;
    socklen_t client_len = sizeof(client_addr);
    char buffer[BUFFER_SIZE];
    
    // 1. 创建套接字
    if ((server_fd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
        perror("socket creation failed");
        exit(EXIT_FAILURE);
    }
    
    // 设置 SO_REUSEADDR 选项，避免地址占用问题
    int opt = 1;
    if (setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt)) == -1) {
        perror("setsockopt failed");
        close(server_fd);
        exit(EXIT_FAILURE);
    }
    
    // 2. 绑定地址和端口
    memset(&server_addr, 0, sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = INADDR_ANY;
    server_addr.sin_port = htons(PORT);
    
    if (bind(server_fd, (struct sockaddr*)&server_addr, sizeof(server_addr)) == -1) {
        perror("bind failed");
        close(server_fd);
        exit(EXIT_FAILURE);
    }
    
    // 3. 开始监听
    if (listen(server_fd, 5) == -1) {
        perror("listen failed");
        close(server_fd);
        exit(EXIT_FAILURE);
    }
    
    printf("Server listening on port %d...\n", PORT);
    
    // 4. 接受客户端连接
    while (1) {
        if ((client_fd = accept(server_fd, (struct sockaddr*)&client_addr, &client_len)) == -1) {
            perror("accept failed");
            continue;
        }
        
        printf("New connection from %s:%d\n", 
               inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));
        
        // 5. 处理客户端请求
        while (1) {
            ssize_t bytes_read = read(client_fd, buffer, BUFFER_SIZE - 1);
            if (bytes_read <= 0) {
                if (bytes_read == 0) {
                    printf("Client disconnected\n");
                } else {
                    perror("read error");
                }
                break;
            }
            
            buffer[bytes_read] = '\0';
            printf("Received: %s", buffer);
            
            // 回显数据
            if (write(client_fd, buffer, bytes_read) == -1) {
                perror("write failed");
                break;
            }
        }
        
        close(client_fd);
    }
    
    close(server_fd);
    return 0;
}

4.2 客户端代码

c复制#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#define PORT 8888
#define BUFFER_SIZE 1024

int main(int argc, char *argv[]) {
    if (argc != 2) {
        fprintf(stderr, "Usage: %s <server_ip>\n", argv[0]);
        exit(EXIT_FAILURE);
    }
    
    int sock_fd;
    struct sockaddr_in server_addr;
    char buffer[BUFFER_SIZE];
    
    // 1. 创建套接字
    if ((sock_fd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
        perror("socket creation failed");
        exit(EXIT_FAILURE);
    }
    
    // 2. 设置服务器地址
    memset(&server_addr, 0, sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(PORT);
    
    if (inet_pton(AF_INET, argv[1], &server_addr.sin_addr) <= 0) {
        perror("invalid address");
        close(sock_fd);
        exit(EXIT_FAILURE);
    }
    
    // 3. 连接服务器
    if (connect(sock_fd, (struct sockaddr*)&server_addr, sizeof(server_addr)) == -1) {
        perror("connection failed");
        close(sock_fd);
        exit(EXIT_FAILURE);
    }
    
    printf("Connected to server at %s:%d\n", argv[1], PORT);
    printf("Enter message (Ctrl+D to quit):\n");
    
    // 4. 数据交互
    while (fgets(buffer, BUFFER_SIZE, stdin) != NULL) {
        if (write(sock_fd, buffer, strlen(buffer)) == -1) {
            perror("write failed");
            break;
        }
        
        ssize_t bytes_read = read(sock_fd, buffer, BUFFER_SIZE - 1);
        if (bytes_read <= 0) {
            if (bytes_read == 0) {
                printf("Server closed connection\n");
            } else {
                perror("read error");
            }
            break;
        }
        
        buffer[bytes_read] = '\0';
        printf("Echo: %s", buffer);
    }
    
    close(sock_fd);
    return 0;
}

5. 常见问题与解决方案

5.1 地址已在使用 (Address already in use)

这个问题通常发生在服务器程序异常退出后立即重启时。原因是 TCP 协议有一个 TIME_WAIT 状态，会保持连接一段时间（通常是 2MSL，约 1-4 分钟）。

解决方案：

设置 SO_REUSEADDR 套接字选项
修改服务器端口号
等待一段时间再重启

5.2 连接被拒绝 (Connection refused)

可能原因：

服务器未运行
防火墙阻止了连接
IP 地址或端口号错误

排查步骤：

检查服务器是否正在运行
使用 netstat -tuln 查看端口监听状态
检查防火墙设置

5.3 数据读写问题

常见问题：

数据不完整
数据乱码
连接意外断开