Check for lowercase 'bool' type at configuration stage. If not available provide a suitable replacement with a type definition of 'unsigned char' in setup_once.h Move definitions of TRUE and FALSE to setup_once.h
		
			
				
	
	
		
			880 lines
		
	
	
		
			24 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			880 lines
		
	
	
		
			24 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/***************************************************************************
 | 
						|
 *                                  _   _ ____  _
 | 
						|
 *  Project                     ___| | | |  _ \| |
 | 
						|
 *                             / __| | | | |_) | |
 | 
						|
 *                            | (__| |_| |  _ <| |___
 | 
						|
 *                             \___|\___/|_| \_\_____|
 | 
						|
 *
 | 
						|
 * Copyright (C) 1998 - 2007, 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>
 | 
						|
 | 
						|
#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_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_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(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 = 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 = 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 = 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;
 | 
						|
 | 
						|
  DEBUGASSERT(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;
 | 
						|
    else
 | 
						|
      allow = data->set.connecttimeout;
 | 
						|
  }
 | 
						|
  else if(data->set.timeout) {
 | 
						|
    allow_total = allow = data->set.timeout;
 | 
						|
  }
 | 
						|
  else if(data->set.connecttimeout) {
 | 
						|
    allow = data->set.connecttimeout;
 | 
						|
  }
 | 
						|
 | 
						|
  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 = 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, 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)
 | 
						|
{
 | 
						|
  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 = 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;
 | 
						|
      else
 | 
						|
        timeout_ms = data->set.connecttimeout;
 | 
						|
    }
 | 
						|
    else if(data->set.timeout)
 | 
						|
      timeout_ms = data->set.timeout;
 | 
						|
    else
 | 
						|
      timeout_ms = data->set.connecttimeout;
 | 
						|
 | 
						|
    /* 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;
 | 
						|
}
 |