mirror of
https://github.com/HikariKnight/quickpassthrough.git
synced 2024-09-19 19:38:45 -04:00
129 lines
3.4 KiB
Go
129 lines
3.4 KiB
Go
package pages
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/HikariKnight/ls-iommu/pkg/errorcheck"
|
|
"github.com/HikariKnight/quickpassthrough/internal/configs"
|
|
lsiommu "github.com/HikariKnight/quickpassthrough/internal/lsiommu"
|
|
"github.com/HikariKnight/quickpassthrough/pkg/command"
|
|
"github.com/HikariKnight/quickpassthrough/pkg/fileio"
|
|
"github.com/HikariKnight/quickpassthrough/pkg/menu"
|
|
"github.com/gookit/color"
|
|
)
|
|
|
|
func SelectGPU(config *configs.Config) {
|
|
// Clear the screen
|
|
command.Clear()
|
|
|
|
// Get the users GPUs
|
|
gpus := lsiommu.GetIOMMU("-g", "-F", "vendor:,prod_name,optional_revision:,device_id")
|
|
|
|
// Generate a list of choices based on the GPUs and get the users selection
|
|
choice := menu.GenIOMMUMenu("Select a GPU to view the IOMMU groups of", gpus)
|
|
|
|
// Parse the choice
|
|
switch choice {
|
|
case "back":
|
|
Welcome()
|
|
case "":
|
|
// If ESC is pressed
|
|
fmt.Println("")
|
|
os.Exit(0)
|
|
default:
|
|
config.Gpu_Group = choice
|
|
viewGPU(config)
|
|
}
|
|
}
|
|
|
|
func viewGPU(config *configs.Config, ext ...int) {
|
|
// Clear the screen
|
|
command.Clear()
|
|
|
|
// Set mode to relative
|
|
mode := "-r"
|
|
|
|
// Set mode to relative extended
|
|
if len(ext) > 0 {
|
|
mode = "-rr"
|
|
}
|
|
|
|
// Get the IOMMU listings for GPUs
|
|
group := lsiommu.GetIOMMU("-g", mode, "-i", config.Gpu_Group, "-F", "vendor:,prod_name,optional_revision:,device_id")
|
|
|
|
// Write a title
|
|
title := color.New(color.BgHiBlue, color.White, color.Bold)
|
|
title.Println("This list should only show devices related to your GPU (usually 1 video, 1 audio device)")
|
|
|
|
// Print all the gpus
|
|
for _, v := range group {
|
|
fmt.Println(v)
|
|
}
|
|
|
|
// Add a new line for tidyness
|
|
fmt.Println("")
|
|
|
|
// Make an empty string
|
|
var choice string
|
|
|
|
// Change choices depending on if we have done an extended search or not
|
|
if len(ext) > 0 {
|
|
choice = menu.YesNoManual("Use this GPU (any extra devices listed may or may not be linked to it) for passthrough?")
|
|
} else {
|
|
choice = menu.YesNoEXT("Use this GPU (and related devices) for passthrough?")
|
|
}
|
|
|
|
// Parse the choice
|
|
switch choice {
|
|
case "":
|
|
// If ESC is pressed
|
|
fmt.Println("")
|
|
os.Exit(0)
|
|
|
|
case "ext":
|
|
// Run an extended relative search
|
|
viewGPU(config, 1)
|
|
|
|
case "n":
|
|
// Go back to selecting a gpu
|
|
SelectGPU(config)
|
|
|
|
case "y":
|
|
// Get the device ids for the selected gpu using ls-iommu
|
|
config.Gpu_IDs = lsiommu.GetIOMMU("-g", mode, "-i", config.Gpu_Group, "--id")
|
|
|
|
// If the kernel_args file already exists
|
|
if fileio.FileExist(config.Path.CMDLINE) {
|
|
// Delete it as we will have to make a new one anyway
|
|
err := os.Remove(config.Path.CMDLINE)
|
|
errorcheck.ErrorCheck(err, fmt.Sprintf("Could not remove %s", config.Path.CMDLINE))
|
|
}
|
|
|
|
// Write initial kernel_arg file
|
|
configs.Set_Cmdline(config.Gpu_IDs)
|
|
|
|
// Go to the vbios dumper page
|
|
genVBIOS_dumper(config)
|
|
|
|
case "manual":
|
|
config.Gpu_IDs = menu.ManualInput(
|
|
"Please manually enter the vendorID:deviceID for every device to use except PCI Express Switches\n"+
|
|
"NOTE: All devices sharing the same IOMMU group will still get pulled into the VM!",
|
|
"xxxx:yyyy,xxxx:yyyy,xxxx:yyyy",
|
|
)
|
|
|
|
// If the kernel_args file already exists
|
|
if fileio.FileExist(config.Path.CMDLINE) {
|
|
// Delete it as we will have to make a new one anyway
|
|
err := os.Remove(config.Path.CMDLINE)
|
|
errorcheck.ErrorCheck(err, fmt.Sprintf("Could not remove %s", config.Path.CMDLINE))
|
|
}
|
|
|
|
// Write initial kernel_arg file
|
|
configs.Set_Cmdline(config.Gpu_IDs)
|
|
|
|
// Go to the vbios dumper page
|
|
genVBIOS_dumper(config)
|
|
}
|
|
}
|