1 //===-- YAMLModuleTester.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 "TestingSupport/Symbol/YAMLModuleTester.h"
10 #include "Plugins/TypeSystem/Clang/TypeSystemClang.h"
11 #include "lldb/Core/Section.h"
12 #include "llvm/ObjectYAML/DWARFEmitter.h"
13 
14 using namespace lldb_private;
15 
16 /// A mock module holding an object file parsed from YAML.
17 class YAMLModule : public lldb_private::Module {
18 public:
19   YAMLModule(ArchSpec &arch) : Module(FileSpec("test"), arch) {}
20   void SetObjectFile(lldb::ObjectFileSP obj_file) { m_objfile_sp = obj_file; }
21   ObjectFile *GetObjectFile() override { return m_objfile_sp.get(); }
22 };
23 
24 /// A mock object file that can be parsed from YAML.
25 class YAMLObjectFile : public lldb_private::ObjectFile {
26   const lldb::ModuleSP m_module_sp;
27   llvm::StringMap<std::unique_ptr<llvm::MemoryBuffer>> &m_section_map;
28   /// Because there is only one DataExtractor in the ObjectFile
29   /// interface, all sections are copied into a contiguous buffer.
30   std::vector<char> m_buffer;
31 
32 public:
33   YAMLObjectFile(const lldb::ModuleSP &module_sp,
34                  llvm::StringMap<std::unique_ptr<llvm::MemoryBuffer>> &map)
35       : ObjectFile(module_sp, &module_sp->GetFileSpec(), /*file_offset*/ 0,
36                    /*length*/ 0, /*data_sp*/ nullptr, /*data_offset*/ 0),
37         m_module_sp(module_sp), m_section_map(map) {}
38 
39   /// Callback for initializing the module's list of sections.
40   void CreateSections(SectionList &unified_section_list) override {
41     lldb::offset_t total_bytes = 0;
42     for (auto &entry : m_section_map)
43       total_bytes += entry.getValue()->getBufferSize();
44     m_buffer.reserve(total_bytes);
45     m_data =
46         DataExtractor(m_buffer.data(), total_bytes, lldb::eByteOrderLittle, 4);
47 
48     lldb::user_id_t sect_id = 1;
49     for (auto &entry : m_section_map) {
50       llvm::StringRef name = entry.getKey();
51       lldb::SectionType sect_type =
52           llvm::StringSwitch<lldb::SectionType>(name)
53               .Case("debug_info", lldb::eSectionTypeDWARFDebugInfo)
54               .Case("debug_abbrev", lldb::eSectionTypeDWARFDebugAbbrev);
55       auto &membuf = entry.getValue();
56       lldb::addr_t file_vm_addr = 0;
57       lldb::addr_t vm_size = 0;
58       lldb::offset_t file_offset = m_buffer.size();
59       lldb::offset_t file_size = membuf->getBufferSize();
60       m_buffer.resize(file_offset + file_size);
61       memcpy(m_buffer.data() + file_offset, membuf->getBufferStart(),
62              file_size);
63       uint32_t log2align = 0;
64       uint32_t flags = 0;
65       auto section_sp = std::make_shared<lldb_private::Section>(
66           m_module_sp, this, sect_id++, ConstString(name), sect_type,
67           file_vm_addr, vm_size, file_offset, file_size, log2align, flags);
68       unified_section_list.AddSection(section_sp);
69     }
70   }
71 
72   /// \{
73   /// Stub methods that aren't needed here.
74   ConstString GetPluginName() override { return ConstString("YAMLObjectFile"); }
75   uint32_t GetPluginVersion() override { return 0; }
76   void Dump(Stream *s) override {}
77   uint32_t GetAddressByteSize() const override { return 8; }
78   uint32_t GetDependentModules(FileSpecList &file_list) override { return 0; }
79   bool IsExecutable() const override { return 0; }
80   ArchSpec GetArchitecture() override { return {}; }
81   Symtab *GetSymtab() override { return nullptr; }
82   bool IsStripped() override { return false; }
83   UUID GetUUID() override { return {}; }
84   lldb::ByteOrder GetByteOrder() const override {
85     return lldb::eByteOrderLittle;
86   }
87   bool ParseHeader() override { return false; }
88   Type CalculateType() override { return {}; }
89   Strata CalculateStrata() override { return {}; }
90   /// \}
91 };
92 
93 YAMLModuleTester::YAMLModuleTester(llvm::StringRef yaml_data,
94                                    llvm::StringRef triple) {
95   auto sections_map = llvm::DWARFYAML::EmitDebugSections(yaml_data, true);
96   if (!sections_map)
97     return;
98   m_sections_map = std::move(*sections_map);
99   ArchSpec arch(triple);
100   m_module_sp = std::make_shared<YAMLModule>(arch);
101   m_objfile_sp = std::make_shared<YAMLObjectFile>(m_module_sp, m_sections_map);
102   static_cast<YAMLModule *>(m_module_sp.get())->SetObjectFile(m_objfile_sp);
103 
104   lldb::user_id_t uid = 0;
105   llvm::StringRef raw_debug_info = m_sections_map["debug_info"]->getBuffer();
106   lldb_private::DataExtractor debug_info(
107       raw_debug_info.data(), raw_debug_info.size(),
108       m_objfile_sp->GetByteOrder(), m_objfile_sp->GetAddressByteSize());
109   lldb::offset_t offset_ptr = 0;
110   m_symfile_dwarf = std::make_unique<SymbolFileDWARF>(m_objfile_sp, nullptr);
111   llvm::Expected<DWARFUnitSP> dwarf_unit = DWARFUnit::extract(
112       *m_symfile_dwarf, uid,
113       *static_cast<lldb_private::DWARFDataExtractor *>(&debug_info),
114       DIERef::DebugInfo, &offset_ptr, nullptr);
115   if (dwarf_unit)
116     m_dwarf_unit = dwarf_unit.get();
117 }
118