mirror of
https://github.com/Atmosphere-NX/Atmosphere
synced 2024-11-14 00:56:35 +00:00
61 lines
2.7 KiB
C++
61 lines
2.7 KiB
C++
/*
|
|
* Copyright (c) 2018-2020 Atmosphère-NX
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms and conditions of the GNU General Public License,
|
|
* version 2, as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
|
* more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
#pragma once
|
|
#include <mesosphere/kern_common.hpp>
|
|
#include <mesosphere/kern_k_condition_variable.hpp>
|
|
|
|
namespace ams::kern {
|
|
|
|
class KAddressArbiter {
|
|
public:
|
|
using ThreadTree = KConditionVariable::ThreadTree;
|
|
private:
|
|
ThreadTree m_tree;
|
|
public:
|
|
constexpr KAddressArbiter() : m_tree() { /* ... */ }
|
|
|
|
Result SignalToAddress(uintptr_t addr, ams::svc::SignalType type, s32 value, s32 count) {
|
|
switch (type) {
|
|
case ams::svc::SignalType_Signal:
|
|
return this->Signal(addr, count);
|
|
case ams::svc::SignalType_SignalAndIncrementIfEqual:
|
|
return this->SignalAndIncrementIfEqual(addr, value, count);
|
|
case ams::svc::SignalType_SignalAndModifyByWaitingCountIfEqual:
|
|
return this->SignalAndModifyByWaitingCountIfEqual(addr, value, count);
|
|
MESOSPHERE_UNREACHABLE_DEFAULT_CASE();
|
|
}
|
|
}
|
|
|
|
Result WaitForAddress(uintptr_t addr, ams::svc::ArbitrationType type, s32 value, s64 timeout) {
|
|
switch (type) {
|
|
case ams::svc::ArbitrationType_WaitIfLessThan:
|
|
return this->WaitIfLessThan(addr, value, false, timeout);
|
|
case ams::svc::ArbitrationType_DecrementAndWaitIfLessThan:
|
|
return this->WaitIfLessThan(addr, value, true, timeout);
|
|
case ams::svc::ArbitrationType_WaitIfEqual:
|
|
return this->WaitIfEqual(addr, value, timeout);
|
|
MESOSPHERE_UNREACHABLE_DEFAULT_CASE();
|
|
}
|
|
}
|
|
private:
|
|
Result Signal(uintptr_t addr, s32 count);
|
|
Result SignalAndIncrementIfEqual(uintptr_t addr, s32 value, s32 count);
|
|
Result SignalAndModifyByWaitingCountIfEqual(uintptr_t addr, s32 value, s32 count);
|
|
Result WaitIfLessThan(uintptr_t addr, s32 value, bool decrement, s64 timeout);
|
|
Result WaitIfEqual(uintptr_t addr, s32 value, s64 timeout);
|
|
};
|
|
|
|
}
|