nold/nstool
1
0
Fork 0
mirror of https://github.com/jakcron/nstool synced 2024-11-22 13:39:28 +00:00
Code Issues Projects Releases Packages Wiki Activity

Misc

Browse source
This commit is contained in:
jakcron 2021-10-13 13:39:45 +08:00
parent 054776b586
commit afd1c97272
2 changed files with 16 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
src/IniProcess.cpp
View file

@ -1,20 +1,12 @@
#include <iostream>
#include <iomanip>
#include <fnd/io.h>
#include <fnd/SimpleFile.h>
#include <fnd/SimpleTextOutput.h>
#include <fnd/OffsetAdjustedIFile.h>
#include <fnd/Vec.h>
#include "IniProcess.h" #include "IniProcess.h"
#include "KipProcess.h"
#include "KipProcess.h"
nstool::IniProcess::IniProcess() : nstool::IniProcess::IniProcess() :
mModuleName("nstool::IniProcess"), mModuleName("nstool::IniProcess"),
mFile(), mFile(),
mCliOutputMode(true, false, false, false), mCliOutputMode(true, false, false, false),
mVerify(false), mVerify(false),
mDoExtractKip(false),
mKipExtractPath() mKipExtractPath()
{ {
} }
@ -84,20 +76,20 @@ void nstool::IniProcess::importHeader()
void nstool::IniProcess::importKipList() void nstool::IniProcess::importKipList()
{ {
// kip pos info // kip pos info
size_t kip_pos = sizeof(nn::hac::sIniHeader); int64_t kip_pos = sizeof(nn::hac::sIniHeader);
size_t kip_size = 0; int64_t kip_size = 0;
// tmp data to determine size // tmp data to determine size
tc::ByteData hdr_raw; nn::hac::sKipHeader hdr_raw;
nn::hac::KernelInitialProcessHeader hdr; nn::hac::KernelInitialProcessHeader hdr;
hdr_raw.alloc(sizeof(nn::hac::sKipHeader));
for (size_t i = 0; i < mHdr.getKipNum(); i++) for (size_t i = 0; i < mHdr.getKipNum(); i++)
{ {
(*mFile)->read(hdr_raw.data(), kip_pos, hdr_raw.size()); mFile->seek(kip_pos, tc::io::SeekOrigin::Begin);
hdr.fromBytes(hdr_raw.data(), hdr_raw.size()); mFile->read((byte_t*)&hdr_raw, sizeof(hdr_raw));
hdr.fromBytes((byte_t*)&hdr_raw, sizeof(hdr_raw));
kip_size = getKipSizeFromHeader(hdr); kip_size = getKipSizeFromHeader(hdr);
mKipList.push_back(new fnd::OffsetAdjustedIFile(mFile, kip_pos, kip_size)); mKipList.push_back(std::make_shared<tc::io::SubStream>(tc::io::SubStream(mFile, kip_pos, kip_size)));
kip_pos += kip_size; kip_pos += kip_size;
} }
} }
@ -171,7 +163,7 @@ void nstool::IniProcess::extractKipList()
} }
} }
size_t nstool::IniProcess::getKipSizeFromHeader(const nn::hac::KernelInitialProcessHeader& hdr) const int64_t nstool::IniProcess::getKipSizeFromHeader(const nn::hac::KernelInitialProcessHeader& hdr) const
{ {
return sizeof(nn::hac::sKipHeader) + hdr.getTextSegmentInfo().file_layout.size + hdr.getRoSegmentInfo().file_layout.size + hdr.getDataSegmentInfo().file_layout.size; return sizeof(nn::hac::sKipHeader) + hdr.getTextSegmentInfo().file_layout.size + hdr.getRoSegmentInfo().file_layout.size + hdr.getDataSegmentInfo().file_layout.size;
} }

9
src/IniProcess.h
View file

@ -31,7 +31,12 @@ private:
tc::Optional<tc::io::Path> mKipExtractPath; tc::Optional<tc::io::Path> mKipExtractPath;
nn::hac::IniHeader mHdr; nn::hac::IniHeader mHdr;
std::vector<std::shared_ptr<tc::io::IStream>> mKipList; struct InnerKipInfo
{
nn::hac::KernelInitialProcessHeader hdr;
std::shared_ptr<tc::io::IStream> stream;
};
std::vector<InnerKipInfo> mKipList;
void importHeader(); void importHeader();
void importKipList(); void importKipList();
@ -39,7 +44,7 @@ private:
void displayKipList(); void displayKipList();
void extractKipList(); void extractKipList();
size_t getKipSizeFromHeader(const nn::hac::KernelInitialProcessHeader& hdr) const; int64_t getKipSizeFromHeader(const nn::hac::KernelInitialProcessHeader& hdr) const;
}; };
} }