testusbdlg.cpp

一个关于usb控制的源码,内有readme。

        goto GetRootHubNameError;
    }

    success = DeviceIoControl(HostController,
                              IOCTL_USB_GET_ROOT_HUB_NAME,
                              NULL,
                              0,
                              rootHubNameW,
                              nBytes,
                              &nBytes,
                              NULL);

    if (!success)
    {
       goto GetRootHubNameError;
    }

    rootHubNameA = WideStrToMultiStr(rootHubNameW->RootHubName);
    free(rootHubNameW);

    return rootHubNameA;


GetRootHubNameError:
    if (rootHubNameW != NULL)
    {
        free(rootHubNameW);
        rootHubNameW = NULL;
		ShowMessage("出错!\r\n");
    }

    return NULL;
}

PCHAR CTestUSBDlg::GetDriverKeyName(HANDLE Hub, ULONG ConnectionIndex)
{
    BOOL                                success;
    ULONG                               nBytes;
    USB_NODE_CONNECTION_DRIVERKEY_NAME  driverKeyName;
    PUSB_NODE_CONNECTION_DRIVERKEY_NAME driverKeyNameW;
    PCHAR                               driverKeyNameA;

    driverKeyNameW = NULL;
    driverKeyNameA = NULL;

    driverKeyName.ConnectionIndex = ConnectionIndex;

    success = DeviceIoControl(Hub,
                              IOCTL_USB_GET_NODE_CONNECTION_DRIVERKEY_NAME,
                              &driverKeyName,
                              sizeof(driverKeyName),
                              &driverKeyName,
                              sizeof(driverKeyName),
                              &nBytes,
                              NULL);

    if (!success)
    {
        goto GetDriverKeyNameError;
    }

    nBytes = driverKeyName.ActualLength;

    if (nBytes <= sizeof(driverKeyName))
    {
        goto GetDriverKeyNameError;
    }

    driverKeyNameW = (PUSB_NODE_CONNECTION_DRIVERKEY_NAME)malloc(nBytes);

    if (driverKeyNameW == NULL)
    {
        goto GetDriverKeyNameError;
    }

    driverKeyNameW->ConnectionIndex = ConnectionIndex;

    success = DeviceIoControl(Hub,
                              IOCTL_USB_GET_NODE_CONNECTION_DRIVERKEY_NAME,
                              driverKeyNameW,
                              nBytes,
                              driverKeyNameW,
                              nBytes,
                              &nBytes,
                              NULL);

    if (!success)
    {
        goto GetDriverKeyNameError;
    }
    driverKeyNameA = WideStrToMultiStr(driverKeyNameW->DriverKeyName);
    free(driverKeyNameW);

    return driverKeyNameA;


GetDriverKeyNameError:
    if (driverKeyNameW != NULL)
    {
        free(driverKeyNameW);
        driverKeyNameW = NULL;
    }

    return NULL;
}

PUSB_DESCRIPTOR_REQUEST CTestUSBDlg::GetConfigDescriptor(HANDLE hHubDevice, ULONG ConnectionIndex, UCHAR DescriptorIndex)
{
   BOOL    success;
    ULONG   nBytes;
    ULONG   nBytesReturned;

    UCHAR   configDescReqBuf[sizeof(USB_DESCRIPTOR_REQUEST) +
                             sizeof(USB_CONFIGURATION_DESCRIPTOR)];

    PUSB_DESCRIPTOR_REQUEST         configDescReq;
    PUSB_CONFIGURATION_DESCRIPTOR   configDesc;


    nBytes = sizeof(configDescReqBuf);

    configDescReq = (PUSB_DESCRIPTOR_REQUEST)configDescReqBuf;
    configDesc = (PUSB_CONFIGURATION_DESCRIPTOR)(configDescReq+1);

    memset(configDescReq, 0, nBytes);
    configDescReq->ConnectionIndex = ConnectionIndex;

    configDescReq->SetupPacket.wValue = (USB_CONFIGURATION_DESCRIPTOR_TYPE << 8)
                                        | DescriptorIndex;

    configDescReq->SetupPacket.wLength = (USHORT)(nBytes - sizeof(USB_DESCRIPTOR_REQUEST));

    success = DeviceIoControl(hHubDevice,
                              IOCTL_USB_GET_DESCRIPTOR_FROM_NODE_CONNECTION,
                              configDescReq,
                              nBytes,
                              configDescReq,
                              nBytes,
                              &nBytesReturned,
                              NULL);

    if (!success)
    {

        return NULL;
    }

    if (nBytes != nBytesReturned)
    {
        return NULL;
    }

    if (configDesc->wTotalLength < sizeof(USB_CONFIGURATION_DESCRIPTOR))
    {

        return NULL;
    }
    nBytes = sizeof(USB_DESCRIPTOR_REQUEST) + configDesc->wTotalLength;

    configDescReq = (PUSB_DESCRIPTOR_REQUEST)malloc(nBytes);

    if (configDescReq == NULL)
    {

        return NULL;
    }

    configDesc = (PUSB_CONFIGURATION_DESCRIPTOR)(configDescReq+1);

    configDescReq->ConnectionIndex = ConnectionIndex;

    configDescReq->SetupPacket.wValue = (USB_CONFIGURATION_DESCRIPTOR_TYPE << 8)
                                        | DescriptorIndex;

    configDescReq->SetupPacket.wLength = (USHORT)(nBytes - sizeof(USB_DESCRIPTOR_REQUEST));

    success = DeviceIoControl(hHubDevice,
                              IOCTL_USB_GET_DESCRIPTOR_FROM_NODE_CONNECTION,
                              configDescReq,
                              nBytes,
                              configDescReq,
                              nBytes,
                              &nBytesReturned,
                              NULL);

    if (!success)
    {
        free(configDescReq);
        return NULL;
    }

    if (nBytes != nBytesReturned)
    {
        free(configDescReq);
        return NULL;
    }

    if (configDesc->wTotalLength != (nBytes - sizeof(USB_DESCRIPTOR_REQUEST)))
    {

        free(configDescReq);
        return NULL;
    }

    return configDescReq;
}

void CTestUSBDlg::ShowCenctInfo(PUSB_NODE_CONNECTION_INFORMATION connectionInfo)
{
		CString strShow;
		strShow.Format("connection index:[Port:%d]\r\n",connectionInfo->ConnectionIndex);
		ShowMessage(strShow);
			strShow.Format("Device class:0x%X,idVendor:0x%X,\r\nidProduct:0x%X,\r\n ",
				connectionInfo->DeviceDescriptor.bDeviceClass,
				connectionInfo->DeviceDescriptor.idVendor,
				connectionInfo->DeviceDescriptor.idProduct);
			ShowMessage(strShow);
			  strShow.Format("iManufacturer:0x%X,\r\niSerialNumber:0x%X,\r\n",
				  connectionInfo->DeviceDescriptor.iManufacturer,
				  connectionInfo->DeviceDescriptor.iSerialNumber);
			  ShowMessage(strShow);
			  if(connectionInfo->LowSpeed)
				  strShow="bus speed：low,\r\n";
			  else
				  strShow="bus speed：full,\r\n";
			  ShowMessage(strShow);
			  strShow.Format("Device Address:0x%X,\r\nOpen Pipes:0x%X\r\n",
			  connectionInfo->DeviceAddress,connectionInfo->NumberOfOpenPipes);
			  ShowMessage(strShow);

}

PCHAR CTestUSBDlg::WideStrToMultiStr(PWCHAR WideStr)
{
   ULONG nBytes;
    PCHAR MultiStr;
    nBytes = WideCharToMultiByte(
                 CP_ACP,
                 0,
                 WideStr,
                 -1,
                 NULL,
                 0,
                 NULL,
                 NULL);

    if (nBytes == 0)
    {
        return NULL;
    }
    MultiStr =(PCHAR) malloc(nBytes);

    if (MultiStr == NULL)
    {
        return NULL;
    }
    nBytes = WideCharToMultiByte(
                 CP_ACP,
                 0,
                 WideStr,
                 -1,
                 MultiStr,
                 nBytes,
                 NULL,
                 NULL);

    if (nBytes == 0)
    {
        free(MultiStr);
        return NULL;
    }

    return MultiStr;
}

void CTestUSBDlg::ShowHidInfo()//不能打开USB鼠标。能打开USB游戏手柄等
{
	CString strShow;
	strShow="正在查找系统中HID类的GUID标识...\r\n";
	ShowMessage(strShow);
	GUID guidHID;
	HidD_GetHidGuid(&guidHID);

	strShow="HID类的GUID标识为:\r\n";
	ShowMessage(strShow);
	strShow.Format("%08x-%04x-%04x-%02x-%02x-%02x-%02x-%02x-%02x-%02x-%02x\r\n",
		guidHID.Data1,guidHID.Data2,guidHID.Data3,guidHID.Data4[0],
		guidHID.Data4[1],guidHID.Data4[2],guidHID.Data4[3],guidHID.Data4[4],
		guidHID.Data4[5],guidHID.Data4[6],guidHID.Data4[7]);
	ShowMessage(strShow);

		HDEVINFO hDevInfo = SetupDiGetClassDevs(&guidHID,NULL,0,
		DIGCF_PRESENT|DIGCF_DEVICEINTERFACE );
	
	if(hDevInfo==INVALID_HANDLE_VALUE)
	{
		ShowMessage("查找USB设备出错!\r\n");
		return;
	}
	else
		ShowMessage("找到USB设备.\r\n");
	UINT nIndex = 0;
	strShow= "正在查找可用的USB设备...\r\n";
	ShowMessage(strShow);

	SP_DEVICE_INTERFACE_DATA strtInterfaceData;

	strtInterfaceData.cbSize=sizeof(SP_DEVICE_INTERFACE_DATA);
	BOOL bSuccess ;
	DWORD index=0;
	CString temp;
	for(int i=0;i<10;i++)
	{
	bSuccess= SetupDiEnumDeviceInterfaces(hDevInfo,NULL,&guidHID,index,
		&strtInterfaceData);
	if (!bSuccess)
	{	
		ShowMessage("查找USB设备出错!\r\n");
		break;
	}
	else
	{
		if(strtInterfaceData.Flags==SPINT_ACTIVE )
		{
			temp.Format("%d 连接 \r\n",index+1);
			ShowMessage(temp);
			ShowMore(hDevInfo,strtInterfaceData);
		}
	}
	index++;

	}
	temp.Format("总共%d个设备连接",index);
	ShowMessage(temp);
}

void CTestUSBDlg::ShowMore(HDEVINFO hDevInfo, SP_DEVICE_INTERFACE_DATA strtInterfaceData)
{
	CString strShow;
PSP_DEVICE_INTERFACE_DETAIL_DATA strtDetailData;

	DWORD strSzie=0,requiesize=0;
	SetupDiGetDeviceInterfaceDetail(hDevInfo,&strtInterfaceData,NULL,0,&strSzie,NULL);

	requiesize=strSzie;
	strtDetailData=(PSP_DEVICE_INTERFACE_DETAIL_DATA)malloc(requiesize);
	strtDetailData->cbSize=sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA);

	SP_DEVINFO_DATA infodata;
	infodata.cbSize=sizeof(SP_DEVINFO_DATA);
	if (!SetupDiGetDeviceInterfaceDetail(hDevInfo,&strtInterfaceData,
		strtDetailData,strSzie,&requiesize,NULL))
	{
		ShowMessage("查找设备路径时出错!");
		SetupDiDestroyDeviceInfoList(hDevInfo);
		free(strtDetailData);
		return;
	}
	strShow.Format("设备路径:\r\n%s\r\n",strtDetailData->DevicePath);
	ShowMessage(strShow);
	ShowMessage("打开通信端口....\r\n");
		HANDLE hCom = CreateFile (
                  strtDetailData->DevicePath,
                  GENERIC_READ | GENERIC_WRITE,
                  FILE_SHARE_READ | FILE_SHARE_WRITE,
                  NULL, 
                  OPEN_EXISTING, 0, 
                  NULL);
	if (hCom == INVALID_HANDLE_VALUE)
	{
	ShowMessage("无法打开通信端口。不能读取设备信息!(设备连接可能不正确?)\r\n");
		SetupDiDestroyDeviceInfoList(hDevInfo);
		free(strtDetailData);
		return;
	}
	ShowMessage("打开通信端口成功！\r\n");
		HIDD_ATTRIBUTES strtAttrib;
	ShowMessage("正在读取设备的标识...\r\n");

	strtAttrib.Size=sizeof(HIDD_ATTRIBUTES);
	if (!HidD_GetAttributes(hCom,&strtAttrib))
	{
		ShowMessage("查询设备状态时出错!");
		CloseHandle(hCom);

		free(strtDetailData);
		return;	
	}

	ShowMessage("已读取，如下所示:\r\n");
	strShow.Format("VendorID:0x%X\r\n",strtAttrib.VendorID);
	ShowMessage(strShow);
	strShow.Format("ProductID:0x%X\r\n",strtAttrib.ProductID);
	ShowMessage(strShow);
	strShow.Format("VersionNumber:0x%X\r\n",strtAttrib.VersionNumber);
	ShowMessage(strShow);

}
BOOL GetDisksProperty(HANDLE hDevice, PSTORAGE_DEVICE_DESCRIPTOR pDevDesc)
{
	STORAGE_PROPERTY_QUERY	Query;	
	DWORD dwOutBytes;				
	BOOL bResult;					

	Query.PropertyId = StorageDeviceProperty;
	Query.QueryType = PropertyStandardQuery;

	bResult = ::DeviceIoControl(hDevice,			
			IOCTL_STORAGE_QUERY_PROPERTY,			
			&Query, sizeof(STORAGE_PROPERTY_QUERY),	
			pDevDesc, pDevDesc->Size,				
			&dwOutBytes,							
			(LPOVERLAPPED)NULL);					

	return bResult;
}
BOOL CTestUSBDlg::PreTranslateMessage(MSG* pMsg) 
{
	// TODO: Add your specialized code here and/or call the base class
	
	return CDialog::PreTranslateMessage(pMsg);
}

char CTestUSBDlg::chFirstDriveFromMask(ULONG unitmask)
{
      char i;
      for (i = 0; i < 26; ++i)  
      {
           if (unitmask & 0x1) 
				break;
            unitmask = unitmask >> 1;
      }
    return (i + 'A');
}

void CTestUSBDlg::ShowUdisk()
{
	int k = 0;
	DWORD			MaxDriveSet, CurDriveSet;
	DWORD			drive, drivetype;
	TCHAR			szBuf[300];
	HANDLE			hDevice;
	PSTORAGE_DEVICE_DESCRIPTOR pDevDesc;

		char* p;

		CString str;
	for(k=0; k<26; k++)
		szMoveDiskName[k] = '\0';	
	k = 1;		
	MaxDriveSet = CurDriveSet = 0;

	MaxDriveSet = GetLogicalDrives();
	CurDriveSet = MaxDriveSet;
	for ( drive = 0; drive < 32; ++drive )  
	{
		if ( MaxDriveSet & (1 << drive) )  
		{
			DWORD temp = 1<<drive;
			_stprintf( szDrvName, _T("%c:\\"), 'A'+drive );
			if(GetDriveType( szDrvName )== DRIVE_REMOVABLE)
			{
					drivetype = DRVREMOVE;

					sprintf(szBuf, "\\\\?\\%c:", 'A'+drive);
					hDevice = CreateFile(szBuf, GENERIC_READ,
							FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, NULL, NULL);

					if (hDevice != INVALID_HANDLE_VALUE)
					{

						pDevDesc = (PSTORAGE_DEVICE_DESCRIPTOR)new BYTE[sizeof(STORAGE_DEVICE_DESCRIPTOR) + 512 - 1];

						pDevDesc->Size = sizeof(STORAGE_DEVICE_DESCRIPTOR) + 512 - 1;

						if(GetDisksProperty(hDevice, pDevDesc))
						{
								szMoveDiskName[k] = chFirstDriveFromMask(temp);
								szMoveDiskName[0]=k;
								k++;
								ShowMessage("\r\n找到U盘，名称:\r\n");
								p= (char*)pDevDesc;
								str = (pDevDesc->VendorIdOffset ? &p[pDevDesc->VendorIdOffset]:"(NULL)");
								ShowMessage(str);

								ShowMessage("\r\n产品ID:\r\n");
								str=(pDevDesc->ProductIdOffset ? &p[pDevDesc->ProductIdOffset]:"(NULL)");
								ShowMessage(str);
								ShowMessage("\r\n产品版本:\r\n");
								str = (pDevDesc->ProductRevisionOffset ? &p[pDevDesc->ProductRevisionOffset] : "(NULL)");
								ShowMessage(str);
								ShowMessage("\r\n序列号:\r\n");
								str = (pDevDesc->SerialNumberOffset ? &p[pDevDesc->SerialNumberOffset] : "(NULL)");
								ShowMessage(str);
						}

						delete pDevDesc;
						CloseHandle(hDevice);
					}
			}
			}

		}
		str.Format("\r\n U盘符:\r\n%s",&szMoveDiskName[1]);

		if(szMoveDiskName[0]==0)
		{
			str="\r\n没有U盘\r\n";
			m_showMessage.Empty();
		}
	ShowMessage(str);
}