1.connect函数的超时时间设置只对TCP有效 UDP由于是无连接的connect都会返回success

有两种方法:

第一种方法 默认的socket是阻塞模式 我们只需要设置其为非阻塞模式,然后调用select去查询其状态 代码如下: 

static int connect_server(char *ip, int port)
{
    int sockfd = -1;
    struct sockaddr_in servaddr;
    int flags = 0;

    if ((sockfd = socket(AF_INET, SOCK_STREAM , 0)) < 0)
    {
        printf("create socket error: %s(errno: %d)\n", strerror(errno), errno);
        return sockfd;
    }

    memset(&servaddr, 0, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(port);
    if (inet_pton(AF_INET, ip, &servaddr.sin_addr) <= 0)
    {
        printf("inet_pton error for %s\n", ip);
        close(sockfd);
        sockfd = -1;
        return sockfd;
    }

    //把链路设置为非阻塞
    flags = fcntl(sockfd, F_GETFL, 0);
    fcntl(sockfd, F_SETFL, flags | O_NONBLOCK);
    
    if (connect(sockfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) < 0)
    {
        if (errno != EINPROGRESS)       /* EINPROGRESS 表示连接正在建立的过程中 */
        {
            printf("connect error: %s(errno: %d)\n", strerror(errno), errno);
            close(sockfd);
            sockfd = -1;
        }
        else
        {
            int ret;
            fd_set write_fds;
            struct timeval timeout;
            
            timeout.tv_sec = 0;
	        timeout.tv_usec = 100 * 1000;       /* 连接超时时长:100ms */

            FD_ZERO(&write_fds);
            FD_SET(sockfd, &write_fds);
  
            ret = select(sockfd + 1, NULL, &write_fds, NULL, &timeout);
            switch (ret)
            {
                case -1:        /* select错误 */
                {
                    printf("connect error: %s(errno: %d)\n", strerror(errno), errno);
                    close(sockfd);
                    sockfd = -1;
                    break;
                }
                case 0:         /* 超时 */
                {
                    FK_TRACE_WARN("select timeout...\n");
                    close(sockfd);
                    sockfd = -1;
                    break;
                }
                default:
                {
                    int error = -1;
                    socklen_t optLen = sizeof(socklen_t);
                    
                    getsockopt(sockfd, SOL_SOCKET, SO_ERROR, (char *)&error, &optLen);       /* 通过 getsockopt 替代 FD_ISSET 判断是否连接 */
                    if (error != 0)
                    {
                        printf("connect error: %s(errno: %d)\n", strerror(errno), errno);
                        close(sockfd);
                        sockfd = -1;
                    }
                    
                    break;
                }
            }
        }
    }

    return sockfd;
}

第二种是 默认其为阻塞模式  通过setsockopt 函数设置TCP_SYNCNT 值 头文件如下 

表示失败是tcp重连的次数 每次重连时间会翻倍 依次是 1 2 4 8 16 32 64 ....

1+2+4+8+16+32+64

实测发现 这个值是从0开始的  TCP_SYNCNT 值为2 那么 就是超时时间就是1+2+4 = 7秒

如果是3 就是1+2+4+8 = 15秒

#include <netinet/tcp.h>
#define TCP_SYNCNT 7

设置方法如下:

//Add tcp connect timeout processing
	printf("QMCY 20230113 add tcp connect time\n");
	int syncnt = 2; //1+2+4+8+16+32+64
	setsockopt(FDevSocket, IPPROTO_TCP, TCP_SYNCNT, &syncnt, sizeof(syncnt));

2.设置socket发送和接收超时时间 对TCP/UDP都有效

代码如下:

timeout.tv_sec =5;
	timeout.tv_usec = 0;

    if (setsockopt(socket_fd, SOL_SOCKET, SO_RCVTIMEO, (char*)&timeout, sizeof(timeout)) == -1) 
	{
		printf("setsockopt:%08x  SO_RCVTIMEO error is %d ",
			socket_fd,SOCKET_ERRNO());						
		goto ERROR;

    }

    if (setsockopt(socket_fd, SOL_SOCKET, SO_SNDTIMEO, (char*)&timeout, sizeof(timeout)) == -1) 
	{
			printf("setsockopt:%08x  SO_SNDTIMEO error is %d ",
				socket_fd,SOCKET_ERRNO());						
			goto ERROR;
	}

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐