servermonitor: ServerMonitor/Objects/ServerMonitor.cs annotate

annotate ServerMonitor/Objects/ServerMonitor.cs @ 25:781d8b980be1

Fix checks not getting scheduled when a server is enabled.

author	Brad Greco <brad@bgreco.net>
date	Thu, 30 May 2019 21:41:14 -0400
parents	68d7834dc28e
children	437442cd8090

rev	line source
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	1 using Microsoft.Win32;
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	2 using Renci.SshNet;
5 b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	3 using Renci.SshNet.Common;
b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	4 using ServerMonitorApp.Properties;
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	5 using System;
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	6 using System.Collections.Generic;
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	7 using System.IO;
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	8 using System.Linq;
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	9 using System.Net.NetworkInformation;
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	10 using System.Threading;
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	11 using System.Threading.Tasks;
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	12 using System.Windows.Forms;
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	13 using System.Xml.Serialization;
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	14
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	15 namespace ServerMonitorApp
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	16 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	17 /// <summary>Central class for scheduling and executing checks against remote servers.</summary>
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	18 public class ServerMonitor
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	19 {
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	20 private readonly string configFileDir;
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	21 private readonly Logger logger;
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	22 // Cancellation tokens for executing checks, keyed by check ID.
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	23 private readonly Dictionary<int, CancellationTokenSource> tokens = new Dictionary<int, CancellationTokenSource>();
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	24 // SSH private keys, keyed by the path to the private key file.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	25 // A value of NULL indicates that the private key is inaccessible or encrypted.
5 b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	26 private readonly Dictionary<string, PrivateKeyFile> privateKeys = new Dictionary<string, PrivateKeyFile>();
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	27 // IDs of all checks that have been paused due to network unavailability,
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	28 // or due to the system being suspended.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	29 // Not to be confused with checks that have been disabled by the user.
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	30 private readonly List<int> pausedChecks = new List<int>();
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	31 private bool running, networkAvailable, suspend;
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	32 // List of check execution tasks that have been started.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	33 // A check task begins by sleeping until the next scheduled execution time,
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	34 // then executes.
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	35 private Dictionary<Task<CheckResult>, int> tasks;
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	36 private ServerSummaryForm mainForm;
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	37
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	38 /// <summary>Fires when the status of a check changes.</summary>
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	39 public event EventHandler<CheckStatusChangedEventArgs> CheckStatusChanged;
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	40
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	41 /// <summary>The collection of registered servers.</summary>
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	42 public List<Server> Servers { get; private set; } = new List<Server>();
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	43
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	44 /// <summary>A collection of all checks belonging to all registerd servers.</summary>
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	45 public IEnumerable<Check> Checks => Servers.SelectMany(s => s.Checks);
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	46
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	47 /// <summary>Path to the file that stores server and check configuration.</summary>
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	48 public string ConfigFile { get; private set; }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	49
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	50 /// <summary>Path to the file that stores server and check configuration.</summary>
5 b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	51 public IEnumerable<string> LockedKeys { get { return privateKeys.Where(kvp => kvp.Value == null).Select(kvp => kvp.Key); } }
b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	52
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	53 /// <summary>ServerMonitor constructor.</summary>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	54 /// <param name="mainForm">A reference to the main form.</param>
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	55 public ServerMonitor(ServerSummaryForm mainForm)
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	56 {
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	57 this.mainForm = mainForm;
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	58 // Store configuration in %appdata%\ServerMonitor
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	59 configFileDir = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData), "ServerMonitor");
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	60 ConfigFile = Path.Combine(configFileDir, "servers.xml");
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	61 logger = new Logger(Path.Combine(configFileDir, "monitor.log"));
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	62 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	63
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	64 /// <summary>Registers a new server with the server monitor.</summary>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	65 /// <param name="server">The server to be added.</param>
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	66 public void AddServer(Server server)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	67 {
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	68 Servers.Add(server);
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	69 SaveServers();
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	70 }
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	71
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	72 /// <summary>Deletes a server from the server monitor.</summary>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	73 /// <param name="server">The server to be deleted.</param>
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	74 public void DeleteServer(Server server)
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	75 {
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	76 Servers.Remove(server);
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	77 SaveServers();
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	78 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	79
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	80 /// <summary>Loads all servers and checks from the config file.</summary>
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	81 public void LoadServers()
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	82 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	83 bool triedBackup = false;
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	84
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	85 Read:
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	86 TextReader reader = null;
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	87 try
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	88 {
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	89 reader = new StreamReader(ConfigFile);
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	90 XmlSerializer serializer = CreateXmlSerializer();
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	91 Servers.Clear();
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	92 Servers.AddRange((List<Server>)serializer.Deserialize(reader));
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	93 // Do some more set-up now that the servers and checks have been loaded.
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	94 foreach (Server server in Servers)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	95 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	96 // Read private keys into memory if they are accessible and not encrypted.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	97 // If PrivateKeyFile != null, it means same the key has already been loaded for
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	98 // a different server and nothing more needs to be done.
6 c1dffaac66fa - Don't show multiple password dialogs for the same key if the first one was cancelled. Brad Greco <brad@bgreco.net> parents: 5 diff changeset	99 if (server.LoginType == LoginType.PrivateKey && server.PrivateKeyFile == null)
5 b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	100 OpenPrivateKey(server.KeyFile);
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	101 foreach (Check check in server.Checks)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	102 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	103 // Update the checks so they know what server they belong to.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	104 // Would rather do this in the Server object on deserialization, but
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	105 // that doesn't work when using the XML serializer for some reason.
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	106 check.Server = server;
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	107 // If the program last exited while the check was running, change its status
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	108 // to the result of its last execution (since, at this point, the check is
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	109 // not running).
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	110 if (check.Status == CheckStatus.Running)
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	111 check.Status = check.LastRunStatus;
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	112 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	113 server.CheckModified += Server_CheckModified;
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	114 server.EnabledChanged += Server_EnabledChanged;
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	115 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	116 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	117 // If the file doesn't exist, no special handling is needed. It will be created later.
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	118 catch (FileNotFoundException) { }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	119 catch (DirectoryNotFoundException) { }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	120 catch (InvalidOperationException)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	121 {
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	122 reader?.Close();
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	123 // If there was an error parsing the XML, try again with the backup config file.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	124 if (!triedBackup)
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	125 {
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	126 File.Copy(ConfigFile, ConfigFile + ".error", true);
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	127 string backupConfig = ConfigFile + ".bak";
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	128 if (File.Exists(backupConfig))
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	129 {
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	130 File.Copy(backupConfig, ConfigFile, true);
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	131 }
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	132 triedBackup = true;
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	133 goto Read;
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	134 }
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	135 else
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	136 {
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	137 // If there was an error reading the backup file too, give up.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	138 throw;
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	139 }
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	140 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	141 finally
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	142 {
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	143 reader?.Close();
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	144 }
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	145 Application.ApplicationExit += Application_ApplicationExit;
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	146 NetworkChange.NetworkAddressChanged += NetworkChange_NetworkAddressChanged;
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	147 SystemEvents.PowerModeChanged += SystemEvents_PowerModeChanged;
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	148
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	149 // Remove old entries from the log file according to user preferences.
6 c1dffaac66fa - Don't show multiple password dialogs for the same key if the first one was cancelled. Brad Greco <brad@bgreco.net> parents: 5 diff changeset	150 logger.TrimLog();
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	151
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	152 Run();
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	153 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	154
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	155 /// <summary>Saves all servers and checks to the config file.</summary>
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	156 public void SaveServers()
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	157 {
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	158 GenerateIds();
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	159 TextWriter writer = null;
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	160 XmlSerializer serializer = null;
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	161 try
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	162 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	163 // Make a backup first in case something goes wrong in the middle of writing.
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	164 File.Copy(ConfigFile, ConfigFile + ".bak", true);
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	165 }
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	166 catch { }
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	167 try
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	168 {
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	169 writer = new StreamWriter(ConfigFile);
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	170 serializer = CreateXmlSerializer();
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	171 serializer.Serialize(writer, Servers);
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	172 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	173 catch (DirectoryNotFoundException)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	174 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	175 // If the directory does not exist, create it and try again.
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	176 Directory.CreateDirectory(configFileDir);
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	177 writer = new StreamWriter(ConfigFile);
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	178 serializer = CreateXmlSerializer();
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	179 serializer.Serialize(writer, Servers);
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	180 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	181 finally
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	182 {
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	183 writer?.Close();
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	184 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	185 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	186
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	187 /// <summary>Main server monitor loop. Schedules and executes checks.</summary>
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	188 private async void Run()
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	189 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	190 // Do not run again if already running or if the system is suspending or resuming.
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	191 if (running \|\| suspend)
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	192 return;
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	193
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	194 running = true;
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	195
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	196 // If the network is available, immediately execute checks that were supposed to run
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	197 // earlier but could not due to network unavailability or the system being suspended.
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	198 networkAvailable = Helpers.IsNetworkAvailable();
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	199 if (networkAvailable)
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	200 {
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	201 foreach (int id in pausedChecks)
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	202 {
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	203 await ExecuteCheckAsync(Checks.FirstOrDefault(c => c.Id == id));
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	204 }
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	205 pausedChecks.Clear();
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	206 }
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	207
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	208 // Schedule all checks to run according to their schedules.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	209 // Each check will sleep until it is scheduled to run, then execute.
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	210 tasks = Checks.ToDictionary(c => ScheduleExecuteCheckAsync(c), c => c.Id);
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	211 while (tasks.Count > 0)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	212 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	213 // When any check is done sleeping and executing, remove the completed
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	214 // task and queue a new task to schedule it again.
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	215 Task<CheckResult> task = await Task.WhenAny(tasks.Keys);
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	216 tasks.Remove(task);
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	217 try
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	218 {
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	219 CheckResult result = await task;
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	220 // Do not schedule the task again if it is now disabled.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	221 // Result will be null if a scheduled check was disabled.
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	222 if (result != null && result.CheckStatus != CheckStatus.Disabled)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	223 tasks.Add(ScheduleExecuteCheckAsync(result.Check), result.Check.Id);
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	224 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	225 catch (OperationCanceledException)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	226 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	227 // When a server's state changes to Disabled, any checks that are executing
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	228 // are immediately cancelled. Silently catch these expected exceptions.
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	229 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	230 }
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	231 // If there are no enabled checks scheduled, exit the main loop.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	232 // It will be restarted when a check or server is enabled.
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	233 running = false;
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	234 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	235
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	236 /// <summary>Schedules a check to be run on its schedule.</summary>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	237 /// <param name="check">The check to execute.</param>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	238 /// <returns>The async check result.</returns>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	239 private async Task<CheckResult> ScheduleExecuteCheckAsync(Check check)
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	240 {
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	241 // Do not schedule or execute the check if it or its server is disabled.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	242 if (!check.Enabled \|\| !check.Server.Enabled)
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	243 return await Task.FromResult(new CheckResult(check, CheckStatus.Disabled, null));
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	244
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	245 // Create a cancellation token that will be used to cancel the check if it or
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	246 // its server is disabled while it is executing.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	247 CancellationTokenSource cts = new CancellationTokenSource();
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	248 tokens[check.Id] = cts;
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	249
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	250 // Sleep until next time the check is supposed to be executed.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	251 // Use the LastScheduledRunTime so manual executions by the user do not
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	252 // interfere with the schedule.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	253 check.NextRunTime = check.Schedule.GetNextTime(check.LastScheduledRunTime);
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	254 int delay = Math.Max(0, (int)(check.NextRunTime - DateTime.Now).TotalMilliseconds);
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	255 await Task.Delay(delay, cts.Token);
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	256 check.LastScheduledRunTime = check.NextRunTime;
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	257
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	258 // Execute the check if not cancelled.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	259 if (!cts.IsCancellationRequested)
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	260 {
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	261 // If the network is available, execute the check.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	262 // Otherwise, add it to the list of paused checks to be executed
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	263 // when the network becomes available again.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	264 if (networkAvailable)
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	265 {
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	266 return await ExecuteCheckAsync(check, cts.Token);
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	267 }
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	268 else
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	269 {
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	270 if (!pausedChecks.Contains(check.Id))
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	271 pausedChecks.Add(check.Id);
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	272 }
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	273 }
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	274 return await Task.FromResult(new CheckResult(check, CheckStatus.Disabled, null));
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	275 }
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	276
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	277 /// <summary>Executes a check asynchronously.</summary>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	278 /// <param name="check">The check to execute.</param>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	279 /// <param name="token">A chancellation token that may be used to cancel the check execution.</param>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	280 /// <returns>The async check result.</returns>
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	281 public async Task<CheckResult> ExecuteCheckAsync(Check check, CancellationToken token = default(CancellationToken))
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	282 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	283 // Update the status.
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	284 check.Status = CheckStatus.Running;
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	285 OnCheckStatusChanged(check);
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	286
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	287 // Execute the check.
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	288 CheckResult result = await check.ExecuteAsync(token);
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	289
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	290 // Increment the consecutive failure counter on failue, or reset
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	291 // the counter on success.
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	292 if (result.Failed)
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	293 check.ConsecutiveFailures++;
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	294 else
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	295 check.ConsecutiveFailures = 0;
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	296
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	297 OnCheckStatusChanged(check, result);
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	298 HandleResultAsync(result);
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	299 return result;
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	300 }
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	301
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	302 /// <summary>Handles the result of a check execution.</summary>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	303 /// <param name="result">The result.</param>
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	304 private void HandleResultAsync(CheckResult result)
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	305 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	306 // Log the result.
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	307 logger.Log(result);
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	308
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	309 // Notify the user of failure according to user preferences.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	310 // If the check succeeded, result.FailAction will be None.
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	311 if (result.Check.ConsecutiveFailures >= result.Check.MaxConsecutiveFailures)
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	312 {
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	313 if (result.FailAction == FailAction.FlashTaskbar)
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	314 mainForm.AlertServerForm(result.Check);
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	315 if (result.FailAction.In(FailAction.FlashTaskbar, FailAction.NotificationBalloon))
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	316 mainForm.ShowBalloon(result);
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	317 }
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	318 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	319
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	320 /// <summary>Reads all check results from the log for a server.</summary>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	321 /// <param name="server">The server whose check results should be read.</param>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	322 /// <returns>A list of all check results found in the log file for the given server.</returns>
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	323 public IList<CheckResult> GetLog(Server server)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	324 {
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	325 return logger.Read(server);
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	326 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	327
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	328 /// <summary>Saves the check settings and notifies event subscribers when the status of a check changes.</summary>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	329 /// <param name="check">The check whose status has changed.</param>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	330 /// <param name="result">The check result that caused the status to change, if any.</param>
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	331 private void OnCheckStatusChanged(Check check, CheckResult result = null)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	332 {
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	333 SaveServers();
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	334 CheckStatusChanged?.Invoke(check, new CheckStatusChangedEventArgs(check, result));
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	335 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	336
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	337 /// <summary>Handles user modifications to a check's settings.</summary>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	338 /// <param name="sender">The check that was modified.</param>
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	339 private void Server_CheckModified(object sender, EventArgs e)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	340 {
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	341 Check check = (Check)sender;
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	342
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	343 // No need to mess with the task queue if not currently running.
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	344 if (running)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	345 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	346 Task<CheckResult> task = tasks.FirstOrDefault(kvp => kvp.Value == check.Id).Key;
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	347 if (task == null)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	348 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	349 // No tasks associated with the check, so schedule a new one.
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	350 tasks.Add(ScheduleExecuteCheckAsync(check), check.Id);
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	351 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	352 else
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	353 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	354 // Check was modified or deleted, so remove any waiting tasks.
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	355 CancelCheck(check);
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	356 if (check.Server != null)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	357 {
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	358 // If the check was not deleted, schedule the new check.
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	359 // But only if it's still running, otherwise restarting the monitor below
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	360 // will create a duplicate run.
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	361 if (running)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	362 tasks.Add(ScheduleExecuteCheckAsync(check), check.Id);
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	363 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	364 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	365 }
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	366 // Run again in case removing a task above caused it to stop.
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	367 Run();
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	368 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	369
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	370 /// <summary>Handles the enabled state of a server changing.</summary>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	371 /// <param name="sender">The server that was enabled or disabled.</param>
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	372 private void Server_EnabledChanged(object sender, EventArgs e)
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	373 {
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	374 Server server = (Server)sender;
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	375
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	376 // Make sure the monitor is running. If no servers were enabled before this
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	377 // one was enabled, it is not running.
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	378 if (server.Enabled)
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	379 {
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	380 Run();
25 781d8b980be1 Fix checks not getting scheduled when a server is enabled. Brad Greco <brad@bgreco.net> parents: 17 diff changeset	381 // Schedule all checks to run.
781d8b980be1 Fix checks not getting scheduled when a server is enabled. Brad Greco <brad@bgreco.net> parents: 17 diff changeset	382 foreach (Check check in server.Checks)
781d8b980be1 Fix checks not getting scheduled when a server is enabled. Brad Greco <brad@bgreco.net> parents: 17 diff changeset	383 {
781d8b980be1 Fix checks not getting scheduled when a server is enabled. Brad Greco <brad@bgreco.net> parents: 17 diff changeset	384 Server_CheckModified(check, new EventArgs());
781d8b980be1 Fix checks not getting scheduled when a server is enabled. Brad Greco <brad@bgreco.net> parents: 17 diff changeset	385 }
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	386 }
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	387 else
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	388 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	389 // Cancel all queued and executing checks belonging to a
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	390 // server that was disabled.
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	391 foreach (Check check in server.Checks)
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	392 {
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	393 CancelCheck(check);
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	394 }
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	395 }
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	396 }
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	397
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	398 /// <summary>Cancels a check that may be executing.</summary>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	399 /// <param name="check">The check to cancel.</param>
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	400 private void CancelCheck(Check check)
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	401 {
6 c1dffaac66fa - Don't show multiple password dialogs for the same key if the first one was cancelled. Brad Greco <brad@bgreco.net> parents: 5 diff changeset	402 if (tasks == null)
c1dffaac66fa - Don't show multiple password dialogs for the same key if the first one was cancelled. Brad Greco <brad@bgreco.net> parents: 5 diff changeset	403 return;
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	404
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	405 // Find the waiting or executing task for the check and remove it.
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	406 Task<CheckResult> task = tasks.FirstOrDefault(kvp => kvp.Value == check.Id).Key;
5 b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	407 if (task != null)
b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	408 tasks.Remove(task);
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	409
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	410 // Remove it from the list of paused checks so it doesn't get restarted later.
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	411 pausedChecks.RemoveAll(id => id == check.Id);
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	412
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	413 // Cancel the current execution.
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	414 if (tokens.TryGetValue(check.Id, out CancellationTokenSource cts))
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	415 cts.Cancel();
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	416 }
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	417
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	418 /// <summary>Handles network state changing.</summary>
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	419 private void NetworkChange_NetworkAddressChanged(object sender, EventArgs e)
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	420 {
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	421 networkAvailable = Helpers.IsNetworkAvailable();
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	422 // If the network is available, it might not have been before.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	423 // This method is not called from the correct thread, so special
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	424 // handling is needed to start it on the UI thread again.
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	425 if (networkAvailable)
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	426 mainForm.Invoke((MethodInvoker)(() => Run()));
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	427 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	428
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	429 /// <summary>Handles system power mode changes.</summary>
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	430 private async void SystemEvents_PowerModeChanged(object sender, PowerModeChangedEventArgs e)
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	431 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	432 // If the system is being suspended, cancel all waiting and executing checks.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	433 // Once all the checks are removed, the main loop will exit.
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	434 if (e.Mode == PowerModes.Suspend)
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	435 {
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	436 foreach (Check check in Checks)
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	437 {
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	438 CancelCheck(check);
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	439 }
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	440 suspend = true;
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	441 }
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	442 else if (e.Mode == PowerModes.Resume)
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	443 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	444 // When resuming from suspend, examine each check to find out if it was
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	445 // scheduled to be executed during the time period when the systems was
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	446 // suspended. Add them to the paused checks list, to be executed almost
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	447 // immediately.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	448 // Make sure the list is empty to start.
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	449 pausedChecks.Clear();
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	450 foreach (Check check in Checks)
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	451 {
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	452 if (check.Enabled && check.Server.Enabled && check.NextRunTime < DateTime.Now)
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	453 {
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	454 pausedChecks.Add(check.Id);
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	455 }
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	456 }
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	457 // Wait 10 seconds to give things time to quiet down after resuming.
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	458 await Task.Delay(10000);
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	459 suspend = false;
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	460 Run();
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	461 }
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	462 }
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	463
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	464 /// <summary>Unregister system events when exiting.</summary>
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	465 private void Application_ApplicationExit(object sender, EventArgs e)
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	466 {
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	467 NetworkChange.NetworkAddressChanged -= NetworkChange_NetworkAddressChanged;
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	468 SystemEvents.PowerModeChanged -= SystemEvents_PowerModeChanged;
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	469 }
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	470
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	471 /// <summary>Attempts to read a private file.</summary>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	472 /// <param name="path">The path to the private key file.</param>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	473 /// <param name="password">The password used to encrypt the key.</param>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	474 /// <returns>A status indicating the result of the attempt.</returns>
5 b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	475 public KeyStatus OpenPrivateKey(string path, string password = null)
b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	476 {
b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	477 KeyStatus keyStatus;
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	478
5 b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	479 if (path == null)
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	480 keyStatus = KeyStatus.NotAccessible;
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	481
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	482 // Check if the key has already been open and read.
5 b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	483 if (privateKeys.TryGetValue(path, out PrivateKeyFile key) && key != null)
b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	484 keyStatus = KeyStatus.Open;
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	485 else
5 b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	486 {
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	487 try
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	488 {
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	489 key = new PrivateKeyFile(path, password);
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	490 keyStatus = KeyStatus.Open;
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	491 }
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	492 // If the key is encrypted and the password is empty or incorrect,
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	493 // return the NeedPassword status.
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	494 catch (Exception e) when (e is SshPassPhraseNullOrEmptyException \|\| e is InvalidOperationException)
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	495 {
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	496 keyStatus = KeyStatus.NeedPassword;
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	497 }
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	498 // For any other failure reason, return the NotAccessible status.
8 052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	499 catch (Exception)
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	500 {
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	501 keyStatus = KeyStatus.NotAccessible;
052aa62cb42a Single instance. Add autorun option. Add icons. General enhancements. Brad Greco <brad@bgreco.net> parents: 6 diff changeset	502 }
5 b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	503 }
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	504 // A single private key may be used by multiple servers. Update all servers
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	505 // that use this private key with the results of the above operations.
5 b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	506 foreach (Server server in Servers)
b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	507 {
b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	508 if (server.KeyFile == path)
b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	509 {
b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	510 server.PrivateKeyFile = key;
b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	511 server.KeyStatus = keyStatus;
b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	512 }
b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	513 }
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	514 // Keep a reference to this private key so we don't have to re-open
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	515 // it later if the same key is used on a different server.
5 b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	516 privateKeys[path] = key;
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	517
5 b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	518 return keyStatus;
b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	519 }
b6fe203af9d5 Private key passwords and validation Brad Greco <brad@bgreco.net> parents: 4 diff changeset	520
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	521 /// <summary>Generates internal IDs for servers and checks.</summary>
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	522 private void GenerateIds()
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	523 {
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	524 if (Servers.Any())
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	525 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	526 // Start at the maximum ID to make sure IDs are not reused
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	527 // if a server was deleted so old log entries do not get associated
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	528 // with a new server.
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	529 int id = Servers.Max(s => s.Id);
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	530 foreach (Server server in Servers)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	531 {
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	532 if (server.Id == 0)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	533 server.Id = ++id;
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	534 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	535 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	536
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	537 if (Checks.Any())
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	538 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	539 // Start with the max check ID, same reasons as above.
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	540 // Is there a reason this is stored in a setting?
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	541 int id = Math.Max(Settings.Default.MaxCheckId, Checks.Max(c => c.Id));
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	542 foreach (Check check in Checks)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	543 {
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	544 if (check.Id == 0)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	545 check.Id = ++id;
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	546 }
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	547 Settings.Default.MaxCheckId = id;
3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	548 Settings.Default.Save();
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	549 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	550 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	551
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	552 /// <summary>Creates an XML serializer that can handle servers and all check types.</summary>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	553 /// <returns>An XML serializer that can handle servers and all check types.</returns>
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	554 private XmlSerializer CreateXmlSerializer()
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	555 {
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	556 return new XmlSerializer(typeof(List<Server>), Check.CheckTypes);
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	557 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	558 }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	559
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	560 /// <summary>Event arguments for when a check status changes.</summary>
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	561 public class CheckStatusChangedEventArgs : EventArgs
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	562 {
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	563 /// <summary>The check whose status changed.</summary>
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	564 public Check Check { get; private set; }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	565
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	566 /// <summary>The check result that caused the status to change, if any.</summary>
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	567 public CheckResult CheckResult { get; private set; }
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	568
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	569 public CheckStatusChangedEventArgs(Check check, CheckResult result)
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	570 {
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	571 Check = check;
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	572 CheckResult = result;
3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	573 }
4 3142e52cbe69 Lots more progress Brad Greco <brad@bgreco.net> parents: 0 diff changeset	574 }
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	575
17 68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	576 /// <summary>Possible actions that may be taken when a check fails.</summary>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	577 public enum FailAction
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	578 {
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	579 /// <summary>Flashes the Server Monitor tasbar program icon.</summary>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	580 FlashTaskbar = 0,
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	581 /// <summary>Shows a balloon in the notification area.</summary>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	582 NotificationBalloon = 1,
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	583 /// <summary>Take no action.</summary>
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	584 None = 10
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	585 }
68d7834dc28e More comments. Brad Greco <brad@bgreco.net> parents: 8 diff changeset	586
0 3e1a2131f897 Initial commit. Ping check, scheduling, UI working. SSH check mostly working. Brad Greco <brad@bgreco.net> parents: diff changeset	587 }

Mercurial > servermonitor

annotate ServerMonitor/Objects/ServerMonitor.cs @ 25:781d8b980be1