Add UNKNOWN to PartTableType

blp1526 · Feb 2, 2019 · 9aa00fd · 9aa00fd
1 parent 5db89ec
commit 9aa00fd
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/blk/blk.go b/blk/blk.go
@@ -110,7 +110,15 @@ func GetPartTableType(majorMinor string) (string, error) {
 	for _, line := range lines {
 		prefix := "E:ID_PART_TABLE_TYPE="
 		if strings.HasPrefix(line, prefix) {
-			return strings.TrimPrefix(line, prefix), nil
+			rawName := strings.TrimPrefix(line, prefix)
+			switch rawName {
+			case "dos":
+				return "MBR", nil
+			case "gpt":
+				return "GPT", nil
+			default:
+				return "UNKNOWN", nil
+			}
 		}
 	}