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 <asm/io.h>
9 #include <asm/arch/boot_mode.h>
10 
11 #if (CONFIG_ROCKCHIP_BOOT_MODE_REG == 0)
12 
13 int setup_boot_mode(void)
14 {
15 	return 0;
16 }
17 
18 #else
19 
20 void set_back_to_bootrom_dnl_flag(void)
21 {
22 	writel(BOOT_BROM_DOWNLOAD, CONFIG_ROCKCHIP_BOOT_MODE_REG);
23 }
24 
25 /*
26  * detect download key status by adc, most rockchip
27  * based boards use adc sample the download key status,
28  * but there are also some use gpio. So it's better to
29  * make this a weak function that can be override by
30  * some special boards.
31  */
32 #define KEY_DOWN_MIN_VAL	0
33 #define KEY_DOWN_MAX_VAL	30
34 
35 __weak int rockchip_dnl_key_pressed(void)
36 {
37 	unsigned int val;
38 
39 	if (adc_channel_single_shot("saradc", 1, &val)) {
40 		pr_err("%s: adc_channel_single_shot fail!\n", __func__);
41 		return false;
42 	}
43 
44 	if ((val >= KEY_DOWN_MIN_VAL) && (val <= KEY_DOWN_MAX_VAL))
45 		return true;
46 	else
47 		return false;
48 }
49 
50 void rockchip_dnl_mode_check(void)
51 {
52 	if (rockchip_dnl_key_pressed()) {
53 		printf("download key pressed, entering download mode...");
54 		set_back_to_bootrom_dnl_flag();
55 		do_reset(NULL, 0, 0, NULL);
56 	}
57 }
58 
59 int setup_boot_mode(void)
60 {
61 	void *reg = (void *)CONFIG_ROCKCHIP_BOOT_MODE_REG;
62 	int boot_mode = readl(reg);
63 
64 	rockchip_dnl_mode_check();
65 
66 	boot_mode = readl(reg);
67 	debug("%s: boot mode 0x%08x\n", __func__, boot_mode);
68 
69 	/* Clear boot mode */
70 	writel(BOOT_NORMAL, reg);
71 
72 	switch (boot_mode) {
73 	case BOOT_FASTBOOT:
74 		debug("%s: enter fastboot!\n", __func__);
75 		env_set("preboot", "setenv preboot; fastboot usb0");
76 		break;
77 	case BOOT_UMS:
78 		debug("%s: enter UMS!\n", __func__);
79 		env_set("preboot", "setenv preboot; ums mmc 0");
80 		break;
81 	}
82 
83 	return 0;
84 }
85 
86 #endif
87