mirror of
https://github.com/Atmosphere-NX/Atmosphere
synced 2024-12-23 12:51:13 +00:00
102 lines
3.4 KiB
C++
102 lines
3.4 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/>.
|
|
*/
|
|
#pragma once
|
|
#include <stratosphere.hpp>
|
|
#include "htclow_mux_task_manager.hpp"
|
|
#include "htclow_mux_send_buffer.hpp"
|
|
|
|
namespace ams::htclow {
|
|
|
|
class PacketFactory;
|
|
|
|
namespace ctrl {
|
|
|
|
class HtcctrlStateMachine;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
namespace ams::htclow::mux {
|
|
|
|
class ChannelImpl {
|
|
private:
|
|
impl::ChannelInternalType m_channel;
|
|
PacketFactory *m_packet_factory;
|
|
ctrl::HtcctrlStateMachine *m_state_machine;
|
|
TaskManager *m_task_manager;
|
|
os::Event *m_event;
|
|
SendBuffer m_send_buffer;
|
|
RingBuffer m_receive_buffer;
|
|
s16 m_version;
|
|
ChannelConfig m_config;
|
|
u64 m_offset;
|
|
u64 m_total_send_size;
|
|
u64 m_cur_max_data;
|
|
u64 m_prev_max_data;
|
|
util::optional<u64> m_share;
|
|
os::Event m_state_change_event;
|
|
ChannelState m_state;
|
|
public:
|
|
ChannelImpl(impl::ChannelInternalType channel, PacketFactory *pf, ctrl::HtcctrlStateMachine *sm, TaskManager *tm, os::Event *ev);
|
|
|
|
void SetVersion(s16 version);
|
|
|
|
ChannelState GetChannelState() { return m_state; }
|
|
os::EventType *GetChannelStateEvent() { return m_state_change_event.GetBase(); }
|
|
|
|
Result ProcessReceivePacket(const PacketHeader &header, const void *body, size_t body_size);
|
|
|
|
bool QuerySendPacket(PacketHeader *header, PacketBody *body, int *out_body_size);
|
|
|
|
void RemovePacket(const PacketHeader &header);
|
|
|
|
void ShutdownForce();
|
|
|
|
void UpdateState();
|
|
public:
|
|
Result DoConnectBegin(u32 *out_task_id);
|
|
Result DoConnectEnd();
|
|
|
|
Result DoFlush(u32 *out_task_id);
|
|
|
|
Result DoReceiveBegin(u32 *out_task_id, size_t size);
|
|
Result DoReceiveEnd(size_t *out, void *dst, size_t dst_size);
|
|
|
|
Result DoSend(u32 *out_task_id, size_t *out, const void *src, size_t src_size);
|
|
|
|
Result DoShutdown();
|
|
|
|
void SetConfig(const ChannelConfig &config);
|
|
|
|
void SetSendBuffer(void *buf, size_t buf_size, size_t max_packet_size);
|
|
void SetReceiveBuffer(void *buf, size_t buf_size);
|
|
void SetSendBufferWithData(const void *buf, size_t buf_size, size_t max_packet_size);
|
|
private:
|
|
void SetState(ChannelState state);
|
|
void SetStateWithoutCheck(ChannelState state);
|
|
|
|
void SignalSendPacketEvent();
|
|
|
|
Result CheckState(std::initializer_list<ChannelState> states) const;
|
|
Result CheckPacketVersion(s16 version) const;
|
|
|
|
Result ProcessReceiveDataPacket(s16 version, u64 share, u32 offset, const void *body, size_t body_size);
|
|
Result ProcessReceiveMaxDataPacket(s16 version, u64 share);
|
|
Result ProcessReceiveErrorPacket();
|
|
};
|
|
|
|
}
|