mirror of
https://github.com/Atmosphere-NX/Atmosphere
synced 2024-12-26 22:51:12 +00:00
473 lines
16 KiB
C++
473 lines
16 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/>.
|
|
*/
|
|
#include <stratosphere.hpp>
|
|
#include "htclow_mux_channel_impl.hpp"
|
|
#include "../ctrl/htclow_ctrl_state_machine.hpp"
|
|
#include "../htclow_default_channel_config.hpp"
|
|
#include "../htclow_packet_factory.hpp"
|
|
|
|
namespace ams::htclow::mux {
|
|
|
|
ChannelImpl::ChannelImpl(impl::ChannelInternalType channel, PacketFactory *pf, ctrl::HtcctrlStateMachine *sm, TaskManager *tm, os::Event *ev)
|
|
: m_channel(channel), m_packet_factory(pf), m_state_machine(sm), m_task_manager(tm), m_event(ev),
|
|
m_send_buffer(m_channel, pf), m_receive_buffer(), m_version(ProtocolVersion), m_config(DefaultChannelConfig),
|
|
m_offset(0), m_total_send_size(0), m_cur_max_data(0), m_prev_max_data(0), m_share(),
|
|
m_state_change_event(os::EventClearMode_ManualClear), m_state(ChannelState_Unconnectable)
|
|
|
|
{
|
|
this->UpdateState();
|
|
}
|
|
|
|
void ChannelImpl::SetVersion(s16 version) {
|
|
/* Sanity check the version. */
|
|
AMS_ASSERT(version <= ProtocolVersion);
|
|
|
|
/* Set version. */
|
|
m_version = version;
|
|
m_send_buffer.SetVersion(version);
|
|
}
|
|
|
|
Result ChannelImpl::CheckState(std::initializer_list<ChannelState> states) const {
|
|
/* Determine if we have a matching state. */
|
|
bool match = false;
|
|
for (const auto &state : states) {
|
|
match |= m_state == state;
|
|
}
|
|
|
|
/* If we do, we're good. */
|
|
R_SUCCEED_IF(match);
|
|
|
|
/* Otherwise, return appropriate failure error. */
|
|
if (m_state == ChannelState_Disconnected) {
|
|
return htclow::ResultInvalidChannelStateDisconnected();
|
|
} else {
|
|
return htclow::ResultInvalidChannelState();
|
|
}
|
|
}
|
|
|
|
Result ChannelImpl::CheckPacketVersion(s16 version) const {
|
|
R_UNLESS(version == m_version, htclow::ResultChannelVersionNotMatched());
|
|
return ResultSuccess();
|
|
}
|
|
|
|
|
|
Result ChannelImpl::ProcessReceivePacket(const PacketHeader &header, const void *body, size_t body_size) {
|
|
switch (header.packet_type) {
|
|
case PacketType_Data:
|
|
return this->ProcessReceiveDataPacket(header.version, header.share, header.offset, body, body_size);
|
|
case PacketType_MaxData:
|
|
return this->ProcessReceiveMaxDataPacket(header.version, header.share);
|
|
case PacketType_Error:
|
|
return this->ProcessReceiveErrorPacket();
|
|
default:
|
|
return htclow::ResultProtocolError();
|
|
}
|
|
}
|
|
|
|
Result ChannelImpl::ProcessReceiveDataPacket(s16 version, u64 share, u32 offset, const void *body, size_t body_size) {
|
|
/* Check our state. */
|
|
R_TRY(this->CheckState({ChannelState_Connectable, ChannelState_Connected}));
|
|
|
|
/* Check the packet version. */
|
|
R_TRY(this->CheckPacketVersion(version));
|
|
|
|
/* Check that offset matches. */
|
|
R_UNLESS(offset == static_cast<u32>(m_offset), htclow::ResultProtocolError());
|
|
|
|
/* Check for flow control, if we should. */
|
|
if (m_config.flow_control_enabled) {
|
|
/* Check that the share increases monotonically. */
|
|
if (m_share.has_value()) {
|
|
R_UNLESS(m_share.value() <= share, htclow::ResultProtocolError());
|
|
}
|
|
|
|
/* Update our share. */
|
|
m_share = share;
|
|
|
|
/* Signal our event. */
|
|
this->SignalSendPacketEvent();
|
|
}
|
|
|
|
/* Update our offset. */
|
|
m_offset += body_size;
|
|
|
|
/* Write the packet body. */
|
|
R_ABORT_UNLESS(m_receive_buffer.Write(body, body_size));
|
|
|
|
/* Notify the data was received. */
|
|
m_task_manager->NotifyReceiveData(m_channel, m_receive_buffer.GetDataSize());
|
|
|
|
return ResultSuccess();
|
|
}
|
|
|
|
Result ChannelImpl::ProcessReceiveMaxDataPacket(s16 version, u64 share) {
|
|
/* Check our state. */
|
|
R_TRY(this->CheckState({ChannelState_Connectable, ChannelState_Connected}));
|
|
|
|
/* Check the packet version. */
|
|
R_TRY(this->CheckPacketVersion(version));
|
|
|
|
/* Check for flow control, if we should. */
|
|
if (m_config.flow_control_enabled) {
|
|
/* Check that the share increases monotonically. */
|
|
if (m_share.has_value()) {
|
|
R_UNLESS(m_share.value() <= share, htclow::ResultProtocolError());
|
|
}
|
|
|
|
/* Update our share. */
|
|
m_share = share;
|
|
|
|
/* Signal our event. */
|
|
this->SignalSendPacketEvent();
|
|
}
|
|
|
|
return ResultSuccess();
|
|
}
|
|
|
|
Result ChannelImpl::ProcessReceiveErrorPacket() {
|
|
if (m_state == ChannelState_Connected || m_state == ChannelState_Disconnected) {
|
|
this->ShutdownForce();
|
|
}
|
|
return ResultSuccess();
|
|
}
|
|
|
|
bool ChannelImpl::QuerySendPacket(PacketHeader *header, PacketBody *body, int *out_body_size) {
|
|
/* Check our send buffer. */
|
|
if (m_send_buffer.QueryNextPacket(header, body, out_body_size, m_cur_max_data, m_total_send_size, m_share.has_value(), m_share.value_or(0))) {
|
|
/* Update tracking variables. */
|
|
if (header->packet_type == PacketType_Data) {
|
|
m_prev_max_data = m_cur_max_data;
|
|
}
|
|
|
|
return true;
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
void ChannelImpl::RemovePacket(const PacketHeader &header) {
|
|
/* Remove the packet. */
|
|
m_send_buffer.RemovePacket(header);
|
|
|
|
/* Check if the send buffer is now empty. */
|
|
if (m_send_buffer.Empty()) {
|
|
m_task_manager->NotifySendBufferEmpty(m_channel);
|
|
}
|
|
}
|
|
|
|
void ChannelImpl::UpdateState() {
|
|
/* Check if shutdown must be forced. */
|
|
if (m_state_machine->IsUnsupportedServiceChannelToShutdown(m_channel)) {
|
|
this->ShutdownForce();
|
|
}
|
|
|
|
/* Check if we're readied. */
|
|
if (m_state_machine->IsReadied()) {
|
|
m_task_manager->NotifyConnectReady();
|
|
}
|
|
|
|
/* Update our state transition. */
|
|
if (m_state_machine->IsConnectable(m_channel)) {
|
|
if (m_state == ChannelState_Unconnectable) {
|
|
this->SetState(ChannelState_Connectable);
|
|
}
|
|
} else if (m_state_machine->IsUnconnectable()) {
|
|
if (m_state == ChannelState_Connectable) {
|
|
this->SetState(ChannelState_Unconnectable);
|
|
m_state_machine->SetNotConnecting(m_channel);
|
|
} else if (m_state == ChannelState_Connected) {
|
|
this->ShutdownForce();
|
|
}
|
|
}
|
|
}
|
|
|
|
void ChannelImpl::ShutdownForce() {
|
|
/* Clear our send buffer. */
|
|
m_send_buffer.Clear();
|
|
|
|
/* Set our state to shutdown. */
|
|
this->SetState(ChannelState_Disconnected);
|
|
}
|
|
|
|
void ChannelImpl::SetState(ChannelState state) {
|
|
/* Check that we can perform the transition. */
|
|
AMS_ABORT_UNLESS(IsStateTransitionAllowed(m_state, state));
|
|
|
|
/* Perform the transition. */
|
|
this->SetStateWithoutCheck(state);
|
|
}
|
|
|
|
void ChannelImpl::SetStateWithoutCheck(ChannelState state) {
|
|
/* Change our state. */
|
|
if (m_state != state) {
|
|
m_state = state;
|
|
m_state_change_event.Signal();
|
|
}
|
|
|
|
/* If relevant, notify disconnect. */
|
|
if (m_state == ChannelState_Disconnected) {
|
|
m_task_manager->NotifyDisconnect(m_channel);
|
|
}
|
|
}
|
|
|
|
void ChannelImpl::SignalSendPacketEvent() {
|
|
if (m_event != nullptr) {
|
|
m_event->Signal();
|
|
}
|
|
}
|
|
|
|
Result ChannelImpl::DoConnectBegin(u32 *out_task_id) {
|
|
/* Check our state. */
|
|
R_TRY(this->CheckState({ChannelState_Connectable}));
|
|
|
|
/* Set ourselves as connecting. */
|
|
m_state_machine->SetConnecting(m_channel);
|
|
|
|
/* Allocate a task. */
|
|
u32 task_id{};
|
|
R_TRY(m_task_manager->AllocateTask(std::addressof(task_id), m_channel));
|
|
|
|
/* Configure the task. */
|
|
m_task_manager->ConfigureConnectTask(task_id);
|
|
|
|
/* If we're ready, complete the task immediately. */
|
|
if (m_state_machine->IsReadied()) {
|
|
m_task_manager->CompleteTask(task_id, EventTrigger_ConnectReady);
|
|
}
|
|
|
|
/* Set the output task id. */
|
|
*out_task_id = task_id;
|
|
return ResultSuccess();
|
|
}
|
|
|
|
Result ChannelImpl::DoConnectEnd() {
|
|
/* Check our state. */
|
|
R_TRY(this->CheckState({ChannelState_Connectable}));
|
|
|
|
/* Perform handshake, if we should. */
|
|
if (m_config.handshake_enabled) {
|
|
/* Set our current max data. */
|
|
m_cur_max_data = m_receive_buffer.GetBufferSize();
|
|
|
|
/* Make a max data packet. */
|
|
auto packet = m_packet_factory->MakeMaxDataPacket(m_channel, m_version, m_cur_max_data);
|
|
R_UNLESS(packet, htclow::ResultOutOfMemory());
|
|
|
|
/* Send the packet. */
|
|
m_send_buffer.AddPacket(std::move(packet));
|
|
|
|
/* Signal that we have an packet to send. */
|
|
this->SignalSendPacketEvent();
|
|
|
|
/* Set our prev max data. */
|
|
m_prev_max_data = m_cur_max_data;
|
|
} else {
|
|
/* Set our share. */
|
|
m_share = m_config.initial_counter_max_data;
|
|
|
|
/* If we're not empty, signal. */
|
|
if (!m_send_buffer.Empty()) {
|
|
this->SignalSendPacketEvent();
|
|
}
|
|
}
|
|
|
|
/* Set our state as connected. */
|
|
this->SetState(ChannelState_Connected);
|
|
|
|
return ResultSuccess();
|
|
}
|
|
|
|
Result ChannelImpl::DoFlush(u32 *out_task_id) {
|
|
/* Check our state. */
|
|
R_TRY(this->CheckState({ChannelState_Connected}));
|
|
|
|
/* Allocate a task. */
|
|
u32 task_id{};
|
|
R_TRY(m_task_manager->AllocateTask(std::addressof(task_id), m_channel));
|
|
|
|
/* Configure the task. */
|
|
m_task_manager->ConfigureFlushTask(task_id);
|
|
|
|
/* If we're already flushed, complete the task immediately. */
|
|
if (m_send_buffer.Empty()) {
|
|
m_task_manager->CompleteTask(task_id, EventTrigger_SendBufferEmpty);
|
|
}
|
|
|
|
/* Set the output task id. */
|
|
*out_task_id = task_id;
|
|
return ResultSuccess();
|
|
}
|
|
|
|
Result ChannelImpl::DoReceiveBegin(u32 *out_task_id, size_t size) {
|
|
/* Check our state. */
|
|
R_TRY(this->CheckState({ChannelState_Connected, ChannelState_Disconnected}));
|
|
|
|
/* Allocate a task. */
|
|
u32 task_id{};
|
|
R_TRY(m_task_manager->AllocateTask(std::addressof(task_id), m_channel));
|
|
|
|
/* Configure the task. */
|
|
m_task_manager->ConfigureReceiveTask(task_id, size);
|
|
|
|
/* Check if the task is already complete. */
|
|
if (m_receive_buffer.GetDataSize() >= size) {
|
|
m_task_manager->CompleteTask(task_id, EventTrigger_ReceiveData);
|
|
} else if (m_state == ChannelState_Disconnected) {
|
|
m_task_manager->CompleteTask(task_id, EventTrigger_Disconnect);
|
|
}
|
|
|
|
/* Set the output task id. */
|
|
*out_task_id = task_id;
|
|
return ResultSuccess();
|
|
|
|
}
|
|
Result ChannelImpl::DoReceiveEnd(size_t *out, void *dst, size_t dst_size) {
|
|
/* Check our state. */
|
|
R_TRY(this->CheckState({ChannelState_Connected, ChannelState_Disconnected}));
|
|
|
|
/* If we have nowhere to receive, we're done. */
|
|
if (dst_size == 0) {
|
|
*out = 0;
|
|
return ResultSuccess();
|
|
}
|
|
|
|
/* Get the amount of receivable data. */
|
|
const size_t receivable = m_receive_buffer.GetDataSize();
|
|
const size_t received = std::min(dst_size, receivable);
|
|
|
|
/* Read the data. */
|
|
R_ABORT_UNLESS(m_receive_buffer.Read(dst, received));
|
|
|
|
/* Handle flow control, if we should. */
|
|
if (m_config.flow_control_enabled) {
|
|
/* Read our fields. */
|
|
const auto prev_max_data = m_prev_max_data;
|
|
const auto next_max_data = m_cur_max_data + received;
|
|
const auto max_packet_size = m_config.max_packet_size;
|
|
const auto offset = m_offset;
|
|
|
|
/* Update our current max data. */
|
|
m_cur_max_data = next_max_data;
|
|
|
|
/* If we can, send a max data packet. */
|
|
if (prev_max_data - offset < max_packet_size + sizeof(PacketHeader)) {
|
|
/* Make a max data packet. */
|
|
auto packet = m_packet_factory->MakeMaxDataPacket(m_channel, m_version, next_max_data);
|
|
R_UNLESS(packet, htclow::ResultOutOfMemory());
|
|
|
|
/* Send the packet. */
|
|
m_send_buffer.AddPacket(std::move(packet));
|
|
|
|
/* Signal that we have an packet to send. */
|
|
this->SignalSendPacketEvent();
|
|
|
|
/* Set our prev max data. */
|
|
m_prev_max_data = m_cur_max_data;
|
|
}
|
|
}
|
|
|
|
/* Set the output size. */
|
|
*out = received;
|
|
return ResultSuccess();
|
|
}
|
|
|
|
Result ChannelImpl::DoSend(u32 *out_task_id, size_t *out, const void *src, size_t src_size) {
|
|
/* Check our state. */
|
|
R_TRY(this->CheckState({ChannelState_Connected}));
|
|
|
|
/* Allocate a task. */
|
|
u32 task_id{};
|
|
R_TRY(m_task_manager->AllocateTask(std::addressof(task_id), m_channel));
|
|
|
|
/* Send the data. */
|
|
const size_t sent = m_send_buffer.AddData(src, src_size);
|
|
|
|
/* Add the size to our total. */
|
|
m_total_send_size += sent;
|
|
|
|
/* Signal our event. */
|
|
this->SignalSendPacketEvent();
|
|
|
|
/* Configure the task. */
|
|
m_task_manager->ConfigureSendTask(task_id);
|
|
|
|
/* If we sent all the data, we're done. */
|
|
if (sent == src_size) {
|
|
m_task_manager->CompleteTask(task_id, EventTrigger_SendComplete);
|
|
}
|
|
|
|
/* Set the output. */
|
|
*out_task_id = task_id;
|
|
*out = sent;
|
|
|
|
return ResultSuccess();
|
|
}
|
|
|
|
Result ChannelImpl::DoShutdown() {
|
|
/* Check our state. */
|
|
R_TRY(this->CheckState({ChannelState_Connected}));
|
|
|
|
/* Set our state. */
|
|
this->SetState(ChannelState_Disconnected);
|
|
return ResultSuccess();
|
|
}
|
|
|
|
void ChannelImpl::SetConfig(const ChannelConfig &config) {
|
|
/* Check our state. */
|
|
R_ABORT_UNLESS(this->CheckState({ChannelState_Unconnectable, ChannelState_Connectable}));
|
|
|
|
/* Set our config. */
|
|
m_config = config;
|
|
|
|
/* Set flow control for our send buffer. */
|
|
m_send_buffer.SetFlowControlEnabled(m_config.flow_control_enabled);
|
|
}
|
|
|
|
void ChannelImpl::SetSendBuffer(void *buf, size_t buf_size, size_t max_packet_size) {
|
|
/* Set buffer. */
|
|
m_send_buffer.SetBuffer(buf, buf_size);
|
|
|
|
/* Determine true max packet size. */
|
|
if (m_config.flow_control_enabled) {
|
|
max_packet_size = std::min(max_packet_size, m_config.max_packet_size);
|
|
}
|
|
|
|
/* Set max packet size. */
|
|
m_send_buffer.SetMaxPacketSize(max_packet_size);
|
|
}
|
|
|
|
void ChannelImpl::SetReceiveBuffer(void *buf, size_t buf_size) {
|
|
/* Set the buffer. */
|
|
m_receive_buffer.Initialize(buf, buf_size);
|
|
}
|
|
|
|
void ChannelImpl::SetSendBufferWithData(const void *buf, size_t buf_size, size_t max_packet_size) {
|
|
/* Set buffer. */
|
|
m_send_buffer.SetReadOnlyBuffer(buf, buf_size);
|
|
|
|
/* Determine true max packet size. */
|
|
if (m_config.flow_control_enabled) {
|
|
max_packet_size = std::min(max_packet_size, m_config.max_packet_size);
|
|
}
|
|
|
|
/* Set max packet size. */
|
|
m_send_buffer.SetMaxPacketSize(max_packet_size);
|
|
|
|
/* Set our total send size. */
|
|
m_total_send_size = buf_size;
|
|
}
|
|
|
|
}
|