Add hook for GetCommandLineA

kevoreilly · Dec 2, 2022 · 4c31b16 · 4c31b16
1 parent 4a336c9
commit 4c31b16
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/hook_misc.c b/hook_misc.c
@@ -1758,6 +1758,14 @@ HOOKDEF(int, WINAPI, WideCharToMultiByte,
 	return Old_WideCharToMultiByte(CodePage, dwFlags, lpWideCharStr, cchWideChar, lpMultiByteStr, cbMultiByte, lpDefaultChar, lpUsedDefaultChar);
 }
 
+HOOKDEF(LPSTR, WINAPI, GetCommandLineA,
+	void
+) {
+	LPSTR ret = Old_GetCommandLineA();
+	LOQ_nonnull("misc", "s", "CommandLine", ret);
+	return ret;
+}
+
 HOOKDEF(LPWSTR, WINAPI, GetCommandLineW,
 	void
 ) {

diff --git a/hooks.c b/hooks.c
@@ -318,6 +318,7 @@ hook_t full_hooks[] = {
 #endif
 	HOOK(msvcrt, memcpy),
 	//HOOK(ntdll, RtlMoveMemory),
+	HOOK(kernel32, GetCommandLineA),
 	HOOK(kernel32, GetCommandLineW),
 	HOOK(kernel32, OutputDebugStringA),
 	HOOK(kernel32, OutputDebugStringW),

diff --git a/hooks.h b/hooks.h
@@ -3393,6 +3393,10 @@ HOOKDEF(int, WINAPI, WideCharToMultiByte,
 	__out_opt	LPBOOL	lpUsedDefaultChar
 );
 
+HOOKDEF(LPSTR, WINAPI, GetCommandLineA,
+	void
+);
+
 HOOKDEF(LPWSTR, WINAPI, GetCommandLineW,
 	void
 );
-Original file line number
+Diff line change
@@ Expand Up / @@ -3393,6 +3393,10 @@ HOOKDEF(int, WINAPI, WideCharToMultiByte, @@
     	__out_opt	LPBOOL	lpUsedDefaultChar
     );
+    HOOKDEF(LPSTR, WINAPI, GetCommandLineA,
+    	void
+    );
     HOOKDEF(LPWSTR, WINAPI, GetCommandLineW,
     	void
     );
@@ Expand Down @@