mirror of
https://github.com/Atmosphere-NX/Atmosphere.git
synced 2024-12-05 18:12:05 +00:00
91 lines
4.3 KiB
C++
91 lines
4.3 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_overlay_manager.hpp"
|
|
#include "fusee_external_package.hpp"
|
|
#include "fusee_fatal.hpp"
|
|
|
|
namespace ams::nxboot {
|
|
|
|
namespace {
|
|
|
|
constinit u8 g_secmon_debug_storage[secmon::MemoryRegionPhysicalIramSecureMonitorDebug.GetSize()];
|
|
|
|
ALWAYS_INLINE void *GetOverlayDestination() {
|
|
return reinterpret_cast<void *>(0x4002C000);
|
|
}
|
|
|
|
void LoadMemoryTrainingOverlay(fs::FileHandle archive_file) {
|
|
Result result;
|
|
u32 verif_hash;
|
|
u32 store_hash;
|
|
if (fuse::GetSocType() == fuse::SocType_Erista) {
|
|
result = fs::ReadFile(archive_file, AMS_OFFSETOF(ExternalPackage, ovl_mtc_erista), GetOverlayDestination(), sizeof(ExternalPackage{}.ovl_mtc_erista));
|
|
verif_hash = reinterpret_cast<const u32 *>(GetOverlayDestination())[-2];
|
|
store_hash = reinterpret_cast<const u32 *>(GetOverlayDestination())[(sizeof(ExternalPackage{}.ovl_mtc_erista) / sizeof(u32)) - 1];
|
|
} else /* if (fuse::GetSocType() == fuse::SocType_Mariko) */ {
|
|
result = fs::ReadFile(archive_file, AMS_OFFSETOF(ExternalPackage, ovl_mtc_mariko), GetOverlayDestination(), sizeof(ExternalPackage{}.ovl_mtc_mariko));
|
|
verif_hash = reinterpret_cast<const u32 *>(GetOverlayDestination())[-1];
|
|
store_hash = reinterpret_cast<const u32 *>(GetOverlayDestination())[(sizeof(ExternalPackage{}.ovl_mtc_mariko) / sizeof(u32)) - 1];
|
|
}
|
|
|
|
if (R_FAILED(result)) {
|
|
ShowFatalError("Failed to load MTC overlay: 0x%08" PRIx32 "\n", result.GetValue());
|
|
}
|
|
|
|
if (verif_hash != store_hash) {
|
|
ShowFatalError("Incorrect fusee version! (program=0x%08" PRIx32 ", mtc=0x%08" PRIx32 ")\n", verif_hash, store_hash);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
void LoadOverlay(fs::FileHandle archive_file, OverlayId ovl) {
|
|
switch (ovl) {
|
|
case OverlayId_MemoryTraining:
|
|
LoadMemoryTrainingOverlay(archive_file);
|
|
break;
|
|
}
|
|
}
|
|
|
|
void SaveMemoryTrainingOverlay() {
|
|
if (fuse::GetSocType() == fuse::SocType_Erista) {
|
|
/* NOTE: Erista does not do memory clock restoration. */
|
|
/* std::memcpy(const_cast<u8 *>(GetExternalPackage().ovl_mtc_erista), GetOverlayDestination(), sizeof(ExternalPackage{}.ovl_mtc_erista)); */
|
|
} else /* if (fuse::GetSocType() == fuse::SocType_Mariko) */ {
|
|
std::memcpy(const_cast<u8 *>(GetExternalPackage().ovl_mtc_mariko), GetOverlayDestination(), sizeof(ExternalPackage{}.ovl_mtc_mariko) - 0x2000);
|
|
}
|
|
}
|
|
|
|
void RestoreMemoryTrainingOverlay() {
|
|
if (fuse::GetSocType() == fuse::SocType_Erista) {
|
|
/* NOTE: Erista does not do memory clock restoration. */
|
|
/* std::memcpy(GetOverlayDestination(), GetExternalPackage().ovl_mtc_erista, sizeof(ExternalPackage{}.ovl_mtc_erista)); */
|
|
} else /* if (fuse::GetSocType() == fuse::SocType_Mariko) */ {
|
|
std::memcpy(g_secmon_debug_storage, secmon::MemoryRegionPhysicalIramSecureMonitorDebug.GetPointer<void>(), sizeof(g_secmon_debug_storage));
|
|
std::memcpy(GetOverlayDestination(), GetExternalPackage().ovl_mtc_mariko, sizeof(ExternalPackage{}.ovl_mtc_mariko) - 0x2000);
|
|
}
|
|
}
|
|
|
|
void RestoreSecureMonitorOverlay() {
|
|
if (fuse::GetSocType() == fuse::SocType_Erista) {
|
|
/* NOTE: Erista does not do memory clock restoration. */
|
|
} else /* if (fuse::GetSocType() == fuse::SocType_Mariko) */ {
|
|
std::memcpy(secmon::MemoryRegionPhysicalIramSecureMonitorDebug.GetPointer<void>(), g_secmon_debug_storage, sizeof(g_secmon_debug_storage));
|
|
}
|
|
}
|
|
|
|
}
|