mirror of
https://github.com/Atmosphere-NX/Atmosphere.git
synced 2024-11-27 06:12:15 +00:00
ad636f7216
* Stratosphere: Use modern C++ idioms in some places * algorithms like std::for_each are used instead of raw loops * Stratosphere: Replace more raw loops with algorithms * Stratosphere: Add a utility predicate function to test for equality with a reference element This can be used to rewrite some common raw loops using algorithms instead * fs.mitm: Use variant * fs.mitm: Use enum class * fs.mitm: Turn RomFSSourceInfo::Cleanup into a destructor This obsoletes the need for a custom deleter in other places * fs.mitm: Use enum class some more * fs.mitm: Use unique_ptr * fs.mitm: Simplify initialization * Stratosphere: Simplify initialization * fs.mitm: Use unique_ptr (fix memory leak along the way) The previous code was using "delete" rather than "delete[]" * fs.mitm: Use vector::emplace_back rather than push_back emplace_back constructs elements in-place, hence avoiding a redundant element copy. * Stratosphere: Replace more raw loops with algorithms * Stratosphere: Use unique_ptr * fs.mitm: Replace more raw loops with algorithms * Stratosphere: Prefer move-construction over copy-construction when moving sink parameters around
468 lines
13 KiB
C++
468 lines
13 KiB
C++
#include <switch.h>
|
|
#include <algorithm>
|
|
#include <stratosphere/servicesession.hpp>
|
|
#include "sm_registration.hpp"
|
|
#include "meta_tools.hpp"
|
|
|
|
static std::array<Registration::Process, REGISTRATION_LIST_MAX_PROCESS> g_process_list = {0};
|
|
static std::array<Registration::Service, REGISTRATION_LIST_MAX_SERVICE> g_service_list = {0};
|
|
|
|
static u64 g_initial_process_id_low = 0;
|
|
static u64 g_initial_process_id_high = 0;
|
|
static bool g_determined_initial_process_ids = false;
|
|
|
|
u64 GetServiceNameLength(u64 service) {
|
|
u64 service_name_len = 0;
|
|
while (service & 0xFF) {
|
|
service_name_len++;
|
|
service >>= 8;
|
|
}
|
|
return service_name_len;
|
|
}
|
|
|
|
/* Utilities. */
|
|
Registration::Process *Registration::GetProcessForPid(u64 pid) {
|
|
auto process_it = std::find_if(g_process_list.begin(), g_process_list.end(), member_equals_fn(&Process::pid, pid));
|
|
if (process_it == g_process_list.end()) {
|
|
return nullptr;
|
|
}
|
|
return &*process_it;
|
|
}
|
|
|
|
Registration::Process *Registration::GetFreeProcess() {
|
|
return GetProcessForPid(0);
|
|
}
|
|
|
|
Registration::Service *Registration::GetService(u64 service_name) {
|
|
auto service_it = std::find_if(g_service_list.begin(), g_service_list.end(), member_equals_fn(&Service::service_name, service_name));
|
|
if (service_it == g_service_list.end()) {
|
|
return nullptr;
|
|
}
|
|
return &*service_it;
|
|
}
|
|
|
|
Registration::Service *Registration::GetFreeService() {
|
|
return GetService(0);
|
|
}
|
|
|
|
bool Registration::IsValidForSac(u8 *sac, size_t sac_size, u64 service, bool is_host) {
|
|
u8 cur_ctrl;
|
|
u64 cur_service;
|
|
u64 service_for_compare;
|
|
bool cur_is_host;
|
|
size_t remaining = sac_size;
|
|
while (remaining) {
|
|
cur_ctrl = *sac++;
|
|
remaining--;
|
|
size_t cur_size = (cur_ctrl & 7) + 1;
|
|
if (cur_size > remaining) {
|
|
break;
|
|
}
|
|
cur_is_host = (cur_ctrl & 0x80) != 0;
|
|
cur_service = 0;
|
|
for (unsigned int i = 0; i < cur_size; i++) {
|
|
cur_service |= ((u64)sac[i]) << (8 * i);
|
|
}
|
|
/* Check if the last byte is a wildcard ('*') */
|
|
service_for_compare = service;
|
|
if (sac[cur_size - 1] == '*') {
|
|
u64 mask = U64_MAX;
|
|
for (unsigned int i = 0; i < 8 - (cur_size - 1); i++) {
|
|
mask >>= 8;
|
|
}
|
|
/* Mask cur_service, service with 0xFF.. up until the wildcard. */
|
|
cur_service &= mask;
|
|
service_for_compare &= mask;
|
|
}
|
|
if (cur_service == service_for_compare && (is_host == cur_is_host)) {
|
|
return true;
|
|
}
|
|
sac += cur_size;
|
|
remaining -= cur_size;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
|
|
bool Registration::ValidateSacAgainstRestriction(u8 *r_sac, size_t r_sac_size, u8 *sac, size_t sac_size) {
|
|
u8 cur_ctrl;
|
|
u64 cur_service;
|
|
bool cur_is_host;
|
|
size_t remaining = sac_size;
|
|
while (remaining) {
|
|
cur_ctrl = *sac++;
|
|
remaining--;
|
|
size_t cur_size = (cur_ctrl & 7) + 1;
|
|
if (cur_size > remaining) {
|
|
break;
|
|
}
|
|
cur_is_host = (cur_ctrl & 0x80) != 0;
|
|
cur_service = 0;
|
|
for (unsigned int i = 0; i < cur_size; i++) {
|
|
cur_service |= ((u64)sac[i]) << (8 * i);
|
|
}
|
|
if (!IsValidForSac(r_sac, r_sac_size, cur_service, cur_is_host)) {
|
|
return false;
|
|
}
|
|
sac += cur_size;
|
|
remaining -= cur_size;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void Registration::CacheInitialProcessIdLimits() {
|
|
if (g_determined_initial_process_ids) {
|
|
return;
|
|
}
|
|
if (kernelAbove500()) {
|
|
svcGetSystemInfo(&g_initial_process_id_low, 2, 0, 0);
|
|
svcGetSystemInfo(&g_initial_process_id_high, 2, 0, 1);
|
|
} else {
|
|
g_initial_process_id_low = 0;
|
|
g_initial_process_id_high = REGISTRATION_INITIAL_PID_MAX;
|
|
}
|
|
g_determined_initial_process_ids = true;
|
|
}
|
|
|
|
bool Registration::IsInitialProcess(u64 pid) {
|
|
CacheInitialProcessIdLimits();
|
|
return g_initial_process_id_low <= pid && pid <= g_initial_process_id_high;
|
|
}
|
|
|
|
u64 Registration::GetInitialProcessId() {
|
|
CacheInitialProcessIdLimits();
|
|
return g_initial_process_id_low;
|
|
}
|
|
|
|
/* Process management. */
|
|
Result Registration::RegisterProcess(u64 pid, u8 *acid_sac, size_t acid_sac_size, u8 *aci0_sac, size_t aci0_sac_size) {
|
|
Registration::Process *proc = GetFreeProcess();
|
|
if (aci0_sac_size > REGISTRATION_MAX_SAC_SIZE) {
|
|
return 0x1215;
|
|
}
|
|
|
|
if (proc == NULL) {
|
|
return 0x215;
|
|
}
|
|
|
|
if (aci0_sac_size && !ValidateSacAgainstRestriction(acid_sac, acid_sac_size, aci0_sac, aci0_sac_size)) {
|
|
return 0x1015;
|
|
}
|
|
|
|
proc->pid = pid;
|
|
proc->sac_size = aci0_sac_size;
|
|
std::copy(aci0_sac, aci0_sac + aci0_sac_size, proc->sac);
|
|
return 0;
|
|
}
|
|
|
|
Result Registration::UnregisterProcess(u64 pid) {
|
|
Registration::Process *proc = GetProcessForPid(pid);
|
|
if (proc == NULL) {
|
|
return 0x415;
|
|
}
|
|
|
|
proc->pid = 0;
|
|
return 0;
|
|
}
|
|
|
|
/* Service management. */
|
|
bool Registration::HasService(u64 service) {
|
|
return std::any_of(g_service_list.begin(), g_service_list.end(), member_equals_fn(&Service::service_name, service));
|
|
}
|
|
|
|
Result Registration::GetServiceHandle(u64 pid, u64 service, Handle *out) {
|
|
Registration::Service *target_service = GetService(service);
|
|
if (target_service == NULL) {
|
|
/* Note: This defers the result until later. */
|
|
return RESULT_DEFER_SESSION;
|
|
}
|
|
|
|
*out = 0;
|
|
Result rc;
|
|
if (target_service->mitm_pid == 0 || target_service->mitm_pid == pid) {
|
|
rc = svcConnectToPort(out, target_service->port_h);
|
|
} else {
|
|
IpcCommand c;
|
|
ipcInitialize(&c);
|
|
struct {
|
|
u64 magic;
|
|
u64 cmd_id;
|
|
u64 pid;
|
|
} *info = ((decltype(info))ipcPrepareHeader(&c, sizeof(*info)));
|
|
info->magic = SFCI_MAGIC;
|
|
info->cmd_id = 65000;
|
|
info->pid = pid;
|
|
rc = ipcDispatch(target_service->mitm_query_h);
|
|
if (R_SUCCEEDED(rc)) {
|
|
IpcParsedCommand r;
|
|
ipcParse(&r);
|
|
struct {
|
|
u64 magic;
|
|
u64 result;
|
|
u64 should_mitm;
|
|
} *resp = ((decltype(resp))r.Raw);
|
|
rc = resp->result;
|
|
if (R_SUCCEEDED(rc)) {
|
|
if (resp->should_mitm) {
|
|
rc = svcConnectToPort(out, target_service->mitm_port_h);
|
|
} else {
|
|
rc = svcConnectToPort(out, target_service->port_h);
|
|
}
|
|
}
|
|
}
|
|
if (R_FAILED(rc)) {
|
|
rc = svcConnectToPort(out, target_service->port_h);
|
|
}
|
|
}
|
|
if (R_FAILED(rc)) {
|
|
if ((rc & 0x3FFFFF) == 0xE01) {
|
|
return 0x615;
|
|
}
|
|
}
|
|
|
|
return rc;
|
|
}
|
|
|
|
Result Registration::GetServiceForPid(u64 pid, u64 service, Handle *out) {
|
|
if (!service) {
|
|
return 0xC15;
|
|
}
|
|
|
|
u64 service_name_len = GetServiceNameLength(service);
|
|
|
|
/* If the service has bytes after a null terminator, that's no good. */
|
|
if (service_name_len != 8 && (service >> (8 * service_name_len))) {
|
|
return 0xC15;
|
|
}
|
|
|
|
if (!IsInitialProcess(pid)) {
|
|
Registration::Process *proc = GetProcessForPid(pid);
|
|
if (proc == NULL) {
|
|
return 0x415;
|
|
}
|
|
|
|
if (!IsValidForSac(proc->sac, proc->sac_size, service, false)) {
|
|
return 0x1015;
|
|
}
|
|
}
|
|
|
|
return GetServiceHandle(pid, service, out);
|
|
}
|
|
|
|
Result Registration::RegisterServiceForPid(u64 pid, u64 service, u64 max_sessions, bool is_light, Handle *out) {
|
|
if (!service) {
|
|
return 0xC15;
|
|
}
|
|
|
|
u64 service_name_len = GetServiceNameLength(service);
|
|
|
|
/* If the service has bytes after a null terminator, that's no good. */
|
|
if (service_name_len != 8 && (service >> (8 * service_name_len))) {
|
|
return 0xC15;
|
|
}
|
|
|
|
if (!IsInitialProcess(pid)) {
|
|
Registration::Process *proc = GetProcessForPid(pid);
|
|
if (proc == NULL) {
|
|
return 0x415;
|
|
}
|
|
|
|
if (!IsValidForSac(proc->sac, proc->sac_size, service, true)) {
|
|
return 0x1015;
|
|
}
|
|
}
|
|
|
|
if (HasService(service)) {
|
|
return 0x815;
|
|
}
|
|
|
|
#ifdef SM_MINIMUM_SESSION_LIMIT
|
|
if (max_sessions < SM_MINIMUM_SESSION_LIMIT) {
|
|
max_sessions = SM_MINIMUM_SESSION_LIMIT;
|
|
}
|
|
#endif
|
|
|
|
Registration::Service *free_service = GetFreeService();
|
|
if (free_service == NULL) {
|
|
return 0xA15;
|
|
}
|
|
|
|
*out = 0;
|
|
*free_service = (const Registration::Service){0};
|
|
Result rc = svcCreatePort(out, &free_service->port_h, max_sessions, is_light, (char *)&free_service->service_name);
|
|
|
|
if (R_SUCCEEDED(rc)) {
|
|
free_service->service_name = service;
|
|
free_service->owner_pid = pid;
|
|
free_service->max_sessions = max_sessions;
|
|
free_service->is_light = is_light;
|
|
}
|
|
|
|
return rc;
|
|
}
|
|
|
|
Result Registration::RegisterServiceForSelf(u64 service, u64 max_sessions, bool is_light, Handle *out) {
|
|
u64 pid;
|
|
Result rc = svcGetProcessId(&pid, CUR_PROCESS_HANDLE);
|
|
if (R_FAILED(rc)) {
|
|
return rc;
|
|
}
|
|
|
|
u64 service_name_len = GetServiceNameLength(service);
|
|
|
|
/* If the service has bytes after a null terminator, that's no good. */
|
|
if (service_name_len != 8 && (service >> (8 * service_name_len))) {
|
|
return 0xC15;
|
|
}
|
|
|
|
if (HasService(service)) {
|
|
return 0x815;
|
|
}
|
|
|
|
#ifdef SM_MINIMUM_SESSION_LIMIT
|
|
if (max_sessions < SM_MINIMUM_SESSION_LIMIT) {
|
|
max_sessions = SM_MINIMUM_SESSION_LIMIT;
|
|
}
|
|
#endif
|
|
|
|
Registration::Service *free_service = GetFreeService();
|
|
if (free_service == NULL) {
|
|
return 0xA15;
|
|
}
|
|
|
|
*out = 0;
|
|
*free_service = (const Registration::Service){0};
|
|
rc = svcCreatePort(out, &free_service->port_h, max_sessions, is_light, (char *)&free_service->service_name);
|
|
|
|
if (R_SUCCEEDED(rc)) {
|
|
free_service->service_name = service;
|
|
free_service->owner_pid = pid;
|
|
free_service->max_sessions = max_sessions;
|
|
free_service->is_light = is_light;
|
|
}
|
|
|
|
return rc;
|
|
}
|
|
|
|
Result Registration::UnregisterServiceForPid(u64 pid, u64 service) {
|
|
if (!service) {
|
|
return 0xC15;
|
|
}
|
|
|
|
u64 service_name_len = GetServiceNameLength(service);
|
|
|
|
/* If the service has bytes after a null terminator, that's no good. */
|
|
if (service_name_len != 8 && (service >> (8 * service_name_len))) {
|
|
return 0xC15;
|
|
}
|
|
|
|
Registration::Service *target_service = GetService(service);
|
|
if (target_service == NULL) {
|
|
return 0xE15;
|
|
}
|
|
|
|
if (!IsInitialProcess(pid) && target_service->owner_pid != pid) {
|
|
return 0x1015;
|
|
}
|
|
|
|
svcCloseHandle(target_service->port_h);
|
|
svcCloseHandle(target_service->mitm_port_h);
|
|
svcCloseHandle(target_service->mitm_query_h);
|
|
*target_service = (const Registration::Service){0};
|
|
return 0;
|
|
}
|
|
|
|
|
|
Result Registration::InstallMitmForPid(u64 pid, u64 service, Handle *out, Handle *query_out) {
|
|
if (!service) {
|
|
return 0xC15;
|
|
}
|
|
|
|
u64 service_name_len = GetServiceNameLength(service);
|
|
|
|
/* If the service has bytes after a null terminator, that's no good. */
|
|
if (service_name_len != 8 && (service >> (8 * service_name_len))) {
|
|
return 0xC15;
|
|
}
|
|
|
|
/* Verify we're allowed to mitm the service. */
|
|
if (!IsInitialProcess(pid)) {
|
|
Registration::Process *proc = GetProcessForPid(pid);
|
|
if (proc == NULL) {
|
|
return 0x415;
|
|
}
|
|
|
|
if (!IsValidForSac(proc->sac, proc->sac_size, service, true)) {
|
|
return 0x1015;
|
|
}
|
|
}
|
|
|
|
/* Verify the service exists. */
|
|
Registration::Service *target_service = GetService(service);
|
|
if (target_service == NULL) {
|
|
return 0xE15;
|
|
}
|
|
|
|
/* Verify the service isn't already being mitm'd. */
|
|
if (target_service->mitm_pid != 0) {
|
|
return 0x815;
|
|
}
|
|
|
|
*out = 0;
|
|
u64 x = 0;
|
|
Result rc = svcCreatePort(out, &target_service->mitm_port_h, target_service->max_sessions, target_service->is_light, (char *)&x);
|
|
|
|
if (R_SUCCEEDED(rc) && R_SUCCEEDED((rc = svcCreateSession(query_out, &target_service->mitm_query_h, 0, 0)))) {
|
|
target_service->mitm_pid = pid;
|
|
}
|
|
|
|
return rc;
|
|
}
|
|
|
|
Result Registration::UninstallMitmForPid(u64 pid, u64 service) {
|
|
if (!service) {
|
|
return 0xC15;
|
|
}
|
|
|
|
u64 service_name_len = GetServiceNameLength(service);
|
|
|
|
/* If the service has bytes after a null terminator, that's no good. */
|
|
if (service_name_len != 8 && (service >> (8 * service_name_len))) {
|
|
return 0xC15;
|
|
}
|
|
|
|
Registration::Service *target_service = GetService(service);
|
|
if (target_service == NULL) {
|
|
return 0xE15;
|
|
}
|
|
|
|
if (!IsInitialProcess(pid) && target_service->mitm_pid != pid) {
|
|
return 0x1015;
|
|
}
|
|
|
|
svcCloseHandle(target_service->mitm_port_h);
|
|
svcCloseHandle(target_service->mitm_query_h);
|
|
target_service->mitm_pid = 0;
|
|
return 0;
|
|
}
|
|
|
|
Result Registration::AssociatePidTidForMitm(u64 pid, u64 tid) {
|
|
for (auto &service : g_service_list) {
|
|
if (service.mitm_pid) {
|
|
IpcCommand c;
|
|
ipcInitialize(&c);
|
|
struct {
|
|
u64 magic;
|
|
u64 cmd_id;
|
|
u64 pid;
|
|
u64 tid;
|
|
} *info = ((decltype(info))ipcPrepareHeader(&c, sizeof(*info)));
|
|
info->magic = SFCI_MAGIC;
|
|
info->cmd_id = 65001;
|
|
info->pid = pid;
|
|
info->tid = tid;
|
|
ipcDispatch(service.mitm_query_h);
|
|
}
|
|
}
|
|
return 0x0;
|
|
}
|