cmclark00/retro-imager
1
0
Fork 0
mirror of https://github.com/cmclark00/retro-imager.git synced 2025-05-18 16:05:21 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

drivelist: let GetEnumeratorName() return std::string instead of char *

Browse source 
Returning a pointer to a character array on the stack that is
no longer reserved after the function returns can lead to
undefined behavior.
This commit is contained in:
Floris Bos 2021-01-11 18:37:39 +01:00
parent 8e87933320
commit 7c1c111699
1 changed files with 8 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

16
dependencies/drivelist/src/windows/list.cpp vendored
View file

@ -69,7 +69,7 @@ std::string WCharToUtf8String(const wchar_t* wstr) {
return result;
}
char* GetEnumeratorName(HDEVINFO hDeviceInfo, SP_DEVINFO_DATA deviceInfoData) {
std::string GetEnumeratorName(HDEVINFO hDeviceInfo, SP_DEVINFO_DATA deviceInfoData) {
char buffer[MAX_PATH];
ZeroMemory(&buffer, sizeof(buffer));
@ -83,7 +83,7 @@ char* GetEnumeratorName(HDEVINFO hDeviceInfo, SP_DEVINFO_DATA deviceInfoData) {
hasEnumeratorName = SetupDiGetDevicePropertyW(hDeviceInfo, &deviceInfoData, &DEVPKEY_Device_EnumeratorName, NULL, (LPBYTE) buffer, sizeof(buffer), NULL, 0);
}*/
return hasEnumeratorName ? buffer : NULL;
return hasEnumeratorName ? std::string(buffer) : std::string();
}
std::string GetFriendlyName(HDEVINFO hDeviceInfo,
@ -619,7 +619,7 @@ std::vector<DeviceDescriptor> ListStorageDevices() {
std::vector<DeviceDescriptor> deviceList;
DWORD i;
char *enumeratorName;
std::string enumeratorName;
DeviceDescriptor device;
hDeviceInfo = SetupDiGetClassDevsA(
@ -636,21 +636,21 @@ std::vector<DeviceDescriptor> ListStorageDevices() {
for (i = 0; SetupDiEnumDeviceInfo(hDeviceInfo, i, &deviceInfoData); i++) {
enumeratorName = GetEnumeratorName(hDeviceInfo, deviceInfoData);
printf("[INFO] Enumerating %s\n", enumeratorName);
printf("[INFO] Enumerating %s\n", enumeratorName.c_str());
// If it failed to get the SPDRP_ENUMERATOR_NAME, skip it
//if (enumeratorName == NULL) {
//if (enumeratorName.empty()) {
// continue;
//}
device = DeviceDescriptor();
device.enumerator = enumeratorName ? std::string(enumeratorName) : "";
device.enumerator = enumeratorName;
device.description = GetFriendlyName(hDeviceInfo, deviceInfoData);
device.isRemovable = IsRemovableDevice(hDeviceInfo, deviceInfoData);
device.isVirtual = IsVirtualHardDrive(hDeviceInfo, deviceInfoData);
device.isSCSI = enumeratorName ? IsSCSIDevice(enumeratorName) : false;
device.isUSB = enumeratorName ? IsUSBDevice(enumeratorName) : false;
device.isSCSI = IsSCSIDevice(enumeratorName);
device.isUSB = IsUSBDevice(enumeratorName);
device.isCard = device.enumerator == "SD";
device.isSystem = !device.isRemovable &&
(device.enumerator == "SCSI" || device.enumerator == "IDE");