zoukankan      html  css  js  c++  java
  • 第十九篇:不为客户连接创建子进程的并发回射服务器(select实现)

    前言

           在此前,我已经介绍了一种并发回射服务器实现。它通过调用fork函数为每个客户请求创建一个子进程。同时,我还为此服务器添加了自动消除僵尸子进程的机制。现在请想想,在客户量非常大的情况下,这种为每个客户请求都创建子进程的做法是不是太费资源了?我们可不可以在不为每个客户请求都创建子进程的前提下实现并发回射服务器

           答案自然是肯定的,这也是此文要讲述的方法。

    实现思路

           服务器建立好监听套接字之后,进入以下循环中:

    1. 调用select函数,先使之阻塞于监听套接字描述符(之后阻塞于监听套接字描述符和已连接套接字描述符);

    2. 当监听套接字描述符被激活,就调用connect函数建议一个客户连接,并将返回的已连接套接字描述符也登记到select函数的监听描述符集中。(这里我们需要再设立一个client数组记录下所有的已连接套接字描述符在select函数的监听描述符集中的位置);

    3. 遍历这个client数组,一旦发现有已连接的监听套接字被激活,则做出相应处理。

    代码实现

     1 #include    "unp.h"
     2 
     3 int
     4 main(int argc, char **argv)
     5 {
     6     int                    i, maxi, maxfd, listenfd, connfd, sockfd;
     7     int                    nready, client[FD_SETSIZE];
     8     ssize_t                n;
     9     fd_set                rset, allset;
    10     char                buf[MAXLINE];
    11     socklen_t            clilen;
    12     struct sockaddr_in    cliaddr, servaddr;
    13 
    14     listenfd = Socket(AF_INET, SOCK_STREAM, 0);
    15 
    16     bzero(&servaddr, sizeof(servaddr));
    17     servaddr.sin_family      = AF_INET;
    18     servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    19     servaddr.sin_port        = htons(SERV_PORT);
    20 
    21     Bind(listenfd, (SA *) &servaddr, sizeof(servaddr));
    22 
    23     Listen(listenfd, LISTENQ);
    24 
    25     // maxfd 表示select函数的监听描述符集中的最大值( 作为第一个参数传递给select )。
    26     maxfd = listenfd;            
    27     // maxi 表示client数组的边界
    28     maxi = -1;                
    29     for (i = 0; i < FD_SETSIZE; i++)
    30         client[i] = -1;            
    31     FD_ZERO(&allset);
    32     FD_SET(listenfd, &allset);
    33 
    34     for ( ; ; ) {
    35         rset = allset;        
    36         nready = Select(maxfd+1, &rset, NULL, NULL, NULL);
    37 
    38         if (FD_ISSET(listenfd, &rset)) {    
    39             clilen = sizeof(cliaddr);
    40             // 和客户建立连接
    41             connfd = Accept(listenfd, (SA *) &cliaddr, &clilen);
    42 #ifdef    NOTDEF
    43             printf("new client: %s, port %d
    ",
    44                     Inet_ntop(AF_INET, &cliaddr.sin_addr, 4, NULL),
    45                     ntohs(cliaddr.sin_port));
    46 #endif
    47             // 将已连接套接字描述符登记进client数组
    48             for (i = 0; i < FD_SETSIZE; i++)
    49                 if (client[i] < 0) {
    50                     client[i] = connfd;    
    51                     break;
    52                 }
    53             if (i == FD_SETSIZE)
    54                 err_quit("too many clients");
    55 
    56             // 将已连接套接字描述符登记进select函数的监听套接字描述符集中
    57             FD_SET(connfd, &allset);
    58 
    59             /*
    60              * 下面两个if语句在获取到已连接描述符后更新maxfd和maxi。
    61             */
    62             if (connfd > maxfd)
    63                 maxfd = connfd;            
    64             if (i > maxi)
    65                 maxi = i;            
    66 
    67             // 如果所有被监听的信号已经处理完毕了,则立即启动下一次循环。
    68             if (--nready <= 0)
    69                 continue;                
    70         }
    71 
    72         // 检查select监听描述符集中所有的已连接套接字描述符是否有被激活的,有则做出相应处理。
    73         for (i = 0; i <= maxi; i++) {
    74             if ( (sockfd = client[i]) < 0)
    75                 continue;
    76             if (FD_ISSET(sockfd, &rset)) {
    77                 if ( (n = Read(sockfd, buf, MAXLINE)) == 0) {
    78                     Close(sockfd);
    79                     FD_CLR(sockfd, &allset);
    80                     client[i] = -1;
    81                 } else
    82                     Writen(sockfd, buf, n);
    83 
    84             // 如果所有被监听的信号已经处理完毕了,则立即启动下一次循环。
    85                 if (--nready <= 0)
    86                     break;                
    87             }
    88         }
    89     }
    90 }

    说明

    1. 设置client数组是必须的,它用来标记select的监听描述符集中哪些位处于被监听状态;

    2. 使用这种技术的程序容易收到拒绝服务攻击。

  • 相关阅读:
    Java 简单算法--打印乘法口诀(只使用一次循环)
    Java简单算法--求100以内素数
    ubuntu 16.04 chrome flash player 过期
    java 网络API访问 web 站点
    java scoket (UDP通信模型)简易聊天室
    leetcode1105 Filling Bookcase Shelves
    leetcode1140 Stone Game II
    leetcode1186 Maximum Subarray Sum with One Deletion
    leetcode31 Next Permutation
    leetcode834 Sum of Distances in Tree
  • 原文地址:https://www.cnblogs.com/muchen/p/6877469.html
Copyright © 2011-2022 走看看