admin/omath
1
0
Fork 0
mirror of https://github.com/orange-cpp/omath.git synced 2026-04-18 18:23:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactored test

Browse Source
This commit is contained in:
orange
2026-03-15 17:20:28 +03:00
parent 4f1c42d6f6
commit 130277c1ae
1 changed files with 20 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
tests/general/unit_test_elf_scanner.cpp
View File

@@ -3,9 +3,11 @@
// //
#include <algorithm> #include <algorithm>
#include <cstring> #include <cstring>
#include <filesystem>
#include <fstream> #include <fstream>
#include <gtest/gtest.h> #include <gtest/gtest.h>
#include <omath/utility/elf_pattern_scan.hpp> #include <omath/utility/elf_pattern_scan.hpp>
#include <span>
#include <vector> #include <vector>
using namespace omath; using namespace omath;
@@ -188,26 +190,25 @@ TEST(unit_test_elf_pattern_scan_memory, missing_section_returns_nullopt)
TEST(unit_test_elf_pattern_scan_memory, matches_file_scan) TEST(unit_test_elf_pattern_scan_memory, matches_file_scan)
{ {
// Read test binary itself and compare memory scan vs file scan // Write our synthetic ELF to a temp file and verify memory scan == file scan
std::ifstream file("/proc/self/exe", std::ios::binary); const std::vector<std::uint8_t> code = {0x48, 0x89, 0xE5, 0xDE, 0xAD, 0xBE, 0xEF, 0x00};
if (!file.is_open()) const auto buf = make_elf64_with_text_section(code);
GTEST_SKIP() << "Cannot open /proc/self/exe";
const std::string raw{std::istreambuf_iterator<char>(file), {}}; const auto tmp_path = std::filesystem::temp_directory_path() / "omath_elf_test.elf";
std::vector<std::byte> data(raw.size());
std::transform(raw.begin(), raw.end(), data.begin(),
[](char c) { return std::byte{static_cast<unsigned char>(c)}; });
constexpr std::string_view pattern = "7F 45 4C 46"; // ELF magic at start of .text unlikely; use any known bytes
const auto file_result = ElfPatternScanner::scan_for_pattern_in_file("/proc/self/exe", pattern, ".text");
const auto mem_result =
ElfPatternScanner::scan_for_pattern_in_memory_file(std::span<const std::byte>{data}, pattern, ".text");
EXPECT_EQ(file_result.has_value(), mem_result.has_value());
if (file_result && mem_result)
{ {
EXPECT_EQ(file_result->virtual_base_addr, mem_result->virtual_base_addr); std::ofstream out(tmp_path, std::ios::binary);
EXPECT_EQ(file_result->raw_base_addr, mem_result->raw_base_addr); out.write(reinterpret_cast<const char*>(buf.data()), static_cast<std::streamsize>(buf.size()));
EXPECT_EQ(file_result->target_offset, mem_result->target_offset);
} }
const auto file_result = ElfPatternScanner::scan_for_pattern_in_file(tmp_path, "48 89 E5 DE AD", ".text");
const auto mem_result =
ElfPatternScanner::scan_for_pattern_in_memory_file(std::span<const std::byte>{buf}, "48 89 E5 DE AD", ".text");
std::filesystem::remove(tmp_path);
ASSERT_TRUE(file_result.has_value());
ASSERT_TRUE(mem_result.has_value());
EXPECT_EQ(file_result->virtual_base_addr, mem_result->virtual_base_addr);
EXPECT_EQ(file_result->raw_base_addr, mem_result->raw_base_addr);
EXPECT_EQ(file_result->target_offset, mem_result->target_offset);
} }