tedious 1.12.3 → 1.14.0

package/lib/message-io.js

@@ -1,235 +1,205 @@
-var _regeneratorRuntime = require('babel-runtime/regenerator').default;
-
-// Generated by CoffeeScript 1.10.0
-var EventEmitter,
-    MessageIO,
-    Packet,
-    ReadablePacketStream,
-    StreamParser,
-    TYPE,
-    crypto,
-    isPacketComplete,
-    packetHeaderLength,
-    packetLength,
-    tls,
-    extend = function (child, parent) {
-  for (var key in parent) {
-    if (hasProp.call(parent, key)) child[key] = parent[key];
-  }function ctor() {
-    this.constructor = child;
-  }ctor.prototype = parent.prototype;child.prototype = new ctor();child.__super__ = parent.prototype;return child;
-},
-    hasProp = ({}).hasOwnProperty,
-    bind = function (fn, me) {
-  return function () {
-    return fn.apply(me, arguments);
-  };
-};
-
-tls = require('tls');
-
-crypto = require('crypto');
+'use strict';
 
-require('./buffertools');
-
-EventEmitter = require('events').EventEmitter;
+var _get = require('babel-runtime/helpers/get')['default'];
 
-isPacketComplete = require('./packet').isPacketComplete;
+var _inherits = require('babel-runtime/helpers/inherits')['default'];
 
-packetLength = require('./packet').packetLength;
+var _createClass = require('babel-runtime/helpers/create-class')['default'];
 
-packetHeaderLength = require('./packet').HEADER_LENGTH;
+var _classCallCheck = require('babel-runtime/helpers/class-call-check')['default'];
 
-Packet = require('./packet').Packet;
+var tls = require('tls');
+var crypto = require('crypto');
+var EventEmitter = require('events').EventEmitter;
+var Transform = require('readable-stream').Transform;
 
-TYPE = require('./packet').TYPE;
+require('./buffertools');
 
-StreamParser = require('./stream-parser');
+var Packet = require('./packet').Packet;
+var TYPE = require('./packet').TYPE;
+var packetHeaderLength = require('./packet').HEADER_LENGTH;
 
-ReadablePacketStream = (function (superClass) {
-  extend(ReadablePacketStream, superClass);
+var ReadablePacketStream = (function (_Transform) {
+  _inherits(ReadablePacketStream, _Transform);
 
   function ReadablePacketStream() {
-    ReadablePacketStream.__super__.constructor.call(this);
+    _classCallCheck(this, ReadablePacketStream);
+
+    _get(Object.getPrototypeOf(ReadablePacketStream.prototype), 'constructor', this).call(this, { objectMode: true });
+
+    this.buffer = new Buffer(0);
+    this.position = 0;
   }
 
-  ReadablePacketStream.prototype.parser = _regeneratorRuntime.mark(function callee$1$0() {
-    var data, length, packetId, spid, status, type, window;
-    return _regeneratorRuntime.wrap(function callee$1$0$(context$2$0) {
-      while (1) switch (context$2$0.prev = context$2$0.next) {
-        case 0:
-          if (!true) {
-            context$2$0.next = 25;
-            break;
-          }
-
-          context$2$0.next = 3;
-          return this.readUInt8();
-
-        case 3:
-          type = context$2$0.sent;
-          context$2$0.next = 6;
-          return this.readUInt8();
-
-        case 6:
-          status = context$2$0.sent;
-          context$2$0.next = 9;
-          return this.readUInt16BE();
-
-        case 9:
-          length = context$2$0.sent;
-          context$2$0.next = 12;
-          return this.readUInt16BE();
-
-        case 12:
-          spid = context$2$0.sent;
-          context$2$0.next = 15;
-          return this.readUInt8();
-
-        case 15:
-          packetId = context$2$0.sent;
-          context$2$0.next = 18;
-          return this.readUInt8();
-
-        case 18:
-          window = context$2$0.sent;
-          context$2$0.next = 21;
-          return this.readBuffer(length - packetHeaderLength);
-
-        case 21:
-          data = context$2$0.sent;
-
-          this.push({
-            data: data,
-            isLast: status & 0x01 ? function () {
-              return true;
-            } : function () {
-              return false;
-            }
-          });
-          context$2$0.next = 0;
-          break;
+  _createClass(ReadablePacketStream, [{
+    key: '_transform',
+    value: function _transform(chunk, encoding, callback) {
+      if (this.position === this.buffer.length) {
+        // If we have fully consumed the previous buffer,
+        // we can just replace it with the new chunk
+        this.buffer = chunk;
+      } else {
+        // If we haven't fully consumed the previous buffer,
+        // we simply concatenate the leftovers and the new chunk.
+        this.buffer = Buffer.concat([this.buffer.slice(this.position), chunk], this.buffer.length - this.position + chunk.length);
+      }
+
+      this.position = 0;
 
-        case 25:
-          return context$2$0.abrupt('return', void 0);
+      // The packet header is always 8 bytes of length.
+      while (this.buffer.length >= this.position + packetHeaderLength) {
+        // Get the full packet length
+        var _length = this.buffer.readUInt16BE(this.position + 2);
 
-        case 26:
-        case 'end':
-          return context$2$0.stop();
+        if (this.buffer.length >= this.position + _length) {
+          var data = this.buffer.slice(this.position, this.position + _length);
+          this.position += _length;
+          this.push(new Packet(data));
+        } else {
+          // Not enough data to provide the next packet. Stop here and wait for
+          // the next call to `_transform`.
+          break;
+        }
       }
-    }, callee$1$0, this);
-  });
+
+      callback();
+    }
+  }]);
 
   return ReadablePacketStream;
-})(StreamParser);
+})(Transform);
 
-MessageIO = (function (superClass) {
-  extend(MessageIO, superClass);
+module.exports = (function (_EventEmitter) {
+  _inherits(MessageIO, _EventEmitter);
 
   function MessageIO(socket, _packetSize, debug) {
+    var _this = this;
+
+    _classCallCheck(this, MessageIO);
+
+    _get(Object.getPrototypeOf(MessageIO.prototype), 'constructor', this).call(this);
+
     this.socket = socket;
     this._packetSize = _packetSize;
     this.debug = debug;
-    this.sendPacket = bind(this.sendPacket, this);
+    this.sendPacket = this.sendPacket.bind(this);
+
     this.packetStream = new ReadablePacketStream();
-    this.packetStream.on('data', (function (_this) {
-      return function (packet) {
-        _this.emit('data', packet.data);
-        if (packet.isLast()) {
-          return _this.emit('message');
-        }
-      };
-    })(this));
+    this.packetStream.on('data', function (packet) {
+      _this.logPacket('Received', packet);
+      _this.emit('data', packet.data());
+      if (packet.isLast()) {
+        _this.emit('message');
+      }
+    });
+
     this.socket.pipe(this.packetStream);
     this.packetDataSize = this._packetSize - packetHeaderLength;
   }
 
-  MessageIO.prototype.packetSize = function (packetSize) {
-    if (arguments.length > 0) {
-      this.debug.log("Packet size changed from " + this._packetSize + " to " + packetSize);
-      this._packetSize = packetSize;
-      this.packetDataSize = this._packetSize - packetHeaderLength;
+  _createClass(MessageIO, [{
+    key: 'packetSize',
+    value: function packetSize(_packetSize2) {
+      if (arguments.length > 0) {
+        this.debug.log('Packet size changed from ' + this._packetSize + ' to ' + _packetSize2);
+        this._packetSize = _packetSize2;
+        this.packetDataSize = this._packetSize - packetHeaderLength;
+      }
+      return this._packetSize;
+    }
+  }, {
+    key: 'startTls',
+    value: function startTls(credentialsDetails) {
+      var _this2 = this;
+
+      var credentials = tls.createSecureContext ? tls.createSecureContext(credentialsDetails) : crypto.createCredentials(credentialsDetails);
+
+      this.securePair = tls.createSecurePair(credentials);
+      this.tlsNegotiationComplete = false;
+
+      this.securePair.on('secure', function () {
+        var cipher = _this2.securePair.cleartext.getCipher();
+        _this2.debug.log('TLS negotiated (' + cipher.name + ', ' + cipher.version + ')');
+        _this2.emit('secure', _this2.securePair.cleartext);
+        _this2.encryptAllFutureTraffic();
+      });
+
+      this.securePair.encrypted.on('data', function (data) {
+        _this2.sendMessage(TYPE.PRELOGIN, data);
+      });
+
+      // On Node >= 0.12, the encrypted stream automatically starts spewing out
+      // data once we attach a `data` listener. But on Node <= 0.10.x, this is not
+      // the case. We need to kick the cleartext stream once to get the
+      // encrypted end of the secure pair to emit the TLS handshake data.
+      this.securePair.cleartext.write('');
+    }
+  }, {
+    key: 'encryptAllFutureTraffic',
+    value: function encryptAllFutureTraffic() {
+      this.socket.unpipe(this.packetStream);
+      this.securePair.encrypted.removeAllListeners('data');
+      this.socket.pipe(this.securePair.encrypted);
+      this.securePair.encrypted.pipe(this.socket);
+      this.securePair.cleartext.pipe(this.packetStream);
+      this.tlsNegotiationComplete = true;
     }
-    return this._packetSize;
-  };
-
-  MessageIO.prototype.startTls = function (credentialsDetails) {
-    var credentials;
-    credentials = tls.createSecureContext ? tls.createSecureContext(credentialsDetails) : crypto.createCredentials(credentialsDetails);
-    this.securePair = tls.createSecurePair(credentials);
-    this.tlsNegotiationComplete = false;
-    this.securePair.on('secure', (function (_this) {
-      return function () {
-        var cipher;
-        cipher = _this.securePair.cleartext.getCipher();
-        _this.debug.log("TLS negotiated (" + cipher.name + ", " + cipher.version + ")");
-        _this.emit('secure', _this.securePair.cleartext);
-        return _this.encryptAllFutureTraffic();
-      };
-    })(this));
-    this.securePair.encrypted.on('data', (function (_this) {
-      return function (data) {
-        return _this.sendMessage(TYPE.PRELOGIN, data);
-      };
-    })(this));
-    return this.securePair.cleartext.write('');
-  };
-
-  MessageIO.prototype.encryptAllFutureTraffic = function () {
-    this.socket.unpipe(this.packetStream);
-    this.securePair.encrypted.removeAllListeners('data');
-    this.socket.pipe(this.securePair.encrypted);
-    this.securePair.encrypted.pipe(this.socket);
-    this.securePair.cleartext.pipe(this.packetStream);
-    return this.tlsNegotiationComplete = true;
-  };
-
-  MessageIO.prototype.tlsHandshakeData = function (data) {
-    return this.securePair.encrypted.write(data);
-  };
-
-  MessageIO.prototype.sendMessage = function (packetType, data, resetConnection) {
-    var i, numberOfPackets, packet, packetNumber, packetPayload, payloadEnd, payloadStart, ref, results;
-    if (data) {
-      numberOfPackets = Math.floor((data.length - 1) / this.packetDataSize) + 1;
-    } else {
-      numberOfPackets = 1;
-      data = new Buffer(0);
+  }, {
+    key: 'tlsHandshakeData',
+    value: function tlsHandshakeData(data) {
+      this.securePair.encrypted.write(data);
     }
-    results = [];
-    for (packetNumber = i = 0, ref = numberOfPackets - 1; 0 <= ref ? i <= ref : i >= ref; packetNumber = 0 <= ref ? ++i : --i) {
-      payloadStart = packetNumber * this.packetDataSize;
-      if (packetNumber < numberOfPackets - 1) {
-        payloadEnd = payloadStart + this.packetDataSize;
+
+    // TODO listen for 'drain' event when socket.write returns false.
+    // TODO implement incomplete request cancelation (2.2.1.6)
+  }, {
+    key: 'sendMessage',
+    value: function sendMessage(packetType, data, resetConnection) {
+      var numberOfPackets = undefined;
+      if (data) {
+        numberOfPackets = Math.floor((data.length - 1) / this.packetDataSize) + 1;
       } else {
-        payloadEnd = data.length;
+        numberOfPackets = 1;
+        data = new Buffer(0);
+      }
+
+      for (var packetNumber = 0; packetNumber < numberOfPackets; packetNumber++) {
+        var payloadStart = packetNumber * this.packetDataSize;
+
+        var payloadEnd = undefined;
+        if (packetNumber < numberOfPackets - 1) {
+          payloadEnd = payloadStart + this.packetDataSize;
+        } else {
+          payloadEnd = data.length;
+        }
+
+        var packetPayload = data.slice(payloadStart, payloadEnd);
+
+        var packet = new Packet(packetType);
+        packet.last(packetNumber === numberOfPackets - 1);
+        packet.resetConnection(resetConnection);
+        packet.packetId(packetNumber + 1);
+        packet.addData(packetPayload);
+        this.sendPacket(packet);
+      }
+    }
+  }, {
+    key: 'sendPacket',
+    value: function sendPacket(packet) {
+      this.logPacket('Sent', packet);
+      if (this.securePair && this.tlsNegotiationComplete) {
+        this.securePair.cleartext.write(packet.buffer);
+      } else {
+        this.socket.write(packet.buffer);
       }
-      packetPayload = data.slice(payloadStart, payloadEnd);
-      packet = new Packet(packetType);
-      packet.last(packetNumber === numberOfPackets - 1);
-      packet.resetConnection(resetConnection);
-      packet.packetId(packetNumber + 1);
-      packet.addData(packetPayload);
-      results.push(this.sendPacket(packet, packetType));
     }
-    return results;
-  };
-
-  MessageIO.prototype.sendPacket = function (packet, packetType) {
-    this.logPacket('Sent', packet);
-    if (this.securePair && this.tlsNegotiationComplete) {
-      return this.securePair.cleartext.write(packet.buffer);
-    } else {
-      return this.socket.write(packet.buffer);
+  }, {
+    key: 'logPacket',
+    value: function logPacket(direction, packet) {
+      this.debug.packet(direction, packet);
+      return this.debug.data(packet);
     }
-  };
-
-  MessageIO.prototype.logPacket = function (direction, packet) {
-    this.debug.packet(direction, packet);
-    return this.debug.data(packet);
-  };
+  }]);
 
   return MessageIO;
-})(EventEmitter);
-
-module.exports = MessageIO;
+})(EventEmitter);