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