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