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

. ./tool/ventoy_lib.sh

print_usage() {
longpanda's avatar
longpanda committed
6
    
7
8
9
10
    echo 'Usage:  Ventoy2Disk.sh CMD [ OPTION ] /dev/sdX'
    echo '  CMD:'
    echo '   -i  install ventoy to sdX (fail if disk already installed with ventoy)'
    echo '   -I  force install ventoy to sdX (no matter installed or not)'
longpanda's avatar
longpanda committed
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)'
longpanda's avatar
longpanda committed
18
    echo '   -L          Label of the 1st exfat partition (default is ventoy)'
19
20
21
    echo ''
}

longpanda's avatar
longpanda committed
22

longpanda's avatar
longpanda committed
23
VTNEW_LABEL='ventoy'
longpanda's avatar
longpanda committed
24
RESERVE_SIZE_MB=0
25
26
27
28
29
30
31
32
while [ -n "$1" ]; do
    if [ "$1" = "-i" ]; then
        MODE="install"
    elif [ "$1" = "-I" ]; then
        MODE="install"
        FORCE="Y"
    elif [ "$1" = "-u" ]; then
        MODE="update"
longpanda's avatar
longpanda committed
33
34
    elif [ "$1" = "-l" ]; then
        MODE="list"
35
36
    elif [ "$1" = "-s" ]; then
        SECUREBOOT="YES"
longpanda's avatar
longpanda committed
37
38
    elif [ "$1" = "-S" ]; then
        SECUREBOOT="NO"
longpanda's avatar
longpanda committed
39
40
    elif [ "$1" = "-g" ]; then
        VTGPT="YES"
longpanda's avatar
longpanda committed
41
42
43
    elif [ "$1" = "-L" ]; then
        shift
        VTNEW_LABEL=$1
longpanda's avatar
longpanda committed
44
45
46
47
    elif [ "$1" = "-r" ]; then
        RESERVE_SPACE="YES"
        shift
        RESERVE_SIZE_MB=$1
longpanda's avatar
longpanda committed
48
49
    elif [ "$1" = "-V" ] || [ "$1" = "--version" ]; then
        exit 0
longpanda's avatar
longpanda committed
50
    elif [ "$1" = "-h" ] || [ "$1" = "--help" ]; then
longpanda's avatar
longpanda committed
51
52
        print_usage
        exit 0
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
    else
        if ! [ -b "$1" ]; then
            vterr "$1 is NOT a valid device"
            print_usage
            exit 1
        fi
        DISK=$1
    fi
    
    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
76
77
78
79
80
    vterr  "$DISK is a partition, please use the whole disk."
    echo   "For example:"
    vterr  "    sudo sh Ventoy2Disk.sh -i /dev/sdX1 <=== This is wrong"
    vtinfo "    sudo sh Ventoy2Disk.sh -i /dev/sdX  <=== This is right"
    echo ""
81
82
83
    exit 1
fi

longpanda's avatar
longpanda committed
84
if [ -n "$RESERVE_SPACE" -a "$MODE" = "install" ]; then
longpanda's avatar
longpanda committed
85
86
87
88
89
90
91
92
    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

longpanda's avatar
longpanda committed
93
#check access 
94
95
96
97
98
99
100
101
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
102
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
109
110
111
    vterr "Some tools can not run in current system. Please check log.txt for detail."
    exit 1
fi

longpanda's avatar
longpanda committed
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
if [ "$MODE" = "list" ]; then
    version=$(get_disk_ventoy_version $DISK)
    if [ $? -eq 0 ]; then
        echo "Ventoy Version in Disk: $version"
        
        vtPart1Type=$(dd if=$DISK bs=1 count=1 skip=450 status=none | hexdump -n1 -e  '1/1 "%02X"')
        if [ "$vtPart1Type" = "EE" ]; then            
            echo "Disk Partition Style  : GPT"
        else
            echo "Disk Partition Style  : MBR"
        fi
        
        vtPART2=$(get_disk_part_name $DISK 2)        
        rm -rf ./tmpmntp2 && mkdir ./tmpmntp2
        mount $vtPART2 ./tmpmntp2 > /dev/null 2>&1

        if [ -e ./tmpmntp2/EFI/BOOT/MokManager.efi ]; then
            echo "Secure Boot Support   : YES"
        else
            echo "Secure Boot Support   : NO"
        fi        
        umount ./tmpmntp2 > /dev/null 2>&1
        rm -rf ./tmpmntp2
    else
        echo "Ventoy Version: NA"
    fi
    echo ""
    exit 0
fi

longpanda's avatar
longpanda committed
142
#check mountpoint
143
144
145
146
147
148
149
150
151
152
153
grep "^$DISK" /proc/mounts | while read mtline; do
    mtpnt=$(echo $mtline | awk '{print $2}')
    vtdebug "Trying to umount $mtpnt ..."
    umount $mtpnt >/dev/null 2>&1
done

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

longpanda's avatar
longpanda committed
154
155
156
157
158
159
#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
160
161
fi

longpanda's avatar
longpanda committed
162
163
164
165
166
167
168
169
170
171
172
#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
        vterr "tmp_mnt directory exit, please delete it first."
        exit 1
    fi
fi

173
174
175
176

if [ "$MODE" = "install" ]; then
    vtdebug "install ventoy ..."

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
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
    fi
    
    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."
            vtwarn "OR if you really want to reinstall ventoy to $DISK, please use -I option."
            vtwarn ""
            exit 1
        fi
    fi
    
    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
218
219
220
221
222
223
224
    if [ -n "$RESERVE_SPACE" ]; then
        sum_size_mb=$(expr $RESERVE_SIZE_MB + $VENTOY_PART_SIZE_MB)
        reserve_sector_num=$(expr $sum_size_mb \* 2048)
        
        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
longpanda's avatar
longpanda committed
228
229
230
231
232
233
    echo "Size : $disk_size_gb GB"    
    if [ -n "$VTGPT" ]; then
        echo "Style: GPT"
    else
        echo "Style: MBR"
    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
262
263
264
265
266
267
268
269
270
271
    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

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

longpanda's avatar
longpanda committed
272
273
274
275
276
277
278
    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
279
280
281
282
283
284
285
286
287
288
289

    # 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

longpanda's avatar
longpanda committed
290
291
292
    PART1=$(get_disk_part_name $DISK 1)  
    PART2=$(get_disk_part_name $DISK 2)  

longpanda's avatar
longpanda committed
293
    mkexfatfs -n "$VTNEW_LABEL" -s $cluster_sectors ${PART1}
294
295

    vtinfo "writing data to disk ..."
longpanda's avatar
longpanda committed
296
    
297
    dd status=none conv=fsync if=./boot/boot.img of=$DISK bs=1 count=446
longpanda's avatar
longpanda committed
298
299
300
    
    if [ -n "$VTGPT" ]; then
        echo -en '\x22' | dd status=none of=$DISK conv=fsync bs=1 count=1 seek=92        
longpanda's avatar
longpanda committed
301
        xzcat ./boot/core.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=2014 seek=34
longpanda's avatar
longpanda committed
302
303
        echo -en '\x23' | dd of=$DISK conv=fsync bs=1 count=1 seek=17908 status=none
    else
longpanda's avatar
longpanda committed
304
        xzcat ./boot/core.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=2047 seek=1
longpanda's avatar
longpanda committed
305
306
    fi
    
longpanda's avatar
longpanda committed
307
    xzcat ./ventoy/ventoy.disk.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=$VENTOY_SECTOR_NUM seek=$part2_start_sector
308
    
longpanda's avatar
longpanda committed
309
310
311
312
    #test UUID
    testUUIDStr=$(vtoy_gen_uuid | hexdump -C)
    vtdebug "test uuid: $testUUIDStr"
    
313
    #disk uuid
longpanda's avatar
longpanda committed
314
    vtoy_gen_uuid | dd status=none conv=fsync of=${DISK} seek=384 bs=1 count=16
315
316
    
    #disk signature
longpanda's avatar
longpanda committed
317
    vtoy_gen_uuid | dd status=none conv=fsync of=${DISK} skip=12 seek=440 bs=1 count=4
318
319
320
321
322
323
324

    vtinfo "sync data ..."
    sync
    
    vtinfo "esp partition processing ..."
    
    sleep 1
longpanda's avatar
longpanda committed
325
    mtpnt=$(grep "^${PART2}" /proc/mounts | awk '{print $2}')
326
327
328
329
    if [ -n "$mtpnt" ]; then
        umount $mtpnt >/dev/null 2>&1
    fi
    
longpanda's avatar
longpanda committed
330
    if [ "$SECUREBOOT" != "YES" ]; then        
331
332
333
334
        mkdir ./tmp_mnt
        
        vtdebug "mounting part2 ...."
        for tt in 1 2 3; do
longpanda's avatar
longpanda committed
335
            if mount ${PART2} ./tmp_mnt; then
336
337
338
339
                vtdebug "mounting part2 success"
                break
            fi
            
longpanda's avatar
longpanda committed
340
            mtpnt=$(grep "^${PART2}" /proc/mounts | awk '{print $2}')
341
342
343
344
345
            if [ -n "$mtpnt" ]; then
                umount $mtpnt >/dev/null 2>&1
            fi
            sleep 2
        done
longpanda's avatar
longpanda committed
346
        
347
348
        rm -f ./tmp_mnt/EFI/BOOT/BOOTX64.EFI
        rm -f ./tmp_mnt/EFI/BOOT/grubx64.efi
longpanda's avatar
longpanda committed
349
350
        rm -f ./tmp_mnt/EFI/BOOT/BOOTIA32.EFI
        rm -f ./tmp_mnt/EFI/BOOT/grubia32.efi
351
        rm -f ./tmp_mnt/EFI/BOOT/MokManager.efi
longpanda's avatar
longpanda committed
352
        rm -f ./tmp_mnt/EFI/BOOT/mmia32.efi
353
354
        rm -f ./tmp_mnt/ENROLL_THIS_KEY_IN_MOKMANAGER.cer
        mv ./tmp_mnt/EFI/BOOT/grubx64_real.efi  ./tmp_mnt/EFI/BOOT/BOOTX64.EFI
longpanda's avatar
longpanda committed
355
        mv ./tmp_mnt/EFI/BOOT/grubia32_real.efi  ./tmp_mnt/EFI/BOOT/BOOTIA32.EFI
356
        
longpanda's avatar
longpanda committed
357
358
359
360
361
362
363
364
365
366
        for tt in 1 2 3; do
            if umount ./tmp_mnt; then
                vtdebug "umount part2 success"
                rm -rf ./tmp_mnt
                break
            else
                vtdebug "umount part2 failed, now retry..."
                sleep 1
            fi
        done
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
    fi

    echo ""
    vtinfo "Install Ventoy to $DISK successfully finished."
    echo ""
    
else
    vtdebug "update ventoy ..."
    
    oldver=$(get_disk_ventoy_version $DISK)
    if [ $? -ne 0 ]; then
        vtwarn "$DISK does not contain ventoy or data corupted"
        echo ""
        vtwarn "Please use -i option if you want to install ventoy to $DISK"
        echo ""
        exit 1
    fi

longpanda's avatar
longpanda committed
385
386
387
388
389
390
391
392
393
    #reserve secure boot option
    if [ -z "$SECUREBOOT" ]; then
        if check_disk_secure_boot $DISK; then
            SECUREBOOT="YES"
        else
            SECUREBOOT="NO"
        fi
    fi

394
395
396
397
398
399
400
401
402
403
404
405
406
    curver=$(cat ./ventoy/version)

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

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

    PART2=$(get_disk_part_name $DISK 2)
407
408
    SHORT_PART2=${PART2#/dev/}
    part2_start=$(cat /sys/class/block/$SHORT_PART2/start)
409
    
longpanda's avatar
longpanda committed
410
    PART1_TYPE=$(dd if=$DISK bs=1 count=1 skip=450 status=none | hexdump -n1 -e  '1/1 "%02X"')
411
    
longpanda's avatar
longpanda committed
412
413
414
415
416
417
418
419
420
421
422
423
    #reserve disk uuid
    rm -f ./diskuuid.bin
    dd status=none conv=fsync if=${DISK} skip=384 bs=1 count=16 of=./diskuuid.bin
    
    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
424
    if [ "$PART1_TYPE" = "EE" ]; then
longpanda's avatar
longpanda committed
425
426
        vtdebug "This is GPT partition style ..."    
        echo -en '\x22' | dd status=none of=$DISK conv=fsync bs=1 count=1 seek=92
longpanda's avatar
longpanda committed
427
        xzcat ./boot/core.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=2014 seek=34
longpanda's avatar
update  
longpanda committed
428
429
430
        echo -en '\x23' | dd of=$DISK conv=fsync bs=1 count=1 seek=17908 status=none
    else
        vtdebug "This is MBR partition style ..."
longpanda's avatar
longpanda committed
431
    
longpanda's avatar
longpanda committed
432
433
        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"')
longpanda's avatar
update  
longpanda committed
434
435
436
437
438
439
440
        
        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
441
        xzcat ./boot/core.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=2047 seek=1
longpanda's avatar
longpanda committed
442
    fi
longpanda's avatar
longpanda committed
443

longpanda's avatar
longpanda committed
444
445
446
    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
447
    xzcat ./ventoy/ventoy.disk.img.xz | dd status=none conv=fsync of=$DISK bs=512 count=$VENTOY_SECTOR_NUM seek=$part2_start
448
449
450
451
452
453
454
455

    sync
    
    if [ "$SECUREBOOT" != "YES" ]; then
        mkdir ./tmp_mnt
        
        vtdebug "mounting part2 ...."
        for tt in 1 2 3; do
longpanda's avatar
longpanda committed
456
            if mount ${PART2} ./tmp_mnt; then
457
458
459
460
461
                vtdebug "mounting part2 success"
                break
            fi
            sleep 2
        done
longpanda's avatar
longpanda committed
462
        
463
464
        rm -f ./tmp_mnt/EFI/BOOT/BOOTX64.EFI
        rm -f ./tmp_mnt/EFI/BOOT/grubx64.efi
longpanda's avatar
longpanda committed
465
466
        rm -f ./tmp_mnt/EFI/BOOT/BOOTIA32.EFI
        rm -f ./tmp_mnt/EFI/BOOT/grubia32.efi
467
        rm -f ./tmp_mnt/EFI/BOOT/MokManager.efi
longpanda's avatar
longpanda committed
468
        rm -f ./tmp_mnt/EFI/BOOT/mmia32.efi
469
470
        rm -f ./tmp_mnt/ENROLL_THIS_KEY_IN_MOKMANAGER.cer
        mv ./tmp_mnt/EFI/BOOT/grubx64_real.efi  ./tmp_mnt/EFI/BOOT/BOOTX64.EFI
longpanda's avatar
longpanda committed
471
472
        mv ./tmp_mnt/EFI/BOOT/grubia32_real.efi  ./tmp_mnt/EFI/BOOT/BOOTIA32.EFI
        
473
        
longpanda's avatar
longpanda committed
474
475
476
477
478
479
480
481
482
483
        for tt in 1 2 3; do
            if umount ./tmp_mnt; then
                vtdebug "umount part2 success"
                rm -rf ./tmp_mnt
                break
            else
                vtdebug "umount part2 failed, now retry..."
                sleep 1
            fi
        done        
484
485
486
487
488
489
490
491
492
    fi

    echo ""
    vtinfo "Update Ventoy to $DISK successfully finished."
    echo ""
    
fi