From bc4bec8a60720b1f911fa960ab633eec4afc975e Mon Sep 17 00:00:00 2001 From: Zach Hilman Date: Mon, 1 Oct 2018 09:12:14 -0400 Subject: fsmitm_romfsbuild: Add support for stubbing and IPS patches in LFS --- src/core/file_sys/fsmitm_romfsbuild.cpp | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'src/core/file_sys/fsmitm_romfsbuild.cpp') diff --git a/src/core/file_sys/fsmitm_romfsbuild.cpp b/src/core/file_sys/fsmitm_romfsbuild.cpp index 2a913ce82..4aa8ae369 100644 --- a/src/core/file_sys/fsmitm_romfsbuild.cpp +++ b/src/core/file_sys/fsmitm_romfsbuild.cpp @@ -26,6 +26,7 @@ #include "common/alignment.h" #include "common/assert.h" #include "core/file_sys/fsmitm_romfsbuild.h" +#include "core/file_sys/ips_layer.h" #include "core/file_sys/vfs.h" #include "core/file_sys/vfs_vector.h" @@ -138,6 +139,9 @@ void RomFSBuildContext::VisitDirectory(VirtualDir root_romfs, for (const auto& kv : entries) { if (kv.second == VfsEntryType::Directory) { + if (dir->GetSubdirectory(kv.first + ".stub") != nullptr) + continue; + const auto child = std::make_shared(); // Set child's path. child->cur_path_ofs = parent->path_len + 1; @@ -151,6 +155,9 @@ void RomFSBuildContext::VisitDirectory(VirtualDir root_romfs, child_dirs.push_back(child); } } else { + if (dir->GetFile(kv.first + ".stub") != nullptr) + continue; + const auto child = std::make_shared(); // Set child's path. child->cur_path_ofs = parent->path_len + 1; @@ -162,6 +169,13 @@ void RomFSBuildContext::VisitDirectory(VirtualDir root_romfs, child->source = root_romfs->GetFileRelative(child->path); + if (dir->GetFile(kv.first + ".ips") != nullptr) { + const auto ips = dir->GetFile(kv.first + ".ips"); + auto patched = PatchIPS(child->source, ips); + if (patched != nullptr) + child->source = std::move(patched); + } + child->size = child->source->GetSize(); AddFile(parent, child); -- cgit v1.2.3 From c1e069c066a609103c6b880585738ecabe48ddd2 Mon Sep 17 00:00:00 2001 From: Zach Hilman Date: Tue, 2 Oct 2018 08:56:56 -0400 Subject: fsmitm_romfsbuild: Extract stubs and IPS to romfs_ext dir --- src/core/file_sys/fsmitm_romfsbuild.cpp | 34 ++++++++++++++++++--------------- 1 file changed, 19 insertions(+), 15 deletions(-) (limited to 'src/core/file_sys/fsmitm_romfsbuild.cpp') diff --git a/src/core/file_sys/fsmitm_romfsbuild.cpp b/src/core/file_sys/fsmitm_romfsbuild.cpp index 4aa8ae369..47b7526c7 100644 --- a/src/core/file_sys/fsmitm_romfsbuild.cpp +++ b/src/core/file_sys/fsmitm_romfsbuild.cpp @@ -124,7 +124,7 @@ static u64 romfs_get_hash_table_count(u64 num_entries) { return count; } -void RomFSBuildContext::VisitDirectory(VirtualDir root_romfs, +void RomFSBuildContext::VisitDirectory(VirtualDir root_romfs, VirtualDir ext, std::shared_ptr parent) { std::vector> child_dirs; @@ -139,15 +139,15 @@ void RomFSBuildContext::VisitDirectory(VirtualDir root_romfs, for (const auto& kv : entries) { if (kv.second == VfsEntryType::Directory) { - if (dir->GetSubdirectory(kv.first + ".stub") != nullptr) - continue; - const auto child = std::make_shared(); // Set child's path. child->cur_path_ofs = parent->path_len + 1; child->path_len = child->cur_path_ofs + static_cast(kv.first.size()); child->path = parent->path + "/" + kv.first; + if (ext != nullptr && ext->GetFileRelative(child->path + ".stub") != nullptr) + continue; + // Sanity check on path_len ASSERT(child->path_len < FS_MAX_PATH); @@ -155,25 +155,28 @@ void RomFSBuildContext::VisitDirectory(VirtualDir root_romfs, child_dirs.push_back(child); } } else { - if (dir->GetFile(kv.first + ".stub") != nullptr) - continue; - const auto child = std::make_shared(); // Set child's path. child->cur_path_ofs = parent->path_len + 1; child->path_len = child->cur_path_ofs + static_cast(kv.first.size()); child->path = parent->path + "/" + kv.first; + if (ext != nullptr && ext->GetFileRelative(child->path + ".stub") != nullptr) + continue; + // Sanity check on path_len ASSERT(child->path_len < FS_MAX_PATH); child->source = root_romfs->GetFileRelative(child->path); - if (dir->GetFile(kv.first + ".ips") != nullptr) { - const auto ips = dir->GetFile(kv.first + ".ips"); - auto patched = PatchIPS(child->source, ips); - if (patched != nullptr) - child->source = std::move(patched); + if (ext != nullptr) { + const auto ips = ext->GetFileRelative(child->path + ".ips"); + + if (ips != nullptr) { + auto patched = PatchIPS(child->source, ips); + if (patched != nullptr) + child->source = std::move(patched); + } } child->size = child->source->GetSize(); @@ -183,7 +186,7 @@ void RomFSBuildContext::VisitDirectory(VirtualDir root_romfs, } for (auto& child : child_dirs) { - this->VisitDirectory(root_romfs, child); + this->VisitDirectory(root_romfs, ext, child); } } @@ -222,14 +225,15 @@ bool RomFSBuildContext::AddFile(std::shared_ptr pare return true; } -RomFSBuildContext::RomFSBuildContext(VirtualDir base_) : base(std::move(base_)) { +RomFSBuildContext::RomFSBuildContext(VirtualDir base_, VirtualDir ext_) + : base(std::move(base_)), ext(std::move(ext_)) { root = std::make_shared(); root->path = "\0"; directories.emplace(root->path, root); num_dirs = 1; dir_table_size = 0x18; - VisitDirectory(base, root); + VisitDirectory(base, ext, root); } RomFSBuildContext::~RomFSBuildContext() = default; -- cgit v1.2.3