使用Select就可以完成非阻塞方式工作的程序,它能够监视我们需要监视的文件描述符的变化情况——读写或是异常。
Select的函数格式:
int select(int maxfdp,fd_set *readfds,fd_set *writefds,fd_set *errorfds,struct timeval*timeout);
两个结构体:
fd_set:存放的是文件描述符的集合。
struct timeval:用来代表时间值,有两个成员,一个是秒数,另一个是微妙数。
参数:
int maxfdp:是指集合中所有文件描述符的范围。
fd_set *readfds:是指向fd_set结构的指针,这个集合中应该包括文件描述符,我们是要监视这些文件描述符的读变化的,即我们关心是否可以从这些文件中读取数据了,如果这个集合中有一个文件可读,select就会返回一个大于0的值,表示有文件可读,如果没有可读的文件,则根据timeout参数再判断是否超时,若超出timeout的时间,select返回0,若发生错误返回负值。可以传入NULL值,表示不关心任何文件的读变化。
fd_set *writefds:是指向fd_set结构的指针,这个集合中应该包括文件描述符,我们是要监视这些文件描述符的写变化的,即我们关心是否可以向这些文件中写入数据了,如果这个集合中有一个文件可写,select就会返回一个大于0的值,表示有文件可写,如果没有可写的文件,则根据timeout参数再判断是否超时,若超出timeout的时间,select返回0,若发生错误返回负值。可以传入NULL值,表示不关心任何文件的写变化。
fd_set *errorfds:同上面两个参数的意图,用来监视文件错误异常。
struct timeval*timeout:是select的超时时间,这个参数至关重要,它可以使select处于三种状态。
- 若将NULL以形参传入,即不传入时间结构,就是将select置于阻塞状态,一定等到监视文件描述符集合中某个文件描述符发生变化为止;
- 若将时间值设为0秒0毫秒,就变成一个纯粹的非阻塞函数,不管文件描述符是否有变化,都立刻返回继续执行,文件无变化返回0,有变化返回一个正值;
- timeout的值大于0,这就是等待的超时时间,即select在timeout时间内阻塞,超时时间之内有事件到来就返回了,否则在超时后不管怎样一定返回,返回值同上述。
操作fd_set集合的宏:
FD_ZERO(fd_set *): 清空集合;
FD_SET(int,fd_set*): 将一个给定的文件描述符加入集合之中;
FD_CLR(int,fd_set*): 将一个给定的文件描述符从集合中删除;
FD_ISSET(int,fd_set*): 检查集合中指定的文件描述符是否可以读写
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <stdlib.h>
#include <string.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#define SIZE 7
int main()
{
int sockfd;
int ret;
int fd[SIZE];
int i;
struct sockaddr_in server_addr; //服务器地址信息
struct sockaddr_in client_addr; //客户端地址信息
char buf[32] = {0};
sockfd = socket(PF_INET, SOCK_STREAM, 0); //创建socket
if(-1 == sockfd)
{
perror("socket");
exit(1);
}
memset(&server_addr, 0, sizeof(server_addr));
server_addr.sin_family = PF_INET;
server_addr.sin_port = 8000;
server_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
ret = bind(sockfd, (struct sockaddr *)&server_addr, sizeof(server_addr)); //绑定信息
if(-1 == ret)
{
perror("bind");
}
ret = listen(sockfd, 10); //监听
if(-1 == ret)
{
perror("listen");
exit(1);
}
fd_set readfd, tmpfd; //监听的文件描述符的集合
FD_ZERO(&readfd); //清空监听集合
FD_SET(sockfd, &readfd); //将服务器创建的通道加入监听集合
FD_SET(0, &readfd);
int maxfd = sockfd; //集合中所存文件描述符的范围
int length = sizeof(client_addr);
while(1)
{
tmpfd = readfd;
ret = select(maxfd + 1, &tmpfd, NULL, NULL, NULL); //监听的文件描述符的集合是否有读变化,参数5传入NULL,则如果集合中没有读状态的变化为阻塞状态。
if(-1 == ret)
{
perror("select");
}
if(FD_ISSET(sockfd, &tmpfd)) //判断发生状态变化的是否为 sockfd 描述符,如果是表示有 connect 从客户端来 把对应的fd[i]记录下来
{
for(i = 0; i < SIZE; i++)
{
if(fd[i] == 0)
{
break;
}
}
fd[i] = accept(sockfd, (struct sockaddr *)&client_addr, &length); //接受客户端的请求
if(-1 == fd[i])
{
perror("accept");
exit(1);
}
printf("accept client %d\n",fd[i]);
FD_SET(fd[i], &readfd); //将文件描述符fd[i]加入监听集合,表示监听fd[i]所代表的客户端
if(fd[i] > maxfd) //更新文件中文件描述符范围
{
maxfd = fd[i];
}
}
else //判断发生状态变化的不是 sockfd 描述符,则表示有 send 从客户端来, 判断是那个fd。
{
for(i = 0; i < SIZE; i++)
{
if(FD_ISSET(fd[i],&TmpFd)) //判断发生读状态变化的是哪个fd,并进行相应的操作。
{
ret = recv(fd[i], buf, sizeof(buf), 0);
if(-1 == ret)
{
perror("recv");
}
if(!strcmp(buf, "bye")) //服务器端接收到客户端 send 的数据为 “bye”
{
printf("client %d exit!\n",fd[i]);
FD_CLR(fd[i], &readfd); //清空相应的文件描述符
close(fd[i]); //关闭与相应客户端的通讯
fd[i] = 0;
break;
}
printf("%s\n",buf);
memset(buf,0,sizeof(buf));
break;
}
}
}
}
return 0;
}