mika/Atmosphere
1
0
Fork 0
mirror of https://github.com/Atmosphere-NX/Atmosphere.git synced 2025-07-05 09:02:15 +02:00
Code Issues Packages Projects Releases Wiki Activity

sm: add query registrations extension

Browse Source
This commit is contained in:
misson20000 2018-12-20 16:18:10 -08:00
parent 34c16e211f
commit 0169d7d77c
5 changed files with 51 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
stratosphere/sm/source/sm_manager_service.cpp
View File

@ -35,3 +35,10 @@ void ManagerService::AtmosphereHasMitm(Out<bool> out, SmServiceName service) {
out.SetValue(Registration::HasMitm(smEncodeName(service.name)));
}
Result ManagerService::AtmosphereQueryRegistrations(u64 offset, OutBuffer<ServiceRecord> records, Out<u64> out_count, Out<u64> record_size) {
uint64_t count = records.num_elements;
Registration::QueryRegistrations(offset, records.buffer, &count);
out_count.SetValue(count);
record_size.SetValue(sizeof(ServiceRecord));
return 0;
}

3
stratosphere/sm/source/sm_manager_service.hpp
View File

@ -26,6 +26,7 @@ enum ManagerServiceCmd {
Manager_Cmd_AtmosphereEndInitDefers = 65000,
Manager_Cmd_AtmosphereHasMitm = 65001,
Manager_Cmd_AtmosphereQueryRegistrations = 65002,
};
class ManagerService final : public IServiceObject {
@ -35,6 +36,7 @@ class ManagerService final : public IServiceObject {
virtual Result UnregisterProcess(u64 pid);
virtual void AtmosphereEndInitDefers();
virtual void AtmosphereHasMitm(Out<bool> out, SmServiceName service);
virtual Result AtmosphereQueryRegistrations(u64 offset, OutBuffer<ServiceRecord> records, Out<u64> out_count, Out<u64> record_size);
public:
DEFINE_SERVICE_DISPATCH_TABLE {
MakeServiceCommandMeta<Manager_Cmd_RegisterProcess, &ManagerService::RegisterProcess>(),
@ -42,5 +44,6 @@ class ManagerService final : public IServiceObject {
MakeServiceCommandMeta<Manager_Cmd_AtmosphereEndInitDefers, &ManagerService::AtmosphereEndInitDefers>(),
MakeServiceCommandMeta<Manager_Cmd_AtmosphereHasMitm, &ManagerService::AtmosphereHasMitm>(),
MakeServiceCommandMeta<Manager_Cmd_AtmosphereQueryRegistrations, &ManagerService::AtmosphereQueryRegistrations>(),
};
};

26
stratosphere/sm/source/sm_registration.cpp
View File

@ -562,3 +562,29 @@ Result Registration::AssociatePidTidForMitm(u64 pid, u64 tid) {
}
return 0x0;
}
void Registration::QueryRegistrations(u64 offset, ServiceRecord *out, u64 *count) {
u64 space = *count;
*count = 0;
for(auto i = g_service_list.begin() + offset; i < g_service_list.end() && space > 0; i++) {
if(i->service_name != 0) {
if(offset > 0) {
offset--;
} else {
ServiceRecord *out_record = out++;
out_record->service_name = i->service_name;
out_record->owner_pid = i->owner_pid;
out_record->max_sessions = i->max_sessions;
out_record->mitm_pid = i->mitm_pid;
out_record->mitm_waiting_ack_pid = i->mitm_waiting_ack_pid;
out_record->is_light = i->is_light;
out_record->mitm_waiting_ack = i->mitm_waiting_ack;
space--;
(*count)++;
}
}
}
}

3
stratosphere/sm/source/sm_registration.hpp
View File

@ -17,6 +17,8 @@
#pragma once
#include <switch.h>
#include "sm_types.hpp"
#define REGISTRATION_LIST_MAX_PROCESS (0x40)
#define REGISTRATION_LIST_MAX_SERVICE (0x100)
#define REGISTRATION_MAX_SAC_SIZE (0x200)
@ -81,4 +83,5 @@ class Registration {
static Result UninstallMitmForPid(u64 pid, u64 service);
static Result AcknowledgeMitmSessionForPid(u64 pid, u64 service, Handle *out, u64 *out_pid);
static Result AssociatePidTidForMitm(u64 pid, u64 tid);
static void QueryRegistrations(u64 offset, ServiceRecord *out, u64 *count);
};

11
stratosphere/sm/source/sm_types.hpp
View File

@ -20,3 +20,14 @@ struct SmServiceName {
};
static_assert(__alignof__(SmServiceName) == 1, "SmServiceName definition!");
// for QueryRegistrations extension
struct ServiceRecord {
u64 service_name;
u64 owner_pid;
u64 max_sessions;
u64 mitm_pid;
u64 mitm_waiting_ack_pid;
bool is_light;
bool mitm_waiting_ack;
};