msgpackr 1.11.11 → 2.0.0

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "msgpackr",
   "author": "Kris Zyp",
-  "version": "1.11.11",
+  "version": "2.0.0",
   "description": "Ultra-fast MessagePack implementation with extensions for records and structured cloning",
   "license": "MIT",
   "types": "./index.d.ts",

package/stream.js

@@ -1,57 +1,62 @@
-import { Transform } from 'stream'
-import { Packr } from './pack.js'
-import { Unpackr } from './unpack.js'
-var DEFAULT_OPTIONS = {objectMode: true}
+import { Transform } from 'stream';
+import { Packr } from './pack.js';
+import { Unpackr } from './unpack.js';
+var DEFAULT_OPTIONS = {objectMode: true};
 
 export class PackrStream extends Transform {
 	constructor(options) {
 		if (!options)
-			options = {}
-		options.writableObjectMode = true
-		super(options)
-		options.sequential = true
-		this.packr = options.packr || new Packr(options)
+			options = {};
+		options.writableObjectMode = true;
+		super(options);
+		options.sequential = true;
+		this.packr = options.packr || new Packr(options);
 	}
 	_transform(value, encoding, callback) {
-		this.push(this.packr.pack(value))
-		callback()
+		this.push(this.packr.pack(value));
+		callback();
 	}
 }
 
 export class UnpackrStream extends Transform {
 	constructor(options) {
 		if (!options)
-			options = {}
-		options.objectMode = true
-		super(options)
-		options.structures = []
-		this.unpackr = options.unpackr || new Unpackr(options)
+			options = {};
+		options.objectMode = true;
+		super(options);
+		options.structures = [];
+		this.maxIncompleteBufferSize = options.maxIncompleteBufferSize !== undefined ? options.maxIncompleteBufferSize : 0x4000000;
+		this.unpackr = options.unpackr || new Unpackr(options);
 	}
 	_transform(chunk, encoding, callback) {
 		if (this.incompleteBuffer) {
-			chunk = Buffer.concat([this.incompleteBuffer, chunk])
-			this.incompleteBuffer = null
+			chunk = Buffer.concat([this.incompleteBuffer, chunk]);
+			this.incompleteBuffer = null;
 		}
-		let values
+		let values;
 		try {
-			values = this.unpackr.unpackMultiple(chunk)
+			values = this.unpackr.unpackMultiple(chunk);
 		} catch(error) {
 			if (error.incomplete) {
-				this.incompleteBuffer = chunk.slice(error.lastPosition)
-				values = error.values
-			}
-			else
-				throw error
-		} finally {
-			for (let value of values || []) {
-				if (value === null)
-					value = this.getNullValue()
-				this.push(value)
+				let incompleteBuffer = chunk.slice(error.lastPosition);
+				if (incompleteBuffer.length > this.maxIncompleteBufferSize) {
+					this.incompleteBuffer = null;
+					return callback(new Error('Maximum incomplete buffer size exceeded'));
+				}
+				this.incompleteBuffer = incompleteBuffer;
+				values = error.values;
+			} else {
+				return callback(error);
 			}
 		}
-		if (callback) callback()
+		for (let value of values || []) {
+			if (value === null)
+				value = this.getNullValue();
+			this.push(value);
+		}
+		callback();
 	}
 	getNullValue() {
-		return Symbol.for(null)
+		return Symbol.for(null);
 	}
 }