]> git.saurik.com Git - apt.git/blob - apt-pkg/acquire-item.cc
* merged with apt--mvo
[apt.git] / apt-pkg / acquire-item.cc
1 // -*- mode: cpp; mode: fold -*-
2 // Description /*{{{*/
3 // $Id: acquire-item.cc,v 1.46.2.9 2004/01/16 18:51:11 mdz Exp $
4 /* ######################################################################
5
6 Acquire Item - Item to acquire
7
8 Each item can download to exactly one file at a time. This means you
9 cannot create an item that fetches two uri's to two files at the same
10 time. The pkgAcqIndex class creates a second class upon instantiation
11 to fetch the other index files because of this.
12
13 ##################################################################### */
14 /*}}}*/
15 // Include Files /*{{{*/
16 #ifdef __GNUG__
17 #pragma implementation "apt-pkg/acquire-item.h"
18 #endif
19 #include <apt-pkg/acquire-item.h>
20 #include <apt-pkg/configuration.h>
21 #include <apt-pkg/sourcelist.h>
22 #include <apt-pkg/vendorlist.h>
23 #include <apt-pkg/error.h>
24 #include <apt-pkg/strutl.h>
25 #include <apt-pkg/fileutl.h>
26 #include <apt-pkg/md5.h>
27
28 #include <apti18n.h>
29
30 #include <sys/stat.h>
31 #include <unistd.h>
32 #include <errno.h>
33 #include <string>
34 #include <stdio.h>
35 /*}}}*/
36
37 using namespace std;
38
39 // Acquire::Item::Item - Constructor /*{{{*/
40 // ---------------------------------------------------------------------
41 /* */
42 pkgAcquire::Item::Item(pkgAcquire *Owner) : Owner(Owner), FileSize(0),
43 PartialSize(0), Mode(0), ID(0), Complete(false),
44 Local(false), QueueCounter(0)
45 {
46 Owner->Add(this);
47 Status = StatIdle;
48 }
49 /*}}}*/
50 // Acquire::Item::~Item - Destructor /*{{{*/
51 // ---------------------------------------------------------------------
52 /* */
53 pkgAcquire::Item::~Item()
54 {
55 Owner->Remove(this);
56 }
57 /*}}}*/
58 // Acquire::Item::Failed - Item failed to download /*{{{*/
59 // ---------------------------------------------------------------------
60 /* We return to an idle state if there are still other queues that could
61 fetch this object */
62 void pkgAcquire::Item::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
63 {
64 Status = StatIdle;
65 ErrorText = LookupTag(Message,"Message");
66 if (QueueCounter <= 1)
67 {
68 /* This indicates that the file is not available right now but might
69 be sometime later. If we do a retry cycle then this should be
70 retried [CDROMs] */
71 if (Cnf->LocalOnly == true &&
72 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
73 {
74 Status = StatIdle;
75 Dequeue();
76 return;
77 }
78
79 Status = StatError;
80 Dequeue();
81 }
82 }
83 /*}}}*/
84 // Acquire::Item::Start - Item has begun to download /*{{{*/
85 // ---------------------------------------------------------------------
86 /* Stash status and the file size. Note that setting Complete means
87 sub-phases of the acquire process such as decompresion are operating */
88 void pkgAcquire::Item::Start(string /*Message*/,unsigned long Size)
89 {
90 Status = StatFetching;
91 if (FileSize == 0 && Complete == false)
92 FileSize = Size;
93 }
94 /*}}}*/
95 // Acquire::Item::Done - Item downloaded OK /*{{{*/
96 // ---------------------------------------------------------------------
97 /* */
98 void pkgAcquire::Item::Done(string Message,unsigned long Size,string,
99 pkgAcquire::MethodConfig *Cnf)
100 {
101 // We just downloaded something..
102 string FileName = LookupTag(Message,"Filename");
103 if (Complete == false && FileName == DestFile)
104 {
105 if (Owner->Log != 0)
106 Owner->Log->Fetched(Size,atoi(LookupTag(Message,"Resume-Point","0").c_str()));
107 }
108
109 if (FileSize == 0)
110 FileSize= Size;
111
112 Status = StatDone;
113 ErrorText = string();
114 Owner->Dequeue(this);
115 }
116 /*}}}*/
117 // Acquire::Item::Rename - Rename a file /*{{{*/
118 // ---------------------------------------------------------------------
119 /* This helper function is used by alot of item methods as thier final
120 step */
121 void pkgAcquire::Item::Rename(string From,string To)
122 {
123 if (rename(From.c_str(),To.c_str()) != 0)
124 {
125 char S[300];
126 snprintf(S,sizeof(S),_("rename failed, %s (%s -> %s)."),strerror(errno),
127 From.c_str(),To.c_str());
128 Status = StatError;
129 ErrorText = S;
130 }
131 }
132 /*}}}*/
133
134 // AcqIndex::AcqIndex - Constructor /*{{{*/
135 // ---------------------------------------------------------------------
136 /* The package file is added to the queue and a second class is
137 instantiated to fetch the revision file */
138 pkgAcqIndex::pkgAcqIndex(pkgAcquire *Owner,
139 string URI,string URIDesc,string ShortDesc,
140 string ExpectedMD5, string comprExt) :
141 Item(Owner), RealURI(URI), ExpectedMD5(ExpectedMD5)
142 {
143 Decompression = false;
144 Erase = false;
145
146 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
147 DestFile += URItoFileName(URI);
148
149 if(comprExt.empty())
150 {
151 // autoselect
152 if(FileExists("/usr/bin/bzip2"))
153 Desc.URI = URI + ".bz2";
154 else
155 Desc.URI = URI + ".gz";
156 } else {
157 Desc.URI = URI + comprExt;
158 }
159
160 Desc.Description = URIDesc;
161 Desc.Owner = this;
162 Desc.ShortDesc = ShortDesc;
163
164 QueueURI(Desc);
165 }
166 /*}}}*/
167 // AcqIndex::Custom600Headers - Insert custom request headers /*{{{*/
168 // ---------------------------------------------------------------------
169 /* The only header we use is the last-modified header. */
170 string pkgAcqIndex::Custom600Headers()
171 {
172 string Final = _config->FindDir("Dir::State::lists");
173 Final += URItoFileName(RealURI);
174
175 struct stat Buf;
176 if (stat(Final.c_str(),&Buf) != 0)
177 return "\nIndex-File: true";
178
179 return "\nIndex-File: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
180 }
181 /*}}}*/
182
183 void pkgAcqIndex::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
184 {
185 // no .bz2 found, retry with .gz
186 if(Desc.URI.substr(Desc.URI.size()-3,Desc.URI.size()-1) == "bz2") {
187 Desc.URI = Desc.URI.substr(0,Desc.URI.size()-3) + "gz";
188
189 // retry with a gzip one
190 new pkgAcqIndex(Owner, RealURI, Desc.Description,Desc.ShortDesc,
191 ExpectedMD5, string(".gz"));
192 Status = StatDone;
193 Complete = false;
194 Dequeue();
195 return;
196 }
197
198
199 Item::Failed(Message,Cnf);
200 }
201
202
203 // AcqIndex::Done - Finished a fetch /*{{{*/
204 // ---------------------------------------------------------------------
205 /* This goes through a number of states.. On the initial fetch the
206 method could possibly return an alternate filename which points
207 to the uncompressed version of the file. If this is so the file
208 is copied into the partial directory. In all other cases the file
209 is decompressed with a gzip uri. */
210 void pkgAcqIndex::Done(string Message,unsigned long Size,string MD5,
211 pkgAcquire::MethodConfig *Cfg)
212 {
213 Item::Done(Message,Size,MD5,Cfg);
214
215 if (Decompression == true)
216 {
217 if (_config->FindB("Debug::pkgAcquire::Auth", false))
218 {
219 std::cerr << std::endl << RealURI << ": Computed MD5: " << MD5;
220 std::cerr << " Expected MD5: " << ExpectedMD5 << std::endl;
221 }
222
223 if (MD5.empty())
224 {
225 MD5Summation sum;
226 FileFd Fd(DestFile, FileFd::ReadOnly);
227 sum.AddFD(Fd.Fd(), Fd.Size());
228 Fd.Close();
229 MD5 = (string)sum.Result();
230 }
231
232 if (!ExpectedMD5.empty() && MD5 != ExpectedMD5)
233 {
234 Status = StatAuthError;
235 ErrorText = _("MD5Sum mismatch");
236 Rename(DestFile,DestFile + ".FAILED");
237 return;
238 }
239 // Done, move it into position
240 string FinalFile = _config->FindDir("Dir::State::lists");
241 FinalFile += URItoFileName(RealURI);
242 Rename(DestFile,FinalFile);
243 chmod(FinalFile.c_str(),0644);
244
245 /* We restore the original name to DestFile so that the clean operation
246 will work OK */
247 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
248 DestFile += URItoFileName(RealURI);
249
250 // Remove the compressed version.
251 if (Erase == true)
252 unlink(DestFile.c_str());
253 return;
254 }
255
256 Erase = false;
257 Complete = true;
258
259 // Handle the unzipd case
260 string FileName = LookupTag(Message,"Alt-Filename");
261 if (FileName.empty() == false)
262 {
263 // The files timestamp matches
264 if (StringToBool(LookupTag(Message,"Alt-IMS-Hit"),false) == true)
265 return;
266
267 Decompression = true;
268 Local = true;
269 DestFile += ".decomp";
270 Desc.URI = "copy:" + FileName;
271 QueueURI(Desc);
272 Mode = "copy";
273 return;
274 }
275
276 FileName = LookupTag(Message,"Filename");
277 if (FileName.empty() == true)
278 {
279 Status = StatError;
280 ErrorText = "Method gave a blank filename";
281 }
282
283 // The files timestamp matches
284 if (StringToBool(LookupTag(Message,"IMS-Hit"),false) == true)
285 return;
286
287 if (FileName == DestFile)
288 Erase = true;
289 else
290 Local = true;
291
292 string compExt = Desc.URI.substr(Desc.URI.size()-3,Desc.URI.size()-1);
293 char *decompProg;
294 if(compExt == "bz2")
295 decompProg = "bzip2";
296 else if(compExt == ".gz")
297 decompProg = "gzip";
298 else {
299 _error->Error("Unsupported extension: %s", compExt.c_str());
300 return;
301 }
302
303 Decompression = true;
304 DestFile += ".decomp";
305 Desc.URI = string(decompProg) + ":" + FileName;
306 QueueURI(Desc);
307 Mode = decompProg;
308 }
309
310 pkgAcqMetaSig::pkgAcqMetaSig(pkgAcquire *Owner,
311 string URI,string URIDesc,string ShortDesc,
312 string MetaIndexURI, string MetaIndexURIDesc,
313 string MetaIndexShortDesc,
314 const vector<IndexTarget*>* IndexTargets,
315 indexRecords* MetaIndexParser) :
316 Item(Owner), RealURI(URI), MetaIndexURI(MetaIndexURI),
317 MetaIndexURIDesc(MetaIndexURIDesc), MetaIndexShortDesc(MetaIndexShortDesc)
318 {
319 this->MetaIndexParser = MetaIndexParser;
320 this->IndexTargets = IndexTargets;
321 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
322 DestFile += URItoFileName(URI);
323
324 // remove any partial downloaded sig-file. it may confuse proxies
325 // and is too small to warrant a partial download anyway
326 unlink(DestFile.c_str());
327
328 // Create the item
329 Desc.Description = URIDesc;
330 Desc.Owner = this;
331 Desc.ShortDesc = ShortDesc;
332 Desc.URI = URI;
333
334
335 string Final = _config->FindDir("Dir::State::lists");
336 Final += URItoFileName(RealURI);
337 struct stat Buf;
338 if (stat(Final.c_str(),&Buf) == 0)
339 {
340 // File was already in place. It needs to be re-verified
341 // because Release might have changed, so Move it into partial
342 Rename(Final,DestFile);
343 // unlink the file and do not try to use I-M-S and Last-Modified
344 // if the users proxy is broken
345 if(_config->FindB("Acquire::BrokenProxy", false) == true) {
346 std::cerr << "forcing re-get of the signature file as requested" << std::endl;
347 unlink(DestFile.c_str());
348 }
349 }
350
351 QueueURI(Desc);
352 }
353 /*}}}*/
354 // pkgAcqMetaSig::Custom600Headers - Insert custom request headers /*{{{*/
355 // ---------------------------------------------------------------------
356 /* The only header we use is the last-modified header. */
357 string pkgAcqMetaSig::Custom600Headers()
358 {
359 struct stat Buf;
360 if (stat(DestFile.c_str(),&Buf) != 0)
361 return "\nIndex-File: true";
362
363 return "\nIndex-File: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
364 }
365
366 void pkgAcqMetaSig::Done(string Message,unsigned long Size,string MD5,
367 pkgAcquire::MethodConfig *Cfg)
368 {
369 Item::Done(Message,Size,MD5,Cfg);
370
371 string FileName = LookupTag(Message,"Filename");
372 if (FileName.empty() == true)
373 {
374 Status = StatError;
375 ErrorText = "Method gave a blank filename";
376 return;
377 }
378
379 if (FileName != DestFile)
380 {
381 // We have to copy it into place
382 Local = true;
383 Desc.URI = "copy:" + FileName;
384 QueueURI(Desc);
385 return;
386 }
387
388 Complete = true;
389
390 // queue a pkgAcqMetaIndex to be verified against the sig we just retrieved
391 new pkgAcqMetaIndex(Owner, MetaIndexURI, MetaIndexURIDesc, MetaIndexShortDesc,
392 DestFile, IndexTargets, MetaIndexParser);
393
394 }
395 /*}}}*/
396 void pkgAcqMetaSig::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
397 {
398 // Delete any existing sigfile, so that this source isn't
399 // mistakenly trusted
400 string Final = _config->FindDir("Dir::State::lists") + URItoFileName(RealURI);
401 unlink(Final.c_str());
402
403 // if we get a timeout if fail
404 if(LookupTag(Message,"FailReason") == "Timeout" ||
405 LookupTag(Message,"FailReason") == "TmpResolveFailure") {
406 Item::Failed(Message,Cnf);
407 return;
408 }
409
410 // queue a pkgAcqMetaIndex with no sigfile
411 new pkgAcqMetaIndex(Owner, MetaIndexURI, MetaIndexURIDesc, MetaIndexShortDesc,
412 "", IndexTargets, MetaIndexParser);
413
414 if (Cnf->LocalOnly == true ||
415 StringToBool(LookupTag(Message,"Transient-Failure"),false) == false)
416 {
417 // Ignore this
418 Status = StatDone;
419 Complete = false;
420 Dequeue();
421 return;
422 }
423
424 Item::Failed(Message,Cnf);
425 }
426
427 pkgAcqMetaIndex::pkgAcqMetaIndex(pkgAcquire *Owner,
428 string URI,string URIDesc,string ShortDesc,
429 string SigFile,
430 const vector<struct IndexTarget*>* IndexTargets,
431 indexRecords* MetaIndexParser) :
432 Item(Owner), RealURI(URI), SigFile(SigFile)
433 {
434 this->AuthPass = false;
435 this->MetaIndexParser = MetaIndexParser;
436 this->IndexTargets = IndexTargets;
437 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
438 DestFile += URItoFileName(URI);
439
440 // Create the item
441 Desc.Description = URIDesc;
442 Desc.Owner = this;
443 Desc.ShortDesc = ShortDesc;
444 Desc.URI = URI;
445
446 QueueURI(Desc);
447 }
448
449 /*}}}*/
450 // pkgAcqMetaIndex::Custom600Headers - Insert custom request headers /*{{{*/
451 // ---------------------------------------------------------------------
452 /* The only header we use is the last-modified header. */
453 string pkgAcqMetaIndex::Custom600Headers()
454 {
455 string Final = _config->FindDir("Dir::State::lists");
456 Final += URItoFileName(RealURI);
457
458 struct stat Buf;
459 if (stat(Final.c_str(),&Buf) != 0)
460 return "\nIndex-File: true";
461
462 return "\nIndex-File: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
463 }
464
465 void pkgAcqMetaIndex::Done(string Message,unsigned long Size,string MD5,
466 pkgAcquire::MethodConfig *Cfg)
467 {
468 Item::Done(Message,Size,MD5,Cfg);
469
470 // MetaIndexes are done in two passes: one to download the
471 // metaindex with an appropriate method, and a second to verify it
472 // with the gpgv method
473
474 if (AuthPass == true)
475 {
476 AuthDone(Message);
477 }
478 else
479 {
480 RetrievalDone(Message);
481 if (!Complete)
482 // Still more retrieving to do
483 return;
484
485 if (SigFile == "")
486 {
487 // There was no signature file, so we are finished. Download
488 // the indexes without verification.
489 QueueIndexes(false);
490 }
491 else
492 {
493 // There was a signature file, so pass it to gpgv for
494 // verification
495
496 if (_config->FindB("Debug::pkgAcquire::Auth", false))
497 std::cerr << "Metaindex acquired, queueing gpg verification ("
498 << SigFile << "," << DestFile << ")\n";
499 AuthPass = true;
500 Desc.URI = "gpgv:" + SigFile;
501 QueueURI(Desc);
502 Mode = "gpgv";
503 }
504 }
505 }
506
507 void pkgAcqMetaIndex::RetrievalDone(string Message)
508 {
509 // We have just finished downloading a Release file (it is not
510 // verified yet)
511
512 string FileName = LookupTag(Message,"Filename");
513 if (FileName.empty() == true)
514 {
515 Status = StatError;
516 ErrorText = "Method gave a blank filename";
517 return;
518 }
519
520 if (FileName != DestFile)
521 {
522 Local = true;
523 Desc.URI = "copy:" + FileName;
524 QueueURI(Desc);
525 return;
526 }
527
528 Complete = true;
529
530 string FinalFile = _config->FindDir("Dir::State::lists");
531 FinalFile += URItoFileName(RealURI);
532
533 // The files timestamp matches
534 if (StringToBool(LookupTag(Message,"IMS-Hit"),false) == false)
535 {
536 // Move it into position
537 Rename(DestFile,FinalFile);
538 }
539 DestFile = FinalFile;
540 }
541
542 void pkgAcqMetaIndex::AuthDone(string Message)
543 {
544 // At this point, the gpgv method has succeeded, so there is a
545 // valid signature from a key in the trusted keyring. We
546 // perform additional verification of its contents, and use them
547 // to verify the indexes we are about to download
548
549 if (!MetaIndexParser->Load(DestFile))
550 {
551 Status = StatAuthError;
552 ErrorText = MetaIndexParser->ErrorText;
553 return;
554 }
555
556 if (!VerifyVendor())
557 {
558 return;
559 }
560
561 if (_config->FindB("Debug::pkgAcquire::Auth", false))
562 std::cerr << "Signature verification succeeded: "
563 << DestFile << std::endl;
564
565 // Download further indexes with verification
566 QueueIndexes(true);
567
568 // Done, move signature file into position
569
570 string VerifiedSigFile = _config->FindDir("Dir::State::lists") +
571 URItoFileName(RealURI) + ".gpg";
572 Rename(SigFile,VerifiedSigFile);
573 chmod(VerifiedSigFile.c_str(),0644);
574 }
575
576 void pkgAcqMetaIndex::QueueIndexes(bool verify)
577 {
578 for (vector <struct IndexTarget*>::const_iterator Target = IndexTargets->begin();
579 Target != IndexTargets->end();
580 Target++)
581 {
582 string ExpectedIndexMD5;
583 if (verify)
584 {
585 const indexRecords::checkSum *Record = MetaIndexParser->Lookup((*Target)->MetaKey);
586 if (!Record)
587 {
588 Status = StatAuthError;
589 ErrorText = "Unable to find expected entry "
590 + (*Target)->MetaKey + " in Meta-index file (malformed Release file?)";
591 return;
592 }
593 ExpectedIndexMD5 = Record->MD5Hash;
594 if (_config->FindB("Debug::pkgAcquire::Auth", false))
595 {
596 std::cerr << "Queueing: " << (*Target)->URI << std::endl;
597 std::cerr << "Expected MD5: " << ExpectedIndexMD5 << std::endl;
598 }
599 if (ExpectedIndexMD5.empty())
600 {
601 Status = StatAuthError;
602 ErrorText = "Unable to find MD5 sum for "
603 + (*Target)->MetaKey + " in Meta-index file";
604 return;
605 }
606 }
607
608 // Queue Packages file
609 new pkgAcqIndex(Owner, (*Target)->URI, (*Target)->Description,
610 (*Target)->ShortDesc, ExpectedIndexMD5);
611 }
612 }
613
614 bool pkgAcqMetaIndex::VerifyVendor()
615 {
616 // // Maybe this should be made available from above so we don't have
617 // // to read and parse it every time?
618 // pkgVendorList List;
619 // List.ReadMainList();
620
621 // const Vendor* Vndr = NULL;
622 // for (std::vector<string>::const_iterator I = GPGVOutput.begin(); I != GPGVOutput.end(); I++)
623 // {
624 // string::size_type pos = (*I).find("VALIDSIG ");
625 // if (_config->FindB("Debug::Vendor", false))
626 // std::cerr << "Looking for VALIDSIG in \"" << (*I) << "\": pos " << pos
627 // << std::endl;
628 // if (pos != std::string::npos)
629 // {
630 // string Fingerprint = (*I).substr(pos+sizeof("VALIDSIG"));
631 // if (_config->FindB("Debug::Vendor", false))
632 // std::cerr << "Looking for \"" << Fingerprint << "\" in vendor..." <<
633 // std::endl;
634 // Vndr = List.FindVendor(Fingerprint) != "";
635 // if (Vndr != NULL);
636 // break;
637 // }
638 // }
639
640 string Transformed = MetaIndexParser->GetExpectedDist();
641
642 if (Transformed == "../project/experimental")
643 {
644 Transformed = "experimental";
645 }
646
647 string::size_type pos = Transformed.rfind('/');
648 if (pos != string::npos)
649 {
650 Transformed = Transformed.substr(0, pos);
651 }
652
653 if (Transformed == ".")
654 {
655 Transformed = "";
656 }
657
658 if (_config->FindB("Debug::pkgAcquire::Auth", false))
659 {
660 std::cerr << "Got Codename: " << MetaIndexParser->GetDist() << std::endl;
661 std::cerr << "Expecting Dist: " << MetaIndexParser->GetExpectedDist() << std::endl;
662 std::cerr << "Transformed Dist: " << Transformed << std::endl;
663 }
664
665 if (MetaIndexParser->CheckDist(Transformed) == false)
666 {
667 // This might become fatal one day
668 // Status = StatAuthError;
669 // ErrorText = "Conflicting distribution; expected "
670 // + MetaIndexParser->GetExpectedDist() + " but got "
671 // + MetaIndexParser->GetDist();
672 // return false;
673 if (!Transformed.empty())
674 {
675 _error->Warning("Conflicting distribution: %s (expected %s but got %s)",
676 Desc.Description.c_str(),
677 Transformed.c_str(),
678 MetaIndexParser->GetDist().c_str());
679 }
680 }
681
682 return true;
683 }
684 /*}}}*/
685 // pkgAcqMetaIndex::Failed - no Release file present or no signature
686 // file present /*{{{*/
687 // ---------------------------------------------------------------------
688 /* */
689 void pkgAcqMetaIndex::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
690 {
691 if (AuthPass == true)
692 {
693 // gpgv method failed
694 _error->Warning("GPG error: %s: %s",
695 Desc.Description.c_str(),
696 LookupTag(Message,"Message").c_str());
697 }
698
699 // No Release file was present, or verification failed, so fall
700 // back to queueing Packages files without verification
701 QueueIndexes(false);
702 }
703
704 /*}}}*/
705
706 // AcqArchive::AcqArchive - Constructor /*{{{*/
707 // ---------------------------------------------------------------------
708 /* This just sets up the initial fetch environment and queues the first
709 possibilitiy */
710 pkgAcqArchive::pkgAcqArchive(pkgAcquire *Owner,pkgSourceList *Sources,
711 pkgRecords *Recs,pkgCache::VerIterator const &Version,
712 string &StoreFilename) :
713 Item(Owner), Version(Version), Sources(Sources), Recs(Recs),
714 StoreFilename(StoreFilename), Vf(Version.FileList()),
715 Trusted(false)
716 {
717 Retries = _config->FindI("Acquire::Retries",0);
718
719 if (Version.Arch() == 0)
720 {
721 _error->Error(_("I wasn't able to locate a file for the %s package. "
722 "This might mean you need to manually fix this package. "
723 "(due to missing arch)"),
724 Version.ParentPkg().Name());
725 return;
726 }
727
728 /* We need to find a filename to determine the extension. We make the
729 assumption here that all the available sources for this version share
730 the same extension.. */
731 // Skip not source sources, they do not have file fields.
732 for (; Vf.end() == false; Vf++)
733 {
734 if ((Vf.File()->Flags & pkgCache::Flag::NotSource) != 0)
735 continue;
736 break;
737 }
738
739 // Does not really matter here.. we are going to fail out below
740 if (Vf.end() != true)
741 {
742 // If this fails to get a file name we will bomb out below.
743 pkgRecords::Parser &Parse = Recs->Lookup(Vf);
744 if (_error->PendingError() == true)
745 return;
746
747 // Generate the final file name as: package_version_arch.foo
748 StoreFilename = QuoteString(Version.ParentPkg().Name(),"_:") + '_' +
749 QuoteString(Version.VerStr(),"_:") + '_' +
750 QuoteString(Version.Arch(),"_:.") +
751 "." + flExtension(Parse.FileName());
752 }
753
754 // check if we have one trusted source for the package. if so, switch
755 // to "TrustedOnly" mode
756 for (pkgCache::VerFileIterator i = Version.FileList(); i.end() == false; i++)
757 {
758 pkgIndexFile *Index;
759 if (Sources->FindIndex(i.File(),Index) == false)
760 continue;
761 if (_config->FindB("Debug::pkgAcquire::Auth", false))
762 {
763 std::cerr << "Checking index: " << Index->Describe()
764 << "(Trusted=" << Index->IsTrusted() << ")\n";
765 }
766 if (Index->IsTrusted()) {
767 Trusted = true;
768 break;
769 }
770 }
771
772 // Select a source
773 if (QueueNext() == false && _error->PendingError() == false)
774 _error->Error(_("I wasn't able to locate file for the %s package. "
775 "This might mean you need to manually fix this package."),
776 Version.ParentPkg().Name());
777 }
778 /*}}}*/
779 // AcqArchive::QueueNext - Queue the next file source /*{{{*/
780 // ---------------------------------------------------------------------
781 /* This queues the next available file version for download. It checks if
782 the archive is already available in the cache and stashs the MD5 for
783 checking later. */
784 bool pkgAcqArchive::QueueNext()
785 {
786 for (; Vf.end() == false; Vf++)
787 {
788 // Ignore not source sources
789 if ((Vf.File()->Flags & pkgCache::Flag::NotSource) != 0)
790 continue;
791
792 // Try to cross match against the source list
793 pkgIndexFile *Index;
794 if (Sources->FindIndex(Vf.File(),Index) == false)
795 continue;
796
797 // only try to get a trusted package from another source if that source
798 // is also trusted
799 if(Trusted && !Index->IsTrusted())
800 continue;
801
802 // Grab the text package record
803 pkgRecords::Parser &Parse = Recs->Lookup(Vf);
804 if (_error->PendingError() == true)
805 return false;
806
807 string PkgFile = Parse.FileName();
808 MD5 = Parse.MD5Hash();
809 if (PkgFile.empty() == true)
810 return _error->Error(_("The package index files are corrupted. No Filename: "
811 "field for package %s."),
812 Version.ParentPkg().Name());
813
814 Desc.URI = Index->ArchiveURI(PkgFile);
815 Desc.Description = Index->ArchiveInfo(Version);
816 Desc.Owner = this;
817 Desc.ShortDesc = Version.ParentPkg().Name();
818
819 // See if we already have the file. (Legacy filenames)
820 FileSize = Version->Size;
821 string FinalFile = _config->FindDir("Dir::Cache::Archives") + flNotDir(PkgFile);
822 struct stat Buf;
823 if (stat(FinalFile.c_str(),&Buf) == 0)
824 {
825 // Make sure the size matches
826 if ((unsigned)Buf.st_size == Version->Size)
827 {
828 Complete = true;
829 Local = true;
830 Status = StatDone;
831 StoreFilename = DestFile = FinalFile;
832 return true;
833 }
834
835 /* Hmm, we have a file and its size does not match, this means it is
836 an old style mismatched arch */
837 unlink(FinalFile.c_str());
838 }
839
840 // Check it again using the new style output filenames
841 FinalFile = _config->FindDir("Dir::Cache::Archives") + flNotDir(StoreFilename);
842 if (stat(FinalFile.c_str(),&Buf) == 0)
843 {
844 // Make sure the size matches
845 if ((unsigned)Buf.st_size == Version->Size)
846 {
847 Complete = true;
848 Local = true;
849 Status = StatDone;
850 StoreFilename = DestFile = FinalFile;
851 return true;
852 }
853
854 /* Hmm, we have a file and its size does not match, this shouldnt
855 happen.. */
856 unlink(FinalFile.c_str());
857 }
858
859 DestFile = _config->FindDir("Dir::Cache::Archives") + "partial/" + flNotDir(StoreFilename);
860
861 // Check the destination file
862 if (stat(DestFile.c_str(),&Buf) == 0)
863 {
864 // Hmm, the partial file is too big, erase it
865 if ((unsigned)Buf.st_size > Version->Size)
866 unlink(DestFile.c_str());
867 else
868 PartialSize = Buf.st_size;
869 }
870
871 // Create the item
872 Local = false;
873 Desc.URI = Index->ArchiveURI(PkgFile);
874 Desc.Description = Index->ArchiveInfo(Version);
875 Desc.Owner = this;
876 Desc.ShortDesc = Version.ParentPkg().Name();
877 QueueURI(Desc);
878
879 Vf++;
880 return true;
881 }
882 return false;
883 }
884 /*}}}*/
885 // AcqArchive::Done - Finished fetching /*{{{*/
886 // ---------------------------------------------------------------------
887 /* */
888 void pkgAcqArchive::Done(string Message,unsigned long Size,string Md5Hash,
889 pkgAcquire::MethodConfig *Cfg)
890 {
891 Item::Done(Message,Size,Md5Hash,Cfg);
892
893 // Check the size
894 if (Size != Version->Size)
895 {
896 Status = StatError;
897 ErrorText = _("Size mismatch");
898 return;
899 }
900
901 // Check the md5
902 if (Md5Hash.empty() == false && MD5.empty() == false)
903 {
904 if (Md5Hash != MD5)
905 {
906 Status = StatError;
907 ErrorText = _("MD5Sum mismatch");
908 if(FileExists(DestFile))
909 Rename(DestFile,DestFile + ".FAILED");
910 return;
911 }
912 }
913
914 // Grab the output filename
915 string FileName = LookupTag(Message,"Filename");
916 if (FileName.empty() == true)
917 {
918 Status = StatError;
919 ErrorText = "Method gave a blank filename";
920 return;
921 }
922
923 Complete = true;
924
925 // Reference filename
926 if (FileName != DestFile)
927 {
928 StoreFilename = DestFile = FileName;
929 Local = true;
930 return;
931 }
932
933 // Done, move it into position
934 string FinalFile = _config->FindDir("Dir::Cache::Archives");
935 FinalFile += flNotDir(StoreFilename);
936 Rename(DestFile,FinalFile);
937
938 StoreFilename = DestFile = FinalFile;
939 Complete = true;
940 }
941 /*}}}*/
942 // AcqArchive::Failed - Failure handler /*{{{*/
943 // ---------------------------------------------------------------------
944 /* Here we try other sources */
945 void pkgAcqArchive::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
946 {
947 ErrorText = LookupTag(Message,"Message");
948
949 /* We don't really want to retry on failed media swaps, this prevents
950 that. An interesting observation is that permanent failures are not
951 recorded. */
952 if (Cnf->Removable == true &&
953 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
954 {
955 // Vf = Version.FileList();
956 while (Vf.end() == false) Vf++;
957 StoreFilename = string();
958 Item::Failed(Message,Cnf);
959 return;
960 }
961
962 if (QueueNext() == false)
963 {
964 // This is the retry counter
965 if (Retries != 0 &&
966 Cnf->LocalOnly == false &&
967 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
968 {
969 Retries--;
970 Vf = Version.FileList();
971 if (QueueNext() == true)
972 return;
973 }
974
975 StoreFilename = string();
976 Item::Failed(Message,Cnf);
977 }
978 }
979 /*}}}*/
980 // AcqArchive::IsTrusted - Determine whether this archive comes from a
981 // trusted source /*{{{*/
982 // ---------------------------------------------------------------------
983 bool pkgAcqArchive::IsTrusted()
984 {
985 return Trusted;
986 }
987
988 // AcqArchive::Finished - Fetching has finished, tidy up /*{{{*/
989 // ---------------------------------------------------------------------
990 /* */
991 void pkgAcqArchive::Finished()
992 {
993 if (Status == pkgAcquire::Item::StatDone &&
994 Complete == true)
995 return;
996 StoreFilename = string();
997 }
998 /*}}}*/
999
1000 // AcqFile::pkgAcqFile - Constructor /*{{{*/
1001 // ---------------------------------------------------------------------
1002 /* The file is added to the queue */
1003 pkgAcqFile::pkgAcqFile(pkgAcquire *Owner,string URI,string MD5,
1004 unsigned long Size,string Dsc,string ShortDesc) :
1005 Item(Owner), Md5Hash(MD5)
1006 {
1007 Retries = _config->FindI("Acquire::Retries",0);
1008
1009 DestFile = flNotDir(URI);
1010
1011 // Create the item
1012 Desc.URI = URI;
1013 Desc.Description = Dsc;
1014 Desc.Owner = this;
1015
1016 // Set the short description to the archive component
1017 Desc.ShortDesc = ShortDesc;
1018
1019 // Get the transfer sizes
1020 FileSize = Size;
1021 struct stat Buf;
1022 if (stat(DestFile.c_str(),&Buf) == 0)
1023 {
1024 // Hmm, the partial file is too big, erase it
1025 if ((unsigned)Buf.st_size > Size)
1026 unlink(DestFile.c_str());
1027 else
1028 PartialSize = Buf.st_size;
1029 }
1030
1031 QueueURI(Desc);
1032 }
1033 /*}}}*/
1034 // AcqFile::Done - Item downloaded OK /*{{{*/
1035 // ---------------------------------------------------------------------
1036 /* */
1037 void pkgAcqFile::Done(string Message,unsigned long Size,string MD5,
1038 pkgAcquire::MethodConfig *Cnf)
1039 {
1040 // Check the md5
1041 if (Md5Hash.empty() == false && MD5.empty() == false)
1042 {
1043 if (Md5Hash != MD5)
1044 {
1045 Status = StatError;
1046 ErrorText = "MD5Sum mismatch";
1047 Rename(DestFile,DestFile + ".FAILED");
1048 return;
1049 }
1050 }
1051
1052 Item::Done(Message,Size,MD5,Cnf);
1053
1054 string FileName = LookupTag(Message,"Filename");
1055 if (FileName.empty() == true)
1056 {
1057 Status = StatError;
1058 ErrorText = "Method gave a blank filename";
1059 return;
1060 }
1061
1062 Complete = true;
1063
1064 // The files timestamp matches
1065 if (StringToBool(LookupTag(Message,"IMS-Hit"),false) == true)
1066 return;
1067
1068 // We have to copy it into place
1069 if (FileName != DestFile)
1070 {
1071 Local = true;
1072 if (_config->FindB("Acquire::Source-Symlinks",true) == false ||
1073 Cnf->Removable == true)
1074 {
1075 Desc.URI = "copy:" + FileName;
1076 QueueURI(Desc);
1077 return;
1078 }
1079
1080 // Erase the file if it is a symlink so we can overwrite it
1081 struct stat St;
1082 if (lstat(DestFile.c_str(),&St) == 0)
1083 {
1084 if (S_ISLNK(St.st_mode) != 0)
1085 unlink(DestFile.c_str());
1086 }
1087
1088 // Symlink the file
1089 if (symlink(FileName.c_str(),DestFile.c_str()) != 0)
1090 {
1091 ErrorText = "Link to " + DestFile + " failure ";
1092 Status = StatError;
1093 Complete = false;
1094 }
1095 }
1096 }
1097 /*}}}*/
1098 // AcqFile::Failed - Failure handler /*{{{*/
1099 // ---------------------------------------------------------------------
1100 /* Here we try other sources */
1101 void pkgAcqFile::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
1102 {
1103 ErrorText = LookupTag(Message,"Message");
1104
1105 // This is the retry counter
1106 if (Retries != 0 &&
1107 Cnf->LocalOnly == false &&
1108 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
1109 {
1110 Retries--;
1111 QueueURI(Desc);
1112 return;
1113 }
1114
1115 Item::Failed(Message,Cnf);
1116 }
1117 /*}}}*/