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