9a07ed53eb
Now that we have most of core free of shadowing, we can enable the warning as an error to catch anything that may be remaining and also eliminate this class of logic bug entirely.
125 lines
4.3 KiB
C++
125 lines
4.3 KiB
C++
// Copyright 2018 yuzu emulator team
|
|
// Licensed under GPLv2 or any later version
|
|
// Refer to the license.txt file included.
|
|
|
|
#include <cstring>
|
|
#include "common/common_types.h"
|
|
#include "common/logging/log.h"
|
|
#include "common/swap.h"
|
|
#include "core/file_sys/nca_metadata.h"
|
|
#include "core/file_sys/vfs.h"
|
|
|
|
namespace FileSys {
|
|
|
|
CNMT::CNMT(VirtualFile file) {
|
|
if (file->ReadObject(&header) != sizeof(CNMTHeader))
|
|
return;
|
|
|
|
// If type is {Application, Update, AOC} has opt-header.
|
|
if (header.type >= TitleType::Application && header.type <= TitleType::AOC) {
|
|
if (file->ReadObject(&opt_header, sizeof(CNMTHeader)) != sizeof(OptionalHeader)) {
|
|
LOG_WARNING(Loader, "Failed to read optional header.");
|
|
}
|
|
}
|
|
|
|
for (u16 i = 0; i < header.number_content_entries; ++i) {
|
|
auto& next = content_records.emplace_back(ContentRecord{});
|
|
if (file->ReadObject(&next, sizeof(CNMTHeader) + i * sizeof(ContentRecord) +
|
|
header.table_offset) != sizeof(ContentRecord)) {
|
|
content_records.erase(content_records.end() - 1);
|
|
}
|
|
}
|
|
|
|
for (u16 i = 0; i < header.number_meta_entries; ++i) {
|
|
auto& next = meta_records.emplace_back(MetaRecord{});
|
|
if (file->ReadObject(&next, sizeof(CNMTHeader) + i * sizeof(MetaRecord) +
|
|
header.table_offset) != sizeof(MetaRecord)) {
|
|
meta_records.erase(meta_records.end() - 1);
|
|
}
|
|
}
|
|
}
|
|
|
|
CNMT::CNMT(CNMTHeader header_, OptionalHeader opt_header_,
|
|
std::vector<ContentRecord> content_records_, std::vector<MetaRecord> meta_records_)
|
|
: header(std::move(header_)), opt_header(std::move(opt_header_)),
|
|
content_records(std::move(content_records_)), meta_records(std::move(meta_records_)) {}
|
|
|
|
CNMT::~CNMT() = default;
|
|
|
|
u64 CNMT::GetTitleID() const {
|
|
return header.title_id;
|
|
}
|
|
|
|
u32 CNMT::GetTitleVersion() const {
|
|
return header.title_version;
|
|
}
|
|
|
|
TitleType CNMT::GetType() const {
|
|
return header.type;
|
|
}
|
|
|
|
const std::vector<ContentRecord>& CNMT::GetContentRecords() const {
|
|
return content_records;
|
|
}
|
|
|
|
const std::vector<MetaRecord>& CNMT::GetMetaRecords() const {
|
|
return meta_records;
|
|
}
|
|
|
|
bool CNMT::UnionRecords(const CNMT& other) {
|
|
bool change = false;
|
|
for (const auto& rec : other.content_records) {
|
|
const auto iter = std::find_if(content_records.begin(), content_records.end(),
|
|
[&rec](const ContentRecord& r) {
|
|
return r.nca_id == rec.nca_id && r.type == rec.type;
|
|
});
|
|
if (iter == content_records.end()) {
|
|
content_records.emplace_back(rec);
|
|
++header.number_content_entries;
|
|
change = true;
|
|
}
|
|
}
|
|
for (const auto& rec : other.meta_records) {
|
|
const auto iter =
|
|
std::find_if(meta_records.begin(), meta_records.end(), [&rec](const MetaRecord& r) {
|
|
return r.title_id == rec.title_id && r.title_version == rec.title_version &&
|
|
r.type == rec.type;
|
|
});
|
|
if (iter == meta_records.end()) {
|
|
meta_records.emplace_back(rec);
|
|
++header.number_meta_entries;
|
|
change = true;
|
|
}
|
|
}
|
|
return change;
|
|
}
|
|
|
|
std::vector<u8> CNMT::Serialize() const {
|
|
const bool has_opt_header =
|
|
header.type >= TitleType::Application && header.type <= TitleType::AOC;
|
|
const auto dead_zone = header.table_offset + sizeof(CNMTHeader);
|
|
std::vector<u8> out(
|
|
std::max(sizeof(CNMTHeader) + (has_opt_header ? sizeof(OptionalHeader) : 0), dead_zone) +
|
|
content_records.size() * sizeof(ContentRecord) + meta_records.size() * sizeof(MetaRecord));
|
|
memcpy(out.data(), &header, sizeof(CNMTHeader));
|
|
|
|
// Optional Header
|
|
if (has_opt_header) {
|
|
memcpy(out.data() + sizeof(CNMTHeader), &opt_header, sizeof(OptionalHeader));
|
|
}
|
|
|
|
u64_le offset = header.table_offset;
|
|
|
|
for (const auto& rec : content_records) {
|
|
memcpy(out.data() + offset + sizeof(CNMTHeader), &rec, sizeof(ContentRecord));
|
|
offset += sizeof(ContentRecord);
|
|
}
|
|
|
|
for (const auto& rec : meta_records) {
|
|
memcpy(out.data() + offset + sizeof(CNMTHeader), &rec, sizeof(MetaRecord));
|
|
offset += sizeof(MetaRecord);
|
|
}
|
|
|
|
return out;
|
|
}
|
|
} // namespace FileSys
|