111 lines
2.6 KiB
C++
111 lines
2.6 KiB
C++
#include "PKG.h"
|
|
#include "FsFile.h"
|
|
#include <direct.h>
|
|
|
|
PKG::PKG()
|
|
{
|
|
}
|
|
|
|
|
|
PKG::~PKG()
|
|
{
|
|
}
|
|
|
|
bool PKG::open(const std::string& filepath) {
|
|
FsFile file;
|
|
if (!file.Open(filepath, fsRead))
|
|
{
|
|
return false;
|
|
}
|
|
pkgSize = file.getFileSize();
|
|
file.ReadBE(pkgheader);
|
|
//we have already checked magic should be ok
|
|
|
|
//find title id it is part of pkg_content_id starting at offset 0x40
|
|
file.Seek(0x47, fsSeekSet);//skip first 7 characters of content_id
|
|
file.Read(&pkgTitleID, sizeof(pkgTitleID));
|
|
|
|
file.Close();
|
|
|
|
return true;
|
|
}
|
|
bool PKG::extract(const std::string& filepath, const std::string& extractPath, std::string& failreason)
|
|
{
|
|
this->extractPath = extractPath;
|
|
FsFile file;
|
|
if (!file.Open(filepath, fsRead))
|
|
{
|
|
return false;
|
|
}
|
|
pkgSize = file.getFileSize();
|
|
|
|
file.ReadBE(pkgheader);
|
|
|
|
if (pkgheader.pkg_size > pkgSize)
|
|
{
|
|
failreason = "PKG file size is different";
|
|
return false;
|
|
}
|
|
if ((pkgheader.pkg_content_size + pkgheader.pkg_content_offset) > pkgheader.pkg_size)
|
|
{
|
|
failreason = "Content size is bigger than pkg size";
|
|
return false;
|
|
}
|
|
file.Seek(0, fsSeekSet);
|
|
pkg = (U08*)mmap(pkgSize, file.fileDescr());
|
|
if (pkg == nullptr)
|
|
{
|
|
failreason = "Can't allocate size for image";
|
|
return false;
|
|
}
|
|
|
|
file.Read(pkg, pkgSize);
|
|
|
|
U32 offset = pkgheader.pkg_table_entry_offset;
|
|
U32 n_files = pkgheader.pkg_table_entry_count;
|
|
|
|
|
|
for (U32 i = 0; i < n_files; i++) {
|
|
PKGEntry entry = (PKGEntry&)pkg[offset + i * 0x20];
|
|
ReadBE(entry);
|
|
//try to figure out the name
|
|
std::string name = getEntryNameByType(entry.id);
|
|
if (!name.empty())
|
|
{
|
|
std::size_t pos = name.find("/");//check if we have a directory (assuming we only have 1 level of subdirectories)
|
|
if (pos != std::string::npos)
|
|
{
|
|
//directory found
|
|
std::string dir = name.substr(0, pos+1);//include '/' as well
|
|
std::string path = extractPath + dir;
|
|
_mkdir(path.c_str());//make dir
|
|
std::string file = name.substr(pos + 1);//read filename
|
|
FsFile out;
|
|
out.Open(path + file, fsWrite);
|
|
out.Write(pkg + entry.offset, entry.size);
|
|
out.Close();
|
|
|
|
}
|
|
//found an name use it
|
|
FsFile out;
|
|
out.Open(extractPath + name, fsWrite);
|
|
out.Write(pkg + entry.offset, entry.size);
|
|
out.Close();
|
|
}
|
|
else
|
|
{
|
|
//just print with id
|
|
FsFile out;
|
|
out.Open(extractPath + std::to_string(entry.id), fsWrite);
|
|
out.Write(pkg + entry.offset, entry.size);
|
|
out.Close();
|
|
}
|
|
}
|
|
//extract pfs_image.dat
|
|
FsFile out;
|
|
out.Open(extractPath + "pfs_image.dat", fsWrite);
|
|
out.Write(pkg + pkgheader.pfs_image_offset, pkgheader.pfs_image_size);
|
|
out.Close();
|
|
munmap(pkg);
|
|
return true;
|
|
} |