mirror of
https://github.com/jakcron/nstool
synced 2024-12-27 15:11:12 +00:00
138 lines
3.6 KiB
C++
138 lines
3.6 KiB
C++
#include "AssetProcess.h"
|
|
|
|
#include "util.h"
|
|
|
|
nstool::AssetProcess::AssetProcess() :
|
|
mModuleName("nstool::AssetProcess::"),
|
|
mFile(),
|
|
mCliOutputMode(true, false, false, false),
|
|
mVerify(false)
|
|
{
|
|
}
|
|
|
|
void nstool::AssetProcess::process()
|
|
{
|
|
importHeader();
|
|
if (mCliOutputMode.show_basic_info)
|
|
displayHeader();
|
|
processSections();
|
|
}
|
|
|
|
void nstool::AssetProcess::setInputFile(const std::shared_ptr<tc::io::IStream>& file)
|
|
{
|
|
mFile = file;
|
|
}
|
|
|
|
void nstool::AssetProcess::setCliOutputMode(CliOutputMode type)
|
|
{
|
|
mCliOutputMode = type;
|
|
}
|
|
|
|
void nstool::AssetProcess::setVerifyMode(bool verify)
|
|
{
|
|
mVerify = verify;
|
|
}
|
|
|
|
void nstool::AssetProcess::setListFs(bool list)
|
|
{
|
|
mRomfs.setListFs(list);
|
|
}
|
|
|
|
void nstool::AssetProcess::setIconExtractPath(const tc::io::Path& path)
|
|
{
|
|
mIconExtractPath = path;
|
|
}
|
|
|
|
void nstool::AssetProcess::setNacpExtractPath(const tc::io::Path& path)
|
|
{
|
|
mNacpExtractPath = path;
|
|
}
|
|
|
|
void nstool::AssetProcess::setRomfsExtractPath(const tc::io::Path& path)
|
|
{
|
|
mRomfs.setExtractPath(path);
|
|
}
|
|
|
|
|
|
void nstool::AssetProcess::importHeader()
|
|
{
|
|
if (mFile == nullptr)
|
|
{
|
|
throw tc::Exception(mModuleName, "No file reader set.");
|
|
}
|
|
|
|
if (mFile->length() < tc::io::IOUtil::castSizeToInt64(sizeof(nn::hac::sAssetHeader)))
|
|
{
|
|
throw tc::Exception(mModuleName, "Corrupt ASET: file too small");
|
|
}
|
|
|
|
tc::ByteData scratch = tc::ByteData(sizeof(nn::hac::sAssetHeader));
|
|
mFile->seek(0, tc::io::SeekOrigin::Begin);
|
|
mFile->read(scratch.data(), scratch.size());
|
|
|
|
mHdr.fromBytes(scratch.data(), scratch.size());
|
|
}
|
|
|
|
void nstool::AssetProcess::processSections()
|
|
{
|
|
int64_t file_size = mFile->length();
|
|
|
|
if (mHdr.getIconInfo().size > 0 && mIconExtractPath.isSet())
|
|
{
|
|
if ((mHdr.getIconInfo().size + mHdr.getIconInfo().offset) > file_size)
|
|
throw tc::Exception(mModuleName, "ASET geometry for icon beyond file size");
|
|
|
|
std::string icon_extract_path_str;
|
|
tc::io::PathUtil::pathToUnixUTF8(mIconExtractPath.get(), icon_extract_path_str);
|
|
|
|
fmt::print("Saving {:s}...", icon_extract_path_str);
|
|
writeSubStreamToFile(mFile, mHdr.getIconInfo().offset, mHdr.getIconInfo().size, mIconExtractPath.get());
|
|
}
|
|
|
|
if (mHdr.getNacpInfo().size > 0)
|
|
{
|
|
if ((mHdr.getNacpInfo().size + mHdr.getNacpInfo().offset) > file_size)
|
|
throw tc::Exception(mModuleName, "ASET geometry for nacp beyond file size");
|
|
|
|
if (mNacpExtractPath.isSet())
|
|
{
|
|
writeSubStreamToFile(mFile, mHdr.getNacpInfo().offset, mHdr.getNacpInfo().size, mNacpExtractPath.get());
|
|
}
|
|
|
|
mNacp.setInputFile(std::make_shared<tc::io::SubStream>(mFile, mHdr.getNacpInfo().offset, mHdr.getNacpInfo().size));
|
|
mNacp.setCliOutputMode(mCliOutputMode);
|
|
mNacp.setVerifyMode(mVerify);
|
|
|
|
mNacp.process();
|
|
}
|
|
|
|
if (mHdr.getRomfsInfo().size > 0)
|
|
{
|
|
if ((mHdr.getRomfsInfo().size + mHdr.getRomfsInfo().offset) > file_size)
|
|
throw tc::Exception(mModuleName, "ASET geometry for romfs beyond file size");
|
|
|
|
mRomfs.setInputFile(std::make_shared<tc::io::SubStream>(mFile, mHdr.getRomfsInfo().offset, mHdr.getRomfsInfo().size));
|
|
mRomfs.setCliOutputMode(mCliOutputMode);
|
|
mRomfs.setVerifyMode(mVerify);
|
|
|
|
mRomfs.process();
|
|
}
|
|
}
|
|
|
|
void nstool::AssetProcess::displayHeader()
|
|
{
|
|
if (mCliOutputMode.show_layout)
|
|
{
|
|
fmt::print("[ASET Header]\n");
|
|
fmt::print(" Icon:\n");
|
|
fmt::print(" Offset: 0x{:x}\n", mHdr.getIconInfo().offset);
|
|
fmt::print(" Size: 0x{:x}\n", mHdr.getIconInfo().size);
|
|
fmt::print(" NACP:\n");
|
|
fmt::print(" Offset: 0x{:x}\n", mHdr.getNacpInfo().offset);
|
|
fmt::print(" Size: 0x{:x}\n", mHdr.getNacpInfo().size);
|
|
fmt::print(" RomFS:\n");
|
|
fmt::print(" Offset: 0x{:x}\n", mHdr.getRomfsInfo().offset);
|
|
fmt::print(" Size: 0x{:x}\n", mHdr.getRomfsInfo().size);
|
|
}
|
|
}
|
|
|