mirror of
https://github.com/MacRimi/ProxMenux.git
synced 2026-04-25 08:56:21 +00:00
update pci_passthrough_helpers.sh
This commit is contained in:
52
scripts/global/pci_passthrough_helpers.sh
Normal file
52
scripts/global/pci_passthrough_helpers.sh
Normal file
@@ -0,0 +1,52 @@
|
||||
#!/usr/bin/env bash
|
||||
|
||||
if [[ -n "${__PROXMENUX_PCI_PASSTHROUGH_HELPERS__}" ]]; then
|
||||
return 0
|
||||
fi
|
||||
__PROXMENUX_PCI_PASSTHROUGH_HELPERS__=1
|
||||
|
||||
function _pci_is_iommu_active() {
|
||||
grep -qE 'intel_iommu=on|amd_iommu=on' /proc/cmdline 2>/dev/null || return 1
|
||||
[[ -d /sys/kernel/iommu_groups ]] || return 1
|
||||
find /sys/kernel/iommu_groups -mindepth 1 -maxdepth 1 -type d -print -quit 2>/dev/null | grep -q .
|
||||
}
|
||||
|
||||
function _pci_next_hostpci_index() {
|
||||
local vmid="$1"
|
||||
local idx=0
|
||||
local hostpci_existing
|
||||
|
||||
hostpci_existing=$(qm config "$vmid" 2>/dev/null) || return 1
|
||||
while grep -q "^hostpci${idx}:" <<< "$hostpci_existing"; do
|
||||
idx=$((idx + 1))
|
||||
done
|
||||
echo "$idx"
|
||||
}
|
||||
|
||||
function _pci_slot_assigned_to_vm() {
|
||||
local pci_full="$1"
|
||||
local vmid="$2"
|
||||
local slot_base
|
||||
slot_base="${pci_full#0000:}"
|
||||
slot_base="${slot_base%.*}"
|
||||
|
||||
qm config "$vmid" 2>/dev/null \
|
||||
| grep -qE "^hostpci[0-9]+:.*(0000:)?${slot_base}(\\.[0-7])?([,[:space:]]|$)"
|
||||
}
|
||||
|
||||
function _pci_function_assigned_to_vm() {
|
||||
local pci_full="$1"
|
||||
local vmid="$2"
|
||||
local bdf slot func pattern
|
||||
bdf="${pci_full#0000:}"
|
||||
slot="${bdf%.*}"
|
||||
func="${bdf##*.}"
|
||||
|
||||
if [[ "$func" == "0" ]]; then
|
||||
pattern="^hostpci[0-9]+:.*(0000:)?(${bdf}|${slot})([,:[:space:]]|$)"
|
||||
else
|
||||
pattern="^hostpci[0-9]+:.*(0000:)?${bdf}([,[:space:]]|$)"
|
||||
fi
|
||||
|
||||
qm config "$vmid" 2>/dev/null | grep -qE "$pattern"
|
||||
}
|
||||
138
scripts/global/vm_storage_helpers.sh
Normal file
138
scripts/global/vm_storage_helpers.sh
Normal file
@@ -0,0 +1,138 @@
|
||||
#!/usr/bin/env bash
|
||||
|
||||
if [[ -n "${__PROXMENUX_VM_STORAGE_HELPERS__}" ]]; then
|
||||
return 0
|
||||
fi
|
||||
__PROXMENUX_VM_STORAGE_HELPERS__=1
|
||||
|
||||
function _array_contains() {
|
||||
local needle="$1"
|
||||
shift
|
||||
local item
|
||||
for item in "$@"; do
|
||||
[[ "$item" == "$needle" ]] && return 0
|
||||
done
|
||||
return 1
|
||||
}
|
||||
|
||||
function _refresh_host_storage_cache() {
|
||||
MOUNTED_DISKS=$(lsblk -ln -o NAME,MOUNTPOINT | awk '$2!="" {print "/dev/" $1}')
|
||||
SWAP_DISKS=$(swapon --noheadings --raw --show=NAME 2>/dev/null)
|
||||
LVM_DEVICES=$(pvs --noheadings -o pv_name 2> >(grep -v 'File descriptor .* leaked') | xargs -r -n1 readlink -f | sort -u)
|
||||
CONFIG_DATA=$(grep -vE '^\s*#' /etc/pve/qemu-server/*.conf /etc/pve/lxc/*.conf 2>/dev/null)
|
||||
|
||||
ZFS_DISKS=""
|
||||
local zfs_raw entry path base_disk
|
||||
zfs_raw=$(zpool list -v -H 2>/dev/null | awk '{print $1}' | grep -v '^NAME$' | grep -v '^-' | grep -v '^mirror')
|
||||
for entry in $zfs_raw; do
|
||||
path=""
|
||||
if [[ "$entry" == wwn-* || "$entry" == ata-* ]]; then
|
||||
[[ -e "/dev/disk/by-id/$entry" ]] && path=$(readlink -f "/dev/disk/by-id/$entry")
|
||||
elif [[ "$entry" == /dev/* ]]; then
|
||||
path="$entry"
|
||||
fi
|
||||
if [[ -n "$path" ]]; then
|
||||
base_disk=$(lsblk -no PKNAME "$path" 2>/dev/null)
|
||||
[[ -n "$base_disk" ]] && ZFS_DISKS+="/dev/$base_disk"$'\n'
|
||||
fi
|
||||
done
|
||||
ZFS_DISKS=$(echo "$ZFS_DISKS" | sort -u)
|
||||
}
|
||||
|
||||
function _disk_is_host_system_used() {
|
||||
local disk="$1"
|
||||
local disk_real part fstype part_path
|
||||
DISK_USAGE_REASON=""
|
||||
|
||||
while read -r part fstype; do
|
||||
[[ -z "$part" ]] && continue
|
||||
part_path="/dev/$part"
|
||||
|
||||
if grep -qFx "$part_path" <<< "$MOUNTED_DISKS"; then
|
||||
DISK_USAGE_REASON="$(translate "Mounted filesystem detected") ($part_path)"
|
||||
return 0
|
||||
fi
|
||||
if grep -qFx "$part_path" <<< "$SWAP_DISKS"; then
|
||||
DISK_USAGE_REASON="$(translate "Swap partition detected") ($part_path)"
|
||||
return 0
|
||||
fi
|
||||
case "$fstype" in
|
||||
zfs_member)
|
||||
DISK_USAGE_REASON="$(translate "ZFS member detected") ($part_path)"
|
||||
return 0
|
||||
;;
|
||||
linux_raid_member)
|
||||
DISK_USAGE_REASON="$(translate "RAID member detected") ($part_path)"
|
||||
return 0
|
||||
;;
|
||||
LVM2_member)
|
||||
DISK_USAGE_REASON="$(translate "LVM physical volume detected") ($part_path)"
|
||||
return 0
|
||||
;;
|
||||
esac
|
||||
done < <(lsblk -ln -o NAME,FSTYPE "$disk" 2>/dev/null)
|
||||
|
||||
disk_real=$(readlink -f "$disk" 2>/dev/null)
|
||||
if [[ -n "$disk_real" && -n "$LVM_DEVICES" ]] && grep -qFx "$disk_real" <<< "$LVM_DEVICES"; then
|
||||
DISK_USAGE_REASON="$(translate "Disk is part of host LVM")"
|
||||
return 0
|
||||
fi
|
||||
if [[ -n "$ZFS_DISKS" && "$ZFS_DISKS" == *"$disk"* ]]; then
|
||||
DISK_USAGE_REASON="$(translate "Disk is part of a host ZFS pool")"
|
||||
return 0
|
||||
fi
|
||||
return 1
|
||||
}
|
||||
|
||||
function _disk_used_in_guest_configs() {
|
||||
local disk="$1"
|
||||
local real_path
|
||||
real_path=$(readlink -f "$disk" 2>/dev/null)
|
||||
|
||||
if [[ -n "$real_path" ]] && grep -Fq "$real_path" <<< "$CONFIG_DATA"; then
|
||||
return 0
|
||||
fi
|
||||
|
||||
local symlink
|
||||
for symlink in /dev/disk/by-id/*; do
|
||||
[[ -e "$symlink" ]] || continue
|
||||
if [[ "$(readlink -f "$symlink")" == "$real_path" ]] && grep -Fq "$symlink" <<< "$CONFIG_DATA"; then
|
||||
return 0
|
||||
fi
|
||||
done
|
||||
return 1
|
||||
}
|
||||
|
||||
function _controller_block_devices() {
|
||||
local pci_full="$1"
|
||||
local pci_root="/sys/bus/pci/devices/$pci_full"
|
||||
[[ -d "$pci_root" ]] || return 0
|
||||
|
||||
local sys_block dev_name cur base
|
||||
# Walk /sys/block and resolve each block device back to its ancestor PCI device.
|
||||
# This avoids unbounded recursive scans while still handling NVMe/SATA paths.
|
||||
for sys_block in /sys/block/*; do
|
||||
[[ -e "$sys_block/device" ]] || continue
|
||||
dev_name=$(basename "$sys_block")
|
||||
[[ -b "/dev/$dev_name" ]] || continue
|
||||
|
||||
cur=$(readlink -f "$sys_block/device" 2>/dev/null)
|
||||
[[ -n "$cur" ]] || continue
|
||||
|
||||
while [[ "$cur" != "/" ]]; do
|
||||
base=$(basename "$cur")
|
||||
if [[ "$base" == "$pci_full" ]]; then
|
||||
echo "/dev/$dev_name"
|
||||
break
|
||||
fi
|
||||
cur=$(dirname "$cur")
|
||||
done
|
||||
done
|
||||
}
|
||||
|
||||
function _vm_is_q35() {
|
||||
local vmid="$1"
|
||||
local machine_line
|
||||
machine_line=$(qm config "$vmid" 2>/dev/null | awk -F': ' '/^machine:/ {print $2}')
|
||||
[[ "$machine_line" == *q35* ]]
|
||||
}
|
||||
Reference in New Issue
Block a user