1346 lines
53 KiB
C++
1346 lines
53 KiB
C++
#include "Quest.hh"
|
|
|
|
#include <algorithm>
|
|
#include <filesystem>
|
|
#include <mutex>
|
|
#include <phosg/Encoding.hh>
|
|
#include <phosg/Filesystem.hh>
|
|
#include <phosg/Hash.hh>
|
|
#include <phosg/Random.hh>
|
|
#include <phosg/Strings.hh>
|
|
#include <phosg/Tools.hh>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
|
|
#include "CommandFormats.hh"
|
|
#include "Compression.hh"
|
|
#include "Loggers.hh"
|
|
#include "PSOEncryption.hh"
|
|
#include "QuestScript.hh"
|
|
#include "SaveFileFormats.hh"
|
|
#include "Text.hh"
|
|
|
|
QuestCategoryIndex::Category::Category(uint32_t category_id, const phosg::JSON& json) : category_id(category_id) {
|
|
this->enabled_flags = json.get_int(0);
|
|
this->directory_name = json.get_string(1);
|
|
this->name = json.get_string(2);
|
|
this->description = json.get_string(3);
|
|
}
|
|
|
|
QuestCategoryIndex::QuestCategoryIndex(const phosg::JSON& json) {
|
|
uint32_t next_category_id = 1;
|
|
for (const auto& it : json.as_list()) {
|
|
this->categories.emplace_back(std::make_shared<Category>(next_category_id++, *it));
|
|
}
|
|
}
|
|
|
|
std::shared_ptr<const QuestCategoryIndex::Category> QuestCategoryIndex::at(uint32_t category_id) const {
|
|
return this->categories.at(category_id - 1);
|
|
}
|
|
|
|
// GCI decoding logic
|
|
|
|
template <bool BE>
|
|
struct PSOMemCardDLQFileEncryptedHeaderT {
|
|
U32T<BE> round2_seed;
|
|
// To compute checksum, set checksum to zero, then compute the CRC32 of the entire data section, including this
|
|
// header struct (but not the unencrypted header struct).
|
|
U32T<BE> checksum;
|
|
le_uint32_t decompressed_size;
|
|
le_uint32_t round3_seed;
|
|
// Data follows here.
|
|
} __packed_ws_be__(PSOMemCardDLQFileEncryptedHeaderT, 0x10);
|
|
using PSOVMSDLQFileEncryptedHeader = PSOMemCardDLQFileEncryptedHeaderT<false>;
|
|
using PSOGCIDLQFileEncryptedHeader = PSOMemCardDLQFileEncryptedHeaderT<true>;
|
|
|
|
template <bool BE>
|
|
std::string decrypt_download_quest_data_section(
|
|
const void* data_section, size_t size, uint32_t seed, bool skip_checksum = false, bool is_ep3_trial = false) {
|
|
std::string decrypted = decrypt_data_section<BE>(data_section, size, seed);
|
|
|
|
size_t orig_size = decrypted.size();
|
|
decrypted.resize((decrypted.size() + 3) & (~3));
|
|
|
|
// Other PSO save files have the round 2 seed at the end, not at the beginning. Presumably this is because the
|
|
// system, character, and Guild Card files are a constant size, but download quest files can vary in size.
|
|
using HeaderT = PSOMemCardDLQFileEncryptedHeaderT<BE>;
|
|
auto* header = reinterpret_cast<HeaderT*>(decrypted.data());
|
|
PSOV2Encryption round2_crypt(header->round2_seed);
|
|
round2_crypt.encrypt_t<BE>(decrypted.data() + 4, (decrypted.size() - 4));
|
|
|
|
if (is_ep3_trial) {
|
|
phosg::StringReader r(decrypted);
|
|
r.skip(16);
|
|
if (r.readx(15) != "SONICTEAM,SEGA.") {
|
|
throw std::runtime_error("Episode 3 GCI file is not a quest");
|
|
}
|
|
r.skip(9);
|
|
|
|
// Some Ep3 trial download quests don't have a stop opcode in the PRS stream; it seems the client just
|
|
// automatically stops when the correct amount of data has been produced. To handle this, we allow the PRS stream
|
|
// to be unterminated here.
|
|
size_t decompressed_size = prs_decompress_size(
|
|
r.getv(r.remaining(), false), r.remaining(), sizeof(Episode3::MapDefinitionTrial), true);
|
|
if (decompressed_size < sizeof(Episode3::MapDefinitionTrial)) {
|
|
throw std::runtime_error(std::format(
|
|
"decompressed size ({}) does not match expected size ({})",
|
|
decompressed_size, sizeof(Episode3::MapDefinitionTrial)));
|
|
}
|
|
return decrypted.substr(0x28);
|
|
|
|
} else {
|
|
if (header->decompressed_size & 0xFFF00000) {
|
|
throw std::runtime_error(std::format("decompressed_size too large ({:08X})", header->decompressed_size));
|
|
}
|
|
|
|
if (!skip_checksum) {
|
|
uint32_t expected_crc = header->checksum;
|
|
header->checksum = 0;
|
|
uint32_t actual_crc = phosg::crc32(decrypted.data(), orig_size);
|
|
header->checksum = expected_crc;
|
|
if (expected_crc != actual_crc && expected_crc != phosg::bswap32(actual_crc)) {
|
|
throw std::runtime_error(std::format(
|
|
"incorrect decrypted data section checksum: expected {:08X}; received {:08X}", expected_crc, actual_crc));
|
|
}
|
|
}
|
|
|
|
// Unlike the above rounds, round 3 is always little-endian (it corresponds to the round of encryption done on the
|
|
// server before sending the file to the client in the first place)
|
|
PSOV2Encryption(header->round3_seed).decrypt(decrypted.data() + sizeof(HeaderT), decrypted.size() - sizeof(HeaderT));
|
|
decrypted.resize(orig_size);
|
|
|
|
// Some download quest GCI files have decompressed_size fields that are 8 bytes smaller than the actual
|
|
// decompressed size of the data. They seem to work fine, so we accept both cases as correct.
|
|
size_t decompressed_size = prs_decompress_size(
|
|
decrypted.data() + sizeof(HeaderT), decrypted.size() - sizeof(HeaderT));
|
|
size_t expected_decompressed_size = header->decompressed_size;
|
|
if ((decompressed_size != expected_decompressed_size) && (decompressed_size != expected_decompressed_size - 8)) {
|
|
throw std::runtime_error(std::format(
|
|
"decompressed size ({}) does not match expected size ({})", decompressed_size, expected_decompressed_size));
|
|
}
|
|
|
|
return decrypted.substr(sizeof(HeaderT));
|
|
}
|
|
}
|
|
|
|
std::string decrypt_vms_v1_data_section(const void* data_section, size_t size) {
|
|
phosg::StringReader r(data_section, size);
|
|
uint32_t expected_decompressed_size = r.get_u32l();
|
|
uint32_t seed = r.get_u32l();
|
|
|
|
std::string data = r.read(r.remaining());
|
|
|
|
size_t orig_size = data.size();
|
|
data.resize((orig_size + 3) & (~3));
|
|
PSOV2Encryption(seed).decrypt(data.data(), data.size());
|
|
data.resize(orig_size);
|
|
|
|
size_t actual_decompressed_size = prs_decompress_size(data);
|
|
if (actual_decompressed_size != expected_decompressed_size) {
|
|
throw std::runtime_error(std::format(
|
|
"decompressed size ({}) does not match size in header ({})",
|
|
actual_decompressed_size, expected_decompressed_size));
|
|
}
|
|
|
|
return data;
|
|
}
|
|
|
|
template <bool BE>
|
|
std::string find_seed_and_decrypt_download_quest_data_section(
|
|
const void* data_section, size_t size, bool skip_checksum, bool is_ep3_trial, size_t num_threads) {
|
|
std::mutex result_lock;
|
|
std::string result;
|
|
uint64_t result_seed = phosg::parallel_blocks<uint64_t>([&](uint64_t seed, size_t) {
|
|
try {
|
|
std::string ret = decrypt_download_quest_data_section<BE>(data_section, size, seed, skip_checksum, is_ep3_trial);
|
|
std::lock_guard g(result_lock);
|
|
result = std::move(ret);
|
|
return true;
|
|
} catch (const std::runtime_error& e) {
|
|
return false;
|
|
}
|
|
},
|
|
0, 0x100000000, 0x1000, num_threads);
|
|
|
|
if (!result.empty() && (result_seed < 0x100000000)) {
|
|
static_game_data_log.info_f("Found seed {:08X}", result_seed);
|
|
return result;
|
|
} else {
|
|
throw std::runtime_error("no seed found");
|
|
}
|
|
}
|
|
|
|
struct PSODownloadQuestHeader {
|
|
le_uint32_t size;
|
|
le_uint32_t encryption_seed;
|
|
} __packed_ws__(PSODownloadQuestHeader, 8);
|
|
|
|
void VersionedQuest::assert_valid() const {
|
|
if (this->meta.category_id == 0xFFFFFFFF) {
|
|
throw std::runtime_error("category ID is not set");
|
|
}
|
|
if (this->meta.quest_number == 0xFFFFFFFF) {
|
|
throw std::runtime_error("quest number is not set");
|
|
}
|
|
if (this->meta.version == Version::UNKNOWN) {
|
|
throw std::runtime_error("version is not set");
|
|
}
|
|
if (this->meta.language == Language::UNKNOWN) {
|
|
throw std::runtime_error("language is not set");
|
|
}
|
|
|
|
uint8_t num_floors = is_v1(this->meta.version) ? 0x10 : 0x12;
|
|
uint8_t num_areas;
|
|
if (is_v1(this->meta.version)) {
|
|
num_areas = 0x10;
|
|
} else if (is_v2(this->meta.version) && (this->meta.version != Version::GC_NTE)) {
|
|
num_areas = 0x12;
|
|
} else if (is_v3(this->meta.version) || (this->meta.version == Version::GC_NTE)) {
|
|
num_areas = 0x24;
|
|
} else if (is_ep3(this->meta.version)) {
|
|
num_areas = 0x00; // Floor overrides not allowed on Ep3
|
|
} else if (is_v4(this->meta.version)) {
|
|
num_areas = 0x2F;
|
|
} else {
|
|
throw std::logic_error("unhandled quest version");
|
|
}
|
|
for (size_t floor = 0; floor < num_floors; floor++) {
|
|
const auto& fa = this->meta.floor_assignments[floor];
|
|
if (fa.floor != floor) {
|
|
throw std::logic_error("floor assignment is inconsistent");
|
|
}
|
|
if ((fa.area != 0xFF) && (fa.area > num_areas)) {
|
|
throw std::runtime_error(std::format("floor assignment 0x{:02X} specifies invalid area 0x{:02X}", floor, fa.area));
|
|
}
|
|
}
|
|
|
|
switch (this->meta.episode) {
|
|
case Episode::EP1:
|
|
break;
|
|
case Episode::EP2:
|
|
if (is_v1_or_v2(this->meta.version)) {
|
|
throw std::runtime_error("v1 or v2 quest specifies Episode 2");
|
|
}
|
|
break;
|
|
case Episode::EP3:
|
|
if (!is_ep3(this->meta.version)) {
|
|
throw std::runtime_error("non-Ep3 quest specifies Episode 3");
|
|
}
|
|
break;
|
|
case Episode::EP4:
|
|
if (!is_v4(this->meta.version)) {
|
|
throw std::runtime_error("non-v4 quest specifies Episode 4");
|
|
}
|
|
break;
|
|
case Episode::NONE:
|
|
throw std::runtime_error("episode is not set");
|
|
default:
|
|
throw std::runtime_error("episode is not valid");
|
|
}
|
|
|
|
if (!this->bin_contents) {
|
|
throw std::runtime_error("bin file is missing");
|
|
}
|
|
if (!this->dat_contents) {
|
|
throw std::runtime_error("dat file is missing");
|
|
}
|
|
if (!this->map_file) {
|
|
throw std::runtime_error("parsed map file is missing");
|
|
}
|
|
if (this->meta.allowed_drop_modes &&
|
|
!(this->meta.allowed_drop_modes & (1 << static_cast<size_t>(this->meta.default_drop_mode)))) {
|
|
throw std::runtime_error("default drop mode is not allowed");
|
|
}
|
|
}
|
|
|
|
std::string VersionedQuest::bin_filename() const {
|
|
return std::format("quest{}.bin", this->meta.quest_number);
|
|
}
|
|
|
|
std::string VersionedQuest::dat_filename() const {
|
|
return std::format("quest{}.dat", this->meta.quest_number);
|
|
}
|
|
|
|
std::string VersionedQuest::pvr_filename() const {
|
|
return std::format("quest{}.pvr", this->meta.quest_number);
|
|
}
|
|
|
|
std::string VersionedQuest::xb_filename() const {
|
|
return std::format("quest{}_{}.dat",
|
|
this->meta.quest_number, static_cast<char>(tolower(char_for_language(this->meta.language))));
|
|
}
|
|
|
|
std::string VersionedQuest::encode_qst() const {
|
|
std::unordered_map<std::string, std::shared_ptr<const std::string>> files;
|
|
files.emplace(std::format("quest{}.bin", this->meta.quest_number), this->bin_contents);
|
|
files.emplace(std::format("quest{}.dat", this->meta.quest_number), this->dat_contents);
|
|
if (this->pvr_contents) {
|
|
files.emplace(std::format("quest{}.pvr", this->meta.quest_number), this->pvr_contents);
|
|
}
|
|
std::string xb_filename = std::format("quest{}_{}.dat",
|
|
this->meta.quest_number, static_cast<char>(tolower(char_for_language(this->meta.language))));
|
|
return encode_qst_file(files, this->meta.name, this->meta.quest_number, xb_filename, this->meta.version, this->is_dlq_encoded);
|
|
}
|
|
|
|
Quest::Quest(std::shared_ptr<const VersionedQuest> initial_version) : meta(initial_version->meta), supermap(nullptr) {
|
|
this->add_version(initial_version);
|
|
}
|
|
|
|
phosg::JSON Quest::json() const {
|
|
auto versions_json = phosg::JSON::list();
|
|
for (const auto& [_, vq] : this->versions) {
|
|
versions_json.emplace_back(phosg::JSON::dict({
|
|
{"Version", phosg::name_for_enum(vq->meta.version)},
|
|
{"Language", ::name_for_language(vq->meta.language)},
|
|
{"Name", vq->meta.name},
|
|
{"ShortDescription", vq->meta.short_description},
|
|
{"LongDescription", vq->meta.long_description},
|
|
{"BINFileSize", vq->bin_contents ? vq->bin_contents->size() : phosg::JSON(nullptr)},
|
|
{"DATFileSize", vq->dat_contents ? vq->dat_contents->size() : phosg::JSON(nullptr)},
|
|
{"PVRFileSize", vq->pvr_contents ? vq->pvr_contents->size() : phosg::JSON(nullptr)},
|
|
}));
|
|
}
|
|
|
|
return phosg::JSON::dict({{"Metadata", this->meta.json()}, {"Versions", std::move(versions_json)}});
|
|
}
|
|
|
|
uint32_t Quest::versions_key(Version v, Language language) {
|
|
return (static_cast<uint32_t>(v) << 8) | static_cast<uint8_t>(language);
|
|
}
|
|
|
|
const std::string& Quest::name_for_language(Language language) const {
|
|
size_t lang_index = static_cast<size_t>(language);
|
|
if (!this->names_by_language.at(lang_index).empty()) {
|
|
return this->names_by_language[lang_index];
|
|
}
|
|
constexpr size_t english_lang_index = static_cast<size_t>(Language::ENGLISH);
|
|
if (!this->names_by_language[english_lang_index].empty()) {
|
|
return this->names_by_language[english_lang_index];
|
|
}
|
|
for (const std::string& name : this->names_by_language) {
|
|
if (!name.empty()) {
|
|
return name;
|
|
}
|
|
}
|
|
return this->meta.name;
|
|
}
|
|
|
|
void Quest::add_version(std::shared_ptr<const VersionedQuest> vq) {
|
|
this->meta.assert_compatible(vq->meta);
|
|
if (this->meta.create_item_mask_entries.empty()) {
|
|
this->meta.create_item_mask_entries = vq->meta.create_item_mask_entries;
|
|
}
|
|
this->versions.emplace(this->versions_key(vq->meta.version, vq->meta.language), vq);
|
|
|
|
size_t lang_index = static_cast<size_t>(vq->meta.language);
|
|
auto& name_by_language = this->names_by_language.at(lang_index);
|
|
if (name_by_language.empty()) {
|
|
name_by_language = vq->meta.name;
|
|
}
|
|
}
|
|
|
|
std::shared_ptr<const SuperMap> Quest::get_supermap(int64_t random_seed) const {
|
|
if (this->supermap) {
|
|
return this->supermap;
|
|
}
|
|
|
|
bool save_to_cache = true;
|
|
bool any_map_file_present = false;
|
|
std::array<std::shared_ptr<const MapFile>, NUM_VERSIONS> map_files;
|
|
for (Version v : ALL_NON_PATCH_VERSIONS) {
|
|
auto vq = this->version(v, Language::ENGLISH);
|
|
if (vq && vq->map_file) {
|
|
auto map_file = vq->map_file;
|
|
if (map_file->has_random_sections()) {
|
|
if (random_seed < 0) {
|
|
return nullptr;
|
|
}
|
|
save_to_cache = false;
|
|
map_file = map_file->materialize_random_sections(random_seed);
|
|
}
|
|
map_files.at(static_cast<size_t>(v)) = map_file;
|
|
any_map_file_present = true;
|
|
}
|
|
}
|
|
|
|
if (!any_map_file_present) {
|
|
return nullptr;
|
|
}
|
|
|
|
auto supermap = std::make_shared<SuperMap>(map_files, this->meta.get_floor_to_area());
|
|
if (save_to_cache) {
|
|
this->supermap = supermap;
|
|
}
|
|
static_game_data_log.info_f("Constructed {} supermap for quest {} ({})",
|
|
save_to_cache ? "cacheable" : "temporary", this->meta.quest_number, this->meta.name);
|
|
|
|
return supermap;
|
|
}
|
|
|
|
bool Quest::has_version(Version v, Language language) const {
|
|
return this->versions.count(this->versions_key(v, language));
|
|
}
|
|
|
|
bool Quest::has_version_any_language(Version v) const {
|
|
uint32_t k = this->versions_key(v, Language::JAPANESE);
|
|
auto it = this->versions.lower_bound(k);
|
|
return ((it != this->versions.end()) && ((it->first & 0xFF00) == k));
|
|
}
|
|
|
|
std::shared_ptr<const VersionedQuest> Quest::version(Version v, Language language) const {
|
|
// Return the requested version, if it exists
|
|
try {
|
|
return this->versions.at(this->versions_key(v, language));
|
|
} catch (const std::out_of_range&) {
|
|
}
|
|
|
|
// Return the English version, if it exists
|
|
try {
|
|
return this->versions.at(this->versions_key(v, Language::ENGLISH));
|
|
} catch (const std::out_of_range&) {
|
|
}
|
|
|
|
// Return the first language, if it exists
|
|
auto it = this->versions.lower_bound(this->versions_key(v, Language::JAPANESE));
|
|
if ((it == this->versions.end()) || ((it->first & 0xFF00) != this->versions_key(v, Language::JAPANESE))) {
|
|
return nullptr;
|
|
}
|
|
return it->second;
|
|
}
|
|
|
|
QuestIndex::QuestIndex(
|
|
const std::string& directory, std::shared_ptr<const QuestCategoryIndex> category_index, bool raise_on_any_failure)
|
|
: directory(directory), category_index(category_index) {
|
|
|
|
struct FileData {
|
|
std::string filename;
|
|
std::shared_ptr<const std::string> data;
|
|
};
|
|
struct BINFileData {
|
|
std::string filename;
|
|
std::shared_ptr<const AssembledQuestScript> assembled;
|
|
std::shared_ptr<const std::string> data;
|
|
};
|
|
struct DATFileData {
|
|
std::string filename;
|
|
std::shared_ptr<const std::string> data;
|
|
std::shared_ptr<const MapFile> map_file;
|
|
};
|
|
std::map<std::string, BINFileData> bin_files;
|
|
std::map<std::string, DATFileData> dat_files;
|
|
std::map<std::string, FileData> pvr_files;
|
|
std::map<std::string, FileData> json_files;
|
|
std::map<std::string, uint32_t> categories;
|
|
for (const auto& cat : this->category_index->categories) {
|
|
auto add_file = [&](std::map<std::string, FileData>& files, const std::string& basename, const std::string& filename, std::string&& value, bool check_chunk_size) {
|
|
if (categories.emplace(basename, cat->category_id).first->second != cat->category_id) {
|
|
throw std::runtime_error("file " + basename + " exists in multiple categories");
|
|
}
|
|
auto data_ptr = std::make_shared<std::string>(std::move(value));
|
|
if (!files.emplace(basename, FileData{filename, data_ptr}).second) {
|
|
throw std::runtime_error("file " + basename + " already exists");
|
|
}
|
|
// There is a bug in the client that prevents quests from loading properly if any file's size is a multiple of
|
|
// 0x400. See the comments on the 13 command in CommandFormats.hh for more details.
|
|
if (check_chunk_size && !(data_ptr->size() & 0x3FF)) {
|
|
data_ptr->push_back(0x00);
|
|
}
|
|
};
|
|
|
|
auto add_bin_file = [&](const std::string& basename, const std::string& filename, std::string&& data, std::shared_ptr<AssembledQuestScript> assembled) {
|
|
if (categories.emplace(basename, cat->category_id).first->second != cat->category_id) {
|
|
throw std::runtime_error("bin file " + basename + " exists in multiple categories");
|
|
}
|
|
auto data_ptr = std::make_shared<std::string>(std::move(data));
|
|
auto emplace_ret = bin_files.emplace(basename, BINFileData{});
|
|
if (!emplace_ret.second) {
|
|
throw std::runtime_error("bin file " + basename + " already exists");
|
|
}
|
|
auto& entry = emplace_ret.first->second;
|
|
entry.filename = filename;
|
|
entry.data = data_ptr;
|
|
entry.assembled = assembled;
|
|
if (!(data_ptr->size() & 0x3FF)) {
|
|
data_ptr->push_back(0x00);
|
|
}
|
|
};
|
|
auto add_dat_file = [&](const std::string& basename, const std::string& filename, std::string&& data) {
|
|
if (categories.emplace(basename, cat->category_id).first->second != cat->category_id) {
|
|
throw std::runtime_error("dat file " + basename + " exists in multiple categories");
|
|
}
|
|
auto data_ptr = std::make_shared<std::string>(std::move(data));
|
|
auto map_file = std::make_shared<MapFile>(std::make_shared<std::string>(prs_decompress(*data_ptr)));
|
|
if (!dat_files.emplace(basename, DATFileData{filename, data_ptr, map_file}).second) {
|
|
throw std::runtime_error("dat file " + basename + " already exists");
|
|
}
|
|
if (!(data_ptr->size() & 0x3FF)) {
|
|
data_ptr->push_back(0x00);
|
|
}
|
|
};
|
|
|
|
std::string cat_path = directory + "/" + cat->directory_name;
|
|
if (!std::filesystem::is_directory(cat_path)) {
|
|
static_game_data_log.warning_f("Quest category directory {} is missing; skipping it", cat_path);
|
|
continue;
|
|
}
|
|
for (const auto& item : std::filesystem::directory_iterator(cat_path)) {
|
|
std::string filename = item.path().filename().string();
|
|
if (filename == ".DS_Store") {
|
|
continue;
|
|
}
|
|
|
|
std::string file_path = cat_path + "/" + filename;
|
|
std::shared_ptr<AssembledQuestScript> assembled;
|
|
try {
|
|
std::string orig_filename = filename;
|
|
std::string file_data;
|
|
if (filename.ends_with(".gci")) {
|
|
file_data = decode_gci_data(phosg::load_file(file_path));
|
|
filename.resize(filename.size() - 4);
|
|
} else if (filename.ends_with(".vms")) {
|
|
file_data = decode_vms_data(phosg::load_file(file_path));
|
|
filename.resize(filename.size() - 4);
|
|
} else if (filename.ends_with(".dlq")) {
|
|
file_data = decode_dlq_data(phosg::load_file(file_path));
|
|
filename.resize(filename.size() - 4);
|
|
} else if (filename.ends_with(".bin.txt")) {
|
|
std::string include_dir = phosg::dirname(file_path);
|
|
assembled = std::make_shared<AssembledQuestScript>(assemble_quest_script(
|
|
phosg::load_file(file_path),
|
|
{include_dir, "system/quests/includes"},
|
|
{include_dir, "system/quests/includes", "system/client-functions/System"}));
|
|
file_data = std::move(assembled->data);
|
|
filename.resize(filename.size() - 4);
|
|
if (filename.ends_with(".bin")) {
|
|
filename.push_back('d');
|
|
}
|
|
} else {
|
|
file_data = phosg::load_file(file_path);
|
|
}
|
|
|
|
size_t dot_pos = filename.rfind('.');
|
|
std::string file_basename;
|
|
std::string extension;
|
|
if (dot_pos != std::string::npos) {
|
|
file_basename = phosg::tolower(filename.substr(0, dot_pos));
|
|
extension = phosg::tolower(filename.substr(dot_pos + 1));
|
|
} else {
|
|
file_basename = phosg::tolower(filename);
|
|
}
|
|
|
|
if (extension == "json") {
|
|
add_file(json_files, file_basename, orig_filename, std::move(file_data), false);
|
|
} else if (extension == "bin" || extension == "mnm") {
|
|
add_bin_file(file_basename, orig_filename, std::move(file_data), assembled);
|
|
} else if (extension == "bind" || extension == "mnmd") {
|
|
add_bin_file(file_basename, orig_filename, prs_compress_optimal(file_data), assembled);
|
|
} else if (extension == "dat") {
|
|
add_dat_file(file_basename, orig_filename, std::move(file_data));
|
|
} else if (extension == "datd") {
|
|
add_dat_file(file_basename, orig_filename, prs_compress_optimal(file_data));
|
|
} else if (extension == "pvr") {
|
|
add_file(pvr_files, file_basename, orig_filename, std::move(file_data), true);
|
|
} else if (extension == "qst") {
|
|
auto files = decode_qst_data(file_data);
|
|
for (auto& it : files) {
|
|
if (it.first.ends_with(".bin")) {
|
|
add_bin_file(file_basename, orig_filename, std::move(it.second), nullptr);
|
|
} else if (it.first.ends_with(".dat")) {
|
|
add_dat_file(file_basename, orig_filename, std::move(it.second));
|
|
} else if (it.first.ends_with(".pvr")) {
|
|
add_file(pvr_files, file_basename, orig_filename, std::move(it.second), true);
|
|
} else {
|
|
throw std::runtime_error("qst file contains unsupported file type: " + it.first);
|
|
}
|
|
}
|
|
}
|
|
|
|
} catch (const std::exception& e) {
|
|
if (raise_on_any_failure) {
|
|
throw std::runtime_error(std::format("({}) {}", filename, e.what()));
|
|
}
|
|
static_game_data_log.warning_f("({}) Failed to load quest file: ({})", filename, e.what());
|
|
}
|
|
}
|
|
}
|
|
|
|
// All quests have a bin file (even in Episode 3, though its format is different), so we use bin_files as the primary
|
|
// list of all quests that should be indexed
|
|
std::unordered_map<const FileData*, std::shared_ptr<const phosg::JSON>> parsed_json_files;
|
|
for (auto& [basename, entry] : bin_files) {
|
|
try {
|
|
auto vq = std::make_shared<VersionedQuest>();
|
|
|
|
// Quest .bin filenames are like K###-VERS-LANG.EXT, where:
|
|
// K can be any character (usually it's q)
|
|
// # = quest number (does not have to match the internal quest number)
|
|
// VERS = PSO version that the quest is for (dc, pc, gc, etc.)
|
|
// LANG = client language (j, e, g, f, s)
|
|
// EXT = file type (bin, bind, bin.dlq, qst, etc.)
|
|
// EXT has already been stripped off by the time we get here, so we just parse the remaining fields.
|
|
std::string quest_number_token, version_token, language_token;
|
|
{
|
|
std::vector<std::string> filename_tokens = phosg::split(basename, '-');
|
|
if (filename_tokens.size() != 3) {
|
|
throw std::invalid_argument("incorrect filename format");
|
|
}
|
|
quest_number_token = std::move(filename_tokens[0]);
|
|
version_token = std::move(filename_tokens[1]);
|
|
language_token = std::move(filename_tokens[2]);
|
|
}
|
|
vq->meta.category_id = categories.at(basename);
|
|
|
|
if (entry.assembled) {
|
|
vq->meta.quest_number = entry.assembled->meta.quest_number;
|
|
vq->meta.version = entry.assembled->meta.version;
|
|
vq->meta.language = entry.assembled->meta.language;
|
|
} else {
|
|
// Get the number from the first token
|
|
if (quest_number_token.empty()) {
|
|
throw std::runtime_error("quest number token is missing");
|
|
}
|
|
vq->meta.quest_number = strtoull(quest_number_token.c_str() + 1, nullptr, 10);
|
|
|
|
// Get the version from the second token
|
|
static const std::unordered_map<std::string, Version> name_to_version({
|
|
{"dn", Version::DC_NTE},
|
|
{"dp", Version::DC_11_2000},
|
|
{"d1", Version::DC_V1},
|
|
{"dc", Version::DC_V2},
|
|
{"pcn", Version::PC_NTE},
|
|
{"pc", Version::PC_V2},
|
|
{"gcn", Version::GC_NTE},
|
|
{"gc", Version::GC_V3},
|
|
{"gc3t", Version::GC_EP3_NTE},
|
|
{"gc3", Version::GC_EP3},
|
|
{"xb", Version::XB_V3},
|
|
{"bb", Version::BB_V4},
|
|
});
|
|
vq->meta.version = name_to_version.at(version_token);
|
|
|
|
// Get the language from the last token
|
|
if (language_token.size() != 1) {
|
|
throw std::runtime_error("language token is not a single character");
|
|
}
|
|
vq->meta.language = language_for_char(language_token[0]);
|
|
}
|
|
|
|
auto bin_decompressed = prs_decompress(*entry.data);
|
|
populate_quest_metadata_from_script(vq->meta, bin_decompressed.data(), bin_decompressed.size(), vq->meta.version, vq->meta.language);
|
|
|
|
// Find the corresponding dat and pvr files with the same basename as the bin file; if not found, look for them
|
|
// without the language suffix
|
|
const DATFileData* dat_filedata = nullptr;
|
|
const FileData* pvr_filedata = nullptr;
|
|
try {
|
|
dat_filedata = &dat_files.at(basename);
|
|
} catch (const std::out_of_range&) {
|
|
try {
|
|
dat_filedata = &dat_files.at(quest_number_token + "-" + version_token);
|
|
} catch (const std::out_of_range&) {
|
|
throw std::runtime_error("no dat file found for bin file " + basename);
|
|
}
|
|
}
|
|
try {
|
|
pvr_filedata = &pvr_files.at(basename);
|
|
} catch (const std::out_of_range&) {
|
|
try {
|
|
pvr_filedata = &pvr_files.at(quest_number_token + "-" + version_token);
|
|
} catch (const std::out_of_range&) {
|
|
// pvr files aren't required (and most quests do not have them), so don't fail if it's missing
|
|
}
|
|
}
|
|
vq->bin_contents = entry.data;
|
|
if (dat_filedata) {
|
|
vq->dat_contents = dat_filedata->data;
|
|
vq->map_file = dat_filedata->map_file;
|
|
}
|
|
if (pvr_filedata) {
|
|
vq->pvr_contents = pvr_filedata->data;
|
|
}
|
|
|
|
// Load the quest's metadata JSON file, if it exists
|
|
const FileData* json_filedata = nullptr;
|
|
try {
|
|
json_filedata = &json_files.at(basename);
|
|
} catch (const std::out_of_range&) {
|
|
try {
|
|
json_filedata = &json_files.at(quest_number_token + "-" + version_token);
|
|
} catch (const std::out_of_range&) {
|
|
try {
|
|
json_filedata = &json_files.at(quest_number_token);
|
|
} catch (const std::out_of_range&) {
|
|
}
|
|
}
|
|
}
|
|
if (json_filedata) {
|
|
try {
|
|
vq->json_contents = parsed_json_files.at(json_filedata);
|
|
} catch (const std::out_of_range&) {
|
|
vq->json_contents = std::make_shared<phosg::JSON>(phosg::JSON::parse(*json_filedata->data));
|
|
parsed_json_files.emplace(json_filedata, vq->json_contents);
|
|
}
|
|
vq->meta.apply_json_overrides(*vq->json_contents);
|
|
}
|
|
|
|
vq->assert_valid();
|
|
|
|
auto category_name = this->category_index->at(vq->meta.category_id)->name;
|
|
std::string filenames_str = entry.filename;
|
|
if (dat_filedata) {
|
|
filenames_str += std::format("/{}", dat_filedata->filename);
|
|
}
|
|
if (pvr_filedata) {
|
|
filenames_str += std::format("/{}", pvr_filedata->filename);
|
|
}
|
|
if (json_filedata) {
|
|
filenames_str += std::format("/{}", json_filedata->filename);
|
|
}
|
|
auto q_it = this->quests_by_number.find(vq->meta.quest_number);
|
|
if (q_it != this->quests_by_number.end()) {
|
|
q_it->second->add_version(vq);
|
|
static_game_data_log.debug_f("({}) Added {} {} version of quest {} ({})",
|
|
filenames_str,
|
|
phosg::name_for_enum(vq->meta.version),
|
|
char_for_language(vq->meta.language),
|
|
vq->meta.quest_number,
|
|
vq->meta.name);
|
|
} else {
|
|
auto q = std::make_shared<Quest>(vq);
|
|
this->quests_by_number.emplace(vq->meta.quest_number, q);
|
|
this->quests_by_name.emplace(vq->meta.name, q);
|
|
this->quests_by_category_id_and_number[q->meta.category_id].emplace(vq->meta.quest_number, q);
|
|
static_game_data_log.debug_f("({}) Created {} {} quest {} ({}) ({}, {} ({}), {})",
|
|
filenames_str,
|
|
phosg::name_for_enum(vq->meta.version),
|
|
char_for_language(vq->meta.language),
|
|
vq->meta.quest_number,
|
|
vq->meta.name,
|
|
name_for_episode(vq->meta.episode),
|
|
category_name,
|
|
vq->meta.category_id,
|
|
vq->meta.joinable ? "joinable" : "not joinable");
|
|
}
|
|
} catch (const std::exception& e) {
|
|
if (raise_on_any_failure) {
|
|
throw std::runtime_error(std::format("({}) {}", basename, e.what()));
|
|
}
|
|
static_game_data_log.warning_f("({}) Failed to index quest file: {}", basename, e.what());
|
|
}
|
|
}
|
|
}
|
|
|
|
phosg::JSON QuestIndex::json() const {
|
|
auto categories_json = phosg::JSON::dict();
|
|
for (const auto& cat : this->category_index->categories) {
|
|
auto dict = phosg::JSON::dict({
|
|
{"CategoryID", cat->category_id},
|
|
{"Flags", cat->enabled_flags},
|
|
{"DirectoryName", cat->directory_name},
|
|
{"Name", cat->name},
|
|
{"Description", cat->description},
|
|
});
|
|
categories_json.emplace(cat->name, std::move(dict));
|
|
}
|
|
|
|
auto quests_json = phosg::JSON::list();
|
|
for (const auto& [_, q] : this->quests_by_number) {
|
|
quests_json.emplace_back(q->json());
|
|
}
|
|
|
|
return phosg::JSON::dict({
|
|
{"Directory", this->directory},
|
|
{"Categories", std::move(categories_json)},
|
|
{"Quests", std::move(quests_json)},
|
|
});
|
|
}
|
|
|
|
std::shared_ptr<const Quest> QuestIndex::get(uint32_t quest_number) const {
|
|
try {
|
|
return this->quests_by_number.at(quest_number);
|
|
} catch (const std::out_of_range&) {
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
std::shared_ptr<const Quest> QuestIndex::get(const std::string& name) const {
|
|
try {
|
|
return this->quests_by_name.at(name);
|
|
} catch (const std::out_of_range&) {
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
std::vector<std::shared_ptr<const QuestCategoryIndex::Category>> QuestIndex::categories(
|
|
QuestMenuType menu_type, Episode episode, uint16_t version_flags, IncludeCondition include_condition) const {
|
|
std::vector<std::shared_ptr<const QuestCategoryIndex::Category>> ret;
|
|
for (const auto& cat : this->category_index->categories) {
|
|
if (cat->check_flag(menu_type) && !this->filter(episode, version_flags, cat->category_id, include_condition, 1).empty()) {
|
|
ret.emplace_back(cat);
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
std::vector<std::pair<QuestIndex::IncludeState, std::shared_ptr<const Quest>>> QuestIndex::filter(
|
|
Episode episode,
|
|
uint16_t version_flags,
|
|
uint32_t category_id,
|
|
IncludeCondition include_condition,
|
|
size_t limit) const {
|
|
auto cat = this->category_index->at(category_id);
|
|
Episode effective_episode = cat->enable_episode_filter() ? episode : Episode::NONE;
|
|
|
|
std::vector<std::pair<IncludeState, std::shared_ptr<const Quest>>> ret;
|
|
auto category_it = this->quests_by_category_id_and_number.find(category_id);
|
|
if (category_it == this->quests_by_category_id_and_number.end()) {
|
|
return ret;
|
|
}
|
|
for (auto it : category_it->second) {
|
|
if ((effective_episode != Episode::NONE) && (it.second->meta.episode != effective_episode)) {
|
|
continue;
|
|
}
|
|
bool all_required_versions_present = true;
|
|
for (size_t v_s = 0; v_s < NUM_VERSIONS; v_s++) {
|
|
if ((version_flags & (1 << v_s)) && !it.second->has_version_any_language(static_cast<Version>(v_s))) {
|
|
all_required_versions_present = false;
|
|
break;
|
|
}
|
|
}
|
|
if (!all_required_versions_present) {
|
|
continue;
|
|
}
|
|
IncludeState state = include_condition ? include_condition(it.second) : IncludeState::AVAILABLE;
|
|
if (state == IncludeState::HIDDEN) {
|
|
continue;
|
|
}
|
|
ret.emplace_back(make_pair(state, it.second));
|
|
if (limit && (ret.size() >= limit)) {
|
|
break;
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
std::string encode_download_quest_data(
|
|
const std::string& compressed_data, size_t decompressed_size, uint32_t encryption_seed) {
|
|
// Download quest files are like normal (PRS-compressed) quest files, but they are encrypted with PSO V2 encryption
|
|
// (even on V3 / PSO GC), and a small header (PSODownloadQuestHeader) is prepended to the encrypted data.
|
|
|
|
if (encryption_seed == 0) {
|
|
encryption_seed = phosg::random_object<uint32_t>();
|
|
}
|
|
if (decompressed_size == 0) {
|
|
decompressed_size = prs_decompress_size(compressed_data);
|
|
}
|
|
|
|
std::string data(8, '\0');
|
|
auto* header = reinterpret_cast<PSODownloadQuestHeader*>(data.data());
|
|
header->size = decompressed_size;
|
|
header->encryption_seed = encryption_seed;
|
|
data += compressed_data;
|
|
|
|
// Add extra bytes if necessary so encryption won't fail; the data size must be a multiple of 4 for PSO V2 encryption
|
|
size_t original_size = data.size();
|
|
data.resize((data.size() + 3) & (~3));
|
|
|
|
PSOV2Encryption encr(encryption_seed);
|
|
encr.encrypt(data.data() + sizeof(PSODownloadQuestHeader), data.size() - sizeof(PSODownloadQuestHeader));
|
|
data.resize(original_size);
|
|
|
|
return data;
|
|
}
|
|
|
|
std::shared_ptr<VersionedQuest> VersionedQuest::create_download_quest(Language override_language) const {
|
|
// The download flag needs to be set in the bin header, or else the client will ignore it when scanning for download
|
|
// quests in an offline game. To set this flag, we need to decompress the quest's .bin file, set the flag, then
|
|
// recompress it again.
|
|
|
|
std::string decompressed_bin = prs_decompress(*this->bin_contents);
|
|
|
|
void* data_ptr = decompressed_bin.data();
|
|
switch (this->meta.version) {
|
|
case Version::DC_NTE:
|
|
if (decompressed_bin.size() < sizeof(PSOQuestHeaderDCNTE)) {
|
|
throw std::runtime_error("bin file is too small for header");
|
|
}
|
|
// There's no known language field in this version, so we don't write anything here
|
|
break;
|
|
case Version::DC_11_2000:
|
|
case Version::DC_V1:
|
|
case Version::DC_V2:
|
|
if (decompressed_bin.size() < sizeof(PSOQuestHeaderDC)) {
|
|
throw std::runtime_error("bin file is too small for header");
|
|
}
|
|
if (override_language != Language::UNKNOWN) {
|
|
reinterpret_cast<PSOQuestHeaderDC*>(data_ptr)->language = override_language;
|
|
}
|
|
break;
|
|
case Version::PC_NTE:
|
|
case Version::PC_V2:
|
|
if (decompressed_bin.size() < sizeof(PSOQuestHeaderPC)) {
|
|
throw std::runtime_error("bin file is too small for header");
|
|
}
|
|
if (override_language != Language::UNKNOWN) {
|
|
reinterpret_cast<PSOQuestHeaderPC*>(data_ptr)->language = override_language;
|
|
}
|
|
break;
|
|
case Version::GC_NTE:
|
|
case Version::GC_V3:
|
|
case Version::XB_V3:
|
|
if (decompressed_bin.size() < sizeof(PSOQuestHeaderV3)) {
|
|
throw std::runtime_error("bin file is too small for header");
|
|
}
|
|
if (override_language != Language::UNKNOWN) {
|
|
reinterpret_cast<PSOQuestHeaderV3*>(data_ptr)->language = override_language;
|
|
}
|
|
break;
|
|
case Version::BB_V4:
|
|
throw std::invalid_argument("PSOBB does not support download quests");
|
|
default:
|
|
throw std::invalid_argument("unknown game version");
|
|
}
|
|
|
|
std::string compressed_bin = prs_compress(decompressed_bin);
|
|
|
|
// Return a new VersionedQuest object with appropriately-processed .bin and .dat file contents
|
|
auto dlq = std::make_shared<VersionedQuest>(*this);
|
|
dlq->bin_contents = std::make_shared<std::string>(encode_download_quest_data(compressed_bin, decompressed_bin.size()));
|
|
dlq->dat_contents = std::make_shared<std::string>(encode_download_quest_data(*this->dat_contents));
|
|
dlq->pvr_contents = this->pvr_contents;
|
|
dlq->is_dlq_encoded = true;
|
|
return dlq;
|
|
}
|
|
|
|
std::string decode_gci_data(
|
|
const std::string& data, ssize_t find_seed_num_threads, int64_t known_seed, bool skip_checksum) {
|
|
phosg::StringReader r(data);
|
|
const auto& header = r.get<PSOGCIFileHeader>();
|
|
header.check();
|
|
|
|
if (header.is_ep12()) {
|
|
const auto& dlq_header = r.get<PSOGCIDLQFileEncryptedHeader>(false);
|
|
// Unencrypted GCI files appear to always have zeroes in these fields. Encrypted GCI files are highly unlikely to
|
|
// have zeroes in ALL of these fields, so assume it's encrypted if any of them are nonzero.
|
|
if (dlq_header.round2_seed || dlq_header.checksum || dlq_header.round3_seed) {
|
|
if (known_seed >= 0) {
|
|
return decrypt_download_quest_data_section<true>(
|
|
r.getv(header.data_size), header.data_size, known_seed, skip_checksum, false);
|
|
|
|
} else if (header.embedded_seed != 0) {
|
|
return decrypt_download_quest_data_section<true>(
|
|
r.getv(header.data_size), header.data_size, header.embedded_seed, skip_checksum, false);
|
|
|
|
} else {
|
|
if (find_seed_num_threads < 0) {
|
|
throw std::runtime_error("file is encrypted");
|
|
}
|
|
if (find_seed_num_threads == 0) {
|
|
find_seed_num_threads = std::thread::hardware_concurrency();
|
|
}
|
|
return find_seed_and_decrypt_download_quest_data_section<true>(
|
|
r.getv(header.data_size), header.data_size, skip_checksum, false, find_seed_num_threads);
|
|
}
|
|
|
|
} else { // Unencrypted GCI format
|
|
r.skip(sizeof(PSOGCIDLQFileEncryptedHeader));
|
|
std::string compressed_data = r.readx(header.data_size - sizeof(PSOGCIDLQFileEncryptedHeader));
|
|
size_t decompressed_bytes = prs_decompress_size(compressed_data);
|
|
|
|
size_t expected_decompressed_bytes = dlq_header.decompressed_size - 8;
|
|
if (decompressed_bytes < expected_decompressed_bytes) {
|
|
throw std::runtime_error(std::format(
|
|
"GCI decompressed data is smaller than expected size (have 0x{:X} bytes, expected 0x{:X} bytes)",
|
|
decompressed_bytes, expected_decompressed_bytes));
|
|
}
|
|
|
|
return compressed_data;
|
|
}
|
|
|
|
} else if (header.is_ep3()) {
|
|
if (header.is_nte()) {
|
|
if (known_seed >= 0) {
|
|
return decrypt_download_quest_data_section<true>(
|
|
r.getv(header.data_size), header.data_size, known_seed, true, true);
|
|
} else {
|
|
if (find_seed_num_threads < 0) {
|
|
throw std::runtime_error("file is encrypted");
|
|
}
|
|
if (find_seed_num_threads == 0) {
|
|
find_seed_num_threads = std::thread::hardware_concurrency();
|
|
}
|
|
return find_seed_and_decrypt_download_quest_data_section<true>(
|
|
r.getv(header.data_size), header.data_size, true, true, find_seed_num_threads);
|
|
}
|
|
|
|
} else {
|
|
// The first 0x10 bytes in the data segment appear to be unused. In most files I've seen, the last half of it (8
|
|
// bytes) are duplicates of the first 8 bytes of the unscrambled, compressed data, though this is the result of
|
|
// an uninitialized memory bug when the client encodes the file and not an actual constraint on what should be in
|
|
// these 8 bytes.
|
|
r.skip(16);
|
|
// The game treats this field as a 16-byte string (including the \0). The 8 bytes after it appear to be
|
|
// completely unused.
|
|
if (r.readx(15) != "SONICTEAM,SEGA.") {
|
|
throw std::runtime_error("Episode 3 GCI file is not a quest");
|
|
}
|
|
r.skip(9);
|
|
|
|
std::string decrypted = r.readx(header.data_size - 40);
|
|
|
|
// For some reason, Sega decided not to encrypt Episode 3 quest files in the same way as Episodes 1&2 quest files
|
|
// (see above). Instead, they just wrote a fairly trivial XOR loop over the first 0x100 bytes, leaving the
|
|
// remaining bytes completely unencrypted (but still compressed).
|
|
size_t unscramble_size = std::min<size_t>(0x100, data.size());
|
|
decrypt_trivial_gci_data(decrypted.data(), unscramble_size, 0);
|
|
|
|
size_t decompressed_size = prs_decompress_size(decrypted);
|
|
if (decompressed_size != sizeof(Episode3::MapDefinition)) {
|
|
throw std::runtime_error(std::format(
|
|
"decompressed quest is 0x{:X} bytes; expected 0x{:X} bytes",
|
|
decompressed_size, sizeof(Episode3::MapDefinition)));
|
|
}
|
|
return decrypted;
|
|
}
|
|
|
|
} else {
|
|
throw std::runtime_error("unknown game name in GCI header");
|
|
}
|
|
}
|
|
|
|
std::string decode_vms_data(
|
|
const std::string& data, ssize_t find_seed_num_threads, int64_t known_seed, bool skip_checksum) {
|
|
phosg::StringReader r(data);
|
|
const auto& header = r.get<PSOVMSFileHeader>();
|
|
if (!header.checksum_correct()) {
|
|
throw std::runtime_error("VMS file unencrypted header checksum is incorrect");
|
|
}
|
|
r.skip(header.num_icons * 0x200);
|
|
|
|
const void* data_section = r.getv(header.data_size);
|
|
try {
|
|
return decrypt_vms_v1_data_section(data_section, header.data_size);
|
|
} catch (const std::exception& e) {
|
|
}
|
|
|
|
if (known_seed >= 0) {
|
|
return decrypt_download_quest_data_section<false>(data_section, header.data_size, known_seed);
|
|
|
|
} else {
|
|
if (find_seed_num_threads < 0) {
|
|
throw std::runtime_error("file is encrypted");
|
|
}
|
|
if (find_seed_num_threads == 0) {
|
|
find_seed_num_threads = std::thread::hardware_concurrency();
|
|
}
|
|
return find_seed_and_decrypt_download_quest_data_section<false>(
|
|
data_section, header.data_size, skip_checksum, 0, find_seed_num_threads);
|
|
}
|
|
}
|
|
|
|
std::string decode_dlq_data(const std::string& data) {
|
|
phosg::StringReader r(data);
|
|
uint32_t decompressed_size = r.get_u32l();
|
|
uint32_t key = r.get_u32l();
|
|
|
|
// The compressed data size does not need to be a multiple of 4, but the V2 encryption (which is used for all
|
|
// download quests, even in V3) requires the data size to be a multiple of 4. We'll just temporarily stick a few
|
|
// bytes on the end, then throw them away later if needed.
|
|
std::string decrypted = r.read(r.remaining());
|
|
PSOV2Encryption encr(key);
|
|
size_t original_size = data.size();
|
|
decrypted.resize((decrypted.size() + 3) & (~3));
|
|
encr.decrypt(decrypted);
|
|
decrypted.resize(original_size);
|
|
|
|
if (prs_decompress_size(decrypted) != decompressed_size) {
|
|
throw std::runtime_error("decompressed size does not match size in header");
|
|
}
|
|
|
|
return decrypted;
|
|
}
|
|
|
|
template <typename HeaderT, typename OpenFileT>
|
|
static std::unordered_map<std::string, std::string> decode_qst_data_t(const std::string& data) {
|
|
phosg::StringReader r(data);
|
|
|
|
struct File {
|
|
struct Block {
|
|
uint32_t index = 0;
|
|
const void* data = nullptr;
|
|
size_t size = 0;
|
|
};
|
|
std::deque<Block> blocks;
|
|
size_t total_size = 0;
|
|
};
|
|
std::unordered_map<std::string, File> files;
|
|
|
|
QuestFileFormat subformat = QuestFileFormat::QST; // Stand-in for unknown
|
|
bool assemble_in_order = true;
|
|
while (!r.eof()) {
|
|
// Handle BB's implicit 8-byte command alignment
|
|
static constexpr size_t alignment = sizeof(HeaderT);
|
|
size_t next_command_offset = (r.where() + (alignment - 1)) & ~(alignment - 1);
|
|
r.go(next_command_offset);
|
|
if (r.eof()) {
|
|
break;
|
|
}
|
|
|
|
const auto& header = r.get<HeaderT>();
|
|
|
|
if (header.command == 0x44 || header.command == 0x13) {
|
|
if (subformat == QuestFileFormat::QST) {
|
|
subformat = QuestFileFormat::BIN_DAT;
|
|
} else if (subformat != QuestFileFormat::BIN_DAT) {
|
|
throw std::runtime_error("QST file contains mixed download and non-download commands");
|
|
}
|
|
} else if (header.command == 0xA6 || header.command == 0xA7) {
|
|
if (subformat == QuestFileFormat::QST) {
|
|
subformat = QuestFileFormat::BIN_DAT_DLQ;
|
|
} else if (subformat != QuestFileFormat::BIN_DAT_DLQ) {
|
|
throw std::runtime_error("QST file contains mixed download and non-download commands");
|
|
}
|
|
}
|
|
|
|
if (header.command == 0x44 || header.command == 0xA6) {
|
|
if (header.size != sizeof(HeaderT) + sizeof(OpenFileT)) {
|
|
throw std::runtime_error("qst open file command has incorrect size");
|
|
}
|
|
const auto& cmd = r.get<OpenFileT>();
|
|
std::string internal_filename = cmd.filename.decode();
|
|
if (!files.emplace(internal_filename, File{{}, cmd.file_size}).second) {
|
|
throw std::runtime_error("qst opens the same file multiple times: " + internal_filename);
|
|
}
|
|
|
|
} else if (header.command == 0x13 || header.command == 0xA7) {
|
|
// We have to allow larger commands here, because it seems some tools encoded QST files with BB's extra 4 padding
|
|
// bytes included in the command size.
|
|
if (header.size < sizeof(HeaderT) + sizeof(S_WriteFile_13_A7)) {
|
|
throw std::runtime_error("qst write file command has incorrect size");
|
|
}
|
|
const auto& cmd = r.get<S_WriteFile_13_A7>();
|
|
std::string filename = cmd.filename.decode();
|
|
if (cmd.data_size > 0x400) {
|
|
throw std::runtime_error(std::format(
|
|
"qst chunk {}:{:02X} has invalid size {:08X}", filename, header.flag, cmd.data_size));
|
|
}
|
|
|
|
auto& file = files.at(filename);
|
|
size_t offset = header.flag * 0x400;
|
|
if (offset + cmd.data_size > file.total_size) {
|
|
throw std::runtime_error(std::format("qst chunk {}:{:02X} extends beyond end of file", filename, header.flag));
|
|
}
|
|
auto& block = file.blocks.emplace_back();
|
|
block.index = header.flag;
|
|
block.data = cmd.data.data();
|
|
block.size = cmd.data_size;
|
|
|
|
if (header.flag) {
|
|
assemble_in_order = false;
|
|
}
|
|
|
|
} else {
|
|
throw std::runtime_error("invalid command in qst file");
|
|
}
|
|
}
|
|
|
|
// QST is an unfortunately underspecified format, and there are technically many versions of it out there (though I'm
|
|
// sure no one has ever thought much about this). Here we handle one of the format's quirks, in which some QST files
|
|
// have all zeroes for chunk indexes, so we just assume in that case that the chunks are in the correct order in the
|
|
// source file.
|
|
std::unordered_map<std::string, std::string> ret;
|
|
for (const auto& [name, file] : files) {
|
|
std::string& assembled_file = ret.emplace(name, "").first->second;
|
|
if (assemble_in_order) {
|
|
static_game_data_log.warning_f(
|
|
"QST file entry {} is missing block indexes; assuming blocks are assembled in order", name);
|
|
for (const auto& block : file.blocks) {
|
|
assembled_file.append(reinterpret_cast<const char*>(block.data), block.size);
|
|
}
|
|
} else {
|
|
assembled_file.resize(file.total_size, 0x00);
|
|
for (const auto& block : file.blocks) {
|
|
memcpy(assembled_file.data() + block.index * 0x400, block.data, block.size);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (subformat == QuestFileFormat::BIN_DAT_DLQ) {
|
|
for (auto& [_, data] : ret) {
|
|
data = decode_dlq_data(data);
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
std::unordered_map<std::string, std::string> decode_qst_data(const std::string& data) {
|
|
// QST files start with an open file command, but the format differs depending on the PSO version that the qst file
|
|
// is for. We can detect the format from the first 4 bytes in the file:
|
|
// - BB: 58 00 44 00 or 58 00 A6 00
|
|
// - PC: 3C 00 44 ?? or 3C 00 A6 ??
|
|
// - DC/GC: 44 ?? 3C 00 or A6 ?? 3C 00
|
|
// - XB: 44 ?? 54 00 or A6 ?? 54 00
|
|
phosg::StringReader r(data);
|
|
uint32_t signature = r.get_u32b();
|
|
if ((signature == 0x58004400) || (signature == 0x5800A600)) {
|
|
return decode_qst_data_t<PSOCommandHeaderBB, S_OpenFile_BB_44_A6>(data);
|
|
} else if (((signature & 0xFFFFFF00) == 0x3C004400) || ((signature & 0xFFFFFF00) == 0x3C00A600)) {
|
|
return decode_qst_data_t<PSOCommandHeaderPC, S_OpenFile_PC_GC_44_A6>(data);
|
|
} else if (((signature & 0xFF00FFFF) == 0x44003C00) || ((signature & 0xFF00FFFF) == 0xA6003C00)) {
|
|
// In PSO DC, the type field is only one byte, but in V3 it's two bytes and the filename was shifted over by one
|
|
// byte. To detect this, we check if the V3 type field has a reasonable value, and if not, we assume the file is
|
|
// for PSO DC.
|
|
if (r.pget_u16l(sizeof(PSOCommandHeaderDCV3) + offsetof(S_OpenFile_PC_GC_44_A6, type)) > 3) {
|
|
return decode_qst_data_t<PSOCommandHeaderDCV3, S_OpenFile_DC_44_A6>(data);
|
|
} else {
|
|
return decode_qst_data_t<PSOCommandHeaderDCV3, S_OpenFile_PC_GC_44_A6>(data);
|
|
}
|
|
} else if (((signature & 0xFF00FFFF) == 0x44005400) || ((signature & 0xFF00FFFF) == 0xA6005400)) {
|
|
return decode_qst_data_t<PSOCommandHeaderDCV3, S_OpenFile_XB_44_A6>(data);
|
|
} else {
|
|
throw std::runtime_error("invalid qst file format");
|
|
}
|
|
}
|
|
|
|
template <typename HeaderT>
|
|
void add_command_header(phosg::StringWriter& w, uint8_t command, uint8_t flag, uint16_t size) {
|
|
HeaderT header;
|
|
header.command = command;
|
|
header.flag = flag;
|
|
header.size = sizeof(HeaderT) + size;
|
|
w.put(header);
|
|
}
|
|
|
|
template <typename HeaderT, typename CmdT>
|
|
void add_open_file_command_t(
|
|
phosg::StringWriter& w,
|
|
const std::string& name,
|
|
const std::string& filename,
|
|
const std::string&,
|
|
uint32_t,
|
|
size_t file_size,
|
|
bool is_download) {
|
|
add_command_header<HeaderT>(w, is_download ? 0xA6 : 0x44, 0x00, sizeof(CmdT));
|
|
CmdT cmd;
|
|
cmd.name.assign_raw("PSO/" + name);
|
|
cmd.filename.encode(filename);
|
|
cmd.type = 0;
|
|
cmd.file_size = file_size;
|
|
// TODO: It'd be nice to have something like w.emplace(...) to avoid copying the structs into the StringWriter.
|
|
w.put(cmd);
|
|
}
|
|
|
|
template <>
|
|
void add_open_file_command_t<PSOCommandHeaderDCV3, S_OpenFile_XB_44_A6>(
|
|
phosg::StringWriter& w,
|
|
const std::string& name,
|
|
const std::string& filename,
|
|
const std::string& xb_filename,
|
|
uint32_t quest_number,
|
|
size_t file_size,
|
|
bool is_download) {
|
|
add_command_header<PSOCommandHeaderDCV3>(w, is_download ? 0xA6 : 0x44, 0x00, sizeof(S_OpenFile_XB_44_A6));
|
|
S_OpenFile_XB_44_A6 cmd;
|
|
cmd.name.assign_raw("PSO/" + name);
|
|
cmd.filename.encode(filename);
|
|
cmd.type = 0;
|
|
cmd.file_size = file_size;
|
|
cmd.xb_filename.encode(xb_filename);
|
|
cmd.content_meta = 0x30000000 | quest_number;
|
|
w.put(cmd);
|
|
}
|
|
|
|
template <typename HeaderT>
|
|
void add_write_file_commands_t(
|
|
phosg::StringWriter& w,
|
|
const std::string& filename,
|
|
const std::string& data,
|
|
bool is_download,
|
|
bool bb_alignment) {
|
|
for (size_t z = 0; z < data.size(); z += 0x400) {
|
|
size_t chunk_size = std::min<size_t>(data.size() - z, 0x400);
|
|
add_command_header<HeaderT>(w, is_download ? 0xA7 : 0x13, z >> 10, sizeof(S_WriteFile_13_A7));
|
|
S_WriteFile_13_A7 cmd;
|
|
cmd.filename.encode(filename);
|
|
memcpy(cmd.data.data(), &data[z], chunk_size);
|
|
cmd.data_size = chunk_size;
|
|
w.put(cmd);
|
|
// On BB, the write file command size is a multiple of 4 but not a multiple of 8; in QST format the implicit extra
|
|
// 4 bytes are apparently stored in the file.
|
|
if (bb_alignment) {
|
|
w.put_u32(0);
|
|
}
|
|
}
|
|
}
|
|
|
|
std::string encode_qst_file(
|
|
const std::unordered_map<std::string, std::shared_ptr<const std::string>>& files,
|
|
const std::string& name,
|
|
uint32_t quest_number,
|
|
const std::string& xb_filename,
|
|
Version version,
|
|
bool is_dlq_encoded) {
|
|
phosg::StringWriter w;
|
|
|
|
// Some tools expect both open file commands at the beginning, hence this unfortunate abstraction-breaking.
|
|
switch (version) {
|
|
case Version::DC_NTE: // DC NTE doesn't support quests, but we support encoding QST files anyway
|
|
case Version::DC_11_2000:
|
|
case Version::DC_V1:
|
|
case Version::DC_V2:
|
|
for (const auto& it : files) {
|
|
add_open_file_command_t<PSOCommandHeaderDCV3, S_OpenFile_DC_44_A6>(
|
|
w, name, it.first, xb_filename, quest_number, it.second->size(), is_dlq_encoded);
|
|
}
|
|
for (const auto& it : files) {
|
|
add_write_file_commands_t<PSOCommandHeaderDCV3>(w, it.first, *it.second, is_dlq_encoded, false);
|
|
}
|
|
break;
|
|
case Version::PC_NTE:
|
|
case Version::PC_V2:
|
|
for (const auto& it : files) {
|
|
add_open_file_command_t<PSOCommandHeaderPC, S_OpenFile_PC_GC_44_A6>(
|
|
w, name, it.first, xb_filename, quest_number, it.second->size(), is_dlq_encoded);
|
|
}
|
|
for (const auto& it : files) {
|
|
add_write_file_commands_t<PSOCommandHeaderPC>(w, it.first, *it.second, is_dlq_encoded, false);
|
|
}
|
|
break;
|
|
case Version::GC_NTE:
|
|
case Version::GC_V3:
|
|
case Version::GC_EP3_NTE:
|
|
case Version::GC_EP3:
|
|
for (const auto& it : files) {
|
|
add_open_file_command_t<PSOCommandHeaderDCV3, S_OpenFile_PC_GC_44_A6>(
|
|
w, name, it.first, xb_filename, quest_number, it.second->size(), is_dlq_encoded);
|
|
}
|
|
for (const auto& it : files) {
|
|
add_write_file_commands_t<PSOCommandHeaderDCV3>(w, it.first, *it.second, is_dlq_encoded, false);
|
|
}
|
|
break;
|
|
case Version::XB_V3:
|
|
for (const auto& it : files) {
|
|
add_open_file_command_t<PSOCommandHeaderDCV3, S_OpenFile_XB_44_A6>(
|
|
w, name, it.first, xb_filename, quest_number, it.second->size(), is_dlq_encoded);
|
|
}
|
|
for (const auto& it : files) {
|
|
add_write_file_commands_t<PSOCommandHeaderDCV3>(w, it.first, *it.second, is_dlq_encoded, false);
|
|
}
|
|
break;
|
|
case Version::BB_V4:
|
|
for (const auto& it : files) {
|
|
add_open_file_command_t<PSOCommandHeaderBB, S_OpenFile_BB_44_A6>(
|
|
w, name, it.first, xb_filename, quest_number, it.second->size(), is_dlq_encoded);
|
|
}
|
|
for (const auto& it : files) {
|
|
add_write_file_commands_t<PSOCommandHeaderBB>(w, it.first, *it.second, is_dlq_encoded, true);
|
|
}
|
|
break;
|
|
default:
|
|
throw std::logic_error("invalid game version");
|
|
}
|
|
|
|
return std::move(w.str());
|
|
}
|