2 * (C) Copyright David Gibson <dwg@au1.ibm.com>, IBM Corporation. 2005.
5 * This program is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU General Public License as
7 * published by the Free Software Foundation; either version 2 of the
8 * License, or (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * General Public License for more details.
15 * You should have received a copy of the GNU General Public License
16 * along with this program; if not, write to the Free Software
17 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
24 * Tree building functions
27 void add_label(struct label
**labels
, char *label
)
31 /* Make sure the label isn't already there */
32 for_each_label_withdel(*labels
, new)
33 if (streq(new->label
, label
)) {
38 new = xmalloc(sizeof(*new));
39 memset(new, 0, sizeof(*new));
45 void delete_labels(struct label
**labels
)
49 for_each_label(*labels
, label
)
53 struct property
*build_property(char *name
, struct data val
)
55 struct property
*new = xmalloc(sizeof(*new));
57 memset(new, 0, sizeof(*new));
65 struct property
*build_property_delete(char *name
)
67 struct property
*new = xmalloc(sizeof(*new));
69 memset(new, 0, sizeof(*new));
77 struct property
*chain_property(struct property
*first
, struct property
*list
)
79 assert(first
->next
== NULL
);
85 struct property
*reverse_properties(struct property
*first
)
87 struct property
*p
= first
;
88 struct property
*head
= NULL
;
89 struct property
*next
;
100 struct node
*build_node(struct property
*proplist
, struct node
*children
)
102 struct node
*new = xmalloc(sizeof(*new));
105 memset(new, 0, sizeof(*new));
107 new->proplist
= reverse_properties(proplist
);
108 new->children
= children
;
110 for_each_child(new, child
) {
117 struct node
*build_node_delete(void)
119 struct node
*new = xmalloc(sizeof(*new));
121 memset(new, 0, sizeof(*new));
128 struct node
*name_node(struct node
*node
, char *name
)
130 assert(node
->name
== NULL
);
137 struct node
*merge_nodes(struct node
*old_node
, struct node
*new_node
)
139 struct property
*new_prop
, *old_prop
;
140 struct node
*new_child
, *old_child
;
143 old_node
->deleted
= 0;
145 /* Add new node labels to old node */
146 for_each_label_withdel(new_node
->labels
, l
)
147 add_label(&old_node
->labels
, l
->label
);
149 /* Move properties from the new node to the old node. If there
150 * is a collision, replace the old value with the new */
151 while (new_node
->proplist
) {
152 /* Pop the property off the list */
153 new_prop
= new_node
->proplist
;
154 new_node
->proplist
= new_prop
->next
;
155 new_prop
->next
= NULL
;
157 if (new_prop
->deleted
) {
158 delete_property_by_name(old_node
, new_prop
->name
);
163 /* Look for a collision, set new value if there is */
164 for_each_property_withdel(old_node
, old_prop
) {
165 if (streq(old_prop
->name
, new_prop
->name
)) {
166 /* Add new labels to old property */
167 for_each_label_withdel(new_prop
->labels
, l
)
168 add_label(&old_prop
->labels
, l
->label
);
170 old_prop
->val
= new_prop
->val
;
171 old_prop
->deleted
= 0;
178 /* if no collision occurred, add property to the old node. */
180 add_property(old_node
, new_prop
);
183 /* Move the override child nodes into the primary node. If
184 * there is a collision, then merge the nodes. */
185 while (new_node
->children
) {
186 /* Pop the child node off the list */
187 new_child
= new_node
->children
;
188 new_node
->children
= new_child
->next_sibling
;
189 new_child
->parent
= NULL
;
190 new_child
->next_sibling
= NULL
;
192 if (new_child
->deleted
) {
193 delete_node_by_name(old_node
, new_child
->name
);
198 /* Search for a collision. Merge if there is */
199 for_each_child_withdel(old_node
, old_child
) {
200 if (streq(old_child
->name
, new_child
->name
)) {
201 merge_nodes(old_child
, new_child
);
207 /* if no collision occurred, add child to the old node. */
209 add_child(old_node
, new_child
);
212 /* The new node contents are now merged into the old node. Free
219 struct node
* add_orphan_node(struct node
*dt
, struct node
*new_node
, char *ref
)
221 static unsigned int next_orphan_fragment
= 0;
224 struct data d
= empty_data
;
227 d
= data_add_marker(d
, REF_PHANDLE
, ref
);
228 d
= data_append_integer(d
, 0xffffffff, 32);
230 p
= build_property("target", d
);
232 xasprintf(&name
, "fragment@%u",
233 next_orphan_fragment
++);
234 name_node(new_node
, "__overlay__");
235 node
= build_node(p
, new_node
);
236 name_node(node
, name
);
242 struct node
*chain_node(struct node
*first
, struct node
*list
)
244 assert(first
->next_sibling
== NULL
);
246 first
->next_sibling
= list
;
250 void add_property(struct node
*node
, struct property
*prop
)
263 void delete_property_by_name(struct node
*node
, char *name
)
265 struct property
*prop
= node
->proplist
;
268 if (streq(prop
->name
, name
)) {
269 delete_property(prop
);
276 void delete_property(struct property
*prop
)
279 delete_labels(&prop
->labels
);
282 void add_child(struct node
*parent
, struct node
*child
)
286 child
->next_sibling
= NULL
;
287 child
->parent
= parent
;
289 p
= &parent
->children
;
291 p
= &((*p
)->next_sibling
);
296 void delete_node_by_name(struct node
*parent
, char *name
)
298 struct node
*node
= parent
->children
;
301 if (streq(node
->name
, name
)) {
305 node
= node
->next_sibling
;
309 void delete_node(struct node
*node
)
311 struct property
*prop
;
315 for_each_child(node
, child
)
317 for_each_property(node
, prop
)
318 delete_property(prop
);
319 delete_labels(&node
->labels
);
322 void append_to_property(struct node
*node
,
323 char *name
, const void *data
, int len
)
328 p
= get_property(node
, name
);
330 d
= data_append_data(p
->val
, data
, len
);
333 d
= data_append_data(empty_data
, data
, len
);
334 p
= build_property(name
, d
);
335 add_property(node
, p
);
339 struct reserve_info
*build_reserve_entry(uint64_t address
, uint64_t size
)
341 struct reserve_info
*new = xmalloc(sizeof(*new));
343 memset(new, 0, sizeof(*new));
345 new->address
= address
;
351 struct reserve_info
*chain_reserve_entry(struct reserve_info
*first
,
352 struct reserve_info
*list
)
354 assert(first
->next
== NULL
);
360 struct reserve_info
*add_reserve_entry(struct reserve_info
*list
,
361 struct reserve_info
*new)
363 struct reserve_info
*last
;
370 for (last
= list
; last
->next
; last
= last
->next
)
378 struct dt_info
*build_dt_info(unsigned int dtsflags
,
379 struct reserve_info
*reservelist
,
380 struct node
*tree
, uint32_t boot_cpuid_phys
)
384 dti
= xmalloc(sizeof(*dti
));
385 dti
->dtsflags
= dtsflags
;
386 dti
->reservelist
= reservelist
;
388 dti
->boot_cpuid_phys
= boot_cpuid_phys
;
394 * Tree accessor functions
397 const char *get_unitname(struct node
*node
)
399 if (node
->name
[node
->basenamelen
] == '\0')
402 return node
->name
+ node
->basenamelen
+ 1;
405 struct property
*get_property(struct node
*node
, const char *propname
)
407 struct property
*prop
;
409 for_each_property(node
, prop
)
410 if (streq(prop
->name
, propname
))
416 cell_t
propval_cell(struct property
*prop
)
418 assert(prop
->val
.len
== sizeof(cell_t
));
419 return fdt32_to_cpu(*((fdt32_t
*)prop
->val
.val
));
422 cell_t
propval_cell_n(struct property
*prop
, int n
)
424 assert(prop
->val
.len
/ sizeof(cell_t
) >= n
);
425 return fdt32_to_cpu(*((fdt32_t
*)prop
->val
.val
+ n
));
428 struct property
*get_property_by_label(struct node
*tree
, const char *label
,
431 struct property
*prop
;
436 for_each_property(tree
, prop
) {
439 for_each_label(prop
->labels
, l
)
440 if (streq(l
->label
, label
))
444 for_each_child(tree
, c
) {
445 prop
= get_property_by_label(c
, label
, node
);
454 struct marker
*get_marker_label(struct node
*tree
, const char *label
,
455 struct node
**node
, struct property
**prop
)
463 for_each_property(tree
, p
) {
466 for_each_marker_of_type(m
, LABEL
)
467 if (streq(m
->ref
, label
))
471 for_each_child(tree
, c
) {
472 m
= get_marker_label(c
, label
, node
, prop
);
482 struct node
*get_subnode(struct node
*node
, const char *nodename
)
486 for_each_child(node
, child
)
487 if (streq(child
->name
, nodename
))
493 struct node
*get_node_by_path(struct node
*tree
, const char *path
)
498 if (!path
|| ! (*path
)) {
504 while (path
[0] == '/')
507 p
= strchr(path
, '/');
509 for_each_child(tree
, child
) {
510 if (p
&& (strlen(child
->name
) == p
-path
) &&
511 strprefixeq(path
, p
- path
, child
->name
))
512 return get_node_by_path(child
, p
+1);
513 else if (!p
&& streq(path
, child
->name
))
520 struct node
*get_node_by_label(struct node
*tree
, const char *label
)
522 struct node
*child
, *node
;
525 assert(label
&& (strlen(label
) > 0));
527 for_each_label(tree
->labels
, l
)
528 if (streq(l
->label
, label
))
531 for_each_child(tree
, child
) {
532 node
= get_node_by_label(child
, label
);
540 struct node
*get_node_by_phandle(struct node
*tree
, cell_t phandle
)
542 struct node
*child
, *node
;
544 if ((phandle
== 0) || (phandle
== -1)) {
545 assert(generate_fixups
);
549 if (tree
->phandle
== phandle
) {
555 for_each_child(tree
, child
) {
556 node
= get_node_by_phandle(child
, phandle
);
564 struct node
*get_node_by_ref(struct node
*tree
, const char *ref
)
568 else if (ref
[0] == '/')
569 return get_node_by_path(tree
, ref
);
571 return get_node_by_label(tree
, ref
);
574 cell_t
get_node_phandle(struct node
*root
, struct node
*node
)
576 static cell_t phandle
= 1; /* FIXME: ick, static local */
578 if ((node
->phandle
!= 0) && (node
->phandle
!= -1))
579 return node
->phandle
;
581 while (get_node_by_phandle(root
, phandle
))
584 node
->phandle
= phandle
;
586 if (!get_property(node
, "linux,phandle")
587 && (phandle_format
& PHANDLE_LEGACY
))
589 build_property("linux,phandle",
590 data_append_cell(empty_data
, phandle
)));
592 if (!get_property(node
, "phandle")
593 && (phandle_format
& PHANDLE_EPAPR
))
595 build_property("phandle",
596 data_append_cell(empty_data
, phandle
)));
598 /* If the node *does* have a phandle property, we must
599 * be dealing with a self-referencing phandle, which will be
600 * fixed up momentarily in the caller */
602 return node
->phandle
;
605 uint32_t guess_boot_cpuid(struct node
*tree
)
607 struct node
*cpus
, *bootcpu
;
608 struct property
*reg
;
610 cpus
= get_node_by_path(tree
, "/cpus");
615 bootcpu
= cpus
->children
;
619 reg
= get_property(bootcpu
, "reg");
620 if (!reg
|| (reg
->val
.len
!= sizeof(uint32_t)))
623 /* FIXME: Sanity check node? */
625 return propval_cell(reg
);
628 static int cmp_reserve_info(const void *ax
, const void *bx
)
630 const struct reserve_info
*a
, *b
;
632 a
= *((const struct reserve_info
* const *)ax
);
633 b
= *((const struct reserve_info
* const *)bx
);
635 if (a
->address
< b
->address
)
637 else if (a
->address
> b
->address
)
639 else if (a
->size
< b
->size
)
641 else if (a
->size
> b
->size
)
647 static void sort_reserve_entries(struct dt_info
*dti
)
649 struct reserve_info
*ri
, **tbl
;
652 for (ri
= dti
->reservelist
;
660 tbl
= xmalloc(n
* sizeof(*tbl
));
662 for (ri
= dti
->reservelist
;
667 qsort(tbl
, n
, sizeof(*tbl
), cmp_reserve_info
);
669 dti
->reservelist
= tbl
[0];
670 for (i
= 0; i
< (n
-1); i
++)
671 tbl
[i
]->next
= tbl
[i
+1];
672 tbl
[n
-1]->next
= NULL
;
677 static int cmp_prop(const void *ax
, const void *bx
)
679 const struct property
*a
, *b
;
681 a
= *((const struct property
* const *)ax
);
682 b
= *((const struct property
* const *)bx
);
684 return strcmp(a
->name
, b
->name
);
687 static void sort_properties(struct node
*node
)
690 struct property
*prop
, **tbl
;
692 for_each_property_withdel(node
, prop
)
698 tbl
= xmalloc(n
* sizeof(*tbl
));
700 for_each_property_withdel(node
, prop
)
703 qsort(tbl
, n
, sizeof(*tbl
), cmp_prop
);
705 node
->proplist
= tbl
[0];
706 for (i
= 0; i
< (n
-1); i
++)
707 tbl
[i
]->next
= tbl
[i
+1];
708 tbl
[n
-1]->next
= NULL
;
713 static int cmp_subnode(const void *ax
, const void *bx
)
715 const struct node
*a
, *b
;
717 a
= *((const struct node
* const *)ax
);
718 b
= *((const struct node
* const *)bx
);
720 return strcmp(a
->name
, b
->name
);
723 static void sort_subnodes(struct node
*node
)
726 struct node
*subnode
, **tbl
;
728 for_each_child_withdel(node
, subnode
)
734 tbl
= xmalloc(n
* sizeof(*tbl
));
736 for_each_child_withdel(node
, subnode
)
739 qsort(tbl
, n
, sizeof(*tbl
), cmp_subnode
);
741 node
->children
= tbl
[0];
742 for (i
= 0; i
< (n
-1); i
++)
743 tbl
[i
]->next_sibling
= tbl
[i
+1];
744 tbl
[n
-1]->next_sibling
= NULL
;
749 static void sort_node(struct node
*node
)
753 sort_properties(node
);
755 for_each_child_withdel(node
, c
)
759 void sort_tree(struct dt_info
*dti
)
761 sort_reserve_entries(dti
);
765 /* utility helper to avoid code duplication */
766 static struct node
*build_and_name_child_node(struct node
*parent
, char *name
)
770 node
= build_node(NULL
, NULL
);
771 name_node(node
, xstrdup(name
));
772 add_child(parent
, node
);
777 static struct node
*build_root_node(struct node
*dt
, char *name
)
781 an
= get_subnode(dt
, name
);
783 an
= build_and_name_child_node(dt
, name
);
786 die("Could not build root node /%s\n", name
);
791 static bool any_label_tree(struct dt_info
*dti
, struct node
*node
)
798 for_each_child(node
, c
)
799 if (any_label_tree(dti
, c
))
805 static void generate_label_tree_internal(struct dt_info
*dti
,
806 struct node
*an
, struct node
*node
,
809 struct node
*dt
= dti
->dt
;
814 /* if there are labels */
817 /* now add the label in the node */
818 for_each_label(node
->labels
, l
) {
820 /* check whether the label already exists */
821 p
= get_property(an
, l
->label
);
823 fprintf(stderr
, "WARNING: label %s already"
824 " exists in /%s", l
->label
,
830 p
= build_property(l
->label
,
831 data_copy_mem(node
->fullpath
,
832 strlen(node
->fullpath
) + 1));
836 /* force allocation of a phandle for this node */
838 (void)get_node_phandle(dt
, node
);
841 for_each_child(node
, c
)
842 generate_label_tree_internal(dti
, an
, c
, allocph
);
845 static bool any_fixup_tree(struct dt_info
*dti
, struct node
*node
)
848 struct property
*prop
;
851 for_each_property(node
, prop
) {
852 m
= prop
->val
.markers
;
853 for_each_marker_of_type(m
, REF_PHANDLE
) {
854 if (!get_node_by_ref(dti
->dt
, m
->ref
))
859 for_each_child(node
, c
) {
860 if (any_fixup_tree(dti
, c
))
867 static void add_fixup_entry(struct dt_info
*dti
, struct node
*fn
,
868 struct node
*node
, struct property
*prop
,
873 /* m->ref can only be a REF_PHANDLE, but check anyway */
874 assert(m
->type
== REF_PHANDLE
);
876 /* there shouldn't be any ':' in the arguments */
877 if (strchr(node
->fullpath
, ':') || strchr(prop
->name
, ':'))
878 die("arguments should not contain ':'\n");
880 xasprintf(&entry
, "%s:%s:%u",
881 node
->fullpath
, prop
->name
, m
->offset
);
882 append_to_property(fn
, m
->ref
, entry
, strlen(entry
) + 1);
887 static void generate_fixups_tree_internal(struct dt_info
*dti
,
891 struct node
*dt
= dti
->dt
;
893 struct property
*prop
;
895 struct node
*refnode
;
897 for_each_property(node
, prop
) {
898 m
= prop
->val
.markers
;
899 for_each_marker_of_type(m
, REF_PHANDLE
) {
900 refnode
= get_node_by_ref(dt
, m
->ref
);
902 add_fixup_entry(dti
, fn
, node
, prop
, m
);
906 for_each_child(node
, c
)
907 generate_fixups_tree_internal(dti
, fn
, c
);
910 static bool any_local_fixup_tree(struct dt_info
*dti
, struct node
*node
)
913 struct property
*prop
;
916 for_each_property(node
, prop
) {
917 m
= prop
->val
.markers
;
918 for_each_marker_of_type(m
, REF_PHANDLE
) {
919 if (get_node_by_ref(dti
->dt
, m
->ref
))
924 for_each_child(node
, c
) {
925 if (any_local_fixup_tree(dti
, c
))
932 static void add_local_fixup_entry(struct dt_info
*dti
,
933 struct node
*lfn
, struct node
*node
,
934 struct property
*prop
, struct marker
*m
,
935 struct node
*refnode
)
937 struct node
*wn
, *nwn
; /* local fixup node, walk node, new */
942 /* walk back retreiving depth */
944 for (wn
= node
; wn
; wn
= wn
->parent
)
947 /* allocate name array */
948 compp
= xmalloc(sizeof(*compp
) * depth
);
950 /* store names in the array */
951 for (wn
= node
, i
= depth
- 1; wn
; wn
= wn
->parent
, i
--)
954 /* walk the path components creating nodes if they don't exist */
955 for (wn
= lfn
, i
= 1; i
< depth
; i
++, wn
= nwn
) {
956 /* if no node exists, create it */
957 nwn
= get_subnode(wn
, compp
[i
]);
959 nwn
= build_and_name_child_node(wn
, compp
[i
]);
964 value_32
= cpu_to_fdt32(m
->offset
);
965 append_to_property(wn
, prop
->name
, &value_32
, sizeof(value_32
));
968 static void generate_local_fixups_tree_internal(struct dt_info
*dti
,
972 struct node
*dt
= dti
->dt
;
974 struct property
*prop
;
976 struct node
*refnode
;
978 for_each_property(node
, prop
) {
979 m
= prop
->val
.markers
;
980 for_each_marker_of_type(m
, REF_PHANDLE
) {
981 refnode
= get_node_by_ref(dt
, m
->ref
);
983 add_local_fixup_entry(dti
, lfn
, node
, prop
, m
, refnode
);
987 for_each_child(node
, c
)
988 generate_local_fixups_tree_internal(dti
, lfn
, c
);
991 void generate_label_tree(struct dt_info
*dti
, char *name
, bool allocph
)
993 if (!any_label_tree(dti
, dti
->dt
))
995 generate_label_tree_internal(dti
, build_root_node(dti
->dt
, name
),
999 void generate_fixups_tree(struct dt_info
*dti
, char *name
)
1001 if (!any_fixup_tree(dti
, dti
->dt
))
1003 generate_fixups_tree_internal(dti
, build_root_node(dti
->dt
, name
),
1007 void generate_local_fixups_tree(struct dt_info
*dti
, char *name
)
1009 if (!any_local_fixup_tree(dti
, dti
->dt
))
1011 generate_local_fixups_tree_internal(dti
, build_root_node(dti
->dt
, name
),