驱动加载类
需要管理员权限
#pragma once typedef struct _tagColse_NetWork { BOOLEAN m_bColseOrNot; }ST_COLSE_NETWORK, *PST_COLSE_NETWORK; class CLoadSys { public: CLoadSys(); ~CLoadSys(); BOOL LoadDriver(WCHAR * lpszDriverName, WCHAR * lpszDriverPath); BOOL UnloadDriver(WCHAR * szSvrName); };
#include "stdafx.h" #include "LoadSys.h" #include <winsvc.h> CLoadSys::CLoadSys() { } CLoadSys::~CLoadSys() { } //装载NT驱动程序 BOOL CLoadSys::LoadDriver(WCHAR* lpszDriverName, WCHAR* lpszDriverPath) { WCHAR szDriverImagePath[256] = { 0 }; //得到完整的驱动路径 GetFullPathName(lpszDriverPath, 256, szDriverImagePath, NULL); BOOL bRet = FALSE; SC_HANDLE hServiceMgr = NULL;//SCM管理器的句柄 SC_HANDLE hServiceDDK = NULL;//NT驱动程序的服务句柄 //打开服务控制管理器 hServiceMgr = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS); if (hServiceMgr == NULL) { 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, //GroupOrder HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Control\GroupOrderList NULL, NULL, NULL, NULL); DWORD dwRtn; //判断服务是否失败 if (hServiceDDK == NULL) { dwRtn = GetLastError(); if (dwRtn != ERROR_IO_PENDING && dwRtn != ERROR_SERVICE_EXISTS) { //由于其他原因创建服务失败 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) { //如果打开服务也失败,则意味错误 bRet = FALSE; goto BeforeLeave; } } //启动项服务 bRet = StartService(hServiceDDK, NULL, NULL); if (!bRet) { DWORD dwRtn = GetLastError(); if (dwRtn != ERROR_IO_PENDING && dwRtn != ERROR_SERVICE_ALREADY_RUNNING) { bRet = FALSE; goto BeforeLeave; } else { if (dwRtn == ERROR_IO_PENDING) { //设备被挂住 bRet = FALSE; goto BeforeLeave; } else { bRet = TRUE; // goto BeforeLeave; } } } bRet = TRUE; //离开前关闭句柄 BeforeLeave: if (hServiceDDK) { CloseServiceHandle(hServiceDDK); } if (hServiceMgr) { CloseServiceHandle(hServiceMgr); } return bRet; } //卸载驱动程序 BOOL CLoadSys::UnloadDriver(WCHAR * 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) { bRet = FALSE; goto BeforeLeave; } //打开驱动所对应的服务 hServiceDDK = OpenService(hServiceMgr, szSvrName, SERVICE_ALL_ACCESS); if (hServiceDDK == NULL) { bRet = FALSE; goto BeforeLeave; } //停止驱动程序,如果停止失败,只有重新启动才能,再动态加载。 if (!ControlService(hServiceDDK, SERVICE_CONTROL_STOP, &SvrSta)) { bRet = FALSE; goto BeforeLeave; } //动态卸载驱动程序。 if (!DeleteService(hServiceDDK)) { //卸载失败 bRet = FALSE; goto BeforeLeave; } bRet = TRUE; BeforeLeave: //离开前关闭打开的句柄 if (hServiceDDK) { CloseServiceHandle(hServiceDDK); } if (hServiceMgr) { CloseServiceHandle(hServiceMgr); } return bRet; }
爱程序 不爱bug
爱生活 不爱黑眼圈
我和你们一样 我和你们不一样
我不是凡客 我要做geek