epoll通信的C语言demo

服务端：

cpp 复制代码

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <errno.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/epoll.h>

#define PORT 8080
#define MAX_EVENTS 10
#define BUFFER_SIZE 1024

// 设置非阻塞套接字
int set_non_blocking(int sockfd) {
    int flags = fcntl(sockfd, F_GETFL, 0);
    if (flags == -1) {
        perror("fcntl get");
        return -1;
    }
    
    if (fcntl(sockfd, F_SETFL, flags | O_NONBLOCK) == -1) {
        perror("fcntl set");
        return -1;
    }
    return 0;
}

int main() {
    int listen_sock, conn_sock, epoll_fd;
    struct sockaddr_in server_addr, client_addr;
    socklen_t addrlen = sizeof(client_addr);
    struct epoll_event ev, events[MAX_EVENTS];
    char buffer[BUFFER_SIZE];

    // 创建监听套接字
    if ((listen_sock = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
        perror("socket");
        exit(EXIT_FAILURE);
    }

    // 设置地址结构
    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(listen_sock, (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1) {
        perror("bind");
        close(listen_sock);
        exit(EXIT_FAILURE);
    }

    // 开始监听
    if (listen(listen_sock, SOMAXCONN) == -1) {
        perror("listen");
        close(listen_sock);
        exit(EXIT_FAILURE);
    }

    // 设置监听套接字为非阻塞
    set_non_blocking(listen_sock);

    // 创建 epoll 实例
    epoll_fd = epoll_create1(0);
    if (epoll_fd == -1) {
        perror("epoll_create1");
        close(listen_sock);
        exit(EXIT_FAILURE);
    }

    // 将监听套接字添加到 epoll 中
    ev.events = EPOLLIN;
    ev.data.fd = listen_sock;
    if (epoll_ctl(epoll_fd, EPOLL_CTL_ADD, listen_sock, &ev) == -1) {
        perror("epoll_ctl: listen_sock");
        close(epoll_fd);
        close(listen_sock);
        exit(EXIT_FAILURE);
    }

    printf("服务器正在运行，等待连接...\n");

    // 主循环
    while (1) {
        int nfds = epoll_wait(epoll_fd, events, MAX_EVENTS, -1);
        if (nfds == -1) {
            perror("epoll_wait");
            break;
        }

        for (int n = 0; n < nfds; n++) {
            if (events[n].data.fd == listen_sock) {
                // 有新的连接
                conn_sock = accept(listen_sock, (struct sockaddr *)&client_addr, &addrlen);
                if (conn_sock == -1) {
                    perror("accept");
                    continue;
                }

                printf("新连接：%d\n", conn_sock);
                set_non_blocking(conn_sock);

                ev.events = EPOLLIN;
                ev.data.fd = conn_sock;
                if (epoll_ctl(epoll_fd, EPOLL_CTL_ADD, conn_sock, &ev) == -1) {
                    perror("epoll_ctl: conn_sock");
                    close(conn_sock);
                }

            } else {
                // 有数据可读
                conn_sock = events[n].data.fd;
                ssize_t count = read(conn_sock, buffer, BUFFER_SIZE);
                if (count == -1) {
                    perror("read");
                    close(conn_sock);
                } else if (count == 0) {
                    // 客户端关闭连接
                    printf("关闭连接：%d\n", conn_sock);
                    close(conn_sock);
                } else {
                    // 回显数据
                    write(conn_sock, buffer, count);
                }
            }
        }
    }

    close(epoll_fd);
    close(listen_sock);
    return 0;
}

客户端

cpp 复制代码

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <errno.h>

#define SERVER_ADDR "127.0.0.1" // 服务器地址
#define PORT 8080                // 服务器端口
#define BUFFER_SIZE 1024         // 缓冲区大小

int main() {
    int sockfd;
    struct sockaddr_in server_addr;
    char buffer[BUFFER_SIZE];

    // 创建套接字
    if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
        perror("socket");
        exit(EXIT_FAILURE);
    }

    // 设置服务器地址结构
    memset(&server_addr, 0, sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(PORT);

    // 将点分十进制的字符串 IP 地址转换为网络字节顺序的地址
    if (inet_pton(AF_INET, SERVER_ADDR, &server_addr.sin_addr) <= 0) {
        perror("inet_pton");
        close(sockfd);
        exit(EXIT_FAILURE);
    }

    // 连接到服务器
    if (connect(sockfd, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0) {
        perror("connect");
        close(sockfd);
        exit(EXIT_FAILURE);
    }

    printf("连接到服务器 %s:%d\n", SERVER_ADDR, PORT);

    while (1) {
        // 从标准输入读取数据
        printf("请输入要发送的消息（输入 'exit' 退出）：");
        fgets(buffer, BUFFER_SIZE, stdin);

        // 去除末尾换行符
        buffer[strcspn(buffer, "\n")] = 0;

        // 如果输入 "exit"，则退出循环
        if (strcmp(buffer, "exit") == 0) {
            break;
        }

        // 发送消息到服务器
        send(sockfd, buffer, strlen(buffer), 0);

        // 接收服务器的回复
        ssize_t count = recv(sockfd, buffer, BUFFER_SIZE, 0);
        if (count <= 0) {
            if (count == 0) {
                printf("服务器关闭连接\n");
            } else {
                perror("recv");
            }
            break;
        }

        // 打印服务器返回的消息
        buffer[count] = '\0'; // 确保以 NULL 结尾
        printf("从服务器收到消息: %s\n", buffer);
    }

    // 关闭套接字
    close(sockfd);
    return 0;
}

编译运行即可