服务器编程入门(13) Linux套接字设置超时的三种方法

摘要:

    本文介绍在套接字的I/O操作上设置超时的三种方法。


服务器编程入门(13) Linux套接字设置超时的三种方法

图片可能有点宽,看不到的童鞋可以点击图片查看完整图片。。


1 调用alarm

使用SIGALRM为connect设置超时

设置方法:

  1. 监听SIGALRM信号,
  2. 设置sig_alrm处理函数,
  3. 在阻塞函数前调用alarm函数设置超时时间,
  4. 正常返回后,重置超时事件为0
服务器编程入门(13) Linux套接字设置超时的三种方法
void handle_msg(int sockfd) {

    char sendbuf[BUFSIZE];
    char recvbuf[BUFSIZE];
    
    signal(SIGALRM, sig_alrm);    //监听SIGALRM信号
 
    while(1) {
        memset( sendbuf, \0, BUFSIZE );
        memset( recvbuf, \0, BUFSIZE );
        
        printf("%s", "send msg:");
        gets(sendbuf);

        if (strlen(sendbuf) > 0)
            send(sockfd,sendbuf,strlen(sendbuf),0);

        if ( !strcmp(sendbuf, "exit"))
            break;

        alarm(5);       //设置超时事件为5s,同时设置服务器回射前sleep 10秒,以让recv函数超时
        if (recv(sockfd,recvbuf,BUFSIZE,0) > 0) {
            alarm(0);
            printf("recv back:%s\n\n", recvbuf);
        }
        else {
            if (errno == EINTR)
                fprintf(stderr,
                        "socket timeout\n");
            else
                fprintf(stderr,
                        "receive error\n");
        }
    }
    close( sockfd );
    return;
}


static void sig_alrm(int signo) {
    fprintf(stderr,
            "recv SIGALRM, return.\n");
    return;
}
服务器编程入门(13) Linux套接字设置超时的三种方法

运行截图:

服务器编程入门(13) Linux套接字设置超时的三种方法

虽然设置了SIGALRM信号处理函数,但是如图所示,本例依然可以等待读取回射信息,因为信号处理函数里只是return。


2 使用select阻塞等待I/O

设置方法:

使用select的内置时间限制,阻塞在select代替recv函数的阻塞。

服务器编程入门(13) Linux套接字设置超时的三种方法
void handle_msg(int sockfd) {

    char sendbuf[BUFSIZE];
    char recvbuf[BUFSIZE];

    while(1) {
        memset( sendbuf, \0, BUFSIZE );
        memset( recvbuf, \0, BUFSIZE );
        
        printf("%s", "send msg:");
        gets(sendbuf);

        if (strlen(sendbuf) > 0)
             send(sockfd,sendbuf,strlen(sendbuf),0);

        if ( !strcmp(sendbuf, "exit"))
            break;
        
        if (readable_timeo(sockfd, 5) == 0) {
            fprintf(stderr,
                    "socket timeout\n");
        }
        else{
            recv(sockfd,recvbuf,BUFSIZE,0);
            printf("recv back:%s\n\n", recvbuf);
        }
    }
    close( sockfd );
    return;
}

int readable_timeo(int fd, int sec) {
    fd_set rset;
    struct timeval tv;
    
    FD_ZERO(&rset);
    FD_SET(fd, &rset);

    tv.tv_sec = sec;
    tv.tv_usec = 0;

    return select(fd+1, &rset, NULL, NULL, &tv);
}
服务器编程入门(13) Linux套接字设置超时的三种方法

 

运行截图:

服务器编程入门(13) Linux套接字设置超时的三种方法

由运行截图可以看到,超时警告运行正常。

需要注意一点的是,虽然客户端在超时之后继续发送消息,但是服务器回射的消息(hello world)依然被接收,这导致我们再次发送消息时,从缓冲区中读出了延迟收到的hello world。

这里只是演示超时技术,因此对此并不做进一步处理。


3 使用SO_RCVTIMEO套接字选项

使用SO_RCVTIMEO套接字选项为recv设置超时

设置方法:

  • 使用setsockopt函数对套接字进行设置
  • 一旦设置了某个描述符,其超时设置将应用于该描述符上的所有读操作
  • SO_RCVTIMEO仅用于读操作,SO_SNDTIMEO仅用于写操作,两者都不能用于为connect设置超时
  • 如果套接字超时,被阻塞的函数将返回一个EWOULDBLOCK错误
服务器编程入门(13) Linux套接字设置超时的三种方法
void handle_msg(int sockfd) {

    char sendbuf[BUFSIZE];
    char recvbuf[BUFSIZE + 1];
    int n;
    struct timeval tv;

    tv.tv_sec = 5;
    tv.tv_usec = 0;
    setsockopt( sockfd, SOL_SOCKET, SO_RCVTIMEO,
               &tv, sizeof(tv) );

    while(1) {
         memset( sendbuf, \0, BUFSIZE );
        memset( recvbuf, \0, BUFSIZE );
        
        printf("%s", "send msg:");
        gets(sendbuf);

        if (strlen(sendbuf) > 0)
             send(sockfd,sendbuf,strlen(sendbuf),0);

        if ( !strcmp(sendbuf, "exit"))
            break;

        if ( (n=recv(sockfd,recvbuf,BUFSIZE,0)) < 0 ) {
            if (errno == EWOULDBLOCK) {
                fprintf(stderr,
                        "socket timeout\n");
                continue;
            }
            else
                fprintf(stderr,
                        "recv error");
        }
        else{
            printf("recv back:%s\n\n", recvbuf);
        }
    }
    close( sockfd );
    return;
}
服务器编程入门(13) Linux套接字设置超时的三种方法

运行截图:

服务器编程入门(13) Linux套接字设置超时的三种方法

可以看到,超时警报成功运行,但依然有上一例中的延迟接收的情况发生,不作处理。


 

示例源码上传到了github上,地址:https://github.com/zs634134578/UNP/tree/tryTimeout

 

 

参考资料:

《UNIX网络编程 卷1:套接字联网API(第3版)》

服务器编程入门(13) Linux套接字设置超时的三种方法,布布扣,bubuko.com

服务器编程入门(13) Linux套接字设置超时的三种方法

上一篇:Eclipse中Ant的配置与测试


下一篇:Linux操作命令--持续更新