书上源码防火墙程序出问题高手帮看看怎么回事

catastrophe

书上源码防火墙程序出问题高手帮看看怎么回事 [复制链接]

下面是设置过滤函数的程序，是书上的源程序，不知道为什么执行不成功，在客户程序中使用GetLastError()得到2，找不到指定文件，初次接触驱动程序，希望老虾们给看看

1. 
   
2. NTSTATUS SetFilterFunction(PacketFilterExtensionPtr filterFun)
   
3. {       
   
4.         NTSTATUS status = STATUS_SUCCESS;
   
5. 
   
6.                 // 取得IP过滤驱动设备对象。下面代码执行后，pDeviceObj变量将指向IP过滤驱动设备对象
   
7.         PDEVICE_OBJECT pDeviceObj;
   
8.         PFILE_OBJECT pFileObj;
   
9.         // 初始化IP过滤驱动的名称
   
10.         UNICODE_STRING ustrFilterDriver;
    
11.         RtlInitUnicodeString(&ustrFilterDriver, L"\\Device\\IpFilterDriver");
    
12.         // 取得设备对象指针       
    
13.         status = IoGetDeviceObjectPointer(&ustrFilterDriver, FILE_ALL_ACCESS, &pFileObj, &pDeviceObj);
    
14.         if(!NT_SUCCESS(status))
    
15.         {
    
16.                 return status;
    
17.         }
    
18. 
    
19.                 // 使用到IP过滤驱动中设备对象的指针创建一个IRP       
    
20.         // 填充PF_SET_EXTENSION_HOOK_INFO结构
    
21.         PF_SET_EXTENSION_HOOK_INFO filterData;
    
22.         filterData.ExtensionPointer = filterFun;
    
23. 
    
24.         // 我们需要初始化一个事件对象。
    
25.         // 构建IRP时需要使用这个事件内核对象，当IP过滤取得接受到此IRP，完成工作以后会将它置位
    
26.         KEVENT event;
    
27.         KeInitializeEvent(&event, NotificationEvent, FALSE);
    
28. 
    
29.         // 为设备控制请求申请和构建一个IRP
    
30.         PIRP pIrp;
    
31.         IO_STATUS_BLOCK ioStatus;
    
32.         pIrp = IoBuildDeviceIoControlRequest(IOCTL_PF_SET_EXTENSION_POINTER,  // io control code
    
33.                 pDeviceObj,
    
34.                 (PVOID) &filterData,
    
35.                 sizeof(PF_SET_EXTENSION_HOOK_INFO),
    
36.                 NULL,
    
37.                 0,
    
38.                 FALSE,
    
39.                 &event,
    
40.                 &ioStatus);
    
41.         if(pIrp == NULL)
    
42.         {
    
43.                 // 如果不能申请空间，返回对应的错误代码
    
44.                 return STATUS_INSUFFICIENT_RESOURCES;
    
45.         }
    
46.        
    
47.                 // 请求安装钩子回调函数
    
48.         // 发送此IRP到IP过滤驱动
    
49.         status = IoCallDriver(pDeviceObj, pIrp);
    
50.         // 等待IP过滤驱动的通知
    
51.         if(status == STATUS_PENDING)
    
52.         {
    
53.                 KeWaitForSingleObject(&event, Executive, KernelMode, FALSE, NULL);
    
54.         }
    
55.         status = ioStatus.Status;
    
56. 
    
57.         // 清除资源
    
58.         if(pFileObj != NULL)
    
59.                 ObDereferenceObject(pFileObj);
    
60.         return status;
    
61. }

复制代码

yoyo_xmu

1. NTSTATUS DispatchIoctl(PDEVICE_OBJECT pDevObj, PIRP pIrp)//处理IRP
   
2. {
   
3.         NTSTATUS status = STATUS_SUCCESS;
   
4. 
   
5.         // 取得此IRP（pIrp）的I/O堆栈指针
   
6.         PIO_STACK_LOCATION pIrpStack = IoGetCurrentIrpStackLocation(pIrp);
   
7. 
   
8.         // 取得I/O控制代码
   
9.         ULONG uIoControlCode = pIrpStack->Parameters.DeviceIoControl.IoControlCode;
   
10.         // 取得I/O缓冲区指针和它的长度
    
11.         PVOID pIoBuffer = pIrp->AssociatedIrp.SystemBuffer;
    
12.         ULONG uInSize = pIrpStack->Parameters.DeviceIoControl.InputBufferLength;
    
13.        
    
14.         // 响应用户的命令
    
15.         switch(uIoControlCode)
    
16.         {
    
17.         case START_IP_HOOK:        // 开始过滤
    
18.                 status = SetFilterFunction(FilterPackets);
    
19.                 break;
    
20.       
    
21. ........
    
22.         return status;
    
23. }
    

复制代码

客户程序：

1. 
   
2. if(m_pFilterDrv->IoControl(START_IP_HOOK, NULL, 0, NULL, 0) == -1)//m_pFilterDrv为IP过滤钩子驱动指针
   
3.         {
   
4.                 str.Format("%d",GetLastError());
   
5.                 MessageBox(str,"error");
   
6.                 return;
   
7.         }
   

复制代码

bayasina

跪求了．．．．．．

zgfjcc

status = IoGetDeviceObjectPointer(&ustrFilterDriver, SYNCHRONIZE | GENERIC_READ | GENERIC_WRITE, &pFileObj, &pDeviceObj);

处理DeviceControl完成时要
pIrp->IoStatus.Status = status;
pIrp->IoStatus.Information = 0;

qq413243608

看看你的代码！
monkeycd@163.com

kerrigan

还不行啊，cnzdgs 你要代码吗？

卖蛋筒

taianmonkey  代码已经发送

xyw6813927

不明LZ在说什么