added windows support

Author: StackOverflowExcept1on

.github/workflows/ci.yml

@@ -1,16 +1,27 @@
 name: CI
 on:
   push:
+    branches: [ master ]
   pull_request:
+    branches: [ master ]
   workflow_dispatch:
 
 jobs:
-  build:
-    if: "!contains(github.event.head_commit.message, '[ci skip]')"
+  linux-build:
     runs-on: ubuntu-latest
 
     steps:
-      - uses: actions/checkout@v3
+      - uses: actions/checkout@v4
 
       - name: Build project
         run: ./build.sh
+
+  windows-build:
+    runs-on: windows-latest
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Build project
+        run: build.bat
+        shell: cmd

.gitignore

@@ -1,3 +1,6 @@
 .idea
 build
-cmake-build-*
+cmake-build-debug
+cmake-build-minsizerel
+cmake-build-release
+cmake-build-relwithdebinfo

CMakeLists.txt

@@ -1,7 +1,7 @@
-cmake_minimum_required(VERSION 3.11)
+cmake_minimum_required(VERSION 3.14)
 project(function-interceptor)
 
-set(CMAKE_CXX_STANDARD 20)
+set(CMAKE_CXX_STANDARD 23)
 set(CMAKE_CXX_STANDARD_REQUIRED ON)
 set(CMAKE_POSITION_INDEPENDENT_CODE ON)
 

README.md

@@ -6,12 +6,12 @@ This project demonstrates how to hook functions through the [funchook](https://g
 
 ### Requirements
 
-- g++ compiler
+- g++ / msvc compiler
 - cmake
 - python3
 - frida: https://frida.re
 
-### Preparing
+### Preparing (Linux)
 
 ```bash
 # install dependencies
@@ -24,7 +24,17 @@ sudo sysctl kernel.yama.ptrace_scope=0
 ./build.sh
 ```
 
-### Running
+### Preparing (Windows)
+
+```bat
+REM install dependencies
+pip install frida
+
+REM build c++ project
+build.bat
+```
+
+### Running (Linux)
 
 ```bash
 # run program (it is assumed that in different terminals)
@@ -34,6 +44,21 @@ LD_LIBRARY_PATH=./build/bin ./build/bin/program
 ./scripts/loader.py $(pidof program) ./build/bin/libhook.so
 ```
 
+### Running (Windows)
+
+```bat
+REM run program (it is assumed that in different terminals)
+.\build\bin\program.exe
+
+REM find the PID
+frida-ps
+
+REM attach .\build\bin\hook.dll to PID
+python .\scripts\loader.py PID .\build\bin\hook.dll
+```
+
+### Output
+
 ```
 exampleFunction(a=1, b=2, c=3)
 ret = 48

build.bat

@@ -0,0 +1,16 @@
+@echo off
+
+set CMAKE_BUILD_TYPE="Release"
+
+if not exist build mkdir build
+cd build
+
+set vswhere="%ProgramFiles(x86)%\Microsoft Visual Studio\Installer\vswhere.exe"
+set cmakeLookup=call %vswhere% -latest -requires Microsoft.VisualStudio.Component.VC.CMake.Project -find Common7\IDE\CommonExtensions\Microsoft\CMake\CMake\bin\cmake.exe
+
+for /f "tokens=*" %%i in ('%cmakeLookup%') do set cmake="%%i"
+
+%cmake% -DCMAKE_VERBOSE_MAKEFILE:BOOL=ON ..
+%cmake% --build . --config %CMAKE_BUILD_TYPE% --target INSTALL
+
+cd ..

clean.bat

@@ -0,0 +1,17 @@
+@echo off
+
+setlocal ENABLEDELAYEDEXPANSION
+
+for /f "tokens=*" %%i in (.gitignore) do (
+  set filename=%%i
+  set filename=!filename:/=\!
+  if exist !filename!\* (
+    for /d %%d in (!filename!) do (
+      del /s /f /q %%d\*.*
+      for /f %%f in ('dir /ad /b %%d\') do rd /s /q %%d\%%f
+      rd %%d
+    )
+  ) else if exist !filename! (
+    del /f /q !filename!
+  )
+)

demo/module/include/module.h

@@ -1,9 +1,9 @@
 #ifndef MODULE_H
 #define MODULE_H
 
-#if defined(WIN32)
+#ifdef _WIN32
 #define DLL_EXPORT __declspec(dllexport)
-#elif defined(__GNUC__)
+#else
 #define DLL_EXPORT __attribute__((visibility("default")))
 #endif
 

demo/program/main.cpp

@@ -1,12 +1,15 @@
 #include <module.h>
 #include <cstdio>
-#include <unistd.h>
+#include <thread>
+#include <chrono>
+
+using namespace std::chrono_literals;
 
 int main() {
     for (int i = 0; i < 30; ++i) {
         int ret = exampleFunction(1, 2, 3);
         printf("ret = %d\n", ret);
-        sleep(1);
+        std::this_thread::sleep_for(1s);
     }
     return 0;
 }

hook/CMakeLists.txt

@@ -2,7 +2,11 @@ project(hook)
 
 include(FetchContent)
 
-FetchContent_Declare(funchook GIT_REPOSITORY https://github.com/kubo/funchook)
+set(FUNCHOOK_BUILD_SHARED OFF CACHE BOOL "")
+set(FUNCHOOK_BUILD_STATIC ON CACHE BOOL "")
+set(FUNCHOOK_BUILD_TESTS OFF CACHE BOOL "")
+set(FUNCHOOK_INSTALL OFF CACHE BOOL "")
+FetchContent_Declare(funchook GIT_REPOSITORY https://github.com/StackOverflowExcept1on/funchook)
 FetchContent_MakeAvailable(funchook)
 
 add_library(${PROJECT_NAME} SHARED library.cpp)

hook/library.cpp

@@ -1,11 +1,15 @@
 #include <funchook.h>
 
 #include <cstdio>
+#ifdef _WIN32
+#include <windows.h>
+#else
 #include <dlfcn.h>
+#endif
 
-#if defined(WIN32)
+#ifdef _WIN32
 #define DLL_EXPORT __declspec(dllexport)
-#elif defined(__GNUC__)
+#else
 #define DLL_EXPORT __attribute__((visibility("default")))
 #endif
 
@@ -24,9 +28,17 @@ extern "C" DLL_EXPORT void onStartup() {
     funchook_t *funchook = funchook_create();
     int ret;
 
+#ifdef _WIN32
+    void *handle = GetModuleHandleA("module.dll");
+#else
     void *handle = dlopen("libmodule.so", RTLD_NOW);
+#endif
     printf("handle = %p\n", handle);
+#ifdef _WIN32
+    exampleFunction_original = (exampleFunction_fptr_t) GetProcAddress((HMODULE) handle, "exampleFunction");
+#else
     exampleFunction_original = (exampleFunction_fptr_t) dlsym(handle, "exampleFunction");
+#endif
     printf("exampleFunction_original = %p\n", (void *) exampleFunction_original);
 
     ret = funchook_prepare(funchook, (void **) &exampleFunction_original, (void *) exampleFunction_hook);