Add hook for GetCommandLineA

kevoreilly · kevoreilly · commit 4c31b16a17e3 · 2022-12-02T11:42:32.000Z
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
 );
