epoll io多路复用示例

epoll io多路复用示例

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


int main(int argc, char* argv[])
{
    if(argc < 2)
    {
        printf("Param Error\n");
        return -1;
    }

    //atoi把参数 str 所指向的字符串转换为一个整数（类型为 int 型）
    int port = atoi(argv[1]);
    
    //初始化socket套接字
    int socketfd = socket(AF_INET, SOCK_STREAM, 0);
    //初始化信息
    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_port = htons(port);
    addr.sin_addr.s_addr = INADDR_ANY;

    //绑定信息与套接字
    if(bind(socketfd, (struct sockaddr*)&addr, sizeof(struct sockaddr_in)) < 0)
    {
        perror("bind");
        return -2;
    }

    //监听请求
    if(listen(socketfd, 5) < 0)
    {
        perror("listen");
        return -3;
    }
    
    //epoll_create参数现已弃用,最早是红黑数的规模
    int epfd = epoll_create(1);
    //事件存放
    struct epoll_event events[1024] = {0};
    
    //单个事件
    struct epoll_event ev;
    //要监听的事件 EPOLLIN 表示对应的文件描述符可以读（包括对端SOCKET正常关闭）
    ev.events = EPOLLIN;
    //事件的数据是一个fd
    ev.data.fd = socketfd;

    /*
    epoll_ctl向epoll对象中添加、修改或者删除事件，返回0表示成功，否则返回–1
    epfd参数 epoll_create()的返回值
    第二个参数 表示动作 EPOLL_CTL_ADD表示注册新的fd到epfd中
    socketfd 需要监听的fd
    &ev 告诉内核需要监听什么事
    */
    epoll_ctl(epfd, EPOLL_CTL_ADD, socketfd, &ev);

    while(1)
    {
        /*
        参数epfd      epoll的描述符
        参数events    用来从内核得到事件的集合 
                      分配好的epoll_event结构体数组，epoll将会把发生的事件复制到events数组中 
                      events不可以是空指针，内核只负责把数据复制到这个events数组中，不会去帮助我们在用户态中分配内存
        参数3         告之内核这个events有多大
        参数timeout   超时时间 (毫秒，0会立即返回，-1将不确定，也有说法说是永久阻塞)
        */
        //该函数返回需要处理的事件数目，如返回0表示已超时。如果返回–1，则表示出现错误，需要检查 errno错误码判断错误类型。
        int nready = epoll_wait(epfd, events, 1024, 5);
        if(nready == -1) continue;

        int i = 0;
        for (i = 0; i < nready; i++)
        {
            if(events[i].data.fd == socketfd) //代表这个事件是listen
            {
                struct sockaddr_in clinet_addr;
                memset(&clinet_addr, 0, sizeof(struct sockaddr_in));
                socklen_t client_addr_len = sizeof(clinet_addr);

                //接收一个请求返回一个套接字clientfd
                int clientfd = accept(socketfd, (struct sockaddr*)&clinet_addr, &client_addr_len);
                //可写|边沿触发
                ev.events = EPOLLIN | EPOLLET;
                ev.data.fd = clientfd;
                epoll_ctl(epfd, EPOLL_CTL_ADD, clientfd, &ev);
            }
            else
            {
                //这个不是一个listen 可以直接处理
                int clientfd = events[i].data.fd;
                char buffer[1024] = {0};
                //接收信息
                int len = recv(clientfd, buffer, 1024, 0);
                if(len <= 0)
                {
                    close(clientfd);
                    break;
                }
                else
                {
                printf("Recv : %s %d byte(s)\n", buffer, len);
                }
            }
        }
    }
}

测试