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 = getenv("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 = getenv("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;
 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         len = fdt_pack_reg(blob, tmp, start, size, banks);
 454
 455         err = fdt_setprop(blob, nodeoffset, "reg", tmp, len);
 456         if (err < 0) {
 457                 printf("WARNING: could not set %s %s.\n",
 458                                 "reg", fdt_strerror(err));
 459                 return err;
 460         }
 461         return 0;
 462 }
 463 #endif
 464
 465 int fdt_fixup_memory(void *blob, u64 start, u64 size)
 466 {
 467         return fdt_fixup_memory_banks(blob, &start, &size, 1);
 468 }
 469
 470 void fdt_fixup_ethernet(void *fdt)
 471 {
 472         int i, j, prop;
 473         char *tmp, *end;
 474         char mac[16];
 475         const char *path;
 476         unsigned char mac_addr[ARP_HLEN];
 477         int offset;
 478
 479         if (fdt_path_offset(fdt, "/aliases") < 0)
 480                 return;
 481
 482         /* Cycle through all aliases */
 483         for (prop = 0; ; prop++) {
 484                 const char *name;
 485                 int len = strlen("ethernet");
 486
 487                 /* FDT might have been edited, recompute the offset */
 488                 offset = fdt_first_property_offset(fdt,
 489                         fdt_path_offset(fdt, "/aliases"));
 490                 /* Select property number 'prop' */
 491                 for (i = 0; i < prop; i++)
 492                         offset = fdt_next_property_offset(fdt, offset);
 493
 494                 if (offset < 0)
 495                         break;
 496
 497                 path = fdt_getprop_by_offset(fdt, offset, &name, NULL);
 498                 if (!strncmp(name, "ethernet", len)) {
 499                         i = trailing_strtol(name);
 500                         if (i != -1) {
 501                                 if (i == 0)
 502                                         strcpy(mac, "ethaddr");
 503                                 else
 504                                         sprintf(mac, "eth%daddr", i);
 505                         } else {
 506                                 continue;
 507                         }
 508                         tmp = getenv(mac);
 509                         if (!tmp)
 510                                 continue;
 511
 512                         for (j = 0; j < 6; j++) {
 513                                 mac_addr[j] = tmp ?
 514                                               simple_strtoul(tmp, &end, 16) : 0;
 515                                 if (tmp)
 516                                         tmp = (*end) ? end + 1 : end;
 517                         }
 518
 519                         do_fixup_by_path(fdt, path, "mac-address",
 520                                          &mac_addr, 6, 0);
 521                         do_fixup_by_path(fdt, path, "local-mac-address",
 522                                          &mac_addr, 6, 1);
 523                 }
 524         }
 525 }
 526
 527 /* Resize the fdt to its actual size + a bit of padding */
 528 int fdt_shrink_to_minimum(void *blob, uint extrasize)
 529 {
 530         int i;
 531         uint64_t addr, size;
 532         int total, ret;
 533         uint actualsize;
 534
 535         if (!blob)
 536                 return 0;
 537
 538         total = fdt_num_mem_rsv(blob);
 539         for (i = 0; i < total; i++) {
 540                 fdt_get_mem_rsv(blob, i, &addr, &size);
 541                 if (addr == (uintptr_t)blob) {
 542                         fdt_del_mem_rsv(blob, i);
 543                         break;
 544                 }
 545         }
 546
 547         /*
 548          * Calculate the actual size of the fdt
 549          * plus the size needed for 5 fdt_add_mem_rsv, one
 550          * for the fdt itself and 4 for a possible initrd
 551          * ((initrd-start + initrd-end) * 2 (name & value))
 552          */
 553         actualsize = fdt_off_dt_strings(blob) +
 554                 fdt_size_dt_strings(blob) + 5 * sizeof(struct fdt_reserve_entry);
 555
 556         actualsize += extrasize;
 557         /* Make it so the fdt ends on a page boundary */
 558         actualsize = ALIGN(actualsize + ((uintptr_t)blob & 0xfff), 0x1000);
 559         actualsize = actualsize - ((uintptr_t)blob & 0xfff);
 560
 561         /* Change the fdt header to reflect the correct size */
 562         fdt_set_totalsize(blob, actualsize);
 563
 564         /* Add the new reservation */
 565         ret = fdt_add_mem_rsv(blob, (uintptr_t)blob, actualsize);
 566         if (ret < 0)
 567                 return ret;
 568
 569         return actualsize;
 570 }
 571
 572 #ifdef CONFIG_PCI
 573 #define CONFIG_SYS_PCI_NR_INBOUND_WIN 4
 574
 575 #define FDT_PCI_PREFETCH        (0x40000000)
 576 #define FDT_PCI_MEM32           (0x02000000)
 577 #define FDT_PCI_IO              (0x01000000)
 578 #define FDT_PCI_MEM64           (0x03000000)
 579
 580 int fdt_pci_dma_ranges(void *blob, int phb_off, struct pci_controller *hose) {
 581
 582         int addrcell, sizecell, len, r;
 583         u32 *dma_range;
 584         /* sized based on pci addr cells, size-cells, & address-cells */
 585         u32 dma_ranges[(3 + 2 + 2) * CONFIG_SYS_PCI_NR_INBOUND_WIN];
 586
 587         addrcell = fdt_getprop_u32_default(blob, "/", "#address-cells", 1);
 588         sizecell = fdt_getprop_u32_default(blob, "/", "#size-cells", 1);
 589
 590         dma_range = &dma_ranges[0];
 591         for (r = 0; r < hose->region_count; r++) {
 592                 u64 bus_start, phys_start, size;
 593
 594                 /* skip if !PCI_REGION_SYS_MEMORY */
 595                 if (!(hose->regions[r].flags & PCI_REGION_SYS_MEMORY))
 596                         continue;
 597
 598                 bus_start = (u64)hose->regions[r].bus_start;
 599                 phys_start = (u64)hose->regions[r].phys_start;
 600                 size = (u64)hose->regions[r].size;
 601
 602                 dma_range[0] = 0;
 603                 if (size >= 0x100000000ull)
 604                         dma_range[0] |= FDT_PCI_MEM64;
 605                 else
 606                         dma_range[0] |= FDT_PCI_MEM32;
 607                 if (hose->regions[r].flags & PCI_REGION_PREFETCH)
 608                         dma_range[0] |= FDT_PCI_PREFETCH;
 609 #ifdef CONFIG_SYS_PCI_64BIT
 610                 dma_range[1] = bus_start >> 32;
 611 #else
 612                 dma_range[1] = 0;
 613 #endif
 614                 dma_range[2] = bus_start & 0xffffffff;
 615
 616                 if (addrcell == 2) {
 617                         dma_range[3] = phys_start >> 32;
 618                         dma_range[4] = phys_start & 0xffffffff;
 619                 } else {
 620                         dma_range[3] = phys_start & 0xffffffff;
 621                 }
 622
 623                 if (sizecell == 2) {
 624                         dma_range[3 + addrcell + 0] = size >> 32;
 625                         dma_range[3 + addrcell + 1] = size & 0xffffffff;
 626                 } else {
 627                         dma_range[3 + addrcell + 0] = size & 0xffffffff;
 628                 }
 629
 630                 dma_range += (3 + addrcell + sizecell);
 631         }
 632
 633         len = dma_range - &dma_ranges[0];
 634         if (len)
 635                 fdt_setprop(blob, phb_off, "dma-ranges", &dma_ranges[0], len*4);
 636
 637         return 0;
 638 }
 639 #endif
 640
 641 #ifdef CONFIG_FDT_FIXUP_NOR_FLASH_SIZE
 642 /*
 643  * Provide a weak default function to return the flash bank size.
 644  * There might be multiple non-identical flash chips connected to one
 645  * chip-select, so we need to pass an index as well.
 646  */
 647 u32 __flash_get_bank_size(int cs, int idx)
 648 {
 649         extern flash_info_t flash_info[];
 650
 651         /*
 652          * As default, a simple 1:1 mapping is provided. Boards with
 653          * a different mapping need to supply a board specific mapping
 654          * routine.
 655          */
 656         return flash_info[cs].size;
 657 }
 658 u32 flash_get_bank_size(int cs, int idx)
 659         __attribute__((weak, alias("__flash_get_bank_size")));
 660
 661 /*
 662  * This function can be used to update the size in the "reg" property
 663  * of all NOR FLASH device nodes. This is necessary for boards with
 664  * non-fixed NOR FLASH sizes.
 665  */
 666 int fdt_fixup_nor_flash_size(void *blob)
 667 {
 668         char compat[][16] = { "cfi-flash", "jedec-flash" };
 669         int off;
 670         int len;
 671         struct fdt_property *prop;
 672         u32 *reg, *reg2;
 673         int i;
 674
 675         for (i = 0; i < 2; i++) {
 676                 off = fdt_node_offset_by_compatible(blob, -1, compat[i]);
 677                 while (off != -FDT_ERR_NOTFOUND) {
 678                         int idx;
 679
 680                         /*
 681                          * Found one compatible node, so fixup the size
 682                          * int its reg properties
 683                          */
 684                         prop = fdt_get_property_w(blob, off, "reg", &len);
 685                         if (prop) {
 686                                 int tuple_size = 3 * sizeof(reg);
 687
 688                                 /*
 689                                  * There might be multiple reg-tuples,
 690                                  * so loop through them all
 691                                  */
 692                                 reg = reg2 = (u32 *)&prop->data[0];
 693                                 for (idx = 0; idx < (len / tuple_size); idx++) {
 694                                         /*
 695                                          * Update size in reg property
 696                                          */
 697                                         reg[2] = flash_get_bank_size(reg[0],
 698                                                                      idx);
 699
 700                                         /*
 701                                          * Point to next reg tuple
 702                                          */
 703                                         reg += 3;
 704                                 }
 705
 706                                 fdt_setprop(blob, off, "reg", reg2, len);
 707                         }
 708
 709                         /* Move to next compatible node */
 710                         off = fdt_node_offset_by_compatible(blob, off,
 711                                                             compat[i]);
 712                 }
 713         }
 714
 715         return 0;
 716 }
 717 #endif
 718
 719 int fdt_increase_size(void *fdt, int add_len)
 720 {
 721         int newlen;
 722
 723         newlen = fdt_totalsize(fdt) + add_len;
 724
 725         /* Open in place with a new len */
 726         return fdt_open_into(fdt, fdt, newlen);
 727 }
 728
 729 #ifdef CONFIG_FDT_FIXUP_PARTITIONS
 730 #include <jffs2/load_kernel.h>
 731 #include <mtd_node.h>
 732
 733 struct reg_cell {
 734         unsigned int r0;
 735         unsigned int r1;
 736 };
 737
 738 int fdt_del_subnodes(const void *blob, int parent_offset)
 739 {
 740         int off, ndepth;
 741         int ret;
 742
 743         for (ndepth = 0, off = fdt_next_node(blob, parent_offset, &ndepth);
 744              (off >= 0) && (ndepth > 0);
 745              off = fdt_next_node(blob, off, &ndepth)) {
 746                 if (ndepth == 1) {
 747                         debug("delete %s: offset: %x\n",
 748                                 fdt_get_name(blob, off, 0), off);
 749                         ret = fdt_del_node((void *)blob, off);
 750                         if (ret < 0) {
 751                                 printf("Can't delete node: %s\n",
 752                                         fdt_strerror(ret));
 753                                 return ret;
 754                         } else {
 755                                 ndepth = 0;
 756                                 off = parent_offset;
 757                         }
 758                 }
 759         }
 760         return 0;
 761 }
 762
 763 int fdt_del_partitions(void *blob, int parent_offset)
 764 {
 765         const void *prop;
 766         int ndepth = 0;
 767         int off;
 768         int ret;
 769
 770         off = fdt_next_node(blob, parent_offset, &ndepth);
 771         if (off > 0 && ndepth == 1) {
 772                 prop = fdt_getprop(blob, off, "label", NULL);
 773                 if (prop == NULL) {
 774                         /*
 775                          * Could not find label property, nand {}; node?
 776                          * Check subnode, delete partitions there if any.
 777                          */
 778                         return fdt_del_partitions(blob, off);
 779                 } else {
 780                         ret = fdt_del_subnodes(blob, parent_offset);
 781                         if (ret < 0) {
 782                                 printf("Can't remove subnodes: %s\n",
 783                                         fdt_strerror(ret));
 784                                 return ret;
 785                         }
 786                 }
 787         }
 788         return 0;
 789 }
 790
 791 int fdt_node_set_part_info(void *blob, int parent_offset,
 792                            struct mtd_device *dev)
 793 {
 794         struct list_head *pentry;
 795         struct part_info *part;
 796         struct reg_cell cell;
 797         int off, ndepth = 0;
 798         int part_num, ret;
 799         char buf[64];
 800
 801         ret = fdt_del_partitions(blob, parent_offset);
 802         if (ret < 0)
 803                 return ret;
 804
 805         /*
 806          * Check if it is nand {}; subnode, adjust
 807          * the offset in this case
 808          */
 809         off = fdt_next_node(blob, parent_offset, &ndepth);
 810         if (off > 0 && ndepth == 1)
 811                 parent_offset = off;
 812
 813         part_num = 0;
 814         list_for_each_prev(pentry, &dev->parts) {
 815                 int newoff;
 816
 817                 part = list_entry(pentry, struct part_info, link);
 818
 819                 debug("%2d: %-20s0x%08llx\t0x%08llx\t%d\n",
 820                         part_num, part->name, part->size,
 821                         part->offset, part->mask_flags);
 822
 823                 sprintf(buf, "partition@%llx", part->offset);
 824 add_sub:
 825                 ret = fdt_add_subnode(blob, parent_offset, buf);
 826                 if (ret == -FDT_ERR_NOSPACE) {
 827                         ret = fdt_increase_size(blob, 512);
 828                         if (!ret)
 829                                 goto add_sub;
 830                         else
 831                                 goto err_size;
 832                 } else if (ret < 0) {
 833                         printf("Can't add partition node: %s\n",
 834                                 fdt_strerror(ret));
 835                         return ret;
 836                 }
 837                 newoff = ret;
 838
 839                 /* Check MTD_WRITEABLE_CMD flag */
 840                 if (part->mask_flags & 1) {
 841 add_ro:
 842                         ret = fdt_setprop(blob, newoff, "read_only", NULL, 0);
 843                         if (ret == -FDT_ERR_NOSPACE) {
 844                                 ret = fdt_increase_size(blob, 512);
 845                                 if (!ret)
 846                                         goto add_ro;
 847                                 else
 848                                         goto err_size;
 849                         } else if (ret < 0)
 850                                 goto err_prop;
 851                 }
 852
 853                 cell.r0 = cpu_to_fdt32(part->offset);
 854                 cell.r1 = cpu_to_fdt32(part->size);
 855 add_reg:
 856                 ret = fdt_setprop(blob, newoff, "reg", &cell, sizeof(cell));
 857                 if (ret == -FDT_ERR_NOSPACE) {
 858                         ret = fdt_increase_size(blob, 512);
 859                         if (!ret)
 860                                 goto add_reg;
 861                         else
 862                                 goto err_size;
 863                 } else if (ret < 0)
 864                         goto err_prop;
 865
 866 add_label:
 867                 ret = fdt_setprop_string(blob, newoff, "label", part->name);
 868                 if (ret == -FDT_ERR_NOSPACE) {
 869                         ret = fdt_increase_size(blob, 512);
 870                         if (!ret)
 871                                 goto add_label;
 872                         else
 873                                 goto err_size;
 874                 } else if (ret < 0)
 875                         goto err_prop;
 876
 877                 part_num++;
 878         }
 879         return 0;
 880 err_size:
 881         printf("Can't increase blob size: %s\n", fdt_strerror(ret));
 882         return ret;
 883 err_prop:
 884         printf("Can't add property: %s\n", fdt_strerror(ret));
 885         return ret;
 886 }
 887
 888 /*
 889  * Update partitions in nor/nand nodes using info from
 890  * mtdparts environment variable. The nodes to update are
 891  * specified by node_info structure which contains mtd device
 892  * type and compatible string: E. g. the board code in
 893  * ft_board_setup() could use:
 894  *
 895  *      struct node_info nodes[] = {
 896  *              { "fsl,mpc5121-nfc",    MTD_DEV_TYPE_NAND, },
 897  *              { "cfi-flash",          MTD_DEV_TYPE_NOR,  },
 898  *      };
 899  *
 900  *      fdt_fixup_mtdparts(blob, nodes, ARRAY_SIZE(nodes));
 901  */
 902 void fdt_fixup_mtdparts(void *blob, void *node_info, int node_info_size)
 903 {
 904         struct node_info *ni = node_info;
 905         struct mtd_device *dev;
 906         int i, idx;
 907         int noff;
 908
 909         if (mtdparts_init() != 0)
 910                 return;
 911
 912         for (i = 0; i < node_info_size; i++) {
 913                 idx = 0;
 914                 noff = fdt_node_offset_by_compatible(blob, -1, ni[i].compat);
 915                 while (noff != -FDT_ERR_NOTFOUND) {
 916                         debug("%s: %s, mtd dev type %d\n",
 917                                 fdt_get_name(blob, noff, 0),
 918                                 ni[i].compat, ni[i].type);
 919                         dev = device_find(ni[i].type, idx++);
 920                         if (dev) {
 921                                 if (fdt_node_set_part_info(blob, noff, dev))
 922                                         return; /* return on error */
 923                         }
 924
 925                         /* Jump to next flash node */
 926                         noff = fdt_node_offset_by_compatible(blob, noff,
 927                                                              ni[i].compat);
 928                 }
 929         }
 930 }
 931 #endif
 932
 933 void fdt_del_node_and_alias(void *blob, const char *alias)
 934 {
 935         int off = fdt_path_offset(blob, alias);
 936
 937         if (off < 0)
 938                 return;
 939
 940         fdt_del_node(blob, off);
 941
 942         off = fdt_path_offset(blob, "/aliases");
 943         fdt_delprop(blob, off, alias);
 944 }
 945
 946 /* Max address size we deal with */
 947 #define OF_MAX_ADDR_CELLS       4
 948 #define OF_BAD_ADDR     FDT_ADDR_T_NONE
 949 #define OF_CHECK_COUNTS(na, ns) ((na) > 0 && (na) <= OF_MAX_ADDR_CELLS && \
 950                         (ns) > 0)
 951
 952 /* Debug utility */
 953 #ifdef DEBUG
 954 static void of_dump_addr(const char *s, const fdt32_t *addr, int na)
 955 {
 956         printf("%s", s);
 957         while(na--)
 958                 printf(" %08x", *(addr++));
 959         printf("\n");
 960 }
 961 #else
 962 static void of_dump_addr(const char *s, const fdt32_t *addr, int na) { }
 963 #endif
 964
 965 /**
 966  * struct of_bus - Callbacks for bus specific translators
 967  * @name:       A string used to identify this bus in debug output.
 968  * @addresses:  The name of the DT property from which addresses are
 969  *              to be read, typically "reg".
 970  * @match:      Return non-zero if the node whose parent is at
 971  *              parentoffset in the FDT blob corresponds to a bus
 972  *              of this type, otherwise return zero. If NULL a match
 973  *              is assumed.
 974  * @count_cells:Count how many cells (be32 values) a node whose parent
 975  *              is at parentoffset in the FDT blob will require to
 976  *              represent its address (written to *addrc) & size
 977  *              (written to *sizec).
 978  * @map:        Map the address addr from the address space of this
 979  *              bus to that of its parent, making use of the ranges
 980  *              read from DT to an array at range. na and ns are the
 981  *              number of cells (be32 values) used to hold and address
 982  *              or size, respectively, for this bus. pna is the number
 983  *              of cells used to hold an address for the parent bus.
 984  *              Returns the address in the address space of the parent
 985  *              bus.
 986  * @translate:  Update the value of the address cells at addr within an
 987  *              FDT by adding offset to it. na specifies the number of
 988  *              cells used to hold the address being translated. Returns
 989  *              zero on success, non-zero on error.
 990  *
 991  * Each bus type will include a struct of_bus in the of_busses array,
 992  * providing implementations of some or all of the functions used to
 993  * match the bus & handle address translation for its children.
 994  */
 995 struct of_bus {
 996         const char      *name;
 997         const char      *addresses;
 998         int             (*match)(const void *blob, int parentoffset);
 999         void            (*count_cells)(const void *blob, int parentoffset,
1000                                 int *addrc, int *sizec);
1001         u64             (*map)(fdt32_t *addr, const fdt32_t *range,
1002                                 int na, int ns, int pna);
1003         int             (*translate)(fdt32_t *addr, u64 offset, int na);
1004 };
1005
1006 /* Default translator (generic bus) */
1007 void of_bus_default_count_cells(const void *blob, int parentoffset,
1008                                         int *addrc, int *sizec)
1009 {
1010         const fdt32_t *prop;
1011
1012         if (addrc)
1013                 *addrc = fdt_address_cells(blob, parentoffset);
1014
1015         if (sizec) {
1016                 prop = fdt_getprop(blob, parentoffset, "#size-cells", NULL);
1017                 if (prop)
1018                         *sizec = be32_to_cpup(prop);
1019                 else
1020                         *sizec = 1;
1021         }
1022 }
1023
1024 static u64 of_bus_default_map(fdt32_t *addr, const fdt32_t *range,
1025                 int na, int ns, int pna)
1026 {
1027         u64 cp, s, da;
1028
1029         cp = of_read_number(range, na);
1030         s  = of_read_number(range + na + pna, ns);
1031         da = of_read_number(addr, na);
1032
1033         debug("OF: default map, cp=%" PRIu64 ", s=%" PRIu64
1034               ", da=%" PRIu64 "\n", cp, s, da);
1035
1036         if (da < cp || da >= (cp + s))
1037                 return OF_BAD_ADDR;
1038         return da - cp;
1039 }
1040
1041 static int of_bus_default_translate(fdt32_t *addr, u64 offset, int na)
1042 {
1043         u64 a = of_read_number(addr, na);
1044         memset(addr, 0, na * 4);
1045         a += offset;
1046         if (na > 1)
1047                 addr[na - 2] = cpu_to_fdt32(a >> 32);
1048         addr[na - 1] = cpu_to_fdt32(a & 0xffffffffu);
1049
1050         return 0;
1051 }
1052
1053 #ifdef CONFIG_OF_ISA_BUS
1054
1055 /* ISA bus translator */
1056 static int of_bus_isa_match(const void *blob, int parentoffset)
1057 {
1058         const char *name;
1059
1060         name = fdt_get_name(blob, parentoffset, NULL);
1061         if (!name)
1062                 return 0;
1063
1064         return !strcmp(name, "isa");
1065 }
1066
1067 static void of_bus_isa_count_cells(const void *blob, int parentoffset,
1068                                    int *addrc, int *sizec)
1069 {
1070         if (addrc)
1071                 *addrc = 2;
1072         if (sizec)
1073                 *sizec = 1;
1074 }
1075
1076 static u64 of_bus_isa_map(fdt32_t *addr, const fdt32_t *range,
1077                           int na, int ns, int pna)
1078 {
1079         u64 cp, s, da;
1080
1081         /* Check address type match */
1082         if ((addr[0] ^ range[0]) & cpu_to_be32(1))
1083                 return OF_BAD_ADDR;
1084
1085         cp = of_read_number(range + 1, na - 1);
1086         s  = of_read_number(range + na + pna, ns);
1087         da = of_read_number(addr + 1, na - 1);
1088
1089         debug("OF: ISA map, cp=%" PRIu64 ", s=%" PRIu64
1090               ", da=%" PRIu64 "\n", cp, s, da);
1091
1092         if (da < cp || da >= (cp + s))
1093                 return OF_BAD_ADDR;
1094         return da - cp;
1095 }
1096
1097 static int of_bus_isa_translate(fdt32_t *addr, u64 offset, int na)
1098 {
1099         return of_bus_default_translate(addr + 1, offset, na - 1);
1100 }
1101
1102 #endif /* CONFIG_OF_ISA_BUS */
1103
1104 /* Array of bus specific translators */
1105 static struct of_bus of_busses[] = {
1106 #ifdef CONFIG_OF_ISA_BUS
1107         /* ISA */
1108         {
1109                 .name = "isa",
1110                 .addresses = "reg",
1111                 .match = of_bus_isa_match,
1112                 .count_cells = of_bus_isa_count_cells,
1113                 .map = of_bus_isa_map,
1114                 .translate = of_bus_isa_translate,
1115         },
1116 #endif /* CONFIG_OF_ISA_BUS */
1117         /* Default */
1118         {
1119                 .name = "default",
1120                 .addresses = "reg",
1121                 .count_cells = of_bus_default_count_cells,
1122                 .map = of_bus_default_map,
1123                 .translate = of_bus_default_translate,
1124         },
1125 };
1126
1127 static struct of_bus *of_match_bus(const void *blob, int parentoffset)
1128 {
1129         struct of_bus *bus;
1130
1131         if (ARRAY_SIZE(of_busses) == 1)
1132                 return of_busses;
1133
1134         for (bus = of_busses; bus; bus++) {
1135                 if (!bus->match || bus->match(blob, parentoffset))
1136                         return bus;
1137         }
1138
1139         /*
1140          * We should always have matched the default bus at least, since
1141          * it has a NULL match field. If we didn't then it somehow isn't
1142          * in the of_busses array or something equally catastrophic has
1143          * gone wrong.
1144          */
1145         assert(0);
1146         return NULL;
1147 }
1148
1149 static int of_translate_one(const void *blob, int parent, struct of_bus *bus,
1150                             struct of_bus *pbus, fdt32_t *addr,
1151                             int na, int ns, int pna, const char *rprop)
1152 {
1153         const fdt32_t *ranges;
1154         int rlen;
1155         int rone;
1156         u64 offset = OF_BAD_ADDR;
1157
1158         /* Normally, an absence of a "ranges" property means we are
1159          * crossing a non-translatable boundary, and thus the addresses
1160          * below the current not cannot be converted to CPU physical ones.
1161          * Unfortunately, while this is very clear in the spec, it's not
1162          * what Apple understood, and they do have things like /uni-n or
1163          * /ht nodes with no "ranges" property and a lot of perfectly
1164          * useable mapped devices below them. Thus we treat the absence of
1165          * "ranges" as equivalent to an empty "ranges" property which means
1166          * a 1:1 translation at that level. It's up to the caller not to try
1167          * to translate addresses that aren't supposed to be translated in
1168          * the first place. --BenH.
1169          */
1170         ranges = fdt_getprop(blob, parent, rprop, &rlen);
1171         if (ranges == NULL || rlen == 0) {
1172                 offset = of_read_number(addr, na);
1173                 memset(addr, 0, pna * 4);
1174                 debug("OF: no ranges, 1:1 translation\n");
1175                 goto finish;
1176         }
1177
1178         debug("OF: walking ranges...\n");
1179
1180         /* Now walk through the ranges */
1181         rlen /= 4;
1182         rone = na + pna + ns;
1183         for (; rlen >= rone; rlen -= rone, ranges += rone) {
1184                 offset = bus->map(addr, ranges, na, ns, pna);
1185                 if (offset != OF_BAD_ADDR)
1186                         break;
1187         }
1188         if (offset == OF_BAD_ADDR) {
1189                 debug("OF: not found !\n");
1190                 return 1;
1191         }
1192         memcpy(addr, ranges + na, 4 * pna);
1193
1194  finish:
1195         of_dump_addr("OF: parent translation for:", addr, pna);
1196         debug("OF: with offset: %" PRIu64 "\n", offset);
1197
1198         /* Translate it into parent bus space */
1199         return pbus->translate(addr, offset, pna);
1200 }
1201
1202 /*
1203  * Translate an address from the device-tree into a CPU physical address,
1204  * this walks up the tree and applies the various bus mappings on the
1205  * way.
1206  *
1207  * Note: We consider that crossing any level with #size-cells == 0 to mean
1208  * that translation is impossible (that is we are not dealing with a value
1209  * that can be mapped to a cpu physical address). This is not really specified
1210  * that way, but this is traditionally the way IBM at least do things
1211  */
1212 static u64 __of_translate_address(const void *blob, int node_offset,
1213                                   const fdt32_t *in_addr, const char *rprop)
1214 {
1215         int parent;
1216         struct of_bus *bus, *pbus;
1217         fdt32_t addr[OF_MAX_ADDR_CELLS];
1218         int na, ns, pna, pns;
1219         u64 result = OF_BAD_ADDR;
1220
1221         debug("OF: ** translation for device %s **\n",
1222                 fdt_get_name(blob, node_offset, NULL));
1223
1224         /* Get parent & match bus type */
1225         parent = fdt_parent_offset(blob, node_offset);
1226         if (parent < 0)
1227                 goto bail;
1228         bus = of_match_bus(blob, parent);
1229
1230         /* Cound address cells & copy address locally */
1231         bus->count_cells(blob, parent, &na, &ns);
1232         if (!OF_CHECK_COUNTS(na, ns)) {
1233                 printf("%s: Bad cell count for %s\n", __FUNCTION__,
1234                        fdt_get_name(blob, node_offset, NULL));
1235                 goto bail;
1236         }
1237         memcpy(addr, in_addr, na * 4);
1238
1239         debug("OF: bus is %s (na=%d, ns=%d) on %s\n",
1240             bus->name, na, ns, fdt_get_name(blob, parent, NULL));
1241         of_dump_addr("OF: translating address:", addr, na);
1242
1243         /* Translate */
1244         for (;;) {
1245                 /* Switch to parent bus */
1246                 node_offset = parent;
1247                 parent = fdt_parent_offset(blob, node_offset);
1248
1249                 /* If root, we have finished */
1250                 if (parent < 0) {
1251                         debug("OF: reached root node\n");
1252                         result = of_read_number(addr, na);
1253                         break;
1254                 }
1255
1256                 /* Get new parent bus and counts */
1257                 pbus = of_match_bus(blob, parent);
1258                 pbus->count_cells(blob, parent, &pna, &pns);
1259                 if (!OF_CHECK_COUNTS(pna, pns)) {
1260                         printf("%s: Bad cell count for %s\n", __FUNCTION__,
1261                                 fdt_get_name(blob, node_offset, NULL));
1262                         break;
1263                 }
1264
1265                 debug("OF: parent bus is %s (na=%d, ns=%d) on %s\n",
1266                     pbus->name, pna, pns, fdt_get_name(blob, parent, NULL));
1267
1268                 /* Apply bus translation */
1269                 if (of_translate_one(blob, node_offset, bus, pbus,
1270                                         addr, na, ns, pna, rprop))
1271                         break;
1272
1273                 /* Complete the move up one level */
1274                 na = pna;
1275                 ns = pns;
1276                 bus = pbus;
1277
1278                 of_dump_addr("OF: one level translation:", addr, na);
1279         }
1280  bail:
1281
1282         return result;
1283 }
1284
1285 u64 fdt_translate_address(const void *blob, int node_offset,
1286                           const fdt32_t *in_addr)
1287 {
1288         return __of_translate_address(blob, node_offset, in_addr, "ranges");
1289 }
1290
1291 /**
1292  * fdt_node_offset_by_compat_reg: Find a node that matches compatiable and
1293  * who's reg property matches a physical cpu address
1294  *
1295  * @blob: ptr to device tree
1296  * @compat: compatiable string to match
1297  * @compat_off: property name
1298  *
1299  */
1300 int fdt_node_offset_by_compat_reg(void *blob, const char *compat,
1301                                         phys_addr_t compat_off)
1302 {
1303         int len, off = fdt_node_offset_by_compatible(blob, -1, compat);
1304         while (off != -FDT_ERR_NOTFOUND) {
1305                 const fdt32_t *reg = fdt_getprop(blob, off, "reg", &len);
1306                 if (reg) {
1307                         if (compat_off == fdt_translate_address(blob, off, reg))
1308                                 return off;
1309                 }
1310                 off = fdt_node_offset_by_compatible(blob, off, compat);
1311         }
1312
1313         return -FDT_ERR_NOTFOUND;
1314 }
1315
1316 /**
1317  * fdt_alloc_phandle: Return next free phandle value
1318  *
1319  * @blob: ptr to device tree
1320  */
1321 int fdt_alloc_phandle(void *blob)
1322 {
1323         int offset;
1324         uint32_t phandle = 0;
1325
1326         for (offset = fdt_next_node(blob, -1, NULL); offset >= 0;
1327              offset = fdt_next_node(blob, offset, NULL)) {
1328                 phandle = max(phandle, fdt_get_phandle(blob, offset));
1329         }
1330
1331         return phandle + 1;
1332 }
1333
1334 /*
1335  * fdt_set_phandle: Create a phandle property for the given node
1336  *
1337  * @fdt: ptr to device tree
1338  * @nodeoffset: node to update
1339  * @phandle: phandle value to set (must be unique)
1340  */
1341 int fdt_set_phandle(void *fdt, int nodeoffset, uint32_t phandle)
1342 {
1343         int ret;
1344
1345 #ifdef DEBUG
1346         int off = fdt_node_offset_by_phandle(fdt, phandle);
1347
1348         if ((off >= 0) && (off != nodeoffset)) {
1349                 char buf[64];
1350
1351                 fdt_get_path(fdt, nodeoffset, buf, sizeof(buf));
1352                 printf("Trying to update node %s with phandle %u ",
1353                        buf, phandle);
1354
1355                 fdt_get_path(fdt, off, buf, sizeof(buf));
1356                 printf("that already exists in node %s.\n", buf);
1357                 return -FDT_ERR_BADPHANDLE;
1358         }
1359 #endif
1360
1361         ret = fdt_setprop_cell(fdt, nodeoffset, "phandle", phandle);
1362         if (ret < 0)
1363                 return ret;
1364
1365         /*
1366          * For now, also set the deprecated "linux,phandle" property, so that we
1367          * don't break older kernels.
1368          */
1369         ret = fdt_setprop_cell(fdt, nodeoffset, "linux,phandle", phandle);
1370
1371         return ret;
1372 }
1373
1374 /*
1375  * fdt_create_phandle: Create a phandle property for the given node
1376  *
1377  * @fdt: ptr to device tree
1378  * @nodeoffset: node to update
1379  */
1380 unsigned int fdt_create_phandle(void *fdt, int nodeoffset)
1381 {
1382         /* see if there is a phandle already */
1383         int phandle = fdt_get_phandle(fdt, nodeoffset);
1384
1385         /* if we got 0, means no phandle so create one */
1386         if (phandle == 0) {
1387                 int ret;
1388
1389                 phandle = fdt_alloc_phandle(fdt);
1390                 ret = fdt_set_phandle(fdt, nodeoffset, phandle);
1391                 if (ret < 0) {
1392                         printf("Can't set phandle %u: %s\n", phandle,
1393                                fdt_strerror(ret));
1394                         return 0;
1395                 }
1396         }
1397
1398         return phandle;
1399 }
1400
1401 /*
1402  * fdt_set_node_status: Set status for the given node
1403  *
1404  * @fdt: ptr to device tree
1405  * @nodeoffset: node to update
1406  * @status: FDT_STATUS_OKAY, FDT_STATUS_DISABLED,
1407  *          FDT_STATUS_FAIL, FDT_STATUS_FAIL_ERROR_CODE
1408  * @error_code: optional, only used if status is FDT_STATUS_FAIL_ERROR_CODE
1409  */
1410 int fdt_set_node_status(void *fdt, int nodeoffset,
1411                         enum fdt_status status, unsigned int error_code)
1412 {
1413         char buf[16];
1414         int ret = 0;
1415
1416         if (nodeoffset < 0)
1417                 return nodeoffset;
1418
1419         switch (status) {
1420         case FDT_STATUS_OKAY:
1421                 ret = fdt_setprop_string(fdt, nodeoffset, "status", "okay");
1422                 break;
1423         case FDT_STATUS_DISABLED:
1424                 ret = fdt_setprop_string(fdt, nodeoffset, "status", "disabled");
1425                 break;
1426         case FDT_STATUS_FAIL:
1427                 ret = fdt_setprop_string(fdt, nodeoffset, "status", "fail");
1428                 break;
1429         case FDT_STATUS_FAIL_ERROR_CODE:
1430                 sprintf(buf, "fail-%d", error_code);
1431                 ret = fdt_setprop_string(fdt, nodeoffset, "status", buf);
1432                 break;
1433         default:
1434                 printf("Invalid fdt status: %x\n", status);
1435                 ret = -1;
1436                 break;
1437         }
1438
1439         return ret;
1440 }
1441
1442 /*
1443  * fdt_set_status_by_alias: Set status for the given node given an alias
1444  *
1445  * @fdt: ptr to device tree
1446  * @alias: alias of node to update
1447  * @status: FDT_STATUS_OKAY, FDT_STATUS_DISABLED,
1448  *          FDT_STATUS_FAIL, FDT_STATUS_FAIL_ERROR_CODE
1449  * @error_code: optional, only used if status is FDT_STATUS_FAIL_ERROR_CODE
1450  */
1451 int fdt_set_status_by_alias(void *fdt, const char* alias,
1452                             enum fdt_status status, unsigned int error_code)
1453 {
1454         int offset = fdt_path_offset(fdt, alias);
1455
1456         return fdt_set_node_status(fdt, offset, status, error_code);
1457 }
1458
1459 #if defined(CONFIG_VIDEO) || defined(CONFIG_LCD)
1460 int fdt_add_edid(void *blob, const char *compat, unsigned char *edid_buf)
1461 {
1462         int noff;
1463         int ret;
1464
1465         noff = fdt_node_offset_by_compatible(blob, -1, compat);
1466         if (noff != -FDT_ERR_NOTFOUND) {
1467                 debug("%s: %s\n", fdt_get_name(blob, noff, 0), compat);
1468 add_edid:
1469                 ret = fdt_setprop(blob, noff, "edid", edid_buf, 128);
1470                 if (ret == -FDT_ERR_NOSPACE) {
1471                         ret = fdt_increase_size(blob, 512);
1472                         if (!ret)
1473                                 goto add_edid;
1474                         else
1475                                 goto err_size;
1476                 } else if (ret < 0) {
1477                         printf("Can't add property: %s\n", fdt_strerror(ret));
1478                         return ret;
1479                 }
1480         }
1481         return 0;
1482 err_size:
1483         printf("Can't increase blob size: %s\n", fdt_strerror(ret));
1484         return ret;
1485 }
1486 #endif
1487
1488 /*
1489  * Verify the physical address of device tree node for a given alias
1490  *
1491  * This function locates the device tree node of a given alias, and then
1492  * verifies that the physical address of that device matches the given
1493  * parameter.  It displays a message if there is a mismatch.
1494  *
1495  * Returns 1 on success, 0 on failure
1496  */
1497 int fdt_verify_alias_address(void *fdt, int anode, const char *alias, u64 addr)
1498 {
1499         const char *path;
1500         const fdt32_t *reg;
1501         int node, len;
1502         u64 dt_addr;
1503
1504         path = fdt_getprop(fdt, anode, alias, NULL);
1505         if (!path) {
1506                 /* If there's no such alias, then it's not a failure */
1507                 return 1;
1508         }
1509
1510         node = fdt_path_offset(fdt, path);
1511         if (node < 0) {
1512                 printf("Warning: device tree alias '%s' points to invalid "
1513                        "node %s.\n", alias, path);
1514                 return 0;
1515         }
1516
1517         reg = fdt_getprop(fdt, node, "reg", &len);
1518         if (!reg) {
1519                 printf("Warning: device tree node '%s' has no address.\n",
1520                        path);
1521                 return 0;
1522         }
1523
1524         dt_addr = fdt_translate_address(fdt, node, reg);
1525         if (addr != dt_addr) {
1526                 printf("Warning: U-Boot configured device %s at address %"
1527                        PRIx64 ",\n but the device tree has it address %"
1528                        PRIx64 ".\n", alias, addr, dt_addr);
1529                 return 0;
1530         }
1531
1532         return 1;
1533 }
1534
1535 /*
1536  * Returns the base address of an SOC or PCI node
1537  */
1538 u64 fdt_get_base_address(void *fdt, int node)
1539 {
1540         int size;
1541         u32 naddr;
1542         const fdt32_t *prop;
1543
1544         naddr = fdt_address_cells(fdt, node);
1545
1546         prop = fdt_getprop(fdt, node, "ranges", &size);
1547
1548         return prop ? fdt_translate_address(fdt, node, prop + naddr) : 0;
1549 }
1550
1551 /*
1552  * Read a property of size <prop_len>. Currently only supports 1 or 2 cells.
1553  */
1554 static int fdt_read_prop(const fdt32_t *prop, int prop_len, int cell_off,
1555                          uint64_t *val, int cells)
1556 {
1557         const fdt32_t *prop32 = &prop[cell_off];
1558         const fdt64_t *prop64 = (const fdt64_t *)&prop[cell_off];
1559
1560         if ((cell_off + cells) > prop_len)
1561                 return -FDT_ERR_NOSPACE;
1562
1563         switch (cells) {
1564         case 1:
1565                 *val = fdt32_to_cpu(*prop32);
1566                 break;
1567         case 2:
1568                 *val = fdt64_to_cpu(*prop64);
1569                 break;
1570         default:
1571                 return -FDT_ERR_NOSPACE;
1572         }
1573
1574         return 0;
1575 }
1576
1577 /**
1578  * fdt_read_range - Read a node's n'th range property
1579  *
1580  * @fdt: ptr to device tree
1581  * @node: offset of node
1582  * @n: range index
1583  * @child_addr: pointer to storage for the "child address" field
1584  * @addr: pointer to storage for the CPU view translated physical start
1585  * @len: pointer to storage for the range length
1586  *
1587  * Convenience function that reads and interprets a specific range out of
1588  * a number of the "ranges" property array.
1589  */
1590 int fdt_read_range(void *fdt, int node, int n, uint64_t *child_addr,
1591                    uint64_t *addr, uint64_t *len)
1592 {
1593         int pnode = fdt_parent_offset(fdt, node);
1594         const fdt32_t *ranges;
1595         int pacells;
1596         int acells;
1597         int scells;
1598         int ranges_len;
1599         int cell = 0;
1600         int r = 0;
1601
1602         /*
1603          * The "ranges" property is an array of
1604          * { <child address> <parent address> <size in child address space> }
1605          *
1606          * All 3 elements can span a diffent number of cells. Fetch their size.
1607          */
1608         pacells = fdt_getprop_u32_default_node(fdt, pnode, 0, "#address-cells", 1);
1609         acells = fdt_getprop_u32_default_node(fdt, node, 0, "#address-cells", 1);
1610         scells = fdt_getprop_u32_default_node(fdt, node, 0, "#size-cells", 1);
1611
1612         /* Now try to get the ranges property */
1613         ranges = fdt_getprop(fdt, node, "ranges", &ranges_len);
1614         if (!ranges)
1615                 return -FDT_ERR_NOTFOUND;
1616         ranges_len /= sizeof(uint32_t);
1617
1618         /* Jump to the n'th entry */
1619         cell = n * (pacells + acells + scells);
1620
1621         /* Read <child address> */
1622         if (child_addr) {
1623                 r = fdt_read_prop(ranges, ranges_len, cell, child_addr,
1624                                   acells);
1625                 if (r)
1626                         return r;
1627         }
1628         cell += acells;
1629
1630         /* Read <parent address> */
1631         if (addr)
1632                 *addr = fdt_translate_address(fdt, node, ranges + cell);
1633         cell += pacells;
1634
1635         /* Read <size in child address space> */
1636         if (len) {
1637                 r = fdt_read_prop(ranges, ranges_len, cell, len, scells);
1638                 if (r)
1639                         return r;
1640         }
1641
1642         return 0;
1643 }
1644
1645 /**
1646  * fdt_setup_simplefb_node - Fill and enable a simplefb node
1647  *
1648  * @fdt: ptr to device tree
1649  * @node: offset of the simplefb node
1650  * @base_address: framebuffer base address
1651  * @width: width in pixels
1652  * @height: height in pixels
1653  * @stride: bytes per line
1654  * @format: pixel format string
1655  *
1656  * Convenience function to fill and enable a simplefb node.
1657  */
1658 int fdt_setup_simplefb_node(void *fdt, int node, u64 base_address, u32 width,
1659                             u32 height, u32 stride, const char *format)
1660 {
1661         char name[32];
1662         fdt32_t cells[4];
1663         int i, addrc, sizec, ret;
1664
1665         of_bus_default_count_cells(fdt, fdt_parent_offset(fdt, node),
1666                                    &addrc, &sizec);
1667         i = 0;
1668         if (addrc == 2)
1669                 cells[i++] = cpu_to_fdt32(base_address >> 32);
1670         cells[i++] = cpu_to_fdt32(base_address);
1671         if (sizec == 2)
1672                 cells[i++] = 0;
1673         cells[i++] = cpu_to_fdt32(height * stride);
1674
1675         ret = fdt_setprop(fdt, node, "reg", cells, sizeof(cells[0]) * i);
1676         if (ret < 0)
1677                 return ret;
1678
1679         snprintf(name, sizeof(name), "framebuffer@%" PRIx64, base_address);
1680         ret = fdt_set_name(fdt, node, name);
1681         if (ret < 0)
1682                 return ret;
1683
1684         ret = fdt_setprop_u32(fdt, node, "width", width);
1685         if (ret < 0)
1686                 return ret;
1687
1688         ret = fdt_setprop_u32(fdt, node, "height", height);
1689         if (ret < 0)
1690                 return ret;
1691
1692         ret = fdt_setprop_u32(fdt, node, "stride", stride);
1693         if (ret < 0)
1694                 return ret;
1695
1696         ret = fdt_setprop_string(fdt, node, "format", format);
1697         if (ret < 0)
1698                 return ret;
1699
1700         ret = fdt_setprop_string(fdt, node, "status", "okay");
1701         if (ret < 0)
1702                 return ret;
1703
1704         return 0;
1705 }
1706
1707 /*
1708  * Update native-mode in display-timings from display environment variable.
1709  * The node to update are specified by path.
1710  */
1711 int fdt_fixup_display(void *blob, const char *path, const char *display)
1712 {
1713         int off, toff;
1714
1715         if (!display || !path)
1716                 return -FDT_ERR_NOTFOUND;
1717
1718         toff = fdt_path_offset(blob, path);
1719         if (toff >= 0)
1720                 toff = fdt_subnode_offset(blob, toff, "display-timings");
1721         if (toff < 0)
1722                 return toff;
1723
1724         for (off = fdt_first_subnode(blob, toff);
1725              off >= 0;
1726              off = fdt_next_subnode(blob, off)) {
1727                 uint32_t h = fdt_get_phandle(blob, off);
1728                 debug("%s:0x%x\n", fdt_get_name(blob, off, NULL),
1729                       fdt32_to_cpu(h));
1730                 if (strcasecmp(fdt_get_name(blob, off, NULL), display) == 0)
1731                         return fdt_setprop_u32(blob, toff, "native-mode", h);
1732         }
1733         return toff;
1734 }