mirror of
https://github.com/ventoy/Ventoy.git
synced 2025-10-05 17:56:17 +00:00
Add Ventoy Sparse Image Option
This commit is contained in:
@@ -11,6 +11,8 @@
|
||||
|
||||
#include "ff.h" /* Obtains integer types */
|
||||
#include "diskio.h" /* Declarations of disk functions */
|
||||
#include "../Ventoy2Disk.h"
|
||||
|
||||
|
||||
/* Definitions of physical drive number for each drive */
|
||||
#define DEV_RAM 0 /* Example: Map Ramdisk to physical drive 0 */
|
||||
@@ -23,12 +25,41 @@ static UINT8 g_MbrSector[512];
|
||||
HANDLE g_hPhyDrive;
|
||||
UINT64 g_SectorCount;
|
||||
|
||||
FILE *g_VentoyImgFp = NULL;
|
||||
VTSI_SEGMENT *g_VentoySegment = NULL;
|
||||
int g_VentoyMaxSeg = 0;
|
||||
int g_VentoyCurSeg = -1;
|
||||
UINT64 g_VentoyDataOffset = 0;
|
||||
|
||||
void disk_io_set_param(HANDLE Handle, UINT64 SectorCount)
|
||||
{
|
||||
g_hPhyDrive = Handle;
|
||||
g_SectorCount = SectorCount;
|
||||
}
|
||||
|
||||
void disk_io_set_imghook(FILE *fp, VTSI_SEGMENT *segment, int maxseg, UINT64 data_offset)
|
||||
{
|
||||
g_VentoyImgFp = fp;
|
||||
g_VentoySegment = segment;
|
||||
g_VentoyMaxSeg = maxseg;
|
||||
|
||||
memset(segment, 0, maxseg * sizeof(VTSI_SEGMENT));
|
||||
g_VentoyCurSeg = -1;
|
||||
g_VentoyDataOffset = data_offset;
|
||||
}
|
||||
|
||||
void disk_io_reset_imghook(int *psegnum, UINT64 *pDataOffset)
|
||||
{
|
||||
*psegnum = g_VentoyCurSeg + 1;
|
||||
*pDataOffset = g_VentoyDataOffset;
|
||||
|
||||
g_VentoyImgFp = NULL;
|
||||
g_VentoySegment = NULL;
|
||||
g_VentoyMaxSeg = 0;
|
||||
g_VentoyCurSeg = -1;
|
||||
g_VentoyDataOffset = 0;
|
||||
}
|
||||
|
||||
/*-----------------------------------------------------------------------*/
|
||||
/* Get Drive Status */
|
||||
/*-----------------------------------------------------------------------*/
|
||||
@@ -126,6 +157,13 @@ DRESULT disk_read (
|
||||
BOOL bRet;
|
||||
LARGE_INTEGER liCurrentPosition;
|
||||
|
||||
//Log("xxx disk_read: sector:%ld count:%ld", (long)sector, (long)count);
|
||||
|
||||
if (g_VentoyImgFp)
|
||||
{
|
||||
return RES_OK;
|
||||
}
|
||||
|
||||
liCurrentPosition.QuadPart = sector * 512;
|
||||
SetFilePointerEx(g_hPhyDrive, liCurrentPosition, &liCurrentPosition, FILE_BEGIN);
|
||||
|
||||
@@ -162,6 +200,9 @@ DRESULT disk_write (
|
||||
DWORD dwSize;
|
||||
BOOL bRet;
|
||||
LARGE_INTEGER liCurrentPosition;
|
||||
VTSI_SEGMENT *CurSeg = NULL;
|
||||
|
||||
Log("==== disk_write: sector:%ld count:%ld", (long)sector, (long)count);
|
||||
|
||||
// skip MBR
|
||||
if (sector == 0)
|
||||
@@ -177,6 +218,31 @@ DRESULT disk_write (
|
||||
count--;
|
||||
}
|
||||
|
||||
if (g_VentoyImgFp)
|
||||
{
|
||||
CurSeg = g_VentoySegment + g_VentoyCurSeg;
|
||||
|
||||
if (g_VentoyCurSeg >= 0 && CurSeg->sector_num > 0 && sector == CurSeg->disk_start_sector + CurSeg->sector_num)
|
||||
{
|
||||
CurSeg->sector_num += count; //merge
|
||||
}
|
||||
else
|
||||
{
|
||||
g_VentoyCurSeg++;
|
||||
CurSeg++;
|
||||
|
||||
CurSeg->disk_start_sector = sector;
|
||||
CurSeg->data_offset = g_VentoyDataOffset;
|
||||
CurSeg->sector_num = count;
|
||||
}
|
||||
|
||||
g_VentoyDataOffset += count * 512;
|
||||
|
||||
fwrite(buff, 1, count * 512, g_VentoyImgFp);
|
||||
return RES_OK;
|
||||
}
|
||||
|
||||
|
||||
liCurrentPosition.QuadPart = sector * 512;
|
||||
SetFilePointerEx(g_hPhyDrive, liCurrentPosition, &liCurrentPosition, FILE_BEGIN);
|
||||
|
||||
|
@@ -6067,6 +6067,11 @@ FRESULT f_mkfs (
|
||||
/* Initialize the root directory */
|
||||
mem_set(buf, 0, sz_buf * ss);
|
||||
buf[SZDIRE * 0 + 0] = ET_VLABEL; /* Volume label entry (no label) */
|
||||
|
||||
//default label: Ventoy (unicode)
|
||||
buf[1] = 6; //label length
|
||||
memcpy(buf + 2, L"Ventoy", 2 * buf[1]);
|
||||
|
||||
buf[SZDIRE * 1 + 0] = ET_BITMAP; /* Bitmap entry */
|
||||
st_dword(buf + SZDIRE * 1 + 20, 2); /* cluster */
|
||||
st_dword(buf + SZDIRE * 1 + 24, szb_bit); /* size */
|
||||
|
Reference in New Issue
Block a user