【网络编程】学习笔记06 I/O多路复用之epoll

epoll是Linux下多路复用IO接口select/poll的增强版本，它能显著提高程序在大量并发连接中只有少量活跃的情况下的系统CPU利用率，因为它会复用文件描述符集合来传递结果而不用迫使开发者每次等待事件之前都必须重新准备要被监听的文件描述符集合，另一个原因就是获取事件的时候，它无需遍历整个被监听的描述符集，只要遍历那些被IO事件唤醒而加入ready队列的描述符即可。

 

epoll的server端：

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

#define SERV_PORT 9527
#define MAXLINE 8192
#define OPEN_MAX 5000

int main(int argc,char *argv[])
{
    int listenfd = 0,connfd = 0,sockfd,i;
    int n,num = 0;
    ssize_t nready,efd,res;
    char buf[MAXLINE], str[INET_ADDRSTRLEN];
    socklen_t clilen;

    struct sockaddr_in cliaddr,servaddr;

    listenfd = socket(AF_INET,SOCK_STREAM,0);
    int opt = 1;
    setsockopt(listenfd,SOL_SOCKET,SO_REUSEADDR,&opt,sizeof(opt));  //端口复用
    memset(&servaddr,0,sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servaddr.sin_port = htons(SERV_PORT);
    bind(listenfd,(struct sockaddr *)&servaddr, sizeof(servaddr));
    listen(listenfd,20);

    efd = epoll_create(OPEN_MAX);    //创建epoll模型，efd指向红黑树根节点

    struct epoll_event tep, ep[OPEN_MAX];       //tep:epoll_ctl参数，ep[]:epoll_wait参数;
    tep.events = EPOLLIN;
    tep.data.fd = listenfd;                     //指定listenfd的监听事件为“读”

    res = epoll_ctl(efd, EPOLL_CTL_ADD, listenfd, &tep); //将listenfd及对应的结构体设置到树上，efd可找到该树

    for(;;){
        //epoll为server阻塞监听事件，ep为struct epoll_event类型数组，OPEN_MAX为数组容量，-1表示永久阻塞
        nready = epoll_wait(efd,ep,OPEN_MAX,-1);

        for(i=0;i<nready;i++){
            if(!(ep[i].events & EPOLLIN))
                continue;
            if(ep[i].data.fd == listenfd){      //判断满足事件的fd是不是lfd
                clilen = sizeof(cliaddr);
                connfd = accept(listenfd,(struct sockaddr *)&cliaddr,&clilen); //接受连接

                tep.events = EPOLLIN; 
                tep.data.fd = connfd;

                res = epoll_ctl(efd,EPOLL_CTL_ADD,connfd,&tep);  //加入红黑树
            }else{                      //不是lfd
                sockfd = ep[i].data.fd;
                n = read(sockfd,buf,MAXLINE);

                if(n == 0){                                            //读到0，说明客户端关闭连接
                    res = epoll_ctl(efd, EPOLL_CTL_DEL, sockfd, NULL); //将该文件描述符从红黑树摘除
                    close(sockfd);                                     //关闭与该客户端的连接
                }
                else if(n < 0){        //出错
                    res = epoll_ctl(efd, EPOLL_CTL_DEL, sockfd, NULL); //摘除节点
                    close(sockfd);
                }
                else{                  //读到了字节数
                    for(i=0;i<n;i++)
                       buf[i] = toupper(buf[i]);

                    write(STDOUT_FILENO,buf,n);
                    write(sockfd,buf,n);
                }
            }
        }
    }
    close(listenfd);
    return 0;
}