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