55 lines
1.5 KiB
C
55 lines
1.5 KiB
C
|
/*
|
||
|
|
* Copyright (C) 2018 Intel Corporation
|
||
|
|
*
|
||
|
|
* SPDX-License-Identifier: MIT
|
||
|
|
*
|
||
|
|
*/
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
#include "elf/types.h"
|
||
|
|
|
||
|
|
#include <memory>
|
||
|
|
#include <string>
|
||
|
|
#include <unordered_map>
|
||
|
|
#include <vector>
|
||
|
|
|
||
|
|
struct PTField {
|
||
|
|
uint8_t size;
|
||
|
|
std::string name;
|
||
|
|
};
|
||
|
|
|
||
|
|
struct BinaryHeader {
|
||
|
|
std::vector<PTField> fields; // (size, name)
|
||
|
|
uint32_t size;
|
||
|
|
};
|
||
|
|
struct PatchToken : BinaryHeader {
|
||
|
|
std::string name;
|
||
|
|
};
|
||
|
|
|
||
|
|
using PTMap = std::unordered_map<uint8_t, std::unique_ptr<PatchToken>>;
|
||
|
|
|
||
|
|
class BinaryDecoder {
|
||
|
|
public:
|
||
|
|
BinaryDecoder() = default;
|
||
|
|
BinaryDecoder(const std::string &file, const std::string &patch, const std::string &dump)
|
||
|
|
: binaryFile(file), pathToPatch(patch), pathToDump(dump){};
|
||
|
|
int decode();
|
||
|
|
int validateInput(uint32_t argc, const char **argv);
|
||
|
|
|
||
|
|
protected:
|
||
|
|
BinaryHeader programHeader, kernelHeader;
|
||
|
|
CLElfLib::ElfBinaryStorage binary;
|
||
|
|
PTMap patchTokens;
|
||
|
|
std::string binaryFile, pathToPatch, pathToDump;
|
||
|
|
void dumpField(void *&binaryPtr, const PTField &field, std::ostream &ptmFile);
|
||
|
|
uint8_t getSize(const std::string &typeStr);
|
||
|
|
void *getDevBinary();
|
||
|
|
void parseTokens();
|
||
|
|
void printHelp();
|
||
|
|
int processBinary(void *&ptr, std::ostream &ptmFile);
|
||
|
|
void processKernel(void *&ptr, std::ostream &ptmFile);
|
||
|
|
void readPatchTokens(void *&patchListPtr, uint32_t patchListSize, std::ostream &ptmFile);
|
||
|
|
uint32_t readStructFields(const std::vector<std::string> &patchList,
|
||
|
|
const size_t &structPos, std::vector<PTField> &fields);
|
||
|
|
};
|