socket 通信之select

　　对于socket 通信，大家很多都用的单线程通信。同时只能监听一个端口,只能响应一个服务，select的方式可以解决多个socket 被连接的问题。一次可以分配多个资源，只要一个连接便可以进行通信。在网络已经有很多的select 的例子。不过很多例子没有真正体现到select的精妙之处。此函数主要是对多个文件描述符进行监听，直到某一个或者多个被连接。

　　首先我们介绍fd_set这个结构：

　　fd_set可以理解为一个集合，这个集合中存放的是文件描述符(file descriptor)，即文件句柄，这可以是我们所说的普通意义的文件，当然Unix下任何设备、管道、FIFO等都是文件形式，全部包括在内，所以毫无疑问一个socket就是一个文件，socket句柄就是一个文件描述符。fd_set集合可以通过一些宏由人为来操作，比如清空集合 FD_ZERO(fd_set *)，将一个给定的文件描述符加入集合之中FD_SET(int ,fd_set *)，将一个给定的文件描述符从集合中删除FD_CLR(int ,fd_set*)。

　　在select使用这个结构之前，我们需要调用FD_SET,设置对应socket的标志位，网络生很多的例子错误就在此，这里必须要绑定多个socket才是真正体会了select的多fd监听。而且不能随意指定，比如FD_SET(0,XX），这样的fd实际已经被系统占用了。在select成功返回后检查集合中指定的文件描述符是否可以读写FD_ISSET(int ,fd_set* )。进而根据对应的fd进行读写操作。不多说，直接粘贴代码：

 #include <iostream>
 #include <sys/times.h>
 #include <sys/types.h>
 #include <unistd.h>
 #include <sys/socket.h>
 #include <sys/select.h>
 #include <cstdlib>
 #include <cstdio>
 #include <cstring>
 #include <string>
 #include <signal.h>
 #include <netinet/in.h>
 #include <arpa/inet.h>
 #include <errno.h>
 using namespace std;

 #define max(a,b) ((a)>(b)?(a):(b))

 static int listen_socket(int port)
 {
     sockaddr_in s_in;
     int s;
     int yes;
     if((s = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == -)
     {
         cout<<"socket error"<<strerror(errno)<<endl;
         return -;
     }
     yes = ;
     if(setsockopt(s, SOL_SOCKET, SO_REUSEADDR, (char*)&yes,sizeof(yes)) == -)
     {
         cout<<"setsockopt error"<<strerror(errno)<<endl;
         close(s);
         return -;
     }

     memset(&s_in, , sizeof(s_in));
     s_in.sin_port = htons(port);
     s_in.sin_family =  AF_INET;
     s_in.sin_addr.s_addr = inet_addr("127.0.0.1");
     if(bind(s,(sockaddr*)&s_in, sizeof(s_in)) == -)
     {
         cout<<"bind error"<<strerror(errno)<<endl;
         close(s);
         return -;
     }
     listen(s,);
     cout<<"socket -> setsockopt-> bind->listen"<<port<<endl;
     return s;
 }

 int main(int argi, char* args[])
 {
     int h, h1;
     int fd1 = -, fd2 = -;

     if(argi !=  )
     {
     cout<<"Usage server listen_port1 listen_port2"<<endl;
         exit(-);

     }

     int listen_port1 = atoi(args[]);
     fd1 = listen_socket(listen_port1);
     int listen_port2 = atoi(args[]);
     fd2 = listen_socket(listen_port2);//这里是建立了两个socket：cyjwdm0503
     if(fd1==- || fd2==-)
     {
         exit(-);
     }
     for(;;)
     {
         int r, nfds = ;
         fd_set rfd,sfd;

         FD_ZERO(&rfd);
         FD_ZERO(&sfd);
         FD_SET(fd1, &rfd);//绑定对应的两个socket
         FD_SET(fd2,&rfd);

         int maxnum = max(, fd1);
          maxnum = max(fd2, maxnum);
         cout<<"maxnum:"<<maxnum<<"\t"<<h<<endl;

         r = select(maxnum+, &rfd, &sfd, NULL, NULL);//select的参赛为rfd，最大值为socket+1
         if( r == - && errno == EINTR)
         {
             cout<<strerror(errno)<<endl;
                 continue;
         }
         if( r == -)
         {
             cout<<"select error"<<strerror(errno)<<"\t"<<errno<<endl;
             return -;
         }

         //        for(int index=1; index<=5; index++)
         {
         int select_fd = ;
         if(FD_ISSET(fd1,&rfd))
             select_fd = fd1;
         else if(FD_ISSET(fd2,&rfd))
             select_fd = fd2;
         else
             select_fd = -;
         if(select_fd != -)
         {
                 cout<<"select retrun fd"<<select_fd<<endl;
             sockaddr_in accept_addr;
             socklen_t le = sizeof(accept_addr);//注意在accept时候的sockaddr_in 长度要初始化
             int acc = accept(ddd, (sockaddr*)&accept_addr, &le);

             cout<<"accept port"<<accept_addr.sin_port<<endl;            

             if(- == acc)
             {
                 cout<<"accept error"<<strerror(errno)<<endl;
                 return -;
             }
             char buffer[] = "";
             int ret = recv(acc, buffer, sizeof(buffer),);
             if(ret <= )
             {
                 cout<<"client close:"<<strerror(errno)<<"\t"<<errno<<endl;
             }
             else
             {
                 cout<<"buffer from client:"<<buffer<<endl;
             }

         }
         else
             cout<<"select_fd = -1"<<endl;
         }
         cout<<"for over"<<endl;        

     }
 }

　　select 在监听对应的socket的rfd集合时，如果有对应的客户端链接两个socket之中的某一个，select 边能够返回对应的信息，然后调用FD_ISSET，获取对应被链接的socket ，进行accept ，收发信息。

　　下面为客户端的示例代码：

 #include <iostream>
 #include <sys/socket.h>
 #include <unistd.h>
 #include <sys/types.h>
 #include <netdb.h>
 #include <cstdio>
 #include <cstdlib>
 #include <cstring>
 #include <string>
 #include <iostream>
 #include <netinet/in.h>
 #include <arpa/inet.h>
 #include <errno.h>

 using namespace std;

 int main(int argi, char* args[])
 {
     if( argi<)
     {
         cout<<args[]<<"\t"<<"host port msg..."<<endl;
         return -;
     }
     sockaddr_in s_in;
     memset(&s_in, , sizeof(s_in));
     s_in.sin_family = AF_INET;
     s_in.sin_addr.s_addr = inet_addr("127.0.0.1");//(args[1]);
     s_in.sin_port = htons(atoi(args[]));

     cout<<"port"<<"\t"<<args[]<<endl;
     int s = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
     if( s == -)
     {
         cout<<"socket error"<<strerror(errno)<<endl;
     }

     socklen_t length  = sizeof(sockaddr_in);
     if(- == connect(s, (sockaddr*)&s_in, length))
     {
         cout<<"connect error"<<strerror(errno)<<endl;
         close(s);
         return -;
     }
     for(;;)
     {
     char buffer[];
          if(strcmp(args[2],"q")==)
         break;
     ssize_t size =    send(s, buffer, strlen(buffer),);
     cout<<"socket num:"<<s<<endl;
 　　 sprintf(buffer,"%s%s",args[2],"www.cnblogs.com/cyjwdm0503");
     size = send(s,args[],strlen(args[]),);
     if( - == size)
     {
         cout<<"write error"<<strerror(errno)<<endl;
         close(s);
         return -;
     }
     }
     close(s);

 }

虽然这里是以Linux为示例，Windows和这个也类似的，转载请注明来源地址http://www.cnblogs.com/cyjwdm0503

上面如果有问题，请最好留言给我，大家交流。