Windows 驅動加載程序代碼

#include <windows.h>    
#include <winsvc.h>    
#include <conio.h>    
#include <stdio.h>  
#define DRIVER_NAME "MySYS"  
#define DRIVER_PATH "./MySYS.sys"  


BOOL LoadNTDriver(char* lpszDriverName,char* lpszDriverPath)  
{  
    char szDriverImagePath[256];  
    //得到完整的驅動路徑  
    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 )    
    {  
        //OpenSCManager失敗  
        printf( "OpenSCManager() Faild %d ! /n", GetLastError() );  
        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,    
        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 );    
            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 )    
        {  
            //如果打開服務也失敗，則意味錯誤  
            dwRtn = GetLastError();    
            printf( "OpenService() Faild %d ! /n", dwRtn );    
            bRet = FALSE;  
            goto BeforeLeave;  
        }    
        else   
        {  
            printf( "OpenService() ok ! /n" );  
        }  
    }    
    else    
    {  
        printf( "CrateService() ok ! /n" );  
    }  
    //開啟此項服務  
    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 );    
            bRet = FALSE;  
            goto BeforeLeave;  
        }    
        else    
        {    
            if( dwRtn == ERROR_IO_PENDING )    
            {    
                //設備被掛住  
                printf( "StartService() Faild ERROR_IO_PENDING ! /n");  
                bRet = FALSE;  
                goto BeforeLeave;  
            }    
            else    
            {    
                //服務已經開啟  
                printf( "StartService() Faild ERROR_SERVICE_ALREADY_RUNNING ! /n");  
                bRet = TRUE;  
                goto BeforeLeave;  
            }    
        }    
    }  
    bRet = TRUE;  
//離開前關閉句柄  
BeforeLeave:  
    if(hServiceDDK)  
    {  
        CloseServiceHandle(hServiceDDK);  
    }  
    if(hServiceMgr)  
    {  
        CloseServiceHandle(hServiceMgr);  
    }  
    return bRet;  
}



//卸載驅動程序    
BOOL UnloadNTDriver( char * 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 )    
    {  
        //帶開SCM管理器失敗  
        printf( "OpenSCManager() Faild %d ! /n", GetLastError() );    
        bRet = FALSE;  
        goto BeforeLeave;  
    }    
    else    
    {  
        //帶開SCM管理器失敗成功  
        printf( "OpenSCManager() ok ! /n" );    
    }  
    //打開驅動所對應的服務  
    hServiceDDK = OpenService( hServiceMgr, szSvrName, SERVICE_ALL_ACCESS );    
    if( hServiceDDK == NULL )    
    {  
        //打開驅動所對應的服務失敗  
        printf( "OpenService() Faild %d ! /n", GetLastError() );    
        bRet = FALSE;  
        goto BeforeLeave;  
    }    
    else    
    {    
        printf( "OpenService() ok ! /n" );    
    }    
    //停止驅動程序，如果停止失敗，只有重新啟動才能，再動態加載。    
    if( !ControlService( hServiceDDK, SERVICE_CONTROL_STOP , &SvrSta ) )    
    {    
        printf( "ControlService() Faild %d !/n", GetLastError() );    
    }    
    else    
    {  
        //打開驅動所對應的失敗  
        printf( "ControlService() ok !/n" );    
    }    
    //動態卸載驅動程序。    
    if( !DeleteService( hServiceDDK ) )    
    {  
        //卸載失敗  
        printf( "DeleteSrevice() Faild %d !/n", GetLastError() );    
    }    
    else    
    {    
        //卸載成功  
        printf( "DelServer:eleteSrevice() ok !/n" );    
    }    
    bRet = TRUE;  
BeforeLeave:  
//離開前關閉打開的句柄  
    if(hServiceDDK)  
    {  
        CloseServiceHandle(hServiceDDK);  
    }  
    if(hServiceMgr)  
    {  
        CloseServiceHandle(hServiceMgr);  
    }  
    return bRet;      
}   
void TestDriver()  
{  
    //測試驅動程序    
    HANDLE hDevice = CreateFile("////.//HelloDDK",    
        GENERIC_WRITE | GENERIC_READ,    
        0,    
        NULL,    
        OPEN_EXISTING,    
        0,    
        NULL);    
    if( hDevice != INVALID_HANDLE_VALUE )    
    {  
        printf( "Create Device ok ! /n" );    
    }  
    else    
    {  
        printf( "Create Device faild %d ! /n", GetLastError() );    
    }  
    CloseHandle( hDevice );  
}









int main(int argc, char* argv[])    
{  
    //加載驅動  
    BOOL bRet = LoadNTDriver(DRIVER_NAME,DRIVER_PATH);  
    if (!bRet)  
    {  
        printf("LoadNTDriver error/n");  
        return 0;  
    }  
    //加載成功  
    printf( "press any to create device!/n" );    
    getch();    
    TestDriver();  
    //這時候你可以通過注冊表，或其他查看符號連接的軟件驗證。    
    printf( "press any to unload the driver!/n" );    
    getch();    
    //卸載驅動  
    UnloadNTDriver(DRIVER_NAME);  
    if (!bRet)  
    {  
        printf("UnloadNTDriver error/n");  
        return 0;  
    }  
    return 0;    
}