Fix IOException when Console window unavailable (#2237)

src/BenchmarkDotNet/Running/BenchmarkRunnerClean.cs

@@ -1,4 +1,4 @@
-using System;
+using System;
 using System.Collections.Generic;
 using System.Collections.Immutable;
 using System.IO;
@@ -17,7 +17,6 @@
 using BenchmarkDotNet.Jobs;
 using BenchmarkDotNet.Loggers;
 using BenchmarkDotNet.Mathematics;
-using BenchmarkDotNet.Portability;
 using BenchmarkDotNet.Reports;
 using BenchmarkDotNet.Toolchains;
 using BenchmarkDotNet.Toolchains.Parameters;
@@ -168,14 +167,14 @@ private static Summary Run(BenchmarkRunInfo benchmarkRunInfo,
             var cultureInfo = config.CultureInfo ?? DefaultCultureInfo.Instance;
             var reports = new List<BenchmarkReport>();
             string title = GetTitle(new[] { benchmarkRunInfo });
-            var consoleTitle = RuntimeInformation.IsWindows() ? Console.Title : string.Empty;
+            using var consoleTitler = new ConsoleTitler("");
 
             logger.WriteLineInfo($"// Found {benchmarks.Length} benchmarks:");
             foreach (var benchmark in benchmarks)
                 logger.WriteLineInfo($"//   {benchmark.DisplayInfo}");
             logger.WriteLine();
 
-            UpdateTitle(totalBenchmarkCount, benchmarksToRunCount);
+            UpdateTitle(totalBenchmarkCount, benchmarksToRunCount, consoleTitler);
 
             using (var powerManagementApplier = new PowerManagementApplier(logger))
             {
@@ -241,15 +240,10 @@ private static Summary Run(BenchmarkRunInfo benchmarkRunInfo,
 
                     benchmarksToRunCount -= stop ? benchmarks.Length - i : 1;
 
-                    LogProgress(logger, in runsChronometer, totalBenchmarkCount, benchmarksToRunCount, taskbarProgress);
+                    LogProgress(logger, in runsChronometer, totalBenchmarkCount, benchmarksToRunCount, consoleTitler, taskbarProgress);
                 }
             }
 
-            if (RuntimeInformation.IsWindows())
-            {
-                Console.Title = consoleTitle;
-            }
-
             var runEnd = runsChronometer.GetElapsed();
 
             return new Summary(title,
@@ -652,15 +646,12 @@ private static void Cleanup(HashSet<string> artifactsToCleanup)
             }
         }
 
-        private static void UpdateTitle(int totalBenchmarkCount, int benchmarksToRunCount)
+        private static void UpdateTitle(int totalBenchmarkCount, int benchmarksToRunCount, ConsoleTitler consoleTitler)
         {
-            if (!Console.IsOutputRedirected && (RuntimeInformation.IsWindows() || RuntimeInformation.IsLinux() || RuntimeInformation.IsMacOS()))
-            {
-                Console.Title = $"{benchmarksToRunCount}/{totalBenchmarkCount} Remaining";
-            }
+            consoleTitler.UpdateTitle(() => $"{benchmarksToRunCount}/{totalBenchmarkCount} Remaining");
         }
 
-        private static void LogProgress(ILogger logger, in StartedClock runsChronometer, int totalBenchmarkCount, int benchmarksToRunCount, TaskbarProgress taskbarProgress)
+        private static void LogProgress(ILogger logger, in StartedClock runsChronometer, int totalBenchmarkCount, int benchmarksToRunCount, ConsoleTitler consoleTitler, TaskbarProgress taskbarProgress)
         {
             int executedBenchmarkCount = totalBenchmarkCount - benchmarksToRunCount;
             TimeSpan fromNow = GetEstimatedFinishTime(runsChronometer, benchmarksToRunCount, executedBenchmarkCount);
@@ -669,10 +660,8 @@ private static void LogProgress(ILogger logger, in StartedClock runsChronometer,
                 $" Estimated finish {estimatedEnd:yyyy-MM-dd H:mm} ({(int)fromNow.TotalHours}h {fromNow.Minutes}m from now) **";
             logger.WriteLineHeader(message);
 
-            if (!Console.IsOutputRedirected && (RuntimeInformation.IsWindows() || RuntimeInformation.IsLinux() || RuntimeInformation.IsMacOS()))
-            {
-                Console.Title = $"{benchmarksToRunCount}/{totalBenchmarkCount} Remaining - {(int)fromNow.TotalHours}h {fromNow.Minutes}m to finish";
-            }
+            consoleTitler.UpdateTitle (() => $"{benchmarksToRunCount}/{totalBenchmarkCount} Remaining - {(int)fromNow.TotalHours}h {fromNow.Minutes}m to finish");
+
             taskbarProgress.SetProgress((float) executedBenchmarkCount / totalBenchmarkCount);
         }
 

src/BenchmarkDotNet/Running/ConsoleTitler.cs

@@ -0,0 +1,87 @@
+using System;
+using System.IO;
+using BenchmarkDotNet.Portability;
+
+namespace BenchmarkDotNet.Running
+{
+    /// <summary>
+    /// Updates Console.Title, subject to platform capabilities and Console availability.
+    /// Restores the original (or fallback) title upon disposal.
+    /// </summary>
+    internal class ConsoleTitler : IDisposable
+    {
+        /// <summary>
+        /// Whether this instance has any effect. This will be false if the platform doesn't support Console retitling,
+        /// or if Console output is redirected.
+        /// </summary>
+        public bool IsEnabled { get; private set; }
+
+        private string oldConsoleTitle;
+
+        /// <summary>
+        /// Constructs a ConsoleTitler
+        /// </summary>
+        /// <param name="fallbackTitle">What to restore Console.Title to upon disposal (for platforms with write-only Console.Title)</param>
+        public ConsoleTitler(string fallbackTitle)
+        {
+            // Return without enabling if Console output is redirected, or if we're not on a platform that supports Console retitling.
+            if (Console.IsOutputRedirected || !PlatformSupportsTitleWrite())
+            {
+                return;
+            }
+
+            try
+            {
+                oldConsoleTitle = PlatformSupportsTitleRead() ? Console.Title : fallbackTitle;
+                IsEnabled = true;
+            }
+            catch (IOException)
+            {
+                // We're unable to read Console.Title on a platform that supports it. This can happen when no console
+                // window is available due to the application being Windows Forms, WPF, Windows Service or a daemon.
+                // Because we won't be able to write Console.Title either, return without enabling the titler.
+            }
+        }
+
+#if NET6_0_OR_GREATER
+        [System.Runtime.Versioning.SupportedOSPlatformGuard("windows")]
+#endif
+        private static bool PlatformSupportsTitleRead() => RuntimeInformation.IsWindows();
+
+        private static bool PlatformSupportsTitleWrite() =>
+            RuntimeInformation.IsWindows() ||
+            RuntimeInformation.IsLinux() ||
+            RuntimeInformation.IsMacOS();
+
+        /// <summary>
+        /// Updates Console.Title if enabled.
+        /// </summary>
+        public void UpdateTitle(string title)
+        {
+            if (IsEnabled)
+            {
+                Console.Title = title;
+            }
+        }
+
+        /// <summary>
+        /// Updates Console.Title if enabled, using a Func to avoid potential string-building cost.
+        /// </summary>
+        public void UpdateTitle(Func<string> title)
+        {
+            if (IsEnabled)
+            {
+                Console.Title = title();
+            }
+        }
+
+        public void Dispose()
+        {
+            if (IsEnabled && oldConsoleTitle != null)
+            {
+                Console.Title = oldConsoleTitle;
+                IsEnabled = false;
+            }
+        }
+    }
+}