IMG/cpio/ventoy/hook/debian/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() {
  23
  24     vtlog "ventoy_os_install_dmsetup $1 ..."
  25
  26     vt_usb_disk=$1
  27
  28     # dump iso file location
  29     $VTOY_PATH/tool/vtoydm -i -f $VTOY_PATH/ventoy_image_map -d ${vt_usb_disk} > $VTOY_PATH/iso_file_list
  30
  31     # install dmsetup
  32     LINE=$($GREP ' dmsetup.*\.udeb'  $VTOY_PATH/iso_file_list)
  33     if [ $? -eq 0 ]; then
  34         install_udeb_from_line "$LINE" ${vt_usb_disk}
  35     fi
  36
  37     # install libdevmapper
  38     LINE=$($GREP ' libdevmapper.*\.udeb'  $VTOY_PATH/iso_file_list)
  39     if [ $? -eq 0 ]; then
  40         install_udeb_from_line "$LINE" ${vt_usb_disk}
  41     fi
  42
  43     # install md-modules
  44     LINE=$($GREP ' md-modules.*\.udeb'  $VTOY_PATH/iso_file_list)
  45     if [ $? -eq 0 ]; then
  46         LINTCNT=$($GREP -c ' md-modules.*\.udeb'  $VTOY_PATH/iso_file_list)
  47         if [ $LINTCNT -gt 1 ]; then
  48             vtlog "more than one pkgs, need to filter..."
  49             VER=$($BUSYBOX_PATH/uname -r)
  50             LINE=$($GREP ' md-modules.*\.udeb'  $VTOY_PATH/iso_file_list | $GREP $VER)
  51         fi
  52         install_udeb_from_line "$LINE" ${vt_usb_disk}
  53     fi
  54
  55     # insmod md-mod if needed
  56     if $GREP -q 'device-mapper' /proc/devices; then
  57         vtlog "device mapper module is loaded"
  58     else
  59         vtlog "device mapper module is NOT loaded, now load it..."
  60
  61         VER=$($BUSYBOX_PATH/uname -r)
  62         KO=$($FIND /lib/modules/$VER/kernel/drivers/md -name "dm-mod*")
  63         vtlog "KO=$KO"
  64
  65         insmod $KO
  66     fi
  67
  68     vtlog "dmsetup install finish, now check it..."
  69     if dmsetup info >> $VTLOG 2>&1; then
  70         vtlog "dmsetup work ok"
  71     else
  72         vtlog "dmsetup not work, now try to load eglibc ..."
  73
  74         # install eglibc (some ubuntu 32 bit version need it)
  75         LINE=$($GREP 'libc6-.*\.udeb'  $VTOY_PATH/iso_file_list)
  76         if [ $? -eq 0 ]; then
  77             install_udeb_from_line "$LINE" ${vt_usb_disk}
  78         fi
  79
  80         if dmsetup info >> $VTLOG 2>&1; then
  81             vtlog "dmsetup work ok after retry"
  82         else
  83             vtlog "dmsetup still not work after retry"
  84         fi
  85     fi
  86 }
  87
  88 if is_ventoy_hook_finished || not_ventoy_disk "${1:0:-1}"; then
  89     exit 0
  90 fi
  91
  92 vtlog "==== $0 $* ===="
  93
  94 dmsetup_path=$(ventoy_find_bin_path dmsetup)
  95 if [ -z "$dmsetup_path" ]; then
  96     ventoy_os_install_dmsetup "/dev/${1:0:-1}"
  97 fi
  98
  99 if ! $GREP -q 'device-mapper' /proc/devices; then
 100     ventoy_os_install_dmsetup "/dev/${1:0:-1}"
 101 fi
 102
 103 ventoy_udev_disk_common_hook $*
 104
 105 #
 106 # Some distro default only accept usb partitions as install medium.
 107 # So if ventoy is installed on a non-USB device, we just mount /cdrom here except
 108 # for these has boot=live or boot=casper parameter in cmdline
 109 #
 110 VT_BUS_USB=""
 111 if [ -n "$ID_BUS" ]; then
 112     if echo $ID_BUS | $GREP -q -i usb; then
 113         VT_BUS_USB="YES"
 114     fi
 115 else
 116     if $BUSYBOX_PATH/ls -l /sys/class/block/${1:0:-1} | $GREP -q -i usb; then
 117         VT_BUS_USB="YES"
 118     fi
 119 fi
 120
 121 if [ -n "$VT_BUS_USB" ]; then
 122     vtlog "$1 is USB device"
 123 else
 124     vtlog "$1 is NOT USB device (bus $ID_BUS)"
 125
 126     if $EGREP -q 'boot=|casper' /proc/cmdline; then
 127         vtlog "boot=, or casper, don't mount"
 128     else
 129         vtlog "No boot param, need to mount"
 130         $BUSYBOX_PATH/mkdir /cdrom
 131
 132         if [ -b $VTOY_DM_PATH ]; then
 133             vtlog "mount $VTOY_DM_PATH ..."
 134             $BUSYBOX_PATH/mount -t iso9660 $VTOY_DM_PATH  /cdrom
 135         else
 136             vtlog "mount /dev/$1 ..."
 137             $BUSYBOX_PATH/mount -t iso9660 /dev/$1  /cdrom
 138         fi
 139     fi
 140 fi
 141
 142
 143 # OK finish
 144 set_ventoy_hook_finish