Preliminary support for HOS 13.0.0
- pcv/am offsets in ldr_patcher - ptm offsets in .pchtxt - drop Erista support TODOs: - Update patcher.te to TegraScript v3 - Recompile sys-clk against new libnx - Generate pcv/ptm/am patches manually
This commit is contained in:
158
Source/Atmosphere/ldr_patcher.cpp
Normal file
158
Source/Atmosphere/ldr_patcher.cpp
Normal file
@@ -0,0 +1,158 @@
|
||||
// placed in Atmosphere/stratosphere/loader/source/
|
||||
/*
|
||||
* Copyright (c) 2018-2020 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 <stratosphere.hpp>
|
||||
#include "ldr_patcher.hpp"
|
||||
#include "ldr_pcv_patch.hpp"
|
||||
|
||||
namespace ams::ldr {
|
||||
|
||||
namespace {
|
||||
|
||||
constexpr const char *NsoPatchesDirectory = "exefs_patches";
|
||||
|
||||
/* Exefs patches want to prevent modification of header, */
|
||||
/* and also want to adjust offset relative to mapped location. */
|
||||
constexpr size_t NsoPatchesProtectedSize = sizeof(NsoHeader);
|
||||
constexpr size_t NsoPatchesProtectedOffset = sizeof(NsoHeader);
|
||||
|
||||
constexpr const char * const LoaderSdMountName = "#amsldr-sdpatch";
|
||||
static_assert(sizeof(LoaderSdMountName) <= fs::MountNameLengthMax);
|
||||
|
||||
constinit os::SdkMutex g_ldr_sd_lock;
|
||||
constinit bool g_mounted_sd;
|
||||
|
||||
constinit os::SdkMutex g_embedded_patch_lock;
|
||||
constinit bool g_got_embedded_patch_settings;
|
||||
constinit bool g_force_enable_usb30;
|
||||
|
||||
bool EnsureSdCardMounted() {
|
||||
std::scoped_lock lk(g_ldr_sd_lock);
|
||||
|
||||
if (g_mounted_sd) {
|
||||
return true;
|
||||
}
|
||||
|
||||
if (!cfg::IsSdCardInitialized()) {
|
||||
return false;
|
||||
}
|
||||
|
||||
if (R_FAILED(fs::MountSdCard(LoaderSdMountName))) {
|
||||
return false;
|
||||
}
|
||||
|
||||
return (g_mounted_sd = true);
|
||||
}
|
||||
|
||||
bool IsUsb30ForceEnabled() {
|
||||
std::scoped_lock lk(g_embedded_patch_lock);
|
||||
|
||||
if (!g_got_embedded_patch_settings) {
|
||||
g_force_enable_usb30 = spl::IsUsb30ForceEnabled();
|
||||
g_got_embedded_patch_settings = true;
|
||||
}
|
||||
|
||||
return g_force_enable_usb30;
|
||||
}
|
||||
|
||||
consteval u8 ParseNybble(char c) {
|
||||
AMS_ASSUME(('0' <= c && c <= '9') || ('A' <= c && c <= 'F') || ('a' <= c && c <= 'f'));
|
||||
if ('0' <= c && c <= '9') {
|
||||
return c - '0' + 0x0;
|
||||
} else if ('A' <= c && c <= 'F') {
|
||||
return c - 'A' + 0xA;
|
||||
} else /* if ('a' <= c && c <= 'f') */ {
|
||||
return c - 'a' + 0xa;
|
||||
}
|
||||
}
|
||||
|
||||
consteval ro::ModuleId ParseModuleId(const char *str) {
|
||||
/* Parse a static module id. */
|
||||
ro::ModuleId module_id = {};
|
||||
|
||||
size_t ofs = 0;
|
||||
while (str[0] != 0) {
|
||||
AMS_ASSUME(ofs < sizeof(module_id));
|
||||
AMS_ASSUME(str[1] != 0);
|
||||
|
||||
module_id.build_id[ofs] = (ParseNybble(str[0]) << 4) | (ParseNybble(str[1]) << 0);
|
||||
|
||||
str += 2;
|
||||
ofs++;
|
||||
}
|
||||
|
||||
return module_id;
|
||||
}
|
||||
|
||||
struct EmbeddedPatchEntry {
|
||||
uintptr_t offset;
|
||||
const void * const data;
|
||||
size_t size;
|
||||
};
|
||||
|
||||
struct EmbeddedPatch {
|
||||
ro::ModuleId module_id;
|
||||
size_t num_entries;
|
||||
const EmbeddedPatchEntry *entries;
|
||||
};
|
||||
|
||||
#include "ldr_embedded_usb_patches.inc"
|
||||
|
||||
}
|
||||
|
||||
/* Apply IPS patches. */
|
||||
void LocateAndApplyIpsPatchesToModule(const u8 *build_id, uintptr_t mapped_nso, size_t mapped_size) {
|
||||
for(int i = 0; i < VERS; i++)
|
||||
{
|
||||
if(memcmp(PcvModuleId[i], build_id, sizeof(PcvModuleId[i])) == 0) {
|
||||
ApplyPcvPatch(reinterpret_cast<u8 *>(mapped_nso), mapped_size, i);
|
||||
return; // Return here since pcv module loads before sd card can be mounted
|
||||
}
|
||||
else if(memcmp(AmModuleId[i], build_id, sizeof(AmModuleId[i])) == 0) {
|
||||
ApplyCopyrightPatch(reinterpret_cast<u8 *>(mapped_nso), mapped_size, i);
|
||||
}
|
||||
}
|
||||
|
||||
if (!EnsureSdCardMounted()) {
|
||||
return;
|
||||
}
|
||||
|
||||
ro::ModuleId module_id;
|
||||
std::memcpy(&module_id.build_id, build_id, sizeof(module_id.build_id));
|
||||
ams::patcher::LocateAndApplyIpsPatchesToModule(LoaderSdMountName, NsoPatchesDirectory, NsoPatchesProtectedSize, NsoPatchesProtectedOffset, &module_id, reinterpret_cast<u8 *>(mapped_nso), mapped_size);
|
||||
}
|
||||
|
||||
/* Apply embedded patches. */
|
||||
void ApplyEmbeddedPatchesToModule(const u8 *build_id, uintptr_t mapped_nso, size_t mapped_size) {
|
||||
/* Make module id. */
|
||||
ro::ModuleId module_id;
|
||||
std::memcpy(&module_id.build_id, build_id, sizeof(module_id.build_id));
|
||||
|
||||
if (IsUsb30ForceEnabled()) {
|
||||
for (const auto &patch : Usb30ForceEnablePatches) {
|
||||
if (std::memcmp(std::addressof(patch.module_id), std::addressof(module_id), sizeof(module_id)) == 0) {
|
||||
for (size_t i = 0; i < patch.num_entries; ++i) {
|
||||
const auto &entry = patch.entries[i];
|
||||
if (entry.offset + entry.size <= mapped_size) {
|
||||
std::memcpy(reinterpret_cast<void *>(mapped_nso + entry.offset), entry.data, entry.size);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user