用C++ 和 Win32 SDK 写的简单键盘锁

最近总要在笔记本电脑前面写东西，所以一般把本子放在键盘上。但是又容易不小心按了哪个键。所以写了个键盘锁，这样锁起来之后按什么键都不起作用了。主要是使用了 Hook 技术。

可能有谁也会需要到，就贴在这里了。

另外，关于气泡提示中的 timeout 的变量好像不是很准，我设置了 1000 毫秒，但是基本上要等上 10 秒左右才能消失。暂时还不知道是什么原因。


#define _WIN32_IE 0x0500


#include <windows.h>




enum




...{


WM_TRAYICON = WM_USER + 100, //自定义托盘消息


ID_LOCK,


ID_UNLOCK,


ID_EXIT


};




HHOOK KBhook = NULL;//钩子句柄




LRESULT CALLBACK WndProc(HWND, UINT, WPARAM, LPARAM);


LRESULT CALLBACK KeyboardHookProc(int nCode, WPARAM wParam, LPARAM lParam);//键盘钩子回调函数




void TrayMessage(HWND hwnd, LPCTSTR szText);//显示托盘气泡信息


void DestroyTrayIcon(HWND hwnd);//删除托盘图标


void CreateTrayMenu(HWND hwnd);//建立托盘菜单




int EnableKeyboardCapture();//激活键盘钩子


int DisableKeyboardCapture();//解除键盘钩子




void OnLock(HWND hwnd);


void OnUnlock(HWND hwnd);






int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, PSTR szCmdLine, int iCmdShow)




...{


HWND hwnd;


MSG msg;


NOTIFYICONDATA IconData;


WNDCLASS wndclass;




static TCHAR szAppName[] = ...{"Keyboard Lock"};




//初始化窗口


wndclass.style = CS_HREDRAW | CS_VREDRAW;


wndclass.lpfnWndProc = WndProc;


wndclass.cbClsExtra = 0;


wndclass.cbWndExtra = 0;


wndclass.hCursor = LoadCursor(NULL, IDI_APPLICATION);


wndclass.hIcon = LoadIcon(NULL, IDC_ARROW);


wndclass.hbrBackground = (HBRUSH)GetStockObject(WHITE_BRUSH);


wndclass.hInstance = hInstance;


wndclass.lpszMenuName = NULL;


wndclass.lpszClassName = szAppName;




RegisterClass(&wndclass);




hwnd = CreateWindow(szAppName, szAppName, WS_OVERLAPPEDWINDOW, 0, 0, 0, 0, 0, 0, hInstance, NULL);




//这里之后不需要显示窗口，新建托盘图标，操作全部在托盘图标上完成


//Warning: HWND 和 HINSTANCE 不对的话，就会有问题




IconData.cbSize = sizeof(NOTIFYICONDATA);


IconData.hWnd = hwnd;


IconData.uID = (UINT)hInstance;


IconData.hIcon = LoadIcon(0, MAKEINTRESOURCE(IDI_INFORMATION));


IconData.uFlags = NIF_MESSAGE + NIF_ICON + NIF_TIP;


IconData.uCallbackMessage = WM_TRAYICON;




strcpy(IconData.szTip, "键盘锁");




Shell_NotifyIcon(NIM_ADD, &IconData);




while(GetMessage(&msg, NULL, 0, 0))




...{


TranslateMessage(&msg);


DispatchMessage(&msg);


}




return msg.wParam;


}






LRESULT CALLBACK WndProc(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)




...{


switch (message)




...{


case WM_COMMAND:


if(HIWORD(wParam) == 0)




...{


//菜单动作的消息


switch (LOWORD(wParam))




...{


case ID_LOCK:


OnLock(hwnd);


break;




case ID_UNLOCK:


OnUnlock(hwnd);


break;




case ID_EXIT:


::DestroyWindow(hwnd);


break;


}


}


break;




case WM_TRAYICON:


switch (lParam)




...{


case WM_LBUTTONDOWN:


CreateTrayMenu(hwnd);


break;


}


break;




case WM_DESTROY:


DestroyTrayIcon(hwnd);


PostQuitMessage(0);


break;


}




return DefWindowProc(hwnd, message, wParam, lParam);


}






LRESULT CALLBACK KeyboardHookProc(int nCode, WPARAM wParam, LPARAM lParam)




...{


//是键盘的动作全部忽略


if (nCode == HC_ACTION)


return 1;




return CallNextHookEx(NULL, nCode, wParam, lParam);


}






int EnableKeyboardCapture()




...{


// 13 表示使用低级键盘钩子，和 WM_KEYBOARD 不同


KBhook = !KBhook ? SetWindowsHookEx(13, (HOOKPROC)KeyboardHookProc, (HINSTANCE)GetModuleHandle(NULL), 0) : KBhook;




return (KBhook ? 1 : -1);


}






int DisableKeyboardCapture()




...{


if (KBhook == NULL)




...{


return 0;


}


else




...{


BOOL flag;




flag = UnhookWindowsHookEx(KBhook);


KBhook = NULL;




return (flag ? 1 : -1);


}


}






void TrayMessage(HWND hwnd, LPCTSTR szText)




...{


NOTIFYICONDATA nid;




nid.cbSize = sizeof(NOTIFYICONDATA);


nid.hWnd = hwnd;


nid.uID = (UINT)GetModuleHandle(NULL);


nid.uFlags = NIF_INFO | NIF_ICON;


nid.dwInfoFlags = NIIF_INFO;


nid.uTimeout = 1000;


nid.hIcon = LoadIcon(0, MAKEINTRESOURCE(IDI_INFORMATION));




wsprintf(nid.szInfo, szText);


strcpy(nid.szInfoTitle, "键盘锁");




Shell_NotifyIcon(NIM_MODIFY, &nid);


}






void DestroyTrayIcon(HWND hwnd)




...{


NOTIFYICONDATA nid;


nid.cbSize = sizeof(nid);


nid.uID = (UINT)GetModuleHandle(NULL);


nid.hWnd = hwnd;


nid.uFlags = 0;


Shell_NotifyIcon(NIM_DELETE, &nid);


}






void CreateTrayMenu(HWND hwnd)




...{


HMENU hMenu;


hMenu = CreatePopupMenu();


AppendMenu(hMenu, MF_STRING, ID_LOCK, "锁定键盘");


AppendMenu(hMenu, MF_STRING, ID_UNLOCK, "解锁键盘");


AppendMenu(hMenu, MF_SEPARATOR, 0, NULL);


AppendMenu(hMenu, MF_STRING, ID_EXIT, "退出");




POINT pt;


GetCursorPos(&pt);




TrackPopupMenu(hMenu, TPM_RIGHTALIGN, pt.x, pt.y, 0, hwnd, NULL);


}






void OnUnlock(HWND hwnd)




...{


switch (DisableKeyboardCapture())




...{


case 0:


TrayMessage(hwnd, "键盘没有锁定");


break;




case 1:


TrayMessage(hwnd, "键盘已经解锁");


break;




case -1:


TrayMessage(hwnd, "键盘解锁错误");


break;


}


}






void OnLock(HWND hwnd)




...{


switch (EnableKeyboardCapture())




...{


case 1:


TrayMessage(hwnd, "键盘已经锁定");


break;




case -1:


TrayMessage(hwnd, "键盘锁定错误");


break;


}


}