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} };
|
||||||
|
|
||||||
|
|
56
src/format.c
56
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…
Reference in a new issue