c++加载驱动文件

来源：互联网发布：三枪内裤怎么样知乎编辑：程序博客网时间：2024/06/05 12:50
#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; }
阅读全文
0 0