From fbf99d9116409fd59686dfd3f891547b111d65cb Mon Sep 17 00:00:00 2001
From: Joseph Chen <chenjh@rock-chips.com>
Date: Tue, 30 Jul 2019 15:15:45 +0800
Subject: [PATCH] cmd: add "download" to enter rockusb/bootrom mode

Effect order: rockusb => bootrom.

Change-Id: Ie206f565e1c79e72c173c25f6970c93c8306332f
Signed-off-by: Joseph Chen <chenjh@rock-chips.com>
---
 cmd/Makefile   |  2 +-
 cmd/download.c | 27 +++++++++++++++++++++++++++
 2 files changed, 28 insertions(+), 1 deletion(-)
 create mode 100644 cmd/download.c

diff --git a/cmd/Makefile b/cmd/Makefile
index 353972e0a6..e485988fc8 100644
--- a/cmd/Makefile
+++ b/cmd/Makefile
@@ -81,7 +81,7 @@ obj-$(CONFIG_CMD_CRAMFS) += cramfs.o
 obj-$(CONFIG_LED_STATUS_CMD) += legacy_led.o
 obj-$(CONFIG_CMD_LED) += led.o
 obj-$(CONFIG_CMD_LICENSE) += license.o
-obj-y += load.o
+obj-y += load.o download.o
 obj-$(CONFIG_CMD_LOAD_ANDROID) += load_android.o android_cmds.o
 obj-$(CONFIG_CMD_LOG) += log.o
 obj-$(CONFIG_ID_EEPROM) += mac.o
diff --git a/cmd/download.c b/cmd/download.c
new file mode 100644
index 0000000000..232c474977
--- /dev/null
+++ b/cmd/download.c
@@ -0,0 +1,27 @@
+/*
+ * (C) Copyright 2019 Rockchip Electronics Co., Ltd
+ *
+ * SPDX-License-Identifier:     GPL-2.0+
+ */
+
+#include <common.h>
+#include <command.h>
+
+static int do_download(cmd_tbl_t *cmdtp, int flag,
+		       int argc, char * const argv[])
+{
+#ifdef CONFIG_CMD_ROCKUSB
+	run_command("rockusb 0 $devtype $devnum", 0);
+#endif
+#ifdef CONFIG_CMD_GO
+	printf("Enter rockusb failed, fallback to bootrom...\n");
+	flushc();
+	run_command("rbrom", 0);
+#endif
+	return 0;
+}
+
+U_BOOT_CMD(
+	download, 1, 1, do_download,
+	"enter rockusb/bootrom download mode", ""
+);