add support for oem, vendor and prod_name in formated output

HikariKnight · Apr 8, 2023 · 1e41609 · 1e41609
1 parent e1ca657
commit 1e41609
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/pkg/iommu/genoutput.go b/pkg/iommu/genoutput.go
@@ -45,6 +45,18 @@ func GenDeviceLine(group int, device *pci.Device, pArg *params.Params) string {
 			formated_line = append(formated_line, fmt.Sprintf("%s %s", device.Vendor.Name, device.Product.Name))
 		case "name:":
 			formated_line = append(formated_line, fmt.Sprintf("%s %s:", device.Vendor.Name, device.Product.Name))
+		case "prod_name":
+			formated_line = append(formated_line, device.Product.Name)
+		case "prod_name:":
+			formated_line = append(formated_line, fmt.Sprintf("%s:", device.Product.Name))
+		case "oem":
+			formated_line = append(formated_line, device.Subsystem.Name)
+		case "oem:":
+			formated_line = append(formated_line, fmt.Sprintf("%s:", device.Subsystem.Name))
+		case "vendor":
+			formated_line = append(formated_line, device.Vendor.Name)
+		case "vendor:":
+			formated_line = append(formated_line, fmt.Sprintf("%s:", device.Vendor.Name))
 		case "device_id":
 			formated_line = append(formated_line, fmt.Sprintf("[%s:%s]", device.Vendor.ID, device.Product.ID))
 		case "device_id:":

diff --git a/pkg/params/params.go b/pkg/params/params.go
@@ -112,7 +112,7 @@ func NewParams() *Params {
 
 	format := parser.String("F", "format", &argparse.Options{
 		Required: false,
-		Help:     "Formats the device line output the way you want it (omit what you do not want)\n\t\t Supported objects: pciaddr, subclass_name, subclass_name:, subclass_id, subclass_id:, name, name:, device_id, device_id:, revision, optional_revision",
+		Help:     "Formats the device line output the way you want it (omit what you do not want)\n\t\t Supported objects: pciaddr, subclass_name, subclass_name:, subclass_id, subclass_id:, name, name:, device_id, device_id:, vendor, vendor:, oem, oem:, prod_name, prod_name:, revision, optional_revision",
 		Default:  "pciaddr,subclass_name,subclass_id,name,device_id,optional_revision",
 	})