add vbios extration page

internal/pages/02_select_gpu.go

@@ -86,6 +86,7 @@ func viewGPU(id string, ext ...int) {
 
 	case "y":
 		// Go to the select a usb controller
-		selectUSB()
+		//selectUSB()
+		genVBIOS_dumper(id)
 	}
 }

internal/pages/03_vbios_extract.go

@@ -0,0 +1,50 @@
+package pages
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"github.com/HikariKnight/quickpassthrough/pkg/command"
+	"github.com/HikariKnight/quickpassthrough/pkg/menu"
+)
+
+func genVBIOS_dumper(id string) {
+	// Clear the scren
+	command.Clear()
+
+	// Get the program directory
+	exe, _ := os.Executable()
+	scriptdir := filepath.Dir(exe)
+
+	// If we are using go run use the working directory instead
+	if strings.Contains(scriptdir, "/tmp/go-build") {
+		scriptdir, _ = os.Getwd()
+	}
+
+	// Get the vbios path
+	//vbios_path := lsiommu.GetIOMMU("-g", "-i", id, "--rom")[0]
+
+	// Tell users about the VBIOS dumper script
+	fmt.Print(
+		"For some GPUs, you will need to dump the VBIOS and pass the\n",
+		"rom to the VM along with the card in order to get a functional passthrough.\n",
+		"In many cases you can find your vbios at https://www.techpowerup.com/vgabios/\n",
+		"\n",
+		"You can also attempt to dump your own vbios using the script in\n",
+		fmt.Sprintf("%s/utils/dump_vbios.sh\n", scriptdir),
+		"\n",
+	)
+
+	// Get the OK press
+	choice := menu.Ok("Make sure you run the script with the display-manager stopped using ssh or tty!")
+
+	// If OK is pressed
+	if choice == "next" {
+		selectUSB()
+	} else {
+		fmt.Println("")
+		os.Exit(0)
+	}
+}

pkg/menu/ok.go

@@ -0,0 +1,16 @@
+package menu
+
+import "github.com/nexidian/gocliselect"
+
+// Make a YesNo menu
+func Ok(msg string) string {
+	// Make the menu
+	menu := gocliselect.NewMenu(msg)
+	menu.AddItem("OK", "next")
+
+	// Display the menu
+	choice := menu.Display()
+
+	// Return the value selected
+	return choice
+}