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