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