2 #************************************************************************************
3 # Copyright (c) 2020, longpanda <admin@ventoy.net>
5 # This program is free software; you can redistribute it and/or
6 # modify it under the terms of the GNU General Public License as
7 # published by the Free Software Foundation; either version 3 of the
8 # License, or (at your option) any later version.
10 # This program is distributed in the hope that it will be useful, but
11 # WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 # General Public License for more details.
15 # You should have received a copy of the GNU General Public License
16 # along with this program; if not, see <http://www.gnu.org/licenses/>.
18 #************************************************************************************
21 BUSYBOX_PATH
=$VTOY_PATH/busybox
23 FIND
=$BUSYBOX_PATH/find
24 GREP
=$BUSYBOX_PATH/grep
25 EGREP
=$BUSYBOX_PATH/egrep
29 SLEEP
=$BUSYBOX_PATH/sleep
30 HEAD
=$BUSYBOX_PATH/head
31 VTOY_DM_PATH
=/dev
/mapper
/ventoy
32 VTOY_DEBUG_LEVEL
=$($BUSYBOX_PATH/hexdump -n 1 -s 450 -e '1/1 "%02x"' $VTOY_PATH/ventoy_os_param)
34 if [ "$VTOY_DEBUG_LEVEL" = "01" ]; then
35 if [ -e /dev
/console
]; then
41 if [ "$VTLOG" = "$VTOY_PATH/log" ]; then
44 echo -e "\033[32m $* \033[0m" > $VTLOG
50 if [ "$VTLOG" = "$VTOY_PATH/log" ]; then
53 echo -e "\n\033[31m $* \033[0m" > $VTLOG
59 is_ventoy_hook_finished
() {
60 [ -e $VTOY_PATH/hook_finish
]
63 set_ventoy_hook_finish
() {
64 echo 'Y' > $VTOY_PATH/hook_finish
67 get_ventoy_disk_name
() {
68 line
=$($VTOY_PATH/tool/vtoydump -f /ventoy/ventoy_os_param)
76 get_ventoy_iso_name
() {
77 line
=$($VTOY_PATH/tool/vtoydump -f /ventoy/ventoy_os_param)
85 wait_for_usb_disk_ready
() {
88 usb_disk
=$(get_ventoy_disk_name)
89 vtlog
"wait_for_usb_disk_ready $usb_disk ..."
91 if echo $usb_disk | $EGREP -q "nvme|mmc|nbd"; then
97 if [ -e "${vtpart2}" ]; then
98 vtlog
"wait_for_usb_disk_ready $usb_disk finish"
102 if [ $vtloop -gt 2 ]; then
103 if [ "$VTLOG" != "$VTOY_PATH/log" ]; then
104 $VTOY_PATH/tool
/vtoydump
-f /ventoy
/ventoy_os_param
-v > $VTLOG
113 check_usb_disk_ready
() {
114 if echo $1 | $EGREP -q "nvme|mmc|nbd"; then
124 if $VTOY_PATH/tool
/vtoydump
-f $VTOY_PATH/ventoy_os_param
-c "$1"; then
132 if echo $1 | $EGREP -q "nvme.*p$|mmc.*p$|nbd.*p$"; then
138 if $VTOY_PATH/tool
/vtoydump
-f $VTOY_PATH/ventoy_os_param
-c "$vtDiskName"; then
145 ventoy_get_vblade_bin
() {
146 if $VTOY_PATH/tool
/vblade_64
-t >>$VTLOG 2>&1; then
147 echo $VTOY_PATH/tool
/vblade_64
149 echo $VTOY_PATH/tool
/vblade_32
153 ventoy_find_bin_path
() {
154 if $BUSYBOX_PATH/which "$1" > /dev
/null
; then
155 $BUSYBOX_PATH/which "$1"; return
158 for vt_path
in '/bin' '/sbin' '/usr/bin' '/usr/sbin' '/usr/local/bin' '/usr/local/sbin' '/root/bin'; do
159 if [ -e "$vt_path/$1" ]; then
160 echo "$vt_path/$1"; return
168 ventoy_find_bin_run
() {
170 if [ "$1" = "sudo" ]; then
175 vtbinpath
=$(ventoy_find_bin_path "$1")
176 if [ -n "$vtbinpath" ]; then
179 if [ $vtsudo -eq 0 ]; then
180 vtlog
"$vtbinpath $*"
183 vtlog
"sudo $vtbinpath $*"
189 ventoy_get_module_postfix
() {
190 vtKerVer
=$($BUSYBOX_PATH/uname -r)
191 vtLine
=$($FIND /lib/modules/$vtKerVer/ -name *.ko* | $HEAD -n1)
192 vtComp
=${vtLine##*/*.ko}
196 ventoy_check_dm_module
() {
197 if $GREP -q 'device-mapper' /proc
/devices
; then
198 $BUSYBOX_PATH/true
; return
201 vtlog
"device-mapper NOT found in /proc/devices, try to load kernel module"
202 $BUSYBOX_PATH/modprobe dm_mod
>>$VTLOG 2>&1
203 $BUSYBOX_PATH/modprobe dm
-mod >>$VTLOG 2>&1
205 if ! $GREP -q 'device-mapper' /proc
/devices
; then
206 vtlog
"modprobe failed, now try to insmod ko..."
208 $FIND /lib
/modules
/ -name "dm-mod.ko*" | while read vtline
; do
209 vtlog
"insmode $vtline "
210 $BUSYBOX_PATH/insmod
$vtline >>$VTLOG 2>&1
214 if $GREP -q 'device-mapper' /proc
/devices
; then
215 vtlog
"device-mapper found in /proc/devices after retry"
216 $BUSYBOX_PATH/true
; return
218 vtlog
"device-mapper still NOT found in /proc/devices after retry"
219 $BUSYBOX_PATH/false
; return
223 create_ventoy_device_mapper
() {
224 vtlog
"create_ventoy_device_mapper $*"
226 VT_DM_BIN
=$(ventoy_find_bin_path dmsetup)
227 if [ -z "$VT_DM_BIN" ]; then
228 vtlog
"no dmsetup avaliable, lastly try inbox dmsetup"
229 VT_DM_BIN
=$VTOY_PATH/tool
/dmsetup
232 vtlog
"dmsetup avaliable in system $VT_DM_BIN"
234 if ventoy_check_dm_module
"$1"; then
235 vtlog
"device-mapper module check success"
237 vterr
"Error: no dm module avaliable"
240 $VTOY_PATH/tool
/vtoydm
-p -f $VTOY_PATH/ventoy_image_map
-d $1 > $VTOY_PATH/ventoy_dm_table
242 $VT_DM_BIN create ventoy
$VTOY_PATH/ventoy_dm_table
>>$VTLOG 2>&1
244 $VT_DM_BIN "$2" create ventoy
$VTOY_PATH/ventoy_dm_table
>>$VTLOG 2>&1
248 create_persistent_device_mapper
() {
249 vtlog
"create_persistent_device_mapper $*"
251 VT_DM_BIN
=$(ventoy_find_bin_path dmsetup)
252 if [ -z "$VT_DM_BIN" ]; then
253 vtlog
"no dmsetup avaliable, lastly try inbox dmsetup"
254 VT_DM_BIN
=$VTOY_PATH/tool
/dmsetup
257 vtlog
"dmsetup avaliable in system $VT_DM_BIN"
259 if ventoy_check_dm_module
"$1"; then
260 vtlog
"device-mapper module check success"
262 vterr
"Error: no dm module avaliable"
265 $VTOY_PATH/tool
/vtoydm
-p -f $VTOY_PATH/ventoy_persistent_map
-d $1 > $VTOY_PATH/persistent_dm_table
266 $VT_DM_BIN create vtoy_persistent
$VTOY_PATH/persistent_dm_table
>>$VTLOG 2>&1
271 wait_for_ventoy_dm_disk_label
() {
272 DM
=$($BUSYBOX_PATH/readlink $VTOY_DM_PATH)
273 vtlog
"wait_for_ventoy_dm_disk_label $DM ..."
275 for i
in 0 1 2 3 4 5 6 7 8 9; do
276 vtlog
"i=$i ####### ls /dev/disk/by-label/"
277 ls -l /dev
/disk
/by
-label/ >> $VTLOG
279 if ls -l /dev
/disk
/by
-label/ | $GREP -q "$DM"; then
288 if ! [ -e "$1" ]; then
293 if [ -d /tmp
/vtoy_udeb
]; then
294 $BUSYBOX_PATH/rm -rf /tmp
/vtoy_udeb
297 $BUSYBOX_PATH/mkdir -p /tmp
/vtoy_udeb
298 $BUSYBOX_PATH/cp -a "$1" /tmp
/vtoy_udeb
/
300 CURDIR
=$($BUSYBOX_PATH/pwd)
303 $BUSYBOX_PATH/ar x
"$1"
305 if [ -e 'data.tar.gz' ]; then
306 $BUSYBOX_PATH/tar -xzf data.
tar.gz
-C /
307 elif [ -e 'data.tar.xz' ]; then
308 $BUSYBOX_PATH/tar -xJf data.
tar.xz
-C /
309 elif [ -e 'data.tar.bz2' ]; then
310 $BUSYBOX_PATH/tar -xjf data.
tar.bz2
-C /
311 elif [ -e 'data.tar.lzma' ]; then
312 $BUSYBOX_PATH/tar -xaf data.
tar.lzma
-C /
315 if [ -e 'control.tar.gz' ]; then
316 $BUSYBOX_PATH/tar -xzf control.
tar.gz
-C /
317 elif [ -e 'control.tar.xz' ]; then
318 $BUSYBOX_PATH/tar -xJf control.
tar.xz
-C /
319 elif [ -e 'control.tar.bz2' ]; then
320 $BUSYBOX_PATH/tar -xjf control.
tar.bz2
-C /
321 elif [ -e 'control.tar.lzma' ]; then
322 $BUSYBOX_PATH/tar -xaf control.
tar.lzma
-C /
326 $BUSYBOX_PATH/rm -rf /tmp
/vtoy_udeb
331 install_udeb_from_line
() {
332 vtlog
"install_udeb_from_line $1"
334 if ! [ -b "$2" ]; then
335 vterr
"disk #$2# not exist"
339 sector
=$(echo $1 | $AWK '{print $(NF-1)}')
340 length=$(echo $1 | $AWK '{print $NF}')
341 vtlog "sector=$sector length=$length"
343 $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o /tmp/xxx.udeb
344 if [ -e /tmp/xxx.udeb ]; then
345 vtlog "extract udeb file from iso success"
347 vterr "extract udeb file from iso fail"
351 install_udeb_pkg /tmp/xxx.udeb
352 $BUSYBOX_PATH/rm -f /tmp/xxx.udeb
355 extract_file_from_line() {
356 vtlog "extract_file_from_line $1 disk=#$2#"
357 if ! [ -b "$2" ]; then
358 vterr "disk #$2# not exist"
362 sector=$(echo $1 | $AWK '{print $(NF-1)}')
363 length
=$(echo $1 | $AWK '{print $NF}')
364 vtlog
"sector=$sector length=$length"
366 $VTOY_PATH/tool
/vtoydm
-e -f $VTOY_PATH/ventoy_image_map
-d ${2} -s $sector -l $length -o $3
368 vtlog
"extract file from iso success"
371 vterr
"extract file from iso fail"
376 extract_rpm_from_line
() {
377 vtlog
"extract_rpm_from_line $1 disk=#$2#"
379 if ! [ -b "$2" ]; then
380 vterr
"disk #$2# not exist"
384 sector
=$(echo $1 | $AWK '{print $(NF-1)}')
385 length=$(echo $1 | $AWK '{print $NF}')
386 vtlog "sector=$sector length=$length"
388 $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o /tmp/xxx.rpm
389 if [ -e /tmp/xxx.rpm ]; then
390 vtlog "extract rpm file from iso success"
392 vterr "extract rpm file from iso fail"
396 CURPWD=$($BUSYBOX_PATH/pwd)
398 $BUSYBOX_PATH/mkdir -p $VTOY_PATH/rpm
400 vtlog "extract rpm..."
401 $BUSYBOX_PATH/rpm2cpio /tmp/xxx.rpm | $BUSYBOX_PATH/cpio -idm 2>>$VTLOG
404 $BUSYBOX_PATH/rm -f /tmp/xxx.rpm
407 install_rpm_from_line() {
408 vtlog "install_rpm_from_line $1 disk=#$2#"
410 if ! [ -b "$2" ]; then
411 vterr "disk #$2# not exist"
415 sector=$(echo $1 | $AWK '{print $(NF-1)}')
416 length
=$(echo $1 | $AWK '{print $NF}')
417 vtlog
"sector=$sector length=$length"
419 $VTOY_PATH/tool
/vtoydm
-e -f $VTOY_PATH/ventoy_image_map
-d ${2} -s $sector -l $length -o /tmp
/xxx.rpm
420 if [ -e /tmp
/xxx.rpm
]; then
421 vtlog
"extract rpm file from iso success"
423 vterr
"extract rpm file from iso fail"
427 CURPWD
=$($BUSYBOX_PATH/pwd)
430 vtlog
"install rpm..."
431 $BUSYBOX_PATH/rpm2cpio
/tmp
/xxx.rpm
| $BUSYBOX_PATH/cpio -idm 2>>$VTLOG
434 $BUSYBOX_PATH/rm -f /tmp
/xxx.rpm
437 dump_whole_iso_file
() {
438 $VTOY_PATH/tool
/vtoydm
-p -f $VTOY_PATH/ventoy_image_map
-d $usb_disk | while read vtline
; do
439 vtlog
"dmtable line: $vtline"
440 vtcount
=$(echo $vtline | $AWK '{print $2}')
441 vtoffset
=$(echo $vtline | $AWK '{print $NF}')
442 $BUSYBOX_PATH/dd if=$usb_disk of
="$1" bs
=512 count
=$vtcount skip
=$vtoffset oflag
=append conv
=notrunc
446 ventoy_copy_device_mapper
() {
447 if [ -L $VTOY_DM_PATH ]; then
448 vtlog
"replace block device link $1..."
449 $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_
${1#/dev/}
450 VT_MAPPER_LINK
=$($BUSYBOX_PATH/readlink $VTOY_DM_PATH)
451 $BUSYBOX_PATH/cp -a "/dev/mapper/$VT_MAPPER_LINK" "$1"
452 elif [ -b $VTOY_DM_PATH ]; then
453 vtlog
"replace block device $1..."
454 $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_
${1#/dev/}
455 $BUSYBOX_PATH/cp -a "$VTOY_DM_PATH" "$1"
458 vtlog
"$VTOY_DM_PATH not exist, now check /dev/dm-X ..."
459 VT_DM_BIN
=$(ventoy_find_bin_path dmsetup)
460 if [ -z "$VT_DM_BIN" ]; then
461 vtlog
"no dmsetup avaliable, lastly try inbox dmsetup"
462 VT_DM_BIN
=$VTOY_PATH/tool
/dmsetup
465 DM_VT_ID
=$($VT_DM_BIN ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.
*\
([0-9][0-9]*\
).
*/\
1 \
2/')
466 vtlog "DM_VT_ID=$DM_VT_ID ..."
467 $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_${1#/dev/}
468 $BUSYBOX_PATH/mknod -m 0666 "$1" b $DM_VT_ID
472 # create link for device-mapper
473 ventoy_create_persistent_link() {
474 blkdev_num=$($VTOY_PATH/tool/dmsetup ls | grep vtoy_persistent | sed 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1:\2/')
475 vtDM
=$(ventoy_find_dm_id ${blkdev_num})
477 if ! [ -d /dev
/disk
/by
-label ]; then
478 mkdir -p /dev
/disk
/by
-label
481 VTLABEL
=$($BUSYBOX_PATH/blkid /dev/$vtDM | $SED 's/.*LABEL="\([^"]*\)".*/\1/')
482 if [ -z "$VTLABEL" ]; then
486 vtlog "Persistent Label
: ##${VTLABEL}##"
488 if ! [ -e /dev
/disk
/by
-label/$VTLABEL ]; then
490 cd /dev
/disk
/by
-label
491 ln -s ..
/..
/$vtDM $VTLABEL
496 ventoy_udev_disk_common_hook
() {
497 if echo $1 | $EGREP -q "nvme.*p[0-9]$|mmc.*p[0-9]$|nbd.*p[0-9]$"; then
503 if [ -e /vtoy
/vtoy
]; then
509 # create device mapper for iso image file
510 if create_ventoy_device_mapper
"/dev/$VTDISK" $VTRWMOD; then
511 vtlog
"==== create ventoy device mapper success ===="
513 vtlog
"==== create ventoy device mapper failed ===="
517 if $GREP -q "/dev/$VTDISK" /proc
/mounts
; then
518 $GREP "/dev/$VTDISK" /proc
/mounts
| while read vtLine
; do
519 vtPart
=$(echo $vtLine | $AWK '{print $1}')
520 vtMnt
=$(echo $vtLine | $AWK '{print $2}')
521 vtlog
"$vtPart is mounted on $vtMnt now umount it ..."
522 $BUSYBOX_PATH/umount
$vtMnt
526 if create_ventoy_device_mapper
"/dev/$VTDISK" $VTRWMOD; then
527 vtlog
"==== create ventoy device mapper success after retry ===="
529 vtlog
"==== create ventoy device mapper failed after retry ===="
534 if [ "$2" = "noreplace" ]; then
535 vtlog
"no need to replace block device"
537 ventoy_copy_device_mapper
"/dev/$1"
540 if [ -f $VTOY_PATH/ventoy_persistent_map
]; then
541 create_persistent_device_mapper
"/dev/$VTDISK"
542 ventoy_create_persistent_link
546 ventoy_create_dev_ventoy_part
() {
547 blkdev_num
=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.
*\
([0-9][0-9]*\
).
*/\
1 \
2/')
548 $BUSYBOX_PATH/mknod -m 0666 /dev/ventoy b $blkdev_num
550 if [ -e /vtoy_dm_table ]; then
553 $CAT /vtoy_dm_table | while read vtline; do
554 echo $vtline > /ventoy/dm_table_part${vtPartid}
555 $VTOY_PATH/tool/dmsetup create ventoy${vtPartid} /ventoy/dm_table_part${vtPartid}
557 blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy${vtPartid} | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
558 $BUSYBOX_PATH/mknod
-m 0666 /dev
/ventoy
${vtPartid} b
$blkdev_num
560 vtPartid
=$(expr $vtPartid + 1)
566 ventoy_create_chromeos_ventoy_part
() {
567 blkdev_num
=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.
*\
([0-9][0-9]*\
).
*/\
1 \
2/')
568 $BUSYBOX_PATH/mknod -m 0666 /dev/ventoy b $blkdev_num
570 if [ -e /vtoy_dm_table ]; then
573 $CAT /vtoy_dm_table | while read vtline; do
574 echo $vtline > /ventoy/dm_table_part${vtPartid}
576 if [ $vtPartid -eq $1 ]; then
577 $VTOY_PATH/tool/dmsetup create ventoy${vtPartid} /ventoy/dm_table_part${vtPartid} --readonly
579 $VTOY_PATH/tool/dmsetup create ventoy${vtPartid} /ventoy/dm_table_part${vtPartid}
582 blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy${vtPartid} | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
583 $BUSYBOX_PATH/mknod
-m 0666 /dev
/ventoy
${vtPartid} b
$blkdev_num
585 vtPartid
=$(expr $vtPartid + 1)
590 is_inotify_ventoy_part
() {
591 if echo $1 | $GREP -q "2$"; then
592 if ! [ -e /sys
/block
/$1 ]; then
593 if [ -e /sys
/class
/block
/$1 ]; then
594 if echo $1 | $EGREP -q "nvme|mmc|nbd"; then
595 vtShortName
=${1:0:-2}
597 vtShortName
=${1:0:-1}
600 if [ -e /dev
/$vtShortName ]; then
601 $VTOY_PATH/tool
/vtoydump
-f $VTOY_PATH/ventoy_os_param
-c $vtShortName
611 ventoy_find_dm_id
() {
612 for vt
in $($BUSYBOX_PATH/ls /sys/block/); do
613 if [ "${vt:0:3}" = "dm-" ]; then
614 vtMajorMinor
=$($CAT /sys/block/$vt/dev)
615 if [ "$vtMajorMinor" = "$1" ]; then
624 ventoy_swap_device
() {
625 mv $1 $VTOY_PATH/swap_tmp_dev
627 mv $VTOY_PATH/swap_tmp_dev
$2
630 ventoy_extract_vtloopex
() {
632 $BUSYBOX_PATH/mkdir -p $VTOY_PATH/partmnt
$VTOY_PATH/vtloopex
633 $BUSYBOX_PATH/mount
-o ro
-t vfat
$1 $VTOY_PATH/partmnt
634 cd $VTOY_PATH/vtloopex
635 $CAT $VTOY_PATH/partmnt
/ventoy
/vtloopex.
cpio | $BUSYBOX_PATH/cpio -idm >> $VTLOG 2>&1
636 $BUSYBOX_PATH/umount
$VTOY_PATH/partmnt
637 $BUSYBOX_PATH/rm -rf $VTOY_PATH/partmnt
640 cd $VTOY_PATH/vtloopex
/$2/
641 $BUSYBOX_PATH/tar -xJf vtloopex.
tar.xz
647 ventoy_check_install_module_xz
() {
648 if [ -f "${1}.xz" ]; then
649 $BUSYBOX_PATH/xz
-d "${1}.xz"
650 $BUSYBOX_PATH/insmod
"$1"