socket通信进阶一，使用epoll

之前的例子是使用的简单的socket编程，这样可以实现简单的一对一的连接，如果我们的服务端可能有很多的客户端呢？此时该怎么办?
     这里我们用使用epoll，为什么是epoll？epoll是在linux2.6中添加的。
     它有三个特性：mmap，红黑树，双链表。这三个特性使得它的性能优于select，poll。
      mmap将用户空间和内核空间映射到同一块内存中，使得数据传递不再是select，poll中使用的句柄拷贝，而是直接访问就可以了。
      先介绍下它的函数：

#include <sys/epoll.h> 
int epoll_create(int size);
int epoll_create1(int flags)
//Since Linux 2.6.8, the size argument is ignored, but must be greater than  zero; see NOTES below.

作为函数返回值，epoll_create()返回了代表新创建的epoll 实例的文件描述符。这个文件描述符在其他几个epoll 系统调用中用来表示epoll 实例。
当这个文件描述符不再需要时，应该通过close()来关闭。当所有与epoll 实例相关的文件描述符都被关闭时，实例被销毁，相关的资源都返还给系统。
（多个文件描述符可能引用到相同的epoll 实例，这是由于调用了fork()或者dup()这样类似的函数所致。）

#include <sys/epoll.h>
int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event);

int epfd:
修改由文件描述符epfd 所代表的epoll 实例中的兴趣列表,

int op
参数op 用来指定需要执行的操作，它可以是如下几种值:
EPOLL_CTL_ADD
将描述符fd 添加到epoll 实例epfd 中的兴趣列表中去。对于fd 上我们感兴趣的事件，都
指定在ev 所指向的结构体中，下面会详细介绍。如果我们试图向兴趣列表中添加一个已存在
的文件描述符，epoll_ctl()将出现EEXIST 错误。

EPOLL_CTL_MOD
修改描述符fd 上设定的事件，需要用到由ev 所指向的结构体中的信息。如果我们试图修
改不在兴趣列表中的文件描述符，epoll_ctl()将出现ENOENT 错误。

EPOLL_CTL_DEL
将文件描述符fd 从epfd 的兴趣列表中移除。该操作忽略参数ev。如果我们试图移除一个
不在epfd 的兴趣列表中的文件描述符，epoll_ctl()将出现ENOENT 错误。关闭一个文件描述
符会自动将其从所有的epoll 实例的兴趣列表中移除。

int fd
指明了要修改兴趣列表中的哪一个文件描述符的设定。该参数可以是代表管道、
FIFO、套接字、POSIX 消息队列、inotify 实例、终端、设备，甚至是另一个epoll 实例的文件
描述符（例如，我们可以为受检查的描述符建立起一种层次关系）。但是，这里fd 不能作为普
通文件或目录的文件描述符（会出现EPERM 错误）。

struct epoll_event *event:
参数event 是指向结构体epoll_event 的指针，结构体的定义如下：

struct epoll_event
{
  uint32_t events;
  epoll_data_t data;
};

结构体epoll_event 中的data 字段的类型为：

typedef union epoll_data
{
   void *ptr;
   int fd;
   uint32_t u32;
   uint64_t u64;
}epoll_data_t;

参数ev 为文件描述符fd 所做的设置如下:
    1.结构体 epoll_event 中的 events 字段是一个位掩码，它指定了我们为待检查的描述
符 fd 上所感兴趣的事件集合。
    2.data 字段是一个联合体，当描述符fd 稍后成为就绪态时，联合体的成员可用来指定传
回给调用进程的信息。

int epoll_wait(int epfd, struct epoll_event *evlist,int maxevents, int timeout);
struct epoll_event *evlist,int maxevents:
参数evlist 所指向的结构体数组中返回的是有关就绪态文件描述符的信息。（结构体epoll_event 已经在上一节中描述。）数组evlist 的空间由调用者负责申请，所包含的元素个数在参数maxevents 中指定。

在数组evlist中，每个元素返回的都是单个就绪态文件描述符的信息。events 字段返回了在该描述符上已经发生的事件掩码。
data 字段返回的是我们在描述符上使用 epoll_ctl()注册感兴趣的事件时在ev.data 中所指定的值。
注意，data 字段是唯一可获知同这个事件相关的文件描述符号的途径。因此，当我们调用epoll_ctl()将文件描述符添加到兴趣列表中时，应该要么将ev.data.fd 设为文件描述符号，要么将ev.data.ptr 设为指向包含文件描述符号的结构体。
 
int timeout:
参数timeout 用来确定epoll_wait()的阻塞行为，有如下几种。
   如果timeout 等于?1，调用将一直阻塞，直到兴趣列表中的文件描述符上有事件产生，或者直到捕获到一个信号为止。
   如果timeout 等于0，执行一次非阻塞式的检查，看兴趣列表中的文件描述符上产生了哪个事件。
   如果timeout 大于0，调用将阻塞至多timeout 毫秒，直到文件描述符上有事件发生，或者直到捕获到一个信号为止。
使用epoll的实例，功能与上篇相同：

#include <cstdio>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <cstdlib>
#include <cstring>
#include <unistd.h>
#include <netinet/in.h>
#include <sys/epoll.h>
#include <cerrno>
#include <iostream>
#include <string>
#include <vector>
#include <cmath>
#include <sys/resource.h>    

using namespace std;

#define    MAXEPOLL    10000
#define    MAXLINE        1024
#define MAX_MSG_SIZE 256
#define SERVER_PORT 9987
 
 
#define BACKLOG 2

struct msg_req{
    int len;
    char msg[MAX_MSG_SIZE];
};

struct msg_rsp{
  int len;
  char msg[MAX_MSG_SIZE];
}; 
 
vector<string> split(string strtem,char a)
{
    vector<string> strvec;
    string::size_type pos1, pos2;
    pos2 = strtem.find(a);
    pos1 = 0;
    while (string::npos != pos2)
    {
        strvec.push_back(strtem.substr(pos1, pos2 - pos1));
        pos1 = pos2 + 1;
        pos2 = strtem.find(a, pos1);
    }
    strvec.push_back(strtem.substr(pos1));
    return strvec;
}

void process(struct msg_req &req,struct msg_rsp &rsp)
{
  int len=strlen(req.msg);
  printf("len=%d\n.msg=",len,req.msg);
  
  std::string str_cmd(req.msg);
  vector<string> res_list=split(str_cmd,',');
  float a=atof(res_list[0].c_str()); 
  float b=atof(res_list[1].c_str());
  float c=sqrt((pow(a,2))+(pow(b,2)));
  string str_result=std::to_string(c);
  strcpy(rsp.msg,str_result.c_str());
  return;
}
 
int main(){ 
    int         conn_fd;
    int         epoll_fd;
    int         nread;
    int         cur_fds;
    int         wait_fds;
    
    int sock_fd,client_fd; /*sock_fd：监听socket；client_fd：数据传输socket */ 
    struct sockaddr_in ser_addr; /* 本机地址信息 */ 
    struct sockaddr_in cli_addr; /* 客户端地址信息 */ 
    char msg[MAX_MSG_SIZE];/* 缓冲区*/ 
    int ser_sockfd=socket(AF_INET,SOCK_STREAM,0);/*创建连接的SOCKET */ 
    if(ser_sockfd<0) 
    { 
        /*创建失败 */ 
        fprintf(stderr,"socker Error:%s\n",strerror(errno)); 
        exit(1); 
    } 
    struct     epoll_event    ev;
    struct     epoll_event    evs[MAXEPOLL];
    struct     rlimit    rlt;
    rlt.rlim_max = rlt.rlim_cur = MAXEPOLL;

    /* 初始化服务器地址*/ 
    socklen_t addrlen=sizeof(struct sockaddr_in); 
    bzero(&ser_addr,addrlen); 
    ser_addr.sin_family=AF_INET; 
    ser_addr.sin_addr.s_addr=htonl(INADDR_ANY); 
    ser_addr.sin_port=htons(SERVER_PORT); 
    if(bind(ser_sockfd,(struct sockaddr*)&ser_addr,sizeof(struct sockaddr_in))<0)
    { /*绑定失败 */ 
        fprintf(stderr,"Bind Error:%s\n",strerror(errno)); 
        exit(1); 
    }

    
    /*侦听客户端请求*/ 
    if(listen(ser_sockfd,BACKLOG)<0)
    { 
        fprintf(stderr,"Listen Error:%s\n",strerror(errno)); 
        close(ser_sockfd); 
        exit(1); 
    } 


    epoll_fd = epoll_create( MAXEPOLL ); // 
    ev.events = EPOLLIN | EPOLLET;    //表示对应的文件描述符可以读|采用边缘触发事件通知
    ev.data.fd = ser_sockfd;

    if( epoll_ctl( epoll_fd, EPOLL_CTL_ADD, ser_sockfd, &ev ) < 0 )
    {
        printf("Epoll Error : %d\n", errno);
        exit( EXIT_FAILURE );
    }
    
    cur_fds = 1;
    struct msg_req req;
    struct msg_rsp rsp;
    while(1)
    {
     if( ( wait_fds = epoll_wait( epoll_fd, evs, cur_fds, -1 ) ) == -1 ) //
     {
            printf( "Epoll Wait Error : %d\n", errno );
            exit( -1 );
        }
        for( i = 0; i < wait_fds; i++ )
        {
            if( evs[i].data.fd == ser_sockfd && cur_fds < MAXEPOLL )
            {
              if( ( conn_fd = accept( ser_sockfd, (struct sockaddr *)&cli_addr, &addrlen ) ) == -1 )
              {
                    printf("Accept Error : %d\n", errno);
                    exit( -2 );
             }

             printf( "Server get from client !\n"/*, inet_ntoa(cliaddr.sin_addr), cliaddr.sin_port */);

             ev.events = EPOLLIN | EPOLLET;
             ev.data.fd = conn_fd;
             if( epoll_ctl( epoll_fd, EPOLL_CTL_ADD, conn_fd, &ev ) < 0 )
             {
                 printf("Epoll Error : %d\n", errno);
                 exit( -3 );
             }

             ++cur_fds;
             continue;
   
            }
           
            
            req.msg[0]='\0';
            rsp.msg[0]='\0';
            nread = read( evs[i].data.fd, req.msg, sizeof(msg) );
            
            printf("recv:%s\n",req.msg);
            if( nread <= 0 )
            {
                close( evs[i].data.fd );
                epoll_ctl( epoll_fd, EPOLL_CTL_DEL, evs[i].data.fd, &ev );
                --cur_fds;
                continue;
            }

            process(req,rsp);
            printf("send:%s\n",rsp.msg);
            write( evs[i].data.fd, rsp.msg, nread );

        }

    }

    close(ser_sockfd); 
    return 0; 
 }