驱动学习笔记二：加载驱动

来源：互联网发布：360网络联盟编辑：程序博客网时间：2024/04/30 13:40
下面的代码是使用的SCM加载的驱动，非是什么猥琐方法...思想委琐的人可以去蹲墙角去了。驱动名和驱动路径之前定义过了...如果想写个加载工具，是可以参考下InstDrv工具源代码的。方法都一样的，呵呵
#include <windows.h>   
#include <winsvc.h>   
#include <conio.h>   
#include <stdio.h> 
#define DRIVER_NAME "HelloDDK" 
#define DRIVER_PATH "..//MyDriver//MyDriver_Check//HelloDDK.sys" 
//装载NT驱动程序 
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;  
}  