本篇是上篇《 c++实现服务器在线重启》涉及到的一个小实验,具体是验证accept等库函数在被信号中断后,如何设置它们在信号处理函数完成后的动作。如果有什么不懂的地方可以参见这里。
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <netinet/in.h>
#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <errno.h>
#include <unistd.h>
#include <syslog.h>
#include <string.h>
#include <signal.h>
#include <fstream>
#include <ctime>
#include <iostream>
#define MAXLINE 4096
void handler(int sig) {
std::cout<<"hello"<<std::endl;
}
int main(int argc, char const* argv[])
{
int listenfd, connfd;
struct sockaddr_in servaddr;
char buff[4096];
int n;
if( (listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1 ){
printf("create socket error: %s(errno: %d)\n",strerror(errno),errno);
exit(0);
}
// timeval timeout = {1, 0};
// if (setsockopt(listenfd, SOL_SOCKET, SO_RCVTIMEO, (char*)&timeout, sizeof(timeval)) < 0)
// {
// exit(1);
// }
memset(&servaddr, 0, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
servaddr.sin_port = htons(6666);
if( bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) == -1){
printf("bind socket error: %s(errno: %d)\n",strerror(errno),errno);
exit(0);
}
if( listen(listenfd, 10) == -1){
printf("listen socket error: %s(errno: %d)\n",strerror(errno),errno);
exit(0);
}
struct sigaction action;
action.sa_handler = handler;
sigemptyset(&action.sa_mask);
// action.sa_flags = 0;
action.sa_flags |= SA_RESTART;
sigaction(SIGALRM, &action, NULL);
alarm(3);
printf("======waiting for client's request======\n");
if((connfd = accept(listenfd, (struct sockaddr*)NULL, NULL)) == -1){
printf("accept socket error: %s(errno: %d)",strerror(errno),errno);
}
n = recv(connfd, buff, MAXLINE, 0);
buff[n] = '\0';
std::cout<<buff<<std::endl;
close(connfd);
close(listenfd);
//close the log
closelog();
while (1) {
std::cout<<"wait..."<<std::endl;
sleep(1);
}
return 0;
}
当action.sa_flags被设置为SA_RESTART时,程序没有执行while部分,既没有输出wait,说明accept函数在中断后被重启。
#action.sa_flags |= SA_RESTART;
[wangqingze@localhost socket]$ ./server
======waiting for client's request======
hello
当action.sa_flags被设置成0时,程序执行while部分,说明accept函数在中断后没有被重启。
[wangqingze@localhost socket]$ ./server
======waiting for client's request======
hello
accept socket error: Interrupted system call(errno: 4)
wait...
wait...
wait...
wait...
wait...
wait...
^C
按照官方文档的说明,setsockopt函数会影响accept函数的行为,但是程序证明没有,应该是有什么地方设置不正确。
#action.sa_flags = 0;
#setsockopt(listenfd, SOL_SOCKET, SO_RCVTIMEO, (char*)&timeout, sizeof(timeval))
[wangqingze@localhost socket]$ ./server
======waiting for client's request======
accept socket error: Resource temporarily unavailable(errno: 11)
wait...
wait...
hello
wait...
wait...
wait...
wait...
wait...
wait...
^C
#action.sa_flags |= SA_RESTART;
#setsockopt(listenfd, SOL_SOCKET, SO_RCVTIMEO, (char*)&timeout, sizeof(timeval))
[wangqingze@localhost socket]$ ./server
======waiting for client's request======
accept socket error: Resource temporarily unavailable(errno: 11)
wait...
wait...
hello
wait...
wait...
wait...
wait...
wait...
wait...
wait...
wait...
wait...
^C