Add files via upload

新建文件夹 (4)/distros/dracut/vtoy.sh

@@ -0,0 +1,134 @@
+#!/bin/sh
+#************************************************************************************
+# Copyright (c) 2020, longpanda <admin@ventoy.net>
+# 
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of the GNU General Public License as
+# published by the Free Software Foundation; either version 3 of the
+# License, or (at your option) any later version.
+# 
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# General Public License for more details.
+# 
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, see <http://www.gnu.org/licenses/>.
+# 
+#************************************************************************************
+
+. ./tools/efi_legacy_grub.sh
+
+if [ -e /lib/dracut/dracut-install ]; then
+    vtmodpath=/lib/dracut/modules.d/99ventoy
+else
+    vtmodpath=/usr/lib/dracut/modules.d/99ventoy
+fi
+
+if [ -d /etc/dracut.conf.d ]; then
+    dracutConfPath=/etc/dracut.conf.d
+else
+    dracutConfPath=/usr/lib/dracut/dracut.conf.d
+fi
+
+
+rm -f /bin/vtoydump /bin/vtoypartx /bin/vtoytool /bin/vtoydmpatch
+rm -f $dracutConfPath/ventoy.conf
+rm -rf $vtmodpath
+mkdir -p $vtmodpath
+
+cp -a $vtdumpcmd /bin/vtoydump
+cp -a $partxcmd /bin/vtoypartx
+cp -a $vtoytool /bin/vtoytool
+cat /bin/vtoydump $dmpatchko > /bin/vtoydmpatch
+cp -a ./distros/$initrdtool/module-setup.sh $vtmodpath/
+cp -a ./distros/$initrdtool/ventoy-settled.sh $vtmodpath/
+
+#early centos release doesn't have require_binaries
+if [ -e $vtmodpath/../../dracut-functions ]; then
+    if grep -q require_binaries $vtmodpath/../../dracut-functions; then
+        :
+    else
+        sed "/require_binaries/d" -i $vtmodpath/module-setup.sh
+    fi
+fi
+
+
+for md in $(cat ./tools/vtoydrivers); do
+    if [ -n "$md" ]; then
+        if modinfo -n $md 2>/dev/null | grep -q '\.ko'; then
+            extdrivers="$extdrivers $md"
+        fi
+    fi
+done
+
+
+#generate dracut conf file
+cat >$dracutConfPath/ventoy.conf <<EOF
+add_dracutmodules+=" ventoy "
+force_drivers+=" $extdrivers "
+EOF
+
+
+echo "updating the initramfs, please wait ..."
+dracut -f --no-hostonly
+
+kv=$(uname -r)
+for k in $(ls /lib/modules); do
+    if [ "$k" != "$kv" ]; then
+        echo "updating initramfs for $k please wait ..."
+        dracut -f --no-hostonly --kver $k
+    fi
+done
+
+
+disable_grub_os_probe
+
+#wrapper grub-probe
+echo "grub mkconfig ..."
+PROBE_PATH=$(find_grub_probe_path)
+EDITENV_PATH=$(find_grub_editenv_path)
+MKCONFIG_PATH=$(find_grub_mkconfig_path)
+echo "PROBE_PATH=$PROBE_PATH EDITENV_PATH=$EDITENV_PATH MKCONFIG_PATH=$MKCONFIG_PATH"
+
+if [ -e "$PROBE_PATH" -a -e "$MKCONFIG_PATH" ]; then
+    wrapper_grub_probe $PROBE_PATH
+    if [ -e "$EDITENV_PATH" ]; then
+        wrapper_grub_editenv $EDITENV_PATH
+    fi
+
+    GRUB_CFG_PATH=$(find_grub_config_path)
+    if [ -f "$GRUB_CFG_PATH" ]; then
+        echo "$MKCONFIG_PATH -o $GRUB_CFG_PATH"
+        $MKCONFIG_PATH -o $GRUB_CFG_PATH
+    else
+        echo "$MKCONFIG_PATH null"
+        $MKCONFIG_PATH > /dev/null 2>&1
+    fi
+fi
+
+
+if [ -e /sys/firmware/efi ]; then
+    if [ -e /dev/mapper/ventoy ]; then
+        echo "This is ventoy enviroment"
+    else
+        update_grub_config
+        install_legacy_bios_grub
+    fi
+    
+    if [ "$1" = "-s" ]; then
+        recover_shim_efi
+    else
+        replace_shim_efi
+    fi
+    
+    if [ -d /boot/EFI/EFI/mageia ]; then
+        if ! [ -d /boot/EFI/EFI/boot ]; then
+            mkdir -p /boot/EFI/EFI/boot
+            if [ -f /boot/EFI/EFI/mageia/grubx64.efi ]; then
+                cp -a /boot/EFI/EFI/mageia/grubx64.efi /boot/EFI/EFI/boot/bootx64.efi
+            fi
+        fi
+    fi
+    
+fi