From a2b2529576e47b18431ecdb34a746a768b0db78e Mon Sep 17 00:00:00 2001
From: Jeffrey Townsend <jeffrey.townsend@bigswitch.com>
Date: Fri, 15 Jan 2016 12:49:03 -0800
Subject: [PATCH] Improve partition descrimination.

---
 .../installer/legacy/builds/amd64-installer.sh   | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/builds/amd64/installer/legacy/builds/amd64-installer.sh b/builds/amd64/installer/legacy/builds/amd64-installer.sh
index 8bce42b0..a75ba06d 100644
--- a/builds/amd64/installer/legacy/builds/amd64-installer.sh
+++ b/builds/amd64/installer/legacy/builds/amd64-installer.sh
@@ -143,12 +143,16 @@ do_handle_partitions()
 
   installer_say "Examining $DEV part $part"
 
-  if [ "$label" = "ONIE-BOOT" ] || [ "$label" = "PLATFORM-DIAG" ] || [ "$label" = "GRUB-BOOT" ]; then
-      installer_say "Partition $DEV$part: $label: Preserving..."
-  else
-      installer_say "Partition $DEV$part: $label: Deleting..."
-      parted $DEV rm $part || return 1
-  fi
+
+  case "$label" in
+      ONIE-BOOT|GRUB-BOOT|*-DIAG)
+          installer_say "Partition $DEV$part: $label: Preserving..."
+          ;;
+      *)
+          installer_say "Partition $DEV$part: $label: Deleting..."
+          parted $DEV rm $part || return 1
+          ;;
+  esac
 
   return 0
 }