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