lysann/mach-detours
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity Actions

transaction tests

Browse source
This commit is contained in:
Lysann Tranvouez 2025-10-02 22:31:45 +02:00
parent 058069d1f3
commit 1fbdb703eb
3 changed files with 87 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

1
tests/CMakeLists.txt
View file

@ -3,6 +3,7 @@
add_executable(mach_detours_tests add_executable(mach_detours_tests
test_dylib_function.cpp test_dylib_function.cpp
test_local_function.cpp test_local_function.cpp
test_transaction.cpp
) )
# The target function must be in a shared library because otherwise it might be in the same code page as the test.cpp functions. # The target function must be in a shared library because otherwise it might be in the same code page as the test.cpp functions.

34
tests/test_dylib_function.cpp
View file

@ -5,10 +5,10 @@
#include <mach_detours.h> #include <mach_detours.h>
#include "lib_function.h" #include "lib_function.h"
int (*realLibFunction)() = libFunction; static int (*realLibFunction)() = libFunction;
static int libFunctionDetourCounter = 0; static int libFunctionDetourCounter = 0;
int libFunctionDetour() static int libFunctionDetour()
{ {
libFunctionDetourCounter++; libFunctionDetourCounter++;
return 94; return 94;
@ -75,34 +75,4 @@ TEST_CASE( "Overriding custom function in dylib", "[dylib]" )
CHECK( libFunctionCounter == 1 ); CHECK( libFunctionCounter == 1 );
CHECK( libFunctionDetourCounter == 0 ); CHECK( libFunctionDetourCounter == 0 );
} }
SECTION( "aborting transaction means no detour" )
{
REQUIRE( detour_transaction_begin() == err_none );
REQUIRE( detour_attach(reinterpret_cast<detour_func_t*>(&realLibFunction), reinterpret_cast<detour_func_t>(libFunctionDetour)) == err_none );
CHECK( detour_transaction_abort() == err_none );
CHECK( realLibFunction == libFunction );
CHECK( libFunctionCounter == 0 );
CHECK( libFunctionDetourCounter == 0 );
CHECK( libFunction() == 42 );
CHECK( libFunctionCounter == 1 );
CHECK( libFunctionDetourCounter == 0 );
}
SECTION( "an error in a transaction means no detour" )
{
REQUIRE( detour_transaction_begin() == err_none );
REQUIRE( detour_attach(reinterpret_cast<detour_func_t*>(&realLibFunction), reinterpret_cast<detour_func_t>(libFunctionDetour)) == err_none );
// cannot detach because trampoline is not yet in place
CHECK( detour_detach(reinterpret_cast<detour_func_t*>(&realLibFunction), reinterpret_cast<detour_func_t>(libFunctionDetour)) == KERN_FAILURE );
CHECK( detour_transaction_commit() == KERN_FAILURE );
CHECK( realLibFunction == libFunction );
CHECK( libFunctionCounter == 0 );
CHECK( libFunctionDetourCounter == 0 );
CHECK( libFunction() == 42 );
CHECK( libFunctionCounter == 1 );
CHECK( libFunctionDetourCounter == 0 );
}
} }

84
tests/test_transaction.cpp Normal file
View file

@ -0,0 +1,84 @@
// Copyright (c) Lysann Tranvouez. All rights reserved.
#include <catch2/catch_test_macros.hpp>
#include <mach_detours.h>
#include "lib_function.h"
static int (*realLibFunction)() = libFunction;
static int libFunctionDetourCounter = 0;
static int libFunctionDetour()
{
libFunctionDetourCounter++;
return 94;
}
TEST_CASE( "Must start a transaction to attach", "[transaction]" )
{
libFunctionCounter = 0;
libFunctionDetourCounter = 0;
CHECK( detour_attach(reinterpret_cast<detour_func_t*>(&realLibFunction), reinterpret_cast<detour_func_t>(libFunctionDetour)) == detour_err_wrong_thread );
CHECK( libFunction() == 42 );
CHECK( libFunctionCounter == 1 );
CHECK( libFunctionDetourCounter == 0 );
}
TEST_CASE( "Must start a transaction to detach", "[transaction]" )
{
libFunctionCounter = 0;
libFunctionDetourCounter = 0;
REQUIRE( detour_transaction_begin() == err_none );
REQUIRE( detour_attach_and_commit(reinterpret_cast<detour_func_t*>(&realLibFunction), reinterpret_cast<detour_func_t>(libFunctionDetour)) == err_none );
CHECK( detour_detach_and_commit(reinterpret_cast<detour_func_t*>(&realLibFunction), reinterpret_cast<detour_func_t>(libFunctionDetour)) == detour_err_wrong_thread );
CHECK( libFunction() == 94 );
CHECK( libFunctionCounter == 0 );
CHECK( libFunctionDetourCounter == 1 );
CHECK( detour_transaction_begin() == err_none );
CHECK( detour_detach_and_commit(reinterpret_cast<detour_func_t*>(&realLibFunction), reinterpret_cast<detour_func_t>(libFunctionDetour)) == err_none );
CHECK( libFunction() == 42 );
CHECK( libFunctionCounter == 1 );
CHECK( libFunctionDetourCounter == 1 );
}
TEST_CASE( "aborting transaction means no detour", "[transaction]" )
{
libFunctionCounter = 0;
libFunctionDetourCounter = 0;
REQUIRE( detour_transaction_begin() == err_none );
REQUIRE( detour_attach(reinterpret_cast<detour_func_t*>(&realLibFunction), reinterpret_cast<detour_func_t>(libFunctionDetour)) == err_none );
CHECK( detour_transaction_abort() == err_none );
CHECK( realLibFunction == libFunction );
CHECK( libFunctionCounter == 0 );
CHECK( libFunctionDetourCounter == 0 );
CHECK( libFunction() == 42 );
CHECK( libFunctionCounter == 1 );
CHECK( libFunctionDetourCounter == 0 );
}
TEST_CASE( "an error in a transaction means no detour", "[transaction]" )
{
libFunctionCounter = 0;
libFunctionDetourCounter = 0;
REQUIRE( detour_transaction_begin() == err_none );
REQUIRE( detour_attach(reinterpret_cast<detour_func_t*>(&realLibFunction), reinterpret_cast<detour_func_t>(libFunctionDetour)) == err_none );
// cannot detach because trampoline is not yet in place
CHECK( detour_detach(reinterpret_cast<detour_func_t*>(&realLibFunction), reinterpret_cast<detour_func_t>(libFunctionDetour)) == KERN_FAILURE );
CHECK( detour_transaction_commit() == KERN_FAILURE );
CHECK( realLibFunction == libFunction );
CHECK( libFunctionCounter == 0 );
CHECK( libFunctionDetourCounter == 0 );
CHECK( libFunction() == 42 );
CHECK( libFunctionCounter == 1 );
CHECK( libFunctionDetourCounter == 0 );
}