2018-01-20 20:59:17 +01:00
|
|
|
// Copyright 2018 yuzu emulator team
|
|
|
|
// Licensed under GPLv2 or any later version
|
|
|
|
// Refer to the license.txt file included.
|
|
|
|
|
2018-02-14 06:53:51 +01:00
|
|
|
#include <cinttypes>
|
2018-09-30 20:04:48 +02:00
|
|
|
#include <cstring>
|
2018-01-20 20:59:17 +01:00
|
|
|
#include "common/common_funcs.h"
|
2018-01-20 21:48:37 +01:00
|
|
|
#include "common/file_util.h"
|
2018-01-20 20:59:17 +01:00
|
|
|
#include "common/logging/log.h"
|
2019-04-22 23:56:56 +02:00
|
|
|
#include "core/core.h"
|
2018-07-19 03:07:11 +02:00
|
|
|
#include "core/file_sys/content_archive.h"
|
2018-07-28 18:32:16 +02:00
|
|
|
#include "core/file_sys/control_metadata.h"
|
2018-08-26 01:04:48 +02:00
|
|
|
#include "core/file_sys/patch_manager.h"
|
2018-08-21 02:36:36 +02:00
|
|
|
#include "core/file_sys/romfs_factory.h"
|
2021-02-13 02:58:31 +01:00
|
|
|
#include "core/hle/kernel/k_page_table.h"
|
2018-08-31 18:21:34 +02:00
|
|
|
#include "core/hle/kernel/kernel.h"
|
2018-01-20 20:59:17 +01:00
|
|
|
#include "core/hle/kernel/process.h"
|
2018-01-21 00:20:46 +01:00
|
|
|
#include "core/hle/service/filesystem/filesystem.h"
|
2018-01-20 20:59:17 +01:00
|
|
|
#include "core/loader/deconstructed_rom_directory.h"
|
|
|
|
#include "core/loader/nso.h"
|
|
|
|
|
|
|
|
namespace Loader {
|
|
|
|
|
2018-08-26 01:04:48 +02:00
|
|
|
AppLoader_DeconstructedRomDirectory::AppLoader_DeconstructedRomDirectory(FileSys::VirtualFile file_,
|
|
|
|
bool override_update)
|
|
|
|
: AppLoader(std::move(file_)), override_update(override_update) {
|
2018-07-28 18:32:16 +02:00
|
|
|
const auto dir = file->GetContainingDirectory();
|
|
|
|
|
2018-09-04 01:00:51 +02:00
|
|
|
// Title ID
|
|
|
|
const auto npdm = dir->GetFile("main.npdm");
|
|
|
|
if (npdm != nullptr) {
|
|
|
|
const auto res = metadata.Load(npdm);
|
|
|
|
if (res == ResultStatus::Success)
|
|
|
|
title_id = metadata.GetTitleID();
|
|
|
|
}
|
|
|
|
|
2018-07-28 18:32:16 +02:00
|
|
|
// Icon
|
|
|
|
FileSys::VirtualFile icon_file = nullptr;
|
|
|
|
for (const auto& language : FileSys::LANGUAGE_NAMES) {
|
2018-07-28 18:35:02 +02:00
|
|
|
icon_file = dir->GetFile("icon_" + std::string(language) + ".dat");
|
2018-07-28 18:32:16 +02:00
|
|
|
if (icon_file != nullptr) {
|
|
|
|
icon_data = icon_file->ReadAllBytes();
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (icon_data.empty()) {
|
|
|
|
// Any png, jpeg, or bmp file
|
|
|
|
const auto& files = dir->GetFiles();
|
|
|
|
const auto icon_iter =
|
|
|
|
std::find_if(files.begin(), files.end(), [](const FileSys::VirtualFile& file) {
|
|
|
|
return file->GetExtension() == "png" || file->GetExtension() == "jpg" ||
|
|
|
|
file->GetExtension() == "bmp" || file->GetExtension() == "jpeg";
|
|
|
|
});
|
|
|
|
if (icon_iter != files.end())
|
|
|
|
icon_data = (*icon_iter)->ReadAllBytes();
|
|
|
|
}
|
|
|
|
|
|
|
|
// Metadata
|
|
|
|
FileSys::VirtualFile nacp_file = dir->GetFile("control.nacp");
|
|
|
|
if (nacp_file == nullptr) {
|
|
|
|
const auto& files = dir->GetFiles();
|
|
|
|
const auto nacp_iter =
|
|
|
|
std::find_if(files.begin(), files.end(), [](const FileSys::VirtualFile& file) {
|
|
|
|
return file->GetExtension() == "nacp";
|
|
|
|
});
|
|
|
|
if (nacp_iter != files.end())
|
|
|
|
nacp_file = *nacp_iter;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (nacp_file != nullptr) {
|
|
|
|
FileSys::NACP nacp(nacp_file);
|
|
|
|
name = nacp.GetApplicationName();
|
|
|
|
}
|
|
|
|
}
|
2018-07-08 05:24:51 +02:00
|
|
|
|
2018-08-06 00:27:05 +02:00
|
|
|
AppLoader_DeconstructedRomDirectory::AppLoader_DeconstructedRomDirectory(
|
2018-08-26 01:04:48 +02:00
|
|
|
FileSys::VirtualDir directory, bool override_update)
|
|
|
|
: AppLoader(directory->GetFile("main")), dir(std::move(directory)),
|
|
|
|
override_update(override_update) {}
|
2018-08-06 00:27:05 +02:00
|
|
|
|
2018-07-19 03:07:11 +02:00
|
|
|
FileType AppLoader_DeconstructedRomDirectory::IdentifyType(const FileSys::VirtualFile& file) {
|
|
|
|
if (FileSys::IsDirectoryExeFS(file->GetContainingDirectory())) {
|
2018-01-20 20:59:17 +01:00
|
|
|
return FileType::DeconstructedRomDirectory;
|
|
|
|
}
|
|
|
|
|
|
|
|
return FileType::Error;
|
|
|
|
}
|
|
|
|
|
2019-04-09 23:03:04 +02:00
|
|
|
AppLoader_DeconstructedRomDirectory::LoadResult AppLoader_DeconstructedRomDirectory::Load(
|
2020-09-16 14:19:25 +02:00
|
|
|
Kernel::Process& process, Core::System& system) {
|
2018-01-20 20:59:17 +01:00
|
|
|
if (is_loaded) {
|
2019-04-09 23:03:04 +02:00
|
|
|
return {ResultStatus::ErrorAlreadyLoaded, {}};
|
2018-01-20 20:59:17 +01:00
|
|
|
}
|
|
|
|
|
2018-08-06 00:27:05 +02:00
|
|
|
if (dir == nullptr) {
|
2019-04-09 23:03:04 +02:00
|
|
|
if (file == nullptr) {
|
|
|
|
return {ResultStatus::ErrorNullFile, {}};
|
|
|
|
}
|
|
|
|
|
2018-08-07 16:37:38 +02:00
|
|
|
dir = file->GetContainingDirectory();
|
2018-08-06 00:27:05 +02:00
|
|
|
}
|
|
|
|
|
2018-08-26 01:04:48 +02:00
|
|
|
// Read meta to determine title ID
|
|
|
|
FileSys::VirtualFile npdm = dir->GetFile("main.npdm");
|
2019-04-09 23:03:04 +02:00
|
|
|
if (npdm == nullptr) {
|
|
|
|
return {ResultStatus::ErrorMissingNPDM, {}};
|
|
|
|
}
|
2018-02-25 15:01:37 +01:00
|
|
|
|
2019-04-09 23:03:04 +02:00
|
|
|
const ResultStatus result = metadata.Load(npdm);
|
2018-02-25 15:01:37 +01:00
|
|
|
if (result != ResultStatus::Success) {
|
2019-04-09 23:03:04 +02:00
|
|
|
return {result, {}};
|
2018-02-25 15:01:37 +01:00
|
|
|
}
|
2018-08-26 01:04:48 +02:00
|
|
|
|
|
|
|
if (override_update) {
|
2020-11-18 13:53:10 +01:00
|
|
|
const FileSys::PatchManager patch_manager(
|
|
|
|
metadata.GetTitleID(), system.GetFileSystemController(), system.GetContentProvider());
|
2018-08-26 01:04:48 +02:00
|
|
|
dir = patch_manager.PatchExeFS(dir);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Reread in case PatchExeFS affected the main.npdm
|
|
|
|
npdm = dir->GetFile("main.npdm");
|
2019-04-09 23:03:04 +02:00
|
|
|
if (npdm == nullptr) {
|
|
|
|
return {ResultStatus::ErrorMissingNPDM, {}};
|
|
|
|
}
|
2018-08-26 01:04:48 +02:00
|
|
|
|
2019-04-09 23:03:04 +02:00
|
|
|
const ResultStatus result2 = metadata.Load(npdm);
|
2018-08-26 01:04:48 +02:00
|
|
|
if (result2 != ResultStatus::Success) {
|
2019-04-09 23:03:04 +02:00
|
|
|
return {result2, {}};
|
2018-08-26 01:04:48 +02:00
|
|
|
}
|
2018-02-25 11:35:07 +01:00
|
|
|
metadata.Print();
|
|
|
|
|
2020-04-09 22:12:31 +02:00
|
|
|
const auto static_modules = {"rtld", "main", "subsdk0", "subsdk1", "subsdk2", "subsdk3",
|
|
|
|
"subsdk4", "subsdk5", "subsdk6", "subsdk7", "sdk"};
|
|
|
|
|
|
|
|
// Use the NSO module loader to figure out the code layout
|
|
|
|
std::size_t code_size{};
|
|
|
|
for (const auto& module : static_modules) {
|
|
|
|
const FileSys::VirtualFile module_file{dir->GetFile(module)};
|
|
|
|
if (!module_file) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
2020-09-16 14:19:25 +02:00
|
|
|
const bool should_pass_arguments = std::strcmp(module, "rtld") == 0;
|
|
|
|
const auto tentative_next_load_addr = AppLoader_NSO::LoadModule(
|
|
|
|
process, system, *module_file, code_size, should_pass_arguments, false);
|
2020-04-09 22:12:31 +02:00
|
|
|
if (!tentative_next_load_addr) {
|
|
|
|
return {ResultStatus::ErrorLoadingNSO, {}};
|
|
|
|
}
|
|
|
|
|
|
|
|
code_size = *tentative_next_load_addr;
|
2018-12-20 05:50:20 +01:00
|
|
|
}
|
|
|
|
|
2020-04-09 22:12:31 +02:00
|
|
|
// Setup the process code layout
|
|
|
|
if (process.LoadFromMetadata(metadata, code_size).IsError()) {
|
|
|
|
return {ResultStatus::ErrorUnableToParseKernelMetadata, {}};
|
|
|
|
}
|
2018-09-23 02:09:32 +02:00
|
|
|
|
2018-01-20 20:59:17 +01:00
|
|
|
// Load NSO modules
|
2019-05-26 17:40:41 +02:00
|
|
|
modules.clear();
|
2020-04-09 22:12:31 +02:00
|
|
|
const VAddr base_address{process.PageTable().GetCodeRegionStart()};
|
|
|
|
VAddr next_load_addr{base_address};
|
2020-11-18 13:53:10 +01:00
|
|
|
const FileSys::PatchManager pm{metadata.GetTitleID(), system.GetFileSystemController(),
|
|
|
|
system.GetContentProvider()};
|
2020-04-09 22:12:31 +02:00
|
|
|
for (const auto& module : static_modules) {
|
|
|
|
const FileSys::VirtualFile module_file{dir->GetFile(module)};
|
|
|
|
if (!module_file) {
|
2018-10-15 03:41:58 +02:00
|
|
|
continue;
|
2018-01-20 20:59:17 +01:00
|
|
|
}
|
2018-10-15 03:41:58 +02:00
|
|
|
|
2020-04-09 22:12:31 +02:00
|
|
|
const VAddr load_addr{next_load_addr};
|
2020-09-16 14:19:25 +02:00
|
|
|
const bool should_pass_arguments = std::strcmp(module, "rtld") == 0;
|
|
|
|
const auto tentative_next_load_addr = AppLoader_NSO::LoadModule(
|
|
|
|
process, system, *module_file, load_addr, should_pass_arguments, true, pm);
|
2018-10-15 03:41:58 +02:00
|
|
|
if (!tentative_next_load_addr) {
|
2019-04-09 23:03:04 +02:00
|
|
|
return {ResultStatus::ErrorLoadingNSO, {}};
|
2018-10-15 03:41:58 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
next_load_addr = *tentative_next_load_addr;
|
2019-05-26 17:40:41 +02:00
|
|
|
modules.insert_or_assign(load_addr, module);
|
2018-10-15 03:41:58 +02:00
|
|
|
LOG_DEBUG(Loader, "loaded module {} @ 0x{:X}", module, load_addr);
|
2018-01-20 20:59:17 +01:00
|
|
|
}
|
|
|
|
|
2018-02-05 09:01:10 +01:00
|
|
|
// Find the RomFS by searching for a ".romfs" file in this directory
|
2018-07-19 03:07:11 +02:00
|
|
|
const auto& files = dir->GetFiles();
|
|
|
|
const auto romfs_iter =
|
|
|
|
std::find_if(files.begin(), files.end(), [](const FileSys::VirtualFile& file) {
|
|
|
|
return file->GetName().find(".romfs") != std::string::npos;
|
|
|
|
});
|
2018-01-21 00:20:46 +01:00
|
|
|
|
2018-07-08 05:24:51 +02:00
|
|
|
// Register the RomFS if a ".romfs" file was found
|
2018-07-19 03:07:11 +02:00
|
|
|
if (romfs_iter != files.end() && *romfs_iter != nullptr) {
|
|
|
|
romfs = *romfs_iter;
|
2020-09-17 00:29:24 +02:00
|
|
|
system.GetFileSystemController().RegisterRomFS(std::make_unique<FileSys::RomFSFactory>(
|
|
|
|
*this, system.GetContentProvider(), system.GetFileSystemController()));
|
2018-07-08 05:24:51 +02:00
|
|
|
}
|
2018-01-21 00:20:46 +01:00
|
|
|
|
2018-07-06 16:51:32 +02:00
|
|
|
is_loaded = true;
|
2019-04-09 23:03:04 +02:00
|
|
|
return {ResultStatus::Success,
|
|
|
|
LoadParameters{metadata.GetMainThreadPriority(), metadata.GetMainThreadStackSize()}};
|
2018-01-21 00:20:46 +01:00
|
|
|
}
|
|
|
|
|
2018-07-19 03:07:11 +02:00
|
|
|
ResultStatus AppLoader_DeconstructedRomDirectory::ReadRomFS(FileSys::VirtualFile& dir) {
|
|
|
|
if (romfs == nullptr)
|
2018-08-10 03:06:44 +02:00
|
|
|
return ResultStatus::ErrorNoRomFS;
|
2018-07-19 03:07:11 +02:00
|
|
|
dir = romfs;
|
2018-07-08 05:24:51 +02:00
|
|
|
return ResultStatus::Success;
|
|
|
|
}
|
|
|
|
|
2018-07-28 18:32:16 +02:00
|
|
|
ResultStatus AppLoader_DeconstructedRomDirectory::ReadIcon(std::vector<u8>& buffer) {
|
|
|
|
if (icon_data.empty())
|
2018-08-10 03:06:44 +02:00
|
|
|
return ResultStatus::ErrorNoIcon;
|
2018-07-28 18:32:16 +02:00
|
|
|
buffer = icon_data;
|
|
|
|
return ResultStatus::Success;
|
|
|
|
}
|
|
|
|
|
|
|
|
ResultStatus AppLoader_DeconstructedRomDirectory::ReadProgramId(u64& out_program_id) {
|
|
|
|
out_program_id = title_id;
|
|
|
|
return ResultStatus::Success;
|
|
|
|
}
|
|
|
|
|
|
|
|
ResultStatus AppLoader_DeconstructedRomDirectory::ReadTitle(std::string& title) {
|
|
|
|
if (name.empty())
|
2018-08-10 03:06:44 +02:00
|
|
|
return ResultStatus::ErrorNoControl;
|
2018-07-28 18:32:16 +02:00
|
|
|
title = name;
|
|
|
|
return ResultStatus::Success;
|
|
|
|
}
|
|
|
|
|
2018-08-26 16:53:31 +02:00
|
|
|
bool AppLoader_DeconstructedRomDirectory::IsRomFSUpdatable() const {
|
2018-08-26 01:05:04 +02:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2019-05-26 17:40:41 +02:00
|
|
|
ResultStatus AppLoader_DeconstructedRomDirectory::ReadNSOModules(Modules& modules) {
|
|
|
|
if (!is_loaded) {
|
|
|
|
return ResultStatus::ErrorNotInitialized;
|
|
|
|
}
|
|
|
|
|
|
|
|
modules = this->modules;
|
|
|
|
return ResultStatus::Success;
|
|
|
|
}
|
|
|
|
|
2018-01-20 20:59:17 +01:00
|
|
|
} // namespace Loader
|