添加项目文件。
This commit is contained in:
69
Etw Syscall/libpeconv-master/tests/test_fix_dotnet.cpp
Normal file
69
Etw Syscall/libpeconv-master/tests/test_fix_dotnet.cpp
Normal file
@@ -0,0 +1,69 @@
|
||||
#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;
|
||||
}
|
||||
Reference in New Issue
Block a user