пятница, 4 сентября 2009 г.

Глобальный хук на мышку в .NET

Понадобилось мне как-то реализовать возможность авто-логаута для проекта с системой аутентификации. То есть, после простоя проекта без использования, он должен был скрываться и запрашивать пароль заново. Проект основывался на Windows Forms. Поискав в Интернете информацию, мне не понадобилось много времени, чтобы приладить хук на мышку к моему приложению.

Вот код, который понадобился мне для реализации этой возможности.

static class Program
{
private static FormMain m_formMain = null;
private static LowLevelMouseProc _proc = HookCallback;
private static IntPtr _hookID = IntPtr.Zero;

private static IntPtr SetHook(LowLevelMouseProc proc)
{
using (Process curProcess = Process.GetCurrentProcess())
using (ProcessModule curModule = curProcess.MainModule)
{
return SetWindowsHookEx(WH_MOUSE_LL, proc,
GetModuleHandle(curModule.ModuleName), 0);
}
}

private delegate IntPtr LowLevelMouseProc(int nCode, IntPtr wParam, IntPtr lParam);

private static IntPtr HookCallback(
int nCode, IntPtr wParam, IntPtr lParam)
{
if (nCode >= 0 && MouseMessages.WM_MOUSEMOVE == (MouseMessages)wParam)
{
if (m_formMain != null)
m_formMain.ResetLogoutTimer();
}
return CallNextHookEx(_hookID, nCode, wParam, lParam);
}

private const int WH_MOUSE_LL = 14;

private enum MouseMessages
{
WM_LBUTTONDOWN = 0x0201,
WM_LBUTTONUP = 0x0202,
WM_MOUSEMOVE = 0x0200,
WM_MOUSEWHEEL = 0x020A,
WM_RBUTTONDOWN = 0x0204,
WM_RBUTTONUP = 0x0205
}

[StructLayout(LayoutKind.Sequential)]
private struct POINT
{
public int x;
public int y;
}

[StructLayout(LayoutKind.Sequential)]
private struct MSLLHOOKSTRUCT
{
public POINT pt;
public uint mouseData;
public uint flags;
public uint time;
public IntPtr dwExtraInfo;
}

[DllImport("user32.dll", CharSet = CharSet.Auto, SetLastError = true)]
private static extern IntPtr SetWindowsHookEx(int idHook,
LowLevelMouseProc 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);

///
/// The main entry point for the application.
///

[STAThread]
static void Main()
{
Application.EnableVisualStyles();
_hookID = SetHook(_proc);
m_formMain = new FormMain();
Application.Run(m_formMain);
UnhookWindowsHookEx(_hookID);
}
}


FormMain - класс главной формы моего приложения.

Функция-callback - HookCallback позволяет нам фиксировать различные действия мышки. Таким образом, чтобы сбросить таймер авто-логаута необходимо отловить сообщение WM_MOUSEMOVE и вызвать соответствующие функции. Каждый раз, когда мышка двигается на экране (даже если приложение свернуто), таймер начинает заново с нуля отсчитывать секунды бездействия пользователя.