0
回答
菜鸟之驱动开发5
利用AWS快速构建适用于生产的无服务器应用程序,免费试用12个月>>>   
 

在前面4节中,我们学习了一个驱动的基本写法,也算是驱动已入门了。今天要介绍的就是在应用层加载与卸载驱动,有两种方法:1.修改注册表,2.用SCM-Service Control Manager管理驱动。我们今天学习第二种方法,主要学习如何加载驱动与卸载驱动。

我用的是VS2010的开发环境,创建一个基于MFC 对话框的工程。

拖两个BUTTON,一个用于加载驱动,一个用于卸载驱动。

 

加载驱动代码如下:

//装载NT驱动程序
BOOL LoadNTDriver(LPCTSTR lpDriverName,LPCTSTR lpDriverPathName)
{
	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失败
		TRACE( "OpenSCManager() Faild %d ! \n", GetLastError() );
		bRet = FALSE;
		goto BExit;
	}
	else
	{
		////OpenSCManager成功
		TRACE( "OpenSCManager() ok ! \n" );  
	}

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

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

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

	//开启此项服务
	bRet= StartService( hServiceDDK, NULL, NULL );  
	if( !bRet )  //开启服务不成功
	{  
		TRACE( "StartService() 失败 服务可能已经开启%d ! \n", dwRtn );  
	}
	bRet = TRUE;
	//离开前关闭句柄
BExit:
	if(hServiceDDK)
	{
		CloseServiceHandle(hServiceDDK);
	}
	if(hServiceMgr)
	{
		CloseServiceHandle(hServiceMgr);
	}
	return bRet;
}
//--------------------------------------------------------------------------


卸载代码如下:

//卸载驱动程序  
BOOL UnLoadSys( LPCTSTR szSvrName )  
{
	//一定义所用到的变量
	BOOL bRet = FALSE;
	SC_HANDLE hSCM=NULL;//SCM管理器的句柄,用来存放OpenSCManager的返回值
	SC_HANDLE hService=NULL;//NT驱动程序的服务句柄,用来存放OpenService的返回值
	SERVICE_STATUS SvrSta;
	//二打开SCM管理器
	hSCM = OpenSCManager( NULL, NULL, SC_MANAGER_ALL_ACCESS );  
	if( hSCM == NULL )  
	{
		//带开SCM管理器失败
		TRACE( "OpenSCManager() Faild %d ! \n", GetLastError() );  
		bRet = FALSE;
		goto BeforeLeave;
	}  
	else  
	{
		//打开SCM管理器成功
		TRACE( "OpenSCManager() ok ! \n" );  
	}
	//三打开驱动所对应的服务
	hService = OpenService( hSCM, szSvrName, SERVICE_ALL_ACCESS );  

	if( hService == NULL )  
	{
		//打开驱动所对应的服务失败 退出
		TRACE( "OpenService() Faild %d ! \n", GetLastError() );  
		bRet = FALSE;
		goto BeforeLeave;
	}  
	else  
	{  
		TRACE( "OpenService() ok ! \n" );  //打开驱动所对应的服务 成功
	}  
	//四停止驱动程序,如果停止失败,只有重新启动才能,再动态加载。  
	if( !ControlService( hService, SERVICE_CONTROL_STOP , &SvrSta ) )  
	{  
		TRACE( "用ControlService() 停止驱动程序失败 错误号:%d !\n", GetLastError() );  
	}  
	else  
	{
		//停止驱动程序成功
		TRACE( "用ControlService() 停止驱动程序成功 !\n" );  
	}  
	//五动态卸载驱动服务。  
	if( !DeleteService( hService ) )  //TRUE//FALSE
	{
		//卸载失败
		TRACE( "卸载失败:DeleteSrevice()错误号:%d !\n", GetLastError() );  
	}  
	else  
	{  
		//卸载成功
		TRACE ( "卸载成功 !\n" );  

	}  
	bRet = TRUE;
	//六 离开前关闭打开的句柄
BeforeLeave:
	if(hService>0)
	{
		CloseServiceHandle(hService);
	}
	if(hSCM>0)
	{
		CloseServiceHandle(hSCM);
	}
	return bRet;	
}


代码里有详细注释,很容易明白。代码来自郁金香老师教程。要用SCM的函数需要包含一个头文件:

#include <WinSvc.h>

 

因为我们是MFC的工程, 默认编译出来的可执行文件在没有MFC动态库的系统中是无法远行的,所以我把工程修改为静态连接的方式,以减速少对环境的依赖。

修改如下:

项目-》配置属性-》常规:MFC的使用  选择:在静态库中使用MFC

项目-》配置属性-》C/C++-》生成代码:运行库  选择:多线程调试(MTD)


具体参看完整源码


原文链接:http://blog.csdn.net/favormm/article/details/6699941
<无标签>
举报
长平狐
发帖于5年前 0回/89阅
顶部