nod/lib/DiscBase.cpp

117 lines
3.4 KiB
C++
Raw Normal View History

2015-07-02 18:33:55 +00:00
#include "NOD/DiscBase.hpp"
#include "NOD/IFileIO.hpp"
2015-06-26 20:01:15 +00:00
#ifndef _WIN32
#include <unistd.h>
#endif
2015-06-26 20:01:15 +00:00
namespace NOD
{
2015-06-30 00:07:46 +00:00
void DiscBase::IPartition::parseFST(IPartReadStream& s)
2015-06-29 05:35:25 +00:00
{
2015-06-30 00:07:46 +00:00
std::unique_ptr<uint8_t[]> fst(new uint8_t[m_fstSz]);
s.seek(m_fstOff);
s.read(fst.get(), m_fstSz);
const FSTNode* nodes = (FSTNode*)fst.get();
/* Root node indicates the count of all contained nodes */
uint32_t nodeCount = nodes[0].getLength();
const char* names = (char*)fst.get() + 12 * nodeCount;
m_nodes.clear();
m_nodes.reserve(nodeCount);
/* Construct nodes */
for (uint32_t n=0 ; n<nodeCount ; ++n)
{
const FSTNode& node = nodes[n];
2015-06-30 06:46:19 +00:00
m_nodes.emplace_back(*this, node, n ? names + node.getNameOffset() : "");
2015-06-30 00:07:46 +00:00
}
/* Setup dir-child iterators */
for (std::vector<Node>::iterator it=m_nodes.begin();
it != m_nodes.end();
++it)
{
Node& node = *it;
if (node.m_kind == Node::NODE_DIRECTORY)
{
node.m_childrenBegin = it + 1;
node.m_childrenEnd = m_nodes.begin() + node.m_discLength;
}
}
2015-06-29 05:35:25 +00:00
}
2015-07-10 03:11:30 +00:00
void DiscBase::IPartition::parseDOL(IPartReadStream& s)
{
/* Read Dol header */
s.read(&m_dolHead, sizeof(DOLHeader));
/* Calculate Dol size */
uint32_t dolSize = SBig(m_dolHead.textOff[0]) - sizeof(DOLHeader);
for (uint32_t i = 0 ; i < 7 ; i++)
dolSize += SBig(m_dolHead.textSizes[i]);
for (uint32_t i = 0 ; i < 11 ; i++)
dolSize += SBig(m_dolHead.dataSizes[i]);
m_dolSz = dolSize;
}
2015-07-02 18:33:55 +00:00
void DiscBase::IPartition::Node::extractToDirectory(const SystemString& basePath, bool force)
2015-06-30 06:46:19 +00:00
{
2015-07-02 18:33:55 +00:00
SystemStringView nameView(getName());
SystemString path = basePath + _S("/") + nameView.sys_str();
2015-06-30 06:46:19 +00:00
if (m_kind == NODE_DIRECTORY)
{
2015-07-02 20:37:07 +00:00
if (Mkdir(path.c_str(), 0755) && errno != EEXIST)
{
SystemUTF8View pathView(path);
throw std::runtime_error("unable to mkdir '" + pathView.utf8_str() + "'");
}
for (Node& subnode : *this)
subnode.extractToDirectory(path, force);
2015-06-30 06:46:19 +00:00
}
else if (m_kind == NODE_FILE)
{
2015-07-02 20:37:07 +00:00
Sstat theStat;
if (force || Stat(path.c_str(), &theStat))
2015-06-30 06:46:19 +00:00
{
m_hddFile = NewFileIO(path);
std::unique_ptr<IPartReadStream> rs = beginReadStream();
std::unique_ptr<IFileIO::IWriteStream> ws = m_hddFile->beginWriteStream();
ws->copyFromDisc(*rs.get(), m_discLength);
}
}
}
2015-07-02 18:33:55 +00:00
void DiscBase::IPartition::extractToDirectory(const SystemString& path, bool force)
{
2015-07-02 20:37:07 +00:00
Sstat theStat;
if (Mkdir(path.c_str(), 0755) && errno != EEXIST)
{
SystemUTF8View pathView(path);
throw std::runtime_error("unable to mkdir '" + pathView.utf8_str() + "'");
}
/* Extract Apploader */
2015-07-02 20:37:07 +00:00
SystemString apploaderPath = path + _S("/apploader.bin");
if (force || Stat(apploaderPath.c_str(), &theStat))
{
2015-07-10 03:11:30 +00:00
std::unique_ptr<uint8_t[]> buf = getApploaderBuf();
NewFileIO(apploaderPath)->beginWriteStream()->write(buf.get(), m_apploaderSz);
}
2015-07-03 03:49:31 +00:00
/* Extract Dol */
SystemString dolPath = path + _S("/main.dol");
if (force || Stat(dolPath.c_str(), &theStat))
{
2015-07-10 03:11:30 +00:00
std::unique_ptr<uint8_t[]> buf = getDOLBuf();
NewFileIO(dolPath)->beginWriteStream()->write(buf.get(), m_dolSz);
2015-07-03 03:49:31 +00:00
}
/* Extract Filesystem */
m_nodes[0].extractToDirectory(path, force);
}
2015-06-26 20:01:15 +00:00
}