2018-04-19 05:00:10 +00:00
|
|
|
#include <switch.h>
|
|
|
|
#include <algorithm>
|
|
|
|
#include <cstdio>
|
|
|
|
#include "ldr_registration.hpp"
|
|
|
|
|
|
|
|
static Registration::List g_registration_list = {0};
|
|
|
|
static u64 g_num_registered = 1;
|
|
|
|
|
2018-04-19 05:15:17 +00:00
|
|
|
Registration::Process *Registration::get_free_process() {
|
2018-04-19 05:00:10 +00:00
|
|
|
unsigned int i;
|
|
|
|
for (i = 0; i < REGISTRATION_LIST_MAX; i++) {
|
|
|
|
if (!g_registration_list.processes[i].in_use) {
|
|
|
|
return &g_registration_list.processes[i];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2018-04-19 05:15:17 +00:00
|
|
|
Registration::Process *Registration::get_process(u64 index) {
|
2018-04-19 05:00:10 +00:00
|
|
|
unsigned int i;
|
|
|
|
for (i = 0; !g_registration_list.processes[i].in_use || g_registration_list.processes[i].index != index; i++) {
|
|
|
|
if (i >= REGISTRATION_LIST_MAX) {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return &g_registration_list.processes[i];
|
|
|
|
}
|
|
|
|
|
2018-04-19 22:14:48 +00:00
|
|
|
Registration::Process *Registration::get_process_by_process_id(u64 pid) {
|
|
|
|
unsigned int i;
|
|
|
|
for (i = 0; !g_registration_list.processes[i].in_use || g_registration_list.processes[i].process_id != pid; i++) {
|
|
|
|
if (i >= REGISTRATION_LIST_MAX) {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return &g_registration_list.processes[i];
|
|
|
|
}
|
|
|
|
|
2018-04-19 05:00:10 +00:00
|
|
|
bool Registration::register_tid_sid(const TidSid *tid_sid, u64 *out_index) {
|
|
|
|
Registration::Process *free_process = get_free_process();
|
|
|
|
if (free_process == NULL) {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Reset the process. */
|
|
|
|
*free_process = {0};
|
|
|
|
free_process->tid_sid = *tid_sid;
|
|
|
|
free_process->in_use = true;
|
|
|
|
free_process->index = g_num_registered++;
|
|
|
|
*out_index = free_process->index;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool Registration::unregister_index(u64 index) {
|
|
|
|
Registration::Process *target_process = get_process(index);
|
|
|
|
if (target_process == NULL) {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Reset the process. */
|
|
|
|
*target_process = {0};
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
void Registration::set_process_id_and_tid_min(u64 index, u64 process_id, u64 tid_min) {
|
|
|
|
Registration::Process *target_process = get_process(index);
|
|
|
|
if (target_process == NULL) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
target_process->process_id = process_id;
|
|
|
|
target_process->title_id_min = tid_min;
|
|
|
|
}
|
|
|
|
|
|
|
|
void Registration::add_nso_info(u64 index, u64 base_address, u64 size, const unsigned char *build_id) {
|
|
|
|
Registration::Process *target_process = get_process(index);
|
|
|
|
if (target_process == NULL) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
for (unsigned int i = 0; i < NSO_INFO_MAX; i++) {
|
|
|
|
if (!target_process->nso_infos[i].in_use) {
|
2018-04-19 22:14:48 +00:00
|
|
|
target_process->nso_infos[i].info.base_address = base_address;
|
|
|
|
target_process->nso_infos[i].info.size = size;
|
|
|
|
std::copy(build_id, build_id + sizeof(target_process->nso_infos[i].info.build_id), target_process->nso_infos[i].info.build_id);
|
2018-04-19 05:00:10 +00:00
|
|
|
target_process->nso_infos[i].in_use = true;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
2018-04-19 22:14:48 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
Result Registration::get_nso_infos_for_process_id(Registration::NsoInfo *out, u32 max_out, u64 process_id, u32 *num_written) {
|
|
|
|
Registration::Process *target_process = get_process_by_process_id(process_id);
|
|
|
|
if (target_process == NULL) {
|
|
|
|
return 0x1009;
|
|
|
|
}
|
|
|
|
u32 cur = 0;
|
|
|
|
|
|
|
|
if (max_out > 0) {
|
|
|
|
for (unsigned int i = 0; i < NSO_INFO_MAX && cur < max_out; i++) {
|
|
|
|
if (target_process->nso_infos[i].in_use) {
|
|
|
|
out[cur++] = target_process->nso_infos[i].info;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
*num_written = cur;
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|