1 | //===- ExternalASTMerger.cpp - Merging External AST Interface ---*- C++ -*-===// |
2 | // |
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
4 | // See https://llvm.org/LICENSE.txt for license information. |
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
6 | // |
7 | //===----------------------------------------------------------------------===// |
8 | // |
9 | // This file implements the ExternalASTMerger, which vends a combination of |
10 | // ASTs from several different ASTContext/FileManager pairs |
11 | // |
12 | //===----------------------------------------------------------------------===// |
13 | |
14 | #include "clang/AST/ASTContext.h" |
15 | #include "clang/AST/Decl.h" |
16 | #include "clang/AST/DeclCXX.h" |
17 | #include "clang/AST/DeclObjC.h" |
18 | #include "clang/AST/DeclTemplate.h" |
19 | #include "clang/AST/ExternalASTMerger.h" |
20 | |
21 | using namespace clang; |
22 | |
23 | namespace { |
24 | |
25 | template <typename T> struct Source { |
26 | T t; |
27 | Source(T t) : t(t) {} |
28 | operator T() { return t; } |
29 | template <typename U = T> U &get() { return t; } |
30 | template <typename U = T> const U &get() const { return t; } |
31 | template <typename U> operator Source<U>() { return Source<U>(t); } |
32 | }; |
33 | |
34 | typedef std::pair<Source<NamedDecl *>, ASTImporter *> Candidate; |
35 | |
36 | /// For the given DC, return the DC that is safe to perform lookups on. This is |
37 | /// the DC we actually want to work with most of the time. |
38 | const DeclContext *CanonicalizeDC(const DeclContext *DC) { |
39 | if (isa<LinkageSpecDecl>(Val: DC)) |
40 | return DC->getRedeclContext(); |
41 | return DC; |
42 | } |
43 | |
44 | Source<const DeclContext *> |
45 | LookupSameContext(Source<TranslationUnitDecl *> SourceTU, const DeclContext *DC, |
46 | ASTImporter &ReverseImporter) { |
47 | DC = CanonicalizeDC(DC); |
48 | if (DC->isTranslationUnit()) { |
49 | return SourceTU; |
50 | } |
51 | Source<const DeclContext *> SourceParentDC = |
52 | LookupSameContext(SourceTU, DC: DC->getParent(), ReverseImporter); |
53 | if (!SourceParentDC) { |
54 | // If we couldn't find the parent DC in this TranslationUnit, give up. |
55 | return nullptr; |
56 | } |
57 | auto *ND = cast<NamedDecl>(Val: DC); |
58 | DeclarationName Name = ND->getDeclName(); |
59 | auto SourceNameOrErr = ReverseImporter.Import(FromName: Name); |
60 | if (!SourceNameOrErr) { |
61 | llvm::consumeError(Err: SourceNameOrErr.takeError()); |
62 | return nullptr; |
63 | } |
64 | Source<DeclarationName> SourceName = *SourceNameOrErr; |
65 | DeclContext::lookup_result SearchResult = |
66 | SourceParentDC.get()->lookup(Name: SourceName.get()); |
67 | |
68 | // There are two cases here. First, we might not find the name. |
69 | // We might also find multiple copies, in which case we have no |
70 | // guarantee that the one we wanted is the one we pick. (E.g., |
71 | // if we have two specializations of the same template it is |
72 | // very hard to determine which is the one you want.) |
73 | // |
74 | // The Origins map fixes this problem by allowing the origin to be |
75 | // explicitly recorded, so we trigger that recording by returning |
76 | // nothing (rather than a possibly-inaccurate guess) here. |
77 | if (SearchResult.isSingleResult()) { |
78 | NamedDecl *SearchResultDecl = SearchResult.front(); |
79 | if (isa<DeclContext>(Val: SearchResultDecl) && |
80 | SearchResultDecl->getKind() == DC->getDeclKind()) |
81 | return cast<DeclContext>(Val: SearchResultDecl)->getPrimaryContext(); |
82 | return nullptr; // This type of lookup is unsupported |
83 | } else { |
84 | return nullptr; |
85 | } |
86 | } |
87 | |
88 | /// A custom implementation of ASTImporter, for ExternalASTMerger's purposes. |
89 | /// |
90 | /// There are several modifications: |
91 | /// |
92 | /// - It enables lazy lookup (via the HasExternalLexicalStorage flag and a few |
93 | /// others), which instructs Clang to refer to ExternalASTMerger. Also, it |
94 | /// forces MinimalImport to true, which is necessary to make this work. |
95 | /// - It maintains a reverse importer for use with names. This allows lookup of |
96 | /// arbitrary names in the source context. |
97 | /// - It updates the ExternalASTMerger's origin map as needed whenever a |
98 | /// it sees a DeclContext. |
99 | class LazyASTImporter : public ASTImporter { |
100 | private: |
101 | ExternalASTMerger &Parent; |
102 | ASTImporter Reverse; |
103 | const ExternalASTMerger::OriginMap &FromOrigins; |
104 | /// @see ExternalASTMerger::ImporterSource::Temporary |
105 | bool TemporarySource; |
106 | /// Map of imported declarations back to the declarations they originated |
107 | /// from. |
108 | llvm::DenseMap<Decl *, Decl *> ToOrigin; |
109 | /// @see ExternalASTMerger::ImporterSource::Merger |
110 | ExternalASTMerger *SourceMerger; |
111 | llvm::raw_ostream &logs() { return Parent.logs(); } |
112 | public: |
113 | LazyASTImporter(ExternalASTMerger &_Parent, ASTContext &ToContext, |
114 | FileManager &ToFileManager, |
115 | const ExternalASTMerger::ImporterSource &S, |
116 | std::shared_ptr<ASTImporterSharedState> SharedState) |
117 | : ASTImporter(ToContext, ToFileManager, S.getASTContext(), |
118 | S.getFileManager(), |
119 | /*MinimalImport=*/true, SharedState), |
120 | Parent(_Parent), |
121 | Reverse(S.getASTContext(), S.getFileManager(), ToContext, ToFileManager, |
122 | /*MinimalImport=*/true), |
123 | FromOrigins(S.getOriginMap()), TemporarySource(S.isTemporary()), |
124 | SourceMerger(S.getMerger()) {} |
125 | |
126 | llvm::Expected<Decl *> ImportImpl(Decl *FromD) override { |
127 | if (!TemporarySource || !SourceMerger) |
128 | return ASTImporter::ImportImpl(From: FromD); |
129 | |
130 | // If we get here, then this source is importing from a temporary ASTContext |
131 | // that also has another ExternalASTMerger attached. It could be |
132 | // possible that the current ExternalASTMerger and the temporary ASTContext |
133 | // share a common ImporterSource, which means that the temporary |
134 | // AST could contain declarations that were imported from a source |
135 | // that this ExternalASTMerger can access directly. Instead of importing |
136 | // such declarations from the temporary ASTContext, they should instead |
137 | // be directly imported by this ExternalASTMerger from the original |
138 | // source. This way the ExternalASTMerger can safely do a minimal import |
139 | // without creating incomplete declarations originated from a temporary |
140 | // ASTContext. If we would try to complete such declarations later on, we |
141 | // would fail to do so as their temporary AST could be deleted (which means |
142 | // that the missing parts of the minimally imported declaration in that |
143 | // ASTContext were also deleted). |
144 | // |
145 | // The following code tracks back any declaration that needs to be |
146 | // imported from the temporary ASTContext to a persistent ASTContext. |
147 | // Then the ExternalASTMerger tries to import from the persistent |
148 | // ASTContext directly by using the associated ASTImporter. If that |
149 | // succeeds, this ASTImporter just maps the declarations imported by |
150 | // the other (persistent) ASTImporter to this (temporary) ASTImporter. |
151 | // The steps can be visualized like this: |
152 | // |
153 | // Target AST <--- 3. Indirect import --- Persistent AST |
154 | // ^ of persistent decl ^ |
155 | // | | |
156 | // 1. Current import 2. Tracking back to persistent decl |
157 | // 4. Map persistent decl | |
158 | // & pretend we imported. | |
159 | // | | |
160 | // Temporary AST -------------------------------' |
161 | |
162 | // First, ask the ExternalASTMerger of the source where the temporary |
163 | // declaration originated from. |
164 | Decl *Persistent = SourceMerger->FindOriginalDecl(D: FromD); |
165 | // FromD isn't from a persistent AST, so just do a normal import. |
166 | if (!Persistent) |
167 | return ASTImporter::ImportImpl(From: FromD); |
168 | // Now ask the current ExternalASTMerger to try import the persistent |
169 | // declaration into the target. |
170 | ASTContext &PersistentCtx = Persistent->getASTContext(); |
171 | ASTImporter &OtherImporter = Parent.ImporterForOrigin(OriginContext&: PersistentCtx); |
172 | // Check that we never end up in the current Importer again. |
173 | assert((&PersistentCtx != &getFromContext()) && (&OtherImporter != this) && |
174 | "Delegated to same Importer?" ); |
175 | auto DeclOrErr = OtherImporter.Import(FromD: Persistent); |
176 | // Errors when importing the persistent decl are treated as if we |
177 | // had errors with importing the temporary decl. |
178 | if (!DeclOrErr) |
179 | return DeclOrErr.takeError(); |
180 | Decl *D = *DeclOrErr; |
181 | // Tell the current ASTImporter that this has already been imported |
182 | // to prevent any further queries for the temporary decl. |
183 | MapImported(From: FromD, To: D); |
184 | return D; |
185 | } |
186 | |
187 | /// Implements the ASTImporter interface for tracking back a declaration |
188 | /// to its original declaration it came from. |
189 | Decl *GetOriginalDecl(Decl *To) override { |
190 | return ToOrigin.lookup(Val: To); |
191 | } |
192 | |
193 | /// Whenever a DeclContext is imported, ensure that ExternalASTSource's origin |
194 | /// map is kept up to date. Also set the appropriate flags. |
195 | void Imported(Decl *From, Decl *To) override { |
196 | ToOrigin[To] = From; |
197 | |
198 | if (auto *ToDC = dyn_cast<DeclContext>(Val: To)) { |
199 | const bool LoggingEnabled = Parent.LoggingEnabled(); |
200 | if (LoggingEnabled) |
201 | logs() << "(ExternalASTMerger*)" << (void*)&Parent |
202 | << " imported (DeclContext*)" << (void*)ToDC |
203 | << ", (ASTContext*)" << (void*)&getToContext() |
204 | << " from (DeclContext*)" << (void*)llvm::cast<DeclContext>(Val: From) |
205 | << ", (ASTContext*)" << (void*)&getFromContext() |
206 | << "\n" ; |
207 | Source<DeclContext *> FromDC( |
208 | cast<DeclContext>(Val: From)->getPrimaryContext()); |
209 | if (FromOrigins.count(x: FromDC) && |
210 | Parent.HasImporterForOrigin(OriginContext&: *FromOrigins.at(k: FromDC).AST)) { |
211 | if (LoggingEnabled) |
212 | logs() << "(ExternalASTMerger*)" << (void*)&Parent |
213 | << " forced origin (DeclContext*)" |
214 | << (void*)FromOrigins.at(k: FromDC).DC |
215 | << ", (ASTContext*)" |
216 | << (void*)FromOrigins.at(k: FromDC).AST |
217 | << "\n" ; |
218 | Parent.ForceRecordOrigin(ToDC, Origin: FromOrigins.at(k: FromDC)); |
219 | } else { |
220 | if (LoggingEnabled) |
221 | logs() << "(ExternalASTMerger*)" << (void*)&Parent |
222 | << " maybe recording origin (DeclContext*)" << (void*)FromDC |
223 | << ", (ASTContext*)" << (void*)&getFromContext() |
224 | << "\n" ; |
225 | Parent.MaybeRecordOrigin(ToDC, Origin: {.DC: FromDC, .AST: &getFromContext()}); |
226 | } |
227 | } |
228 | if (auto *ToTag = dyn_cast<TagDecl>(Val: To)) { |
229 | ToTag->setHasExternalLexicalStorage(); |
230 | ToTag->getPrimaryContext()->setMustBuildLookupTable(); |
231 | assert(Parent.CanComplete(ToTag)); |
232 | } else if (auto *ToNamespace = dyn_cast<NamespaceDecl>(Val: To)) { |
233 | ToNamespace->setHasExternalVisibleStorage(); |
234 | assert(Parent.CanComplete(ToNamespace)); |
235 | } else if (auto *ToContainer = dyn_cast<ObjCContainerDecl>(Val: To)) { |
236 | ToContainer->setHasExternalLexicalStorage(); |
237 | ToContainer->getPrimaryContext()->setMustBuildLookupTable(); |
238 | assert(Parent.CanComplete(ToContainer)); |
239 | } |
240 | } |
241 | ASTImporter &GetReverse() { return Reverse; } |
242 | }; |
243 | |
244 | bool HasDeclOfSameType(llvm::ArrayRef<Candidate> Decls, const Candidate &C) { |
245 | if (isa<FunctionDecl>(Val: C.first.get())) |
246 | return false; |
247 | return llvm::any_of(Range&: Decls, P: [&](const Candidate &D) { |
248 | return C.first.get()->getKind() == D.first.get()->getKind(); |
249 | }); |
250 | } |
251 | |
252 | } // end namespace |
253 | |
254 | ASTImporter &ExternalASTMerger::ImporterForOrigin(ASTContext &OriginContext) { |
255 | for (const std::unique_ptr<ASTImporter> &I : Importers) |
256 | if (&I->getFromContext() == &OriginContext) |
257 | return *I; |
258 | llvm_unreachable("We should have an importer for this origin!" ); |
259 | } |
260 | |
261 | namespace { |
262 | LazyASTImporter &LazyImporterForOrigin(ExternalASTMerger &Merger, |
263 | ASTContext &OriginContext) { |
264 | return static_cast<LazyASTImporter &>( |
265 | Merger.ImporterForOrigin(OriginContext)); |
266 | } |
267 | } |
268 | |
269 | bool ExternalASTMerger::HasImporterForOrigin(ASTContext &OriginContext) { |
270 | for (const std::unique_ptr<ASTImporter> &I : Importers) |
271 | if (&I->getFromContext() == &OriginContext) |
272 | return true; |
273 | return false; |
274 | } |
275 | |
276 | template <typename CallbackType> |
277 | void ExternalASTMerger::ForEachMatchingDC(const DeclContext *DC, |
278 | CallbackType Callback) { |
279 | if (Origins.count(x: DC)) { |
280 | ExternalASTMerger::DCOrigin Origin = Origins[DC]; |
281 | LazyASTImporter &Importer = LazyImporterForOrigin(Merger&: *this, OriginContext&: *Origin.AST); |
282 | Callback(Importer, Importer.GetReverse(), Origin.DC); |
283 | } else { |
284 | bool DidCallback = false; |
285 | for (const std::unique_ptr<ASTImporter> &Importer : Importers) { |
286 | Source<TranslationUnitDecl *> SourceTU = |
287 | Importer->getFromContext().getTranslationUnitDecl(); |
288 | ASTImporter &Reverse = |
289 | static_cast<LazyASTImporter *>(Importer.get())->GetReverse(); |
290 | if (auto SourceDC = LookupSameContext(SourceTU, DC, ReverseImporter&: Reverse)) { |
291 | DidCallback = true; |
292 | if (Callback(*Importer, Reverse, SourceDC)) |
293 | break; |
294 | } |
295 | } |
296 | if (!DidCallback && LoggingEnabled()) |
297 | logs() << "(ExternalASTMerger*)" << (void*)this |
298 | << " asserting for (DeclContext*)" << (const void*)DC |
299 | << ", (ASTContext*)" << (void*)&Target.AST |
300 | << "\n" ; |
301 | assert(DidCallback && "Couldn't find a source context matching our DC" ); |
302 | } |
303 | } |
304 | |
305 | void ExternalASTMerger::CompleteType(TagDecl *Tag) { |
306 | assert(Tag->hasExternalLexicalStorage()); |
307 | ForEachMatchingDC(Tag, [&](ASTImporter &Forward, ASTImporter &Reverse, |
308 | Source<const DeclContext *> SourceDC) -> bool { |
309 | auto *SourceTag = const_cast<TagDecl *>(cast<TagDecl>(Val: SourceDC.get())); |
310 | if (SourceTag->hasExternalLexicalStorage()) |
311 | SourceTag->getASTContext().getExternalSource()->CompleteType(SourceTag); |
312 | if (!SourceTag->getDefinition()) |
313 | return false; |
314 | Forward.MapImported(SourceTag, Tag); |
315 | if (llvm::Error Err = Forward.ImportDefinition(SourceTag)) |
316 | llvm::consumeError(Err: std::move(Err)); |
317 | Tag->setCompleteDefinition(SourceTag->isCompleteDefinition()); |
318 | return true; |
319 | }); |
320 | } |
321 | |
322 | void ExternalASTMerger::CompleteType(ObjCInterfaceDecl *Interface) { |
323 | assert(Interface->hasExternalLexicalStorage()); |
324 | ForEachMatchingDC( |
325 | Interface, [&](ASTImporter &Forward, ASTImporter &Reverse, |
326 | Source<const DeclContext *> SourceDC) -> bool { |
327 | auto *SourceInterface = const_cast<ObjCInterfaceDecl *>( |
328 | cast<ObjCInterfaceDecl>(Val: SourceDC.get())); |
329 | if (SourceInterface->hasExternalLexicalStorage()) |
330 | SourceInterface->getASTContext().getExternalSource()->CompleteType( |
331 | SourceInterface); |
332 | if (!SourceInterface->getDefinition()) |
333 | return false; |
334 | Forward.MapImported(SourceInterface, Interface); |
335 | if (llvm::Error Err = Forward.ImportDefinition(SourceInterface)) |
336 | llvm::consumeError(Err: std::move(Err)); |
337 | return true; |
338 | }); |
339 | } |
340 | |
341 | bool ExternalASTMerger::CanComplete(DeclContext *Interface) { |
342 | assert(Interface->hasExternalLexicalStorage() || |
343 | Interface->hasExternalVisibleStorage()); |
344 | bool FoundMatchingDC = false; |
345 | ForEachMatchingDC(DC: Interface, |
346 | Callback: [&](ASTImporter &Forward, ASTImporter &Reverse, |
347 | Source<const DeclContext *> SourceDC) -> bool { |
348 | FoundMatchingDC = true; |
349 | return true; |
350 | }); |
351 | return FoundMatchingDC; |
352 | } |
353 | |
354 | namespace { |
355 | bool IsSameDC(const DeclContext *D1, const DeclContext *D2) { |
356 | if (isa<ObjCContainerDecl>(Val: D1) && isa<ObjCContainerDecl>(Val: D2)) |
357 | return true; // There are many cases where Objective-C is ambiguous. |
358 | if (auto *T1 = dyn_cast<TagDecl>(Val: D1)) |
359 | if (auto *T2 = dyn_cast<TagDecl>(Val: D2)) |
360 | if (T1->getFirstDecl() == T2->getFirstDecl()) |
361 | return true; |
362 | return D1 == D2 || D1 == CanonicalizeDC(DC: D2); |
363 | } |
364 | } |
365 | |
366 | void ExternalASTMerger::MaybeRecordOrigin(const DeclContext *ToDC, |
367 | DCOrigin Origin) { |
368 | LazyASTImporter &Importer = LazyImporterForOrigin(Merger&: *this, OriginContext&: *Origin.AST); |
369 | ASTImporter &Reverse = Importer.GetReverse(); |
370 | Source<const DeclContext *> FoundFromDC = |
371 | LookupSameContext(SourceTU: Origin.AST->getTranslationUnitDecl(), DC: ToDC, ReverseImporter&: Reverse); |
372 | const bool DoRecord = !FoundFromDC || !IsSameDC(D1: FoundFromDC.get(), D2: Origin.DC); |
373 | if (DoRecord) |
374 | RecordOriginImpl(ToDC, Origin, importer&: Importer); |
375 | if (LoggingEnabled()) |
376 | logs() << "(ExternalASTMerger*)" << (void*)this |
377 | << (DoRecord ? " decided " : " decided NOT" ) |
378 | << " to record origin (DeclContext*)" << (void*)Origin.DC |
379 | << ", (ASTContext*)" << (void*)&Origin.AST |
380 | << "\n" ; |
381 | } |
382 | |
383 | void ExternalASTMerger::ForceRecordOrigin(const DeclContext *ToDC, |
384 | DCOrigin Origin) { |
385 | RecordOriginImpl(ToDC, Origin, importer&: ImporterForOrigin(OriginContext&: *Origin.AST)); |
386 | } |
387 | |
388 | void ExternalASTMerger::RecordOriginImpl(const DeclContext *ToDC, DCOrigin Origin, |
389 | ASTImporter &Importer) { |
390 | Origins[ToDC] = Origin; |
391 | Importer.ASTImporter::MapImported(From: cast<Decl>(Val: Origin.DC), To: const_cast<Decl*>(cast<Decl>(Val: ToDC))); |
392 | } |
393 | |
394 | ExternalASTMerger::ExternalASTMerger(const ImporterTarget &Target, |
395 | llvm::ArrayRef<ImporterSource> Sources) : LogStream(&llvm::nulls()), Target(Target) { |
396 | SharedState = std::make_shared<ASTImporterSharedState>( |
397 | args&: *Target.AST.getTranslationUnitDecl()); |
398 | AddSources(Sources); |
399 | } |
400 | |
401 | Decl *ExternalASTMerger::FindOriginalDecl(Decl *D) { |
402 | assert(&D->getASTContext() == &Target.AST); |
403 | for (const auto &I : Importers) |
404 | if (auto Result = I->GetOriginalDecl(To: D)) |
405 | return Result; |
406 | return nullptr; |
407 | } |
408 | |
409 | void ExternalASTMerger::AddSources(llvm::ArrayRef<ImporterSource> Sources) { |
410 | for (const ImporterSource &S : Sources) { |
411 | assert(&S.getASTContext() != &Target.AST); |
412 | // Check that the associated merger actually imports into the source AST. |
413 | assert(!S.getMerger() || &S.getMerger()->Target.AST == &S.getASTContext()); |
414 | Importers.push_back(x: std::make_unique<LazyASTImporter>( |
415 | args&: *this, args&: Target.AST, args&: Target.FM, args: S, args&: SharedState)); |
416 | } |
417 | } |
418 | |
419 | void ExternalASTMerger::RemoveSources(llvm::ArrayRef<ImporterSource> Sources) { |
420 | if (LoggingEnabled()) |
421 | for (const ImporterSource &S : Sources) |
422 | logs() << "(ExternalASTMerger*)" << (void *)this |
423 | << " removing source (ASTContext*)" << (void *)&S.getASTContext() |
424 | << "\n" ; |
425 | llvm::erase_if(C&: Importers, |
426 | P: [&Sources](std::unique_ptr<ASTImporter> &Importer) -> bool { |
427 | for (const ImporterSource &S : Sources) { |
428 | if (&Importer->getFromContext() == &S.getASTContext()) |
429 | return true; |
430 | } |
431 | return false; |
432 | }); |
433 | for (OriginMap::iterator OI = Origins.begin(), OE = Origins.end(); OI != OE; ) { |
434 | std::pair<const DeclContext *, DCOrigin> Origin = *OI; |
435 | bool Erase = false; |
436 | for (const ImporterSource &S : Sources) { |
437 | if (&S.getASTContext() == Origin.second.AST) { |
438 | Erase = true; |
439 | break; |
440 | } |
441 | } |
442 | if (Erase) |
443 | OI = Origins.erase(position: OI); |
444 | else |
445 | ++OI; |
446 | } |
447 | } |
448 | |
449 | template <typename DeclTy> |
450 | static bool importSpecializations(DeclTy *D, ASTImporter *Importer) { |
451 | for (auto *Spec : D->specializations()) { |
452 | auto ImportedSpecOrError = Importer->Import(Spec); |
453 | if (!ImportedSpecOrError) { |
454 | llvm::consumeError(Err: ImportedSpecOrError.takeError()); |
455 | return true; |
456 | } |
457 | } |
458 | return false; |
459 | } |
460 | |
461 | /// Imports specializations from template declarations that can be specialized. |
462 | static bool importSpecializationsIfNeeded(Decl *D, ASTImporter *Importer) { |
463 | if (!isa<TemplateDecl>(Val: D)) |
464 | return false; |
465 | if (auto *FunctionTD = dyn_cast<FunctionTemplateDecl>(Val: D)) |
466 | return importSpecializations(D: FunctionTD, Importer); |
467 | else if (auto *ClassTD = dyn_cast<ClassTemplateDecl>(Val: D)) |
468 | return importSpecializations(D: ClassTD, Importer); |
469 | else if (auto *VarTD = dyn_cast<VarTemplateDecl>(Val: D)) |
470 | return importSpecializations(D: VarTD, Importer); |
471 | return false; |
472 | } |
473 | |
474 | bool ExternalASTMerger::FindExternalVisibleDeclsByName(const DeclContext *DC, |
475 | DeclarationName Name) { |
476 | llvm::SmallVector<NamedDecl *, 1> Decls; |
477 | llvm::SmallVector<Candidate, 4> Candidates; |
478 | |
479 | auto FilterFoundDecl = [&Candidates](const Candidate &C) { |
480 | if (!HasDeclOfSameType(Decls: Candidates, C)) |
481 | Candidates.push_back(Elt: C); |
482 | }; |
483 | |
484 | ForEachMatchingDC(DC, |
485 | Callback: [&](ASTImporter &Forward, ASTImporter &Reverse, |
486 | Source<const DeclContext *> SourceDC) -> bool { |
487 | auto FromNameOrErr = Reverse.Import(FromName: Name); |
488 | if (!FromNameOrErr) { |
489 | llvm::consumeError(Err: FromNameOrErr.takeError()); |
490 | return false; |
491 | } |
492 | DeclContextLookupResult Result = |
493 | SourceDC.get()->lookup(Name: *FromNameOrErr); |
494 | for (NamedDecl *FromD : Result) { |
495 | FilterFoundDecl(std::make_pair(x&: FromD, y: &Forward)); |
496 | } |
497 | return false; |
498 | }); |
499 | |
500 | if (Candidates.empty()) |
501 | return false; |
502 | |
503 | Decls.reserve(N: Candidates.size()); |
504 | for (const Candidate &C : Candidates) { |
505 | Decl *LookupRes = C.first.get(); |
506 | ASTImporter *Importer = C.second; |
507 | auto NDOrErr = Importer->Import(FromD: LookupRes); |
508 | NamedDecl *ND = cast<NamedDecl>(llvm::cantFail(std::move(NDOrErr))); |
509 | assert(ND); |
510 | // If we don't import specialization, they are not available via lookup |
511 | // because the lookup result is imported TemplateDecl and it does not |
512 | // reference its specializations until they are imported explicitly. |
513 | bool IsSpecImportFailed = |
514 | importSpecializationsIfNeeded(D: LookupRes, Importer); |
515 | assert(!IsSpecImportFailed); |
516 | (void)IsSpecImportFailed; |
517 | Decls.push_back(Elt: ND); |
518 | } |
519 | SetExternalVisibleDeclsForName(DC, Name, Decls); |
520 | return true; |
521 | } |
522 | |
523 | void ExternalASTMerger::FindExternalLexicalDecls( |
524 | const DeclContext *DC, llvm::function_ref<bool(Decl::Kind)> IsKindWeWant, |
525 | SmallVectorImpl<Decl *> &Result) { |
526 | ForEachMatchingDC(DC, Callback: [&](ASTImporter &Forward, ASTImporter &Reverse, |
527 | Source<const DeclContext *> SourceDC) -> bool { |
528 | for (const Decl *SourceDecl : SourceDC.get()->decls()) { |
529 | if (IsKindWeWant(SourceDecl->getKind())) { |
530 | auto ImportedDeclOrErr = Forward.Import(FromD: SourceDecl); |
531 | if (ImportedDeclOrErr) |
532 | assert(!(*ImportedDeclOrErr) || |
533 | IsSameDC((*ImportedDeclOrErr)->getDeclContext(), DC)); |
534 | else |
535 | llvm::consumeError(Err: ImportedDeclOrErr.takeError()); |
536 | } |
537 | } |
538 | return false; |
539 | }); |
540 | } |
541 | |