89 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			89 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			C++
		
	
	
	
| //===-- NameToDIE.cpp -----------------------------------------------------===//
 | |
| //
 | |
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 | |
| // See https://llvm.org/LICENSE.txt for license information.
 | |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 | |
| //
 | |
| //===----------------------------------------------------------------------===//
 | |
| 
 | |
| #include "NameToDIE.h"
 | |
| #include "DWARFUnit.h"
 | |
| #include "lldb/Symbol/ObjectFile.h"
 | |
| #include "lldb/Utility/ConstString.h"
 | |
| #include "lldb/Utility/RegularExpression.h"
 | |
| #include "lldb/Utility/Stream.h"
 | |
| #include "lldb/Utility/StreamString.h"
 | |
| 
 | |
| using namespace lldb;
 | |
| using namespace lldb_private;
 | |
| 
 | |
| void NameToDIE::Finalize() {
 | |
|   m_map.Sort();
 | |
|   m_map.SizeToFit();
 | |
| }
 | |
| 
 | |
| void NameToDIE::Insert(ConstString name, const DIERef &die_ref) {
 | |
|   m_map.Append(name, die_ref);
 | |
| }
 | |
| 
 | |
| bool NameToDIE::Find(ConstString name,
 | |
|                      llvm::function_ref<bool(DIERef ref)> callback) const {
 | |
|   for (const auto &entry : m_map.equal_range(name))
 | |
|     if (!callback(entry.value))
 | |
|       return false;
 | |
|   return true;
 | |
| }
 | |
| 
 | |
| bool NameToDIE::Find(const RegularExpression ®ex,
 | |
|                      llvm::function_ref<bool(DIERef ref)> callback) const {
 | |
|   for (const auto &entry : m_map)
 | |
|     if (regex.Execute(entry.cstring.GetCString())) {
 | |
|       if (!callback(entry.value))
 | |
|         return false;
 | |
|     }
 | |
|   return true;
 | |
| }
 | |
| 
 | |
| void NameToDIE::FindAllEntriesForUnit(
 | |
|     const DWARFUnit &unit,
 | |
|     llvm::function_ref<bool(DIERef ref)> callback) const {
 | |
|   const uint32_t size = m_map.GetSize();
 | |
|   for (uint32_t i = 0; i < size; ++i) {
 | |
|     const DIERef &die_ref = m_map.GetValueAtIndexUnchecked(i);
 | |
|     if (unit.GetSymbolFileDWARF().GetDwoNum() == die_ref.dwo_num() &&
 | |
|         unit.GetDebugSection() == die_ref.section() &&
 | |
|         unit.GetOffset() <= die_ref.die_offset() &&
 | |
|         die_ref.die_offset() < unit.GetNextUnitOffset()) {
 | |
|       if (!callback(die_ref))
 | |
|         return;
 | |
|     }
 | |
|   }
 | |
| }
 | |
| 
 | |
| void NameToDIE::Dump(Stream *s) {
 | |
|   const uint32_t size = m_map.GetSize();
 | |
|   for (uint32_t i = 0; i < size; ++i) {
 | |
|     s->Format("{0} \"{1}\"\n", m_map.GetValueAtIndexUnchecked(i),
 | |
|               m_map.GetCStringAtIndexUnchecked(i));
 | |
|   }
 | |
| }
 | |
| 
 | |
| void NameToDIE::ForEach(
 | |
|     std::function<bool(ConstString name, const DIERef &die_ref)> const
 | |
|         &callback) const {
 | |
|   const uint32_t size = m_map.GetSize();
 | |
|   for (uint32_t i = 0; i < size; ++i) {
 | |
|     if (!callback(m_map.GetCStringAtIndexUnchecked(i),
 | |
|                   m_map.GetValueAtIndexUnchecked(i)))
 | |
|       break;
 | |
|   }
 | |
| }
 | |
| 
 | |
| void NameToDIE::Append(const NameToDIE &other) {
 | |
|   const uint32_t size = other.m_map.GetSize();
 | |
|   for (uint32_t i = 0; i < size; ++i) {
 | |
|     m_map.Append(other.m_map.GetCStringAtIndexUnchecked(i),
 | |
|                  other.m_map.GetValueAtIndexUnchecked(i));
 | |
|   }
 | |
| }
 |