GRUB2/MOD_SRC/grub-2.04/grub-core/ventoy/ventoy_def.h

   1 /******************************************************************************
   2  * ventoy_def.h
   3  *
   4  * Copyright (c) 2020, longpanda <admin@ventoy.net>
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU General Public License as
   8  * published by the Free Software Foundation; either version 3 of the
   9  * License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful, but
  12  * WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, see <http://www.gnu.org/licenses/>.
  18  *
  19  */
  20
  21 #ifndef __VENTOY_DEF_H__
  22 #define __VENTOY_DEF_H__
  23
  24 #define VTOY_MAX_SCRIPT_BUF    (4 * 1024 * 1024)
  25
  26 #define VTOY_PART_BUF_LEN  (128 * 1024)
  27
  28 #define VTOY_FILT_MIN_FILE_SIZE  32768
  29
  30 #define VTOY_SIZE_1GB     1073741824
  31 #define VTOY_SIZE_512KB  (512 * 1024)
  32 #define VTOY_SIZE_1KB     1024
  33
  34 #define JSON_SUCCESS    0
  35 #define JSON_FAILED     1
  36 #define JSON_NOT_FOUND  2
  37
  38 #define ulong unsigned long
  39 #define ulonglong  unsigned long long
  40
  41 #define vtoy_to_upper(c) (((char)(c) >= 'a' && (char)(c) <= 'z') ? ((char)(c) - 'a' + 'A') : (char)(c))
  42
  43 #define VENTOY_CMD_RETURN(err)  grub_errno = (err); return (err)
  44 #define VENTOY_FILE_TYPE    (GRUB_FILE_TYPE_NO_DECOMPRESS | GRUB_FILE_TYPE_LINUX_INITRD)
  45
  46 #define ventoy_env_op1(op, a) grub_env_##op(a)
  47 #define ventoy_env_op2(op, a, b) grub_env_##op((a), (b))
  48
  49 #define ventoy_get_env(key)         ventoy_env_op1(get, key)
  50 #define ventoy_set_env(key, val)    ventoy_env_op2(set, key, val)
  51
  52 #define VTOY_WARNING  "!!!!!!!!!!!!! WARNING !!!!!!!!!!!!!"
  53
  54 typedef struct ventoy_initrd_ctx
  55 {
  56     const char *path_prefix;
  57     const char *dir_prefix;
  58 }ventoy_initrd_ctx;
  59
  60 typedef struct cmd_para
  61 {
  62     const char *name;
  63     grub_extcmd_func_t func;
  64     grub_command_flags_t flags;
  65     const struct grub_arg_option *parser;
  66
  67     const char *summary;
  68     const char *description;
  69
  70     grub_extcmd_t cmd;
  71 }cmd_para;
  72
  73 #define ventoy_align_2k(value)  ((value + 2047) / 2048 * 2048)
  74 #define ventoy_align(value, align)  (((value) + ((align) - 1)) & (~((align) - 1)))
  75
  76 #pragma pack(1)
  77 typedef struct cpio_newc_header
  78 {
  79     char  c_magic[6];
  80     char  c_ino[8];
  81     char  c_mode[8];
  82     char  c_uid[8];
  83     char  c_gid[8];
  84     char  c_nlink[8];
  85     char  c_mtime[8];
  86     char  c_filesize[8];
  87     char  c_devmajor[8];
  88     char  c_devminor[8];
  89     char  c_rdevmajor[8];
  90     char  c_rdevminor[8];
  91     char  c_namesize[8];
  92     char  c_check[8];
  93 }cpio_newc_header;
  94 #pragma pack()
  95
  96
  97 #define cmd_raw_name ctxt->extcmd->cmd->name
  98 #define check_free(p, func) if (p) { func(p); p = NULL; }
  99 #define grub_check_free(p) if (p) { grub_free(p); p = NULL; }
 100
 101 typedef int (*grub_char_check_func)(int c);
 102 #define ventoy_is_decimal(str)  ventoy_string_check(str, grub_isdigit)
 103
 104
 105 #pragma pack(1)
 106 typedef struct ventoy_patch_vhd
 107 {
 108     grub_uint8_t  part_offset_or_guid[16];
 109     grub_uint32_t reserved1;
 110     grub_uint32_t part_type;
 111     grub_uint8_t  disk_signature_or_guid[16];
 112     grub_uint8_t  reserved2[16];
 113     grub_uint8_t  vhd_file_path[1];
 114 }ventoy_patch_vhd;
 115 #pragma pack()
 116
 117 // El Torito Boot Record Volume Descriptor
 118 #pragma pack(1)
 119 typedef struct eltorito_descriptor
 120 {
 121     grub_uint8_t type;
 122     grub_uint8_t id[5];
 123     grub_uint8_t version;
 124     grub_uint8_t system_id[32];
 125     grub_uint8_t reserved[32];
 126     grub_uint32_t sector;
 127 }eltorito_descriptor;
 128
 129 typedef struct ventoy_iso9660_override
 130 {
 131     grub_uint32_t first_sector;
 132     grub_uint32_t first_sector_be;
 133     grub_uint32_t size;
 134     grub_uint32_t size_be;
 135 }ventoy_iso9660_override;
 136
 137 typedef struct ventoy_udf_override
 138 {
 139     grub_uint32_t length;
 140     grub_uint32_t position;
 141 }ventoy_udf_override;
 142
 143 typedef struct ventoy_iso9660_vd
 144 {
 145     grub_uint8_t type;
 146     grub_uint8_t id[5];
 147     grub_uint8_t ver;
 148     grub_uint8_t res;
 149     char sys[32];
 150     char vol[32];
 151 }ventoy_iso9660_vd;
 152
 153 #pragma pack()
 154
 155 #define img_type_iso 0
 156 #define img_type_wim 1
 157 #define img_type_efi 2
 158 #define img_type_img 3
 159 #define img_type_vhd 4
 160
 161 typedef struct img_info
 162 {
 163     int pathlen;
 164     char path[512];
 165     char name[256];
 166
 167     const char *alias;
 168     const char *class;
 169     const char *menu_prefix;
 170
 171     int id;
 172     int type;
 173     grub_uint64_t size;
 174     int select;
 175     int unsupport;
 176
 177     void *parent;
 178
 179     struct img_info *next;
 180     struct img_info *prev;
 181 }img_info;
 182
 183 typedef struct img_iterator_node
 184 {
 185     struct img_iterator_node *next;
 186     img_info **tail;
 187     char dir[400];
 188     int dirlen;
 189     int isocnt;
 190     int done;
 191     int select;
 192
 193     struct img_iterator_node *parent;
 194     struct img_iterator_node *firstchild;
 195
 196     void *firstiso;
 197 }img_iterator_node;
 198
 199
 200
 201 typedef struct initrd_info
 202 {
 203     char name[256];
 204
 205     grub_uint64_t offset;
 206     grub_uint64_t size;
 207
 208     grub_uint8_t  iso_type; // 0: iso9660  1:udf
 209     grub_uint32_t udf_start_block;
 210
 211     grub_uint64_t override_offset;
 212     grub_uint32_t override_length;
 213     char          override_data[32];
 214
 215     struct initrd_info *next;
 216     struct initrd_info *prev;
 217 }initrd_info;
 218
 219 extern initrd_info *g_initrd_img_list;
 220 extern initrd_info *g_initrd_img_tail;
 221 extern int g_initrd_img_count;
 222 extern int g_valid_initrd_count;
 223
 224 extern img_info *g_ventoy_img_list;
 225 extern int g_ventoy_img_count;
 226
 227 extern grub_uint8_t *g_ventoy_cpio_buf;
 228 extern grub_uint32_t g_ventoy_cpio_size;
 229 extern cpio_newc_header *g_ventoy_initrd_head;
 230 extern grub_uint8_t *g_ventoy_runtime_buf;
 231
 232 extern ventoy_guid  g_ventoy_guid;
 233
 234 extern ventoy_img_chunk_list g_img_chunk_list;
 235 extern ventoy_img_chunk_list g_wimiso_chunk_list;
 236 extern char *g_wimiso_path;
 237
 238 extern int g_ventoy_debug;
 239 void ventoy_debug(const char *fmt, ...);
 240 #define debug(fmt, ...) if (g_ventoy_debug) ventoy_debug("[VTOY]: "fmt, __VA_ARGS__)
 241
 242 #define vtoy_ssprintf(buf, pos, fmt, ...) \
 243     pos += grub_snprintf(buf + pos, VTOY_MAX_SCRIPT_BUF - pos, fmt, __VA_ARGS__)
 244
 245 #define FLAG_HEADER_RESERVED          0x00000001
 246 #define FLAG_HEADER_COMPRESSION       0x00000002
 247 #define FLAG_HEADER_READONLY          0x00000004
 248 #define FLAG_HEADER_SPANNED           0x00000008
 249 #define FLAG_HEADER_RESOURCE_ONLY     0x00000010
 250 #define FLAG_HEADER_METADATA_ONLY     0x00000020
 251 #define FLAG_HEADER_WRITE_IN_PROGRESS 0x00000040
 252 #define FLAG_HEADER_RP_FIX            0x00000080 // reparse point fixup
 253 #define FLAG_HEADER_COMPRESS_RESERVED 0x00010000
 254 #define FLAG_HEADER_COMPRESS_XPRESS   0x00020000
 255 #define FLAG_HEADER_COMPRESS_LZX      0x00040000
 256 #define FLAG_HEADER_COMPRESS_LZMS     0x00080000
 257
 258 #define RESHDR_FLAG_FREE 0x01
 259 #define RESHDR_FLAG_METADATA 0x02
 260 #define RESHDR_FLAG_COMPRESSED 0x04
 261 #define RESHDR_FLAG_SPANNED 0x08
 262
 263 #pragma pack(1)
 264
 265 /* A WIM resource header */
 266 typedef struct wim_resource_header
 267 {
 268     grub_uint64_t size_in_wim:56; /* Compressed length */
 269     grub_uint64_t flags:8;        /* flags  */
 270     grub_uint64_t offset;         /* Offset */
 271     grub_uint64_t raw_size;       /* Uncompressed length */
 272 }wim_resource_header;
 273
 274 /* WIM resource header length mask */
 275 #define WIM_RESHDR_ZLEN_MASK 0x00ffffffffffffffULL
 276
 277 /* WIM resource header flags */
 278 typedef enum wim_resource_header_flags
 279 {
 280     WIM_RESHDR_METADATA = ( 0x02ULL << 56 ),       /* Resource contains metadata */
 281     WIM_RESHDR_COMPRESSED = ( 0x04ULL << 56 ),     /* Resource is compressed */
 282     WIM_RESHDR_PACKED_STREAMS = ( 0x10ULL << 56 ), /* Resource is compressed using packed streams */
 283 }wim_resource_header_flags;
 284
 285 #define WIM_HEAD_SIGNATURE   "MSWIM\0\0"
 286
 287 /* WIM header */
 288 typedef struct wim_header
 289 {
 290     grub_uint8_t signature[8];          /* Signature */
 291     grub_uint32_t header_len;           /* Header length */
 292     grub_uint32_t version;              /* Verson */
 293     grub_uint32_t flags;                /* Flags */
 294     grub_uint32_t chunk_len;            /* Chunk length */
 295     grub_uint8_t guid[16];              /* GUID */
 296     grub_uint16_t part;                 /* Part number */
 297     grub_uint16_t parts;                /* Total number of parts */
 298     grub_uint32_t images;               /* number of images */
 299     wim_resource_header lookup;    /* Lookup table */
 300     wim_resource_header xml;       /* XML data */
 301     wim_resource_header metadata;  /* Boot metadata */
 302     grub_uint32_t boot_index;           /* Boot index */
 303     wim_resource_header integrity; /* Integrity table */
 304     grub_uint8_t reserved[60];          /* Reserved */
 305 } wim_header;
 306
 307 /* WIM header flags */
 308 typedef enum wim_header_flags
 309 {
 310     WIM_HDR_XPRESS = 0x00020000, /* WIM uses Xpress compresson */
 311     WIM_HDR_LZX = 0x00040000,    /* WIM uses LZX compression */
 312 }wim_header_flags;
 313
 314 /* A WIM file hash */
 315 typedef struct wim_hash
 316 {
 317     /* SHA-1 hash */
 318     grub_uint8_t sha1[20];
 319 }wim_hash;
 320
 321 /* A WIM lookup table entry */
 322 typedef struct wim_lookup_entry
 323 {
 324     wim_resource_header resource; /* Resource header */
 325     grub_uint16_t part;           /* Part number */
 326     grub_uint32_t refcnt;         /* Reference count */
 327     wim_hash hash;                /* Hash */
 328 }wim_lookup_entry;
 329
 330 /* WIM chunk length */
 331 #define WIM_CHUNK_LEN 32768
 332
 333 /* A WIM chunk buffer */
 334 typedef struct wim_chunk_buffer
 335 {
 336     grub_uint8_t data[WIM_CHUNK_LEN]; /*Data */
 337 }wim_chunk_buffer;
 338
 339 /* Security data */
 340 typedef struct wim_security_header
 341 {
 342     grub_uint32_t len;   /* Length */
 343     grub_uint32_t count; /* Number of entries */
 344 }wim_security_header;
 345
 346 /* Directory entry */
 347 typedef struct wim_directory_entry
 348 {
 349     grub_uint64_t len;                 /* Length */
 350     grub_uint32_t attributes;     /* Attributes */
 351     grub_uint32_t security;       /* Security ID */
 352     grub_uint64_t subdir;              /* Subdirectory offset */
 353     grub_uint8_t reserved1[16];   /* Reserved */
 354     grub_uint64_t created;             /* Creation time */
 355     grub_uint64_t accessed;            /* Last access time */
 356     grub_uint64_t written;             /* Last written time */
 357     wim_hash hash;                /* Hash */
 358     grub_uint8_t reserved2[12];   /* Reserved */
 359     grub_uint16_t streams;        /* Streams */
 360     grub_uint16_t short_name_len; /* Short name length */
 361     grub_uint16_t name_len;       /* Name length */
 362 }wim_directory_entry;
 363
 364 /** Normal file */
 365 #define WIM_ATTR_NORMAL 0x00000080UL
 366
 367 /** No security information exists for this file */
 368 #define WIM_NO_SECURITY 0xffffffffUL
 369
 370 #pragma pack()
 371
 372
 373 typedef struct wim_tail
 374 {
 375     grub_uint32_t wim_raw_size;
 376     grub_uint32_t wim_align_size;
 377
 378     grub_uint8_t  iso_type;
 379     grub_uint64_t file_offset;
 380     grub_uint32_t udf_start_block;
 381     grub_uint64_t fe_entry_size_offset;
 382     grub_uint64_t override_offset;
 383     grub_uint32_t override_len;
 384     grub_uint8_t  override_data[32];
 385
 386     wim_header wim_header;
 387
 388     wim_hash bin_hash;
 389     grub_uint32_t jump_exe_len;
 390     grub_uint8_t *jump_bin_data;
 391     grub_uint32_t bin_raw_len;
 392     grub_uint32_t bin_align_len;
 393
 394     grub_uint8_t *new_meta_data;
 395     grub_uint32_t new_meta_len;
 396     grub_uint32_t new_meta_align_len;
 397
 398     grub_uint8_t *new_lookup_data;
 399     grub_uint32_t new_lookup_len;
 400     grub_uint32_t new_lookup_align_len;
 401 }wim_tail;
 402
 403 typedef struct wim_patch
 404 {
 405     int pathlen;
 406     char path[256];
 407
 408     wim_hash old_hash;
 409     wim_tail wim_data;
 410     wim_lookup_entry *replace_look;
 411
 412     int valid;
 413
 414     struct wim_patch *next;
 415 }wim_patch;
 416
 417
 418 typedef enum _JSON_TYPE
 419 {
 420     JSON_TYPE_NUMBER = 0,
 421     JSON_TYPE_STRING,
 422     JSON_TYPE_BOOL,
 423     JSON_TYPE_ARRAY,
 424     JSON_TYPE_OBJECT,
 425     JSON_TYPE_NULL,
 426     JSON_TYPE_BUTT
 427 }JSON_TYPE;
 428
 429
 430 typedef struct _VTOY_JSON
 431 {
 432     struct _VTOY_JSON *pstPrev;
 433     struct _VTOY_JSON *pstNext;
 434     struct _VTOY_JSON *pstChild;
 435
 436     JSON_TYPE enDataType;
 437     union
 438     {
 439         char  *pcStrVal;
 440         int   iNumVal;
 441         grub_uint64_t lValue;
 442     }unData;
 443
 444     char *pcName;
 445 }VTOY_JSON;
 446
 447 typedef struct _JSON_PARSE
 448 {
 449     char *pcKey;
 450     void *pDataBuf;
 451     grub_uint32_t  uiBufSize;
 452 }JSON_PARSE;
 453
 454 #define JSON_NEW_ITEM(pstJson, ret) \
 455 { \
 456     (pstJson) = (VTOY_JSON *)grub_zalloc(sizeof(VTOY_JSON)); \
 457     if (NULL == (pstJson)) \
 458     { \
 459         json_debug("Failed to alloc memory for json.\n"); \
 460         return (ret); \
 461     } \
 462 }
 463
 464 typedef int (*ventoy_plugin_entry_pf)(VTOY_JSON *json, const char *isodisk);
 465 typedef int (*ventoy_plugin_check_pf)(VTOY_JSON *json, const char *isodisk);
 466
 467 typedef struct plugin_entry
 468 {
 469     const char *key;
 470     ventoy_plugin_entry_pf entryfunc;
 471     ventoy_plugin_check_pf checkfunc;
 472 }plugin_entry;
 473
 474
 475 void ventoy_fill_os_param(grub_file_t file, ventoy_os_param *param);
 476 grub_err_t ventoy_cmd_isolinux_initrd_collect(grub_extcmd_context_t ctxt, int argc, char **args);
 477 grub_err_t ventoy_cmd_grub_initrd_collect(grub_extcmd_context_t ctxt, int argc, char **args);
 478 grub_err_t ventoy_cmd_specify_initrd_file(grub_extcmd_context_t ctxt, int argc, char **args);
 479 grub_err_t ventoy_cmd_dump_initrd_list(grub_extcmd_context_t ctxt, int argc, char **args);
 480 grub_err_t ventoy_cmd_clear_initrd_list(grub_extcmd_context_t ctxt, int argc, char **args);
 481 grub_uint32_t ventoy_get_iso_boot_catlog(grub_file_t file);
 482 int ventoy_has_efi_eltorito(grub_file_t file, grub_uint32_t sector);
 483 grub_err_t ventoy_cmd_linux_chain_data(grub_extcmd_context_t ctxt, int argc, char **args);
 484 grub_err_t ventoy_cmd_linux_locate_initrd(grub_extcmd_context_t ctxt, int argc, char **args);
 485 grub_err_t ventoy_cmd_initrd_count(grub_extcmd_context_t ctxt, int argc, char **args);
 486 grub_err_t ventoy_cmd_valid_initrd_count(grub_extcmd_context_t ctxt, int argc, char **args);
 487 grub_err_t ventoy_cmd_load_cpio(grub_extcmd_context_t ctxt, int argc, char **args);
 488 grub_err_t ventoy_cmd_cpio_busybox_64(grub_extcmd_context_t ctxt, int argc, char **args);
 489 grub_err_t ventoy_cmd_trailer_cpio(grub_extcmd_context_t ctxt, int argc, char **args);
 490 int ventoy_cpio_newc_fill_head(void *buf, int filesize, const void *filedata, const char *name);
 491 grub_file_t ventoy_grub_file_open(enum grub_file_type type, const char *fmt, ...);
 492 grub_uint64_t ventoy_grub_get_file_size(const char *fmt, ...);
 493 int ventoy_is_file_exist(const char *fmt, ...);
 494 int ventoy_is_dir_exist(const char *fmt, ...);
 495 int ventoy_fill_data(grub_uint32_t buflen, char *buffer);
 496 grub_err_t ventoy_cmd_load_plugin(grub_extcmd_context_t ctxt, int argc, char **args);
 497 grub_err_t ventoy_cmd_wimdows_reset(grub_extcmd_context_t ctxt, int argc, char **args);
 498 grub_err_t ventoy_cmd_windows_chain_data(grub_extcmd_context_t ctxt, int argc, char **args);
 499 grub_err_t ventoy_cmd_wim_chain_data(grub_extcmd_context_t ctxt, int argc, char **args);
 500 grub_err_t ventoy_cmd_dump_wim_patch(grub_extcmd_context_t ctxt, int argc, char **args);
 501
 502 VTOY_JSON *vtoy_json_find_item
 503 (
 504     VTOY_JSON *pstJson,
 505     JSON_TYPE  enDataType,
 506     const char *szKey
 507 );
 508 int vtoy_json_parse_value
 509 (
 510     char *pcNewStart,
 511     char *pcRawStart,
 512     VTOY_JSON *pstJson,
 513     const char *pcData,
 514     const char **ppcEnd
 515 );
 516 VTOY_JSON * vtoy_json_create(void);
 517 int vtoy_json_parse(VTOY_JSON *pstJson, const char *szJsonData);
 518
 519 int vtoy_json_scan_parse
 520 (
 521     const VTOY_JSON    *pstJson,
 522     grub_uint32_t       uiParseNum,
 523     JSON_PARSE         *pstJsonParse
 524 );
 525
 526 int vtoy_json_scan_array
 527 (
 528      VTOY_JSON *pstJson,
 529      const char *szKey,
 530      VTOY_JSON **ppstArrayItem
 531 );
 532
 533 int vtoy_json_scan_array_ex
 534 (
 535      VTOY_JSON *pstJson,
 536      const char *szKey,
 537      VTOY_JSON **ppstArrayItem
 538 );
 539 int vtoy_json_scan_object
 540 (
 541      VTOY_JSON *pstJson,
 542      const char *szKey,
 543     VTOY_JSON **ppstObjectItem
 544 );
 545 int vtoy_json_get_int
 546 (
 547     VTOY_JSON *pstJson,
 548     const char *szKey,
 549     int *piValue
 550 );
 551 int vtoy_json_get_uint
 552 (
 553     VTOY_JSON *pstJson,
 554     const char *szKey,
 555     grub_uint32_t *puiValue
 556 );
 557 int vtoy_json_get_uint64
 558 (
 559     VTOY_JSON *pstJson,
 560     const char *szKey,
 561     grub_uint64_t *pui64Value
 562 );
 563 int vtoy_json_get_bool
 564 (
 565     VTOY_JSON *pstJson,
 566     const char *szKey,
 567     grub_uint8_t *pbValue
 568 );
 569 int vtoy_json_get_string
 570 (
 571      VTOY_JSON *pstJson,
 572      const char *szKey,
 573      grub_uint32_t  uiBufLen,
 574      char *pcBuf
 575 );
 576 const char * vtoy_json_get_string_ex(VTOY_JSON *pstJson,  const char *szKey);
 577 int vtoy_json_destroy(VTOY_JSON *pstJson);
 578
 579
 580 grub_uint32_t CalculateCrc32
 581 (
 582     const void       *Buffer,
 583     grub_uint32_t          Length,
 584     grub_uint32_t          InitValue
 585 );
 586
 587 static inline int ventoy_isspace (int c)
 588 {
 589     return (c == '\n' || c == '\r' || c == ' ' || c == '\t');
 590 }
 591
 592 static inline int ventoy_is_word_end(int c)
 593 {
 594     return (c == 0 || c == ',' || ventoy_isspace(c));
 595 }
 596
 597 #pragma pack(1)
 598 typedef struct ventoy_part_table
 599 {
 600     grub_uint8_t  Active; // 0x00  0x80
 601
 602     grub_uint8_t  StartHead;
 603     grub_uint16_t StartSector : 6;
 604     grub_uint16_t StartCylinder : 10;
 605
 606     grub_uint8_t  FsFlag;
 607
 608     grub_uint8_t  EndHead;
 609     grub_uint16_t EndSector : 6;
 610     grub_uint16_t EndCylinder : 10;
 611
 612     grub_uint32_t StartSectorId;
 613     grub_uint32_t SectorCount;
 614 }ventoy_part_table;
 615
 616 typedef struct ventoy_mbr_head
 617 {
 618     grub_uint8_t BootCode[446];
 619     ventoy_part_table PartTbl[4];
 620     grub_uint8_t Byte55;
 621     grub_uint8_t ByteAA;
 622 }ventoy_mbr_head;
 623
 624 typedef struct ventoy_gpt_head
 625 {
 626     char   Signature[8]; /* EFI PART */
 627     grub_uint8_t  Version[4];
 628     grub_uint32_t Length;
 629     grub_uint32_t Crc;
 630     grub_uint8_t  Reserved1[4];
 631     grub_uint64_t EfiStartLBA;
 632     grub_uint64_t EfiBackupLBA;
 633     grub_uint64_t PartAreaStartLBA;
 634     grub_uint64_t PartAreaEndLBA;
 635     grub_uint8_t  DiskGuid[16];
 636     grub_uint64_t PartTblStartLBA;
 637     grub_uint32_t PartTblTotNum;
 638     grub_uint32_t PartTblEntryLen;
 639     grub_uint32_t PartTblCrc;
 640     grub_uint8_t  Reserved2[420];
 641 }ventoy_gpt_head;
 642
 643 typedef struct ventoy_gpt_part_tbl
 644 {
 645     grub_uint8_t  PartType[16];
 646     grub_uint8_t  PartGuid[16];
 647     grub_uint64_t StartLBA;
 648     grub_uint64_t LastLBA;
 649     grub_uint64_t Attr;
 650     grub_uint16_t Name[36];
 651 }ventoy_gpt_part_tbl;
 652
 653 typedef struct ventoy_gpt_info
 654 {
 655     ventoy_mbr_head MBR;
 656     ventoy_gpt_head Head;
 657     ventoy_gpt_part_tbl PartTbl[128];
 658 }ventoy_gpt_info;
 659 #pragma pack()
 660
 661 typedef struct ventoy_video_mode
 662 {
 663     grub_uint32_t width;
 664     grub_uint32_t height;
 665     grub_uint32_t bpp;
 666 }ventoy_video_mode;
 667
 668
 669
 670 typedef struct file_fullpath
 671 {
 672     char path[256];
 673 }file_fullpath;
 674
 675 typedef struct install_template
 676 {
 677     int pathlen;
 678     char isopath[256];
 679
 680     int autosel;
 681     int cursel;
 682     int templatenum;
 683     file_fullpath *templatepath;
 684
 685     struct install_template *next;
 686 }install_template;
 687
 688 typedef struct persistence_config
 689 {
 690     int pathlen;
 691     char isopath[256];
 692
 693     int autosel;
 694     int cursel;
 695     int backendnum;
 696     file_fullpath *backendpath;
 697
 698     struct persistence_config *next;
 699 }persistence_config;
 700
 701 #define vtoy_alias_image_file 0
 702 #define vtoy_alias_directory  1
 703
 704 typedef struct menu_alias
 705 {
 706     int type;
 707     int pathlen;
 708     char isopath[256];
 709     char alias[256];
 710
 711     struct menu_alias *next;
 712 }menu_alias;
 713
 714 #define vtoy_class_image_file  0
 715 #define vtoy_class_directory   1
 716
 717 typedef struct menu_class
 718 {
 719     int  type;
 720     int  patlen;
 721     char pattern[256];
 722     char class[64];
 723
 724     struct menu_class *next;
 725 }menu_class;
 726
 727 typedef struct injection_config
 728 {
 729     int pathlen;
 730     char isopath[256];
 731     char archive[256];
 732
 733     struct injection_config *next;
 734 }injection_config;
 735
 736 typedef struct auto_memdisk
 737 {
 738     int pathlen;
 739     char isopath[256];
 740
 741     struct auto_memdisk *next;
 742 }auto_memdisk;
 743
 744 extern int g_ventoy_menu_esc;
 745 extern int g_ventoy_suppress_esc;
 746 extern int g_ventoy_last_entry;
 747 extern int g_ventoy_memdisk_mode;
 748 extern int g_ventoy_iso_raw;
 749 extern int g_ventoy_iso_uefi_drv;
 750 extern int g_ventoy_case_insensitive;
 751 extern grub_uint8_t g_ventoy_chain_type;
 752 extern int g_vhdboot_enable;
 753 extern ventoy_gpt_info *g_ventoy_part_info;
 754
 755
 756 #define ventoy_unix_fill_virt(new_data, new_len) \
 757 { \
 758     data_secs = (new_len + 2047) / 2048; \
 759     cur->mem_sector_start   = sector; \
 760     cur->mem_sector_end     = cur->mem_sector_start + data_secs; \
 761     cur->mem_sector_offset  = offset; \
 762     cur->remap_sector_start = 0; \
 763     cur->remap_sector_end   = 0; \
 764     cur->org_sector_start   = 0; \
 765     grub_memcpy(override + offset, new_data, new_len); \
 766     cur++; \
 767     sector += data_secs; \
 768     offset += new_len; \
 769     chain->virt_img_size_in_bytes += data_secs * 2048; \
 770 }
 771
 772 char * ventoy_get_line(char *start);
 773 int ventoy_cmp_img(img_info *img1, img_info *img2);
 774 void ventoy_swap_img(img_info *img1, img_info *img2);
 775 char * ventoy_plugin_get_cur_install_template(const char *isopath);
 776 install_template * ventoy_plugin_find_install_template(const char *isopath);
 777 persistence_config * ventoy_plugin_find_persistent(const char *isopath);
 778 void ventoy_plugin_dump_injection(void);
 779 void ventoy_plugin_dump_auto_install(void);
 780 int ventoy_fill_windows_rtdata(void *buf, char *isopath);
 781 int ventoy_plugin_get_persistent_chunklist(const char *isopath, int index, ventoy_img_chunk_list *chunk_list);
 782 const char * ventoy_plugin_get_injection(const char *isopath);
 783 const char * ventoy_plugin_get_menu_alias(int type, const char *isopath);
 784 const char * ventoy_plugin_get_menu_class(int type, const char *name);
 785 int ventoy_plugin_check_memdisk(const char *isopath);
 786 int ventoy_get_block_list(grub_file_t file, ventoy_img_chunk_list *chunklist, grub_disk_addr_t start);
 787 int ventoy_check_block_list(grub_file_t file, ventoy_img_chunk_list *chunklist, grub_disk_addr_t start);
 788 void ventoy_plugin_dump_persistence(void);
 789 grub_err_t ventoy_cmd_plugin_check_json(grub_extcmd_context_t ctxt, int argc, char **args);
 790 grub_err_t ventoy_cmd_linux_get_main_initrd_index(grub_extcmd_context_t ctxt, int argc, char **args);
 791 grub_err_t ventoy_cmd_collect_wim_patch(grub_extcmd_context_t ctxt, int argc, char **args);
 792 grub_err_t ventoy_cmd_wim_patch_count(grub_extcmd_context_t ctxt, int argc, char **args);
 793 grub_err_t ventoy_cmd_locate_wim_patch(grub_extcmd_context_t ctxt, int argc, char **args);
 794 grub_err_t ventoy_cmd_unix_chain_data(grub_extcmd_context_t ctxt, int argc, char **args);
 795 int ventoy_get_disk_guid(const char *filename, grub_uint8_t *guid);
 796 grub_err_t ventoy_cmd_unix_reset(grub_extcmd_context_t ctxt, int argc, char **args);
 797 grub_err_t ventoy_cmd_unix_replace_conf(grub_extcmd_context_t ctxt, int argc, char **args);
 798 grub_err_t ventoy_cmd_unix_replace_ko(grub_extcmd_context_t ctxt, int argc, char **args);
 799 grub_err_t ventoy_cmd_unix_freebsd_ver(grub_extcmd_context_t ctxt, int argc, char **args);
 800 grub_err_t ventoy_cmd_parse_freenas_ver(grub_extcmd_context_t ctxt, int argc, char **args);
 801 int ventoy_check_device_result(int ret);
 802 int ventoy_check_device(grub_device_t dev);
 803 void ventoy_debug_dump_guid(const char *prefix, grub_uint8_t *guid);
 804 grub_err_t ventoy_cmd_load_vhdboot(grub_extcmd_context_t ctxt, int argc, char **args);
 805 grub_err_t ventoy_cmd_patch_vhdboot(grub_extcmd_context_t ctxt, int argc, char **args);
 806
 807 #endif /* __VENTOY_DEF_H__ */
 808