blob: 9ce0df957d7d20ad415029e1b4d21a2ab5e2db01 [file] [log] [blame]
Chris Lattner30fdc8d2010-06-08 16:52:24 +00001//===-- Symtab.cpp ----------------------------------------------*- C++ -*-===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
10#include <map>
11
12#include "lldb/Core/Module.h"
13#include "lldb/Core/RegularExpression.h"
14#include "lldb/Core/Timer.h"
15#include "lldb/Symbol/ObjectFile.h"
16#include "lldb/Symbol/Symtab.h"
Jim Inghamff5f5ff2011-08-15 01:32:22 +000017#include "lldb/Target/ObjCLanguageRuntime.h"
Chris Lattner30fdc8d2010-06-08 16:52:24 +000018
19using namespace lldb;
20using namespace lldb_private;
21
22
23
24Symtab::Symtab(ObjectFile *objfile) :
Greg Clayton8087ca22010-10-08 04:20:14 +000025 m_objfile (objfile),
26 m_symbols (),
27 m_addr_indexes (),
28 m_name_to_index (),
29 m_mutex (Mutex::eMutexTypeRecursive),
30 m_addr_indexes_computed (false),
31 m_name_indexes_computed (false)
Chris Lattner30fdc8d2010-06-08 16:52:24 +000032{
33}
34
35Symtab::~Symtab()
36{
37}
38
39void
40Symtab::Reserve(uint32_t count)
41{
Greg Clayton8087ca22010-10-08 04:20:14 +000042 // Clients should grab the mutex from this symbol table and lock it manually
43 // when calling this function to avoid performance issues.
Chris Lattner30fdc8d2010-06-08 16:52:24 +000044 m_symbols.reserve (count);
45}
46
47Symbol *
48Symtab::Resize(uint32_t count)
49{
Greg Clayton8087ca22010-10-08 04:20:14 +000050 // Clients should grab the mutex from this symbol table and lock it manually
51 // when calling this function to avoid performance issues.
Chris Lattner30fdc8d2010-06-08 16:52:24 +000052 m_symbols.resize (count);
53 return &m_symbols[0];
54}
55
56uint32_t
57Symtab::AddSymbol(const Symbol& symbol)
58{
Greg Clayton8087ca22010-10-08 04:20:14 +000059 // Clients should grab the mutex from this symbol table and lock it manually
60 // when calling this function to avoid performance issues.
Chris Lattner30fdc8d2010-06-08 16:52:24 +000061 uint32_t symbol_idx = m_symbols.size();
62 m_name_to_index.Clear();
63 m_addr_indexes.clear();
64 m_symbols.push_back(symbol);
Greg Clayton8087ca22010-10-08 04:20:14 +000065 m_addr_indexes_computed = false;
66 m_name_indexes_computed = false;
Chris Lattner30fdc8d2010-06-08 16:52:24 +000067 return symbol_idx;
68}
69
70size_t
71Symtab::GetNumSymbols() const
72{
Greg Clayton8087ca22010-10-08 04:20:14 +000073 Mutex::Locker locker (m_mutex);
Chris Lattner30fdc8d2010-06-08 16:52:24 +000074 return m_symbols.size();
75}
76
77void
Greg Claytone0d378b2011-03-24 21:19:54 +000078Symtab::Dump (Stream *s, Target *target, SortOrder sort_order)
Chris Lattner30fdc8d2010-06-08 16:52:24 +000079{
Greg Clayton8087ca22010-10-08 04:20:14 +000080 Mutex::Locker locker (m_mutex);
81
Greg Clayton89411422010-10-08 00:21:05 +000082// s->Printf("%.*p: ", (int)sizeof(void*) * 2, this);
Chris Lattner30fdc8d2010-06-08 16:52:24 +000083 s->Indent();
84 const FileSpec &file_spec = m_objfile->GetFileSpec();
85 const char * object_name = NULL;
86 if (m_objfile->GetModule())
87 object_name = m_objfile->GetModule()->GetObjectName().GetCString();
88
89 if (file_spec)
Jason Molendafd54b362011-09-20 21:44:10 +000090 s->Printf("Symtab, file = %s/%s%s%s%s, num_symbols = %lu",
Chris Lattner30fdc8d2010-06-08 16:52:24 +000091 file_spec.GetDirectory().AsCString(),
92 file_spec.GetFilename().AsCString(),
93 object_name ? "(" : "",
94 object_name ? object_name : "",
95 object_name ? ")" : "",
96 m_symbols.size());
97 else
Jason Molendafd54b362011-09-20 21:44:10 +000098 s->Printf("Symtab, num_symbols = %lu", m_symbols.size());
Chris Lattner30fdc8d2010-06-08 16:52:24 +000099
100 if (!m_symbols.empty())
101 {
Greg Clayton8087ca22010-10-08 04:20:14 +0000102 switch (sort_order)
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000103 {
Greg Clayton8087ca22010-10-08 04:20:14 +0000104 case eSortOrderNone:
105 {
106 s->PutCString (":\n");
107 DumpSymbolHeader (s);
108 const_iterator begin = m_symbols.begin();
109 const_iterator end = m_symbols.end();
110 for (const_iterator pos = m_symbols.begin(); pos != end; ++pos)
111 {
112 s->Indent();
113 pos->Dump(s, target, std::distance(begin, pos));
114 }
115 }
116 break;
117
118 case eSortOrderByName:
119 {
120 // Although we maintain a lookup by exact name map, the table
121 // isn't sorted by name. So we must make the ordered symbol list
122 // up ourselves.
123 s->PutCString (" (sorted by name):\n");
124 DumpSymbolHeader (s);
125 typedef std::multimap<const char*, const Symbol *, CStringCompareFunctionObject> CStringToSymbol;
126 CStringToSymbol name_map;
127 for (const_iterator pos = m_symbols.begin(), end = m_symbols.end(); pos != end; ++pos)
128 {
129 const char *name = pos->GetMangled().GetName(Mangled::ePreferDemangled).AsCString();
130 if (name && name[0])
131 name_map.insert (std::make_pair(name, &(*pos)));
132 }
133
134 for (CStringToSymbol::const_iterator pos = name_map.begin(), end = name_map.end(); pos != end; ++pos)
135 {
136 s->Indent();
137 pos->second->Dump (s, target, pos->second - &m_symbols[0]);
138 }
139 }
140 break;
141
142 case eSortOrderByAddress:
143 s->PutCString (" (sorted by address):\n");
144 DumpSymbolHeader (s);
145 if (!m_addr_indexes_computed)
146 InitAddressIndexes();
147 const size_t num_symbols = GetNumSymbols();
148 std::vector<uint32_t>::const_iterator pos;
149 std::vector<uint32_t>::const_iterator end = m_addr_indexes.end();
150 for (pos = m_addr_indexes.begin(); pos != end; ++pos)
151 {
152 uint32_t idx = *pos;
153 if (idx < num_symbols)
154 {
155 s->Indent();
156 m_symbols[idx].Dump(s, target, idx);
157 }
158 }
159 break;
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000160 }
161 }
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000162}
163
164void
Greg Claytonf5e56de2010-09-14 23:36:40 +0000165Symtab::Dump(Stream *s, Target *target, std::vector<uint32_t>& indexes) const
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000166{
Greg Clayton8087ca22010-10-08 04:20:14 +0000167 Mutex::Locker locker (m_mutex);
168
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000169 const size_t num_symbols = GetNumSymbols();
Greg Clayton8087ca22010-10-08 04:20:14 +0000170 //s->Printf("%.*p: ", (int)sizeof(void*) * 2, this);
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000171 s->Indent();
Jason Molendafd54b362011-09-20 21:44:10 +0000172 s->Printf("Symtab %lu symbol indexes (%lu symbols total):\n", indexes.size(), m_symbols.size());
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000173 s->IndentMore();
174
175 if (!indexes.empty())
176 {
177 std::vector<uint32_t>::const_iterator pos;
178 std::vector<uint32_t>::const_iterator end = indexes.end();
179 DumpSymbolHeader (s);
180 for (pos = indexes.begin(); pos != end; ++pos)
181 {
182 uint32_t idx = *pos;
183 if (idx < num_symbols)
184 {
185 s->Indent();
Greg Claytonf5e56de2010-09-14 23:36:40 +0000186 m_symbols[idx].Dump(s, target, idx);
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000187 }
188 }
189 }
190 s->IndentLess ();
191}
192
193void
194Symtab::DumpSymbolHeader (Stream *s)
195{
196 s->Indent(" Debug symbol\n");
197 s->Indent(" |Synthetic symbol\n");
198 s->Indent(" ||Externally Visible\n");
199 s->Indent(" |||\n");
200 s->Indent("Index UserID DSX Type File Address/Value Load Address Size Flags Name\n");
201 s->Indent("------- ------ --- ------------ ------------------ ------------------ ------------------ ---------- ----------------------------------\n");
202}
203
Greg Clayton49bd1c82010-09-07 17:36:17 +0000204
205static int
206CompareSymbolID (const void *key, const void *p)
207{
208 const user_id_t match_uid = *(user_id_t*) key;
209 const user_id_t symbol_uid = ((Symbol *)p)->GetID();
210 if (match_uid < symbol_uid)
211 return -1;
212 if (match_uid > symbol_uid)
213 return 1;
214 return 0;
215}
216
217Symbol *
218Symtab::FindSymbolByID (lldb::user_id_t symbol_uid) const
219{
Greg Clayton8087ca22010-10-08 04:20:14 +0000220 Mutex::Locker locker (m_mutex);
221
Greg Clayton49bd1c82010-09-07 17:36:17 +0000222 Symbol *symbol = (Symbol*)::bsearch (&symbol_uid,
223 &m_symbols[0],
224 m_symbols.size(),
Greg Clayton0c38b0d2010-09-12 05:25:16 +0000225 (uint8_t *)&m_symbols[1] - (uint8_t *)&m_symbols[0],
Greg Clayton49bd1c82010-09-07 17:36:17 +0000226 CompareSymbolID);
227 return symbol;
228}
229
230
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000231Symbol *
232Symtab::SymbolAtIndex(uint32_t idx)
233{
Greg Clayton8087ca22010-10-08 04:20:14 +0000234 // Clients should grab the mutex from this symbol table and lock it manually
235 // when calling this function to avoid performance issues.
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000236 if (idx < m_symbols.size())
237 return &m_symbols[idx];
238 return NULL;
239}
240
241
242const Symbol *
243Symtab::SymbolAtIndex(uint32_t idx) const
244{
Greg Clayton8087ca22010-10-08 04:20:14 +0000245 // Clients should grab the mutex from this symbol table and lock it manually
246 // when calling this function to avoid performance issues.
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000247 if (idx < m_symbols.size())
248 return &m_symbols[idx];
249 return NULL;
250}
251
252//----------------------------------------------------------------------
253// InitNameIndexes
254//----------------------------------------------------------------------
255void
256Symtab::InitNameIndexes()
257{
Greg Clayton8087ca22010-10-08 04:20:14 +0000258 // Protected function, no need to lock mutex...
259 if (!m_name_indexes_computed)
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000260 {
Greg Clayton8087ca22010-10-08 04:20:14 +0000261 m_name_indexes_computed = true;
262 Timer scoped_timer (__PRETTY_FUNCTION__, "%s", __PRETTY_FUNCTION__);
263 // Create the name index vector to be able to quickly search by name
264 const size_t count = m_symbols.size();
265 assert(m_objfile != NULL);
266 assert(m_objfile->GetModule() != NULL);
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000267
Greg Clayton8087ca22010-10-08 04:20:14 +0000268#if 1
269 m_name_to_index.Reserve (count);
270#else
271 // TODO: benchmark this to see if we save any memory. Otherwise we
272 // will always keep the memory reserved in the vector unless we pull
273 // some STL swap magic and then recopy...
274 uint32_t actual_count = 0;
275 for (const_iterator pos = m_symbols.begin(), end = m_symbols.end();
276 pos != end;
277 ++pos)
278 {
279 const Mangled &mangled = pos->GetMangled();
280 if (mangled.GetMangledName())
281 ++actual_count;
282
283 if (mangled.GetDemangledName())
284 ++actual_count;
285 }
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000286
Greg Clayton8087ca22010-10-08 04:20:14 +0000287 m_name_to_index.Reserve (actual_count);
288#endif
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000289
Greg Clayton8087ca22010-10-08 04:20:14 +0000290 UniqueCStringMap<uint32_t>::Entry entry;
291
292 for (entry.value = 0; entry.value < count; ++entry.value)
293 {
294 const Symbol *symbol = &m_symbols[entry.value];
295
296 // Don't let trampolines get into the lookup by name map
297 // If we ever need the trampoline symbols to be searchable by name
298 // we can remove this and then possibly add a new bool to any of the
299 // Symtab functions that lookup symbols by name to indicate if they
300 // want trampolines.
301 if (symbol->IsTrampoline())
302 continue;
303
304 const Mangled &mangled = symbol->GetMangled();
305 entry.cstring = mangled.GetMangledName().GetCString();
306 if (entry.cstring && entry.cstring[0])
307 m_name_to_index.Append (entry);
308
309 entry.cstring = mangled.GetDemangledName().GetCString();
310 if (entry.cstring && entry.cstring[0])
311 m_name_to_index.Append (entry);
Jim Inghamff5f5ff2011-08-15 01:32:22 +0000312
313 // If the demangled name turns out to be an ObjC name, and
314 // is a category name, add the version without categories to the index too.
315 ConstString objc_base_name;
316 if (ObjCLanguageRuntime::ParseMethodName (entry.cstring,
317 NULL,
318 NULL,
319 &objc_base_name)
320 && !objc_base_name.IsEmpty())
321 {
322 entry.cstring = objc_base_name.GetCString();
323 m_name_to_index.Append (entry);
324 }
325
Greg Clayton8087ca22010-10-08 04:20:14 +0000326 }
327 m_name_to_index.Sort();
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000328 }
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000329}
330
331uint32_t
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000332Symtab::AppendSymbolIndexesWithType (SymbolType symbol_type, std::vector<uint32_t>& indexes, uint32_t start_idx, uint32_t end_index) const
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000333{
Greg Clayton8087ca22010-10-08 04:20:14 +0000334 Mutex::Locker locker (m_mutex);
335
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000336 uint32_t prev_size = indexes.size();
337
338 const uint32_t count = std::min<uint32_t> (m_symbols.size(), end_index);
339
340 for (uint32_t i = start_idx; i < count; ++i)
341 {
342 if (symbol_type == eSymbolTypeAny || m_symbols[i].GetType() == symbol_type)
343 indexes.push_back(i);
344 }
345
346 return indexes.size() - prev_size;
347}
348
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000349uint32_t
Greg Clayton16b2d2b2011-01-20 06:08:59 +0000350Symtab::AppendSymbolIndexesWithTypeAndFlagsValue (SymbolType symbol_type, uint32_t flags_value, std::vector<uint32_t>& indexes, uint32_t start_idx, uint32_t end_index) const
351{
352 Mutex::Locker locker (m_mutex);
353
354 uint32_t prev_size = indexes.size();
355
356 const uint32_t count = std::min<uint32_t> (m_symbols.size(), end_index);
357
358 for (uint32_t i = start_idx; i < count; ++i)
359 {
360 if ((symbol_type == eSymbolTypeAny || m_symbols[i].GetType() == symbol_type) && m_symbols[i].GetFlags() == flags_value)
361 indexes.push_back(i);
362 }
363
364 return indexes.size() - prev_size;
365}
366
367uint32_t
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000368Symtab::AppendSymbolIndexesWithType (SymbolType symbol_type, Debug symbol_debug_type, Visibility symbol_visibility, std::vector<uint32_t>& indexes, uint32_t start_idx, uint32_t end_index) const
369{
Greg Clayton8087ca22010-10-08 04:20:14 +0000370 Mutex::Locker locker (m_mutex);
371
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000372 uint32_t prev_size = indexes.size();
373
374 const uint32_t count = std::min<uint32_t> (m_symbols.size(), end_index);
375
376 for (uint32_t i = start_idx; i < count; ++i)
377 {
378 if (symbol_type == eSymbolTypeAny || m_symbols[i].GetType() == symbol_type)
379 {
380 if (CheckSymbolAtIndex(i, symbol_debug_type, symbol_visibility))
381 indexes.push_back(i);
382 }
383 }
384
385 return indexes.size() - prev_size;
386}
387
388
389uint32_t
390Symtab::GetIndexForSymbol (const Symbol *symbol) const
391{
392 const Symbol *first_symbol = &m_symbols[0];
393 if (symbol >= first_symbol && symbol < first_symbol + m_symbols.size())
394 return symbol - first_symbol;
395 return UINT32_MAX;
396}
397
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000398struct SymbolSortInfo
399{
400 const bool sort_by_load_addr;
401 const Symbol *symbols;
402};
403
Owen Andersonc7da5f42010-06-16 17:34:05 +0000404namespace {
405 struct SymbolIndexComparator {
406 const std::vector<Symbol>& symbols;
407 SymbolIndexComparator(const std::vector<Symbol>& s) : symbols(s) { }
408 bool operator()(uint32_t index_a, uint32_t index_b) {
409 addr_t value_a;
410 addr_t value_b;
411 if (symbols[index_a].GetValue().GetSection() == symbols[index_b].GetValue().GetSection()) {
412 value_a = symbols[index_a].GetValue ().GetOffset();
413 value_b = symbols[index_b].GetValue ().GetOffset();
414 } else {
415 value_a = symbols[index_a].GetValue ().GetFileAddress();
416 value_b = symbols[index_b].GetValue ().GetFileAddress();
417 }
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000418
Owen Andersonc7da5f42010-06-16 17:34:05 +0000419 if (value_a == value_b) {
420 // The if the values are equal, use the original symbol user ID
421 lldb::user_id_t uid_a = symbols[index_a].GetID();
422 lldb::user_id_t uid_b = symbols[index_b].GetID();
423 if (uid_a < uid_b)
424 return true;
425 if (uid_a > uid_b)
426 return false;
427 return false;
428 } else if (value_a < value_b)
429 return true;
430
431 return false;
432 }
433 };
Eli Friedmana92e3322010-06-10 23:36:31 +0000434}
435
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000436void
437Symtab::SortSymbolIndexesByValue (std::vector<uint32_t>& indexes, bool remove_duplicates) const
438{
Greg Clayton8087ca22010-10-08 04:20:14 +0000439 Mutex::Locker locker (m_mutex);
440
Owen Andersonc7da5f42010-06-16 17:34:05 +0000441 Timer scoped_timer (__PRETTY_FUNCTION__,__PRETTY_FUNCTION__);
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000442 // No need to sort if we have zero or one items...
443 if (indexes.size() <= 1)
444 return;
445
Owen Anderson836af6b2010-06-17 00:51:12 +0000446 // Sort the indexes in place using std::stable_sort.
447 // NOTE: The use of std::stable_sort instead of std::sort here is strictly for performance,
448 // not correctness. The indexes vector tends to be "close" to sorted, which the
449 // stable sort handles better.
Owen Andersonc7da5f42010-06-16 17:34:05 +0000450 std::stable_sort(indexes.begin(), indexes.end(), SymbolIndexComparator(m_symbols));
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000451
452 // Remove any duplicates if requested
453 if (remove_duplicates)
454 std::unique(indexes.begin(), indexes.end());
455}
456
457uint32_t
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000458Symtab::AppendSymbolIndexesWithName (const ConstString& symbol_name, std::vector<uint32_t>& indexes)
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000459{
Greg Clayton8087ca22010-10-08 04:20:14 +0000460 Mutex::Locker locker (m_mutex);
461
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000462 Timer scoped_timer (__PRETTY_FUNCTION__, "%s", __PRETTY_FUNCTION__);
463 if (symbol_name)
464 {
Greg Clayton38e953d2011-09-11 00:20:09 +0000465 const char *symbol_cstr = symbol_name.GetCString();
Greg Clayton8087ca22010-10-08 04:20:14 +0000466 if (!m_name_indexes_computed)
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000467 InitNameIndexes();
468
Greg Clayton38e953d2011-09-11 00:20:09 +0000469 return m_name_to_index.GetValues (symbol_cstr, indexes);
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000470 }
471 return 0;
472}
473
474uint32_t
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000475Symtab::AppendSymbolIndexesWithName (const ConstString& symbol_name, Debug symbol_debug_type, Visibility symbol_visibility, std::vector<uint32_t>& indexes)
476{
Greg Clayton8087ca22010-10-08 04:20:14 +0000477 Mutex::Locker locker (m_mutex);
478
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000479 Timer scoped_timer (__PRETTY_FUNCTION__, "%s", __PRETTY_FUNCTION__);
480 if (symbol_name)
481 {
482 const size_t old_size = indexes.size();
Greg Clayton8087ca22010-10-08 04:20:14 +0000483 if (!m_name_indexes_computed)
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000484 InitNameIndexes();
485
486 const char *symbol_cstr = symbol_name.GetCString();
Greg Clayton38e953d2011-09-11 00:20:09 +0000487
488 std::vector<uint32_t> all_name_indexes;
489 const size_t name_match_count = m_name_to_index.GetValues (symbol_cstr, all_name_indexes);
490 for (size_t i=0; i<name_match_count; ++i)
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000491 {
Greg Clayton38e953d2011-09-11 00:20:09 +0000492 if (CheckSymbolAtIndex(all_name_indexes[i], symbol_debug_type, symbol_visibility))
493 indexes.push_back (all_name_indexes[i]);
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000494 }
495 return indexes.size() - old_size;
496 }
497 return 0;
498}
499
500uint32_t
501Symtab::AppendSymbolIndexesWithNameAndType (const ConstString& symbol_name, SymbolType symbol_type, std::vector<uint32_t>& indexes)
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000502{
Greg Clayton8087ca22010-10-08 04:20:14 +0000503 Mutex::Locker locker (m_mutex);
504
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000505 if (AppendSymbolIndexesWithName(symbol_name, indexes) > 0)
506 {
507 std::vector<uint32_t>::iterator pos = indexes.begin();
508 while (pos != indexes.end())
509 {
510 if (symbol_type == eSymbolTypeAny || m_symbols[*pos].GetType() == symbol_type)
511 ++pos;
512 else
513 indexes.erase(pos);
514 }
515 }
516 return indexes.size();
517}
518
519uint32_t
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000520Symtab::AppendSymbolIndexesWithNameAndType (const ConstString& symbol_name, SymbolType symbol_type, Debug symbol_debug_type, Visibility symbol_visibility, std::vector<uint32_t>& indexes)
521{
Greg Clayton8087ca22010-10-08 04:20:14 +0000522 Mutex::Locker locker (m_mutex);
523
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000524 if (AppendSymbolIndexesWithName(symbol_name, symbol_debug_type, symbol_visibility, indexes) > 0)
525 {
526 std::vector<uint32_t>::iterator pos = indexes.begin();
527 while (pos != indexes.end())
528 {
529 if (symbol_type == eSymbolTypeAny || m_symbols[*pos].GetType() == symbol_type)
530 ++pos;
531 else
532 indexes.erase(pos);
533 }
534 }
535 return indexes.size();
536}
537
538
539uint32_t
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000540Symtab::AppendSymbolIndexesMatchingRegExAndType (const RegularExpression &regexp, SymbolType symbol_type, std::vector<uint32_t>& indexes)
541{
Greg Clayton8087ca22010-10-08 04:20:14 +0000542 Mutex::Locker locker (m_mutex);
543
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000544 uint32_t prev_size = indexes.size();
545 uint32_t sym_end = m_symbols.size();
546
547 for (int i = 0; i < sym_end; i++)
548 {
549 if (symbol_type == eSymbolTypeAny || m_symbols[i].GetType() == symbol_type)
550 {
551 const char *name = m_symbols[i].GetMangled().GetName().AsCString();
552 if (name)
553 {
554 if (regexp.Execute (name))
555 indexes.push_back(i);
556 }
557 }
558 }
559 return indexes.size() - prev_size;
560
561}
562
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000563uint32_t
564Symtab::AppendSymbolIndexesMatchingRegExAndType (const RegularExpression &regexp, SymbolType symbol_type, Debug symbol_debug_type, Visibility symbol_visibility, std::vector<uint32_t>& indexes)
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000565{
Greg Clayton8087ca22010-10-08 04:20:14 +0000566 Mutex::Locker locker (m_mutex);
567
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000568 uint32_t prev_size = indexes.size();
569 uint32_t sym_end = m_symbols.size();
570
571 for (int i = 0; i < sym_end; i++)
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000572 {
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000573 if (symbol_type == eSymbolTypeAny || m_symbols[i].GetType() == symbol_type)
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000574 {
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000575 if (CheckSymbolAtIndex(i, symbol_debug_type, symbol_visibility) == false)
576 continue;
577
578 const char *name = m_symbols[i].GetMangled().GetName().AsCString();
579 if (name)
580 {
581 if (regexp.Execute (name))
582 indexes.push_back(i);
583 }
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000584 }
585 }
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000586 return indexes.size() - prev_size;
587
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000588}
589
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000590Symbol *
591Symtab::FindSymbolWithType (SymbolType symbol_type, Debug symbol_debug_type, Visibility symbol_visibility, uint32_t& start_idx)
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000592{
Greg Clayton8087ca22010-10-08 04:20:14 +0000593 Mutex::Locker locker (m_mutex);
594
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000595 const size_t count = m_symbols.size();
596 for (uint32_t idx = start_idx; idx < count; ++idx)
597 {
598 if (symbol_type == eSymbolTypeAny || m_symbols[idx].GetType() == symbol_type)
599 {
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000600 if (CheckSymbolAtIndex(idx, symbol_debug_type, symbol_visibility))
601 {
602 start_idx = idx;
603 return &m_symbols[idx];
604 }
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000605 }
606 }
607 return NULL;
608}
609
610size_t
611Symtab::FindAllSymbolsWithNameAndType (const ConstString &name, SymbolType symbol_type, std::vector<uint32_t>& symbol_indexes)
612{
Greg Clayton8087ca22010-10-08 04:20:14 +0000613 Mutex::Locker locker (m_mutex);
614
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000615 Timer scoped_timer (__PRETTY_FUNCTION__, "%s", __PRETTY_FUNCTION__);
616 // Initialize all of the lookup by name indexes before converting NAME
617 // to a uniqued string NAME_STR below.
Greg Clayton8087ca22010-10-08 04:20:14 +0000618 if (!m_name_indexes_computed)
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000619 InitNameIndexes();
620
621 if (name)
622 {
623 // The string table did have a string that matched, but we need
624 // to check the symbols and match the symbol_type if any was given.
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000625 AppendSymbolIndexesWithNameAndType (name, symbol_type, symbol_indexes);
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000626 }
627 return symbol_indexes.size();
628}
629
630size_t
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000631Symtab::FindAllSymbolsWithNameAndType (const ConstString &name, SymbolType symbol_type, Debug symbol_debug_type, Visibility symbol_visibility, std::vector<uint32_t>& symbol_indexes)
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000632{
Greg Clayton8087ca22010-10-08 04:20:14 +0000633 Mutex::Locker locker (m_mutex);
634
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000635 Timer scoped_timer (__PRETTY_FUNCTION__, "%s", __PRETTY_FUNCTION__);
636 // Initialize all of the lookup by name indexes before converting NAME
637 // to a uniqued string NAME_STR below.
Greg Clayton8087ca22010-10-08 04:20:14 +0000638 if (!m_name_indexes_computed)
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000639 InitNameIndexes();
640
641 if (name)
642 {
643 // The string table did have a string that matched, but we need
644 // to check the symbols and match the symbol_type if any was given.
645 AppendSymbolIndexesWithNameAndType (name, symbol_type, symbol_debug_type, symbol_visibility, symbol_indexes);
646 }
647 return symbol_indexes.size();
648}
649
650size_t
651Symtab::FindAllSymbolsMatchingRexExAndType (const RegularExpression &regex, SymbolType symbol_type, Debug symbol_debug_type, Visibility symbol_visibility, std::vector<uint32_t>& symbol_indexes)
652{
Greg Clayton8087ca22010-10-08 04:20:14 +0000653 Mutex::Locker locker (m_mutex);
654
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000655 AppendSymbolIndexesMatchingRegExAndType(regex, symbol_type, symbol_debug_type, symbol_visibility, symbol_indexes);
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000656 return symbol_indexes.size();
657}
658
659Symbol *
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000660Symtab::FindFirstSymbolWithNameAndType (const ConstString &name, SymbolType symbol_type, Debug symbol_debug_type, Visibility symbol_visibility)
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000661{
Greg Clayton8087ca22010-10-08 04:20:14 +0000662 Mutex::Locker locker (m_mutex);
663
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000664 Timer scoped_timer (__PRETTY_FUNCTION__, "%s", __PRETTY_FUNCTION__);
Greg Clayton8087ca22010-10-08 04:20:14 +0000665 if (!m_name_indexes_computed)
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000666 InitNameIndexes();
667
668 if (name)
669 {
670 std::vector<uint32_t> matching_indexes;
671 // The string table did have a string that matched, but we need
672 // to check the symbols and match the symbol_type if any was given.
Greg Claytonbcf2cfb2010-09-11 03:13:28 +0000673 if (AppendSymbolIndexesWithNameAndType (name, symbol_type, symbol_debug_type, symbol_visibility, matching_indexes))
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000674 {
675 std::vector<uint32_t>::const_iterator pos, end = matching_indexes.end();
676 for (pos = matching_indexes.begin(); pos != end; ++pos)
677 {
678 Symbol *symbol = SymbolAtIndex(*pos);
679
680 if (symbol->Compare(name, symbol_type))
681 return symbol;
682 }
683 }
684 }
685 return NULL;
686}
687
688typedef struct
689{
690 const Symtab *symtab;
691 const addr_t file_addr;
692 Symbol *match_symbol;
693 const uint32_t *match_index_ptr;
694 addr_t match_offset;
695} SymbolSearchInfo;
696
697static int
698SymbolWithFileAddress (SymbolSearchInfo *info, const uint32_t *index_ptr)
699{
700 const Symbol *curr_symbol = info->symtab->SymbolAtIndex (index_ptr[0]);
701 if (curr_symbol == NULL)
702 return -1;
703
704 const addr_t info_file_addr = info->file_addr;
705
706 // lldb::Symbol::GetAddressRangePtr() will only return a non NULL address
707 // range if the symbol has a section!
708 const AddressRange *curr_range = curr_symbol->GetAddressRangePtr();
709 if (curr_range)
710 {
711 const addr_t curr_file_addr = curr_range->GetBaseAddress().GetFileAddress();
712 if (info_file_addr < curr_file_addr)
713 return -1;
714 if (info_file_addr > curr_file_addr)
715 return +1;
716 info->match_symbol = const_cast<Symbol *>(curr_symbol);
717 info->match_index_ptr = index_ptr;
718 return 0;
719 }
720
721 return -1;
722}
723
724static int
725SymbolWithClosestFileAddress (SymbolSearchInfo *info, const uint32_t *index_ptr)
726{
727 const Symbol *symbol = info->symtab->SymbolAtIndex (index_ptr[0]);
728 if (symbol == NULL)
729 return -1;
730
731 const addr_t info_file_addr = info->file_addr;
732 const AddressRange *curr_range = symbol->GetAddressRangePtr();
733 if (curr_range)
734 {
735 const addr_t curr_file_addr = curr_range->GetBaseAddress().GetFileAddress();
736 if (info_file_addr < curr_file_addr)
737 return -1;
738
739 // Since we are finding the closest symbol that is greater than or equal
740 // to 'info->file_addr' we set the symbol here. This will get set
741 // multiple times, but after the search is done it will contain the best
742 // symbol match
743 info->match_symbol = const_cast<Symbol *>(symbol);
744 info->match_index_ptr = index_ptr;
745 info->match_offset = info_file_addr - curr_file_addr;
746
747 if (info_file_addr > curr_file_addr)
748 return +1;
749 return 0;
750 }
751 return -1;
752}
753
754static SymbolSearchInfo
755FindIndexPtrForSymbolContainingAddress(Symtab* symtab, addr_t file_addr, const uint32_t* indexes, uint32_t num_indexes)
756{
757 SymbolSearchInfo info = { symtab, file_addr, NULL, NULL, 0 };
Greg Claytone0d378b2011-03-24 21:19:54 +0000758 ::bsearch (&info,
759 indexes,
760 num_indexes,
761 sizeof(uint32_t),
762 (ComparisonFunction)SymbolWithClosestFileAddress);
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000763 return info;
764}
765
766
767void
768Symtab::InitAddressIndexes()
769{
Greg Clayton8087ca22010-10-08 04:20:14 +0000770 // Protected function, no need to lock mutex...
771 if (!m_addr_indexes_computed && !m_symbols.empty())
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000772 {
Greg Clayton8087ca22010-10-08 04:20:14 +0000773 m_addr_indexes_computed = true;
774#if 0
775 // The old was to add only code, trampoline or data symbols...
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000776 AppendSymbolIndexesWithType (eSymbolTypeCode, m_addr_indexes);
777 AppendSymbolIndexesWithType (eSymbolTypeTrampoline, m_addr_indexes);
778 AppendSymbolIndexesWithType (eSymbolTypeData, m_addr_indexes);
Greg Clayton8087ca22010-10-08 04:20:14 +0000779#else
780 // The new way adds all symbols with valid addresses that are section
781 // offset.
782 const_iterator begin = m_symbols.begin();
783 const_iterator end = m_symbols.end();
784 for (const_iterator pos = m_symbols.begin(); pos != end; ++pos)
785 {
786 if (pos->GetAddressRangePtr())
787 m_addr_indexes.push_back (std::distance(begin, pos));
788 }
789#endif
790 SortSymbolIndexesByValue (m_addr_indexes, false);
791 m_addr_indexes.push_back (UINT32_MAX); // Terminator for bsearch since we might need to look at the next symbol
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000792 }
793}
794
795size_t
796Symtab::CalculateSymbolSize (Symbol *symbol)
797{
Greg Clayton8087ca22010-10-08 04:20:14 +0000798 Mutex::Locker locker (m_mutex);
799
Greg Clayton471b31c2010-07-20 22:52:08 +0000800 if (m_symbols.empty())
801 return 0;
802
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000803 // Make sure this symbol is from this symbol table...
Greg Clayton471b31c2010-07-20 22:52:08 +0000804 if (symbol < &m_symbols.front() || symbol > &m_symbols.back())
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000805 return 0;
806
807 // See if this symbol already has a byte size?
808 size_t byte_size = symbol->GetByteSize();
809
810 if (byte_size)
811 {
812 // It does, just return it
813 return byte_size;
814 }
815
816 // Else if this is an address based symbol, figure out the delta between
817 // it and the next address based symbol
818 if (symbol->GetAddressRangePtr())
819 {
Greg Clayton8087ca22010-10-08 04:20:14 +0000820 if (!m_addr_indexes_computed)
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000821 InitAddressIndexes();
822 const size_t num_addr_indexes = m_addr_indexes.size();
Greg Clayton471b31c2010-07-20 22:52:08 +0000823 SymbolSearchInfo info = FindIndexPtrForSymbolContainingAddress(this, symbol->GetAddressRangePtr()->GetBaseAddress().GetFileAddress(), &m_addr_indexes.front(), num_addr_indexes);
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000824 if (info.match_index_ptr != NULL)
825 {
826 const lldb::addr_t curr_file_addr = symbol->GetAddressRangePtr()->GetBaseAddress().GetFileAddress();
827 // We can figure out the address range of all symbols except the
828 // last one by taking the delta between the current symbol and
829 // the next symbol
830
Greg Clayton471b31c2010-07-20 22:52:08 +0000831 for (uint32_t addr_index = info.match_index_ptr - &m_addr_indexes.front() + 1;
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000832 addr_index < num_addr_indexes;
833 ++addr_index)
834 {
835 Symbol *next_symbol = SymbolAtIndex(m_addr_indexes[addr_index]);
836 if (next_symbol == NULL)
837 break;
838
839 assert (next_symbol->GetAddressRangePtr());
840 const lldb::addr_t next_file_addr = next_symbol->GetAddressRangePtr()->GetBaseAddress().GetFileAddress();
841 if (next_file_addr > curr_file_addr)
842 {
843 byte_size = next_file_addr - curr_file_addr;
844 symbol->GetAddressRangePtr()->SetByteSize(byte_size);
845 symbol->SetSizeIsSynthesized(true);
846 break;
847 }
848 }
849 }
850 }
851 return byte_size;
852}
853
854Symbol *
855Symtab::FindSymbolWithFileAddress (addr_t file_addr)
856{
Greg Clayton8087ca22010-10-08 04:20:14 +0000857 Mutex::Locker locker (m_mutex);
858
859 if (!m_addr_indexes_computed)
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000860 InitAddressIndexes();
861
862 SymbolSearchInfo info = { this, file_addr, NULL, NULL, 0 };
863
Greg Claytone0d378b2011-03-24 21:19:54 +0000864 uint32_t* match = (uint32_t*)::bsearch (&info,
865 &m_addr_indexes[0],
866 m_addr_indexes.size(),
867 sizeof(uint32_t),
868 (ComparisonFunction)SymbolWithFileAddress);
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000869 if (match)
870 return SymbolAtIndex (*match);
871 return NULL;
872}
873
874
875Symbol *
876Symtab::FindSymbolContainingFileAddress (addr_t file_addr, const uint32_t* indexes, uint32_t num_indexes)
877{
Greg Clayton8087ca22010-10-08 04:20:14 +0000878 Mutex::Locker locker (m_mutex);
879
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000880 SymbolSearchInfo info = { this, file_addr, NULL, NULL, 0 };
881
Greg Claytone0d378b2011-03-24 21:19:54 +0000882 ::bsearch (&info,
883 indexes,
884 num_indexes,
885 sizeof(uint32_t),
886 (ComparisonFunction)SymbolWithClosestFileAddress);
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000887
888 if (info.match_symbol)
889 {
Greg Clayton89411422010-10-08 00:21:05 +0000890 if (info.match_offset == 0)
891 {
892 // We found an exact match!
893 return info.match_symbol;
894 }
895
896 const size_t symbol_byte_size = CalculateSymbolSize(info.match_symbol);
897
898 if (symbol_byte_size == 0)
899 {
900 // We weren't able to find the size of the symbol so lets just go
901 // with that match we found in our search...
902 return info.match_symbol;
903 }
904
905 // We were able to figure out a symbol size so lets make sure our
906 // offset puts "file_addr" in the symbol's address range.
907 if (info.match_offset < symbol_byte_size)
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000908 return info.match_symbol;
909 }
910 return NULL;
911}
912
913Symbol *
914Symtab::FindSymbolContainingFileAddress (addr_t file_addr)
915{
Greg Clayton8087ca22010-10-08 04:20:14 +0000916 Mutex::Locker locker (m_mutex);
917
918 if (!m_addr_indexes_computed)
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000919 InitAddressIndexes();
920
921 return FindSymbolContainingFileAddress (file_addr, &m_addr_indexes[0], m_addr_indexes.size());
922}
923