mirror of
https://github.com/Atmosphere-NX/Atmosphere.git
synced 2024-11-30 07:42:13 +00:00
c2d9ac8f5c
Also greatly refactors libstratosphere, and does a lot of other things. There is a lot of code in this one.
59 lines
2.3 KiB
C++
59 lines
2.3 KiB
C++
#pragma once
|
|
#include <switch.h>
|
|
|
|
#define REGISTRATION_LIST_MAX_PROCESS (0x40)
|
|
#define REGISTRATION_LIST_MAX_SERVICE (0x100)
|
|
#define REGISTRATION_MAX_SAC_SIZE (0x200)
|
|
#define REGISTRATION_INITIAL_PID_MAX 0x50
|
|
|
|
class Registration {
|
|
public:
|
|
struct Process {
|
|
u64 pid;
|
|
u64 sac_size;
|
|
u8 sac[REGISTRATION_MAX_SAC_SIZE];
|
|
};
|
|
|
|
struct Service {
|
|
u64 service_name;
|
|
u64 owner_pid;
|
|
Handle port_h;
|
|
|
|
/* Debug. */
|
|
u64 max_sessions;
|
|
bool is_light;
|
|
|
|
/* Extension. */
|
|
u64 mitm_pid;
|
|
Handle mitm_port_h;
|
|
Handle mitm_query_h;
|
|
};
|
|
|
|
/* Utilities. */
|
|
static Registration::Process *GetProcessForPid(u64 pid);
|
|
static Registration::Process *GetFreeProcess();
|
|
static Registration::Service *GetService(u64 service);
|
|
static Registration::Service *GetFreeService();
|
|
static bool IsValidForSac(u8 *sac, size_t sac_size, u64 service, bool is_host);
|
|
static bool ValidateSacAgainstRestriction(u8 *r_sac, size_t r_sac_size, u8 *sac, size_t sac_size);
|
|
static void CacheInitialProcessIdLimits();
|
|
static bool IsInitialProcess(u64 pid);
|
|
static u64 GetInitialProcessId();
|
|
|
|
/* Process management. */
|
|
static Result RegisterProcess(u64 pid, u8 *acid_sac, size_t acid_sac_size, u8 *aci0_sac, size_t aci0_sac_size);
|
|
static Result UnregisterProcess(u64 pid);
|
|
|
|
/* Service management. */
|
|
static bool HasService(u64 service);
|
|
static Result GetServiceHandle(u64 pid, u64 service, Handle *out);
|
|
static Result GetServiceForPid(u64 pid, u64 service, Handle *out);
|
|
static Result RegisterServiceForPid(u64 pid, u64 service, u64 max_sessions, bool is_light, Handle *out);
|
|
static Result RegisterServiceForSelf(u64 service, u64 max_sessions, bool is_light, Handle *out);
|
|
static Result UnregisterServiceForPid(u64 pid, u64 service);
|
|
|
|
/* Extension. */
|
|
static Result InstallMitmForPid(u64 pid, u64 service, Handle *out, Handle *query_out);
|
|
static Result UninstallMitmForPid(u64 pid, u64 service);
|
|
static Result AssociatePidTidForMitm(u64 pid, u64 tid);
|
|
};
|