Better disconnect - should resolve issues with SSH

core/client.js

@@ -511,7 +511,12 @@ Client.prototype.end = function () {
         //
         //  We can end up calling 'end' before TTY/etc. is established, e.g. with SSH
         //
+        if(_.isFunction(this.disconnect)) {
+            return this.disconnect();
+        } else {
+            //  legacy fallback
             return this.output.end.apply(this.output, arguments);
+        }
     } catch(e) {
         //  ie TypeError
     }

core/servers/login/ssh.js

@@ -278,13 +278,17 @@ function SSHClient(clientConn) {
         });
     });
 
-    clientConn.on('end', () => {
+    clientConn.once('end', () => {
-        self.emit('end');   //  remove client connection/tracking
+        return self.emit('end');   //  remove client connection/tracking
     });
 
     clientConn.on('error', err => {
         self.log.warn( { error : err.message, code : err.code }, 'SSH connection error');
     });
+
+    this.disconnect = function() {
+        return clientConn.end();
+    };
 }
 
 util.inherits(SSHClient, baseClient.Client);

core/servers/login/telnet.js

@@ -550,6 +550,15 @@ function TelnetClient(input, output) {
             this.emit('ready', { firstMenu : Config().loginServers.telnet.firstMenu } );
         }
     };
+
+    this.disconnect = function() {
+        try {
+            return this.output.end.apply(this.output, arguments);
+        }
+        catch(e) {
+            //  nothing
+        }
+    };
 }
 
 util.inherits(TelnetClient, baseClient.Client);