Merge pull request #782 from lioncash/file

loader/nro: Minor changes
This commit is contained in:
bunnei 2018-07-23 19:06:30 -07:00 committed by GitHub
commit 10dd03dec5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 33 additions and 14 deletions

View File

@ -6,6 +6,7 @@
#include <vector> #include <vector>
#include "common/common_funcs.h" #include "common/common_funcs.h"
#include "common/common_types.h"
#include "common/file_util.h" #include "common/file_util.h"
#include "common/logging/log.h" #include "common/logging/log.h"
#include "common/swap.h" #include "common/swap.h"
@ -68,22 +69,27 @@ static_assert(sizeof(AssetHeader) == 0x38, "AssetHeader has incorrect size.");
AppLoader_NRO::AppLoader_NRO(FileSys::VirtualFile file) : AppLoader(file) { AppLoader_NRO::AppLoader_NRO(FileSys::VirtualFile file) : AppLoader(file) {
NroHeader nro_header{}; NroHeader nro_header{};
if (file->ReadObject(&nro_header) != sizeof(NroHeader)) if (file->ReadObject(&nro_header) != sizeof(NroHeader)) {
return; return;
}
if (file->GetSize() >= nro_header.file_size + sizeof(AssetHeader)) { if (file->GetSize() >= nro_header.file_size + sizeof(AssetHeader)) {
u64 offset = nro_header.file_size; const u64 offset = nro_header.file_size;
AssetHeader asset_header{}; AssetHeader asset_header{};
if (file->ReadObject(&asset_header, offset) != sizeof(AssetHeader)) if (file->ReadObject(&asset_header, offset) != sizeof(AssetHeader)) {
return; return;
}
if (asset_header.format_version != 0) if (asset_header.format_version != 0) {
LOG_WARNING(Loader, LOG_WARNING(Loader,
"NRO Asset Header has format {}, currently supported format is 0. If " "NRO Asset Header has format {}, currently supported format is 0. If "
"strange glitches occur with metadata, check NRO assets.", "strange glitches occur with metadata, check NRO assets.",
asset_header.format_version); asset_header.format_version);
if (asset_header.magic != Common::MakeMagic('A', 'S', 'E', 'T')) }
if (asset_header.magic != Common::MakeMagic('A', 'S', 'E', 'T')) {
return; return;
}
if (asset_header.nacp.size > 0) { if (asset_header.nacp.size > 0) {
nacp = std::make_unique<FileSys::NACP>(std::make_shared<FileSys::OffsetVfsFile>( nacp = std::make_unique<FileSys::NACP>(std::make_shared<FileSys::OffsetVfsFile>(
@ -101,6 +107,8 @@ AppLoader_NRO::AppLoader_NRO(FileSys::VirtualFile file) : AppLoader(file) {
} }
} }
AppLoader_NRO::~AppLoader_NRO() = default;
FileType AppLoader_NRO::IdentifyType(const FileSys::VirtualFile& file) { FileType AppLoader_NRO::IdentifyType(const FileSys::VirtualFile& file) {
// Read NSO header // Read NSO header
NroHeader nro_header{}; NroHeader nro_header{};
@ -130,8 +138,9 @@ bool AppLoader_NRO::LoadNro(FileSys::VirtualFile file, VAddr load_base) {
// Build program image // Build program image
Kernel::SharedPtr<Kernel::CodeSet> codeset = Kernel::CodeSet::Create(""); Kernel::SharedPtr<Kernel::CodeSet> codeset = Kernel::CodeSet::Create("");
std::vector<u8> program_image = file->ReadBytes(PageAlignSize(nro_header.file_size)); std::vector<u8> program_image = file->ReadBytes(PageAlignSize(nro_header.file_size));
if (program_image.size() != PageAlignSize(nro_header.file_size)) if (program_image.size() != PageAlignSize(nro_header.file_size)) {
return {}; return {};
}
for (std::size_t i = 0; i < nro_header.segments.size(); ++i) { for (std::size_t i = 0; i < nro_header.segments.size(); ++i) {
codeset->segments[i].addr = nro_header.segments[i].offset; codeset->segments[i].addr = nro_header.segments[i].offset;
@ -187,29 +196,37 @@ ResultStatus AppLoader_NRO::Load(Kernel::SharedPtr<Kernel::Process>& process) {
} }
ResultStatus AppLoader_NRO::ReadIcon(std::vector<u8>& buffer) { ResultStatus AppLoader_NRO::ReadIcon(std::vector<u8>& buffer) {
if (icon_data.empty()) if (icon_data.empty()) {
return ResultStatus::ErrorNotUsed; return ResultStatus::ErrorNotUsed;
}
buffer = icon_data; buffer = icon_data;
return ResultStatus::Success; return ResultStatus::Success;
} }
ResultStatus AppLoader_NRO::ReadProgramId(u64& out_program_id) { ResultStatus AppLoader_NRO::ReadProgramId(u64& out_program_id) {
if (nacp == nullptr) if (nacp == nullptr) {
return ResultStatus::ErrorNotUsed; return ResultStatus::ErrorNotUsed;
}
out_program_id = nacp->GetTitleId(); out_program_id = nacp->GetTitleId();
return ResultStatus::Success; return ResultStatus::Success;
} }
ResultStatus AppLoader_NRO::ReadRomFS(FileSys::VirtualFile& dir) { ResultStatus AppLoader_NRO::ReadRomFS(FileSys::VirtualFile& dir) {
if (romfs == nullptr) if (romfs == nullptr) {
return ResultStatus::ErrorNotUsed; return ResultStatus::ErrorNotUsed;
}
dir = romfs; dir = romfs;
return ResultStatus::Success; return ResultStatus::Success;
} }
ResultStatus AppLoader_NRO::ReadTitle(std::string& title) { ResultStatus AppLoader_NRO::ReadTitle(std::string& title) {
if (nacp == nullptr) if (nacp == nullptr) {
return ResultStatus::ErrorNotUsed; return ResultStatus::ErrorNotUsed;
}
title = nacp->GetApplicationName(); title = nacp->GetApplicationName();
return ResultStatus::Success; return ResultStatus::Success;
} }

View File

@ -6,19 +6,21 @@
#include <string> #include <string>
#include "common/common_types.h" #include "common/common_types.h"
#include "core/file_sys/control_metadata.h"
#include "core/hle/kernel/kernel.h" #include "core/hle/kernel/kernel.h"
#include "core/loader/linker.h" #include "core/loader/linker.h"
#include "core/loader/loader.h" #include "core/loader/loader.h"
namespace Loader { namespace FileSys {
class NACP;
}
struct AssetHeader; namespace Loader {
/// Loads an NRO file /// Loads an NRO file
class AppLoader_NRO final : public AppLoader, Linker { class AppLoader_NRO final : public AppLoader, Linker {
public: public:
AppLoader_NRO(FileSys::VirtualFile file); explicit AppLoader_NRO(FileSys::VirtualFile file);
~AppLoader_NRO() override;
/** /**
* Returns the type of the file * Returns the type of the file