blob: c0ab8cbb4996eae87cd0eac9d65013a8919d1917 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
# -----------------------------------------------------------------------------
#
# Copyright (c) 2009..2021 bwLehrpool-Projektteam
#
# This program/file is free software distributed under the GPL version 2.
# See https://www.gnu.org/licenses/old-licenses/gpl-2.0-standalone.html
#
# If you have any feedback please consult https://bwlehrpool.de and
# send your feedback to support@bwlehrpool.de.
#
# General information about bwLehrpool can be found at https://bwlehrpool.de
#
# -----------------------------------------------------------------------------
# Utils and functions to setup PCI passthrough
# -----------------------------------------------------------------------------
# Function to get state of passthrough
# Return : 0 (true) if passthrough is enabled, otherwise 1 (false)
function passthrough_pci_enabled() {
local passthrough_iommu="$(kernel_cmdln_parse_option "iommu")"
local passthrough_pci_ids=($(kernel_cmdln_parse_pci_ids))
if [ "${passthrough_iommu}" == "pt" ] && ["${#passthrough_pci_ids[@]}" -gt 0 ]; then
return 0
else
return 1
fi
}
# Function to check validitiy of the passthrough configuration
# Return : 0 (true) if passthrough is configured properly, otherwise 1 (false)
function passthrough_pci_check() {
local passthrough_iommu_intel="$(kernel_cmdln_parse_option "intel_iommu")"
local passthrough_iommu_amd="$(kernel_cmdln_parse_option "amd_iommu")"
if [ "${passthrough_iommu_intel}" != "on" ] && [ "${passthrough_iommu_amd}" != "on" ]; then
return 1
else
return 0
fi
}
# Function to get PCI address of a PCI device specified by its vendor and product ID
# Parameter 1: vendor and product ID with the following pattern: "<VENDOR ID>:<PRODUCT ID>"
# Return : PCI address of the specified PCI device of form "<PCI DOMAIN>:<PCI BUS>:<PCI DEVICE>.<PCI FUNCTION>"
function passthrough_pci_lookup_address() {
local passthrough_pci_addresses="$(lspci -n -D -d "${1}" | cut -d ' ' -f 1)"
passthrough_pci_addresses=($(echo "${passthrough_pci_addresses}" | tr '\n' ' '))
if [ "${#passthrough_pci_addresses[@]}" -eq 1 ]; then
echo "${passthrough_pci_addresses[0]}"
return 0
else
return 1
fi
}
# Function to setup PCI passthrough
# Return : PCI-IDs and addresses of the specified passthrough PCI devices as tuples of form
# "<VENDOR ID>:<PRODUCT ID>,<PCI DOMAIN>:<PCI BUS>:<PCI DEVICE>.<PCI FUNCTION>"
function passthrough_pci_setup() {
# check if passthrough is enabled
if passthrough_pci_enabled; then
# check if passthrough is configured properly
if ! passthrough_pci_check; then
return 1;
fi
writelog "Passthrough of PCI devices is enabled successfully"
# parse PCI-IDs from the kernel command line
local passthrough_pci_ids=($(kernel_cmdln_parse_pci_ids))
# lookup PCI address of each specified PCI-ID
local device_pci_address
local pt_gpu_pci_ids
for device_vendor_id in "${passthrough_pci_ids[@]}"; do
device_pci_address="$(passthrough_pci_lookup_address "${device_vendor_id}")"
if [ "${?}" -eq 0 ]; then
pt_gpu_pci_ids+=("${device_vendor_id},${device_pci_address}")
else
writelog "Failed to lookup PCI address for '${device_vendor_id}'"
fi
done
echo "${pt_gpu_pci_ids[@]}"
fi
return 0;
}
|
