6c48a35180
* Fix: don't need sudo if we're root + other aesthetics
* Heavy refactoring, see PR #28
* Fix: avoid silent fatalities
demo: https://tcp.ac/i/JMSUc.gif
* Fix: Inverse check on `IsRoot`
* D.R.Y: check for permissions error in `common.ErrorCheck`
Reduce cognitive complexity.
* Fix: Issue with copying
* Resolve https://github.com/HikariKnight/quickpassthrough/pull/28#discussion_r1646535918
* Resolve https://github.com/HikariKnight/quickpassthrough/pull/28#discussion_r1646606680 and https://github.com/HikariKnight/quickpassthrough/pull/28#discussion_r1646594105
* Revert "Resolve https://github.com/HikariKnight/quickpassthrough/pull/28#discussion_r1646606680 and https://github.com/HikariKnight/quickpassthrough/pull/28#discussion_r1646594105"
This reverts commit ce15213009.
* Resolve https://github.com/HikariKnight/quickpassthrough/pull/28#discussion_r1646730751
92 lines
1.9 KiB
Go
92 lines
1.9 KiB
Go
package pages
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/gookit/color"
|
|
|
|
"github.com/HikariKnight/quickpassthrough/internal/configs"
|
|
"github.com/HikariKnight/quickpassthrough/internal/lsiommu"
|
|
"github.com/HikariKnight/quickpassthrough/pkg/command"
|
|
"github.com/HikariKnight/quickpassthrough/pkg/menu"
|
|
)
|
|
|
|
func selectUSB(config *configs.Config) {
|
|
// Clear the screen
|
|
command.Clear()
|
|
|
|
// Get the users GPUs
|
|
usbs := lsiommu.GetIOMMU("-u", "-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 USB to view the IOMMU groups of", usbs, 1)
|
|
|
|
// Parse the choice
|
|
switch choice {
|
|
case "back":
|
|
disableVideo(config)
|
|
|
|
case "":
|
|
// If ESC is pressed
|
|
fmt.Println("")
|
|
os.Exit(0)
|
|
|
|
default:
|
|
// View the selected GPU
|
|
viewUSB(choice, config)
|
|
}
|
|
}
|
|
|
|
func viewUSB(id string, 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 USB controllers
|
|
group := lsiommu.GetIOMMU("-u", mode, "-i", id, "-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 the USB controller")
|
|
|
|
// Print all the usb controllers
|
|
for _, v := range group {
|
|
fmt.Println(v)
|
|
}
|
|
|
|
// Add a new line for tidyness
|
|
fmt.Println("")
|
|
|
|
// Make an empty string
|
|
var choice string
|
|
|
|
// Ask if we shall use the devices for passthrough
|
|
if len(ext) == 0 {
|
|
choice = menu.YesNo("Use all listed devices for passthrough?")
|
|
} else {
|
|
choice = menu.YesNoEXT("Use all listed devices for passthrough?")
|
|
}
|
|
|
|
// Parse the choice
|
|
switch choice {
|
|
case "":
|
|
// If ESC is pressed
|
|
fmt.Println("")
|
|
os.Exit(0)
|
|
|
|
case "n":
|
|
// Go back to selecting a gpu
|
|
selectUSB(config)
|
|
|
|
case "y":
|
|
// Go to the select a usb controller
|
|
}
|
|
}
|