Linux学习（二十五）：网络超时检测
1、介绍
在网络通信中，很多操作会使得进程阻塞，TCP套接字中的recv/accept/connect，UDP套接字中的recvfrom。
超时检测的必要性：
1、避免进程在没有数据时无限制地阻塞
2、当设定的时间到时，进程从原操作返回继续运行
超时检测的必要性：
1、避免进程在没有数据时无限制地阻塞
2、当设定的时间到时，进程从原操作返回继续运行
网络超时检测的实质是：设定一定的时间，在时间到达之前，阻塞等待数据的到来，如果时间到时还没有数据，则变成非阻塞状态
当客户端连接服务器之后，服务器会开辟一块空间与客户端进行通信，如果在一定时间之内客户端没有雨服务器进行通信，则会浪费服务器空间，所以使用超时检测断开客户端的连接，减少服务器的资源的浪费。
2、使用setsocketopt实现超时检测
这个函数功能还有很多，我们这一节只介绍和超时检测相关的，在后面的广播和组播中还会用到。设置超时时间时的optval的类型为
struct timeval
{
__time_t tv_sec; 秒
__suseconds_t tv_usec; 微秒
};
{
__time_t tv_sec; 秒
__suseconds_t tv_usec; 微秒
};
例程：
服务器设定5s接收超时时间，注意，设置后，网络相关的阻塞函数都有效accept和recv函数都会有5s的检测超时机制（accpet和recv其实都是一种，都是在接收数据）
#include //printf
#include //inet_addr htons
#include
#include //socket bind listen accept connect
#include //sockaddr_in
#include //exit
#include //close
#include
#include
#define N 128
#define errlog(errmsg) do{\
perror(errmsg);\
printf("%s --> %s --> %d\n", __FILE__, __func__, __LINE__);\
exit(1);\
}while(0)
int main(int argc, const char *argv[])
{
int sockfd, acceptfd;
struct sockaddr_in serveraddr, clientaddr;
socklen_t addrlen = sizeof(serveraddr);
char buf[N] = {};
ssize_t bytes;
if(argc < 3)
{
printf("您输入的参数太少了: %s \n", argv[0]);
exit(1);
}
//第一步：创建套接字
if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
{
errlog("fail to socket");
}
//第二步：填充服务器网络信息结构体
//inet_addr：将点分十进制ip地址转化为网络字节序的整型数据
//htons：将主机字节序转化为网络字节序
//atoi：将数字型字符串转化为整型数据
serveraddr.sin_family = AF_INET;
serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
serveraddr.sin_port = htons(atoi(argv[2]));
//第三步：将套接字域网络信息结构体绑定
if(bind(sockfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr)) < 0)
{
errlog("fail to bind");
}
//第四步：将套接字设置为监听状态
if(listen(sockfd, 5) < 0)
{
errlog("fail to listen");
}
//使用setsockopt实现网络超时检测
//注意：使用setscokopt设置的超时时间，设置一次，永久有效，并且与网络相关的阻塞函数都有效
struct timeval time_out;
time_out.tv_sec = 5;
time_out.tv_usec = 0;
if(setsockopt(sockfd, SOL_SOCKET, SO_RCVTIMEO, &time_out, sizeof(time_out)) < 0)
{
errlog("fail to setsockopt");
}
while(1)
{
//第五步：阻塞等待客户端的连接请求
if((acceptfd = accept(sockfd, (struct sockaddr *)&clientaddr, &addrlen)) < 0)
{
//printf("errno = %d\n", errno);
if(errno == 11)
{
printf("accept timeout ...\n");
}
else
{
errlog("fail to accept");
}
}
else
{
//打印客户端的ip地址、端口号
printf("%s --- %d\n", inet_ntoa(clientaddr.sin_addr), ntohs(clientaddr.sin_port));
while(1)
{
if((bytes = recv(acceptfd, buf, N, 0)) < 0)
{
if(errno == 11)
{
printf("recv timeout, client must be quited!!!\n");
strcpy(buf, "TIMEO");
send(acceptfd, buf, N, 0);
close(acceptfd);
break;
}
else
{
errlog("fail to recv");
}
}
else if(bytes == 0)
{
printf("NO DATA\n");
exit(1);
}
else
{
if(strncmp(buf, "quit", 4) == 0)
{
printf("client is quited ...\n");
break;
}
else
{
printf("client : %s\n", buf);
strcat(buf, " *_*");
if(send(acceptfd, buf, N, 0) < 0)
{
errlog("fail to send");
}
}
}
}
}
}
close(acceptfd);
close(sockfd);
return 0;
}
客户端
#include //printf
#include //inet_addr htons
#include
#include //socket bind listen accept connect
#include //sockaddr_in
#include //exit
#include //close
#include
#define N 128
#define errlog(errmsg) do{\
perror(errmsg);\
printf("%s --> %s --> %d\n", __FILE__, __func__, __LINE__);\
exit(1);\
}while(0)
int main(int argc, const char *argv[])
{
int sockfd;
struct sockaddr_in serveraddr;
socklen_t addrlen = sizeof(serveraddr);
char buf[N] = {};
if(argc < 3)
{
printf("您输入的参数太少了: %s \n", argv[0]);
exit(1);
}
//第一步：创建套接字
if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
{
errlog("fail to socket");
}
//第二步：填充服务器网络信息结构体
//inet_addr：将点分十进制ip地址转化为网络字节序的整型数据
//htons：将主机字节序转化为网络字节序
//atoi：将数字型字符串转化为整型数据
serveraddr.sin_family = AF_INET;
serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
serveraddr.sin_port = htons(atoi(argv[2]));
//第三步：发送客户端的连接请求
if(connect(sockfd, (struct sockaddr *)&serveraddr, addrlen) < 0)
{
errlog("fail to connect");
}
while(1)
{
fgets(buf, N, stdin);
buf[strlen(buf) - 1] = '\0';
if(send(sockfd, buf, N, 0) < 0)
{
errlog("fail to send");
}
if(strncmp(buf, "quit", 4) == 0)
{
printf("client quit ...\n");
break;
}
else
{
if(recv(sockfd, buf, N, 0) < 0)
{
errlog("fail to recv");
}
printf("server : %s\n", buf);
}
}
close(sockfd);
return 0;
}
服务器：
客户端
服务器在等待连接，客户端过一段时间再连接，可以看到服务器打印出超时信息，客户端连接后，不及时发送数据，服务器也会提示接收超时，客户端退出后，服务器再次进入accept超时检测。
2、select实现超时检测
在Linux学习（二十三）：IO模型中我们学习了IO多路复用功能select函数，select函数中的最后一个参数就是设置超时时间，可以利用select函数实现网络超时检测。
我们用上一节Linux学习（二十四）：服务器模型select函数章节的服务器代码稍作修改即可。
服务器：
#include //printf
#include //inet_addr htons
#include
#include //socket bind listen accept connect
#include //sockaddr_in
#include //exit
#include //close
#include
#include
#include
#define N 128
#define errlog(errmsg) do{\
perror(errmsg);\
printf("%s --> %s --> %d\n", __FILE__, __func__, __LINE__);\
exit(1);\
}while(0)
int main(int argc, const char *argv[])
{
int sockfd, acceptfd;
struct sockaddr_in serveraddr, clientaddr;
socklen_t addrlen = sizeof(serveraddr);
char buf[N] = {};
ssize_t bytes;
if(argc < 3)
{
printf("您输入的参数太少了: %s \n", argv[0]);
exit(1);
}
//第一步：创建套接字
if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
{
errlog("fail to socket");
}
printf("scokfd = %d\n", sockfd);
//第二步：填充服务器网络信息结构体
//inet_addr：将点分十进制ip地址转化为网络字节序的整型数据
//htons：将主机字节序转化为网络字节序
//atoi：将数字型字符串转化为整型数据
serveraddr.sin_family = AF_INET;
serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
serveraddr.sin_port = htons(atoi(argv[2]));
//第三步：将套接字域网络信息结构体绑定
if(bind(sockfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr)) < 0)
{
errlog("fail to bind");
}
//第四步：将套接字设置为监听状态
if(listen(sockfd, 5) < 0)
{
errlog("fail to listen");
}
//使用select函数实现网络超时检测
//注意：select函数返回后，会移除除当前文件描述符以外其他所有的
//注意：使用select函数设置的超时时间，设置一次，只有效一次，所以需要每次都设置
// 由于只考虑select函数的阻塞，所以没办法区分到底是哪个io操作引起的超时，一般
// 用于一个io操作的select使用
fd_set readfds, tempfds;
int maxfd, i, ret;
struct timeval time_out;
//第一步：清空集合
FD_ZERO(&readfds);
//第二步：将需要的文件描述符添加到集合里面
FD_SET(sockfd, &readfds);
maxfd = sockfd;
while(1)
{
time_out.tv_sec = 5;
time_out.tv_usec = 0;
tempfds = readfds;
//第三步：调用函数阻塞等待文件描述符准备就绪
if((ret = select(maxfd + 1, &tempfds, NULL, NULL, &time_out)) < 0)
{
errlog("fail to select");
}
else if(ret == 0)
{
printf("timeout......\n");
}
else
{
//由于返回之后集合里面只剩下一个文件描述符，所以判断是哪一个，执行相应的操作
for(i = 0; i < maxfd + 1; i++)
{
if(FD_ISSET(i, &tempfds))
{
if(i == sockfd)
{
//第五步：阻塞等待客户端的连接请求
if((acceptfd = accept(sockfd, (struct sockaddr *)&clientaddr, &addrlen)) < 0)
{
errlog("fail to accept");
}
printf("acceptfd = %d\n", acceptfd);
//打印客户端的ip地址、端口号
printf("%s --- %d\n", inet_ntoa(clientaddr.sin_addr), ntohs(clientaddr.sin_port));
//将需要的文件描述符添加到集合里面
FD_SET(acceptfd, &readfds);
//确定最大的文件描述符
maxfd = maxfd > acceptfd ? maxfd : acceptfd;
}
else
{
if((bytes = recv(i, buf, N, 0)) < 0)
{
errlog("fail to recv");
}
else if(bytes == 0)
{
printf("NO DATA\n");
FD_CLR(i, &readfds);
close(i);
break;
}
else
{
if(strncmp(buf, "quit", 4) == 0)
{
printf("client is quited ...\n");
FD_CLR(i, &readfds);
close(i);
break;
}
else
{
printf("client : %s\n", buf);
strcat(buf, " *_*");
if(send(i, buf, N, 0) < 0)
{
errlog("fail to send");
}
}
}
}
}
}
}
}
close(acceptfd);
close(sockfd);
return 0;
}
3、利用信号实现超时检测
在学习进程通信时，我们学了一个唯一的异步通信方式--信号，我们可以使用alarm闹钟信号的方式实现超时检测。要注意以下几点：
1、闹钟信号响的时候，默认会退出进程，我们使用signal函数改变闹钟信号的执行结果，输出打印超时信息即可。
2、信号响应时会中断当前的系统调用，信号函数执行完毕后，默认会继续执行之前的系统调用，这是Linux的自重启属性。例如accept函数被alarm信号打断，执行响应handler完毕后，仍会回到accetp继续一直等待，无法实现重复超时检测，需要取消自重启属性，然后通过错误标识号errno来实现超时检测。这里要用到sigaction函数
参数中struct sigaction如下：
struct sigaction {
void (*sa_handler)(int); 信号处理函数
void (*sa_sigaction)(int, siginfo_t *, void *); 信号处理函数，sa_flag为SA_SIGINFO时使用
sigset_t sa_mask; 掩码（关于阻塞）
int sa_flags; 标志位（ SA_RESTART 自重启属性）
void (*sa_restorer)(void); 没有用
};
void (*sa_handler)(int); 信号处理函数
void (*sa_sigaction)(int, siginfo_t *, void *); 信号处理函数，sa_flag为SA_SIGINFO时使用
sigset_t sa_mask; 掩码（关于阻塞）
int sa_flags; 标志位（ SA_RESTART 自重启属性）
void (*sa_restorer)(void); 没有用
};
一般设置第一个参数sa_handler和标志位参数sa_flags，注意标志位都是用位来表征的，所以我们要用与或来操作。
服务器：
#include //printf
#include //inet_addr htons
#include
#include //socket bind listen accept connect
#include //sockaddr_in
#include //exit
#include //close
#include
#include
#include
#define N 128
#define errlog(errmsg) do{\
perror(errmsg);\
printf("%s --> %s --> %d\n", __FILE__, __func__, __LINE__);\
exit(1);\
}while(0)
void handler(int sig)
{
//printf("timeout.......\n");
}
int main(int argc, const char *argv[])
{
int sockfd, acceptfd;
struct sockaddr_in serveraddr, clientaddr;
socklen_t addrlen = sizeof(serveraddr);
char buf[N] = {};
ssize_t bytes;
if(argc < 3)
{
printf("您输入的参数太少了: %s \n", argv[0]);
exit(1);
}
//第一步：创建套接字
if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
{
errlog("fail to socket");
}
//第二步：填充服务器网络信息结构体
//inet_addr：将点分十进制ip地址转化为网络字节序的整型数据
//htons：将主机字节序转化为网络字节序
//atoi：将数字型字符串转化为整型数据
serveraddr.sin_family = AF_INET;
serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
serveraddr.sin_port = htons(atoi(argv[2]));
//第三步：将套接字域网络信息结构体绑定
if(bind(sockfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr)) < 0)
{
errlog("fail to bind");
}
//第四步：将套接字设置为监听状态
if(listen(sockfd, 5) < 0)
{
errlog("fail to listen");
}
//使用alarm闹钟实现网络超时检测
//第一步：获取旧的行为
struct sigaction act;
if(sigaction(SIGALRM, NULL, &act) < 0)
{
perror("fail to sigaction");
return -1;
}
//第二步：修改行为
act.sa_handler = handler;
act.sa_flags = act.sa_flags & (~SA_RESTART);
//act.sa_flags |= SA_RESTART;
//第三步：将新的行为写回去
if(sigaction(SIGALRM, &act, NULL) < 0)
{
perror("fail to sigaction");
return -1;
}
while(1)
{
alarm(5);
//第五步：阻塞等待客户端的连接请求
if((acceptfd = accept(sockfd, (struct sockaddr *)&clientaddr, &addrlen)) < 0)
{
//printf("errno = %d\n", errno);
if(errno == 4)
{
printf("accept timeout ...\n");
}
else
{
errlog("fail to accept");
}
}
else
{
//打印客户端的ip地址、端口号
printf("%s --- %d\n", inet_ntoa(clientaddr.sin_addr), ntohs(clientaddr.sin_port));
while(1)
{
alarm(5);
if((bytes = recv(acceptfd, buf, N, 0)) < 0)
{
if(errno == 4)
{
printf("recv timeout, client must be quited!!!\n");
strcpy(buf, "TIMEO");
send(acceptfd, buf, N, 0);
close(acceptfd);
break;
}
else
{
errlog("fail to recv");
}
}
else if(bytes == 0)
{
printf("NO DATA\n");
exit(1);
}
else
{
if(strncmp(buf, "quit", 4) == 0)
{
printf("client is quited ...\n");
break;
}
else
{
printf("client : %s\n", buf);
strcat(buf, " *_*");
if(send(acceptfd, buf, N, 0) < 0)
{
errlog("fail to send");
}
}
}
}
}
}
close(acceptfd);
close(sockfd);
return 0;
}
#include //printf
#include //inet_addr htons
#include
#include //socket bind listen accept connect
#include //sockaddr_in
#include //exit
#include //close
#include
#define N 128
#define errlog(errmsg) do{\
perror(errmsg);\
printf("%s --> %s --> %d\n", __FILE__, __func__, __LINE__);\
exit(1);\
}while(0)
int main(int argc, const char *argv[])
{
int sockfd;
struct sockaddr_in serveraddr;
socklen_t addrlen = sizeof(serveraddr);
char buf[N] = {};
if(argc < 3)
{
printf("您输入的参数太少了: %s \n", argv[0]);
exit(1);
}
//第一步：创建套接字
if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
{
errlog("fail to socket");
}
//第二步：填充服务器网络信息结构体
//inet_addr：将点分十进制ip地址转化为网络字节序的整型数据
//htons：将主机字节序转化为网络字节序
//atoi：将数字型字符串转化为整型数据
serveraddr.sin_family = AF_INET;
serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
serveraddr.sin_port = htons(atoi(argv[2]));
#if 0
//客户端也可以自己指定自己的信息
struct sockaddr_in clientaddr;
clientaddr.sin_family = AF_INET;
clientaddr.sin_addr.s_addr = inet_addr(argv[3]);
clientaddr.sin_port = htons(atoi(argv[4]));
if(bind(sockfd, (struct sockaddr *)&clientaddr, addrlen) < 0)
{
errlog("fail to bind");
}
#endif
//第三步：发送客户端的连接请求
if(connect(sockfd, (struct sockaddr *)&serveraddr, addrlen) < 0)
{
errlog("fail to connect");
}
while(1)
{
fgets(buf, N, stdin);
buf[strlen(buf) - 1] = '\0';
if(send(sockfd, buf, N, 0) < 0)
{
errlog("fail to send");
}
if(strncmp(buf, "quit", 4) == 0)
{
printf("client quit ...\n");
break;
}
else
{
if(recv(sockfd, buf, N, 0) < 0)
{
errlog("fail to recv");
}
if(strncmp(buf, "TIMEO", 5) == 0)
{
printf("timeout, quit...\n");
break;
}
printf("server : %s\n", buf);
}
}
close(sockfd);
return 0;
}
服务器在接收超时后向客户端发送超时信息，客户端再次输入数据时将会被退出，表明已经断开连接，需要从新连接
执行结果，服务器：
客户端：
