mirror of
https://github.com/orange-cpp/omath.git
synced 2026-04-19 00:23:26 +00:00
67 lines
2.7 KiB
Lua
67 lines
2.7 KiB
Lua
-- PatternScanner tests: generic scan over a Lua string buffer
|
|
|
|
function PatternScanner_FindsExactPattern()
|
|
local buf = "\x90\x01\x02\x03\x04"
|
|
local offset = omath.PatternScanner.scan(buf, "90 01 02")
|
|
assert(offset ~= nil, "expected pattern to be found")
|
|
assert(offset == 0, "expected offset 0, got " .. tostring(offset))
|
|
end
|
|
|
|
function PatternScanner_FindsPatternAtNonZeroOffset()
|
|
local buf = "\x00\x00\xAB\xCD\xEF"
|
|
local offset = omath.PatternScanner.scan(buf, "AB CD EF")
|
|
assert(offset ~= nil, "expected pattern to be found")
|
|
assert(offset == 2, "expected offset 2, got " .. tostring(offset))
|
|
end
|
|
|
|
function PatternScanner_WildcardMatches()
|
|
local buf = "\xDE\xAD\xBE\xEF"
|
|
local offset = omath.PatternScanner.scan(buf, "DE ?? BE")
|
|
assert(offset ~= nil, "expected wildcard match")
|
|
assert(offset == 0)
|
|
end
|
|
|
|
function PatternScanner_ReturnsNilWhenNotFound()
|
|
local buf = "\x01\x02\x03"
|
|
local offset = omath.PatternScanner.scan(buf, "AA BB CC")
|
|
assert(offset == nil, "expected nil for not-found pattern")
|
|
end
|
|
|
|
function PatternScanner_ReturnsNilForEmptyBuffer()
|
|
local offset = omath.PatternScanner.scan("", "90 01")
|
|
assert(offset == nil)
|
|
end
|
|
|
|
-- PePatternScanner tests: scan_in_module uses FAKE_MODULE_BASE injected from C++
|
|
-- The fake module contains {0x90, 0x01, 0x02, 0x03, 0x04} placed at raw offset 0x200
|
|
|
|
function PeScanner_FindsExactPattern()
|
|
local addr = omath.PePatternScanner.scan_in_module(FAKE_MODULE_BASE, "90 01 02")
|
|
assert(addr ~= nil, "expected pattern to be found in module")
|
|
local offset = addr - FAKE_MODULE_BASE
|
|
assert(offset == 0x200, string.format("expected offset 0x200, got 0x%X", offset))
|
|
end
|
|
|
|
function PeScanner_WildcardMatches()
|
|
local addr = omath.PePatternScanner.scan_in_module(FAKE_MODULE_BASE, "90 ?? 02")
|
|
assert(addr ~= nil, "expected wildcard match in module")
|
|
local offset = addr - FAKE_MODULE_BASE
|
|
assert(offset == 0x200, string.format("expected offset 0x200, got 0x%X", offset))
|
|
end
|
|
|
|
function PeScanner_ReturnsNilWhenNotFound()
|
|
local addr = omath.PePatternScanner.scan_in_module(FAKE_MODULE_BASE, "AA BB CC DD")
|
|
assert(addr == nil, "expected nil for not-found pattern")
|
|
end
|
|
|
|
function PeScanner_CustomSectionFallsBackToNil()
|
|
-- Request a section that doesn't exist in our fake module
|
|
local addr = omath.PePatternScanner.scan_in_module(FAKE_MODULE_BASE, "90 01 02", ".rdata")
|
|
assert(addr == nil, "expected nil for wrong section name")
|
|
end
|
|
|
|
-- SectionScanResult: verify the type is registered and tostring works on a C++-returned value
|
|
function SectionScanResult_TypeIsRegistered()
|
|
assert(omath.SectionScanResult ~= nil, "SectionScanResult type should be registered")
|
|
end
|