Add tests directory and TestDLL project; improve CMake configuration

Author: woldann

CMakeLists.txt

@@ -18,6 +18,9 @@ add_subdirectory(Neptune)
 add_subdirectory(NThread)
 add_subdirectory(NThreadOSUtils)
 
+# Testleri derlemeye dahil et
+add_subdirectory(tests)
+
 set(GHOSTINJECTOR_INCLUDE_DIR ${CMAKE_CURRENT_SOURCE_DIR}/include)
 set(GHOSTINJECTOR_SOURCE_DIR ${CMAKE_CURRENT_SOURCE_DIR}/src)
 

src/main.c

@@ -32,22 +32,24 @@
 #include "ntmem.h"
 #include "ntosutilswin.h"
 
-void print_usage() {
-    printf("GhostInjector - DLL Injection tool for Windows processes\n\n");
-    printf("Examples:\n");
-    printf("  ghostinjector.exe 1234 mydll.dll\n");
-    printf("  ghostinjector.exe 5678 first.dll second.dll third.dll\n\n");
-    printf("Usage:\n");
-    printf("  ghostinjector.exe <process_id> <dll_path> [dll_path2 ...]\n");
-    printf("  ghostinjector.exe -h | --help\n");
+void print_usage()
+{
+	printf("GhostInjector - DLL Injection tool for Windows processes\n\n");
+	printf("Examples:\n");
+	printf("  ghostinjector.exe 1234 mydll.dll\n");
+	printf("  ghostinjector.exe 5678 first.dll second.dll third.dll\n\n");
+	printf("Usage:\n");
+	printf("  ghostinjector.exe <process_id> <dll_path> [dll_path2 ...]\n");
+	printf("  ghostinjector.exe -h | --help\n");
 }
 
 int main(int argc, char *argv[])
 {
 	if (HAS_ERR(neptune_init()))
 		return EXIT_FAILURE;
 
-	if (argc < 3 || strcmp(argv[1], "-h") == 0 || strcmp(argv[1], "--help") == 0) {
+	if (argc < 3 || strcmp(argv[1], "-h") == 0 ||
+	    strcmp(argv[1], "--help") == 0) {
 		print_usage();
 		neptune_destroy();
 		return EXIT_SUCCESS;
@@ -109,7 +111,8 @@ int main(int argc, char *argv[])
 		const char *dll_path = argv[i];
 
 #ifdef LOG_LEVEL_1
-		LOG_INFO("Injecting DLL [%d/%d]: %s", i - 1, argc - 2, dll_path);
+		LOG_INFO("Injecting DLL [%d/%d]: %s", i - 1, argc - 2,
+			 dll_path);
 #endif
 
 		size_t dll_path_len = strlen(dll_path);
@@ -120,7 +123,7 @@ int main(int argc, char *argv[])
 #ifdef LOG_LEVEL_1
 			LOG_ERROR("ntm_create failed for %s", dll_path);
 #endif
-			continue; 
+			continue;
 		}
 
 		void *local = NTM_LOCAL(ntmem);
@@ -139,7 +142,8 @@ int main(int argc, char *argv[])
 		LOG_INFO("DLL Path Address(%p)", dll_path_addr);
 #endif
 
-		void *load_library_ret = ntu_ucall((void *)load_library_func, dll_path_addr);
+		void *load_library_ret =
+			ntu_ucall((void *)load_library_func, dll_path_addr);
 
 #ifdef LOG_LEVEL_1
 		LOG_INFO("LoadLibrary returned: %p", load_library_ret);

tests/CMakeLists.txt

@@ -0,0 +1,2 @@
+# Bu dizindeki diğer test projelerini ekleyin
+add_subdirectory(TestDLL)

tests/TestDLL/CMakeLists.txt

@@ -0,0 +1,20 @@
+cmake_minimum_required(VERSION 3.10)
+project(TestDLL C)
+
+set(CMAKE_C_STANDARD 99)
+
+string(TOLOWER ${PROJECT_NAME} library_name)
+
+add_library(${library_name} SHARED src/dllmain.c)
+
+target_include_directories(${library_name} PRIVATE ../../Neptune/include)
+
+target_link_libraries(${library_name} PRIVATE Neptune)
+
+target_compile_definitions(${library_name} PRIVATE
+  LOG_LEVEL_2
+  "LOG_FILE_PATH=L\"C:\\\\ghostinjectortest.log\""
+  LOG_ON_STDOUT=0
+)
+
+set_target_properties(${library_name} PROPERTIES PREFIX "")

tests/TestDLL/src/dllmain.c

@@ -0,0 +1,53 @@
+/**
+ * MIT License
+ *
+ * Copyright (c) 2025 Serkan Aksoy
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ * 
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software. 
+ * 
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+
+#include <windows.h>
+#include "neptune.h"
+#include "log.h"
+#include "nerror.h"
+
+BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
+{
+	(void)hinstDLL;
+	(void)lpvReserved;
+
+	switch (fdwReason) {
+	case DLL_PROCESS_ATTACH:
+		if (HAS_ERR(neptune_init())) {
+			return FALSE;
+		}
+
+		DWORD pid = GetCurrentProcessId();
+		DWORD tid = GetCurrentThreadId();
+
+		LOG_INFO("Injected into PID: %lu, Thread ID: %lu", pid, tid);
+		break;
+
+	case DLL_PROCESS_DETACH:
+		neptune_destroy();
+		break;
+	}
+
+	return TRUE;
+}

toolchain-mingw.cmake

@@ -4,5 +4,8 @@ set(CMAKE_C_COMPILER x86_64-w64-mingw32-gcc)
 set(CMAKE_CXX_COMPILER x86_64-w64-mingw32-g++)
 set(CMAKE_RC_COMPILER x86_64-w64-mingw32-windres)
 
+set(CMAKE_EXE_LINKER_FLAGS "-s")
+set(CMAKE_SHARED_LINKER_FLAGS "-s")
+
 # Optimize for size in release builds
 string(APPEND CMAKE_C_FLAGS_RELEASE " -Os")