find all PIDS, error on admin

LeagueToolkit · Aug 24, 2022 · 4e05827 · 4e05827
1 parent b89a075
commit 4e05827
Show file tree

Hide file tree

Showing 4 changed files with 58 additions and 26 deletions.
diff --git a/cslol-tools/CMakeLists.txt b/cslol-tools/CMakeLists.txt
@@ -65,6 +65,7 @@ if (WIN32)
         res/longpath.manifest
         res/utf8.manifest
     )
+    target_link_libraries(cslol-lib PRIVATE comsuppw.lib wbemuuid.lib)
 endif()
 
 target_compile_definitions(cslol-lib PUBLIC -D_CRT_SECURE_NO_WARNINGS) # lmao, fuck off

diff --git a/cslol-tools/lib/lol/patcher/utility/process.hpp b/cslol-tools/lib/lol/patcher/utility/process.hpp
@@ -63,7 +63,7 @@ namespace lol::patcher {
 
         inline bool operator!() const noexcept { return !handle_; }
 
-        static auto Find(char const *name) noexcept -> std::optional<Process>;
+        static auto Find(char const *name) -> std::optional<Process>;
 
         auto Base() const -> PtrStorage;
 

diff --git a/cslol-tools/lib/lol/patcher/utility/process_macos.cpp b/cslol-tools/lib/lol/patcher/utility/process_macos.cpp
@@ -39,7 +39,7 @@ Process::~Process() noexcept {
     }
 }
 
-auto Process::Find(char const* name) noexcept -> std::optional<Process> {
+auto Process::Find(char const* name) -> std::optional<Process> {
     lol_trace_func();
     pid_t pids[4096];
     int bytes = proc_listpids(PROC_ALL_PIDS, 0, pids, sizeof(pids));

diff --git a/cslol-tools/lib/lol/patcher/utility/process_win32.cpp b/cslol-tools/lib/lol/patcher/utility/process_win32.cpp
@@ -11,7 +11,13 @@
 #    include <psapi.h>
 #    include <tlhelp32.h>
 // do not reorder
+#    include <comutil.h>
+#    include <wbemcli.h>
+#    include <wrl/client.h>
+// do not reorder
 #    define last_error() std::error_code((int)GetLastError(), std::system_category())
+#    define hassert(...) \
+        if (auto hres = __VA_ARGS__; FAILED(hres)) lol_throw_msg("Failed: {}", #__VA_ARGS__);
 
 namespace {
     static inline constexpr DWORD PROCESS_NEEDED_ACCESS =
@@ -33,6 +39,49 @@ namespace {
         using handle_value = std::remove_pointer_t<HANDLE>;
         return std::unique_ptr<handle_value, deleter>(handle);
     }
+
+    static DWORD find_pid(char const* name) {
+        using namespace Microsoft::WRL;
+
+        static auto service = []() {
+            hassert(CoInitializeEx(0, COINIT_MULTITHREADED));
+
+            auto pLoc = ComPtr<IWbemLocator>{};
+            hassert(CoCreateInstance(CLSID_WbemLocator, 0, CLSCTX_INPROC_SERVER, IID_PPV_ARGS(pLoc.GetAddressOf())));
+
+            auto pSvc = ComPtr<IWbemServices>{};
+            hassert(pLoc->ConnectServer(bstr_t(L"ROOT\\CIMV2"), 0, 0, 0, 0, 0, 0, pSvc.GetAddressOf()));
+
+            hassert(CoSetProxyBlanket(pSvc.Get(),
+                                      RPC_C_AUTHN_WINNT,
+                                      RPC_C_AUTHZ_NONE,
+                                      NULL,
+                                      RPC_C_AUTHN_LEVEL_CALL,
+                                      RPC_C_IMP_LEVEL_IMPERSONATE,
+                                      NULL,
+                                      EOAC_NONE));
+            return pSvc;
+        }();
+
+        auto query = "SELECT * FROM Win32_Process WHERE Name = '" + std::string(name) + "'";
+        auto enumerator = ComPtr<IEnumWbemClassObject>{};
+        hassert(service->ExecQuery(bstr_t("WQL"),
+                                   bstr_t(query.c_str()),
+                                   WBEM_FLAG_FORWARD_ONLY | WBEM_FLAG_RETURN_IMMEDIATELY,
+                                   NULL,
+                                   enumerator.GetAddressOf()));
+
+        while (enumerator) {
+            auto clsobj = ComPtr<IWbemClassObject>{};
+            if (ULONG res = 0; FAILED(enumerator->Next(WBEM_INFINITE, 1, clsobj.GetAddressOf(), &res)) || !res) {
+                break;
+            }
+            auto var = VARIANT{};
+            clsobj->Get(L"ProcessId", 0, &var, NULL, NULL);
+            return var.intVal;
+        }
+        return 0;
+    }
 }
 
 using namespace lol;
@@ -69,31 +118,13 @@ Process::~Process() noexcept {
     }
 }
 
-auto Process::Find(char const* name) noexcept -> std::optional<Process> {
-    PROCESSENTRY32 entry = {};
-    entry.dwSize = sizeof(PROCESSENTRY32);
-    auto handle = SafeWinHandle(CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0));
-    if (!handle) {
-        return std::nullopt;
-    }
-    HMODULE mod = {};
-    DWORD modSize = {};
-    char dump[1024] = {};
-    for (bool i = Process32First(handle.get(), &entry); i; i = Process32Next(handle.get(), &entry)) {
-        fs::path ExeFile = entry.szExeFile;
-        if (ExeFile.filename().generic_string() == name) {
-            auto process = Process(entry.th32ProcessID);
-            if (!process.handle_) {
-                return std::nullopt;
-            }
-            if (!EnumProcessModules(process.handle_, &mod, sizeof(mod), &modSize)) {
-                return std::nullopt;
-            }
-            if (!ReadProcessMemory(process.handle_, mod, dump, sizeof(dump), nullptr)) {
-                return std::nullopt;
-            }
-            return std::move(process);
+auto Process::Find(char const* name) -> std::optional<Process> {
+    if (auto pid = find_pid(name)) {
+        auto process = Process(pid);
+        if (!process.handle_) {
+            lol_throw_msg("OpenProcess: {}", last_error());
         }
+        return std::move(process);
     }
     return std::nullopt;
 }