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  * Provide a weak default function to return the flash bank size.
 595  * There might be multiple non-identical flash chips connected to one
 596  * chip-select, so we need to pass an index as well.
 597  */
 598 u32 __flash_get_bank_size(int cs, int idx)
 599 {
 600         extern flash_info_t flash_info[];
 601
 602         /*
 603          * As default, a simple 1:1 mapping is provided. Boards with
 604          * a different mapping need to supply a board specific mapping
 605          * routine.
 606          */
 607         return flash_info[cs].size;
 608 }
 609 u32 flash_get_bank_size(int cs, int idx)
 610         __attribute__((weak, alias("__flash_get_bank_size")));
 611
 612 /*
 613  * This function can be used to update the size in the "reg" property
 614  * of all NOR FLASH device nodes. This is necessary for boards with
 615  * non-fixed NOR FLASH sizes.
 616  */
 617 int fdt_fixup_nor_flash_size(void *blob)
 618 {
 619         char compat[][16] = { "cfi-flash", "jedec-flash" };
 620         int off;
 621         int len;
 622         struct fdt_property *prop;
 623         u32 *reg, *reg2;
 624         int i;
 625
 626         for (i = 0; i < 2; i++) {
 627                 off = fdt_node_offset_by_compatible(blob, -1, compat[i]);
 628                 while (off != -FDT_ERR_NOTFOUND) {
 629                         int idx;
 630
 631                         /*
 632                          * Found one compatible node, so fixup the size
 633                          * int its reg properties
 634                          */
 635                         prop = fdt_get_property_w(blob, off, "reg", &len);
 636                         if (prop) {
 637                                 int tuple_size = 3 * sizeof(reg);
 638
 639                                 /*
 640                                  * There might be multiple reg-tuples,
 641                                  * so loop through them all
 642                                  */
 643                                 reg = reg2 = (u32 *)&prop->data[0];
 644                                 for (idx = 0; idx < (len / tuple_size); idx++) {
 645                                         /*
 646                                          * Update size in reg property
 647                                          */
 648                                         reg[2] = flash_get_bank_size(reg[0],
 649                                                                      idx);
 650
 651                                         /*
 652                                          * Point to next reg tuple
 653                                          */
 654                                         reg += 3;
 655                                 }
 656
 657                                 fdt_setprop(blob, off, "reg", reg2, len);
 658                         }
 659
 660                         /* Move to next compatible node */
 661                         off = fdt_node_offset_by_compatible(blob, off,
 662                                                             compat[i]);
 663                 }
 664         }
 665
 666         return 0;
 667 }
 668 #endif
 669
 670 #ifdef CONFIG_FDT_FIXUP_PARTITIONS
 671 #include <jffs2/load_kernel.h>
 672 #include <mtd_node.h>
 673
 674 struct reg_cell {
 675         unsigned int r0;
 676         unsigned int r1;
 677 };
 678
 679 int fdt_del_subnodes(const void *blob, int parent_offset)
 680 {
 681         int off, ndepth;
 682         int ret;
 683
 684         for (ndepth = 0, off = fdt_next_node(blob, parent_offset, &ndepth);
 685              (off >= 0) && (ndepth > 0);
 686              off = fdt_next_node(blob, off, &ndepth)) {
 687                 if (ndepth == 1) {
 688                         debug("delete %s: offset: %x\n",
 689                                 fdt_get_name(blob, off, 0), off);
 690                         ret = fdt_del_node((void *)blob, off);
 691                         if (ret < 0) {
 692                                 printf("Can't delete node: %s\n",
 693                                         fdt_strerror(ret));
 694                                 return ret;
 695                         } else {
 696                                 ndepth = 0;
 697                                 off = parent_offset;
 698                         }
 699                 }
 700         }
 701         return 0;
 702 }
 703
 704 int fdt_increase_size(void *fdt, int add_len)
 705 {
 706         int newlen;
 707
 708         newlen = fdt_totalsize(fdt) + add_len;
 709
 710         /* Open in place with a new len */
 711         return fdt_open_into(fdt, fdt, newlen);
 712 }
 713
 714 int fdt_del_partitions(void *blob, int parent_offset)
 715 {
 716         const void *prop;
 717         int ndepth = 0;
 718         int off;
 719         int ret;
 720
 721         off = fdt_next_node(blob, parent_offset, &ndepth);
 722         if (off > 0 && ndepth == 1) {
 723                 prop = fdt_getprop(blob, off, "label", NULL);
 724                 if (prop == NULL) {
 725                         /*
 726                          * Could not find label property, nand {}; node?
 727                          * Check subnode, delete partitions there if any.
 728                          */
 729                         return fdt_del_partitions(blob, off);
 730                 } else {
 731                         ret = fdt_del_subnodes(blob, parent_offset);
 732                         if (ret < 0) {
 733                                 printf("Can't remove subnodes: %s\n",
 734                                         fdt_strerror(ret));
 735                                 return ret;
 736                         }
 737                 }
 738         }
 739         return 0;
 740 }
 741
 742 int fdt_node_set_part_info(void *blob, int parent_offset,
 743                            struct mtd_device *dev)
 744 {
 745         struct list_head *pentry;
 746         struct part_info *part;
 747         struct reg_cell cell;
 748         int off, ndepth = 0;
 749         int part_num, ret;
 750         char buf[64];
 751
 752         ret = fdt_del_partitions(blob, parent_offset);
 753         if (ret < 0)
 754                 return ret;
 755
 756         /*
 757          * Check if it is nand {}; subnode, adjust
 758          * the offset in this case
 759          */
 760         off = fdt_next_node(blob, parent_offset, &ndepth);
 761         if (off > 0 && ndepth == 1)
 762                 parent_offset = off;
 763
 764         part_num = 0;
 765         list_for_each_prev(pentry, &dev->parts) {
 766                 int newoff;
 767
 768                 part = list_entry(pentry, struct part_info, link);
 769
 770                 debug("%2d: %-20s0x%08x\t0x%08x\t%d\n",
 771                         part_num, part->name, part->size,
 772                         part->offset, part->mask_flags);
 773
 774                 sprintf(buf, "partition@%x", part->offset);
 775 add_sub:
 776                 ret = fdt_add_subnode(blob, parent_offset, buf);
 777                 if (ret == -FDT_ERR_NOSPACE) {
 778                         ret = fdt_increase_size(blob, 512);
 779                         if (!ret)
 780                                 goto add_sub;
 781                         else
 782                                 goto err_size;
 783                 } else if (ret < 0) {
 784                         printf("Can't add partition node: %s\n",
 785                                 fdt_strerror(ret));
 786                         return ret;
 787                 }
 788                 newoff = ret;
 789
 790                 /* Check MTD_WRITEABLE_CMD flag */
 791                 if (part->mask_flags & 1) {
 792 add_ro:
 793                         ret = fdt_setprop(blob, newoff, "read_only", NULL, 0);
 794                         if (ret == -FDT_ERR_NOSPACE) {
 795                                 ret = fdt_increase_size(blob, 512);
 796                                 if (!ret)
 797                                         goto add_ro;
 798                                 else
 799                                         goto err_size;
 800                         } else if (ret < 0)
 801                                 goto err_prop;
 802                 }
 803
 804                 cell.r0 = cpu_to_fdt32(part->offset);
 805                 cell.r1 = cpu_to_fdt32(part->size);
 806 add_reg:
 807                 ret = fdt_setprop(blob, newoff, "reg", &cell, sizeof(cell));
 808                 if (ret == -FDT_ERR_NOSPACE) {
 809                         ret = fdt_increase_size(blob, 512);
 810                         if (!ret)
 811                                 goto add_reg;
 812                         else
 813                                 goto err_size;
 814                 } else if (ret < 0)
 815                         goto err_prop;
 816
 817 add_label:
 818                 ret = fdt_setprop_string(blob, newoff, "label", part->name);
 819                 if (ret == -FDT_ERR_NOSPACE) {
 820                         ret = fdt_increase_size(blob, 512);
 821                         if (!ret)
 822                                 goto add_label;
 823                         else
 824                                 goto err_size;
 825                 } else if (ret < 0)
 826                         goto err_prop;
 827
 828                 part_num++;
 829         }
 830         return 0;
 831 err_size:
 832         printf("Can't increase blob size: %s\n", fdt_strerror(ret));
 833         return ret;
 834 err_prop:
 835         printf("Can't add property: %s\n", fdt_strerror(ret));
 836         return ret;
 837 }
 838
 839 /*
 840  * Update partitions in nor/nand nodes using info from
 841  * mtdparts environment variable. The nodes to update are
 842  * specified by node_info structure which contains mtd device
 843  * type and compatible string: E. g. the board code in
 844  * ft_board_setup() could use:
 845  *
 846  *      struct node_info nodes[] = {
 847  *              { "fsl,mpc5121-nfc",    MTD_DEV_TYPE_NAND, },
 848  *              { "cfi-flash",          MTD_DEV_TYPE_NOR,  },
 849  *      };
 850  *
 851  *      fdt_fixup_mtdparts(blob, nodes, ARRAY_SIZE(nodes));
 852  */
 853 void fdt_fixup_mtdparts(void *blob, void *node_info, int node_info_size)
 854 {
 855         struct node_info *ni = node_info;
 856         struct mtd_device *dev;
 857         char *parts;
 858         int i, idx;
 859         int noff;
 860
 861         parts = getenv("mtdparts");
 862         if (!parts)
 863                 return;
 864
 865         if (mtdparts_init() != 0)
 866                 return;
 867
 868         for (i = 0; i < node_info_size; i++) {
 869                 idx = 0;
 870                 noff = fdt_node_offset_by_compatible(blob, -1, ni[i].compat);
 871                 while (noff != -FDT_ERR_NOTFOUND) {
 872                         debug("%s: %s, mtd dev type %d\n",
 873                                 fdt_get_name(blob, noff, 0),
 874                                 ni[i].compat, ni[i].type);
 875                         dev = device_find(ni[i].type, idx++);
 876                         if (dev) {
 877                                 if (fdt_node_set_part_info(blob, noff, dev))
 878                                         return; /* return on error */
 879                         }
 880
 881                         /* Jump to next flash node */
 882                         noff = fdt_node_offset_by_compatible(blob, noff,
 883                                                              ni[i].compat);
 884                 }
 885         }
 886 }
 887 #endif
 888
 889 void fdt_del_node_and_alias(void *blob, const char *alias)
 890 {
 891         int off = fdt_path_offset(blob, alias);
 892
 893         if (off < 0)
 894                 return;
 895
 896         fdt_del_node(blob, off);
 897
 898         off = fdt_path_offset(blob, "/aliases");
 899         fdt_delprop(blob, off, alias);
 900 }
 901
 902 /* Helper to read a big number; size is in cells (not bytes) */
 903 static inline u64 of_read_number(const __be32 *cell, int size)
 904 {
 905         u64 r = 0;
 906         while (size--)
 907                 r = (r << 32) | be32_to_cpu(*(cell++));
 908         return r;
 909 }
 910
 911 #define PRu64   "%llx"
 912
 913 /* Max address size we deal with */
 914 #define OF_MAX_ADDR_CELLS       4
 915 #define OF_BAD_ADDR     ((u64)-1)
 916 #define OF_CHECK_COUNTS(na, ns) ((na) > 0 && (na) <= OF_MAX_ADDR_CELLS && \
 917                         (ns) > 0)
 918
 919 /* Debug utility */
 920 #ifdef DEBUG
 921 static void of_dump_addr(const char *s, const u32 *addr, int na)
 922 {
 923         printf("%s", s);
 924         while(na--)
 925                 printf(" %08x", *(addr++));
 926         printf("\n");
 927 }
 928 #else
 929 static void of_dump_addr(const char *s, const u32 *addr, int na) { }
 930 #endif
 931
 932 /* Callbacks for bus specific translators */
 933 struct of_bus {
 934         const char      *name;
 935         const char      *addresses;
 936         void            (*count_cells)(void *blob, int parentoffset,
 937                                 int *addrc, int *sizec);
 938         u64             (*map)(u32 *addr, const u32 *range,
 939                                 int na, int ns, int pna);
 940         int             (*translate)(u32 *addr, u64 offset, int na);
 941 };
 942
 943 /* Default translator (generic bus) */
 944 static void of_bus_default_count_cells(void *blob, int parentoffset,
 945                                         int *addrc, int *sizec)
 946 {
 947         const u32 *prop;
 948
 949         if (addrc) {
 950                 prop = fdt_getprop(blob, parentoffset, "#address-cells", NULL);
 951                 if (prop)
 952                         *addrc = be32_to_cpup(prop);
 953                 else
 954                         *addrc = 2;
 955         }
 956
 957         if (sizec) {
 958                 prop = fdt_getprop(blob, parentoffset, "#size-cells", NULL);
 959                 if (prop)
 960                         *sizec = be32_to_cpup(prop);
 961                 else
 962                         *sizec = 1;
 963         }
 964 }
 965
 966 static u64 of_bus_default_map(u32 *addr, const u32 *range,
 967                 int na, int ns, int pna)
 968 {
 969         u64 cp, s, da;
 970
 971         cp = of_read_number(range, na);
 972         s  = of_read_number(range + na + pna, ns);
 973         da = of_read_number(addr, na);
 974
 975         debug("OF: default map, cp="PRu64", s="PRu64", da="PRu64"\n",
 976             cp, s, da);
 977
 978         if (da < cp || da >= (cp + s))
 979                 return OF_BAD_ADDR;
 980         return da - cp;
 981 }
 982
 983 static int of_bus_default_translate(u32 *addr, u64 offset, int na)
 984 {
 985         u64 a = of_read_number(addr, na);
 986         memset(addr, 0, na * 4);
 987         a += offset;
 988         if (na > 1)
 989                 addr[na - 2] = a >> 32;
 990         addr[na - 1] = a & 0xffffffffu;
 991
 992         return 0;
 993 }
 994
 995 /* Array of bus specific translators */
 996 static struct of_bus of_busses[] = {
 997         /* Default */
 998         {
 999                 .name = "default",
1000                 .addresses = "reg",
1001                 .count_cells = of_bus_default_count_cells,
1002                 .map = of_bus_default_map,
1003                 .translate = of_bus_default_translate,
1004         },
1005 };
1006
1007 static int of_translate_one(void * blob, int parent, struct of_bus *bus,
1008                             struct of_bus *pbus, u32 *addr,
1009                             int na, int ns, int pna, const char *rprop)
1010 {
1011         const u32 *ranges;
1012         int rlen;
1013         int rone;
1014         u64 offset = OF_BAD_ADDR;
1015
1016         /* Normally, an absence of a "ranges" property means we are
1017          * crossing a non-translatable boundary, and thus the addresses
1018          * below the current not cannot be converted to CPU physical ones.
1019          * Unfortunately, while this is very clear in the spec, it's not
1020          * what Apple understood, and they do have things like /uni-n or
1021          * /ht nodes with no "ranges" property and a lot of perfectly
1022          * useable mapped devices below them. Thus we treat the absence of
1023          * "ranges" as equivalent to an empty "ranges" property which means
1024          * a 1:1 translation at that level. It's up to the caller not to try
1025          * to translate addresses that aren't supposed to be translated in
1026          * the first place. --BenH.
1027          */
1028         ranges = (u32 *)fdt_getprop(blob, parent, rprop, &rlen);
1029         if (ranges == NULL || rlen == 0) {
1030                 offset = of_read_number(addr, na);
1031                 memset(addr, 0, pna * 4);
1032                 debug("OF: no ranges, 1:1 translation\n");
1033                 goto finish;
1034         }
1035
1036         debug("OF: walking ranges...\n");
1037
1038         /* Now walk through the ranges */
1039         rlen /= 4;
1040         rone = na + pna + ns;
1041         for (; rlen >= rone; rlen -= rone, ranges += rone) {
1042                 offset = bus->map(addr, ranges, na, ns, pna);
1043                 if (offset != OF_BAD_ADDR)
1044                         break;
1045         }
1046         if (offset == OF_BAD_ADDR) {
1047                 debug("OF: not found !\n");
1048                 return 1;
1049         }
1050         memcpy(addr, ranges + na, 4 * pna);
1051
1052  finish:
1053         of_dump_addr("OF: parent translation for:", addr, pna);
1054         debug("OF: with offset: "PRu64"\n", offset);
1055
1056         /* Translate it into parent bus space */
1057         return pbus->translate(addr, offset, pna);
1058 }
1059
1060 /*
1061  * Translate an address from the device-tree into a CPU physical address,
1062  * this walks up the tree and applies the various bus mappings on the
1063  * way.
1064  *
1065  * Note: We consider that crossing any level with #size-cells == 0 to mean
1066  * that translation is impossible (that is we are not dealing with a value
1067  * that can be mapped to a cpu physical address). This is not really specified
1068  * that way, but this is traditionally the way IBM at least do things
1069  */
1070 u64 __of_translate_address(void *blob, int node_offset, const u32 *in_addr,
1071                            const char *rprop)
1072 {
1073         int parent;
1074         struct of_bus *bus, *pbus;
1075         u32 addr[OF_MAX_ADDR_CELLS];
1076         int na, ns, pna, pns;
1077         u64 result = OF_BAD_ADDR;
1078
1079         debug("OF: ** translation for device %s **\n",
1080                 fdt_get_name(blob, node_offset, NULL));
1081
1082         /* Get parent & match bus type */
1083         parent = fdt_parent_offset(blob, node_offset);
1084         if (parent < 0)
1085                 goto bail;
1086         bus = &of_busses[0];
1087
1088         /* Cound address cells & copy address locally */
1089         bus->count_cells(blob, parent, &na, &ns);
1090         if (!OF_CHECK_COUNTS(na, ns)) {
1091                 printf("%s: Bad cell count for %s\n", __FUNCTION__,
1092                        fdt_get_name(blob, node_offset, NULL));
1093                 goto bail;
1094         }
1095         memcpy(addr, in_addr, na * 4);
1096
1097         debug("OF: bus is %s (na=%d, ns=%d) on %s\n",
1098             bus->name, na, ns, fdt_get_name(blob, parent, NULL));
1099         of_dump_addr("OF: translating address:", addr, na);
1100
1101         /* Translate */
1102         for (;;) {
1103                 /* Switch to parent bus */
1104                 node_offset = parent;
1105                 parent = fdt_parent_offset(blob, node_offset);
1106
1107                 /* If root, we have finished */
1108                 if (parent < 0) {
1109                         debug("OF: reached root node\n");
1110                         result = of_read_number(addr, na);
1111                         break;
1112                 }
1113
1114                 /* Get new parent bus and counts */
1115                 pbus = &of_busses[0];
1116                 pbus->count_cells(blob, parent, &pna, &pns);
1117                 if (!OF_CHECK_COUNTS(pna, pns)) {
1118                         printf("%s: Bad cell count for %s\n", __FUNCTION__,
1119                                 fdt_get_name(blob, node_offset, NULL));
1120                         break;
1121                 }
1122
1123                 debug("OF: parent bus is %s (na=%d, ns=%d) on %s\n",
1124                     pbus->name, pna, pns, fdt_get_name(blob, parent, NULL));
1125
1126                 /* Apply bus translation */
1127                 if (of_translate_one(blob, node_offset, bus, pbus,
1128                                         addr, na, ns, pna, rprop))
1129                         break;
1130
1131                 /* Complete the move up one level */
1132                 na = pna;
1133                 ns = pns;
1134                 bus = pbus;
1135
1136                 of_dump_addr("OF: one level translation:", addr, na);
1137         }
1138  bail:
1139
1140         return result;
1141 }
1142
1143 u64 fdt_translate_address(void *blob, int node_offset, const u32 *in_addr)
1144 {
1145         return __of_translate_address(blob, node_offset, in_addr, "ranges");
1146 }
1147
1148 /**
1149  * fdt_node_offset_by_compat_reg: Find a node that matches compatiable and
1150  * who's reg property matches a physical cpu address
1151  *
1152  * @blob: ptr to device tree
1153  * @compat: compatiable string to match
1154  * @compat_off: property name
1155  *
1156  */
1157 int fdt_node_offset_by_compat_reg(void *blob, const char *compat,
1158                                         phys_addr_t compat_off)
1159 {
1160         int len, off = fdt_node_offset_by_compatible(blob, -1, compat);
1161         while (off != -FDT_ERR_NOTFOUND) {
1162                 u32 *reg = (u32 *)fdt_getprop(blob, off, "reg", &len);
1163                 if (reg) {
1164                         if (compat_off == fdt_translate_address(blob, off, reg))
1165                                 return off;
1166                 }
1167                 off = fdt_node_offset_by_compatible(blob, off, compat);
1168         }
1169
1170         return -FDT_ERR_NOTFOUND;
1171 }
1172
1173 /**
1174  * fdt_alloc_phandle: Return next free phandle value
1175  *
1176  * @blob: ptr to device tree
1177  */
1178 int fdt_alloc_phandle(void *blob)
1179 {
1180         int offset, len, phandle = 0;
1181         const u32 *val;
1182
1183         for (offset = fdt_next_node(blob, -1, NULL); offset >= 0;
1184              offset = fdt_next_node(blob, offset, NULL)) {
1185                 val = fdt_getprop(blob, offset, "linux,phandle", &len);
1186                 if (val)
1187                         phandle = max(*val, phandle);
1188         }
1189
1190         return phandle + 1;
1191 }