最近在做一个在开发板启动后,程序运行前首先检查一下网络是否已经连通的东西。参考了网上的一些方法,现在整理一下。(使用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; }
|
阅读(8815) | 评论(2) | 转发(0) |