From e13b10fedf0f7d126ad2c6199ff83f5ffe463c27 Mon Sep 17 00:00:00 2001
From: hippoz <paperplanelive@gmail.com>
Date: Thu, 4 Feb 2021 16:01:14 +0200
Subject: [PATCH] packet fragmentation that doesnt work

---
 example/example.html |  7 ++++++-
 example/example.js   |  8 ++++++--
 lib/Parser.js        | 16 +++++++---------
 lib/Socket.js        | 22 +++++++++++++++++++++-
 lib/constants.js     |  3 ++-
 5 files changed, 42 insertions(+), 14 deletions(-)

diff --git a/example/example.html b/example/example.html
index 71d3a2b..46cc05f 100644
--- a/example/example.html
+++ b/example/example.html
@@ -6,12 +6,17 @@
     <title>Wormhole test</title>
 
     <script>
+        let theFunniestStringEver = '';
+        for (let i = 0; i < 1000000; i++) {
+            theFunniestStringEver += 'h';
+        }
+
         // Create WebSocket connection.
         const socket = new WebSocket('ws://localhost:8080/hello');
 
         // Connection opened
         socket.addEventListener('open', function (event) {
-            socket.send('Hello Server!');
+            socket.send(theFunniestStringEver);
         });
 
         socket.addEventListener('error', console.error)
diff --git a/example/example.js b/example/example.js
index 3177e55..aa37c37 100644
--- a/example/example.js
+++ b/example/example.js
@@ -9,9 +9,13 @@ const wormhole = new Wormhole({ urls: [ '/hello' ], httpServer });
 wormhole.on('connect', ({ socket, accept, reject }) => {
     accept();
     const buf = Buffer.from([0x41, 0x41, 0x41, 0x41]);
-    console.log(buf);
     socket.send(buf);
-    socket.on('text', console.log);
+    socket.on('text', (e) => {
+        //console.log('Got text frame', e);
+    });
+    socket.on('binary', () => {
+        //console.log('Got binary frame', e);
+    });
 });
 
 httpServer.listen(8080);
\ No newline at end of file
diff --git a/lib/Parser.js b/lib/Parser.js
index 174b3b3..48570ad 100644
--- a/lib/Parser.js
+++ b/lib/Parser.js
@@ -46,8 +46,6 @@ WebsocketFrame.prototype.toBuffer = function() {
     this.MASK && packet.push(this.MaskingKey);
     packet.push(this.PayloadData);
 
-    console.log(packet);
-
     return Buffer.concat(packet);
 };
 
@@ -58,13 +56,13 @@ const parseWebsocketFrame = (data) => {
     const frame = new WebsocketFrame();
 
     // 1 byte - first byte
-    frame.FIN = (firstByte & 0x01); // 0x01[0b10000000] - Get most significant bit (FIN)
-    frame.RSVx = (firstByte & 0x70) // 0x70[0b01110000] - Get the 3 bits after the most significant bit (RSVx)
-    frame.Opcode = (firstByte & 0x0F) // 0xF[0b00001111] - Get the last 4 bits (Opcode)
+    frame.FIN = (firstByte >> 7); // >> 7 - Get most significant bit (FIN)
+    frame.RSVx = (firstByte & 0x70) >> 4; // 0x70[0b01110000] - Get the 3 bits after the most significant bit (RSVx)
+    frame.Opcode = (firstByte & 0x0F); // 0xF[0b00001111] - Get the last 4 bits (Opcode)
 
     // 1 byte - second byte
-    frame.MASK = (secondByte & 0x01) // 0x01[0b10000000] - Get most significant bit (MASK)
-    frame.PayloadLen = (secondByte & 0x7F) // 0x7F[0b01111111] - Get last 7 bits (Payload len)
+    frame.MASK = (secondByte >> 7); // >> 7 - Get most significant bit (MASK)
+    frame.PayloadLen = (secondByte & 0x7F); // 0x7F[0b01111111] - Get last 7 bits (Payload len)
 
     let maskingKeyOffset = 2; // By default, theres a 2 byte offset. We will modify this in the cases below.
 
@@ -79,11 +77,11 @@ const parseWebsocketFrame = (data) => {
         frame.PayloadLenEx = frame.PayloadLen;
     }
     
-    const maskingKeyEnd = maskingKeyOffset + 4; // (4 bytes because it is a 32 bit value)
+    let maskingKeyEnd = frame.MASK ? maskingKeyOffset + 4 : maskingKeyOffset; // (4 bytes because it is a 32 bit value)    
     if (frame.MASK) frame.MaskingKey = data.buffer.slice(maskingKeyOffset, maskingKeyEnd); // Create a new buffer starting at the masking key offset and ending at the masking key end (duh).
 
     // TODO: Separate extension data from application data
-    frame.PayloadData = data.buffer.slice(maskingKeyEnd, maskingKeyEnd + frame.PayloadLenEx); // Create a new buffer that starts at the end of the masking key and ends at (the end of the masking key + payload length)
+    frame.PayloadData = data.buffer.slice(maskingKeyEnd, Number(BigInt(maskingKeyEnd) + BigInt(frame.PayloadLenEx))); // Create a new buffer that starts at the end of the masking key and ends at (the end of the masking key + payload length)
 
     // TODO: implement unmasked string decoding
     if (frame.MASK) {
diff --git a/lib/Socket.js b/lib/Socket.js
index 75675ca..16ada56 100644
--- a/lib/Socket.js
+++ b/lib/Socket.js
@@ -12,10 +12,31 @@ class Socket extends EventEmitter {
         this._accepted = false;
         this._fateDecided = false; // Wether the decision to accept or reject the socket was made
 
+        this._buffering = undefined;
+
         this._socket.on('data', (e) => {
             if (!this._isConnected()) return;
 
             const packet = this._decodePayload(e.buffer);
+
+            console.log(packet);
+
+            if (this._buffering && packet.FIN === 1) {
+                this.emit('binary', Buffer.concat([this._buffering, packet.PayloadData]));
+                this._buffering = undefined;
+            }
+
+            if (packet.FIN === 0) {
+                if (!this._buffering) {
+                    this._buffering = Buffer.from(packet.PayloadData);
+                } else {
+                    Buffer.concat([this._buffering, Buffer.from(packet.PayloadData)]);
+                }
+                
+                console.log('Buffering', packet);
+                return;
+            }
+
             if (packet.Opcode === constants.opcodes.TEXT_FRAME) {
                 this.emit('text', packet);
             } else if (packet.Opcode === constants.opcodes.BINARY_FRAME) {
@@ -43,7 +64,6 @@ Socket.prototype._sendBuffer = function(buffer) {
 
     frame.MASK = 0;
     const length = buffer.byteLength;
-    console.log(length);
     if (length > 125) {
         frame.PayloadLen = 126;
     } else if (length >= 65536) {
diff --git a/lib/constants.js b/lib/constants.js
index feff026..91d587e 100644
--- a/lib/constants.js
+++ b/lib/constants.js
@@ -9,7 +9,8 @@ module.exports = {
         CLOSED: 'CLOSED'
     },
     opcodes: {
+        'CONTINUATION': 0x00,
         'TEXT_FRAME': 0x01,
         'BINARY_FRAME': 0x02
     }
-}
\ No newline at end of file
+};
\ No newline at end of file