xref: /openbsd-src/gnu/llvm/lldb/source/Symbol/Symtab.cpp (revision f6aab3d83b51b91c24247ad2c2573574de475a82)
1dda28197Spatrick //===-- Symtab.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 <map>
10061da546Spatrick #include <set>
11061da546Spatrick 
12*f6aab3d8Srobert #include "lldb/Core/DataFileCache.h"
13061da546Spatrick #include "lldb/Core/Module.h"
14061da546Spatrick #include "lldb/Core/RichManglingContext.h"
15061da546Spatrick #include "lldb/Core/Section.h"
16061da546Spatrick #include "lldb/Symbol/ObjectFile.h"
17061da546Spatrick #include "lldb/Symbol/Symbol.h"
18061da546Spatrick #include "lldb/Symbol/SymbolContext.h"
19061da546Spatrick #include "lldb/Symbol/Symtab.h"
20be691f3bSpatrick #include "lldb/Target/Language.h"
21*f6aab3d8Srobert #include "lldb/Utility/DataEncoder.h"
22*f6aab3d8Srobert #include "lldb/Utility/Endian.h"
23061da546Spatrick #include "lldb/Utility/RegularExpression.h"
24061da546Spatrick #include "lldb/Utility/Stream.h"
25061da546Spatrick #include "lldb/Utility/Timer.h"
26061da546Spatrick 
27*f6aab3d8Srobert #include "llvm/ADT/ArrayRef.h"
28061da546Spatrick #include "llvm/ADT/StringRef.h"
29*f6aab3d8Srobert #include "llvm/Support/DJB.h"
30061da546Spatrick 
31061da546Spatrick using namespace lldb;
32061da546Spatrick using namespace lldb_private;
33061da546Spatrick 
Symtab(ObjectFile * objfile)34061da546Spatrick Symtab::Symtab(ObjectFile *objfile)
35061da546Spatrick     : m_objfile(objfile), m_symbols(), m_file_addr_to_index(*this),
36be691f3bSpatrick       m_name_to_symbol_indices(), m_mutex(),
37*f6aab3d8Srobert       m_file_addr_to_index_computed(false), m_name_indexes_computed(false),
38*f6aab3d8Srobert       m_loaded_from_cache(false), m_saved_to_cache(false) {
39be691f3bSpatrick   m_name_to_symbol_indices.emplace(std::make_pair(
40be691f3bSpatrick       lldb::eFunctionNameTypeNone, UniqueCStringMap<uint32_t>()));
41be691f3bSpatrick   m_name_to_symbol_indices.emplace(std::make_pair(
42be691f3bSpatrick       lldb::eFunctionNameTypeBase, UniqueCStringMap<uint32_t>()));
43be691f3bSpatrick   m_name_to_symbol_indices.emplace(std::make_pair(
44be691f3bSpatrick       lldb::eFunctionNameTypeMethod, UniqueCStringMap<uint32_t>()));
45be691f3bSpatrick   m_name_to_symbol_indices.emplace(std::make_pair(
46be691f3bSpatrick       lldb::eFunctionNameTypeSelector, UniqueCStringMap<uint32_t>()));
47be691f3bSpatrick }
48061da546Spatrick 
49be691f3bSpatrick Symtab::~Symtab() = default;
50061da546Spatrick 
Reserve(size_t count)51061da546Spatrick void Symtab::Reserve(size_t count) {
52061da546Spatrick   // Clients should grab the mutex from this symbol table and lock it manually
53061da546Spatrick   // when calling this function to avoid performance issues.
54061da546Spatrick   m_symbols.reserve(count);
55061da546Spatrick }
56061da546Spatrick 
Resize(size_t count)57061da546Spatrick Symbol *Symtab::Resize(size_t count) {
58061da546Spatrick   // Clients should grab the mutex from this symbol table and lock it manually
59061da546Spatrick   // when calling this function to avoid performance issues.
60061da546Spatrick   m_symbols.resize(count);
61061da546Spatrick   return m_symbols.empty() ? nullptr : &m_symbols[0];
62061da546Spatrick }
63061da546Spatrick 
AddSymbol(const Symbol & symbol)64061da546Spatrick uint32_t Symtab::AddSymbol(const Symbol &symbol) {
65061da546Spatrick   // Clients should grab the mutex from this symbol table and lock it manually
66061da546Spatrick   // when calling this function to avoid performance issues.
67061da546Spatrick   uint32_t symbol_idx = m_symbols.size();
68be691f3bSpatrick   auto &name_to_index = GetNameToSymbolIndexMap(lldb::eFunctionNameTypeNone);
69be691f3bSpatrick   name_to_index.Clear();
70061da546Spatrick   m_file_addr_to_index.Clear();
71061da546Spatrick   m_symbols.push_back(symbol);
72061da546Spatrick   m_file_addr_to_index_computed = false;
73061da546Spatrick   m_name_indexes_computed = false;
74061da546Spatrick   return symbol_idx;
75061da546Spatrick }
76061da546Spatrick 
GetNumSymbols() const77061da546Spatrick size_t Symtab::GetNumSymbols() const {
78061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
79061da546Spatrick   return m_symbols.size();
80061da546Spatrick }
81061da546Spatrick 
SectionFileAddressesChanged()82061da546Spatrick void Symtab::SectionFileAddressesChanged() {
83be691f3bSpatrick   auto &name_to_index = GetNameToSymbolIndexMap(lldb::eFunctionNameTypeNone);
84be691f3bSpatrick   name_to_index.Clear();
85061da546Spatrick   m_file_addr_to_index_computed = false;
86061da546Spatrick }
87061da546Spatrick 
Dump(Stream * s,Target * target,SortOrder sort_order,Mangled::NamePreference name_preference)88061da546Spatrick void Symtab::Dump(Stream *s, Target *target, SortOrder sort_order,
89061da546Spatrick                   Mangled::NamePreference name_preference) {
90061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
91061da546Spatrick 
92061da546Spatrick   //    s->Printf("%.*p: ", (int)sizeof(void*) * 2, this);
93061da546Spatrick   s->Indent();
94061da546Spatrick   const FileSpec &file_spec = m_objfile->GetFileSpec();
95061da546Spatrick   const char *object_name = nullptr;
96061da546Spatrick   if (m_objfile->GetModule())
97061da546Spatrick     object_name = m_objfile->GetModule()->GetObjectName().GetCString();
98061da546Spatrick 
99061da546Spatrick   if (file_spec)
100061da546Spatrick     s->Printf("Symtab, file = %s%s%s%s, num_symbols = %" PRIu64,
101061da546Spatrick               file_spec.GetPath().c_str(), object_name ? "(" : "",
102061da546Spatrick               object_name ? object_name : "", object_name ? ")" : "",
103061da546Spatrick               (uint64_t)m_symbols.size());
104061da546Spatrick   else
105061da546Spatrick     s->Printf("Symtab, num_symbols = %" PRIu64 "", (uint64_t)m_symbols.size());
106061da546Spatrick 
107061da546Spatrick   if (!m_symbols.empty()) {
108061da546Spatrick     switch (sort_order) {
109061da546Spatrick     case eSortOrderNone: {
110061da546Spatrick       s->PutCString(":\n");
111061da546Spatrick       DumpSymbolHeader(s);
112061da546Spatrick       const_iterator begin = m_symbols.begin();
113061da546Spatrick       const_iterator end = m_symbols.end();
114061da546Spatrick       for (const_iterator pos = m_symbols.begin(); pos != end; ++pos) {
115061da546Spatrick         s->Indent();
116061da546Spatrick         pos->Dump(s, target, std::distance(begin, pos), name_preference);
117061da546Spatrick       }
118*f6aab3d8Srobert     }
119*f6aab3d8Srobert     break;
120061da546Spatrick 
121061da546Spatrick     case eSortOrderByName: {
122061da546Spatrick       // Although we maintain a lookup by exact name map, the table isn't
123061da546Spatrick       // sorted by name. So we must make the ordered symbol list up ourselves.
124061da546Spatrick       s->PutCString(" (sorted by name):\n");
125061da546Spatrick       DumpSymbolHeader(s);
126061da546Spatrick 
127061da546Spatrick       std::multimap<llvm::StringRef, const Symbol *> name_map;
128061da546Spatrick       for (const_iterator pos = m_symbols.begin(), end = m_symbols.end();
129061da546Spatrick            pos != end; ++pos) {
130061da546Spatrick         const char *name = pos->GetName().AsCString();
131061da546Spatrick         if (name && name[0])
132061da546Spatrick           name_map.insert(std::make_pair(name, &(*pos)));
133061da546Spatrick       }
134061da546Spatrick 
135061da546Spatrick       for (const auto &name_to_symbol : name_map) {
136061da546Spatrick         const Symbol *symbol = name_to_symbol.second;
137061da546Spatrick         s->Indent();
138061da546Spatrick         symbol->Dump(s, target, symbol - &m_symbols[0], name_preference);
139061da546Spatrick       }
140061da546Spatrick     } break;
141061da546Spatrick 
142061da546Spatrick     case eSortOrderByAddress:
143061da546Spatrick       s->PutCString(" (sorted by address):\n");
144061da546Spatrick       DumpSymbolHeader(s);
145061da546Spatrick       if (!m_file_addr_to_index_computed)
146061da546Spatrick         InitAddressIndexes();
147061da546Spatrick       const size_t num_entries = m_file_addr_to_index.GetSize();
148061da546Spatrick       for (size_t i = 0; i < num_entries; ++i) {
149061da546Spatrick         s->Indent();
150061da546Spatrick         const uint32_t symbol_idx = m_file_addr_to_index.GetEntryRef(i).data;
151061da546Spatrick         m_symbols[symbol_idx].Dump(s, target, symbol_idx, name_preference);
152061da546Spatrick       }
153061da546Spatrick       break;
154061da546Spatrick     }
155061da546Spatrick   } else {
156061da546Spatrick     s->PutCString("\n");
157061da546Spatrick   }
158061da546Spatrick }
159061da546Spatrick 
Dump(Stream * s,Target * target,std::vector<uint32_t> & indexes,Mangled::NamePreference name_preference) const160061da546Spatrick void Symtab::Dump(Stream *s, Target *target, std::vector<uint32_t> &indexes,
161061da546Spatrick                   Mangled::NamePreference name_preference) const {
162061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
163061da546Spatrick 
164061da546Spatrick   const size_t num_symbols = GetNumSymbols();
165061da546Spatrick   // s->Printf("%.*p: ", (int)sizeof(void*) * 2, this);
166061da546Spatrick   s->Indent();
167061da546Spatrick   s->Printf("Symtab %" PRIu64 " symbol indexes (%" PRIu64 " symbols total):\n",
168061da546Spatrick             (uint64_t)indexes.size(), (uint64_t)m_symbols.size());
169061da546Spatrick   s->IndentMore();
170061da546Spatrick 
171061da546Spatrick   if (!indexes.empty()) {
172061da546Spatrick     std::vector<uint32_t>::const_iterator pos;
173061da546Spatrick     std::vector<uint32_t>::const_iterator end = indexes.end();
174061da546Spatrick     DumpSymbolHeader(s);
175061da546Spatrick     for (pos = indexes.begin(); pos != end; ++pos) {
176061da546Spatrick       size_t idx = *pos;
177061da546Spatrick       if (idx < num_symbols) {
178061da546Spatrick         s->Indent();
179061da546Spatrick         m_symbols[idx].Dump(s, target, idx, name_preference);
180061da546Spatrick       }
181061da546Spatrick     }
182061da546Spatrick   }
183061da546Spatrick   s->IndentLess();
184061da546Spatrick }
185061da546Spatrick 
DumpSymbolHeader(Stream * s)186061da546Spatrick void Symtab::DumpSymbolHeader(Stream *s) {
187061da546Spatrick   s->Indent("               Debug symbol\n");
188061da546Spatrick   s->Indent("               |Synthetic symbol\n");
189061da546Spatrick   s->Indent("               ||Externally Visible\n");
190061da546Spatrick   s->Indent("               |||\n");
191061da546Spatrick   s->Indent("Index   UserID DSX Type            File Address/Value Load "
192061da546Spatrick             "Address       Size               Flags      Name\n");
193061da546Spatrick   s->Indent("------- ------ --- --------------- ------------------ "
194061da546Spatrick             "------------------ ------------------ ---------- "
195061da546Spatrick             "----------------------------------\n");
196061da546Spatrick }
197061da546Spatrick 
CompareSymbolID(const void * key,const void * p)198061da546Spatrick static int CompareSymbolID(const void *key, const void *p) {
199061da546Spatrick   const user_id_t match_uid = *(const user_id_t *)key;
200061da546Spatrick   const user_id_t symbol_uid = ((const Symbol *)p)->GetID();
201061da546Spatrick   if (match_uid < symbol_uid)
202061da546Spatrick     return -1;
203061da546Spatrick   if (match_uid > symbol_uid)
204061da546Spatrick     return 1;
205061da546Spatrick   return 0;
206061da546Spatrick }
207061da546Spatrick 
FindSymbolByID(lldb::user_id_t symbol_uid) const208061da546Spatrick Symbol *Symtab::FindSymbolByID(lldb::user_id_t symbol_uid) const {
209061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
210061da546Spatrick 
211061da546Spatrick   Symbol *symbol =
212061da546Spatrick       (Symbol *)::bsearch(&symbol_uid, &m_symbols[0], m_symbols.size(),
213061da546Spatrick                           sizeof(m_symbols[0]), CompareSymbolID);
214061da546Spatrick   return symbol;
215061da546Spatrick }
216061da546Spatrick 
SymbolAtIndex(size_t idx)217061da546Spatrick Symbol *Symtab::SymbolAtIndex(size_t idx) {
218061da546Spatrick   // Clients should grab the mutex from this symbol table and lock it manually
219061da546Spatrick   // when calling this function to avoid performance issues.
220061da546Spatrick   if (idx < m_symbols.size())
221061da546Spatrick     return &m_symbols[idx];
222061da546Spatrick   return nullptr;
223061da546Spatrick }
224061da546Spatrick 
SymbolAtIndex(size_t idx) const225061da546Spatrick const Symbol *Symtab::SymbolAtIndex(size_t idx) const {
226061da546Spatrick   // Clients should grab the mutex from this symbol table and lock it manually
227061da546Spatrick   // when calling this function to avoid performance issues.
228061da546Spatrick   if (idx < m_symbols.size())
229061da546Spatrick     return &m_symbols[idx];
230061da546Spatrick   return nullptr;
231061da546Spatrick }
232061da546Spatrick 
lldb_skip_name(llvm::StringRef mangled,Mangled::ManglingScheme scheme)233061da546Spatrick static bool lldb_skip_name(llvm::StringRef mangled,
234061da546Spatrick                            Mangled::ManglingScheme scheme) {
235061da546Spatrick   switch (scheme) {
236061da546Spatrick   case Mangled::eManglingSchemeItanium: {
237061da546Spatrick     if (mangled.size() < 3 || !mangled.startswith("_Z"))
238061da546Spatrick       return true;
239061da546Spatrick 
240061da546Spatrick     // Avoid the following types of symbols in the index.
241061da546Spatrick     switch (mangled[2]) {
242061da546Spatrick     case 'G': // guard variables
243061da546Spatrick     case 'T': // virtual tables, VTT structures, typeinfo structures + names
244061da546Spatrick     case 'Z': // named local entities (if we eventually handle
245061da546Spatrick               // eSymbolTypeData, we will want this back)
246061da546Spatrick       return true;
247061da546Spatrick 
248061da546Spatrick     default:
249061da546Spatrick       break;
250061da546Spatrick     }
251061da546Spatrick 
252061da546Spatrick     // Include this name in the index.
253061da546Spatrick     return false;
254061da546Spatrick   }
255061da546Spatrick 
256061da546Spatrick   // No filters for this scheme yet. Include all names in indexing.
257061da546Spatrick   case Mangled::eManglingSchemeMSVC:
258be691f3bSpatrick   case Mangled::eManglingSchemeRustV0:
259*f6aab3d8Srobert   case Mangled::eManglingSchemeD:
260be691f3bSpatrick     return false;
261be691f3bSpatrick 
262061da546Spatrick   // Don't try and demangle things we can't categorize.
263061da546Spatrick   case Mangled::eManglingSchemeNone:
264061da546Spatrick     return true;
265061da546Spatrick   }
266061da546Spatrick   llvm_unreachable("unknown scheme!");
267061da546Spatrick }
268061da546Spatrick 
InitNameIndexes()269061da546Spatrick void Symtab::InitNameIndexes() {
270061da546Spatrick   // Protected function, no need to lock mutex...
271061da546Spatrick   if (!m_name_indexes_computed) {
272061da546Spatrick     m_name_indexes_computed = true;
273*f6aab3d8Srobert     ElapsedTime elapsed(m_objfile->GetModule()->GetSymtabIndexTime());
274be691f3bSpatrick     LLDB_SCOPED_TIMER();
275be691f3bSpatrick 
276be691f3bSpatrick     // Collect all loaded language plugins.
277be691f3bSpatrick     std::vector<Language *> languages;
278be691f3bSpatrick     Language::ForEach([&languages](Language *l) {
279be691f3bSpatrick       languages.push_back(l);
280be691f3bSpatrick       return true;
281be691f3bSpatrick     });
282be691f3bSpatrick 
283be691f3bSpatrick     auto &name_to_index = GetNameToSymbolIndexMap(lldb::eFunctionNameTypeNone);
284be691f3bSpatrick     auto &basename_to_index =
285be691f3bSpatrick         GetNameToSymbolIndexMap(lldb::eFunctionNameTypeBase);
286be691f3bSpatrick     auto &method_to_index =
287be691f3bSpatrick         GetNameToSymbolIndexMap(lldb::eFunctionNameTypeMethod);
288be691f3bSpatrick     auto &selector_to_index =
289be691f3bSpatrick         GetNameToSymbolIndexMap(lldb::eFunctionNameTypeSelector);
290061da546Spatrick     // Create the name index vector to be able to quickly search by name
291061da546Spatrick     const size_t num_symbols = m_symbols.size();
292be691f3bSpatrick     name_to_index.Reserve(num_symbols);
293061da546Spatrick 
294061da546Spatrick     // The "const char *" in "class_contexts" and backlog::value_type::second
295061da546Spatrick     // must come from a ConstString::GetCString()
296061da546Spatrick     std::set<const char *> class_contexts;
297061da546Spatrick     std::vector<std::pair<NameToIndexMap::Entry, const char *>> backlog;
298061da546Spatrick     backlog.reserve(num_symbols / 2);
299061da546Spatrick 
300061da546Spatrick     // Instantiation of the demangler is expensive, so better use a single one
301061da546Spatrick     // for all entries during batch processing.
302061da546Spatrick     RichManglingContext rmc;
303061da546Spatrick     for (uint32_t value = 0; value < num_symbols; ++value) {
304061da546Spatrick       Symbol *symbol = &m_symbols[value];
305061da546Spatrick 
306061da546Spatrick       // Don't let trampolines get into the lookup by name map If we ever need
307061da546Spatrick       // the trampoline symbols to be searchable by name we can remove this and
308061da546Spatrick       // then possibly add a new bool to any of the Symtab functions that
309be691f3bSpatrick       // lookup symbols by name to indicate if they want trampolines. We also
310be691f3bSpatrick       // don't want any synthetic symbols with auto generated names in the
311be691f3bSpatrick       // name lookups.
312be691f3bSpatrick       if (symbol->IsTrampoline() || symbol->IsSyntheticWithAutoGeneratedName())
313061da546Spatrick         continue;
314061da546Spatrick 
315061da546Spatrick       // If the symbol's name string matched a Mangled::ManglingScheme, it is
316061da546Spatrick       // stored in the mangled field.
317061da546Spatrick       Mangled &mangled = symbol->GetMangled();
318061da546Spatrick       if (ConstString name = mangled.GetMangledName()) {
319be691f3bSpatrick         name_to_index.Append(name, value);
320061da546Spatrick 
321061da546Spatrick         if (symbol->ContainsLinkerAnnotations()) {
322061da546Spatrick           // If the symbol has linker annotations, also add the version without
323061da546Spatrick           // the annotations.
324061da546Spatrick           ConstString stripped = ConstString(
325061da546Spatrick               m_objfile->StripLinkerSymbolAnnotations(name.GetStringRef()));
326be691f3bSpatrick           name_to_index.Append(stripped, value);
327061da546Spatrick         }
328061da546Spatrick 
329061da546Spatrick         const SymbolType type = symbol->GetType();
330061da546Spatrick         if (type == eSymbolTypeCode || type == eSymbolTypeResolver) {
331*f6aab3d8Srobert           if (mangled.GetRichManglingInfo(rmc, lldb_skip_name)) {
332061da546Spatrick             RegisterMangledNameEntry(value, class_contexts, backlog, rmc);
333*f6aab3d8Srobert             continue;
334*f6aab3d8Srobert           }
335061da546Spatrick         }
336061da546Spatrick       }
337061da546Spatrick 
338061da546Spatrick       // Symbol name strings that didn't match a Mangled::ManglingScheme, are
339061da546Spatrick       // stored in the demangled field.
340dda28197Spatrick       if (ConstString name = mangled.GetDemangledName()) {
341be691f3bSpatrick         name_to_index.Append(name, value);
342061da546Spatrick 
343061da546Spatrick         if (symbol->ContainsLinkerAnnotations()) {
344061da546Spatrick           // If the symbol has linker annotations, also add the version without
345061da546Spatrick           // the annotations.
346061da546Spatrick           name = ConstString(
347061da546Spatrick               m_objfile->StripLinkerSymbolAnnotations(name.GetStringRef()));
348be691f3bSpatrick           name_to_index.Append(name, value);
349061da546Spatrick         }
350061da546Spatrick 
351061da546Spatrick         // If the demangled name turns out to be an ObjC name, and is a category
352061da546Spatrick         // name, add the version without categories to the index too.
353be691f3bSpatrick         for (Language *lang : languages) {
354be691f3bSpatrick           for (auto variant : lang->GetMethodNameVariants(name)) {
355be691f3bSpatrick             if (variant.GetType() & lldb::eFunctionNameTypeSelector)
356be691f3bSpatrick               selector_to_index.Append(variant.GetName(), value);
357be691f3bSpatrick             else if (variant.GetType() & lldb::eFunctionNameTypeFull)
358be691f3bSpatrick               name_to_index.Append(variant.GetName(), value);
359be691f3bSpatrick             else if (variant.GetType() & lldb::eFunctionNameTypeMethod)
360be691f3bSpatrick               method_to_index.Append(variant.GetName(), value);
361be691f3bSpatrick             else if (variant.GetType() & lldb::eFunctionNameTypeBase)
362be691f3bSpatrick               basename_to_index.Append(variant.GetName(), value);
363be691f3bSpatrick           }
364061da546Spatrick         }
365061da546Spatrick       }
366061da546Spatrick     }
367061da546Spatrick 
368061da546Spatrick     for (const auto &record : backlog) {
369061da546Spatrick       RegisterBacklogEntry(record.first, record.second, class_contexts);
370061da546Spatrick     }
371061da546Spatrick 
372be691f3bSpatrick     name_to_index.Sort();
373be691f3bSpatrick     name_to_index.SizeToFit();
374be691f3bSpatrick     selector_to_index.Sort();
375be691f3bSpatrick     selector_to_index.SizeToFit();
376be691f3bSpatrick     basename_to_index.Sort();
377be691f3bSpatrick     basename_to_index.SizeToFit();
378be691f3bSpatrick     method_to_index.Sort();
379be691f3bSpatrick     method_to_index.SizeToFit();
380061da546Spatrick   }
381061da546Spatrick }
382061da546Spatrick 
RegisterMangledNameEntry(uint32_t value,std::set<const char * > & class_contexts,std::vector<std::pair<NameToIndexMap::Entry,const char * >> & backlog,RichManglingContext & rmc)383061da546Spatrick void Symtab::RegisterMangledNameEntry(
384061da546Spatrick     uint32_t value, std::set<const char *> &class_contexts,
385061da546Spatrick     std::vector<std::pair<NameToIndexMap::Entry, const char *>> &backlog,
386061da546Spatrick     RichManglingContext &rmc) {
387061da546Spatrick   // Only register functions that have a base name.
388*f6aab3d8Srobert   llvm::StringRef base_name = rmc.ParseFunctionBaseName();
389061da546Spatrick   if (base_name.empty())
390061da546Spatrick     return;
391061da546Spatrick 
392061da546Spatrick   // The base name will be our entry's name.
393061da546Spatrick   NameToIndexMap::Entry entry(ConstString(base_name), value);
394*f6aab3d8Srobert   llvm::StringRef decl_context = rmc.ParseFunctionDeclContextName();
395061da546Spatrick 
396061da546Spatrick   // Register functions with no context.
397061da546Spatrick   if (decl_context.empty()) {
398061da546Spatrick     // This has to be a basename
399be691f3bSpatrick     auto &basename_to_index =
400be691f3bSpatrick         GetNameToSymbolIndexMap(lldb::eFunctionNameTypeBase);
401be691f3bSpatrick     basename_to_index.Append(entry);
402061da546Spatrick     // If there is no context (no namespaces or class scopes that come before
403061da546Spatrick     // the function name) then this also could be a fullname.
404be691f3bSpatrick     auto &name_to_index = GetNameToSymbolIndexMap(lldb::eFunctionNameTypeNone);
405be691f3bSpatrick     name_to_index.Append(entry);
406061da546Spatrick     return;
407061da546Spatrick   }
408061da546Spatrick 
409061da546Spatrick   // Make sure we have a pool-string pointer and see if we already know the
410061da546Spatrick   // context name.
411061da546Spatrick   const char *decl_context_ccstr = ConstString(decl_context).GetCString();
412061da546Spatrick   auto it = class_contexts.find(decl_context_ccstr);
413061da546Spatrick 
414be691f3bSpatrick   auto &method_to_index =
415be691f3bSpatrick       GetNameToSymbolIndexMap(lldb::eFunctionNameTypeMethod);
416061da546Spatrick   // Register constructors and destructors. They are methods and create
417061da546Spatrick   // declaration contexts.
418061da546Spatrick   if (rmc.IsCtorOrDtor()) {
419be691f3bSpatrick     method_to_index.Append(entry);
420061da546Spatrick     if (it == class_contexts.end())
421061da546Spatrick       class_contexts.insert(it, decl_context_ccstr);
422061da546Spatrick     return;
423061da546Spatrick   }
424061da546Spatrick 
425061da546Spatrick   // Register regular methods with a known declaration context.
426061da546Spatrick   if (it != class_contexts.end()) {
427be691f3bSpatrick     method_to_index.Append(entry);
428061da546Spatrick     return;
429061da546Spatrick   }
430061da546Spatrick 
431061da546Spatrick   // Regular methods in unknown declaration contexts are put to the backlog. We
432061da546Spatrick   // will revisit them once we processed all remaining symbols.
433061da546Spatrick   backlog.push_back(std::make_pair(entry, decl_context_ccstr));
434061da546Spatrick }
435061da546Spatrick 
RegisterBacklogEntry(const NameToIndexMap::Entry & entry,const char * decl_context,const std::set<const char * > & class_contexts)436061da546Spatrick void Symtab::RegisterBacklogEntry(
437061da546Spatrick     const NameToIndexMap::Entry &entry, const char *decl_context,
438061da546Spatrick     const std::set<const char *> &class_contexts) {
439be691f3bSpatrick   auto &method_to_index =
440be691f3bSpatrick       GetNameToSymbolIndexMap(lldb::eFunctionNameTypeMethod);
441061da546Spatrick   auto it = class_contexts.find(decl_context);
442061da546Spatrick   if (it != class_contexts.end()) {
443be691f3bSpatrick     method_to_index.Append(entry);
444061da546Spatrick   } else {
445061da546Spatrick     // If we got here, we have something that had a context (was inside
446061da546Spatrick     // a namespace or class) yet we don't know the entry
447be691f3bSpatrick     method_to_index.Append(entry);
448be691f3bSpatrick     auto &basename_to_index =
449be691f3bSpatrick         GetNameToSymbolIndexMap(lldb::eFunctionNameTypeBase);
450be691f3bSpatrick     basename_to_index.Append(entry);
451061da546Spatrick   }
452061da546Spatrick }
453061da546Spatrick 
PreloadSymbols()454061da546Spatrick void Symtab::PreloadSymbols() {
455061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
456061da546Spatrick   InitNameIndexes();
457061da546Spatrick }
458061da546Spatrick 
AppendSymbolNamesToMap(const IndexCollection & indexes,bool add_demangled,bool add_mangled,NameToIndexMap & name_to_index_map) const459061da546Spatrick void Symtab::AppendSymbolNamesToMap(const IndexCollection &indexes,
460061da546Spatrick                                     bool add_demangled, bool add_mangled,
461061da546Spatrick                                     NameToIndexMap &name_to_index_map) const {
462be691f3bSpatrick   LLDB_SCOPED_TIMER();
463061da546Spatrick   if (add_demangled || add_mangled) {
464061da546Spatrick     std::lock_guard<std::recursive_mutex> guard(m_mutex);
465061da546Spatrick 
466061da546Spatrick     // Create the name index vector to be able to quickly search by name
467061da546Spatrick     const size_t num_indexes = indexes.size();
468061da546Spatrick     for (size_t i = 0; i < num_indexes; ++i) {
469061da546Spatrick       uint32_t value = indexes[i];
470061da546Spatrick       assert(i < m_symbols.size());
471061da546Spatrick       const Symbol *symbol = &m_symbols[value];
472061da546Spatrick 
473061da546Spatrick       const Mangled &mangled = symbol->GetMangled();
474061da546Spatrick       if (add_demangled) {
475dda28197Spatrick         if (ConstString name = mangled.GetDemangledName())
476061da546Spatrick           name_to_index_map.Append(name, value);
477061da546Spatrick       }
478061da546Spatrick 
479061da546Spatrick       if (add_mangled) {
480061da546Spatrick         if (ConstString name = mangled.GetMangledName())
481061da546Spatrick           name_to_index_map.Append(name, value);
482061da546Spatrick       }
483061da546Spatrick     }
484061da546Spatrick   }
485061da546Spatrick }
486061da546Spatrick 
AppendSymbolIndexesWithType(SymbolType symbol_type,std::vector<uint32_t> & indexes,uint32_t start_idx,uint32_t end_index) const487061da546Spatrick uint32_t Symtab::AppendSymbolIndexesWithType(SymbolType symbol_type,
488061da546Spatrick                                              std::vector<uint32_t> &indexes,
489061da546Spatrick                                              uint32_t start_idx,
490061da546Spatrick                                              uint32_t end_index) const {
491061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
492061da546Spatrick 
493061da546Spatrick   uint32_t prev_size = indexes.size();
494061da546Spatrick 
495061da546Spatrick   const uint32_t count = std::min<uint32_t>(m_symbols.size(), end_index);
496061da546Spatrick 
497061da546Spatrick   for (uint32_t i = start_idx; i < count; ++i) {
498061da546Spatrick     if (symbol_type == eSymbolTypeAny || m_symbols[i].GetType() == symbol_type)
499061da546Spatrick       indexes.push_back(i);
500061da546Spatrick   }
501061da546Spatrick 
502061da546Spatrick   return indexes.size() - prev_size;
503061da546Spatrick }
504061da546Spatrick 
AppendSymbolIndexesWithTypeAndFlagsValue(SymbolType symbol_type,uint32_t flags_value,std::vector<uint32_t> & indexes,uint32_t start_idx,uint32_t end_index) const505061da546Spatrick uint32_t Symtab::AppendSymbolIndexesWithTypeAndFlagsValue(
506061da546Spatrick     SymbolType symbol_type, uint32_t flags_value,
507061da546Spatrick     std::vector<uint32_t> &indexes, uint32_t start_idx,
508061da546Spatrick     uint32_t end_index) const {
509061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
510061da546Spatrick 
511061da546Spatrick   uint32_t prev_size = indexes.size();
512061da546Spatrick 
513061da546Spatrick   const uint32_t count = std::min<uint32_t>(m_symbols.size(), end_index);
514061da546Spatrick 
515061da546Spatrick   for (uint32_t i = start_idx; i < count; ++i) {
516061da546Spatrick     if ((symbol_type == eSymbolTypeAny ||
517061da546Spatrick          m_symbols[i].GetType() == symbol_type) &&
518061da546Spatrick         m_symbols[i].GetFlags() == flags_value)
519061da546Spatrick       indexes.push_back(i);
520061da546Spatrick   }
521061da546Spatrick 
522061da546Spatrick   return indexes.size() - prev_size;
523061da546Spatrick }
524061da546Spatrick 
AppendSymbolIndexesWithType(SymbolType symbol_type,Debug symbol_debug_type,Visibility symbol_visibility,std::vector<uint32_t> & indexes,uint32_t start_idx,uint32_t end_index) const525061da546Spatrick uint32_t Symtab::AppendSymbolIndexesWithType(SymbolType symbol_type,
526061da546Spatrick                                              Debug symbol_debug_type,
527061da546Spatrick                                              Visibility symbol_visibility,
528061da546Spatrick                                              std::vector<uint32_t> &indexes,
529061da546Spatrick                                              uint32_t start_idx,
530061da546Spatrick                                              uint32_t end_index) const {
531061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
532061da546Spatrick 
533061da546Spatrick   uint32_t prev_size = indexes.size();
534061da546Spatrick 
535061da546Spatrick   const uint32_t count = std::min<uint32_t>(m_symbols.size(), end_index);
536061da546Spatrick 
537061da546Spatrick   for (uint32_t i = start_idx; i < count; ++i) {
538061da546Spatrick     if (symbol_type == eSymbolTypeAny ||
539061da546Spatrick         m_symbols[i].GetType() == symbol_type) {
540061da546Spatrick       if (CheckSymbolAtIndex(i, symbol_debug_type, symbol_visibility))
541061da546Spatrick         indexes.push_back(i);
542061da546Spatrick     }
543061da546Spatrick   }
544061da546Spatrick 
545061da546Spatrick   return indexes.size() - prev_size;
546061da546Spatrick }
547061da546Spatrick 
GetIndexForSymbol(const Symbol * symbol) const548061da546Spatrick uint32_t Symtab::GetIndexForSymbol(const Symbol *symbol) const {
549061da546Spatrick   if (!m_symbols.empty()) {
550061da546Spatrick     const Symbol *first_symbol = &m_symbols[0];
551061da546Spatrick     if (symbol >= first_symbol && symbol < first_symbol + m_symbols.size())
552061da546Spatrick       return symbol - first_symbol;
553061da546Spatrick   }
554061da546Spatrick   return UINT32_MAX;
555061da546Spatrick }
556061da546Spatrick 
557061da546Spatrick struct SymbolSortInfo {
558061da546Spatrick   const bool sort_by_load_addr;
559061da546Spatrick   const Symbol *symbols;
560061da546Spatrick };
561061da546Spatrick 
562061da546Spatrick namespace {
563061da546Spatrick struct SymbolIndexComparator {
564061da546Spatrick   const std::vector<Symbol> &symbols;
565061da546Spatrick   std::vector<lldb::addr_t> &addr_cache;
566061da546Spatrick 
567061da546Spatrick   // Getting from the symbol to the Address to the File Address involves some
568061da546Spatrick   // work. Since there are potentially many symbols here, and we're using this
569061da546Spatrick   // for sorting so we're going to be computing the address many times, cache
570061da546Spatrick   // that in addr_cache. The array passed in has to be the same size as the
571061da546Spatrick   // symbols array passed into the member variable symbols, and should be
572061da546Spatrick   // initialized with LLDB_INVALID_ADDRESS.
573061da546Spatrick   // NOTE: You have to make addr_cache externally and pass it in because
574061da546Spatrick   // std::stable_sort
575061da546Spatrick   // makes copies of the comparator it is initially passed in, and you end up
576061da546Spatrick   // spending huge amounts of time copying this array...
577061da546Spatrick 
SymbolIndexComparator__anon76e44db90211::SymbolIndexComparator578061da546Spatrick   SymbolIndexComparator(const std::vector<Symbol> &s,
579061da546Spatrick                         std::vector<lldb::addr_t> &a)
580061da546Spatrick       : symbols(s), addr_cache(a) {
581061da546Spatrick     assert(symbols.size() == addr_cache.size());
582061da546Spatrick   }
operator ()__anon76e44db90211::SymbolIndexComparator583061da546Spatrick   bool operator()(uint32_t index_a, uint32_t index_b) {
584061da546Spatrick     addr_t value_a = addr_cache[index_a];
585061da546Spatrick     if (value_a == LLDB_INVALID_ADDRESS) {
586061da546Spatrick       value_a = symbols[index_a].GetAddressRef().GetFileAddress();
587061da546Spatrick       addr_cache[index_a] = value_a;
588061da546Spatrick     }
589061da546Spatrick 
590061da546Spatrick     addr_t value_b = addr_cache[index_b];
591061da546Spatrick     if (value_b == LLDB_INVALID_ADDRESS) {
592061da546Spatrick       value_b = symbols[index_b].GetAddressRef().GetFileAddress();
593061da546Spatrick       addr_cache[index_b] = value_b;
594061da546Spatrick     }
595061da546Spatrick 
596061da546Spatrick     if (value_a == value_b) {
597061da546Spatrick       // The if the values are equal, use the original symbol user ID
598061da546Spatrick       lldb::user_id_t uid_a = symbols[index_a].GetID();
599061da546Spatrick       lldb::user_id_t uid_b = symbols[index_b].GetID();
600061da546Spatrick       if (uid_a < uid_b)
601061da546Spatrick         return true;
602061da546Spatrick       if (uid_a > uid_b)
603061da546Spatrick         return false;
604061da546Spatrick       return false;
605061da546Spatrick     } else if (value_a < value_b)
606061da546Spatrick       return true;
607061da546Spatrick 
608061da546Spatrick     return false;
609061da546Spatrick   }
610061da546Spatrick };
611061da546Spatrick }
612061da546Spatrick 
SortSymbolIndexesByValue(std::vector<uint32_t> & indexes,bool remove_duplicates) const613061da546Spatrick void Symtab::SortSymbolIndexesByValue(std::vector<uint32_t> &indexes,
614061da546Spatrick                                       bool remove_duplicates) const {
615061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
616be691f3bSpatrick   LLDB_SCOPED_TIMER();
617061da546Spatrick   // No need to sort if we have zero or one items...
618061da546Spatrick   if (indexes.size() <= 1)
619061da546Spatrick     return;
620061da546Spatrick 
621061da546Spatrick   // Sort the indexes in place using std::stable_sort.
622061da546Spatrick   // NOTE: The use of std::stable_sort instead of llvm::sort here is strictly
623061da546Spatrick   // for performance, not correctness.  The indexes vector tends to be "close"
624061da546Spatrick   // to sorted, which the stable sort handles better.
625061da546Spatrick 
626061da546Spatrick   std::vector<lldb::addr_t> addr_cache(m_symbols.size(), LLDB_INVALID_ADDRESS);
627061da546Spatrick 
628061da546Spatrick   SymbolIndexComparator comparator(m_symbols, addr_cache);
629061da546Spatrick   std::stable_sort(indexes.begin(), indexes.end(), comparator);
630061da546Spatrick 
631061da546Spatrick   // Remove any duplicates if requested
632061da546Spatrick   if (remove_duplicates) {
633061da546Spatrick     auto last = std::unique(indexes.begin(), indexes.end());
634061da546Spatrick     indexes.erase(last, indexes.end());
635061da546Spatrick   }
636061da546Spatrick }
637061da546Spatrick 
GetNameIndexes(ConstString symbol_name,std::vector<uint32_t> & indexes)638be691f3bSpatrick uint32_t Symtab::GetNameIndexes(ConstString symbol_name,
639be691f3bSpatrick                                 std::vector<uint32_t> &indexes) {
640be691f3bSpatrick   auto &name_to_index = GetNameToSymbolIndexMap(lldb::eFunctionNameTypeNone);
641be691f3bSpatrick   const uint32_t count = name_to_index.GetValues(symbol_name, indexes);
642be691f3bSpatrick   if (count)
643be691f3bSpatrick     return count;
644be691f3bSpatrick   // Synthetic symbol names are not added to the name indexes, but they start
645be691f3bSpatrick   // with a prefix and end with a the symbol UserID. This allows users to find
646be691f3bSpatrick   // these symbols without having to add them to the name indexes. These
647be691f3bSpatrick   // queries will not happen very often since the names don't mean anything, so
648be691f3bSpatrick   // performance is not paramount in this case.
649be691f3bSpatrick   llvm::StringRef name = symbol_name.GetStringRef();
650be691f3bSpatrick   // String the synthetic prefix if the name starts with it.
651be691f3bSpatrick   if (!name.consume_front(Symbol::GetSyntheticSymbolPrefix()))
652be691f3bSpatrick     return 0; // Not a synthetic symbol name
653be691f3bSpatrick 
654be691f3bSpatrick   // Extract the user ID from the symbol name
655be691f3bSpatrick   unsigned long long uid = 0;
656be691f3bSpatrick   if (getAsUnsignedInteger(name, /*Radix=*/10, uid))
657be691f3bSpatrick     return 0; // Failed to extract the user ID as an integer
658be691f3bSpatrick   Symbol *symbol = FindSymbolByID(uid);
659be691f3bSpatrick   if (symbol == nullptr)
660be691f3bSpatrick     return 0;
661be691f3bSpatrick   const uint32_t symbol_idx = GetIndexForSymbol(symbol);
662be691f3bSpatrick   if (symbol_idx == UINT32_MAX)
663be691f3bSpatrick     return 0;
664be691f3bSpatrick   indexes.push_back(symbol_idx);
665be691f3bSpatrick   return 1;
666be691f3bSpatrick }
667be691f3bSpatrick 
AppendSymbolIndexesWithName(ConstString symbol_name,std::vector<uint32_t> & indexes)668061da546Spatrick uint32_t Symtab::AppendSymbolIndexesWithName(ConstString symbol_name,
669061da546Spatrick                                              std::vector<uint32_t> &indexes) {
670061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
671061da546Spatrick 
672061da546Spatrick   if (symbol_name) {
673061da546Spatrick     if (!m_name_indexes_computed)
674061da546Spatrick       InitNameIndexes();
675061da546Spatrick 
676be691f3bSpatrick     return GetNameIndexes(symbol_name, indexes);
677061da546Spatrick   }
678061da546Spatrick   return 0;
679061da546Spatrick }
680061da546Spatrick 
AppendSymbolIndexesWithName(ConstString symbol_name,Debug symbol_debug_type,Visibility symbol_visibility,std::vector<uint32_t> & indexes)681061da546Spatrick uint32_t Symtab::AppendSymbolIndexesWithName(ConstString symbol_name,
682061da546Spatrick                                              Debug symbol_debug_type,
683061da546Spatrick                                              Visibility symbol_visibility,
684061da546Spatrick                                              std::vector<uint32_t> &indexes) {
685061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
686061da546Spatrick 
687be691f3bSpatrick   LLDB_SCOPED_TIMER();
688061da546Spatrick   if (symbol_name) {
689061da546Spatrick     const size_t old_size = indexes.size();
690061da546Spatrick     if (!m_name_indexes_computed)
691061da546Spatrick       InitNameIndexes();
692061da546Spatrick 
693061da546Spatrick     std::vector<uint32_t> all_name_indexes;
694061da546Spatrick     const size_t name_match_count =
695be691f3bSpatrick         GetNameIndexes(symbol_name, all_name_indexes);
696061da546Spatrick     for (size_t i = 0; i < name_match_count; ++i) {
697061da546Spatrick       if (CheckSymbolAtIndex(all_name_indexes[i], symbol_debug_type,
698061da546Spatrick                              symbol_visibility))
699061da546Spatrick         indexes.push_back(all_name_indexes[i]);
700061da546Spatrick     }
701061da546Spatrick     return indexes.size() - old_size;
702061da546Spatrick   }
703061da546Spatrick   return 0;
704061da546Spatrick }
705061da546Spatrick 
706061da546Spatrick uint32_t
AppendSymbolIndexesWithNameAndType(ConstString symbol_name,SymbolType symbol_type,std::vector<uint32_t> & indexes)707061da546Spatrick Symtab::AppendSymbolIndexesWithNameAndType(ConstString symbol_name,
708061da546Spatrick                                            SymbolType symbol_type,
709061da546Spatrick                                            std::vector<uint32_t> &indexes) {
710061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
711061da546Spatrick 
712061da546Spatrick   if (AppendSymbolIndexesWithName(symbol_name, indexes) > 0) {
713061da546Spatrick     std::vector<uint32_t>::iterator pos = indexes.begin();
714061da546Spatrick     while (pos != indexes.end()) {
715061da546Spatrick       if (symbol_type == eSymbolTypeAny ||
716061da546Spatrick           m_symbols[*pos].GetType() == symbol_type)
717061da546Spatrick         ++pos;
718061da546Spatrick       else
719061da546Spatrick         pos = indexes.erase(pos);
720061da546Spatrick     }
721061da546Spatrick   }
722061da546Spatrick   return indexes.size();
723061da546Spatrick }
724061da546Spatrick 
AppendSymbolIndexesWithNameAndType(ConstString symbol_name,SymbolType symbol_type,Debug symbol_debug_type,Visibility symbol_visibility,std::vector<uint32_t> & indexes)725061da546Spatrick uint32_t Symtab::AppendSymbolIndexesWithNameAndType(
726061da546Spatrick     ConstString symbol_name, SymbolType symbol_type,
727061da546Spatrick     Debug symbol_debug_type, Visibility symbol_visibility,
728061da546Spatrick     std::vector<uint32_t> &indexes) {
729061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
730061da546Spatrick 
731061da546Spatrick   if (AppendSymbolIndexesWithName(symbol_name, symbol_debug_type,
732061da546Spatrick                                   symbol_visibility, indexes) > 0) {
733061da546Spatrick     std::vector<uint32_t>::iterator pos = indexes.begin();
734061da546Spatrick     while (pos != indexes.end()) {
735061da546Spatrick       if (symbol_type == eSymbolTypeAny ||
736061da546Spatrick           m_symbols[*pos].GetType() == symbol_type)
737061da546Spatrick         ++pos;
738061da546Spatrick       else
739061da546Spatrick         pos = indexes.erase(pos);
740061da546Spatrick     }
741061da546Spatrick   }
742061da546Spatrick   return indexes.size();
743061da546Spatrick }
744061da546Spatrick 
AppendSymbolIndexesMatchingRegExAndType(const RegularExpression & regexp,SymbolType symbol_type,std::vector<uint32_t> & indexes,Mangled::NamePreference name_preference)745061da546Spatrick uint32_t Symtab::AppendSymbolIndexesMatchingRegExAndType(
746061da546Spatrick     const RegularExpression &regexp, SymbolType symbol_type,
747*f6aab3d8Srobert     std::vector<uint32_t> &indexes, Mangled::NamePreference name_preference) {
748061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
749061da546Spatrick 
750061da546Spatrick   uint32_t prev_size = indexes.size();
751061da546Spatrick   uint32_t sym_end = m_symbols.size();
752061da546Spatrick 
753061da546Spatrick   for (uint32_t i = 0; i < sym_end; i++) {
754061da546Spatrick     if (symbol_type == eSymbolTypeAny ||
755061da546Spatrick         m_symbols[i].GetType() == symbol_type) {
756*f6aab3d8Srobert       const char *name =
757*f6aab3d8Srobert           m_symbols[i].GetMangled().GetName(name_preference).AsCString();
758061da546Spatrick       if (name) {
759061da546Spatrick         if (regexp.Execute(name))
760061da546Spatrick           indexes.push_back(i);
761061da546Spatrick       }
762061da546Spatrick     }
763061da546Spatrick   }
764061da546Spatrick   return indexes.size() - prev_size;
765061da546Spatrick }
766061da546Spatrick 
AppendSymbolIndexesMatchingRegExAndType(const RegularExpression & regexp,SymbolType symbol_type,Debug symbol_debug_type,Visibility symbol_visibility,std::vector<uint32_t> & indexes,Mangled::NamePreference name_preference)767061da546Spatrick uint32_t Symtab::AppendSymbolIndexesMatchingRegExAndType(
768061da546Spatrick     const RegularExpression &regexp, SymbolType symbol_type,
769061da546Spatrick     Debug symbol_debug_type, Visibility symbol_visibility,
770*f6aab3d8Srobert     std::vector<uint32_t> &indexes, Mangled::NamePreference name_preference) {
771061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
772061da546Spatrick 
773061da546Spatrick   uint32_t prev_size = indexes.size();
774061da546Spatrick   uint32_t sym_end = m_symbols.size();
775061da546Spatrick 
776061da546Spatrick   for (uint32_t i = 0; i < sym_end; i++) {
777061da546Spatrick     if (symbol_type == eSymbolTypeAny ||
778061da546Spatrick         m_symbols[i].GetType() == symbol_type) {
779061da546Spatrick       if (!CheckSymbolAtIndex(i, symbol_debug_type, symbol_visibility))
780061da546Spatrick         continue;
781061da546Spatrick 
782*f6aab3d8Srobert       const char *name =
783*f6aab3d8Srobert           m_symbols[i].GetMangled().GetName(name_preference).AsCString();
784061da546Spatrick       if (name) {
785061da546Spatrick         if (regexp.Execute(name))
786061da546Spatrick           indexes.push_back(i);
787061da546Spatrick       }
788061da546Spatrick     }
789061da546Spatrick   }
790061da546Spatrick   return indexes.size() - prev_size;
791061da546Spatrick }
792061da546Spatrick 
FindSymbolWithType(SymbolType symbol_type,Debug symbol_debug_type,Visibility symbol_visibility,uint32_t & start_idx)793061da546Spatrick Symbol *Symtab::FindSymbolWithType(SymbolType symbol_type,
794061da546Spatrick                                    Debug symbol_debug_type,
795061da546Spatrick                                    Visibility symbol_visibility,
796061da546Spatrick                                    uint32_t &start_idx) {
797061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
798061da546Spatrick 
799061da546Spatrick   const size_t count = m_symbols.size();
800061da546Spatrick   for (size_t idx = start_idx; idx < count; ++idx) {
801061da546Spatrick     if (symbol_type == eSymbolTypeAny ||
802061da546Spatrick         m_symbols[idx].GetType() == symbol_type) {
803061da546Spatrick       if (CheckSymbolAtIndex(idx, symbol_debug_type, symbol_visibility)) {
804061da546Spatrick         start_idx = idx;
805061da546Spatrick         return &m_symbols[idx];
806061da546Spatrick       }
807061da546Spatrick     }
808061da546Spatrick   }
809061da546Spatrick   return nullptr;
810061da546Spatrick }
811061da546Spatrick 
812061da546Spatrick void
FindAllSymbolsWithNameAndType(ConstString name,SymbolType symbol_type,std::vector<uint32_t> & symbol_indexes)813061da546Spatrick Symtab::FindAllSymbolsWithNameAndType(ConstString name,
814061da546Spatrick                                       SymbolType symbol_type,
815061da546Spatrick                                       std::vector<uint32_t> &symbol_indexes) {
816061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
817061da546Spatrick 
818061da546Spatrick   // Initialize all of the lookup by name indexes before converting NAME to a
819061da546Spatrick   // uniqued string NAME_STR below.
820061da546Spatrick   if (!m_name_indexes_computed)
821061da546Spatrick     InitNameIndexes();
822061da546Spatrick 
823061da546Spatrick   if (name) {
824061da546Spatrick     // The string table did have a string that matched, but we need to check
825061da546Spatrick     // the symbols and match the symbol_type if any was given.
826061da546Spatrick     AppendSymbolIndexesWithNameAndType(name, symbol_type, symbol_indexes);
827061da546Spatrick   }
828061da546Spatrick }
829061da546Spatrick 
FindAllSymbolsWithNameAndType(ConstString name,SymbolType symbol_type,Debug symbol_debug_type,Visibility symbol_visibility,std::vector<uint32_t> & symbol_indexes)830061da546Spatrick void Symtab::FindAllSymbolsWithNameAndType(
831061da546Spatrick     ConstString name, SymbolType symbol_type, Debug symbol_debug_type,
832061da546Spatrick     Visibility symbol_visibility, std::vector<uint32_t> &symbol_indexes) {
833061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
834061da546Spatrick 
835be691f3bSpatrick   LLDB_SCOPED_TIMER();
836061da546Spatrick   // Initialize all of the lookup by name indexes before converting NAME to a
837061da546Spatrick   // uniqued string NAME_STR below.
838061da546Spatrick   if (!m_name_indexes_computed)
839061da546Spatrick     InitNameIndexes();
840061da546Spatrick 
841061da546Spatrick   if (name) {
842061da546Spatrick     // The string table did have a string that matched, but we need to check
843061da546Spatrick     // the symbols and match the symbol_type if any was given.
844061da546Spatrick     AppendSymbolIndexesWithNameAndType(name, symbol_type, symbol_debug_type,
845061da546Spatrick                                        symbol_visibility, symbol_indexes);
846061da546Spatrick   }
847061da546Spatrick }
848061da546Spatrick 
FindAllSymbolsMatchingRexExAndType(const RegularExpression & regex,SymbolType symbol_type,Debug symbol_debug_type,Visibility symbol_visibility,std::vector<uint32_t> & symbol_indexes,Mangled::NamePreference name_preference)849061da546Spatrick void Symtab::FindAllSymbolsMatchingRexExAndType(
850061da546Spatrick     const RegularExpression &regex, SymbolType symbol_type,
851061da546Spatrick     Debug symbol_debug_type, Visibility symbol_visibility,
852*f6aab3d8Srobert     std::vector<uint32_t> &symbol_indexes,
853*f6aab3d8Srobert     Mangled::NamePreference name_preference) {
854061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
855061da546Spatrick 
856061da546Spatrick   AppendSymbolIndexesMatchingRegExAndType(regex, symbol_type, symbol_debug_type,
857*f6aab3d8Srobert                                           symbol_visibility, symbol_indexes,
858*f6aab3d8Srobert                                           name_preference);
859061da546Spatrick }
860061da546Spatrick 
FindFirstSymbolWithNameAndType(ConstString name,SymbolType symbol_type,Debug symbol_debug_type,Visibility symbol_visibility)861061da546Spatrick Symbol *Symtab::FindFirstSymbolWithNameAndType(ConstString name,
862061da546Spatrick                                                SymbolType symbol_type,
863061da546Spatrick                                                Debug symbol_debug_type,
864061da546Spatrick                                                Visibility symbol_visibility) {
865061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
866be691f3bSpatrick   LLDB_SCOPED_TIMER();
867061da546Spatrick   if (!m_name_indexes_computed)
868061da546Spatrick     InitNameIndexes();
869061da546Spatrick 
870061da546Spatrick   if (name) {
871061da546Spatrick     std::vector<uint32_t> matching_indexes;
872061da546Spatrick     // The string table did have a string that matched, but we need to check
873061da546Spatrick     // the symbols and match the symbol_type if any was given.
874061da546Spatrick     if (AppendSymbolIndexesWithNameAndType(name, symbol_type, symbol_debug_type,
875061da546Spatrick                                            symbol_visibility,
876061da546Spatrick                                            matching_indexes)) {
877061da546Spatrick       std::vector<uint32_t>::const_iterator pos, end = matching_indexes.end();
878061da546Spatrick       for (pos = matching_indexes.begin(); pos != end; ++pos) {
879061da546Spatrick         Symbol *symbol = SymbolAtIndex(*pos);
880061da546Spatrick 
881061da546Spatrick         if (symbol->Compare(name, symbol_type))
882061da546Spatrick           return symbol;
883061da546Spatrick       }
884061da546Spatrick     }
885061da546Spatrick   }
886061da546Spatrick   return nullptr;
887061da546Spatrick }
888061da546Spatrick 
889061da546Spatrick typedef struct {
890061da546Spatrick   const Symtab *symtab;
891061da546Spatrick   const addr_t file_addr;
892061da546Spatrick   Symbol *match_symbol;
893061da546Spatrick   const uint32_t *match_index_ptr;
894061da546Spatrick   addr_t match_offset;
895061da546Spatrick } SymbolSearchInfo;
896061da546Spatrick 
897061da546Spatrick // Add all the section file start address & size to the RangeVector, recusively
898061da546Spatrick // adding any children sections.
AddSectionsToRangeMap(SectionList * sectlist,RangeVector<addr_t,addr_t> & section_ranges)899061da546Spatrick static void AddSectionsToRangeMap(SectionList *sectlist,
900061da546Spatrick                                   RangeVector<addr_t, addr_t> &section_ranges) {
901061da546Spatrick   const int num_sections = sectlist->GetNumSections(0);
902061da546Spatrick   for (int i = 0; i < num_sections; i++) {
903061da546Spatrick     SectionSP sect_sp = sectlist->GetSectionAtIndex(i);
904061da546Spatrick     if (sect_sp) {
905061da546Spatrick       SectionList &child_sectlist = sect_sp->GetChildren();
906061da546Spatrick 
907061da546Spatrick       // If this section has children, add the children to the RangeVector.
908061da546Spatrick       // Else add this section to the RangeVector.
909061da546Spatrick       if (child_sectlist.GetNumSections(0) > 0) {
910061da546Spatrick         AddSectionsToRangeMap(&child_sectlist, section_ranges);
911061da546Spatrick       } else {
912061da546Spatrick         size_t size = sect_sp->GetByteSize();
913061da546Spatrick         if (size > 0) {
914061da546Spatrick           addr_t base_addr = sect_sp->GetFileAddress();
915061da546Spatrick           RangeVector<addr_t, addr_t>::Entry entry;
916061da546Spatrick           entry.SetRangeBase(base_addr);
917061da546Spatrick           entry.SetByteSize(size);
918061da546Spatrick           section_ranges.Append(entry);
919061da546Spatrick         }
920061da546Spatrick       }
921061da546Spatrick     }
922061da546Spatrick   }
923061da546Spatrick }
924061da546Spatrick 
InitAddressIndexes()925061da546Spatrick void Symtab::InitAddressIndexes() {
926061da546Spatrick   // Protected function, no need to lock mutex...
927061da546Spatrick   if (!m_file_addr_to_index_computed && !m_symbols.empty()) {
928061da546Spatrick     m_file_addr_to_index_computed = true;
929061da546Spatrick 
930061da546Spatrick     FileRangeToIndexMap::Entry entry;
931061da546Spatrick     const_iterator begin = m_symbols.begin();
932061da546Spatrick     const_iterator end = m_symbols.end();
933061da546Spatrick     for (const_iterator pos = m_symbols.begin(); pos != end; ++pos) {
934061da546Spatrick       if (pos->ValueIsAddress()) {
935061da546Spatrick         entry.SetRangeBase(pos->GetAddressRef().GetFileAddress());
936061da546Spatrick         entry.SetByteSize(pos->GetByteSize());
937061da546Spatrick         entry.data = std::distance(begin, pos);
938061da546Spatrick         m_file_addr_to_index.Append(entry);
939061da546Spatrick       }
940061da546Spatrick     }
941061da546Spatrick     const size_t num_entries = m_file_addr_to_index.GetSize();
942061da546Spatrick     if (num_entries > 0) {
943061da546Spatrick       m_file_addr_to_index.Sort();
944061da546Spatrick 
945061da546Spatrick       // Create a RangeVector with the start & size of all the sections for
946061da546Spatrick       // this objfile.  We'll need to check this for any FileRangeToIndexMap
947061da546Spatrick       // entries with an uninitialized size, which could potentially be a large
948061da546Spatrick       // number so reconstituting the weak pointer is busywork when it is
949061da546Spatrick       // invariant information.
950061da546Spatrick       SectionList *sectlist = m_objfile->GetSectionList();
951061da546Spatrick       RangeVector<addr_t, addr_t> section_ranges;
952061da546Spatrick       if (sectlist) {
953061da546Spatrick         AddSectionsToRangeMap(sectlist, section_ranges);
954061da546Spatrick         section_ranges.Sort();
955061da546Spatrick       }
956061da546Spatrick 
957061da546Spatrick       // Iterate through the FileRangeToIndexMap and fill in the size for any
958061da546Spatrick       // entries that didn't already have a size from the Symbol (e.g. if we
959061da546Spatrick       // have a plain linker symbol with an address only, instead of debug info
960061da546Spatrick       // where we get an address and a size and a type, etc.)
961061da546Spatrick       for (size_t i = 0; i < num_entries; i++) {
962061da546Spatrick         FileRangeToIndexMap::Entry *entry =
963061da546Spatrick             m_file_addr_to_index.GetMutableEntryAtIndex(i);
964061da546Spatrick         if (entry->GetByteSize() == 0) {
965061da546Spatrick           addr_t curr_base_addr = entry->GetRangeBase();
966061da546Spatrick           const RangeVector<addr_t, addr_t>::Entry *containing_section =
967061da546Spatrick               section_ranges.FindEntryThatContains(curr_base_addr);
968061da546Spatrick 
969061da546Spatrick           // Use the end of the section as the default max size of the symbol
970061da546Spatrick           addr_t sym_size = 0;
971061da546Spatrick           if (containing_section) {
972061da546Spatrick             sym_size =
973061da546Spatrick                 containing_section->GetByteSize() -
974061da546Spatrick                 (entry->GetRangeBase() - containing_section->GetRangeBase());
975061da546Spatrick           }
976061da546Spatrick 
977061da546Spatrick           for (size_t j = i; j < num_entries; j++) {
978061da546Spatrick             FileRangeToIndexMap::Entry *next_entry =
979061da546Spatrick                 m_file_addr_to_index.GetMutableEntryAtIndex(j);
980061da546Spatrick             addr_t next_base_addr = next_entry->GetRangeBase();
981061da546Spatrick             if (next_base_addr > curr_base_addr) {
982061da546Spatrick               addr_t size_to_next_symbol = next_base_addr - curr_base_addr;
983061da546Spatrick 
984061da546Spatrick               // Take the difference between this symbol and the next one as
985061da546Spatrick               // its size, if it is less than the size of the section.
986061da546Spatrick               if (sym_size == 0 || size_to_next_symbol < sym_size) {
987061da546Spatrick                 sym_size = size_to_next_symbol;
988061da546Spatrick               }
989061da546Spatrick               break;
990061da546Spatrick             }
991061da546Spatrick           }
992061da546Spatrick 
993061da546Spatrick           if (sym_size > 0) {
994061da546Spatrick             entry->SetByteSize(sym_size);
995061da546Spatrick             Symbol &symbol = m_symbols[entry->data];
996061da546Spatrick             symbol.SetByteSize(sym_size);
997061da546Spatrick             symbol.SetSizeIsSynthesized(true);
998061da546Spatrick           }
999061da546Spatrick         }
1000061da546Spatrick       }
1001061da546Spatrick 
1002061da546Spatrick       // Sort again in case the range size changes the ordering
1003061da546Spatrick       m_file_addr_to_index.Sort();
1004061da546Spatrick     }
1005061da546Spatrick   }
1006061da546Spatrick }
1007061da546Spatrick 
Finalize()1008*f6aab3d8Srobert void Symtab::Finalize() {
1009061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
1010*f6aab3d8Srobert   // Calculate the size of symbols inside InitAddressIndexes.
1011061da546Spatrick   InitAddressIndexes();
1012*f6aab3d8Srobert   // Shrink to fit the symbols so we don't waste memory
1013*f6aab3d8Srobert   if (m_symbols.capacity() > m_symbols.size()) {
1014*f6aab3d8Srobert     collection new_symbols(m_symbols.begin(), m_symbols.end());
1015*f6aab3d8Srobert     m_symbols.swap(new_symbols);
1016*f6aab3d8Srobert   }
1017*f6aab3d8Srobert   SaveToCache();
1018061da546Spatrick }
1019061da546Spatrick 
FindSymbolAtFileAddress(addr_t file_addr)1020061da546Spatrick Symbol *Symtab::FindSymbolAtFileAddress(addr_t file_addr) {
1021061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
1022061da546Spatrick   if (!m_file_addr_to_index_computed)
1023061da546Spatrick     InitAddressIndexes();
1024061da546Spatrick 
1025061da546Spatrick   const FileRangeToIndexMap::Entry *entry =
1026061da546Spatrick       m_file_addr_to_index.FindEntryStartsAt(file_addr);
1027061da546Spatrick   if (entry) {
1028061da546Spatrick     Symbol *symbol = SymbolAtIndex(entry->data);
1029061da546Spatrick     if (symbol->GetFileAddress() == file_addr)
1030061da546Spatrick       return symbol;
1031061da546Spatrick   }
1032061da546Spatrick   return nullptr;
1033061da546Spatrick }
1034061da546Spatrick 
FindSymbolContainingFileAddress(addr_t file_addr)1035061da546Spatrick Symbol *Symtab::FindSymbolContainingFileAddress(addr_t file_addr) {
1036061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
1037061da546Spatrick 
1038061da546Spatrick   if (!m_file_addr_to_index_computed)
1039061da546Spatrick     InitAddressIndexes();
1040061da546Spatrick 
1041061da546Spatrick   const FileRangeToIndexMap::Entry *entry =
1042061da546Spatrick       m_file_addr_to_index.FindEntryThatContains(file_addr);
1043061da546Spatrick   if (entry) {
1044061da546Spatrick     Symbol *symbol = SymbolAtIndex(entry->data);
1045061da546Spatrick     if (symbol->ContainsFileAddress(file_addr))
1046061da546Spatrick       return symbol;
1047061da546Spatrick   }
1048061da546Spatrick   return nullptr;
1049061da546Spatrick }
1050061da546Spatrick 
ForEachSymbolContainingFileAddress(addr_t file_addr,std::function<bool (Symbol *)> const & callback)1051061da546Spatrick void Symtab::ForEachSymbolContainingFileAddress(
1052061da546Spatrick     addr_t file_addr, std::function<bool(Symbol *)> const &callback) {
1053061da546Spatrick   std::lock_guard<std::recursive_mutex> guard(m_mutex);
1054061da546Spatrick 
1055061da546Spatrick   if (!m_file_addr_to_index_computed)
1056061da546Spatrick     InitAddressIndexes();
1057061da546Spatrick 
1058061da546Spatrick   std::vector<uint32_t> all_addr_indexes;
1059061da546Spatrick 
1060061da546Spatrick   // Get all symbols with file_addr
1061061da546Spatrick   const size_t addr_match_count =
1062061da546Spatrick       m_file_addr_to_index.FindEntryIndexesThatContain(file_addr,
1063061da546Spatrick                                                        all_addr_indexes);
1064061da546Spatrick 
1065061da546Spatrick   for (size_t i = 0; i < addr_match_count; ++i) {
1066061da546Spatrick     Symbol *symbol = SymbolAtIndex(all_addr_indexes[i]);
1067061da546Spatrick     if (symbol->ContainsFileAddress(file_addr)) {
1068061da546Spatrick       if (!callback(symbol))
1069061da546Spatrick         break;
1070061da546Spatrick     }
1071061da546Spatrick   }
1072061da546Spatrick }
1073061da546Spatrick 
SymbolIndicesToSymbolContextList(std::vector<uint32_t> & symbol_indexes,SymbolContextList & sc_list)1074061da546Spatrick void Symtab::SymbolIndicesToSymbolContextList(
1075061da546Spatrick     std::vector<uint32_t> &symbol_indexes, SymbolContextList &sc_list) {
1076061da546Spatrick   // No need to protect this call using m_mutex all other method calls are
1077061da546Spatrick   // already thread safe.
1078061da546Spatrick 
1079061da546Spatrick   const bool merge_symbol_into_function = true;
1080061da546Spatrick   size_t num_indices = symbol_indexes.size();
1081061da546Spatrick   if (num_indices > 0) {
1082061da546Spatrick     SymbolContext sc;
1083061da546Spatrick     sc.module_sp = m_objfile->GetModule();
1084061da546Spatrick     for (size_t i = 0; i < num_indices; i++) {
1085061da546Spatrick       sc.symbol = SymbolAtIndex(symbol_indexes[i]);
1086061da546Spatrick       if (sc.symbol)
1087061da546Spatrick         sc_list.AppendIfUnique(sc, merge_symbol_into_function);
1088061da546Spatrick     }
1089061da546Spatrick   }
1090061da546Spatrick }
1091061da546Spatrick 
FindFunctionSymbols(ConstString name,uint32_t name_type_mask,SymbolContextList & sc_list)1092061da546Spatrick void Symtab::FindFunctionSymbols(ConstString name, uint32_t name_type_mask,
1093061da546Spatrick                                  SymbolContextList &sc_list) {
1094061da546Spatrick   std::vector<uint32_t> symbol_indexes;
1095061da546Spatrick 
1096061da546Spatrick   // eFunctionNameTypeAuto should be pre-resolved by a call to
1097061da546Spatrick   // Module::LookupInfo::LookupInfo()
1098061da546Spatrick   assert((name_type_mask & eFunctionNameTypeAuto) == 0);
1099061da546Spatrick 
1100061da546Spatrick   if (name_type_mask & (eFunctionNameTypeBase | eFunctionNameTypeFull)) {
1101061da546Spatrick     std::vector<uint32_t> temp_symbol_indexes;
1102061da546Spatrick     FindAllSymbolsWithNameAndType(name, eSymbolTypeAny, temp_symbol_indexes);
1103061da546Spatrick 
1104061da546Spatrick     unsigned temp_symbol_indexes_size = temp_symbol_indexes.size();
1105061da546Spatrick     if (temp_symbol_indexes_size > 0) {
1106061da546Spatrick       std::lock_guard<std::recursive_mutex> guard(m_mutex);
1107061da546Spatrick       for (unsigned i = 0; i < temp_symbol_indexes_size; i++) {
1108061da546Spatrick         SymbolContext sym_ctx;
1109061da546Spatrick         sym_ctx.symbol = SymbolAtIndex(temp_symbol_indexes[i]);
1110061da546Spatrick         if (sym_ctx.symbol) {
1111061da546Spatrick           switch (sym_ctx.symbol->GetType()) {
1112061da546Spatrick           case eSymbolTypeCode:
1113061da546Spatrick           case eSymbolTypeResolver:
1114061da546Spatrick           case eSymbolTypeReExported:
1115*f6aab3d8Srobert           case eSymbolTypeAbsolute:
1116061da546Spatrick             symbol_indexes.push_back(temp_symbol_indexes[i]);
1117061da546Spatrick             break;
1118061da546Spatrick           default:
1119061da546Spatrick             break;
1120061da546Spatrick           }
1121061da546Spatrick         }
1122061da546Spatrick       }
1123061da546Spatrick     }
1124061da546Spatrick   }
1125061da546Spatrick 
1126061da546Spatrick   if (!m_name_indexes_computed)
1127061da546Spatrick     InitNameIndexes();
1128061da546Spatrick 
1129be691f3bSpatrick   for (lldb::FunctionNameType type :
1130be691f3bSpatrick        {lldb::eFunctionNameTypeBase, lldb::eFunctionNameTypeMethod,
1131be691f3bSpatrick         lldb::eFunctionNameTypeSelector}) {
1132be691f3bSpatrick     if (name_type_mask & type) {
1133be691f3bSpatrick       auto map = GetNameToSymbolIndexMap(type);
1134be691f3bSpatrick 
1135061da546Spatrick       const UniqueCStringMap<uint32_t>::Entry *match;
1136be691f3bSpatrick       for (match = map.FindFirstValueForName(name); match != nullptr;
1137be691f3bSpatrick            match = map.FindNextValueForName(match)) {
1138061da546Spatrick         symbol_indexes.push_back(match->value);
1139061da546Spatrick       }
1140061da546Spatrick     }
1141061da546Spatrick   }
1142061da546Spatrick 
1143061da546Spatrick   if (!symbol_indexes.empty()) {
1144*f6aab3d8Srobert     llvm::sort(symbol_indexes);
1145061da546Spatrick     symbol_indexes.erase(
1146061da546Spatrick         std::unique(symbol_indexes.begin(), symbol_indexes.end()),
1147061da546Spatrick         symbol_indexes.end());
1148061da546Spatrick     SymbolIndicesToSymbolContextList(symbol_indexes, sc_list);
1149061da546Spatrick   }
1150061da546Spatrick }
1151061da546Spatrick 
GetParent(Symbol * child_symbol) const1152061da546Spatrick const Symbol *Symtab::GetParent(Symbol *child_symbol) const {
1153061da546Spatrick   uint32_t child_idx = GetIndexForSymbol(child_symbol);
1154061da546Spatrick   if (child_idx != UINT32_MAX && child_idx > 0) {
1155061da546Spatrick     for (uint32_t idx = child_idx - 1; idx != UINT32_MAX; --idx) {
1156061da546Spatrick       const Symbol *symbol = SymbolAtIndex(idx);
1157061da546Spatrick       const uint32_t sibling_idx = symbol->GetSiblingIndex();
1158061da546Spatrick       if (sibling_idx != UINT32_MAX && sibling_idx > child_idx)
1159061da546Spatrick         return symbol;
1160061da546Spatrick     }
1161061da546Spatrick   }
1162061da546Spatrick   return nullptr;
1163061da546Spatrick }
1164*f6aab3d8Srobert 
GetCacheKey()1165*f6aab3d8Srobert std::string Symtab::GetCacheKey() {
1166*f6aab3d8Srobert   std::string key;
1167*f6aab3d8Srobert   llvm::raw_string_ostream strm(key);
1168*f6aab3d8Srobert   // Symbol table can come from different object files for the same module. A
1169*f6aab3d8Srobert   // module can have one object file as the main executable and might have
1170*f6aab3d8Srobert   // another object file in a separate symbol file.
1171*f6aab3d8Srobert   strm << m_objfile->GetModule()->GetCacheKey() << "-symtab-"
1172*f6aab3d8Srobert       << llvm::format_hex(m_objfile->GetCacheHash(), 10);
1173*f6aab3d8Srobert   return strm.str();
1174*f6aab3d8Srobert }
1175*f6aab3d8Srobert 
SaveToCache()1176*f6aab3d8Srobert void Symtab::SaveToCache() {
1177*f6aab3d8Srobert   DataFileCache *cache = Module::GetIndexCache();
1178*f6aab3d8Srobert   if (!cache)
1179*f6aab3d8Srobert     return; // Caching is not enabled.
1180*f6aab3d8Srobert   InitNameIndexes(); // Init the name indexes so we can cache them as well.
1181*f6aab3d8Srobert   const auto byte_order = endian::InlHostByteOrder();
1182*f6aab3d8Srobert   DataEncoder file(byte_order, /*addr_size=*/8);
1183*f6aab3d8Srobert   // Encode will return false if the symbol table's object file doesn't have
1184*f6aab3d8Srobert   // anything to make a signature from.
1185*f6aab3d8Srobert   if (Encode(file))
1186*f6aab3d8Srobert     if (cache->SetCachedData(GetCacheKey(), file.GetData()))
1187*f6aab3d8Srobert       SetWasSavedToCache();
1188*f6aab3d8Srobert }
1189*f6aab3d8Srobert 
1190*f6aab3d8Srobert constexpr llvm::StringLiteral kIdentifierCStrMap("CMAP");
1191*f6aab3d8Srobert 
EncodeCStrMap(DataEncoder & encoder,ConstStringTable & strtab,const UniqueCStringMap<uint32_t> & cstr_map)1192*f6aab3d8Srobert static void EncodeCStrMap(DataEncoder &encoder, ConstStringTable &strtab,
1193*f6aab3d8Srobert                           const UniqueCStringMap<uint32_t> &cstr_map) {
1194*f6aab3d8Srobert   encoder.AppendData(kIdentifierCStrMap);
1195*f6aab3d8Srobert   encoder.AppendU32(cstr_map.GetSize());
1196*f6aab3d8Srobert   for (const auto &entry: cstr_map) {
1197*f6aab3d8Srobert     // Make sure there are no empty strings.
1198*f6aab3d8Srobert     assert((bool)entry.cstring);
1199*f6aab3d8Srobert     encoder.AppendU32(strtab.Add(entry.cstring));
1200*f6aab3d8Srobert     encoder.AppendU32(entry.value);
1201*f6aab3d8Srobert   }
1202*f6aab3d8Srobert }
1203*f6aab3d8Srobert 
DecodeCStrMap(const DataExtractor & data,lldb::offset_t * offset_ptr,const StringTableReader & strtab,UniqueCStringMap<uint32_t> & cstr_map)1204*f6aab3d8Srobert bool DecodeCStrMap(const DataExtractor &data, lldb::offset_t *offset_ptr,
1205*f6aab3d8Srobert                    const StringTableReader &strtab,
1206*f6aab3d8Srobert                    UniqueCStringMap<uint32_t> &cstr_map) {
1207*f6aab3d8Srobert   llvm::StringRef identifier((const char *)data.GetData(offset_ptr, 4), 4);
1208*f6aab3d8Srobert   if (identifier != kIdentifierCStrMap)
1209*f6aab3d8Srobert     return false;
1210*f6aab3d8Srobert   const uint32_t count = data.GetU32(offset_ptr);
1211*f6aab3d8Srobert   cstr_map.Reserve(count);
1212*f6aab3d8Srobert   for (uint32_t i=0; i<count; ++i)
1213*f6aab3d8Srobert   {
1214*f6aab3d8Srobert     llvm::StringRef str(strtab.Get(data.GetU32(offset_ptr)));
1215*f6aab3d8Srobert     uint32_t value = data.GetU32(offset_ptr);
1216*f6aab3d8Srobert     // No empty strings in the name indexes in Symtab
1217*f6aab3d8Srobert     if (str.empty())
1218*f6aab3d8Srobert       return false;
1219*f6aab3d8Srobert     cstr_map.Append(ConstString(str), value);
1220*f6aab3d8Srobert   }
1221*f6aab3d8Srobert   // We must sort the UniqueCStringMap after decoding it since it is a vector
1222*f6aab3d8Srobert   // of UniqueCStringMap::Entry objects which contain a ConstString and type T.
1223*f6aab3d8Srobert   // ConstString objects are sorted by "const char *" and then type T and
1224*f6aab3d8Srobert   // the "const char *" are point values that will depend on the order in which
1225*f6aab3d8Srobert   // ConstString objects are created and in which of the 256 string pools they
1226*f6aab3d8Srobert   // are created in. So after we decode all of the entries, we must sort the
1227*f6aab3d8Srobert   // name map to ensure name lookups succeed. If we encode and decode within
1228*f6aab3d8Srobert   // the same process we wouldn't need to sort, so unit testing didn't catch
1229*f6aab3d8Srobert   // this issue when first checked in.
1230*f6aab3d8Srobert   cstr_map.Sort();
1231*f6aab3d8Srobert   return true;
1232*f6aab3d8Srobert }
1233*f6aab3d8Srobert 
1234*f6aab3d8Srobert constexpr llvm::StringLiteral kIdentifierSymbolTable("SYMB");
1235*f6aab3d8Srobert constexpr uint32_t CURRENT_CACHE_VERSION = 1;
1236*f6aab3d8Srobert 
1237*f6aab3d8Srobert /// The encoding format for the symbol table is as follows:
1238*f6aab3d8Srobert ///
1239*f6aab3d8Srobert /// Signature signature;
1240*f6aab3d8Srobert /// ConstStringTable strtab;
1241*f6aab3d8Srobert /// Identifier four character code: 'SYMB'
1242*f6aab3d8Srobert /// uint32_t version;
1243*f6aab3d8Srobert /// uint32_t num_symbols;
1244*f6aab3d8Srobert /// Symbol symbols[num_symbols];
1245*f6aab3d8Srobert /// uint8_t num_cstr_maps;
1246*f6aab3d8Srobert /// UniqueCStringMap<uint32_t> cstr_maps[num_cstr_maps]
Encode(DataEncoder & encoder) const1247*f6aab3d8Srobert bool Symtab::Encode(DataEncoder &encoder) const {
1248*f6aab3d8Srobert   // Name indexes must be computed before calling this function.
1249*f6aab3d8Srobert   assert(m_name_indexes_computed);
1250*f6aab3d8Srobert 
1251*f6aab3d8Srobert   // Encode the object file's signature
1252*f6aab3d8Srobert   CacheSignature signature(m_objfile);
1253*f6aab3d8Srobert   if (!signature.Encode(encoder))
1254*f6aab3d8Srobert     return false;
1255*f6aab3d8Srobert   ConstStringTable strtab;
1256*f6aab3d8Srobert 
1257*f6aab3d8Srobert   // Encoder the symbol table into a separate encoder first. This allows us
1258*f6aab3d8Srobert   // gather all of the strings we willl need in "strtab" as we will need to
1259*f6aab3d8Srobert   // write the string table out before the symbol table.
1260*f6aab3d8Srobert   DataEncoder symtab_encoder(encoder.GetByteOrder(),
1261*f6aab3d8Srobert                               encoder.GetAddressByteSize());
1262*f6aab3d8Srobert   symtab_encoder.AppendData(kIdentifierSymbolTable);
1263*f6aab3d8Srobert   // Encode the symtab data version.
1264*f6aab3d8Srobert   symtab_encoder.AppendU32(CURRENT_CACHE_VERSION);
1265*f6aab3d8Srobert   // Encode the number of symbols.
1266*f6aab3d8Srobert   symtab_encoder.AppendU32(m_symbols.size());
1267*f6aab3d8Srobert   // Encode the symbol data for all symbols.
1268*f6aab3d8Srobert   for (const auto &symbol: m_symbols)
1269*f6aab3d8Srobert     symbol.Encode(symtab_encoder, strtab);
1270*f6aab3d8Srobert 
1271*f6aab3d8Srobert   // Emit a byte for how many C string maps we emit. We will fix this up after
1272*f6aab3d8Srobert   // we emit the C string maps since we skip emitting C string maps if they are
1273*f6aab3d8Srobert   // empty.
1274*f6aab3d8Srobert   size_t num_cmaps_offset = symtab_encoder.GetByteSize();
1275*f6aab3d8Srobert   uint8_t num_cmaps = 0;
1276*f6aab3d8Srobert   symtab_encoder.AppendU8(0);
1277*f6aab3d8Srobert   for (const auto &pair: m_name_to_symbol_indices) {
1278*f6aab3d8Srobert     if (pair.second.IsEmpty())
1279*f6aab3d8Srobert       continue;
1280*f6aab3d8Srobert     ++num_cmaps;
1281*f6aab3d8Srobert     symtab_encoder.AppendU8(pair.first);
1282*f6aab3d8Srobert     EncodeCStrMap(symtab_encoder, strtab, pair.second);
1283*f6aab3d8Srobert   }
1284*f6aab3d8Srobert   if (num_cmaps > 0)
1285*f6aab3d8Srobert     symtab_encoder.PutU8(num_cmaps_offset, num_cmaps);
1286*f6aab3d8Srobert 
1287*f6aab3d8Srobert   // Now that all strings have been gathered, we will emit the string table.
1288*f6aab3d8Srobert   strtab.Encode(encoder);
1289*f6aab3d8Srobert   // Followed the the symbol table data.
1290*f6aab3d8Srobert   encoder.AppendData(symtab_encoder.GetData());
1291*f6aab3d8Srobert   return true;
1292*f6aab3d8Srobert }
1293*f6aab3d8Srobert 
Decode(const DataExtractor & data,lldb::offset_t * offset_ptr,bool & signature_mismatch)1294*f6aab3d8Srobert bool Symtab::Decode(const DataExtractor &data, lldb::offset_t *offset_ptr,
1295*f6aab3d8Srobert                     bool &signature_mismatch) {
1296*f6aab3d8Srobert   signature_mismatch = false;
1297*f6aab3d8Srobert   CacheSignature signature;
1298*f6aab3d8Srobert   StringTableReader strtab;
1299*f6aab3d8Srobert   { // Scope for "elapsed" object below so it can measure the time parse.
1300*f6aab3d8Srobert     ElapsedTime elapsed(m_objfile->GetModule()->GetSymtabParseTime());
1301*f6aab3d8Srobert     if (!signature.Decode(data, offset_ptr))
1302*f6aab3d8Srobert       return false;
1303*f6aab3d8Srobert     if (CacheSignature(m_objfile) != signature) {
1304*f6aab3d8Srobert       signature_mismatch = true;
1305*f6aab3d8Srobert       return false;
1306*f6aab3d8Srobert     }
1307*f6aab3d8Srobert     // We now decode the string table for all strings in the data cache file.
1308*f6aab3d8Srobert     if (!strtab.Decode(data, offset_ptr))
1309*f6aab3d8Srobert       return false;
1310*f6aab3d8Srobert 
1311*f6aab3d8Srobert     // And now we can decode the symbol table with string table we just decoded.
1312*f6aab3d8Srobert     llvm::StringRef identifier((const char *)data.GetData(offset_ptr, 4), 4);
1313*f6aab3d8Srobert     if (identifier != kIdentifierSymbolTable)
1314*f6aab3d8Srobert       return false;
1315*f6aab3d8Srobert     const uint32_t version = data.GetU32(offset_ptr);
1316*f6aab3d8Srobert     if (version != CURRENT_CACHE_VERSION)
1317*f6aab3d8Srobert       return false;
1318*f6aab3d8Srobert     const uint32_t num_symbols = data.GetU32(offset_ptr);
1319*f6aab3d8Srobert     if (num_symbols == 0)
1320*f6aab3d8Srobert       return true;
1321*f6aab3d8Srobert     m_symbols.resize(num_symbols);
1322*f6aab3d8Srobert     SectionList *sections = m_objfile->GetModule()->GetSectionList();
1323*f6aab3d8Srobert     for (uint32_t i=0; i<num_symbols; ++i) {
1324*f6aab3d8Srobert       if (!m_symbols[i].Decode(data, offset_ptr, sections, strtab))
1325*f6aab3d8Srobert         return false;
1326*f6aab3d8Srobert     }
1327*f6aab3d8Srobert   }
1328*f6aab3d8Srobert 
1329*f6aab3d8Srobert   { // Scope for "elapsed" object below so it can measure the time to index.
1330*f6aab3d8Srobert     ElapsedTime elapsed(m_objfile->GetModule()->GetSymtabIndexTime());
1331*f6aab3d8Srobert     const uint8_t num_cstr_maps = data.GetU8(offset_ptr);
1332*f6aab3d8Srobert     for (uint8_t i=0; i<num_cstr_maps; ++i) {
1333*f6aab3d8Srobert       uint8_t type = data.GetU8(offset_ptr);
1334*f6aab3d8Srobert       UniqueCStringMap<uint32_t> &cstr_map =
1335*f6aab3d8Srobert           GetNameToSymbolIndexMap((lldb::FunctionNameType)type);
1336*f6aab3d8Srobert       if (!DecodeCStrMap(data, offset_ptr, strtab, cstr_map))
1337*f6aab3d8Srobert         return false;
1338*f6aab3d8Srobert     }
1339*f6aab3d8Srobert     m_name_indexes_computed = true;
1340*f6aab3d8Srobert   }
1341*f6aab3d8Srobert   return true;
1342*f6aab3d8Srobert }
1343*f6aab3d8Srobert 
LoadFromCache()1344*f6aab3d8Srobert bool Symtab::LoadFromCache() {
1345*f6aab3d8Srobert   DataFileCache *cache = Module::GetIndexCache();
1346*f6aab3d8Srobert   if (!cache)
1347*f6aab3d8Srobert     return false;
1348*f6aab3d8Srobert 
1349*f6aab3d8Srobert   std::unique_ptr<llvm::MemoryBuffer> mem_buffer_up =
1350*f6aab3d8Srobert       cache->GetCachedData(GetCacheKey());
1351*f6aab3d8Srobert   if (!mem_buffer_up)
1352*f6aab3d8Srobert     return false;
1353*f6aab3d8Srobert   DataExtractor data(mem_buffer_up->getBufferStart(),
1354*f6aab3d8Srobert                      mem_buffer_up->getBufferSize(),
1355*f6aab3d8Srobert                      m_objfile->GetByteOrder(),
1356*f6aab3d8Srobert                      m_objfile->GetAddressByteSize());
1357*f6aab3d8Srobert   bool signature_mismatch = false;
1358*f6aab3d8Srobert   lldb::offset_t offset = 0;
1359*f6aab3d8Srobert   const bool result = Decode(data, &offset, signature_mismatch);
1360*f6aab3d8Srobert   if (signature_mismatch)
1361*f6aab3d8Srobert     cache->RemoveCacheFile(GetCacheKey());
1362*f6aab3d8Srobert   if (result)
1363*f6aab3d8Srobert     SetWasLoadedFromCache();
1364*f6aab3d8Srobert   return result;
1365*f6aab3d8Srobert }
1366