当前位置:   article > 正文

linux select函数详解_timeout.tv_sec = 1; timeout.tv_usec = 0;什么意思

timeout.tv_sec = 1; timeout.tv_usec = 0;什么意思


select是等待IO读取阻塞的时侯基本不占CPU,






原文地址:http://blog.csdn.net/lingfengtengfei/article/details/12392449


linux select函数详解

分类: linux 奇淫技巧 笔试面试精选   7217人阅读  评论(6)  收藏  举报

Linux中,我们可以使用select函数实现I/O端口的复用,传递给 select函数的参数会告诉内核:

      •我们所关心的文件描述符

      •对每个描述符,我们所关心的状态。(我们是要想从一个文件描述符中读或者写,还是关注一个描述符中是否出现异常)

      •我们要等待多长时间。(我们可以等待无限长的时间,等待固定的一段时间,或者根本就不等待)

    select函数返回后,内核告诉我们一下信息:

      •对我们的要求已经做好准备的描述符的个数

      •对于三种条件哪些描述符已经做好准备.(读,写,异常)

   有了这些返回信息,我们可以调用合适的I/O函数(通常是 read  write),并且这些函数不会再阻塞.

 

#include <sys/select.h>   

    int select(int maxfdp1, fd_set *readset, fd_set *writeset, fd_set *exceptset,struct timeval *timeout);

   

   返回:做好准备的文件描述符的个数,超时为0,错误为 -1.

   

   首先我们先看一下最后一个参数。它指明我们要等待的时间:

struct timeval{      

        long tv_sec;   /* */

        long tv_usec;  /*微秒 */   

    }

   

   有三种情况:

    timeout == NULL  等待无限长的时间。等待可以被一个信号中断。当有一个描述符做好准备或者是捕获到一个信号时函数会返回。如果捕获到一个信号, select函数将返回 -1,并将变量 erro设为 EINTR

    timeout->tv_sec == 0 &&timeout->tv_usec == 0不等待,直接返回。加入描述符集的描述符都会被测试,并且返回满足要求的描述符的个数。这种方法通过轮询,无阻塞地获得了多个文件描述符状态。

    timeout->tv_sec !=0 ||timeout->tv_usec!= 0 等待指定的时间。当有描述符符合条件或者超过超时时间的话,函数返回。在超时时间即将用完但又没有描述符合条件的话,返回 0。对于第一种情况,等待也会被信号所中断。

   

   中间的三个参数 readset, writset, exceptset,指向描述符集。这些参数指明了我们关心哪些描述符,和需要满足什么条件(可写,可读,异常)。一个文件描述集保存在 fd_set 类型中。fd_set类型变量每一位代表了一个描述符。我们也可以认为它只是一个由很多二进制位构成的数组。如下图所示:

   

   对于 fd_set类型的变量我们所能做的就是声明一个变量,为变量赋一个同种类型变量的值,或者使用以下几个宏来控制它:

 

#include <sys/select.h>   

int FD_ZERO(int fd, fd_set *fdset);   

int FD_CLR(int fd, fd_set *fdset);   

int FD_SET(int fd, fd_set *fd_set);   

int FD_ISSET(int fd, fd_set *fdset);

    FD_ZERO宏将一个 fd_set类型变量的所有位都设为 0,使用FD_SET将变量的某个位置位。清除某个位时可以使用 FD_CLR,我们可以使用FD_ISSET来测试某个位是否被置位。

   当声明了一个文件描述符集后,必须用FD_ZERO将所有位置零。之后将我们所感兴趣的描述符所对应的位置位,操作如下:

 

fd_set rset;   

int fd;   

FD_ZERO(&rset);   

FD_SET(fd, &rset);   

FD_SET(stdin, &rset);</span>

    

    select返回后,用FD_ISSET测试给定位是否置位:

 

if(FD_ISSET(fd, &rset)   

{ ... }

具体解释select的参数:

1intmaxfdp是一个整数值,是指集合中所有文件描述符的范围,即所有文件描述符的最大值加1,不能错。

说明:对于这个原理的解释可以看上边fd_set的详细解释,fd_set是以位图的形式来存储这些文件描述符。maxfdp也就是定义了位图中有效的位的个数。

2fd_set*readfds是指向fd_set结构的指针,这个集合中应该包括文件描述符,我们是要监视这些文件描述符的读变化的,即我们关心是否可以从这些文件中读取数据了,如果这个集合中有一个文件可读,select就会返回一个大于0的值,表示有文件可读;如果没有可读的文件,则根据timeout参数再判断是否超时,若超出timeout的时间,select返回0,若发生错误返回负值。可以传入NULL值,表示不关心任何文件的读变化。

3fd_set*writefds是指向fd_set结构的指针,这个集合中应该包括文件描述符,我们是要监视这些文件描述符的写变化的,即我们关心是否可以向这些文件中写入数据了,如果这个集合中有一个文件可写,select就会返回一个大于0的值,表示有文件可写,如果没有可写的文件,则根据timeout参数再判断是否超时,若超出timeout的时间,select返回0,若发生错误返回负值。可以传入NULL值,表示不关心任何文件的写变化。

4fd_set*errorfds同上面两个参数的意图,用来监视文件错误异常文件。

5structtimeval* timeoutselect的超时时间,这个参数至关重要,它可以使select处于三种状态,第一,若将NULL以形参传入,即不传入时间结构,就是将select置于阻塞状态,一定等到监视文件描述符集合中某个文件描述符发生变化为止;第二,若将时间值设为00毫秒,就变成一个纯粹的非阻塞函数,不管文件描述符是否有变化,都立刻返回继续执行,文件无变化返回0,有变化返回一个正值;第三,timeout的值大于0,这就是等待的超时时间,即 selecttimeout时间内阻塞,超时时间之内有事件到来就返回了,否则在超时后不管怎样一定返回,返回值同上述。

说明:

函数返回:

1)当监视的相应的文件描述符集中满足条件时,比如说读文件描述符集中有数据到来时,内核(I/O)根据状态修改文件描述符集,并返回一个大于0的数。

2)当没有满足条件的文件描述符,且设置的timeval监控时间超时时,select函数会返回一个为0的值。

3)当select返回负值时,发生错误。

理解select模型:

理解select模型的关键在于理解fd_set,为说明方便,取fd_set长度为1字节,fd_set中的每一bit可以对应一个文件描述符fd。则1字节长的fd_set最大可以对应8fd

1)执行fd_set set;FD_ZERO(&set);set用位表示是0000,0000

2)若fd5,执行FD_SET(fd,&set);set变为0001,0000(5位置为1)

3)若再加入fd2fd=1,set变为0001,0011

4)执行select(6,&set,0,0,0)阻塞等待

5)若fd=1,fd=2上都发生可读事件,则select返回,此时set变为0000,0011。注意:没有事件发生的fd=5被清空。

基于上面的讨论,可以轻松得出select模型的特点:

1)可监控的文件描述符个数取决与sizeof(fd_set)的值。我这边服务器上sizeof(fd_set)512,每bit表示一个文件描述符,则我服务器上支持的最大文件描述符是512*8=4096。据说可调,另有说虽然可调,但调整上限受于编译内核时的变量值。

2)将fd加入select监控集的同时,还要再使用一个数据结构array保存放到select监控集中的fd,一是用于再select返回后,array作为源数据和fd_set进行FD_ISSET判断。二是select返回后会把以前加入的但并无事件发生的fd清空,则每次开始 select前都要重新从array取得fd逐一加入(FD_ZERO最先),扫描array的同时取得fd最大值maxfd,用于select的第一个参数。

3)可见select模型必须在select前循环array(加fd,取maxfd),select返回后循环arrayFD_ISSET判断是否有时间发生)。

基本原理

Resize icon

 select()系统调用代码走读

调用顺序如下:sys_select() à core_sys_select() à do_select() à fop->poll()

 

 

 

 

 

 

计算机生成了可选文字:stdinS七dOU七S七derrlistensocke七fdClient【]readSet[0][1]一1[2][3][FDSETSIZE一l]图1.正在监听客户端的连接

 

计算机生成了可选文字:re〕dSet115七ensocke七fdClient【][0]级IT[aXi=0S七dOU七~stdi诊‘s七derr)〔厂”rs'Cli’吐[l]一112][3][FDSETSIZE一1]其中第一个可用的套接字为4,所以第一个。liert为4,maxi为。“ent数组当前使用项的最大下标。图2.第1个客户端建立连接后

 

计算机生成了可选文字:115七ensocke七fd反firs七client眨SeCOndC"’吐厂―---readSetClient【]S七dOUt[0]…s'd'rr)一n·1d七S[l]Ir.SX立=1[3]其中第一个可用的套接字为5,所以第2个。!1ent为5,maxi为。Iient数组当前使用项的最大下标.[FDSETSIZE一1]图3.第2个客户端建立连接后.

 

计算机生成了可选文字:re日dSetClient[]“七derr〕115七ensocke七firs七clien七七ermina七ed[0]111]5i_」,_。七dou七…5'"'n)I'r己<卜=l.fdofdZfd4交SeC'ndC"’吐fds1入…d一f[2]一1一1[FDSETSIZE一1]图4.第1个客户端断开连接后

参考资料:

http://blog.csdn.net/tianmohust/article/details/6595998 

http://www.cnblogs.com/jinmu190/archive/2010/11/21/1883184.html




原文地址:http://blog.csdn.net/turkeyzhou/article/details/8609360



linux select 多路复用机制

分类: Socket/multi Thread UNIX/LINUX C/C++   5749人阅读  评论(3)  收藏  举报

目录(?)[+]

函数作用:

系统提供select函数来实现多路复用输入/输出模型。select系统调用是用来让我们的程序监视多个文件句柄的状态变化的。程序会停在select这里等待,直到被监视的文件句柄有一个或多个发生了状态改变。关于文件句柄,其实就是一个整数,我们最熟悉的句柄是0、1、2三个,0是标准输入,1是标准输出,2是标准错误输出。0、1、2是整数表示的,对应的FILE *结构的表示就是stdin、stdout、stderr。

函数原型:

  1. int select(int maxfd,fd_set *rdset,fd_set *wrset, \  
  2.            fd_set *exset,struct timeval *timeout);  

参数说明:

参数maxfd是需要监视的最大的文件描述符值+1;rdset,wrset,exset分别对应于需要检测的可读文件描述符的集合,可写文件描述符的集 合及异常文件描述符的集合。struct timeval结构用于描述一段时间长度,如果在这个时间内,需要监视的描述符没有事件发生则函数返回,返回值为0。

下面的宏提供了处理这三种描述词组的方式:
FD_CLR(inr fd,fd_set* set);用来清除描述词组set中相关fd 的位
FD_ISSET(int fd,fd_set *set);用来测试描述词组set中相关fd 的位是否为真
FD_SET(int fd,fd_set*set);用来设置描述词组set中相关fd的位
FD_ZERO(fd_set *set);用来清除描述词组set的全部位

参数timeout为结构timeval,用来设置select()的等待时间,其结构定义如下:

  1. struct timeval  
  2. {  
  3.     time_t tv_sec;//second  
  4.     time_t tv_usec;//minisecond  
  5. };  

如果参数timeout设为:

NULL,则表示select()没有timeout,select将一直被阻塞,直到某个文件描述符上发生了事件。

0:仅检测描述符集合的状态,然后立即返回,并不等待外部事件的发生。

特定的时间值:如果在指定的时间段里没有事件发生,select将超时返回。

函数返回值:

执行成功则返回文件描述词状态已改变的个数,如果返回0代表在描述词状态改变前已超过timeout时间,没有返回;当有错误发生时则返回-1,错误原因存于errno,此时参数readfds,writefds,exceptfds和timeout的值变成不可预测。错误值可能为:
EBADF 文件描述词为无效的或该文件已关闭
EINTR 此调用被信号所中断
EINVAL 参数n 为负值。
ENOMEM 核心内存不足

常见的程序片段如下:

fs_set readset;
FD_ZERO(&readset);
FD_SET(fd,&readset);
select(fd+1,&readset,NULL,NULL,NULL);
if(FD_ISSET(fd,readset){……}

理解select模型:

理解select模型的关键在于理解fd_set,为说明方便,取fd_set长度为1字节,fd_set中的每一bit可以对应一个文件描述符fd。则1字节长的fd_set最大可以对应8个fd。

(1)执行fd_set set; FD_ZERO(&set);则set用位表示是0000,0000。

(2)若fd=5,执行FD_SET(fd,&set);后set变为0001,0000(第5位置为1)

(3)若再加入fd=2,fd=1,则set变为0001,0011

(4)执行select(6,&set,0,0,0)阻塞等待

(5)若fd=1,fd=2上都发生可读事件,则select返回,此时set变为0000,0011。注意:没有事件发生的fd=5被清空。

 基于上面的讨论,可以轻松得出select模型的特点:

  (1)可监控的文件描述符个数取决与sizeof(fd_set)的值。我这边服务 器上sizeof(fd_set)=512,每bit表示一个文件描述符,则我服务器上支持的最大文件描述符是512*8=4096。据说可调,另有说虽 然可调,但调整上限受于编译内核时的变量值。本人对调整fd_set的大小不太感兴趣,参考http://www.cppblog.com /CppExplore/archive/2008/03/21/45061.html中的模型2(1)可以有效突破select可监控的文件描述符上 限。

  (2)将fd加入select监控集的同时,还要再使用一个数据结构array保存放到select监控集中的fd,一是用于再select 返回后,array作为源数据和fd_set进行FD_ISSET判断。二是select返回后会把以前加入的但并无事件发生的fd清空,则每次开始 select前都要重新从array取得fd逐一加入(FD_ZERO最先),扫描array的同时取得fd最大值maxfd,用于select的第一个 参数。

  (3)可见select模型必须在select前循环array(加fd,取maxfd),select返回后循环array(FD_ISSET判断是否有时间发生)。

下面给一个伪码说明基本select模型的服务器模型:

  1. array[slect_len];  
  2.  nSock=0;  
  3.  array[nSock++]=listen_fd;(之前listen port已绑定并listen)  
  4.  maxfd=listen_fd;  
  5.   
  6.  while(1){  
  7.   
  8.   FD_ZERO(&set);  
  9.   
  10.   foreach (fd in array)  
  11.   {  
  12.       fd大于maxfd,则maxfd=fd  
  13.       FD_SET(fd,&set)  
  14.   }  
  15.   
  16.   res=select(maxfd+1,&set,0,0,0);  
  17.   
  18.   if(FD_ISSET(listen_fd,&set))  
  19.   {  
  20.       newfd=accept(listen_fd);  
  21.       array[nsock++]=newfd;  
  22.       if(--res<=0) continue;  
  23.   }  
  24.   
  25.   foreach 下标1开始 (fd in array)  
  26.   {  
  27.       if(FD_ISSET(fd,&tyle="COLOR: #ff0000">set))  
  28.       执行读等相关操作  
  29.       如果错误或者关闭,则要删除该fd,将array中相应位置和最后一个元素互换就好,nsock减一  
  30.       if(--res<=0) continue;  
  31.   }  
  32.   
  33.  }  
检测键盘有无输入,完整的程序如下:

  1. #include<sys/time.h>  
  2. #include<sys/types.h>  
  3. #include<unistd.h>  
  4. #include<string.h>  
  5. #include<stdlib.h>  
  6. #include<stdio.h>  
  7. int main()  
  8. {  
  9.         char buf[10]="";  
  10.         fd_set rdfds;  
  11.         struct timeval tv;  
  12.         int ret;  
  13.         FD_ZERO(&rdfds);  
  14.         FD_SET(0,&rdfds);   //文件描述符0表示stdin键盘输入  
  15.         tv.tv_sec = 3;  
  16.         tv.tv_usec = 500;  
  17.         ret = select(1,&rdfds,NULL,NULL,&tv);  
  18.         if(ret<0)  
  19.               printf("\n selcet");  
  20.         else if(ret == 0)  
  21.               printf("\n timeout");  
  22.         else  
  23.               printf("\n ret = %d",ret);  
  24.   
  25.         if(FD_ISSET(1,&rdfds))  //如果有输入,从stdin中获取输入字符  
  26.         {  
  27.               printf("\n reading");  
  28.               fread(buf,9,1,stdin);  
  29.          }  
  30.          write(1,buf,strlen(buf));  
  31.          printf("\n %d \n",strlen(buf));  
  32.          return 0;  
  33. }  
  34. //执行结果ret = 1.  

利用Select模型,设计的web服务器:

  1. #include <stdio.h>  
  2. #include <stdlib.h>  
  3. #include <unistd.h>  
  4. #include <errno.h>  
  5. #include <string.h>  
  6. #include <sys/types.h>  
  7. #include <sys/socket.h>  
  8. #include <netinet/in.h>  
  9. #include <arpa/inet.h>  
  10.   
  11. #define MYPORT 88960    // the port users will be connecting to  
  12.   
  13. #define BACKLOG 10     // how many pending connections queue will hold  
  14.   
  15. #define BUF_SIZE 200  
  16.   
  17. int fd_A[BACKLOG];    // accepted connection fd  
  18. int conn_amount;    // current connection amount  
  19.   
  20. void showclient()  
  21. {  
  22.     int i;  
  23.     printf("client amount: %d\n", conn_amount);  
  24.     for (i = 0; i < BACKLOG; i++) {  
  25.         printf("[%d]:%d  ", i, fd_A[i]);  
  26.     }  
  27.     printf("\n\n");  
  28. }  
  29.   
  30. int main(void)  
  31. {  
  32.     int sock_fd, new_fd;  // listen on sock_fd, new connection on new_fd  
  33.     struct sockaddr_in server_addr;    // server address information  
  34.     struct sockaddr_in client_addr; // connector's address information  
  35.     socklen_t sin_size;  
  36.     int yes = 1;  
  37.     char buf[BUF_SIZE];  
  38.     int ret;  
  39.     int i;  
  40.   
  41.     if ((sock_fd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {  
  42.         perror("socket");  
  43.         exit(1);  
  44.     }  
  45.   
  46.     if (setsockopt(sock_fd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int)) == -1) {  
  47.         perror("setsockopt");  
  48.         exit(1);  
  49.     }  
  50.       
  51.     server_addr.sin_family = AF_INET;         // host byte order  
  52.     server_addr.sin_port = htons(MYPORT);     // short, network byte order  
  53.     server_addr.sin_addr.s_addr = INADDR_ANY; // automatically fill with my IP  
  54.     memset(server_addr.sin_zero, '\0'sizeof(server_addr.sin_zero));  
  55.   
  56.     if (bind(sock_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1) {  
  57.         perror("bind");  
  58.         exit(1);  
  59.     }  
  60.   
  61.     if (listen(sock_fd, BACKLOG) == -1) {  
  62.         perror("listen");  
  63.         exit(1);  
  64.     }  
  65.   
  66.     printf("listen port %d\n", MYPORT);  
  67.   
  68.     fd_set fdsr;  
  69.     int maxsock;  
  70.     struct timeval tv;  
  71.   
  72.     conn_amount = 0;  
  73.     sin_size = sizeof(client_addr);  
  74.     maxsock = sock_fd;  
  75.     while (1) {  
  76.         // initialize file descriptor set  
  77.         FD_ZERO(&fdsr);  
  78.         FD_SET(sock_fd, &fdsr);  
  79.   
  80.         // timeout setting  
  81.         tv.tv_sec = 30;  
  82.         tv.tv_usec = 0;  
  83.   
  84.         // add active connection to fd set  
  85.         for (i = 0; i < BACKLOG; i++) {  
  86.             if (fd_A[i] != 0) {  
  87.                 FD_SET(fd_A[i], &fdsr);  
  88.             }  
  89.         }  
  90.   
  91.         ret = select(maxsock + 1, &fdsr, NULL, NULL, &tv);  
  92.         if (ret < 0) {  
  93.             perror("select");  
  94.             break;  
  95.         } else if (ret == 0) {  
  96.             printf("timeout\n");  
  97.             continue;  
  98.         }  
  99.   
  100.         // check every fd in the set  
  101.         for (i = 0; i < conn_amount; i++) {  
  102.             if (FD_ISSET(fd_A[i], &fdsr)) {  
  103.                 ret = recv(fd_A[i], buf, sizeof(buf), 0);  
  104.                   
  105.                 char str[] = "Good,very nice!\n";  
  106.                   
  107.                 send(fd_A[i],str,sizeof(str) + 1, 0);  
  108.                   
  109.                   
  110.                 if (ret <= 0) {        // client close  
  111.                     printf("client[%d] close\n", i);  
  112.                     close(fd_A[i]);  
  113.                     FD_CLR(fd_A[i], &fdsr);  
  114.                     fd_A[i] = 0;  
  115.                 } else {        // receive data  
  116.                     if (ret < BUF_SIZE)  
  117.                         memset(&buf[ret], '\0', 1);  
  118.                     printf("client[%d] send:%s\n", i, buf);  
  119.                 }  
  120.             }  
  121.         }  
  122.   
  123.         // check whether a new connection comes  
  124.         if (FD_ISSET(sock_fd, &fdsr)) {  
  125.             new_fd = accept(sock_fd, (struct sockaddr *)&client_addr, &sin_size);  
  126.             if (new_fd <= 0) {  
  127.                 perror("accept");  
  128.                 continue;  
  129.             }  
  130.   
  131.             // add to fd queue  
  132.             if (conn_amount < BACKLOG) {  
  133.                 fd_A[conn_amount++] = new_fd;  
  134.                 printf("new connection client[%d] %s:%d\n", conn_amount,  
  135.                         inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));  
  136.                 if (new_fd > maxsock)  
  137.                     maxsock = new_fd;  
  138.             }  
  139.             else {  
  140.                 printf("max connections arrive, exit\n");  
  141.                 send(new_fd, "bye", 4, 0);  
  142.                 close(new_fd);  
  143.                 break;  
  144.             }  
  145.         }  
  146.         showclient();  
  147.     }  
  148.   
  149.     // close other connections  
  150.     for (i = 0; i < BACKLOG; i++) {  
  151.         if (fd_A[i] != 0) {  
  152.             close(fd_A[i]);  
  153.         }  
  154.     }  
  155.   
  156.     exit(0);  
  157. }  
补充部分:

1 基本原理

Resize icon

注:select 原理图,摘自 IBM iSeries 信息中心

1 数据结构与函数原型

1.1 select

  • 函数原型
    1. int select(
    2. int nfds,
    3. fd_set *readset,
    4. fd_set *writeset,
    5. fd_set* exceptset,
    6. struct timeval *timeout
    7. );
  • 头文件
    • select位于:
      #include <sys/select.h>
      
    • struct timeval位于:
      #include <sys/time.h>
      
  • 返回值

    返回对应位仍然为1的fd的总数。

  • 参数
    • nfds:第一个参数是:最大的文件描述符值+1;
    • readset:可读描述符集合;
    • writeset:可写描述符集合;
    • exceptset:异常描述符;
    • timeout:select 的监听时长,如果这短时间内所监听的 socket 没有事件发生。

1.2 fd_set

1.2.1 清空描述符集合
FD_ZERO(fd_set *)
1.2.2 向描述符集合添加指定描述符
FD_SET(int, fd_set *)
1.2.3 从描述符集合删除指定描述符
FD_CLR(int, fd_set *)
1.2.4 检测指定描述符是否在描述符集合中
FD_ISSET(int, fd_set *)
1.2.5 描述符最大数量
#define FD_SETSIZE 1024

1.3 描述符集合

可读描述符集合中可读的描述符,为1,其他为0;可写也类似。异常描述符集合中有异常等待处理的描述符的值为1,其他为0。

1.4 ioctl

  • 函数原型:

      int ioctl(int handle, int cmd,[int *argdx, int argcx]);
    
  • 头文件:

      #include <sys/ioctl.h>
    
  • 返回值:

    • 0 - 成功
    • 1 - 失败

2 示例

程序各部分的解释在注释中。

  1. #include <sys/socket.h>
  2. #include <string.h>
  3. #include <sys/time.h>
  4. #include <netinet/in.h>
  5. #include <sys/ioctl.h>
  6. #include <stdlib.h>
  7. #include <errno.h>
  8. #include <stdio.h>
  9. #include <unistd.h>
  10. #define TRUE 1
  11. #define FALSE 0
  12. int main(int argc, char *argv[])
  13. {
  14. int i, len, rc, on = TRUE;
  15. int listen_sd, new_sd = 0, max_sd;
  16. int desc_ready;
  17. char buffer[80];
  18. int close_conn, end_server = FALSE;
  19. struct sockaddr_in server_addr;
  20. struct timeval timeout;
  21. struct fd_set master_set, working_set;
  22. // Listen
  23. listen_sd = socket(AF_INET, SOCK_STREAM, 0);
  24. if (listen_sd < 0)
  25. {
  26. perror("socket() failed");
  27. exit(-1);
  28. }
  29. // Set socket options
  30. rc = setsockopt(listen_sd, SOL_SOCKET, SO_REUSEADDR, (char *) &on, sizeof(on));
  31. if (rc < 0)
  32. {
  33. perror("setsockopt() failed");
  34. close(listen_sd);
  35. exit(-1);
  36. }
  37. // Set IO control
  38. rc = ioctl(listen_sd, FIONBIO, (char *) &on);
  39. if (rc < 0)
  40. {
  41. perror("ioctl() failed");
  42. close(listen_sd);
  43. exit(-1);
  44. }
  45. // Bind
  46. memset(&server_addr, 0, sizeof(server_addr));
  47. server_addr.sin_family = AF_INET;
  48. server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
  49. server_addr.sin_port = htons(atoi(argv[1]));
  50. rc = bind(listen_sd, (struct sockaddr *) &server_addr, sizeof(server_addr));
  51. if (rc < 0)
  52. {
  53. perror("bind() failed\n");
  54. close(listen_sd);
  55. exit(-1);
  56. }
  57. // Listen
  58. rc = listen(listen_sd, 32);
  59. if (rc < 0)
  60. {
  61. perror("listen() failed\n");
  62. close(listen_sd);
  63. exit(-1);
  64. }
  65. // Intialize sd set
  66. FD_ZERO(&master_set);
  67. max_sd = listen_sd;
  68. FD_SET(listen_sd, &master_set);
  69. timeout.tv_sec = 3 * 60;
  70. timeout.tv_usec = 0;
  71. // Start
  72. do
  73. {
  74. // Copy master_set into working_set
  75. memcpy(&working_set, &master_set, sizeof(master_set));
  76. printf("Waiting on select()...\n");
  77. rc = select(max_sd + 1, &working_set, NULL, NULL, &timeout);
  78. if (rc < 0)
  79. {
  80. perror(" select() failed\n");
  81. break;
  82. }
  83. if (rc == 0)
  84. {
  85. printf(" select() timed out. End program.\n");
  86. break;
  87. }
  88. desc_ready = rc; // number of sds ready in working_set
  89. // Check each sd in working_set
  90. for (i = 0; i <= max_sd && desc_ready > 0; ++i)
  91. {
  92. // Check to see if this sd is ready
  93. if (FD_ISSET(i, &working_set))
  94. {
  95. --desc_ready;
  96. // Check to see if this is the listening sd
  97. if (i == listen_sd)
  98. {
  99. printf(" Listeing socket is readable\n");
  100. do
  101. {
  102. // Accept
  103. new_sd = accept(listen_sd, NULL, NULL);
  104. // Nothing to be accepted
  105. if (new_sd < 0)
  106. {
  107. // All have been accepted
  108. if (errno != EWOULDBLOCK)
  109. {
  110. perror(" accept() failed\n");
  111. end_server = TRUE;
  112. }
  113. break;
  114. }
  115. // Insert new_sd into master_set
  116. printf(" New incoming connection - %d\n", new_sd);
  117. FD_SET(new_sd, &master_set);
  118. if (new_sd > max_sd)
  119. {
  120. max_sd = new_sd;
  121. }
  122. }
  123. while (new_sd != -1);
  124. }
  125. // This is not the listening sd
  126. else
  127. {
  128. close_conn = FALSE;
  129. printf(" Descriptor %d is avaliable\n", i);
  130. do
  131. {
  132. rc = recv(i, buffer, sizeof(buffer), 0);
  133. // Receive data on sd "i", until failure occurs
  134. if (rc < 0)
  135. {
  136. // Normal failure
  137. if (errno != EWOULDBLOCK)
  138. {
  139. perror(" recv() failed\n");
  140. close_conn = TRUE;
  141. }
  142. break;
  143. }
  144. // The connection has been closed by the client
  145. if (rc == 0)
  146. {
  147. printf(" Connection closed\n");
  148. close_conn = TRUE;
  149. break;
  150. }
  151. /* Receiving data succeeded and echo it back
  152. the to client */
  153. len = rc;
  154. printf(" %d bytes received\n", len);
  155. rc = send(i, buffer, len, 0);
  156. if (rc < 0)
  157. {
  158. perror(" send() failed");
  159. close_conn = TRUE;
  160. break;
  161. }
  162. }
  163. while (TRUE);
  164. // If unknown failure occured
  165. if (close_conn)
  166. {
  167. // Close the sd and remove it from master_set
  168. close(i);
  169. FD_CLR(i, &master_set);
  170. // If this is the max sd
  171. if (i == max_sd)
  172. {
  173. // Find the max sd in master_set now
  174. while (FD_ISSET(max_sd, &master_set) == FALSE)
  175. {
  176. --max_sd;
  177. }
  178. } // End of if (i == max_sd)
  179. } // End of if (close_conn)
  180. }
  181. }
  182. }
  183. }
  184. while (end_server == FALSE);
  185. /* Close each sd in master_set */
  186. for (i = 0; i < max_sd; ++i)
  187. {
  188. if (FD_ISSET(i, &master_set))
  189. {
  190. close(i);
  191. }
  192. }
  193. return 0;
  194. }





声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/weixin_40725706/article/detail/225067
推荐阅读
相关标签
  

闽ICP备14008679号