第二講:我的第一個驅動
阿新 • • 發佈:2018-11-12
原文:http://blog.csdn.net/caperingrabbit/article/details/5285288
配置好了開發環境之後,下面就要通過具體的程式來了解驅動的開發了。下面我們以一個WDM驅動的框架來實現Windows驅動程式的HelloWorld。
作為一個驅動程式,首先應該寫的是它的入口函式,這點跟MFC的WinMain或者C++中的Main函式一樣,驅動的入口函式使用DriverEntry。在入口函式中主要實現的功能是一些分發例程的註冊以及其他的需要初始化的事務。
DriverEntry的原型是NTSTATUS DriverEntry(IN PDRIVER_OBJECT pDriverObject,IN PUNICODE_STRING pRegistryPath),
- extern"C" //一般用C++進行Windows驅動開發,在入口函式前面要加這個關鍵詞,主要是為了在驅動程式中能夠呼叫C的函式
- NTSTATUS DriverEntry(IN PDRIVER_OBJECT pDriverObject,
- IN PUNICODE_STRING pRegistryPath
- )
- {
- KdPrint(("Hello!welcome to the driver entry!/n"));//KdPrint是一個巨集,類似於MFC中的Trace
- //對於通用的例程,這裡也使用一個通用的處理函式進行註冊
- for(ULONG i = 0; i <= IRP_MJ_MAXIMUM_FUNCTION; i++)
- {
- pDriverObject->MajorFunction[i] = DispatchRoutine;
- }
- //驅動的功能例程的註冊,這是幾個比較特殊的例程,我們分別註冊,
- //而驅動開發的主要工作就是完成他們的這些例程的實現函式,為了簡單方便講解,這裡使用的和上邊一樣
- pDriverObject->DriverExtension->AddDevice = MyDeviceAdd;
- pDriverObject->DriverUnload = DriverUnload;
- pDriverObject->MajorFunction[IRP_MJ_CREATE] = DispatchRoutine;
- pDriverObject->MajorFunction[IRP_MJ_CLOSE] = DispatchRoutine;
- pDriverObject->MajorFunction[IRP_MJ_READ] = DispatchRoutine;
- pDriverObject->MajorFunction[IRP_MJ_WRITE] = DispatchRoutine;
- return STATUS_SUCCESS;
- }
完成了入口函式之後,接下來就要實現上面註冊的分發例程的功能函數了。作為一個WDM驅動程式,首先需要建立裝置物件,這個任務就交給AddDevice例程,通過上面的註冊,實現部分放在MyDeviceAdd。
[cpp] view plain copy
- NTSTATUS MyDeviceAdd(IN PDRIVER_OBJECT DriverObject,
- IN PDEVICE_OBJECT PhysicalDeviceObject)
- {
- PAGED_CODE();//DDK提供的一個巨集
- KdPrint(("Enter MyDrivers/n"));
- NTSTATUS status;
- PDEVICE_OBJECT fdo;
- UNICODE_STRING devName;
- RtlInitUnicodeString(&devName,L"//Device//MyDevice");
- //建立裝置物件
- status = IoCreateDevice(
- DriverObject,
- sizeof(DEVICE_EXTENSION),
- &(UNICODE_STRING)devName,
- FILE_DEVICE_UNKNOWN,
- 0,
- FALSE,
- &fdo);
- if( !NT_SUCCESS(status))
- return status;
- //獲得裝置擴充套件的資料結構。這個資料結構一般是由使用者自己定義,儲存一些裝置相關的資訊
- PDEVICE_EXTENSION pdx = (PDEVICE_EXTENSION)fdo->DeviceExtension;
- //配置裝置擴充套件的內容
- pdx->fdo = fdo;
- //裝置的掛載
- pdx->NextStackDevice = IoAttachDeviceToDeviceStack(fdo, PhysicalDeviceObject);
- UNICODE_STRING symLinkName;
- RtlInitUnicodeString(&symLinkName,L"//DosDevices//MyDriver");
- pdx->ustrDeviceName = devName;
- pdx->ustrSymLinkName = symLinkName;
- status = IoCreateSymbolicLink(&(UNICODE_STRING)symLinkName,&(UNICODE_STRING)devName);
- if( !NT_SUCCESS(status))
- {
- IoDeleteSymbolicLink(&pdx->ustrSymLinkName);
- status = IoCreateSymbolicLink(&symLinkName,&devName);
- if( !NT_SUCCESS(status))
- {
- return status;
- }
- }
- //設定裝置的IO狀態
- fdo->Flags |= DO_BUFFERED_IO | DO_POWER_PAGABLE;
- fdo->Flags &= ~DO_DEVICE_INITIALIZING;
- KdPrint(("LeaveMyDrivers/n"));
- return STATUS_SUCCESS;
- }
生成了裝置物件之後,接下來就要處理髮向裝置的各種請求了。這裡就用一個通用的處理例程來示範一下。其實這個處理函式只是將這些請求轉交給真實的裝置而已,沒有實際的功能。
[cpp] view plain copy
- NTSTATUS HelloWDMDispatchRoutine(IN PDEVICE_OBJECT fdo,
- IN PIRP Irp)
- {
- PAGED_CODE();
- KdPrint(("Enter HelloWDMDispatchRoutine/n"));
- Irp->IoStatus.Status = STATUS_SUCCESS;
- Irp->IoStatus.Information = 0; //
- IoCompleteRequest( Irp, IO_NO_INCREMENT );
- KdPrint(("Leave HelloWDMDispatchRoutine/n"));
- return STATUS_SUCCESS;
- }
- 執行完所有的請求之後,在最後解除安裝驅動。對於WDM驅動,標準的程式其實解除安裝例程沒什麼實質內容,一般都是有Pnp裡邊的RemoveDevice來處理的,這裡只是一個演示,演示一下這個例程的功能和內容。
- VOID DriverUnload(IN PDRIVER_OBJECT pDriverObject)
- {
- PDEVICE_OBJECT pNextObj;
- KdPrint(("Enter DriverUnload/n"));
- pNextObj = pDriverObject->DeviceObject;
- while (pNextObj != NULL)
- {
- PDEVICE_EXTENSION pDevExt = (PDEVICE_EXTENSION)
- pNextObj->DeviceExtension;
- //刪除符號連結
- UNICODE_STRING pLinkName = pDevExt->ustrSymLinkName;
- IoDeleteSymbolicLink(&pLinkName);
- pNextObj = pNextObj->NextDevice;
- IoDeleteDevice( pDevExt->pDevice );
- }
- }
這樣,一個驅動的主要部分就完成了。接下來完成一些細碎的工作,編譯一下,一個驅動就出現了。
標頭檔案
[cpp] view plain copy
- #ifdef __cplusplus
- extern "C"
- {
- #endif
- #include <wdm.h>
- #ifdef __cplusplus
- }
- #endif
- typedef struct _DEVICE_EXTENSION
- {
- PDEVICE_OBJECT fdo;
- PDEVICE_OBJECT NextStackDevice;
- UNICODE_STRING ustrDeviceName; // 裝置名
- UNICODE_STRING ustrSymLinkName; // 符號連結名
- } DEVICE_EXTENSION, *PDEVICE_EXTENSION;
- #define PAGEDCODE code_seg("PAGE")
- #define LOCKEDCODE code_seg()
- #define INITCODE code_seg("INIT")
- #define PAGEDDATA data_seg("PAGE")
- #define LOCKEDDATA data_seg()
- #define INITDATA data_seg("INIT")
- #define arraysize(p) (sizeof(p)/sizeof((p)[0]))
- NTSTATUS MyDeviceAdd(IN PDRIVER_OBJECT DriverObject,
- IN PDEVICE_OBJECT PhysicalDeviceObject);
- NTSTATUS DispatchRoutine(IN PDEVICE_OBJECT fdo,
- IN PIRP Irp);
- void DriverUnload(IN PDRIVER_OBJECT DriverObject);
- extern "C"
- NTSTATUS DriverEntry(IN PDRIVER_OBJECT DriverObject,
- IN PUNICODE_STRING RegistryPath);
新建一個文字檔案,內容如下:
[cpp] view plain copy
- #
- # DO NOT EDIT THIS FILE!!! Edit ./sources. If you want to add a new source
- # file to this component. This file merely indirects to the real make file
- # that is shared by all the driver components of the Windows NT DDK
- #
- !INCLUDE $(NTMAKEENV)/makefile.def
重新命名檔案,改為Makefile,字尾沒有。然後重新建立一個文字檔案,寫入如下內容:
[cpp] view plain copy
- TARGETNAME=HellOWORLD
- TARGETTYPE=DRIVER
- DRIVERTYPE=WDM
- TARGETPATH=OBJ
- INCLUDES=$(BASEDIR)/inc;/
- $(BASEDIR)/inc/ddk;/
- SOURCES=MyDrivers.cpp/
儲存,更改檔名為Sources,字尾沒有,這樣進入DDK的編譯環境,已編譯,一個字尾為.sys的驅動檔案就出現了。
好了,一個簡單的驅動就誕生了。