注册 登录  
 加关注
   显示下一条  |  关闭
温馨提示!由于新浪微博认证机制调整,您的新浪微博帐号绑定已过期,请重新绑定!立即重新绑定新浪微博》  |  关闭

从C开始

 
 
 

日志

 
 

NT式驱动  

2010-12-07 15:34:40|  分类: 驱动编程 |  标签: |举报 |字号 订阅

  下载LOFTER 我的照片书  |
//Driver.h 

#pragma once

#ifdef __cplusplus 
extern "C"
{
#endif
#include <ntddk.h> //NT式驱动包含此头文件
#ifdef __cplusplus
}
#endif

#define PAGEDCODE code_seg("PAGE") 
#define LOCKEDCODE code_seg()
#define INITCODE code_sg("INIT")

#define PAGEDDATA data_seg("PAGE")
#define LOCKEDDATA data_seg()
#define INITDATA data_seg("INIT")

#define arraysize(a) (sizeof(a)/sizeof((a)[1])) //数组元素个数

typedef struct _DEVICE_EXTENSION  //定义设备扩展结构体
{
PDEVICE_OBJECT pDeviceObj; //设备
UNICODE_STRING ustrDeviceName; //设备名称
UNICODE_STRING ustrSymLinkName; //链接符号名称
} DEVICE_EXTENSION, *PDEVICE_EXTENSION;

//函数声明
 NTSTATUS CreateDevice(IN PDRIVER_OBJECT pDriberObj); //创建设备函数
 VOID HelloDDKUnload(IN PDRIVER_OBJECT pDriverObj); //驱动卸载例程
 NTSTATUS HelloDDKDispatchRoutine(IN PDEVICE_OBJECT pDeviceObj, IN PIRP pIrp); //派遣例程



//Driver.cpp

#include "Driver.h"

/************************************************************************
* 函数名称:DriverEntry
* 功能描述:初始化驱动程序,定位和申请硬件资源,创建内核对象
* 参数列表:
      pDriverObj:从I/O管理器中传进来的驱动对象
      pRegisterPath:驱动程序在注册表的中的路径
* 返回 值:返回初始化驱动状态
*************************************************************************/
#pragma INITCODE
extern "C" NTSTATUS DriverEntry( IN PDRIVER_OBJECT pDriverObj, IN PUNICODE_STRING pRegisterPath )
{
KdPrint(("Entry DriverEntry!\n"));
//注册其他驱动调用的函数入口
pDriverObj->DriverUnload = HelloDDKUnload;
pDriverObj->MajorFunction[IRP_MJ_CREATE] = HelloDDKDispatchRoutine;
pDriverObj->MajorFunction[IRP_MJ_CLOSE] = HelloDDKDispatchRoutine;
pDriverObj->MajorFunction[IRP_MJ_WRITE] = HelloDDKDispatchRoutine;
pDriverObj->MajorFunction[IRP_MJ_READ] = HelloDDKDispatchRoutine;

NTSTATUS status;
status = CreateDevice(pDriverObj); //创建设备对象
KdPrint(("Leave DriverEntry!\n"));
return STATUS_SUCCESS;
}

/************************************************************************
* 函数名称:CreateDevice
* 功能描述:初始化设备对象
* 参数列表:
      pDriverObject:从I/O管理器中传进来的驱动对象
* 返回 值:返回初始化状态
*************************************************************************/
#pragma INITCODE
NTSTATUS CreateDevice(IN PDRIVER_OBJECT pDriberObj)
{
KdPrint(("Entry CreateDevice!\n"));
NTSTATUS status;
PDEVICE_OBJECT pDeviceObj;
PDEVICE_EXTENSION pDeviceExtension;

UNICODE_STRING ustrDeviceName;
RtlInitUnicodeString(&ustrDeviceName, L"\\Device\\MyDDKDevice"); //要创建的设备对象名称
//创建设备对象
status = IoCreateDevice(pDriberObj, sizeof(DEVICE_EXTENSION), &(UNICODE_STRING)ustrDeviceName,
FILE_DEVICE_UNKNOWN, 0, TRUE, &pDeviceObj);

if (!NT_SUCCESS(status))
{
return status;
}
pDeviceObj->Flags |= DO_BUFFERED_IO;
pDeviceExtension = (PDEVICE_EXTENSION)pDeviceObj->DeviceExtension; //填充设备扩展结构各个字段
pDeviceExtension->pDeviceObj = pDeviceObj;
pDeviceExtension->ustrDeviceName = ustrDeviceName;
UNICODE_STRING ustrSymLinkName;
RtlInitUnicodeString(&ustrSymLinkName, L"\\??\\HelloDDK");//符号链接名
pDeviceExtension->ustrSymLinkName = ustrSymLinkName;
//创建符号链接
status = IoCreateSymbolicLink(&ustrSymLinkName, &ustrDeviceName);
if (!NT_SUCCESS(status))
{
IoDeleteDevice(pDeviceObj);
return status;
}
KdPrint(("Leave CreateDevice!\n"));
return STATUS_SUCCESS;
}

/************************************************************************
* 函数名称:HelloDDKUnload
* 功能描述:负责驱动程序的卸载操作
* 参数列表:
      pDriverObj:驱动对象
* 返回 值:返回状态
*************************************************************************/
#pragma PAGEDCODE
VOID HelloDDKUnload(IN PDRIVER_OBJECT pDriverObj)
{
KdPrint(("Entry UnloadDriver!\n"));
PDEVICE_OBJECT pNextDeviceObj;
pNextDeviceObj = pDriverObj->DeviceObject;
//遍历删除驱动的设备对象
while (pNextDeviceObj != NULL)
{
PDEVICE_EXTENSION pDeviceExten = (PDEVICE_EXTENSION)pNextDeviceObj->DeviceExtension;
UNICODE_STRING ustrDelSymlinkName = pDeviceExten->ustrSymLinkName; 
IoDeleteSymbolicLink(&ustrDelSymlinkName);//删除符号链接
pNextDeviceObj = pNextDeviceObj->NextDevice;
IoDeleteDevice(pDeviceExten->pDeviceObj);//删除设备对象
}
KdPrint(("Leave UnloadDriver!\n"));
}

/************************************************************************
* 函数名称:HelloDDKDispatchRoutine
* 功能描述:对读IRP进行处理
* 参数列表:
      pDeviceObj:功能设备对象
      pIrp:从IO请求包
* 返回 值:返回状态
*************************************************************************/
#pragma PAGEDCODE
NTSTATUS HelloDDKDispatchRoutine(IN PDEVICE_OBJECT pDeviceObj, IN PIRP pIrp)
{
KdPrint(("Enter HelloDDKDispatchRoutine!\n"));
NTSTATUS status = STATUS_SUCCESS;
pIrp->IoStatus.Status = status;
pIrp->IoStatus.Information = 0;
IoCompleteRequest(pIrp, IO_NO_INCREMENT); //设置IRP请求为完成状态
KdPrint(("Leave HelloDDKDispatchRoutine\n"));
return status;
}


//sources
TARGETNAME=MzfFirstDriver
TARGETTYPE=DRIVER
TARGETPATH=OBJ

INCLUDES=$(BASEDIR)\inc;\
         $(BASEDIR)\inc\ddk;\

SOURCES=Driver.cpp \

  评论这张
 
阅读(481)| 评论(0)
推荐 转载

历史上的今天

评论

<#--最新日志,群博日志--> <#--推荐日志--> <#--引用记录--> <#--博主推荐--> <#--随机阅读--> <#--首页推荐--> <#--历史上的今天--> <#--被推荐日志--> <#--上一篇,下一篇--> <#-- 热度 --> <#-- 网易新闻广告 --> <#--右边模块结构--> <#--评论模块结构--> <#--引用模块结构--> <#--博主发起的投票-->
 
 
 
 
 
 
 
 
 
 
 
 
 
 

页脚

网易公司版权所有 ©1997-2018