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
  34 if [ "$VTOY_DEBUG_LEVEL" = "01" ]; then
  35     if [ -e /dev/console ]; then
  36         VTLOG=/dev/console
  37     fi
  38 fi
  39
  40 vtlog() {
  41     if [ "$VTLOG" = "$VTOY_PATH/log" ]; then
  42         echo "$*" >>$VTLOG
  43     else
  44         echo -e "\033[32m $* \033[0m" > $VTLOG
  45         $SLEEP 2
  46     fi
  47 }
  48
  49 vterr() {
  50     if [ "$VTLOG" = "$VTOY_PATH/log" ]; then
  51         echo "$*" >>$VTLOG
  52     else
  53         echo -e "\n\033[31m $* \033[0m" > $VTLOG
  54         $SLEEP 30
  55     fi
  56 }
  57
  58
  59 is_ventoy_hook_finished() {
  60         [ -e $VTOY_PATH/hook_finish ]
  61 }
  62
  63 set_ventoy_hook_finish() {
  64         echo 'Y' > $VTOY_PATH/hook_finish
  65 }
  66
  67 get_ventoy_disk_name() {
  68     line=$($VTOY_PATH/tool/vtoydump -f /ventoy/ventoy_os_param)
  69     if [ $? -eq 0 ]; then
  70         echo ${line%%#*}
  71     else
  72         echo "unknown"
  73     fi
  74 }
  75
  76 get_ventoy_iso_name() {
  77         line=$($VTOY_PATH/tool/vtoydump -f /ventoy/ventoy_os_param)
  78     if [ $? -eq 0 ]; then
  79         echo ${line##*#}
  80     else
  81         echo "unknown"
  82     fi
  83 }
  84
  85 wait_for_usb_disk_ready() {
  86         while [ -n "Y" ]; do
  87                 usb_disk=$(get_ventoy_disk_name)
  88         vtlog "wait_for_usb_disk_ready $usb_disk ..."
  89
  90                 if [ -e "${usb_disk}2" ]; then
  91             vtlog "wait_for_usb_disk_ready $usb_disk finish"
  92                         break
  93                 else
  94                         $SLEEP 0.3
  95                 fi
  96         done
  97 }
  98
  99 is_ventoy_disk() {
 100     if $VTOY_PATH/tool/vtoydump -f $VTOY_PATH/ventoy_os_param -c "$1"; then
 101         $BUSYBOX_PATH/true
 102     else
 103         $BUSYBOX_PATH/false
 104     fi
 105 }
 106
 107 not_ventoy_disk() {
 108     if $VTOY_PATH/tool/vtoydump -f $VTOY_PATH/ventoy_os_param -c "$1"; then
 109         $BUSYBOX_PATH/false
 110     else
 111         $BUSYBOX_PATH/true
 112     fi
 113 }
 114
 115 ventoy_get_vblade_bin() {
 116     if $VTOY_PATH/tool/vblade_64 -t >>$VTLOG 2>&1; then
 117         echo $VTOY_PATH/tool/vblade_64
 118     else
 119         echo $VTOY_PATH/tool/vblade_32
 120     fi
 121 }
 122
 123 ventoy_find_bin_path() {
 124     if $BUSYBOX_PATH/which "$1" > /dev/null; then
 125         $BUSYBOX_PATH/which "$1"; return
 126     fi
 127
 128     for vt_path in '/bin' '/sbin' '/usr/bin' '/usr/sbin' '/usr/local/bin' '/usr/local/sbin' '/root/bin'; do
 129         if [ -e "$vt_path/$1" ]; then
 130             echo "$vt_path/$1"; return
 131         fi
 132     done
 133
 134     echo ""
 135 }
 136
 137
 138 ventoy_find_bin_run() {
 139     vtsudo=0
 140     if [ "$1" = "sudo" ]; then
 141         shift
 142         vtsudo=1
 143     fi
 144
 145     vtbinpath=$(ventoy_find_bin_path "$1")
 146     if [ -n "$vtbinpath" ]; then
 147         shift
 148
 149         if [ $vtsudo -eq 0 ]; then
 150             vtlog "$vtbinpath $*"
 151             $vtbinpath $*
 152         else
 153             vtlog "sudo $vtbinpath $*"
 154             sudo $vtbinpath $*
 155         fi
 156     fi
 157 }
 158
 159 ventoy_get_module_postfix() {
 160     vtKerVer=$($BUSYBOX_PATH/uname -r)
 161     vtLine=$($FIND /lib/modules/$vtKerVer/ -name *.ko* | $HEAD -n1)
 162     vtComp=${vtLine##*/*.ko}
 163     echo "ko$vtComp"
 164 }
 165
 166 ventoy_check_dm_module() {
 167     if $GREP -q 'device-mapper' /proc/devices; then
 168         $BUSYBOX_PATH/true; return
 169     fi
 170
 171     vtlog "device-mapper NOT found in /proc/devices, try to load kernel module"
 172     $BUSYBOX_PATH/modprobe dm_mod >>$VTLOG 2>&1
 173     $BUSYBOX_PATH/modprobe dm-mod >>$VTLOG 2>&1
 174
 175     if ! $GREP -q 'device-mapper' /proc/devices; then
 176         vtlog "modprobe failed, now try to insmod ko..."
 177
 178         $FIND /lib/modules/ -name "dm-mod.ko*" | while read vtline; do
 179             vtlog "insmode $vtline "
 180             $BUSYBOX_PATH/insmod $vtline >>$VTLOG 2>&1
 181         done
 182     fi
 183
 184     if $GREP -q 'device-mapper' /proc/devices; then
 185         vtlog "device-mapper found in /proc/devices after retry"
 186         $BUSYBOX_PATH/true; return
 187     else
 188         vtlog "device-mapper still NOT found in /proc/devices after retry"
 189         $BUSYBOX_PATH/false; return
 190     fi
 191 }
 192
 193 create_ventoy_device_mapper() {
 194     vtlog "create_ventoy_device_mapper $*"
 195
 196     VT_DM_BIN=$(ventoy_find_bin_path dmsetup)
 197     if [ -z "$VT_DM_BIN" ]; then
 198         vtlog "no dmsetup avaliable, lastly try inbox dmsetup"
 199         VT_DM_BIN=$VTOY_PATH/tool/dmsetup
 200     fi
 201
 202     vtlog "dmsetup avaliable in system $VT_DM_BIN"
 203
 204     if ventoy_check_dm_module "$1"; then
 205         vtlog "device-mapper module check success"
 206     else
 207         vterr "Error: no dm module avaliable"
 208     fi
 209
 210     $VTOY_PATH/tool/vtoydm -p -f $VTOY_PATH/ventoy_image_map -d $1 > $VTOY_PATH/ventoy_dm_table
 211     if [ -z "$2" ]; then
 212         $VT_DM_BIN create ventoy $VTOY_PATH/ventoy_dm_table >>$VTLOG 2>&1
 213     else
 214         $VT_DM_BIN "$2" create ventoy $VTOY_PATH/ventoy_dm_table >>$VTLOG 2>&1
 215     fi
 216 }
 217
 218 create_persistent_device_mapper() {
 219     vtlog "create_persistent_device_mapper $*"
 220
 221     VT_DM_BIN=$(ventoy_find_bin_path dmsetup)
 222     if [ -z "$VT_DM_BIN" ]; then
 223         vtlog "no dmsetup avaliable, lastly try inbox dmsetup"
 224         VT_DM_BIN=$VTOY_PATH/tool/dmsetup
 225     fi
 226
 227     vtlog "dmsetup avaliable in system $VT_DM_BIN"
 228
 229     if ventoy_check_dm_module "$1"; then
 230         vtlog "device-mapper module check success"
 231     else
 232         vterr "Error: no dm module avaliable"
 233     fi
 234
 235     $VTOY_PATH/tool/vtoydm -p -f $VTOY_PATH/ventoy_persistent_map -d $1 > $VTOY_PATH/persistent_dm_table
 236     $VT_DM_BIN create vtoy_persistent $VTOY_PATH/persistent_dm_table >>$VTLOG 2>&1
 237 }
 238
 239
 240
 241 wait_for_ventoy_dm_disk_label() {
 242     DM=$($BUSYBOX_PATH/readlink $VTOY_DM_PATH)
 243     vtlog "wait_for_ventoy_dm_disk_label $DM ..."
 244
 245     for i in 0 1 2 3 4 5 6 7 8 9; do
 246         vtlog "i=$i ####### ls /dev/disk/by-label/"
 247         ls -l /dev/disk/by-label/ >> $VTLOG
 248
 249         if ls -l /dev/disk/by-label/ | $GREP -q "$DM"; then
 250             break
 251         else
 252             $SLEEP 1
 253         fi
 254     done
 255 }
 256
 257 install_udeb_pkg() {
 258     if ! [ -e "$1" ]; then
 259         $BUSYBOX_PATH/false
 260         return
 261     fi
 262
 263     if [ -d /tmp/vtoy_udeb ]; then
 264         $BUSYBOX_PATH/rm -rf /tmp/vtoy_udeb
 265     fi
 266
 267     $BUSYBOX_PATH/mkdir -p /tmp/vtoy_udeb
 268     $BUSYBOX_PATH/cp -a "$1" /tmp/vtoy_udeb/
 269
 270     CURDIR=$($BUSYBOX_PATH/pwd)
 271     cd /tmp/vtoy_udeb
 272
 273     $BUSYBOX_PATH/ar x "$1"
 274
 275     if [ -e 'data.tar.gz' ]; then
 276         $BUSYBOX_PATH/tar -xzf data.tar.gz -C /
 277     elif [ -e 'data.tar.xz' ]; then
 278         $BUSYBOX_PATH/tar -xJf data.tar.xz -C /
 279     elif [ -e 'data.tar.bz2' ]; then
 280         $BUSYBOX_PATH/tar -xjf data.tar.bz2 -C /
 281     elif [ -e 'data.tar.lzma' ]; then
 282         $BUSYBOX_PATH/tar -xaf data.tar.lzma -C /
 283     fi
 284
 285     if [ -e 'control.tar.gz' ]; then
 286         $BUSYBOX_PATH/tar -xzf control.tar.gz -C /
 287     elif [ -e 'control.tar.xz' ]; then
 288         $BUSYBOX_PATH/tar -xJf control.tar.xz -C /
 289     elif [ -e 'control.tar.bz2' ]; then
 290         $BUSYBOX_PATH/tar -xjf control.tar.bz2 -C /
 291     elif [ -e 'control.tar.lzma' ]; then
 292         $BUSYBOX_PATH/tar -xaf control.tar.lzma -C /
 293     fi
 294
 295     cd $CURDIR
 296     $BUSYBOX_PATH/rm -rf /tmp/vtoy_udeb
 297     $BUSYBOX_PATH/true
 298 }
 299
 300
 301 install_udeb_from_line() {
 302     vtlog "install_udeb_from_line $1"
 303
 304     if ! [ -b "$2" ]; then
 305         vterr "disk #$2# not exist"
 306         return
 307     fi
 308
 309     sector=$(echo $1 | $AWK '{print $(NF-1)}')
 310     length=$(echo $1 | $AWK '{print $NF}')
 311     vtlog "sector=$sector  length=$length"
 312
 313     $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o /tmp/xxx.udeb
 314     if [ -e /tmp/xxx.udeb ]; then
 315         vtlog "extract udeb file from iso success"
 316     else
 317         vterr "extract udeb file from iso fail"
 318         return
 319     fi
 320
 321     install_udeb_pkg /tmp/xxx.udeb
 322     $BUSYBOX_PATH/rm -f /tmp/xxx.udeb
 323 }
 324
 325 extract_file_from_line() {
 326     vtlog "extract_file_from_line $1 disk=#$2#"
 327     if ! [ -b "$2" ]; then
 328         vterr "disk #$2# not exist"
 329         return
 330     fi
 331
 332     sector=$(echo $1 | $AWK '{print $(NF-1)}')
 333     length=$(echo $1 | $AWK '{print $NF}')
 334     vtlog "sector=$sector  length=$length"
 335
 336     $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o $3
 337     if [ -e $3 ]; then
 338         vtlog "extract file from iso success"
 339         $BUSYBOX_PATH/true
 340     else
 341         vterr "extract file from iso fail"
 342         $BUSYBOX_PATH/false
 343     fi
 344 }
 345
 346 install_rpm_from_line() {
 347     vtlog "install_rpm_from_line $1 disk=#$2#"
 348
 349     if ! [ -b "$2" ]; then
 350         vterr "disk #$2# not exist"
 351         return
 352     fi
 353
 354     sector=$(echo $1 | $AWK '{print $(NF-1)}')
 355     length=$(echo $1 | $AWK '{print $NF}')
 356     vtlog "sector=$sector  length=$length"
 357
 358     $VTOY_PATH/tool/vtoydm -e -f $VTOY_PATH/ventoy_image_map -d ${2} -s $sector -l $length -o /tmp/xxx.rpm
 359     if [ -e /tmp/xxx.rpm ]; then
 360         vtlog "extract rpm file from iso success"
 361     else
 362         vterr "extract rpm file from iso fail"
 363         return
 364     fi
 365
 366     CURPWD=$($BUSYBOX_PATH/pwd)
 367
 368     cd /
 369     vtlog "install rpm..."
 370     $BUSYBOX_PATH/rpm2cpio /tmp/xxx.rpm | $BUSYBOX_PATH/cpio -idm 2>>$VTLOG
 371     cd $CURPWD
 372
 373     $BUSYBOX_PATH/rm -f /tmp/xxx.rpm
 374 }
 375
 376 dump_whole_iso_file() {
 377    $VTOY_PATH/tool/vtoydm -p -f $VTOY_PATH/ventoy_image_map -d $usb_disk | while read vtline; do
 378         vtlog "dmtable line: $vtline"
 379         vtcount=$(echo $vtline | $AWK '{print $2}')
 380         vtoffset=$(echo $vtline | $AWK '{print $NF}')
 381         $BUSYBOX_PATH/dd if=$usb_disk of="$1" bs=512 count=$vtcount skip=$vtoffset oflag=append conv=notrunc
 382     done
 383 }
 384
 385 ventoy_copy_device_mapper() {
 386     if [ -L $VTOY_DM_PATH ]; then
 387         vtlog "replace block device link $1..."
 388         $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_${1#/dev/}
 389         VT_MAPPER_LINK=$($BUSYBOX_PATH/readlink $VTOY_DM_PATH)
 390         $BUSYBOX_PATH/cp -a "/dev/mapper/$VT_MAPPER_LINK" "$1"
 391     elif [ -b $VTOY_DM_PATH ]; then
 392         vtlog "replace block device $1..."
 393         $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_${1#/dev/}
 394         $BUSYBOX_PATH/cp -a "$VTOY_DM_PATH" "$1"
 395     else
 396
 397         vtlog "$VTOY_DM_PATH not exist, now check /dev/dm-X ..."
 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         DM_VT_ID=$($VT_DM_BIN ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
 405         vtlog "DM_VT_ID=$DM_VT_ID ..."
 406         $BUSYBOX_PATH/mv "$1" $VTOY_PATH/dev_backup_${1#/dev/}
 407         $BUSYBOX_PATH/mknod -m 0666 "$1" b $DM_VT_ID
 408     fi
 409 }
 410
 411 # create link for device-mapper
 412 ventoy_create_persistent_link() {
 413     blkdev_num=$($VTOY_PATH/tool/dmsetup ls | grep vtoy_persistent | sed 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1:\2/')
 414     vtDM=$(ventoy_find_dm_id ${blkdev_num})
 415
 416     if ! [ -d /dev/disk/by-label ]; then
 417         mkdir -p /dev/disk/by-label
 418     fi
 419
 420     VTLABEL=$($BUSYBOX_PATH/blkid /dev/$vtDM | $SED 's/.*LABEL="\([^"]*\)".*/\1/')
 421     if [ -z "$VTLABEL" ]; then
 422         VTLABEL=casper-rw
 423     fi
 424
 425     vtlog "Persistent Label: ##${VTLABEL}##"
 426
 427     if ! [ -e /dev/disk/by-label/$VTLABEL ]; then
 428         vtOldDir=$PWD
 429         cd /dev/disk/by-label
 430         ln -s ../../$vtDM $VTLABEL
 431         cd $vtOldDir
 432     fi
 433 }
 434
 435 ventoy_udev_disk_common_hook() {
 436
 437     VTDISK="${1:0:-1}"
 438
 439     if [ -e /vtoy/vtoy ]; then
 440         VTRWMOD=""
 441     else
 442         VTRWMOD="--readonly"
 443     fi
 444
 445     # create device mapper for iso image file
 446     if create_ventoy_device_mapper "/dev/$VTDISK" $VTRWMOD; then
 447         vtlog "==== create ventoy device mapper success ===="
 448     else
 449         vtlog "==== create ventoy device mapper failed ===="
 450
 451         $SLEEP 3
 452
 453         if $GREP -q "/dev/$VTDISK" /proc/mounts; then
 454             $GREP "/dev/$VTDISK" /proc/mounts | while read vtLine; do
 455                 vtPart=$(echo $vtLine | $AWK '{print $1}')
 456                 vtMnt=$(echo $vtLine | $AWK '{print $2}')
 457                 vtlog "$vtPart is mounted on $vtMnt  now umount it ..."
 458                 $BUSYBOX_PATH/umount $vtMnt
 459             done
 460         fi
 461
 462         if create_ventoy_device_mapper "/dev/$VTDISK" $VTRWMOD; then
 463             vtlog "==== create ventoy device mapper success after retry ===="
 464         else
 465             vtlog "==== create ventoy device mapper failed after retry ===="
 466             return
 467         fi
 468     fi
 469
 470     if [ "$2" = "noreplace" ]; then
 471         vtlog "no need to replace block device"
 472     else
 473         ventoy_copy_device_mapper "/dev/$1"
 474     fi
 475
 476     if [ -f $VTOY_PATH/ventoy_persistent_map ]; then
 477         create_persistent_device_mapper "/dev/$VTDISK"
 478         ventoy_create_persistent_link
 479     fi
 480 }
 481
 482 ventoy_create_dev_ventoy_part() {
 483     blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
 484     $BUSYBOX_PATH/mknod -m 0666 /dev/ventoy b $blkdev_num
 485
 486     if [ -e /vtoy_dm_table ]; then
 487         vtPartid=1
 488
 489         $CAT /vtoy_dm_table | while read vtline; do
 490             echo $vtline > /ventoy/dm_table_part${vtPartid}
 491             $VTOY_PATH/tool/dmsetup create ventoy${vtPartid} /ventoy/dm_table_part${vtPartid}
 492
 493             blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy${vtPartid} | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
 494             $BUSYBOX_PATH/mknod -m 0666 /dev/ventoy${vtPartid} b $blkdev_num
 495
 496             vtPartid=$(expr $vtPartid + 1)
 497         done
 498     fi
 499 }
 500
 501 is_inotify_ventoy_part() {
 502     if echo $1 | $GREP -q "2$"; then
 503         if ! [ -e /sys/block/$1 ]; then
 504             if [ -e /sys/class/block/$1 ]; then
 505                 if [ -e /dev/${1:0:-1} ]; then
 506                     $VTOY_PATH/tool/vtoydump -f $VTOY_PATH/ventoy_os_param -c ${1:0:-1}
 507                     return
 508                 fi
 509             fi
 510         fi
 511     fi
 512
 513     [ "1" = "0" ]
 514 }
 515
 516 ventoy_find_dm_id() {
 517     for vt in $($BUSYBOX_PATH/ls /sys/block/); do
 518         if [ "${vt:0:3}" = "dm-" ]; then
 519             vtMajorMinor=$($CAT /sys/block/$vt/dev)
 520             if [ "$vtMajorMinor" = "$1" ]; then
 521                 echo ${vt}
 522                 return
 523             fi
 524         fi
 525     done
 526     echo 'xx'
 527 }
 528
 529 ventoy_swap_device() {
 530     mv $1 $VTOY_PATH/swap_tmp_dev
 531     mv $2 $1
 532     mv $VTOY_PATH/swap_tmp_dev $2
 533 }
 534