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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
/*
* Copyright (C) 2020 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include "shared/source/device_binary_format/elf/elf_decoder.h"
#include "shared/source/device_binary_format/elf/elf_encoder.h"
#include "shared/source/device_binary_format/elf/zebin_elf.h"
#include "igfxfmid.h"
#include <vector>
extern PRODUCT_FAMILY productFamily;
namespace ZebinTestData {
struct ValidEmptyProgram {
ValidEmptyProgram() {
NEO::Elf::ElfEncoder<> enc;
enc.getElfFileHeader().type = NEO::Elf::ET_ZEBIN_EXE;
enc.getElfFileHeader().machine = productFamily;
enc.appendSection(NEO::Elf::SHT_ZEBIN_ZEINFO, NEO::Elf::SectionsNamesZebin::zeInfo, std::string{"---\nkernels : \n...\n"});
storage = enc.encode();
recalcPtr();
}
virtual void recalcPtr() {
elfHeader = reinterpret_cast<NEO::Elf::ElfFileHeader<NEO::Elf::EI_CLASS_64> *>(storage.data());
}
template <typename SectionHeaderEnumT>
NEO::Elf::ElfSectionHeader<NEO::Elf::EI_CLASS_64> &appendSection(SectionHeaderEnumT sectionType, NEO::ConstStringRef sectionLabel, const ArrayRef<const uint8_t> sectionData) {
std::string err, warn;
auto decoded = NEO::Elf::decodeElf(storage, err, warn);
NEO::Elf::ElfEncoder<NEO::Elf::EI_CLASS_64> enc;
enc.getElfFileHeader() = *decoded.elfFileHeader;
int sectionIt = 0;
auto sectionHeaderNamesData = decoded.sectionHeaders[decoded.elfFileHeader->shStrNdx].data;
NEO::ConstStringRef sectionHeaderNamesString(reinterpret_cast<const char *>(sectionHeaderNamesData.begin()), sectionHeaderNamesData.size());
for (const auto §ion : decoded.sectionHeaders) {
switch (section.header->type) {
case NEO::Elf::SHN_UNDEF:
break;
case NEO::Elf::SHT_STRTAB:
if (decoded.elfFileHeader->shStrNdx != sectionIt) {
enc.appendSection(section.header->type, sectionHeaderNamesString.data() + section.header->name, section.data);
}
break;
default:
enc.appendSection(section.header->type, sectionHeaderNamesString.data() + section.header->name, section.data);
break;
}
++sectionIt;
}
enc.appendSection(static_cast<NEO::Elf::SECTION_HEADER_TYPE>(sectionType), sectionLabel, sectionData);
storage = enc.encode();
recalcPtr();
decoded = NEO::Elf::decodeElf(storage, err, warn);
sectionHeaderNamesData = decoded.sectionHeaders[decoded.elfFileHeader->shStrNdx].data;
sectionHeaderNamesString = NEO::ConstStringRef(reinterpret_cast<const char *>(sectionHeaderNamesData.begin()), sectionHeaderNamesData.size());
for (const auto §ion : decoded.sectionHeaders) {
if ((sectionType == section.header->type) && (sectionLabel == sectionHeaderNamesString.data() + section.header->name)) {
return const_cast<NEO::Elf::ElfSectionHeader<NEO::Elf::EI_CLASS_64> &>(*section.header);
}
}
UNREACHABLE();
}
template <typename SectionHeaderEnumT>
void removeSection(SectionHeaderEnumT sectionType, NEO::ConstStringRef sectionLabel) {
std::string err, warn;
auto decoded = NEO::Elf::decodeElf(storage, err, warn);
NEO::Elf::ElfEncoder<NEO::Elf::EI_CLASS_64> enc;
enc.getElfFileHeader() = *decoded.elfFileHeader;
int sectionIt = 0;
auto sectionHeaderNamesData = decoded.sectionHeaders[decoded.elfFileHeader->shStrNdx].data;
NEO::ConstStringRef sectionHeaderNamesString(reinterpret_cast<const char *>(sectionHeaderNamesData.begin()), sectionHeaderNamesData.size());
for (const auto §ion : decoded.sectionHeaders) {
bool add = true;
switch (section.header->type) {
case NEO::Elf::SHN_UNDEF:
add = false;
break;
case NEO::Elf::SHT_STRTAB:
add = (decoded.elfFileHeader->shStrNdx != sectionIt);
break;
default:
add = ((section.header->type != sectionType) || (sectionHeaderNamesString.data() + section.header->name != sectionLabel));
break;
}
if (add) {
enc.appendSection(section.header->type, sectionHeaderNamesString.data() + section.header->name, section.data);
}
++sectionIt;
}
if (decoded.elfFileHeader->shNum <= 3) {
enc.appendSection(NEO::Elf::SHT_STRTAB, "", {});
}
storage = enc.encode();
recalcPtr();
}
NEO::Elf::ElfFileHeader<NEO::Elf::EI_CLASS_64> *elfHeader;
std::vector<uint8_t> storage;
};
} // namespace ZebinTestData
|
