IMG/cpio/ventoy/hook/ewe/ventoy-disk.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 vtlog "######### $0 $* ############"
  23
  24 if is_ventoy_hook_finished; then
  25     exit 0
  26 fi
  27
  28 wait_for_usb_disk_ready
  29
  30 vtdiskname=$(get_ventoy_disk_name)
  31 if [ "$vtdiskname" = "unknown" ]; then
  32     vtlog "ventoy disk not found"
  33     exit 0
  34 fi
  35
  36 ventoy_udev_disk_common_hook "${vtdiskname#/dev/}2" "noreplace"
  37
  38 blkdev_num=$($VTOY_PATH/tool/dmsetup ls | $GREP ventoy | $SED 's/.*(\([0-9][0-9]*\),.*\([0-9][0-9]*\).*/\1:\2/')
  39 vtDM=$(ventoy_find_dm_id ${blkdev_num})
  40 vtlog "blkdev_num=$blkdev_num vtDM=$vtDM ..."
  41
  42 while [ -n "Y" ]; do
  43     if [ -b /dev/$vtDM ]; then
  44         break
  45     else
  46         sleep 0.3
  47     fi
  48 done
  49
  50 if [ -n "$1" ]; then
  51     vtlog "ln -s /dev/$vtDM $1"
  52
  53     if [ -e "$1" ]; then
  54         vtlog "$1 already exist"
  55     else
  56         ln -s /dev/$vtDM "$1"
  57     fi
  58 else
  59     vtLABEL=$($BUSYBOX_PATH/blkid /dev/$vtDM | $SED 's/.*LABEL="\([^"]*\)".*/\1/')
  60     vtlog "vtLABEL is $vtLABEL"
  61
  62     if [ -z "$vtLABEL" ]; then
  63         vtLABEL=$($SED "s/.*label=\([^ ]*\)/\1/" /proc/cmdline)
  64         vtlog "vtLABEL is $vtLABEL from cmdline"
  65     fi
  66
  67     if [ -e "/dev/disk/by-label/$vtLABEL" ]; then
  68         vtlog "$1 already exist"
  69     else
  70         ln -s /dev/$vtDM "/dev/disk/by-label/$vtLABEL"
  71     fi
  72 fi
  73
  74 # OK finish
  75 set_ventoy_hook_finish