mirror of
https://github.com/pbatard/rufus.git
synced 2024-08-14 23:57:05 +00:00
8738e7a7de
* The legacy code we used for writing disk images used the size of the source image as the maximum number of bytes we should copy, which is fine for uncompressed DD or VHD images, but not so much for compressed VHDX ones. So we now make sure to use the actual size of the virtual disk, which we obtain when mounting the VHD/VHDX. * Also fix log progress update as well as a MinGW warning.
142 lines
6 KiB
C
142 lines
6 KiB
C
/*
|
|
* Rufus: The Reliable USB Formatting Utility
|
|
* Virtual Disk Handling definitions and prototypes
|
|
* Copyright © 2022-2024 Pete Batard <pete@akeo.ie>
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <stdint.h>
|
|
#include <windows.h>
|
|
#include <virtdisk.h>
|
|
|
|
#pragma once
|
|
|
|
#define WIM_MAGIC 0x0000004D4957534DULL // "MSWIM\0\0\0"
|
|
#define WIM_HAS_API_EXTRACT 1
|
|
#define WIM_HAS_7Z_EXTRACT 2
|
|
#define WIM_HAS_API_APPLY 4
|
|
#define WIM_HAS_EXTRACT(r) (r & (WIM_HAS_API_EXTRACT|WIM_HAS_7Z_EXTRACT))
|
|
|
|
#define SECONDS_SINCE_JAN_1ST_2000 946684800
|
|
|
|
#define INVALID_CALLBACK_VALUE 0xFFFFFFFF
|
|
|
|
#define WIM_FLAG_RESERVED 0x00000001
|
|
#define WIM_FLAG_VERIFY 0x00000002
|
|
#define WIM_FLAG_INDEX 0x00000004
|
|
#define WIM_FLAG_NO_APPLY 0x00000008
|
|
#define WIM_FLAG_NO_DIRACL 0x00000010
|
|
#define WIM_FLAG_NO_FILEACL 0x00000020
|
|
#define WIM_FLAG_SHARE_WRITE 0x00000040
|
|
#define WIM_FLAG_FILEINFO 0x00000080
|
|
#define WIM_FLAG_NO_RP_FIX 0x00000100
|
|
|
|
// Bitmask for the kind of progress we want to report in the WIM progress callback
|
|
#define WIM_REPORT_PROGRESS 0x00000001
|
|
#define WIM_REPORT_PROCESS 0x00000002
|
|
#define WIM_REPORT_FILEINFO 0x00000004
|
|
|
|
#define WIM_GENERIC_READ GENERIC_READ
|
|
#define WIM_OPEN_EXISTING OPEN_EXISTING
|
|
#define WIM_UNDOCUMENTED_BULLSHIT 0x20000000
|
|
|
|
#define MBR_SIZE 512 // Might need to review this once we see bootable 4k systems
|
|
|
|
// TODO: Remove this once MinGW has been updated
|
|
#ifndef VIRTUAL_STORAGE_TYPE_DEVICE_VHDX
|
|
#define VIRTUAL_STORAGE_TYPE_DEVICE_VHDX 3
|
|
#endif
|
|
#define VIRTUAL_STORAGE_TYPE_DEVICE_FFU 99
|
|
#define CREATE_VIRTUAL_DISK_VERSION_2 2
|
|
#define CREATE_VIRTUAL_DISK_FLAG_CREATE_BACKING_STORAGE 8
|
|
|
|
#ifndef CREATE_VIRTUAL_DISK_PARAMETERS_DEFAULT_BLOCK_SIZE
|
|
#define CREATE_VIRTUAL_DISK_PARAMETERS_DEFAULT_BLOCK_SIZE 0
|
|
#endif
|
|
|
|
#ifndef CREATE_VIRTUAL_DISK_PARAMETERS_DEFAULT_SECTOR_SIZE
|
|
#define CREATE_VIRTUAL_DISK_PARAMETERS_DEFAULT_SECTOR_SIZE 0
|
|
#endif
|
|
|
|
typedef struct
|
|
{
|
|
CREATE_VIRTUAL_DISK_VERSION Version;
|
|
union
|
|
{
|
|
struct
|
|
{
|
|
GUID UniqueId;
|
|
ULONGLONG MaximumSize;
|
|
ULONG BlockSizeInBytes;
|
|
ULONG SectorSizeInBytes;
|
|
PCWSTR ParentPath;
|
|
PCWSTR SourcePath;
|
|
} Version1;
|
|
struct
|
|
{
|
|
GUID UniqueId;
|
|
ULONGLONG MaximumSize;
|
|
ULONG BlockSizeInBytes;
|
|
ULONG SectorSizeInBytes;
|
|
ULONG PhysicalSectorSizeInBytes;
|
|
PCWSTR ParentPath;
|
|
PCWSTR SourcePath;
|
|
OPEN_VIRTUAL_DISK_FLAG OpenFlags;
|
|
VIRTUAL_STORAGE_TYPE ParentVirtualStorageType;
|
|
VIRTUAL_STORAGE_TYPE SourceVirtualStorageType;
|
|
GUID ResiliencyGuid;
|
|
} Version2;
|
|
};
|
|
} STOPGAP_CREATE_VIRTUAL_DISK_PARAMETERS;
|
|
|
|
// From https://docs.microsoft.com/en-us/previous-versions/msdn10/dd834960(v=msdn.10)
|
|
// as well as https://msfn.org/board/topic/150700-wimgapi-wimmountimage-progressbar/
|
|
enum WIMMessage {
|
|
WIM_MSG = WM_APP + 0x1476,
|
|
WIM_MSG_TEXT,
|
|
WIM_MSG_PROGRESS, // Indicates an update in the progress of an image application.
|
|
WIM_MSG_PROCESS, // Enables the caller to prevent a file or a directory from being captured or applied.
|
|
WIM_MSG_SCANNING, // Indicates that volume information is being gathered during an image capture.
|
|
WIM_MSG_SETRANGE, // Indicates the number of files that will be captured or applied.
|
|
WIM_MSG_SETPOS, // Indicates the number of files that have been captured or applied.
|
|
WIM_MSG_STEPIT, // Indicates that a file has been either captured or applied.
|
|
WIM_MSG_COMPRESS, // Enables the caller to prevent a file resource from being compressed during a capture.
|
|
WIM_MSG_ERROR, // Alerts the caller that an error has occurred while capturing or applying an image.
|
|
WIM_MSG_ALIGNMENT, // Enables the caller to align a file resource on a particular alignment boundary.
|
|
WIM_MSG_RETRY, // Sent when the file is being reapplied because of a network timeout.
|
|
WIM_MSG_SPLIT, // Enables the caller to align a file resource on a particular alignment boundary.
|
|
WIM_MSG_FILEINFO, // Used in conjunction with WimApplyImages()'s WIM_FLAG_FILEINFO flag to provide detailed file info.
|
|
WIM_MSG_INFO, // Sent when an info message is available.
|
|
WIM_MSG_WARNING, // Sent when a warning message is available.
|
|
WIM_MSG_CHK_PROCESS,
|
|
WIM_MSG_SUCCESS = 0,
|
|
WIM_MSG_ABORT_IMAGE = -1
|
|
};
|
|
|
|
extern uint8_t WimExtractCheck(BOOL bSilent);
|
|
extern BOOL WimExtractFile(const char* wim_image, int index, const char* src, const char* dst, BOOL bSilent);
|
|
extern BOOL WimExtractFile_API(const char* image, int index, const char* src, const char* dst, BOOL bSilent);
|
|
extern BOOL WimExtractFile_7z(const char* image, int index, const char* src, const char* dst, BOOL bSilent);
|
|
extern BOOL WimApplyImage(const char* image, int index, const char* dst);
|
|
extern char* WimMountImage(const char* image, int index);
|
|
extern BOOL WimUnmountImage(const char* image, int index, BOOL commit);
|
|
extern char* WimGetExistingMountPoint(const char* image, int index);
|
|
extern BOOL WimIsValidIndex(const char* image, int index);
|
|
extern int8_t IsBootableImage(const char* path);
|
|
extern char* VhdMountImageAndGetSize(const char* path, uint64_t* disksize);
|
|
#define VhdMountImage(path) VhdMountImageAndGetSize(path, NULL)
|
|
extern void VhdUnmountImage(void);
|
|
extern void VhdSaveImage(void);
|
|
extern void IsoSaveImage(void);
|