2021-06-27 21:25:29 +00:00
|
|
|
/***
|
|
|
|
Copyright (C) 2021 J Reece Wilson (a/k/a "Reece"). All rights reserved.
|
|
|
|
|
2023-03-12 15:27:28 +00:00
|
|
|
File: AuMutex.NT.cpp
|
2021-06-27 21:25:29 +00:00
|
|
|
Date: 2021-6-12
|
|
|
|
Author: Reece
|
|
|
|
***/
|
2021-09-30 14:57:41 +00:00
|
|
|
#include <Source/RuntimeInternal.hpp>
|
2022-11-17 07:46:07 +00:00
|
|
|
#include "AuMutex.Generic.hpp"
|
2023-03-12 15:27:28 +00:00
|
|
|
#include "SMPYield.hpp"
|
2021-06-27 21:25:29 +00:00
|
|
|
|
|
|
|
#if !defined(_AURUNTIME_GENERICMUTEX)
|
2022-11-17 07:46:07 +00:00
|
|
|
#include "AuMutex.NT.hpp"
|
2021-06-27 21:25:29 +00:00
|
|
|
|
|
|
|
namespace Aurora::Threading::Primitives
|
|
|
|
{
|
|
|
|
Mutex::Mutex()
|
|
|
|
{
|
2023-02-16 16:01:21 +00:00
|
|
|
if (!pWaitOnAddress)
|
2022-12-28 10:22:44 +00:00
|
|
|
{
|
|
|
|
::InitializeSRWLock(&this->atomicHolder_);
|
|
|
|
::InitializeConditionVariable(&this->wakeup_);
|
|
|
|
}
|
2022-11-28 16:01:08 +00:00
|
|
|
this->state_ = 0;
|
2021-06-27 21:25:29 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
Mutex::~Mutex()
|
|
|
|
{
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
bool Mutex::HasOSHandle(AuMach &mach)
|
|
|
|
{
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool Mutex::TryLock()
|
|
|
|
{
|
2023-03-12 15:27:28 +00:00
|
|
|
return DoTryIf([=]()
|
|
|
|
{
|
|
|
|
return ::_interlockedbittestandset(&this->state_, 0) == 0;
|
|
|
|
});
|
2021-06-27 21:25:29 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
bool Mutex::HasLockImplementation()
|
|
|
|
{
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
void Mutex::Lock()
|
|
|
|
{
|
|
|
|
auto status = Lock(0);
|
|
|
|
SysAssert(status, "Couldn't lock Mutex object");
|
|
|
|
}
|
|
|
|
|
2023-03-12 15:27:28 +00:00
|
|
|
bool Mutex::Lock(AuUInt64 uTimeout)
|
|
|
|
{
|
|
|
|
return LockNS(AuMSToNS<AuUInt64>(uTimeout));
|
|
|
|
}
|
|
|
|
|
|
|
|
bool Mutex::LockNS(AuUInt64 uTimeout)
|
2021-06-27 21:25:29 +00:00
|
|
|
{
|
|
|
|
bool returnValue = false;
|
|
|
|
|
2022-12-28 10:22:44 +00:00
|
|
|
if (this->TryLock())
|
|
|
|
{
|
|
|
|
return true;
|
|
|
|
}
|
2021-06-27 21:25:29 +00:00
|
|
|
|
2023-03-12 15:27:28 +00:00
|
|
|
AuUInt64 uStartTime = Time::SteadyClockNS();
|
|
|
|
AuUInt64 uEndTime = uStartTime + uTimeout;
|
2021-06-27 21:25:29 +00:00
|
|
|
|
2023-02-16 16:01:21 +00:00
|
|
|
if (pWaitOnAddress)
|
2021-06-27 21:25:29 +00:00
|
|
|
{
|
2022-12-28 10:22:44 +00:00
|
|
|
auto state = this->state_;
|
|
|
|
while (::_interlockedbittestandset(&this->state_, 0) != 0)
|
2021-06-27 21:25:29 +00:00
|
|
|
{
|
2022-12-28 10:22:44 +00:00
|
|
|
AuUInt32 uTimeoutMS = INFINITE;
|
|
|
|
|
2023-03-12 15:27:28 +00:00
|
|
|
if (uTimeout != 0)
|
2021-06-27 21:25:29 +00:00
|
|
|
{
|
2023-03-12 15:27:28 +00:00
|
|
|
uStartTime = Time::SteadyClockNS();
|
2022-12-28 10:22:44 +00:00
|
|
|
if (uStartTime >= uEndTime)
|
|
|
|
{
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2023-03-12 15:27:28 +00:00
|
|
|
uTimeoutMS = AuNSToMS<AuInt64>(uEndTime - uStartTime);
|
2022-12-28 10:22:44 +00:00
|
|
|
}
|
|
|
|
|
2023-03-12 15:27:28 +00:00
|
|
|
if (!uTimeoutMS)
|
2022-12-28 10:22:44 +00:00
|
|
|
{
|
2023-03-12 15:27:28 +00:00
|
|
|
SMPPause();
|
|
|
|
AuThreading::ContextYield();
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
(void)pWaitOnAddress(&this->state_, &state, sizeof(this->state_), uTimeoutMS);
|
2021-06-27 21:25:29 +00:00
|
|
|
}
|
2022-02-18 11:52:52 +00:00
|
|
|
|
2022-12-28 10:22:44 +00:00
|
|
|
state = this->state_;
|
2021-06-27 21:25:29 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 10:22:44 +00:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
::AcquireSRWLockShared(&this->atomicHolder_);
|
|
|
|
|
|
|
|
BOOL status = false;
|
|
|
|
while (!this->TryLock())
|
2021-06-27 21:25:29 +00:00
|
|
|
{
|
2022-12-28 10:22:44 +00:00
|
|
|
AuUInt32 uTimeoutMS = INFINITE;
|
|
|
|
|
2023-03-12 15:27:28 +00:00
|
|
|
if (uTimeout != 0)
|
2022-12-28 10:22:44 +00:00
|
|
|
{
|
2023-03-12 15:27:28 +00:00
|
|
|
uStartTime = Time::SteadyClockNS();
|
2022-12-28 10:22:44 +00:00
|
|
|
if (uStartTime >= uEndTime)
|
|
|
|
{
|
|
|
|
goto exitWin32;
|
|
|
|
}
|
|
|
|
|
2023-03-12 15:27:28 +00:00
|
|
|
uTimeoutMS = AuNSToMS<AuInt64>(uEndTime - uStartTime);
|
2022-12-28 10:22:44 +00:00
|
|
|
}
|
|
|
|
|
2023-03-12 15:27:28 +00:00
|
|
|
if (!uTimeoutMS)
|
|
|
|
{
|
|
|
|
::ReleaseSRWLockShared(&this->atomicHolder_);
|
|
|
|
SMPPause();
|
|
|
|
AuThreading::ContextYield();
|
|
|
|
::AcquireSRWLockShared(&this->atomicHolder_);
|
|
|
|
}
|
|
|
|
else
|
2022-12-28 10:22:44 +00:00
|
|
|
{
|
2023-03-12 15:27:28 +00:00
|
|
|
(void)SleepConditionVariableSRW(&this->wakeup_, &this->atomicHolder_, uTimeoutMS, CONDITION_VARIABLE_LOCKMODE_SHARED);
|
2022-12-28 10:22:44 +00:00
|
|
|
}
|
2021-06-27 21:25:29 +00:00
|
|
|
}
|
|
|
|
|
2022-12-28 10:22:44 +00:00
|
|
|
returnValue = true;
|
2021-06-27 21:25:29 +00:00
|
|
|
|
2022-12-28 10:22:44 +00:00
|
|
|
exitWin32:
|
|
|
|
::ReleaseSRWLockShared(&this->atomicHolder_);
|
|
|
|
return returnValue;
|
|
|
|
}
|
2021-06-27 21:25:29 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void Mutex::Unlock()
|
|
|
|
{
|
2023-02-16 16:01:21 +00:00
|
|
|
if (!pWaitOnAddress)
|
2022-12-28 10:22:44 +00:00
|
|
|
{
|
|
|
|
::AcquireSRWLockExclusive(&this->atomicHolder_);
|
|
|
|
this->state_ = 0;
|
|
|
|
::ReleaseSRWLockExclusive(&this->atomicHolder_);
|
|
|
|
::WakeAllConditionVariable(&this->wakeup_);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
this->state_ = 0;
|
2023-02-16 16:01:21 +00:00
|
|
|
pWakeByAddressSingle((void *)&this->state_);
|
2022-12-28 10:22:44 +00:00
|
|
|
}
|
2021-06-27 21:25:29 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
AUKN_SYM IWaitable *MutexNew()
|
|
|
|
{
|
|
|
|
return _new Mutex();
|
|
|
|
}
|
|
|
|
|
2022-11-17 07:46:07 +00:00
|
|
|
AUKN_SYM void MutexRelease(IWaitable *pMutex)
|
2021-06-27 21:25:29 +00:00
|
|
|
{
|
2022-11-17 07:46:07 +00:00
|
|
|
AuSafeDelete<Mutex *>(pMutex);
|
2021-06-27 21:25:29 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|