ventoy_lib.sh 12.5 KB
Newer Older
longpanda's avatar
longpanda committed
1
2
3
4
#!/bin/sh

#Ventoy partition 32MB
VENTOY_PART_SIZE=33554432
longpanda's avatar
longpanda committed
5
VENTOY_PART_SIZE_MB=32
longpanda's avatar
longpanda committed
6
7
8
9
10
11
12
13
14
15
16
17
18
VENTOY_SECTOR_SIZE=512
VENTOY_SECTOR_NUM=65536

ventoy_false() {
    [ "1" = "2" ]
}

ventoy_true() {
    [ "1" = "1" ]
}


vtinfo() {
longpanda's avatar
longpanda committed
19
    echo -e "\033[32m$*\033[0m"
longpanda's avatar
longpanda committed
20
21
22
}

vtwarn() {
longpanda's avatar
longpanda committed
23
    echo -e "\033[33m$*\033[0m"
longpanda's avatar
longpanda committed
24
25
26
27
}


vterr() {
longpanda's avatar
longpanda committed
28
    echo -e "\033[31m$*\033[0m"
longpanda's avatar
longpanda committed
29
30
31
32
33
34
}

vtdebug() {
    echo "$*" >> ./log.txt
}

longpanda's avatar
longpanda committed
35
36
37
38
39
40
41
42
43
44
45
vtoy_gen_uuid() {
    if  uuid -F BIN > /dev/null 2>&1; then
        uuid -F BIN
    elif uuidgen -V > /dev/null 2>&1; then
        a=$(uuidgen | sed 's/-//g')
        echo -en "\x${a:0:2}\x${a:2:2}\x${a:4:2}\x${a:6:2}\x${a:8:2}\x${a:10:2}\x${a:12:2}\x${a:14:2}\x${a:16:2}\x${a:18:2}\x${a:20:2}\x${a:22:2}\x${a:24:2}\x${a:26:2}\x${a:28:2}\x${a:30:2}"        
    elif python -V > /dev/null 2>&1; then
        a=$(python -c 'import sys,uuid; sys.stdout.write(uuid.uuid4().hex)')
        echo -en "\x${a:0:2}\x${a:2:2}\x${a:4:2}\x${a:6:2}\x${a:8:2}\x${a:10:2}\x${a:12:2}\x${a:14:2}\x${a:16:2}\x${a:18:2}\x${a:20:2}\x${a:22:2}\x${a:24:2}\x${a:26:2}\x${a:28:2}\x${a:30:2}"
    elif [ -e /dev/urandom ]; then
        dd if=/dev/urandom bs=1 count=16 status=none
longpanda's avatar
longpanda committed
46
    else
longpanda's avatar
longpanda committed
47
48
49
        datestr=$(date +%N%N%N%N%N)
        a=${datestr:0:32}
        echo -en "\x${a:0:2}\x${a:2:2}\x${a:4:2}\x${a:6:2}\x${a:8:2}\x${a:10:2}\x${a:12:2}\x${a:14:2}\x${a:16:2}\x${a:18:2}\x${a:20:2}\x${a:22:2}\x${a:24:2}\x${a:26:2}\x${a:28:2}\x${a:30:2}"
longpanda's avatar
longpanda committed
50
    fi
longpanda's avatar
longpanda committed
51
52
53
}

check_tool_work_ok() {
longpanda's avatar
longpanda committed
54
    
longpanda's avatar
longpanda committed
55
    if echo 1 | hexdump > /dev/null; then
longpanda's avatar
longpanda committed
56
57
58
59
60
61
62
        vtdebug "hexdump test ok ..."
    else
        vtdebug "hexdump test fail ..."
        ventoy_false
        return
    fi
   
longpanda's avatar
longpanda committed
63
64
    if mkexfatfs -V > /dev/null; then
        vtdebug "mkexfatfs test ok ..."
longpanda's avatar
longpanda committed
65
    else
longpanda's avatar
longpanda committed
66
        vtdebug "mkexfatfs test fail ..."
longpanda's avatar
longpanda committed
67
68
69
70
        ventoy_false
        return
    fi
    
longpanda's avatar
longpanda committed
71
72
    if vtoycli fat -T; then
        vtdebug "vtoycli fat test ok ..."
longpanda's avatar
longpanda committed
73
    else
longpanda's avatar
longpanda committed
74
        vtdebug "vtoycli fat test fail ..."
longpanda's avatar
longpanda committed
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
        ventoy_false
        return
    fi
    
    vtdebug "tool check success ..."
    ventoy_true
}


get_disk_part_name() {
    DISK=$1
    
    if echo $DISK | grep -q "/dev/loop"; then
        echo ${DISK}p${2}
    elif echo $DISK | grep -q "/dev/nvme[0-9][0-9]*n[0-9]"; then
        echo ${DISK}p${2}
longpanda's avatar
longpanda committed
91
92
    elif echo $DISK | grep -q "/dev/mmcblk[0-9]"; then
        echo ${DISK}p${2}
93
94
    elif echo $DISK | grep -q "/dev/nbd[0-9]"; then
        echo ${DISK}p${2}
95
96
    elif echo $DISK | grep -q "/dev/zd[0-9]"; then
        echo ${DISK}p${2}
longpanda's avatar
longpanda committed
97
98
99
100
101
    else
        echo ${DISK}${2}
    fi
}

longpanda's avatar
longpanda committed
102
103
104
105
106
107
108
109
check_umount_disk() {
    DiskOrPart="$1"
    grep "^${DiskOrPart}" /proc/mounts | while read mtline; do
        mtpnt=$(echo $mtline | awk '{print $2}')
        vtdebug "Trying to umount $mtpnt ..."
        umount $mtpnt >/dev/null 2>&1
    done
}
longpanda's avatar
longpanda committed
110
111
112
113
114
115
116
117
118
119

get_ventoy_version_from_cfg() {
    if grep -q 'set.*VENTOY_VERSION=' $1; then
        grep 'set.*VENTOY_VERSION=' $1 | awk -F'"' '{print $2}'
    else
        echo 'none'
    fi
}

is_disk_contains_ventoy() {
longpanda's avatar
longpanda committed
120
    DISK=$1    
longpanda's avatar
longpanda committed
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
    
    PART1=$(get_disk_part_name $1 1)  
    PART2=$(get_disk_part_name $1 2)  
    
    if [ -e /sys/class/block/${PART2#/dev/}/size ]; then    
        SIZE=$(cat /sys/class/block/${PART2#/dev/}/size)
    else
        SIZE=0
    fi

    if ! [ -b $PART1 ]; then
        vtdebug "$PART1 not exist"
        ventoy_false
        return
    fi
    
    if ! [ -b $PART2 ]; then
        vtdebug "$PART2 not exist"
        ventoy_false
        return
    fi
    
longpanda's avatar
longpanda committed
143
144
    PART1_TYPE=$(dd if=$DISK bs=1 count=1 skip=450 status=none | hexdump -n1 -e  '1/1 "%02X"')
    PART2_TYPE=$(dd if=$DISK bs=1 count=1 skip=466 status=none | hexdump -n1 -e  '1/1 "%02X"')
longpanda's avatar
longpanda committed
145
    
longpanda's avatar
longpanda committed
146
147
148
149
150
151
152
    # if [ "$PART1_TYPE" != "EE" ]; then
        # if [ "$PART2_TYPE" != "EF" ]; then
            # vtdebug "part2 type is $PART2_TYPE not EF"
            # ventoy_false
            # return
        # fi
    # fi
longpanda's avatar
longpanda committed
153
    
longpanda's avatar
longpanda committed
154
    # PART1_TYPE=$(dd if=$DISK bs=1 count=1 skip=450 status=none | hexdump -n1 -e  '1/1 "%02X"')
longpanda's avatar
longpanda committed
155
156
157
158
159
    # if [ "$PART1_TYPE" != "07" ]; then
        # vtdebug "part1 type is $PART2_TYPE not 07"
        # ventoy_false
        # return
    # fi
longpanda's avatar
longpanda committed
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
    
    if [ -e /sys/class/block/${PART1#/dev/}/start ]; then
        PART1_START=$(cat /sys/class/block/${PART1#/dev/}/start)
    fi
    
    if [ "$PART1_START" != "2048" ]; then
        vtdebug "part1 start is $PART1_START not 2048"
        ventoy_false
        return
    fi 
    
    if [ "$VENTOY_SECTOR_NUM" != "$SIZE" ]; then
        vtdebug "part2 size is $SIZE not $VENTOY_SECTOR_NUM"
        ventoy_false
        return
    fi
    
    ventoy_true
}

longpanda's avatar
longpanda committed
180
check_disk_secure_boot() {
longpanda's avatar
longpanda committed
181
182
183
184
185
186
187
    if ! is_disk_contains_ventoy $1; then
        ventoy_false
        return
    fi
    
    PART2=$(get_disk_part_name $1 2)    
    
longpanda's avatar
longpanda committed
188
    vtoycli fat -s $PART2
longpanda's avatar
longpanda committed
189
190
191
192
193
194
195
}

get_disk_ventoy_version() {

    if ! is_disk_contains_ventoy $1; then
        ventoy_false
        return
longpanda's avatar
longpanda committed
196
197
    fi
    
longpanda's avatar
longpanda committed
198
199
    PART2=$(get_disk_part_name $1 2)    
    
longpanda's avatar
longpanda committed
200
    ParseVer=$(vtoycli fat $PART2)
longpanda's avatar
longpanda committed
201
202
203
204
205
206
207
208
209
210
    if [ $? -eq 0 ]; then
        vtdebug "Ventoy version in $PART2 is $ParseVer"
        echo $ParseVer
        ventoy_true
        return
    fi
    
    ventoy_false
}

longpanda's avatar
longpanda committed
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
wait_and_create_part() {
    vPART1=$1
    vPART2=$2
    echo 'Wait for partitions ...'
    for i in 0 1 2 3 4 5 6 7 8 9; do
        if ls -l $vPART1 2>/dev/null | grep -q '^b'; then
            if ls -l $vPART2 2>/dev/null | grep -q '^b'; then
                break
            fi
        else
            echo "Wait for $vPART1/$vPART2 ..."
            sleep 1
        fi
    done

    if ls -l $vPART1 2>/dev/null | grep -q '^b'; then
        echo "$vPART1 exist OK"
    else
        MajorMinor=$(sed "s/:/ /" /sys/class/block/${vPART1#/dev/}/dev)        
        echo "mknod -m 0660 $vPART1 b $MajorMinor ..."
        mknod -m 0660 $vPART1 b $MajorMinor
    fi
    
    if ls -l $vPART2 2>/dev/null | grep -q '^b'; then
        echo "$vPART2 exist OK"
    else
        MajorMinor=$(sed "s/:/ /" /sys/class/block/${vPART2#/dev/}/dev)        
        echo "mknod -m 0660 $vPART2 b $MajorMinor ..."
        mknod -m 0660 $vPART2 b $MajorMinor        
    fi

    if ls -l $vPART1 2>/dev/null | grep -q '^b'; then
        if ls -l $vPART2 2>/dev/null | grep -q '^b'; then
            echo "partition exist OK"
        fi
    else
        echo "[FAIL] $vPART1/$vPART2 does not exist"
        exit 1
    fi
}


longpanda's avatar
longpanda committed
253
format_ventoy_disk_mbr() {
longpanda's avatar
longpanda committed
254
255
256
    reserve_mb=$1
    DISK=$2
    PARTTOOL=$3
257
    
longpanda's avatar
longpanda committed
258
    PART1=$(get_disk_part_name $DISK 1)
longpanda's avatar
longpanda committed
259
260
261
262
    PART2=$(get_disk_part_name $DISK 2)
    
    sector_num=$(cat /sys/block/${DISK#/dev/}/size)
    
longpanda's avatar
longpanda committed
263
264
265
266
267
268
269
270
271
272
    part1_start_sector=2048 
    
    if [ $reserve_mb -gt 0 ]; then
        reserve_sector_num=$(expr $reserve_mb \* 2048)
        part1_end_sector=$(expr $sector_num - $reserve_sector_num - $VENTOY_SECTOR_NUM - 1)
    else
        part1_end_sector=$(expr $sector_num - $VENTOY_SECTOR_NUM - 1)
    fi
    
    part2_start_sector=$(expr $part1_end_sector + 1)
longpanda's avatar
longpanda committed
273
274
275
276
277
278
279
280
    
    modsector=$(expr $part2_start_sector % 8)
    if [ $modsector -gt 0 ]; then
        vtdebug "modsector:$modsector need to be aligned with 4KB"
        part1_end_sector=$(expr $part1_end_sector - $modsector)
        part2_start_sector=$(expr $part1_end_sector + 1)
    fi
    
longpanda's avatar
longpanda committed
281
282
283
    part2_end_sector=$(expr $part2_start_sector + $VENTOY_SECTOR_NUM - 1)

    export part2_start_sector
longpanda's avatar
longpanda committed
284

longpanda's avatar
longpanda committed
285
286
287
    vtdebug "part1_start_sector=$part1_start_sector  part1_end_sector=$part1_end_sector"
    vtdebug "part2_start_sector=$part2_start_sector  part2_end_sector=$part2_end_sector"

288
289
290
291
292
    if [ -e $PART1 ]; then
        echo "delete $PART1"
        rm -f $PART1
    fi

longpanda's avatar
longpanda committed
293
294
295
296
297
298
    if [ -e $PART2 ]; then
        echo "delete $PART2"
        rm -f $PART2
    fi

    echo ""
longpanda's avatar
longpanda committed
299
    echo "Create partitions on $DISK by $PARTTOOL in MBR style ..."
longpanda's avatar
longpanda committed
300
    
301
    if [ "$PARTTOOL" = "parted" ]; then
longpanda's avatar
longpanda committed
302
303
304
305
306
307
        vtdebug "format disk by parted ..."
        parted -a none --script $DISK \
            mklabel msdos \
            unit s \
            mkpart primary ntfs $part1_start_sector $part1_end_sector \
            mkpart primary fat16 $part2_start_sector $part2_end_sector \
longpanda's avatar
longpanda committed
308
            set 1 boot on \
longpanda's avatar
longpanda committed
309
            quit
310
311
312

        sync
        echo -en '\xEF' | dd of=$DISK conv=fsync bs=1 count=1 seek=466 > /dev/null 2>&1
longpanda's avatar
longpanda committed
313
314
    else
    vtdebug "format disk by fdisk ..."
longpanda's avatar
longpanda committed
315
    
longpanda's avatar
longpanda committed
316
fdisk $DISK >>./log.txt 2>&1 <<EOF
longpanda's avatar
longpanda committed
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
o
n
p
1
$part1_start_sector
$part1_end_sector
n
p
2
$part2_start_sector
$part2_end_sector
t
1
7
t
2
ef
a
longpanda's avatar
longpanda committed
335
1
longpanda's avatar
longpanda committed
336
337
w
EOF
longpanda's avatar
longpanda committed
338
339
    fi
   
longpanda's avatar
longpanda committed
340
    udevadm trigger --name-match=$DISK >/dev/null 2>&1
longpanda's avatar
longpanda committed
341
342
    partprobe >/dev/null 2>&1
    sleep 3
343
    echo "Done"
longpanda's avatar
longpanda committed
344

345
346
347
348

    echo 'Wait for partitions ...'
    for i in 0 1 2 3 4 5 6 7 8 9; do
        if [ -b $PART1 -a -b $PART2 ]; then
longpanda's avatar
longpanda committed
349
350
            break
        else
351
            echo "Wait for $PART1/$PART2 ..."
longpanda's avatar
longpanda committed
352
353
            sleep 1
        fi
longpanda's avatar
longpanda committed
354
    done
longpanda's avatar
longpanda committed
355

356
357
358
359
360
361
    if ! [ -b $PART1 ]; then
        MajorMinor=$(sed "s/:/ /" /sys/class/block/${PART1#/dev/}/dev)        
        echo "mknod -m 0660 $PART1 b $MajorMinor ..."
        mknod -m 0660 $PART1 b $MajorMinor
    fi
    
longpanda's avatar
longpanda committed
362
363
    if ! [ -b $PART2 ]; then
        MajorMinor=$(sed "s/:/ /" /sys/class/block/${PART2#/dev/}/dev)        
longpanda's avatar
longpanda committed
364
        echo "mknod -m 0660 $PART2 b $MajorMinor ..."
365
366
367
368
369
370
371
372
        mknod -m 0660 $PART2 b $MajorMinor        
    fi

    if [ -b $PART1 -a -b $PART2 ]; then
        echo "partition exist OK"
    else
        echo "[FAIL] $PART1/$PART2 does not exist"
        exit 1
longpanda's avatar
longpanda committed
373
    fi
longpanda's avatar
longpanda committed
374

375
    echo "create efi fat fs $PART2 ..."
longpanda's avatar
longpanda committed
376
    for i in 0 1 2 3 4 5 6 7 8 9; do
longpanda's avatar
longpanda committed
377
378
        check_umount_disk "$PART2"

longpanda's avatar
longpanda committed
379
        if mkfs.vfat -F 16 -n VTOYEFI -s 1 $PART2; then
longpanda's avatar
longpanda committed
380
381
382
383
384
385
386
387
            echo 'success'
            break
        else
            echo "$? retry ..."
            sleep 2
        fi
    done
}
388
389


longpanda's avatar
longpanda committed
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
format_ventoy_disk_gpt() {
    reserve_mb=$1
    DISK=$2
    PARTTOOL=$3
    
    PART1=$(get_disk_part_name $DISK 1)
    PART2=$(get_disk_part_name $DISK 2)
    
    sector_num=$(cat /sys/block/${DISK#/dev/}/size)
    
    part1_start_sector=2048 
    
    if [ $reserve_mb -gt 0 ]; then
        reserve_sector_num=$(expr $reserve_mb \* 2048 + 33)
        part1_end_sector=$(expr $sector_num - $reserve_sector_num - $VENTOY_SECTOR_NUM - 1)
    else
        part1_end_sector=$(expr $sector_num - $VENTOY_SECTOR_NUM - 34)
    fi
    
    part2_start_sector=$(expr $part1_end_sector + 1)
longpanda's avatar
longpanda committed
410
411
412
413
414
415
416
417
    
    modsector=$(expr $part2_start_sector % 8)
    if [ $modsector -gt 0 ]; then
        vtdebug "modsector:$modsector need to be aligned with 4KB"
        part1_end_sector=$(expr $part1_end_sector - $modsector)
        part2_start_sector=$(expr $part1_end_sector + 1)
    fi
    
longpanda's avatar
longpanda committed
418
419
420
421
422
423
424
    part2_end_sector=$(expr $part2_start_sector + $VENTOY_SECTOR_NUM - 1)

    export part2_start_sector

    vtdebug "part1_start_sector=$part1_start_sector  part1_end_sector=$part1_end_sector"
    vtdebug "part2_start_sector=$part2_start_sector  part2_end_sector=$part2_end_sector"

425
426
427
428
429
    if [ -e $PART1 ]; then
        echo "delete $PART1"
        rm -f $PART1
    fi

longpanda's avatar
longpanda committed
430
431
432
433
434
435
436
437
438
    if [ -e $PART2 ]; then
        echo "delete $PART2"
        rm -f $PART2
    fi

    echo ""
    echo "Create partitions on $DISK by $PARTTOOL in GPT style ..."
    
    vtdebug "format disk by parted ..."
longpanda's avatar
longpanda committed
439
440
441
442
443
    
    if [ "$TOOLDIR" != "aarch64" ]; then
        vt_set_efi_type="set 2 msftdata on"
    fi    
    
longpanda's avatar
longpanda committed
444
445
446
447
448
    parted -a none --script $DISK \
        mklabel gpt \
        unit s \
        mkpart Ventoy ntfs $part1_start_sector $part1_end_sector \
        mkpart VTOYEFI fat16 $part2_start_sector $part2_end_sector \
longpanda's avatar
longpanda committed
449
        $vt_set_efi_type \
longpanda's avatar
longpanda committed
450
451
        set 2 hidden on \
        quit
longpanda's avatar
longpanda committed
452

longpanda's avatar
update  
longpanda committed
453
454
    sync
    
longpanda's avatar
longpanda committed
455
    vtoycli gpt -f $DISK
longpanda's avatar
longpanda committed
456
457
    sync

longpanda's avatar
longpanda committed
458
    udevadm trigger --name-match=$DISK >/dev/null 2>&1
longpanda's avatar
longpanda committed
459
460
461
462
    partprobe >/dev/null 2>&1
    sleep 3
    echo "Done"

463
464
465
    echo 'Wait for partitions ...'
    for i in 0 1 2 3 4 5 6 7 8 9; do
        if [ -b $PART1 -a -b $PART2 ]; then
longpanda's avatar
longpanda committed
466
467
            break
        else
468
            echo "Wait for $PART1/$PART2 ..."
longpanda's avatar
longpanda committed
469
470
471
472
            sleep 1
        fi
    done

473
474
475
476
477
478
    if ! [ -b $PART1 ]; then
        MajorMinor=$(sed "s/:/ /" /sys/class/block/${PART1#/dev/}/dev)        
        echo "mknod -m 0660 $PART1 b $MajorMinor ..."
        mknod -m 0660 $PART1 b $MajorMinor
    fi
    
longpanda's avatar
longpanda committed
479
480
481
    if ! [ -b $PART2 ]; then
        MajorMinor=$(sed "s/:/ /" /sys/class/block/${PART2#/dev/}/dev)        
        echo "mknod -m 0660 $PART2 b $MajorMinor ..."
482
483
484
485
486
487
488
489
        mknod -m 0660 $PART2 b $MajorMinor        
    fi

    if [ -b $PART1 -a -b $PART2 ]; then
        echo "partition exist OK"
    else
        echo "[FAIL] $PART1/$PART2 does not exist"
        exit 1
longpanda's avatar
longpanda committed
490
491
492
    fi

    echo "create efi fat fs $PART2 ..."
longpanda's avatar
longpanda committed
493
    
longpanda's avatar
longpanda committed
494
    for i in 0 1 2 3 4 5 6 7 8 9; do
longpanda's avatar
longpanda committed
495
496
        check_umount_disk "$PART2"
        
497
        if mkfs.vfat -F 16 -n VTOYEFI -s 1 $PART2; then
longpanda's avatar
longpanda committed
498
499
500
501
502
503
504
505
506
507
508
            echo 'success'
            break
        else
            echo "$? retry ..."
            sleep 2
        fi
    done
}



509
510