2014-05-20 20:03:45 -07:00
|
|
|
// Copyright 2014 Citra Emulator Project
|
2014-12-16 21:38:14 -08:00
|
|
|
// Licensed under GPLv2 or any later version
|
2014-11-19 00:49:13 -08:00
|
|
|
// Refer to the license.txt file included.
|
2014-05-20 20:03:45 -07:00
|
|
|
|
|
|
|
#include <map>
|
|
|
|
#include <vector>
|
2015-01-31 13:22:40 -08:00
|
|
|
#include <boost/range/algorithm_ext/erase.hpp>
|
2015-05-06 00:06:12 -07:00
|
|
|
#include "common/assert.h"
|
2017-01-01 13:59:30 -08:00
|
|
|
#include "core/core.h"
|
2014-05-20 20:03:45 -07:00
|
|
|
#include "core/hle/kernel/kernel.h"
|
2016-09-20 23:52:38 -07:00
|
|
|
#include "core/hle/kernel/mutex.h"
|
2014-05-20 20:03:45 -07:00
|
|
|
#include "core/hle/kernel/thread.h"
|
|
|
|
|
|
|
|
namespace Kernel {
|
|
|
|
|
2015-01-20 15:33:23 -08:00
|
|
|
void ReleaseThreadMutexes(Thread* thread) {
|
2015-01-31 13:22:40 -08:00
|
|
|
for (auto& mtx : thread->held_mutexes) {
|
2017-01-01 13:59:30 -08:00
|
|
|
mtx->lock_count = 0;
|
|
|
|
mtx->holding_thread = nullptr;
|
|
|
|
mtx->WakeupAllWaitingThreads();
|
2014-12-07 12:44:21 -08:00
|
|
|
}
|
2015-01-31 13:22:40 -08:00
|
|
|
thread->held_mutexes.clear();
|
2014-12-07 12:44:21 -08:00
|
|
|
}
|
|
|
|
|
2016-09-18 18:01:46 -07:00
|
|
|
Mutex::Mutex() {}
|
|
|
|
Mutex::~Mutex() {}
|
2015-01-31 16:56:59 -08:00
|
|
|
|
2017-10-09 20:56:20 -07:00
|
|
|
SharedPtr<Mutex> Mutex::Create(bool initial_locked, VAddr addr, std::string name) {
|
2015-01-22 17:12:19 -08:00
|
|
|
SharedPtr<Mutex> mutex(new Mutex);
|
2014-05-20 20:03:45 -07:00
|
|
|
|
2015-02-09 19:06:09 -08:00
|
|
|
mutex->lock_count = 0;
|
2017-10-09 20:56:20 -07:00
|
|
|
mutex->addr = addr;
|
2015-01-22 17:12:19 -08:00
|
|
|
mutex->name = std::move(name);
|
2015-01-20 15:33:23 -08:00
|
|
|
mutex->holding_thread = nullptr;
|
2014-05-20 20:03:45 -07:00
|
|
|
|
2017-01-01 13:53:22 -08:00
|
|
|
// Acquire mutex with current thread if initialized as locked
|
2015-01-22 17:12:19 -08:00
|
|
|
if (initial_locked)
|
2017-01-01 13:53:22 -08:00
|
|
|
mutex->Acquire(GetCurrentThread());
|
2014-05-20 20:03:45 -07:00
|
|
|
|
2015-01-31 18:14:40 -08:00
|
|
|
return mutex;
|
2014-05-20 20:03:45 -07:00
|
|
|
}
|
|
|
|
|
2017-01-01 13:53:22 -08:00
|
|
|
bool Mutex::ShouldWait(Thread* thread) const {
|
|
|
|
return lock_count > 0 && thread != holding_thread;
|
2015-01-22 17:12:19 -08:00
|
|
|
}
|
|
|
|
|
2017-01-01 13:53:22 -08:00
|
|
|
void Mutex::Acquire(Thread* thread) {
|
|
|
|
ASSERT_MSG(!ShouldWait(thread), "object unavailable!");
|
2015-01-22 17:12:19 -08:00
|
|
|
|
2017-01-01 13:59:30 -08:00
|
|
|
// Actually "acquire" the mutex only if we don't already have it
|
2015-02-09 19:06:09 -08:00
|
|
|
if (lock_count == 0) {
|
2017-01-01 13:59:30 -08:00
|
|
|
priority = thread->current_priority;
|
2015-02-09 19:06:09 -08:00
|
|
|
thread->held_mutexes.insert(this);
|
2017-01-01 13:59:30 -08:00
|
|
|
holding_thread = thread;
|
2017-01-02 16:38:08 -08:00
|
|
|
thread->UpdatePriority();
|
2017-01-01 13:59:30 -08:00
|
|
|
Core::System::GetInstance().PrepareReschedule();
|
2015-02-09 19:06:09 -08:00
|
|
|
}
|
2015-01-22 17:12:19 -08:00
|
|
|
|
2015-02-09 19:06:09 -08:00
|
|
|
lock_count++;
|
2015-01-22 17:12:19 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
void Mutex::Release() {
|
2017-01-01 13:59:30 -08:00
|
|
|
// Only release if the mutex is held
|
2015-02-09 19:06:09 -08:00
|
|
|
if (lock_count > 0) {
|
|
|
|
lock_count--;
|
|
|
|
|
2017-01-01 13:59:30 -08:00
|
|
|
// Yield to the next thread only if we've fully released the mutex
|
2015-02-09 19:06:09 -08:00
|
|
|
if (lock_count == 0) {
|
|
|
|
holding_thread->held_mutexes.erase(this);
|
2017-01-02 16:38:08 -08:00
|
|
|
holding_thread->UpdatePriority();
|
2017-01-01 13:59:30 -08:00
|
|
|
holding_thread = nullptr;
|
|
|
|
WakeupAllWaitingThreads();
|
2017-01-01 08:57:02 -08:00
|
|
|
Core::System::GetInstance().PrepareReschedule();
|
2015-02-09 19:06:09 -08:00
|
|
|
}
|
|
|
|
}
|
2014-12-07 12:57:28 -08:00
|
|
|
}
|
2015-01-20 15:16:45 -08:00
|
|
|
|
2017-01-01 13:59:30 -08:00
|
|
|
void Mutex::AddWaitingThread(SharedPtr<Thread> thread) {
|
|
|
|
WaitObject::AddWaitingThread(thread);
|
2017-01-02 16:38:08 -08:00
|
|
|
thread->pending_mutexes.insert(this);
|
|
|
|
UpdatePriority();
|
2017-01-02 10:53:10 -08:00
|
|
|
}
|
2017-01-01 13:59:30 -08:00
|
|
|
|
2017-01-02 10:53:10 -08:00
|
|
|
void Mutex::RemoveWaitingThread(Thread* thread) {
|
|
|
|
WaitObject::RemoveWaitingThread(thread);
|
2017-01-02 16:38:08 -08:00
|
|
|
thread->pending_mutexes.erase(this);
|
|
|
|
UpdatePriority();
|
|
|
|
}
|
|
|
|
|
|
|
|
void Mutex::UpdatePriority() {
|
|
|
|
if (!holding_thread)
|
|
|
|
return;
|
|
|
|
|
2017-09-26 16:26:09 -07:00
|
|
|
u32 best_priority = THREADPRIO_LOWEST;
|
2017-01-02 16:38:08 -08:00
|
|
|
for (auto& waiter : GetWaitingThreads()) {
|
|
|
|
if (waiter->current_priority < best_priority)
|
|
|
|
best_priority = waiter->current_priority;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (best_priority != priority) {
|
|
|
|
priority = best_priority;
|
|
|
|
holding_thread->UpdatePriority();
|
|
|
|
}
|
2017-01-01 13:59:30 -08:00
|
|
|
}
|
|
|
|
|
2014-05-20 20:03:45 -07:00
|
|
|
} // namespace
|