mirror of
https://github.com/SkimrMe/funky_tv
synced 2026-01-29 19:15:23 +00:00
Add files via upload
This commit is contained in:
SkimrMe
GitHub
27
新建文件夹 (4)/distros/dracut/check.sh
Normal file
27
新建文件夹 (4)/distros/dracut/check.sh
Normal file
@@ -0,0 +1,27 @@
|
||||
#!/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/>.
|
||||
#
|
||||
#************************************************************************************
|
||||
|
||||
vtoy_check_dracut() {
|
||||
if which dracut >/dev/null 2>&1; then
|
||||
if dracut --list-modules | grep -q '^dm$'; then
|
||||
[ -n "1" ]; return
|
||||
fi
|
||||
fi
|
||||
[ -z "1" ]
|
||||
}
|
||||
20
新建文件夹 (4)/distros/dracut/module-setup.sh
Normal file
20
新建文件夹 (4)/distros/dracut/module-setup.sh
Normal file
@@ -0,0 +1,20 @@
|
||||
#!/bin/bash
|
||||
# -*- mode: shell-script; indent-tabs-mode: nil; sh-basic-offset: 4; -*-
|
||||
# ex: ts=8 sw=4 sts=4 et filetype=sh
|
||||
|
||||
check() {
|
||||
require_binaries sed grep awk || return 1
|
||||
return 255
|
||||
}
|
||||
|
||||
depends() {
|
||||
echo dm
|
||||
return 0
|
||||
}
|
||||
|
||||
|
||||
install() {
|
||||
inst_multiple sed grep awk dd sort head find basename xzcat zcat vtoydump vtoypartx vtoydmpatch vtoytool
|
||||
inst_hook initqueue/settled 99 "$moddir/ventoy-settled.sh"
|
||||
dracut_need_initqueue
|
||||
}
|
||||
370
新建文件夹 (4)/distros/dracut/ventoy-settled.sh
Normal file
370
新建文件夹 (4)/distros/dracut/ventoy-settled.sh
Normal file
@@ -0,0 +1,370 @@
|
||||
#!/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/>.
|
||||
#
|
||||
#************************************************************************************
|
||||
|
||||
type getarg >/dev/null 2>&1 || . /lib/dracut-lib.sh
|
||||
|
||||
###########################
|
||||
###########################
|
||||
#AUTO_INSERT_COMMON_FUNC
|
||||
|
||||
ventoy_check_efivars() {
|
||||
if [ -e /sys/firmware/efi ]; then
|
||||
if grep -q efivar /proc/mounts; then
|
||||
:
|
||||
else
|
||||
if [ -e /sys/firmware/efi/efivars ]; then
|
||||
mount -t efivarfs efivarfs /sys/firmware/efi/efivars >/dev/null 2>&1
|
||||
fi
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
ventoy_log() {
|
||||
echo "$@" >> /tmp/vtoy.log
|
||||
}
|
||||
|
||||
ventoy_need_dm_patch() {
|
||||
if vtoydump -R > /dev/null 2>&1; then
|
||||
[ 1 -eq 1 ]; return
|
||||
fi
|
||||
|
||||
if grep -q "VTOY_LINUX_REMOUNT=1" /proc/cmdline; then
|
||||
[ 1 -eq 1 ]; return
|
||||
fi
|
||||
|
||||
[ 1 -eq 0 ]
|
||||
}
|
||||
|
||||
ventoy_check_insmod() {
|
||||
if [ -f /bin/kmod ]; then
|
||||
[ -f /bin/insmod ] || ln -s /bin/kmod /bin/insmod
|
||||
[ -f /bin/lsmod ] || ln -s /bin/kmod /bin/lsmod
|
||||
fi
|
||||
}
|
||||
|
||||
ventoy_need_proc_ibt() {
|
||||
vtKv=$(uname -r)
|
||||
vtMajor=$(echo $vtKv | awk -F. '{print $1}')
|
||||
vtMinor=$(echo $vtKv | awk -F. '{print $2}')
|
||||
|
||||
#ibt was supported since linux kernel 5.18
|
||||
if [ $vtMajor -lt 5 ]; then
|
||||
false; return
|
||||
elif [ $vtMajor -eq 5 ]; then
|
||||
if [ $vtMajor -lt 18 ]; then
|
||||
false; return
|
||||
fi
|
||||
fi
|
||||
|
||||
if grep -q ' ibt=off' /proc/cmdline; then
|
||||
false; return
|
||||
fi
|
||||
|
||||
#hardware CPU doesn't support IBT
|
||||
if vtoytool vtoykmod -I; then
|
||||
:
|
||||
else
|
||||
false; return
|
||||
fi
|
||||
|
||||
#dot.CONFIG not enabled
|
||||
if grep -q ' ibt_restore$' /proc/kallsyms; then
|
||||
:
|
||||
else
|
||||
false; return
|
||||
fi
|
||||
|
||||
true
|
||||
}
|
||||
|
||||
|
||||
ventoy_do_dm_patch() {
|
||||
vtDmPatchDebug=0
|
||||
ventoy_log 'ventoy_do_dm_patch'
|
||||
if [ -f /tmp/dm_patch.ko ]; then
|
||||
if grep -q 'dm_patch' /proc/modules; then
|
||||
ventoy_log "dm_patch already exist"
|
||||
else
|
||||
ventoy_log "dm_patch reinstall"
|
||||
insmod /tmp/dm_patch.ko >>/tmp/vtoy.log 2>&1
|
||||
if grep -q 'dm_patch' /proc/modules; then
|
||||
ventoy_log "dm_patch reinstall success"
|
||||
else
|
||||
ventoy_log "dm_patch reinstall failed"
|
||||
fi
|
||||
fi
|
||||
return
|
||||
fi
|
||||
|
||||
if [ -f /bin/vtoydump ]; then
|
||||
vtHeadSize=$(stat -c '%s' /bin/vtoydump)
|
||||
dd if=/bin/vtoydmpatch of=/tmp/dm_patch.ko bs=1 skip=$vtHeadSize >/dev/null 2>&1
|
||||
elif [ -f /sbin/vtoydump ]; then
|
||||
vtHeadSize=$(stat -c '%s' /sbin/vtoydump)
|
||||
dd if=/sbin/vtoydmpatch of=/tmp/dm_patch.ko bs=1 skip=$vtHeadSize >/dev/null 2>&1
|
||||
else
|
||||
ventoy_log 'vtoydump not found'
|
||||
return
|
||||
fi
|
||||
|
||||
if ! grep -m1 -q dm_get_table_device /proc/kallsyms; then
|
||||
ventoy_log "modprobe dm_mod"
|
||||
modprobe dm_mod >>/tmp/vtoy.log 2>&1
|
||||
fi
|
||||
|
||||
cat /proc/kallsyms | sort > /tmp/kallsyms
|
||||
|
||||
if grep -m1 -q 'open_table_device.isra' /tmp/kallsyms; then
|
||||
vtLine=$(vtoytool vtoyksym open_table_device.isra /tmp/kallsyms)
|
||||
else
|
||||
vtLine=$(vtoytool vtoyksym dm_get_table_device /tmp/kallsyms)
|
||||
fi
|
||||
get_addr=$(echo $vtLine | awk '{print $1}')
|
||||
get_size=$(echo $vtLine | awk '{print $2}')
|
||||
|
||||
vtLine=$(vtoytool vtoyksym blkdev_get_by_dev /tmp/kallsyms)
|
||||
blkdev_get_addr=$(echo $vtLine | awk '{print $1}')
|
||||
|
||||
vtLine=$(vtoytool vtoyksym blkdev_put /tmp/kallsyms)
|
||||
blkdev_put_addr=$(echo $vtLine | awk '{print $1}')
|
||||
|
||||
vtLine=$(vtoytool vtoyksym bdev_open_by_dev /tmp/kallsyms)
|
||||
bdev_open_addr=$(echo $vtLine | awk '{print $1}')
|
||||
|
||||
vtLine=$(vtoytool vtoyksym bdev_file_open_by_dev /tmp/kallsyms)
|
||||
bdev_file_open_by_dev=$(echo $vtLine | awk '{print $1}')
|
||||
|
||||
if grep -m1 -q 'close_table_device.isra' /tmp/kallsyms; then
|
||||
vtLine=$(vtoytool vtoyksym close_table_device.isra /tmp/kallsyms)
|
||||
else
|
||||
vtLine=$(vtoytool vtoyksym dm_put_table_device /tmp/kallsyms)
|
||||
fi
|
||||
put_addr=$(echo $vtLine | awk '{print $1}')
|
||||
put_size=$(echo $vtLine | awk '{print $2}')
|
||||
|
||||
ro_addr=$(grep ' set_memory_ro$' /proc/kallsyms | awk '{print $1}')
|
||||
rw_addr=$(grep ' set_memory_rw$' /proc/kallsyms | awk '{print $1}')
|
||||
kprobe_reg_addr=$(grep ' register_kprobe$' /proc/kallsyms | awk '{print $1}')
|
||||
kprobe_unreg_addr=$(grep ' unregister_kprobe$' /proc/kallsyms | awk '{print $1}')
|
||||
|
||||
if [ "$VTOY_DEBUG_LEVEL" = "01" ]; then
|
||||
vtDmPatchDebug=1
|
||||
fi
|
||||
|
||||
if grep -q 'dmpatch_debug' /proc/cmdline; then
|
||||
vtDmPatchDebug=1
|
||||
fi
|
||||
|
||||
if [ $vtDmPatchDebug -eq 1 ]; then
|
||||
printk_addr=$(grep ' printk$' /proc/kallsyms | awk '{print $1}')
|
||||
vtDebug="-v"
|
||||
elif grep -q "vtdebug" /proc/cmdline; then
|
||||
printk_addr=$(grep ' printk$' /proc/kallsyms | awk '{print $1}')
|
||||
vtDebug="-v"
|
||||
else
|
||||
printk_addr=0
|
||||
fi
|
||||
|
||||
if [ -z "$printk_addr" ]; then
|
||||
printk_addr=$(grep ' _printk$' /proc/kallsyms | awk '{print $1}')
|
||||
fi
|
||||
|
||||
if ventoy_need_proc_ibt; then
|
||||
ventoy_log "need to proc IBT"
|
||||
vtIBT='0x8888'
|
||||
else
|
||||
ventoy_log "NO need to proc IBT"
|
||||
vtIBT='0'
|
||||
fi
|
||||
|
||||
#printk_addr=$(grep ' printk$' /proc/kallsyms | awk '{print $1}')
|
||||
#vtDebug="-v"
|
||||
|
||||
ventoy_log get_addr=$get_addr get_size=$get_size vtDebug=$vtDebug
|
||||
ventoy_log put_addr=$put_addr put_size=$put_size
|
||||
ventoy_log blkdev_get_addr=$blkdev_get_addr blkdev_put_addr=$blkdev_put_addr
|
||||
ventoy_log kprobe_reg_addr=$kprobe_reg_addr kprobe_unreg_addr=$kprobe_unreg_addr
|
||||
ventoy_log ro_addr=$ro_addr rw_addr=$rw_addr printk_addr=$printk_addr
|
||||
ventoy_log bdev_open_addr=$bdev_open_addr bdev_file_open_by_dev=$bdev_file_open_by_dev
|
||||
|
||||
if [ "$get_addr" = "0" -o "$put_addr" = "0" ]; then
|
||||
ventoy_log "Invalid symbol address"
|
||||
return
|
||||
fi
|
||||
if [ "$ro_addr" = "0" -o "$rw_addr" = "0" ]; then
|
||||
ventoy_log "Invalid symbol address"
|
||||
return
|
||||
fi
|
||||
|
||||
vtKv=$(uname -r)
|
||||
vtKVMajor=$(echo $vtKv | awk -F. '{print $1}')
|
||||
vtKVMinor=$(echo $vtKv | awk -F. '{print $2}')
|
||||
vtKVSubMinor=$(echo $vtKv | awk -F. '{print $3}')
|
||||
|
||||
if [ ! -d /lib/modules/$vtKv ]; then
|
||||
ventoy_log "No modules directory found"
|
||||
return
|
||||
elif [ -d /lib/modules/$vtKv/kernel/fs ]; then
|
||||
vtModPath=$(find /lib/modules/$vtKv/kernel/fs/ -name "*.ko*" | head -n1)
|
||||
else
|
||||
vtModPath=$(find /lib/modules/$vtKv/kernel/ -name "xfs.ko*" | head -n1)
|
||||
fi
|
||||
|
||||
if [ -z "$vtModPath" ]; then
|
||||
vtModPath=$(find /lib/modules/$vtKv/kernel/ -name "*.ko*" | head -n1)
|
||||
fi
|
||||
|
||||
vtModName=$(basename $vtModPath)
|
||||
[ -f /tmp/$vtModName ] && rm -f /tmp/$vtModName
|
||||
|
||||
ventoy_log "template module is $vtModPath $vtModName"
|
||||
|
||||
if [ -z "$vtModPath" ]; then
|
||||
ventoy_log "No template module found"
|
||||
return
|
||||
elif echo $vtModPath | grep -q "[.]ko$"; then
|
||||
cp -a $vtModPath /tmp/$vtModName
|
||||
elif echo $vtModPath | grep -q "[.]ko[.]xz$"; then
|
||||
xzcat $vtModPath > /tmp/$vtModName
|
||||
elif echo $vtModPath | grep -q "[.]ko[.]gz$"; then
|
||||
zcat $vtModPath > /tmp/$vtModName
|
||||
elif echo $vtModPath | grep -q "[.]ko[.]zst$"; then
|
||||
zstdcat $vtModPath > /tmp/$vtModName
|
||||
else
|
||||
ventoy_log "unsupport module type"
|
||||
return
|
||||
fi
|
||||
|
||||
|
||||
#step1: modify vermagic/mod crc/relocation
|
||||
vtoytool vtoykmod -u $vtKVMajor $vtKVMinor /tmp/dm_patch.ko /tmp/$vtModName $vtDebug >>/tmp/vtoy.log 2>&1
|
||||
|
||||
#step2: fill parameters
|
||||
vtPgsize=$(vtoytool vtoyksym -p)
|
||||
vtoytool vtoykmod -f /tmp/dm_patch.ko $vtPgsize 0x$printk_addr 0x$ro_addr 0x$rw_addr $get_addr $get_size $put_addr $put_size 0x$kprobe_reg_addr 0x$kprobe_unreg_addr $vtKVMajor $vtIBT $vtKVMinor $blkdev_get_addr $blkdev_put_addr $vtKVSubMinor $bdev_open_addr $bdev_file_open_by_dev $vtDebug >>/tmp/vtoy.log 2>&1
|
||||
|
||||
ventoy_check_insmod
|
||||
insmod /tmp/dm_patch.ko >>/tmp/vtoy.log 2>&1
|
||||
|
||||
if grep -q 'dm_patch' /proc/modules; then
|
||||
ventoy_log "dm_patch success"
|
||||
else
|
||||
ventoy_log "dm_patch failed"
|
||||
fi
|
||||
}
|
||||
|
||||
|
||||
ventoy_dm_patch_proc_begin() {
|
||||
if ventoy_need_dm_patch; then
|
||||
export vtLevel1=$(cat /proc/sys/kernel/printk | awk '{print $1}')
|
||||
export vtLevel2=$(cat /proc/sys/kernel/printk | awk '{print $2}')
|
||||
export vtLevel3=$(cat /proc/sys/kernel/printk | awk '{print $3}')
|
||||
export vtLevel4=$(cat /proc/sys/kernel/printk | awk '{print $4}')
|
||||
|
||||
#suppress printk message
|
||||
echo 0 $vtLevel2 0 $vtLevel4 > /proc/sys/kernel/printk
|
||||
fi
|
||||
}
|
||||
|
||||
ventoy_dm_patch_install() {
|
||||
if ventoy_need_dm_patch; then
|
||||
ventoy_do_dm_patch
|
||||
fi
|
||||
}
|
||||
|
||||
ventoy_dm_patch_remove() {
|
||||
if ventoy_need_dm_patch; then
|
||||
if grep -q 'dm_patch' /proc/modules; then
|
||||
ventoy_log "remove dm_patch"
|
||||
rmmod dm_patch
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
ventoy_dm_create_ventoy() {
|
||||
ventoy_dm_patch_install
|
||||
dmsetup create ventoy /ventoy_table
|
||||
vret=$?
|
||||
ventoy_dm_patch_remove
|
||||
return $vret
|
||||
}
|
||||
|
||||
ventoy_dm_patch_proc_end() {
|
||||
if ventoy_need_dm_patch; then
|
||||
#recover printk level
|
||||
echo $vtLevel1 $vtLevel2 $vtLevel3 $vtLevel4 > /proc/sys/kernel/printk
|
||||
fi
|
||||
}
|
||||
|
||||
#check for efivarfs
|
||||
ventoy_check_efivars
|
||||
|
||||
if ! vtoydump > /dev/null 2>&1; then
|
||||
info 'vtoydump failed'
|
||||
return
|
||||
fi
|
||||
|
||||
#already done
|
||||
if dmsetup ls | grep -q ventoy; then
|
||||
info 'ventoy already exist'
|
||||
return
|
||||
fi
|
||||
|
||||
ventoy_dm_patch_proc_begin
|
||||
|
||||
#flush multipath before dmsetup
|
||||
multipath -F > /dev/null 2>&1
|
||||
|
||||
vtoydump -L > /ventoy_table
|
||||
if ventoy_dm_create_ventoy; then
|
||||
:
|
||||
else
|
||||
sleep 3
|
||||
multipath -F > /dev/null 2>&1
|
||||
ventoy_dm_create_ventoy
|
||||
fi
|
||||
|
||||
DEVDM=/dev/mapper/ventoy
|
||||
|
||||
loop=0
|
||||
while ! [ -e $DEVDM ]; do
|
||||
sleep 0.5
|
||||
let loop+=1
|
||||
if [ $loop -gt 10 ]; then
|
||||
echo "Waiting for ventoy device ..." > /dev/console
|
||||
fi
|
||||
|
||||
if [ $loop -gt 10 -a $loop -lt 15 ]; then
|
||||
multipath -F > /dev/null 2>&1
|
||||
ventoy_dm_create_ventoy
|
||||
fi
|
||||
done
|
||||
|
||||
for ID in $(vtoypartx $DEVDM -oNR | grep -v NR); do
|
||||
PART_START=$(vtoypartx $DEVDM -n$ID -oSTART,SECTORS | grep -v START | awk '{print $1}')
|
||||
PART_SECTOR=$(vtoypartx $DEVDM -n$ID -oSTART,SECTORS | grep -v START | awk '{print $2}')
|
||||
|
||||
echo "0 $PART_SECTOR linear $DEVDM $PART_START" > /ventoy_part_table
|
||||
dmsetup create ventoy$ID /ventoy_part_table
|
||||
done
|
||||
|
||||
rm -f /ventoy_table
|
||||
rm -f /ventoy_part_table
|
||||
|
||||
ventoy_dm_patch_proc_end
|
||||
134
新建文件夹 (4)/distros/dracut/vtoy.sh
Normal file
134
新建文件夹 (4)/distros/dracut/vtoy.sh
Normal file
@@ -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
|
||||
48
新建文件夹 (4)/distros/initramfstool/check.sh
Normal file
48
新建文件夹 (4)/distros/initramfstool/check.sh
Normal file
@@ -0,0 +1,48 @@
|
||||
#!/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/>.
|
||||
#
|
||||
#************************************************************************************
|
||||
|
||||
vtoy_false() {
|
||||
[ -z "1" ]
|
||||
}
|
||||
|
||||
vtoy_true() {
|
||||
[ -n "1" ]
|
||||
}
|
||||
|
||||
|
||||
vtoy_check_initramfs_tool() {
|
||||
if which update-initramfs >/dev/null 2>&1; then
|
||||
|
||||
if ! [ -f /usr/share/initramfs-tools/hook-functions ]; then
|
||||
vtoy_false; return
|
||||
fi
|
||||
|
||||
if ! [ -f /usr/share/initramfs-tools/hooks/dmsetup ]; then
|
||||
vtoy_false; return
|
||||
fi
|
||||
|
||||
if ! [ -d /etc/initramfs-tools/scripts/local-top ]; then
|
||||
vtoy_false; return
|
||||
fi
|
||||
|
||||
vtoy_true
|
||||
else
|
||||
vtoy_false
|
||||
fi
|
||||
}
|
||||
56
新建文件夹 (4)/distros/initramfstool/vtoy-hook.sh
Normal file
56
新建文件夹 (4)/distros/initramfstool/vtoy-hook.sh
Normal file
@@ -0,0 +1,56 @@
|
||||
#!/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/>.
|
||||
#
|
||||
#************************************************************************************
|
||||
|
||||
PREREQ="dmsetup"
|
||||
prereqs() {
|
||||
echo "$PREREQ"
|
||||
}
|
||||
|
||||
case $1 in
|
||||
prereqs)
|
||||
prereqs
|
||||
exit 0
|
||||
;;
|
||||
esac
|
||||
|
||||
. /usr/share/initramfs-tools/hook-functions
|
||||
|
||||
# Begin real processing below this line
|
||||
|
||||
for md in $(cat /sbin/vtoydrivers); do
|
||||
if [ -n "$md" ]; then
|
||||
if modinfo -n $md 2>/dev/null | grep -q '\.ko'; then
|
||||
force_load $md
|
||||
fi
|
||||
fi
|
||||
done
|
||||
|
||||
for ef in dd sort head find basename xzcat zcat; do
|
||||
for vp in /bin /sbin /usr/bin /usr/sbin; do
|
||||
if [ -f $vp/$ef ]; then
|
||||
copy_exec $vp/$ef /sbin
|
||||
break
|
||||
fi
|
||||
done
|
||||
done
|
||||
|
||||
copy_exec /sbin/vtoytool /sbin
|
||||
copy_exec /sbin/vtoydmpatch /sbin
|
||||
copy_exec /sbin/vtoypartx /sbin
|
||||
copy_exec /sbin/vtoydump /sbin
|
||||
377
新建文件夹 (4)/distros/initramfstool/vtoy-local-top.sh
Normal file
377
新建文件夹 (4)/distros/initramfstool/vtoy-local-top.sh
Normal file
@@ -0,0 +1,377 @@
|
||||
#!/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/>.
|
||||
#
|
||||
#************************************************************************************
|
||||
|
||||
###########################
|
||||
###########################
|
||||
#AUTO_INSERT_COMMON_FUNC
|
||||
|
||||
ventoy_check_efivars() {
|
||||
if [ -e /sys/firmware/efi ]; then
|
||||
if grep -q efivar /proc/mounts; then
|
||||
:
|
||||
else
|
||||
if [ -e /sys/firmware/efi/efivars ]; then
|
||||
mount -t efivarfs efivarfs /sys/firmware/efi/efivars >/dev/null 2>&1
|
||||
fi
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
ventoy_log() {
|
||||
echo "$@" >> /tmp/vtoy.log
|
||||
}
|
||||
|
||||
ventoy_need_dm_patch() {
|
||||
if vtoydump -R > /dev/null 2>&1; then
|
||||
[ 1 -eq 1 ]; return
|
||||
fi
|
||||
|
||||
if grep -q "VTOY_LINUX_REMOUNT=1" /proc/cmdline; then
|
||||
[ 1 -eq 1 ]; return
|
||||
fi
|
||||
|
||||
[ 1 -eq 0 ]
|
||||
}
|
||||
|
||||
ventoy_check_insmod() {
|
||||
if [ -f /bin/kmod ]; then
|
||||
[ -f /bin/insmod ] || ln -s /bin/kmod /bin/insmod
|
||||
[ -f /bin/lsmod ] || ln -s /bin/kmod /bin/lsmod
|
||||
fi
|
||||
}
|
||||
|
||||
ventoy_need_proc_ibt() {
|
||||
vtKv=$(uname -r)
|
||||
vtMajor=$(echo $vtKv | awk -F. '{print $1}')
|
||||
vtMinor=$(echo $vtKv | awk -F. '{print $2}')
|
||||
|
||||
#ibt was supported since linux kernel 5.18
|
||||
if [ $vtMajor -lt 5 ]; then
|
||||
false; return
|
||||
elif [ $vtMajor -eq 5 ]; then
|
||||
if [ $vtMajor -lt 18 ]; then
|
||||
false; return
|
||||
fi
|
||||
fi
|
||||
|
||||
if grep -q ' ibt=off' /proc/cmdline; then
|
||||
false; return
|
||||
fi
|
||||
|
||||
#hardware CPU doesn't support IBT
|
||||
if vtoytool vtoykmod -I; then
|
||||
:
|
||||
else
|
||||
false; return
|
||||
fi
|
||||
|
||||
#dot.CONFIG not enabled
|
||||
if grep -q ' ibt_restore$' /proc/kallsyms; then
|
||||
:
|
||||
else
|
||||
false; return
|
||||
fi
|
||||
|
||||
true
|
||||
}
|
||||
|
||||
|
||||
ventoy_do_dm_patch() {
|
||||
vtDmPatchDebug=0
|
||||
ventoy_log 'ventoy_do_dm_patch'
|
||||
if [ -f /tmp/dm_patch.ko ]; then
|
||||
if grep -q 'dm_patch' /proc/modules; then
|
||||
ventoy_log "dm_patch already exist"
|
||||
else
|
||||
ventoy_log "dm_patch reinstall"
|
||||
insmod /tmp/dm_patch.ko >>/tmp/vtoy.log 2>&1
|
||||
if grep -q 'dm_patch' /proc/modules; then
|
||||
ventoy_log "dm_patch reinstall success"
|
||||
else
|
||||
ventoy_log "dm_patch reinstall failed"
|
||||
fi
|
||||
fi
|
||||
return
|
||||
fi
|
||||
|
||||
if [ -f /bin/vtoydump ]; then
|
||||
vtHeadSize=$(stat -c '%s' /bin/vtoydump)
|
||||
dd if=/bin/vtoydmpatch of=/tmp/dm_patch.ko bs=1 skip=$vtHeadSize >/dev/null 2>&1
|
||||
elif [ -f /sbin/vtoydump ]; then
|
||||
vtHeadSize=$(stat -c '%s' /sbin/vtoydump)
|
||||
dd if=/sbin/vtoydmpatch of=/tmp/dm_patch.ko bs=1 skip=$vtHeadSize >/dev/null 2>&1
|
||||
else
|
||||
ventoy_log 'vtoydump not found'
|
||||
return
|
||||
fi
|
||||
|
||||
if ! grep -m1 -q dm_get_table_device /proc/kallsyms; then
|
||||
ventoy_log "modprobe dm_mod"
|
||||
modprobe dm_mod >>/tmp/vtoy.log 2>&1
|
||||
fi
|
||||
|
||||
cat /proc/kallsyms | sort > /tmp/kallsyms
|
||||
|
||||
if grep -m1 -q 'open_table_device.isra' /tmp/kallsyms; then
|
||||
vtLine=$(vtoytool vtoyksym open_table_device.isra /tmp/kallsyms)
|
||||
else
|
||||
vtLine=$(vtoytool vtoyksym dm_get_table_device /tmp/kallsyms)
|
||||
fi
|
||||
get_addr=$(echo $vtLine | awk '{print $1}')
|
||||
get_size=$(echo $vtLine | awk '{print $2}')
|
||||
|
||||
vtLine=$(vtoytool vtoyksym blkdev_get_by_dev /tmp/kallsyms)
|
||||
blkdev_get_addr=$(echo $vtLine | awk '{print $1}')
|
||||
|
||||
vtLine=$(vtoytool vtoyksym blkdev_put /tmp/kallsyms)
|
||||
blkdev_put_addr=$(echo $vtLine | awk '{print $1}')
|
||||
|
||||
vtLine=$(vtoytool vtoyksym bdev_open_by_dev /tmp/kallsyms)
|
||||
bdev_open_addr=$(echo $vtLine | awk '{print $1}')
|
||||
|
||||
vtLine=$(vtoytool vtoyksym bdev_file_open_by_dev /tmp/kallsyms)
|
||||
bdev_file_open_by_dev=$(echo $vtLine | awk '{print $1}')
|
||||
|
||||
if grep -m1 -q 'close_table_device.isra' /tmp/kallsyms; then
|
||||
vtLine=$(vtoytool vtoyksym close_table_device.isra /tmp/kallsyms)
|
||||
else
|
||||
vtLine=$(vtoytool vtoyksym dm_put_table_device /tmp/kallsyms)
|
||||
fi
|
||||
put_addr=$(echo $vtLine | awk '{print $1}')
|
||||
put_size=$(echo $vtLine | awk '{print $2}')
|
||||
|
||||
ro_addr=$(grep ' set_memory_ro$' /proc/kallsyms | awk '{print $1}')
|
||||
rw_addr=$(grep ' set_memory_rw$' /proc/kallsyms | awk '{print $1}')
|
||||
kprobe_reg_addr=$(grep ' register_kprobe$' /proc/kallsyms | awk '{print $1}')
|
||||
kprobe_unreg_addr=$(grep ' unregister_kprobe$' /proc/kallsyms | awk '{print $1}')
|
||||
|
||||
if [ "$VTOY_DEBUG_LEVEL" = "01" ]; then
|
||||
vtDmPatchDebug=1
|
||||
fi
|
||||
|
||||
if grep -q 'dmpatch_debug' /proc/cmdline; then
|
||||
vtDmPatchDebug=1
|
||||
fi
|
||||
|
||||
if [ $vtDmPatchDebug -eq 1 ]; then
|
||||
printk_addr=$(grep ' printk$' /proc/kallsyms | awk '{print $1}')
|
||||
vtDebug="-v"
|
||||
elif grep -q "vtdebug" /proc/cmdline; then
|
||||
printk_addr=$(grep ' printk$' /proc/kallsyms | awk '{print $1}')
|
||||
vtDebug="-v"
|
||||
else
|
||||
printk_addr=0
|
||||
fi
|
||||
|
||||
if [ -z "$printk_addr" ]; then
|
||||
printk_addr=$(grep ' _printk$' /proc/kallsyms | awk '{print $1}')
|
||||
fi
|
||||
|
||||
if ventoy_need_proc_ibt; then
|
||||
ventoy_log "need to proc IBT"
|
||||
vtIBT='0x8888'
|
||||
else
|
||||
ventoy_log "NO need to proc IBT"
|
||||
vtIBT='0'
|
||||
fi
|
||||
|
||||
#printk_addr=$(grep ' printk$' /proc/kallsyms | awk '{print $1}')
|
||||
#vtDebug="-v"
|
||||
|
||||
ventoy_log get_addr=$get_addr get_size=$get_size vtDebug=$vtDebug
|
||||
ventoy_log put_addr=$put_addr put_size=$put_size
|
||||
ventoy_log blkdev_get_addr=$blkdev_get_addr blkdev_put_addr=$blkdev_put_addr
|
||||
ventoy_log kprobe_reg_addr=$kprobe_reg_addr kprobe_unreg_addr=$kprobe_unreg_addr
|
||||
ventoy_log ro_addr=$ro_addr rw_addr=$rw_addr printk_addr=$printk_addr
|
||||
ventoy_log bdev_open_addr=$bdev_open_addr bdev_file_open_by_dev=$bdev_file_open_by_dev
|
||||
|
||||
if [ "$get_addr" = "0" -o "$put_addr" = "0" ]; then
|
||||
ventoy_log "Invalid symbol address"
|
||||
return
|
||||
fi
|
||||
if [ "$ro_addr" = "0" -o "$rw_addr" = "0" ]; then
|
||||
ventoy_log "Invalid symbol address"
|
||||
return
|
||||
fi
|
||||
|
||||
vtKv=$(uname -r)
|
||||
vtKVMajor=$(echo $vtKv | awk -F. '{print $1}')
|
||||
vtKVMinor=$(echo $vtKv | awk -F. '{print $2}')
|
||||
vtKVSubMinor=$(echo $vtKv | awk -F. '{print $3}')
|
||||
|
||||
if [ ! -d /lib/modules/$vtKv ]; then
|
||||
ventoy_log "No modules directory found"
|
||||
return
|
||||
elif [ -d /lib/modules/$vtKv/kernel/fs ]; then
|
||||
vtModPath=$(find /lib/modules/$vtKv/kernel/fs/ -name "*.ko*" | head -n1)
|
||||
else
|
||||
vtModPath=$(find /lib/modules/$vtKv/kernel/ -name "xfs.ko*" | head -n1)
|
||||
fi
|
||||
|
||||
if [ -z "$vtModPath" ]; then
|
||||
vtModPath=$(find /lib/modules/$vtKv/kernel/ -name "*.ko*" | head -n1)
|
||||
fi
|
||||
|
||||
vtModName=$(basename $vtModPath)
|
||||
[ -f /tmp/$vtModName ] && rm -f /tmp/$vtModName
|
||||
|
||||
ventoy_log "template module is $vtModPath $vtModName"
|
||||
|
||||
if [ -z "$vtModPath" ]; then
|
||||
ventoy_log "No template module found"
|
||||
return
|
||||
elif echo $vtModPath | grep -q "[.]ko$"; then
|
||||
cp -a $vtModPath /tmp/$vtModName
|
||||
elif echo $vtModPath | grep -q "[.]ko[.]xz$"; then
|
||||
xzcat $vtModPath > /tmp/$vtModName
|
||||
elif echo $vtModPath | grep -q "[.]ko[.]gz$"; then
|
||||
zcat $vtModPath > /tmp/$vtModName
|
||||
elif echo $vtModPath | grep -q "[.]ko[.]zst$"; then
|
||||
zstdcat $vtModPath > /tmp/$vtModName
|
||||
else
|
||||
ventoy_log "unsupport module type"
|
||||
return
|
||||
fi
|
||||
|
||||
|
||||
#step1: modify vermagic/mod crc/relocation
|
||||
vtoytool vtoykmod -u $vtKVMajor $vtKVMinor /tmp/dm_patch.ko /tmp/$vtModName $vtDebug >>/tmp/vtoy.log 2>&1
|
||||
|
||||
#step2: fill parameters
|
||||
vtPgsize=$(vtoytool vtoyksym -p)
|
||||
vtoytool vtoykmod -f /tmp/dm_patch.ko $vtPgsize 0x$printk_addr 0x$ro_addr 0x$rw_addr $get_addr $get_size $put_addr $put_size 0x$kprobe_reg_addr 0x$kprobe_unreg_addr $vtKVMajor $vtIBT $vtKVMinor $blkdev_get_addr $blkdev_put_addr $vtKVSubMinor $bdev_open_addr $bdev_file_open_by_dev $vtDebug >>/tmp/vtoy.log 2>&1
|
||||
|
||||
ventoy_check_insmod
|
||||
insmod /tmp/dm_patch.ko >>/tmp/vtoy.log 2>&1
|
||||
|
||||
if grep -q 'dm_patch' /proc/modules; then
|
||||
ventoy_log "dm_patch success"
|
||||
else
|
||||
ventoy_log "dm_patch failed"
|
||||
fi
|
||||
}
|
||||
|
||||
|
||||
ventoy_dm_patch_proc_begin() {
|
||||
if ventoy_need_dm_patch; then
|
||||
export vtLevel1=$(cat /proc/sys/kernel/printk | awk '{print $1}')
|
||||
export vtLevel2=$(cat /proc/sys/kernel/printk | awk '{print $2}')
|
||||
export vtLevel3=$(cat /proc/sys/kernel/printk | awk '{print $3}')
|
||||
export vtLevel4=$(cat /proc/sys/kernel/printk | awk '{print $4}')
|
||||
|
||||
#suppress printk message
|
||||
echo 0 $vtLevel2 0 $vtLevel4 > /proc/sys/kernel/printk
|
||||
fi
|
||||
}
|
||||
|
||||
ventoy_dm_patch_install() {
|
||||
if ventoy_need_dm_patch; then
|
||||
ventoy_do_dm_patch
|
||||
fi
|
||||
}
|
||||
|
||||
ventoy_dm_patch_remove() {
|
||||
if ventoy_need_dm_patch; then
|
||||
if grep -q 'dm_patch' /proc/modules; then
|
||||
ventoy_log "remove dm_patch"
|
||||
rmmod dm_patch
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
ventoy_dm_create_ventoy() {
|
||||
ventoy_dm_patch_install
|
||||
dmsetup create ventoy /ventoy_table
|
||||
vret=$?
|
||||
ventoy_dm_patch_remove
|
||||
return $vret
|
||||
}
|
||||
|
||||
ventoy_dm_patch_proc_end() {
|
||||
if ventoy_need_dm_patch; then
|
||||
#recover printk level
|
||||
echo $vtLevel1 $vtLevel2 $vtLevel3 $vtLevel4 > /proc/sys/kernel/printk
|
||||
fi
|
||||
}
|
||||
|
||||
|
||||
vtoy_wait_for_device() {
|
||||
while ! vtoydump > /dev/null 2>&1; do
|
||||
sleep 0.5
|
||||
done
|
||||
}
|
||||
|
||||
vtoy_device_mapper_proc() {
|
||||
#flush multipath before dmsetup
|
||||
multipath -F > /dev/null 2>&1
|
||||
|
||||
vtoydump -L > /ventoy_table
|
||||
if ventoy_dm_create_ventoy; then
|
||||
:
|
||||
else
|
||||
sleep 3
|
||||
multipath -F > /dev/null 2>&1
|
||||
ventoy_dm_create_ventoy
|
||||
fi
|
||||
|
||||
|
||||
DEVDM=/dev/mapper/ventoy
|
||||
|
||||
loop=0
|
||||
while ! [ -e $DEVDM ]; do
|
||||
sleep 0.5
|
||||
let loop+=1
|
||||
if [ $loop -gt 10 ]; then
|
||||
echo "Waiting for ventoy device ..." > /dev/console
|
||||
fi
|
||||
|
||||
if [ $loop -gt 10 -a $loop -lt 15 ]; then
|
||||
multipath -F > /dev/null 2>&1
|
||||
ventoy_dm_create_ventoy
|
||||
fi
|
||||
done
|
||||
|
||||
for ID in $(vtoypartx $DEVDM -oNR | grep -v NR); do
|
||||
PART_START=$(vtoypartx $DEVDM -n$ID -oSTART,SECTORS | grep -v START | awk '{print $1}')
|
||||
PART_SECTOR=$(vtoypartx $DEVDM -n$ID -oSTART,SECTORS | grep -v START | awk '{print $2}')
|
||||
|
||||
echo "0 $PART_SECTOR linear $DEVDM $PART_START" > /ventoy_part_table
|
||||
dmsetup create ventoy$ID /ventoy_part_table
|
||||
done
|
||||
|
||||
rm -f /ventoy_table
|
||||
rm -f /ventoy_part_table
|
||||
}
|
||||
|
||||
case $1 in
|
||||
prereqs)
|
||||
exit 0
|
||||
;;
|
||||
esac
|
||||
|
||||
#check for efivarfs
|
||||
ventoy_check_efivars
|
||||
|
||||
if vtoydump -c > /dev/null 2>&1; then
|
||||
vtoy_wait_for_device
|
||||
ventoy_dm_patch_proc_begin
|
||||
vtoy_device_mapper_proc
|
||||
ventoy_dm_patch_proc_end
|
||||
fi
|
||||
|
||||
119
新建文件夹 (4)/distros/initramfstool/vtoy.sh
Normal file
119
新建文件夹 (4)/distros/initramfstool/vtoy.sh
Normal file
@@ -0,0 +1,119 @@
|
||||
#!/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/>.
|
||||
#
|
||||
#************************************************************************************
|
||||
|
||||
vtoy_clean_env() {
|
||||
rm -f /sbin/vtoydump /sbin/vtoypartx /sbin/vtoytool /sbin/vtoydmpatch /sbin/vtoydrivers
|
||||
rm -f /usr/share/initramfs-tools/hooks/vtoy-hook.sh
|
||||
rm -f /etc/initramfs-tools/scripts/local-top/vtoy-local-top.sh
|
||||
}
|
||||
|
||||
vtoy_efi_fixup() {
|
||||
if [ -d /boot/efi/EFI ]; then
|
||||
for f in 'boot/bootx64.efi' 'boot/BOOTX64.efi' 'boot/BOOTX64.EFI' 'BOOT/bootx64.efi' 'BOOT/BOOTX64.efi' 'BOOT/BOOTX64.EFI'; do
|
||||
if [ -f /boot/efi/EFI/$f ]; then
|
||||
return
|
||||
fi
|
||||
done
|
||||
fi
|
||||
|
||||
Dirs=$(ls /boot/efi/EFI)
|
||||
|
||||
if ! [ -d /boot/efi/EFI/boot ]; then
|
||||
mkdir -p /boot/efi/EFI/boot
|
||||
fi
|
||||
|
||||
for d in $Dirs; do
|
||||
for e in 'grubx64.efi' 'GRUBX64.EFI' 'bootx64.efi' 'BOOTX64.EFI'; do
|
||||
if [ -f "/boot/efi/EFI/$d/$e" ]; then
|
||||
cp -a "/boot/efi/EFI/$d/$e" /boot/efi/EFI/boot/bootx64.efi
|
||||
return
|
||||
fi
|
||||
done
|
||||
done
|
||||
}
|
||||
|
||||
vtoy_fix_elementary_partuuid() {
|
||||
if grep -q 'elementary OS' /etc/os-release; then
|
||||
if grep -q '^PARTUUID=.* /boot/efi ' /etc/fstab; then
|
||||
part=$(grep ' /boot/efi ' /proc/mounts | awk '{print $1}')
|
||||
uuid=$(blkid -s UUID -o value $part)
|
||||
sed -i "s#^PARTUUID=.* /boot/efi #UUID=$uuid /boot/efi #g" /etc/fstab
|
||||
echo "Fix elementary OS PARTUUID $part"
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
. ./tools/efi_legacy_grub.sh
|
||||
|
||||
vtoy_clean_env
|
||||
|
||||
cp -a $vtdumpcmd /sbin/vtoydump
|
||||
cp -a $partxcmd /sbin/vtoypartx
|
||||
cp -a $vtoytool /sbin/vtoytool
|
||||
cat /sbin/vtoydump $dmpatchko > /sbin/vtoydmpatch
|
||||
cp -a ./tools/vtoydrivers /sbin/vtoydrivers
|
||||
cp -a ./distros/$initrdtool/vtoy-hook.sh /usr/share/initramfs-tools/hooks/
|
||||
cp -a ./distros/$initrdtool/vtoy-local-top.sh /etc/initramfs-tools/scripts/local-top/
|
||||
|
||||
echo "updating the initramfs, please wait ..."
|
||||
update-initramfs -u
|
||||
|
||||
|
||||
disable_grub_os_probe
|
||||
|
||||
#wrapper grub-probe
|
||||
echo "grub mkconfig ..."
|
||||
PROBE_PATH=$(find_grub_probe_path)
|
||||
MKCONFIG_PATH=$(find_grub_mkconfig_path)
|
||||
echo "PROBE_PATH=$PROBE_PATH MKCONFIG_PATH=$MKCONFIG_PATH"
|
||||
|
||||
if [ -e "$PROBE_PATH" -a -e "$MKCONFIG_PATH" ]; then
|
||||
wrapper_grub_probe $PROBE_PATH
|
||||
|
||||
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
|
||||
|
||||
|
||||
|
||||
#efi fixup
|
||||
if [ -e /sys/firmware/efi ]; then
|
||||
if [ -e /dev/mapper/ventoy ]; then
|
||||
echo "This is ventoy enviroment"
|
||||
else
|
||||
vtoy_fix_elementary_partuuid
|
||||
update_grub_config
|
||||
install_legacy_bios_grub
|
||||
fi
|
||||
|
||||
if [ "$1" = "-s" ]; then
|
||||
recover_shim_efi
|
||||
else
|
||||
replace_shim_efi
|
||||
fi
|
||||
|
||||
vtoy_efi_fixup
|
||||
fi
|
||||
|
||||
33
新建文件夹 (4)/distros/mkinitcpio/check.sh
Normal file
33
新建文件夹 (4)/distros/mkinitcpio/check.sh
Normal file
@@ -0,0 +1,33 @@
|
||||
#!/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/>.
|
||||
#
|
||||
#************************************************************************************
|
||||
|
||||
vtoy_check_mkinitcpio() {
|
||||
if which mkinitcpio >/dev/null 2>&1; then
|
||||
|
||||
for vtfile in '/etc/mkinitcpio.conf' '/usr/lib/initcpio/install/lvm2'; do
|
||||
if ! [ -f $vtfile ]; then
|
||||
[ -z "1" ]; return
|
||||
fi
|
||||
done
|
||||
|
||||
[ -n "1" ]
|
||||
else
|
||||
[ -z "1" ]
|
||||
fi
|
||||
}
|
||||
382
新建文件夹 (4)/distros/mkinitcpio/ventoy-hook.sh
Normal file
382
新建文件夹 (4)/distros/mkinitcpio/ventoy-hook.sh
Normal file
@@ -0,0 +1,382 @@
|
||||
#!/bin/bash
|
||||
#************************************************************************************
|
||||
# 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/>.
|
||||
#
|
||||
#************************************************************************************
|
||||
|
||||
###########################
|
||||
###########################
|
||||
#AUTO_INSERT_COMMON_FUNC
|
||||
|
||||
ventoy_check_efivars() {
|
||||
if [ -e /sys/firmware/efi ]; then
|
||||
if grep -q efivar /proc/mounts; then
|
||||
:
|
||||
else
|
||||
if [ -e /sys/firmware/efi/efivars ]; then
|
||||
mount -t efivarfs efivarfs /sys/firmware/efi/efivars >/dev/null 2>&1
|
||||
fi
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
ventoy_log() {
|
||||
echo "$@" >> /tmp/vtoy.log
|
||||
}
|
||||
|
||||
ventoy_need_dm_patch() {
|
||||
if vtoydump -R > /dev/null 2>&1; then
|
||||
[ 1 -eq 1 ]; return
|
||||
fi
|
||||
|
||||
if grep -q "VTOY_LINUX_REMOUNT=1" /proc/cmdline; then
|
||||
[ 1 -eq 1 ]; return
|
||||
fi
|
||||
|
||||
[ 1 -eq 0 ]
|
||||
}
|
||||
|
||||
ventoy_check_insmod() {
|
||||
if [ -f /bin/kmod ]; then
|
||||
[ -f /bin/insmod ] || ln -s /bin/kmod /bin/insmod
|
||||
[ -f /bin/lsmod ] || ln -s /bin/kmod /bin/lsmod
|
||||
fi
|
||||
}
|
||||
|
||||
ventoy_need_proc_ibt() {
|
||||
vtKv=$(uname -r)
|
||||
vtMajor=$(echo $vtKv | awk -F. '{print $1}')
|
||||
vtMinor=$(echo $vtKv | awk -F. '{print $2}')
|
||||
|
||||
#ibt was supported since linux kernel 5.18
|
||||
if [ $vtMajor -lt 5 ]; then
|
||||
false; return
|
||||
elif [ $vtMajor -eq 5 ]; then
|
||||
if [ $vtMajor -lt 18 ]; then
|
||||
false; return
|
||||
fi
|
||||
fi
|
||||
|
||||
if grep -q ' ibt=off' /proc/cmdline; then
|
||||
false; return
|
||||
fi
|
||||
|
||||
#hardware CPU doesn't support IBT
|
||||
if vtoytool vtoykmod -I; then
|
||||
:
|
||||
else
|
||||
false; return
|
||||
fi
|
||||
|
||||
#dot.CONFIG not enabled
|
||||
if grep -q ' ibt_restore$' /proc/kallsyms; then
|
||||
:
|
||||
else
|
||||
false; return
|
||||
fi
|
||||
|
||||
true
|
||||
}
|
||||
|
||||
|
||||
ventoy_do_dm_patch() {
|
||||
vtDmPatchDebug=0
|
||||
ventoy_log 'ventoy_do_dm_patch'
|
||||
if [ -f /tmp/dm_patch.ko ]; then
|
||||
if grep -q 'dm_patch' /proc/modules; then
|
||||
ventoy_log "dm_patch already exist"
|
||||
else
|
||||
ventoy_log "dm_patch reinstall"
|
||||
insmod /tmp/dm_patch.ko >>/tmp/vtoy.log 2>&1
|
||||
if grep -q 'dm_patch' /proc/modules; then
|
||||
ventoy_log "dm_patch reinstall success"
|
||||
else
|
||||
ventoy_log "dm_patch reinstall failed"
|
||||
fi
|
||||
fi
|
||||
return
|
||||
fi
|
||||
|
||||
if [ -f /bin/vtoydump ]; then
|
||||
vtHeadSize=$(stat -c '%s' /bin/vtoydump)
|
||||
dd if=/bin/vtoydmpatch of=/tmp/dm_patch.ko bs=1 skip=$vtHeadSize >/dev/null 2>&1
|
||||
elif [ -f /sbin/vtoydump ]; then
|
||||
vtHeadSize=$(stat -c '%s' /sbin/vtoydump)
|
||||
dd if=/sbin/vtoydmpatch of=/tmp/dm_patch.ko bs=1 skip=$vtHeadSize >/dev/null 2>&1
|
||||
else
|
||||
ventoy_log 'vtoydump not found'
|
||||
return
|
||||
fi
|
||||
|
||||
if ! grep -m1 -q dm_get_table_device /proc/kallsyms; then
|
||||
ventoy_log "modprobe dm_mod"
|
||||
modprobe dm_mod >>/tmp/vtoy.log 2>&1
|
||||
fi
|
||||
|
||||
cat /proc/kallsyms | sort > /tmp/kallsyms
|
||||
|
||||
if grep -m1 -q 'open_table_device.isra' /tmp/kallsyms; then
|
||||
vtLine=$(vtoytool vtoyksym open_table_device.isra /tmp/kallsyms)
|
||||
else
|
||||
vtLine=$(vtoytool vtoyksym dm_get_table_device /tmp/kallsyms)
|
||||
fi
|
||||
get_addr=$(echo $vtLine | awk '{print $1}')
|
||||
get_size=$(echo $vtLine | awk '{print $2}')
|
||||
|
||||
vtLine=$(vtoytool vtoyksym blkdev_get_by_dev /tmp/kallsyms)
|
||||
blkdev_get_addr=$(echo $vtLine | awk '{print $1}')
|
||||
|
||||
vtLine=$(vtoytool vtoyksym blkdev_put /tmp/kallsyms)
|
||||
blkdev_put_addr=$(echo $vtLine | awk '{print $1}')
|
||||
|
||||
vtLine=$(vtoytool vtoyksym bdev_open_by_dev /tmp/kallsyms)
|
||||
bdev_open_addr=$(echo $vtLine | awk '{print $1}')
|
||||
|
||||
vtLine=$(vtoytool vtoyksym bdev_file_open_by_dev /tmp/kallsyms)
|
||||
bdev_file_open_by_dev=$(echo $vtLine | awk '{print $1}')
|
||||
|
||||
if grep -m1 -q 'close_table_device.isra' /tmp/kallsyms; then
|
||||
vtLine=$(vtoytool vtoyksym close_table_device.isra /tmp/kallsyms)
|
||||
else
|
||||
vtLine=$(vtoytool vtoyksym dm_put_table_device /tmp/kallsyms)
|
||||
fi
|
||||
put_addr=$(echo $vtLine | awk '{print $1}')
|
||||
put_size=$(echo $vtLine | awk '{print $2}')
|
||||
|
||||
ro_addr=$(grep ' set_memory_ro$' /proc/kallsyms | awk '{print $1}')
|
||||
rw_addr=$(grep ' set_memory_rw$' /proc/kallsyms | awk '{print $1}')
|
||||
kprobe_reg_addr=$(grep ' register_kprobe$' /proc/kallsyms | awk '{print $1}')
|
||||
kprobe_unreg_addr=$(grep ' unregister_kprobe$' /proc/kallsyms | awk '{print $1}')
|
||||
|
||||
if [ "$VTOY_DEBUG_LEVEL" = "01" ]; then
|
||||
vtDmPatchDebug=1
|
||||
fi
|
||||
|
||||
if grep -q 'dmpatch_debug' /proc/cmdline; then
|
||||
vtDmPatchDebug=1
|
||||
fi
|
||||
|
||||
if [ $vtDmPatchDebug -eq 1 ]; then
|
||||
printk_addr=$(grep ' printk$' /proc/kallsyms | awk '{print $1}')
|
||||
vtDebug="-v"
|
||||
elif grep -q "vtdebug" /proc/cmdline; then
|
||||
printk_addr=$(grep ' printk$' /proc/kallsyms | awk '{print $1}')
|
||||
vtDebug="-v"
|
||||
else
|
||||
printk_addr=0
|
||||
fi
|
||||
|
||||
if [ -z "$printk_addr" ]; then
|
||||
printk_addr=$(grep ' _printk$' /proc/kallsyms | awk '{print $1}')
|
||||
fi
|
||||
|
||||
if ventoy_need_proc_ibt; then
|
||||
ventoy_log "need to proc IBT"
|
||||
vtIBT='0x8888'
|
||||
else
|
||||
ventoy_log "NO need to proc IBT"
|
||||
vtIBT='0'
|
||||
fi
|
||||
|
||||
#printk_addr=$(grep ' printk$' /proc/kallsyms | awk '{print $1}')
|
||||
#vtDebug="-v"
|
||||
|
||||
ventoy_log get_addr=$get_addr get_size=$get_size vtDebug=$vtDebug
|
||||
ventoy_log put_addr=$put_addr put_size=$put_size
|
||||
ventoy_log blkdev_get_addr=$blkdev_get_addr blkdev_put_addr=$blkdev_put_addr
|
||||
ventoy_log kprobe_reg_addr=$kprobe_reg_addr kprobe_unreg_addr=$kprobe_unreg_addr
|
||||
ventoy_log ro_addr=$ro_addr rw_addr=$rw_addr printk_addr=$printk_addr
|
||||
ventoy_log bdev_open_addr=$bdev_open_addr bdev_file_open_by_dev=$bdev_file_open_by_dev
|
||||
|
||||
if [ "$get_addr" = "0" -o "$put_addr" = "0" ]; then
|
||||
ventoy_log "Invalid symbol address"
|
||||
return
|
||||
fi
|
||||
if [ "$ro_addr" = "0" -o "$rw_addr" = "0" ]; then
|
||||
ventoy_log "Invalid symbol address"
|
||||
return
|
||||
fi
|
||||
|
||||
vtKv=$(uname -r)
|
||||
vtKVMajor=$(echo $vtKv | awk -F. '{print $1}')
|
||||
vtKVMinor=$(echo $vtKv | awk -F. '{print $2}')
|
||||
vtKVSubMinor=$(echo $vtKv | awk -F. '{print $3}')
|
||||
|
||||
if [ ! -d /lib/modules/$vtKv ]; then
|
||||
ventoy_log "No modules directory found"
|
||||
return
|
||||
elif [ -d /lib/modules/$vtKv/kernel/fs ]; then
|
||||
vtModPath=$(find /lib/modules/$vtKv/kernel/fs/ -name "*.ko*" | head -n1)
|
||||
else
|
||||
vtModPath=$(find /lib/modules/$vtKv/kernel/ -name "xfs.ko*" | head -n1)
|
||||
fi
|
||||
|
||||
if [ -z "$vtModPath" ]; then
|
||||
vtModPath=$(find /lib/modules/$vtKv/kernel/ -name "*.ko*" | head -n1)
|
||||
fi
|
||||
|
||||
vtModName=$(basename $vtModPath)
|
||||
[ -f /tmp/$vtModName ] && rm -f /tmp/$vtModName
|
||||
|
||||
ventoy_log "template module is $vtModPath $vtModName"
|
||||
|
||||
if [ -z "$vtModPath" ]; then
|
||||
ventoy_log "No template module found"
|
||||
return
|
||||
elif echo $vtModPath | grep -q "[.]ko$"; then
|
||||
cp -a $vtModPath /tmp/$vtModName
|
||||
elif echo $vtModPath | grep -q "[.]ko[.]xz$"; then
|
||||
xzcat $vtModPath > /tmp/$vtModName
|
||||
elif echo $vtModPath | grep -q "[.]ko[.]gz$"; then
|
||||
zcat $vtModPath > /tmp/$vtModName
|
||||
elif echo $vtModPath | grep -q "[.]ko[.]zst$"; then
|
||||
zstdcat $vtModPath > /tmp/$vtModName
|
||||
else
|
||||
ventoy_log "unsupport module type"
|
||||
return
|
||||
fi
|
||||
|
||||
|
||||
#step1: modify vermagic/mod crc/relocation
|
||||
vtoytool vtoykmod -u $vtKVMajor $vtKVMinor /tmp/dm_patch.ko /tmp/$vtModName $vtDebug >>/tmp/vtoy.log 2>&1
|
||||
|
||||
#step2: fill parameters
|
||||
vtPgsize=$(vtoytool vtoyksym -p)
|
||||
vtoytool vtoykmod -f /tmp/dm_patch.ko $vtPgsize 0x$printk_addr 0x$ro_addr 0x$rw_addr $get_addr $get_size $put_addr $put_size 0x$kprobe_reg_addr 0x$kprobe_unreg_addr $vtKVMajor $vtIBT $vtKVMinor $blkdev_get_addr $blkdev_put_addr $vtKVSubMinor $bdev_open_addr $bdev_file_open_by_dev $vtDebug >>/tmp/vtoy.log 2>&1
|
||||
|
||||
ventoy_check_insmod
|
||||
insmod /tmp/dm_patch.ko >>/tmp/vtoy.log 2>&1
|
||||
|
||||
if grep -q 'dm_patch' /proc/modules; then
|
||||
ventoy_log "dm_patch success"
|
||||
else
|
||||
ventoy_log "dm_patch failed"
|
||||
fi
|
||||
}
|
||||
|
||||
|
||||
ventoy_dm_patch_proc_begin() {
|
||||
if ventoy_need_dm_patch; then
|
||||
export vtLevel1=$(cat /proc/sys/kernel/printk | awk '{print $1}')
|
||||
export vtLevel2=$(cat /proc/sys/kernel/printk | awk '{print $2}')
|
||||
export vtLevel3=$(cat /proc/sys/kernel/printk | awk '{print $3}')
|
||||
export vtLevel4=$(cat /proc/sys/kernel/printk | awk '{print $4}')
|
||||
|
||||
#suppress printk message
|
||||
echo 0 $vtLevel2 0 $vtLevel4 > /proc/sys/kernel/printk
|
||||
fi
|
||||
}
|
||||
|
||||
ventoy_dm_patch_install() {
|
||||
if ventoy_need_dm_patch; then
|
||||
ventoy_do_dm_patch
|
||||
fi
|
||||
}
|
||||
|
||||
ventoy_dm_patch_remove() {
|
||||
if ventoy_need_dm_patch; then
|
||||
if grep -q 'dm_patch' /proc/modules; then
|
||||
ventoy_log "remove dm_patch"
|
||||
rmmod dm_patch
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
ventoy_dm_create_ventoy() {
|
||||
ventoy_dm_patch_install
|
||||
dmsetup create ventoy /ventoy_table
|
||||
vret=$?
|
||||
ventoy_dm_patch_remove
|
||||
return $vret
|
||||
}
|
||||
|
||||
ventoy_dm_patch_proc_end() {
|
||||
if ventoy_need_dm_patch; then
|
||||
#recover printk level
|
||||
echo $vtLevel1 $vtLevel2 $vtLevel3 $vtLevel4 > /proc/sys/kernel/printk
|
||||
fi
|
||||
}
|
||||
|
||||
|
||||
vtoy_wait_for_device() {
|
||||
local i=100
|
||||
|
||||
while [ $i -gt 0 ]; do
|
||||
if vtoydump > /dev/null 2>&1; then
|
||||
break
|
||||
else
|
||||
sleep 0.2
|
||||
fi
|
||||
i=$((i - 1))
|
||||
done
|
||||
}
|
||||
|
||||
vtoy_device_mapper_proc() {
|
||||
#flush multipath before dmsetup
|
||||
multipath -F > /dev/null 2>&1
|
||||
|
||||
vtoydump -L > /ventoy_table
|
||||
if ventoy_dm_create_ventoy; then
|
||||
:
|
||||
else
|
||||
sleep 3
|
||||
multipath -F > /dev/null 2>&1
|
||||
ventoy_dm_create_ventoy
|
||||
fi
|
||||
|
||||
DEVDM=/dev/mapper/ventoy
|
||||
|
||||
loop=0
|
||||
while ! [ -e $DEVDM ]; do
|
||||
sleep 0.5
|
||||
let loop+=1
|
||||
if [ $loop -gt 10 ]; then
|
||||
echo "Waiting for ventoy device ..." > /dev/console
|
||||
fi
|
||||
|
||||
if [ $loop -gt 10 -a $loop -lt 15 ]; then
|
||||
multipath -F > /dev/null 2>&1
|
||||
ventoy_dm_create_ventoy
|
||||
fi
|
||||
done
|
||||
|
||||
for ID in $(vtoypartx $DEVDM -oNR | grep -v NR); do
|
||||
PART_START=$(vtoypartx $DEVDM -n$ID -oSTART,SECTORS | grep -v START | awk '{print $1}')
|
||||
PART_SECTOR=$(vtoypartx $DEVDM -n$ID -oSTART,SECTORS | grep -v START | awk '{print $2}')
|
||||
|
||||
echo "0 $PART_SECTOR linear $DEVDM $PART_START" > /ventoy_part_table
|
||||
dmsetup create ventoy$ID /ventoy_part_table
|
||||
done
|
||||
|
||||
rm -f /ventoy_table
|
||||
rm -f /ventoy_part_table
|
||||
}
|
||||
|
||||
run_hook() {
|
||||
#check for efivarfs
|
||||
ventoy_check_efivars
|
||||
|
||||
if vtoydump -c >/dev/null 2>/dev/null; then
|
||||
vtoy_wait_for_device
|
||||
if vtoydump > /dev/null 2>&1; then
|
||||
ventoy_dm_patch_proc_begin
|
||||
vtoy_device_mapper_proc
|
||||
ventoy_dm_patch_proc_end
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
|
||||
48
新建文件夹 (4)/distros/mkinitcpio/ventoy-install.sh
Normal file
48
新建文件夹 (4)/distros/mkinitcpio/ventoy-install.sh
Normal file
@@ -0,0 +1,48 @@
|
||||
#!/bin/bash
|
||||
#************************************************************************************
|
||||
# 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/>.
|
||||
#
|
||||
#************************************************************************************
|
||||
|
||||
build() {
|
||||
add_binary "dd"
|
||||
add_binary "sort"
|
||||
add_binary "head"
|
||||
add_binary "find"
|
||||
add_binary "xzcat"
|
||||
add_binary "zcat"
|
||||
add_binary "basename"
|
||||
add_binary "vtoydump"
|
||||
add_binary "vtoypartx"
|
||||
add_binary "vtoydmpatch"
|
||||
add_binary "vtoytool"
|
||||
|
||||
for md in $(cat /sbin/vtoydrivers); do
|
||||
if [ -n "$md" ]; then
|
||||
if modinfo -n $md 2>/dev/null | grep -q '\.ko'; then
|
||||
add_module $md
|
||||
fi
|
||||
fi
|
||||
done
|
||||
|
||||
add_runscript
|
||||
}
|
||||
|
||||
help() {
|
||||
cat <<HELPEOF
|
||||
This hook enables ventoy in initramfs.
|
||||
HELPEOF
|
||||
}
|
||||
93
新建文件夹 (4)/distros/mkinitcpio/vtoy.sh
Normal file
93
新建文件夹 (4)/distros/mkinitcpio/vtoy.sh
Normal file
@@ -0,0 +1,93 @@
|
||||
#!/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
|
||||
|
||||
vtoy_clean_env() {
|
||||
rm -f /sbin/vtoydump /sbin/vtoypartx /sbin/vtoytool /sbin/vtoydmpatch /sbin/vtoydrivers
|
||||
rm -f /usr/lib/initcpio/hooks/ventoy
|
||||
rm -f /usr/lib/initcpio/install/ventoy
|
||||
}
|
||||
|
||||
vtoy_clean_env
|
||||
|
||||
cp -a $vtdumpcmd /sbin/vtoydump
|
||||
cp -a $partxcmd /sbin/vtoypartx
|
||||
cp -a $vtoytool /sbin/vtoytool
|
||||
cat /sbin/vtoydump $dmpatchko > /sbin/vtoydmpatch
|
||||
cp -a ./tools/vtoydrivers /sbin/vtoydrivers
|
||||
cp -a ./distros/$initrdtool/ventoy-install.sh /usr/lib/initcpio/install/ventoy
|
||||
cp -a ./distros/$initrdtool/ventoy-hook.sh /usr/lib/initcpio/hooks/ventoy
|
||||
|
||||
echo "updating the initramfs, please wait ..."
|
||||
|
||||
if ! grep -q '^HOOKS=.*ventoy' /etc/mkinitcpio.conf; then
|
||||
if grep -q '^HOOKS=.*lvm' /etc/mkinitcpio.conf; then
|
||||
exthook='ventoy'
|
||||
else
|
||||
exthook='lvm2 ventoy'
|
||||
fi
|
||||
|
||||
if grep -q '^HOOKS=.*encrypt' /etc/mkinitcpio.conf; then
|
||||
sed "s/\(^HOOKS=.*\)encrypt/\1 $exthook encrypt/" -i /etc/mkinitcpio.conf
|
||||
elif grep -q "^HOOKS=\"" /etc/mkinitcpio.conf; then
|
||||
sed "s/^HOOKS=\"\(.*\)\"/HOOKS=\"\1 $exthook\"/" -i /etc/mkinitcpio.conf
|
||||
elif grep -q "^HOOKS=(" /etc/mkinitcpio.conf; then
|
||||
sed "s/^HOOKS=(\(.*\))/HOOKS=(\1 $exthook)/" -i /etc/mkinitcpio.conf
|
||||
fi
|
||||
fi
|
||||
|
||||
mkinitcpio -P
|
||||
|
||||
disable_grub_os_probe
|
||||
|
||||
#wrapper grub-probe
|
||||
echo "grub mkconfig ..."
|
||||
PROBE_PATH=$(find_grub_probe_path)
|
||||
MKCONFIG_PATH=$(find_grub_mkconfig_path)
|
||||
echo "PROBE_PATH=$PROBE_PATH MKCONFIG_PATH=$MKCONFIG_PATH"
|
||||
|
||||
if [ -e "$PROBE_PATH" -a -e "$MKCONFIG_PATH" ]; then
|
||||
wrapper_grub_probe $PROBE_PATH
|
||||
|
||||
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
|
||||
fi
|
||||
Reference in New Issue
Block a user