From 38063cef55b9c3c47b2ead6258b0261ae7e62502 Mon Sep 17 00:00:00 2001
From: Kroese <kroese@users.noreply.github.com>
Date: Wed, 22 May 2024 21:02:36 +0200
Subject: [PATCH] Update disk.sh

---
 src/disk.sh | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/disk.sh b/src/disk.sh
index 9fd45f8..341eb6e 100644
--- a/src/disk.sh
+++ b/src/disk.sh
@@ -514,8 +514,17 @@ case "${DISK_TYPE,,}" in
   * ) error "Invalid DISK_TYPE, value \"$DISK_TYPE\" is unrecognized!" && exit 80 ;;
 esac
 
+DRIVER_TYPE="ide"
 MEDIA_TYPE="$DISK_TYPE"
 
+case "${MACHINE,,}" in
+  "virt" )
+    DRIVER_TYPE="usb" ;;
+  "pc-q35-2"* )
+    DISK_TYPE="blk"
+    MEDIA_TYPE="ide" ;;
+esac
+
 if [ -f "$BOOT" ] && [ -s "$BOOT" ]; then
   DISK_OPTS=$(addMedia "$BOOT" "$MEDIA_TYPE" "0" "$BOOT_INDEX" "0x5")
 fi