epoll使用

功能描述：
    使用epoll方式监听网络数据，监听客户端连接和接收客户端发送数据。

实验环境
    系统:ubuntu16.04
    地址:192.168.31.14
    端口:1234
    GCC版本:gcc version 5.4.0 20160609
    
测试工具：
    使用NetAssist网络调试助手

目的：epoll的使用    

流程：
    1 创建服务端，得到 listen_sock
    2 使用epoll_ctl 添加listen_sock的监测
    3 使用epoll_wait 等待EPOLLIN事件
    4 处理EPOLLIN事件，使用accept接收客户端，得到客户端conn_sock
    5 使用epoll_ctl 添加客户端conn_sock的监测
    6 使用epoll_wait 等待EPOLLIN事件
    7 处理EPOLLIN事件，使用do_use_fd，读取客户端conn_sock内容

参考：   man socket 
              man epoll
              man epoll_ctl

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

#define MAX_EVENTS 10

int get_listen_sock(const char * ip, int port)
{
        int server_socket = socket(AF_INET, SOCK_STREAM, 0);
        struct sockaddr_in server_addr;
        memset(&server_addr, 0, sizeof(server_addr));
        server_addr.sin_family = AF_INET;
        server_addr.sin_addr.s_addr = inet_addr(ip);
        server_addr.sin_port = htons(port);
        bind(server_socket, (struct sockaddr*)&server_addr, sizeof(server_addr));
        listen(server_socket, 20);
        return server_socket;
}

int setnonblocking(int fd) 
{
        int old_option = fcntl(fd, F_GETFL);
        int new_option = old_option | O_NONBLOCK;
        fcntl(fd, F_SETFL, new_option);
        return old_option;

} 

void do_use_fd(int fd)
{
        unsigned char buf[1024];
        int r = read(fd, buf, 1024);
        printf("recv:%s\n", buf);
}
void main()
{
        struct epoll_event ev, events[MAX_EVENTS];
        int listen_sock, conn_sock, nfds, epollfd,n;
        struct sockaddr_in local;
        socklen_t addrlen;
        /* Code to set up listening socket, 'listen_sock',
           (socket(), bind(), listen()) omitted */
        listen_sock = get_listen_sock("192.168.31.14", 1234);

        epollfd = epoll_create1(0);
        if (epollfd == -1) {
                perror("epoll_create1");
                exit(EXIT_FAILURE);
        }

        ev.events = EPOLLIN;
        ev.data.fd = listen_sock;
        if (epoll_ctl(epollfd, EPOLL_CTL_ADD, listen_sock, &ev) == -1) {
                perror("epoll_ctl: listen_sock");
                exit(EXIT_FAILURE);
        }

        for (;;) {
                nfds = epoll_wait(epollfd, events, MAX_EVENTS, -1);
                if (nfds == -1) {
                        perror("epoll_wait");
                        exit(EXIT_FAILURE);
                }

                for (n = 0; n < nfds; ++n) {
                        if (events[n].data.fd == listen_sock) {
                                conn_sock = accept(listen_sock,
                                                (struct sockaddr *) &local, &addrlen);
                                if (conn_sock == -1) {
                                        perror("accept");
                                        exit(EXIT_FAILURE);
                                }
                                setnonblocking(conn_sock);
                                ev.events = EPOLLIN | EPOLLET;
                                ev.data.fd = conn_sock;
                                if (epoll_ctl(epollfd, EPOLL_CTL_ADD, conn_sock,
                                                        &ev) == -1) {
                                        perror("epoll_ctl: conn_sock");
                                        exit(EXIT_FAILURE);
                                }
                        } else {
                                do_use_fd(events[n].data.fd);
                        }
                }
        }

}

测试结果：