mirrors/rufus
1
0
Fork 0
mirror of https://github.com/pbatard/rufus.git synced 2025-05-29 22:15:21 -04:00
Code Issues Projects Releases Packages Wiki Activity

[vhd] fix VHD detection on localized versions of Windows

Browse source 
* SPDRP_FRIENDLYNAME is localized, so we can't use it.
* Closes #417
This commit is contained in:
Pete Batard 2014-12-20 17:30:12 +00:00
parent d81f83c110
commit a4e12f5c12
2 changed files with 21 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

23
src/usb.c
View file

@ -109,12 +109,10 @@ static __inline BOOL IsVHD(const char* buffer)
int i;
// List of the Friendly Names of the VHD devices we know
const char* vhd_name[] = {
"Arsenal Virtual",
"Kernsafe Virtual",
"Microsoft Virtual",
"MS Virtual",
"Msft Virtual",
// "VMware Virtual" // Would list primary disks on VMWare instances, so we avoid it
"Arsenal_________Virtual_",
"KernSafeVirtual_________",
"Msft____Virtual_Disk____",
// "VMware__VMware_Virtual_S" // Would list primary disks on VMWare instances, so we avoid it
};
for (i = 0; i < ARRAYSIZE(vhd_name); i++)
@ -259,16 +257,21 @@ BOOL GetUSBDevices(DWORD devnum)
is_SCSI = (safe_stricmp(buffer, scsi_name) == 0);
if ((safe_stricmp(buffer, storage_name[0]) != 0) && (!is_SCSI))
continue;
memset(buffer, 0, sizeof(buffer));
// We can't use the friendly name to find if a drive is a VHD, as friendly name string gets translated
// according to your locale, so we poke the Hardware ID
memset(&props, 0, sizeof(props));
memset(buffer, 0, sizeof(buffer));
props.is_VHD = SetupDiGetDeviceRegistryPropertyA(dev_info, &dev_info_data, SPDRP_HARDWAREID,
&datatype, (LPBYTE)buffer, sizeof(buffer), &size) && IsVHD(buffer);
memset(buffer, 0, sizeof(buffer));
if (!SetupDiGetDeviceRegistryPropertyA(dev_info, &dev_info_data, SPDRP_FRIENDLYNAME,
&datatype, (LPBYTE)buffer, sizeof(buffer), &size)) {
uprintf("SetupDiGetDeviceRegistryProperty (Friendly Name) failed: %s\n", WindowsErrorString());
// We can afford a failure on this call - just replace the name with "USB Storage Device (Generic)"
safe_strcpy(buffer, sizeof(buffer), lmprintf(MSG_045));
} else if (IsVHD(buffer)) {
props.is_VHD = TRUE;
} else if (devid_list != NULL) {
} else if ((!props.is_VHD) && (devid_list != NULL)) {
// Get the properties of the device. We could avoid doing this lookup every time by keeping
// a lookup table, but there shouldn't be that many USB storage devices connected...
// NB: Each of these Device IDs have an _only_ child, from which we get the Device Instance match.