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