Atmosphere/fusee/program/source/fusee_fatal.cpp
2022-03-26 00:14:36 -07:00

125 lines
4.2 KiB
C++

/*
* Copyright (c) Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
* more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include <exosphere.hpp>
#include "fusee_fatal.hpp"
#include "fusee_external_package.hpp"
#include "fs/fusee_fs_api.hpp"
namespace ams::nxboot {
namespace {
constexpr inline const uintptr_t PMC = secmon::MemoryRegionPhysicalDevicePmc.GetAddress();
Result SaveFatalErrorContext(const ams::impl::FatalErrorContext *ctx) {
/* Create and open the file. */
fs::FileHandle file;
{
/* Generate the file path. */
char path[0x40];
util::TSNPrintf(path, sizeof(path), "sdmc:/atmosphere/fatal_errors/report_%016" PRIx64 ".bin", ctx->report_identifier);
/* Create the file. */
R_TRY(fs::CreateFile(path, sizeof(*ctx)));
/* Open the file. */
R_TRY(fs::OpenFile(std::addressof(file), path, fs::OpenMode_ReadWrite));
}
/* Ensure we close the file when done with it. */
ON_SCOPE_EXIT { fs::CloseFile(file); };
/* Write the context to the file. */
R_TRY(fs::WriteFile(file, 0, ctx, sizeof(*ctx), fs::WriteOption::Flush));
R_SUCCEED();
}
}
NORETURN void RebootToSelf() {
/* Patch SDRAM init to perform an SVC immediately after second write. */
reg::Write(PMC + APBDEV_PMC_SCRATCH45, 0x2E38DFFF);
reg::Write(PMC + APBDEV_PMC_SCRATCH46, 0x6001DC28);
/* Set SVC handler to jump to reboot stub in IRAM. */
reg::Write(PMC + APBDEV_PMC_SCRATCH33, 0x4003F000);
reg::Write(PMC + APBDEV_PMC_SCRATCH40, 0x6000F208);
/* Set boot as warmboot. */
reg::Write(PMC + APBDEV_PMC_SCRATCH0, (1 << 0));
/* Copy reboot stub into high IRAM. */
std::memcpy(reinterpret_cast<void *>(0x4003F000), GetExternalPackage().reboot_stub, sizeof(GetExternalPackage().reboot_stub));
/* Copy our main payload into low IRAM. */
std::memcpy(reinterpret_cast<void *>(0x40010000), GetExternalPackage().fusee, sizeof(GetExternalPackage().fusee));
/* Reboot. */
reg::Write(PMC + APBDEV_PMC_CNTRL, PMC_REG_BITS_ENUM(CNTRL_MAIN_RESET, ENABLE));
/* Wait for the reboot to take. */
AMS_INFINITE_LOOP();
}
void SaveAndShowFatalError() {
/* Get the context (at static location in memory). */
ams::impl::FatalErrorContext *f_ctx = reinterpret_cast<ams::impl::FatalErrorContext *>(0x4003E000);
/* Check for valid magic. */
if (f_ctx->magic != ams::impl::FatalErrorContext::Magic) {
return;
}
/* Show the fatal error. */
ShowFatalError(f_ctx, SaveFatalErrorContext(f_ctx));
/* Clear the magic. */
f_ctx->magic = ~f_ctx->magic;
/* Wait for reboot. */
WaitForReboot();
}
void WaitForReboot() {
/* Wait for power button to be pressed. */
while (!pmic::IsPowerButtonPressed()) {
util::WaitMicroSeconds(100);
}
/* If not erista, just do a normal reboot. */
if (fuse::GetSocType() != fuse::SocType_Erista) {
/* Reboot. */
pmic::ShutdownSystem(true);
/* Wait for our reboot to complete. */
AMS_INFINITE_LOOP();
}
/* Reboot to self, if we can. */
if (GetExternalPackage().header.magic == ExternalPackageHeader::Magic) {
RebootToSelf();
} else {
/* Just do a normal reboot. */
pmic::ShutdownSystem(true);
/* Wait for our reboot to complete. */
AMS_INFINITE_LOOP();
}
}
}