2020-12-30 08:18:06 +01:00
|
|
|
// Copyright 2021 yuzu Emulator Project
|
|
|
|
// Licensed under GPLv2 or any later version
|
|
|
|
// Refer to the license.txt file included.
|
|
|
|
|
|
|
|
#include "core/arm/exclusive_monitor.h"
|
|
|
|
#include "core/core.h"
|
|
|
|
#include "core/hle/kernel/k_address_arbiter.h"
|
|
|
|
#include "core/hle/kernel/k_scheduler.h"
|
|
|
|
#include "core/hle/kernel/k_scoped_scheduler_lock_and_sleep.h"
|
2020-12-31 08:01:08 +01:00
|
|
|
#include "core/hle/kernel/k_thread.h"
|
2020-12-30 08:18:06 +01:00
|
|
|
#include "core/hle/kernel/kernel.h"
|
|
|
|
#include "core/hle/kernel/svc_results.h"
|
|
|
|
#include "core/hle/kernel/time_manager.h"
|
|
|
|
#include "core/memory.h"
|
|
|
|
|
|
|
|
namespace Kernel {
|
|
|
|
|
|
|
|
KAddressArbiter::KAddressArbiter(Core::System& system_)
|
|
|
|
: system{system_}, kernel{system.Kernel()} {}
|
|
|
|
KAddressArbiter::~KAddressArbiter() = default;
|
|
|
|
|
|
|
|
namespace {
|
|
|
|
|
|
|
|
bool ReadFromUser(Core::System& system, s32* out, VAddr address) {
|
|
|
|
*out = system.Memory().Read32(address);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool DecrementIfLessThan(Core::System& system, s32* out, VAddr address, s32 value) {
|
|
|
|
auto& monitor = system.Monitor();
|
2021-08-07 07:45:18 +02:00
|
|
|
const auto current_core = system.Kernel().CurrentPhysicalCoreIndex();
|
2020-12-30 08:18:06 +01:00
|
|
|
|
|
|
|
// TODO(bunnei): We should disable interrupts here via KScopedInterruptDisable.
|
|
|
|
// TODO(bunnei): We should call CanAccessAtomic(..) here.
|
|
|
|
|
|
|
|
// Load the value from the address.
|
|
|
|
const s32 current_value = static_cast<s32>(monitor.ExclusiveRead32(current_core, address));
|
|
|
|
|
|
|
|
// Compare it to the desired one.
|
|
|
|
if (current_value < value) {
|
|
|
|
// If less than, we want to try to decrement.
|
|
|
|
const s32 decrement_value = current_value - 1;
|
|
|
|
|
|
|
|
// Decrement and try to store.
|
|
|
|
if (!monitor.ExclusiveWrite32(current_core, address, static_cast<u32>(decrement_value))) {
|
|
|
|
// If we failed to store, try again.
|
|
|
|
DecrementIfLessThan(system, out, address, value);
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// Otherwise, clear our exclusive hold and finish
|
|
|
|
monitor.ClearExclusive();
|
|
|
|
}
|
|
|
|
|
|
|
|
// We're done.
|
|
|
|
*out = current_value;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool UpdateIfEqual(Core::System& system, s32* out, VAddr address, s32 value, s32 new_value) {
|
|
|
|
auto& monitor = system.Monitor();
|
2021-08-07 07:45:18 +02:00
|
|
|
const auto current_core = system.Kernel().CurrentPhysicalCoreIndex();
|
2020-12-30 08:18:06 +01:00
|
|
|
|
|
|
|
// TODO(bunnei): We should disable interrupts here via KScopedInterruptDisable.
|
|
|
|
// TODO(bunnei): We should call CanAccessAtomic(..) here.
|
|
|
|
|
|
|
|
// Load the value from the address.
|
|
|
|
const s32 current_value = static_cast<s32>(monitor.ExclusiveRead32(current_core, address));
|
|
|
|
|
|
|
|
// Compare it to the desired one.
|
|
|
|
if (current_value == value) {
|
|
|
|
// If equal, we want to try to write the new value.
|
|
|
|
|
|
|
|
// Try to store.
|
|
|
|
if (!monitor.ExclusiveWrite32(current_core, address, static_cast<u32>(new_value))) {
|
|
|
|
// If we failed to store, try again.
|
|
|
|
UpdateIfEqual(system, out, address, value, new_value);
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// Otherwise, clear our exclusive hold and finish.
|
|
|
|
monitor.ClearExclusive();
|
|
|
|
}
|
|
|
|
|
|
|
|
// We're done.
|
|
|
|
*out = current_value;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
} // namespace
|
|
|
|
|
|
|
|
ResultCode KAddressArbiter::Signal(VAddr addr, s32 count) {
|
|
|
|
// Perform signaling.
|
|
|
|
s32 num_waiters{};
|
|
|
|
{
|
|
|
|
KScopedSchedulerLock sl(kernel);
|
|
|
|
|
|
|
|
auto it = thread_tree.nfind_light({addr, -1});
|
|
|
|
while ((it != thread_tree.end()) && (count <= 0 || num_waiters < count) &&
|
|
|
|
(it->GetAddressArbiterKey() == addr)) {
|
2020-12-31 08:01:08 +01:00
|
|
|
KThread* target_thread = std::addressof(*it);
|
2021-11-10 04:21:20 +01:00
|
|
|
target_thread->SetWaitResult(ResultSuccess);
|
2020-12-30 08:18:06 +01:00
|
|
|
|
|
|
|
ASSERT(target_thread->IsWaitingForAddressArbiter());
|
|
|
|
target_thread->Wakeup();
|
|
|
|
|
|
|
|
it = thread_tree.erase(it);
|
|
|
|
target_thread->ClearAddressArbiter();
|
|
|
|
++num_waiters;
|
|
|
|
}
|
|
|
|
}
|
2021-05-21 07:05:04 +02:00
|
|
|
return ResultSuccess;
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
ResultCode KAddressArbiter::SignalAndIncrementIfEqual(VAddr addr, s32 value, s32 count) {
|
|
|
|
// Perform signaling.
|
|
|
|
s32 num_waiters{};
|
|
|
|
{
|
|
|
|
KScopedSchedulerLock sl(kernel);
|
|
|
|
|
|
|
|
// Check the userspace value.
|
|
|
|
s32 user_value{};
|
2021-02-04 03:56:20 +01:00
|
|
|
if (!UpdateIfEqual(system, &user_value, addr, value, value + 1)) {
|
|
|
|
LOG_ERROR(Kernel, "Invalid current memory!");
|
2021-02-13 00:43:01 +01:00
|
|
|
return ResultInvalidCurrentMemory;
|
2021-02-04 03:56:20 +01:00
|
|
|
}
|
2021-02-05 01:56:57 +01:00
|
|
|
if (user_value != value) {
|
2021-02-13 00:43:01 +01:00
|
|
|
return ResultInvalidState;
|
2021-02-05 01:56:57 +01:00
|
|
|
}
|
2020-12-30 08:18:06 +01:00
|
|
|
|
|
|
|
auto it = thread_tree.nfind_light({addr, -1});
|
|
|
|
while ((it != thread_tree.end()) && (count <= 0 || num_waiters < count) &&
|
|
|
|
(it->GetAddressArbiterKey() == addr)) {
|
2020-12-31 08:01:08 +01:00
|
|
|
KThread* target_thread = std::addressof(*it);
|
2021-11-10 04:21:20 +01:00
|
|
|
target_thread->SetWaitResult(ResultSuccess);
|
2020-12-30 08:18:06 +01:00
|
|
|
|
|
|
|
ASSERT(target_thread->IsWaitingForAddressArbiter());
|
|
|
|
target_thread->Wakeup();
|
|
|
|
|
|
|
|
it = thread_tree.erase(it);
|
|
|
|
target_thread->ClearAddressArbiter();
|
|
|
|
++num_waiters;
|
|
|
|
}
|
|
|
|
}
|
2021-05-21 07:05:04 +02:00
|
|
|
return ResultSuccess;
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
ResultCode KAddressArbiter::SignalAndModifyByWaitingCountIfEqual(VAddr addr, s32 value, s32 count) {
|
|
|
|
// Perform signaling.
|
|
|
|
s32 num_waiters{};
|
|
|
|
{
|
2021-02-04 03:45:52 +01:00
|
|
|
[[maybe_unused]] const KScopedSchedulerLock sl(kernel);
|
2020-12-30 08:18:06 +01:00
|
|
|
|
|
|
|
auto it = thread_tree.nfind_light({addr, -1});
|
|
|
|
// Determine the updated value.
|
|
|
|
s32 new_value{};
|
2021-02-04 03:45:52 +01:00
|
|
|
if (count <= 0) {
|
|
|
|
if (it != thread_tree.end() && it->GetAddressArbiterKey() == addr) {
|
|
|
|
new_value = value - 2;
|
2020-12-30 08:18:06 +01:00
|
|
|
} else {
|
2021-02-04 03:45:52 +01:00
|
|
|
new_value = value + 1;
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
} else {
|
2021-02-04 03:45:52 +01:00
|
|
|
if (it != thread_tree.end() && it->GetAddressArbiterKey() == addr) {
|
2020-12-30 08:18:06 +01:00
|
|
|
auto tmp_it = it;
|
|
|
|
s32 tmp_num_waiters{};
|
2021-02-04 03:45:52 +01:00
|
|
|
while (++tmp_it != thread_tree.end() && tmp_it->GetAddressArbiterKey() == addr) {
|
|
|
|
if (tmp_num_waiters++ >= count) {
|
|
|
|
break;
|
|
|
|
}
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
|
2021-02-04 03:45:52 +01:00
|
|
|
if (tmp_num_waiters < count) {
|
2020-12-30 08:18:06 +01:00
|
|
|
new_value = value - 1;
|
|
|
|
} else {
|
|
|
|
new_value = value;
|
|
|
|
}
|
2021-02-04 03:45:52 +01:00
|
|
|
} else {
|
|
|
|
new_value = value + 1;
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Check the userspace value.
|
|
|
|
s32 user_value{};
|
|
|
|
bool succeeded{};
|
|
|
|
if (value != new_value) {
|
2021-02-04 03:49:41 +01:00
|
|
|
succeeded = UpdateIfEqual(system, &user_value, addr, value, new_value);
|
2020-12-30 08:18:06 +01:00
|
|
|
} else {
|
2021-02-04 03:49:41 +01:00
|
|
|
succeeded = ReadFromUser(system, &user_value, addr);
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
|
2021-02-04 03:56:20 +01:00
|
|
|
if (!succeeded) {
|
|
|
|
LOG_ERROR(Kernel, "Invalid current memory!");
|
2021-02-13 00:43:01 +01:00
|
|
|
return ResultInvalidCurrentMemory;
|
2021-02-04 03:56:20 +01:00
|
|
|
}
|
2021-02-05 01:56:57 +01:00
|
|
|
if (user_value != value) {
|
2021-02-13 00:43:01 +01:00
|
|
|
return ResultInvalidState;
|
2021-02-05 01:56:57 +01:00
|
|
|
}
|
2020-12-30 08:18:06 +01:00
|
|
|
|
|
|
|
while ((it != thread_tree.end()) && (count <= 0 || num_waiters < count) &&
|
|
|
|
(it->GetAddressArbiterKey() == addr)) {
|
2020-12-31 08:01:08 +01:00
|
|
|
KThread* target_thread = std::addressof(*it);
|
2021-11-10 04:21:20 +01:00
|
|
|
target_thread->SetWaitResult(ResultSuccess);
|
2020-12-30 08:18:06 +01:00
|
|
|
|
|
|
|
ASSERT(target_thread->IsWaitingForAddressArbiter());
|
|
|
|
target_thread->Wakeup();
|
|
|
|
|
|
|
|
it = thread_tree.erase(it);
|
|
|
|
target_thread->ClearAddressArbiter();
|
|
|
|
++num_waiters;
|
|
|
|
}
|
|
|
|
}
|
2021-05-21 07:05:04 +02:00
|
|
|
return ResultSuccess;
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
ResultCode KAddressArbiter::WaitIfLessThan(VAddr addr, s32 value, bool decrement, s64 timeout) {
|
|
|
|
// Prepare to wait.
|
2020-12-31 08:01:08 +01:00
|
|
|
KThread* cur_thread = kernel.CurrentScheduler()->GetCurrentThread();
|
2020-12-30 08:18:06 +01:00
|
|
|
|
|
|
|
{
|
2021-01-20 06:05:24 +01:00
|
|
|
KScopedSchedulerLockAndSleep slp{kernel, cur_thread, timeout};
|
2020-12-30 08:18:06 +01:00
|
|
|
|
|
|
|
// Check that the thread isn't terminating.
|
|
|
|
if (cur_thread->IsTerminationRequested()) {
|
|
|
|
slp.CancelSleep();
|
2021-02-13 00:43:01 +01:00
|
|
|
return ResultTerminationRequested;
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// Set the synced object.
|
2021-11-10 04:21:20 +01:00
|
|
|
cur_thread->SetWaitResult(ResultTimedOut);
|
2020-12-30 08:18:06 +01:00
|
|
|
|
|
|
|
// Read the value from userspace.
|
|
|
|
s32 user_value{};
|
|
|
|
bool succeeded{};
|
|
|
|
if (decrement) {
|
2021-02-04 03:49:41 +01:00
|
|
|
succeeded = DecrementIfLessThan(system, &user_value, addr, value);
|
2020-12-30 08:18:06 +01:00
|
|
|
} else {
|
2021-02-04 03:49:41 +01:00
|
|
|
succeeded = ReadFromUser(system, &user_value, addr);
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
if (!succeeded) {
|
|
|
|
slp.CancelSleep();
|
2021-02-13 00:43:01 +01:00
|
|
|
return ResultInvalidCurrentMemory;
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// Check that the value is less than the specified one.
|
|
|
|
if (user_value >= value) {
|
|
|
|
slp.CancelSleep();
|
2021-02-13 00:43:01 +01:00
|
|
|
return ResultInvalidState;
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// Check that the timeout is non-zero.
|
|
|
|
if (timeout == 0) {
|
|
|
|
slp.CancelSleep();
|
2021-02-13 00:43:01 +01:00
|
|
|
return ResultTimedOut;
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// Set the arbiter.
|
2021-02-04 03:49:41 +01:00
|
|
|
cur_thread->SetAddressArbiter(&thread_tree, addr);
|
2020-12-30 08:18:06 +01:00
|
|
|
thread_tree.insert(*cur_thread);
|
|
|
|
cur_thread->SetState(ThreadState::Waiting);
|
2021-01-10 23:29:02 +01:00
|
|
|
cur_thread->SetWaitReasonForDebugging(ThreadWaitReasonForDebugging::Arbitration);
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// Cancel the timer wait.
|
2021-01-20 06:05:24 +01:00
|
|
|
kernel.TimeManager().UnscheduleTimeEvent(cur_thread);
|
2020-12-30 08:18:06 +01:00
|
|
|
|
|
|
|
// Remove from the address arbiter.
|
|
|
|
{
|
|
|
|
KScopedSchedulerLock sl(kernel);
|
|
|
|
|
|
|
|
if (cur_thread->IsWaitingForAddressArbiter()) {
|
|
|
|
thread_tree.erase(thread_tree.iterator_to(*cur_thread));
|
|
|
|
cur_thread->ClearAddressArbiter();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Get the result.
|
2021-11-10 04:21:20 +01:00
|
|
|
return cur_thread->GetWaitResult();
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
ResultCode KAddressArbiter::WaitIfEqual(VAddr addr, s32 value, s64 timeout) {
|
|
|
|
// Prepare to wait.
|
2020-12-31 08:01:08 +01:00
|
|
|
KThread* cur_thread = kernel.CurrentScheduler()->GetCurrentThread();
|
2020-12-30 08:18:06 +01:00
|
|
|
|
|
|
|
{
|
2021-01-20 06:05:24 +01:00
|
|
|
KScopedSchedulerLockAndSleep slp{kernel, cur_thread, timeout};
|
2020-12-30 08:18:06 +01:00
|
|
|
|
|
|
|
// Check that the thread isn't terminating.
|
|
|
|
if (cur_thread->IsTerminationRequested()) {
|
|
|
|
slp.CancelSleep();
|
2021-02-13 00:43:01 +01:00
|
|
|
return ResultTerminationRequested;
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// Set the synced object.
|
2021-11-10 04:21:20 +01:00
|
|
|
cur_thread->SetWaitResult(ResultTimedOut);
|
2020-12-30 08:18:06 +01:00
|
|
|
|
|
|
|
// Read the value from userspace.
|
|
|
|
s32 user_value{};
|
2021-02-04 03:49:41 +01:00
|
|
|
if (!ReadFromUser(system, &user_value, addr)) {
|
2020-12-30 08:18:06 +01:00
|
|
|
slp.CancelSleep();
|
2021-02-13 00:43:01 +01:00
|
|
|
return ResultInvalidCurrentMemory;
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// Check that the value is equal.
|
|
|
|
if (value != user_value) {
|
|
|
|
slp.CancelSleep();
|
2021-02-13 00:43:01 +01:00
|
|
|
return ResultInvalidState;
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// Check that the timeout is non-zero.
|
|
|
|
if (timeout == 0) {
|
|
|
|
slp.CancelSleep();
|
2021-02-13 00:43:01 +01:00
|
|
|
return ResultTimedOut;
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// Set the arbiter.
|
2021-02-04 03:49:41 +01:00
|
|
|
cur_thread->SetAddressArbiter(&thread_tree, addr);
|
2020-12-30 08:18:06 +01:00
|
|
|
thread_tree.insert(*cur_thread);
|
|
|
|
cur_thread->SetState(ThreadState::Waiting);
|
2021-01-10 23:29:02 +01:00
|
|
|
cur_thread->SetWaitReasonForDebugging(ThreadWaitReasonForDebugging::Arbitration);
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// Cancel the timer wait.
|
2021-01-20 06:05:24 +01:00
|
|
|
kernel.TimeManager().UnscheduleTimeEvent(cur_thread);
|
2020-12-30 08:18:06 +01:00
|
|
|
|
|
|
|
// Remove from the address arbiter.
|
|
|
|
{
|
|
|
|
KScopedSchedulerLock sl(kernel);
|
|
|
|
|
|
|
|
if (cur_thread->IsWaitingForAddressArbiter()) {
|
|
|
|
thread_tree.erase(thread_tree.iterator_to(*cur_thread));
|
|
|
|
cur_thread->ClearAddressArbiter();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Get the result.
|
2021-11-10 04:21:20 +01:00
|
|
|
return cur_thread->GetWaitResult();
|
2020-12-30 08:18:06 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
} // namespace Kernel
|