add a iommu device in the automated VM deployment and set memballon to disable

Only allow selection of files that starts with virtual (this script is only used to deploy virtual machine)

Reworked some comments

README.md

@@ -17,7 +17,7 @@ chmod +x deploy-vm.sh
 
 Executing: ./scripts/core-count.sh
 System has more than 2 core (nproc --all: 6).
-[...]]
+[...]
 10. virtual-desktop-hypervisor
 [...]
 Enter the number of the file you want to select: 10

scripts/core-count.sh

@@ -4,9 +4,7 @@
 core_count=$(nproc --all)
 
 # Check if nproc --all returns a numerical value greater than 2
-if (( core_count > 2 )); then
- echo "System has more than 2 core (nproc --all: $core_count)."
-else
+if (( core_count < 2 )); then
   echo "Warning: System has only $core_count core)."
   echo "The script requires at least four cores"
   exit 1  # Exit with an error code to indicate the condition is not met

scripts/deploy-distro.sh

@@ -5,17 +5,17 @@ DEFAULT_MEMORY=4096
 DEFAULT_DISK_SIZE=10
 
 # Prompt user for VM memory size
-read -r -p "Provide memory desired VM memory in MB or press Enter to keep default value of $DEFAULT_MEMORY): " memory_size
+read -r -p "Provide desired VM memory in MB or press Enter to keep default value of $DEFAULT_MEMORY MB): " memory_size
 memory_size=${memory_size:-$DEFAULT_MEMORY}
 
 # Validate memory size
 if ! [[ "$memory_size" =~ ^[0-9]+$ ]] || (( memory_size < 2048 )); then
-    echo "Invalid memory size.  Must be a number greater than or equal to 2048. Using default value of $DEFAULT_MEMORY."
+    echo "Invalid memory size.  Must be a number greater than or equal to 2048. Using default value of $DEFAULT_MEMORY MB."
     memory_size=$DEFAULT_MEMORY
 fi
 
 # Prompt user for VM disk size
-read -r -p "Provide desired disk size of VM in GB or press Enter to use default disk size of $DEFAULT_DISK_SIZE: " disk_size
+read -r -p "Provide desired disk size of VM in GB or press Enter to use default disk size of $DEFAULT_DISK_SIZE GB: " disk_size
 disk_size=${disk_size:-$DEFAULT_DISK_SIZE}
 
 # Validate disk size
@@ -24,44 +24,52 @@ if ! [[ "$disk_size" =~ ^[0-9]+$ ]] || (( disk_size < 10 )); then
     disk_size=$DEFAULT_DISK_SIZE
 fi
 
-# Define URI options for qemu
-uri_options=("qemu:///system" "qemu:///session")
+# Set the choices
+CHOICE_SYSTEM="qemu:///system"
+CHOICE_SESSION="qemu:///session"
 
-# Prompt user to select URI
-select uri in "${uri_options[@]}"; do
-    if [[ -n "$uri" ]]; then
-        break  # Exit the select loop if a valid option is chosen
+# Display the choices to the user
+echo "Please select an option or press Enter to keep default value of $CHOICE_SESSION):"
+echo "1) $CHOICE_SYSTEM (system-based or rootfull virtual machine)"
+echo "2) $CHOICE_SESSION (session-based or rootless virtual machine)"
+
+# Prompt the user for input
+IFS= read -r -p "Enter your choice (1 or 2): " user_choice
+
+# Validate the user's input
+if [[ ! "$user_choice" =~ ^[12]$ ]]; then
+  echo "Invalid choice.  Defaulting to session-based VM."
+  uri="$CHOICE_SESSION"  # Default to session-based if input is invalid
 else
-        echo "Invalid selection. Please choose a valid URI."
-    fi
-done
-
-case "$uri" in
-    qemu:///session)
-        disk_path="$HOME/.local/share/libvirt/images/"
+  # Determine the selected option
+  case "$user_choice" in
+    1)
+      uri="$CHOICE_SYSTEM"
       ;;
-    qemu:///system)
+    2)
+      uri="$CHOICE_SESSION"
+      ;;
+    *)
+      echo "Unexpected error: Invalid choice. This should not happen due to validation."
+      exit 1
+      ;;
+  esac
+
+  # Conditional variable assignment based on URI
+  if [[ "$uri" == "qemu:///system" ]]; then
     disk_path="/var/lib/libvirt/images/"
-        ;;
-    *)
-        echo "Invalid URI selected. Exiting."
-        exit 1
-esac
-
-case "$uri" in
-    qemu:///session)
-        network_type="user"
-        ;;
-    qemu:///system)
     network_type="default"
-        ;;
-    *)
-        echo "Invalid URI selected. Exiting."
-        exit 1
-esac
+  elif [[ "$uri" == "qemu:///session" ]]; then
+    disk_path="$HOME/.local/share/libvirt/images/"
+    network_type="user"
+  fi
+fi
 
-# Get a list of files in "dishes" directory, restricted to those starting with "virtual"
-mapfile -t dish_name < <(find "dishes/" -maxdepth 1 -type f -printf "%f\n" | sed 's/\.[^.]*$//')
+# Display the selected option (optional)
+echo "You selected: $uri"
+
+# Get a list of files in "dishes" directory
+mapfile -t dish_name < <(find "dishes/" -maxdepth 1 -type f \( -name "virtual*" \) -printf "%f\n" | sed 's/\.[^.]*$//')
 
 # Check if there are any files
 if [ ${#dish_name[@]} -eq 0 ]; then
@@ -114,6 +122,8 @@ virt-install \
     --input type=keyboard,bus=virtio \
     --input type=mouse,bus=virtio \
     --rng /dev/urandom,model=virtio \
+    --iommu model=virtio \
+    --memballoon none \
     --disk path="${disk_path}/${vm_name}.img",format=raw,bus=virtio,cache=writeback,size="$disk_size" \
     --location=https://download.fedoraproject.org/pub/fedora/linux/releases/42/Everything/x86_64/os/ \
     --initrd-inject ./dishes/"$vm_name".cfg \

scripts/system-memory.sh

@@ -9,6 +9,4 @@ total_memory_mb=$(( total_memory / 1024 ))
 if [[ "$total_memory_mb" -lt "4096" ]]; then
     echo "Not enough RAM: The system has only ${total_memory_mb}MiB of RAM, but at least 4096 is required."
     exit 1
-else
-    echo "Sufficient memory available. System has ${total_memory_mb}MiB of RAM."
 fi