connect_nonb.lc

unix网络编程卷1：套接口API的全书源码

#include    "unp.h"##  1 ##src/lib/connect_nonb.c##

int##  2 ##src/lib/connect_nonb.c##
connect_nonb(int sockfd, const SA *saptr, socklen_t salen, int nsec)##  3 ##src/lib/connect_nonb.c##
{##  4 ##src/lib/connect_nonb.c##
    int     flags, n, error;##  5 ##src/lib/connect_nonb.c##
    socklen_t len;##  6 ##src/lib/connect_nonb.c##
    fd_set  rset, wset;##  7 ##src/lib/connect_nonb.c##
    struct timeval tval;##  8 ##src/lib/connect_nonb.c##

    flags = Fcntl(sockfd, F_GETFL, 0);##  9 ##src/lib/connect_nonb.c##
    Fcntl(sockfd, F_SETFL, flags | O_NONBLOCK);## 10 ##src/lib/connect_nonb.c##

    error = 0;## 11 ##src/lib/connect_nonb.c##
    if ((n = connect(sockfd, saptr, salen)) < 0)## 12 ##src/lib/connect_nonb.c##
        if (errno != EINPROGRESS)## 13 ##src/lib/connect_nonb.c##
            return (-1);## 14 ##src/lib/connect_nonb.c##

    /* Do whatever we want while the connect is taking place. */## 15 ##src/lib/connect_nonb.c##

    if (n == 0)## 16 ##src/lib/connect_nonb.c##
        goto done;              /* connect completed immediately */## 17 ##src/lib/connect_nonb.c##

    FD_ZERO(&rset);## 18 ##src/lib/connect_nonb.c##
    FD_SET(sockfd, &rset);## 19 ##src/lib/connect_nonb.c##
    wset = rset;## 20 ##src/lib/connect_nonb.c##
    tval.tv_sec = nsec;## 21 ##src/lib/connect_nonb.c##
    tval.tv_usec = 0;## 22 ##src/lib/connect_nonb.c##

    if ((n = Select(sockfd + 1, &rset, &wset, NULL,## 23 ##src/lib/connect_nonb.c##
                    nsec ? &tval : NULL)) == 0) {## 24 ##src/lib/connect_nonb.c##
        close(sockfd);          /* timeout */## 25 ##src/lib/connect_nonb.c##
        errno = ETIMEDOUT;## 26 ##src/lib/connect_nonb.c##
        return (-1);## 27 ##src/lib/connect_nonb.c##
    }## 28 ##src/lib/connect_nonb.c##

    if (FD_ISSET(sockfd, &rset) || FD_ISSET(sockfd, &wset)) {## 29 ##src/lib/connect_nonb.c##
        len = sizeof(error);## 30 ##src/lib/connect_nonb.c##
        if (getsockopt(sockfd, SOL_SOCKET, SO_ERROR, &error, &len) < 0)## 31 ##src/lib/connect_nonb.c##
            return (-1);        /* Solaris pending error */## 32 ##src/lib/connect_nonb.c##
    } else## 33 ##src/lib/connect_nonb.c##
        err_quit("select error: sockfd not set");## 34 ##src/lib/connect_nonb.c##

  done:## 35 ##src/lib/connect_nonb.c##
    Fcntl(sockfd, F_SETFL, flags);  /* restore file status flags */## 36 ##src/lib/connect_nonb.c##

    if (error) {## 37 ##src/lib/connect_nonb.c##
        close(sockfd);          /* just in case */## 38 ##src/lib/connect_nonb.c##
        errno = error;## 39 ##src/lib/connect_nonb.c##
        return (-1);## 40 ##src/lib/connect_nonb.c##
    }## 41 ##src/lib/connect_nonb.c##
    return (0);## 42 ##src/lib/connect_nonb.c##
}## 43 ##src/lib/connect_nonb.c##