40b6d274558a1ae5edc0f18fc4eb8e1421ba06d4
[oweals/u-boot.git] / lib / libfdt / fdt_overlay.c
1 #include "libfdt_env.h"
2
3 #include <fdt.h>
4 #include <libfdt.h>
5
6 #include "libfdt_internal.h"
7
8 /**
9  * overlay_get_target_phandle - retrieves the target phandle of a fragment
10  * @fdto: pointer to the device tree overlay blob
11  * @fragment: node offset of the fragment in the overlay
12  *
13  * overlay_get_target_phandle() retrieves the target phandle of an
14  * overlay fragment when that fragment uses a phandle (target
15  * property) instead of a path (target-path property).
16  *
17  * returns:
18  *      the phandle pointed by the target property
19  *      0, if the phandle was not found
20  *      -1, if the phandle was malformed
21  */
22 static uint32_t overlay_get_target_phandle(const void *fdto, int fragment)
23 {
24         const uint32_t *val;
25         int len;
26
27         val = fdt_getprop(fdto, fragment, "target", &len);
28         if (!val)
29                 return 0;
30
31         if ((*val == (uint32_t)-1) || (len != sizeof(*val)))
32                 return (uint32_t)-1;
33
34         return fdt32_to_cpu(*val);
35 }
36
37 /**
38  * overlay_get_target - retrieves the target phandle of a fragment
39  * @fdt: Base device tree blob
40  * @fdto: Device tree overlay blob
41  * @fragment: node offset of the fragment in the overlay
42  *
43  * overlay_get_target() retrieves the target phandle in the base
44  * device tree of a fragment, no matter how the actual targetting is
45  * done (through a phandle or a path)
46  *
47  * returns:
48  *      the targetted node offset in the base device tree
49  *      Negative error code on error
50  */
51 static int overlay_get_target(const void *fdt, const void *fdto,
52                               int fragment)
53 {
54         uint32_t phandle;
55         const char *path;
56
57         /* Try first to do a phandle based lookup */
58         phandle = overlay_get_target_phandle(fdto, fragment);
59         if (phandle == (uint32_t)-1)
60                 return -FDT_ERR_BADPHANDLE;
61
62         if (phandle)
63                 return fdt_node_offset_by_phandle(fdt, phandle);
64
65         /* And then a path based lookup */
66         path = fdt_getprop(fdto, fragment, "target-path", NULL);
67         if (!path)
68                 return -FDT_ERR_NOTFOUND;
69
70         return fdt_path_offset(fdt, path);
71 }
72
73 /**
74  * overlay_phandle_add_offset - Increases a phandle by an offset
75  * @fdt: Base device tree blob
76  * @node: Device tree overlay blob
77  * @name: Name of the property to modify (phandle or linux,phandle)
78  * @delta: offset to apply
79  *
80  * overlay_phandle_add_offset() increments a node phandle by a given
81  * offset.
82  *
83  * returns:
84  *      0 on success.
85  *      Negative error code on error
86  */
87 static int overlay_phandle_add_offset(void *fdt, int node,
88                                       const char *name, uint32_t delta)
89 {
90         const uint32_t *val;
91         uint32_t adj_val;
92         int len;
93
94         val = fdt_getprop(fdt, node, name, &len);
95         if (!val)
96                 return len;
97
98         if (len != sizeof(*val))
99                 return -FDT_ERR_BADSTRUCTURE;
100
101         adj_val = fdt32_to_cpu(*val);
102         if ((adj_val + delta) < adj_val)
103                 return -FDT_ERR_BADPHANDLE;
104
105         adj_val += delta;
106         return fdt_setprop_inplace_u32(fdt, node, name, adj_val);
107 }
108
109 /**
110  * overlay_adjust_node_phandles - Offsets the phandles of a node
111  * @fdto: Device tree overlay blob
112  * @node: Offset of the node we want to adjust
113  * @delta: Offset to shift the phandles of
114  *
115  * overlay_adjust_node_phandles() adds a constant to all the phandles
116  * of a given node. This is mainly use as part of the overlay
117  * application process, when we want to update all the overlay
118  * phandles to not conflict with the overlays of the base device tree.
119  *
120  * returns:
121  *      0 on success
122  *      Negative error code on failure
123  */
124 static int overlay_adjust_node_phandles(void *fdto, int node,
125                                         uint32_t delta)
126 {
127         bool found = false;
128         int child;
129         int ret;
130
131         ret = overlay_phandle_add_offset(fdto, node, "phandle", delta);
132         if (ret && ret != -FDT_ERR_NOTFOUND)
133                 return ret;
134
135         if (!ret)
136                 found = true;
137
138         ret = overlay_phandle_add_offset(fdto, node, "linux,phandle", delta);
139         if (ret && ret != -FDT_ERR_NOTFOUND)
140                 return ret;
141
142         /*
143          * If neither phandle nor linux,phandle have been found return
144          * an error.
145          */
146         if (!found && !ret)
147                 return ret;
148
149         fdt_for_each_subnode(fdto, child, node)
150                 overlay_adjust_node_phandles(fdto, child, delta);
151
152         return 0;
153 }
154
155 /**
156  * overlay_adjust_local_phandles - Adjust the phandles of a whole overlay
157  * @fdto: Device tree overlay blob
158  * @delta: Offset to shift the phandles of
159  *
160  * overlay_adjust_local_phandles() adds a constant to all the
161  * phandles of an overlay. This is mainly use as part of the overlay
162  * application process, when we want to update all the overlay
163  * phandles to not conflict with the overlays of the base device tree.
164  *
165  * returns:
166  *      0 on success
167  *      Negative error code on failure
168  */
169 static int overlay_adjust_local_phandles(void *fdto, uint32_t delta)
170 {
171         /*
172          * Start adjusting the phandles from the overlay root
173          */
174         return overlay_adjust_node_phandles(fdto, 0, delta);
175 }
176
177 /**
178  * overlay_update_local_node_references - Adjust the overlay references
179  * @fdto: Device tree overlay blob
180  * @tree_node: Node offset of the node to operate on
181  * @fixup_node: Node offset of the matching local fixups node
182  * @delta: Offset to shift the phandles of
183  *
184  * overlay_update_local_nodes_references() update the phandles
185  * pointing to a node within the device tree overlay by adding a
186  * constant delta.
187  *
188  * This is mainly used as part of a device tree application process,
189  * where you want the device tree overlays phandles to not conflict
190  * with the ones from the base device tree before merging them.
191  *
192  * returns:
193  *      0 on success
194  *      Negative error code on failure
195  */
196 static int overlay_update_local_node_references(void *fdto,
197                                                 int tree_node,
198                                                 int fixup_node,
199                                                 uint32_t delta)
200 {
201         int fixup_prop;
202         int fixup_child;
203         int ret;
204
205         fdt_for_each_property_offset(fixup_prop, fdto, fixup_node) {
206                 const unsigned char *fixup_val, *tree_val;
207                 const char *name;
208                 int fixup_len;
209                 int tree_len;
210                 int i;
211
212                 fixup_val = fdt_getprop_by_offset(fdto, fixup_prop,
213                                                   &name, &fixup_len);
214                 if (!fixup_val)
215                         return fixup_len;
216
217                 tree_val = fdt_getprop(fdto, tree_node, name, &tree_len);
218                 if (!tree_val)
219                         return tree_len;
220
221                 for (i = 0; i < fixup_len; i += sizeof(uint32_t)) {
222                         uint32_t adj_val, index;
223
224                         index = *(uint32_t *)(fixup_val + i);
225                         index = fdt32_to_cpu(index);
226
227                         /*
228                          * phandles to fixup can be unaligned.
229                          *
230                          * Use a memcpy for the architectures that do
231                          * not support unaligned accesses.
232                          */
233                         memcpy(&adj_val, tree_val + index, sizeof(uint32_t));
234
235                         adj_val = fdt32_to_cpu(adj_val);
236                         adj_val += delta;
237                         adj_val = cpu_to_fdt32(adj_val);
238
239                         ret = fdt_setprop_inplace_namelen_partial(fdto,
240                                                                   tree_node,
241                                                                   name,
242                                                                   strlen(name),
243                                                                   index,
244                                                                   &adj_val,
245                                                                   sizeof(adj_val));
246                         if (ret)
247                                 return ret;
248                 }
249         }
250
251         fdt_for_each_subnode(fdto, fixup_child, fixup_node) {
252                 const char *fixup_child_name = fdt_get_name(fdto, fixup_child,
253                                                             NULL);
254                 int tree_child;
255
256                 tree_child = fdt_subnode_offset(fdto, tree_node,
257                                                 fixup_child_name);
258                 if (tree_child < 0)
259                         return tree_child;
260
261                 ret = overlay_update_local_node_references(fdto,
262                                                            tree_child,
263                                                            fixup_child,
264                                                            delta);
265                 if (ret)
266                         return ret;
267         }
268
269         return 0;
270 }
271
272 /**
273  * overlay_update_local_references - Adjust the overlay references
274  * @fdto: Device tree overlay blob
275  * @delta: Offset to shift the phandles of
276  *
277  * overlay_update_local_references() update all the phandles pointing
278  * to a node within the device tree overlay by adding a constant
279  * delta to not conflict with the base overlay.
280  *
281  * This is mainly used as part of a device tree application process,
282  * where you want the device tree overlays phandles to not conflict
283  * with the ones from the base device tree before merging them.
284  *
285  * returns:
286  *      0 on success
287  *      Negative error code on failure
288  */
289 static int overlay_update_local_references(void *fdto, uint32_t delta)
290 {
291         int fixups;
292
293         fixups = fdt_path_offset(fdto, "/__local_fixups__");
294         if (fixups < 0) {
295                 /* There's no local phandles to adjust, bail out */
296                 if (fixups == -FDT_ERR_NOTFOUND)
297                         return 0;
298
299                 return fixups;
300         }
301
302         /*
303          * Update our local references from the root of the tree
304          */
305         return overlay_update_local_node_references(fdto, 0, fixups,
306                                                     delta);
307 }
308
309 /**
310  * overlay_fixup_one_phandle - Set an overlay phandle to the base one
311  * @fdt: Base Device Tree blob
312  * @fdto: Device tree overlay blob
313  * @symbols_off: Node offset of the symbols node in the base device tree
314  * @path: Path to a node holding a phandle in the overlay
315  * @path_len: number of path characters to consider
316  * @name: Name of the property holding the phandle reference in the overlay
317  * @name_len: number of name characters to consider
318  * @index: Index in the overlay property where the phandle is stored
319  * @label: Label of the node referenced by the phandle
320  *
321  * overlay_fixup_one_phandle() resolves an overlay phandle pointing to
322  * a node in the base device tree.
323  *
324  * This is part of the device tree overlay application process, when
325  * you want all the phandles in the overlay to point to the actual
326  * base dt nodes.
327  *
328  * returns:
329  *      0 on success
330  *      Negative error code on failure
331  */
332 static int overlay_fixup_one_phandle(void *fdt, void *fdto,
333                                      int symbols_off,
334                                      const char *path, uint32_t path_len,
335                                      const char *name, uint32_t name_len,
336                                      int index, const char *label)
337 {
338         const char *symbol_path;
339         uint32_t phandle;
340         int symbol_off, fixup_off;
341         int prop_len;
342
343         symbol_path = fdt_getprop(fdt, symbols_off, label,
344                                   &prop_len);
345         if (!symbol_path)
346                 return -FDT_ERR_NOTFOUND;
347
348         symbol_off = fdt_path_offset(fdt, symbol_path);
349         if (symbol_off < 0)
350                 return symbol_off;
351
352         phandle = fdt_get_phandle(fdt, symbol_off);
353         if (!phandle)
354                 return -FDT_ERR_NOTFOUND;
355
356         fixup_off = fdt_path_offset_namelen(fdto, path, path_len);
357         if (fixup_off < 0)
358                 return fixup_off;
359
360         phandle = cpu_to_fdt32(phandle);
361         return fdt_setprop_inplace_namelen_partial(fdto, fixup_off,
362                                                    name, name_len, index,
363                                                    &phandle, sizeof(phandle));
364 };
365
366 /**
367  * overlay_fixup_phandle - Set an overlay phandle to the base one
368  * @fdt: Base Device Tree blob
369  * @fdto: Device tree overlay blob
370  * @symbols_off: Node offset of the symbols node in the base device tree
371  * @property: Property offset in the overlay holding the list of fixups
372  *
373  * overlay_fixup_phandle() resolves all the overlay phandles pointed
374  * to in a __local_fixup__ property, and updates them to match the
375  * phandles in use in the base device tree.
376  *
377  * This is part of the device tree overlay application process, when
378  * you want all the phandles in the overlay to point to the actual
379  * base dt nodes.
380  *
381  * returns:
382  *      0 on success
383  *      Negative error code on failure
384  */
385 static int overlay_fixup_phandle(void *fdt, void *fdto, int symbols_off,
386                                  int property)
387 {
388         const char *value;
389         const char *label;
390         int len;
391
392         value = fdt_getprop_by_offset(fdto, property,
393                                       &label, &len);
394         if (!value)
395                 return len;
396
397         do {
398                 const char *prop_string = value;
399                 const char *path, *name;
400                 uint32_t prop_len = strlen(value);
401                 uint32_t path_len, name_len;
402                 char *sep, *endptr;
403                 int index;
404                 int ret;
405
406                 path = prop_string;
407                 sep = memchr(prop_string, ':', prop_len);
408                 if (*sep != ':')
409                         return -FDT_ERR_BADSTRUCTURE;
410                 path_len = sep - path;
411
412                 name = sep + 1;
413                 sep = memchr(name, ':', prop_len);
414                 if (*sep != ':')
415                         return -FDT_ERR_BADSTRUCTURE;
416                 name_len = sep - name;
417
418                 index = strtoul(sep + 1, &endptr, 10);
419                 if ((*endptr != '\0') || (endptr <= (sep + 1)))
420                         return -FDT_ERR_BADSTRUCTURE;
421
422                 len -= prop_len + 1;
423                 value += prop_len + 1;
424
425                 ret = overlay_fixup_one_phandle(fdt, fdto, symbols_off,
426                                                 path, path_len, name, name_len,
427                                                 index, label);
428                 if (ret)
429                         return ret;
430         } while (len > 0);
431
432         return 0;
433 }
434
435 /**
436  * overlay_fixup_phandles - Resolve the overlay phandles to the base
437  *                          device tree
438  * @fdt: Base Device Tree blob
439  * @fdto: Device tree overlay blob
440  *
441  * overlay_fixup_phandles() resolves all the overlay phandles pointing
442  * to nodes in the base device tree.
443  *
444  * This is one of the steps of the device tree overlay application
445  * process, when you want all the phandles in the overlay to point to
446  * the actual base dt nodes.
447  *
448  * returns:
449  *      0 on success
450  *      Negative error code on failure
451  */
452 static int overlay_fixup_phandles(void *fdt, void *fdto)
453 {
454         int fixups_off, symbols_off;
455         int property;
456
457         symbols_off = fdt_path_offset(fdt, "/__symbols__");
458         fixups_off = fdt_path_offset(fdto, "/__fixups__");
459
460         fdt_for_each_property_offset(property, fdto, fixups_off) {
461                 int ret;
462
463                 ret = overlay_fixup_phandle(fdt, fdto, symbols_off, property);
464                 if (ret)
465                         return ret;
466         }
467
468         return 0;
469 }
470
471 /**
472  * overlay_apply_node - Merge an overlay fragment into the base device tree
473  * @fdt: Base Device Tree blob
474  * @target: Node offset in the base device tree to apply the fragment to
475  * @fdto: Device tree overlay blob
476  * @fragment: Node offset in the overlay holding the changes to merge
477  *
478  * overlay_apply_node() merges an overlay fragment into a target base
479  * device tree node pointed.
480  *
481  * This is part of the final step in the device tree overlay
482  * application process, when all the phandles have been adjusted and
483  * resolved and you just have to merge overlay into the base device
484  * tree.
485  *
486  * returns:
487  *      0 on success
488  *      Negative error code on failure
489  */
490 static int overlay_apply_node(void *fdt, int target,
491                               void *fdto, int fragment)
492 {
493         int property;
494         int node;
495
496         fdt_for_each_property_offset(property, fdto, fragment) {
497                 const char *name;
498                 const void *prop;
499                 int prop_len;
500                 int ret;
501
502                 prop = fdt_getprop_by_offset(fdto, property, &name,
503                                              &prop_len);
504                 if (prop_len == -FDT_ERR_NOTFOUND)
505                         return -FDT_ERR_INTERNAL;
506                 if (prop_len < 0)
507                         return prop_len;
508
509                 ret = fdt_setprop(fdt, target, name, prop, prop_len);
510                 if (ret)
511                         return ret;
512         }
513
514         fdt_for_each_subnode(fdto, node, fragment) {
515                 const char *name = fdt_get_name(fdto, node, NULL);
516                 int nnode;
517                 int ret;
518
519                 nnode = fdt_add_subnode(fdt, target, name);
520                 if (nnode == -FDT_ERR_EXISTS)
521                         nnode = fdt_subnode_offset(fdt, target, name);
522
523                 if (nnode < 0)
524                         return nnode;
525
526                 ret = overlay_apply_node(fdt, nnode, fdto, node);
527                 if (ret)
528                         return ret;
529         }
530
531         return 0;
532 }
533
534 /**
535  * overlay_merge - Merge an overlay into its base device tree
536  * @fdt: Base Device Tree blob
537  * @fdto: Device tree overlay blob
538  *
539  * overlay_merge() merges an overlay into its base device tree.
540  *
541  * This is the final step in the device tree overlay application
542  * process, when all the phandles have been adjusted and resolved and
543  * you just have to merge overlay into the base device tree.
544  *
545  * returns:
546  *      0 on success
547  *      Negative error code on failure
548  */
549 static int overlay_merge(void *dt, void *dto)
550 {
551         int fragment;
552
553         fdt_for_each_subnode(dto, fragment, 0) {
554                 int overlay;
555                 int target;
556                 int ret;
557
558                 target = overlay_get_target(dt, dto, fragment);
559                 if (target < 0)
560                         continue;
561
562                 overlay = fdt_subnode_offset(dto, fragment, "__overlay__");
563                 if (overlay < 0)
564                         return overlay;
565
566                 ret = overlay_apply_node(dt, target, dto, overlay);
567                 if (ret)
568                         return ret;
569         }
570
571         return 0;
572 }
573
574 int fdt_overlay_apply(void *fdt, void *fdto)
575 {
576         uint32_t delta = fdt_get_max_phandle(fdt) + 1;
577         int ret;
578
579         FDT_CHECK_HEADER(fdt);
580         FDT_CHECK_HEADER(fdto);
581
582         ret = overlay_adjust_local_phandles(fdto, delta);
583         if (ret)
584                 goto err;
585
586         ret = overlay_update_local_references(fdto, delta);
587         if (ret)
588                 goto err;
589
590         ret = overlay_fixup_phandles(fdt, fdto);
591         if (ret)
592                 goto err;
593
594         ret = overlay_merge(fdt, fdto);
595         if (ret)
596                 goto err;
597
598         /*
599          * The overlay has been damaged, erase its magic.
600          */
601         fdt_set_magic(fdto, ~0);
602
603         return 0;
604
605 err:
606         /*
607          * The overlay might have been damaged, erase its magic.
608          */
609         fdt_set_magic(fdto, ~0);
610
611         /*
612          * The base device tree might have been damaged, erase its
613          * magic.
614          */
615         fdt_set_magic(fdt, ~0);
616
617         return ret;
618 }