contrib/msgbeep.cpp

   1 #include <easyhook.h>
   2 #include <stdio.h>
   3
   4 HOOK_TRACE_INFO hinfo = {NULL};
   5 ULONG acl_entries[128] = {};
   6 int thread_num = 0;
   7
   8 BOOL WINAPI MessageBeep_hook(UINT type)
   9 {
  10    printf("Beep!\n");
  11    return TRUE;
  12 }
  13
  14 void hook_install()
  15 {
  16    LhInstallHook(MessageBeep, MessageBeep_hook, 0, &hinfo);
  17 }
  18
  19 void hook_uninstall()
  20 {
  21    LhUninstallAllHooks();
  22    LhWaitForPendingRemovals();
  23 }
  24
  25 void hook_enable_thread()
  26 {
  27    LhSetInclusiveACL(acl_entries, ++thread_num, &hinfo);
  28 }
  29
  30 BOOL WINAPI DllMain(HINSTANCE instance, DWORD reason, LPVOID)
  31 {
  32    switch(reason) {
  33    case DLL_PROCESS_ATTACH:
  34       hook_install();
  35       /* go through to enable current thread */
  36    case DLL_THREAD_ATTACH:
  37       hook_enable_thread();
  38       break;
  39    case DLL_THREAD_DETACH:
  40       break;
  41    case DLL_PROCESS_DETACH:
  42       hook_uninstall();
  43    }
  44    return TRUE;
  45 }