create_image.sh check part_type before creating

tjmnmk · Mar 21, 2021 · 07ad2e7 · 07ad2e7
1 parent a8ed54c
commit 07ad2e7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/create_image.sh b/create_image.sh
@@ -36,6 +36,11 @@ else
     size="${size}k"
     part_type=ntfs
 fi
+
+if [ "$part_type" != "ntfs" ] && [ "$part_type" != "fat32" ]; then
+    echo "$part_type is not supported, choose ntfs or fat32" 1>&2
+    exit 1
+fi
 
 echo "Creating $size image..."
 
@@ -50,7 +55,6 @@ elif [ "$part_type" = "fat32" ]; then
     mkfs.vfat "${dev}p1"
     fatlabel "${dev}p1" RPIHDD
 else
-    echo "$part_type is not supported, choose ntfs or fat32"
     exit 1
 fi