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