void TestPeFile()
{
hadesmem::Process const process(::GetCurrentProcessId());
hadesmem::PeFile pe_file_1(
process, ::GetModuleHandleW(nullptr), hadesmem::PeFileType::kImage, 0);
hadesmem::PeFile pe_file_2(pe_file_1);
BOOST_TEST_EQ(pe_file_1, pe_file_2);
pe_file_1 = pe_file_2;
BOOST_TEST_EQ(pe_file_1, pe_file_2);
hadesmem::PeFile pe_file_3(std::move(pe_file_2));
BOOST_TEST_EQ(pe_file_3, pe_file_1);
pe_file_2 = std::move(pe_file_3);
BOOST_TEST_EQ(pe_file_1, pe_file_2);
hadesmem::PeFile pe_file_this(
process, ::GetModuleHandleW(nullptr), hadesmem::PeFileType::kImage, 0);
BOOST_TEST_EQ(pe_file_this.GetBase(), ::GetModuleHandle(nullptr));
BOOST_TEST(pe_file_this.GetType() == hadesmem::PeFileType::kImage);
BOOST_TEST_EQ(hadesmem::RvaToVa(process, pe_file_this, 0),
static_cast<void*>(nullptr));
hadesmem::PeFile pe_file_ntdll(
process, ::GetModuleHandleW(L"ntdll"), hadesmem::PeFileType::kImage, 0);
BOOST_TEST_EQ(pe_file_ntdll.GetBase(), ::GetModuleHandle(L"ntdll"));
BOOST_TEST(pe_file_ntdll.GetType() == hadesmem::PeFileType::kImage);
BOOST_TEST_EQ(hadesmem::RvaToVa(process, pe_file_ntdll, 0),
static_cast<void*>(nullptr));
BOOST_TEST_EQ(pe_file_this, pe_file_this);
BOOST_TEST_NE(pe_file_this, pe_file_ntdll);
BOOST_TEST_NE(pe_file_ntdll, pe_file_this);
if (pe_file_this > pe_file_ntdll)
{
BOOST_TEST(pe_file_this > pe_file_ntdll);
BOOST_TEST(pe_file_this >= pe_file_ntdll);
BOOST_TEST(!(pe_file_this < pe_file_ntdll));
BOOST_TEST(!(pe_file_this <= pe_file_ntdll));
}
else
{
BOOST_TEST(pe_file_ntdll > pe_file_this);
BOOST_TEST(pe_file_ntdll >= pe_file_this);
BOOST_TEST(!(pe_file_ntdll < pe_file_this));
BOOST_TEST(!(pe_file_ntdll <= pe_file_this));
}
std::stringstream test_str_1;
test_str_1.imbue(std::locale::classic());
test_str_1 << pe_file_this;
std::stringstream test_str_2;
test_str_2.imbue(std::locale::classic());
test_str_2 << pe_file_this.GetBase();
BOOST_TEST_EQ(test_str_1.str(), test_str_2.str());
std::stringstream test_str_3;
test_str_3.imbue(std::locale::classic());
test_str_3 << pe_file_ntdll.GetBase();
BOOST_TEST_NE(test_str_1.str(), test_str_3.str());
}
void TestThreadList()
{
using ThreadListIterCat =
std::iterator_traits<hadesmem::ThreadList::iterator>::iterator_category;
HADESMEM_DETAIL_STATIC_ASSERT(
std::is_base_of<std::input_iterator_tag, ThreadListIterCat>::value);
using ThreadListConstIterCat = std::iterator_traits<
hadesmem::ThreadList::const_iterator>::iterator_category;
HADESMEM_DETAIL_STATIC_ASSERT(
std::is_base_of<std::input_iterator_tag, ThreadListConstIterCat>::value);
{
hadesmem::detail::SmartHandle wait_thread_1(
::CreateThread(nullptr, 0, &SleepWrapper, nullptr, 0, nullptr));
hadesmem::detail::SmartHandle wait_thread_2(
::CreateThread(nullptr, 0, &SleepWrapper, nullptr, 0, nullptr));
}
hadesmem::ThreadList const thread_list_1(::GetCurrentProcessId());
hadesmem::ThreadList thread_list_2(thread_list_1);
hadesmem::ThreadList thread_list_3(std::move(thread_list_2));
thread_list_2 = std::move(thread_list_3);
BOOST_TEST(std::begin(thread_list_2) != std::end(thread_list_2));
auto iter = std::begin(thread_list_1);
BOOST_TEST(iter != std::end(thread_list_1));
BOOST_TEST(++iter != std::end(thread_list_1));
BOOST_TEST_NE(iter->GetId(), 0U);
DWORD const second_id = iter->GetId();
BOOST_TEST(++iter != std::end(thread_list_1));
DWORD const third_id = iter->GetId();
BOOST_TEST_NE(second_id, third_id);
}
int main(int, char*[])
{
#if !defined(BOOST_UUID_TEST_RANDOM_MOCK) // Positive Testing
boost::uuids::detail::random_provider provider;
boost::array<unsigned int, 2> ints;
// test generator()
ints[0] = 0;
ints[1] = 0;
provider.generate(ints.begin(), ints.end());
BOOST_TEST_NE(ints[0], ints[1]);
// test name()
BOOST_TEST_GT(strlen(provider.name()), 4u);
// test get_random_bytes()
char buf1[64];
char buf2[64];
provider.get_random_bytes(buf1, 64);
provider.get_random_bytes(buf2, 64);
BOOST_TEST_NE(0, memcmp(buf1, buf2, 64));
#else // Negative Testing
if (expectations_capable())
{
// Test fail acquiring context if the provider supports it
if (provider_acquires_context())
{
expect_next_call_success(false);
BOOST_TEST_THROWS(boost::uuids::detail::random_provider(),
boost::uuids::entropy_error);
BOOST_TEST(expectations_met());
}
// Test fail acquiring entropy
if (provider_acquires_context())
{
expect_next_call_success(true);
}
expect_next_call_success(false);
// 4 is important for the posix negative test (partial read) to work properly
// as it sees a size of 4, returns 1, causing a 2nd loop to read 3 more bytes...
char buf[4];
BOOST_TEST_THROWS(boost::uuids::detail::random_provider().get_random_bytes(buf, 4),
boost::uuids::entropy_error);
BOOST_TEST(expectations_met());
}
#endif
return boost::report_errors();
}
int main(int, char*[])
{
object o1(1);
object o2 = o1;
BOOST_TEST_EQ(o1, o2);
o2.set_state(2);
BOOST_TEST_EQ(o1, o2);
object o3;
o3.set_state(3);
BOOST_TEST_NE(o1, o3);
BOOST_TEST_NE(o2, o3);
return boost::report_errors();
}
int main()
{
boost::hash< boost::intrusive_ptr<X> > hasher;
boost::intrusive_ptr<X> p1, p2( p1 ), p3( new X ), p4( p3 ), p5( new X );
BOOST_TEST_EQ( p1, p2 );
BOOST_TEST_EQ( hasher( p1 ), hasher( p2 ) );
BOOST_TEST_NE( p1, p3 );
BOOST_TEST_NE( hasher( p1 ), hasher( p3 ) );
BOOST_TEST_EQ( p3, p4 );
BOOST_TEST_EQ( hasher( p3 ), hasher( p4 ) );
BOOST_TEST_NE( p3, p5 );
BOOST_TEST_NE( hasher( p3 ), hasher( p5 ) );
return boost::report_errors();
}
void TestRegionListAlgorithm()
{
hadesmem::Process const process(::GetCurrentProcessId());
hadesmem::RegionList const region_list_1(process);
for (auto const& region : region_list_1)
{
hadesmem::Region const other(process, region.GetBase());
BOOST_TEST_EQ(region, other);
if (region.GetState() != MEM_FREE)
{
BOOST_TEST_NE(region.GetBase(), static_cast<void*>(nullptr));
BOOST_TEST_NE(region.GetAllocBase(), static_cast<void*>(nullptr));
BOOST_TEST_NE(region.GetAllocProtect(), 0U);
BOOST_TEST_NE(region.GetType(), 0U);
}
region.GetProtect();
BOOST_TEST_NE(region.GetSize(), 0U);
BOOST_TEST_NE(region.GetState(), 0U);
}
HMODULE const user32_mod = GetModuleHandle(L"user32.dll");
auto user32_iter = std::find_if(std::begin(region_list_1),
std::end(region_list_1),
[user32_mod](hadesmem::Region const& region)
{
return region.GetBase() == user32_mod;
});
BOOST_TEST(user32_iter != std::end(region_list_1));
}
int main(int, char*[])
{
uuid_class u1;
uuid_class u2;
BOOST_TEST_NE(u1, u2);
BOOST_TEST_EQ(u1.is_nil(), false);
BOOST_TEST_EQ(u2.is_nil(), false);
u2 = u1;
BOOST_TEST_EQ(u1, u2);
return boost::report_errors();
}
int main()
{
int x = 0;
// BOOST_TEST
BOOST_TEST( x == 0 );
BOOST_TEST( ++x == 1 );
BOOST_TEST( x++ == 1 );
BOOST_TEST( x == 2? true: false );
BOOST_TEST( x == 2? &x: 0 );
// BOOST_TEST_EQ
BOOST_TEST_EQ( x, 2 );
BOOST_TEST_EQ( ++x, 3 );
BOOST_TEST_EQ( x++, 3 );
int y = 4;
BOOST_TEST_EQ( ++x, ++y );
BOOST_TEST_EQ( x++, y++ );
// BOOST_TEST_NE
BOOST_TEST_NE( ++x, y );
BOOST_TEST_NE( &x, &y );
// BOOST_TEST_THROWS
BOOST_TEST_THROWS( throw X(), X );
BOOST_TEST_THROWS( throw 1, int );
BOOST_TEST_THROWS( f(true), X );
BOOST_TEST_THROWS( f(false), int );
return boost::report_errors();
}
void TestThreadListAlgorithm()
{
hadesmem::ThreadList const thread_list_1(::GetCurrentProcessId());
for (auto const& entry : thread_list_1)
{
BOOST_TEST_EQ(entry.GetUsage(), 0UL);
BOOST_TEST_NE(entry.GetId(), 0UL);
BOOST_TEST_EQ(entry.GetOwnerId(), ::GetCurrentProcessId());
BOOST_TEST(entry.GetBasePriority() >= 0L);
BOOST_TEST(entry.GetBasePriority() <= 31L);
BOOST_TEST_EQ(entry.GetDeltaPriority(), 0L);
BOOST_TEST_EQ(entry.GetFlags(), 0UL);
}
auto const this_iter = std::find_if(std::begin(thread_list_1),
std::end(thread_list_1),
[](hadesmem::ThreadEntry const& thread)
{
return thread.GetId() == ::GetCurrentThreadId();
});
BOOST_TEST(this_iter != std::end(thread_list_1));
}
void TestRegion()
{
hadesmem::Process const process(::GetCurrentProcessId());
hadesmem::Region const first_region(process, nullptr);
hadesmem::Region first_region_2(first_region);
hadesmem::Region first_region_3(std::move(first_region_2));
first_region_2 = std::move(first_region_3);
BOOST_TEST_EQ(first_region, first_region_2);
if (first_region.GetState() != MEM_FREE)
{
BOOST_TEST_NE(first_region.GetBase(), static_cast<void*>(nullptr));
BOOST_TEST_NE(first_region.GetAllocBase(), static_cast<void*>(nullptr));
BOOST_TEST_NE(first_region.GetAllocProtect(), 0U);
BOOST_TEST_NE(first_region.GetType(), 0U);
}
first_region.GetProtect();
BOOST_TEST_NE(first_region.GetSize(), 0U);
BOOST_TEST_NE(first_region.GetState(), 0U);
hadesmem::Region const first_region_other(
process, static_cast<PBYTE>(first_region.GetBase()) + 1);
BOOST_TEST_EQ(first_region.GetBase(), first_region_other.GetBase());
BOOST_TEST_EQ(first_region.GetAllocBase(), first_region_other.GetAllocBase());
BOOST_TEST_EQ(first_region.GetAllocProtect(),
first_region_other.GetAllocProtect());
BOOST_TEST_EQ(first_region.GetSize(), first_region_other.GetSize());
BOOST_TEST_EQ(first_region.GetState(), first_region_other.GetState());
BOOST_TEST_EQ(first_region.GetProtect(), first_region_other.GetProtect());
BOOST_TEST_EQ(first_region.GetType(), first_region_other.GetType());
hadesmem::Region const second_region(
process,
static_cast<char const* const>(first_region.GetBase()) +
first_region.GetSize());
BOOST_TEST_NE(first_region, second_region);
BOOST_TEST(first_region < second_region);
BOOST_TEST(first_region <= second_region);
BOOST_TEST(second_region > first_region);
BOOST_TEST(second_region >= first_region);
BOOST_TEST(!(first_region > second_region));
BOOST_TEST(!(first_region >= second_region));
BOOST_TEST(!(second_region < first_region));
BOOST_TEST(!(second_region <= first_region));
BOOST_TEST(first_region >= first_region);
BOOST_TEST(first_region <= first_region);
BOOST_TEST_EQ(first_region, first_region);
std::stringstream second_region_str_1;
second_region_str_1.imbue(std::locale::classic());
second_region_str_1 << second_region.GetBase();
std::stringstream second_region_str_2;
second_region_str_2.imbue(std::locale::classic());
second_region_str_2 << second_region;
BOOST_TEST(second_region_str_1.str() == second_region_str_2.str());
std::wstringstream second_region_str_3;
second_region_str_3.imbue(std::locale::classic());
second_region_str_3 << second_region.GetBase();
std::wstringstream second_region_str_4;
second_region_str_4.imbue(std::locale::classic());
second_region_str_4 << second_region;
BOOST_TEST(second_region_str_3.str() == second_region_str_4.str());
}
void TestTlsDir()
{
// Use TLS to ensure that at least one module has a TLS dir
thread_local static std::int32_t tls_dummy = 0;
hadesmem::Process const process(::GetCurrentProcessId());
hadesmem::PeFile pe_file_1(
process, ::GetModuleHandleW(nullptr), hadesmem::PeFileType::kImage, 0);
hadesmem::TlsDir tls_dir_1(process, pe_file_1);
hadesmem::TlsDir tls_dir_2(tls_dir_1);
BOOST_TEST_EQ(tls_dir_1, tls_dir_2);
tls_dir_1 = tls_dir_2;
BOOST_TEST_EQ(tls_dir_1, tls_dir_2);
hadesmem::TlsDir tls_dir_3(std::move(tls_dir_2));
BOOST_TEST_EQ(tls_dir_3, tls_dir_1);
tls_dir_2 = std::move(tls_dir_3);
BOOST_TEST_EQ(tls_dir_1, tls_dir_2);
hadesmem::ModuleList modules(process);
for (auto const& mod : modules)
{
hadesmem::PeFile const cur_pe_file(
process, mod.GetHandle(), hadesmem::PeFileType::kImage, 0);
std::unique_ptr<hadesmem::TlsDir> cur_tls_dir;
try
{
cur_tls_dir = std::make_unique<hadesmem::TlsDir>(process, cur_pe_file);
}
catch (std::exception const& /*e*/)
{
continue;
}
++tls_dummy;
auto const tls_dir_raw =
hadesmem::Read<IMAGE_TLS_DIRECTORY>(process, cur_tls_dir->GetBase());
cur_tls_dir->SetStartAddressOfRawData(
cur_tls_dir->GetStartAddressOfRawData());
cur_tls_dir->SetEndAddressOfRawData(cur_tls_dir->GetEndAddressOfRawData());
cur_tls_dir->SetAddressOfIndex(cur_tls_dir->GetAddressOfIndex());
cur_tls_dir->SetAddressOfCallBacks(cur_tls_dir->GetAddressOfCallBacks());
cur_tls_dir->SetSizeOfZeroFill(cur_tls_dir->GetSizeOfZeroFill());
cur_tls_dir->SetCharacteristics(cur_tls_dir->GetCharacteristics());
std::vector<ULONGLONG> callbacks;
cur_tls_dir->GetCallbacks(std::back_inserter(callbacks));
cur_tls_dir->UpdateWrite();
cur_tls_dir->UpdateRead();
auto const tls_dir_raw_new =
hadesmem::Read<IMAGE_TLS_DIRECTORY>(process, cur_tls_dir->GetBase());
BOOST_TEST_EQ(
std::memcmp(&tls_dir_raw, &tls_dir_raw_new, sizeof(tls_dir_raw)), 0);
std::stringstream test_str_1;
test_str_1.imbue(std::locale::classic());
test_str_1 << *cur_tls_dir;
std::stringstream test_str_2;
test_str_2.imbue(std::locale::classic());
test_str_2 << cur_tls_dir->GetBase();
BOOST_TEST_EQ(test_str_1.str(), test_str_2.str());
}
BOOST_TEST_NE(tls_dummy, 0);
}
void TestSectionList()
{
hadesmem::Process const process(::GetCurrentProcessId());
hadesmem::PeFile pe_file_1(
process, ::GetModuleHandleW(nullptr), hadesmem::PeFileType::Image, 0);
hadesmem::NtHeaders nt_headers_1(process, pe_file_1);
BOOST_TEST(nt_headers_1.GetNumberOfSections() >= 1);
hadesmem::Section section_1(process, pe_file_1, 0);
hadesmem::Section section_2(section_1);
BOOST_TEST_EQ(section_1, section_2);
section_1 = section_2;
BOOST_TEST_EQ(section_1, section_2);
hadesmem::Section section_3(std::move(section_2));
BOOST_TEST_EQ(section_3, section_1);
section_2 = std::move(section_3);
BOOST_TEST_EQ(section_1, section_2);
hadesmem::ModuleList modules(process);
for (auto const& mod : modules)
{
hadesmem::PeFile const pe_file(
process, mod.GetHandle(), hadesmem::PeFileType::Image, 0);
hadesmem::NtHeaders const nt_headers(process, pe_file);
WORD const num_sections = nt_headers.GetNumberOfSections();
// Assume every module has at least one section.
hadesmem::SectionList sections(process, pe_file);
WORD section_count = 0;
for (auto& section : sections)
{
section_count = static_cast<WORD>(section_count + 1);
auto const section_header_raw =
hadesmem::Read<IMAGE_SECTION_HEADER>(process, section.GetBase());
section.SetName(section.GetName());
section.SetVirtualAddress(section.GetVirtualAddress());
section.SetVirtualSize(section.GetVirtualSize());
section.SetSizeOfRawData(section.GetSizeOfRawData());
section.SetPointerToRawData(section.GetPointerToRawData());
section.SetPointerToRelocations(section.GetPointerToRelocations());
section.SetPointerToLinenumbers(section.GetPointerToLinenumbers());
section.SetNumberOfRelocations(section.GetNumberOfRelocations());
section.SetNumberOfLinenumbers(section.GetNumberOfLinenumbers());
section.SetCharacteristics(section.GetCharacteristics());
section.UpdateWrite();
section.UpdateRead();
auto const section_header_raw_new =
hadesmem::Read<IMAGE_SECTION_HEADER>(process, section.GetBase());
BOOST_TEST_EQ(std::memcmp(§ion_header_raw,
§ion_header_raw_new,
sizeof(section_header_raw)),
0);
std::stringstream test_str_1;
test_str_1.imbue(std::locale::classic());
test_str_1 << section;
std::stringstream test_str_2;
test_str_2.imbue(std::locale::classic());
test_str_2 << section.GetBase();
BOOST_TEST_EQ(test_str_1.str(), test_str_2.str());
if (mod.GetHandle() != GetModuleHandle(L"ntdll"))
{
hadesmem::PeFile const pe_file_ntdll(process,
::GetModuleHandleW(L"ntdll"),
hadesmem::PeFileType::Image,
0);
hadesmem::Section const section_ntdll(process, pe_file_ntdll, 0);
std::stringstream test_str_3;
test_str_3.imbue(std::locale::classic());
test_str_3 << section_ntdll.GetBase();
BOOST_TEST_NE(test_str_1.str(), test_str_3.str());
}
}
BOOST_TEST(section_count == num_sections);
// Assume every module has a '.text' section.
auto text_iter = std::find_if(std::begin(sections),
std::end(sections),
[](hadesmem::Section const& section)
{
return section.GetName() == ".data";
});
BOOST_TEST(text_iter != std::end(sections));
}
}
void TestNtHeaders()
{
hadesmem::Process const process(::GetCurrentProcessId());
hadesmem::PeFile pe_file_1(
process, ::GetModuleHandleW(nullptr), hadesmem::PeFileType::Image, 0);
hadesmem::NtHeaders nt_headers_1(process, pe_file_1);
hadesmem::NtHeaders nt_headers_2(nt_headers_1);
BOOST_TEST_EQ(nt_headers_1, nt_headers_2);
nt_headers_1 = nt_headers_2;
BOOST_TEST_EQ(nt_headers_1, nt_headers_2);
hadesmem::NtHeaders nt_headers_3(std::move(nt_headers_2));
BOOST_TEST_EQ(nt_headers_3, nt_headers_1);
nt_headers_2 = std::move(nt_headers_3);
BOOST_TEST_EQ(nt_headers_1, nt_headers_2);
hadesmem::ModuleList modules(process);
for (auto const& mod : modules)
{
hadesmem::PeFile const cur_pe_file(
process, mod.GetHandle(), hadesmem::PeFileType::Image, 0);
hadesmem::NtHeaders cur_nt_headers(process, cur_pe_file);
auto const nt_headers_raw =
hadesmem::Read<IMAGE_NT_HEADERS>(process, cur_nt_headers.GetBase());
BOOST_TEST_EQ(cur_nt_headers.IsValid(), true);
cur_nt_headers.EnsureValid();
cur_nt_headers.SetSignature(cur_nt_headers.GetSignature());
cur_nt_headers.SetMachine(cur_nt_headers.GetMachine());
cur_nt_headers.SetNumberOfSections(cur_nt_headers.GetNumberOfSections());
cur_nt_headers.SetTimeDateStamp(cur_nt_headers.GetTimeDateStamp());
cur_nt_headers.SetPointerToSymbolTable(
cur_nt_headers.GetPointerToSymbolTable());
cur_nt_headers.SetNumberOfSymbols(cur_nt_headers.GetNumberOfSymbols());
cur_nt_headers.SetSizeOfOptionalHeader(
cur_nt_headers.GetSizeOfOptionalHeader());
cur_nt_headers.SetCharacteristics(cur_nt_headers.GetCharacteristics());
cur_nt_headers.SetMagic(cur_nt_headers.GetMagic());
cur_nt_headers.SetMajorLinkerVersion(
cur_nt_headers.GetMajorLinkerVersion());
cur_nt_headers.SetMinorLinkerVersion(
cur_nt_headers.GetMinorLinkerVersion());
cur_nt_headers.SetSizeOfCode(cur_nt_headers.GetSizeOfCode());
cur_nt_headers.SetSizeOfInitializedData(
cur_nt_headers.GetSizeOfInitializedData());
cur_nt_headers.SetSizeOfUninitializedData(
cur_nt_headers.GetSizeOfUninitializedData());
cur_nt_headers.SetAddressOfEntryPoint(
cur_nt_headers.GetAddressOfEntryPoint());
cur_nt_headers.SetBaseOfCode(cur_nt_headers.GetBaseOfCode());
#if defined(HADESMEM_DETAIL_ARCH_X86)
cur_nt_headers.SetBaseOfData(cur_nt_headers.GetBaseOfData());
#endif
cur_nt_headers.SetImageBase(cur_nt_headers.GetImageBase());
cur_nt_headers.SetSectionAlignment(cur_nt_headers.GetSectionAlignment());
cur_nt_headers.SetFileAlignment(cur_nt_headers.GetFileAlignment());
cur_nt_headers.SetMajorOperatingSystemVersion(
cur_nt_headers.GetMajorOperatingSystemVersion());
cur_nt_headers.SetMinorOperatingSystemVersion(
cur_nt_headers.GetMinorOperatingSystemVersion());
cur_nt_headers.SetMajorImageVersion(cur_nt_headers.GetMajorImageVersion());
cur_nt_headers.SetMinorImageVersion(cur_nt_headers.GetMinorImageVersion());
cur_nt_headers.SetMajorSubsystemVersion(
cur_nt_headers.GetMajorSubsystemVersion());
cur_nt_headers.SetMinorSubsystemVersion(
cur_nt_headers.GetMinorSubsystemVersion());
cur_nt_headers.SetWin32VersionValue(cur_nt_headers.GetWin32VersionValue());
cur_nt_headers.SetSizeOfImage(cur_nt_headers.GetSizeOfImage());
cur_nt_headers.SetSizeOfHeaders(cur_nt_headers.GetSizeOfHeaders());
cur_nt_headers.SetCheckSum(cur_nt_headers.GetCheckSum());
cur_nt_headers.SetSubsystem(cur_nt_headers.GetSubsystem());
cur_nt_headers.SetDllCharacteristics(
cur_nt_headers.GetDllCharacteristics());
cur_nt_headers.SetSizeOfStackReserve(
cur_nt_headers.GetSizeOfStackReserve());
cur_nt_headers.SetSizeOfStackCommit(cur_nt_headers.GetSizeOfStackCommit());
cur_nt_headers.SetSizeOfHeapReserve(cur_nt_headers.GetSizeOfHeapReserve());
cur_nt_headers.SetSizeOfHeapCommit(cur_nt_headers.GetSizeOfHeapCommit());
cur_nt_headers.SetLoaderFlags(cur_nt_headers.GetLoaderFlags());
cur_nt_headers.SetNumberOfRvaAndSizes(
cur_nt_headers.GetNumberOfRvaAndSizes());
for (std::size_t i = 0; i < cur_nt_headers.GetNumberOfRvaAndSizes(); ++i)
{
auto data_dir = static_cast<hadesmem::PeDataDir>(i);
cur_nt_headers.SetDataDirectoryVirtualAddress(
data_dir, cur_nt_headers.GetDataDirectoryVirtualAddress(data_dir));
cur_nt_headers.SetDataDirectorySize(
data_dir, cur_nt_headers.GetDataDirectorySize(data_dir));
}
cur_nt_headers.UpdateWrite();
cur_nt_headers.UpdateRead();
auto const nt_headers_raw_new =
hadesmem::Read<IMAGE_NT_HEADERS>(process, cur_nt_headers.GetBase());
BOOST_TEST_EQ(
std::memcmp(&nt_headers_raw, &nt_headers_raw_new, sizeof(nt_headers_raw)),
0);
std::stringstream test_str_1;
test_str_1.imbue(std::locale::classic());
test_str_1 << cur_nt_headers;
std::stringstream test_str_2;
test_str_2.imbue(std::locale::classic());
test_str_2 << cur_nt_headers.GetBase();
BOOST_TEST_EQ(test_str_1.str(), test_str_2.str());
if (mod.GetHandle() != ::GetModuleHandle(L"ntdll"))
{
hadesmem::PeFile const pe_file_ntdll(
process, ::GetModuleHandleW(L"ntdll"), hadesmem::PeFileType::Image, 0);
hadesmem::NtHeaders const nt_headers_ntdll(process, pe_file_ntdll);
std::stringstream test_str_3;
test_str_3.imbue(std::locale::classic());
test_str_3 << nt_headers_ntdll.GetBase();
BOOST_TEST_NE(test_str_1.str(), test_str_3.str());
}
}
}
//SHA Test Vector for Hashing Byte-Oriented Messages
//http://csrc.nist.gov/cryptval/shs.htm
//http://csrc.nist.gov/cryptval/shs/shabytetestvectors.zip
//values from SHA1ShortMsg.txt
void test_short_messages()
{
struct test_case
{
char const* message;
unsigned int digest[5];
};
test_case cases[] =
{ { "",
{ 0xda39a3ee, 0x5e6b4b0d, 0x3255bfef, 0x95601890, 0xafd80709 } }
, { "a8",
{ 0x99f2aa95, 0xe36f95c2, 0xacb0eaf2, 0x3998f030, 0x638f3f15 } }
, { "3000",
{ 0xf944dcd6, 0x35f9801f, 0x7ac90a40, 0x7fbc4799, 0x64dec024 } }
, { "42749e",
{ 0xa444319e, 0x9b6cc1e8, 0x464c511e, 0xc0969c37, 0xd6bb2619 } }
, { "9fc3fe08",
{ 0x16a0ff84, 0xfcc156fd, 0x5d3ca3a7, 0x44f20a23, 0x2d172253 } }
, { "b5c1c6f1af",
{ 0xfec9deeb, 0xfcdedaf6, 0x6dda525e, 0x1be43597, 0xa73a1f93 } }
, { "e47571e5022e",
{ 0x8ce05118, 0x1f0ed5e9, 0xd0c498f6, 0xbc4caf44, 0x8d20deb5 } }
, { "3e1b28839fb758",
{ 0x67da5383, 0x7d89e03b, 0xf652ef09, 0xc369a341, 0x5937cfd3 } }
, { "a81350cbb224cb90",
{ 0x305e4ff9, 0x888ad855, 0xa78573cd, 0xdf4c5640, 0xcce7e946 } }
, { "c243d167923dec3ce1",
{ 0x5902b77b, 0x3265f023, 0xf9bbc396, 0xba1a93fa, 0x3509bde7 } }
, { "50ac18c59d6a37a29bf4",
{ 0xfcade5f5, 0xd156bf6f, 0x9af97bdf, 0xa9c19bcc, 0xfb4ff6ab } }
, { "98e2b611ad3b1cccf634f6",
{ 0x1d20fbe0, 0x0533c10e, 0x3cbd6b27, 0x088a5de0, 0xc632c4b5 } }
, { "73fe9afb68e1e8712e5d4eec",
{ 0x7e1b7e0f, 0x7a8f3455, 0xa9c03e95, 0x80fd63ae, 0x205a2d93 } }
, { "9e701ed7d412a9226a2a130e66",
{ 0x706f0677, 0x146307b2, 0x0bb0e8d6, 0x311e3299, 0x66884d13 } }
, { "6d3ee90413b0a7cbf69e5e6144ca",
{ 0xa7241a70, 0x3aaf0d53, 0xfe142f86, 0xbf2e8492, 0x51fa8dff } }
, { "fae24d56514efcb530fd4802f5e71f",
{ 0x400f5354, 0x6916d33a, 0xd01a5e6d, 0xf66822df, 0xbdc4e9e6 } }
, { "c5a22dd6eda3fe2bdc4ddb3ce6b35fd1",
{ 0xfac8ab93, 0xc1ae6c16, 0xf0311872, 0xb984f729, 0xdc928ccd } }
, { "d98cded2adabf08fda356445c781802d95",
{ 0xfba6d750, 0xc18da58f, 0x6e2aab10, 0x112b9a5e, 0xf3301b3b } }
, { "bcc6d7087a84f00103ccb32e5f5487a751a2",
{ 0x29d27c2d, 0x44c205c8, 0x107f0351, 0xb05753ac, 0x708226b6 } }
, { "36ecacb1055434190dbbc556c48bafcb0feb0d",
{ 0xb971bfc1, 0xebd6f359, 0xe8d74cb7, 0xecfe7f89, 0x8d0ba845 } }
, { "5ff9edb69e8f6bbd498eb4537580b7fba7ad31d0",
{ 0x96d08c43, 0x0094b9fc, 0xc164ad2f, 0xb6f72d0a, 0x24268f68 } }
, { "c95b441d8270822a46a798fae5defcf7b26abace36",
{ 0xa287ea75, 0x2a593d52, 0x09e28788, 0x1a09c49f, 0xa3f0beb1 } }
, { "83104c1d8a55b28f906f1b72cb53f68cbb097b44f860",
{ 0xa06c7137, 0x79cbd885, 0x19ed4a58, 0x5ac0cb8a, 0x5e9d612b } }
, { "755175528d55c39c56493d697b790f099a5ce741f7754b",
{ 0xbff7d52c, 0x13a36881, 0x32a1d407, 0xb1ab40f5, 0xb5ace298 } }
, { "088fc38128bbdb9fd7d65228b3184b3faac6c8715f07272f",
{ 0xc7566b91, 0xd7b6f56b, 0xdfcaa978, 0x1a7b6841, 0xaacb17e9 } }
, { "a4a586eb9245a6c87e3adf1009ac8a49f46c07e14185016895",
{ 0xffa30c0b, 0x5c550ea4, 0xb1e34f8a, 0x60ec9295, 0xa1e06ac1 } }
, { "8e7c555270c006092c2a3189e2a526b873e2e269f0fb28245256",
{ 0x29e66ed2, 0x3e914351, 0xe872aa76, 0x1df6e4f1, 0xa07f4b81 } }
, { "a5f3bfa6bb0ba3b59f6b9cbdef8a558ec565e8aa3121f405e7f2f0",
{ 0xb28cf5e5, 0xb806a014, 0x91d41f69, 0xbd924876, 0x5c5dc292 } }
, { "589054f0d2bd3c2c85b466bfd8ce18e6ec3e0b87d944cd093ba36469",
{ 0x60224fb7, 0x2c460696, 0x52cd78bc, 0xd08029ef, 0x64da62f3 } }
, { "a0abb12083b5bbc78128601bf1cbdbc0fdf4b862b24d899953d8da0ff3",
{ 0xb72c4a86, 0xf72608f2, 0x4c05f3b9, 0x088ef92f, 0xba431df7 } }
, { "82143f4cea6fadbf998e128a8811dc75301cf1db4f079501ea568da68eeb",
{ 0x73779ad5, 0xd6b71b9b, 0x8328ef72, 0x20ff12eb, 0x167076ac } }
, { "9f1231dd6df1ff7bc0b0d4f989d048672683ce35d956d2f57913046267e6f3",
{ 0xa09671d4, 0x452d7cf5, 0x0015c914, 0xa1e31973, 0xd20cc1a0 } }
, { "041c512b5eed791f80d3282f3a28df263bb1df95e1239a7650e5670fc2187919",
{ 0xe88cdcd2, 0x33d99184, 0xa6fd260b, 0x8fca1b7f, 0x7687aee0 } }
, { "17e81f6ae8c2e5579d69dafa6e070e7111461552d314b691e7a3e7a4feb3fae418",
{ 0x010def22, 0x850deb11, 0x68d525e8, 0xc84c2811, 0x6cb8a269 } }
, { "d15976b23a1d712ad28fad04d805f572026b54dd64961fda94d5355a0cc98620cf77",
{ 0xaeaa40ba, 0x1717ed54, 0x39b1e6ea, 0x901b294b, 0xa500f9ad } }
, { "09fce4d434f6bd32a44e04b848ff50ec9f642a8a85b37a264dc73f130f22838443328f",
{ 0xc6433791, 0x238795e3, 0x4f080a5f, 0x1f1723f0, 0x65463ca0 } }
, { "f17af27d776ec82a257d8d46d2b46b639462c56984cc1be9c1222eadb8b26594a25c709d",
{ 0xe21e22b8, 0x9c1bb944, 0xa32932e6, 0xb2a2f20d, 0x491982c3 } }
, { "b13ce635d6f8758143ffb114f2f601cb20b6276951416a2f94fbf4ad081779d79f4f195b22",
{ 0x575323a9, 0x661f5d28, 0x387964d2, 0xba6ab92c, 0x17d05a8a } }
, { "5498793f60916ff1c918dde572cdea76da8629ba4ead6d065de3dfb48de94d234cc1c5002910",
{ 0xfeb44494, 0xaf72f245, 0xbfe68e86, 0xc4d7986d, 0x57c11db7 } }
, { "498a1e0b39fa49582ae688cd715c86fbaf8a81b8b11b4d1594c49c902d197c8ba8a621fd6e3be5",
{ 0xcff2290b, 0x3648ba28, 0x31b98dde, 0x436a72f9, 0xebf51eee } }
, { "3a36ae71521f9af628b3e34dcb0d4513f84c78ee49f10416a98857150b8b15cb5c83afb4b570376e",
{ 0x9b4efe9d, 0x27b96590, 0x5b0c3dab, 0x67b8d7c9, 0xebacd56c } }
, { "dcc76b40ae0ea3ba253e92ac50fcde791662c5b6c948538cffc2d95e9de99cac34dfca38910db2678f",
{ 0xafedb0ff, 0x156205bc, 0xd831cbdb, 0xda43db8b, 0x0588c113 } }
, { "5b5ec6ec4fd3ad9c4906f65c747fd4233c11a1736b6b228b92e90cddabb0c7c2fcf9716d3fad261dff33",
{ 0x8deb1e85, 0x8f88293a, 0x5e5e4d52, 0x1a34b2a4, 0xefa70fc4 } }
, { "df48a37b29b1d6de4e94717d60cdb4293fcf170bba388bddf7a9035a15d433f20fd697c3e4c8b8c5f590ab",
{ 0x95cbdac0, 0xf74afa69, 0xcebd0e5c, 0x7defbc6f, 0xaf0cbeaf } }
, { "1f179b3b82250a65e1b0aee949e218e2f45c7a8dbfd6ba08de05c55acfc226b48c68d7f7057e5675cd96fcfc",
{ 0xf0307bcb, 0x92842e5a, 0xe0cd4f4f, 0x14f3df7f, 0x877fbef2 } }
, { "ee3d72da3a44d971578972a8e6780ce64941267e0f7d0179b214fa97855e1790e888e09fbe3a70412176cb3b54",
{ 0x7b13bb0d, 0xbf14964b, 0xd63b133a, 0xc85e2210, 0x0542ef55 } }
, { "d4d4c7843d312b30f610b3682254c8be96d5f6684503f8fbfbcd15774fc1b084d3741afb8d24aaa8ab9c104f7258",
{ 0xc314d2b6, 0xcf439be6, 0x78d2a74e, 0x890d96cf, 0xac1c02ed } }
, { "32c094944f5936a190a0877fb9178a7bf60ceae36fd530671c5b38c5dbd5e6a6c0d615c2ac8ad04b213cc589541cf6",
{ 0x4d0be361, 0xe410b47a, 0x9d67d8ce, 0x0bb6a8e0, 0x1c53c078 } }
, { "e5d3180c14bf27a5409fa12b104a8fd7e9639609bfde6ee82bbf9648be2546d29688a65e2e3f3da47a45ac14343c9c02",
{ 0xe5353431, 0xffae097f, 0x675cbf49, 0x8869f6fb, 0xb6e1c9f2 } }
, { "e7b6e4b69f724327e41e1188a37f4fe38b1dba19cbf5a7311d6e32f1038e97ab506ee05aebebc1eed09fc0e357109818b9",
{ 0xb8720a70, 0x68a085c0, 0x18ab1896, 0x1de2765a, 0xa6cd9ac4 } }
, { "bc880cb83b8ac68ef2fedc2da95e7677ce2aa18b0e2d8b322701f67af7d5e7a0d96e9e33326ccb7747cfff0852b961bfd475",
{ 0xb0732181, 0x568543ba, 0x85f2b6da, 0x602b4b06, 0x5d9931aa } }
, { "235ea9c2ba7af25400f2e98a47a291b0bccdaad63faa2475721fda5510cc7dad814bce8dabb611790a6abe56030b798b75c944",
{ 0x9c22674c, 0xf3222c3b, 0xa9216726, 0x94aafee4, 0xce67b96b } }
, { "07e3e29fed63104b8410f323b975fd9fba53f636af8c4e68a53fb202ca35dd9ee07cb169ec5186292e44c27e5696a967f5e67709",
{ 0xd128335f, 0x4cecca90, 0x66cdae08, 0x958ce656, 0xff0b4cfc } }
, { "65d2a1dd60a517eb27bfbf530cf6a5458f9d5f4730058bd9814379547f34241822bf67e6335a6d8b5ed06abf8841884c636a25733f",
{ 0x0b67c57a, 0xc578de88, 0xa2ae055c, 0xaeaec8bb, 0x9b0085a0 } }
, { "dcc86b3bd461615bab739d8daafac231c0f462e819ad29f9f14058f3ab5b75941d4241ea2f17ebb8a458831b37a9b16dead4a76a9b0e",
{ 0xc766f912, 0xa89d4ccd, 0xa88e0cce, 0x6a713ef5, 0xf178b596 } }
, { "4627d54f0568dc126b62a8c35fb46a9ac5024400f2995e51635636e1afc4373dbb848eb32df23914230560b82477e9c3572647a7f2bb92",
{ 0x9aa3925a, 0x9dcb177b, 0x15ccff9b, 0x78e70cf3, 0x44858779 } }
, { "ba531affd4381168ef24d8b275a84d9254c7f5cc55fded53aa8024b2c5c5c8aa7146fe1d1b83d62b70467e9a2e2cb67b3361830adbab28d7",
{ 0x4811fa30, 0x042fc076, 0xacf37c8e, 0x2274d025, 0x307e5943 } }
, { "8764dcbcf89dcf4282eb644e3d568bdccb4b13508bfa7bfe0ffc05efd1390be22109969262992d377691eb4f77f3d59ea8466a74abf57b2ef4",
{ 0x67430184, 0x50c97307, 0x61ee2b13, 0x0df9b91c, 0x1e118150 } }
, { "497d9df9ddb554f3d17870b1a31986c1be277bc44feff713544217a9f579623d18b5ffae306c25a45521d2759a72c0459b58957255ab592f3be4",
{ 0x71ad4a19, 0xd37d92a5, 0xe6ef3694, 0xddbeb5aa, 0x61ada645 } }
, { "72c3c2e065aefa8d9f7a65229e818176eef05da83f835107ba90ec2e95472e73e538f783b416c04654ba8909f26a12db6e5c4e376b7615e4a25819",
{ 0xa7d9dc68, 0xdacefb7d, 0x61161860, 0x48cb355c, 0xc548e11d } }
, { "7cc9894454d0055ab5069a33984e2f712bef7e3124960d33559f5f3b81906bb66fe64da13c153ca7f5cabc89667314c32c01036d12ecaf5f9a78de98",
{ 0x142e429f, 0x0522ba5a, 0xbf5131fa, 0x81df82d3, 0x55b96909 } }
, { "74e8404d5a453c5f4d306f2cfa338ca65501c840ddab3fb82117933483afd6913c56aaf8a0a0a6b2a342fc3d9dc7599f4a850dfa15d06c61966d74ea59",
{ 0xef72db70, 0xdcbcab99, 0x1e963797, 0x6c6faf00, 0xd22caae9 } }
, { "46fe5ed326c8fe376fcc92dc9e2714e2240d3253b105adfbb256ff7a19bc40975c604ad7c0071c4fd78a7cb64786e1bece548fa4833c04065fe593f6fb10",
{ 0xf220a745, 0x7f4588d6, 0x39dc2140, 0x7c942e98, 0x43f8e26b } }
, { "836dfa2524d621cf07c3d2908835de859e549d35030433c796b81272fd8bc0348e8ddbc7705a5ad1fdf2155b6bc48884ac0cd376925f069a37849c089c8645",
{ 0xddd2117b, 0x6e309c23, 0x3ede85f9, 0x62a0c2fc, 0x215e5c69 } }
, { "7e3a4c325cb9c52b88387f93d01ae86d42098f5efa7f9457388b5e74b6d28b2438d42d8b64703324d4aa25ab6aad153ae30cd2b2af4d5e5c00a8a2d0220c6116",
{ 0xa3054427, 0xcdb13f16, 0x4a610b34, 0x8702724c, 0x808a0dcc } }
};
char const xdigits[17] = "0123456789abcdef";
char const*const xdigits_end = xdigits+16;
for (std::size_t i=0; i!=sizeof(cases)/sizeof(cases[0]); ++i) {
test_case const& tc = cases[i];
boost::uuids::detail::sha1 sha;
std::size_t message_length = std::strlen(tc.message);
BOOST_TEST_EQ(message_length % 2, 0u);
for (std::size_t b=0; b<message_length; b+=2) {
char c = tc.message[b];
char const* f = std::find(xdigits, xdigits_end, c);
BOOST_TEST_NE(f, xdigits_end);
unsigned char byte = static_cast<unsigned char>(std::distance(&xdigits[0], f));
c = tc.message[b+1];
f = std::find(xdigits, xdigits_end, c);
BOOST_TEST_NE(f, xdigits_end);
byte <<= 4;
byte |= static_cast<unsigned char>(std::distance(&xdigits[0], f));
sha.process_byte(byte);
}
unsigned int digest[5];
sha.get_digest(digest);
BOOST_TEST_SHA1_DIGEST(digest, tc.digest);
}
}
void TestThisThread()
{
hadesmem::Thread thread(::GetCurrentThreadId());
BOOST_TEST_EQ(thread, thread);
hadesmem::Thread thread_new(::GetCurrentThreadId());
BOOST_TEST_EQ(thread, thread_new);
hadesmem::Thread const thread_moved(std::move(thread_new));
hadesmem::Thread thread_copy(thread);
BOOST_TEST_EQ(thread_copy, thread);
thread = thread_copy;
BOOST_TEST_EQ(thread, thread_copy);
thread_new = std::move(thread_copy);
BOOST_TEST_EQ(thread, thread_new);
BOOST_TEST(thread >= thread);
BOOST_TEST(thread <= thread);
BOOST_TEST_EQ(thread.GetId(), ::GetCurrentThreadId());
std::stringstream thread_str_1;
thread_str_1.imbue(std::locale::classic());
thread_str_1 << thread;
DWORD thread_id_1;
thread_str_1 >> thread_id_1;
BOOST_TEST_EQ(thread.GetId(), thread_id_1);
std::wstringstream thread_str_2;
thread_str_2.imbue(std::locale::classic());
thread_str_2 << thread;
DWORD thread_id_2;
thread_str_2 >> thread_id_2;
BOOST_TEST_EQ(thread.GetId(), thread_id_2);
hadesmem::detail::SmartHandle quit_event(
::CreateEvent(nullptr, TRUE, FALSE, nullptr));
DWORD other_id = 0;
hadesmem::detail::SmartHandle wait_thread(
::CreateThread(nullptr, 0, &WaitFunc, &quit_event, 0, &other_id));
// Disgusting hack to work around a potential race condition where we suspend
// the thread before it's actually waiting on our event, and instead it's
// still in its initialization routine and ends up holding a lock that the
// main thread will need (e.g. from malloc). See thread_140224_144550.dmp.
::Sleep(5 * 1000);
auto const cleanup_thread_func = [&]()
{
BOOST_TEST_NE(::SetEvent(quit_event.GetHandle()), 0);
BOOST_TEST_EQ(::WaitForSingleObject(wait_thread.GetHandle(), INFINITE),
WAIT_OBJECT_0);
};
{
ScopeExit<decltype(cleanup_thread_func)> cleanup_thread(
cleanup_thread_func);
hadesmem::Thread const other_thread(other_id);
BOOST_TEST_EQ(hadesmem::SuspendThread(other_thread), 0UL);
BOOST_TEST_EQ(hadesmem::SuspendThread(other_thread), 1UL);
BOOST_TEST_EQ(hadesmem::ResumeThread(other_thread), 2UL);
BOOST_TEST_EQ(hadesmem::ResumeThread(other_thread), 1UL);
{
hadesmem::SuspendedThread const suspend_thread(other_thread.GetId());
CONTEXT const full_context =
hadesmem::GetThreadContext(other_thread, CONTEXT_FULL);
CONTEXT const all_context =
hadesmem::GetThreadContext(other_thread, CONTEXT_ALL);
hadesmem::SetThreadContext(other_thread, full_context);
hadesmem::SetThreadContext(other_thread, all_context);
BOOST_TEST_THROWS(hadesmem::GetThreadContext(thread, CONTEXT_FULL),
hadesmem::Error);
}
{
hadesmem::SuspendedProcess const suspend_process(::GetCurrentProcessId());
}
}
}
void TestDosHeader()
{
hadesmem::Process const process(::GetCurrentProcessId());
hadesmem::PeFile pe_file_1(
process, ::GetModuleHandleW(nullptr), hadesmem::PeFileType::Image, 0);
hadesmem::DosHeader dos_header_1(process, pe_file_1);
hadesmem::DosHeader dos_header_2(dos_header_1);
BOOST_TEST_EQ(dos_header_1, dos_header_2);
dos_header_1 = dos_header_2;
BOOST_TEST_EQ(dos_header_1, dos_header_2);
hadesmem::DosHeader dos_header_3(std::move(dos_header_2));
BOOST_TEST_EQ(dos_header_3, dos_header_1);
dos_header_2 = std::move(dos_header_3);
BOOST_TEST_EQ(dos_header_1, dos_header_2);
hadesmem::ModuleList modules(process);
for (auto const& mod : modules)
{
hadesmem::PeFile const cur_pe_file(
process, mod.GetHandle(), hadesmem::PeFileType::Image, 0);
hadesmem::DosHeader cur_dos_header(process, cur_pe_file);
auto const dos_header_raw =
hadesmem::Read<IMAGE_DOS_HEADER>(process, cur_dos_header.GetBase());
BOOST_TEST_EQ(cur_dos_header.IsValid(), true);
cur_dos_header.EnsureValid();
cur_dos_header.SetMagic(cur_dos_header.GetMagic());
cur_dos_header.SetBytesOnLastPage(cur_dos_header.GetBytesOnLastPage());
cur_dos_header.SetPagesInFile(cur_dos_header.GetPagesInFile());
cur_dos_header.SetRelocations(cur_dos_header.GetRelocations());
cur_dos_header.SetSizeOfHeaderInParagraphs(
cur_dos_header.GetSizeOfHeaderInParagraphs());
cur_dos_header.SetMinExtraParagraphs(
cur_dos_header.GetMinExtraParagraphs());
cur_dos_header.SetMaxExtraParagraphs(
cur_dos_header.GetMaxExtraParagraphs());
cur_dos_header.SetInitialSS(cur_dos_header.GetInitialSS());
cur_dos_header.SetInitialSP(cur_dos_header.GetInitialSP());
cur_dos_header.SetChecksum(cur_dos_header.GetChecksum());
cur_dos_header.SetInitialIP(cur_dos_header.GetInitialIP());
cur_dos_header.SetInitialCS(cur_dos_header.GetInitialCS());
cur_dos_header.SetRelocTableFileAddr(
cur_dos_header.GetRelocTableFileAddr());
cur_dos_header.SetOverlayNum(cur_dos_header.GetOverlayNum());
cur_dos_header.SetReservedWords1(cur_dos_header.GetReservedWords1());
cur_dos_header.SetOEMID(cur_dos_header.GetOEMID());
cur_dos_header.SetOEMInfo(cur_dos_header.GetOEMInfo());
cur_dos_header.SetReservedWords2(cur_dos_header.GetReservedWords2());
cur_dos_header.SetNewHeaderOffset(cur_dos_header.GetNewHeaderOffset());
cur_dos_header.UpdateWrite();
cur_dos_header.UpdateRead();
auto const dos_header_raw_new =
hadesmem::Read<IMAGE_DOS_HEADER>(process, cur_pe_file.GetBase());
BOOST_TEST_EQ(
std::memcmp(&dos_header_raw, &dos_header_raw_new, sizeof(dos_header_raw)),
0);
std::stringstream test_str_1;
test_str_1.imbue(std::locale::classic());
test_str_1 << cur_dos_header;
std::stringstream test_str_2;
test_str_2.imbue(std::locale::classic());
test_str_2 << cur_dos_header.GetBase();
BOOST_TEST_EQ(test_str_1.str(), test_str_2.str());
if (mod.GetHandle() != ::GetModuleHandleW(L"ntdll"))
{
hadesmem::PeFile const pe_file_ntdll(
process, ::GetModuleHandleW(L"ntdll"), hadesmem::PeFileType::Image, 0);
hadesmem::DosHeader const dos_header_ntdll(process, pe_file_ntdll);
std::stringstream test_str_3;
test_str_3.imbue(std::locale::classic());
test_str_3 << dos_header_ntdll.GetBase();
BOOST_TEST_NE(test_str_1.str(), test_str_3.str());
}
}
}
