Working through ssh disconnect / leak problems

Pepperdash Core/Pepperdash Core/Comm/GenericSshClient.cs

@@ -26,7 +26,7 @@ namespace PepperDash.Core
 	//*****************************************************************************************************
 	//*****************************************************************************************************
 
-	public class GenericSshClient : Device, IBasicCommunication
+	public class GenericSshClient : Device, IBasicCommunication, IAutoReconnect
 	{
 		public event EventHandler<GenericCommMethodReceiveBytesArgs> BytesReceived;
 		public event EventHandler<GenericCommMethodReceiveTextArgs> TextReceived;
@@ -66,12 +66,12 @@ namespace PepperDash.Core
 		/// <summary>
 		/// Millisecond value, determines the timeout period in between reconnect attempts
 		/// </summary>
-		public ushort AutoReconnectIntervalMs { get; set; }
+		public int AutoReconnectIntervalMs { get; set; }
 
 		SshClient Client;
 		ShellStream TheStream;
 		CTimer ReconnectTimer;
-		bool ReconnectTimerRunning;
+		//bool ReconnectTimerRunning;
 
 		public GenericSshClient(string key, string hostname, int port, string username, string password) :
 			base(key)
@@ -82,6 +82,21 @@ namespace PepperDash.Core
 			Port = port;
 			Username = username;
 			Password = password;
+			CrestronEnvironment.ProgramStatusEventHandler += new ProgramStatusEventHandler(CrestronEnvironment_ProgramStatusEventHandler);
+		}
+
+		void CrestronEnvironment_ProgramStatusEventHandler(eProgramStatusEventType programEventType)
+		{
+			if (programEventType == eProgramStatusEventType.Stopping)
+			{
+				if (Client != null)
+				{
+					Debug.Console(2, this, "Closing connection");
+					Client.Disconnect();
+					Client.Dispose();
+					Debug.Console(2, this, "Connection closed");
+				}
+			}
 		}
 
 		/// <summary>
@@ -89,13 +104,22 @@ namespace PepperDash.Core
 		/// </summary>
 		public void Connect()
 		{
-			ReconnectTimerRunning = false;
+			Debug.Console(1, this, "attempting connect, IsConnected={0}", Client != null ? Client.IsConnected : false);
+			
+			//ReconnectTimerRunning = false;
+			if (ReconnectTimer != null)
+			{
+				ReconnectTimer.Stop();
+				ReconnectTimer = null;
+			}
+
+			if (IsConnected)
+				return;
+
 			if (Hostname != null && Hostname != string.Empty && Port > 0 &&
 				Username != null && Password != null)
 			{
-				Debug.Console(1, this, "attempting connect, IsConnected={0}", IsConnected);
+
-				if (!IsConnected)
-				{
 				UStatus = 1;
 				IsConnected = false;
 
@@ -104,7 +128,15 @@ namespace PepperDash.Core
 				kauth.AuthenticationPrompt += new EventHandler<AuthenticationPromptEventArgs>(kauth_AuthenticationPrompt);
 				PasswordAuthenticationMethod pauth = new PasswordAuthenticationMethod(Username, Password);
 				ConnectionInfo connectionInfo = new ConnectionInfo(Hostname, Port, Username, pauth, kauth);
+				// always spin up new client in case parameters have changed
+				if (Client != null)
+				{
+					Client.Disconnect();
+					Client = null;
+					//Client.Dispose();
+				}				
 				Client = new SshClient(connectionInfo);
+				
 				Client.ErrorOccurred += Client_ErrorOccurred;
 				try
 				{
@@ -117,8 +149,6 @@ namespace PepperDash.Core
 						Debug.Console(1, this, "Connected");
 						TheStream = Client.CreateShellStream("PDTShell", 100, 80, 100, 200, 65534);
 						TheStream.DataReceived += Stream_DataReceived;						
-							//TheStream.ErrorOccurred += Stream_ErrorOccurred;
-							
 					}
 					return;
 				}
@@ -142,7 +172,6 @@ namespace PepperDash.Core
 						Debug.Console(0, this, "Error on connect:\r({0})", e);
 				}	
 			}
-			}
 			else
 			{
 				Debug.Console(0, this, "Connect failed.  Check hostname, port, username and password are set or not null");
@@ -160,7 +189,11 @@ namespace PepperDash.Core
 		public void Disconnect()
 		{
 			// Stop trying reconnects, if we are
-			if(ReconnectTimer != null) ReconnectTimer.Stop();
+			if (ReconnectTimer != null)
+			{
+				ReconnectTimer.Stop();
+				ReconnectTimer = null;
+			}
 			// Otherwise just close up
 			if (Client != null) // && Client.IsConnected) <-- Doesn't always report properly...
 			{
@@ -181,10 +214,13 @@ namespace PepperDash.Core
 				AutoReconnect, AutoReconnectIntervalMs);
 			if (AutoReconnect)
 			{
-				if (ReconnectTimer == null || !ReconnectTimerRunning)
+				if (ReconnectTimer == null)// || !ReconnectTimerRunning)
 				{
-					ReconnectTimer = new CTimer(o => Connect(), AutoReconnectIntervalMs);
+					ReconnectTimer = new CTimer(o =>
-					ReconnectTimerRunning = true;
+					{
+						Connect();
+						ReconnectTimer = null;
+					}, AutoReconnectIntervalMs);
 					Debug.Console(1, this, "Attempting connection in {0} seconds",
 						(float)(AutoReconnectIntervalMs / 1000));
 				}
@@ -251,6 +287,8 @@ namespace PepperDash.Core
 				Debug.Console(0, this, "SSH client error: {0}", e.Exception);
 				UStatus = 4;
 			}
+			Client.Disconnect();
+			Client = null;
 			Debug.Console(1, this, "Disconnected by remote");
 			IsConnected = false;
 			HandleConnectionFailure();

Pepperdash Core/Pepperdash Core/Comm/GenericTcpIpClient.cs

@@ -11,7 +11,7 @@ using Newtonsoft.Json.Linq;
 
 namespace PepperDash.Core
 {
-	public class GenericTcpIpClient : Device, IBasicCommunication
+	public class GenericTcpIpClient : Device, IBasicCommunication, IAutoReconnect
 	{
 		public event EventHandler<GenericCommMethodReceiveBytesArgs> BytesReceived;
 		public event EventHandler<GenericCommMethodReceiveTextArgs> TextReceived;
@@ -21,6 +21,9 @@ namespace PepperDash.Core
 		public string Status { get { return Client.ClientStatus.ToString(); } }
 		public string ConnectionFailure { get { return Client.ClientStatus.ToString(); } }
 
+		public bool AutoReconnect { get; set; }
+		public int AutoReconnectIntervalMs { get; set; }
+
 		public bool Connected
 		{
 			get { return Client.ClientStatus == SocketStatus.SOCKET_STATUS_CONNECTED; }
@@ -173,9 +176,21 @@ namespace PepperDash.Core
 		/// </summary>
 		public int BufferSize { get; set; }
 
+		/// <summary>
+		/// Defaults to true
+		/// </summary>
+		public bool AutoReconnect { get; set; }
+
+		/// <summary>
+		/// Defaults to 5000ms
+		/// </summary>
+		public int AutoReconnectIntervalMs { get; set; }
+
 		public TcpIpConfig()
 		{
 			BufferSize = 32768;
+			AutoReconnect = true;
+			AutoReconnectIntervalMs = 5000;
 		}
 	}
 

Pepperdash Core/Pepperdash Core/CommunicationExtras.cs

@@ -23,6 +23,12 @@ namespace PepperDash.Core
 		void Connect();
 	}
 
+	public interface IAutoReconnect
+	{
+		bool AutoReconnect { get; set; }
+		int AutoReconnectIntervalMs { get; set; }
+	}
+
 	/// <summary>
 	/// 
 	/// </summary>

Pepperdash Core/Pepperdash Core/PepperDash_Core.csproj

@@ -82,7 +82,7 @@
         <Programmer />
         <ArchiveFilename>C:\Users\hvolm\Desktop\working\pepperdash-simplsharp-core\Pepperdash Core\Pepperdash Core\bin\PepperDash_Core.clz</ArchiveFilename>
         <MinFirmwareVersion>1.007.0017</MinFirmwareVersion>
-        <CompiledOn>8/3/2016 4:33:09 PM</CompiledOn>
+        <CompiledOn>8/4/2016 12:58:52 PM</CompiledOn>
         <AdditionalInfo />
         <EmbedSourceArchive>False</EmbedSourceArchive>
         <CopyTo />

Pepperdash Core/Pepperdash Core/bin/PepperDash_Core.config

@@ -10,7 +10,7 @@
     <ArchiveName />
   </RequiredInfo>
   <OptionalInfo>
-    <CompiledOn>8/3/2016 4:33:09 PM</CompiledOn>
+    <CompiledOn>8/4/2016 12:58:52 PM</CompiledOn>
-    <CompilerRev>1.0.0.27993</CompilerRev>
+    <CompilerRev>1.0.0.21565</CompilerRev>
   </OptionalInfo>
 </ProgramInfo>

Pepperdash Core/Pepperdash Core/bin/manifest.info

@@ -1,4 +1,4 @@
-MainAssembly=PepperDash_Core.dll:ca423b7b3c2fe76c3d6a8b9cb9ff67cd
+MainAssembly=PepperDash_Core.dll:1e2af81f50d343b460b7e2684b9db8fe
 MainAssemblyMinFirmwareVersion=1.007.0017
 ü
 DependencySource=Newtonsoft.Json.Compact.dll:ea996aa2ec65aa1878e7c9d09e37a896