mirror of
https://github.com/Atmosphere-NX/Atmosphere.git
synced 2024-12-30 06:06:02 +00:00
86 lines
3.2 KiB
C++
86 lines
3.2 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_synchronization_object.hpp>
|
|
#include <mesosphere/kern_k_server_session.hpp>
|
|
#include <mesosphere/kern_k_client_session.hpp>
|
|
#include <mesosphere/kern_slab_helpers.hpp>
|
|
|
|
namespace ams::kern {
|
|
|
|
class KClientPort;
|
|
class KProcess;
|
|
|
|
class KSession final : public KAutoObjectWithSlabHeapAndContainer<KSession, KAutoObjectWithList> {
|
|
MESOSPHERE_AUTOOBJECT_TRAITS(KSession, KAutoObject);
|
|
private:
|
|
enum class State : u8 {
|
|
Invalid = 0,
|
|
Normal = 1,
|
|
ClientClosed = 2,
|
|
ServerClosed = 3,
|
|
};
|
|
private:
|
|
KServerSession m_server;
|
|
KClientSession m_client;
|
|
std::atomic<std::underlying_type<State>::type> m_atomic_state;
|
|
KClientPort *m_port;
|
|
uintptr_t m_name;
|
|
KProcess *m_process;
|
|
bool m_initialized;
|
|
private:
|
|
ALWAYS_INLINE void SetState(State state) {
|
|
m_atomic_state = static_cast<u8>(state);
|
|
}
|
|
|
|
ALWAYS_INLINE State GetState() const {
|
|
return static_cast<State>(m_atomic_state.load());
|
|
}
|
|
public:
|
|
constexpr KSession()
|
|
: m_server(), m_client(), m_atomic_state(static_cast<std::underlying_type<State>::type>(State::Invalid)), m_port(), m_name(), m_process(), m_initialized()
|
|
{
|
|
/* ... */
|
|
}
|
|
|
|
virtual ~KSession() { /* ... */ }
|
|
|
|
void Initialize(KClientPort *client_port, uintptr_t name);
|
|
virtual void Finalize() override;
|
|
|
|
virtual bool IsInitialized() const override { return m_initialized; }
|
|
virtual uintptr_t GetPostDestroyArgument() const override { return reinterpret_cast<uintptr_t>(m_process); }
|
|
|
|
static void PostDestroy(uintptr_t arg);
|
|
|
|
void OnServerClosed();
|
|
void OnClientClosed();
|
|
|
|
bool IsServerClosed() const { return this->GetState() != State::Normal; }
|
|
bool IsClientClosed() const { return this->GetState() != State::Normal; }
|
|
|
|
Result OnRequest(KSessionRequest *request) { return m_server.OnRequest(request); }
|
|
|
|
KClientSession &GetClientSession() { return m_client; }
|
|
KServerSession &GetServerSession() { return m_server; }
|
|
const KClientSession &GetClientSession() const { return m_client; }
|
|
const KServerSession &GetServerSession() const { return m_server; }
|
|
|
|
const KClientPort *GetParent() const { return m_port; }
|
|
};
|
|
|
|
}
|