mirror of
https://github.com/decompals/wibo.git
synced 2025-10-15 14:45:12 +00:00
42 lines
1.8 KiB
C
42 lines
1.8 KiB
C
#include <windows.h>
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
|
|
#include "test_assert.h"
|
|
|
|
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());
|
|
|
|
printf("external_exports: sum=%d attached=%d\n", sum, attached);
|
|
return EXIT_SUCCESS;
|
|
}
|