在前面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; } //--------------------------------------------------------------------------
卸载代码如下:
代码里有详细注释,很容易明白。代码来自郁金香老师教程。要用SCM的函数需要包含一个头文件:
因为我们是MFC的工程, 默认编译出来的可执行文件在没有MFC动态库的系统中是无法远行的,所以我把工程修改为静态连接的方式,以减速少对环境的依赖。
修改如下:
项目-》配置属性-》常规:MFC的使用 选择:在静态库中使用MFC
项目-》配置属性-》C/C++-》生成代码:运行库 选择:多线程调试(MTD)
具体参看完整源码。