1 | //===-- BreakpointLocationList.cpp ----------------------------------------===// |
2 | // |
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
4 | // See https://llvm.org/LICENSE.txt for license information. |
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
6 | // |
7 | //===----------------------------------------------------------------------===// |
8 | |
9 | #include "lldb/Breakpoint/BreakpointLocationList.h" |
10 | |
11 | #include "lldb/Breakpoint/Breakpoint.h" |
12 | #include "lldb/Breakpoint/BreakpointLocation.h" |
13 | #include "lldb/Core/Module.h" |
14 | #include "lldb/Core/Section.h" |
15 | #include "lldb/Target/SectionLoadList.h" |
16 | #include "lldb/Target/Target.h" |
17 | #include "lldb/Utility/ArchSpec.h" |
18 | |
19 | using namespace lldb; |
20 | using namespace lldb_private; |
21 | |
22 | BreakpointLocationList::BreakpointLocationList(Breakpoint &owner) |
23 | : m_owner(owner), m_next_id(0), m_new_location_recorder(nullptr) {} |
24 | |
25 | BreakpointLocationList::~BreakpointLocationList() = default; |
26 | |
27 | BreakpointLocationSP |
28 | BreakpointLocationList::Create(const Address &addr, |
29 | bool resolve_indirect_symbols) { |
30 | std::lock_guard<std::recursive_mutex> guard(m_mutex); |
31 | // The location ID is just the size of the location list + 1 |
32 | lldb::break_id_t bp_loc_id = ++m_next_id; |
33 | BreakpointLocationSP bp_loc_sp( |
34 | new BreakpointLocation(bp_loc_id, m_owner, addr, LLDB_INVALID_THREAD_ID, |
35 | m_owner.IsHardware(), resolve_indirect_symbols)); |
36 | m_locations.push_back(x: bp_loc_sp); |
37 | m_address_to_location[addr] = bp_loc_sp; |
38 | return bp_loc_sp; |
39 | } |
40 | |
41 | bool BreakpointLocationList::ShouldStop(StoppointCallbackContext *context, |
42 | lldb::break_id_t break_id) { |
43 | BreakpointLocationSP bp = FindByID(breakID: break_id); |
44 | if (bp) { |
45 | // Let the BreakpointLocation decide if it should stop here (could not have |
46 | // reached it's target hit count yet, or it could have a callback that |
47 | // decided it shouldn't stop (shared library loads/unloads). |
48 | return bp->ShouldStop(context); |
49 | } |
50 | // We should stop here since this BreakpointLocation isn't valid anymore or |
51 | // it doesn't exist. |
52 | return true; |
53 | } |
54 | |
55 | lldb::break_id_t BreakpointLocationList::FindIDByAddress(const Address &addr) { |
56 | BreakpointLocationSP bp_loc_sp = FindByAddress(addr); |
57 | if (bp_loc_sp) { |
58 | return bp_loc_sp->GetID(); |
59 | } |
60 | return LLDB_INVALID_BREAK_ID; |
61 | } |
62 | |
63 | static bool Compare(BreakpointLocationSP lhs, lldb::break_id_t val) { |
64 | return lhs->GetID() < val; |
65 | } |
66 | |
67 | BreakpointLocationSP |
68 | BreakpointLocationList::FindByID(lldb::break_id_t break_id) const { |
69 | std::lock_guard<std::recursive_mutex> guard(m_mutex); |
70 | collection::const_iterator end = m_locations.end(); |
71 | collection::const_iterator pos = |
72 | llvm::lower_bound(Range: m_locations, Value&: break_id, C: Compare); |
73 | if (pos != end && (*pos)->GetID() == break_id) |
74 | return *(pos); |
75 | else |
76 | return BreakpointLocationSP(); |
77 | } |
78 | |
79 | size_t BreakpointLocationList::FindInModule( |
80 | Module *module, BreakpointLocationCollection &bp_loc_list) { |
81 | std::lock_guard<std::recursive_mutex> guard(m_mutex); |
82 | const size_t orig_size = bp_loc_list.GetSize(); |
83 | collection::iterator pos, end = m_locations.end(); |
84 | |
85 | for (pos = m_locations.begin(); pos != end; ++pos) { |
86 | BreakpointLocationSP break_loc = (*pos); |
87 | SectionSP section_sp(break_loc->GetAddress().GetSection()); |
88 | if (section_sp && section_sp->GetModule().get() == module) { |
89 | bp_loc_list.Add(bp_loc_sp: break_loc); |
90 | } |
91 | } |
92 | return bp_loc_list.GetSize() - orig_size; |
93 | } |
94 | |
95 | const BreakpointLocationSP |
96 | BreakpointLocationList::FindByAddress(const Address &addr) const { |
97 | std::lock_guard<std::recursive_mutex> guard(m_mutex); |
98 | BreakpointLocationSP bp_loc_sp; |
99 | if (!m_locations.empty()) { |
100 | Address so_addr; |
101 | |
102 | if (addr.IsSectionOffset()) { |
103 | so_addr = addr; |
104 | } else { |
105 | // Try and resolve as a load address if possible. |
106 | m_owner.GetTarget().ResolveLoadAddress(load_addr: addr.GetOffset(), so_addr); |
107 | if (!so_addr.IsValid()) { |
108 | // The address didn't resolve, so just set to passed in addr. |
109 | so_addr = addr; |
110 | } |
111 | } |
112 | |
113 | addr_map::const_iterator pos = m_address_to_location.find(x: so_addr); |
114 | if (pos != m_address_to_location.end()) |
115 | bp_loc_sp = pos->second; |
116 | } |
117 | |
118 | return bp_loc_sp; |
119 | } |
120 | |
121 | void BreakpointLocationList::Dump(Stream *s) const { |
122 | s->Printf(format: "%p: " , static_cast<const void *>(this)); |
123 | // s->Indent(); |
124 | std::lock_guard<std::recursive_mutex> guard(m_mutex); |
125 | s->Printf(format: "BreakpointLocationList with %" PRIu64 " BreakpointLocations:\n" , |
126 | (uint64_t)m_locations.size()); |
127 | s->IndentMore(); |
128 | collection::const_iterator pos, end = m_locations.end(); |
129 | for (pos = m_locations.begin(); pos != end; ++pos) |
130 | (*pos)->Dump(s); |
131 | s->IndentLess(); |
132 | } |
133 | |
134 | BreakpointLocationSP BreakpointLocationList::GetByIndex(size_t i) { |
135 | std::lock_guard<std::recursive_mutex> guard(m_mutex); |
136 | BreakpointLocationSP bp_loc_sp; |
137 | if (i < m_locations.size()) |
138 | bp_loc_sp = m_locations[i]; |
139 | |
140 | return bp_loc_sp; |
141 | } |
142 | |
143 | const BreakpointLocationSP BreakpointLocationList::GetByIndex(size_t i) const { |
144 | std::lock_guard<std::recursive_mutex> guard(m_mutex); |
145 | BreakpointLocationSP bp_loc_sp; |
146 | if (i < m_locations.size()) |
147 | bp_loc_sp = m_locations[i]; |
148 | |
149 | return bp_loc_sp; |
150 | } |
151 | |
152 | void BreakpointLocationList::ClearAllBreakpointSites() { |
153 | std::lock_guard<std::recursive_mutex> guard(m_mutex); |
154 | collection::iterator pos, end = m_locations.end(); |
155 | for (pos = m_locations.begin(); pos != end; ++pos) |
156 | (*pos)->ClearBreakpointSite(); |
157 | } |
158 | |
159 | void BreakpointLocationList::ResolveAllBreakpointSites() { |
160 | std::lock_guard<std::recursive_mutex> guard(m_mutex); |
161 | collection::iterator pos, end = m_locations.end(); |
162 | |
163 | for (pos = m_locations.begin(); pos != end; ++pos) { |
164 | if ((*pos)->IsEnabled()) |
165 | (*pos)->ResolveBreakpointSite(); |
166 | } |
167 | } |
168 | |
169 | uint32_t BreakpointLocationList::GetHitCount() const { |
170 | uint32_t hit_count = 0; |
171 | std::lock_guard<std::recursive_mutex> guard(m_mutex); |
172 | collection::const_iterator pos, end = m_locations.end(); |
173 | for (pos = m_locations.begin(); pos != end; ++pos) |
174 | hit_count += (*pos)->GetHitCount(); |
175 | return hit_count; |
176 | } |
177 | |
178 | void BreakpointLocationList::ResetHitCount() { |
179 | std::lock_guard<std::recursive_mutex> guard(m_mutex); |
180 | for (auto &loc : m_locations) |
181 | loc->ResetHitCount(); |
182 | } |
183 | |
184 | size_t BreakpointLocationList::GetNumResolvedLocations() const { |
185 | std::lock_guard<std::recursive_mutex> guard(m_mutex); |
186 | size_t resolve_count = 0; |
187 | collection::const_iterator pos, end = m_locations.end(); |
188 | for (pos = m_locations.begin(); pos != end; ++pos) { |
189 | if ((*pos)->IsResolved()) |
190 | ++resolve_count; |
191 | } |
192 | return resolve_count; |
193 | } |
194 | |
195 | void BreakpointLocationList::GetDescription(Stream *s, |
196 | lldb::DescriptionLevel level) { |
197 | std::lock_guard<std::recursive_mutex> guard(m_mutex); |
198 | collection::iterator pos, end = m_locations.end(); |
199 | |
200 | for (pos = m_locations.begin(); pos != end; ++pos) { |
201 | s->Printf(format: " " ); |
202 | (*pos)->GetDescription(s, level); |
203 | } |
204 | } |
205 | |
206 | BreakpointLocationSP BreakpointLocationList::AddLocation( |
207 | const Address &addr, bool resolve_indirect_symbols, bool *new_location) { |
208 | std::lock_guard<std::recursive_mutex> guard(m_mutex); |
209 | |
210 | if (new_location) |
211 | *new_location = false; |
212 | BreakpointLocationSP bp_loc_sp(FindByAddress(addr)); |
213 | if (!bp_loc_sp) { |
214 | bp_loc_sp = Create(addr, resolve_indirect_symbols); |
215 | if (bp_loc_sp) { |
216 | bp_loc_sp->ResolveBreakpointSite(); |
217 | |
218 | if (new_location) |
219 | *new_location = true; |
220 | if (m_new_location_recorder) { |
221 | m_new_location_recorder->Add(bp_loc_sp); |
222 | } |
223 | } |
224 | } |
225 | return bp_loc_sp; |
226 | } |
227 | |
228 | void BreakpointLocationList::SwapLocation( |
229 | BreakpointLocationSP to_location_sp, |
230 | BreakpointLocationSP from_location_sp) { |
231 | if (!from_location_sp || !to_location_sp) |
232 | return; |
233 | |
234 | m_address_to_location.erase(x: to_location_sp->GetAddress()); |
235 | to_location_sp->SwapLocation(swap_from: from_location_sp); |
236 | RemoveLocation(bp_loc_sp: from_location_sp); |
237 | m_address_to_location[to_location_sp->GetAddress()] = to_location_sp; |
238 | to_location_sp->ResolveBreakpointSite(); |
239 | } |
240 | |
241 | bool BreakpointLocationList::RemoveLocation( |
242 | const lldb::BreakpointLocationSP &bp_loc_sp) { |
243 | if (bp_loc_sp) { |
244 | std::lock_guard<std::recursive_mutex> guard(m_mutex); |
245 | |
246 | m_address_to_location.erase(x: bp_loc_sp->GetAddress()); |
247 | |
248 | size_t num_locations = m_locations.size(); |
249 | for (size_t idx = 0; idx < num_locations; idx++) { |
250 | if (m_locations[idx].get() == bp_loc_sp.get()) { |
251 | RemoveLocationByIndex(idx); |
252 | return true; |
253 | } |
254 | } |
255 | } |
256 | return false; |
257 | } |
258 | |
259 | void BreakpointLocationList::RemoveLocationByIndex(size_t idx) { |
260 | assert (idx < m_locations.size()); |
261 | m_address_to_location.erase(x: m_locations[idx]->GetAddress()); |
262 | m_locations.erase(position: m_locations.begin() + idx); |
263 | } |
264 | |
265 | void BreakpointLocationList::RemoveInvalidLocations(const ArchSpec &arch) { |
266 | std::lock_guard<std::recursive_mutex> guard(m_mutex); |
267 | size_t idx = 0; |
268 | // Don't cache m_location.size() as it will change since we might remove |
269 | // locations from our vector... |
270 | while (idx < m_locations.size()) { |
271 | BreakpointLocation *bp_loc = m_locations[idx].get(); |
272 | if (bp_loc->GetAddress().SectionWasDeleted()) { |
273 | // Section was deleted which means this breakpoint comes from a module |
274 | // that is no longer valid, so we should remove it. |
275 | RemoveLocationByIndex(idx); |
276 | continue; |
277 | } |
278 | if (arch.IsValid()) { |
279 | ModuleSP module_sp(bp_loc->GetAddress().GetModule()); |
280 | if (module_sp) { |
281 | if (!arch.IsCompatibleMatch(rhs: module_sp->GetArchitecture())) { |
282 | // The breakpoint was in a module whose architecture is no longer |
283 | // compatible with "arch", so we need to remove it |
284 | RemoveLocationByIndex(idx); |
285 | continue; |
286 | } |
287 | } |
288 | } |
289 | // Only increment the index if we didn't remove the locations at index |
290 | // "idx" |
291 | ++idx; |
292 | } |
293 | } |
294 | |
295 | void BreakpointLocationList::StartRecordingNewLocations( |
296 | BreakpointLocationCollection &new_locations) { |
297 | std::lock_guard<std::recursive_mutex> guard(m_mutex); |
298 | assert(m_new_location_recorder == nullptr); |
299 | m_new_location_recorder = &new_locations; |
300 | } |
301 | |
302 | void BreakpointLocationList::StopRecordingNewLocations() { |
303 | std::lock_guard<std::recursive_mutex> guard(m_mutex); |
304 | m_new_location_recorder = nullptr; |
305 | } |
306 | |
307 | void BreakpointLocationList::Compact() { |
308 | lldb::break_id_t highest_id = 0; |
309 | |
310 | for (BreakpointLocationSP loc_sp : m_locations) { |
311 | lldb::break_id_t cur_id = loc_sp->GetID(); |
312 | if (cur_id > highest_id) |
313 | highest_id = cur_id; |
314 | } |
315 | m_next_id = highest_id; |
316 | } |
317 | |