VB 驱动控制类

1. Option Explicit
2. Private Const STANDARD_RIGHTS_REQUIRED = 
3. Private Const SC_MANAGER_CONNECT = 
4. Private Const SC_MANAGER_CREATE_SERVICE = 
5. Private Const SC_MANAGER_ENUMERATE_SERVICE = 
6. Private Const SC_MANAGER_LOCK = 
7. Private Const SC_MANAGER_QUERY_LOCK_STATUS = 
8. Private Const SC_MANAGER_MODIFY_BOOT_CONFIG = 
11. Private Const SC_MANAGER_ALL_ACCESS = (STANDARD_RIGHTS_REQUIRED Or _
12.                                        SC_MANAGER_CONNECT Or _
13.                                        SC_MANAGER_CREATE_SERVICE Or _
14.                                        SC_MANAGER_ENUMERATE_SERVICE Or _
15.                                        SC_MANAGER_LOCK Or _
16.                                        SC_MANAGER_QUERY_LOCK_STATUS Or _
17.                                        SC_MANAGER_MODIFY_BOOT_CONFIG)
19. Private Const SERVICE_KERNEL_DRIVER = 1
21. Private Type SERVICE_STATUS
22.     dwCurrentState   As Long
23.     dwServiceType   As Long
24.     dwControlsAccepted   As Long
25.     dwWin32ExitCode   As Long
26.     dwServiceSpecificExitCode   As Long
27.     dwCheckPoint   As Long
28.     dwWaitHint   As Long
29. End Type
30. Private Const SERVICE_AUTO_START = 2
31. Private Const SERVICE_DEMAND_START = 3
32. Private Const SERVICE_ERROR_NORMAL = 1
34. Private Const SERVICE_CONTROL_STOP = 
35. Private Const SERVICE_CONTROL_PAUSE = 
36. Private Const SERVICE_CONTROL_CONTINUE = 
37. Private Const SERVICE_CONTROL_INTERROGATE = 
38. Private Const SERVICE_CONTROL_SHUTDOWN = 
41. Private Const SERVICE_QUERY_CONFIG = &H1
42. Private Const SERVICE_CHANGE_CONFIG = &H2
43. Private Const SERVICE_QUERY_STATUS = &H4
44. Private Const SERVICE_ENUMERATE_DEPENDENTS = &H8
45. Private Const SERVICE_START = &H10
46. Private Const SERVICE_STOP = &H20
47. Private Const SERVICE_PAUSE_CONTINUE = &H40
48. Private Const SERVICE_INTERROGATE = &H80
49. Private Const SERVICE_USER_DEFINED_CONTROL = &H100
50. Private Const SERVICE_ALL_ACCESS = (STANDARD_RIGHTS_REQUIRED Or _
51.                                     SERVICE_QUERY_CONFIG Or _
52.                                     SERVICE_CHANGE_CONFIG Or _
53.                                     SERVICE_QUERY_STATUS Or _
54.                                     SERVICE_ENUMERATE_DEPENDENTS Or _
55.                                     SERVICE_START Or _
56.                                     SERVICE_STOP Or _
57.                                     SERVICE_PAUSE_CONTINUE Or _
58.                                     SERVICE_INTERROGATE Or _
59.                                     SERVICE_USER_DEFINED_CONTROL)
61. Private Declare Function OpenSCManager _
62.       Lib "advapi32" Alias "OpenSCManagerW" _
63.       (ByVal lpMachineName As Long, ByVal lpDatabaseName As Long, _
64.       ByVal dwDesiredAccess As Long) As Long
65.       
66. Private Declare Function CreateService _
67.       Lib "advapi32" Alias "CreateServiceW" _
68.       (ByVal hSCManager As Long, ByVal lpServiceName As Long, _
69.       ByVal lpDisplayName As Long, ByVal dwDesiredAccess As Long, _
70.       ByVal dwServiceType As Long, ByVal dwStartType As Long, _
71.       ByVal dwErrorControl As Long, ByVal lpBinaryPathName As Long, _
72.       ByVal lpLoadOrderGroup As Long, ByVal lpdwTagId As Long, _
73.       ByVal lpDependencies As Long, ByVal lpServiceStartName As Long, _
74.       ByVal lpPassword As Long) As Long
75.       
76. Private Declare Function DeleteService _
77.       Lib "advapi32" (ByVal hService As Long) As Long
78.       
79. Private Declare Function CloseServiceHandle _
80.       Lib "advapi32" (ByVal hSCObject As Long) As Long
81.       
82. Private Declare Function OpenService _
83.       Lib "advapi32" Alias "OpenServiceW" _
84.       (ByVal hSCManager As Long, ByVal lpServiceName As Long, _
85.       ByVal dwDesiredAccess As Long) As Long
86.       
87. Private Declare Function StartService Lib "advapi32" _
88.         Alias "StartServiceW" (ByVal hService As Long, _
89.         ByVal dwNumServiceArgs As Long, ByVal lpServiceArgVectors As Long) As Long
90.         
91. Private Declare Function ChangeServiceConfig2 Lib "advapi32" Alias "ChangeServiceConfig2W" (ByVal hService As Long, _
92.         ByVal dwInfoLevel As Long, lpInfo As Any) As Long
93.         
94. Private Declare Function ControlService Lib "advapi32" _
95.         (ByVal hService As Long, ByVal dwControl As Long, _
96.         lpServiceStatus As SERVICE_STATUS) As Long
98.                                       
100. '''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
101. '驱动控制相关
102. Private Const OPEN_EXISTING As Long = 3
103. Private Const GENERIC_READ As Long = 
104. Private Const GENERIC_WRITE As Long = 
105. Private Const FILE_ATTRIBUTE_NORMAL = 
106. Private Const FILE_DEVICE_UNKNOWN As Long = 
107. Private Const FILE_SHARE_READ = 
108. Private Const FILE_SHARE_WRITE = 
110. Private Declare Function CreateFile Lib "KERNEL32" Alias "CreateFileA" (ByVal lpFileName As String, ByVal dwDesiredAccess As Long, ByVal dwShareMode As Long, lpSecurityAttributes As Any, ByVal dwCreationDisposition As Long, ByVal dwFlagsAndAttributes As Long, ByVal hTemplateFile As Long) As Long
111. Private Declare Function CloseHandle Lib "KERNEL32.DLL" (ByVal hObject As Long) As Long
112. Public Declare Function DeviceIoControl Lib "KERNEL32.DLL" (ByVal hDevice As Long, ByVal dwIoControlCode As Long, ByRef lpInBuffer As Any, ByVal nInBufferSize As Long, ByRef lpOutBuffer As Any, ByVal nOutBufferSize As Long, ByRef lpBytesReturned As Long, ByRef lpOverlapped As Any) As Long
114. Private mstrDriverName As String '驱动名称
115. Private mstrDisplayName As String
116. Private mhManager As Long
117. Private mhDriver As Long
118. Private mblnShare As Boolean
121. Public Function CHLoadDriver(ByVal lpDriverPath As String) As Boolean
122.     Dim hService As Long
123.     Dim blnOk As Boolean
124.     mhManager = OpenSCManager(0, 0, SC_MANAGER_ALL_ACCESS)
125.     If mhManager = 0 Then Exit Function
126.     hService = CreateService(mhManager, _
127.                              StrPtr(mstrDriverName), _
128.                              StrPtr(mstrDisplayName), _
129.                              SERVICE_ALL_ACCESS, _
130.                              SERVICE_KERNEL_DRIVER, _
131.                              SERVICE_DEMAND_START, _
132.                              SERVICE_ERROR_NORMAL, _
133.                              StrPtr(lpDriverPath), _
134.                              0, _
135.                              0, _
136.                              0, _
137.                              0, _
138.                              0)
139.     If hService Then
140.         If StartService(hService, 0, 0) Then
141.             blnOk = True
142.         End If
143.         Call CloseServiceHandle(hService)
144.     Else
145.         hService = OpenService(mhManager, StrPtr(mstrDriverName), SERVICE_START)
146.         If hService Then
147.             If StartService(hService, 0, 0) Then
148.                 blnOk = True
149.             End If
150.             Call CloseServiceHandle(hService)
151.         End If
152.     End If
153.     LoadDriver = blnOk
154. End Function
156. Public Function CHUnLoadDriver() As Boolean
157.     Dim objState As SERVICE_STATUS
158.     Dim hService As Long
159.     Dim blnOk As Boolean
160.     hService = OpenService(mhManager, StrPtr(mstrDriverName), SERVICE_ALL_ACCESS)
161.     If hService Then
162.         If ControlService(hService, SERVICE_CONTROL_STOP, objState) Then
163.             If DeleteService(hService) Then
164.                 blnOk = True
165.             End If
166.         End If
167.         Call CloseServiceHandle(hService)
168.     End If
169.     Call CloseServiceHandle(mhManager)
170.     UnLoadDriver = blnOk
171. End Function
173. Public Function CHDeviceIoControl(ByVal dwIoControlCode As Long, ByVal lpInBuffer As Long, ByVal nInBufferSize As Long, ByVal lpOutBuffer As Long, ByVal nOutBufferSize As Long, ByRef lpBytesReturned As Long) As Boolean
174.     CHDeviceIoControl = DeviceIoControl(mhDriver, dwIoControlCode, lpInBuffer, nInBufferSize, lpOutBuffer, nOutBufferSize, lpBytesReturned, ByVal 0&)
175. End Function
177. Public Property Get DriverHandle() As Long
178.     If mhDriver = 0 Then
179.         If mblnShare Then
180.             mhDriver = CreateFile("//./" & mstrDriverName, GENERIC_READ Or GENERIC_WRITE, FILE_SHARE_READ Or FILE_SHARE_WRITE, ByVal 0&, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0)
181.         Else
182.             mhDriver = CreateFile("//./" & mstrDriverName, GENERIC_READ Or GENERIC_WRITE, 0, ByVal 0&, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0)
183.         End If
184.     End If
185.     DriverHandle = mhDriver
186. End Property
188. 'Private Property Let DriverName(ByVal New_Value As Long)
189. '    mhDriver = New_Value
190. 'End Property
192. Public Property Get ShareControl() As Boolean
193.     ShareControl = mblnShare
194. End Property
196. Public Property Let ShareControl(ByVal New_Value As Boolean)
197.     mblnShare = New_Value
198. End Property
200. Public Property Get DriverName() As String
201.     DriverName = mstrDriverName
202. End Property
204. Public Property Let DriverName(ByVal New_Value As String)
205.     mstrDriverName = New_Value
206. End Property
208. Public Property Get DisplayName() As String
209.     DriverName = mstrDisplayName
210. End Property
212. Public Property Let DisplayName(ByVal New_Value As String)
213.     mstrDisplayName = New_Value
214. End Property
216. Private Sub Class_Initialize()
217.     mblnShare = True
218. End Sub
220. Private Sub Class_Terminate()
221.     CloseHandle mhDriver
222. End Sub