mirror of
				https://github.com/pbatard/rufus.git
				synced 2024-08-14 23:57:05 +00:00 
			
		
		
		
	[core] refactor CreatePartition()
* Also add groundwork for optional creation of MSR (see #432) * Also fix a few minor issues
This commit is contained in:
		
							parent
							
								
									0440e1fb3d
								
							
						
					
					
						commit
						ec761dfb41
					
				
					 3 changed files with 129 additions and 123 deletions
				
			
		
							
								
								
									
										232
									
								
								src/drive.c
									
										
									
									
									
								
							
							
						
						
									
										232
									
								
								src/drive.c
									
										
									
									
									
								
							|  | @ -39,7 +39,11 @@ | ||||||
| 
 | 
 | ||||||
| #if !defined(PARTITION_BASIC_DATA_GUID) | #if !defined(PARTITION_BASIC_DATA_GUID) | ||||||
| const GUID PARTITION_BASIC_DATA_GUID =  | const GUID PARTITION_BASIC_DATA_GUID =  | ||||||
| 	{ 0xebd0a0a2, 0xb9e5, 0x4433, {0x87, 0xc0, 0x68, 0xb6, 0xb7, 0x26, 0x99, 0xc7} }; | 	{ 0xebd0a0a2L, 0xb9e5, 0x4433, {0x87, 0xc0, 0x68, 0xb6, 0xb7, 0x26, 0x99, 0xc7} }; | ||||||
|  | #endif | ||||||
|  | #if !defined(PARTITION_MSFT_RESERVED_GUID) | ||||||
|  | const GUID PARTITION_MSFT_RESERVED_GUID = | ||||||
|  | 	{ 0xe3c9e316L, 0x0b5c, 0x4db8, {0x81, 0x7d, 0xf9, 0x2d, 0xf0, 0x02, 0x15, 0xae} }; | ||||||
| #endif | #endif | ||||||
| 
 | 
 | ||||||
| /*
 | /*
 | ||||||
|  | @ -912,33 +916,133 @@ BOOL CreatePartition(HANDLE hDrive, int partition_style, int file_system, BOOL m | ||||||
| 	CREATE_DISK CreateDisk = {PARTITION_STYLE_RAW, {{0}}}; | 	CREATE_DISK CreateDisk = {PARTITION_STYLE_RAW, {{0}}}; | ||||||
| 	DRIVE_LAYOUT_INFORMATION_EX4 DriveLayoutEx = {0}; | 	DRIVE_LAYOUT_INFORMATION_EX4 DriveLayoutEx = {0}; | ||||||
| 	BOOL r; | 	BOOL r; | ||||||
| 	DWORD size, bufsize; | 	DWORD i, size, bufsize, pn = 0; | ||||||
| 	LONGLONG size_in_sectors, extra_size_in_tracks = 1; | 	LONGLONG size_in_sectors, extra_part_size_in_tracks = 0; | ||||||
|  | 	BOOL add_msr = FALSE; //TRUE;
 | ||||||
| 
 | 
 | ||||||
| 	PrintInfoDebug(0, MSG_238, PartitionTypeName[partition_style]); | 	PrintInfoDebug(0, MSG_238, PartitionTypeName[partition_style]); | ||||||
|  | 
 | ||||||
| 	if (uefi_togo_size == 0) | 	if (uefi_togo_size == 0) | ||||||
| 		uefi_togo_size = GetResourceSize(hMainInstance, MAKEINTRESOURCEA(IDR_UEFI_TOGO), _RT_RCDATA, "uefi-togo.img"); | 		uefi_togo_size = GetResourceSize(hMainInstance, MAKEINTRESOURCEA(IDR_UEFI_TOGO), _RT_RCDATA, "uefi-togo.img"); | ||||||
| 
 | 
 | ||||||
|  | 	// Compute the start offset of our first partition
 | ||||||
| 	if ((partition_style == PARTITION_STYLE_GPT) || (!IsChecked(IDC_EXTRA_PARTITION))) { | 	if ((partition_style == PARTITION_STYLE_GPT) || (!IsChecked(IDC_EXTRA_PARTITION))) { | ||||||
| 		// Go with the MS 1 MB wastage at the beginning...
 | 		// Go with the MS 1 MB wastage at the beginning...
 | ||||||
| 		DriveLayoutEx.PartitionEntry[0].StartingOffset.QuadPart = 1024*1024; | 		DriveLayoutEx.PartitionEntry[pn].StartingOffset.QuadPart = 1024*1024; | ||||||
| 	} else { | 	} else { | ||||||
| 		// Align on Cylinder
 | 		// Align on Cylinder
 | ||||||
| 		DriveLayoutEx.PartitionEntry[0].StartingOffset.QuadPart =  | 		DriveLayoutEx.PartitionEntry[pn].StartingOffset.QuadPart =  | ||||||
| 			SelectedDrive.Geometry.BytesPerSector * SelectedDrive.Geometry.SectorsPerTrack; | 			SelectedDrive.Geometry.BytesPerSector * SelectedDrive.Geometry.SectorsPerTrack; | ||||||
| 	} | 	} | ||||||
| 	size_in_sectors = (SelectedDrive.DiskSize - DriveLayoutEx.PartitionEntry[0].StartingOffset.QuadPart) / SelectedDrive.Geometry.BytesPerSector; | 
 | ||||||
| 	// Align on track boundary if the extra part option is checked
 | 	// If required, set the MSR partition (GPT only - must be created before the data part)
 | ||||||
|  | 	if ((partition_style == PARTITION_STYLE_GPT) && (add_msr)) { | ||||||
|  | 		uprintf("Adding MSR partition"); | ||||||
|  | 		DriveLayoutEx.PartitionEntry[pn].PartitionLength.QuadPart = 128*1024*1024; | ||||||
|  | 		DriveLayoutEx.PartitionEntry[pn].Gpt.PartitionType = PARTITION_MSFT_RESERVED_GUID; | ||||||
|  | 		IGNORE_RETVAL(CoCreateGuid(&DriveLayoutEx.PartitionEntry[pn].Gpt.PartitionId)); | ||||||
|  | 		wcscpy(DriveLayoutEx.PartitionEntry[pn].Gpt.Name, L"Microsoft reserved partition"); | ||||||
|  | 		pn++; | ||||||
|  | 		DriveLayoutEx.PartitionEntry[pn].StartingOffset.QuadPart = DriveLayoutEx.PartitionEntry[pn-1].StartingOffset.QuadPart + | ||||||
|  | 				DriveLayoutEx.PartitionEntry[pn-1].PartitionLength.QuadPart; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Set our main data partition
 | ||||||
|  | 	size_in_sectors = (SelectedDrive.DiskSize - DriveLayoutEx.PartitionEntry[pn].StartingOffset.QuadPart) / | ||||||
|  | 		// Need 33 sectors at the end for secondary GPT
 | ||||||
|  | 		SelectedDrive.Geometry.BytesPerSector - ((partition_style == PARTITION_STYLE_GPT)?33:0); | ||||||
|  | 	// Adjust the size according to extra partitions
 | ||||||
| 	if ((add_uefi_togo) || ((partition_style == PARTITION_STYLE_MBR) && (IsChecked(IDC_EXTRA_PARTITION)))) { | 	if ((add_uefi_togo) || ((partition_style == PARTITION_STYLE_MBR) && (IsChecked(IDC_EXTRA_PARTITION)))) { | ||||||
| 		if (add_uefi_togo)	// Already set to 1 track in non To_Go mode
 | 		if (add_uefi_togo)	{ | ||||||
| 			extra_size_in_tracks = (MIN_EXTRA_PART_SIZE + SelectedDrive.Geometry.SectorsPerTrack - 1) / | 			extra_part_size_in_tracks = (MIN_EXTRA_PART_SIZE + SelectedDrive.Geometry.SectorsPerTrack - 1) / | ||||||
| 				SelectedDrive.Geometry.SectorsPerTrack; | 				SelectedDrive.Geometry.SectorsPerTrack; | ||||||
| 		uprintf("Reserving %d tracks for extra partition", extra_size_in_tracks); | 		} else { | ||||||
| 		size_in_sectors = ((size_in_sectors / SelectedDrive.Geometry.SectorsPerTrack) - extra_size_in_tracks) * | 			extra_part_size_in_tracks = 1;	// One track for the extra part in non togo mode
 | ||||||
|  | 		} | ||||||
|  | 		uprintf("Reserving %d tracks for extra partition", extra_part_size_in_tracks); | ||||||
|  | 		size_in_sectors = ((size_in_sectors / SelectedDrive.Geometry.SectorsPerTrack) - extra_part_size_in_tracks) * | ||||||
| 			SelectedDrive.Geometry.SectorsPerTrack; | 			SelectedDrive.Geometry.SectorsPerTrack; | ||||||
| 		if (size_in_sectors <= 0) | 		if (size_in_sectors <= 0) | ||||||
| 			return FALSE; | 			return FALSE; | ||||||
| 	} | 	} | ||||||
|  | 	DriveLayoutEx.PartitionEntry[pn].PartitionLength.QuadPart = size_in_sectors * SelectedDrive.Geometry.BytesPerSector; | ||||||
|  | 	if (partition_style == PARTITION_STYLE_MBR) { | ||||||
|  | 		DriveLayoutEx.PartitionEntry[pn].Mbr.BootIndicator = IsChecked(IDC_BOOT); | ||||||
|  | 		DriveLayoutEx.PartitionEntry[pn].Mbr.HiddenSectors = SelectedDrive.Geometry.SectorsPerTrack; | ||||||
|  | 		switch (file_system) { | ||||||
|  | 		case FS_FAT16: | ||||||
|  | 			DriveLayoutEx.PartitionEntry[pn].Mbr.PartitionType = 0x0e;	// FAT16 LBA
 | ||||||
|  | 			break; | ||||||
|  | 		case FS_NTFS: | ||||||
|  | 		case FS_EXFAT: | ||||||
|  | 		case FS_UDF: | ||||||
|  | 		case FS_REFS: | ||||||
|  | 			DriveLayoutEx.PartitionEntry[pn].Mbr.PartitionType = 0x07; | ||||||
|  | 			break; | ||||||
|  | 		case FS_FAT32: | ||||||
|  | 			DriveLayoutEx.PartitionEntry[pn].Mbr.PartitionType = 0x0c;	// FAT32 LBA
 | ||||||
|  | 			break; | ||||||
|  | 		default: | ||||||
|  | 			uprintf("Unsupported file system\n"); | ||||||
|  | 			return FALSE; | ||||||
|  | 		} | ||||||
|  | 	} else { | ||||||
|  | 		DriveLayoutEx.PartitionEntry[pn].Gpt.PartitionType = PARTITION_BASIC_DATA_GUID; | ||||||
|  | 		IGNORE_RETVAL(CoCreateGuid(&DriveLayoutEx.PartitionEntry[pn].Gpt.PartitionId)); | ||||||
|  | 		wcscpy(DriveLayoutEx.PartitionEntry[pn].Gpt.Name, L"Microsoft Basic Data"); | ||||||
|  | 	} | ||||||
|  | 	pn++; | ||||||
|  | 
 | ||||||
|  | 	// Set the optional extra partition
 | ||||||
|  | 	if (IsChecked(IDC_EXTRA_PARTITION) || (add_uefi_togo)) { | ||||||
|  | 		// Should end on a track boundary
 | ||||||
|  | 		DriveLayoutEx.PartitionEntry[pn].StartingOffset.QuadPart = DriveLayoutEx.PartitionEntry[pn-1].StartingOffset.QuadPart + | ||||||
|  | 				DriveLayoutEx.PartitionEntry[pn-1].PartitionLength.QuadPart; | ||||||
|  | 		DriveLayoutEx.PartitionEntry[pn].PartitionLength.QuadPart = (add_uefi_togo)?uefi_togo_size: | ||||||
|  | 			extra_part_size_in_tracks * SelectedDrive.Geometry.SectorsPerTrack * SelectedDrive.Geometry.BytesPerSector; | ||||||
|  | 		if (partition_style == PARTITION_STYLE_GPT) { | ||||||
|  | 			if (add_uefi_togo)	// already set to UNUSED GUID (000...000) otherwise
 | ||||||
|  | 				DriveLayoutEx.PartitionEntry[pn].Gpt.PartitionType = PARTITION_BASIC_DATA_GUID; | ||||||
|  | 			IGNORE_RETVAL(CoCreateGuid(&DriveLayoutEx.PartitionEntry[pn].Gpt.PartitionId)); | ||||||
|  | 			wcscpy(DriveLayoutEx.PartitionEntry[pn].Gpt.Name, (add_uefi_togo)?L"UEFI:TOGO":L"Rufus Extra Partition"); | ||||||
|  | 		} else { | ||||||
|  | 			DriveLayoutEx.PartitionEntry[pn].Mbr.PartitionType = (add_uefi_togo)?0x01:RUFUS_EXTRA_PARTITION_TYPE; | ||||||
|  | 			DriveLayoutEx.PartitionEntry[pn].Mbr.HiddenSectors = SelectedDrive.Geometry.SectorsPerTrack; | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// We need to write the TOGO partition before we refresh the disk
 | ||||||
|  | 		if (add_uefi_togo) { | ||||||
|  | 			uprintf("Writing UEFI:TOGO partition..."); | ||||||
|  | 			if (!SetFilePointerEx(hDrive, DriveLayoutEx.PartitionEntry[pn].StartingOffset, NULL, FILE_BEGIN)) { | ||||||
|  | 				uprintf("Unable to set position"); | ||||||
|  | 				safe_closehandle(hDrive); | ||||||
|  | 				return FALSE; | ||||||
|  | 			} | ||||||
|  | 			buffer = GetResource(hMainInstance, MAKEINTRESOURCEA(IDR_UEFI_TOGO), _RT_RCDATA, "uefi-togo.img", &bufsize, FALSE); | ||||||
|  | 			if (buffer == NULL) { | ||||||
|  | 				uprintf("Could not access uefi-togo.img"); | ||||||
|  | 				safe_closehandle(hDrive); | ||||||
|  | 				return FALSE; | ||||||
|  | 			} | ||||||
|  | 			r = WriteFile(hDrive, buffer, bufsize, &size, NULL); | ||||||
|  | 			if ((!r) || (size != bufsize)) { | ||||||
|  | 				if (!r) | ||||||
|  | 					uprintf("Write error: %s", WindowsErrorString()); | ||||||
|  | 				else | ||||||
|  | 					uprintf("Write error: Wrote %d bytes, expected %d bytes\n", size, bufsize); | ||||||
|  | 				safe_closehandle(hDrive); | ||||||
|  | 				return FALSE; | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		pn++; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Initialize the remaining partition data
 | ||||||
|  | 	for (i = 0; i < pn; i++) { | ||||||
|  | 		DriveLayoutEx.PartitionEntry[i].PartitionNumber = i+1; | ||||||
|  | 		DriveLayoutEx.PartitionEntry[i].PartitionStyle = partition_style; | ||||||
|  | 		DriveLayoutEx.PartitionEntry[i].RewritePartition = TRUE; | ||||||
|  | 	} | ||||||
| 
 | 
 | ||||||
| 	switch (partition_style) { | 	switch (partition_style) { | ||||||
| 	case PARTITION_STYLE_MBR: | 	case PARTITION_STYLE_MBR: | ||||||
|  | @ -952,7 +1056,6 @@ BOOL CreatePartition(HANDLE hDrive, int partition_style, int file_system, BOOL m | ||||||
| 		DriveLayoutEx.PartitionStyle = PARTITION_STYLE_MBR; | 		DriveLayoutEx.PartitionStyle = PARTITION_STYLE_MBR; | ||||||
| 		DriveLayoutEx.PartitionCount = 4;	// Must be multiple of 4 for MBR
 | 		DriveLayoutEx.PartitionCount = 4;	// Must be multiple of 4 for MBR
 | ||||||
| 		DriveLayoutEx.Type.Mbr.Signature = CreateDisk.Mbr.Signature; | 		DriveLayoutEx.Type.Mbr.Signature = CreateDisk.Mbr.Signature; | ||||||
| 		DriveLayoutEx.PartitionEntry[0].PartitionStyle = PARTITION_STYLE_MBR; |  | ||||||
| 		// TODO: CHS fixup (32 sectors/track) through a cheat mode, if requested
 | 		// TODO: CHS fixup (32 sectors/track) through a cheat mode, if requested
 | ||||||
| 		// NB: disk geometry is computed by BIOS & co. by finding a match between LBA and CHS value of first partition
 | 		// NB: disk geometry is computed by BIOS & co. by finding a match between LBA and CHS value of first partition
 | ||||||
| 		//     ms-sys's write_partition_number_of_heads() and write_partition_start_sector_number() can be used if needed
 | 		//     ms-sys's write_partition_number_of_heads() and write_partition_start_sector_number() can be used if needed
 | ||||||
|  | @ -967,123 +1070,26 @@ BOOL CreatePartition(HANDLE hDrive, int partition_style, int file_system, BOOL m | ||||||
| 		CreateDisk.Gpt.MaxPartitionCount = MAX_GPT_PARTITIONS; | 		CreateDisk.Gpt.MaxPartitionCount = MAX_GPT_PARTITIONS; | ||||||
| 
 | 
 | ||||||
| 		DriveLayoutEx.PartitionStyle = PARTITION_STYLE_GPT; | 		DriveLayoutEx.PartitionStyle = PARTITION_STYLE_GPT; | ||||||
| 		DriveLayoutEx.PartitionCount = (add_uefi_togo)?2:1; | 		DriveLayoutEx.PartitionCount = pn; | ||||||
| 		// At the very least, a GPT disk has 34 reserved sectors at the beginning and 33 at the end.
 | 		// At the very least, a GPT disk has 34 reserved sectors at the beginning and 33 at the end.
 | ||||||
| 		DriveLayoutEx.Type.Gpt.StartingUsableOffset.QuadPart = 34 * SelectedDrive.Geometry.BytesPerSector; | 		DriveLayoutEx.Type.Gpt.StartingUsableOffset.QuadPart = 34 * SelectedDrive.Geometry.BytesPerSector; | ||||||
| 		DriveLayoutEx.Type.Gpt.UsableLength.QuadPart = SelectedDrive.DiskSize - (34+33) * SelectedDrive.Geometry.BytesPerSector; | 		DriveLayoutEx.Type.Gpt.UsableLength.QuadPart = SelectedDrive.DiskSize - (34+33) * SelectedDrive.Geometry.BytesPerSector; | ||||||
| 		DriveLayoutEx.Type.Gpt.MaxPartitionCount = MAX_GPT_PARTITIONS; | 		DriveLayoutEx.Type.Gpt.MaxPartitionCount = MAX_GPT_PARTITIONS; | ||||||
| 		DriveLayoutEx.Type.Gpt.DiskId = CreateDisk.Gpt.DiskId; | 		DriveLayoutEx.Type.Gpt.DiskId = CreateDisk.Gpt.DiskId; | ||||||
| 		DriveLayoutEx.PartitionEntry[0].PartitionStyle = PARTITION_STYLE_GPT; |  | ||||||
| 
 |  | ||||||
| 		size_in_sectors -= 33;	// Need 33 sectors at the end for secondary GPT
 |  | ||||||
| 		break; | 		break; | ||||||
| 	default: |  | ||||||
| 		break; |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	DriveLayoutEx.PartitionEntry[0].PartitionLength.QuadPart = size_in_sectors * SelectedDrive.Geometry.BytesPerSector; |  | ||||||
| 	DriveLayoutEx.PartitionEntry[0].PartitionNumber = 1; |  | ||||||
| 	DriveLayoutEx.PartitionEntry[0].RewritePartition = TRUE; |  | ||||||
| 
 |  | ||||||
| 	switch (partition_style) { |  | ||||||
| 	case PARTITION_STYLE_MBR: |  | ||||||
| 		DriveLayoutEx.PartitionEntry[0].Mbr.BootIndicator = IsChecked(IDC_BOOT); |  | ||||||
| 		DriveLayoutEx.PartitionEntry[0].Mbr.HiddenSectors = SelectedDrive.Geometry.SectorsPerTrack; |  | ||||||
| 		switch (file_system) { |  | ||||||
| 		case FS_FAT16: |  | ||||||
| 			DriveLayoutEx.PartitionEntry[0].Mbr.PartitionType = 0x0e;	// FAT16 LBA
 |  | ||||||
| 			break; |  | ||||||
| 		case FS_NTFS: |  | ||||||
| 		case FS_EXFAT: |  | ||||||
| 		case FS_UDF: |  | ||||||
| 		case FS_REFS: |  | ||||||
| 			DriveLayoutEx.PartitionEntry[0].Mbr.PartitionType = 0x07; |  | ||||||
| 			break; |  | ||||||
| 		case FS_FAT32: |  | ||||||
| 			DriveLayoutEx.PartitionEntry[0].Mbr.PartitionType = 0x0c;	// FAT32 LBA
 |  | ||||||
| 			break; |  | ||||||
| 		default: |  | ||||||
| 			uprintf("Unsupported file system\n"); |  | ||||||
| 			return FALSE; |  | ||||||
| 		} |  | ||||||
| 		// Create an extra partition on request
 |  | ||||||
| 		if (IsChecked(IDC_EXTRA_PARTITION) || (add_uefi_togo)) { |  | ||||||
| 			DriveLayoutEx.PartitionEntry[1].PartitionStyle = PARTITION_STYLE_MBR; |  | ||||||
| 			// Should end on a track boundary
 |  | ||||||
| 			DriveLayoutEx.PartitionEntry[1].StartingOffset.QuadPart = DriveLayoutEx.PartitionEntry[0].StartingOffset.QuadPart + |  | ||||||
| 				DriveLayoutEx.PartitionEntry[0].PartitionLength.QuadPart; |  | ||||||
| 			if (add_uefi_togo) { |  | ||||||
| 				DriveLayoutEx.PartitionEntry[1].PartitionLength.QuadPart = uefi_togo_size; |  | ||||||
| 			} else { |  | ||||||
| 				DriveLayoutEx.PartitionEntry[1].PartitionLength.QuadPart = extra_size_in_tracks * |  | ||||||
| 					SelectedDrive.Geometry.SectorsPerTrack * SelectedDrive.Geometry.BytesPerSector; |  | ||||||
| 			} |  | ||||||
| 			DriveLayoutEx.PartitionEntry[1].PartitionNumber = 2; |  | ||||||
| 			DriveLayoutEx.PartitionEntry[1].RewritePartition = TRUE; |  | ||||||
| 			DriveLayoutEx.PartitionEntry[1].Mbr.BootIndicator = FALSE; |  | ||||||
| 			DriveLayoutEx.PartitionEntry[1].Mbr.HiddenSectors = SelectedDrive.Geometry.SectorsPerTrack*SelectedDrive.Geometry.BytesPerSector; |  | ||||||
| 			DriveLayoutEx.PartitionEntry[1].Mbr.PartitionType = (add_uefi_togo)?0x01:RUFUS_EXTRA_PARTITION_TYPE; |  | ||||||
| 		} |  | ||||||
| 		// For the remaining partitions, PartitionStyle & PartitionType have already
 |  | ||||||
| 		// been zeroed => already set to MBR/unused
 |  | ||||||
| 		break; |  | ||||||
| 	case PARTITION_STYLE_GPT: |  | ||||||
| 		DriveLayoutEx.PartitionEntry[0].Gpt.PartitionType = PARTITION_BASIC_DATA_GUID; |  | ||||||
| 		IGNORE_RETVAL(CoCreateGuid(&DriveLayoutEx.PartitionEntry[0].Gpt.PartitionId)); |  | ||||||
| 		wcscpy(DriveLayoutEx.PartitionEntry[0].Gpt.Name, L"Microsoft Basic Data"); |  | ||||||
| 		if (add_uefi_togo) { |  | ||||||
| 			DriveLayoutEx.PartitionEntry[1].Gpt.PartitionType = PARTITION_BASIC_DATA_GUID; |  | ||||||
| 			IGNORE_RETVAL(CoCreateGuid(&DriveLayoutEx.PartitionEntry[1].Gpt.PartitionId)); |  | ||||||
| 			wcscpy(DriveLayoutEx.PartitionEntry[1].Gpt.Name, L"UEFI:TOGO"); |  | ||||||
| 			DriveLayoutEx.PartitionEntry[1].PartitionNumber = 2; |  | ||||||
| 			DriveLayoutEx.PartitionEntry[1].RewritePartition = TRUE; |  | ||||||
| 			DriveLayoutEx.PartitionEntry[1].StartingOffset.QuadPart = DriveLayoutEx.PartitionEntry[0].StartingOffset.QuadPart + |  | ||||||
| 				DriveLayoutEx.PartitionEntry[0].PartitionLength.QuadPart; |  | ||||||
| 			DriveLayoutEx.PartitionEntry[1].PartitionLength.QuadPart = uefi_togo_size; |  | ||||||
| 		} |  | ||||||
| 		break; |  | ||||||
| 	default: |  | ||||||
| 		break; |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	// We need to write the extra partition before we refresh the disk
 |  | ||||||
| 	if (add_uefi_togo) { |  | ||||||
| 		uprintf("Writing UEFI:TOGO partition..."); |  | ||||||
| 		if (!SetFilePointerEx(hDrive, DriveLayoutEx.PartitionEntry[1].StartingOffset, NULL, FILE_BEGIN)) { |  | ||||||
| 			uprintf("Unable to set position"); |  | ||||||
| 			safe_closehandle(hDrive); |  | ||||||
| 			return FALSE; |  | ||||||
| 		} |  | ||||||
| 		buffer = GetResource(hMainInstance, MAKEINTRESOURCEA(IDR_UEFI_TOGO), _RT_RCDATA, "uefi-togo.img", &bufsize, FALSE); |  | ||||||
| 		if (buffer == NULL) { |  | ||||||
| 			uprintf("Could not access uefi-togo.img"); |  | ||||||
| 			safe_closehandle(hDrive); |  | ||||||
| 			return FALSE; |  | ||||||
| 		} |  | ||||||
| 		r = WriteFile(hDrive, buffer, bufsize, &size, NULL); |  | ||||||
| 		if ((!r) || (size != bufsize)) { |  | ||||||
| 			if (!r) |  | ||||||
| 				uprintf("Write error: %s", WindowsErrorString()); |  | ||||||
| 			else |  | ||||||
| 				uprintf("Write error: Wrote %d bytes, expected %d bytes\n", size, bufsize); |  | ||||||
| 			safe_closehandle(hDrive); |  | ||||||
| 			return FALSE; |  | ||||||
| 		} |  | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	// If you don't call IOCTL_DISK_CREATE_DISK, the next call will fail
 | 	// If you don't call IOCTL_DISK_CREATE_DISK, the next call will fail
 | ||||||
| 	size = sizeof(CreateDisk); | 	size = sizeof(CreateDisk); | ||||||
| 	r = DeviceIoControl(hDrive, IOCTL_DISK_CREATE_DISK, | 	r = DeviceIoControl(hDrive, IOCTL_DISK_CREATE_DISK, (BYTE*)&CreateDisk, size, NULL, 0, &size, NULL ); | ||||||
| 			(BYTE*)&CreateDisk, size, NULL, 0, &size, NULL ); |  | ||||||
| 	if (!r) { | 	if (!r) { | ||||||
| 		uprintf("Could not reset disk: %s\n", WindowsErrorString()); | 		uprintf("Could not reset disk: %s\n", WindowsErrorString()); | ||||||
| 		safe_closehandle(hDrive); | 		safe_closehandle(hDrive); | ||||||
| 		return FALSE; | 		return FALSE; | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	size = sizeof(DriveLayoutEx) - ((partition_style == PARTITION_STYLE_GPT)?((add_uefi_togo?2:3)*sizeof(PARTITION_INFORMATION_EX)):0); | 	size = sizeof(DriveLayoutEx) - ((partition_style == PARTITION_STYLE_GPT)?((4-pn)*sizeof(PARTITION_INFORMATION_EX)):0); | ||||||
| 	r = DeviceIoControl(hDrive, IOCTL_DISK_SET_DRIVE_LAYOUT_EX, | 	r = DeviceIoControl(hDrive, IOCTL_DISK_SET_DRIVE_LAYOUT_EX, (BYTE*)&DriveLayoutEx, size, NULL, 0, &size, NULL ); | ||||||
| 			(BYTE*)&DriveLayoutEx, size, NULL, 0, &size, NULL ); |  | ||||||
| 	if (!r) { | 	if (!r) { | ||||||
| 		uprintf("Could not set drive layout: %s\n", WindowsErrorString()); | 		uprintf("Could not set drive layout: %s\n", WindowsErrorString()); | ||||||
| 		safe_closehandle(hDrive); | 		safe_closehandle(hDrive); | ||||||
|  |  | ||||||
|  | @ -1778,12 +1778,12 @@ DWORD WINAPI FormatThread(void* param) | ||||||
| 					} | 					} | ||||||
| 					if (!SetupWinToGo(drive_name)) { | 					if (!SetupWinToGo(drive_name)) { | ||||||
| 						if (!IS_ERROR(FormatStatus)) | 						if (!IS_ERROR(FormatStatus)) | ||||||
| 							FormatStatus = ERROR_SEVERITY_ERROR|FAC(FACILITY_STORAGE)|ERROR_ISO_EXTRACT; | 							FormatStatus = ERROR_SEVERITY_ERROR|FAC(FACILITY_STORAGE)|APPERR(ERROR_ISO_EXTRACT); | ||||||
| 						goto out; | 						goto out; | ||||||
| 					} | 					} | ||||||
| 				} else if (!ExtractISO(image_path, drive_name, FALSE)) { | 				} else if (!ExtractISO(image_path, drive_name, FALSE)) { | ||||||
| 					if (!IS_ERROR(FormatStatus)) | 					if (!IS_ERROR(FormatStatus)) | ||||||
| 						FormatStatus = ERROR_SEVERITY_ERROR|FAC(FACILITY_STORAGE)|ERROR_ISO_EXTRACT; | 						FormatStatus = ERROR_SEVERITY_ERROR|FAC(FACILITY_STORAGE)|APPERR(ERROR_ISO_EXTRACT); | ||||||
| 					goto out; | 					goto out; | ||||||
| 				} | 				} | ||||||
| 				if (iso_report.has_kolibrios) { | 				if (iso_report.has_kolibrios) { | ||||||
|  |  | ||||||
							
								
								
									
										16
									
								
								src/rufus.rc
									
										
									
									
									
								
							
							
						
						
									
										16
									
								
								src/rufus.rc
									
										
									
									
									
								
							|  | @ -32,7 +32,7 @@ LANGUAGE LANG_NEUTRAL, SUBLANG_NEUTRAL | ||||||
| 
 | 
 | ||||||
| IDD_DIALOG DIALOGEX 12, 12, 242, 376 | IDD_DIALOG DIALOGEX 12, 12, 242, 376 | ||||||
| STYLE DS_SETFONT | DS_MODALFRAME | DS_CENTER | WS_MINIMIZEBOX | WS_POPUP | WS_CAPTION | WS_SYSMENU | STYLE DS_SETFONT | DS_MODALFRAME | DS_CENTER | WS_MINIMIZEBOX | WS_POPUP | WS_CAPTION | WS_SYSMENU | ||||||
| CAPTION "Rufus 2.0.0.577" | CAPTION "Rufus 2.0.0.578" | ||||||
| FONT 8, "Segoe UI", 400, 0, 0x1 | FONT 8, "Segoe UI", 400, 0, 0x1 | ||||||
| BEGIN | BEGIN | ||||||
|     DEFPUSHBUTTON   "Start",IDC_START,127,339,50,14 |     DEFPUSHBUTTON   "Start",IDC_START,127,339,50,14 | ||||||
|  | @ -157,7 +157,7 @@ END | ||||||
| 
 | 
 | ||||||
| IDD_DIALOG_XP DIALOGEX 12, 12, 242, 376 | IDD_DIALOG_XP DIALOGEX 12, 12, 242, 376 | ||||||
| STYLE DS_SETFONT | DS_MODALFRAME | DS_CENTER | WS_MINIMIZEBOX | WS_POPUP | WS_CAPTION | WS_SYSMENU | STYLE DS_SETFONT | DS_MODALFRAME | DS_CENTER | WS_MINIMIZEBOX | WS_POPUP | WS_CAPTION | WS_SYSMENU | ||||||
| CAPTION "Rufus 2.0.0.577" | CAPTION "Rufus 2.0.0.578" | ||||||
| FONT 8, "MS Shell Dlg", 400, 0, 0x1 | FONT 8, "MS Shell Dlg", 400, 0, 0x1 | ||||||
| BEGIN | BEGIN | ||||||
|     DEFPUSHBUTTON   "Start",IDC_START,127,339,50,14 |     DEFPUSHBUTTON   "Start",IDC_START,127,339,50,14 | ||||||
|  | @ -283,7 +283,7 @@ END | ||||||
| IDD_DIALOG_RTL DIALOGEX 12, 12, 242, 376 | IDD_DIALOG_RTL DIALOGEX 12, 12, 242, 376 | ||||||
| STYLE DS_SETFONT | DS_MODALFRAME | DS_CENTER | WS_MINIMIZEBOX | WS_POPUP | WS_CAPTION | WS_SYSMENU | STYLE DS_SETFONT | DS_MODALFRAME | DS_CENTER | WS_MINIMIZEBOX | WS_POPUP | WS_CAPTION | WS_SYSMENU | ||||||
| EXSTYLE WS_EX_RTLREADING | WS_EX_APPWINDOW | WS_EX_LAYOUTRTL | EXSTYLE WS_EX_RTLREADING | WS_EX_APPWINDOW | WS_EX_LAYOUTRTL | ||||||
| CAPTION "Rufus 2.0.0.577" | CAPTION "Rufus 2.0.0.578" | ||||||
| FONT 8, "Segoe UI", 400, 0, 0x1 | FONT 8, "Segoe UI", 400, 0, 0x1 | ||||||
| BEGIN | BEGIN | ||||||
|     DEFPUSHBUTTON   "Start",IDC_START,127,339,50,14 |     DEFPUSHBUTTON   "Start",IDC_START,127,339,50,14 | ||||||
|  | @ -415,7 +415,7 @@ END | ||||||
| IDD_DIALOG_RTL_XP DIALOGEX 12, 12, 242, 376 | IDD_DIALOG_RTL_XP DIALOGEX 12, 12, 242, 376 | ||||||
| STYLE DS_SETFONT | DS_MODALFRAME | DS_CENTER | WS_MINIMIZEBOX | WS_POPUP | WS_CAPTION | WS_SYSMENU | STYLE DS_SETFONT | DS_MODALFRAME | DS_CENTER | WS_MINIMIZEBOX | WS_POPUP | WS_CAPTION | WS_SYSMENU | ||||||
| EXSTYLE WS_EX_RTLREADING | WS_EX_APPWINDOW | WS_EX_LAYOUTRTL | EXSTYLE WS_EX_RTLREADING | WS_EX_APPWINDOW | WS_EX_LAYOUTRTL | ||||||
| CAPTION "Rufus 2.0.0.577" | CAPTION "Rufus 2.0.0.578" | ||||||
| FONT 8, "MS Shell Dlg", 400, 0, 0x1 | FONT 8, "MS Shell Dlg", 400, 0, 0x1 | ||||||
| BEGIN | BEGIN | ||||||
|     DEFPUSHBUTTON   "Start",IDC_START,127,339,50,14 |     DEFPUSHBUTTON   "Start",IDC_START,127,339,50,14 | ||||||
|  | @ -671,8 +671,8 @@ END | ||||||
| // | // | ||||||
| 
 | 
 | ||||||
| VS_VERSION_INFO VERSIONINFO | VS_VERSION_INFO VERSIONINFO | ||||||
|  FILEVERSION 2,0,0,577 |  FILEVERSION 2,0,0,578 | ||||||
|  PRODUCTVERSION 2,0,0,577 |  PRODUCTVERSION 2,0,0,578 | ||||||
|  FILEFLAGSMASK 0x3fL |  FILEFLAGSMASK 0x3fL | ||||||
| #ifdef _DEBUG | #ifdef _DEBUG | ||||||
|  FILEFLAGS 0x1L |  FILEFLAGS 0x1L | ||||||
|  | @ -689,13 +689,13 @@ BEGIN | ||||||
|         BEGIN |         BEGIN | ||||||
|             VALUE "CompanyName", "Akeo Consulting (http://akeo.ie)" |             VALUE "CompanyName", "Akeo Consulting (http://akeo.ie)" | ||||||
|             VALUE "FileDescription", "Rufus" |             VALUE "FileDescription", "Rufus" | ||||||
|             VALUE "FileVersion", "2.0.0.577" |             VALUE "FileVersion", "2.0.0.578" | ||||||
|             VALUE "InternalName", "Rufus" |             VALUE "InternalName", "Rufus" | ||||||
|             VALUE "LegalCopyright", "© 2011-2015 Pete Batard (GPL v3)" |             VALUE "LegalCopyright", "© 2011-2015 Pete Batard (GPL v3)" | ||||||
|             VALUE "LegalTrademarks", "http://www.gnu.org/copyleft/gpl.html" |             VALUE "LegalTrademarks", "http://www.gnu.org/copyleft/gpl.html" | ||||||
|             VALUE "OriginalFilename", "rufus.exe" |             VALUE "OriginalFilename", "rufus.exe" | ||||||
|             VALUE "ProductName", "Rufus" |             VALUE "ProductName", "Rufus" | ||||||
|             VALUE "ProductVersion", "2.0.0.577" |             VALUE "ProductVersion", "2.0.0.578" | ||||||
|         END |         END | ||||||
|     END |     END | ||||||
|     BLOCK "VarFileInfo" |     BLOCK "VarFileInfo" | ||||||
|  |  | ||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue