initial commit to prepare for rewrite into golang for actual use
This commit is contained in:
HikariKnight
parent
0d4f63f371
commit
5da1a88296
21 changed files with 129 additions and 1 deletions
|
|
@ -1,188 +0,0 @@
|
|||
#!/bin/bash
|
||||
# shellcheck disable=SC1091
|
||||
|
||||
# Function to configure systemd-boot using kernelstub
|
||||
function set_KERNELSTUB () {
|
||||
# Separator
|
||||
printf "
|
||||
############################################################
|
||||
"
|
||||
|
||||
# Tell what we are going to do
|
||||
echo "Adding vfio kernel arguments to systemd-boot using kernelstub"
|
||||
|
||||
# Get the config paths
|
||||
source "${SCRIPTDIR}/lib/paths.sh"
|
||||
|
||||
# Check if systemd-boot already has vfio parameters from before
|
||||
KERNELSTUB_TEST=$(sudo kernelstub -p 2>&1 | grep "Kernel Boot Options" | perl -pe "s/.+Kernel Boot Options:\..+(vfio_pci.ids=.+ ).+/\1/")
|
||||
|
||||
# If there are already vfio_pci parameters in kernelstub
|
||||
if [[ "$KERNELSTUB_TEST" =~ vfio_pci.ids ]] ;
|
||||
then
|
||||
# Remove the old parameters
|
||||
sudo kernelstub -d "$KERNELSTUB_TEST"
|
||||
sudo kernelstub -d "vfio_pci.disable_vga=1"
|
||||
sudo kernelstub -d "vfio_pci.disable_vga=0"
|
||||
fi
|
||||
|
||||
# Apply new parameters
|
||||
CMDLINE=$(cat "${SCRIPTDIR}/config/kernel_args")
|
||||
sudo kernelstub -a "$CMDLINE"
|
||||
}
|
||||
|
||||
# Function to configure grub
|
||||
function set_GRUB () {
|
||||
# Separator
|
||||
printf "
|
||||
############################################################
|
||||
"
|
||||
|
||||
# Get the config paths
|
||||
source "$SCRIPTDIR/lib/paths.sh"
|
||||
|
||||
local CMDLINE
|
||||
CMDLINE=$(cat "$SCRIPTDIR/config/kernel_args")
|
||||
|
||||
# HIGHLY EXPERIMENTAL!
|
||||
local GRUB_CMDLINE
|
||||
local GRUB_CMDLINE_LINUX
|
||||
|
||||
# Check if there is a GRUB_CMDLINE_LINUX_DEFAULT line in grub config
|
||||
if grep -q "GRUB_CMDLINE_LINUX_DEFAULT=" "$SCRIPTDIR/$DEFAULT/grub" ;
|
||||
then
|
||||
# Update the GRUB_CMDLINE_LINUX_DEFAULT line
|
||||
GRUB_CMDLINE=$(grep -P "^GRUB_CMDLINE_LINUX_DEFAULT" "/etc/default/grub" | perl -pe "s/GRUB_CMDLINE_LINUX_DEFAULT=\"(.+)\"/\1/" | perl -pe "s/(amd|intel)_iommu=(on|1)|iommu=(pt|on)|vfio_pci.ids=.+|vfio_pci.disable_vga=\d{1}//g" | perl -pe "s/(^\s+|\s+$)//g")
|
||||
GRUB_CMDLINE_LINUX=$(grep -P "^GRUB_CMDLINE_LINUX_DEFAULT" "/etc/default/grub")
|
||||
perl -pi -e "s/${GRUB_CMDLINE_LINUX}/GRUB_CMDLINE_LINUX_DEFAULT=\"${GRUB_CMDLINE} ${CMDLINE}\"/" "${SCRIPTDIR}/$DEFAULT/grub"
|
||||
else
|
||||
# Update the GRUB_CMDLINE_LINUX line
|
||||
GRUB_CMDLINE=$(grep -P "^GRUB_CMDLINE_LINUX" "/etc/default/grub" | perl -pe "s/GRUB_CMDLINE_LINUX=\"(.+)\"/\1/" | perl -pe "s/(amd|intel)_iommu=(on|1)|iommu=(pt|on)|vfio_pci.ids=.+|vfio_pci.disable_vga=\d{1}//g" | perl -pe "s/(^\s+|\s+$)//g")
|
||||
GRUB_CMDLINE_LINUX=$(grep -P "^GRUB_CMDLINE_LINUX" "/etc/default/grub")
|
||||
perl -pi -e "s/${GRUB_CMDLINE_LINUX}/GRUB_CMDLINE_LINUX=\"${GRUB_CMDLINE} ${CMDLINE}\"/" "${SCRIPTDIR}/$DEFAULT/grub"
|
||||
fi
|
||||
|
||||
|
||||
echo "The script will now replace your default grub file with a new one.
|
||||
Then attempt to update grub and generate a new grub.cfg.
|
||||
If generating the grub.cfg file fails, you can find a backup of your grub default file here:
|
||||
$SCRIPTDIR/backup/etc/default/grub
|
||||
"
|
||||
read -r -p "Press ENTER to continue"
|
||||
|
||||
sudo cp -v "$SCRIPTDIR/$DEFAULT/grub" "/etc/default/grub"
|
||||
|
||||
# Generate grub.cfg
|
||||
if [ -d "/boot/grub" ];
|
||||
then
|
||||
sudo grub-mkconfig -o "/boot/grub/grub.cfg"
|
||||
else
|
||||
sudo grub-mkconfig -o "/boot/grub2/grub.cfg"
|
||||
fi
|
||||
|
||||
echo ""
|
||||
read -r -p "Please verify there were no errors generating the grub.cfg file, then press ENTER"
|
||||
}
|
||||
|
||||
function show_FINISH () {
|
||||
# Separator
|
||||
printf "
|
||||
############################################################
|
||||
"
|
||||
|
||||
# Get the config paths
|
||||
source "$SCRIPTDIR/lib/paths.sh"
|
||||
|
||||
local CMDLINE
|
||||
CMDLINE=$(cat "$SCRIPTDIR/config/kernel_args")
|
||||
|
||||
echo "Configuration is now complete!"
|
||||
|
||||
if [ "$1" == 0 ];
|
||||
then
|
||||
printf "For VFIO to work properly you need to make sure these kernel parameters are in your bootloader entry:
|
||||
#-----------------------------------------------#
|
||||
%s
|
||||
#-----------------------------------------------#
|
||||
|
||||
" "$CMDLINE"
|
||||
fi
|
||||
|
||||
echo "Restart your system and run
|
||||
\"$SCRIPTDIR/vfio-verify\"
|
||||
to check if your GPU is properly set up.
|
||||
|
||||
If the graphic card is bound to vfio-pci then you can
|
||||
proceed to add it to your virtual machines.
|
||||
|
||||
A backup the files we replaced on your system can be found inside
|
||||
$SCRIPTDIR/backup/
|
||||
In order to restore these files just copy them back to your system and
|
||||
rebuild your initramfs image.
|
||||
|
||||
You can remove the the vfio_pci kernel arguments from the linux line in your bootloader
|
||||
to disable/unbind the graphic card from the vfio driver on boot.
|
||||
|
||||
The files inside \"$SCRIPTDIR/$QUICKEMU\" are currently unused files, however they provide
|
||||
the required information that the QuickEMU project can hook into and use to add support for VFIO enabled VMs.
|
||||
|
||||
The PCI Devices with these IDs are what you should add to your VMs using Virt Manager:
|
||||
NOTE: Some AMD GPUs will require the vendor-reset kernel module from https://github.com/gnif/vendor-reset to be installed!"
|
||||
|
||||
source "${SCRIPTDIR}/config/quickemu/qemu-vfio_vars.conf"
|
||||
|
||||
for dev in "${GPU_PCI_ID[@]}"
|
||||
do
|
||||
echo "* $dev"
|
||||
done
|
||||
for dev in "${USB_CTL_ID[@]}"
|
||||
do
|
||||
echo "* $dev"
|
||||
done
|
||||
|
||||
echo "
|
||||
To add the graphic card to your VM using qemu directly, use the following arguments:"
|
||||
for dev in "${GPU_PCI_ID[@]}"
|
||||
do
|
||||
|
||||
echo -n "-device vfio-pci,host=$dev "
|
||||
done
|
||||
printf "\n"
|
||||
|
||||
echo "
|
||||
For performance tuning and advanced configuration look at:
|
||||
https://github.com/HikariKnight/vfio-setup-docs/wiki"
|
||||
}
|
||||
|
||||
function set_CMDLINE () {
|
||||
# Make a variable to tell if
|
||||
local BOOTLOADER_AUTOCONFIG
|
||||
BOOTLOADER_AUTOCONFIG=0
|
||||
|
||||
# If kernelstub is detected (program to manage systemd-boot)
|
||||
if which kernelstub > /dev/null 2>&1 ;
|
||||
then
|
||||
# Configure kernelstub
|
||||
set_KERNELSTUB
|
||||
BOOTLOADER_AUTOCONFIG=1
|
||||
fi
|
||||
|
||||
# If grub exists
|
||||
if which grub-mkconfig > /dev/null 2>&1 ;
|
||||
then
|
||||
# Configure grub
|
||||
set_GRUB
|
||||
BOOTLOADER_AUTOCONFIG=1
|
||||
fi
|
||||
|
||||
show_FINISH $BOOTLOADER_AUTOCONFIG
|
||||
}
|
||||
|
||||
|
||||
function main () {
|
||||
SCRIPTDIR=$(dirname "$(realpath "$0")" | perl -pe "s/\/\.\.\/lib//" | perl -pe "s/\/lib$//")
|
||||
|
||||
set_CMDLINE
|
||||
}
|
||||
|
||||
main
|
||||
Loading…
Add table
Add a link
Reference in a new issue