]>
Commit | Line | Data |
---|---|---|
0837bd25 AL |
1 | // -*- mode: cpp; mode: fold -*- |
2 | // Description /*{{{*/ | |
7db98ffc | 3 | // $Id: connect.cc,v 1.10.2.1 2004/01/16 18:58:50 mdz Exp $ |
0837bd25 AL |
4 | /* ###################################################################### |
5 | ||
6 | Connect - Replacement connect call | |
7da2b375 AL |
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 | ||
0837bd25 AL |
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> | |
36280399 | 21 | #include <sstream> |
0837bd25 AL |
22 | |
23 | // Internet stuff | |
24 | #include <netinet/in.h> | |
25 | #include <sys/socket.h> | |
26 | #include <arpa/inet.h> | |
27 | #include <netdb.h> | |
28 | ||
29 | #include "rfc2553emu.h" | |
d77559ac | 30 | #include <apti18n.h> |
0837bd25 AL |
31 | /*}}}*/ |
32 | ||
33 | static string LastHost; | |
34 | static int LastPort = 0; | |
35 | static struct addrinfo *LastHostAddr = 0; | |
36 | static struct addrinfo *LastUsed = 0; | |
37 | ||
b2e465d6 AL |
38 | // RotateDNS - Select a new server from a DNS rotation /*{{{*/ |
39 | // --------------------------------------------------------------------- | |
40 | /* This is called during certain errors in order to recover by selecting a | |
41 | new server */ | |
42 | void RotateDNS() | |
43 | { | |
44 | if (LastUsed != 0 && LastUsed->ai_next != 0) | |
45 | LastUsed = LastUsed->ai_next; | |
46 | else | |
47 | LastUsed = LastHostAddr; | |
48 | } | |
49 | /*}}}*/ | |
0837bd25 AL |
50 | // DoConnect - Attempt a connect operation /*{{{*/ |
51 | // --------------------------------------------------------------------- | |
52 | /* This helper function attempts a connection to a single address. */ | |
53 | static bool DoConnect(struct addrinfo *Addr,string Host, | |
54 | unsigned long TimeOut,int &Fd,pkgAcqMethod *Owner) | |
55 | { | |
56 | // Show a status indicator | |
57 | char Name[NI_MAXHOST]; | |
28006885 | 58 | char Service[NI_MAXSERV]; |
b2e465d6 AL |
59 | |
60 | Name[0] = 0; | |
28006885 | 61 | Service[0] = 0; |
0837bd25 | 62 | getnameinfo(Addr->ai_addr,Addr->ai_addrlen, |
28006885 AL |
63 | Name,sizeof(Name),Service,sizeof(Service), |
64 | NI_NUMERICHOST|NI_NUMERICSERV); | |
dc738e7a | 65 | Owner->Status(_("Connecting to %s (%s)"),Host.c_str(),Name); |
b2e465d6 AL |
66 | |
67 | /* If this is an IP rotation store the IP we are using.. If something goes | |
68 | wrong this will get tacked onto the end of the error message */ | |
69 | if (LastHostAddr->ai_next != 0) | |
70 | { | |
36280399 MV |
71 | std::stringstream ss; |
72 | ioprintf(ss, _("[IP: %s %s]"),Name,Service); | |
73 | Owner->SetIP(ss.str()); | |
74 | } | |
b2e465d6 | 75 | |
0837bd25 AL |
76 | // Get a socket |
77 | if ((Fd = socket(Addr->ai_family,Addr->ai_socktype, | |
78 | Addr->ai_protocol)) < 0) | |
dc738e7a | 79 | return _error->Errno("socket",_("Could not create a socket for %s (f=%u t=%u p=%u)"), |
b2e465d6 | 80 | Name,Addr->ai_family,Addr->ai_socktype,Addr->ai_protocol); |
0837bd25 AL |
81 | |
82 | SetNonBlock(Fd,true); | |
83 | if (connect(Fd,Addr->ai_addr,Addr->ai_addrlen) < 0 && | |
84 | errno != EINPROGRESS) | |
dc738e7a AL |
85 | return _error->Errno("connect",_("Cannot initiate the connection " |
86 | "to %s:%s (%s)."),Host.c_str(),Service,Name); | |
0837bd25 AL |
87 | |
88 | /* This implements a timeout for connect by opening the connection | |
89 | nonblocking */ | |
24057ad6 | 90 | if (WaitFd(Fd,true,TimeOut) == false) { |
36280399 | 91 | Owner->SetFailReason("Timeout"); |
dc738e7a AL |
92 | return _error->Error(_("Could not connect to %s:%s (%s), " |
93 | "connection timed out"),Host.c_str(),Service,Name); | |
24057ad6 | 94 | } |
b2e465d6 | 95 | |
0837bd25 AL |
96 | // Check the socket for an error condition |
97 | unsigned int Err; | |
98 | unsigned int Len = sizeof(Err); | |
99 | if (getsockopt(Fd,SOL_SOCKET,SO_ERROR,&Err,&Len) != 0) | |
dc738e7a | 100 | return _error->Errno("getsockopt",_("Failed")); |
0837bd25 AL |
101 | |
102 | if (Err != 0) | |
28006885 AL |
103 | { |
104 | errno = Err; | |
75dd8af1 | 105 | if(errno == ECONNREFUSED) |
36280399 | 106 | Owner->SetFailReason("ConnectionRefused"); |
dc738e7a | 107 | return _error->Errno("connect",_("Could not connect to %s:%s (%s)."),Host.c_str(), |
28006885 AL |
108 | Service,Name); |
109 | } | |
110 | ||
0837bd25 AL |
111 | return true; |
112 | } | |
113 | /*}}}*/ | |
c141b9a9 | 114 | // Connect - Connect to a server /*{{{*/ |
0837bd25 AL |
115 | // --------------------------------------------------------------------- |
116 | /* Performs a connection to the server */ | |
9505213b | 117 | bool Connect(string Host,int Port,const char *Service,int DefPort,int &Fd, |
0837bd25 AL |
118 | unsigned long TimeOut,pkgAcqMethod *Owner) |
119 | { | |
120 | if (_error->PendingError() == true) | |
121 | return false; | |
28006885 AL |
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); | |
0837bd25 AL |
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 | { | |
dc738e7a | 135 | Owner->Status(_("Connecting to %s"),Host.c_str()); |
0837bd25 | 136 | |
0837bd25 AL |
137 | // Free the old address structure |
138 | if (LastHostAddr != 0) | |
139 | { | |
140 | freeaddrinfo(LastHostAddr); | |
141 | LastHostAddr = 0; | |
28006885 | 142 | LastUsed = 0; |
0837bd25 AL |
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; | |
28006885 | 149 | Hints.ai_protocol = 0; |
0837bd25 AL |
150 | |
151 | // Resolve both the host and service simultaneously | |
9505213b | 152 | while (1) |
c141b9a9 | 153 | { |
9505213b | 154 | int Res; |
28006885 | 155 | if ((Res = getaddrinfo(Host.c_str(),ServStr,&Hints,&LastHostAddr)) != 0 || |
9505213b AL |
156 | LastHostAddr == 0) |
157 | { | |
72472b95 | 158 | if (Res == EAI_NONAME || Res == EAI_SERVICE) |
9505213b AL |
159 | { |
160 | if (DefPort != 0) | |
161 | { | |
28006885 | 162 | snprintf(ServStr,sizeof(ServStr),"%u",DefPort); |
9505213b AL |
163 | DefPort = 0; |
164 | continue; | |
165 | } | |
59271f62 | 166 | Owner->SetFailReason("ResolveFailure"); |
dc738e7a | 167 | return _error->Error(_("Could not resolve '%s'"),Host.c_str()); |
9505213b AL |
168 | } |
169 | ||
4fe6e0c2 | 170 | if (Res == EAI_AGAIN) |
25182152 | 171 | { |
36280399 | 172 | Owner->SetFailReason("TmpResolveFailure"); |
dc738e7a | 173 | return _error->Error(_("Temporary failure resolving '%s'"), |
4fe6e0c2 | 174 | Host.c_str()); |
25182152 | 175 | } |
dc738e7a | 176 | return _error->Error(_("Something wicked happened resolving '%s:%s' (%i)"), |
b2e465d6 | 177 | Host.c_str(),ServStr,Res); |
9505213b AL |
178 | } |
179 | break; | |
c141b9a9 AL |
180 | } |
181 | ||
0837bd25 AL |
182 | LastHost = Host; |
183 | LastPort = Port; | |
0837bd25 AL |
184 | } |
185 | ||
28006885 | 186 | // When we have an IP rotation stay with the last IP. |
0837bd25 AL |
187 | struct addrinfo *CurHost = LastHostAddr; |
188 | if (LastUsed != 0) | |
189 | CurHost = LastUsed; | |
190 | ||
191 | while (CurHost != 0) | |
192 | { | |
193 | if (DoConnect(CurHost,Host,TimeOut,Fd,Owner) == true) | |
194 | { | |
195 | LastUsed = CurHost; | |
196 | return true; | |
197 | } | |
198 | close(Fd); | |
199 | Fd = -1; | |
200 | ||
28006885 AL |
201 | // Ignore UNIX domain sockets |
202 | do | |
203 | { | |
204 | CurHost = CurHost->ai_next; | |
205 | } | |
206 | while (CurHost != 0 && CurHost->ai_family == AF_UNIX); | |
b2e465d6 AL |
207 | |
208 | /* If we reached the end of the search list then wrap around to the | |
209 | start */ | |
210 | if (CurHost == 0 && LastUsed != 0) | |
211 | CurHost = LastHostAddr; | |
212 | ||
213 | // Reached the end of the search cycle | |
214 | if (CurHost == LastUsed) | |
215 | break; | |
216 | ||
0837bd25 AL |
217 | if (CurHost != 0) |
218 | _error->Discard(); | |
b2e465d6 | 219 | } |
28006885 | 220 | |
dd1fd92b | 221 | if (_error->PendingError() == true) |
b2e465d6 | 222 | return false; |
dc738e7a | 223 | return _error->Error(_("Unable to connect to %s %s:"),Host.c_str(),ServStr); |
0837bd25 AL |
224 | } |
225 | /*}}}*/ |