From beab97643d8ceafb9eea0bd12e134286abe53d2a Mon Sep 17 00:00:00 2001
From: Xabier Napal <xabier.napal@empro-ms.com>
Date: Fri, 21 Aug 2020 14:23:55 +0200
Subject: [PATCH] Close and destroy underlying socket after any connection
 error

---
 lib/connect.js    | 11 +++++++++--
 lib/connection.js | 33 +++++++++++++++++++++++++++------
 2 files changed, 36 insertions(+), 8 deletions(-)

diff --git a/lib/connect.js b/lib/connect.js
index 603f1981..be14d8e2 100644
--- a/lib/connect.js
+++ b/lib/connect.js
@@ -150,8 +150,11 @@ function connect(url, socketOptions, openCallback) {
       if (timeout) sock.setTimeout(0);
       if (err === null) {
         openCallback(null, c);
+      } else {
+        sock.end();
+        sock.destroy();
+        openCallback(err);
       }
-      else openCallback(err);
     });
   }
 
@@ -174,7 +177,11 @@ function connect(url, socketOptions, openCallback) {
   }
 
   sock.once('error', function(err) {
-    if (!sockok) openCallback(err);
+    if (!sockok) {
+      sock.end();
+      sock.destroy();
+      openCallback(err);
+    }
   });
 
 }
diff --git a/lib/connection.js b/lib/connection.js
index f65138c5..1954960b 100644
--- a/lib/connection.js
+++ b/lib/connection.js
@@ -179,8 +179,15 @@ C.open = function(allFields, openCallback0) {
   function send(Method) {
     // This can throw an exception if there's some problem with the
     // options; e.g., something is a string instead of a number.
-    try { self.sendMethod(0, Method, tunedOptions); }
-    catch (err) { bail(err); }
+    try {
+      self.sendMethod(0, Method, tunedOptions);
+    } catch (err) {
+      bail(err);
+
+      // We are in an inconsistent state so we can't continue,
+      // rethrowing the exception will stop further execution.
+      throw err;
+    }
   }
 
   function negotiate(server, desired) {
@@ -206,8 +213,13 @@ C.open = function(allFields, openCallback0) {
       return;
     }
     self.serverProperties = start.fields.serverProperties;
-    send(defs.ConnectionStartOk);
-    wait(afterStartOk);
+    try {
+      send(defs.ConnectionStartOk);
+      wait(afterStartOk);
+    } catch (_) {
+      // Exit, callback with error already called,
+      return;
+    }
   }
 
   function afterStartOk(reply) {
@@ -228,8 +240,13 @@ C.open = function(allFields, openCallback0) {
         negotiate(fields.channelMax, allFields.channelMax);
       tunedOptions.heartbeat =
         negotiate(fields.heartbeat, allFields.heartbeat);
-      send(defs.ConnectionTuneOk);
-      send(defs.ConnectionOpen);
+      try {
+        send(defs.ConnectionTuneOk);
+        send(defs.ConnectionOpen);
+      } catch (_) {
+        // Exit, callback with error already called,
+        return;
+      }
       expect(defs.ConnectionOpenOk, onOpenOk);
       break;
     default:
@@ -257,6 +274,10 @@ C.open = function(allFields, openCallback0) {
   // If the server closes the connection, it's probably because of
   // something we did
   function endWhileOpening(err) {
+    self.stream.removeListener('end', endWhileOpening);
+    self.stream.removeListener('error', endWhileOpening);
+    self.stream.end();
+
     bail(err || new Error('Socket closed abruptly ' +
                           'during opening handshake'));
   }