Linux网络编程——tcp并发服务器（epoll实现）

转载：http://blog.csdn.net/lianghe_work/article/details/46551871

通过epoll实现tcp并发回执服务器（客户端给服务器发啥，服务器就给客户端回啥）

代码如下：

 #include
 #include
 #include
 #include
 #include
 #include
 #include
 #include
 #include
 #include
 #include
 #define OPEN_MAX 100

 int main(int argc, char *argv[])
 {
     struct epoll_event event;   // 告诉内核要监听什么事件
     struct epoll_event wait_event; //内核监听完的结果

     //1.创建tcp监听套接字
     int sockfd = socket(AF_INET, SOCK_STREAM, 0);

     //2.绑定sockfd
     struct sockaddr_in my_addr;
     bzero(&my_addr, sizeof(my_addr));
     my_addr.sin_family = AF_INET;
     my_addr.sin_port = htons(8001);
     my_addr.sin_addr.s_addr = htonl(INADDR_ANY);
     bind(sockfd, (struct sockaddr *)&my_addr, sizeof(my_addr));

     //3.监听listen
     listen(sockfd, 10);

     //4.epoll相应参数准备
     int fd[OPEN_MAX];
     int i = 0, maxi = 0;
     memset(fd,-1, sizeof(fd));
     fd[0] = sockfd;

     int epfd = epoll_create(10); // 创建一个 epoll 的句柄，参数要大于 0， 没有太大意义
     if( -1 == epfd ){
         perror ("epoll_create");
         return -1;
     }

     event.data.fd = sockfd;     //监听套接字
     event.events = EPOLLIN; // 表示对应的文件描述符可以读

     //5.事件注册函数，将监听套接字描述符 sockfd 加入监听事件
     int ret = epoll_ctl(epfd, EPOLL_CTL_ADD, sockfd, &event);
     if(-1 == ret){
         perror("epoll_ctl");
         return -1;
     }

     //6.对已连接的客户端的数据处理
     while(1)
     {
         // 监视并等待多个文件（标准输入，udp套接字）描述符的属性变化（是否可读）
         // 没有属性变化，这个函数会阻塞，直到有变化才往下执行，这里没有设置超时
         ret = epoll_wait(epfd, &wait_event, maxi+1, -1);

         //6.1监测sockfd(监听套接字)是否存在连接
         if(( sockfd == wait_event.data.fd )
             && ( EPOLLIN == wait_event.events & EPOLLIN ) )
         {
             struct sockaddr_in cli_addr;
             int clilen = sizeof(cli_addr);

             //6.1.1 从tcp完成连接中提取客户端
             int connfd = accept(sockfd, (struct sockaddr *)&cli_addr, &clilen);

             //6.1.2 将提取到的connfd放入fd数组中，以便下面轮询客户端套接字
             for(i=1; i
             {
                 if(fd[i] < 0)
                 {
                     fd[i] = connfd;
                     event.data.fd = connfd; //监听套接字
                     event.events = EPOLLIN; // 表示对应的文件描述符可以读

                     //6.1.3.事件注册函数，将监听套接字描述符 connfd 加入监听事件
                     ret = epoll_ctl(epfd, EPOLL_CTL_ADD, connfd, &event);
                     if(-1 == ret){
                         perror("epoll_ctl");
                         return -1;
                     }

                     break;
                 }
             }

             //6.1.4 maxi更新
             if(i > maxi)
                 maxi = i;

             //6.1.5 如果没有就绪的描述符，就继续epoll监测，否则继续向下看
             if(--ret <= 0)
                 continue;
         }

         //6.2继续响应就绪的描述符
         for(i=1; i<=maxi; i++)
         {
             if(fd[i] < 0)
                 continue;

             if(( fd[i] == wait_event.data.fd )
             && ( EPOLLIN == wait_event.events & (EPOLLIN|EPOLLERR) ))
             {
                 int len = 0;
                 char buf[128] = "";

                 //6.2.1接受客户端数据
                 if((len = recv(fd[i], buf, sizeof(buf), 0)) < 0)
                 {
                     if(errno == ECONNRESET)//tcp连接超时、RST
                     {
                         close(fd[i]);
                         fd[i] = -1;
                     }
                     else
                         perror("read error:");
                 }
                 else if(len == 0)//客户端关闭连接
                 {
                     close(fd[i]);
                     fd[i] = -1;
                 }
                 else//正常接收到服务器的数据
                     send(fd[i], buf, len, 0);

                 //6.2.2所有的就绪描述符处理完了，就退出当前的for循环，继续poll监测
                 if(--ret <= 0)
                     break;
             }
         }
     }
     return 0;
 }

运行结果：

源码下载：