任务管理器状态恢复机制

Author: JiuLing-zhang

src/ComputerLock/Platforms/TaskManagerHook.cs

@@ -1,33 +1,94 @@
 using Microsoft.Win32;
+using System.IO;
 
 namespace ComputerLock.Platforms;
 internal class TaskManagerHook
 {
+    private static readonly string OriginalStateFilePath = Path.Combine(Path.GetTempPath(), $"{AppBase.FriendlyName}_OriginalTaskMgr");
     private const string RegKey = @"Software\Microsoft\Windows\CurrentVersion\Policies\System";
+    private const string DisableTaskMgrValueName = "DisableTaskMgr";
+
+    private readonly ILogger _logger;
+
     private enum TaskManagerStateEnum
     {
         Enabled = 0,
         Disabled = 1
     }
 
-    public void EnabledTaskManager()
+    public TaskManagerHook(ILogger logger)
     {
-        SetState(TaskManagerStateEnum.Enabled);
+        _logger = logger;
     }
 
-    public void DisabledTaskManager()
+    public void Lock()
     {
+        var currentState = GetCurrentState();
+        SaveOriginalState(currentState);
         SetState(TaskManagerStateEnum.Disabled);
     }
 
+    public void Unlock()
+    {
+        try
+        {
+            string content = File.ReadAllText(OriginalStateFilePath);
+            if (Enum.TryParse<TaskManagerStateEnum>(content, out var originalState))
+            {
+                SetState(originalState);
+            }
+
+            File.Delete(OriginalStateFilePath);
+        }
+        catch (Exception ex)
+        {
+            _logger.Error("恢复任务管理器状态失败", ex);
+        }
+    }
+
+    /// <summary>
+    /// 崩溃恢复
+    /// </summary>
+    public void RecoverFromCrash()
+    {
+        if (File.Exists(OriginalStateFilePath))
+        {
+            Unlock();
+        }
+    }
+
+    private TaskManagerStateEnum GetCurrentState()
+    {
+        using var registryKey = Registry.CurrentUser.OpenSubKey(RegKey);
+        if (registryKey == null)
+        {
+            return TaskManagerStateEnum.Enabled;
+        }
+
+        var value = registryKey.GetValue(DisableTaskMgrValueName, 0);
+        return (int)value == 1 ? TaskManagerStateEnum.Disabled : TaskManagerStateEnum.Enabled;
+    }
+
+    private void SaveOriginalState(TaskManagerStateEnum state)
+    {
+        try
+        {
+            File.WriteAllText(OriginalStateFilePath, ((int)state).ToString());
+        }
+        catch (Exception ex)
+        {
+            _logger.Error($"保存任务管理器原始状态失败", ex);
+        }
+    }
+
     private void SetState(TaskManagerStateEnum state)
     {
-        var registryKey = Registry.CurrentUser.CreateSubKey(RegKey);
+        using var registryKey = Registry.CurrentUser.CreateSubKey(RegKey);
         if (registryKey == null)
         {
-            throw new Exception("注册表扫描失败");
+            _logger.Error($"注册表键不存在或无法创建:{RegKey}");
+            return;
         }
-        registryKey.SetValue("DisableTaskMgr", (int)state);
-        registryKey.Close();
+        registryKey.SetValue(DisableTaskMgrValueName, (int)state);
     }
 }

src/ComputerLock/Services/GlobalLockService.cs

@@ -34,10 +34,7 @@ public GlobalLockService(ILogger logger, AppSettings appSettings, UserActivityMo
         _hotkeyHook = hotkeyHook;
         _taskManagerHook = taskManagerHook;
 
-        // 防止锁屏时系统崩溃、重启等问题导致任务栏被禁用
-        // 启动时默认启用一次        
-        // TODO 这里需要重新设计，不应该在这里启用任务管理器，因为有可能任务管理器是被其他程序主动禁用的
-        _taskManagerHook.EnabledTaskManager();
+        _taskManagerHook.RecoverFromCrash();
         _mouseHook = mouseHook;
         _systemKeyHook = systemKeyHook;
         _serviceProvider = serviceProvider;
@@ -69,6 +66,7 @@ private void InitActivityMonitor()
 
     private void InitUserInputHandling()
     {
+        _logger.Info("系统 -> 准备监听用户输入状态");
         _systemKeyHook.OnUserInput += (_, _) =>
         {
             if (_appSettings.LockTips && IsLocked)
@@ -191,7 +189,7 @@ public void Lock()
         _activityMonitor.StopMonitoring();
 
         _logger.Info("全局锁定 -> 禁用任务管理器和系统键");
-        _taskManagerHook.DisabledTaskManager();
+        _taskManagerHook.Lock();
 
         if (_appSettings.IsHideMouseCursor)
         {
@@ -349,7 +347,7 @@ private void SystemUnlock()
         AutoLockStart();
 
         _logger.Info("系统解锁 -> 恢复任务管理器和系统键");
-        _taskManagerHook.EnabledTaskManager();
+        _taskManagerHook.Unlock();
 
         if (_appSettings.IsHideMouseCursor)
         {