Ryujinx-git/Ryujinx/OsHle/Mutex.cs

95 lines
2.5 KiB
C#
Raw Normal View History

2018-02-04 23:08:20 +00:00
using ChocolArm64.Memory;
using Ryujinx.OsHle.Handles;
using System.Collections.Concurrent;
2018-02-04 23:08:20 +00:00
namespace Ryujinx.OsHle
{
class Mutex
{
private const int MutexHasListenersMask = 0x40000000;
private Process Process;
2018-02-04 23:08:20 +00:00
private long MutexAddress;
private object EnterWaitLock;
private ConcurrentQueue<HThread> WaitingThreads;
public Mutex(Process Process, long MutexAddress, int OwnerThreadHandle)
2018-02-04 23:08:20 +00:00
{
this.Process = Process;
2018-02-04 23:08:20 +00:00
this.MutexAddress = MutexAddress;
//Process.Memory.WriteInt32(MutexAddress, OwnerThreadHandle);
2018-02-04 23:08:20 +00:00
EnterWaitLock = new object();
WaitingThreads = new ConcurrentQueue<HThread>();
2018-02-04 23:08:20 +00:00
}
public void WaitForLock(HThread RequestingThread, int RequestingThreadHandle)
2018-02-04 23:08:20 +00:00
{
lock (EnterWaitLock)
{
int CurrentThreadHandle = ReadMutexValue() & ~MutexHasListenersMask;
2018-02-04 23:08:20 +00:00
if (CurrentThreadHandle == RequestingThreadHandle ||
CurrentThreadHandle == 0)
{
return;
}
WriteMutexValue(CurrentThreadHandle | MutexHasListenersMask);
2018-02-04 23:08:20 +00:00
WaitingThreads.Enqueue(RequestingThread);
2018-02-04 23:08:20 +00:00
}
Process.Scheduler.WaitForSignal(RequestingThread);
2018-02-04 23:08:20 +00:00
}
public void GiveUpLock(int ThreadHandle)
{
lock (EnterWaitLock)
{
int CurrentThread = ReadMutexValue() & ~MutexHasListenersMask;
2018-02-04 23:08:20 +00:00
if (CurrentThread == ThreadHandle)
{
Unlock();
}
}
}
public void Unlock()
{
lock (EnterWaitLock)
{
int HasListeners = WaitingThreads.Count > 1 ? MutexHasListenersMask : 0;
WriteMutexValue(HasListeners);
HThread[] UnlockedThreads = new HThread[WaitingThreads.Count];
int Index = 0;
while (WaitingThreads.TryDequeue(out HThread Thread))
2018-02-04 23:08:20 +00:00
{
UnlockedThreads[Index++] = Thread;
2018-02-04 23:08:20 +00:00
}
Process.Scheduler.Signal(UnlockedThreads);
2018-02-04 23:08:20 +00:00
}
}
private int ReadMutexValue()
{
return AMemoryHelper.ReadInt32Exclusive(Process.Memory, MutexAddress);
}
private void WriteMutexValue(int Value)
{
AMemoryHelper.WriteInt32Exclusive(Process.Memory, MutexAddress, Value);
}
2018-02-04 23:08:20 +00:00
}
}