驱动程序的开发,向来是令人感到有所畏惧的,可能很多人像我一样,看了很久却还是一头雾水,不得其门而入。我们今天就通过一个简单的程序来使读者学会初步的驱动程序开发。
在开发Windows驱动程序之前,我们需要首先安装DDK,Win98及其以前的VxD我们就不要再考虑了;Windows 2000 DDK也比较老了点,很少有人使用了,微软最新的WDK貌似都不支持2000了;在本文中假设我们已经安装了Win XP DDK或2003 DDK(2003会包括XP)。
驱动开发通常有两种环境,一种是使用任意文本编辑器来编写代码,然后通过编写makefile和sources文件在WinDDK的命令行环境下使用build命令编译;另一种方法是使用各种各样的方法以图可以利用IDE的环境来搭建驱动框架,比如使用驱动向导文件或一些小工具,我经常使用的是EasySys这个小工具。
废话不多说了,假设我们已经通过EasySys生成了一个驱动框架,现在我们就向里面添加代码。下面我们编写了一个函数,它的作用是根据形参来创建一个文件,形参中给出了将要被创建的文件完整路径。(注意,以下程序有许多错误,具体请看后面的讲解)
BOOL CreateFileTest( IN PUNICODE_STRING FileName)
{
BOOL bRet = FALSE;
HANDLE hFile = NULL;
NTSTATUS status;
IO_STATUS_BLOCK Io_Status_Block;
// 初始化文件路径
OBJECT_ATTRIBUTES obj_attrib;
InitializeObjectAttributes(&obj_attrib,&FileName,OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE,NULL,NULL );
// 创建文件
status = ZwCreateFile(hFile,GENERIC_ALL,NULL,Io_Status_Block,NULL,FILE_ATTRIBUTE_NORMAL,FILE_SHARE_READ,FILE_CREATE,FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT,NULL,0 );
if (Io_Status_Block.Information != FILE_CREATED)
{
bRet = FALSE;
}
else
{
bRet = TRUE;
}
if (hFile)
{
ZwClose(hFile);
}
return bRet;
}
当我们激动地编译上述程序时,却很郁闷地发现编译器报出了十几个错误,不要急,这是大多数新手都必须要跨过的一个坎儿。很多时候你能看懂别人的程序并不等于你就能自己写程序,很有可能你看懂了别人的程序,但自己写的过程中却出现了很多错误,因为这中间有很多细节性的东西。下面我们来分析上述代码中存在的错误:
除此之外,上述代码还写的不够规范,比如判断函数调用是否成功的代码就有问题,下面是我们修改后的正确代码(请注意它与原代码的不同之处):
NTSTATUS CreateFileTest( IN PUNICODE_STRING FileName )
{
HANDLE hFile = NULL;
NTSTATUS status;
IO_STATUS_BLOCK Io_Status_Block;
// 初始化文件路径
OBJECT_ATTRIBUTES obj_attrib;
InitializeObjectAttributes( &obj_attrib, FileName, OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE,
NULL, NULL );
// 创建文件
status = ZwCreateFile( &hFile, GENERIC_ALL, &obj_attrib, &Io_Status_Block, NULL,
FILE_ATTRIBUTE_NORMAL, FILE_SHARE_READ, FILE_CREATE, // 创建文件
FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT, NULL, 0 );
if (NT_SUCCESS(status))
{
status = STATUS_SUCCESS;
}
else
{
status = Io_Status_Block.Status;
}
KdPrint(("hFile = %08X", hFile));
// 关闭句柄
if (hFile)
{
ZwClose(hFile);
}
return status;
}
下面我们在DriverEntry中编写代码测试上述函数,如下所示:
// 测试函数
RtlInitUnicodeString( &ustrFileName, L"\??\C:\nokyo.dat");
Status = CreateFileTest(&ustrFileName);
if (NT_SUCCESS(Status))
{
KdPrint(("[FileTest] CreateFile Success!"));
}
else
{
KdPrint(("[FileTest] CreateFile Error(%08X)!", Status));
}
注意ustrFileName变量必须在函数开头进行定义,上述代码的运行效果如下图所示。