]> glassweightruler.freedombox.rocks Git - Ventoy.git/blob - GRUB2/MOD_SRC/grub-2.04/grub-core/ventoy/ventoy_def.h
projects / Ventoy.git / blob
? search:


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