mirror of
				https://github.com/pbatard/rufus.git
				synced 2024-08-14 23:57:05 +00:00 
			
		
		
		
	[iso] add dumping of ISOs from optical media (Alt-O)
This commit is contained in:
		
							parent
							
								
									abead192d3
								
							
						
					
					
						commit
						5113be0a39
					
				
					 6 changed files with 227 additions and 42 deletions
				
			
		
							
								
								
									
										98
									
								
								src/dev.c
									
										
									
									
									
								
							
							
						
						
									
										98
									
								
								src/dev.c
									
										
									
									
									
								
							|  | @ -146,6 +146,104 @@ static __inline BOOL IsRemovable(const char* buffer) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  | BOOL GetOpticalMedia(IMG_SAVE* img_save) | ||||||
|  | { | ||||||
|  | 	static char str[MAX_PATH]; | ||||||
|  | 	static char label[33]; | ||||||
|  | 	int k; | ||||||
|  | 	BYTE geometry[256], *buffer = NULL; | ||||||
|  | 	PDISK_GEOMETRY_EX DiskGeometry = (PDISK_GEOMETRY_EX)(void*)geometry; | ||||||
|  | 	DWORD i, j, size, datatype; | ||||||
|  | 	HDEVINFO dev_info = NULL; | ||||||
|  | 	SP_DEVINFO_DATA dev_info_data; | ||||||
|  | 	SP_DEVICE_INTERFACE_DATA devint_data; | ||||||
|  | 	PSP_DEVICE_INTERFACE_DETAIL_DATA_A devint_detail_data; | ||||||
|  | 	HANDLE hDrive = INVALID_HANDLE_VALUE; | ||||||
|  | 	LARGE_INTEGER li; | ||||||
|  | 
 | ||||||
|  | 	dev_info = SetupDiGetClassDevsA(&_GUID_DEVINTERFACE_CDROM, NULL, NULL, DIGCF_PRESENT | DIGCF_DEVICEINTERFACE); | ||||||
|  | 	if (dev_info == INVALID_HANDLE_VALUE) { | ||||||
|  | 		uprintf("SetupDiGetClassDevs (Interface) failed: %s\n", WindowsErrorString()); | ||||||
|  | 		return FALSE; | ||||||
|  | 	} | ||||||
|  | 	dev_info_data.cbSize = sizeof(dev_info_data); | ||||||
|  | 	for (i = 0; SetupDiEnumDeviceInfo(dev_info, i, &dev_info_data); i++) { | ||||||
|  | 		memset(str, 0, sizeof(str)); | ||||||
|  | 		if (!SetupDiGetDeviceRegistryPropertyU(dev_info, &dev_info_data, SPDRP_FRIENDLYNAME, | ||||||
|  | 			&datatype, (LPBYTE)str, sizeof(str), &size)) { | ||||||
|  | 			uprintf("SetupDiGetDeviceRegistryProperty (Friendly Name) failed: %s\n", WindowsErrorString()); | ||||||
|  | 			safe_strcpy(str, sizeof(str), "Generic Optical Drive"); | ||||||
|  | 		} | ||||||
|  | 		uprintf("Found '%s' optical device", str); | ||||||
|  | 		devint_data.cbSize = sizeof(devint_data); | ||||||
|  | 		devint_detail_data = NULL; | ||||||
|  | 		for (j = 0; ; j++) { | ||||||
|  | 			safe_closehandle(hDrive); | ||||||
|  | 			safe_free(devint_detail_data); | ||||||
|  | 			safe_free(buffer); | ||||||
|  | 
 | ||||||
|  | 			if (!SetupDiEnumDeviceInterfaces(dev_info, &dev_info_data, &_GUID_DEVINTERFACE_CDROM, j, &devint_data)) { | ||||||
|  | 				if (GetLastError() != ERROR_NO_MORE_ITEMS) { | ||||||
|  | 					uprintf("SetupDiEnumDeviceInterfaces failed: %s\n", WindowsErrorString()); | ||||||
|  | 				} | ||||||
|  | 				break; | ||||||
|  | 			} | ||||||
|  | 
 | ||||||
|  | 			if (!SetupDiGetDeviceInterfaceDetailA(dev_info, &devint_data, NULL, 0, &size, NULL)) { | ||||||
|  | 				if (GetLastError() == ERROR_INSUFFICIENT_BUFFER) { | ||||||
|  | 					devint_detail_data = (PSP_DEVICE_INTERFACE_DETAIL_DATA_A)calloc(1, size); | ||||||
|  | 					if (devint_detail_data == NULL) { | ||||||
|  | 						uprintf("Unable to allocate data for SP_DEVICE_INTERFACE_DETAIL_DATA\n"); | ||||||
|  | 						continue; | ||||||
|  | 					} | ||||||
|  | 					devint_detail_data->cbSize = sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA_A); | ||||||
|  | 				} else { | ||||||
|  | 					uprintf("SetupDiGetDeviceInterfaceDetail (dummy) failed: %s\n", WindowsErrorString()); | ||||||
|  | 					continue; | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			if (devint_detail_data == NULL) { | ||||||
|  | 				uprintf("SetupDiGetDeviceInterfaceDetail (dummy) - no data was allocated\n"); | ||||||
|  | 				continue; | ||||||
|  | 			} | ||||||
|  | 			if (!SetupDiGetDeviceInterfaceDetailA(dev_info, &devint_data, devint_detail_data, size, &size, NULL)) { | ||||||
|  | 				uprintf("SetupDiGetDeviceInterfaceDetail (actual) failed: %s\n", WindowsErrorString()); | ||||||
|  | 				continue; | ||||||
|  | 			} | ||||||
|  | 
 | ||||||
|  | 			// Get the size of the inserted media (if any)
 | ||||||
|  | 			hDrive = CreateFileA(devint_detail_data->DevicePath, GENERIC_READ, | ||||||
|  | 				FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN, NULL); | ||||||
|  | 			if (hDrive == INVALID_HANDLE_VALUE) | ||||||
|  | 				continue; | ||||||
|  | 			if (!DeviceIoControl(hDrive, IOCTL_DISK_GET_DRIVE_GEOMETRY_EX, | ||||||
|  | 				NULL, 0, geometry, sizeof(geometry), &size, NULL)) | ||||||
|  | 				continue; | ||||||
|  | 			// Rewritable media usually has a one sector
 | ||||||
|  | 			if (DiskGeometry->DiskSize.QuadPart <= 4096) | ||||||
|  | 				continue; | ||||||
|  | 			// Read the label directly, since it's a massive PITA to get it from Windows
 | ||||||
|  | 			li.QuadPart = 0x8000LL; | ||||||
|  | 			buffer = malloc(2048); | ||||||
|  | 			if ((buffer != NULL) && (SetFilePointerEx(hDrive, li, NULL, FILE_BEGIN)) && | ||||||
|  | 				ReadFile(hDrive, buffer, 2048, &size, NULL) && (size == 2048)) { | ||||||
|  | 				safe_strcpy(label, sizeof(label), (char*)&buffer[0x28]); | ||||||
|  | 				for (k = safe_strlen(label) - 1; (k >= 0) && (label[k] == 0x20); k--) | ||||||
|  | 					label[k] = 0; | ||||||
|  | 				img_save->Label = label; | ||||||
|  | 			} | ||||||
|  | 			safe_strcpy(str, sizeof(str), devint_detail_data->DevicePath); | ||||||
|  | 			img_save->DevicePath = str; | ||||||
|  | 			img_save->DeviceSize = DiskGeometry->DiskSize.QuadPart; | ||||||
|  | 			safe_closehandle(hDrive); | ||||||
|  | 			safe_free(devint_detail_data); | ||||||
|  | 			safe_free(buffer); | ||||||
|  | 			return TRUE; | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return FALSE; | ||||||
|  | } | ||||||
|  | 
 | ||||||
| /* For debugging user reports of HDDs vs UFDs */ | /* For debugging user reports of HDDs vs UFDs */ | ||||||
| //#define FORCED_DEVICE
 | //#define FORCED_DEVICE
 | ||||||
| #ifdef FORCED_DEVICE | #ifdef FORCED_DEVICE | ||||||
|  |  | ||||||
|  | @ -176,6 +176,8 @@ typedef struct _USB_NODE_CONNECTION_INFORMATION_EX_V2 { | ||||||
| 
 | 
 | ||||||
| const GUID _GUID_DEVINTERFACE_DISK = | const GUID _GUID_DEVINTERFACE_DISK = | ||||||
| 	{ 0x53f56307L, 0xb6bf, 0x11d0, {0x94, 0xf2, 0x00, 0xa0, 0xc9, 0x1e, 0xfb, 0x8b} }; | 	{ 0x53f56307L, 0xb6bf, 0x11d0, {0x94, 0xf2, 0x00, 0xa0, 0xc9, 0x1e, 0xfb, 0x8b} }; | ||||||
|  | const GUID _GUID_DEVINTERFACE_CDROM = | ||||||
|  | 	{ 0x53f56308L, 0xb6bf, 0x11d0, {0x94, 0xf2, 0x00, 0xa0, 0xc9, 0x1e, 0xfb, 0x8b} }; | ||||||
| const GUID _GUID_DEVINTERFACE_USB_HUB = | const GUID _GUID_DEVINTERFACE_USB_HUB = | ||||||
| 	{ 0xf18a0e88L, 0xc30c, 0x11d0, {0x88, 0x15, 0x00, 0xa0, 0xc9, 0x06, 0xbe, 0xd8} }; | 	{ 0xf18a0e88L, 0xc30c, 0x11d0, {0x88, 0x15, 0x00, 0xa0, 0xc9, 0x06, 0xbe, 0xd8} }; | ||||||
| 
 | 
 | ||||||
|  |  | ||||||
							
								
								
									
										50
									
								
								src/format.c
									
										
									
									
									
								
							
							
						
						
									
										50
									
								
								src/format.c
									
										
									
									
									
								
							|  | @ -1992,7 +1992,7 @@ DWORD WINAPI SaveImageThread(void* param) | ||||||
| { | { | ||||||
| 	BOOL s; | 	BOOL s; | ||||||
| 	DWORD rSize, wSize; | 	DWORD rSize, wSize; | ||||||
| 	VHD_SAVE *vhd_save = (VHD_SAVE*)param; | 	IMG_SAVE *img_save = (IMG_SAVE*)param; | ||||||
| 	HANDLE hPhysicalDrive = INVALID_HANDLE_VALUE; | 	HANDLE hPhysicalDrive = INVALID_HANDLE_VALUE; | ||||||
| 	HANDLE hDestImage = INVALID_HANDLE_VALUE; | 	HANDLE hDestImage = INVALID_HANDLE_VALUE; | ||||||
| 	LARGE_INTEGER li; | 	LARGE_INTEGER li; | ||||||
|  | @ -2002,39 +2002,57 @@ DWORD WINAPI SaveImageThread(void* param) | ||||||
| 
 | 
 | ||||||
| 	PrintInfoDebug(0, MSG_225); | 	PrintInfoDebug(0, MSG_225); | ||||||
| 	LastRefresh = 0; | 	LastRefresh = 0; | ||||||
| 	hPhysicalDrive = GetPhysicalHandle(vhd_save->DeviceNum, FALSE, TRUE); | 	switch (img_save->Type) { | ||||||
|  | 	case IMG_SAVE_TYPE_VHD: | ||||||
|  | 		hPhysicalDrive = GetPhysicalHandle(img_save->DeviceNum, FALSE, TRUE); | ||||||
|  | 		break; | ||||||
|  | 	case IMG_SAVE_TYPE_ISO: | ||||||
|  | 		hPhysicalDrive = CreateFileA(img_save->DevicePath, GENERIC_READ, FILE_SHARE_READ, | ||||||
|  | 			NULL, OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN, NULL); | ||||||
|  | 		break; | ||||||
|  | 	default: | ||||||
|  | 		uprintf("invalid image type"); | ||||||
|  | 	} | ||||||
| 	if (hPhysicalDrive == INVALID_HANDLE_VALUE) { | 	if (hPhysicalDrive == INVALID_HANDLE_VALUE) { | ||||||
| 		FormatStatus = ERROR_SEVERITY_ERROR|FAC(FACILITY_STORAGE)|ERROR_OPEN_FAILED; | 		FormatStatus = ERROR_SEVERITY_ERROR|FAC(FACILITY_STORAGE)|ERROR_OPEN_FAILED; | ||||||
| 		goto out; | 		goto out; | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	// Write an image file
 | 	// Write an image file
 | ||||||
| 	// We poked the MBR and other stuff, so we need to rewind
 | 	// We may have poked the MBR and other stuff, so need to rewind
 | ||||||
| 	li.QuadPart = 0; | 	li.QuadPart = 0; | ||||||
| 	if (!SetFilePointerEx(hPhysicalDrive, li, NULL, FILE_BEGIN)) | 	if (!SetFilePointerEx(hPhysicalDrive, li, NULL, FILE_BEGIN)) | ||||||
| 		uprintf("Warning: Unable to rewind device position - wrong data might be copied!"); | 		uprintf("Warning: Unable to rewind device position - wrong data might be copied!"); | ||||||
| 	hDestImage = CreateFileU(vhd_save->path, GENERIC_WRITE, FILE_SHARE_WRITE, NULL, | 	hDestImage = CreateFileU(img_save->ImagePath, GENERIC_WRITE, FILE_SHARE_WRITE, NULL, | ||||||
| 		CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); | 		CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); | ||||||
| 	if (hDestImage == INVALID_HANDLE_VALUE) { | 	if (hDestImage == INVALID_HANDLE_VALUE) { | ||||||
| 		uprintf("Could not open image '%s': %s", vhd_save->path, WindowsErrorString()); | 		uprintf("Could not open image '%s': %s", img_save->ImagePath, WindowsErrorString()); | ||||||
| 		FormatStatus = ERROR_SEVERITY_ERROR|FAC(FACILITY_STORAGE)|ERROR_OPEN_FAILED; | 		FormatStatus = ERROR_SEVERITY_ERROR|FAC(FACILITY_STORAGE)|ERROR_OPEN_FAILED; | ||||||
| 		goto out; | 		goto out; | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 	uprintf("Saving to image '%s'...", vhd_save->path); | 	buffer = (uint8_t*)_mm_malloc(img_save->BufSize, 16); | ||||||
| 	buffer = (uint8_t*)_mm_malloc(DD_BUFFER_SIZE, 16); |  | ||||||
| 	if (buffer == NULL) { | 	if (buffer == NULL) { | ||||||
| 		FormatStatus = ERROR_SEVERITY_ERROR|FAC(FACILITY_STORAGE)|ERROR_NOT_ENOUGH_MEMORY; | 		FormatStatus = ERROR_SEVERITY_ERROR|FAC(FACILITY_STORAGE)|ERROR_NOT_ENOUGH_MEMORY; | ||||||
| 		uprintf("could not allocate buffer"); | 		uprintf("could not allocate buffer"); | ||||||
| 		goto out; | 		goto out; | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
|  | 	uprintf("Will use a buffer size of %s", SizeToHumanReadable(img_save->BufSize, FALSE, FALSE)); | ||||||
|  | 	uprintf("Saving to image '%s'...", img_save->ImagePath); | ||||||
|  | 
 | ||||||
| 	// Don't bother trying for something clever, using double buffering overlapped and whatnot:
 | 	// Don't bother trying for something clever, using double buffering overlapped and whatnot:
 | ||||||
| 	// With Windows' default optimizations, sync read + sync write for sequential operations
 | 	// With Windows' default optimizations, sync read + sync write for sequential operations
 | ||||||
| 	// will be as fast, if not faster, than whatever async scheme you can come up with.
 | 	// will be as fast, if not faster, than whatever async scheme you can come up with.
 | ||||||
| 	for (wb = 0; ; wb += wSize) { | 	for (wb = 0; ; wb += wSize) { | ||||||
|  | 		if (img_save->Type == IMG_SAVE_TYPE_ISO) { | ||||||
|  | 			// Optical drives do not appear to increment the sectors to read automatically
 | ||||||
|  | 			li.QuadPart = wb; | ||||||
|  | 			if (!SetFilePointerEx(hPhysicalDrive, li, NULL, FILE_BEGIN)) | ||||||
|  | 				uprintf("Warning: Unable to set device position - wrong data might be copied!"); | ||||||
|  | 		} | ||||||
| 		s = ReadFile(hPhysicalDrive, buffer, | 		s = ReadFile(hPhysicalDrive, buffer, | ||||||
| 			(DWORD)MIN(DD_BUFFER_SIZE, SelectedDrive.DiskSize - wb), &rSize, NULL); | 			(DWORD)MIN(img_save->BufSize, img_save->DeviceSize - wb), &rSize, NULL); | ||||||
| 		if (!s) { | 		if (!s) { | ||||||
| 			FormatStatus = ERROR_SEVERITY_ERROR|FAC(FACILITY_STORAGE)|ERROR_READ_FAULT; | 			FormatStatus = ERROR_SEVERITY_ERROR|FAC(FACILITY_STORAGE)|ERROR_READ_FAULT; | ||||||
| 			uprintf("read error: %s", WindowsErrorString()); | 			uprintf("read error: %s", WindowsErrorString()); | ||||||
|  | @ -2044,7 +2062,7 @@ DWORD WINAPI SaveImageThread(void* param) | ||||||
| 			break; | 			break; | ||||||
| 		if (_GetTickCount64() > LastRefresh + MAX_REFRESH) { | 		if (_GetTickCount64() > LastRefresh + MAX_REFRESH) { | ||||||
| 			LastRefresh = _GetTickCount64(); | 			LastRefresh = _GetTickCount64(); | ||||||
| 			format_percent = (100.0f*wb)/(1.0f*SelectedDrive.DiskSize); | 			format_percent = (100.0f*wb)/(1.0f*img_save->DeviceSize); | ||||||
| 			PrintInfo(0, MSG_261, format_percent); | 			PrintInfo(0, MSG_261, format_percent); | ||||||
| 			UpdateProgress(OP_FORMAT, format_percent); | 			UpdateProgress(OP_FORMAT, format_percent); | ||||||
| 		} | 		} | ||||||
|  | @ -2068,21 +2086,23 @@ DWORD WINAPI SaveImageThread(void* param) | ||||||
| 		} | 		} | ||||||
| 		if (i >= WRITE_RETRIES) goto out; | 		if (i >= WRITE_RETRIES) goto out; | ||||||
| 	} | 	} | ||||||
| 	if (wb != SelectedDrive.DiskSize) { | 	if (wb != img_save->DeviceSize) { | ||||||
| 		uprintf("Error: wrote %" PRIu64 " bytes, expected %" PRIu64, wb, SelectedDrive.DiskSize); | 		uprintf("Error: wrote %s, expected %s", SizeToHumanReadable(wb, FALSE, FALSE), | ||||||
|  | 			SizeToHumanReadable(img_save->DeviceSize, FALSE, FALSE)); | ||||||
| 		FormatStatus = ERROR_SEVERITY_ERROR|FAC(FACILITY_STORAGE)|ERROR_WRITE_FAULT; | 		FormatStatus = ERROR_SEVERITY_ERROR|FAC(FACILITY_STORAGE)|ERROR_WRITE_FAULT; | ||||||
| 		goto out; | 		goto out; | ||||||
| 	} | 	} | ||||||
| 	uprintf("%" PRIu64 " bytes written", wb); | 	if (img_save->Type == IMG_SAVE_TYPE_VHD) { | ||||||
| 		uprintf("Appending VHD footer..."); | 		uprintf("Appending VHD footer..."); | ||||||
| 	if (!AppendVHDFooter(vhd_save->path)) { | 		if (!AppendVHDFooter(img_save->ImagePath)) { | ||||||
| 			FormatStatus = ERROR_SEVERITY_ERROR|FAC(FACILITY_STORAGE)|ERROR_WRITE_FAULT; | 			FormatStatus = ERROR_SEVERITY_ERROR|FAC(FACILITY_STORAGE)|ERROR_WRITE_FAULT; | ||||||
| 			goto out; | 			goto out; | ||||||
| 		} | 		} | ||||||
| 	uprintf("Done"); | 	} | ||||||
|  | 	uprintf("Operation complete (Wrote %s).", SizeToHumanReadable(wb, FALSE, FALSE)); | ||||||
| 
 | 
 | ||||||
| out: | out: | ||||||
| 	safe_free(vhd_save->path); | 	safe_free(img_save->ImagePath); | ||||||
| 	safe_mm_free(buffer); | 	safe_mm_free(buffer); | ||||||
| 	safe_closehandle(hDestImage); | 	safe_closehandle(hDestImage); | ||||||
| 	safe_unlockclose(hPhysicalDrive); | 	safe_unlockclose(hPhysicalDrive); | ||||||
|  |  | ||||||
							
								
								
									
										90
									
								
								src/rufus.c
									
										
									
									
									
								
							
							
						
						
									
										90
									
								
								src/rufus.c
									
										
									
									
									
								
							|  | @ -960,7 +960,7 @@ static void DisplayISOProps(void) | ||||||
| 	int i; | 	int i; | ||||||
| 
 | 
 | ||||||
| 	uprintf("ISO label: '%s'", img_report.label); | 	uprintf("ISO label: '%s'", img_report.label); | ||||||
| 	uprintf("  Size: %" PRIu64 " bytes", img_report.projected_size); | 	uprintf("  Size: %s", SizeToHumanReadable( img_report.projected_size, FALSE, FALSE)); | ||||||
| 	PRINT_ISO_PROP(img_report.has_4GB_file, "  Has a >4GB file"); | 	PRINT_ISO_PROP(img_report.has_4GB_file, "  Has a >4GB file"); | ||||||
| 	PRINT_ISO_PROP(img_report.has_long_filename, "  Has a >64 chars filename"); | 	PRINT_ISO_PROP(img_report.has_long_filename, "  Has a >64 chars filename"); | ||||||
| 	PRINT_ISO_PROP(HAS_SYSLINUX(img_report), "  Uses: Syslinux/Isolinux v%s", img_report.sl_version_str); | 	PRINT_ISO_PROP(HAS_SYSLINUX(img_report), "  Uses: Syslinux/Isolinux v%s", img_report.sl_version_str); | ||||||
|  | @ -1994,19 +1994,22 @@ void SetBoot(int fs, int tt) | ||||||
| 
 | 
 | ||||||
| void SaveVHD(void) | void SaveVHD(void) | ||||||
| { | { | ||||||
| 	static VHD_SAVE vhd_save; | 	static IMG_SAVE img_save = { 0 }; | ||||||
| 	char filename[128]; | 	char filename[128]; | ||||||
| 	char path[MAX_PATH]; | 	char path[MAX_PATH]; | ||||||
| 	int DriveIndex = ComboBox_GetCurSel(hDeviceList); | 	int DriveIndex = ComboBox_GetCurSel(hDeviceList); | ||||||
| 	EXT_DECL(vhd_ext, filename, __VA_GROUP__("*.vhd"), __VA_GROUP__("VHD File")); | 	EXT_DECL(img_ext, filename, __VA_GROUP__("*.vhd"), __VA_GROUP__(lmprintf(MSG_095))); | ||||||
| 	ULARGE_INTEGER free_space; | 	ULARGE_INTEGER free_space; | ||||||
| 
 | 
 | ||||||
| 	if (DriveIndex >= 0) | 	if (DriveIndex >= 0) | ||||||
| 		safe_sprintf(filename, sizeof(filename), "%s.vhd", DriveLabel.String[DriveIndex]); | 		safe_sprintf(filename, sizeof(filename), "%s.vhd", DriveLabel.String[DriveIndex]); | ||||||
| 	if ((DriveIndex != CB_ERR) && (!format_op_in_progress) && (format_thid == NULL)) { | 	if ((DriveIndex != CB_ERR) && (!format_op_in_progress) && (format_thid == NULL)) { | ||||||
| 		vhd_save.DeviceNum = (DWORD)ComboBox_GetItemData(hDeviceList, DriveIndex); | 		img_save.Type = IMG_SAVE_TYPE_VHD; | ||||||
| 		vhd_save.path = FileDialog(TRUE, NULL, &vhd_ext, 0); | 		img_save.DeviceNum = (DWORD)ComboBox_GetItemData(hDeviceList, DriveIndex); | ||||||
| 		if (vhd_save.path != NULL) { | 		img_save.ImagePath = FileDialog(TRUE, NULL, &img_ext, 0); | ||||||
|  | 		img_save.BufSize = DD_BUFFER_SIZE; | ||||||
|  | 		img_save.DeviceSize = SelectedDrive.DiskSize; | ||||||
|  | 		if (img_save.ImagePath != NULL) { | ||||||
| 			// Reset all progress bars
 | 			// Reset all progress bars
 | ||||||
| 			SendMessage(hProgress, PBM_SETSTATE, (WPARAM)PBST_NORMAL, 0); | 			SendMessage(hProgress, PBM_SETSTATE, (WPARAM)PBST_NORMAL, 0); | ||||||
| 			SetTaskbarProgressState(TASKBAR_NORMAL); | 			SetTaskbarProgressState(TASKBAR_NORMAL); | ||||||
|  | @ -2015,14 +2018,14 @@ void SaveVHD(void) | ||||||
| 			FormatStatus = 0; | 			FormatStatus = 0; | ||||||
| 			format_op_in_progress = TRUE; | 			format_op_in_progress = TRUE; | ||||||
| 			free_space.QuadPart = 0; | 			free_space.QuadPart = 0; | ||||||
| 			if ((GetVolumePathNameA(vhd_save.path, path, sizeof(path))) | 			if ((GetVolumePathNameA(img_save.ImagePath, path, sizeof(path))) | ||||||
| 				&& (GetDiskFreeSpaceExA(path, &free_space, NULL, NULL)) | 				&& (GetDiskFreeSpaceExA(path, &free_space, NULL, NULL)) | ||||||
| 				&& ((LONGLONG)free_space.QuadPart > (SelectedDrive.DiskSize + 512))) { | 				&& ((LONGLONG)free_space.QuadPart > (SelectedDrive.DiskSize + 512))) { | ||||||
| 				// Disable all controls except cancel
 | 				// Disable all controls except cancel
 | ||||||
| 				EnableControls(FALSE); | 				EnableControls(FALSE); | ||||||
| 				FormatStatus = 0; | 				FormatStatus = 0; | ||||||
| 				InitProgress(TRUE); | 				InitProgress(TRUE); | ||||||
| 				format_thid = CreateThread(NULL, 0, SaveImageThread, &vhd_save, 0, NULL); | 				format_thid = CreateThread(NULL, 0, SaveImageThread, &img_save, 0, NULL); | ||||||
| 				if (format_thid != NULL) { | 				if (format_thid != NULL) { | ||||||
| 					uprintf("\r\nSave to VHD operation started"); | 					uprintf("\r\nSave to VHD operation started"); | ||||||
| 					PrintInfo(0, -1); | 					PrintInfo(0, -1); | ||||||
|  | @ -2030,25 +2033,22 @@ void SaveVHD(void) | ||||||
| 					safe_sprintf(szTimer, sizeof(szTimer), "00:00:00"); | 					safe_sprintf(szTimer, sizeof(szTimer), "00:00:00"); | ||||||
| 					SendMessageA(hStatus, SB_SETTEXTA, SBT_OWNERDRAW | SB_SECTION_RIGHT, (LPARAM)szTimer); | 					SendMessageA(hStatus, SB_SETTEXTA, SBT_OWNERDRAW | SB_SECTION_RIGHT, (LPARAM)szTimer); | ||||||
| 					SetTimer(hMainDialog, TID_APP_TIMER, 1000, ClockTimer); | 					SetTimer(hMainDialog, TID_APP_TIMER, 1000, ClockTimer); | ||||||
| 				} | 				} else { | ||||||
| 				else { |  | ||||||
| 					uprintf("Unable to start VHD save thread"); | 					uprintf("Unable to start VHD save thread"); | ||||||
| 					FormatStatus = ERROR_SEVERITY_ERROR | FAC(FACILITY_STORAGE) | APPERR(ERROR_CANT_START_THREAD); | 					FormatStatus = ERROR_SEVERITY_ERROR | FAC(FACILITY_STORAGE) | APPERR(ERROR_CANT_START_THREAD); | ||||||
| 					safe_free(vhd_save.path); | 					safe_free(img_save.ImagePath); | ||||||
| 					PostMessage(hMainDialog, UM_FORMAT_COMPLETED, (WPARAM)FALSE, 0); | 					PostMessage(hMainDialog, UM_FORMAT_COMPLETED, (WPARAM)FALSE, 0); | ||||||
| 					format_op_in_progress = FALSE; | 					format_op_in_progress = FALSE; | ||||||
| 				} | 				} | ||||||
| 			} | 			} else { | ||||||
| 			else { |  | ||||||
| 				if (free_space.QuadPart == 0) { | 				if (free_space.QuadPart == 0) { | ||||||
| 					uprintf("Unable to isolate drive name for VHD save"); | 					uprintf("Unable to isolate drive name for VHD save"); | ||||||
| 					FormatStatus = ERROR_SEVERITY_ERROR | FAC(FACILITY_STORAGE) | ERROR_PATH_NOT_FOUND; | 					FormatStatus = ERROR_SEVERITY_ERROR | FAC(FACILITY_STORAGE) | ERROR_PATH_NOT_FOUND; | ||||||
| 				} | 				} else { | ||||||
| 				else { |  | ||||||
| 					uprintf("The VHD size is too large for the target drive"); | 					uprintf("The VHD size is too large for the target drive"); | ||||||
| 					FormatStatus = ERROR_SEVERITY_ERROR | FAC(FACILITY_STORAGE) | ERROR_FILE_TOO_LARGE; | 					FormatStatus = ERROR_SEVERITY_ERROR | FAC(FACILITY_STORAGE) | ERROR_FILE_TOO_LARGE; | ||||||
| 				} | 				} | ||||||
| 				safe_free(vhd_save.path); | 				safe_free(img_save.ImagePath); | ||||||
| 				PostMessage(hMainDialog, UM_FORMAT_COMPLETED, (WPARAM)FALSE, 0); | 				PostMessage(hMainDialog, UM_FORMAT_COMPLETED, (WPARAM)FALSE, 0); | ||||||
| 				format_op_in_progress = FALSE; | 				format_op_in_progress = FALSE; | ||||||
| 			} | 			} | ||||||
|  | @ -2056,6 +2056,57 @@ void SaveVHD(void) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  | void SaveISO(void) | ||||||
|  | { | ||||||
|  | 	static IMG_SAVE img_save = { 0 }; | ||||||
|  | 	char filename[33] = "disc_image.iso"; | ||||||
|  | 	EXT_DECL(img_ext, filename, __VA_GROUP__("*.iso"), __VA_GROUP__(lmprintf(MSG_036))); | ||||||
|  | 
 | ||||||
|  | 	if ((format_op_in_progress) || (format_thid != NULL)) | ||||||
|  | 		return; | ||||||
|  | 
 | ||||||
|  | 	img_save.Type = IMG_SAVE_TYPE_ISO; | ||||||
|  | 	if (!GetOpticalMedia(&img_save)) { | ||||||
|  | 		uprintf("No dumpable optical media found."); | ||||||
|  | 		return; | ||||||
|  | 	} | ||||||
|  | 	// Adjust the buffer size according to the disc size so that we get a decent speed.
 | ||||||
|  | 	for (img_save.BufSize = 32 * MB; | ||||||
|  | 		(img_save.BufSize > 8 * MB) && (img_save.DeviceSize <= img_save.BufSize * 64); | ||||||
|  | 		img_save.BufSize /= 2); | ||||||
|  | 	if ((img_save.Label != NULL) && (img_save.Label[0] != 0)) | ||||||
|  | 		safe_sprintf(filename, sizeof(filename), "%s.iso", img_save.Label); | ||||||
|  | 	uprintf("ISO media size %s", SizeToHumanReadable(img_save.DeviceSize, FALSE, FALSE)); | ||||||
|  | 
 | ||||||
|  | 	img_save.ImagePath = FileDialog(TRUE, NULL, &img_ext, 0); | ||||||
|  | 	if (img_save.ImagePath == NULL) | ||||||
|  | 		return; | ||||||
|  | 	// Reset all progress bars
 | ||||||
|  | 	SendMessage(hProgress, PBM_SETSTATE, (WPARAM)PBST_NORMAL, 0); | ||||||
|  | 	SetTaskbarProgressState(TASKBAR_NORMAL); | ||||||
|  | 	SetTaskbarProgressValue(0, MAX_PROGRESS); | ||||||
|  | 	SendMessage(hProgress, PBM_SETPOS, 0, 0); | ||||||
|  | 	FormatStatus = 0; | ||||||
|  | 	format_op_in_progress = TRUE; | ||||||
|  | 	// Disable all controls except cancel
 | ||||||
|  | 	EnableControls(FALSE); | ||||||
|  | 	InitProgress(TRUE); | ||||||
|  | 	format_thid = CreateThread(NULL, 0, SaveImageThread, &img_save, 0, NULL); | ||||||
|  | 	if (format_thid != NULL) { | ||||||
|  | 		uprintf("\r\nSave to ISO operation started"); | ||||||
|  | 		PrintInfo(0, -1); | ||||||
|  | 		timer = 0; | ||||||
|  | 		safe_sprintf(szTimer, sizeof(szTimer), "00:00:00"); | ||||||
|  | 		SendMessageA(hStatus, SB_SETTEXTA, SBT_OWNERDRAW | SB_SECTION_RIGHT, (LPARAM)szTimer); | ||||||
|  | 		SetTimer(hMainDialog, TID_APP_TIMER, 1000, ClockTimer); | ||||||
|  | 	} else { | ||||||
|  | 		uprintf("Unable to start ISO save thread"); | ||||||
|  | 		FormatStatus = ERROR_SEVERITY_ERROR | FAC(FACILITY_STORAGE) | APPERR(ERROR_CANT_START_THREAD); | ||||||
|  | 		safe_free(img_save.ImagePath); | ||||||
|  | 		PostMessage(hMainDialog, UM_FORMAT_COMPLETED, (WPARAM)FALSE, 0); | ||||||
|  | 		format_op_in_progress = FALSE; | ||||||
|  | 	} | ||||||
|  | } | ||||||
| 
 | 
 | ||||||
| /*
 | /*
 | ||||||
|  * Main dialog callback |  * Main dialog callback | ||||||
|  | @ -3115,7 +3166,7 @@ relaunch: | ||||||
| 
 | 
 | ||||||
| 	// Do our own event processing and process "magic" commands
 | 	// Do our own event processing and process "magic" commands
 | ||||||
| 	while(GetMessage(&msg, NULL, 0, 0)) { | 	while(GetMessage(&msg, NULL, 0, 0)) { | ||||||
| 		// ** *****  **** *  ******** *
 | 		// ** *****  **** ** ******** *
 | ||||||
| 		// .,ABCDEFGHIJKLMNOPQRSTUVWXYZ
 | 		// .,ABCDEFGHIJKLMNOPQRSTUVWXYZ
 | ||||||
| 
 | 
 | ||||||
| 		// Ctrl-A => Select the log data
 | 		// Ctrl-A => Select the log data
 | ||||||
|  | @ -3225,6 +3276,11 @@ relaunch: | ||||||
| 			PrintStatusTimeout(lmprintf(MSG_260), enable_ntfs_compression); | 			PrintStatusTimeout(lmprintf(MSG_260), enable_ntfs_compression); | ||||||
| 			continue; | 			continue; | ||||||
| 		} | 		} | ||||||
|  | 		// Alt-O => Save from Optical drive to ISO
 | ||||||
|  | 		if ((msg.message == WM_SYSKEYDOWN) && (msg.wParam == 'O')) { | ||||||
|  | 			SaveISO(); | ||||||
|  | 			continue; | ||||||
|  | 		} | ||||||
| 		// Alt-Q => Disable file indexing (for file systems that support it)
 | 		// Alt-Q => Disable file indexing (for file systems that support it)
 | ||||||
| 		if ((msg.message == WM_SYSKEYDOWN) && (msg.wParam == 'Q')) { | 		if ((msg.message == WM_SYSKEYDOWN) && (msg.wParam == 'Q')) { | ||||||
| 			disable_file_indexing = !disable_file_indexing; | 			disable_file_indexing = !disable_file_indexing; | ||||||
|  |  | ||||||
							
								
								
									
										13
									
								
								src/rufus.h
									
										
									
									
									
								
							
							
						
						
									
										13
									
								
								src/rufus.h
									
										
									
									
									
								
							|  | @ -303,10 +303,18 @@ typedef struct { | ||||||
| 	char* release_notes; | 	char* release_notes; | ||||||
| } RUFUS_UPDATE; | } RUFUS_UPDATE; | ||||||
| 
 | 
 | ||||||
|  | #define IMG_SAVE_TYPE_VHD 1 | ||||||
|  | #define IMG_SAVE_TYPE_ISO 2 | ||||||
|  | 
 | ||||||
| typedef struct { | typedef struct { | ||||||
|  | 	DWORD Type; | ||||||
| 	DWORD DeviceNum; | 	DWORD DeviceNum; | ||||||
| 	char* path; | 	DWORD BufSize; | ||||||
| } VHD_SAVE; | 	LONGLONG DeviceSize; | ||||||
|  | 	char* DevicePath; | ||||||
|  | 	char* ImagePath; | ||||||
|  | 	char* Label; | ||||||
|  | } IMG_SAVE; | ||||||
| 
 | 
 | ||||||
| /*
 | /*
 | ||||||
|  * Structure and macros used for the extensions specification of FileDialog() |  * Structure and macros used for the extensions specification of FileDialog() | ||||||
|  | @ -432,6 +440,7 @@ extern DWORD GetResourceSize(HMODULE module, char* name, char* type, const char* | ||||||
| extern DWORD RunCommand(const char* cmdline, const char* dir, BOOL log); | extern DWORD RunCommand(const char* cmdline, const char* dir, BOOL log); | ||||||
| extern BOOL CompareGUID(const GUID *guid1, const GUID *guid2); | extern BOOL CompareGUID(const GUID *guid1, const GUID *guid2); | ||||||
| extern BOOL GetDevices(DWORD devnum); | extern BOOL GetDevices(DWORD devnum); | ||||||
|  | extern BOOL GetOpticalMedia(IMG_SAVE* img_save); | ||||||
| extern BOOL SetLGP(BOOL bRestore, BOOL* bExistingKey, const char* szPath, const char* szPolicy, DWORD dwValue); | extern BOOL SetLGP(BOOL bRestore, BOOL* bExistingKey, const char* szPath, const char* szPolicy, DWORD dwValue); | ||||||
| extern LONG GetEntryWidth(HWND hDropDown, const char* entry); | extern LONG GetEntryWidth(HWND hDropDown, const char* entry); | ||||||
| extern DWORD DownloadFile(const char* url, const char* file, HWND hProgressDialog); | extern DWORD DownloadFile(const char* url, const char* file, HWND hProgressDialog); | ||||||
|  |  | ||||||
							
								
								
									
										10
									
								
								src/rufus.rc
									
										
									
									
									
								
							
							
						
						
									
										10
									
								
								src/rufus.rc
									
										
									
									
									
								
							|  | @ -33,7 +33,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 | ||||||
| EXSTYLE WS_EX_ACCEPTFILES | EXSTYLE WS_EX_ACCEPTFILES | ||||||
| CAPTION "Rufus 2.11.1006" | CAPTION "Rufus 2.11.1007" | ||||||
| FONT 8, "Segoe UI Symbol", 400, 0, 0x0 | FONT 8, "Segoe UI Symbol", 400, 0, 0x0 | ||||||
| BEGIN | BEGIN | ||||||
|     LTEXT           "Device",IDS_DEVICE_TXT,9,6,200,8 |     LTEXT           "Device",IDS_DEVICE_TXT,9,6,200,8 | ||||||
|  | @ -320,8 +320,8 @@ END | ||||||
| // | // | ||||||
| 
 | 
 | ||||||
| VS_VERSION_INFO VERSIONINFO | VS_VERSION_INFO VERSIONINFO | ||||||
|  FILEVERSION 2,11,1006,0 |  FILEVERSION 2,11,1007,0 | ||||||
|  PRODUCTVERSION 2,11,1006,0 |  PRODUCTVERSION 2,11,1007,0 | ||||||
|  FILEFLAGSMASK 0x3fL |  FILEFLAGSMASK 0x3fL | ||||||
| #ifdef _DEBUG | #ifdef _DEBUG | ||||||
|  FILEFLAGS 0x1L |  FILEFLAGS 0x1L | ||||||
|  | @ -338,13 +338,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.11.1006" |             VALUE "FileVersion", "2.11.1007" | ||||||
|             VALUE "InternalName", "Rufus" |             VALUE "InternalName", "Rufus" | ||||||
|             VALUE "LegalCopyright", "© 2011-2016 Pete Batard (GPL v3)" |             VALUE "LegalCopyright", "© 2011-2016 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.11.1006" |             VALUE "ProductVersion", "2.11.1007" | ||||||
|         END |         END | ||||||
|     END |     END | ||||||
|     BLOCK "VarFileInfo" |     BLOCK "VarFileInfo" | ||||||
|  |  | ||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue