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