AuroraSupport/AuroraRuntime
2
1
Fork 0
Code Issues 27 Pull Requests Releases Wiki Activity

[+] NtWaitForMultipleObjects awareness

Browse Source
This commit is contained in:
Reece Wilson 2023-12-22 05:01:04 +00:00
parent e05bb8f5dc
commit 24fec2301c
3 changed files with 24 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Source/AuProcAddresses.NT.cpp
View File

@ -241,6 +241,7 @@ namespace Aurora
ADD_GET_PROC(Nt, NtTerminateProcess)
ADD_GET_PROC(Nt, NtQuerySymbolicLinkObject)
ADD_GET_PROC(Nt, NtOpenSymbolicLinkObject)
ADD_GET_PROC(Nt, NtWaitForMultipleObjects)
ADD_GET_PROC_BI(Kernel32, KernelBase, VirtualAlloc2)
ADD_GET_PROC_BI(Kernel32, KernelBase, MapViewOfFile3)

9
Source/AuProcAddresses.NT.hpp
View File

@ -25,6 +25,7 @@ struct _SP_DEVICE_INTERFACE_DETAIL_DATA_W;
struct _NETRESOURCEW;
enum _SE_OBJECT_TYPE;
enum _MINIDUMP_TYPE;
enum _OBJECT_WAIT_TYPE;
//#if defined(AURORA_COMPILER_MSVC)
struct _IP_ADAPTER_ADDRESSES_LH;
@ -94,6 +95,14 @@ namespace Aurora
PVOID Address
);
inline NTSTATUS(__stdcall *pNtWaitForMultipleObjects)(
ULONG ObjectCount,
PHANDLE ObjectsArray,
_OBJECT_WAIT_TYPE WaitType,
BOOLEAN Alertable,
AuUInt64 * pTimeOut
);
inline NTSTATUS(__stdcall *pNtQuerySymbolicLinkObject)(
HANDLE LinkHandle,
PUNICODE_STRING LinkTarget,

16
Source/IO/Loop/LoopQueue.NT.cpp
View File

@ -914,7 +914,7 @@ namespace Aurora::IO::Loop
bool active = this->hEvent_ == INVALID_HANDLE_VALUE;
while (count != index )
while (count != index)
{
auto next = AuMin(count - index, AuUInt32(MAXIMUM_WAIT_OBJECTS));
@ -960,7 +960,19 @@ namespace Aurora::IO::Loop
{
do
{
status = ::WaitForMultipleObjectsEx(next, this->handleArrayOr_.data() + index, false, sleepMS, true);
if (pNtWaitForMultipleObjects)
{
AuUInt64 uTimeout = -(AuMSToNS<AuUInt64>(1) / 2ull / 100ull);
if (AuSwInfo::IsWindows10OrGreater())
{
uTimeout /= 2ull;
}
status = pNtWaitForMultipleObjects(next, this->handleArrayOr_.data() + index, (_OBJECT_WAIT_TYPE)1, TRUE, &uTimeout);
}
else
{
status = ::WaitForMultipleObjectsEx(next, this->handleArrayOr_.data() + index, false, sleepMS, true);
}
}
while (status == WAIT_IO_COMPLETION);
}