]>
Commit | Line | Data |
---|---|---|
1 | // -*- mode: cpp; mode: fold -*- | |
2 | // Description /*{{{*/ | |
3 | // $Id: acquire-item.cc,v 1.46.2.9 2004/01/16 18:51:11 mdz Exp $ | |
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. | |
12 | ||
13 | ##################################################################### */ | |
14 | /*}}}*/ | |
15 | // Include Files /*{{{*/ | |
16 | #include <config.h> | |
17 | ||
18 | #include <apt-pkg/acquire-item.h> | |
19 | #include <apt-pkg/configuration.h> | |
20 | #include <apt-pkg/aptconfiguration.h> | |
21 | #include <apt-pkg/sourcelist.h> | |
22 | #include <apt-pkg/error.h> | |
23 | #include <apt-pkg/strutl.h> | |
24 | #include <apt-pkg/fileutl.h> | |
25 | #include <apt-pkg/sha1.h> | |
26 | #include <apt-pkg/tagfile.h> | |
27 | #include <apt-pkg/indexrecords.h> | |
28 | #include <apt-pkg/acquire.h> | |
29 | #include <apt-pkg/hashes.h> | |
30 | #include <apt-pkg/indexfile.h> | |
31 | #include <apt-pkg/pkgcache.h> | |
32 | #include <apt-pkg/cacheiterators.h> | |
33 | #include <apt-pkg/pkgrecords.h> | |
34 | ||
35 | #include <stddef.h> | |
36 | #include <stdlib.h> | |
37 | #include <string.h> | |
38 | #include <iostream> | |
39 | #include <vector> | |
40 | #include <sys/stat.h> | |
41 | #include <unistd.h> | |
42 | #include <errno.h> | |
43 | #include <string> | |
44 | #include <sstream> | |
45 | #include <stdio.h> | |
46 | #include <ctime> | |
47 | #include <sys/types.h> | |
48 | #include <pwd.h> | |
49 | #include <grp.h> | |
50 | ||
51 | #include <apti18n.h> | |
52 | /*}}}*/ | |
53 | ||
54 | using namespace std; | |
55 | ||
56 | static void printHashSumComparision(std::string const &URI, HashStringList const &Expected, HashStringList const &Actual) /*{{{*/ | |
57 | { | |
58 | if (_config->FindB("Debug::Acquire::HashSumMismatch", false) == false) | |
59 | return; | |
60 | std::cerr << std::endl << URI << ":" << std::endl << " Expected Hash: " << std::endl; | |
61 | for (HashStringList::const_iterator hs = Expected.begin(); hs != Expected.end(); ++hs) | |
62 | std::cerr << "\t- " << hs->toStr() << std::endl; | |
63 | std::cerr << " Actual Hash: " << std::endl; | |
64 | for (HashStringList::const_iterator hs = Actual.begin(); hs != Actual.end(); ++hs) | |
65 | std::cerr << "\t- " << hs->toStr() << std::endl; | |
66 | } | |
67 | /*}}}*/ | |
68 | static void ChangeOwnerAndPermissionOfFile(char const * const requester, char const * const file, char const * const user, char const * const group, mode_t const mode) | |
69 | { | |
70 | // ensure the file is owned by root and has good permissions | |
71 | struct passwd const * const pw = getpwnam(user); | |
72 | struct group const * const gr = getgrnam(group); | |
73 | if (getuid() == 0) // if we aren't root, we can't chown, so don't try it | |
74 | { | |
75 | if (pw != NULL && gr != NULL && chown(file, pw->pw_uid, gr->gr_gid) != 0) | |
76 | _error->WarningE(requester, "chown to %s:%s of file %s failed", user, group, file); | |
77 | } | |
78 | if (chmod(file, mode) != 0) | |
79 | _error->WarningE(requester, "chmod 0%o of file %s failed", mode, file); | |
80 | } | |
81 | static std::string GetPartialFileName(std::string const &file) | |
82 | { | |
83 | std::string DestFile = _config->FindDir("Dir::State::lists") + "partial/"; | |
84 | DestFile += file; | |
85 | return DestFile; | |
86 | } | |
87 | static std::string GetPartialFileNameFromURI(std::string const &uri) | |
88 | { | |
89 | return GetPartialFileName(URItoFileName(uri)); | |
90 | } | |
91 | ||
92 | ||
93 | // Acquire::Item::Item - Constructor /*{{{*/ | |
94 | #if __GNUC__ >= 4 | |
95 | #pragma GCC diagnostic push | |
96 | #pragma GCC diagnostic ignored "-Wdeprecated-declarations" | |
97 | #endif | |
98 | pkgAcquire::Item::Item(pkgAcquire *Owner, | |
99 | HashStringList const &ExpectedHashes, | |
100 | pkgAcqMetaBase *TransactionManager) | |
101 | : Owner(Owner), FileSize(0), PartialSize(0), Mode(0), ID(0), Complete(false), | |
102 | Local(false), QueueCounter(0), TransactionManager(TransactionManager), | |
103 | ExpectedAdditionalItems(0), ExpectedHashes(ExpectedHashes) | |
104 | { | |
105 | Owner->Add(this); | |
106 | Status = StatIdle; | |
107 | if(TransactionManager != NULL) | |
108 | TransactionManager->Add(this); | |
109 | } | |
110 | #if __GNUC__ >= 4 | |
111 | #pragma GCC diagnostic pop | |
112 | #endif | |
113 | /*}}}*/ | |
114 | // Acquire::Item::~Item - Destructor /*{{{*/ | |
115 | // --------------------------------------------------------------------- | |
116 | /* */ | |
117 | pkgAcquire::Item::~Item() | |
118 | { | |
119 | Owner->Remove(this); | |
120 | } | |
121 | /*}}}*/ | |
122 | // Acquire::Item::Failed - Item failed to download /*{{{*/ | |
123 | // --------------------------------------------------------------------- | |
124 | /* We return to an idle state if there are still other queues that could | |
125 | fetch this object */ | |
126 | void pkgAcquire::Item::Failed(string Message,pkgAcquire::MethodConfig *Cnf) | |
127 | { | |
128 | if(ErrorText == "") | |
129 | ErrorText = LookupTag(Message,"Message"); | |
130 | UsedMirror = LookupTag(Message,"UsedMirror"); | |
131 | if (QueueCounter <= 1) | |
132 | { | |
133 | /* This indicates that the file is not available right now but might | |
134 | be sometime later. If we do a retry cycle then this should be | |
135 | retried [CDROMs] */ | |
136 | if (Cnf != NULL && Cnf->LocalOnly == true && | |
137 | StringToBool(LookupTag(Message,"Transient-Failure"),false) == true) | |
138 | { | |
139 | Status = StatIdle; | |
140 | Dequeue(); | |
141 | return; | |
142 | } | |
143 | ||
144 | Status = StatError; | |
145 | Complete = false; | |
146 | Dequeue(); | |
147 | } | |
148 | else | |
149 | Status = StatIdle; | |
150 | ||
151 | // check fail reason | |
152 | string FailReason = LookupTag(Message, "FailReason"); | |
153 | if(FailReason == "MaximumSizeExceeded") | |
154 | Rename(DestFile, DestFile+".FAILED"); | |
155 | ||
156 | // report mirror failure back to LP if we actually use a mirror | |
157 | if(FailReason.size() != 0) | |
158 | ReportMirrorFailure(FailReason); | |
159 | else | |
160 | ReportMirrorFailure(ErrorText); | |
161 | } | |
162 | /*}}}*/ | |
163 | // Acquire::Item::Start - Item has begun to download /*{{{*/ | |
164 | // --------------------------------------------------------------------- | |
165 | /* Stash status and the file size. Note that setting Complete means | |
166 | sub-phases of the acquire process such as decompresion are operating */ | |
167 | void pkgAcquire::Item::Start(string /*Message*/,unsigned long long Size) | |
168 | { | |
169 | Status = StatFetching; | |
170 | if (FileSize == 0 && Complete == false) | |
171 | FileSize = Size; | |
172 | } | |
173 | /*}}}*/ | |
174 | // Acquire::Item::Done - Item downloaded OK /*{{{*/ | |
175 | // --------------------------------------------------------------------- | |
176 | /* */ | |
177 | void pkgAcquire::Item::Done(string Message,unsigned long long Size,HashStringList const &/*Hash*/, | |
178 | pkgAcquire::MethodConfig * /*Cnf*/) | |
179 | { | |
180 | // We just downloaded something.. | |
181 | string FileName = LookupTag(Message,"Filename"); | |
182 | UsedMirror = LookupTag(Message,"UsedMirror"); | |
183 | if (Complete == false && !Local && FileName == DestFile) | |
184 | { | |
185 | if (Owner->Log != 0) | |
186 | Owner->Log->Fetched(Size,atoi(LookupTag(Message,"Resume-Point","0").c_str())); | |
187 | } | |
188 | ||
189 | if (FileSize == 0) | |
190 | FileSize= Size; | |
191 | Status = StatDone; | |
192 | ErrorText = string(); | |
193 | Owner->Dequeue(this); | |
194 | } | |
195 | /*}}}*/ | |
196 | // Acquire::Item::Rename - Rename a file /*{{{*/ | |
197 | // --------------------------------------------------------------------- | |
198 | /* This helper function is used by a lot of item methods as their final | |
199 | step */ | |
200 | bool pkgAcquire::Item::Rename(string From,string To) | |
201 | { | |
202 | if (rename(From.c_str(),To.c_str()) != 0) | |
203 | { | |
204 | char S[300]; | |
205 | snprintf(S,sizeof(S),_("rename failed, %s (%s -> %s)."),strerror(errno), | |
206 | From.c_str(),To.c_str()); | |
207 | Status = StatError; | |
208 | ErrorText += S; | |
209 | return false; | |
210 | } | |
211 | return true; | |
212 | } | |
213 | /*}}}*/ | |
214 | ||
215 | void pkgAcquire::Item::QueueURI(ItemDesc &Item) | |
216 | { | |
217 | if (RealFileExists(DestFile)) | |
218 | ChangeOwnerAndPermissionOfFile("GetPartialFileName", DestFile.c_str(), "_apt", "root", 0600); | |
219 | Owner->Enqueue(Item); | |
220 | } | |
221 | void pkgAcquire::Item::Dequeue() | |
222 | { | |
223 | Owner->Dequeue(this); | |
224 | } | |
225 | ||
226 | bool pkgAcquire::Item::RenameOnError(pkgAcquire::Item::RenameOnErrorState const error)/*{{{*/ | |
227 | { | |
228 | if(FileExists(DestFile)) | |
229 | Rename(DestFile, DestFile + ".FAILED"); | |
230 | ||
231 | switch (error) | |
232 | { | |
233 | case HashSumMismatch: | |
234 | ErrorText = _("Hash Sum mismatch"); | |
235 | Status = StatAuthError; | |
236 | ReportMirrorFailure("HashChecksumFailure"); | |
237 | break; | |
238 | case SizeMismatch: | |
239 | ErrorText = _("Size mismatch"); | |
240 | Status = StatAuthError; | |
241 | ReportMirrorFailure("SizeFailure"); | |
242 | break; | |
243 | case InvalidFormat: | |
244 | ErrorText = _("Invalid file format"); | |
245 | Status = StatError; | |
246 | // do not report as usually its not the mirrors fault, but Portal/Proxy | |
247 | break; | |
248 | case SignatureError: | |
249 | ErrorText = _("Signature error"); | |
250 | Status = StatError; | |
251 | break; | |
252 | case NotClearsigned: | |
253 | ErrorText = _("Does not start with a cleartext signature"); | |
254 | Status = StatError; | |
255 | break; | |
256 | } | |
257 | return false; | |
258 | } | |
259 | /*}}}*/ | |
260 | void pkgAcquire::Item::SetActiveSubprocess(const std::string &subprocess)/*{{{*/ | |
261 | { | |
262 | ActiveSubprocess = subprocess; | |
263 | #if __GNUC__ >= 4 | |
264 | #pragma GCC diagnostic push | |
265 | #pragma GCC diagnostic ignored "-Wdeprecated-declarations" | |
266 | #endif | |
267 | Mode = ActiveSubprocess.c_str(); | |
268 | #if __GNUC__ >= 4 | |
269 | #pragma GCC diagnostic pop | |
270 | #endif | |
271 | } | |
272 | /*}}}*/ | |
273 | // Acquire::Item::ReportMirrorFailure /*{{{*/ | |
274 | // --------------------------------------------------------------------- | |
275 | void pkgAcquire::Item::ReportMirrorFailure(string FailCode) | |
276 | { | |
277 | // we only act if a mirror was used at all | |
278 | if(UsedMirror.empty()) | |
279 | return; | |
280 | #if 0 | |
281 | std::cerr << "\nReportMirrorFailure: " | |
282 | << UsedMirror | |
283 | << " Uri: " << DescURI() | |
284 | << " FailCode: " | |
285 | << FailCode << std::endl; | |
286 | #endif | |
287 | const char *Args[40]; | |
288 | unsigned int i = 0; | |
289 | string report = _config->Find("Methods::Mirror::ProblemReporting", | |
290 | "/usr/lib/apt/apt-report-mirror-failure"); | |
291 | if(!FileExists(report)) | |
292 | return; | |
293 | Args[i++] = report.c_str(); | |
294 | Args[i++] = UsedMirror.c_str(); | |
295 | Args[i++] = DescURI().c_str(); | |
296 | Args[i++] = FailCode.c_str(); | |
297 | Args[i++] = NULL; | |
298 | pid_t pid = ExecFork(); | |
299 | if(pid < 0) | |
300 | { | |
301 | _error->Error("ReportMirrorFailure Fork failed"); | |
302 | return; | |
303 | } | |
304 | else if(pid == 0) | |
305 | { | |
306 | execvp(Args[0], (char**)Args); | |
307 | std::cerr << "Could not exec " << Args[0] << std::endl; | |
308 | _exit(100); | |
309 | } | |
310 | if(!ExecWait(pid, "report-mirror-failure")) | |
311 | { | |
312 | _error->Warning("Couldn't report problem to '%s'", | |
313 | _config->Find("Methods::Mirror::ProblemReporting").c_str()); | |
314 | } | |
315 | } | |
316 | /*}}}*/ | |
317 | // AcqDiffIndex::AcqDiffIndex - Constructor /*{{{*/ | |
318 | // --------------------------------------------------------------------- | |
319 | /* Get the DiffIndex file first and see if there are patches available | |
320 | * If so, create a pkgAcqIndexDiffs fetcher that will get and apply the | |
321 | * patches. If anything goes wrong in that process, it will fall back to | |
322 | * the original packages file | |
323 | */ | |
324 | pkgAcqDiffIndex::pkgAcqDiffIndex(pkgAcquire *Owner, | |
325 | pkgAcqMetaBase *TransactionManager, | |
326 | IndexTarget const * const Target, | |
327 | HashStringList const &ExpectedHashes, | |
328 | indexRecords *MetaIndexParser) | |
329 | : pkgAcqBaseIndex(Owner, TransactionManager, Target, ExpectedHashes, | |
330 | MetaIndexParser), PackagesFileReadyInPartial(false) | |
331 | { | |
332 | ||
333 | Debug = _config->FindB("Debug::pkgAcquire::Diffs",false); | |
334 | ||
335 | RealURI = Target->URI; | |
336 | Desc.Owner = this; | |
337 | Desc.Description = Target->Description + ".diff/Index"; | |
338 | Desc.ShortDesc = Target->ShortDesc; | |
339 | Desc.URI = Target->URI + ".diff/Index"; | |
340 | ||
341 | DestFile = GetPartialFileNameFromURI(Desc.URI); | |
342 | ||
343 | if(Debug) | |
344 | std::clog << "pkgAcqDiffIndex: " << Desc.URI << std::endl; | |
345 | ||
346 | // look for the current package file | |
347 | CurrentPackagesFile = _config->FindDir("Dir::State::lists"); | |
348 | CurrentPackagesFile += URItoFileName(RealURI); | |
349 | ||
350 | // FIXME: this file:/ check is a hack to prevent fetching | |
351 | // from local sources. this is really silly, and | |
352 | // should be fixed cleanly as soon as possible | |
353 | if(!FileExists(CurrentPackagesFile) || | |
354 | Desc.URI.substr(0,strlen("file:/")) == "file:/") | |
355 | { | |
356 | // we don't have a pkg file or we don't want to queue | |
357 | Failed("No index file, local or canceld by user", NULL); | |
358 | return; | |
359 | } | |
360 | ||
361 | if(Debug) | |
362 | std::clog << "pkgAcqDiffIndex::pkgAcqDiffIndex(): " | |
363 | << CurrentPackagesFile << std::endl; | |
364 | ||
365 | QueueURI(Desc); | |
366 | ||
367 | } | |
368 | /*}}}*/ | |
369 | // AcqIndex::Custom600Headers - Insert custom request headers /*{{{*/ | |
370 | // --------------------------------------------------------------------- | |
371 | /* The only header we use is the last-modified header. */ | |
372 | string pkgAcqDiffIndex::Custom600Headers() const | |
373 | { | |
374 | string Final = _config->FindDir("Dir::State::lists"); | |
375 | Final += URItoFileName(Desc.URI); | |
376 | ||
377 | if(Debug) | |
378 | std::clog << "Custom600Header-IMS: " << Final << std::endl; | |
379 | ||
380 | struct stat Buf; | |
381 | if (stat(Final.c_str(),&Buf) != 0) | |
382 | return "\nIndex-File: true"; | |
383 | ||
384 | return "\nIndex-File: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime); | |
385 | } | |
386 | /*}}}*/ | |
387 | bool pkgAcqDiffIndex::ParseDiffIndex(string IndexDiffFile) /*{{{*/ | |
388 | { | |
389 | // failing here is fine: our caller will take care of trying to | |
390 | // get the complete file if patching fails | |
391 | if(Debug) | |
392 | std::clog << "pkgAcqDiffIndex::ParseIndexDiff() " << IndexDiffFile | |
393 | << std::endl; | |
394 | ||
395 | FileFd Fd(IndexDiffFile,FileFd::ReadOnly); | |
396 | pkgTagFile TF(&Fd); | |
397 | if (_error->PendingError() == true) | |
398 | return false; | |
399 | ||
400 | pkgTagSection Tags; | |
401 | if(unlikely(TF.Step(Tags) == false)) | |
402 | return false; | |
403 | ||
404 | HashStringList ServerHashes; | |
405 | unsigned long long ServerSize = 0; | |
406 | ||
407 | for (char const * const * type = HashString::SupportedHashes(); *type != NULL; ++type) | |
408 | { | |
409 | std::string tagname = *type; | |
410 | tagname.append("-Current"); | |
411 | std::string const tmp = Tags.FindS(tagname.c_str()); | |
412 | if (tmp.empty() == true) | |
413 | continue; | |
414 | ||
415 | string hash; | |
416 | unsigned long long size; | |
417 | std::stringstream ss(tmp); | |
418 | ss >> hash >> size; | |
419 | if (unlikely(hash.empty() == true)) | |
420 | continue; | |
421 | if (unlikely(ServerSize != 0 && ServerSize != size)) | |
422 | continue; | |
423 | ServerHashes.push_back(HashString(*type, hash)); | |
424 | ServerSize = size; | |
425 | } | |
426 | ||
427 | if (ServerHashes.usable() == false) | |
428 | { | |
429 | if (Debug == true) | |
430 | std::clog << "pkgAcqDiffIndex: " << IndexDiffFile << ": Did not find a good hashsum in the index" << std::endl; | |
431 | return false; | |
432 | } | |
433 | ||
434 | if (ServerHashes != HashSums()) | |
435 | { | |
436 | if (Debug == true) | |
437 | { | |
438 | std::clog << "pkgAcqDiffIndex: " << IndexDiffFile << ": Index has different hashes than parser, probably older, so fail pdiffing" << std::endl; | |
439 | printHashSumComparision(CurrentPackagesFile, ServerHashes, HashSums()); | |
440 | } | |
441 | return false; | |
442 | } | |
443 | ||
444 | if (ServerHashes.VerifyFile(CurrentPackagesFile) == true) | |
445 | { | |
446 | // we have the same sha1 as the server so we are done here | |
447 | if(Debug) | |
448 | std::clog << "pkgAcqDiffIndex: Package file " << CurrentPackagesFile << " is up-to-date" << std::endl; | |
449 | ||
450 | // list cleanup needs to know that this file as well as the already | |
451 | // present index is ours, so we create an empty diff to save it for us | |
452 | new pkgAcqIndexDiffs(Owner, TransactionManager, Target, | |
453 | ExpectedHashes, MetaIndexParser); | |
454 | return true; | |
455 | } | |
456 | ||
457 | FileFd fd(CurrentPackagesFile, FileFd::ReadOnly); | |
458 | Hashes LocalHashesCalc; | |
459 | LocalHashesCalc.AddFD(fd); | |
460 | HashStringList const LocalHashes = LocalHashesCalc.GetHashStringList(); | |
461 | ||
462 | if(Debug) | |
463 | std::clog << "Server-Current: " << ServerHashes.find(NULL)->toStr() << " and we start at " | |
464 | << fd.Name() << " " << fd.FileSize() << " " << LocalHashes.find(NULL)->toStr() << std::endl; | |
465 | ||
466 | // parse all of (provided) history | |
467 | vector<DiffInfo> available_patches; | |
468 | bool firstAcceptedHashes = true; | |
469 | for (char const * const * type = HashString::SupportedHashes(); *type != NULL; ++type) | |
470 | { | |
471 | if (LocalHashes.find(*type) == NULL) | |
472 | continue; | |
473 | ||
474 | std::string tagname = *type; | |
475 | tagname.append("-History"); | |
476 | std::string const tmp = Tags.FindS(tagname.c_str()); | |
477 | if (tmp.empty() == true) | |
478 | continue; | |
479 | ||
480 | string hash, filename; | |
481 | unsigned long long size; | |
482 | std::stringstream ss(tmp); | |
483 | ||
484 | while (ss >> hash >> size >> filename) | |
485 | { | |
486 | if (unlikely(hash.empty() == true || filename.empty() == true)) | |
487 | continue; | |
488 | ||
489 | // see if we have a record for this file already | |
490 | std::vector<DiffInfo>::iterator cur = available_patches.begin(); | |
491 | for (; cur != available_patches.end(); ++cur) | |
492 | { | |
493 | if (cur->file != filename || unlikely(cur->result_size != size)) | |
494 | continue; | |
495 | cur->result_hashes.push_back(HashString(*type, hash)); | |
496 | break; | |
497 | } | |
498 | if (cur != available_patches.end()) | |
499 | continue; | |
500 | if (firstAcceptedHashes == true) | |
501 | { | |
502 | DiffInfo next; | |
503 | next.file = filename; | |
504 | next.result_hashes.push_back(HashString(*type, hash)); | |
505 | next.result_size = size; | |
506 | next.patch_size = 0; | |
507 | available_patches.push_back(next); | |
508 | } | |
509 | else | |
510 | { | |
511 | if (Debug == true) | |
512 | std::clog << "pkgAcqDiffIndex: " << IndexDiffFile << ": File " << filename | |
513 | << " wasn't in the list for the first parsed hash! (history)" << std::endl; | |
514 | break; | |
515 | } | |
516 | } | |
517 | firstAcceptedHashes = false; | |
518 | } | |
519 | ||
520 | if (unlikely(available_patches.empty() == true)) | |
521 | { | |
522 | if (Debug) | |
523 | std::clog << "pkgAcqDiffIndex: " << IndexDiffFile << ": " | |
524 | << "Couldn't find any patches for the patch series." << std::endl; | |
525 | return false; | |
526 | } | |
527 | ||
528 | for (char const * const * type = HashString::SupportedHashes(); *type != NULL; ++type) | |
529 | { | |
530 | if (LocalHashes.find(*type) == NULL) | |
531 | continue; | |
532 | ||
533 | std::string tagname = *type; | |
534 | tagname.append("-Patches"); | |
535 | std::string const tmp = Tags.FindS(tagname.c_str()); | |
536 | if (tmp.empty() == true) | |
537 | continue; | |
538 | ||
539 | string hash, filename; | |
540 | unsigned long long size; | |
541 | std::stringstream ss(tmp); | |
542 | ||
543 | while (ss >> hash >> size >> filename) | |
544 | { | |
545 | if (unlikely(hash.empty() == true || filename.empty() == true)) | |
546 | continue; | |
547 | ||
548 | // see if we have a record for this file already | |
549 | std::vector<DiffInfo>::iterator cur = available_patches.begin(); | |
550 | for (; cur != available_patches.end(); ++cur) | |
551 | { | |
552 | if (cur->file != filename) | |
553 | continue; | |
554 | if (unlikely(cur->patch_size != 0 && cur->patch_size != size)) | |
555 | continue; | |
556 | cur->patch_hashes.push_back(HashString(*type, hash)); | |
557 | cur->patch_size = size; | |
558 | break; | |
559 | } | |
560 | if (cur != available_patches.end()) | |
561 | continue; | |
562 | if (Debug == true) | |
563 | std::clog << "pkgAcqDiffIndex: " << IndexDiffFile << ": File " << filename | |
564 | << " wasn't in the list for the first parsed hash! (patches)" << std::endl; | |
565 | break; | |
566 | } | |
567 | } | |
568 | ||
569 | bool foundStart = false; | |
570 | for (std::vector<DiffInfo>::iterator cur = available_patches.begin(); | |
571 | cur != available_patches.end(); ++cur) | |
572 | { | |
573 | if (LocalHashes != cur->result_hashes) | |
574 | continue; | |
575 | ||
576 | available_patches.erase(available_patches.begin(), cur); | |
577 | foundStart = true; | |
578 | break; | |
579 | } | |
580 | ||
581 | if (foundStart == false || unlikely(available_patches.empty() == true)) | |
582 | { | |
583 | if (Debug) | |
584 | std::clog << "pkgAcqDiffIndex: " << IndexDiffFile << ": " | |
585 | << "Couldn't find the start of the patch series." << std::endl; | |
586 | return false; | |
587 | } | |
588 | ||
589 | // patching with too many files is rather slow compared to a fast download | |
590 | unsigned long const fileLimit = _config->FindI("Acquire::PDiffs::FileLimit", 0); | |
591 | if (fileLimit != 0 && fileLimit < available_patches.size()) | |
592 | { | |
593 | if (Debug) | |
594 | std::clog << "Need " << available_patches.size() << " diffs (Limit is " << fileLimit | |
595 | << ") so fallback to complete download" << std::endl; | |
596 | return false; | |
597 | } | |
598 | ||
599 | // calculate the size of all patches we have to get | |
600 | // note that all sizes are uncompressed, while we download compressed files | |
601 | unsigned long long patchesSize = 0; | |
602 | for (std::vector<DiffInfo>::const_iterator cur = available_patches.begin(); | |
603 | cur != available_patches.end(); ++cur) | |
604 | patchesSize += cur->patch_size; | |
605 | unsigned long long const sizeLimit = ServerSize * _config->FindI("Acquire::PDiffs::SizeLimit", 100); | |
606 | if (false && sizeLimit > 0 && (sizeLimit/100) < patchesSize) | |
607 | { | |
608 | if (Debug) | |
609 | std::clog << "Need " << patchesSize << " bytes (Limit is " << sizeLimit/100 | |
610 | << ") so fallback to complete download" << std::endl; | |
611 | return false; | |
612 | } | |
613 | ||
614 | // FIXME: make this use the method | |
615 | PackagesFileReadyInPartial = true; | |
616 | std::string const Partial = GetPartialFileNameFromURI(RealURI); | |
617 | ||
618 | FileFd From(CurrentPackagesFile, FileFd::ReadOnly); | |
619 | FileFd To(Partial, FileFd::WriteEmpty); | |
620 | if(CopyFile(From, To) == false) | |
621 | return _error->Errno("CopyFile", "failed to copy"); | |
622 | ||
623 | if(Debug) | |
624 | std::cerr << "Done copying " << CurrentPackagesFile | |
625 | << " -> " << Partial | |
626 | << std::endl; | |
627 | ||
628 | // we have something, queue the diffs | |
629 | string::size_type const last_space = Description.rfind(" "); | |
630 | if(last_space != string::npos) | |
631 | Description.erase(last_space, Description.size()-last_space); | |
632 | ||
633 | /* decide if we should download patches one by one or in one go: | |
634 | The first is good if the server merges patches, but many don't so client | |
635 | based merging can be attempt in which case the second is better. | |
636 | "bad things" will happen if patches are merged on the server, | |
637 | but client side merging is attempt as well */ | |
638 | bool pdiff_merge = _config->FindB("Acquire::PDiffs::Merge", true); | |
639 | if (pdiff_merge == true) | |
640 | { | |
641 | // reprepro adds this flag if it has merged patches on the server | |
642 | std::string const precedence = Tags.FindS("X-Patch-Precedence"); | |
643 | pdiff_merge = (precedence != "merged"); | |
644 | } | |
645 | ||
646 | if (pdiff_merge == false) | |
647 | { | |
648 | new pkgAcqIndexDiffs(Owner, TransactionManager, Target, ExpectedHashes, | |
649 | MetaIndexParser, available_patches); | |
650 | } | |
651 | else | |
652 | { | |
653 | std::vector<pkgAcqIndexMergeDiffs*> *diffs = new std::vector<pkgAcqIndexMergeDiffs*>(available_patches.size()); | |
654 | for(size_t i = 0; i < available_patches.size(); ++i) | |
655 | (*diffs)[i] = new pkgAcqIndexMergeDiffs(Owner, TransactionManager, | |
656 | Target, | |
657 | ExpectedHashes, | |
658 | MetaIndexParser, | |
659 | available_patches[i], | |
660 | diffs); | |
661 | } | |
662 | ||
663 | Complete = false; | |
664 | Status = StatDone; | |
665 | Dequeue(); | |
666 | return true; | |
667 | } | |
668 | /*}}}*/ | |
669 | void pkgAcqDiffIndex::Failed(string Message,pkgAcquire::MethodConfig * Cnf)/*{{{*/ | |
670 | { | |
671 | if(Debug) | |
672 | std::clog << "pkgAcqDiffIndex failed: " << Desc.URI << " with " << Message << std::endl | |
673 | << "Falling back to normal index file acquire" << std::endl; | |
674 | ||
675 | new pkgAcqIndex(Owner, TransactionManager, Target, ExpectedHashes, MetaIndexParser); | |
676 | ||
677 | Item::Failed(Message,Cnf); | |
678 | Status = StatDone; | |
679 | } | |
680 | /*}}}*/ | |
681 | void pkgAcqDiffIndex::Done(string Message,unsigned long long Size,HashStringList const &Hashes, /*{{{*/ | |
682 | pkgAcquire::MethodConfig *Cnf) | |
683 | { | |
684 | if(Debug) | |
685 | std::clog << "pkgAcqDiffIndex::Done(): " << Desc.URI << std::endl; | |
686 | ||
687 | Item::Done(Message, Size, Hashes, Cnf); | |
688 | ||
689 | // verify the index target | |
690 | if(Target && Target->MetaKey != "" && MetaIndexParser && Hashes.usable()) | |
691 | { | |
692 | std::string IndexMetaKey = Target->MetaKey + ".diff/Index"; | |
693 | indexRecords::checkSum *Record = MetaIndexParser->Lookup(IndexMetaKey); | |
694 | if(Record && Record->Hashes.usable() && Hashes != Record->Hashes) | |
695 | { | |
696 | RenameOnError(HashSumMismatch); | |
697 | printHashSumComparision(RealURI, Record->Hashes, Hashes); | |
698 | Failed(Message, Cnf); | |
699 | return; | |
700 | } | |
701 | ||
702 | } | |
703 | ||
704 | string FinalFile; | |
705 | FinalFile = _config->FindDir("Dir::State::lists"); | |
706 | FinalFile += URItoFileName(Desc.URI); | |
707 | ||
708 | if(StringToBool(LookupTag(Message,"IMS-Hit"),false)) | |
709 | DestFile = FinalFile; | |
710 | ||
711 | if(!ParseDiffIndex(DestFile)) | |
712 | return Failed("Message: Couldn't parse pdiff index", Cnf); | |
713 | ||
714 | // queue for final move | |
715 | TransactionManager->TransactionStageCopy(this, DestFile, FinalFile); | |
716 | ||
717 | Complete = true; | |
718 | Status = StatDone; | |
719 | Dequeue(); | |
720 | return; | |
721 | } | |
722 | /*}}}*/ | |
723 | // AcqIndexDiffs::AcqIndexDiffs - Constructor /*{{{*/ | |
724 | // --------------------------------------------------------------------- | |
725 | /* The package diff is added to the queue. one object is constructed | |
726 | * for each diff and the index | |
727 | */ | |
728 | pkgAcqIndexDiffs::pkgAcqIndexDiffs(pkgAcquire *Owner, | |
729 | pkgAcqMetaBase *TransactionManager, | |
730 | struct IndexTarget const * const Target, | |
731 | HashStringList const &ExpectedHashes, | |
732 | indexRecords *MetaIndexParser, | |
733 | vector<DiffInfo> diffs) | |
734 | : pkgAcqBaseIndex(Owner, TransactionManager, Target, ExpectedHashes, MetaIndexParser), | |
735 | available_patches(diffs) | |
736 | { | |
737 | DestFile = GetPartialFileNameFromURI(Target->URI); | |
738 | ||
739 | Debug = _config->FindB("Debug::pkgAcquire::Diffs",false); | |
740 | ||
741 | RealURI = Target->URI; | |
742 | Desc.Owner = this; | |
743 | Description = Target->Description; | |
744 | Desc.ShortDesc = Target->ShortDesc; | |
745 | ||
746 | if(available_patches.empty() == true) | |
747 | { | |
748 | // we are done (yeah!), check hashes against the final file | |
749 | DestFile = _config->FindDir("Dir::State::lists"); | |
750 | DestFile += URItoFileName(Target->URI); | |
751 | Finish(true); | |
752 | } | |
753 | else | |
754 | { | |
755 | // get the next diff | |
756 | State = StateFetchDiff; | |
757 | QueueNextDiff(); | |
758 | } | |
759 | } | |
760 | /*}}}*/ | |
761 | void pkgAcqIndexDiffs::Failed(string Message,pkgAcquire::MethodConfig * /*Cnf*/)/*{{{*/ | |
762 | { | |
763 | if(Debug) | |
764 | std::clog << "pkgAcqIndexDiffs failed: " << Desc.URI << " with " << Message << std::endl | |
765 | << "Falling back to normal index file acquire" << std::endl; | |
766 | new pkgAcqIndex(Owner, TransactionManager, Target, ExpectedHashes, MetaIndexParser); | |
767 | Finish(); | |
768 | } | |
769 | /*}}}*/ | |
770 | // Finish - helper that cleans the item out of the fetcher queue /*{{{*/ | |
771 | void pkgAcqIndexDiffs::Finish(bool allDone) | |
772 | { | |
773 | if(Debug) | |
774 | std::clog << "pkgAcqIndexDiffs::Finish(): " | |
775 | << allDone << " " | |
776 | << Desc.URI << std::endl; | |
777 | ||
778 | // we restore the original name, this is required, otherwise | |
779 | // the file will be cleaned | |
780 | if(allDone) | |
781 | { | |
782 | if(HashSums().usable() && !HashSums().VerifyFile(DestFile)) | |
783 | { | |
784 | RenameOnError(HashSumMismatch); | |
785 | Dequeue(); | |
786 | return; | |
787 | } | |
788 | ||
789 | // queue for copy | |
790 | std::string FinalFile = _config->FindDir("Dir::State::lists"); | |
791 | FinalFile += URItoFileName(RealURI); | |
792 | TransactionManager->TransactionStageCopy(this, DestFile, FinalFile); | |
793 | ||
794 | // this is for the "real" finish | |
795 | Complete = true; | |
796 | Status = StatDone; | |
797 | Dequeue(); | |
798 | if(Debug) | |
799 | std::clog << "\n\nallDone: " << DestFile << "\n" << std::endl; | |
800 | return; | |
801 | } | |
802 | ||
803 | if(Debug) | |
804 | std::clog << "Finishing: " << Desc.URI << std::endl; | |
805 | Complete = false; | |
806 | Status = StatDone; | |
807 | Dequeue(); | |
808 | return; | |
809 | } | |
810 | /*}}}*/ | |
811 | bool pkgAcqIndexDiffs::QueueNextDiff() /*{{{*/ | |
812 | { | |
813 | // calc sha1 of the just patched file | |
814 | std::string const FinalFile = GetPartialFileNameFromURI(RealURI); | |
815 | ||
816 | if(!FileExists(FinalFile)) | |
817 | { | |
818 | Failed("Message: No FinalFile " + FinalFile + " available", NULL); | |
819 | return false; | |
820 | } | |
821 | ||
822 | FileFd fd(FinalFile, FileFd::ReadOnly); | |
823 | Hashes LocalHashesCalc; | |
824 | LocalHashesCalc.AddFD(fd); | |
825 | HashStringList const LocalHashes = LocalHashesCalc.GetHashStringList(); | |
826 | ||
827 | if(Debug) | |
828 | std::clog << "QueueNextDiff: " << FinalFile << " (" << LocalHashes.find(NULL)->toStr() << ")" << std::endl; | |
829 | ||
830 | if (unlikely(LocalHashes.usable() == false || ExpectedHashes.usable() == false)) | |
831 | { | |
832 | Failed("Local/Expected hashes are not usable", NULL); | |
833 | return false; | |
834 | } | |
835 | ||
836 | ||
837 | // final file reached before all patches are applied | |
838 | if(LocalHashes == ExpectedHashes) | |
839 | { | |
840 | Finish(true); | |
841 | return true; | |
842 | } | |
843 | ||
844 | // remove all patches until the next matching patch is found | |
845 | // this requires the Index file to be ordered | |
846 | for(vector<DiffInfo>::iterator I = available_patches.begin(); | |
847 | available_patches.empty() == false && | |
848 | I != available_patches.end() && | |
849 | I->result_hashes != LocalHashes; | |
850 | ++I) | |
851 | { | |
852 | available_patches.erase(I); | |
853 | } | |
854 | ||
855 | // error checking and falling back if no patch was found | |
856 | if(available_patches.empty() == true) | |
857 | { | |
858 | Failed("No patches left to reach target", NULL); | |
859 | return false; | |
860 | } | |
861 | ||
862 | // queue the right diff | |
863 | Desc.URI = RealURI + ".diff/" + available_patches[0].file + ".gz"; | |
864 | Desc.Description = Description + " " + available_patches[0].file + string(".pdiff"); | |
865 | DestFile = GetPartialFileNameFromURI(RealURI + ".diff/" + available_patches[0].file); | |
866 | ||
867 | if(Debug) | |
868 | std::clog << "pkgAcqIndexDiffs::QueueNextDiff(): " << Desc.URI << std::endl; | |
869 | ||
870 | QueueURI(Desc); | |
871 | ||
872 | return true; | |
873 | } | |
874 | /*}}}*/ | |
875 | void pkgAcqIndexDiffs::Done(string Message,unsigned long long Size, HashStringList const &Hashes, /*{{{*/ | |
876 | pkgAcquire::MethodConfig *Cnf) | |
877 | { | |
878 | if(Debug) | |
879 | std::clog << "pkgAcqIndexDiffs::Done(): " << Desc.URI << std::endl; | |
880 | ||
881 | Item::Done(Message, Size, Hashes, Cnf); | |
882 | ||
883 | // FIXME: verify this download too before feeding it to rred | |
884 | std::string const FinalFile = GetPartialFileNameFromURI(RealURI); | |
885 | ||
886 | // success in downloading a diff, enter ApplyDiff state | |
887 | if(State == StateFetchDiff) | |
888 | { | |
889 | FileFd fd(DestFile, FileFd::ReadOnly, FileFd::Gzip); | |
890 | class Hashes LocalHashesCalc; | |
891 | LocalHashesCalc.AddFD(fd); | |
892 | HashStringList const LocalHashes = LocalHashesCalc.GetHashStringList(); | |
893 | ||
894 | if (fd.Size() != available_patches[0].patch_size || | |
895 | available_patches[0].patch_hashes != LocalHashes) | |
896 | { | |
897 | Failed("Patch has Size/Hashsum mismatch", NULL); | |
898 | return; | |
899 | } | |
900 | ||
901 | // rred excepts the patch as $FinalFile.ed | |
902 | Rename(DestFile,FinalFile+".ed"); | |
903 | ||
904 | if(Debug) | |
905 | std::clog << "Sending to rred method: " << FinalFile << std::endl; | |
906 | ||
907 | State = StateApplyDiff; | |
908 | Local = true; | |
909 | Desc.URI = "rred:" + FinalFile; | |
910 | QueueURI(Desc); | |
911 | SetActiveSubprocess("rred"); | |
912 | return; | |
913 | } | |
914 | ||
915 | ||
916 | // success in download/apply a diff, queue next (if needed) | |
917 | if(State == StateApplyDiff) | |
918 | { | |
919 | // remove the just applied patch | |
920 | available_patches.erase(available_patches.begin()); | |
921 | unlink((FinalFile + ".ed").c_str()); | |
922 | ||
923 | // move into place | |
924 | if(Debug) | |
925 | { | |
926 | std::clog << "Moving patched file in place: " << std::endl | |
927 | << DestFile << " -> " << FinalFile << std::endl; | |
928 | } | |
929 | Rename(DestFile,FinalFile); | |
930 | chmod(FinalFile.c_str(),0644); | |
931 | ||
932 | // see if there is more to download | |
933 | if(available_patches.empty() == false) { | |
934 | new pkgAcqIndexDiffs(Owner, TransactionManager, Target, | |
935 | ExpectedHashes, MetaIndexParser, | |
936 | available_patches); | |
937 | return Finish(); | |
938 | } else | |
939 | // update | |
940 | DestFile = FinalFile; | |
941 | return Finish(true); | |
942 | } | |
943 | } | |
944 | /*}}}*/ | |
945 | // AcqIndexMergeDiffs::AcqIndexMergeDiffs - Constructor /*{{{*/ | |
946 | pkgAcqIndexMergeDiffs::pkgAcqIndexMergeDiffs(pkgAcquire *Owner, | |
947 | pkgAcqMetaBase *TransactionManager, | |
948 | struct IndexTarget const * const Target, | |
949 | HashStringList const &ExpectedHashes, | |
950 | indexRecords *MetaIndexParser, | |
951 | DiffInfo const &patch, | |
952 | std::vector<pkgAcqIndexMergeDiffs*> const * const allPatches) | |
953 | : pkgAcqBaseIndex(Owner, TransactionManager, Target, ExpectedHashes, MetaIndexParser), | |
954 | patch(patch), allPatches(allPatches), State(StateFetchDiff) | |
955 | { | |
956 | Debug = _config->FindB("Debug::pkgAcquire::Diffs",false); | |
957 | ||
958 | RealURI = Target->URI; | |
959 | Desc.Owner = this; | |
960 | Description = Target->Description; | |
961 | Desc.ShortDesc = Target->ShortDesc; | |
962 | ||
963 | Desc.URI = RealURI + ".diff/" + patch.file + ".gz"; | |
964 | Desc.Description = Description + " " + patch.file + string(".pdiff"); | |
965 | ||
966 | DestFile = GetPartialFileNameFromURI(RealURI + ".diff/" + patch.file); | |
967 | ||
968 | if(Debug) | |
969 | std::clog << "pkgAcqIndexMergeDiffs: " << Desc.URI << std::endl; | |
970 | ||
971 | QueueURI(Desc); | |
972 | } | |
973 | /*}}}*/ | |
974 | void pkgAcqIndexMergeDiffs::Failed(string Message,pkgAcquire::MethodConfig * Cnf)/*{{{*/ | |
975 | { | |
976 | if(Debug) | |
977 | std::clog << "pkgAcqIndexMergeDiffs failed: " << Desc.URI << " with " << Message << std::endl; | |
978 | ||
979 | Item::Failed(Message,Cnf); | |
980 | Status = StatDone; | |
981 | ||
982 | // check if we are the first to fail, otherwise we are done here | |
983 | State = StateDoneDiff; | |
984 | for (std::vector<pkgAcqIndexMergeDiffs *>::const_iterator I = allPatches->begin(); | |
985 | I != allPatches->end(); ++I) | |
986 | if ((*I)->State == StateErrorDiff) | |
987 | return; | |
988 | ||
989 | // first failure means we should fallback | |
990 | State = StateErrorDiff; | |
991 | std::clog << "Falling back to normal index file acquire" << std::endl; | |
992 | new pkgAcqIndex(Owner, TransactionManager, Target, ExpectedHashes, MetaIndexParser); | |
993 | } | |
994 | /*}}}*/ | |
995 | void pkgAcqIndexMergeDiffs::Done(string Message,unsigned long long Size,HashStringList const &Hashes, /*{{{*/ | |
996 | pkgAcquire::MethodConfig *Cnf) | |
997 | { | |
998 | if(Debug) | |
999 | std::clog << "pkgAcqIndexMergeDiffs::Done(): " << Desc.URI << std::endl; | |
1000 | ||
1001 | Item::Done(Message,Size,Hashes,Cnf); | |
1002 | ||
1003 | // FIXME: verify download before feeding it to rred | |
1004 | string const FinalFile = GetPartialFileNameFromURI(RealURI); | |
1005 | ||
1006 | if (State == StateFetchDiff) | |
1007 | { | |
1008 | FileFd fd(DestFile, FileFd::ReadOnly, FileFd::Gzip); | |
1009 | class Hashes LocalHashesCalc; | |
1010 | LocalHashesCalc.AddFD(fd); | |
1011 | HashStringList const LocalHashes = LocalHashesCalc.GetHashStringList(); | |
1012 | ||
1013 | if (fd.Size() != patch.patch_size || patch.patch_hashes != LocalHashes) | |
1014 | { | |
1015 | Failed("Patch has Size/Hashsum mismatch", NULL); | |
1016 | return; | |
1017 | } | |
1018 | ||
1019 | // rred expects the patch as $FinalFile.ed.$patchname.gz | |
1020 | Rename(DestFile, FinalFile + ".ed." + patch.file + ".gz"); | |
1021 | ||
1022 | // check if this is the last completed diff | |
1023 | State = StateDoneDiff; | |
1024 | for (std::vector<pkgAcqIndexMergeDiffs *>::const_iterator I = allPatches->begin(); | |
1025 | I != allPatches->end(); ++I) | |
1026 | if ((*I)->State != StateDoneDiff) | |
1027 | { | |
1028 | if(Debug) | |
1029 | std::clog << "Not the last done diff in the batch: " << Desc.URI << std::endl; | |
1030 | return; | |
1031 | } | |
1032 | ||
1033 | // this is the last completed diff, so we are ready to apply now | |
1034 | State = StateApplyDiff; | |
1035 | ||
1036 | if(Debug) | |
1037 | std::clog << "Sending to rred method: " << FinalFile << std::endl; | |
1038 | ||
1039 | Local = true; | |
1040 | Desc.URI = "rred:" + FinalFile; | |
1041 | QueueURI(Desc); | |
1042 | SetActiveSubprocess("rred"); | |
1043 | return; | |
1044 | } | |
1045 | // success in download/apply all diffs, clean up | |
1046 | else if (State == StateApplyDiff) | |
1047 | { | |
1048 | // see if we really got the expected file | |
1049 | if(ExpectedHashes.usable() && !ExpectedHashes.VerifyFile(DestFile)) | |
1050 | { | |
1051 | RenameOnError(HashSumMismatch); | |
1052 | return; | |
1053 | } | |
1054 | ||
1055 | ||
1056 | std::string FinalFile = _config->FindDir("Dir::State::lists"); | |
1057 | FinalFile += URItoFileName(RealURI); | |
1058 | ||
1059 | // move the result into place | |
1060 | if(Debug) | |
1061 | std::clog << "Queue patched file in place: " << std::endl | |
1062 | << DestFile << " -> " << FinalFile << std::endl; | |
1063 | ||
1064 | // queue for copy by the transaction manager | |
1065 | TransactionManager->TransactionStageCopy(this, DestFile, FinalFile); | |
1066 | ||
1067 | // ensure the ed's are gone regardless of list-cleanup | |
1068 | for (std::vector<pkgAcqIndexMergeDiffs *>::const_iterator I = allPatches->begin(); | |
1069 | I != allPatches->end(); ++I) | |
1070 | { | |
1071 | std::string const PartialFile = GetPartialFileNameFromURI(RealURI); | |
1072 | std::string patch = PartialFile + ".ed." + (*I)->patch.file + ".gz"; | |
1073 | std::cerr << patch << std::endl; | |
1074 | unlink(patch.c_str()); | |
1075 | } | |
1076 | ||
1077 | // all set and done | |
1078 | Complete = true; | |
1079 | if(Debug) | |
1080 | std::clog << "allDone: " << DestFile << "\n" << std::endl; | |
1081 | } | |
1082 | } | |
1083 | /*}}}*/ | |
1084 | // AcqBaseIndex::VerifyHashByMetaKey - verify hash for the given metakey /*{{{*/ | |
1085 | bool pkgAcqBaseIndex::VerifyHashByMetaKey(HashStringList const &Hashes) | |
1086 | { | |
1087 | if(MetaKey != "" && Hashes.usable()) | |
1088 | { | |
1089 | indexRecords::checkSum *Record = MetaIndexParser->Lookup(MetaKey); | |
1090 | if(Record && Record->Hashes.usable() && Hashes != Record->Hashes) | |
1091 | { | |
1092 | printHashSumComparision(RealURI, Record->Hashes, Hashes); | |
1093 | return false; | |
1094 | } | |
1095 | } | |
1096 | return true; | |
1097 | } | |
1098 | /*}}}*/ | |
1099 | // AcqIndex::AcqIndex - Constructor /*{{{*/ | |
1100 | // --------------------------------------------------------------------- | |
1101 | /* The package file is added to the queue and a second class is | |
1102 | instantiated to fetch the revision file */ | |
1103 | pkgAcqIndex::pkgAcqIndex(pkgAcquire *Owner, | |
1104 | string URI,string URIDesc,string ShortDesc, | |
1105 | HashStringList const &ExpectedHash) | |
1106 | : pkgAcqBaseIndex(Owner, 0, NULL, ExpectedHash, NULL) | |
1107 | { | |
1108 | RealURI = URI; | |
1109 | ||
1110 | AutoSelectCompression(); | |
1111 | Init(URI, URIDesc, ShortDesc); | |
1112 | ||
1113 | if(_config->FindB("Debug::Acquire::Transaction", false) == true) | |
1114 | std::clog << "New pkgIndex with TransactionManager " | |
1115 | << TransactionManager << std::endl; | |
1116 | } | |
1117 | /*}}}*/ | |
1118 | // AcqIndex::AcqIndex - Constructor /*{{{*/ | |
1119 | pkgAcqIndex::pkgAcqIndex(pkgAcquire *Owner, | |
1120 | pkgAcqMetaBase *TransactionManager, | |
1121 | IndexTarget const *Target, | |
1122 | HashStringList const &ExpectedHash, | |
1123 | indexRecords *MetaIndexParser) | |
1124 | : pkgAcqBaseIndex(Owner, TransactionManager, Target, ExpectedHash, | |
1125 | MetaIndexParser) | |
1126 | { | |
1127 | RealURI = Target->URI; | |
1128 | ||
1129 | // autoselect the compression method | |
1130 | AutoSelectCompression(); | |
1131 | Init(Target->URI, Target->Description, Target->ShortDesc); | |
1132 | ||
1133 | if(_config->FindB("Debug::Acquire::Transaction", false) == true) | |
1134 | std::clog << "New pkgIndex with TransactionManager " | |
1135 | << TransactionManager << std::endl; | |
1136 | } | |
1137 | /*}}}*/ | |
1138 | // AcqIndex::AutoSelectCompression - Select compression /*{{{*/ | |
1139 | void pkgAcqIndex::AutoSelectCompression() | |
1140 | { | |
1141 | std::vector<std::string> types = APT::Configuration::getCompressionTypes(); | |
1142 | CompressionExtensions = ""; | |
1143 | if (ExpectedHashes.usable()) | |
1144 | { | |
1145 | for (std::vector<std::string>::const_iterator t = types.begin(); | |
1146 | t != types.end(); ++t) | |
1147 | { | |
1148 | std::string CompressedMetaKey = string(Target->MetaKey).append(".").append(*t); | |
1149 | if (*t == "uncompressed" || | |
1150 | MetaIndexParser->Exists(CompressedMetaKey) == true) | |
1151 | CompressionExtensions.append(*t).append(" "); | |
1152 | } | |
1153 | } | |
1154 | else | |
1155 | { | |
1156 | for (std::vector<std::string>::const_iterator t = types.begin(); t != types.end(); ++t) | |
1157 | CompressionExtensions.append(*t).append(" "); | |
1158 | } | |
1159 | if (CompressionExtensions.empty() == false) | |
1160 | CompressionExtensions.erase(CompressionExtensions.end()-1); | |
1161 | } | |
1162 | /*}}}*/ | |
1163 | // AcqIndex::Init - defered Constructor /*{{{*/ | |
1164 | void pkgAcqIndex::Init(string const &URI, string const &URIDesc, | |
1165 | string const &ShortDesc) | |
1166 | { | |
1167 | Stage = STAGE_DOWNLOAD; | |
1168 | ||
1169 | DestFile = GetPartialFileNameFromURI(URI); | |
1170 | ||
1171 | CurrentCompressionExtension = CompressionExtensions.substr(0, CompressionExtensions.find(' ')); | |
1172 | if (CurrentCompressionExtension == "uncompressed") | |
1173 | { | |
1174 | Desc.URI = URI; | |
1175 | if(Target) | |
1176 | MetaKey = string(Target->MetaKey); | |
1177 | } | |
1178 | else | |
1179 | { | |
1180 | Desc.URI = URI + '.' + CurrentCompressionExtension; | |
1181 | DestFile = DestFile + '.' + CurrentCompressionExtension; | |
1182 | if(Target) | |
1183 | MetaKey = string(Target->MetaKey) + '.' + CurrentCompressionExtension; | |
1184 | } | |
1185 | ||
1186 | // load the filesize | |
1187 | if(MetaIndexParser) | |
1188 | { | |
1189 | indexRecords::checkSum *Record = MetaIndexParser->Lookup(MetaKey); | |
1190 | if(Record) | |
1191 | FileSize = Record->Size; | |
1192 | ||
1193 | InitByHashIfNeeded(MetaKey); | |
1194 | } | |
1195 | ||
1196 | Desc.Description = URIDesc; | |
1197 | Desc.Owner = this; | |
1198 | Desc.ShortDesc = ShortDesc; | |
1199 | ||
1200 | QueueURI(Desc); | |
1201 | } | |
1202 | /*}}}*/ | |
1203 | // AcqIndex::AdjustForByHash - modify URI for by-hash support /*{{{*/ | |
1204 | void pkgAcqIndex::InitByHashIfNeeded(const std::string MetaKey) | |
1205 | { | |
1206 | // TODO: | |
1207 | // - (maybe?) add support for by-hash into the sources.list as flag | |
1208 | // - make apt-ftparchive generate the hashes (and expire?) | |
1209 | std::string HostKnob = "APT::Acquire::" + ::URI(Desc.URI).Host + "::By-Hash"; | |
1210 | if(_config->FindB("APT::Acquire::By-Hash", false) == true || | |
1211 | _config->FindB(HostKnob, false) == true || | |
1212 | MetaIndexParser->GetSupportsAcquireByHash()) | |
1213 | { | |
1214 | indexRecords::checkSum *Record = MetaIndexParser->Lookup(MetaKey); | |
1215 | if(Record) | |
1216 | { | |
1217 | // FIXME: should we really use the best hash here? or a fixed one? | |
1218 | const HashString *TargetHash = Record->Hashes.find(""); | |
1219 | std::string ByHash = "/by-hash/" + TargetHash->HashType() + "/" + TargetHash->HashValue(); | |
1220 | size_t trailing_slash = Desc.URI.find_last_of("/"); | |
1221 | Desc.URI = Desc.URI.replace( | |
1222 | trailing_slash, | |
1223 | Desc.URI.substr(trailing_slash+1).size()+1, | |
1224 | ByHash); | |
1225 | } else { | |
1226 | _error->Warning( | |
1227 | "Fetching ByHash requested but can not find record for %s", | |
1228 | MetaKey.c_str()); | |
1229 | } | |
1230 | } | |
1231 | } | |
1232 | /*}}}*/ | |
1233 | // AcqIndex::Custom600Headers - Insert custom request headers /*{{{*/ | |
1234 | // --------------------------------------------------------------------- | |
1235 | /* The only header we use is the last-modified header. */ | |
1236 | string pkgAcqIndex::Custom600Headers() const | |
1237 | { | |
1238 | string Final = GetFinalFilename(); | |
1239 | ||
1240 | string msg = "\nIndex-File: true"; | |
1241 | struct stat Buf; | |
1242 | if (stat(Final.c_str(),&Buf) == 0) | |
1243 | msg += "\nLast-Modified: " + TimeRFC1123(Buf.st_mtime); | |
1244 | ||
1245 | return msg; | |
1246 | } | |
1247 | /*}}}*/ | |
1248 | // pkgAcqIndex::Failed - getting the indexfile failed /*{{{*/ | |
1249 | void pkgAcqIndex::Failed(string Message,pkgAcquire::MethodConfig *Cnf) | |
1250 | { | |
1251 | size_t const nextExt = CompressionExtensions.find(' '); | |
1252 | if (nextExt != std::string::npos) | |
1253 | { | |
1254 | CompressionExtensions = CompressionExtensions.substr(nextExt+1); | |
1255 | Init(RealURI, Desc.Description, Desc.ShortDesc); | |
1256 | return; | |
1257 | } | |
1258 | ||
1259 | // on decompression failure, remove bad versions in partial/ | |
1260 | if (Stage == STAGE_DECOMPRESS_AND_VERIFY) | |
1261 | { | |
1262 | unlink(EraseFileName.c_str()); | |
1263 | } | |
1264 | ||
1265 | Item::Failed(Message,Cnf); | |
1266 | ||
1267 | /// cancel the entire transaction | |
1268 | TransactionManager->AbortTransaction(); | |
1269 | } | |
1270 | /*}}}*/ | |
1271 | // pkgAcqIndex::GetFinalFilename - Return the full final file path /*{{{*/ | |
1272 | std::string pkgAcqIndex::GetFinalFilename() const | |
1273 | { | |
1274 | std::string FinalFile = _config->FindDir("Dir::State::lists"); | |
1275 | FinalFile += URItoFileName(RealURI); | |
1276 | if (_config->FindB("Acquire::GzipIndexes",false) == true) | |
1277 | FinalFile += '.' + CurrentCompressionExtension; | |
1278 | return FinalFile; | |
1279 | } | |
1280 | /*}}}*/ | |
1281 | // AcqIndex::ReverifyAfterIMS - Reverify index after an ims-hit /*{{{*/ | |
1282 | void pkgAcqIndex::ReverifyAfterIMS() | |
1283 | { | |
1284 | // update destfile to *not* include the compression extension when doing | |
1285 | // a reverify (as its uncompressed on disk already) | |
1286 | DestFile = GetPartialFileNameFromURI(RealURI); | |
1287 | ||
1288 | // adjust DestFile if its compressed on disk | |
1289 | if (_config->FindB("Acquire::GzipIndexes",false) == true) | |
1290 | DestFile += '.' + CurrentCompressionExtension; | |
1291 | ||
1292 | // copy FinalFile into partial/ so that we check the hash again | |
1293 | string FinalFile = GetFinalFilename(); | |
1294 | Stage = STAGE_DECOMPRESS_AND_VERIFY; | |
1295 | Desc.URI = "copy:" + FinalFile; | |
1296 | QueueURI(Desc); | |
1297 | } | |
1298 | /*}}}*/ | |
1299 | // AcqIndex::ValidateFile - Validate the content of the downloaded file /*{{{*/ | |
1300 | bool pkgAcqIndex::ValidateFile(const std::string &FileName) | |
1301 | { | |
1302 | // FIXME: this can go away once we only ever download stuff that | |
1303 | // has a valid hash and we never do GET based probing | |
1304 | // FIXME2: this also leaks debian-isms into the code and should go therefore | |
1305 | ||
1306 | /* Always validate the index file for correctness (all indexes must | |
1307 | * have a Package field) (LP: #346386) (Closes: #627642) | |
1308 | */ | |
1309 | FileFd fd(FileName, FileFd::ReadOnly, FileFd::Extension); | |
1310 | // Only test for correctness if the content of the file is not empty | |
1311 | // (empty is ok) | |
1312 | if (fd.Size() > 0) | |
1313 | { | |
1314 | pkgTagSection sec; | |
1315 | pkgTagFile tag(&fd); | |
1316 | ||
1317 | // all our current indexes have a field 'Package' in each section | |
1318 | if (_error->PendingError() == true || | |
1319 | tag.Step(sec) == false || | |
1320 | sec.Exists("Package") == false) | |
1321 | return false; | |
1322 | } | |
1323 | return true; | |
1324 | } | |
1325 | /*}}}*/ | |
1326 | // AcqIndex::Done - Finished a fetch /*{{{*/ | |
1327 | // --------------------------------------------------------------------- | |
1328 | /* This goes through a number of states.. On the initial fetch the | |
1329 | method could possibly return an alternate filename which points | |
1330 | to the uncompressed version of the file. If this is so the file | |
1331 | is copied into the partial directory. In all other cases the file | |
1332 | is decompressed with a compressed uri. */ | |
1333 | void pkgAcqIndex::Done(string Message, | |
1334 | unsigned long long Size, | |
1335 | HashStringList const &Hashes, | |
1336 | pkgAcquire::MethodConfig *Cfg) | |
1337 | { | |
1338 | Item::Done(Message,Size,Hashes,Cfg); | |
1339 | ||
1340 | switch(Stage) | |
1341 | { | |
1342 | case STAGE_DOWNLOAD: | |
1343 | StageDownloadDone(Message, Hashes, Cfg); | |
1344 | break; | |
1345 | case STAGE_DECOMPRESS_AND_VERIFY: | |
1346 | StageDecompressDone(Message, Hashes, Cfg); | |
1347 | break; | |
1348 | } | |
1349 | } | |
1350 | /*}}}*/ | |
1351 | // AcqIndex::StageDownloadDone - Queue for decompress and verify /*{{{*/ | |
1352 | void pkgAcqIndex::StageDownloadDone(string Message, | |
1353 | HashStringList const &Hashes, | |
1354 | pkgAcquire::MethodConfig *Cfg) | |
1355 | { | |
1356 | // First check if the calculcated Hash of the (compressed) downloaded | |
1357 | // file matches the hash we have in the MetaIndexRecords for this file | |
1358 | if(VerifyHashByMetaKey(Hashes) == false) | |
1359 | { | |
1360 | RenameOnError(HashSumMismatch); | |
1361 | Failed(Message, Cfg); | |
1362 | return; | |
1363 | } | |
1364 | ||
1365 | Complete = true; | |
1366 | ||
1367 | // Handle the unzipd case | |
1368 | string FileName = LookupTag(Message,"Alt-Filename"); | |
1369 | if (FileName.empty() == false) | |
1370 | { | |
1371 | Stage = STAGE_DECOMPRESS_AND_VERIFY; | |
1372 | Local = true; | |
1373 | DestFile += ".decomp"; | |
1374 | Desc.URI = "copy:" + FileName; | |
1375 | QueueURI(Desc); | |
1376 | SetActiveSubprocess("copy"); | |
1377 | return; | |
1378 | } | |
1379 | ||
1380 | FileName = LookupTag(Message,"Filename"); | |
1381 | if (FileName.empty() == true) | |
1382 | { | |
1383 | Status = StatError; | |
1384 | ErrorText = "Method gave a blank filename"; | |
1385 | } | |
1386 | ||
1387 | // Methods like e.g. "file:" will give us a (compressed) FileName that is | |
1388 | // not the "DestFile" we set, in this case we uncompress from the local file | |
1389 | if (FileName != DestFile) | |
1390 | Local = true; | |
1391 | else | |
1392 | EraseFileName = FileName; | |
1393 | ||
1394 | // we need to verify the file against the current Release file again | |
1395 | // on if-modfied-since hit to avoid a stale attack against us | |
1396 | if(StringToBool(LookupTag(Message,"IMS-Hit"),false) == true) | |
1397 | { | |
1398 | // do not reverify cdrom sources as apt-cdrom may rewrite the Packages | |
1399 | // file when its doing the indexcopy | |
1400 | if (RealURI.substr(0,6) == "cdrom:") | |
1401 | return; | |
1402 | ||
1403 | // The files timestamp matches, reverify by copy into partial/ | |
1404 | EraseFileName = ""; | |
1405 | ReverifyAfterIMS(); | |
1406 | return; | |
1407 | } | |
1408 | ||
1409 | // If we have compressed indexes enabled, queue for hash verification | |
1410 | if (_config->FindB("Acquire::GzipIndexes",false)) | |
1411 | { | |
1412 | DestFile = GetPartialFileNameFromURI(RealURI + '.' + CurrentCompressionExtension); | |
1413 | EraseFileName = ""; | |
1414 | Stage = STAGE_DECOMPRESS_AND_VERIFY; | |
1415 | Desc.URI = "copy:" + FileName; | |
1416 | QueueURI(Desc); | |
1417 | SetActiveSubprocess("copy"); | |
1418 | return; | |
1419 | } | |
1420 | ||
1421 | // get the binary name for your used compression type | |
1422 | string decompProg; | |
1423 | if(CurrentCompressionExtension == "uncompressed") | |
1424 | decompProg = "copy"; | |
1425 | else | |
1426 | decompProg = _config->Find(string("Acquire::CompressionTypes::").append(CurrentCompressionExtension),""); | |
1427 | if(decompProg.empty() == true) | |
1428 | { | |
1429 | _error->Error("Unsupported extension: %s", CurrentCompressionExtension.c_str()); | |
1430 | return; | |
1431 | } | |
1432 | ||
1433 | // queue uri for the next stage | |
1434 | Stage = STAGE_DECOMPRESS_AND_VERIFY; | |
1435 | DestFile += ".decomp"; | |
1436 | Desc.URI = decompProg + ":" + FileName; | |
1437 | QueueURI(Desc); | |
1438 | SetActiveSubprocess(decompProg); | |
1439 | } | |
1440 | /*}}}*/ | |
1441 | // pkgAcqIndex::StageDecompressDone - Final verification /*{{{*/ | |
1442 | void pkgAcqIndex::StageDecompressDone(string Message, | |
1443 | HashStringList const &Hashes, | |
1444 | pkgAcquire::MethodConfig *Cfg) | |
1445 | { | |
1446 | if (ExpectedHashes.usable() && ExpectedHashes != Hashes) | |
1447 | { | |
1448 | Desc.URI = RealURI; | |
1449 | RenameOnError(HashSumMismatch); | |
1450 | printHashSumComparision(RealURI, ExpectedHashes, Hashes); | |
1451 | Failed(Message, Cfg); | |
1452 | return; | |
1453 | } | |
1454 | ||
1455 | if(!ValidateFile(DestFile)) | |
1456 | { | |
1457 | RenameOnError(InvalidFormat); | |
1458 | Failed(Message, Cfg); | |
1459 | return; | |
1460 | } | |
1461 | ||
1462 | // remove the compressed version of the file | |
1463 | unlink(EraseFileName.c_str()); | |
1464 | ||
1465 | // Done, queue for rename on transaction finished | |
1466 | TransactionManager->TransactionStageCopy(this, DestFile, GetFinalFilename()); | |
1467 | ||
1468 | return; | |
1469 | } | |
1470 | /*}}}*/ | |
1471 | // AcqIndexTrans::pkgAcqIndexTrans - Constructor /*{{{*/ | |
1472 | // --------------------------------------------------------------------- | |
1473 | /* The Translation file is added to the queue */ | |
1474 | pkgAcqIndexTrans::pkgAcqIndexTrans(pkgAcquire *Owner, | |
1475 | string URI,string URIDesc,string ShortDesc) | |
1476 | : pkgAcqIndex(Owner, URI, URIDesc, ShortDesc, HashStringList()) | |
1477 | { | |
1478 | } | |
1479 | pkgAcqIndexTrans::pkgAcqIndexTrans(pkgAcquire *Owner, | |
1480 | pkgAcqMetaBase *TransactionManager, | |
1481 | IndexTarget const * const Target, | |
1482 | HashStringList const &ExpectedHashes, | |
1483 | indexRecords *MetaIndexParser) | |
1484 | : pkgAcqIndex(Owner, TransactionManager, Target, ExpectedHashes, MetaIndexParser) | |
1485 | { | |
1486 | // load the filesize | |
1487 | indexRecords::checkSum *Record = MetaIndexParser->Lookup(string(Target->MetaKey)); | |
1488 | if(Record) | |
1489 | FileSize = Record->Size; | |
1490 | } | |
1491 | /*}}}*/ | |
1492 | // AcqIndexTrans::Custom600Headers - Insert custom request headers /*{{{*/ | |
1493 | string pkgAcqIndexTrans::Custom600Headers() const | |
1494 | { | |
1495 | string Final = GetFinalFilename(); | |
1496 | ||
1497 | struct stat Buf; | |
1498 | if (stat(Final.c_str(),&Buf) != 0) | |
1499 | return "\nFail-Ignore: true\nIndex-File: true"; | |
1500 | return "\nFail-Ignore: true\nIndex-File: true\nLast-Modified: " + TimeRFC1123(Buf.st_mtime); | |
1501 | } | |
1502 | /*}}}*/ | |
1503 | // AcqIndexTrans::Failed - Silence failure messages for missing files /*{{{*/ | |
1504 | void pkgAcqIndexTrans::Failed(string Message,pkgAcquire::MethodConfig *Cnf) | |
1505 | { | |
1506 | size_t const nextExt = CompressionExtensions.find(' '); | |
1507 | if (nextExt != std::string::npos) | |
1508 | { | |
1509 | CompressionExtensions = CompressionExtensions.substr(nextExt+1); | |
1510 | Init(RealURI, Desc.Description, Desc.ShortDesc); | |
1511 | Status = StatIdle; | |
1512 | return; | |
1513 | } | |
1514 | ||
1515 | Item::Failed(Message,Cnf); | |
1516 | ||
1517 | // FIXME: this is used often (e.g. in pkgAcqIndexTrans) so refactor | |
1518 | if (Cnf->LocalOnly == true || | |
1519 | StringToBool(LookupTag(Message,"Transient-Failure"),false) == false) | |
1520 | { | |
1521 | // Ignore this | |
1522 | Status = StatDone; | |
1523 | } | |
1524 | } | |
1525 | /*}}}*/ | |
1526 | // AcqMetaBase::Add - Add a item to the current Transaction /*{{{*/ | |
1527 | void pkgAcqMetaBase::Add(Item *I) | |
1528 | { | |
1529 | Transaction.push_back(I); | |
1530 | } | |
1531 | /*}}}*/ | |
1532 | // AcqMetaBase::AbortTransaction - Abort the current Transaction /*{{{*/ | |
1533 | void pkgAcqMetaBase::AbortTransaction() | |
1534 | { | |
1535 | if(_config->FindB("Debug::Acquire::Transaction", false) == true) | |
1536 | std::clog << "AbortTransaction: " << TransactionManager << std::endl; | |
1537 | ||
1538 | // ensure the toplevel is in error state too | |
1539 | for (std::vector<Item*>::iterator I = Transaction.begin(); | |
1540 | I != Transaction.end(); ++I) | |
1541 | { | |
1542 | if(_config->FindB("Debug::Acquire::Transaction", false) == true) | |
1543 | std::clog << " Cancel: " << (*I)->DestFile << std::endl; | |
1544 | // the transaction will abort, so stop anything that is idle | |
1545 | if ((*I)->Status == pkgAcquire::Item::StatIdle) | |
1546 | (*I)->Status = pkgAcquire::Item::StatDone; | |
1547 | ||
1548 | // kill files in partial | |
1549 | std::string const PartialFile = GetPartialFileName(flNotDir((*I)->DestFile)); | |
1550 | if(FileExists(PartialFile)) | |
1551 | Rename(PartialFile, PartialFile + ".FAILED"); | |
1552 | } | |
1553 | } | |
1554 | /*}}}*/ | |
1555 | // AcqMetaBase::TransactionHasError - Check for errors in Transaction /*{{{*/ | |
1556 | bool pkgAcqMetaBase::TransactionHasError() | |
1557 | { | |
1558 | for (pkgAcquire::ItemIterator I = Transaction.begin(); | |
1559 | I != Transaction.end(); ++I) | |
1560 | if((*I)->Status != pkgAcquire::Item::StatDone && | |
1561 | (*I)->Status != pkgAcquire::Item::StatIdle) | |
1562 | return true; | |
1563 | ||
1564 | return false; | |
1565 | } | |
1566 | /*}}}*/ | |
1567 | // AcqMetaBase::CommitTransaction - Commit a transaction /*{{{*/ | |
1568 | void pkgAcqMetaBase::CommitTransaction() | |
1569 | { | |
1570 | if(_config->FindB("Debug::Acquire::Transaction", false) == true) | |
1571 | std::clog << "CommitTransaction: " << this << std::endl; | |
1572 | ||
1573 | // move new files into place *and* remove files that are not | |
1574 | // part of the transaction but are still on disk | |
1575 | for (std::vector<Item*>::iterator I = Transaction.begin(); | |
1576 | I != Transaction.end(); ++I) | |
1577 | { | |
1578 | if((*I)->PartialFile != "") | |
1579 | { | |
1580 | if(_config->FindB("Debug::Acquire::Transaction", false) == true) | |
1581 | std::clog << "mv " << (*I)->PartialFile << " -> "<< (*I)->DestFile << " " | |
1582 | << (*I)->DescURI() << std::endl; | |
1583 | ||
1584 | Rename((*I)->PartialFile, (*I)->DestFile); | |
1585 | ChangeOwnerAndPermissionOfFile("CommitTransaction", (*I)->DestFile.c_str(), "root", "root", 0644); | |
1586 | ||
1587 | } else { | |
1588 | if(_config->FindB("Debug::Acquire::Transaction", false) == true) | |
1589 | std::clog << "rm " | |
1590 | << (*I)->DestFile | |
1591 | << " " | |
1592 | << (*I)->DescURI() | |
1593 | << std::endl; | |
1594 | unlink((*I)->DestFile.c_str()); | |
1595 | } | |
1596 | // mark that this transaction is finished | |
1597 | (*I)->TransactionManager = 0; | |
1598 | } | |
1599 | } | |
1600 | /*}}}*/ | |
1601 | // AcqMetaBase::TransactionStageCopy - Stage a file for copying /*{{{*/ | |
1602 | void pkgAcqMetaBase::TransactionStageCopy(Item *I, | |
1603 | const std::string &From, | |
1604 | const std::string &To) | |
1605 | { | |
1606 | I->PartialFile = From; | |
1607 | I->DestFile = To; | |
1608 | } | |
1609 | /*}}}*/ | |
1610 | // AcqMetaBase::TransactionStageRemoval - Sage a file for removal /*{{{*/ | |
1611 | void pkgAcqMetaBase::TransactionStageRemoval(Item *I, | |
1612 | const std::string &FinalFile) | |
1613 | { | |
1614 | I->PartialFile = ""; | |
1615 | I->DestFile = FinalFile; | |
1616 | } | |
1617 | /*}}}*/ | |
1618 | // AcqMetaBase::GenerateAuthWarning - Check gpg authentication error /*{{{*/ | |
1619 | bool pkgAcqMetaBase::CheckStopAuthentication(const std::string &RealURI, | |
1620 | const std::string &Message) | |
1621 | { | |
1622 | // FIXME: this entire function can do now that we disallow going to | |
1623 | // a unauthenticated state and can cleanly rollback | |
1624 | ||
1625 | string Final = _config->FindDir("Dir::State::lists") + URItoFileName(RealURI); | |
1626 | ||
1627 | if(FileExists(Final)) | |
1628 | { | |
1629 | Status = StatTransientNetworkError; | |
1630 | _error->Warning(_("An error occurred during the signature " | |
1631 | "verification. The repository is not updated " | |
1632 | "and the previous index files will be used. " | |
1633 | "GPG error: %s: %s\n"), | |
1634 | Desc.Description.c_str(), | |
1635 | LookupTag(Message,"Message").c_str()); | |
1636 | RunScripts("APT::Update::Auth-Failure"); | |
1637 | return true; | |
1638 | } else if (LookupTag(Message,"Message").find("NODATA") != string::npos) { | |
1639 | /* Invalid signature file, reject (LP: #346386) (Closes: #627642) */ | |
1640 | _error->Error(_("GPG error: %s: %s"), | |
1641 | Desc.Description.c_str(), | |
1642 | LookupTag(Message,"Message").c_str()); | |
1643 | Status = StatError; | |
1644 | return true; | |
1645 | } else { | |
1646 | _error->Warning(_("GPG error: %s: %s"), | |
1647 | Desc.Description.c_str(), | |
1648 | LookupTag(Message,"Message").c_str()); | |
1649 | } | |
1650 | // gpgv method failed | |
1651 | ReportMirrorFailure("GPGFailure"); | |
1652 | return false; | |
1653 | } | |
1654 | /*}}}*/ | |
1655 | // AcqMetaSig::AcqMetaSig - Constructor /*{{{*/ | |
1656 | pkgAcqMetaSig::pkgAcqMetaSig(pkgAcquire *Owner, | |
1657 | pkgAcqMetaBase *TransactionManager, | |
1658 | string URI,string URIDesc,string ShortDesc, | |
1659 | string MetaIndexFile, | |
1660 | const vector<IndexTarget*>* IndexTargets, | |
1661 | indexRecords* MetaIndexParser) : | |
1662 | pkgAcqMetaBase(Owner, IndexTargets, MetaIndexParser, | |
1663 | HashStringList(), TransactionManager), | |
1664 | RealURI(URI), MetaIndexFile(MetaIndexFile), URIDesc(URIDesc), | |
1665 | ShortDesc(ShortDesc) | |
1666 | { | |
1667 | DestFile = _config->FindDir("Dir::State::lists") + "partial/"; | |
1668 | DestFile += URItoFileName(RealURI); | |
1669 | ||
1670 | // remove any partial downloaded sig-file in partial/. | |
1671 | // it may confuse proxies and is too small to warrant a | |
1672 | // partial download anyway | |
1673 | unlink(DestFile.c_str()); | |
1674 | ||
1675 | // set the TransactionManager | |
1676 | if(_config->FindB("Debug::Acquire::Transaction", false) == true) | |
1677 | std::clog << "New pkgAcqMetaSig with TransactionManager " | |
1678 | << TransactionManager << std::endl; | |
1679 | ||
1680 | // Create the item | |
1681 | Desc.Description = URIDesc; | |
1682 | Desc.Owner = this; | |
1683 | Desc.ShortDesc = ShortDesc; | |
1684 | Desc.URI = URI; | |
1685 | ||
1686 | QueueURI(Desc); | |
1687 | } | |
1688 | /*}}}*/ | |
1689 | pkgAcqMetaSig::~pkgAcqMetaSig() /*{{{*/ | |
1690 | { | |
1691 | } | |
1692 | /*}}}*/ | |
1693 | // pkgAcqMetaSig::Custom600Headers - Insert custom request headers /*{{{*/ | |
1694 | // --------------------------------------------------------------------- | |
1695 | string pkgAcqMetaSig::Custom600Headers() const | |
1696 | { | |
1697 | std::string Header = GetCustom600Headers(RealURI); | |
1698 | return Header; | |
1699 | } | |
1700 | /*}}}*/ | |
1701 | // pkgAcqMetaSig::Done - The signature was downloaded/verified /*{{{*/ | |
1702 | // --------------------------------------------------------------------- | |
1703 | /* The only header we use is the last-modified header. */ | |
1704 | void pkgAcqMetaSig::Done(string Message,unsigned long long Size, | |
1705 | HashStringList const &Hashes, | |
1706 | pkgAcquire::MethodConfig *Cfg) | |
1707 | { | |
1708 | Item::Done(Message, Size, Hashes, Cfg); | |
1709 | ||
1710 | if(AuthPass == false) | |
1711 | { | |
1712 | if(CheckDownloadDone(Message, RealURI) == true) | |
1713 | { | |
1714 | // destfile will be modified to point to MetaIndexFile for the | |
1715 | // gpgv method, so we need to save it here | |
1716 | MetaIndexFileSignature = DestFile; | |
1717 | QueueForSignatureVerify(MetaIndexFile, MetaIndexFileSignature); | |
1718 | } | |
1719 | return; | |
1720 | } | |
1721 | else | |
1722 | { | |
1723 | if(CheckAuthDone(Message, RealURI) == true) | |
1724 | { | |
1725 | std::string FinalFile = _config->FindDir("Dir::State::lists"); | |
1726 | FinalFile += URItoFileName(RealURI); | |
1727 | TransactionManager->TransactionStageCopy(this, MetaIndexFileSignature, FinalFile); | |
1728 | } | |
1729 | } | |
1730 | } | |
1731 | /*}}}*/ | |
1732 | void pkgAcqMetaSig::Failed(string Message,pkgAcquire::MethodConfig *Cnf)/*{{{*/ | |
1733 | { | |
1734 | string Final = _config->FindDir("Dir::State::lists") + URItoFileName(RealURI); | |
1735 | ||
1736 | // check if we need to fail at this point | |
1737 | if (AuthPass == true && CheckStopAuthentication(RealURI, Message)) | |
1738 | return; | |
1739 | ||
1740 | // FIXME: meh, this is not really elegant | |
1741 | string InReleaseURI = RealURI.replace(RealURI.rfind("Release.gpg"), 12, | |
1742 | "InRelease"); | |
1743 | string FinalInRelease = _config->FindDir("Dir::State::lists") + URItoFileName(InReleaseURI); | |
1744 | ||
1745 | if (RealFileExists(Final) || RealFileExists(FinalInRelease)) | |
1746 | { | |
1747 | std::string downgrade_msg; | |
1748 | strprintf(downgrade_msg, _("The repository '%s' is no longer signed."), | |
1749 | URIDesc.c_str()); | |
1750 | if(_config->FindB("Acquire::AllowDowngradeToInsecureRepositories")) | |
1751 | { | |
1752 | // meh, the users wants to take risks (we still mark the packages | |
1753 | // from this repository as unauthenticated) | |
1754 | _error->Warning("%s", downgrade_msg.c_str()); | |
1755 | _error->Warning(_("This is normally not allowed, but the option " | |
1756 | "Acquire::AllowDowngradeToInsecureRepositories was " | |
1757 | "given to override it.")); | |
1758 | ||
1759 | } else { | |
1760 | _error->Error("%s", downgrade_msg.c_str()); | |
1761 | Rename(MetaIndexFile, MetaIndexFile+".FAILED"); | |
1762 | Item::Failed("Message: " + downgrade_msg, Cnf); | |
1763 | TransactionManager->AbortTransaction(); | |
1764 | return; | |
1765 | } | |
1766 | } | |
1767 | ||
1768 | // this ensures that any file in the lists/ dir is removed by the | |
1769 | // transaction | |
1770 | DestFile = GetPartialFileNameFromURI(RealURI); | |
1771 | TransactionManager->TransactionStageRemoval(this, DestFile); | |
1772 | ||
1773 | // only allow going further if the users explicitely wants it | |
1774 | if(_config->FindB("Acquire::AllowInsecureRepositories") == true) | |
1775 | { | |
1776 | // we parse the indexes here because at this point the user wanted | |
1777 | // a repository that may potentially harm him | |
1778 | MetaIndexParser->Load(MetaIndexFile); | |
1779 | QueueIndexes(true); | |
1780 | } | |
1781 | else | |
1782 | { | |
1783 | _error->Warning("Use --allow-insecure-repositories to force the update"); | |
1784 | } | |
1785 | ||
1786 | Item::Failed(Message,Cnf); | |
1787 | ||
1788 | // FIXME: this is used often (e.g. in pkgAcqIndexTrans) so refactor | |
1789 | if (Cnf->LocalOnly == true || | |
1790 | StringToBool(LookupTag(Message,"Transient-Failure"),false) == false) | |
1791 | { | |
1792 | // Ignore this | |
1793 | Status = StatDone; | |
1794 | } | |
1795 | } | |
1796 | /*}}}*/ | |
1797 | pkgAcqMetaIndex::pkgAcqMetaIndex(pkgAcquire *Owner, /*{{{*/ | |
1798 | pkgAcqMetaBase *TransactionManager, | |
1799 | string URI,string URIDesc,string ShortDesc, | |
1800 | string MetaIndexSigURI,string MetaIndexSigURIDesc, string MetaIndexSigShortDesc, | |
1801 | const vector<IndexTarget*>* IndexTargets, | |
1802 | indexRecords* MetaIndexParser) : | |
1803 | pkgAcqMetaBase(Owner, IndexTargets, MetaIndexParser, HashStringList(), | |
1804 | TransactionManager), | |
1805 | RealURI(URI), URIDesc(URIDesc), ShortDesc(ShortDesc), | |
1806 | MetaIndexSigURI(MetaIndexSigURI), MetaIndexSigURIDesc(MetaIndexSigURIDesc), | |
1807 | MetaIndexSigShortDesc(MetaIndexSigShortDesc) | |
1808 | { | |
1809 | if(TransactionManager == NULL) | |
1810 | { | |
1811 | this->TransactionManager = this; | |
1812 | this->TransactionManager->Add(this); | |
1813 | } | |
1814 | ||
1815 | if(_config->FindB("Debug::Acquire::Transaction", false) == true) | |
1816 | std::clog << "New pkgAcqMetaIndex with TransactionManager " | |
1817 | << this->TransactionManager << std::endl; | |
1818 | ||
1819 | ||
1820 | Init(URIDesc, ShortDesc); | |
1821 | } | |
1822 | /*}}}*/ | |
1823 | // pkgAcqMetaIndex::Init - Delayed constructor /*{{{*/ | |
1824 | void pkgAcqMetaIndex::Init(std::string URIDesc, std::string ShortDesc) | |
1825 | { | |
1826 | DestFile = GetPartialFileNameFromURI(RealURI); | |
1827 | ||
1828 | // Create the item | |
1829 | Desc.Description = URIDesc; | |
1830 | Desc.Owner = this; | |
1831 | Desc.ShortDesc = ShortDesc; | |
1832 | Desc.URI = RealURI; | |
1833 | ||
1834 | // we expect more item | |
1835 | ExpectedAdditionalItems = IndexTargets->size(); | |
1836 | QueueURI(Desc); | |
1837 | } | |
1838 | /*}}}*/ | |
1839 | // pkgAcqMetaIndex::Custom600Headers - Insert custom request headers /*{{{*/ | |
1840 | // --------------------------------------------------------------------- | |
1841 | string pkgAcqMetaIndex::Custom600Headers() const | |
1842 | { | |
1843 | return GetCustom600Headers(RealURI); | |
1844 | } | |
1845 | /*}}}*/ | |
1846 | void pkgAcqMetaIndex::Done(string Message,unsigned long long Size, /*{{{*/ | |
1847 | HashStringList const &Hashes, | |
1848 | pkgAcquire::MethodConfig *Cfg) | |
1849 | { | |
1850 | Item::Done(Message,Size,Hashes,Cfg); | |
1851 | ||
1852 | if(CheckDownloadDone(Message, RealURI)) | |
1853 | { | |
1854 | // we have a Release file, now download the Signature, all further | |
1855 | // verify/queue for additional downloads will be done in the | |
1856 | // pkgAcqMetaSig::Done() code | |
1857 | std::string MetaIndexFile = DestFile; | |
1858 | new pkgAcqMetaSig(Owner, TransactionManager, | |
1859 | MetaIndexSigURI, MetaIndexSigURIDesc, | |
1860 | MetaIndexSigShortDesc, MetaIndexFile, IndexTargets, | |
1861 | MetaIndexParser); | |
1862 | ||
1863 | string FinalFile = _config->FindDir("Dir::State::lists"); | |
1864 | FinalFile += URItoFileName(RealURI); | |
1865 | TransactionManager->TransactionStageCopy(this, DestFile, FinalFile); | |
1866 | } | |
1867 | } | |
1868 | /*}}}*/ | |
1869 | bool pkgAcqMetaBase::CheckAuthDone(string Message, const string &RealURI) /*{{{*/ | |
1870 | { | |
1871 | // At this point, the gpgv method has succeeded, so there is a | |
1872 | // valid signature from a key in the trusted keyring. We | |
1873 | // perform additional verification of its contents, and use them | |
1874 | // to verify the indexes we are about to download | |
1875 | ||
1876 | if (!MetaIndexParser->Load(DestFile)) | |
1877 | { | |
1878 | Status = StatAuthError; | |
1879 | ErrorText = MetaIndexParser->ErrorText; | |
1880 | return false; | |
1881 | } | |
1882 | ||
1883 | if (!VerifyVendor(Message, RealURI)) | |
1884 | { | |
1885 | return false; | |
1886 | } | |
1887 | ||
1888 | if (_config->FindB("Debug::pkgAcquire::Auth", false)) | |
1889 | std::cerr << "Signature verification succeeded: " | |
1890 | << DestFile << std::endl; | |
1891 | ||
1892 | // Download further indexes with verification | |
1893 | // | |
1894 | // it would be really nice if we could simply do | |
1895 | // if (IMSHit == false) QueueIndexes(true) | |
1896 | // and skip the download if the Release file has not changed | |
1897 | // - but right now the list cleaner will needs to be tricked | |
1898 | // to not delete all our packages/source indexes in this case | |
1899 | QueueIndexes(true); | |
1900 | ||
1901 | return true; | |
1902 | } | |
1903 | /*}}}*/ | |
1904 | // pkgAcqMetaBase::GetCustom600Headers - Get header for AcqMetaBase /*{{{*/ | |
1905 | // --------------------------------------------------------------------- | |
1906 | string pkgAcqMetaBase::GetCustom600Headers(const string &RealURI) const | |
1907 | { | |
1908 | std::string Header = "\nIndex-File: true"; | |
1909 | std::string MaximumSize; | |
1910 | strprintf(MaximumSize, "\nMaximum-Size: %i", | |
1911 | _config->FindI("Acquire::MaxReleaseFileSize", 10*1000*1000)); | |
1912 | Header += MaximumSize; | |
1913 | ||
1914 | string FinalFile = _config->FindDir("Dir::State::lists"); | |
1915 | FinalFile += URItoFileName(RealURI); | |
1916 | ||
1917 | struct stat Buf; | |
1918 | if (stat(FinalFile.c_str(),&Buf) == 0) | |
1919 | Header += "\nLast-Modified: " + TimeRFC1123(Buf.st_mtime); | |
1920 | ||
1921 | return Header; | |
1922 | } | |
1923 | /*}}}*/ | |
1924 | // pkgAcqMetaBase::QueueForSignatureVerify /*{{{*/ | |
1925 | void pkgAcqMetaBase::QueueForSignatureVerify(const std::string &MetaIndexFile, | |
1926 | const std::string &MetaIndexFileSignature) | |
1927 | { | |
1928 | AuthPass = true; | |
1929 | Desc.URI = "gpgv:" + MetaIndexFileSignature; | |
1930 | DestFile = MetaIndexFile; | |
1931 | QueueURI(Desc); | |
1932 | SetActiveSubprocess("gpgv"); | |
1933 | } | |
1934 | /*}}}*/ | |
1935 | // pkgAcqMetaBase::CheckDownloadDone /*{{{*/ | |
1936 | bool pkgAcqMetaBase::CheckDownloadDone(const std::string &Message, | |
1937 | const std::string &RealURI) | |
1938 | { | |
1939 | // We have just finished downloading a Release file (it is not | |
1940 | // verified yet) | |
1941 | ||
1942 | string FileName = LookupTag(Message,"Filename"); | |
1943 | if (FileName.empty() == true) | |
1944 | { | |
1945 | Status = StatError; | |
1946 | ErrorText = "Method gave a blank filename"; | |
1947 | return false; | |
1948 | } | |
1949 | ||
1950 | if (FileName != DestFile) | |
1951 | { | |
1952 | Local = true; | |
1953 | Desc.URI = "copy:" + FileName; | |
1954 | QueueURI(Desc); | |
1955 | return false; | |
1956 | } | |
1957 | ||
1958 | // make sure to verify against the right file on I-M-S hit | |
1959 | IMSHit = StringToBool(LookupTag(Message,"IMS-Hit"),false); | |
1960 | if(IMSHit) | |
1961 | { | |
1962 | string FinalFile = _config->FindDir("Dir::State::lists"); | |
1963 | FinalFile += URItoFileName(RealURI); | |
1964 | DestFile = FinalFile; | |
1965 | } | |
1966 | ||
1967 | // set Item to complete as the remaining work is all local (verify etc) | |
1968 | Complete = true; | |
1969 | ||
1970 | return true; | |
1971 | } | |
1972 | /*}}}*/ | |
1973 | void pkgAcqMetaBase::QueueIndexes(bool verify) /*{{{*/ | |
1974 | { | |
1975 | bool transInRelease = false; | |
1976 | { | |
1977 | std::vector<std::string> const keys = MetaIndexParser->MetaKeys(); | |
1978 | for (std::vector<std::string>::const_iterator k = keys.begin(); k != keys.end(); ++k) | |
1979 | // FIXME: Feels wrong to check for hardcoded string here, but what should we do else… | |
1980 | if (k->find("Translation-") != std::string::npos) | |
1981 | { | |
1982 | transInRelease = true; | |
1983 | break; | |
1984 | } | |
1985 | } | |
1986 | ||
1987 | // at this point the real Items are loaded in the fetcher | |
1988 | ExpectedAdditionalItems = 0; | |
1989 | for (vector <IndexTarget*>::const_iterator Target = IndexTargets->begin(); | |
1990 | Target != IndexTargets->end(); | |
1991 | ++Target) | |
1992 | { | |
1993 | HashStringList ExpectedIndexHashes; | |
1994 | const indexRecords::checkSum *Record = MetaIndexParser->Lookup((*Target)->MetaKey); | |
1995 | bool compressedAvailable = false; | |
1996 | if (Record == NULL) | |
1997 | { | |
1998 | if ((*Target)->IsOptional() == true) | |
1999 | { | |
2000 | std::vector<std::string> types = APT::Configuration::getCompressionTypes(); | |
2001 | for (std::vector<std::string>::const_iterator t = types.begin(); t != types.end(); ++t) | |
2002 | if (MetaIndexParser->Exists((*Target)->MetaKey + "." + *t) == true) | |
2003 | { | |
2004 | compressedAvailable = true; | |
2005 | break; | |
2006 | } | |
2007 | } | |
2008 | else if (verify == true) | |
2009 | { | |
2010 | Status = StatAuthError; | |
2011 | strprintf(ErrorText, _("Unable to find expected entry '%s' in Release file (Wrong sources.list entry or malformed file)"), (*Target)->MetaKey.c_str()); | |
2012 | return; | |
2013 | } | |
2014 | } | |
2015 | else | |
2016 | { | |
2017 | ExpectedIndexHashes = Record->Hashes; | |
2018 | if (_config->FindB("Debug::pkgAcquire::Auth", false)) | |
2019 | { | |
2020 | std::cerr << "Queueing: " << (*Target)->URI << std::endl | |
2021 | << "Expected Hash:" << std::endl; | |
2022 | for (HashStringList::const_iterator hs = ExpectedIndexHashes.begin(); hs != ExpectedIndexHashes.end(); ++hs) | |
2023 | std::cerr << "\t- " << hs->toStr() << std::endl; | |
2024 | std::cerr << "For: " << Record->MetaKeyFilename << std::endl; | |
2025 | } | |
2026 | if (verify == true && ExpectedIndexHashes.empty() == true && (*Target)->IsOptional() == false) | |
2027 | { | |
2028 | Status = StatAuthError; | |
2029 | strprintf(ErrorText, _("Unable to find hash sum for '%s' in Release file"), (*Target)->MetaKey.c_str()); | |
2030 | return; | |
2031 | } | |
2032 | } | |
2033 | ||
2034 | if ((*Target)->IsOptional() == true) | |
2035 | { | |
2036 | if (transInRelease == false || Record != NULL || compressedAvailable == true) | |
2037 | { | |
2038 | if (_config->FindB("Acquire::PDiffs",true) == true && transInRelease == true && | |
2039 | MetaIndexParser->Exists((*Target)->MetaKey + ".diff/Index") == true) | |
2040 | new pkgAcqDiffIndex(Owner, TransactionManager, *Target, ExpectedIndexHashes, MetaIndexParser); | |
2041 | else | |
2042 | new pkgAcqIndexTrans(Owner, TransactionManager, *Target, ExpectedIndexHashes, MetaIndexParser); | |
2043 | } | |
2044 | continue; | |
2045 | } | |
2046 | ||
2047 | /* Queue Packages file (either diff or full packages files, depending | |
2048 | on the users option) - we also check if the PDiff Index file is listed | |
2049 | in the Meta-Index file. Ideal would be if pkgAcqDiffIndex would test this | |
2050 | instead, but passing the required info to it is to much hassle */ | |
2051 | if(_config->FindB("Acquire::PDiffs",true) == true && (verify == false || | |
2052 | MetaIndexParser->Exists((*Target)->MetaKey + ".diff/Index") == true)) | |
2053 | new pkgAcqDiffIndex(Owner, TransactionManager, *Target, ExpectedIndexHashes, MetaIndexParser); | |
2054 | else | |
2055 | new pkgAcqIndex(Owner, TransactionManager, *Target, ExpectedIndexHashes, MetaIndexParser); | |
2056 | } | |
2057 | } | |
2058 | /*}}}*/ | |
2059 | bool pkgAcqMetaBase::VerifyVendor(string Message, const string &RealURI)/*{{{*/ | |
2060 | { | |
2061 | string::size_type pos; | |
2062 | ||
2063 | // check for missing sigs (that where not fatal because otherwise we had | |
2064 | // bombed earlier) | |
2065 | string missingkeys; | |
2066 | string msg = _("There is no public key available for the " | |
2067 | "following key IDs:\n"); | |
2068 | pos = Message.find("NO_PUBKEY "); | |
2069 | if (pos != std::string::npos) | |
2070 | { | |
2071 | string::size_type start = pos+strlen("NO_PUBKEY "); | |
2072 | string Fingerprint = Message.substr(start, Message.find("\n")-start); | |
2073 | missingkeys += (Fingerprint); | |
2074 | } | |
2075 | if(!missingkeys.empty()) | |
2076 | _error->Warning("%s", (msg + missingkeys).c_str()); | |
2077 | ||
2078 | string Transformed = MetaIndexParser->GetExpectedDist(); | |
2079 | ||
2080 | if (Transformed == "../project/experimental") | |
2081 | { | |
2082 | Transformed = "experimental"; | |
2083 | } | |
2084 | ||
2085 | pos = Transformed.rfind('/'); | |
2086 | if (pos != string::npos) | |
2087 | { | |
2088 | Transformed = Transformed.substr(0, pos); | |
2089 | } | |
2090 | ||
2091 | if (Transformed == ".") | |
2092 | { | |
2093 | Transformed = ""; | |
2094 | } | |
2095 | ||
2096 | if (_config->FindB("Acquire::Check-Valid-Until", true) == true && | |
2097 | MetaIndexParser->GetValidUntil() > 0) { | |
2098 | time_t const invalid_since = time(NULL) - MetaIndexParser->GetValidUntil(); | |
2099 | if (invalid_since > 0) | |
2100 | // TRANSLATOR: The first %s is the URL of the bad Release file, the second is | |
2101 | // the time since then the file is invalid - formated in the same way as in | |
2102 | // the download progress display (e.g. 7d 3h 42min 1s) | |
2103 | return _error->Error( | |
2104 | _("Release file for %s is expired (invalid since %s). " | |
2105 | "Updates for this repository will not be applied."), | |
2106 | RealURI.c_str(), TimeToStr(invalid_since).c_str()); | |
2107 | } | |
2108 | ||
2109 | if (_config->FindB("Debug::pkgAcquire::Auth", false)) | |
2110 | { | |
2111 | std::cerr << "Got Codename: " << MetaIndexParser->GetDist() << std::endl; | |
2112 | std::cerr << "Expecting Dist: " << MetaIndexParser->GetExpectedDist() << std::endl; | |
2113 | std::cerr << "Transformed Dist: " << Transformed << std::endl; | |
2114 | } | |
2115 | ||
2116 | if (MetaIndexParser->CheckDist(Transformed) == false) | |
2117 | { | |
2118 | // This might become fatal one day | |
2119 | // Status = StatAuthError; | |
2120 | // ErrorText = "Conflicting distribution; expected " | |
2121 | // + MetaIndexParser->GetExpectedDist() + " but got " | |
2122 | // + MetaIndexParser->GetDist(); | |
2123 | // return false; | |
2124 | if (!Transformed.empty()) | |
2125 | { | |
2126 | _error->Warning(_("Conflicting distribution: %s (expected %s but got %s)"), | |
2127 | Desc.Description.c_str(), | |
2128 | Transformed.c_str(), | |
2129 | MetaIndexParser->GetDist().c_str()); | |
2130 | } | |
2131 | } | |
2132 | ||
2133 | return true; | |
2134 | } | |
2135 | /*}}}*/ | |
2136 | // pkgAcqMetaIndex::Failed - no Release file present /*{{{*/ | |
2137 | void pkgAcqMetaIndex::Failed(string Message, | |
2138 | pkgAcquire::MethodConfig * Cnf) | |
2139 | { | |
2140 | pkgAcquire::Item::Failed(Message, Cnf); | |
2141 | Status = StatDone; | |
2142 | ||
2143 | string FinalFile = _config->FindDir("Dir::State::lists") + URItoFileName(RealURI); | |
2144 | ||
2145 | _error->Warning(_("The repository '%s' does not have a Release file. " | |
2146 | "This is deprecated, please contact the owner of the " | |
2147 | "repository."), URIDesc.c_str()); | |
2148 | ||
2149 | // No Release file was present so fall | |
2150 | // back to queueing Packages files without verification | |
2151 | // only allow going further if the users explicitely wants it | |
2152 | if(_config->FindB("Acquire::AllowInsecureRepositories") == true) | |
2153 | { | |
2154 | // Done, queue for rename on transaction finished | |
2155 | if (FileExists(DestFile)) | |
2156 | TransactionManager->TransactionStageCopy(this, DestFile, FinalFile); | |
2157 | ||
2158 | // queue without any kind of hashsum support | |
2159 | QueueIndexes(false); | |
2160 | } else { | |
2161 | // warn if the repository is unsinged | |
2162 | _error->Warning("Use --allow-insecure-repositories to force the update"); | |
2163 | TransactionManager->AbortTransaction(); | |
2164 | Status = StatError; | |
2165 | return; | |
2166 | } | |
2167 | } | |
2168 | /*}}}*/ | |
2169 | void pkgAcqMetaIndex::Finished() /*{{{*/ | |
2170 | { | |
2171 | if(_config->FindB("Debug::Acquire::Transaction", false) == true) | |
2172 | std::clog << "Finished: " << DestFile <<std::endl; | |
2173 | if(TransactionManager != NULL && | |
2174 | TransactionManager->TransactionHasError() == false) | |
2175 | TransactionManager->CommitTransaction(); | |
2176 | } | |
2177 | /*}}}*/ | |
2178 | pkgAcqMetaClearSig::pkgAcqMetaClearSig(pkgAcquire *Owner, /*{{{*/ | |
2179 | string const &URI, string const &URIDesc, string const &ShortDesc, | |
2180 | string const &MetaIndexURI, string const &MetaIndexURIDesc, string const &MetaIndexShortDesc, | |
2181 | string const &MetaSigURI, string const &MetaSigURIDesc, string const &MetaSigShortDesc, | |
2182 | const vector<IndexTarget*>* IndexTargets, | |
2183 | indexRecords* MetaIndexParser) : | |
2184 | pkgAcqMetaIndex(Owner, NULL, URI, URIDesc, ShortDesc, MetaSigURI, MetaSigURIDesc,MetaSigShortDesc, IndexTargets, MetaIndexParser), | |
2185 | MetaIndexURI(MetaIndexURI), MetaIndexURIDesc(MetaIndexURIDesc), MetaIndexShortDesc(MetaIndexShortDesc), | |
2186 | MetaSigURI(MetaSigURI), MetaSigURIDesc(MetaSigURIDesc), MetaSigShortDesc(MetaSigShortDesc) | |
2187 | { | |
2188 | // index targets + (worst case:) Release/Release.gpg | |
2189 | ExpectedAdditionalItems = IndexTargets->size() + 2; | |
2190 | ||
2191 | } | |
2192 | /*}}}*/ | |
2193 | pkgAcqMetaClearSig::~pkgAcqMetaClearSig() /*{{{*/ | |
2194 | { | |
2195 | } | |
2196 | /*}}}*/ | |
2197 | // pkgAcqMetaClearSig::Custom600Headers - Insert custom request headers /*{{{*/ | |
2198 | // --------------------------------------------------------------------- | |
2199 | string pkgAcqMetaClearSig::Custom600Headers() const | |
2200 | { | |
2201 | string Header = GetCustom600Headers(RealURI); | |
2202 | Header += "\nFail-Ignore: true"; | |
2203 | return Header; | |
2204 | } | |
2205 | /*}}}*/ | |
2206 | // pkgAcqMetaClearSig::Done - We got a file /*{{{*/ | |
2207 | // --------------------------------------------------------------------- | |
2208 | void pkgAcqMetaClearSig::Done(std::string Message,unsigned long long /*Size*/, | |
2209 | HashStringList const &/*Hashes*/, | |
2210 | pkgAcquire::MethodConfig *Cnf) | |
2211 | { | |
2212 | // if we expect a ClearTextSignature (InRelase), ensure that | |
2213 | // this is what we get and if not fail to queue a | |
2214 | // Release/Release.gpg, see #346386 | |
2215 | if (FileExists(DestFile) && !StartsWithGPGClearTextSignature(DestFile)) | |
2216 | { | |
2217 | pkgAcquire::Item::Failed(Message, Cnf); | |
2218 | RenameOnError(NotClearsigned); | |
2219 | TransactionManager->AbortTransaction(); | |
2220 | return; | |
2221 | } | |
2222 | ||
2223 | if(AuthPass == false) | |
2224 | { | |
2225 | if(CheckDownloadDone(Message, RealURI) == true) | |
2226 | QueueForSignatureVerify(DestFile, DestFile); | |
2227 | return; | |
2228 | } | |
2229 | else | |
2230 | { | |
2231 | if(CheckAuthDone(Message, RealURI) == true) | |
2232 | { | |
2233 | string FinalFile = _config->FindDir("Dir::State::lists"); | |
2234 | FinalFile += URItoFileName(RealURI); | |
2235 | ||
2236 | // queue for copy in place | |
2237 | TransactionManager->TransactionStageCopy(this, DestFile, FinalFile); | |
2238 | } | |
2239 | } | |
2240 | } | |
2241 | /*}}}*/ | |
2242 | void pkgAcqMetaClearSig::Failed(string Message,pkgAcquire::MethodConfig *Cnf) /*{{{*/ | |
2243 | { | |
2244 | Item::Failed(Message, Cnf); | |
2245 | ||
2246 | // we failed, we will not get additional items from this method | |
2247 | ExpectedAdditionalItems = 0; | |
2248 | ||
2249 | if (AuthPass == false) | |
2250 | { | |
2251 | // Queue the 'old' InRelease file for removal if we try Release.gpg | |
2252 | // as otherwise the file will stay around and gives a false-auth | |
2253 | // impression (CVE-2012-0214) | |
2254 | string FinalFile = _config->FindDir("Dir::State::lists"); | |
2255 | FinalFile.append(URItoFileName(RealURI)); | |
2256 | TransactionManager->TransactionStageRemoval(this, FinalFile); | |
2257 | Status = StatDone; | |
2258 | ||
2259 | new pkgAcqMetaIndex(Owner, TransactionManager, | |
2260 | MetaIndexURI, MetaIndexURIDesc, MetaIndexShortDesc, | |
2261 | MetaSigURI, MetaSigURIDesc, MetaSigShortDesc, | |
2262 | IndexTargets, MetaIndexParser); | |
2263 | } | |
2264 | else | |
2265 | { | |
2266 | if(CheckStopAuthentication(RealURI, Message)) | |
2267 | return; | |
2268 | ||
2269 | _error->Warning(_("The data from '%s' is not signed. Packages " | |
2270 | "from that repository can not be authenticated."), | |
2271 | URIDesc.c_str()); | |
2272 | ||
2273 | // No Release file was present, or verification failed, so fall | |
2274 | // back to queueing Packages files without verification | |
2275 | // only allow going further if the users explicitely wants it | |
2276 | if(_config->FindB("Acquire::AllowInsecureRepositories") == true) | |
2277 | { | |
2278 | Status = StatDone; | |
2279 | ||
2280 | /* Always move the meta index, even if gpgv failed. This ensures | |
2281 | * that PackageFile objects are correctly filled in */ | |
2282 | if (FileExists(DestFile)) | |
2283 | { | |
2284 | string FinalFile = _config->FindDir("Dir::State::lists"); | |
2285 | FinalFile += URItoFileName(RealURI); | |
2286 | /* InRelease files become Release files, otherwise | |
2287 | * they would be considered as trusted later on */ | |
2288 | RealURI = RealURI.replace(RealURI.rfind("InRelease"), 9, | |
2289 | "Release"); | |
2290 | FinalFile = FinalFile.replace(FinalFile.rfind("InRelease"), 9, | |
2291 | "Release"); | |
2292 | ||
2293 | // Done, queue for rename on transaction finished | |
2294 | TransactionManager->TransactionStageCopy(this, DestFile, FinalFile); | |
2295 | } | |
2296 | QueueIndexes(false); | |
2297 | } else { | |
2298 | // warn if the repository is unsigned | |
2299 | _error->Warning("Use --allow-insecure-repositories to force the update"); | |
2300 | TransactionManager->AbortTransaction(); | |
2301 | Status = StatError; | |
2302 | } | |
2303 | } | |
2304 | } | |
2305 | /*}}}*/ | |
2306 | // AcqArchive::AcqArchive - Constructor /*{{{*/ | |
2307 | // --------------------------------------------------------------------- | |
2308 | /* This just sets up the initial fetch environment and queues the first | |
2309 | possibilitiy */ | |
2310 | pkgAcqArchive::pkgAcqArchive(pkgAcquire *Owner,pkgSourceList *Sources, | |
2311 | pkgRecords *Recs,pkgCache::VerIterator const &Version, | |
2312 | string &StoreFilename) : | |
2313 | Item(Owner, HashStringList()), Version(Version), Sources(Sources), Recs(Recs), | |
2314 | StoreFilename(StoreFilename), Vf(Version.FileList()), | |
2315 | Trusted(false) | |
2316 | { | |
2317 | Retries = _config->FindI("Acquire::Retries",0); | |
2318 | ||
2319 | if (Version.Arch() == 0) | |
2320 | { | |
2321 | _error->Error(_("I wasn't able to locate a file for the %s package. " | |
2322 | "This might mean you need to manually fix this package. " | |
2323 | "(due to missing arch)"), | |
2324 | Version.ParentPkg().FullName().c_str()); | |
2325 | return; | |
2326 | } | |
2327 | ||
2328 | /* We need to find a filename to determine the extension. We make the | |
2329 | assumption here that all the available sources for this version share | |
2330 | the same extension.. */ | |
2331 | // Skip not source sources, they do not have file fields. | |
2332 | for (; Vf.end() == false; ++Vf) | |
2333 | { | |
2334 | if ((Vf.File()->Flags & pkgCache::Flag::NotSource) != 0) | |
2335 | continue; | |
2336 | break; | |
2337 | } | |
2338 | ||
2339 | // Does not really matter here.. we are going to fail out below | |
2340 | if (Vf.end() != true) | |
2341 | { | |
2342 | // If this fails to get a file name we will bomb out below. | |
2343 | pkgRecords::Parser &Parse = Recs->Lookup(Vf); | |
2344 | if (_error->PendingError() == true) | |
2345 | return; | |
2346 | ||
2347 | // Generate the final file name as: package_version_arch.foo | |
2348 | StoreFilename = QuoteString(Version.ParentPkg().Name(),"_:") + '_' + | |
2349 | QuoteString(Version.VerStr(),"_:") + '_' + | |
2350 | QuoteString(Version.Arch(),"_:.") + | |
2351 | "." + flExtension(Parse.FileName()); | |
2352 | } | |
2353 | ||
2354 | // check if we have one trusted source for the package. if so, switch | |
2355 | // to "TrustedOnly" mode - but only if not in AllowUnauthenticated mode | |
2356 | bool const allowUnauth = _config->FindB("APT::Get::AllowUnauthenticated", false); | |
2357 | bool const debugAuth = _config->FindB("Debug::pkgAcquire::Auth", false); | |
2358 | bool seenUntrusted = false; | |
2359 | for (pkgCache::VerFileIterator i = Version.FileList(); i.end() == false; ++i) | |
2360 | { | |
2361 | pkgIndexFile *Index; | |
2362 | if (Sources->FindIndex(i.File(),Index) == false) | |
2363 | continue; | |
2364 | ||
2365 | if (debugAuth == true) | |
2366 | std::cerr << "Checking index: " << Index->Describe() | |
2367 | << "(Trusted=" << Index->IsTrusted() << ")" << std::endl; | |
2368 | ||
2369 | if (Index->IsTrusted() == true) | |
2370 | { | |
2371 | Trusted = true; | |
2372 | if (allowUnauth == false) | |
2373 | break; | |
2374 | } | |
2375 | else | |
2376 | seenUntrusted = true; | |
2377 | } | |
2378 | ||
2379 | // "allow-unauthenticated" restores apts old fetching behaviour | |
2380 | // that means that e.g. unauthenticated file:// uris are higher | |
2381 | // priority than authenticated http:// uris | |
2382 | if (allowUnauth == true && seenUntrusted == true) | |
2383 | Trusted = false; | |
2384 | ||
2385 | // Select a source | |
2386 | if (QueueNext() == false && _error->PendingError() == false) | |
2387 | _error->Error(_("Can't find a source to download version '%s' of '%s'"), | |
2388 | Version.VerStr(), Version.ParentPkg().FullName(false).c_str()); | |
2389 | } | |
2390 | /*}}}*/ | |
2391 | // AcqArchive::QueueNext - Queue the next file source /*{{{*/ | |
2392 | // --------------------------------------------------------------------- | |
2393 | /* This queues the next available file version for download. It checks if | |
2394 | the archive is already available in the cache and stashs the MD5 for | |
2395 | checking later. */ | |
2396 | bool pkgAcqArchive::QueueNext() | |
2397 | { | |
2398 | for (; Vf.end() == false; ++Vf) | |
2399 | { | |
2400 | // Ignore not source sources | |
2401 | if ((Vf.File()->Flags & pkgCache::Flag::NotSource) != 0) | |
2402 | continue; | |
2403 | ||
2404 | // Try to cross match against the source list | |
2405 | pkgIndexFile *Index; | |
2406 | if (Sources->FindIndex(Vf.File(),Index) == false) | |
2407 | continue; | |
2408 | ||
2409 | // only try to get a trusted package from another source if that source | |
2410 | // is also trusted | |
2411 | if(Trusted && !Index->IsTrusted()) | |
2412 | continue; | |
2413 | ||
2414 | // Grab the text package record | |
2415 | pkgRecords::Parser &Parse = Recs->Lookup(Vf); | |
2416 | if (_error->PendingError() == true) | |
2417 | return false; | |
2418 | ||
2419 | string PkgFile = Parse.FileName(); | |
2420 | ExpectedHashes = Parse.Hashes(); | |
2421 | ||
2422 | if (PkgFile.empty() == true) | |
2423 | return _error->Error(_("The package index files are corrupted. No Filename: " | |
2424 | "field for package %s."), | |
2425 | Version.ParentPkg().Name()); | |
2426 | ||
2427 | Desc.URI = Index->ArchiveURI(PkgFile); | |
2428 | Desc.Description = Index->ArchiveInfo(Version); | |
2429 | Desc.Owner = this; | |
2430 | Desc.ShortDesc = Version.ParentPkg().FullName(true); | |
2431 | ||
2432 | // See if we already have the file. (Legacy filenames) | |
2433 | FileSize = Version->Size; | |
2434 | string FinalFile = _config->FindDir("Dir::Cache::Archives") + flNotDir(PkgFile); | |
2435 | struct stat Buf; | |
2436 | if (stat(FinalFile.c_str(),&Buf) == 0) | |
2437 | { | |
2438 | // Make sure the size matches | |
2439 | if ((unsigned long long)Buf.st_size == Version->Size) | |
2440 | { | |
2441 | Complete = true; | |
2442 | Local = true; | |
2443 | Status = StatDone; | |
2444 | StoreFilename = DestFile = FinalFile; | |
2445 | return true; | |
2446 | } | |
2447 | ||
2448 | /* Hmm, we have a file and its size does not match, this means it is | |
2449 | an old style mismatched arch */ | |
2450 | unlink(FinalFile.c_str()); | |
2451 | } | |
2452 | ||
2453 | // Check it again using the new style output filenames | |
2454 | FinalFile = _config->FindDir("Dir::Cache::Archives") + flNotDir(StoreFilename); | |
2455 | if (stat(FinalFile.c_str(),&Buf) == 0) | |
2456 | { | |
2457 | // Make sure the size matches | |
2458 | if ((unsigned long long)Buf.st_size == Version->Size) | |
2459 | { | |
2460 | Complete = true; | |
2461 | Local = true; | |
2462 | Status = StatDone; | |
2463 | StoreFilename = DestFile = FinalFile; | |
2464 | return true; | |
2465 | } | |
2466 | ||
2467 | /* Hmm, we have a file and its size does not match, this shouldn't | |
2468 | happen.. */ | |
2469 | unlink(FinalFile.c_str()); | |
2470 | } | |
2471 | ||
2472 | DestFile = _config->FindDir("Dir::Cache::Archives") + "partial/" + flNotDir(StoreFilename); | |
2473 | ||
2474 | // Check the destination file | |
2475 | if (stat(DestFile.c_str(),&Buf) == 0) | |
2476 | { | |
2477 | // Hmm, the partial file is too big, erase it | |
2478 | if ((unsigned long long)Buf.st_size > Version->Size) | |
2479 | unlink(DestFile.c_str()); | |
2480 | else | |
2481 | { | |
2482 | PartialSize = Buf.st_size; | |
2483 | ChangeOwnerAndPermissionOfFile("pkgAcqArchive::QueueNext", FinalFile.c_str(), "_apt", "root", 0600); | |
2484 | } | |
2485 | } | |
2486 | ||
2487 | // Disables download of archives - useful if no real installation follows, | |
2488 | // e.g. if we are just interested in proposed installation order | |
2489 | if (_config->FindB("Debug::pkgAcqArchive::NoQueue", false) == true) | |
2490 | { | |
2491 | Complete = true; | |
2492 | Local = true; | |
2493 | Status = StatDone; | |
2494 | StoreFilename = DestFile = FinalFile; | |
2495 | return true; | |
2496 | } | |
2497 | ||
2498 | // Create the item | |
2499 | Local = false; | |
2500 | QueueURI(Desc); | |
2501 | ||
2502 | ++Vf; | |
2503 | return true; | |
2504 | } | |
2505 | return false; | |
2506 | } | |
2507 | /*}}}*/ | |
2508 | // AcqArchive::Done - Finished fetching /*{{{*/ | |
2509 | // --------------------------------------------------------------------- | |
2510 | /* */ | |
2511 | void pkgAcqArchive::Done(string Message,unsigned long long Size, HashStringList const &CalcHashes, | |
2512 | pkgAcquire::MethodConfig *Cfg) | |
2513 | { | |
2514 | Item::Done(Message, Size, CalcHashes, Cfg); | |
2515 | ||
2516 | // Check the size | |
2517 | if (Size != Version->Size) | |
2518 | { | |
2519 | RenameOnError(SizeMismatch); | |
2520 | return; | |
2521 | } | |
2522 | ||
2523 | // FIXME: could this empty() check impose *any* sort of security issue? | |
2524 | if(ExpectedHashes.usable() && ExpectedHashes != CalcHashes) | |
2525 | { | |
2526 | RenameOnError(HashSumMismatch); | |
2527 | printHashSumComparision(DestFile, ExpectedHashes, CalcHashes); | |
2528 | return; | |
2529 | } | |
2530 | ||
2531 | // Grab the output filename | |
2532 | string FileName = LookupTag(Message,"Filename"); | |
2533 | if (FileName.empty() == true) | |
2534 | { | |
2535 | Status = StatError; | |
2536 | ErrorText = "Method gave a blank filename"; | |
2537 | return; | |
2538 | } | |
2539 | ||
2540 | // Reference filename | |
2541 | if (FileName != DestFile) | |
2542 | { | |
2543 | StoreFilename = DestFile = FileName; | |
2544 | Local = true; | |
2545 | Complete = true; | |
2546 | return; | |
2547 | } | |
2548 | ||
2549 | // Done, move it into position | |
2550 | string FinalFile = _config->FindDir("Dir::Cache::Archives"); | |
2551 | FinalFile += flNotDir(StoreFilename); | |
2552 | Rename(DestFile,FinalFile); | |
2553 | ChangeOwnerAndPermissionOfFile("pkgAcqArchive::Done", FinalFile.c_str(), "root", "root", 0644); | |
2554 | StoreFilename = DestFile = FinalFile; | |
2555 | Complete = true; | |
2556 | } | |
2557 | /*}}}*/ | |
2558 | // AcqArchive::Failed - Failure handler /*{{{*/ | |
2559 | // --------------------------------------------------------------------- | |
2560 | /* Here we try other sources */ | |
2561 | void pkgAcqArchive::Failed(string Message,pkgAcquire::MethodConfig *Cnf) | |
2562 | { | |
2563 | ErrorText = LookupTag(Message,"Message"); | |
2564 | ||
2565 | /* We don't really want to retry on failed media swaps, this prevents | |
2566 | that. An interesting observation is that permanent failures are not | |
2567 | recorded. */ | |
2568 | if (Cnf->Removable == true && | |
2569 | StringToBool(LookupTag(Message,"Transient-Failure"),false) == true) | |
2570 | { | |
2571 | // Vf = Version.FileList(); | |
2572 | while (Vf.end() == false) ++Vf; | |
2573 | StoreFilename = string(); | |
2574 | Item::Failed(Message,Cnf); | |
2575 | return; | |
2576 | } | |
2577 | ||
2578 | if (QueueNext() == false) | |
2579 | { | |
2580 | // This is the retry counter | |
2581 | if (Retries != 0 && | |
2582 | Cnf->LocalOnly == false && | |
2583 | StringToBool(LookupTag(Message,"Transient-Failure"),false) == true) | |
2584 | { | |
2585 | Retries--; | |
2586 | Vf = Version.FileList(); | |
2587 | if (QueueNext() == true) | |
2588 | return; | |
2589 | } | |
2590 | ||
2591 | StoreFilename = string(); | |
2592 | Item::Failed(Message,Cnf); | |
2593 | } | |
2594 | } | |
2595 | /*}}}*/ | |
2596 | // AcqArchive::IsTrusted - Determine whether this archive comes from a trusted source /*{{{*/ | |
2597 | // --------------------------------------------------------------------- | |
2598 | APT_PURE bool pkgAcqArchive::IsTrusted() const | |
2599 | { | |
2600 | return Trusted; | |
2601 | } | |
2602 | /*}}}*/ | |
2603 | // AcqArchive::Finished - Fetching has finished, tidy up /*{{{*/ | |
2604 | // --------------------------------------------------------------------- | |
2605 | /* */ | |
2606 | void pkgAcqArchive::Finished() | |
2607 | { | |
2608 | if (Status == pkgAcquire::Item::StatDone && | |
2609 | Complete == true) | |
2610 | return; | |
2611 | StoreFilename = string(); | |
2612 | } | |
2613 | /*}}}*/ | |
2614 | // AcqFile::pkgAcqFile - Constructor /*{{{*/ | |
2615 | // --------------------------------------------------------------------- | |
2616 | /* The file is added to the queue */ | |
2617 | pkgAcqFile::pkgAcqFile(pkgAcquire *Owner,string URI, HashStringList const &Hashes, | |
2618 | unsigned long long Size,string Dsc,string ShortDesc, | |
2619 | const string &DestDir, const string &DestFilename, | |
2620 | bool IsIndexFile) : | |
2621 | Item(Owner, Hashes), IsIndexFile(IsIndexFile) | |
2622 | { | |
2623 | Retries = _config->FindI("Acquire::Retries",0); | |
2624 | ||
2625 | if(!DestFilename.empty()) | |
2626 | DestFile = DestFilename; | |
2627 | else if(!DestDir.empty()) | |
2628 | DestFile = DestDir + "/" + flNotDir(URI); | |
2629 | else | |
2630 | DestFile = flNotDir(URI); | |
2631 | ||
2632 | // Create the item | |
2633 | Desc.URI = URI; | |
2634 | Desc.Description = Dsc; | |
2635 | Desc.Owner = this; | |
2636 | ||
2637 | // Set the short description to the archive component | |
2638 | Desc.ShortDesc = ShortDesc; | |
2639 | ||
2640 | // Get the transfer sizes | |
2641 | FileSize = Size; | |
2642 | struct stat Buf; | |
2643 | if (stat(DestFile.c_str(),&Buf) == 0) | |
2644 | { | |
2645 | // Hmm, the partial file is too big, erase it | |
2646 | if ((Size > 0) && (unsigned long long)Buf.st_size > Size) | |
2647 | unlink(DestFile.c_str()); | |
2648 | else | |
2649 | { | |
2650 | PartialSize = Buf.st_size; | |
2651 | ChangeOwnerAndPermissionOfFile("pkgAcqFile", DestFile.c_str(), "_apt", "root", 0600); | |
2652 | } | |
2653 | } | |
2654 | ||
2655 | QueueURI(Desc); | |
2656 | } | |
2657 | /*}}}*/ | |
2658 | // AcqFile::Done - Item downloaded OK /*{{{*/ | |
2659 | // --------------------------------------------------------------------- | |
2660 | /* */ | |
2661 | void pkgAcqFile::Done(string Message,unsigned long long Size,HashStringList const &CalcHashes, | |
2662 | pkgAcquire::MethodConfig *Cnf) | |
2663 | { | |
2664 | Item::Done(Message,Size,CalcHashes,Cnf); | |
2665 | ||
2666 | // Check the hash | |
2667 | if(ExpectedHashes.usable() && ExpectedHashes != CalcHashes) | |
2668 | { | |
2669 | RenameOnError(HashSumMismatch); | |
2670 | printHashSumComparision(DestFile, ExpectedHashes, CalcHashes); | |
2671 | return; | |
2672 | } | |
2673 | ||
2674 | string FileName = LookupTag(Message,"Filename"); | |
2675 | if (FileName.empty() == true) | |
2676 | { | |
2677 | Status = StatError; | |
2678 | ErrorText = "Method gave a blank filename"; | |
2679 | return; | |
2680 | } | |
2681 | ||
2682 | Complete = true; | |
2683 | ||
2684 | // The files timestamp matches | |
2685 | if (StringToBool(LookupTag(Message,"IMS-Hit"),false) == true) | |
2686 | return; | |
2687 | ||
2688 | // We have to copy it into place | |
2689 | if (FileName != DestFile) | |
2690 | { | |
2691 | Local = true; | |
2692 | if (_config->FindB("Acquire::Source-Symlinks",true) == false || | |
2693 | Cnf->Removable == true) | |
2694 | { | |
2695 | Desc.URI = "copy:" + FileName; | |
2696 | QueueURI(Desc); | |
2697 | return; | |
2698 | } | |
2699 | ||
2700 | // Erase the file if it is a symlink so we can overwrite it | |
2701 | struct stat St; | |
2702 | if (lstat(DestFile.c_str(),&St) == 0) | |
2703 | { | |
2704 | if (S_ISLNK(St.st_mode) != 0) | |
2705 | unlink(DestFile.c_str()); | |
2706 | } | |
2707 | ||
2708 | // Symlink the file | |
2709 | if (symlink(FileName.c_str(),DestFile.c_str()) != 0) | |
2710 | { | |
2711 | ErrorText = "Link to " + DestFile + " failure "; | |
2712 | Status = StatError; | |
2713 | Complete = false; | |
2714 | } | |
2715 | } | |
2716 | } | |
2717 | /*}}}*/ | |
2718 | // AcqFile::Failed - Failure handler /*{{{*/ | |
2719 | // --------------------------------------------------------------------- | |
2720 | /* Here we try other sources */ | |
2721 | void pkgAcqFile::Failed(string Message,pkgAcquire::MethodConfig *Cnf) | |
2722 | { | |
2723 | ErrorText = LookupTag(Message,"Message"); | |
2724 | ||
2725 | // This is the retry counter | |
2726 | if (Retries != 0 && | |
2727 | Cnf->LocalOnly == false && | |
2728 | StringToBool(LookupTag(Message,"Transient-Failure"),false) == true) | |
2729 | { | |
2730 | Retries--; | |
2731 | QueueURI(Desc); | |
2732 | return; | |
2733 | } | |
2734 | ||
2735 | Item::Failed(Message,Cnf); | |
2736 | } | |
2737 | /*}}}*/ | |
2738 | // AcqIndex::Custom600Headers - Insert custom request headers /*{{{*/ | |
2739 | // --------------------------------------------------------------------- | |
2740 | /* The only header we use is the last-modified header. */ | |
2741 | string pkgAcqFile::Custom600Headers() const | |
2742 | { | |
2743 | if (IsIndexFile) | |
2744 | return "\nIndex-File: true"; | |
2745 | return ""; | |
2746 | } | |
2747 | /*}}}*/ |