1 #define SERV_PORT 4466   //服务端口号
  2 const char *LOCAL_ADDR = "127.0.0.1";//绑定服务地址
  3
  4 bool setnonblocking(int sock)//设置socket为非阻塞方式
  5 {
  6     int opts;
  7
  8     opts=fcntl(sock,F_GETFL);
  9     if(opts<0)
 10     {
 11         perror("fcntl(sock,GETFL)");
 12         return false;
 13     }
 14     opts = opts|O_NONBLOCK;
 15     if(fcntl(sock,F_SETFL,opts)<0)
 16     {
 17         perror("fcntl(sock,SETFL,opts)");
 18         return false;
 19     }
 20     return true;
 21 }
 22
 23 int main()
 24 {
 25 　　int i, maxi, listenfd, new_fd, sockfd,epfd,nfds;
 26 　　ssize_t n;
 27 　　char line[MAXLINE];
 28 　　socklen_t clilen;
 29 　　struct epoll_event ev,events[20];//ev用于注册事件,数组用于回传要处理的事件
 30 　　struct sockaddr_in clientaddr, serveraddr;
 31
 32 　　listenfd = socket(AF_INET, SOCK_STREAM, 0);//生成socket文件描述符
 33 　　setnonblocking(listenfd);//把socket设置为非阻塞方式
 34 　　　
 35 　　epfd=epoll_create(256);//生成用于处理accept的epoll专用的文件描述符
 36 　　ev.data.fd=listenfd;//设置与要处理的事件相关的文件描述符
 37 　　ev.events=EPOLLIN|EPOLLET;//设置要处理的事件类型
 38 　　epoll_ctl(epfd,EPOLL_CTL_ADD,listenfd,&ev);//注册epoll事件
 39
 40        //设置服务器端地址信息
 41 　　bzero(&serveraddr, sizeof(serveraddr));
 42 　　serveraddr.sin_family = AF_INET;
 43 　　char *local_addr= LOCAL_ADDR;
 44 　　inet_aton(local_addr,&(serveraddr.sin_addr));
 45 　　serveraddr.sin_port=htons(SERV_PORT);
 46
 47 　　bind(listenfd,(sockaddr *)&serveraddr, sizeof(serveraddr));//绑定socket连接
 48 　　listen(listenfd, LISTENQ);//监听
 49
 50 　　maxi = 0;
 51 　　for ( ; ; )
 52       {
 53          /* epoll_wait：等待epoll事件的发生，并将发生的sokct fd和事件类型放入到events数组中；
 54           * nfds：为发生的事件的个数。
 55           * 注：事件发生后，注册在epfd上的socket fd的事件类型会被清空，所以如果下一个循环你
 56          * 还要关注这个socket fd的话，则需要用epoll_ctl(epfd,EPOLL_CTL_MOD,listenfd,&ev)来
 57          * 重新设置socket fd的事件类型
 58          */
 59 　　    nfds=epoll_wait(epfd,events,20,500);
 60
 61 　　    //处理所发生的所有事件
 62 　　    for(i=0;i<nfds;++i)
 63 　　    {
 64 　　        if(events[i].data.fd==listenfd)//事件发生在listenfd上
 65 　　        {
 66                     /* 获取发生事件端口信息，存于clientaddr中；
 67                    * new_fd：返回的新的socket描述符，用它来对该事件进行recv/send操作*/
 68 　　            new_fd = accept(listenfd,(struct sockaddr *)&clientaddr, &clilen);
 69 　　            if(connfd<0)
 70                    {
 71 　　                perror("connfd<0");
 72 　　                exit(1);
 73 　　            }
 74 　　            setnonblocking(connfd);
 75 　　            char *str = inet_ntoa(clientaddr.sin_addr);
 76 　　            ev.data.fd=connfd;//设置用于读操作的文件描述符
 77 　　            ev.events=EPOLLIN|EPOLLET;//设置用于注测的读操作事件
 78 　　            epoll_ctl(epfd,EPOLL_CTL_ADD,connfd,&ev);//注册ev
 79 　　        }
 80 　　        else if(events[i].events&EPOLLIN)
 81 　　        {
 82 　　            if ( (sockfd = events[i].data.fd) < 0)
 83                        continue;
 84
 85 　　            if ( (n = read(sockfd, line, MAXLINE)) < 0)
 86                    {
 87 　　                if (errno == ECONNRESET)
 88                       {
 89 　　                    close(sockfd);
 90 　　                    events[i].data.fd = -1;
 91 　　                }
 92                       else
 93 　　                    std::cout<<"readline error"<<std::endl;
 94 　　            }
 95                   else if (n == 0)
 96                   {
 97 　　                close(sockfd);
 98 　　                events[i].data.fd = -1;
 99 　　           }
100 　　           ev.data.fd=sockfd;//设置用于写操作的文件描述符
101 　　           ev.events=EPOLLOUT|EPOLLET;//设置用于注测的写操作事件
102 　　           epoll_ctl(epfd,EPOLL_CTL_MOD,sockfd,&ev);//修改sockfd上要处理的事件为EPOLLOUT
103 　　      }
104 　　     else if(events[i].events&EPOLLOUT)
105 　　     {
106 　　         sockfd = events[i].data.fd;
107 　　          write(sockfd, line, n);
108 　　
109 　　          ev.data.fd=sockfd;//设置用于读操作的文件描述符
110 　　          ev.events=EPOLLIN|EPOLLET;//设置用于注测的读操作事件
111 　　          epoll_ctl(epfd,EPOLL_CTL_MOD,sockfd,&ev);//修改sockfd上要处理的事件为EPOLIN
112 　　     }
113 　　 }
114 　}
115 }