mirror of
https://github.com/Atmosphere-NX/Atmosphere
synced 2024-11-14 00:56:35 +00:00
htc: skeleton HtcsManagerImpl, implement HtcsMonitor
This commit is contained in:
parent
cb5a706659
commit
10255f7f51
20 changed files with 652 additions and 17 deletions
|
@ -20,6 +20,8 @@
|
||||||
namespace ams::htclow {
|
namespace ams::htclow {
|
||||||
|
|
||||||
enum class ModuleId : u8 {
|
enum class ModuleId : u8 {
|
||||||
|
Unknown = 0,
|
||||||
|
|
||||||
Htcfs = 1,
|
Htcfs = 1,
|
||||||
|
|
||||||
Htcmisc = 3,
|
Htcmisc = 3,
|
||||||
|
@ -31,4 +33,34 @@ namespace ams::htclow {
|
||||||
ModuleId _id;
|
ModuleId _id;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
constexpr void InitializeModule(ModuleType *out, ModuleId id) {
|
||||||
|
*out = {
|
||||||
|
._is_initialized = true,
|
||||||
|
._id = id,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
constexpr void FinalizeModule(ModuleType *out) {
|
||||||
|
*out = {
|
||||||
|
._is_initialized = false,
|
||||||
|
._id = ModuleId::Unknown,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
class Module final {
|
||||||
|
private:
|
||||||
|
ModuleType m_impl;
|
||||||
|
public:
|
||||||
|
constexpr explicit Module(ModuleId id) : m_impl() {
|
||||||
|
InitializeModule(std::addressof(m_impl), id);
|
||||||
|
}
|
||||||
|
|
||||||
|
constexpr ~Module() {
|
||||||
|
FinalizeModule(std::addressof(m_impl));
|
||||||
|
}
|
||||||
|
|
||||||
|
ModuleType *GetBase() { return std::addressof(m_impl); }
|
||||||
|
const ModuleType *GetBase() const { return std::addressof(m_impl); }
|
||||||
|
};
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -16,3 +16,5 @@
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
#include <stratosphere/htcs/htcs_types.hpp>
|
#include <stratosphere/htcs/htcs_types.hpp>
|
||||||
|
#include <stratosphere/htcs/impl/htcs_manager_holder.hpp>
|
||||||
|
#include <stratosphere/htcs/impl/htcs_channel_ids.hpp>
|
||||||
|
|
|
@ -0,0 +1,24 @@
|
||||||
|
/*
|
||||||
|
* 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 <vapours.hpp>
|
||||||
|
#include <stratosphere/htclow/htclow_channel_types.hpp>
|
||||||
|
|
||||||
|
namespace ams::htcs::impl {
|
||||||
|
|
||||||
|
constexpr inline htclow::ChannelId HtcsClientChannelId = 0;
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,33 @@
|
||||||
|
/*
|
||||||
|
* 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 <vapours.hpp>
|
||||||
|
#include <stratosphere/htcs/htcs_types.hpp>
|
||||||
|
|
||||||
|
namespace ams::htcs::impl {
|
||||||
|
|
||||||
|
class HtcsManager;
|
||||||
|
|
||||||
|
namespace HtcsManagerHolder {
|
||||||
|
|
||||||
|
void AddReference();
|
||||||
|
void Release();
|
||||||
|
|
||||||
|
HtcsManager *GetHtcsManager();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -24,6 +24,8 @@ namespace ams::htc::server::driver {
|
||||||
IDriver *m_driver;
|
IDriver *m_driver;
|
||||||
public:
|
public:
|
||||||
DriverManager(IDriver *driver) : m_driver(driver) { /* ... */ }
|
DriverManager(IDriver *driver) : m_driver(driver) { /* ... */ }
|
||||||
|
|
||||||
|
IDriver *GetDriver() { return m_driver; }
|
||||||
};
|
};
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -104,6 +104,7 @@ namespace ams::htc::server {
|
||||||
m_rpc_client.Close();
|
m_rpc_client.Close();
|
||||||
m_rpc_client.Cancel();
|
m_rpc_client.Cancel();
|
||||||
m_rpc_client.Wait();
|
m_rpc_client.Wait();
|
||||||
|
this->SetClientConnectionEvent(false);
|
||||||
};
|
};
|
||||||
|
|
||||||
/* Wait to become disconnected. */
|
/* Wait to become disconnected. */
|
||||||
|
@ -111,12 +112,7 @@ namespace ams::htc::server {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Set ourselves as disconnected. */
|
|
||||||
this->SetClientConnectionEvent(false);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Set ourselves as disconnected. */
|
|
||||||
this->SetClientConnectionEvent(false);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void HtcmiscImpl::ServerThread() {
|
void HtcmiscImpl::ServerThread() {
|
||||||
|
@ -147,6 +143,7 @@ namespace ams::htc::server {
|
||||||
m_rpc_server.Close();
|
m_rpc_server.Close();
|
||||||
m_rpc_server.Cancel();
|
m_rpc_server.Cancel();
|
||||||
m_rpc_server.Wait();
|
m_rpc_server.Wait();
|
||||||
|
this->SetServerConnectionEvent(false);
|
||||||
};
|
};
|
||||||
|
|
||||||
/* Wait to become disconnected. */
|
/* Wait to become disconnected. */
|
||||||
|
@ -154,12 +151,7 @@ namespace ams::htc::server {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Set ourselves as disconnected. */
|
|
||||||
this->SetServerConnectionEvent(false);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Set ourselves as disconnected. */
|
|
||||||
this->SetServerConnectionEvent(false);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void HtcmiscImpl::SetClientConnectionEvent(bool en) {
|
void HtcmiscImpl::SetClientConnectionEvent(bool en) {
|
||||||
|
|
|
@ -15,6 +15,7 @@
|
||||||
*/
|
*/
|
||||||
#include <stratosphere.hpp>
|
#include <stratosphere.hpp>
|
||||||
#include "htc_observer.hpp"
|
#include "htc_observer.hpp"
|
||||||
|
#include "../../htcs/impl/htcs_manager.hpp"
|
||||||
|
|
||||||
namespace ams::htc::server {
|
namespace ams::htc::server {
|
||||||
|
|
||||||
|
@ -29,7 +30,7 @@ namespace ams::htc::server {
|
||||||
m_is_service_available(false)
|
m_is_service_available(false)
|
||||||
{
|
{
|
||||||
/* Initialize htcs library. */
|
/* Initialize htcs library. */
|
||||||
/* TODO: AMS_ABORT("htcs::impl::HtcsManagerHolder::AddReference();"); */
|
htcs::impl::HtcsManagerHolder::AddReference();
|
||||||
|
|
||||||
/* Update our event state. */
|
/* Update our event state. */
|
||||||
this->UpdateEvent();
|
this->UpdateEvent();
|
||||||
|
@ -76,10 +77,13 @@ namespace ams::htc::server {
|
||||||
this->UpdateEvent();
|
this->UpdateEvent();
|
||||||
};
|
};
|
||||||
|
|
||||||
|
/* Get the htcs manager. */
|
||||||
|
auto * const htcs_manager = htcs::impl::HtcsManagerHolder::GetHtcsManager();
|
||||||
|
|
||||||
/* Get the events we're waiting on. */
|
/* Get the events we're waiting on. */
|
||||||
os::EventType * const stop_event = m_stop_event.GetBase();
|
os::EventType * const stop_event = m_stop_event.GetBase();
|
||||||
os::EventType * const conn_event = m_misc_impl.GetConnectionEvent();
|
os::EventType * const conn_event = m_misc_impl.GetConnectionEvent();
|
||||||
os::EventType * const htcs_event = nullptr /* TODO: htcs::impl::HtcsManagerHolder::GetHtcsManager()->GetServiceAvailabilityEvent() */;
|
os::EventType * const htcs_event = htcs_manager->GetServiceAvailabilityEvent();
|
||||||
|
|
||||||
/* Loop until we're asked to stop. */
|
/* Loop until we're asked to stop. */
|
||||||
while (!m_stopped) {
|
while (!m_stopped) {
|
||||||
|
@ -98,7 +102,7 @@ namespace ams::htc::server {
|
||||||
case 2:
|
case 2:
|
||||||
/* Htcs event, update our service status. */
|
/* Htcs event, update our service status. */
|
||||||
os::ClearEvent(htcs_event);
|
os::ClearEvent(htcs_event);
|
||||||
m_is_service_available = false /* TODO: htcs::impl::HtcsManagerHolder::GetHtcsManager()->IsServiceAvailable() */;
|
m_is_service_available = htcs_manager->IsServiceAvailable();
|
||||||
break;
|
break;
|
||||||
AMS_UNREACHABLE_DEFAULT_CASE();
|
AMS_UNREACHABLE_DEFAULT_CASE();
|
||||||
}
|
}
|
||||||
|
|
|
@ -27,7 +27,7 @@ namespace ams::htc::server::rpc {
|
||||||
}
|
}
|
||||||
|
|
||||||
HtcmiscRpcServer::HtcmiscRpcServer(driver::IDriver *driver, htclow::ChannelId channel)
|
HtcmiscRpcServer::HtcmiscRpcServer(driver::IDriver *driver, htclow::ChannelId channel)
|
||||||
: m_00(0),
|
: m_allocator(nullptr),
|
||||||
m_driver(driver),
|
m_driver(driver),
|
||||||
m_channel_id(channel),
|
m_channel_id(channel),
|
||||||
m_receive_thread_stack(g_receive_thread_stack),
|
m_receive_thread_stack(g_receive_thread_stack),
|
||||||
|
|
|
@ -25,7 +25,7 @@ namespace ams::htc::server::rpc {
|
||||||
/* TODO: where is this value coming from, again? */
|
/* TODO: where is this value coming from, again? */
|
||||||
static constexpr size_t BufferSize = 1_KB;
|
static constexpr size_t BufferSize = 1_KB;
|
||||||
private:
|
private:
|
||||||
u64 m_00;
|
mem::StandardAllocator *m_allocator;
|
||||||
driver::IDriver *m_driver;
|
driver::IDriver *m_driver;
|
||||||
htclow::ChannelId m_channel_id;
|
htclow::ChannelId m_channel_id;
|
||||||
void *m_receive_thread_stack;
|
void *m_receive_thread_stack;
|
||||||
|
|
|
@ -33,7 +33,7 @@ namespace ams::htc::server::rpc {
|
||||||
}
|
}
|
||||||
|
|
||||||
RpcClient::RpcClient(driver::IDriver *driver, htclow::ChannelId channel)
|
RpcClient::RpcClient(driver::IDriver *driver, htclow::ChannelId channel)
|
||||||
: m_00(0),
|
: m_allocator(nullptr),
|
||||||
m_driver(driver),
|
m_driver(driver),
|
||||||
m_channel_id(channel),
|
m_channel_id(channel),
|
||||||
m_receive_thread_stack(g_receive_thread_stack),
|
m_receive_thread_stack(g_receive_thread_stack),
|
||||||
|
@ -53,6 +53,53 @@ namespace ams::htc::server::rpc {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
RpcClient::RpcClient(mem::StandardAllocator *allocator, driver::IDriver *driver, htclow::ChannelId channel)
|
||||||
|
: m_allocator(allocator),
|
||||||
|
m_driver(driver),
|
||||||
|
m_channel_id(channel),
|
||||||
|
m_receive_thread_stack(m_allocator->Allocate(ThreadStackSize, os::ThreadStackAlignment)),
|
||||||
|
m_send_thread_stack(m_allocator->Allocate(ThreadStackSize, os::ThreadStackAlignment)),
|
||||||
|
m_mutex(g_rpc_mutex),
|
||||||
|
m_task_id_free_list(g_task_id_free_list),
|
||||||
|
m_task_table(g_task_table),
|
||||||
|
m_task_active(),
|
||||||
|
m_task_queue(),
|
||||||
|
m_cancelled(false),
|
||||||
|
m_thread_running(false)
|
||||||
|
{
|
||||||
|
/* Initialize all events. */
|
||||||
|
for (size_t i = 0; i < MaxRpcCount; ++i) {
|
||||||
|
os::InitializeEvent(std::addressof(m_receive_buffer_available_events[i]), false, os::EventClearMode_AutoClear);
|
||||||
|
os::InitializeEvent(std::addressof(m_send_buffer_available_events[i]), false, os::EventClearMode_AutoClear);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
RpcClient::~RpcClient() {
|
||||||
|
/* Finalize all events. */
|
||||||
|
for (size_t i = 0; i < MaxRpcCount; ++i) {
|
||||||
|
os::FinalizeEvent(std::addressof(m_receive_buffer_available_events[i]));
|
||||||
|
os::FinalizeEvent(std::addressof(m_send_buffer_available_events[i]));
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Free the thread stacks. */
|
||||||
|
if (m_allocator != nullptr) {
|
||||||
|
m_allocator->Free(m_receive_thread_stack);
|
||||||
|
m_allocator->Free(m_send_thread_stack);
|
||||||
|
}
|
||||||
|
m_receive_thread_stack = nullptr;
|
||||||
|
m_send_thread_stack = nullptr;
|
||||||
|
|
||||||
|
/* Free all tasks. */
|
||||||
|
for (u32 i = 0; i < MaxRpcCount; ++i) {
|
||||||
|
if (m_task_active[i]) {
|
||||||
|
std::scoped_lock lk(m_mutex);
|
||||||
|
|
||||||
|
m_task_table.Delete(i);
|
||||||
|
m_task_id_free_list.Free(i);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
void RpcClient::Open() {
|
void RpcClient::Open() {
|
||||||
R_ABORT_UNLESS(m_driver->Open(m_channel_id));
|
R_ABORT_UNLESS(m_driver->Open(m_channel_id));
|
||||||
}
|
}
|
||||||
|
|
|
@ -27,7 +27,7 @@ namespace ams::htc::server::rpc {
|
||||||
/* TODO: where is this value coming from, again? */
|
/* TODO: where is this value coming from, again? */
|
||||||
static constexpr size_t BufferSize = 0xE400;
|
static constexpr size_t BufferSize = 0xE400;
|
||||||
private:
|
private:
|
||||||
u64 m_00;
|
mem::StandardAllocator *m_allocator;
|
||||||
driver::IDriver *m_driver;
|
driver::IDriver *m_driver;
|
||||||
htclow::ChannelId m_channel_id;
|
htclow::ChannelId m_channel_id;
|
||||||
void *m_receive_thread_stack;
|
void *m_receive_thread_stack;
|
||||||
|
@ -53,6 +53,8 @@ namespace ams::htc::server::rpc {
|
||||||
Result SendThread();
|
Result SendThread();
|
||||||
public:
|
public:
|
||||||
RpcClient(driver::IDriver *driver, htclow::ChannelId channel);
|
RpcClient(driver::IDriver *driver, htclow::ChannelId channel);
|
||||||
|
RpcClient(mem::StandardAllocator *allocator, driver::IDriver *driver, htclow::ChannelId channel);
|
||||||
|
~RpcClient();
|
||||||
public:
|
public:
|
||||||
void Open();
|
void Open();
|
||||||
void Close();
|
void Close();
|
||||||
|
|
39
libraries/libstratosphere/source/htcs/impl/htcs_manager.cpp
Normal file
39
libraries/libstratosphere/source/htcs/impl/htcs_manager.cpp
Normal file
|
@ -0,0 +1,39 @@
|
||||||
|
/*
|
||||||
|
* 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 <stratosphere.hpp>
|
||||||
|
#include "htcs_manager.hpp"
|
||||||
|
#include "htcs_manager_impl.hpp"
|
||||||
|
|
||||||
|
namespace ams::htcs::impl {
|
||||||
|
|
||||||
|
HtcsManager::HtcsManager(mem::StandardAllocator *allocator, htclow::HtclowManager *htclow_manager) : m_allocator(allocator), m_impl(static_cast<HtcsManagerImpl *>(allocator->Allocate(sizeof(HtcsManagerImpl), alignof(HtcsManagerImpl)))) {
|
||||||
|
std::construct_at(m_impl, m_allocator, htclow_manager);
|
||||||
|
}
|
||||||
|
|
||||||
|
HtcsManager::~HtcsManager() {
|
||||||
|
std::destroy_at(m_impl);
|
||||||
|
m_allocator->Free(m_impl);
|
||||||
|
}
|
||||||
|
|
||||||
|
os::EventType *HtcsManager::GetServiceAvailabilityEvent() {
|
||||||
|
return m_impl->GetServiceAvailabilityEvent();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool HtcsManager::IsServiceAvailable() {
|
||||||
|
return m_impl->IsServiceAvailable();
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
37
libraries/libstratosphere/source/htcs/impl/htcs_manager.hpp
Normal file
37
libraries/libstratosphere/source/htcs/impl/htcs_manager.hpp
Normal file
|
@ -0,0 +1,37 @@
|
||||||
|
/*
|
||||||
|
* 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 <stratosphere.hpp>
|
||||||
|
#include "../../htclow/htclow_manager.hpp"
|
||||||
|
|
||||||
|
namespace ams::htcs::impl {
|
||||||
|
|
||||||
|
class HtcsManagerImpl;
|
||||||
|
|
||||||
|
class HtcsManager {
|
||||||
|
private:
|
||||||
|
mem::StandardAllocator *m_allocator;
|
||||||
|
HtcsManagerImpl *m_impl;
|
||||||
|
public:
|
||||||
|
HtcsManager(mem::StandardAllocator *allocator, htclow::HtclowManager *htclow_manager);
|
||||||
|
~HtcsManager();
|
||||||
|
public:
|
||||||
|
os::EventType *GetServiceAvailabilityEvent();
|
||||||
|
|
||||||
|
bool IsServiceAvailable();
|
||||||
|
};
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,79 @@
|
||||||
|
/*
|
||||||
|
* 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 <stratosphere.hpp>
|
||||||
|
#include "htcs_manager.hpp"
|
||||||
|
|
||||||
|
namespace ams::htcs::impl::HtcsManagerHolder {
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
constinit os::SdkMutex g_holder_mutex;
|
||||||
|
constinit int g_holder_reference_count = 0;
|
||||||
|
|
||||||
|
mem::StandardAllocator g_allocator;
|
||||||
|
|
||||||
|
constinit HtcsManager *g_manager = nullptr;
|
||||||
|
|
||||||
|
alignas(os::MemoryPageSize) u8 g_heap_buffer[416_KB];
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
void AddReference() {
|
||||||
|
std::scoped_lock lk(g_holder_mutex);
|
||||||
|
|
||||||
|
if ((g_holder_reference_count++) == 0) {
|
||||||
|
/* Add reference to the htclow manager. */
|
||||||
|
htclow::HtclowManagerHolder::AddReference();
|
||||||
|
|
||||||
|
/* Initialize the allocator for the manager. */
|
||||||
|
g_allocator.Initialize(g_heap_buffer, sizeof(g_heap_buffer));
|
||||||
|
|
||||||
|
/* Allocate the manager. */
|
||||||
|
g_manager = static_cast<HtcsManager *>(g_allocator.Allocate(sizeof(HtcsManager), alignof(HtcsManager)));
|
||||||
|
|
||||||
|
/* Construct the manager. */
|
||||||
|
std::construct_at(g_manager, std::addressof(g_allocator), htclow::HtclowManagerHolder::GetHtclowManager());
|
||||||
|
}
|
||||||
|
|
||||||
|
AMS_ASSERT(g_holder_reference_count > 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
void Release() {
|
||||||
|
std::scoped_lock lk(g_holder_mutex);
|
||||||
|
|
||||||
|
AMS_ASSERT(g_holder_reference_count > 0);
|
||||||
|
|
||||||
|
if ((--g_holder_reference_count) == 0) {
|
||||||
|
/* Destroy the manager. */
|
||||||
|
std::destroy_at(g_manager);
|
||||||
|
g_allocator.Free(g_manager);
|
||||||
|
g_manager = nullptr;
|
||||||
|
|
||||||
|
/* Finalize the allocator. */
|
||||||
|
g_allocator.Finalize();
|
||||||
|
|
||||||
|
/* Release reference to the htclow manager. */
|
||||||
|
htclow::HtclowManagerHolder::Release();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
HtcsManager *GetHtcsManager() {
|
||||||
|
std::scoped_lock lk(g_holder_mutex);
|
||||||
|
|
||||||
|
return g_manager;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,49 @@
|
||||||
|
/*
|
||||||
|
* 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 <stratosphere.hpp>
|
||||||
|
#include "htcs_manager.hpp"
|
||||||
|
#include "htcs_manager_impl.hpp"
|
||||||
|
|
||||||
|
namespace ams::htcs::impl {
|
||||||
|
|
||||||
|
HtcsManagerImpl::HtcsManagerImpl(mem::StandardAllocator *allocator, htclow::HtclowManager *htclow_manager)
|
||||||
|
: m_allocator(allocator),
|
||||||
|
m_driver(htclow_manager, htclow::ModuleId::Htcs),
|
||||||
|
m_driver_manager(std::addressof(m_driver)),
|
||||||
|
m_rpc_client(m_allocator, std::addressof(m_driver), HtcsClientChannelId),
|
||||||
|
m_data_channel_manager(std::addressof(m_rpc_client), htclow_manager),
|
||||||
|
m_service(m_allocator, m_driver_manager.GetDriver(), std::addressof(m_rpc_client), std::addressof(m_data_channel_manager)),
|
||||||
|
m_monitor(m_allocator, m_driver_manager.GetDriver(), std::addressof(m_rpc_client), std::addressof(m_service))
|
||||||
|
{
|
||||||
|
/* Start the monitor. */
|
||||||
|
m_monitor.Start();
|
||||||
|
}
|
||||||
|
|
||||||
|
HtcsManagerImpl::~HtcsManagerImpl() {
|
||||||
|
/* Cancel our monitor. */
|
||||||
|
m_monitor.Cancel();
|
||||||
|
m_monitor.Wait();
|
||||||
|
}
|
||||||
|
|
||||||
|
os::EventType *HtcsManagerImpl::GetServiceAvailabilityEvent() {
|
||||||
|
return m_monitor.GetServiceAvailabilityEvent();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool HtcsManagerImpl::IsServiceAvailable() {
|
||||||
|
return m_monitor.IsServiceAvailable();
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,46 @@
|
||||||
|
/*
|
||||||
|
* 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 <stratosphere.hpp>
|
||||||
|
#include "../../htclow/htclow_manager.hpp"
|
||||||
|
#include "../../htc/server/driver/htc_htclow_driver.hpp"
|
||||||
|
#include "../../htc/server/driver/htc_driver_manager.hpp"
|
||||||
|
#include "../../htc/server/rpc/htc_rpc_client.hpp"
|
||||||
|
#include "rpc/htcs_data_channel_manager.hpp"
|
||||||
|
#include "htcs_service.hpp"
|
||||||
|
#include "htcs_monitor.hpp"
|
||||||
|
|
||||||
|
namespace ams::htcs::impl {
|
||||||
|
|
||||||
|
class HtcsManagerImpl {
|
||||||
|
private:
|
||||||
|
mem::StandardAllocator *m_allocator;
|
||||||
|
htc::server::driver::HtclowDriver m_driver;
|
||||||
|
htc::server::driver::DriverManager m_driver_manager;
|
||||||
|
htc::server::rpc::RpcClient m_rpc_client;
|
||||||
|
rpc::DataChannelManager m_data_channel_manager;
|
||||||
|
HtcsService m_service;
|
||||||
|
HtcsMonitor m_monitor;
|
||||||
|
public:
|
||||||
|
HtcsManagerImpl(mem::StandardAllocator *allocator, htclow::HtclowManager *htclow_manager);
|
||||||
|
~HtcsManagerImpl();
|
||||||
|
public:
|
||||||
|
os::EventType *GetServiceAvailabilityEvent();
|
||||||
|
|
||||||
|
bool IsServiceAvailable();
|
||||||
|
};
|
||||||
|
|
||||||
|
}
|
103
libraries/libstratosphere/source/htcs/impl/htcs_monitor.cpp
Normal file
103
libraries/libstratosphere/source/htcs/impl/htcs_monitor.cpp
Normal file
|
@ -0,0 +1,103 @@
|
||||||
|
/*
|
||||||
|
* 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 <stratosphere.hpp>
|
||||||
|
#include "htcs_manager.hpp"
|
||||||
|
#include "htcs_manager_impl.hpp"
|
||||||
|
|
||||||
|
namespace ams::htcs::impl {
|
||||||
|
|
||||||
|
HtcsMonitor::HtcsMonitor(mem::StandardAllocator *allocator, htc::server::driver::IDriver *drv, htc::server::rpc::RpcClient *rc, HtcsService *srv)
|
||||||
|
: m_allocator(allocator),
|
||||||
|
m_driver(drv),
|
||||||
|
m_rpc_client(rc),
|
||||||
|
m_service(srv),
|
||||||
|
m_monitor_thread_stack(m_allocator->Allocate(os::MemoryPageSize, os::ThreadStackAlignment)),
|
||||||
|
m_mutex(),
|
||||||
|
m_cancel_event(os::EventClearMode_ManualClear),
|
||||||
|
m_service_availability_event(os::EventClearMode_ManualClear),
|
||||||
|
m_cancelled(false),
|
||||||
|
m_is_service_available(false)
|
||||||
|
{
|
||||||
|
/* ... */
|
||||||
|
}
|
||||||
|
|
||||||
|
HtcsMonitor::~HtcsMonitor() {
|
||||||
|
/* Free thread stack. */
|
||||||
|
m_allocator->Free(m_monitor_thread_stack);
|
||||||
|
}
|
||||||
|
|
||||||
|
void HtcsMonitor::Start() {
|
||||||
|
/* Create the monitor thread. */
|
||||||
|
R_ABORT_UNLESS(os::CreateThread(std::addressof(m_monitor_thread), ThreadEntry, this, m_monitor_thread_stack, os::MemoryPageSize, AMS_GET_SYSTEM_THREAD_PRIORITY(htc, HtcsMonitor)));
|
||||||
|
|
||||||
|
/* Set thread name. */
|
||||||
|
os::SetThreadNamePointer(std::addressof(m_monitor_thread), AMS_GET_SYSTEM_THREAD_NAME(htc, HtcsMonitor));
|
||||||
|
|
||||||
|
/* Start the monitor thread. */
|
||||||
|
os::StartThread(std::addressof(m_monitor_thread));
|
||||||
|
}
|
||||||
|
|
||||||
|
void HtcsMonitor::Cancel() {
|
||||||
|
/* Cancel, and signal. */
|
||||||
|
m_cancelled = true;
|
||||||
|
m_cancel_event.Signal();
|
||||||
|
}
|
||||||
|
|
||||||
|
void HtcsMonitor::Wait() {
|
||||||
|
/* Wait for the thread. */
|
||||||
|
os::WaitThread(std::addressof(m_monitor_thread));
|
||||||
|
os::DestroyThread(std::addressof(m_monitor_thread));
|
||||||
|
}
|
||||||
|
|
||||||
|
void HtcsMonitor::ThreadBody() {
|
||||||
|
/* Loop so long as we're not cancelled. */
|
||||||
|
while (!m_cancelled) {
|
||||||
|
/* Open the rpc client. */
|
||||||
|
m_rpc_client->Open();
|
||||||
|
|
||||||
|
/* Ensure we close, if something goes wrong. */
|
||||||
|
auto client_guard = SCOPE_GUARD { m_rpc_client->Close(); };
|
||||||
|
|
||||||
|
/* Wait for the rpc server. */
|
||||||
|
if (m_rpc_client->WaitAny(htclow::ChannelState_Connectable, m_cancel_event.GetBase()) != 0) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Start the rpc client. */
|
||||||
|
if (R_FAILED(m_rpc_client->Start())) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* We're available! */
|
||||||
|
this->SetServiceAvailability(true);
|
||||||
|
client_guard.Cancel();
|
||||||
|
|
||||||
|
/* We're available, so we want to cleanup when we're done. */
|
||||||
|
ON_SCOPE_EXIT {
|
||||||
|
m_rpc_client->Close();
|
||||||
|
m_rpc_client->Cancel();
|
||||||
|
m_rpc_client->Wait();
|
||||||
|
this->SetServiceAvailability(false);
|
||||||
|
};
|
||||||
|
|
||||||
|
/* Wait to become disconnected. */
|
||||||
|
if (m_rpc_client->WaitAny(htclow::ChannelState_Disconnected, m_cancel_event.GetBase()) != 0) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
73
libraries/libstratosphere/source/htcs/impl/htcs_monitor.hpp
Normal file
73
libraries/libstratosphere/source/htcs/impl/htcs_monitor.hpp
Normal file
|
@ -0,0 +1,73 @@
|
||||||
|
/*
|
||||||
|
* 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 <stratosphere.hpp>
|
||||||
|
#include "../../htc/server/driver/htc_i_driver.hpp"
|
||||||
|
#include "../../htc/server/rpc/htc_rpc_client.hpp"
|
||||||
|
#include "htcs_service.hpp"
|
||||||
|
|
||||||
|
namespace ams::htcs::impl {
|
||||||
|
|
||||||
|
class HtcsMonitor {
|
||||||
|
private:
|
||||||
|
mem::StandardAllocator *m_allocator;
|
||||||
|
htc::server::driver::IDriver *m_driver;
|
||||||
|
htc::server::rpc::RpcClient *m_rpc_client;
|
||||||
|
HtcsService *m_service;
|
||||||
|
void *m_monitor_thread_stack;
|
||||||
|
os::ThreadType m_monitor_thread;
|
||||||
|
os::SdkMutex m_mutex;
|
||||||
|
os::Event m_cancel_event;
|
||||||
|
os::Event m_service_availability_event;
|
||||||
|
bool m_cancelled;
|
||||||
|
bool m_is_service_available;
|
||||||
|
private:
|
||||||
|
static void ThreadEntry(void *arg) {
|
||||||
|
static_cast<HtcsMonitor *>(arg)->ThreadBody();
|
||||||
|
}
|
||||||
|
|
||||||
|
void ThreadBody();
|
||||||
|
public:
|
||||||
|
HtcsMonitor(mem::StandardAllocator *allocator, htc::server::driver::IDriver *drv, htc::server::rpc::RpcClient *rc, HtcsService *srv);
|
||||||
|
~HtcsMonitor();
|
||||||
|
public:
|
||||||
|
void Start();
|
||||||
|
void Cancel();
|
||||||
|
void Wait();
|
||||||
|
|
||||||
|
os::EventType *GetServiceAvailabilityEvent() { return m_service_availability_event.GetBase(); }
|
||||||
|
|
||||||
|
bool IsServiceAvailable() {
|
||||||
|
/* Lock ourselves. */
|
||||||
|
std::scoped_lock lk(m_mutex);
|
||||||
|
|
||||||
|
/* Get availability. */
|
||||||
|
return m_is_service_available;
|
||||||
|
}
|
||||||
|
private:
|
||||||
|
void SetServiceAvailability(bool available) {
|
||||||
|
/* Lock ourselves. */
|
||||||
|
std::scoped_lock lk(m_mutex);
|
||||||
|
|
||||||
|
/* Set availability. */
|
||||||
|
m_is_service_available = available;
|
||||||
|
|
||||||
|
/* Signal availability change. */
|
||||||
|
m_service_availability_event.Signal();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
}
|
37
libraries/libstratosphere/source/htcs/impl/htcs_service.hpp
Normal file
37
libraries/libstratosphere/source/htcs/impl/htcs_service.hpp
Normal file
|
@ -0,0 +1,37 @@
|
||||||
|
/*
|
||||||
|
* 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 <stratosphere.hpp>
|
||||||
|
#include "../../htc/server/driver/htc_i_driver.hpp"
|
||||||
|
#include "../../htc/server/rpc/htc_rpc_client.hpp"
|
||||||
|
#include "rpc/htcs_data_channel_manager.hpp"
|
||||||
|
|
||||||
|
namespace ams::htcs::impl {
|
||||||
|
|
||||||
|
class HtcsService {
|
||||||
|
private:
|
||||||
|
mem::StandardAllocator *m_allocator;
|
||||||
|
htc::server::driver::IDriver *m_driver;
|
||||||
|
htc::server::rpc::RpcClient *m_rpc_client;
|
||||||
|
rpc::DataChannelManager *m_data_channel_manager;
|
||||||
|
public:
|
||||||
|
HtcsService(mem::StandardAllocator *allocator, htc::server::driver::IDriver *drv, htc::server::rpc::RpcClient *rc, rpc::DataChannelManager *dcm)
|
||||||
|
: m_allocator(allocator), m_driver(drv), m_rpc_client(rc), m_data_channel_manager(dcm) { /* ... */ }
|
||||||
|
public:
|
||||||
|
/* TODO */
|
||||||
|
};
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,34 @@
|
||||||
|
/*
|
||||||
|
* 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 <stratosphere.hpp>
|
||||||
|
#include "../../../htclow/htclow_manager.hpp"
|
||||||
|
#include "../../../htc/server/rpc/htc_rpc_client.hpp"
|
||||||
|
|
||||||
|
namespace ams::htcs::impl::rpc {
|
||||||
|
|
||||||
|
class DataChannelManager {
|
||||||
|
private:
|
||||||
|
htc::server::rpc::RpcClient* m_rpc_client;
|
||||||
|
htclow::HtclowManager *m_htclow_manager;
|
||||||
|
htclow::Module m_module;
|
||||||
|
public:
|
||||||
|
DataChannelManager(htc::server::rpc::RpcClient *client, htclow::HtclowManager *htclow_manager) : m_rpc_client(client), m_htclow_manager(htclow_manager), m_module(htclow::ModuleId::Htcs) { /* ... */ }
|
||||||
|
public:
|
||||||
|
/* TODO */
|
||||||
|
};
|
||||||
|
|
||||||
|
}
|
Loading…
Reference in a new issue