一個epoll例子（二）

/* 使用 epoll 寫的回射服務器將從client中接收到的數據再返回給client  */  
#include <iostream> 
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>  
#include <sys/epoll.h>  
#include <netinet/in.h>  
#include <arpa/inet.h>  
#include <fcntl.h>  
#include <unistd.h>  
#include <stdio.h>  
#include <errno.h>  
  
using namespace std;  
  
#define MAXLINE 100  
#define OPEN_MAX 100  
#define LISTENQ 20  
#define SERV_PORT 5000  
#define INFTIM 1000  
  
void setnonblocking(int sock)  
{  
    int opts;  
     opts=fcntl(sock,F_GETFL);  
    if(opts<0)  
    {  
        perror("fcntl(sock,GETFL)");  
        exit(1);  
    }  
     opts = opts|O_NONBLOCK;  
    if(fcntl(sock,F_SETFL,opts)<0)  
    {  
        perror("fcntl(sock,SETFL,opts)");  
        exit(1);  
    }  
}  
  
int main(int argc, char* argv[])  
{  
    int i, maxi, listenfd, connfd, sockfd,epfd,nfds, portnumber;  
    ssize_t n;  
    char line[MAXLINE];  
    socklen_t clilen;  
    string szTemp("");  
  
    if ( 2 == argc )  
    {  
        if( (portnumber = atoi(argv[1])) < 0 )  
        {  
            fprintf(stderr,"Usage:%s portnumber\a\n",argv[0]);  
            return 1;  
        }  
    }  
    else  
    {  
        fprintf(stderr,"Usage:%s portnumber\a\n",argv[0]);  
        return 1;  
    }  
  
  
  
    //聲明epoll_event結構體的變量,ev用於註冊事件,數組用於回傳要處理的事件  
    struct epoll_event ev, events[20];  
      
    //創建一個epoll的句柄，size用來告訴內核這個監聽的數目一共有多大  
    epfd = epoll_create(256); //生成用於處理accept的epoll專用的文件描述符  
      
    struct sockaddr_in clientaddr;  
    struct sockaddr_in serveraddr;  
    listenfd = socket(AF_INET, SOCK_STREAM, 0);  
      
    //把socket設置爲非阻塞方式  
    setnonblocking(listenfd);  
  
    //設置與要處理的事件相關的文件描述符  
    ev.data.fd=listenfd;  
      
    //設置要處理的事件類型  
    ev.events=EPOLLIN|EPOLLET;  
  
    //註冊epoll事件  
    epoll_ctl(epfd,EPOLL_CTL_ADD,listenfd,&ev);  
      
    bzero(&serveraddr, sizeof(serveraddr)); /*配置Server socket的相關信息 */  
    serveraddr.sin_family = AF_INET;  
    const char *local_addr="127.0.0.1";  
    inet_aton(local_addr,&(serveraddr.sin_addr));//htons(portnumber);  
    serveraddr.sin_port=htons(portnumber);  
    bind(listenfd,(sockaddr *)&serveraddr, sizeof(serveraddr));  
      
    listen(listenfd, LISTENQ);  
      
    maxi = 0;  
      
    for ( ; ; ) {  
          
        //等待epoll事件的發生  
        //返回需要處理的事件數目nfds，如返回0表示已超時。  
        nfds=epoll_wait(epfd,events,maxi,500);  
          
        //處理所發生的所有事件  
        for(i=0; i < nfds; ++i)  
        {  
            //如果新監測到一個SOCKET用戶連接到了綁定的SOCKET端口，建立新的連接。  
            if(events[i].data.fd == listenfd)  
            {  
                connfd = accept(listenfd,(sockaddr *)&clientaddr, &clilen);  
                if(connfd < 0)  
                {  
                    perror("connfd < 0");  
                    exit(1);  
                }  
                setnonblocking(connfd);  
                char *str = inet_ntoa(clientaddr.sin_addr);  
                cout << "accapt a connection from " << str << endl;  
                  
                //設置用於讀操作的文件描述符  
                ev.data.fd=connfd;  
                  
                //設置用於註冊的讀操作事件  
                ev.events=EPOLLIN|EPOLLET;  
  
                //註冊ev  
                epoll_ctl(epfd,EPOLL_CTL_ADD,connfd,&ev); /* 添加 */  
                
            }  
            //如果是已經連接的用戶，並且收到數據，那麼進行讀入。  
            else if(events[i].events&EPOLLIN)  
            {  
                cout << "EPOLLIN" << endl;  
                if ( (sockfd = events[i].data.fd) < 0)  
                    continue;  
                if ( (n = recv(sockfd, line, sizeof(line), 0)) < 0)   
                {    
                    // Connection Reset:你連接的那一端已經斷開了，而你卻還試着在對方已斷開的socketfd上讀寫數據！  
                    if (errno == ECONNRESET)  
                    {  
                        close(sockfd);  
                        events[i].data.fd = -1;  
                    }   
                    else  
                        std::cout<<"readline error"<<std::endl;  
                }   
                else if (n == 0) //讀入的數據爲空  
                {  
                    close(sockfd);  
                    events[i].data.fd = -1;  
                }  
                  
                szTemp = "";  
                szTemp += line;  
                szTemp = szTemp.substr(0,szTemp.find('\r')); /* remove the enter key */  
                memset(line,0,100); /* clear the buffer */  
                //line[n] = '\0';  
                cout << "Readin: " << szTemp << endl;  
                 
                //設置用於寫操作的文件描述符  
                ev.data.fd=sockfd;  
                  
                //設置用於註冊的寫操作事件  
                ev.events=EPOLLOUT|EPOLLET;  
                  
                //修改sockfd上要處理的事件爲EPOLLOUT  
                epoll_ctl(epfd,EPOLL_CTL_MOD,sockfd,&ev); /* 修改 */  
  
            }  
            else if(events[i].events&EPOLLOUT) // 如果有數據發送 
            {  
                sockfd = events[i].data.fd;  
                szTemp = "Server:" + szTemp + "\n";  
                send(sockfd, szTemp.c_str(), szTemp.size(), 0);  
                  
                  
                //設置用於讀操作的文件描述符  
                ev.data.fd=sockfd;  
                 
                //設置用於註冊的讀操作事件  
                ev.events=EPOLLIN|EPOLLET;  
                  
                //修改sockfd上要處理的事件爲EPOLIN  
                epoll_ctl(epfd,EPOLL_CTL_MOD,sockfd,&ev); /* 修改 */  
            }  
        } //(over)處理所發生的所有事件  
    } //(over)等待epoll事件的發生  
      
    close(epfd);  
    return 0;  
}