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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 防火墙与安全工具 > 查看源码
IpHookGlobal.h
资源名称：ipfilter.zip　[点击查看]
上传用户：nnxzhh
上传日期：2007-01-11
资源大小：742k
文件大小：3k
源码类别：

防火墙与安全工具

开发平台：

WINDOWS

IpHookGlobal.h：源码内容
  1. ///////////////////////////////////////////////////////////////////////////////
  2. //
  3. // (C) Copyright 1999 - 2000 Mark Roddy
  4. // All Rights Reserved
  5. //
  6. // Hollis Technology Solutions
  7. // 94 Dow Road
  8. // Hollis, NH 03049
  9. // info@hollistech.com
  10. //
  11. // Synopsis: 
  12. // 
  13. //
  14. // Version Information:
  15. //
  16. // $Header: /iphook/sys/driver/IpHookGlobal.h 4     1/27/00 10:35p Markr $ 
  17. //
  18. ///////////////////////////////////////////////////////////////////////////////
  19. #pragma once
  20. //
  21. // this was a simple structure but alas it has developed class
  22. // it ought to be a class in fact and as it is the single global
  23. // data object it ought to enforce its singleton nature.
  24. //
  25. class IP_HOOK_GLOBAL_DATA {
  27. private:
  29. //
  30. // we track the thread that started a hook
  31. //
  32. PVOID owningContext;
  33. PDEVICE_OBJECT IpfDeviceObject;
  34. PFILE_OBJECT IpfFileObject;
  35. ULONG IpHookSequence;
  36. //
  37. // we don't know what the actual situation
  38. // is here - are we called serially at our
  39. // hook function? Are we called at DISPATCH_LEVEL?
  40. //
  41. // until we figure this out use a spinlock.
  42. // 
  43. spinLock Lock;
  45. LIST_ENTRY queue;
  46. LIST_ENTRY cancelQueue;
  47. PIRP currentIrp;
  48. PIPHOOK_BUFFER currentBuffer;
  50. //
  51. // a worker thread wakes up every now and then and flushes the
  52. // current buffer
  53. //
  54. HANDLE  workerThread;
  55. PVOID workerThreadObject;
  56. KEVENT  stopEvent;
  59. public:
  61. NTSTATUS createThread();
  63. NTSTATUS initDevice();
  65. void queueRequest(PIRP Irp);
  67. PIRP popQueue(BOOLEAN locked = FALSE);
  69. PIPHOOK_DATA getBuffer();
  71. void flushBuffer(BOOLEAN locked = FALSE);
  73. void flushBufferIfData();
  75. void releaseAllBuffers(BOOLEAN locked = FALSE);
  77. IP_HOOK_GLOBAL_DATA();
  79. ~IP_HOOK_GLOBAL_DATA();
  81. BOOLEAN IsOwner(PVOID context)
  82. {
  83. if (context == owningContext) {
  85. return TRUE;
  86. }
  88. return FALSE;
  89. }
  91. BOOLEAN owned()
  92. {
  93. return (owningContext != NULL);
  94. }
  96. PVOID owner()
  97. {
  98. return owningContext;
  99. }
  101. BOOLEAN setOwner(PVOID context)
  102. {
  103. //
  104. // we sort of kinda hafta serialize this
  105. //
  106. PVOID inUse = 
  107. InterlockedCompareExchangePointer(&owningContext, 
  108.   context,
  109.   NULL);
  111. if (inUse != NULL) {
  113. return FALSE;
  114. }
  116. return TRUE;
  117. }
  119. void freeOwner(PVOID context)
  120. {
  121. PVOID lockContext;
  123. lock(lockContext);
  125. if (IsOwner(context)) {
  127. releaseAllBuffers(TRUE);
  129. InterlockedExchangePointer(&owningContext, NULL);
  130. }
  132. unlock(lockContext);
  133. }
  135. ULONG getSequence()
  136. {
  137. return InterlockedIncrement((PLONG)&IpHookSequence);
  138. }
  140. PDEVICE_OBJECT getIpDevice()
  141. {
  142. return IpfDeviceObject;
  143. }
  145. void lock(PVOID& context)
  146. {
  147. Lock.lock(context);
  148. }
  150. void unlock(PVOID context)
  151. {
  152. Lock.unlock(context);
  153. }
  155. };
  156. ///////////////////////////////////////////////////////////////////////////////
  157. // 
  158. // Change History Log
  159. //
  160. // $Log: /iphook/sys/driver/IpHookGlobal.h $
  161. // 
  162. // 4     1/27/00 10:35p Markr
  163. // Prepare to release!
  164. //
  165. ///////////////////////////////////////////////////////////////////////////////