// // Created by Иван_Архипов on 07.11.2017. // #include "SubDirectory.h" #include "DatFile.h" #include "DatException.h" #include "Subfile.h" #include "BinaryData.h" #include "Subfiles/TextSubfile.h" #include "Subfiles/DdsSubfile.h" #include "Subfiles/FontSubfile.h" #include "Subfiles/JpgSubfile.h" #include "Subfiles/OggSubfile.h" #include "Subfiles/WavSubfile.h" #include "Subfiles/UnknownSubfile.h" namespace LOTRO_DAT { SubDirectory::SubDirectory() { offset_ = 0; } SubDirectory::~SubDirectory() { subfiles_.clear(); subdirs_.clear(); } SubDirectory::SubDirectory(long long offset, DatFile *dat, long long max_subdirs) : dat_(dat), offset_(offset), max_subdirs_(max_subdirs) { try { MakeSubDirectories(); } catch (std::exception &e) { fprintf(stderr, "Caught %s exception.", e.what()); fprintf(stderr, "Unable to initialize directory at offset %lld. Initializing it as empty directory...\n", offset); subdirs_.clear(); subfiles_.clear(); return; } try { MakeSubFiles(); } catch (std::exception &e) { fprintf(stderr, "Caught %s exception.", e.what()); fprintf(stderr, "Unable to initialize directory at offset %lld. Initializing it as empty directory...\n", offset); subdirs_.clear(); subfiles_.clear(); return; } } void SubDirectory::MakeSubDirectories() { BinaryData data(1024); dat_->ReadData(data, 63 * 8, offset_); if (data.ToNumber<4>(0) != 0 || data.ToNumber<4>(4) != 0) { std::string err = std::string("Bad SubDirectory::MakeSubDirectories - first 8 bytes are not equal to 0 at offset ") + std::to_string(offset_); throw DatException(err.c_str(), SUBDIR_EXCEPTION); } for (unsigned int i = 8; i < 63 * 8; i += 8) { if (data.ToNumber<4>(i) == 0 || data.ToNumber<4>(i + 4) == 0) break; try { SubDirectory *subdir = new SubDirectory(data.ToNumber<4>(i + 4), dat_); subdirs_.push_back(subdir); } catch (std::exception &e) { fprintf(stderr, "Caught %s exception.", e.what()); fprintf(stderr, "Making SubDirectory at offset %lld failed, continuing\n", data.ToNumber<4>(i + 4)); } } } void SubDirectory::MakeSubFiles() { BinaryData data(64 * 32); dat_->ReadData(data, 64 * 32, offset_ + 63 * 8); for (unsigned int i = 0; i < 61 * 32; i += 32) { if (data.ToNumber<4>(i + 8) < 0x32 || data.ToNumber<4>(i + 12) < 0x32) continue; BinaryData mfile_id(4); dat_->ReadData(mfile_id, 4, data.ToNumber<4>(i + 12) + 8); if (data.ToNumber<4>(i + 8) != mfile_id.ToNumber<4>(0)) { continue; } subfiles_.push_back( MakeSubfile( offset_ + 63 * 8 + i, data.ToNumber<4>(i), // fragments_count data.ToNumber<4>(i + 4), // unknown1 data.ToNumber<4>(i + 8), // file_id data.ToNumber<4>(i + 12), // file_offset data.ToNumber<4>(i + 16), // block_size data.ToNumber<4>(i + 20), // timestamp data.ToNumber<4>(i + 24), // version data.ToNumber<4>(i + 28) // block_size ) ); } } void SubDirectory::MakeDictionary(std::unordered_map &dict) { for (Subfile *i : subfiles_) { if (dict.count(i->file_id()) == 0 || dict[i->file_id()]->timestamp() < i->timestamp()) dict[i->file_id()] = i; } for (SubDirectory *i : subdirs_) i->MakeDictionary(dict); } void SubDirectory::UpdateDirectories(std::unordered_set &patched_files, std::unordered_map &dict) { for (unsigned i = 0; i < subfiles_.size(); i++) { long long file_id = subfiles_[i]->file_id(); if (patched_files.count(file_id) != 0) { BinaryData data(32); dat_->ReadData(data, 32, subfiles_[i]->dictionary_offset()); auto subfile = dict[file_id]; dat_->WriteData(BinaryData::FromNumber<4>(subfile->file_offset()), 4, subfiles_[i]->dictionary_offset() + 12); dat_->WriteData(BinaryData::FromNumber<4>(subfile->file_size()), 4, subfiles_[i]->dictionary_offset() + 16); dat_->WriteData(BinaryData::FromNumber<4>(subfile->block_size()), 4, subfiles_[i]->dictionary_offset() + 28); } } for (SubDirectory *i : subdirs_) i->UpdateDirectories(patched_files, dict); } Subfile *SubDirectory::MakeSubfile(long long dictionary_offset, long long fragments_count, long long unknown1, long long file_id, long long file_offset, long long file_size, long long timestamp, long long version, long long block_size) { FILE_TYPE type = GetSubfileType(file_id, file_offset); switch (type) { case TEXT: return dynamic_cast(new TextSubfile(dat_, dictionary_offset, fragments_count, unknown1, file_id, file_offset, file_size, timestamp, version, block_size)); case JPG: return dynamic_cast(new JpgSubfile(dat_, dictionary_offset, fragments_count, unknown1, file_id, file_offset, file_size, timestamp, version, block_size)); case DDS: return dynamic_cast(new DdsSubfile(dat_, dictionary_offset, fragments_count, unknown1, file_id, file_offset, file_size, timestamp, version, block_size)); case WAV: return dynamic_cast(new WavSubfile(dat_, dictionary_offset, fragments_count, unknown1, file_id, file_offset, file_size, timestamp, version, block_size)); case OGG: return dynamic_cast(new OggSubfile(dat_, dictionary_offset, fragments_count, unknown1, file_id, file_offset, file_size, timestamp, version, block_size)); case FONT: return dynamic_cast(new FontSubfile(dat_, dictionary_offset, fragments_count, unknown1, file_id, file_offset, file_size, timestamp, version, block_size)); case UNKNOWN: return dynamic_cast(new UnknownSubfile(dat_, dictionary_offset, fragments_count, unknown1, file_id, file_offset, file_size, timestamp, version, block_size)); } throw DatException("Bad SubDirectory::MakeSubfile() - unable to recognize file_type!", SUBFILE_EXCEPTION); } FILE_TYPE SubDirectory::GetSubfileType(long long file_id, long long file_offset) const { // Text check based on file_id if ((file_id >> 24ll) == 0x25ll) return TEXT; // Font check based on file_id if ((file_id >> 24ll) == 0x42ll) return FONT; BinaryData header(64); try { dat_->ReadData(header, 64, (unsigned) file_offset + 8); }catch (DatException &e) { if (e.type() == READ_EXCEPTION) { std::string err = "Bad Subfile::getExtension() - unable to read header of file with id = " + std::to_string(file_id) + " and offset = " + std::to_string(file_offset); throw DatException(err.c_str(), SUBFILE_EXCEPTION); } else throw e; } // jpeg / dds check if ((file_id >> 24ll) == 0x41ll) { long long soi = header.ToNumber<2>(24); long long marker = header.ToNumber<2>(26); //auto markerSize = header.ToNumber(28); //auto four = header.ToNumber(30); if ((soi == 0xD8FFll && marker == 0xE0FFll) || marker == 0xE1FFll) return JPG; return DDS; } // Ogg and Wav check if (header[8] == 0x4F && header[9] == 0x67 && header[10] == 0x67 && header[11] == 0x53) return OGG; if (header[8] == 0x52 && header[9] == 0x49 && header[10] == 0x46 && header[11] == 0x46) return WAV; return UNKNOWN; } };