直接上源码
注释什么的都在源码上：
因为没有 打印机 实现 下载程序 超级终端 可以看到
/download/yfly13/863785
#include <ntddk.h>#define NTSTRSAFE_LIB#include <ntstrsafe.h>#ifndef SetFlag#define SetFlag(_F,_SF) ((_F) |= (_SF))#endif#ifndef ClearFlag#define ClearFlag(_F,_SF)((_F) &= ~(_SF))#endif#define CCP_MAX_COM_ID 32 //作者假设计算机上最多有32个串口// 过滤设备和真实设备static PDEVICE_OBJECT s_fltobj[CCP_MAX_COM_ID] = { 0 };static PDEVICE_OBJECT s_nextobj[CCP_MAX_COM_ID] = { 0 };// 打开一个端口设备PDEVICE_OBJECT ccpOpenCom(ULONG id,NTSTATUS *status){//外面输入的是串口ID，这里会改写成字符串的形式UNICODE_STRING name_str;static WCHAR name[32] = { 0 };PFILE_OBJECT fileobj = NULL;PDEVICE_OBJECT devobj = NULL;// 输入字符串。memset(name,0,sizeof(WCHAR)*32);RtlStringCchPrintfW(name,32,L"\\Device\\Serial%d",id);RtlInitUnicodeString(&name_str,name);// 打开设备对象*status = IoGetDeviceObjectPointer(&name_str, FILE_ALL_ACCESS, &fileobj, &devobj);if (*status == STATUS_SUCCESS)//fileobj 这里无用ObDereferenceObject(fileobj);//必须解除引用 不然会内存泄漏return devobj;}NTSTATUSccpAttachDevice(PDRIVER_OBJECT driver, PDEVICE_OBJECT oldobj, //通过名称得到的设备对象PDEVICE_OBJECT *fltobj, //过滤设备PDEVICE_OBJECT *next) //真实设备{NTSTATUS status;PDEVICE_OBJECT topdev = NULL;// 生成设备，然后绑定之。status = IoCreateDevice(driver,0,NULL,//过滤设备一般不需要名称oldobj->DeviceType, //保持和被绑定的一样0,FALSE,fltobj);//过滤设备if (status != STATUS_SUCCESS)return status;// 拷贝重要标志位。if(oldobj->Flags & DO_BUFFERED_IO)(*fltobj)->Flags |= DO_BUFFERED_IO;if(oldobj->Flags & DO_DIRECT_IO)(*fltobj)->Flags |= DO_DIRECT_IO;if(oldobj->Flags & DO_BUFFERED_IO)(*fltobj)->Flags |= DO_BUFFERED_IO;if(oldobj->Characteristics & FILE_DEVICE_SECURE_OPEN)(*fltobj)->Characteristics |= FILE_DEVICE_SECURE_OPEN;(*fltobj)->Flags |= DO_POWER_PAGABLE;// 绑定一个设备到另一个设备上topdev = IoAttachDeviceToDeviceStack(*fltobj,oldobj);//过滤设备， 要被绑定的设备栈中的设备if (topdev == NULL){// 如果绑定失败了，销毁设备，重新来过。IoDeleteDevice(*fltobj);*fltobj = NULL;status = STATUS_UNSUCCESSFUL;return status;}*next = topdev;//返回的最终被绑定的设备// 设置这个设备已经启动。(*fltobj)->Flags = (*fltobj)->Flags & ~DO_DEVICE_INITIALIZING;return STATUS_SUCCESS;}// 这个函数绑定所有的串口。void ccpAttachAllComs(PDRIVER_OBJECT driver){ULONG i;PDEVICE_OBJECT com_ob;NTSTATUS status;for(i = 0;i<CCP_MAX_COM_ID;i++){// 获得object对象。com_ob = ccpOpenCom(i,&status);if(com_ob == NULL)continue;// 在这里绑定。并不管绑定是否成功。ccpAttachDevice(driver,com_ob,&s_fltobj[i],&s_nextobj[i]);// 取消object引用。}}#define DELAY_ONE_MICROSECOND (-10)#define DELAY_ONE_MILLISECOND (DELAY_ONE_MICROSECOND*1000)#define DELAY_ONE_SECOND (DELAY_ONE_MILLISECOND*1000)void ccpUnload(PDRIVER_OBJECT drv){ULONG i;LARGE_INTEGER interval;// 首先解除绑定for(i=0;i<CCP_MAX_COM_ID;i++){if(s_nextobj[i] != NULL)IoDetachDevice(s_nextobj[i]);}// 睡眠5秒。等待所有irp处理结束interval.QuadPart = (5*1000 * DELAY_ONE_MILLISECOND);KeDelayExecutionThread(KernelMode,FALSE,&interval);// 删除这些设备for(i=0;i<CCP_MAX_COM_ID;i++){if(s_fltobj[i] != NULL)IoDeleteDevice(s_fltobj[i]);}KdPrint(("enter my unliad\n"));}NTSTATUS ccpDispatch(PDEVICE_OBJECT device,PIRP irp){PIO_STACK_LOCATION irpsp = IoGetCurrentIrpStackLocation(irp);NTSTATUS status;ULONG i,j;// 首先得知道发送给了哪个设备。设备一共最多CCP_MAX_COM_ID// 个，是前面的代码保存好的，都在s_fltobj中。for(i=0;i < CCP_MAX_COM_ID;i++){if(s_fltobj[i] == device){// 所有电源操作，全部直接放过。if(irpsp->MajorFunction == IRP_MJ_POWER){// 直接发送，然后返回说已经被处理了。PoStartNextPowerIrp(irp);IoSkipCurrentIrpStackLocation(irp);return PoCallDriver(s_nextobj[i],irp);}// 此外我们只过滤写请求。写请求的话，获得缓冲区以及其长度。// 然后打印一下。if(irpsp->MajorFunction == IRP_MJ_WRITE){// 如果是写，先获得长度ULONG len = irpsp->Parameters.Write.Length;// 然后获得缓冲区PUCHAR buf = NULL;if(irp->MdlAddress != NULL)buf = (PUCHAR)MmGetSystemAddressForMdlSafe(irp->MdlAddress,NormalPagePriority);elsebuf = (PUCHAR)irp->UserBuffer;if(buf == NULL)buf = (PUCHAR)irp->AssociatedIrp.SystemBuffer;// 打印内容for(j=0;j<len;++j){DbgPrint("comcap: Send Data: %2x\r\n", buf[j]);}}// 这些请求直接下发执行即可。我们并不禁止或者改变它。IoSkipCurrentIrpStackLocation(irp);//将请求发送到对应的真是设备,s_nextobj为真实设备保存地return IoCallDriver(s_nextobj[i],irp);}}// 如果根本就不在被绑定的设备中，那是有问题的，直接返回参数错误。irp->IoStatus.Information = 0;irp->IoStatus.Status = STATUS_INVALID_PARAMETER;IoCompleteRequest(irp,IO_NO_INCREMENT);return STATUS_SUCCESS;}NTSTATUS DriverEntry(PDRIVER_OBJECT driver, PUNICODE_STRING reg_path){size_t i;// 所有的分发函数都设置成一样的。KdPrint(("enter my driverentry\n"));for(i=0;i<IRP_MJ_MAXIMUM_FUNCTION;i++){driver->MajorFunction[i] = ccpDispatch;}// 支持动态卸载。driver->DriverUnload = ccpUnload;// 绑定所有的串口。ccpAttachAllComs(driver);// 直接返回成功即可。return STATUS_SUCCESS;}
如果觉得《寒江独钓-串口过滤》对你有帮助，请点赞、收藏，并留下你的观点哦！
