天书笔记（2）---使用C++开发内核程序

1.使用C接口的标准声明

在C++中使用C编写的接口，要在前面加 extern "c"。

extern "C"{ #include <ntddk.h> }; extern "C" NTSTATUS DriverEntry(PDRIVER_OBJECT driver_object,PUNICODE_STRING reg_path) { return STATUS_SUCCESS; }

2.重载操作符

void *__cdecl operator new(unsigned int size) { void *pt=ExAllocatePool(NonPagedPool,size); if (pt!=NULL) { memset(pt,0,size); } return pt; } void __cdecl operator delete(void *p) { ExFreePool(p); } void __cdecl operator delete [](void *p) { ExFreePool(p); }

3.使用类静态成员函数

使用C++的时候总是要注意，把一个类的成员函数当做普通的C函数使用是不对的。至少类的成员函数都知道this指针，而C函数不知道，所以它们的接口实际上是不一样的。而静态函数可以，静态函数调用的结果不会访问或者修改任何对象（非static）数据成员。

下面实现一个MyDriver的类，把分发函数作为虚函数以便于继承

extern "C"{ #include <ntddk.h> }; void *__cdecl operator new(unsigned int size) { void *pt=ExAllocatePool(NonPagedPool,size); if (pt!=NULL) { memset(pt,0,size); } return pt; } class MyDriver { public: MyDriver(PDRIVER_OBJECT driver); virtual NTSTATUS OnDispatch(PDEVICE_OBJECT dev,PIRP irp)//测试：直接返回失败 { return STATUS_UNSUCCESSFUL; } static MyDriver *d_my_driver;//静态成员，总是记录唯一被实例化的MyDriver指针 private: //静态成员函数，用来作为Dispatch函数使用 static NTSTATUS sDispatch(PDEVICE_OBJECT dev,PIRP irp); PDRIVER_OBJECT d_driver; } MyDriver::MyDriver(PDRIVER_OBJECT driver) : d_driver(driver) { size_t i; for (i=0;i<=IRP_MJ_MAXIMUM_FUNCTION;i++) { driver->MajorFunction[i]=sDispatch; } d_my_driver=this; } NTSTATUS MyDriver::sDispatch(PDEVICE_OBJECT dev,PIRP irp) { return d_my_driver->OnDispatch(dev,irp); } MyDriver *MyDriver::d_my_driver=NULL; extern "C" NTSTATUS DriverEntry(PDRIVER_OBJECT driver_object,PUNICODE_STRING reg_path) { MyDriver *MyDriver::d_my_driver=new MyDriver(driver_object); return STATUS_SUCCESS; }

代码照着书敲得可哥就是没编译成功。。。

driver->MajorFunction[i]=sDispatch为什么不直接给我们的OnDispatch，因为类成员函数（右）跟普通C函数（左）接口不同，所以又定义了一个静态函数，这个静态函数来作为分发函数的指针，这个静态函数里面调用虚函数OnDispatch（要调用这个虚函数因为静态函数里面没有this指针，所以去定义了一个类型为MyDriver*的静态成员，这等于是一个全局公共的指针，只有singleton才可以这样做），以便我们派生。