?? driver.cpp
字號:
/************************************************************************
* 文件名稱:Driver.cpp
* 作 者:張帆
* 完成日期:2007-11-1
*************************************************************************/
#include "Driver.h"
/************************************************************************
* 函數名稱:DriverEntry
* 功能描述:初始化驅動程序，定位和申請硬件資源，創建內核對象
* 參數列表:
pDriverObject:從I/O管理器中傳進來的驅動對象
pRegistryPath:驅動程序在注冊表的中的路徑
* 返回 值:返回初始化驅動狀態
*************************************************************************/
#pragma INITCODE
extern "C" NTSTATUS DriverEntry (
IN PDRIVER_OBJECT pDriverObject,
IN PUNICODE_STRING pRegistryPath )
{
NTSTATUS ntStatus;
KdPrint(("DriverB:Enter B DriverEntry\n"));
//注冊其他驅動調用函數入口
pDriverObject->DriverUnload = HelloDDKUnload;
pDriverObject->MajorFunction[IRP_MJ_CREATE] = HelloDDKCreate;
pDriverObject->MajorFunction[IRP_MJ_CLOSE] = HelloDDKClose;
pDriverObject->MajorFunction[IRP_MJ_WRITE] = HelloDDKDispatchRoutine;
pDriverObject->MajorFunction[IRP_MJ_READ] = HelloDDKRead;
UNICODE_STRING DeviceName;
RtlInitUnicodeString( &DeviceName, L"\\Device\\MyDDKDeviceA" );
PDEVICE_OBJECT DeviceObject = NULL;
PFILE_OBJECT FileObject = NULL;
//尋找DriverA創建的設備對象
ntStatus = IoGetDeviceObjectPointer(&DeviceName,FILE_ALL_ACCESS,&FileObject,&DeviceObject);
if (!NT_SUCCESS(ntStatus))
{
KdPrint(("DriverB:IoGetDeviceObjectPointer() 0x%x\n", ntStatus ));
return ntStatus;
}
//創建自己的驅動設備對象
ntStatus = CreateDevice(pDriverObject);
if ( !NT_SUCCESS( ntStatus ) )
{
ObDereferenceObject( FileObject );
DbgPrint( "IoCreateDevice() 0x%x!\n", ntStatus );
return ntStatus;
}
PDEVICE_EXTENSION pdx = (PDEVICE_EXTENSION) pDriverObject->DeviceObject->DeviceExtension;
PDEVICE_OBJECT FilterDeviceObject = pdx->pDevice;
//將自己的設備對象掛載在DriverA的設備對象上
PDEVICE_OBJECT TargetDevice = IoAttachDeviceToDeviceStack( FilterDeviceObject,
DeviceObject );
//將底層設備對象記錄下來
pdx->TargetDevice = TargetDevice;
if ( !TargetDevice )
{
ObDereferenceObject( FileObject );
IoDeleteDevice( FilterDeviceObject );
DbgPrint( "IoAttachDeviceToDeviceStack() 0x%x!\n", ntStatus );
return STATUS_INSUFFICIENT_RESOURCES;
}
FilterDeviceObject->DeviceType = TargetDevice->DeviceType;
FilterDeviceObject->Characteristics = TargetDevice->Characteristics;
FilterDeviceObject->Flags &= ~DO_DEVICE_INITIALIZING;
FilterDeviceObject->Flags |= ( TargetDevice->Flags & ( DO_DIRECT_IO |
DO_BUFFERED_IO ) );
ObDereferenceObject( FileObject );
KdPrint(("DriverB:B attached A successfully!\n"));
KdPrint(("DriverB:Leave B DriverEntry\n"));
return ntStatus;
}
/************************************************************************
* 函數名稱:CreateDevice
* 功能描述:初始化設備對象
* 參數列表:
pDriverObject:從I/O管理器中傳進來的驅動對象
* 返回 值:返回初始化狀態
*************************************************************************/
#pragma INITCODE
NTSTATUS CreateDevice (
IN PDRIVER_OBJECT pDriverObject)
{
NTSTATUS ntStatus;
PDEVICE_OBJECT pDevObj;
PDEVICE_EXTENSION pDevExt;
//創建設備名稱
UNICODE_STRING devName;
RtlInitUnicodeString(&devName,L"\\Device\\MyDDKDeviceB");
//創建設備
ntStatus = IoCreateDevice( pDriverObject,
sizeof(DEVICE_EXTENSION),
&(UNICODE_STRING)devName,
FILE_DEVICE_UNKNOWN,
0, TRUE,
&pDevObj );
if (!NT_SUCCESS(ntStatus))
return ntStatus;
pDevObj->Flags |= DO_DIRECT_IO;
pDevExt = (PDEVICE_EXTENSION)pDevObj->DeviceExtension;
pDevExt->pDevice = pDevObj;
pDevExt->ustrDeviceName = devName;
return STATUS_SUCCESS;
}
/************************************************************************
* 函數名稱:HelloDDKUnload
* 功能描述:負責驅動程序的卸載操作
* 參數列表:
pDriverObject:驅動對象
* 返回 值:返回狀態
*************************************************************************/
#pragma PAGEDCODE
VOID HelloDDKUnload (IN PDRIVER_OBJECT pDriverObject)
{
PDEVICE_OBJECT pNextObj;
KdPrint(("DriverB:Enter B DriverUnload\n"));
pNextObj = pDriverObject->DeviceObject;
while (pNextObj != NULL)
{
PDEVICE_EXTENSION pDevExt = (PDEVICE_EXTENSION)
pNextObj->DeviceExtension;
pNextObj = pNextObj->NextDevice;
//從設備棧中彈出
IoDetachDevice( pDevExt->TargetDevice);
//刪除該設備對象
IoDeleteDevice( pDevExt->pDevice );
}
KdPrint(("DriverB:Enter B DriverUnload\n"));
}
/************************************************************************
* 函數名稱:HelloDDKDispatchRoutine
* 功能描述:對讀IRP進行處理
* 參數列表:
pDevObj:功能設備對象
pIrp:從IO請求包
* 返回 值:返回狀態
*************************************************************************/
#pragma PAGEDCODE
NTSTATUS HelloDDKDispatchRoutine(IN PDEVICE_OBJECT pDevObj,
IN PIRP pIrp)
{
KdPrint(("DriverB:Enter B HelloDDKDispatchRoutine\n"));
NTSTATUS ntStatus = STATUS_SUCCESS;
// 完成IRP
pIrp->IoStatus.Status = ntStatus;
pIrp->IoStatus.Information = 0; // bytes xfered
IoCompleteRequest( pIrp, IO_NO_INCREMENT );
KdPrint(("DriverB:Leave B HelloDDKDispatchRoutine\n"));
return ntStatus;
}
#pragma PAGEDCODE
NTSTATUS HelloDDKCreate(IN PDEVICE_OBJECT pDevObj,
IN PIRP pIrp)
{
KdPrint(("DriverB:Enter B HelloDDKCreate\n"));
NTSTATUS ntStatus = STATUS_SUCCESS;
//
// // 完成IRP
// pIrp->IoStatus.Status = ntStatus;
// pIrp->IoStatus.Information = 0; // bytes xfered
// IoCompleteRequest( pIrp, IO_NO_INCREMENT );
PDEVICE_EXTENSION pdx = (PDEVICE_EXTENSION)pDevObj->DeviceExtension;
IoSkipCurrentIrpStackLocation (pIrp);
ntStatus = IoCallDriver(pdx->TargetDevice, pIrp);
KdPrint(("DriverB:Leave B HelloDDKCreate\n"));
return ntStatus;
}
const int MAX_PACKAGE_SIZE=1024;
#pragma PAGEDCODE
NTSTATUS
HelloDDKReadCompletion(
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp,
IN PVOID Context
)
{
KdPrint(("DriverB:Enter B HelloDDKReadCompletion\n"));
PMYDRIVER_RW_CONTEXT rwContext = (PMYDRIVER_RW_CONTEXT) Context;
NTSTATUS ntStatus = Irp->IoStatus.Status;
ULONG stageLength;
if(rwContext && NT_SUCCESS(ntStatus))
{
//已經傳送了多少字節
rwContext->Numxfer += Irp->IoStatus.Information;
if(rwContext->Length)
{
//設定下一階段讀取字節數
if(rwContext->Length > MAX_PACKAGE_SIZE)
{
stageLength = MAX_PACKAGE_SIZE;
}
else
{
stageLength = rwContext->Length;
}
//重新利用MDL
MmPrepareMdlForReuse(rwContext->NewMdl);
IoBuildPartialMdl(Irp->MdlAddress,
rwContext->NewMdl,
(PVOID) rwContext->VirtualAddress,
stageLength);
rwContext->VirtualAddress += stageLength;
rwContext->Length -= stageLength;
IoCopyCurrentIrpStackLocationToNext(Irp);
PIO_STACK_LOCATION nextStack = IoGetNextIrpStackLocation(Irp);
nextStack->Parameters.Read.Length = stageLength;
IoSetCompletionRoutine(Irp,
HelloDDKReadCompletion,
rwContext,
TRUE,
TRUE,
TRUE);
IoCallDriver(rwContext->DeviceExtension->TargetDevice,
Irp);
return STATUS_MORE_PROCESSING_REQUIRED;
}
else
{
//最后一次傳輸
Irp->IoStatus.Information = rwContext->Numxfer;
}
}
KdPrint(("DriverB:Leave B HelloDDKReadCompletion\n"));
return STATUS_MORE_PROCESSING_REQUIRED;
}
#pragma PAGEDCODE
NTSTATUS HelloDDKRead(IN PDEVICE_OBJECT pDevObj,
IN PIRP pIrp)
{
KdPrint(("DriverB:Enter B HelloDDKRead\n"));
NTSTATUS status = STATUS_SUCCESS;
PDEVICE_EXTENSION pDevExt = (PDEVICE_EXTENSION)
pDevObj->DeviceExtension;
PIO_STACK_LOCATION stack = IoGetCurrentIrpStackLocation(pIrp);
ULONG totalLength;
ULONG stageLength;
PMDL mdl;
PVOID virtualAddress;
PMYDRIVER_RW_CONTEXT rwContext = NULL;
PIO_STACK_LOCATION nextStack;
if (!pIrp->MdlAddress)
{
status = STATUS_UNSUCCESSFUL;
totalLength = 0;
goto HelloDDKRead_EXIT;
}
//獲取MDL的虛擬地址
virtualAddress = MmGetMdlVirtualAddress(pIrp->MdlAddress);
//獲取MDL的長度
totalLength = MmGetMdlByteCount(pIrp->MdlAddress);
KdPrint(("DriverB:(pIrp->MdlAddress)MmGetMdlVirtualAddress:%08X\n",MmGetMdlVirtualAddress(pIrp->MdlAddress)));
KdPrint(("DriverB:(pIrp->MdlAddress)MmGetMdlByteCount:%d\n",MmGetMdlByteCount(pIrp->MdlAddress)));
//將總的傳輸，分成幾個階段，這里設定每次階段的長度
if(totalLength > MAX_PACKAGE_SIZE)
{
stageLength = MAX_PACKAGE_SIZE;
}else
{
stageLength = totalLength;
}
//創建新的MDL
mdl = IoAllocateMdl((PVOID) virtualAddress,
totalLength,
FALSE,
FALSE,
NULL);
KdPrint(("DriverB:(new mdl)MmGetMdlVirtualAddress:%08X\n",MmGetMdlVirtualAddress(mdl)));
KdPrint(("DriverB:(new mdl)MmGetMdlByteCount:%d\n",MmGetMdlByteCount(mdl)));
if(mdl == NULL)
{
KdPrint(("DriverB:Failed to alloc mem for mdl\n"));
status = STATUS_INSUFFICIENT_RESOURCES;
goto HelloDDKRead_EXIT;
}
//將IRP的MDL做重新映射
IoBuildPartialMdl(pIrp->MdlAddress,
mdl,
(PVOID) virtualAddress,
stageLength);
KdPrint(("DriverB:(new mdl)MmGetMdlVirtualAddress:%08X\n",MmGetMdlVirtualAddress(mdl)));
KdPrint(("DriverB:(new mdl)MmGetMdlByteCount:%d\n",MmGetMdlByteCount(mdl)));
rwContext = (PMYDRIVER_RW_CONTEXT)
ExAllocatePool(NonPagedPool,sizeof(MYDRIVER_RW_CONTEXT));
rwContext->NewMdl = mdl;
rwContext->PreviousMdl = pIrp->MdlAddress;
rwContext->Length = totalLength - stageLength;//還剩下多少沒讀取
rwContext->Numxfer = 0; //讀了多少字節
rwContext->VirtualAddress = ((ULONG_PTR)virtualAddress + stageLength);//下一階段開始讀取的地址
rwContext->DeviceExtension = pDevExt;
//拷貝到底層堆棧
IoCopyCurrentIrpStackLocationToNext(pIrp);
nextStack = IoGetNextIrpStackLocation(pIrp);
//根據底層驅動的實現，底層驅動有可能讀取這個數值，也有可能讀取mdl的length。
nextStack->Parameters.Read.Length = stageLength;
pIrp->MdlAddress = mdl;
//設定完成例程
IoSetCompletionRoutine(pIrp,
(PIO_COMPLETION_ROUTINE)HelloDDKReadCompletion,
rwContext,
TRUE,
TRUE,
TRUE);
IoCallDriver(pDevExt->TargetDevice,pIrp);
pIrp->MdlAddress = rwContext->PreviousMdl;
IoFreeMdl(rwContext->NewMdl);
HelloDDKRead_EXIT:
// 完成IRP
pIrp->IoStatus.Status = status;
pIrp->IoStatus.Information = totalLength; // bytes xfered
IoCompleteRequest( pIrp, IO_NO_INCREMENT );
KdPrint(("DriverB:Leave B HelloDDKRead\n"));
return status;
}
#pragma PAGEDCODE
NTSTATUS HelloDDKClose(IN PDEVICE_OBJECT pDevObj,
IN PIRP pIrp)
{
KdPrint(("DriverB:Enter B HelloDDKClose\n"));
NTSTATUS ntStatus = STATUS_SUCCESS;
PDEVICE_EXTENSION pdx = (PDEVICE_EXTENSION)pDevObj->DeviceExtension;
IoSkipCurrentIrpStackLocation (pIrp);
ntStatus = IoCallDriver(pdx->TargetDevice, pIrp);
KdPrint(("DriverB:Leave B HelloDDKClose\n"));
return ntStatus;
}?? 快捷鍵說明
復制代碼
Ctrl + C
搜索代碼
Ctrl + F
全屏模式
F11
切換主題
Ctrl + Shift + D
顯示快捷鍵
?
增大字號
Ctrl + =
減小字號
Ctrl + -