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