Linux非阻塞IO（四）非阻塞IO中connect的实现

我们为客户端的编写再做一些工作。

这次我们使用非阻塞IO实现connect函数。

int connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen);

非阻塞IO有以下用处：

1.将三次握手的处理过程生下来，处理其他事情。

2.使用这个同时建立多个连接。

3.实现超时connect功能，本节实现的connect就可以指定时间，超时后算作错误处理。

 

在阻塞IO中，调用connect后一般会阻塞，直到确定连接成功或者失败。

在Non-Blocking IO中，connect往往会立刻返回，此时connect就有两种结果。

一是连接成功

二是返回-1，errno置为EINPROGRESS，这种一般是因为网络延迟，所以连接不能马上建立，我们需要使用poll来监听sockfd。

所以接下来我们需要向poll注册sockfd的写事件。

《TCP/IP详解》第二卷指出以下的一些规则：

当连接建立时，sockfd可写。

当遇到错误时，sockfd既可读又可写。

我们设置一个超时时间，当poll返回时，如果sockfd可写，此时有两种情况：

一是连接确实建立成功

二是连接发生错误

我们需要某些手段辨别究竟是否发生了错误。

这里我们采用socket选项，检测的是SO_ERROR，代码如下：

int get_sockfd_error(int sockfd)

{

    int err;

    socklen_t socklen = sizeof(err);

    if(getsockopt(sockfd, SOL_SOCKET, SO_ERROR, &err, &socklen) == -1)

        ERR_EXIT("getsock_error  error");

    if (err == 0)

        return 0; //无错误

    else

        return err;

}

如果sockfd无错误，则返回0，否则返回错误代码。

 

在非阻塞connect的实现中，我们通常需要先把fd设置为非阻塞，最后再重新设置为阻塞。这样做，是为了满足阻塞与非阻塞的需求。

因为在阻塞IO中，有时候也会使用非阻塞connect。

 

实现代码如下：

int nonblocking_connect(int sockfd, const char *des_host, uint16_t des_port, int timeout)

{

    if(des_host == NULL)

    {

        fprintf(stderr, "des_host can not be NULL\n");

        exit(EXIT_FAILURE);

    }



    SAI servaddr;

    memset(&servaddr, 0, sizeof servaddr);

    servaddr.sin_family = AF_INET;

    servaddr.sin_port = htons(des_port);

    if(inet_aton(des_host, &servaddr.sin_addr) == 0)

    {

        //DNS

        //struct hostent *gethostbyname(const char *name);

        struct hostent *hp = gethostbyname(des_host);

        if(hp == NULL)

            ERR_EXIT("gethostbyname");

        servaddr.sin_addr = *(struct in_addr*)hp->h_addr;

    }



    //设置为非阻塞

    activate_nonblock(sockfd); 



    //connect会立刻返回

    int ret = connect(sockfd, (SA*)&servaddr, sizeof servaddr);

    if(ret == -1)

    {

        if(errno != EINPROGRESS) //连接失败

            ERR_EXIT("connect error");

        struct pollfd pfd[1];

        pfd[0].fd = sockfd;

        pfd[0].events = POLLOUT;



        ret = poll(pfd, 1, timeout);



        if(ret == 0)

        {

            errno = ETIMEDOUT;

            ret = -1; //连接超时，此时判定为失败

        }

        //sockfd可写，此时需要检查套接字选项，查看是否发生错误

        else if(ret == 1 && pfd[0].revents & (POLLOUT | POLLWRBAND))

        {

            int err;

            //检查sockfd错误

            if((err = get_sockfd_error(sockfd)))

            {

                errno = err;

                return -1;

            }

        }

    }



    //重新设置为阻塞

    deactivate_nonblock(sockfd);

    return ret;

}

 

读者可以自行测试。

 

下节使用非阻塞connect实现客户端。