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