开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 书籍源码 > 查看源码
Hook.cpp
资源名称：c++时尚编程百例__1-20.rar　[点击查看]
上传用户：aokegd
上传日期：2009-12-14
资源大小：1276k
文件大小：5k
源码类别：

书籍源码

开发平台：

Visual C++

Hook.cpp：源码内容
  1. // Hook.cpp : Defines the initialization routines for the DLL.
  2. #include "stdafx.h"
  3. #include "hook.h"
  4. #include <windowsx.h>
  5. #ifdef _DEBUG
  6. #define new DEBUG_NEW
  7. #undef THIS_FILE
  8. static char THIS_FILE[] = __FILE__;
  9. #endif
  10. #define MAX_KEY 100
  11. #define CTRLBIT 0x04
  12. #define ALTBIT  0x02
  13. #define SHIFTBIT 0x01 
  14. /////////////////////////////////////////////////////////////////
  15. #pragma data_seg("shareddata")
  16. HHOOK hHook =NULL;
  17. UINT nHookCount =0;
  18. static UCHAR HotKey[MAX_KEY] = {0}; //hotkey
  19. static UCHAR HotKeyMask[MAX_KEY] = {0}; //flag for hotkey, value is VK_CONTRL or VK_NEMU or VK_SHIFT
  20. static HWND hCallWnd[MAX_KEY] = {0}; //window associated with hotkey
  21. static int KeyCount =0;
  22. static UCHAR MaskBits =0;    //00000 Ctrl Alt Shift
  23. #pragma data_seg()
  24. ////////////////////////////////////////////////////////////////
  25. HINSTANCE hins;
  26. void VerifyWindow();
  27. /////////////////////////////////////////////////////////////////////////////
  28. // CHookApp
  29. BEGIN_MESSAGE_MAP(CHookApp, CWinApp)
  30. //{{AFX_MSG_MAP(CHookApp)
  31. // NOTE - the ClassWizard will add and remove mapping macros here.
  32. //    DO NOT EDIT what you see in these blocks of generated code!
  33. //}}AFX_MSG_MAP
  34. END_MESSAGE_MAP()
  36. /////////////////////////////////////////////////////////////////////////////
  37. // CHookApp construction
  39. CHookApp::CHookApp()
  40. {
  41. // TODO: add construction code here,
  42. // Place all significant initialization in InitInstance
  43. }
  45. /////////////////////////////////////////////////////////////////////////////
  46. // The one and only CHookApp object
  48. CHookApp theApp;
  49. LRESULT CALLBACK KeyboardProc(int nCode,WPARAM wParam,LPARAM lParam)
  50. {
  51. BOOL bProcessed=FALSE;
  52. if(HC_ACTION==nCode)
  53. {
  54. if((lParam&0xc0000000)==0xc0000000){// Key up
  55. switch(wParam)
  56. {
  57. case VK_MENU:
  58. MaskBits&=~ALTBIT;
  59. break;
  60. case VK_CONTROL:
  61. MaskBits&=~CTRLBIT;
  62. break;
  63. case VK_SHIFT:
  64. MaskBits&=~SHIFTBIT;
  65. break;
  66. default: //judge the key and send message
  67. break;
  68. }
  69. for(int index=0;index<MAX_KEY;index++){
  70. if(hCallWnd[index]==NULL)
  71. continue;
  72. if(IsWindow(hCallWnd[index])&&(HotKey[index]==wParam)&&(HotKeyMask[index]==MaskBits)){
  73. SendMessage(hCallWnd[index],WM_HOTKEY,wParam,WM_KEYUP);
  74. bProcessed=TRUE;
  75. }
  76. }
  77. }
  78. else if((lParam&0xc000ffff)==1){ //Key down
  79. switch(wParam)
  80. {
  81. case VK_MENU:
  82. MaskBits|=ALTBIT;
  83. break;
  84. case VK_CONTROL:
  85. MaskBits|=CTRLBIT;
  86. break;
  87. case VK_SHIFT:
  88. MaskBits|=SHIFTBIT;
  89. break;
  90. default: //judge the key and send message
  91. break;
  92. }
  93. for(int index=0;index<MAX_KEY;index++){
  94. if(hCallWnd[index]==NULL)
  95. continue;
  96. if(IsWindow(hCallWnd[index])&&(HotKey[index]==wParam)&&(HotKeyMask[index]==MaskBits)){
  97. SendMessage(hCallWnd[index],WM_HOTKEY,wParam,WM_KEYDOWN);
  98. bProcessed=TRUE;
  99. }
  100. }
  101. }
  102. if(!bProcessed){
  103. for(int index=0;index<MAX_KEY;index++){
  104. if(hCallWnd[index]==NULL)
  105. continue;
  106. if(IsWindow(hCallWnd[index])&&(HotKey[index]==0)&&(HotKeyMask[index]==0))
  107. SendMessage(hCallWnd[index],WM_HOTKEY,WM_HOTKEY,lParam);
  108. }
  109. }
  110. }
  111. return CallNextHookEx( hHook, nCode, wParam, lParam );
  112. }
  114. BOOL InitHotkey()
  115. {
  116. if(hHook!=NULL){
  117. nHookCount++;
  118. return TRUE;
  119. }
  120. else
  121. hHook=SetWindowsHookEx(WH_KEYBOARD,(HOOKPROC)KeyboardProc,hins,0);
  122. if(hHook!=NULL)
  123. nHookCount++;
  124. return (hHook!=NULL);
  125. }
  126. BOOL UnInit()
  127. {   
  128. if(nHookCount>1){
  129. nHookCount--;
  130. return TRUE;
  131. }
  132. BOOL unhooked = UnhookWindowsHookEx(hHook);
  133. if(unhooked==TRUE){
  134. nHookCount=0;
  135. hHook=NULL;
  136. }
  137. return unhooked;
  138. } 
  140. BOOL __declspec(dllexport) __stdcall AddHotkey(HWND hWnd,UCHAR cKey,UCHAR cMask)
  141. {
  142. BOOL bAdded=FALSE;
  143. for(int index=0;index<MAX_KEY;index++){
  144. if(hCallWnd[index]==0){
  145. hCallWnd[index]=hWnd;
  146. HotKey[index]=cKey;
  147. HotKeyMask[index]=cMask;
  148. bAdded=TRUE;
  149. KeyCount++;
  150. break;
  151. }
  152. }
  153. return bAdded;
  154. }
  156. BOOL __declspec(dllexport) __stdcall DeleteHotkey(HWND hWnd,UCHAR cKey,UCHAR cMask)
  157. {
  158. BOOL bRemoved=FALSE;
  159. for(int index=0;index<MAX_KEY;index++){
  160. if(hCallWnd[index]==hWnd){
  161. if(HotKey[index]==cKey&&HotKeyMask[index]==cMask){
  162. hCallWnd[index]=NULL;
  163. HotKey[index]=0;
  164. HotKeyMask[index]=0;
  165. bRemoved=TRUE;
  166. KeyCount--;
  167. break;
  168. }
  169. }
  170. }
  171. return bRemoved;
  172. }
  174. void VerifyWindow()
  175. {
  176. for(int i=0;i<MAX_KEY;i++){
  177. if(hCallWnd[i]!=NULL){
  178. if(!IsWindow(hCallWnd[i])){
  179. hCallWnd[i]=NULL;
  180. HotKey[i]=0;
  181. HotKeyMask[i]=0;
  182. KeyCount--;
  183. }
  184. }
  185. }
  186. }
  188. BOOL CHookApp::InitInstance() 
  189. {
  190. AFX_MANAGE_STATE(AfxGetStaticModuleState());
  191. hins=AfxGetInstanceHandle();
  192. InitHotkey();
  193. return CWinApp::InitInstance();
  194. }
  196. int CHookApp::ExitInstance() 
  197. {
  198. VerifyWindow();
  199. UnInit();
  200. return CWinApp::ExitInstance();
  201. }