mirror of
https://github.com/pbatard/rufus.git
synced 2024-08-14 23:57:05 +00:00
f88faf1a4f
* Also fix the VID:PID population of USB card readers * Also improve enumeration debugging * Also add an unofficial cheat mode to list non USB *REMOVABLE* drives * Closes #693
182 lines
6.1 KiB
C
182 lines
6.1 KiB
C
/*
|
|
* Rufus: The Reliable USB Formatting Utility
|
|
* USB device listing
|
|
* Copyright © 2014-2016 Pete Batard <pete@akeo.ie>
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <windows.h>
|
|
|
|
#define USB_SPEED_UNKNOWN 0
|
|
#define USB_SPEED_LOW 1
|
|
#define USB_SPEED_FULL 2
|
|
#define USB_SPEED_HIGH 3
|
|
#define USB_SPEED_SUPER_OR_LATER 4
|
|
#define USB_SPEED_MAX 5
|
|
|
|
/* List of the properties we are interested in */
|
|
typedef struct usb_device_props {
|
|
uint32_t vid;
|
|
uint32_t pid;
|
|
uint32_t speed;
|
|
uint32_t port;
|
|
BOOLEAN is_USB;
|
|
BOOLEAN is_SCSI;
|
|
BOOLEAN is_CARD;
|
|
BOOLEAN is_UASP;
|
|
BOOLEAN is_VHD;
|
|
BOOLEAN is_Removable;
|
|
BOOLEAN is_LowerSpeed;
|
|
} usb_device_props;
|
|
|
|
/*
|
|
* Windows DDK API definitions. Most of it copied from MinGW's includes
|
|
*/
|
|
typedef DWORD DEVNODE, DEVINST;
|
|
typedef DEVNODE *PDEVNODE, *PDEVINST;
|
|
typedef DWORD RETURN_TYPE;
|
|
typedef RETURN_TYPE CONFIGRET;
|
|
typedef CHAR *DEVINSTID_A;
|
|
|
|
#define CR_SUCCESS 0x00000000
|
|
#define CR_NO_SUCH_DEVNODE 0x0000000D
|
|
#define CM_GETIDLIST_FILTER_SERVICE 0x00000002
|
|
#define CM_REMOVAL_POLICY_EXPECT_NO_REMOVAL 0x00000001
|
|
#define CM_REMOVAL_POLICY_EXPECT_ORDERLY_REMOVAL 0x00000002
|
|
#define CM_REMOVAL_POLICY_EXPECT_SURPRISE_REMOVAL 0x00000003
|
|
// /!\ The following flag is only available on Windows 7 or later!
|
|
#define CM_GETIDLIST_FILTER_PRESENT 0x00000100
|
|
#define CM_DRP_ADDRESS 0x0000001D
|
|
|
|
#ifndef METHOD_BUFFERED
|
|
#define METHOD_BUFFERED 0
|
|
#endif
|
|
#ifndef FILE_ANY_ACCESS
|
|
#define FILE_ANY_ACCESS 0x00000000
|
|
#endif
|
|
#ifndef FILE_DEVICE_UNKNOWN
|
|
#define FILE_DEVICE_UNKNOWN 0x00000022
|
|
#endif
|
|
#ifndef FILE_DEVICE_USB
|
|
#define FILE_DEVICE_USB FILE_DEVICE_UNKNOWN
|
|
#endif
|
|
|
|
#ifndef CTL_CODE
|
|
#define CTL_CODE(DeviceType, Function, Method, Access)( \
|
|
((DeviceType) << 16) | ((Access) << 14) | ((Function) << 2) | (Method))
|
|
#endif
|
|
|
|
typedef enum USB_CONNECTION_STATUS {
|
|
NoDeviceConnected,
|
|
DeviceConnected,
|
|
DeviceFailedEnumeration,
|
|
DeviceGeneralFailure,
|
|
DeviceCausedOvercurrent,
|
|
DeviceNotEnoughPower,
|
|
DeviceNotEnoughBandwidth,
|
|
DeviceHubNestedTooDeeply,
|
|
DeviceInLegacyHub
|
|
} USB_CONNECTION_STATUS, *PUSB_CONNECTION_STATUS;
|
|
|
|
typedef enum USB_HUB_NODE {
|
|
UsbHub,
|
|
UsbMIParent
|
|
} USB_HUB_NODE;
|
|
|
|
/* Cfgmgr32.dll interface */
|
|
DECLSPEC_IMPORT CONFIGRET WINAPI CM_Get_Device_IDA(DEVINST dnDevInst, PCSTR Buffer, ULONG BufferLen, ULONG ulFlags);
|
|
DECLSPEC_IMPORT CONFIGRET WINAPI CM_Get_Device_ID_List_SizeA(PULONG pulLen, PCSTR pszFilter, ULONG ulFlags);
|
|
DECLSPEC_IMPORT CONFIGRET WINAPI CM_Get_Device_ID_ListA(PCSTR pszFilter, PCHAR Buffer, ULONG BufferLen, ULONG ulFlags);
|
|
DECLSPEC_IMPORT CONFIGRET WINAPI CM_Locate_DevNodeA(PDEVINST pdnDevInst, DEVINSTID_A pDeviceID, ULONG ulFlags);
|
|
DECLSPEC_IMPORT CONFIGRET WINAPI CM_Get_Child(PDEVINST pdnDevInst, DEVINST dnDevInst, ULONG ulFlags);
|
|
DECLSPEC_IMPORT CONFIGRET WINAPI CM_Get_Parent(PDEVINST pdnDevInst, DEVINST dnDevInst, ULONG ulFlags);
|
|
DECLSPEC_IMPORT CONFIGRET WINAPI CM_Get_Sibling(PDEVINST pdnDevInst, DEVINST dnDevInst, ULONG ulFlags);
|
|
// This last one is unknown from MinGW32 and needs to be fetched from the DLL
|
|
PF_TYPE_DECL(WINAPI, CONFIGRET, CM_Get_DevNode_Registry_PropertyA, (DEVINST, ULONG, PULONG, PVOID, PULONG, ULONG));
|
|
|
|
#define USB_GET_NODE_CONNECTION_INFORMATION_EX 274
|
|
#define USB_GET_NODE_CONNECTION_INFORMATION_EX_V2 279
|
|
|
|
#define IOCTL_USB_GET_NODE_CONNECTION_INFORMATION_EX \
|
|
CTL_CODE(FILE_DEVICE_USB, USB_GET_NODE_CONNECTION_INFORMATION_EX, METHOD_BUFFERED, FILE_ANY_ACCESS)
|
|
#define IOCTL_USB_GET_NODE_CONNECTION_INFORMATION_EX_V2 \
|
|
CTL_CODE(FILE_DEVICE_USB, USB_GET_NODE_CONNECTION_INFORMATION_EX_V2, METHOD_BUFFERED, FILE_ANY_ACCESS)
|
|
|
|
/* Most of the structures below need to be packed */
|
|
#pragma pack(push, 1)
|
|
|
|
typedef struct _USB_DEVICE_DESCRIPTOR {
|
|
UCHAR bLength;
|
|
UCHAR bDescriptorType;
|
|
USHORT bcdUSB;
|
|
UCHAR bDeviceClass;
|
|
UCHAR bDeviceSubClass;
|
|
UCHAR bDeviceProtocol;
|
|
UCHAR bMaxPacketSize0;
|
|
USHORT idVendor;
|
|
USHORT idProduct;
|
|
USHORT bcdDevice;
|
|
UCHAR iManufacturer;
|
|
UCHAR iProduct;
|
|
UCHAR iSerialNumber;
|
|
UCHAR bNumConfigurations;
|
|
} USB_DEVICE_DESCRIPTOR, *PUSB_DEVICE_DESCRIPTOR;
|
|
|
|
typedef struct USB_NODE_CONNECTION_INFORMATION_EX {
|
|
ULONG ConnectionIndex;
|
|
USB_DEVICE_DESCRIPTOR DeviceDescriptor;
|
|
UCHAR CurrentConfigurationValue;
|
|
UCHAR Speed;
|
|
BOOLEAN DeviceIsHub;
|
|
USHORT DeviceAddress;
|
|
ULONG NumberOfOpenPipes;
|
|
USB_CONNECTION_STATUS ConnectionStatus;
|
|
// USB_PIPE_INFO PipeList[0];
|
|
} USB_NODE_CONNECTION_INFORMATION_EX, *PUSB_NODE_CONNECTION_INFORMATION_EX;
|
|
|
|
typedef union _USB_PROTOCOLS {
|
|
ULONG ul;
|
|
struct {
|
|
ULONG Usb110:1;
|
|
ULONG Usb200:1;
|
|
ULONG Usb300:1;
|
|
ULONG ReservedMBZ:29;
|
|
};
|
|
} USB_PROTOCOLS, *PUSB_PROTOCOLS;
|
|
|
|
typedef union _USB_NODE_CONNECTION_INFORMATION_EX_V2_FLAGS {
|
|
ULONG ul;
|
|
struct {
|
|
ULONG DeviceIsOperatingAtSuperSpeedOrHigher:1;
|
|
ULONG DeviceIsSuperSpeedCapableOrHigher:1;
|
|
ULONG ReservedMBZ:30;
|
|
};
|
|
} USB_NODE_CONNECTION_INFORMATION_EX_V2_FLAGS, *PUSB_NODE_CONNECTION_INFORMATION_EX_V2_FLAGS;
|
|
|
|
typedef struct _USB_NODE_CONNECTION_INFORMATION_EX_V2 {
|
|
ULONG ConnectionIndex;
|
|
ULONG Length;
|
|
USB_PROTOCOLS SupportedUsbProtocols;
|
|
USB_NODE_CONNECTION_INFORMATION_EX_V2_FLAGS Flags;
|
|
} USB_NODE_CONNECTION_INFORMATION_EX_V2, *PUSB_NODE_CONNECTION_INFORMATION_EX_V2;
|
|
|
|
#pragma pack(pop)
|
|
|
|
const GUID _GUID_DEVINTERFACE_DISK =
|
|
{ 0x53f56307L, 0xb6bf, 0x11d0, {0x94, 0xf2, 0x00, 0xa0, 0xc9, 0x1e, 0xfb, 0x8b} };
|
|
const GUID _GUID_DEVINTERFACE_USB_HUB =
|
|
{ 0xf18a0e88L, 0xc30c, 0x11d0, {0x88, 0x15, 0x00, 0xa0, 0xc9, 0x06, 0xbe, 0xd8} };
|
|
|
|
#define DEVID_HTAB_SIZE 257
|