- #include <WINDOWS.H>
- #include <TCHAR.H>
- #include <SETUPAPI.H>
- //#include <SHLWAPI.H>
- #pragma comment(lib, "setupapi.lib")
- //#pragma comment(lib, "shlwapi.lib")
- HDEVINFO hDevInfo = NULL;
- // 出错信息
- void FormatMSG(DWORD dwError, LPTSTR * o)
- {
- LPTSTR oo=(LPTSTR)o;
- BOOL bOk = 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指向。
- HMODULE hDll = 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);
- }
- }
- }
- BOOL ChangeStatus(DWORD NewStatus, DWORD SelectedItem, HDEVINFO hDevInfo)
- {
- LPTSTR lpszMsg = NULL;
- HCURSOR hCursor = NULL;
- try
- {
- SP_PROPCHANGE_PARAMS PropChangeParams = {sizeof(SP_CLASSINSTALL_HEADER)};
- SP_DEVINFO_DATA DeviceInfoData = {sizeof(SP_DEVINFO_DATA)};
- hCursor = SetCursor(LoadCursor(NULL, IDC_WAIT));
- // Get a handle to the Selected Item.
- if (!SetupDiEnumDeviceInfo(hDevInfo, SelectedItem, &DeviceInfoData))
- {
- FormatMSG(GetLastError(), &lpszMsg);
- throw lpszMsg;
- }
- // Set the PropChangeParams structure.
- 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);
- throw lpszMsg;
- }
- // Call the ClassInstaller and perform the change.
- if (!SetupDiCallClassInstaller(DIF_PROPERTYCHANGE,hDevInfo,&DeviceInfoData))
- {
- FormatMSG(GetLastError(), &lpszMsg);
- throw lpszMsg;
- }
- SetCursor(hCursor);
- return TRUE;
- }
- catch (TCHAR * pszError)
- {
- SetCursor(hCursor);
- ::MessageBox(NULL,pszError,_T("提示"),MB_OK);
- if (NULL != lpszMsg)
- {
- LocalFree((HLOCAL)lpszMsg);
- }
- return FALSE;
- }
- }
- // 这些设备的启用和禁用主要有此函数实现
- // 参数说明,nStatus 可取3个值,-1 :啥也不做
- // 0 禁用设备,1启用设备。
- // nIndex用于控制是对光驱,软驱还是对USB启用或禁用
- // 也是可取3个值,0 代表软驱,1 代表光驱
- // 2 代表USB
- // 譬如要对光驱实现禁用,可以这样调用此函数
- // ControlDisk(0, 1);
- BOOL ControlDisk(int nStatus, int nIndex)
- {
- if (-1 == nStatus)
- {
- return FALSE;
- }
- LPTSTR lpszMsg = NULL;
- try
- {
- TCHAR * GUIDString = NULL;
- GUID guid;
- ZeroMemory(&guid, sizeof(GUID));
- switch(nIndex)
- {
- case 0: // 0 代表软驱
- GUIDString = _T("4D36E980-E325-11CE-BFC1-08002BE10318");
- UuidFromString((unsigned char *)GUIDString, &guid);
- break;
- case 1: // 1 代表光驱
- GUIDString = _T("4D36E965-E325-11CE-BFC1-08002BE10318");
- UuidFromString((unsigned char *)GUIDString, &guid);
- break;
- case 2: // 2 代表USB
- GUIDString = _T("36FC9E60-C465-11CF-8056-444553540000");
- UuidFromString((unsigned char *)GUIDString, &guid);
- break;
- }
- hDevInfo = SetupDiGetClassDevs(&guid,NULL,NULL,DIGCF_PRESENT);
- if (INVALID_HANDLE_VALUE != hDevInfo)
- {lpszMsg="hahazx";
- FormatMSG(GetLastError(), &lpszMsg);
- throw lpszMsg;
- }
- DWORD i;
- SP_DEVINFO_DATA DeviceInfoData;
- 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);
- }
- else if (0 == nStatus)
- {::MessageBox(NULL,"禁用","error",MB_OK);
- ChangeStatus(DICS_DISABLE, i, hDevInfo);
- }
- }
- // 释放 device information set
- return SetupDiDestroyDeviceInfoList(hDevInfo);
- }
- catch (TCHAR * pszError)
- {
- ::MessageBox(NULL,pszError,_T("提示"),MB_OK);
- if (NULL != lpszMsg)
- {
- LocalFree((HLOCAL)lpszMsg);
- }
- return FALSE;
- }
- return FALSE;
- }
- void main()
- {
- ControlDisk(1, 2);
- }
原文链接: http://blog.csdn.net/yincheng01/article/details/4244982