mirror of
https://github.com/Atmosphere-NX/Atmosphere
synced 2024-12-22 20:31:14 +00:00
htcs: declare all rpc tasks
This commit is contained in:
Michael Scire
SciresM
parent
536e3e99a8
commit
f71943c03a
2 changed files with 328 additions and 5 deletions
|
|
@ -28,14 +28,16 @@ namespace ams::htc::server::rpc {
|
|||
struct RpcTaskFunctionTraits {
|
||||
public:
|
||||
template<typename R, typename C, typename... A>
|
||||
static std::tuple<A...> GetArgsImpl(R(C::*)(A...));
|
||||
static std::tuple<A...> GetSetArgumentsImpl(R(C::*)(A...));
|
||||
template<typename R, typename C, typename... A>
|
||||
static std::tuple<A...> GetGetResultImpl(R(C::*)(A...) const);
|
||||
};
|
||||
|
||||
template<typename T> requires IsRpcTask<T>
|
||||
using RpcTaskArgumentsType = decltype(RpcTaskFunctionTraits::GetArgsImpl(&T::SetArguments));
|
||||
using RpcTaskArgumentsType = decltype(RpcTaskFunctionTraits::GetSetArgumentsImpl(&T::SetArguments));
|
||||
|
||||
template<typename T> requires IsRpcTask<T>
|
||||
using RpcTaskResultType = decltype(RpcTaskFunctionTraits::GetArgsImpl(&T::GetResult));
|
||||
using RpcTaskResultType = decltype(RpcTaskFunctionTraits::GetGetResultImpl(&T::GetResult));
|
||||
|
||||
template<typename T, typename... Args>
|
||||
concept IsRpcTaskArgumentsType = IsRpcTask<T> && std::same_as<std::tuple<Args...>, RpcTaskArgumentsType<T>>;
|
||||
|
|
|
|||
|
|
@ -48,16 +48,337 @@ namespace ams::htcs::impl::rpc {
|
|||
s16 GetVersion() const { return m_version; }
|
||||
};
|
||||
|
||||
class HtcsSignalingTask : public HtcsTask {
|
||||
private:
|
||||
os::SystemEventType m_system_event;
|
||||
bool m_is_valid;
|
||||
public:
|
||||
HtcsSignalingTask(HtcsTaskType type);
|
||||
virtual ~HtcsSignalingTask();
|
||||
|
||||
bool IsValid() const { return m_is_valid; }
|
||||
|
||||
void Complete() {
|
||||
os::SignalSystemEvent(std::addressof(m_system_event));
|
||||
HtcsTask::Complete();
|
||||
}
|
||||
public:
|
||||
virtual void Cancel(htc::server::rpc::RpcTaskCancelReason reason) override {
|
||||
HtcsTask::Cancel(reason);
|
||||
os::SignalSystemEvent(std::addressof(m_system_event));
|
||||
}
|
||||
|
||||
virtual os::SystemEventType *GetSystemEvent() override { return std::addressof(m_system_event); }
|
||||
};
|
||||
|
||||
class ReceiveTask : public HtcsSignalingTask {
|
||||
private:
|
||||
s32 m_handle;
|
||||
s64 m_size;
|
||||
htcs::MessageFlag m_flags;
|
||||
void *m_buffer;
|
||||
s64 m_buffer_size;
|
||||
htcs::SocketError m_err;
|
||||
s64 m_result_size;
|
||||
public:
|
||||
ReceiveTask() : HtcsSignalingTask(HtcsTaskType::Receive) { /* ... */ }
|
||||
|
||||
s32 GetHandle() const { return m_handle; }
|
||||
s64 GetSize() const { return m_size; }
|
||||
htcs::MessageFlag GetFlags() const { return m_flags; }
|
||||
void *GetBuffer() const { return m_buffer; }
|
||||
s64 GetBufferSize() const { return m_buffer_size; }
|
||||
|
||||
s64 GetResultSize() const {
|
||||
AMS_ASSERT(this->GetTaskState() == htc::server::rpc::RpcTaskState::Completed);
|
||||
return m_result_size;
|
||||
}
|
||||
public:
|
||||
Result SetArguments(s32 handle, s64 size, htcs::MessageFlag flags);
|
||||
void Complete(htcs::SocketError err, s64 size);
|
||||
Result GetResult(htcs::SocketError *out_err, s64 *out_size) const;
|
||||
public:
|
||||
virtual Result ProcessResponse(const char *data, size_t size) override;
|
||||
virtual Result CreateRequest(size_t *out, char *data, size_t size, u32 task_id) override;
|
||||
virtual Result CreateNotification(size_t *out, char *data, size_t size, u32 task_id) override;
|
||||
};
|
||||
|
||||
class SendTask : public HtcsSignalingTask {
|
||||
private:
|
||||
os::Event m_ready_event;
|
||||
s32 m_handle;
|
||||
s64 m_size;
|
||||
htcs::MessageFlag m_flags;
|
||||
void *m_buffer;
|
||||
s64 m_buffer_size;
|
||||
htcs::SocketError m_err;
|
||||
s64 m_result_size;
|
||||
public:
|
||||
SendTask() : HtcsSignalingTask(HtcsTaskType::Send), m_ready_event(os::EventClearMode_ManualClear) { /* ... */ }
|
||||
|
||||
s32 GetHandle() const { return m_handle; }
|
||||
s64 GetSize() const { return m_size; }
|
||||
htcs::MessageFlag GetFlags() const { return m_flags; }
|
||||
void *GetBuffer() const { return m_buffer; }
|
||||
s64 GetBufferSize() const { return m_buffer_size; }
|
||||
|
||||
void SetBuffer(const void *buffer, s64 buffer_size);
|
||||
void NotifyDataChannelReady();
|
||||
void WaitNotification();
|
||||
public:
|
||||
Result SetArguments(s32 handle, s64 size, htcs::MessageFlag flags);
|
||||
void Complete(htcs::SocketError err, s64 size);
|
||||
Result GetResult(htcs::SocketError *out_err, s64 *out_size) const;
|
||||
public:
|
||||
virtual void Cancel(htc::server::rpc::RpcTaskCancelReason reason) override;
|
||||
virtual Result ProcessResponse(const char *data, size_t size) override;
|
||||
virtual Result CreateRequest(size_t *out, char *data, size_t size, u32 task_id) override;
|
||||
virtual Result ProcessNotification(const char *data, size_t size) override;
|
||||
virtual Result CreateNotification(size_t *out, char *data, size_t size, u32 task_id) override;
|
||||
};
|
||||
|
||||
class ShutdownTask : public HtcsTask {
|
||||
private:
|
||||
s32 m_handle;
|
||||
ShutdownType m_how;
|
||||
htcs::SocketError m_err;
|
||||
public:
|
||||
ShutdownTask() : HtcsTask(HtcsTaskType::Shutdown) { /* ... */ }
|
||||
|
||||
s32 GetHandle() const { return m_handle; }
|
||||
ShutdownType GetHow() const { return m_how; }
|
||||
public:
|
||||
Result SetArguments(s32 handle, ShutdownType how);
|
||||
void Complete(htcs::SocketError err);
|
||||
Result GetResult(htcs::SocketError *out_err) const;
|
||||
public:
|
||||
virtual Result ProcessResponse(const char *data, size_t size) override;
|
||||
virtual Result CreateRequest(size_t *out, char *data, size_t size, u32 task_id) override;
|
||||
};
|
||||
|
||||
class CloseTask : public HtcsTask {
|
||||
private:
|
||||
s32 m_handle;
|
||||
htcs::SocketError m_err;
|
||||
public:
|
||||
CloseTask() : HtcsTask(HtcsTaskType::Close) { /* ... */ }
|
||||
|
||||
s32 GetHandle() const { return m_handle; }
|
||||
public:
|
||||
Result SetArguments(s32 handle);
|
||||
void Complete(htcs::SocketError err);
|
||||
Result GetResult(htcs::SocketError *out_err) const;
|
||||
public:
|
||||
virtual Result ProcessResponse(const char *data, size_t size) override;
|
||||
virtual Result CreateRequest(size_t *out, char *data, size_t size, u32 task_id) override;
|
||||
};
|
||||
|
||||
class ConnectTask : public HtcsTask {
|
||||
private:
|
||||
s32 m_handle;
|
||||
HtcsPeerName m_peer_name;
|
||||
HtcsPortName m_port_name;
|
||||
htcs::SocketError m_err;
|
||||
public:
|
||||
ConnectTask() : HtcsTask(HtcsTaskType::Connect) { /* ... */ }
|
||||
|
||||
s32 GetHandle() const { return m_handle; }
|
||||
const HtcsPeerName &GetPeerName() const { return m_peer_name; }
|
||||
const HtcsPortName &GetPortName() const { return m_port_name; }
|
||||
public:
|
||||
Result SetArguments(s32 handle, const HtcsPeerName &peer_name, const HtcsPortName &port_name);
|
||||
void Complete(htcs::SocketError err);
|
||||
Result GetResult(htcs::SocketError *out_err) const;
|
||||
public:
|
||||
virtual Result ProcessResponse(const char *data, size_t size) override;
|
||||
virtual Result CreateRequest(size_t *out, char *data, size_t size, u32 task_id) override;
|
||||
};
|
||||
|
||||
class ListenTask : public HtcsTask {
|
||||
private:
|
||||
s32 m_handle;
|
||||
s32 m_backlog;
|
||||
htcs::SocketError m_err;
|
||||
public:
|
||||
ListenTask() : HtcsTask(HtcsTaskType::Listen) { /* ... */ }
|
||||
|
||||
s32 GetHandle() const { return m_handle; }
|
||||
s32 GetBacklog() const { return m_backlog; }
|
||||
public:
|
||||
Result SetArguments(s32 handle, s32 backlog);
|
||||
void Complete(htcs::SocketError err);
|
||||
Result GetResult(htcs::SocketError *out_err) const;
|
||||
public:
|
||||
virtual Result ProcessResponse(const char *data, size_t size) override;
|
||||
virtual Result CreateRequest(size_t *out, char *data, size_t size, u32 task_id) override;
|
||||
};
|
||||
|
||||
class AcceptTask : public HtcsSignalingTask {
|
||||
private:
|
||||
s32 m_server_handle;
|
||||
htcs::SocketError m_err;
|
||||
s32 m_desc;
|
||||
public:
|
||||
AcceptTask() : HtcsSignalingTask(HtcsTaskType::Accept) { /* ... */ }
|
||||
|
||||
s32 GetServerHandle() const { return m_server_handle; }
|
||||
public:
|
||||
Result SetArguments(s32 server_handle);
|
||||
void Complete(htcs::SocketError err, s32 desc);
|
||||
Result GetResult(htcs::SocketError *out_err, s32 *out_desc, s32 server_handle) const;
|
||||
public:
|
||||
virtual Result ProcessResponse(const char *data, size_t size) override;
|
||||
virtual Result CreateRequest(size_t *out, char *data, size_t size, u32 task_id) override;
|
||||
};
|
||||
|
||||
class SocketTask : public HtcsTask {
|
||||
private:
|
||||
htcs::SocketError m_err;
|
||||
s32 m_desc;
|
||||
public:
|
||||
SocketTask() : HtcsTask(HtcsTaskType::Socket) { /* ... */ }
|
||||
|
||||
public:
|
||||
Result SetArguments();
|
||||
void Complete(htcs::SocketError err, s32 desc);
|
||||
Result GetResult(htcs::SocketError *out_err, s32 *out_desc);
|
||||
Result GetResult(htcs::SocketError *out_err, s32 *out_desc) const;
|
||||
public:
|
||||
virtual Result ProcessResponse(const char *data, size_t size) override;
|
||||
virtual Result CreateRequest(size_t *out, char *data, size_t size, u32 task_id) override;
|
||||
};
|
||||
|
||||
class BindTask : public HtcsTask {
|
||||
private:
|
||||
s32 m_handle;
|
||||
HtcsPeerName m_peer_name;
|
||||
HtcsPortName m_port_name;
|
||||
htcs::SocketError m_err;
|
||||
public:
|
||||
BindTask() : HtcsTask(HtcsTaskType::Bind) { /* ... */ }
|
||||
|
||||
s32 GetHandle() const { return m_handle; }
|
||||
const HtcsPeerName &GetPeerName() const { return m_peer_name; }
|
||||
const HtcsPortName &GetPortName() const { return m_port_name; }
|
||||
public:
|
||||
Result SetArguments(s32 handle, const HtcsPeerName &peer_name, const HtcsPortName &port_name);
|
||||
void Complete(htcs::SocketError err);
|
||||
Result GetResult(htcs::SocketError *out_err) const;
|
||||
public:
|
||||
virtual Result ProcessResponse(const char *data, size_t size) override;
|
||||
virtual Result CreateRequest(size_t *out, char *data, size_t size, u32 task_id) override;
|
||||
};
|
||||
|
||||
class FcntlTask : public HtcsTask {
|
||||
private:
|
||||
s32 m_handle;
|
||||
s32 m_command;
|
||||
s32 m_value;
|
||||
htcs::SocketError m_err;
|
||||
s32 m_res;
|
||||
public:
|
||||
FcntlTask() : HtcsTask(HtcsTaskType::Fcntl) { /* ... */ }
|
||||
|
||||
s32 GetHandle() const { return m_handle; }
|
||||
s32 GetCommand() const { return m_command; }
|
||||
s32 GetValue() const { return m_value; }
|
||||
public:
|
||||
Result SetArguments(s32 handle, s32 command, s32 value);
|
||||
void Complete(htcs::SocketError err);
|
||||
Result GetResult(htcs::SocketError *out_err, s32 *out_res) const;
|
||||
public:
|
||||
virtual Result ProcessResponse(const char *data, size_t size) override;
|
||||
virtual Result CreateRequest(size_t *out, char *data, size_t size, u32 task_id) override;
|
||||
};
|
||||
|
||||
class ReceiveSmallTask : public HtcsSignalingTask {
|
||||
private:
|
||||
s32 m_handle;
|
||||
s64 m_size;
|
||||
htcs::MessageFlag m_flags;
|
||||
char m_buffer[0xE000];
|
||||
htcs::SocketError m_err;
|
||||
s64 m_result_size;
|
||||
public:
|
||||
ReceiveSmallTask() : HtcsSignalingTask(HtcsTaskType::ReceiveSmall) { /* ... */ }
|
||||
|
||||
s32 GetHandle() const { return m_handle; }
|
||||
s64 GetSize() const { return m_size; }
|
||||
htcs::MessageFlag GetFlags() const { return m_flags; }
|
||||
void *GetBuffer() { return m_buffer; }
|
||||
s64 GetBufferSize() const { return static_cast<s64>(sizeof(m_buffer)); }
|
||||
|
||||
s64 GetResultSize() const {
|
||||
AMS_ASSERT(this->GetTaskState() == htc::server::rpc::RpcTaskState::Completed);
|
||||
return m_result_size;
|
||||
}
|
||||
public:
|
||||
Result SetArguments(s32 handle, s64 size, htcs::MessageFlag flags);
|
||||
void Complete(htcs::SocketError err, s64 size);
|
||||
Result GetResult(htcs::SocketError *out_err, s64 *out_size) const;
|
||||
public:
|
||||
virtual Result ProcessResponse(const char *data, size_t size) override;
|
||||
virtual Result CreateRequest(size_t *out, char *data, size_t size, u32 task_id) override;
|
||||
virtual bool IsReceiveBufferRequired() override;
|
||||
};
|
||||
|
||||
class SendSmallTask : public HtcsSignalingTask {
|
||||
private:
|
||||
os::Event m_ready_event;
|
||||
s32 m_handle;
|
||||
s64 m_size;
|
||||
htcs::MessageFlag m_flags;
|
||||
char m_buffer[0xE000];
|
||||
s64 m_buffer_size;
|
||||
htcs::SocketError m_err;
|
||||
s64 m_result_size;
|
||||
public:
|
||||
SendSmallTask() : HtcsSignalingTask(HtcsTaskType::SendSmall), m_ready_event(os::EventClearMode_ManualClear) { /* ... */ }
|
||||
|
||||
s32 GetHandle() const { return m_handle; }
|
||||
s64 GetSize() const { return m_size; }
|
||||
htcs::MessageFlag GetFlags() const { return m_flags; }
|
||||
void *GetBuffer() { return m_buffer; }
|
||||
s64 GetBufferSize() const { return m_buffer_size; }
|
||||
|
||||
void SetBuffer(const void *buffer, s64 buffer_size);
|
||||
void NotifyDataChannelReady();
|
||||
void WaitNotification();
|
||||
public:
|
||||
Result SetArguments(s32 handle, s64 size, htcs::MessageFlag flags);
|
||||
void Complete(htcs::SocketError err, s64 size);
|
||||
Result GetResult(htcs::SocketError *out_err, s64 *out_size) const;
|
||||
public:
|
||||
virtual void Cancel(htc::server::rpc::RpcTaskCancelReason reason) override;
|
||||
virtual Result ProcessResponse(const char *data, size_t size) override;
|
||||
virtual Result CreateRequest(size_t *out, char *data, size_t size, u32 task_id) override;
|
||||
virtual bool IsSendBufferRequired() override;
|
||||
};
|
||||
|
||||
class SelectTask : public HtcsSignalingTask {
|
||||
private:
|
||||
s32 m_handles[SocketCountMax * 3];
|
||||
s32 m_read_handle_count;
|
||||
s32 m_write_handle_count;
|
||||
s32 m_exception_handle_count;
|
||||
s64 m_tv_sec;
|
||||
s64 m_tv_usec;
|
||||
htcs::SocketError m_err;
|
||||
s32 m_out_handles[SocketCountMax * 3];
|
||||
s32 m_out_read_handle_count;
|
||||
s32 m_out_write_handle_count;
|
||||
s32 m_out_exception_handle_count;
|
||||
public:
|
||||
SelectTask() : HtcsSignalingTask(HtcsTaskType::Select) { /* ... */ }
|
||||
|
||||
const s32 *GetHandles() const { return m_handles; }
|
||||
s32 GetReadHandleCount() const { return m_read_handle_count; }
|
||||
s32 GetWriteHandleCount() const { return m_write_handle_count; }
|
||||
s32 GetExceptionHandleCount() const { return m_exception_handle_count; }
|
||||
s64 GetTimeoutSeconds() const { return m_tv_sec; }
|
||||
s64 GetTimeoutMicroSeconds() const { return m_tv_usec; }
|
||||
public:
|
||||
Result SetArguments(Span<const int> read_handles, Span<const int> write_handles, Span<const int> exception_handles, s64 tv_sec, s64 tv_usec);
|
||||
void Complete(htcs::SocketError err, s32 read_handle_count, s32 write_handle_count, s32 exception_handle_count, const void *body, s64 body_size);
|
||||
Result GetResult(htcs::SocketError *out_err, bool *out_empty, Span<int> read_handles, Span<int> write_handles, Span<int> exception_handles) const;
|
||||
public:
|
||||
virtual Result ProcessResponse(const char *data, size_t size) override;
|
||||
virtual Result CreateRequest(size_t *out, char *data, size_t size, u32 task_id) override;
|
||||
|
|
|
|||
Loading…
Reference in a new issue