| 1 | //===-- LibCxxVector.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 "LibCxx.h" |
| 10 | |
| 11 | #include "lldb/DataFormatters/FormattersHelpers.h" |
| 12 | #include "lldb/Utility/ConstString.h" |
| 13 | #include "lldb/ValueObject/ValueObject.h" |
| 14 | #include "lldb/lldb-enumerations.h" |
| 15 | #include "lldb/lldb-forward.h" |
| 16 | #include <optional> |
| 17 | |
| 18 | using namespace lldb; |
| 19 | using namespace lldb_private; |
| 20 | using namespace lldb_private::formatters; |
| 21 | |
| 22 | namespace lldb_private { |
| 23 | namespace formatters { |
| 24 | class LibcxxStdVectorSyntheticFrontEnd : public SyntheticChildrenFrontEnd { |
| 25 | public: |
| 26 | LibcxxStdVectorSyntheticFrontEnd(lldb::ValueObjectSP valobj_sp); |
| 27 | |
| 28 | ~LibcxxStdVectorSyntheticFrontEnd() override; |
| 29 | |
| 30 | llvm::Expected<uint32_t> CalculateNumChildren() override; |
| 31 | |
| 32 | lldb::ValueObjectSP GetChildAtIndex(uint32_t idx) override; |
| 33 | |
| 34 | lldb::ChildCacheState Update() override; |
| 35 | |
| 36 | llvm::Expected<size_t> GetIndexOfChildWithName(ConstString name) override; |
| 37 | |
| 38 | private: |
| 39 | ValueObject *m_start = nullptr; |
| 40 | ValueObject *m_finish = nullptr; |
| 41 | CompilerType m_element_type; |
| 42 | uint32_t m_element_size = 0; |
| 43 | }; |
| 44 | |
| 45 | class LibcxxVectorBoolSyntheticFrontEnd : public SyntheticChildrenFrontEnd { |
| 46 | public: |
| 47 | LibcxxVectorBoolSyntheticFrontEnd(lldb::ValueObjectSP valobj_sp); |
| 48 | |
| 49 | llvm::Expected<uint32_t> CalculateNumChildren() override; |
| 50 | |
| 51 | lldb::ValueObjectSP GetChildAtIndex(uint32_t idx) override; |
| 52 | |
| 53 | lldb::ChildCacheState Update() override; |
| 54 | |
| 55 | llvm::Expected<size_t> GetIndexOfChildWithName(ConstString name) override; |
| 56 | |
| 57 | private: |
| 58 | CompilerType m_bool_type; |
| 59 | ExecutionContextRef m_exe_ctx_ref; |
| 60 | uint64_t m_count = 0; |
| 61 | lldb::addr_t m_base_data_address = 0; |
| 62 | std::map<size_t, lldb::ValueObjectSP> m_children; |
| 63 | }; |
| 64 | |
| 65 | } // namespace formatters |
| 66 | } // namespace lldb_private |
| 67 | |
| 68 | lldb_private::formatters::LibcxxStdVectorSyntheticFrontEnd:: |
| 69 | LibcxxStdVectorSyntheticFrontEnd(lldb::ValueObjectSP valobj_sp) |
| 70 | : SyntheticChildrenFrontEnd(*valobj_sp), m_element_type() { |
| 71 | if (valobj_sp) |
| 72 | Update(); |
| 73 | } |
| 74 | |
| 75 | lldb_private::formatters::LibcxxStdVectorSyntheticFrontEnd:: |
| 76 | ~LibcxxStdVectorSyntheticFrontEnd() { |
| 77 | // these need to stay around because they are child objects who will follow |
| 78 | // their parent's life cycle |
| 79 | // delete m_start; |
| 80 | // delete m_finish; |
| 81 | } |
| 82 | |
| 83 | llvm::Expected<uint32_t> lldb_private::formatters:: |
| 84 | LibcxxStdVectorSyntheticFrontEnd::CalculateNumChildren() { |
| 85 | if (!m_start || !m_finish) |
| 86 | return llvm::createStringError( |
| 87 | Fmt: "Failed to determine start/end of vector data." ); |
| 88 | |
| 89 | uint64_t start_val = m_start->GetValueAsUnsigned(fail_value: 0); |
| 90 | uint64_t finish_val = m_finish->GetValueAsUnsigned(fail_value: 0); |
| 91 | |
| 92 | // A default-initialized empty vector. |
| 93 | if (start_val == 0 && finish_val == 0) |
| 94 | return 0; |
| 95 | |
| 96 | if (start_val == 0) |
| 97 | return llvm::createStringError(Fmt: "Invalid value for start of vector." ); |
| 98 | |
| 99 | if (finish_val == 0) |
| 100 | return llvm::createStringError(Fmt: "Invalid value for end of vector." ); |
| 101 | |
| 102 | if (start_val > finish_val) |
| 103 | return llvm::createStringError( |
| 104 | Fmt: "Start of vector data begins after end pointer." ); |
| 105 | |
| 106 | size_t num_children = (finish_val - start_val); |
| 107 | if (num_children % m_element_size) |
| 108 | return llvm::createStringError(Fmt: "Size not multiple of element size." ); |
| 109 | |
| 110 | return num_children / m_element_size; |
| 111 | } |
| 112 | |
| 113 | lldb::ValueObjectSP |
| 114 | lldb_private::formatters::LibcxxStdVectorSyntheticFrontEnd::GetChildAtIndex( |
| 115 | uint32_t idx) { |
| 116 | if (!m_start || !m_finish) |
| 117 | return lldb::ValueObjectSP(); |
| 118 | |
| 119 | uint64_t offset = idx * m_element_size; |
| 120 | offset = offset + m_start->GetValueAsUnsigned(fail_value: 0); |
| 121 | StreamString name; |
| 122 | name.Printf(format: "[%" PRIu64 "]" , (uint64_t)idx); |
| 123 | return CreateValueObjectFromAddress(name: name.GetString(), address: offset, |
| 124 | exe_ctx: m_backend.GetExecutionContextRef(), |
| 125 | type: m_element_type); |
| 126 | } |
| 127 | |
| 128 | static ValueObjectSP GetDataPointer(ValueObject &root) { |
| 129 | if (auto cap_sp = root.GetChildMemberWithName(name: "__cap_" )) |
| 130 | return cap_sp; |
| 131 | |
| 132 | ValueObjectSP cap_sp = root.GetChildMemberWithName(name: "__end_cap_" ); |
| 133 | if (!cap_sp) |
| 134 | return nullptr; |
| 135 | |
| 136 | if (!isOldCompressedPairLayout(pair_obj&: *cap_sp)) |
| 137 | return nullptr; |
| 138 | |
| 139 | return GetFirstValueOfLibCXXCompressedPair(pair&: *cap_sp); |
| 140 | } |
| 141 | |
| 142 | lldb::ChildCacheState |
| 143 | lldb_private::formatters::LibcxxStdVectorSyntheticFrontEnd::Update() { |
| 144 | m_start = m_finish = nullptr; |
| 145 | ValueObjectSP data_sp(GetDataPointer(root&: m_backend)); |
| 146 | |
| 147 | if (!data_sp) |
| 148 | return lldb::ChildCacheState::eRefetch; |
| 149 | |
| 150 | m_element_type = data_sp->GetCompilerType().GetPointeeType(); |
| 151 | llvm::Expected<uint64_t> size_or_err = m_element_type.GetByteSize(exe_scope: nullptr); |
| 152 | if (!size_or_err) |
| 153 | LLDB_LOG_ERRORV(GetLog(LLDBLog::DataFormatters), size_or_err.takeError(), |
| 154 | "{0}" ); |
| 155 | else { |
| 156 | m_element_size = *size_or_err; |
| 157 | |
| 158 | if (m_element_size > 0) { |
| 159 | // store raw pointers or end up with a circular dependency |
| 160 | m_start = m_backend.GetChildMemberWithName(name: "__begin_" ).get(); |
| 161 | m_finish = m_backend.GetChildMemberWithName(name: "__end_" ).get(); |
| 162 | } |
| 163 | } |
| 164 | return lldb::ChildCacheState::eRefetch; |
| 165 | } |
| 166 | |
| 167 | llvm::Expected<size_t> |
| 168 | lldb_private::formatters::LibcxxStdVectorSyntheticFrontEnd:: |
| 169 | GetIndexOfChildWithName(ConstString name) { |
| 170 | if (!m_start || !m_finish) |
| 171 | return llvm::createStringError(Fmt: "Type has no child named '%s'" , |
| 172 | Vals: name.AsCString()); |
| 173 | auto optional_idx = formatters::ExtractIndexFromString(item_name: name.GetCString()); |
| 174 | if (!optional_idx) { |
| 175 | return llvm::createStringError(Fmt: "Type has no child named '%s'" , |
| 176 | Vals: name.AsCString()); |
| 177 | } |
| 178 | return *optional_idx; |
| 179 | } |
| 180 | |
| 181 | lldb_private::formatters::LibcxxVectorBoolSyntheticFrontEnd:: |
| 182 | LibcxxVectorBoolSyntheticFrontEnd(lldb::ValueObjectSP valobj_sp) |
| 183 | : SyntheticChildrenFrontEnd(*valobj_sp), m_bool_type(), m_exe_ctx_ref(), |
| 184 | m_children() { |
| 185 | if (valobj_sp) { |
| 186 | Update(); |
| 187 | m_bool_type = |
| 188 | valobj_sp->GetCompilerType().GetBasicTypeFromAST(basic_type: lldb::eBasicTypeBool); |
| 189 | } |
| 190 | } |
| 191 | |
| 192 | llvm::Expected<uint32_t> lldb_private::formatters:: |
| 193 | LibcxxVectorBoolSyntheticFrontEnd::CalculateNumChildren() { |
| 194 | return m_count; |
| 195 | } |
| 196 | |
| 197 | lldb::ValueObjectSP |
| 198 | lldb_private::formatters::LibcxxVectorBoolSyntheticFrontEnd::GetChildAtIndex( |
| 199 | uint32_t idx) { |
| 200 | auto iter = m_children.find(x: idx), end = m_children.end(); |
| 201 | if (iter != end) |
| 202 | return iter->second; |
| 203 | if (idx >= m_count) |
| 204 | return {}; |
| 205 | if (m_base_data_address == 0 || m_count == 0) |
| 206 | return {}; |
| 207 | if (!m_bool_type) |
| 208 | return {}; |
| 209 | size_t byte_idx = (idx >> 3); // divide by 8 to get byte index |
| 210 | size_t bit_index = (idx & 7); // efficient idx % 8 for bit index |
| 211 | lldb::addr_t byte_location = m_base_data_address + byte_idx; |
| 212 | ProcessSP process_sp(m_exe_ctx_ref.GetProcessSP()); |
| 213 | if (!process_sp) |
| 214 | return {}; |
| 215 | uint8_t byte = 0; |
| 216 | uint8_t mask = 0; |
| 217 | Status err; |
| 218 | size_t bytes_read = process_sp->ReadMemory(vm_addr: byte_location, buf: &byte, size: 1, error&: err); |
| 219 | if (err.Fail() || bytes_read == 0) |
| 220 | return {}; |
| 221 | mask = 1 << bit_index; |
| 222 | bool bit_set = ((byte & mask) != 0); |
| 223 | std::optional<uint64_t> size = |
| 224 | llvm::expectedToOptional(E: m_bool_type.GetByteSize(exe_scope: nullptr)); |
| 225 | if (!size) |
| 226 | return {}; |
| 227 | WritableDataBufferSP buffer_sp(new DataBufferHeap(*size, 0)); |
| 228 | if (bit_set && buffer_sp && buffer_sp->GetBytes()) { |
| 229 | // regardless of endianness, anything non-zero is true |
| 230 | *(buffer_sp->GetBytes()) = 1; |
| 231 | } |
| 232 | StreamString name; |
| 233 | name.Printf(format: "[%" PRIu64 "]" , (uint64_t)idx); |
| 234 | ValueObjectSP retval_sp(CreateValueObjectFromData( |
| 235 | name: name.GetString(), |
| 236 | data: DataExtractor(buffer_sp, process_sp->GetByteOrder(), |
| 237 | process_sp->GetAddressByteSize()), |
| 238 | exe_ctx: m_exe_ctx_ref, type: m_bool_type)); |
| 239 | if (retval_sp) |
| 240 | m_children[idx] = retval_sp; |
| 241 | return retval_sp; |
| 242 | } |
| 243 | |
| 244 | lldb::ChildCacheState |
| 245 | lldb_private::formatters::LibcxxVectorBoolSyntheticFrontEnd::Update() { |
| 246 | m_children.clear(); |
| 247 | ValueObjectSP valobj_sp = m_backend.GetSP(); |
| 248 | if (!valobj_sp) |
| 249 | return lldb::ChildCacheState::eRefetch; |
| 250 | m_exe_ctx_ref = valobj_sp->GetExecutionContextRef(); |
| 251 | ValueObjectSP size_sp(valobj_sp->GetChildMemberWithName(name: "__size_" )); |
| 252 | if (!size_sp) |
| 253 | return lldb::ChildCacheState::eRefetch; |
| 254 | m_count = size_sp->GetValueAsUnsigned(fail_value: 0); |
| 255 | if (!m_count) |
| 256 | return lldb::ChildCacheState::eReuse; |
| 257 | ValueObjectSP begin_sp(valobj_sp->GetChildMemberWithName(name: "__begin_" )); |
| 258 | if (!begin_sp) { |
| 259 | m_count = 0; |
| 260 | return lldb::ChildCacheState::eRefetch; |
| 261 | } |
| 262 | m_base_data_address = begin_sp->GetValueAsUnsigned(fail_value: 0); |
| 263 | if (!m_base_data_address) { |
| 264 | m_count = 0; |
| 265 | return lldb::ChildCacheState::eRefetch; |
| 266 | } |
| 267 | return lldb::ChildCacheState::eRefetch; |
| 268 | } |
| 269 | |
| 270 | llvm::Expected<size_t> |
| 271 | lldb_private::formatters::LibcxxVectorBoolSyntheticFrontEnd:: |
| 272 | GetIndexOfChildWithName(ConstString name) { |
| 273 | if (!m_count || !m_base_data_address) |
| 274 | return llvm::createStringError(Fmt: "Type has no child named '%s'" , |
| 275 | Vals: name.AsCString()); |
| 276 | auto optional_idx = ExtractIndexFromString(item_name: name.AsCString()); |
| 277 | if (!optional_idx) { |
| 278 | return llvm::createStringError(Fmt: "Type has no child named '%s'" , |
| 279 | Vals: name.AsCString()); |
| 280 | } |
| 281 | uint32_t idx = *optional_idx; |
| 282 | if (idx >= CalculateNumChildrenIgnoringErrors()) |
| 283 | return llvm::createStringError(Fmt: "Type has no child named '%s'" , |
| 284 | Vals: name.AsCString()); |
| 285 | return idx; |
| 286 | } |
| 287 | |
| 288 | lldb_private::SyntheticChildrenFrontEnd * |
| 289 | lldb_private::formatters::LibcxxStdVectorSyntheticFrontEndCreator( |
| 290 | CXXSyntheticChildren *, lldb::ValueObjectSP valobj_sp) { |
| 291 | if (!valobj_sp) |
| 292 | return nullptr; |
| 293 | CompilerType type = valobj_sp->GetCompilerType(); |
| 294 | if (!type.IsValid() || type.GetNumTemplateArguments() == 0) |
| 295 | return nullptr; |
| 296 | CompilerType arg_type = type.GetTypeTemplateArgument(idx: 0); |
| 297 | if (arg_type.GetTypeName() == "bool" ) |
| 298 | return new LibcxxVectorBoolSyntheticFrontEnd(valobj_sp); |
| 299 | return new LibcxxStdVectorSyntheticFrontEnd(valobj_sp); |
| 300 | } |
| 301 | |