套接字超时处理的三种方法

最新推荐文章于 2024-04-20 09:51:01 发布

原创最新推荐文章于 2024-04-20 09:51:01 发布 · 1.2w 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

unix网络编程同时被 2 个专栏收录

10 篇文章

订阅专栏

UNPV探索

10 篇文章

订阅专栏

本文介绍了处理套接字超时的三种方法：一是利用SIGALRM信号结合connect设置超时，当alarm信号触发时中断connect；二是使用select为accept设置超时，确保等待连接不会无限期阻塞；三是通过setsockopt设置SO_RCVTIMEO，为recvfrom提供超时功能，避免在接收数据时阻塞。这些方法有助于优化网络通信的效率和稳定性。

一、使用SIGALRM为connect 设置超时

signal 捕捉到alarm信号之后会中断connect函数，导致返回值为EINTR，因此可提前设置connect超时返回，而不用等到最长 75s的connect原始超时时长

1、Signal 返回值为 typede void Sigfunc(int); -- 可以保存和恢复旧的信号处理函数

2、alarm(); 可覆盖之前的定时，返回值为之前定时的剩余时间，返回0表示之前未设置定时

3、connect() 返回 < 0，设置为失败,超时; 若成功, == 0；connect失败后需关闭close(sockfd);并重新 socket() 获取套接字

/* include connect_timeo */
#include    "unp.h"

static void connect_alarm(int);

int
connect_timeo(int sockfd, const SA *saptr, socklen_t salen, int nsec)
{
    Sigfunc *sigfunc;
    int     n;  

    sigfunc = Signal(SIGALRM, connect_alarm);
    if (alarm(nsec) != 0)
        err_msg("connect_timeo: alarm was already set");

    if ( (n = connect(sockfd, saptr, salen)) < 0) {
        close(sockfd);
        if (errno == EINTR)
            errno = ETIMEDOUT;
    }   
    alarm(0);                   /* turn off the alarm */
    Signal(SIGALRM, sigfunc);   /* restore previous signal handler */

    return(n);
}

static void
connect_alarm(int signo)
{
    return;     /* just interrupt the connect() */
}
/* end connect_timeo */

void
Connect_timeo(int fd, const SA *sa, socklen_t salen, int sec)
{
    if (connect_timeo(fd, sa, salen, sec) < 0)
        err_sys("connect_timeo error");             // 出错或者超时 都 < 0

    // 连接成功 connect_timeo 返回 0
}

二、使用select为accept 设置超时

fd_set fdset;
struct timeval tv;

FD_ZERO(&fdset);
FD_SET(info->sockfd, &fdset);
    
tv.tv_sec = 9;
tv.tv_usec = 0;

iRet = select(info->sockfd + 1, &fdset, NULL, NULL, &tv);
if (iRet > 0)
{
    /*成功则为非负描述符*/
    connfd = accept(info->sockfd, (struct sockaddr *)&cliAddr, &iAddrLen);
    if (connfd < 0)
    {
        printf("[%s %d] accept failed. \r\n", __FILE__, __LINE__);
        return -1;
    }
    else
    {
        info->connfd = connfd;
        printf("[%s %d] accept succ. connfd = %d\r\n", __FILE__, __LINE__, connfd);
        return 0;
    }
}
else
{
    printf("[%s %d] select timeout or Failed\r\n", __FILE__, __LINE__);
    return -1;
}

三、使用setsockopt设置SO_RCVTIMEO为recvfrom设置超时

errno == EWOULDBLOCK

#include    "unp.h"

void
dg_cli(FILE *fp, int sockfd, const SA *pservaddr, socklen_t servlen)
{
    int             n;  
    char            sendline[MAXLINE], recvline[MAXLINE + 1]; 
    struct timeval  tv; 

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

    while (Fgets(sendline, MAXLINE, fp) != NULL) {

        Sendto(sockfd, sendline, strlen(sendline), 0, pservaddr, servlen);

        n = recvfrom(sockfd, recvline, MAXLINE, 0, NULL, NULL);
        if (n < 0) {
            if (errno == EWOULDBLOCK) {
                fprintf(stderr, "socket timeout\n");
                continue;
            } else
                err_sys("recvfrom error");
        }   

        recvline[n] = 0;    /* null terminate */
        Fputs(recvline, stdout);
    }   
}