Sometimes, the connect time-out can take too much time when the target is unavailable. To resolve this issue, we can use non-blocking socket mode to select the timeout.
bool connect(char *host,int port, int timeout)
{
TIMEVAL Timeout;
Timeout.tv_sec = timeout;
Timeout.tv_usec = 0;
struct sockaddr_in address;
sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
address.sin_addr.s_addr = inet_addr(host);
address.sin_port = htons(port);
address.sin_family = AF_INET;
unsigned long iMode = 1;
int iResult = ioctlsocket(sock, FIONBIO, &iMode);
if (iResult != NO_ERROR)
{
printf("ioctlsocket failed with error: %ld\n", iResult);
}
if(connect(sock,(struct sockaddr *)&address,sizeof(address))==false)
{
return false;
}
iMode = 0;
iResult = ioctlsocket(sock, FIONBIO, &iMode);
if (iResult != NO_ERROR)
{
printf("ioctlsocket failed with error: %ld\n", iResult);
}
fd_set Write, Err;
FD_ZERO(&Write);
FD_ZERO(&Err);
FD_SET(sock, &Write);
FD_SET(sock, &Err);
select(0,NULL,&Write,&Err,&Timeout);
if(FD_ISSET(sock, &Write))
{
return true;
}
return false;
}