merged from lp:~donkult/apt/sid
[ntk/apt.git] / apt-pkg / cacheset.cc
1 // -*- mode: cpp; mode: fold -*-
2 // Description /*{{{*/
3 /* ######################################################################
4
5 Simple wrapper around a std::set to provide a similar interface to
6 a set of cache structures as to the complete set of all structures
7 in the pkgCache. Currently only Package is supported.
8
9 ##################################################################### */
10 /*}}}*/
11 // Include Files /*{{{*/
12 #include <config.h>
13
14 #include <apt-pkg/aptconfiguration.h>
15 #include <apt-pkg/cachefile.h>
16 #include <apt-pkg/cachefilter.h>
17 #include <apt-pkg/cacheset.h>
18 #include <apt-pkg/error.h>
19 #include <apt-pkg/strutl.h>
20 #include <apt-pkg/versionmatch.h>
21 #include <apt-pkg/pkgrecords.h>
22 #include <apt-pkg/policy.h>
23
24 #include <vector>
25
26 #include <regex.h>
27
28 #include <apti18n.h>
29 /*}}}*/
30 namespace APT {
31 // FromTask - Return all packages in the cache from a specific task /*{{{*/
32 bool PackageContainerInterface::FromTask(PackageContainerInterface * const pci, pkgCacheFile &Cache, std::string pattern, CacheSetHelper &helper) {
33 size_t const archfound = pattern.find_last_of(':');
34 std::string arch = "native";
35 if (archfound != std::string::npos) {
36 arch = pattern.substr(archfound+1);
37 pattern.erase(archfound);
38 }
39
40 if (pattern[pattern.length() -1] != '^')
41 return false;
42 pattern.erase(pattern.length()-1);
43
44 if (unlikely(Cache.GetPkgCache() == 0 || Cache.GetDepCache() == 0))
45 return false;
46
47 bool const wasEmpty = pci->empty();
48 if (wasEmpty == true)
49 pci->setConstructor(TASK);
50
51 // get the records
52 pkgRecords Recs(Cache);
53
54 // build regexp for the task
55 regex_t Pattern;
56 char S[300];
57 snprintf(S, sizeof(S), "^Task:.*[, ]%s([, ]|$)", pattern.c_str());
58 if(regcomp(&Pattern,S, REG_EXTENDED | REG_NOSUB | REG_NEWLINE) != 0) {
59 _error->Error("Failed to compile task regexp");
60 return false;
61 }
62
63 bool found = false;
64 for (pkgCache::GrpIterator Grp = Cache->GrpBegin(); Grp.end() == false; ++Grp) {
65 pkgCache::PkgIterator Pkg = Grp.FindPkg(arch);
66 if (Pkg.end() == true)
67 continue;
68 pkgCache::VerIterator ver = Cache[Pkg].CandidateVerIter(Cache);
69 if(ver.end() == true)
70 continue;
71
72 pkgRecords::Parser &parser = Recs.Lookup(ver.FileList());
73 const char *start, *end;
74 parser.GetRec(start,end);
75 unsigned int const length = end - start;
76 char buf[length];
77 strncpy(buf, start, length);
78 buf[length-1] = '\0';
79 if (regexec(&Pattern, buf, 0, 0, 0) != 0)
80 continue;
81
82 pci->insert(Pkg);
83 helper.showTaskSelection(Pkg, pattern);
84 found = true;
85 }
86 regfree(&Pattern);
87
88 if (found == false) {
89 helper.canNotFindTask(pci, Cache, pattern);
90 pci->setConstructor(UNKNOWN);
91 return false;
92 }
93
94 if (wasEmpty == false && pci->getConstructor() != UNKNOWN)
95 pci->setConstructor(UNKNOWN);
96
97 return true;
98 }
99 /*}}}*/
100 // FromRegEx - Return all packages in the cache matching a pattern /*{{{*/
101 bool PackageContainerInterface::FromRegEx(PackageContainerInterface * const pci, pkgCacheFile &Cache, std::string pattern, CacheSetHelper &helper) {
102 static const char * const isregex = ".?+*|[^$";
103 if (pattern.find_first_of(isregex) == std::string::npos)
104 return false;
105
106 bool const wasEmpty = pci->empty();
107 if (wasEmpty == true)
108 pci->setConstructor(REGEX);
109
110 size_t archfound = pattern.find_last_of(':');
111 std::string arch = "native";
112 if (archfound != std::string::npos) {
113 arch = pattern.substr(archfound+1);
114 if (arch.find_first_of(isregex) == std::string::npos)
115 pattern.erase(archfound);
116 else
117 arch = "native";
118 }
119
120 if (unlikely(Cache.GetPkgCache() == 0))
121 return false;
122
123 APT::CacheFilter::PackageNameMatchesRegEx regexfilter(pattern);
124
125 bool found = false;
126 for (pkgCache::GrpIterator Grp = Cache.GetPkgCache()->GrpBegin(); Grp.end() == false; ++Grp) {
127 if (regexfilter(Grp) == false)
128 continue;
129 pkgCache::PkgIterator Pkg = Grp.FindPkg(arch);
130 if (Pkg.end() == true) {
131 if (archfound == std::string::npos) {
132 std::vector<std::string> archs = APT::Configuration::getArchitectures();
133 for (std::vector<std::string>::const_iterator a = archs.begin();
134 a != archs.end() && Pkg.end() != true; ++a)
135 Pkg = Grp.FindPkg(*a);
136 }
137 if (Pkg.end() == true)
138 continue;
139 }
140
141 pci->insert(Pkg);
142 helper.showRegExSelection(Pkg, pattern);
143 found = true;
144 }
145
146 if (found == false) {
147 helper.canNotFindRegEx(pci, Cache, pattern);
148 pci->setConstructor(UNKNOWN);
149 return false;
150 }
151
152 if (wasEmpty == false && pci->getConstructor() != UNKNOWN)
153 pci->setConstructor(UNKNOWN);
154
155 return true;
156 }
157 /*}}}*/
158 // FromName - Returns the package defined by this string /*{{{*/
159 pkgCache::PkgIterator PackageContainerInterface::FromName(pkgCacheFile &Cache,
160 std::string const &str, CacheSetHelper &helper) {
161 std::string pkg = str;
162 size_t archfound = pkg.find_last_of(':');
163 std::string arch;
164 if (archfound != std::string::npos) {
165 arch = pkg.substr(archfound+1);
166 pkg.erase(archfound);
167 }
168
169 if (Cache.GetPkgCache() == 0)
170 return pkgCache::PkgIterator(Cache, 0);
171
172 pkgCache::PkgIterator Pkg(Cache, 0);
173 if (arch.empty() == true) {
174 pkgCache::GrpIterator Grp = Cache.GetPkgCache()->FindGrp(pkg);
175 if (Grp.end() == false)
176 Pkg = Grp.FindPreferredPkg();
177 } else
178 Pkg = Cache.GetPkgCache()->FindPkg(pkg, arch);
179
180 if (Pkg.end() == true)
181 return helper.canNotFindPkgName(Cache, str);
182 return Pkg;
183 }
184 /*}}}*/
185 // FromString - Return all packages matching a specific string /*{{{*/
186 bool PackageContainerInterface::FromString(PackageContainerInterface * const pci, pkgCacheFile &Cache, std::string const &str, CacheSetHelper &helper) {
187 bool found = true;
188 _error->PushToStack();
189
190 pkgCache::PkgIterator Pkg = FromName(Cache, str, helper);
191 if (Pkg.end() == false)
192 pci->insert(Pkg);
193 else if (FromTask(pci, Cache, str, helper) == false &&
194 FromRegEx(pci, Cache, str, helper) == false)
195 {
196 helper.canNotFindPackage(pci, Cache, str);
197 found = false;
198 }
199
200 if (found == true)
201 _error->RevertToStack();
202 else
203 _error->MergeWithStack();
204 return found;
205 }
206 /*}}}*/
207 // FromCommandLine - Return all packages specified on commandline /*{{{*/
208 bool PackageContainerInterface::FromCommandLine(PackageContainerInterface * const pci, pkgCacheFile &Cache, const char **cmdline, CacheSetHelper &helper) {
209 bool found = false;
210 for (const char **I = cmdline; *I != 0; ++I)
211 found |= PackageContainerInterface::FromString(pci, Cache, *I, helper);
212 return found;
213 }
214 /*}}}*/
215 // FromModifierCommandLine - helper doing the work for PKG:GroupedFromCommandLine /*{{{*/
216 bool PackageContainerInterface::FromModifierCommandLine(unsigned short &modID, PackageContainerInterface * const pci,
217 pkgCacheFile &Cache, const char * cmdline,
218 std::list<Modifier> const &mods, CacheSetHelper &helper) {
219 std::string str = cmdline;
220 unsigned short fallback = modID;
221 bool modifierPresent = false;
222 for (std::list<Modifier>::const_iterator mod = mods.begin();
223 mod != mods.end(); ++mod) {
224 size_t const alength = strlen(mod->Alias);
225 switch(mod->Pos) {
226 case Modifier::POSTFIX:
227 if (str.compare(str.length() - alength, alength,
228 mod->Alias, 0, alength) != 0)
229 continue;
230 str.erase(str.length() - alength);
231 modID = mod->ID;
232 break;
233 case Modifier::PREFIX:
234 continue;
235 case Modifier::NONE:
236 continue;
237 }
238 modifierPresent = true;
239 break;
240 }
241 if (modifierPresent == true) {
242 bool const errors = helper.showErrors(false);
243 pkgCache::PkgIterator Pkg = FromName(Cache, cmdline, helper);
244 helper.showErrors(errors);
245 if (Pkg.end() == false) {
246 pci->insert(Pkg);
247 modID = fallback;
248 return true;
249 }
250 }
251 return FromString(pci, Cache, str, helper);
252 }
253 /*}}}*/
254 // FromModifierCommandLine - helper doing the work for VER:GroupedFromCommandLine /*{{{*/
255 bool VersionContainerInterface::FromModifierCommandLine(unsigned short &modID,
256 VersionContainerInterface * const vci,
257 pkgCacheFile &Cache, const char * cmdline,
258 std::list<Modifier> const &mods,
259 CacheSetHelper &helper) {
260 Version select = NEWEST;
261 std::string str = cmdline;
262 bool modifierPresent = false;
263 unsigned short fallback = modID;
264 for (std::list<Modifier>::const_iterator mod = mods.begin();
265 mod != mods.end(); ++mod) {
266 if (modID == fallback && mod->ID == fallback)
267 select = mod->SelectVersion;
268 size_t const alength = strlen(mod->Alias);
269 switch(mod->Pos) {
270 case Modifier::POSTFIX:
271 if (str.compare(str.length() - alength, alength,
272 mod->Alias, 0, alength) != 0)
273 continue;
274 str.erase(str.length() - alength);
275 modID = mod->ID;
276 select = mod->SelectVersion;
277 break;
278 case Modifier::PREFIX:
279 continue;
280 case Modifier::NONE:
281 continue;
282 }
283 modifierPresent = true;
284 break;
285 }
286 if (modifierPresent == true) {
287 bool const errors = helper.showErrors(false);
288 bool const found = VersionContainerInterface::FromString(vci, Cache, cmdline, select, helper, true);
289 helper.showErrors(errors);
290 if (found == true) {
291 modID = fallback;
292 return true;
293 }
294 }
295 return FromString(vci, Cache, str, select, helper);
296 }
297 /*}}}*/
298 // FromCommandLine - Return all versions specified on commandline /*{{{*/
299 bool VersionContainerInterface::FromCommandLine(VersionContainerInterface * const vci,
300 pkgCacheFile &Cache, const char **cmdline,
301 Version const &fallback, CacheSetHelper &helper) {
302 bool found = false;
303 for (const char **I = cmdline; *I != 0; ++I)
304 found |= VersionContainerInterface::FromString(vci, Cache, *I, fallback, helper);
305 return found;
306 }
307 /*}}}*/
308 // FromString - Returns all versions spedcified by a string /*{{{*/
309 bool VersionContainerInterface::FromString(VersionContainerInterface * const vci,
310 pkgCacheFile &Cache, std::string pkg,
311 Version const &fallback, CacheSetHelper &helper,
312 bool const onlyFromName) {
313 std::string ver;
314 bool verIsRel = false;
315 size_t const vertag = pkg.find_last_of("/=");
316 if (vertag != std::string::npos) {
317 ver = pkg.substr(vertag+1);
318 verIsRel = (pkg[vertag] == '/');
319 pkg.erase(vertag);
320 }
321 PackageSet pkgset;
322 if (onlyFromName == false)
323 PackageContainerInterface::FromString(&pkgset, Cache, pkg, helper);
324 else {
325 pkgset.insert(PackageContainerInterface::FromName(Cache, pkg, helper));
326 }
327
328 bool errors = true;
329 if (pkgset.getConstructor() != PackageSet::UNKNOWN)
330 errors = helper.showErrors(false);
331
332 bool found = false;
333 for (PackageSet::const_iterator P = pkgset.begin();
334 P != pkgset.end(); ++P) {
335 if (vertag == std::string::npos) {
336 found |= VersionContainerInterface::FromPackage(vci, Cache, P, fallback, helper);
337 continue;
338 }
339 pkgCache::VerIterator V;
340 if (ver == "installed")
341 V = getInstalledVer(Cache, P, helper);
342 else if (ver == "candidate")
343 V = getCandidateVer(Cache, P, helper);
344 else if (ver == "newest") {
345 if (P->VersionList != 0)
346 V = P.VersionList();
347 else
348 V = helper.canNotFindNewestVer(Cache, P);
349 } else {
350 pkgVersionMatch Match(ver, (verIsRel == true ? pkgVersionMatch::Release :
351 pkgVersionMatch::Version));
352 V = Match.Find(P);
353 if (V.end() == true) {
354 if (verIsRel == true)
355 _error->Error(_("Release '%s' for '%s' was not found"),
356 ver.c_str(), P.FullName(true).c_str());
357 else
358 _error->Error(_("Version '%s' for '%s' was not found"),
359 ver.c_str(), P.FullName(true).c_str());
360 continue;
361 }
362 }
363 if (V.end() == true)
364 continue;
365 helper.showSelectedVersion(P, V, ver, verIsRel);
366 vci->insert(V);
367 found = true;
368 }
369 if (pkgset.getConstructor() != PackageSet::UNKNOWN)
370 helper.showErrors(errors);
371 return found;
372 }
373 /*}}}*/
374 // FromPackage - versions from package based on fallback /*{{{*/
375 bool VersionContainerInterface::FromPackage(VersionContainerInterface * const vci,
376 pkgCacheFile &Cache,
377 pkgCache::PkgIterator const &P,
378 Version const &fallback,
379 CacheSetHelper &helper) {
380 pkgCache::VerIterator V;
381 bool showErrors;
382 bool found = false;
383 switch(fallback) {
384 case ALL:
385 if (P->VersionList != 0)
386 for (V = P.VersionList(); V.end() != true; ++V)
387 found |= vci->insert(V);
388 else
389 helper.canNotFindAllVer(vci, Cache, P);
390 break;
391 case CANDANDINST:
392 found |= vci->insert(getInstalledVer(Cache, P, helper));
393 found |= vci->insert(getCandidateVer(Cache, P, helper));
394 break;
395 case CANDIDATE:
396 found |= vci->insert(getCandidateVer(Cache, P, helper));
397 break;
398 case INSTALLED:
399 found |= vci->insert(getInstalledVer(Cache, P, helper));
400 break;
401 case CANDINST:
402 showErrors = helper.showErrors(false);
403 V = getCandidateVer(Cache, P, helper);
404 if (V.end() == true)
405 V = getInstalledVer(Cache, P, helper);
406 helper.showErrors(showErrors);
407 if (V.end() == false)
408 found |= vci->insert(V);
409 else
410 helper.canNotFindInstCandVer(vci, Cache, P);
411 break;
412 case INSTCAND:
413 showErrors = helper.showErrors(false);
414 V = getInstalledVer(Cache, P, helper);
415 if (V.end() == true)
416 V = getCandidateVer(Cache, P, helper);
417 helper.showErrors(showErrors);
418 if (V.end() == false)
419 found |= vci->insert(V);
420 else
421 helper.canNotFindInstCandVer(vci, Cache, P);
422 break;
423 case NEWEST:
424 if (P->VersionList != 0)
425 found |= vci->insert(P.VersionList());
426 else
427 helper.canNotFindNewestVer(Cache, P);
428 break;
429 }
430 return found;
431 }
432 /*}}}*/
433 // getCandidateVer - Returns the candidate version of the given package /*{{{*/
434 pkgCache::VerIterator VersionContainerInterface::getCandidateVer(pkgCacheFile &Cache,
435 pkgCache::PkgIterator const &Pkg, CacheSetHelper &helper) {
436 pkgCache::VerIterator Cand;
437 if (Cache.IsPolicyBuilt() == true || Cache.IsDepCacheBuilt() == false) {
438 if (unlikely(Cache.GetPolicy() == 0))
439 return pkgCache::VerIterator(Cache);
440 Cand = Cache.GetPolicy()->GetCandidateVer(Pkg);
441 } else {
442 Cand = Cache[Pkg].CandidateVerIter(Cache);
443 }
444 if (Cand.end() == true)
445 return helper.canNotFindCandidateVer(Cache, Pkg);
446 return Cand;
447 }
448 /*}}}*/
449 // getInstalledVer - Returns the installed version of the given package /*{{{*/
450 pkgCache::VerIterator VersionContainerInterface::getInstalledVer(pkgCacheFile &Cache,
451 pkgCache::PkgIterator const &Pkg, CacheSetHelper &helper) {
452 if (Pkg->CurrentVer == 0)
453 return helper.canNotFindInstalledVer(Cache, Pkg);
454 return Pkg.CurrentVer();
455 }
456 /*}}}*/
457
458 // canNotFindPkgName - handle the case no package has this name /*{{{*/
459 pkgCache::PkgIterator CacheSetHelper::canNotFindPkgName(pkgCacheFile &Cache,
460 std::string const &str) {
461 if (ShowError == true)
462 _error->Insert(ErrorType, _("Unable to locate package %s"), str.c_str());
463 return pkgCache::PkgIterator(Cache, 0);
464 }
465 /*}}}*/
466 // canNotFindTask - handle the case no package is found for a task /*{{{*/
467 void CacheSetHelper::canNotFindTask(PackageContainerInterface * const pci, pkgCacheFile &Cache, std::string pattern) {
468 if (ShowError == true)
469 _error->Insert(ErrorType, _("Couldn't find task '%s'"), pattern.c_str());
470 }
471 /*}}}*/
472 // canNotFindRegEx - handle the case no package is found by a regex /*{{{*/
473 void CacheSetHelper::canNotFindRegEx(PackageContainerInterface * const pci, pkgCacheFile &Cache, std::string pattern) {
474 if (ShowError == true)
475 _error->Insert(ErrorType, _("Couldn't find any package by regex '%s'"), pattern.c_str());
476 }
477 /*}}}*/
478 // canNotFindPackage - handle the case no package is found from a string/*{{{*/
479 void CacheSetHelper::canNotFindPackage(PackageContainerInterface * const pci, pkgCacheFile &Cache, std::string const &str) {
480 }
481 /*}}}*/
482 // canNotFindAllVer /*{{{*/
483 void CacheSetHelper::canNotFindAllVer(VersionContainerInterface * const vci, pkgCacheFile &Cache,
484 pkgCache::PkgIterator const &Pkg) {
485 if (ShowError == true)
486 _error->Insert(ErrorType, _("Can't select versions from package '%s' as it is purely virtual"), Pkg.FullName(true).c_str());
487 }
488 /*}}}*/
489 // canNotFindInstCandVer /*{{{*/
490 void CacheSetHelper::canNotFindInstCandVer(VersionContainerInterface * const vci, pkgCacheFile &Cache,
491 pkgCache::PkgIterator const &Pkg) {
492 if (ShowError == true)
493 _error->Insert(ErrorType, _("Can't select installed nor candidate version from package '%s' as it has neither of them"), Pkg.FullName(true).c_str());
494 }
495 /*}}}*/
496 // canNotFindInstCandVer /*{{{*/
497 void CacheSetHelper::canNotFindCandInstVer(VersionContainerInterface * const vci, pkgCacheFile &Cache,
498 pkgCache::PkgIterator const &Pkg) {
499 if (ShowError == true)
500 _error->Insert(ErrorType, _("Can't select installed nor candidate version from package '%s' as it has neither of them"), Pkg.FullName(true).c_str());
501 }
502 /*}}}*/
503 // canNotFindNewestVer /*{{{*/
504 pkgCache::VerIterator CacheSetHelper::canNotFindNewestVer(pkgCacheFile &Cache,
505 pkgCache::PkgIterator const &Pkg) {
506 if (ShowError == true)
507 _error->Insert(ErrorType, _("Can't select newest version from package '%s' as it is purely virtual"), Pkg.FullName(true).c_str());
508 return pkgCache::VerIterator(Cache, 0);
509 }
510 /*}}}*/
511 // canNotFindCandidateVer /*{{{*/
512 pkgCache::VerIterator CacheSetHelper::canNotFindCandidateVer(pkgCacheFile &Cache,
513 pkgCache::PkgIterator const &Pkg) {
514 if (ShowError == true)
515 _error->Insert(ErrorType, _("Can't select candidate version from package %s as it has no candidate"), Pkg.FullName(true).c_str());
516 return pkgCache::VerIterator(Cache, 0);
517 }
518 /*}}}*/
519 // canNotFindInstalledVer /*{{{*/
520 pkgCache::VerIterator CacheSetHelper::canNotFindInstalledVer(pkgCacheFile &Cache,
521 pkgCache::PkgIterator const &Pkg) {
522 if (ShowError == true)
523 _error->Insert(ErrorType, _("Can't select installed version from package %s as it is not installed"), Pkg.FullName(true).c_str());
524 return pkgCache::VerIterator(Cache, 0);
525 }
526 /*}}}*/
527 // showTaskSelection /*{{{*/
528 void CacheSetHelper::showTaskSelection(pkgCache::PkgIterator const &pkg,
529 std::string const &pattern) {
530 }
531 /*}}}*/
532 // showRegExSelection /*{{{*/
533 void CacheSetHelper::showRegExSelection(pkgCache::PkgIterator const &pkg,
534 std::string const &pattern) {
535 }
536 /*}}}*/
537 // showSelectedVersion /*{{{*/
538 void CacheSetHelper::showSelectedVersion(pkgCache::PkgIterator const &Pkg,
539 pkgCache::VerIterator const Ver,
540 std::string const &ver,
541 bool const verIsRel) {
542 }
543 /*}}}*/
544 }