Blob stamping for vfio devices + DRPC

Coopydood · Feb 22, 2024 · 45fbc02 · 45fbc02
1 parent 6621765
commit 45fbc02
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/main.py b/main.py
@@ -337,7 +337,12 @@ def clear(): print("\n" * 150)
             os.system("cp ./resources/ovmf/OVMF_VARS.fd ./ovmf/OVMF_VARS.fd")
 
     if discordRPC == 1:
-        subprocess.Popen(["python","./scripts/drpc.py","--os",macOSVer])
+        if os.path.exists("./blobs/user/USR_VFIO_DEVICES.apb"):
+            vfioDevs = open("./blobs/user/USR_VFIO_DEVICES.apb")
+            vfioDevs = vfioDevs.read()
+            subprocess.Popen(["python","./scripts/drpc.py","--os",macOSVer,"--pt",vfioDevs])
+        else:
+            subprocess.Popen(["python","./scripts/drpc.py","--os",macOSVer])
     if REQUIRES_SUDO == 1:
         print(color.YELLOW+color.BOLD+"\n   ⚠ "+color.END+color.BOLD+"SUPERUSER PRIVILEGES"+color.END+"\n   This script uses physical device passthrough,\n   and needs superuser priviledges to run.\n\n   Press CTRL+C to cancel.\n"+color.END)
         if discordRPC == 0:

diff --git a/scripts/extras/vfio-passthrough.py b/scripts/extras/vfio-passthrough.py
@@ -500,6 +500,9 @@ def autoAPSelect():
 
                             apFileM = apFileM.replace("VFIO_DEVICES="+str(currentAmount),"VFIO_DEVICES="+str(totalVD))
 
+                            blob = open("./blobs/USR_VFIO_DEVICES.apb","w")
+                            blob.write(totalVD)
+                            blob.close()
 
                             apFileM = apFileM.replace("-device qxl-vga,vgamem_mb=128,vram_size_mb=128    ","#-device qxl-vga,vgamem_mb=128,vram_size_mb=128   # DISABLED BY VFIO-PCI PASSTHROUGH ASSISTANT")
                             apFileM = apFileM.replace("/OVMF_VARS.fd","/OVMF_VARS_PT.fd")