connect设置超时的方法

在使用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\n",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\n", t2 - t1);
			if(res < 0)
			{
				printf("network error in connect\n");
			}
			else if(res == 0)
			{
				printf("connect time out\n");
			}
			else if (1 == res)
			{
				if(FD_ISSET(sock_fd,&wset))
				{
					printf("connect succeed.\n");
					fcntl(sock_fd,F_SETFL,fcntl(sock_fd,F_GETFL,0) & ~O_NONBLOCK);
					ret = 0;
				}
				else
				{
					printf("other error when select:%s\n",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连接超时了。

最后,别忘了把套接口设置会阻塞状态,毕竟阻塞状态加线程方便控制。


 

阅读更多

如何设置connect超时

06-23

代码如下:rnrn[code=C/C++]// 装入Winsock然后连接服务器rn__declspec (dllexport) int LoadWinsockM(HWND hWnd,char *szIP,HWND m_child_wnd)rnrn WSADATA wsd;rn DWORD dwLen;rn int nRet,nZero;rn LPBYTE pBuf;rn WSAPROTOCOL_INFO Protocol;rn char szMessage[81];rn if (WSAStartup(0x202,&wsd) != 0)rn rn MessageBox(NULL,"hehe","Client Socket Error",MB_OK);rn return 0;rn rn dwLen = 0;rn nRet = WSAEnumProtocols(NULL,NULL,&dwLen);rn if (nRet == SOCKET_ERROR)rn rn if (WSAGetLastError() != WSAENOBUFS)rn return 0;rn rn pBuf = malloc(dwLen);rn nRet = SelectProtocols(SETFLAGS,NOTSETFLAGS,(LPWSAPROTOCOL_INFO)pBuf,&dwLen,&Protocol);rn free(pBuf);rn m_data[m_index].sClient = WSASocket(AF_INET,SOCK_STREAM,IPPROTO_IP,NULL,0,SOCK_STREAM);rn if (m_data[m_index].sClient == INVALID_SOCKET)rn return 1;rn nZero = 0;rn setsockopt(m_data[m_index].sClient,SOL_SOCKET,SO_RCVBUF,(char *)&nZero,sizeof(nZero));rn m_data[m_index].server.sin_family = AF_INET;rn m_data[m_index].server.sin_port = htons(m_data[m_index].port);rn m_data[m_index].server.sin_addr.s_addr = inet_addr(szIP);rnrn if (connect(m_data[m_index].sClient,(struct sockaddr *)&m_data[m_index].server,sizeof(m_data[m_index].server)) == SOCKET_ERROR)rn rn int i=GetLastError();rn return 0;rn rnrn GetResolution();rn SetGrids(hWnd);rnrn memset(szMessage,'\0',sizeof(szMessage));rn sprintf(szMessage,"WM_COMP;%d;0;0;0;\0",10);rn SendCommandM(hWnd,1,szMessage,NULL);rnrn UpdateRegionalScreen(hWnd,m_data[m_index].sClient,TRUE,m_child_wnd);rnrn return m_data[m_index].sClient;rnrn[/code]rnrn其中rn if (connect(m_data[m_index].sClient,(struct sockaddr *)&m_data[m_index].server,sizeof(m_data[m_index].server)) == SOCKET_ERROR)rn这一句,如果要链接的机子关机了,也就是说是个无效IP,这个地方会一直等下去,直到windows认为连不上了为止,rnrn我要问的就是,能不能在这之前设置一下超时呢?rn我想超过5秒就不让他等了,让它运行下面的代码,怎么设置这个时间呢?用什么API?rn

没有更多推荐了,返回首页