bootstd: Allow scanning for global bootmeths separately
[platform/kernel/u-boot.git] / common / fdt_support.c
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * (C) Copyright 2007
4  * Gerald Van Baren, Custom IDEAS, vanbaren@cideas.com
5  *
6  * Copyright 2010-2011 Freescale Semiconductor, Inc.
7  */
8
9 #include <common.h>
10 #include <env.h>
11 #include <log.h>
12 #include <mapmem.h>
13 #include <net.h>
14 #include <stdio_dev.h>
15 #include <linux/ctype.h>
16 #include <linux/types.h>
17 #include <asm/global_data.h>
18 #include <linux/libfdt.h>
19 #include <fdt_support.h>
20 #include <exports.h>
21 #include <fdtdec.h>
22 #include <version.h>
23
24 /**
25  * fdt_getprop_u32_default_node - Return a node's property or a default
26  *
27  * @fdt: ptr to device tree
28  * @off: offset of node
29  * @cell: cell offset in property
30  * @prop: property name
31  * @dflt: default value if the property isn't found
32  *
33  * Convenience function to return a node's property or a default value if
34  * the property doesn't exist.
35  */
36 u32 fdt_getprop_u32_default_node(const void *fdt, int off, int cell,
37                                 const char *prop, const u32 dflt)
38 {
39         const fdt32_t *val;
40         int len;
41
42         val = fdt_getprop(fdt, off, prop, &len);
43
44         /* Check if property exists */
45         if (!val)
46                 return dflt;
47
48         /* Check if property is long enough */
49         if (len < ((cell + 1) * sizeof(uint32_t)))
50                 return dflt;
51
52         return fdt32_to_cpu(*val);
53 }
54
55 /**
56  * fdt_getprop_u32_default - Find a node and return it's property or a default
57  *
58  * @fdt: ptr to device tree
59  * @path: path of node
60  * @prop: property name
61  * @dflt: default value if the property isn't found
62  *
63  * Convenience function to find a node and return it's property or a
64  * default value if it doesn't exist.
65  */
66 u32 fdt_getprop_u32_default(const void *fdt, const char *path,
67                                 const char *prop, const u32 dflt)
68 {
69         int off;
70
71         off = fdt_path_offset(fdt, path);
72         if (off < 0)
73                 return dflt;
74
75         return fdt_getprop_u32_default_node(fdt, off, 0, prop, dflt);
76 }
77
78 /**
79  * fdt_find_and_setprop: Find a node and set it's property
80  *
81  * @fdt: ptr to device tree
82  * @node: path of node
83  * @prop: property name
84  * @val: ptr to new value
85  * @len: length of new property value
86  * @create: flag to create the property if it doesn't exist
87  *
88  * Convenience function to directly set a property given the path to the node.
89  */
90 int fdt_find_and_setprop(void *fdt, const char *node, const char *prop,
91                          const void *val, int len, int create)
92 {
93         int nodeoff = fdt_path_offset(fdt, node);
94
95         if (nodeoff < 0)
96                 return nodeoff;
97
98         if ((!create) && (fdt_get_property(fdt, nodeoff, prop, NULL) == NULL))
99                 return 0; /* create flag not set; so exit quietly */
100
101         return fdt_setprop(fdt, nodeoff, prop, val, len);
102 }
103
104 /**
105  * fdt_find_or_add_subnode() - find or possibly add a subnode of a given node
106  *
107  * @fdt: pointer to the device tree blob
108  * @parentoffset: structure block offset of a node
109  * @name: name of the subnode to locate
110  *
111  * fdt_subnode_offset() finds a subnode of the node with a given name.
112  * If the subnode does not exist, it will be created.
113  */
114 int fdt_find_or_add_subnode(void *fdt, int parentoffset, const char *name)
115 {
116         int offset;
117
118         offset = fdt_subnode_offset(fdt, parentoffset, name);
119
120         if (offset == -FDT_ERR_NOTFOUND)
121                 offset = fdt_add_subnode(fdt, parentoffset, name);
122
123         if (offset < 0)
124                 printf("%s: %s: %s\n", __func__, name, fdt_strerror(offset));
125
126         return offset;
127 }
128
129 #if defined(CONFIG_OF_STDOUT_VIA_ALIAS) && defined(CONFIG_CONS_INDEX)
130 static int fdt_fixup_stdout(void *fdt, int chosenoff)
131 {
132         int err;
133         int aliasoff;
134         char sername[9] = { 0 };
135         const void *path;
136         int len;
137         char tmp[256]; /* long enough */
138
139         sprintf(sername, "serial%d", CONFIG_CONS_INDEX - 1);
140
141         aliasoff = fdt_path_offset(fdt, "/aliases");
142         if (aliasoff < 0) {
143                 err = aliasoff;
144                 goto noalias;
145         }
146
147         path = fdt_getprop(fdt, aliasoff, sername, &len);
148         if (!path) {
149                 err = len;
150                 goto noalias;
151         }
152
153         /* fdt_setprop may break "path" so we copy it to tmp buffer */
154         memcpy(tmp, path, len);
155
156         err = fdt_setprop(fdt, chosenoff, "linux,stdout-path", tmp, len);
157         if (err < 0)
158                 printf("WARNING: could not set linux,stdout-path %s.\n",
159                        fdt_strerror(err));
160
161         return err;
162
163 noalias:
164         printf("WARNING: %s: could not read %s alias: %s\n",
165                __func__, sername, fdt_strerror(err));
166
167         return 0;
168 }
169 #else
170 static int fdt_fixup_stdout(void *fdt, int chosenoff)
171 {
172         return 0;
173 }
174 #endif
175
176 static inline int fdt_setprop_uxx(void *fdt, int nodeoffset, const char *name,
177                                   uint64_t val, int is_u64)
178 {
179         if (is_u64)
180                 return fdt_setprop_u64(fdt, nodeoffset, name, val);
181         else
182                 return fdt_setprop_u32(fdt, nodeoffset, name, (uint32_t)val);
183 }
184
185 int fdt_root(void *fdt)
186 {
187         char *serial;
188         int err;
189
190         err = fdt_check_header(fdt);
191         if (err < 0) {
192                 printf("fdt_root: %s\n", fdt_strerror(err));
193                 return err;
194         }
195
196         serial = env_get("serial#");
197         if (serial) {
198                 err = fdt_setprop(fdt, 0, "serial-number", serial,
199                                   strlen(serial) + 1);
200
201                 if (err < 0) {
202                         printf("WARNING: could not set serial-number %s.\n",
203                                fdt_strerror(err));
204                         return err;
205                 }
206         }
207
208         return 0;
209 }
210
211 int fdt_initrd(void *fdt, ulong initrd_start, ulong initrd_end)
212 {
213         int   nodeoffset;
214         int   err, j, total;
215         int is_u64;
216         uint64_t addr, size;
217
218         /* just return if the size of initrd is zero */
219         if (initrd_start == initrd_end)
220                 return 0;
221
222         /* find or create "/chosen" node. */
223         nodeoffset = fdt_find_or_add_subnode(fdt, 0, "chosen");
224         if (nodeoffset < 0)
225                 return nodeoffset;
226
227         total = fdt_num_mem_rsv(fdt);
228
229         /*
230          * Look for an existing entry and update it.  If we don't find
231          * the entry, we will j be the next available slot.
232          */
233         for (j = 0; j < total; j++) {
234                 err = fdt_get_mem_rsv(fdt, j, &addr, &size);
235                 if (addr == initrd_start) {
236                         fdt_del_mem_rsv(fdt, j);
237                         break;
238                 }
239         }
240
241         err = fdt_add_mem_rsv(fdt, initrd_start, initrd_end - initrd_start);
242         if (err < 0) {
243                 printf("fdt_initrd: %s\n", fdt_strerror(err));
244                 return err;
245         }
246
247         is_u64 = (fdt_address_cells(fdt, 0) == 2);
248
249         err = fdt_setprop_uxx(fdt, nodeoffset, "linux,initrd-start",
250                               (uint64_t)initrd_start, is_u64);
251
252         if (err < 0) {
253                 printf("WARNING: could not set linux,initrd-start %s.\n",
254                        fdt_strerror(err));
255                 return err;
256         }
257
258         err = fdt_setprop_uxx(fdt, nodeoffset, "linux,initrd-end",
259                               (uint64_t)initrd_end, is_u64);
260
261         if (err < 0) {
262                 printf("WARNING: could not set linux,initrd-end %s.\n",
263                        fdt_strerror(err));
264
265                 return err;
266         }
267
268         return 0;
269 }
270
271 /**
272  * board_fdt_chosen_bootargs - boards may override this function to use
273  *                             alternative kernel command line arguments
274  */
275 __weak char *board_fdt_chosen_bootargs(void)
276 {
277         return env_get("bootargs");
278 }
279
280 int fdt_chosen(void *fdt)
281 {
282         int   nodeoffset;
283         int   err;
284         char  *str;             /* used to set string properties */
285
286         err = fdt_check_header(fdt);
287         if (err < 0) {
288                 printf("fdt_chosen: %s\n", fdt_strerror(err));
289                 return err;
290         }
291
292         /* find or create "/chosen" node. */
293         nodeoffset = fdt_find_or_add_subnode(fdt, 0, "chosen");
294         if (nodeoffset < 0)
295                 return nodeoffset;
296
297         str = board_fdt_chosen_bootargs();
298
299         if (str) {
300                 err = fdt_setprop(fdt, nodeoffset, "bootargs", str,
301                                   strlen(str) + 1);
302                 if (err < 0) {
303                         printf("WARNING: could not set bootargs %s.\n",
304                                fdt_strerror(err));
305                         return err;
306                 }
307         }
308
309         /* add u-boot version */
310         err = fdt_setprop(fdt, nodeoffset, "u-boot,version", PLAIN_VERSION,
311                           strlen(PLAIN_VERSION) + 1);
312         if (err < 0) {
313                 printf("WARNING: could not set u-boot,version %s.\n",
314                        fdt_strerror(err));
315                 return err;
316         }
317
318         return fdt_fixup_stdout(fdt, nodeoffset);
319 }
320
321 void do_fixup_by_path(void *fdt, const char *path, const char *prop,
322                       const void *val, int len, int create)
323 {
324 #if defined(DEBUG)
325         int i;
326         debug("Updating property '%s/%s' = ", path, prop);
327         for (i = 0; i < len; i++)
328                 debug(" %.2x", *(u8*)(val+i));
329         debug("\n");
330 #endif
331         int rc = fdt_find_and_setprop(fdt, path, prop, val, len, create);
332         if (rc)
333                 printf("Unable to update property %s:%s, err=%s\n",
334                         path, prop, fdt_strerror(rc));
335 }
336
337 void do_fixup_by_path_u32(void *fdt, const char *path, const char *prop,
338                           u32 val, int create)
339 {
340         fdt32_t tmp = cpu_to_fdt32(val);
341         do_fixup_by_path(fdt, path, prop, &tmp, sizeof(tmp), create);
342 }
343
344 void do_fixup_by_prop(void *fdt,
345                       const char *pname, const void *pval, int plen,
346                       const char *prop, const void *val, int len,
347                       int create)
348 {
349         int off;
350 #if defined(DEBUG)
351         int i;
352         debug("Updating property '%s' = ", prop);
353         for (i = 0; i < len; i++)
354                 debug(" %.2x", *(u8*)(val+i));
355         debug("\n");
356 #endif
357         off = fdt_node_offset_by_prop_value(fdt, -1, pname, pval, plen);
358         while (off != -FDT_ERR_NOTFOUND) {
359                 if (create || (fdt_get_property(fdt, off, prop, NULL) != NULL))
360                         fdt_setprop(fdt, off, prop, val, len);
361                 off = fdt_node_offset_by_prop_value(fdt, off, pname, pval, plen);
362         }
363 }
364
365 void do_fixup_by_prop_u32(void *fdt,
366                           const char *pname, const void *pval, int plen,
367                           const char *prop, u32 val, int create)
368 {
369         fdt32_t tmp = cpu_to_fdt32(val);
370         do_fixup_by_prop(fdt, pname, pval, plen, prop, &tmp, 4, create);
371 }
372
373 void do_fixup_by_compat(void *fdt, const char *compat,
374                         const char *prop, const void *val, int len, int create)
375 {
376         int off = -1;
377 #if defined(DEBUG)
378         int i;
379         debug("Updating property '%s' = ", prop);
380         for (i = 0; i < len; i++)
381                 debug(" %.2x", *(u8*)(val+i));
382         debug("\n");
383 #endif
384         fdt_for_each_node_by_compatible(off, fdt, -1, compat)
385                 if (create || (fdt_get_property(fdt, off, prop, NULL) != NULL))
386                         fdt_setprop(fdt, off, prop, val, len);
387 }
388
389 void do_fixup_by_compat_u32(void *fdt, const char *compat,
390                             const char *prop, u32 val, int create)
391 {
392         fdt32_t tmp = cpu_to_fdt32(val);
393         do_fixup_by_compat(fdt, compat, prop, &tmp, 4, create);
394 }
395
396 #ifdef CONFIG_ARCH_FIXUP_FDT_MEMORY
397 /*
398  * fdt_pack_reg - pack address and size array into the "reg"-suitable stream
399  */
400 static int fdt_pack_reg(const void *fdt, void *buf, u64 *address, u64 *size,
401                         int n)
402 {
403         int i;
404         int address_cells = fdt_address_cells(fdt, 0);
405         int size_cells = fdt_size_cells(fdt, 0);
406         char *p = buf;
407
408         for (i = 0; i < n; i++) {
409                 if (address_cells == 2)
410                         *(fdt64_t *)p = cpu_to_fdt64(address[i]);
411                 else
412                         *(fdt32_t *)p = cpu_to_fdt32(address[i]);
413                 p += 4 * address_cells;
414
415                 if (size_cells == 2)
416                         *(fdt64_t *)p = cpu_to_fdt64(size[i]);
417                 else
418                         *(fdt32_t *)p = cpu_to_fdt32(size[i]);
419                 p += 4 * size_cells;
420         }
421
422         return p - (char *)buf;
423 }
424
425 #if CONFIG_NR_DRAM_BANKS > 4
426 #define MEMORY_BANKS_MAX CONFIG_NR_DRAM_BANKS
427 #else
428 #define MEMORY_BANKS_MAX 4
429 #endif
430
431 /**
432  * fdt_fixup_memory_banks - Update DT memory node
433  * @blob: Pointer to DT blob
434  * @start: Pointer to memory start addresses array
435  * @size: Pointer to memory sizes array
436  * @banks: Number of memory banks
437  *
438  * Return: 0 on success, negative value on failure
439  *
440  * Based on the passed number of banks and arrays, the function is able to
441  * update existing DT memory nodes to match run time detected/changed memory
442  * configuration. Implementation is handling one specific case with only one
443  * memory node where multiple tuples could be added/updated.
444  * The case where multiple memory nodes with a single tuple (base, size) are
445  * used, this function is only updating the first memory node without removing
446  * others.
447  */
448 int fdt_fixup_memory_banks(void *blob, u64 start[], u64 size[], int banks)
449 {
450         int err, nodeoffset;
451         int len, i;
452         u8 tmp[MEMORY_BANKS_MAX * 16]; /* Up to 64-bit address + 64-bit size */
453
454         if (banks > MEMORY_BANKS_MAX) {
455                 printf("%s: num banks %d exceeds hardcoded limit %d."
456                        " Recompile with higher MEMORY_BANKS_MAX?\n",
457                        __FUNCTION__, banks, MEMORY_BANKS_MAX);
458                 return -1;
459         }
460
461         err = fdt_check_header(blob);
462         if (err < 0) {
463                 printf("%s: %s\n", __FUNCTION__, fdt_strerror(err));
464                 return err;
465         }
466
467         /* find or create "/memory" node. */
468         nodeoffset = fdt_find_or_add_subnode(blob, 0, "memory");
469         if (nodeoffset < 0)
470                         return nodeoffset;
471
472         err = fdt_setprop(blob, nodeoffset, "device_type", "memory",
473                         sizeof("memory"));
474         if (err < 0) {
475                 printf("WARNING: could not set %s %s.\n", "device_type",
476                                 fdt_strerror(err));
477                 return err;
478         }
479
480         for (i = 0; i < banks; i++) {
481                 if (start[i] == 0 && size[i] == 0)
482                         break;
483         }
484
485         banks = i;
486
487         if (!banks)
488                 return 0;
489
490         len = fdt_pack_reg(blob, tmp, start, size, banks);
491
492         err = fdt_setprop(blob, nodeoffset, "reg", tmp, len);
493         if (err < 0) {
494                 printf("WARNING: could not set %s %s.\n",
495                                 "reg", fdt_strerror(err));
496                 return err;
497         }
498         return 0;
499 }
500
501 int fdt_set_usable_memory(void *blob, u64 start[], u64 size[], int areas)
502 {
503         int err, nodeoffset;
504         int len;
505         u8 tmp[8 * 16]; /* Up to 64-bit address + 64-bit size */
506
507         if (areas > 8) {
508                 printf("%s: num areas %d exceeds hardcoded limit %d\n",
509                        __func__, areas, 8);
510                 return -1;
511         }
512
513         err = fdt_check_header(blob);
514         if (err < 0) {
515                 printf("%s: %s\n", __func__, fdt_strerror(err));
516                 return err;
517         }
518
519         /* find or create "/memory" node. */
520         nodeoffset = fdt_find_or_add_subnode(blob, 0, "memory");
521         if (nodeoffset < 0)
522                 return nodeoffset;
523
524         len = fdt_pack_reg(blob, tmp, start, size, areas);
525
526         err = fdt_setprop(blob, nodeoffset, "linux,usable-memory", tmp, len);
527         if (err < 0) {
528                 printf("WARNING: could not set %s %s.\n",
529                        "reg", fdt_strerror(err));
530                 return err;
531         }
532
533         return 0;
534 }
535 #endif
536
537 int fdt_fixup_memory(void *blob, u64 start, u64 size)
538 {
539         return fdt_fixup_memory_banks(blob, &start, &size, 1);
540 }
541
542 void fdt_fixup_ethernet(void *fdt)
543 {
544         int i = 0, j, prop;
545         char *tmp, *end;
546         char mac[16];
547         const char *path;
548         unsigned char mac_addr[ARP_HLEN];
549         int offset;
550 #ifdef FDT_SEQ_MACADDR_FROM_ENV
551         int nodeoff;
552         const struct fdt_property *fdt_prop;
553 #endif
554
555         if (fdt_path_offset(fdt, "/aliases") < 0)
556                 return;
557
558         /* Cycle through all aliases */
559         for (prop = 0; ; prop++) {
560                 const char *name;
561
562                 /* FDT might have been edited, recompute the offset */
563                 offset = fdt_first_property_offset(fdt,
564                         fdt_path_offset(fdt, "/aliases"));
565                 /* Select property number 'prop' */
566                 for (j = 0; j < prop; j++)
567                         offset = fdt_next_property_offset(fdt, offset);
568
569                 if (offset < 0)
570                         break;
571
572                 path = fdt_getprop_by_offset(fdt, offset, &name, NULL);
573                 if (!strncmp(name, "ethernet", 8)) {
574                         /* Treat plain "ethernet" same as "ethernet0". */
575                         if (!strcmp(name, "ethernet")
576 #ifdef FDT_SEQ_MACADDR_FROM_ENV
577                          || !strcmp(name, "ethernet0")
578 #endif
579                         )
580                                 i = 0;
581 #ifndef FDT_SEQ_MACADDR_FROM_ENV
582                         else
583                                 i = trailing_strtol(name);
584 #endif
585                         if (i != -1) {
586                                 if (i == 0)
587                                         strcpy(mac, "ethaddr");
588                                 else
589                                         sprintf(mac, "eth%daddr", i);
590                         } else {
591                                 continue;
592                         }
593 #ifdef FDT_SEQ_MACADDR_FROM_ENV
594                         nodeoff = fdt_path_offset(fdt, path);
595                         fdt_prop = fdt_get_property(fdt, nodeoff, "status",
596                                                     NULL);
597                         if (fdt_prop && !strcmp(fdt_prop->data, "disabled"))
598                                 continue;
599                         i++;
600 #endif
601                         tmp = env_get(mac);
602                         if (!tmp)
603                                 continue;
604
605                         for (j = 0; j < 6; j++) {
606                                 mac_addr[j] = tmp ?
607                                               hextoul(tmp, &end) : 0;
608                                 if (tmp)
609                                         tmp = (*end) ? end + 1 : end;
610                         }
611
612                         do_fixup_by_path(fdt, path, "mac-address",
613                                          &mac_addr, 6, 0);
614                         do_fixup_by_path(fdt, path, "local-mac-address",
615                                          &mac_addr, 6, 1);
616                 }
617         }
618 }
619
620 int fdt_record_loadable(void *blob, u32 index, const char *name,
621                         uintptr_t load_addr, u32 size, uintptr_t entry_point,
622                         const char *type, const char *os, const char *arch)
623 {
624         int err, node;
625
626         err = fdt_check_header(blob);
627         if (err < 0) {
628                 printf("%s: %s\n", __func__, fdt_strerror(err));
629                 return err;
630         }
631
632         /* find or create "/fit-images" node */
633         node = fdt_find_or_add_subnode(blob, 0, "fit-images");
634         if (node < 0)
635                 return node;
636
637         /* find or create "/fit-images/<name>" node */
638         node = fdt_find_or_add_subnode(blob, node, name);
639         if (node < 0)
640                 return node;
641
642         fdt_setprop_u64(blob, node, "load", load_addr);
643         if (entry_point != -1)
644                 fdt_setprop_u64(blob, node, "entry", entry_point);
645         fdt_setprop_u32(blob, node, "size", size);
646         if (type)
647                 fdt_setprop_string(blob, node, "type", type);
648         if (os)
649                 fdt_setprop_string(blob, node, "os", os);
650         if (arch)
651                 fdt_setprop_string(blob, node, "arch", arch);
652
653         return node;
654 }
655
656 /* Resize the fdt to its actual size + a bit of padding */
657 int fdt_shrink_to_minimum(void *blob, uint extrasize)
658 {
659         int i;
660         uint64_t addr, size;
661         int total, ret;
662         uint actualsize;
663         int fdt_memrsv = 0;
664
665         if (!blob)
666                 return 0;
667
668         total = fdt_num_mem_rsv(blob);
669         for (i = 0; i < total; i++) {
670                 fdt_get_mem_rsv(blob, i, &addr, &size);
671                 if (addr == (uintptr_t)blob) {
672                         fdt_del_mem_rsv(blob, i);
673                         fdt_memrsv = 1;
674                         break;
675                 }
676         }
677
678         /*
679          * Calculate the actual size of the fdt
680          * plus the size needed for 5 fdt_add_mem_rsv, one
681          * for the fdt itself and 4 for a possible initrd
682          * ((initrd-start + initrd-end) * 2 (name & value))
683          */
684         actualsize = fdt_off_dt_strings(blob) +
685                 fdt_size_dt_strings(blob) + 5 * sizeof(struct fdt_reserve_entry);
686
687         actualsize += extrasize;
688         /* Make it so the fdt ends on a page boundary */
689         actualsize = ALIGN(actualsize + ((uintptr_t)blob & 0xfff), 0x1000);
690         actualsize = actualsize - ((uintptr_t)blob & 0xfff);
691
692         /* Change the fdt header to reflect the correct size */
693         fdt_set_totalsize(blob, actualsize);
694
695         if (fdt_memrsv) {
696                 /* Add the new reservation */
697                 ret = fdt_add_mem_rsv(blob, map_to_sysmem(blob), actualsize);
698                 if (ret < 0)
699                         return ret;
700         }
701
702         return actualsize;
703 }
704
705 /**
706  * fdt_delete_disabled_nodes: Delete all nodes with status == "disabled"
707  *
708  * @blob: ptr to device tree
709  */
710 int fdt_delete_disabled_nodes(void *blob)
711 {
712         while (1) {
713                 int ret, offset;
714
715                 offset = fdt_node_offset_by_prop_value(blob, -1, "status",
716                                                        "disabled", 9);
717                 if (offset < 0)
718                         break;
719
720                 ret = fdt_del_node(blob, offset);
721                 if (ret < 0)
722                         return ret;
723         }
724
725         return 0;
726 }
727
728 #ifdef CONFIG_PCI
729 #define CONFIG_SYS_PCI_NR_INBOUND_WIN 4
730
731 #define FDT_PCI_PREFETCH        (0x40000000)
732 #define FDT_PCI_MEM32           (0x02000000)
733 #define FDT_PCI_IO              (0x01000000)
734 #define FDT_PCI_MEM64           (0x03000000)
735
736 int fdt_pci_dma_ranges(void *blob, int phb_off, struct pci_controller *hose) {
737
738         int addrcell, sizecell, len, r;
739         u32 *dma_range;
740         /* sized based on pci addr cells, size-cells, & address-cells */
741         u32 dma_ranges[(3 + 2 + 2) * CONFIG_SYS_PCI_NR_INBOUND_WIN];
742
743         addrcell = fdt_getprop_u32_default(blob, "/", "#address-cells", 1);
744         sizecell = fdt_getprop_u32_default(blob, "/", "#size-cells", 1);
745
746         dma_range = &dma_ranges[0];
747         for (r = 0; r < hose->region_count; r++) {
748                 u64 bus_start, phys_start, size;
749
750                 /* skip if !PCI_REGION_SYS_MEMORY */
751                 if (!(hose->regions[r].flags & PCI_REGION_SYS_MEMORY))
752                         continue;
753
754                 bus_start = (u64)hose->regions[r].bus_start;
755                 phys_start = (u64)hose->regions[r].phys_start;
756                 size = (u64)hose->regions[r].size;
757
758                 dma_range[0] = 0;
759                 if (size >= 0x100000000ull)
760                         dma_range[0] |= cpu_to_fdt32(FDT_PCI_MEM64);
761                 else
762                         dma_range[0] |= cpu_to_fdt32(FDT_PCI_MEM32);
763                 if (hose->regions[r].flags & PCI_REGION_PREFETCH)
764                         dma_range[0] |= cpu_to_fdt32(FDT_PCI_PREFETCH);
765 #ifdef CONFIG_SYS_PCI_64BIT
766                 dma_range[1] = cpu_to_fdt32(bus_start >> 32);
767 #else
768                 dma_range[1] = 0;
769 #endif
770                 dma_range[2] = cpu_to_fdt32(bus_start & 0xffffffff);
771
772                 if (addrcell == 2) {
773                         dma_range[3] = cpu_to_fdt32(phys_start >> 32);
774                         dma_range[4] = cpu_to_fdt32(phys_start & 0xffffffff);
775                 } else {
776                         dma_range[3] = cpu_to_fdt32(phys_start & 0xffffffff);
777                 }
778
779                 if (sizecell == 2) {
780                         dma_range[3 + addrcell + 0] =
781                                 cpu_to_fdt32(size >> 32);
782                         dma_range[3 + addrcell + 1] =
783                                 cpu_to_fdt32(size & 0xffffffff);
784                 } else {
785                         dma_range[3 + addrcell + 0] =
786                                 cpu_to_fdt32(size & 0xffffffff);
787                 }
788
789                 dma_range += (3 + addrcell + sizecell);
790         }
791
792         len = dma_range - &dma_ranges[0];
793         if (len)
794                 fdt_setprop(blob, phb_off, "dma-ranges", &dma_ranges[0], len*4);
795
796         return 0;
797 }
798 #endif
799
800 int fdt_increase_size(void *fdt, int add_len)
801 {
802         int newlen;
803
804         newlen = fdt_totalsize(fdt) + add_len;
805
806         /* Open in place with a new len */
807         return fdt_open_into(fdt, fdt, newlen);
808 }
809
810 #ifdef CONFIG_FDT_FIXUP_PARTITIONS
811 #include <jffs2/load_kernel.h>
812 #include <mtd_node.h>
813
814 static int fdt_del_subnodes(const void *blob, int parent_offset)
815 {
816         int off, ndepth;
817         int ret;
818
819         for (ndepth = 0, off = fdt_next_node(blob, parent_offset, &ndepth);
820              (off >= 0) && (ndepth > 0);
821              off = fdt_next_node(blob, off, &ndepth)) {
822                 if (ndepth == 1) {
823                         debug("delete %s: offset: %x\n",
824                                 fdt_get_name(blob, off, 0), off);
825                         ret = fdt_del_node((void *)blob, off);
826                         if (ret < 0) {
827                                 printf("Can't delete node: %s\n",
828                                         fdt_strerror(ret));
829                                 return ret;
830                         } else {
831                                 ndepth = 0;
832                                 off = parent_offset;
833                         }
834                 }
835         }
836         return 0;
837 }
838
839 static int fdt_del_partitions(void *blob, int parent_offset)
840 {
841         const void *prop;
842         int ndepth = 0;
843         int off;
844         int ret;
845
846         off = fdt_next_node(blob, parent_offset, &ndepth);
847         if (off > 0 && ndepth == 1) {
848                 prop = fdt_getprop(blob, off, "label", NULL);
849                 if (prop == NULL) {
850                         /*
851                          * Could not find label property, nand {}; node?
852                          * Check subnode, delete partitions there if any.
853                          */
854                         return fdt_del_partitions(blob, off);
855                 } else {
856                         ret = fdt_del_subnodes(blob, parent_offset);
857                         if (ret < 0) {
858                                 printf("Can't remove subnodes: %s\n",
859                                         fdt_strerror(ret));
860                                 return ret;
861                         }
862                 }
863         }
864         return 0;
865 }
866
867 static int fdt_node_set_part_info(void *blob, int parent_offset,
868                                   struct mtd_device *dev)
869 {
870         struct list_head *pentry;
871         struct part_info *part;
872         int off, ndepth = 0;
873         int part_num, ret;
874         int sizecell;
875         char buf[64];
876
877         ret = fdt_del_partitions(blob, parent_offset);
878         if (ret < 0)
879                 return ret;
880
881         /*
882          * Check if size/address is 1 or 2 cells.
883          * We assume #address-cells and #size-cells have same value.
884          */
885         sizecell = fdt_getprop_u32_default_node(blob, parent_offset,
886                                                 0, "#size-cells", 1);
887
888         /*
889          * Check if it is nand {}; subnode, adjust
890          * the offset in this case
891          */
892         off = fdt_next_node(blob, parent_offset, &ndepth);
893         if (off > 0 && ndepth == 1)
894                 parent_offset = off;
895
896         part_num = 0;
897         list_for_each_prev(pentry, &dev->parts) {
898                 int newoff;
899
900                 part = list_entry(pentry, struct part_info, link);
901
902                 debug("%2d: %-20s0x%08llx\t0x%08llx\t%d\n",
903                         part_num, part->name, part->size,
904                         part->offset, part->mask_flags);
905
906                 sprintf(buf, "partition@%llx", part->offset);
907 add_sub:
908                 ret = fdt_add_subnode(blob, parent_offset, buf);
909                 if (ret == -FDT_ERR_NOSPACE) {
910                         ret = fdt_increase_size(blob, 512);
911                         if (!ret)
912                                 goto add_sub;
913                         else
914                                 goto err_size;
915                 } else if (ret < 0) {
916                         printf("Can't add partition node: %s\n",
917                                 fdt_strerror(ret));
918                         return ret;
919                 }
920                 newoff = ret;
921
922                 /* Check MTD_WRITEABLE_CMD flag */
923                 if (part->mask_flags & 1) {
924 add_ro:
925                         ret = fdt_setprop(blob, newoff, "read_only", NULL, 0);
926                         if (ret == -FDT_ERR_NOSPACE) {
927                                 ret = fdt_increase_size(blob, 512);
928                                 if (!ret)
929                                         goto add_ro;
930                                 else
931                                         goto err_size;
932                         } else if (ret < 0)
933                                 goto err_prop;
934                 }
935
936 add_reg:
937                 if (sizecell == 2) {
938                         ret = fdt_setprop_u64(blob, newoff,
939                                               "reg", part->offset);
940                         if (!ret)
941                                 ret = fdt_appendprop_u64(blob, newoff,
942                                                          "reg", part->size);
943                 } else {
944                         ret = fdt_setprop_u32(blob, newoff,
945                                               "reg", part->offset);
946                         if (!ret)
947                                 ret = fdt_appendprop_u32(blob, newoff,
948                                                          "reg", part->size);
949                 }
950
951                 if (ret == -FDT_ERR_NOSPACE) {
952                         ret = fdt_increase_size(blob, 512);
953                         if (!ret)
954                                 goto add_reg;
955                         else
956                                 goto err_size;
957                 } else if (ret < 0)
958                         goto err_prop;
959
960 add_label:
961                 ret = fdt_setprop_string(blob, newoff, "label", part->name);
962                 if (ret == -FDT_ERR_NOSPACE) {
963                         ret = fdt_increase_size(blob, 512);
964                         if (!ret)
965                                 goto add_label;
966                         else
967                                 goto err_size;
968                 } else if (ret < 0)
969                         goto err_prop;
970
971                 part_num++;
972         }
973         return 0;
974 err_size:
975         printf("Can't increase blob size: %s\n", fdt_strerror(ret));
976         return ret;
977 err_prop:
978         printf("Can't add property: %s\n", fdt_strerror(ret));
979         return ret;
980 }
981
982 /*
983  * Update partitions in nor/nand nodes using info from
984  * mtdparts environment variable. The nodes to update are
985  * specified by node_info structure which contains mtd device
986  * type and compatible string: E. g. the board code in
987  * ft_board_setup() could use:
988  *
989  *      struct node_info nodes[] = {
990  *              { "fsl,mpc5121-nfc",    MTD_DEV_TYPE_NAND, },
991  *              { "cfi-flash",          MTD_DEV_TYPE_NOR,  },
992  *      };
993  *
994  *      fdt_fixup_mtdparts(blob, nodes, ARRAY_SIZE(nodes));
995  */
996 void fdt_fixup_mtdparts(void *blob, const struct node_info *node_info,
997                         int node_info_size)
998 {
999         struct mtd_device *dev;
1000         int i, idx;
1001         int noff, parts;
1002         bool inited = false;
1003
1004         for (i = 0; i < node_info_size; i++) {
1005                 idx = 0;
1006
1007                 fdt_for_each_node_by_compatible(noff, blob, -1,
1008                                                 node_info[i].compat) {
1009                         const char *prop;
1010
1011                         prop = fdt_getprop(blob, noff, "status", NULL);
1012                         if (prop && !strcmp(prop, "disabled"))
1013                                 continue;
1014
1015                         debug("%s: %s, mtd dev type %d\n",
1016                                 fdt_get_name(blob, noff, 0),
1017                                 node_info[i].compat, node_info[i].type);
1018
1019                         if (!inited) {
1020                                 if (mtdparts_init() != 0)
1021                                         return;
1022                                 inited = true;
1023                         }
1024
1025                         dev = device_find(node_info[i].type, idx++);
1026                         if (dev) {
1027                                 parts = fdt_subnode_offset(blob, noff,
1028                                                            "partitions");
1029                                 if (parts < 0)
1030                                         parts = noff;
1031
1032                                 if (fdt_node_set_part_info(blob, parts, dev))
1033                                         return; /* return on error */
1034                         }
1035                 }
1036         }
1037 }
1038 #endif
1039
1040 void fdt_del_node_and_alias(void *blob, const char *alias)
1041 {
1042         int off = fdt_path_offset(blob, alias);
1043
1044         if (off < 0)
1045                 return;
1046
1047         fdt_del_node(blob, off);
1048
1049         off = fdt_path_offset(blob, "/aliases");
1050         fdt_delprop(blob, off, alias);
1051 }
1052
1053 /* Max address size we deal with */
1054 #define OF_MAX_ADDR_CELLS       4
1055 #define OF_BAD_ADDR     FDT_ADDR_T_NONE
1056 #define OF_CHECK_COUNTS(na, ns) ((na) > 0 && (na) <= OF_MAX_ADDR_CELLS && \
1057                         (ns) > 0)
1058
1059 /* Debug utility */
1060 #ifdef DEBUG
1061 static void of_dump_addr(const char *s, const fdt32_t *addr, int na)
1062 {
1063         printf("%s", s);
1064         while(na--)
1065                 printf(" %08x", *(addr++));
1066         printf("\n");
1067 }
1068 #else
1069 static void of_dump_addr(const char *s, const fdt32_t *addr, int na) { }
1070 #endif
1071
1072 /**
1073  * struct of_bus - Callbacks for bus specific translators
1074  * @name:       A string used to identify this bus in debug output.
1075  * @addresses:  The name of the DT property from which addresses are
1076  *              to be read, typically "reg".
1077  * @match:      Return non-zero if the node whose parent is at
1078  *              parentoffset in the FDT blob corresponds to a bus
1079  *              of this type, otherwise return zero. If NULL a match
1080  *              is assumed.
1081  * @count_cells:Count how many cells (be32 values) a node whose parent
1082  *              is at parentoffset in the FDT blob will require to
1083  *              represent its address (written to *addrc) & size
1084  *              (written to *sizec).
1085  * @map:        Map the address addr from the address space of this
1086  *              bus to that of its parent, making use of the ranges
1087  *              read from DT to an array at range. na and ns are the
1088  *              number of cells (be32 values) used to hold and address
1089  *              or size, respectively, for this bus. pna is the number
1090  *              of cells used to hold an address for the parent bus.
1091  *              Returns the address in the address space of the parent
1092  *              bus.
1093  * @translate:  Update the value of the address cells at addr within an
1094  *              FDT by adding offset to it. na specifies the number of
1095  *              cells used to hold the address being translated. Returns
1096  *              zero on success, non-zero on error.
1097  *
1098  * Each bus type will include a struct of_bus in the of_busses array,
1099  * providing implementations of some or all of the functions used to
1100  * match the bus & handle address translation for its children.
1101  */
1102 struct of_bus {
1103         const char      *name;
1104         const char      *addresses;
1105         int             (*match)(const void *blob, int parentoffset);
1106         void            (*count_cells)(const void *blob, int parentoffset,
1107                                 int *addrc, int *sizec);
1108         u64             (*map)(fdt32_t *addr, const fdt32_t *range,
1109                                 int na, int ns, int pna);
1110         int             (*translate)(fdt32_t *addr, u64 offset, int na);
1111 };
1112
1113 /* Default translator (generic bus) */
1114 void fdt_support_default_count_cells(const void *blob, int parentoffset,
1115                                         int *addrc, int *sizec)
1116 {
1117         const fdt32_t *prop;
1118
1119         if (addrc)
1120                 *addrc = fdt_address_cells(blob, parentoffset);
1121
1122         if (sizec) {
1123                 prop = fdt_getprop(blob, parentoffset, "#size-cells", NULL);
1124                 if (prop)
1125                         *sizec = be32_to_cpup(prop);
1126                 else
1127                         *sizec = 1;
1128         }
1129 }
1130
1131 static u64 of_bus_default_map(fdt32_t *addr, const fdt32_t *range,
1132                 int na, int ns, int pna)
1133 {
1134         u64 cp, s, da;
1135
1136         cp = fdt_read_number(range, na);
1137         s  = fdt_read_number(range + na + pna, ns);
1138         da = fdt_read_number(addr, na);
1139
1140         debug("OF: default map, cp=%llx, s=%llx, da=%llx\n", cp, s, da);
1141
1142         if (da < cp || da >= (cp + s))
1143                 return OF_BAD_ADDR;
1144         return da - cp;
1145 }
1146
1147 static int of_bus_default_translate(fdt32_t *addr, u64 offset, int na)
1148 {
1149         u64 a = fdt_read_number(addr, na);
1150         memset(addr, 0, na * 4);
1151         a += offset;
1152         if (na > 1)
1153                 addr[na - 2] = cpu_to_fdt32(a >> 32);
1154         addr[na - 1] = cpu_to_fdt32(a & 0xffffffffu);
1155
1156         return 0;
1157 }
1158
1159 #ifdef CONFIG_OF_ISA_BUS
1160
1161 /* ISA bus translator */
1162 static int of_bus_isa_match(const void *blob, int parentoffset)
1163 {
1164         const char *name;
1165
1166         name = fdt_get_name(blob, parentoffset, NULL);
1167         if (!name)
1168                 return 0;
1169
1170         return !strcmp(name, "isa");
1171 }
1172
1173 static void of_bus_isa_count_cells(const void *blob, int parentoffset,
1174                                    int *addrc, int *sizec)
1175 {
1176         if (addrc)
1177                 *addrc = 2;
1178         if (sizec)
1179                 *sizec = 1;
1180 }
1181
1182 static u64 of_bus_isa_map(fdt32_t *addr, const fdt32_t *range,
1183                           int na, int ns, int pna)
1184 {
1185         u64 cp, s, da;
1186
1187         /* Check address type match */
1188         if ((addr[0] ^ range[0]) & cpu_to_be32(1))
1189                 return OF_BAD_ADDR;
1190
1191         cp = fdt_read_number(range + 1, na - 1);
1192         s  = fdt_read_number(range + na + pna, ns);
1193         da = fdt_read_number(addr + 1, na - 1);
1194
1195         debug("OF: ISA map, cp=%llx, s=%llx, da=%llx\n", cp, s, da);
1196
1197         if (da < cp || da >= (cp + s))
1198                 return OF_BAD_ADDR;
1199         return da - cp;
1200 }
1201
1202 static int of_bus_isa_translate(fdt32_t *addr, u64 offset, int na)
1203 {
1204         return of_bus_default_translate(addr + 1, offset, na - 1);
1205 }
1206
1207 #endif /* CONFIG_OF_ISA_BUS */
1208
1209 /* Array of bus specific translators */
1210 static struct of_bus of_busses[] = {
1211 #ifdef CONFIG_OF_ISA_BUS
1212         /* ISA */
1213         {
1214                 .name = "isa",
1215                 .addresses = "reg",
1216                 .match = of_bus_isa_match,
1217                 .count_cells = of_bus_isa_count_cells,
1218                 .map = of_bus_isa_map,
1219                 .translate = of_bus_isa_translate,
1220         },
1221 #endif /* CONFIG_OF_ISA_BUS */
1222         /* Default */
1223         {
1224                 .name = "default",
1225                 .addresses = "reg",
1226                 .count_cells = fdt_support_default_count_cells,
1227                 .map = of_bus_default_map,
1228                 .translate = of_bus_default_translate,
1229         },
1230 };
1231
1232 static struct of_bus *of_match_bus(const void *blob, int parentoffset)
1233 {
1234         struct of_bus *bus;
1235
1236         if (ARRAY_SIZE(of_busses) == 1)
1237                 return of_busses;
1238
1239         for (bus = of_busses; bus; bus++) {
1240                 if (!bus->match || bus->match(blob, parentoffset))
1241                         return bus;
1242         }
1243
1244         /*
1245          * We should always have matched the default bus at least, since
1246          * it has a NULL match field. If we didn't then it somehow isn't
1247          * in the of_busses array or something equally catastrophic has
1248          * gone wrong.
1249          */
1250         assert(0);
1251         return NULL;
1252 }
1253
1254 static int of_translate_one(const void *blob, int parent, struct of_bus *bus,
1255                             struct of_bus *pbus, fdt32_t *addr,
1256                             int na, int ns, int pna, const char *rprop)
1257 {
1258         const fdt32_t *ranges;
1259         int rlen;
1260         int rone;
1261         u64 offset = OF_BAD_ADDR;
1262
1263         /* Normally, an absence of a "ranges" property means we are
1264          * crossing a non-translatable boundary, and thus the addresses
1265          * below the current not cannot be converted to CPU physical ones.
1266          * Unfortunately, while this is very clear in the spec, it's not
1267          * what Apple understood, and they do have things like /uni-n or
1268          * /ht nodes with no "ranges" property and a lot of perfectly
1269          * useable mapped devices below them. Thus we treat the absence of
1270          * "ranges" as equivalent to an empty "ranges" property which means
1271          * a 1:1 translation at that level. It's up to the caller not to try
1272          * to translate addresses that aren't supposed to be translated in
1273          * the first place. --BenH.
1274          */
1275         ranges = fdt_getprop(blob, parent, rprop, &rlen);
1276         if (ranges == NULL || rlen == 0) {
1277                 offset = fdt_read_number(addr, na);
1278                 memset(addr, 0, pna * 4);
1279                 debug("OF: no ranges, 1:1 translation\n");
1280                 goto finish;
1281         }
1282
1283         debug("OF: walking ranges...\n");
1284
1285         /* Now walk through the ranges */
1286         rlen /= 4;
1287         rone = na + pna + ns;
1288         for (; rlen >= rone; rlen -= rone, ranges += rone) {
1289                 offset = bus->map(addr, ranges, na, ns, pna);
1290                 if (offset != OF_BAD_ADDR)
1291                         break;
1292         }
1293         if (offset == OF_BAD_ADDR) {
1294                 debug("OF: not found !\n");
1295                 return 1;
1296         }
1297         memcpy(addr, ranges + na, 4 * pna);
1298
1299  finish:
1300         of_dump_addr("OF: parent translation for:", addr, pna);
1301         debug("OF: with offset: %llu\n", offset);
1302
1303         /* Translate it into parent bus space */
1304         return pbus->translate(addr, offset, pna);
1305 }
1306
1307 /*
1308  * Translate an address from the device-tree into a CPU physical address,
1309  * this walks up the tree and applies the various bus mappings on the
1310  * way.
1311  *
1312  * Note: We consider that crossing any level with #size-cells == 0 to mean
1313  * that translation is impossible (that is we are not dealing with a value
1314  * that can be mapped to a cpu physical address). This is not really specified
1315  * that way, but this is traditionally the way IBM at least do things
1316  */
1317 static u64 __of_translate_address(const void *blob, int node_offset,
1318                                   const fdt32_t *in_addr, const char *rprop)
1319 {
1320         int parent;
1321         struct of_bus *bus, *pbus;
1322         fdt32_t addr[OF_MAX_ADDR_CELLS];
1323         int na, ns, pna, pns;
1324         u64 result = OF_BAD_ADDR;
1325
1326         debug("OF: ** translation for device %s **\n",
1327                 fdt_get_name(blob, node_offset, NULL));
1328
1329         /* Get parent & match bus type */
1330         parent = fdt_parent_offset(blob, node_offset);
1331         if (parent < 0)
1332                 goto bail;
1333         bus = of_match_bus(blob, parent);
1334
1335         /* Cound address cells & copy address locally */
1336         bus->count_cells(blob, parent, &na, &ns);
1337         if (!OF_CHECK_COUNTS(na, ns)) {
1338                 printf("%s: Bad cell count for %s\n", __FUNCTION__,
1339                        fdt_get_name(blob, node_offset, NULL));
1340                 goto bail;
1341         }
1342         memcpy(addr, in_addr, na * 4);
1343
1344         debug("OF: bus is %s (na=%d, ns=%d) on %s\n",
1345             bus->name, na, ns, fdt_get_name(blob, parent, NULL));
1346         of_dump_addr("OF: translating address:", addr, na);
1347
1348         /* Translate */
1349         for (;;) {
1350                 /* Switch to parent bus */
1351                 node_offset = parent;
1352                 parent = fdt_parent_offset(blob, node_offset);
1353
1354                 /* If root, we have finished */
1355                 if (parent < 0) {
1356                         debug("OF: reached root node\n");
1357                         result = fdt_read_number(addr, na);
1358                         break;
1359                 }
1360
1361                 /* Get new parent bus and counts */
1362                 pbus = of_match_bus(blob, parent);
1363                 pbus->count_cells(blob, parent, &pna, &pns);
1364                 if (!OF_CHECK_COUNTS(pna, pns)) {
1365                         printf("%s: Bad cell count for %s\n", __FUNCTION__,
1366                                 fdt_get_name(blob, node_offset, NULL));
1367                         break;
1368                 }
1369
1370                 debug("OF: parent bus is %s (na=%d, ns=%d) on %s\n",
1371                     pbus->name, pna, pns, fdt_get_name(blob, parent, NULL));
1372
1373                 /* Apply bus translation */
1374                 if (of_translate_one(blob, node_offset, bus, pbus,
1375                                         addr, na, ns, pna, rprop))
1376                         break;
1377
1378                 /* Complete the move up one level */
1379                 na = pna;
1380                 ns = pns;
1381                 bus = pbus;
1382
1383                 of_dump_addr("OF: one level translation:", addr, na);
1384         }
1385  bail:
1386
1387         return result;
1388 }
1389
1390 u64 fdt_translate_address(const void *blob, int node_offset,
1391                           const fdt32_t *in_addr)
1392 {
1393         return __of_translate_address(blob, node_offset, in_addr, "ranges");
1394 }
1395
1396 u64 fdt_translate_dma_address(const void *blob, int node_offset,
1397                               const fdt32_t *in_addr)
1398 {
1399         return __of_translate_address(blob, node_offset, in_addr, "dma-ranges");
1400 }
1401
1402 int fdt_get_dma_range(const void *blob, int node, phys_addr_t *cpu,
1403                       dma_addr_t *bus, u64 *size)
1404 {
1405         bool found_dma_ranges = false;
1406         struct of_bus *bus_node;
1407         const fdt32_t *ranges;
1408         int na, ns, pna, pns;
1409         int parent = node;
1410         int ret = 0;
1411         int len;
1412
1413         /* Find the closest dma-ranges property */
1414         while (parent >= 0) {
1415                 ranges = fdt_getprop(blob, parent, "dma-ranges", &len);
1416
1417                 /* Ignore empty ranges, they imply no translation required */
1418                 if (ranges && len > 0)
1419                         break;
1420
1421                 /* Once we find 'dma-ranges', then a missing one is an error */
1422                 if (found_dma_ranges && !ranges) {
1423                         ret = -EINVAL;
1424                         goto out;
1425                 }
1426
1427                 if (ranges)
1428                         found_dma_ranges = true;
1429
1430                 parent = fdt_parent_offset(blob, parent);
1431         }
1432
1433         if (!ranges || parent < 0) {
1434                 debug("no dma-ranges found for node %s\n",
1435                       fdt_get_name(blob, node, NULL));
1436                 ret = -ENOENT;
1437                 goto out;
1438         }
1439
1440         /* switch to that node */
1441         node = parent;
1442         parent = fdt_parent_offset(blob, node);
1443         if (parent < 0) {
1444                 printf("Found dma-ranges in root node, shouldn't happen\n");
1445                 ret = -EINVAL;
1446                 goto out;
1447         }
1448
1449         /* Get the address sizes both for the bus and its parent */
1450         bus_node = of_match_bus(blob, node);
1451         bus_node->count_cells(blob, node, &na, &ns);
1452         if (!OF_CHECK_COUNTS(na, ns)) {
1453                 printf("%s: Bad cell count for %s\n", __FUNCTION__,
1454                        fdt_get_name(blob, node, NULL));
1455                 return -EINVAL;
1456                 goto out;
1457         }
1458
1459         bus_node = of_match_bus(blob, parent);
1460         bus_node->count_cells(blob, parent, &pna, &pns);
1461         if (!OF_CHECK_COUNTS(pna, pns)) {
1462                 printf("%s: Bad cell count for %s\n", __FUNCTION__,
1463                        fdt_get_name(blob, parent, NULL));
1464                 return -EINVAL;
1465                 goto out;
1466         }
1467
1468         *bus = fdt_read_number(ranges, na);
1469         *cpu = fdt_translate_dma_address(blob, node, ranges + na);
1470         *size = fdt_read_number(ranges + na + pna, ns);
1471 out:
1472         return ret;
1473 }
1474
1475 /**
1476  * fdt_node_offset_by_compat_reg: Find a node that matches compatiable and
1477  * who's reg property matches a physical cpu address
1478  *
1479  * @blob: ptr to device tree
1480  * @compat: compatiable string to match
1481  * @compat_off: property name
1482  *
1483  */
1484 int fdt_node_offset_by_compat_reg(void *blob, const char *compat,
1485                                         phys_addr_t compat_off)
1486 {
1487         int len, off;
1488
1489         fdt_for_each_node_by_compatible(off, blob, -1, compat) {
1490                 const fdt32_t *reg = fdt_getprop(blob, off, "reg", &len);
1491                 if (reg && compat_off == fdt_translate_address(blob, off, reg))
1492                         return off;
1493         }
1494
1495         return -FDT_ERR_NOTFOUND;
1496 }
1497
1498 static int vnode_offset_by_pathf(void *blob, const char *fmt, va_list ap)
1499 {
1500         char path[512];
1501         int len;
1502
1503         len = vsnprintf(path, sizeof(path), fmt, ap);
1504         if (len < 0 || len + 1 > sizeof(path))
1505                 return -FDT_ERR_NOSPACE;
1506
1507         return fdt_path_offset(blob, path);
1508 }
1509
1510 /**
1511  * fdt_node_offset_by_pathf: Find node offset by sprintf formatted path
1512  *
1513  * @blob: ptr to device tree
1514  * @fmt: path format
1515  * @ap: vsnprintf arguments
1516  */
1517 int fdt_node_offset_by_pathf(void *blob, const char *fmt, ...)
1518 {
1519         va_list ap;
1520         int res;
1521
1522         va_start(ap, fmt);
1523         res = vnode_offset_by_pathf(blob, fmt, ap);
1524         va_end(ap);
1525
1526         return res;
1527 }
1528
1529 /*
1530  * fdt_set_phandle: Create a phandle property for the given node
1531  *
1532  * @fdt: ptr to device tree
1533  * @nodeoffset: node to update
1534  * @phandle: phandle value to set (must be unique)
1535  */
1536 int fdt_set_phandle(void *fdt, int nodeoffset, uint32_t phandle)
1537 {
1538         int ret;
1539
1540 #ifdef DEBUG
1541         int off = fdt_node_offset_by_phandle(fdt, phandle);
1542
1543         if ((off >= 0) && (off != nodeoffset)) {
1544                 char buf[64];
1545
1546                 fdt_get_path(fdt, nodeoffset, buf, sizeof(buf));
1547                 printf("Trying to update node %s with phandle %u ",
1548                        buf, phandle);
1549
1550                 fdt_get_path(fdt, off, buf, sizeof(buf));
1551                 printf("that already exists in node %s.\n", buf);
1552                 return -FDT_ERR_BADPHANDLE;
1553         }
1554 #endif
1555
1556         ret = fdt_setprop_cell(fdt, nodeoffset, "phandle", phandle);
1557
1558         return ret;
1559 }
1560
1561 /*
1562  * fdt_create_phandle: Get or create a phandle property for the given node
1563  *
1564  * @fdt: ptr to device tree
1565  * @nodeoffset: node to update
1566  */
1567 unsigned int fdt_create_phandle(void *fdt, int nodeoffset)
1568 {
1569         /* see if there is a phandle already */
1570         uint32_t phandle = fdt_get_phandle(fdt, nodeoffset);
1571
1572         /* if we got 0, means no phandle so create one */
1573         if (phandle == 0) {
1574                 int ret;
1575
1576                 ret = fdt_generate_phandle(fdt, &phandle);
1577                 if (ret < 0) {
1578                         printf("Can't generate phandle: %s\n",
1579                                fdt_strerror(ret));
1580                         return 0;
1581                 }
1582
1583                 ret = fdt_set_phandle(fdt, nodeoffset, phandle);
1584                 if (ret < 0) {
1585                         printf("Can't set phandle %u: %s\n", phandle,
1586                                fdt_strerror(ret));
1587                         return 0;
1588                 }
1589         }
1590
1591         return phandle;
1592 }
1593
1594 /**
1595  * fdt_create_phandle_by_compatible: Get or create a phandle for first node with
1596  *                                   given compatible
1597  *
1598  * @fdt: ptr to device tree
1599  * @compat: node's compatible string
1600  */
1601 unsigned int fdt_create_phandle_by_compatible(void *fdt, const char *compat)
1602 {
1603         int offset = fdt_node_offset_by_compatible(fdt, -1, compat);
1604
1605         if (offset < 0) {
1606                 printf("Can't find node with compatible \"%s\": %s\n", compat,
1607                        fdt_strerror(offset));
1608                 return 0;
1609         }
1610
1611         return fdt_create_phandle(fdt, offset);
1612 }
1613
1614 /**
1615  * fdt_create_phandle_by_pathf: Get or create a phandle for node given by
1616  *                              sprintf-formatted path
1617  *
1618  * @fdt: ptr to device tree
1619  * @fmt, ...: path format string and arguments to pass to sprintf
1620  */
1621 unsigned int fdt_create_phandle_by_pathf(void *fdt, const char *fmt, ...)
1622 {
1623         va_list ap;
1624         int offset;
1625
1626         va_start(ap, fmt);
1627         offset = vnode_offset_by_pathf(fdt, fmt, ap);
1628         va_end(ap);
1629
1630         if (offset < 0) {
1631                 printf("Can't find node by given path: %s\n",
1632                        fdt_strerror(offset));
1633                 return 0;
1634         }
1635
1636         return fdt_create_phandle(fdt, offset);
1637 }
1638
1639 /*
1640  * fdt_set_node_status: Set status for the given node
1641  *
1642  * @fdt: ptr to device tree
1643  * @nodeoffset: node to update
1644  * @status: FDT_STATUS_OKAY, FDT_STATUS_DISABLED, FDT_STATUS_FAIL
1645  */
1646 int fdt_set_node_status(void *fdt, int nodeoffset, enum fdt_status status)
1647 {
1648         int ret = 0;
1649
1650         if (nodeoffset < 0)
1651                 return nodeoffset;
1652
1653         switch (status) {
1654         case FDT_STATUS_OKAY:
1655                 ret = fdt_setprop_string(fdt, nodeoffset, "status", "okay");
1656                 break;
1657         case FDT_STATUS_DISABLED:
1658                 ret = fdt_setprop_string(fdt, nodeoffset, "status", "disabled");
1659                 break;
1660         case FDT_STATUS_FAIL:
1661                 ret = fdt_setprop_string(fdt, nodeoffset, "status", "fail");
1662                 break;
1663         default:
1664                 printf("Invalid fdt status: %x\n", status);
1665                 ret = -1;
1666                 break;
1667         }
1668
1669         return ret;
1670 }
1671
1672 /*
1673  * fdt_set_status_by_alias: Set status for the given node given an alias
1674  *
1675  * @fdt: ptr to device tree
1676  * @alias: alias of node to update
1677  * @status: FDT_STATUS_OKAY, FDT_STATUS_DISABLED, FDT_STATUS_FAIL
1678  */
1679 int fdt_set_status_by_alias(void *fdt, const char* alias,
1680                             enum fdt_status status)
1681 {
1682         int offset = fdt_path_offset(fdt, alias);
1683
1684         return fdt_set_node_status(fdt, offset, status);
1685 }
1686
1687 /**
1688  * fdt_set_status_by_compatible: Set node status for first node with given
1689  *                               compatible
1690  *
1691  * @fdt: ptr to device tree
1692  * @compat: node's compatible string
1693  * @status: FDT_STATUS_OKAY, FDT_STATUS_DISABLED, FDT_STATUS_FAIL
1694  */
1695 int fdt_set_status_by_compatible(void *fdt, const char *compat,
1696                                  enum fdt_status status)
1697 {
1698         int offset = fdt_node_offset_by_compatible(fdt, -1, compat);
1699
1700         if (offset < 0)
1701                 return offset;
1702
1703         return fdt_set_node_status(fdt, offset, status);
1704 }
1705
1706 /**
1707  * fdt_set_status_by_pathf: Set node status for node given by sprintf-formatted
1708  *                          path
1709  *
1710  * @fdt: ptr to device tree
1711  * @status: FDT_STATUS_OKAY, FDT_STATUS_DISABLED, FDT_STATUS_FAIL
1712  * @fmt, ...: path format string and arguments to pass to sprintf
1713  */
1714 int fdt_set_status_by_pathf(void *fdt, enum fdt_status status, const char *fmt,
1715                             ...)
1716 {
1717         va_list ap;
1718         int offset;
1719
1720         va_start(ap, fmt);
1721         offset = vnode_offset_by_pathf(fdt, fmt, ap);
1722         va_end(ap);
1723
1724         if (offset < 0)
1725                 return offset;
1726
1727         return fdt_set_node_status(fdt, offset, status);
1728 }
1729
1730 #if defined(CONFIG_LCD)
1731 int fdt_add_edid(void *blob, const char *compat, unsigned char *edid_buf)
1732 {
1733         int noff;
1734         int ret;
1735
1736         noff = fdt_node_offset_by_compatible(blob, -1, compat);
1737         if (noff != -FDT_ERR_NOTFOUND) {
1738                 debug("%s: %s\n", fdt_get_name(blob, noff, 0), compat);
1739 add_edid:
1740                 ret = fdt_setprop(blob, noff, "edid", edid_buf, 128);
1741                 if (ret == -FDT_ERR_NOSPACE) {
1742                         ret = fdt_increase_size(blob, 512);
1743                         if (!ret)
1744                                 goto add_edid;
1745                         else
1746                                 goto err_size;
1747                 } else if (ret < 0) {
1748                         printf("Can't add property: %s\n", fdt_strerror(ret));
1749                         return ret;
1750                 }
1751         }
1752         return 0;
1753 err_size:
1754         printf("Can't increase blob size: %s\n", fdt_strerror(ret));
1755         return ret;
1756 }
1757 #endif
1758
1759 /*
1760  * Verify the physical address of device tree node for a given alias
1761  *
1762  * This function locates the device tree node of a given alias, and then
1763  * verifies that the physical address of that device matches the given
1764  * parameter.  It displays a message if there is a mismatch.
1765  *
1766  * Returns 1 on success, 0 on failure
1767  */
1768 int fdt_verify_alias_address(void *fdt, int anode, const char *alias, u64 addr)
1769 {
1770         const char *path;
1771         const fdt32_t *reg;
1772         int node, len;
1773         u64 dt_addr;
1774
1775         path = fdt_getprop(fdt, anode, alias, NULL);
1776         if (!path) {
1777                 /* If there's no such alias, then it's not a failure */
1778                 return 1;
1779         }
1780
1781         node = fdt_path_offset(fdt, path);
1782         if (node < 0) {
1783                 printf("Warning: device tree alias '%s' points to invalid "
1784                        "node %s.\n", alias, path);
1785                 return 0;
1786         }
1787
1788         reg = fdt_getprop(fdt, node, "reg", &len);
1789         if (!reg) {
1790                 printf("Warning: device tree node '%s' has no address.\n",
1791                        path);
1792                 return 0;
1793         }
1794
1795         dt_addr = fdt_translate_address(fdt, node, reg);
1796         if (addr != dt_addr) {
1797                 printf("Warning: U-Boot configured device %s at address %llu,\n"
1798                        "but the device tree has it address %llx.\n",
1799                        alias, addr, dt_addr);
1800                 return 0;
1801         }
1802
1803         return 1;
1804 }
1805
1806 /*
1807  * Returns the base address of an SOC or PCI node
1808  */
1809 u64 fdt_get_base_address(const void *fdt, int node)
1810 {
1811         int size;
1812         const fdt32_t *prop;
1813
1814         prop = fdt_getprop(fdt, node, "reg", &size);
1815
1816         return prop ? fdt_translate_address(fdt, node, prop) : OF_BAD_ADDR;
1817 }
1818
1819 /*
1820  * Read a property of size <prop_len>. Currently only supports 1 or 2 cells,
1821  * or 3 cells specially for a PCI address.
1822  */
1823 static int fdt_read_prop(const fdt32_t *prop, int prop_len, int cell_off,
1824                          uint64_t *val, int cells)
1825 {
1826         const fdt32_t *prop32;
1827         const unaligned_fdt64_t *prop64;
1828
1829         if ((cell_off + cells) > prop_len)
1830                 return -FDT_ERR_NOSPACE;
1831
1832         prop32 = &prop[cell_off];
1833
1834         /*
1835          * Special handling for PCI address in PCI bus <ranges>
1836          *
1837          * PCI child address is made up of 3 cells. Advance the cell offset
1838          * by 1 so that the PCI child address can be correctly read.
1839          */
1840         if (cells == 3)
1841                 cell_off += 1;
1842         prop64 = (const fdt64_t *)&prop[cell_off];
1843
1844         switch (cells) {
1845         case 1:
1846                 *val = fdt32_to_cpu(*prop32);
1847                 break;
1848         case 2:
1849         case 3:
1850                 *val = fdt64_to_cpu(*prop64);
1851                 break;
1852         default:
1853                 return -FDT_ERR_NOSPACE;
1854         }
1855
1856         return 0;
1857 }
1858
1859 /**
1860  * fdt_read_range - Read a node's n'th range property
1861  *
1862  * @fdt: ptr to device tree
1863  * @node: offset of node
1864  * @n: range index
1865  * @child_addr: pointer to storage for the "child address" field
1866  * @addr: pointer to storage for the CPU view translated physical start
1867  * @len: pointer to storage for the range length
1868  *
1869  * Convenience function that reads and interprets a specific range out of
1870  * a number of the "ranges" property array.
1871  */
1872 int fdt_read_range(void *fdt, int node, int n, uint64_t *child_addr,
1873                    uint64_t *addr, uint64_t *len)
1874 {
1875         int pnode = fdt_parent_offset(fdt, node);
1876         const fdt32_t *ranges;
1877         int pacells;
1878         int acells;
1879         int scells;
1880         int ranges_len;
1881         int cell = 0;
1882         int r = 0;
1883
1884         /*
1885          * The "ranges" property is an array of
1886          * { <child address> <parent address> <size in child address space> }
1887          *
1888          * All 3 elements can span a diffent number of cells. Fetch their size.
1889          */
1890         pacells = fdt_getprop_u32_default_node(fdt, pnode, 0, "#address-cells", 1);
1891         acells = fdt_getprop_u32_default_node(fdt, node, 0, "#address-cells", 1);
1892         scells = fdt_getprop_u32_default_node(fdt, node, 0, "#size-cells", 1);
1893
1894         /* Now try to get the ranges property */
1895         ranges = fdt_getprop(fdt, node, "ranges", &ranges_len);
1896         if (!ranges)
1897                 return -FDT_ERR_NOTFOUND;
1898         ranges_len /= sizeof(uint32_t);
1899
1900         /* Jump to the n'th entry */
1901         cell = n * (pacells + acells + scells);
1902
1903         /* Read <child address> */
1904         if (child_addr) {
1905                 r = fdt_read_prop(ranges, ranges_len, cell, child_addr,
1906                                   acells);
1907                 if (r)
1908                         return r;
1909         }
1910         cell += acells;
1911
1912         /* Read <parent address> */
1913         if (addr)
1914                 *addr = fdt_translate_address(fdt, node, ranges + cell);
1915         cell += pacells;
1916
1917         /* Read <size in child address space> */
1918         if (len) {
1919                 r = fdt_read_prop(ranges, ranges_len, cell, len, scells);
1920                 if (r)
1921                         return r;
1922         }
1923
1924         return 0;
1925 }
1926
1927 /**
1928  * fdt_setup_simplefb_node - Fill and enable a simplefb node
1929  *
1930  * @fdt: ptr to device tree
1931  * @node: offset of the simplefb node
1932  * @base_address: framebuffer base address
1933  * @width: width in pixels
1934  * @height: height in pixels
1935  * @stride: bytes per line
1936  * @format: pixel format string
1937  *
1938  * Convenience function to fill and enable a simplefb node.
1939  */
1940 int fdt_setup_simplefb_node(void *fdt, int node, u64 base_address, u32 width,
1941                             u32 height, u32 stride, const char *format)
1942 {
1943         char name[32];
1944         fdt32_t cells[4];
1945         int i, addrc, sizec, ret;
1946
1947         fdt_support_default_count_cells(fdt, fdt_parent_offset(fdt, node),
1948                                         &addrc, &sizec);
1949         i = 0;
1950         if (addrc == 2)
1951                 cells[i++] = cpu_to_fdt32(base_address >> 32);
1952         cells[i++] = cpu_to_fdt32(base_address);
1953         if (sizec == 2)
1954                 cells[i++] = 0;
1955         cells[i++] = cpu_to_fdt32(height * stride);
1956
1957         ret = fdt_setprop(fdt, node, "reg", cells, sizeof(cells[0]) * i);
1958         if (ret < 0)
1959                 return ret;
1960
1961         snprintf(name, sizeof(name), "framebuffer@%llx", base_address);
1962         ret = fdt_set_name(fdt, node, name);
1963         if (ret < 0)
1964                 return ret;
1965
1966         ret = fdt_setprop_u32(fdt, node, "width", width);
1967         if (ret < 0)
1968                 return ret;
1969
1970         ret = fdt_setprop_u32(fdt, node, "height", height);
1971         if (ret < 0)
1972                 return ret;
1973
1974         ret = fdt_setprop_u32(fdt, node, "stride", stride);
1975         if (ret < 0)
1976                 return ret;
1977
1978         ret = fdt_setprop_string(fdt, node, "format", format);
1979         if (ret < 0)
1980                 return ret;
1981
1982         ret = fdt_setprop_string(fdt, node, "status", "okay");
1983         if (ret < 0)
1984                 return ret;
1985
1986         return 0;
1987 }
1988
1989 /*
1990  * Update native-mode in display-timings from display environment variable.
1991  * The node to update are specified by path.
1992  */
1993 int fdt_fixup_display(void *blob, const char *path, const char *display)
1994 {
1995         int off, toff;
1996
1997         if (!display || !path)
1998                 return -FDT_ERR_NOTFOUND;
1999
2000         toff = fdt_path_offset(blob, path);
2001         if (toff >= 0)
2002                 toff = fdt_subnode_offset(blob, toff, "display-timings");
2003         if (toff < 0)
2004                 return toff;
2005
2006         for (off = fdt_first_subnode(blob, toff);
2007              off >= 0;
2008              off = fdt_next_subnode(blob, off)) {
2009                 uint32_t h = fdt_get_phandle(blob, off);
2010                 debug("%s:0x%x\n", fdt_get_name(blob, off, NULL),
2011                       fdt32_to_cpu(h));
2012                 if (strcasecmp(fdt_get_name(blob, off, NULL), display) == 0)
2013                         return fdt_setprop_u32(blob, toff, "native-mode", h);
2014         }
2015         return toff;
2016 }
2017
2018 #ifdef CONFIG_OF_LIBFDT_OVERLAY
2019 /**
2020  * fdt_overlay_apply_verbose - Apply an overlay with verbose error reporting
2021  *
2022  * @fdt: ptr to device tree
2023  * @fdto: ptr to device tree overlay
2024  *
2025  * Convenience function to apply an overlay and display helpful messages
2026  * in the case of an error
2027  */
2028 int fdt_overlay_apply_verbose(void *fdt, void *fdto)
2029 {
2030         int err;
2031         bool has_symbols;
2032
2033         err = fdt_path_offset(fdt, "/__symbols__");
2034         has_symbols = err >= 0;
2035
2036         err = fdt_overlay_apply(fdt, fdto);
2037         if (err < 0) {
2038                 printf("failed on fdt_overlay_apply(): %s\n",
2039                                 fdt_strerror(err));
2040                 if (!has_symbols) {
2041                         printf("base fdt does did not have a /__symbols__ node\n");
2042                         printf("make sure you've compiled with -@\n");
2043                 }
2044         }
2045         return err;
2046 }
2047 #endif
2048
2049 /**
2050  * fdt_valid() - Check if an FDT is valid. If not, change it to NULL
2051  *
2052  * @blobp: Pointer to FDT pointer
2053  * Return: 1 if OK, 0 if bad (in which case *blobp is set to NULL)
2054  */
2055 int fdt_valid(struct fdt_header **blobp)
2056 {
2057         const void *blob = *blobp;
2058         int err;
2059
2060         if (!blob) {
2061                 printf("The address of the fdt is invalid (NULL).\n");
2062                 return 0;
2063         }
2064
2065         err = fdt_check_header(blob);
2066         if (err == 0)
2067                 return 1;       /* valid */
2068
2069         if (err < 0) {
2070                 printf("libfdt fdt_check_header(): %s", fdt_strerror(err));
2071                 /*
2072                  * Be more informative on bad version.
2073                  */
2074                 if (err == -FDT_ERR_BADVERSION) {
2075                         if (fdt_version(blob) <
2076                             FDT_FIRST_SUPPORTED_VERSION) {
2077                                 printf(" - too old, fdt %d < %d",
2078                                        fdt_version(blob),
2079                                        FDT_FIRST_SUPPORTED_VERSION);
2080                         }
2081                         if (fdt_last_comp_version(blob) >
2082                             FDT_LAST_SUPPORTED_VERSION) {
2083                                 printf(" - too new, fdt %d > %d",
2084                                        fdt_version(blob),
2085                                        FDT_LAST_SUPPORTED_VERSION);
2086                         }
2087                 }
2088                 printf("\n");
2089                 *blobp = NULL;
2090                 return 0;
2091         }
2092         return 1;
2093 }