驱动day 8-优快云博客

本文链接：https://blog.youkuaiyun.com/weixin_45544618/article/details/131355707

该代码示例展示了一个使用epoll进行事件驱动的TCP服务器，它创建一个监听socket，绑定到8080端口，并通过epoll_wait等待事件。当有新连接时，服务器接受连接并将客户端socket添加到epoll实例中。当检测到已连接客户端有数据可读时，服务器读取并回显数据。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <sys/epoll.h>

#define MAX_EVENTS 10
#define MAX_BUFFER_SIZE 1024

int main() {
    int listen_sock, client_sock, epollfd;
    struct epoll_event event, events[MAX_EVENTS];
    char buffer[MAX_BUFFER_SIZE];

    // 创建主监听socket
    listen_sock = socket(AF_INET, SOCK_STREAM, 0);
    // 设置socket地址重用
    int reuse = 1;
    setsockopt(listen_sock, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse));

    struct sockaddr_in server_addr;
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    server_addr.sin_port = htons(8080);

    bind(listen_sock, (struct sockaddr*)&server_addr, sizeof(server_addr));
    listen(listen_sock, 10);

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

    // 添加主监听socket到epoll事件集合
    event.data.fd = listen_sock;
    event.events = EPOLLIN;
    epoll_ctl(epollfd, EPOLL_CTL_ADD, listen_sock, &event);

    while (1) {
        // 等待事件发生
        int num_ready = epoll_wait(epollfd, events, MAX_EVENTS, -1);
        if (num_ready == -1) {
            perror("epoll_wait");
            exit(EXIT_FAILURE);
        }

        // 处理每个事件
        for (int i = 0; i < num_ready; i++) {
            if (events[i].data.fd == listen_sock) {
                // 有新的连接
                struct sockaddr_in client_addr;
                socklen_t addrlen = sizeof(client_addr);
                client_sock = accept(listen_sock, (struct sockaddr*)&client_addr, &addrlen);

                // 将新的客户端socket添加到epoll事件集合
                event.data.fd = client_sock;
                event.events = EPOLLIN;
                epoll_ctl(epollfd, EPOLL_CTL_ADD, client_sock, &event);
            } else {
                // 已连接的客户端有数据可读
                int client_fd = events[i].data.fd;
                memset(buffer, 0, sizeof(buffer));
                int num_bytes = read(client_fd, buffer, sizeof(buffer));
                if (num_bytes <= 0) {
                    // 客户端断开连接
                    epoll_ctl(epollfd, EPOLL_CTL_DEL, client_fd, NULL);
                    close(client_fd);
                } else {
                    // 处理客户端发送的数据
                    // TODO: 进行相应的处理

                    // 回复客户端消息
                    write(client_fd, buffer, strlen(buffer));
                }
            }
        }
    }

    // 清理资源
    close(listen_sock);
    close(epollfd);

    return 0;
}