Fix the long delay for Tails

[Ventoy.git] / Ventoy2Disk / Ventoy2Disk / Ventoy2Disk.h

diff --git a/Ventoy2Disk/Ventoy2Disk/Ventoy2Disk.h b/Ventoy2Disk/Ventoy2Disk/Ventoy2Disk.h
index 153141847372f659ba2c01daa857377ac09093ae..dca9eb9fca96d411663ff857ea3997bdb4033188 100644 (file)
--- a/Ventoy2Disk/Ventoy2Disk/Ventoy2Disk.h
+++ b/Ventoy2Disk/Ventoy2Disk/Ventoy2Disk.h
@@ -53,6 +53,7 @@
 }\r
 \r
 #define LASTERR     GetLastError()\r
+#define RET_LASTERR (ret ? 0 : LASTERR)\r
 \r
 #pragma pack(1)\r
 typedef struct PART_TABLE\r
@@ -148,17 +149,18 @@ typedef struct PHY_DRIVE_INFO
     STORAGE_BUS_TYPE BusType;\r
 \r
     CHAR DriveLetters[64];\r
-\r
+   \r
     CHAR VentoyVersion[32];\r
 \r
     BOOL SecureBootSupport;\r
     MBR_HEAD MBR;\r
+    UINT64 Part2GPTAttr;\r
 }PHY_DRIVE_INFO;\r
 \r
 typedef enum PROGRESS_POINT\r
 {\r
     PT_START = 0,\r
-    PT_LOCK_FOR_CLEAN,\r
+    PT_LOCK_FOR_CLEAN = 8,\r
     PT_DEL_ALL_PART,\r
     PT_LOCK_FOR_WRITE,\r
     PT_FORMAT_PART1,\r
@@ -185,6 +187,7 @@ extern HFONT g_language_normal_font;
 extern HFONT g_language_bold_font;\r
 extern int g_FilterUSB;\r
 \r
+void TraceOut(const char *Fmt, ...);\r
 void Log(const char *Fmt, ...);\r
 BOOL IsPathExist(BOOL Dir, const char *Fmt, ...);\r
 void DumpWindowsVersion(void);\r
@@ -207,11 +210,12 @@ int Ventoy2DiskInit(void);
 int Ventoy2DiskDestroy(void);\r
 PHY_DRIVE_INFO * GetPhyDriveInfoById(int Id);\r
 int ParseCmdLineOption(LPSTR lpCmdLine);\r
-int InstallVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive, int PartStyle);\r
-int UpdateVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive);\r
+int InstallVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive, int PartStyle, int TryId);\r
+int UpdateVentoy2PhyDrive(PHY_DRIVE_INFO *pPhyDrive, int TryId);\r
 int VentoyFillBackupGptHead(VTOY_GPT_INFO *pInfo, VTOY_GPT_HDR *pHead);\r
 int VentoyFillWholeGpt(UINT64 DiskSizeBytes, VTOY_GPT_INFO *pInfo);\r
 void SetProgressBarPos(int Pos);\r
+int SaveBufToFile(const CHAR *FileName, const void *Buffer, int BufLen);\r
 int ReadWholeFileToBuf(const CHAR *FileName, int ExtLen, void **Bufer, int *BufLen);\r
 int INIT unxz(unsigned char *in, int in_size,\r
     int(*fill)(void *dest, unsigned int size),\r
@@ -231,5 +235,67 @@ UINT32 VentoyCrc32(void *Buffer, UINT32 Length);
     liCurrentPosition.QuadPart = pos; \\r
     SetFilePointerEx(hDrive, liCurrentPosition, &liCurrentPosition, FILE_BEGIN)\\r
 \r
+#define SECURE_ICON_STRING _UICON(UNICODE_LOCK)\r
+\r
+extern int g_WriteImage;\r
+\r
+#define VTSI_IMG_MAGIC 0x0000594F544E4556ULL  // "VENTOY\0\0"\r
+\r
+#pragma pack(1)\r
+\r
+/*\r
+ +---------------------------------\r
+ + sector 0 ~ sector N-1\r
+ +     data area\r
+ +---------------------------------\r
+ + sector N ~ \r
+ +     segment[0]\r
+ +     segment[1]\r
+ +     segment[2]\r
+ +      ......\r
+ +     segment[M-1]\r
+ +     align data (aligned with 512)\r
+ +---------------------------------\r
+ +     footer\r
+ +---------------------------------\r
+ *\r
+ * All the integers are in little endian\r
+ * The sector size is fixed 512 for ventoy image file.\r
+ *\r
+ */\r
+\r
+#define VTSI_IMG_MAX_SEG   128\r
+\r
+typedef struct {\r
+    UINT64 disk_start_sector;\r
+    UINT64 sector_num;\r
+    UINT64 data_offset;\r
+}VTSI_SEGMENT;\r
+\r
+typedef struct {\r
+    UINT64 magic;\r
+    UINT32 version;\r
+    UINT64 disk_size;\r
+    UINT32 disk_signature;\r
+    UINT32 foot_chksum;\r
+\r
+    UINT32 segment_num;\r
+    UINT32 segment_chksum;\r
+    UINT64 segment_offset;\r
+\r
+    UINT8  reserved[512 - 44];\r
+}VTSI_FOOTER;\r
+#pragma pack()\r
+extern int __static_assert__[sizeof(VTSI_FOOTER) == 512 ? 1 : -1];\r
+\r
+\r
+#define SAFE_FREE(ptr) if (ptr) { free(ptr); (ptr) = NULL; }\r
+int InstallVentoy2FileImage(PHY_DRIVE_INFO *pPhyDrive, int PartStyle);\r
+void disk_io_set_imghook(FILE *fp, VTSI_SEGMENT *segment, int maxseg, UINT64 data_offset);\r
+void disk_io_reset_imghook(int *psegnum, UINT64 *pDataOffset);\r
+\r
+\r
+#define VTSI_SUPPORT 1\r
+\r
 \r
 #endif\r