2020-04-29 18:02:36 +02:00
|
|
|
#include "DynamicLinkingHelper.h"
|
2020-12-26 14:17:50 +01:00
|
|
|
#include <cstring>
|
|
|
|
#include "../utils/logger.h"
|
2020-04-29 18:02:36 +02:00
|
|
|
|
2020-05-03 12:30:15 +02:00
|
|
|
dyn_linking_function_t *DynamicLinkingHelper::getOrAddFunctionEntryByName(dyn_linking_relocation_data_t *data, const char *functionName) {
|
2020-12-26 14:17:50 +01:00
|
|
|
if (data == nullptr) {
|
|
|
|
return nullptr;
|
2020-04-29 18:02:36 +02:00
|
|
|
}
|
2020-12-26 14:17:50 +01:00
|
|
|
if (functionName == nullptr) {
|
|
|
|
return nullptr;
|
2020-04-29 18:02:36 +02:00
|
|
|
}
|
2020-12-26 14:17:50 +01:00
|
|
|
dyn_linking_function_t *result = nullptr;
|
2020-05-03 12:30:15 +02:00
|
|
|
for (int32_t i = 0; i < DYN_LINK_FUNCTION_LIST_LENGTH; i++) {
|
|
|
|
dyn_linking_function_t *curEntry = &(data->functions[i]);
|
|
|
|
if (strlen(curEntry->functionName) == 0) {
|
|
|
|
if (strlen(functionName) > DYN_LINK_FUNCTION_NAME_LENGTH) {
|
2020-12-28 14:40:53 +01:00
|
|
|
DEBUG_FUNCTION_LINE("Failed to add function name, it's too long.");
|
2020-12-26 14:17:50 +01:00
|
|
|
return nullptr;
|
2020-04-29 18:02:36 +02:00
|
|
|
}
|
2020-05-03 12:30:15 +02:00
|
|
|
strncpy(curEntry->functionName, functionName, DYN_LINK_FUNCTION_NAME_LENGTH);
|
2020-04-29 18:02:36 +02:00
|
|
|
result = curEntry;
|
|
|
|
break;
|
|
|
|
}
|
2020-05-03 12:30:15 +02:00
|
|
|
if (strncmp(curEntry->functionName, functionName, DYN_LINK_FUNCTION_NAME_LENGTH) == 0) {
|
2020-04-29 18:02:36 +02:00
|
|
|
result = curEntry;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2020-05-03 12:30:15 +02:00
|
|
|
dyn_linking_import_t *DynamicLinkingHelper::getOrAddFunctionImportByName(dyn_linking_relocation_data_t *data, const char *importName) {
|
2020-04-29 18:02:36 +02:00
|
|
|
return getOrAddImport(data, importName, false);
|
|
|
|
}
|
|
|
|
|
2020-05-03 12:30:15 +02:00
|
|
|
dyn_linking_import_t *DynamicLinkingHelper::getOrAddDataImportByName(dyn_linking_relocation_data_t *data, const char *importName) {
|
2020-04-29 18:02:36 +02:00
|
|
|
return getOrAddImport(data, importName, true);
|
|
|
|
}
|
|
|
|
|
2020-05-03 12:30:15 +02:00
|
|
|
dyn_linking_import_t *DynamicLinkingHelper::getOrAddImport(dyn_linking_relocation_data_t *data, const char *importName, bool isData) {
|
2020-12-26 14:17:50 +01:00
|
|
|
if (importName == nullptr || data == nullptr) {
|
|
|
|
return nullptr;
|
2020-04-29 18:02:36 +02:00
|
|
|
}
|
2020-12-26 14:17:50 +01:00
|
|
|
dyn_linking_import_t *result = nullptr;
|
2020-05-03 12:30:15 +02:00
|
|
|
for (int32_t i = 0; i < DYN_LINK_IMPORT_LIST_LENGTH; i++) {
|
|
|
|
dyn_linking_import_t *curEntry = &(data->imports[i]);
|
|
|
|
if (strlen(curEntry->importName) == 0) {
|
|
|
|
if (strlen(importName) > DYN_LINK_IMPORT_NAME_LENGTH) {
|
2020-12-28 14:40:53 +01:00
|
|
|
DEBUG_FUNCTION_LINE("Failed to add Import, it's too long.");
|
2020-12-26 14:17:50 +01:00
|
|
|
return nullptr;
|
2020-04-29 18:02:36 +02:00
|
|
|
}
|
2020-05-03 12:30:15 +02:00
|
|
|
strncpy(curEntry->importName, importName, DYN_LINK_IMPORT_NAME_LENGTH);
|
2020-04-29 18:02:36 +02:00
|
|
|
curEntry->isData = isData;
|
|
|
|
result = curEntry;
|
|
|
|
break;
|
|
|
|
}
|
2020-05-03 12:30:15 +02:00
|
|
|
if (strncmp(curEntry->importName, importName, DYN_LINK_IMPORT_NAME_LENGTH) == 0 && (curEntry->isData == isData)) {
|
2020-04-29 18:02:36 +02:00
|
|
|
return curEntry;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2020-05-03 12:30:15 +02:00
|
|
|
bool DynamicLinkingHelper::addReloationEntry(dyn_linking_relocation_data_t *linking_data, dyn_linking_relocation_entry_t *linking_entries, uint32_t linking_entry_length, const RelocationData &relocationData) {
|
|
|
|
return addReloationEntry(linking_data, linking_entries, linking_entry_length, relocationData.getType(), relocationData.getOffset(), relocationData.getAddend(), relocationData.getDestination(), relocationData.getName(),
|
|
|
|
relocationData.getImportRPLInformation());
|
2020-04-29 18:02:36 +02:00
|
|
|
}
|
|
|
|
|
2020-05-03 12:30:15 +02:00
|
|
|
bool DynamicLinkingHelper::addReloationEntry(dyn_linking_relocation_data_t *linking_data, dyn_linking_relocation_entry_t *linking_entries, uint32_t linking_entry_length, char type, size_t offset, int32_t addend, const void *destination,
|
|
|
|
const std::string &name, const ImportRPLInformation &rplInfo) {
|
|
|
|
dyn_linking_import_t *importInfoGbl = DynamicLinkingHelper::getOrAddImport(linking_data, rplInfo.getName().c_str(), rplInfo.isData());
|
2020-12-26 14:17:50 +01:00
|
|
|
if (importInfoGbl == nullptr) {
|
2020-12-28 14:40:53 +01:00
|
|
|
DEBUG_FUNCTION_LINE("Getting import info failed. Probably maximum of %d rpl files to import reached.", DYN_LINK_IMPORT_LIST_LENGTH);
|
2020-04-29 18:02:36 +02:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2020-05-03 12:30:15 +02:00
|
|
|
dyn_linking_function_t *functionInfo = DynamicLinkingHelper::getOrAddFunctionEntryByName(linking_data, name.c_str());
|
2020-12-26 14:17:50 +01:00
|
|
|
if (functionInfo == nullptr) {
|
2020-12-28 14:40:53 +01:00
|
|
|
DEBUG_FUNCTION_LINE("Getting import info failed. Probably maximum of %d function to be relocated reached.", DYN_LINK_FUNCTION_LIST_LENGTH);
|
2020-04-29 18:02:36 +02:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
return addReloationEntry(linking_entries, linking_entry_length, type, offset, addend, destination, functionInfo, importInfoGbl);
|
|
|
|
}
|
|
|
|
|
2020-05-03 12:30:15 +02:00
|
|
|
bool DynamicLinkingHelper::addReloationEntry(dyn_linking_relocation_entry_t *linking_entries, uint32_t linking_entry_length, char type, size_t offset, int32_t addend, const void *destination, dyn_linking_function_t *functionName,
|
|
|
|
dyn_linking_import_t *importInfo) {
|
|
|
|
for (uint32_t i = 0; i < linking_entry_length; i++) {
|
|
|
|
dyn_linking_relocation_entry_t *curEntry = &(linking_entries[i]);
|
2020-12-26 14:17:50 +01:00
|
|
|
if (curEntry->functionEntry != nullptr) {
|
2020-04-29 18:02:36 +02:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
curEntry->type = type;
|
|
|
|
curEntry->offset = offset;
|
|
|
|
curEntry->addend = addend;
|
2020-05-03 12:30:15 +02:00
|
|
|
curEntry->destination = (void *) destination;
|
2020-04-29 18:02:36 +02:00
|
|
|
curEntry->functionEntry = functionName;
|
|
|
|
curEntry->importEntry = importInfo;
|
|
|
|
return true;
|
|
|
|
}
|
2020-05-03 10:21:05 +02:00
|
|
|
DEBUG_FUNCTION_LINE("Failed to find empty slot for saving relocations entry. We ned more than %d slots.", linking_entry_length);
|
2020-04-29 18:02:36 +02:00
|
|
|
return false;
|
|
|
|
}
|