Implement get_module for Win32

BasedInc · Aug 16, 2023 · 35ea12b · 35ea12b
1 parent 3cb8e60
commit 35ea12b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/include/libhat/Process.hpp b/include/libhat/Process.hpp
@@ -1,18 +1,18 @@
 #pragma once
 
-#include <ranges>
+#include <span>
 #include <string_view>
 
 namespace hat::process {
 
     // TODO: Consider using a typedef or class instead? idk
     enum class module_t : uintptr_t {};
 
-    /// Returns the module for the curent process's base executable
+    /// Returns the module for the current process's base executable
     auto get_process_module() -> module_t;
 
     /// Returns a module by its given name in the current process
-    auto get_module(std::string_view name) -> module_t;
+    auto get_module(const std::string& name) -> module_t;
 
     /// Returns the module located at the specified base address
     auto module_at(uintptr_t address) -> module_t;

diff --git a/src/os/Win32.cpp b/src/os/Win32.cpp
@@ -118,6 +118,10 @@ namespace hat::process {
         return module_t{reinterpret_cast<uintptr_t>(GetModuleHandleA(nullptr))};
     }
 
+    module_t get_module(const std::string& name) {
+        return module_at(reinterpret_cast<uintptr_t>(GetModuleHandleA(name.c_str())));
+    }
+
     std::span<std::byte> get_module_data(module_t mod) {
         auto* const scanBytes = reinterpret_cast<std::byte*>(mod);
         auto* const ntHeaders = GetNTHeaders(mod);