MFC 驱动加载工具（安装/运行/停止/卸载）

题外话

一、了解驱动

二、驱动加载流程

2.1 加载和卸载NT驱动的4个步骤

2.2 对应函数实现流程和代码

三、一些注意事项

题外话

虽然网上已经有很多驱动加载工具和相关代码，但自己实现一次有助于理解如何使用代码实现驱动加载。因此本篇文章只是一个学习总结。再说点题外话，不管之前耗费多久时间完成一个项目，事后最好写个总结，后期可以自己翻翻，重温一下~因为我之前完成任何一个项目，都是不懂就搜，但下次遇到相同问题时还是得上网搜(⊙o⊙)…所以想试试这种总结方法会不会让我下次遇到相同问题就不要搜百度，来搜自己博客就好啦~O(∩_∩)O哈哈~

罗里吧嗦了一堆，那开始进入正题吧！(〃'▽'〃)

一、了解驱动

如果对驱动已经有了初步的认识的同学可以跳过这部分~因为我当时是对“驱动”这个概念有点懵d(´ω｀*)所以这里做个简单介绍，目的是加深自己对驱动的理解~

驱动程序是一个软件组件，可让操作系统和设备彼此通信。即驱动程序是设备驱动程序，是一种可以使计算机和设备通信的特殊程序。相当于硬件的接口，操作系统只有通过这个接口，才能控制硬件设备的工作，假如某设备的驱动程序未能正确安装，便不能正常工作。如下图所示：

驱动文件一般由扩展名 CAT、 INF、 SYS、 DLL 等文件格式组成的。

二、驱动加载流程

编写代码加载驱动的原理相当于在注册表中写入相应的字段，因此设备驱动程序的动态加载主要由SCM（服务控制管理程序）系统组件完成。

2.1 加载和卸载NT驱动的4个步骤

Step1：安装——为NT驱动创建新的服务：LoadNTDriver（自定义的函数名）

Step2：运行——开启该服务：StartNTDriver（自定义的函数名）

Step3：停止——关闭此服务：StopNTDriver（自定义的函数名）

Step4：卸载——删除NT驱动所创建的服务：UnLoadNTDriver（自定义的函数名）

2.2 对应函数实现流程和代码

2.2.1 安装【LoadNTDriver】

流程图：

代码如下：

//装载NT驱动程序
BOOL CLoadNtDriver::loadNTDriver(CString strDriverPath)
{wchar_t  szDriverImagePath[256];CString  strDriverImagePath;DWORD   dwRtn;//得到完整的驱动路径GetFullPathName(strDriverPath, 256, szDriverImagePath, NULL);strDriverImagePath = szDriverImagePath;if (strDriverImagePath.IsEmpty()){MessageBoxW(0, L" CLoadDriver::installDriver: 请输入路径名", NULL, MB_OK | MB_ICONWARNING);return FALSE;}if (!pathIsFile(strDriverImagePath)){strDriverImagePath += L" loadNTDriver:文件不存在";MessageBoxW(0, strDriverImagePath, NULL, MB_OK | MB_ICONWARNING);return FALSE;}// 提取驱动名wchar_t szServerName[40], szShowName[40];getDriverName(strDriverImagePath, szServerName, szShowName);//打开服务控制管理器m_hServiceMgr = OpenSCManager(NULL, NULL,  SC_MANAGER_CONNECT|SC_MANAGER_CREATE_SERVICE|SC_MANAGER_ENUMERATE_SERVICE);if (m_hServiceMgr == NULL){//OpenSCManager失败showErrorInfo(GetLastError(), __LINE__, L"loadNTDriver");return FALSE;}else{//OpenSCManager成功showErrorInfo(GetLastError(), __LINE__, L"loadNTDriver()");}//创建驱动所对应的服务m_hServiceDDK = CreateService(m_hServiceMgr,szServerName,                //驱动程序的在注册表中的名字  szShowName,                // 注册表驱动程序的 DisplayName 值  SERVICE_ALL_ACCESS,            // 加载驱动程序的访问权限  SERVICE_KERNEL_DRIVER,        // 表示加载的服务是驱动程序  SERVICE_DEMAND_START,        // 注册表驱动程序的 Start 值  SERVICE_ERROR_IGNORE,        // 注册表驱动程序的 ErrorControl 值  strDriverImagePath,            // 注册表驱动程序的 ImagePath 值  NULL,NULL,NULL,NULL,NULL);//判断服务是否失败if (m_hServiceDDK == NULL){dwRtn = GetLastError();if (dwRtn != ERROR_IO_PENDING && dwRtn != ERROR_SERVICE_EXISTS){//由于其他原因创建服务失败showErrorInfo(GetLastError(), __LINE__, L"loadNTDriver()");releaseHandle();return FALSE;}else{//服务创建失败，是由于服务已经创立过showErrorInfo(GetLastError(), __LINE__, L"loadNTDriver()");m_hServiceDDK = OpenService(m_hServiceMgr, szServerName, SERVICE_ALL_ACCESS);}}return TRUE;
}

2.2.2 运行【StartNTDriver】

流程图：

PS：其实启动驱动就是开启在注册表中注册的该项服务~

代码如下：

// 启动驱动
BOOL CLoadNtDriver::startNTDriver( CString strDriverPath )
{//开启此项服务if (!StartService(m_hServiceDDK, NULL, NULL)){DWORD dwRtn = GetLastError();if (dwRtn != ERROR_IO_PENDING && dwRtn != ERROR_SERVICE_ALREADY_RUNNING){showErrorInfo(GetLastError(), __LINE__, L"startNTDriver()");releaseHandle();return FALSE;}else{if (dwRtn == ERROR_IO_PENDING){//设备被挂住showErrorInfo(GetLastError(), __LINE__, L"startNTDriver()");releaseHandle();return FALSE;}}}//服务已经开启showErrorInfo(GetLastError(), __LINE__, L"startNTDriver()");return TRUE;
}

2.2.3 停止【StopNTDriver】

代码如下：

// 停止驱动
BOOL CLoadNtDriver::stopNTDriver( CString strDrivePath )
{//停止驱动服务SERVICE_STATUS svcsta = { 0 };if(!ControlService(m_hServiceDDK, SERVICE_CONTROL_STOP, &svcsta)){showErrorInfo(GetLastError(), __LINE__, L"stopNTDriver()");releaseHandle();return FALSE;}//服务已经停止showErrorInfo(GetLastError(), __LINE__, L"stopNTDriver()");return TRUE;
}

2.2.1 卸载【UnLoadNTDriver】

代码如下：

// 卸载驱动
BOOL CLoadNtDriver::unloadNTDriver(CString strDriverPath)
{if (m_hServiceDDK == NULL){//打开驱动所对应的服务失败showErrorInfo(GetLastError(), __LINE__, L"unloadNTDriver()");releaseHandle();return FALSE;}//动态卸载驱动程序。  if (!DeleteService(m_hServiceDDK)){//卸载失败showErrorInfo(GetLastError(), __LINE__, L"unloadNTDriver()");releaseHandle();return FALSE;}showErrorInfo(GetLastError(), __LINE__, L"unloadNTDriver()");releaseHandle();return TRUE;
}

三、一些注意事项

1. 记得释放句柄！释放句柄后还要将句柄置为NULL。如果不释放句柄就会出现只能实现一次安装卸载一个驱动文件，再次点击安装该驱动文件（并没有将该工具关闭重启情况下）就会安装失败~

2.本驱动加载工具是用来加载自己写的驱动文件，不要试图加载/运行/停止/卸载系统的驱动文件！！！我在虚拟机上试过几次加载系统驱动文件，然后……就蓝屏了……(╥╯^╰╥)

该工具后期还会加点其他功能，实现完后再添加上来吧~到时顺便把截图也放上来，完善代码后再将完整代码发上来~

嘻嘻嘻，感谢你看到这(❁´◡`❁)*✲ﾟ*

如果这篇文章对你有帮助的话，可以给个赞哟~Ｏ(≧▽≦)Ｏ