flash-wrs: verify memory mode input

Signed-off-by: Federico Vaga <federico.vaga@gmail.com> Conflicts: build/flash-wrs

flash-wrs: verify memory mode input
Signed-off-by: Federico Vaga <federico.vaga@gmail.com> Conflicts: build/flash-wrs
4aa7876f · Federico Vaga · Alessandro Rubini · d0be3bc6 · 4aa7876f
Commit 4aa7876f authored Jul 16, 2012 by Federico Vaga Committed by Alessandro Rubini Jul 17, 2012
Hide whitespace changes
Inline Side-by-side

Showing with 9 additions and 1 deletion

flash-wrs build/flash-wrs +9 -1

No files found.
--- a/build/flash-wrs
+++ b/build/flash-wrs
@@ -118,7 +118,15 @@ while [ $# -ge 1 ]; do
 		rootfs=/tmp/wrs-image.jffs2.img
 		shift;;
 	-h|--help) 	showhelp; shift;;
-	-m|--memmode)	memmode="$2";  shift; shift;;
+	-m|--memmode)
+	    # verify memory mode
+	    if [ "$2" != "nf" ] && [ "$2" != "df" ] && [ "$2" != "test" ] && [ "$2" != "default" ]; then
+		echo "Error: Invalid memory mode"
+		showhelp
+		exit 1
+	    fi
+	    memmode="$2"
+	    shift; shift;;
 	--skip)	skip=1; shift;;
 	/* ) DEV="-s $1"; shift ;;
 	*:* ) MAC="$1"; shift ;;