oot support

CMakeLists.txt

@@ -19,6 +19,8 @@ option(BUILD_PM64 "Build with Paper Mario support" ON)
 option(BUILD_FZERO "Build with F-Zero X support" ON)
 option(BUILD_MARIO_ARTIST "Build with Mario Artist support" ON)
 option(BUILD_NAUDIO "Build with NAudio support" ON)
+option(BUILD_OOT "Build with Ocarina of Time support" ON)
+option(PORT_VERSION_ENDIANNESS "Include endianness byte in portVersion file" OFF)
 
 if(EMSCRIPTEN)
     set(BUILD_SM64 OFF) # TODO: This is broken for some reason
@@ -70,7 +72,7 @@ include_directories(${CMAKE_CURRENT_SOURCE_DIR})
 include_directories(${CMAKE_CURRENT_SOURCE_DIR}/lib)
 include_directories(${CMAKE_CURRENT_SOURCE_DIR}/src)
 file(GLOB_RECURSE CXX_FILES ${CMAKE_CURRENT_SOURCE_DIR}/src/*.cpp ${CMAKE_CURRENT_SOURCE_DIR}/src/**/*.cpp ${CMAKE_CURRENT_SOURCE_DIR}/lib/strhash64/*.cpp)
-file(GLOB C_FILES ${CMAKE_CURRENT_SOURCE_DIR}/src/*.c ${CMAKE_CURRENT_SOURCE_DIR}/src/**/*.c ${CMAKE_CURRENT_SOURCE_DIR}/lib/libmio0/*.c ${CMAKE_CURRENT_SOURCE_DIR}/lib/libyay0/*.c)
+file(GLOB C_FILES ${CMAKE_CURRENT_SOURCE_DIR}/src/*.c ${CMAKE_CURRENT_SOURCE_DIR}/src/**/*.c ${CMAKE_CURRENT_SOURCE_DIR}/lib/libmio0/*.c ${CMAKE_CURRENT_SOURCE_DIR}/lib/libyay0/*.c ${CMAKE_CURRENT_SOURCE_DIR}/lib/libyaz0/*.c)
 set(SRC_DIR ${CXX_FILES} ${C_FILES} ${LGFXD_FILES})
 
 if(BUILD_SM64)
@@ -115,6 +117,16 @@ else()
     list(FILTER SRC_DIR EXCLUDE REGEX "${CMAKE_CURRENT_SOURCE_DIR}/src/factories/naudio/*")
 endif()
 
+if(PORT_VERSION_ENDIANNESS)
+    add_definitions(-DPORT_VERSION_ENDIANNESS)
+endif()
+
+if(BUILD_OOT)
+    add_definitions(-DOOT_SUPPORT)
+else()
+    list(FILTER SRC_DIR EXCLUDE REGEX "${CMAKE_CURRENT_SOURCE_DIR}/src/factories/oot/*")
+endif()
+
 if(ENABLE_ASAN)
     add_compile_options(-fsanitize=address)
     add_link_options(-fsanitize=address)

lib/libyaz0/yaz0.c

@@ -0,0 +1,57 @@
+#include "yaz0.h"
+#include "libmio0/utils.h"
+#include <stdlib.h>
+#include <string.h>
+
+uint8_t* yaz0_decode(const uint8_t* in, uint32_t* out_size) {
+    if (strncmp((const char*)in, "Yaz0", 4) != 0) {
+        return NULL;
+    }
+
+    uint32_t decompressed_size = read_u32_be(in + 4);
+    uint8_t* out = malloc(decompressed_size);
+    if (!out) {
+        return NULL;
+    }
+
+    uint32_t src = YAZ0_HEADER_LENGTH;
+    uint32_t dst = 0;
+    uint8_t group_head = 0;
+    int bits_left = 0;
+
+    while (dst < decompressed_size) {
+        if (bits_left == 0) {
+            group_head = in[src++];
+            bits_left = 8;
+        }
+
+        if (group_head & 0x80) {
+            /* literal byte */
+            out[dst++] = in[src++];
+        } else {
+            /* back-reference */
+            uint8_t b1 = in[src++];
+            uint8_t b2 = in[src++];
+
+            uint32_t dist = ((b1 & 0x0F) << 8) | b2;
+            uint32_t copy_src = dst - (dist + 1);
+            uint32_t count;
+
+            if (b1 >> 4) {
+                count = (b1 >> 4) + 2;
+            } else {
+                count = in[src++] + 0x12;
+            }
+
+            for (uint32_t i = 0; i < count; i++) {
+                out[dst++] = out[copy_src + i];
+            }
+        }
+
+        group_head <<= 1;
+        bits_left--;
+    }
+
+    *out_size = decompressed_size;
+    return out;
+}

lib/libyaz0/yaz0.h

@@ -0,0 +1,7 @@
+#pragma once
+
+#include <stdint.h>
+
+#define YAZ0_HEADER_LENGTH 16
+
+extern uint8_t* yaz0_decode(const uint8_t* in, uint32_t* out_size);

src/AliasManager.cpp

@@ -0,0 +1,23 @@
+#include "AliasManager.h"
+#include "utils/TorchUtils.h"
+
+static AliasManager sAliasManagerInstance;
+AliasManager* AliasManager::Instance = &sAliasManagerInstance;
+
+void AliasManager::Register(const std::string& primaryPath, const std::string& aliasPath) {
+    mAliases[primaryPath].push_back(aliasPath);
+}
+
+void AliasManager::WriteAliases(const std::string& primaryPath, BinaryWrapper* wrapper,
+                                const std::vector<char>& data) {
+    if (!Torch::contains(mAliases, primaryPath)) return;
+
+    for (auto& alias : mAliases[primaryPath]) {
+        wrapper->AddFile(alias, data);
+    }
+    mAliases.erase(primaryPath);
+}
+
+void AliasManager::Clear() {
+    mAliases.clear();
+}

src/AliasManager.h

@@ -0,0 +1,19 @@
+#pragma once
+
+#include <string>
+#include <vector>
+#include <unordered_map>
+#include "archive/BinaryWrapper.h"
+
+class AliasManager {
+public:
+    static AliasManager* Instance;
+
+    void Register(const std::string& primaryPath, const std::string& aliasPath);
+    void WriteAliases(const std::string& primaryPath, BinaryWrapper* wrapper,
+                      const std::vector<char>& data);
+    void Clear();
+
+private:
+    std::unordered_map<std::string, std::vector<std::string>> mAliases;
+};