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