don't crash if no serverConfig and use default encoding

Author: sidorares

lib/connection.js

@@ -87,7 +87,7 @@ function Connection(opts) {
 
   this.clientEncoding = CharsetToEncoding[this.config.charsetNumber];
 
-  this.stream.once('error', connection._handleNetworkError.bind(this));
+  this.stream.on('error', connection._handleNetworkError.bind(this));
 
   // see https://gist.github.com/khoomeister/4985691#use-that-instead-of-bind
   this.packetParser = new PacketParser(function(p) {
@@ -123,11 +123,17 @@ function Connection(opts) {
   if (!this.config.isServer) {
     handshakeCommand = new Commands.ClientHandshake(this.config.clientFlags);
     handshakeCommand.on('end', function() {
+      // this happens when handshake finishes early and first packet is error
+      // and not server hello ( for example, 'Too many connactions' error)
+      if (!handshakeCommand.handshake) {
+        return;
+      }
       connection._handshakePacket = handshakeCommand.handshake;
       connection.threadId = handshakeCommand.handshake.connectionId;
       connection.emit('connect', handshakeCommand.handshake);
     });
     handshakeCommand.on('error', function(err) {
+      connection._closing = true;
       connection._notifyError(err);
     });
     this.addCommand(handshakeCommand);
@@ -174,6 +180,7 @@ Connection.prototype._handleFatalError = function(err) {
 };
 
 Connection.prototype._handleNetworkError = function(err) {
+  console.log('_handleNetworkError AAAAA NETWORK ERROR');
   this._handleFatalError(err);
 };
 
@@ -212,7 +219,16 @@ Connection.prototype._notifyError = function(err) {
     connection._command.onResult(err);
     connection._command = null;
   } else {
-    bubbleErrorToConnection = true;
+    // connection handshake is special because we allow it to be implicit
+    // if error happened during handshake, but there are others commands in queue
+    // then bubble error to other commands and not to connection
+    if (
+      !(connection._command &&
+        connection._command.constructor == Commands.ClientHandshake &&
+        connection._commands.length > 0)
+    ) {
+      bubbleErrorToConnection = true;
+    }
   }
   while ((command = connection._commands.shift())) {
     if (command.onResult) {
@@ -900,7 +916,7 @@ Connection.prototype.writeOk = function(args) {
 Connection.prototype.writeError = function(args) {
   // if we want to send error before initial hello was sent, use default encoding
   var encoding = this.serverConfig ? this.serverConfig.encoding : 'cesu8';
-  this.writePacket(Packets.Error.toPacket(args, this.serverConfig.encoding));
+  this.writePacket(Packets.Error.toPacket(args, encoding));
 };
 
 Connection.prototype.serverHandshake = function serverHandshake(args) {