fabi/Xenia-Canary
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

XexGetProcedureAddress by name implemented.

Browse Source
This commit is contained in:
Dr. Chat
2015-05-04 17:38:34 -05:00
parent 78921c1a7e
commit 49c8efdc6d
8 changed files with 101 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
src/xenia/kernel/util/xex2.cc
View File

@@ -49,6 +49,8 @@ typedef struct xe_xex2 {
size_t count;
xe_xex2_import_info_t *infos;
} library_imports[16];
xe_xex2_export_table *export_table;
} xe_xex2_t;
int xe_xex2_read_header(const uint8_t *addr, const size_t length,
@@ -237,6 +239,11 @@ int xe_xex2_read_header(const uint8_t *addr, const size_t length,
case XEX_HEADER_DEFAULT_HEAP_SIZE:
header->exe_heap_size = opt_header->value;
break;
case XEX_HEADER_EXPORTS_BY_NAME: {
// IMAGE_DATA_DIRECTORY (w/ offset from PE file base)
header->export_table_offset = xe::load_and_swap<uint32_t>(pp);
// size = xe::load_and_swap<uint32_t>(pp + 0x04);
} break;
case XEX_HEADER_IMPORT_LIBRARIES: {
const size_t max_count = xe::countof(header->import_libraries);
size_t count = xe::load_and_swap<uint32_t>(pp + 0x08);
@@ -869,6 +876,20 @@ int xe_xex2_load_pe(xe_xex2_ref xex) {
xex->sections->push_back(section);
}
if (header->export_table_offset) {
// This table is located inside of the PE (for some reason)
xe_xex2_export_table *table = (xe_xex2_export_table *)xex->memory->TranslateVirtual(xex->header.loader_info.export_table);
xex->export_table = table;
// cmp magic...
table->base = xe::load_and_swap<uint32_t>(&table->base);
table->imagebaseaddr = xe::load_and_swap<uint32_t>(&table->imagebaseaddr);
table->count = xe::load_and_swap<uint32_t>(&table->count);
for (int i = 0; i < table->count; i++) {
table->ordOffset[i] = xe::load_and_swap<uint32_t>(&table->ordOffset[i]);
}
}
// DumpTLSDirectory(pImageBase, pNTHeader, (PIMAGE_TLS_DIRECTORY32)0);
// DumpExportsSection(pImageBase, pNTHeader);
return 0;
@@ -982,3 +1003,41 @@ int xe_xex2_get_import_infos(xe_xex2_ref xex,
*out_import_infos = xex->library_imports[library_index].infos;
return 0;
}
int xe_xex2_lookup_export(xe_xex2_ref xex, const char *name,
PEExport &peexport) {
auto header = xe_xex2_get_header(xex);
// no exports :(
if (!header->export_table_offset) {
return 1;
}
uint64_t loadaddr = (uint64_t)xex->memory->TranslateVirtual(header->loader_info.load_address);
IMAGE_EXPORT_DIRECTORY *e = (PIMAGE_EXPORT_DIRECTORY)(loadaddr + header->export_table_offset);
// e->AddressOfX RVAs are relative to the IMAGE_EXPORT_DIRECTORY!
uint32_t* function_table = (uint32_t*)((uint64_t)e + e->AddressOfFunctions); // Functions relative to base
uint32_t* name_table = (uint32_t*)((uint64_t)e + e->AddressOfNames); // Names relative to directory
uint16_t* ordinal_table = (uint16_t*)((uint64_t)e + e->AddressOfNameOrdinals); // Table of ordinals
const char* mod_name = (const char*)((uint64_t)e + e->Name);
for (int i = 0; i < e->NumberOfNames; i++) {
const char *fn_name = (const char *)((uint64_t)e + name_table[i]);
uint16_t ordinal = ordinal_table[i];
uint64_t addr = (uint64_t)(loadaddr + function_table[ordinal]);
if (!strcmp(name, fn_name)) {
// We have a match!
peexport.name = fn_name;
peexport.addr = addr;
peexport.ordinal = ordinal;
return 0;
}
}
// No match
return 1;
}