1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * (C) Copyright 2016 Rockchip Electronics Co., Ltd
4  */
5 
6 #include <common.h>
7 #include <adc.h>
8 #include <command.h>
9 #include <env.h>
10 #include <log.h>
11 #include <asm/io.h>
12 #include <asm/arch-rockchip/boot_mode.h>
13 #include <dm/device.h>
14 #include <dm/uclass.h>
15 
16 #if (CONFIG_ROCKCHIP_BOOT_MODE_REG == 0)
17 
setup_boot_mode(void)18 int setup_boot_mode(void)
19 {
20 	return 0;
21 }
22 
23 #else
24 
set_back_to_bootrom_dnl_flag(void)25 void set_back_to_bootrom_dnl_flag(void)
26 {
27 	writel(BOOT_BROM_DOWNLOAD, CONFIG_ROCKCHIP_BOOT_MODE_REG);
28 }
29 
30 /*
31  * detect download key status by adc, most rockchip
32  * based boards use adc sample the download key status,
33  * but there are also some use gpio. So it's better to
34  * make this a weak function that can be override by
35  * some special boards.
36  */
37 #define KEY_DOWN_MIN_VAL	0
38 #define KEY_DOWN_MAX_VAL	30
39 
rockchip_dnl_key_pressed(void)40 __weak int rockchip_dnl_key_pressed(void)
41 {
42 	unsigned int val;
43 	struct udevice *dev;
44 	struct uclass *uc;
45 	int ret;
46 
47 	ret = uclass_get(UCLASS_ADC, &uc);
48 	if (ret)
49 		return false;
50 
51 	ret = -ENODEV;
52 	uclass_foreach_dev(dev, uc) {
53 		if (!strncmp(dev->name, "saradc", 6)) {
54 			ret = adc_channel_single_shot(dev->name, 1, &val);
55 			break;
56 		}
57 	}
58 
59 	if (ret == -ENODEV) {
60 		pr_warn("%s: no saradc device found\n", __func__);
61 		return false;
62 	} else if (ret) {
63 		pr_err("%s: adc_channel_single_shot fail!\n", __func__);
64 		return false;
65 	}
66 
67 	if ((val >= KEY_DOWN_MIN_VAL) && (val <= KEY_DOWN_MAX_VAL))
68 		return true;
69 	else
70 		return false;
71 }
72 
rockchip_dnl_mode_check(void)73 void rockchip_dnl_mode_check(void)
74 {
75 	if (rockchip_dnl_key_pressed()) {
76 		printf("download key pressed, entering download mode...");
77 		set_back_to_bootrom_dnl_flag();
78 		do_reset(NULL, 0, 0, NULL);
79 	}
80 }
81 
setup_boot_mode(void)82 int setup_boot_mode(void)
83 {
84 	void *reg = (void *)CONFIG_ROCKCHIP_BOOT_MODE_REG;
85 	int boot_mode = readl(reg);
86 
87 	rockchip_dnl_mode_check();
88 
89 	boot_mode = readl(reg);
90 	debug("%s: boot mode 0x%08x\n", __func__, boot_mode);
91 
92 	/* Clear boot mode */
93 	writel(BOOT_NORMAL, reg);
94 
95 	switch (boot_mode) {
96 	case BOOT_FASTBOOT:
97 		debug("%s: enter fastboot!\n", __func__);
98 		env_set("preboot", "setenv preboot; fastboot usb0");
99 		break;
100 	case BOOT_UMS:
101 		debug("%s: enter UMS!\n", __func__);
102 		env_set("preboot", "setenv preboot; ums mmc 0");
103 		break;
104 	}
105 
106 	return 0;
107 }
108 
109 #endif
110