|
|
楼主 |
发表于 2006-11-20 09:51:04
|
显示全部楼层
#include "stdio.h"
#include "windows.h"
#include "rpcdce.h"
#include <setupapi.h>
#include <shlwapi.h>
#include "tchar.h"
//#include "stdlib.h"
//#include "Winbase.h"
#pragma comment(lib, "rpcrt4.lib")
#pragma comment(lib, "setupapi.lib")
#pragma comment(lib, "shlwapi.lib")
HDEVINFO hDevInfo = NULL;
void FormatMSG(DWORD dwError, LPTSTR * lpszMsg)
{
BOOL bOk = FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER, NULL, dwError,
MAKELANGID(LANG_CHINESE, SUBLANG_CHINESE_SIMPLIFIED), (LPTSTR)lpszMsg, 0, NULL);
if(!bOk)
{
HMODULE hDll = LoadLibraryEx(("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)lpszMsg,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));
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,"提示",MB_OK);
if (NULL != lpszMsg)
{
LocalFree((HLOCAL)lpszMsg);
}
return FALSE;
}
}
_declspec(dllexport) 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:
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 36FC9E60-C465-11CF-8056-444553540000
GUIDString = _T("71A27CDD-812A-11D0-BEC7-08002BE2092F");
UuidFromString((unsigned char *)GUIDString, &guid);
break;
}
hDevInfo = SetupDiGetClassDevs(&guid,NULL,NULL,DIGCF_PRESENT);
if (INVALID_HANDLE_VALUE == hDevInfo)
{
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)
{
ChangeStatus(DICS_ENABLE, i, hDevInfo);
}
else if (0 == nStatus)
{
ChangeStatus(DICS_DISABLE, i, hDevInfo);
}
}
return SetupDiDestroyDeviceInfoList(hDevInfo);
}
catch (TCHAR * pszError)
{
//::MessageBox(NULL,pszError,_T("提示"),MB_OK);
if (NULL != lpszMsg)
{
LocalFree((HLOCAL)lpszMsg);
}
return FALSE;
}
return FALSE;
}
以上就是dll程序了(没做头文件,不知道需要否????).labview调用ControlDisk函数.
当输入参数0,2时;卸载usb设备,labview程序就会暂时的未响应.
当输入参数1,2时;启用usb设备,labview程序正常.
请高手指点一下,谢谢 |
|