fdisk: fix possible memory leak

karelzak · karelzak · commit 138117a83e79 · 2025-04-07T14:16:17.000+02:00
Signed-off-by: Karel Zak <kzak@redhat.com> (cherry picked from commit 7171968)
diff --git a/disk-utils/fdisk.c b/disk-utils/fdisk.c
@@ -652,27 +652,24 @@ static int strtosize_sectors(const char *str, unsigned long sector_size,
 	size_t len = strlen(str);
 	int insec = 0;
 	int rc;
+	char *buf = NULL;
 
 	if (!len)
 		return 0;
 
 	if (str[len - 1] == 'S' || str[len - 1] == 's') {
 		insec = 1;
-		str = strndup(str, len - 1); /* strip trailing 's' */
+		str = buf = strndup(str, len - 1); /* strip trailing 's' */
 		if (!str)
 			return -errno;
 	}
 
 	rc = strtosize(str, res);
-	if (rc)
-		return rc;
-
-	if (insec) {
+	if (rc == 0 && insec)
 		*res *= sector_size;
-		free((void *)str);
-	}
 
-	return 0;
+	free(buf);
+	return rc;
 }
 
 void resize_partition(struct fdisk_context *cxt)
