VentoyWorker.sh 18.7 KB
Newer Older
1
2
3
4
5
#!/bin/sh

. ./tool/ventoy_lib.sh

print_usage() {
6

7
8
    echo 'Usage:  Ventoy2Disk.sh CMD [ OPTION ] /dev/sdX'
    echo '  CMD:'
9
    echo '   -i  install Ventoy to sdX (fails if disk already installed with Ventoy)'
10
    echo '   -I  force install Ventoy to sdX (no matter if installed or not)'
11
    echo '   -u  update Ventoy in sdX'
longpanda's avatar
longpanda committed
12
    echo '   -l  list Ventoy information in sdX'
13
14
    echo ''
    echo '  OPTION: (optional)'
longpanda's avatar
longpanda committed
15
    echo '   -r SIZE_MB  preserve some space at the bottom of the disk (only for install)'
longpanda's avatar
longpanda committed
16
    echo '   -s/-S       enable/disable secure boot support (default is disabled)'
longpanda's avatar
longpanda committed
17
    echo '   -g          use GPT partition style, default is MBR (only for install)'
18
    echo '   -L          Label of the 1st exfat partition (default is Ventoy)'
longpanda's avatar
longpanda committed
19
    echo '   -n          try non-destructive installation (only for install)'
20
21
22
    echo ''
}

longpanda's avatar
longpanda committed
23

24
VTNEW_LABEL='Ventoy'
longpanda's avatar
longpanda committed
25
RESERVE_SIZE_MB=0
26
27
28
29
30
31
while [ -n "$1" ]; do
    if [ "$1" = "-i" ]; then
        MODE="install"
    elif [ "$1" = "-I" ]; then
        MODE="install"
        FORCE="Y"
longpanda's avatar
longpanda committed
32
33
    elif [ "$1" = "-n" ]; then
        NONDESTRUCTIVE="Y"
34
35
    elif [ "$1" = "-u" ]; then
        MODE="update"
longpanda's avatar
longpanda committed
36
37
    elif [ "$1" = "-l" ]; then
        MODE="list"
38
39
    elif [ "$1" = "-s" ]; then
        SECUREBOOT="YES"
longpanda's avatar
longpanda committed
40
41
    elif [ "$1" = "-S" ]; then
        SECUREBOOT="NO"
longpanda's avatar
longpanda committed
42
43
    elif [ "$1" = "-g" ]; then
        VTGPT="YES"
longpanda's avatar
longpanda committed
44
45
46
    elif [ "$1" = "-L" ]; then
        shift
        VTNEW_LABEL=$1
longpanda's avatar
longpanda committed
47
48
49
50
    elif [ "$1" = "-r" ]; then
        RESERVE_SPACE="YES"
        shift
        RESERVE_SIZE_MB=$1
longpanda's avatar
longpanda committed
51
52
    elif [ "$1" = "-V" ] || [ "$1" = "--version" ]; then
        exit 0
longpanda's avatar
longpanda committed
53
    elif [ "$1" = "-h" ] || [ "$1" = "--help" ]; then
longpanda's avatar
longpanda committed
54
55
        print_usage
        exit 0
56
57
58
59
60
61
62
    else
        if ! [ -b "$1" ]; then
            vterr "$1 is NOT a valid device"
            print_usage
            exit 1
        fi
        DISK=$1
63
64
65
66
67
68
        # Resolve symlinks now, will be needed to look up information about the device in
        # the /sys/ filesystem, for example /sys/class/block/${DISK#/dev/}/start
        # The main use case is supporting /dev/disk/by-id/ symlinks instead of raw devices
        if [ -L "$DISK" ]; then
            DISK=$(readlink -e -n "$DISK")
        fi
69
    fi
70

71
72
73
74
75
76
77
78
79
80
81
82
83
84
    shift
done

if [ -z "$MODE" ]; then
    print_usage
    exit 1
fi

if ! [ -b "$DISK" ]; then
    vterr "Disk $DISK does not exist"
    exit 1
fi

if [ -e /sys/class/block/${DISK#/dev/}/start ]; then
longpanda's avatar
longpanda committed
85
86
    vterr  "$DISK is a partition, please use the whole disk."
    echo   "For example:"
longpanda's avatar
longpanda committed
87
88
    vterr  "    sudo sh Ventoy2Disk.sh -i /dev/sdb1 <=== This is wrong"
    vtinfo "    sudo sh Ventoy2Disk.sh -i /dev/sdb  <=== This is right"
longpanda's avatar
longpanda committed
89
    echo ""
90
91
92
    exit 1
fi

longpanda's avatar
longpanda committed
93
if [ -n "$RESERVE_SPACE" -a "$MODE" = "install" ]; then
longpanda's avatar
longpanda committed
94
95
96
97
98
99
100
101
102
    if echo $RESERVE_SIZE_MB | grep -q '^[0-9][0-9]*$'; then
        vtdebug "User will reserve $RESERVE_SIZE_MB MB disk space"
    else
        vterr "$RESERVE_SIZE_MB is invalid for reserved space"
        exit 1
    fi
fi

vtdebug "MODE=$MODE FORCE=$FORCE RESERVE_SPACE=$RESERVE_SPACE RESERVE_SIZE_MB=$RESERVE_SIZE_MB"
103

longpanda's avatar
longpanda committed
104
105
106
107
#check tools
if check_tool_work_ok; then
    vtdebug "check tool work ok"
else
108
    vterr "Some tools can not run on current system. Please check log.txt for details."
109
110
111
    exit 1
fi

longpanda's avatar
longpanda committed
112
113
114
115
if [ "$MODE" = "list" ]; then
    version=$(get_disk_ventoy_version $DISK)
    if [ $? -eq 0 ]; then
        echo "Ventoy Version in Disk: $version"
116

longpanda's avatar
longpanda committed
117
        vtPart1Type=$(dd if=$DISK bs=1 count=1 skip=450 status=none | hexdump -n1 -e  '1/1 "%02X"')
118
        if [ "$vtPart1Type" = "EE" ]; then
longpanda's avatar
longpanda committed
119
120
121
122
            echo "Disk Partition Style  : GPT"
        else
            echo "Disk Partition Style  : MBR"
        fi
123

124
        if check_disk_secure_boot $DISK; then
longpanda's avatar
longpanda committed
125
126
127
            echo "Secure Boot Support   : YES"
        else
            echo "Secure Boot Support   : NO"
128
        fi
longpanda's avatar
longpanda committed
129
130
131
132
133
134
135
    else
        echo "Ventoy Version: NA"
    fi
    echo ""
    exit 0
fi

longpanda's avatar
longpanda committed
136
#check mountpoint
longpanda's avatar
longpanda committed
137
check_umount_disk "$DISK"
138
139
140
141
142
143

if grep "$DISK" /proc/mounts; then
    vterr "$DISK is already mounted, please umount it first!"
    exit 1
fi

longpanda's avatar
longpanda committed
144
145
146
147
148
149
#check swap partition
if swapon --help 2>&1 | grep -q '^ \-s,'; then
    if swapon -s | grep -q "^${DISK}[0-9]"; then
        vterr "$DISK is used as swap, please swapoff it first!"
        exit 1
    fi
150
151
fi

152
#check access
longpanda's avatar
longpanda committed
153
154
155
156
157
158
159
160
161
if dd if="$DISK" of=/dev/null bs=1 count=1 >/dev/null 2>&1; then
    vtdebug "root permission check ok ..."
else
    vterr "Failed to access $DISK, maybe root privilege is needed!"
    echo ''
    exit 1
fi


longpanda's avatar
longpanda committed
162
163
164
165
166
167
#check tmp_mnt directory
if [ -d ./tmp_mnt ]; then
    vtdebug "There is a tmp_mnt directory, now delete it."
    umount ./tmp_mnt >/dev/null 2>&1
    rm -rf ./tmp_mnt
    if [ -d ./tmp_mnt ]; then
168
        vterr "tmp_mnt directory exists, please delete it first."
longpanda's avatar
longpanda committed
169
170
171
172
        exit 1
    fi
fi

173

longpanda's avatar
longpanda committed
174
if [ "$MODE" = "install" -a -z "$NONDESTRUCTIVE" ]; then
175
    vtdebug "install Ventoy ..."
176

longpanda's avatar
longpanda committed
177
178
179
180
    if [ -n "$VTGPT" ]; then
        if parted -v > /dev/null 2>&1; then
            PARTTOOL='parted'
        else
longpanda's avatar
longpanda committed
181
182
            vterr "parted is not found in the system, Ventoy can't create new partitions without it."
            vterr "You should install \"GNU parted\" first."
longpanda's avatar
longpanda committed
183
184
            exit 1
        fi
185
    else
longpanda's avatar
longpanda committed
186
187
188
189
190
        if parted -v > /dev/null 2>&1; then
            PARTTOOL='parted'
        elif fdisk -v >/dev/null 2>&1; then
            PARTTOOL='fdisk'
        else
longpanda's avatar
longpanda committed
191
            vterr "Both parted and fdisk are not found in the system, Ventoy can't create new partitions."
longpanda's avatar
longpanda committed
192
193
            exit 1
        fi
194
    fi
195

196
197
198
199
200
    version=$(get_disk_ventoy_version $DISK)
    if [ $? -eq 0 ]; then
        if [ -z "$FORCE" ]; then
            vtwarn "$DISK already contains a Ventoy with version $version"
            vtwarn "Use -u option to do a safe upgrade operation."
201
            vtwarn "OR if you really want to reinstall Ventoy to $DISK, please use -I option."
202
203
204
205
            vtwarn ""
            exit 1
        fi
    fi
206

207
208
209
    disk_sector_num=$(cat /sys/block/${DISK#/dev/}/size)
    disk_size_gb=$(expr $disk_sector_num / 2097152)

longpanda's avatar
update  
longpanda committed
210
    if [ $disk_sector_num -gt 4294967296 ] && [ -z "$VTGPT" ]; then
211
212
213
214
        vterr "$DISK is over 2TB size, MBR will not work on it."
        exit 1
    fi

longpanda's avatar
longpanda committed
215
216
217
    if [ -n "$RESERVE_SPACE" ]; then
        sum_size_mb=$(expr $RESERVE_SIZE_MB + $VENTOY_PART_SIZE_MB)
        reserve_sector_num=$(expr $sum_size_mb \* 2048)
218

longpanda's avatar
longpanda committed
219
220
221
222
223
224
        if [ $disk_sector_num -le $reserve_sector_num ]; then
            vterr "Can't reserve $RESERVE_SIZE_MB MB space from $DISK"
            exit 1
        fi
    fi

225
226
227
    #Print disk info
    echo "Disk : $DISK"
    parted -s $DISK p 2>&1 | grep Model
228
    echo "Size : $disk_size_gb GB"
longpanda's avatar
longpanda committed
229
230
231
232
    if [ -n "$VTGPT" ]; then
        echo "Style: GPT"
    else
        echo "Style: MBR"
233
    fi
234
235
    echo ''

longpanda's avatar
longpanda committed
236
237
238
239
240
    if [ -n "$RESERVE_SPACE" ]; then
        echo "You will reserve $RESERVE_SIZE_MB MB disk space "
    fi
    echo ''

241
242
243
244
245
    vtwarn "Attention:"
    vtwarn "You will install Ventoy to $DISK."
    vtwarn "All the data on the disk $DISK will be lost!!!"
    echo ""

longpanda's avatar
longpanda committed
246
    read -p 'Continue? (y/n) '  Answer
247
248
249
250
251
252
253
254
    if [ "$Answer" != "y" ]; then
        if [ "$Answer" != "Y" ]; then
            exit 0
        fi
    fi

    echo ""
    vtwarn "All the data on the disk $DISK will be lost!!!"
longpanda's avatar
longpanda committed
255
    read -p 'Double-check. Continue? (y/n) '  Answer
256
257
258
259
260
261
    if [ "$Answer" != "y" ]; then
        if [ "$Answer" != "Y" ]; then
            exit 0
        fi
    fi

262
    if [ $disk_sector_num -le $VENTOY_SECTOR_NUM ]; then
263
264
265
266
        vterr "No enough space in disk $DISK"
        exit 1
    fi

longpanda's avatar
longpanda committed
267
268
269
270
    # check and umount
    check_umount_disk "$DISK"

    if ! dd if=/dev/zero of=$DISK bs=64 count=512 status=none conv=fsync; then
271
272
273
274
        vterr "Write data to $DISK failed, please check whether it's in use."
        exit 1
    fi

longpanda's avatar
longpanda committed
275
276
277
278
279
280
281
    if [ -n "$VTGPT" ]; then
        vtdebug "format_ventoy_disk_gpt $RESERVE_SIZE_MB $DISK $PARTTOOL ..."
        format_ventoy_disk_gpt $RESERVE_SIZE_MB $DISK $PARTTOOL
    else
        vtdebug "format_ventoy_disk_mbr $RESERVE_SIZE_MB $DISK $PARTTOOL ..."
        format_ventoy_disk_mbr $RESERVE_SIZE_MB $DISK $PARTTOOL
    fi
282
283
284
285
286
287
288
289
290
291
292

    # format part1

    # DiskSize > 32GB  Cluster Size use 128KB
    # DiskSize < 32GB  Cluster Size use 32KB
    if [ $disk_size_gb -gt 32 ]; then
        cluster_sectors=256
    else
        cluster_sectors=64
    fi

293
294
    PART1=$(get_disk_part_name $DISK 1)
    PART2=$(get_disk_part_name $DISK 2)
longpanda's avatar
longpanda committed
295

longpanda's avatar
longpanda committed
296
297
298
299
    #clean part2
    dd status=none conv=fsync if=/dev/zero of=$DISK bs=512 count=32 seek=$part2_start_sector

    #format part1
longpanda's avatar
longpanda committed
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
    wait_and_create_part ${PART1} ${PART2}    
    if [ -b ${PART1} ]; then
        vtinfo "Format partition 1 ${PART1} ..."
        mkexfatfs -n "$VTNEW_LABEL" -s $cluster_sectors ${PART1}
        if [ $? -ne 0 ]; then
            echo "mkexfatfs failed, now retry..."
            mkexfatfs -n "$VTNEW_LABEL" -s $cluster_sectors ${PART1}
            if [ $? -ne 0 ]; then
                echo "######### mkexfatfs failed, exit ########"
                exit 1
            fi
        else
            echo "mkexfatfs success"
        fi        
    else
        vterr "${PART1} NOT exist"
    fi
317

318
    vtinfo "writing data to disk ..."
319
    dd status=none conv=fsync if=./boot/boot.img of=$DISK bs=1 count=446
320

longpanda's avatar
longpanda committed
321
    if [ -n "$VTGPT" ]; then
322
        echo -en '\x22' | dd status=none of=$DISK conv=fsync bs=1 count=1 seek=92
longpanda's avatar
longpanda committed
323
        xzcat ./boot/core.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=2014 seek=34
longpanda's avatar
longpanda committed
324
325
        echo -en '\x23' | dd of=$DISK conv=fsync bs=1 count=1 seek=17908 status=none
    else
longpanda's avatar
longpanda committed
326
        xzcat ./boot/core.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=2047 seek=1
longpanda's avatar
longpanda committed
327
328
    fi
    
longpanda's avatar
longpanda committed
329
330
    # check and umount
    check_umount_disk "$DISK"
331

longpanda's avatar
longpanda committed
332
    xzcat ./ventoy/ventoy.disk.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=$VENTOY_SECTOR_NUM seek=$part2_start_sector
333

longpanda's avatar
longpanda committed
334
335
336
    #test UUID
    testUUIDStr=$(vtoy_gen_uuid | hexdump -C)
    vtdebug "test uuid: $testUUIDStr"
337

338
    #disk uuid
longpanda's avatar
longpanda committed
339
    vtoy_gen_uuid | dd status=none conv=fsync of=${DISK} seek=384 bs=1 count=16
340

341
    #disk signature
longpanda's avatar
longpanda committed
342
    vtoy_gen_uuid | dd status=none conv=fsync of=${DISK} skip=12 seek=440 bs=1 count=4
343
344
345

    vtinfo "sync data ..."
    sync
346

347
    vtinfo "esp partition processing ..."
348

longpanda's avatar
longpanda committed
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
    if [ "$SECUREBOOT" != "YES" ]; then 
        sleep 2
        check_umount_disk "$DISK"  
        vtoycli partresize -s $DISK $part2_start_sector
    fi

    echo ""
    vtinfo "Install Ventoy to $DISK successfully finished."
    echo ""

elif [ "$MODE" = "install" -a -n "$NONDESTRUCTIVE" ]; then
    vtdebug "non-destructive install Ventoy ..."

    version=$(get_disk_ventoy_version $DISK)
    if [ $? -eq 0 ]; then
        if [ -z "$FORCE" ]; then
            vtwarn "$DISK already contains a Ventoy with version $version."
            vtwarn "You can not do and don not need non-destructive installation."
            vtwarn ""
            exit 1
        fi
    fi
371
    
longpanda's avatar
longpanda committed
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
    disk_sector_num=$(cat /sys/block/${DISK#/dev/}/size)
    disk_size_gb=$(expr $disk_sector_num / 2097152)

    if vtoycli partresize -t $DISK; then
        OldStyle="GPT"
    else
        OldStyle="MBR"
    fi

    #Print disk info
    echo "Disk : $DISK"
    parted -s $DISK p 2>&1 | grep Model
    echo "Size : $disk_size_gb GB" 
    echo "Style: $OldStyle"    
    echo ''

    vtwarn "Attention:"
    vtwarn "Ventoy will try non-destructive installation on $DISK if possible."
    echo ""

    read -p 'Continue? (y/n) '  Answer
    if [ "$Answer" != "y" ]; then
        if [ "$Answer" != "Y" ]; then
            exit 0
        fi
    fi

    if [ $disk_sector_num -le $VENTOY_SECTOR_NUM ]; then  
        vterr "No enough space in disk $DISK"
        exit 1
    fi

    PART1=$(get_disk_part_name $DISK 1)  
    PART2=$(get_disk_part_name $DISK 2)  

    #Part1 size in MB aligned with 4KB
    PART1_SECTORS=$(cat /sys/class/block/${PART1#/dev/}/size)
    PART1_4K=$(expr $PART1_SECTORS / 8)
    PART1_MB=$(expr $PART1_4K / 256)
    PART1_NEW_MB=$(expr $PART1_MB - 32)

    echo "$PART1 is ${PART1_MB}MB"

    #check partition layout
    echo "check partition layout ..."
    vtoycli partresize -c $DISK
    vtRet=$?
    if [ $vtRet -eq 0 ]; then
        exit 1
    else
        # check and umount
        check_umount_disk "$DISK"
        sleep 1
        check_umount_disk "$DISK"
426
    
longpanda's avatar
longpanda committed
427
428
429
430
431
        if [ $vtRet -eq 1 ]; then
            echo "Free space enough, start install..."
            part2_start_sector=$(expr $PART1_SECTORS + 2048)
        elif [ $vtRet -eq 2 ]; then
            echo "We need to shrink partition 1 firstly ..."
432
            
longpanda's avatar
longpanda committed
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
            PART1_BLKID=$(blkid $PART1)
            blkid $PART1
            
            if echo $PART1_BLKID | egrep -q -i 'TYPE=ntfs|TYPE=.ntfs'; then
                echo "Partition 1 contains NTFS filesystem"
                
                which ntfsresize
                if [ $? -ne 0 ]; then
                    echo "###[FAIL] ntfsresize not found. Please install ntfs-3g package."
                    exit 1
                fi
                
                echo "ntfsfix -b -d $PART1 ..."
                ntfsfix -b -d $PART1
                
                echo "ntfsresize --size ${PART1_NEW_MB}Mi $PART1 ..."
                ntfsresize -f --size ${PART1_NEW_MB}Mi $PART1
                if [ $? -ne 0 ]; then
                    echo "###[FAIL] ntfsresize failed." 
                    exit 1
                fi
            elif echo $PART1_BLKID | egrep -q -i 'TYPE=ext[2-4]|TYPE=.ext[2-4]'; then
                echo "Partition 1 contains EXT filesystem"
                
                which resize2fs
                if [ $? -ne 0 ]; then
                    echo "###[FAIL] resize2fs not found. Please install e2fsprogs package."
                    exit 1
                fi
                
                echo "e2fsck -f $PART1 ..."
                e2fsck -f $PART1
                
                echo "resize2fs $PART1 ${PART1_NEW_MB}M ..."
                resize2fs $PART1 ${PART1_NEW_MB}M
                if [ $? -ne 0 ]; then
                    echo "###[FAIL] resize2fs failed." 
                    exit 1
                fi
longpanda's avatar
longpanda committed
472
            else
longpanda's avatar
longpanda committed
473
474
                echo "###[FAIL] Unsupported filesystem in partition 1."
                exit 1
longpanda's avatar
longpanda committed
475
            fi
longpanda's avatar
longpanda committed
476
477
478
479
480
            
            sync
            PART1_NEW_END_MB=$(expr $PART1_NEW_MB + 1)
            part2_start_sector=$(expr $PART1_NEW_END_MB \* 2048)
        fi
481
482
    fi

longpanda's avatar
longpanda committed
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
    vtinfo "writing data to disk part2_start=$part2_start_sector ..."
    
    dd status=none conv=fsync if=./boot/boot.img of=$DISK bs=1 count=440
    
    if [ "$OldStyle" = "GPT" ]; then
        echo -en '\x22' | dd status=none of=$DISK conv=fsync bs=1 count=1 seek=92        
        xzcat ./boot/core.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=2014 seek=34
        echo -en '\x23' | dd of=$DISK conv=fsync bs=1 count=1 seek=17908 status=none
    else
        xzcat ./boot/core.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=2047 seek=1
    fi
    
    xzcat ./ventoy/ventoy.disk.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=$VENTOY_SECTOR_NUM seek=$part2_start_sector
    
    #test UUID
    testUUIDStr=$(vtoy_gen_uuid | hexdump -C)
    vtdebug "test uuid: $testUUIDStr"
    
    #disk uuid
    vtoy_gen_uuid | dd status=none conv=fsync of=${DISK} seek=384 bs=1 count=16
    
    vtinfo "sync data ..."
    sync
    
    vtinfo "esp partition processing ..."
    if [ "$SECUREBOOT" != "YES" ]; then
        vtoycli partresize -s $DISK $part2_start_sector
    fi

    vtinfo "update partition table $DISK $part2_start_sector ..."
    vtoycli partresize -p $DISK $part2_start_sector
    if [ $? -eq 0 ]; then
        sync
        echo ""
        vtinfo "Ventoy non-destructive installation on $DISK successfully finished."
        echo ""
    else
        echo ""
        vterr "Ventoy non-destructive installation on $DISK failed."
        echo ""
    fi
524
525
    
else
526
    vtdebug "update Ventoy ..."
527
528
529
    
    oldver=$(get_disk_ventoy_version $DISK)
    if [ $? -ne 0 ]; then
530
531
532
533
534
535
536
537
538
        if is_disk_contains_ventoy $DISK; then
            oldver="Unknown"
        else
            vtwarn "$DISK does not contain Ventoy or data corrupted"
            echo ""
            vtwarn "Please use -i option if you want to install ventoy to $DISK"
            echo ""
            exit 1
        fi
539
540
    fi

longpanda's avatar
longpanda committed
541
542
543
544
545
546
547
548
549
    #reserve secure boot option
    if [ -z "$SECUREBOOT" ]; then
        if check_disk_secure_boot $DISK; then
            SECUREBOOT="YES"
        else
            SECUREBOOT="NO"
        fi
    fi

550
551
552
553
554
    curver=$(cat ./ventoy/version)

    vtinfo "Upgrade operation is safe, all the data in the 1st partition (iso files and other) will be unchanged!"
    echo ""

555
    read -p "Update Ventoy  $oldver ===> $curver   Continue? (y/n) "  Answer
556
557
558
559
560
561
562
    if [ "$Answer" != "y" ]; then
        if [ "$Answer" != "Y" ]; then
            exit 0
        fi
    fi

    PART2=$(get_disk_part_name $DISK 2)
563
564
    SHORT_PART2=${PART2#/dev/}
    part2_start=$(cat /sys/class/block/$SHORT_PART2/start)
565

longpanda's avatar
longpanda committed
566
    PART1_TYPE=$(dd if=$DISK bs=1 count=1 skip=450 status=none | hexdump -n1 -e  '1/1 "%02X"')
567

longpanda's avatar
longpanda committed
568
569
570
    #reserve disk uuid
    rm -f ./diskuuid.bin
    dd status=none conv=fsync if=${DISK} skip=384 bs=1 count=16 of=./diskuuid.bin
571

longpanda's avatar
longpanda committed
572
573
574
575
576
577
578
579
    dd status=none conv=fsync if=./boot/boot.img of=$DISK bs=1 count=440
    dd status=none conv=fsync if=./diskuuid.bin of=$DISK bs=1 count=16 seek=384
    rm -f ./diskuuid.bin

    #reserve data
    rm -f ./rsvdata.bin
    dd status=none conv=fsync if=${DISK} skip=2040 bs=512 count=8 of=./rsvdata.bin

longpanda's avatar
update  
longpanda committed
580
    if [ "$PART1_TYPE" = "EE" ]; then
581
        vtdebug "This is GPT partition style ..."
longpanda's avatar
longpanda committed
582
        echo -en '\x22' | dd status=none of=$DISK conv=fsync bs=1 count=1 seek=92
longpanda's avatar
longpanda committed
583
        xzcat ./boot/core.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=2014 seek=34
longpanda's avatar
update  
longpanda committed
584
585
586
        echo -en '\x23' | dd of=$DISK conv=fsync bs=1 count=1 seek=17908 status=none
    else
        vtdebug "This is MBR partition style ..."
587

longpanda's avatar
longpanda committed
588
589
        PART1_ACTIVE=$(dd if=$DISK bs=1 count=1 skip=446 status=none | hexdump -n1 -e  '1/1 "%02X"')
        PART2_ACTIVE=$(dd if=$DISK bs=1 count=1 skip=462 status=none | hexdump -n1 -e  '1/1 "%02X"')
590

longpanda's avatar
update  
longpanda committed
591
592
593
594
595
596
        vtdebug "PART1_ACTIVE=$PART1_ACTIVE  PART2_ACTIVE=$PART2_ACTIVE"
        if [ "$PART1_ACTIVE" = "00" ] && [ "$PART2_ACTIVE" = "80" ]; then
            vtdebug "change 1st partition active, 2nd partition inactive ..."
            echo -en '\x80' | dd of=$DISK conv=fsync bs=1 count=1 seek=446 status=none
            echo -en '\x00' | dd of=$DISK conv=fsync bs=1 count=1 seek=462 status=none
        fi
longpanda's avatar
longpanda committed
597
        xzcat ./boot/core.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=2047 seek=1
longpanda's avatar
longpanda committed
598
    fi
longpanda's avatar
longpanda committed
599

longpanda's avatar
longpanda committed
600
601
602
    dd status=none conv=fsync if=./rsvdata.bin seek=2040 bs=512 count=8 of=${DISK}
    rm -f ./rsvdata.bin

longpanda's avatar
longpanda committed
603
604
    check_umount_disk "$DISK"
    
longpanda's avatar
longpanda committed
605
    xzcat ./ventoy/ventoy.disk.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=$VENTOY_SECTOR_NUM seek=$part2_start
606
    sync
longpanda's avatar
longpanda committed
607

longpanda's avatar
longpanda committed
608
    vtinfo "esp partition processing ..."
609
    if [ "$SECUREBOOT" != "YES" ]; then
longpanda's avatar
longpanda committed
610
611
612
        sleep 2
        check_umount_disk "$DISK"
        vtoycli partresize -s $DISK $part2_start
613
614
615
    fi

    echo ""
616
    vtinfo "Update Ventoy on $DISK successfully finished."
617
    echo ""
618

619
620
621
fi