mirror of
https://github.com/Atmosphere-NX/Atmosphere.git
synced 2025-05-24 03:36:52 -04:00
sm/tipc: minor cleanup
This commit is contained in:
Michael Scire
parent
124a1a1ea0
commit
0189c5f1e6
15 changed files with 394 additions and 599 deletions
|
|
@ -19,91 +19,188 @@
|
|||
|
||||
namespace ams::sm {
|
||||
|
||||
UserService::~UserService() {
|
||||
if (m_initialized) {
|
||||
impl::OnClientDisconnected(m_process_id);
|
||||
namespace {
|
||||
|
||||
constexpr const u8 CmifResponseToQueryPointerBufferSize[] = {
|
||||
0x00, 0x00, 0x00, 0x00, 0x07, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
||||
0x53, 0x46, 0x43, 0x4F, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
||||
0x00, 0x00, 0x00, 0x00,
|
||||
};
|
||||
|
||||
static_assert(CmifCommandType_Request == 4);
|
||||
|
||||
constexpr const u8 CmifExpectedRequestHeaderForRegisterClientAndDetachClient[] = {
|
||||
0x04, 0x00, 0x00, 0x00, 0x0A, 0x00, 0x00, 0x80, 0x01, 0x00, 0x00, 0x00,
|
||||
};
|
||||
|
||||
constexpr const u8 CmifResponseToRegisterClientAndDetachClient[] = {
|
||||
0x00, 0x00, 0x00, 0x00, 0x06, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
||||
0x53, 0x46, 0x43, 0x4F, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
||||
};
|
||||
|
||||
constexpr const u8 CmifExpectedRequestHeaderForGetServiceHandleAndUnregisterService[] = {
|
||||
0x04, 0x00, 0x00, 0x00, 0x0A, 0x00, 0x00, 0x00
|
||||
};
|
||||
|
||||
constexpr const u8 CmifResponseToGetServiceHandleAndRegisterService[] = {
|
||||
0x00, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x80, 0x20, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
||||
0x53, 0x46, 0x43, 0x4F, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
||||
};
|
||||
|
||||
constexpr const u8 CmifResponseToUnregisterService[] = {
|
||||
0x00, 0x00, 0x00, 0x00, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
||||
0x53, 0x46, 0x43, 0x4F, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
||||
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
||||
};
|
||||
|
||||
constexpr const u8 CmifExpectedRequestHeaderForRegisterService[] = {
|
||||
0x04, 0x00, 0x00, 0x00, 0x0C, 0x00, 0x00, 0x00
|
||||
};
|
||||
|
||||
static_assert(tipc::ResultRequestDeferred().GetValue() == 0xC823);
|
||||
constexpr const u8 CmifResponseToForceProcessorDeferral[] = {
|
||||
0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x23, 0xC8, 0x00, 0x00,
|
||||
};
|
||||
|
||||
}
|
||||
|
||||
Result UserService::ProcessDefaultServiceCommand(const svc::ipc::MessageBuffer &message_buffer) {
|
||||
/* Parse the hipc headers. */
|
||||
const svc::ipc::MessageBuffer::MessageHeader message_header(message_buffer);
|
||||
const svc::ipc::MessageBuffer::SpecialHeader special_header(message_buffer, message_header);
|
||||
|
||||
/* Get the request tag. */
|
||||
const auto tag = message_header.GetTag();
|
||||
|
||||
/* Handle the case where we received a control request. */
|
||||
if (tag == CmifCommandType_Control || tag == CmifCommandType_ControlWithContext) {
|
||||
/* The only control/control with context command which we support is QueryPointerBufferSize. */
|
||||
/* We do not have a pointer buffer, and so our pointer buffer size is zero. */
|
||||
/* Return the relevant hardcoded response. */
|
||||
std::memcpy(message_buffer.GetBufferForDebug(), CmifResponseToQueryPointerBufferSize, sizeof(CmifResponseToQueryPointerBufferSize));
|
||||
return ResultSuccess();
|
||||
}
|
||||
|
||||
/* We only support request (with context), from this point forwards. */
|
||||
R_UNLESS((tag == CmifCommandType_Request || tag == CmifCommandType_RequestWithContext), tipc::ResultInvalidMethod());
|
||||
|
||||
/* For ease of parsing, we will alter the tag to be Request when it is RequestWithContext. */
|
||||
if (tag == CmifCommandType_RequestWithContext) {
|
||||
message_buffer.Set(svc::ipc::MessageBuffer::MessageHeader(CmifCommandType_Request,
|
||||
message_header.GetHasSpecialHeader(),
|
||||
message_header.GetPointerCount(),
|
||||
message_header.GetSendCount(),
|
||||
message_header.GetReceiveCount(),
|
||||
message_header.GetExchangeCount(),
|
||||
message_header.GetRawCount(),
|
||||
message_header.GetReceiveListCount()));
|
||||
}
|
||||
|
||||
/* NOTE: Nintendo only supports RegisterClient and GetServiceHandle. However, it would break */
|
||||
/* a substantial amount of homebrew system modules, if we were to not provide shims for some */
|
||||
/* other commands (RegisterService, and UnregisterService, in particular). As such, we will */
|
||||
/* do the nice thing, and accommodate this by providing backwards compatibility shims for */
|
||||
/* those commands. */
|
||||
|
||||
/* Please note, though, that the atmosphere extensions are not shimmed, as performing all the required */
|
||||
/* parsing for that seems unreasonable. Also, if you are using atmosphere extensions, you are probably */
|
||||
/* used to my breaking shit regularly anyway, and I don't expect much of a fuss to be raised by you. */
|
||||
/* I invite you to demonstrate to me that this expectation does not reflect reality. */
|
||||
const u8 * const raw_message_buffer = static_cast<const u8 *>(message_buffer.GetBufferForDebug());
|
||||
u8 * const out_message_buffer = static_cast<u8 *>(message_buffer.GetBufferForDebug());
|
||||
|
||||
if (std::memcmp(raw_message_buffer, CmifExpectedRequestHeaderForRegisterClientAndDetachClient, sizeof(CmifExpectedRequestHeaderForRegisterClientAndDetachClient)) == 0) {
|
||||
/* Get the command id. */
|
||||
const u32 command_id = *reinterpret_cast<const u32 *>(raw_message_buffer + 0x28);
|
||||
|
||||
/* Get the client process id. */
|
||||
u64 client_process_id;
|
||||
std::memcpy(std::addressof(client_process_id), raw_message_buffer + 0xC, sizeof(client_process_id));
|
||||
|
||||
if (command_id == 0) {
|
||||
/* Invoke the handler for RegisterClient. */
|
||||
R_ABORT_UNLESS(this->RegisterClient(tipc::ClientProcessId{ client_process_id }));
|
||||
} else if (command_id == 4) {
|
||||
/* Invoke the handler for DetachClient. */
|
||||
R_ABORT_UNLESS(this->DetachClient(tipc::ClientProcessId{ client_process_id }));
|
||||
} else {
|
||||
return tipc::ResultInvalidMethod();
|
||||
}
|
||||
|
||||
/* Serialize output. */
|
||||
std::memcpy(out_message_buffer, CmifResponseToRegisterClientAndDetachClient, sizeof(CmifResponseToRegisterClientAndDetachClient));
|
||||
} else if (std::memcmp(raw_message_buffer, CmifExpectedRequestHeaderForGetServiceHandleAndUnregisterService, sizeof(CmifExpectedRequestHeaderForGetServiceHandleAndUnregisterService)) == 0) {
|
||||
/* Get the command id. */
|
||||
const u32 command_id = *reinterpret_cast<const u32 *>(raw_message_buffer + 0x18);
|
||||
|
||||
/* Get the service_name. */
|
||||
sm::ServiceName service_name;
|
||||
std::memcpy(std::addressof(service_name), raw_message_buffer + 0x20, sizeof(service_name));
|
||||
|
||||
if (command_id == 1) {
|
||||
/* Invoke the handler for GetServiceHandle. */
|
||||
svc::Handle out_handle = svc::InvalidHandle;
|
||||
const Result result = this->GetServiceHandle(std::addressof(out_handle), service_name);
|
||||
|
||||
/* Serialize output. */
|
||||
if (R_SUCCEEDED(result) || !tipc::ResultRequestDeferred::Includes(result)) {
|
||||
std::memcpy(out_message_buffer + 0x00, CmifResponseToGetServiceHandleAndRegisterService, sizeof(CmifResponseToGetServiceHandleAndRegisterService));
|
||||
std::memcpy(out_message_buffer + 0x0C, std::addressof(out_handle), sizeof(out_handle));
|
||||
std::memcpy(out_message_buffer + 0x18, std::addressof(result), sizeof(result));
|
||||
} else {
|
||||
std::memcpy(out_message_buffer, CmifResponseToForceProcessorDeferral, sizeof(CmifResponseToForceProcessorDeferral));
|
||||
}
|
||||
} else if (command_id == 3) {
|
||||
/* Invoke the handler for UnregisterService. */
|
||||
const Result result = this->UnregisterService(service_name);
|
||||
|
||||
/* Serialize output. */
|
||||
if (R_SUCCEEDED(result) || !tipc::ResultRequestDeferred::Includes(result)) {
|
||||
std::memcpy(out_message_buffer + 0x00, CmifResponseToUnregisterService, sizeof(CmifResponseToUnregisterService));
|
||||
std::memcpy(out_message_buffer + 0x18, std::addressof(result), sizeof(result));
|
||||
} else {
|
||||
std::memcpy(out_message_buffer, CmifResponseToForceProcessorDeferral, sizeof(CmifResponseToForceProcessorDeferral));
|
||||
}
|
||||
} else {
|
||||
return tipc::ResultInvalidMethod();
|
||||
}
|
||||
} else if (std::memcmp(raw_message_buffer, CmifExpectedRequestHeaderForRegisterService, sizeof(CmifExpectedRequestHeaderForRegisterService)) == 0) {
|
||||
/* Get the command id. */
|
||||
const u32 command_id = *reinterpret_cast<const u32 *>(raw_message_buffer + 0x18);
|
||||
|
||||
/* Get the service_name. */
|
||||
sm::ServiceName service_name;
|
||||
std::memcpy(std::addressof(service_name), raw_message_buffer + 0x20, sizeof(service_name));
|
||||
|
||||
/* Get "is light". */
|
||||
bool is_light;
|
||||
is_light = (raw_message_buffer[0x28] & 0x01) != 0;
|
||||
|
||||
/* Get the max sessions. */
|
||||
u32 max_sessions;
|
||||
std::memcpy(std::addressof(max_sessions), raw_message_buffer + 0x2C, sizeof(max_sessions));
|
||||
|
||||
if (command_id == 2) {
|
||||
/* Invoke the handler for RegisterService. */
|
||||
svc::Handle out_handle = svc::InvalidHandle;
|
||||
const Result result = this->RegisterService(std::addressof(out_handle), service_name, max_sessions, is_light);
|
||||
|
||||
/* Serialize output. */
|
||||
if (R_SUCCEEDED(result) || !tipc::ResultRequestDeferred::Includes(result)) {
|
||||
std::memcpy(out_message_buffer + 0x00, CmifResponseToGetServiceHandleAndRegisterService, sizeof(CmifResponseToGetServiceHandleAndRegisterService));
|
||||
std::memcpy(out_message_buffer + 0x0C, std::addressof(out_handle), sizeof(out_handle));
|
||||
std::memcpy(out_message_buffer + 0x18, std::addressof(result), sizeof(result));
|
||||
} else {
|
||||
std::memcpy(out_message_buffer, CmifResponseToForceProcessorDeferral, sizeof(CmifResponseToForceProcessorDeferral));
|
||||
}
|
||||
} else {
|
||||
return tipc::ResultInvalidMethod();
|
||||
}
|
||||
} else {
|
||||
return tipc::ResultInvalidMethod();
|
||||
}
|
||||
}
|
||||
|
||||
Result UserService::RegisterClient(const tipc::ClientProcessId client_process_id) {
|
||||
m_process_id = client_process_id.value;
|
||||
m_initialized = true;
|
||||
return ResultSuccess();
|
||||
}
|
||||
|
||||
Result UserService::EnsureInitialized() {
|
||||
R_UNLESS(m_initialized, sm::ResultInvalidClient());
|
||||
return ResultSuccess();
|
||||
}
|
||||
|
||||
Result UserService::GetServiceHandle(tipc::OutMoveHandle out_h, ServiceName service) {
|
||||
R_TRY(this->EnsureInitialized());
|
||||
return impl::GetServiceHandle(out_h.GetHandlePointer(), m_process_id, service);
|
||||
}
|
||||
|
||||
Result UserService::RegisterService(tipc::OutMoveHandle out_h, ServiceName service, u32 max_sessions, bool is_light) {
|
||||
R_TRY(this->EnsureInitialized());
|
||||
return impl::RegisterService(out_h.GetHandlePointer(), m_process_id, service, max_sessions, is_light);
|
||||
}
|
||||
|
||||
Result UserService::UnregisterService(ServiceName service) {
|
||||
R_TRY(this->EnsureInitialized());
|
||||
return impl::UnregisterService(m_process_id, service);
|
||||
}
|
||||
|
||||
Result UserService::DetachClient(const tipc::ClientProcessId client_process_id) {
|
||||
AMS_UNUSED(client_process_id);
|
||||
|
||||
m_initialized = false;
|
||||
return ResultSuccess();
|
||||
}
|
||||
|
||||
Result UserService::AtmosphereInstallMitm(tipc::OutMoveHandle srv_h, tipc::OutMoveHandle qry_h, ServiceName service) {
|
||||
R_TRY(this->EnsureInitialized());
|
||||
return impl::InstallMitm(srv_h.GetHandlePointer(), qry_h.GetHandlePointer(), m_process_id, service);
|
||||
}
|
||||
|
||||
Result UserService::AtmosphereUninstallMitm(ServiceName service) {
|
||||
R_TRY(this->EnsureInitialized());
|
||||
return impl::UninstallMitm(m_process_id, service);
|
||||
}
|
||||
|
||||
Result UserService::AtmosphereAcknowledgeMitmSession(tipc::Out<MitmProcessInfo> client_info, tipc::OutMoveHandle fwd_h, ServiceName service) {
|
||||
R_TRY(this->EnsureInitialized());
|
||||
return impl::AcknowledgeMitmSession(client_info.GetPointer(), fwd_h.GetHandlePointer(), m_process_id, service);
|
||||
}
|
||||
|
||||
Result UserService::AtmosphereHasMitm(tipc::Out<bool> out, ServiceName service) {
|
||||
R_TRY(this->EnsureInitialized());
|
||||
return impl::HasMitm(out.GetPointer(), service);
|
||||
}
|
||||
|
||||
Result UserService::AtmosphereWaitMitm(ServiceName service) {
|
||||
R_TRY(this->EnsureInitialized());
|
||||
return impl::WaitMitm(service);
|
||||
}
|
||||
|
||||
Result UserService::AtmosphereDeclareFutureMitm(ServiceName service) {
|
||||
R_TRY(this->EnsureInitialized());
|
||||
return impl::DeclareFutureMitm(m_process_id, service);
|
||||
}
|
||||
|
||||
Result UserService::AtmosphereClearFutureMitm(ServiceName service) {
|
||||
R_TRY(this->EnsureInitialized());
|
||||
return impl::ClearFutureMitm(m_process_id, service);
|
||||
}
|
||||
|
||||
Result UserService::AtmosphereHasService(tipc::Out<bool> out, ServiceName service) {
|
||||
R_TRY(this->EnsureInitialized());
|
||||
return impl::HasService(out.GetPointer(), service);
|
||||
}
|
||||
|
||||
Result UserService::AtmosphereWaitService(ServiceName service) {
|
||||
R_TRY(this->EnsureInitialized());
|
||||
return impl::WaitService(service);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
/* Include the backwards compatibility shim for CMIF. */
|
||||
#include "sm_user_service_cmif_shim.inc"
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue