System.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "Common.hpp"
 
 
namespace pwn::System
{
 
namespace details
{
Result<PVOID>
QueryInternal(const SYSTEM_INFORMATION_CLASS, const usize);
} // namespace details
 
auto
PageSize() -> u32;
 
u32
ProcessId(const HANDLE hProcess = ::GetCurrentProcess());
 
 
auto
ParentProcessId(const u32 dwProcessId) -> Result<u32>;
 
auto
PidOf(std::wstring_view const targetProcessName) -> Result<std::vector<u32>>;
 
const std::wstring
ComputerName();
 
Result<std::wstring>
UserName();
 
Result<std::wstring>
ModuleName(HMODULE hModule);
 
Result<std::wstring>
FileName();
 
std::tuple<u32, u32, u32>
WindowsVersion();
 
template<class T>
Result<std::shared_ptr<T>>
Query(SYSTEM_INFORMATION_CLASS SystemInformationClass)
{
    auto res = details::QueryInternal(SystemInformationClass, sizeof(T));
    if ( Failed(res) )
    {
        return Error(res);
    }
 
    const auto p = reinterpret_cast<T*>(Value(res));
    auto deleter = [](T* x)
    {
        ::LocalFree(x);
    };
    return Ok(std::shared_ptr<T>(p, deleter));
}
 
 
Result<std::tuple<u8, u8, u8, u8, u8>>
ProcessorCount();
 
 
Result<std::vector<RTL_PROCESS_MODULE_INFORMATION>>
Modules();
 
 
Result<std::vector<SYSTEM_HANDLE_TABLE_ENTRY_INFO>>
Handles();
 
 
Result<std::vector<std::tuple<u32, u32>>>
Threads();
} // namespace pwn::System