From a8c46485203d3ab00ef478bbf9daa7450df14dfd Mon Sep 17 00:00:00 2001
From: bunnei <ericbunnie@gmail.com>
Date: Thu, 19 Jun 2014 00:11:45 -0400
Subject: [PATCH] NCCH: Added RomFS loading.

---
 src/core/loader/ncch.cpp | 29 ++++++++++++++++++++++++++++-
 src/core/loader/ncch.h   |  8 ++++++++
 2 files changed, 36 insertions(+), 1 deletion(-)

diff --git a/src/core/loader/ncch.cpp b/src/core/loader/ncch.cpp
index 4cf805ba0..6423da8f9 100644
--- a/src/core/loader/ncch.cpp
+++ b/src/core/loader/ncch.cpp
@@ -178,6 +178,32 @@ const ResultStatus AppLoader_NCCH::LoadSectionExeFS(File::IOFile& file, const ch
     return ResultStatus::ErrorNotUsed;
 } 
 
+/**
+ * Reads RomFS of an NCCH file into AppLoader
+ * @param file Handle to file to read from
+ * @return ResultStatus result of function
+ */
+const ResultStatus AppLoader_NCCH::LoadRomFS(File::IOFile& file) {
+    // Check if the NCCH has a RomFS...
+    if (ncch_header.romfs_offset != 0 && ncch_header.romfs_size != 0) {
+        u32 romfs_offset = ncch_offset + (ncch_header.romfs_offset * kBlockSize) + 0x1000;
+        u32 romfs_size = (ncch_header.romfs_size * kBlockSize) - 0x1000;
+
+        INFO_LOG(LOADER, "RomFS offset:    0x%08X", romfs_offset);
+        INFO_LOG(LOADER, "RomFS size:      0x%08X", romfs_size);
+
+        romfs.resize(romfs_size);
+
+        file.Seek(romfs_offset, 0);
+        file.ReadBytes(&romfs[0], romfs_size);
+
+        return ResultStatus::Success;
+    } else {
+        NOTICE_LOG(LOADER, "RomFS unused");
+    }
+    return ResultStatus::ErrorNotUsed;
+}
+
 /**
  * Loads an NCCH file (e.g. from a CCI, or the first NCCH in a CXI)
  * @param error_string Pointer to string to put error message if an error has occurred
@@ -193,7 +219,6 @@ const ResultStatus AppLoader_NCCH::Load() {
     File::IOFile file(filename, "rb");
 
     if (file.IsOpen()) {
-        NCCH_Header ncch_header;
         file.ReadBytes(&ncch_header, sizeof(NCCH_Header));
 
         // Skip NCSD header and load first NCCH (NCSD is just a container of NCCH files)...
@@ -237,6 +262,8 @@ const ResultStatus AppLoader_NCCH::Load() {
         LoadSectionExeFS(file, "icon", icon);
         LoadSectionExeFS(file, "logo", logo);
 
+        LoadRomFS(file);
+
         is_loaded = true; // Set state to loaded
 
         LoadExec(); // Load the executable into memory for booting
diff --git a/src/core/loader/ncch.h b/src/core/loader/ncch.h
index 525a5aef5..939b144a6 100644
--- a/src/core/loader/ncch.h
+++ b/src/core/loader/ncch.h
@@ -168,6 +168,13 @@ private:
     const ResultStatus LoadSectionExeFS(File::IOFile& file, const char* name, 
         std::vector<u8>& buffer);
 
+    /**
+     * Reads RomFS of an NCCH file into AppLoader
+     * @param file Handle to file to read from
+     * @return ResultStatus result of function
+     */
+    const ResultStatus LoadRomFS(File::IOFile& file);
+
     /**
      * Loads .code section into memory for booting
      * @return ResultStatus result of function
@@ -182,6 +189,7 @@ private:
     u32             ncch_offset; // Offset to NCCH header, can be 0 or after NCSD header
     u32             exefs_offset;
     
+    NCCH_Header     ncch_header;
     ExeFs_Header    exefs_header;
     ExHeader_Header exheader_header;
 };