enable test support for mkinitcpio and dracut

lib/apply_CHANGES.sh

@@ -36,6 +36,8 @@ function make_BACKUP () {
 
         fi
 
+        if [ -d "/etc/modprobe.d" ];
+        then
             mkdir -p "$BACKUPDIR/etc/modprobe.d"
 
             # If a vfio.conf file exists, backup that too
@@ -45,6 +47,8 @@ function make_BACKUP () {
 
             fi
 
+        fi
+
         printf "Backup completed!\n"
 
     else
@@ -79,6 +83,13 @@ Rebuilding initramfs"
 Rebuilding initramfs"
         sudo dracut -f -kver "$(uname -r)"
 
+    elif [ -f "/etc/mkinitcpio.conf" ];
+    then
+        cp -v "$SCRIPTDIR/$MKINITCPIO" "/etc/mkinitcpio.conf"
+        echo "
+Rebuilding initramfs"
+        sudo mkinitcpio -P
+
     else
         echo "
 Unsupported initramfs infrastructure

lib/get_GPU_GROUP.sh

@@ -49,7 +49,24 @@ USB_CTL_ID=()
             "$SCRIPTDIR/lib/get_GPU_ROM.sh" "$ROM_PCI_ID"
 
             # Start setting up modules
+            if [ -d "/etc/initramfs-tools" ];
+            then
                 exec "$SCRIPTDIR/lib/set_INITRAMFSTOOLS.sh" "$GPU_DEVID"
+            
+            elif [ -d "/etc/dracut.conf" ];
+            then
+                exec "$SCRIPTDIR/lib/set_DRACUT.sh" "$GPU_DEVID"
+            
+            elif [ -f "/etc/mkinitcpio.conf" ];
+            then
+                exec "$SCRIPTDIR/lib/set_MKINITCPIO.sh" "$GPU_DEVID"
+            else
+                # Bind GPU to VFIO
+                "$SCRIPTDIR/lib/set_VFIO.sh" "$1"
+
+                # Configure modprobe
+                "$SCRIPTDIR/lib/set_MODPROBE.sh" "$1"
+            fi
         ;;
         *)
             exec "$SCRIPTDIR/lib/get_GPU.sh"

lib/set_DRACUT.sh

@@ -16,9 +16,7 @@ function set_DRACUT () {
     "$SCRIPTDIR/lib/set_VFIO.sh" "$1"
 
     # Configure modprobe
-    "$SCRIPTDIR/lib/set_MODPROBE.sh" "$1"
-
-    exec "$SCRIPTDIR/lib/get_USB_CTL.sh"
+    exec "$SCRIPTDIR/lib/set_MODPROBE.sh" "$1"
 }
 
 function main () {

lib/set_INITRAMFSTOOLS.sh

@@ -52,9 +52,7 @@ function set_INITRAMFSTOOLS () {
     "$SCRIPTDIR/lib/set_VFIO.sh" "$1"
 
     # Configure modprobe
-    "$SCRIPTDIR/lib/set_MODPROBE.sh" "$1"
-
-    exec "$SCRIPTDIR/lib/get_USB_CTL.sh"
+    exec "$SCRIPTDIR/lib/set_MODPROBE.sh" "$1"
 }
 
 

lib/set_MKINITCPIO.sh

@@ -17,6 +17,11 @@ function set_MKINITCPIO () {
         perl -pi -e "s/MODULES=\(${CURRENTMODULES}\)/MODULES=\(vfio vfio_iommu_type1 vfio_pci vfio_virqfd ${MODULES}\)/" "$SCRIPTDIR/$MKINITCPIO"
     fi
 
+    # Bind GPU to VFIO
+    "$SCRIPTDIR/lib/set_VFIO.sh" "$1"
+
+    # Configure modprobe
+    exec "$SCRIPTDIR/lib/set_MODPROBE.sh" "$1"
 }
 
 function main () {

lib/set_MODPROBE.sh

@@ -26,6 +26,8 @@ softdep nouveau pre: vfio vfio_pci
 softdep amdgpu pre: vfio vfio_pci
 softdep radeon pre: vfio vfio_pci
 "  "${GPU_DEVID}"  > "$SCRIPTDIR/$MODPROBE/vfio.conf"
+
+    exec "$SCRIPTDIR/lib/get_USB_CTL.sh"
 }
 
 function main () {