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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 防火墙与安全工具 > 查看源码
Init.cpp
资源名称：FireWallSys.rar　[点击查看]
上传用户：yitai_qhd
上传日期：2008-04-24
资源大小：31k
文件大小：2k
源码类别：

防火墙与安全工具

开发平台：

Visual C++

Init.cpp：源码内容
  1. #include "hookndis.h"
  2. #include "init.h"
  3. #include "hook.h"
  4. #include "myfun.h"
  5. #include "fun.h"
  6. #include "memory.h"
  7. #include "WINDEF.H"
  9. #pragma code_seg("INIT") 
  11. extern "C"
  12. NTSTATUS DriverEntry( IN PDRIVER_OBJECT DriverObject,
  13.  IN PUNICODE_STRING RegistryPath)
  14. {
  15. PVOID m_NdisBaseAddress=NULL;
  16. NTSTATUS status = STATUS_SUCCESS;
  17. if(GetNdisModuleAddress(m_NdisBaseAddress) && m_NdisBaseAddress != NULL)
  18. {
  19. if(HookFun(m_NdisBaseAddress, "NdisSend", MY_NdisSend, (ULONG*)&m_pNdisSend) == NULL)
  20.             dprintf(("Hook NdisSend Failuren"));
  21.         else
  22.             dprintf(("Hook NdisSend Successn"));
  24.         if(HookFun(m_NdisBaseAddress, "NdisRegisterProtocol", MY_NdisRegisterProtocol, (ULONG*)&m_pNdisRegisterProtocol) == NULL
  25. )
  26.             dprintf(("Hook NdisRegisterProtocol Failuren"));
  27.         else
  28.             dprintf(("Hook NdisRegisterProtocol Successn"));
  30.         if(HookFun(m_NdisBaseAddress, "NdisOpenAdapter",     MY_NdisOpenAdapter, (ULONG*)&m_pNdisOpenAdapter) == NULL
  31. )
  32.             dprintf(("Hook NdisOpenAdapter Failuren"));
  33.         else
  34.             dprintf(("Hook NdisOpenAdapter Successn"));
  35. }
  36.     DriverObject->MajorFunction[IRP_MJ_CREATE] = Open;
  37.     DriverObject->MajorFunction[IRP_MJ_CLOSE] = Close;
  38.     DriverObject->MajorFunction[IRP_MJ_READ] = Read;
  39.     DriverObject->MajorFunction[IRP_MJ_WRITE] = Write;
  40.     DriverObject->MajorFunction[IRP_MJ_CLEANUP] = Cleanup;
  41. DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL]  = IoControl;
  42.     return status;
  43. }
  44. #pragma code_seg() 
  48. #pragma code_seg("PAGE") 
  50. VOID DbpUnload(IN PDRIVER_OBJECT DriverObject)
  51. {
  52. }
  55. #pragma code_seg() 
  56. //
  57. // IRP_MJ_CREATE
  58. //
  59. NTSTATUS
  60. Open(
  61.     IN PDEVICE_OBJECT DeviceObject,
  62.     IN PIRP Irp
  63. )
  64. {
  65. return 0;
  66. }
  68. //
  69. // IRP_MJ_CLOSE
  70. //
  71. NTSTATUS
  72. Close(
  73.     IN PDEVICE_OBJECT DeviceObject,
  74.     IN PIRP Irp
  75. )
  76. {
  77. return 0;
  78. }
  80. //
  81. // IRP_MJ_CLEANUP
  82. //
  83. NTSTATUS
  84. Cleanup(
  85.     IN PDEVICE_OBJECT DeviceObject,
  86.     IN PIRP Irp
  87. )
  88. {
  89. return 0;
  90. }
  92. //
  93. // IRP_MJ_READ
  94. //
  95. NTSTATUS
  96. Read(
  97.     IN PDEVICE_OBJECT DeviceObject,
  98.     IN PIRP Irp
  99. )
  100. {
  101. return 0;
  102. }
  104. //
  105. // IRP_MJ_WRITE
  106. //
  107. NTSTATUS
  108. Write(
  109.     IN PDEVICE_OBJECT DeviceObject,
  110.     IN PIRP Irp
  111. )
  112. {
  113. return 0;
  114. }
  116. NTSTATUS
  117. IoControl(
  118.     IN PDEVICE_OBJECT DeviceObject,
  119.     IN PIRP Irp
  120. )
  121. {
  123. return STATUS_SUCCESS;
  124. }