Merge tag 'u-boot-atmel-fixes-2020.07-a' of https://gitlab.denx.de/u-boot/custodians...

[oweals/u-boot.git] / arch / arm / mach-rockchip / boot_mode.c

diff --git a/arch/arm/mach-rockchip/boot_mode.c b/arch/arm/mach-rockchip/boot_mode.c
index 46524900def4dbfb629b22a36e9d8abb67b0a61c..2158934159a5b997a102d2b3649352b635927a98 100644 (file)
--- a/arch/arm/mach-rockchip/boot_mode.c
+++ b/arch/arm/mach-rockchip/boot_mode.c
@@ -1,33 +1,109 @@
+// SPDX-License-Identifier: GPL-2.0+
 /*
  * (C) Copyright 2016 Rockchip Electronics Co., Ltd
- *
- * SPDX-License-Identifier:     GPL-2.0+
  */
 
 #include <common.h>
+#include <adc.h>
+#include <command.h>
+#include <env.h>
+#include <log.h>
 #include <asm/io.h>
-#include <asm/arch/boot_mode.h>
+#include <asm/arch-rockchip/boot_mode.h>
+#include <dm/device.h>
+#include <dm/uclass.h>
+
+#if (CONFIG_ROCKCHIP_BOOT_MODE_REG == 0)
+
+int setup_boot_mode(void)
+{
+       return 0;
+}
+
+#else
+
+void set_back_to_bootrom_dnl_flag(void)
+{
+       writel(BOOT_BROM_DOWNLOAD, CONFIG_ROCKCHIP_BOOT_MODE_REG);
+}
+
+/*
+ * detect download key status by adc, most rockchip
+ * based boards use adc sample the download key status,
+ * but there are also some use gpio. So it's better to
+ * make this a weak function that can be override by
+ * some special boards.
+ */
+#define KEY_DOWN_MIN_VAL       0
+#define KEY_DOWN_MAX_VAL       30
+
+__weak int rockchip_dnl_key_pressed(void)
+{
+       unsigned int val;
+       struct udevice *dev;
+       struct uclass *uc;
+       int ret;
+
+       ret = uclass_get(UCLASS_ADC, &uc);
+       if (ret)
+               return false;
+
+       ret = -ENODEV;
+       uclass_foreach_dev(dev, uc) {
+               if (!strncmp(dev->name, "saradc", 6)) {
+                       ret = adc_channel_single_shot(dev->name, 1, &val);
+                       break;
+               }
+       }
+
+       if (ret == -ENODEV) {
+               pr_warn("%s: no saradc device found\n", __func__);
+               return false;
+       } else if (ret) {
+               pr_err("%s: adc_channel_single_shot fail!\n", __func__);
+               return false;
+       }
+
+       if ((val >= KEY_DOWN_MIN_VAL) && (val <= KEY_DOWN_MAX_VAL))
+               return true;
+       else
+               return false;
+}
+
+void rockchip_dnl_mode_check(void)
+{
+       if (rockchip_dnl_key_pressed()) {
+               printf("download key pressed, entering download mode...");
+               set_back_to_bootrom_dnl_flag();
+               do_reset(NULL, 0, 0, NULL);
+       }
+}
 
 int setup_boot_mode(void)
 {
        void *reg = (void *)CONFIG_ROCKCHIP_BOOT_MODE_REG;
        int boot_mode = readl(reg);
 
-       debug("boot mode %x.\n", boot_mode);
+       rockchip_dnl_mode_check();
+
+       boot_mode = readl(reg);
+       debug("%s: boot mode 0x%08x\n", __func__, boot_mode);
 
        /* Clear boot mode */
        writel(BOOT_NORMAL, reg);
 
        switch (boot_mode) {
        case BOOT_FASTBOOT:
-               printf("enter fastboot!\n");
+               debug("%s: enter fastboot!\n", __func__);
                env_set("preboot", "setenv preboot; fastboot usb0");
                break;
        case BOOT_UMS:
-               printf("enter UMS!\n");
+               debug("%s: enter UMS!\n", __func__);
                env_set("preboot", "setenv preboot; ums mmc 0");
                break;
        }
 
        return 0;
 }
+
+#endif