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_DIR_DEPTH  32
  25
  26 #define VTOY_MAX_SCRIPT_BUF    (4 * 1024 * 1024)
  27
  28 #define VTOY_PART_BUF_LEN  (128 * 1024)
  29
  30 #define VTOY_FILT_MIN_FILE_SIZE  32768
  31
  32 #define VTOY_LINUX_SYSTEMD_MENU_MAX_BUF    16384
  33
  34 #define VTOY_SIZE_1GB     1073741824
  35 #define VTOY_SIZE_1MB     (1024 * 1024)
  36 #define VTOY_SIZE_2MB     (2 * 1024 * 1024)
  37 #define VTOY_SIZE_4MB     (4 * 1024 * 1024)
  38 #define VTOY_SIZE_512KB   (512 * 1024)
  39 #define VTOY_SIZE_1KB     1024
  40 #define VTOY_SIZE_32KB    (32 * 1024)
  41 #define VTOY_SIZE_128KB   (128 * 1024)
  42
  43 #define JSON_SUCCESS    0
  44 #define JSON_FAILED     1
  45 #define JSON_NOT_FOUND  2
  46
  47 #define WINDATA_FLAG_TEMPLATE   1
  48
  49 #define ulong unsigned long
  50 #define ulonglong  unsigned long long
  51
  52 #define vtoy_to_upper(c) (((char)(c) >= 'a' && (char)(c) <= 'z') ? ((char)(c) - 'a' + 'A') : (char)(c))
  53
  54 #define VENTOY_CMD_RETURN(err)  grub_errno = (err); return (err)
  55 #define VENTOY_FILE_TYPE    (GRUB_FILE_TYPE_NO_DECOMPRESS | GRUB_FILE_TYPE_LINUX_INITRD)
  56
  57 #define ventoy_env_op1(op, a) grub_env_##op(a)
  58 #define ventoy_env_op2(op, a, b) grub_env_##op((a), (b))
  59
  60 #define ventoy_get_env(key)         ventoy_env_op1(get, key)
  61 #define ventoy_set_env(key, val)    ventoy_env_op2(set, key, val)
  62
  63 #define VTOY_WARNING  "!!!!!!!!!!!!! WARNING !!!!!!!!!!!!!"
  64
  65 #define VTOY_PLAT_I386_UEFI     0x49413332
  66 #define VTOY_PLAT_ARM64_UEFI    0x41413634
  67 #define VTOY_PLAT_X86_64_UEFI   0x55454649
  68 #define VTOY_PLAT_X86_LEGACY    0x42494f53
  69 #define VTOY_PLAT_MIPS_UEFI     0x4D495053
  70
  71 #define VTOY_COMM_CPIO  "ventoy.cpio"
  72 #if defined(__arm__) || defined(__aarch64__)
  73 #define VTOY_ARCH_CPIO  "ventoy_arm64.cpio"
  74 #elif defined(__mips__)
  75 #define VTOY_ARCH_CPIO  "ventoy_mips64.cpio"
  76 #else
  77 #define VTOY_ARCH_CPIO  "ventoy_x86.cpio"
  78 #endif
  79
  80 #define ventoy_varg_4(arg) arg[0], arg[1], arg[2], arg[3]
  81 #define ventoy_varg_8(arg) arg[0], arg[1], arg[2], arg[3], arg[4], arg[5], arg[6], arg[7]
  82
  83 #define VTOY_PWD_CORRUPTED(err) \
  84 {\
  85     grub_printf("\n\n Password corrupted, will reboot after 5 seconds.\n\n"); \
  86     grub_refresh(); \
  87     grub_sleep(5); \
  88     grub_exit(); \
  89     return (err);\
  90 }
  91
  92 #define VTOY_APPEND_NEWBUF(buf) \
  93 {\
  94     char *__c = buf;\
  95     while (*__c)\
  96     {\
  97         newbuf[pos++] = *__c;\
  98         __c++;\
  99     }\
 100 }
 101
 102 #define VTOY_SKIP_SPACE(s) \
 103     while (ventoy_isspace(*s)) \
 104     {\
 105         s++;\
 106     }
 107
 108 #define VTOY_SKIP_SPACE_NEXT(s, initial) \
 109     s += initial;\
 110     while (ventoy_isspace(*s)) \
 111     {\
 112         s++;\
 113     }
 114
 115 typedef enum VTOY_FILE_FLT
 116 {
 117     VTOY_FILE_FLT_ISO = 0, /* .iso */
 118     VTOY_FILE_FLT_WIM,     /* .wim */
 119     VTOY_FILE_FLT_EFI,     /* .efi */
 120     VTOY_FILE_FLT_IMG,     /* .img */
 121     VTOY_FILE_FLT_VHD,     /* .vhd(x) */
 122     VTOY_FILE_FLT_VTOY,    /* .vtoy */
 123
 124     VTOY_FILE_FLT_BUTT
 125 }VTOY_FILE_FLT;
 126
 127 #define FILE_FLT(type) (0 == g_vtoy_file_flt[VTOY_FILE_FLT_##type])
 128
 129 typedef struct ventoy_initrd_ctx
 130 {
 131     const char *path_prefix;
 132     const char *dir_prefix;
 133 }ventoy_initrd_ctx;
 134
 135 typedef struct cmd_para
 136 {
 137     const char *name;
 138     grub_extcmd_func_t func;
 139     grub_command_flags_t flags;
 140     const struct grub_arg_option *parser;
 141
 142     const char *summary;
 143     const char *description;
 144
 145     grub_extcmd_t cmd;
 146 }cmd_para;
 147
 148 #define ventoy_align_2k(value)  ((value + 2047) / 2048 * 2048)
 149 #define ventoy_align(value, align)  (((value) + ((align) - 1)) & (~((align) - 1)))
 150
 151 #pragma pack(1)
 152 typedef struct cpio_newc_header
 153 {
 154     char  c_magic[6];
 155     char  c_ino[8];
 156     char  c_mode[8];
 157     char  c_uid[8];
 158     char  c_gid[8];
 159     char  c_nlink[8];
 160     char  c_mtime[8];
 161     char  c_filesize[8];
 162     char  c_devmajor[8];
 163     char  c_devminor[8];
 164     char  c_rdevmajor[8];
 165     char  c_rdevminor[8];
 166     char  c_namesize[8];
 167     char  c_check[8];
 168 }cpio_newc_header;
 169 #pragma pack()
 170
 171
 172 #define cmd_raw_name ctxt->extcmd->cmd->name
 173 #define check_free(p, func) if (p) { func(p); p = NULL; }
 174 #define grub_check_free(p) if (p) { grub_free(p); p = NULL; }
 175
 176 typedef int (*grub_char_check_func)(int c);
 177 #define ventoy_is_decimal(str)  ventoy_string_check(str, grub_isdigit)
 178
 179 #define OFFSET_OF(TYPE, MEMBER) ((grub_size_t) &((TYPE *)0)->MEMBER)
 180
 181 #pragma pack(1)
 182 typedef struct ventoy_patch_vhd
 183 {
 184     grub_uint8_t  part_offset_or_guid[16];
 185     grub_uint32_t reserved1;
 186     grub_uint32_t part_type;
 187     grub_uint8_t  disk_signature_or_guid[16];
 188     grub_uint8_t  reserved2[16];
 189     grub_uint8_t  vhd_file_path[1];
 190 }ventoy_patch_vhd;
 191 #pragma pack()
 192
 193 // El Torito Boot Record Volume Descriptor
 194 #pragma pack(1)
 195 typedef struct eltorito_descriptor
 196 {
 197     grub_uint8_t type;
 198     grub_uint8_t id[5];
 199     grub_uint8_t version;
 200     grub_uint8_t system_id[32];
 201     grub_uint8_t reserved[32];
 202     grub_uint32_t sector;
 203 }eltorito_descriptor;
 204
 205 typedef struct ventoy_iso9660_override
 206 {
 207     grub_uint32_t first_sector;
 208     grub_uint32_t first_sector_be;
 209     grub_uint32_t size;
 210     grub_uint32_t size_be;
 211 }ventoy_iso9660_override;
 212
 213 typedef struct ventoy_udf_override
 214 {
 215     grub_uint32_t length;
 216     grub_uint32_t position;
 217 }ventoy_udf_override;
 218
 219 typedef struct ventoy_iso9660_vd
 220 {
 221     grub_uint8_t type;
 222     grub_uint8_t id[5];
 223     grub_uint8_t ver;
 224     grub_uint8_t res;
 225     char sys[32];
 226     char vol[32];
 227     grub_uint8_t res2[8];
 228     grub_uint32_t space;
 229 }ventoy_iso9660_vd;
 230
 231 /* https://wiki.osdev.org/El-Torito */
 232 typedef struct boot_info_table
 233 {
 234     grub_uint32_t bi_data0;
 235     grub_uint32_t bi_data1;
 236     grub_uint32_t bi_PrimaryVolumeDescriptor;
 237     grub_uint32_t bi_BootFileLocation;
 238     grub_uint32_t bi_BootFileLength;
 239     grub_uint32_t bi_Checksum;
 240     grub_uint8_t bi_Reserved[40];
 241 }boot_info_table;
 242
 243 #pragma pack()
 244
 245 #define img_type_start 0
 246 #define img_type_iso   0
 247 #define img_type_wim   1
 248 #define img_type_efi   2
 249 #define img_type_img   3
 250 #define img_type_vhd   4
 251 #define img_type_vtoy  5
 252 #define img_type_max   6
 253
 254 typedef struct img_info
 255 {
 256     int pathlen;
 257     char path[512];
 258     char name[256];
 259
 260     const char *alias;
 261     const char *tip1;
 262     const char *tip2;
 263     const char *class;
 264     const char *menu_prefix;
 265
 266     int id;
 267     int type;
 268     int plugin_list_index;
 269     grub_uint64_t size;
 270     int select;
 271     int unsupport;
 272
 273     void *parent;
 274
 275     struct img_info *next;
 276     struct img_info *prev;
 277 }img_info;
 278
 279 typedef struct img_iterator_node
 280 {
 281     struct img_iterator_node *next;
 282     img_info **tail;
 283     char dir[400];
 284     int dirlen;
 285     int level;
 286     int isocnt;
 287     int done;
 288     int select;
 289
 290     int plugin_list_index;
 291
 292     struct img_iterator_node *parent;
 293     struct img_iterator_node *firstchild;
 294
 295     void *firstiso;
 296 }img_iterator_node;
 297
 298
 299
 300 typedef struct initrd_info
 301 {
 302     char name[256];
 303
 304     grub_uint64_t offset;
 305     grub_uint64_t size;
 306
 307     grub_uint8_t  iso_type; // 0: iso9660  1:udf
 308     grub_uint32_t udf_start_block;
 309
 310     grub_uint64_t override_offset;
 311     grub_uint32_t override_length;
 312     char          override_data[32];
 313
 314     struct initrd_info *next;
 315     struct initrd_info *prev;
 316 }initrd_info;
 317
 318 extern initrd_info *g_initrd_img_list;
 319 extern initrd_info *g_initrd_img_tail;
 320 extern int g_initrd_img_count;
 321 extern int g_valid_initrd_count;
 322
 323 extern img_info *g_ventoy_img_list;
 324 extern int g_ventoy_img_count;
 325
 326 extern grub_uint8_t *g_ventoy_cpio_buf;
 327 extern grub_uint32_t g_ventoy_cpio_size;
 328 extern cpio_newc_header *g_ventoy_initrd_head;
 329 extern grub_uint8_t *g_ventoy_runtime_buf;
 330
 331 extern ventoy_guid  g_ventoy_guid;
 332
 333 extern ventoy_img_chunk_list g_img_chunk_list;
 334 extern ventoy_img_chunk_list g_wimiso_chunk_list;
 335 extern char *g_wimiso_path;
 336 extern grub_uint32_t g_wimiso_size;
 337 extern char g_arch_mode_suffix[64];
 338 extern const char *g_menu_prefix[img_type_max];
 339
 340 extern int g_ventoy_debug;
 341 void ventoy_debug(const char *fmt, ...);
 342 #define debug(fmt, args...) if (g_ventoy_debug) ventoy_debug("[VTOY]: "fmt, ##args)
 343
 344 #define vtoy_ssprintf(buf, pos, fmt, args...) \
 345     pos += grub_snprintf(buf + pos, VTOY_MAX_SCRIPT_BUF - pos, fmt, ##args)
 346
 347 #define vtoy_len_ssprintf(buf, pos, len, fmt, args...) \
 348     pos += grub_snprintf(buf + pos, len - pos, fmt, ##args)
 349
 350 #define browser_ssprintf(mbuf, fmt, args...) \
 351     (mbuf)->pos += grub_snprintf((mbuf)->buf + (mbuf)->pos, (mbuf)->max - (mbuf)->pos, fmt, ##args)
 352
 353 #define vtoy_dummy_menuentry(buf, pos, len, title, class) \
 354     vtoy_len_ssprintf(buf, pos, len, "menuentry \"%s\" --class=\"%s\" {\n  echo \"\"\n}\n", title, class)
 355
 356
 357 #define FLAG_HEADER_RESERVED          0x00000001
 358 #define FLAG_HEADER_COMPRESSION       0x00000002
 359 #define FLAG_HEADER_READONLY          0x00000004
 360 #define FLAG_HEADER_SPANNED           0x00000008
 361 #define FLAG_HEADER_RESOURCE_ONLY     0x00000010
 362 #define FLAG_HEADER_METADATA_ONLY     0x00000020
 363 #define FLAG_HEADER_WRITE_IN_PROGRESS 0x00000040
 364 #define FLAG_HEADER_RP_FIX            0x00000080 // reparse point fixup
 365 #define FLAG_HEADER_COMPRESS_RESERVED 0x00010000
 366 #define FLAG_HEADER_COMPRESS_XPRESS   0x00020000
 367 #define FLAG_HEADER_COMPRESS_LZX      0x00040000
 368 #define FLAG_HEADER_COMPRESS_LZMS     0x00080000
 369
 370 #define RESHDR_FLAG_FREE 0x01
 371 #define RESHDR_FLAG_METADATA 0x02
 372 #define RESHDR_FLAG_COMPRESSED 0x04
 373 #define RESHDR_FLAG_SPANNED 0x08
 374
 375 #pragma pack(1)
 376
 377 /* A WIM resource header */
 378 typedef struct wim_resource_header
 379 {
 380     grub_uint64_t size_in_wim:56; /* Compressed length */
 381     grub_uint64_t flags:8;        /* flags  */
 382     grub_uint64_t offset;         /* Offset */
 383     grub_uint64_t raw_size;       /* Uncompressed length */
 384 }wim_resource_header;
 385
 386 /* WIM resource header length mask */
 387 #define WIM_RESHDR_ZLEN_MASK 0x00ffffffffffffffULL
 388
 389 /* WIM resource header flags */
 390 typedef enum wim_resource_header_flags
 391 {
 392     WIM_RESHDR_METADATA = ( 0x02ULL << 56 ),       /* Resource contains metadata */
 393     WIM_RESHDR_COMPRESSED = ( 0x04ULL << 56 ),     /* Resource is compressed */
 394     WIM_RESHDR_PACKED_STREAMS = ( 0x10ULL << 56 ), /* Resource is compressed using packed streams */
 395 }wim_resource_header_flags;
 396
 397 #define WIM_HEAD_SIGNATURE   "MSWIM\0\0"
 398
 399 /* WIM header */
 400 typedef struct wim_header
 401 {
 402     grub_uint8_t signature[8];          /* Signature */
 403     grub_uint32_t header_len;           /* Header length */
 404     grub_uint32_t version;              /* Verson */
 405     grub_uint32_t flags;                /* Flags */
 406     grub_uint32_t chunk_len;            /* Chunk length */
 407     grub_uint8_t guid[16];              /* GUID */
 408     grub_uint16_t part;                 /* Part number */
 409     grub_uint16_t parts;                /* Total number of parts */
 410     grub_uint32_t images;               /* number of images */
 411     wim_resource_header lookup;    /* Lookup table */
 412     wim_resource_header xml;       /* XML data */
 413     wim_resource_header metadata;  /* Boot metadata */
 414     grub_uint32_t boot_index;           /* Boot index */
 415     wim_resource_header integrity; /* Integrity table */
 416     grub_uint8_t reserved[60];          /* Reserved */
 417 } wim_header;
 418
 419 /* WIM header flags */
 420 typedef enum wim_header_flags
 421 {
 422     WIM_HDR_XPRESS = 0x00020000, /* WIM uses Xpress compresson */
 423     WIM_HDR_LZX = 0x00040000,    /* WIM uses LZX compression */
 424 }wim_header_flags;
 425
 426 /* A WIM file hash */
 427 typedef struct wim_hash
 428 {
 429     /* SHA-1 hash */
 430     grub_uint8_t sha1[20];
 431 }wim_hash;
 432
 433 /* A WIM lookup table entry */
 434 typedef struct wim_lookup_entry
 435 {
 436     wim_resource_header resource; /* Resource header */
 437     grub_uint16_t part;           /* Part number */
 438     grub_uint32_t refcnt;         /* Reference count */
 439     wim_hash hash;                /* Hash */
 440 }wim_lookup_entry;
 441
 442 /* WIM chunk length */
 443 #define WIM_CHUNK_LEN 32768
 444
 445 /* A WIM chunk buffer */
 446 typedef struct wim_chunk_buffer
 447 {
 448     grub_uint8_t data[WIM_CHUNK_LEN]; /*Data */
 449 }wim_chunk_buffer;
 450
 451 /* Security data */
 452 typedef struct wim_security_header
 453 {
 454     grub_uint32_t len;   /* Length */
 455     grub_uint32_t count; /* Number of entries */
 456 }wim_security_header;
 457
 458 typedef struct wim_stream_entry
 459 {
 460     grub_uint64_t len;
 461     grub_uint64_t unused1;
 462     wim_hash hash;
 463     grub_uint16_t name_len;
 464     /* name */
 465 }wim_stream_entry;
 466
 467 /* Directory entry */
 468 typedef struct wim_directory_entry
 469 {
 470     grub_uint64_t len;                 /* Length */
 471     grub_uint32_t attributes;     /* Attributes */
 472     grub_uint32_t security;       /* Security ID */
 473     grub_uint64_t subdir;              /* Subdirectory offset */
 474     grub_uint8_t reserved1[16];   /* Reserved */
 475     grub_uint64_t created;             /* Creation time */
 476     grub_uint64_t accessed;            /* Last access time */
 477     grub_uint64_t written;             /* Last written time */
 478     wim_hash hash;                /* Hash */
 479     grub_uint8_t reserved2[12];   /* Reserved */
 480     grub_uint16_t streams;        /* Streams */
 481     grub_uint16_t short_name_len; /* Short name length */
 482     grub_uint16_t name_len;       /* Name length */
 483 }wim_directory_entry;
 484
 485 /** Normal file */
 486 #define WIM_ATTR_NORMAL 0x00000080UL
 487
 488 /** No security information exists for this file */
 489 #define WIM_NO_SECURITY 0xffffffffUL
 490
 491 typedef struct reg_vk
 492 {
 493     grub_uint32_t res1;
 494     grub_uint16_t sig;
 495     grub_uint16_t namesize;
 496     grub_uint32_t datasize;
 497     grub_uint32_t dataoffset;
 498     grub_uint32_t datatype;
 499     grub_uint16_t flag;
 500     grub_uint16_t res2;
 501 }reg_vk;
 502
 503 #pragma pack()
 504
 505
 506 typedef struct wim_tail
 507 {
 508     grub_uint32_t wim_raw_size;
 509     grub_uint32_t wim_align_size;
 510
 511     grub_uint8_t  iso_type;
 512     grub_uint64_t file_offset;
 513     grub_uint32_t udf_start_block;
 514     grub_uint64_t fe_entry_size_offset;
 515     grub_uint64_t override_offset;
 516     grub_uint32_t override_len;
 517     grub_uint8_t  override_data[32];
 518
 519     wim_header wim_header;
 520
 521     wim_hash bin_hash;
 522     grub_uint32_t jump_exe_len;
 523     grub_uint8_t *jump_bin_data;
 524     grub_uint32_t bin_raw_len;
 525     grub_uint32_t bin_align_len;
 526     grub_uint32_t windata_flag;
 527
 528     grub_uint8_t *new_meta_data;
 529     grub_uint32_t new_meta_len;
 530     grub_uint32_t new_meta_align_len;
 531
 532     grub_uint8_t *new_lookup_data;
 533     grub_uint32_t new_lookup_len;
 534     grub_uint32_t new_lookup_align_len;
 535 }wim_tail;
 536
 537 typedef struct wim_patch
 538 {
 539     int pathlen;
 540     char path[256];
 541
 542     wim_hash old_hash;
 543     wim_tail wim_data;
 544     wim_lookup_entry *replace_look;
 545
 546     int valid;
 547
 548     struct wim_patch *next;
 549 }wim_patch;
 550
 551
 552 typedef enum _JSON_TYPE
 553 {
 554     JSON_TYPE_NUMBER = 0,
 555     JSON_TYPE_STRING,
 556     JSON_TYPE_BOOL,
 557     JSON_TYPE_ARRAY,
 558     JSON_TYPE_OBJECT,
 559     JSON_TYPE_NULL,
 560     JSON_TYPE_BUTT
 561 }JSON_TYPE;
 562
 563
 564 typedef struct _VTOY_JSON
 565 {
 566     struct _VTOY_JSON *pstPrev;
 567     struct _VTOY_JSON *pstNext;
 568     struct _VTOY_JSON *pstChild;
 569
 570     JSON_TYPE enDataType;
 571     union
 572     {
 573         char  *pcStrVal;
 574         int   iNumVal;
 575         grub_uint64_t lValue;
 576     }unData;
 577
 578     char *pcName;
 579 }VTOY_JSON;
 580
 581 typedef struct _JSON_PARSE
 582 {
 583     char *pcKey;
 584     void *pDataBuf;
 585     grub_uint32_t  uiBufSize;
 586 }JSON_PARSE;
 587
 588 #define JSON_NEW_ITEM(pstJson, ret) \
 589 { \
 590     (pstJson) = (VTOY_JSON *)grub_zalloc(sizeof(VTOY_JSON)); \
 591     if (NULL == (pstJson)) \
 592     { \
 593         json_debug("Failed to alloc memory for json.\n"); \
 594         return (ret); \
 595     } \
 596 }
 597
 598 typedef int (*ventoy_plugin_entry_pf)(VTOY_JSON *json, const char *isodisk);
 599 typedef int (*ventoy_plugin_check_pf)(VTOY_JSON *json, const char *isodisk);
 600
 601 typedef struct plugin_entry
 602 {
 603     const char *key;
 604     ventoy_plugin_entry_pf entryfunc;
 605     ventoy_plugin_check_pf checkfunc;
 606     int flag;
 607 }plugin_entry;
 608
 609 typedef struct replace_fs_dir
 610 {
 611     grub_device_t dev;
 612     grub_fs_t fs;
 613     char fullpath[512];
 614     char initrd[512];
 615     int curpos;
 616     int dircnt;
 617     int filecnt;
 618 }replace_fs_dir;
 619
 620 typedef struct chk_case_fs_dir
 621 {
 622     grub_device_t dev;
 623     grub_fs_t fs;
 624 }chk_case_fs_dir;
 625
 626 int ventoy_str_all_digit(const char *str);
 627 int ventoy_strcmp(const char *pattern, const char *str);
 628 int ventoy_strncmp (const char *pattern, const char *str, grub_size_t n);
 629 void ventoy_fill_os_param(grub_file_t file, ventoy_os_param *param);
 630 grub_err_t ventoy_cmd_isolinux_initrd_collect(grub_extcmd_context_t ctxt, int argc, char **args);
 631 grub_err_t ventoy_cmd_grub_initrd_collect(grub_extcmd_context_t ctxt, int argc, char **args);
 632 grub_err_t ventoy_cmd_specify_initrd_file(grub_extcmd_context_t ctxt, int argc, char **args);
 633 grub_err_t ventoy_cmd_dump_initrd_list(grub_extcmd_context_t ctxt, int argc, char **args);
 634 grub_err_t ventoy_cmd_clear_initrd_list(grub_extcmd_context_t ctxt, int argc, char **args);
 635 grub_uint32_t ventoy_get_iso_boot_catlog(grub_file_t file);
 636 int ventoy_has_efi_eltorito(grub_file_t file, grub_uint32_t sector);
 637 grub_err_t ventoy_cmd_linux_chain_data(grub_extcmd_context_t ctxt, int argc, char **args);
 638 grub_err_t ventoy_cmd_linux_systemd_menu(grub_extcmd_context_t ctxt, int argc, char **args);
 639 grub_err_t ventoy_cmd_linux_limine_menu(grub_extcmd_context_t ctxt, int argc, char **args);
 640 grub_err_t ventoy_cmd_linux_locate_initrd(grub_extcmd_context_t ctxt, int argc, char **args);
 641 grub_err_t ventoy_cmd_initrd_count(grub_extcmd_context_t ctxt, int argc, char **args);
 642 grub_err_t ventoy_cmd_valid_initrd_count(grub_extcmd_context_t ctxt, int argc, char **args);
 643 grub_err_t ventoy_cmd_load_cpio(grub_extcmd_context_t ctxt, int argc, char **args);
 644 grub_err_t ventoy_cmd_append_ext_sector(grub_extcmd_context_t ctxt, int argc, char **args);
 645 grub_err_t ventoy_cmd_skip_svd(grub_extcmd_context_t ctxt, int argc, char **args);
 646 grub_err_t ventoy_cmd_cpio_busybox_64(grub_extcmd_context_t ctxt, int argc, char **args);
 647 grub_err_t ventoy_cmd_trailer_cpio(grub_extcmd_context_t ctxt, int argc, char **args);
 648 int ventoy_cpio_newc_fill_head(void *buf, int filesize, const void *filedata, const char *name);
 649 grub_file_t ventoy_grub_file_open(enum grub_file_type type, const char *fmt, ...);
 650 grub_uint64_t ventoy_grub_get_file_size(const char *fmt, ...);
 651 int ventoy_is_dir_exist(const char *fmt, ...);
 652 int ventoy_fill_data(grub_uint32_t buflen, char *buffer);
 653 grub_err_t ventoy_cmd_load_plugin(grub_extcmd_context_t ctxt, int argc, char **args);
 654 grub_err_t ventoy_cmd_wimdows_reset(grub_extcmd_context_t ctxt, int argc, char **args);
 655 grub_err_t ventoy_cmd_is_pe64(grub_extcmd_context_t ctxt, int argc, char **args);
 656 grub_err_t ventoy_cmd_windows_chain_data(grub_extcmd_context_t ctxt, int argc, char **args);
 657 grub_err_t ventoy_cmd_windows_wimboot_data(grub_extcmd_context_t ctxt, int argc, char **args);
 658 grub_err_t ventoy_cmd_wim_chain_data(grub_extcmd_context_t ctxt, int argc, char **args);
 659 grub_err_t ventoy_cmd_wim_check_bootable(grub_extcmd_context_t ctxt, int argc, char **args);
 660 grub_err_t ventoy_cmd_is_standard_winiso(grub_extcmd_context_t ctxt, int argc, char **args);
 661 grub_err_t ventoy_cmd_dump_wim_patch(grub_extcmd_context_t ctxt, int argc, char **args);
 662 grub_err_t ventoy_cmd_sel_wimboot(grub_extcmd_context_t ctxt, int argc, char **args);
 663 grub_err_t ventoy_cmd_set_wim_prompt(grub_extcmd_context_t ctxt, int argc, char **args);
 664 grub_ssize_t ventoy_load_file_with_prompt(grub_file_t file, void *buf, grub_ssize_t size);
 665 int ventoy_need_prompt_load_file(void);
 666
 667 VTOY_JSON *vtoy_json_find_item
 668 (
 669     VTOY_JSON *pstJson,
 670     JSON_TYPE  enDataType,
 671     const char *szKey
 672 );
 673 int vtoy_json_parse_value
 674 (
 675     char *pcNewStart,
 676     char *pcRawStart,
 677     VTOY_JSON *pstJson,
 678     const char *pcData,
 679     const char **ppcEnd
 680 );
 681 VTOY_JSON * vtoy_json_create(void);
 682 int vtoy_json_parse(VTOY_JSON *pstJson, const char *szJsonData);
 683
 684 int vtoy_json_scan_parse
 685 (
 686     const VTOY_JSON    *pstJson,
 687     grub_uint32_t       uiParseNum,
 688     JSON_PARSE         *pstJsonParse
 689 );
 690
 691 int vtoy_json_scan_array
 692 (
 693      VTOY_JSON *pstJson,
 694      const char *szKey,
 695      VTOY_JSON **ppstArrayItem
 696 );
 697
 698 int vtoy_json_scan_array_ex
 699 (
 700      VTOY_JSON *pstJson,
 701      const char *szKey,
 702      VTOY_JSON **ppstArrayItem
 703 );
 704 int vtoy_json_scan_object
 705 (
 706      VTOY_JSON *pstJson,
 707      const char *szKey,
 708     VTOY_JSON **ppstObjectItem
 709 );
 710 int vtoy_json_get_int
 711 (
 712     VTOY_JSON *pstJson,
 713     const char *szKey,
 714     int *piValue
 715 );
 716 int vtoy_json_get_uint
 717 (
 718     VTOY_JSON *pstJson,
 719     const char *szKey,
 720     grub_uint32_t *puiValue
 721 );
 722 int vtoy_json_get_uint64
 723 (
 724     VTOY_JSON *pstJson,
 725     const char *szKey,
 726     grub_uint64_t *pui64Value
 727 );
 728 int vtoy_json_get_bool
 729 (
 730     VTOY_JSON *pstJson,
 731     const char *szKey,
 732     grub_uint8_t *pbValue
 733 );
 734 int vtoy_json_get_string
 735 (
 736      VTOY_JSON *pstJson,
 737      const char *szKey,
 738      grub_uint32_t  uiBufLen,
 739      char *pcBuf
 740 );
 741 const char * vtoy_json_get_string_ex(VTOY_JSON *pstJson,  const char *szKey);
 742 int vtoy_json_destroy(VTOY_JSON *pstJson);
 743
 744
 745 grub_uint32_t CalculateCrc32
 746 (
 747     const void       *Buffer,
 748     grub_uint32_t          Length,
 749     grub_uint32_t          InitValue
 750 );
 751
 752 static inline int ventoy_isspace (int c)
 753 {
 754     return (c == '\n' || c == '\r' || c == ' ' || c == '\t');
 755 }
 756
 757 static inline int ventoy_is_word_end(int c)
 758 {
 759     return (c == 0 || c == ',' || ventoy_isspace(c));
 760 }
 761
 762 #pragma pack(1)
 763 typedef struct ventoy_part_table
 764 {
 765     grub_uint8_t  Active; // 0x00  0x80
 766
 767     grub_uint8_t  StartHead;
 768     grub_uint16_t StartSector : 6;
 769     grub_uint16_t StartCylinder : 10;
 770
 771     grub_uint8_t  FsFlag;
 772
 773     grub_uint8_t  EndHead;
 774     grub_uint16_t EndSector : 6;
 775     grub_uint16_t EndCylinder : 10;
 776
 777     grub_uint32_t StartSectorId;
 778     grub_uint32_t SectorCount;
 779 }ventoy_part_table;
 780
 781 typedef struct ventoy_mbr_head
 782 {
 783     grub_uint8_t BootCode[446];
 784     ventoy_part_table PartTbl[4];
 785     grub_uint8_t Byte55;
 786     grub_uint8_t ByteAA;
 787 }ventoy_mbr_head;
 788
 789 typedef struct ventoy_gpt_head
 790 {
 791     char   Signature[8]; /* EFI PART */
 792     grub_uint8_t  Version[4];
 793     grub_uint32_t Length;
 794     grub_uint32_t Crc;
 795     grub_uint8_t  Reserved1[4];
 796     grub_uint64_t EfiStartLBA;
 797     grub_uint64_t EfiBackupLBA;
 798     grub_uint64_t PartAreaStartLBA;
 799     grub_uint64_t PartAreaEndLBA;
 800     grub_uint8_t  DiskGuid[16];
 801     grub_uint64_t PartTblStartLBA;
 802     grub_uint32_t PartTblTotNum;
 803     grub_uint32_t PartTblEntryLen;
 804     grub_uint32_t PartTblCrc;
 805     grub_uint8_t  Reserved2[420];
 806 }ventoy_gpt_head;
 807
 808 typedef struct ventoy_gpt_part_tbl
 809 {
 810     grub_uint8_t  PartType[16];
 811     grub_uint8_t  PartGuid[16];
 812     grub_uint64_t StartLBA;
 813     grub_uint64_t LastLBA;
 814     grub_uint64_t Attr;
 815     grub_uint16_t Name[36];
 816 }ventoy_gpt_part_tbl;
 817
 818 typedef struct ventoy_gpt_info
 819 {
 820     ventoy_mbr_head MBR;
 821     ventoy_gpt_head Head;
 822     ventoy_gpt_part_tbl PartTbl[128];
 823 }ventoy_gpt_info;
 824
 825 typedef struct vhd_footer_t
 826 {
 827     char             cookie[8];    // Cookie
 828     grub_uint32_t    features;     // Features
 829     grub_uint32_t    ffversion;    // File format version
 830     grub_uint32_t    dataoffset;   // Data offset
 831     grub_uint32_t    timestamp;    // Timestamp
 832     grub_uint32_t    creatorapp;   // Creator application
 833     grub_uint32_t    creatorver;   // Creator version
 834     grub_uint32_t    creatorhos;   // Creator host OS
 835     grub_uint32_t    origsize;     // Original size
 836     grub_uint32_t    currsize;     // Current size
 837     grub_uint32_t    diskgeom;     // Disk geometry
 838     grub_uint32_t    disktype;     // Disk type
 839     grub_uint32_t    checksum;     // Checksum
 840     grub_uint8_t     uniqueid[16]; // Unique ID
 841     grub_uint8_t     savedst;      // Saved state
 842 }vhd_footer_t;
 843
 844 #define VDI_IMAGE_FILE_INFO   "<<< Oracle VM VirtualBox Disk Image >>>\n"
 845
 846 /** Image signature. */
 847 #define VDI_IMAGE_SIGNATURE   (0xbeda107f)
 848
 849 typedef struct VDIPREHEADER
 850 {
 851     /** Just text info about image type, for eyes only. */
 852     char            szFileInfo[64];
 853     /** The image signature (VDI_IMAGE_SIGNATURE). */
 854     grub_uint32_t   u32Signature;
 855     /** The image version (VDI_IMAGE_VERSION). */
 856     grub_uint32_t   u32Version;
 857 } VDIPREHEADER, *PVDIPREHEADER;
 858
 859 #pragma pack()
 860
 861 typedef struct ventoy_video_mode
 862 {
 863     grub_uint32_t width;
 864     grub_uint32_t height;
 865     grub_uint32_t bpp;
 866 }ventoy_video_mode;
 867
 868
 869
 870 typedef struct file_fullpath
 871 {
 872     char path[256];
 873     int vlnk_add;
 874 }file_fullpath;
 875
 876 typedef struct theme_list
 877 {
 878     file_fullpath theme;
 879     struct theme_list *next;
 880 }theme_list;
 881
 882 #define auto_install_type_file   0
 883 #define auto_install_type_parent 1
 884 typedef struct install_template
 885 {
 886     int type;
 887     int pathlen;
 888     char isopath[256];
 889
 890     int timeout;
 891     int autosel;
 892     int cursel;
 893     int templatenum;
 894     file_fullpath *templatepath;
 895
 896     char *filebuf;
 897     int filelen;
 898
 899     struct install_template *next;
 900 }install_template;
 901
 902 typedef struct dudfile
 903 {
 904     int size;
 905     char *buf;
 906 }dudfile;
 907
 908 typedef struct dud
 909 {
 910     int pathlen;
 911     char isopath[256];
 912
 913     int dudnum;
 914     file_fullpath *dudpath;
 915     dudfile *files;
 916
 917     struct dud *next;
 918 }dud;
 919
 920 typedef struct persistence_config
 921 {
 922     int pathlen;
 923     char isopath[256];
 924
 925     int timeout;
 926     int autosel;
 927     int cursel;
 928     int backendnum;
 929     file_fullpath *backendpath;
 930
 931     struct persistence_config *next;
 932 }persistence_config;
 933
 934 #define vtoy_alias_image_file 0
 935 #define vtoy_alias_directory  1
 936
 937 typedef struct menu_alias
 938 {
 939     int type;
 940     int pathlen;
 941     char isopath[256];
 942     char alias[256];
 943
 944     struct menu_alias *next;
 945 }menu_alias;
 946
 947 #define vtoy_tip_image_file 0
 948 #define vtoy_tip_directory  1
 949 typedef struct menu_tip
 950 {
 951     int type;
 952     int pathlen;
 953     char isopath[256];
 954     char tip1[1024];
 955     char tip2[1024];
 956
 957     struct menu_tip *next;
 958 }menu_tip;
 959
 960
 961 #define vtoy_class_image_file  0
 962 #define vtoy_class_directory   1
 963
 964 typedef struct menu_class
 965 {
 966     int  type;
 967     int  patlen;
 968     int  parent;
 969     char pattern[256];
 970     char class[64];
 971
 972     struct menu_class *next;
 973 }menu_class;
 974
 975 #define vtoy_custom_boot_image_file  0
 976 #define vtoy_custom_boot_directory   1
 977
 978 typedef struct custom_boot
 979 {
 980     int  type;
 981     int  pathlen;
 982     char path[256];
 983     char cfg[256];
 984
 985     struct custom_boot *next;
 986 }custom_boot;
 987
 988 #define vtoy_max_replace_file_size  (1024 * 1024)
 989 typedef struct conf_replace
 990 {
 991     int pathlen;
 992     int img;
 993     char isopath[256];
 994     char orgconf[256];
 995     char newconf[256];
 996
 997     struct conf_replace *next;
 998 }conf_replace;
 999
1000 #define injection_type_file   0
1001 #define injection_type_parent 1
1002 typedef struct injection_config
1003 {
1004     int type;
1005     int pathlen;
1006     char isopath[256];
1007     char archive[256];
1008
1009     struct injection_config *next;
1010 }injection_config;
1011
1012 typedef struct auto_memdisk
1013 {
1014     int pathlen;
1015     char isopath[256];
1016
1017     struct auto_memdisk *next;
1018 }auto_memdisk;
1019
1020 typedef struct image_list
1021 {
1022     int pathlen;
1023     char isopath[256];
1024
1025     struct image_list *next;
1026 }image_list;
1027
1028 #define VTOY_PASSWORD_NONE       0
1029 #define VTOY_PASSWORD_TXT        1
1030 #define VTOY_PASSWORD_MD5        2
1031 #define VTOY_PASSWORD_SALT_MD5   3
1032
1033 typedef struct vtoy_password
1034 {
1035     int type;
1036     char text[128];
1037     char salt[64];
1038     grub_uint8_t md5[16];
1039 }vtoy_password;
1040
1041 #define vtoy_menu_pwd_file   0
1042 #define vtoy_menu_pwd_parent 1
1043
1044 typedef struct menu_password
1045 {
1046     int type;
1047     int pathlen;
1048     char isopath[256];
1049
1050     vtoy_password password;
1051
1052     struct menu_password *next;
1053 }menu_password;
1054
1055 extern int g_ventoy_menu_esc;
1056 extern int g_ventoy_secondary_menu_on;
1057 extern int g_ventoy_suppress_esc;
1058 extern int g_ventoy_suppress_esc_default;
1059 extern int g_ventoy_last_entry;
1060 extern int g_ventoy_memdisk_mode;
1061 extern int g_ventoy_iso_raw;
1062 extern int g_ventoy_grub2_mode;
1063 extern int g_ventoy_wimboot_mode;
1064 extern int g_ventoy_iso_uefi_drv;
1065 extern int g_ventoy_case_insensitive;
1066 extern int g_ventoy_fn_mutex;
1067 extern grub_uint8_t g_ventoy_chain_type;
1068 extern int g_vhdboot_enable;
1069 extern int g_default_menu_mode;
1070 extern char g_ventoy_hotkey_tip[256];
1071 extern int g_ventoy_menu_refresh;
1072
1073 #define VENTOY_IMG_WHITE_LIST   1
1074 #define VENTOY_IMG_BLACK_LIST   2
1075 extern int g_plugin_image_list;
1076
1077 extern ventoy_gpt_info *g_ventoy_part_info;
1078 extern int g_conf_replace_count;
1079 extern grub_uint64_t g_conf_replace_offset[VTOY_MAX_CONF_REPLACE];
1080 extern grub_uint64_t g_svd_replace_offset;
1081 extern conf_replace *g_conf_replace_node[VTOY_MAX_CONF_REPLACE];
1082 extern grub_uint8_t *g_conf_replace_new_buf[VTOY_MAX_CONF_REPLACE];
1083 extern int g_conf_replace_new_len[VTOY_MAX_CONF_REPLACE];
1084 extern int g_conf_replace_new_len_align[VTOY_MAX_CONF_REPLACE];
1085 extern int g_ventoy_disk_bios_id;
1086 extern grub_uint64_t g_ventoy_disk_size;
1087 extern grub_uint64_t g_ventoy_disk_part_size[2];
1088 extern grub_uint32_t g_ventoy_plat_data;
1089
1090 #define ventoy_unix_fill_virt(new_data, new_len) \
1091 { \
1092     data_secs = (new_len + 2047) / 2048; \
1093     cur->mem_sector_start   = sector; \
1094     cur->mem_sector_end     = cur->mem_sector_start + data_secs; \
1095     cur->mem_sector_offset  = offset; \
1096     cur->remap_sector_start = 0; \
1097     cur->remap_sector_end   = 0; \
1098     cur->org_sector_start   = 0; \
1099     grub_memcpy(override + offset, new_data, new_len); \
1100     cur++; \
1101     sector += data_secs; \
1102     offset += new_len; \
1103     chain->virt_img_size_in_bytes += data_secs * 2048; \
1104 }
1105
1106 #define ventoy_syscall0(name) grub_##name()
1107 #define ventoy_syscall1(name, a) grub_##name(a)
1108
1109 void ventoy_str_tolower(char *str);
1110 void ventoy_str_toupper(char *str);
1111 char * ventoy_get_line(char *start);
1112 char *ventoy_str_last(char *str, char ch);
1113 int ventoy_cmp_img(img_info *img1, img_info *img2);
1114 void ventoy_swap_img(img_info *img1, img_info *img2);
1115 char * ventoy_plugin_get_cur_install_template(const char *isopath, install_template **cur);
1116 install_template * ventoy_plugin_find_install_template(const char *isopath);
1117 persistence_config * ventoy_plugin_find_persistent(const char *isopath);
1118 grub_uint64_t ventoy_get_vtoy_partsize(int part);
1119 void ventoy_plugin_dump_injection(void);
1120 void ventoy_plugin_dump_auto_install(void);
1121 int ventoy_plugin_get_persistent_chunklist(const char *isopath, int index, ventoy_img_chunk_list *chunk_list);
1122 const char * ventoy_plugin_get_injection(const char *isopath);
1123 const char * ventoy_plugin_get_menu_alias(int type, const char *isopath);
1124 const menu_tip * ventoy_plugin_get_menu_tip(int type, const char *isopath);
1125 const char * ventoy_plugin_get_menu_class(int type, const char *name, const char *path);
1126 int ventoy_plugin_check_memdisk(const char *isopath);
1127 int ventoy_plugin_get_image_list_index(int type, const char *name);
1128 int ventoy_plugin_find_conf_replace(const char *iso, conf_replace *nodes[VTOY_MAX_CONF_REPLACE]);
1129 dud * ventoy_plugin_find_dud(const char *iso);
1130 int ventoy_plugin_load_dud(dud *node, const char *isopart);
1131 int ventoy_get_block_list(grub_file_t file, ventoy_img_chunk_list *chunklist, grub_disk_addr_t start);
1132 int ventoy_check_block_list(grub_file_t file, ventoy_img_chunk_list *chunklist, grub_disk_addr_t start);
1133 void ventoy_plugin_dump_persistence(void);
1134 grub_err_t ventoy_cmd_set_theme(grub_extcmd_context_t ctxt, int argc, char **args);
1135 grub_err_t ventoy_cmd_set_theme_path(grub_extcmd_context_t ctxt, int argc, char **args);
1136 grub_err_t ventoy_cmd_select_theme_cfg(grub_extcmd_context_t ctxt, int argc, char **args);
1137 grub_err_t ventoy_cmd_plugin_check_json(grub_extcmd_context_t ctxt, int argc, char **args);
1138 grub_err_t ventoy_cmd_check_password(grub_extcmd_context_t ctxt, int argc, char **args);
1139 grub_err_t ventoy_cmd_linux_get_main_initrd_index(grub_extcmd_context_t ctxt, int argc, char **args);
1140 grub_err_t ventoy_cmd_collect_wim_patch(grub_extcmd_context_t ctxt, int argc, char **args);
1141 grub_err_t ventoy_cmd_wim_patch_count(grub_extcmd_context_t ctxt, int argc, char **args);
1142 grub_err_t ventoy_cmd_locate_wim_patch(grub_extcmd_context_t ctxt, int argc, char **args);
1143 grub_err_t ventoy_cmd_sel_winpe_wim(grub_extcmd_context_t ctxt, int argc, char **args);
1144 grub_err_t ventoy_cmd_unix_chain_data(grub_extcmd_context_t ctxt, int argc, char **args);
1145 int ventoy_get_disk_guid(const char *filename, grub_uint8_t *guid, grub_uint8_t *signature);
1146 grub_err_t ventoy_cmd_unix_reset(grub_extcmd_context_t ctxt, int argc, char **args);
1147 grub_err_t ventoy_cmd_unix_check_vlnk(grub_extcmd_context_t ctxt, int argc, char **args);
1148 grub_err_t ventoy_cmd_unix_replace_conf(grub_extcmd_context_t ctxt, int argc, char **args);
1149 grub_err_t ventoy_cmd_unix_replace_grub_conf(grub_extcmd_context_t ctxt, int argc, char **args);
1150 grub_err_t ventoy_cmd_unix_replace_ko(grub_extcmd_context_t ctxt, int argc, char **args);
1151 grub_err_t ventoy_cmd_unix_ko_fillmap(grub_extcmd_context_t ctxt, int argc, char **args);
1152 grub_err_t ventoy_cmd_unix_fill_image_desc(grub_extcmd_context_t ctxt, int argc, char **args);
1153 grub_err_t ventoy_cmd_unix_gzip_newko(grub_extcmd_context_t ctxt, int argc, char **args);
1154 grub_err_t ventoy_cmd_unix_freebsd_ver(grub_extcmd_context_t ctxt, int argc, char **args);
1155 grub_err_t ventoy_cmd_parse_freenas_ver(grub_extcmd_context_t ctxt, int argc, char **args);
1156 grub_err_t ventoy_cmd_unix_freebsd_ver_elf(grub_extcmd_context_t ctxt, int argc, char **args);
1157 int ventoy_check_device_result(int ret);
1158 int ventoy_check_device(grub_device_t dev);
1159 void ventoy_debug_dump_guid(const char *prefix, grub_uint8_t *guid);
1160 grub_err_t ventoy_cmd_load_vhdboot(grub_extcmd_context_t ctxt, int argc, char **args);
1161 grub_err_t ventoy_cmd_patch_vhdboot(grub_extcmd_context_t ctxt, int argc, char **args);
1162 grub_err_t ventoy_cmd_raw_chain_data(grub_extcmd_context_t ctxt, int argc, char **args);
1163 grub_err_t ventoy_cmd_get_vtoy_type(grub_extcmd_context_t ctxt, int argc, char **args);
1164 int ventoy_check_password(const vtoy_password *pwd, int retry);
1165 int ventoy_plugin_add_custom_boot(const char *vcfgpath);
1166 const char * ventoy_plugin_get_custom_boot(const char *isopath);
1167 grub_err_t ventoy_cmd_dump_custom_boot(grub_extcmd_context_t ctxt, int argc, char **args);
1168 int ventoy_gzip_compress(void *mem_in, int mem_in_len, void *mem_out, int mem_out_len);
1169 int ventoy_load_part_table(const char *diskname);
1170 int ventoy_env_init(void);
1171 int ventoy_register_all_cmd(void);
1172 int ventoy_unregister_all_cmd(void);
1173 int ventoy_chain_file_size(const char *path);
1174 int ventoy_chain_file_read(const char *path, int offset, int len, void *buf);
1175
1176 #define VTOY_CMD_CHECK(a) if (33554432 != g_ventoy_disk_part_size[a]) ventoy_syscall0(exit)
1177
1178 #define vtoy_theme_random_boot_second  0
1179 #define vtoy_theme_random_boot_day     1
1180 #define vtoy_theme_random_boot_month   2
1181
1182 #define ventoy_env_export(env, name) \
1183 {\
1184     grub_env_set((env), (name));\
1185     grub_env_export(env);\
1186 }
1187
1188 #define ret_goto_end(a) ret = a; goto end;
1189
1190 extern ventoy_grub_param *g_grub_param;
1191
1192 #pragma pack(1)
1193 #define VENTOY_UNIX_SEG_MAGIC0    0x11223344
1194 #define VENTOY_UNIX_SEG_MAGIC1    0x55667788
1195 #define VENTOY_UNIX_SEG_MAGIC2    0x99aabbcc
1196 #define VENTOY_UNIX_SEG_MAGIC3    0xddeeff00
1197 #define VENTOY_UNIX_MAX_SEGNUM    40960
1198 struct g_ventoy_seg {
1199     grub_uint64_t seg_start_bytes;
1200     grub_uint64_t seg_end_bytes;
1201 };
1202
1203 struct g_ventoy_map{
1204     grub_uint32_t magic1[4];
1205     grub_uint32_t magic2[4];
1206     grub_uint64_t segnum;
1207     grub_uint64_t disksize;
1208     grub_uint8_t diskuuid[16];
1209     struct g_ventoy_seg seglist[VENTOY_UNIX_MAX_SEGNUM];
1210     grub_uint32_t magic3[4];
1211 };
1212 #pragma pack()
1213
1214 typedef struct ventoy_vlnk_part
1215 {
1216     grub_uint32_t disksig;
1217     grub_uint64_t partoffset;
1218     char disk[64];
1219     char device[64];
1220     grub_device_t dev;
1221     grub_fs_t fs;
1222     int probe;
1223     struct ventoy_vlnk_part *next;
1224 }ventoy_vlnk_part;
1225
1226
1227 typedef struct browser_mbuf
1228 {
1229     int max;
1230     int pos;
1231     char *buf;
1232 }browser_mbuf;
1233
1234 typedef struct browser_node
1235 {
1236     int  dir;
1237     char menuentry[1024];
1238     char filename[512];
1239     struct browser_node *prev;
1240     struct browser_node *next;
1241 }browser_node;
1242
1243 typedef struct var_node
1244 {
1245     char var[128];
1246     char val[256];
1247
1248     struct var_node *next;
1249 }var_node;
1250
1251 typedef struct systemd_menu_ctx
1252 {
1253     char *dev;
1254     char *buf;
1255     int pos;
1256     int len;
1257 }systemd_menu_ctx;
1258
1259 typedef struct ctrl_var_cfg
1260 {
1261     const char *name;
1262     int value;
1263 }ctrl_var_cfg;
1264
1265 #define vtoy_check_goto_out(p)  if (!p) goto out
1266
1267 extern char *g_tree_script_buf;
1268 extern int g_tree_script_pos;
1269 extern int g_tree_script_pre;
1270 extern int g_tree_view_menu_style;
1271 extern int g_sort_case_sensitive;
1272 extern int g_wimboot_enable;
1273 extern int g_filt_dot_underscore_file;
1274 extern int g_vtoy_file_flt[VTOY_FILE_FLT_BUTT];
1275 extern const char *g_menu_class[img_type_max];
1276 extern char g_iso_path[256];
1277 int ventoy_add_vlnk_file(char *dir, const char *name);
1278 grub_err_t ventoy_cmd_browser_dir(grub_extcmd_context_t ctxt, int argc, char **args);
1279 grub_err_t ventoy_cmd_browser_disk(grub_extcmd_context_t ctxt, int argc, char **args);
1280 int ventoy_get_fs_type(const char *fs);
1281 int ventoy_img_name_valid(const char *filename, grub_size_t namelen);
1282 void * ventoy_alloc_chain(grub_size_t size);
1283 int ventoy_plugin_load_menu_lang(int init, const char *lang);
1284 const char *ventoy_get_vmenu_title(const char *vMenu);
1285 grub_err_t ventoy_cmd_cur_menu_lang(grub_extcmd_context_t ctxt, int argc, char **args);
1286 extern int ventoy_menu_push_key(int code);
1287
1288 #endif /* __VENTOY_DEF_H__ */
1289