425 lines
		
	
	
		
			15 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			425 lines
		
	
	
		
			15 KiB
		
	
	
	
		
			C++
		
	
	
	
| //===-- BreakpointResolverName.cpp ------------------------------*- C++ -*-===//
 | |
| //
 | |
| //                     The LLVM Compiler Infrastructure
 | |
| //
 | |
| // This file is distributed under the University of Illinois Open Source
 | |
| // License. See LICENSE.TXT for details.
 | |
| //
 | |
| //===----------------------------------------------------------------------===//
 | |
| 
 | |
| // C Includes
 | |
| // C++ Includes
 | |
| // Other libraries and framework includes
 | |
| // Project includes
 | |
| #include "lldb/Breakpoint/BreakpointResolverName.h"
 | |
| 
 | |
| #include "Plugins/Language/CPlusPlus/CPlusPlusLanguage.h"
 | |
| #include "Plugins/Language/ObjC/ObjCLanguage.h"
 | |
| #include "lldb/Breakpoint/BreakpointLocation.h"
 | |
| #include "lldb/Core/Module.h"
 | |
| #include "lldb/Symbol/Block.h"
 | |
| #include "lldb/Symbol/Function.h"
 | |
| #include "lldb/Symbol/Symbol.h"
 | |
| #include "lldb/Symbol/SymbolContext.h"
 | |
| #include "lldb/Utility/Log.h"
 | |
| #include "lldb/Utility/StreamString.h"
 | |
| 
 | |
| using namespace lldb;
 | |
| using namespace lldb_private;
 | |
| 
 | |
| BreakpointResolverName::BreakpointResolverName(
 | |
|     Breakpoint *bkpt, const char *name_cstr, uint32_t name_type_mask,
 | |
|     LanguageType language, Breakpoint::MatchType type, lldb::addr_t offset,
 | |
|     bool skip_prologue)
 | |
|     : BreakpointResolver(bkpt, BreakpointResolver::NameResolver, offset),
 | |
|       m_class_name(), m_regex(), m_match_type(type), m_language(language),
 | |
|       m_skip_prologue(skip_prologue) {
 | |
|   if (m_match_type == Breakpoint::Regexp) {
 | |
|     if (!m_regex.Compile(llvm::StringRef::withNullAsEmpty(name_cstr))) {
 | |
|       Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_BREAKPOINTS));
 | |
| 
 | |
|       if (log)
 | |
|         log->Warning("function name regexp: \"%s\" did not compile.",
 | |
|                      name_cstr);
 | |
|     }
 | |
|   } else {
 | |
|     AddNameLookup(ConstString(name_cstr), name_type_mask);
 | |
|   }
 | |
| }
 | |
| 
 | |
| BreakpointResolverName::BreakpointResolverName(
 | |
|     Breakpoint *bkpt, const char *names[], size_t num_names,
 | |
|     uint32_t name_type_mask, LanguageType language, lldb::addr_t offset,
 | |
|     bool skip_prologue)
 | |
|     : BreakpointResolver(bkpt, BreakpointResolver::NameResolver, offset),
 | |
|       m_match_type(Breakpoint::Exact), m_language(language),
 | |
|       m_skip_prologue(skip_prologue) {
 | |
|   for (size_t i = 0; i < num_names; i++) {
 | |
|     AddNameLookup(ConstString(names[i]), name_type_mask);
 | |
|   }
 | |
| }
 | |
| 
 | |
| BreakpointResolverName::BreakpointResolverName(
 | |
|     Breakpoint *bkpt, std::vector<std::string> names, uint32_t name_type_mask,
 | |
|     LanguageType language, lldb::addr_t offset, bool skip_prologue)
 | |
|     : BreakpointResolver(bkpt, BreakpointResolver::NameResolver, offset),
 | |
|       m_match_type(Breakpoint::Exact), m_language(language),
 | |
|       m_skip_prologue(skip_prologue) {
 | |
|   for (const std::string &name : names) {
 | |
|     AddNameLookup(ConstString(name.c_str(), name.size()), name_type_mask);
 | |
|   }
 | |
| }
 | |
| 
 | |
| BreakpointResolverName::BreakpointResolverName(Breakpoint *bkpt,
 | |
|                                                RegularExpression &func_regex,
 | |
|                                                lldb::LanguageType language,
 | |
|                                                lldb::addr_t offset,
 | |
|                                                bool skip_prologue)
 | |
|     : BreakpointResolver(bkpt, BreakpointResolver::NameResolver, offset),
 | |
|       m_class_name(nullptr), m_regex(func_regex),
 | |
|       m_match_type(Breakpoint::Regexp), m_language(language),
 | |
|       m_skip_prologue(skip_prologue) {}
 | |
| 
 | |
| BreakpointResolverName::~BreakpointResolverName() = default;
 | |
| 
 | |
| BreakpointResolverName::BreakpointResolverName(
 | |
|     const BreakpointResolverName &rhs)
 | |
|     : BreakpointResolver(rhs.m_breakpoint, BreakpointResolver::NameResolver,
 | |
|                          rhs.m_offset),
 | |
|       m_lookups(rhs.m_lookups), m_class_name(rhs.m_class_name),
 | |
|       m_regex(rhs.m_regex), m_match_type(rhs.m_match_type),
 | |
|       m_language(rhs.m_language), m_skip_prologue(rhs.m_skip_prologue) {}
 | |
| 
 | |
| BreakpointResolver *BreakpointResolverName::CreateFromStructuredData(
 | |
|     Breakpoint *bkpt, const StructuredData::Dictionary &options_dict,
 | |
|     Status &error) {
 | |
|   LanguageType language = eLanguageTypeUnknown;
 | |
|   llvm::StringRef language_name;
 | |
|   bool success = options_dict.GetValueForKeyAsString(
 | |
|       GetKey(OptionNames::LanguageName), language_name);
 | |
|   if (success) {
 | |
|     language = Language::GetLanguageTypeFromString(language_name);
 | |
|     if (language == eLanguageTypeUnknown) {
 | |
|       error.SetErrorStringWithFormatv("BRN::CFSD: Unknown language: {0}.",
 | |
|                                       language_name);
 | |
|       return nullptr;
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   lldb::addr_t offset = 0;
 | |
|   success =
 | |
|       options_dict.GetValueForKeyAsInteger(GetKey(OptionNames::Offset), offset);
 | |
|   if (!success) {
 | |
|     error.SetErrorStringWithFormat("BRN::CFSD: Missing offset entry.");
 | |
|     return nullptr;
 | |
|   }
 | |
| 
 | |
|   bool skip_prologue;
 | |
|   success = options_dict.GetValueForKeyAsBoolean(
 | |
|       GetKey(OptionNames::SkipPrologue), skip_prologue);
 | |
|   if (!success) {
 | |
|     error.SetErrorStringWithFormat("BRN::CFSD: Missing Skip prologue entry.");
 | |
|     return nullptr;
 | |
|   }
 | |
| 
 | |
|   llvm::StringRef regex_text;
 | |
|   success = options_dict.GetValueForKeyAsString(
 | |
|       GetKey(OptionNames::RegexString), regex_text);
 | |
|   if (success) {
 | |
|     RegularExpression regex(regex_text);
 | |
|     return new BreakpointResolverName(bkpt, regex, language, offset,
 | |
|                                       skip_prologue);
 | |
|   } else {
 | |
|     StructuredData::Array *names_array;
 | |
|     success = options_dict.GetValueForKeyAsArray(
 | |
|         GetKey(OptionNames::SymbolNameArray), names_array);
 | |
|     if (!success) {
 | |
|       error.SetErrorStringWithFormat("BRN::CFSD: Missing symbol names entry.");
 | |
|       return nullptr;
 | |
|     }
 | |
|     StructuredData::Array *names_mask_array;
 | |
|     success = options_dict.GetValueForKeyAsArray(
 | |
|         GetKey(OptionNames::NameMaskArray), names_mask_array);
 | |
|     if (!success) {
 | |
|       error.SetErrorStringWithFormat(
 | |
|           "BRN::CFSD: Missing symbol names mask entry.");
 | |
|       return nullptr;
 | |
|     }
 | |
| 
 | |
|     size_t num_elem = names_array->GetSize();
 | |
|     if (num_elem != names_mask_array->GetSize()) {
 | |
|       error.SetErrorString(
 | |
|           "BRN::CFSD: names and names mask arrays have different sizes.");
 | |
|       return nullptr;
 | |
|     }
 | |
| 
 | |
|     if (num_elem == 0) {
 | |
|       error.SetErrorString(
 | |
|           "BRN::CFSD: no name entry in a breakpoint by name breakpoint.");
 | |
|       return nullptr;
 | |
|     }
 | |
|     std::vector<std::string> names;
 | |
|     std::vector<uint32_t> name_masks;
 | |
|     for (size_t i = 0; i < num_elem; i++) {
 | |
|       uint32_t name_mask;
 | |
|       llvm::StringRef name;
 | |
| 
 | |
|       success = names_array->GetItemAtIndexAsString(i, name);
 | |
|       if (!success) {
 | |
|         error.SetErrorString("BRN::CFSD: name entry is not a string.");
 | |
|         return nullptr;
 | |
|       }
 | |
|       success = names_mask_array->GetItemAtIndexAsInteger(i, name_mask);
 | |
|       if (!success) {
 | |
|         error.SetErrorString("BRN::CFSD: name mask entry is not an integer.");
 | |
|         return nullptr;
 | |
|       }
 | |
|       names.push_back(name);
 | |
|       name_masks.push_back(name_mask);
 | |
|     }
 | |
| 
 | |
|     BreakpointResolverName *resolver = new BreakpointResolverName(
 | |
|         bkpt, names[0].c_str(), name_masks[0], language,
 | |
|         Breakpoint::MatchType::Exact, offset, skip_prologue);
 | |
|     for (size_t i = 1; i < num_elem; i++) {
 | |
|       resolver->AddNameLookup(ConstString(names[i]), name_masks[i]);
 | |
|     }
 | |
|     return resolver;
 | |
|   }
 | |
| }
 | |
| 
 | |
| StructuredData::ObjectSP BreakpointResolverName::SerializeToStructuredData() {
 | |
|   StructuredData::DictionarySP options_dict_sp(
 | |
|       new StructuredData::Dictionary());
 | |
| 
 | |
|   if (m_regex.IsValid()) {
 | |
|     options_dict_sp->AddStringItem(GetKey(OptionNames::RegexString),
 | |
|                                    m_regex.GetText());
 | |
|   } else {
 | |
|     StructuredData::ArraySP names_sp(new StructuredData::Array());
 | |
|     StructuredData::ArraySP name_masks_sp(new StructuredData::Array());
 | |
|     for (auto lookup : m_lookups) {
 | |
|       names_sp->AddItem(StructuredData::StringSP(
 | |
|           new StructuredData::String(lookup.GetName().AsCString())));
 | |
|       name_masks_sp->AddItem(StructuredData::IntegerSP(
 | |
|           new StructuredData::Integer(lookup.GetNameTypeMask())));
 | |
|     }
 | |
|     options_dict_sp->AddItem(GetKey(OptionNames::SymbolNameArray), names_sp);
 | |
|     options_dict_sp->AddItem(GetKey(OptionNames::NameMaskArray), name_masks_sp);
 | |
|   }
 | |
|   if (m_language != eLanguageTypeUnknown)
 | |
|     options_dict_sp->AddStringItem(
 | |
|         GetKey(OptionNames::LanguageName),
 | |
|         Language::GetNameForLanguageType(m_language));
 | |
|   options_dict_sp->AddBooleanItem(GetKey(OptionNames::SkipPrologue),
 | |
|                                   m_skip_prologue);
 | |
| 
 | |
|   return WrapOptionsDict(options_dict_sp);
 | |
| }
 | |
| 
 | |
| void BreakpointResolverName::AddNameLookup(const ConstString &name,
 | |
|                                            uint32_t name_type_mask) {
 | |
|   ObjCLanguage::MethodName objc_method(name.GetCString(), false);
 | |
|   if (objc_method.IsValid(false)) {
 | |
|     std::vector<ConstString> objc_names;
 | |
|     objc_method.GetFullNames(objc_names, true);
 | |
|     for (ConstString objc_name : objc_names) {
 | |
|       Module::LookupInfo lookup;
 | |
|       lookup.SetName(name);
 | |
|       lookup.SetLookupName(objc_name);
 | |
|       lookup.SetNameTypeMask(eFunctionNameTypeFull);
 | |
|       m_lookups.push_back(lookup);
 | |
|     }
 | |
|   } else {
 | |
|     Module::LookupInfo lookup(name, name_type_mask, m_language);
 | |
|     m_lookups.push_back(lookup);
 | |
|   }
 | |
| }
 | |
| 
 | |
| // FIXME: Right now we look at the module level, and call the module's
 | |
| // "FindFunctions".
 | |
| // Greg says he will add function tables, maybe at the CompileUnit level to
 | |
| // accelerate function
 | |
| // lookup.  At that point, we should switch the depth to CompileUnit, and look
 | |
| // in these tables.
 | |
| 
 | |
| Searcher::CallbackReturn
 | |
| BreakpointResolverName::SearchCallback(SearchFilter &filter,
 | |
|                                        SymbolContext &context, Address *addr,
 | |
|                                        bool containing) {
 | |
|   SymbolContextList func_list;
 | |
|   // SymbolContextList sym_list;
 | |
| 
 | |
|   uint32_t i;
 | |
|   bool new_location;
 | |
|   Address break_addr;
 | |
|   assert(m_breakpoint != nullptr);
 | |
| 
 | |
|   Log *log(lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_BREAKPOINTS));
 | |
| 
 | |
|   if (m_class_name) {
 | |
|     if (log)
 | |
|       log->Warning("Class/method function specification not supported yet.\n");
 | |
|     return Searcher::eCallbackReturnStop;
 | |
|   }
 | |
|   bool filter_by_cu =
 | |
|       (filter.GetFilterRequiredItems() & eSymbolContextCompUnit) != 0;
 | |
|   bool filter_by_language = (m_language != eLanguageTypeUnknown);
 | |
|   const bool include_symbols = !filter_by_cu;
 | |
|   const bool include_inlines = true;
 | |
|   const bool append = true;
 | |
| 
 | |
|   switch (m_match_type) {
 | |
|   case Breakpoint::Exact:
 | |
|     if (context.module_sp) {
 | |
|       for (const auto &lookup : m_lookups) {
 | |
|         const size_t start_func_idx = func_list.GetSize();
 | |
|         context.module_sp->FindFunctions(
 | |
|             lookup.GetLookupName(), nullptr, lookup.GetNameTypeMask(),
 | |
|             include_symbols, include_inlines, append, func_list);
 | |
| 
 | |
|         const size_t end_func_idx = func_list.GetSize();
 | |
| 
 | |
|         if (start_func_idx < end_func_idx)
 | |
|           lookup.Prune(func_list, start_func_idx);
 | |
|       }
 | |
|     }
 | |
|     break;
 | |
|   case Breakpoint::Regexp:
 | |
|     if (context.module_sp) {
 | |
|       context.module_sp->FindFunctions(
 | |
|           m_regex,
 | |
|           !filter_by_cu, // include symbols only if we aren't filtering by CU
 | |
|           include_inlines, append, func_list);
 | |
|     }
 | |
|     break;
 | |
|   case Breakpoint::Glob:
 | |
|     if (log)
 | |
|       log->Warning("glob is not supported yet.");
 | |
|     break;
 | |
|   }
 | |
| 
 | |
|   // If the filter specifies a Compilation Unit, remove the ones that don't pass
 | |
|   // at this point.
 | |
|   if (filter_by_cu || filter_by_language) {
 | |
|     uint32_t num_functions = func_list.GetSize();
 | |
| 
 | |
|     for (size_t idx = 0; idx < num_functions; idx++) {
 | |
|       bool remove_it = false;
 | |
|       SymbolContext sc;
 | |
|       func_list.GetContextAtIndex(idx, sc);
 | |
|       if (filter_by_cu) {
 | |
|         if (!sc.comp_unit || !filter.CompUnitPasses(*sc.comp_unit))
 | |
|           remove_it = true;
 | |
|       }
 | |
| 
 | |
|       if (filter_by_language) {
 | |
|         LanguageType sym_language = sc.GetLanguage();
 | |
|         if ((Language::GetPrimaryLanguage(sym_language) !=
 | |
|              Language::GetPrimaryLanguage(m_language)) &&
 | |
|             (sym_language != eLanguageTypeUnknown)) {
 | |
|           remove_it = true;
 | |
|         }
 | |
|       }
 | |
| 
 | |
|       if (remove_it) {
 | |
|         func_list.RemoveContextAtIndex(idx);
 | |
|         num_functions--;
 | |
|         idx--;
 | |
|       }
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   // Remove any duplicates between the function list and the symbol list
 | |
|   SymbolContext sc;
 | |
|   if (func_list.GetSize()) {
 | |
|     for (i = 0; i < func_list.GetSize(); i++) {
 | |
|       if (func_list.GetContextAtIndex(i, sc)) {
 | |
|         bool is_reexported = false;
 | |
| 
 | |
|         if (sc.block && sc.block->GetInlinedFunctionInfo()) {
 | |
|           if (!sc.block->GetStartAddress(break_addr))
 | |
|             break_addr.Clear();
 | |
|         } else if (sc.function) {
 | |
|           break_addr = sc.function->GetAddressRange().GetBaseAddress();
 | |
|           if (m_skip_prologue && break_addr.IsValid()) {
 | |
|             const uint32_t prologue_byte_size =
 | |
|                 sc.function->GetPrologueByteSize();
 | |
|             if (prologue_byte_size)
 | |
|               break_addr.SetOffset(break_addr.GetOffset() + prologue_byte_size);
 | |
|           }
 | |
|         } else if (sc.symbol) {
 | |
|           if (sc.symbol->GetType() == eSymbolTypeReExported) {
 | |
|             const Symbol *actual_symbol =
 | |
|                 sc.symbol->ResolveReExportedSymbol(m_breakpoint->GetTarget());
 | |
|             if (actual_symbol) {
 | |
|               is_reexported = true;
 | |
|               break_addr = actual_symbol->GetAddress();
 | |
|             }
 | |
|           } else {
 | |
|             break_addr = sc.symbol->GetAddress();
 | |
|           }
 | |
| 
 | |
|           if (m_skip_prologue && break_addr.IsValid()) {
 | |
|             const uint32_t prologue_byte_size =
 | |
|                 sc.symbol->GetPrologueByteSize();
 | |
|             if (prologue_byte_size)
 | |
|               break_addr.SetOffset(break_addr.GetOffset() + prologue_byte_size);
 | |
|           }
 | |
|         }
 | |
| 
 | |
|         if (break_addr.IsValid()) {
 | |
|           if (filter.AddressPasses(break_addr)) {
 | |
|             BreakpointLocationSP bp_loc_sp(
 | |
|                 AddLocation(break_addr, &new_location));
 | |
|             bp_loc_sp->SetIsReExported(is_reexported);
 | |
|             if (bp_loc_sp && new_location && !m_breakpoint->IsInternal()) {
 | |
|               if (log) {
 | |
|                 StreamString s;
 | |
|                 bp_loc_sp->GetDescription(&s, lldb::eDescriptionLevelVerbose);
 | |
|                 log->Printf("Added location: %s\n", s.GetData());
 | |
|               }
 | |
|             }
 | |
|           }
 | |
|         }
 | |
|       }
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   return Searcher::eCallbackReturnContinue;
 | |
| }
 | |
| 
 | |
| Searcher::Depth BreakpointResolverName::GetDepth() {
 | |
|   return Searcher::eDepthModule;
 | |
| }
 | |
| 
 | |
| void BreakpointResolverName::GetDescription(Stream *s) {
 | |
|   if (m_match_type == Breakpoint::Regexp)
 | |
|     s->Printf("regex = '%s'", m_regex.GetText().str().c_str());
 | |
|   else {
 | |
|     size_t num_names = m_lookups.size();
 | |
|     if (num_names == 1)
 | |
|       s->Printf("name = '%s'", m_lookups[0].GetName().GetCString());
 | |
|     else {
 | |
|       s->Printf("names = {");
 | |
|       for (size_t i = 0; i < num_names; i++) {
 | |
|         s->Printf("%s'%s'", (i == 0 ? "" : ", "),
 | |
|                   m_lookups[i].GetName().GetCString());
 | |
|       }
 | |
|       s->Printf("}");
 | |
|     }
 | |
|   }
 | |
|   if (m_language != eLanguageTypeUnknown) {
 | |
|     s->Printf(", language = %s", Language::GetNameForLanguageType(m_language));
 | |
|   }
 | |
| }
 | |
| 
 | |
| void BreakpointResolverName::Dump(Stream *s) const {}
 | |
| 
 | |
| lldb::BreakpointResolverSP
 | |
| BreakpointResolverName::CopyForBreakpoint(Breakpoint &breakpoint) {
 | |
|   lldb::BreakpointResolverSP ret_sp(new BreakpointResolverName(*this));
 | |
|   ret_sp->SetBreakpoint(&breakpoint);
 | |
|   return ret_sp;
 | |
| }
 |