基于select实现socket的超时功能

今天遇到一个要求,套接字连上之后,由于在recv数据时就阻塞住了,一般会要求等待5s之后假如超时就返回失败,通过select实现了一下要求:

#define BUF_SIZE 128
#define RESPONSE_TIMEOUT 5

int ReciveResponse(int sfd, void *out_receive_data, int receive_data_size) {
    if (!out_receive_data || receive_data_size == 0) { 
        perror("No data to read!");
        return 0;
    }
    char *receive_buf = (char *)out_receive_data;
    fd_set readfds;
    struct timeval tv;
    int selectRt;
    int bytes = 0;

    FD_ZERO(&readfds);
    FD_SET(sfd, &readfds);
    tv.tv_sec = 1;
    tv.tv_usec = 5000;

    time_t start = time(NULL);
    while (time(NULL) - start < RESPONSE_TIMEOUT) {
        selectRt = select(sfd + 1, &readfds, NULL, NULL, &tv);
        if (-1 == selectRt) {
            perror("Select failed to open!");
            break;
        } else if (0 == selectRt) {
            continue;
        } else {
            if (-1 == (bytes = recv(sfd, out_receive_data, sizeof(out_receive_data), 0))) {
                perror("recv failed!");
                break;
            }
            receive_buf[bytes] = 0;
        }
    }
    return bytes;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
C++中使用socket进行网络编程时,可以通过设置超时时间来控制网络操作的等待时间。超时时间是指在进行网络操作(如连接、发送、接收等)时,如果在指定的时间内没有完成操作,则会触发超时。 在C++中,可以使用以下方法来设置socket超时时间: 1. 使用`setsockopt`函数设置超时时间: ```cpp #include <sys/types.h> #include <sys/socket.h> #include <netinet/in.h> #include <arpa/inet.h> #include <unistd.h> #include <fcntl.h> #include <errno.h> int setSocketTimeout(int sockfd, int timeout) { struct timeval tv; tv.tv_sec = timeout; tv.tv_usec = 0; if (setsockopt(sockfd, SOL_SOCKET, SO_RCVTIMEO, (const char*)&tv, sizeof(tv)) < 0) { perror("setsockopt"); return -1; } if (setsockopt(sockfd, SOL_SOCKET, SO_SNDTIMEO, (const char*)&tv, sizeof(tv)) < 0) { perror("setsockopt"); return -1; } return 0; } ``` 上述代码中,`setSocketTimeout`函数用于设置socket的接收和发送超时时间,`sockfd`为socket文件描述符,`timeout`为超时时间(单位为秒)。该函数使用`setsockopt`函数设置`SO_RCVTIMEO`和`SO_SNDTIMEO`选项,分别用于设置接收和发送超时时间。 2. 使用`select`函数实现超时: ```cpp #include <sys/types.h> #include <sys/socket.h> #include <netinet/in.h> #include <arpa/inet.h> #include <unistd.h> #include <fcntl.h> #include <errno.h> int socketTimeout(int sockfd, int timeout) { fd_set fds; FD_ZERO(&fds); FD_SET(sockfd, &fds); struct timeval tv; tv.tv_sec = timeout; tv.tv_usec = 0; int ret = select(sockfd + 1, &fds, NULL, NULL, &tv); if (ret == -1) { perror("select"); return -1; } else if (ret == 0) { printf("Timeout\n"); return -1; } return 0; } ``` 上述代码中,`socketTimeout`函数使用`select`函数实现超时。`sockfd`为socket文件描述符,`timeout`为超时时间(单位为秒)。该函数将socket文件描述符添加到`fd_set`集合中,然后使用`select`函数进行等待,如果在指定的超时时间内有数据可读,则返回0,否则返回-1。 需要注意的是,以上方法只能设置socket的读写超时时间,无法设置连接超时时间。如果需要设置连接超时时间,可以使用非阻塞socket和`connect`函数的超时机制。 相关问题: 1. 什么是socket超时? 2. 如何在C++设置socket超时时间? 3. 什么是非阻塞socket? 4. 如何使用非阻塞socket和connect函数的超时机制实现连接超时
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值