使用SIGALRM信号为阻塞操作设置超时

我们经常会遇到为阻塞操作设置超时的问题，比如说阻塞套接字read读取设置10秒超时，其中一个办法就是调用alarm函数，它在指定超时时期产生SIGALRM信号，使得阻塞操作中断。

但其弊端在于：

1、可能干扰进程中现有的alarm调用，如timer、settimer、sleep等。

2、在多线程化程序中正确使用信号却非常困难；所以建议只是在未线程化或单线程化的程序中使用本技术。

 

#include <stdio.h>  
#include <netinet/in.h>//for struct sockaddr_in  
#include <string.h>//for memset  
#include <signal.h>//for signal  
  
typedef void (*sighandler_t)(int);  
  
static void  
read_alarm(int signo)  
{  
    return;  
}  
  
int main()  
{  
    int conn_sock;  
    struct sockaddr_in ser_addr;  
    int ret;  
    char buf[1024];  
    sighandler_t src_sig;  
    struct sigaction sa_alarm;  
  
    conn_sock = socket(AF_INET, SOCK_STREAM, 0);  
    if (conn_sock < 0)  
    {  
        perror("socket error");  
        return -1;  
    }  
  
    memset(&ser_addr, 0, sizeof(ser_addr));  
    ser_addr.sin_family = AF_INET;  
    ser_addr.sin_port = htons(135);  
    inet_pton(AF_INET, "127.0.0.1", &ser_addr.sin_addr);  
  
    ret = connect(conn_sock, (struct sockaddr *)&ser_addr, sizeof(ser_addr));  
    if (ret < 0)  
    {  
        perror("connect error");  
        return -1;  
    }  
  
    //src_sig = signal(SIGALRM, read_alarm);  
    sa_alarm.sa_flags = SA_RESETHAND;  
    sa_alarm.sa_handler = read_alarm;  
    sigaction(SIGALRM, &sa_alarm, NULL);  
    alarm(10);  
    ret = read(conn_sock, buf, sizeof(buf));  
    if (ret < 0)  
    {  
        perror("read error");  
    }  
    else if (ret == 0)  
    {  
        printf("close by peer\n");  
    }  
    else  
    {  
        printf("recv %d bytes\n", ret);  
    }  
    alarm(0);  
    //signal(SIGALRM, src_sig);  
    return 0;  
}  

 

运行结果：

read error: Interrupted system call 

 

在linux系统中使用signal函数，read超时10秒后不会被中断，因为中断后会自动重启被中断的read。

需要采用segaction函数来验证，当read超过10秒时，alarm发出SIGALRM信号，read被中断。