From 94163632062b29107569c819855ecf046d737574 Mon Sep 17 00:00:00 2001
From: FIFARenderZ <sebastiaandirks@gmail.com>
Date: Tue, 13 Apr 2021 20:28:43 +0200
Subject: [PATCH] Add a vGPU name check script to make using KVM hypervisors
 easier (#28)

---
 scripts/vgpu-name.sh | 68 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 68 insertions(+)
 create mode 100644 scripts/vgpu-name.sh

diff --git a/scripts/vgpu-name.sh b/scripts/vgpu-name.sh
new file mode 100644
index 0000000..42fa226
--- /dev/null
+++ b/scripts/vgpu-name.sh
@@ -0,0 +1,68 @@
+#!/bin/bash
+
+# This file is part of the "vgpu_unlock" project, and is distributed under the MIT License.
+# See the LICENSE file for more details.
+
+help_function()
+{
+	echo "This script checks for the internal names of vGPU profiles."
+	echo
+	echo "Usage: $0 [-h] [-p <[VGPU_NAME]|ALL>]"
+	echo "Options:"
+	echo "-h   Display the help page."
+	echo "-p   Display selected VGPU_NAME or ALL profiles."
+	exit 0;
+}
+
+process_id()
+{
+	echo "$(cat "$1"/name): $(basename "$1" | cut -d '/' -f 8)"
+}
+
+list_all_ids()
+{
+	for gpu in $(lspci -Dd 10de: -s 0 -n | awk '$2 ~ /030[02]/{print $1}'); do
+		if [ -d "/sys/bus/pci/devices/$gpu/mdev_supported_types/" ]; then
+			for gpu_type in /sys/bus/pci/devices/"$gpu"/mdev_supported_types/* ; do
+				process_id "$gpu_type"
+			done
+			exit
+		fi
+	done
+}
+
+get_pciid()
+{
+	for gpu in $(lspci -Dd 10de: -s 0 -n | awk '$2 ~ /030[02]/{print $1}'); do
+		if [ -d "/sys/bus/pci/devices/$gpu/mdev_supported_types/" ]; then
+			if [ ! -z "$(grep -l "$1" /sys/bus/pci/devices/"$gpu"/mdev_supported_types/nvidia-*/name)" ]; then
+				grep -l "$1" /sys/bus/pci/devices/"$gpu"/mdev_supported_types/nvidia-*/name | cut -d '/' -f 8
+				exit
+			fi
+		fi
+	done
+}
+
+while getopts ":p:" option; do
+	case "${option}" in
+		p)
+			p=${OPTARG}
+			;;
+		*)
+			help_function
+			;;
+   esac
+done
+shift $((OPTIND-1))
+
+if [ -z "${p}" ]; then
+	help_function
+fi
+
+if [ "$p" = "ALL" ]; then
+	list_all_ids
+else
+	get_pciid "$p"
+fi
+
+exit 1