1*0b57cec5SDimitry Andric //===-- NameToDIE.cpp -----------------------------------------------------===//
2*0b57cec5SDimitry Andric //
3*0b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*0b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
5*0b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6*0b57cec5SDimitry Andric //
7*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
8*0b57cec5SDimitry Andric 
9*0b57cec5SDimitry Andric #include "NameToDIE.h"
10*0b57cec5SDimitry Andric #include "DWARFUnit.h"
11*0b57cec5SDimitry Andric #include "lldb/Symbol/ObjectFile.h"
12*0b57cec5SDimitry Andric #include "lldb/Utility/ConstString.h"
13*0b57cec5SDimitry Andric #include "lldb/Utility/RegularExpression.h"
14*0b57cec5SDimitry Andric #include "lldb/Utility/Stream.h"
15*0b57cec5SDimitry Andric #include "lldb/Utility/StreamString.h"
16*0b57cec5SDimitry Andric 
17*0b57cec5SDimitry Andric using namespace lldb;
18*0b57cec5SDimitry Andric using namespace lldb_private;
19*0b57cec5SDimitry Andric 
Finalize()20*0b57cec5SDimitry Andric void NameToDIE::Finalize() {
21*0b57cec5SDimitry Andric   m_map.Sort();
22*0b57cec5SDimitry Andric   m_map.SizeToFit();
23*0b57cec5SDimitry Andric }
24*0b57cec5SDimitry Andric 
Insert(ConstString name,const DIERef & die_ref)25*0b57cec5SDimitry Andric void NameToDIE::Insert(ConstString name, const DIERef &die_ref) {
26*0b57cec5SDimitry Andric   m_map.Append(name, die_ref);
27*0b57cec5SDimitry Andric }
28*0b57cec5SDimitry Andric 
Find(ConstString name,llvm::function_ref<bool (DIERef ref)> callback) const29*0b57cec5SDimitry Andric bool NameToDIE::Find(ConstString name,
30*0b57cec5SDimitry Andric                      llvm::function_ref<bool(DIERef ref)> callback) const {
31*0b57cec5SDimitry Andric   for (const auto &entry : m_map.equal_range(name))
32*0b57cec5SDimitry Andric     if (!callback(entry.value))
33*0b57cec5SDimitry Andric       return false;
34*0b57cec5SDimitry Andric   return true;
35*0b57cec5SDimitry Andric }
36*0b57cec5SDimitry Andric 
Find(const RegularExpression & regex,llvm::function_ref<bool (DIERef ref)> callback) const37*0b57cec5SDimitry Andric bool NameToDIE::Find(const RegularExpression &regex,
38*0b57cec5SDimitry Andric                      llvm::function_ref<bool(DIERef ref)> callback) const {
39*0b57cec5SDimitry Andric   for (const auto &entry : m_map)
40*0b57cec5SDimitry Andric     if (regex.Execute(entry.cstring.GetCString())) {
41*0b57cec5SDimitry Andric       if (!callback(entry.value))
42*0b57cec5SDimitry Andric         return false;
43*0b57cec5SDimitry Andric     }
44*0b57cec5SDimitry Andric   return true;
45*0b57cec5SDimitry Andric }
46*0b57cec5SDimitry Andric 
FindAllEntriesForUnit(const DWARFUnit & unit,llvm::function_ref<bool (DIERef ref)> callback) const47*0b57cec5SDimitry Andric void NameToDIE::FindAllEntriesForUnit(
48*0b57cec5SDimitry Andric     const DWARFUnit &unit,
49*0b57cec5SDimitry Andric     llvm::function_ref<bool(DIERef ref)> callback) const {
50*0b57cec5SDimitry Andric   const uint32_t size = m_map.GetSize();
51*0b57cec5SDimitry Andric   for (uint32_t i = 0; i < size; ++i) {
52*0b57cec5SDimitry Andric     const DIERef &die_ref = m_map.GetValueAtIndexUnchecked(i);
53*0b57cec5SDimitry Andric     if (unit.GetSymbolFileDWARF().GetDwoNum() == die_ref.dwo_num() &&
54*0b57cec5SDimitry Andric         unit.GetDebugSection() == die_ref.section() &&
55*0b57cec5SDimitry Andric         unit.GetOffset() <= die_ref.die_offset() &&
56*0b57cec5SDimitry Andric         die_ref.die_offset() < unit.GetNextUnitOffset()) {
57*0b57cec5SDimitry Andric       if (!callback(die_ref))
58*0b57cec5SDimitry Andric         return;
59*0b57cec5SDimitry Andric     }
60*0b57cec5SDimitry Andric   }
61*0b57cec5SDimitry Andric }
62*0b57cec5SDimitry Andric 
Dump(Stream * s)63*0b57cec5SDimitry Andric void NameToDIE::Dump(Stream *s) {
64*0b57cec5SDimitry Andric   const uint32_t size = m_map.GetSize();
65*0b57cec5SDimitry Andric   for (uint32_t i = 0; i < size; ++i) {
66*0b57cec5SDimitry Andric     s->Format("{0} \"{1}\"\n", m_map.GetValueAtIndexUnchecked(i),
67*0b57cec5SDimitry Andric               m_map.GetCStringAtIndexUnchecked(i));
68*0b57cec5SDimitry Andric   }
69*0b57cec5SDimitry Andric }
70*0b57cec5SDimitry Andric 
ForEach(std::function<bool (ConstString name,const DIERef & die_ref)> const & callback) const71*0b57cec5SDimitry Andric void NameToDIE::ForEach(
72*0b57cec5SDimitry Andric     std::function<bool(ConstString name, const DIERef &die_ref)> const
73*0b57cec5SDimitry Andric         &callback) const {
74*0b57cec5SDimitry Andric   const uint32_t size = m_map.GetSize();
75*0b57cec5SDimitry Andric   for (uint32_t i = 0; i < size; ++i) {
76*0b57cec5SDimitry Andric     if (!callback(m_map.GetCStringAtIndexUnchecked(i),
77*0b57cec5SDimitry Andric                   m_map.GetValueAtIndexUnchecked(i)))
78*0b57cec5SDimitry Andric       break;
79   }
80 }
81 
Append(const NameToDIE & other)82 void NameToDIE::Append(const NameToDIE &other) {
83   const uint32_t size = other.m_map.GetSize();
84   for (uint32_t i = 0; i < size; ++i) {
85     m_map.Append(other.m_map.GetCStringAtIndexUnchecked(i),
86                  other.m_map.GetValueAtIndexUnchecked(i));
87   }
88 }
89