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