Refactor main module resolution & HMODULE handle behavior

This commit is contained in:
2025-09-29 13:50:27 -06:00
parent 4a2ba45620
commit c17953b318
13 changed files with 355 additions and 224 deletions

View File

@@ -7,23 +7,32 @@
int main(void) {
typedef int(__stdcall *add_numbers_fn)(int, int);
typedef int(__stdcall *was_attached_fn)(void);
typedef HMODULE(__stdcall *observed_main_module_fn)(void);
HMODULE initial_main = GetModuleHandleW(NULL);
TEST_CHECK(initial_main != NULL);
HMODULE mod = LoadLibraryA("external_exports.dll");
TEST_CHECK_MSG(mod != NULL, "LoadLibraryA failed: %lu", (unsigned long)GetLastError());
FARPROC raw_add_numbers = GetProcAddress(mod, "add_numbers@8");
FARPROC raw_was_attached = GetProcAddress(mod, "was_attached@0");
FARPROC raw_observed_main = GetProcAddress(mod, "observed_main_module@0");
TEST_CHECK_MSG(raw_add_numbers != NULL, "GetProcAddress(add_numbers@8) failed: %lu", (unsigned long)GetLastError());
TEST_CHECK_MSG(raw_was_attached != NULL, "GetProcAddress(was_attached@0) failed: %lu", (unsigned long)GetLastError());
TEST_CHECK_MSG(raw_observed_main != NULL, "GetProcAddress(observed_main_module@0) failed: %lu", (unsigned long)GetLastError());
add_numbers_fn add_numbers = (add_numbers_fn)(uintptr_t)raw_add_numbers;
was_attached_fn was_attached = (was_attached_fn)(uintptr_t)raw_was_attached;
observed_main_module_fn observed_main = (observed_main_module_fn)(uintptr_t)raw_observed_main;
int sum = add_numbers(2, 40);
int attached = was_attached();
HMODULE observed_main_module = observed_main();
TEST_CHECK_EQ(42, sum);
TEST_CHECK_EQ(1, attached);
TEST_CHECK_EQ(initial_main, observed_main_module);
TEST_CHECK_MSG(FreeLibrary(mod) != 0, "FreeLibrary failed: %lu", (unsigned long)GetLastError());