-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathGlobalActivityHook.cs
More file actions
111 lines (90 loc) · 3.99 KB
/
GlobalActivityHook.cs
File metadata and controls
111 lines (90 loc) · 3.99 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
using System;
using System.Diagnostics;
using System.Runtime.InteropServices;
using System.Windows.Forms;
namespace WorkTimer
{
/// <summary>
/// Captures global mouse and keyboard activity using low-level Windows hooks.
/// </summary>
public class GlobalActivityHook : IDisposable
{
private readonly IntPtr _mouseHookID = IntPtr.Zero;
private readonly IntPtr _keyboardHookID = IntPtr.Zero;
// Keep delegates as member variables to prevent them from being garbage collected
private readonly HookProc _mouseProc;
private readonly HookProc _keyboardProc;
private const int WH_MOUSE_LL = 14;
private const int WH_KEYBOARD_LL = 13;
private const int WM_MOUSEMOVE = 0x0200;
private const int WM_KEYDOWN = 0x0100;
public event EventHandler<ActivityEventArgs>? OnActivity;
public GlobalActivityHook()
{
_mouseProc = MouseHookCallback;
_keyboardProc = KeyboardHookCallback;
_mouseHookID = SetHook(_mouseProc, WH_MOUSE_LL);
_keyboardHookID = SetHook(_keyboardProc, WH_KEYBOARD_LL);
}
private IntPtr SetHook(HookProc proc, int hookType)
{
using (Process curProcess = Process.GetCurrentProcess())
{
// Process.MainModule can be null in some circumstances (e.g., 64-bit processes under WOW64).
// We add a null check here to make the code more robust and resolve the compiler warning.
ProcessModule? curModule = curProcess.MainModule;
if (curModule == null)
{
// If we can't get the module, we can't set the hook.
// Throw an exception to indicate a fatal setup error.
throw new InvalidOperationException("Could not get main module of the current process to set a global hook.");
}
using (curModule)
{
return SetWindowsHookEx(hookType, proc, GetModuleHandle(curModule.ModuleName), 0);
}
}
}
private IntPtr MouseHookCallback(int nCode, IntPtr wParam, IntPtr lParam)
{
if (nCode >= 0 && wParam == WM_MOUSEMOVE)
{
OnActivity?.Invoke(this, new ActivityEventArgs(false));
}
return CallNextHookEx(_mouseHookID, nCode, wParam, lParam);
}
private IntPtr KeyboardHookCallback(int nCode, IntPtr wParam, IntPtr lParam)
{
if (nCode >= 0 && wParam == WM_KEYDOWN)
{
OnActivity?.Invoke(this, new ActivityEventArgs(true));
}
return CallNextHookEx(_keyboardHookID, nCode, wParam, lParam);
}
public class ActivityEventArgs : EventArgs
{
public bool IsKeyboard { get; }
public ActivityEventArgs(bool isKeyboard)
{
IsKeyboard = isKeyboard;
}
}
#region IDisposable and P/Invoke
public void Dispose()
{
UnhookWindowsHookEx(_mouseHookID);
UnhookWindowsHookEx(_keyboardHookID);
}
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);
#endregion
}
}