#include "amuse/ContainerRegistry.hpp" #include "amuse/Common.hpp" #include #include #include #include #include #include #if _WIN32 #define WIN32_LEAN_AND_MEAN #include #include static void *memmem(const void *haystack, size_t hlen, const void *needle, size_t nlen) { int needle_first; const uint8_t *p = static_cast(haystack); size_t plen = hlen; if (!nlen) return NULL; needle_first = *(unsigned char *)needle; while (plen >= nlen && (p = static_cast(memchr(p, needle_first, plen - nlen + 1)))) { if (!memcmp(p, needle, nlen)) return (void *)p; p++; plen = hlen - (p - static_cast(haystack)); } return NULL; } amuse::SystemString StrToSys(const std::string& str) { std::wstring ret; int len = MultiByteToWideChar(CP_UTF8, 0, str.c_str(), str.size(), nullptr, 0); ret.assign(len, L'\0'); MultiByteToWideChar(CP_UTF8, 0, str.c_str(), str.size(), &ret[0], len); return ret; } #else amuse::SystemString StrToSys(const std::string& str) { return str; } #endif namespace amuse { const SystemChar* ContainerRegistry::TypeToName(Type tp) { switch (tp) { case Type::Invalid: default: return nullptr; case Type::Raw4: return _S("4 RAW Chunks"); case Type::MetroidPrime: return _S("Metroid Prime (GCN)"); case Type::MetroidPrime2: return _S("Metroid Prime 2 (GCN)"); case Type::RogueSquadronPC: return _S("Star Wars - Rogue Squadron (PC)"); case Type::RogueSquadronN64: return _S("Star Wars - Rogue Squadron (N64)"); case Type::BattleForNabooPC: return _S("Star Wars Episode I - Battle for Naboo (PC)"); case Type::BattleForNabooN64: return _S("Star Wars Episode I - Battle for Naboo (N64)"); case Type::RogueSquadron2: return _S("Star Wars - Rogue Squadron 2 (GCN)"); case Type::RogueSquadron3: return _S("Star Wars - Rogue Squadron 3 (GCN)"); } } static size_t FileLength(FILE* fp) { FSeek(fp, 0, SEEK_END); int64_t endPos = FTell(fp); fseek(fp, 0, SEEK_SET); return size_t(endPos); } static SystemString ReadString(FILE* fp) { char byte; SystemString ret; while (fread(&byte, 1, 1, fp) == 1 && byte != 0) ret.push_back(byte); return ret; } static bool IsChunkExtension(const SystemChar* path, const SystemChar*& dotOut) { const SystemChar* ext = StrRChr(path, _S('.')); if (ext) { if (!CompareCaseInsensitive(ext, _S(".poo")) || !CompareCaseInsensitive(ext, _S(".pool")) || !CompareCaseInsensitive(ext, _S(".pro")) || !CompareCaseInsensitive(ext, _S(".proj")) || !CompareCaseInsensitive(ext, _S(".sdi")) || !CompareCaseInsensitive(ext, _S(".sdir")) || !CompareCaseInsensitive(ext, _S(".sam")) || !CompareCaseInsensitive(ext, _S(".samp"))) { dotOut = ext; return true; } } return false; } static bool IsSongExtension(const SystemChar* path, const SystemChar*& dotOut) { const SystemChar* ext = StrRChr(path, _S('.')); if (ext) { if (!CompareCaseInsensitive(ext, _S(".son")) || !CompareCaseInsensitive(ext, _S(".sng")) || !CompareCaseInsensitive(ext, _S(".song"))) { dotOut = ext; return true; } } return false; } static bool ValidateMP1(FILE* fp) { if (FileLength(fp) > 40 * 1024 * 1024) return false; uint32_t magic; fread(&magic, 1, 4, fp); magic = SBig(magic); if (magic == 0x00030005) { FSeek(fp, 8, SEEK_SET); uint32_t nameCount; fread(&nameCount, 1, 4, fp); nameCount = SBig(nameCount); for (uint32_t i=0 ; i> LoadMP1(FILE* fp) { std::vector> ret; FileLength(fp); uint32_t magic; fread(&magic, 1, 4, fp); magic = SBig(magic); if (magic == 0x00030005) { FSeek(fp, 8, SEEK_SET); uint32_t nameCount; fread(&nameCount, 1, 4, fp); nameCount = SBig(nameCount); for (uint32_t i=0 ; i pool(new uint8_t[poolLen]); fread(pool.get(), 1, poolLen, fp); uint32_t projLen; fread(&projLen, 1, 4, fp); projLen = SBig(projLen); std::unique_ptr proj(new uint8_t[projLen]); fread(proj.get(), 1, projLen, fp); uint32_t sampLen; fread(&sampLen, 1, 4, fp); sampLen = SBig(sampLen); std::unique_ptr samp(new uint8_t[sampLen]); fread(samp.get(), 1, sampLen, fp); uint32_t sdirLen; fread(&sdirLen, 1, 4, fp); sdirLen = SBig(sdirLen); std::unique_ptr sdir(new uint8_t[sdirLen]); fread(sdir.get(), 1, sdirLen, fp); ret.emplace_back(std::move(name), IntrusiveAudioGroupData{proj.release(), projLen, pool.release(), poolLen, sdir.release(), sdirLen, samp.release(), sampLen, GCNDataTag{}}); } } FSeek(fp, origPos, SEEK_SET); } } } return ret; } static bool ValidateMP1Songs(FILE* fp) { if (FileLength(fp) > 40 * 1024 * 1024) return false; uint32_t magic; fread(&magic, 1, 4, fp); magic = SBig(magic); if (magic == 0x00030005) { FSeek(fp, 8, SEEK_SET); uint32_t nameCount; fread(&nameCount, 1, 4, fp); nameCount = SBig(nameCount); for (uint32_t i=0 ; i> LoadMP1Songs(FILE* fp) { std::vector> ret; FileLength(fp); uint32_t magic; fread(&magic, 1, 4, fp); magic = SBig(magic); if (magic == 0x00030005) { FSeek(fp, 8, SEEK_SET); uint32_t nameCount; fread(&nameCount, 1, 4, fp); nameCount = SBig(nameCount); std::unordered_map names; names.reserve(nameCount); for (uint32_t i=0 ; i song(new uint8_t[sonLength]); fread(song.get(), 1, sonLength, fp); auto search = names.find(id); if (search != names.end()) ret.emplace_back(std::move(search->second), ContainerRegistry::SongData(std::move(song), sonLength, groupId, midiSetup)); else { SystemChar name[128]; SNPrintf(name, 128, _S("%08X"), id); ret.emplace_back(name, ContainerRegistry::SongData(std::move(song), sonLength, groupId, midiSetup)); } FSeek(fp, origPos, SEEK_SET); } } } return ret; } static bool ValidateMP2(FILE* fp) { if (FileLength(fp) > 40 * 1024 * 1024) return false; uint32_t magic; fread(&magic, 1, 4, fp); magic = SBig(magic); if (magic == 0x00030005) { FSeek(fp, 8, SEEK_SET); uint32_t nameCount; fread(&nameCount, 1, 4, fp); nameCount = SBig(nameCount); for (uint32_t i=0 ; i(testBuf)) == 0x1) return true; else return false; } FSeek(fp, origPos, SEEK_SET); } } } return false; } static std::vector> LoadMP2(FILE* fp) { std::vector> ret; FileLength(fp); uint32_t magic; fread(&magic, 1, 4, fp); magic = SBig(magic); if (magic == 0x00030005) { FSeek(fp, 8, SEEK_SET); uint32_t nameCount; fread(&nameCount, 1, 4, fp); nameCount = SBig(nameCount); for (uint32_t i=0 ; i(testBuf)) == 0x1) { FSeek(fp, offset + 4, SEEK_SET); SystemString name = ReadString(fp); FSeek(fp, 2, SEEK_CUR); uint32_t poolSz; fread(&poolSz, 1, 4, fp); poolSz = SBig(poolSz); uint32_t projSz; fread(&projSz, 1, 4, fp); projSz = SBig(projSz); uint32_t sdirSz; fread(&sdirSz, 1, 4, fp); sdirSz = SBig(sdirSz); uint32_t sampSz; fread(&sampSz, 1, 4, fp); sampSz = SBig(sampSz); if (projSz && poolSz && sdirSz && sampSz) { std::unique_ptr pool(new uint8_t[poolSz]); fread(pool.get(), 1, poolSz, fp); std::unique_ptr proj(new uint8_t[projSz]); fread(proj.get(), 1, projSz, fp); std::unique_ptr sdir(new uint8_t[sdirSz]); fread(sdir.get(), 1, sdirSz, fp); std::unique_ptr samp(new uint8_t[sampSz]); fread(samp.get(), 1, sampSz, fp); ret.emplace_back(std::move(name), IntrusiveAudioGroupData{proj.release(), projSz, pool.release(), poolSz, sdir.release(), sdirSz, samp.release(), sampSz, GCNDataTag{}}); } } } FSeek(fp, origPos, SEEK_SET); } } } return ret; } struct RS1FSTEntry { uint32_t offset; uint32_t decompSz; uint32_t compSz; uint16_t type; uint16_t childCount; char name[16]; void swapBig() { offset = SBig(offset); decompSz = SBig(decompSz); compSz = SBig(compSz); type = SBig(type); childCount = SBig(childCount); } }; static void SwapN64Rom16(void* data, size_t size) { uint16_t* words = reinterpret_cast(data); for (size_t i=0 ; i(data); for (size_t i=0 ; i 100 * 1024 * 1024) return false; uint32_t fstOff; uint32_t fstSz; if (fread(&fstOff, 1, 4, fp) == 4 && fread(&fstSz, 1, 4, fp) == 4) { if (fstOff + fstSz <= endPos) { FSeek(fp, fstOff, SEEK_SET); uint32_t elemCount = fstSz / 32; std::unique_ptr entries(new RS1FSTEntry[elemCount]); fread(entries.get(), fstSz, 1, fp); uint8_t foundComps = 0; for (uint32_t i=0 ; i> LoadRS1PC(FILE* fp) { std::vector> ret; size_t endPos = FileLength(fp); uint32_t fstOff; uint32_t fstSz; if (fread(&fstOff, 1, 4, fp) == 4 && fread(&fstSz, 1, 4, fp) == 4) { if (fstOff + fstSz <= endPos) { FSeek(fp, fstOff, SEEK_SET); uint32_t elemCount = fstSz / 32; std::unique_ptr entries(new RS1FSTEntry[elemCount]); fread(entries.get(), fstSz, 1, fp); std::unique_ptr proj; size_t projSz = 0; std::unique_ptr pool; size_t poolSz = 0; std::unique_ptr sdir; size_t sdirSz = 0; std::unique_ptr samp; size_t sampSz = 0; for (uint32_t i=0 ; i 32 * 1024 * 1024) return false; /* N64 ROM definitely won't exceed 32MB */ std::unique_ptr data(new uint8_t[endPos]); fread(data.get(), 1, endPos, fp); if ((data[0] & 0x80) != 0x80 && (data[3] & 0x80) == 0x80) SwapN64Rom32(data.get(), endPos); else if ((data[0] & 0x80) != 0x80 && (data[1] & 0x80) == 0x80) SwapN64Rom16(data.get(), endPos); #if 0 const uint32_t* gameId = reinterpret_cast(&data[59]); if (*gameId != 0x4553524e && *gameId != 0x4a53524e && *gameId != 0x5053524e) return false; /* GameId not 'NRSE', 'NRSJ', or 'NRSP' */ #endif const uint8_t* dataSeg = reinterpret_cast(memmem(data.get(), endPos, "dbg_data\0\0\0\0\0\0\0\0", 16)); if (dataSeg) { dataSeg += 28; size_t fstEnd = SBig(*reinterpret_cast(dataSeg)); dataSeg += 4; size_t fstOff = SBig(*reinterpret_cast(dataSeg)); if (endPos <= size_t(dataSeg - data.get()) + fstOff || endPos <= size_t(dataSeg - data.get()) + fstEnd) return false; const RS1FSTEntry* entry = reinterpret_cast(dataSeg + fstOff); const RS1FSTEntry* lastEnt = reinterpret_cast(dataSeg + fstEnd); uint8_t foundComps = 0; for (; entry != lastEnt ; ++entry) { if (!strncmp("proj_SND", entry->name, 16)) foundComps |= 1; else if (!strncmp("pool_SND", entry->name, 16)) foundComps |= 2; else if (!strncmp("sdir_SND", entry->name, 16)) foundComps |= 4; else if (!strncmp("samp_SND", entry->name, 16)) foundComps |= 8; } if (foundComps == 0xf) return true; } return false; } static std::vector> LoadRS1N64(FILE* fp) { std::vector> ret; size_t endPos = FileLength(fp); std::unique_ptr data(new uint8_t[endPos]); fread(data.get(), 1, endPos, fp); if ((data[0] & 0x80) != 0x80 && (data[3] & 0x80) == 0x80) SwapN64Rom32(data.get(), endPos); else if ((data[0] & 0x80) != 0x80 && (data[1] & 0x80) == 0x80) SwapN64Rom16(data.get(), endPos); const uint8_t* dataSeg = reinterpret_cast(memmem(data.get(), endPos, "dbg_data\0\0\0\0\0\0\0\0", 16)); if (dataSeg) { dataSeg += 28; size_t fstEnd = SBig(*reinterpret_cast(dataSeg)); dataSeg += 4; size_t fstOff = SBig(*reinterpret_cast(dataSeg)); if (endPos <= size_t(dataSeg - data.get()) + fstOff || endPos <= size_t(dataSeg - data.get()) + fstEnd) return ret; const RS1FSTEntry* entry = reinterpret_cast(dataSeg + fstOff); const RS1FSTEntry* lastEnt = reinterpret_cast(dataSeg + fstEnd); std::unique_ptr proj; size_t projSz = 0; std::unique_ptr pool; size_t poolSz = 0; std::unique_ptr sdir; size_t sdirSz = 0; std::unique_ptr samp; size_t sampSz = 0; for (; entry != lastEnt ; ++entry) { RS1FSTEntry ent = *entry; ent.swapBig(); if (!strncmp("proj_SND", ent.name, 16)) { if (ent.compSz == 0xffffffff) { proj.reset(new uint8_t[ent.decompSz]); memmove(proj.get(), dataSeg + ent.offset, ent.decompSz); } else { proj.reset(new uint8_t[ent.decompSz]); uLongf outSz = ent.decompSz; uncompress(proj.get(), &outSz, dataSeg + ent.offset, ent.compSz); } projSz = ent.decompSz; } else if (!strncmp("pool_SND", ent.name, 16)) { if (ent.compSz == 0xffffffff) { pool.reset(new uint8_t[ent.decompSz]); memmove(pool.get(), dataSeg + ent.offset, ent.decompSz); } else { pool.reset(new uint8_t[ent.decompSz]); uLongf outSz = ent.decompSz; uncompress(pool.get(), &outSz, dataSeg + ent.offset, ent.compSz); } poolSz = ent.decompSz; } else if (!strncmp("sdir_SND", ent.name, 16)) { if (ent.compSz == 0xffffffff) { sdir.reset(new uint8_t[ent.decompSz]); memmove(sdir.get(), dataSeg + ent.offset, ent.decompSz); } else { sdir.reset(new uint8_t[ent.decompSz]); uLongf outSz = ent.decompSz; uncompress(sdir.get(), &outSz, dataSeg + ent.offset, ent.compSz); } sdirSz = ent.decompSz; } else if (!strncmp("samp_SND", ent.name, 16)) { if (ent.compSz == 0xffffffff) { samp.reset(new uint8_t[ent.decompSz]); memmove(samp.get(), dataSeg + ent.offset, ent.decompSz); } else { samp.reset(new uint8_t[ent.decompSz]); uLongf outSz = ent.decompSz; uncompress(samp.get(), &outSz, dataSeg + ent.offset, ent.compSz); } sampSz = ent.decompSz; } } ret.emplace_back(_S("Group"), IntrusiveAudioGroupData{proj.release(), projSz, pool.release(), poolSz, sdir.release(), sdirSz, samp.release(), sampSz, false, N64DataTag{}}); } return ret; } static bool ValidateBFNPC(FILE* fp) { size_t endPos = FileLength(fp); if (endPos > 100 * 1024 * 1024) return false; uint32_t fstOff; uint32_t fstSz; if (fread(&fstOff, 1, 4, fp) == 4 && fread(&fstSz, 1, 4, fp) == 4) { if (fstOff + fstSz <= endPos) { FSeek(fp, fstOff, SEEK_SET); uint32_t elemCount = fstSz / 32; std::unique_ptr entries(new RS1FSTEntry[elemCount]); fread(entries.get(), fstSz, 1, fp); uint8_t foundComps = 0; for (uint32_t i=0 ; i> LoadBFNPC(FILE* fp) { std::vector> ret; size_t endPos = FileLength(fp); uint32_t fstOff; uint32_t fstSz; if (fread(&fstOff, 1, 4, fp) == 4 && fread(&fstSz, 1, 4, fp) == 4) { if (fstOff + fstSz <= endPos) { FSeek(fp, fstOff, SEEK_SET); uint32_t elemCount = fstSz / 32; std::unique_ptr entries(new RS1FSTEntry[elemCount]); fread(entries.get(), fstSz, 1, fp); std::unique_ptr proj; size_t projSz = 0; std::unique_ptr pool; size_t poolSz = 0; std::unique_ptr sdir; size_t sdirSz = 0; std::unique_ptr samp; size_t sampSz = 0; for (uint32_t i=0 ; i 32 * 1024 * 1024) return false; /* N64 ROM definitely won't exceed 32MB */ std::unique_ptr data(new uint8_t[endPos]); fread(data.get(), 1, endPos, fp); if ((data[0] & 0x80) != 0x80 && (data[3] & 0x80) == 0x80) SwapN64Rom32(data.get(), endPos); else if ((data[0] & 0x80) != 0x80 && (data[1] & 0x80) == 0x80) SwapN64Rom16(data.get(), endPos); #if 0 const uint32_t* gameId = reinterpret_cast(&data[59]); if (*gameId != 0x4553524e && *gameId != 0x4a53524e && *gameId != 0x5053524e) return false; /* GameId not 'NRSE', 'NRSJ', or 'NRSP' */ #endif const uint8_t* dataSeg = reinterpret_cast(memmem(data.get(), endPos, "dbg_data\0\0\0\0\0\0\0\0", 16)); if (dataSeg) { dataSeg += 28; size_t fstEnd = SBig(*reinterpret_cast(dataSeg)); dataSeg += 4; size_t fstOff = SBig(*reinterpret_cast(dataSeg)); if (endPos <= size_t(dataSeg - data.get()) + fstOff || endPos <= size_t(dataSeg - data.get()) + fstEnd) return false; const RS1FSTEntry* entry = reinterpret_cast(dataSeg + fstOff); const RS1FSTEntry* lastEnt = reinterpret_cast(dataSeg + fstEnd); uint8_t foundComps = 0; for (; entry != lastEnt ; ++entry) { if (!strncmp("proj", entry->name, 16)) foundComps |= 1; else if (!strncmp("pool", entry->name, 16)) foundComps |= 2; else if (!strncmp("sdir", entry->name, 16)) foundComps |= 4; else if (!strncmp("samp", entry->name, 16)) foundComps |= 8; } if (foundComps == 0xf) return true; } return false; } static std::vector> LoadBFNN64(FILE* fp) { std::vector> ret; size_t endPos = FileLength(fp); std::unique_ptr data(new uint8_t[endPos]); fread(data.get(), 1, endPos, fp); if ((data[0] & 0x80) != 0x80 && (data[3] & 0x80) == 0x80) SwapN64Rom32(data.get(), endPos); else if ((data[0] & 0x80) != 0x80 && (data[1] & 0x80) == 0x80) SwapN64Rom16(data.get(), endPos); const uint8_t* dataSeg = reinterpret_cast(memmem(data.get(), endPos, "dbg_data\0\0\0\0\0\0\0\0", 16)); if (dataSeg) { dataSeg += 28; size_t fstEnd = SBig(*reinterpret_cast(dataSeg)); dataSeg += 4; size_t fstOff = SBig(*reinterpret_cast(dataSeg)); if (endPos <= size_t(dataSeg - data.get()) + fstOff || endPos <= size_t(dataSeg - data.get()) + fstEnd) return ret; const RS1FSTEntry* entry = reinterpret_cast(dataSeg + fstOff); const RS1FSTEntry* lastEnt = reinterpret_cast(dataSeg + fstEnd); std::unique_ptr proj; size_t projSz = 0; std::unique_ptr pool; size_t poolSz = 0; std::unique_ptr sdir; size_t sdirSz = 0; std::unique_ptr samp; size_t sampSz = 0; for (; entry != lastEnt ; ++entry) { RS1FSTEntry ent = *entry; ent.swapBig(); if (!strncmp("proj", ent.name, 16)) { if (ent.compSz == 0xffffffff) { proj.reset(new uint8_t[ent.decompSz]); memmove(proj.get(), dataSeg + ent.offset, ent.decompSz); } else { proj.reset(new uint8_t[ent.decompSz]); uLongf outSz = ent.decompSz; uncompress(proj.get(), &outSz, dataSeg + ent.offset, ent.compSz); } projSz = ent.decompSz; } else if (!strncmp("pool", ent.name, 16)) { if (ent.compSz == 0xffffffff) { pool.reset(new uint8_t[ent.decompSz]); memmove(pool.get(), dataSeg + ent.offset, ent.decompSz); } else { pool.reset(new uint8_t[ent.decompSz]); uLongf outSz = ent.decompSz; uncompress(pool.get(), &outSz, dataSeg + ent.offset, ent.compSz); } poolSz = ent.decompSz; } else if (!strncmp("sdir", ent.name, 16)) { if (ent.compSz == 0xffffffff) { sdir.reset(new uint8_t[ent.decompSz]); memmove(sdir.get(), dataSeg + ent.offset, ent.decompSz); } else { sdir.reset(new uint8_t[ent.decompSz]); uLongf outSz = ent.decompSz; uncompress(sdir.get(), &outSz, dataSeg + ent.offset, ent.compSz); } sdirSz = ent.decompSz; } else if (!strncmp("samp", ent.name, 16)) { if (ent.compSz == 0xffffffff) { samp.reset(new uint8_t[ent.decompSz]); memmove(samp.get(), dataSeg + ent.offset, ent.decompSz); } else { samp.reset(new uint8_t[ent.decompSz]); uLongf outSz = ent.decompSz; uncompress(samp.get(), &outSz, dataSeg + ent.offset, ent.compSz); } sampSz = ent.decompSz; } } ret.emplace_back(_S("Group"), IntrusiveAudioGroupData{proj.release(), projSz, pool.release(), poolSz, sdir.release(), sdirSz, samp.release(), sampSz, true, N64DataTag{}}); } return ret; } struct RS2FSTEntry { uint64_t offset; uint64_t decompSz; uint64_t compSz; uint64_t type; char name[32]; void swapBig() { offset = SBig(offset); decompSz = SBig(decompSz); compSz = SBig(compSz); type = SBig(type); } }; struct RS23GroupHead { uint32_t projOff; uint32_t projLen; uint32_t poolOff; uint32_t poolLen; uint32_t sdirOff; uint32_t sdirLen; uint32_t sampOff; uint32_t sampLen; uint32_t unkOff; uint32_t unkLen; uint32_t sonCount; uint32_t sonIdxBeginOff; uint32_t sonIdxEndOff; void swapBig() { projOff = SBig(projOff); projLen = SBig(projLen); poolOff = SBig(poolOff); poolLen = SBig(poolLen); sdirOff = SBig(sdirOff); sdirLen = SBig(sdirLen); sampOff = SBig(sampOff); sampLen = SBig(sampLen); unkOff = SBig(unkOff); unkLen = SBig(unkLen); sonCount = SBig(sonCount); sonIdxBeginOff = SBig(sonIdxBeginOff); sonIdxEndOff = SBig(sonIdxEndOff); } }; struct RS23SONHead { uint32_t offset; uint32_t length; uint16_t groupId; uint16_t setupId; void swapBig() { offset = SBig(offset); length = SBig(length); groupId = SBig(groupId); setupId = SBig(setupId); } }; static bool ValidateRS2(FILE* fp) { size_t endPos = FileLength(fp); if (endPos > 600 * 1024 * 1024) return false; uint64_t fstOff; fread(&fstOff, 1, 8, fp); fstOff = SBig(fstOff); uint64_t fstSz; fread(&fstSz, 1, 8, fp); fstSz = SBig(fstSz); if (fstOff + fstSz > endPos) return false; FSeek(fp, int64_t(fstOff), SEEK_SET); for (size_t i=0 ; i> LoadRS2(FILE* fp) { std::vector> ret; size_t endPos = FileLength(fp); uint64_t fstOff; fread(&fstOff, 1, 8, fp); fstOff = SBig(fstOff); uint64_t fstSz; fread(&fstSz, 1, 8, fp); fstSz = SBig(fstSz); if (fstOff + fstSz > endPos) return ret; FSeek(fp, int64_t(fstOff), SEEK_SET); for (size_t i=0 ; i audData(new uint8_t[entry.decompSz]); fread(audData.get(), 1, entry.decompSz, fp); uint32_t indexOff = SBig(*reinterpret_cast(audData.get() + 4)); uint32_t groupCount = SBig(*reinterpret_cast(audData.get() + indexOff)); const uint32_t* groupOffs = reinterpret_cast(audData.get() + indexOff + 4); for (uint32_t j=0 ; j(groupData); head.swapBig(); std::unique_ptr pool(new uint8_t[head.poolLen]); memmove(pool.get(), audData.get() + head.poolOff, head.poolLen); std::unique_ptr proj(new uint8_t[head.projLen]); memmove(proj.get(), audData.get() + head.projOff, head.projLen); std::unique_ptr sdir(new uint8_t[head.sdirLen]); memmove(sdir.get(), audData.get() + head.sdirOff, head.sdirLen); std::unique_ptr samp(new uint8_t[head.sampLen]); memmove(samp.get(), audData.get() + head.sampOff, head.sampLen); if (head.projLen && head.poolLen && head.sdirLen && head.sampLen) { SystemChar name[128]; SNPrintf(name, 128, _S("GroupFile%02u"), j); ret.emplace_back(name, IntrusiveAudioGroupData{proj.release(), head.projLen, pool.release(), head.poolLen, sdir.release(), head.sdirLen, samp.release(), head.sampLen, GCNDataTag{}}); } } break; } } return ret; } static std::vector> LoadRS2Songs(FILE* fp) { std::vector> ret; size_t endPos = FileLength(fp); uint64_t fstOff; fread(&fstOff, 1, 8, fp); fstOff = SBig(fstOff); uint64_t fstSz; fread(&fstSz, 1, 8, fp); fstSz = SBig(fstSz); if (fstOff + fstSz > endPos) return ret; FSeek(fp, int64_t(fstOff), SEEK_SET); for (size_t i=0 ; i audData(new uint8_t[entry.decompSz]); fread(audData.get(), 1, entry.decompSz, fp); uint32_t indexOff = SBig(*reinterpret_cast(audData.get() + 4)); uint32_t groupCount = SBig(*reinterpret_cast(audData.get() + indexOff)); const uint32_t* groupOffs = reinterpret_cast(audData.get() + indexOff + 4); for (uint32_t j=0 ; j(groupData); head.swapBig(); if (!head.sonCount) continue; const RS23SONHead* sonData = reinterpret_cast(audData.get() + head.sonIdxBeginOff); for (uint32_t s=0 ; s song(new uint8_t[sonHead.length]); memmove(song.get(), audData.get() + sonHead.offset, sonHead.length); ret.emplace_back(name, ContainerRegistry::SongData(std::move(song), sonHead.length, sonHead.groupId, sonHead.setupId)); } } break; } } return ret; } struct RS3FSTEntry { uint64_t offset; uint64_t decompSz; uint64_t compSz; uint64_t type; char name[128]; void swapBig() { offset = SBig(offset); decompSz = SBig(decompSz); compSz = SBig(compSz); type = SBig(type); } }; static bool ValidateRS3(FILE* fp) { size_t endPos = FileLength(fp); if (endPos > 600 * 1024 * 1024) return false; uint64_t fstOff; fread(&fstOff, 1, 8, fp); fstOff = SBig(fstOff); uint64_t fstSz; fread(&fstSz, 1, 8, fp); fstSz = SBig(fstSz); if (fstOff + fstSz > endPos) return false; FSeek(fp, int64_t(fstOff), SEEK_SET); for (size_t i=0 ; i> LoadRS3(FILE* fp) { std::vector> ret; size_t endPos = FileLength(fp); uint64_t fstOff; fread(&fstOff, 1, 8, fp); fstOff = SBig(fstOff); uint64_t fstSz; fread(&fstSz, 1, 8, fp); fstSz = SBig(fstSz); if (fstOff + fstSz > endPos) return ret; FSeek(fp, int64_t(fstOff), SEEK_SET); for (size_t i=0 ; i audData(new uint8_t[entry.decompSz]); fread(audData.get(), 1, entry.decompSz, fp); uint32_t indexOff = SBig(*reinterpret_cast(audData.get() + 4)); uint32_t groupCount = SBig(*reinterpret_cast(audData.get() + indexOff)); const uint32_t* groupOffs = reinterpret_cast(audData.get() + indexOff + 4); for (uint32_t j=0 ; j(groupData); head.swapBig(); std::unique_ptr pool(new uint8_t[head.poolLen]); memmove(pool.get(), audData.get() + head.poolOff, head.poolLen); std::unique_ptr proj(new uint8_t[head.projLen]); memmove(proj.get(), audData.get() + head.projOff, head.projLen); std::unique_ptr sdir(new uint8_t[head.sdirLen]); memmove(sdir.get(), audData.get() + head.sdirOff, head.sdirLen); std::unique_ptr samp(new uint8_t[head.sampLen]); memmove(samp.get(), audData.get() + head.sampOff, head.sampLen); if (head.projLen && head.poolLen && head.sdirLen && head.sampLen) { SystemChar name[128]; SNPrintf(name, 128, _S("GroupFile%02u"), j); ret.emplace_back(name, IntrusiveAudioGroupData{proj.release(), head.projLen, pool.release(), head.poolLen, sdir.release(), head.sdirLen, samp.release(), head.sampLen, GCNDataTag{}}); } } break; } } return ret; } ContainerRegistry::Type ContainerRegistry::DetectContainerType(const SystemChar* path) { FILE* fp; /* See if provided file is one of four raw chunks */ const SystemChar* dot = nullptr; if (IsChunkExtension(path, dot)) { SystemChar newpath[1024]; /* Project */ SNPrintf(newpath, 1024, _S("%.*s.pro"), int(dot - path), path); fp = FOpen(newpath, _S("rb")); if (!fp) { SNPrintf(newpath, 1024, _S("%.*s.proj"), int(dot - path), path); fp = FOpen(newpath, _S("rb")); if (!fp) return Type::Invalid; } fclose(fp); /* Pool */ SNPrintf(newpath, 1024, _S("%.*s.poo"), int(dot - path), path); fp = FOpen(newpath, _S("rb")); if (!fp) { SNPrintf(newpath, 1024, _S("%.*s.pool"), int(dot - path), path); fp = FOpen(newpath, _S("rb")); if (!fp) return Type::Invalid; } fclose(fp); /* Sample Directory */ SNPrintf(newpath, 1024, _S("%.*s.sdi"), int(dot - path), path); fp = FOpen(newpath, _S("rb")); if (!fp) { SNPrintf(newpath, 1024, _S("%.*s.sdir"), int(dot - path), path); fp = FOpen(newpath, _S("rb")); if (!fp) return Type::Invalid; } fclose(fp); /* Sample */ SNPrintf(newpath, 1024, _S("%.*s.sam"), int(dot - path), path); fp = FOpen(newpath, _S("rb")); if (!fp) { SNPrintf(newpath, 1024, _S("%.*s.samp"), int(dot - path), path); fp = FOpen(newpath, _S("rb")); if (!fp) return Type::Invalid; } fclose(fp); return Type::Raw4; } /* Now attempt single-file case */ fp = FOpen(path, _S("rb")); if (fp) { if (ValidateMP1(fp)) { fclose(fp); return Type::MetroidPrime; } if (ValidateMP2(fp)) { fclose(fp); return Type::MetroidPrime2; } if (ValidateRS1PC(fp)) { fclose(fp); return Type::RogueSquadronPC; } if (ValidateRS1N64(fp)) { fclose(fp); return Type::RogueSquadronN64; } if (ValidateBFNPC(fp)) { fclose(fp); return Type::BattleForNabooPC; } if (ValidateBFNN64(fp)) { fclose(fp); return Type::BattleForNabooN64; } if (ValidateRS2(fp)) { fclose(fp); return Type::RogueSquadron2; } if (ValidateRS3(fp)) { fclose(fp); return Type::RogueSquadron3; } fclose(fp); } return Type::Invalid; } std::vector> ContainerRegistry::LoadContainer(const SystemChar* path) { Type typeOut; return LoadContainer(path, typeOut); }; std::vector> ContainerRegistry::LoadContainer(const SystemChar* path, Type& typeOut) { FILE* fp; typeOut = Type::Invalid; /* See if provided file is one of four raw chunks */ const SystemChar* dot = nullptr; if (IsChunkExtension(path, dot)) { std::vector> ret; /* Project */ SystemChar projPath[1024]; SNPrintf(projPath, 1024, _S("%.*s.pro"), int(dot - path), path); fp = FOpen(projPath, _S("rb")); if (!fp) { SNPrintf(projPath, 1024, _S("%.*s.proj"), int(dot - path), path); fp = FOpen(projPath, _S("rb")); if (!fp) return ret; } fclose(fp); /* Pool */ SystemChar poolPath[1024]; SNPrintf(poolPath, 1024, _S("%.*s.poo"), int(dot - path), path); fp = FOpen(poolPath, _S("rb")); if (!fp) { SNPrintf(poolPath, 1024, _S("%.*s.pool"), int(dot - path), path); fp = FOpen(poolPath, _S("rb")); if (!fp) return ret; } fclose(fp); /* Sample Directory */ SystemChar sdirPath[1024]; SNPrintf(sdirPath, 1024, _S("%.*s.sdi"), int(dot - path), path); fp = FOpen(sdirPath, _S("rb")); if (!fp) { SNPrintf(sdirPath, 1024, _S("%.*s.sdir"), int(dot - path), path); fp = FOpen(sdirPath, _S("rb")); if (!fp) return ret; } fclose(fp); /* Sample */ SystemChar sampPath[1024]; SNPrintf(sampPath, 1024, _S("%.*s.sam"), int(dot - path), path); fp = FOpen(sampPath, _S("rb")); if (!fp) { SNPrintf(sampPath, 1024, _S("%.*s.samp"), int(dot - path), path); fp = FOpen(sampPath, _S("rb")); if (!fp) return ret; } fclose(fp); fp = FOpen(projPath, _S("rb")); size_t projLen = FileLength(fp); if (!projLen) return ret; std::unique_ptr proj(new uint8_t[projLen]); fread(proj.get(), 1, projLen, fp); fp = FOpen(poolPath, _S("rb")); size_t poolLen = FileLength(fp); if (!poolLen) return ret; std::unique_ptr pool(new uint8_t[poolLen]); fread(pool.get(), 1, poolLen, fp); fp = FOpen(sdirPath, _S("rb")); size_t sdirLen = FileLength(fp); if (!sdirLen) return ret; std::unique_ptr sdir(new uint8_t[sdirLen]); fread(sdir.get(), 1, sdirLen, fp); fp = FOpen(sampPath, _S("rb")); size_t sampLen = FileLength(fp); if (!sampLen) return ret; std::unique_ptr samp(new uint8_t[sampLen]); fread(samp.get(), 1, sampLen, fp); fclose(fp); /* SDIR-based format detection */ if (*reinterpret_cast(sdir.get() + 8) == 0x0) ret.emplace_back(_S("Group"), IntrusiveAudioGroupData{proj.release(), projLen, pool.release(), poolLen, sdir.release(), sdirLen, samp.release(), sampLen, GCNDataTag{}}); else if (sdir[9] == 0x0) ret.emplace_back(_S("Group"), IntrusiveAudioGroupData{proj.release(), projLen, pool.release(), poolLen, sdir.release(), sdirLen, samp.release(), sampLen, false, N64DataTag{}}); else ret.emplace_back(_S("Group"), IntrusiveAudioGroupData{proj.release(), projLen, pool.release(), poolLen, sdir.release(), sdirLen, samp.release(), sampLen, false, PCDataTag{}}); typeOut = Type::Raw4; return ret; } /* Now attempt single-file case */ fp = FOpen(path, _S("rb")); if (fp) { if (ValidateMP1(fp)) { auto ret = LoadMP1(fp); fclose(fp); typeOut = Type::MetroidPrime; return ret; } if (ValidateMP2(fp)) { auto ret = LoadMP2(fp); fclose(fp); typeOut = Type::MetroidPrime2; return ret; } if (ValidateRS1PC(fp)) { auto ret = LoadRS1PC(fp); fclose(fp); typeOut = Type::RogueSquadronPC; return ret; } if (ValidateRS1N64(fp)) { auto ret = LoadRS1N64(fp); fclose(fp); typeOut = Type::RogueSquadronN64; return ret; } if (ValidateBFNPC(fp)) { auto ret = LoadBFNPC(fp); fclose(fp); typeOut = Type::BattleForNabooPC; return ret; } if (ValidateBFNN64(fp)) { auto ret = LoadBFNN64(fp); fclose(fp); typeOut = Type::BattleForNabooN64; return ret; } if (ValidateRS2(fp)) { auto ret = LoadRS2(fp); fclose(fp); typeOut = Type::RogueSquadron2; return ret; } if (ValidateRS3(fp)) { auto ret = LoadRS3(fp); fclose(fp); typeOut = Type::RogueSquadron3; return ret; } fclose(fp); } return {}; } std::vector> ContainerRegistry::LoadSongs(const SystemChar* path) { FILE* fp; /* See if provided file is a raw song */ const SystemChar* dot = nullptr; if (IsSongExtension(path, dot)) { fp = FOpen(path, _S("rb")); size_t fLen = FileLength(fp); if (!fLen) { fclose(fp); return {}; } std::unique_ptr song(new uint8_t[fLen]); fread(song.get(), 1, fLen, fp); fclose(fp); std::vector> ret; ret.emplace_back(_S("Song"), SongData(std::move(song), fLen, -1, -1)); return ret; } /* Now attempt archive-file case */ fp = FOpen(path, _S("rb")); if (fp) { if (ValidateMP1Songs(fp)) { auto ret = LoadMP1Songs(fp); fclose(fp); return ret; } #if 0 if (ValidateRS1PCSongs(fp)) { auto ret = LoadRS1PCSongs(fp); fclose(fp); return ret; } if (ValidateRS1N64Songs(fp)) { auto ret = LoadRS1N64Songs(fp); fclose(fp); return ret; } if (ValidateBFNPCSongs(fp)) { auto ret = LoadBFNPCSongs(fp); fclose(fp); return ret; } if (ValidateBFNN64Songs(fp)) { auto ret = LoadBFNN64Songs(fp); fclose(fp); return ret; } #endif if (ValidateRS2(fp)) { auto ret = LoadRS2Songs(fp); fclose(fp); return ret; } fclose(fp); } return {}; } }