1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
//===----------------------- ELFTypesTest.cpp -----------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "llvm/Object/ELFTypes.h"
#include "llvm/Testing/Support/Error.h"
#include "gtest/gtest.h"
#include <iostream>
using namespace llvm;
using namespace llvm::object;
template <typename ELFT> using Elf_Note = typename ELFT::Note;
template <class ELFT> struct NoteTestData {
std::vector<uint8_t> Data;
const Elf_Note_Impl<ELFT> getElfNote(StringRef Name, uint32_t Type,
ArrayRef<uint8_t> Desc) {
constexpr uint64_t Align = 4;
Data.resize(alignTo(sizeof(Elf_Nhdr_Impl<ELFT>) + Name.size(), Align) +
alignTo(Desc.size(), Align),
0);
Elf_Nhdr_Impl<ELFT> *Nhdr =
reinterpret_cast<Elf_Nhdr_Impl<ELFT> *>(Data.data());
Nhdr->n_namesz = (Name == "") ? 0 : Name.size() + 1;
Nhdr->n_descsz = Desc.size();
Nhdr->n_type = Type;
auto NameOffset = Data.begin() + sizeof(*Nhdr);
std::copy(Name.begin(), Name.end(), NameOffset);
auto DescOffset =
Data.begin() + alignTo(sizeof(*Nhdr) + Nhdr->n_namesz, Align);
std::copy(Desc.begin(), Desc.end(), DescOffset);
return Elf_Note_Impl<ELFT>(*Nhdr);
}
};
TEST(ELFTypesTest, NoteTest) {
static const uint8_t Random[] = {1, 2, 3, 4, 5, 6, 7, 8, 9};
ArrayRef<uint8_t> RandomData = ArrayRef(Random);
NoteTestData<ELF64LE> TestData;
auto Note1 = TestData.getElfNote(StringRef("AMD"), ELF::NT_AMDGPU_METADATA,
RandomData);
EXPECT_EQ(Note1.getName(), "AMD");
EXPECT_EQ(Note1.getType(), ELF::NT_AMDGPU_METADATA);
EXPECT_EQ(Note1.getDesc(4), RandomData);
EXPECT_EQ(Note1.getDescAsStringRef(4),
StringRef(reinterpret_cast<const char *>(Random), sizeof(Random)));
auto Note2 = TestData.getElfNote("", ELF::NT_AMDGPU_METADATA, RandomData);
EXPECT_EQ(Note2.getName(), "");
auto Note3 =
TestData.getElfNote("AMD", ELF::NT_AMDGPU_METADATA, ArrayRef<uint8_t>());
EXPECT_EQ(Note3.getDescAsStringRef(4), StringRef(""));
}
TEST(ELFTypesTest, BBEntryMetadataEncodingTest) {
const std::array<BBAddrMap::BBEntry::Metadata, 6> Decoded = {
{{false, false, false, false},
{true, false, false, false},
{false, true, false, false},
{false, false, true, false},
{false, false, false, true},
{true, true, true, true}}};
const std::array<uint32_t, 6> Encoded = {{0, 1, 2, 4, 8, 15}};
for (size_t i = 0; i < Decoded.size(); ++i)
EXPECT_EQ(Decoded[i].encode(), Encoded[i]);
for (size_t i = 0; i < Encoded.size(); ++i) {
Expected<BBAddrMap::BBEntry::Metadata> MetadataOrError =
BBAddrMap::BBEntry::Metadata::decode(Encoded[i]);
ASSERT_THAT_EXPECTED(MetadataOrError, Succeeded());
EXPECT_EQ(*MetadataOrError, Decoded[i]);
}
}
TEST(ELFTypesTest, BBEntryMetadataInvalidEncodingTest) {
const std::array<std::string, 2> Errors = {
"invalid encoding for BBEntry::Metadata: 0xffff",
"invalid encoding for BBEntry::Metadata: 0x100001"};
const std::array<uint32_t, 2> Values = {{0xFFFF, 0x100001}};
for (size_t i = 0; i < Values.size(); ++i) {
EXPECT_THAT_ERROR(
BBAddrMap::BBEntry::Metadata::decode(Values[i]).takeError(),
FailedWithMessage(Errors[i]));
}
}
|
