]>
Commit | Line | Data |
---|---|---|
ee0167c4 | 1 | // Includes /*{{{*/ |
453b82a3 DK |
2 | #include <config.h> |
3 | ||
b9179170 | 4 | #include <apt-pkg/cachefile.h> |
b9179170 | 5 | #include <apt-pkg/cacheset.h> |
b9179170 | 6 | #include <apt-pkg/cmndline.h> |
b9179170 | 7 | #include <apt-pkg/pkgrecords.h> |
b9179170 | 8 | #include <apt-pkg/policy.h> |
453b82a3 DK |
9 | #include <apt-pkg/progress.h> |
10 | #include <apt-pkg/cacheiterators.h> | |
11 | #include <apt-pkg/configuration.h> | |
12 | #include <apt-pkg/depcache.h> | |
13 | #include <apt-pkg/macros.h> | |
14 | #include <apt-pkg/pkgcache.h> | |
b9179170 | 15 | |
453b82a3 DK |
16 | #include <apt-private/private-cacheset.h> |
17 | #include <apt-private/private-output.h> | |
18 | #include <apt-private/private-search.h> | |
631800b1 | 19 | #include <apt-private/private-show.h> |
453b82a3 DK |
20 | |
21 | #include <string.h> | |
b9179170 | 22 | #include <iostream> |
453b82a3 | 23 | #include <sstream> |
b9179170 | 24 | #include <map> |
453b82a3 DK |
25 | #include <string> |
26 | #include <utility> | |
b9179170 | 27 | |
453b82a3 | 28 | #include <apti18n.h> |
ee0167c4 | 29 | /*}}}*/ |
b9179170 | 30 | |
631800b1 | 31 | static bool FullTextSearch(CommandLine &CmdL) /*{{{*/ |
b9179170 MV |
32 | { |
33 | pkgCacheFile CacheFile; | |
34 | pkgCache *Cache = CacheFile.GetPkgCache(); | |
35 | pkgDepCache::Policy *Plcy = CacheFile.GetPolicy(); | |
b9179170 MV |
36 | if (unlikely(Cache == NULL || Plcy == NULL)) |
37 | return false; | |
38 | ||
859093da DK |
39 | // Make sure there is at least one argument |
40 | unsigned int const NumPatterns = CmdL.FileSize() -1; | |
41 | if (NumPatterns < 1) | |
42 | return _error->Error(_("You must give at least one search pattern")); | |
43 | ||
44 | #define APT_FREE_PATTERNS() for (std::vector<regex_t>::iterator P = Patterns.begin(); \ | |
45 | P != Patterns.end(); ++P) { regfree(&(*P)); } | |
46 | ||
47 | // Compile the regex pattern | |
48 | std::vector<regex_t> Patterns; | |
49 | for (unsigned int I = 0; I != NumPatterns; ++I) | |
50 | { | |
51 | regex_t pattern; | |
52 | if (regcomp(&pattern, CmdL.FileList[I + 1], REG_EXTENDED | REG_ICASE | REG_NOSUB) != 0) | |
53 | { | |
54 | APT_FREE_PATTERNS(); | |
55 | return _error->Error("Regex compilation error"); | |
56 | } | |
57 | Patterns.push_back(pattern); | |
58 | } | |
59 | ||
b9179170 | 60 | std::map<std::string, std::string> output_map; |
b9179170 MV |
61 | |
62 | LocalitySortedVersionSet bag; | |
1bef0dd5 | 63 | OpTextProgress progress(*_config); |
b9179170 | 64 | progress.OverallProgress(0, 100, 50, _("Sorting")); |
25594bb5 | 65 | GetLocalitySortedVersionSet(CacheFile, &bag, &progress); |
b9179170 MV |
66 | LocalitySortedVersionSet::iterator V = bag.begin(); |
67 | ||
68 | progress.OverallProgress(50, 100, 50, _("Full Text Search")); | |
69 | progress.SubProgress(bag.size()); | |
206b6bb3 | 70 | pkgRecords records(CacheFile); |
1a68655d DK |
71 | |
72 | std::string format = "${color:highlight}${Package}${color:neutral}/${Origin} ${Version} ${Architecture}${ }${apt:Status}\n"; | |
73 | if (_config->FindB("APT::Cache::ShowFull",false) == false) | |
74 | format += " ${Description}\n"; | |
75 | else | |
76 | format += " ${LongDescription}\n"; | |
77 | ||
631800b1 | 78 | bool const NamesOnly = _config->FindB("APT::Cache::NamesOnly", false); |
b9179170 | 79 | int Done = 0; |
25594bb5 | 80 | std::vector<bool> PkgsDone(Cache->Head().PackageCount, false); |
9ce3cfc9 | 81 | for ( ;V != bag.end(); ++V) |
b9179170 MV |
82 | { |
83 | if (Done%500 == 0) | |
84 | progress.Progress(Done); | |
9ce3cfc9 | 85 | ++Done; |
859093da | 86 | |
206b6bb3 | 87 | // we want to list each package only once |
25594bb5 DK |
88 | pkgCache::PkgIterator const P = V.ParentPkg(); |
89 | if (PkgsDone[P->ID] == true) | |
206b6bb3 DK |
90 | continue; |
91 | ||
25594bb5 | 92 | char const * const PkgName = P.Name(); |
b9179170 | 93 | pkgCache::DescIterator Desc = V.TranslatedDescription(); |
ace5a062 | 94 | std::string LongDesc = ""; |
fe4a5378 JAK |
95 | if (Desc.end() == false) |
96 | { | |
ace5a062 JAK |
97 | pkgRecords::Parser &parser = records.Lookup(Desc.FileList()); |
98 | LongDesc = parser.LongDesc(); | |
99 | } | |
859093da | 100 | |
b9179170 | 101 | bool all_found = true; |
859093da DK |
102 | for (std::vector<regex_t>::const_iterator pattern = Patterns.begin(); |
103 | pattern != Patterns.end(); ++pattern) | |
b9179170 | 104 | { |
206b6bb3 | 105 | if (regexec(&(*pattern), PkgName, 0, 0, 0) == 0) |
859093da | 106 | continue; |
206b6bb3 | 107 | else if (NamesOnly == false && regexec(&(*pattern), LongDesc.c_str(), 0, 0, 0) == 0) |
859093da DK |
108 | continue; |
109 | // search patterns are AND, so one failing fails all | |
110 | all_found = false; | |
111 | break; | |
b9179170 | 112 | } |
859093da | 113 | if (all_found == true) |
b9179170 | 114 | { |
25594bb5 DK |
115 | PkgsDone[P->ID] = true; |
116 | std::stringstream outs; | |
117 | ListSingleVersion(CacheFile, records, V, outs, format); | |
118 | output_map.insert(std::make_pair<std::string, std::string>( | |
119 | PkgName, outs.str())); | |
b9179170 MV |
120 | } |
121 | } | |
859093da | 122 | APT_FREE_PATTERNS(); |
b9179170 MV |
123 | progress.Done(); |
124 | ||
125 | // FIXME: SORT! and make sorting flexible (alphabetic, by pkg status) | |
126 | // output the sorted map | |
206b6bb3 | 127 | std::map<std::string, std::string>::const_iterator K; |
9ce3cfc9 | 128 | for (K = output_map.begin(); K != output_map.end(); ++K) |
b9179170 MV |
129 | std::cout << (*K).second << std::endl; |
130 | ||
131 | return true; | |
132 | } | |
ee0167c4 | 133 | /*}}}*/ |
631800b1 DK |
134 | // LocalitySort - Sort a version list by package file locality /*{{{*/ |
135 | static int LocalityCompare(const void * const a, const void * const b) | |
136 | { | |
137 | pkgCache::VerFile const * const A = *(pkgCache::VerFile const * const * const)a; | |
138 | pkgCache::VerFile const * const B = *(pkgCache::VerFile const * const * const)b; | |
139 | ||
140 | if (A == 0 && B == 0) | |
141 | return 0; | |
142 | if (A == 0) | |
143 | return 1; | |
144 | if (B == 0) | |
145 | return -1; | |
146 | ||
147 | if (A->File == B->File) | |
148 | return A->Offset - B->Offset; | |
149 | return A->File - B->File; | |
150 | } | |
151 | void LocalitySort(pkgCache::VerFile ** const begin, unsigned long long const Count,size_t const Size) | |
152 | { | |
153 | qsort(begin,Count,Size,LocalityCompare); | |
154 | } | |
155 | static void LocalitySort(pkgCache::DescFile ** const begin, unsigned long long const Count,size_t const Size) | |
156 | { | |
157 | qsort(begin,Count,Size,LocalityCompare); | |
158 | } | |
159 | /*}}}*/ | |
160 | // Search - Perform a search /*{{{*/ | |
161 | // --------------------------------------------------------------------- | |
162 | /* This searches the package names and package descriptions for a pattern */ | |
163 | struct ExDescFile | |
164 | { | |
165 | pkgCache::DescFile *Df; | |
166 | pkgCache::VerIterator V; | |
167 | map_id_t ID; | |
168 | }; | |
169 | static bool Search(CommandLine &CmdL) | |
170 | { | |
171 | bool const ShowFull = _config->FindB("APT::Cache::ShowFull",false); | |
172 | unsigned int const NumPatterns = CmdL.FileSize() -1; | |
173 | ||
174 | pkgCacheFile CacheFile; | |
175 | pkgCache *Cache = CacheFile.GetPkgCache(); | |
176 | pkgDepCache::Policy *Plcy = CacheFile.GetPolicy(); | |
177 | if (unlikely(Cache == NULL || Plcy == NULL)) | |
178 | return false; | |
179 | ||
180 | // Make sure there is at least one argument | |
181 | if (NumPatterns < 1) | |
182 | return _error->Error(_("You must give at least one search pattern")); | |
183 | ||
184 | // Compile the regex pattern | |
185 | regex_t *Patterns = new regex_t[NumPatterns]; | |
186 | memset(Patterns,0,sizeof(*Patterns)*NumPatterns); | |
187 | for (unsigned I = 0; I != NumPatterns; I++) | |
188 | { | |
189 | if (regcomp(&Patterns[I],CmdL.FileList[I+1],REG_EXTENDED | REG_ICASE | | |
190 | REG_NOSUB) != 0) | |
191 | { | |
192 | for (; I != 0; I--) | |
193 | regfree(&Patterns[I]); | |
194 | return _error->Error("Regex compilation error"); | |
195 | } | |
196 | } | |
197 | ||
198 | if (_error->PendingError() == true) | |
199 | { | |
200 | for (unsigned I = 0; I != NumPatterns; I++) | |
201 | regfree(&Patterns[I]); | |
202 | return false; | |
203 | } | |
204 | ||
205 | size_t const descCount = Cache->HeaderP->GroupCount + 1; | |
206 | ExDescFile *DFList = new ExDescFile[descCount]; | |
207 | memset(DFList,0,sizeof(*DFList) * descCount); | |
208 | ||
209 | bool *PatternMatch = new bool[descCount * NumPatterns]; | |
210 | memset(PatternMatch,false,sizeof(*PatternMatch) * descCount * NumPatterns); | |
211 | ||
212 | // Map versions that we want to write out onto the VerList array. | |
213 | bool const NamesOnly = _config->FindB("APT::Cache::NamesOnly",false); | |
214 | for (pkgCache::GrpIterator G = Cache->GrpBegin(); G.end() == false; ++G) | |
215 | { | |
216 | size_t const PatternOffset = G->ID * NumPatterns; | |
217 | size_t unmatched = 0, matched = 0; | |
218 | for (unsigned I = 0; I < NumPatterns; ++I) | |
219 | { | |
220 | if (PatternMatch[PatternOffset + I] == true) | |
221 | ++matched; | |
222 | else if (regexec(&Patterns[I],G.Name(),0,0,0) == 0) | |
223 | PatternMatch[PatternOffset + I] = true; | |
224 | else | |
225 | ++unmatched; | |
226 | } | |
227 | ||
228 | // already dealt with this package? | |
229 | if (matched == NumPatterns) | |
230 | continue; | |
231 | ||
232 | // Doing names only, drop any that don't match.. | |
233 | if (NamesOnly == true && unmatched == NumPatterns) | |
234 | continue; | |
235 | ||
236 | // Find the proper version to use | |
237 | pkgCache::PkgIterator P = G.FindPreferredPkg(); | |
238 | if (P.end() == true) | |
239 | continue; | |
240 | pkgCache::VerIterator V = Plcy->GetCandidateVer(P); | |
241 | if (V.end() == false) | |
242 | { | |
243 | pkgCache::DescIterator const D = V.TranslatedDescription(); | |
244 | //FIXME: packages without a description can't be found | |
245 | if (D.end() == true) | |
246 | continue; | |
247 | DFList[G->ID].Df = D.FileList(); | |
248 | DFList[G->ID].V = V; | |
249 | DFList[G->ID].ID = G->ID; | |
250 | } | |
251 | ||
252 | if (unmatched == NumPatterns) | |
253 | continue; | |
254 | ||
255 | // Include all the packages that provide matching names too | |
256 | for (pkgCache::PrvIterator Prv = P.ProvidesList() ; Prv.end() == false; ++Prv) | |
257 | { | |
258 | pkgCache::VerIterator V = Plcy->GetCandidateVer(Prv.OwnerPkg()); | |
259 | if (V.end() == true) | |
260 | continue; | |
261 | ||
262 | unsigned long id = Prv.OwnerPkg().Group()->ID; | |
263 | pkgCache::DescIterator const D = V.TranslatedDescription(); | |
264 | //FIXME: packages without a description can't be found | |
265 | if (D.end() == true) | |
266 | continue; | |
267 | DFList[id].Df = D.FileList(); | |
268 | DFList[id].V = V; | |
269 | DFList[id].ID = id; | |
270 | ||
271 | size_t const PrvPatternOffset = id * NumPatterns; | |
272 | for (unsigned I = 0; I < NumPatterns; ++I) | |
273 | PatternMatch[PrvPatternOffset + I] |= PatternMatch[PatternOffset + I]; | |
274 | } | |
275 | } | |
276 | ||
277 | LocalitySort(&DFList->Df, Cache->HeaderP->GroupCount, sizeof(*DFList)); | |
278 | ||
279 | // Create the text record parser | |
280 | pkgRecords Recs(*Cache); | |
281 | // Iterate over all the version records and check them | |
282 | for (ExDescFile *J = DFList; J->Df != 0; ++J) | |
283 | { | |
284 | pkgRecords::Parser &P = Recs.Lookup(pkgCache::DescFileIterator(*Cache,J->Df)); | |
285 | size_t const PatternOffset = J->ID * NumPatterns; | |
286 | ||
287 | if (NamesOnly == false) | |
288 | { | |
289 | std::string const LongDesc = P.LongDesc(); | |
290 | for (unsigned I = 0; I < NumPatterns; ++I) | |
291 | { | |
292 | if (PatternMatch[PatternOffset + I] == true) | |
293 | continue; | |
294 | else if (regexec(&Patterns[I],LongDesc.c_str(),0,0,0) == 0) | |
295 | PatternMatch[PatternOffset + I] = true; | |
296 | } | |
297 | } | |
298 | ||
299 | bool matchedAll = true; | |
300 | for (unsigned I = 0; I < NumPatterns; ++I) | |
301 | if (PatternMatch[PatternOffset + I] == false) | |
302 | { | |
303 | matchedAll = false; | |
304 | break; | |
305 | } | |
306 | ||
307 | if (matchedAll == true) | |
308 | { | |
309 | if (ShowFull == true) | |
310 | DisplayRecordV1(CacheFile, J->V, std::cout); | |
311 | else | |
312 | printf("%s - %s\n",P.Name().c_str(),P.ShortDesc().c_str()); | |
313 | } | |
314 | } | |
315 | ||
316 | delete [] DFList; | |
317 | delete [] PatternMatch; | |
318 | for (unsigned I = 0; I != NumPatterns; I++) | |
319 | regfree(&Patterns[I]); | |
320 | delete [] Patterns; | |
321 | if (ferror(stdout)) | |
322 | return _error->Error("Write to stdout failed"); | |
323 | return true; | |
324 | } | |
325 | /*}}}*/ | |
326 | bool DoSearch(CommandLine &CmdL) /*{{{*/ | |
327 | { | |
328 | int const ShowVersion = _config->FindI("APT::Cache::Search::Version", 1); | |
329 | if (ShowVersion <= 1) | |
330 | return Search(CmdL); | |
331 | return FullTextSearch(CmdL); | |
332 | } | |
333 |