]> git.saurik.com Git - apt.git/blame - apt-pkg/acquire-item.cc
pick up Translation-* even if only compressed available
[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
69c2ecbd 563 if(available_patches.empty() == true)
2ac3eeb6 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 1291 if (SigFile == DestFile)
0aec7d5c 1292 {
fe0f7911 1293 SigFile = FinalFile;
0aec7d5c
DK
1294 // constructor of pkgAcqMetaClearSig moved it out of the way,
1295 // now move it back in on IMS hit for the 'old' file
1296 string const OldClearSig = DestFile + ".reverify";
1297 if (RealFileExists(OldClearSig) == true)
1298 Rename(OldClearSig, FinalFile);
1299 }
fce72602
MV
1300 DestFile = FinalFile;
1301 }
b3d44315 1302 Complete = true;
b3d44315 1303}
92fcbfc1
DK
1304 /*}}}*/
1305void pkgAcqMetaIndex::AuthDone(string Message) /*{{{*/
b3d44315
MV
1306{
1307 // At this point, the gpgv method has succeeded, so there is a
1308 // valid signature from a key in the trusted keyring. We
1309 // perform additional verification of its contents, and use them
1310 // to verify the indexes we are about to download
1311
1312 if (!MetaIndexParser->Load(DestFile))
1313 {
1314 Status = StatAuthError;
1315 ErrorText = MetaIndexParser->ErrorText;
1316 return;
1317 }
1318
ce424cd4 1319 if (!VerifyVendor(Message))
b3d44315
MV
1320 {
1321 return;
1322 }
1323
1324 if (_config->FindB("Debug::pkgAcquire::Auth", false))
1325 std::cerr << "Signature verification succeeded: "
1326 << DestFile << std::endl;
1327
1328 // Download further indexes with verification
1329 QueueIndexes(true);
1330
fe0f7911
DK
1331 // is it a clearsigned MetaIndex file?
1332 if (DestFile == SigFile)
1333 return;
1334
b3d44315 1335 // Done, move signature file into position
b3d44315
MV
1336 string VerifiedSigFile = _config->FindDir("Dir::State::lists") +
1337 URItoFileName(RealURI) + ".gpg";
1338 Rename(SigFile,VerifiedSigFile);
1339 chmod(VerifiedSigFile.c_str(),0644);
1340}
92fcbfc1
DK
1341 /*}}}*/
1342void pkgAcqMetaIndex::QueueIndexes(bool verify) /*{{{*/
b3d44315 1343{
0901c5d0 1344#if 0
4fdb6123 1345 /* Reject invalid, existing Release files (LP: #346386) (Closes: #627642)
0901c5d0
JAK
1346 * FIXME: Disabled; it breaks unsigned repositories without hashes */
1347 if (!verify && FileExists(DestFile) && !MetaIndexParser->Load(DestFile))
1348 {
1349 Status = StatError;
1350 ErrorText = MetaIndexParser->ErrorText;
1351 return;
1352 }
1353#endif
8e3900d0
DK
1354 bool transInRelease = false;
1355 {
1356 std::vector<std::string> const keys = MetaIndexParser->MetaKeys();
1357 for (std::vector<std::string>::const_iterator k = keys.begin(); k != keys.end(); ++k)
1358 // FIXME: Feels wrong to check for hardcoded string here, but what should we do else…
1359 if (k->find("Translation-") != std::string::npos)
1360 {
1361 transInRelease = true;
1362 break;
1363 }
1364 }
1365
b3d44315
MV
1366 for (vector <struct IndexTarget*>::const_iterator Target = IndexTargets->begin();
1367 Target != IndexTargets->end();
f7f0d6c7 1368 ++Target)
b3d44315 1369 {
495e5cb2 1370 HashString ExpectedIndexHash;
1207cf3f 1371 const indexRecords::checkSum *Record = MetaIndexParser->Lookup((*Target)->MetaKey);
a5b9f489 1372 bool compressedAvailable = false;
1207cf3f 1373 if (Record == NULL)
b3d44315 1374 {
a5b9f489
DK
1375 if ((*Target)->IsOptional() == true)
1376 {
1377 std::vector<std::string> types = APT::Configuration::getCompressionTypes();
1378 for (std::vector<std::string>::const_iterator t = types.begin(); t != types.end(); ++t)
1379 if (MetaIndexParser->Exists(string((*Target)->MetaKey).append(".").append(*t)) == true)
1380 {
1381 compressedAvailable = true;
1382 break;
1383 }
1384 }
1385 else if (verify == true)
ab53c018 1386 {
1207cf3f
DK
1387 Status = StatAuthError;
1388 strprintf(ErrorText, _("Unable to find expected entry '%s' in Release file (Wrong sources.list entry or malformed file)"), (*Target)->MetaKey.c_str());
1389 return;
ab53c018 1390 }
1207cf3f
DK
1391 }
1392 else
1393 {
1394 ExpectedIndexHash = Record->Hash;
1395 if (_config->FindB("Debug::pkgAcquire::Auth", false))
ab53c018 1396 {
1207cf3f
DK
1397 std::cerr << "Queueing: " << (*Target)->URI << std::endl;
1398 std::cerr << "Expected Hash: " << ExpectedIndexHash.toStr() << std::endl;
1399 std::cerr << "For: " << Record->MetaKeyFilename << std::endl;
1400 }
1401 if (verify == true && ExpectedIndexHash.empty() == true && (*Target)->IsOptional() == false)
1402 {
1403 Status = StatAuthError;
1404 strprintf(ErrorText, _("Unable to find hash sum for '%s' in Release file"), (*Target)->MetaKey.c_str());
1405 return;
ab53c018
DK
1406 }
1407 }
1408
1409 if ((*Target)->IsOptional() == true)
1410 {
1411 if ((*Target)->IsSubIndex() == true)
1412 new pkgAcqSubIndex(Owner, (*Target)->URI, (*Target)->Description,
1413 (*Target)->ShortDesc, ExpectedIndexHash);
a5b9f489 1414 else if (transInRelease == false || Record != NULL || compressedAvailable == true)
8e3900d0 1415 {
f55602cb
DK
1416 if (_config->FindB("Acquire::PDiffs",true) == true && transInRelease == true &&
1417 MetaIndexParser->Exists(string((*Target)->MetaKey).append(".diff/Index")) == true)
1418 new pkgAcqDiffIndex(Owner, (*Target)->URI, (*Target)->Description,
1419 (*Target)->ShortDesc, ExpectedIndexHash);
1420 else
1421 new pkgAcqIndexTrans(Owner, *Target, ExpectedIndexHash, MetaIndexParser);
8e3900d0 1422 }
ab53c018 1423 continue;
b3d44315 1424 }
e1430400
DK
1425
1426 /* Queue Packages file (either diff or full packages files, depending
1427 on the users option) - we also check if the PDiff Index file is listed
1428 in the Meta-Index file. Ideal would be if pkgAcqDiffIndex would test this
1429 instead, but passing the required info to it is to much hassle */
1430 if(_config->FindB("Acquire::PDiffs",true) == true && (verify == false ||
1431 MetaIndexParser->Exists(string((*Target)->MetaKey).append(".diff/Index")) == true))
2ac3eeb6 1432 new pkgAcqDiffIndex(Owner, (*Target)->URI, (*Target)->Description,
495e5cb2 1433 (*Target)->ShortDesc, ExpectedIndexHash);
e1430400 1434 else
5d885723 1435 new pkgAcqIndex(Owner, *Target, ExpectedIndexHash, MetaIndexParser);
b3d44315
MV
1436 }
1437}
92fcbfc1
DK
1438 /*}}}*/
1439bool pkgAcqMetaIndex::VerifyVendor(string Message) /*{{{*/
b3d44315 1440{
ce424cd4
MV
1441 string::size_type pos;
1442
1443 // check for missing sigs (that where not fatal because otherwise we had
1444 // bombed earlier)
1445 string missingkeys;
400ad7a4 1446 string msg = _("There is no public key available for the "
ce424cd4
MV
1447 "following key IDs:\n");
1448 pos = Message.find("NO_PUBKEY ");
1449 if (pos != std::string::npos)
1450 {
1451 string::size_type start = pos+strlen("NO_PUBKEY ");
1452 string Fingerprint = Message.substr(start, Message.find("\n")-start);
1453 missingkeys += (Fingerprint);
1454 }
1455 if(!missingkeys.empty())
1456 _error->Warning("%s", string(msg+missingkeys).c_str());
b3d44315
MV
1457
1458 string Transformed = MetaIndexParser->GetExpectedDist();
1459
1460 if (Transformed == "../project/experimental")
1461 {
1462 Transformed = "experimental";
1463 }
1464
ce424cd4 1465 pos = Transformed.rfind('/');
b3d44315
MV
1466 if (pos != string::npos)
1467 {
1468 Transformed = Transformed.substr(0, pos);
1469 }
1470
1471 if (Transformed == ".")
1472 {
1473 Transformed = "";
1474 }
1475
0323317c
DK
1476 if (_config->FindB("Acquire::Check-Valid-Until", true) == true &&
1477 MetaIndexParser->GetValidUntil() > 0) {
1478 time_t const invalid_since = time(NULL) - MetaIndexParser->GetValidUntil();
1479 if (invalid_since > 0)
1480 // TRANSLATOR: The first %s is the URL of the bad Release file, the second is
1481 // the time since then the file is invalid - formated in the same way as in
1482 // the download progress display (e.g. 7d 3h 42min 1s)
457bea86
MV
1483 return _error->Error(
1484 _("Release file for %s is expired (invalid since %s). "
1485 "Updates for this repository will not be applied."),
1486 RealURI.c_str(), TimeToStr(invalid_since).c_str());
1ddb8596
DK
1487 }
1488
b3d44315
MV
1489 if (_config->FindB("Debug::pkgAcquire::Auth", false))
1490 {
1491 std::cerr << "Got Codename: " << MetaIndexParser->GetDist() << std::endl;
1492 std::cerr << "Expecting Dist: " << MetaIndexParser->GetExpectedDist() << std::endl;
1493 std::cerr << "Transformed Dist: " << Transformed << std::endl;
1494 }
1495
1496 if (MetaIndexParser->CheckDist(Transformed) == false)
1497 {
1498 // This might become fatal one day
1499// Status = StatAuthError;
1500// ErrorText = "Conflicting distribution; expected "
1501// + MetaIndexParser->GetExpectedDist() + " but got "
1502// + MetaIndexParser->GetDist();
1503// return false;
1504 if (!Transformed.empty())
1505 {
1ddb8596 1506 _error->Warning(_("Conflicting distribution: %s (expected %s but got %s)"),
b3d44315
MV
1507 Desc.Description.c_str(),
1508 Transformed.c_str(),
1509 MetaIndexParser->GetDist().c_str());
1510 }
1511 }
1512
1513 return true;
1514}
92fcbfc1
DK
1515 /*}}}*/
1516// pkgAcqMetaIndex::Failed - no Release file present or no signature file present /*{{{*/
b3d44315
MV
1517// ---------------------------------------------------------------------
1518/* */
1519void pkgAcqMetaIndex::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
1520{
1521 if (AuthPass == true)
1522 {
fce72602 1523 // gpgv method failed, if we have a good signature
39f38a81
DK
1524 string LastGoodSigFile = _config->FindDir("Dir::State::lists").append("partial/").append(URItoFileName(RealURI));
1525 if (DestFile != SigFile)
1526 LastGoodSigFile.append(".gpg");
1527 LastGoodSigFile.append(".reverify");
fe0f7911 1528
fce72602 1529 if(FileExists(LastGoodSigFile))
f381d68d 1530 {
39f38a81 1531 string VerifiedSigFile = _config->FindDir("Dir::State::lists") + URItoFileName(RealURI);
fe0f7911 1532 if (DestFile != SigFile)
39f38a81
DK
1533 VerifiedSigFile.append(".gpg");
1534 Rename(LastGoodSigFile, VerifiedSigFile);
fce72602 1535 Status = StatTransientNetworkError;
b5595da9 1536 _error->Warning(_("An error occurred during the signature "
fce72602 1537 "verification. The repository is not updated "
2493f4b5 1538 "and the previous index files will be used. "
76b8e5a5 1539 "GPG error: %s: %s\n"),
fce72602
MV
1540 Desc.Description.c_str(),
1541 LookupTag(Message,"Message").c_str());
5d149bfc 1542 RunScripts("APT::Update::Auth-Failure");
f381d68d 1543 return;
0901c5d0 1544 } else if (LookupTag(Message,"Message").find("NODATA") != string::npos) {
4fdb6123 1545 /* Invalid signature file, reject (LP: #346386) (Closes: #627642) */
0901c5d0
JAK
1546 _error->Error(_("GPG error: %s: %s"),
1547 Desc.Description.c_str(),
1548 LookupTag(Message,"Message").c_str());
1549 return;
fce72602
MV
1550 } else {
1551 _error->Warning(_("GPG error: %s: %s"),
1552 Desc.Description.c_str(),
1553 LookupTag(Message,"Message").c_str());
f381d68d 1554 }
f381d68d 1555 // gpgv method failed
59271f62 1556 ReportMirrorFailure("GPGFailure");
b3d44315
MV
1557 }
1558
7ea7ac9e
JAK
1559 /* Always move the meta index, even if gpgv failed. This ensures
1560 * that PackageFile objects are correctly filled in */
a235ddf8 1561 if (FileExists(DestFile)) {
7ea7ac9e
JAK
1562 string FinalFile = _config->FindDir("Dir::State::lists");
1563 FinalFile += URItoFileName(RealURI);
1564 /* InRelease files become Release files, otherwise
1565 * they would be considered as trusted later on */
1566 if (SigFile == DestFile) {
1567 RealURI = RealURI.replace(RealURI.rfind("InRelease"), 9,
1568 "Release");
1569 FinalFile = FinalFile.replace(FinalFile.rfind("InRelease"), 9,
1570 "Release");
1571 SigFile = FinalFile;
1572 }
1573 Rename(DestFile,FinalFile);
1574 chmod(FinalFile.c_str(),0644);
1575
1576 DestFile = FinalFile;
1577 }
1578
b3d44315
MV
1579 // No Release file was present, or verification failed, so fall
1580 // back to queueing Packages files without verification
1581 QueueIndexes(false);
1582}
681d76d0 1583 /*}}}*/
fe0f7911
DK
1584pkgAcqMetaClearSig::pkgAcqMetaClearSig(pkgAcquire *Owner, /*{{{*/
1585 string const &URI, string const &URIDesc, string const &ShortDesc,
1586 string const &MetaIndexURI, string const &MetaIndexURIDesc, string const &MetaIndexShortDesc,
1587 string const &MetaSigURI, string const &MetaSigURIDesc, string const &MetaSigShortDesc,
1588 const vector<struct IndexTarget*>* IndexTargets,
1589 indexRecords* MetaIndexParser) :
1590 pkgAcqMetaIndex(Owner, URI, URIDesc, ShortDesc, "", IndexTargets, MetaIndexParser),
1591 MetaIndexURI(MetaIndexURI), MetaIndexURIDesc(MetaIndexURIDesc), MetaIndexShortDesc(MetaIndexShortDesc),
1592 MetaSigURI(MetaSigURI), MetaSigURIDesc(MetaSigURIDesc), MetaSigShortDesc(MetaSigShortDesc)
1593{
1594 SigFile = DestFile;
39f38a81
DK
1595
1596 // keep the old InRelease around in case of transistent network errors
1597 string const Final = _config->FindDir("Dir::State::lists") + URItoFileName(RealURI);
1598 struct stat Buf;
1599 if (stat(Final.c_str(),&Buf) == 0)
1600 {
1601 string const LastGoodSig = DestFile + ".reverify";
1602 Rename(Final,LastGoodSig);
1603 }
fe0f7911
DK
1604}
1605 /*}}}*/
8d6c5839
MV
1606// pkgAcqMetaClearSig::Custom600Headers - Insert custom request headers /*{{{*/
1607// ---------------------------------------------------------------------
1608// FIXME: this can go away once the InRelease file is used widely
1609string pkgAcqMetaClearSig::Custom600Headers()
1610{
1611 string Final = _config->FindDir("Dir::State::lists");
1612 Final += URItoFileName(RealURI);
1613
1614 struct stat Buf;
1615 if (stat(Final.c_str(),&Buf) != 0)
0aec7d5c
DK
1616 {
1617 Final = DestFile + ".reverify";
1618 if (stat(Final.c_str(),&Buf) != 0)
1619 return "\nIndex-File: true\nFail-Ignore: true\n";
1620 }
8d6c5839
MV
1621
1622 return "\nIndex-File: true\nFail-Ignore: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
1623}
1624 /*}}}*/
fe0f7911
DK
1625void pkgAcqMetaClearSig::Failed(string Message,pkgAcquire::MethodConfig *Cnf) /*{{{*/
1626{
1627 if (AuthPass == false)
1628 {
de498a52
DK
1629 // Remove the 'old' InRelease file if we try Release.gpg now as otherwise
1630 // the file will stay around and gives a false-auth impression (CVE-2012-0214)
1631 string FinalFile = _config->FindDir("Dir::State::lists");
1632 FinalFile.append(URItoFileName(RealURI));
1633 if (FileExists(FinalFile))
1634 unlink(FinalFile.c_str());
1635
fe0f7911
DK
1636 new pkgAcqMetaSig(Owner,
1637 MetaSigURI, MetaSigURIDesc, MetaSigShortDesc,
1638 MetaIndexURI, MetaIndexURIDesc, MetaIndexShortDesc,
1639 IndexTargets, MetaIndexParser);
1640 if (Cnf->LocalOnly == true ||
1641 StringToBool(LookupTag(Message, "Transient-Failure"), false) == false)
1642 Dequeue();
1643 }
1644 else
1645 pkgAcqMetaIndex::Failed(Message, Cnf);
1646}
1647 /*}}}*/
03e39e59
AL
1648// AcqArchive::AcqArchive - Constructor /*{{{*/
1649// ---------------------------------------------------------------------
17caf1b1
AL
1650/* This just sets up the initial fetch environment and queues the first
1651 possibilitiy */
03e39e59 1652pkgAcqArchive::pkgAcqArchive(pkgAcquire *Owner,pkgSourceList *Sources,
30e1eab5
AL
1653 pkgRecords *Recs,pkgCache::VerIterator const &Version,
1654 string &StoreFilename) :
1655 Item(Owner), Version(Version), Sources(Sources), Recs(Recs),
b3d44315
MV
1656 StoreFilename(StoreFilename), Vf(Version.FileList()),
1657 Trusted(false)
03e39e59 1658{
7d8afa39 1659 Retries = _config->FindI("Acquire::Retries",0);
813c8eea
AL
1660
1661 if (Version.Arch() == 0)
bdae53f1 1662 {
d1f1f6a8 1663 _error->Error(_("I wasn't able to locate a file for the %s package. "
7a3c2ab0
AL
1664 "This might mean you need to manually fix this package. "
1665 "(due to missing arch)"),
813c8eea 1666 Version.ParentPkg().Name());
bdae53f1
AL
1667 return;
1668 }
813c8eea 1669
b2e465d6
AL
1670 /* We need to find a filename to determine the extension. We make the
1671 assumption here that all the available sources for this version share
1672 the same extension.. */
1673 // Skip not source sources, they do not have file fields.
69c2ecbd 1674 for (; Vf.end() == false; ++Vf)
b2e465d6
AL
1675 {
1676 if ((Vf.File()->Flags & pkgCache::Flag::NotSource) != 0)
1677 continue;
1678 break;
1679 }
1680
1681 // Does not really matter here.. we are going to fail out below
1682 if (Vf.end() != true)
1683 {
1684 // If this fails to get a file name we will bomb out below.
1685 pkgRecords::Parser &Parse = Recs->Lookup(Vf);
1686 if (_error->PendingError() == true)
1687 return;
1688
1689 // Generate the final file name as: package_version_arch.foo
1690 StoreFilename = QuoteString(Version.ParentPkg().Name(),"_:") + '_' +
1691 QuoteString(Version.VerStr(),"_:") + '_' +
1692 QuoteString(Version.Arch(),"_:.") +
1693 "." + flExtension(Parse.FileName());
1694 }
b3d44315
MV
1695
1696 // check if we have one trusted source for the package. if so, switch
1697 // to "TrustedOnly" mode
f7f0d6c7 1698 for (pkgCache::VerFileIterator i = Version.FileList(); i.end() == false; ++i)
b3d44315
MV
1699 {
1700 pkgIndexFile *Index;
1701 if (Sources->FindIndex(i.File(),Index) == false)
1702 continue;
1703 if (_config->FindB("Debug::pkgAcquire::Auth", false))
1704 {
1705 std::cerr << "Checking index: " << Index->Describe()
1706 << "(Trusted=" << Index->IsTrusted() << ")\n";
1707 }
1708 if (Index->IsTrusted()) {
1709 Trusted = true;
1710 break;
1711 }
1712 }
1713
a3371852
MV
1714 // "allow-unauthenticated" restores apts old fetching behaviour
1715 // that means that e.g. unauthenticated file:// uris are higher
1716 // priority than authenticated http:// uris
1717 if (_config->FindB("APT::Get::AllowUnauthenticated",false) == true)
1718 Trusted = false;
1719
03e39e59 1720 // Select a source
b185acc2 1721 if (QueueNext() == false && _error->PendingError() == false)
2d5102e8 1722 _error->Error(_("I wasn't able to locate a file for the %s package. "
b2e465d6 1723 "This might mean you need to manually fix this package."),
b185acc2
AL
1724 Version.ParentPkg().Name());
1725}
1726 /*}}}*/
1727// AcqArchive::QueueNext - Queue the next file source /*{{{*/
1728// ---------------------------------------------------------------------
17caf1b1
AL
1729/* This queues the next available file version for download. It checks if
1730 the archive is already available in the cache and stashs the MD5 for
1731 checking later. */
b185acc2 1732bool pkgAcqArchive::QueueNext()
a722b2c5
DK
1733{
1734 string const ForceHash = _config->Find("Acquire::ForceHash");
f7f0d6c7 1735 for (; Vf.end() == false; ++Vf)
03e39e59
AL
1736 {
1737 // Ignore not source sources
1738 if ((Vf.File()->Flags & pkgCache::Flag::NotSource) != 0)
1739 continue;
1740
1741 // Try to cross match against the source list
b2e465d6
AL
1742 pkgIndexFile *Index;
1743 if (Sources->FindIndex(Vf.File(),Index) == false)
1744 continue;
03e39e59 1745
b3d44315
MV
1746 // only try to get a trusted package from another source if that source
1747 // is also trusted
1748 if(Trusted && !Index->IsTrusted())
1749 continue;
1750
03e39e59
AL
1751 // Grab the text package record
1752 pkgRecords::Parser &Parse = Recs->Lookup(Vf);
1753 if (_error->PendingError() == true)
b185acc2 1754 return false;
03e39e59 1755
b2e465d6 1756 string PkgFile = Parse.FileName();
a722b2c5
DK
1757 if (ForceHash.empty() == false)
1758 {
d9b9e9e2
MV
1759 if(stringcasecmp(ForceHash, "sha512") == 0)
1760 ExpectedHash = HashString("SHA512", Parse.SHA512Hash());
ee5f5d25 1761 else if(stringcasecmp(ForceHash, "sha256") == 0)
a722b2c5
DK
1762 ExpectedHash = HashString("SHA256", Parse.SHA256Hash());
1763 else if (stringcasecmp(ForceHash, "sha1") == 0)
1764 ExpectedHash = HashString("SHA1", Parse.SHA1Hash());
1765 else
1766 ExpectedHash = HashString("MD5Sum", Parse.MD5Hash());
1767 }
1768 else
1769 {
1770 string Hash;
d9b9e9e2
MV
1771 if ((Hash = Parse.SHA512Hash()).empty() == false)
1772 ExpectedHash = HashString("SHA512", Hash);
1773 else if ((Hash = Parse.SHA256Hash()).empty() == false)
a722b2c5
DK
1774 ExpectedHash = HashString("SHA256", Hash);
1775 else if ((Hash = Parse.SHA1Hash()).empty() == false)
1776 ExpectedHash = HashString("SHA1", Hash);
1777 else
1778 ExpectedHash = HashString("MD5Sum", Parse.MD5Hash());
1779 }
03e39e59 1780 if (PkgFile.empty() == true)
b2e465d6
AL
1781 return _error->Error(_("The package index files are corrupted. No Filename: "
1782 "field for package %s."),
1783 Version.ParentPkg().Name());
a6568219 1784
b3d44315
MV
1785 Desc.URI = Index->ArchiveURI(PkgFile);
1786 Desc.Description = Index->ArchiveInfo(Version);
1787 Desc.Owner = this;
1788 Desc.ShortDesc = Version.ParentPkg().Name();
1789
17caf1b1 1790 // See if we already have the file. (Legacy filenames)
a6568219
AL
1791 FileSize = Version->Size;
1792 string FinalFile = _config->FindDir("Dir::Cache::Archives") + flNotDir(PkgFile);
1793 struct stat Buf;
1794 if (stat(FinalFile.c_str(),&Buf) == 0)
1795 {
1796 // Make sure the size matches
73da43e9 1797 if ((unsigned long long)Buf.st_size == Version->Size)
a6568219
AL
1798 {
1799 Complete = true;
1800 Local = true;
1801 Status = StatDone;
30e1eab5 1802 StoreFilename = DestFile = FinalFile;
b185acc2 1803 return true;
a6568219
AL
1804 }
1805
6b1ff003
AL
1806 /* Hmm, we have a file and its size does not match, this means it is
1807 an old style mismatched arch */
a6568219
AL
1808 unlink(FinalFile.c_str());
1809 }
17caf1b1
AL
1810
1811 // Check it again using the new style output filenames
1812 FinalFile = _config->FindDir("Dir::Cache::Archives") + flNotDir(StoreFilename);
1813 if (stat(FinalFile.c_str(),&Buf) == 0)
1814 {
1815 // Make sure the size matches
73da43e9 1816 if ((unsigned long long)Buf.st_size == Version->Size)
17caf1b1
AL
1817 {
1818 Complete = true;
1819 Local = true;
1820 Status = StatDone;
1821 StoreFilename = DestFile = FinalFile;
1822 return true;
1823 }
1824
1825 /* Hmm, we have a file and its size does not match, this shouldnt
1826 happen.. */
1827 unlink(FinalFile.c_str());
1828 }
1829
1830 DestFile = _config->FindDir("Dir::Cache::Archives") + "partial/" + flNotDir(StoreFilename);
6b1ff003
AL
1831
1832 // Check the destination file
1833 if (stat(DestFile.c_str(),&Buf) == 0)
1834 {
1835 // Hmm, the partial file is too big, erase it
73da43e9 1836 if ((unsigned long long)Buf.st_size > Version->Size)
6b1ff003
AL
1837 unlink(DestFile.c_str());
1838 else
1839 PartialSize = Buf.st_size;
1840 }
de31189f
DK
1841
1842 // Disables download of archives - useful if no real installation follows,
1843 // e.g. if we are just interested in proposed installation order
1844 if (_config->FindB("Debug::pkgAcqArchive::NoQueue", false) == true)
1845 {
1846 Complete = true;
1847 Local = true;
1848 Status = StatDone;
1849 StoreFilename = DestFile = FinalFile;
1850 return true;
1851 }
1852
03e39e59 1853 // Create the item
b2e465d6
AL
1854 Local = false;
1855 Desc.URI = Index->ArchiveURI(PkgFile);
1856 Desc.Description = Index->ArchiveInfo(Version);
03e39e59
AL
1857 Desc.Owner = this;
1858 Desc.ShortDesc = Version.ParentPkg().Name();
1859 QueueURI(Desc);
b185acc2 1860
f7f0d6c7 1861 ++Vf;
b185acc2 1862 return true;
03e39e59 1863 }
b185acc2
AL
1864 return false;
1865}
03e39e59
AL
1866 /*}}}*/
1867// AcqArchive::Done - Finished fetching /*{{{*/
1868// ---------------------------------------------------------------------
1869/* */
73da43e9 1870void pkgAcqArchive::Done(string Message,unsigned long long Size,string CalcHash,
459681d3 1871 pkgAcquire::MethodConfig *Cfg)
03e39e59 1872{
495e5cb2 1873 Item::Done(Message,Size,CalcHash,Cfg);
03e39e59
AL
1874
1875 // Check the size
1876 if (Size != Version->Size)
1877 {
bdae53f1 1878 Status = StatError;
b2e465d6 1879 ErrorText = _("Size mismatch");
03e39e59
AL
1880 return;
1881 }
1882
495e5cb2 1883 // Check the hash
8a8feb29 1884 if(ExpectedHash.toStr() != CalcHash)
03e39e59 1885 {
495e5cb2 1886 Status = StatError;
9498d182 1887 ErrorText = _("Hash Sum mismatch");
495e5cb2
MV
1888 if(FileExists(DestFile))
1889 Rename(DestFile,DestFile + ".FAILED");
1890 return;
03e39e59 1891 }
a6568219
AL
1892
1893 // Grab the output filename
03e39e59
AL
1894 string FileName = LookupTag(Message,"Filename");
1895 if (FileName.empty() == true)
1896 {
1897 Status = StatError;
1898 ErrorText = "Method gave a blank filename";
1899 return;
1900 }
a6568219
AL
1901
1902 Complete = true;
30e1eab5
AL
1903
1904 // Reference filename
a6568219
AL
1905 if (FileName != DestFile)
1906 {
30e1eab5 1907 StoreFilename = DestFile = FileName;
a6568219
AL
1908 Local = true;
1909 return;
1910 }
1911
1912 // Done, move it into position
1913 string FinalFile = _config->FindDir("Dir::Cache::Archives");
17caf1b1 1914 FinalFile += flNotDir(StoreFilename);
a6568219 1915 Rename(DestFile,FinalFile);
03e39e59 1916
30e1eab5 1917 StoreFilename = DestFile = FinalFile;
03e39e59
AL
1918 Complete = true;
1919}
1920 /*}}}*/
db890fdb
AL
1921// AcqArchive::Failed - Failure handler /*{{{*/
1922// ---------------------------------------------------------------------
1923/* Here we try other sources */
7d8afa39 1924void pkgAcqArchive::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
db890fdb
AL
1925{
1926 ErrorText = LookupTag(Message,"Message");
b2e465d6
AL
1927
1928 /* We don't really want to retry on failed media swaps, this prevents
1929 that. An interesting observation is that permanent failures are not
1930 recorded. */
1931 if (Cnf->Removable == true &&
1932 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
1933 {
1934 // Vf = Version.FileList();
f7f0d6c7 1935 while (Vf.end() == false) ++Vf;
b2e465d6
AL
1936 StoreFilename = string();
1937 Item::Failed(Message,Cnf);
1938 return;
1939 }
1940
db890fdb 1941 if (QueueNext() == false)
7d8afa39
AL
1942 {
1943 // This is the retry counter
1944 if (Retries != 0 &&
1945 Cnf->LocalOnly == false &&
1946 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
1947 {
1948 Retries--;
1949 Vf = Version.FileList();
1950 if (QueueNext() == true)
1951 return;
1952 }
1953
9dbb421f 1954 StoreFilename = string();
7d8afa39
AL
1955 Item::Failed(Message,Cnf);
1956 }
db890fdb
AL
1957}
1958 /*}}}*/
92fcbfc1 1959// AcqArchive::IsTrusted - Determine whether this archive comes from a trusted source /*{{{*/
b3d44315
MV
1960// ---------------------------------------------------------------------
1961bool pkgAcqArchive::IsTrusted()
1962{
1963 return Trusted;
1964}
92fcbfc1 1965 /*}}}*/
ab559b35
AL
1966// AcqArchive::Finished - Fetching has finished, tidy up /*{{{*/
1967// ---------------------------------------------------------------------
1968/* */
1969void pkgAcqArchive::Finished()
1970{
1971 if (Status == pkgAcquire::Item::StatDone &&
1972 Complete == true)
1973 return;
1974 StoreFilename = string();
1975}
1976 /*}}}*/
36375005
AL
1977// AcqFile::pkgAcqFile - Constructor /*{{{*/
1978// ---------------------------------------------------------------------
1979/* The file is added to the queue */
8a8feb29 1980pkgAcqFile::pkgAcqFile(pkgAcquire *Owner,string URI,string Hash,
73da43e9 1981 unsigned long long Size,string Dsc,string ShortDesc,
77278c2b
MV
1982 const string &DestDir, const string &DestFilename,
1983 bool IsIndexFile) :
1984 Item(Owner), ExpectedHash(Hash), IsIndexFile(IsIndexFile)
36375005 1985{
08cfc005
AL
1986 Retries = _config->FindI("Acquire::Retries",0);
1987
46e00f9d
MV
1988 if(!DestFilename.empty())
1989 DestFile = DestFilename;
1990 else if(!DestDir.empty())
1991 DestFile = DestDir + "/" + flNotDir(URI);
1992 else
1993 DestFile = flNotDir(URI);
1994
36375005
AL
1995 // Create the item
1996 Desc.URI = URI;
1997 Desc.Description = Dsc;
1998 Desc.Owner = this;
1999
2000 // Set the short description to the archive component
2001 Desc.ShortDesc = ShortDesc;
2002
2003 // Get the transfer sizes
2004 FileSize = Size;
2005 struct stat Buf;
2006 if (stat(DestFile.c_str(),&Buf) == 0)
2007 {
2008 // Hmm, the partial file is too big, erase it
73da43e9 2009 if ((unsigned long long)Buf.st_size > Size)
36375005
AL
2010 unlink(DestFile.c_str());
2011 else
2012 PartialSize = Buf.st_size;
2013 }
092ae175 2014
36375005
AL
2015 QueueURI(Desc);
2016}
2017 /*}}}*/
2018// AcqFile::Done - Item downloaded OK /*{{{*/
2019// ---------------------------------------------------------------------
2020/* */
73da43e9 2021void pkgAcqFile::Done(string Message,unsigned long long Size,string CalcHash,
459681d3 2022 pkgAcquire::MethodConfig *Cnf)
36375005 2023{
495e5cb2
MV
2024 Item::Done(Message,Size,CalcHash,Cnf);
2025
8a8feb29 2026 // Check the hash
2c941d89 2027 if(!ExpectedHash.empty() && ExpectedHash.toStr() != CalcHash)
b3c39978 2028 {
495e5cb2 2029 Status = StatError;
f5eb830c 2030 ErrorText = _("Hash Sum mismatch");
495e5cb2
MV
2031 Rename(DestFile,DestFile + ".FAILED");
2032 return;
b3c39978
AL
2033 }
2034
36375005
AL
2035 string FileName = LookupTag(Message,"Filename");
2036 if (FileName.empty() == true)
2037 {
2038 Status = StatError;
2039 ErrorText = "Method gave a blank filename";
2040 return;
2041 }
2042
2043 Complete = true;
2044
2045 // The files timestamp matches
2046 if (StringToBool(LookupTag(Message,"IMS-Hit"),false) == true)
2047 return;
2048
2049 // We have to copy it into place
2050 if (FileName != DestFile)
2051 {
2052 Local = true;
459681d3
AL
2053 if (_config->FindB("Acquire::Source-Symlinks",true) == false ||
2054 Cnf->Removable == true)
917ae805
AL
2055 {
2056 Desc.URI = "copy:" + FileName;
2057 QueueURI(Desc);
2058 return;
2059 }
2060
83ab33fc
AL
2061 // Erase the file if it is a symlink so we can overwrite it
2062 struct stat St;
2063 if (lstat(DestFile.c_str(),&St) == 0)
2064 {
2065 if (S_ISLNK(St.st_mode) != 0)
2066 unlink(DestFile.c_str());
2067 }
2068
2069 // Symlink the file
917ae805
AL
2070 if (symlink(FileName.c_str(),DestFile.c_str()) != 0)
2071 {
83ab33fc 2072 ErrorText = "Link to " + DestFile + " failure ";
917ae805
AL
2073 Status = StatError;
2074 Complete = false;
2075 }
36375005
AL
2076 }
2077}
2078 /*}}}*/
08cfc005
AL
2079// AcqFile::Failed - Failure handler /*{{{*/
2080// ---------------------------------------------------------------------
2081/* Here we try other sources */
2082void pkgAcqFile::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
2083{
2084 ErrorText = LookupTag(Message,"Message");
2085
2086 // This is the retry counter
2087 if (Retries != 0 &&
2088 Cnf->LocalOnly == false &&
2089 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
2090 {
2091 Retries--;
2092 QueueURI(Desc);
2093 return;
2094 }
2095
2096 Item::Failed(Message,Cnf);
2097}
2098 /*}}}*/
77278c2b
MV
2099// AcqIndex::Custom600Headers - Insert custom request headers /*{{{*/
2100// ---------------------------------------------------------------------
2101/* The only header we use is the last-modified header. */
2102string pkgAcqFile::Custom600Headers()
2103{
2104 if (IsIndexFile)
2105 return "\nIndex-File: true";
61a07c57 2106 return "";
77278c2b
MV
2107}
2108 /*}}}*/