epoll 示例代码

server.cpp

#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <math.h>
#include <string.h>
#include <pthread.h> 
#include <signal.h>
#include <netinet/in.h>
#include <sys/file.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/epoll.h>

#include <set>
using namespace std;
#define MAX_EVENTS 1024 
#define MAX_BUF_LEN 1024 
set<int> connectFds;
void stop(int signo) {  
        printf("Stop\n");  
        set<int>::iterator it;
        for(it = connectFds.begin(); it != connectFds.end(); it++){
                close(*it);
        }
        connectFds.clear();
        exit(EXIT_SUCCESS);  
}  
void *handleConnection(void *arg){
        int *data = (int*) arg;
        int connectFd = *data, len;
        char buf[MAX_BUF_LEN]; 
        printf("current thread id is %u, connectFd:%d\n",(unsigned)pthread_self(), connectFd); 
        connectFds.insert(connectFd);
        while(true){
                len = read(connectFd, buf, MAX_BUF_LEN);
                if(len <= 0) break; //连接断开
                printf("Get message:%s\n", buf);
                if(strcmp(buf, "quit") == 0 || 
                        strcmp(buf, "exit") == 0){
                        break;
                }
        }
        close(connectFd);
        connectFds.erase(connectFd);
        printf("Close connection\n");
}
int main(int argc, char** argv) {
        int epollFd, listenFd, connectFd, nfds;
        uint16_t serverPort = 9999;
        epoll_event ev, events[MAX_EVENTS];
        sockaddr_in serverAddr;
        /*{{{sock*/
        listenFd = socket(AF_INET, SOCK_STREAM, 0);
        if(listenFd == -1){
                printf("socket failed\n");
                exit(EXIT_FAILURE);
        }
        memset(&serverAddr, 0, sizeof(serverAddr));
        serverAddr.sin_family = AF_INET;
        serverAddr.sin_addr.s_addr = htonl(INADDR_ANY);
        serverAddr.sin_port = htons(serverPort);
        bind(listenFd, (sockaddr*) &serverAddr, sizeof(serverAddr));
        listen(listenFd, 16);
        printf("Listening ...\n");
        /*}}}*/
        epollFd = epoll_create(512);
        if(epollFd == -1){
                printf("epoll_create failed\n");
                exit(EXIT_FAILURE);
        }
        ev.events = EPOLLIN | EPOLLOUT;
        ev.data.fd = listenFd;
        if(epoll_ctl(epollFd, EPOLL_CTL_ADD, listenFd, &ev) == -1){
                printf("epoll_ctl add event failed\n");
                exit(EXIT_FAILURE);
        }
        signal(SIGINT, stop);   
        while(true){
                nfds = epoll_wait(epollFd, events, MAX_EVENTS, -1);
                if(nfds == -1){
                        printf("epoll_wait failed\n");
                        exit(EXIT_FAILURE);
                }
                for(int i = 0; i < nfds; i++){
                        if(events[i].data.fd == listenFd){
                                connectFd = accept(listenFd, (sockaddr*)NULL, NULL);
                                printf("Connected ...\n");
                                pthread_t thread;
                                pthread_create(&thread, NULL, handleConnection, (void *) &connectFd);
                        }
                }
        }
        printf("Success over !\n");
        return 0;
}

client.cpp

#include <stdio.h>  
#include <stdlib.h>  
#include <string.h>  
#include <errno.h>  
#include <unistd.h>
#include <sys/types.h>  
#include <sys/socket.h>  
#include <arpa/inet.h>
#include <netinet/in.h>
#define MAX_BUF_LEN 1024 
int main(int argc, char** argv) {
        int sockFd, sendLen;
        uint16_t serverPort = 9999;
        char buf[MAX_BUF_LEN];
        sockaddr_in serverAddr;

        sockFd = socket(AF_INET, SOCK_STREAM, 0);
        memset(&serverAddr, 0, sizeof(serverAddr));
        serverAddr.sin_family = AF_INET;
        serverAddr.sin_port = htons(serverPort);
        inet_pton(AF_INET, "127.0.0.1", &serverAddr.sin_addr);
        if(connect(sockFd, (sockaddr*)&serverAddr, sizeof(serverAddr)) == -1){
                printf("connect failed\n");
                exit(EXIT_FAILURE);
        }
        while(scanf("%s", buf) != EOF){
                if((sendLen = send(sockFd, buf, strlen(buf), 0)) < 0){
                        printf("send message error : %s", strerror(errno));
                        break;
                }
                printf("sendLen : %d\n", sendLen);
                if(strcmp(buf, "quit") == 0 || 
                        strcmp(buf, "exit") == 0){
                        break;
                }
        }
        close(sockFd);
        return 0;
}