linux c select 服务器源码简介

Myth丶恋晨 2021-09-07 06:14 198阅读 0赞

直接看调用主函数中while循环的部分

每次调用select都会改变FD\_SET的值， 所以要两个变量，allset，以及一个rset

先看看是否有客户端请求连接，如果有处理连接。

这里用新开一个数组维护文件描述符。这样的好处是如果用动态数组，可以动态维护文件描述符。

遍历整个客户端的数组，然后来看看有没有可以读的信息

select函数的缺点是每次要将FD\_SET拷贝到内核区，是基于轮询机制的IO多路复用。

while (1) {   
            rset = allset;                                          /* 每次循环时都从新设置select监控信号集 */
            nready = select(maxfd+1, &rset, NULL, NULL, NULL);
            if (nready < 0)
                perr_exit("select error");
     
            if (FD_ISSET(listenfd, &rset)) {                        /* 说明有新的客户端链接请求 */
     
                clie_addr_len = sizeof(clie_addr);
                connfd = Accept(listenfd, (struct sockaddr *)&clie_addr, &clie_addr_len);       /* Accept 不会阻塞 */
                printf("received from %s at PORT %d\n",
                        inet_ntop(AF_INET, &clie_addr.sin_addr, str, sizeof(str)),
                        ntohs(clie_addr.sin_port));
     
                for (i = 0; i < FD_SETSIZE; i++)
                    if (client[i] < 0) {                            /* 找client[]中没有使用的位置 */
                        client[i] = connfd;                         /* 保存accept返回的文件描述符到client[]里 */
                        break;
                    }
     
                if (i == FD_SETSIZE) {                              /* 达到select能监控的文件个数上限 1024 */
                    fputs("too many clients\n", stderr);
                    exit(1);
                }
     
                FD_SET(connfd, &allset);                            /* 向监控文件描述符集合allset添加新的文件描述符connfd */
                if (connfd > maxfd)
                    maxfd = connfd;                                 /* select第一个参数需要 */
     
                if (i > maxi)
                    maxi = i;                                       /* 保证maxi存的总是client[]最后一个元素下标 */
     
                if (--nready == 0)
                    continue;
            } 
     
            for (i = 0; i <= maxi; i++) {                               /* 检测哪个clients 有数据就绪 */
     
                if ((sockfd = client[i]) < 0)
                    continue;
                if (FD_ISSET(sockfd, &rset)) {
     
                    if ((n = Read(sockfd, buf, sizeof(buf))) == 0) {    /* 当client关闭链接时,服务器端也关闭对应链接 */
                        Close(sockfd);
                        FD_CLR(sockfd, &allset);                        /* 解除select对此文件描述符的监控 */
                        client[i] = -1;
                    } else if (n > 0) {
                        for (j = 0; j < n; j++)
                            buf[j] = toupper(buf[j]);
                        Write(sockfd, buf, n);
                        Write(STDOUT_FILENO, buf, n);
                    }
                    if (--nready == 0)
                        break;                                          /* 跳出for, 但还在while中 */
                }
            }
        }