linux accept

本篇是上篇《 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