X-Git-Url: https://git.sur5r.net/?a=blobdiff_plain;f=common%2Fcmd_demo.c;h=209dc4a57ce71b57fc8ef93c0961aa38e7b54894;hb=08fc7b8fac1d27d7e9f964309cbc32de0abd2c0d;hp=bcb34d904569d98612317dcc4e0712db6edd6410;hpb=8e3da9dd113699eed2fa05fcde3c55a2ff410913;p=u-boot

diff --git a/common/cmd_demo.c b/common/cmd_demo.c
index bcb34d9045..209dc4a57c 100644
--- a/common/cmd_demo.c
+++ b/common/cmd_demo.c
@@ -9,6 +9,7 @@
 
 #include <common.h>
 #include <dm-demo.h>
+#include <mapmem.h>
 #include <asm/io.h>
 
 struct udevice *demo_dev;
@@ -97,7 +98,9 @@ static int do_demo(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
 				ARRAY_SIZE(demo_commands));
 	argc -= 2;
 	argv += 2;
-	if (!demo_cmd || argc > demo_cmd->maxargs)
+
+	if ((!demo_cmd || argc > demo_cmd->maxargs) ||
+	    ((demo_cmd->name[0] != 'l') && (argc < 1)))
 		return CMD_RET_USAGE;
 
 	if (argc) {