1 //===----------------------- ELFTypesTest.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 #include "llvm/Object/ELFTypes.h" 9 #include "gtest/gtest.h" 10 #include <iostream> 11 12 using namespace llvm; 13 using namespace llvm::object; 14 15 template <typename ELFT> using Elf_Note = typename ELFT::Note; 16 17 template <class ELFT> struct NoteTestData { 18 std::vector<uint8_t> Data; 19 20 const Elf_Note_Impl<ELFT> getElfNote(StringRef Name, uint32_t Type, 21 ArrayRef<uint8_t> Desc) { 22 constexpr uint64_t Align = 4; 23 Data.resize(alignTo(sizeof(Elf_Nhdr_Impl<ELFT>) + Name.size(), Align) + 24 alignTo(Desc.size(), Align), 25 0); 26 27 Elf_Nhdr_Impl<ELFT> *Nhdr = 28 reinterpret_cast<Elf_Nhdr_Impl<ELFT> *>(Data.data()); 29 Nhdr->n_namesz = (Name == "") ? 0 : Name.size() + 1; 30 Nhdr->n_descsz = Desc.size(); 31 Nhdr->n_type = Type; 32 33 auto NameOffset = Data.begin() + sizeof(*Nhdr); 34 std::copy(Name.begin(), Name.end(), NameOffset); 35 36 auto DescOffset = 37 Data.begin() + alignTo(sizeof(*Nhdr) + Nhdr->n_namesz, Align); 38 std::copy(Desc.begin(), Desc.end(), DescOffset); 39 40 return Elf_Note_Impl<ELFT>(*Nhdr); 41 } 42 }; 43 44 TEST(ELFTypesTest, NoteTest) { 45 static const uint8_t Random[] = {1, 2, 3, 4, 5, 6, 7, 8, 9}; 46 ArrayRef<uint8_t> RandomData = ArrayRef(Random); 47 NoteTestData<ELF64LE> TestData; 48 49 auto Note1 = TestData.getElfNote(StringRef("AMD"), ELF::NT_AMDGPU_METADATA, 50 RandomData); 51 EXPECT_EQ(Note1.getName(), "AMD"); 52 EXPECT_EQ(Note1.getType(), ELF::NT_AMDGPU_METADATA); 53 EXPECT_EQ(Note1.getDesc(4), RandomData); 54 EXPECT_EQ(Note1.getDescAsStringRef(4), 55 StringRef(reinterpret_cast<const char *>(Random), sizeof(Random))); 56 57 auto Note2 = TestData.getElfNote("", ELF::NT_AMDGPU_METADATA, RandomData); 58 EXPECT_EQ(Note2.getName(), ""); 59 60 auto Note3 = 61 TestData.getElfNote("AMD", ELF::NT_AMDGPU_METADATA, ArrayRef<uint8_t>()); 62 EXPECT_EQ(Note3.getDescAsStringRef(4), StringRef("")); 63 } 64