Martin Michelsen
105 lines
3.2 KiB
C++
105 lines
3.2 KiB
C++
#include "BMLArchive.hh"
|
|
|
|
#include <phosg/Encoding.hh>
|
|
#include <phosg/Filesystem.hh>
|
|
#include <stdexcept>
|
|
|
|
#include "Text.hh"
|
|
#include "Types.hh"
|
|
|
|
template <bool BE>
|
|
struct BMLHeaderT {
|
|
parray<uint8_t, 0x04> unknown_a1;
|
|
U32T<BE> num_entries;
|
|
parray<uint8_t, 0x38> unknown_a2;
|
|
} __packed_ws_be__(BMLHeaderT, 0x40);
|
|
using BMLHeader = BMLHeaderT<false>;
|
|
using BMLHeaderBE = BMLHeaderT<true>;
|
|
|
|
template <bool BE>
|
|
struct BMLHeaderEntryT {
|
|
pstring<TextEncoding::ASCII, 0x20> filename;
|
|
U32T<BE> compressed_size;
|
|
parray<uint8_t, 0x04> unknown_a1;
|
|
U32T<BE> decompressed_size;
|
|
U32T<BE> compressed_gvm_size;
|
|
U32T<BE> decompressed_gvm_size;
|
|
parray<uint8_t, 0x0C> unknown_a2;
|
|
} __packed_ws_be__(BMLHeaderEntryT, 0x40);
|
|
using BMLHeaderEntry = BMLHeaderEntryT<false>;
|
|
using BMLHeaderEntryBE = BMLHeaderEntryT<true>;
|
|
|
|
template <bool BE>
|
|
void BMLArchive::load_t() {
|
|
phosg::StringReader r(*this->data);
|
|
|
|
const auto& header = r.get<BMLHeaderT<BE>>();
|
|
|
|
size_t offset = 0x800;
|
|
while (this->entries.size() < header.num_entries) {
|
|
const auto& entry = r.get<BMLHeaderEntryT<BE>>();
|
|
|
|
if (offset + entry.compressed_size > this->data->size()) {
|
|
throw std::runtime_error("BML data entry extends beyond end of data");
|
|
}
|
|
size_t data_offset = offset;
|
|
offset = (offset + entry.compressed_size + 0x1F) & (~0x1F);
|
|
|
|
if (offset + entry.compressed_gvm_size > this->data->size()) {
|
|
throw std::runtime_error("BML GVM entry extends beyond end of data");
|
|
}
|
|
size_t gvm_offset = offset;
|
|
offset = (offset + entry.compressed_gvm_size + 0x1F) & (~0x1F);
|
|
|
|
this->entries.emplace(entry.filename.decode(), Entry{data_offset, entry.compressed_size, gvm_offset, entry.compressed_gvm_size});
|
|
}
|
|
}
|
|
|
|
BMLArchive::BMLArchive(std::shared_ptr<const std::string> data, bool big_endian) : data(data) {
|
|
if (big_endian) {
|
|
this->load_t<true>();
|
|
} else {
|
|
this->load_t<false>();
|
|
}
|
|
}
|
|
|
|
const std::unordered_map<std::string, BMLArchive::Entry> BMLArchive::all_entries() const {
|
|
return this->entries;
|
|
}
|
|
|
|
std::pair<const void*, size_t> BMLArchive::get(const std::string& name) const {
|
|
try {
|
|
const auto& entry = this->entries.at(name);
|
|
return std::make_pair(this->data->data() + entry.offset, entry.size);
|
|
} catch (const std::out_of_range&) {
|
|
throw std::out_of_range("BML does not contain file: " + name);
|
|
}
|
|
}
|
|
|
|
std::pair<const void*, size_t> BMLArchive::get_gvm(const std::string& name) const {
|
|
try {
|
|
const auto& entry = this->entries.at(name);
|
|
return std::make_pair(this->data->data() + entry.gvm_offset, entry.gvm_size);
|
|
} catch (const std::out_of_range&) {
|
|
throw std::out_of_range("BML does not contain file: " + name);
|
|
}
|
|
}
|
|
|
|
std::string BMLArchive::get_copy(const std::string& name) const {
|
|
try {
|
|
const auto& entry = this->entries.at(name);
|
|
return this->data->substr(entry.offset, entry.size);
|
|
} catch (const std::out_of_range&) {
|
|
throw std::out_of_range("BML does not contain file: " + name);
|
|
}
|
|
}
|
|
|
|
phosg::StringReader BMLArchive::get_reader(const std::string& name) const {
|
|
try {
|
|
const auto& entry = this->entries.at(name);
|
|
return phosg::StringReader(this->data->data() + entry.offset, entry.size);
|
|
} catch (const std::out_of_range&) {
|
|
throw std::out_of_range("BML does not contain file: " + name);
|
|
}
|
|
}
|