Only convert the current message to a string once the last fragment has been received

Author: aeaton-overleaf

lib/transports/websocket/hybi-16.js

@@ -281,7 +281,7 @@ function Parser (opts) {
   this.expectOffset = 0;
   this.expectBuffer = null;
   this.expectHandler = null;
-  this.currentMessage = '';
+  this.currentMessage = [];
   this._maxBuffer = (opts && opts.maxBuffer) || 10E7;
   this._dataLength = 0;
 
@@ -290,10 +290,10 @@ function Parser (opts) {
     // text
     '1': function(data) {
       var finish = function(mask, data) {
-        self.currentMessage += self.unmask(mask, data);
+        self.currentMessage.push(self.unmask(mask, data, true));
         if (self.state.lastFragment) {
-          self.emit('data', self.currentMessage);
-          self.currentMessage = '';
+          self.emit('data', self.concatBuffers(self.currentMessage).toString('utf8'));
+          self.currentMessage = [];
         }
         self.endPacket();
       }
@@ -338,11 +338,10 @@ function Parser (opts) {
     // binary
     '2': function(data) {
       var finish = function(mask, data) {
-        if (typeof self.currentMessage == 'string') self.currentMessage = []; // build a buffer list
         self.currentMessage.push(self.unmask(mask, data, true));
         if (self.state.lastFragment) {
           self.emit('binary', self.concatBuffers(self.currentMessage));
-          self.currentMessage = '';
+          self.currentMessage = [];
         }
         self.endPacket();
       }