驱动开发：实现驱动加载卸载工具

原创

LyShark 2022-05-23 15:48:23 博主文章分类：《Windows 内核安全编程技术实践》 ©著作权

文章标签 Windows 驱动开发数据库句柄控件 文章分类 C/C++ 后端开发 yyds干货盘点

©著作权归作者所有：来自51CTO博客作者LyShark的原创作品，请联系作者获取转载授权，否则将追究法律责任

驱动程序加载工具有许多，最常用的当属KmdManager工具，如果驱动程序需要对外发布那我们必须自己编写实现一个驱动加载工具，当需要使用驱动时可以拉起自己的驱动，如下将实现一个简单的驱动加载工具，该工具可以实现基本的，安装，加载，关闭，卸载等操作日常使用完全没问题。

installDvr 驱动安装

#include <iostream>
#include <Windows.h>

using namespace std;

// 安装驱动
BOOL installDvr(CONST WCHAR drvPath[50], CONST WCHAR serviceName[20])
{

  // 打开服务控制管理器数据库
  SC_HANDLE schSCManager = OpenSCManager(
    NULL,                   // 目标计算机的名称,NULL：连接本地计算机上的服务控制管理器
    NULL,                   // 服务控制管理器数据库的名称，NULL：打开 SERVICES_ACTIVE_DATABASE 数据库
    SC_MANAGER_ALL_ACCESS   // 所有权限
    );
  
  if (schSCManager == NULL)
  {
    CloseServiceHandle(schSCManager);
    return FALSE;
  }

  // 创建服务对象，添加至服务控制管理器数据库
  SC_HANDLE schService = CreateService
    (
    schSCManager,               // 服务控件管理器数据库的句柄
    serviceName,                // 要安装的服务的名称
    serviceName,                // 用户界面程序用来标识服务的显示名称
    SERVICE_ALL_ACCESS,         // 对服务的访问权限：所有全权限
    SERVICE_KERNEL_DRIVER,      // 服务类型：驱动服务
    SERVICE_DEMAND_START,       // 服务启动选项：进程调用 StartService 时启动
    SERVICE_ERROR_IGNORE,       // 如果无法启动：忽略错误继续运行
    drvPath,                    // 驱动文件绝对路径，如果包含空格需要多加双引号
    NULL,                       // 服务所属的负载订购组：服务不属于某个组
    NULL,                       // 接收订购组唯一标记值：不接收
    NULL,                       // 服务加载顺序数组：服务没有依赖项
    NULL,                       // 运行服务的账户名：使用 LocalSystem 账户
    NULL                        // LocalSystem 账户密码
    );
  
  if (schService == NULL)
  {
    CloseServiceHandle(schService);
    CloseServiceHandle(schSCManager);
    return FALSE;
  }

  CloseServiceHandle(schService);
  CloseServiceHandle(schSCManager);
  return TRUE;
}

int main(int argc, char *argv[])
{
  if (installDvr(L"C:\\WinDDK.sys", L"service") == TRUE)
  {
    cout << "驱动已安装" << endl;
  }

  getchar();
  return 0;
}

startDvr 启动驱动

#include <iostream>
#include <Windows.h>

using namespace std;

// 启动服务
BOOL startDvr(CONST WCHAR serviceName[20])
{

  // 打开服务控制管理器数据库
  SC_HANDLE schSCManager = OpenSCManager
    (
    NULL,                   // 目标计算机的名称,NULL：连接本地计算机上的服务控制管理器
    NULL,                   // 服务控制管理器数据库的名称，NULL：打开 SERVICES_ACTIVE_DATABASE 数据库
    SC_MANAGER_ALL_ACCESS   // 所有权限
    );
  
  if (schSCManager == NULL)
  {
    CloseServiceHandle(schSCManager);
    return FALSE;
  }

  // 打开服务
  SC_HANDLE hs = OpenService
    (
    schSCManager,           // 服务控件管理器数据库的句柄
    serviceName,            // 要打开的服务名
    SERVICE_ALL_ACCESS      // 服务访问权限：所有权限
    );
  
  if (hs == NULL)
  {
    CloseServiceHandle(hs);
    CloseServiceHandle(schSCManager);
    return FALSE;
  }
  
  if (StartService(hs, 0, 0) == 0)
  {
    CloseServiceHandle(hs);
    CloseServiceHandle(schSCManager);
    return FALSE;
  }

  CloseServiceHandle(hs);
  CloseServiceHandle(schSCManager);
  return TRUE;
}

int main(int argc, char *argv[])
{
  if (startDvr(L"service") == TRUE)
  {
    cout << "驱动服务" << endl;
  }

  getchar();
  return 0;
}

stopDvr 停止驱动

#include <iostream>
#include <Windows.h>

using namespace std;

// 停止服务
BOOL stopDvr(CONST WCHAR serviceName[20])
{

  // 打开服务控制管理器数据库
  SC_HANDLE schSCManager = OpenSCManager
    (
    NULL,                   // 目标计算机的名称,NULL：连接本地计算机上的服务控制管理器
    NULL,                   // 服务控制管理器数据库的名称，NULL：打开 SERVICES_ACTIVE_DATABASE 数据库
    SC_MANAGER_ALL_ACCESS   // 所有权限
    );

  if (schSCManager == NULL)
  {
    CloseServiceHandle(schSCManager);
    return FALSE;
  }

  // 打开服务
  SC_HANDLE hs = OpenService
    (
    schSCManager,           // 服务控件管理器数据库的句柄
    serviceName,            // 要打开的服务名
    SERVICE_ALL_ACCESS      // 服务访问权限：所有权限
    );
  
  if (hs == NULL)
  {
    CloseServiceHandle(hs);
    CloseServiceHandle(schSCManager);
    return FALSE;
  }

  // 如果服务正在运行
  SERVICE_STATUS status;
  if (QueryServiceStatus(hs, &status) == 0)
  {
    CloseServiceHandle(hs);
    CloseServiceHandle(schSCManager);
    return FALSE;
  }

  if (status.dwCurrentState != SERVICE_STOPPED &&
    status.dwCurrentState != SERVICE_STOP_PENDING
    )
  {
    // 发送关闭服务请求
    if (ControlService(
      hs,                         // 服务句柄
      SERVICE_CONTROL_STOP,       // 控制码：通知服务应该停止
      &status                     // 接收最新的服务状态信息
      ) == 0) {
      CloseServiceHandle(hs);
      CloseServiceHandle(schSCManager);
      return FALSE;
    }

    // 判断超时
    INT timeOut = 0;
    while (status.dwCurrentState != SERVICE_STOPPED)
    {
      timeOut++;
      QueryServiceStatus(hs, &status);
      Sleep(50);
    }

    if (timeOut > 80)
    {
      CloseServiceHandle(hs);
      CloseServiceHandle(schSCManager);
      return FALSE;
    }
  }

  CloseServiceHandle(hs);
  CloseServiceHandle(schSCManager);
  return TRUE;
}

int main(int argc, char *argv[])
{
  if (stopDvr(L"service") == TRUE)
  {
    cout << "停止驱动服务" << endl;
  }

  getchar();
  return 0;
}

unloadDvr 卸载驱动

#include <iostream>
#include <Windows.h>

using namespace std;

// 卸载驱动
BOOL unloadDvr(CONST WCHAR serviceName[20])
{

  // 打开服务控制管理器数据库
  SC_HANDLE schSCManager = OpenSCManager
    (
    NULL,                   // 目标计算机的名称,NULL：连接本地计算机上的服务控制管理器
    NULL,                   // 服务控制管理器数据库的名称，NULL：打开 SERVICES_ACTIVE_DATABASE 数据库
    SC_MANAGER_ALL_ACCESS   // 所有权限
    );

  if (schSCManager == NULL)
  {
    CloseServiceHandle(schSCManager);
    return FALSE;
  }

  // 打开服务
  SC_HANDLE hs = OpenService
    (
    schSCManager,           // 服务控件管理器数据库的句柄
    serviceName,            // 要打开的服务名
    SERVICE_ALL_ACCESS      // 服务访问权限：所有权限
    );

  if (hs == NULL)
  {
    CloseServiceHandle(hs);
    CloseServiceHandle(schSCManager);
    return FALSE;
  }

  // 删除服务
  if (DeleteService(hs) == 0)
  {
    CloseServiceHandle(hs);
    CloseServiceHandle(schSCManager);
    return FALSE;
  }

  CloseServiceHandle(hs);
  CloseServiceHandle(schSCManager);
  return TRUE;
}

int main(int argc, char *argv[])
{
  if (unloadDvr(L"service") == TRUE)
  {
    cout << "卸载驱动服务" << endl;
  }

  getchar();
  return 0;
}