Linux使用socket选项的定时器

socket选项SO_RCVTIMEO和SO_SNDTIMEO

这个选项是专门针对socket系列的函数的，函数在下图中说明了。如果超时，就设置错误码，之后在流程中根据错误码判断是否超时。

先给出超时的错误码：

直接给出代码实例：

#include <stdio.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <errno.h>
#include <fcntl.h>
#include <unistd.h>
#include <string.h>
#include <strings.h>
#include <assert.h>

int time_connect (const char* ip, int port, int time) {
    int ret = 0;

    struct sockaddr_in address;
    bzero (&address, sizeof (address) );
    address.sin_family = AF_INET;
    assert (inet_pton (AF_INET, ip, &address.sin_addr) != -1);
    address.sin_port = htons (port);

    int sockfd = socket(AF_INET, SOCK_STREAM, 0);  // 非阻塞模式
    assert(sockfd > 0);


    struct timeval timeout;
    timeout.tv_sec = time;
    timeout.tv_usec = 0;
    socklen_t len = sizeof(timeout);
    // 这里设置超时机制
    ret = setsockopt(sockfd, SOL_SOCKET, SO_SNDTIMEO, &timeout, len);
    assert(ret != -1);

    ret = connect(sockfd, (struct sockaddr*)&address, sizeof(address));
    if (ret < 0) {
        // 超时对应的错误码
        if (errno == EINPROGRESS) {
            perror("connecting timeout, process timeout logic");
        } else {
            perror("connect() error\n");
        }

        return -1;
    }

    return sockfd;
}

int main(int argc, char* argv[]) {
    if (argc != 3) {
        printf("Usage: %s <ip of server> <port of server>\n", argv[0]);
        return 1;
    }

    const char* ip = argv[1];
    int port = atoi(argv[2]);
    if (port < 1024 || port > 65535) {
        perror("port error\n");
        return 1;
    }

    int sockfd = time_connect(ip, port, 5);
    if (sockfd < 0) {
        return 1;
    }

    return 0;
}