1053 lines
		
	
	
		
			29 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			1053 lines
		
	
	
		
			29 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /***************************************************************************
 | |
|  *                                  _   _ ____  _
 | |
|  *  Project                     ___| | | |  _ \| |
 | |
|  *                             / __| | | | |_) | |
 | |
|  *                            | (__| |_| |  _ <| |___
 | |
|  *                             \___|\___/|_| \_\_____|
 | |
|  *
 | |
|  * Copyright (C) 1998 - 2008, 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: connect.c,v 1.207 2008-11-06 17:19:57 yangtse Exp $
 | |
|  ***************************************************************************/
 | |
| 
 | |
| #include "setup.h"
 | |
| 
 | |
| #ifndef WIN32
 | |
| /* headers for non-win32 */
 | |
| #ifdef HAVE_SYS_TIME_H
 | |
| #include <sys/time.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_SYS_UN_H
 | |
| #include <sys/un.h> /* for sockaddr_un */
 | |
| #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_ARPA_INET_H
 | |
| #include <arpa/inet.h>
 | |
| #endif
 | |
| #ifdef HAVE_STDLIB_H
 | |
| #include <stdlib.h> /* required for free() prototype, without it, this crashes */
 | |
| #endif              /* on macos 68K */
 | |
| 
 | |
| #if (defined(HAVE_FIONBIO) && defined(NETWARE))
 | |
| #include <sys/filio.h>
 | |
| #endif
 | |
| #ifdef NETWARE
 | |
| #undef in_addr_t
 | |
| #define in_addr_t unsigned long
 | |
| #endif
 | |
| #ifdef VMS
 | |
| #include <in.h>
 | |
| #include <inet.h>
 | |
| #endif
 | |
| 
 | |
| #endif  /* !WIN32 */
 | |
| 
 | |
| #include <stdio.h>
 | |
| #include <errno.h>
 | |
| #include <string.h>
 | |
| 
 | |
| #define _MPRINTF_REPLACE /* use our functions only */
 | |
| #include <curl/mprintf.h>
 | |
| 
 | |
| #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"
 | |
| #include "sslgen.h" /* for Curl_ssl_check_cxn() */
 | |
| 
 | |
| /* The last #include file should be: */
 | |
| #include "memdebug.h"
 | |
| 
 | |
| #ifdef __SYMBIAN32__
 | |
| /* This isn't actually supported under Symbian OS */
 | |
| #undef SO_NOSIGPIPE
 | |
| #endif
 | |
| 
 | |
| struct Curl_sockaddr_ex {
 | |
|   int family;
 | |
|   int socktype;
 | |
|   int protocol;
 | |
|   unsigned int addrlen;
 | |
|   union {
 | |
|     struct sockaddr addr;
 | |
|     struct Curl_sockaddr_storage buff;
 | |
|   } _sa_ex_u;
 | |
| };
 | |
| #define sa_addr _sa_ex_u.addr
 | |
| 
 | |
| 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_timeleft() returns the amount of milliseconds left allowed for the
 | |
|  * transfer/connection. If the value is negative, the timeout time has already
 | |
|  * elapsed.
 | |
|  *
 | |
|  * If 'nowp' is non-NULL, it points to the current time.
 | |
|  * 'duringconnect' is FALSE if not during a connect, as then of course the
 | |
|  * connect timeout is not taken into account!
 | |
|  */
 | |
| long Curl_timeleft(struct connectdata *conn,
 | |
|                    struct timeval *nowp,
 | |
|                    bool duringconnect)
 | |
| {
 | |
|   struct SessionHandle *data = conn->data;
 | |
|   int timeout_set = 0;
 | |
|   long timeout_ms = duringconnect?DEFAULT_CONNECT_TIMEOUT:0;
 | |
|   struct timeval now;
 | |
| 
 | |
|   /* if a timeout is set, use the most restrictive one */
 | |
| 
 | |
|   if(data->set.timeout > 0)
 | |
|     timeout_set |= 1;
 | |
|   if(duringconnect && (data->set.connecttimeout > 0))
 | |
|     timeout_set |= 2;
 | |
| 
 | |
|   switch (timeout_set) {
 | |
|   case 1:
 | |
|     timeout_ms = data->set.timeout;
 | |
|     break;
 | |
|   case 2:
 | |
|     timeout_ms = data->set.connecttimeout;
 | |
|     break;
 | |
|   case 3:
 | |
|     if(data->set.timeout < data->set.connecttimeout)
 | |
|       timeout_ms = data->set.timeout;
 | |
|     else
 | |
|       timeout_ms = data->set.connecttimeout;
 | |
|     break;
 | |
|   default:
 | |
|     /* use the default */
 | |
|     if(!duringconnect)
 | |
|       /* if we're not during connect, there's no default timeout so if we're
 | |
|          at zero we better just return zero and not make it a negative number
 | |
|          by the math below */
 | |
|       return 0;
 | |
|     break;
 | |
|   }
 | |
| 
 | |
|   if(!nowp) {
 | |
|     now = Curl_tvnow();
 | |
|     nowp = &now;
 | |
|   }
 | |
| 
 | |
|   /* substract elapsed time */
 | |
|   timeout_ms -= Curl_tvdiff(*nowp, data->progress.t_startsingle);
 | |
| 
 | |
|   return timeout_ms;
 | |
| }
 | |
| 
 | |
| 
 | |
| /*
 | |
|  * 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(FALSE != 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_socket_ready(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 & CURL_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, int af)
 | |
| {
 | |
|   struct SessionHandle *data = conn->data;
 | |
|   struct sockaddr_in me;
 | |
| #ifdef ENABLE_IPV6
 | |
|   struct sockaddr_in6 me6;
 | |
| #endif
 | |
|   struct sockaddr *sock = NULL;  /* bind to this address */
 | |
|   socklen_t socksize = 0; /* size of the data sock points to */
 | |
|   struct Curl_dns_entry *h=NULL;
 | |
|   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;
 | |
|   const char *dev = data->set.str[STRING_DEVICE];
 | |
|   int error;
 | |
| 
 | |
|   /*************************************************************
 | |
|    * Select device to bind socket to
 | |
|    *************************************************************/
 | |
|   if(dev && (strlen(dev)<255) ) {
 | |
|     char myhost[256] = "";
 | |
|     int rc;
 | |
|     bool was_iface = FALSE;
 | |
| 
 | |
|     if(Curl_if2ip(af, dev, myhost, sizeof(myhost))) {
 | |
|       /*
 | |
|        * We now have the numerical IP address 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;
 | |
|       }
 | |
|     }
 | |
| 
 | |
|     if(!was_iface) {
 | |
|       /*
 | |
|        * This was not an interface, resolve the name as a host name
 | |
|        * or IP number
 | |
|        */
 | |
| 
 | |
|       /*
 | |
|        * Temporarily force name resolution to use only the address type
 | |
|        * of the connection. The resolve functions should really be changed
 | |
|        * to take a type parameter instead.
 | |
|        */
 | |
|       long ipver = data->set.ip_version;
 | |
|       if (af == AF_INET)
 | |
|         data->set.ip_version = CURL_IPRESOLVE_V4;
 | |
| #ifdef ENABLE_IPV6
 | |
|       else if (af == AF_INET6)
 | |
|         data->set.ip_version = CURL_IPRESOLVE_V6;
 | |
| #endif
 | |
| 
 | |
|       rc = Curl_resolv(conn, dev, 0, &h);
 | |
|       if(rc == CURLRESOLV_PENDING)
 | |
|         (void)Curl_wait_for_resolv(conn, &h);
 | |
|       data->set.ip_version = ipver;
 | |
| 
 | |
|       if(h) {
 | |
|         /* convert the resolved address, sizeof myhost >= INET_ADDRSTRLEN */
 | |
|         Curl_printable_address(h->addr, myhost, sizeof myhost);
 | |
|       }
 | |
|     }
 | |
| 
 | |
|     if(!*myhost || !h) {
 | |
|       /* need to fix this
 | |
|          h=Curl_gethost(data,
 | |
|          getmyhost(*myhost,sizeof(myhost)),
 | |
|          hostent_buf,
 | |
|          sizeof(hostent_buf));
 | |
|       */
 | |
|       failf(data, "Couldn't bind to '%s'", dev);
 | |
|       if(h)
 | |
|         Curl_resolv_unlock(data, h);
 | |
|       return CURLE_INTERFACE_FAILED;
 | |
|     }
 | |
| 
 | |
|     infof(data, "Bind local address to %s\n", myhost);
 | |
| 
 | |
|     sock = h->addr->ai_addr;
 | |
|     socksize = h->addr->ai_addrlen;
 | |
| 
 | |
| #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,
 | |
|                      dev, strlen(dev)+1) != 0) {
 | |
|         error = SOCKERRNO;
 | |
|         infof(data, "SO_BINDTODEVICE %s failed with errno %d: %s; will do regular bind\n",
 | |
|               dev, error, Curl_strerror(conn, error));
 | |
|         /* This is typically "errno 1, error: Operation not permitted" if
 | |
|            you're not running as root or another suitable privileged user */
 | |
|       }
 | |
|     }
 | |
| #endif
 | |
|   }
 | |
|   else if(port) {
 | |
|     /* if a local port number is requested but no local IP, extract the
 | |
|        address from the socket */
 | |
|     if(af == AF_INET) {
 | |
|       memset(&me, 0, sizeof(me));
 | |
|       me.sin_family = AF_INET;
 | |
|       me.sin_addr.s_addr = INADDR_ANY;
 | |
| 
 | |
|       sock = (struct sockaddr *)&me;
 | |
|       socksize = sizeof(me);
 | |
| 
 | |
|     }
 | |
| #ifdef ENABLE_IPV6
 | |
|     else { /* AF_INET6 */
 | |
|       memset(&me6, 0, sizeof(me6));
 | |
|       me6.sin6_family = AF_INET6;
 | |
|       /* in6addr_any isn't always available and since me6 has just been
 | |
|          cleared, it's not strictly necessary to use it here */
 | |
|       /*me6.sin6_addr = in6addr_any;*/
 | |
| 
 | |
|       sock = (struct sockaddr *)&me6;
 | |
|       socksize = sizeof(me6);
 | |
|     }
 | |
| #endif
 | |
|   }
 | |
|   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)
 | |
|       me.sin_port = htons(port);
 | |
| #ifdef ENABLE_IPV6
 | |
|     else
 | |
|       me6.sin6_port = htons(port);
 | |
| #endif
 | |
| 
 | |
|     if( bind(sockfd, sock, socksize) >= 0) {
 | |
|       /* we succeeded to bind */
 | |
|       struct Curl_sockaddr_storage add;
 | |
|       socklen_t size = sizeof(add);
 | |
|       memset(&add, 0, sizeof(struct Curl_sockaddr_storage));
 | |
|       if(getsockname(sockfd, (struct sockaddr *) &add, &size) < 0) {
 | |
|         data->state.os_errno = error = SOCKERRNO;
 | |
|         failf(data, "getsockname() failed with errno %d: %s",
 | |
|               error, Curl_strerror(conn, error));
 | |
|         if(h)
 | |
|           Curl_resolv_unlock(data, h);
 | |
|         return CURLE_INTERFACE_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);
 | |
|       conn->bits.bound = TRUE;
 | |
|       if(h)
 | |
|         Curl_resolv_unlock(data, h);
 | |
|       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 = error = SOCKERRNO;
 | |
|   failf(data, "bind failed with errno %d: %s",
 | |
|         error, Curl_strerror(conn, error));
 | |
|   if(h)
 | |
|     Curl_resolv_unlock(data, h);
 | |
| 
 | |
|   return CURLE_INTERFACE_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(0 != getsockopt(sockfd, SOL_SOCKET, SO_ERROR, (void *)&err, &errSize))
 | |
|     err = SOCKERRNO;
 | |
| #ifdef _WIN32_WCE
 | |
|   /* Old WinCE versions don't support SO_ERROR */
 | |
|   if(WSAENOPROTOOPT == err) {
 | |
|     SET_SOCKERRNO(0);
 | |
|     err = 0;
 | |
|   }
 | |
| #endif
 | |
| #ifdef __minix
 | |
|   /* Minix 3.1.x doesn't support getsockopt on UDP sockets */
 | |
|   if(EBADIOCTL == err) {
 | |
|     SET_SOCKERRNO(0);
 | |
|     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 = SOCKERRNO;
 | |
| #endif
 | |
|   return rc;
 | |
| }
 | |
| 
 | |
| /* Used within the multi interface. Try next IP address, return TRUE if no
 | |
|    more address exists or error */
 | |
| 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;
 | |
|       break;
 | |
|     }
 | |
|     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;
 | |
| 
 | |
|   DEBUGASSERT(sockindex >= FIRSTSOCKET && sockindex <= SECONDARYSOCKET);
 | |
| 
 | |
|   *connected = FALSE; /* a very negative world view is best */
 | |
| 
 | |
|   if(conn->bits.tcpconnect) {
 | |
|     /* we are connected already! */
 | |
|     long allow_total = 0;
 | |
| 
 | |
|     /* subtract the most strict timeout of the ones */
 | |
|     if(data->set.timeout)
 | |
|       allow_total = data->set.timeout;
 | |
| 
 | |
|     Curl_expire(data, allow_total);
 | |
|     *connected = TRUE;
 | |
|     return CURLE_OK;
 | |
|   }
 | |
| 
 | |
|   /* figure out how long time we have left to connect */
 | |
|   allow = Curl_timeleft(conn, NULL, TRUE);
 | |
| 
 | |
|   if(allow < 0) {
 | |
|     /* time-out, bail out, go home */
 | |
|     failf(data, "Connection time-out");
 | |
|     return CURLE_OPERATION_TIMEDOUT;
 | |
|   }
 | |
| 
 | |
|   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)) {
 | |
|       failf(data, "Failed connect to %s:%d; %s",
 | |
|             conn->host.name, conn->port, Curl_strerror(conn, error));
 | |
|       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 = 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;
 | |
| 
 | |
| #if 0
 | |
|   /* The use of getprotobyname() is disabled since it isn't thread-safe on
 | |
|      numerous systems. On these getprotobyname_r() should be used instead, but
 | |
|      that exists in at least one 4 arg version and one 5 arg version, and
 | |
|      since the proto number rarely changes anyway we now just use the hard
 | |
|      coded number. The "proper" fix would need a configure check for the
 | |
|      correct function much in the same style the gethostbyname_r versions are
 | |
|      detected. */
 | |
|   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, 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, 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)
 | |
| {
 | |
|   struct Curl_sockaddr_ex addr;
 | |
|   char addr_buf[128];
 | |
|   int rc;
 | |
|   int error;
 | |
|   bool isconnected;
 | |
|   struct SessionHandle *data = conn->data;
 | |
|   curl_socket_t sockfd;
 | |
|   CURLcode res;
 | |
|   const void *iptoprint;
 | |
|   struct sockaddr_in * const sa4 = (void *)&addr.sa_addr;
 | |
| #ifdef ENABLE_IPV6
 | |
|   struct sockaddr_in6 * const sa6 = (void *)&addr.sa_addr;
 | |
| #endif
 | |
| 
 | |
|   /*
 | |
|    * The Curl_sockaddr_ex structure is basically libcurl's external API
 | |
|    * curl_sockaddr structure with enough space available to directly hold
 | |
|    * any protocol-specific address structures. The variable declared here
 | |
|    * will be used to pass / receive data to/from the fopensocket callback
 | |
|    * if this has been set, before that, it is initialized from parameters.
 | |
|    */
 | |
| 
 | |
|   addr.family = ai->ai_family;
 | |
|   addr.socktype = conn->socktype;
 | |
|   addr.protocol = ai->ai_protocol;
 | |
|   addr.addrlen = ai->ai_addrlen;
 | |
| 
 | |
|   if(addr.addrlen > sizeof(struct Curl_sockaddr_storage))
 | |
|      addr.addrlen = sizeof(struct Curl_sockaddr_storage);
 | |
|   memcpy(&addr.sa_addr, ai->ai_addr, addr.addrlen);
 | |
| 
 | |
|   *connected = FALSE; /* default is not connected */
 | |
| 
 | |
|   if(data->set.fopensocket)
 | |
|    /*
 | |
|     * If the opensocket callback is set, all the destination address information
 | |
|     * is passed to the callback. Depending on this information the callback may
 | |
|     * opt to abort the connection, this is indicated returning CURL_SOCKET_BAD;
 | |
|     * otherwise it will return a not-connected socket. When the callback returns
 | |
|     * a valid socket the destination address information might have been changed
 | |
|     * and this 'new' address will actually be used here to connect.
 | |
|     */
 | |
|     sockfd = data->set.fopensocket(data->set.opensocket_client,
 | |
|                                    CURLSOCKTYPE_IPCXN,
 | |
|                                    (struct curl_sockaddr *)&addr);
 | |
|   else
 | |
|     /* opensocket callback not set, so simply create the socket now */
 | |
|     sockfd = socket(addr.family, addr.socktype, addr.protocol);
 | |
| 
 | |
|   if(sockfd == CURL_SOCKET_BAD)
 | |
|     /* no socket, no connection */
 | |
|     return CURL_SOCKET_BAD;
 | |
| 
 | |
| #ifdef ENABLE_IPV6
 | |
|   if (conn->scope && (addr.family == AF_INET6))
 | |
|     sa6->sin6_scope_id = conn->scope;
 | |
| #endif
 | |
| 
 | |
|   /* FIXME: do we have Curl_printable_address-like with struct sockaddr* as
 | |
|      argument? */
 | |
| #if defined(HAVE_SYS_UN_H) && defined(AF_UNIX)
 | |
|   if(addr.family == AF_UNIX) {
 | |
|     infof(data, "  Trying %s... ",
 | |
|           ((const struct sockaddr_un*)(&addr.sa_addr))->sun_path);
 | |
|     snprintf(data->info.ip, MAX_IPADR_LEN, "%s",
 | |
|              ((const struct sockaddr_un*)(&addr.sa_addr))->sun_path);
 | |
|     strcpy(conn->ip_addr_str, data->info.ip);
 | |
|   }
 | |
|   else
 | |
| #endif
 | |
|   {
 | |
| #ifdef ENABLE_IPV6
 | |
|     if(addr.family == AF_INET6) {
 | |
|       iptoprint = &sa6->sin6_addr;
 | |
|       conn->bits.ipv6 = TRUE;
 | |
|     }
 | |
|     else
 | |
| #endif
 | |
|     {
 | |
|       iptoprint = &sa4->sin_addr;
 | |
|     }
 | |
| 
 | |
|     if(Curl_inet_ntop(addr.family, iptoprint, addr_buf,
 | |
|                       sizeof(addr_buf)) != NULL) {
 | |
|       infof(data, "  Trying %s... ", addr_buf);
 | |
|       snprintf(data->info.ip, MAX_IPADR_LEN, "%s", addr_buf);
 | |
|       strcpy(conn->ip_addr_str, data->info.ip);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   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, addr.family);
 | |
|   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, &addr.sa_addr, addr.addrlen);
 | |
|   else
 | |
|     rc = 0;
 | |
| 
 | |
|   if(-1 == rc) {
 | |
|     error = 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,
 | |
|                           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;
 | |
|   long timeout_per_addr;
 | |
| 
 | |
|   DEBUGASSERT(sockconn);
 | |
|   *connected = FALSE; /* default to not connected */
 | |
| 
 | |
|   /* get the timeout left */
 | |
|   timeout_ms = Curl_timeleft(conn, &before, TRUE);
 | |
| 
 | |
|   if(timeout_ms < 0) {
 | |
|     /* a precaution, no need to continue if time already is up */
 | |
|     failf(data, "Connection time-out");
 | |
|     return CURLE_OPERATION_TIMEDOUT;
 | |
|   }
 | |
|   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_TIMEDOUT;
 | |
|     }
 | |
|     before = after;
 | |
|   }  /* end of connect-to-each-address loop */
 | |
| 
 | |
|   *sockconn = sockfd;    /* the socket descriptor we've connected */
 | |
| 
 | |
|   if(sockfd == CURL_SOCKET_BAD) {
 | |
|     /* no good connect was made */
 | |
|     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;
 | |
| 
 | |
|   data->info.numconnects++; /* to track the number of connections made */
 | |
| 
 | |
|   return CURLE_OK;
 | |
| }
 | |
| 
 | |
| /*
 | |
|  * Used to extract socket and connectdata struct for the most recent
 | |
|  * transfer on the given SessionHandle.
 | |
|  *
 | |
|  * The socket 'long' will be -1 in case of failure!
 | |
|  */
 | |
| CURLcode Curl_getconnectinfo(struct SessionHandle *data,
 | |
|                              long *param_longp,
 | |
|                              struct connectdata **connp)
 | |
| {
 | |
|   if((data->state.lastconnect != -1) &&
 | |
|      (data->state.connc->connects[data->state.lastconnect] != NULL)) {
 | |
|     struct connectdata *c =
 | |
|       data->state.connc->connects[data->state.lastconnect];
 | |
|     if(connp)
 | |
|       /* only store this if the caller cares for it */
 | |
|       *connp = c;
 | |
|     *param_longp = c->sock[FIRSTSOCKET];
 | |
|     /* we have a socket connected, let's determine if the server shut down */
 | |
|     /* determine if ssl */
 | |
|     if(c->ssl[FIRSTSOCKET].use) {
 | |
|       /* use the SSL context */
 | |
|       if(!Curl_ssl_check_cxn(c))
 | |
|         *param_longp = -1;   /* FIN received */
 | |
|     }
 | |
| /* Minix 3.1 doesn't support any flags on recv; just assume socket is OK */
 | |
| #ifdef MSG_PEEK
 | |
|     else {
 | |
|       /* use the socket */
 | |
|       char buf;
 | |
|       if(recv((RECV_TYPE_ARG1)c->sock[FIRSTSOCKET], (RECV_TYPE_ARG2)&buf,
 | |
|               (RECV_TYPE_ARG3)1, (RECV_TYPE_ARG4)MSG_PEEK) == 0) {
 | |
|         *param_longp = -1;   /* FIN received */
 | |
|       }
 | |
|     }
 | |
| #endif
 | |
|   }
 | |
|   else
 | |
|     *param_longp = -1;
 | |
| 
 | |
|   return CURLE_OK;
 | |
| }
 |