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


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