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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > PlugIns编程 > 查看源码
HookBase.h
资源名称：MSNPlugin_src.zip　[点击查看]
上传用户：kittypts
上传日期：2018-02-11
资源大小：241k
文件大小：2k
源码类别：

PlugIns编程

开发平台：

Visual C++

HookBase.h：源码内容
  1. /*
  2. *
  3. *  This file is
  4. *    Copyright (C) 2006-2008 Nektra S.A.
  5. *  
  6. *  This program is free software; you can redistribute it and/or modify
  7. *  it under the terms of the GNU Lesser General Public License as published by
  8. *  the Free Software Foundation; either version 2, or (at your option)
  9. *  any later version.
  10. *  
  11. *  This program is distributed in the hope that it will be useful,
  12. *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14. *  GNU General Public License for more details.
  15. *  
  16. */
  17. #ifndef _HOOKBASE_H_
  18. #define _HOOKBASE_H_
  20. #include "FunctionWrapper.h"
  21. #include "VirtualCode.h"
  22. #include "LocalFunction.h"
  23. #include "HookTypes.h"
  25. /**
  26. Base class for local hooks.
  27. */
  28. class NktHookBase : public INktHookSimple
  29. {
  30. public:
  31. NktHookBase();
  32. ~NktHookBase();
  34. /**
  35. Call hooked function with specified context.
  36. */
  37. void CallFunction(NktHookCallContext*);
  39. /**
  40. Skip call and and use retVal on return.
  41. */
  42. void SkipCall(NktHookCallContext*, INT_PTR retVal);
  44. /**
  45. Global hooks enable:
  46. */
  47. static BOOL GetGlobalEnabled();
  48. static void SetGlobalEnabled(BOOL enable);
  50. /**
  51. Force return value in call.
  52. */
  53. void SetReturnValue(NktHookCallContext*, INT_PTR retVal);
  55. /**
  56. Force lastError value in call.
  57. */
  58. void SetLastError(NktHookCallContext*, INT_PTR err);
  60. /**
  61. Disable reports from calling thread.
  62. */
  63. static HANDLE DisableThreadReports();
  65. /**
  66. Restore previous state of thread reports.
  67. */
  68. static void RestoreThreadReports(HANDLE);
  70. /**
  71. Tells if reports are disabled for the calling thread.
  72. */
  73. static BOOL IsThreadDisabled();
  75. /**
  76. Enable / Disable reports
  77. */
  78. void SetEnabled(BOOL);
  79. BOOL GetEnabled() const;
  81. /**
  82. Get hook flags
  83. */
  84. int GetFlags() const;
  86. protected:
  87. /**
  88. Implement call of hooked function.
  89. */
  90. virtual void RawCallFunction(NktHookCallContext*) = ABSTRACT;
  92. protected:
  93. void SetHandler(const NktLocalFunction&);
  94. const NktFunctionWrapper& GetHandler() const;
  96. void SetFlags(int);
  98. void SetUserData(INT_PTR);
  99. INT_PTR GetUserData() const;
  101. /**
  102. Internal: First Handler.
  103. */
  104. void __fastcall InternalHandler(NktRegisters* reg, INT_PTR tag);
  106. private:
  107. NktLocalFunction _handler;
  108. int _flags;
  109. INT_PTR _userData;
  110. BOOL _disabled;
  112. protected:
  113. static BOOL hooksDisabled;
  114. };
  116. #endif //_HOOKBASE_H_