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