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


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