在使用TCP的connect连接服务器时,在默认情况下系统使用的是阻塞式socket,如果服务器当前不可用,则connect会等待知道超时时间到达,而这个超时时间是系统内核规定的,并不能使用setSocketOpt来设置,这个函数只能设置send和recv的超时,为了能够随意控制connect的超时时间,可以使用select。大致的过程就是先将socket设置成非阻塞,使用select去轮询套接口,再根据套接口去判断连接状态。
int connectServer(int sock_fd,unsigned int port,char* ip) { struct sockaddr_in servaddr; bzero(&servaddr, sizeof(servaddr)); servaddr.sin_family = AF_INET; servaddr.sin_port = htons(port); inet_aton(ip, &servaddr.sin_addr ); fcntl(sock_fd,F_SETFL,fcntl(sock_fd,F_GETFL,0)|O_NONBLOCK); int connected = connect(sock_fd, (struct sockaddr*)&servaddr, sizeof(struct sockaddr_in)); int ret = -1; if (connected != 0 ) { if(errno != EINPROGRESS) printf("connect error :%s ",strerror(errno)); else { struct timeval tm = {2, 0}; fd_set wset,rset; FD_ZERO(&wset); FD_ZERO(&rset); FD_SET(sock_fd,&wset); FD_SET(sock_fd,&rset); long t1 = time(NULL); int res = select(sock_fd+1,&rset,&wset,NULL,&tm); long t2 = time(NULL); printf("interval time: %ld ", t2 - t1); if(res < 0) { printf("network error in connect "); } else if(res == 0) { printf("connect time out "); } else if (1 == res) { if(FD_ISSET(sock_fd,&wset)) { printf("connect succeed. "); fcntl(sock_fd,F_SETFL,fcntl(sock_fd,F_GETFL,0) & ~O_NONBLOCK); ret = 0; } else { printf("other error when select:%s ",strerror(errno)); } } } } return ret; }
程序先把socket设置成非阻塞,connect在非阻塞模式下会立刻返回,如果没有其他错误,返回值等于0。当connect不能立刻建立连接时,会返回一个EINPROGRESS,表示连接正在建立的过程中,这时我们可以使用select去轮询套接口,而select的轮询超时时间可以根据自己的需要去设置,最主要的是轮询的集合一定要是读和写的集合,即select的第二和第三个参数要赋值,待select返回就可以去判断返回值来确定connect的进程状态了。如果返回值小于0,说明connect的进程出现了错误,如果是等于0则说明connect超时,如果等于1,并且套接口此时的状态是可写,则说明了connect已经成功建立(关于这点大概是因为服务器接收了连接后,不会立刻想socket写数据,这是客户端就只能轮询到可写的socket,我觉得如果服务器接受连接并立刻写数据,在客户端就可能是返回2,这时学要同时判断socket的可读和可写了);其他情况的话就算是其他错误吧,至此,我们只需要设置select的超时值就可以随心所欲地实现自己想要的connect连接超时了。
最后,别忘了把套接口设置会阻塞状态,毕竟阻塞状态加线程方便控制。