limit number of unknown chunks in bnk

ArthurHeitmann · Dec 7, 2024 · 89896c4 · 89896c4
1 parent c87728d
commit 89896c4
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/lib/fileTypeUtils/audio/bnkIO.dart b/lib/fileTypeUtils/audio/bnkIO.dart
@@ -37,6 +37,7 @@ class BnkFile extends ChunkWithSize {
   BnkFile(this.chunks);
 
   BnkFile.read(ByteDataWrapper bytes) {
+    int unknownChunkCount = 0;
     while (bytes.position < bytes.length) {
       var chunk = _makeNextChunk(bytes);
       if (chunk is BnkHeader) {
@@ -45,6 +46,13 @@ class BnkFile extends ChunkWithSize {
           throw Exception("Unsupported BNK version ${chunk.version}");
         }
       }
+      else if (chunk is BnkUnknownChunk) {
+        unknownChunkCount++;
+        if (unknownChunkCount > 10) {
+          showToast("Warning: More than 10 unknown chunks found. Stopping parsing.");
+          throw Exception("Too many unknown chunks");
+        }
+      }
       chunks.add(chunk);
     }
   }