Windows下完成端口移植Linux下的epoll（续）

#include <stdio.h> 
#include <sys/socket.h> 
#include <sys/epoll.h> 
#include <fcntl.h> 
#include <string.h> 
#include <errno.h> 
#include <netinet/in.h> 
#include <arpa/inet.h> 

#define MAXSIZE         64000 
#define MAXEPS            256 
//#define EVENTS            100 
#define LISTENQ         32 
#define SERV_PORT     8000 

int setnonblock(int sock) 
{ 
     int flags = fcntl(sock, F_GETFL, 0); 

     if(-1 == flags) { 
            perror("fcntl(sock, F_GETFL)"); 
            return -1; 
     } 

     flags |= O_NONBLOCK; 

     if(-1 == fcntl(sock, F_SETFL, flags)) { 
            perror("fcntl(sock, F_SETFT, flags)"); 
            return -2; 
     } 

     return 0; 
} 

int main(int argc, char *argv[]) 
{ 
     int i, maxi, listenfd, connfd, sockfd, epfd, nfds; 
     ssize_t n; 
    
     char buf[MAXSIZE];        

     socklen_t clilen; 
        
     struct epoll_event ev, events[20]; 

     epfd = epoll_create(MAXEPS); 

     struct sockaddr_in clientaddr; 
     struct sockaddr_in serveraddr; 

     listenfd = socket(AF_INET, SOCK_STREAM, 0); 

     setnonblock(listenfd); 

     ev.data.fd = listenfd; 

     ev.events = EPOLLIN | EPOLLET; 

     epoll_ctl(epfd, EPOLL_CTL_ADD, listenfd, &ev); 
     
     memset(&serveraddr, 0, sizeof(serveraddr)); 
     serveraddr.sin_family = AF_INET; 

/* 
     char *local_addr = "10.0.2.15"; 
     inet_aton(local_addr, &(serveraddr.sin_addr)); 
*/ 
     serveraddr.sin_addr.s_addr = htonl(INADDR_ANY); 
     serveraddr.sin_port = htons(SERV_PORT); 

     if(bind(listenfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr)) != 0) { 
            perror("bind failed"); 
            return -1; 
     } 

     if(listen(listenfd, LISTENQ) != 0) { 
            perror("listen failed"); 
            return -2; 
     } 

     maxi = 0; 

     printf("began to accept...\n"); 

     for(;;) { 
            nfds = epoll_wait(epfd, events, 32, 10000); 

            if(-1 == m_nfds)    { 
                 if(EINTR == errno) { 
            continue; 
    } 
    return -1; 
            } 

         for(i=0; i<nfds; ++i) { 
                 if(events[i].data.fd == listenfd) { 
                        connfd = accept(listenfd, (struct sockaddr *)&clientaddr, &clilen); 

                        if(connfd < 0) { 
                             perror("accept failed"); 
                             return -3; 
                        } 

         printf("accepted..\n"); 

                        setnonblock(connfd); 

                        ev.data.fd = connfd; 
                        ev.events = EPOLLIN | EPOLLET; 

                        epoll_ctl(epfd, EPOLL_CTL_ADD, connfd, &ev); 
                 } 
                 else if(events[i].events & EPOLLIN) { 
                        if((sockfd = events[i].data.fd) < 0) 
                             continue; 

                        if((n = read(sockfd, buf, MAXSIZE)) < 0) { 
                             if(errno == ECONNRESET) {    
                                    close(sockfd); 
                                    events[i].data.fd = -1; 
                             } else { 
                                    perror("read failed"); 
                             } 
                        } 
                        else if(0 == n) { 
                             close(sockfd); 
                             events[i].data.fd = -1; 
                        } 

         printf("Read the buf: %s\n", buf); 

                        ev.data.fd = sockfd; 
                        ev.events = EPOLLOUT | EPOLLET; 
                         
                        epoll_ctl(epfd, EPOLL_CTL_MOD, sockfd, &ev); 
                 } 
                 else if(events[i].events & EPOLLOUT) { 
                        sockfd = events[i].data.fd; 
                        char *sndbuf = "I get your message!"; 
                        write(sockfd,sndbuf, 10); 

                        ev.data.fd = sockfd; 
                        ev.events = EPOLLIN | EPOLLET; 

                        epoll_ctl(epfd, EPOLL_CTL_MOD, sockfd, &ev); 
                 } 
            } 
     } 
        
     return 0; 
}