methods/connect.cc

   1 // -*- mode: cpp; mode: fold -*-
   2 // Description                                                          /*{{{*/
   3 // $Id: connect.cc,v 1.10.2.1 2004/01/16 18:58:50 mdz Exp $
   4 /* ######################################################################
   5
   6    Connect - Replacement connect call
   7
   8    This was originally authored by Jason Gunthorpe <jgg@debian.org>
   9    and is placed in the Public Domain, do with it what you will.
  10
  11    ##################################################################### */
  12                                                                         /*}}}*/
  13 // Include Files                                                        /*{{{*/
  14 #include "connect.h"
  15 #include <apt-pkg/error.h>
  16 #include <apt-pkg/fileutl.h>
  17
  18 #include <stdio.h>
  19 #include <errno.h>
  20 #include <unistd.h>
  21
  22 // Internet stuff
  23 #include <netinet/in.h>
  24 #include <sys/socket.h>
  25 #include <arpa/inet.h>
  26 #include <netdb.h>
  27
  28 #include "rfc2553emu.h"
  29 #include <apti18n.h>
  30                                                                         /*}}}*/
  31
  32 static string LastHost;
  33 static int LastPort = 0;
  34 static struct addrinfo *LastHostAddr = 0;
  35 static struct addrinfo *LastUsed = 0;
  36
  37 // RotateDNS - Select a new server from a DNS rotation                  /*{{{*/
  38 // ---------------------------------------------------------------------
  39 /* This is called during certain errors in order to recover by selecting a
  40    new server */
  41 void RotateDNS()
  42 {
  43    if (LastUsed != 0 && LastUsed->ai_next != 0)
  44       LastUsed = LastUsed->ai_next;
  45    else
  46       LastUsed = LastHostAddr;
  47 }
  48                                                                         /*}}}*/
  49 // DoConnect - Attempt a connect operation                              /*{{{*/
  50 // ---------------------------------------------------------------------
  51 /* This helper function attempts a connection to a single address. */
  52 static bool DoConnect(struct addrinfo *Addr,string Host,
  53                       unsigned long TimeOut,int &Fd,pkgAcqMethod *Owner)
  54 {
  55    // Show a status indicator
  56    char Name[NI_MAXHOST];
  57    char Service[NI_MAXSERV];
  58
  59    Name[0] = 0;
  60    Service[0] = 0;
  61    getnameinfo(Addr->ai_addr,Addr->ai_addrlen,
  62                Name,sizeof(Name),Service,sizeof(Service),
  63                NI_NUMERICHOST|NI_NUMERICSERV);
  64    Owner->Status(_("Connecting to %s (%s)"),Host.c_str(),Name);
  65
  66    /* If this is an IP rotation store the IP we are using.. If something goes
  67       wrong this will get tacked onto the end of the error message */
  68    if (LastHostAddr->ai_next != 0)
  69    {
  70       char Name2[NI_MAXHOST + NI_MAXSERV + 10];
  71       snprintf(Name2,sizeof(Name2),_("[IP: %s %s]"),Name,Service);
  72       Owner->SetFailExtraMsg(string(Name2));
  73    }
  74    else
  75       Owner->SetFailExtraMsg("");
  76
  77    // Get a socket
  78    if ((Fd = socket(Addr->ai_family,Addr->ai_socktype,
  79                     Addr->ai_protocol)) < 0)
  80       return _error->Errno("socket",_("Could not create a socket for %s (f=%u t=%u p=%u)"),
  81                            Name,Addr->ai_family,Addr->ai_socktype,Addr->ai_protocol);
  82
  83    SetNonBlock(Fd,true);
  84    if (connect(Fd,Addr->ai_addr,Addr->ai_addrlen) < 0 &&
  85        errno != EINPROGRESS)
  86       return _error->Errno("connect",_("Cannot initiate the connection "
  87                            "to %s:%s (%s)."),Host.c_str(),Service,Name);
  88
  89    /* This implements a timeout for connect by opening the connection
  90       nonblocking */
  91    if (WaitFd(Fd,true,TimeOut) == false) {
  92       Owner->SetFailExtraMsg("\nFailReason: Timeout");
  93       return _error->Error(_("Could not connect to %s:%s (%s), "
  94                            "connection timed out"),Host.c_str(),Service,Name);
  95    }
  96
  97    // Check the socket for an error condition
  98    unsigned int Err;
  99    unsigned int Len = sizeof(Err);
 100    if (getsockopt(Fd,SOL_SOCKET,SO_ERROR,&Err,&Len) != 0)
 101       return _error->Errno("getsockopt",_("Failed"));
 102
 103    if (Err != 0)
 104    {
 105       errno = Err;
 106       if(errno == ECONNREFUSED)
 107          Owner->SetFailExtraMsg("\nFailReason: ConnectionRefused");
 108       return _error->Errno("connect",_("Could not connect to %s:%s (%s)."),Host.c_str(),
 109                            Service,Name);
 110    }
 111
 112    return true;
 113 }
 114                                                                         /*}}}*/
 115 // Connect - Connect to a server                                        /*{{{*/
 116 // ---------------------------------------------------------------------
 117 /* Performs a connection to the server */
 118 bool Connect(string Host,int Port,const char *Service,int DefPort,int &Fd,
 119              unsigned long TimeOut,pkgAcqMethod *Owner)
 120 {
 121    if (_error->PendingError() == true)
 122       return false;
 123
 124    // Convert the port name/number
 125    char ServStr[300];
 126    if (Port != 0)
 127       snprintf(ServStr,sizeof(ServStr),"%u",Port);
 128    else
 129       snprintf(ServStr,sizeof(ServStr),"%s",Service);
 130
 131    /* We used a cached address record.. Yes this is against the spec but
 132       the way we have setup our rotating dns suggests that this is more
 133       sensible */
 134    if (LastHost != Host || LastPort != Port)
 135    {
 136       Owner->Status(_("Connecting to %s"),Host.c_str());
 137
 138       // Free the old address structure
 139       if (LastHostAddr != 0)
 140       {
 141          freeaddrinfo(LastHostAddr);
 142          LastHostAddr = 0;
 143          LastUsed = 0;
 144       }
 145
 146       // We only understand SOCK_STREAM sockets.
 147       struct addrinfo Hints;
 148       memset(&Hints,0,sizeof(Hints));
 149       Hints.ai_socktype = SOCK_STREAM;
 150       Hints.ai_protocol = 0;
 151
 152       // Resolve both the host and service simultaneously
 153       while (1)
 154       {
 155          int Res;
 156          if ((Res = getaddrinfo(Host.c_str(),ServStr,&Hints,&LastHostAddr)) != 0 ||
 157              LastHostAddr == 0)
 158          {
 159             if (Res == EAI_NONAME || Res == EAI_SERVICE)
 160             {
 161                if (DefPort != 0)
 162                {
 163                   snprintf(ServStr,sizeof(ServStr),"%u",DefPort);
 164                   DefPort = 0;
 165                   continue;
 166                }
 167                Owner->SetFailExtraMsg("\nFailReason: ResolveFailure");
 168                return _error->Error(_("Could not resolve '%s'"),Host.c_str());
 169             }
 170
 171             if (Res == EAI_AGAIN)
 172             {
 173                Owner->SetFailExtraMsg("\nFailReason: TmpResolveFailure");
 174                return _error->Error(_("Temporary failure resolving '%s'"),
 175                                     Host.c_str());
 176             }
 177             return _error->Error(_("Something wicked happened resolving '%s:%s' (%i)"),
 178                                  Host.c_str(),ServStr,Res);
 179          }
 180          break;
 181       }
 182
 183       LastHost = Host;
 184       LastPort = Port;
 185    }
 186
 187    // When we have an IP rotation stay with the last IP.
 188    struct addrinfo *CurHost = LastHostAddr;
 189    if (LastUsed != 0)
 190        CurHost = LastUsed;
 191
 192    while (CurHost != 0)
 193    {
 194       if (DoConnect(CurHost,Host,TimeOut,Fd,Owner) == true)
 195       {
 196          LastUsed = CurHost;
 197          return true;
 198       }
 199       close(Fd);
 200       Fd = -1;
 201
 202       // Ignore UNIX domain sockets
 203       do
 204       {
 205          CurHost = CurHost->ai_next;
 206       }
 207       while (CurHost != 0 && CurHost->ai_family == AF_UNIX);
 208
 209       /* If we reached the end of the search list then wrap around to the
 210          start */
 211       if (CurHost == 0 && LastUsed != 0)
 212          CurHost = LastHostAddr;
 213
 214       // Reached the end of the search cycle
 215       if (CurHost == LastUsed)
 216          break;
 217
 218       if (CurHost != 0)
 219          _error->Discard();
 220    }
 221
 222    if (_error->PendingError() == true)
 223       return false;
 224    return _error->Error(_("Unable to connect to %s %s:"),Host.c_str(),ServStr);
 225 }
 226                                                                         /*}}}*/