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