merged from apt--mvo
[ntk/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 /*{{{*/
0118833a
AL
16#include <apt-pkg/acquire-item.h>
17#include <apt-pkg/configuration.h>
b2e465d6 18#include <apt-pkg/sourcelist.h>
b3d44315 19#include <apt-pkg/vendorlist.h>
03e39e59 20#include <apt-pkg/error.h>
cdcc6d34 21#include <apt-pkg/strutl.h>
36375005 22#include <apt-pkg/fileutl.h>
b3d44315 23#include <apt-pkg/md5.h>
ac5b205a
MV
24#include <apt-pkg/sha1.h>
25#include <apt-pkg/tagfile.h>
0a8a80e5 26
b2e465d6
AL
27#include <apti18n.h>
28
0a8a80e5
AL
29#include <sys/stat.h>
30#include <unistd.h>
c88edf1d 31#include <errno.h>
5819a761 32#include <string>
ac5b205a 33#include <sstream>
c88edf1d 34#include <stdio.h>
0118833a
AL
35 /*}}}*/
36
b3d44315 37using namespace std;
5819a761 38
0118833a
AL
39// Acquire::Item::Item - Constructor /*{{{*/
40// ---------------------------------------------------------------------
41/* */
8267fe24 42pkgAcquire::Item::Item(pkgAcquire *Owner) : Owner(Owner), FileSize(0),
6b1ff003
AL
43 PartialSize(0), Mode(0), ID(0), Complete(false),
44 Local(false), QueueCounter(0)
0118833a
AL
45{
46 Owner->Add(this);
c88edf1d 47 Status = StatIdle;
0118833a
AL
48}
49 /*}}}*/
50// Acquire::Item::~Item - Destructor /*{{{*/
51// ---------------------------------------------------------------------
52/* */
53pkgAcquire::Item::~Item()
54{
55 Owner->Remove(this);
56}
57 /*}}}*/
c88edf1d
AL
58// Acquire::Item::Failed - Item failed to download /*{{{*/
59// ---------------------------------------------------------------------
93bf083d
AL
60/* We return to an idle state if there are still other queues that could
61 fetch this object */
7d8afa39 62void pkgAcquire::Item::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
c88edf1d 63{
93bf083d 64 Status = StatIdle;
db890fdb 65 ErrorText = LookupTag(Message,"Message");
c88edf1d 66 if (QueueCounter <= 1)
93bf083d 67 {
a72ace20 68 /* This indicates that the file is not available right now but might
7d8afa39 69 be sometime later. If we do a retry cycle then this should be
17caf1b1 70 retried [CDROMs] */
7d8afa39
AL
71 if (Cnf->LocalOnly == true &&
72 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
a72ace20
AL
73 {
74 Status = StatIdle;
681d76d0 75 Dequeue();
a72ace20
AL
76 return;
77 }
78
93bf083d 79 Status = StatError;
681d76d0 80 Dequeue();
93bf083d 81 }
c88edf1d
AL
82}
83 /*}}}*/
8267fe24
AL
84// Acquire::Item::Start - Item has begun to download /*{{{*/
85// ---------------------------------------------------------------------
17caf1b1
AL
86/* Stash status and the file size. Note that setting Complete means
87 sub-phases of the acquire process such as decompresion are operating */
727f18af 88void pkgAcquire::Item::Start(string /*Message*/,unsigned long Size)
8267fe24
AL
89{
90 Status = StatFetching;
91 if (FileSize == 0 && Complete == false)
92 FileSize = Size;
93}
94 /*}}}*/
c88edf1d
AL
95// Acquire::Item::Done - Item downloaded OK /*{{{*/
96// ---------------------------------------------------------------------
97/* */
495e5cb2 98void pkgAcquire::Item::Done(string Message,unsigned long Size,string Hash,
459681d3 99 pkgAcquire::MethodConfig *Cnf)
c88edf1d 100{
b98f2859
AL
101 // We just downloaded something..
102 string FileName = LookupTag(Message,"Filename");
8f30ca30
MV
103 // we only inform the Log class if it was actually not a local thing
104 if (Complete == false && !Local && FileName == DestFile)
b98f2859
AL
105 {
106 if (Owner->Log != 0)
107 Owner->Log->Fetched(Size,atoi(LookupTag(Message,"Resume-Point","0").c_str()));
108 }
aa0e1101
AL
109
110 if (FileSize == 0)
111 FileSize= Size;
b98f2859 112
c88edf1d
AL
113 Status = StatDone;
114 ErrorText = string();
115 Owner->Dequeue(this);
116}
117 /*}}}*/
8b89e57f
AL
118// Acquire::Item::Rename - Rename a file /*{{{*/
119// ---------------------------------------------------------------------
120/* This helper function is used by alot of item methods as thier final
121 step */
122void pkgAcquire::Item::Rename(string From,string To)
123{
124 if (rename(From.c_str(),To.c_str()) != 0)
125 {
126 char S[300];
0fcd01de 127 snprintf(S,sizeof(S),_("rename failed, %s (%s -> %s)."),strerror(errno),
8b89e57f
AL
128 From.c_str(),To.c_str());
129 Status = StatError;
130 ErrorText = S;
7a3c2ab0 131 }
8b89e57f
AL
132}
133 /*}}}*/
0118833a 134
2237bd01
MV
135
136// AcqDiffIndex::AcqDiffIndex - Constructor
ac5b205a 137// ---------------------------------------------------------------------
2237bd01
MV
138/* Get the DiffIndex file first and see if there are patches availabe
139 * If so, create a pkgAcqIndexDiffs fetcher that will get and apply the
140 * patches. If anything goes wrong in that process, it will fall back to
141 * the original packages file
ac5b205a 142 */
2237bd01
MV
143pkgAcqDiffIndex::pkgAcqDiffIndex(pkgAcquire *Owner,
144 string URI,string URIDesc,string ShortDesc,
495e5cb2
MV
145 HashString ExpectedHash)
146 : Item(Owner), RealURI(URI), ExpectedHash(ExpectedHash),
147 Description(URIDesc)
ac5b205a
MV
148{
149
ac5b205a
MV
150 Debug = _config->FindB("Debug::pkgAcquire::Diffs",false);
151
2237bd01 152 Desc.Description = URIDesc + "/DiffIndex";
ac5b205a
MV
153 Desc.Owner = this;
154 Desc.ShortDesc = ShortDesc;
2237bd01
MV
155 Desc.URI = URI + ".diff/Index";
156
157 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
158 DestFile += URItoFileName(URI) + string(".DiffIndex");
159
160 if(Debug)
161 std::clog << "pkgAcqDiffIndex: " << Desc.URI << std::endl;
ac5b205a 162
2237bd01 163 // look for the current package file
ac5b205a
MV
164 CurrentPackagesFile = _config->FindDir("Dir::State::lists");
165 CurrentPackagesFile += URItoFileName(RealURI);
166
b4e57d2d
MV
167 // FIXME: this file:/ check is a hack to prevent fetching
168 // from local sources. this is really silly, and
169 // should be fixed cleanly as soon as possible
ac5b205a 170 if(!FileExists(CurrentPackagesFile) ||
81fcf9e2 171 Desc.URI.substr(0,strlen("file:/")) == "file:/")
2ac3eeb6 172 {
ac5b205a
MV
173 // we don't have a pkg file or we don't want to queue
174 if(Debug)
b4e57d2d 175 std::clog << "No index file, local or canceld by user" << std::endl;
ac5b205a
MV
176 Failed("", NULL);
177 return;
178 }
179
2ac3eeb6 180 if(Debug)
2237bd01
MV
181 std::clog << "pkgAcqIndexDiffs::pkgAcqIndexDiffs(): "
182 << CurrentPackagesFile << std::endl;
2ac3eeb6 183
ac5b205a 184 QueueURI(Desc);
2237bd01 185
ac5b205a
MV
186}
187
6cb30d01
MV
188// AcqIndex::Custom600Headers - Insert custom request headers /*{{{*/
189// ---------------------------------------------------------------------
190/* The only header we use is the last-modified header. */
2237bd01 191string pkgAcqDiffIndex::Custom600Headers()
6cb30d01 192{
6cb30d01
MV
193 string Final = _config->FindDir("Dir::State::lists");
194 Final += URItoFileName(RealURI) + string(".IndexDiff");
195
196 if(Debug)
197 std::clog << "Custom600Header-IMS: " << Final << std::endl;
198
199 struct stat Buf;
200 if (stat(Final.c_str(),&Buf) != 0)
201 return "\nIndex-File: true";
202
203 return "\nIndex-File: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
204}
205
2237bd01
MV
206
207bool pkgAcqDiffIndex::ParseDiffIndex(string IndexDiffFile)
208{
209 if(Debug)
210 std::clog << "pkgAcqIndexDiffs::ParseIndexDiff() " << IndexDiffFile
211 << std::endl;
212
213 pkgTagSection Tags;
214 string ServerSha1;
215 vector<DiffInfo> available_patches;
216
217 FileFd Fd(IndexDiffFile,FileFd::ReadOnly);
218 pkgTagFile TF(&Fd);
219 if (_error->PendingError() == true)
220 return false;
221
222 if(TF.Step(Tags) == true)
223 {
224 string local_sha1;
002d9943
MV
225 bool found = false;
226 DiffInfo d;
227 string size;
228
2237bd01
MV
229 string tmp = Tags.FindS("SHA1-Current");
230 std::stringstream ss(tmp);
231 ss >> ServerSha1;
232
233 FileFd fd(CurrentPackagesFile, FileFd::ReadOnly);
234 SHA1Summation SHA1;
235 SHA1.AddFD(fd.Fd(), fd.Size());
236 local_sha1 = string(SHA1.Result());
237
2ac3eeb6
MV
238 if(local_sha1 == ServerSha1)
239 {
240 // we have the same sha1 as the server
2237bd01
MV
241 if(Debug)
242 std::clog << "Package file is up-to-date" << std::endl;
002d9943
MV
243 // set found to true, this will queue a pkgAcqIndexDiffs with
244 // a empty availabe_patches
245 found = true;
2ac3eeb6
MV
246 }
247 else
248 {
002d9943
MV
249 if(Debug)
250 std::clog << "SHA1-Current: " << ServerSha1 << std::endl;
251
252 // check the historie and see what patches we need
253 string history = Tags.FindS("SHA1-History");
254 std::stringstream hist(history);
2ac3eeb6
MV
255 while(hist >> d.sha1 >> size >> d.file)
256 {
002d9943
MV
257 d.size = atoi(size.c_str());
258 // read until the first match is found
259 if(d.sha1 == local_sha1)
260 found=true;
261 // from that point on, we probably need all diffs
2ac3eeb6
MV
262 if(found)
263 {
002d9943
MV
264 if(Debug)
265 std::clog << "Need to get diff: " << d.file << std::endl;
266 available_patches.push_back(d);
267 }
2237bd01
MV
268 }
269 }
270
05aab406
MV
271 // we have something, queue the next diff
272 if(found)
2ac3eeb6 273 {
2237bd01 274 // queue the diffs
495e5cb2 275 string::size_type last_space = Description.rfind(" ");
05aab406
MV
276 if(last_space != string::npos)
277 Description.erase(last_space, Description.size()-last_space);
2237bd01 278 new pkgAcqIndexDiffs(Owner, RealURI, Description, Desc.ShortDesc,
495e5cb2 279 ExpectedHash, available_patches);
2237bd01
MV
280 Complete = false;
281 Status = StatDone;
282 Dequeue();
283 return true;
284 }
285 }
05aab406
MV
286
287 // Nothing found, report and return false
288 // Failing here is ok, if we return false later, the full
289 // IndexFile is queued
290 if(Debug)
291 std::clog << "Can't find a patch in the index file" << std::endl;
2237bd01
MV
292 return false;
293}
294
295void pkgAcqDiffIndex::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
296{
297 if(Debug)
298 std::clog << "pkgAcqDiffIndex failed: " << Desc.URI << std::endl
299 << "Falling back to normal index file aquire" << std::endl;
300
002d9943 301 new pkgAcqIndex(Owner, RealURI, Description, Desc.ShortDesc,
495e5cb2 302 ExpectedHash);
2237bd01
MV
303
304 Complete = false;
305 Status = StatDone;
306 Dequeue();
307}
308
309void pkgAcqDiffIndex::Done(string Message,unsigned long Size,string Md5Hash,
310 pkgAcquire::MethodConfig *Cnf)
311{
312 if(Debug)
313 std::clog << "pkgAcqDiffIndex::Done(): " << Desc.URI << std::endl;
314
315 Item::Done(Message,Size,Md5Hash,Cnf);
316
317 string FinalFile;
318 FinalFile = _config->FindDir("Dir::State::lists")+URItoFileName(RealURI);
319
320 // sucess in downloading the index
321 // rename the index
322 FinalFile += string(".IndexDiff");
323 if(Debug)
324 std::clog << "Renaming: " << DestFile << " -> " << FinalFile
325 << std::endl;
326 Rename(DestFile,FinalFile);
327 chmod(FinalFile.c_str(),0644);
328 DestFile = FinalFile;
329
330 if(!ParseDiffIndex(DestFile))
331 return Failed("", NULL);
332
333 Complete = true;
334 Status = StatDone;
335 Dequeue();
336 return;
337}
338
339
340
341// AcqIndexDiffs::AcqIndexDiffs - Constructor
342// ---------------------------------------------------------------------
343/* The package diff is added to the queue. one object is constructed
344 * for each diff and the index
345 */
346pkgAcqIndexDiffs::pkgAcqIndexDiffs(pkgAcquire *Owner,
347 string URI,string URIDesc,string ShortDesc,
59d5cc74 348 HashString ExpectedHash,
495e5cb2
MV
349 vector<DiffInfo> diffs)
350 : Item(Owner), RealURI(URI), ExpectedHash(ExpectedHash),
2237bd01
MV
351 available_patches(diffs)
352{
353
354 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
355 DestFile += URItoFileName(URI);
356
357 Debug = _config->FindB("Debug::pkgAcquire::Diffs",false);
358
05aab406 359 Description = URIDesc;
2237bd01
MV
360 Desc.Owner = this;
361 Desc.ShortDesc = ShortDesc;
362
2ac3eeb6
MV
363 if(available_patches.size() == 0)
364 {
2237bd01
MV
365 // we are done (yeah!)
366 Finish(true);
2ac3eeb6
MV
367 }
368 else
369 {
2237bd01
MV
370 // get the next diff
371 State = StateFetchDiff;
372 QueueNextDiff();
373 }
374}
375
376
ac5b205a
MV
377void pkgAcqIndexDiffs::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
378{
2237bd01
MV
379 if(Debug)
380 std::clog << "pkgAcqIndexDiffs failed: " << Desc.URI << std::endl
381 << "Falling back to normal index file aquire" << std::endl;
382 new pkgAcqIndex(Owner, RealURI, Description,Desc.ShortDesc,
495e5cb2 383 ExpectedHash);
ac5b205a
MV
384 Finish();
385}
386
387
388// helper that cleans the item out of the fetcher queue
389void pkgAcqIndexDiffs::Finish(bool allDone)
390{
391 // we restore the original name, this is required, otherwise
392 // the file will be cleaned
2ac3eeb6
MV
393 if(allDone)
394 {
ac5b205a
MV
395 DestFile = _config->FindDir("Dir::State::lists");
396 DestFile += URItoFileName(RealURI);
2d4722e2 397
495e5cb2 398 if(!ExpectedHash.empty() && !ExpectedHash.VerifyFile(DestFile))
2d4722e2
MV
399 {
400 Status = StatAuthError;
401 ErrorText = _("MD5Sum mismatch");
402 Rename(DestFile,DestFile + ".FAILED");
403 Dequeue();
404 return;
405 }
406
407 // this is for the "real" finish
ac5b205a 408 Complete = true;
cffc2ddd 409 Status = StatDone;
ac5b205a
MV
410 Dequeue();
411 if(Debug)
412 std::clog << "\n\nallDone: " << DestFile << "\n" << std::endl;
413 return;
ac5b205a
MV
414 }
415
416 if(Debug)
417 std::clog << "Finishing: " << Desc.URI << std::endl;
418 Complete = false;
419 Status = StatDone;
420 Dequeue();
421 return;
422}
423
424
ac5b205a
MV
425
426bool pkgAcqIndexDiffs::QueueNextDiff()
427{
3de9ff77 428
94dc9d7d
MV
429 // calc sha1 of the just patched file
430 string FinalFile = _config->FindDir("Dir::State::lists");
431 FinalFile += URItoFileName(RealURI);
432
433 FileFd fd(FinalFile, FileFd::ReadOnly);
434 SHA1Summation SHA1;
435 SHA1.AddFD(fd.Fd(), fd.Size());
436 string local_sha1 = string(SHA1.Result());
3de9ff77
MV
437 if(Debug)
438 std::clog << "QueueNextDiff: "
439 << FinalFile << " (" << local_sha1 << ")"<<std::endl;
94dc9d7d 440
26d27645
MV
441 // remove all patches until the next matching patch is found
442 // this requires the Index file to be ordered
94dc9d7d 443 for(vector<DiffInfo>::iterator I=available_patches.begin();
2ac3eeb6
MV
444 available_patches.size() > 0 &&
445 I != available_patches.end() &&
446 (*I).sha1 != local_sha1;
447 I++)
448 {
26d27645 449 available_patches.erase(I);
59a704f0 450 }
94dc9d7d
MV
451
452 // error checking and falling back if no patch was found
2ac3eeb6
MV
453 if(available_patches.size() == 0)
454 {
94dc9d7d
MV
455 Failed("", NULL);
456 return false;
457 }
6cb30d01 458
94dc9d7d 459 // queue the right diff
2237bd01 460 Desc.URI = string(RealURI) + ".diff/" + available_patches[0].file + ".gz";
05aab406 461 Desc.Description = Description + " " + available_patches[0].file + string(".pdiff");
ac5b205a 462 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
2237bd01 463 DestFile += URItoFileName(RealURI + ".diff/" + available_patches[0].file);
ac5b205a
MV
464
465 if(Debug)
466 std::clog << "pkgAcqIndexDiffs::QueueNextDiff(): " << Desc.URI << std::endl;
467
468 QueueURI(Desc);
469
470 return true;
471}
472
ac5b205a
MV
473
474
475void pkgAcqIndexDiffs::Done(string Message,unsigned long Size,string Md5Hash,
476 pkgAcquire::MethodConfig *Cnf)
477{
478 if(Debug)
479 std::clog << "pkgAcqIndexDiffs::Done(): " << Desc.URI << std::endl;
480
481 Item::Done(Message,Size,Md5Hash,Cnf);
482
4a0a786f
MV
483 string FinalFile;
484 FinalFile = _config->FindDir("Dir::State::lists")+URItoFileName(RealURI);
6cb30d01 485
4a0a786f
MV
486 // sucess in downloading a diff, enter ApplyDiff state
487 if(State == StateFetchDiff)
488 {
489
490 if(Debug)
491 std::clog << "Sending to gzip method: " << FinalFile << std::endl;
492
493 string FileName = LookupTag(Message,"Filename");
494 State = StateUnzipDiff;
b7347826 495 Local = true;
4a0a786f
MV
496 Desc.URI = "gzip:" + FileName;
497 DestFile += ".decomp";
498 QueueURI(Desc);
499 Mode = "gzip";
500 return;
501 }
502
503 // sucess in downloading a diff, enter ApplyDiff state
504 if(State == StateUnzipDiff)
505 {
506
507 // rred excepts the patch as $FinalFile.ed
508 Rename(DestFile,FinalFile+".ed");
509
510 if(Debug)
511 std::clog << "Sending to rred method: " << FinalFile << std::endl;
512
513 State = StateApplyDiff;
b7347826 514 Local = true;
4a0a786f
MV
515 Desc.URI = "rred:" + FinalFile;
516 QueueURI(Desc);
517 Mode = "rred";
518 return;
519 }
520
521
522 // success in download/apply a diff, queue next (if needed)
523 if(State == StateApplyDiff)
524 {
525 // remove the just applied patch
94dc9d7d 526 available_patches.erase(available_patches.begin());
ac5b205a 527
4a0a786f 528 // move into place
59a704f0
MV
529 if(Debug)
530 {
4a0a786f
MV
531 std::clog << "Moving patched file in place: " << std::endl
532 << DestFile << " -> " << FinalFile << std::endl;
59a704f0 533 }
4a0a786f 534 Rename(DestFile,FinalFile);
1790e0cf 535 chmod(FinalFile.c_str(),0644);
4a0a786f
MV
536
537 // see if there is more to download
94dc9d7d 538 if(available_patches.size() > 0) {
ac5b205a 539 new pkgAcqIndexDiffs(Owner, RealURI, Description, Desc.ShortDesc,
495e5cb2 540 ExpectedHash, available_patches);
4a0a786f
MV
541 return Finish();
542 } else
543 return Finish(true);
ac5b205a 544 }
ac5b205a
MV
545}
546
547
0118833a
AL
548// AcqIndex::AcqIndex - Constructor /*{{{*/
549// ---------------------------------------------------------------------
550/* The package file is added to the queue and a second class is
b2e465d6
AL
551 instantiated to fetch the revision file */
552pkgAcqIndex::pkgAcqIndex(pkgAcquire *Owner,
b3d44315 553 string URI,string URIDesc,string ShortDesc,
495e5cb2
MV
554 HashString ExpectedHash, string comprExt)
555 : Item(Owner), RealURI(URI), ExpectedHash(ExpectedHash)
0118833a 556{
8b89e57f 557 Decompression = false;
bfd22fc0 558 Erase = false;
13e8426f 559
0a8a80e5 560 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
b2e465d6 561 DestFile += URItoFileName(URI);
8267fe24 562
b3d44315
MV
563 if(comprExt.empty())
564 {
13e8426f 565 // autoselect the compression method
4577fda2 566 if(FileExists("/bin/bzip2"))
13e8426f
MV
567 CompressionExtension = ".bz2";
568 else
569 CompressionExtension = ".gz";
b3d44315 570 } else {
13e8426f 571 CompressionExtension = comprExt;
b3d44315 572 }
13e8426f 573 Desc.URI = URI + CompressionExtension;
b3d44315 574
b2e465d6 575 Desc.Description = URIDesc;
8267fe24 576 Desc.Owner = this;
b2e465d6 577 Desc.ShortDesc = ShortDesc;
8267fe24
AL
578
579 QueueURI(Desc);
0118833a
AL
580}
581 /*}}}*/
0a8a80e5 582// AcqIndex::Custom600Headers - Insert custom request headers /*{{{*/
0118833a 583// ---------------------------------------------------------------------
0a8a80e5
AL
584/* The only header we use is the last-modified header. */
585string pkgAcqIndex::Custom600Headers()
0118833a 586{
0a8a80e5 587 string Final = _config->FindDir("Dir::State::lists");
b2e465d6 588 Final += URItoFileName(RealURI);
0a8a80e5
AL
589
590 struct stat Buf;
591 if (stat(Final.c_str(),&Buf) != 0)
a72ace20 592 return "\nIndex-File: true";
0118833a 593
a72ace20 594 return "\nIndex-File: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
0118833a
AL
595}
596 /*}}}*/
debc84b2
MZ
597
598void pkgAcqIndex::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
599{
600 // no .bz2 found, retry with .gz
46e00f9d 601 if(Desc.URI.substr(Desc.URI.size()-3) == "bz2") {
debc84b2 602 Desc.URI = Desc.URI.substr(0,Desc.URI.size()-3) + "gz";
b3d44315
MV
603
604 // retry with a gzip one
605 new pkgAcqIndex(Owner, RealURI, Desc.Description,Desc.ShortDesc,
495e5cb2 606 ExpectedHash, string(".gz"));
b3d44315
MV
607 Status = StatDone;
608 Complete = false;
609 Dequeue();
debc84b2 610 return;
17ff0930
MV
611 }
612
613 // on decompression failure, remove bad versions in partial/
614 if(Decompression && Erase) {
615 string s = _config->FindDir("Dir::State::lists") + "partial/";
616 s += URItoFileName(RealURI);
617 unlink(s.c_str());
debc84b2
MZ
618 }
619
debc84b2
MZ
620 Item::Failed(Message,Cnf);
621}
622
623
8b89e57f
AL
624// AcqIndex::Done - Finished a fetch /*{{{*/
625// ---------------------------------------------------------------------
626/* This goes through a number of states.. On the initial fetch the
627 method could possibly return an alternate filename which points
628 to the uncompressed version of the file. If this is so the file
629 is copied into the partial directory. In all other cases the file
630 is decompressed with a gzip uri. */
495e5cb2 631void pkgAcqIndex::Done(string Message,unsigned long Size,string Hash,
459681d3 632 pkgAcquire::MethodConfig *Cfg)
8b89e57f 633{
495e5cb2 634 Item::Done(Message,Size,Hash,Cfg);
8b89e57f
AL
635
636 if (Decompression == true)
637 {
b3d44315
MV
638 if (_config->FindB("Debug::pkgAcquire::Auth", false))
639 {
495e5cb2
MV
640 std::cerr << std::endl << RealURI << ": Computed Hash: " << Hash;
641 std::cerr << " Expected Hash: " << ExpectedHash.toStr() << std::endl;
b3d44315
MV
642 }
643
8a8feb29 644 if (!ExpectedHash.empty() && ExpectedHash.toStr() != Hash)
b3d44315
MV
645 {
646 Status = StatAuthError;
9498d182 647 ErrorText = _("Hash Sum mismatch");
b3d44315
MV
648 Rename(DestFile,DestFile + ".FAILED");
649 return;
650 }
8b89e57f
AL
651 // Done, move it into position
652 string FinalFile = _config->FindDir("Dir::State::lists");
b2e465d6 653 FinalFile += URItoFileName(RealURI);
8b89e57f 654 Rename(DestFile,FinalFile);
7a3c2ab0 655 chmod(FinalFile.c_str(),0644);
bfd22fc0 656
7a7fa5f0
AL
657 /* We restore the original name to DestFile so that the clean operation
658 will work OK */
659 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
b2e465d6 660 DestFile += URItoFileName(RealURI);
7a7fa5f0 661
bfd22fc0
AL
662 // Remove the compressed version.
663 if (Erase == true)
bfd22fc0 664 unlink(DestFile.c_str());
8b89e57f
AL
665 return;
666 }
bfd22fc0
AL
667
668 Erase = false;
8267fe24 669 Complete = true;
bfd22fc0 670
8b89e57f
AL
671 // Handle the unzipd case
672 string FileName = LookupTag(Message,"Alt-Filename");
673 if (FileName.empty() == false)
674 {
675 // The files timestamp matches
676 if (StringToBool(LookupTag(Message,"Alt-IMS-Hit"),false) == true)
677 return;
8b89e57f 678 Decompression = true;
a6568219 679 Local = true;
8b89e57f 680 DestFile += ".decomp";
8267fe24
AL
681 Desc.URI = "copy:" + FileName;
682 QueueURI(Desc);
b98f2859 683 Mode = "copy";
8b89e57f
AL
684 return;
685 }
686
687 FileName = LookupTag(Message,"Filename");
688 if (FileName.empty() == true)
689 {
690 Status = StatError;
691 ErrorText = "Method gave a blank filename";
692 }
693
694 // The files timestamp matches
695 if (StringToBool(LookupTag(Message,"IMS-Hit"),false) == true)
696 return;
bfd22fc0
AL
697
698 if (FileName == DestFile)
699 Erase = true;
8267fe24 700 else
a6568219 701 Local = true;
8b89e57f 702
46e00f9d 703 string compExt = Desc.URI.substr(Desc.URI.size()-3);
1450dbaa 704 const char *decompProg;
debc84b2
MZ
705 if(compExt == "bz2")
706 decompProg = "bzip2";
707 else if(compExt == ".gz")
708 decompProg = "gzip";
709 else {
710 _error->Error("Unsupported extension: %s", compExt.c_str());
711 return;
712 }
713
8b89e57f
AL
714 Decompression = true;
715 DestFile += ".decomp";
debc84b2 716 Desc.URI = string(decompProg) + ":" + FileName;
8267fe24 717 QueueURI(Desc);
debc84b2 718 Mode = decompProg;
8b89e57f 719}
8b89e57f 720
a52f938b
OS
721// AcqIndexTrans::pkgAcqIndexTrans - Constructor /*{{{*/
722// ---------------------------------------------------------------------
723/* The Translation file is added to the queue */
724pkgAcqIndexTrans::pkgAcqIndexTrans(pkgAcquire *Owner,
495e5cb2
MV
725 string URI,string URIDesc,string ShortDesc)
726 : pkgAcqIndex(Owner, URI, URIDesc, ShortDesc, HashString(), "")
a52f938b
OS
727{
728}
729
730 /*}}}*/
731// AcqIndexTrans::Failed - Silence failure messages for missing files /*{{{*/
732// ---------------------------------------------------------------------
733/* */
734void pkgAcqIndexTrans::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
735{
736 if (Cnf->LocalOnly == true ||
737 StringToBool(LookupTag(Message,"Transient-Failure"),false) == false)
738 {
739 // Ignore this
740 Status = StatDone;
741 Complete = false;
742 Dequeue();
743 return;
744 }
745
746 Item::Failed(Message,Cnf);
747}
748 /*}}}*/
749
b3d44315
MV
750pkgAcqMetaSig::pkgAcqMetaSig(pkgAcquire *Owner,
751 string URI,string URIDesc,string ShortDesc,
752 string MetaIndexURI, string MetaIndexURIDesc,
753 string MetaIndexShortDesc,
754 const vector<IndexTarget*>* IndexTargets,
755 indexRecords* MetaIndexParser) :
756 Item(Owner), RealURI(URI), MetaIndexURI(MetaIndexURI),
46e00f9d
MV
757 MetaIndexURIDesc(MetaIndexURIDesc), MetaIndexShortDesc(MetaIndexShortDesc),
758 MetaIndexParser(MetaIndexParser), IndexTargets(IndexTargets)
0118833a 759{
0a8a80e5 760 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
b2e465d6 761 DestFile += URItoFileName(URI);
b3d44315 762
47eb38f4
MV
763 // remove any partial downloaded sig-file in partial/.
764 // it may confuse proxies and is too small to warrant a
765 // partial download anyway
f6237efd
MV
766 unlink(DestFile.c_str());
767
8267fe24 768 // Create the item
b2e465d6 769 Desc.Description = URIDesc;
8267fe24 770 Desc.Owner = this;
b3d44315
MV
771 Desc.ShortDesc = ShortDesc;
772 Desc.URI = URI;
b3d44315
MV
773
774 string Final = _config->FindDir("Dir::State::lists");
775 Final += URItoFileName(RealURI);
776 struct stat Buf;
777 if (stat(Final.c_str(),&Buf) == 0)
778 {
ef942597
MV
779 // File was already in place. It needs to be re-downloaded/verified
780 // because Release might have changed, we do give it a differnt
781 // name than DestFile because otherwise the http method will
782 // send If-Range requests and there are too many broken servers
783 // out there that do not understand them
784 LastGoodSig = DestFile+".reverify";
785 Rename(Final,LastGoodSig);
b3d44315 786 }
8267fe24 787
8267fe24 788 QueueURI(Desc);
0118833a
AL
789}
790 /*}}}*/
b3d44315 791// pkgAcqMetaSig::Custom600Headers - Insert custom request headers /*{{{*/
0118833a 792// ---------------------------------------------------------------------
0a8a80e5 793/* The only header we use is the last-modified header. */
b3d44315 794string pkgAcqMetaSig::Custom600Headers()
0118833a 795{
0a8a80e5 796 struct stat Buf;
ef942597 797 if (stat(LastGoodSig.c_str(),&Buf) != 0)
a72ace20 798 return "\nIndex-File: true";
a789b983 799
a72ace20 800 return "\nIndex-File: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
0118833a 801}
b3d44315
MV
802
803void pkgAcqMetaSig::Done(string Message,unsigned long Size,string MD5,
804 pkgAcquire::MethodConfig *Cfg)
c88edf1d 805{
459681d3 806 Item::Done(Message,Size,MD5,Cfg);
c88edf1d
AL
807
808 string FileName = LookupTag(Message,"Filename");
809 if (FileName.empty() == true)
810 {
811 Status = StatError;
812 ErrorText = "Method gave a blank filename";
8b89e57f 813 return;
c88edf1d 814 }
8b89e57f 815
c88edf1d
AL
816 if (FileName != DestFile)
817 {
b3d44315 818 // We have to copy it into place
a6568219 819 Local = true;
8267fe24
AL
820 Desc.URI = "copy:" + FileName;
821 QueueURI(Desc);
c88edf1d
AL
822 return;
823 }
b3d44315
MV
824
825 Complete = true;
826
ef942597
MV
827 // put the last known good file back on i-m-s hit (it will
828 // be re-verified again)
829 // Else do nothing, we have the new file in DestFile then
830 if(StringToBool(LookupTag(Message,"IMS-Hit"),false) == true)
831 Rename(LastGoodSig, DestFile);
832
b3d44315 833 // queue a pkgAcqMetaIndex to be verified against the sig we just retrieved
fce72602
MV
834 new pkgAcqMetaIndex(Owner, MetaIndexURI, MetaIndexURIDesc,
835 MetaIndexShortDesc, DestFile, IndexTargets,
836 MetaIndexParser);
b3d44315 837
c88edf1d
AL
838}
839 /*}}}*/
b3d44315 840void pkgAcqMetaSig::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
681d76d0 841{
47eb38f4 842 string Final = _config->FindDir("Dir::State::lists") + URItoFileName(RealURI);
a789b983 843
75dd8af1 844 // if we get a network error we fail gracefully
7e5f33eb
MV
845 if(Status == StatTransientNetworkError)
846 {
24057ad6 847 Item::Failed(Message,Cnf);
484dbb81 848 // move the sigfile back on transient network failures
47eb38f4 849 if(FileExists(DestFile))
ef942597 850 Rename(LastGoodSig,Final);
7e5f33eb
MV
851
852 // set the status back to , Item::Failed likes to reset it
853 Status = pkgAcquire::Item::StatTransientNetworkError;
24057ad6
MV
854 return;
855 }
856
75dd8af1 857 // Delete any existing sigfile when the acquire failed
75dd8af1
MV
858 unlink(Final.c_str());
859
b3d44315
MV
860 // queue a pkgAcqMetaIndex with no sigfile
861 new pkgAcqMetaIndex(Owner, MetaIndexURI, MetaIndexURIDesc, MetaIndexShortDesc,
862 "", IndexTargets, MetaIndexParser);
863
681d76d0
AL
864 if (Cnf->LocalOnly == true ||
865 StringToBool(LookupTag(Message,"Transient-Failure"),false) == false)
866 {
2b154e53
AL
867 // Ignore this
868 Status = StatDone;
869 Complete = false;
681d76d0
AL
870 Dequeue();
871 return;
872 }
873
874 Item::Failed(Message,Cnf);
875}
b3d44315
MV
876
877pkgAcqMetaIndex::pkgAcqMetaIndex(pkgAcquire *Owner,
878 string URI,string URIDesc,string ShortDesc,
879 string SigFile,
880 const vector<struct IndexTarget*>* IndexTargets,
881 indexRecords* MetaIndexParser) :
21fd1746
OS
882 Item(Owner), RealURI(URI), SigFile(SigFile), IndexTargets(IndexTargets),
883 MetaIndexParser(MetaIndexParser), AuthPass(false), IMSHit(false)
b3d44315 884{
b3d44315
MV
885 DestFile = _config->FindDir("Dir::State::lists") + "partial/";
886 DestFile += URItoFileName(URI);
887
888 // Create the item
889 Desc.Description = URIDesc;
890 Desc.Owner = this;
891 Desc.ShortDesc = ShortDesc;
892 Desc.URI = URI;
893
894 QueueURI(Desc);
895}
896
897 /*}}}*/
898// pkgAcqMetaIndex::Custom600Headers - Insert custom request headers /*{{{*/
899// ---------------------------------------------------------------------
900/* The only header we use is the last-modified header. */
901string pkgAcqMetaIndex::Custom600Headers()
902{
903 string Final = _config->FindDir("Dir::State::lists");
904 Final += URItoFileName(RealURI);
905
906 struct stat Buf;
907 if (stat(Final.c_str(),&Buf) != 0)
908 return "\nIndex-File: true";
909
910 return "\nIndex-File: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime);
911}
912
495e5cb2 913void pkgAcqMetaIndex::Done(string Message,unsigned long Size,string Hash,
b3d44315
MV
914 pkgAcquire::MethodConfig *Cfg)
915{
495e5cb2 916 Item::Done(Message,Size,Hash,Cfg);
b3d44315
MV
917
918 // MetaIndexes are done in two passes: one to download the
919 // metaindex with an appropriate method, and a second to verify it
920 // with the gpgv method
921
922 if (AuthPass == true)
923 {
924 AuthDone(Message);
fce72602
MV
925
926 // all cool, move Release file into place
927 Complete = true;
928
929 string FinalFile = _config->FindDir("Dir::State::lists");
930 FinalFile += URItoFileName(RealURI);
931 Rename(DestFile,FinalFile);
932 chmod(FinalFile.c_str(),0644);
933 DestFile = FinalFile;
b3d44315
MV
934 }
935 else
936 {
937 RetrievalDone(Message);
f3224a73
MV
938 if (!Complete)
939 // Still more retrieving to do
940 return;
b3d44315
MV
941
942 if (SigFile == "")
943 {
944 // There was no signature file, so we are finished. Download
945 // the indexes without verification.
946 QueueIndexes(false);
947 }
948 else
949 {
950 // There was a signature file, so pass it to gpgv for
951 // verification
952
953 if (_config->FindB("Debug::pkgAcquire::Auth", false))
954 std::cerr << "Metaindex acquired, queueing gpg verification ("
955 << SigFile << "," << DestFile << ")\n";
956 AuthPass = true;
957 Desc.URI = "gpgv:" + SigFile;
958 QueueURI(Desc);
959 Mode = "gpgv";
960 }
961 }
962}
963
964void pkgAcqMetaIndex::RetrievalDone(string Message)
965{
966 // We have just finished downloading a Release file (it is not
967 // verified yet)
968
969 string FileName = LookupTag(Message,"Filename");
970 if (FileName.empty() == true)
971 {
972 Status = StatError;
973 ErrorText = "Method gave a blank filename";
974 return;
975 }
976
977 if (FileName != DestFile)
978 {
979 Local = true;
980 Desc.URI = "copy:" + FileName;
981 QueueURI(Desc);
982 return;
983 }
984
fce72602 985 // make sure to verify against the right file on I-M-S hit
f381d68d 986 IMSHit = StringToBool(LookupTag(Message,"IMS-Hit"),false);
fce72602
MV
987 if(IMSHit)
988 {
989 string FinalFile = _config->FindDir("Dir::State::lists");
990 FinalFile += URItoFileName(RealURI);
991 DestFile = FinalFile;
992 }
f3224a73 993 Complete = true;
b3d44315
MV
994}
995
996void pkgAcqMetaIndex::AuthDone(string Message)
997{
998 // At this point, the gpgv method has succeeded, so there is a
999 // valid signature from a key in the trusted keyring. We
1000 // perform additional verification of its contents, and use them
1001 // to verify the indexes we are about to download
1002
1003 if (!MetaIndexParser->Load(DestFile))
1004 {
1005 Status = StatAuthError;
1006 ErrorText = MetaIndexParser->ErrorText;
1007 return;
1008 }
1009
ce424cd4 1010 if (!VerifyVendor(Message))
b3d44315
MV
1011 {
1012 return;
1013 }
1014
1015 if (_config->FindB("Debug::pkgAcquire::Auth", false))
1016 std::cerr << "Signature verification succeeded: "
1017 << DestFile << std::endl;
1018
1019 // Download further indexes with verification
1020 QueueIndexes(true);
1021
1022 // Done, move signature file into position
b3d44315
MV
1023 string VerifiedSigFile = _config->FindDir("Dir::State::lists") +
1024 URItoFileName(RealURI) + ".gpg";
1025 Rename(SigFile,VerifiedSigFile);
1026 chmod(VerifiedSigFile.c_str(),0644);
1027}
1028
1029void pkgAcqMetaIndex::QueueIndexes(bool verify)
1030{
1031 for (vector <struct IndexTarget*>::const_iterator Target = IndexTargets->begin();
1032 Target != IndexTargets->end();
1033 Target++)
1034 {
495e5cb2 1035 HashString ExpectedIndexHash;
b3d44315
MV
1036 if (verify)
1037 {
1038 const indexRecords::checkSum *Record = MetaIndexParser->Lookup((*Target)->MetaKey);
1039 if (!Record)
1040 {
1041 Status = StatAuthError;
1042 ErrorText = "Unable to find expected entry "
1043 + (*Target)->MetaKey + " in Meta-index file (malformed Release file?)";
1044 return;
1045 }
495e5cb2 1046 ExpectedIndexHash = Record->Hash;
b3d44315
MV
1047 if (_config->FindB("Debug::pkgAcquire::Auth", false))
1048 {
1049 std::cerr << "Queueing: " << (*Target)->URI << std::endl;
495e5cb2 1050 std::cerr << "Expected Hash: " << ExpectedIndexHash.toStr() << std::endl;
b3d44315 1051 }
495e5cb2 1052 if (ExpectedIndexHash.empty())
b3d44315
MV
1053 {
1054 Status = StatAuthError;
495e5cb2 1055 ErrorText = "Unable to find hash sum for "
b3d44315
MV
1056 + (*Target)->MetaKey + " in Meta-index file";
1057 return;
1058 }
1059 }
1060
2ac3eeb6
MV
1061 // Queue Packages file (either diff or full packages files, depending
1062 // on the users option)
27299daf 1063 if(_config->FindB("Acquire::PDiffs",true) == true)
2ac3eeb6 1064 new pkgAcqDiffIndex(Owner, (*Target)->URI, (*Target)->Description,
495e5cb2 1065 (*Target)->ShortDesc, ExpectedIndexHash);
2ac3eeb6 1066 else
81fcf9e2 1067 new pkgAcqIndex(Owner, (*Target)->URI, (*Target)->Description,
495e5cb2 1068 (*Target)->ShortDesc, ExpectedIndexHash);
b3d44315
MV
1069 }
1070}
1071
ce424cd4 1072bool pkgAcqMetaIndex::VerifyVendor(string Message)
b3d44315
MV
1073{
1074// // Maybe this should be made available from above so we don't have
1075// // to read and parse it every time?
1076// pkgVendorList List;
1077// List.ReadMainList();
1078
1079// const Vendor* Vndr = NULL;
1080// for (std::vector<string>::const_iterator I = GPGVOutput.begin(); I != GPGVOutput.end(); I++)
1081// {
1082// string::size_type pos = (*I).find("VALIDSIG ");
1083// if (_config->FindB("Debug::Vendor", false))
1084// std::cerr << "Looking for VALIDSIG in \"" << (*I) << "\": pos " << pos
1085// << std::endl;
1086// if (pos != std::string::npos)
1087// {
1088// string Fingerprint = (*I).substr(pos+sizeof("VALIDSIG"));
1089// if (_config->FindB("Debug::Vendor", false))
1090// std::cerr << "Looking for \"" << Fingerprint << "\" in vendor..." <<
1091// std::endl;
1092// Vndr = List.FindVendor(Fingerprint) != "";
1093// if (Vndr != NULL);
1094// break;
1095// }
1096// }
ce424cd4
MV
1097 string::size_type pos;
1098
1099 // check for missing sigs (that where not fatal because otherwise we had
1100 // bombed earlier)
1101 string missingkeys;
400ad7a4 1102 string msg = _("There is no public key available for the "
ce424cd4
MV
1103 "following key IDs:\n");
1104 pos = Message.find("NO_PUBKEY ");
1105 if (pos != std::string::npos)
1106 {
1107 string::size_type start = pos+strlen("NO_PUBKEY ");
1108 string Fingerprint = Message.substr(start, Message.find("\n")-start);
1109 missingkeys += (Fingerprint);
1110 }
1111 if(!missingkeys.empty())
1112 _error->Warning("%s", string(msg+missingkeys).c_str());
b3d44315
MV
1113
1114 string Transformed = MetaIndexParser->GetExpectedDist();
1115
1116 if (Transformed == "../project/experimental")
1117 {
1118 Transformed = "experimental";
1119 }
1120
ce424cd4 1121 pos = Transformed.rfind('/');
b3d44315
MV
1122 if (pos != string::npos)
1123 {
1124 Transformed = Transformed.substr(0, pos);
1125 }
1126
1127 if (Transformed == ".")
1128 {
1129 Transformed = "";
1130 }
1131
1132 if (_config->FindB("Debug::pkgAcquire::Auth", false))
1133 {
1134 std::cerr << "Got Codename: " << MetaIndexParser->GetDist() << std::endl;
1135 std::cerr << "Expecting Dist: " << MetaIndexParser->GetExpectedDist() << std::endl;
1136 std::cerr << "Transformed Dist: " << Transformed << std::endl;
1137 }
1138
1139 if (MetaIndexParser->CheckDist(Transformed) == false)
1140 {
1141 // This might become fatal one day
1142// Status = StatAuthError;
1143// ErrorText = "Conflicting distribution; expected "
1144// + MetaIndexParser->GetExpectedDist() + " but got "
1145// + MetaIndexParser->GetDist();
1146// return false;
1147 if (!Transformed.empty())
1148 {
1149 _error->Warning("Conflicting distribution: %s (expected %s but got %s)",
1150 Desc.Description.c_str(),
1151 Transformed.c_str(),
1152 MetaIndexParser->GetDist().c_str());
1153 }
1154 }
1155
1156 return true;
1157}
1158 /*}}}*/
1159// pkgAcqMetaIndex::Failed - no Release file present or no signature
1160// file present /*{{{*/
1161// ---------------------------------------------------------------------
1162/* */
1163void pkgAcqMetaIndex::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
1164{
1165 if (AuthPass == true)
1166 {
fce72602
MV
1167 // gpgv method failed, if we have a good signature
1168 string LastGoodSigFile = _config->FindDir("Dir::State::lists") +
1169 "partial/" + URItoFileName(RealURI) + ".gpg.reverify";
1170 if(FileExists(LastGoodSigFile))
f381d68d 1171 {
fce72602
MV
1172 string VerifiedSigFile = _config->FindDir("Dir::State::lists") +
1173 URItoFileName(RealURI) + ".gpg";
1174 Rename(LastGoodSigFile,VerifiedSigFile);
1175 Status = StatTransientNetworkError;
1176 _error->Warning(_("A error occurred during the signature "
1177 "verification. The repository is not updated "
1178 "and the previous index files will be used."
76b8e5a5 1179 "GPG error: %s: %s\n"),
fce72602
MV
1180 Desc.Description.c_str(),
1181 LookupTag(Message,"Message").c_str());
f381d68d 1182 return;
fce72602
MV
1183 } else {
1184 _error->Warning(_("GPG error: %s: %s"),
1185 Desc.Description.c_str(),
1186 LookupTag(Message,"Message").c_str());
f381d68d 1187 }
b3d44315
MV
1188 }
1189
1190 // No Release file was present, or verification failed, so fall
1191 // back to queueing Packages files without verification
1192 QueueIndexes(false);
1193}
1194
681d76d0 1195 /*}}}*/
03e39e59
AL
1196
1197// AcqArchive::AcqArchive - Constructor /*{{{*/
1198// ---------------------------------------------------------------------
17caf1b1
AL
1199/* This just sets up the initial fetch environment and queues the first
1200 possibilitiy */
03e39e59 1201pkgAcqArchive::pkgAcqArchive(pkgAcquire *Owner,pkgSourceList *Sources,
30e1eab5
AL
1202 pkgRecords *Recs,pkgCache::VerIterator const &Version,
1203 string &StoreFilename) :
1204 Item(Owner), Version(Version), Sources(Sources), Recs(Recs),
b3d44315
MV
1205 StoreFilename(StoreFilename), Vf(Version.FileList()),
1206 Trusted(false)
03e39e59 1207{
7d8afa39 1208 Retries = _config->FindI("Acquire::Retries",0);
813c8eea
AL
1209
1210 if (Version.Arch() == 0)
bdae53f1 1211 {
d1f1f6a8 1212 _error->Error(_("I wasn't able to locate a file for the %s package. "
7a3c2ab0
AL
1213 "This might mean you need to manually fix this package. "
1214 "(due to missing arch)"),
813c8eea 1215 Version.ParentPkg().Name());
bdae53f1
AL
1216 return;
1217 }
813c8eea 1218
b2e465d6
AL
1219 /* We need to find a filename to determine the extension. We make the
1220 assumption here that all the available sources for this version share
1221 the same extension.. */
1222 // Skip not source sources, they do not have file fields.
1223 for (; Vf.end() == false; Vf++)
1224 {
1225 if ((Vf.File()->Flags & pkgCache::Flag::NotSource) != 0)
1226 continue;
1227 break;
1228 }
1229
1230 // Does not really matter here.. we are going to fail out below
1231 if (Vf.end() != true)
1232 {
1233 // If this fails to get a file name we will bomb out below.
1234 pkgRecords::Parser &Parse = Recs->Lookup(Vf);
1235 if (_error->PendingError() == true)
1236 return;
1237
1238 // Generate the final file name as: package_version_arch.foo
1239 StoreFilename = QuoteString(Version.ParentPkg().Name(),"_:") + '_' +
1240 QuoteString(Version.VerStr(),"_:") + '_' +
1241 QuoteString(Version.Arch(),"_:.") +
1242 "." + flExtension(Parse.FileName());
1243 }
b3d44315
MV
1244
1245 // check if we have one trusted source for the package. if so, switch
1246 // to "TrustedOnly" mode
1247 for (pkgCache::VerFileIterator i = Version.FileList(); i.end() == false; i++)
1248 {
1249 pkgIndexFile *Index;
1250 if (Sources->FindIndex(i.File(),Index) == false)
1251 continue;
1252 if (_config->FindB("Debug::pkgAcquire::Auth", false))
1253 {
1254 std::cerr << "Checking index: " << Index->Describe()
1255 << "(Trusted=" << Index->IsTrusted() << ")\n";
1256 }
1257 if (Index->IsTrusted()) {
1258 Trusted = true;
1259 break;
1260 }
1261 }
1262
a3371852
MV
1263 // "allow-unauthenticated" restores apts old fetching behaviour
1264 // that means that e.g. unauthenticated file:// uris are higher
1265 // priority than authenticated http:// uris
1266 if (_config->FindB("APT::Get::AllowUnauthenticated",false) == true)
1267 Trusted = false;
1268
03e39e59 1269 // Select a source
b185acc2 1270 if (QueueNext() == false && _error->PendingError() == false)
b2e465d6
AL
1271 _error->Error(_("I wasn't able to locate file for the %s package. "
1272 "This might mean you need to manually fix this package."),
b185acc2
AL
1273 Version.ParentPkg().Name());
1274}
1275 /*}}}*/
1276// AcqArchive::QueueNext - Queue the next file source /*{{{*/
1277// ---------------------------------------------------------------------
17caf1b1
AL
1278/* This queues the next available file version for download. It checks if
1279 the archive is already available in the cache and stashs the MD5 for
1280 checking later. */
b185acc2 1281bool pkgAcqArchive::QueueNext()
b2e465d6 1282{
03e39e59
AL
1283 for (; Vf.end() == false; Vf++)
1284 {
1285 // Ignore not source sources
1286 if ((Vf.File()->Flags & pkgCache::Flag::NotSource) != 0)
1287 continue;
1288
1289 // Try to cross match against the source list
b2e465d6
AL
1290 pkgIndexFile *Index;
1291 if (Sources->FindIndex(Vf.File(),Index) == false)
1292 continue;
03e39e59 1293
b3d44315
MV
1294 // only try to get a trusted package from another source if that source
1295 // is also trusted
1296 if(Trusted && !Index->IsTrusted())
1297 continue;
1298
03e39e59
AL
1299 // Grab the text package record
1300 pkgRecords::Parser &Parse = Recs->Lookup(Vf);
1301 if (_error->PendingError() == true)
b185acc2 1302 return false;
03e39e59 1303
b2e465d6 1304 string PkgFile = Parse.FileName();
495e5cb2 1305 if(Parse.SHA256Hash() != "")
8a8feb29 1306 ExpectedHash = HashString("SHA256", Parse.SHA256Hash());
495e5cb2 1307 else if (Parse.SHA1Hash() != "")
8a8feb29 1308 ExpectedHash = HashString("SHA1", Parse.SHA1Hash());
495e5cb2 1309 else
8a8feb29 1310 ExpectedHash = HashString("MD5Sum", Parse.MD5Hash());
03e39e59 1311 if (PkgFile.empty() == true)
b2e465d6
AL
1312 return _error->Error(_("The package index files are corrupted. No Filename: "
1313 "field for package %s."),
1314 Version.ParentPkg().Name());
a6568219 1315
b3d44315
MV
1316 Desc.URI = Index->ArchiveURI(PkgFile);
1317 Desc.Description = Index->ArchiveInfo(Version);
1318 Desc.Owner = this;
1319 Desc.ShortDesc = Version.ParentPkg().Name();
1320
17caf1b1 1321 // See if we already have the file. (Legacy filenames)
a6568219
AL
1322 FileSize = Version->Size;
1323 string FinalFile = _config->FindDir("Dir::Cache::Archives") + flNotDir(PkgFile);
1324 struct stat Buf;
1325 if (stat(FinalFile.c_str(),&Buf) == 0)
1326 {
1327 // Make sure the size matches
1328 if ((unsigned)Buf.st_size == Version->Size)
1329 {
1330 Complete = true;
1331 Local = true;
1332 Status = StatDone;
30e1eab5 1333 StoreFilename = DestFile = FinalFile;
b185acc2 1334 return true;
a6568219
AL
1335 }
1336
6b1ff003
AL
1337 /* Hmm, we have a file and its size does not match, this means it is
1338 an old style mismatched arch */
a6568219
AL
1339 unlink(FinalFile.c_str());
1340 }
17caf1b1
AL
1341
1342 // Check it again using the new style output filenames
1343 FinalFile = _config->FindDir("Dir::Cache::Archives") + flNotDir(StoreFilename);
1344 if (stat(FinalFile.c_str(),&Buf) == 0)
1345 {
1346 // Make sure the size matches
1347 if ((unsigned)Buf.st_size == Version->Size)
1348 {
1349 Complete = true;
1350 Local = true;
1351 Status = StatDone;
1352 StoreFilename = DestFile = FinalFile;
1353 return true;
1354 }
1355
1356 /* Hmm, we have a file and its size does not match, this shouldnt
1357 happen.. */
1358 unlink(FinalFile.c_str());
1359 }
1360
1361 DestFile = _config->FindDir("Dir::Cache::Archives") + "partial/" + flNotDir(StoreFilename);
6b1ff003
AL
1362
1363 // Check the destination file
1364 if (stat(DestFile.c_str(),&Buf) == 0)
1365 {
1366 // Hmm, the partial file is too big, erase it
1367 if ((unsigned)Buf.st_size > Version->Size)
1368 unlink(DestFile.c_str());
1369 else
1370 PartialSize = Buf.st_size;
1371 }
1372
03e39e59 1373 // Create the item
b2e465d6
AL
1374 Local = false;
1375 Desc.URI = Index->ArchiveURI(PkgFile);
1376 Desc.Description = Index->ArchiveInfo(Version);
03e39e59
AL
1377 Desc.Owner = this;
1378 Desc.ShortDesc = Version.ParentPkg().Name();
1379 QueueURI(Desc);
b185acc2
AL
1380
1381 Vf++;
1382 return true;
03e39e59 1383 }
b185acc2
AL
1384 return false;
1385}
03e39e59
AL
1386 /*}}}*/
1387// AcqArchive::Done - Finished fetching /*{{{*/
1388// ---------------------------------------------------------------------
1389/* */
495e5cb2 1390void pkgAcqArchive::Done(string Message,unsigned long Size,string CalcHash,
459681d3 1391 pkgAcquire::MethodConfig *Cfg)
03e39e59 1392{
495e5cb2 1393 Item::Done(Message,Size,CalcHash,Cfg);
03e39e59
AL
1394
1395 // Check the size
1396 if (Size != Version->Size)
1397 {
bdae53f1 1398 Status = StatError;
b2e465d6 1399 ErrorText = _("Size mismatch");
03e39e59
AL
1400 return;
1401 }
1402
495e5cb2 1403 // Check the hash
8a8feb29 1404 if(ExpectedHash.toStr() != CalcHash)
03e39e59 1405 {
495e5cb2 1406 Status = StatError;
9498d182 1407 ErrorText = _("Hash Sum mismatch");
495e5cb2
MV
1408 if(FileExists(DestFile))
1409 Rename(DestFile,DestFile + ".FAILED");
1410 return;
03e39e59 1411 }
a6568219
AL
1412
1413 // Grab the output filename
03e39e59
AL
1414 string FileName = LookupTag(Message,"Filename");
1415 if (FileName.empty() == true)
1416 {
1417 Status = StatError;
1418 ErrorText = "Method gave a blank filename";
1419 return;
1420 }
a6568219
AL
1421
1422 Complete = true;
30e1eab5
AL
1423
1424 // Reference filename
a6568219
AL
1425 if (FileName != DestFile)
1426 {
30e1eab5 1427 StoreFilename = DestFile = FileName;
a6568219
AL
1428 Local = true;
1429 return;
1430 }
1431
1432 // Done, move it into position
1433 string FinalFile = _config->FindDir("Dir::Cache::Archives");
17caf1b1 1434 FinalFile += flNotDir(StoreFilename);
a6568219 1435 Rename(DestFile,FinalFile);
03e39e59 1436
30e1eab5 1437 StoreFilename = DestFile = FinalFile;
03e39e59
AL
1438 Complete = true;
1439}
1440 /*}}}*/
db890fdb
AL
1441// AcqArchive::Failed - Failure handler /*{{{*/
1442// ---------------------------------------------------------------------
1443/* Here we try other sources */
7d8afa39 1444void pkgAcqArchive::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
db890fdb
AL
1445{
1446 ErrorText = LookupTag(Message,"Message");
b2e465d6
AL
1447
1448 /* We don't really want to retry on failed media swaps, this prevents
1449 that. An interesting observation is that permanent failures are not
1450 recorded. */
1451 if (Cnf->Removable == true &&
1452 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
1453 {
1454 // Vf = Version.FileList();
1455 while (Vf.end() == false) Vf++;
1456 StoreFilename = string();
1457 Item::Failed(Message,Cnf);
1458 return;
1459 }
1460
db890fdb 1461 if (QueueNext() == false)
7d8afa39
AL
1462 {
1463 // This is the retry counter
1464 if (Retries != 0 &&
1465 Cnf->LocalOnly == false &&
1466 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
1467 {
1468 Retries--;
1469 Vf = Version.FileList();
1470 if (QueueNext() == true)
1471 return;
1472 }
1473
9dbb421f 1474 StoreFilename = string();
7d8afa39
AL
1475 Item::Failed(Message,Cnf);
1476 }
db890fdb
AL
1477}
1478 /*}}}*/
b3d44315
MV
1479// AcqArchive::IsTrusted - Determine whether this archive comes from a
1480// trusted source /*{{{*/
1481// ---------------------------------------------------------------------
1482bool pkgAcqArchive::IsTrusted()
1483{
1484 return Trusted;
1485}
1486
ab559b35
AL
1487// AcqArchive::Finished - Fetching has finished, tidy up /*{{{*/
1488// ---------------------------------------------------------------------
1489/* */
1490void pkgAcqArchive::Finished()
1491{
1492 if (Status == pkgAcquire::Item::StatDone &&
1493 Complete == true)
1494 return;
1495 StoreFilename = string();
1496}
1497 /*}}}*/
36375005
AL
1498
1499// AcqFile::pkgAcqFile - Constructor /*{{{*/
1500// ---------------------------------------------------------------------
1501/* The file is added to the queue */
8a8feb29 1502pkgAcqFile::pkgAcqFile(pkgAcquire *Owner,string URI,string Hash,
46e00f9d
MV
1503 unsigned long Size,string Dsc,string ShortDesc,
1504 const string &DestDir, const string &DestFilename) :
8a8feb29 1505 Item(Owner), ExpectedHash(Hash)
36375005 1506{
08cfc005
AL
1507 Retries = _config->FindI("Acquire::Retries",0);
1508
46e00f9d
MV
1509 if(!DestFilename.empty())
1510 DestFile = DestFilename;
1511 else if(!DestDir.empty())
1512 DestFile = DestDir + "/" + flNotDir(URI);
1513 else
1514 DestFile = flNotDir(URI);
1515
36375005
AL
1516 // Create the item
1517 Desc.URI = URI;
1518 Desc.Description = Dsc;
1519 Desc.Owner = this;
1520
1521 // Set the short description to the archive component
1522 Desc.ShortDesc = ShortDesc;
1523
1524 // Get the transfer sizes
1525 FileSize = Size;
1526 struct stat Buf;
1527 if (stat(DestFile.c_str(),&Buf) == 0)
1528 {
1529 // Hmm, the partial file is too big, erase it
1530 if ((unsigned)Buf.st_size > Size)
1531 unlink(DestFile.c_str());
1532 else
1533 PartialSize = Buf.st_size;
1534 }
092ae175 1535
36375005
AL
1536 QueueURI(Desc);
1537}
1538 /*}}}*/
1539// AcqFile::Done - Item downloaded OK /*{{{*/
1540// ---------------------------------------------------------------------
1541/* */
495e5cb2 1542void pkgAcqFile::Done(string Message,unsigned long Size,string CalcHash,
459681d3 1543 pkgAcquire::MethodConfig *Cnf)
36375005 1544{
495e5cb2
MV
1545 Item::Done(Message,Size,CalcHash,Cnf);
1546
8a8feb29 1547 // Check the hash
2c941d89 1548 if(!ExpectedHash.empty() && ExpectedHash.toStr() != CalcHash)
b3c39978 1549 {
495e5cb2 1550 Status = StatError;
9498d182 1551 ErrorText = "Hash Sum mismatch";
495e5cb2
MV
1552 Rename(DestFile,DestFile + ".FAILED");
1553 return;
b3c39978
AL
1554 }
1555
36375005
AL
1556 string FileName = LookupTag(Message,"Filename");
1557 if (FileName.empty() == true)
1558 {
1559 Status = StatError;
1560 ErrorText = "Method gave a blank filename";
1561 return;
1562 }
1563
1564 Complete = true;
1565
1566 // The files timestamp matches
1567 if (StringToBool(LookupTag(Message,"IMS-Hit"),false) == true)
1568 return;
1569
1570 // We have to copy it into place
1571 if (FileName != DestFile)
1572 {
1573 Local = true;
459681d3
AL
1574 if (_config->FindB("Acquire::Source-Symlinks",true) == false ||
1575 Cnf->Removable == true)
917ae805
AL
1576 {
1577 Desc.URI = "copy:" + FileName;
1578 QueueURI(Desc);
1579 return;
1580 }
1581
83ab33fc
AL
1582 // Erase the file if it is a symlink so we can overwrite it
1583 struct stat St;
1584 if (lstat(DestFile.c_str(),&St) == 0)
1585 {
1586 if (S_ISLNK(St.st_mode) != 0)
1587 unlink(DestFile.c_str());
1588 }
1589
1590 // Symlink the file
917ae805
AL
1591 if (symlink(FileName.c_str(),DestFile.c_str()) != 0)
1592 {
83ab33fc 1593 ErrorText = "Link to " + DestFile + " failure ";
917ae805
AL
1594 Status = StatError;
1595 Complete = false;
1596 }
36375005
AL
1597 }
1598}
1599 /*}}}*/
08cfc005
AL
1600// AcqFile::Failed - Failure handler /*{{{*/
1601// ---------------------------------------------------------------------
1602/* Here we try other sources */
1603void pkgAcqFile::Failed(string Message,pkgAcquire::MethodConfig *Cnf)
1604{
1605 ErrorText = LookupTag(Message,"Message");
1606
1607 // This is the retry counter
1608 if (Retries != 0 &&
1609 Cnf->LocalOnly == false &&
1610 StringToBool(LookupTag(Message,"Transient-Failure"),false) == true)
1611 {
1612 Retries--;
1613 QueueURI(Desc);
1614 return;
1615 }
1616
1617 Item::Failed(Message,Cnf);
1618}
1619 /*}}}*/