make all fields work with :

HikariKnight · Apr 8, 2023 · a2f4e2a · a2f4e2a
1 parent 1e41609
commit a2f4e2a
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/pkg/iommu/genoutput.go b/pkg/iommu/genoutput.go
@@ -33,6 +33,8 @@ func GenDeviceLine(group int, device *pci.Device, pArg *params.Params) string {
 		switch object {
 		case "pciaddr":
 			formated_line = append(formated_line, device.Address)
+		case "pciaddr:":
+			formated_line = append(formated_line, fmt.Sprintf("%s:", device.Address))
 		case "subclass_name":
 			formated_line = append(formated_line, device.Subclass.Name)
 		case "subclass_name:":
@@ -63,11 +65,20 @@ func GenDeviceLine(group int, device *pci.Device, pArg *params.Params) string {
 			formated_line = append(formated_line, fmt.Sprintf("[%s:%s]:", device.Vendor.ID, device.Product.ID))
 		case "revision":
 			formated_line = append(formated_line, fmt.Sprintf("(rev %s)", device.Revision[len(device.Revision)-2:]))
+		case "revision:":
+			formated_line = append(formated_line, fmt.Sprintf("(rev %s):", device.Revision[len(device.Revision)-2:]))
 		case "optional_revision":
 			// Else only show it if the device is not on revision 00
 			if device.Revision != "0x00" {
 				formated_line = append(formated_line, fmt.Sprintf("(rev %s)", device.Revision[len(device.Revision)-2:]))
 			}
+		case "optional_revision:":
+			// Else only show it if the device is not on revision 00
+			if device.Revision != "0x00" {
+				formated_line = append(formated_line, fmt.Sprintf("(rev %s):", device.Revision[len(device.Revision)-2:]))
+			} else {
+				formated_line = append(formated_line, ":")
+			}
 		}
 	}