sq

Signed-off-by: Egor Lazarchuk <yegorlz@amazon.co.uk>

Author: ShadowCurse

tools/devtool

@@ -713,6 +713,12 @@ downgrade_to_shared_kvm_lock() {
     echo "Downgraded to shared KVM lock"
 }
 
+# Get the reference count of a kernel module from /proc/modules.
+# Returns 0 if the module is not loaded.
+get_kvm_refcount() {
+    awk -v mod="$1" '$1 == mod {print $3; found=1} END {if (!found) print 0}' /proc/modules
+}
+
 # Reload KVM modules with the given vendor module and kvm params.
 # Always enables avic=1 on AMD. Unloads first if already loaded.
 # Usage: reload_kvm_modules <vendor_mod> [kvm_param...]
@@ -814,16 +820,17 @@ setup_kvm() {
     fi
 
     if [[ $need_kvm_reload -eq 1 ]]; then
-        echo "Reloading KVM modules"
-        upgrade_to_exclusive_kvm_lock
-        if ! reload_kvm_modules "$vendor_mod" "${kvm_extra_params[@]}"; then
-            if [[ -c /dev/kvm ]]; then
-                say_warn "Could not reload KVM modules (in use), continuing with current parameters"
-            else
-                die "Could not reload KVM modules and /dev/kvm not found"
-            fi
+        local refcount
+        refcount=$(get_kvm_refcount "$vendor_mod")
+        if [[ "$refcount" -gt 0 ]]; then
+            say_warn "KVM module $vendor_mod is in use (refcount=$refcount), skipping reload"
+        else
+            echo "Reloading KVM modules"
+            upgrade_to_exclusive_kvm_lock
+            reload_kvm_modules "$vendor_mod" "${kvm_extra_params[@]}"
+            ok_or_die "Could not reload kvm modules"
+            downgrade_to_shared_kvm_lock
         fi
-        downgrade_to_shared_kvm_lock
     fi
 
     tail -v $itlb_multihit $nx_huge_pages