135 lines
3.8 KiB
C++
135 lines
3.8 KiB
C++
/****************************************************************************
|
|
* Copyright (C) 2018 Maschell
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
****************************************************************************/
|
|
|
|
#ifndef _PLUGIN_DATA_H_
|
|
#define _PLUGIN_DATA_H_
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <map>
|
|
#include "FunctionData.h"
|
|
#include "HookData.h"
|
|
#include "PluginInformation.h"
|
|
#include "RelocationData.h"
|
|
#include "ImportRPLInformation.h"
|
|
#include <utils/logger.h>
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
#include <libelf.h>
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
class PluginData {
|
|
public:
|
|
PluginData(PluginInformation * pluginInformation) {
|
|
this->pluginInformation = pluginInformation;
|
|
}
|
|
|
|
~PluginData() {
|
|
for(size_t i = 0; i< function_data_list.size(); i++) {
|
|
if(function_data_list[i] != NULL) {
|
|
delete function_data_list[i];
|
|
}
|
|
}
|
|
|
|
for(size_t i = 0; i< hook_data_list.size(); i++) {
|
|
if(hook_data_list[i] != NULL) {
|
|
delete hook_data_list[i];
|
|
}
|
|
}
|
|
|
|
for(size_t i = 0; i< relocation_data_list.size(); i++) {
|
|
if(relocation_data_list[i] != NULL) {
|
|
delete relocation_data_list[i];
|
|
}
|
|
}
|
|
|
|
for(size_t i = 0; i< importRPLInformation_list.size(); i++) {
|
|
if(importRPLInformation_list[i] != NULL) {
|
|
delete importRPLInformation_list[i];
|
|
}
|
|
}
|
|
}
|
|
|
|
void addFunctionData(FunctionData * function_data) {
|
|
function_data_list.push_back(function_data);
|
|
}
|
|
|
|
std::vector<FunctionData *> getFunctionDataList() {
|
|
return function_data_list;
|
|
}
|
|
|
|
void addHookData(HookData * hook_data) {
|
|
hook_data_list.push_back(hook_data);
|
|
}
|
|
|
|
std::vector<HookData *> getHookDataList() {
|
|
return hook_data_list;
|
|
}
|
|
|
|
void addRelocationData(RelocationData * relocation_data) {
|
|
relocation_data_list.push_back(relocation_data);
|
|
}
|
|
|
|
std::vector<RelocationData *> getRelocationDataList() {
|
|
return relocation_data_list;
|
|
}
|
|
|
|
void addImportRPLInformation(ImportRPLInformation * importRPLInformation) {
|
|
importRPLInformation_list.push_back(importRPLInformation);
|
|
}
|
|
|
|
std::vector<ImportRPLInformation *> getImportRPLInformationList() {
|
|
return importRPLInformation_list;
|
|
}
|
|
|
|
/**
|
|
Returns a ImportRPLInformation for a given section header index.
|
|
|
|
\param section_header_index: ID of section in elf, started counting at 1.
|
|
|
|
\return A pointer to the corresponding ImportRPLInformation, return NULL if no corresponding information was found.
|
|
**/
|
|
ImportRPLInformation * getImportRPLInformationBySectionHeaderIndex(int32_t section_header_index);
|
|
|
|
|
|
PluginInformation * getPluginInformation() {
|
|
return pluginInformation;
|
|
}
|
|
|
|
uint32_t getMemoryForCommonBySymbol(size_t symbol, size_t align, size_t size);
|
|
|
|
private:
|
|
|
|
PluginInformation * pluginInformation;
|
|
|
|
std::vector<FunctionData *> function_data_list;
|
|
std::vector<HookData *> hook_data_list;
|
|
std::vector<RelocationData *> relocation_data_list;
|
|
std::vector<ImportRPLInformation *> importRPLInformation_list;
|
|
|
|
std::map<size_t, uint32_t> memoryBySymbol;
|
|
};
|
|
|
|
|
|
#endif
|