thirdpartymirrors/UnleashedRecomp
1
0
Fork 0
mirror of https://github.com/hedge-dev/UnleashedRecomp.git synced 2025-10-30 07:11:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Implement critical sections using WaitOnAddress.

Browse source
This commit is contained in:
Skyth 2024-11-04 13:16:58 +03:00
parent cd99cf2c04
commit fa5fc9aae7
2 changed files with 17 additions and 54 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
UnleashedRecomp/CMakeLists.txt
View file

@ -132,6 +132,7 @@ target_link_libraries(UnleashedRecomp PRIVATE
PkgConfig::tomlplusplus
zstd::libzstd_static
unofficial::concurrentqueue::concurrentqueue
Synchronization
)
target_include_directories(UnleashedRecomp PRIVATE

70
UnleashedRecomp/kernel/imports.cpp
View file

@ -586,62 +586,32 @@ struct Semaphore : HostObject<XKSEMAPHORE>
}
};
extern "C" NTSYSAPI NTSTATUS NTAPI NtReleaseKeyedEvent(
IN HANDLE KeyedEventHandle,
IN PVOID Key,
IN BOOLEAN Alertable,
IN PLARGE_INTEGER Timeout OPTIONAL);
// https://devblogs.microsoft.com/oldnewthing/20160825-00/?p=94165
void RtlLeaveCriticalSection(XRTL_CRITICAL_SECTION* cs)
{
//printf("!!! STUB !!! RtlLeaveCriticalSection\n");
if (--cs->RecursionCount != 0)
{
InterlockedDecrement(&cs->LockCount);
cs->RecursionCount--;
if (cs->RecursionCount != 0) {
return;
}
cs->OwningThread = NULL;
if (InterlockedDecrement(&cs->LockCount) != -1)
NtReleaseKeyedEvent(nullptr, cs, FALSE, nullptr);
InterlockedExchange(&cs->OwningThread, 0);
WakeByAddressSingle(&cs->OwningThread);
}
extern "C" NTSYSAPI NTSTATUS NTAPI NtWaitForKeyedEvent(
IN HANDLE KeyedEventHandle,
IN PVOID Key,
IN BOOLEAN Alertable,
IN PLARGE_INTEGER Timeout OPTIONAL);
void RtlEnterCriticalSection(XRTL_CRITICAL_SECTION* cs)
{
//printf("!!! STUB !!! RtlEnterCriticalSection\n");
const uint32_t thread = static_cast<uint32_t>(GetPPCContext()->r13.u64);
if (cs->OwningThread == thread)
DWORD thisThread = GetCurrentThreadId();
while (true)
{
InterlockedIncrement(&cs->LockCount);
++cs->RecursionCount;
return;
}
uint32_t spinCount = cs->Header.Absolute * 256;
while (spinCount--)
{
if (InterlockedCompareExchange(&cs->LockCount, 0, -1) == -1)
{
cs->OwningThread = thread;
cs->RecursionCount = 1;
DWORD previousOwner = InterlockedCompareExchangeAcquire(&cs->OwningThread, thisThread, 0);
if (previousOwner == 0 || previousOwner == thisThread) {
cs->RecursionCount++;
return;
}
WaitOnAddress(&cs->OwningThread, &previousOwner, sizeof(previousOwner), INFINITE);
}
if (InterlockedIncrement(&cs->LockCount) != 0)
NtWaitForKeyedEvent(nullptr, cs, FALSE, nullptr);
cs->OwningThread = thread;
cs->RecursionCount = 1;
}
void RtlImageXexHeaderField()
@ -1110,19 +1080,11 @@ void XexGetModuleHandle()
bool RtlTryEnterCriticalSection(XRTL_CRITICAL_SECTION* cs)
{
const uint32_t thread = static_cast<uint32_t>(GetPPCContext()->r13.u64);
DWORD thisThread = GetCurrentThreadId();
DWORD previousOwner = InterlockedCompareExchangeAcquire(&cs->OwningThread, thisThread, 0);
if (InterlockedCompareExchange(&cs->LockCount, 0, -1) == -1)
{
cs->OwningThread = thread;
cs->RecursionCount = 1;
return true;
}
if (cs->OwningThread == thread)
{
InterlockedIncrement(&cs->LockCount);
++cs->RecursionCount;
if (previousOwner == 0 || previousOwner == thisThread) {
cs->RecursionCount++;
return true;
}