Use ProtoDef skipChecks flag to allow full NBT array size support

README.md

@@ -58,10 +58,18 @@ Minecraft Java Edition uses big-endian format, and Bedrock uses little-endian.
 
 Returns a buffer with a serialized nbt `value`. 
 
-### parseUncompressed(data, format='big')
+### parseUncompressed(data, format='big', options?= {noArraySizeCheck?: boolean})
 
 Takes a buffer `data` and returns a parsed nbt value.
 
+The `options` parameter is optional. When `noArraySizeCheck` is `true`, an array size check is disabled which allows for parsing of large arrays.
+
+### parseAs(data, type, options?= {noArraySizeCheck?: boolean})
+
+Takes a buffer `data` and returns a parsed nbt value. If the buffer is gzipped, it will unzip the data first.
+
+The `options` parameter is optional. When `noArraySizeCheck` is `true`, an array size check is disabled which allows for parsing of large arrays.
+
 
 ### simplify(nbt)
 

nbt.js

@@ -55,8 +55,11 @@ function writeUncompressed (value, proto = 'big') {
   return protos[proto].createPacketBuffer('nbt', value)
 }
 
-function parseUncompressed (data, proto = 'big') {
+function parseUncompressed (data, proto = 'big', options = {}) {
   if (proto === true) proto = 'little'
+
+  protos[proto].setVariable('noArraySizeCheck', options.noArraySizeCheck)
+
   return protos[proto].parsePacketBuffer('nbt', data, data.startOffset).data
 }
 
@@ -70,7 +73,7 @@ const hasGzipHeader = function (data) {
 const hasBedrockLevelHeader = (data) =>
   data[1] === 0 && data[2] === 0 && data[3] === 0
 
-async function parseAs (data, type) {
+async function parseAs (data, type, options = {}) {
   if (hasGzipHeader(data)) {
     data = await new Promise((resolve, reject) => {
       zlib.gunzip(data, (error, uncompressed) => {
@@ -79,7 +82,11 @@ async function parseAs (data, type) {
       })
     })
   }
+
+  protos[type].setVariable('noArraySizeCheck', options.noArraySizeCheck)
+
   const parsed = protos[type].parsePacketBuffer('nbt', data, data.startOffset)
+
   parsed.metadata.buffer = data
   parsed.type = type
   return parsed

package.json

@@ -33,7 +33,7 @@
     "fix": "standard --fix"
   },
   "dependencies": {
-    "protodef": "^1.9.0"
+    "protodef": "bdkopen/node-protodef#add-skip-array-checks-flag"
   },
   "license": "MIT"
 }

typings/index.d.ts

@@ -64,6 +64,10 @@ declare module 'prismarine-nbt'{
     [TagType.LongArray]: LongArray;
   }
 
+  interface ParseOptions {
+    noArraySizeCheck?: boolean;
+  }
+
   export type NBTFormat = 'big' | 'little' | 'littleVarint'
 
   export type NBT = Tags['compound'] & {name: string};
@@ -74,7 +78,8 @@ declare module 'prismarine-nbt'{
     size: number
   }
   export function writeUncompressed(value: NBT, format?: NBTFormat): Buffer;
-  export function parseUncompressed(value: Buffer, format?: NBTFormat): NBT;
+  export function parseUncompressed(value: Buffer, format?: NBTFormat, options?: ParseOptions): NBT;
+  export function parseAs(value: Buffer, type: NBTFormat, options?: ParseOptions): Promise<{parsed: NBT, type: NBTFormat, metadata: Metadata}>;
 
   export function parse(data: Buffer, nbtType?: NBTFormat): Promise<{parsed: NBT, type: NBTFormat, metadata: Metadata}>;
   export function simplify(data: Tags[TagType]): any