Rewrite disk type checking in udevProcessStorage

Error out on parsing errors and use a local const char pointer
instead of chained ifs to check whether we found a match.
This commit is contained in:
Ján Tomko 2016-06-03 18:24:43 +02:00
parent 4ccf6886c8
commit 71cddab042
1 changed files with 18 additions and 14 deletions

View File

@ -1019,30 +1019,34 @@ static int udevProcessStorage(struct udev_device *device,
"ID_TYPE",
&data->storage.drive_type) != PROPERTY_FOUND ||
STREQ(def->caps->data.storage.drive_type, "generic")) {
int tmp_int = 0;
int val = 0;
const char *str = NULL;
/* All floppy drives have the ID_DRIVE_FLOPPY prop. This is
* needed since legacy floppies don't have a drive_type */
if (udevGetIntProperty(device, "ID_DRIVE_FLOPPY",
&tmp_int, 0) == PROPERTY_FOUND &&
tmp_int == 1) {
if (udevGetIntProperty(device, "ID_DRIVE_FLOPPY", &val, 0) == PROPERTY_ERROR)
goto out;
else if (val == 1)
str = "floppy";
if (VIR_STRDUP(data->storage.drive_type, "floppy") < 0)
if (!str) {
if (udevGetIntProperty(device, "ID_CDROM", &val, 0) == PROPERTY_ERROR)
goto out;
} else if (udevGetIntProperty(device, "ID_CDROM",
&tmp_int, 0) == PROPERTY_FOUND &&
tmp_int == 1) {
else if (val == 1)
str = "cd";
}
if (VIR_STRDUP(data->storage.drive_type, "cd") < 0)
if (!str) {
if (udevGetIntProperty(device, "ID_DRIVE_FLASH_SD", &val, 0) == PROPERTY_ERROR)
goto out;
} else if (udevGetIntProperty(device, "ID_DRIVE_FLASH_SD",
&tmp_int, 0) == PROPERTY_FOUND &&
tmp_int == 1) {
if (val == 1)
str = "sd";
}
if (VIR_STRDUP(data->storage.drive_type, "sd") < 0)
if (str) {
if (VIR_STRDUP(data->storage.drive_type, str) < 0)
goto out;
} else {
/* If udev doesn't have it, perhaps we can guess it. */
if (udevKludgeStorageType(def) != 0)
goto out;