119 lines
3.8 KiB
C++
119 lines
3.8 KiB
C++
// Copyright 2018 yuzu emulator team
|
|
// Licensed under GPLv2 or any later version
|
|
// Refer to the license.txt file included.
|
|
|
|
#include <algorithm>
|
|
#include <cstddef>
|
|
#include <cstring>
|
|
#include <iterator>
|
|
#include <utility>
|
|
|
|
#include "common/file_util.h"
|
|
#include "common/logging/log.h"
|
|
#include "core/file_sys/partition_filesystem.h"
|
|
#include "core/file_sys/vfs_offset.h"
|
|
#include "core/loader/loader.h"
|
|
|
|
namespace FileSys {
|
|
|
|
bool PartitionFilesystem::Header::HasValidMagicValue() const {
|
|
return magic == Common::MakeMagic('H', 'F', 'S', '0') ||
|
|
magic == Common::MakeMagic('P', 'F', 'S', '0');
|
|
}
|
|
|
|
PartitionFilesystem::PartitionFilesystem(std::shared_ptr<VfsFile> file) {
|
|
// At least be as large as the header
|
|
if (file->GetSize() < sizeof(Header)) {
|
|
status = Loader::ResultStatus::ErrorBadPFSHeader;
|
|
return;
|
|
}
|
|
|
|
// For cartridges, HFSs can get very large, so we need to calculate the size up to
|
|
// the actual content itself instead of just blindly reading in the entire file.
|
|
if (sizeof(Header) != file->ReadObject(&pfs_header)) {
|
|
status = Loader::ResultStatus::ErrorBadPFSHeader;
|
|
return;
|
|
}
|
|
|
|
if (!pfs_header.HasValidMagicValue()) {
|
|
status = Loader::ResultStatus::ErrorBadPFSHeader;
|
|
return;
|
|
}
|
|
|
|
is_hfs = pfs_header.magic == Common::MakeMagic('H', 'F', 'S', '0');
|
|
|
|
size_t entry_size = is_hfs ? sizeof(HFSEntry) : sizeof(PFSEntry);
|
|
size_t metadata_size =
|
|
sizeof(Header) + (pfs_header.num_entries * entry_size) + pfs_header.strtab_size;
|
|
|
|
// Actually read in now...
|
|
std::vector<u8> file_data = file->ReadBytes(metadata_size);
|
|
const size_t total_size = file_data.size();
|
|
|
|
if (total_size != metadata_size) {
|
|
status = Loader::ResultStatus::ErrorIncorrectPFSFileSize;
|
|
return;
|
|
}
|
|
|
|
size_t entries_offset = sizeof(Header);
|
|
size_t strtab_offset = entries_offset + (pfs_header.num_entries * entry_size);
|
|
content_offset = strtab_offset + pfs_header.strtab_size;
|
|
for (u16 i = 0; i < pfs_header.num_entries; i++) {
|
|
FSEntry entry;
|
|
|
|
memcpy(&entry, &file_data[entries_offset + (i * entry_size)], sizeof(FSEntry));
|
|
std::string name(
|
|
reinterpret_cast<const char*>(&file_data[strtab_offset + entry.strtab_offset]));
|
|
|
|
pfs_files.emplace_back(std::make_shared<OffsetVfsFile>(
|
|
file, entry.size, content_offset + entry.offset, std::move(name)));
|
|
}
|
|
|
|
status = Loader::ResultStatus::Success;
|
|
}
|
|
|
|
Loader::ResultStatus PartitionFilesystem::GetStatus() const {
|
|
return status;
|
|
}
|
|
|
|
std::vector<std::shared_ptr<VfsFile>> PartitionFilesystem::GetFiles() const {
|
|
return pfs_files;
|
|
}
|
|
|
|
std::vector<std::shared_ptr<VfsDirectory>> PartitionFilesystem::GetSubdirectories() const {
|
|
return pfs_dirs;
|
|
}
|
|
|
|
std::string PartitionFilesystem::GetName() const {
|
|
return is_hfs ? "HFS0" : "PFS0";
|
|
}
|
|
|
|
std::shared_ptr<VfsDirectory> PartitionFilesystem::GetParentDirectory() const {
|
|
// TODO(DarkLordZach): Add support for nested containers.
|
|
return nullptr;
|
|
}
|
|
|
|
void PartitionFilesystem::PrintDebugInfo() const {
|
|
LOG_DEBUG(Service_FS, "Magic: {:.4}", pfs_header.magic);
|
|
LOG_DEBUG(Service_FS, "Files: {}", pfs_header.num_entries);
|
|
for (u32 i = 0; i < pfs_header.num_entries; i++) {
|
|
LOG_DEBUG(Service_FS, " > File {}: {} (0x{:X} bytes)", i,
|
|
pfs_files[i]->GetName(), pfs_files[i]->GetSize());
|
|
}
|
|
}
|
|
|
|
bool PartitionFilesystem::ReplaceFileWithSubdirectory(VirtualFile file, VirtualDir dir) {
|
|
const auto iter = std::find(pfs_files.begin(), pfs_files.end(), file);
|
|
if (iter == pfs_files.end())
|
|
return false;
|
|
|
|
const std::ptrdiff_t offset = std::distance(pfs_files.begin(), iter);
|
|
pfs_files[offset] = std::move(pfs_files.back());
|
|
pfs_files.pop_back();
|
|
|
|
pfs_dirs.emplace_back(std::move(dir));
|
|
|
|
return true;
|
|
}
|
|
} // namespace FileSys
|