97 lines
2.5 KiB
C++
97 lines
2.5 KiB
C++
|
#include "test_hooking_local.h"
|
||
|
|
|
||
|
|
#include <peconv.h>
|
||
|
|
using namespace peconv;
|
||
|
|
|
||
|
|
#include <iostream>
|
||
|
|
#include <string>
|
||
|
|
#include <map>
|
||
|
|
|
||
|
|
#define FAKE_NAME "fake_module_name"
|
||
|
|
|
||
|
|
namespace test11 {
|
||
|
|
int _stdcall my_MessageBoxA(
|
||
|
|
_In_opt_ HWND hWnd,
|
||
|
|
_In_opt_ LPCSTR lpText,
|
||
|
|
_In_opt_ LPCSTR lpCaption,
|
||
|
|
_In_ UINT uType)
|
||
|
|
{
|
||
|
|
std::cout << "TITLE: [" << lpCaption << "]" << std::endl;
|
||
|
|
std::cout << "MESSAGE: [" << lpText << "]" << std::endl;
|
||
|
|
return 1337;
|
||
|
|
}
|
||
|
|
|
||
|
|
int _stdcall my_MessageBoxW(
|
||
|
|
_In_opt_ HWND hWnd,
|
||
|
|
_In_opt_ LPCWSTR lpText,
|
||
|
|
_In_opt_ LPCWSTR lpCaption,
|
||
|
|
_In_ UINT uType)
|
||
|
|
{
|
||
|
|
std::wcout << L"TITLE: [" << lpCaption << L"]" << std::endl;
|
||
|
|
std::wcout << L"MESSAGE: [" << lpText << L"]" << std::endl;
|
||
|
|
return 1338;
|
||
|
|
}
|
||
|
|
|
||
|
|
int __cdecl my_rand(void)
|
||
|
|
{
|
||
|
|
return 44;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD
|
||
|
|
WINAPI
|
||
|
|
my_GetModuleFileNameA(
|
||
|
|
IN OPTIONAL HMODULE hModule,
|
||
|
|
OUT LPSTR lpFilename,
|
||
|
|
IN DWORD nSize
|
||
|
|
)
|
||
|
|
{
|
||
|
|
const char fake_name[] = FAKE_NAME;
|
||
|
|
size_t to_copy = strlen(fake_name);
|
||
|
|
if (to_copy < nSize) to_copy = nSize;
|
||
|
|
|
||
|
|
memcpy(lpFilename, fake_name, to_copy);
|
||
|
|
return to_copy;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
int tests::hook_self_local()
|
||
|
|
{
|
||
|
|
char normal_name[MAX_PATH] = { 0 };
|
||
|
|
GetModuleFileNameA(NULL, normal_name, MAX_PATH);
|
||
|
|
|
||
|
|
HMODULE user32_lib = LoadLibraryA("user32.dll");
|
||
|
|
HMODULE kernel32_lib = LoadLibraryA("kernel32.dll");
|
||
|
|
PatchBackup backup;
|
||
|
|
|
||
|
|
peconv::redirect_to_local(GetProcAddress(user32_lib, "MessageBoxA"), &test11::my_MessageBoxA);
|
||
|
|
peconv::redirect_to_local(GetProcAddress(kernel32_lib, "GetModuleFileNameA"), &test11::my_GetModuleFileNameA, &backup);
|
||
|
|
peconv::redirect_to_local(rand, &test11::my_rand);
|
||
|
|
|
||
|
|
char module_name[MAX_PATH] = { 0 };
|
||
|
|
GetModuleFileNameA(NULL, module_name, MAX_PATH);
|
||
|
|
MessageBoxA(0, module_name, "Module Name", MB_OK);
|
||
|
|
|
||
|
|
if (strcmp(FAKE_NAME, module_name) != 0) {
|
||
|
|
std::cout << "Failed!";
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
srand(10000);
|
||
|
|
int rand_val = rand();
|
||
|
|
if (rand_val != 44) {
|
||
|
|
std::cout << "Failed: " << rand_val << "\n";
|
||
|
|
return -2;
|
||
|
|
}
|
||
|
|
if (!backup.applyBackup()) {
|
||
|
|
std::cout << "Failed! Cannot apply backup.";
|
||
|
|
return -3;
|
||
|
|
}
|
||
|
|
GetModuleFileNameA(NULL, module_name, MAX_PATH);
|
||
|
|
MessageBoxA(0, module_name, "Module Name", MB_OK);
|
||
|
|
|
||
|
|
if (strcmp(normal_name, module_name) != 0) {
|
||
|
|
std::cout << "Failed!";
|
||
|
|
return -4;
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|