From 3dbd4d9a573bbbeacef2729bbb4d0b6e658d206e Mon Sep 17 00:00:00 2001
From: holly <onlinegamer748@gmail.com>
Date: Tue, 24 Aug 2021 15:12:21 +0100
Subject: [PATCH] Support memory pre-allocation for all writable types

---
 server/bufferStuff.js | 102 ++++++++++++++++++++++++++++++------------
 1 file changed, 74 insertions(+), 28 deletions(-)

diff --git a/server/bufferStuff.js b/server/bufferStuff.js
index e5272c5..ffc5bb5 100644
--- a/server/bufferStuff.js
+++ b/server/bufferStuff.js
@@ -6,6 +6,12 @@
 */
 
 module.exports.Writer = class {
+	// bufferStuff supports pre-allocating memory for the buffer
+	// if you pass in a size of 0 then it will just dynamicly allocate at the
+	// cost of performance
+
+	// NOTE: In pre-allocation mode if you exceed the size of the buffer
+	// 		 that you set it will cause a crash.
 	constructor(size = 0) {
 		this.buffer = Buffer.alloc(size);
 		this.offset = 0;
@@ -49,61 +55,101 @@ module.exports.Writer = class {
 	}
 
 	writeByteArray(data = [0]) {
-		const buff = Buffer.alloc(data.length);
+		if (this.baseSize == 0) {
+			const buff = Buffer.alloc(data.length);
 
-		for (let byte of data) {
-			buff.writeInt8(byte);
+			for (let byte of data) {
+				buff.writeInt8(byte);
+			}
+
+			this.writeBuffer(buff);
+		} else {
+			for (let byte of data) {
+				this.buffer.writeInt8(byte);
+				this.offset += 1;
+			}
 		}
-
-		this.writeBuffer(buff);
 	}
 
 	writeShort(data = 0) {
-		const buff = Buffer.alloc(2);
-		buff.writeIntBE(data, 0, 2);
+		if (this.baseSize == 0) {
+			const buff = Buffer.alloc(2);
+			buff.writeIntBE(data, 0, 2);
 
-		this.writeBuffer(buff);
+			this.writeBuffer(buff);
+		} else {
+			this.buffer.writeIntBE(data, this.offset, 2);
+			this.offset += 2;
+		}
 	}
 
 	writeShortArray(data = [0]) {
-		const buff = Buffer.alloc(data.length * 2);
-		let offset = 0;
+		if (this.baseSize == 0) {
+			const buff = Buffer.alloc(data.length * 2);
+			let offset = 0;
 
-		for (let short of data) {
-			buff.writeIntBE(short, offset, 2);
-			offset += 2;
+			for (let short of data) {
+				buff.writeIntBE(short, offset, 2);
+				offset += 2;
+			}
+
+			this.writeBuffer(buff);
+		} else {
+			for (let short of data) {
+				this.buffer.writeIntBE(short, this.offset, 2);
+				this.offset += 2;
+			}
 		}
-
-		this.writeBuffer(buff);
 	}
 
 	writeInt(data = 0) {
-		const buff = Buffer.alloc(4);
-		buff.writeIntBE(data, 0, 4);
+		if (this.baseSize == 0) {
+			const buff = Buffer.alloc(4);
+			buff.writeIntBE(data, 0, 4);
 
-		this.writeBuffer(buff);
+			this.writeBuffer(buff);
+		} else {
+			this.buffer.writeIntBE(data, this.offset, 4);
+			this.offset += 4;
+		}
 	}
 
 	writeLong(data = 0) {
-		const buff = Buffer.alloc(8);
-		if (data instanceof BigInt) buff.writeBigInt64BE(data, 0);
-		else buff.writeBigInt64BE(BigInt(data), 0);
+		if (this.baseSize) {
+			const buff = Buffer.alloc(8);
+			if (data instanceof BigInt) buff.writeBigInt64BE(data, 0);
+			else buff.writeBigInt64BE(BigInt(data), 0);
 
-		this.writeBuffer(buff);
+			this.writeBuffer(buff);
+		} else {
+			if (data instanceof BigInt) this.buffer.writeBigInt64BE(data, this.offset);
+			else this.buffer.writeBigInt64BE(BigInt(data), this.offset);
+			this.offset += 8;
+		}
 	}
 
 	writeFloat(data = 0.0) {
-		const buff = Buffer.alloc(4);
-		buff.writeFloatBE(data, 0);
+		if (this.baseSize == 0) {
+			const buff = Buffer.alloc(4);
+			buff.writeFloatBE(data, 0);
 
-		this.writeBuffer(buff);
+			this.writeBuffer(buff);
+		} else {
+			this.buffer.writeFloatBE(data, this.offset);
+			this.offset += 4;
+		}
 	}
 
 	writeDouble(data = 0.0) {
-		const buff = Buffer.alloc(8);
-		buff.writeDoubleBE(data, 0);
+		if (this.baseSize == 0) {
+			const buff = Buffer.alloc(8);
+			buff.writeDoubleBE(data, 0);
 
-		this.writeBuffer(buff);
+			this.writeBuffer(buff);
+		} else {
+			this.buffer.writeDoubleBE(data, this.offset);
+			this.offset += 8;
+		}
 	}
 
 	writeString(string = "") {