Atmosphere/stratosphere/pm/source/pm_boot2.cpp

165 lines
6.2 KiB
C++
Raw Normal View History

2018-05-01 16:58:19 +00:00
#include <cstdlib>
#include <cstdint>
#include <cstdio>
2018-05-01 16:58:19 +00:00
#include <cstring>
#include <dirent.h>
2018-05-01 16:58:19 +00:00
#include <malloc.h>
#include <switch.h>
#include <stratosphere.hpp>
2018-07-27 09:23:53 +00:00
#include "pm_boot2.hpp"
#include "pm_registration.hpp"
2018-05-01 16:58:19 +00:00
static bool IsHexadecimal(const char *str) {
while (*str) {
if (('0' <= *str && *str <= '9') ||
('a' <= *str && *str <= 'f') ||
('A' <= *str && *str <= 'F')) {
str++;
} else {
return false;
}
}
return true;
}
2018-07-27 09:37:45 +00:00
static void LaunchTitle(Boot2KnownTitleId title_id, FsStorageId storage_id, u32 launch_flags, u64 *pid) {
2018-05-01 16:58:19 +00:00
u64 local_pid;
2018-07-27 09:23:53 +00:00
2018-07-27 09:37:45 +00:00
Result rc = Registration::LaunchProcessByTidSid(Registration::TidSid{(u64)title_id, storage_id}, launch_flags, &local_pid);
2018-05-01 16:58:19 +00:00
switch (rc) {
case 0xCE01:
/* Out of resource! */
/* TODO: Panic(). */
break;
case 0xDE01:
/* Out of memory! */
/* TODO: Panic(). */
break;
case 0xD001:
2018-05-01 17:04:39 +00:00
/* Limit Reached! */
2018-05-01 16:58:19 +00:00
/* TODO: Panic(). */
break;
default:
/* We don't care about other issues. */
break;
}
if (pid) {
*pid = local_pid;
}
}
2018-07-27 09:23:53 +00:00
static bool ShouldForceMaintenanceMode() {
2018-05-01 16:58:19 +00:00
/* TODO: Contact set:sys, retrieve boot!force_maintenance, read plus/minus buttons. */
return false;
}
2018-07-27 09:37:45 +00:00
static const std::tuple<Boot2KnownTitleId, bool> g_additional_launch_programs[] = {
{Boot2KnownTitleId::am, true}, /* am */
{Boot2KnownTitleId::nvservices, true}, /* nvservices */
{Boot2KnownTitleId::nvnflinger, true}, /* nvnflinger */
{Boot2KnownTitleId::vi, true}, /* vi */
{Boot2KnownTitleId::ns, true}, /* ns */
{Boot2KnownTitleId::lm, true}, /* lm */
{Boot2KnownTitleId::ppc, true}, /* ppc */
{Boot2KnownTitleId::ptm, true}, /* ptm */
{Boot2KnownTitleId::hid, true}, /* hid */
{Boot2KnownTitleId::audio, true}, /* audio */
{Boot2KnownTitleId::lbl, true}, /* lbl */
{Boot2KnownTitleId::wlan, true}, /* wlan */
{Boot2KnownTitleId::bluetooth, true}, /* bluetooth */
{Boot2KnownTitleId::bsdsockets, true}, /* bsdsockets */
{Boot2KnownTitleId::nifm, true}, /* nifm */
{Boot2KnownTitleId::ldn, true}, /* ldn */
{Boot2KnownTitleId::account, true}, /* account */
{Boot2KnownTitleId::friends, false}, /* friends */
{Boot2KnownTitleId::nfc, true}, /* nfc */
{Boot2KnownTitleId::jpegdec, true}, /* jpegdec */
{Boot2KnownTitleId::capsrv, true}, /* capsrv */
{Boot2KnownTitleId::ssl, true}, /* ssl */
{Boot2KnownTitleId::nim, true}, /* nim */
{Boot2KnownTitleId::bcat, false}, /* bcat */
{Boot2KnownTitleId::erpt, true}, /* erpt */
{Boot2KnownTitleId::es, true}, /* es */
{Boot2KnownTitleId::pctl, true}, /* pctl */
{Boot2KnownTitleId::btm, true}, /* btm */
{Boot2KnownTitleId::eupld, false}, /* eupld */
{Boot2KnownTitleId::glue, true}, /* glue */
{Boot2KnownTitleId::eclct, true}, /* eclct */
{Boot2KnownTitleId::npns, false}, /* npns */
{Boot2KnownTitleId::fatal, true}, /* fatal */
{Boot2KnownTitleId::ro, true}, /* ro */
{Boot2KnownTitleId::profiler, true}, /* profiler */
{Boot2KnownTitleId::sdb, true}, /* sdb */
{Boot2KnownTitleId::migration, true}, /* migration */
{Boot2KnownTitleId::grc, true}, /* grc */
2018-05-01 16:58:19 +00:00
};
static void MountSdCard() {
Handle tmp_hnd = 0;
static const char * const required_active_services[] = {"pcv", "gpio", "pinmux", "psc:c"};
for (unsigned int i = 0; i < sizeof(required_active_services) / sizeof(required_active_services[0]); i++) {
if (R_FAILED(smGetServiceOriginal(&tmp_hnd, smEncodeName(required_active_services[i])))) {
/* TODO: Panic */
} else {
svcCloseHandle(tmp_hnd);
}
}
fsdevMountSdmc();
}
2018-07-27 09:23:53 +00:00
void EmbeddedBoot2::Main() {
/* psc, bus, pcv is the minimal set of required titles to get SD card. */
/* bus depends on pcie, and pcv depends on settings. */
2018-05-01 16:58:19 +00:00
/* Launch psc. */
2018-07-27 09:37:45 +00:00
LaunchTitle(Boot2KnownTitleId::psc, FsStorageId_NandSystem, 0, NULL);
2018-05-01 16:58:19 +00:00
/* Launch pcie. */
2018-07-27 09:37:45 +00:00
LaunchTitle(Boot2KnownTitleId::pcie, FsStorageId_NandSystem, 0, NULL);
2018-07-27 09:23:53 +00:00
/* Launch bus. */
2018-07-27 09:37:45 +00:00
LaunchTitle(Boot2KnownTitleId::bus, FsStorageId_NandSystem, 0, NULL);
/* Launch settings. */
LaunchTitle(Boot2KnownTitleId::settings, FsStorageId_NandSystem, 0, NULL);
2018-07-27 09:23:53 +00:00
/* Launch pcv. */
2018-07-27 09:37:45 +00:00
LaunchTitle(Boot2KnownTitleId::pcv, FsStorageId_NandSystem, 0, NULL);
2018-05-01 16:58:19 +00:00
/* At this point, the SD card can be mounted. */
MountSdCard();
/* Launch usb. */
LaunchTitle(Boot2KnownTitleId::usb, FsStorageId_NandSystem, 0, NULL);
/* Launch tma. */
LaunchTitle(Boot2KnownTitleId::tma, FsStorageId_NandSystem, 0, NULL);
/* Launch default programs. */
2018-05-01 16:58:19 +00:00
bool maintenance = ShouldForceMaintenanceMode();
for (auto &launch_program : g_additional_launch_programs) {
if (!maintenance || std::get<bool>(launch_program)) {
2018-07-27 09:37:45 +00:00
LaunchTitle(std::get<Boot2KnownTitleId>(launch_program), FsStorageId_NandSystem, 0, NULL);
2018-05-01 16:58:19 +00:00
}
}
/* Allow for user-customizable programs. */
DIR *titles_dir = opendir("sdmc:/atmosphere/titles");
struct dirent *ent;
if (titles_dir != NULL) {
while ((ent = readdir(titles_dir)) != NULL) {
if (strlen(ent->d_name) == 0x10 && IsHexadecimal(ent->d_name)) {
u64 title_id = strtoul(ent->d_name, NULL, 16);
char title_path[FS_MAX_PATH] = {0};
strcpy(title_path, "sdmc:/atmosphere/titles/");
strcat(title_path, ent->d_name);
strcat(title_path, "/boot2.flag");
FILE *f_flag = fopen(title_path, "rb");
if (f_flag != NULL) {
fclose(f_flag);
LaunchTitle((Boot2KnownTitleId)title_id, FsStorageId_None, 0, NULL);
}
}
}
closedir(titles_dir);
}
/* We no longer need the SD card. */
fsdevUnmountAll();
}