]> git.saurik.com Git - apt.git/blame - methods/ftp.cc
stop displaying time of build in online help
[apt.git] / methods / ftp.cc
CommitLineData
30b30ec1
AL
1// -*- mode: cpp; mode: fold -*-
2// Description /*{{{*/
7db98ffc 3// $Id: ftp.cc,v 1.31.2.1 2004/01/16 18:58:50 mdz Exp $
30b30ec1
AL
4/* ######################################################################
5
1e3f4083 6 FTP Acquire Method - This is the FTP acquire method for APT.
30b30ec1
AL
7
8 This is a very simple implementation that does not try to optimize
9 at all. Commands are sent syncronously with the FTP server (as the
10 rfc recommends, but it is not really necessary..) and no tricks are
11 done to speed things along.
934b6582
AL
12
13 RFC 2428 describes the IPv6 FTP behavior
14
30b30ec1
AL
15 ##################################################################### */
16 /*}}}*/
17// Include Files /*{{{*/
ea542140
DK
18#include <config.h>
19
30b30ec1
AL
20#include <apt-pkg/fileutl.h>
21#include <apt-pkg/acquire-method.h>
22#include <apt-pkg/error.h>
63b1700f 23#include <apt-pkg/hashes.h>
f1c081b6 24#include <apt-pkg/netrc.h>
472ff00e 25#include <apt-pkg/configuration.h>
453b82a3 26#include <apt-pkg/strutl.h>
30b30ec1 27
453b82a3
DK
28#include <ctype.h>
29#include <stdlib.h>
30#include <string.h>
30b30ec1
AL
31#include <sys/stat.h>
32#include <sys/time.h>
30b30ec1
AL
33#include <unistd.h>
34#include <signal.h>
35#include <stdio.h>
36#include <errno.h>
37#include <stdarg.h>
076cc664 38#include <iostream>
30b30ec1
AL
39
40// Internet stuff
41#include <netinet/in.h>
42#include <sys/socket.h>
43#include <arpa/inet.h>
44#include <netdb.h>
45
6d13bbca 46#include "rfc2553emu.h"
0837bd25 47#include "connect.h"
ce0ae89a 48#include "ftp.h"
453b82a3 49
ea542140 50#include <apti18n.h>
30b30ec1
AL
51 /*}}}*/
52
076cc664
AL
53using namespace std;
54
b2e465d6
AL
55/* This table is for the EPRT and EPSV commands, it maps the OS address
56 family to the IETF address families */
57struct AFMap
58{
59 unsigned long Family;
60 unsigned long IETFFamily;
61};
62
63#ifndef AF_INET6
67c3067f 64struct AFMap AFMap[] = {{AF_INET,1},{0, 0}};
b2e465d6 65#else
67c3067f 66struct AFMap AFMap[] = {{AF_INET,1},{AF_INET6,2},{0, 0}};
b2e465d6
AL
67#endif
68
30b30ec1
AL
69unsigned long TimeOut = 120;
70URI Proxy;
ce0ae89a
AL
71string FtpMethod::FailFile;
72int FtpMethod::FailFd = -1;
73time_t FtpMethod::FailTime = 0;
30b30ec1
AL
74
75// FTPConn::FTPConn - Constructor /*{{{*/
76// ---------------------------------------------------------------------
77/* */
25613a61 78FTPConn::FTPConn(URI Srv) : Len(0), ServerFd(-1), DataFd(-1),
f5a34606 79 DataListenFd(-1), ServerName(Srv),
25613a61
DK
80 ForceExtended(false), TryPassive(true),
81 PeerAddrLen(0), ServerAddrLen(0)
30b30ec1 82{
ce0ae89a 83 Debug = _config->FindB("Debug::Acquire::Ftp",false);
b2e465d6 84 PasvAddr = 0;
dcaa1185 85 Buffer[0] = '\0';
30b30ec1
AL
86}
87 /*}}}*/
88// FTPConn::~FTPConn - Destructor /*{{{*/
89// ---------------------------------------------------------------------
90/* */
91FTPConn::~FTPConn()
92{
93 Close();
94}
95 /*}}}*/
96// FTPConn::Close - Close down the connection /*{{{*/
97// ---------------------------------------------------------------------
98/* Just tear down the socket and data socket */
99void FTPConn::Close()
100{
101 close(ServerFd);
102 ServerFd = -1;
103 close(DataFd);
104 DataFd = -1;
105 close(DataListenFd);
106 DataListenFd = -1;
b2e465d6
AL
107
108 if (PasvAddr != 0)
109 freeaddrinfo(PasvAddr);
110 PasvAddr = 0;
30b30ec1
AL
111}
112 /*}}}*/
113// FTPConn::Open - Open a new connection /*{{{*/
114// ---------------------------------------------------------------------
115/* Connect to the server using a non-blocking connection and perform a
116 login. */
ce0ae89a 117bool FTPConn::Open(pkgAcqMethod *Owner)
30b30ec1
AL
118{
119 // Use the already open connection if possible.
120 if (ServerFd != -1)
121 return true;
122
123 Close();
b2e465d6 124
30b30ec1 125 // Determine the proxy setting
788a8f42
EL
126 string SpecificProxy = _config->Find("Acquire::ftp::Proxy::" + ServerName.Host);
127 if (!SpecificProxy.empty())
30b30ec1 128 {
788a8f42
EL
129 if (SpecificProxy == "DIRECT")
130 Proxy = "";
131 else
132 Proxy = SpecificProxy;
30b30ec1
AL
133 }
134 else
788a8f42
EL
135 {
136 string DefProxy = _config->Find("Acquire::ftp::Proxy");
137 if (!DefProxy.empty())
138 {
139 Proxy = DefProxy;
140 }
141 else
142 {
143 char* result = getenv("ftp_proxy");
144 Proxy = result ? result : "";
145 }
146 }
147
f8081133
AL
148 // Parse no_proxy, a , separated list of domains
149 if (getenv("no_proxy") != 0)
150 {
151 if (CheckDomainList(ServerName.Host,getenv("no_proxy")) == true)
152 Proxy = "";
153 }
154
30b30ec1 155 // Determine what host and port to use based on the proxy settings
6d13bbca 156 int Port = 0;
30b30ec1
AL
157 string Host;
158 if (Proxy.empty() == true)
159 {
160 if (ServerName.Port != 0)
161 Port = ServerName.Port;
30b30ec1
AL
162 Host = ServerName.Host;
163 }
164 else
165 {
166 if (Proxy.Port != 0)
167 Port = Proxy.Port;
168 Host = Proxy.Host;
169 }
6d13bbca 170
b2e465d6
AL
171 /* Connect to the remote server. Since FTP is connection oriented we
172 want to make sure we get a new server every time we reconnect */
173 RotateDNS();
9505213b 174 if (Connect(Host,Port,"ftp",21,ServerFd,TimeOut,Owner) == false)
0837bd25 175 return false;
c968dc2f
AL
176
177 // Login must be before getpeername otherwise dante won't work.
dc738e7a 178 Owner->Status(_("Logging in"));
c968dc2f 179 bool Res = Login();
b2e465d6
AL
180
181 // Get the remote server's address
182 PeerAddrLen = sizeof(PeerAddr);
183 if (getpeername(ServerFd,(sockaddr *)&PeerAddr,&PeerAddrLen) != 0)
dc738e7a 184 return _error->Errno("getpeername",_("Unable to determine the peer name"));
30b30ec1 185
b2e465d6
AL
186 // Get the local machine's address
187 ServerAddrLen = sizeof(ServerAddr);
188 if (getsockname(ServerFd,(sockaddr *)&ServerAddr,&ServerAddrLen) != 0)
dc738e7a 189 return _error->Errno("getsockname",_("Unable to determine the local name"));
b2e465d6 190
c968dc2f 191 return Res;
30b30ec1
AL
192}
193 /*}}}*/
194// FTPConn::Login - Login to the remote server /*{{{*/
195// ---------------------------------------------------------------------
196/* This performs both normal login and proxy login using a simples script
197 stored in the config file. */
198bool FTPConn::Login()
199{
200 unsigned int Tag;
201 string Msg;
202
203 // Setup the variables needed for authentication
204 string User = "anonymous";
44a38e53 205 string Pass = "apt_get_ftp_2.1@debian.linux.user";
30b30ec1
AL
206
207 // Fill in the user/pass
208 if (ServerName.User.empty() == false)
209 User = ServerName.User;
210 if (ServerName.Password.empty() == false)
211 Pass = ServerName.Password;
212
213 // Perform simple login
214 if (Proxy.empty() == true)
215 {
216 // Read the initial response
217 if (ReadResp(Tag,Msg) == false)
218 return false;
219 if (Tag >= 400)
db0db9fe 220 return _error->Error(_("The server refused the connection and said: %s"),Msg.c_str());
30b30ec1
AL
221
222 // Send the user
223 if (WriteMsg(Tag,Msg,"USER %s",User.c_str()) == false)
224 return false;
225 if (Tag >= 400)
dc738e7a 226 return _error->Error(_("USER failed, server said: %s"),Msg.c_str());
30b30ec1 227
2391e7b5
AL
228 if (Tag == 331) { // 331 User name okay, need password.
229 // Send the Password
230 if (WriteMsg(Tag,Msg,"PASS %s",Pass.c_str()) == false)
231 return false;
232 if (Tag >= 400)
233 return _error->Error(_("PASS failed, server said: %s"),Msg.c_str());
234 }
30b30ec1
AL
235
236 // Enter passive mode
10861bb5
AL
237 if (_config->Exists("Acquire::FTP::Passive::" + ServerName.Host) == true)
238 TryPassive = _config->FindB("Acquire::FTP::Passive::" + ServerName.Host,true);
30b30ec1 239 else
b2e465d6 240 TryPassive = _config->FindB("Acquire::FTP::Passive",true);
30b30ec1
AL
241 }
242 else
243 {
244 // Read the initial response
245 if (ReadResp(Tag,Msg) == false)
246 return false;
247 if (Tag >= 400)
1169dbfa 248 return _error->Error(_("The server refused the connection and said: %s"),Msg.c_str());
30b30ec1
AL
249
250 // Perform proxy script execution
251 Configuration::Item const *Opts = _config->Tree("Acquire::ftp::ProxyLogin");
252 if (Opts == 0 || Opts->Child == 0)
dc738e7a
AL
253 return _error->Error(_("A proxy server was specified but no login "
254 "script, Acquire::ftp::ProxyLogin is empty."));
30b30ec1
AL
255 Opts = Opts->Child;
256
257 // Iterate over the entire login script
258 for (; Opts != 0; Opts = Opts->Next)
259 {
260 if (Opts->Value.empty() == true)
261 continue;
262
263 // Substitute the variables into the command
264 char SitePort[20];
ce0ae89a
AL
265 if (ServerName.Port != 0)
266 sprintf(SitePort,"%u",ServerName.Port);
267 else
10861bb5 268 strcpy(SitePort,"21");
30b30ec1
AL
269 string Tmp = Opts->Value;
270 Tmp = SubstVar(Tmp,"$(PROXY_USER)",Proxy.User);
271 Tmp = SubstVar(Tmp,"$(PROXY_PASS)",Proxy.Password);
272 Tmp = SubstVar(Tmp,"$(SITE_USER)",User);
273 Tmp = SubstVar(Tmp,"$(SITE_PASS)",Pass);
274 Tmp = SubstVar(Tmp,"$(SITE_PORT)",SitePort);
275 Tmp = SubstVar(Tmp,"$(SITE)",ServerName.Host);
276
277 // Send the command
278 if (WriteMsg(Tag,Msg,"%s",Tmp.c_str()) == false)
279 return false;
280 if (Tag >= 400)
dc738e7a 281 return _error->Error(_("Login script command '%s' failed, server said: %s"),Tmp.c_str(),Msg.c_str());
30b30ec1
AL
282 }
283
284 // Enter passive mode
285 TryPassive = false;
10861bb5
AL
286 if (_config->Exists("Acquire::FTP::Passive::" + ServerName.Host) == true)
287 TryPassive = _config->FindB("Acquire::FTP::Passive::" + ServerName.Host,true);
30b30ec1
AL
288 else
289 {
10861bb5
AL
290 if (_config->Exists("Acquire::FTP::Proxy::Passive") == true)
291 TryPassive = _config->FindB("Acquire::FTP::Proxy::Passive",true);
30b30ec1 292 else
10861bb5 293 TryPassive = _config->FindB("Acquire::FTP::Passive",true);
30b30ec1
AL
294 }
295 }
296
b2e465d6
AL
297 // Force the use of extended commands
298 if (_config->Exists("Acquire::FTP::ForceExtended::" + ServerName.Host) == true)
299 ForceExtended = _config->FindB("Acquire::FTP::ForceExtended::" + ServerName.Host,true);
300 else
301 ForceExtended = _config->FindB("Acquire::FTP::ForceExtended",false);
302
30b30ec1
AL
303 // Binary mode
304 if (WriteMsg(Tag,Msg,"TYPE I") == false)
305 return false;
306 if (Tag >= 400)
dc738e7a 307 return _error->Error(_("TYPE failed, server said: %s"),Msg.c_str());
30b30ec1
AL
308
309 return true;
310}
311 /*}}}*/
312// FTPConn::ReadLine - Read a line from the server /*{{{*/
313// ---------------------------------------------------------------------
314/* This performs a very simple buffered read. */
315bool FTPConn::ReadLine(string &Text)
316{
2e90f6e0
AL
317 if (ServerFd == -1)
318 return false;
319
30b30ec1
AL
320 // Suck in a line
321 while (Len < sizeof(Buffer))
322 {
323 // Scan the buffer for a new line
324 for (unsigned int I = 0; I != Len; I++)
325 {
326 // Escape some special chars
327 if (Buffer[I] == 0)
328 Buffer[I] = '?';
329
330 // End of line?
331 if (Buffer[I] != '\n')
332 continue;
333
334 I++;
335 Text = string(Buffer,I);
336 memmove(Buffer,Buffer+I,Len - I);
337 Len -= I;
338 return true;
339 }
340
341 // Wait for some data..
342 if (WaitFd(ServerFd,false,TimeOut) == false)
ce0ae89a
AL
343 {
344 Close();
dc738e7a 345 return _error->Error(_("Connection timeout"));
ce0ae89a 346 }
30b30ec1
AL
347
348 // Suck it back
2e90f6e0 349 int Res = read(ServerFd,Buffer + Len,sizeof(Buffer) - Len);
b2e465d6 350 if (Res == 0)
dc738e7a 351 _error->Error(_("Server closed the connection"));
30b30ec1 352 if (Res <= 0)
ce0ae89a 353 {
dc738e7a 354 _error->Errno("read",_("Read error"));
ce0ae89a 355 Close();
f93d1355 356 return false;
ce0ae89a 357 }
30b30ec1
AL
358 Len += Res;
359 }
360
dc738e7a 361 return _error->Error(_("A response overflowed the buffer."));
30b30ec1
AL
362}
363 /*}}}*/
364// FTPConn::ReadResp - Read a full response from the server /*{{{*/
365// ---------------------------------------------------------------------
366/* This reads a reply code from the server, it handles both p */
367bool FTPConn::ReadResp(unsigned int &Ret,string &Text)
368{
369 // Grab the first line of the response
370 string Msg;
371 if (ReadLine(Msg) == false)
372 return false;
373
374 // Get the ID code
375 char *End;
376 Ret = strtol(Msg.c_str(),&End,10);
377 if (End - Msg.c_str() != 3)
dc738e7a 378 return _error->Error(_("Protocol corruption"));
30b30ec1
AL
379
380 // All done ?
381 Text = Msg.c_str()+4;
382 if (*End == ' ')
383 {
384 if (Debug == true)
ce0ae89a 385 cerr << "<- '" << QuoteString(Text,"") << "'" << endl;
30b30ec1
AL
386 return true;
387 }
388
389 if (*End != '-')
dc738e7a 390 return _error->Error(_("Protocol corruption"));
30b30ec1
AL
391
392 /* Okay, here we do the continued message trick. This is foolish, but
393 proftpd follows the protocol as specified and wu-ftpd doesn't, so
394 we filter. I wonder how many clients break if you use proftpd and
395 put a '- in the 3rd spot in the message? */
396 char Leader[4];
397 strncpy(Leader,Msg.c_str(),3);
398 Leader[3] = 0;
399 while (ReadLine(Msg) == true)
400 {
401 // Short, it must be using RFC continuation..
402 if (Msg.length() < 4)
403 {
404 Text += Msg;
405 continue;
406 }
407
408 // Oops, finished
409 if (strncmp(Msg.c_str(),Leader,3) == 0 && Msg[3] == ' ')
410 {
411 Text += Msg.c_str()+4;
412 break;
413 }
414
415 // This message has the wu-ftpd style reply code prefixed
416 if (strncmp(Msg.c_str(),Leader,3) == 0 && Msg[3] == '-')
417 {
418 Text += Msg.c_str()+4;
419 continue;
420 }
421
422 // Must be RFC style prefixing
423 Text += Msg;
424 }
425
426 if (Debug == true && _error->PendingError() == false)
ce0ae89a 427 cerr << "<- '" << QuoteString(Text,"") << "'" << endl;
30b30ec1
AL
428
429 return !_error->PendingError();
430}
431 /*}}}*/
432// FTPConn::WriteMsg - Send a message to the server /*{{{*/
433// ---------------------------------------------------------------------
434/* Simple printf like function.. */
435bool FTPConn::WriteMsg(unsigned int &Ret,string &Text,const char *Fmt,...)
436{
437 va_list args;
438 va_start(args,Fmt);
439
440 // sprintf the description
441 char S[400];
442 vsnprintf(S,sizeof(S) - 4,Fmt,args);
443 strcat(S,"\r\n");
11d0fb91 444 va_end(args);
30b30ec1
AL
445
446 if (Debug == true)
ce0ae89a 447 cerr << "-> '" << QuoteString(S,"") << "'" << endl;
30b30ec1
AL
448
449 // Send it off
450 unsigned long Len = strlen(S);
451 unsigned long Start = 0;
452 while (Len != 0)
453 {
454 if (WaitFd(ServerFd,true,TimeOut) == false)
ce0ae89a
AL
455 {
456 Close();
dc738e7a 457 return _error->Error(_("Connection timeout"));
ce0ae89a 458 }
30b30ec1
AL
459
460 int Res = write(ServerFd,S + Start,Len);
461 if (Res <= 0)
ce0ae89a 462 {
db0db9fe 463 _error->Errno("write",_("Write error"));
ce0ae89a 464 Close();
f93d1355 465 return false;
ce0ae89a
AL
466 }
467
30b30ec1
AL
468 Len -= Res;
469 Start += Res;
470 }
471
472 return ReadResp(Ret,Text);
473}
474 /*}}}*/
475// FTPConn::GoPasv - Enter Passive mode /*{{{*/
476// ---------------------------------------------------------------------
477/* Try to enter passive mode, the return code does not indicate if passive
478 mode could or could not be established, only if there was a fatal error.
b2e465d6 479 We have to enter passive mode every time we make a data connection :| */
30b30ec1
AL
480bool FTPConn::GoPasv()
481{
b2e465d6
AL
482 /* The PASV command only works on IPv4 sockets, even though it could
483 in theory suppory IPv6 via an all zeros reply */
484 if (((struct sockaddr *)&PeerAddr)->sa_family != AF_INET ||
485 ForceExtended == true)
486 return ExtGoPasv();
487
488 if (PasvAddr != 0)
489 freeaddrinfo(PasvAddr);
490 PasvAddr = 0;
491
30b30ec1
AL
492 // Try to enable pasv mode
493 unsigned int Tag;
494 string Msg;
495 if (WriteMsg(Tag,Msg,"PASV") == false)
496 return false;
497
498 // Unsupported function
499 string::size_type Pos = Msg.find('(');
500 if (Tag >= 400 || Pos == string::npos)
30b30ec1 501 return true;
30b30ec1
AL
502
503 // Scan it
504 unsigned a0,a1,a2,a3,p0,p1;
505 if (sscanf(Msg.c_str() + Pos,"(%u,%u,%u,%u,%u,%u)",&a0,&a1,&a2,&a3,&p0,&p1) != 6)
b2e465d6
AL
506 return true;
507
508 /* Some evil servers return 0 to mean their addr. We can actually speak
509 to these servers natively using IPv6 */
510 if (a0 == 0 && a1 == 0 && a2 == 0 && a3 == 0)
30b30ec1 511 {
b2e465d6
AL
512 // Get the IP in text form
513 char Name[NI_MAXHOST];
514 char Service[NI_MAXSERV];
515 getnameinfo((struct sockaddr *)&PeerAddr,PeerAddrLen,
516 Name,sizeof(Name),Service,sizeof(Service),
517 NI_NUMERICHOST|NI_NUMERICSERV);
518
519 struct addrinfo Hints;
520 memset(&Hints,0,sizeof(Hints));
521 Hints.ai_socktype = SOCK_STREAM;
522 Hints.ai_family = ((struct sockaddr *)&PeerAddr)->sa_family;
523 Hints.ai_flags |= AI_NUMERICHOST;
524
525 // Get a new passive address.
526 char Port[100];
527 snprintf(Port,sizeof(Port),"%u",(p0 << 8) + p1);
528 if (getaddrinfo(Name,Port,&Hints,&PasvAddr) != 0)
529 return true;
30b30ec1
AL
530 return true;
531 }
532
b2e465d6
AL
533 struct addrinfo Hints;
534 memset(&Hints,0,sizeof(Hints));
535 Hints.ai_socktype = SOCK_STREAM;
536 Hints.ai_family = AF_INET;
537 Hints.ai_flags |= AI_NUMERICHOST;
30b30ec1 538
b2e465d6
AL
539 // Get a new passive address.
540 char Port[100];
541 snprintf(Port,sizeof(Port),"%u",(p0 << 8) + p1);
542 char Name[100];
543 snprintf(Name,sizeof(Name),"%u.%u.%u.%u",a0,a1,a2,a3);
544 if (getaddrinfo(Name,Port,&Hints,&PasvAddr) != 0)
545 return true;
546 return true;
547}
548 /*}}}*/
549// FTPConn::ExtGoPasv - Enter Extended Passive mode /*{{{*/
550// ---------------------------------------------------------------------
551/* Try to enter extended passive mode. See GoPasv above and RFC 2428 */
552bool FTPConn::ExtGoPasv()
553{
554 if (PasvAddr != 0)
555 freeaddrinfo(PasvAddr);
556 PasvAddr = 0;
557
558 // Try to enable pasv mode
559 unsigned int Tag;
560 string Msg;
561 if (WriteMsg(Tag,Msg,"EPSV") == false)
562 return false;
563
564 // Unsupported function
565 string::size_type Pos = Msg.find('(');
566 if (Tag >= 400 || Pos == string::npos)
567 return true;
568
569 // Scan it
570 string::const_iterator List[4];
571 unsigned Count = 0;
572 Pos++;
f7f0d6c7 573 for (string::const_iterator I = Msg.begin() + Pos; I < Msg.end(); ++I)
b2e465d6
AL
574 {
575 if (*I != Msg[Pos])
576 continue;
577 if (Count >= 4)
578 return true;
579 List[Count++] = I;
580 }
581 if (Count != 4)
582 return true;
583
584 // Break it up ..
585 unsigned long Proto = 0;
586 unsigned long Port = 0;
587 string IP;
588 IP = string(List[1]+1,List[2]);
589 Port = atoi(string(List[2]+1,List[3]).c_str());
590 if (IP.empty() == false)
591 Proto = atoi(string(List[0]+1,List[1]).c_str());
592
593 if (Port == 0)
594 return false;
595
596 // String version of the port
597 char PStr[100];
598 snprintf(PStr,sizeof(PStr),"%lu",Port);
599
600 // Get the IP in text form
601 struct addrinfo Hints;
602 memset(&Hints,0,sizeof(Hints));
603 Hints.ai_socktype = SOCK_STREAM;
604 Hints.ai_flags |= AI_NUMERICHOST;
605
606 /* The RFC defined case, connect to the old IP/protocol using the
607 new port. */
608 if (IP.empty() == true)
30b30ec1 609 {
b2e465d6
AL
610 // Get the IP in text form
611 char Name[NI_MAXHOST];
612 char Service[NI_MAXSERV];
613 getnameinfo((struct sockaddr *)&PeerAddr,PeerAddrLen,
614 Name,sizeof(Name),Service,sizeof(Service),
615 NI_NUMERICHOST|NI_NUMERICSERV);
616 IP = Name;
617 Hints.ai_family = ((struct sockaddr *)&PeerAddr)->sa_family;
30b30ec1
AL
618 }
619 else
620 {
b2e465d6
AL
621 // Get the family..
622 Hints.ai_family = 0;
623 for (unsigned J = 0; AFMap[J].Family != 0; J++)
624 if (AFMap[J].IETFFamily == Proto)
625 Hints.ai_family = AFMap[J].Family;
626 if (Hints.ai_family == 0)
627 return true;
30b30ec1
AL
628 }
629
b2e465d6 630 // Get a new passive address.
cddbc86d 631 if (getaddrinfo(IP.c_str(),PStr,&Hints,&PasvAddr) != 0)
b2e465d6 632 return true;
30b30ec1
AL
633
634 return true;
635}
636 /*}}}*/
637// FTPConn::Size - Return the size of a file /*{{{*/
638// ---------------------------------------------------------------------
639/* Grab the file size from the server, 0 means no size or empty file */
650faab0 640bool FTPConn::Size(const char *Path,unsigned long long &Size)
30b30ec1
AL
641{
642 // Query the size
643 unsigned int Tag;
644 string Msg;
ce0ae89a 645 Size = 0;
30b30ec1
AL
646 if (WriteMsg(Tag,Msg,"SIZE %s",Path) == false)
647 return false;
648
649 char *End;
650faab0 650 Size = strtoull(Msg.c_str(),&End,10);
30b30ec1 651 if (Tag >= 400 || End == Msg.c_str())
ce0ae89a
AL
652 Size = 0;
653 return true;
30b30ec1
AL
654}
655 /*}}}*/
656// FTPConn::ModTime - Return the modification time of the file /*{{{*/
657// ---------------------------------------------------------------------
658/* Like Size no error is returned if the command is not supported. If the
659 command fails then time is set to the current time of day to fool
660 date checks. */
661bool FTPConn::ModTime(const char *Path, time_t &Time)
662{
663 Time = time(&Time);
664
665 // Query the mod time
666 unsigned int Tag;
667 string Msg;
668 if (WriteMsg(Tag,Msg,"MDTM %s",Path) == false)
669 return false;
670 if (Tag >= 400 || Msg.empty() == true || isdigit(Msg[0]) == 0)
671 return true;
672
673 // Parse it
96cc64a5 674 return FTPMDTMStrToTime(Msg.c_str(), Time);
30b30ec1
AL
675}
676 /*}}}*/
677// FTPConn::CreateDataFd - Get a data connection /*{{{*/
678// ---------------------------------------------------------------------
679/* Create the data connection. Call FinalizeDataFd after this though.. */
680bool FTPConn::CreateDataFd()
681{
682 close(DataFd);
683 DataFd = -1;
684
685 // Attempt to enter passive mode.
686 if (TryPassive == true)
687 {
688 if (GoPasv() == false)
689 return false;
690
691 // Oops, didn't work out, don't bother trying again.
b2e465d6 692 if (PasvAddr == 0)
30b30ec1
AL
693 TryPassive = false;
694 }
695
696 // Passive mode?
b2e465d6 697 if (PasvAddr != 0)
30b30ec1
AL
698 {
699 // Get a socket
b2e465d6
AL
700 if ((DataFd = socket(PasvAddr->ai_family,PasvAddr->ai_socktype,
701 PasvAddr->ai_protocol)) < 0)
dc738e7a 702 return _error->Errno("socket",_("Could not create a socket"));
30b30ec1
AL
703
704 // Connect to the server
705 SetNonBlock(DataFd,true);
b2e465d6 706 if (connect(DataFd,PasvAddr->ai_addr,PasvAddr->ai_addrlen) < 0 &&
30b30ec1 707 errno != EINPROGRESS)
dc738e7a 708 return _error->Errno("socket",_("Could not create a socket"));
30b30ec1
AL
709
710 /* This implements a timeout for connect by opening the connection
6d13bbca 711 nonblocking */
9b148dad 712 if (WaitFd(DataFd,true,TimeOut) == false)
dc738e7a 713 return _error->Error(_("Could not connect data socket, connection timed out"));
30b30ec1
AL
714 unsigned int Err;
715 unsigned int Len = sizeof(Err);
9b148dad 716 if (getsockopt(DataFd,SOL_SOCKET,SO_ERROR,&Err,&Len) != 0)
dc738e7a 717 return _error->Errno("getsockopt",_("Failed"));
30b30ec1 718 if (Err != 0)
dc738e7a 719 return _error->Error(_("Could not connect passive socket."));
b2e465d6 720
30b30ec1
AL
721 return true;
722 }
723
724 // Port mode :<
2de438e7
AL
725 close(DataListenFd);
726 DataListenFd = -1;
727
b2e465d6 728 // Get the information for a listening socket.
ce3c2407 729 struct addrinfo *BindAddr = NULL;
b2e465d6
AL
730 struct addrinfo Hints;
731 memset(&Hints,0,sizeof(Hints));
732 Hints.ai_socktype = SOCK_STREAM;
733 Hints.ai_flags |= AI_PASSIVE;
734 Hints.ai_family = ((struct sockaddr *)&ServerAddr)->sa_family;
cddbc86d 735 if (getaddrinfo(0,"0",&Hints,&BindAddr) != 0 || BindAddr == NULL)
dc738e7a 736 return _error->Error(_("getaddrinfo was unable to get a listening socket"));
b2e465d6
AL
737
738 // Construct the socket
739 if ((DataListenFd = socket(BindAddr->ai_family,BindAddr->ai_socktype,
740 BindAddr->ai_protocol)) < 0)
741 {
742 freeaddrinfo(BindAddr);
dc738e7a 743 return _error->Errno("socket",_("Could not create a socket"));
b2e465d6 744 }
30b30ec1 745
2de438e7 746 // Bind and listen
b2e465d6
AL
747 if (bind(DataListenFd,BindAddr->ai_addr,BindAddr->ai_addrlen) < 0)
748 {
749 freeaddrinfo(BindAddr);
dc738e7a 750 return _error->Errno("bind",_("Could not bind a socket"));
b2e465d6
AL
751 }
752 freeaddrinfo(BindAddr);
2de438e7 753 if (listen(DataListenFd,1) < 0)
dc738e7a 754 return _error->Errno("listen",_("Could not listen on the socket"));
2de438e7
AL
755 SetNonBlock(DataListenFd,true);
756
757 // Determine the name to send to the remote
b2e465d6
AL
758 struct sockaddr_storage Addr;
759 socklen_t AddrLen = sizeof(Addr);
760 if (getsockname(DataListenFd,(sockaddr *)&Addr,&AddrLen) < 0)
dc738e7a
AL
761 return _error->Errno("getsockname",_("Could not determine the socket's name"));
762
30b30ec1 763
b2e465d6
AL
764 // Reverse the address. We need the server address and the data port.
765 char Name[NI_MAXHOST];
766 char Service[NI_MAXSERV];
767 char Service2[NI_MAXSERV];
768 getnameinfo((struct sockaddr *)&Addr,AddrLen,
769 Name,sizeof(Name),Service,sizeof(Service),
770 NI_NUMERICHOST|NI_NUMERICSERV);
771 getnameinfo((struct sockaddr *)&ServerAddr,ServerAddrLen,
772 Name,sizeof(Name),Service2,sizeof(Service2),
773 NI_NUMERICHOST|NI_NUMERICSERV);
774
775 // Send off an IPv4 address in the old port format
776 if (((struct sockaddr *)&Addr)->sa_family == AF_INET &&
777 ForceExtended == false)
778 {
779 // Convert the dots in the quad into commas
780 for (char *I = Name; *I != 0; I++)
781 if (*I == '.')
782 *I = ',';
783 unsigned long Port = atoi(Service);
784
785 // Send the port command
786 unsigned int Tag;
787 string Msg;
788 if (WriteMsg(Tag,Msg,"PORT %s,%d,%d",
789 Name,
790 (int)(Port >> 8) & 0xff, (int)(Port & 0xff)) == false)
791 return false;
792 if (Tag >= 400)
dc738e7a 793 return _error->Error(_("Unable to send PORT command"));
b2e465d6
AL
794 return true;
795 }
796
797 // Construct an EPRT command
798 unsigned Proto = 0;
799 for (unsigned J = 0; AFMap[J].Family != 0; J++)
800 if (AFMap[J].Family == ((struct sockaddr *)&Addr)->sa_family)
801 Proto = AFMap[J].IETFFamily;
802 if (Proto == 0)
dc738e7a 803 return _error->Error(_("Unknown address family %u (AF_*)"),
b2e465d6
AL
804 ((struct sockaddr *)&Addr)->sa_family);
805
806 // Send the EPRT command
30b30ec1
AL
807 unsigned int Tag;
808 string Msg;
b2e465d6 809 if (WriteMsg(Tag,Msg,"EPRT |%u|%s|%s|",Proto,Name,Service) == false)
30b30ec1
AL
810 return false;
811 if (Tag >= 400)
dc738e7a 812 return _error->Error(_("EPRT failed, server said: %s"),Msg.c_str());
30b30ec1
AL
813 return true;
814}
815 /*}}}*/
816// FTPConn::Finalize - Complete the Data connection /*{{{*/
817// ---------------------------------------------------------------------
818/* If the connection is in port mode this waits for the other end to hook
819 up to us. */
820bool FTPConn::Finalize()
821{
822 // Passive mode? Do nothing
b2e465d6 823 if (PasvAddr != 0)
30b30ec1
AL
824 return true;
825
826 // Close any old socket..
827 close(DataFd);
828 DataFd = -1;
829
830 // Wait for someone to connect..
831 if (WaitFd(DataListenFd,false,TimeOut) == false)
dc738e7a 832 return _error->Error(_("Data socket connect timed out"));
30b30ec1
AL
833
834 // Accept the connection
835 struct sockaddr_in Addr;
836 socklen_t Len = sizeof(Addr);
837 DataFd = accept(DataListenFd,(struct sockaddr *)&Addr,&Len);
838 if (DataFd < 0)
dc738e7a 839 return _error->Errno("accept",_("Unable to accept connection"));
30b30ec1 840
2de438e7
AL
841 close(DataListenFd);
842 DataListenFd = -1;
843
30b30ec1
AL
844 return true;
845}
846 /*}}}*/
847// FTPConn::Get - Get a file /*{{{*/
848// ---------------------------------------------------------------------
849/* This opens a data connection, sends REST and RETR and then
850 transfers the file over. */
650faab0 851bool FTPConn::Get(const char *Path,FileFd &To,unsigned long long Resume,
ee279506
MV
852 Hashes &Hash,bool &Missing, unsigned long long MaximumSize,
853 pkgAcqMethod *Owner)
30b30ec1 854{
ce0ae89a 855 Missing = false;
30b30ec1
AL
856 if (CreateDataFd() == false)
857 return false;
858
859 unsigned int Tag;
ce0ae89a 860 string Msg;
30b30ec1
AL
861 if (Resume != 0)
862 {
863 if (WriteMsg(Tag,Msg,"REST %u",Resume) == false)
864 return false;
865 if (Tag >= 400)
866 Resume = 0;
867 }
868
869 if (To.Truncate(Resume) == false)
870 return false;
10861bb5
AL
871
872 if (To.Seek(0) == false)
873 return false;
874
875 if (Resume != 0)
876 {
109eb151 877 if (Hash.AddFD(To,Resume) == false)
10861bb5 878 {
dc738e7a 879 _error->Errno("read",_("Problem hashing file"));
10861bb5
AL
880 return false;
881 }
882 }
30b30ec1
AL
883
884 // Send the get command
885 if (WriteMsg(Tag,Msg,"RETR %s",Path) == false)
886 return false;
887
888 if (Tag >= 400)
ce0ae89a
AL
889 {
890 if (Tag == 550)
891 Missing = true;
dc738e7a 892 return _error->Error(_("Unable to fetch file, server said '%s'"),Msg.c_str());
ce0ae89a 893 }
30b30ec1
AL
894
895 // Finish off the data connection
896 if (Finalize() == false)
897 return false;
898
899 // Copy loop
900 unsigned char Buffer[4096];
901 while (1)
902 {
903 // Wait for some data..
904 if (WaitFd(DataFd,false,TimeOut) == false)
25dbb396
AL
905 {
906 Close();
dc738e7a 907 return _error->Error(_("Data socket timed out"));
25dbb396
AL
908 }
909
30b30ec1
AL
910 // Read the data..
911 int Res = read(DataFd,Buffer,sizeof(Buffer));
912 if (Res == 0)
913 break;
914 if (Res < 0)
915 {
916 if (errno == EAGAIN)
917 continue;
918 break;
919 }
10861bb5 920
63b1700f 921 Hash.Add(Buffer,Res);
30b30ec1 922 if (To.Write(Buffer,Res) == false)
25dbb396
AL
923 {
924 Close();
30b30ec1 925 return false;
5b33fab8
MV
926 }
927
c48eea97 928 if (MaximumSize > 0 && To.Tell() > MaximumSize)
ee279506
MV
929 {
930 Owner->SetFailReason("MaximumSizeExceeded");
5b33fab8 931 return _error->Error("Writing more data than expected (%llu > %llu)",
c48eea97 932 To.Tell(), MaximumSize);
ee279506 933 }
30b30ec1
AL
934 }
935
936 // All done
937 close(DataFd);
938 DataFd = -1;
939
940 // Read the closing message from the server
941 if (ReadResp(Tag,Msg) == false)
942 return false;
943 if (Tag >= 400)
dc738e7a 944 return _error->Error(_("Data transfer failed, server said '%s'"),Msg.c_str());
30b30ec1
AL
945 return true;
946}
947 /*}}}*/
948
ce0ae89a
AL
949// FtpMethod::FtpMethod - Constructor /*{{{*/
950// ---------------------------------------------------------------------
951/* */
952FtpMethod::FtpMethod() : pkgAcqMethod("1.0",SendConfig)
30b30ec1 953{
ce0ae89a
AL
954 signal(SIGTERM,SigTerm);
955 signal(SIGINT,SigTerm);
30b30ec1 956
ce0ae89a
AL
957 Server = 0;
958 FailFd = -1;
959}
960 /*}}}*/
961// FtpMethod::SigTerm - Handle a fatal signal /*{{{*/
962// ---------------------------------------------------------------------
1e3f4083 963/* This closes and timestamps the open file. This is necessary to get
ce0ae89a
AL
964 resume behavoir on user abort */
965void FtpMethod::SigTerm(int)
966{
967 if (FailFd == -1)
ffe9323a 968 _exit(100);
9ce3cfc9 969
ce0ae89a 970 // Timestamp
246bbb61 971 struct timeval times[2];
9ce3cfc9
DK
972 times[0].tv_sec = FailTime;
973 times[1].tv_sec = FailTime;
246bbb61
DK
974 times[0].tv_usec = times[1].tv_usec = 0;
975 utimes(FailFile.c_str(), times);
9ce3cfc9
DK
976
977 close(FailFd);
978
ffe9323a 979 _exit(100);
ce0ae89a
AL
980}
981 /*}}}*/
982// FtpMethod::Configuration - Handle a configuration message /*{{{*/
983// ---------------------------------------------------------------------
984/* We stash the desired pipeline depth */
985bool FtpMethod::Configuration(string Message)
986{
987 if (pkgAcqMethod::Configuration(Message) == false)
988 return false;
989
990 TimeOut = _config->FindI("Acquire::Ftp::Timeout",TimeOut);
9983999d
MV
991
992 // no more active ftp, sorry
993 DropPrivsOrDie();
994
ce0ae89a
AL
995 return true;
996}
997 /*}}}*/
998// FtpMethod::Fetch - Fetch a file /*{{{*/
999// ---------------------------------------------------------------------
10861bb5 1000/* Fetch a single file, called by the base class.. */
ce0ae89a
AL
1001bool FtpMethod::Fetch(FetchItem *Itm)
1002{
1003 URI Get = Itm->Uri;
1004 const char *File = Get.Path.c_str();
1005 FetchResult Res;
1006 Res.Filename = Itm->DestFile;
1007 Res.IMSHit = false;
1de1f703
MV
1008
1009 maybe_add_auth (Get, _config->FindFile("Dir::Etc::netrc"));
1010
ce0ae89a
AL
1011 // Connect to the server
1012 if (Server == 0 || Server->Comp(Get) == false)
30b30ec1 1013 {
ce0ae89a
AL
1014 delete Server;
1015 Server = new FTPConn(Get);
1016 }
1017
1018 // Could not connect is a transient error..
1019 if (Server->Open(this) == false)
1020 {
b3e53cec 1021 Server->Close();
ce0ae89a
AL
1022 Fail(true);
1023 return true;
1024 }
30b30ec1 1025
ce0ae89a 1026 // Get the files information
dc738e7a 1027 Status(_("Query"));
650faab0 1028 unsigned long long Size;
ce0ae89a
AL
1029 if (Server->Size(File,Size) == false ||
1030 Server->ModTime(File,FailTime) == false)
1031 {
1032 Fail(true);
1033 return true;
1034 }
1035 Res.Size = Size;
1036
1037 // See if it is an IMS hit
1038 if (Itm->LastModified == FailTime)
1039 {
1040 Res.Size = 0;
1041 Res.IMSHit = true;
1042 URIDone(Res);
1043 return true;
1044 }
1045
1046 // See if the file exists
1047 struct stat Buf;
1048 if (stat(Itm->DestFile.c_str(),&Buf) == 0)
1049 {
650faab0 1050 if (Size == (unsigned long long)Buf.st_size && FailTime == Buf.st_mtime)
30b30ec1 1051 {
ce0ae89a
AL
1052 Res.Size = Buf.st_size;
1053 Res.LastModified = Buf.st_mtime;
7ef72446 1054 Res.ResumePoint = Buf.st_size;
ce0ae89a
AL
1055 URIDone(Res);
1056 return true;
30b30ec1
AL
1057 }
1058
ce0ae89a 1059 // Resume?
650faab0 1060 if (FailTime == Buf.st_mtime && Size > (unsigned long long)Buf.st_size)
ce0ae89a
AL
1061 Res.ResumePoint = Buf.st_size;
1062 }
1063
1064 // Open the file
63b1700f 1065 Hashes Hash;
ce0ae89a
AL
1066 {
1067 FileFd Fd(Itm->DestFile,FileFd::WriteAny);
1068 if (_error->PendingError() == true)
1069 return false;
1070
1071 URIStart(Res);
1072
1073 FailFile = Itm->DestFile;
1074 FailFile.c_str(); // Make sure we dont do a malloc in the signal handler
1075 FailFd = Fd.Fd();
1076
1077 bool Missing;
ee279506 1078 if (Server->Get(File,Fd,Res.ResumePoint,Hash,Missing,Itm->MaximumSize,this) == false)
30b30ec1 1079 {
0dfc0829 1080 Fd.Close();
9ce3cfc9 1081
0dfc0829 1082 // Timestamp
246bbb61 1083 struct timeval times[2];
9ce3cfc9
DK
1084 times[0].tv_sec = FailTime;
1085 times[1].tv_sec = FailTime;
246bbb61
DK
1086 times[0].tv_usec = times[1].tv_usec = 0;
1087 utimes(FailFile.c_str(), times);
9ce3cfc9 1088
fc5f5417
MV
1089 // If the file is missing we hard fail and delete the destfile
1090 // otherwise transient fail
1091 if (Missing == true) {
1092 unlink(FailFile.c_str());
ce0ae89a 1093 return false;
fc5f5417 1094 }
ce0ae89a
AL
1095 Fail(true);
1096 return true;
30b30ec1 1097 }
ce0ae89a
AL
1098
1099 Res.Size = Fd.Size();
9ce3cfc9
DK
1100
1101 // Timestamp
246bbb61 1102 struct timeval times[2];
9ce3cfc9
DK
1103 times[0].tv_sec = FailTime;
1104 times[1].tv_sec = FailTime;
246bbb61
DK
1105 times[0].tv_usec = times[1].tv_usec = 0;
1106 utimes(Fd.Name().c_str(), times);
9ce3cfc9 1107 FailFd = -1;
30b30ec1 1108 }
9ce3cfc9 1109
ce0ae89a 1110 Res.LastModified = FailTime;
a7c835af 1111 Res.TakeHashes(Hash);
ce0ae89a
AL
1112
1113 URIDone(Res);
9ce3cfc9 1114
ce0ae89a
AL
1115 return true;
1116}
1117 /*}}}*/
1118
65512241 1119int main(int, const char *argv[])
ce0ae89a 1120{
b25423f6
MZ
1121 setlocale(LC_ALL, "");
1122
d4489322
AL
1123 /* See if we should be come the http client - we do this for http
1124 proxy urls */
1125 if (getenv("ftp_proxy") != 0)
1126 {
1127 URI Proxy = string(getenv("ftp_proxy"));
f8081133 1128
f8081133 1129 // Run the HTTP method
d4489322
AL
1130 if (Proxy.Access == "http")
1131 {
1132 // Copy over the environment setting
1133 char S[300];
1134 snprintf(S,sizeof(S),"http_proxy=%s",getenv("ftp_proxy"));
1135 putenv(S);
b8564336 1136 putenv((char *)"no_proxy=");
d4489322
AL
1137
1138 // Run the http method
91bb3e2e 1139 string Path = flNotFile(argv[0]) + "http";
42ab8223 1140 execl(Path.c_str(),Path.c_str(),(char *)NULL);
dc738e7a 1141 cerr << _("Unable to invoke ") << Path << endl;
d4489322
AL
1142 exit(100);
1143 }
1144 }
1145
ce0ae89a 1146 FtpMethod Mth;
7b18d559 1147
ce0ae89a 1148 return Mth.Run();
30b30ec1 1149}