vc++实现控制USB设备启用与否

winzenghua

• #include<WINDOWS.H>
  
• #include<TCHAR.H>
  
• #include<SETUPAPI.H>
  
• //#include<SHLWAPI.H>
  
• #pragmacomment(lib,"setupapi.lib")
  
• //#pragmacomment(lib,"shlwapi.lib")
  
• HDEVINFOhDevInfo=NULL;
  
• //出错信息
  
• voidFormatMSG(DWORDdwError,LPTSTR*o)
  
• {
  
• LPTSTRoo=(LPTSTR)o;
  
• BOOLbOk=FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_ALLOCATE_BUFFER,NULL,dwError,
  
• MAKELANGID(LANG_CHINESE,SUBLANG_CHINESE_SIMPLIFIED),oo,0,NULL);
  
• if(!bOk)//将**变成*2007/01/10
  
• {//新建的字符串由oo指向。
  
• HMODULEhDll=LoadLibraryEx(_T("netmsg.dll"),NULL,DONT_RESOLVE_DLL_REFERENCES);
  
• if(NULL!=hDll)
  
• {
  
• FormatMessage(FORMAT_MESSAGE_FROM_HMODULE|
  
• FORMAT_MESSAGE_FROM_SYSTEM,hDll,
  
• dwError,
  
• MAKELANGID(LANG_CHINESE,SUBLANG_CHINESE_SIMPLIFIED),
  
• (LPTSTR)o,0,NULL);
  
• FreeLibrary(hDll);
  
• }
  
• }
  
• }
  
• BOOLChangeStatus(DWORDNewStatus,DWORDSelectedItem,HDEVINFOhDevInfo)
  
• {
  
• LPTSTRlpszMsg=NULL;
  
• HCURSORhCursor=NULL;
  
• try
  
• {
  
• SP_PROPCHANGE_PARAMSPropChangeParams={sizeof(SP_CLASSINSTALL_HEADER)};
  
• SP_DEVINFO_DATADeviceInfoData={sizeof(SP_DEVINFO_DATA)};
  
• hCursor=SetCursor(LoadCursor(NULL,IDC_WAIT));
  
• //GetahandletotheSelectedItem.
  
• if(!SetupDiEnumDeviceInfo(hDevInfo,SelectedItem,&DeviceInfoData))
  
• {
  
• FormatMSG(GetLastError(),&lpszMsg);
  
• throwlpszMsg;
  
• }
  
• //SetthePropChangeParamsstructure.
  
• PropChangeParams.ClassInstallHeader.InstallFunction=DIF_PROPERTYCHANGE;
  
• PropChangeParams.Scope=DICS_FLAG_GLOBAL;
  
• PropChangeParams.StateChange=NewStatus;
  
• if(!SetupDiSetClassInstallParams(hDevInfo,&DeviceInfoData,(SP_CLASSINSTALL_HEADER*)&PropChangeParams,
  
• sizeof(PropChangeParams)))
  
• {
  
• FormatMSG(GetLastError(),&lpszMsg);
  
• throwlpszMsg;
  
• }
  
• 
  
• //CalltheClassInstallerandperformthechange.
  
• if(!SetupDiCallClassInstaller(DIF_PROPERTYCHANGE,hDevInfo,&DeviceInfoData))
  
• {
  
• FormatMSG(GetLastError(),&lpszMsg);
  
• throwlpszMsg;
  
• }
  
• SetCursor(hCursor);
  
• returnTRUE;
  
• }
  
• catch(TCHAR*pszError)
  
• {
  
• SetCursor(hCursor);
  
• ::MessageBox(NULL,pszError,_T("提示"),MB_OK);
  
• if(NULL!=lpszMsg)
  
• {
  
• LocalFree((HLOCAL)lpszMsg);
  
• }
  
• returnFALSE;
  
• }
  
• }
  
• 
  
• //这些设备的启用和禁用主要有此函数实现
  
• //参数说明,nStatus可取3个值，-1：啥也不做
  
• //0禁用设备，1启用设备。
  
• //nIndex用于控制是对光驱，软驱还是对USB启用或禁用
  
• //也是可取3个值，0代表软驱，1代表光驱
  
• //2代表USB
  
• //譬如要对光驱实现禁用，可以这样调用此函数
  
• //ControlDisk(0,1);
  
• BOOLControlDisk(intnStatus,intnIndex)
  
• {
  
• if(-1==nStatus)
  
• {
  
• returnFALSE;
  
• }
  
• LPTSTRlpszMsg=NULL;
  
• try
  
• {
  
• TCHAR*GUIDString=NULL;
  
• GUIDguid;
  
• ZeroMemory(&guid,sizeof(GUID));
  
• switch(nIndex)
  
• {
  
• case0://0代表软驱
  
• GUIDString=_T("4D36E980-E325-11CE-BFC1-08002BE10318");
  
• UuidFromString((unsignedchar*)GUIDString,&guid);
  
• break;
  
• case1://1代表光驱
  
• GUIDString=_T("4D36E965-E325-11CE-BFC1-08002BE10318");
  
• UuidFromString((unsignedchar*)GUIDString,&guid);
  
• break;
  
• case2://2代表USB
  
• GUIDString=_T("36FC9E60-C465-11CF-8056-444553540000");
  
• UuidFromString((unsignedchar*)GUIDString,&guid);
  
• break;
  
• }
  
• hDevInfo=SetupDiGetClassDevs(&guid,NULL,NULL,DIGCF_PRESENT);
  
• if(INVALID_HANDLE_VALUE!=hDevInfo)
  
• {lpszMsg="hahazx";
  
• FormatMSG(GetLastError(),&lpszMsg);
  
• 
  
• throwlpszMsg;
  
• }
  
• DWORDi;
  
• SP_DEVINFO_DATADeviceInfoData;
  
• ZeroMemory(&DeviceInfoData,sizeof(SP_DEVINFO_DATA));
  
• DeviceInfoData.cbSize=sizeof(SP_DEVINFO_DATA);
  
• for(i=0;SetupDiEnumDeviceInfo(hDevInfo,i,&DeviceInfoData);++i)
  
• {
  
• if(1==nStatus)
  
• {::MessageBox(NULL,"启用","correct",MB_OK);
  
• ChangeStatus(DICS_ENABLE,i,hDevInfo);
  
• }
  
• elseif(0==nStatus)
  
• {::MessageBox(NULL,"禁用","error",MB_OK);
  
• ChangeStatus(DICS_DISABLE,i,hDevInfo);
  
• }
  
• }
  
• //释放deviceinformationset
  
• returnSetupDiDestroyDeviceInfoList(hDevInfo);
  
• }
  
• catch(TCHAR*pszError)
  
• {
  
• ::MessageBox(NULL,pszError,_T("提示"),MB_OK);
  
• if(NULL!=lpszMsg)
  
• {
  
• LocalFree((HLOCAL)lpszMsg);
  
• }
  
• returnFALSE;
  
• }
  
• returnFALSE;
  
• }
  
• voidmain()
  
• {
  
• ControlDisk(1,2);
  
• }