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 ventoy_need_proc_ibt() {
 236     vtKv=$($BUSYBOX_PATH/uname -r)
 237     vtMajor=$(echo $vtKv | $AWK -F. '{print $1}')
 238     vtMinor=$(echo $vtKv | $AWK -F. '{print $2}')
 239
 240     #ibt was supported since linux kernel 5.18
 241     if [ $vtMajor -lt 5 ]; then
 242         $BUSYBOX_PATH/false; return
 243     elif [ $vtMajor -eq 5 ]; then
 244         if [ $vtMajor -lt 18 ]; then
 245             $BUSYBOX_PATH/false; return
 246         fi
 247     fi
 248
 249     if $GREP -q ' ibt=off' /proc/cmdline; then
 250         $BUSYBOX_PATH/false; return
 251     fi
 252
 253     #hardware CPU doesn't support IBT
 254     if $VTOY_PATH/tool/vtoykmod -I; then
 255         :
 256     else
 257         $BUSYBOX_PATH/false; return
 258     fi
 259
 260     #dot.CONFIG not enabled
 261     if $GREP -q ' ibt_restore$' /proc/kallsyms; then
 262         :
 263     else
 264         $BUSYBOX_PATH/false; return
 265     fi
 266
 267     $BUSYBOX_PATH/true
 268 }
 269
 270
 271 ventoy_need_dm_patch() {
 272     if [ "$VTOY_LINUX_REMOUNT" != "01" ]; then
 273         if $GREP -q 'VTOY_LINUX_REMOUNT=1' /proc/cmdline; then
 274             :
 275         else
 276             $BUSYBOX_PATH/false; return
 277         fi
 278     fi
 279
 280     if $GREP -q 'device-mapper' /proc/devices; then
 281         :
 282     else
 283         $BUSYBOX_PATH/false; return
 284     fi
 285
 286     if $GREP -q 'dm_patch' /proc/modules; then
 287         $BUSYBOX_PATH/false; return
 288     fi
 289
 290     vtMajorVer=$($BUSYBOX_PATH/uname -r | $AWK -F. '{print $1}')
 291     vtMinorVer=$($BUSYBOX_PATH/uname -r | $AWK -F. '{print $2}')
 292
 293     if [ $vtMajorVer -lt 3 ]; then
 294         $BUSYBOX_PATH/false; return
 295     elif [ $vtMajorVer -eq 3 -a $vtMinorVer -lt 10 ]; then
 296         $BUSYBOX_PATH/false; return
 297     fi
 298
 299
 300     $BUSYBOX_PATH/true
 301 }
 302
 303 ventoy_dm_patch() {
 304     vtDmPatchDebug=0
 305     vtMType=$($BUSYBOX_PATH/uname -m)
 306
 307     vtlog "######### ventoy_dm_patch ############"
 308
 309     if echo $vtMType | $EGREP -i -q "x86.64|amd64"; then
 310         vtKoName=dm_patch_64.ko
 311     elif echo $vtMType | $EGREP -i -q "i[3-6]86"; then
 312         vtKoName=dm_patch_32.ko
 313     else
 314         vtlog "unsupported machine type $vtMType"
 315         return
 316     fi
 317
 318     if ventoy_need_proc_ibt; then
 319         vtlog "need to proc IBT"
 320         vtKoName=dm_patch_ibt_64.ko
 321         vtIBT='0x8888'
 322     else
 323         vtlog "NO need to proc IBT"
 324         vtIBT='0'
 325     fi
 326
 327
 328     if [ -f $VTOY_PATH/tool/$vtKoName ]; then
 329         vtlog "/ventoy/tool/$vtKoName exist OK"
 330     else
 331         vtlog "/ventoy/tool/$vtKoName NOT exist"
 332         return
 333     fi
 334
 335     $CAT /proc/kallsyms | $BUSYBOX_PATH/sort > $VTOY_PATH/kallsyms
 336
 337     if $GREP -m1 -q 'open_table_device.isra' $VTOY_PATH/kallsyms; then
 338         vtISRA=$($GREP -m1 'open_table_device.isra' $VTOY_PATH/kallsyms | $AWK '{print $3}')
 339         vtLine=$($VTOY_PATH/tool/vtoyksym $vtISRA $VTOY_PATH/kallsyms)
 340         vtlog "get $vtISRA address $vtLine"
 341     else
 342         vtLine=$($VTOY_PATH/tool/vtoyksym dm_get_table_device $VTOY_PATH/kallsyms)
 343         vtlog "get dm_get_table_device address $vtLine"
 344     fi
 345     get_addr=$(echo $vtLine | $AWK '{print $1}')
 346     get_size=$(echo $vtLine | $AWK '{print $2}')
 347
 348     vtLine=$($VTOY_PATH/tool/vtoyksym blkdev_get_by_dev $VTOY_PATH/kallsyms)
 349     vtlog "get blkdev_get_by_dev address $vtLine"
 350     blkdev_get_addr=$(echo $vtLine | $AWK '{print $1}')
 351
 352     vtLine=$($VTOY_PATH/tool/vtoyksym blkdev_put $VTOY_PATH/kallsyms)
 353     vtlog "get blkdev_put address $vtLine"
 354     blkdev_put_addr=$(echo $vtLine | $AWK '{print $1}')
 355
 356     vtLine=$($VTOY_PATH/tool/vtoyksym bdev_open_by_dev $VTOY_PATH/kallsyms)
 357     vtlog "get bdev_open_by_dev address $vtLine"
 358     bdev_open_addr=$(echo $vtLine | $AWK '{print $1}')
 359
 360     vtLine=$($VTOY_PATH/tool/vtoyksym bdev_file_open_by_dev $VTOY_PATH/kallsyms)
 361     vtlog "get bdev_file_open_by_dev address $vtLine"
 362     bdev_file_open_addr=$(echo $vtLine | $AWK '{print $1}')
 363
 364
 365     if $GREP -m1 -q 'close_table_device.isra' $VTOY_PATH/kallsyms; then
 366         vtLine=$($VTOY_PATH/tool/vtoyksym close_table_device.isra $VTOY_PATH/kallsyms)
 367         vtlog "get close_table_device.isra address $vtLine"
 368     else
 369         vtLine=$($VTOY_PATH/tool/vtoyksym dm_put_table_device $VTOY_PATH/kallsyms)
 370         vtlog "get dm_put_table_device address $vtLine"
 371     fi
 372     put_addr=$(echo $vtLine | $AWK '{print $1}')
 373     put_size=$(echo $vtLine | $AWK '{print $2}')
 374
 375     ro_addr=$($GREP ' set_memory_ro$' /proc/kallsyms | $AWK '{print $1}')
 376     rw_addr=$($GREP ' set_memory_rw$' /proc/kallsyms | $AWK '{print $1}')
 377     kprobe_reg_addr=$($GREP ' register_kprobe$' /proc/kallsyms | $AWK '{print $1}')
 378     kprobe_unreg_addr=$($GREP ' unregister_kprobe$' /proc/kallsyms | $AWK '{print $1}')
 379
 380     if [ "$VTOY_DEBUG_LEVEL" = "01" ]; then
 381         vtDmPatchDebug=1
 382     fi
 383
 384     if $GREP -q 'dmpatch_debug' /proc/cmdline; then
 385         vtDmPatchDebug=1
 386     fi
 387
 388
 389     if [ $vtDmPatchDebug -eq 1 ]; then
 390         printk_addr=$($GREP ' printk$' /proc/kallsyms | $AWK '{print $1}')
 391         if [ -z "$printk_addr" ]; then
 392             printk_addr=$($GREP ' _printk$' /proc/kallsyms | $AWK '{print $1}')
 393         fi
 394         vtDebug="-v"
 395     else
 396         printk_addr=0
 397     fi
 398
 399     #printk_addr=$($GREP ' printk$' /proc/kallsyms | $AWK '{print $1}')
 400     #vtDebug="-v"
 401
 402     vtlog get_addr=$get_addr  get_size=$get_size
 403     vtlog put_addr=$put_addr  put_size=$put_size
 404     vtlog blkdev_get_addr=$blkdev_get_addr blkdev_put_addr=$blkdev_put_addr
 405     vtlog kprobe_reg_addr=$kprobe_reg_addr  kprobe_unreg_addr=$kprobe_unreg_addr
 406     vtlog ro_addr=$ro_addr  rw_addr=$rw_addr  printk_addr=$printk_addr bdev_open_addr=$bdev_open_addr bdev_file_open_addr=$bdev_file_open_addr
 407
 408     if [ "$get_addr" = "0" -o "$put_addr" = "0" ]; then
 409         vtlog "Invalid symbol address"
 410         return
 411     fi
 412     if [ "$ro_addr" = "0" -o "$rw_addr" = "0" ]; then
 413         vtlog "Invalid symbol address"
 414         return
 415     fi
 416
 417
 418     vtKv=$($BUSYBOX_PATH/uname -r)
 419     vtKVMajor=$(echo $vtKv | $AWK -F. '{print $1}')
 420     vtKVMinor=$(echo $vtKv | $AWK -F. '{print $2}')
 421     vtKVSubMinor=$(echo $vtKv | $AWK -F. '{print $3}')
 422
 423     if [ ! -d /lib/modules/$vtKv ]; then
 424         vtlog "No modules directory found"
 425         return
 426     elif [ -d /lib/modules/$vtKv/kernel/fs ]; then
 427         vtModPath=$($FIND /lib/modules/$vtKv/kernel/fs/ -name "*.ko*" | $HEAD -n1)
 428     elif [ -d /lib/modules/$vtKv/kernel ]; then
 429         vtModPath=$($FIND /lib/modules/$vtKv/kernel/ -name "xfs.ko*" | $HEAD -n1)
 430     elif [ -d /lib/modules/$vtKv/initrd ]; then
 431         vtModPath=$($FIND /lib/modules/$vtKv/initrd/ -name "xfs.ko*" | $HEAD -n1)
 432     fi
 433
 434
 435     if [ -z "$vtModPath" ]; then
 436         vtModPath=$($FIND /lib/modules/$vtKv/ -name "*.ko*" | $HEAD -n1)
 437     fi
 438
 439     vtModName=$($BUSYBOX_PATH/basename $vtModPath)
 440
 441     vtlog "template module is $vtModPath $vtModName"
 442
 443     if [ -z "$vtModPath" ]; then
 444         vtlog "No template module found"
 445         return
 446     elif echo $vtModPath | $GREP -q "[.]ko$"; then
 447         $BUSYBOX_PATH/cp -a $vtModPath  $VTOY_PATH/$vtModName
 448     elif echo $vtModPath | $GREP -q "[.]ko[.]xz$"; then
 449         $BUSYBOX_PATH/xzcat $vtModPath > $VTOY_PATH/$vtModName
 450     elif echo $vtModPath | $GREP -q "[.]ko[.]gz$"; then
 451         $BUSYBOX_PATH/zcat $vtModPath > $VTOY_PATH/$vtModName
 452     elif echo $vtModPath | $GREP -q "[.]ko[.]zst$"; then
 453         $VTOY_PATH/tool/zstdcat $vtModPath > $VTOY_PATH/$vtModName
 454     else
 455         vtlog "unsupport module type"
 456         return
 457     fi
 458
 459
 460
 461     #step1: modify vermagic/mod crc/relocation
 462     vtlog "$VTOY_PATH/tool/vtoykmod -u $vtKVMajor $vtKVMinor $VTOY_PATH/tool/$vtKoName $VTOY_PATH/$vtModName $vtDebug"
 463     $VTOY_PATH/tool/vtoykmod -u $vtKVMajor $vtKVMinor $VTOY_PATH/tool/$vtKoName $VTOY_PATH/$vtModName $vtDebug >>$VTLOG 2>&1
 464
 465     #step2: fill parameters
 466     vtPgsize=$($VTOY_PATH/tool/vtoyksym -p)
 467
 468     vtPrams="$VTOY_PATH/tool/$vtKoName $vtPgsize 0x$printk_addr 0x$ro_addr 0x$rw_addr $get_addr $get_size $put_addr $put_size 0x$kprobe_reg_addr 0x$kprobe_unreg_addr $vtKVMajor $vtIBT $vtKVMinor $blkdev_get_addr $blkdev_put_addr $vtKVSubMinor $bdev_open_addr $bdev_file_open_addr  $vtDebug"
 469
 470
 471     vtlog "$VTOY_PATH/tool/vtoykmod -f $vtPrams"
 472     $VTOY_PATH/tool/vtoykmod -f $vtPrams >>$VTLOG 2>&1
 473
 474
 475     vtlog "insmod $VTOY_PATH/tool/$vtKoName"
 476     $BUSYBOX_PATH/insmod $VTOY_PATH/tool/$vtKoName  >>$VTLOG 2>&1
 477
 478     if $GREP -q 'dm_patch' /proc/modules; then
 479         vtlog "dm_patch module OK"
 480         echo "done" > $VTOY_PATH/dm_patch_done
 481     else
 482         vtlog "dm_patch module FAILED"
 483     fi
 484
 485 }
 486
 487 create_ventoy_device_mapper() {
 488     vtlog "create_ventoy_device_mapper $*"
 489
 490     VT_DM_BIN=$(ventoy_find_bin_path dmsetup)
 491     if [ -z "$VT_DM_BIN" ]; then
 492         vtlog "no dmsetup avaliable, lastly try inbox dmsetup"
 493         VT_DM_BIN=$VTOY_PATH/tool/dmsetup
 494     fi
 495
 496     vtlog "dmsetup avaliable in system $VT_DM_BIN"
 497
 498     if ventoy_check_dm_module "$1"; then
 499         vtlog "device-mapper module check success"
 500     else
 501         vterr "Error: no dm module avaliable"
 502     fi
 503
 504     $VTOY_PATH/tool/vtoydm -p -f $VTOY_PATH/ventoy_image_map -d $1 > $VTOY_PATH/ventoy_dm_table
 505
 506
 507     vtLevel1=$($CAT /proc/sys/kernel/printk | $AWK '{print $1}')
 508     vtLevel2=$($CAT /proc/sys/kernel/printk | $AWK '{print $2}')
 509     vtLevel3=$($CAT /proc/sys/kernel/printk | $AWK '{print $3}')
 510     vtLevel4=$($CAT /proc/sys/kernel/printk | $AWK '{print $4}')
 511     if ventoy_need_dm_patch; then
 512         ventoy_dm_patch
 513         #suppress printk message
 514         echo 0 $vtLevel2 0 $vtLevel4 > /proc/sys/kernel/printk
 515     fi
 516
 517     if [ -z "$2" ]; then
 518         $VT_DM_BIN create ventoy $VTOY_PATH/ventoy_dm_table >>$VTLOG 2>&1
 519     else
 520         $VT_DM_BIN "$2" create ventoy $VTOY_PATH/ventoy_dm_table >>$VTLOG 2>&1
 521     fi
 522
 523     if ventoy_need_dm_patch; then
 524         #recover printk level
 525         echo $vtLevel1 $vtLevel2 $vtLevel3 $vtLevel4 > /proc/sys/kernel/printk
 526     fi
 527 }
 528
 529 create_persistent_device_mapper() {
 530     vtlog "create_persistent_device_mapper $*"
 531
 532     VT_DM_BIN=$(ventoy_find_bin_path dmsetup)
 533     if [ -z "$VT_DM_BIN" ]; then
 534         vtlog "no dmsetup avaliable, lastly try inbox dmsetup"
 535         VT_DM_BIN=$VTOY_PATH/tool/dmsetup
 536     fi
 537
 538     vtlog "dmsetup avaliable in system $VT_DM_BIN"
 539
 540     if ventoy_check_dm_module "$1"; then
 541         vtlog "device-mapper module check success"
 542     else
 543         vterr "Error: no dm module avaliable"
 544     fi
 545
 546     $VTOY_PATH/tool/vtoydm -p -f $VTOY_PATH/ventoy_persistent_map -d $1 > $VTOY_PATH/persistent_dm_table
 547
 548
 549     vtLevel1=$($CAT /proc/sys/kernel/printk | $AWK '{print $1}')
 550     vtLevel2=$($CAT /proc/sys/kernel/printk | $AWK '{print $2}')
 551     vtLevel3=$($CAT /proc/sys/kernel/printk | $AWK '{print $3}')
 552     vtLevel4=$($CAT /proc/sys/kernel/printk | $AWK '{print $4}')
 553     if [ -f $VTOY_PATH/dm_patch_done ]; then
 554         #suppress printk message
 555         echo 0 $vtLevel2 0 $vtLevel4 > /proc/sys/kernel/printk
 556     fi
 557
 558     $VT_DM_BIN create vtoy_persistent $VTOY_PATH/persistent_dm_table >>$VTLOG 2>&1
 559
 560     if [ -f $VTOY_PATH/dm_patch_done ]; then
 561         #recover printk level
 562         echo $vtLevel1 $vtLevel2 $vtLevel3 $vtLevel4 > /proc/sys/kernel/printk
 563     fi
 564 }
 565
 566
 567
 568 wait_for_ventoy_dm_disk_label() {
 569     DM=$($BUSYBOX_PATH/readlink $VTOY_DM_PATH)
 570     vtlog "wait_for_ventoy_dm_disk_label $DM ..."
 571
 572     for i in 0 1 2 3 4 5 6 7 8 9; do
 573         vtlog "i=$i ####### ls /dev/disk/by-label/"
 574         ls -l /dev/disk/by-label/ >> $VTLOG
 575
 576         if ls -l /dev/disk/by-label/ | $GREP -q "$DM"; then
 577             break
 578         else
 579             $SLEEP 1
 580         fi
 581     done
 582 }
 583
 584 install_udeb_pkg() {
 585     if ! [ -e "$1" ]; then
 586         $BUSYBOX_PATH/false
 587         return
 588     fi
 589
 590     if [ -d /tmp/vtoy_udeb ]; then
 591         $BUSYBOX_PATH/rm -rf /tmp/vtoy_udeb
 592     fi
 593
 594     $BUSYBOX_PATH/mkdir -p /tmp/vtoy_udeb
 595     $BUSYBOX_PATH/cp -a "$1" /tmp/vtoy_udeb/
 596
 597     CURDIR=$($BUSYBOX_PATH/pwd)
 598     cd /tmp/vtoy_udeb
 599
 600     $BUSYBOX_PATH/ar x "$1"
 601
 602     if [ -e 'data.tar.gz' ]; then
 603         $BUSYBOX_PATH/tar -xzf data.tar.gz -C /
 604     elif [ -e 'data.tar.xz' ]; then
 605         $BUSYBOX_PATH/tar -xJf data.tar.xz -C /
 606     elif [ -e 'data.tar.bz2' ]; then
 607         $BUSYBOX_PATH/tar -xjf data.tar.bz2 -C /
 608     elif [ -e 'data.tar.lzma' ]; then
 609         $BUSYBOX_PATH/tar -xaf data.tar.lzma -C /
 610     fi
 611
 612     if [ -e 'control.tar.gz' ]; then
 613         $BUSYBOX_PATH/tar -xzf control.tar.gz -C /
 614     elif [ -e 'control.tar.xz' ]; then
 615         $BUSYBOX_PATH/tar -xJf control.tar.xz -C /
 616     elif [ -e 'control.tar.bz2' ]; then
 617         $BUSYBOX_PATH/tar -xjf control.tar.bz2 -C /
 618     elif [ -e 'control.tar.lzma' ]; then
 619         $BUSYBOX_PATH/tar -xaf control.tar.lzma -C /
 620     fi
 621
 622     cd $CURDIR
 623     $BUSYBOX_PATH/rm -rf /tmp/vtoy_udeb
 624     $BUSYBOX_PATH/true
 625 }
 626
 627
 628 install_udeb_from_line() {
 629     vtlog "install_udeb_from_line $1"
 630
 631     if ! [ -b "$2" ]; then
 632         vterr "disk #$2# not exist"
 633         return
 634     fi
 635
 636     sector=$(echo $1 | $AWK '{print $(NF-1)}')
 637     length=$(echo $1 | $AWK '{print $NF}')
 638     vtlog "sector=$sector  length=$length"
 639
 640     $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o /tmp/xxx.udeb
 641     if [ -e /tmp/xxx.udeb ]; then
 642         vtlog "extract udeb file from iso success"
 643     else
 644         vterr "extract udeb file from iso fail"
 645         return
 646     fi
 647
 648     install_udeb_pkg /tmp/xxx.udeb
 649     $BUSYBOX_PATH/rm -f /tmp/xxx.udeb
 650 }
 651
 652 extract_file_from_line() {
 653     vtlog "extract_file_from_line $1 disk=#$2#"
 654     if ! [ -b "$2" ]; then
 655         vterr "disk #$2# not exist"
 656         return
 657     fi
 658
 659     sector=$(echo $1 | $AWK '{print $(NF-1)}')
 660     length=$(echo $1 | $AWK '{print $NF}')
 661     vtlog "sector=$sector  length=$length"
 662
 663     $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o $3
 664     if [ -e $3 ]; then
 665         vtlog "extract file from iso success"
 666         $BUSYBOX_PATH/true
 667     else
 668         vterr "extract file from iso fail"
 669         $BUSYBOX_PATH/false
 670     fi
 671 }
 672
 673 extract_rpm_from_line() {
 674     vtlog "extract_rpm_from_line $1 disk=#$2#"
 675
 676     if ! [ -b "$2" ]; then
 677         vterr "disk #$2# not exist"
 678         return
 679     fi
 680
 681     sector=$(echo $1 | $AWK '{print $(NF-1)}')
 682     length=$(echo $1 | $AWK '{print $NF}')
 683     vtlog "sector=$sector  length=$length"
 684
 685     $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o /tmp/xxx.rpm
 686     if [ -e /tmp/xxx.rpm ]; then
 687         vtlog "extract rpm file from iso success"
 688     else
 689         vterr "extract rpm file from iso fail"
 690         return
 691     fi
 692
 693     CURPWD=$($BUSYBOX_PATH/pwd)
 694
 695     $BUSYBOX_PATH/mkdir -p $VTOY_PATH/rpm
 696     cd $VTOY_PATH/rpm
 697     vtlog "extract rpm..."
 698     $BUSYBOX_PATH/rpm2cpio /tmp/xxx.rpm | $BUSYBOX_PATH/cpio -idm 2>>$VTLOG
 699     cd $CURPWD
 700
 701     $BUSYBOX_PATH/rm -f /tmp/xxx.rpm
 702 }
 703
 704 install_rpm_from_line() {
 705     vtlog "install_rpm_from_line $1 disk=#$2#"
 706
 707     if ! [ -b "$2" ]; then
 708         vterr "disk #$2# not exist"
 709         return
 710     fi
 711
 712     sector=$(echo $1 | $AWK '{print $(NF-1)}')
 713     length=$(echo $1 | $AWK '{print $NF}')
 714     vtlog "sector=$sector  length=$length"
 715
 716     $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o /tmp/xxx.rpm
 717     if [ -e /tmp/xxx.rpm ]; then
 718         vtlog "extract rpm file from iso success"
 719     else
 720         vterr "extract rpm file from iso fail"
 721         return
 722     fi
 723
 724     CURPWD=$($BUSYBOX_PATH/pwd)
 725
 726     cd /
 727     vtlog "install rpm..."
 728     $BUSYBOX_PATH/rpm2cpio /tmp/xxx.rpm | $BUSYBOX_PATH/cpio -idm 2>>$VTLOG
 729     cd $CURPWD
 730
 731     $BUSYBOX_PATH/rm -f /tmp/xxx.rpm
 732 }
 733
 734 dump_whole_iso_file() {
 735    $VTOY_PATH/tool/vtoydm -p -f $VTOY_PATH/ventoy_image_map -d $usb_disk | while read vtline; do
 736         vtlog "dmtable line: $vtline"
 737         vtcount=$(echo $vtline | $AWK '{print $2}')
 738         vtoffset=$(echo $vtline | $AWK '{print $NF}')
 739         $BUSYBOX_PATH/dd if=$usb_disk of="$1" bs=512 count=$vtcount skip=$vtoffset oflag=append conv=notrunc
 740     done
 741 }
 742
 743 ventoy_copy_device_mapper() {
 744     if [ -L $VTOY_DM_PATH ]; then
 745         vtlog "replace block device link $1..."
 746         $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_${1#/dev/}
 747         VT_MAPPER_LINK=$($BUSYBOX_PATH/readlink $VTOY_DM_PATH)
 748         $BUSYBOX_PATH/cp -a "/dev/mapper/$VT_MAPPER_LINK" "$1"
 749     elif [ -b $VTOY_DM_PATH ]; then
 750         vtlog "replace block device $1..."
 751         $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_${1#/dev/}
 752         $BUSYBOX_PATH/cp -a "$VTOY_DM_PATH" "$1"
 753     else
 754
 755         vtlog "$VTOY_DM_PATH not exist, now check /dev/dm-X ..."
 756         VT_DM_BIN=$(ventoy_find_bin_path dmsetup)
 757         if [ -z "$VT_DM_BIN" ]; then
 758             vtlog "no dmsetup avaliable, lastly try inbox dmsetup"
 759             VT_DM_BIN=$VTOY_PATH/tool/dmsetup
 760         fi
 761
 762         DM_VT_ID=$($VT_DM_BIN ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
 763         vtlog "DM_VT_ID=$DM_VT_ID ..."
 764         $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_${1#/dev/}
 765         $BUSYBOX_PATH/mknod -m 0666 "$1" b $DM_VT_ID
 766     fi
 767 }
 768
 769 # create link for device-mapper
 770 ventoy_create_persistent_link() {
 771     blkdev_num=$($VTOY_PATH/tool/dmsetup ls | grep vtoy_persistent | sed 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1:\2/')
 772     vtDM=$(ventoy_find_dm_id ${blkdev_num})
 773
 774     if ! [ -d /dev/disk/by-label ]; then
 775         mkdir -p /dev/disk/by-label
 776     fi
 777
 778     VTLABEL=$($BUSYBOX_PATH/blkid /dev/$vtDM | $SED 's/.*LABEL="\([^"]*\)".*/\1/')
 779     if [ -z "$VTLABEL" ]; then
 780         VTLABEL=casper-rw
 781     fi
 782
 783     vtlog "Persistent Label: ##${VTLABEL}##"
 784
 785     if ! [ -e /dev/disk/by-label/$VTLABEL ]; then
 786         vtOldDir=$PWD
 787         cd /dev/disk/by-label
 788         ln -s ../../$vtDM $VTLABEL
 789         cd $vtOldDir
 790     fi
 791 }
 792
 793 ventoy_partname_to_diskname() {
 794     if echo $1 | $EGREP -q "nvme.*p[0-9]$|mmc.*p[0-9]$|nbd.*p[0-9]$"; then
 795         echo -n "${1:0:-2}"
 796     else
 797         echo -n "${1:0:-1}"
 798     fi
 799 }
 800
 801 ventoy_diskname_to_partname() {
 802     if echo $1 | $EGREP -q "nvme.*p[0-9]$|mmc.*p[0-9]$|nbd.*p[0-9]$"; then
 803         echo -n "${1}p$2"
 804     else
 805         echo -n "${1}$2"
 806     fi
 807 }
 808
 809 ventoy_udev_disk_common_hook() {
 810     if echo $1 | $EGREP -q "nvme.*p[0-9]$|mmc.*p[0-9]$|nbd.*p[0-9]$"; then
 811         VTDISK="${1:0:-2}"
 812     else
 813         VTDISK="${1:0:-1}"
 814     fi
 815
 816     if [ -e /vtoy/vtoy ]; then
 817         VTRWMOD=""
 818     else
 819         VTRWMOD="--readonly"
 820     fi
 821
 822     # create device mapper for iso image file
 823     if create_ventoy_device_mapper "/dev/$VTDISK" $VTRWMOD; then
 824         vtlog "==== create ventoy device mapper success ===="
 825     else
 826         vtlog "==== create ventoy device mapper failed ===="
 827
 828         $SLEEP 3
 829
 830         if $GREP -q "/dev/$VTDISK" /proc/mounts; then
 831             $GREP "/dev/$VTDISK" /proc/mounts | while read vtLine; do
 832                 vtPart=$(echo $vtLine | $AWK '{print $1}')
 833                 vtMnt=$(echo $vtLine | $AWK '{print $2}')
 834                 vtlog "$vtPart is mounted on $vtMnt  now umount it ..."
 835                 $BUSYBOX_PATH/umount $vtMnt
 836             done
 837         fi
 838
 839         if create_ventoy_device_mapper "/dev/$VTDISK" $VTRWMOD; then
 840             vtlog "==== create ventoy device mapper success after retry ===="
 841         else
 842             vtlog "==== create ventoy device mapper failed after retry ===="
 843             return
 844         fi
 845     fi
 846
 847     if [ "$2" = "noreplace" ]; then
 848         vtlog "no need to replace block device"
 849     else
 850         ventoy_copy_device_mapper "/dev/$1"
 851     fi
 852
 853     if [ -f $VTOY_PATH/ventoy_persistent_map ]; then
 854         create_persistent_device_mapper "/dev/$VTDISK"
 855         ventoy_create_persistent_link
 856     fi
 857
 858     if $GREP -q 'dm_patch' /proc/modules; then
 859         vtlog "remove dm_patch module."
 860         $BUSYBOX_PATH/rmmod dm_patch
 861     fi
 862 }
 863
 864 ventoy_create_dev_ventoy_part() {
 865     blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
 866     $BUSYBOX_PATH/mknod -m 0666 /dev/ventoy b $blkdev_num
 867
 868     if [ -e /vtoy_dm_table ]; then
 869         vtPartid=1
 870
 871         vtLevel1=$($CAT /proc/sys/kernel/printk | $AWK '{print $1}')
 872         vtLevel2=$($CAT /proc/sys/kernel/printk | $AWK '{print $2}')
 873         vtLevel3=$($CAT /proc/sys/kernel/printk | $AWK '{print $3}')
 874         vtLevel4=$($CAT /proc/sys/kernel/printk | $AWK '{print $4}')
 875         if [ -f $VTOY_PATH/dm_patch_done ]; then
 876             #suppress printk message
 877             echo 0 $vtLevel2 0 $vtLevel4 > /proc/sys/kernel/printk
 878         fi
 879
 880         $CAT /vtoy_dm_table | while read vtline; do
 881             echo $vtline > /ventoy/dm_table_part${vtPartid}
 882             $VTOY_PATH/tool/dmsetup create ventoy${vtPartid} /ventoy/dm_table_part${vtPartid}
 883
 884             blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy${vtPartid} | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
 885             $BUSYBOX_PATH/mknod -m 0666 /dev/ventoy${vtPartid} b $blkdev_num
 886
 887             vtPartid=$(expr $vtPartid + 1)
 888         done
 889
 890         if [ -f $VTOY_PATH/dm_patch_done ]; then
 891             #recover printk level
 892             echo $vtLevel1 $vtLevel2 $vtLevel3 $vtLevel4 > /proc/sys/kernel/printk
 893         fi
 894     fi
 895 }
 896
 897
 898 ventoy_create_chromeos_ventoy_part() {
 899     blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
 900     $BUSYBOX_PATH/mknod -m 0666 /dev/ventoy b $blkdev_num
 901
 902     if [ -e /vtoy_dm_table ]; then
 903         vtPartid=1
 904
 905         $CAT /vtoy_dm_table | while read vtline; do
 906             echo $vtline > /ventoy/dm_table_part${vtPartid}
 907
 908             if [ $vtPartid -eq $1 ]; then
 909                 $VTOY_PATH/tool/dmsetup create ventoy${vtPartid} /ventoy/dm_table_part${vtPartid} --readonly
 910             else
 911                 $VTOY_PATH/tool/dmsetup create ventoy${vtPartid} /ventoy/dm_table_part${vtPartid}
 912             fi
 913
 914             blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy${vtPartid} | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
 915             $BUSYBOX_PATH/mknod -m 0666 /dev/ventoy${vtPartid} b $blkdev_num
 916
 917             vtPartid=$(expr $vtPartid + 1)
 918         done
 919     fi
 920 }
 921
 922 is_inotify_ventoy_part() {
 923     if echo $1 | $GREP -q "2$"; then
 924         if ! [ -e /sys/block/$1 ]; then
 925             if [ -e /sys/class/block/$1 ]; then
 926                 if echo $1 | $EGREP -q "nvme|mmc|nbd"; then
 927                     vtShortName=${1:0:-2}
 928                 else
 929                     vtShortName=${1:0:-1}
 930                 fi
 931
 932                 if [ -e /dev/$vtShortName ]; then
 933                     if [ "$VTOY_VLNK_BOOT" = "01" ]; then
 934                         vtOrgDiskName=$($VTOY_PATH/tool/vtoydump -t $VTOY_PATH/ventoy_os_param)
 935                         [ "$vtOrgDiskName" = "/dev/$vtShortName" ]
 936                     else
 937                         $VTOY_PATH/tool/vtoydump -f $VTOY_PATH/ventoy_os_param -c $vtShortName
 938                     fi
 939                     return
 940                 fi
 941             fi
 942         fi
 943     fi
 944
 945     [ "1" = "0" ]
 946 }
 947
 948 ventoy_find_dm_id() {
 949     for vt in $($BUSYBOX_PATH/ls /sys/block/); do
 950         if [ "${vt:0:3}" = "dm-" ]; then
 951             vtMajorMinor=$($CAT /sys/block/$vt/dev)
 952             if [ "$vtMajorMinor" = "$1" ]; then
 953                 echo ${vt}
 954                 return
 955             fi
 956         fi
 957     done
 958     echo 'xx'
 959 }
 960
 961 ventoy_swap_device() {
 962     mv $1 $VTOY_PATH/swap_tmp_dev
 963     mv $2 $1
 964     mv $VTOY_PATH/swap_tmp_dev $2
 965 }
 966
 967 ventoy_extract_vtloopex() {
 968     vtCurPwd=$PWD
 969     $BUSYBOX_PATH/mkdir -p $VTOY_PATH/partmnt $VTOY_PATH/vtloopex
 970     $BUSYBOX_PATH/mount -o ro -t vfat $1  $VTOY_PATH/partmnt
 971     cd $VTOY_PATH/vtloopex
 972     $CAT $VTOY_PATH/partmnt/ventoy/vtloopex.cpio | $BUSYBOX_PATH/cpio -idm >> $VTLOG 2>&1
 973     $BUSYBOX_PATH/umount $VTOY_PATH/partmnt
 974     $BUSYBOX_PATH/rm -rf $VTOY_PATH/partmnt
 975
 976     if [ -n "$2" ]; then
 977         cd $VTOY_PATH/vtloopex/$2/
 978         $BUSYBOX_PATH/tar -xJf vtloopex.tar.xz
 979     fi
 980
 981     cd $vtCurPwd
 982 }
 983
 984 ventoy_check_install_module_xz() {
 985     if [ -f "${1}.xz" ]; then
 986         $BUSYBOX_PATH/xz -d  "${1}.xz"
 987         $BUSYBOX_PATH/insmod "$1"
 988     fi
 989 }
 990
 991 ventoy_check_umount() {
 992     for vtLoop in 0 1 2 3 4 5 6 7 8 9; do
 993         $BUSYBOX_PATH/umount "$1" > /dev/null 2>&1
 994         if $BUSYBOX_PATH/mountpoint -q "$1"; then
 995             $SLEEP 1
 996         else
 997             break
 998         fi
 999     done
1000 }
1001
1002 ventoy_wait_dir() {
1003     vtdir=$1
1004     vtsec=0
1005
1006     while [ $vtsec -lt $2 ]; do
1007         if [ -d "$vtdir" ]; then
1008             break
1009         else
1010             $SLEEP 1
1011             vtsec=$(expr $vtsec + 1)
1012         fi
1013     done
1014 }