Atmosphere/libraries/libvapours/source/prfile2/prfile2_critical_section.cpp
2021-02-03 12:45:25 -08:00

143 lines
4.6 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/>.
*/
#if defined(ATMOSPHERE_IS_STRATOSPHERE)
#include <stratosphere.hpp>
#elif defined(ATMOSPHERE_IS_MESOSPHERE)
#include <mesosphere.hpp>
#elif defined(ATMOSPHERE_IS_EXOSPHERE)
#include <exosphere.hpp>
#else
#include <vapours.hpp>
#endif
namespace ams::prfile2 {
#if defined(AMS_PRFILE2_THREAD_SAFE)
struct CriticalSection::Resource {
os::MutexType mutex;
bool in_use;
};
namespace {
constexpr inline const auto NumCriticalSectionResources = 5 /* TODO: NumVolumes */;
constinit os::SdkMutex g_crit_resource_mutex;
constinit CriticalSection::Resource g_crit_resources[NumCriticalSectionResources];
CriticalSection::Resource *AllocateCriticalSectionResource() {
std::scoped_lock lk(g_crit_resource_mutex);
for (auto &resource : g_crit_resources) {
if (!resource.in_use) {
resource.in_use = true;
return std::addressof(resource);
}
}
AMS_ABORT("Failed to allocate critical section resource");
}
void AcquireCriticalSectionResource(CriticalSection::Resource *resource) {
return os::LockMutex(std::addressof(resource->mutex));
}
void ReleaseCriticalSectionResource(CriticalSection::Resource *resource) {
return os::UnlockMutex(std::addressof(resource->mutex));
}
}
void InitializeCriticalSection(CriticalSection *cs) {
/* Check pre-condition. */
AMS_ASSERT(cs->state == CriticalSection::State_NotInitialized);
/* Perform initialization. */
if (cs->state == CriticalSection::State_NotInitialized) {
cs->resource = AllocateCriticalSectionResource();
cs->owner = 0;
cs->state = CriticalSection::State_Initialized;
}
}
void FinalizeCriticalSection(CriticalSection *cs) {
/* Check pre-condition. */
AMS_ASSERT(cs->state == CriticalSection::State_Initialized);
/* TODO */
AMS_UNUSED(cs);
AMS_ABORT("prfile2::FinalizeCriticalSection");
}
void EnterCriticalSection(CriticalSection *cs) {
/* Check pre-condition. */
AMS_ASSERT(cs->state == CriticalSection::State_Initialized);
if (AMS_LIKELY(cs->lock_count == 0)) {
/* Acquire the lock .*/
AcquireCriticalSectionResource(cs->resource);
system::GetCurrentContextId(std::addressof(cs->owner));
} else {
/* Get the current context id. */
u64 current_context_id;
system::GetCurrentContextId(std::addressof(current_context_id));
/* If the lock isn't already held, acquire it. */
if (cs->owner != current_context_id) {
AcquireCriticalSectionResource(cs->resource);
cs->owner = current_context_id;
}
}
/* Increment the lock count. */
++cs->lock_count;
}
void ExitCriticalSection(CriticalSection *cs) {
/* Check pre-condition. */
AMS_ASSERT(cs->state == CriticalSection::State_Initialized);
/* Unlock. */
if ((--cs->lock_count) == 0) {
cs->owner = 0;
ReleaseCriticalSectionResource(cs->resource);
}
}
#else
void InitializeCriticalSection(CriticalSection *cs) {
AMS_UNUSED(cs);
}
void FinalizeCriticalSection(CriticalSection *cs) {
AMS_UNUSED(cs);
}
void EnterCriticalSection(CriticalSection *cs) {
AMS_UNUSED(cs);
}
void ExitCriticalSection(CriticalSection *cs) {
AMS_UNUSED(cs);
}
#endif
}