IMG/cpio/ventoy/hook/ventoy-hook-lib.sh

   1 #!/ventoy/busybox/sh
   2 #************************************************************************************
   3 # Copyright (c) 2020, longpanda <admin@ventoy.net>
   4 #
   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.
   9 #
  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.
  14 #
  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/>.
  17 #
  18 #************************************************************************************
  19
  20 VTOY_PATH=/ventoy
  21 BUSYBOX_PATH=$VTOY_PATH/busybox
  22 VTLOG=$VTOY_PATH/log
  23 FIND=$BUSYBOX_PATH/find
  24 GREP=$BUSYBOX_PATH/grep
  25 EGREP=$BUSYBOX_PATH/egrep
  26 CAT=$BUSYBOX_PATH/cat
  27 AWK=$BUSYBOX_PATH/awk
  28 SED=$BUSYBOX_PATH/sed
  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)
  33 VTOY_LINUX_REMOUNT=$($BUSYBOX_PATH/hexdump -n 1 -s 454 -e '1/1 "%02x"' $VTOY_PATH/ventoy_os_param)
  34 VTOY_VLNK_BOOT=$($BUSYBOX_PATH/hexdump -n 1 -s 455 -e '1/1 "%02x"' $VTOY_PATH/ventoy_os_param)
  35
  36 if [ "$VTOY_DEBUG_LEVEL" = "01" ]; then
  37     if [ -e /dev/console ]; then
  38         VTLOG=/dev/console
  39     fi
  40 fi
  41
  42 vtlog() {
  43     if [ "$VTLOG" = "$VTOY_PATH/log" ]; then
  44         echo "$*" >>$VTLOG
  45     else
  46         echo -e "\033[32m $* \033[0m" > $VTLOG
  47         $SLEEP 2
  48     fi
  49 }
  50
  51 vterr() {
  52     if [ "$VTLOG" = "$VTOY_PATH/log" ]; then
  53         echo "$*" >>$VTLOG
  54     else
  55         echo -e "\n\033[31m $* \033[0m" > $VTLOG
  56         $SLEEP 30
  57     fi
  58 }
  59
  60
  61 is_ventoy_hook_finished() {
  62     [ -e $VTOY_PATH/hook_finish ]
  63 }
  64
  65 set_ventoy_hook_finish() {
  66     echo 'Y' > $VTOY_PATH/hook_finish
  67 }
  68
  69 get_ventoy_disk_name() {
  70     if [ "$VTOY_VLNK_BOOT" = "01" ]; then
  71         $VTOY_PATH/tool/vtoydump -t /ventoy/ventoy_os_param
  72     else
  73         line=$($VTOY_PATH/tool/vtoydump -f /ventoy/ventoy_os_param)
  74         if [ $? -eq 0 ]; then
  75             echo ${line%%#*}
  76         else
  77             echo "unknown"
  78         fi
  79     fi
  80 }
  81
  82 get_ventoy_iso_name() {
  83     line=$($VTOY_PATH/tool/vtoydump -f /ventoy/ventoy_os_param)
  84     if [ $? -eq 0 ]; then
  85         echo ${line##*#}
  86     else
  87         echo "unknown"
  88     fi
  89 }
  90
  91 wait_for_usb_disk_ready() {
  92     vtloop=0
  93     while [ -n "Y" ]; do
  94         usb_disk=$(get_ventoy_disk_name)
  95         vtlog "wait_for_usb_disk_ready $usb_disk ..."
  96
  97         if echo $usb_disk | $EGREP -q "nvme|mmc|nbd"; then
  98             vtpart2=${usb_disk}p2
  99         else
 100             vtpart2=${usb_disk}2
 101         fi
 102
 103         if [ -e "${vtpart2}" ]; then
 104             vtlog "wait_for_usb_disk_ready $usb_disk finish"
 105             break
 106         else
 107             let vtloop=vtloop+1
 108             if [ $vtloop -gt 2 ]; then
 109                 if [ "$VTLOG" != "$VTOY_PATH/log" ]; then
 110                     $VTOY_PATH/tool/vtoydump -f /ventoy/ventoy_os_param -v > $VTLOG
 111                 fi
 112             fi
 113             $SLEEP 0.3
 114         fi
 115     done
 116 }
 117
 118
 119 check_usb_disk_ready() {
 120     if echo $1 | $EGREP -q "nvme|mmc|nbd"; then
 121         vtpart2=${1}p2
 122     else
 123         vtpart2=${1}2
 124     fi
 125
 126     [ -e "${vtpart2}" ]
 127 }
 128
 129 not_ventoy_disk() {
 130     if echo $1 | $EGREP -q "nvme.*p$|mmc.*p$|nbd.*p$"; then
 131         vtDiskName=${1:0:-1}
 132     else
 133         vtDiskName=$1
 134     fi
 135
 136     if [ "$VTOY_VLNK_BOOT" = "01" ]; then
 137         vtVtoyDisk=$($VTOY_PATH/tool/vtoydump -t $VTOY_PATH/ventoy_os_param)
 138         [ "$vtVtoyDisk" != "/dev/$vtDiskName" ]
 139     else
 140         if $VTOY_PATH/tool/vtoydump -f $VTOY_PATH/ventoy_os_param -c "$vtDiskName"; then
 141             $BUSYBOX_PATH/false
 142         else
 143             $BUSYBOX_PATH/true
 144         fi
 145     fi
 146 }
 147
 148 ventoy_get_vblade_bin() {
 149     if $VTOY_PATH/tool/vblade_64 -t >>$VTLOG 2>&1; then
 150         echo $VTOY_PATH/tool/vblade_64
 151     else
 152         echo $VTOY_PATH/tool/vblade_32
 153     fi
 154 }
 155
 156 ventoy_find_bin_path() {
 157     if $BUSYBOX_PATH/which "$1" > /dev/null; then
 158         $BUSYBOX_PATH/which "$1"; return
 159     fi
 160
 161     for vt_path in '/bin' '/sbin' '/usr/bin' '/usr/sbin' '/usr/local/bin' '/usr/local/sbin' '/root/bin'; do
 162         if [ -e "$vt_path/$1" ]; then
 163             echo "$vt_path/$1"; return
 164         fi
 165     done
 166
 167     echo ""
 168 }
 169
 170
 171 ventoy_find_bin_run() {
 172     vtsudo=0
 173     if [ "$1" = "sudo" ]; then
 174         shift
 175         vtsudo=1
 176     fi
 177
 178     vtbinpath=$(ventoy_find_bin_path "$1")
 179     if [ -n "$vtbinpath" ]; then
 180         shift
 181
 182         if [ $vtsudo -eq 0 ]; then
 183             vtlog "$vtbinpath $*"
 184             $vtbinpath $*
 185         else
 186             vtlog "sudo $vtbinpath $*"
 187             sudo $vtbinpath $*
 188         fi
 189     fi
 190 }
 191
 192 ventoy_get_module_postfix() {
 193     vtKerVer=$($BUSYBOX_PATH/uname -r)
 194     vtLine=$($FIND /lib/modules/$vtKerVer/ -name *.ko* | $HEAD -n1)
 195     vtComp=${vtLine##*/*.ko}
 196     echo "ko$vtComp"
 197 }
 198
 199 ventoy_check_dm_module() {
 200     if $GREP -q 'device-mapper' /proc/devices; then
 201         $BUSYBOX_PATH/true; return
 202     fi
 203
 204     vtlog "device-mapper NOT found in /proc/devices, try to load kernel module"
 205     $BUSYBOX_PATH/modprobe dm_mod >>$VTLOG 2>&1
 206     $BUSYBOX_PATH/modprobe dm-mod >>$VTLOG 2>&1
 207
 208     if ! $GREP -q 'device-mapper' /proc/devices; then
 209         vtlog "modprobe failed, now try to insmod ko..."
 210
 211         $FIND /lib/modules/ -name "dm-mod.ko*" | while read vtline; do
 212             vtlog "insmod $vtline "
 213             $BUSYBOX_PATH/insmod $vtline >>$VTLOG 2>&1
 214             if [ $? -eq 0 ]; then
 215                 vtlog "insmod success"
 216             else
 217                 vtlog "insmod failed, try decompress"
 218                 if echo $vtline | $GREP -q "\.zst"; then
 219                     $VTOY_PATH/tool/zstdcat $vtline > $VTOY_PATH/extract_dm_mod.ko
 220                     $BUSYBOX_PATH/insmod $VTOY_PATH/extract_dm_mod.ko >>$VTLOG 2>&1
 221                 fi
 222             fi
 223         done
 224     fi
 225
 226     if $GREP -q 'device-mapper' /proc/devices; then
 227         vtlog "device-mapper found in /proc/devices after retry"
 228         $BUSYBOX_PATH/true; return
 229     else
 230         vtlog "device-mapper still NOT found in /proc/devices after retry"
 231         $BUSYBOX_PATH/false; return
 232     fi
 233 }
 234
 235
 236 create_ventoy_device_mapper() {
 237     vtlog "create_ventoy_device_mapper $*"
 238
 239     VT_DM_BIN=$(ventoy_find_bin_path dmsetup)
 240     if [ -z "$VT_DM_BIN" ]; then
 241         vtlog "no dmsetup avaliable, lastly try inbox dmsetup"
 242         VT_DM_BIN=$VTOY_PATH/tool/dmsetup
 243     fi
 244
 245     vtlog "dmsetup avaliable in system $VT_DM_BIN"
 246
 247     if ventoy_check_dm_module "$1"; then
 248         vtlog "device-mapper module check success"
 249     else
 250         vterr "Error: no dm module avaliable"
 251     fi
 252
 253     $VTOY_PATH/tool/vtoydm -p -f $VTOY_PATH/ventoy_image_map -d $1 > $VTOY_PATH/ventoy_dm_table
 254     $VTOY_PATH/tool/vtoydm -r -f $VTOY_PATH/ventoy_image_map -d $1 > $VTOY_PATH/ventoy_raw_table
 255
 256     if [ -z "$2" ]; then
 257         $VT_DM_BIN create ventoy $VTOY_PATH/ventoy_dm_table >>$VTLOG 2>&1
 258     else
 259         $VT_DM_BIN "$2" create ventoy $VTOY_PATH/ventoy_dm_table >>$VTLOG 2>&1
 260     fi
 261
 262     RAWDISKNAME=$($HEAD -n1 $VTOY_PATH/ventoy_raw_table | $AWK '{print $4}')
 263     $VT_DM_BIN create ${RAWDISKNAME#/dev/}  $VTOY_PATH/ventoy_raw_table >>$VTLOG 2>&1
 264 }
 265
 266 create_persistent_device_mapper() {
 267     vtlog "create_persistent_device_mapper $*"
 268
 269     VT_DM_BIN=$(ventoy_find_bin_path dmsetup)
 270     if [ -z "$VT_DM_BIN" ]; then
 271         vtlog "no dmsetup avaliable, lastly try inbox dmsetup"
 272         VT_DM_BIN=$VTOY_PATH/tool/dmsetup
 273     fi
 274
 275     vtlog "dmsetup avaliable in system $VT_DM_BIN"
 276
 277     if ventoy_check_dm_module "$1"; then
 278         vtlog "device-mapper module check success"
 279     else
 280         vterr "Error: no dm module avaliable"
 281     fi
 282
 283     $VTOY_PATH/tool/vtoydm -p -f $VTOY_PATH/ventoy_persistent_map -d $1 > $VTOY_PATH/persistent_dm_table
 284     $VT_DM_BIN create vtoy_persistent $VTOY_PATH/persistent_dm_table >>$VTLOG 2>&1
 285 }
 286
 287
 288
 289 wait_for_ventoy_dm_disk_label() {
 290     DM=$($BUSYBOX_PATH/readlink $VTOY_DM_PATH)
 291     vtlog "wait_for_ventoy_dm_disk_label $DM ..."
 292
 293     for i in 0 1 2 3 4 5 6 7 8 9; do
 294         vtlog "i=$i ####### ls /dev/disk/by-label/"
 295         ls -l /dev/disk/by-label/ >> $VTLOG
 296
 297         if ls -l /dev/disk/by-label/ | $GREP -q "$DM"; then
 298             break
 299         else
 300             $SLEEP 1
 301         fi
 302     done
 303 }
 304
 305 install_udeb_pkg() {
 306     if ! [ -e "$1" ]; then
 307         $BUSYBOX_PATH/false
 308         return
 309     fi
 310
 311     if [ -d /tmp/vtoy_udeb ]; then
 312         $BUSYBOX_PATH/rm -rf /tmp/vtoy_udeb
 313     fi
 314
 315     $BUSYBOX_PATH/mkdir -p /tmp/vtoy_udeb
 316     $BUSYBOX_PATH/cp -a "$1" /tmp/vtoy_udeb/
 317
 318     CURDIR=$($BUSYBOX_PATH/pwd)
 319     cd /tmp/vtoy_udeb
 320
 321     $BUSYBOX_PATH/ar x "$1"
 322
 323     if [ -e 'data.tar.gz' ]; then
 324         $BUSYBOX_PATH/tar -xzf data.tar.gz -C /
 325     elif [ -e 'data.tar.xz' ]; then
 326         $BUSYBOX_PATH/tar -xJf data.tar.xz -C /
 327     elif [ -e 'data.tar.bz2' ]; then
 328         $BUSYBOX_PATH/tar -xjf data.tar.bz2 -C /
 329     elif [ -e 'data.tar.lzma' ]; then
 330         $BUSYBOX_PATH/tar -xaf data.tar.lzma -C /
 331     fi
 332
 333     if [ -e 'control.tar.gz' ]; then
 334         $BUSYBOX_PATH/tar -xzf control.tar.gz -C /
 335     elif [ -e 'control.tar.xz' ]; then
 336         $BUSYBOX_PATH/tar -xJf control.tar.xz -C /
 337     elif [ -e 'control.tar.bz2' ]; then
 338         $BUSYBOX_PATH/tar -xjf control.tar.bz2 -C /
 339     elif [ -e 'control.tar.lzma' ]; then
 340         $BUSYBOX_PATH/tar -xaf control.tar.lzma -C /
 341     fi
 342
 343     cd $CURDIR
 344     $BUSYBOX_PATH/rm -rf /tmp/vtoy_udeb
 345     $BUSYBOX_PATH/true
 346 }
 347
 348
 349 install_udeb_from_line() {
 350     vtlog "install_udeb_from_line $1"
 351
 352     if ! [ -b "$2" ]; then
 353         vterr "disk #$2# not exist"
 354         return
 355     fi
 356
 357     sector=$(echo $1 | $AWK '{print $(NF-1)}')
 358     length=$(echo $1 | $AWK '{print $NF}')
 359     vtlog "sector=$sector  length=$length"
 360
 361     $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o /tmp/xxx.udeb
 362     if [ -e /tmp/xxx.udeb ]; then
 363         vtlog "extract udeb file from iso success"
 364     else
 365         vterr "extract udeb file from iso fail"
 366         return
 367     fi
 368
 369     install_udeb_pkg /tmp/xxx.udeb
 370     $BUSYBOX_PATH/rm -f /tmp/xxx.udeb
 371 }
 372
 373 extract_file_from_line() {
 374     vtlog "extract_file_from_line $1 disk=#$2#"
 375     if ! [ -b "$2" ]; then
 376         vterr "disk #$2# not exist"
 377         return
 378     fi
 379
 380     sector=$(echo $1 | $AWK '{print $(NF-1)}')
 381     length=$(echo $1 | $AWK '{print $NF}')
 382     vtlog "sector=$sector  length=$length"
 383
 384     $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o $3
 385     if [ -e $3 ]; then
 386         vtlog "extract file from iso success"
 387         $BUSYBOX_PATH/true
 388     else
 389         vterr "extract file from iso fail"
 390         $BUSYBOX_PATH/false
 391     fi
 392 }
 393
 394 extract_rpm_from_line() {
 395     vtlog "extract_rpm_from_line $1 disk=#$2#"
 396
 397     if ! [ -b "$2" ]; then
 398         vterr "disk #$2# not exist"
 399         return
 400     fi
 401
 402     sector=$(echo $1 | $AWK '{print $(NF-1)}')
 403     length=$(echo $1 | $AWK '{print $NF}')
 404     vtlog "sector=$sector  length=$length"
 405
 406     $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o /tmp/xxx.rpm
 407     if [ -e /tmp/xxx.rpm ]; then
 408         vtlog "extract rpm file from iso success"
 409     else
 410         vterr "extract rpm file from iso fail"
 411         return
 412     fi
 413
 414     CURPWD=$($BUSYBOX_PATH/pwd)
 415
 416     $BUSYBOX_PATH/mkdir -p $VTOY_PATH/rpm
 417     cd $VTOY_PATH/rpm
 418     vtlog "extract rpm..."
 419     $BUSYBOX_PATH/rpm2cpio /tmp/xxx.rpm | $BUSYBOX_PATH/cpio -idm 2>>$VTLOG
 420     cd $CURPWD
 421
 422     $BUSYBOX_PATH/rm -f /tmp/xxx.rpm
 423 }
 424
 425 install_rpm_from_line() {
 426     vtlog "install_rpm_from_line $1 disk=#$2#"
 427
 428     if ! [ -b "$2" ]; then
 429         vterr "disk #$2# not exist"
 430         return
 431     fi
 432
 433     sector=$(echo $1 | $AWK '{print $(NF-1)}')
 434     length=$(echo $1 | $AWK '{print $NF}')
 435     vtlog "sector=$sector  length=$length"
 436
 437     $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o /tmp/xxx.rpm
 438     if [ -e /tmp/xxx.rpm ]; then
 439         vtlog "extract rpm file from iso success"
 440     else
 441         vterr "extract rpm file from iso fail"
 442         return
 443     fi
 444
 445     CURPWD=$($BUSYBOX_PATH/pwd)
 446
 447     cd /
 448     vtlog "install rpm..."
 449     $BUSYBOX_PATH/rpm2cpio /tmp/xxx.rpm | $BUSYBOX_PATH/cpio -idm 2>>$VTLOG
 450     cd $CURPWD
 451
 452     $BUSYBOX_PATH/rm -f /tmp/xxx.rpm
 453 }
 454
 455 dump_whole_iso_file() {
 456    $VTOY_PATH/tool/vtoydm -p -f $VTOY_PATH/ventoy_image_map -d $usb_disk | while read vtline; do
 457         vtlog "dmtable line: $vtline"
 458         vtcount=$(echo $vtline | $AWK '{print $2}')
 459         vtoffset=$(echo $vtline | $AWK '{print $NF}')
 460         $BUSYBOX_PATH/dd if=$usb_disk of="$1" bs=512 count=$vtcount skip=$vtoffset oflag=append conv=notrunc
 461     done
 462 }
 463
 464 ventoy_copy_device_mapper() {
 465     if [ -L $VTOY_DM_PATH ]; then
 466         vtlog "replace block device link $1..."
 467         $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_${1#/dev/}
 468         VT_MAPPER_LINK=$($BUSYBOX_PATH/readlink $VTOY_DM_PATH)
 469         $BUSYBOX_PATH/cp -a "/dev/mapper/$VT_MAPPER_LINK" "$1"
 470     elif [ -b $VTOY_DM_PATH ]; then
 471         vtlog "replace block device $1..."
 472         $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_${1#/dev/}
 473         $BUSYBOX_PATH/cp -a "$VTOY_DM_PATH" "$1"
 474     else
 475
 476         vtlog "$VTOY_DM_PATH not exist, now check /dev/dm-X ..."
 477         VT_DM_BIN=$(ventoy_find_bin_path dmsetup)
 478         if [ -z "$VT_DM_BIN" ]; then
 479             vtlog "no dmsetup avaliable, lastly try inbox dmsetup"
 480             VT_DM_BIN=$VTOY_PATH/tool/dmsetup
 481         fi
 482
 483         DM_VT_ID=$($VT_DM_BIN ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
 484         vtlog "DM_VT_ID=$DM_VT_ID ..."
 485         $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_${1#/dev/}
 486         $BUSYBOX_PATH/mknod -m 0666 "$1" b $DM_VT_ID
 487     fi
 488 }
 489
 490 # create link for device-mapper
 491 ventoy_create_persistent_link() {
 492     blkdev_num=$($VTOY_PATH/tool/dmsetup ls | grep vtoy_persistent | sed 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1:\2/')
 493     vtDM=$(ventoy_find_dm_id ${blkdev_num})
 494
 495     if ! [ -d /dev/disk/by-label ]; then
 496         mkdir -p /dev/disk/by-label
 497     fi
 498
 499     VTLABEL=$($BUSYBOX_PATH/blkid /dev/$vtDM | $SED 's/.*LABEL="\([^"]*\)".*/\1/')
 500     if [ -z "$VTLABEL" ]; then
 501         VTLABEL=casper-rw
 502     fi
 503
 504     vtlog "Persistent Label: ##${VTLABEL}##"
 505
 506     if ! [ -e /dev/disk/by-label/$VTLABEL ]; then
 507         vtOldDir=$PWD
 508         cd /dev/disk/by-label
 509         ln -s ../../$vtDM $VTLABEL
 510         cd $vtOldDir
 511     fi
 512 }
 513
 514 ventoy_partname_to_diskname() {
 515     if echo $1 | $EGREP -q "nvme.*p[0-9]$|mmc.*p[0-9]$|nbd.*p[0-9]$"; then
 516         echo -n "${1:0:-2}"
 517     else
 518         echo -n "${1:0:-1}"
 519     fi
 520 }
 521
 522 ventoy_diskname_to_partname() {
 523     if echo $1 | $EGREP -q "nvme.*p[0-9]$|mmc.*p[0-9]$|nbd.*p[0-9]$"; then
 524         echo -n "${1}p$2"
 525     else
 526         echo -n "${1}$2"
 527     fi
 528 }
 529
 530 ventoy_udev_disk_common_hook() {
 531     if echo $1 | $EGREP -q "nvme.*p[0-9]$|mmc.*p[0-9]$|nbd.*p[0-9]$"; then
 532         VTDISK="${1:0:-2}"
 533     else
 534         VTDISK="${1:0:-1}"
 535     fi
 536
 537     if [ -e /vtoy/vtoy ]; then
 538         VTRWMOD=""
 539     else
 540         VTRWMOD="--readonly"
 541     fi
 542
 543     # create device mapper for iso image file
 544     if create_ventoy_device_mapper "/dev/$VTDISK" $VTRWMOD; then
 545         vtlog "==== create ventoy device mapper success ===="
 546     else
 547         vtlog "==== create ventoy device mapper failed ===="
 548
 549         $SLEEP 3
 550
 551         if $GREP -q "/dev/$VTDISK" /proc/mounts; then
 552             $GREP "/dev/$VTDISK" /proc/mounts | while read vtLine; do
 553                 vtPart=$(echo $vtLine | $AWK '{print $1}')
 554                 vtMnt=$(echo $vtLine | $AWK '{print $2}')
 555                 vtlog "$vtPart is mounted on $vtMnt  now umount it ..."
 556                 $BUSYBOX_PATH/umount $vtMnt
 557             done
 558         fi
 559
 560         if create_ventoy_device_mapper "/dev/$VTDISK" $VTRWMOD; then
 561             vtlog "==== create ventoy device mapper success after retry ===="
 562         else
 563             vtlog "==== create ventoy device mapper failed after retry ===="
 564             return
 565         fi
 566     fi
 567
 568     if [ "$2" = "noreplace" ]; then
 569         vtlog "no need to replace block device"
 570     else
 571         ventoy_copy_device_mapper "/dev/$1"
 572     fi
 573
 574     if [ -f $VTOY_PATH/ventoy_persistent_map ]; then
 575         create_persistent_device_mapper "/dev/$VTDISK"
 576         ventoy_create_persistent_link
 577     fi
 578 }
 579
 580 ventoy_create_dev_ventoy_part() {
 581     blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
 582     $BUSYBOX_PATH/mknod -m 0666 /dev/ventoy b $blkdev_num
 583
 584     if [ -e /vtoy_dm_table ]; then
 585         vtPartid=1
 586         $CAT /vtoy_dm_table | while read vtline; do
 587             echo $vtline > /ventoy/dm_table_part${vtPartid}
 588             $VTOY_PATH/tool/dmsetup create ventoy${vtPartid} /ventoy/dm_table_part${vtPartid}
 589
 590             blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy${vtPartid} | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
 591             $BUSYBOX_PATH/mknod -m 0666 /dev/ventoy${vtPartid} b $blkdev_num
 592
 593             vtPartid=$(expr $vtPartid + 1)
 594         done
 595     fi
 596 }
 597
 598
 599 ventoy_create_chromeos_ventoy_part() {
 600     blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
 601     $BUSYBOX_PATH/mknod -m 0666 /dev/ventoy b $blkdev_num
 602
 603     if [ -e /vtoy_dm_table ]; then
 604         vtPartid=1
 605
 606         $CAT /vtoy_dm_table | while read vtline; do
 607             echo $vtline > /ventoy/dm_table_part${vtPartid}
 608
 609             if [ $vtPartid -eq $1 ]; then
 610                 $VTOY_PATH/tool/dmsetup create ventoy${vtPartid} /ventoy/dm_table_part${vtPartid} --readonly
 611             else
 612                 $VTOY_PATH/tool/dmsetup create ventoy${vtPartid} /ventoy/dm_table_part${vtPartid}
 613             fi
 614
 615             blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy${vtPartid} | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
 616             $BUSYBOX_PATH/mknod -m 0666 /dev/ventoy${vtPartid} b $blkdev_num
 617
 618             vtPartid=$(expr $vtPartid + 1)
 619         done
 620     fi
 621 }
 622
 623 is_inotify_ventoy_part() {
 624     if echo $1 | $GREP -q "2$"; then
 625         if ! [ -e /sys/block/$1 ]; then
 626             if [ -e /sys/class/block/$1 ]; then
 627                 if echo $1 | $EGREP -q "nvme|mmc|nbd"; then
 628                     vtShortName=${1:0:-2}
 629                 else
 630                     vtShortName=${1:0:-1}
 631                 fi
 632
 633                 if [ -e /dev/$vtShortName ]; then
 634                     if [ "$VTOY_VLNK_BOOT" = "01" ]; then
 635                         vtOrgDiskName=$($VTOY_PATH/tool/vtoydump -t $VTOY_PATH/ventoy_os_param)
 636                         [ "$vtOrgDiskName" = "/dev/$vtShortName" ]
 637                     else
 638                         $VTOY_PATH/tool/vtoydump -f $VTOY_PATH/ventoy_os_param -c $vtShortName
 639                     fi
 640                     return
 641                 fi
 642             fi
 643         fi
 644     fi
 645
 646     [ "1" = "0" ]
 647 }
 648
 649 ventoy_find_dm_id() {
 650     for vt in $($BUSYBOX_PATH/ls /sys/block/); do
 651         if [ "${vt:0:3}" = "dm-" ]; then
 652             vtMajorMinor=$($CAT /sys/block/$vt/dev)
 653             if [ "$vtMajorMinor" = "$1" ]; then
 654                 echo ${vt}
 655                 return
 656             fi
 657         fi
 658     done
 659     echo 'xx'
 660 }
 661
 662 ventoy_swap_device() {
 663     mv $1 $VTOY_PATH/swap_tmp_dev
 664     mv $2 $1
 665     mv $VTOY_PATH/swap_tmp_dev $2
 666 }
 667
 668 ventoy_extract_vtloopex() {
 669     vtCurPwd=$PWD
 670     $BUSYBOX_PATH/mkdir -p $VTOY_PATH/partmnt $VTOY_PATH/vtloopex
 671     $BUSYBOX_PATH/mount -o ro -t vfat $1  $VTOY_PATH/partmnt
 672     cd $VTOY_PATH/vtloopex
 673     $CAT $VTOY_PATH/partmnt/ventoy/vtloopex.cpio | $BUSYBOX_PATH/cpio -idm >> $VTLOG 2>&1
 674     $BUSYBOX_PATH/umount $VTOY_PATH/partmnt
 675     $BUSYBOX_PATH/rm -rf $VTOY_PATH/partmnt
 676
 677     if [ -n "$2" ]; then
 678         cd $VTOY_PATH/vtloopex/$2/
 679         $BUSYBOX_PATH/tar -xJf vtloopex.tar.xz
 680     fi
 681
 682     cd $vtCurPwd
 683 }
 684
 685 ventoy_check_install_module_xz() {
 686     if [ -f "${1}.xz" ]; then
 687         $BUSYBOX_PATH/xz -d  "${1}.xz"
 688         $BUSYBOX_PATH/insmod "$1"
 689     fi
 690 }
 691
 692 ventoy_check_umount() {
 693     for vtLoop in 0 1 2 3 4 5 6 7 8 9; do
 694         $BUSYBOX_PATH/umount "$1" > /dev/null 2>&1
 695         if $BUSYBOX_PATH/mountpoint -q "$1"; then
 696             $SLEEP 1
 697         else
 698             break
 699         fi
 700     done
 701 }
 702
 703 ventoy_wait_dir() {
 704     vtdir=$1
 705     vtsec=0
 706
 707     while [ $vtsec -lt $2 ]; do
 708         if [ -d "$vtdir" ]; then
 709             break
 710         else
 711             $SLEEP 1
 712             vtsec=$(expr $vtsec + 1)
 713         fi
 714     done
 715 }