一个简单的epoll使用例子

核心代码,所有代码在这里下载

#include "unixnet.h"
#include "chat.h"
#include "sys/epoll.h"

/**
 * 改进说明:使用epoll进行事件回调,多用户可以处于输入命令状态(不再阻塞在登陆处)
 */

int main (int argc, char *argv[])
{
    int listen_fd;
    socklen_t cli_len;
    struct sockaddr_in cli_addr,serv_addr;
    int ret,flags;
    int re_use_addr=1;
    char recv_buf[MAXLINE];
    int i;
    //epoll 描述符
    int efd;
    struct epoll_event event;
    struct epoll_event events[MAXUSERS];
    //初始化槽位
    for(i=0; i<MAXUSERS; i++)
    {
        chater[i].slot_status =SLOT_FREED ;
        chater[i].sock_fd =-1;
        chater[i].cmd .cmd_type =-1;
    }
    /*AF_INET指定ipv4,SOCK_STREAM制定流模式,0为tcp*/
    listen_fd = socket(AF_INET,SOCK_STREAM,0);
    if(listen_fd==-1)
    {
        perror("create listen fd");
        exit(1);
    }
    //服务器地址清零
    bzero(&serv_addr ,sizeof(serv_addr));
    serv_addr.sin_family=AF_INET;
    //接受所有ip的请求
    serv_addr.sin_addr.s_addr=htonl(INADDR_ANY);
    //设置端口
    serv_addr.sin_port=htons(SERVER_PORT );
    setsockopt(listen_fd, SOL_SOCKET,SO_REUSEADDR,(void
    *)&re_use_addr,sizeof(int));
    //绑定描述符到端口
    ret=bind(listen_fd,(struct sockaddr
    *)&serv_addr,sizeof(serv_addr));
    if(ret<0)
    {
        perror("bind server port");
        exit(1);
    }
    //开始监听端口请求,并设置最大的请求队列
    listen(listen_fd,LISTENQ);
    //设置为非阻塞io
    if((flags=fcntl(listen_fd,F_GETFL,flags)<0))
    {
        perror("F_SETFL error");
    }
    //使用位域
    flags|=O_NONBLOCK;
    if(fcntl(listen_fd,F_SETFL,flags)<0)
    {
        perror("F_SETFL error");
    }
    //创建一个epoll句柄
    efd = epoll_create(MAXUSERS);
    //对连接套接字设置监听事件,并设置触发模式,最后加入到epoll中进行回调
    event.data.fd = listen_fd;
    event.events = EPOLLET| EPOLLIN;
    epoll_ctl(efd, EPOLL_CTL_ADD, listen_fd, &event);
    printf("listen_fd %d\n",listen_fd);
    //循环处理所有事件
    while(1){
        int n,i;
        //等待直到事件到来,规律性sleep,所有不会导致cpu 100%
        n = epoll_wait(efd, events, MAXUSERS,-1);
        for(i = 0;i < n;++i){
            //如果是一个异常事件,则释放资源
            if ((events[i].events & EPOLLERR) ||
                (events[i].events & EPOLLHUP) ||
                (!(events[i].events & EPOLLIN))) {
                fprintf (stderr, "epoll error\n");
                free_slot(findSlotIndexByConnFd(events[i].data.fd));
                epoll_ctl(efd,EPOLL_CTL_DEL,events[i].data.fd,NULL);
                continue;
                //此时代表获得了一个连接建立事件,可以创建一个连接,本质上也是一个EPOLLIN
            }else if(listen_fd == events[i].data.fd){
                int infd = -1;
                //accept接受连接直到失败(由于使用ET触发所以需要使用循环以保所有事件都得到处理)
                while((infd = accept(listen_fd,(struct sockaddr *)&cli_addr,&cli_len)) > 0){
                    //获得一个空槽,并将连接设置为非阻塞
                    get_free_slot(infd, "no_login");
                    event.data.fd = infd;
                    event.events = EPOLLIN | EPOLLET;
                    epoll_ctl(efd, EPOLL_CTL_ADD, infd, &event);
                    writen(infd,message[2],strlen(message [2]));
                    printf("accept one\n");
                }
                if(infd == -1){
                    printf("%d\n",errno);
                }
                //在这里处理退出事件|用户名登陆事件|正常的cmd命令
            }else{
                int infd = events[i].data.fd;
                int index = -1;
                char *ptr;
                index = findSlotIndexByConnFd(infd);
                //从内核读数据到用户缓冲区
                ptr=&(chater[index].buffer[chater[index].next_char]);
                if(index != -1){
                    while((n=read(infd ,ptr,1))==1)
                    {
                        if(*ptr=='\n')
                        {
                            *(ptr+1)='\0';
                            chater[index].next_char =0;
                            break;
                        }
                        //在这里丢弃过长的字符串
                        if(++chater[index].next_char ==MAXLINE)
                            --chater[index].next_char ;
                        else
                            ++ptr;
                    }
                    //退出事件:当n==0代表连接丢失,当n==-1且errno!=EWOULDBLOCK代表不是因为暂时没有数据抛出错误
                    if(n == 0 || (n == -1&&errno!=EWOULDBLOCK)){
                        free_slot(findSlotIndexByConnFd(events[i].data.fd));
                        epoll_ctl(efd,EPOLL_CTL_DEL,events[i].data.fd,NULL);
                        printf("user %s use slot_index %d logout\n\n",chater[index].user_id, index);
                        continue;
                    }
                    //登陆用户名
                    if(!chater[index].user_assigned){
                        copyStringSkipSpace(chater[index].user_id,chater[index].buffer);
                        chater[index].user_assigned = 1;
                        printf("user %s use slot_index %d login\n\n",chater[index].user_id, index);
                        //处理cmd.
                    }else{
                        handle_cmd(index);
                    }
                }
            }
        }
    }
}