載入NT驅動的類 C++

標頭檔案

#pragma once
#include <windows.h>
#include <winsvc.h>  
#include <tchar.h>

class CLoadNtDriver
{
public:
    CLoadNtDriver();
    ~CLoadNtDriver();

    BOOL pathIsFile(CString strPath);

    BOOL loadNTDriver(CString strDriverPath);
    BOOL unloadNTDriver(CString strSvrName);
    void showErrorInfo(UINT nErrCode, UINT nLine, LPCTSTR = _T(""), UINT = 0);


    CString m_strDvrName;
    wchar_t m_szServerName[40];
    wchar_t m_szShowName[40];
    
};

 

 

原始檔

#include "stdafx.h"
#include "LoadNtDriver.h"
#include <strsafe.h>


CLoadNtDriver::CLoadNtDriver()
{
}


CLoadNtDriver::~CLoadNtDriver()
{
}

BOOL CLoadNtDriver::pathIsFile(CString strPath)
{
    // 過濾路徑是資料夾的情況
    //  Code by Lthis 

    if (PathIsDirectory(strPath))
        return FALSE;

    if (PathFileExists(strPath))
        return TRUE;

    return FALSE;
}


void CLoadNtDriver::showErrorInfo(UINT nErrCode, UINT nLine, LPCTSTR lpFuncName, UINT nType){
    LPTSTR lpMsgBuf = NULL;
    TCHAR szMessage[256] = { 0 };
    TCHAR szCaption[32]  = { 0 };

    FormatMessage(0x1300, NULL, nErrCode, 0x400, (LPTSTR)&lpMsgBuf, 64, NULL);
    StringCchPrintf(szMessage, 256, _T("Error code:0x%X:%s\n"), nErrCode, lpMsgBuf);
    StringCchPrintf(szCaption, 32, _T("%s (Error Line:%d)"), lpFuncName, nLine);
    StringCchCat(szMessage, 256+32+1, szCaption);

    switch (nType)
    {
    case 0:
        OutputDebugString(szMessage);
        break;
    case 1:
        MessageBox(NULL, szMessage, szCaption, 0);
        break;
    default:
        break;
    }
}

//裝載NT驅動程式
BOOL CLoadNtDriver::loadNTDriver(CString strDriverPath)
{
    wchar_t  szDriverImagePath[256];
    CString  strDriverImagePath;
    //得到完整的驅動路徑
    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;
    }

    // 提取驅動名
    int nPos;
    nPos = strDriverImagePath.ReverseFind(_T('\\'));
    
    m_strDvrName = strDriverImagePath.Right(nPos - 1);
    nPos = m_strDvrName.ReverseFind(_T('.'));
    m_strDvrName = m_strDvrName.Left(nPos);

    swprintf_s(m_szServerName, m_strDvrName + L"_ServerName");
    swprintf_s(m_szShowName, m_strDvrName + L"_ShowName");


    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失敗
        showErrorInfo(GetLastError(), __LINE__, L"loadNTDriver");
        bRet = FALSE;
        goto BeforeLeave;
    }
    else{
        //OpenSCManager成功
        showErrorInfo(GetLastError(), __LINE__, L"loadNTDriver()");
    }

    //建立驅動所對應的服務
    hServiceDDK = CreateService(hServiceMgr,
        m_szServerName,                //驅動程式的在登錄檔中的名字  
        m_szShowName,                // 登錄檔驅動程式的 DisplayName 值  
        SERVICE_ALL_ACCESS,            // 載入驅動程式的訪問許可權  
        SERVICE_KERNEL_DRIVER,        // 表示載入的服務是驅動程式  
        SERVICE_DEMAND_START,        // 登錄檔驅動程式的 Start 值  
        SERVICE_ERROR_IGNORE,        // 登錄檔驅動程式的 ErrorControl 值  
        strDriverImagePath,            // 登錄檔驅動程式的 ImagePath 值  
        NULL,
        NULL,
        NULL,
        NULL,
        NULL);

    DWORD dwRtn;

    //判斷服務是否失敗
    if (hServiceDDK == NULL)
    {
        dwRtn = GetLastError();
        if (dwRtn != ERROR_IO_PENDING && dwRtn != ERROR_SERVICE_EXISTS)
        {
            //由於其他原因建立服務失敗
            //printf("CrateService() Faild %d ! \n", dwRtn);
            showErrorInfo(GetLastError(), __LINE__, L"loadNTDriver()");
            bRet = FALSE;
            goto BeforeLeave;
        }
        else
        {
            //服務建立失敗，是由於服務已經創立過
            //printf("CrateService() Faild Service is ERROR_IO_PENDING or ERROR_SERVICE_EXISTS! \n");
            showErrorInfo(GetLastError(), __LINE__, L"loadNTDriver()");
        }

        // 驅動程式已經載入，只需要開啟  
        hServiceDDK = OpenService(hServiceMgr, m_szServerName, SERVICE_ALL_ACCESS);
        if (hServiceDDK == NULL)
        {
            //如果開啟服務也失敗，則意味錯誤
            dwRtn = GetLastError();
            printf("OpenService() Faild %d ! \n", dwRtn);
            bRet = FALSE;
            goto BeforeLeave;
        }
    }

    //開啟此項服務
    bRet = StartService(hServiceDDK, NULL, NULL);
    if (!bRet)
    {
        DWORD dwRtn = GetLastError();
        if (dwRtn != ERROR_IO_PENDING && dwRtn != ERROR_SERVICE_ALREADY_RUNNING)
        {
            //printf("StartService() Faild %d ! \n", dwRtn);
            showErrorInfo(GetLastError(), __LINE__, L"loadNTDriver()");
            bRet = FALSE;
            goto BeforeLeave;
        }
        else
        {
            if (dwRtn == ERROR_IO_PENDING)
            {
                //裝置被掛住
                //printf("StartService() Faild ERROR_IO_PENDING ! \n");
                showErrorInfo(GetLastError(), __LINE__, L"loadNTDriver()");
                bRet = FALSE;
                goto BeforeLeave;
            }
            else
            {
                //服務已經開啟
                //printf("StartService() Faild ERROR_SERVICE_ALREADY_RUNNING ! \n");
                showErrorInfo(GetLastError(), __LINE__, L"loadNTDriver()");
                bRet = TRUE;
                goto BeforeLeave;
            }
        }
    }
    bRet = TRUE;
    //離開前關閉控制程式碼
BeforeLeave:
    if (hServiceDDK){
        CloseServiceHandle(hServiceDDK);
    }
    if (hServiceMgr){
        CloseServiceHandle(hServiceMgr);
    }

    return bRet;
}

/**************************************************************** 
// FunctionName:  unloadNTDriver
// Function :      解除安裝NT驅動
// Parameter:      strSvrName,驅動服務名，類中m_szServerName儲存，
                    傳遞該值即可
// Author: 張帆
// Create: 2015-3-23 20:57:50 
// Checked: Lthis 2015-3-23 
****************************************************************/
BOOL CLoadNtDriver::unloadNTDriver(CString strSvrName)
{
    BOOL bRet = FALSE;
    SC_HANDLE hServiceMgr = NULL;//SCM管理器的控制程式碼
    SC_HANDLE hServiceDDK = NULL;//NT驅動程式的服務控制程式碼
    SERVICE_STATUS SvrSta;
    
    CString strError;
    //開啟SCM管理器
    hServiceMgr = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
    if (hServiceMgr == NULL)
    {
        //開啟SCM管理器失敗
        //"unloadNTDriver -> OpenSCManager() Faild ! \n";
        showErrorInfo(GetLastError(), __LINE__, L"unloadNTDriver()");
        bRet = FALSE;
        goto BeforeLeave;
    }

    //開啟驅動所對應的服務
    hServiceDDK = OpenService(hServiceMgr, strSvrName, SERVICE_ALL_ACCESS);

    if (hServiceDDK == NULL){
        //開啟驅動所對應的服務失敗
        //strError = L"unloadNTDriver -> OpenService() Faild  ! !";
        showErrorInfo(GetLastError(), __LINE__, L"unloadNTDriver()");
        goto BeforeLeave;
    }

    //停止驅動程式，如果停止失敗，只有重新啟動才能，再動態載入。  
    if (!ControlService(hServiceDDK, SERVICE_CONTROL_STOP, &SvrSta))
    {
        //printf("ControlService() Faild %d !\n", GetLastError());
        showErrorInfo(GetLastError(), __LINE__, L"unloadNTDriver()");
    }

    //動態解除安裝驅動程式。  
    if (!DeleteService(hServiceDDK))
    {
        //解除安裝失敗
        //printf("DeleteSrevice() Faild %d !\n", GetLastError());
        showErrorInfo(GetLastError(), __LINE__, L"unloadNTDriver()");

    }
    bRet = TRUE;
BeforeLeave:
    //離開前關閉開啟的控制程式碼
    if (hServiceDDK){
        CloseServiceHandle(hServiceDDK);
    }
    if (hServiceMgr){
        CloseServiceHandle(hServiceMgr);
    }
    return bRet;
}