restore MAX_PARTITIONS

Alex313031 · Oct 29, 2024 · 10119d1 · 10119d1
1 parent d770df3
commit 10119d1
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/drive.c b/src/drive.c
@@ -2547,9 +2547,8 @@ BOOL CreatePartition(HANDLE hDrive, int partition_style, int file_system, BOOL m
 	// "The goggles, they do nothing!"
 	RefreshDriveLayout(hDrive);
 
-	const int max_parts = 4;
 	size = sizeof(DriveLayoutEx) - ((partition_style == PARTITION_STYLE_GPT) ?
-		((max_parts - pi) * sizeof(PARTITION_INFORMATION_EX)) : 0);
+		((MAX_PARTITIONS - pi) * sizeof(PARTITION_INFORMATION_EX)) : 0);
 	// The DRIVE_LAYOUT_INFORMATION_EX used by Microsoft, with its 1-sized array, is designed to overrun...
 	// coverity[overrun-buffer-arg]
 	if (!DeviceIoControl(hDrive, IOCTL_DISK_SET_DRIVE_LAYOUT_EX, (BYTE*)&DriveLayoutEx, size, NULL, 0, &size, NULL)) {