scripts: refactor prepare_vdpa for distro-specific setup

Split the vDPA preparation flow into helper functions for
building modules, validating availability, loading modules,
and creating devices.

Build the vdpa_sim modules only on Ubuntu, where the script
installs dependencies and compiles them from the matching
kernel source. On other distributions, reuse the installed
kernel modules and verify that they are available before
continuing.

This makes the script easier to follow and supports systems
such as Azure Linux, where the modules are provided by the
kernel package.

Signed-off-by: Muminul Islam <muislam@microsoft.com>
This commit is contained in:
Muminul Islam
2026-03-13 18:15:12 -07:00
committed by Rob Bradford
parent 8b60b38281
commit c53bc3d170

View File

@@ -1,35 +1,73 @@
#!/usr/bin/env bash #!/usr/bin/env bash
set -x set -x
sudo apt install -y libncurses-dev gawk flex bison openssl libssl-dev dkms libelf-dev libudev-dev libpci-dev libiberty-dev autoconf git make dpkg-dev libmnl-dev pkg-config iproute2 build_install_vdpa_sim_modules_ubuntu() {
sudo sed -i -- 's/# deb-src/deb-src/g' /etc/apt/sources.list sudo apt install -y libncurses-dev gawk flex bison openssl libssl-dev dkms libelf-dev libudev-dev libpci-dev libiberty-dev autoconf git make dpkg-dev libmnl-dev pkg-config iproute2
sudo apt update sudo sed -i -- 's/# deb-src/deb-src/g' /etc/apt/sources.list
apt-get source linux-image-unsigned-"$(uname -r)" sudo apt update
pushd linux-azure*/drivers/vdpa/vdpa_sim/ || exit apt-get source linux-image-unsigned-"$(uname -r)"
# REUSE-IgnoreStart pushd linux-azure*/drivers/vdpa/vdpa_sim/ || exit
cat <<'EOF' >Makefile # REUSE-IgnoreStart
cat <<'EOF' >Makefile
# SPDX-License-Identifier: GPL-2.0 # SPDX-License-Identifier: GPL-2.0
obj-m += vdpa_sim.o obj-m += vdpa_sim.o
obj-m += vdpa_sim_net.o obj-m += vdpa_sim_net.o
obj-m += vdpa_sim_blk.o obj-m += vdpa_sim_blk.o
EOF EOF
# REUSE-IgnoreEnd # REUSE-IgnoreEnd
make -C /lib/modules/"$(uname -r)"/build M="$PWD" make -C /lib/modules/"$(uname -r)"/build M="$PWD"
sudo make -C /lib/modules/"$(uname -r)"/build M="$PWD" modules_install sudo make -C /lib/modules/"$(uname -r)"/build M="$PWD" modules_install
popd || exit popd || exit
sudo depmod -a sudo depmod -a
sudo modprobe vdpa }
sudo modprobe vhost_vdpa
sudo modprobe vdpa_sim check_vdpa_sim_modules() {
sudo modprobe vdpa_sim_blk for module in $MODULES; do
sudo modprobe vdpa_sim_net modinfo "$module" || {
# Create /dev/vhost-vdpa-0 echo "Module $module is not installed. Please build and install it first."
sudo vdpa dev add name vdpa-blk1 mgmtdev vdpasim_blk exit 1
# Create /dev/vhost-vdpa-1 }
sudo vdpa dev add name vdpa-blk2 mgmtdev vdpasim_blk done
# Create /dev/vhost-vdpa-2 }
sudo vdpa dev add name vdpa-net1 mgmtdev vdpasim_net
sudo chmod 660 /dev/vhost-vdpa-0 modproobe_modules() {
sudo chmod 660 /dev/vhost-vdpa-1 for module in $MODULES; do
sudo chmod 660 /dev/vhost-vdpa-2 sudo modprobe "$module" || {
vdpa dev show -jp echo "Failed to load module $module. Please check if it is installed correctly."
exit 1
}
done
}
prepare_vdpa() {
# Create /dev/vhost-vdpa-0
sudo vdpa dev add name vdpa-blk1 mgmtdev vdpasim_blk
# Create /dev/vhost-vdpa-1
sudo vdpa dev add name vdpa-blk2 mgmtdev vdpasim_blk
# Create /dev/vhost-vdpa-2
sudo vdpa dev add name vdpa-net1 mgmtdev vdpasim_net
sudo chmod 660 /dev/vhost-vdpa-0
sudo chmod 660 /dev/vhost-vdpa-1
sudo chmod 660 /dev/vhost-vdpa-2
vdpa dev show -jp
}
MODULES="vdpa vhost_vdpa vdpa_sim vdpa_sim_blk vdpa_sim_net"
DISTRO_NAME="ubuntu"
if [[ -f /etc/lsb-release ]]; then
DISTRO_NAME=$(grep DISTRIB_ID /etc/lsb-release | cut -d '=' -f 2)
# Converts the value of the DISTRO_NAME variable to lowercase letters.
DISTRO_NAME=$(echo "$DISTRO_NAME" | tr '[:upper:]' '[:lower:]')
echo "Distribution Name: $DISTRO_NAME"
fi
if [[ "$DISTRO_NAME" == "ubuntu" ]]; then
build_install_vdpa_sim_modules_ubuntu
fi
# For other distros, we assume the modules are already built and installed
# For Azure Linux, the modules are included in the kernel and should be available by default
check_vdpa_sim_modules
modproobe_modules
prepare_vdpa