Linux网络编程 — 设置connect函数连接超时

时间：2009-07-08 来源：_mystic

    最近在做一个在开发板启动后，程序运行前首先检查一下网络是否已经连通的东西。参考了网上的一些方法，现在整理一下。(使用system(ping IPaddress)也可以办到)     其实问题的关键是，如何设置connect函数能够在指定的时间内返回。如果直接connect一个IP地址，不成功的话将会是一个漫长的等待时间。测试了一下2分钟内没有反映，但是网上有人说是75秒,还需要进一步确认啊！过程如下：     (1)调用fcntl，将socket置为非阻塞模式;
    (2)connect();
    (3)判断connect()的返回值,一般情况会返回-1,这时你还必须判断错误码如果是EINPROGRESS,那说明connect还在继续;如果错误码不是前者那么就是有问题了,不必往下执行,必须关掉socket;待下次重联;
    (4)select();设置好函数中的超时时间,将select()中的read和write置位,在超时时间内,如果select返回1,即描述字变为了可写,那么连接成功。如果返回2,即描述字变为即可读又可写,那么出错。如果返回0,那么超时。代码如下：

#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <unistd.h>
#include <sys/time.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <errno.h>

#define TIMEOUT 5
#define SERV_PORT 53

int main(int argc, char *argv[])
{
    if(argc != 2)
    {
        printf("Usage: client xxx.xxx.xxx.xxx\nExample: client 192.168.0.1\n");
        exit(1);
    }

    int sockfd, flags, res;
    struct sockaddr_in servaddr;
    fd_set fdr, fdw;
    struct timeval timeout;

    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if(sockfd < 0) {
        perror("Netwrok test...\n");
        return -1;
    }

    /* set socket fd noblock */
    if((flags = fcntl(sockfd, F_GETFL, 0)) < 0) {
        perror("Netwrok test...\n");
        close(sockfd);
        return -1;
    }

    if(fcntl(sockfd, F_SETFL, flags | O_NONBLOCK) < 0) {
        perror("Network test...\n");
        close(sockfd);
        return -1;
    }

    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    inet_pton(AF_INET, argv[1], &servaddr.sin_addr);
    servaddr.sin_port = htons(SERV_PORT);

    if(connect(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr)) != 0) {
        if(errno != EINPROGRESS) { // EINPROGRESS

            perror("Network test...\n");
            close(sockfd);
            return -1;
        }
    }
    else {
        printf("Connected\n");
        return 0;
    }

    FD_ZERO(&fdr);
    FD_ZERO(&fdw);
    FD_SET(sockfd, &fdr);
    FD_SET(sockfd, &fdw);

    timeout.tv_sec = TIMEOUT;
    timeout.tv_usec = 0;

    res = select(sockfd + 1, &fdr, &fdw, NULL, &timeout);
    if(res < 0) {
        perror("Network test...\n");
        close(sockfd);
        return -1;
    }
    if(res == 0) {
        printf("Connect server timeout")
        close(sockfd);
        return -1;
    }
    if(res == 1) {
        if(FD_ISSET(sockfd, &fdw))
        {
            printf("Connected\n");
            close(sockfd);
            return 0;
        }

    }
    /* Not necessary */
    if(res == 2) {
        printf("Connect server timeout");
        close(sockfd);
        return -1;
    }

    printf("Connect server timeout");
    close(sockfd);
    return -1;
}