]> git.saurik.com Git - apt.git/blob - apt-pkg/acquire-item.cc
b7ca82c0c1f199441ba37b030e87a6f12e3876ff
[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 #include <config.h>
17
18 #include <apt-pkg/acquire-item.h>
19 #include <apt-pkg/configuration.h>
20 #include <apt-pkg/aptconfiguration.h>
21 #include <apt-pkg/sourcelist.h>
22 #include <apt-pkg/error.h>
23 #include <apt-pkg/strutl.h>
24 #include <apt-pkg/fileutl.h>
25 #include <apt-pkg/sha1.h>
26 #include <apt-pkg/tagfile.h>
27 #include <apt-pkg/indexrecords.h>
28 #include <apt-pkg/acquire.h>
29 #include <apt-pkg/hashes.h>
30 #include <apt-pkg/indexfile.h>
31 #include <apt-pkg/pkgcache.h>
32 #include <apt-pkg/cacheiterators.h>
33 #include <apt-pkg/pkgrecords.h>
34
35 #include <stddef.h>
36 #include <stdlib.h>
37 #include <string.h>
38 #include <iostream>
39 #include <vector>
40 #include <sys/stat.h>
41 #include <unistd.h>
42 #include <errno.h>
43 #include <string>
44 #include <sstream>
45 #include <stdio.h>
46 #include <ctime>
47
48 #include <apti18n.h>
49 /*}}}*/
50
51 using namespace std;
52
53 static void printHashSumComparision(std::string const &URI, HashStringList const &Expected, HashStringList const &Actual) /*{{{*/
54 {
55 if (_config->FindB("Debug::Acquire::HashSumMismatch", false) == false)
56 return;
57 std::cerr << std::endl << URI << ":" << std::endl << " Expected Hash: " << std::endl;
58 for (HashStringList::const_iterator hs = Expected.begin(); hs != Expected.end(); ++hs)
59 std::cerr << "\t- " << hs->toStr() << std::endl;
60 std::cerr << " Actual Hash: " << std::endl;
61 for (HashStringList::const_iterator hs = Actual.begin(); hs != Actual.end(); ++hs)
62 std::cerr << "\t- " << hs->toStr() << std::endl;
63 }
64 /*}}}*/
65
66 // Acquire::Item::Item - Constructor /*{{{*/
67 pkgAcquire::Item::Item(pkgAcquire *Owner, HashStringList const &ExpectedHashes) :
68 Owner(Owner), FileSize(0), PartialSize(0), Mode(0), ID(0), Complete(false),
69 Local(false), QueueCounter(0), ExpectedAdditionalItems(0),
70 ExpectedHashes(ExpectedHashes)
71 {
72 Owner->Add(this);
73 Status = StatIdle;
74 }
75 /*}}}*/
76 // Acquire::Item::~Item - Destructor /*{{{*/
77 // ---------------------------------------------------------------------
78 /* */
79 pkgAcquire::Item::~Item()
80 {
81 Owner->Remove(this);
82 }
83 /*}}}*/
84 // Acquire::Item::Failed - Item failed to download /*{{{*/
85 // ---------------------------------------------------------------------
86 /* We return to an idle state if there are still other queues that could
87 fetch this object */
88 void pkgAcquire::Item::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
89 {
90 Status = StatIdle;
91 ErrorText = LookupTag(Message,"Message");
92 UsedMirror = LookupTag(Message,"UsedMirror");
93 if (QueueCounter <= 1)
94 {
95 /* This indicates that the file is not available right now but might
96 be sometime later. If we do a retry cycle then this should be
97 retried [CDROMs] */
98 if (Cnf->LocalOnly == true &&
99 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
100 {
101 Status = StatIdle;
102 Dequeue();
103 return;
104 }
105
106 Status = StatError;
107 Dequeue();
108 }
109
110 // report mirror failure back to LP if we actually use a mirror
111 string FailReason = LookupTag(Message, "FailReason");
112 if(FailReason.size() != 0)
113 ReportMirrorFailure(FailReason);
114 else
115 ReportMirrorFailure(ErrorText);
116 }
117 /*}}}*/
118 // Acquire::Item::Start - Item has begun to download /*{{{*/
119 // ---------------------------------------------------------------------
120 /* Stash status and the file size. Note that setting Complete means
121 sub-phases of the acquire process such as decompresion are operating */
122 void pkgAcquire::Item::Start(string /*Message*/,unsigned long long Size)
123 {
124 Status = StatFetching;
125 if (FileSize == 0 && Complete == false)
126 FileSize = Size;
127 }
128 /*}}}*/
129 // Acquire::Item::Done - Item downloaded OK /*{{{*/
130 // ---------------------------------------------------------------------
131 /* */
132 void pkgAcquire::Item::Done(string Message,unsigned long long Size,HashStringList const &/*Hash*/,
133 pkgAcquire::MethodConfig * /*Cnf*/)
134 {
135 // We just downloaded something..
136 string FileName = LookupTag(Message,"Filename");
137 UsedMirror = LookupTag(Message,"UsedMirror");
138 if (Complete == false && !Local && FileName == DestFile)
139 {
140 if (Owner->Log != 0)
141 Owner->Log->Fetched(Size,atoi(LookupTag(Message,"Resume-Point","0").c_str()));
142 }
143
144 if (FileSize == 0)
145 FileSize= Size;
146 Status = StatDone;
147 ErrorText = string();
148 Owner->Dequeue(this);
149 }
150 /*}}}*/
151 // Acquire::Item::Rename - Rename a file /*{{{*/
152 // ---------------------------------------------------------------------
153 /* This helper function is used by a lot of item methods as their final
154 step */
155 void pkgAcquire::Item::Rename(string From,string To)
156 {
157 if (rename(From.c_str(),To.c_str()) != 0)
158 {
159 char S[300];
160 snprintf(S,sizeof(S),_("rename failed, %s (%s -> %s)."),strerror(errno),
161 From.c_str(),To.c_str());
162 Status = StatError;
163 ErrorText = S;
164 }
165 }
166 /*}}}*/
167 bool pkgAcquire::Item::RenameOnError(pkgAcquire::Item::RenameOnErrorState const error)/*{{{*/
168 {
169 if(FileExists(DestFile))
170 Rename(DestFile, DestFile + ".FAILED");
171
172 switch (error)
173 {
174 case HashSumMismatch:
175 ErrorText = _("Hash Sum mismatch");
176 Status = StatAuthError;
177 ReportMirrorFailure("HashChecksumFailure");
178 break;
179 case SizeMismatch:
180 ErrorText = _("Size mismatch");
181 Status = StatAuthError;
182 ReportMirrorFailure("SizeFailure");
183 break;
184 case InvalidFormat:
185 ErrorText = _("Invalid file format");
186 Status = StatError;
187 // do not report as usually its not the mirrors fault, but Portal/Proxy
188 break;
189 }
190 return false;
191 }
192 /*}}}*/
193 // Acquire::Item::ReportMirrorFailure /*{{{*/
194 // ---------------------------------------------------------------------
195 void pkgAcquire::Item::ReportMirrorFailure(string FailCode)
196 {
197 // we only act if a mirror was used at all
198 if(UsedMirror.empty())
199 return;
200 #if 0
201 std::cerr << "\nReportMirrorFailure: "
202 << UsedMirror
203 << " Uri: " << DescURI()
204 << " FailCode: "
205 << FailCode << std::endl;
206 #endif
207 const char *Args[40];
208 unsigned int i = 0;
209 string report = _config->Find("Methods::Mirror::ProblemReporting",
210 "/usr/lib/apt/apt-report-mirror-failure");
211 if(!FileExists(report))
212 return;
213 Args[i++] = report.c_str();
214 Args[i++] = UsedMirror.c_str();
215 Args[i++] = DescURI().c_str();
216 Args[i++] = FailCode.c_str();
217 Args[i++] = NULL;
218 pid_t pid = ExecFork();
219 if(pid < 0)
220 {
221 _error->Error("ReportMirrorFailure Fork failed");
222 return;
223 }
224 else if(pid == 0)
225 {
226 execvp(Args[0], (char**)Args);
227 std::cerr << "Could not exec " << Args[0] << std::endl;
228 _exit(100);
229 }
230 if(!ExecWait(pid, "report-mirror-failure"))
231 {
232 _error->Warning("Couldn't report problem to '%s'",
233 _config->Find("Methods::Mirror::ProblemReporting").c_str());
234 }
235 }
236 /*}}}*/
237 // AcqSubIndex::AcqSubIndex - Constructor /*{{{*/
238 // ---------------------------------------------------------------------
239 /* Get a sub-index file based on checksums from a 'master' file and
240 possibly query additional files */
241 pkgAcqSubIndex::pkgAcqSubIndex(pkgAcquire *Owner, string const &URI,
242 string const &URIDesc, string const &ShortDesc,
243 HashStringList const &ExpectedHashes)
244 : Item(Owner, ExpectedHashes)
245 {
246 /* XXX: Beware: Currently this class does nothing (of value) anymore ! */
247 Debug = _config->FindB("Debug::pkgAcquire::SubIndex",false);
248
249 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
250 DestFile += URItoFileName(URI);
251
252 Desc.URI = URI;
253 Desc.Description = URIDesc;
254 Desc.Owner = this;
255 Desc.ShortDesc = ShortDesc;
256
257 QueueURI(Desc);
258
259 if(Debug)
260 std::clog << "pkgAcqSubIndex: " << Desc.URI << std::endl;
261 }
262 /*}}}*/
263 // AcqSubIndex::Custom600Headers - Insert custom request headers /*{{{*/
264 // ---------------------------------------------------------------------
265 /* The only header we use is the last-modified header. */
266 string pkgAcqSubIndex::Custom600Headers() const
267 {
268 string Final = _config->FindDir("Dir::State::lists");
269 Final += URItoFileName(Desc.URI);
270
271 struct stat Buf;
272 if (stat(Final.c_str(),&Buf) != 0)
273 return "\nIndex-File: true\nFail-Ignore: true\n";
274 return "\nIndex-File: true\nFail-Ignore: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
275 }
276 /*}}}*/
277 void pkgAcqSubIndex::Failed(string Message,pkgAcquire::MethodConfig * /*Cnf*/)/*{{{*/
278 {
279 if(Debug)
280 std::clog << "pkgAcqSubIndex failed: " << Desc.URI << " with " << Message << std::endl;
281
282 Complete = false;
283 Status = StatDone;
284 Dequeue();
285
286 // No good Index is provided
287 }
288 /*}}}*/
289 void pkgAcqSubIndex::Done(string Message,unsigned long long Size,HashStringList const &Hashes, /*{{{*/
290 pkgAcquire::MethodConfig *Cnf)
291 {
292 if(Debug)
293 std::clog << "pkgAcqSubIndex::Done(): " << Desc.URI << std::endl;
294
295 string FileName = LookupTag(Message,"Filename");
296 if (FileName.empty() == true)
297 {
298 Status = StatError;
299 ErrorText = "Method gave a blank filename";
300 return;
301 }
302
303 if (FileName != DestFile)
304 {
305 Local = true;
306 Desc.URI = "copy:" + FileName;
307 QueueURI(Desc);
308 return;
309 }
310
311 Item::Done(Message, Size, Hashes, Cnf);
312
313 string FinalFile = _config->FindDir("Dir::State::lists")+URItoFileName(Desc.URI);
314
315 /* Downloaded invalid transindex => Error (LP: #346386) (Closes: #627642) */
316 indexRecords SubIndexParser;
317 if (FileExists(DestFile) == true && !SubIndexParser.Load(DestFile)) {
318 Status = StatError;
319 ErrorText = SubIndexParser.ErrorText;
320 return;
321 }
322
323 // success in downloading the index
324 // rename the index
325 if(Debug)
326 std::clog << "Renaming: " << DestFile << " -> " << FinalFile << std::endl;
327 Rename(DestFile,FinalFile);
328 chmod(FinalFile.c_str(),0644);
329 DestFile = FinalFile;
330
331 if(ParseIndex(DestFile) == false)
332 return Failed("", NULL);
333
334 Complete = true;
335 Status = StatDone;
336 Dequeue();
337 return;
338 }
339 /*}}}*/
340 bool pkgAcqSubIndex::ParseIndex(string const &IndexFile) /*{{{*/
341 {
342 indexRecords SubIndexParser;
343 if (FileExists(IndexFile) == false || SubIndexParser.Load(IndexFile) == false)
344 return false;
345 // so something with the downloaded index
346 return true;
347 }
348 /*}}}*/
349 // AcqDiffIndex::AcqDiffIndex - Constructor /*{{{*/
350 // ---------------------------------------------------------------------
351 /* Get the DiffIndex file first and see if there are patches available
352 * If so, create a pkgAcqIndexDiffs fetcher that will get and apply the
353 * patches. If anything goes wrong in that process, it will fall back to
354 * the original packages file
355 */
356 pkgAcqDiffIndex::pkgAcqDiffIndex(pkgAcquire *Owner,
357 IndexTarget const * const Target,
358 HashStringList const &ExpectedHashes,
359 indexRecords *MetaIndexParser)
360 : pkgAcqBaseIndex(Owner, Target, ExpectedHashes, MetaIndexParser)
361 {
362
363 Debug = _config->FindB("Debug::pkgAcquire::Diffs",false);
364
365 RealURI = Target->URI;
366 Desc.Owner = this;
367 Desc.Description = Target->Description + "/DiffIndex";
368 Desc.ShortDesc = Target->ShortDesc;
369 Desc.URI = Target->URI + ".diff/Index";
370
371 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
372 DestFile += URItoFileName(Target->URI) + string(".DiffIndex");
373
374 if(Debug)
375 std::clog << "pkgAcqDiffIndex: " << Desc.URI << std::endl;
376
377 // look for the current package file
378 CurrentPackagesFile = _config->FindDir("Dir::State::lists");
379 CurrentPackagesFile += URItoFileName(RealURI);
380
381 // FIXME: this file:/ check is a hack to prevent fetching
382 // from local sources. this is really silly, and
383 // should be fixed cleanly as soon as possible
384 if(!FileExists(CurrentPackagesFile) ||
385 Desc.URI.substr(0,strlen("file:/")) == "file:/")
386 {
387 // we don't have a pkg file or we don't want to queue
388 if(Debug)
389 std::clog << "No index file, local or canceld by user" << std::endl;
390 Failed("", NULL);
391 return;
392 }
393
394 if(Debug)
395 std::clog << "pkgAcqDiffIndex::pkgAcqDiffIndex(): "
396 << CurrentPackagesFile << std::endl;
397
398 QueueURI(Desc);
399
400 }
401 /*}}}*/
402 // AcqIndex::Custom600Headers - Insert custom request headers /*{{{*/
403 // ---------------------------------------------------------------------
404 /* The only header we use is the last-modified header. */
405 string pkgAcqDiffIndex::Custom600Headers() const
406 {
407 string Final = _config->FindDir("Dir::State::lists");
408 Final += URItoFileName(RealURI) + string(".IndexDiff");
409
410 if(Debug)
411 std::clog << "Custom600Header-IMS: " << Final << std::endl;
412
413 struct stat Buf;
414 if (stat(Final.c_str(),&Buf) != 0)
415 return "\nIndex-File: true";
416
417 return "\nIndex-File: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
418 }
419 /*}}}*/
420 bool pkgAcqDiffIndex::ParseDiffIndex(string IndexDiffFile) /*{{{*/
421 {
422 if(Debug)
423 std::clog << "pkgAcqDiffIndex::ParseIndexDiff() " << IndexDiffFile
424 << std::endl;
425
426 pkgTagSection Tags;
427 string ServerSha1;
428 vector<DiffInfo> available_patches;
429
430 FileFd Fd(IndexDiffFile,FileFd::ReadOnly);
431 pkgTagFile TF(&Fd);
432 if (_error->PendingError() == true)
433 return false;
434
435 if(TF.Step(Tags) == true)
436 {
437 bool found = false;
438 DiffInfo d;
439 string size;
440
441 string const tmp = Tags.FindS("SHA1-Current");
442 std::stringstream ss(tmp);
443 ss >> ServerSha1 >> size;
444 unsigned long const ServerSize = atol(size.c_str());
445
446 FileFd fd(CurrentPackagesFile, FileFd::ReadOnly);
447 SHA1Summation SHA1;
448 SHA1.AddFD(fd);
449 string const local_sha1 = SHA1.Result();
450
451 if(local_sha1 == ServerSha1)
452 {
453 // we have the same sha1 as the server so we are done here
454 if(Debug)
455 std::clog << "Package file is up-to-date" << std::endl;
456 // list cleanup needs to know that this file as well as the already
457 // present index is ours, so we create an empty diff to save it for us
458 new pkgAcqIndexDiffs(Owner, Target, ExpectedHashes, MetaIndexParser,
459 ServerSha1, available_patches);
460 return true;
461 }
462 else
463 {
464 if(Debug)
465 std::clog << "SHA1-Current: " << ServerSha1 << " and we start at "<< fd.Name() << " " << fd.Size() << " " << local_sha1 << std::endl;
466
467 // check the historie and see what patches we need
468 string const history = Tags.FindS("SHA1-History");
469 std::stringstream hist(history);
470 while(hist >> d.sha1 >> size >> d.file)
471 {
472 // read until the first match is found
473 // from that point on, we probably need all diffs
474 if(d.sha1 == local_sha1)
475 found=true;
476 else if (found == false)
477 continue;
478
479 if(Debug)
480 std::clog << "Need to get diff: " << d.file << std::endl;
481 available_patches.push_back(d);
482 }
483
484 if (available_patches.empty() == false)
485 {
486 // patching with too many files is rather slow compared to a fast download
487 unsigned long const fileLimit = _config->FindI("Acquire::PDiffs::FileLimit", 0);
488 if (fileLimit != 0 && fileLimit < available_patches.size())
489 {
490 if (Debug)
491 std::clog << "Need " << available_patches.size() << " diffs (Limit is " << fileLimit
492 << ") so fallback to complete download" << std::endl;
493 return false;
494 }
495
496 // see if the patches are too big
497 found = false; // it was true and it will be true again at the end
498 d = *available_patches.begin();
499 string const firstPatch = d.file;
500 unsigned long patchesSize = 0;
501 std::stringstream patches(Tags.FindS("SHA1-Patches"));
502 while(patches >> d.sha1 >> size >> d.file)
503 {
504 if (firstPatch == d.file)
505 found = true;
506 else if (found == false)
507 continue;
508
509 patchesSize += atol(size.c_str());
510 }
511 unsigned long const sizeLimit = ServerSize * _config->FindI("Acquire::PDiffs::SizeLimit", 100);
512 if (sizeLimit > 0 && (sizeLimit/100) < patchesSize)
513 {
514 if (Debug)
515 std::clog << "Need " << patchesSize << " bytes (Limit is " << sizeLimit/100
516 << ") so fallback to complete download" << std::endl;
517 return false;
518 }
519 }
520 }
521
522 // we have something, queue the next diff
523 if(found)
524 {
525 // queue the diffs
526 string::size_type const last_space = Description.rfind(" ");
527 if(last_space != string::npos)
528 Description.erase(last_space, Description.size()-last_space);
529
530 /* decide if we should download patches one by one or in one go:
531 The first is good if the server merges patches, but many don't so client
532 based merging can be attempt in which case the second is better.
533 "bad things" will happen if patches are merged on the server,
534 but client side merging is attempt as well */
535 bool pdiff_merge = _config->FindB("Acquire::PDiffs::Merge", true);
536 if (pdiff_merge == true)
537 {
538 // reprepro adds this flag if it has merged patches on the server
539 std::string const precedence = Tags.FindS("X-Patch-Precedence");
540 pdiff_merge = (precedence != "merged");
541 }
542
543 if (pdiff_merge == false)
544 {
545 new pkgAcqIndexDiffs(Owner, Target, ExpectedHashes, MetaIndexParser,
546 ServerSha1, available_patches);
547 }
548 else
549 {
550 std::vector<pkgAcqIndexMergeDiffs*> *diffs = new std::vector<pkgAcqIndexMergeDiffs*>(available_patches.size());
551 for(size_t i = 0; i < available_patches.size(); ++i)
552 (*diffs)[i] = new pkgAcqIndexMergeDiffs(Owner, Target,
553 ExpectedHashes,
554 MetaIndexParser,
555 available_patches[i],
556 diffs);
557 }
558
559 Complete = false;
560 Status = StatDone;
561 Dequeue();
562 return true;
563 }
564 }
565
566 // Nothing found, report and return false
567 // Failing here is ok, if we return false later, the full
568 // IndexFile is queued
569 if(Debug)
570 std::clog << "Can't find a patch in the index file" << std::endl;
571 return false;
572 }
573 /*}}}*/
574 void pkgAcqDiffIndex::Failed(string Message,pkgAcquire::MethodConfig * /*Cnf*/)/*{{{*/
575 {
576 if(Debug)
577 std::clog << "pkgAcqDiffIndex failed: " << Desc.URI << " with " << Message << std::endl
578 << "Falling back to normal index file acquire" << std::endl;
579
580 new pkgAcqIndex(Owner, Target, ExpectedHashes, MetaIndexParser);
581
582 Complete = false;
583 Status = StatDone;
584 Dequeue();
585 }
586 /*}}}*/
587 void pkgAcqDiffIndex::Done(string Message,unsigned long long Size,HashStringList const &Hashes, /*{{{*/
588 pkgAcquire::MethodConfig *Cnf)
589 {
590 if(Debug)
591 std::clog << "pkgAcqDiffIndex::Done(): " << Desc.URI << std::endl;
592
593 Item::Done(Message, Size, Hashes, Cnf);
594
595 string FinalFile;
596 FinalFile = _config->FindDir("Dir::State::lists")+URItoFileName(RealURI);
597
598 // success in downloading the index
599 // rename the index
600 FinalFile += string(".IndexDiff");
601 if(Debug)
602 std::clog << "Renaming: " << DestFile << " -> " << FinalFile
603 << std::endl;
604 Rename(DestFile,FinalFile);
605 chmod(FinalFile.c_str(),0644);
606 DestFile = FinalFile;
607
608 if(!ParseDiffIndex(DestFile))
609 return Failed("", NULL);
610
611 Complete = true;
612 Status = StatDone;
613 Dequeue();
614 return;
615 }
616 /*}}}*/
617 // AcqIndexDiffs::AcqIndexDiffs - Constructor /*{{{*/
618 // ---------------------------------------------------------------------
619 /* The package diff is added to the queue. one object is constructed
620 * for each diff and the index
621 */
622 pkgAcqIndexDiffs::pkgAcqIndexDiffs(pkgAcquire *Owner,
623 struct IndexTarget const * const Target,
624 HashStringList const &ExpectedHashes,
625 indexRecords *MetaIndexParser,
626 string ServerSha1,
627 vector<DiffInfo> diffs)
628 : pkgAcqBaseIndex(Owner, Target, ExpectedHashes, MetaIndexParser),
629 available_patches(diffs), ServerSha1(ServerSha1)
630 {
631
632 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
633 DestFile += URItoFileName(Target->URI);
634
635 Debug = _config->FindB("Debug::pkgAcquire::Diffs",false);
636
637 RealURI = Target->URI;
638 Desc.Owner = this;
639 Description = Target->Description;
640 Desc.ShortDesc = Target->ShortDesc;
641
642 if(available_patches.empty() == true)
643 {
644 // we are done (yeah!)
645 Finish(true);
646 }
647 else
648 {
649 // get the next diff
650 State = StateFetchDiff;
651 QueueNextDiff();
652 }
653 }
654 /*}}}*/
655 void pkgAcqIndexDiffs::Failed(string Message,pkgAcquire::MethodConfig * /*Cnf*/)/*{{{*/
656 {
657 if(Debug)
658 std::clog << "pkgAcqIndexDiffs failed: " << Desc.URI << " with " << Message << std::endl
659 << "Falling back to normal index file acquire" << std::endl;
660 new pkgAcqIndex(Owner, Target, ExpectedHashes, MetaIndexParser);
661 Finish();
662 }
663 /*}}}*/
664 // Finish - helper that cleans the item out of the fetcher queue /*{{{*/
665 void pkgAcqIndexDiffs::Finish(bool allDone)
666 {
667 // we restore the original name, this is required, otherwise
668 // the file will be cleaned
669 if(allDone)
670 {
671 DestFile = _config->FindDir("Dir::State::lists");
672 DestFile += URItoFileName(RealURI);
673
674 if(HashSums().usable() && !HashSums().VerifyFile(DestFile))
675 {
676 RenameOnError(HashSumMismatch);
677 Dequeue();
678 return;
679 }
680
681 // this is for the "real" finish
682 Complete = true;
683 Status = StatDone;
684 Dequeue();
685 if(Debug)
686 std::clog << "\n\nallDone: " << DestFile << "\n" << std::endl;
687 return;
688 }
689
690 if(Debug)
691 std::clog << "Finishing: " << Desc.URI << std::endl;
692 Complete = false;
693 Status = StatDone;
694 Dequeue();
695 return;
696 }
697 /*}}}*/
698 bool pkgAcqIndexDiffs::QueueNextDiff() /*{{{*/
699 {
700
701 // calc sha1 of the just patched file
702 string FinalFile = _config->FindDir("Dir::State::lists");
703 FinalFile += URItoFileName(RealURI);
704
705 FileFd fd(FinalFile, FileFd::ReadOnly);
706 SHA1Summation SHA1;
707 SHA1.AddFD(fd);
708 string local_sha1 = string(SHA1.Result());
709 if(Debug)
710 std::clog << "QueueNextDiff: "
711 << FinalFile << " (" << local_sha1 << ")"<<std::endl;
712
713 // final file reached before all patches are applied
714 if(local_sha1 == ServerSha1)
715 {
716 Finish(true);
717 return true;
718 }
719
720 // remove all patches until the next matching patch is found
721 // this requires the Index file to be ordered
722 for(vector<DiffInfo>::iterator I=available_patches.begin();
723 available_patches.empty() == false &&
724 I != available_patches.end() &&
725 I->sha1 != local_sha1;
726 ++I)
727 {
728 available_patches.erase(I);
729 }
730
731 // error checking and falling back if no patch was found
732 if(available_patches.empty() == true)
733 {
734 Failed("", NULL);
735 return false;
736 }
737
738 // queue the right diff
739 Desc.URI = RealURI + ".diff/" + available_patches[0].file + ".gz";
740 Desc.Description = Description + " " + available_patches[0].file + string(".pdiff");
741 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
742 DestFile += URItoFileName(RealURI + ".diff/" + available_patches[0].file);
743
744 if(Debug)
745 std::clog << "pkgAcqIndexDiffs::QueueNextDiff(): " << Desc.URI << std::endl;
746
747 QueueURI(Desc);
748
749 return true;
750 }
751 /*}}}*/
752 void pkgAcqIndexDiffs::Done(string Message,unsigned long long Size, HashStringList const &Hashes, /*{{{*/
753 pkgAcquire::MethodConfig *Cnf)
754 {
755 if(Debug)
756 std::clog << "pkgAcqIndexDiffs::Done(): " << Desc.URI << std::endl;
757
758 Item::Done(Message, Size, Hashes, Cnf);
759
760 string FinalFile;
761 FinalFile = _config->FindDir("Dir::State::lists")+URItoFileName(RealURI);
762
763 // success in downloading a diff, enter ApplyDiff state
764 if(State == StateFetchDiff)
765 {
766
767 // rred excepts the patch as $FinalFile.ed
768 Rename(DestFile,FinalFile+".ed");
769
770 if(Debug)
771 std::clog << "Sending to rred method: " << FinalFile << std::endl;
772
773 State = StateApplyDiff;
774 Local = true;
775 Desc.URI = "rred:" + FinalFile;
776 QueueURI(Desc);
777 Mode = "rred";
778 return;
779 }
780
781
782 // success in download/apply a diff, queue next (if needed)
783 if(State == StateApplyDiff)
784 {
785 // remove the just applied patch
786 available_patches.erase(available_patches.begin());
787 unlink((FinalFile + ".ed").c_str());
788
789 // move into place
790 if(Debug)
791 {
792 std::clog << "Moving patched file in place: " << std::endl
793 << DestFile << " -> " << FinalFile << std::endl;
794 }
795 Rename(DestFile,FinalFile);
796 chmod(FinalFile.c_str(),0644);
797
798 // see if there is more to download
799 if(available_patches.empty() == false) {
800 new pkgAcqIndexDiffs(Owner, Target,
801 ExpectedHashes, MetaIndexParser,
802 ServerSha1, available_patches);
803 return Finish();
804 } else
805 return Finish(true);
806 }
807 }
808 /*}}}*/
809 // AcqIndexMergeDiffs::AcqIndexMergeDiffs - Constructor /*{{{*/
810 pkgAcqIndexMergeDiffs::pkgAcqIndexMergeDiffs(pkgAcquire *Owner,
811 struct IndexTarget const * const Target,
812 HashStringList const &ExpectedHashes,
813 indexRecords *MetaIndexParser,
814 DiffInfo const &patch,
815 std::vector<pkgAcqIndexMergeDiffs*> const * const allPatches)
816 : pkgAcqBaseIndex(Owner, Target, ExpectedHashes, MetaIndexParser),
817 patch(patch), allPatches(allPatches), State(StateFetchDiff)
818 {
819
820 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
821 DestFile += URItoFileName(Target->URI);
822
823 Debug = _config->FindB("Debug::pkgAcquire::Diffs",false);
824
825 RealURI = Target->URI;
826 Desc.Owner = this;
827 Description = Target->Description;
828 Desc.ShortDesc = Target->ShortDesc;
829
830 Desc.URI = RealURI + ".diff/" + patch.file + ".gz";
831 Desc.Description = Description + " " + patch.file + string(".pdiff");
832 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
833 DestFile += URItoFileName(RealURI + ".diff/" + patch.file);
834
835 if(Debug)
836 std::clog << "pkgAcqIndexMergeDiffs: " << Desc.URI << std::endl;
837
838 QueueURI(Desc);
839 }
840 /*}}}*/
841 void pkgAcqIndexMergeDiffs::Failed(string Message,pkgAcquire::MethodConfig * /*Cnf*/)/*{{{*/
842 {
843 if(Debug)
844 std::clog << "pkgAcqIndexMergeDiffs failed: " << Desc.URI << " with " << Message << std::endl;
845 Complete = false;
846 Status = StatDone;
847 Dequeue();
848
849 // check if we are the first to fail, otherwise we are done here
850 State = StateDoneDiff;
851 for (std::vector<pkgAcqIndexMergeDiffs *>::const_iterator I = allPatches->begin();
852 I != allPatches->end(); ++I)
853 if ((*I)->State == StateErrorDiff)
854 return;
855
856 // first failure means we should fallback
857 State = StateErrorDiff;
858 std::clog << "Falling back to normal index file acquire" << std::endl;
859 new pkgAcqIndex(Owner, Target, ExpectedHashes, MetaIndexParser);
860 }
861 /*}}}*/
862 void pkgAcqIndexMergeDiffs::Done(string Message,unsigned long long Size,HashStringList const &Hashes, /*{{{*/
863 pkgAcquire::MethodConfig *Cnf)
864 {
865 if(Debug)
866 std::clog << "pkgAcqIndexMergeDiffs::Done(): " << Desc.URI << std::endl;
867
868 Item::Done(Message,Size,Hashes,Cnf);
869
870 string const FinalFile = _config->FindDir("Dir::State::lists") + URItoFileName(RealURI);
871
872 if (State == StateFetchDiff)
873 {
874 // rred expects the patch as $FinalFile.ed.$patchname.gz
875 Rename(DestFile, FinalFile + ".ed." + patch.file + ".gz");
876
877 // check if this is the last completed diff
878 State = StateDoneDiff;
879 for (std::vector<pkgAcqIndexMergeDiffs *>::const_iterator I = allPatches->begin();
880 I != allPatches->end(); ++I)
881 if ((*I)->State != StateDoneDiff)
882 {
883 if(Debug)
884 std::clog << "Not the last done diff in the batch: " << Desc.URI << std::endl;
885 return;
886 }
887
888 // this is the last completed diff, so we are ready to apply now
889 State = StateApplyDiff;
890
891 if(Debug)
892 std::clog << "Sending to rred method: " << FinalFile << std::endl;
893
894 Local = true;
895 Desc.URI = "rred:" + FinalFile;
896 QueueURI(Desc);
897 Mode = "rred";
898 return;
899 }
900 // success in download/apply all diffs, clean up
901 else if (State == StateApplyDiff)
902 {
903 // see if we really got the expected file
904 if(ExpectedHashes.usable() && !ExpectedHashes.VerifyFile(DestFile))
905 {
906 RenameOnError(HashSumMismatch);
907 return;
908 }
909
910 // move the result into place
911 if(Debug)
912 std::clog << "Moving patched file in place: " << std::endl
913 << DestFile << " -> " << FinalFile << std::endl;
914 Rename(DestFile, FinalFile);
915 chmod(FinalFile.c_str(), 0644);
916
917 // otherwise lists cleanup will eat the file
918 DestFile = FinalFile;
919
920 // ensure the ed's are gone regardless of list-cleanup
921 for (std::vector<pkgAcqIndexMergeDiffs *>::const_iterator I = allPatches->begin();
922 I != allPatches->end(); ++I)
923 {
924 std::string patch = FinalFile + ".ed." + (*I)->patch.file + ".gz";
925 unlink(patch.c_str());
926 }
927
928 // all set and done
929 Complete = true;
930 if(Debug)
931 std::clog << "allDone: " << DestFile << "\n" << std::endl;
932 }
933 }
934 /*}}}*/
935 // AcqIndex::AcqIndex - Constructor /*{{{*/
936 // ---------------------------------------------------------------------
937 /* The package file is added to the queue and a second class is
938 instantiated to fetch the revision file */
939 pkgAcqIndex::pkgAcqIndex(pkgAcquire *Owner,
940 string URI,string URIDesc,string ShortDesc,
941 HashStringList const &ExpectedHash, string comprExt)
942 : pkgAcqBaseIndex(Owner, NULL, ExpectedHash, NULL), RealURI(URI)
943 {
944 if(comprExt.empty() == true)
945 {
946 // autoselect the compression method
947 std::vector<std::string> types = APT::Configuration::getCompressionTypes();
948 for (std::vector<std::string>::const_iterator t = types.begin(); t != types.end(); ++t)
949 comprExt.append(*t).append(" ");
950 if (comprExt.empty() == false)
951 comprExt.erase(comprExt.end()-1);
952 }
953 CompressionExtension = comprExt;
954
955 Verify = true;
956
957 Init(URI, URIDesc, ShortDesc);
958 }
959 pkgAcqIndex::pkgAcqIndex(pkgAcquire *Owner, IndexTarget const *Target,
960 HashStringList const &ExpectedHash,
961 indexRecords *MetaIndexParser)
962 : pkgAcqBaseIndex(Owner, Target, ExpectedHash, MetaIndexParser),
963 RealURI(Target->URI)
964 {
965 // autoselect the compression method
966 std::vector<std::string> types = APT::Configuration::getCompressionTypes();
967 CompressionExtension = "";
968 if (ExpectedHashes.usable())
969 {
970 for (std::vector<std::string>::const_iterator t = types.begin(); t != types.end(); ++t)
971 if (*t == "uncompressed" || MetaIndexParser->Exists(string(Target->MetaKey).append(".").append(*t)) == true)
972 CompressionExtension.append(*t).append(" ");
973 }
974 else
975 {
976 for (std::vector<std::string>::const_iterator t = types.begin(); t != types.end(); ++t)
977 CompressionExtension.append(*t).append(" ");
978 }
979 if (CompressionExtension.empty() == false)
980 CompressionExtension.erase(CompressionExtension.end()-1);
981
982 // only verify non-optional targets, see acquire-item.h for a FIXME
983 // to make this more flexible
984 if (Target->IsOptional())
985 Verify = false;
986 else
987 Verify = true;
988
989 Init(Target->URI, Target->Description, Target->ShortDesc);
990 }
991 /*}}}*/
992 // AcqIndex::Init - defered Constructor /*{{{*/
993 void pkgAcqIndex::Init(string const &URI, string const &URIDesc, string const &ShortDesc) {
994 Decompression = false;
995 Erase = false;
996
997 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
998 DestFile += URItoFileName(URI);
999
1000 std::string const comprExt = CompressionExtension.substr(0, CompressionExtension.find(' '));
1001 std::string MetaKey;
1002 if (comprExt == "uncompressed")
1003 {
1004 Desc.URI = URI;
1005 if(Target)
1006 MetaKey = string(Target->MetaKey);
1007 }
1008 else
1009 {
1010 Desc.URI = URI + '.' + comprExt;
1011 if(Target)
1012 MetaKey = string(Target->MetaKey) + '.' + comprExt;
1013 }
1014
1015 // load the filesize
1016 if(MetaIndexParser)
1017 {
1018 indexRecords::checkSum *Record = MetaIndexParser->Lookup(MetaKey);
1019 if(Record)
1020 FileSize = Record->Size;
1021
1022 // TODO:
1023 // - (maybe?) add support for by-hash into the sources.list as flag
1024 // - make apt-ftparchive generate the hashes (and expire?)
1025 // do the request by-hash
1026 std::string HostKnob = "APT::Acquire::" + ::URI(URI).Host + "::By-Hash";
1027 if(_config->FindB("APT::Acquire::By-Hash", false) == true ||
1028 _config->FindB(HostKnob, false) == true ||
1029 MetaIndexParser->GetSupportsAcquireByHash())
1030 {
1031 indexRecords::checkSum *Record = MetaIndexParser->Lookup(MetaKey);
1032 if(Record)
1033 {
1034 // FIXME: should we really use the best hash here? or a fixed one?
1035 const HashString *TargetHash = Record->Hashes.find("");
1036 std::string ByHash = "/by-hash/" + TargetHash->HashType() + "/" + TargetHash->HashValue();
1037 size_t trailing_slash = Desc.URI.find_last_of("/");
1038 Desc.URI = Desc.URI.replace(
1039 trailing_slash,
1040 Desc.URI.substr(trailing_slash+1).size()+1,
1041 ByHash);
1042 } else {
1043 _error->Warning("By-Hash requested but can not find record for %s",
1044 MetaKey.c_str());
1045 }
1046 }
1047 }
1048
1049 Desc.Description = URIDesc;
1050 Desc.Owner = this;
1051 Desc.ShortDesc = ShortDesc;
1052
1053 QueueURI(Desc);
1054 }
1055 /*}}}*/
1056 // AcqIndex::Custom600Headers - Insert custom request headers /*{{{*/
1057 // ---------------------------------------------------------------------
1058 /* The only header we use is the last-modified header. */
1059 string pkgAcqIndex::Custom600Headers() const
1060 {
1061 string Final = _config->FindDir("Dir::State::lists");
1062 Final += URItoFileName(RealURI);
1063 if (_config->FindB("Acquire::GzipIndexes",false))
1064 Final += ".gz";
1065
1066 string msg = "\nIndex-File: true";
1067 // FIXME: this really should use "IndexTarget::IsOptional()" but that
1068 // seems to be difficult without breaking ABI
1069 if (ShortDesc().find("Translation") != 0)
1070 msg += "\nFail-Ignore: true";
1071 struct stat Buf;
1072 if (stat(Final.c_str(),&Buf) == 0)
1073 msg += "\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
1074
1075 return msg;
1076 }
1077 /*}}}*/
1078 void pkgAcqIndex::Failed(string Message,pkgAcquire::MethodConfig *Cnf) /*{{{*/
1079 {
1080 size_t const nextExt = CompressionExtension.find(' ');
1081 if (nextExt != std::string::npos)
1082 {
1083 CompressionExtension = CompressionExtension.substr(nextExt+1);
1084 Init(RealURI, Desc.Description, Desc.ShortDesc);
1085 return;
1086 }
1087
1088 // on decompression failure, remove bad versions in partial/
1089 if (Decompression && Erase) {
1090 string s = _config->FindDir("Dir::State::lists") + "partial/";
1091 s.append(URItoFileName(RealURI));
1092 unlink(s.c_str());
1093 }
1094
1095 Item::Failed(Message,Cnf);
1096 }
1097 /*}}}*/
1098 // AcqIndex::Done - Finished a fetch /*{{{*/
1099 // ---------------------------------------------------------------------
1100 /* This goes through a number of states.. On the initial fetch the
1101 method could possibly return an alternate filename which points
1102 to the uncompressed version of the file. If this is so the file
1103 is copied into the partial directory. In all other cases the file
1104 is decompressed with a gzip uri. */
1105 void pkgAcqIndex::Done(string Message,unsigned long long Size,HashStringList const &Hashes,
1106 pkgAcquire::MethodConfig *Cfg)
1107 {
1108 Item::Done(Message,Size,Hashes,Cfg);
1109
1110 if (Decompression == true)
1111 {
1112 if (ExpectedHashes.usable() && ExpectedHashes != Hashes)
1113 {
1114 RenameOnError(HashSumMismatch);
1115 printHashSumComparision(RealURI, ExpectedHashes, Hashes);
1116 return;
1117 }
1118
1119 /* Verify the index file for correctness (all indexes must
1120 * have a Package field) (LP: #346386) (Closes: #627642) */
1121 if (Verify == true)
1122 {
1123 FileFd fd(DestFile, FileFd::ReadOnly);
1124 // Only test for correctness if the file is not empty (empty is ok)
1125 if (fd.FileSize() > 0)
1126 {
1127 pkgTagSection sec;
1128 pkgTagFile tag(&fd);
1129
1130 // all our current indexes have a field 'Package' in each section
1131 if (_error->PendingError() == true || tag.Step(sec) == false || sec.Exists("Package") == false)
1132 {
1133 RenameOnError(InvalidFormat);
1134 return;
1135 }
1136 }
1137 }
1138
1139 // Done, move it into position
1140 string FinalFile = _config->FindDir("Dir::State::lists");
1141 FinalFile += URItoFileName(RealURI);
1142 Rename(DestFile,FinalFile);
1143 chmod(FinalFile.c_str(),0644);
1144
1145 /* We restore the original name to DestFile so that the clean operation
1146 will work OK */
1147 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
1148 DestFile += URItoFileName(RealURI);
1149
1150 // Remove the compressed version.
1151 if (Erase == true)
1152 unlink(DestFile.c_str());
1153
1154 return;
1155 }
1156
1157 Erase = false;
1158 Complete = true;
1159
1160 // Handle the unzipd case
1161 string FileName = LookupTag(Message,"Alt-Filename");
1162 if (FileName.empty() == false)
1163 {
1164 // The files timestamp matches
1165 if (StringToBool(LookupTag(Message,"Alt-IMS-Hit"),false) == true)
1166 return;
1167 Decompression = true;
1168 Local = true;
1169 DestFile += ".decomp";
1170 Desc.URI = "copy:" + FileName;
1171 QueueURI(Desc);
1172 Mode = "copy";
1173 return;
1174 }
1175
1176 FileName = LookupTag(Message,"Filename");
1177 if (FileName.empty() == true)
1178 {
1179 Status = StatError;
1180 ErrorText = "Method gave a blank filename";
1181 }
1182
1183 std::string const compExt = CompressionExtension.substr(0, CompressionExtension.find(' '));
1184
1185 // The files timestamp matches
1186 if (StringToBool(LookupTag(Message,"IMS-Hit"),false) == true) {
1187 if (_config->FindB("Acquire::GzipIndexes",false) && compExt == "gz")
1188 // Update DestFile for .gz suffix so that the clean operation keeps it
1189 DestFile += ".gz";
1190 return;
1191 }
1192
1193 if (FileName == DestFile)
1194 Erase = true;
1195 else
1196 Local = true;
1197
1198 string decompProg;
1199
1200 // If we enable compressed indexes and already have gzip, keep it
1201 if (_config->FindB("Acquire::GzipIndexes",false) && compExt == "gz" && !Local) {
1202 string FinalFile = _config->FindDir("Dir::State::lists");
1203 FinalFile += URItoFileName(RealURI) + ".gz";
1204 Rename(DestFile,FinalFile);
1205 chmod(FinalFile.c_str(),0644);
1206
1207 // Update DestFile for .gz suffix so that the clean operation keeps it
1208 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
1209 DestFile += URItoFileName(RealURI) + ".gz";
1210 return;
1211 }
1212
1213 // get the binary name for your used compression type
1214 decompProg = _config->Find(string("Acquire::CompressionTypes::").append(compExt),"");
1215 if(decompProg.empty() == false);
1216 else if(compExt == "uncompressed")
1217 decompProg = "copy";
1218 else {
1219 _error->Error("Unsupported extension: %s", compExt.c_str());
1220 return;
1221 }
1222
1223 Decompression = true;
1224 DestFile += ".decomp";
1225 Desc.URI = decompProg + ":" + FileName;
1226 QueueURI(Desc);
1227
1228 // FIXME: this points to a c++ string that goes out of scope
1229 Mode = decompProg.c_str();
1230 }
1231 /*}}}*/
1232 // AcqIndexTrans::pkgAcqIndexTrans - Constructor /*{{{*/
1233 // ---------------------------------------------------------------------
1234 /* The Translation file is added to the queue */
1235 pkgAcqIndexTrans::pkgAcqIndexTrans(pkgAcquire *Owner,
1236 string URI,string URIDesc,string ShortDesc)
1237 : pkgAcqIndex(Owner, URI, URIDesc, ShortDesc, HashStringList(), "")
1238 {
1239 }
1240 pkgAcqIndexTrans::pkgAcqIndexTrans(pkgAcquire *Owner, IndexTarget const * const Target,
1241 HashStringList const &ExpectedHashes, indexRecords *MetaIndexParser)
1242 : pkgAcqIndex(Owner, Target, ExpectedHashes, MetaIndexParser)
1243 {
1244 // load the filesize
1245 indexRecords::checkSum *Record = MetaIndexParser->Lookup(string(Target->MetaKey));
1246 if(Record)
1247 FileSize = Record->Size;
1248 }
1249 /*}}}*/
1250 // AcqIndexTrans::Custom600Headers - Insert custom request headers /*{{{*/
1251 // ---------------------------------------------------------------------
1252 string pkgAcqIndexTrans::Custom600Headers() const
1253 {
1254 string Final = _config->FindDir("Dir::State::lists");
1255 Final += URItoFileName(RealURI);
1256
1257 struct stat Buf;
1258 if (stat(Final.c_str(),&Buf) != 0)
1259 return "\nFail-Ignore: true\nIndex-File: true";
1260 return "\nFail-Ignore: true\nIndex-File: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
1261 }
1262 /*}}}*/
1263 // AcqIndexTrans::Failed - Silence failure messages for missing files /*{{{*/
1264 // ---------------------------------------------------------------------
1265 /* */
1266 void pkgAcqIndexTrans::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
1267 {
1268 size_t const nextExt = CompressionExtension.find(' ');
1269 if (nextExt != std::string::npos)
1270 {
1271 CompressionExtension = CompressionExtension.substr(nextExt+1);
1272 Init(RealURI, Desc.Description, Desc.ShortDesc);
1273 Status = StatIdle;
1274 return;
1275 }
1276
1277 if (Cnf->LocalOnly == true ||
1278 StringToBool(LookupTag(Message,"Transient-Failure"),false) == false)
1279 {
1280 // Ignore this
1281 Status = StatDone;
1282 Complete = false;
1283 Dequeue();
1284 return;
1285 }
1286
1287 Item::Failed(Message,Cnf);
1288 }
1289 /*}}}*/
1290 pkgAcqMetaSig::pkgAcqMetaSig(pkgAcquire *Owner, /*{{{*/
1291 string URI,string URIDesc,string ShortDesc,
1292 string MetaIndexURI, string MetaIndexURIDesc,
1293 string MetaIndexShortDesc,
1294 const vector<IndexTarget*>* IndexTargets,
1295 indexRecords* MetaIndexParser) :
1296 Item(Owner, HashStringList()), RealURI(URI), MetaIndexURI(MetaIndexURI),
1297 MetaIndexURIDesc(MetaIndexURIDesc), MetaIndexShortDesc(MetaIndexShortDesc),
1298 MetaIndexParser(MetaIndexParser), IndexTargets(IndexTargets)
1299 {
1300 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
1301 DestFile += URItoFileName(URI);
1302
1303 // remove any partial downloaded sig-file in partial/.
1304 // it may confuse proxies and is too small to warrant a
1305 // partial download anyway
1306 unlink(DestFile.c_str());
1307
1308 // Create the item
1309 Desc.Description = URIDesc;
1310 Desc.Owner = this;
1311 Desc.ShortDesc = ShortDesc;
1312 Desc.URI = URI;
1313
1314 string Final = _config->FindDir("Dir::State::lists");
1315 Final += URItoFileName(RealURI);
1316 if (RealFileExists(Final) == true)
1317 {
1318 // File was already in place. It needs to be re-downloaded/verified
1319 // because Release might have changed, we do give it a different
1320 // name than DestFile because otherwise the http method will
1321 // send If-Range requests and there are too many broken servers
1322 // out there that do not understand them
1323 LastGoodSig = DestFile+".reverify";
1324 Rename(Final,LastGoodSig);
1325 }
1326
1327 // we expect the indextargets + one additional Release file
1328 ExpectedAdditionalItems = IndexTargets->size() + 1;
1329
1330 QueueURI(Desc);
1331 }
1332 /*}}}*/
1333 pkgAcqMetaSig::~pkgAcqMetaSig() /*{{{*/
1334 {
1335 // if the file was never queued undo file-changes done in the constructor
1336 if (QueueCounter == 1 && Status == StatIdle && FileSize == 0 && Complete == false &&
1337 LastGoodSig.empty() == false)
1338 {
1339 string const Final = _config->FindDir("Dir::State::lists") + URItoFileName(RealURI);
1340 if (RealFileExists(Final) == false && RealFileExists(LastGoodSig) == true)
1341 Rename(LastGoodSig, Final);
1342 }
1343
1344 }
1345 /*}}}*/
1346 // pkgAcqMetaSig::Custom600Headers - Insert custom request headers /*{{{*/
1347 // ---------------------------------------------------------------------
1348 /* The only header we use is the last-modified header. */
1349 string pkgAcqMetaSig::Custom600Headers() const
1350 {
1351 struct stat Buf;
1352 if (stat(LastGoodSig.c_str(),&Buf) != 0)
1353 return "\nIndex-File: true";
1354
1355 return "\nIndex-File: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
1356 }
1357
1358 void pkgAcqMetaSig::Done(string Message,unsigned long long Size, HashStringList const &Hashes,
1359 pkgAcquire::MethodConfig *Cfg)
1360 {
1361 Item::Done(Message, Size, Hashes, Cfg);
1362
1363 string FileName = LookupTag(Message,"Filename");
1364 if (FileName.empty() == true)
1365 {
1366 Status = StatError;
1367 ErrorText = "Method gave a blank filename";
1368 return;
1369 }
1370
1371 if (FileName != DestFile)
1372 {
1373 // We have to copy it into place
1374 Local = true;
1375 Desc.URI = "copy:" + FileName;
1376 QueueURI(Desc);
1377 return;
1378 }
1379
1380 Complete = true;
1381
1382 // at this point pkgAcqMetaIndex takes over
1383 ExpectedAdditionalItems = 0;
1384
1385 // put the last known good file back on i-m-s hit (it will
1386 // be re-verified again)
1387 // Else do nothing, we have the new file in DestFile then
1388 if(StringToBool(LookupTag(Message,"IMS-Hit"),false) == true)
1389 Rename(LastGoodSig, DestFile);
1390
1391 // queue a pkgAcqMetaIndex to be verified against the sig we just retrieved
1392 new pkgAcqMetaIndex(Owner, MetaIndexURI, MetaIndexURIDesc,
1393 MetaIndexShortDesc, DestFile, IndexTargets,
1394 MetaIndexParser);
1395
1396 }
1397 /*}}}*/
1398 void pkgAcqMetaSig::Failed(string Message,pkgAcquire::MethodConfig *Cnf)/*{{{*/
1399 {
1400 string Final = _config->FindDir("Dir::State::lists") + URItoFileName(RealURI);
1401
1402 // at this point pkgAcqMetaIndex takes over
1403 ExpectedAdditionalItems = 0;
1404
1405 // if we get a network error we fail gracefully
1406 if(Status == StatTransientNetworkError)
1407 {
1408 Item::Failed(Message,Cnf);
1409 // move the sigfile back on transient network failures
1410 if(FileExists(LastGoodSig))
1411 Rename(LastGoodSig,Final);
1412
1413 // set the status back to , Item::Failed likes to reset it
1414 Status = pkgAcquire::Item::StatTransientNetworkError;
1415 return;
1416 }
1417
1418 // Delete any existing sigfile when the acquire failed
1419 unlink(Final.c_str());
1420
1421 // queue a pkgAcqMetaIndex with no sigfile
1422 new pkgAcqMetaIndex(Owner, MetaIndexURI, MetaIndexURIDesc, MetaIndexShortDesc,
1423 "", IndexTargets, MetaIndexParser);
1424
1425 if (Cnf->LocalOnly == true ||
1426 StringToBool(LookupTag(Message,"Transient-Failure"),false) == false)
1427 {
1428 // Ignore this
1429 Status = StatDone;
1430 Complete = false;
1431 Dequeue();
1432 return;
1433 }
1434
1435 Item::Failed(Message,Cnf);
1436 }
1437 /*}}}*/
1438 pkgAcqMetaIndex::pkgAcqMetaIndex(pkgAcquire *Owner, /*{{{*/
1439 string URI,string URIDesc,string ShortDesc,
1440 string SigFile,
1441 const vector<IndexTarget*>* IndexTargets,
1442 indexRecords* MetaIndexParser) :
1443 Item(Owner, HashStringList()), RealURI(URI), SigFile(SigFile), IndexTargets(IndexTargets),
1444 MetaIndexParser(MetaIndexParser), AuthPass(false), IMSHit(false)
1445 {
1446 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
1447 DestFile += URItoFileName(URI);
1448
1449 // Create the item
1450 Desc.Description = URIDesc;
1451 Desc.Owner = this;
1452 Desc.ShortDesc = ShortDesc;
1453 Desc.URI = URI;
1454
1455 // we expect more item
1456 ExpectedAdditionalItems = IndexTargets->size();
1457
1458 QueueURI(Desc);
1459 }
1460 /*}}}*/
1461 // pkgAcqMetaIndex::Custom600Headers - Insert custom request headers /*{{{*/
1462 // ---------------------------------------------------------------------
1463 /* The only header we use is the last-modified header. */
1464 string pkgAcqMetaIndex::Custom600Headers() const
1465 {
1466 string Final = _config->FindDir("Dir::State::lists");
1467 Final += URItoFileName(RealURI);
1468
1469 struct stat Buf;
1470 if (stat(Final.c_str(),&Buf) != 0)
1471 return "\nIndex-File: true";
1472
1473 return "\nIndex-File: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
1474 }
1475 /*}}}*/
1476 void pkgAcqMetaIndex::Done(string Message,unsigned long long Size,HashStringList const &Hashes, /*{{{*/
1477 pkgAcquire::MethodConfig *Cfg)
1478 {
1479 Item::Done(Message,Size,Hashes,Cfg);
1480
1481 // MetaIndexes are done in two passes: one to download the
1482 // metaindex with an appropriate method, and a second to verify it
1483 // with the gpgv method
1484
1485 if (AuthPass == true)
1486 {
1487 AuthDone(Message);
1488
1489 // all cool, move Release file into place
1490 Complete = true;
1491 }
1492 else
1493 {
1494 RetrievalDone(Message);
1495 if (!Complete)
1496 // Still more retrieving to do
1497 return;
1498
1499 if (SigFile == "")
1500 {
1501 // There was no signature file, so we are finished. Download
1502 // the indexes and do only hashsum verification if possible
1503 MetaIndexParser->Load(DestFile);
1504 QueueIndexes(false);
1505 }
1506 else
1507 {
1508 // FIXME: move this into pkgAcqMetaClearSig::Done on the next
1509 // ABI break
1510
1511 // if we expect a ClearTextSignature (InRelase), ensure that
1512 // this is what we get and if not fail to queue a
1513 // Release/Release.gpg, see #346386
1514 if (SigFile == DestFile && !StartsWithGPGClearTextSignature(DestFile))
1515 {
1516 Failed(Message, Cfg);
1517 return;
1518 }
1519
1520 // There was a signature file, so pass it to gpgv for
1521 // verification
1522 if (_config->FindB("Debug::pkgAcquire::Auth", false))
1523 std::cerr << "Metaindex acquired, queueing gpg verification ("
1524 << SigFile << "," << DestFile << ")\n";
1525 AuthPass = true;
1526 Desc.URI = "gpgv:" + SigFile;
1527 QueueURI(Desc);
1528 Mode = "gpgv";
1529 return;
1530 }
1531 }
1532
1533 if (Complete == true)
1534 {
1535 string FinalFile = _config->FindDir("Dir::State::lists");
1536 FinalFile += URItoFileName(RealURI);
1537 if (SigFile == DestFile)
1538 SigFile = FinalFile;
1539 Rename(DestFile,FinalFile);
1540 chmod(FinalFile.c_str(),0644);
1541 DestFile = FinalFile;
1542 }
1543 }
1544 /*}}}*/
1545 void pkgAcqMetaIndex::RetrievalDone(string Message) /*{{{*/
1546 {
1547 // We have just finished downloading a Release file (it is not
1548 // verified yet)
1549
1550 string FileName = LookupTag(Message,"Filename");
1551 if (FileName.empty() == true)
1552 {
1553 Status = StatError;
1554 ErrorText = "Method gave a blank filename";
1555 return;
1556 }
1557
1558 if (FileName != DestFile)
1559 {
1560 Local = true;
1561 Desc.URI = "copy:" + FileName;
1562 QueueURI(Desc);
1563 return;
1564 }
1565
1566 // make sure to verify against the right file on I-M-S hit
1567 IMSHit = StringToBool(LookupTag(Message,"IMS-Hit"),false);
1568 if(IMSHit)
1569 {
1570 string FinalFile = _config->FindDir("Dir::State::lists");
1571 FinalFile += URItoFileName(RealURI);
1572 if (SigFile == DestFile)
1573 {
1574 SigFile = FinalFile;
1575 // constructor of pkgAcqMetaClearSig moved it out of the way,
1576 // now move it back in on IMS hit for the 'old' file
1577 string const OldClearSig = DestFile + ".reverify";
1578 if (RealFileExists(OldClearSig) == true)
1579 Rename(OldClearSig, FinalFile);
1580 }
1581 DestFile = FinalFile;
1582 }
1583 Complete = true;
1584 }
1585 /*}}}*/
1586 void pkgAcqMetaIndex::AuthDone(string Message) /*{{{*/
1587 {
1588 // At this point, the gpgv method has succeeded, so there is a
1589 // valid signature from a key in the trusted keyring. We
1590 // perform additional verification of its contents, and use them
1591 // to verify the indexes we are about to download
1592
1593 if (!MetaIndexParser->Load(DestFile))
1594 {
1595 Status = StatAuthError;
1596 ErrorText = MetaIndexParser->ErrorText;
1597 return;
1598 }
1599
1600 if (!VerifyVendor(Message))
1601 {
1602 return;
1603 }
1604
1605 if (_config->FindB("Debug::pkgAcquire::Auth", false))
1606 std::cerr << "Signature verification succeeded: "
1607 << DestFile << std::endl;
1608
1609 // Download further indexes with verification
1610 QueueIndexes(true);
1611
1612 // is it a clearsigned MetaIndex file?
1613 if (DestFile == SigFile)
1614 return;
1615
1616 // Done, move signature file into position
1617 string VerifiedSigFile = _config->FindDir("Dir::State::lists") +
1618 URItoFileName(RealURI) + ".gpg";
1619 Rename(SigFile,VerifiedSigFile);
1620 chmod(VerifiedSigFile.c_str(),0644);
1621 }
1622 /*}}}*/
1623 void pkgAcqMetaIndex::QueueIndexes(bool verify) /*{{{*/
1624 {
1625 #if 0
1626 /* Reject invalid, existing Release files (LP: #346386) (Closes: #627642)
1627 * FIXME: Disabled; it breaks unsigned repositories without hashes */
1628 if (!verify && FileExists(DestFile) && !MetaIndexParser->Load(DestFile))
1629 {
1630 Status = StatError;
1631 ErrorText = MetaIndexParser->ErrorText;
1632 return;
1633 }
1634 #endif
1635 bool transInRelease = false;
1636 {
1637 std::vector<std::string> const keys = MetaIndexParser->MetaKeys();
1638 for (std::vector<std::string>::const_iterator k = keys.begin(); k != keys.end(); ++k)
1639 // FIXME: Feels wrong to check for hardcoded string here, but what should we do else…
1640 if (k->find("Translation-") != std::string::npos)
1641 {
1642 transInRelease = true;
1643 break;
1644 }
1645 }
1646
1647 // at this point the real Items are loaded in the fetcher
1648 ExpectedAdditionalItems = 0;
1649
1650 for (vector <IndexTarget*>::const_iterator Target = IndexTargets->begin();
1651 Target != IndexTargets->end();
1652 ++Target)
1653 {
1654 HashStringList ExpectedIndexHashes;
1655 const indexRecords::checkSum *Record = MetaIndexParser->Lookup((*Target)->MetaKey);
1656 bool compressedAvailable = false;
1657 if (Record == NULL)
1658 {
1659 if ((*Target)->IsOptional() == true)
1660 {
1661 std::vector<std::string> types = APT::Configuration::getCompressionTypes();
1662 for (std::vector<std::string>::const_iterator t = types.begin(); t != types.end(); ++t)
1663 if (MetaIndexParser->Exists((*Target)->MetaKey + "." + *t) == true)
1664 {
1665 compressedAvailable = true;
1666 break;
1667 }
1668 }
1669 else if (verify == true)
1670 {
1671 Status = StatAuthError;
1672 strprintf(ErrorText, _("Unable to find expected entry '%s' in Release file (Wrong sources.list entry or malformed file)"), (*Target)->MetaKey.c_str());
1673 return;
1674 }
1675 }
1676 else
1677 {
1678 ExpectedIndexHashes = Record->Hashes;
1679 if (_config->FindB("Debug::pkgAcquire::Auth", false))
1680 {
1681 std::cerr << "Queueing: " << (*Target)->URI << std::endl
1682 << "Expected Hash:" << std::endl;
1683 for (HashStringList::const_iterator hs = ExpectedIndexHashes.begin(); hs != ExpectedIndexHashes.end(); ++hs)
1684 std::cerr << "\t- " << hs->toStr() << std::endl;
1685 std::cerr << "For: " << Record->MetaKeyFilename << std::endl;
1686 }
1687 if (verify == true && ExpectedIndexHashes.empty() == true && (*Target)->IsOptional() == false)
1688 {
1689 Status = StatAuthError;
1690 strprintf(ErrorText, _("Unable to find hash sum for '%s' in Release file"), (*Target)->MetaKey.c_str());
1691 return;
1692 }
1693 }
1694
1695 if ((*Target)->IsOptional() == true)
1696 {
1697 if ((*Target)->IsSubIndex() == true)
1698 new pkgAcqSubIndex(Owner, (*Target)->URI, (*Target)->Description,
1699 (*Target)->ShortDesc, ExpectedIndexHashes);
1700 else if (transInRelease == false || Record != NULL || compressedAvailable == true)
1701 {
1702 if (_config->FindB("Acquire::PDiffs",true) == true && transInRelease == true &&
1703 MetaIndexParser->Exists((*Target)->MetaKey + ".diff/Index") == true)
1704 new pkgAcqDiffIndex(Owner, *Target, ExpectedIndexHashes, MetaIndexParser);
1705 else
1706 new pkgAcqIndexTrans(Owner, *Target, ExpectedIndexHashes, MetaIndexParser);
1707 }
1708 continue;
1709 }
1710
1711 /* Queue Packages file (either diff or full packages files, depending
1712 on the users option) - we also check if the PDiff Index file is listed
1713 in the Meta-Index file. Ideal would be if pkgAcqDiffIndex would test this
1714 instead, but passing the required info to it is to much hassle */
1715 if(_config->FindB("Acquire::PDiffs",true) == true && (verify == false ||
1716 MetaIndexParser->Exists((*Target)->MetaKey + ".diff/Index") == true))
1717 new pkgAcqDiffIndex(Owner, *Target, ExpectedIndexHashes, MetaIndexParser);
1718 else
1719 new pkgAcqIndex(Owner, *Target, ExpectedIndexHashes, MetaIndexParser);
1720 }
1721 }
1722 /*}}}*/
1723 bool pkgAcqMetaIndex::VerifyVendor(string Message) /*{{{*/
1724 {
1725 string::size_type pos;
1726
1727 // check for missing sigs (that where not fatal because otherwise we had
1728 // bombed earlier)
1729 string missingkeys;
1730 string msg = _("There is no public key available for the "
1731 "following key IDs:\n");
1732 pos = Message.find("NO_PUBKEY ");
1733 if (pos != std::string::npos)
1734 {
1735 string::size_type start = pos+strlen("NO_PUBKEY ");
1736 string Fingerprint = Message.substr(start, Message.find("\n")-start);
1737 missingkeys += (Fingerprint);
1738 }
1739 if(!missingkeys.empty())
1740 _error->Warning("%s", (msg + missingkeys).c_str());
1741
1742 string Transformed = MetaIndexParser->GetExpectedDist();
1743
1744 if (Transformed == "../project/experimental")
1745 {
1746 Transformed = "experimental";
1747 }
1748
1749 pos = Transformed.rfind('/');
1750 if (pos != string::npos)
1751 {
1752 Transformed = Transformed.substr(0, pos);
1753 }
1754
1755 if (Transformed == ".")
1756 {
1757 Transformed = "";
1758 }
1759
1760 if (_config->FindB("Acquire::Check-Valid-Until", true) == true &&
1761 MetaIndexParser->GetValidUntil() > 0) {
1762 time_t const invalid_since = time(NULL) - MetaIndexParser->GetValidUntil();
1763 if (invalid_since > 0)
1764 // TRANSLATOR: The first %s is the URL of the bad Release file, the second is
1765 // the time since then the file is invalid - formated in the same way as in
1766 // the download progress display (e.g. 7d 3h 42min 1s)
1767 return _error->Error(
1768 _("Release file for %s is expired (invalid since %s). "
1769 "Updates for this repository will not be applied."),
1770 RealURI.c_str(), TimeToStr(invalid_since).c_str());
1771 }
1772
1773 if (_config->FindB("Debug::pkgAcquire::Auth", false))
1774 {
1775 std::cerr << "Got Codename: " << MetaIndexParser->GetDist() << std::endl;
1776 std::cerr << "Expecting Dist: " << MetaIndexParser->GetExpectedDist() << std::endl;
1777 std::cerr << "Transformed Dist: " << Transformed << std::endl;
1778 }
1779
1780 if (MetaIndexParser->CheckDist(Transformed) == false)
1781 {
1782 // This might become fatal one day
1783 // Status = StatAuthError;
1784 // ErrorText = "Conflicting distribution; expected "
1785 // + MetaIndexParser->GetExpectedDist() + " but got "
1786 // + MetaIndexParser->GetDist();
1787 // return false;
1788 if (!Transformed.empty())
1789 {
1790 _error->Warning(_("Conflicting distribution: %s (expected %s but got %s)"),
1791 Desc.Description.c_str(),
1792 Transformed.c_str(),
1793 MetaIndexParser->GetDist().c_str());
1794 }
1795 }
1796
1797 return true;
1798 }
1799 /*}}}*/
1800 // pkgAcqMetaIndex::Failed - no Release file present or no signature file present /*{{{*/
1801 // ---------------------------------------------------------------------
1802 /* */
1803 void pkgAcqMetaIndex::Failed(string Message,pkgAcquire::MethodConfig * /*Cnf*/)
1804 {
1805 if (AuthPass == true)
1806 {
1807 // gpgv method failed, if we have a good signature
1808 string LastGoodSigFile = _config->FindDir("Dir::State::lists").append("partial/").append(URItoFileName(RealURI));
1809 if (DestFile != SigFile)
1810 LastGoodSigFile.append(".gpg");
1811 LastGoodSigFile.append(".reverify");
1812
1813 if(FileExists(LastGoodSigFile))
1814 {
1815 string VerifiedSigFile = _config->FindDir("Dir::State::lists") + URItoFileName(RealURI);
1816 if (DestFile != SigFile)
1817 VerifiedSigFile.append(".gpg");
1818 Rename(LastGoodSigFile, VerifiedSigFile);
1819 Status = StatTransientNetworkError;
1820 _error->Warning(_("An error occurred during the signature "
1821 "verification. The repository is not updated "
1822 "and the previous index files will be used. "
1823 "GPG error: %s: %s\n"),
1824 Desc.Description.c_str(),
1825 LookupTag(Message,"Message").c_str());
1826 RunScripts("APT::Update::Auth-Failure");
1827 return;
1828 } else if (LookupTag(Message,"Message").find("NODATA") != string::npos) {
1829 /* Invalid signature file, reject (LP: #346386) (Closes: #627642) */
1830 _error->Error(_("GPG error: %s: %s"),
1831 Desc.Description.c_str(),
1832 LookupTag(Message,"Message").c_str());
1833 return;
1834 } else {
1835 _error->Warning(_("GPG error: %s: %s"),
1836 Desc.Description.c_str(),
1837 LookupTag(Message,"Message").c_str());
1838 }
1839 // gpgv method failed
1840 ReportMirrorFailure("GPGFailure");
1841 }
1842
1843 /* Always move the meta index, even if gpgv failed. This ensures
1844 * that PackageFile objects are correctly filled in */
1845 if (FileExists(DestFile)) {
1846 string FinalFile = _config->FindDir("Dir::State::lists");
1847 FinalFile += URItoFileName(RealURI);
1848 /* InRelease files become Release files, otherwise
1849 * they would be considered as trusted later on */
1850 if (SigFile == DestFile) {
1851 RealURI = RealURI.replace(RealURI.rfind("InRelease"), 9,
1852 "Release");
1853 FinalFile = FinalFile.replace(FinalFile.rfind("InRelease"), 9,
1854 "Release");
1855 SigFile = FinalFile;
1856 }
1857 Rename(DestFile,FinalFile);
1858 chmod(FinalFile.c_str(),0644);
1859
1860 DestFile = FinalFile;
1861 }
1862
1863 // No Release file was present, or verification failed, so fall
1864 // back to queueing Packages files without verification
1865 QueueIndexes(false);
1866 }
1867 /*}}}*/
1868 pkgAcqMetaClearSig::pkgAcqMetaClearSig(pkgAcquire *Owner, /*{{{*/
1869 string const &URI, string const &URIDesc, string const &ShortDesc,
1870 string const &MetaIndexURI, string const &MetaIndexURIDesc, string const &MetaIndexShortDesc,
1871 string const &MetaSigURI, string const &MetaSigURIDesc, string const &MetaSigShortDesc,
1872 const vector<IndexTarget*>* IndexTargets,
1873 indexRecords* MetaIndexParser) :
1874 pkgAcqMetaIndex(Owner, URI, URIDesc, ShortDesc, "", IndexTargets, MetaIndexParser),
1875 MetaIndexURI(MetaIndexURI), MetaIndexURIDesc(MetaIndexURIDesc), MetaIndexShortDesc(MetaIndexShortDesc),
1876 MetaSigURI(MetaSigURI), MetaSigURIDesc(MetaSigURIDesc), MetaSigShortDesc(MetaSigShortDesc)
1877 {
1878 SigFile = DestFile;
1879
1880 // index targets + (worst case:) Release/Release.gpg
1881 ExpectedAdditionalItems = IndexTargets->size() + 2;
1882
1883
1884 // keep the old InRelease around in case of transistent network errors
1885 string const Final = _config->FindDir("Dir::State::lists") + URItoFileName(RealURI);
1886 if (RealFileExists(Final) == true)
1887 {
1888 string const LastGoodSig = DestFile + ".reverify";
1889 Rename(Final,LastGoodSig);
1890 }
1891 }
1892 /*}}}*/
1893 pkgAcqMetaClearSig::~pkgAcqMetaClearSig() /*{{{*/
1894 {
1895 // if the file was never queued undo file-changes done in the constructor
1896 if (QueueCounter == 1 && Status == StatIdle && FileSize == 0 && Complete == false)
1897 {
1898 string const Final = _config->FindDir("Dir::State::lists") + URItoFileName(RealURI);
1899 string const LastGoodSig = DestFile + ".reverify";
1900 if (RealFileExists(Final) == false && RealFileExists(LastGoodSig) == true)
1901 Rename(LastGoodSig, Final);
1902 }
1903 }
1904 /*}}}*/
1905 // pkgAcqMetaClearSig::Custom600Headers - Insert custom request headers /*{{{*/
1906 // ---------------------------------------------------------------------
1907 // FIXME: this can go away once the InRelease file is used widely
1908 string pkgAcqMetaClearSig::Custom600Headers() const
1909 {
1910 string Final = _config->FindDir("Dir::State::lists");
1911 Final += URItoFileName(RealURI);
1912
1913 struct stat Buf;
1914 if (stat(Final.c_str(),&Buf) != 0)
1915 {
1916 Final = DestFile + ".reverify";
1917 if (stat(Final.c_str(),&Buf) != 0)
1918 return "\nIndex-File: true\nFail-Ignore: true\n";
1919 }
1920
1921 return "\nIndex-File: true\nFail-Ignore: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
1922 }
1923 /*}}}*/
1924 void pkgAcqMetaClearSig::Failed(string Message,pkgAcquire::MethodConfig *Cnf) /*{{{*/
1925 {
1926 // we failed, we will not get additional items from this method
1927 ExpectedAdditionalItems = 0;
1928
1929 if (AuthPass == false)
1930 {
1931 // Remove the 'old' InRelease file if we try Release.gpg now as otherwise
1932 // the file will stay around and gives a false-auth impression (CVE-2012-0214)
1933 string FinalFile = _config->FindDir("Dir::State::lists");
1934 FinalFile.append(URItoFileName(RealURI));
1935 if (FileExists(FinalFile))
1936 unlink(FinalFile.c_str());
1937
1938 new pkgAcqMetaSig(Owner,
1939 MetaSigURI, MetaSigURIDesc, MetaSigShortDesc,
1940 MetaIndexURI, MetaIndexURIDesc, MetaIndexShortDesc,
1941 IndexTargets, MetaIndexParser);
1942 if (Cnf->LocalOnly == true ||
1943 StringToBool(LookupTag(Message, "Transient-Failure"), false) == false)
1944 Dequeue();
1945 }
1946 else
1947 pkgAcqMetaIndex::Failed(Message, Cnf);
1948 }
1949 /*}}}*/
1950 // AcqArchive::AcqArchive - Constructor /*{{{*/
1951 // ---------------------------------------------------------------------
1952 /* This just sets up the initial fetch environment and queues the first
1953 possibilitiy */
1954 pkgAcqArchive::pkgAcqArchive(pkgAcquire *Owner,pkgSourceList *Sources,
1955 pkgRecords *Recs,pkgCache::VerIterator const &Version,
1956 string &StoreFilename) :
1957 Item(Owner, HashStringList()), Version(Version), Sources(Sources), Recs(Recs),
1958 StoreFilename(StoreFilename), Vf(Version.FileList()),
1959 Trusted(false)
1960 {
1961 Retries = _config->FindI("Acquire::Retries",0);
1962
1963 if (Version.Arch() == 0)
1964 {
1965 _error->Error(_("I wasn't able to locate a file for the %s package. "
1966 "This might mean you need to manually fix this package. "
1967 "(due to missing arch)"),
1968 Version.ParentPkg().FullName().c_str());
1969 return;
1970 }
1971
1972 /* We need to find a filename to determine the extension. We make the
1973 assumption here that all the available sources for this version share
1974 the same extension.. */
1975 // Skip not source sources, they do not have file fields.
1976 for (; Vf.end() == false; ++Vf)
1977 {
1978 if ((Vf.File()->Flags & pkgCache::Flag::NotSource) != 0)
1979 continue;
1980 break;
1981 }
1982
1983 // Does not really matter here.. we are going to fail out below
1984 if (Vf.end() != true)
1985 {
1986 // If this fails to get a file name we will bomb out below.
1987 pkgRecords::Parser &Parse = Recs->Lookup(Vf);
1988 if (_error->PendingError() == true)
1989 return;
1990
1991 // Generate the final file name as: package_version_arch.foo
1992 StoreFilename = QuoteString(Version.ParentPkg().Name(),"_:") + '_' +
1993 QuoteString(Version.VerStr(),"_:") + '_' +
1994 QuoteString(Version.Arch(),"_:.") +
1995 "." + flExtension(Parse.FileName());
1996 }
1997
1998 // check if we have one trusted source for the package. if so, switch
1999 // to "TrustedOnly" mode - but only if not in AllowUnauthenticated mode
2000 bool const allowUnauth = _config->FindB("APT::Get::AllowUnauthenticated", false);
2001 bool const debugAuth = _config->FindB("Debug::pkgAcquire::Auth", false);
2002 bool seenUntrusted = false;
2003 for (pkgCache::VerFileIterator i = Version.FileList(); i.end() == false; ++i)
2004 {
2005 pkgIndexFile *Index;
2006 if (Sources->FindIndex(i.File(),Index) == false)
2007 continue;
2008
2009 if (debugAuth == true)
2010 std::cerr << "Checking index: " << Index->Describe()
2011 << "(Trusted=" << Index->IsTrusted() << ")" << std::endl;
2012
2013 if (Index->IsTrusted() == true)
2014 {
2015 Trusted = true;
2016 if (allowUnauth == false)
2017 break;
2018 }
2019 else
2020 seenUntrusted = true;
2021 }
2022
2023 // "allow-unauthenticated" restores apts old fetching behaviour
2024 // that means that e.g. unauthenticated file:// uris are higher
2025 // priority than authenticated http:// uris
2026 if (allowUnauth == true && seenUntrusted == true)
2027 Trusted = false;
2028
2029 // Select a source
2030 if (QueueNext() == false && _error->PendingError() == false)
2031 _error->Error(_("Can't find a source to download version '%s' of '%s'"),
2032 Version.VerStr(), Version.ParentPkg().FullName(false).c_str());
2033 }
2034 /*}}}*/
2035 // AcqArchive::QueueNext - Queue the next file source /*{{{*/
2036 // ---------------------------------------------------------------------
2037 /* This queues the next available file version for download. It checks if
2038 the archive is already available in the cache and stashs the MD5 for
2039 checking later. */
2040 bool pkgAcqArchive::QueueNext()
2041 {
2042 for (; Vf.end() == false; ++Vf)
2043 {
2044 // Ignore not source sources
2045 if ((Vf.File()->Flags & pkgCache::Flag::NotSource) != 0)
2046 continue;
2047
2048 // Try to cross match against the source list
2049 pkgIndexFile *Index;
2050 if (Sources->FindIndex(Vf.File(),Index) == false)
2051 continue;
2052
2053 // only try to get a trusted package from another source if that source
2054 // is also trusted
2055 if(Trusted && !Index->IsTrusted())
2056 continue;
2057
2058 // Grab the text package record
2059 pkgRecords::Parser &Parse = Recs->Lookup(Vf);
2060 if (_error->PendingError() == true)
2061 return false;
2062
2063 string PkgFile = Parse.FileName();
2064 ExpectedHashes = Parse.Hashes();
2065
2066 if (PkgFile.empty() == true)
2067 return _error->Error(_("The package index files are corrupted. No Filename: "
2068 "field for package %s."),
2069 Version.ParentPkg().Name());
2070
2071 Desc.URI = Index->ArchiveURI(PkgFile);
2072 Desc.Description = Index->ArchiveInfo(Version);
2073 Desc.Owner = this;
2074 Desc.ShortDesc = Version.ParentPkg().FullName(true);
2075
2076 // See if we already have the file. (Legacy filenames)
2077 FileSize = Version->Size;
2078 string FinalFile = _config->FindDir("Dir::Cache::Archives") + flNotDir(PkgFile);
2079 struct stat Buf;
2080 if (stat(FinalFile.c_str(),&Buf) == 0)
2081 {
2082 // Make sure the size matches
2083 if ((unsigned long long)Buf.st_size == Version->Size)
2084 {
2085 Complete = true;
2086 Local = true;
2087 Status = StatDone;
2088 StoreFilename = DestFile = FinalFile;
2089 return true;
2090 }
2091
2092 /* Hmm, we have a file and its size does not match, this means it is
2093 an old style mismatched arch */
2094 unlink(FinalFile.c_str());
2095 }
2096
2097 // Check it again using the new style output filenames
2098 FinalFile = _config->FindDir("Dir::Cache::Archives") + flNotDir(StoreFilename);
2099 if (stat(FinalFile.c_str(),&Buf) == 0)
2100 {
2101 // Make sure the size matches
2102 if ((unsigned long long)Buf.st_size == Version->Size)
2103 {
2104 Complete = true;
2105 Local = true;
2106 Status = StatDone;
2107 StoreFilename = DestFile = FinalFile;
2108 return true;
2109 }
2110
2111 /* Hmm, we have a file and its size does not match, this shouldn't
2112 happen.. */
2113 unlink(FinalFile.c_str());
2114 }
2115
2116 DestFile = _config->FindDir("Dir::Cache::Archives") + "partial/" + flNotDir(StoreFilename);
2117
2118 // Check the destination file
2119 if (stat(DestFile.c_str(),&Buf) == 0)
2120 {
2121 // Hmm, the partial file is too big, erase it
2122 if ((unsigned long long)Buf.st_size > Version->Size)
2123 unlink(DestFile.c_str());
2124 else
2125 PartialSize = Buf.st_size;
2126 }
2127
2128 // Disables download of archives - useful if no real installation follows,
2129 // e.g. if we are just interested in proposed installation order
2130 if (_config->FindB("Debug::pkgAcqArchive::NoQueue", false) == true)
2131 {
2132 Complete = true;
2133 Local = true;
2134 Status = StatDone;
2135 StoreFilename = DestFile = FinalFile;
2136 return true;
2137 }
2138
2139 // Create the item
2140 Local = false;
2141 QueueURI(Desc);
2142
2143 ++Vf;
2144 return true;
2145 }
2146 return false;
2147 }
2148 /*}}}*/
2149 // AcqArchive::Done - Finished fetching /*{{{*/
2150 // ---------------------------------------------------------------------
2151 /* */
2152 void pkgAcqArchive::Done(string Message,unsigned long long Size, HashStringList const &CalcHashes,
2153 pkgAcquire::MethodConfig *Cfg)
2154 {
2155 Item::Done(Message, Size, CalcHashes, Cfg);
2156
2157 // Check the size
2158 if (Size != Version->Size)
2159 {
2160 RenameOnError(SizeMismatch);
2161 return;
2162 }
2163
2164 // FIXME: could this empty() check impose *any* sort of security issue?
2165 if(ExpectedHashes.usable() && ExpectedHashes != CalcHashes)
2166 {
2167 RenameOnError(HashSumMismatch);
2168 printHashSumComparision(DestFile, ExpectedHashes, CalcHashes);
2169 return;
2170 }
2171
2172 // Grab the output filename
2173 string FileName = LookupTag(Message,"Filename");
2174 if (FileName.empty() == true)
2175 {
2176 Status = StatError;
2177 ErrorText = "Method gave a blank filename";
2178 return;
2179 }
2180
2181 Complete = true;
2182
2183 // Reference filename
2184 if (FileName != DestFile)
2185 {
2186 StoreFilename = DestFile = FileName;
2187 Local = true;
2188 return;
2189 }
2190
2191 // Done, move it into position
2192 string FinalFile = _config->FindDir("Dir::Cache::Archives");
2193 FinalFile += flNotDir(StoreFilename);
2194 Rename(DestFile,FinalFile);
2195
2196 StoreFilename = DestFile = FinalFile;
2197 Complete = true;
2198 }
2199 /*}}}*/
2200 // AcqArchive::Failed - Failure handler /*{{{*/
2201 // ---------------------------------------------------------------------
2202 /* Here we try other sources */
2203 void pkgAcqArchive::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
2204 {
2205 ErrorText = LookupTag(Message,"Message");
2206
2207 /* We don't really want to retry on failed media swaps, this prevents
2208 that. An interesting observation is that permanent failures are not
2209 recorded. */
2210 if (Cnf->Removable == true &&
2211 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
2212 {
2213 // Vf = Version.FileList();
2214 while (Vf.end() == false) ++Vf;
2215 StoreFilename = string();
2216 Item::Failed(Message,Cnf);
2217 return;
2218 }
2219
2220 if (QueueNext() == false)
2221 {
2222 // This is the retry counter
2223 if (Retries != 0 &&
2224 Cnf->LocalOnly == false &&
2225 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
2226 {
2227 Retries--;
2228 Vf = Version.FileList();
2229 if (QueueNext() == true)
2230 return;
2231 }
2232
2233 StoreFilename = string();
2234 Item::Failed(Message,Cnf);
2235 }
2236 }
2237 /*}}}*/
2238 // AcqArchive::IsTrusted - Determine whether this archive comes from a trusted source /*{{{*/
2239 // ---------------------------------------------------------------------
2240 APT_PURE bool pkgAcqArchive::IsTrusted() const
2241 {
2242 return Trusted;
2243 }
2244 /*}}}*/
2245 // AcqArchive::Finished - Fetching has finished, tidy up /*{{{*/
2246 // ---------------------------------------------------------------------
2247 /* */
2248 void pkgAcqArchive::Finished()
2249 {
2250 if (Status == pkgAcquire::Item::StatDone &&
2251 Complete == true)
2252 return;
2253 StoreFilename = string();
2254 }
2255 /*}}}*/
2256 // AcqFile::pkgAcqFile - Constructor /*{{{*/
2257 // ---------------------------------------------------------------------
2258 /* The file is added to the queue */
2259 pkgAcqFile::pkgAcqFile(pkgAcquire *Owner,string URI, HashStringList const &Hashes,
2260 unsigned long long Size,string Dsc,string ShortDesc,
2261 const string &DestDir, const string &DestFilename,
2262 bool IsIndexFile) :
2263 Item(Owner, Hashes), IsIndexFile(IsIndexFile)
2264 {
2265 Retries = _config->FindI("Acquire::Retries",0);
2266
2267 if(!DestFilename.empty())
2268 DestFile = DestFilename;
2269 else if(!DestDir.empty())
2270 DestFile = DestDir + "/" + flNotDir(URI);
2271 else
2272 DestFile = flNotDir(URI);
2273
2274 // Create the item
2275 Desc.URI = URI;
2276 Desc.Description = Dsc;
2277 Desc.Owner = this;
2278
2279 // Set the short description to the archive component
2280 Desc.ShortDesc = ShortDesc;
2281
2282 // Get the transfer sizes
2283 FileSize = Size;
2284 struct stat Buf;
2285 if (stat(DestFile.c_str(),&Buf) == 0)
2286 {
2287 // Hmm, the partial file is too big, erase it
2288 if ((Size > 0) && (unsigned long long)Buf.st_size > Size)
2289 unlink(DestFile.c_str());
2290 else
2291 PartialSize = Buf.st_size;
2292 }
2293
2294 QueueURI(Desc);
2295 }
2296 /*}}}*/
2297 // AcqFile::Done - Item downloaded OK /*{{{*/
2298 // ---------------------------------------------------------------------
2299 /* */
2300 void pkgAcqFile::Done(string Message,unsigned long long Size,HashStringList const &CalcHashes,
2301 pkgAcquire::MethodConfig *Cnf)
2302 {
2303 Item::Done(Message,Size,CalcHashes,Cnf);
2304
2305 // Check the hash
2306 if(ExpectedHashes.usable() && ExpectedHashes != CalcHashes)
2307 {
2308 RenameOnError(HashSumMismatch);
2309 printHashSumComparision(DestFile, ExpectedHashes, CalcHashes);
2310 return;
2311 }
2312
2313 string FileName = LookupTag(Message,"Filename");
2314 if (FileName.empty() == true)
2315 {
2316 Status = StatError;
2317 ErrorText = "Method gave a blank filename";
2318 return;
2319 }
2320
2321 Complete = true;
2322
2323 // The files timestamp matches
2324 if (StringToBool(LookupTag(Message,"IMS-Hit"),false) == true)
2325 return;
2326
2327 // We have to copy it into place
2328 if (FileName != DestFile)
2329 {
2330 Local = true;
2331 if (_config->FindB("Acquire::Source-Symlinks",true) == false ||
2332 Cnf->Removable == true)
2333 {
2334 Desc.URI = "copy:" + FileName;
2335 QueueURI(Desc);
2336 return;
2337 }
2338
2339 // Erase the file if it is a symlink so we can overwrite it
2340 struct stat St;
2341 if (lstat(DestFile.c_str(),&St) == 0)
2342 {
2343 if (S_ISLNK(St.st_mode) != 0)
2344 unlink(DestFile.c_str());
2345 }
2346
2347 // Symlink the file
2348 if (symlink(FileName.c_str(),DestFile.c_str()) != 0)
2349 {
2350 ErrorText = "Link to " + DestFile + " failure ";
2351 Status = StatError;
2352 Complete = false;
2353 }
2354 }
2355 }
2356 /*}}}*/
2357 // AcqFile::Failed - Failure handler /*{{{*/
2358 // ---------------------------------------------------------------------
2359 /* Here we try other sources */
2360 void pkgAcqFile::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
2361 {
2362 ErrorText = LookupTag(Message,"Message");
2363
2364 // This is the retry counter
2365 if (Retries != 0 &&
2366 Cnf->LocalOnly == false &&
2367 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
2368 {
2369 Retries--;
2370 QueueURI(Desc);
2371 return;
2372 }
2373
2374 Item::Failed(Message,Cnf);
2375 }
2376 /*}}}*/
2377 // AcqIndex::Custom600Headers - Insert custom request headers /*{{{*/
2378 // ---------------------------------------------------------------------
2379 /* The only header we use is the last-modified header. */
2380 string pkgAcqFile::Custom600Headers() const
2381 {
2382 if (IsIndexFile)
2383 return "\nIndex-File: true";
2384 return "";
2385 }
2386 /*}}}*/