バカのドライバー勉強ノート1

　　不知不觉间学驱动好像过了一年了，不过丫的又因为没怎么用所以结构跟常用的函数都记不住orz，于是就做个笔记当复习好了。
    一个简单的驱动如下：

#include<ntddk.h>

typedef struct _DEVICE_EXTENSION{
	PDEVICE_OBJECT pDevice;
	UNICODE_STRING devicename;
	UNICODE_STRING Symlink;
}DEVICE_EXTENSION,*PDEVICE_EXTENSION;


#pragma code_seg("PAGE")
NTSTATUS MyDispath(IN PDEVICE_OBJECT pdevobj,IN PIRP pirp)
{
	KdPrint(("Dispath call!"));
	return STATUS_SUCCESS;
}

#pragma code_seg("PAGE")
NTSTATUS MyUnload(IN PDRIVER_OBJECT pdrvobj)
{
	NTSTATUS status;
	PDEVICE_EXTENSION pdrvex ;
	pdrvex  = (PDEVICE_EXTENSION)pdrvobj->DeviceObject->DeviceExtension ;
	status = IoDeleteSymbolicLink(&(pdrvex->Symlink));
	if(!NT_SUCCESS(status))
	{
		KdPrint(("Delete Symbolic link fault!"));
		return status;
	}
	IoDeleteDevice(pdrvobj->DeviceObject);
	
	KdPrint(("Delete Device successfully!"));
	
	
	return STATUS_SUCCESS;
}

#pragma code_seg("INIT")
NTSTATUS DriverEntry(IN PDRIVER_OBJECT pobj , IN PUNICODE_STRING path)
{

	PDEVICE_OBJECT pdev;
	PDEVICE_EXTENSION pext;
	NTSTATUS st;
	UNICODE_STRING symboliclink;
	UNICODE_STRING devicename;

	
	KdPrint(("DriverEntry is called!!"));
	
	RtlInitUnicodeString(&devicename,L"\\Device\\SimpleDriver");
	RtlInitUnicodeString( &symboliclink, L"\\DosDevices\\SimpleDriver");


	st = IoCreateDevice(pobj,
		    sizeof(DEVICE_EXTENSION),
		    &devicename,
		    FILE_DEVICE_UNKNOWN,
		    0, 
		    TRUE,
		    &pdev);

	if(!NT_SUCCESS(st))
	{
		KdPrint(("Create Device Fault!"));
		return st;
	}

	pext = (PDEVICE_EXTENSION)pdev->DeviceExtension ;
	pext->pDevice = pdev;
	pext->devicename = devicename;

	st = IoCreateSymbolicLink(&symboliclink,&devicename);

	if(!NT_SUCCESS(st))
	{
		KdPrint(("Create Symbolic Link fault!"));
		IoDeleteDevice(pdev);
		return st;
	}

	pobj->MajorFunction[IRP_MJ_READ]=
	pobj->MajorFunction[IRP_MJ_WRITE]=
	pobj->MajorFunction[IRP_MJ_CREATE]=
	pobj->MajorFunction[IRP_MJ_CLOSE]=MyDispath;
	pobj->DriverUnload = MyUnload;

	pext->Symlink = symboliclink;
	KdPrint(("Create Device Success!"));	
	return STATUS_SUCCESS;
}

　　简单解释一下：
    DriverEntry是驱动的入口函数，相当于Win32应用程序的入口函数WinMain跟普通C/C++程序的main。第一个参数为DRIVER_OBJECT类型的指针，是保存着驱动相关参数的结构。第二个参数为一个UNICODE_STRING的指针，保存着驱动写进注册表的相关信息。
    IoCreateDevice函数用于生成一个设备对象，一个驱动可以有多个设备对象。值得注意的是IoCreateDevice第二个参数可以指定设备扩展结构的大小。设备扩展是用户自定义的结构大小，保存在设备对象中。可以将相关的信息保存在其中而避免使用全局变量。
    用户空间的程序可以通过符号链接来打开设备，而IoCreateSymbolicLink函数则是为设备名创建相应的符号链接。设备名只能是"\Device\xxxxx"这样的结构，而符号链接则可以是"\\.\xxxxxx"或者"\DosDevice\xxxxxx"这样的格式。
    注意的一点是，在驱动中没有特殊情况的话，一般都使用Unicode字符。
    在成功生成设备对象后，可以给驱动指定回调函数。在内核中驱动之间的信息是用IRP来传递的。可以将IRP视为Win32应用程序中的消息，而以下的代码：

	pobj->MajorFunction[IRP_MJ_READ]=
	pobj->MajorFunction[IRP_MJ_WRITE]=
	pobj->MajorFunction[IRP_MJ_CREATE]=
	pobj->MajorFunction[IRP_MJ_CLOSE]=MyDispath;
	pobj->DriverUnload = MyUnload;

    则跟MFC应用程序中的消息映射类似。
　  另外KdPrint宏则是用来输出调试信息的函数，跟DbgPrint类似，但在Free模式下编译的话则会被定义为空。
    打开DebugView，随便找些东西将驱动加载了，就能看到输出的调试信息了。