Chinaunix首页 | 论坛 | 博客
  • 博客访问: 1557418
  • 博文数量: 347
  • 博客积分: 9328
  • 博客等级: 中将
  • 技术积分: 2680
  • 用 户 组: 普通用户
  • 注册时间: 2010-07-29 23:45
文章分类

全部博文(347)

文章存档

2016年(1)

2013年(4)

2012年(207)

2011年(85)

2010年(50)

分类: C/C++

2012-11-17 22:05:59

原文:

select()函数主要是建立在fd_set类型的基础上的。fd_set(它比较重要所以先介绍一下)是一组文件描述字(fd)的集合,它用一位来表示一个fd(下面会仔细介绍),对于fd_set类型通过下面四个宏来操作:  
    fd_set set; 
    FD_ZERO(&set);       /* 将set清零使集合中不含任何fd*/ 
    FD_SET(fd, &set);    /* 将fd加入set集合 */ 
    FD_CLR(fd, &set);    /* 将fd从set集合中清除 */ 
    FD_ISSET(fd, &set);  /* 测试fd是否在set集合中*/       
过去,一个fd_set通常只能包含中定义常量FD_SETSIZE,它是数
据类型fd_set的描述字数量,其值通常是1024,这样就能表示    fd_set set; 
FD_ZERO(&set);      /*将set的所有位置0,如set在内存中占8位则将set置为 
00000000*/ 
FD_SET(0, &set);    /* 将set的第0位置1,如set原来是00000000,则现在变为10000000,这样fd==1的文件描述字就被加进set中了 */ 
FD_CLR(4, &set);    /*将set的第4位置0,如set原来是10001000,则现在变为10000000,这样fd==4的文件描述字就被从set中清除了 */  
FD_ISSET(5, &set);  /* 测试set的第5位是否为1,如果set原来是10000100,则返回非零,表明fd==5的文件描述字在set中;否则返回0*/  
――――――――――――――――――――――――――――――――――――――― 
注意fd的最大值必须―――――――――――――――――――――――――――――――――――――――  
select函数的接口比较简单: 
    int select(int nfds, fd_set *readset, fd_set *writeset, 
fd_set* exceptset, struct timeval *timeout);  
功能: 
测试指定的fd可读?可写?有异常条件待处理?      
参数: 
nfds     
需要检查的文件描述字个数(即检查到fd_set的第几位),数值应该比三组fd_set中
所含的最大fd值更大,一般设为三组fd_set中所含的最大fd值加1(如在readset,writeset,exceptset中所含最大的fd为
5,则nfds=6,因为fd是从0开始的)。设这个值是为提高效率,使函数不必检查fd_set的所有1024位。 
readset    
     用来检查可读性的一组文件描述字。 
writeset 
     用来检查可写性的一组文件描述字。 
exceptset 
     用来检查是否有异常条件出现的文件描述字。(注:错误不包括在异常条件之内) 
timeout 
有三种可能: 
1.        timeout=NULL(阻塞:直到有一个fd位被置为1函数才返回) 
2.        timeout所指向的结构设为非零时间(等待固定时间:有一个fd位被置为1或者时间耗尽,函数均返回) 
3.        timeout所指向的结构,时间设为0(非阻塞:函数检查完每个fd后立即返回)  
返回值:      
返回对应位仍然为1的fd的总数。  
Remarks: 
三组fd_set均将某些fd位置0,只有那些可读,可写以及有异常条件待处理的fd位仍然为1。 
使用select函数的过程一般是: 
先调用宏FD_ZERO将指定的fd_set清零,然后调用宏FD_SET将需要测试的fd加入fd_set,接着调用函数select测试fd_set中的所有fd,最后用宏FD_ISSET检查某个fd在函数select调用后,相应位是否仍然为1。  
以下是一个测试单个文件描述字可读性的例子: 

  1. int isready(int fd)
  2. {
  3.     int rc;
  4.     fd_set fds;
  5.     struct timeval tv;
  6.     FD_ZERO(&fds);
  7.     FD_SET(fd,&fds);
  8.     tv.tv_sec = tv.tv_usec = 0;
  9.     rc = select(fd+1, &fds, NULL, NULL, &tv);
  10.     if (rc)
  11.         return -1;
  12.     return FD_ISSET(fd,&fds) ? 1 : 0;
  13. }
      
下面还有一个复杂一些的应用: 
//这段代码将指定测试Socket的描述字的可读可写性,因为Socket使用的也是fd 
  1. uint32 SocketWait(TSocket *s,bool rd,bool wr,uint32 timems)
  2. {
  3.      fd_set rfds,wfds;
  4. #ifdef _WIN32
  5.      TIMEVAL tv;
  6. #else
  7.      struct timeval tv;
  8. #endif /* _WIN32 */
  9.      FD_ZERO(&rfds);
  10.      FD_ZERO(&wfds);
  11.      if (rd) //TRUE
  12.           FD_SET(*s,&rfds); //添加要测试的描述字
  13.      if (wr) //FALSE
  14.           FD_SET(*s,&wfds);
  15.      tv.tv_sec=timems/1000; //second
  16.      tv.tv_usec=timems%1000; //ms
  17.      for (;;) //如果errno==EINTR,反复测试缓冲区的可读性
  18.           switch(select((*s)+1,&rfds,&wfds,NULL,
  19.               (timems==TIME_INFINITE?NULL:&tv))) //测试在规定的时间内套接口接收缓冲区中是否有数据可读
  20.          { //0--超时,-1--出错
  21.          case 0: /* time out */
  22.               return 0;
  23.          case (-1): /* socket error */
  24.               if (SocketError()==EINTR)
  25.                    break;
  26.               return 0; //有错但不是EINTR
  27.           default:
  28.               if (FD_ISSET(*s,&rfds)) //如果s是fds中的一员返回非0,否则返回0
  29.                    return 1;
  30.               if (FD_ISSET(*s,&wfds))
  31.                    return 2;
  32.               return 0;
  33.          };
  34. }

阅读(1342) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~