Linux程序设计之IO多路复用(API:epoll)

1.select和poll机制经常需要把用户区的文件描述符数据复制到内核中，而epoll则是直接将文件描述符信息存放在内核区中，当文件描述符信息发生变化时（有读写操作或者进程连接等操作出现），通知进程有文件描述符发生变化，不需要频繁的在用户和内核区之间切换，可以节约系统的资源，提高系统的响应速度。

2.epoll流程（图来源：登录—专业IT笔试面试备考平台_牛客网）：

3.代码：

/*IO多路复用之epoll*/
//利用IO多路复用实现一个服务器可被多个客户端连接访问

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

int main()
{
    //1.创建用于监听的socket
    int lfd = socket(AF_INET,SOCK_STREAM,0);
    struct sockaddr_in saddr;
    saddr.sin_family = AF_INET;
    saddr.sin_addr.s_addr = INADDR_ANY;
    saddr.sin_port = htons(9999);

    //2.绑定
    bind(lfd,(struct sockaddr*)&saddr,sizeof(saddr));

    //3.监听
    listen(lfd,128);

    //调用epoll_create()创建一个epoll实列
    int epfd = epoll_create(100);

    //将监听的文件描述符相关的检测信息添加到epoll实列中
    struct epoll_event epev;
    epev.events = EPOLLIN;
    epev.data.fd = lfd;
    epoll_ctl(epfd,EPOLL_CTL_ADD,lfd,&epev);    

    struct epoll_event epevs[1024];
    while(1)
    {
        //等待(检测)在epoll实列中是否有文件描述符的信息发生了变化
        int ret = epoll_wait(epfd,epevs,1024,-1);
        if(ret == -1)
        {
            perror("epoll_wait");
            exit(-1);
        }
        printf("ret = %d\n",ret);

        for(int i = 0; i < ret ;i++)
        {
            int curfd = epevs[i].data.fd;
            //检测到是监听的文件描述符数据到达，即有客户端连接进来了
            if(curfd == lfd)
            {
                    struct sockaddr_in cliaddr;
                    int len = sizeof(cliaddr);
                    int cfd = accept(lfd,(struct sockaddr *)&cliaddr,&len);

                    epev.events = EPOLLIN;
                    epev.data.fd = cfd;
                    epoll_ctl(epfd,EPOLL_CTL_ADD,cfd,&epev);
            }
            else    //有数据到达，需要通信
            {
                char buf[1024] = {0};
                int len = read(curfd,buf,sizeof(buf));
                if(len == -1)
                {
                    perror("read");
                    exit(-1);
                }
                else if(len == 0)
                {
                    printf("client closed...\n");
                    epoll_ctl(epfd,EPOLL_CTL_DEL,curfd,NULL);
                    close(curfd);
                }
                else if(len > 0)
                {
                    printf("read buf = %s\n",buf);
                    write(curfd,buf,strlen(buf) + 1);
                }
            }
        }

    }
    
    close(lfd);
    close(epfd);

    return 0;
}

/*TCP/IP通信之客户端*/
#include <stdio.h>
#include <stdlib.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>

int main()
{
    //1.创建socket套接字
    int fd = socket(AF_INET,SOCK_STREAM,0);
    if(fd == -1)
    {
        perror("socket");
        exit(-1);
    }

    //2.与服务器进行连接
    struct sockaddr_in serveraddr;
    serveraddr.sin_family = AF_INET;
    //主机字节序转为网络字节序
    inet_pton(AF_INET,"127.0.0.1",&serveraddr.sin_addr.s_addr);
    serveraddr.sin_port = htons(9999);  //客户端去访问服务端的9999端口，服务器端口是固定的，而客户端端口是随机的
    int ret = connect(fd,(struct sockaddr *)&serveraddr,sizeof(serveraddr));
    if(ret == -1)
    {
        perror("connect");
        exit(-1);
    }
    
    char buf[1024];
    int i = 0;
    while(1)
    {
        //3.通信
        //给服务器端发送数据
        // char buf[] = "hello,i am client";
        sprintf(buf,"data : %d\n",i++);
        // printf("test");
        // memset(buf,0,strlen(buf));
        // scanf("%s",buf); //scanf("%s")无法输入空格，只能获取空格和\n之前的数据
        // gets(buf);  //会产生警告
        ret = write(fd,buf,strlen(buf)+1);
        if(ret == -1)
        {
            perror("write");
            exit(-1);
        }

        //获取服务器端的数据
        char recebuf[1024] = {0};
        int len = read(fd,recebuf,sizeof(recebuf));
        if(len == -1)
        {
            perror("read");
            exit(-1);
        }
        else if(len > 0)
        {
            printf("receive server data : %s\n",recebuf);
        }
        else if(len == 0)
        {
            //服务器端断开连接
            printf("server close!!!\n");
            break;
        }
        sleep(1);
    }

    //4.关闭文件描述符
    close(fd);

    return 0;

}

4.运行结果：

 5.总结：epoll不仅会告诉进程有几个文件描述符的信息发生了变化，也会告诉进程是哪几个文件描述符的信息发生了变化。同时，将文件描述符的信息保存在内核中，避免了频繁在用户和内核之间切换，能够提高程序的效率。