fix crash on Windows logout

Author: gabsoftware

GabNetStats/MainForm.cs

@@ -183,6 +183,23 @@ public MainForm()
             InitializeComponent();
         }
 
+        protected override void WndProc(ref Message m)
+        {
+            const int WM_QUERYENDSESSION = 0x0011;
+            const int WM_ENDSESSION = 0x0016;
+
+            if (m.Msg == WM_QUERYENDSESSION || m.Msg == WM_ENDSESSION)
+            {
+                Program.IsWindowsShuttingDown = true;
+                // Signal threads to stop and save state cleanly
+                try { workerCancellationTokenSource.Cancel(); } catch { }
+                Settings.Default.Save();
+                m.Result = (IntPtr)1; // Tell Windows: OK to shut down
+                return;
+            }
+            base.WndProc(ref m);
+        }
+
         private void applyIconSet()
         {
             string desiredSet = Settings.Default.IconSet;

GabNetStats/Program.cs

@@ -11,6 +11,9 @@ static class Program
         /// </summary>
         private static Mutex m_Mutex;
 
+        /// <summary>Set to true when Windows initiates a shutdown, to suppress restart on crash.</summary>
+        public static bool IsWindowsShuttingDown { get; set; }
+
         [STAThread]
         static void Main()
         {
@@ -30,16 +33,16 @@ static void Main()
             }
             catch (Exception ex)
             {
-                if (ex.GetType() != typeof(ThreadAbortException))
+                if (!IsWindowsShuttingDown && ex.GetType() != typeof(ThreadAbortException))
                 {
                     MessageBox.Show(
                         Res.str_ErrorCrash +
                         "\n\n" + "Thread : " +
                         Thread.CurrentThread.Name +
                         "\n\n" +
                         ex.ToString(), "GabNetStats", MessageBoxButtons.OK, MessageBoxIcon.Error);
+                    Application.Restart();
                 }
-                Application.Restart();
             }
         }
     }

GabNetStats/frmBalloon.cs

@@ -203,8 +203,12 @@ private void frmBalloon_FormClosing(object sender, FormClosingEventArgs e)
         {
             counter = 0;
             Settings.Default.Save();
-            e.Cancel = true;
-            this.Hide();
+            if (e.CloseReason == CloseReason.UserClosing)
+            {
+                e.Cancel = true;
+                this.Hide();
+            }
+            // For WindowsShutDown, ApplicationExitCall, etc., let it close
         }
 
         private void frmBalloon_VisibleChanged(object sender, EventArgs e)