2020-03-14 06:12:47 +00:00
|
|
|
/*
|
|
|
|
* 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/>.
|
|
|
|
*/
|
|
|
|
#include <mesosphere.hpp>
|
|
|
|
|
|
|
|
namespace ams::kern {
|
|
|
|
|
|
|
|
void KClientPort::Initialize(KPort *parent, s32 max_sessions) {
|
|
|
|
/* Set member variables. */
|
2020-12-18 01:18:47 +00:00
|
|
|
m_num_sessions = 0;
|
|
|
|
m_peak_sessions = 0;
|
|
|
|
m_parent = parent;
|
|
|
|
m_max_sessions = max_sessions;
|
2020-03-14 06:12:47 +00:00
|
|
|
}
|
|
|
|
|
2020-07-09 22:49:51 +01:00
|
|
|
void KClientPort::OnSessionFinalized() {
|
|
|
|
KScopedSchedulerLock sl;
|
|
|
|
|
2020-12-18 01:18:47 +00:00
|
|
|
const auto prev = m_num_sessions--;
|
|
|
|
if (prev == m_max_sessions) {
|
2020-07-09 22:49:51 +01:00
|
|
|
this->NotifyAvailable();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void KClientPort::OnServerClosed() {
|
|
|
|
MESOSPHERE_ASSERT_THIS();
|
|
|
|
}
|
|
|
|
|
2020-03-14 06:12:47 +00:00
|
|
|
bool KClientPort::IsLight() const {
|
|
|
|
return this->GetParent()->IsLight();
|
|
|
|
}
|
|
|
|
|
2021-04-07 22:45:38 +01:00
|
|
|
bool KClientPort::IsServerClosed() const {
|
|
|
|
return this->GetParent()->IsServerClosed();
|
|
|
|
}
|
|
|
|
|
2020-03-14 06:12:47 +00:00
|
|
|
void KClientPort::Destroy() {
|
|
|
|
/* Note with our parent that we're closed. */
|
2020-12-18 01:18:47 +00:00
|
|
|
m_parent->OnClientClosed();
|
2020-03-14 06:12:47 +00:00
|
|
|
|
|
|
|
/* Close our reference to our parent. */
|
2020-12-18 01:18:47 +00:00
|
|
|
m_parent->Close();
|
2020-03-14 06:12:47 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
bool KClientPort::IsSignaled() const {
|
|
|
|
MESOSPHERE_ASSERT_THIS();
|
2020-12-18 01:18:47 +00:00
|
|
|
return m_num_sessions < m_max_sessions;
|
2020-03-14 06:12:47 +00:00
|
|
|
}
|
|
|
|
|
2020-07-09 22:49:51 +01:00
|
|
|
Result KClientPort::CreateSession(KClientSession **out) {
|
|
|
|
MESOSPHERE_ASSERT_THIS();
|
|
|
|
|
|
|
|
/* Reserve a new session from the resource limit. */
|
|
|
|
KScopedResourceReservation session_reservation(GetCurrentProcessPointer(), ams::svc::LimitableResource_SessionCountMax);
|
|
|
|
R_UNLESS(session_reservation.Succeeded(), svc::ResultLimitReached());
|
|
|
|
|
|
|
|
/* Update the session counts. */
|
|
|
|
{
|
|
|
|
/* Atomically increment the number of sessions. */
|
|
|
|
s32 new_sessions;
|
|
|
|
{
|
2020-12-18 01:18:47 +00:00
|
|
|
const auto max = m_max_sessions;
|
|
|
|
auto cur_sessions = m_num_sessions.load(std::memory_order_acquire);
|
2020-07-09 22:49:51 +01:00
|
|
|
do {
|
|
|
|
R_UNLESS(cur_sessions < max, svc::ResultOutOfSessions());
|
|
|
|
new_sessions = cur_sessions + 1;
|
2020-12-18 01:18:47 +00:00
|
|
|
} while (!m_num_sessions.compare_exchange_weak(cur_sessions, new_sessions, std::memory_order_relaxed));
|
2020-07-09 22:49:51 +01:00
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Atomically update the peak session tracking. */
|
|
|
|
{
|
2020-12-18 01:18:47 +00:00
|
|
|
auto peak = m_peak_sessions.load(std::memory_order_acquire);
|
2020-07-09 22:49:51 +01:00
|
|
|
do {
|
|
|
|
if (peak >= new_sessions) {
|
|
|
|
break;
|
|
|
|
}
|
2020-12-18 01:18:47 +00:00
|
|
|
} while (!m_peak_sessions.compare_exchange_weak(peak, new_sessions, std::memory_order_relaxed));
|
2020-07-09 22:49:51 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Create a new session. */
|
|
|
|
KSession *session = KSession::Create();
|
|
|
|
if (session == nullptr) {
|
|
|
|
/* Decrement the session count. */
|
2020-12-18 01:18:47 +00:00
|
|
|
const auto prev = m_num_sessions--;
|
|
|
|
if (prev == m_max_sessions) {
|
2020-07-09 22:49:51 +01:00
|
|
|
this->NotifyAvailable();
|
|
|
|
}
|
|
|
|
|
|
|
|
return svc::ResultOutOfResource();
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Initialize the session. */
|
2020-12-18 01:18:47 +00:00
|
|
|
session->Initialize(this, m_parent->GetName());
|
2020-07-09 22:49:51 +01:00
|
|
|
|
|
|
|
/* Commit the session reservation. */
|
|
|
|
session_reservation.Commit();
|
|
|
|
|
|
|
|
/* Register the session. */
|
|
|
|
KSession::Register(session);
|
|
|
|
auto session_guard = SCOPE_GUARD {
|
|
|
|
session->GetClientSession().Close();
|
|
|
|
session->GetServerSession().Close();
|
|
|
|
};
|
|
|
|
|
|
|
|
/* Enqueue the session with our parent. */
|
2020-12-18 01:18:47 +00:00
|
|
|
R_TRY(m_parent->EnqueueSession(std::addressof(session->GetServerSession())));
|
2020-07-09 22:49:51 +01:00
|
|
|
|
|
|
|
/* We succeeded, so set the output. */
|
|
|
|
session_guard.Cancel();
|
|
|
|
*out = std::addressof(session->GetClientSession());
|
|
|
|
return ResultSuccess();
|
2020-07-14 10:24:26 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
Result KClientPort::CreateLightSession(KLightClientSession **out) {
|
|
|
|
MESOSPHERE_ASSERT_THIS();
|
|
|
|
|
|
|
|
/* Reserve a new session from the resource limit. */
|
|
|
|
KScopedResourceReservation session_reservation(GetCurrentProcessPointer(), ams::svc::LimitableResource_SessionCountMax);
|
|
|
|
R_UNLESS(session_reservation.Succeeded(), svc::ResultLimitReached());
|
|
|
|
|
|
|
|
/* Update the session counts. */
|
|
|
|
{
|
|
|
|
/* Atomically increment the number of sessions. */
|
|
|
|
s32 new_sessions;
|
|
|
|
{
|
2020-12-18 01:18:47 +00:00
|
|
|
const auto max = m_max_sessions;
|
|
|
|
auto cur_sessions = m_num_sessions.load(std::memory_order_acquire);
|
2020-07-14 10:24:26 +01:00
|
|
|
do {
|
|
|
|
R_UNLESS(cur_sessions < max, svc::ResultOutOfSessions());
|
|
|
|
new_sessions = cur_sessions + 1;
|
2020-12-18 01:18:47 +00:00
|
|
|
} while (!m_num_sessions.compare_exchange_weak(cur_sessions, new_sessions, std::memory_order_relaxed));
|
2020-07-14 10:24:26 +01:00
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Atomically update the peak session tracking. */
|
|
|
|
{
|
2020-12-18 01:18:47 +00:00
|
|
|
auto peak = m_peak_sessions.load(std::memory_order_acquire);
|
2020-07-14 10:24:26 +01:00
|
|
|
do {
|
|
|
|
if (peak >= new_sessions) {
|
|
|
|
break;
|
|
|
|
}
|
2020-12-18 01:18:47 +00:00
|
|
|
} while (!m_peak_sessions.compare_exchange_weak(peak, new_sessions, std::memory_order_relaxed));
|
2020-07-14 10:24:26 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Create a new session. */
|
|
|
|
KLightSession *session = KLightSession::Create();
|
|
|
|
if (session == nullptr) {
|
|
|
|
/* Decrement the session count. */
|
2020-12-18 01:18:47 +00:00
|
|
|
const auto prev = m_num_sessions--;
|
|
|
|
if (prev == m_max_sessions) {
|
2020-07-14 10:24:26 +01:00
|
|
|
this->NotifyAvailable();
|
|
|
|
}
|
|
|
|
|
|
|
|
return svc::ResultOutOfResource();
|
|
|
|
}
|
2020-07-09 22:49:51 +01:00
|
|
|
|
2020-07-14 10:24:26 +01:00
|
|
|
/* Initialize the session. */
|
2020-12-18 01:18:47 +00:00
|
|
|
session->Initialize(this, m_parent->GetName());
|
2020-07-14 10:24:26 +01:00
|
|
|
|
|
|
|
/* Commit the session reservation. */
|
|
|
|
session_reservation.Commit();
|
|
|
|
|
|
|
|
/* Register the session. */
|
|
|
|
KLightSession::Register(session);
|
|
|
|
auto session_guard = SCOPE_GUARD {
|
|
|
|
session->GetClientSession().Close();
|
|
|
|
session->GetServerSession().Close();
|
|
|
|
};
|
|
|
|
|
|
|
|
/* Enqueue the session with our parent. */
|
2020-12-18 01:18:47 +00:00
|
|
|
R_TRY(m_parent->EnqueueSession(std::addressof(session->GetServerSession())));
|
2020-07-14 10:24:26 +01:00
|
|
|
|
|
|
|
/* We succeeded, so set the output. */
|
|
|
|
session_guard.Cancel();
|
|
|
|
*out = std::addressof(session->GetClientSession());
|
|
|
|
return ResultSuccess();
|
2020-07-09 22:49:51 +01:00
|
|
|
}
|
|
|
|
|
2020-03-14 06:12:47 +00:00
|
|
|
}
|