驱动开发之 加载NT式驱动程序实例

最新推荐文章于 2024-02-06 14:06:56 发布
最新推荐文章于 2024-02-06 14:06:56 发布
阅读量1.6k 收藏
点赞数
分类专栏： Driver

使用SCM组件加载NT驱动

下面是Driver.cpp

#include "Driver.h"

/************************************************************************
* 函数名称:DriverEntry
* 功能描述:初始化驱动程序，定位和申请硬件资源，创建内核对象
* 参数列表:
      pDriverObject:从I/O管理器中传进来的驱动对象
      pRegistryPath:驱动程序在注册表的中的路径
* 返回 值:返回初始化驱动状态
*************************************************************************/
#pragma INITCODE
extern "C" NTSTATUS DriverEntry (
			IN PDRIVER_OBJECT pDriverObject,
			IN PUNICODE_STRING pRegistryPath	) 
{
	NTSTATUS status;
	KdPrint(("Enter DriverEntry\n"));

	//注册其他驱动调用函数入口
	pDriverObject->DriverUnload = HelloDDKUnload;
	pDriverObject->MajorFunction[IRP_MJ_CREATE] = HelloDDKDispatchRoutine;
	pDriverObject->MajorFunction[IRP_MJ_CLOSE] = HelloDDKDispatchRoutine;
	pDriverObject->MajorFunction[IRP_MJ_WRITE] = HelloDDKDispatchRoutine;
	pDriverObject->MajorFunction[IRP_MJ_READ] = HelloDDKDispatchRoutine;
	
	//创建驱动设备对象
	status = CreateDevice(pDriverObject);

	KdPrint(("Leave DriverEntry\n"));
	return status;
}

/************************************************************************
* 函数名称:CreateDevice
* 功能描述:初始化设备对象
* 参数列表:
      pDriverObject:从I/O管理器中传进来的驱动对象
* 返回 值:返回初始化状态
*************************************************************************/
#pragma INITCODE
NTSTATUS CreateDevice (
		IN PDRIVER_OBJECT	pDriverObject) 
{
	NTSTATUS status;
	PDEVICE_OBJECT pDevObj;
	PDEVICE_EXTENSION pDevExt;
	
	//创建设备名称
	UNICODE_STRING devName;
	RtlInitUnicodeString(&devName,L"\\Device\\MyDDKDevice");
	
	//创建设备
	status = IoCreateDevice( pDriverObject,
						sizeof(DEVICE_EXTENSION),
						&(UNICODE_STRING)devName,
						FILE_DEVICE_UNKNOWN,
						0, TRUE,
						&pDevObj );
	if (!NT_SUCCESS(status))
		return status;

	pDevObj->Flags |= DO_BUFFERED_IO;
	pDevExt = (PDEVICE_EXTENSION)pDevObj->DeviceExtension;
	pDevExt->pDevice = pDevObj;
	pDevExt->ustrDeviceName = devName;
	//创建符号链接
	UNICODE_STRING symLinkName;
	RtlInitUnicodeString(&symLinkName,L"\\??\\HelloDDK");
	pDevExt->ustrSymLinkName = symLinkName;
	status = IoCreateSymbolicLink( &symLinkName,&devName );
	if (!NT_SUCCESS(status)) 
	{
		IoDeleteDevice( pDevObj );
		return status;
	}
	return STATUS_SUCCESS;
}

/************************************************************************
* 函数名称:HelloDDKUnload
* 功能描述:负责驱动程序的卸载操作
* 参数列表:
      pDriverObject:驱动对象
* 返回 值:返回状态
*************************************************************************/
#pragma PAGEDCODE
VOID HelloDDKUnload (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 );
	}
	KdPrint(("Leave DriverUnload\n"));
}

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

#include <windows.h>  
#include <winsvc.h>  
#include <conio.h>  
#include <stdio.h>

#define DRIVER_NAME "HelloDDK"


#define DRIVER_PATH "..\\..\\MyDriver\\MyDriver_Check\\HelloDDK.sys"

//装载NT驱动程序
BOOL LoadNTDriver(char* lpszDriverName,char* lpszDriverPath)
{
	char szDriverImagePath[256];
	//得到完整的驱动路径
	GetFullPathName(lpszDriverPath, 256, szDriverImagePath, NULL);
	printf("szDriverImagePath:%s\n",szDriverImagePath);
	BOOL bRet = FALSE;

	SC_HANDLE hServiceMgr=NULL;//SCM管理器的句柄
	SC_HANDLE hServiceDDK=NULL;//NT驱动程序的服务句柄

	//打开服务控制管理器
	hServiceMgr = OpenSCManager( NULL, NULL, SC_MANAGER_ALL_ACCESS );

	if( hServiceMgr == NULL )  
	{
		//OpenSCManager失败
		printf( "OpenSCManager() Faild %d ! \n", GetLastError() );
		bRet = FALSE;
		goto BeforeLeave;
	}
	else
	{
		OpenSCManager成功
		printf( "OpenSCManager() ok ! \n" );  
	}

	//创建驱动所对应的服务
	hServiceDDK = CreateService( hServiceMgr,
		lpszDriverName, //驱动程序的在注册表中的名字  
		lpszDriverName, // 注册表驱动程序的 DisplayName 值  
		SERVICE_ALL_ACCESS, // 加载驱动程序的访问权限  
		SERVICE_KERNEL_DRIVER,// 表示加载的服务是驱动程序  
		SERVICE_DEMAND_START, // 注册表驱动程序的 Start 值  
		SERVICE_ERROR_IGNORE, // 注册表驱动程序的 ErrorControl 值  
		szDriverImagePath, // 注册表驱动程序的 ImagePath 值  
		NULL,  
		NULL,  
		NULL,  
		NULL,  
		NULL);  

	DWORD dwRtn;
	//判断服务是否失败
	if( hServiceDDK == NULL )  
	{  
		dwRtn = GetLastError();
		if( dwRtn != ERROR_IO_PENDING && dwRtn != ERROR_SERVICE_EXISTS )  
		{  
			//由于其他原因创建服务失败
			printf( "CrateService() Faild %d ! \n", dwRtn );  
			bRet = FALSE;
			goto BeforeLeave;
		}  
		else  
		{
			//服务创建失败，是由于服务已经创立过
			printf( "CrateService() Faild Service is ERROR_IO_PENDING or ERROR_SERVICE_EXISTS! \n" );  
		}

		// 驱动程序已经加载，只需要打开  
		hServiceDDK = OpenService( hServiceMgr, lpszDriverName, SERVICE_ALL_ACCESS );  
		if( hServiceDDK == NULL )  
		{
			//如果打开服务也失败，则意味错误
			dwRtn = GetLastError();  
			printf( "OpenService() Faild %d ! \n", dwRtn );  
			bRet = FALSE;
			goto BeforeLeave;
		}  
		else 
		{
			printf( "OpenService() ok ! \n" );
		}
	}  
	else  
	{
		printf( "CrateService() ok ! \n" );
	}

	//开启此项服务
	bRet= StartService( hServiceDDK, NULL, NULL );  
	if( !bRet )  
	{  
		DWORD dwRtn = GetLastError();  
		if( dwRtn != ERROR_IO_PENDING && dwRtn != ERROR_SERVICE_ALREADY_RUNNING )  
		{  
			printf( "StartService() Faild %d ! \n", dwRtn );  
			bRet = FALSE;
			goto BeforeLeave;
		}  
		else  
		{  
			if( dwRtn == ERROR_IO_PENDING )  
			{  
				//设备被挂住
				printf( "StartService() Faild ERROR_IO_PENDING ! \n");
				bRet = FALSE;
				goto BeforeLeave;
			}  
			else  
			{  
				//服务已经开启
				printf( "StartService() Faild ERROR_SERVICE_ALREADY_RUNNING ! \n");
				bRet = TRUE;
				goto BeforeLeave;
			}  
		}  
	}
	bRet = TRUE;
//离开前关闭句柄
BeforeLeave:
	if(hServiceDDK)
	{
		CloseServiceHandle(hServiceDDK);
	}
	if(hServiceMgr)
	{
		CloseServiceHandle(hServiceMgr);
	}
	return bRet;
}

//卸载驱动程序  
BOOL UnloadNTDriver( char * szSvrName )  
{
	BOOL bRet = FALSE;
	SC_HANDLE hServiceMgr=NULL;//SCM管理器的句柄
	SC_HANDLE hServiceDDK=NULL;//NT驱动程序的服务句柄
	SERVICE_STATUS SvrSta;
	//打开SCM管理器
	hServiceMgr = OpenSCManager( NULL, NULL, SC_MANAGER_ALL_ACCESS );  
	if( hServiceMgr == NULL )  
	{
		//带开SCM管理器失败
		printf( "OpenSCManager() Faild %d ! \n", GetLastError() );  
		bRet = FALSE;
		goto BeforeLeave;
	}  
	else  
	{
		//带开SCM管理器失败成功
		printf( "OpenSCManager() ok ! \n" );  
	}
	//打开驱动所对应的服务
	hServiceDDK = OpenService( hServiceMgr, szSvrName, SERVICE_ALL_ACCESS );  

	if( hServiceDDK == NULL )  
	{
		//打开驱动所对应的服务失败
		printf( "OpenService() Faild %d ! \n", GetLastError() );  
		bRet = FALSE;
		goto BeforeLeave;
	}  
	else  
	{  
		printf( "OpenService() ok ! \n" );  
	}  
	//停止驱动程序，如果停止失败，只有重新启动才能，再动态加载。  
	if( !ControlService( hServiceDDK, SERVICE_CONTROL_STOP , &SvrSta ) )  
	{  
		printf( "ControlService() Faild %d !\n", GetLastError() );  
	}  
	else  
	{
		//打开驱动所对应的失败
		printf( "ControlService() ok !\n" );  
	}  
	//动态卸载驱动程序。  
	if( !DeleteService( hServiceDDK ) )  
	{
		//卸载失败
		printf( "DeleteSrevice() Faild %d !\n", GetLastError() );  
	}  
	else  
	{  
		//卸载成功
		printf( "DelServer:eleteSrevice() ok !\n" );  
	}  
	bRet = TRUE;
BeforeLeave:
//离开前关闭打开的句柄
	if(hServiceDDK)
	{
		CloseServiceHandle(hServiceDDK);
	}
	if(hServiceMgr)
	{
		CloseServiceHandle(hServiceMgr);
	}
	return bRet;	
} 

void TestDriver()
{
	//测试驱动程序  
	HANDLE hDevice = CreateFile("\\\\.\\HelloDDK",  
		GENERIC_WRITE | GENERIC_READ,  
		0,  
		NULL,  
		OPEN_EXISTING,  
		0,  
		NULL);  
	if( hDevice != INVALID_HANDLE_VALUE )  
	{
		printf( "Create Device ok ! \n" );  
	}
	else  
	{
		printf( "Create Device faild %d ! \n", GetLastError() );  
	}
	CloseHandle( hDevice );
} 

int main(int argc, char* argv[])  
{
	getchar();
	//加载驱动
	BOOL bRet = LoadNTDriver(DRIVER_NAME,DRIVER_PATH);
	if (!bRet)
	{
		printf("LoadNTDriver error\n");
		getchar(); 

		UnloadNTDriver(DRIVER_NAME);
		//getchar();
		return 0;
	}
	//加载成功

	printf( "press any to create device!\n" );  
	getchar();  

	TestDriver();

	//这时候你可以通过注册表，或其他查看符号连接的软件验证。  
	printf( "press any to unload the driver!\n" );  
	getchar();  

	//卸载驱动
	UnloadNTDriver(DRIVER_NAME);
	if (!bRet)
	{
		printf("UnloadNTDriver error\n");
		getchar(); 
		return 0;
	}
	getchar(); 
	return 0;  
}

用到的函数介绍： http://blog.csdn.net/liyun123gx/article/details/37599357

细粉条
关注
专栏目录
NT式驱动加载器
曾是土木人
06-25 3067
引言 设备驱动程序的动态加载主要是由服务控制管理程序（Service Control Manager，SCM）系统组件完成，SCM组件可以提供许多服务，如，启动、停止和控制服务。 编写加载驱动程序，主要是操作SCM组件。 一般，操作SCM组件的相关API函数有如下几个： SC_HANDLE WINAPI OpenSCManager(   __in_opt  LPCTSTR lpM
NT式驱动程序
raiky的专栏
07-01 1770
<br />对于NT式驱动程序，主要的函数是DriverEntry例程、卸载程序以及各个IRP的派遣函数，基本结构如下：<br /> <br />#include "Driver.h"<br /> <br />#pragma INITCODE<br />extern "C" NTSTATUS DriverEntry (IN PDRIVER_OBJECT pDriverObject<br />                                                      IN PU
NT式驱动三种加载方式详解（工具+手动+代码）
01-15
本文为自己整理的NT式驱动加载方式，主要是通过学习《驱动开发技术详解》所做的笔记和实验，详细介绍了工具加载、手动加载以及代码加载三种NT式驱动加载方式。并附有自己截取的图片以及编写的代码。
NT式驱动的动态加载
jhy8421的专栏
03-06 708
以下均为学习后的个人见解，若果有误，欢迎纠正，谢谢 NT式驱动的加载分为手动加载和动态加载，因为本片文章主要讲解动态加载，所以大体概念讲解一下手动加载 手动加载： 第一步是在注册表中添加新项，路径为HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Services\驱动名称 第二步是添加子键，DisplayName、ErrorControl、Group
win驱动开发笔记-驱动加载
weixin_30265103的博客
04-11 252
手动加载 代码加载(win32) /*安装驱动程序流程： 1、调用OpenSCManager()打开服务控制管理器 2、调用CreateService()创建一个服务，服务类型为内核驱动 3、调用OpenService()取得服务句柄 启动服务 4、调用StartService()启动服务 停止服务 4、调用ControlService()停止...
NT驱动加载源码-2018.rar_NT驱动加载_驱动
09-23
在Windows操作系统中，NT驱动程序是系统核心层的软件组件，它们允许硬件设备与操作系统进行交互。本资源“NT驱动加载源码-2018.rar”提供了关于NT驱动加载的源代码，对于想要深入理解驱动开发或者进行相关研究的人来...
Windows驱动开发系列之一：小白入门经典.pdf
11-09
- **第一个NT式驱动程序**: - 使用`ntifs.h`头文件。 - `UnloadDriver`: 卸载驱动程序的回调函数。 - `DriverEntry`: 相当于驱动程序的入口点，初始化驱动程序并设置卸载回调函数。 - **禁用驱动程序强制签名**:...
NT式驱动加载工具 - SYSLoader
01-12
SYSLoader是一款基于MFC（Microsoft Foundation Classes）开发的NT式驱动加载工具，主要目标是帮助开发者和研究人员在Windows操作系统环境中方便地加载和测试内核模式驱动程序。NT式驱动指的是遵循Windows NT内核...
驱动开发之 加载NT式驱动
Lydia的博客
07-09 1187
设备驱动程序的动态加载主要由服务控制飞、管理che
WDM_Driver.zip_PCI驱动_VisualDDK_ddk pci_pci驱动开发_wdm pci驱动开发
09-23
例如，DriverEntry函数是驱动程序的入口点，它在驱动加载时被调用，用于初始化驱动；IRP（I/O Request Packet）则是驱动处理设备请求的主要方式，通过完成函数来响应系统或用户的应用请求。 此外，对于PCI设备，还...
一个简单的WDM驱动例子
11-21
一个简单的WDM驱动例子，包括驱动和用户模式的调用测试。（例子来自于《windows驱动开发技术详解》，只是稍微做了一些改变）
64位驱动加载工具(NT式驱动)
05-28
64位驱动加载工具NT式驱动，直接把驱动拖入窗体，点击确认即可加载,
动态加载驱动程序源代码
06-14
源代码包含三种不同的驱动加载方法,使用ZwSetSystemInformation函数加载驱动,使用NtLoadDriver函数加载驱动,使用服务控制管理器加载驱动,还包括三种线程注入技术,使用RtlCreateUserThread 函数注入线程,使用CreateRemoteThread函数注入线程,使用NtCreateThreadEx函数注入线程,源代码包含C,C#的demo
32位/64位WINDOWS驱动开发之驱动加载
a756598009的博客
06-17 833
加载驱动的过程1、用OpenSCManager 打开服务控制管理器2、用CreateService创建对应服务3、如果驱动服务已经创建过，则用OpenService打开服务4、用StartService加载启动驱动服务5、清理工作，用CloseServiceHandle关闭释放句柄MFC操作如下应用层：首先加两个按钮 加载驱动 ID IDC_BUTTON2_LOAD_SYS卸载驱动 ID IDC_BUTTON1_UNLOAD_SYS。
驱动开发：实现驱动加载卸载工具
最新发布
rjc_lihui的专栏
02-06 129
工具，如果驱动程序需要对外发布那我们必须自己编写实现一个驱动加载工具，当需要使用驱动时可以拉起自己的驱动，如下将实现一个简单的驱动加载工具，该工具可以实现基本的，安装，加载，关闭，卸载等操作日常使用完全没问题。本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。驱动程序加载工具有许多，最常用的当属。
NT式驱动
收集学习过程中对自己有帮助的牛人文章
11-29 644
转载自：http://mzf2008.blog.163.com/blog/static/35599786201011733440667/ //Driver.h  #pragma once #ifdef __cplusplus  extern "C" { #endif #include //NT式驱动包含此头文件 #ifdef __cplusplus
驱动模块编译及挂载的问题,简单makefile的编写
mmpire之blog
08-06 1728
驱动模块编译: 在gcc命令行中加入-I /usr/src/linux-2.4.22-1.2115/include/ 如:gcc -Wall -c -O2 -I/usr/src/linux-2.4.22-1.2115/include/ a.c (注:-I与目录之间的空格可有可无) 要想去掉编译警告信息"no license",在include的下一行加上 MODULE_LICENSE(
NT式驱动的基本结构1 - 驱动加载过程与驱动入口函数
wendyWJGU的博客
01-26 541
NT式驱动的基本结构 一
NT式驱动基本结构
BpLife
12-04 1156
1.对于NT式驱动来说，主要的函数是DriveEntry函数，卸载函数，以及各个IRP的派遣函数 2.驱动加载过程与驱动入口函数（DriverEntry）    NTSTATUS DriveEntry(IN PDRIVER_OBJECT pDriverobject,   IN PUNICODE_STRING pRegistryPath);//函数名可以自己定义 DriverEntry主
大数据驱动的嵌入式音乐识别软件
代码中的`#include "driver.h"`和`#include "driver.tmh"`暗示这部分可能涉及到驱动程序的开发，如`DriverEntry`函数是驱动程序初始化的关键部分，它定义了驱动加载时的操作。 2. **内存管理**: `#ifdefALLOC_...
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值