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