1 /******************************************************************************
4 * Copyright (c) 2020, longpanda <admin@ventoy.net>
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.
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.
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/>.
21 #ifndef __VENTOY_DEF_H__
22 #define __VENTOY_DEF_H__
24 #define VTOY_MAX_DIR_DEPTH 32
26 #define VTOY_MAX_SCRIPT_BUF (4 * 1024 * 1024)
28 #define VTOY_PART_BUF_LEN (128 * 1024)
30 #define VTOY_FILT_MIN_FILE_SIZE 32768
32 #define VTOY_LINUX_SYSTEMD_MENU_MAX_BUF 16384
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)
43 #define JSON_SUCCESS 0
45 #define JSON_NOT_FOUND 2
47 #define WINDATA_FLAG_TEMPLATE 1
49 #define ulong unsigned long
50 #define ulonglong unsigned long long
52 #define vtoy_to_upper(c) (((char)(c) >= 'a' && (char)(c) <= 'z') ? ((char)(c) - 'a' + 'A') : (char)(c))
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)
57 #define ventoy_env_op1(op, a) grub_env_##op(a)
58 #define ventoy_env_op2(op, a, b) grub_env_##op((a), (b))
60 #define ventoy_get_env(key) ventoy_env_op1(get, key)
61 #define ventoy_set_env(key, val) ventoy_env_op2(set, key, val)
63 #define VTOY_WARNING "!!!!!!!!!!!!! WARNING !!!!!!!!!!!!!"
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
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"
77 #define VTOY_ARCH_CPIO "ventoy_x86.cpio"
80 #define ventoy_left_key "VTLE_LFT"
81 #define ventoy_top_key "VTLE_TOP"
82 #define ventoy_color_key "VTLE_CLR"
84 #define ventoy_varg_4(arg) arg[0], arg[1], arg[2], arg[3]
85 #define ventoy_varg_8(arg) arg[0], arg[1], arg[2], arg[3], arg[4], arg[5], arg[6], arg[7]
87 #define VTOY_PWD_CORRUPTED(err) \
89 grub_printf("\n\n Password corrupted, will reboot after 5 seconds.\n\n"); \
96 #define VTOY_APPEND_NEWBUF(buf) \
101 newbuf[pos++] = *__c;\
106 #define VTOY_SKIP_SPACE(s) \
107 while (ventoy_isspace(*s)) \
112 #define VTOY_SKIP_SPACE_NEXT(s, initial) \
114 while (ventoy_isspace(*s)) \
119 #define VTOY_SKIP_SPACE_NEXT_EX(s, base, initial) \
121 while (ventoy_isspace(*s)) \
126 #define VTOY_GOTO_END(v) ret = v; goto end
128 typedef enum VTOY_FILE_FLT
130 VTOY_FILE_FLT_ISO
= 0, /* .iso */
131 VTOY_FILE_FLT_WIM
, /* .wim */
132 VTOY_FILE_FLT_EFI
, /* .efi */
133 VTOY_FILE_FLT_IMG
, /* .img */
134 VTOY_FILE_FLT_VHD
, /* .vhd(x) */
135 VTOY_FILE_FLT_VTOY
, /* .vtoy */
140 #define FILE_FLT(type) (0 == g_vtoy_file_flt[VTOY_FILE_FLT_##type])
142 typedef struct ventoy_initrd_ctx
144 const char *path_prefix
;
145 const char *dir_prefix
;
148 typedef struct cmd_para
151 grub_extcmd_func_t func
;
152 grub_command_flags_t flags
;
153 const struct grub_arg_option
*parser
;
156 const char *description
;
161 #define ventoy_align_2k(value) ((value + 2047) / 2048 * 2048)
162 #define ventoy_align(value, align) (((value) + ((align) - 1)) & (~((align) - 1)))
165 typedef struct cpio_newc_header
185 #define cmd_raw_name ctxt->extcmd->cmd->name
186 #define check_free(p, func) if (p) { func(p); p = NULL; }
187 #define grub_check_free(p) if (p) { grub_free(p); p = NULL; }
189 typedef int (*grub_char_check_func
)(int c
);
190 #define ventoy_is_decimal(str) ventoy_string_check(str, grub_isdigit)
192 #define OFFSET_OF(TYPE, MEMBER) ((grub_size_t) &((TYPE *)0)->MEMBER)
195 typedef struct ventoy_patch_vhd
197 grub_uint8_t part_offset_or_guid
[16];
198 grub_uint32_t reserved1
;
199 grub_uint32_t part_type
;
200 grub_uint8_t disk_signature_or_guid
[16];
201 grub_uint8_t reserved2
[16];
202 grub_uint8_t vhd_file_path
[1];
206 // El Torito Boot Record Volume Descriptor
208 typedef struct eltorito_descriptor
212 grub_uint8_t version
;
213 grub_uint8_t system_id
[32];
214 grub_uint8_t reserved
[32];
215 grub_uint32_t sector
;
216 }eltorito_descriptor
;
218 typedef struct ventoy_iso9660_override
220 grub_uint32_t first_sector
;
221 grub_uint32_t first_sector_be
;
223 grub_uint32_t size_be
;
224 }ventoy_iso9660_override
;
226 typedef struct ventoy_udf_override
228 grub_uint32_t length
;
229 grub_uint32_t position
;
230 }ventoy_udf_override
;
232 typedef struct ventoy_iso9660_vd
240 grub_uint8_t res2
[8];
244 /* https://wiki.osdev.org/El-Torito */
245 typedef struct boot_info_table
247 grub_uint32_t bi_data0
;
248 grub_uint32_t bi_data1
;
249 grub_uint32_t bi_PrimaryVolumeDescriptor
;
250 grub_uint32_t bi_BootFileLocation
;
251 grub_uint32_t bi_BootFileLength
;
252 grub_uint32_t bi_Checksum
;
253 grub_uint8_t bi_Reserved
[40];
258 #define img_type_start 0
259 #define img_type_iso 0
260 #define img_type_wim 1
261 #define img_type_efi 2
262 #define img_type_img 3
263 #define img_type_vhd 4
264 #define img_type_vtoy 5
265 #define img_type_max 6
267 typedef struct img_info
277 const char *menu_prefix
;
281 int plugin_list_index
;
288 struct img_info
*next
;
289 struct img_info
*prev
;
292 typedef struct img_iterator_node
294 struct img_iterator_node
*next
;
303 int plugin_list_index
;
305 struct img_iterator_node
*parent
;
306 struct img_iterator_node
*firstchild
;
313 typedef struct initrd_info
317 grub_uint64_t offset
;
320 grub_uint8_t iso_type
; // 0: iso9660 1:udf
321 grub_uint32_t udf_start_block
;
323 grub_uint64_t override_offset
;
324 grub_uint32_t override_length
;
325 char override_data
[32];
327 struct initrd_info
*next
;
328 struct initrd_info
*prev
;
331 extern initrd_info
*g_initrd_img_list
;
332 extern initrd_info
*g_initrd_img_tail
;
333 extern int g_initrd_img_count
;
334 extern int g_valid_initrd_count
;
336 extern img_info
*g_ventoy_img_list
;
337 extern int g_ventoy_img_count
;
339 extern grub_uint8_t
*g_ventoy_cpio_buf
;
340 extern grub_uint32_t g_ventoy_cpio_size
;
341 extern cpio_newc_header
*g_ventoy_initrd_head
;
342 extern grub_uint8_t
*g_ventoy_runtime_buf
;
344 extern ventoy_guid g_ventoy_guid
;
346 extern ventoy_img_chunk_list g_img_chunk_list
;
347 extern ventoy_img_chunk_list g_wimiso_chunk_list
;
348 extern char *g_wimiso_path
;
349 extern grub_uint32_t g_wimiso_size
;
350 extern char g_arch_mode_suffix
[64];
351 extern const char *g_menu_prefix
[img_type_max
];
353 extern int g_ventoy_debug
;
354 void ventoy_debug(const char *fmt
, ...);
355 #define debug(fmt, args...) if (g_ventoy_debug) ventoy_debug("[VTOY]: "fmt, ##args)
357 #define vtoy_ssprintf(buf, pos, fmt, args...) \
358 pos += grub_snprintf(buf + pos, VTOY_MAX_SCRIPT_BUF - pos, fmt, ##args)
360 #define vtoy_len_ssprintf(buf, pos, len, fmt, args...) \
361 pos += grub_snprintf(buf + pos, len - pos, fmt, ##args)
363 #define browser_ssprintf(mbuf, fmt, args...) \
364 (mbuf)->pos += grub_snprintf((mbuf)->buf + (mbuf)->pos, (mbuf)->max - (mbuf)->pos, fmt, ##args)
366 #define vtoy_dummy_menuentry(buf, pos, len, title, class) \
367 vtoy_len_ssprintf(buf, pos, len, "menuentry \"%s\" --class=\"%s\" {\n echo \"\"\n}\n", title, class)
370 #define FLAG_HEADER_RESERVED 0x00000001
371 #define FLAG_HEADER_COMPRESSION 0x00000002
372 #define FLAG_HEADER_READONLY 0x00000004
373 #define FLAG_HEADER_SPANNED 0x00000008
374 #define FLAG_HEADER_RESOURCE_ONLY 0x00000010
375 #define FLAG_HEADER_METADATA_ONLY 0x00000020
376 #define FLAG_HEADER_WRITE_IN_PROGRESS 0x00000040
377 #define FLAG_HEADER_RP_FIX 0x00000080 // reparse point fixup
378 #define FLAG_HEADER_COMPRESS_RESERVED 0x00010000
379 #define FLAG_HEADER_COMPRESS_XPRESS 0x00020000
380 #define FLAG_HEADER_COMPRESS_LZX 0x00040000
381 #define FLAG_HEADER_COMPRESS_LZMS 0x00080000
383 #define RESHDR_FLAG_FREE 0x01
384 #define RESHDR_FLAG_METADATA 0x02
385 #define RESHDR_FLAG_COMPRESSED 0x04
386 #define RESHDR_FLAG_SPANNED 0x08
390 /* A WIM resource header */
391 typedef struct wim_resource_header
393 grub_uint64_t size_in_wim
:56; /* Compressed length */
394 grub_uint64_t flags
:8; /* flags */
395 grub_uint64_t offset
; /* Offset */
396 grub_uint64_t raw_size
; /* Uncompressed length */
397 }wim_resource_header
;
399 /* WIM resource header length mask */
400 #define WIM_RESHDR_ZLEN_MASK 0x00ffffffffffffffULL
402 /* WIM resource header flags */
403 typedef enum wim_resource_header_flags
405 WIM_RESHDR_METADATA
= ( 0x02ULL
<< 56 ), /* Resource contains metadata */
406 WIM_RESHDR_COMPRESSED
= ( 0x04ULL
<< 56 ), /* Resource is compressed */
407 WIM_RESHDR_PACKED_STREAMS
= ( 0x10ULL
<< 56 ), /* Resource is compressed using packed streams */
408 }wim_resource_header_flags
;
410 #define WIM_HEAD_SIGNATURE "MSWIM\0\0"
413 typedef struct wim_header
415 grub_uint8_t signature
[8]; /* Signature */
416 grub_uint32_t header_len
; /* Header length */
417 grub_uint32_t version
; /* Verson */
418 grub_uint32_t flags
; /* Flags */
419 grub_uint32_t chunk_len
; /* Chunk length */
420 grub_uint8_t guid
[16]; /* GUID */
421 grub_uint16_t part
; /* Part number */
422 grub_uint16_t parts
; /* Total number of parts */
423 grub_uint32_t images
; /* number of images */
424 wim_resource_header lookup
; /* Lookup table */
425 wim_resource_header xml
; /* XML data */
426 wim_resource_header metadata
; /* Boot metadata */
427 grub_uint32_t boot_index
; /* Boot index */
428 wim_resource_header integrity
; /* Integrity table */
429 grub_uint8_t reserved
[60]; /* Reserved */
432 /* WIM header flags */
433 typedef enum wim_header_flags
435 WIM_HDR_XPRESS
= 0x00020000, /* WIM uses Xpress compresson */
436 WIM_HDR_LZX
= 0x00040000, /* WIM uses LZX compression */
439 /* A WIM file hash */
440 typedef struct wim_hash
443 grub_uint8_t sha1
[20];
446 /* A WIM lookup table entry */
447 typedef struct wim_lookup_entry
449 wim_resource_header resource
; /* Resource header */
450 grub_uint16_t part
; /* Part number */
451 grub_uint32_t refcnt
; /* Reference count */
452 wim_hash hash
; /* Hash */
455 /* WIM chunk length */
456 #define WIM_CHUNK_LEN 32768
458 /* A WIM chunk buffer */
459 typedef struct wim_chunk_buffer
461 grub_uint8_t data
[WIM_CHUNK_LEN
]; /*Data */
465 typedef struct wim_security_header
467 grub_uint32_t len
; /* Length */
468 grub_uint32_t count
; /* Number of entries */
469 }wim_security_header
;
471 typedef struct wim_stream_entry
474 grub_uint64_t unused1
;
476 grub_uint16_t name_len
;
480 /* Directory entry */
481 typedef struct wim_directory_entry
483 grub_uint64_t len
; /* Length */
484 grub_uint32_t attributes
; /* Attributes */
485 grub_uint32_t security
; /* Security ID */
486 grub_uint64_t subdir
; /* Subdirectory offset */
487 grub_uint8_t reserved1
[16]; /* Reserved */
488 grub_uint64_t created
; /* Creation time */
489 grub_uint64_t accessed
; /* Last access time */
490 grub_uint64_t written
; /* Last written time */
491 wim_hash hash
; /* Hash */
492 grub_uint8_t reserved2
[12]; /* Reserved */
493 grub_uint16_t streams
; /* Streams */
494 grub_uint16_t short_name_len
; /* Short name length */
495 grub_uint16_t name_len
; /* Name length */
496 }wim_directory_entry
;
499 #define WIM_ATTR_NORMAL 0x00000080UL
501 /** No security information exists for this file */
502 #define WIM_NO_SECURITY 0xffffffffUL
504 typedef struct reg_vk
508 grub_uint16_t namesize
;
509 grub_uint32_t datasize
;
510 grub_uint32_t dataoffset
;
511 grub_uint32_t datatype
;
519 typedef struct wim_tail
521 grub_uint32_t wim_raw_size
;
522 grub_uint32_t wim_align_size
;
524 grub_uint8_t iso_type
;
525 grub_uint64_t file_offset
;
526 grub_uint32_t udf_start_block
;
527 grub_uint64_t fe_entry_size_offset
;
528 grub_uint64_t override_offset
;
529 grub_uint32_t override_len
;
530 grub_uint8_t override_data
[32];
532 wim_header wim_header
;
535 grub_uint32_t jump_exe_len
;
536 grub_uint8_t
*jump_bin_data
;
537 grub_uint32_t bin_raw_len
;
538 grub_uint32_t bin_align_len
;
539 grub_uint32_t windata_flag
;
541 grub_uint8_t
*new_meta_data
;
542 grub_uint32_t new_meta_len
;
543 grub_uint32_t new_meta_align_len
;
545 grub_uint8_t
*new_lookup_data
;
546 grub_uint32_t new_lookup_len
;
547 grub_uint32_t new_lookup_align_len
;
550 typedef struct wim_patch
557 wim_lookup_entry
*replace_look
;
561 struct wim_patch
*next
;
565 typedef enum _JSON_TYPE
567 JSON_TYPE_NUMBER
= 0,
577 typedef struct _VTOY_JSON
579 struct _VTOY_JSON
*pstPrev
;
580 struct _VTOY_JSON
*pstNext
;
581 struct _VTOY_JSON
*pstChild
;
583 JSON_TYPE enDataType
;
588 grub_uint64_t lValue
;
594 typedef struct _JSON_PARSE
598 grub_uint32_t uiBufSize
;
601 #define JSON_NEW_ITEM(pstJson, ret) \
603 (pstJson) = (VTOY_JSON *)grub_zalloc(sizeof(VTOY_JSON)); \
604 if (NULL == (pstJson)) \
606 json_debug("Failed to alloc memory for json.\n"); \
611 typedef int (*ventoy_plugin_entry_pf
)(VTOY_JSON
*json
, const char *isodisk
);
612 typedef int (*ventoy_plugin_check_pf
)(VTOY_JSON
*json
, const char *isodisk
);
614 typedef struct plugin_entry
617 ventoy_plugin_entry_pf entryfunc
;
618 ventoy_plugin_check_pf checkfunc
;
622 typedef struct replace_fs_dir
633 typedef struct chk_case_fs_dir
639 int ventoy_str_all_digit(const char *str
);
640 int ventoy_str_all_alnum(const char *str
);
641 int ventoy_str_len_alnum(const char *str
, int len
);
642 char * ventoy_str_basename(char *path
);
643 grub_err_t
ventoy_env_int_set(const char *name
, int value
);
644 int ventoy_str_chrcnt(const char *str
, char c
);
645 int ventoy_strcmp(const char *pattern
, const char *str
);
646 int ventoy_strncmp (const char *pattern
, const char *str
, grub_size_t n
);
647 void ventoy_fill_os_param(grub_file_t file
, ventoy_os_param
*param
);
648 grub_err_t
ventoy_cmd_isolinux_initrd_collect(grub_extcmd_context_t ctxt
, int argc
, char **args
);
649 grub_err_t
ventoy_cmd_grub_initrd_collect(grub_extcmd_context_t ctxt
, int argc
, char **args
);
650 grub_err_t
ventoy_cmd_specify_initrd_file(grub_extcmd_context_t ctxt
, int argc
, char **args
);
651 grub_err_t
ventoy_cmd_dump_initrd_list(grub_extcmd_context_t ctxt
, int argc
, char **args
);
652 grub_err_t
ventoy_cmd_clear_initrd_list(grub_extcmd_context_t ctxt
, int argc
, char **args
);
653 grub_uint32_t
ventoy_get_iso_boot_catlog(grub_file_t file
);
654 int ventoy_has_efi_eltorito(grub_file_t file
, grub_uint32_t sector
);
655 grub_err_t
ventoy_cmd_linux_chain_data(grub_extcmd_context_t ctxt
, int argc
, char **args
);
656 grub_err_t
ventoy_cmd_linux_systemd_menu(grub_extcmd_context_t ctxt
, int argc
, char **args
);
657 grub_err_t
ventoy_cmd_linux_limine_menu(grub_extcmd_context_t ctxt
, int argc
, char **args
);
658 grub_err_t
ventoy_cmd_linux_locate_initrd(grub_extcmd_context_t ctxt
, int argc
, char **args
);
659 grub_err_t
ventoy_cmd_initrd_count(grub_extcmd_context_t ctxt
, int argc
, char **args
);
660 grub_err_t
ventoy_cmd_valid_initrd_count(grub_extcmd_context_t ctxt
, int argc
, char **args
);
661 grub_err_t
ventoy_cmd_load_cpio(grub_extcmd_context_t ctxt
, int argc
, char **args
);
662 grub_err_t
ventoy_cmd_append_ext_sector(grub_extcmd_context_t ctxt
, int argc
, char **args
);
663 grub_err_t
ventoy_cmd_skip_svd(grub_extcmd_context_t ctxt
, int argc
, char **args
);
664 grub_err_t
ventoy_cmd_cpio_busybox_64(grub_extcmd_context_t ctxt
, int argc
, char **args
);
665 grub_err_t
ventoy_cmd_trailer_cpio(grub_extcmd_context_t ctxt
, int argc
, char **args
);
666 int ventoy_cpio_newc_fill_head(void *buf
, int filesize
, const void *filedata
, const char *name
);
667 grub_file_t
ventoy_grub_file_open(enum grub_file_type type
, const char *fmt
, ...);
668 grub_uint64_t
ventoy_grub_get_file_size(const char *fmt
, ...);
669 int ventoy_is_dir_exist(const char *fmt
, ...);
670 int ventoy_fill_data(grub_uint32_t buflen
, char *buffer
);
671 grub_err_t
ventoy_cmd_load_plugin(grub_extcmd_context_t ctxt
, int argc
, char **args
);
672 grub_err_t
ventoy_cmd_wimdows_reset(grub_extcmd_context_t ctxt
, int argc
, char **args
);
673 grub_err_t
ventoy_cmd_is_pe64(grub_extcmd_context_t ctxt
, int argc
, char **args
);
674 grub_err_t
ventoy_cmd_windows_chain_data(grub_extcmd_context_t ctxt
, int argc
, char **args
);
675 grub_err_t
ventoy_cmd_windows_wimboot_data(grub_extcmd_context_t ctxt
, int argc
, char **args
);
676 grub_err_t
ventoy_cmd_wim_chain_data(grub_extcmd_context_t ctxt
, int argc
, char **args
);
677 grub_err_t
ventoy_cmd_wim_check_bootable(grub_extcmd_context_t ctxt
, int argc
, char **args
);
678 grub_err_t
ventoy_cmd_is_standard_winiso(grub_extcmd_context_t ctxt
, int argc
, char **args
);
679 grub_err_t
ventoy_cmd_dump_wim_patch(grub_extcmd_context_t ctxt
, int argc
, char **args
);
680 grub_err_t
ventoy_cmd_sel_wimboot(grub_extcmd_context_t ctxt
, int argc
, char **args
);
681 grub_err_t
ventoy_cmd_set_wim_prompt(grub_extcmd_context_t ctxt
, int argc
, char **args
);
682 grub_ssize_t
ventoy_load_file_with_prompt(grub_file_t file
, void *buf
, grub_ssize_t size
);
683 int ventoy_need_prompt_load_file(void);
685 VTOY_JSON
*vtoy_json_find_item
688 JSON_TYPE enDataType
,
691 int vtoy_json_parse_value
699 VTOY_JSON
* vtoy_json_create(void);
700 int vtoy_json_parse(VTOY_JSON
*pstJson
, const char *szJsonData
);
702 int vtoy_json_scan_parse
704 const VTOY_JSON
*pstJson
,
705 grub_uint32_t uiParseNum
,
706 JSON_PARSE
*pstJsonParse
709 int vtoy_json_scan_array
713 VTOY_JSON
**ppstArrayItem
716 int vtoy_json_scan_array_ex
720 VTOY_JSON
**ppstArrayItem
722 int vtoy_json_scan_object
726 VTOY_JSON
**ppstObjectItem
728 int vtoy_json_get_int
734 int vtoy_json_get_uint
738 grub_uint32_t
*puiValue
740 int vtoy_json_get_uint64
744 grub_uint64_t
*pui64Value
746 int vtoy_json_get_bool
750 grub_uint8_t
*pbValue
752 int vtoy_json_get_string
756 grub_uint32_t uiBufLen
,
759 const char * vtoy_json_get_string_ex(VTOY_JSON
*pstJson
, const char *szKey
);
760 int vtoy_json_destroy(VTOY_JSON
*pstJson
);
763 grub_uint32_t CalculateCrc32
766 grub_uint32_t Length
,
767 grub_uint32_t InitValue
770 static inline int ventoy_isspace (int c
)
772 return (c
== '\n' || c
== '\r' || c
== ' ' || c
== '\t');
775 static inline int ventoy_is_word_end(int c
)
777 return (c
== 0 || c
== ',' || ventoy_isspace(c
));
781 typedef struct ventoy_part_table
783 grub_uint8_t Active
; // 0x00 0x80
785 grub_uint8_t StartHead
;
786 grub_uint16_t StartSector
: 6;
787 grub_uint16_t StartCylinder
: 10;
791 grub_uint8_t EndHead
;
792 grub_uint16_t EndSector
: 6;
793 grub_uint16_t EndCylinder
: 10;
795 grub_uint32_t StartSectorId
;
796 grub_uint32_t SectorCount
;
799 typedef struct ventoy_mbr_head
801 grub_uint8_t BootCode
[446];
802 ventoy_part_table PartTbl
[4];
807 typedef struct ventoy_gpt_head
809 char Signature
[8]; /* EFI PART */
810 grub_uint8_t Version
[4];
811 grub_uint32_t Length
;
813 grub_uint8_t Reserved1
[4];
814 grub_uint64_t EfiStartLBA
;
815 grub_uint64_t EfiBackupLBA
;
816 grub_uint64_t PartAreaStartLBA
;
817 grub_uint64_t PartAreaEndLBA
;
818 grub_uint8_t DiskGuid
[16];
819 grub_uint64_t PartTblStartLBA
;
820 grub_uint32_t PartTblTotNum
;
821 grub_uint32_t PartTblEntryLen
;
822 grub_uint32_t PartTblCrc
;
823 grub_uint8_t Reserved2
[420];
826 typedef struct ventoy_gpt_part_tbl
828 grub_uint8_t PartType
[16];
829 grub_uint8_t PartGuid
[16];
830 grub_uint64_t StartLBA
;
831 grub_uint64_t LastLBA
;
833 grub_uint16_t Name
[36];
834 }ventoy_gpt_part_tbl
;
836 typedef struct ventoy_gpt_info
839 ventoy_gpt_head Head
;
840 ventoy_gpt_part_tbl PartTbl
[128];
843 typedef struct vhd_footer_t
845 char cookie
[8]; // Cookie
846 grub_uint32_t features
; // Features
847 grub_uint32_t ffversion
; // File format version
848 grub_uint32_t dataoffset
; // Data offset
849 grub_uint32_t timestamp
; // Timestamp
850 grub_uint32_t creatorapp
; // Creator application
851 grub_uint32_t creatorver
; // Creator version
852 grub_uint32_t creatorhos
; // Creator host OS
853 grub_uint32_t origsize
; // Original size
854 grub_uint32_t currsize
; // Current size
855 grub_uint32_t diskgeom
; // Disk geometry
856 grub_uint32_t disktype
; // Disk type
857 grub_uint32_t checksum
; // Checksum
858 grub_uint8_t uniqueid
[16]; // Unique ID
859 grub_uint8_t savedst
; // Saved state
862 #define VDI_IMAGE_FILE_INFO "<<< Oracle VM VirtualBox Disk Image >>>\n"
864 /** Image signature. */
865 #define VDI_IMAGE_SIGNATURE (0xbeda107f)
867 typedef struct VDIPREHEADER
869 /** Just text info about image type, for eyes only. */
871 /** The image signature (VDI_IMAGE_SIGNATURE). */
872 grub_uint32_t u32Signature
;
873 /** The image version (VDI_IMAGE_VERSION). */
874 grub_uint32_t u32Version
;
875 } VDIPREHEADER
, *PVDIPREHEADER
;
879 typedef struct ventoy_video_mode
882 grub_uint32_t height
;
888 typedef struct file_fullpath
894 typedef struct theme_list
897 struct theme_list
*next
;
900 #define auto_install_type_file 0
901 #define auto_install_type_parent 1
902 typedef struct install_template
912 file_fullpath
*templatepath
;
917 struct install_template
*next
;
920 typedef struct dudfile
932 file_fullpath
*dudpath
;
938 typedef struct persistence_config
947 file_fullpath
*backendpath
;
949 struct persistence_config
*next
;
952 #define vtoy_alias_image_file 0
953 #define vtoy_alias_directory 1
955 typedef struct menu_alias
962 struct menu_alias
*next
;
965 #define vtoy_tip_image_file 0
966 #define vtoy_tip_directory 1
967 typedef struct menu_tip
975 struct menu_tip
*next
;
979 #define vtoy_class_image_file 0
980 #define vtoy_class_directory 1
982 typedef struct menu_class
990 struct menu_class
*next
;
993 #define vtoy_custom_boot_image_file 0
994 #define vtoy_custom_boot_directory 1
996 typedef struct custom_boot
1003 struct custom_boot
*next
;
1006 #define vtoy_max_replace_file_size (1024 * 1024)
1007 typedef struct conf_replace
1015 struct conf_replace
*next
;
1018 #define injection_type_file 0
1019 #define injection_type_parent 1
1020 typedef struct injection_config
1027 struct injection_config
*next
;
1030 typedef struct auto_memdisk
1035 struct auto_memdisk
*next
;
1038 typedef struct image_list
1043 struct image_list
*next
;
1046 #define VTOY_PASSWORD_NONE 0
1047 #define VTOY_PASSWORD_TXT 1
1048 #define VTOY_PASSWORD_MD5 2
1049 #define VTOY_PASSWORD_SALT_MD5 3
1051 typedef struct vtoy_password
1056 grub_uint8_t md5
[16];
1059 #define vtoy_menu_pwd_file 0
1060 #define vtoy_menu_pwd_parent 1
1062 typedef struct menu_password
1068 vtoy_password password
;
1070 struct menu_password
*next
;
1073 extern int g_ventoy_menu_esc
;
1074 extern int g_ventoy_secondary_menu_on
;
1075 extern int g_ventoy_suppress_esc
;
1076 extern int g_ventoy_suppress_esc_default
;
1077 extern int g_ventoy_last_entry
;
1078 extern int g_ventoy_memdisk_mode
;
1079 extern int g_ventoy_iso_raw
;
1080 extern int g_ventoy_grub2_mode
;
1081 extern int g_ventoy_wimboot_mode
;
1082 extern int g_ventoy_iso_uefi_drv
;
1083 extern int g_ventoy_case_insensitive
;
1084 extern int g_ventoy_fn_mutex
;
1085 extern grub_uint8_t g_ventoy_chain_type
;
1086 extern int g_vhdboot_enable
;
1087 extern int g_default_menu_mode
;
1088 extern char g_ventoy_hotkey_tip
[256];
1089 extern int g_ventoy_menu_refresh
;
1091 #define VENTOY_IMG_WHITE_LIST 1
1092 #define VENTOY_IMG_BLACK_LIST 2
1093 extern int g_plugin_image_list
;
1095 extern ventoy_gpt_info
*g_ventoy_part_info
;
1096 extern int g_conf_replace_count
;
1097 extern grub_uint64_t g_conf_replace_offset
[VTOY_MAX_CONF_REPLACE
];
1098 extern grub_uint64_t g_svd_replace_offset
;
1099 extern conf_replace
*g_conf_replace_node
[VTOY_MAX_CONF_REPLACE
];
1100 extern grub_uint8_t
*g_conf_replace_new_buf
[VTOY_MAX_CONF_REPLACE
];
1101 extern int g_conf_replace_new_len
[VTOY_MAX_CONF_REPLACE
];
1102 extern int g_conf_replace_new_len_align
[VTOY_MAX_CONF_REPLACE
];
1103 extern int g_ventoy_disk_bios_id
;
1104 extern grub_uint64_t g_ventoy_disk_size
;
1105 extern grub_uint64_t g_ventoy_disk_part_size
[2];
1106 extern grub_uint32_t g_ventoy_plat_data
;
1108 #define ventoy_unix_fill_virt(new_data, new_len) \
1110 data_secs = (new_len + 2047) / 2048; \
1111 cur->mem_sector_start = sector; \
1112 cur->mem_sector_end = cur->mem_sector_start + data_secs; \
1113 cur->mem_sector_offset = offset; \
1114 cur->remap_sector_start = 0; \
1115 cur->remap_sector_end = 0; \
1116 cur->org_sector_start = 0; \
1117 grub_memcpy(override + offset, new_data, new_len); \
1119 sector += data_secs; \
1120 offset += new_len; \
1121 chain->virt_img_size_in_bytes += data_secs * 2048; \
1124 #define ventoy_syscall0(name) grub_##name()
1125 #define ventoy_syscall1(name, a) grub_##name(a)
1127 void ventoy_str_tolower(char *str
);
1128 void ventoy_str_toupper(char *str
);
1129 char * ventoy_get_line(char *start
);
1130 char *ventoy_str_last(char *str
, char ch
);
1131 int ventoy_cmp_img(img_info
*img1
, img_info
*img2
);
1132 void ventoy_swap_img(img_info
*img1
, img_info
*img2
);
1133 char * ventoy_plugin_get_cur_install_template(const char *isopath
, install_template
**cur
);
1134 install_template
* ventoy_plugin_find_install_template(const char *isopath
);
1135 persistence_config
* ventoy_plugin_find_persistent(const char *isopath
);
1136 grub_uint64_t
ventoy_get_vtoy_partsize(int part
);
1137 void ventoy_plugin_dump_injection(void);
1138 void ventoy_plugin_dump_auto_install(void);
1139 int ventoy_plugin_get_persistent_chunklist(const char *isopath
, int index
, ventoy_img_chunk_list
*chunk_list
);
1140 const char * ventoy_plugin_get_injection(const char *isopath
);
1141 const char * ventoy_plugin_get_menu_alias(int type
, const char *isopath
);
1142 const menu_tip
* ventoy_plugin_get_menu_tip(int type
, const char *isopath
);
1143 const char * ventoy_plugin_get_menu_class(int type
, const char *name
, const char *path
);
1144 int ventoy_plugin_check_memdisk(const char *isopath
);
1145 int ventoy_plugin_get_image_list_index(int type
, const char *name
);
1146 int ventoy_plugin_find_conf_replace(const char *iso
, conf_replace
*nodes
[VTOY_MAX_CONF_REPLACE
]);
1147 dud
* ventoy_plugin_find_dud(const char *iso
);
1148 int ventoy_plugin_load_dud(dud
*node
, const char *isopart
);
1149 int ventoy_get_block_list(grub_file_t file
, ventoy_img_chunk_list
*chunklist
, grub_disk_addr_t start
);
1150 int ventoy_check_block_list(grub_file_t file
, ventoy_img_chunk_list
*chunklist
, grub_disk_addr_t start
);
1151 void ventoy_plugin_dump_persistence(void);
1152 grub_err_t
ventoy_cmd_set_theme(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1153 grub_err_t
ventoy_cmd_set_theme_path(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1154 grub_err_t
ventoy_cmd_select_theme_cfg(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1155 grub_err_t
ventoy_cmd_plugin_check_json(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1156 grub_err_t
ventoy_cmd_check_password(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1157 grub_err_t
ventoy_cmd_linux_get_main_initrd_index(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1158 grub_err_t
ventoy_cmd_collect_wim_patch(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1159 grub_err_t
ventoy_cmd_wim_patch_count(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1160 grub_err_t
ventoy_cmd_locate_wim_patch(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1161 grub_err_t
ventoy_cmd_sel_winpe_wim(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1162 grub_err_t
ventoy_cmd_unix_chain_data(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1163 int ventoy_get_disk_guid(const char *filename
, grub_uint8_t
*guid
, grub_uint8_t
*signature
);
1164 grub_err_t
ventoy_cmd_unix_reset(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1165 grub_err_t
ventoy_cmd_unix_check_vlnk(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1166 grub_err_t
ventoy_cmd_unix_replace_conf(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1167 grub_err_t
ventoy_cmd_unix_replace_grub_conf(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1168 grub_err_t
ventoy_cmd_unix_replace_ko(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1169 grub_err_t
ventoy_cmd_unix_ko_fillmap(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1170 grub_err_t
ventoy_cmd_unix_fill_image_desc(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1171 grub_err_t
ventoy_cmd_unix_gzip_newko(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1172 grub_err_t
ventoy_cmd_unix_freebsd_ver(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1173 grub_err_t
ventoy_cmd_parse_freenas_ver(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1174 grub_err_t
ventoy_cmd_unix_freebsd_ver_elf(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1175 int ventoy_check_device_result(int ret
);
1176 int ventoy_check_device(grub_device_t dev
);
1177 void ventoy_debug_dump_guid(const char *prefix
, grub_uint8_t
*guid
);
1178 grub_err_t
ventoy_cmd_load_vhdboot(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1179 grub_err_t
ventoy_cmd_patch_vhdboot(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1180 grub_err_t
ventoy_cmd_raw_chain_data(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1181 grub_err_t
ventoy_cmd_get_vtoy_type(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1182 int ventoy_check_password(const vtoy_password
*pwd
, int retry
);
1183 int ventoy_plugin_add_custom_boot(const char *vcfgpath
);
1184 const char * ventoy_plugin_get_custom_boot(const char *isopath
);
1185 grub_err_t
ventoy_cmd_dump_custom_boot(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1186 int ventoy_gzip_compress(void *mem_in
, int mem_in_len
, void *mem_out
, int mem_out_len
);
1187 int ventoy_load_part_table(const char *diskname
);
1188 int ventoy_env_init(void);
1189 int ventoy_register_all_cmd(void);
1190 int ventoy_unregister_all_cmd(void);
1191 int ventoy_chain_file_size(const char *path
);
1192 int ventoy_chain_file_read(const char *path
, int offset
, int len
, void *buf
);
1194 #define VTOY_CMD_CHECK(a) if (33554432 != g_ventoy_disk_part_size[a]) ventoy_syscall0(exit)
1196 #define vtoy_theme_random_boot_second 0
1197 #define vtoy_theme_random_boot_day 1
1198 #define vtoy_theme_random_boot_month 2
1200 #define ventoy_env_export(env, name) \
1202 grub_env_set((env), (name));\
1203 grub_env_export(env);\
1206 #define ret_goto_end(a) ret = a; goto end;
1208 extern ventoy_grub_param
*g_grub_param
;
1211 #define VENTOY_UNIX_SEG_MAGIC0 0x11223344
1212 #define VENTOY_UNIX_SEG_MAGIC1 0x55667788
1213 #define VENTOY_UNIX_SEG_MAGIC2 0x99aabbcc
1214 #define VENTOY_UNIX_SEG_MAGIC3 0xddeeff00
1215 #define VENTOY_UNIX_MAX_SEGNUM 40960
1216 struct g_ventoy_seg
{
1217 grub_uint64_t seg_start_bytes
;
1218 grub_uint64_t seg_end_bytes
;
1221 struct g_ventoy_map
{
1222 grub_uint32_t magic1
[4];
1223 grub_uint32_t magic2
[4];
1224 grub_uint64_t segnum
;
1225 grub_uint64_t disksize
;
1226 grub_uint8_t diskuuid
[16];
1227 struct g_ventoy_seg seglist
[VENTOY_UNIX_MAX_SEGNUM
];
1228 grub_uint32_t magic3
[4];
1232 typedef struct ventoy_vlnk_part
1234 grub_uint32_t disksig
;
1235 grub_uint64_t partoffset
;
1241 struct ventoy_vlnk_part
*next
;
1245 typedef struct browser_mbuf
1252 typedef struct browser_node
1255 char menuentry
[1024];
1257 struct browser_node
*prev
;
1258 struct browser_node
*next
;
1261 typedef struct var_node
1266 struct var_node
*next
;
1269 typedef struct systemd_menu_ctx
1277 typedef struct global_var_cfg
1284 typedef struct ctrl_var_cfg
1290 #define vtoy_check_goto_out(p) if (!p) goto out
1292 extern char *g_tree_script_buf
;
1293 extern int g_tree_script_pos
;
1294 extern int g_tree_script_pre
;
1295 extern int g_tree_view_menu_style
;
1296 extern int g_sort_case_sensitive
;
1297 extern int g_wimboot_enable
;
1298 extern int g_filt_dot_underscore_file
;
1299 extern int g_vtoy_file_flt
[VTOY_FILE_FLT_BUTT
];
1300 extern const char *g_menu_class
[img_type_max
];
1301 extern char g_iso_path
[256];
1302 int ventoy_add_vlnk_file(char *dir
, const char *name
);
1303 grub_err_t
ventoy_cmd_browser_dir(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1304 grub_err_t
ventoy_cmd_browser_disk(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1305 int ventoy_get_fs_type(const char *fs
);
1306 int ventoy_img_name_valid(const char *filename
, grub_size_t namelen
);
1307 void * ventoy_alloc_chain(grub_size_t size
);
1308 int ventoy_plugin_load_menu_lang(int init
, const char *lang
);
1309 const char *ventoy_get_vmenu_title(const char *vMenu
);
1310 grub_err_t
ventoy_cmd_cur_menu_lang(grub_extcmd_context_t ctxt
, int argc
, char **args
);
1311 extern int ventoy_menu_push_key(int code
);
1312 int ventoy_ctrl_var_init(void);
1313 int ventoy_global_var_init(void);
1315 #endif /* __VENTOY_DEF_H__ */