AuroraRuntime/Source/AuProcAddresses.NT.hpp

/***
    Copyright (C) 2023 J Reece Wilson (a/k/a "Reece"). All rights reserved.

    File: AuProcAddresses.NT.hpp
    Date: 2023-2-16
    Author: Reece
***/
#pragma once

namespace Aurora
{
    void InitNTAddresses();

    static const wchar_t *kSyncDllName       { L"API-MS-Win-Core-Synch-l1-2-0.dll" };
    static const wchar_t *kNtDllName         { L"NTDLL.dll" };
    static const wchar_t *kKernel32DllName   { L"Kernel32.dll" };
    static const wchar_t *kKernelBaseDllName { L"KernelBase.dll" };

    inline BOOL(_stdcall *pWaitOnAddress)(
        volatile VOID *        Address,
        PVOID                  CompareAddress,
        SIZE_T                 AddressSize,
        DWORD                  dwMilliseconds
    );

    inline void(_stdcall *pWakeByAddressSingle)(
        PVOID                  Address
    );

    inline void(_stdcall *pWakeByAddressAll)(
        PVOID                  Address
    );

    inline DWORD(_stdcall *pNtDelayExecution)(
        BOOLEAN                Alertable,
        PLARGE_INTEGER         DelayInterval
    );

    inline PVOID(__stdcall *pVirtualAlloc2)(
        HANDLE                 Process,
        PVOID                  BaseAddress,
        SIZE_T                 Size,
        ULONG                  AllocationType,
        ULONG                  PageProtection,
        MEM_EXTENDED_PARAMETER *ExtendedParameters,
        ULONG                  ParameterCount
    );

    inline PVOID(__stdcall *pMapViewOfFile3)(
        HANDLE                 FileMapping,
        HANDLE                 Process,
        PVOID                  BaseAddress,
        ULONG64                Offset,
        SIZE_T                 ViewSize,
        ULONG                  AllocationType,
        ULONG                  PageProtection,
        MEM_EXTENDED_PARAMETER *ExtendedParameters,
        ULONG                  ParameterCount
    );

    inline PVOID(__stdcall *pUnmapViewOfFile2)(
        HANDLE                 Process,
        PVOID                  BaseAddress,
        ULONG                  UnmapFlags
    );

    inline NTSTATUS(__stdcall *pNtWaitForKeyedEvent)(
        HANDLE                 Handle,
        PVOID                  Key,
        BOOLEAN                Alertable,
        PLARGE_INTEGER         NTTimeout
    );

    inline NTSTATUS(__stdcall *pNtReleaseKeyedEvent)(
        HANDLE                 Handle,
        PVOID                  Key,
        BOOLEAN                Alertable,
        PLARGE_INTEGER         NTTimeout
    );

    inline NTSTATUS(__stdcall *pNtCreateKeyedEvent)(
        HANDLE                 Handle,
        ACCESS_MASK            Access,
        POBJECT_ATTRIBUTES     Attr,
        ULONG                  Flags
    );

    inline NTSTATUS(__stdcall *pNtOpenKeyedEvent)(
        HANDLE                 Handle,
        ACCESS_MASK            Access,
        POBJECT_ATTRIBUTES     Attr,
        ULONG                  Flags
    );
    
    inline NTSTATUS(__stdcall *pRtlWaitOnAddress)(
        const void *           addr,
        const void *           cmp,
        SIZE_T                 size,
        const LARGE_INTEGER *  timeout);

#if defined(AURORA_PLATFORM_WIN32)
    inline NTSTATUS(__stdcall *pRtlGetVersion)(
        PRTL_OSVERSIONINFOW    lpVersionInformation
    );
#endif

    inline BOOL(__stdcall *pGetSystemCpuSetInformation)(
        PSYSTEM_CPU_SET_INFORMATION Information,
        ULONG                  BufferLength,
        PULONG                 ReturnedLength,
        HANDLE                 Process,
        ULONG                  Flags
    );              

    inline BOOL(__stdcall * pGetLogicalProcessorInformation)(
        PSYSTEM_LOGICAL_PROCESSOR_INFORMATION Buffer,
        PDWORD                 ReturnedLength
    );

    inline bool gUseNativeWaitMutex {};
    inline bool gUseNativeWaitCondvar {};
    inline bool gUseNativeWaitSemapahore {};
}