我们先来说说最重要的函数select这个函数,它的原型如下:
int select(int nfds,fd_set *readfds,fd_set *writefds,fd_set *exceptfds,struct timeval *timeout);
select函数可以执行I/O多路转接。传给select的参数告诉内核:
1.我们所关心的描述符
2.对于每个描述符我们所关心的条件(是否想从一个给定的描述符读,写,还是异常条件)
3.愿意等待的时间(永远等待,等待一个给定的时间或者根本不等待)
从select返回时,内核告诉我们:
1.已准备好的描述符的总数量
2.对于读,写,异常这3个条件中的每一个,哪些描述符已经准备好
有了上面的返回信息,我们就可以调用相应的I/O函数(read或者write),并且确知该函数不会被阻塞。
我们再来说说select这个函数的返回值和相应的参数:
正常返回准备就绪的描述符数目,若超时,返回0;出错时返回-1.
参数:
我们先来说最后一个参数,它是一个指向结构体的指针,这个结构体如下:
struct timeval
{
long tv_sec;
long tv_usec;
}
这个参数指明了愿意等待的时间长度,单位为秒和微妙。当它的值为NULL,说明要永远等待。如果捕捉到一个信号则中断无限等待。当所指定的描述符的一个已经准备好或捕捉到一个信号则返回。当这个结构体的两个成员的值均不为0时,则是我们设定的要指定的秒数和微秒数。当指定的一个描述符之一已经准备好,或者当指定的时间已经超过时立即返回。在超时后还没有一个描述符准备好,返回0。
中间的 readfds,writefds,exceptfds这三个参数是指向描述符集的指针。这3个描述符集说明了我们关心的可读,可写或处于异常条件的描述符集合。每个描述符集存储在一个fd_set数据类型中。这个数据类型为每一个描述符保持一位。在这里我们简单的认为是一个很大的字节数组。
第一个参数 nfds 的意思是“最大文件描述符编号值加1”。因为描述符从0 开始,所以要在最大描述符编号上加1.意思是我们我们在所有描述符集当中找出最大的一个,然后加1就行了。这里我们也可以将第一个参数设置为FD_SETSIZE,它是一个常量值,它指定最大描述符数(一般是1024),一般而言,此值有点大。通过我们指定最大的描述符数,内核就只需在此范围内寻找打开的位,大大节省了时间。
我们再来说说fd_set这个数据类型:
这种数据类型的变量经常用到以下函数:
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(fd_set* fdset);
这些接口可实现为宏或者函数。调用FD_ZERO将一个fd_set变量的所有位设置为0.调用FD_SET设置描述符集中的一位。调用FD_CLR可以清除一位。我们可以调用FD_ISSET测试描述符集中的一个指定位是否已经打开。我们要注意,在我们声明了一个描述符集之后,必须调用FD_ZERO函数将这个描述符集置为0,然后再在其中设置我们关心的各个描述符的位。
上面说了那么多,我们来用用这几个函数吧,在下面的程序中,我们用刚说的那几个函数写了一个简单的客户/服务器间通信模型,客户向服务端发送数据,服务端收到后再回显给客户端,我们来看看具体代码:
server端:
#include <stdio.h> #include <stdlib.h> #include <sys/types.h> #include <assert.h> #include <unistd.h> #include <netinet/in.h> #include <arpa/inet.h> #include <sys/socket.h> #define _BACKLOG_ 5 int fds[64]; static void usage(const char* arg) { printf("usage:%s [ip][poort]",arg); } static int startup(char *ip,int port) { assert(ip); int sock=socket(AF_INET,SOCK_STREAM,0); if(sock<0) { perror("socket"); exit(1); } struct sockaddr_in local; local.sin_family=AF_INET; local.sin_port=htons(port); local.sin_addr.s_addr=inet_addr(ip); if(bind(sock,(struct sockaddr*)&local,sizeof(local))<0) { perror("bind"); exit(2); } if(listen(sock,_BACKLOG_)<0) { perror("listen"); exit(3); } return sock; } int main(int argc,char* argv[]) { if(argc!=3) { usage(argv[0]); exit(1); } int port=atoi(argv[2]); char *ip=argv[1]; int listen_sock=startup(ip,port); int done=0; int new_sock=-1; struct sockaddr_in client; socklen_t len=sizeof(client); int max_fd; fd_set reads; fd_set writs; int i=0; int fds_num=sizeof(fds)/sizeof(fds[0]); for( ;i<fds_num;++i) { fds[i]=-1; } fds[0]=listen_sock; max_fd=fds[0]; while(!done) { FD_ZERO(&reads); FD_ZERO(&writs); FD_SET(listen_sock,&reads); struct timeval timeout={5,0}; for(i=1;i<fds_num;++i) { if(fds[i]>0) { FD_SET(fds[i],&reads); if(fds[i]>max_fd) { max_fd=fds[i]; } } } switch(select(max_fd+1,&reads,&writs,NULL,&timeout)) { case 0://超时 printf("select timeout\n"); break; case -1: perror("select"); break; default: { i=0; for(;i<fds_num;++i) { if(fds[i]==listen_sock&&FD_ISSET(fds[i],&reads)) {//监听套接字就绪 new_sock=accept(listen_sock,(struct sockaddr*)&client,&len); if(new_sock<0) { perror("accept"); continue; } printf("get a new connect...%d\n",new_sock); for(i=0;i<fds_num;++i) { if(fds[i]==-1) { fds[i]=new_sock; break; } } if(i==fds_num) { close(new_sock); } }//listen sock else if(fds[i]>0&&FD_ISSET(fds[i],&reads)) { char buf[1024]; ssize_t _s=read(fds[i],buf,sizeof(buf)-1); if(_s>0) { buf[_s]='\0'; printf("client : %s",buf); write(fds[i],buf,_s); } else if(_s==0) { printf("client quit..\n"); close(fds[i]); fds[i]=-1; } else{ } }//nomal socket else{ } } } break; } } return 0; }
下面是client端:
#include <stdio.h> #include <errno.h> #include <stdlib.h> #include <string.h> #include <netinet/in.h> #include <arpa/inet.h> #include <sys/types.h> #include <sys/socket.h> void usage(const char* arg) { printf("%s [remote_ip][remote_port\n",arg); } int main(int argc,char *argv[]) { if(argc != 3) { usage(argv[0]); exit(0); } int port=atoi(argv[2]); char *ip=argv[1]; int sock=socket(AF_INET,SOCK_STREAM,0); if(sock<0) { perror("socket"); exit(1); } struct sockaddr_in remote; remote.sin_family=AF_INET; remote.sin_port=htons(port); remote.sin_addr.s_addr=inet_addr(ip); int ret=connect(sock,(struct sockaddr*)&remote,sizeof(remote)); char buf[1024]; while(1) { printf("please enter: "); fflush(stdout); ssize_t _s=read(0,buf,sizeof(buf)-1); buf[_s]='\0'; write(sock,buf,sizeof(buf)-1); _s=read(sock,buf,sizeof(buf)-1); //回显 if(_s>0) { buf[_s]='\0'; printf("server-->client:%s",buf); } } return 0; }
上面的结果如下:
从上面的结果我我们可以看到,从client端发送的数据,最后被server端hui显给client端了,达到了我们的目的。
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。