add REX::W32::GetComputerNameA/W

shad0wshayd3-FO4 · May 16, 2024 · 5ea83ff · 5ea83ff
1 parent 7526397
commit 5ea83ff
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/CommonLibF4/include/REX/W32/KERNEL32.h b/CommonLibF4/include/REX/W32/KERNEL32.h
@@ -444,6 +444,8 @@ namespace REX::W32
 	bool FindNextFileW(HANDLE a_file, WIN32_FIND_DATAW* a_data) noexcept;
 	bool FlushInstructionCache(HANDLE a_process, const void* a_baseAddr, std::size_t a_size) noexcept;
 	bool FreeLibrary(HMODULE a_module) noexcept;
+	bool GetComputerNameA(char* a_buffer, std::uint32_t* a_size) noexcept;
+	bool GetComputerNameW(wchar_t* a_buffer, std::uint32_t* a_size) noexcept;
 	HMODULE GetCurrentModule() noexcept;
 	HANDLE GetCurrentProcess() noexcept;
 	std::uint32_t GetCurrentThreadId() noexcept;

diff --git a/CommonLibF4/src/REX/W32/KERNEL32.cpp b/CommonLibF4/src/REX/W32/KERNEL32.cpp
@@ -19,6 +19,8 @@ REX_W32_IMPORT(REX::W32::BOOL, FindNextFileA, REX::W32::HANDLE, REX::W32::WIN32_
 REX_W32_IMPORT(REX::W32::BOOL, FindNextFileW, REX::W32::HANDLE, REX::W32::WIN32_FIND_DATAW*);
 REX_W32_IMPORT(REX::W32::BOOL, FlushInstructionCache, REX::W32::HANDLE, const void*, std::size_t);
 REX_W32_IMPORT(REX::W32::BOOL, FreeLibrary, REX::W32::HMODULE);
+REX_W32_IMPORT(REX::W32::BOOL, GetComputerNameA, char*, std::uint32_t*);
+REX_W32_IMPORT(REX::W32::BOOL, GetComputerNameW, wchar_t*, std::uint32_t*);
 REX_W32_IMPORT(REX::W32::HANDLE, GetCurrentProcess);
 REX_W32_IMPORT(std::uint32_t, GetCurrentThreadId);
 REX_W32_IMPORT(std::uint32_t, GetEnvironmentVariableA, const char*, char*, std::uint32_t);
@@ -167,6 +169,16 @@ namespace REX::W32
 		return ::W32_IMPL_FreeLibrary(a_module);
 	}
 
+	bool GetComputerNameA(char* a_buffer, std::uint32_t* a_size) noexcept
+	{
+		return ::W32_IMPL_GetComputerNameA(a_buffer, a_size);
+	}
+
+	bool GetComputerNameW(wchar_t* a_buffer, std::uint32_t* a_size) noexcept
+	{
+		return ::W32_IMPL_GetComputerNameW(a_buffer, a_size);
+	}
+
 	HMODULE GetCurrentModule() noexcept
 	{
 		return reinterpret_cast<HMODULE>(std::addressof(__ImageBase));