]> git.sur5r.net Git - u-boot/blob - common/fdt_support.c
Merge branch 'master' of ../master
[u-boot] / common / fdt_support.c
1 /*
2  * (C) Copyright 2007
3  * Gerald Van Baren, Custom IDEAS, vanbaren@cideas.com
4  *
5  * Copyright 2010 Freescale Semiconductor, Inc.
6  *
7  * See file CREDITS for list of people who contributed to this
8  * project.
9  *
10  * This program is free software; you can redistribute it and/or
11  * modify it under the terms of the GNU General Public License as
12  * published by the Free Software Foundation; either version 2 of
13  * the License, or (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program; if not, write to the Free Software
22  * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
23  * MA 02111-1307 USA
24  */
25
26 #include <common.h>
27 #include <stdio_dev.h>
28 #include <linux/ctype.h>
29 #include <linux/types.h>
30 #include <asm/global_data.h>
31 #include <fdt.h>
32 #include <libfdt.h>
33 #include <fdt_support.h>
34 #include <exports.h>
35
36 /*
37  * Global data (for the gd->bd)
38  */
39 DECLARE_GLOBAL_DATA_PTR;
40
41 /**
42  * fdt_getprop_u32_default - Find a node and return it's property or a default
43  *
44  * @fdt: ptr to device tree
45  * @path: path of node
46  * @prop: property name
47  * @dflt: default value if the property isn't found
48  *
49  * Convenience function to find a node and return it's property or a
50  * default value if it doesn't exist.
51  */
52 u32 fdt_getprop_u32_default(void *fdt, const char *path, const char *prop,
53                                 const u32 dflt)
54 {
55         const u32 *val;
56         int off;
57
58         off = fdt_path_offset(fdt, path);
59         if (off < 0)
60                 return dflt;
61
62         val = fdt_getprop(fdt, off, prop, NULL);
63         if (val)
64                 return *val;
65         else
66                 return dflt;
67 }
68
69 /**
70  * fdt_find_and_setprop: Find a node and set it's property
71  *
72  * @fdt: ptr to device tree
73  * @node: path of node
74  * @prop: property name
75  * @val: ptr to new value
76  * @len: length of new property value
77  * @create: flag to create the property if it doesn't exist
78  *
79  * Convenience function to directly set a property given the path to the node.
80  */
81 int fdt_find_and_setprop(void *fdt, const char *node, const char *prop,
82                          const void *val, int len, int create)
83 {
84         int nodeoff = fdt_path_offset(fdt, node);
85
86         if (nodeoff < 0)
87                 return nodeoff;
88
89         if ((!create) && (fdt_get_property(fdt, nodeoff, prop, 0) == NULL))
90                 return 0; /* create flag not set; so exit quietly */
91
92         return fdt_setprop(fdt, nodeoff, prop, val, len);
93 }
94
95 #ifdef CONFIG_OF_STDOUT_VIA_ALIAS
96
97 #ifdef CONFIG_SERIAL_MULTI
98 static void fdt_fill_multisername(char *sername, size_t maxlen)
99 {
100         const char *outname = stdio_devices[stdout]->name;
101
102         if (strcmp(outname, "serial") > 0)
103                 strncpy(sername, outname, maxlen);
104
105         /* eserial? */
106         if (strcmp(outname + 1, "serial") > 0)
107                 strncpy(sername, outname + 1, maxlen);
108 }
109 #else
110 static inline void fdt_fill_multisername(char *sername, size_t maxlen) {}
111 #endif /* CONFIG_SERIAL_MULTI */
112
113 static int fdt_fixup_stdout(void *fdt, int chosenoff)
114 {
115         int err = 0;
116 #ifdef CONFIG_CONS_INDEX
117         int node;
118         char sername[9] = { 0 };
119         const char *path;
120
121         fdt_fill_multisername(sername, sizeof(sername) - 1);
122         if (!sername[0])
123                 sprintf(sername, "serial%d", CONFIG_CONS_INDEX - 1);
124
125         err = node = fdt_path_offset(fdt, "/aliases");
126         if (node >= 0) {
127                 int len;
128                 path = fdt_getprop(fdt, node, sername, &len);
129                 if (path) {
130                         char *p = malloc(len);
131                         err = -FDT_ERR_NOSPACE;
132                         if (p) {
133                                 memcpy(p, path, len);
134                                 err = fdt_setprop(fdt, chosenoff,
135                                         "linux,stdout-path", p, len);
136                                 free(p);
137                         }
138                 } else {
139                         err = len;
140                 }
141         }
142 #endif
143         if (err < 0)
144                 printf("WARNING: could not set linux,stdout-path %s.\n",
145                                 fdt_strerror(err));
146
147         return err;
148 }
149 #endif
150
151 int fdt_initrd(void *fdt, ulong initrd_start, ulong initrd_end, int force)
152 {
153         int   nodeoffset;
154         int   err, j, total;
155         u32   tmp;
156         const char *path;
157         uint64_t addr, size;
158
159         /* Find the "chosen" node.  */
160         nodeoffset = fdt_path_offset (fdt, "/chosen");
161
162         /* If there is no "chosen" node in the blob return */
163         if (nodeoffset < 0) {
164                 printf("fdt_initrd: %s\n", fdt_strerror(nodeoffset));
165                 return nodeoffset;
166         }
167
168         /* just return if initrd_start/end aren't valid */
169         if ((initrd_start == 0) || (initrd_end == 0))
170                 return 0;
171
172         total = fdt_num_mem_rsv(fdt);
173
174         /*
175          * Look for an existing entry and update it.  If we don't find
176          * the entry, we will j be the next available slot.
177          */
178         for (j = 0; j < total; j++) {
179                 err = fdt_get_mem_rsv(fdt, j, &addr, &size);
180                 if (addr == initrd_start) {
181                         fdt_del_mem_rsv(fdt, j);
182                         break;
183                 }
184         }
185
186         err = fdt_add_mem_rsv(fdt, initrd_start, initrd_end - initrd_start + 1);
187         if (err < 0) {
188                 printf("fdt_initrd: %s\n", fdt_strerror(err));
189                 return err;
190         }
191
192         path = fdt_getprop(fdt, nodeoffset, "linux,initrd-start", NULL);
193         if ((path == NULL) || force) {
194                 tmp = __cpu_to_be32(initrd_start);
195                 err = fdt_setprop(fdt, nodeoffset,
196                         "linux,initrd-start", &tmp, sizeof(tmp));
197                 if (err < 0) {
198                         printf("WARNING: "
199                                 "could not set linux,initrd-start %s.\n",
200                                 fdt_strerror(err));
201                         return err;
202                 }
203                 tmp = __cpu_to_be32(initrd_end);
204                 err = fdt_setprop(fdt, nodeoffset,
205                         "linux,initrd-end", &tmp, sizeof(tmp));
206                 if (err < 0) {
207                         printf("WARNING: could not set linux,initrd-end %s.\n",
208                                 fdt_strerror(err));
209
210                         return err;
211                 }
212         }
213
214         return 0;
215 }
216
217 int fdt_chosen(void *fdt, int force)
218 {
219         int   nodeoffset;
220         int   err;
221         char  *str;             /* used to set string properties */
222         const char *path;
223
224         err = fdt_check_header(fdt);
225         if (err < 0) {
226                 printf("fdt_chosen: %s\n", fdt_strerror(err));
227                 return err;
228         }
229
230         /*
231          * Find the "chosen" node.
232          */
233         nodeoffset = fdt_path_offset (fdt, "/chosen");
234
235         /*
236          * If there is no "chosen" node in the blob, create it.
237          */
238         if (nodeoffset < 0) {
239                 /*
240                  * Create a new node "/chosen" (offset 0 is root level)
241                  */
242                 nodeoffset = fdt_add_subnode(fdt, 0, "chosen");
243                 if (nodeoffset < 0) {
244                         printf("WARNING: could not create /chosen %s.\n",
245                                 fdt_strerror(nodeoffset));
246                         return nodeoffset;
247                 }
248         }
249
250         /*
251          * Create /chosen properites that don't exist in the fdt.
252          * If the property exists, update it only if the "force" parameter
253          * is true.
254          */
255         str = getenv("bootargs");
256         if (str != NULL) {
257                 path = fdt_getprop(fdt, nodeoffset, "bootargs", NULL);
258                 if ((path == NULL) || force) {
259                         err = fdt_setprop(fdt, nodeoffset,
260                                 "bootargs", str, strlen(str)+1);
261                         if (err < 0)
262                                 printf("WARNING: could not set bootargs %s.\n",
263                                         fdt_strerror(err));
264                 }
265         }
266
267 #ifdef CONFIG_OF_STDOUT_VIA_ALIAS
268         path = fdt_getprop(fdt, nodeoffset, "linux,stdout-path", NULL);
269         if ((path == NULL) || force)
270                 err = fdt_fixup_stdout(fdt, nodeoffset);
271 #endif
272
273 #ifdef OF_STDOUT_PATH
274         path = fdt_getprop(fdt, nodeoffset, "linux,stdout-path", NULL);
275         if ((path == NULL) || force) {
276                 err = fdt_setprop(fdt, nodeoffset,
277                         "linux,stdout-path", OF_STDOUT_PATH, strlen(OF_STDOUT_PATH)+1);
278                 if (err < 0)
279                         printf("WARNING: could not set linux,stdout-path %s.\n",
280                                 fdt_strerror(err));
281         }
282 #endif
283
284         return err;
285 }
286
287 void do_fixup_by_path(void *fdt, const char *path, const char *prop,
288                       const void *val, int len, int create)
289 {
290 #if defined(DEBUG)
291         int i;
292         debug("Updating property '%s/%s' = ", path, prop);
293         for (i = 0; i < len; i++)
294                 debug(" %.2x", *(u8*)(val+i));
295         debug("\n");
296 #endif
297         int rc = fdt_find_and_setprop(fdt, path, prop, val, len, create);
298         if (rc)
299                 printf("Unable to update property %s:%s, err=%s\n",
300                         path, prop, fdt_strerror(rc));
301 }
302
303 void do_fixup_by_path_u32(void *fdt, const char *path, const char *prop,
304                           u32 val, int create)
305 {
306         val = cpu_to_fdt32(val);
307         do_fixup_by_path(fdt, path, prop, &val, sizeof(val), create);
308 }
309
310 void do_fixup_by_prop(void *fdt,
311                       const char *pname, const void *pval, int plen,
312                       const char *prop, const void *val, int len,
313                       int create)
314 {
315         int off;
316 #if defined(DEBUG)
317         int i;
318         debug("Updating property '%s' = ", prop);
319         for (i = 0; i < len; i++)
320                 debug(" %.2x", *(u8*)(val+i));
321         debug("\n");
322 #endif
323         off = fdt_node_offset_by_prop_value(fdt, -1, pname, pval, plen);
324         while (off != -FDT_ERR_NOTFOUND) {
325                 if (create || (fdt_get_property(fdt, off, prop, 0) != NULL))
326                         fdt_setprop(fdt, off, prop, val, len);
327                 off = fdt_node_offset_by_prop_value(fdt, off, pname, pval, plen);
328         }
329 }
330
331 void do_fixup_by_prop_u32(void *fdt,
332                           const char *pname, const void *pval, int plen,
333                           const char *prop, u32 val, int create)
334 {
335         val = cpu_to_fdt32(val);
336         do_fixup_by_prop(fdt, pname, pval, plen, prop, &val, 4, create);
337 }
338
339 void do_fixup_by_compat(void *fdt, const char *compat,
340                         const char *prop, const void *val, int len, int create)
341 {
342         int off = -1;
343 #if defined(DEBUG)
344         int i;
345         debug("Updating property '%s' = ", prop);
346         for (i = 0; i < len; i++)
347                 debug(" %.2x", *(u8*)(val+i));
348         debug("\n");
349 #endif
350         off = fdt_node_offset_by_compatible(fdt, -1, compat);
351         while (off != -FDT_ERR_NOTFOUND) {
352                 if (create || (fdt_get_property(fdt, off, prop, 0) != NULL))
353                         fdt_setprop(fdt, off, prop, val, len);
354                 off = fdt_node_offset_by_compatible(fdt, off, compat);
355         }
356 }
357
358 void do_fixup_by_compat_u32(void *fdt, const char *compat,
359                             const char *prop, u32 val, int create)
360 {
361         val = cpu_to_fdt32(val);
362         do_fixup_by_compat(fdt, compat, prop, &val, 4, create);
363 }
364
365 int fdt_fixup_memory(void *blob, u64 start, u64 size)
366 {
367         int err, nodeoffset, len = 0;
368         u8 tmp[16];
369         const u32 *addrcell, *sizecell;
370
371         err = fdt_check_header(blob);
372         if (err < 0) {
373                 printf("%s: %s\n", __FUNCTION__, fdt_strerror(err));
374                 return err;
375         }
376
377         /* update, or add and update /memory node */
378         nodeoffset = fdt_path_offset(blob, "/memory");
379         if (nodeoffset < 0) {
380                 nodeoffset = fdt_add_subnode(blob, 0, "memory");
381                 if (nodeoffset < 0)
382                         printf("WARNING: could not create /memory: %s.\n",
383                                         fdt_strerror(nodeoffset));
384                 return nodeoffset;
385         }
386         err = fdt_setprop(blob, nodeoffset, "device_type", "memory",
387                         sizeof("memory"));
388         if (err < 0) {
389                 printf("WARNING: could not set %s %s.\n", "device_type",
390                                 fdt_strerror(err));
391                 return err;
392         }
393
394         addrcell = fdt_getprop(blob, 0, "#address-cells", NULL);
395         /* use shifts and mask to ensure endianness */
396         if ((addrcell) && (*addrcell == 2)) {
397                 tmp[0] = (start >> 56) & 0xff;
398                 tmp[1] = (start >> 48) & 0xff;
399                 tmp[2] = (start >> 40) & 0xff;
400                 tmp[3] = (start >> 32) & 0xff;
401                 tmp[4] = (start >> 24) & 0xff;
402                 tmp[5] = (start >> 16) & 0xff;
403                 tmp[6] = (start >>  8) & 0xff;
404                 tmp[7] = (start      ) & 0xff;
405                 len = 8;
406         } else {
407                 tmp[0] = (start >> 24) & 0xff;
408                 tmp[1] = (start >> 16) & 0xff;
409                 tmp[2] = (start >>  8) & 0xff;
410                 tmp[3] = (start      ) & 0xff;
411                 len = 4;
412         }
413
414         sizecell = fdt_getprop(blob, 0, "#size-cells", NULL);
415         /* use shifts and mask to ensure endianness */
416         if ((sizecell) && (*sizecell == 2)) {
417                 tmp[0+len] = (size >> 56) & 0xff;
418                 tmp[1+len] = (size >> 48) & 0xff;
419                 tmp[2+len] = (size >> 40) & 0xff;
420                 tmp[3+len] = (size >> 32) & 0xff;
421                 tmp[4+len] = (size >> 24) & 0xff;
422                 tmp[5+len] = (size >> 16) & 0xff;
423                 tmp[6+len] = (size >>  8) & 0xff;
424                 tmp[7+len] = (size      ) & 0xff;
425                 len += 8;
426         } else {
427                 tmp[0+len] = (size >> 24) & 0xff;
428                 tmp[1+len] = (size >> 16) & 0xff;
429                 tmp[2+len] = (size >>  8) & 0xff;
430                 tmp[3+len] = (size      ) & 0xff;
431                 len += 4;
432         }
433
434         err = fdt_setprop(blob, nodeoffset, "reg", tmp, len);
435         if (err < 0) {
436                 printf("WARNING: could not set %s %s.\n",
437                                 "reg", fdt_strerror(err));
438                 return err;
439         }
440         return 0;
441 }
442
443 void fdt_fixup_ethernet(void *fdt)
444 {
445         int node, i, j;
446         char enet[16], *tmp, *end;
447         char mac[16] = "ethaddr";
448         const char *path;
449         unsigned char mac_addr[6];
450
451         node = fdt_path_offset(fdt, "/aliases");
452         if (node < 0)
453                 return;
454
455         i = 0;
456         while ((tmp = getenv(mac)) != NULL) {
457                 sprintf(enet, "ethernet%d", i);
458                 path = fdt_getprop(fdt, node, enet, NULL);
459                 if (!path) {
460                         debug("No alias for %s\n", enet);
461                         sprintf(mac, "eth%daddr", ++i);
462                         continue;
463                 }
464
465                 for (j = 0; j < 6; j++) {
466                         mac_addr[j] = tmp ? simple_strtoul(tmp, &end, 16) : 0;
467                         if (tmp)
468                                 tmp = (*end) ? end+1 : end;
469                 }
470
471                 do_fixup_by_path(fdt, path, "mac-address", &mac_addr, 6, 0);
472                 do_fixup_by_path(fdt, path, "local-mac-address",
473                                 &mac_addr, 6, 1);
474
475                 sprintf(mac, "eth%daddr", ++i);
476         }
477 }
478
479 /* Resize the fdt to its actual size + a bit of padding */
480 int fdt_resize(void *blob)
481 {
482         int i;
483         uint64_t addr, size;
484         int total, ret;
485         uint actualsize;
486
487         if (!blob)
488                 return 0;
489
490         total = fdt_num_mem_rsv(blob);
491         for (i = 0; i < total; i++) {
492                 fdt_get_mem_rsv(blob, i, &addr, &size);
493                 if (addr == (uint64_t)(u32)blob) {
494                         fdt_del_mem_rsv(blob, i);
495                         break;
496                 }
497         }
498
499         /*
500          * Calculate the actual size of the fdt
501          * plus the size needed for 5 fdt_add_mem_rsv, one
502          * for the fdt itself and 4 for a possible initrd
503          * ((initrd-start + initrd-end) * 2 (name & value))
504          */
505         actualsize = fdt_off_dt_strings(blob) +
506                 fdt_size_dt_strings(blob) + 5 * sizeof(struct fdt_reserve_entry);
507
508         /* Make it so the fdt ends on a page boundary */
509         actualsize = ALIGN(actualsize + ((uint)blob & 0xfff), 0x1000);
510         actualsize = actualsize - ((uint)blob & 0xfff);
511
512         /* Change the fdt header to reflect the correct size */
513         fdt_set_totalsize(blob, actualsize);
514
515         /* Add the new reservation */
516         ret = fdt_add_mem_rsv(blob, (uint)blob, actualsize);
517         if (ret < 0)
518                 return ret;
519
520         return actualsize;
521 }
522
523 #ifdef CONFIG_PCI
524 #define CONFIG_SYS_PCI_NR_INBOUND_WIN 4
525
526 #define FDT_PCI_PREFETCH        (0x40000000)
527 #define FDT_PCI_MEM32           (0x02000000)
528 #define FDT_PCI_IO              (0x01000000)
529 #define FDT_PCI_MEM64           (0x03000000)
530
531 int fdt_pci_dma_ranges(void *blob, int phb_off, struct pci_controller *hose) {
532
533         int addrcell, sizecell, len, r;
534         u32 *dma_range;
535         /* sized based on pci addr cells, size-cells, & address-cells */
536         u32 dma_ranges[(3 + 2 + 2) * CONFIG_SYS_PCI_NR_INBOUND_WIN];
537
538         addrcell = fdt_getprop_u32_default(blob, "/", "#address-cells", 1);
539         sizecell = fdt_getprop_u32_default(blob, "/", "#size-cells", 1);
540
541         dma_range = &dma_ranges[0];
542         for (r = 0; r < hose->region_count; r++) {
543                 u64 bus_start, phys_start, size;
544
545                 /* skip if !PCI_REGION_SYS_MEMORY */
546                 if (!(hose->regions[r].flags & PCI_REGION_SYS_MEMORY))
547                         continue;
548
549                 bus_start = (u64)hose->regions[r].bus_start;
550                 phys_start = (u64)hose->regions[r].phys_start;
551                 size = (u64)hose->regions[r].size;
552
553                 dma_range[0] = 0;
554                 if (size >= 0x100000000ull)
555                         dma_range[0] |= FDT_PCI_MEM64;
556                 else
557                         dma_range[0] |= FDT_PCI_MEM32;
558                 if (hose->regions[r].flags & PCI_REGION_PREFETCH)
559                         dma_range[0] |= FDT_PCI_PREFETCH;
560 #ifdef CONFIG_SYS_PCI_64BIT
561                 dma_range[1] = bus_start >> 32;
562 #else
563                 dma_range[1] = 0;
564 #endif
565                 dma_range[2] = bus_start & 0xffffffff;
566
567                 if (addrcell == 2) {
568                         dma_range[3] = phys_start >> 32;
569                         dma_range[4] = phys_start & 0xffffffff;
570                 } else {
571                         dma_range[3] = phys_start & 0xffffffff;
572                 }
573
574                 if (sizecell == 2) {
575                         dma_range[3 + addrcell + 0] = size >> 32;
576                         dma_range[3 + addrcell + 1] = size & 0xffffffff;
577                 } else {
578                         dma_range[3 + addrcell + 0] = size & 0xffffffff;
579                 }
580
581                 dma_range += (3 + addrcell + sizecell);
582         }
583
584         len = dma_range - &dma_ranges[0];
585         if (len)
586                 fdt_setprop(blob, phb_off, "dma-ranges", &dma_ranges[0], len*4);
587
588         return 0;
589 }
590 #endif
591
592 #ifdef CONFIG_FDT_FIXUP_NOR_FLASH_SIZE
593 /*
594  * This function can be used to update the size in the "reg" property
595  * of the NOR FLASH device nodes. This is necessary for boards with
596  * non-fixed NOR FLASH sizes.
597  */
598 int fdt_fixup_nor_flash_size(void *blob, int cs, u32 size)
599 {
600         char compat[][16] = { "cfi-flash", "jedec-flash" };
601         int off;
602         int len;
603         struct fdt_property *prop;
604         u32 *reg;
605         int i;
606
607         for (i = 0; i < 2; i++) {
608                 off = fdt_node_offset_by_compatible(blob, -1, compat[i]);
609                 while (off != -FDT_ERR_NOTFOUND) {
610                         /*
611                          * Found one compatible node, now check if this one
612                          * has the correct CS
613                          */
614                         prop = fdt_get_property_w(blob, off, "reg", &len);
615                         if (prop) {
616                                 reg = (u32 *)&prop->data[0];
617                                 if (reg[0] == cs) {
618                                         reg[2] = size;
619                                         fdt_setprop(blob, off, "reg", reg,
620                                                     3 * sizeof(u32));
621
622                                         return 0;
623                                 }
624                         }
625
626                         /* Move to next compatible node */
627                         off = fdt_node_offset_by_compatible(blob, off,
628                                                             compat[i]);
629                 }
630         }
631
632         return -1;
633 }
634 #endif
635
636 #ifdef CONFIG_FDT_FIXUP_PARTITIONS
637 #include <jffs2/load_kernel.h>
638 #include <mtd_node.h>
639
640 struct reg_cell {
641         unsigned int r0;
642         unsigned int r1;
643 };
644
645 int fdt_del_subnodes(const void *blob, int parent_offset)
646 {
647         int off, ndepth;
648         int ret;
649
650         for (ndepth = 0, off = fdt_next_node(blob, parent_offset, &ndepth);
651              (off >= 0) && (ndepth > 0);
652              off = fdt_next_node(blob, off, &ndepth)) {
653                 if (ndepth == 1) {
654                         debug("delete %s: offset: %x\n",
655                                 fdt_get_name(blob, off, 0), off);
656                         ret = fdt_del_node((void *)blob, off);
657                         if (ret < 0) {
658                                 printf("Can't delete node: %s\n",
659                                         fdt_strerror(ret));
660                                 return ret;
661                         } else {
662                                 ndepth = 0;
663                                 off = parent_offset;
664                         }
665                 }
666         }
667         return 0;
668 }
669
670 int fdt_increase_size(void *fdt, int add_len)
671 {
672         int newlen;
673
674         newlen = fdt_totalsize(fdt) + add_len;
675
676         /* Open in place with a new len */
677         return fdt_open_into(fdt, fdt, newlen);
678 }
679
680 int fdt_del_partitions(void *blob, int parent_offset)
681 {
682         const void *prop;
683         int ndepth = 0;
684         int off;
685         int ret;
686
687         off = fdt_next_node(blob, parent_offset, &ndepth);
688         if (off > 0 && ndepth == 1) {
689                 prop = fdt_getprop(blob, off, "label", NULL);
690                 if (prop == NULL) {
691                         /*
692                          * Could not find label property, nand {}; node?
693                          * Check subnode, delete partitions there if any.
694                          */
695                         return fdt_del_partitions(blob, off);
696                 } else {
697                         ret = fdt_del_subnodes(blob, parent_offset);
698                         if (ret < 0) {
699                                 printf("Can't remove subnodes: %s\n",
700                                         fdt_strerror(ret));
701                                 return ret;
702                         }
703                 }
704         }
705         return 0;
706 }
707
708 int fdt_node_set_part_info(void *blob, int parent_offset,
709                            struct mtd_device *dev)
710 {
711         struct list_head *pentry;
712         struct part_info *part;
713         struct reg_cell cell;
714         int off, ndepth = 0;
715         int part_num, ret;
716         char buf[64];
717
718         ret = fdt_del_partitions(blob, parent_offset);
719         if (ret < 0)
720                 return ret;
721
722         /*
723          * Check if it is nand {}; subnode, adjust
724          * the offset in this case
725          */
726         off = fdt_next_node(blob, parent_offset, &ndepth);
727         if (off > 0 && ndepth == 1)
728                 parent_offset = off;
729
730         part_num = 0;
731         list_for_each_prev(pentry, &dev->parts) {
732                 int newoff;
733
734                 part = list_entry(pentry, struct part_info, link);
735
736                 debug("%2d: %-20s0x%08x\t0x%08x\t%d\n",
737                         part_num, part->name, part->size,
738                         part->offset, part->mask_flags);
739
740                 sprintf(buf, "partition@%x", part->offset);
741 add_sub:
742                 ret = fdt_add_subnode(blob, parent_offset, buf);
743                 if (ret == -FDT_ERR_NOSPACE) {
744                         ret = fdt_increase_size(blob, 512);
745                         if (!ret)
746                                 goto add_sub;
747                         else
748                                 goto err_size;
749                 } else if (ret < 0) {
750                         printf("Can't add partition node: %s\n",
751                                 fdt_strerror(ret));
752                         return ret;
753                 }
754                 newoff = ret;
755
756                 /* Check MTD_WRITEABLE_CMD flag */
757                 if (part->mask_flags & 1) {
758 add_ro:
759                         ret = fdt_setprop(blob, newoff, "read_only", NULL, 0);
760                         if (ret == -FDT_ERR_NOSPACE) {
761                                 ret = fdt_increase_size(blob, 512);
762                                 if (!ret)
763                                         goto add_ro;
764                                 else
765                                         goto err_size;
766                         } else if (ret < 0)
767                                 goto err_prop;
768                 }
769
770                 cell.r0 = cpu_to_fdt32(part->offset);
771                 cell.r1 = cpu_to_fdt32(part->size);
772 add_reg:
773                 ret = fdt_setprop(blob, newoff, "reg", &cell, sizeof(cell));
774                 if (ret == -FDT_ERR_NOSPACE) {
775                         ret = fdt_increase_size(blob, 512);
776                         if (!ret)
777                                 goto add_reg;
778                         else
779                                 goto err_size;
780                 } else if (ret < 0)
781                         goto err_prop;
782
783 add_label:
784                 ret = fdt_setprop_string(blob, newoff, "label", part->name);
785                 if (ret == -FDT_ERR_NOSPACE) {
786                         ret = fdt_increase_size(blob, 512);
787                         if (!ret)
788                                 goto add_label;
789                         else
790                                 goto err_size;
791                 } else if (ret < 0)
792                         goto err_prop;
793
794                 part_num++;
795         }
796         return 0;
797 err_size:
798         printf("Can't increase blob size: %s\n", fdt_strerror(ret));
799         return ret;
800 err_prop:
801         printf("Can't add property: %s\n", fdt_strerror(ret));
802         return ret;
803 }
804
805 /*
806  * Update partitions in nor/nand nodes using info from
807  * mtdparts environment variable. The nodes to update are
808  * specified by node_info structure which contains mtd device
809  * type and compatible string: E. g. the board code in
810  * ft_board_setup() could use:
811  *
812  *      struct node_info nodes[] = {
813  *              { "fsl,mpc5121-nfc",    MTD_DEV_TYPE_NAND, },
814  *              { "cfi-flash",          MTD_DEV_TYPE_NOR,  },
815  *      };
816  *
817  *      fdt_fixup_mtdparts(blob, nodes, ARRAY_SIZE(nodes));
818  */
819 void fdt_fixup_mtdparts(void *blob, void *node_info, int node_info_size)
820 {
821         struct node_info *ni = node_info;
822         struct mtd_device *dev;
823         char *parts;
824         int i, idx;
825         int noff;
826
827         parts = getenv("mtdparts");
828         if (!parts)
829                 return;
830
831         if (mtdparts_init() != 0)
832                 return;
833
834         for (i = 0; i < node_info_size; i++) {
835                 idx = 0;
836                 noff = fdt_node_offset_by_compatible(blob, -1, ni[i].compat);
837                 while (noff != -FDT_ERR_NOTFOUND) {
838                         debug("%s: %s, mtd dev type %d\n",
839                                 fdt_get_name(blob, noff, 0),
840                                 ni[i].compat, ni[i].type);
841                         dev = device_find(ni[i].type, idx++);
842                         if (dev) {
843                                 if (fdt_node_set_part_info(blob, noff, dev))
844                                         return; /* return on error */
845                         }
846
847                         /* Jump to next flash node */
848                         noff = fdt_node_offset_by_compatible(blob, noff,
849                                                              ni[i].compat);
850                 }
851         }
852 }
853 #endif
854
855 void fdt_del_node_and_alias(void *blob, const char *alias)
856 {
857         int off = fdt_path_offset(blob, alias);
858
859         if (off < 0)
860                 return;
861
862         fdt_del_node(blob, off);
863
864         off = fdt_path_offset(blob, "/aliases");
865         fdt_delprop(blob, off, alias);
866 }
867
868 /* Helper to read a big number; size is in cells (not bytes) */
869 static inline u64 of_read_number(const __be32 *cell, int size)
870 {
871         u64 r = 0;
872         while (size--)
873                 r = (r << 32) | be32_to_cpu(*(cell++));
874         return r;
875 }
876
877 static int of_n_cells(const void *blob, int nodeoffset, const char *name)
878 {
879         int np;
880         const int *ip;
881
882         do {
883                 np = fdt_parent_offset(blob, nodeoffset);
884
885                 if (np >= 0)
886                         nodeoffset = np;
887                 ip = (int *)fdt_getprop(blob, nodeoffset, name, NULL);
888                 if (ip)
889                         return be32_to_cpup(ip);
890         } while (np >= 0);
891
892         /* No #<NAME>-cells property for the root node */
893         return 1;
894 }
895
896 int of_n_addr_cells(const void *blob, int nodeoffset)
897 {
898         return of_n_cells(blob, nodeoffset, "#address-cells");
899 }
900
901 int of_n_size_cells(const void *blob, int nodeoffset)
902 {
903         return of_n_cells(blob, nodeoffset, "#size-cells");
904 }
905
906 #define PRu64   "%llx"
907
908 /* Max address size we deal with */
909 #define OF_MAX_ADDR_CELLS       4
910 #define OF_BAD_ADDR     ((u64)-1)
911 #define OF_CHECK_COUNTS(na, ns) ((na) > 0 && (na) <= OF_MAX_ADDR_CELLS && \
912                         (ns) > 0)
913
914 /* Debug utility */
915 #ifdef DEBUG
916 static void of_dump_addr(const char *s, const u32 *addr, int na)
917 {
918         printf("%s", s);
919         while(na--)
920                 printf(" %08x", *(addr++));
921         printf("\n");
922 }
923 #else
924 static void of_dump_addr(const char *s, const u32 *addr, int na) { }
925 #endif
926
927 /* Callbacks for bus specific translators */
928 struct of_bus {
929         const char      *name;
930         const char      *addresses;
931         void            (*count_cells)(void *blob, int offset,
932                                 int *addrc, int *sizec);
933         u64             (*map)(u32 *addr, const u32 *range,
934                                 int na, int ns, int pna);
935         int             (*translate)(u32 *addr, u64 offset, int na);
936 };
937
938 /* Default translator (generic bus) */
939 static void of_bus_default_count_cells(void *blob, int offset,
940                                         int *addrc, int *sizec)
941 {
942         if (addrc)
943                 *addrc = of_n_addr_cells(blob, offset);
944         if (sizec)
945                 *sizec = of_n_size_cells(blob, offset);
946 }
947
948 static u64 of_bus_default_map(u32 *addr, const u32 *range,
949                 int na, int ns, int pna)
950 {
951         u64 cp, s, da;
952
953         cp = of_read_number(range, na);
954         s  = of_read_number(range + na + pna, ns);
955         da = of_read_number(addr, na);
956
957         debug("OF: default map, cp="PRu64", s="PRu64", da="PRu64"\n",
958             cp, s, da);
959
960         if (da < cp || da >= (cp + s))
961                 return OF_BAD_ADDR;
962         return da - cp;
963 }
964
965 static int of_bus_default_translate(u32 *addr, u64 offset, int na)
966 {
967         u64 a = of_read_number(addr, na);
968         memset(addr, 0, na * 4);
969         a += offset;
970         if (na > 1)
971                 addr[na - 2] = a >> 32;
972         addr[na - 1] = a & 0xffffffffu;
973
974         return 0;
975 }
976
977 /* Array of bus specific translators */
978 static struct of_bus of_busses[] = {
979         /* Default */
980         {
981                 .name = "default",
982                 .addresses = "reg",
983                 .count_cells = of_bus_default_count_cells,
984                 .map = of_bus_default_map,
985                 .translate = of_bus_default_translate,
986         },
987 };
988
989 static int of_translate_one(void * blob, int parent, struct of_bus *bus,
990                             struct of_bus *pbus, u32 *addr,
991                             int na, int ns, int pna, const char *rprop)
992 {
993         const u32 *ranges;
994         int rlen;
995         int rone;
996         u64 offset = OF_BAD_ADDR;
997
998         /* Normally, an absence of a "ranges" property means we are
999          * crossing a non-translatable boundary, and thus the addresses
1000          * below the current not cannot be converted to CPU physical ones.
1001          * Unfortunately, while this is very clear in the spec, it's not
1002          * what Apple understood, and they do have things like /uni-n or
1003          * /ht nodes with no "ranges" property and a lot of perfectly
1004          * useable mapped devices below them. Thus we treat the absence of
1005          * "ranges" as equivalent to an empty "ranges" property which means
1006          * a 1:1 translation at that level. It's up to the caller not to try
1007          * to translate addresses that aren't supposed to be translated in
1008          * the first place. --BenH.
1009          */
1010         ranges = (u32 *)fdt_getprop(blob, parent, rprop, &rlen);
1011         if (ranges == NULL || rlen == 0) {
1012                 offset = of_read_number(addr, na);
1013                 memset(addr, 0, pna * 4);
1014                 debug("OF: no ranges, 1:1 translation\n");
1015                 goto finish;
1016         }
1017
1018         debug("OF: walking ranges...\n");
1019
1020         /* Now walk through the ranges */
1021         rlen /= 4;
1022         rone = na + pna + ns;
1023         for (; rlen >= rone; rlen -= rone, ranges += rone) {
1024                 offset = bus->map(addr, ranges, na, ns, pna);
1025                 if (offset != OF_BAD_ADDR)
1026                         break;
1027         }
1028         if (offset == OF_BAD_ADDR) {
1029                 debug("OF: not found !\n");
1030                 return 1;
1031         }
1032         memcpy(addr, ranges + na, 4 * pna);
1033
1034  finish:
1035         of_dump_addr("OF: parent translation for:", addr, pna);
1036         debug("OF: with offset: "PRu64"\n", offset);
1037
1038         /* Translate it into parent bus space */
1039         return pbus->translate(addr, offset, pna);
1040 }
1041
1042 /*
1043  * Translate an address from the device-tree into a CPU physical address,
1044  * this walks up the tree and applies the various bus mappings on the
1045  * way.
1046  *
1047  * Note: We consider that crossing any level with #size-cells == 0 to mean
1048  * that translation is impossible (that is we are not dealing with a value
1049  * that can be mapped to a cpu physical address). This is not really specified
1050  * that way, but this is traditionally the way IBM at least do things
1051  */
1052 u64 __of_translate_address(void *blob, int node_offset, const u32 *in_addr,
1053                            const char *rprop)
1054 {
1055         int parent;
1056         struct of_bus *bus, *pbus;
1057         u32 addr[OF_MAX_ADDR_CELLS];
1058         int na, ns, pna, pns;
1059         u64 result = OF_BAD_ADDR;
1060
1061         debug("OF: ** translation for device %s **\n",
1062                 fdt_get_name(blob, node_offset, NULL));
1063
1064         /* Get parent & match bus type */
1065         parent = fdt_parent_offset(blob, node_offset);
1066         if (parent < 0)
1067                 goto bail;
1068         bus = &of_busses[0];
1069
1070         /* Cound address cells & copy address locally */
1071         bus->count_cells(blob, node_offset, &na, &ns);
1072         if (!OF_CHECK_COUNTS(na, ns)) {
1073                 printf("%s: Bad cell count for %s\n", __FUNCTION__,
1074                        fdt_get_name(blob, node_offset, NULL));
1075                 goto bail;
1076         }
1077         memcpy(addr, in_addr, na * 4);
1078
1079         debug("OF: bus is %s (na=%d, ns=%d) on %s\n",
1080             bus->name, na, ns, fdt_get_name(blob, parent, NULL));
1081         of_dump_addr("OF: translating address:", addr, na);
1082
1083         /* Translate */
1084         for (;;) {
1085                 /* Switch to parent bus */
1086                 node_offset = parent;
1087                 parent = fdt_parent_offset(blob, node_offset);
1088
1089                 /* If root, we have finished */
1090                 if (parent < 0) {
1091                         debug("OF: reached root node\n");
1092                         result = of_read_number(addr, na);
1093                         break;
1094                 }
1095
1096                 /* Get new parent bus and counts */
1097                 pbus = &of_busses[0];
1098                 pbus->count_cells(blob, node_offset, &pna, &pns);
1099                 if (!OF_CHECK_COUNTS(pna, pns)) {
1100                         printf("%s: Bad cell count for %s\n", __FUNCTION__,
1101                                 fdt_get_name(blob, node_offset, NULL));
1102                         break;
1103                 }
1104
1105                 debug("OF: parent bus is %s (na=%d, ns=%d) on %s\n",
1106                     pbus->name, pna, pns, fdt_get_name(blob, parent, NULL));
1107
1108                 /* Apply bus translation */
1109                 if (of_translate_one(blob, node_offset, bus, pbus,
1110                                         addr, na, ns, pna, rprop))
1111                         break;
1112
1113                 /* Complete the move up one level */
1114                 na = pna;
1115                 ns = pns;
1116                 bus = pbus;
1117
1118                 of_dump_addr("OF: one level translation:", addr, na);
1119         }
1120  bail:
1121
1122         return result;
1123 }
1124
1125 u64 fdt_translate_address(void *blob, int node_offset, const u32 *in_addr)
1126 {
1127         return __of_translate_address(blob, node_offset, in_addr, "ranges");
1128 }
1129
1130 /**
1131  * fdt_node_offset_by_compat_reg: Find a node that matches compatiable and
1132  * who's reg property matches a physical cpu address
1133  *
1134  * @blob: ptr to device tree
1135  * @compat: compatiable string to match
1136  * @compat_off: property name
1137  *
1138  */
1139 int fdt_node_offset_by_compat_reg(void *blob, const char *compat,
1140                                         phys_addr_t compat_off)
1141 {
1142         int len, off = fdt_node_offset_by_compatible(blob, -1, compat);
1143         while (off != -FDT_ERR_NOTFOUND) {
1144                 u32 *reg = (u32 *)fdt_getprop(blob, off, "reg", &len);
1145                 if (reg) {
1146                         if (compat_off == fdt_translate_address(blob, off, reg))
1147                                 return off;
1148                 }
1149                 off = fdt_node_offset_by_compatible(blob, off, compat);
1150         }
1151
1152         return -FDT_ERR_NOTFOUND;
1153 }
1154
1155 /**
1156  * fdt_alloc_phandle: Return next free phandle value
1157  *
1158  * @blob: ptr to device tree
1159  */
1160 int fdt_alloc_phandle(void *blob)
1161 {
1162         int offset, len, phandle = 0;
1163         const u32 *val;
1164
1165         for (offset = fdt_next_node(blob, -1, NULL); offset >= 0;
1166              offset = fdt_next_node(blob, offset, NULL)) {
1167                 val = fdt_getprop(blob, offset, "linux,phandle", &len);
1168                 if (val)
1169                         phandle = max(*val, phandle);
1170         }
1171
1172         return phandle + 1;
1173 }