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

想详细彻底地了解poll或看懂下面的代码请参考《Linux网络编程——I/O复用之poll函数》

代码：

 #include <string.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <unistd.h>
 #include <sys/select.h>
 #include <sys/time.h>
 #include <sys/socket.h>
 #include <netinet/in.h>
 #include <arpa/inet.h>
 #include <poll.h>
 #include <errno.h>
 #define OPEN_MAX 100  

 int main(int argc, char *argv[])
 {
     //1.创建tcp监听套接字
     int sockfd = socket(AF_INET, SOCK_STREAM, );  

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

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

     //4.poll相应参数准备
     struct pollfd client[OPEN_MAX];
     int i = , maxi = ;
     for(;i<OPEN_MAX; i++)
         client[i].fd = -;//初始化poll结构中的文件描述符fd  

     client[].fd = sockfd;//需要监测的描述符
     client[].events = POLLIN;//普通或优先级带数据可读  

     //5.对已连接的客户端的数据处理
     while()
     {
         int ret = poll(client, maxi+, -);//对加入poll结构体数组所有元素进行监测  

         //5.1监测sockfd(监听套接字)是否存在连接
         if((client[].revents & POLLIN) == POLLIN )
         {
             struct sockaddr_in cli_addr;
             int clilen = sizeof(cli_addr);
             int connfd = ;
             //5.1.1 从tcp完成连接中提取客户端
             connfd = accept(sockfd, (struct sockaddr *)&cli_addr, &clilen);  

             //5.1.2 将提取到的connfd放入poll结构体数组中，以便于poll函数监测
             for(i=; i<OPEN_MAX; i++)
             {
                 if(client[i].fd < )
                 {
                     client[i].fd = connfd;
                     client[i].events = POLLIN;
                     break;
                 }
             }  

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

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

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

             if(client[i].revents & (POLLIN | POLLERR))
             {
                 int len = ;
                 char buf[] = "";  

                 //5.2.1接受客户端数据
                 if((len = recv(client[i].fd, buf, sizeof(buf), )) < )
                 {
                     if(errno == ECONNRESET)//tcp连接超时、RST
                     {
                         close(client[i].fd);
                         client[i].fd = -;
                     }
                     else
                         perror("read error:");  

                 }
                 else if(len == )//客户端关闭连接
                 {
                     close(client[i].fd);
                     client[i].fd = -;
                 }
                 else//正常接收到服务器的数据
                     send(client[i].fd, buf, len, );  

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

             }
         }
     }
     return ;
 }  

运行结果：