C connect实现Timeout效果（Windows）

Posted 2020-10-16 离水的鱼

int SocketClient::connectTimeOutForWin(SOCKET &connect_fd, const int &timeout, const sockaddr_in &server)
{
    int error=-1, len;
    len = sizeof(int);
    timeval tm;
    fd_set set;
    unsigned long ul = 1;

     //设置为非阻塞模式

    ioctlsocket(connect_fd,FIONBIO,&ul);

    bool ret = false;
    if( connect(connect_fd, (struct sockaddr *)&server, sizeof(struct sockaddr)) == -1)
    {
        tm.tv_sec = timeout;
        tm.tv_usec = 0;
        FD_ZERO(&set);
        FD_SET(connect_fd, &set);
        if( select(-1, NULL, &set, NULL, &tm) > 0)
        {

              getsockopt(connect_fd, SOL_SOCKET, SO_ERROR, (char*)&error, &len);

            if(error == 0) ret = true;
            else ret = false;
        } else ret = false;
    }
    else ret = true;
    ul = 0;


//设置为阻塞模式
    ioctlsocket(connect_fd,FIONBIO,&ul);

    if(!ret){
        closesocket(connect_fd);
//        fprintf(stderr,"Connect server fail in WIN\n");
    }
    return ret;
}