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