优化定时锁定功能

JiuLing-zhang · Feb 7, 2024 · ff38022 · ff38022
1 parent cf7da51
commit ff38022
Showing 1 changed file with 22 additions and 137 deletions.
diff --git a/src/Hooks/UserActivityMonitor.cs b/src/Hooks/UserActivityMonitor.cs
@@ -1,53 +1,37 @@
-using System.Diagnostics;
-using System.Runtime.InteropServices;
+using System.Runtime.InteropServices;
 using System.Timers;
 using Timer = System.Timers.Timer;
 
 namespace ComputerLock.Hooks;
 
 public class UserActivityMonitor : IDisposable
 {
-    private Stopwatch _activityStopwatch;
-    private Timer _timer;
 
-    private IntPtr _keyboardHook;
-    private IntPtr _mouseHook;
-
-    private HookProc _keyboardHookCallback;
-    private HookProc _mouseHookCallback;
-
-    private const int WH_KEYBOARD_LL = 13;
-    private const int WH_MOUSE_LL = 14;
-    public EventHandler OnIdle;
+    [DllImport("user32.dll")]
+    static extern bool GetLastInputInfo(ref LastInputInfo plii);
+    struct LastInputInfo
+    {
+        public uint cbSize;
+        public uint dwTime;
+    }
 
-    private int _autoLockSecond;
+    private Timer? _timer;
+    public EventHandler? OnIdle;
 
+    private int _autoLockMillisecond;
     private bool _isMonitoring = false;
-
     public void Init(int autoLockSecond)
     {
-        _autoLockSecond = autoLockSecond;
-
-        _activityStopwatch = new Stopwatch();
-        _activityStopwatch.Start();
+        _autoLockMillisecond = autoLockSecond * 1000;
 
         _timer = new Timer();
         _timer.Interval = 1000;
         _timer.Elapsed += _timer_Elapsed;
         _timer.Start();
-
-        // 初始化钩子回调函数
-        _keyboardHookCallback = KeyboardHookCallback;
-        _mouseHookCallback = MouseHookCallback;
-
-        // 启动键盘和鼠标钩子
-        _keyboardHook = SetWindowsHookEx(WH_KEYBOARD_LL, _keyboardHookCallback, GetModuleHandle(Process.GetCurrentProcess().MainModule.ModuleName), 0);
-        _mouseHook = SetWindowsHookEx(WH_MOUSE_LL, _mouseHookCallback, GetModuleHandle(Process.GetCurrentProcess().MainModule.ModuleName), 0);
     }
 
     public void StartMonitoring()
     {
-        _activityStopwatch.Restart();
         _isMonitoring = true;
     }
 
@@ -57,125 +41,26 @@ public void StopMonitoring()
     }
     private void _timer_Elapsed(object? sender, ElapsedEventArgs e)
     {
-        if (_isMonitoring)
-        {
-            TimeSpan idleTime = _activityStopwatch.Elapsed;
-            if (idleTime.TotalSeconds > _autoLockSecond)
-            {
-                OnIdle?.Invoke(this, EventArgs.Empty);
-            }
-        }
-    }
-
-    private IntPtr KeyboardHookCallback(int nCode, IntPtr wParam, IntPtr lParam)
-    {
-        if (nCode >= 0 && (wParam == (IntPtr)WM_KEYDOWN || wParam == (IntPtr)WM_SYSKEYDOWN))
+        if (!_isMonitoring)
         {
-            _activityStopwatch.Restart();
+            return;
         }
 
-        return CallNextHookEx(_keyboardHook, nCode, wParam, lParam);
-    }
+        var lastInputInfo = new LastInputInfo();
+        lastInputInfo.cbSize = (uint)Marshal.SizeOf(lastInputInfo);
 
-    private IntPtr MouseHookCallback(int nCode, IntPtr wParam, IntPtr lParam)
-    {
-        if (wParam == (IntPtr)WM_LBUTTONDOWN || wParam == (IntPtr)WM_RBUTTONDOWN || wParam == (IntPtr)WM_MBUTTONDOWN || wParam == (IntPtr)WM_XBUTTONDOWN)
-        {
-            _activityStopwatch.Restart();
-        }
-        else if (wParam == (IntPtr)WM_MOUSEMOVE)
+        if (GetLastInputInfo(ref lastInputInfo))
         {
-            // 处理鼠标移动事件，重置计时器
-            _activityStopwatch.Restart();
+            long elapsedMillisecond = (long)Environment.TickCount64 - lastInputInfo.dwTime;
+            if (elapsedMillisecond > _autoLockMillisecond)
+            {
+                OnIdle?.Invoke(this, EventArgs.Empty);
+            }
         }
-
-        return CallNextHookEx(_mouseHook, nCode, wParam, lParam);
     }
 
-    // 钩子相关的声明和函数
-
-    [StructLayout(LayoutKind.Sequential)]
-    private struct KBDLLHOOKSTRUCT
-    {
-        public Keys key;
-        public int scanCode;
-        public int flags;
-        public int time;
-        public IntPtr dwExtraInfo;
-    }
-
-    [StructLayout(LayoutKind.Sequential)]
-    private struct MSLLHOOKSTRUCT
-    {
-        public Point pt;
-        public int mouseData;
-        public int flags;
-        public int time;
-        public IntPtr dwExtraInfo;
-    }
-
-    private delegate IntPtr HookProc(int nCode, IntPtr wParam, IntPtr lParam);
-
-    [DllImport("user32.dll", CharSet = CharSet.Auto, SetLastError = true)]
-    private static extern IntPtr SetWindowsHookEx(int idHook, HookProc lpfn, IntPtr hMod, uint dwThreadId);
-
-    [DllImport("user32.dll", CharSet = CharSet.Auto, SetLastError = true)]
-    [return: MarshalAs(UnmanagedType.Bool)]
-    private static extern bool UnhookWindowsHookEx(IntPtr hhk);
-
-    [DllImport("user32.dll", CharSet = CharSet.Auto, SetLastError = true)]
-    private static extern IntPtr CallNextHookEx(IntPtr hhk, int nCode, IntPtr wParam, IntPtr lParam);
-
-    [DllImport("kernel32.dll", CharSet = CharSet.Auto, SetLastError = true)]
-    private static extern IntPtr GetModuleHandle(string lpModuleName);
-
-    private const int WM_KEYDOWN = 0x0100;
-    private const int WM_SYSKEYDOWN = 0x0104;
-    private const int WM_MOUSEMOVE = 0x0200;
-    private const int WM_LBUTTONDOWN = 0x0201;
-    private const int WM_RBUTTONDOWN = 0x0204;
-    private const int WM_MBUTTONDOWN = 0x0207;
-    private const int WM_XBUTTONDOWN = 0x020B;
-
-    // 定义其他结构和函数
-
-    [StructLayout(LayoutKind.Sequential)]
-    private struct Point
-    {
-        public int X;
-        public int Y;
-    }
-
-    private const uint INPUT_KEYBOARD = 1;
-    private const uint KEYEVENTF_KEYUP = 0x0002;
-
-    [DllImport("user32.dll")]
-    [return: MarshalAs(UnmanagedType.Bool)]
-    private static extern bool GetCursorPos(out Point lpPoint);
-
-    [DllImport("user32.dll")]
-    private static extern bool ScreenToClient(IntPtr hWnd, ref Point lpPoint);
-
     public void Dispose()
     {
-        // 停止键盘和鼠标钩子
-        if (_keyboardHook != null)
-        {
-            UnhookWindowsHookEx(_keyboardHook);
-        }
-        if (_mouseHook != null)
-        {
-            UnhookWindowsHookEx(_mouseHook);
-        }
-        if (_keyboardHookCallback != null)
-        {
-            _keyboardHookCallback = null;
-        }
-        if (_mouseHookCallback != null)
-        {
-            _mouseHookCallback = null;
-        }
-        _activityStopwatch?.Stop();
         _timer?.Stop();
         _timer?.Dispose();
     }