vxcontrol/MemoryModule
1
0
Fork 0
mirror of https://github.com/vxcontrol/MemoryModule.git synced 2026-07-21 06:05:55 -04:00
Code Issues Packages Projects Releases Wiki Activity

Execute thread-local-storage callbacks (GitHub #11).

Browse Source 
Pull request by @glysbaysb, cleaned up and adjusted to match code style.
This commit is contained in:
Joachim Bauch
2013-12-07 18:16:14 +01:00
parent 147fdde848
commit e75438a006
1 changed files with 21 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
MemoryModule.c
+21
View File
@@ -184,6 +184,24 @@ FinalizeSections(PMEMORYMODULE module)
#endif
}
static void
ExecuteTLS(PMEMORYMODULE module)
{
unsigned char *codeBase = module->codeBase;
PIMAGE_DATA_DIRECTORY directory = GET_HEADER_DICTIONARY(module, IMAGE_DIRECTORY_ENTRY_TLS);
if (directory->VirtualAddress > 0) {
PIMAGE_TLS_DIRECTORY tls = (PIMAGE_TLS_DIRECTORY) (codeBase + directory->VirtualAddress);
PIMAGE_TLS_CALLBACK* callback = (PIMAGE_TLS_CALLBACK *) tls->AddressOfCallBacks;
if (callback) {
while (*callback) {
(*callback)((LPVOID) codeBase, DLL_PROCESS_ATTACH, NULL);
callback++;
}
}
}
}
static void
PerformBaseRelocation(PMEMORYMODULE module, SIZE_T delta)
{
@@ -419,6 +437,9 @@ HMEMORYMODULE MemoryLoadLibraryEx(const void *data,
// sections that are marked as "discardable"
FinalizeSections(result);
// TLS callbacks are executed BEFORE the main loading
ExecuteTLS(result);
// get entry point of loaded library
if (result->headers->OptionalHeader.AddressOfEntryPoint != 0) {
DllEntry = (DllEntryProc) (code + result->headers->OptionalHeader.AddressOfEntryPoint);