]> git.saurik.com Git - apt.git/blob - apt-pkg/acquire-item.cc
dee00c63e7b5e4f554aaa0a6910bbc71de1c7608
[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 the compression method
152 if(FileExists("/usr/bin/bzip2"))
153 CompressionExtension = ".bz2";
154 else
155 CompressionExtension = ".gz";
156 } else {
157 CompressionExtension = comprExt;
158 }
159 Desc.URI = URI + CompressionExtension;
160
161 Desc.Description = URIDesc;
162 Desc.Owner = this;
163 Desc.ShortDesc = ShortDesc;
164
165 QueueURI(Desc);
166 }
167 /*}}}*/
168 // AcqIndex::Custom600Headers - Insert custom request headers /*{{{*/
169 // ---------------------------------------------------------------------
170 /* The only header we use is the last-modified header. */
171 string pkgAcqIndex::Custom600Headers()
172 {
173 string Final = _config->FindDir("Dir::State::lists");
174 Final += URItoFileName(RealURI);
175
176 struct stat Buf;
177 if (stat(Final.c_str(),&Buf) != 0)
178 return "\nIndex-File: true";
179
180 return "\nIndex-File: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
181 }
182 /*}}}*/
183
184 void pkgAcqIndex::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
185 {
186 // no .bz2 found, retry with .gz
187 if(Desc.URI.substr(Desc.URI.size()-3) == "bz2") {
188 Desc.URI = Desc.URI.substr(0,Desc.URI.size()-3) + "gz";
189
190 // retry with a gzip one
191 new pkgAcqIndex(Owner, RealURI, Desc.Description,Desc.ShortDesc,
192 ExpectedMD5, string(".gz"));
193 Status = StatDone;
194 Complete = false;
195 Dequeue();
196 return;
197 }
198
199
200 Item::Failed(Message,Cnf);
201 }
202
203
204 // AcqIndex::Done - Finished a fetch /*{{{*/
205 // ---------------------------------------------------------------------
206 /* This goes through a number of states.. On the initial fetch the
207 method could possibly return an alternate filename which points
208 to the uncompressed version of the file. If this is so the file
209 is copied into the partial directory. In all other cases the file
210 is decompressed with a gzip uri. */
211 void pkgAcqIndex::Done(string Message,unsigned long Size,string MD5,
212 pkgAcquire::MethodConfig *Cfg)
213 {
214 Item::Done(Message,Size,MD5,Cfg);
215
216 if (Decompression == true)
217 {
218 if (_config->FindB("Debug::pkgAcquire::Auth", false))
219 {
220 std::cerr << std::endl << RealURI << ": Computed MD5: " << MD5;
221 std::cerr << " Expected MD5: " << ExpectedMD5 << std::endl;
222 }
223
224 if (MD5.empty())
225 {
226 MD5Summation sum;
227 FileFd Fd(DestFile, FileFd::ReadOnly);
228 sum.AddFD(Fd.Fd(), Fd.Size());
229 Fd.Close();
230 MD5 = (string)sum.Result();
231 }
232
233 if (!ExpectedMD5.empty() && MD5 != ExpectedMD5)
234 {
235 Status = StatAuthError;
236 ErrorText = _("MD5Sum mismatch");
237 Rename(DestFile,DestFile + ".FAILED");
238 return;
239 }
240 // Done, move it into position
241 string FinalFile = _config->FindDir("Dir::State::lists");
242 FinalFile += URItoFileName(RealURI);
243 Rename(DestFile,FinalFile);
244 chmod(FinalFile.c_str(),0644);
245
246 /* We restore the original name to DestFile so that the clean operation
247 will work OK */
248 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
249 DestFile += URItoFileName(RealURI);
250
251 // Remove the compressed version.
252 if (Erase == true)
253 unlink(DestFile.c_str());
254 return;
255 }
256
257 Erase = false;
258 Complete = true;
259
260 // Handle the unzipd case
261 string FileName = LookupTag(Message,"Alt-Filename");
262 if (FileName.empty() == false)
263 {
264 // The files timestamp matches
265 if (StringToBool(LookupTag(Message,"Alt-IMS-Hit"),false) == true)
266 return;
267
268 Decompression = true;
269 Local = true;
270 DestFile += ".decomp";
271 Desc.URI = "copy:" + FileName;
272 QueueURI(Desc);
273 Mode = "copy";
274 return;
275 }
276
277 FileName = LookupTag(Message,"Filename");
278 if (FileName.empty() == true)
279 {
280 Status = StatError;
281 ErrorText = "Method gave a blank filename";
282 }
283
284 // The files timestamp matches
285 if (StringToBool(LookupTag(Message,"IMS-Hit"),false) == true)
286 return;
287
288 if (FileName == DestFile)
289 Erase = true;
290 else
291 Local = true;
292
293 string compExt = Desc.URI.substr(Desc.URI.size()-3);
294 char *decompProg;
295 if(compExt == "bz2")
296 decompProg = "bzip2";
297 else if(compExt == ".gz")
298 decompProg = "gzip";
299 else {
300 _error->Error("Unsupported extension: %s", compExt.c_str());
301 return;
302 }
303
304 Decompression = true;
305 DestFile += ".decomp";
306 Desc.URI = string(decompProg) + ":" + FileName;
307 QueueURI(Desc);
308 Mode = decompProg;
309 }
310
311 pkgAcqMetaSig::pkgAcqMetaSig(pkgAcquire *Owner,
312 string URI,string URIDesc,string ShortDesc,
313 string MetaIndexURI, string MetaIndexURIDesc,
314 string MetaIndexShortDesc,
315 const vector<IndexTarget*>* IndexTargets,
316 indexRecords* MetaIndexParser) :
317 Item(Owner), RealURI(URI), MetaIndexURI(MetaIndexURI),
318 MetaIndexURIDesc(MetaIndexURIDesc), MetaIndexShortDesc(MetaIndexShortDesc),
319 MetaIndexParser(MetaIndexParser), IndexTargets(IndexTargets)
320 {
321 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
322 DestFile += URItoFileName(URI);
323
324 // remove any partial downloaded sig-file in partial/.
325 // it may confuse proxies and is too small to warrant a
326 // partial download anyway
327 unlink(DestFile.c_str());
328
329 // Create the item
330 Desc.Description = URIDesc;
331 Desc.Owner = this;
332 Desc.ShortDesc = ShortDesc;
333 Desc.URI = URI;
334
335
336 string Final = _config->FindDir("Dir::State::lists");
337 Final += URItoFileName(RealURI);
338 struct stat Buf;
339 if (stat(Final.c_str(),&Buf) == 0)
340 {
341 // File was already in place. It needs to be re-verified
342 // because Release might have changed, so Move it into partial
343 Rename(Final,DestFile);
344 }
345
346 QueueURI(Desc);
347 }
348 /*}}}*/
349 // pkgAcqMetaSig::Custom600Headers - Insert custom request headers /*{{{*/
350 // ---------------------------------------------------------------------
351 /* The only header we use is the last-modified header. */
352 string pkgAcqMetaSig::Custom600Headers()
353 {
354 struct stat Buf;
355 if (stat(DestFile.c_str(),&Buf) != 0)
356 return "\nIndex-File: true";
357
358 return "\nIndex-File: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
359 }
360
361 void pkgAcqMetaSig::Done(string Message,unsigned long Size,string MD5,
362 pkgAcquire::MethodConfig *Cfg)
363 {
364 Item::Done(Message,Size,MD5,Cfg);
365
366 string FileName = LookupTag(Message,"Filename");
367 if (FileName.empty() == true)
368 {
369 Status = StatError;
370 ErrorText = "Method gave a blank filename";
371 return;
372 }
373
374 if (FileName != DestFile)
375 {
376 // We have to copy it into place
377 Local = true;
378 Desc.URI = "copy:" + FileName;
379 QueueURI(Desc);
380 return;
381 }
382
383 Complete = true;
384
385 // queue a pkgAcqMetaIndex to be verified against the sig we just retrieved
386 new pkgAcqMetaIndex(Owner, MetaIndexURI, MetaIndexURIDesc, MetaIndexShortDesc,
387 DestFile, IndexTargets, MetaIndexParser);
388
389 }
390 /*}}}*/
391 void pkgAcqMetaSig::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
392 {
393 string Final = _config->FindDir("Dir::State::lists") + URItoFileName(RealURI);
394
395 // if we get a network error we fail gracefully
396 if(LookupTag(Message,"FailReason") == "Timeout" ||
397 LookupTag(Message,"FailReason") == "TmpResolveFailure" ||
398 LookupTag(Message,"FailReason") == "ConnectionRefused") {
399 Item::Failed(Message,Cnf);
400 // move the sigfile back on network failures (and re-authenticated?)
401 if(FileExists(DestFile))
402 Rename(DestFile,Final);
403 return;
404 }
405
406 // Delete any existing sigfile when the acquire failed
407 unlink(Final.c_str());
408
409 // queue a pkgAcqMetaIndex with no sigfile
410 new pkgAcqMetaIndex(Owner, MetaIndexURI, MetaIndexURIDesc, MetaIndexShortDesc,
411 "", IndexTargets, MetaIndexParser);
412
413 if (Cnf->LocalOnly == true ||
414 StringToBool(LookupTag(Message,"Transient-Failure"),false) == false)
415 {
416 // Ignore this
417 Status = StatDone;
418 Complete = false;
419 Dequeue();
420 return;
421 }
422
423 Item::Failed(Message,Cnf);
424 }
425
426 pkgAcqMetaIndex::pkgAcqMetaIndex(pkgAcquire *Owner,
427 string URI,string URIDesc,string ShortDesc,
428 string SigFile,
429 const vector<struct IndexTarget*>* IndexTargets,
430 indexRecords* MetaIndexParser) :
431 Item(Owner), RealURI(URI), SigFile(SigFile), AuthPass(false),
432 MetaIndexParser(MetaIndexParser), IndexTargets(IndexTargets), IMSHit(false)
433 {
434 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
435 DestFile += URItoFileName(URI);
436
437 // Create the item
438 Desc.Description = URIDesc;
439 Desc.Owner = this;
440 Desc.ShortDesc = ShortDesc;
441 Desc.URI = URI;
442
443 QueueURI(Desc);
444 }
445
446 /*}}}*/
447 // pkgAcqMetaIndex::Custom600Headers - Insert custom request headers /*{{{*/
448 // ---------------------------------------------------------------------
449 /* The only header we use is the last-modified header. */
450 string pkgAcqMetaIndex::Custom600Headers()
451 {
452 string Final = _config->FindDir("Dir::State::lists");
453 Final += URItoFileName(RealURI);
454
455 struct stat Buf;
456 if (stat(Final.c_str(),&Buf) != 0)
457 return "\nIndex-File: true";
458
459 return "\nIndex-File: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
460 }
461
462 void pkgAcqMetaIndex::Done(string Message,unsigned long Size,string MD5,
463 pkgAcquire::MethodConfig *Cfg)
464 {
465 Item::Done(Message,Size,MD5,Cfg);
466
467 // MetaIndexes are done in two passes: one to download the
468 // metaindex with an appropriate method, and a second to verify it
469 // with the gpgv method
470
471 if (AuthPass == true)
472 {
473 AuthDone(Message);
474 }
475 else
476 {
477 RetrievalDone(Message);
478 if (!Complete)
479 // Still more retrieving to do
480 return;
481
482 if (SigFile == "")
483 {
484 // There was no signature file, so we are finished. Download
485 // the indexes without verification.
486 QueueIndexes(false);
487 }
488 else
489 {
490 // There was a signature file, so pass it to gpgv for
491 // verification
492
493 if (_config->FindB("Debug::pkgAcquire::Auth", false))
494 std::cerr << "Metaindex acquired, queueing gpg verification ("
495 << SigFile << "," << DestFile << ")\n";
496 AuthPass = true;
497 Desc.URI = "gpgv:" + SigFile;
498 QueueURI(Desc);
499 Mode = "gpgv";
500 }
501 }
502 }
503
504 void pkgAcqMetaIndex::RetrievalDone(string Message)
505 {
506 // We have just finished downloading a Release file (it is not
507 // verified yet)
508
509 string FileName = LookupTag(Message,"Filename");
510 if (FileName.empty() == true)
511 {
512 Status = StatError;
513 ErrorText = "Method gave a blank filename";
514 return;
515 }
516
517 if (FileName != DestFile)
518 {
519 Local = true;
520 Desc.URI = "copy:" + FileName;
521 QueueURI(Desc);
522 return;
523 }
524
525 // see if the download was a IMSHit
526 IMSHit = StringToBool(LookupTag(Message,"IMS-Hit"),false);
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(Message))
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(string Message)
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 string::size_type pos;
640
641 // check for missing sigs (that where not fatal because otherwise we had
642 // bombed earlier)
643 string missingkeys;
644 string msg = _("There are no public key available for the "
645 "following key IDs:\n");
646 pos = Message.find("NO_PUBKEY ");
647 if (pos != std::string::npos)
648 {
649 string::size_type start = pos+strlen("NO_PUBKEY ");
650 string Fingerprint = Message.substr(start, Message.find("\n")-start);
651 missingkeys += (Fingerprint);
652 }
653 if(!missingkeys.empty())
654 _error->Warning("%s", string(msg+missingkeys).c_str());
655
656 string Transformed = MetaIndexParser->GetExpectedDist();
657
658 if (Transformed == "../project/experimental")
659 {
660 Transformed = "experimental";
661 }
662
663 pos = Transformed.rfind('/');
664 if (pos != string::npos)
665 {
666 Transformed = Transformed.substr(0, pos);
667 }
668
669 if (Transformed == ".")
670 {
671 Transformed = "";
672 }
673
674 if (_config->FindB("Debug::pkgAcquire::Auth", false))
675 {
676 std::cerr << "Got Codename: " << MetaIndexParser->GetDist() << std::endl;
677 std::cerr << "Expecting Dist: " << MetaIndexParser->GetExpectedDist() << std::endl;
678 std::cerr << "Transformed Dist: " << Transformed << std::endl;
679 }
680
681 if (MetaIndexParser->CheckDist(Transformed) == false)
682 {
683 // This might become fatal one day
684 // Status = StatAuthError;
685 // ErrorText = "Conflicting distribution; expected "
686 // + MetaIndexParser->GetExpectedDist() + " but got "
687 // + MetaIndexParser->GetDist();
688 // return false;
689 if (!Transformed.empty())
690 {
691 _error->Warning("Conflicting distribution: %s (expected %s but got %s)",
692 Desc.Description.c_str(),
693 Transformed.c_str(),
694 MetaIndexParser->GetDist().c_str());
695 }
696 }
697
698 return true;
699 }
700 /*}}}*/
701 // pkgAcqMetaIndex::Failed - no Release file present or no signature
702 // file present /*{{{*/
703 // ---------------------------------------------------------------------
704 /* */
705 void pkgAcqMetaIndex::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
706 {
707 if (AuthPass == true)
708 {
709 // if we fail the authentication but got the file via a IMS-Hit
710 // this means that the file wasn't downloaded and that it might be
711 // just stale (server problem, proxy etc). we delete what we have
712 // queue it again without i-m-s
713 // alternatively we could just unlink the file and let the user try again
714 if (IMSHit)
715 {
716 Complete = false;
717 Local = false;
718 AuthPass = false;
719 unlink(DestFile.c_str());
720
721 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
722 DestFile += URItoFileName(RealURI);
723 Desc.URI = RealURI;
724 QueueURI(Desc);
725 return;
726 }
727
728 // gpgv method failed
729 _error->Warning("GPG error: %s: %s",
730 Desc.Description.c_str(),
731 LookupTag(Message,"Message").c_str());
732
733 }
734
735 // No Release file was present, or verification failed, so fall
736 // back to queueing Packages files without verification
737 QueueIndexes(false);
738 }
739
740 /*}}}*/
741
742 // AcqArchive::AcqArchive - Constructor /*{{{*/
743 // ---------------------------------------------------------------------
744 /* This just sets up the initial fetch environment and queues the first
745 possibilitiy */
746 pkgAcqArchive::pkgAcqArchive(pkgAcquire *Owner,pkgSourceList *Sources,
747 pkgRecords *Recs,pkgCache::VerIterator const &Version,
748 string &StoreFilename) :
749 Item(Owner), Version(Version), Sources(Sources), Recs(Recs),
750 StoreFilename(StoreFilename), Vf(Version.FileList()),
751 Trusted(false)
752 {
753 Retries = _config->FindI("Acquire::Retries",0);
754
755 if (Version.Arch() == 0)
756 {
757 _error->Error(_("I wasn't able to locate a file for the %s package. "
758 "This might mean you need to manually fix this package. "
759 "(due to missing arch)"),
760 Version.ParentPkg().Name());
761 return;
762 }
763
764 /* We need to find a filename to determine the extension. We make the
765 assumption here that all the available sources for this version share
766 the same extension.. */
767 // Skip not source sources, they do not have file fields.
768 for (; Vf.end() == false; Vf++)
769 {
770 if ((Vf.File()->Flags & pkgCache::Flag::NotSource) != 0)
771 continue;
772 break;
773 }
774
775 // Does not really matter here.. we are going to fail out below
776 if (Vf.end() != true)
777 {
778 // If this fails to get a file name we will bomb out below.
779 pkgRecords::Parser &Parse = Recs->Lookup(Vf);
780 if (_error->PendingError() == true)
781 return;
782
783 // Generate the final file name as: package_version_arch.foo
784 StoreFilename = QuoteString(Version.ParentPkg().Name(),"_:") + '_' +
785 QuoteString(Version.VerStr(),"_:") + '_' +
786 QuoteString(Version.Arch(),"_:.") +
787 "." + flExtension(Parse.FileName());
788 }
789
790 // check if we have one trusted source for the package. if so, switch
791 // to "TrustedOnly" mode
792 for (pkgCache::VerFileIterator i = Version.FileList(); i.end() == false; i++)
793 {
794 pkgIndexFile *Index;
795 if (Sources->FindIndex(i.File(),Index) == false)
796 continue;
797 if (_config->FindB("Debug::pkgAcquire::Auth", false))
798 {
799 std::cerr << "Checking index: " << Index->Describe()
800 << "(Trusted=" << Index->IsTrusted() << ")\n";
801 }
802 if (Index->IsTrusted()) {
803 Trusted = true;
804 break;
805 }
806 }
807
808 // "allow-unauthenticated" restores apts old fetching behaviour
809 // that means that e.g. unauthenticated file:// uris are higher
810 // priority than authenticated http:// uris
811 if (_config->FindB("APT::Get::AllowUnauthenticated",false) == true)
812 Trusted = false;
813
814 // Select a source
815 if (QueueNext() == false && _error->PendingError() == false)
816 _error->Error(_("I wasn't able to locate file for the %s package. "
817 "This might mean you need to manually fix this package."),
818 Version.ParentPkg().Name());
819 }
820 /*}}}*/
821 // AcqArchive::QueueNext - Queue the next file source /*{{{*/
822 // ---------------------------------------------------------------------
823 /* This queues the next available file version for download. It checks if
824 the archive is already available in the cache and stashs the MD5 for
825 checking later. */
826 bool pkgAcqArchive::QueueNext()
827 {
828 for (; Vf.end() == false; Vf++)
829 {
830 // Ignore not source sources
831 if ((Vf.File()->Flags & pkgCache::Flag::NotSource) != 0)
832 continue;
833
834 // Try to cross match against the source list
835 pkgIndexFile *Index;
836 if (Sources->FindIndex(Vf.File(),Index) == false)
837 continue;
838
839 // only try to get a trusted package from another source if that source
840 // is also trusted
841 if(Trusted && !Index->IsTrusted())
842 continue;
843
844 // Grab the text package record
845 pkgRecords::Parser &Parse = Recs->Lookup(Vf);
846 if (_error->PendingError() == true)
847 return false;
848
849 string PkgFile = Parse.FileName();
850 MD5 = Parse.MD5Hash();
851 if (PkgFile.empty() == true)
852 return _error->Error(_("The package index files are corrupted. No Filename: "
853 "field for package %s."),
854 Version.ParentPkg().Name());
855
856 Desc.URI = Index->ArchiveURI(PkgFile);
857 Desc.Description = Index->ArchiveInfo(Version);
858 Desc.Owner = this;
859 Desc.ShortDesc = Version.ParentPkg().Name();
860
861 // See if we already have the file. (Legacy filenames)
862 FileSize = Version->Size;
863 string FinalFile = _config->FindDir("Dir::Cache::Archives") + flNotDir(PkgFile);
864 struct stat Buf;
865 if (stat(FinalFile.c_str(),&Buf) == 0)
866 {
867 // Make sure the size matches
868 if ((unsigned)Buf.st_size == Version->Size)
869 {
870 Complete = true;
871 Local = true;
872 Status = StatDone;
873 StoreFilename = DestFile = FinalFile;
874 return true;
875 }
876
877 /* Hmm, we have a file and its size does not match, this means it is
878 an old style mismatched arch */
879 unlink(FinalFile.c_str());
880 }
881
882 // Check it again using the new style output filenames
883 FinalFile = _config->FindDir("Dir::Cache::Archives") + flNotDir(StoreFilename);
884 if (stat(FinalFile.c_str(),&Buf) == 0)
885 {
886 // Make sure the size matches
887 if ((unsigned)Buf.st_size == Version->Size)
888 {
889 Complete = true;
890 Local = true;
891 Status = StatDone;
892 StoreFilename = DestFile = FinalFile;
893 return true;
894 }
895
896 /* Hmm, we have a file and its size does not match, this shouldnt
897 happen.. */
898 unlink(FinalFile.c_str());
899 }
900
901 DestFile = _config->FindDir("Dir::Cache::Archives") + "partial/" + flNotDir(StoreFilename);
902
903 // Check the destination file
904 if (stat(DestFile.c_str(),&Buf) == 0)
905 {
906 // Hmm, the partial file is too big, erase it
907 if ((unsigned)Buf.st_size > Version->Size)
908 unlink(DestFile.c_str());
909 else
910 PartialSize = Buf.st_size;
911 }
912
913 // Create the item
914 Local = false;
915 Desc.URI = Index->ArchiveURI(PkgFile);
916 Desc.Description = Index->ArchiveInfo(Version);
917 Desc.Owner = this;
918 Desc.ShortDesc = Version.ParentPkg().Name();
919 QueueURI(Desc);
920
921 Vf++;
922 return true;
923 }
924 return false;
925 }
926 /*}}}*/
927 // AcqArchive::Done - Finished fetching /*{{{*/
928 // ---------------------------------------------------------------------
929 /* */
930 void pkgAcqArchive::Done(string Message,unsigned long Size,string Md5Hash,
931 pkgAcquire::MethodConfig *Cfg)
932 {
933 Item::Done(Message,Size,Md5Hash,Cfg);
934
935 // Check the size
936 if (Size != Version->Size)
937 {
938 Status = StatError;
939 ErrorText = _("Size mismatch");
940 return;
941 }
942
943 // Check the md5
944 if (Md5Hash.empty() == false && MD5.empty() == false)
945 {
946 if (Md5Hash != MD5)
947 {
948 Status = StatError;
949 ErrorText = _("MD5Sum mismatch");
950 if(FileExists(DestFile))
951 Rename(DestFile,DestFile + ".FAILED");
952 return;
953 }
954 }
955
956 // Grab the output filename
957 string FileName = LookupTag(Message,"Filename");
958 if (FileName.empty() == true)
959 {
960 Status = StatError;
961 ErrorText = "Method gave a blank filename";
962 return;
963 }
964
965 Complete = true;
966
967 // Reference filename
968 if (FileName != DestFile)
969 {
970 StoreFilename = DestFile = FileName;
971 Local = true;
972 return;
973 }
974
975 // Done, move it into position
976 string FinalFile = _config->FindDir("Dir::Cache::Archives");
977 FinalFile += flNotDir(StoreFilename);
978 Rename(DestFile,FinalFile);
979
980 StoreFilename = DestFile = FinalFile;
981 Complete = true;
982 }
983 /*}}}*/
984 // AcqArchive::Failed - Failure handler /*{{{*/
985 // ---------------------------------------------------------------------
986 /* Here we try other sources */
987 void pkgAcqArchive::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
988 {
989 ErrorText = LookupTag(Message,"Message");
990
991 /* We don't really want to retry on failed media swaps, this prevents
992 that. An interesting observation is that permanent failures are not
993 recorded. */
994 if (Cnf->Removable == true &&
995 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
996 {
997 // Vf = Version.FileList();
998 while (Vf.end() == false) Vf++;
999 StoreFilename = string();
1000 Item::Failed(Message,Cnf);
1001 return;
1002 }
1003
1004 if (QueueNext() == false)
1005 {
1006 // This is the retry counter
1007 if (Retries != 0 &&
1008 Cnf->LocalOnly == false &&
1009 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
1010 {
1011 Retries--;
1012 Vf = Version.FileList();
1013 if (QueueNext() == true)
1014 return;
1015 }
1016
1017 StoreFilename = string();
1018 Item::Failed(Message,Cnf);
1019 }
1020 }
1021 /*}}}*/
1022 // AcqArchive::IsTrusted - Determine whether this archive comes from a
1023 // trusted source /*{{{*/
1024 // ---------------------------------------------------------------------
1025 bool pkgAcqArchive::IsTrusted()
1026 {
1027 return Trusted;
1028 }
1029
1030 // AcqArchive::Finished - Fetching has finished, tidy up /*{{{*/
1031 // ---------------------------------------------------------------------
1032 /* */
1033 void pkgAcqArchive::Finished()
1034 {
1035 if (Status == pkgAcquire::Item::StatDone &&
1036 Complete == true)
1037 return;
1038 StoreFilename = string();
1039 }
1040 /*}}}*/
1041
1042 // AcqFile::pkgAcqFile - Constructor /*{{{*/
1043 // ---------------------------------------------------------------------
1044 /* The file is added to the queue */
1045 pkgAcqFile::pkgAcqFile(pkgAcquire *Owner,string URI,string MD5,
1046 unsigned long Size,string Dsc,string ShortDesc,
1047 const string &DestDir, const string &DestFilename) :
1048 Item(Owner), Md5Hash(MD5)
1049 {
1050 Retries = _config->FindI("Acquire::Retries",0);
1051
1052 if(!DestFilename.empty())
1053 DestFile = DestFilename;
1054 else if(!DestDir.empty())
1055 DestFile = DestDir + "/" + flNotDir(URI);
1056 else
1057 DestFile = flNotDir(URI);
1058
1059 // Create the item
1060 Desc.URI = URI;
1061 Desc.Description = Dsc;
1062 Desc.Owner = this;
1063
1064 // Set the short description to the archive component
1065 Desc.ShortDesc = ShortDesc;
1066
1067 // Get the transfer sizes
1068 FileSize = Size;
1069 struct stat Buf;
1070 if (stat(DestFile.c_str(),&Buf) == 0)
1071 {
1072 // Hmm, the partial file is too big, erase it
1073 if ((unsigned)Buf.st_size > Size)
1074 unlink(DestFile.c_str());
1075 else
1076 PartialSize = Buf.st_size;
1077 }
1078
1079 QueueURI(Desc);
1080 }
1081 /*}}}*/
1082 // AcqFile::Done - Item downloaded OK /*{{{*/
1083 // ---------------------------------------------------------------------
1084 /* */
1085 void pkgAcqFile::Done(string Message,unsigned long Size,string MD5,
1086 pkgAcquire::MethodConfig *Cnf)
1087 {
1088 // Check the md5
1089 if (Md5Hash.empty() == false && MD5.empty() == false)
1090 {
1091 if (Md5Hash != MD5)
1092 {
1093 Status = StatError;
1094 ErrorText = "MD5Sum mismatch";
1095 Rename(DestFile,DestFile + ".FAILED");
1096 return;
1097 }
1098 }
1099
1100 Item::Done(Message,Size,MD5,Cnf);
1101
1102 string FileName = LookupTag(Message,"Filename");
1103 if (FileName.empty() == true)
1104 {
1105 Status = StatError;
1106 ErrorText = "Method gave a blank filename";
1107 return;
1108 }
1109
1110 Complete = true;
1111
1112 // The files timestamp matches
1113 if (StringToBool(LookupTag(Message,"IMS-Hit"),false) == true)
1114 return;
1115
1116 // We have to copy it into place
1117 if (FileName != DestFile)
1118 {
1119 Local = true;
1120 if (_config->FindB("Acquire::Source-Symlinks",true) == false ||
1121 Cnf->Removable == true)
1122 {
1123 Desc.URI = "copy:" + FileName;
1124 QueueURI(Desc);
1125 return;
1126 }
1127
1128 // Erase the file if it is a symlink so we can overwrite it
1129 struct stat St;
1130 if (lstat(DestFile.c_str(),&St) == 0)
1131 {
1132 if (S_ISLNK(St.st_mode) != 0)
1133 unlink(DestFile.c_str());
1134 }
1135
1136 // Symlink the file
1137 if (symlink(FileName.c_str(),DestFile.c_str()) != 0)
1138 {
1139 ErrorText = "Link to " + DestFile + " failure ";
1140 Status = StatError;
1141 Complete = false;
1142 }
1143 }
1144 }
1145 /*}}}*/
1146 // AcqFile::Failed - Failure handler /*{{{*/
1147 // ---------------------------------------------------------------------
1148 /* Here we try other sources */
1149 void pkgAcqFile::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
1150 {
1151 ErrorText = LookupTag(Message,"Message");
1152
1153 // This is the retry counter
1154 if (Retries != 0 &&
1155 Cnf->LocalOnly == false &&
1156 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
1157 {
1158 Retries--;
1159 QueueURI(Desc);
1160 return;
1161 }
1162
1163 Item::Failed(Message,Cnf);
1164 }
1165 /*}}}*/