| 1 | //===-- LibCxxSpan.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 "llvm/ADT/APSInt.h" |
| 15 | #include <optional> |
| 16 | |
| 17 | using namespace lldb; |
| 18 | using namespace lldb_private; |
| 19 | using namespace lldb_private::formatters; |
| 20 | |
| 21 | namespace lldb_private { |
| 22 | namespace formatters { |
| 23 | |
| 24 | class LibcxxStdSpanSyntheticFrontEnd : public SyntheticChildrenFrontEnd { |
| 25 | public: |
| 26 | LibcxxStdSpanSyntheticFrontEnd(lldb::ValueObjectSP valobj_sp); |
| 27 | |
| 28 | ~LibcxxStdSpanSyntheticFrontEnd() override = default; |
| 29 | |
| 30 | llvm::Expected<uint32_t> CalculateNumChildren() override; |
| 31 | |
| 32 | lldb::ValueObjectSP GetChildAtIndex(uint32_t idx) override; |
| 33 | |
| 34 | /// Determines properties of the std::span<> associated with this object |
| 35 | // |
| 36 | // std::span can either be instantiated with a compile-time known |
| 37 | // extent or a std::dynamic_extent (this is the default if only the |
| 38 | // type template argument is provided). The layout of std::span |
| 39 | // depends on whether the extent is dynamic or not. For static |
| 40 | // extents (e.g., std::span<int, 9>): |
| 41 | // |
| 42 | // (std::__1::span<const int, 9>) s = { |
| 43 | // __data = 0x000000016fdff494 |
| 44 | // } |
| 45 | // |
| 46 | // For dynamic extents, e.g., std::span<int>, the layout is: |
| 47 | // |
| 48 | // (std::__1::span<const int, 18446744073709551615>) s = { |
| 49 | // __data = 0x000000016fdff494 |
| 50 | // __size = 6 |
| 51 | // } |
| 52 | // |
| 53 | // This function checks for a '__size' member to determine the number |
| 54 | // of elements in the span. If no such member exists, we get the size |
| 55 | // from the only other place it can be: the template argument. |
| 56 | lldb::ChildCacheState Update() override; |
| 57 | |
| 58 | llvm::Expected<size_t> GetIndexOfChildWithName(ConstString name) override; |
| 59 | |
| 60 | private: |
| 61 | ValueObject *m_start = nullptr; ///< First element of span. Held, not owned. |
| 62 | CompilerType m_element_type{}; ///< Type of span elements. |
| 63 | size_t m_num_elements = 0; ///< Number of elements in span. |
| 64 | uint32_t m_element_size = 0; ///< Size in bytes of each span element. |
| 65 | }; |
| 66 | |
| 67 | lldb_private::formatters::LibcxxStdSpanSyntheticFrontEnd:: |
| 68 | LibcxxStdSpanSyntheticFrontEnd(lldb::ValueObjectSP valobj_sp) |
| 69 | : SyntheticChildrenFrontEnd(*valobj_sp) { |
| 70 | if (valobj_sp) |
| 71 | Update(); |
| 72 | } |
| 73 | |
| 74 | llvm::Expected<uint32_t> lldb_private::formatters:: |
| 75 | LibcxxStdSpanSyntheticFrontEnd::CalculateNumChildren() { |
| 76 | return m_num_elements; |
| 77 | } |
| 78 | |
| 79 | lldb::ValueObjectSP |
| 80 | lldb_private::formatters::LibcxxStdSpanSyntheticFrontEnd::GetChildAtIndex( |
| 81 | uint32_t idx) { |
| 82 | if (!m_start) |
| 83 | return {}; |
| 84 | |
| 85 | uint64_t offset = idx * m_element_size; |
| 86 | offset = offset + m_start->GetValueAsUnsigned(fail_value: 0); |
| 87 | StreamString name; |
| 88 | name.Printf(format: "[%" PRIu64 "]" , (uint64_t)idx); |
| 89 | return CreateValueObjectFromAddress(name: name.GetString(), address: offset, |
| 90 | exe_ctx: m_backend.GetExecutionContextRef(), |
| 91 | type: m_element_type); |
| 92 | } |
| 93 | |
| 94 | lldb::ChildCacheState |
| 95 | lldb_private::formatters::LibcxxStdSpanSyntheticFrontEnd::Update() { |
| 96 | // Get element type. |
| 97 | ValueObjectSP data_type_finder_sp = GetChildMemberWithName( |
| 98 | obj&: m_backend, alternative_names: {ConstString("__data_" ), ConstString("__data" )}); |
| 99 | if (!data_type_finder_sp) |
| 100 | return lldb::ChildCacheState::eRefetch; |
| 101 | |
| 102 | m_element_type = data_type_finder_sp->GetCompilerType().GetPointeeType(); |
| 103 | |
| 104 | // Get element size. |
| 105 | llvm::Expected<uint64_t> size_or_err = m_element_type.GetByteSize(exe_scope: nullptr); |
| 106 | if (!size_or_err) |
| 107 | LLDB_LOG_ERRORV(GetLog(LLDBLog::DataFormatters), size_or_err.takeError(), |
| 108 | "{0}" ); |
| 109 | else { |
| 110 | m_element_size = *size_or_err; |
| 111 | |
| 112 | // Get data. |
| 113 | if (m_element_size > 0) { |
| 114 | m_start = data_type_finder_sp.get(); |
| 115 | } |
| 116 | |
| 117 | // Get number of elements. |
| 118 | if (auto size_sp = GetChildMemberWithName( |
| 119 | obj&: m_backend, alternative_names: {ConstString("__size_" ), ConstString("__size" )})) { |
| 120 | m_num_elements = size_sp->GetValueAsUnsigned(fail_value: 0); |
| 121 | } else if (auto arg = |
| 122 | m_backend.GetCompilerType().GetIntegralTemplateArgument(idx: 1)) { |
| 123 | |
| 124 | m_num_elements = arg->value.GetAPSInt().getLimitedValue(); |
| 125 | } |
| 126 | } |
| 127 | |
| 128 | return lldb::ChildCacheState::eReuse; |
| 129 | } |
| 130 | |
| 131 | llvm::Expected<size_t> lldb_private::formatters:: |
| 132 | LibcxxStdSpanSyntheticFrontEnd::GetIndexOfChildWithName(ConstString name) { |
| 133 | if (!m_start) |
| 134 | return llvm::createStringError(Fmt: "Type has no child named '%s'" , |
| 135 | Vals: name.AsCString()); |
| 136 | auto optional_idx = formatters::ExtractIndexFromString(item_name: name.GetCString()); |
| 137 | if (!optional_idx) { |
| 138 | return llvm::createStringError(Fmt: "Type has no child named '%s'" , |
| 139 | Vals: name.AsCString()); |
| 140 | } |
| 141 | return *optional_idx; |
| 142 | } |
| 143 | |
| 144 | lldb_private::SyntheticChildrenFrontEnd * |
| 145 | LibcxxStdSpanSyntheticFrontEndCreator(CXXSyntheticChildren *, |
| 146 | lldb::ValueObjectSP valobj_sp) { |
| 147 | if (!valobj_sp) |
| 148 | return nullptr; |
| 149 | CompilerType type = valobj_sp->GetCompilerType(); |
| 150 | if (!type.IsValid() || type.GetNumTemplateArguments() != 2) |
| 151 | return nullptr; |
| 152 | return new LibcxxStdSpanSyntheticFrontEnd(valobj_sp); |
| 153 | } |
| 154 | |
| 155 | } // namespace formatters |
| 156 | } // namespace lldb_private |
| 157 | |