select()函数
1、利用select()函数进行超时检测相比较于其他两种方式的特点是:
使用select函数实现超时检测,超时时间设置一次,只会有效一次,所以需要将其放在循环里面
当select函数到达设定的时间时,函数会返回0
2、在进行超时检测时,应注意的是select的返回值。如果利用其进行超时检测,第5个参数必然不能为 NULL ,在其不为空时,超时后返回 0 。
使用select()实现超时检测
#include <sys/select.h>
#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>
int select(int nfds, fd_set *readfds, fd_set *writefds,
fd_set *exceptfds, struct timeval *timeout);
功能:允许一个进程去操作多个文件描述符,阻塞等待一个或者是多个文件描述符准备就绪,
当有一个或者多个文件描述符准备就绪,则函数立即返回
void FD_CLR(int fd, fd_set *set);
将fd移除set集合
int FD_ISSET(int fd, fd_set *set);
判断fd是否存在在set里面
void FD_SET(int fd, fd_set *set);
将fd添加到set集合里面
void FD_ZERO(fd_set *set);
清空set集合
参数:
nfds: 最大的文件描述符加一
readfds: 读文件描述符集合
writefds:写文件描述符集合
exceptfds:其他或者异常文件描述符集合
timeout: 设置超时时间
返回值:
成功:
如果timeout = NULL 则返回准备就绪的文件描述符的个数
如果timeout != NULL 超时后返回0
失败:
-1
+++++++++++++++++++++++++++++++++++++
struct timeval {
int tv_sec; 秒
int tv_usec; 微秒
};
+++++++++++++++++++++++++++++++++++++
struct timeval out_time;
FD_ZERO(&readfds);
FD_SET(sockfd, &readfds);
out_time.tv_sec = 5;
out_time.tv_usec = 0;
if((num = select(maxfd + 1, &readfds, NULL, NULL, &out_time)) < 0)
{
errlog("fail to select");
}
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
下面上实例:
前面我们用select()函数实现并发服务器,在原来代码的基础上我们稍加更改,便可实现超时检测的功能。
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <stdlib.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <sys/select.h>
#include <sys/time.h>
//使用select函数实现TCP并发服务器
#define N 128
#define errlog(errmsg) do{perror(errmsg); exit(1);}while(0)
int main(int argc, const char *argv[])
{
int sockfd;
struct sockaddr_in serveraddr, clientaddr;
int acceptfd;
socklen_t addrlen = sizeof(struct sockaddr_in);
fd_set readfds;
int maxfd;
int num = 0;
fd_set tempfds;
int i = 0;
char buf[N] = {};
//创建套接字
if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
{
errlog("fail to socket");
}
printf("sockfd = %d\n", sockfd);
//填充网络信息结构体
//inet_addr 将点分十进制转化成网络字节
//htons表示将主机字节序转化成网络字节序
//atoi 将字符串转化成整型数据
serveraddr.sin_family = AF_INET;
serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
serveraddr.sin_port = htons(atoi(argv[2]));
//将套接字与IP地址和端口号绑定
if(bind(sockfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr)) < 0)
{
errlog("fail to bind");
}
//将套接字设置为被动监听状态
if(listen(sockfd, 10) < 0)
{
errlog("fail to listen");
}
//使用select函数实现超时检测
//超时时间设置一次,只会有效一次,所以需要将其放在循环里面
struct timeval out_time;
//第一步:清空集合
FD_ZERO(&readfds);
//第二步:将文件描述符添加到集合当中
//注意:当select函数调用成功后,他会清除没有准备就绪的文件描述符,所以需要每次重复添加
FD_SET(sockfd, &readfds);
maxfd = sockfd;
while(1)
{
out_time.tv_sec = 5;
out_time.tv_usec = 0;
tempfds = readfds;
//第三步:调用select函数将添加进去的文件描述符准备就绪
if((num = select(maxfd + 1, &tempfds, NULL, NULL, &out_time)) < 0)
{
errlog("fail to select");
}
else if(num == 0)
{
printf("timeout...\n");
}
else
{
//使用FD_ISSET判断文件描述符
for(i = 0; i < maxfd + 1; i++)
{
if(FD_ISSET(i, &tempfds) == 1)
{
if(i == sockfd)
{
//接收客户端的连接请求
if((acceptfd = accept(sockfd, (struct sockaddr *)&clientaddr, &addrlen)) < 0)
{
errlog("fail to accept");
}
printf("acceptfd = %d\n", acceptfd);
printf("%s ---> %d\n", inet_ntoa(clientaddr.sin_addr), ntohs(clientaddr.sin_port));
FD_SET(acceptfd, &readfds);
maxfd = maxfd > acceptfd ? maxfd : acceptfd;
}
else
{
if(recv(i, buf, N, 0) < 0)
{
errlog("fail to recv");
}
if(strncmp(buf, "quit", 4) == 0)
{
printf("%s is quited...\n", inet_ntoa(clientaddr.sin_addr));
break;
}
else
{
printf("from client >>> %s\n", buf);
strcat(buf, " from server...");
send(i, buf, N, 0);
}
}
}
}
}
}
close(sockfd);
close(acceptfd);
return 0;
}