学步园

// LoadDrives.cpp : 定义控制台应用程序的入口点。<br /> //<br /> #include "stdafx.h"<br /> #include <Windows.h></p> <p>LPCTSTR lpRegSoftPath = "SYSTEM//CurrentControlSet//Services"; //注册表服务(驱动)路径<br /> LPCTSTR lpRegPath = "SYSTEM//CurrentControlSet//Services//helloDDK"; //注册表HelloDDK驱动路径.</p> <p>void DelSvr(TCHAR * szSvrName); //卸载驱动<br /> int StartSvr(TCHAR* szPath,TCHAR * szSvrName);<br /> void CheckReg()<br /> {<br /> HKEY hSoftKey = NULL;<br /> if (RegOpenKeyEx(HKEY_LOCAL_MACHINE,lpRegPath,<br /> 0,KEY_READ|KEY_WRITE,&hSoftKey) != ERROR_SUCCESS)<br /> {<br /> printf("创建");<br /> DWORD dwDisp;<br /> if (RegCreateKeyEx(HKEY_LOCAL_MACHINE,lpRegPath,0,REG_NONE,REG_OPTION_NON_VOLATILE,<br /> KEY_WRITE|KEY_READ,NULL,&hSoftKey,&dwDisp) != ERROR_SUCCESS)<br /> {<br /> printf("创建失败");<br /> }<br /> else<br /> {<br /> TCHAR szDisplayName[] = "HelloDDK_Test";<br /> DWORD dwErrorControl = 1;<br /> TCHAR szGroup[] = "";<br /> TCHAR szImagePath[] = "//??//E://HelloDDK.sys]//??//E://HelloDDK.sys"; //驱动路径<br /> DWORD dwStart = 3;<br /> DWORD dwType = 1;<br /> RegSetValueEx(hSoftKey,"DisplayName",0,REG_SZ,(BYTE *)szDisplayName,(DWORD)sizeof(szDisplayName));<br /> RegSetValueEx(hSoftKey,"ErrorControl",0,REG_DWORD,(CONST BYTE *)&dwErrorControl,sizeof(DWORD));<br /> RegSetValueEx(hSoftKey,"Group",0,REG_SZ,(BYTE *)szGroup,sizeof(szGroup));<br /> RegSetValueEx(hSoftKey,"ImagePath",0,REG_EXPAND_SZ,(BYTE *)szImagePath,sizeof(szImagePath));<br /> RegSetValueEx(hSoftKey,"Start",0,REG_DWORD,(CONST BYTE *)&dwStart,sizeof(DWORD));<br /> RegSetValueEx(hSoftKey,"Type",0,REG_DWORD,(CONST BYTE *)&dwType,sizeof(DWORD));<br /> }<br /> }<br /> RegCloseKey(hSoftKey);<br /> }<br /> int _tmain(int argc, _TCHAR* argv[])<br /> {<br /> //CheckReg(); //检测注册表<br /> //StartSvr("E://helloddk.sys","helloddk");<br /> DelSvr("HelloDDK");<br /> system("pause");<br /> return 0;<br /> }<br /> int StartSvr(TCHAR* szPath,TCHAR * szSvrName)<br /> {<br /> SC_HANDLE hServiceMgr, hServiceTwdm;<br /> BOOL bRtn;<br /> DWORD dwRtn, dwSize = 256;<br /> TCHAR szDir[256];<br /> strcpy(szDir,szPath); //驱动文件路径<br /> LPCTSTR lpszBinaryPathName = TEXT(szDir);<br /> hServiceMgr = OpenSCManager(NULL,NULL,SC_MANAGER_ALL_ACCESS); //打开服务控制管理器<br /> if( hServiceMgr == NULL )<br /> {<br /> printf( "Open SCManager() Faild %d ! /n", GetLastError() );<br /> return 0;<br /> }<br /> else<br /> {<br /> printf( "Open SCManager() ok ! /n" );<br /> }<br /> printf("================按键执行创建或打开服务./n");<br /> system("pause");<br /> //创建驱动服务<br /> hServiceTwdm = CreateService( hServiceMgr,<br /> TEXT(szSvrName), //SYSTEM/CurrentControlSet/Services 驱动程序的在注册表中的名字<br /> TEXT(szSvrName), // 注册表驱动程序的 DisplayName 值<br /> SERVICE_ALL_ACCESS, // 加载驱动程序的访问权限<br /> SERVICE_KERNEL_DRIVER,// 表示加载的服务是驱动程序<br /> SERVICE_DEMAND_START, // 注册表驱动程序的 Start 值<br /> SERVICE_ERROR_IGNORE, // 注册表驱动程序的 ErrorControl 值<br /> lpszBinaryPathName, // 注册表驱动程序的 ImagePath 值<br /> NULL,<br /> NULL,<br /> NULL,<br /> NULL,<br /> NULL);<br /> if( hServiceTwdm == NULL )<br /> {<br /> dwRtn = GetLastError();<br /> if( dwRtn != ERROR_IO_PENDING && dwRtn != ERROR_SERVICE_EXISTS )<br /> {<br /> CloseServiceHandle( hServiceMgr );<br /> printf( "Create Service() Faild %d ! /n", dwRtn );<br /> system("pause");<br /> return 0;<br /> }<br /> else<br /> {<br /> printf( "Create Service() Faild Service is ERROR_IO_PENDING or ERROR_SERVICE_EXISTS! /n" );<br /> }<br /> // 驱动程序已经加载，只需要打开<br /> hServiceTwdm = OpenService( hServiceMgr, TEXT(szSvrName), SERVICE_ALL_ACCESS );<br /> if( hServiceTwdm == NULL )<br /> {<br /> dwRtn = GetLastError();<br /> CloseServiceHandle( hServiceMgr );<br /> printf( "OpenService() Faild %d ! /n", dwRtn );<br /> system("pause");<br /> return 0;<br /> }<br /> else<br /> {<br /> printf( "Open Service() ok ! /n" );<br /> }<br /> }<br /> else<br /> {<br /> printf( "Create Service() ok ! /n" );<br /> }<br /> printf("================按键执行启动驱动服务./n");<br /> system("pause");<br /> // 启动驱动程序，调用驱动程序的 DriverEntry 函数<br /> bRtn = StartService( hServiceTwdm, NULL, NULL );<br /> if( !bRtn )<br /> {<br /> dwRtn = GetLastError();<br /> if( dwRtn != ERROR_IO_PENDING && dwRtn != ERROR_SERVICE_ALREADY_RUNNING )<br /> {<br /> printf( "StartService() Faild %d ! /n", dwRtn );<br /> CloseServiceHandle( hServiceTwdm );<br /> CloseServiceHandle( hServiceMgr );<br /> return 0;<br /> }<br /> else<br /> {<br /> if( dwRtn != ERROR_IO_PENDING )<br /> {<br /> printf( "Start Service() Faild ERROR_IO_PENDING ! /n");<br /> }<br /> else<br /> {<br /> printf( "Start Service() Faild ERROR_SERVICE_ALREADY_RUNNING ! /n");<br /> }<br /> }<br /> }<br /> else<br /> {<br /> printf( "Start Service() 成功 ! /n");<br /> }<br /> printf("===================按键执行关闭服务./n");<br /> system("pause");<br /> SERVICE_STATUS ServiceStatus;<br /> ControlService(hServiceTwdm,SERVICE_CONTROL_STOP,&ServiceStatus);<br /> CloseServiceHandle( hServiceTwdm );<br /> CloseServiceHandle( hServiceMgr );<br /> }<br /> //卸载驱动程序。<br /> void DelSvr(TCHAR* szSvrName)<br /> {<br /> SC_HANDLE hServiceMgr, hServiceTwdm;<br /> SERVICE_STATUS SvrSta;<br /> hServiceMgr = OpenSCManager( NULL, NULL, SC_MANAGER_ALL_ACCESS );<br /> if( hServiceMgr == NULL )<br /> {<br /> printf( "DelSvr::OpenSCManager() Faild %d ! /n", GetLastError() );<br /> return;<br /> }<br /> else<br /> {<br /> printf( "DelSvr::OpenSCManager() ok ! /n" );<br /> }<br /> hServiceTwdm = OpenService( hServiceMgr, TEXT(szSvrName), SERVICE_ALL_ACCESS );<br /> if( hServiceTwdm == NULL )<br /> {<br /> CloseServiceHandle( hServiceMgr );<br /> printf( "DelSvr::OpenService() Faild %d ! /n", GetLastError() );<br /> return;<br /> }<br /> else<br /> {<br /> printf( "DelSvr::OpenService() ok ! /n" );<br /> }<br /> //停止驱动程序，如果停止失败，只有重新启动才能，再动态加载。<br /> if( !ControlService( hServiceTwdm, SERVICE_CONTROL_STOP , &SvrSta ) )<br /> {<br /> printf( "DelSvr::ControlService() Faild %d !/n", GetLastError() );<br /> }<br /> else<br /> {<br /> printf( "DelSvr::ControlService() ok !/n" );<br /> }<br /> //动态卸载驱动程序。<br /> if( !DeleteService( hServiceTwdm ) )<br /> {<br /> printf( "DelSvr:eleteSrevice() Faild %d !/n", GetLastError() );<br /> }<br /> else<br /> {<br /> printf( "DelSvr:eleteSrevice() ok !/n" );<br /> }<br /> CloseServiceHandle( hServiceTwdm );<br /> CloseServiceHandle( hServiceMgr );<br /> return;<br /> }<br />