IMG/cpio/ventoy/hook/vine/udev_disk_hook.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 . /ventoy/hook/ventoy-hook-lib.sh
  21
  22 ventoy_os_install_dmsetup_by_ko() {
  23     vtlog "ventoy_os_install_dmsetup_by_ko $1"
  24
  25     vtVer=$($BUSYBOX_PATH/uname -r)
  26     if $BUSYBOX_PATH/uname -m | $GREP -q 64; then
  27         vtBit=64
  28     else
  29         vtBit=32
  30     fi
  31
  32     ventoy_extract_vtloopex $1  vine
  33     vtLoopExDir=$VTOY_PATH/vtloopex/vine/vtloopex
  34
  35     if [ -e $vtLoopExDir/dm-mod/$vtVer/$vtBit/dm-mod.ko.xz ]; then
  36         $BUSYBOX_PATH/xz -d $vtLoopExDir/dm-mod/$vtVer/$vtBit/dm-mod.ko.xz
  37         insmod $vtLoopExDir/dm-mod/$vtVer/$vtBit/dm-mod.ko
  38     fi
  39 }
  40
  41 ventoy_os_install_dmsetup_by_rpm() {
  42     vtlog "ventoy_os_install_dmsetup_by_rpm $1"
  43
  44     vt_usb_disk=$1
  45
  46     # dump iso file location
  47     $VTOY_PATH/tool/vtoydm -i -f $VTOY_PATH/ventoy_image_map -d ${vt_usb_disk} > $VTOY_PATH/iso_file_list
  48
  49     # install dmsetup
  50     LINE=$($GREP 'kernel-[0-9].*\.rpm'  $VTOY_PATH/iso_file_list)
  51     if [ $? -eq 0 ]; then
  52         extract_rpm_from_line "$LINE" ${vt_usb_disk}
  53     fi
  54
  55     vtKoName=$($BUSYBOX_PATH/find $VTOY_PATH/rpm/ -name dm-mod.ko*)
  56     vtlog "vtKoName=$vtKoName"
  57
  58     insmod $vtKoName
  59
  60     $BUSYBOX_PATH/rm -rf $VTOY_PATH/rpm/
  61 }
  62
  63 if is_ventoy_hook_finished || not_ventoy_disk "${1:0:-1}"; then
  64     exit 0
  65 fi
  66
  67 ventoy_os_install_dmsetup_by_ko "/dev/$1"
  68 if $GREP -q 'device.mapper' /proc/devices; then
  69     vtlog "device-mapper module install OK"
  70 else
  71     ventoy_os_install_dmsetup_by_rpm "/dev/${1:0:-1}"
  72 fi
  73
  74 ventoy_udev_disk_common_hook $*
  75
  76 blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1:\2/')
  77 blkdev_num_dev=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1 \2/')
  78 vtDM=$(ventoy_find_dm_id ${blkdev_num})
  79
  80 [ -b /dev/$vtDM ] || $BUSYBOX_PATH/mknod -m 0660 /dev/$vtDM b $blkdev_num_dev
  81 $BUSYBOX_PATH/rm -rf /dev/mapper
  82
  83 #Create a fake IDE-CDROM driver can't be ide-scsi   /proc has been patched to /vtoy
  84 $BUSYBOX_PATH/mkdir -p /vtoy
  85 $BUSYBOX_PATH/cp -a /proc/ide /vtoy/
  86 $BUSYBOX_PATH/mkdir -p /vtoy/ide/aztcd
  87
  88 echo 'ide' > /vtoy/ide/aztcd/driver
  89 echo 'cdrom' > /vtoy/ide/aztcd/media
  90
  91
  92 # OK finish
  93 set_ventoy_hook_finish
  94
  95 exit 0