70 lines
1.8 KiB
C++
70 lines
1.8 KiB
C++
|
#include "test_fix_dotnet.h"
|
||
|
|
|
||
|
|
#include <peconv.h>
|
||
|
|
#include "..\\libpeconv\src\fix_dot_net_ep.h"
|
||
|
|
|
||
|
|
namespace tests {
|
||
|
|
|
||
|
|
bool test_finding_offset(BYTE *buf, const size_t buf_size, size_t pattern_offset, bool direction_start=true)
|
||
|
|
{
|
||
|
|
const BYTE pattern[] = { 0xFF, 0x25, 0x00, 0x20, 0x40, 0x00 };
|
||
|
|
const ULONGLONG img_base = 0x400000;
|
||
|
|
const DWORD cor_exe_main_thunk = 0x2000;
|
||
|
|
|
||
|
|
// reset buffer:
|
||
|
|
memset(buf, 0, buf_size);
|
||
|
|
|
||
|
|
if (!direction_start) {
|
||
|
|
pattern_offset = (buf_size - sizeof(pattern)) - pattern_offset;
|
||
|
|
}
|
||
|
|
if (buf_size < sizeof(pattern) || (pattern_offset + sizeof(pattern)) > buf_size) {
|
||
|
|
std::cerr << __FILE__ << " incorrect test data!" << std::endl;
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
memcpy(buf + pattern_offset, pattern, sizeof(pattern));
|
||
|
|
|
||
|
|
BYTE* found = search_jump(buf, buf_size, cor_exe_main_thunk, img_base);
|
||
|
|
if (!found) {
|
||
|
|
std::cout << "Not found!\n";
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
size_t diff = found - buf;
|
||
|
|
std::cout << "Fount at offset: " << std::hex << diff << "\n";
|
||
|
|
if (diff == pattern_offset) {
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
int tests::check_finding_jumps()
|
||
|
|
{
|
||
|
|
BYTE buf[0x100] = { 0 };
|
||
|
|
|
||
|
|
bool is_ok = test_finding_offset(buf, sizeof(buf), 0, true);
|
||
|
|
if (is_ok) {
|
||
|
|
std::cout << "Test 1 passed!\n";
|
||
|
|
} else {
|
||
|
|
std::cout << "Test 1 failed!\n";
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
is_ok = test_finding_offset(buf, sizeof(buf), 0, false);
|
||
|
|
if (is_ok) {
|
||
|
|
std::cout << "Test 2 passed!\n";
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
std::cout << "Test 2 failed!\n";
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
is_ok = test_finding_offset(buf, sizeof(buf), 30, false);
|
||
|
|
if (is_ok) {
|
||
|
|
std::cout << "Test 3 passed!\n";
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
std::cout << "Test 2 failed!\n";
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|