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
 313     if [ -d /lib/modules/$vtKv/kernel/fs ]; then
 314         vtModPath=$($FIND /lib/modules/$vtKv/kernel/fs/ -name "*.ko*" | $HEAD -n1)
 315     else
 316         vtModPath=$($FIND /lib/modules/$vtKv/kernel/ -name "xfs.ko*" | $HEAD -n1)
 317     fi
 318
 319     if [ -z "$vtModPath" ]; then
 320         vtModPath=$($FIND /lib/modules/$vtKv/kernel/ -name "*.ko*" | $HEAD -n1)
 321     fi
 322
 323     vtModName=$($BUSYBOX_PATH/basename $vtModPath)
 324
 325     vtlog "template module is $vtModPath $vtModName"
 326
 327     if echo $vtModPath | $GREP -q "[.]ko$"; then
 328         $BUSYBOX_PATH/cp -a $vtModPath  $VTOY_PATH/$vtModName
 329     elif echo $vtModPath | $GREP -q "[.]ko[.]xz$"; then
 330         $BUSYBOX_PATH/xzcat $vtModPath > $VTOY_PATH/$vtModName
 331     elif echo $vtModPath | $GREP -q "[.]ko[.]gz$"; then
 332         $BUSYBOX_PATH/zcat $vtModPath > $VTOY_PATH/$vtModName
 333     else
 334         vtlog "unsupport module type"
 335         return
 336     fi
 337
 338     #step1: modify vermagic/mod crc/relocation
 339     $VTOY_PATH/tool/vtoykmod -u $VTOY_PATH/tool/$vtKoName $VTOY_PATH/$vtModName $vtDebug
 340
 341     #step2: fill parameters
 342     vtPgsize=$($VTOY_PATH/tool/vtoyksym -p)
 343     $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
 344
 345     $BUSYBOX_PATH/insmod $VTOY_PATH/tool/$vtKoName
 346
 347     if $GREP -q 'dm_patch' /proc/modules; then
 348         echo "done" > $VTOY_PATH/dm_patch_done
 349     fi
 350
 351 }
 352
 353 create_ventoy_device_mapper() {
 354     vtlog "create_ventoy_device_mapper $*"
 355
 356     VT_DM_BIN=$(ventoy_find_bin_path dmsetup)
 357     if [ -z "$VT_DM_BIN" ]; then
 358         vtlog "no dmsetup avaliable, lastly try inbox dmsetup"
 359         VT_DM_BIN=$VTOY_PATH/tool/dmsetup
 360     fi
 361
 362     vtlog "dmsetup avaliable in system $VT_DM_BIN"
 363
 364     if ventoy_check_dm_module "$1"; then
 365         vtlog "device-mapper module check success"
 366     else
 367         vterr "Error: no dm module avaliable"
 368     fi
 369
 370     $VTOY_PATH/tool/vtoydm -p -f $VTOY_PATH/ventoy_image_map -d $1 > $VTOY_PATH/ventoy_dm_table
 371
 372
 373     vtLevel1=$($CAT /proc/sys/kernel/printk | $AWK '{print $1}')
 374     vtLevel2=$($CAT /proc/sys/kernel/printk | $AWK '{print $2}')
 375     vtLevel3=$($CAT /proc/sys/kernel/printk | $AWK '{print $3}')
 376     vtLevel4=$($CAT /proc/sys/kernel/printk | $AWK '{print $4}')
 377     if ventoy_need_dm_patch; then
 378         ventoy_dm_patch
 379         #suppress printk message
 380         echo 0 $vtLevel2 0 $vtLevel4 > /proc/sys/kernel/printk
 381     fi
 382
 383     if [ -z "$2" ]; then
 384         $VT_DM_BIN create ventoy $VTOY_PATH/ventoy_dm_table >>$VTLOG 2>&1
 385     else
 386         $VT_DM_BIN "$2" create ventoy $VTOY_PATH/ventoy_dm_table >>$VTLOG 2>&1
 387     fi
 388
 389     if ventoy_need_dm_patch; then
 390         #recover printk level
 391         echo $vtLevel1 $vtLevel2 $vtLevel3 $vtLevel4 > /proc/sys/kernel/printk
 392     fi
 393 }
 394
 395 create_persistent_device_mapper() {
 396     vtlog "create_persistent_device_mapper $*"
 397
 398     VT_DM_BIN=$(ventoy_find_bin_path dmsetup)
 399     if [ -z "$VT_DM_BIN" ]; then
 400         vtlog "no dmsetup avaliable, lastly try inbox dmsetup"
 401         VT_DM_BIN=$VTOY_PATH/tool/dmsetup
 402     fi
 403
 404     vtlog "dmsetup avaliable in system $VT_DM_BIN"
 405
 406     if ventoy_check_dm_module "$1"; then
 407         vtlog "device-mapper module check success"
 408     else
 409         vterr "Error: no dm module avaliable"
 410     fi
 411
 412     $VTOY_PATH/tool/vtoydm -p -f $VTOY_PATH/ventoy_persistent_map -d $1 > $VTOY_PATH/persistent_dm_table
 413
 414
 415     vtLevel1=$($CAT /proc/sys/kernel/printk | $AWK '{print $1}')
 416     vtLevel2=$($CAT /proc/sys/kernel/printk | $AWK '{print $2}')
 417     vtLevel3=$($CAT /proc/sys/kernel/printk | $AWK '{print $3}')
 418     vtLevel4=$($CAT /proc/sys/kernel/printk | $AWK '{print $4}')
 419     if [ -f $VTOY_PATH/dm_patch_done ]; then
 420         #suppress printk message
 421         echo 0 $vtLevel2 0 $vtLevel4 > /proc/sys/kernel/printk
 422     fi
 423
 424     $VT_DM_BIN create vtoy_persistent $VTOY_PATH/persistent_dm_table >>$VTLOG 2>&1
 425
 426     if [ -f $VTOY_PATH/dm_patch_done ]; then
 427         #recover printk level
 428         echo $vtLevel1 $vtLevel2 $vtLevel3 $vtLevel4 > /proc/sys/kernel/printk
 429     fi
 430 }
 431
 432
 433
 434 wait_for_ventoy_dm_disk_label() {
 435     DM=$($BUSYBOX_PATH/readlink $VTOY_DM_PATH)
 436     vtlog "wait_for_ventoy_dm_disk_label $DM ..."
 437
 438     for i in 0 1 2 3 4 5 6 7 8 9; do
 439         vtlog "i=$i ####### ls /dev/disk/by-label/"
 440         ls -l /dev/disk/by-label/ >> $VTLOG
 441
 442         if ls -l /dev/disk/by-label/ | $GREP -q "$DM"; then
 443             break
 444         else
 445             $SLEEP 1
 446         fi
 447     done
 448 }
 449
 450 install_udeb_pkg() {
 451     if ! [ -e "$1" ]; then
 452         $BUSYBOX_PATH/false
 453         return
 454     fi
 455
 456     if [ -d /tmp/vtoy_udeb ]; then
 457         $BUSYBOX_PATH/rm -rf /tmp/vtoy_udeb
 458     fi
 459
 460     $BUSYBOX_PATH/mkdir -p /tmp/vtoy_udeb
 461     $BUSYBOX_PATH/cp -a "$1" /tmp/vtoy_udeb/
 462
 463     CURDIR=$($BUSYBOX_PATH/pwd)
 464     cd /tmp/vtoy_udeb
 465
 466     $BUSYBOX_PATH/ar x "$1"
 467
 468     if [ -e 'data.tar.gz' ]; then
 469         $BUSYBOX_PATH/tar -xzf data.tar.gz -C /
 470     elif [ -e 'data.tar.xz' ]; then
 471         $BUSYBOX_PATH/tar -xJf data.tar.xz -C /
 472     elif [ -e 'data.tar.bz2' ]; then
 473         $BUSYBOX_PATH/tar -xjf data.tar.bz2 -C /
 474     elif [ -e 'data.tar.lzma' ]; then
 475         $BUSYBOX_PATH/tar -xaf data.tar.lzma -C /
 476     fi
 477
 478     if [ -e 'control.tar.gz' ]; then
 479         $BUSYBOX_PATH/tar -xzf control.tar.gz -C /
 480     elif [ -e 'control.tar.xz' ]; then
 481         $BUSYBOX_PATH/tar -xJf control.tar.xz -C /
 482     elif [ -e 'control.tar.bz2' ]; then
 483         $BUSYBOX_PATH/tar -xjf control.tar.bz2 -C /
 484     elif [ -e 'control.tar.lzma' ]; then
 485         $BUSYBOX_PATH/tar -xaf control.tar.lzma -C /
 486     fi
 487
 488     cd $CURDIR
 489     $BUSYBOX_PATH/rm -rf /tmp/vtoy_udeb
 490     $BUSYBOX_PATH/true
 491 }
 492
 493
 494 install_udeb_from_line() {
 495     vtlog "install_udeb_from_line $1"
 496
 497     if ! [ -b "$2" ]; then
 498         vterr "disk #$2# not exist"
 499         return
 500     fi
 501
 502     sector=$(echo $1 | $AWK '{print $(NF-1)}')
 503     length=$(echo $1 | $AWK '{print $NF}')
 504     vtlog "sector=$sector  length=$length"
 505
 506     $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o /tmp/xxx.udeb
 507     if [ -e /tmp/xxx.udeb ]; then
 508         vtlog "extract udeb file from iso success"
 509     else
 510         vterr "extract udeb file from iso fail"
 511         return
 512     fi
 513
 514     install_udeb_pkg /tmp/xxx.udeb
 515     $BUSYBOX_PATH/rm -f /tmp/xxx.udeb
 516 }
 517
 518 extract_file_from_line() {
 519     vtlog "extract_file_from_line $1 disk=#$2#"
 520     if ! [ -b "$2" ]; then
 521         vterr "disk #$2# not exist"
 522         return
 523     fi
 524
 525     sector=$(echo $1 | $AWK '{print $(NF-1)}')
 526     length=$(echo $1 | $AWK '{print $NF}')
 527     vtlog "sector=$sector  length=$length"
 528
 529     $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o $3
 530     if [ -e $3 ]; then
 531         vtlog "extract file from iso success"
 532         $BUSYBOX_PATH/true
 533     else
 534         vterr "extract file from iso fail"
 535         $BUSYBOX_PATH/false
 536     fi
 537 }
 538
 539 extract_rpm_from_line() {
 540     vtlog "extract_rpm_from_line $1 disk=#$2#"
 541
 542     if ! [ -b "$2" ]; then
 543         vterr "disk #$2# not exist"
 544         return
 545     fi
 546
 547     sector=$(echo $1 | $AWK '{print $(NF-1)}')
 548     length=$(echo $1 | $AWK '{print $NF}')
 549     vtlog "sector=$sector  length=$length"
 550
 551     $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o /tmp/xxx.rpm
 552     if [ -e /tmp/xxx.rpm ]; then
 553         vtlog "extract rpm file from iso success"
 554     else
 555         vterr "extract rpm file from iso fail"
 556         return
 557     fi
 558
 559     CURPWD=$($BUSYBOX_PATH/pwd)
 560
 561     $BUSYBOX_PATH/mkdir -p $VTOY_PATH/rpm
 562     cd $VTOY_PATH/rpm
 563     vtlog "extract rpm..."
 564     $BUSYBOX_PATH/rpm2cpio /tmp/xxx.rpm | $BUSYBOX_PATH/cpio -idm 2>>$VTLOG
 565     cd $CURPWD
 566
 567     $BUSYBOX_PATH/rm -f /tmp/xxx.rpm
 568 }
 569
 570 install_rpm_from_line() {
 571     vtlog "install_rpm_from_line $1 disk=#$2#"
 572
 573     if ! [ -b "$2" ]; then
 574         vterr "disk #$2# not exist"
 575         return
 576     fi
 577
 578     sector=$(echo $1 | $AWK '{print $(NF-1)}')
 579     length=$(echo $1 | $AWK '{print $NF}')
 580     vtlog "sector=$sector  length=$length"
 581
 582     $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o /tmp/xxx.rpm
 583     if [ -e /tmp/xxx.rpm ]; then
 584         vtlog "extract rpm file from iso success"
 585     else
 586         vterr "extract rpm file from iso fail"
 587         return
 588     fi
 589
 590     CURPWD=$($BUSYBOX_PATH/pwd)
 591
 592     cd /
 593     vtlog "install rpm..."
 594     $BUSYBOX_PATH/rpm2cpio /tmp/xxx.rpm | $BUSYBOX_PATH/cpio -idm 2>>$VTLOG
 595     cd $CURPWD
 596
 597     $BUSYBOX_PATH/rm -f /tmp/xxx.rpm
 598 }
 599
 600 dump_whole_iso_file() {
 601    $VTOY_PATH/tool/vtoydm -p -f $VTOY_PATH/ventoy_image_map -d $usb_disk | while read vtline; do
 602         vtlog "dmtable line: $vtline"
 603         vtcount=$(echo $vtline | $AWK '{print $2}')
 604         vtoffset=$(echo $vtline | $AWK '{print $NF}')
 605         $BUSYBOX_PATH/dd if=$usb_disk of="$1" bs=512 count=$vtcount skip=$vtoffset oflag=append conv=notrunc
 606     done
 607 }
 608
 609 ventoy_copy_device_mapper() {
 610     if [ -L $VTOY_DM_PATH ]; then
 611         vtlog "replace block device link $1..."
 612         $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_${1#/dev/}
 613         VT_MAPPER_LINK=$($BUSYBOX_PATH/readlink $VTOY_DM_PATH)
 614         $BUSYBOX_PATH/cp -a "/dev/mapper/$VT_MAPPER_LINK" "$1"
 615     elif [ -b $VTOY_DM_PATH ]; then
 616         vtlog "replace block device $1..."
 617         $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_${1#/dev/}
 618         $BUSYBOX_PATH/cp -a "$VTOY_DM_PATH" "$1"
 619     else
 620
 621         vtlog "$VTOY_DM_PATH not exist, now check /dev/dm-X ..."
 622         VT_DM_BIN=$(ventoy_find_bin_path dmsetup)
 623         if [ -z "$VT_DM_BIN" ]; then
 624             vtlog "no dmsetup avaliable, lastly try inbox dmsetup"
 625             VT_DM_BIN=$VTOY_PATH/tool/dmsetup
 626         fi
 627
 628         DM_VT_ID=$($VT_DM_BIN ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
 629         vtlog "DM_VT_ID=$DM_VT_ID ..."
 630         $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_${1#/dev/}
 631         $BUSYBOX_PATH/mknod -m 0666 "$1" b $DM_VT_ID
 632     fi
 633 }
 634
 635 # create link for device-mapper
 636 ventoy_create_persistent_link() {
 637     blkdev_num=$($VTOY_PATH/tool/dmsetup ls | grep vtoy_persistent | sed 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1:\2/')
 638     vtDM=$(ventoy_find_dm_id ${blkdev_num})
 639
 640     if ! [ -d /dev/disk/by-label ]; then
 641         mkdir -p /dev/disk/by-label
 642     fi
 643
 644     VTLABEL=$($BUSYBOX_PATH/blkid /dev/$vtDM | $SED 's/.*LABEL="\([^"]*\)".*/\1/')
 645     if [ -z "$VTLABEL" ]; then
 646         VTLABEL=casper-rw
 647     fi
 648
 649     vtlog "Persistent Label: ##${VTLABEL}##"
 650
 651     if ! [ -e /dev/disk/by-label/$VTLABEL ]; then
 652         vtOldDir=$PWD
 653         cd /dev/disk/by-label
 654         ln -s ../../$vtDM $VTLABEL
 655         cd $vtOldDir
 656     fi
 657 }
 658
 659 ventoy_partname_to_diskname() {
 660     if echo $1 | $EGREP -q "nvme.*p[0-9]$|mmc.*p[0-9]$|nbd.*p[0-9]$"; then
 661         echo -n "${1:0:-2}"
 662     else
 663         echo -n "${1:0:-1}"
 664     fi
 665 }
 666
 667 ventoy_diskname_to_partname() {
 668     if echo $1 | $EGREP -q "nvme.*p[0-9]$|mmc.*p[0-9]$|nbd.*p[0-9]$"; then
 669         echo -n "${1}p$2"
 670     else
 671         echo -n "${1}$2"
 672     fi
 673 }
 674
 675 ventoy_udev_disk_common_hook() {
 676     if echo $1 | $EGREP -q "nvme.*p[0-9]$|mmc.*p[0-9]$|nbd.*p[0-9]$"; then
 677         VTDISK="${1:0:-2}"
 678     else
 679         VTDISK="${1:0:-1}"
 680     fi
 681
 682     if [ -e /vtoy/vtoy ]; then
 683         VTRWMOD=""
 684     else
 685         VTRWMOD="--readonly"
 686     fi
 687
 688     # create device mapper for iso image file
 689     if create_ventoy_device_mapper "/dev/$VTDISK" $VTRWMOD; then
 690         vtlog "==== create ventoy device mapper success ===="
 691     else
 692         vtlog "==== create ventoy device mapper failed ===="
 693
 694         $SLEEP 3
 695
 696         if $GREP -q "/dev/$VTDISK" /proc/mounts; then
 697             $GREP "/dev/$VTDISK" /proc/mounts | while read vtLine; do
 698                 vtPart=$(echo $vtLine | $AWK '{print $1}')
 699                 vtMnt=$(echo $vtLine | $AWK '{print $2}')
 700                 vtlog "$vtPart is mounted on $vtMnt  now umount it ..."
 701                 $BUSYBOX_PATH/umount $vtMnt
 702             done
 703         fi
 704
 705         if create_ventoy_device_mapper "/dev/$VTDISK" $VTRWMOD; then
 706             vtlog "==== create ventoy device mapper success after retry ===="
 707         else
 708             vtlog "==== create ventoy device mapper failed after retry ===="
 709             return
 710         fi
 711     fi
 712
 713     if [ "$2" = "noreplace" ]; then
 714         vtlog "no need to replace block device"
 715     else
 716         ventoy_copy_device_mapper "/dev/$1"
 717     fi
 718
 719     if [ -f $VTOY_PATH/ventoy_persistent_map ]; then
 720         create_persistent_device_mapper "/dev/$VTDISK"
 721         ventoy_create_persistent_link
 722     fi
 723
 724     if $GREP -q 'dm_patch' /proc/modules; then
 725         $BUSYBOX_PATH/rmmod dm_patch
 726     fi
 727 }
 728
 729 ventoy_create_dev_ventoy_part() {
 730     blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
 731     $BUSYBOX_PATH/mknod -m 0666 /dev/ventoy b $blkdev_num
 732
 733     if [ -e /vtoy_dm_table ]; then
 734         vtPartid=1
 735
 736         vtLevel1=$($CAT /proc/sys/kernel/printk | $AWK '{print $1}')
 737         vtLevel2=$($CAT /proc/sys/kernel/printk | $AWK '{print $2}')
 738         vtLevel3=$($CAT /proc/sys/kernel/printk | $AWK '{print $3}')
 739         vtLevel4=$($CAT /proc/sys/kernel/printk | $AWK '{print $4}')
 740         if [ -f $VTOY_PATH/dm_patch_done ]; then
 741             #suppress printk message
 742             echo 0 $vtLevel2 0 $vtLevel4 > /proc/sys/kernel/printk
 743         fi
 744
 745         $CAT /vtoy_dm_table | while read vtline; do
 746             echo $vtline > /ventoy/dm_table_part${vtPartid}
 747             $VTOY_PATH/tool/dmsetup create ventoy${vtPartid} /ventoy/dm_table_part${vtPartid}
 748
 749             blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy${vtPartid} | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
 750             $BUSYBOX_PATH/mknod -m 0666 /dev/ventoy${vtPartid} b $blkdev_num
 751
 752             vtPartid=$(expr $vtPartid + 1)
 753         done
 754
 755         if [ -f $VTOY_PATH/dm_patch_done ]; then
 756             #recover printk level
 757             echo $vtLevel1 $vtLevel2 $vtLevel3 $vtLevel4 > /proc/sys/kernel/printk
 758         fi
 759     fi
 760 }
 761
 762
 763 ventoy_create_chromeos_ventoy_part() {
 764     blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
 765     $BUSYBOX_PATH/mknod -m 0666 /dev/ventoy b $blkdev_num
 766
 767     if [ -e /vtoy_dm_table ]; then
 768         vtPartid=1
 769
 770         $CAT /vtoy_dm_table | while read vtline; do
 771             echo $vtline > /ventoy/dm_table_part${vtPartid}
 772
 773             if [ $vtPartid -eq $1 ]; then
 774                 $VTOY_PATH/tool/dmsetup create ventoy${vtPartid} /ventoy/dm_table_part${vtPartid} --readonly
 775             else
 776                 $VTOY_PATH/tool/dmsetup create ventoy${vtPartid} /ventoy/dm_table_part${vtPartid}
 777             fi
 778
 779             blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy${vtPartid} | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
 780             $BUSYBOX_PATH/mknod -m 0666 /dev/ventoy${vtPartid} b $blkdev_num
 781
 782             vtPartid=$(expr $vtPartid + 1)
 783         done
 784     fi
 785 }
 786
 787 is_inotify_ventoy_part() {
 788     if echo $1 | $GREP -q "2$"; then
 789         if ! [ -e /sys/block/$1 ]; then
 790             if [ -e /sys/class/block/$1 ]; then
 791                 if echo $1 | $EGREP -q "nvme|mmc|nbd"; then
 792                     vtShortName=${1:0:-2}
 793                 else
 794                     vtShortName=${1:0:-1}
 795                 fi
 796
 797                 if [ -e /dev/$vtShortName ]; then
 798                     $VTOY_PATH/tool/vtoydump -f $VTOY_PATH/ventoy_os_param -c $vtShortName
 799                     return
 800                 fi
 801             fi
 802         fi
 803     fi
 804
 805     [ "1" = "0" ]
 806 }
 807
 808 ventoy_find_dm_id() {
 809     for vt in $($BUSYBOX_PATH/ls /sys/block/); do
 810         if [ "${vt:0:3}" = "dm-" ]; then
 811             vtMajorMinor=$($CAT /sys/block/$vt/dev)
 812             if [ "$vtMajorMinor" = "$1" ]; then
 813                 echo ${vt}
 814                 return
 815             fi
 816         fi
 817     done
 818     echo 'xx'
 819 }
 820
 821 ventoy_swap_device() {
 822     mv $1 $VTOY_PATH/swap_tmp_dev
 823     mv $2 $1
 824     mv $VTOY_PATH/swap_tmp_dev $2
 825 }
 826
 827 ventoy_extract_vtloopex() {
 828     vtCurPwd=$PWD
 829     $BUSYBOX_PATH/mkdir -p $VTOY_PATH/partmnt $VTOY_PATH/vtloopex
 830     $BUSYBOX_PATH/mount -o ro -t vfat $1  $VTOY_PATH/partmnt
 831     cd $VTOY_PATH/vtloopex
 832     $CAT $VTOY_PATH/partmnt/ventoy/vtloopex.cpio | $BUSYBOX_PATH/cpio -idm >> $VTLOG 2>&1
 833     $BUSYBOX_PATH/umount $VTOY_PATH/partmnt
 834     $BUSYBOX_PATH/rm -rf $VTOY_PATH/partmnt
 835
 836     if [ -n "$2" ]; then
 837         cd $VTOY_PATH/vtloopex/$2/
 838         $BUSYBOX_PATH/tar -xJf vtloopex.tar.xz
 839     fi
 840
 841     cd $vtCurPwd
 842 }
 843
 844 ventoy_check_install_module_xz() {
 845     if [ -f "${1}.xz" ]; then
 846         $BUSYBOX_PATH/xz -d  "${1}.xz"
 847         $BUSYBOX_PATH/insmod "$1"
 848     fi
 849 }