123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905 |
- /***************************************************************************
- * _ _ ____ _
- * Project ___| | | | _ \| |
- * / __| | | | |_) | |
- * | (__| |_| | _ <| |___
- * \___|\___/|_| \_\_____|
- *
- * Copyright (C) 1998 - 2006, Daniel Stenberg, <daniel@haxx.se>, et al.
- *
- * This software is licensed as described in the file COPYING, which
- * you should have received as part of this distribution. The terms
- * are also available at http://curl.haxx.se/docs/copyright.html.
- *
- * You may opt to use, copy, modify, merge, publish, distribute and/or sell
- * copies of the Software, and permit persons to whom the Software is
- * furnished to do so, under the terms of the COPYING file.
- *
- * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
- * KIND, either express or implied.
- *
- * $Id$
- ***************************************************************************/
- #include "setup.h"
- #ifndef WIN32
- /* headers for non-win32 */
- #ifdef HAVE_SYS_TIME_H
- #include <sys/time.h>
- #endif
- #ifdef HAVE_SYS_TYPES_H
- #include <sys/types.h>
- #endif
- #ifdef HAVE_SYS_SOCKET_H
- #include <sys/socket.h>
- #endif
- #ifdef HAVE_NETINET_IN_H
- #include <netinet/in.h> /* <netinet/tcp.h> may need it */
- #endif
- #ifdef HAVE_NETINET_TCP_H
- #include <netinet/tcp.h> /* for TCP_NODELAY */
- #endif
- #ifdef HAVE_SYS_IOCTL_H
- #include <sys/ioctl.h>
- #endif
- #ifdef HAVE_UNISTD_H
- #include <unistd.h>
- #endif
- #ifdef HAVE_NETDB_H
- #include <netdb.h>
- #endif
- #ifdef HAVE_FCNTL_H
- #include <fcntl.h>
- #endif
- #ifdef HAVE_NETINET_IN_H
- #include <netinet/in.h>
- #endif
- #ifdef HAVE_ARPA_INET_H
- #include <arpa/inet.h>
- #endif
- #ifdef HAVE_STDLIB_H
- #include <stdlib.h> /* required for free() prototype, without it, this crashes
- on macos 68K */
- #endif
- #if (defined(HAVE_FIONBIO) && defined(__NOVELL_LIBC__))
- #include <sys/filio.h>
- #endif
- #if (defined(NETWARE) && defined(__NOVELL_LIBC__))
- #undef in_addr_t
- #define in_addr_t unsigned long
- #endif
- #ifdef VMS
- #include <in.h>
- #include <inet.h>
- #endif
- #endif
- #include <stdio.h>
- #include <errno.h>
- #include <string.h>
- #ifndef TRUE
- #define TRUE 1
- #define FALSE 0
- #endif
- #ifdef USE_WINSOCK
- #define EINPROGRESS WSAEINPROGRESS
- #define EWOULDBLOCK WSAEWOULDBLOCK
- #define EISCONN WSAEISCONN
- #define ENOTSOCK WSAENOTSOCK
- #define ECONNREFUSED WSAECONNREFUSED
- #endif
- #include "urldata.h"
- #include "sendf.h"
- #include "if2ip.h"
- #include "strerror.h"
- #include "connect.h"
- #include "memory.h"
- #include "select.h"
- #include "url.h" /* for Curl_safefree() */
- #include "multiif.h"
- #include "sockaddr.h" /* required for Curl_sockaddr_storage */
- #include "inet_ntop.h"
- /* The last #include file should be: */
- #include "memdebug.h"
- static bool verifyconnect(curl_socket_t sockfd, int *error);
- static curl_socket_t
- singleipconnect(struct connectdata *conn,
- const Curl_addrinfo *ai, /* start connecting to this */
- long timeout_ms,
- bool *connected);
- /*
- * Curl_sockerrno() returns the *socket-related* errno (or equivalent) on this
- * platform to hide platform specific for the function that calls this.
- */
- int Curl_sockerrno(void)
- {
- #ifdef USE_WINSOCK
- return (int)WSAGetLastError();
- #else
- return errno;
- #endif
- }
- /*
- * Curl_nonblock() set the given socket to either blocking or non-blocking
- * mode based on the 'nonblock' boolean argument. This function is highly
- * portable.
- */
- int Curl_nonblock(curl_socket_t sockfd, /* operate on this */
- int nonblock /* TRUE or FALSE */)
- {
- #undef SETBLOCK
- #define SETBLOCK 0
- #ifdef HAVE_O_NONBLOCK
- /* most recent unix versions */
- int flags;
- flags = fcntl(sockfd, F_GETFL, 0);
- if (TRUE == nonblock)
- return fcntl(sockfd, F_SETFL, flags | O_NONBLOCK);
- else
- return fcntl(sockfd, F_SETFL, flags & (~O_NONBLOCK));
- #undef SETBLOCK
- #define SETBLOCK 1
- #endif
- #if defined(HAVE_FIONBIO) && (SETBLOCK == 0)
- /* older unix versions */
- int flags;
- flags = nonblock;
- return ioctl(sockfd, FIONBIO, &flags);
- #undef SETBLOCK
- #define SETBLOCK 2
- #endif
- #if defined(HAVE_IOCTLSOCKET) && (SETBLOCK == 0)
- /* Windows? */
- unsigned long flags;
- flags = nonblock;
- return ioctlsocket(sockfd, FIONBIO, &flags);
- #undef SETBLOCK
- #define SETBLOCK 3
- #endif
- #if defined(HAVE_IOCTLSOCKET_CASE) && (SETBLOCK == 0)
- /* presumably for Amiga */
- return IoctlSocket(sockfd, FIONBIO, (long)nonblock);
- #undef SETBLOCK
- #define SETBLOCK 4
- #endif
- #if defined(HAVE_SO_NONBLOCK) && (SETBLOCK == 0)
- /* BeOS */
- long b = nonblock ? 1 : 0;
- return setsockopt(sockfd, SOL_SOCKET, SO_NONBLOCK, &b, sizeof(b));
- #undef SETBLOCK
- #define SETBLOCK 5
- #endif
- #ifdef HAVE_DISABLED_NONBLOCKING
- return 0; /* returns success */
- #undef SETBLOCK
- #define SETBLOCK 6
- #endif
- #if (SETBLOCK == 0)
- #error "no non-blocking method was found/used/set"
- #endif
- }
- /*
- * waitconnect() waits for a TCP connect on the given socket for the specified
- * number if milliseconds. It returns:
- * 0 fine connect
- * -1 select() error
- * 1 select() timeout
- * 2 select() returned with an error condition fd_set
- */
- #define WAITCONN_CONNECTED 0
- #define WAITCONN_SELECT_ERROR -1
- #define WAITCONN_TIMEOUT 1
- #define WAITCONN_FDSET_ERROR 2
- static
- int waitconnect(curl_socket_t sockfd, /* socket */
- long timeout_msec)
- {
- int rc;
- #ifdef mpeix
- /* Call this function once now, and ignore the results. We do this to
- "clear" the error state on the socket so that we can later read it
- reliably. This is reported necessary on the MPE/iX operating system. */
- (void)verifyconnect(sockfd, NULL);
- #endif
- /* now select() until we get connect or timeout */
- rc = Curl_select(CURL_SOCKET_BAD, sockfd, (int)timeout_msec);
- if(-1 == rc)
- /* error, no connect here, try next */
- return WAITCONN_SELECT_ERROR;
- else if(0 == rc)
- /* timeout, no connect today */
- return WAITCONN_TIMEOUT;
- if(rc & CSELECT_ERR)
- /* error condition caught */
- return WAITCONN_FDSET_ERROR;
- /* we have a connect! */
- return WAITCONN_CONNECTED;
- }
- static CURLcode bindlocal(struct connectdata *conn,
- curl_socket_t sockfd)
- {
- struct SessionHandle *data = conn->data;
- struct sockaddr_in me;
- struct sockaddr *sock = NULL; /* bind to this address */
- socklen_t socksize; /* size of the data sock points to */
- unsigned short port = data->set.localport; /* use this port number, 0 for
- "random" */
- /* how many port numbers to try to bind to, increasing one at a time */
- int portnum = data->set.localportrange;
- /*************************************************************
- * Select device to bind socket to
- *************************************************************/
- if (data->set.device && (strlen(data->set.device)<255) ) {
- struct Curl_dns_entry *h=NULL;
- char myhost[256] = "";
- in_addr_t in;
- int rc;
- bool was_iface = FALSE;
- /* First check if the given name is an IP address */
- in=inet_addr(data->set.device);
- if((in == CURL_INADDR_NONE) &&
- Curl_if2ip(data->set.device, myhost, sizeof(myhost))) {
- /*
- * We now have the numerical IPv4-style x.y.z.w in the 'myhost' buffer
- */
- rc = Curl_resolv(conn, myhost, 0, &h);
- if(rc == CURLRESOLV_PENDING)
- (void)Curl_wait_for_resolv(conn, &h);
- if(h) {
- was_iface = TRUE;
- Curl_resolv_unlock(data, h);
- }
- }
- if(!was_iface) {
- /*
- * This was not an interface, resolve the name as a host name
- * or IP number
- */
- rc = Curl_resolv(conn, data->set.device, 0, &h);
- if(rc == CURLRESOLV_PENDING)
- (void)Curl_wait_for_resolv(conn, &h);
- if(h) {
- if(in == CURL_INADDR_NONE)
- /* convert the resolved address, sizeof myhost >= INET_ADDRSTRLEN */
- Curl_inet_ntop(h->addr->ai_addr->sa_family,
- &((struct sockaddr_in*)h->addr->ai_addr)->sin_addr,
- myhost, sizeof myhost);
- else
- /* we know data->set.device is shorter than the myhost array */
- strcpy(myhost, data->set.device);
- Curl_resolv_unlock(data, h);
- }
- }
- if(! *myhost) {
- /* need to fix this
- h=Curl_gethost(data,
- getmyhost(*myhost,sizeof(myhost)),
- hostent_buf,
- sizeof(hostent_buf));
- */
- failf(data, "Couldn't bind to '%s'", data->set.device);
- return CURLE_HTTP_PORT_FAILED;
- }
- infof(data, "Bind local address to %s\n", myhost);
- #ifdef SO_BINDTODEVICE
- /* I am not sure any other OSs than Linux that provide this feature, and
- * at the least I cannot test. --Ben
- *
- * This feature allows one to tightly bind the local socket to a
- * particular interface. This will force even requests to other local
- * interfaces to go out the external interface.
- *
- */
- if (was_iface) {
- /* Only bind to the interface when specified as interface, not just as a
- * hostname or ip address.
- */
- if (setsockopt(sockfd, SOL_SOCKET, SO_BINDTODEVICE,
- data->set.device, strlen(data->set.device)+1) != 0) {
- /* printf("Failed to BINDTODEVICE, socket: %d device: %s error: %s\n",
- sockfd, data->set.device, Curl_strerror(Curl_sockerrno())); */
- infof(data, "SO_BINDTODEVICE %s failed\n",
- data->set.device);
- /* This is typically "errno 1, error: Operation not permitted" if
- you're not running as root or another suitable privileged user */
- }
- }
- #endif
- in=inet_addr(myhost);
- if (CURL_INADDR_NONE == in) {
- failf(data,"couldn't find my own IP address (%s)", myhost);
- return CURLE_HTTP_PORT_FAILED;
- } /* end of inet_addr */
- if ( h ) {
- Curl_addrinfo *addr = h->addr;
- sock = addr->ai_addr;
- socksize = addr->ai_addrlen;
- }
- else
- return CURLE_HTTP_PORT_FAILED;
- }
- else if(port) {
- /* if a local port number is requested but no local IP, extract the
- address from the socket */
- memset(&me, 0, sizeof(struct sockaddr));
- me.sin_family = AF_INET;
- me.sin_addr.s_addr = INADDR_ANY;
- sock = (struct sockaddr *)&me;
- socksize = sizeof(struct sockaddr);
- }
- else
- /* no local kind of binding was requested */
- return CURLE_OK;
- do {
- /* Set port number to bind to, 0 makes the system pick one */
- if(sock->sa_family == AF_INET)
- ((struct sockaddr_in *)sock)->sin_port = htons(port);
- #ifdef ENABLE_IPV6
- else
- ((struct sockaddr_in6 *)sock)->sin6_port = htons(port);
- #endif
- if( bind(sockfd, sock, socksize) >= 0) {
- /* we succeeded to bind */
- struct Curl_sockaddr_storage add;
- socklen_t size;
- size = sizeof(add);
- if(getsockname(sockfd, (struct sockaddr *) &add, &size) < 0) {
- failf(data, "getsockname() failed");
- return CURLE_HTTP_PORT_FAILED;
- }
- /* We re-use/clobber the port variable here below */
- if(((struct sockaddr *)&add)->sa_family == AF_INET)
- port = ntohs(((struct sockaddr_in *)&add)->sin_port);
- #ifdef ENABLE_IPV6
- else
- port = ntohs(((struct sockaddr_in6 *)&add)->sin6_port);
- #endif
- infof(data, "Local port: %d\n", port);
- return CURLE_OK;
- }
- if(--portnum > 0) {
- infof(data, "Bind to local port %d failed, trying next\n", port);
- port++; /* try next port */
- }
- else
- break;
- } while(1);
- data->state.os_errno = Curl_sockerrno();
- failf(data, "bind failure: %s",
- Curl_strerror(conn, data->state.os_errno));
- return CURLE_HTTP_PORT_FAILED;
- }
- /*
- * verifyconnect() returns TRUE if the connect really has happened.
- */
- static bool verifyconnect(curl_socket_t sockfd, int *error)
- {
- bool rc = TRUE;
- #ifdef SO_ERROR
- int err = 0;
- socklen_t errSize = sizeof(err);
- #ifdef WIN32
- /*
- * In October 2003 we effectively nullified this function on Windows due to
- * problems with it using all CPU in multi-threaded cases.
- *
- * In May 2004, we bring it back to offer more info back on connect failures.
- * Gisle Vanem could reproduce the former problems with this function, but
- * could avoid them by adding this SleepEx() call below:
- *
- * "I don't have Rational Quantify, but the hint from his post was
- * ntdll::NtRemoveIoCompletion(). So I'd assume the SleepEx (or maybe
- * just Sleep(0) would be enough?) would release whatever
- * mutex/critical-section the ntdll call is waiting on.
- *
- * Someone got to verify this on Win-NT 4.0, 2000."
- */
- #ifdef _WIN32_WCE
- Sleep(0);
- #else
- SleepEx(0, FALSE);
- #endif
- #endif
- if( -1 == getsockopt(sockfd, SOL_SOCKET, SO_ERROR,
- (void *)&err, &errSize))
- err = Curl_sockerrno();
- #ifdef _WIN32_WCE
- /* Always returns this error, bug in CE? */
- if(WSAENOPROTOOPT==err)
- err=0;
- #endif
- if ((0 == err) || (EISCONN == err))
- /* we are connected, awesome! */
- rc = TRUE;
- else
- /* This wasn't a successful connect */
- rc = FALSE;
- if (error)
- *error = err;
- #else
- (void)sockfd;
- if (error)
- *error = Curl_sockerrno();
- #endif
- return rc;
- }
- CURLcode Curl_store_ip_addr(struct connectdata *conn)
- {
- char addrbuf[256];
- Curl_printable_address(conn->ip_addr, addrbuf, sizeof(addrbuf));
- /* save the string */
- Curl_safefree(conn->ip_addr_str);
- conn->ip_addr_str = strdup(addrbuf);
- if(!conn->ip_addr_str)
- return CURLE_OUT_OF_MEMORY; /* FAIL */
- #ifdef PF_INET6
- if(conn->ip_addr->ai_family == PF_INET6)
- conn->bits.ipv6 = TRUE;
- #endif
- return CURLE_OK;
- }
- /* Used within the multi interface. Try next IP address, return TRUE if no
- more address exists */
- static bool trynextip(struct connectdata *conn,
- int sockindex,
- bool *connected)
- {
- curl_socket_t sockfd;
- Curl_addrinfo *ai;
- /* first close the failed socket */
- sclose(conn->sock[sockindex]);
- conn->sock[sockindex] = CURL_SOCKET_BAD;
- *connected = FALSE;
- if(sockindex != FIRSTSOCKET)
- return TRUE; /* no next */
- /* try the next address */
- ai = conn->ip_addr->ai_next;
- while (ai) {
- sockfd = singleipconnect(conn, ai, 0L, connected);
- if(sockfd != CURL_SOCKET_BAD) {
- /* store the new socket descriptor */
- conn->sock[sockindex] = sockfd;
- conn->ip_addr = ai;
- Curl_store_ip_addr(conn);
- return FALSE;
- }
- ai = ai->ai_next;
- }
- return TRUE;
- }
- /*
- * Curl_is_connected() is used from the multi interface to check if the
- * firstsocket has connected.
- */
- CURLcode Curl_is_connected(struct connectdata *conn,
- int sockindex,
- bool *connected)
- {
- int rc;
- struct SessionHandle *data = conn->data;
- CURLcode code = CURLE_OK;
- curl_socket_t sockfd = conn->sock[sockindex];
- long allow = DEFAULT_CONNECT_TIMEOUT;
- long allow_total = 0;
- long has_passed;
- curlassert(sockindex >= FIRSTSOCKET && sockindex <= SECONDARYSOCKET);
- *connected = FALSE; /* a very negative world view is best */
- /* Evaluate in milliseconds how much time that has passed */
- has_passed = Curl_tvdiff(Curl_tvnow(), data->progress.t_startsingle);
- /* subtract the most strict timeout of the ones */
- if(data->set.timeout && data->set.connecttimeout) {
- if (data->set.timeout < data->set.connecttimeout)
- allow_total = allow = data->set.timeout*1000;
- else
- allow = data->set.connecttimeout*1000;
- }
- else if(data->set.timeout) {
- allow_total = allow = data->set.timeout*1000;
- }
- else if(data->set.connecttimeout) {
- allow = data->set.connecttimeout*1000;
- }
- if(has_passed > allow ) {
- /* time-out, bail out, go home */
- failf(data, "Connection time-out after %ld ms", has_passed);
- return CURLE_OPERATION_TIMEOUTED;
- }
- if(conn->bits.tcpconnect) {
- /* we are connected already! */
- Curl_expire(data, allow_total);
- *connected = TRUE;
- return CURLE_OK;
- }
- Curl_expire(data, allow);
- /* check for connect without timeout as we want to return immediately */
- rc = waitconnect(sockfd, 0);
- if(WAITCONN_CONNECTED == rc) {
- int error;
- if (verifyconnect(sockfd, &error)) {
- /* we are connected, awesome! */
- *connected = TRUE;
- return CURLE_OK;
- }
- /* nope, not connected for real */
- data->state.os_errno = error;
- infof(data, "Connection failed\n");
- if(trynextip(conn, sockindex, connected)) {
- code = CURLE_COULDNT_CONNECT;
- }
- }
- else if(WAITCONN_TIMEOUT != rc) {
- int error = 0;
- /* nope, not connected */
- if (WAITCONN_FDSET_ERROR == rc) {
- (void)verifyconnect(sockfd, &error);
- data->state.os_errno = error;
- infof(data, "%s\n",Curl_strerror(conn,error));
- }
- else
- infof(data, "Connection failed\n");
- if(trynextip(conn, sockindex, connected)) {
- error = Curl_sockerrno();
- data->state.os_errno = error;
- failf(data, "Failed connect to %s:%d; %s",
- conn->host.name, conn->port, Curl_strerror(conn,error));
- code = CURLE_COULDNT_CONNECT;
- }
- }
- /*
- * If the connection failed here, we should attempt to connect to the "next
- * address" for the given host.
- */
- return code;
- }
- static void tcpnodelay(struct connectdata *conn,
- curl_socket_t sockfd)
- {
- #ifdef TCP_NODELAY
- struct SessionHandle *data= conn->data;
- socklen_t onoff = (socklen_t) data->set.tcp_nodelay;
- int proto = IPPROTO_TCP;
- #ifdef HAVE_GETPROTOBYNAME
- struct protoent *pe = getprotobyname("tcp");
- if (pe)
- proto = pe->p_proto;
- #endif
- if(setsockopt(sockfd, proto, TCP_NODELAY, (void *)&onoff,
- sizeof(onoff)) < 0)
- infof(data, "Could not set TCP_NODELAY: %s\n",
- Curl_strerror(conn, Curl_sockerrno()));
- else
- infof(data,"TCP_NODELAY set\n");
- #else
- (void)conn;
- (void)sockfd;
- #endif
- }
- #ifdef SO_NOSIGPIPE
- /* The preferred method on Mac OS X (10.2 and later) to prevent SIGPIPEs when
- sending data to a dead peer (instead of relying on the 4th argument to send
- being MSG_NOSIGNAL). Possibly also existing and in use on other BSD
- systems? */
- static void nosigpipe(struct connectdata *conn,
- curl_socket_t sockfd)
- {
- struct SessionHandle *data= conn->data;
- int onoff = 1;
- if(setsockopt(sockfd, SOL_SOCKET, SO_NOSIGPIPE, (void *)&onoff,
- sizeof(onoff)) < 0)
- infof(data, "Could not set SO_NOSIGPIPE: %s\n",
- Curl_strerror(conn, Curl_sockerrno()));
- }
- #else
- #define nosigpipe(x,y)
- #endif
- /* singleipconnect() connects to the given IP only, and it may return without
- having connected if used from the multi interface. */
- static curl_socket_t
- singleipconnect(struct connectdata *conn,
- const Curl_addrinfo *ai,
- long timeout_ms,
- bool *connected)
- {
- char addr_buf[128];
- int rc;
- int error;
- bool isconnected;
- struct SessionHandle *data = conn->data;
- curl_socket_t sockfd;
- CURLcode res;
- sockfd = socket(ai->ai_family, conn->socktype, ai->ai_protocol);
- if (sockfd == CURL_SOCKET_BAD)
- return CURL_SOCKET_BAD;
- *connected = FALSE; /* default is not connected */
- Curl_printable_address(ai, addr_buf, sizeof(addr_buf));
- infof(data, " Trying %s... ", addr_buf);
- if(data->set.tcp_nodelay)
- tcpnodelay(conn, sockfd);
- nosigpipe(conn, sockfd);
- if(data->set.fsockopt) {
- /* activate callback for setting socket options */
- error = data->set.fsockopt(data->set.sockopt_client,
- sockfd,
- CURLSOCKTYPE_IPCXN);
- if (error) {
- sclose(sockfd); /* close the socket and bail out */
- return CURL_SOCKET_BAD;
- }
- }
- /* possibly bind the local end to an IP, interface or port */
- res = bindlocal(conn, sockfd);
- if(res) {
- sclose(sockfd); /* close socket and bail out */
- return CURL_SOCKET_BAD;
- }
- /* set socket non-blocking */
- Curl_nonblock(sockfd, TRUE);
- /* Connect TCP sockets, bind UDP */
- if(conn->socktype == SOCK_STREAM)
- rc = connect(sockfd, ai->ai_addr, ai->ai_addrlen);
- else
- rc = 0;
- if(-1 == rc) {
- error = Curl_sockerrno();
- switch (error) {
- case EINPROGRESS:
- case EWOULDBLOCK:
- #if defined(EAGAIN) && EAGAIN != EWOULDBLOCK
- /* On some platforms EAGAIN and EWOULDBLOCK are the
- * same value, and on others they are different, hence
- * the odd #if
- */
- case EAGAIN:
- #endif
- rc = waitconnect(sockfd, timeout_ms);
- break;
- default:
- /* unknown error, fallthrough and try another address! */
- failf(data, "Failed to connect to %s: %s",
- addr_buf, Curl_strerror(conn,error));
- data->state.os_errno = error;
- break;
- }
- }
- /* The 'WAITCONN_TIMEOUT == rc' comes from the waitconnect(), and not from
- connect(). We can be sure of this since connect() cannot return 1. */
- if((WAITCONN_TIMEOUT == rc) &&
- (data->state.used_interface == Curl_if_multi)) {
- /* Timeout when running the multi interface */
- return sockfd;
- }
- isconnected = verifyconnect(sockfd, &error);
- if(!rc && isconnected) {
- /* we are connected, awesome! */
- *connected = TRUE; /* this is a true connect */
- infof(data, "connected\n");
- return sockfd;
- }
- else if(WAITCONN_TIMEOUT == rc)
- infof(data, "Timeout\n");
- else {
- data->state.os_errno = error;
- infof(data, "%s\n", Curl_strerror(conn, error));
- }
- /* connect failed or timed out */
- sclose(sockfd);
- return CURL_SOCKET_BAD;
- }
- /*
- * TCP connect to the given host with timeout, proxy or remote doesn't matter.
- * There might be more than one IP address to try out. Fill in the passed
- * pointer with the connected socket.
- */
- CURLcode Curl_connecthost(struct connectdata *conn, /* context */
- const struct Curl_dns_entry *remotehost, /* use this one */
- curl_socket_t *sockconn, /* the connected socket */
- Curl_addrinfo **addr, /* the one we used */
- bool *connected) /* really connected? */
- {
- struct SessionHandle *data = conn->data;
- curl_socket_t sockfd = CURL_SOCKET_BAD;
- int aliasindex;
- int num_addr;
- Curl_addrinfo *ai;
- Curl_addrinfo *curr_addr;
- struct timeval after;
- struct timeval before = Curl_tvnow();
- /*************************************************************
- * Figure out what maximum time we have left
- *************************************************************/
- long timeout_ms= DEFAULT_CONNECT_TIMEOUT;
- long timeout_per_addr;
- *connected = FALSE; /* default to not connected */
- if(data->set.timeout || data->set.connecttimeout) {
- long has_passed;
- /* Evaluate in milliseconds how much time that has passed */
- has_passed = Curl_tvdiff(Curl_tvnow(), data->progress.t_startsingle);
- #ifndef min
- #define min(a, b) ((a) < (b) ? (a) : (b))
- #endif
- /* get the most strict timeout of the ones converted to milliseconds */
- if(data->set.timeout && data->set.connecttimeout) {
- if (data->set.timeout < data->set.connecttimeout)
- timeout_ms = data->set.timeout*1000;
- else
- timeout_ms = data->set.connecttimeout*1000;
- }
- else if(data->set.timeout)
- timeout_ms = data->set.timeout*1000;
- else
- timeout_ms = data->set.connecttimeout*1000;
- /* subtract the passed time */
- timeout_ms -= has_passed;
- if(timeout_ms < 0) {
- /* a precaution, no need to continue if time already is up */
- failf(data, "Connection time-out");
- return CURLE_OPERATION_TIMEOUTED;
- }
- }
- Curl_expire(data, timeout_ms);
- /* Max time for each address */
- num_addr = Curl_num_addresses(remotehost->addr);
- timeout_per_addr = timeout_ms / num_addr;
- ai = remotehost->addr;
- /* Below is the loop that attempts to connect to all IP-addresses we
- * know for the given host. One by one until one IP succeeds.
- */
- if(data->state.used_interface == Curl_if_multi)
- /* don't hang when doing multi */
- timeout_per_addr = 0;
- /*
- * Connecting with a Curl_addrinfo chain
- */
- for (curr_addr = ai, aliasindex=0; curr_addr;
- curr_addr = curr_addr->ai_next, aliasindex++) {
- /* start connecting to the IP curr_addr points to */
- sockfd = singleipconnect(conn, curr_addr, timeout_per_addr, connected);
- if(sockfd != CURL_SOCKET_BAD)
- break;
- /* get a new timeout for next attempt */
- after = Curl_tvnow();
- timeout_ms -= Curl_tvdiff(after, before);
- if(timeout_ms < 0) {
- failf(data, "connect() timed out!");
- return CURLE_OPERATION_TIMEOUTED;
- }
- before = after;
- } /* end of connect-to-each-address loop */
- if (sockfd == CURL_SOCKET_BAD) {
- /* no good connect was made */
- *sockconn = CURL_SOCKET_BAD;
- failf(data, "couldn't connect to host");
- return CURLE_COULDNT_CONNECT;
- }
- /* leave the socket in non-blocking mode */
- /* store the address we use */
- if(addr)
- *addr = curr_addr;
- /* allow NULL-pointers to get passed in */
- if(sockconn)
- *sockconn = sockfd; /* the socket descriptor we've connected */
- data->info.numconnects++; /* to track the number of connections made */
- return CURLE_OK;
- }
|