From 778cbcb5cc28240aa1caeae4dd02c2a39567a0e1 Mon Sep 17 00:00:00 2001
From: Andrew Tridgell <tridge@samba.org>
Date: Wed, 15 Jan 2014 19:27:03 +1100
Subject: mtd: fixed creation and erase of a single partition

---
 src/systemcmds/mtd/mtd.c | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/systemcmds/mtd/mtd.c b/src/systemcmds/mtd/mtd.c
index 41da63750..5525a8f33 100644
--- a/src/systemcmds/mtd/mtd.c
+++ b/src/systemcmds/mtd/mtd.c
@@ -108,9 +108,8 @@ int mtd_main(int argc, char *argv[])
 		if (!strcmp(argv[1], "start")) {
 
 			/* start mapping according to user request */
-			if (argc > 3) {
+			if (argc >= 3) {
 				mtd_start(argv + 2, argc - 2);
-
 			} else {
 				mtd_start(partition_names_default, n_partitions_default);
 			}
@@ -123,10 +122,11 @@ int mtd_main(int argc, char *argv[])
 			mtd_status();
 
 		if (!strcmp(argv[1], "erase")) {
-			if (argc < 3) {
-				errx(1, "usage: mtd erase <PARTITION_PATH..>");
+			if (argc >= 3) {
+				mtd_erase(argv + 2, argc - 2);
+			} else {
+				mtd_erase(partition_names_default, n_partitions_default);
 			}
-			mtd_erase(argv + 2, argc - 2);
                 }
 	}
 
-- 
cgit v1.2.3