2022-04-23 10:59:50 +02:00
|
|
|
// SPDX-FileCopyrightText: Copyright 2018 yuzu Emulator Project
|
|
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
2018-08-25 17:49:31 +02:00
|
|
|
|
|
|
|
#include <vector>
|
|
|
|
|
|
|
|
#include "common/common_types.h"
|
2019-04-22 23:56:56 +02:00
|
|
|
#include "core/core.h"
|
2018-08-25 17:49:31 +02:00
|
|
|
#include "core/file_sys/content_archive.h"
|
|
|
|
#include "core/file_sys/control_metadata.h"
|
2018-09-04 20:44:40 +02:00
|
|
|
#include "core/file_sys/nca_metadata.h"
|
2018-09-04 00:57:52 +02:00
|
|
|
#include "core/file_sys/patch_manager.h"
|
2018-10-05 14:53:45 +02:00
|
|
|
#include "core/file_sys/registered_cache.h"
|
2018-08-25 17:49:31 +02:00
|
|
|
#include "core/file_sys/submission_package.h"
|
2021-04-24 07:04:28 +02:00
|
|
|
#include "core/hle/kernel/k_process.h"
|
2018-10-05 14:53:45 +02:00
|
|
|
#include "core/hle/service/filesystem/filesystem.h"
|
2018-08-25 17:49:31 +02:00
|
|
|
#include "core/loader/deconstructed_rom_directory.h"
|
|
|
|
#include "core/loader/nca.h"
|
|
|
|
#include "core/loader/nsp.h"
|
|
|
|
|
|
|
|
namespace Loader {
|
|
|
|
|
2021-04-27 18:05:34 +02:00
|
|
|
AppLoader_NSP::AppLoader_NSP(FileSys::VirtualFile file_,
|
2020-11-18 13:53:10 +01:00
|
|
|
const Service::FileSystem::FileSystemController& fsc,
|
2021-07-20 07:10:05 +02:00
|
|
|
const FileSys::ContentProvider& content_provider, u64 program_id,
|
2020-11-25 00:16:24 +01:00
|
|
|
std::size_t program_index)
|
2021-07-20 07:10:05 +02:00
|
|
|
: AppLoader(file_), nsp(std::make_unique<FileSys::NSP>(file_, program_id, program_index)) {
|
2018-08-26 04:42:54 +02:00
|
|
|
|
2020-11-18 13:53:10 +01:00
|
|
|
if (nsp->GetStatus() != ResultStatus::Success) {
|
2018-08-25 17:49:31 +02:00
|
|
|
return;
|
2020-11-18 13:53:10 +01:00
|
|
|
}
|
2018-10-27 16:16:29 +02:00
|
|
|
|
|
|
|
if (nsp->IsExtractedType()) {
|
2023-09-14 20:34:05 +02:00
|
|
|
secondary_loader = std::make_unique<AppLoader_DeconstructedRomDirectory>(
|
|
|
|
nsp->GetExeFS(), false, file->GetName() == "hbl.nsp");
|
2018-10-27 16:16:29 +02:00
|
|
|
} else {
|
2019-06-10 06:21:41 +02:00
|
|
|
const auto control_nca =
|
|
|
|
nsp->GetNCA(nsp->GetProgramTitleID(), FileSys::ContentRecordType::Control);
|
2020-11-18 13:53:10 +01:00
|
|
|
if (control_nca == nullptr || control_nca->GetStatus() != ResultStatus::Success) {
|
2019-06-10 06:21:41 +02:00
|
|
|
return;
|
2020-11-18 13:53:10 +01:00
|
|
|
}
|
2019-06-10 06:21:41 +02:00
|
|
|
|
2020-11-18 13:53:10 +01:00
|
|
|
std::tie(nacp_file, icon_file) = [this, &content_provider, &control_nca, &fsc] {
|
|
|
|
const FileSys::PatchManager pm{nsp->GetProgramTitleID(), fsc, content_provider};
|
|
|
|
return pm.ParseControlNCA(*control_nca);
|
|
|
|
}();
|
2019-06-10 06:21:41 +02:00
|
|
|
|
2018-10-27 16:16:29 +02:00
|
|
|
secondary_loader = std::make_unique<AppLoader_NCA>(
|
2021-07-20 07:10:05 +02:00
|
|
|
nsp->GetNCAFile(nsp->GetProgramTitleID(), FileSys::ContentRecordType::Program));
|
2018-10-27 16:16:29 +02:00
|
|
|
}
|
2018-08-25 17:49:31 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
AppLoader_NSP::~AppLoader_NSP() = default;
|
|
|
|
|
2021-04-27 18:05:34 +02:00
|
|
|
FileType AppLoader_NSP::IdentifyType(const FileSys::VirtualFile& nsp_file) {
|
|
|
|
const FileSys::NSP nsp(nsp_file);
|
2018-08-25 17:49:31 +02:00
|
|
|
|
|
|
|
if (nsp.GetStatus() == ResultStatus::Success) {
|
|
|
|
// Extracted Type case
|
|
|
|
if (nsp.IsExtractedType() && nsp.GetExeFS() != nullptr &&
|
2019-06-10 06:21:41 +02:00
|
|
|
FileSys::IsDirectoryExeFS(nsp.GetExeFS())) {
|
2018-08-25 17:49:31 +02:00
|
|
|
return FileType::NSP;
|
2018-08-26 04:42:54 +02:00
|
|
|
}
|
2018-08-25 17:49:31 +02:00
|
|
|
|
2019-06-10 06:21:41 +02:00
|
|
|
// Non-Extracted Type case
|
2021-07-20 07:10:05 +02:00
|
|
|
const auto program_id = nsp.GetProgramTitleID();
|
2018-08-25 17:49:31 +02:00
|
|
|
if (!nsp.IsExtractedType() &&
|
2021-07-20 07:10:05 +02:00
|
|
|
nsp.GetNCA(program_id, FileSys::ContentRecordType::Program) != nullptr &&
|
|
|
|
AppLoader_NCA::IdentifyType(
|
|
|
|
nsp.GetNCAFile(program_id, FileSys::ContentRecordType::Program)) == FileType::NCA) {
|
2018-08-25 17:49:31 +02:00
|
|
|
return FileType::NSP;
|
2018-08-26 04:42:54 +02:00
|
|
|
}
|
2018-08-25 17:49:31 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
return FileType::Error;
|
|
|
|
}
|
|
|
|
|
2021-04-24 07:04:28 +02:00
|
|
|
AppLoader_NSP::LoadResult AppLoader_NSP::Load(Kernel::KProcess& process, Core::System& system) {
|
2018-08-25 17:49:31 +02:00
|
|
|
if (is_loaded) {
|
2019-04-09 23:03:04 +02:00
|
|
|
return {ResultStatus::ErrorAlreadyLoaded, {}};
|
2018-08-25 17:49:31 +02:00
|
|
|
}
|
|
|
|
|
2021-07-20 07:10:05 +02:00
|
|
|
const auto title_id = nsp->GetProgramTitleID();
|
|
|
|
|
2019-06-10 06:19:23 +02:00
|
|
|
if (!nsp->IsExtractedType() && title_id == 0) {
|
2019-04-09 23:03:04 +02:00
|
|
|
return {ResultStatus::ErrorNSPMissingProgramNCA, {}};
|
|
|
|
}
|
2018-08-25 17:49:31 +02:00
|
|
|
|
2019-04-09 23:03:04 +02:00
|
|
|
const auto nsp_status = nsp->GetStatus();
|
|
|
|
if (nsp_status != ResultStatus::Success) {
|
|
|
|
return {nsp_status, {}};
|
|
|
|
}
|
2018-08-25 17:49:31 +02:00
|
|
|
|
2021-07-20 07:10:05 +02:00
|
|
|
const auto nsp_program_status = nsp->GetProgramStatus();
|
2019-04-09 23:03:04 +02:00
|
|
|
if (nsp_program_status != ResultStatus::Success) {
|
|
|
|
return {nsp_program_status, {}};
|
|
|
|
}
|
2018-08-25 17:49:31 +02:00
|
|
|
|
2019-06-10 06:21:41 +02:00
|
|
|
if (!nsp->IsExtractedType() &&
|
|
|
|
nsp->GetNCA(title_id, FileSys::ContentRecordType::Program) == nullptr) {
|
2019-04-09 23:03:04 +02:00
|
|
|
if (!Core::Crypto::KeyManager::KeyFileExists(false)) {
|
|
|
|
return {ResultStatus::ErrorMissingProductionKeyFile, {}};
|
|
|
|
}
|
|
|
|
|
|
|
|
return {ResultStatus::ErrorNSPMissingProgramNCA, {}};
|
2018-08-25 17:49:31 +02:00
|
|
|
}
|
|
|
|
|
2020-09-16 14:19:25 +02:00
|
|
|
const auto result = secondary_loader->Load(process, system);
|
2019-04-09 23:03:04 +02:00
|
|
|
if (result.first != ResultStatus::Success) {
|
2018-08-25 17:49:31 +02:00
|
|
|
return result;
|
2019-04-09 23:03:04 +02:00
|
|
|
}
|
2018-08-25 17:49:31 +02:00
|
|
|
|
2018-10-05 14:53:45 +02:00
|
|
|
FileSys::VirtualFile update_raw;
|
2019-04-09 23:03:04 +02:00
|
|
|
if (ReadUpdateRaw(update_raw) == ResultStatus::Success && update_raw != nullptr) {
|
2020-09-16 14:19:25 +02:00
|
|
|
system.GetFileSystemController().SetPackedUpdate(std::move(update_raw));
|
2019-04-09 23:03:04 +02:00
|
|
|
}
|
2018-10-05 14:53:45 +02:00
|
|
|
|
2018-08-25 17:49:31 +02:00
|
|
|
is_loaded = true;
|
2019-04-09 23:03:04 +02:00
|
|
|
return result;
|
2018-08-25 17:49:31 +02:00
|
|
|
}
|
|
|
|
|
2023-09-06 07:06:03 +02:00
|
|
|
ResultStatus AppLoader_NSP::VerifyIntegrity(std::function<bool(size_t, size_t)> progress_callback) {
|
|
|
|
// Extracted-type NSPs can't be verified.
|
|
|
|
if (nsp->IsExtractedType()) {
|
|
|
|
return ResultStatus::ErrorIntegrityVerificationNotImplemented;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Get list of all NCAs.
|
|
|
|
const auto ncas = nsp->GetNCAsCollapsed();
|
|
|
|
|
|
|
|
size_t total_size = 0;
|
|
|
|
size_t processed_size = 0;
|
|
|
|
|
|
|
|
// Loop over NCAs, collecting the total size to verify.
|
|
|
|
for (const auto& nca : ncas) {
|
|
|
|
total_size += nca->GetBaseFile()->GetSize();
|
|
|
|
}
|
|
|
|
|
|
|
|
// Loop over NCAs again, verifying each.
|
|
|
|
for (const auto& nca : ncas) {
|
|
|
|
AppLoader_NCA loader_nca(nca->GetBaseFile());
|
|
|
|
|
|
|
|
const auto NcaProgressCallback = [&](size_t nca_processed_size, size_t nca_total_size) {
|
|
|
|
return progress_callback(processed_size + nca_processed_size, total_size);
|
|
|
|
};
|
|
|
|
|
|
|
|
const auto verification_result = loader_nca.VerifyIntegrity(NcaProgressCallback);
|
|
|
|
if (verification_result != ResultStatus::Success) {
|
|
|
|
return verification_result;
|
|
|
|
}
|
|
|
|
|
|
|
|
processed_size += nca->GetBaseFile()->GetSize();
|
|
|
|
}
|
|
|
|
|
|
|
|
return ResultStatus::Success;
|
|
|
|
}
|
|
|
|
|
2021-04-27 18:05:34 +02:00
|
|
|
ResultStatus AppLoader_NSP::ReadRomFS(FileSys::VirtualFile& out_file) {
|
|
|
|
return secondary_loader->ReadRomFS(out_file);
|
2018-09-25 15:17:14 +02:00
|
|
|
}
|
|
|
|
|
2021-04-27 18:05:34 +02:00
|
|
|
ResultStatus AppLoader_NSP::ReadUpdateRaw(FileSys::VirtualFile& out_file) {
|
|
|
|
if (nsp->IsExtractedType()) {
|
2018-09-25 15:18:55 +02:00
|
|
|
return ResultStatus::ErrorNoPackedUpdate;
|
2021-04-27 18:05:34 +02:00
|
|
|
}
|
2018-09-25 15:18:55 +02:00
|
|
|
|
2021-07-20 07:10:05 +02:00
|
|
|
const auto read = nsp->GetNCAFile(FileSys::GetUpdateTitleID(nsp->GetProgramTitleID()),
|
|
|
|
FileSys::ContentRecordType::Program);
|
2018-09-25 15:18:55 +02:00
|
|
|
|
2021-04-27 18:05:34 +02:00
|
|
|
if (read == nullptr) {
|
2018-09-25 15:18:55 +02:00
|
|
|
return ResultStatus::ErrorNoPackedUpdate;
|
2021-04-27 18:05:34 +02:00
|
|
|
}
|
2018-09-25 15:18:55 +02:00
|
|
|
|
2021-04-27 18:05:34 +02:00
|
|
|
const auto nca_test = std::make_shared<FileSys::NCA>(read);
|
|
|
|
if (nca_test->GetStatus() != ResultStatus::ErrorMissingBKTRBaseRomFS) {
|
2018-09-25 15:18:55 +02:00
|
|
|
return nca_test->GetStatus();
|
2021-04-27 18:05:34 +02:00
|
|
|
}
|
2018-09-25 15:18:55 +02:00
|
|
|
|
2021-04-27 18:05:34 +02:00
|
|
|
out_file = read;
|
2018-09-25 15:18:55 +02:00
|
|
|
return ResultStatus::Success;
|
|
|
|
}
|
|
|
|
|
2018-08-25 17:49:31 +02:00
|
|
|
ResultStatus AppLoader_NSP::ReadProgramId(u64& out_program_id) {
|
2021-07-20 07:10:05 +02:00
|
|
|
out_program_id = nsp->GetProgramTitleID();
|
|
|
|
if (out_program_id == 0) {
|
2018-08-25 17:49:31 +02:00
|
|
|
return ResultStatus::ErrorNotInitialized;
|
2021-04-27 18:05:34 +02:00
|
|
|
}
|
2021-07-20 07:10:05 +02:00
|
|
|
return ResultStatus::Success;
|
|
|
|
}
|
2021-04-27 18:05:34 +02:00
|
|
|
|
2021-07-20 07:10:05 +02:00
|
|
|
ResultStatus AppLoader_NSP::ReadProgramIds(std::vector<u64>& out_program_ids) {
|
|
|
|
out_program_ids = nsp->GetProgramTitleIDs();
|
2018-08-25 17:49:31 +02:00
|
|
|
return ResultStatus::Success;
|
|
|
|
}
|
|
|
|
|
|
|
|
ResultStatus AppLoader_NSP::ReadIcon(std::vector<u8>& buffer) {
|
2021-04-27 18:05:34 +02:00
|
|
|
if (icon_file == nullptr) {
|
2018-08-25 17:49:31 +02:00
|
|
|
return ResultStatus::ErrorNoControl;
|
2021-04-27 18:05:34 +02:00
|
|
|
}
|
|
|
|
|
2018-08-25 17:49:31 +02:00
|
|
|
buffer = icon_file->ReadAllBytes();
|
|
|
|
return ResultStatus::Success;
|
|
|
|
}
|
|
|
|
|
|
|
|
ResultStatus AppLoader_NSP::ReadTitle(std::string& title) {
|
2021-04-27 18:05:34 +02:00
|
|
|
if (nacp_file == nullptr) {
|
2018-08-25 17:49:31 +02:00
|
|
|
return ResultStatus::ErrorNoControl;
|
2021-04-27 18:05:34 +02:00
|
|
|
}
|
|
|
|
|
2018-08-25 17:49:31 +02:00
|
|
|
title = nacp_file->GetApplicationName();
|
|
|
|
return ResultStatus::Success;
|
|
|
|
}
|
2018-11-28 20:01:56 +01:00
|
|
|
|
2018-12-07 02:27:50 +01:00
|
|
|
ResultStatus AppLoader_NSP::ReadControlData(FileSys::NACP& nacp) {
|
2021-04-27 18:05:34 +02:00
|
|
|
if (nacp_file == nullptr) {
|
2018-11-28 20:01:56 +01:00
|
|
|
return ResultStatus::ErrorNoControl;
|
2021-04-27 18:05:34 +02:00
|
|
|
}
|
|
|
|
|
2018-12-07 02:27:50 +01:00
|
|
|
nacp = *nacp_file;
|
2018-11-28 20:01:56 +01:00
|
|
|
return ResultStatus::Success;
|
|
|
|
}
|
2018-12-24 22:19:41 +01:00
|
|
|
|
2021-04-27 18:05:34 +02:00
|
|
|
ResultStatus AppLoader_NSP::ReadManualRomFS(FileSys::VirtualFile& out_file) {
|
2019-07-02 01:57:23 +02:00
|
|
|
const auto nca =
|
|
|
|
nsp->GetNCA(nsp->GetProgramTitleID(), FileSys::ContentRecordType::HtmlDocument);
|
2021-04-27 18:05:34 +02:00
|
|
|
if (nsp->GetStatus() != ResultStatus::Success || nca == nullptr) {
|
2018-12-24 22:19:41 +01:00
|
|
|
return ResultStatus::ErrorNoRomFS;
|
2021-04-27 18:05:34 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
out_file = nca->GetRomFS();
|
|
|
|
return out_file == nullptr ? ResultStatus::ErrorNoRomFS : ResultStatus::Success;
|
2018-12-24 22:19:41 +01:00
|
|
|
}
|
2019-01-15 21:56:52 +01:00
|
|
|
|
|
|
|
ResultStatus AppLoader_NSP::ReadBanner(std::vector<u8>& buffer) {
|
|
|
|
return secondary_loader->ReadBanner(buffer);
|
|
|
|
}
|
|
|
|
|
|
|
|
ResultStatus AppLoader_NSP::ReadLogo(std::vector<u8>& buffer) {
|
|
|
|
return secondary_loader->ReadLogo(buffer);
|
|
|
|
}
|
|
|
|
|
2019-05-26 17:40:41 +02:00
|
|
|
ResultStatus AppLoader_NSP::ReadNSOModules(Modules& modules) {
|
|
|
|
return secondary_loader->ReadNSOModules(modules);
|
|
|
|
}
|
|
|
|
|
2018-08-25 17:49:31 +02:00
|
|
|
} // namespace Loader
|