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
|
/*
* Copyright (C) 2021-2025 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include "shared/source/device_binary_format/elf/elf_decoder.h"
#include "shared/source/device_binary_format/zebin/zebin_elf.h"
#include "shared/source/utilities/arrayref.h"
#include <limits>
#include <string>
#include <unordered_map>
#include <vector>
namespace NEO {
class GraphicsAllocation;
class SharedPoolAllocation;
namespace Zebin::Debug {
struct Segments {
struct Segment {
uintptr_t address = std::numeric_limits<uintptr_t>::max();
size_t size = 0;
};
using CPUSegment = Segment;
using GPUSegment = Segment;
using KernelNameIsaTupleT = std::tuple<std::string_view, Segment>;
using KernelNameToSegmentMap = std::unordered_map<std::string, GPUSegment>;
GPUSegment varData;
GPUSegment constData;
CPUSegment stringData;
KernelNameToSegmentMap nameToSegMap;
Segments();
Segments(const SharedPoolAllocation *globalVarAlloc, const SharedPoolAllocation *globalConstAlloc, ArrayRef<const uint8_t> &globalStrings, std::vector<KernelNameIsaTupleT> &kernels);
};
class DebugZebinCreator {
public:
using Elf = NEO::Elf::Elf<NEO::Elf::EI_CLASS_64>;
DebugZebinCreator() = delete;
DebugZebinCreator(Elf &zebin, const Segments &segments) : segments(segments), zebin(zebin) {}
void applyRelocations();
void createDebugZebin();
inline std::vector<uint8_t> getDebugZebin() { return debugZebin; }
protected:
void applyRelocation(uintptr_t addr, uint64_t value, NEO::Zebin::Elf::RelocTypeZebin type);
bool isRelocTypeSupported(NEO::Zebin::Elf::RelocTypeZebin type);
const Segments::Segment *getSegmentByName(ConstStringRef sectionName);
const Segments::Segment *getTextSegmentByName(ConstStringRef textSegmentName);
bool isCpuSegment(ConstStringRef sectionName);
const Segments &segments;
const Elf &zebin;
uint32_t symTabShndx = std::numeric_limits<uint32_t>::max();
std::vector<uint8_t> debugZebin;
};
template <typename T>
void patchWithValue(uintptr_t addr, T value);
std::vector<uint8_t> createDebugZebin(ArrayRef<const uint8_t> zebin, const Segments &segmentData);
} // namespace Zebin::Debug
} // namespace NEO
|