xref: /openbsd-src/gnu/llvm/lldb/source/Plugins/ExpressionParser/Clang/ClangASTSource.cpp (revision f6aab3d83b51b91c24247ad2c2573574de475a82)
1dda28197Spatrick //===-- ClangASTSource.cpp ------------------------------------------------===//
2061da546Spatrick //
3061da546Spatrick // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4061da546Spatrick // See https://llvm.org/LICENSE.txt for license information.
5061da546Spatrick // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6061da546Spatrick //
7061da546Spatrick //===----------------------------------------------------------------------===//
8061da546Spatrick 
9061da546Spatrick #include "ClangASTSource.h"
10061da546Spatrick 
11061da546Spatrick #include "ClangDeclVendor.h"
12061da546Spatrick #include "ClangModulesDeclVendor.h"
13061da546Spatrick 
14061da546Spatrick #include "lldb/Core/Module.h"
15061da546Spatrick #include "lldb/Core/ModuleList.h"
16061da546Spatrick #include "lldb/Symbol/CompilerDeclContext.h"
17061da546Spatrick #include "lldb/Symbol/Function.h"
18061da546Spatrick #include "lldb/Symbol/SymbolFile.h"
19061da546Spatrick #include "lldb/Symbol/TaggedASTType.h"
20061da546Spatrick #include "lldb/Target/Target.h"
21*f6aab3d8Srobert #include "lldb/Utility/LLDBLog.h"
22061da546Spatrick #include "lldb/Utility/Log.h"
23061da546Spatrick #include "clang/AST/ASTContext.h"
24061da546Spatrick #include "clang/AST/RecordLayout.h"
25dda28197Spatrick #include "clang/Basic/SourceManager.h"
26061da546Spatrick 
27dda28197Spatrick #include "Plugins/ExpressionParser/Clang/ClangUtil.h"
28061da546Spatrick #include "Plugins/LanguageRuntime/ObjC/ObjCLanguageRuntime.h"
29dda28197Spatrick #include "Plugins/TypeSystem/Clang/TypeSystemClang.h"
30061da546Spatrick 
31061da546Spatrick #include <memory>
32061da546Spatrick #include <vector>
33061da546Spatrick 
34061da546Spatrick using namespace clang;
35061da546Spatrick using namespace lldb_private;
36061da546Spatrick 
37061da546Spatrick // Scoped class that will remove an active lexical decl from the set when it
38061da546Spatrick // goes out of scope.
39061da546Spatrick namespace {
40061da546Spatrick class ScopedLexicalDeclEraser {
41061da546Spatrick public:
ScopedLexicalDeclEraser(std::set<const clang::Decl * > & decls,const clang::Decl * decl)42061da546Spatrick   ScopedLexicalDeclEraser(std::set<const clang::Decl *> &decls,
43061da546Spatrick                           const clang::Decl *decl)
44061da546Spatrick       : m_active_lexical_decls(decls), m_decl(decl) {}
45061da546Spatrick 
~ScopedLexicalDeclEraser()46061da546Spatrick   ~ScopedLexicalDeclEraser() { m_active_lexical_decls.erase(m_decl); }
47061da546Spatrick 
48061da546Spatrick private:
49061da546Spatrick   std::set<const clang::Decl *> &m_active_lexical_decls;
50061da546Spatrick   const clang::Decl *m_decl;
51061da546Spatrick };
52061da546Spatrick }
53061da546Spatrick 
ClangASTSource(const lldb::TargetSP & target,const std::shared_ptr<ClangASTImporter> & importer)54dda28197Spatrick ClangASTSource::ClangASTSource(
55dda28197Spatrick     const lldb::TargetSP &target,
56dda28197Spatrick     const std::shared_ptr<ClangASTImporter> &importer)
57dda28197Spatrick     : m_lookups_enabled(false), m_target(target), m_ast_context(nullptr),
58dda28197Spatrick       m_ast_importer_sp(importer), m_active_lexical_decls(),
59dda28197Spatrick       m_active_lookups() {
60dda28197Spatrick   assert(m_ast_importer_sp && "No ClangASTImporter passed to ClangASTSource?");
61061da546Spatrick }
62061da546Spatrick 
InstallASTContext(TypeSystemClang & clang_ast_context)63dda28197Spatrick void ClangASTSource::InstallASTContext(TypeSystemClang &clang_ast_context) {
64061da546Spatrick   m_ast_context = &clang_ast_context.getASTContext();
65061da546Spatrick   m_clang_ast_context = &clang_ast_context;
66061da546Spatrick   m_file_manager = &m_ast_context->getSourceManager().getFileManager();
67061da546Spatrick   m_ast_importer_sp->InstallMapCompleter(m_ast_context, *this);
68061da546Spatrick }
69061da546Spatrick 
~ClangASTSource()70061da546Spatrick ClangASTSource::~ClangASTSource() {
71061da546Spatrick   m_ast_importer_sp->ForgetDestination(m_ast_context);
72061da546Spatrick 
73061da546Spatrick   if (!m_target)
74061da546Spatrick     return;
75061da546Spatrick 
76be691f3bSpatrick   // Unregister the current ASTContext as a source for all scratch
77be691f3bSpatrick   // ASTContexts in the ClangASTImporter. Without this the scratch AST might
78be691f3bSpatrick   // query the deleted ASTContext for additional type information.
79be691f3bSpatrick   // We unregister from *all* scratch ASTContexts in case a type got exported
80be691f3bSpatrick   // to a scratch AST that isn't the best fitting scratch ASTContext.
81*f6aab3d8Srobert   lldb::TypeSystemClangSP scratch_ts_sp = ScratchTypeSystemClang::GetForTarget(
82be691f3bSpatrick       *m_target, ScratchTypeSystemClang::DefaultAST, false);
83be691f3bSpatrick 
84*f6aab3d8Srobert   if (!scratch_ts_sp)
85061da546Spatrick     return;
86061da546Spatrick 
87be691f3bSpatrick   ScratchTypeSystemClang *default_scratch_ast =
88*f6aab3d8Srobert       llvm::cast<ScratchTypeSystemClang>(scratch_ts_sp.get());
89be691f3bSpatrick   // Unregister from the default scratch AST (and all sub-ASTs).
90be691f3bSpatrick   default_scratch_ast->ForgetSource(m_ast_context, *m_ast_importer_sp);
91061da546Spatrick }
92061da546Spatrick 
StartTranslationUnit(ASTConsumer * Consumer)93061da546Spatrick void ClangASTSource::StartTranslationUnit(ASTConsumer *Consumer) {
94061da546Spatrick   if (!m_ast_context)
95061da546Spatrick     return;
96061da546Spatrick 
97061da546Spatrick   m_ast_context->getTranslationUnitDecl()->setHasExternalVisibleStorage();
98061da546Spatrick   m_ast_context->getTranslationUnitDecl()->setHasExternalLexicalStorage();
99061da546Spatrick }
100061da546Spatrick 
101061da546Spatrick // The core lookup interface.
FindExternalVisibleDeclsByName(const DeclContext * decl_ctx,DeclarationName clang_decl_name)102061da546Spatrick bool ClangASTSource::FindExternalVisibleDeclsByName(
103061da546Spatrick     const DeclContext *decl_ctx, DeclarationName clang_decl_name) {
104061da546Spatrick   if (!m_ast_context) {
105061da546Spatrick     SetNoExternalVisibleDeclsForName(decl_ctx, clang_decl_name);
106061da546Spatrick     return false;
107061da546Spatrick   }
108061da546Spatrick 
109061da546Spatrick   std::string decl_name(clang_decl_name.getAsString());
110061da546Spatrick 
111061da546Spatrick   switch (clang_decl_name.getNameKind()) {
112061da546Spatrick   // Normal identifiers.
113061da546Spatrick   case DeclarationName::Identifier: {
114061da546Spatrick     clang::IdentifierInfo *identifier_info =
115061da546Spatrick         clang_decl_name.getAsIdentifierInfo();
116061da546Spatrick 
117061da546Spatrick     if (!identifier_info || identifier_info->getBuiltinID() != 0) {
118061da546Spatrick       SetNoExternalVisibleDeclsForName(decl_ctx, clang_decl_name);
119061da546Spatrick       return false;
120061da546Spatrick     }
121061da546Spatrick   } break;
122061da546Spatrick 
123061da546Spatrick   // Operator names.
124061da546Spatrick   case DeclarationName::CXXOperatorName:
125061da546Spatrick   case DeclarationName::CXXLiteralOperatorName:
126061da546Spatrick     break;
127061da546Spatrick 
128061da546Spatrick   // Using directives found in this context.
129061da546Spatrick   // Tell Sema we didn't find any or we'll end up getting asked a *lot*.
130061da546Spatrick   case DeclarationName::CXXUsingDirective:
131061da546Spatrick     SetNoExternalVisibleDeclsForName(decl_ctx, clang_decl_name);
132061da546Spatrick     return false;
133061da546Spatrick 
134061da546Spatrick   case DeclarationName::ObjCZeroArgSelector:
135061da546Spatrick   case DeclarationName::ObjCOneArgSelector:
136061da546Spatrick   case DeclarationName::ObjCMultiArgSelector: {
137061da546Spatrick     llvm::SmallVector<NamedDecl *, 1> method_decls;
138061da546Spatrick 
139dda28197Spatrick     NameSearchContext method_search_context(*m_clang_ast_context, method_decls,
140061da546Spatrick                                             clang_decl_name, decl_ctx);
141061da546Spatrick 
142061da546Spatrick     FindObjCMethodDecls(method_search_context);
143061da546Spatrick 
144061da546Spatrick     SetExternalVisibleDeclsForName(decl_ctx, clang_decl_name, method_decls);
145061da546Spatrick     return (method_decls.size() > 0);
146061da546Spatrick   }
147061da546Spatrick   // These aren't possible in the global context.
148061da546Spatrick   case DeclarationName::CXXConstructorName:
149061da546Spatrick   case DeclarationName::CXXDestructorName:
150061da546Spatrick   case DeclarationName::CXXConversionFunctionName:
151061da546Spatrick   case DeclarationName::CXXDeductionGuideName:
152061da546Spatrick     SetNoExternalVisibleDeclsForName(decl_ctx, clang_decl_name);
153061da546Spatrick     return false;
154061da546Spatrick   }
155061da546Spatrick 
156061da546Spatrick   if (!GetLookupsEnabled()) {
157061da546Spatrick     // Wait until we see a '$' at the start of a name before we start doing any
158061da546Spatrick     // lookups so we can avoid lookup up all of the builtin types.
159061da546Spatrick     if (!decl_name.empty() && decl_name[0] == '$') {
160061da546Spatrick       SetLookupsEnabled(true);
161061da546Spatrick     } else {
162061da546Spatrick       SetNoExternalVisibleDeclsForName(decl_ctx, clang_decl_name);
163061da546Spatrick       return false;
164061da546Spatrick     }
165061da546Spatrick   }
166061da546Spatrick 
167061da546Spatrick   ConstString const_decl_name(decl_name.c_str());
168061da546Spatrick 
169061da546Spatrick   const char *uniqued_const_decl_name = const_decl_name.GetCString();
170061da546Spatrick   if (m_active_lookups.find(uniqued_const_decl_name) !=
171061da546Spatrick       m_active_lookups.end()) {
172061da546Spatrick     // We are currently looking up this name...
173061da546Spatrick     SetNoExternalVisibleDeclsForName(decl_ctx, clang_decl_name);
174061da546Spatrick     return false;
175061da546Spatrick   }
176061da546Spatrick   m_active_lookups.insert(uniqued_const_decl_name);
177061da546Spatrick   llvm::SmallVector<NamedDecl *, 4> name_decls;
178dda28197Spatrick   NameSearchContext name_search_context(*m_clang_ast_context, name_decls,
179dda28197Spatrick                                         clang_decl_name, decl_ctx);
180061da546Spatrick   FindExternalVisibleDecls(name_search_context);
181061da546Spatrick   SetExternalVisibleDeclsForName(decl_ctx, clang_decl_name, name_decls);
182061da546Spatrick   m_active_lookups.erase(uniqued_const_decl_name);
183061da546Spatrick   return (name_decls.size() != 0);
184061da546Spatrick }
185061da546Spatrick 
FindCompleteType(const TagDecl * decl)186dda28197Spatrick TagDecl *ClangASTSource::FindCompleteType(const TagDecl *decl) {
187*f6aab3d8Srobert   Log *log = GetLog(LLDBLog::Expressions);
188061da546Spatrick 
189061da546Spatrick   if (const NamespaceDecl *namespace_context =
190dda28197Spatrick           dyn_cast<NamespaceDecl>(decl->getDeclContext())) {
191061da546Spatrick     ClangASTImporter::NamespaceMapSP namespace_map =
192061da546Spatrick         m_ast_importer_sp->GetNamespaceMap(namespace_context);
193061da546Spatrick 
194061da546Spatrick     if (!namespace_map)
195dda28197Spatrick       return nullptr;
196061da546Spatrick 
197*f6aab3d8Srobert     LLDB_LOGV(log, "      CTD Inspecting namespace map{0} ({1} entries)",
198*f6aab3d8Srobert               namespace_map.get(), namespace_map->size());
199*f6aab3d8Srobert 
200dda28197Spatrick     for (const ClangASTImporter::NamespaceMapItem &item : *namespace_map) {
201dda28197Spatrick       LLDB_LOG(log, "      CTD Searching namespace {0} in module {1}",
202dda28197Spatrick                item.second.GetName(), item.first->GetFileSpec().GetFilename());
203061da546Spatrick 
204061da546Spatrick       TypeList types;
205061da546Spatrick 
206dda28197Spatrick       ConstString name(decl->getName());
207061da546Spatrick 
208dda28197Spatrick       item.first->FindTypesInNamespace(name, item.second, UINT32_MAX, types);
209061da546Spatrick 
210dda28197Spatrick       for (uint32_t ti = 0, te = types.GetSize(); ti != te; ++ti) {
211061da546Spatrick         lldb::TypeSP type = types.GetTypeAtIndex(ti);
212061da546Spatrick 
213061da546Spatrick         if (!type)
214061da546Spatrick           continue;
215061da546Spatrick 
216061da546Spatrick         CompilerType clang_type(type->GetFullCompilerType());
217061da546Spatrick 
218061da546Spatrick         if (!ClangUtil::IsClangType(clang_type))
219061da546Spatrick           continue;
220061da546Spatrick 
221061da546Spatrick         const TagType *tag_type =
222061da546Spatrick             ClangUtil::GetQualType(clang_type)->getAs<TagType>();
223061da546Spatrick 
224061da546Spatrick         if (!tag_type)
225061da546Spatrick           continue;
226061da546Spatrick 
227061da546Spatrick         TagDecl *candidate_tag_decl =
228061da546Spatrick             const_cast<TagDecl *>(tag_type->getDecl());
229061da546Spatrick 
230dda28197Spatrick         if (TypeSystemClang::GetCompleteDecl(
231dda28197Spatrick                 &candidate_tag_decl->getASTContext(), candidate_tag_decl))
232dda28197Spatrick           return candidate_tag_decl;
233061da546Spatrick       }
234061da546Spatrick     }
235061da546Spatrick   } else {
236061da546Spatrick     TypeList types;
237061da546Spatrick 
238dda28197Spatrick     ConstString name(decl->getName());
239061da546Spatrick 
240061da546Spatrick     const ModuleList &module_list = m_target->GetImages();
241061da546Spatrick 
242061da546Spatrick     bool exact_match = false;
243061da546Spatrick     llvm::DenseSet<SymbolFile *> searched_symbol_files;
244061da546Spatrick     module_list.FindTypes(nullptr, name, exact_match, UINT32_MAX,
245061da546Spatrick                           searched_symbol_files, types);
246061da546Spatrick 
247dda28197Spatrick     for (uint32_t ti = 0, te = types.GetSize(); ti != te; ++ti) {
248061da546Spatrick       lldb::TypeSP type = types.GetTypeAtIndex(ti);
249061da546Spatrick 
250061da546Spatrick       if (!type)
251061da546Spatrick         continue;
252061da546Spatrick 
253061da546Spatrick       CompilerType clang_type(type->GetFullCompilerType());
254061da546Spatrick 
255061da546Spatrick       if (!ClangUtil::IsClangType(clang_type))
256061da546Spatrick         continue;
257061da546Spatrick 
258061da546Spatrick       const TagType *tag_type =
259061da546Spatrick           ClangUtil::GetQualType(clang_type)->getAs<TagType>();
260061da546Spatrick 
261061da546Spatrick       if (!tag_type)
262061da546Spatrick         continue;
263061da546Spatrick 
264dda28197Spatrick       TagDecl *candidate_tag_decl = const_cast<TagDecl *>(tag_type->getDecl());
265061da546Spatrick 
266061da546Spatrick       // We have found a type by basename and we need to make sure the decl
267061da546Spatrick       // contexts are the same before we can try to complete this type with
268061da546Spatrick       // another
269dda28197Spatrick       if (!TypeSystemClang::DeclsAreEquivalent(const_cast<TagDecl *>(decl),
270dda28197Spatrick                                                candidate_tag_decl))
271061da546Spatrick         continue;
272061da546Spatrick 
273dda28197Spatrick       if (TypeSystemClang::GetCompleteDecl(&candidate_tag_decl->getASTContext(),
274061da546Spatrick                                            candidate_tag_decl))
275dda28197Spatrick         return candidate_tag_decl;
276061da546Spatrick     }
277061da546Spatrick   }
278dda28197Spatrick   return nullptr;
279dda28197Spatrick }
280dda28197Spatrick 
CompleteType(TagDecl * tag_decl)281dda28197Spatrick void ClangASTSource::CompleteType(TagDecl *tag_decl) {
282*f6aab3d8Srobert   Log *log = GetLog(LLDBLog::Expressions);
283dda28197Spatrick 
284dda28197Spatrick   if (log) {
285dda28197Spatrick     LLDB_LOG(log,
286dda28197Spatrick              "    CompleteTagDecl on (ASTContext*){0} Completing "
287dda28197Spatrick              "(TagDecl*){1} named {2}",
288dda28197Spatrick              m_clang_ast_context->getDisplayName(), tag_decl,
289dda28197Spatrick              tag_decl->getName());
290dda28197Spatrick 
291dda28197Spatrick     LLDB_LOG(log, "      CTD Before:\n{0}", ClangUtil::DumpDecl(tag_decl));
292dda28197Spatrick   }
293dda28197Spatrick 
294dda28197Spatrick   auto iter = m_active_lexical_decls.find(tag_decl);
295dda28197Spatrick   if (iter != m_active_lexical_decls.end())
296dda28197Spatrick     return;
297dda28197Spatrick   m_active_lexical_decls.insert(tag_decl);
298dda28197Spatrick   ScopedLexicalDeclEraser eraser(m_active_lexical_decls, tag_decl);
299dda28197Spatrick 
300dda28197Spatrick   if (!m_ast_importer_sp->CompleteTagDecl(tag_decl)) {
301dda28197Spatrick     // We couldn't complete the type.  Maybe there's a definition somewhere
302dda28197Spatrick     // else that can be completed.
303dda28197Spatrick     if (TagDecl *alternate = FindCompleteType(tag_decl))
304dda28197Spatrick       m_ast_importer_sp->CompleteTagDeclWithOrigin(tag_decl, alternate);
305061da546Spatrick   }
306061da546Spatrick 
307061da546Spatrick   LLDB_LOG(log, "      [CTD] After:\n{0}", ClangUtil::DumpDecl(tag_decl));
308061da546Spatrick }
309061da546Spatrick 
CompleteType(clang::ObjCInterfaceDecl * interface_decl)310061da546Spatrick void ClangASTSource::CompleteType(clang::ObjCInterfaceDecl *interface_decl) {
311*f6aab3d8Srobert   Log *log = GetLog(LLDBLog::Expressions);
312061da546Spatrick 
313dda28197Spatrick   LLDB_LOG(log,
314dda28197Spatrick            "    [CompleteObjCInterfaceDecl] on (ASTContext*){0} '{1}' "
315dda28197Spatrick            "Completing an ObjCInterfaceDecl named {1}",
316dda28197Spatrick            m_ast_context, m_clang_ast_context->getDisplayName(),
317dda28197Spatrick            interface_decl->getName());
318061da546Spatrick   LLDB_LOG(log, "      [COID] Before:\n{0}",
319061da546Spatrick            ClangUtil::DumpDecl(interface_decl));
320061da546Spatrick 
321061da546Spatrick   ClangASTImporter::DeclOrigin original = m_ast_importer_sp->GetDeclOrigin(interface_decl);
322061da546Spatrick 
323061da546Spatrick   if (original.Valid()) {
324061da546Spatrick     if (ObjCInterfaceDecl *original_iface_decl =
325061da546Spatrick             dyn_cast<ObjCInterfaceDecl>(original.decl)) {
326061da546Spatrick       ObjCInterfaceDecl *complete_iface_decl =
327061da546Spatrick           GetCompleteObjCInterface(original_iface_decl);
328061da546Spatrick 
329061da546Spatrick       if (complete_iface_decl && (complete_iface_decl != original_iface_decl)) {
330061da546Spatrick         m_ast_importer_sp->SetDeclOrigin(interface_decl, complete_iface_decl);
331061da546Spatrick       }
332061da546Spatrick     }
333061da546Spatrick   }
334061da546Spatrick 
335061da546Spatrick   m_ast_importer_sp->CompleteObjCInterfaceDecl(interface_decl);
336061da546Spatrick 
337061da546Spatrick   if (interface_decl->getSuperClass() &&
338061da546Spatrick       interface_decl->getSuperClass() != interface_decl)
339061da546Spatrick     CompleteType(interface_decl->getSuperClass());
340061da546Spatrick 
341dda28197Spatrick   LLDB_LOG(log, "      [COID] After:");
342061da546Spatrick   LLDB_LOG(log, "      [COID] {0}", ClangUtil::DumpDecl(interface_decl));
343061da546Spatrick }
344061da546Spatrick 
GetCompleteObjCInterface(const clang::ObjCInterfaceDecl * interface_decl)345061da546Spatrick clang::ObjCInterfaceDecl *ClangASTSource::GetCompleteObjCInterface(
346061da546Spatrick     const clang::ObjCInterfaceDecl *interface_decl) {
347061da546Spatrick   lldb::ProcessSP process(m_target->GetProcessSP());
348061da546Spatrick 
349061da546Spatrick   if (!process)
350061da546Spatrick     return nullptr;
351061da546Spatrick 
352061da546Spatrick   ObjCLanguageRuntime *language_runtime(ObjCLanguageRuntime::Get(*process));
353061da546Spatrick 
354061da546Spatrick   if (!language_runtime)
355061da546Spatrick     return nullptr;
356061da546Spatrick 
357061da546Spatrick   ConstString class_name(interface_decl->getNameAsString().c_str());
358061da546Spatrick 
359061da546Spatrick   lldb::TypeSP complete_type_sp(
360061da546Spatrick       language_runtime->LookupInCompleteClassCache(class_name));
361061da546Spatrick 
362061da546Spatrick   if (!complete_type_sp)
363061da546Spatrick     return nullptr;
364061da546Spatrick 
365061da546Spatrick   TypeFromUser complete_type =
366061da546Spatrick       TypeFromUser(complete_type_sp->GetFullCompilerType());
367061da546Spatrick   lldb::opaque_compiler_type_t complete_opaque_type =
368061da546Spatrick       complete_type.GetOpaqueQualType();
369061da546Spatrick 
370061da546Spatrick   if (!complete_opaque_type)
371061da546Spatrick     return nullptr;
372061da546Spatrick 
373061da546Spatrick   const clang::Type *complete_clang_type =
374061da546Spatrick       QualType::getFromOpaquePtr(complete_opaque_type).getTypePtr();
375061da546Spatrick   const ObjCInterfaceType *complete_interface_type =
376061da546Spatrick       dyn_cast<ObjCInterfaceType>(complete_clang_type);
377061da546Spatrick 
378061da546Spatrick   if (!complete_interface_type)
379061da546Spatrick     return nullptr;
380061da546Spatrick 
381061da546Spatrick   ObjCInterfaceDecl *complete_iface_decl(complete_interface_type->getDecl());
382061da546Spatrick 
383061da546Spatrick   return complete_iface_decl;
384061da546Spatrick }
385061da546Spatrick 
FindExternalLexicalDecls(const DeclContext * decl_context,llvm::function_ref<bool (Decl::Kind)> predicate,llvm::SmallVectorImpl<Decl * > & decls)386061da546Spatrick void ClangASTSource::FindExternalLexicalDecls(
387061da546Spatrick     const DeclContext *decl_context,
388061da546Spatrick     llvm::function_ref<bool(Decl::Kind)> predicate,
389061da546Spatrick     llvm::SmallVectorImpl<Decl *> &decls) {
390061da546Spatrick 
391*f6aab3d8Srobert   Log *log = GetLog(LLDBLog::Expressions);
392061da546Spatrick 
393061da546Spatrick   const Decl *context_decl = dyn_cast<Decl>(decl_context);
394061da546Spatrick 
395061da546Spatrick   if (!context_decl)
396061da546Spatrick     return;
397061da546Spatrick 
398061da546Spatrick   auto iter = m_active_lexical_decls.find(context_decl);
399061da546Spatrick   if (iter != m_active_lexical_decls.end())
400061da546Spatrick     return;
401061da546Spatrick   m_active_lexical_decls.insert(context_decl);
402061da546Spatrick   ScopedLexicalDeclEraser eraser(m_active_lexical_decls, context_decl);
403061da546Spatrick 
404061da546Spatrick   if (log) {
405061da546Spatrick     if (const NamedDecl *context_named_decl = dyn_cast<NamedDecl>(context_decl))
406dda28197Spatrick       LLDB_LOG(log,
407dda28197Spatrick                "FindExternalLexicalDecls on (ASTContext*){0} '{1}' in "
408*f6aab3d8Srobert                "'{2}' ({3}Decl*){4}",
409dda28197Spatrick                m_ast_context, m_clang_ast_context->getDisplayName(),
410061da546Spatrick                context_named_decl->getNameAsString().c_str(),
411061da546Spatrick                context_decl->getDeclKindName(),
412061da546Spatrick                static_cast<const void *>(context_decl));
413061da546Spatrick     else if (context_decl)
414dda28197Spatrick       LLDB_LOG(log,
415dda28197Spatrick                "FindExternalLexicalDecls on (ASTContext*){0} '{1}' in "
416dda28197Spatrick                "({2}Decl*){3}",
417dda28197Spatrick                m_ast_context, m_clang_ast_context->getDisplayName(),
418061da546Spatrick                context_decl->getDeclKindName(),
419061da546Spatrick                static_cast<const void *>(context_decl));
420061da546Spatrick     else
421dda28197Spatrick       LLDB_LOG(log,
422dda28197Spatrick                "FindExternalLexicalDecls on (ASTContext*){0} '{1}' in a "
423dda28197Spatrick                "NULL context",
424dda28197Spatrick                m_ast_context, m_clang_ast_context->getDisplayName());
425061da546Spatrick   }
426061da546Spatrick 
427061da546Spatrick   ClangASTImporter::DeclOrigin original = m_ast_importer_sp->GetDeclOrigin(context_decl);
428061da546Spatrick 
429061da546Spatrick   if (!original.Valid())
430061da546Spatrick     return;
431061da546Spatrick 
432dda28197Spatrick   LLDB_LOG(log, "  FELD Original decl {0} (Decl*){1:x}:\n{2}",
433dda28197Spatrick            static_cast<void *>(original.ctx),
434dda28197Spatrick            static_cast<void *>(original.decl),
435dda28197Spatrick            ClangUtil::DumpDecl(original.decl));
436061da546Spatrick 
437061da546Spatrick   if (ObjCInterfaceDecl *original_iface_decl =
438061da546Spatrick           dyn_cast<ObjCInterfaceDecl>(original.decl)) {
439061da546Spatrick     ObjCInterfaceDecl *complete_iface_decl =
440061da546Spatrick         GetCompleteObjCInterface(original_iface_decl);
441061da546Spatrick 
442061da546Spatrick     if (complete_iface_decl && (complete_iface_decl != original_iface_decl)) {
443061da546Spatrick       original.decl = complete_iface_decl;
444061da546Spatrick       original.ctx = &complete_iface_decl->getASTContext();
445061da546Spatrick 
446061da546Spatrick       m_ast_importer_sp->SetDeclOrigin(context_decl, complete_iface_decl);
447061da546Spatrick     }
448061da546Spatrick   }
449061da546Spatrick 
450061da546Spatrick   if (TagDecl *original_tag_decl = dyn_cast<TagDecl>(original.decl)) {
451061da546Spatrick     ExternalASTSource *external_source = original.ctx->getExternalSource();
452061da546Spatrick 
453061da546Spatrick     if (external_source)
454061da546Spatrick       external_source->CompleteType(original_tag_decl);
455061da546Spatrick   }
456061da546Spatrick 
457061da546Spatrick   const DeclContext *original_decl_context =
458061da546Spatrick       dyn_cast<DeclContext>(original.decl);
459061da546Spatrick 
460061da546Spatrick   if (!original_decl_context)
461061da546Spatrick     return;
462061da546Spatrick 
463061da546Spatrick   // Indicates whether we skipped any Decls of the original DeclContext.
464061da546Spatrick   bool SkippedDecls = false;
465dda28197Spatrick   for (Decl *decl : original_decl_context->decls()) {
466061da546Spatrick     // The predicate function returns true if the passed declaration kind is
467061da546Spatrick     // the one we are looking for.
468061da546Spatrick     // See clang::ExternalASTSource::FindExternalLexicalDecls()
469061da546Spatrick     if (predicate(decl->getKind())) {
470061da546Spatrick       if (log) {
471061da546Spatrick         std::string ast_dump = ClangUtil::DumpDecl(decl);
472061da546Spatrick         if (const NamedDecl *context_named_decl =
473061da546Spatrick                 dyn_cast<NamedDecl>(context_decl))
474dda28197Spatrick           LLDB_LOG(log, "  FELD Adding [to {0}Decl {1}] lexical {2}Decl {3}",
475dda28197Spatrick                    context_named_decl->getDeclKindName(),
476dda28197Spatrick                    context_named_decl->getName(), decl->getDeclKindName(),
477dda28197Spatrick                    ast_dump);
478061da546Spatrick         else
479dda28197Spatrick           LLDB_LOG(log, "  FELD Adding lexical {0}Decl {1}",
480dda28197Spatrick                    decl->getDeclKindName(), ast_dump);
481061da546Spatrick       }
482061da546Spatrick 
483*f6aab3d8Srobert       Decl *copied_decl = CopyDecl(decl);
484*f6aab3d8Srobert 
485*f6aab3d8Srobert       if (!copied_decl)
486*f6aab3d8Srobert         continue;
487061da546Spatrick 
488be691f3bSpatrick       // FIXME: We should add the copied decl to the 'decls' list. This would
489be691f3bSpatrick       // add the copied Decl into the DeclContext and make sure that we
490be691f3bSpatrick       // correctly propagate that we added some Decls back to Clang.
491be691f3bSpatrick       // By leaving 'decls' empty we incorrectly return false from
492be691f3bSpatrick       // DeclContext::LoadLexicalDeclsFromExternalStorage which might cause
493be691f3bSpatrick       // lookup issues later on.
494be691f3bSpatrick       // We can't just add them for now as the ASTImporter already added the
495be691f3bSpatrick       // decl into the DeclContext and this would add it twice.
496*f6aab3d8Srobert 
497*f6aab3d8Srobert       if (FieldDecl *copied_field = dyn_cast<FieldDecl>(copied_decl)) {
498*f6aab3d8Srobert         QualType copied_field_type = copied_field->getType();
499*f6aab3d8Srobert 
500*f6aab3d8Srobert         m_ast_importer_sp->RequireCompleteType(copied_field_type);
501*f6aab3d8Srobert       }
502061da546Spatrick     } else {
503061da546Spatrick       SkippedDecls = true;
504061da546Spatrick     }
505061da546Spatrick   }
506061da546Spatrick 
507061da546Spatrick   // CopyDecl may build a lookup table which may set up ExternalLexicalStorage
508061da546Spatrick   // to false.  However, since we skipped some of the external Decls we must
509061da546Spatrick   // set it back!
510061da546Spatrick   if (SkippedDecls) {
511061da546Spatrick     decl_context->setHasExternalLexicalStorage(true);
512061da546Spatrick     // This sets HasLazyExternalLexicalLookups to true.  By setting this bit we
513061da546Spatrick     // ensure that the lookup table is rebuilt, which means the external source
514061da546Spatrick     // is consulted again when a clang::DeclContext::lookup is called.
515061da546Spatrick     const_cast<DeclContext *>(decl_context)->setMustBuildLookupTable();
516061da546Spatrick   }
517061da546Spatrick }
518061da546Spatrick 
FindExternalVisibleDecls(NameSearchContext & context)519061da546Spatrick void ClangASTSource::FindExternalVisibleDecls(NameSearchContext &context) {
520061da546Spatrick   assert(m_ast_context);
521061da546Spatrick 
522061da546Spatrick   const ConstString name(context.m_decl_name.getAsString().c_str());
523061da546Spatrick 
524*f6aab3d8Srobert   Log *log = GetLog(LLDBLog::Expressions);
525061da546Spatrick 
526061da546Spatrick   if (log) {
527061da546Spatrick     if (!context.m_decl_context)
528dda28197Spatrick       LLDB_LOG(log,
529dda28197Spatrick                "ClangASTSource::FindExternalVisibleDecls on "
530dda28197Spatrick                "(ASTContext*){0} '{1}' for '{2}' in a NULL DeclContext",
531dda28197Spatrick                m_ast_context, m_clang_ast_context->getDisplayName(), name);
532061da546Spatrick     else if (const NamedDecl *context_named_decl =
533061da546Spatrick                  dyn_cast<NamedDecl>(context.m_decl_context))
534dda28197Spatrick       LLDB_LOG(log,
535dda28197Spatrick                "ClangASTSource::FindExternalVisibleDecls on "
536dda28197Spatrick                "(ASTContext*){0} '{1}' for '{2}' in '{3}'",
537dda28197Spatrick                m_ast_context, m_clang_ast_context->getDisplayName(), name,
538dda28197Spatrick                context_named_decl->getName());
539061da546Spatrick     else
540dda28197Spatrick       LLDB_LOG(log,
541dda28197Spatrick                "ClangASTSource::FindExternalVisibleDecls on "
542dda28197Spatrick                "(ASTContext*){0} '{1}' for '{2}' in a '{3}'",
543dda28197Spatrick                m_ast_context, m_clang_ast_context->getDisplayName(), name,
544dda28197Spatrick                context.m_decl_context->getDeclKindName());
545061da546Spatrick   }
546061da546Spatrick 
547dda28197Spatrick   if (isa<NamespaceDecl>(context.m_decl_context)) {
548dda28197Spatrick     LookupInNamespace(context);
549061da546Spatrick   } else if (isa<ObjCInterfaceDecl>(context.m_decl_context)) {
550061da546Spatrick     FindObjCPropertyAndIvarDecls(context);
551061da546Spatrick   } else if (!isa<TranslationUnitDecl>(context.m_decl_context)) {
552061da546Spatrick     // we shouldn't be getting FindExternalVisibleDecls calls for these
553061da546Spatrick     return;
554061da546Spatrick   } else {
555061da546Spatrick     CompilerDeclContext namespace_decl;
556061da546Spatrick 
557dda28197Spatrick     LLDB_LOG(log, "  CAS::FEVD Searching the root namespace");
558061da546Spatrick 
559dda28197Spatrick     FindExternalVisibleDecls(context, lldb::ModuleSP(), namespace_decl);
560061da546Spatrick   }
561061da546Spatrick 
562061da546Spatrick   if (!context.m_namespace_map->empty()) {
563061da546Spatrick     if (log && log->GetVerbose())
564dda28197Spatrick       LLDB_LOG(log, "  CAS::FEVD Registering namespace map {0} ({1} entries)",
565dda28197Spatrick                context.m_namespace_map.get(), context.m_namespace_map->size());
566061da546Spatrick 
567061da546Spatrick     NamespaceDecl *clang_namespace_decl =
568061da546Spatrick         AddNamespace(context, context.m_namespace_map);
569061da546Spatrick 
570061da546Spatrick     if (clang_namespace_decl)
571061da546Spatrick       clang_namespace_decl->setHasExternalVisibleStorage();
572061da546Spatrick   }
573061da546Spatrick }
574061da546Spatrick 
getSema()575061da546Spatrick clang::Sema *ClangASTSource::getSema() {
576061da546Spatrick   return m_clang_ast_context->getSema();
577061da546Spatrick }
578061da546Spatrick 
IgnoreName(const ConstString name,bool ignore_all_dollar_names)579061da546Spatrick bool ClangASTSource::IgnoreName(const ConstString name,
580061da546Spatrick                                 bool ignore_all_dollar_names) {
581061da546Spatrick   static const ConstString id_name("id");
582061da546Spatrick   static const ConstString Class_name("Class");
583061da546Spatrick 
584061da546Spatrick   if (m_ast_context->getLangOpts().ObjC)
585061da546Spatrick     if (name == id_name || name == Class_name)
586061da546Spatrick       return true;
587061da546Spatrick 
588061da546Spatrick   StringRef name_string_ref = name.GetStringRef();
589061da546Spatrick 
590061da546Spatrick   // The ClangASTSource is not responsible for finding $-names.
591061da546Spatrick   return name_string_ref.empty() ||
592061da546Spatrick          (ignore_all_dollar_names && name_string_ref.startswith("$")) ||
593061da546Spatrick          name_string_ref.startswith("_$");
594061da546Spatrick }
595061da546Spatrick 
FindExternalVisibleDecls(NameSearchContext & context,lldb::ModuleSP module_sp,CompilerDeclContext & namespace_decl)596061da546Spatrick void ClangASTSource::FindExternalVisibleDecls(
597061da546Spatrick     NameSearchContext &context, lldb::ModuleSP module_sp,
598dda28197Spatrick     CompilerDeclContext &namespace_decl) {
599061da546Spatrick   assert(m_ast_context);
600061da546Spatrick 
601*f6aab3d8Srobert   Log *log = GetLog(LLDBLog::Expressions);
602061da546Spatrick 
603061da546Spatrick   SymbolContextList sc_list;
604061da546Spatrick 
605061da546Spatrick   const ConstString name(context.m_decl_name.getAsString().c_str());
606061da546Spatrick   if (IgnoreName(name, true))
607061da546Spatrick     return;
608061da546Spatrick 
609061da546Spatrick   if (!m_target)
610061da546Spatrick     return;
611061da546Spatrick 
612dda28197Spatrick   FillNamespaceMap(context, module_sp, namespace_decl);
613dda28197Spatrick 
614dda28197Spatrick   if (context.m_found_type)
615dda28197Spatrick     return;
616dda28197Spatrick 
617dda28197Spatrick   TypeList types;
618dda28197Spatrick   const bool exact_match = true;
619dda28197Spatrick   llvm::DenseSet<lldb_private::SymbolFile *> searched_symbol_files;
620dda28197Spatrick   if (module_sp && namespace_decl)
621dda28197Spatrick     module_sp->FindTypesInNamespace(name, namespace_decl, 1, types);
622dda28197Spatrick   else {
623dda28197Spatrick     m_target->GetImages().FindTypes(module_sp.get(), name, exact_match, 1,
624dda28197Spatrick                                     searched_symbol_files, types);
625dda28197Spatrick   }
626dda28197Spatrick 
627dda28197Spatrick   if (size_t num_types = types.GetSize()) {
628dda28197Spatrick     for (size_t ti = 0; ti < num_types; ++ti) {
629dda28197Spatrick       lldb::TypeSP type_sp = types.GetTypeAtIndex(ti);
630dda28197Spatrick 
631dda28197Spatrick       if (log) {
632dda28197Spatrick         const char *name_string = type_sp->GetName().GetCString();
633dda28197Spatrick 
634dda28197Spatrick         LLDB_LOG(log, "  CAS::FEVD Matching type found for \"{0}\": {1}", name,
635dda28197Spatrick                  (name_string ? name_string : "<anonymous>"));
636dda28197Spatrick       }
637dda28197Spatrick 
638dda28197Spatrick       CompilerType full_type = type_sp->GetFullCompilerType();
639dda28197Spatrick 
640dda28197Spatrick       CompilerType copied_clang_type(GuardedCopyType(full_type));
641dda28197Spatrick 
642dda28197Spatrick       if (!copied_clang_type) {
643dda28197Spatrick         LLDB_LOG(log, "  CAS::FEVD - Couldn't export a type");
644dda28197Spatrick 
645dda28197Spatrick         continue;
646dda28197Spatrick       }
647dda28197Spatrick 
648dda28197Spatrick       context.AddTypeDecl(copied_clang_type);
649dda28197Spatrick 
650dda28197Spatrick       context.m_found_type = true;
651dda28197Spatrick       break;
652dda28197Spatrick     }
653dda28197Spatrick   }
654dda28197Spatrick 
655dda28197Spatrick   if (!context.m_found_type) {
656dda28197Spatrick     // Try the modules next.
657dda28197Spatrick     FindDeclInModules(context, name);
658dda28197Spatrick   }
659dda28197Spatrick 
660dda28197Spatrick   if (!context.m_found_type) {
661dda28197Spatrick     FindDeclInObjCRuntime(context, name);
662dda28197Spatrick   }
663dda28197Spatrick }
664dda28197Spatrick 
FillNamespaceMap(NameSearchContext & context,lldb::ModuleSP module_sp,const CompilerDeclContext & namespace_decl)665dda28197Spatrick void ClangASTSource::FillNamespaceMap(
666dda28197Spatrick     NameSearchContext &context, lldb::ModuleSP module_sp,
667dda28197Spatrick     const CompilerDeclContext &namespace_decl) {
668dda28197Spatrick   const ConstString name(context.m_decl_name.getAsString().c_str());
669dda28197Spatrick   if (IgnoreName(name, true))
670dda28197Spatrick     return;
671dda28197Spatrick 
672*f6aab3d8Srobert   Log *log = GetLog(LLDBLog::Expressions);
673dda28197Spatrick 
674061da546Spatrick   if (module_sp && namespace_decl) {
675061da546Spatrick     CompilerDeclContext found_namespace_decl;
676061da546Spatrick 
677061da546Spatrick     if (SymbolFile *symbol_file = module_sp->GetSymbolFile()) {
678dda28197Spatrick       found_namespace_decl = symbol_file->FindNamespace(name, namespace_decl);
679061da546Spatrick 
680061da546Spatrick       if (found_namespace_decl) {
681061da546Spatrick         context.m_namespace_map->push_back(
682061da546Spatrick             std::pair<lldb::ModuleSP, CompilerDeclContext>(
683061da546Spatrick                 module_sp, found_namespace_decl));
684061da546Spatrick 
685dda28197Spatrick         LLDB_LOG(log, "  CAS::FEVD Found namespace {0} in module {1}", name,
686dda28197Spatrick                  module_sp->GetFileSpec().GetFilename());
687061da546Spatrick       }
688061da546Spatrick     }
689dda28197Spatrick     return;
690dda28197Spatrick   }
691dda28197Spatrick 
692be691f3bSpatrick   for (lldb::ModuleSP image : m_target->GetImages().Modules()) {
693061da546Spatrick     if (!image)
694061da546Spatrick       continue;
695061da546Spatrick 
696061da546Spatrick     CompilerDeclContext found_namespace_decl;
697061da546Spatrick 
698061da546Spatrick     SymbolFile *symbol_file = image->GetSymbolFile();
699061da546Spatrick 
700061da546Spatrick     if (!symbol_file)
701061da546Spatrick       continue;
702061da546Spatrick 
703dda28197Spatrick     found_namespace_decl = symbol_file->FindNamespace(name, namespace_decl);
704061da546Spatrick 
705061da546Spatrick     if (found_namespace_decl) {
706061da546Spatrick       context.m_namespace_map->push_back(
707dda28197Spatrick           std::pair<lldb::ModuleSP, CompilerDeclContext>(image,
708dda28197Spatrick                                                          found_namespace_decl));
709061da546Spatrick 
710dda28197Spatrick       LLDB_LOG(log, "  CAS::FEVD Found namespace {0} in module {1}", name,
711dda28197Spatrick                image->GetFileSpec().GetFilename());
712061da546Spatrick     }
713061da546Spatrick   }
714061da546Spatrick }
715061da546Spatrick 
716061da546Spatrick template <class D> class TaggedASTDecl {
717061da546Spatrick public:
TaggedASTDecl()718061da546Spatrick   TaggedASTDecl() : decl(nullptr) {}
TaggedASTDecl(D * _decl)719061da546Spatrick   TaggedASTDecl(D *_decl) : decl(_decl) {}
IsValid() const720061da546Spatrick   bool IsValid() const { return (decl != nullptr); }
IsInvalid() const721061da546Spatrick   bool IsInvalid() const { return !IsValid(); }
operator ->() const722061da546Spatrick   D *operator->() const { return decl; }
723061da546Spatrick   D *decl;
724061da546Spatrick };
725061da546Spatrick 
726061da546Spatrick template <class D2, template <class D> class TD, class D1>
DynCast(TD<D1> source)727061da546Spatrick TD<D2> DynCast(TD<D1> source) {
728061da546Spatrick   return TD<D2>(dyn_cast<D2>(source.decl));
729061da546Spatrick }
730061da546Spatrick 
731061da546Spatrick template <class D = Decl> class DeclFromParser;
732061da546Spatrick template <class D = Decl> class DeclFromUser;
733061da546Spatrick 
734061da546Spatrick template <class D> class DeclFromParser : public TaggedASTDecl<D> {
735061da546Spatrick public:
DeclFromParser()736061da546Spatrick   DeclFromParser() : TaggedASTDecl<D>() {}
DeclFromParser(D * _decl)737061da546Spatrick   DeclFromParser(D *_decl) : TaggedASTDecl<D>(_decl) {}
738061da546Spatrick 
739061da546Spatrick   DeclFromUser<D> GetOrigin(ClangASTSource &source);
740061da546Spatrick };
741061da546Spatrick 
742061da546Spatrick template <class D> class DeclFromUser : public TaggedASTDecl<D> {
743061da546Spatrick public:
DeclFromUser()744061da546Spatrick   DeclFromUser() : TaggedASTDecl<D>() {}
DeclFromUser(D * _decl)745061da546Spatrick   DeclFromUser(D *_decl) : TaggedASTDecl<D>(_decl) {}
746061da546Spatrick 
747061da546Spatrick   DeclFromParser<D> Import(ClangASTSource &source);
748061da546Spatrick };
749061da546Spatrick 
750061da546Spatrick template <class D>
GetOrigin(ClangASTSource & source)751061da546Spatrick DeclFromUser<D> DeclFromParser<D>::GetOrigin(ClangASTSource &source) {
752061da546Spatrick   ClangASTImporter::DeclOrigin origin = source.GetDeclOrigin(this->decl);
753061da546Spatrick   if (!origin.Valid())
754061da546Spatrick     return DeclFromUser<D>();
755061da546Spatrick   return DeclFromUser<D>(dyn_cast<D>(origin.decl));
756061da546Spatrick }
757061da546Spatrick 
758061da546Spatrick template <class D>
Import(ClangASTSource & source)759061da546Spatrick DeclFromParser<D> DeclFromUser<D>::Import(ClangASTSource &source) {
760061da546Spatrick   DeclFromParser<> parser_generic_decl(source.CopyDecl(this->decl));
761061da546Spatrick   if (parser_generic_decl.IsInvalid())
762061da546Spatrick     return DeclFromParser<D>();
763061da546Spatrick   return DeclFromParser<D>(dyn_cast<D>(parser_generic_decl.decl));
764061da546Spatrick }
765061da546Spatrick 
FindObjCMethodDeclsWithOrigin(NameSearchContext & context,ObjCInterfaceDecl * original_interface_decl,const char * log_info)766061da546Spatrick bool ClangASTSource::FindObjCMethodDeclsWithOrigin(
767dda28197Spatrick     NameSearchContext &context, ObjCInterfaceDecl *original_interface_decl,
768dda28197Spatrick     const char *log_info) {
769061da546Spatrick   const DeclarationName &decl_name(context.m_decl_name);
770061da546Spatrick   clang::ASTContext *original_ctx = &original_interface_decl->getASTContext();
771061da546Spatrick 
772061da546Spatrick   Selector original_selector;
773061da546Spatrick 
774061da546Spatrick   if (decl_name.isObjCZeroArgSelector()) {
775061da546Spatrick     IdentifierInfo *ident = &original_ctx->Idents.get(decl_name.getAsString());
776061da546Spatrick     original_selector = original_ctx->Selectors.getSelector(0, &ident);
777061da546Spatrick   } else if (decl_name.isObjCOneArgSelector()) {
778061da546Spatrick     const std::string &decl_name_string = decl_name.getAsString();
779061da546Spatrick     std::string decl_name_string_without_colon(decl_name_string.c_str(),
780061da546Spatrick                                                decl_name_string.length() - 1);
781061da546Spatrick     IdentifierInfo *ident =
782061da546Spatrick         &original_ctx->Idents.get(decl_name_string_without_colon);
783061da546Spatrick     original_selector = original_ctx->Selectors.getSelector(1, &ident);
784061da546Spatrick   } else {
785061da546Spatrick     SmallVector<IdentifierInfo *, 4> idents;
786061da546Spatrick 
787061da546Spatrick     clang::Selector sel = decl_name.getObjCSelector();
788061da546Spatrick 
789061da546Spatrick     unsigned num_args = sel.getNumArgs();
790061da546Spatrick 
791061da546Spatrick     for (unsigned i = 0; i != num_args; ++i) {
792061da546Spatrick       idents.push_back(&original_ctx->Idents.get(sel.getNameForSlot(i)));
793061da546Spatrick     }
794061da546Spatrick 
795061da546Spatrick     original_selector =
796061da546Spatrick         original_ctx->Selectors.getSelector(num_args, idents.data());
797061da546Spatrick   }
798061da546Spatrick 
799061da546Spatrick   DeclarationName original_decl_name(original_selector);
800061da546Spatrick 
801061da546Spatrick   llvm::SmallVector<NamedDecl *, 1> methods;
802061da546Spatrick 
803dda28197Spatrick   TypeSystemClang::GetCompleteDecl(original_ctx, original_interface_decl);
804061da546Spatrick 
805061da546Spatrick   if (ObjCMethodDecl *instance_method_decl =
806061da546Spatrick           original_interface_decl->lookupInstanceMethod(original_selector)) {
807061da546Spatrick     methods.push_back(instance_method_decl);
808061da546Spatrick   } else if (ObjCMethodDecl *class_method_decl =
809061da546Spatrick                  original_interface_decl->lookupClassMethod(
810061da546Spatrick                      original_selector)) {
811061da546Spatrick     methods.push_back(class_method_decl);
812061da546Spatrick   }
813061da546Spatrick 
814061da546Spatrick   if (methods.empty()) {
815061da546Spatrick     return false;
816061da546Spatrick   }
817061da546Spatrick 
818061da546Spatrick   for (NamedDecl *named_decl : methods) {
819061da546Spatrick     if (!named_decl)
820061da546Spatrick       continue;
821061da546Spatrick 
822061da546Spatrick     ObjCMethodDecl *result_method = dyn_cast<ObjCMethodDecl>(named_decl);
823061da546Spatrick 
824061da546Spatrick     if (!result_method)
825061da546Spatrick       continue;
826061da546Spatrick 
827061da546Spatrick     Decl *copied_decl = CopyDecl(result_method);
828061da546Spatrick 
829061da546Spatrick     if (!copied_decl)
830061da546Spatrick       continue;
831061da546Spatrick 
832061da546Spatrick     ObjCMethodDecl *copied_method_decl = dyn_cast<ObjCMethodDecl>(copied_decl);
833061da546Spatrick 
834061da546Spatrick     if (!copied_method_decl)
835061da546Spatrick       continue;
836061da546Spatrick 
837*f6aab3d8Srobert     Log *log = GetLog(LLDBLog::Expressions);
838061da546Spatrick 
839dda28197Spatrick     LLDB_LOG(log, "  CAS::FOMD found ({0}) {1}", log_info,
840061da546Spatrick              ClangUtil::DumpDecl(copied_method_decl));
841061da546Spatrick 
842061da546Spatrick     context.AddNamedDecl(copied_method_decl);
843061da546Spatrick   }
844061da546Spatrick 
845061da546Spatrick   return true;
846061da546Spatrick }
847061da546Spatrick 
FindDeclInModules(NameSearchContext & context,ConstString name)848dda28197Spatrick void ClangASTSource::FindDeclInModules(NameSearchContext &context,
849dda28197Spatrick                                        ConstString name) {
850*f6aab3d8Srobert   Log *log = GetLog(LLDBLog::Expressions);
851061da546Spatrick 
852be691f3bSpatrick   std::shared_ptr<ClangModulesDeclVendor> modules_decl_vendor =
853be691f3bSpatrick       GetClangModulesDeclVendor();
854dda28197Spatrick   if (!modules_decl_vendor)
855dda28197Spatrick     return;
856dda28197Spatrick 
857dda28197Spatrick   bool append = false;
858dda28197Spatrick   uint32_t max_matches = 1;
859dda28197Spatrick   std::vector<clang::NamedDecl *> decls;
860dda28197Spatrick 
861dda28197Spatrick   if (!modules_decl_vendor->FindDecls(name, append, max_matches, decls))
862dda28197Spatrick     return;
863dda28197Spatrick 
864dda28197Spatrick   LLDB_LOG(log, "  CAS::FEVD Matching entity found for \"{0}\" in the modules",
865dda28197Spatrick            name);
866dda28197Spatrick 
867dda28197Spatrick   clang::NamedDecl *const decl_from_modules = decls[0];
868dda28197Spatrick 
869dda28197Spatrick   if (llvm::isa<clang::TypeDecl>(decl_from_modules) ||
870dda28197Spatrick       llvm::isa<clang::ObjCContainerDecl>(decl_from_modules) ||
871dda28197Spatrick       llvm::isa<clang::EnumConstantDecl>(decl_from_modules)) {
872dda28197Spatrick     clang::Decl *copied_decl = CopyDecl(decl_from_modules);
873dda28197Spatrick     clang::NamedDecl *copied_named_decl =
874dda28197Spatrick         copied_decl ? dyn_cast<clang::NamedDecl>(copied_decl) : nullptr;
875dda28197Spatrick 
876dda28197Spatrick     if (!copied_named_decl) {
877dda28197Spatrick       LLDB_LOG(log, "  CAS::FEVD - Couldn't export a type from the modules");
878dda28197Spatrick 
879dda28197Spatrick       return;
880dda28197Spatrick     }
881dda28197Spatrick 
882dda28197Spatrick     context.AddNamedDecl(copied_named_decl);
883dda28197Spatrick 
884dda28197Spatrick     context.m_found_type = true;
885dda28197Spatrick   }
886dda28197Spatrick }
887dda28197Spatrick 
FindDeclInObjCRuntime(NameSearchContext & context,ConstString name)888dda28197Spatrick void ClangASTSource::FindDeclInObjCRuntime(NameSearchContext &context,
889dda28197Spatrick                                            ConstString name) {
890*f6aab3d8Srobert   Log *log = GetLog(LLDBLog::Expressions);
891dda28197Spatrick 
892dda28197Spatrick   lldb::ProcessSP process(m_target->GetProcessSP());
893dda28197Spatrick 
894dda28197Spatrick   if (!process)
895dda28197Spatrick     return;
896dda28197Spatrick 
897dda28197Spatrick   ObjCLanguageRuntime *language_runtime(ObjCLanguageRuntime::Get(*process));
898dda28197Spatrick 
899dda28197Spatrick   if (!language_runtime)
900dda28197Spatrick     return;
901dda28197Spatrick 
902dda28197Spatrick   DeclVendor *decl_vendor = language_runtime->GetDeclVendor();
903dda28197Spatrick 
904dda28197Spatrick   if (!decl_vendor)
905dda28197Spatrick     return;
906dda28197Spatrick 
907dda28197Spatrick   bool append = false;
908dda28197Spatrick   uint32_t max_matches = 1;
909dda28197Spatrick   std::vector<clang::NamedDecl *> decls;
910dda28197Spatrick 
911dda28197Spatrick   auto *clang_decl_vendor = llvm::cast<ClangDeclVendor>(decl_vendor);
912dda28197Spatrick   if (!clang_decl_vendor->FindDecls(name, append, max_matches, decls))
913dda28197Spatrick     return;
914dda28197Spatrick 
915dda28197Spatrick   LLDB_LOG(log, "  CAS::FEVD Matching type found for \"{0}\" in the runtime",
916dda28197Spatrick            name);
917dda28197Spatrick 
918dda28197Spatrick   clang::Decl *copied_decl = CopyDecl(decls[0]);
919dda28197Spatrick   clang::NamedDecl *copied_named_decl =
920dda28197Spatrick       copied_decl ? dyn_cast<clang::NamedDecl>(copied_decl) : nullptr;
921dda28197Spatrick 
922dda28197Spatrick   if (!copied_named_decl) {
923dda28197Spatrick     LLDB_LOG(log, "  CAS::FEVD - Couldn't export a type from the runtime");
924dda28197Spatrick 
925dda28197Spatrick     return;
926dda28197Spatrick   }
927dda28197Spatrick 
928dda28197Spatrick   context.AddNamedDecl(copied_named_decl);
929dda28197Spatrick }
930dda28197Spatrick 
FindObjCMethodDecls(NameSearchContext & context)931dda28197Spatrick void ClangASTSource::FindObjCMethodDecls(NameSearchContext &context) {
932*f6aab3d8Srobert   Log *log = GetLog(LLDBLog::Expressions);
933061da546Spatrick 
934061da546Spatrick   const DeclarationName &decl_name(context.m_decl_name);
935061da546Spatrick   const DeclContext *decl_ctx(context.m_decl_context);
936061da546Spatrick 
937061da546Spatrick   const ObjCInterfaceDecl *interface_decl =
938061da546Spatrick       dyn_cast<ObjCInterfaceDecl>(decl_ctx);
939061da546Spatrick 
940061da546Spatrick   if (!interface_decl)
941061da546Spatrick     return;
942061da546Spatrick 
943061da546Spatrick   do {
944061da546Spatrick     ClangASTImporter::DeclOrigin original = m_ast_importer_sp->GetDeclOrigin(interface_decl);
945061da546Spatrick 
946061da546Spatrick     if (!original.Valid())
947061da546Spatrick       break;
948061da546Spatrick 
949061da546Spatrick     ObjCInterfaceDecl *original_interface_decl =
950061da546Spatrick         dyn_cast<ObjCInterfaceDecl>(original.decl);
951061da546Spatrick 
952dda28197Spatrick     if (FindObjCMethodDeclsWithOrigin(context, original_interface_decl,
953dda28197Spatrick                                       "at origin"))
954061da546Spatrick       return; // found it, no need to look any further
955061da546Spatrick   } while (false);
956061da546Spatrick 
957061da546Spatrick   StreamString ss;
958061da546Spatrick 
959061da546Spatrick   if (decl_name.isObjCZeroArgSelector()) {
960061da546Spatrick     ss.Printf("%s", decl_name.getAsString().c_str());
961061da546Spatrick   } else if (decl_name.isObjCOneArgSelector()) {
962061da546Spatrick     ss.Printf("%s", decl_name.getAsString().c_str());
963061da546Spatrick   } else {
964061da546Spatrick     clang::Selector sel = decl_name.getObjCSelector();
965061da546Spatrick 
966061da546Spatrick     for (unsigned i = 0, e = sel.getNumArgs(); i != e; ++i) {
967061da546Spatrick       llvm::StringRef r = sel.getNameForSlot(i);
968061da546Spatrick       ss.Printf("%s:", r.str().c_str());
969061da546Spatrick     }
970061da546Spatrick   }
971061da546Spatrick   ss.Flush();
972061da546Spatrick 
973061da546Spatrick   if (ss.GetString().contains("$__lldb"))
974061da546Spatrick     return; // we don't need any results
975061da546Spatrick 
976061da546Spatrick   ConstString selector_name(ss.GetString());
977061da546Spatrick 
978dda28197Spatrick   LLDB_LOG(log,
979dda28197Spatrick            "ClangASTSource::FindObjCMethodDecls on (ASTContext*){0} '{1}' "
980dda28197Spatrick            "for selector [{2} {3}]",
981dda28197Spatrick            m_ast_context, m_clang_ast_context->getDisplayName(),
982dda28197Spatrick            interface_decl->getName(), selector_name);
983061da546Spatrick   SymbolContextList sc_list;
984061da546Spatrick 
985*f6aab3d8Srobert   ModuleFunctionSearchOptions function_options;
986*f6aab3d8Srobert   function_options.include_symbols = false;
987*f6aab3d8Srobert   function_options.include_inlines = false;
988061da546Spatrick 
989061da546Spatrick   std::string interface_name = interface_decl->getNameAsString();
990061da546Spatrick 
991061da546Spatrick   do {
992061da546Spatrick     StreamString ms;
993061da546Spatrick     ms.Printf("-[%s %s]", interface_name.c_str(), selector_name.AsCString());
994061da546Spatrick     ms.Flush();
995061da546Spatrick     ConstString instance_method_name(ms.GetString());
996061da546Spatrick 
997061da546Spatrick     sc_list.Clear();
998*f6aab3d8Srobert     m_target->GetImages().FindFunctions(instance_method_name,
999*f6aab3d8Srobert                                         lldb::eFunctionNameTypeFull,
1000*f6aab3d8Srobert                                         function_options, sc_list);
1001061da546Spatrick 
1002061da546Spatrick     if (sc_list.GetSize())
1003061da546Spatrick       break;
1004061da546Spatrick 
1005061da546Spatrick     ms.Clear();
1006061da546Spatrick     ms.Printf("+[%s %s]", interface_name.c_str(), selector_name.AsCString());
1007061da546Spatrick     ms.Flush();
1008061da546Spatrick     ConstString class_method_name(ms.GetString());
1009061da546Spatrick 
1010061da546Spatrick     sc_list.Clear();
1011*f6aab3d8Srobert     m_target->GetImages().FindFunctions(class_method_name,
1012*f6aab3d8Srobert                                         lldb::eFunctionNameTypeFull,
1013*f6aab3d8Srobert                                         function_options, sc_list);
1014061da546Spatrick 
1015061da546Spatrick     if (sc_list.GetSize())
1016061da546Spatrick       break;
1017061da546Spatrick 
1018061da546Spatrick     // Fall back and check for methods in categories.  If we find methods this
1019061da546Spatrick     // way, we need to check that they're actually in categories on the desired
1020061da546Spatrick     // class.
1021061da546Spatrick 
1022061da546Spatrick     SymbolContextList candidate_sc_list;
1023061da546Spatrick 
1024*f6aab3d8Srobert     m_target->GetImages().FindFunctions(selector_name,
1025*f6aab3d8Srobert                                         lldb::eFunctionNameTypeSelector,
1026*f6aab3d8Srobert                                         function_options, candidate_sc_list);
1027061da546Spatrick 
1028061da546Spatrick     for (uint32_t ci = 0, ce = candidate_sc_list.GetSize(); ci != ce; ++ci) {
1029061da546Spatrick       SymbolContext candidate_sc;
1030061da546Spatrick 
1031061da546Spatrick       if (!candidate_sc_list.GetContextAtIndex(ci, candidate_sc))
1032061da546Spatrick         continue;
1033061da546Spatrick 
1034061da546Spatrick       if (!candidate_sc.function)
1035061da546Spatrick         continue;
1036061da546Spatrick 
1037061da546Spatrick       const char *candidate_name = candidate_sc.function->GetName().AsCString();
1038061da546Spatrick 
1039061da546Spatrick       const char *cursor = candidate_name;
1040061da546Spatrick 
1041061da546Spatrick       if (*cursor != '+' && *cursor != '-')
1042061da546Spatrick         continue;
1043061da546Spatrick 
1044061da546Spatrick       ++cursor;
1045061da546Spatrick 
1046061da546Spatrick       if (*cursor != '[')
1047061da546Spatrick         continue;
1048061da546Spatrick 
1049061da546Spatrick       ++cursor;
1050061da546Spatrick 
1051061da546Spatrick       size_t interface_len = interface_name.length();
1052061da546Spatrick 
1053061da546Spatrick       if (strncmp(cursor, interface_name.c_str(), interface_len))
1054061da546Spatrick         continue;
1055061da546Spatrick 
1056061da546Spatrick       cursor += interface_len;
1057061da546Spatrick 
1058061da546Spatrick       if (*cursor == ' ' || *cursor == '(')
1059061da546Spatrick         sc_list.Append(candidate_sc);
1060061da546Spatrick     }
1061061da546Spatrick   } while (false);
1062061da546Spatrick 
1063061da546Spatrick   if (sc_list.GetSize()) {
1064061da546Spatrick     // We found a good function symbol.  Use that.
1065061da546Spatrick 
1066061da546Spatrick     for (uint32_t i = 0, e = sc_list.GetSize(); i != e; ++i) {
1067061da546Spatrick       SymbolContext sc;
1068061da546Spatrick 
1069061da546Spatrick       if (!sc_list.GetContextAtIndex(i, sc))
1070061da546Spatrick         continue;
1071061da546Spatrick 
1072061da546Spatrick       if (!sc.function)
1073061da546Spatrick         continue;
1074061da546Spatrick 
1075061da546Spatrick       CompilerDeclContext function_decl_ctx = sc.function->GetDeclContext();
1076061da546Spatrick       if (!function_decl_ctx)
1077061da546Spatrick         continue;
1078061da546Spatrick 
1079061da546Spatrick       ObjCMethodDecl *method_decl =
1080dda28197Spatrick           TypeSystemClang::DeclContextGetAsObjCMethodDecl(function_decl_ctx);
1081061da546Spatrick 
1082061da546Spatrick       if (!method_decl)
1083061da546Spatrick         continue;
1084061da546Spatrick 
1085061da546Spatrick       ObjCInterfaceDecl *found_interface_decl =
1086061da546Spatrick           method_decl->getClassInterface();
1087061da546Spatrick 
1088061da546Spatrick       if (!found_interface_decl)
1089061da546Spatrick         continue;
1090061da546Spatrick 
1091061da546Spatrick       if (found_interface_decl->getName() == interface_decl->getName()) {
1092061da546Spatrick         Decl *copied_decl = CopyDecl(method_decl);
1093061da546Spatrick 
1094061da546Spatrick         if (!copied_decl)
1095061da546Spatrick           continue;
1096061da546Spatrick 
1097061da546Spatrick         ObjCMethodDecl *copied_method_decl =
1098061da546Spatrick             dyn_cast<ObjCMethodDecl>(copied_decl);
1099061da546Spatrick 
1100061da546Spatrick         if (!copied_method_decl)
1101061da546Spatrick           continue;
1102061da546Spatrick 
1103dda28197Spatrick         LLDB_LOG(log, "  CAS::FOMD found (in symbols)\n{0}",
1104061da546Spatrick                  ClangUtil::DumpDecl(copied_method_decl));
1105061da546Spatrick 
1106061da546Spatrick         context.AddNamedDecl(copied_method_decl);
1107061da546Spatrick       }
1108061da546Spatrick     }
1109061da546Spatrick 
1110061da546Spatrick     return;
1111061da546Spatrick   }
1112061da546Spatrick 
1113061da546Spatrick   // Try the debug information.
1114061da546Spatrick 
1115061da546Spatrick   do {
1116061da546Spatrick     ObjCInterfaceDecl *complete_interface_decl = GetCompleteObjCInterface(
1117061da546Spatrick         const_cast<ObjCInterfaceDecl *>(interface_decl));
1118061da546Spatrick 
1119061da546Spatrick     if (!complete_interface_decl)
1120061da546Spatrick       break;
1121061da546Spatrick 
1122061da546Spatrick     // We found the complete interface.  The runtime never needs to be queried
1123061da546Spatrick     // in this scenario.
1124061da546Spatrick 
1125061da546Spatrick     DeclFromUser<const ObjCInterfaceDecl> complete_iface_decl(
1126061da546Spatrick         complete_interface_decl);
1127061da546Spatrick 
1128061da546Spatrick     if (complete_interface_decl == interface_decl)
1129061da546Spatrick       break; // already checked this one
1130061da546Spatrick 
1131dda28197Spatrick     LLDB_LOG(log,
1132dda28197Spatrick              "CAS::FOPD trying origin "
1133dda28197Spatrick              "(ObjCInterfaceDecl*){0}/(ASTContext*){1}...",
1134dda28197Spatrick              complete_interface_decl, &complete_iface_decl->getASTContext());
1135061da546Spatrick 
1136dda28197Spatrick     FindObjCMethodDeclsWithOrigin(context, complete_interface_decl,
1137061da546Spatrick                                   "in debug info");
1138061da546Spatrick 
1139061da546Spatrick     return;
1140061da546Spatrick   } while (false);
1141061da546Spatrick 
1142061da546Spatrick   do {
1143061da546Spatrick     // Check the modules only if the debug information didn't have a complete
1144061da546Spatrick     // interface.
1145061da546Spatrick 
1146be691f3bSpatrick     if (std::shared_ptr<ClangModulesDeclVendor> modules_decl_vendor =
1147be691f3bSpatrick             GetClangModulesDeclVendor()) {
1148061da546Spatrick       ConstString interface_name(interface_decl->getNameAsString().c_str());
1149061da546Spatrick       bool append = false;
1150061da546Spatrick       uint32_t max_matches = 1;
1151061da546Spatrick       std::vector<clang::NamedDecl *> decls;
1152061da546Spatrick 
1153061da546Spatrick       if (!modules_decl_vendor->FindDecls(interface_name, append, max_matches,
1154061da546Spatrick                                           decls))
1155061da546Spatrick         break;
1156061da546Spatrick 
1157061da546Spatrick       ObjCInterfaceDecl *interface_decl_from_modules =
1158061da546Spatrick           dyn_cast<ObjCInterfaceDecl>(decls[0]);
1159061da546Spatrick 
1160061da546Spatrick       if (!interface_decl_from_modules)
1161061da546Spatrick         break;
1162061da546Spatrick 
1163dda28197Spatrick       if (FindObjCMethodDeclsWithOrigin(context, interface_decl_from_modules,
1164dda28197Spatrick                                         "in modules"))
1165061da546Spatrick         return;
1166061da546Spatrick     }
1167061da546Spatrick   } while (false);
1168061da546Spatrick 
1169061da546Spatrick   do {
1170061da546Spatrick     // Check the runtime only if the debug information didn't have a complete
1171061da546Spatrick     // interface and the modules don't get us anywhere.
1172061da546Spatrick 
1173061da546Spatrick     lldb::ProcessSP process(m_target->GetProcessSP());
1174061da546Spatrick 
1175061da546Spatrick     if (!process)
1176061da546Spatrick       break;
1177061da546Spatrick 
1178061da546Spatrick     ObjCLanguageRuntime *language_runtime(ObjCLanguageRuntime::Get(*process));
1179061da546Spatrick 
1180061da546Spatrick     if (!language_runtime)
1181061da546Spatrick       break;
1182061da546Spatrick 
1183061da546Spatrick     DeclVendor *decl_vendor = language_runtime->GetDeclVendor();
1184061da546Spatrick 
1185061da546Spatrick     if (!decl_vendor)
1186061da546Spatrick       break;
1187061da546Spatrick 
1188061da546Spatrick     ConstString interface_name(interface_decl->getNameAsString().c_str());
1189061da546Spatrick     bool append = false;
1190061da546Spatrick     uint32_t max_matches = 1;
1191061da546Spatrick     std::vector<clang::NamedDecl *> decls;
1192061da546Spatrick 
1193061da546Spatrick     auto *clang_decl_vendor = llvm::cast<ClangDeclVendor>(decl_vendor);
1194061da546Spatrick     if (!clang_decl_vendor->FindDecls(interface_name, append, max_matches,
1195061da546Spatrick                                       decls))
1196061da546Spatrick       break;
1197061da546Spatrick 
1198061da546Spatrick     ObjCInterfaceDecl *runtime_interface_decl =
1199061da546Spatrick         dyn_cast<ObjCInterfaceDecl>(decls[0]);
1200061da546Spatrick 
1201061da546Spatrick     if (!runtime_interface_decl)
1202061da546Spatrick       break;
1203061da546Spatrick 
1204dda28197Spatrick     FindObjCMethodDeclsWithOrigin(context, runtime_interface_decl,
1205061da546Spatrick                                   "in runtime");
1206061da546Spatrick   } while (false);
1207061da546Spatrick }
1208061da546Spatrick 
FindObjCPropertyAndIvarDeclsWithOrigin(NameSearchContext & context,ClangASTSource & source,DeclFromUser<const ObjCInterfaceDecl> & origin_iface_decl)1209061da546Spatrick static bool FindObjCPropertyAndIvarDeclsWithOrigin(
1210dda28197Spatrick     NameSearchContext &context, ClangASTSource &source,
1211061da546Spatrick     DeclFromUser<const ObjCInterfaceDecl> &origin_iface_decl) {
1212*f6aab3d8Srobert   Log *log = GetLog(LLDBLog::Expressions);
1213061da546Spatrick 
1214061da546Spatrick   if (origin_iface_decl.IsInvalid())
1215061da546Spatrick     return false;
1216061da546Spatrick 
1217061da546Spatrick   std::string name_str = context.m_decl_name.getAsString();
1218061da546Spatrick   StringRef name(name_str);
1219061da546Spatrick   IdentifierInfo &name_identifier(
1220061da546Spatrick       origin_iface_decl->getASTContext().Idents.get(name));
1221061da546Spatrick 
1222061da546Spatrick   DeclFromUser<ObjCPropertyDecl> origin_property_decl(
1223061da546Spatrick       origin_iface_decl->FindPropertyDeclaration(
1224061da546Spatrick           &name_identifier, ObjCPropertyQueryKind::OBJC_PR_query_instance));
1225061da546Spatrick 
1226061da546Spatrick   bool found = false;
1227061da546Spatrick 
1228061da546Spatrick   if (origin_property_decl.IsValid()) {
1229061da546Spatrick     DeclFromParser<ObjCPropertyDecl> parser_property_decl(
1230061da546Spatrick         origin_property_decl.Import(source));
1231061da546Spatrick     if (parser_property_decl.IsValid()) {
1232dda28197Spatrick       LLDB_LOG(log, "  CAS::FOPD found\n{0}",
1233061da546Spatrick                ClangUtil::DumpDecl(parser_property_decl.decl));
1234061da546Spatrick 
1235061da546Spatrick       context.AddNamedDecl(parser_property_decl.decl);
1236061da546Spatrick       found = true;
1237061da546Spatrick     }
1238061da546Spatrick   }
1239061da546Spatrick 
1240061da546Spatrick   DeclFromUser<ObjCIvarDecl> origin_ivar_decl(
1241061da546Spatrick       origin_iface_decl->getIvarDecl(&name_identifier));
1242061da546Spatrick 
1243061da546Spatrick   if (origin_ivar_decl.IsValid()) {
1244061da546Spatrick     DeclFromParser<ObjCIvarDecl> parser_ivar_decl(
1245061da546Spatrick         origin_ivar_decl.Import(source));
1246061da546Spatrick     if (parser_ivar_decl.IsValid()) {
1247dda28197Spatrick       LLDB_LOG(log, "  CAS::FOPD found\n{0}",
1248061da546Spatrick                ClangUtil::DumpDecl(parser_ivar_decl.decl));
1249061da546Spatrick 
1250061da546Spatrick       context.AddNamedDecl(parser_ivar_decl.decl);
1251061da546Spatrick       found = true;
1252061da546Spatrick     }
1253061da546Spatrick   }
1254061da546Spatrick 
1255061da546Spatrick   return found;
1256061da546Spatrick }
1257061da546Spatrick 
FindObjCPropertyAndIvarDecls(NameSearchContext & context)1258061da546Spatrick void ClangASTSource::FindObjCPropertyAndIvarDecls(NameSearchContext &context) {
1259*f6aab3d8Srobert   Log *log = GetLog(LLDBLog::Expressions);
1260061da546Spatrick 
1261061da546Spatrick   DeclFromParser<const ObjCInterfaceDecl> parser_iface_decl(
1262061da546Spatrick       cast<ObjCInterfaceDecl>(context.m_decl_context));
1263061da546Spatrick   DeclFromUser<const ObjCInterfaceDecl> origin_iface_decl(
1264061da546Spatrick       parser_iface_decl.GetOrigin(*this));
1265061da546Spatrick 
1266061da546Spatrick   ConstString class_name(parser_iface_decl->getNameAsString().c_str());
1267061da546Spatrick 
1268dda28197Spatrick   LLDB_LOG(log,
1269dda28197Spatrick            "ClangASTSource::FindObjCPropertyAndIvarDecls on "
1270dda28197Spatrick            "(ASTContext*){0} '{1}' for '{2}.{3}'",
1271dda28197Spatrick            m_ast_context, m_clang_ast_context->getDisplayName(),
1272dda28197Spatrick            parser_iface_decl->getName(), context.m_decl_name.getAsString());
1273061da546Spatrick 
1274dda28197Spatrick   if (FindObjCPropertyAndIvarDeclsWithOrigin(context, *this, origin_iface_decl))
1275061da546Spatrick     return;
1276061da546Spatrick 
1277dda28197Spatrick   LLDB_LOG(log,
1278dda28197Spatrick            "CAS::FOPD couldn't find the property on origin "
1279dda28197Spatrick            "(ObjCInterfaceDecl*){0}/(ASTContext*){1}, searching "
1280061da546Spatrick            "elsewhere...",
1281dda28197Spatrick            origin_iface_decl.decl, &origin_iface_decl->getASTContext());
1282061da546Spatrick 
1283061da546Spatrick   SymbolContext null_sc;
1284061da546Spatrick   TypeList type_list;
1285061da546Spatrick 
1286061da546Spatrick   do {
1287061da546Spatrick     ObjCInterfaceDecl *complete_interface_decl = GetCompleteObjCInterface(
1288061da546Spatrick         const_cast<ObjCInterfaceDecl *>(parser_iface_decl.decl));
1289061da546Spatrick 
1290061da546Spatrick     if (!complete_interface_decl)
1291061da546Spatrick       break;
1292061da546Spatrick 
1293061da546Spatrick     // We found the complete interface.  The runtime never needs to be queried
1294061da546Spatrick     // in this scenario.
1295061da546Spatrick 
1296061da546Spatrick     DeclFromUser<const ObjCInterfaceDecl> complete_iface_decl(
1297061da546Spatrick         complete_interface_decl);
1298061da546Spatrick 
1299061da546Spatrick     if (complete_iface_decl.decl == origin_iface_decl.decl)
1300061da546Spatrick       break; // already checked this one
1301061da546Spatrick 
1302dda28197Spatrick     LLDB_LOG(log,
1303dda28197Spatrick              "CAS::FOPD trying origin "
1304dda28197Spatrick              "(ObjCInterfaceDecl*){0}/(ASTContext*){1}...",
1305dda28197Spatrick              complete_iface_decl.decl, &complete_iface_decl->getASTContext());
1306061da546Spatrick 
1307dda28197Spatrick     FindObjCPropertyAndIvarDeclsWithOrigin(context, *this, complete_iface_decl);
1308061da546Spatrick 
1309061da546Spatrick     return;
1310061da546Spatrick   } while (false);
1311061da546Spatrick 
1312061da546Spatrick   do {
1313061da546Spatrick     // Check the modules only if the debug information didn't have a complete
1314061da546Spatrick     // interface.
1315061da546Spatrick 
1316be691f3bSpatrick     std::shared_ptr<ClangModulesDeclVendor> modules_decl_vendor =
1317be691f3bSpatrick         GetClangModulesDeclVendor();
1318061da546Spatrick 
1319061da546Spatrick     if (!modules_decl_vendor)
1320061da546Spatrick       break;
1321061da546Spatrick 
1322061da546Spatrick     bool append = false;
1323061da546Spatrick     uint32_t max_matches = 1;
1324061da546Spatrick     std::vector<clang::NamedDecl *> decls;
1325061da546Spatrick 
1326061da546Spatrick     if (!modules_decl_vendor->FindDecls(class_name, append, max_matches, decls))
1327061da546Spatrick       break;
1328061da546Spatrick 
1329061da546Spatrick     DeclFromUser<const ObjCInterfaceDecl> interface_decl_from_modules(
1330061da546Spatrick         dyn_cast<ObjCInterfaceDecl>(decls[0]));
1331061da546Spatrick 
1332061da546Spatrick     if (!interface_decl_from_modules.IsValid())
1333061da546Spatrick       break;
1334061da546Spatrick 
1335dda28197Spatrick     LLDB_LOG(log,
1336dda28197Spatrick              "CAS::FOPD[{0}] trying module "
1337dda28197Spatrick              "(ObjCInterfaceDecl*){0}/(ASTContext*){1}...",
1338dda28197Spatrick              interface_decl_from_modules.decl,
1339dda28197Spatrick              &interface_decl_from_modules->getASTContext());
1340061da546Spatrick 
1341dda28197Spatrick     if (FindObjCPropertyAndIvarDeclsWithOrigin(context, *this,
1342061da546Spatrick                                                interface_decl_from_modules))
1343061da546Spatrick       return;
1344061da546Spatrick   } while (false);
1345061da546Spatrick 
1346061da546Spatrick   do {
1347061da546Spatrick     // Check the runtime only if the debug information didn't have a complete
1348061da546Spatrick     // interface and nothing was in the modules.
1349061da546Spatrick 
1350061da546Spatrick     lldb::ProcessSP process(m_target->GetProcessSP());
1351061da546Spatrick 
1352061da546Spatrick     if (!process)
1353061da546Spatrick       return;
1354061da546Spatrick 
1355061da546Spatrick     ObjCLanguageRuntime *language_runtime(ObjCLanguageRuntime::Get(*process));
1356061da546Spatrick 
1357061da546Spatrick     if (!language_runtime)
1358061da546Spatrick       return;
1359061da546Spatrick 
1360061da546Spatrick     DeclVendor *decl_vendor = language_runtime->GetDeclVendor();
1361061da546Spatrick 
1362061da546Spatrick     if (!decl_vendor)
1363061da546Spatrick       break;
1364061da546Spatrick 
1365061da546Spatrick     bool append = false;
1366061da546Spatrick     uint32_t max_matches = 1;
1367061da546Spatrick     std::vector<clang::NamedDecl *> decls;
1368061da546Spatrick 
1369061da546Spatrick     auto *clang_decl_vendor = llvm::cast<ClangDeclVendor>(decl_vendor);
1370061da546Spatrick     if (!clang_decl_vendor->FindDecls(class_name, append, max_matches, decls))
1371061da546Spatrick       break;
1372061da546Spatrick 
1373061da546Spatrick     DeclFromUser<const ObjCInterfaceDecl> interface_decl_from_runtime(
1374061da546Spatrick         dyn_cast<ObjCInterfaceDecl>(decls[0]));
1375061da546Spatrick 
1376061da546Spatrick     if (!interface_decl_from_runtime.IsValid())
1377061da546Spatrick       break;
1378061da546Spatrick 
1379dda28197Spatrick     LLDB_LOG(log,
1380dda28197Spatrick              "CAS::FOPD[{0}] trying runtime "
1381dda28197Spatrick              "(ObjCInterfaceDecl*){0}/(ASTContext*){1}...",
1382dda28197Spatrick              interface_decl_from_runtime.decl,
1383dda28197Spatrick              &interface_decl_from_runtime->getASTContext());
1384061da546Spatrick 
1385dda28197Spatrick     if (FindObjCPropertyAndIvarDeclsWithOrigin(context, *this,
1386dda28197Spatrick                                                interface_decl_from_runtime))
1387061da546Spatrick       return;
1388061da546Spatrick   } while (false);
1389061da546Spatrick }
1390061da546Spatrick 
LookupInNamespace(NameSearchContext & context)1391dda28197Spatrick void ClangASTSource::LookupInNamespace(NameSearchContext &context) {
1392dda28197Spatrick   const NamespaceDecl *namespace_context =
1393dda28197Spatrick       dyn_cast<NamespaceDecl>(context.m_decl_context);
1394dda28197Spatrick 
1395*f6aab3d8Srobert   Log *log = GetLog(LLDBLog::Expressions);
1396dda28197Spatrick 
1397dda28197Spatrick   ClangASTImporter::NamespaceMapSP namespace_map =
1398dda28197Spatrick       m_ast_importer_sp->GetNamespaceMap(namespace_context);
1399dda28197Spatrick 
1400dda28197Spatrick   LLDB_LOGV(log, "  CAS::FEVD Inspecting namespace map {0} ({1} entries)",
1401dda28197Spatrick             namespace_map.get(), namespace_map->size());
1402dda28197Spatrick 
1403dda28197Spatrick   if (!namespace_map)
1404dda28197Spatrick     return;
1405dda28197Spatrick 
1406dda28197Spatrick   for (ClangASTImporter::NamespaceMap::iterator i = namespace_map->begin(),
1407dda28197Spatrick                                                 e = namespace_map->end();
1408dda28197Spatrick        i != e; ++i) {
1409dda28197Spatrick     LLDB_LOG(log, "  CAS::FEVD Searching namespace {0} in module {1}",
1410dda28197Spatrick              i->second.GetName(), i->first->GetFileSpec().GetFilename());
1411dda28197Spatrick 
1412dda28197Spatrick     FindExternalVisibleDecls(context, i->first, i->second);
1413dda28197Spatrick   }
1414dda28197Spatrick }
1415dda28197Spatrick 
1416061da546Spatrick typedef llvm::DenseMap<const FieldDecl *, uint64_t> FieldOffsetMap;
1417061da546Spatrick typedef llvm::DenseMap<const CXXRecordDecl *, CharUnits> BaseOffsetMap;
1418061da546Spatrick 
1419061da546Spatrick template <class D, class O>
ImportOffsetMap(llvm::DenseMap<const D *,O> & destination_map,llvm::DenseMap<const D *,O> & source_map,ClangASTSource & source)1420061da546Spatrick static bool ImportOffsetMap(llvm::DenseMap<const D *, O> &destination_map,
1421061da546Spatrick                             llvm::DenseMap<const D *, O> &source_map,
1422061da546Spatrick                             ClangASTSource &source) {
1423061da546Spatrick   // When importing fields into a new record, clang has a hard requirement that
1424061da546Spatrick   // fields be imported in field offset order.  Since they are stored in a
1425061da546Spatrick   // DenseMap with a pointer as the key type, this means we cannot simply
1426061da546Spatrick   // iterate over the map, as the order will be non-deterministic.  Instead we
1427061da546Spatrick   // have to sort by the offset and then insert in sorted order.
1428061da546Spatrick   typedef llvm::DenseMap<const D *, O> MapType;
1429061da546Spatrick   typedef typename MapType::value_type PairType;
1430061da546Spatrick   std::vector<PairType> sorted_items;
1431061da546Spatrick   sorted_items.reserve(source_map.size());
1432061da546Spatrick   sorted_items.assign(source_map.begin(), source_map.end());
1433*f6aab3d8Srobert   llvm::sort(sorted_items, llvm::less_second());
1434061da546Spatrick 
1435061da546Spatrick   for (const auto &item : sorted_items) {
1436061da546Spatrick     DeclFromUser<D> user_decl(const_cast<D *>(item.first));
1437061da546Spatrick     DeclFromParser<D> parser_decl(user_decl.Import(source));
1438061da546Spatrick     if (parser_decl.IsInvalid())
1439061da546Spatrick       return false;
1440061da546Spatrick     destination_map.insert(
1441061da546Spatrick         std::pair<const D *, O>(parser_decl.decl, item.second));
1442061da546Spatrick   }
1443061da546Spatrick 
1444061da546Spatrick   return true;
1445061da546Spatrick }
1446061da546Spatrick 
1447061da546Spatrick template <bool IsVirtual>
ExtractBaseOffsets(const ASTRecordLayout & record_layout,DeclFromUser<const CXXRecordDecl> & record,BaseOffsetMap & base_offsets)1448061da546Spatrick bool ExtractBaseOffsets(const ASTRecordLayout &record_layout,
1449061da546Spatrick                         DeclFromUser<const CXXRecordDecl> &record,
1450061da546Spatrick                         BaseOffsetMap &base_offsets) {
1451061da546Spatrick   for (CXXRecordDecl::base_class_const_iterator
1452061da546Spatrick            bi = (IsVirtual ? record->vbases_begin() : record->bases_begin()),
1453061da546Spatrick            be = (IsVirtual ? record->vbases_end() : record->bases_end());
1454061da546Spatrick        bi != be; ++bi) {
1455061da546Spatrick     if (!IsVirtual && bi->isVirtual())
1456061da546Spatrick       continue;
1457061da546Spatrick 
1458061da546Spatrick     const clang::Type *origin_base_type = bi->getType().getTypePtr();
1459061da546Spatrick     const clang::RecordType *origin_base_record_type =
1460061da546Spatrick         origin_base_type->getAs<RecordType>();
1461061da546Spatrick 
1462061da546Spatrick     if (!origin_base_record_type)
1463061da546Spatrick       return false;
1464061da546Spatrick 
1465061da546Spatrick     DeclFromUser<RecordDecl> origin_base_record(
1466061da546Spatrick         origin_base_record_type->getDecl());
1467061da546Spatrick 
1468061da546Spatrick     if (origin_base_record.IsInvalid())
1469061da546Spatrick       return false;
1470061da546Spatrick 
1471061da546Spatrick     DeclFromUser<CXXRecordDecl> origin_base_cxx_record(
1472061da546Spatrick         DynCast<CXXRecordDecl>(origin_base_record));
1473061da546Spatrick 
1474061da546Spatrick     if (origin_base_cxx_record.IsInvalid())
1475061da546Spatrick       return false;
1476061da546Spatrick 
1477061da546Spatrick     CharUnits base_offset;
1478061da546Spatrick 
1479061da546Spatrick     if (IsVirtual)
1480061da546Spatrick       base_offset =
1481061da546Spatrick           record_layout.getVBaseClassOffset(origin_base_cxx_record.decl);
1482061da546Spatrick     else
1483061da546Spatrick       base_offset =
1484061da546Spatrick           record_layout.getBaseClassOffset(origin_base_cxx_record.decl);
1485061da546Spatrick 
1486061da546Spatrick     base_offsets.insert(std::pair<const CXXRecordDecl *, CharUnits>(
1487061da546Spatrick         origin_base_cxx_record.decl, base_offset));
1488061da546Spatrick   }
1489061da546Spatrick 
1490061da546Spatrick   return true;
1491061da546Spatrick }
1492061da546Spatrick 
layoutRecordType(const RecordDecl * record,uint64_t & size,uint64_t & alignment,FieldOffsetMap & field_offsets,BaseOffsetMap & base_offsets,BaseOffsetMap & virtual_base_offsets)1493061da546Spatrick bool ClangASTSource::layoutRecordType(const RecordDecl *record, uint64_t &size,
1494061da546Spatrick                                       uint64_t &alignment,
1495061da546Spatrick                                       FieldOffsetMap &field_offsets,
1496061da546Spatrick                                       BaseOffsetMap &base_offsets,
1497061da546Spatrick                                       BaseOffsetMap &virtual_base_offsets) {
1498061da546Spatrick 
1499*f6aab3d8Srobert   Log *log = GetLog(LLDBLog::Expressions);
1500061da546Spatrick 
1501dda28197Spatrick   LLDB_LOG(log,
1502dda28197Spatrick            "LayoutRecordType on (ASTContext*){0} '{1}' for (RecordDecl*)"
1503*f6aab3d8Srobert            "{2} [name = '{3}']",
1504dda28197Spatrick            m_ast_context, m_clang_ast_context->getDisplayName(), record,
1505dda28197Spatrick            record->getName());
1506061da546Spatrick 
1507061da546Spatrick   DeclFromParser<const RecordDecl> parser_record(record);
1508061da546Spatrick   DeclFromUser<const RecordDecl> origin_record(
1509061da546Spatrick       parser_record.GetOrigin(*this));
1510061da546Spatrick 
1511061da546Spatrick   if (origin_record.IsInvalid())
1512061da546Spatrick     return false;
1513061da546Spatrick 
1514061da546Spatrick   FieldOffsetMap origin_field_offsets;
1515061da546Spatrick   BaseOffsetMap origin_base_offsets;
1516061da546Spatrick   BaseOffsetMap origin_virtual_base_offsets;
1517061da546Spatrick 
1518dda28197Spatrick   TypeSystemClang::GetCompleteDecl(
1519061da546Spatrick       &origin_record->getASTContext(),
1520061da546Spatrick       const_cast<RecordDecl *>(origin_record.decl));
1521061da546Spatrick 
1522061da546Spatrick   clang::RecordDecl *definition = origin_record.decl->getDefinition();
1523061da546Spatrick   if (!definition || !definition->isCompleteDefinition())
1524061da546Spatrick     return false;
1525061da546Spatrick 
1526061da546Spatrick   const ASTRecordLayout &record_layout(
1527061da546Spatrick       origin_record->getASTContext().getASTRecordLayout(origin_record.decl));
1528061da546Spatrick 
1529061da546Spatrick   int field_idx = 0, field_count = record_layout.getFieldCount();
1530061da546Spatrick 
1531061da546Spatrick   for (RecordDecl::field_iterator fi = origin_record->field_begin(),
1532061da546Spatrick                                   fe = origin_record->field_end();
1533061da546Spatrick        fi != fe; ++fi) {
1534061da546Spatrick     if (field_idx >= field_count)
1535061da546Spatrick       return false; // Layout didn't go well.  Bail out.
1536061da546Spatrick 
1537061da546Spatrick     uint64_t field_offset = record_layout.getFieldOffset(field_idx);
1538061da546Spatrick 
1539061da546Spatrick     origin_field_offsets.insert(
1540061da546Spatrick         std::pair<const FieldDecl *, uint64_t>(*fi, field_offset));
1541061da546Spatrick 
1542061da546Spatrick     field_idx++;
1543061da546Spatrick   }
1544061da546Spatrick 
1545061da546Spatrick   lldbassert(&record->getASTContext() == m_ast_context);
1546061da546Spatrick 
1547061da546Spatrick   DeclFromUser<const CXXRecordDecl> origin_cxx_record(
1548061da546Spatrick       DynCast<const CXXRecordDecl>(origin_record));
1549061da546Spatrick 
1550061da546Spatrick   if (origin_cxx_record.IsValid()) {
1551061da546Spatrick     if (!ExtractBaseOffsets<false>(record_layout, origin_cxx_record,
1552061da546Spatrick                                    origin_base_offsets) ||
1553061da546Spatrick         !ExtractBaseOffsets<true>(record_layout, origin_cxx_record,
1554061da546Spatrick                                   origin_virtual_base_offsets))
1555061da546Spatrick       return false;
1556061da546Spatrick   }
1557061da546Spatrick 
1558061da546Spatrick   if (!ImportOffsetMap(field_offsets, origin_field_offsets, *this) ||
1559061da546Spatrick       !ImportOffsetMap(base_offsets, origin_base_offsets, *this) ||
1560061da546Spatrick       !ImportOffsetMap(virtual_base_offsets, origin_virtual_base_offsets,
1561061da546Spatrick                        *this))
1562061da546Spatrick     return false;
1563061da546Spatrick 
1564061da546Spatrick   size = record_layout.getSize().getQuantity() * m_ast_context->getCharWidth();
1565061da546Spatrick   alignment = record_layout.getAlignment().getQuantity() *
1566061da546Spatrick               m_ast_context->getCharWidth();
1567061da546Spatrick 
1568061da546Spatrick   if (log) {
1569dda28197Spatrick     LLDB_LOG(log, "LRT returned:");
1570be691f3bSpatrick     LLDB_LOG(log, "LRT   Original = (RecordDecl*){0}",
1571061da546Spatrick              static_cast<const void *>(origin_record.decl));
1572be691f3bSpatrick     LLDB_LOG(log, "LRT   Size = {0}", size);
1573be691f3bSpatrick     LLDB_LOG(log, "LRT   Alignment = {0}", alignment);
1574dda28197Spatrick     LLDB_LOG(log, "LRT   Fields:");
1575061da546Spatrick     for (RecordDecl::field_iterator fi = record->field_begin(),
1576061da546Spatrick                                     fe = record->field_end();
1577061da546Spatrick          fi != fe; ++fi) {
1578dda28197Spatrick       LLDB_LOG(log,
1579*f6aab3d8Srobert                "LRT     (FieldDecl*){0}, Name = '{1}', Type = '{2}', Offset = "
1580*f6aab3d8Srobert                "{3} bits",
1581*f6aab3d8Srobert                *fi, fi->getName(), fi->getType().getAsString(),
1582*f6aab3d8Srobert                field_offsets[*fi]);
1583061da546Spatrick     }
1584061da546Spatrick     DeclFromParser<const CXXRecordDecl> parser_cxx_record =
1585061da546Spatrick         DynCast<const CXXRecordDecl>(parser_record);
1586061da546Spatrick     if (parser_cxx_record.IsValid()) {
1587dda28197Spatrick       LLDB_LOG(log, "LRT   Bases:");
1588061da546Spatrick       for (CXXRecordDecl::base_class_const_iterator
1589061da546Spatrick                bi = parser_cxx_record->bases_begin(),
1590061da546Spatrick                be = parser_cxx_record->bases_end();
1591061da546Spatrick            bi != be; ++bi) {
1592061da546Spatrick         bool is_virtual = bi->isVirtual();
1593061da546Spatrick 
1594061da546Spatrick         QualType base_type = bi->getType();
1595061da546Spatrick         const RecordType *base_record_type = base_type->getAs<RecordType>();
1596061da546Spatrick         DeclFromParser<RecordDecl> base_record(base_record_type->getDecl());
1597061da546Spatrick         DeclFromParser<CXXRecordDecl> base_cxx_record =
1598061da546Spatrick             DynCast<CXXRecordDecl>(base_record);
1599061da546Spatrick 
1600dda28197Spatrick         LLDB_LOG(log,
1601dda28197Spatrick                  "LRT     {0}(CXXRecordDecl*){1}, Name = '{2}', Offset = "
1602dda28197Spatrick                  "{3} chars",
1603dda28197Spatrick                  (is_virtual ? "Virtual " : ""), base_cxx_record.decl,
1604dda28197Spatrick                  base_cxx_record.decl->getName(),
1605061da546Spatrick                  (is_virtual
1606061da546Spatrick                       ? virtual_base_offsets[base_cxx_record.decl].getQuantity()
1607061da546Spatrick                       : base_offsets[base_cxx_record.decl].getQuantity()));
1608061da546Spatrick       }
1609061da546Spatrick     } else {
1610dda28197Spatrick       LLDB_LOG(log, "LRD   Not a CXXRecord, so no bases");
1611061da546Spatrick     }
1612061da546Spatrick   }
1613061da546Spatrick 
1614061da546Spatrick   return true;
1615061da546Spatrick }
1616061da546Spatrick 
CompleteNamespaceMap(ClangASTImporter::NamespaceMapSP & namespace_map,ConstString name,ClangASTImporter::NamespaceMapSP & parent_map) const1617061da546Spatrick void ClangASTSource::CompleteNamespaceMap(
1618061da546Spatrick     ClangASTImporter::NamespaceMapSP &namespace_map, ConstString name,
1619061da546Spatrick     ClangASTImporter::NamespaceMapSP &parent_map) const {
1620061da546Spatrick 
1621*f6aab3d8Srobert   Log *log = GetLog(LLDBLog::Expressions);
1622061da546Spatrick 
1623061da546Spatrick   if (log) {
1624061da546Spatrick     if (parent_map && parent_map->size())
1625dda28197Spatrick       LLDB_LOG(log,
1626dda28197Spatrick                "CompleteNamespaceMap on (ASTContext*){0} '{1}' Searching "
1627dda28197Spatrick                "for namespace {2} in namespace {3}",
1628dda28197Spatrick                m_ast_context, m_clang_ast_context->getDisplayName(), name,
1629dda28197Spatrick                parent_map->begin()->second.GetName());
1630061da546Spatrick     else
1631dda28197Spatrick       LLDB_LOG(log,
1632dda28197Spatrick                "CompleteNamespaceMap on (ASTContext*){0} '{1}' Searching "
1633dda28197Spatrick                "for namespace {2}",
1634dda28197Spatrick                m_ast_context, m_clang_ast_context->getDisplayName(), name);
1635061da546Spatrick   }
1636061da546Spatrick 
1637061da546Spatrick   if (parent_map) {
1638061da546Spatrick     for (ClangASTImporter::NamespaceMap::iterator i = parent_map->begin(),
1639061da546Spatrick                                                   e = parent_map->end();
1640061da546Spatrick          i != e; ++i) {
1641061da546Spatrick       CompilerDeclContext found_namespace_decl;
1642061da546Spatrick 
1643061da546Spatrick       lldb::ModuleSP module_sp = i->first;
1644061da546Spatrick       CompilerDeclContext module_parent_namespace_decl = i->second;
1645061da546Spatrick 
1646061da546Spatrick       SymbolFile *symbol_file = module_sp->GetSymbolFile();
1647061da546Spatrick 
1648061da546Spatrick       if (!symbol_file)
1649061da546Spatrick         continue;
1650061da546Spatrick 
1651061da546Spatrick       found_namespace_decl =
1652dda28197Spatrick           symbol_file->FindNamespace(name, module_parent_namespace_decl);
1653061da546Spatrick 
1654061da546Spatrick       if (!found_namespace_decl)
1655061da546Spatrick         continue;
1656061da546Spatrick 
1657061da546Spatrick       namespace_map->push_back(std::pair<lldb::ModuleSP, CompilerDeclContext>(
1658061da546Spatrick           module_sp, found_namespace_decl));
1659061da546Spatrick 
1660dda28197Spatrick       LLDB_LOG(log, "  CMN Found namespace {0} in module {1}", name,
1661dda28197Spatrick                module_sp->GetFileSpec().GetFilename());
1662061da546Spatrick     }
1663061da546Spatrick   } else {
1664061da546Spatrick     CompilerDeclContext null_namespace_decl;
1665be691f3bSpatrick     for (lldb::ModuleSP image : m_target->GetImages().Modules()) {
1666061da546Spatrick       if (!image)
1667061da546Spatrick         continue;
1668061da546Spatrick 
1669061da546Spatrick       CompilerDeclContext found_namespace_decl;
1670061da546Spatrick 
1671061da546Spatrick       SymbolFile *symbol_file = image->GetSymbolFile();
1672061da546Spatrick 
1673061da546Spatrick       if (!symbol_file)
1674061da546Spatrick         continue;
1675061da546Spatrick 
1676061da546Spatrick       found_namespace_decl =
1677dda28197Spatrick           symbol_file->FindNamespace(name, null_namespace_decl);
1678061da546Spatrick 
1679061da546Spatrick       if (!found_namespace_decl)
1680061da546Spatrick         continue;
1681061da546Spatrick 
1682061da546Spatrick       namespace_map->push_back(std::pair<lldb::ModuleSP, CompilerDeclContext>(
1683061da546Spatrick           image, found_namespace_decl));
1684061da546Spatrick 
1685dda28197Spatrick       LLDB_LOG(log, "  CMN[{0}] Found namespace {0} in module {1}", name,
1686dda28197Spatrick                image->GetFileSpec().GetFilename());
1687061da546Spatrick     }
1688061da546Spatrick   }
1689061da546Spatrick }
1690061da546Spatrick 
AddNamespace(NameSearchContext & context,ClangASTImporter::NamespaceMapSP & namespace_decls)1691061da546Spatrick NamespaceDecl *ClangASTSource::AddNamespace(
1692061da546Spatrick     NameSearchContext &context,
1693061da546Spatrick     ClangASTImporter::NamespaceMapSP &namespace_decls) {
1694061da546Spatrick   if (!namespace_decls)
1695061da546Spatrick     return nullptr;
1696061da546Spatrick 
1697061da546Spatrick   const CompilerDeclContext &namespace_decl = namespace_decls->begin()->second;
1698061da546Spatrick 
1699061da546Spatrick   clang::ASTContext *src_ast =
1700dda28197Spatrick       TypeSystemClang::DeclContextGetTypeSystemClang(namespace_decl);
1701061da546Spatrick   if (!src_ast)
1702061da546Spatrick     return nullptr;
1703061da546Spatrick   clang::NamespaceDecl *src_namespace_decl =
1704dda28197Spatrick       TypeSystemClang::DeclContextGetAsNamespaceDecl(namespace_decl);
1705061da546Spatrick 
1706061da546Spatrick   if (!src_namespace_decl)
1707061da546Spatrick     return nullptr;
1708061da546Spatrick 
1709061da546Spatrick   Decl *copied_decl = CopyDecl(src_namespace_decl);
1710061da546Spatrick 
1711061da546Spatrick   if (!copied_decl)
1712061da546Spatrick     return nullptr;
1713061da546Spatrick 
1714061da546Spatrick   NamespaceDecl *copied_namespace_decl = dyn_cast<NamespaceDecl>(copied_decl);
1715061da546Spatrick 
1716061da546Spatrick   if (!copied_namespace_decl)
1717061da546Spatrick     return nullptr;
1718061da546Spatrick 
1719061da546Spatrick   context.m_decls.push_back(copied_namespace_decl);
1720061da546Spatrick 
1721061da546Spatrick   m_ast_importer_sp->RegisterNamespaceMap(copied_namespace_decl,
1722061da546Spatrick                                           namespace_decls);
1723061da546Spatrick 
1724061da546Spatrick   return dyn_cast<NamespaceDecl>(copied_decl);
1725061da546Spatrick }
1726061da546Spatrick 
CopyDecl(Decl * src_decl)1727061da546Spatrick clang::Decl *ClangASTSource::CopyDecl(Decl *src_decl) {
1728061da546Spatrick   return m_ast_importer_sp->CopyDecl(m_ast_context, src_decl);
1729061da546Spatrick }
1730061da546Spatrick 
GetDeclOrigin(const clang::Decl * decl)1731061da546Spatrick ClangASTImporter::DeclOrigin ClangASTSource::GetDeclOrigin(const clang::Decl *decl) {
1732061da546Spatrick   return m_ast_importer_sp->GetDeclOrigin(decl);
1733061da546Spatrick }
1734061da546Spatrick 
GuardedCopyType(const CompilerType & src_type)1735061da546Spatrick CompilerType ClangASTSource::GuardedCopyType(const CompilerType &src_type) {
1736*f6aab3d8Srobert   auto ts = src_type.GetTypeSystem();
1737*f6aab3d8Srobert   auto src_ast = ts.dyn_cast_or_null<TypeSystemClang>();
1738*f6aab3d8Srobert   if (!src_ast)
1739*f6aab3d8Srobert     return {};
1740061da546Spatrick 
1741dda28197Spatrick   QualType copied_qual_type = ClangUtil::GetQualType(
1742061da546Spatrick       m_ast_importer_sp->CopyType(*m_clang_ast_context, src_type));
1743061da546Spatrick 
1744061da546Spatrick   if (copied_qual_type.getAsOpaquePtr() &&
1745061da546Spatrick       copied_qual_type->getCanonicalTypeInternal().isNull())
1746061da546Spatrick     // this shouldn't happen, but we're hardening because the AST importer
1747061da546Spatrick     // seems to be generating bad types on occasion.
1748*f6aab3d8Srobert     return {};
1749061da546Spatrick 
1750061da546Spatrick   return m_clang_ast_context->GetType(copied_qual_type);
1751061da546Spatrick }
1752be691f3bSpatrick 
1753be691f3bSpatrick std::shared_ptr<ClangModulesDeclVendor>
GetClangModulesDeclVendor()1754be691f3bSpatrick ClangASTSource::GetClangModulesDeclVendor() {
1755be691f3bSpatrick   auto persistent_vars = llvm::cast<ClangPersistentVariables>(
1756be691f3bSpatrick       m_target->GetPersistentExpressionStateForLanguage(lldb::eLanguageTypeC));
1757be691f3bSpatrick   return persistent_vars->GetClangModulesDeclVendor();
1758be691f3bSpatrick }
1759