VC ++ enables or disables USB devices

2. # Include <windows. h>
4. # Include <tchar. h>
6. # Include <setupapi. h>
8. // # Include <shlwapi. h>
10. # Pragma comment (Lib, "setupapi. lib ")
12. // # Pragma comment (Lib, "shlwapi. lib ")
14. Hdevinfo = NULL;
16. // Error message
18. VoidFormatmsg (DWORDDwerror,Lptstr* O)
20. {
22. LptstrOO = (Lptstr) O;
24. BoolBok = formatmessage (format_message_from_system | format_message_allocate_buffer, null, dwerror,
26. Makelangid (lang_chinese, sublang_chinese_simplified), Oo, 0, null );
28. If(! Bok)// Change ** to * 2007/01/10
30. {// The newly created string is directed by oo.
32. HmoduleHdll = LoadLibraryEx (_ T ("Netmsg. dll"), Null, dont_resolve_dll_references );
34. If(Null! = Hdll)
36. {
38. Formatmessage (format_message_from_hmodule |
40. Format_message_from_system, hdll,
42. Dwerror,
44. Makelangid (lang_chinese, sublang_chinese_simplified ),
46. (Lptstr) O, 0, null );
48. Freelibrary (hdll );
50. }
52. }
54. }
56. BoolChangestatus (DWORDNewstatus,DWORDSelecteditem, hdevinfo)
58. {
60. LptstrLpszmsg = NULL;
62. HcursorHcursor = NULL;
64. Try
66. {
68. Sp_propchange_params propchangeparams = {Sizeof(Sp_classinstall_header )};
70. Sp_devinfo_data deviceinfodata = {Sizeof(Sp_devinfo_data )};
72. Hcursor = setcursor (loadcursor (null, idc_wait ));
74. // Get a handle to the selected item.
76. If(! Setupdienumdeviceinfo (hdevinfo, selecteditem, & deviceinfodata ))
78. {
80. Formatmsg (getlasterror (), & lpszmsg );
82. ThrowLpszmsg;
84. }
86. // Set the propchangeparams structure.
88. Propchangeparams. classinstallheader. installfunction = dif_propertychange;
90. Propchangeparams. Scope = dics_flag_global;
92. Propchangeparams. statechange = newstatus;
94. If(! Setupdisetclassinstallparams (hdevinfo, & deviceinfodata, (sp_classinstall_header *) & propchangeparams,
96. Sizeof(Propchangeparams )))
98. {
100. Formatmsg (getlasterror (), & lpszmsg );
102. ThrowLpszmsg;
104. }
108. // Call the classinstaller and perform the change.
110. If(! Setupdicallclassinstaller (dif_propertychange, hdevinfo, & deviceinfodata ))
112. {
114. Formatmsg (getlasterror (), & lpszmsg );
116. ThrowLpszmsg;
118. }
120. Setcursor (hcursor );
122. ReturnTrue;
124. }
126. Catch(Tchar* Pszerror)
128. {
130. Setcursor (hcursor );
132. : MessageBox (null, pszerror, _ T ("Prompt"), Mb_ OK );
134. If(Null! = Lpszmsg)
136. {
138. Localfree ((Hlocal) Lpszmsg );
140. }
142. ReturnFalse;
144. }
146. }
150. // This function is mainly used to enable and disable these devices.
152. // Parameter description, nstatus can have three values,-1: Nothing
154. // 0 disable the device, 1 enable the device.
156. // Nindex is used to control whether to enable or disable the optical drive or USB drive.
158. // Three values are also recommended. 0 indicates the soft drive, and 1 indicates the optical drive.
160. // 2 stands for USB
162. // You can call this function to disable the optical drive.
164. // Controldisk (0, 1 );
166. BoolControldisk (IntNstatus,IntNindex)
168. {
170. If(-1 = nstatus)
172. {
174. ReturnFalse;
176. }
178. LptstrLpszmsg = NULL;
180. Try
182. {
184. Tchar* Guidstring = NULL;
186. Guid;
188. Zeromemory (& guid,Sizeof(Guid ));
190. Switch(Nindex)
192. {
194. Case0:// 0 indicates the soft drive
196. Guidstring = _ T ("4d36e980-e325-11ce-bfc1-08002be10318");
198. Uuidfromstring (unsignedChar*) Guidstring, & guid );
200. Break;
202. Case1:// 1 indicates the optical drive
204. Guidstring = _ T ("4d36e965-e325-11ce-bfc1-08002be10318");
206. Uuidfromstring (unsignedChar*) Guidstring, & guid );
208. Break;
210. Case2:// 2 stands for USB
212. Guidstring = _ T ("36fc9e60-c465-11cf-8056-444553540000");
214. Uuidfromstring (unsignedChar*) Guidstring, & guid );
216. Break;
218. }
220. Hdevinfo = setupdigetclassdevs (& guid, null, null, digcf_present );
222. If(Invalid_handle_value! = Hdevinfo)
224. {Lpszmsg ="Hahazx";
226. Formatmsg (getlasterror (), & lpszmsg );
230. ThrowLpszmsg;
232. }
234. DWORDI;
236. Sp_devinfo_data deviceinfodata;
238. Zeromemory (& deviceinfodata,Sizeof(Sp_devinfo_data ));
240. Deviceinfodata. cbsize =Sizeof(Sp_devinfo_data );
242. For(I = 0; setupdienumdeviceinfo (hdevinfo, I, & deviceinfodata); ++ I)
244. {
246. If(1 = nstatus)
248. {: MessageBox (null,"Enable","Correct", Mb_ OK );
250. Changestatus (dics_enable, I, hdevinfo );
252. }
254. Else If(0 = nstatus)
256. {: MessageBox (null,"Disable","Error", Mb_ OK );
258. Changestatus (dics_disable, I, hdevinfo );
260. }
262. }
264. // Release device information set
266. ReturnSetupdidestroydeviceinfolist (hdevinfo );
268. }
270. Catch(Tchar* Pszerror)
272. {
274. : MessageBox (null, pszerror, _ T ("Prompt"), Mb_ OK );
276. If(Null! = Lpszmsg)
278. {
280. Localfree ((Hlocal) Lpszmsg );
282. }
284. ReturnFalse;
286. }
288. ReturnFalse;
290. }
292. VoidMain ()
294. {
296. Controldisk (1, 2 );
298. }