Link to NtQueryInformationProcess directly

touhouworldcup · Feb 21, 2024 · 097ed89 · 097ed89
1 parent ffa341f
commit 097ed89
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 10 deletions.
diff --git a/thprac/src/thprac/thprac_main.cpp b/thprac/src/thprac/thprac_main.cpp
@@ -102,14 +102,8 @@ bool PromptUser(thprac_prompt_t info, THGameSig* gameSig = nullptr)
 }
 
 uintptr_t GetGameModuleBase(HANDLE hProc) {
-    static decltype(NtQueryInformationProcess)* _NtQueryInformationProcess;
-
-    if (!_NtQueryInformationProcess) {
-        _NtQueryInformationProcess = (decltype(NtQueryInformationProcess)*)GetProcAddress(GetModuleHandleW(L"ntdll.dll"), "NtQueryInformationProcess");
-    }
-
     PROCESS_BASIC_INFORMATION pbi;
-    _NtQueryInformationProcess(hProc, ProcessBasicInformation, &pbi, sizeof(pbi), nullptr);
+    NtQueryInformationProcess(hProc, ProcessBasicInformation, &pbi, sizeof(pbi), nullptr);
 
     LPVOID based = (LPVOID)((uintptr_t)pbi.PebBaseAddress + offsetof(PEB, ImageBaseAddress));
 

diff --git a/thprac/src/thprac/utils/wininternal.h b/thprac/src/thprac/utils/wininternal.h
@@ -224,11 +224,11 @@ typedef enum _PROCESSINFOCLASS {
     ProcessBreakOnTermination = 29
 } PROCESSINFOCLASS;
 
-__kernel_entry NTSTATUS
+extern "C" __kernel_entry NTSTATUS
     NTAPI
     NtQueryInformationProcess(
         IN HANDLE ProcessHandle,
         IN PROCESSINFOCLASS ProcessInformationClass,
         OUT PVOID ProcessInformation,
         IN ULONG ProcessInformationLength,
-        OUT PULONG ReturnLength OPTIONAL);
+        OUT PULONG ReturnLength OPTIONAL);
diff --git a/thprac/thprac.vcxproj b/thprac/thprac.vcxproj
@@ -200,7 +200,7 @@
     <Link>
       <SubSystem>Windows</SubSystem>
       <FixedBaseAddress>false</FixedBaseAddress>
-      <AdditionalDependencies>kernel32.lib;user32.lib;freetype.lib;gdi32.lib</AdditionalDependencies>
+      <AdditionalDependencies>kernel32.lib;ntdll.lib;user32.lib;freetype.lib;gdi32.lib</AdditionalDependencies>
       <GenerateMapFile>true</GenerateMapFile>
       <RandomizedBaseAddress>true</RandomizedBaseAddress>
       <GenerateDebugInformation>true</GenerateDebugInformation>