【C++】驱动加载/卸载/通信代码（兼容WIN32/WIN64）

Golden Blonde

类代码：

1. //Author: Anonymous
   
2. //#include <Windows.h>
   
3. 
   
4. class cDrvCtrl
   
5. {
   
6. public:
   
7.     cDrvCtrl()
   
8.     {
   
9.         m_pSysPath = NULL;
   
10.         m_pServiceName = NULL;
    
11.         m_pDisplayName = NULL;
    
12.         m_hSCManager = NULL;
    
13.         m_hService = NULL;
    
14.         m_hDriver = INVALID_HANDLE_VALUE;
    
15.     }
    
16.     ~cDrvCtrl()
    
17.     {
    
18.         CloseServiceHandle(m_hService);
    
19.         CloseServiceHandle(m_hSCManager);
    
20.         CloseHandle(m_hDriver);
    
21.     }
    
22. public:
    
23.     DWORD m_dwLastError;
    
24.     PCHAR m_pSysPath;
    
25.     PCHAR m_pServiceName;
    
26.     PCHAR m_pDisplayName;
    
27. public:
    
28.     BOOL Install(PCHAR pSysPath,PCHAR pServiceName,PCHAR pDisplayName);
    
29.     BOOL Start();
    
30.     BOOL Stop();
    
31.     BOOL Remove();
    
32.     BOOL Open(PCHAR pLinkName);
    
33.     BOOL IoControl(DWORD dwIoCode, PVOID InBuff, DWORD InBuffLen, PVOID OutBuff, DWORD OutBuffLen);
    
34. protected:
    
35. private:
    
36.     SC_HANDLE m_hSCManager;
    
37.     SC_HANDLE m_hService;
    
38.     HANDLE m_hDriver;
    
39. };
    
40. 
    
41. BOOL cDrvCtrl::Install(PCHAR pSysPath,PCHAR pServiceName,PCHAR pDisplayName)
    
42. {
    
43.     m_pSysPath = pSysPath;
    
44.     m_pServiceName = pServiceName;
    
45.     m_pDisplayName = pDisplayName;
    
46. 
    
47.     m_hSCManager = OpenSCManagerA(NULL,NULL,SC_MANAGER_ALL_ACCESS);
    
48.     if (NULL == m_hSCManager)
    
49.     {
    
50.         m_dwLastError = GetLastError();
    
51.         return FALSE;
    
52.     }
    
53. 
    
54.     m_hService = CreateServiceA(m_hSCManager,m_pServiceName,m_pDisplayName,
    
55.                     SERVICE_ALL_ACCESS,SERVICE_KERNEL_DRIVER,SERVICE_DEMAND_START,SERVICE_ERROR_NORMAL,
    
56.                     m_pSysPath,NULL,NULL,NULL,NULL,NULL);
    
57.     if (NULL == m_hService)
    
58.     {
    
59.         m_dwLastError = GetLastError();
    
60.         if (ERROR_SERVICE_EXISTS == m_dwLastError)
    
61.         {
    
62.             m_hService = OpenServiceA(m_hSCManager,m_pServiceName,SERVICE_ALL_ACCESS);
    
63.             if (NULL == m_hService)
    
64.             {
    
65.                 CloseServiceHandle(m_hSCManager);
    
66.                 return FALSE;
    
67.             }
    
68.         }
    
69.         else
    
70.         {
    
71.             CloseServiceHandle(m_hSCManager);
    
72.             return FALSE;
    
73.         }
    
74.     }
    
75.     return TRUE;
    
76. }
    
77. 
    
78. BOOL cDrvCtrl::Start()
    
79. {
    
80.     if (!StartServiceA(m_hService,NULL,NULL))
    
81.     {
    
82.         m_dwLastError = GetLastError();
    
83.         return FALSE;
    
84.     }
    
85.     return TRUE;
    
86. }
    
87. 
    
88. BOOL cDrvCtrl::Stop()
    
89. {
    
90.     SERVICE_STATUS ss;
    
91.     if (!ControlService(m_hService,SERVICE_CONTROL_STOP,&ss))
    
92.     {
    
93.         m_dwLastError = GetLastError();
    
94.         return FALSE;
    
95.     }
    
96.     return TRUE;
    
97. 
    
98. }
    
99. 
    
100. BOOL cDrvCtrl::Remove()
     
101. {
     
102.     if (!DeleteService(m_hService))
     
103.     {
     
104.         m_dwLastError = GetLastError();
     
105.         return FALSE;
     
106.     }
     
107.     return TRUE;
     
108. }
     
109. 
     
110. BOOL cDrvCtrl::Open(PCHAR pLinkName)//example: \\\\.\\xxoo        
     
111. {
     
112.     if (m_hDriver != INVALID_HANDLE_VALUE)
     
113.         return TRUE;
     
114.     m_hDriver = CreateFileA(pLinkName, GENERIC_READ | GENERIC_WRITE, 0, 0, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);
     
115.     if(m_hDriver != INVALID_HANDLE_VALUE)
     
116.         return TRUE;
     
117.     else
     
118.         return FALSE;
     
119. }
     
120. 
     
121. DWORD CTL_CODE_GEN(DWORD lngFunction)
     
122. {
     
123.     return (FILE_DEVICE_UNKNOWN * 65536) | (FILE_ANY_ACCESS * 16384) | (lngFunction * 4) | METHOD_BUFFERED;
     
124. }
     
125. 
     
126. BOOL cDrvCtrl::IoControl(DWORD dwIoCode, PVOID InBuff, DWORD InBuffLen, PVOID OutBuff, DWORD OutBuffLen)
     
127. {
     
128.     DWORD dw;
     
129.     return DeviceIoControl(m_hDriver,CTL_CODE_GEN(dwIoCode),InBuff,InBuffLen,OutBuff,OutBuffLen,&dw,NULL);
     
130. }

复制代码

范例代码：

1. void test6()
   
2. {
   
3.     cDrvCtrl dc;
   
4.     BOOL b;
   
5.     DWORD dw=0x1234,dw0=0;
   
6.     //load drv
   
7.     b=dc.Install("D:\\temp\\MyDriver.sys","MyDriver","MyDriver");
   
8.     printf("dc.Install=%d\n",b);
   
9.     b=dc.Start();
   
10.     printf("dc.Start=%d\n",b);
    
11.     //call drv
    
12.     b=dc.Open("\\\\.\\MyDriver");
    
13.     printf("dc.Open=%d\n",b);
    
14.     b=dc.IoControl(0x800,&dw,sizeof(dw),&dw0,sizeof(dw0));
    
15.     printf("dc.IoControl=%d\n",b);
    
16.     if(!b)
    
17.     {
    
18.         dw0=GetLastError();
    
19.         printf("LastError=%ld\n",dw0);
    
20.     }
    
21.     else
    
22.     {
    
23.         printf("dw0=%ld\n",dw0);
    
24.     }
    
25.     //unload drv
    
26.     b=dc.Stop();
    
27.     printf("dc.Stop=%d\n",b);
    
28.     b=dc.Remove();
    
29.     printf("dc.Remove=%d\n",b);
    
30. }

复制代码

0xAA55

关键在于使用sc来控制service，通过创建SERVICE_KERNEL_DRIVER来创建一个内核驱动方式的服务。

我突然在想，如果是单片机，想要动态加载驱动，该怎么设计单片机系统。

虽说，单片机没人把指令放RAM上，都是放ROM的。但可以设计一个功能让它从UART等接口动态读取bin然后自己烧自己的ROM。

虽说STM32F103的每个ROM页只能被烧10000次

xiawan

感谢楼主分享~~~

xiawan

楼主大能，感谢感谢