mirror of
https://github.com/SkimrMe/funky_tv
synced 2026-01-29 19:15:23 +00:00
383 lines
11 KiB
Bash
383 lines
11 KiB
Bash
#!/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
|
|
}
|
|
|
|
|