mirror of
https://github.com/Atmosphere-NX/Atmosphere
synced 2024-12-23 12:51:13 +00:00
119 lines
4.7 KiB
C++
119 lines
4.7 KiB
C++
/*
|
|
* Copyright (c) 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 "os_multiple_wait_holder_base.hpp"
|
|
|
|
#if defined(ATMOSPHERE_OS_HORIZON)
|
|
#include "os_multiple_wait_target_impl.os.horizon.hpp"
|
|
#elif defined(ATMOSPHERE_OS_WINDOWS)
|
|
#include "os_multiple_wait_target_impl.os.windows.hpp"
|
|
#elif defined(ATMOSPHERE_OS_LINUX)
|
|
#include "os_multiple_wait_target_impl.os.linux.hpp"
|
|
#elif defined(ATMOSPHERE_OS_MACOS)
|
|
#include "os_multiple_wait_target_impl.os.macos.hpp"
|
|
#else
|
|
#error "Unknown OS for ams::os::MultiWaitTargetImpl"
|
|
#endif
|
|
|
|
namespace ams::os::impl {
|
|
|
|
class MultiWaitImpl {
|
|
public:
|
|
static constexpr size_t MaximumHandleCount = MultiWaitTargetImpl::MaximumHandleCount;
|
|
static constexpr s32 WaitInvalid = -3;
|
|
static constexpr s32 WaitCancelled = -2;
|
|
static constexpr s32 WaitTimedOut = -1;
|
|
using MultiWaitList = util::IntrusiveListMemberTraitsByNonConstexprOffsetOf<&MultiWaitHolderBase::m_multi_wait_node>::ListType;
|
|
private:
|
|
MultiWaitList m_multi_wait_list;
|
|
MultiWaitHolderBase *m_signaled_holder;
|
|
TimeSpan m_current_time;
|
|
InternalCriticalSection m_cs_wait;
|
|
MultiWaitTargetImpl m_target_impl;
|
|
private:
|
|
Result WaitAnyImpl(MultiWaitHolderBase **out, bool infinite, TimeSpan timeout, bool reply, NativeHandle reply_target);
|
|
Result WaitAnyHandleImpl(MultiWaitHolderBase **out, bool infinite, TimeSpan timeout, bool reply, NativeHandle reply_target);
|
|
s32 BuildHandleArray(NativeHandle out_handles[], MultiWaitHolderBase *out_objects[], s32 num);
|
|
|
|
MultiWaitHolderBase *LinkHoldersToObjectList();
|
|
void UnlinkHoldersFromObjectList();
|
|
|
|
MultiWaitHolderBase *RecalculateNextTimeout(TimeSpan *out_min_timeout, TimeSpan end_time);
|
|
|
|
MultiWaitHolderBase *WaitAnyImpl(bool infinite, TimeSpan timeout) {
|
|
MultiWaitHolderBase *holder = nullptr;
|
|
|
|
const Result wait_result = this->WaitAnyImpl(std::addressof(holder), infinite, timeout, false, os::InvalidNativeHandle);
|
|
R_ASSERT(wait_result);
|
|
AMS_UNUSED(wait_result);
|
|
|
|
return holder;
|
|
}
|
|
public:
|
|
/* Wait. */
|
|
MultiWaitHolderBase *WaitAny() {
|
|
return this->WaitAnyImpl(true, TimeSpan::FromNanoSeconds(std::numeric_limits<s64>::max()));
|
|
}
|
|
|
|
MultiWaitHolderBase *TryWaitAny() {
|
|
return this->WaitAnyImpl(false, TimeSpan(0));
|
|
}
|
|
|
|
MultiWaitHolderBase *TimedWaitAny(TimeSpan ts) {
|
|
return this->WaitAnyImpl(false, ts);
|
|
}
|
|
|
|
Result ReplyAndReceive(MultiWaitHolderBase **out, NativeHandle reply_target) {
|
|
return this->WaitAnyImpl(out, true, TimeSpan::FromNanoSeconds(std::numeric_limits<s64>::max()), true, reply_target);
|
|
}
|
|
|
|
/* List management. */
|
|
bool IsEmpty() const {
|
|
return m_multi_wait_list.empty();
|
|
}
|
|
|
|
void LinkMultiWaitHolder(MultiWaitHolderBase &holder_base) {
|
|
m_multi_wait_list.push_back(holder_base);
|
|
}
|
|
|
|
void UnlinkMultiWaitHolder(MultiWaitHolderBase &holder_base) {
|
|
m_multi_wait_list.erase(m_multi_wait_list.iterator_to(holder_base));
|
|
}
|
|
|
|
void UnlinkAll() {
|
|
while (!this->IsEmpty()) {
|
|
m_multi_wait_list.front().SetMultiWait(nullptr);
|
|
m_multi_wait_list.pop_front();
|
|
}
|
|
}
|
|
|
|
void MoveAllFrom(MultiWaitImpl &other) {
|
|
/* Set ourselves as multi wait for all of the other's holders. */
|
|
for (auto &w : other.m_multi_wait_list) {
|
|
w.SetMultiWait(this);
|
|
}
|
|
m_multi_wait_list.splice(m_multi_wait_list.end(), other.m_multi_wait_list);
|
|
}
|
|
|
|
/* Other. */
|
|
TimeSpan GetCurrentTime() const {
|
|
return m_current_time;
|
|
}
|
|
|
|
void SignalAndWakeupThread(MultiWaitHolderBase *holder_base);
|
|
};
|
|
|
|
}
|