]> git.saurik.com Git - apt.git/blame - apt-pkg/acquire-item.cc
Merge branch 'debian/sid' into debian/experimental
[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
955 Init(URI, URIDesc, ShortDesc);
956}
957pkgAcqIndex::pkgAcqIndex(pkgAcquire *Owner, IndexTarget const *Target,
e110d7bf
MV
958 HashStringList const &ExpectedHash,
959 indexRecords *MetaIndexParser)
0b58b3f8
MV
960 : pkgAcqBaseIndex(Owner, Target, ExpectedHash, MetaIndexParser),
961 RealURI(Target->URI)
5d885723
DK
962{
963 // autoselect the compression method
964 std::vector<std::string> types = APT::Configuration::getCompressionTypes();
965 CompressionExtension = "";
b3501edb 966 if (ExpectedHashes.usable())
5d885723
DK
967 {
968 for (std::vector<std::string>::const_iterator t = types.begin(); t != types.end(); ++t)
969 if (*t == "uncompressed" || MetaIndexParser->Exists(string(Target->MetaKey).append(".").append(*t)) == true)
970 CompressionExtension.append(*t).append(" ");
971 }
972 else
973 {
974 for (std::vector<std::string>::const_iterator t = types.begin(); t != types.end(); ++t)
975 CompressionExtension.append(*t).append(" ");
976 }
977 if (CompressionExtension.empty() == false)
978 CompressionExtension.erase(CompressionExtension.end()-1);
979
980 Init(Target->URI, Target->Description, Target->ShortDesc);
981}
982 /*}}}*/
983// AcqIndex::Init - defered Constructor /*{{{*/
984void pkgAcqIndex::Init(string const &URI, string const &URIDesc, string const &ShortDesc) {
8b89e57f 985 Decompression = false;
bfd22fc0 986 Erase = false;
13e8426f 987
0a8a80e5 988 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
b2e465d6 989 DestFile += URItoFileName(URI);
8267fe24 990
5d885723 991 std::string const comprExt = CompressionExtension.substr(0, CompressionExtension.find(' '));
b11f9599 992 std::string MetaKey;
5d885723 993 if (comprExt == "uncompressed")
b11f9599 994 {
5d885723 995 Desc.URI = URI;
e39698a4
MV
996 if(Target)
997 MetaKey = string(Target->MetaKey);
b11f9599 998 }
5d885723 999 else
b11f9599 1000 {
5d885723 1001 Desc.URI = URI + '.' + comprExt;
e39698a4
MV
1002 if(Target)
1003 MetaKey = string(Target->MetaKey) + '.' + comprExt;
b11f9599
MV
1004 }
1005
1006 // load the filesize
e39698a4
MV
1007 if(MetaIndexParser)
1008 {
1009 indexRecords::checkSum *Record = MetaIndexParser->Lookup(MetaKey);
1010 if(Record)
1011 FileSize = Record->Size;
59194959
MV
1012
1013 InitByHashIfNeeded(MetaKey);
e39698a4 1014 }
b3d44315 1015
b2e465d6 1016 Desc.Description = URIDesc;
8267fe24 1017 Desc.Owner = this;
b2e465d6 1018 Desc.ShortDesc = ShortDesc;
5d885723 1019
8267fe24 1020 QueueURI(Desc);
0118833a
AL
1021}
1022 /*}}}*/
59194959
MV
1023// AcqIndex::AdjustForByHash - modify URI for by-hash support /*{{{*/
1024// ---------------------------------------------------------------------
1025/* */
1026void pkgAcqIndex::InitByHashIfNeeded(const std::string MetaKey)
1027{
1028 // TODO:
1029 // - (maybe?) add support for by-hash into the sources.list as flag
1030 // - make apt-ftparchive generate the hashes (and expire?)
1031 std::string HostKnob = "APT::Acquire::" + ::URI(Desc.URI).Host + "::By-Hash";
1032 if(_config->FindB("APT::Acquire::By-Hash", false) == true ||
1033 _config->FindB(HostKnob, false) == true ||
1034 MetaIndexParser->GetSupportsAcquireByHash())
1035 {
1036 indexRecords::checkSum *Record = MetaIndexParser->Lookup(MetaKey);
1037 if(Record)
1038 {
1039 // FIXME: should we really use the best hash here? or a fixed one?
1040 const HashString *TargetHash = Record->Hashes.find("");
1041 std::string ByHash = "/by-hash/" + TargetHash->HashType() + "/" + TargetHash->HashValue();
1042 size_t trailing_slash = Desc.URI.find_last_of("/");
1043 Desc.URI = Desc.URI.replace(
1044 trailing_slash,
1045 Desc.URI.substr(trailing_slash+1).size()+1,
1046 ByHash);
1047 } else {
1048 _error->Warning(
1049 "Fetching ByHash requested but can not find record for %s",
1050 MetaKey.c_str());
1051 }
1052 }
1053}
1054 /*}}}*/
0a8a80e5 1055// AcqIndex::Custom600Headers - Insert custom request headers /*{{{*/
0118833a 1056// ---------------------------------------------------------------------
0a8a80e5 1057/* The only header we use is the last-modified header. */
b3501edb 1058string pkgAcqIndex::Custom600Headers() const
0118833a 1059{
0a8a80e5 1060 string Final = _config->FindDir("Dir::State::lists");
b2e465d6 1061 Final += URItoFileName(RealURI);
01606def 1062 if (_config->FindB("Acquire::GzipIndexes",false))
1063 Final += ".gz";
0a8a80e5 1064
97b65b10
MV
1065 string msg = "\nIndex-File: true";
1066 // FIXME: this really should use "IndexTarget::IsOptional()" but that
1067 // seems to be difficult without breaking ABI
1068 if (ShortDesc().find("Translation") != 0)
1069 msg += "\nFail-Ignore: true";
0a8a80e5 1070 struct stat Buf;
3a1f49c4 1071 if (stat(Final.c_str(),&Buf) == 0)
97b65b10
MV
1072 msg += "\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
1073
1074 return msg;
0118833a
AL
1075}
1076 /*}}}*/
92fcbfc1 1077void pkgAcqIndex::Failed(string Message,pkgAcquire::MethodConfig *Cnf) /*{{{*/
debc84b2 1078{
5d885723
DK
1079 size_t const nextExt = CompressionExtension.find(' ');
1080 if (nextExt != std::string::npos)
e85b4cd5 1081 {
5d885723
DK
1082 CompressionExtension = CompressionExtension.substr(nextExt+1);
1083 Init(RealURI, Desc.Description, Desc.ShortDesc);
6abe2699 1084 return;
0d7a243d
EL
1085 }
1086
17ff0930 1087 // on decompression failure, remove bad versions in partial/
5d885723 1088 if (Decompression && Erase) {
17ff0930 1089 string s = _config->FindDir("Dir::State::lists") + "partial/";
5d885723 1090 s.append(URItoFileName(RealURI));
17ff0930 1091 unlink(s.c_str());
debc84b2
MZ
1092 }
1093
debc84b2
MZ
1094 Item::Failed(Message,Cnf);
1095}
92fcbfc1 1096 /*}}}*/
8b89e57f
AL
1097// AcqIndex::Done - Finished a fetch /*{{{*/
1098// ---------------------------------------------------------------------
1099/* This goes through a number of states.. On the initial fetch the
1100 method could possibly return an alternate filename which points
1101 to the uncompressed version of the file. If this is so the file
1102 is copied into the partial directory. In all other cases the file
1103 is decompressed with a gzip uri. */
b3501edb 1104void pkgAcqIndex::Done(string Message,unsigned long long Size,HashStringList const &Hashes,
459681d3 1105 pkgAcquire::MethodConfig *Cfg)
8b89e57f 1106{
b3501edb 1107 Item::Done(Message,Size,Hashes,Cfg);
8b89e57f
AL
1108
1109 if (Decompression == true)
1110 {
b3501edb 1111 if (ExpectedHashes.usable() && ExpectedHashes != Hashes)
b3d44315 1112 {
3c8030a4 1113 RenameOnError(HashSumMismatch);
b3501edb 1114 printHashSumComparision(RealURI, ExpectedHashes, Hashes);
b3d44315
MV
1115 return;
1116 }
0901c5d0 1117
19818b65
MV
1118 // FIXME: this can go away once we only ever download stuff that
1119 // has a valid hash and we never do GET based probing
1120 //
1121 /* Always verify the index file for correctness (all indexes must
1122 * have a Package field) (LP: #346386) (Closes: #627642)
1123 */
1124 FileFd fd(DestFile, FileFd::ReadOnly);
1125 // Only test for correctness if the file is not empty (empty is ok)
1126 if (fd.FileSize() > 0)
0901c5d0 1127 {
19818b65
MV
1128 pkgTagSection sec;
1129 pkgTagFile tag(&fd);
1130
1131 // all our current indexes have a field 'Package' in each section
1132 if (_error->PendingError() == true || tag.Step(sec) == false || sec.Exists("Package") == false)
1133 {
1134 RenameOnError(InvalidFormat);
1135 return;
a0c3110e 1136 }
0901c5d0
JAK
1137 }
1138
8b89e57f
AL
1139 // Done, move it into position
1140 string FinalFile = _config->FindDir("Dir::State::lists");
b2e465d6 1141 FinalFile += URItoFileName(RealURI);
8b89e57f 1142 Rename(DestFile,FinalFile);
7a3c2ab0 1143 chmod(FinalFile.c_str(),0644);
b11f9599 1144
7a7fa5f0
AL
1145 /* We restore the original name to DestFile so that the clean operation
1146 will work OK */
1147 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
b2e465d6 1148 DestFile += URItoFileName(RealURI);
7a7fa5f0 1149
bfd22fc0
AL
1150 // Remove the compressed version.
1151 if (Erase == true)
bfd22fc0 1152 unlink(DestFile.c_str());
b11f9599 1153
8b89e57f
AL
1154 return;
1155 }
bfd22fc0
AL
1156
1157 Erase = false;
8267fe24 1158 Complete = true;
bfd22fc0 1159
8b89e57f
AL
1160 // Handle the unzipd case
1161 string FileName = LookupTag(Message,"Alt-Filename");
1162 if (FileName.empty() == false)
1163 {
1164 // The files timestamp matches
1165 if (StringToBool(LookupTag(Message,"Alt-IMS-Hit"),false) == true)
1166 return;
8b89e57f 1167 Decompression = true;
a6568219 1168 Local = true;
8b89e57f 1169 DestFile += ".decomp";
8267fe24
AL
1170 Desc.URI = "copy:" + FileName;
1171 QueueURI(Desc);
b98f2859 1172 Mode = "copy";
8b89e57f
AL
1173 return;
1174 }
1175
1176 FileName = LookupTag(Message,"Filename");
1177 if (FileName.empty() == true)
1178 {
1179 Status = StatError;
1180 ErrorText = "Method gave a blank filename";
1181 }
5d885723
DK
1182
1183 std::string const compExt = CompressionExtension.substr(0, CompressionExtension.find(' '));
0b9032b1 1184
8b89e57f 1185 // The files timestamp matches
0b9032b1 1186 if (StringToBool(LookupTag(Message,"IMS-Hit"),false) == true) {
1187 if (_config->FindB("Acquire::GzipIndexes",false) && compExt == "gz")
1188 // Update DestFile for .gz suffix so that the clean operation keeps it
1189 DestFile += ".gz";
8b89e57f 1190 return;
0b9032b1 1191 }
bfd22fc0
AL
1192
1193 if (FileName == DestFile)
1194 Erase = true;
8267fe24 1195 else
a6568219 1196 Local = true;
8b89e57f 1197
e85b4cd5
DK
1198 string decompProg;
1199
bb109d0b 1200 // If we enable compressed indexes and already have gzip, keep it
7aeee365 1201 if (_config->FindB("Acquire::GzipIndexes",false) && compExt == "gz" && !Local) {
bb109d0b 1202 string FinalFile = _config->FindDir("Dir::State::lists");
1203 FinalFile += URItoFileName(RealURI) + ".gz";
bb109d0b 1204 Rename(DestFile,FinalFile);
1205 chmod(FinalFile.c_str(),0644);
1206
1207 // Update DestFile for .gz suffix so that the clean operation keeps it
1208 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
1209 DestFile += URItoFileName(RealURI) + ".gz";
1210 return;
1211 }
1212
e85b4cd5
DK
1213 // get the binary name for your used compression type
1214 decompProg = _config->Find(string("Acquire::CompressionTypes::").append(compExt),"");
1215 if(decompProg.empty() == false);
5d885723 1216 else if(compExt == "uncompressed")
0d7a243d 1217 decompProg = "copy";
debc84b2
MZ
1218 else {
1219 _error->Error("Unsupported extension: %s", compExt.c_str());
1220 return;
1221 }
1222
8b89e57f
AL
1223 Decompression = true;
1224 DestFile += ".decomp";
e85b4cd5 1225 Desc.URI = decompProg + ":" + FileName;
8267fe24 1226 QueueURI(Desc);
70e0c168
MV
1227
1228 // FIXME: this points to a c++ string that goes out of scope
e85b4cd5 1229 Mode = decompProg.c_str();
8b89e57f 1230}
92fcbfc1 1231 /*}}}*/
a52f938b
OS
1232// AcqIndexTrans::pkgAcqIndexTrans - Constructor /*{{{*/
1233// ---------------------------------------------------------------------
1234/* The Translation file is added to the queue */
1235pkgAcqIndexTrans::pkgAcqIndexTrans(pkgAcquire *Owner,
495e5cb2 1236 string URI,string URIDesc,string ShortDesc)
b3501edb 1237 : pkgAcqIndex(Owner, URI, URIDesc, ShortDesc, HashStringList(), "")
a52f938b 1238{
ab53c018 1239}
fa3b260f 1240pkgAcqIndexTrans::pkgAcqIndexTrans(pkgAcquire *Owner, IndexTarget const * const Target,
b3501edb
DK
1241 HashStringList const &ExpectedHashes, indexRecords *MetaIndexParser)
1242 : pkgAcqIndex(Owner, Target, ExpectedHashes, MetaIndexParser)
ab53c018 1243{
1dca8dc5
MV
1244 // load the filesize
1245 indexRecords::checkSum *Record = MetaIndexParser->Lookup(string(Target->MetaKey));
1246 if(Record)
1247 FileSize = Record->Size;
963b16dc
MV
1248}
1249 /*}}}*/
1250// AcqIndexTrans::Custom600Headers - Insert custom request headers /*{{{*/
1251// ---------------------------------------------------------------------
b3501edb 1252string pkgAcqIndexTrans::Custom600Headers() const
963b16dc 1253{
c91d9a63
DK
1254 string Final = _config->FindDir("Dir::State::lists");
1255 Final += URItoFileName(RealURI);
1256
1257 struct stat Buf;
1258 if (stat(Final.c_str(),&Buf) != 0)
a3f7fff8
MV
1259 return "\nFail-Ignore: true\nIndex-File: true";
1260 return "\nFail-Ignore: true\nIndex-File: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
a52f938b 1261}
a52f938b
OS
1262 /*}}}*/
1263// AcqIndexTrans::Failed - Silence failure messages for missing files /*{{{*/
1264// ---------------------------------------------------------------------
1265/* */
1266void pkgAcqIndexTrans::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
1267{
5d885723
DK
1268 size_t const nextExt = CompressionExtension.find(' ');
1269 if (nextExt != std::string::npos)
1270 {
1271 CompressionExtension = CompressionExtension.substr(nextExt+1);
1272 Init(RealURI, Desc.Description, Desc.ShortDesc);
1273 Status = StatIdle;
1274 return;
1275 }
1276
a52f938b
OS
1277 if (Cnf->LocalOnly == true ||
1278 StringToBool(LookupTag(Message,"Transient-Failure"),false) == false)
1279 {
1280 // Ignore this
1281 Status = StatDone;
1282 Complete = false;
1283 Dequeue();
1284 return;
1285 }
5d885723 1286
a52f938b
OS
1287 Item::Failed(Message,Cnf);
1288}
1289 /*}}}*/
92fcbfc1 1290pkgAcqMetaSig::pkgAcqMetaSig(pkgAcquire *Owner, /*{{{*/
b3d44315
MV
1291 string URI,string URIDesc,string ShortDesc,
1292 string MetaIndexURI, string MetaIndexURIDesc,
1293 string MetaIndexShortDesc,
1294 const vector<IndexTarget*>* IndexTargets,
1295 indexRecords* MetaIndexParser) :
fa3b260f 1296 Item(Owner, HashStringList()), RealURI(URI), MetaIndexURI(MetaIndexURI),
46e00f9d
MV
1297 MetaIndexURIDesc(MetaIndexURIDesc), MetaIndexShortDesc(MetaIndexShortDesc),
1298 MetaIndexParser(MetaIndexParser), IndexTargets(IndexTargets)
0118833a 1299{
0a8a80e5 1300 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
b2e465d6 1301 DestFile += URItoFileName(URI);
b3d44315 1302
47eb38f4
MV
1303 // remove any partial downloaded sig-file in partial/.
1304 // it may confuse proxies and is too small to warrant a
1305 // partial download anyway
f6237efd
MV
1306 unlink(DestFile.c_str());
1307
8267fe24 1308 // Create the item
b2e465d6 1309 Desc.Description = URIDesc;
8267fe24 1310 Desc.Owner = this;
b3d44315
MV
1311 Desc.ShortDesc = ShortDesc;
1312 Desc.URI = URI;
b3d44315
MV
1313
1314 string Final = _config->FindDir("Dir::State::lists");
1315 Final += URItoFileName(RealURI);
ffcccd62 1316 if (RealFileExists(Final) == true)
b3d44315 1317 {
ef942597 1318 // File was already in place. It needs to be re-downloaded/verified
1e3f4083 1319 // because Release might have changed, we do give it a different
ef942597
MV
1320 // name than DestFile because otherwise the http method will
1321 // send If-Range requests and there are too many broken servers
1322 // out there that do not understand them
1323 LastGoodSig = DestFile+".reverify";
1324 Rename(Final,LastGoodSig);
b3d44315 1325 }
8267fe24 1326
d0cfa8ad
MV
1327 // we expect the indextargets + one additional Release file
1328 ExpectedAdditionalItems = IndexTargets->size() + 1;
1329
8267fe24 1330 QueueURI(Desc);
ffcccd62
DK
1331}
1332 /*}}}*/
1333pkgAcqMetaSig::~pkgAcqMetaSig() /*{{{*/
1334{
1335 // if the file was never queued undo file-changes done in the constructor
1336 if (QueueCounter == 1 && Status == StatIdle && FileSize == 0 && Complete == false &&
1337 LastGoodSig.empty() == false)
1338 {
1339 string const Final = _config->FindDir("Dir::State::lists") + URItoFileName(RealURI);
1340 if (RealFileExists(Final) == false && RealFileExists(LastGoodSig) == true)
1341 Rename(LastGoodSig, Final);
1342 }
1343
0118833a
AL
1344}
1345 /*}}}*/
b3d44315 1346// pkgAcqMetaSig::Custom600Headers - Insert custom request headers /*{{{*/
0118833a 1347// ---------------------------------------------------------------------
0a8a80e5 1348/* The only header we use is the last-modified header. */
b3501edb 1349string pkgAcqMetaSig::Custom600Headers() const
0118833a 1350{
0a8a80e5 1351 struct stat Buf;
ef942597 1352 if (stat(LastGoodSig.c_str(),&Buf) != 0)
a72ace20 1353 return "\nIndex-File: true";
a789b983 1354
a72ace20 1355 return "\nIndex-File: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
0118833a 1356}
b3d44315 1357
b3501edb 1358void pkgAcqMetaSig::Done(string Message,unsigned long long Size, HashStringList const &Hashes,
b3d44315 1359 pkgAcquire::MethodConfig *Cfg)
c88edf1d 1360{
b3501edb 1361 Item::Done(Message, Size, Hashes, Cfg);
c88edf1d
AL
1362
1363 string FileName = LookupTag(Message,"Filename");
1364 if (FileName.empty() == true)
1365 {
1366 Status = StatError;
1367 ErrorText = "Method gave a blank filename";
8b89e57f 1368 return;
c88edf1d 1369 }
8b89e57f 1370
c88edf1d
AL
1371 if (FileName != DestFile)
1372 {
b3d44315 1373 // We have to copy it into place
a6568219 1374 Local = true;
8267fe24
AL
1375 Desc.URI = "copy:" + FileName;
1376 QueueURI(Desc);
c88edf1d
AL
1377 return;
1378 }
b3d44315
MV
1379
1380 Complete = true;
1381
d0cfa8ad
MV
1382 // at this point pkgAcqMetaIndex takes over
1383 ExpectedAdditionalItems = 0;
1384
ef942597
MV
1385 // put the last known good file back on i-m-s hit (it will
1386 // be re-verified again)
1387 // Else do nothing, we have the new file in DestFile then
1388 if(StringToBool(LookupTag(Message,"IMS-Hit"),false) == true)
1389 Rename(LastGoodSig, DestFile);
1390
b3d44315 1391 // queue a pkgAcqMetaIndex to be verified against the sig we just retrieved
fce72602
MV
1392 new pkgAcqMetaIndex(Owner, MetaIndexURI, MetaIndexURIDesc,
1393 MetaIndexShortDesc, DestFile, IndexTargets,
1394 MetaIndexParser);
b3d44315 1395
c88edf1d
AL
1396}
1397 /*}}}*/
92fcbfc1 1398void pkgAcqMetaSig::Failed(string Message,pkgAcquire::MethodConfig *Cnf)/*{{{*/
681d76d0 1399{
47eb38f4 1400 string Final = _config->FindDir("Dir::State::lists") + URItoFileName(RealURI);
a789b983 1401
d0cfa8ad
MV
1402 // at this point pkgAcqMetaIndex takes over
1403 ExpectedAdditionalItems = 0;
1404
75dd8af1 1405 // if we get a network error we fail gracefully
7e5f33eb
MV
1406 if(Status == StatTransientNetworkError)
1407 {
24057ad6 1408 Item::Failed(Message,Cnf);
484dbb81 1409 // move the sigfile back on transient network failures
7730e095 1410 if(FileExists(LastGoodSig))
ef942597 1411 Rename(LastGoodSig,Final);
7e5f33eb
MV
1412
1413 // set the status back to , Item::Failed likes to reset it
1414 Status = pkgAcquire::Item::StatTransientNetworkError;
24057ad6
MV
1415 return;
1416 }
1417
75dd8af1 1418 // Delete any existing sigfile when the acquire failed
75dd8af1
MV
1419 unlink(Final.c_str());
1420
b3d44315
MV
1421 // queue a pkgAcqMetaIndex with no sigfile
1422 new pkgAcqMetaIndex(Owner, MetaIndexURI, MetaIndexURIDesc, MetaIndexShortDesc,
1423 "", IndexTargets, MetaIndexParser);
1424
681d76d0
AL
1425 if (Cnf->LocalOnly == true ||
1426 StringToBool(LookupTag(Message,"Transient-Failure"),false) == false)
1427 {
2b154e53
AL
1428 // Ignore this
1429 Status = StatDone;
1430 Complete = false;
681d76d0
AL
1431 Dequeue();
1432 return;
1433 }
1434
1435 Item::Failed(Message,Cnf);
1436}
92fcbfc1
DK
1437 /*}}}*/
1438pkgAcqMetaIndex::pkgAcqMetaIndex(pkgAcquire *Owner, /*{{{*/
b3d44315
MV
1439 string URI,string URIDesc,string ShortDesc,
1440 string SigFile,
fa3b260f 1441 const vector<IndexTarget*>* IndexTargets,
b3d44315 1442 indexRecords* MetaIndexParser) :
fa3b260f 1443 Item(Owner, HashStringList()), RealURI(URI), SigFile(SigFile), IndexTargets(IndexTargets),
21fd1746 1444 MetaIndexParser(MetaIndexParser), AuthPass(false), IMSHit(false)
b3d44315 1445{
b3d44315
MV
1446 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
1447 DestFile += URItoFileName(URI);
1448
1449 // Create the item
1450 Desc.Description = URIDesc;
1451 Desc.Owner = this;
1452 Desc.ShortDesc = ShortDesc;
1453 Desc.URI = URI;
1454
d0cfa8ad
MV
1455 // we expect more item
1456 ExpectedAdditionalItems = IndexTargets->size();
1457
b3d44315
MV
1458 QueueURI(Desc);
1459}
b3d44315
MV
1460 /*}}}*/
1461// pkgAcqMetaIndex::Custom600Headers - Insert custom request headers /*{{{*/
1462// ---------------------------------------------------------------------
1463/* The only header we use is the last-modified header. */
b3501edb 1464string pkgAcqMetaIndex::Custom600Headers() const
b3d44315
MV
1465{
1466 string Final = _config->FindDir("Dir::State::lists");
1467 Final += URItoFileName(RealURI);
1468
1469 struct stat Buf;
1470 if (stat(Final.c_str(),&Buf) != 0)
1471 return "\nIndex-File: true";
1472
1473 return "\nIndex-File: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
1474}
92fcbfc1 1475 /*}}}*/
b3501edb 1476void pkgAcqMetaIndex::Done(string Message,unsigned long long Size,HashStringList const &Hashes, /*{{{*/
b3d44315
MV
1477 pkgAcquire::MethodConfig *Cfg)
1478{
b3501edb 1479 Item::Done(Message,Size,Hashes,Cfg);
b3d44315
MV
1480
1481 // MetaIndexes are done in two passes: one to download the
1482 // metaindex with an appropriate method, and a second to verify it
1483 // with the gpgv method
1484
1485 if (AuthPass == true)
1486 {
1487 AuthDone(Message);
fce72602
MV
1488
1489 // all cool, move Release file into place
1490 Complete = true;
b3d44315
MV
1491 }
1492 else
1493 {
1494 RetrievalDone(Message);
1495 if (!Complete)
1496 // Still more retrieving to do
1497 return;
1498
1499 if (SigFile == "")
1500 {
1501 // There was no signature file, so we are finished. Download
1207cf3f 1502 // the indexes and do only hashsum verification if possible
3568a640 1503 MetaIndexParser->Load(DestFile);
1207cf3f 1504 QueueIndexes(false);
b3d44315
MV
1505 }
1506 else
1507 {
479a2beb
MV
1508 // FIXME: move this into pkgAcqMetaClearSig::Done on the next
1509 // ABI break
1510
0854ad8b
MV
1511 // if we expect a ClearTextSignature (InRelase), ensure that
1512 // this is what we get and if not fail to queue a
1513 // Release/Release.gpg, see #346386
fe5804fc 1514 if (SigFile == DestFile && !StartsWithGPGClearTextSignature(DestFile))
0854ad8b
MV
1515 {
1516 Failed(Message, Cfg);
1517 return;
1518 }
1519
b3d44315
MV
1520 // There was a signature file, so pass it to gpgv for
1521 // verification
b3d44315
MV
1522 if (_config->FindB("Debug::pkgAcquire::Auth", false))
1523 std::cerr << "Metaindex acquired, queueing gpg verification ("
1524 << SigFile << "," << DestFile << ")\n";
1525 AuthPass = true;
1526 Desc.URI = "gpgv:" + SigFile;
1527 QueueURI(Desc);
1528 Mode = "gpgv";
56bc3358 1529 return;
b3d44315
MV
1530 }
1531 }
56bc3358
DK
1532
1533 if (Complete == true)
1534 {
1535 string FinalFile = _config->FindDir("Dir::State::lists");
1536 FinalFile += URItoFileName(RealURI);
fe0f7911
DK
1537 if (SigFile == DestFile)
1538 SigFile = FinalFile;
56bc3358
DK
1539 Rename(DestFile,FinalFile);
1540 chmod(FinalFile.c_str(),0644);
1541 DestFile = FinalFile;
1542 }
b3d44315 1543}
92fcbfc1
DK
1544 /*}}}*/
1545void pkgAcqMetaIndex::RetrievalDone(string Message) /*{{{*/
b3d44315
MV
1546{
1547 // We have just finished downloading a Release file (it is not
1548 // verified yet)
1549
1550 string FileName = LookupTag(Message,"Filename");
1551 if (FileName.empty() == true)
1552 {
1553 Status = StatError;
1554 ErrorText = "Method gave a blank filename";
1555 return;
1556 }
1557
1558 if (FileName != DestFile)
1559 {
1560 Local = true;
1561 Desc.URI = "copy:" + FileName;
1562 QueueURI(Desc);
1563 return;
1564 }
1565
fce72602 1566 // make sure to verify against the right file on I-M-S hit
f381d68d 1567 IMSHit = StringToBool(LookupTag(Message,"IMS-Hit"),false);
fce72602
MV
1568 if(IMSHit)
1569 {
1570 string FinalFile = _config->FindDir("Dir::State::lists");
1571 FinalFile += URItoFileName(RealURI);
fe0f7911 1572 if (SigFile == DestFile)
0aec7d5c 1573 {
fe0f7911 1574 SigFile = FinalFile;
0aec7d5c
DK
1575 // constructor of pkgAcqMetaClearSig moved it out of the way,
1576 // now move it back in on IMS hit for the 'old' file
1577 string const OldClearSig = DestFile + ".reverify";
1578 if (RealFileExists(OldClearSig) == true)
1579 Rename(OldClearSig, FinalFile);
1580 }
fce72602
MV
1581 DestFile = FinalFile;
1582 }
b3d44315 1583 Complete = true;
b3d44315 1584}
92fcbfc1
DK
1585 /*}}}*/
1586void pkgAcqMetaIndex::AuthDone(string Message) /*{{{*/
b3d44315
MV
1587{
1588 // At this point, the gpgv method has succeeded, so there is a
1589 // valid signature from a key in the trusted keyring. We
1590 // perform additional verification of its contents, and use them
1591 // to verify the indexes we are about to download
1592
1593 if (!MetaIndexParser->Load(DestFile))
1594 {
1595 Status = StatAuthError;
1596 ErrorText = MetaIndexParser->ErrorText;
1597 return;
1598 }
1599
ce424cd4 1600 if (!VerifyVendor(Message))
b3d44315
MV
1601 {
1602 return;
1603 }
1604
1605 if (_config->FindB("Debug::pkgAcquire::Auth", false))
1606 std::cerr << "Signature verification succeeded: "
1607 << DestFile << std::endl;
1608
1609 // Download further indexes with verification
1610 QueueIndexes(true);
1611
fe0f7911
DK
1612 // is it a clearsigned MetaIndex file?
1613 if (DestFile == SigFile)
1614 return;
1615
b3d44315 1616 // Done, move signature file into position
b3d44315
MV
1617 string VerifiedSigFile = _config->FindDir("Dir::State::lists") +
1618 URItoFileName(RealURI) + ".gpg";
1619 Rename(SigFile,VerifiedSigFile);
1620 chmod(VerifiedSigFile.c_str(),0644);
1621}
92fcbfc1
DK
1622 /*}}}*/
1623void pkgAcqMetaIndex::QueueIndexes(bool verify) /*{{{*/
b3d44315 1624{
0901c5d0 1625#if 0
4fdb6123 1626 /* Reject invalid, existing Release files (LP: #346386) (Closes: #627642)
0901c5d0
JAK
1627 * FIXME: Disabled; it breaks unsigned repositories without hashes */
1628 if (!verify && FileExists(DestFile) && !MetaIndexParser->Load(DestFile))
1629 {
1630 Status = StatError;
1631 ErrorText = MetaIndexParser->ErrorText;
1632 return;
1633 }
1634#endif
8e3900d0
DK
1635 bool transInRelease = false;
1636 {
1637 std::vector<std::string> const keys = MetaIndexParser->MetaKeys();
1638 for (std::vector<std::string>::const_iterator k = keys.begin(); k != keys.end(); ++k)
1639 // FIXME: Feels wrong to check for hardcoded string here, but what should we do else…
1640 if (k->find("Translation-") != std::string::npos)
1641 {
1642 transInRelease = true;
1643 break;
1644 }
1645 }
1646
d0cfa8ad
MV
1647 // at this point the real Items are loaded in the fetcher
1648 ExpectedAdditionalItems = 0;
fa3b260f 1649 for (vector <IndexTarget*>::const_iterator Target = IndexTargets->begin();
b3d44315 1650 Target != IndexTargets->end();
f7f0d6c7 1651 ++Target)
b3d44315 1652 {
b3501edb
DK
1653 HashStringList ExpectedIndexHashes;
1654 const indexRecords::checkSum *Record = MetaIndexParser->Lookup((*Target)->MetaKey);
a5b9f489 1655 bool compressedAvailable = false;
1207cf3f 1656 if (Record == NULL)
b3d44315 1657 {
a5b9f489
DK
1658 if ((*Target)->IsOptional() == true)
1659 {
1660 std::vector<std::string> types = APT::Configuration::getCompressionTypes();
1661 for (std::vector<std::string>::const_iterator t = types.begin(); t != types.end(); ++t)
e788a834 1662 if (MetaIndexParser->Exists((*Target)->MetaKey + "." + *t) == true)
a5b9f489
DK
1663 {
1664 compressedAvailable = true;
1665 break;
1666 }
1667 }
1668 else if (verify == true)
ab53c018 1669 {
1207cf3f
DK
1670 Status = StatAuthError;
1671 strprintf(ErrorText, _("Unable to find expected entry '%s' in Release file (Wrong sources.list entry or malformed file)"), (*Target)->MetaKey.c_str());
1672 return;
ab53c018 1673 }
1207cf3f
DK
1674 }
1675 else
1676 {
b3501edb 1677 ExpectedIndexHashes = Record->Hashes;
1207cf3f 1678 if (_config->FindB("Debug::pkgAcquire::Auth", false))
ab53c018 1679 {
b3501edb
DK
1680 std::cerr << "Queueing: " << (*Target)->URI << std::endl
1681 << "Expected Hash:" << std::endl;
1682 for (HashStringList::const_iterator hs = ExpectedIndexHashes.begin(); hs != ExpectedIndexHashes.end(); ++hs)
1683 std::cerr << "\t- " << hs->toStr() << std::endl;
1207cf3f
DK
1684 std::cerr << "For: " << Record->MetaKeyFilename << std::endl;
1685 }
b3501edb 1686 if (verify == true && ExpectedIndexHashes.empty() == true && (*Target)->IsOptional() == false)
1207cf3f
DK
1687 {
1688 Status = StatAuthError;
1689 strprintf(ErrorText, _("Unable to find hash sum for '%s' in Release file"), (*Target)->MetaKey.c_str());
1690 return;
ab53c018
DK
1691 }
1692 }
1693
1694 if ((*Target)->IsOptional() == true)
1695 {
1696 if ((*Target)->IsSubIndex() == true)
1697 new pkgAcqSubIndex(Owner, (*Target)->URI, (*Target)->Description,
b3501edb 1698 (*Target)->ShortDesc, ExpectedIndexHashes);
a5b9f489 1699 else if (transInRelease == false || Record != NULL || compressedAvailable == true)
8e3900d0 1700 {
f55602cb 1701 if (_config->FindB("Acquire::PDiffs",true) == true && transInRelease == true &&
e788a834 1702 MetaIndexParser->Exists((*Target)->MetaKey + ".diff/Index") == true)
b3501edb 1703 new pkgAcqDiffIndex(Owner, *Target, ExpectedIndexHashes, MetaIndexParser);
f55602cb 1704 else
b3501edb 1705 new pkgAcqIndexTrans(Owner, *Target, ExpectedIndexHashes, MetaIndexParser);
8e3900d0 1706 }
ab53c018 1707 continue;
b3d44315 1708 }
e1430400
DK
1709
1710 /* Queue Packages file (either diff or full packages files, depending
1711 on the users option) - we also check if the PDiff Index file is listed
1712 in the Meta-Index file. Ideal would be if pkgAcqDiffIndex would test this
1713 instead, but passing the required info to it is to much hassle */
1714 if(_config->FindB("Acquire::PDiffs",true) == true && (verify == false ||
e788a834 1715 MetaIndexParser->Exists((*Target)->MetaKey + ".diff/Index") == true))
b3501edb 1716 new pkgAcqDiffIndex(Owner, *Target, ExpectedIndexHashes, MetaIndexParser);
e1430400 1717 else
b3501edb 1718 new pkgAcqIndex(Owner, *Target, ExpectedIndexHashes, MetaIndexParser);
b3d44315
MV
1719 }
1720}
92fcbfc1
DK
1721 /*}}}*/
1722bool pkgAcqMetaIndex::VerifyVendor(string Message) /*{{{*/
b3d44315 1723{
ce424cd4
MV
1724 string::size_type pos;
1725
1726 // check for missing sigs (that where not fatal because otherwise we had
1727 // bombed earlier)
1728 string missingkeys;
400ad7a4 1729 string msg = _("There is no public key available for the "
ce424cd4
MV
1730 "following key IDs:\n");
1731 pos = Message.find("NO_PUBKEY ");
1732 if (pos != std::string::npos)
1733 {
1734 string::size_type start = pos+strlen("NO_PUBKEY ");
1735 string Fingerprint = Message.substr(start, Message.find("\n")-start);
1736 missingkeys += (Fingerprint);
1737 }
1738 if(!missingkeys.empty())
e788a834 1739 _error->Warning("%s", (msg + missingkeys).c_str());
b3d44315
MV
1740
1741 string Transformed = MetaIndexParser->GetExpectedDist();
1742
1743 if (Transformed == "../project/experimental")
1744 {
1745 Transformed = "experimental";
1746 }
1747
ce424cd4 1748 pos = Transformed.rfind('/');
b3d44315
MV
1749 if (pos != string::npos)
1750 {
1751 Transformed = Transformed.substr(0, pos);
1752 }
1753
1754 if (Transformed == ".")
1755 {
1756 Transformed = "";
1757 }
1758
0323317c
DK
1759 if (_config->FindB("Acquire::Check-Valid-Until", true) == true &&
1760 MetaIndexParser->GetValidUntil() > 0) {
1761 time_t const invalid_since = time(NULL) - MetaIndexParser->GetValidUntil();
1762 if (invalid_since > 0)
1763 // TRANSLATOR: The first %s is the URL of the bad Release file, the second is
1764 // the time since then the file is invalid - formated in the same way as in
1765 // the download progress display (e.g. 7d 3h 42min 1s)
457bea86
MV
1766 return _error->Error(
1767 _("Release file for %s is expired (invalid since %s). "
1768 "Updates for this repository will not be applied."),
1769 RealURI.c_str(), TimeToStr(invalid_since).c_str());
1ddb8596
DK
1770 }
1771
b3d44315
MV
1772 if (_config->FindB("Debug::pkgAcquire::Auth", false))
1773 {
1774 std::cerr << "Got Codename: " << MetaIndexParser->GetDist() << std::endl;
1775 std::cerr << "Expecting Dist: " << MetaIndexParser->GetExpectedDist() << std::endl;
1776 std::cerr << "Transformed Dist: " << Transformed << std::endl;
1777 }
1778
1779 if (MetaIndexParser->CheckDist(Transformed) == false)
1780 {
1781 // This might become fatal one day
1782// Status = StatAuthError;
1783// ErrorText = "Conflicting distribution; expected "
1784// + MetaIndexParser->GetExpectedDist() + " but got "
1785// + MetaIndexParser->GetDist();
1786// return false;
1787 if (!Transformed.empty())
1788 {
1ddb8596 1789 _error->Warning(_("Conflicting distribution: %s (expected %s but got %s)"),
b3d44315
MV
1790 Desc.Description.c_str(),
1791 Transformed.c_str(),
1792 MetaIndexParser->GetDist().c_str());
1793 }
1794 }
1795
1796 return true;
1797}
92fcbfc1
DK
1798 /*}}}*/
1799// pkgAcqMetaIndex::Failed - no Release file present or no signature file present /*{{{*/
b3d44315
MV
1800// ---------------------------------------------------------------------
1801/* */
65512241 1802void pkgAcqMetaIndex::Failed(string Message,pkgAcquire::MethodConfig * /*Cnf*/)
b3d44315
MV
1803{
1804 if (AuthPass == true)
1805 {
fce72602 1806 // gpgv method failed, if we have a good signature
39f38a81
DK
1807 string LastGoodSigFile = _config->FindDir("Dir::State::lists").append("partial/").append(URItoFileName(RealURI));
1808 if (DestFile != SigFile)
1809 LastGoodSigFile.append(".gpg");
1810 LastGoodSigFile.append(".reverify");
fe0f7911 1811
fce72602 1812 if(FileExists(LastGoodSigFile))
f381d68d 1813 {
39f38a81 1814 string VerifiedSigFile = _config->FindDir("Dir::State::lists") + URItoFileName(RealURI);
fe0f7911 1815 if (DestFile != SigFile)
39f38a81
DK
1816 VerifiedSigFile.append(".gpg");
1817 Rename(LastGoodSigFile, VerifiedSigFile);
fce72602 1818 Status = StatTransientNetworkError;
b5595da9 1819 _error->Warning(_("An error occurred during the signature "
fce72602 1820 "verification. The repository is not updated "
2493f4b5 1821 "and the previous index files will be used. "
76b8e5a5 1822 "GPG error: %s: %s\n"),
fce72602
MV
1823 Desc.Description.c_str(),
1824 LookupTag(Message,"Message").c_str());
5d149bfc 1825 RunScripts("APT::Update::Auth-Failure");
f381d68d 1826 return;
0901c5d0 1827 } else if (LookupTag(Message,"Message").find("NODATA") != string::npos) {
4fdb6123 1828 /* Invalid signature file, reject (LP: #346386) (Closes: #627642) */
0901c5d0
JAK
1829 _error->Error(_("GPG error: %s: %s"),
1830 Desc.Description.c_str(),
1831 LookupTag(Message,"Message").c_str());
1832 return;
fce72602
MV
1833 } else {
1834 _error->Warning(_("GPG error: %s: %s"),
1835 Desc.Description.c_str(),
1836 LookupTag(Message,"Message").c_str());
f381d68d 1837 }
f381d68d 1838 // gpgv method failed
59271f62 1839 ReportMirrorFailure("GPGFailure");
b3d44315
MV
1840 }
1841
7ea7ac9e
JAK
1842 /* Always move the meta index, even if gpgv failed. This ensures
1843 * that PackageFile objects are correctly filled in */
a235ddf8 1844 if (FileExists(DestFile)) {
7ea7ac9e
JAK
1845 string FinalFile = _config->FindDir("Dir::State::lists");
1846 FinalFile += URItoFileName(RealURI);
1847 /* InRelease files become Release files, otherwise
1848 * they would be considered as trusted later on */
1849 if (SigFile == DestFile) {
1850 RealURI = RealURI.replace(RealURI.rfind("InRelease"), 9,
1851 "Release");
1852 FinalFile = FinalFile.replace(FinalFile.rfind("InRelease"), 9,
1853 "Release");
1854 SigFile = FinalFile;
1855 }
1856 Rename(DestFile,FinalFile);
1857 chmod(FinalFile.c_str(),0644);
1858
1859 DestFile = FinalFile;
1860 }
1861
b3d44315
MV
1862 // No Release file was present, or verification failed, so fall
1863 // back to queueing Packages files without verification
1864 QueueIndexes(false);
1865}
681d76d0 1866 /*}}}*/
fe0f7911
DK
1867pkgAcqMetaClearSig::pkgAcqMetaClearSig(pkgAcquire *Owner, /*{{{*/
1868 string const &URI, string const &URIDesc, string const &ShortDesc,
1869 string const &MetaIndexURI, string const &MetaIndexURIDesc, string const &MetaIndexShortDesc,
1870 string const &MetaSigURI, string const &MetaSigURIDesc, string const &MetaSigShortDesc,
fa3b260f 1871 const vector<IndexTarget*>* IndexTargets,
fe0f7911
DK
1872 indexRecords* MetaIndexParser) :
1873 pkgAcqMetaIndex(Owner, URI, URIDesc, ShortDesc, "", IndexTargets, MetaIndexParser),
1874 MetaIndexURI(MetaIndexURI), MetaIndexURIDesc(MetaIndexURIDesc), MetaIndexShortDesc(MetaIndexShortDesc),
1875 MetaSigURI(MetaSigURI), MetaSigURIDesc(MetaSigURIDesc), MetaSigShortDesc(MetaSigShortDesc)
1876{
1877 SigFile = DestFile;
39f38a81 1878
d0cfa8ad
MV
1879 // index targets + (worst case:) Release/Release.gpg
1880 ExpectedAdditionalItems = IndexTargets->size() + 2;
1881
1882
39f38a81
DK
1883 // keep the old InRelease around in case of transistent network errors
1884 string const Final = _config->FindDir("Dir::State::lists") + URItoFileName(RealURI);
ffcccd62 1885 if (RealFileExists(Final) == true)
39f38a81
DK
1886 {
1887 string const LastGoodSig = DestFile + ".reverify";
1888 Rename(Final,LastGoodSig);
1889 }
fe0f7911
DK
1890}
1891 /*}}}*/
ffcccd62
DK
1892pkgAcqMetaClearSig::~pkgAcqMetaClearSig() /*{{{*/
1893{
1894 // if the file was never queued undo file-changes done in the constructor
1895 if (QueueCounter == 1 && Status == StatIdle && FileSize == 0 && Complete == false)
1896 {
1897 string const Final = _config->FindDir("Dir::State::lists") + URItoFileName(RealURI);
1898 string const LastGoodSig = DestFile + ".reverify";
1899 if (RealFileExists(Final) == false && RealFileExists(LastGoodSig) == true)
1900 Rename(LastGoodSig, Final);
1901 }
1902}
1903 /*}}}*/
8d6c5839
MV
1904// pkgAcqMetaClearSig::Custom600Headers - Insert custom request headers /*{{{*/
1905// ---------------------------------------------------------------------
1906// FIXME: this can go away once the InRelease file is used widely
b3501edb 1907string pkgAcqMetaClearSig::Custom600Headers() const
8d6c5839
MV
1908{
1909 string Final = _config->FindDir("Dir::State::lists");
1910 Final += URItoFileName(RealURI);
1911
1912 struct stat Buf;
1913 if (stat(Final.c_str(),&Buf) != 0)
0aec7d5c
DK
1914 {
1915 Final = DestFile + ".reverify";
1916 if (stat(Final.c_str(),&Buf) != 0)
1917 return "\nIndex-File: true\nFail-Ignore: true\n";
1918 }
8d6c5839
MV
1919
1920 return "\nIndex-File: true\nFail-Ignore: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
1921}
1922 /*}}}*/
fe0f7911
DK
1923void pkgAcqMetaClearSig::Failed(string Message,pkgAcquire::MethodConfig *Cnf) /*{{{*/
1924{
d0cfa8ad
MV
1925 // we failed, we will not get additional items from this method
1926 ExpectedAdditionalItems = 0;
1927
fe0f7911
DK
1928 if (AuthPass == false)
1929 {
de498a52
DK
1930 // Remove the 'old' InRelease file if we try Release.gpg now as otherwise
1931 // the file will stay around and gives a false-auth impression (CVE-2012-0214)
1932 string FinalFile = _config->FindDir("Dir::State::lists");
1933 FinalFile.append(URItoFileName(RealURI));
1934 if (FileExists(FinalFile))
1935 unlink(FinalFile.c_str());
1936
fe0f7911
DK
1937 new pkgAcqMetaSig(Owner,
1938 MetaSigURI, MetaSigURIDesc, MetaSigShortDesc,
1939 MetaIndexURI, MetaIndexURIDesc, MetaIndexShortDesc,
1940 IndexTargets, MetaIndexParser);
1941 if (Cnf->LocalOnly == true ||
1942 StringToBool(LookupTag(Message, "Transient-Failure"), false) == false)
1943 Dequeue();
1944 }
1945 else
1946 pkgAcqMetaIndex::Failed(Message, Cnf);
1947}
1948 /*}}}*/
03e39e59
AL
1949// AcqArchive::AcqArchive - Constructor /*{{{*/
1950// ---------------------------------------------------------------------
17caf1b1
AL
1951/* This just sets up the initial fetch environment and queues the first
1952 possibilitiy */
03e39e59 1953pkgAcqArchive::pkgAcqArchive(pkgAcquire *Owner,pkgSourceList *Sources,
30e1eab5
AL
1954 pkgRecords *Recs,pkgCache::VerIterator const &Version,
1955 string &StoreFilename) :
fa3b260f 1956 Item(Owner, HashStringList()), Version(Version), Sources(Sources), Recs(Recs),
b3d44315
MV
1957 StoreFilename(StoreFilename), Vf(Version.FileList()),
1958 Trusted(false)
03e39e59 1959{
7d8afa39 1960 Retries = _config->FindI("Acquire::Retries",0);
813c8eea
AL
1961
1962 if (Version.Arch() == 0)
bdae53f1 1963 {
d1f1f6a8 1964 _error->Error(_("I wasn't able to locate a file for the %s package. "
7a3c2ab0
AL
1965 "This might mean you need to manually fix this package. "
1966 "(due to missing arch)"),
40f8a8ba 1967 Version.ParentPkg().FullName().c_str());
bdae53f1
AL
1968 return;
1969 }
813c8eea 1970
b2e465d6
AL
1971 /* We need to find a filename to determine the extension. We make the
1972 assumption here that all the available sources for this version share
1973 the same extension.. */
1974 // Skip not source sources, they do not have file fields.
69c2ecbd 1975 for (; Vf.end() == false; ++Vf)
b2e465d6
AL
1976 {
1977 if ((Vf.File()->Flags & pkgCache::Flag::NotSource) != 0)
1978 continue;
1979 break;
1980 }
1981
1982 // Does not really matter here.. we are going to fail out below
1983 if (Vf.end() != true)
1984 {
1985 // If this fails to get a file name we will bomb out below.
1986 pkgRecords::Parser &Parse = Recs->Lookup(Vf);
1987 if (_error->PendingError() == true)
1988 return;
1989
1990 // Generate the final file name as: package_version_arch.foo
1991 StoreFilename = QuoteString(Version.ParentPkg().Name(),"_:") + '_' +
1992 QuoteString(Version.VerStr(),"_:") + '_' +
1993 QuoteString(Version.Arch(),"_:.") +
1994 "." + flExtension(Parse.FileName());
1995 }
b3d44315
MV
1996
1997 // check if we have one trusted source for the package. if so, switch
6c34ccca
DK
1998 // to "TrustedOnly" mode - but only if not in AllowUnauthenticated mode
1999 bool const allowUnauth = _config->FindB("APT::Get::AllowUnauthenticated", false);
2000 bool const debugAuth = _config->FindB("Debug::pkgAcquire::Auth", false);
2001 bool seenUntrusted = false;
f7f0d6c7 2002 for (pkgCache::VerFileIterator i = Version.FileList(); i.end() == false; ++i)
b3d44315
MV
2003 {
2004 pkgIndexFile *Index;
2005 if (Sources->FindIndex(i.File(),Index) == false)
2006 continue;
6c34ccca
DK
2007
2008 if (debugAuth == true)
b3d44315 2009 std::cerr << "Checking index: " << Index->Describe()
6c34ccca
DK
2010 << "(Trusted=" << Index->IsTrusted() << ")" << std::endl;
2011
2012 if (Index->IsTrusted() == true)
2013 {
b3d44315 2014 Trusted = true;
6c34ccca
DK
2015 if (allowUnauth == false)
2016 break;
b3d44315 2017 }
6c34ccca
DK
2018 else
2019 seenUntrusted = true;
b3d44315
MV
2020 }
2021
a3371852
MV
2022 // "allow-unauthenticated" restores apts old fetching behaviour
2023 // that means that e.g. unauthenticated file:// uris are higher
2024 // priority than authenticated http:// uris
6c34ccca 2025 if (allowUnauth == true && seenUntrusted == true)
a3371852
MV
2026 Trusted = false;
2027
03e39e59 2028 // Select a source
b185acc2 2029 if (QueueNext() == false && _error->PendingError() == false)
d57f6084
DK
2030 _error->Error(_("Can't find a source to download version '%s' of '%s'"),
2031 Version.VerStr(), Version.ParentPkg().FullName(false).c_str());
b185acc2
AL
2032}
2033 /*}}}*/
2034// AcqArchive::QueueNext - Queue the next file source /*{{{*/
2035// ---------------------------------------------------------------------
17caf1b1
AL
2036/* This queues the next available file version for download. It checks if
2037 the archive is already available in the cache and stashs the MD5 for
2038 checking later. */
b185acc2 2039bool pkgAcqArchive::QueueNext()
a722b2c5 2040{
f7f0d6c7 2041 for (; Vf.end() == false; ++Vf)
03e39e59
AL
2042 {
2043 // Ignore not source sources
2044 if ((Vf.File()->Flags & pkgCache::Flag::NotSource) != 0)
2045 continue;
2046
2047 // Try to cross match against the source list
b2e465d6
AL
2048 pkgIndexFile *Index;
2049 if (Sources->FindIndex(Vf.File(),Index) == false)
2050 continue;
03e39e59 2051
b3d44315
MV
2052 // only try to get a trusted package from another source if that source
2053 // is also trusted
2054 if(Trusted && !Index->IsTrusted())
2055 continue;
2056
03e39e59
AL
2057 // Grab the text package record
2058 pkgRecords::Parser &Parse = Recs->Lookup(Vf);
2059 if (_error->PendingError() == true)
b185acc2 2060 return false;
b3501edb 2061
b2e465d6 2062 string PkgFile = Parse.FileName();
b3501edb
DK
2063 ExpectedHashes = Parse.Hashes();
2064
03e39e59 2065 if (PkgFile.empty() == true)
b2e465d6
AL
2066 return _error->Error(_("The package index files are corrupted. No Filename: "
2067 "field for package %s."),
2068 Version.ParentPkg().Name());
a6568219 2069
b3d44315
MV
2070 Desc.URI = Index->ArchiveURI(PkgFile);
2071 Desc.Description = Index->ArchiveInfo(Version);
2072 Desc.Owner = this;
40f8a8ba 2073 Desc.ShortDesc = Version.ParentPkg().FullName(true);
b3d44315 2074
17caf1b1 2075 // See if we already have the file. (Legacy filenames)
a6568219
AL
2076 FileSize = Version->Size;
2077 string FinalFile = _config->FindDir("Dir::Cache::Archives") + flNotDir(PkgFile);
2078 struct stat Buf;
2079 if (stat(FinalFile.c_str(),&Buf) == 0)
2080 {
2081 // Make sure the size matches
73da43e9 2082 if ((unsigned long long)Buf.st_size == Version->Size)
a6568219
AL
2083 {
2084 Complete = true;
2085 Local = true;
2086 Status = StatDone;
30e1eab5 2087 StoreFilename = DestFile = FinalFile;
b185acc2 2088 return true;
a6568219
AL
2089 }
2090
6b1ff003
AL
2091 /* Hmm, we have a file and its size does not match, this means it is
2092 an old style mismatched arch */
a6568219
AL
2093 unlink(FinalFile.c_str());
2094 }
17caf1b1
AL
2095
2096 // Check it again using the new style output filenames
2097 FinalFile = _config->FindDir("Dir::Cache::Archives") + flNotDir(StoreFilename);
2098 if (stat(FinalFile.c_str(),&Buf) == 0)
2099 {
2100 // Make sure the size matches
73da43e9 2101 if ((unsigned long long)Buf.st_size == Version->Size)
17caf1b1
AL
2102 {
2103 Complete = true;
2104 Local = true;
2105 Status = StatDone;
2106 StoreFilename = DestFile = FinalFile;
2107 return true;
2108 }
2109
1e3f4083 2110 /* Hmm, we have a file and its size does not match, this shouldn't
17caf1b1
AL
2111 happen.. */
2112 unlink(FinalFile.c_str());
2113 }
2114
2115 DestFile = _config->FindDir("Dir::Cache::Archives") + "partial/" + flNotDir(StoreFilename);
6b1ff003
AL
2116
2117 // Check the destination file
2118 if (stat(DestFile.c_str(),&Buf) == 0)
2119 {
2120 // Hmm, the partial file is too big, erase it
73da43e9 2121 if ((unsigned long long)Buf.st_size > Version->Size)
6b1ff003
AL
2122 unlink(DestFile.c_str());
2123 else
2124 PartialSize = Buf.st_size;
2125 }
de31189f
DK
2126
2127 // Disables download of archives - useful if no real installation follows,
2128 // e.g. if we are just interested in proposed installation order
2129 if (_config->FindB("Debug::pkgAcqArchive::NoQueue", false) == true)
2130 {
2131 Complete = true;
2132 Local = true;
2133 Status = StatDone;
2134 StoreFilename = DestFile = FinalFile;
2135 return true;
2136 }
2137
03e39e59 2138 // Create the item
b2e465d6 2139 Local = false;
03e39e59 2140 QueueURI(Desc);
b185acc2 2141
f7f0d6c7 2142 ++Vf;
b185acc2 2143 return true;
03e39e59 2144 }
b185acc2
AL
2145 return false;
2146}
03e39e59
AL
2147 /*}}}*/
2148// AcqArchive::Done - Finished fetching /*{{{*/
2149// ---------------------------------------------------------------------
2150/* */
b3501edb 2151void pkgAcqArchive::Done(string Message,unsigned long long Size, HashStringList const &CalcHashes,
459681d3 2152 pkgAcquire::MethodConfig *Cfg)
03e39e59 2153{
b3501edb 2154 Item::Done(Message, Size, CalcHashes, Cfg);
03e39e59
AL
2155
2156 // Check the size
2157 if (Size != Version->Size)
2158 {
3c8030a4 2159 RenameOnError(SizeMismatch);
03e39e59
AL
2160 return;
2161 }
b3501edb 2162
0d29b9d4 2163 // FIXME: could this empty() check impose *any* sort of security issue?
b3501edb 2164 if(ExpectedHashes.usable() && ExpectedHashes != CalcHashes)
03e39e59 2165 {
3c8030a4 2166 RenameOnError(HashSumMismatch);
b3501edb 2167 printHashSumComparision(DestFile, ExpectedHashes, CalcHashes);
495e5cb2 2168 return;
03e39e59 2169 }
a6568219
AL
2170
2171 // Grab the output filename
03e39e59
AL
2172 string FileName = LookupTag(Message,"Filename");
2173 if (FileName.empty() == true)
2174 {
2175 Status = StatError;
2176 ErrorText = "Method gave a blank filename";
2177 return;
2178 }
a6568219
AL
2179
2180 Complete = true;
30e1eab5
AL
2181
2182 // Reference filename
a6568219
AL
2183 if (FileName != DestFile)
2184 {
30e1eab5 2185 StoreFilename = DestFile = FileName;
a6568219
AL
2186 Local = true;
2187 return;
2188 }
2189
2190 // Done, move it into position
2191 string FinalFile = _config->FindDir("Dir::Cache::Archives");
17caf1b1 2192 FinalFile += flNotDir(StoreFilename);
a6568219 2193 Rename(DestFile,FinalFile);
03e39e59 2194
30e1eab5 2195 StoreFilename = DestFile = FinalFile;
03e39e59
AL
2196 Complete = true;
2197}
2198 /*}}}*/
db890fdb
AL
2199// AcqArchive::Failed - Failure handler /*{{{*/
2200// ---------------------------------------------------------------------
2201/* Here we try other sources */
7d8afa39 2202void pkgAcqArchive::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
db890fdb
AL
2203{
2204 ErrorText = LookupTag(Message,"Message");
b2e465d6
AL
2205
2206 /* We don't really want to retry on failed media swaps, this prevents
2207 that. An interesting observation is that permanent failures are not
2208 recorded. */
2209 if (Cnf->Removable == true &&
2210 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
2211 {
2212 // Vf = Version.FileList();
f7f0d6c7 2213 while (Vf.end() == false) ++Vf;
b2e465d6
AL
2214 StoreFilename = string();
2215 Item::Failed(Message,Cnf);
2216 return;
2217 }
2218
db890fdb 2219 if (QueueNext() == false)
7d8afa39
AL
2220 {
2221 // This is the retry counter
2222 if (Retries != 0 &&
2223 Cnf->LocalOnly == false &&
2224 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
2225 {
2226 Retries--;
2227 Vf = Version.FileList();
2228 if (QueueNext() == true)
2229 return;
2230 }
2231
9dbb421f 2232 StoreFilename = string();
7d8afa39
AL
2233 Item::Failed(Message,Cnf);
2234 }
db890fdb
AL
2235}
2236 /*}}}*/
92fcbfc1 2237// AcqArchive::IsTrusted - Determine whether this archive comes from a trusted source /*{{{*/
b3d44315 2238// ---------------------------------------------------------------------
b3501edb 2239APT_PURE bool pkgAcqArchive::IsTrusted() const
b3d44315
MV
2240{
2241 return Trusted;
2242}
92fcbfc1 2243 /*}}}*/
ab559b35
AL
2244// AcqArchive::Finished - Fetching has finished, tidy up /*{{{*/
2245// ---------------------------------------------------------------------
2246/* */
2247void pkgAcqArchive::Finished()
2248{
2249 if (Status == pkgAcquire::Item::StatDone &&
2250 Complete == true)
2251 return;
2252 StoreFilename = string();
2253}
2254 /*}}}*/
36375005
AL
2255// AcqFile::pkgAcqFile - Constructor /*{{{*/
2256// ---------------------------------------------------------------------
2257/* The file is added to the queue */
b3501edb 2258pkgAcqFile::pkgAcqFile(pkgAcquire *Owner,string URI, HashStringList const &Hashes,
73da43e9 2259 unsigned long long Size,string Dsc,string ShortDesc,
77278c2b
MV
2260 const string &DestDir, const string &DestFilename,
2261 bool IsIndexFile) :
fa3b260f 2262 Item(Owner, Hashes), IsIndexFile(IsIndexFile)
36375005 2263{
08cfc005
AL
2264 Retries = _config->FindI("Acquire::Retries",0);
2265
46e00f9d
MV
2266 if(!DestFilename.empty())
2267 DestFile = DestFilename;
2268 else if(!DestDir.empty())
2269 DestFile = DestDir + "/" + flNotDir(URI);
2270 else
2271 DestFile = flNotDir(URI);
2272
36375005
AL
2273 // Create the item
2274 Desc.URI = URI;
2275 Desc.Description = Dsc;
2276 Desc.Owner = this;
2277
2278 // Set the short description to the archive component
2279 Desc.ShortDesc = ShortDesc;
2280
2281 // Get the transfer sizes
2282 FileSize = Size;
2283 struct stat Buf;
2284 if (stat(DestFile.c_str(),&Buf) == 0)
2285 {
2286 // Hmm, the partial file is too big, erase it
ed9665ae 2287 if ((Size > 0) && (unsigned long long)Buf.st_size > Size)
36375005
AL
2288 unlink(DestFile.c_str());
2289 else
2290 PartialSize = Buf.st_size;
2291 }
092ae175 2292
36375005
AL
2293 QueueURI(Desc);
2294}
2295 /*}}}*/
2296// AcqFile::Done - Item downloaded OK /*{{{*/
2297// ---------------------------------------------------------------------
2298/* */
b3501edb 2299void pkgAcqFile::Done(string Message,unsigned long long Size,HashStringList const &CalcHashes,
459681d3 2300 pkgAcquire::MethodConfig *Cnf)
36375005 2301{
b3501edb 2302 Item::Done(Message,Size,CalcHashes,Cnf);
495e5cb2 2303
8a8feb29 2304 // Check the hash
b3501edb 2305 if(ExpectedHashes.usable() && ExpectedHashes != CalcHashes)
b3c39978 2306 {
3c8030a4 2307 RenameOnError(HashSumMismatch);
b3501edb 2308 printHashSumComparision(DestFile, ExpectedHashes, CalcHashes);
495e5cb2 2309 return;
b3c39978
AL
2310 }
2311
36375005
AL
2312 string FileName = LookupTag(Message,"Filename");
2313 if (FileName.empty() == true)
2314 {
2315 Status = StatError;
2316 ErrorText = "Method gave a blank filename";
2317 return;
2318 }
2319
2320 Complete = true;
2321
2322 // The files timestamp matches
2323 if (StringToBool(LookupTag(Message,"IMS-Hit"),false) == true)
2324 return;
2325
2326 // We have to copy it into place
2327 if (FileName != DestFile)
2328 {
2329 Local = true;
459681d3
AL
2330 if (_config->FindB("Acquire::Source-Symlinks",true) == false ||
2331 Cnf->Removable == true)
917ae805
AL
2332 {
2333 Desc.URI = "copy:" + FileName;
2334 QueueURI(Desc);
2335 return;
2336 }
2337
83ab33fc
AL
2338 // Erase the file if it is a symlink so we can overwrite it
2339 struct stat St;
2340 if (lstat(DestFile.c_str(),&St) == 0)
2341 {
2342 if (S_ISLNK(St.st_mode) != 0)
2343 unlink(DestFile.c_str());
2344 }
2345
2346 // Symlink the file
917ae805
AL
2347 if (symlink(FileName.c_str(),DestFile.c_str()) != 0)
2348 {
83ab33fc 2349 ErrorText = "Link to " + DestFile + " failure ";
917ae805
AL
2350 Status = StatError;
2351 Complete = false;
2352 }
36375005
AL
2353 }
2354}
2355 /*}}}*/
08cfc005
AL
2356// AcqFile::Failed - Failure handler /*{{{*/
2357// ---------------------------------------------------------------------
2358/* Here we try other sources */
2359void pkgAcqFile::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
2360{
2361 ErrorText = LookupTag(Message,"Message");
2362
2363 // This is the retry counter
2364 if (Retries != 0 &&
2365 Cnf->LocalOnly == false &&
2366 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
2367 {
2368 Retries--;
2369 QueueURI(Desc);
2370 return;
2371 }
2372
2373 Item::Failed(Message,Cnf);
2374}
2375 /*}}}*/
77278c2b
MV
2376// AcqIndex::Custom600Headers - Insert custom request headers /*{{{*/
2377// ---------------------------------------------------------------------
2378/* The only header we use is the last-modified header. */
b3501edb 2379string pkgAcqFile::Custom600Headers() const
77278c2b
MV
2380{
2381 if (IsIndexFile)
2382 return "\nIndex-File: true";
61a07c57 2383 return "";
77278c2b
MV
2384}
2385 /*}}}*/