IMG/cpio/ventoy/hook/ventoy-hook-lib.sh

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