diff --git a/MemoryModule.c b/MemoryModule.c index aefda09..79e843e 100644 --- a/MemoryModule.c +++ b/MemoryModule.c @@ -447,9 +447,9 @@ static void _FreeLibrary(HCUSTOMMODULE module, void *userdata) FreeLibrary((HMODULE) module); } -HMEMORYMODULE MemoryLoadLibrary(const void *data, size_t size) +HMEMORYMODULE MemoryLoadLibrary(const void *data, size_t size, const char * argv[]) { - return MemoryLoadLibraryEx(data, size, _LoadLibrary, _GetProcAddress, _FreeLibrary, NULL); + return MemoryLoadLibraryEx(data, size, _LoadLibrary, _GetProcAddress, _FreeLibrary, argv); } HMEMORYMODULE MemoryLoadLibraryEx(const void *data, size_t size, diff --git a/MemoryModule.h b/MemoryModule.h index 2cd4867..39d23b8 100644 --- a/MemoryModule.h +++ b/MemoryModule.h @@ -49,7 +49,7 @@ typedef void (*CustomFreeLibraryFunc)(HCUSTOMMODULE, void *); * All dependencies are resolved using default LoadLibrary/GetProcAddress * calls through the Windows API. */ -HMEMORYMODULE MemoryLoadLibrary(const void *, size_t); +HMEMORYMODULE MemoryLoadLibrary(const void *, size_t, const char *[]); /** * Load EXE/DLL from memory location with the given size using custom dependency diff --git a/example/DllLoader/DllLoader.cpp b/example/DllLoader/DllLoader.cpp index 4ff1a20..65f0590 100644 --- a/example/DllLoader/DllLoader.cpp +++ b/example/DllLoader/DllLoader.cpp @@ -76,7 +76,7 @@ void LoadFromMemory(void) assert(read == static_cast(size)); fclose(fp); - handle = MemoryLoadLibrary(data, size); + handle = MemoryLoadLibrary(data, size, NULL); if (handle == NULL) { _tprintf(_T("Can't load library from memory.\n")); diff --git a/example/DllLoader/DllLoaderLoader.cpp b/example/DllLoader/DllLoaderLoader.cpp index b7174f5..178be31 100644 --- a/example/DllLoader/DllLoaderLoader.cpp +++ b/example/DllLoader/DllLoaderLoader.cpp @@ -39,7 +39,7 @@ int RunFromMemory(void) assert(read == static_cast(size)); fclose(fp); - handle = MemoryLoadLibrary(data, size); + handle = MemoryLoadLibrary(data, size, NULL); if (handle == NULL) { _tprintf(_T("Can't load library from memory.\n"));