xref: /rk3399_rockchip-uboot/arch/arm/mach-rockchip/kernel_dtb.c (revision 4a2b8db466479ddec6ee85f9fe9d7f934016be9a)
1 /*
2  * (C) Copyright 2019 Rockchip Electronics Co., Ltd
3  *
4  * SPDX-License-Identifier:     GPL-2.0+
5  */
6 #include <common.h>
7 #include <boot_rkimg.h>
8 #include <dm.h>
9 #include <malloc.h>
10 #include <of_live.h>
11 #include <dm/root.h>
12 #include <asm/arch/hotkey.h>
13 
14 DECLARE_GLOBAL_DATA_PTR;
15 
16 /* Here, only fixup cru phandle, pmucru is not included */
17 static int phandles_fixup_cru(const void *fdt)
18 {
19 	const char *props[] = { "clocks", "assigned-clocks" };
20 	struct udevice *dev;
21 	struct uclass *uc;
22 	const char *comp;
23 	u32 id, nclocks;
24 	u32 *clocks;
25 	int phandle, ncells;
26 	int off, offset;
27 	int ret, length;
28 	int i, j;
29 	int first_phandle = -1;
30 
31 	phandle = -ENODATA;
32 	ncells = -ENODATA;
33 
34 	/* fdt points to kernel dtb, getting cru phandle and "#clock-cells" */
35 	for (offset = fdt_next_node(fdt, 0, NULL);
36 	     offset >= 0;
37 	     offset = fdt_next_node(fdt, offset, NULL)) {
38 		comp = fdt_getprop(fdt, offset, "compatible", NULL);
39 		if (!comp)
40 			continue;
41 
42 		/* Actually, this is not a good method to get cru node */
43 		off = strlen(comp) - strlen("-cru");
44 		if (off > 0 && !strncmp(comp + off, "-cru", 4)) {
45 			phandle = fdt_get_phandle(fdt, offset);
46 			ncells = fdtdec_get_int(fdt, offset,
47 						"#clock-cells", -ENODATA);
48 			break;
49 		}
50 	}
51 
52 	if (phandle == -ENODATA || ncells == -ENODATA)
53 		return 0;
54 
55 	debug("%s: target cru: clock-cells:%d, phandle:0x%x\n",
56 	      __func__, ncells, fdt32_to_cpu(phandle));
57 
58 	/* Try to fixup all cru phandle from U-Boot dtb nodes */
59 	for (id = 0; id < UCLASS_COUNT; id++) {
60 		ret = uclass_get(id, &uc);
61 		if (ret)
62 			continue;
63 
64 		if (list_empty(&uc->dev_head))
65 			continue;
66 
67 		list_for_each_entry(dev, &uc->dev_head, uclass_node) {
68 			/* Only U-Boot node go further */
69 			if (!dev_read_bool(dev, "u-boot,dm-pre-reloc") &&
70 			    !dev_read_bool(dev, "u-boot,dm-spl"))
71 				continue;
72 
73 			for (i = 0; i < ARRAY_SIZE(props); i++) {
74 				if (!dev_read_prop(dev, props[i], &length))
75 					continue;
76 
77 				clocks = malloc(length);
78 				if (!clocks)
79 					return -ENOMEM;
80 
81 				/* Read "props[]" which contains cru phandle */
82 				nclocks = length / sizeof(u32);
83 				if (dev_read_u32_array(dev, props[i],
84 						       clocks, nclocks)) {
85 					free(clocks);
86 					continue;
87 				}
88 
89 				/* Fixup with kernel cru phandle */
90 				for (j = 0; j < nclocks; j += (ncells + 1)) {
91 					/*
92 					 * Check: update pmucru phandle with cru
93 					 * phandle by mistake.
94 					 */
95 					if (first_phandle == -1)
96 						first_phandle = clocks[j];
97 
98 					if (clocks[j] != first_phandle) {
99 						debug("WARN: %s: first cru phandle=%d, this=%d\n",
100 						      dev_read_name(dev),
101 						      first_phandle, clocks[j]);
102 						continue;
103 					}
104 
105 					clocks[j] = phandle;
106 				}
107 
108 				/*
109 				 * Override live dt nodes but not fdt nodes,
110 				 * because all U-Boot nodes has been imported
111 				 * to live dt nodes, should use "dev_xxx()".
112 				 */
113 				dev_write_u32_array(dev, props[i],
114 						    clocks, nclocks);
115 				free(clocks);
116 			}
117 		}
118 	}
119 
120 	return 0;
121 }
122 
123 static int phandles_fixup_gpio(const void *fdt, void *ufdt)
124 {
125 	struct udevice *dev;
126 	struct uclass *uc;
127 	const char *prop = "gpios";
128 	const char *comp;
129 	char *gpio_name[10];
130 	int gpio_off[10];
131 	int pinctrl;
132 	int offset;
133 	int i = 0;
134 	int n = 0;
135 
136 	pinctrl = fdt_path_offset(fdt, "/pinctrl");
137 	if (pinctrl < 0)
138 		return 0;
139 
140 	memset(gpio_name, 0, sizeof(gpio_name));
141 	for (offset = fdt_first_subnode(fdt, pinctrl);
142 	     offset >= 0;
143 	     offset = fdt_next_subnode(fdt, offset)) {
144 		/* assume the font nodes are gpio node */
145 		if (++i >= ARRAY_SIZE(gpio_name))
146 			break;
147 
148 		comp = fdt_getprop(fdt, offset, "compatible", NULL);
149 		if (!comp)
150 			continue;
151 
152 		if (!strcmp(comp, "rockchip,gpio-bank")) {
153 			gpio_name[n] = (char *)fdt_get_name(fdt, offset, NULL);
154 			gpio_off[n]  = offset;
155 			n++;
156 		}
157 	}
158 
159 	if (!gpio_name[0])
160 		return 0;
161 
162 	if (uclass_get(UCLASS_KEY, &uc) || list_empty(&uc->dev_head))
163 		return 0;
164 
165 	list_for_each_entry(dev, &uc->dev_head, uclass_node) {
166 		u32 new_phd, phd_old;
167 		char *name;
168 		ofnode ofn;
169 
170 		if (!dev_read_bool(dev, "u-boot,dm-pre-reloc") &&
171 		    !dev_read_bool(dev, "u-boot,dm-spl"))
172 			continue;
173 
174 		if (dev_read_u32_array(dev, prop, &phd_old, 1))
175 			continue;
176 
177 		ofn = ofnode_get_by_phandle(phd_old);
178 		if (!ofnode_valid(ofn))
179 			continue;
180 
181 		name = (char *)ofnode_get_name(ofn);
182 		if (!name)
183 			continue;
184 
185 		for (i = 0; i < ARRAY_SIZE(gpio_name[i]); i++) {
186 			if (gpio_name[i] && !strcmp(name, gpio_name[i])) {
187 				new_phd = fdt_get_phandle(fdt, gpio_off[i]);
188 				dev_write_u32_array(dev, prop, &new_phd, 1);
189 				break;
190 			}
191 		}
192 	}
193 
194 	return 0;
195 }
196 
197 __weak int board_mmc_dm_reinit(struct udevice *dev)
198 {
199 	return 0;
200 }
201 
202 static int mmc_dm_reinit(void)
203 {
204 	struct udevice *dev;
205 	struct uclass *uc;
206 	int ret;
207 
208 	if (uclass_get(UCLASS_MMC, &uc) || list_empty(&uc->dev_head))
209 		return 0;
210 
211 	list_for_each_entry(dev, &uc->dev_head, uclass_node) {
212 		ret = board_mmc_dm_reinit(dev);
213 		if (ret)
214 			return ret;
215 	}
216 
217 	return 0;
218 }
219 
220 int init_kernel_dtb(void)
221 {
222 	ulong fdt_addr;
223 	void *ufdt_blob;
224 	int ret;
225 
226 	fdt_addr = env_get_ulong("fdt_addr_r", 16, 0);
227 	if (!fdt_addr) {
228 		printf("No Found FDT Load Address.\n");
229 		return -1;
230 	}
231 
232 	ret = rockchip_read_dtb_file((void *)fdt_addr);
233 	if (ret < 0) {
234 		if (!fdt_check_header(gd->fdt_blob_kern)) {
235 			fdt_addr = (ulong)memalign(ARCH_DMA_MINALIGN,
236 					fdt_totalsize(gd->fdt_blob_kern));
237 			if (!fdt_addr)
238 				return -ENOMEM;
239 
240 			memcpy((void *)fdt_addr, gd->fdt_blob_kern,
241 			       fdt_totalsize(gd->fdt_blob_kern));
242 			printf("DTB: embedded kern.dtb\n");
243 		} else {
244 			printf("Failed to get kernel dtb, ret=%d\n", ret);
245 			return ret;
246 		}
247 	}
248 
249 	ufdt_blob = (void *)gd->fdt_blob;
250 	gd->fdt_blob = (void *)fdt_addr;
251 
252 	hotkey_run(HK_FDT);
253 
254 	/*
255 	 * There is a phandle miss match between U-Boot and kernel dtb node,
256 	 * we fixup it in U-Boot live dt nodes.
257 	 *
258 	 * CRU:	 all nodes.
259 	 * GPIO: key nodes.
260 	 */
261 	phandles_fixup_cru((void *)gd->fdt_blob);
262 	phandles_fixup_gpio((void *)gd->fdt_blob, (void *)ufdt_blob);
263 
264 	of_live_build((void *)gd->fdt_blob, (struct device_node **)&gd->of_root);
265 	dm_scan_fdt((void *)gd->fdt_blob, false);
266 
267 	/*
268 	 * There maybe something for the mmc devices to do after kernel dtb
269 	 * dm setup, eg: regain the clock device binding from kernel dtb.
270 	 */
271 	mmc_dm_reinit();
272 
273 	/* Reserve 'reserved-memory' */
274 	ret = boot_fdt_add_sysmem_rsv_regions((void *)gd->fdt_blob);
275 	if (ret)
276 		return ret;
277 
278 	return 0;
279 }
280