select用法详解 http://blog.chinaunix.net/uid-23373524-id-2426940.html

  select用法详解  2010-04-09 21:06:49

分类： LINUX

今天弄了下网络编程，为了让套接字不阻塞采用了select的方法。下面结合unix环境高级编程及自己实际使用时遇到的问题解释下select用法。

 

#include

 

int select(int maxfdp1,fd_set *readfds,fd_set *writefds,fd_set *exceptfds,struct timeval *timeout);

 

先说最后一个参数，它指定愿意等待的时间。

struct timeval

{

     long tv_sec;

     long tv_usec;

};

有3种情况:

timeout == NULL

永远等待。如果捕捉到一个信号则中断此无限期等待。当所指定的描述符中的一个已准备好或捕捉到一个信号则返回。如果捕捉到一个信号，则select返回-1，errno设置为EINTR.

 

timeout->tv_sec = 0 && timeout->tv_usec = 0

完全不等待。测试所有指定的描述符并立即返回。这是得到多个描述符的状态而不阻塞select函数的轮询方法。

 

timeout->tv_sec != 0 || timeout->tv_usec != 0

 

等待指定的秒数和微秒数。

 

注意：

（1）千万不要混淆了timeout == NULL与timeout->tv_sec = 0 && timeout->tv_usec = 0这2种情况，结果截然不同。

（2）timeout->tv_sec != 0 || timeout->tv_usec != 0这种情况下，超时过后，就变成了

 timeout->tv_sec = 0 && timeout->tv_usec = 0。

 

中间的三个参数readfds,writefds和exceptfds是指向描述符集的指针。这三个描述符集说明了我们关心的可读、可写或处于异常条件的各个描述符。每个描述符放在一个fd_set数据类型中。这种数据类型为每一可能的描述符保持了一位。

 

对fd_set数据类型可以进行处理的是：分配一个这种类型的变量；将这种类型的一个变量赋予同类型的另一个变量；或对于这种类型的变量使用下列四个函数中的一个。

 

#include 

int  FD_ISSET(int fd,fd_set *fdset);

void FD_CLR(int fd,fd_set *fdset);

void FD_SET(int fd,fd_set *fdset);

void FD_ZERO(int fd,fd_set *fdset);

 

这些接口可实现为宏或函数。具体意见就不多说了。

 

int select(int maxfdp1,fd_set *readfds,fd_set *writefds,fd_set *exceptfds,struct timeval *timeout);

 

因为描述符编号从0开始，所以要在最大描述符编号值上加1.第一个参数实际上是要检查的描述符数（从描述符0开始）

select有三个可能的返回值。

（1）返回值-1表示出错。出错是有可能的，例如在所指定的描述符都没有准备好时捕捉到一个信号。在此种情况下，将不修改其中任何描述符集。

（2）返回值0表示没有描述符准备好。若指定的描述符都没有准备好，而且指定的时间已经超过，则发生此种情况。此时， 所有描述符集皆被清0.

（3）正返回值表示已经准备好的描述符数。

 

注意红色部分，超时后，每次描述符集都需要重新设置。

 

#include 
#include 
#include 
#include 
#include 
#include

int main(int argc,char *argv[])
{
 int listen_fd;
 int client_fd;
 socklen_t clt_len;
 struct sockaddr_in srv_addr;
 struct sockaddr_in clt_addr;
 int port;
 int ret;
 int len;
 int num;
 char recv_buf[1024];
 fd_set read_fds;
 struct timeval wait_time;

 if(argc != 2)
 {
  printf("Usage: %s port_name\n",argv[0]);
  return 1;
 }
 
 port = atoi(argv[1]);

 listen_fd = socket(PF_INET,SOCK_STREAM,0);
 if(listen_fd < 0){
  perror("cannot create socket");
  return 1;
 }

 memset(&srv_addr,0,sizeof(srv_addr));
 srv_addr.sin_family=AF_INET;
 srv_addr.sin_addr.s_addr=htonl(INADDR_ANY);
 srv_addr.sin_port=htons(port);

 ret = bind(listen_fd,(struct sockaddr*)&srv_addr,sizeof(srv_addr));
 if(ret < 0)
 {
  perror("cannot bind the socket");
  return 1;
 }

 ret = listen(listen_fd,1);
 if(ret == -1)
 {
  perror("cannot listen the client connect request");
  close(listen_fd);
  return 1;
 }

    wait_time.tv_sec=0;
 wait_time.tv_usec=0;
 FD_ZERO(&read_fds);
 FD_SET(listen_fd,&read_fds);

  
 while(1)
 { 
     FD_ZERO(&read_fds);
     FD_SET(listen_fd,&read_fds);
   ret = select(listen_fd + 1,&read_fds,NULL,NULL,&wait_time);   
        if(ret > 0)
        {        
            if(FD_ISSET(listen_fd,&read_fds) > 0)
            {                 
                len = sizeof(clt_addr);
                client_fd = accept(listen_fd,(struct sockaddr*)&clt_addr,&len);
                if(client_fd < 0)
                {         
                    perror("cannot accept client connect request");
                    close(listen_fd);
                    return 1;  
                }

                while(1)
                {
                    FD_ZERO(&read_fds);
                    FD_SET(client_fd,&read_fds);    
                    if(select(client_fd + 1,&read_fds,NULL,NULL,&wait_time) > 0)
                    {                
                        if(FD_ISSET(client_fd,&read_fds) > 0)
                        {
                            len = read(client_fd,recv_buf,sizeof(recv_buf));
                            if(len > 0)
                            {
                                recv_buf[len] = 0;
                                printf("%s\n",recv_buf);
                                write(client_fd,recv_buf,len);
                            }
                            else
                            {
                                close(client_fd);
                                break;
                            }
                        }                   
                    }                   
                }           
               
           }     
                              
        } 
        
 }
}