early test

geode-sdk · Jun 2, 2024 · 9c0b4d0 · 9c0b4d0
1 parent 733d3ff
commit 9c0b4d0
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git a/src/generator/X64Generator.cpp b/src/generator/X64Generator.cpp
@@ -156,13 +156,17 @@ std::vector<uint8_t> X64HandlerGenerator::handlerBytes(uint64_t address) {
 	a.mov(FIRST_PARAM, "content");
 
 	// call the pre handler, incrementing
-	a.callip("handlerPre");
+	// a.callip("handlerPre");
 
 	a.mov(m[RBP - 0x10], RAX);
 
 	// recover registers
 	restoreRegisters(a, preservedSize);
 
+	// early test
+	a.pop(RBP);
+	a.ret();
+
 	// call the func
 	m_metadata.m_convention->generateIntoDefault(a, m_metadata.m_abstract);
 

diff --git a/test/Hook.cpp b/test/Hook.cpp
@@ -90,7 +90,7 @@ TEST(HookTest, NoHandler) {
 TEST(HookTest, NoHooks) {
 	HandlerHandle handlerHandle;
 	makeHandler<1>(handlerHandle);
-	EXPECT_EQ(callFunction<1>(), 2);
+	EXPECT_EQ(callFunction<1>(), 1);
 }
 
 TEST(HookTest, MakeWrapper) {
@@ -104,20 +104,20 @@ TEST(HookTest, SingleHook) {
 	HandlerHandle handlerHandle;
 	makeHandler<3>(handlerHandle);
 
-	// makeHook<3>(handlerHandle);
+	makeHook<3>(handlerHandle);
 
-	// EXPECT_EQ(callFunction<3>(), 3);
+	EXPECT_EQ(callFunction<3>(), 3);
 }
 
 // priorityHook -> hook -> function
 TEST(HookTest, PriorityHook) {
 	HandlerHandle handlerHandle;
-	// makeHandler<4>(handlerHandle);
+	makeHandler<4>(handlerHandle);
 
-	// makeHook<4>(handlerHandle);
-	// makePriorityHook<4>(handlerHandle);
+	makeHook<4>(handlerHandle);
+	makePriorityHook<4>(handlerHandle);
 
-	// EXPECT_EQ(callFunction<4>(), 6);
+	EXPECT_EQ(callFunction<4>(), 6);
 }
 
 // priorityHook -> function