grub.cfg 21.7 KB
Newer Older
longpanda's avatar
longpanda committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
#************************************************************************************
# Copyright (c) 2020, longpanda <admin@ventoy.net>
# 
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License as
# published by the Free Software Foundation; either version 3 of the
# License, or (at your option) any later version.
# 
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
# 
# You should have received a copy of the GNU General Public License
# along with this program; if not, see <http://www.gnu.org/licenses/>.
# 
#************************************************************************************

longpanda's avatar
longpanda committed
19
20
21
22
23
24
25
26
27
28
29
30
31
32
function ventoy_pause {
    if [ -n "${vtdebug_flag}" ]; then
        echo "press Enter to continue ......"
        read vtTmpPause
    fi
}

function ventoy_debug_pause {
    if [ -n "${vtdebug_flag}" ]; then
        echo "press Enter to continue ......"
        read vtTmpPause
    fi
}

longpanda's avatar
update  
longpanda committed
33
34
function ventoy_cli_console {      
    if [ "$grub_platform" = "pc" ]; then
longpanda's avatar
longpanda committed
35
36
        #terminal_output  vga_text
        terminal_output  console
longpanda's avatar
update  
longpanda committed
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
    else
        if [ "$vtoy_display_mode" != "CLI" ]; then
            terminal_output  console
        fi
    fi
}

function ventoy_gui_console {  
    if [ "$grub_platform" = "pc" ]; then
        if [ "$vtoy_display_mode" = "CLI" ]; then
            terminal_output  console
        else
            terminal_output  gfxterm
        fi
    else
        if [ "$vtoy_display_mode" != "CLI" ]; then
            terminal_output  gfxterm
        fi
    fi
}
longpanda's avatar
longpanda committed
57

longpanda's avatar
longpanda committed
58
function ventoy_power {
longpanda's avatar
longpanda committed
59
60
61
62
63
64
65
66
67
    configfile $prefix/power.cfg
}

function ventoy_diagnosis {    
    configfile $prefix/debug.cfg
}

function ventoy_localboot {    
    configfile $prefix/localboot.cfg
longpanda's avatar
longpanda committed
68
69
}

longpanda's avatar
longpanda committed
70
function ventoy_ext_menu {    
longpanda's avatar
longpanda committed
71
    if [ -e $vt_plugin_path/ventoy/ventoy_grub.cfg ]; then
longpanda's avatar
longpanda committed
72
73
74
        set ventoy_new_context=1
        configfile $vt_plugin_path/ventoy/ventoy_grub.cfg
        unset ventoy_new_context
longpanda's avatar
longpanda committed
75
76
77
78
79
    else
       echo "ventoy_grub.cfg NOT exist."
       echo -e "\npress ENTER to exit ..."
       read vtInputKey
    fi
longpanda's avatar
longpanda committed
80
81
}

longpanda's avatar
longpanda committed
82
function get_os_type {
longpanda's avatar
longpanda committed
83
84
85
86
    set vtoy_os=Linux
    
    for file in "efi/microsoft/boot/bcd" "sources/boot.wim" "boot/bcd" "bootmgr.efi" "boot/etfsboot.com" ; do        
        if vt_file_exist_nocase (loop)/$file; then        
longpanda's avatar
longpanda committed
87
88
89
90
            set vtoy_os=Windows
            break
        fi
    done
longpanda's avatar
longpanda committed
91

longpanda's avatar
longpanda committed
92
93
94
95
96
    if [ -n "${vtdebug_flag}" ]; then
        echo ISO is $vtoy_os
    fi
}

longpanda's avatar
longpanda committed
97
98
function vt_check_compatible_pe {
    #Check for PE without external tools
longpanda's avatar
longpanda committed
99
    if [ -f $1/HBCD_PE.ini ]; then
longpanda's avatar
longpanda committed
100
        set ventoy_compatible=YES        
longpanda's avatar
longpanda committed
101
102
103
    fi
}

longpanda's avatar
longpanda committed
104
function locate_initrd {
longpanda's avatar
longpanda committed
105
106
    vt_linux_locate_initrd 

longpanda's avatar
longpanda committed
107
    if [ -n "${vtdebug_flag}" ]; then        
longpanda's avatar
longpanda committed
108
        vt_linux_dump_initrd
longpanda's avatar
longpanda committed
109
        ventoy_debug_pause
longpanda's avatar
longpanda committed
110
111
112
    fi
}

longpanda's avatar
longpanda committed
113
114
function locate_wim {
    vt_windows_locate_wim_patch (loop)
longpanda's avatar
longpanda committed
115
    
longpanda's avatar
longpanda committed
116
117
118
119
120
121
    if [ -n "${vtdebug_flag}" ]; then
        echo '###############################################'
        vt_dump_wim_patch
        echo '###############################################'
        ventoy_debug_pause
    fi
longpanda's avatar
longpanda committed
122
123
}

longpanda's avatar
longpanda committed
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
function distro_specify_wim_patch {
    if [ -d (loop)/h3pe ]; then
        vt_windows_collect_wim_patch wim /BOOT/H3_10PE.WIM
        vt_windows_collect_wim_patch wim /BOOT/H3_7PE.WIM
        vt_windows_collect_wim_patch wim /BOOT/H3_8PE.WIM
        vt_windows_collect_wim_patch wim /BOOT/H3_81PE.WIM
    fi
}

function distro_specify_wim_patch_phase2 {
    if [ -f (loop)/boot/boot.wim ]; then
        vt_windows_collect_wim_patch wim /boot/boot.wim
    fi
}


longpanda's avatar
longpanda committed
140
141
142
143
144
145
146
147
148
149
150
151
152
153
function distro_specify_initrd_file {
    if [ -e (loop)/boot/all.rdz ]; then
        vt_linux_specify_initrd_file /boot/all.rdz
    elif [ -e (loop)/boot/xen.gz ]; then 
        if [ -e (loop)/install.img ]; then
            vt_linux_specify_initrd_file /install.img
        fi
    elif [ -d (loop)/casper ]; then 
        if [ -e (loop)/casper/initrd ]; then
            vt_linux_specify_initrd_file /casper/initrd
        fi
        if [ -e (loop)/casper/initrd-oem ]; then
            vt_linux_specify_initrd_file /casper/initrd-oem
        fi
longpanda's avatar
longpanda committed
154
155
156
157
158
159
160
161
162
163
    elif [ -e (loop)/boot/grub/initrd.xz ]; then
        vt_linux_specify_initrd_file /boot/grub/initrd.xz
    elif [ -e (loop)/initrd.gz ]; then
        vt_linux_specify_initrd_file /initrd.gz
    elif [ -e (loop)/slax/boot/initrfs.img ]; then
        vt_linux_specify_initrd_file /slax/boot/initrfs.img
    elif [ -e (loop)/pmagic/initrd.img ]; then
        vt_linux_specify_initrd_file /pmagic/initrd.img
    elif [ -e (loop)/boot/initrd.xz ]; then
        vt_linux_specify_initrd_file /boot/initrd.xz
longpanda's avatar
longpanda committed
164
165
    elif [ -f (loop)/boot/initrd ]; then
        vt_linux_specify_initrd_file /boot/initrd
longpanda's avatar
longpanda committed
166
167
168
169
    elif [ -f (loop)/boot/x86_64/loader/initrd ]; then
        vt_linux_specify_initrd_file /boot/x86_64/loader/initrd
    elif [ -f (loop)/boot/initramfs-x86_64.img ]; then
        vt_linux_specify_initrd_file /boot/initramfs-x86_64.img
longpanda's avatar
longpanda committed
170
        
longpanda's avatar
longpanda committed
171
172
173
174

    fi
}

longpanda's avatar
longpanda committed
175

longpanda's avatar
longpanda committed
176
177
178
function distro_specify_initrd_file_phase2 {
    if [ -f (loop)/boot/initrd.img ]; then
        vt_linux_specify_initrd_file /boot/initrd.img
longpanda's avatar
longpanda committed
179
180
    elif [ -f (loop)/Setup/initrd.gz ]; then
        vt_linux_specify_initrd_file /Setup/initrd.gz
longpanda's avatar
longpanda committed
181
182
183
184
    elif [ -f (loop)/isolinux/initramfs ]; then
        vt_linux_specify_initrd_file /isolinux/initramfs
    elif [ -f (loop)/boot/iniramfs.igz ]; then
        vt_linux_specify_initrd_file /boot/iniramfs.igz
longpanda's avatar
longpanda committed
185
186
    elif [ -f (loop)/initrd-x86_64 ]; then
        vt_linux_specify_initrd_file /initrd-x86_64
longpanda's avatar
longpanda committed
187
    elif [ -f (loop)/live/initrd.img ]; then 
longpanda's avatar
longpanda committed
188
        vt_linux_specify_initrd_file /live/initrd.img
longpanda's avatar
longpanda committed
189
190
    elif [ -f (loop)/initrd.img ]; then 
        vt_linux_specify_initrd_file /initrd.img
longpanda's avatar
longpanda committed
191
192
    elif [ -f (loop)/sysresccd/boot/x86_64/sysresccd.img ]; then 
        vt_linux_specify_initrd_file /sysresccd/boot/x86_64/sysresccd.img
longpanda's avatar
longpanda committed
193
194
    elif [ -f (loop)/CDlinux/initrd ]; then 
        vt_linux_specify_initrd_file /CDlinux/initrd
longpanda's avatar
longpanda committed
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
    elif [ -f (loop)/parabola/boot/x86_64/parabolaiso.img ]; then 
        vt_linux_specify_initrd_file /parabola/boot/x86_64/parabolaiso.img
        if [ -f (loop)/parabola/boot/i686/parabolaiso.img ]; then 
            vt_linux_specify_initrd_file /parabola/boot/i686/parabolaiso.img
        fi
    elif [ -f (loop)/EFI/BOOT/initrd.img ]; then 
        #Qubes
        vt_linux_specify_initrd_file /EFI/BOOT/initrd.img
        if [ "$grub_platform" != "pc" ]; then
            vt_add_replace_file 0 "initrd.img"    
        fi
    elif [ -f (loop)/initrd ]; then 
        vt_linux_specify_initrd_file /initrd
    elif [ -f (loop)/live/initrd1 ]; then 
        vt_linux_specify_initrd_file /live/initrd1
        
longpanda's avatar
longpanda committed
211
212
213
214
215
    fi
}

function uefi_windows_menu_func {
    vt_windows_reset
longpanda's avatar
longpanda committed
216
217
218
219
220
221
222

    if [ "$ventoy_compatible" = "NO" ]; then   
        
        if [ "$ventoy_fs_probe" = "iso9660" ]; then
            loopback -d loop
            vt_iso9660_nojoliet 1            
            loopback loop $1$2
longpanda's avatar
longpanda committed
223
        fi
longpanda's avatar
longpanda committed
224
225
226
227
        
        for file in "efi/microsoft/boot/bcd"; do
            vt_windows_collect_wim_patch bcd (loop)/$file                
        done
longpanda's avatar
longpanda committed
228

longpanda's avatar
longpanda committed
229
230
231
232
233
234
235
        vt_windows_count_wim_patch vt_wim_cnt
        if [ $vt_wim_cnt -eq 0 ]; then
            distro_specify_wim_patch_phase2
        fi
        
        ventoy_debug_pause        
        locate_wim
longpanda's avatar
bug fix  
longpanda committed
236
237
    fi

longpanda's avatar
longpanda committed
238
239
240
    vt_windows_chain_data ${1}${chosen_path}    
    ventoy_debug_pause
    
longpanda's avatar
longpanda committed
241
    if [ -n "$vtoy_chain_mem_addr" ]; then
longpanda's avatar
update  
longpanda committed
242
        ventoy_cli_console
longpanda's avatar
longpanda committed
243
244
245
246
        chainloader ${vtoy_path}/ventoy_x64.efi  env_param=${env_param} isoefi=${LoadIsoEfiDriver} ${vtdebug_flag} mem:${vtoy_chain_mem_addr}:size:${vtoy_chain_mem_size}
        boot
    else
        echo "chain empty failed"
longpanda's avatar
longpanda committed
247
        ventoy_pause
longpanda's avatar
longpanda committed
248
249
250
251
252
    fi
}

function uefi_linux_menu_func {
    if [ "$ventoy_compatible" = "NO" ]; then    
longpanda's avatar
longpanda committed
253
254
255
256
257
258
259
        
        if [ "$ventoy_fs_probe" = "udf" ]; then
            loopback -d loop            
            set ventoy_fs_probe=iso9660
            loopback loop $1$2
        fi
        
longpanda's avatar
longpanda committed
260
        vt_load_cpio  ${vtoy_path}/ventoy.cpio   $2 $1
longpanda's avatar
longpanda committed
261
262
263

        vt_linux_clear_initrd
        
longpanda's avatar
longpanda committed
264
265
266
        if [ -d (loop)/pmagic ]; then
            vt_linux_specify_initrd_file /pmagic/initrd.img
        else
longpanda's avatar
longpanda committed
267
268
            for file in "boot/grub/grub.cfg" "EFI/BOOT/grub.cfg" "EFI/boot/grub.cfg" "efi/boot/grub.cfg" "EFI/BOOT/BOOTX64.conf" "/grub/grub.cfg"; do
                if [ -e (loop)/$file ]; then                    
longpanda's avatar
longpanda committed
269
270
271
272
                    vt_linux_parse_initrd_grub  file  (loop)/$file
                fi
            done
        fi
longpanda's avatar
longpanda committed
273

longpanda's avatar
longpanda committed
274
        # special process for special distros
longpanda's avatar
longpanda committed
275
276
277
278
        if [ -d (loop)/loader/entries ]; then
            vt_linux_parse_initrd_grub  dir  (loop)/loader/entries/
        elif [ -d (loop)/boot/grub ]; then
            vt_linux_parse_initrd_grub  dir  (loop)/boot/grub/
longpanda's avatar
longpanda committed
279
280
        fi
        
longpanda's avatar
longpanda committed
281
        distro_specify_initrd_file
longpanda's avatar
longpanda committed
282
        
longpanda's avatar
longpanda committed
283
284
285
286
287
288
289
290
291
292
293
        vt_linux_initrd_count vtcount
        if [ $vtcount -eq 0 ]; then
            distro_specify_initrd_file_phase2
            
            if [ "$vt_efi_dir" = "NO" ]; then
                if [ -f (loop)/efi.img ];  then
                    vt_add_replace_file 0 "initrd"
                fi
            fi
        fi
        
longpanda's avatar
longpanda committed
294
        locate_initrd
longpanda's avatar
longpanda committed
295
296
297
298
299
300
301
302
303
304
305
306
        
        if [ -d (loop)/loader/entries ]; then
            vt_linux_get_main_initrd_index vtindex
            
            if [ -d (loop)/arch ]; then
                if [ -f (loop)/arch/boot/x86_64/archiso.img ]; then
                    vt_add_replace_file $vtindex "EFI\\archiso\\archiso.img"
                elif [ -f (loop)/boot/initramfs_x86_64.img ]; then
                    vt_add_replace_file $vtindex "boot\\initramfs_x86_64.img"
                fi
            elif [ -f (loop)/EFI/BOOT/initrd.gz ]; then
                vt_add_replace_file $vtindex "EFI\\BOOT\\initrd.gz"
longpanda's avatar
longpanda committed
307
308
            elif [ -f (loop)/loader/entries/thinstation.conf ]; then
                vt_add_replace_file $vtindex "boot\\initrd"
longpanda's avatar
longpanda committed
309
            fi
longpanda's avatar
longpanda committed
310
311
312
313
        elif [ -d (loop)/EFI/boot/entries ]; then
            if [ -f (loop)/parabola/boot/x86_64/parabolaiso.img ]; then
                vt_add_replace_file 0 "EFI\\parabolaiso\\parabolaiso.img"
            fi
longpanda's avatar
longpanda committed
314
315
316
317
318
        elif [ -e (loop)/syslinux/alt0/full.cz ]; then
            vt_add_replace_file 0 "EFI\\BOOT\\full.cz"            
            set FirstTryBootFile='@EFI@BOOT@grubx64.efi'
        fi
        
longpanda's avatar
longpanda committed
319
320
321
322
323
    fi
    
    vt_linux_chain_data ${1}${chosen_path}

    if [ -n "$vtoy_chain_mem_addr" ]; then
longpanda's avatar
update  
longpanda committed
324
        ventoy_cli_console       
longpanda's avatar
longpanda committed
325
        chainloader ${vtoy_path}/ventoy_x64.efi  env_param=${env_param} isoefi=${LoadIsoEfiDriver} FirstTry=${FirstTryBootFile} ${vtdebug_flag} mem:${vtoy_chain_mem_addr}:size:${vtoy_chain_mem_size}
longpanda's avatar
longpanda committed
326
327
328
        boot
    else
        echo "chain empty failed"
longpanda's avatar
longpanda committed
329
        ventoy_pause
longpanda's avatar
longpanda committed
330
331
332
333
334
335
336
337
338
339
    fi
}


function uefi_iso_menu_func {

    if [ -d (loop)/ ]; then
        loopback -d loop
    fi

longpanda's avatar
bug fix  
longpanda committed
340
341
    if [ -n "$vtisouefi" ]; then
        set LoadIsoEfiDriver=on
longpanda's avatar
longpanda committed
342
        unset vtisouefi
longpanda's avatar
longpanda committed
343
    elif vt_check_mode 2; then
longpanda's avatar
longpanda committed
344
        set LoadIsoEfiDriver=on
longpanda's avatar
bug fix  
longpanda committed
345
346
347
348
    else
        unset LoadIsoEfiDriver
    fi

longpanda's avatar
longpanda committed
349
    vt_chosen_img_path chosen_path
350
351
    vt_select_auto_install ${chosen_path}
    vt_select_persistence ${chosen_path}
longpanda's avatar
longpanda committed
352
353
354
355
356

    if vt_is_udf ${1}${chosen_path}; then
        set ventoy_fs_probe=udf
    else
        set ventoy_fs_probe=iso9660
longpanda's avatar
longpanda committed
357
        vt_iso9660_nojoliet 0
longpanda's avatar
longpanda committed
358
359
360
361
362
    fi

    loopback loop ${1}${chosen_path}
    get_os_type (loop)
    
longpanda's avatar
longpanda committed
363
364
365
366
367
368
369
370
    if [ -d (loop)/EFI ]; then
        set vt_efi_dir=YES
    elif [ -d (loop)/efi ]; then
        set vt_efi_dir=YES
    else
        set vt_efi_dir=NO
    fi
    
longpanda's avatar
longpanda committed
371
372
373
    if [ -n "$vtcompat" ]; then
        set ventoy_compatible=YES
        unset vtcompat
longpanda's avatar
longpanda committed
374
    elif vt_check_mode 1; then
longpanda's avatar
longpanda committed
375
376
377
378
        set ventoy_compatible=YES
    else
        vt_check_compatible (loop)
    fi
longpanda's avatar
longpanda committed
379
380
381
382
    
    vt_img_sector ${1}${chosen_path}
    
    if [ "$vtoy_os" = "Windows" ]; then
longpanda's avatar
longpanda committed
383
        vt_check_compatible_pe (loop)        
longpanda's avatar
longpanda committed
384
        uefi_windows_menu_func  $1 ${chosen_path}
longpanda's avatar
longpanda committed
385
    else
longpanda's avatar
longpanda committed
386
        uefi_linux_menu_func  $1 ${chosen_path}
longpanda's avatar
longpanda committed
387
388
    fi

longpanda's avatar
update  
longpanda committed
389
    ventoy_gui_console
longpanda's avatar
longpanda committed
390
391
}

longpanda's avatar
longpanda committed
392
393
394
395
396
397
function uefi_iso_memdisk {    
    vt_chosen_img_path chosen_path

    echo 'Loading ISO file to memory ...'
    vt_load_iso_to_mem ${1}${chosen_path} vtoy_iso_buf
    
longpanda's avatar
update  
longpanda committed
398
    ventoy_cli_console
longpanda's avatar
longpanda committed
399
400
    chainloader ${vtoy_path}/ventoy_x64.efi memdisk env_param=${env_param} isoefi=${LoadIsoEfiDriver} ${vtdebug_flag} mem:${vtoy_iso_buf_addr}:size:${vtoy_iso_buf_size}
    boot
longpanda's avatar
longpanda committed
401
    
longpanda's avatar
update  
longpanda committed
402
    ventoy_gui_console
longpanda's avatar
longpanda committed
403
404
405
}


longpanda's avatar
longpanda committed
406
407
408
function legacy_windows_menu_func {
    vt_windows_reset
    
longpanda's avatar
longpanda committed
409
410
411
412
413
414
    if [ "$ventoy_compatible" = "NO" ]; then   
        
        if [ "$ventoy_fs_probe" = "iso9660" ]; then
            loopback -d loop
            vt_iso9660_nojoliet 1
            loopback loop $1$2
longpanda's avatar
longpanda committed
415
        fi
longpanda's avatar
longpanda committed
416
417
418
419
420
421
422
423
424
425
426
427
428
429
        
        for file in "boot/bcd" "/efi/microsoft/boot/bcd" "SSTR/BCD"; do
            vt_windows_collect_wim_patch bcd (loop)/$file                
        done
        
        distro_specify_wim_patch

        vt_windows_count_wim_patch vt_wim_cnt
        if [ $vt_wim_cnt -eq 0 ]; then
            distro_specify_wim_patch_phase2
        fi
        
        ventoy_debug_pause        
        locate_wim
longpanda's avatar
longpanda committed
430
    fi
longpanda's avatar
longpanda committed
431
432
433

    vt_windows_chain_data ${1}${chosen_path}    
    ventoy_debug_pause    
longpanda's avatar
longpanda committed
434
435
    
    if [ -n "$vtoy_chain_mem_addr" ]; then
longpanda's avatar
longpanda committed
436
        linux16   $vtoy_path/ipxe.krn ${vtdebug_flag} ibft mem:${vtoy_chain_mem_addr}:size:${vtoy_chain_mem_size}
longpanda's avatar
longpanda committed
437
438
439
        boot
    else
        echo "chain empty failed"
longpanda's avatar
longpanda committed
440
        ventoy_pause
longpanda's avatar
longpanda committed
441
442
443
444
445
    fi
}

function legacy_linux_menu_func {
    if [ "$ventoy_compatible" = "NO" ]; then
longpanda's avatar
longpanda committed
446
447
448
449
450
451
452

        if [ "$ventoy_fs_probe" = "udf" ]; then
            loopback -d loop
            set ventoy_fs_probe=iso9660
            loopback loop $1$2
        fi
    
longpanda's avatar
longpanda committed
453
        vt_load_cpio  $vtoy_path/ventoy.cpio  $2 $1
longpanda's avatar
longpanda committed
454
455
456

        vt_linux_clear_initrd
        
longpanda's avatar
longpanda committed
457
458
459
460
461
462
463
464
465
        if [ -d (loop)/pmagic ]; then
            vt_linux_specify_initrd_file /pmagic/initrd.img
        else
            for dir in "isolinux" "boot/isolinux" "boot/x86_64/loader" "syslinux" "boot/syslinux"; do
                if [ -d (loop)/$dir ]; then
                    vt_linux_parse_initrd_isolinux   (loop)/$dir/
                fi
            done
        fi
longpanda's avatar
longpanda committed
466
467
        
        # special process for special distros
longpanda's avatar
longpanda committed
468
469
470
471
472
473
474
475
476
        #archlinux
        if [ -d (loop)/arch/boot/syslinux ]; then
            vt_linux_parse_initrd_isolinux   (loop)/arch/boot/syslinux/  /arch/
            vt_linux_parse_initrd_isolinux   (loop)/arch/boot/syslinux/  /arch/boot/syslinux/
            
        #manjaro
        elif [ -d (loop)/manjaro ]; then
            if [ -e (loop)/boot/grub/kernels.cfg ]; then
                vt_linux_parse_initrd_grub  file  (loop)/boot/grub/kernels.cfg
longpanda's avatar
longpanda committed
477
            fi
longpanda's avatar
longpanda committed
478
479
        elif [ -e (loop)/boot/grub/grub.cfg ]; then                
            vt_linux_parse_initrd_grub  file  (loop)/boot/grub/grub.cfg
longpanda's avatar
longpanda committed
480
481
        fi
        
longpanda's avatar
longpanda committed
482
        distro_specify_initrd_file
longpanda's avatar
longpanda committed
483
        
longpanda's avatar
longpanda committed
484
485
486
487
488
        vt_linux_initrd_count vtcount
        if [ $vtcount -eq 0 ]; then
            distro_specify_initrd_file_phase2
        fi
        
longpanda's avatar
longpanda committed
489
490
491
        locate_initrd
    fi
    
longpanda's avatar
longpanda committed
492
493
    vt_linux_chain_data ${1}${chosen_path}    
    ventoy_debug_pause
longpanda's avatar
longpanda committed
494
    
longpanda's avatar
longpanda committed
495
496
    if [ -n "$vtoy_chain_mem_addr" ]; then
        linux16   $vtoy_path/ipxe.krn ${vtdebug_flag}  mem:${vtoy_chain_mem_addr}:size:${vtoy_chain_mem_size}
longpanda's avatar
longpanda committed
497
498
499
        boot
    else
        echo "chain empty failed"
longpanda's avatar
longpanda committed
500
        ventoy_pause
longpanda's avatar
longpanda committed
501
502
503
504
505
506
507
508
509
510
    fi
}

function legacy_iso_menu_func {

    if [ -d (loop)/ ]; then
        loopback -d loop
    fi

    vt_chosen_img_path chosen_path
511
512
    vt_select_auto_install ${chosen_path}
    vt_select_persistence ${chosen_path}
longpanda's avatar
longpanda committed
513
514
515
516
517

    if vt_is_udf ${1}${chosen_path}; then
        set ventoy_fs_probe=udf
    else
        set ventoy_fs_probe=iso9660
longpanda's avatar
longpanda committed
518
519
        vt_iso9660_nojoliet 0
    fi
longpanda's avatar
longpanda committed
520
    
longpanda's avatar
longpanda committed
521
    loopback loop ${1}${chosen_path}
longpanda's avatar
longpanda committed
522
    get_os_type (loop)
longpanda's avatar
longpanda committed
523

longpanda's avatar
longpanda committed
524
525
526
    if [ -n "$vtcompat" ]; then
        set ventoy_compatible=YES
        unset vtcompat
longpanda's avatar
longpanda committed
527
    elif vt_check_mode 1; then
longpanda's avatar
longpanda committed
528
529
530
531
        set ventoy_compatible=YES
    else
        vt_check_compatible (loop)
    fi
longpanda's avatar
longpanda committed
532
533
    
    vt_img_sector ${1}${chosen_path}
longpanda's avatar
bug fix  
longpanda committed
534

longpanda's avatar
longpanda committed
535
    if [ "$vtoy_os" = "Windows" ]; then
longpanda's avatar
longpanda committed
536
        vt_check_compatible_pe (loop)        
longpanda's avatar
longpanda committed
537
        legacy_windows_menu_func  $1 ${chosen_path}
longpanda's avatar
longpanda committed
538
    else
longpanda's avatar
longpanda committed
539
        legacy_linux_menu_func  $1 ${chosen_path}
longpanda's avatar
longpanda committed
540
541
542
    fi
}

longpanda's avatar
longpanda committed
543
544
function legacy_iso_memdisk {
    vt_chosen_img_path chosen_path
longpanda's avatar
longpanda committed
545

longpanda's avatar
longpanda committed
546
547
548
549
550
    linux16   $vtoy_path/memdisk iso raw    
    echo "Loading ISO file to memory ..."
    initrd16  ${1}${chosen_path}
    boot
}
longpanda's avatar
longpanda committed
551

552
function iso_common_menuentry {
longpanda's avatar
longpanda committed
553
554
    if [ "$grub_platform" = "pc" ]; then
        if vt_check_mode 0; then
longpanda's avatar
longpanda committed
555
            legacy_iso_memdisk $vtoy_iso_part
longpanda's avatar
longpanda committed
556
        else
longpanda's avatar
longpanda committed
557
            legacy_iso_menu_func $vtoy_iso_part
longpanda's avatar
longpanda committed
558
559
560
        fi
    else
        if vt_check_mode 0; then
longpanda's avatar
longpanda committed
561
            uefi_iso_memdisk $vtoy_iso_part 
longpanda's avatar
longpanda committed
562
        else
longpanda's avatar
longpanda committed
563
            uefi_iso_menu_func $vtoy_iso_part                
longpanda's avatar
longpanda committed
564
565
566
567
        fi
    fi      
}

longpanda's avatar
longpanda committed
568
function common_unsupport_menuentry {
569
    echo -e "\n The name of the iso file could NOT contain space or non-ascii characters. \n"
longpanda's avatar
longpanda committed
570
571
572
    echo -e " 文件名中不能有中文或空格 \n"    
    echo -e "\npress ENTER to exit (请按 回车 键返回) ..."    
    read vtInputKey
573
574
}

longpanda's avatar
longpanda committed
575
576
577
function iso_unsupport_menuentry {
    common_unsupport_menuentry
}
578
579
580

function wim_common_menuentry {
    vt_chosen_img_path chosen_path
longpanda's avatar
longpanda committed
581
    vt_wim_chain_data ${vtoy_iso_part}${chosen_path}
longpanda's avatar
longpanda committed
582
583
    
    ventoy_debug_pause    
584
585
586
587
588
    
    if [ -n "$vtoy_chain_mem_addr" ]; then
        if [ "$grub_platform" = "pc" ]; then
            linux16   $vtoy_path/ipxe.krn ${vtdebug_flag}  mem:${vtoy_chain_mem_addr}:size:${vtoy_chain_mem_size}
        else
longpanda's avatar
update  
longpanda committed
589
            ventoy_cli_console
590
            chainloader ${vtoy_path}/ventoy_x64.efi  env_param=${env_param} isoefi=${LoadIsoEfiDriver} ${vtdebug_flag} mem:${vtoy_chain_mem_addr}:size:${vtoy_chain_mem_size}
longpanda's avatar
update  
longpanda committed
591
            ventoy_gui_console
592
593
594
595
        fi
        boot
    else
        echo "chain empty failed"
longpanda's avatar
longpanda committed
596
        ventoy_pause
597
598
599
600
    fi
}

function wim_unsupport_menuentry {
longpanda's avatar
longpanda committed
601
602
603
604
605
606
    common_unsupport_menuentry
}

function efi_common_menuentry {
    vt_chosen_img_path chosen_path
    
longpanda's avatar
update  
longpanda committed
607
    ventoy_cli_console    
longpanda's avatar
longpanda committed
608
    chainloader ${vtoy_iso_part}${chosen_path}
longpanda's avatar
longpanda committed
609
    boot
longpanda's avatar
update  
longpanda committed
610
    ventoy_gui_console
longpanda's avatar
longpanda committed
611
612
613
614
}

function efi_unsupport_menuentry {
    common_unsupport_menuentry
615
}
longpanda's avatar
longpanda committed
616

longpanda's avatar
longpanda committed
617
618


longpanda's avatar
longpanda committed
619
620
621
622
623
624
625
626
#############################################################
#############################################################
#############################################################
#######                 Main Process              ###########
#############################################################
#############################################################
#############################################################

longpanda's avatar
longpanda committed
627
set VENTOY_VERSION="1.0.16"
longpanda's avatar
longpanda committed
628
629
630
631
632

# Default menu display mode, you can change it as you want.
#    0: List mode   
#    1: TreeView mode
set VTOY_DEFAULT_MENU_MODE=0
longpanda's avatar
longpanda committed
633

634
set VTOY_MEM_DISK_STR="[Memdisk]"
longpanda's avatar
longpanda committed
635
set VTOY_ISO_RAW_STR="Compatible Mode"
longpanda's avatar
longpanda committed
636
637
set VTOY_ISO_UEFI_DRV_STR="UEFI FS"

longpanda's avatar
longpanda committed
638
set VTOY_F2_CMD="ventoy_power"
longpanda's avatar
longpanda committed
639
640
set VTOY_F4_CMD="ventoy_localboot"
set VTOY_F5_CMD="ventoy_diagnosis"
longpanda's avatar
longpanda committed
641
set VTOY_F6_CMD="ventoy_ext_menu"
longpanda's avatar
longpanda committed
642
643
644
645
646
647

if [ "$grub_platform" = "pc" ]; then
    set VTOY_TEXT_MENU_VER="Ventoy $VENTOY_VERSION BIOS  www.ventoy.net"
else
    set VTOY_TEXT_MENU_VER="Ventoy $VENTOY_VERSION UEFI  www.ventoy.net"
fi
longpanda's avatar
longpanda committed
648

longpanda's avatar
longpanda committed
649
650
651
vt_device $root  vtoy_dev

if [ "$vtoy_dev" = "tftp" ]; then
longpanda's avatar
longpanda committed
652
    set vtoy_path=($root)
longpanda's avatar
longpanda committed
653
    for vtid in 0 1 2 3; do
longpanda's avatar
longpanda committed
654
        if [ -d (hd$vtid,2)/ventoy ]; then
longpanda's avatar
longpanda committed
655
            set vtoy_iso_part=(hd$vtid,1)
656
            set vtoy_efi_part=(hd$vtid,2)
longpanda's avatar
longpanda committed
657
658
659
            break
        fi
    done
longpanda's avatar
longpanda committed
660
    loadfont ascii
longpanda's avatar
longpanda committed
661

longpanda's avatar
longpanda committed
662
663
    if [ -f $vtoy_iso_part/ventoy/ventoy.json ]; then
       set vt_plugin_path=$vtoy_iso_part
longpanda's avatar
longpanda committed
664
665
666
667
    else
        set vt_plugin_path=$prefix
        vt_load_plugin $vt_plugin_path
    fi
longpanda's avatar
longpanda committed
668
else
longpanda's avatar
longpanda committed
669
670
671
672
673
674
    if [ "$prefix" = "(ventoydisk)/grub" ]; then
        set vtoy_path=(ventoydisk)/ventoy
    else
        set vtoy_path=($root)/ventoy
    fi

longpanda's avatar
longpanda committed
675
    set vtoy_iso_part=($vtoy_dev,1)
676
    set vtoy_efi_part=($vtoy_dev,2)
longpanda's avatar
longpanda committed
677
    loadfont unicode
longpanda's avatar
longpanda committed
678
    set vt_plugin_path=$vtoy_iso_part
longpanda's avatar
longpanda committed
679
680
681
fi


682
#Load Plugin
longpanda's avatar
longpanda committed
683
684
if [ -f $vtoy_iso_part/ventoy/ventoy.json ]; then
   vt_load_plugin $vtoy_iso_part
longpanda's avatar
longpanda committed
685
686
fi

longpanda's avatar
longpanda committed
687
688
689
690
691
692
if [ -n "$VTOY_MENU_TIMEOUT" ]; then
    set timeout=$VTOY_MENU_TIMEOUT
else
    unset timeout
fi

longpanda's avatar
longpanda committed
693
694
if [ -f $vtoy_iso_part/ventoy/ventoy_wimboot.img ]; then
    vt_load_wimboot $vtoy_iso_part/ventoy/ventoy_wimboot.img
695
696
697
698
elif [ -f $vtoy_efi_part/ventoy/ventoy_wimboot.img ]; then
    vt_load_wimboot $vtoy_efi_part/ventoy/ventoy_wimboot.img
fi

699
700
701

if [ $VTOY_DEFAULT_MENU_MODE -eq 0 ]; then
    set VTOY_F3_CMD="vt_dynamic_menu 1 1"
longpanda's avatar
longpanda committed
702
    set VTOY_HOTKEY_TIP="F1:Memdisk  F2:Power  F3:TreeView  F4:Localboot  F5:Debug  F6:ExMenu"
703
704
else
    set VTOY_F3_CMD="vt_dynamic_menu 1 0"
longpanda's avatar
longpanda committed
705
    set VTOY_HOTKEY_TIP="F1:Memdisk  F2:Power  F3:ListView  F4:Localboot  F5:Debug  F6:ExMenu"
706
707
708
fi


longpanda's avatar
longpanda committed
709
710
711
712
713
if [ -n "$vtoy_gfxmode" ]; then
    set gfxmode=$vtoy_gfxmode
else
    set gfxmode=1920x1080,1366x768,1024x768
fi
longpanda's avatar
longpanda committed
714

longpanda's avatar
longpanda committed
715
716
if [ "$vtoy_display_mode" = "CLI" ]; then
    terminal_output  console
longpanda's avatar
update  
longpanda committed
717
718
719
720
721
722
else    
    if [ -n "$vtoy_theme" ]; then
        set theme=$vtoy_theme
    else
        set theme=$prefix/themes/ventoy/theme.txt
    fi
longpanda's avatar
longpanda committed
723
724
725
    terminal_output  gfxterm
fi

longpanda's avatar
longpanda committed
726
727
728
729
730
731
732
733
734
735
736
#export necessary variable
export theme
export gfxmode
export vtoy_dev
export vtoy_iso_part
export vtoy_efi_part
export VENTOY_VERSION




longpanda's avatar
longpanda committed
737
738
#colect all image files (iso files)
set ventoy_img_count=0
longpanda's avatar
longpanda committed
739
vt_list_img $vtoy_iso_part ventoy_img_count
longpanda's avatar
longpanda committed
740

longpanda's avatar
longpanda committed
741
742
#Main menu
if [ $ventoy_img_count -gt 0 ]; then
longpanda's avatar
longpanda committed
743
744
745
746
747
    if [ $VTOY_DEFAULT_MENU_MODE -eq 0 ]; then
        vt_dynamic_menu 0 0 
    else
        vt_dynamic_menu 0 1
    fi
longpanda's avatar
longpanda committed
748
else
749
750
751
752
753
754
755
756
    if [ -n "$VTOY_NO_ISO_TIP" ]; then
        NO_ISO_MENU="No ISO files found, $VTOY_NO_ISO_TIP"
    elif [ -n "$VTOY_DEFAULT_SEARCH_ROOT" ]; then
        NO_ISO_MENU="No ISO files found, please check VTOY_DEFAULT_SEARCH_ROOT"
    else
        NO_ISO_MENU="No ISO files found"
    fi
    menuentry "$NO_ISO_MENU (Press enter to reboot ...)" {
longpanda's avatar
longpanda committed
757
758
759
760
        echo -e "\n    Rebooting ... "
        reboot
    }
fi