Loading...
1/*
2 * Functions for working with device tree overlays
3 *
4 * Copyright (C) 2012 Pantelis Antoniou <panto@antoniou-consulting.com>
5 * Copyright (C) 2012 Texas Instruments Inc.
6 *
7 * This program is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU General Public License
9 * version 2 as published by the Free Software Foundation.
10 */
11#undef DEBUG
12#include <linux/kernel.h>
13#include <linux/module.h>
14#include <linux/of.h>
15#include <linux/of_device.h>
16#include <linux/string.h>
17#include <linux/ctype.h>
18#include <linux/errno.h>
19#include <linux/string.h>
20#include <linux/slab.h>
21#include <linux/err.h>
22#include <linux/idr.h>
23
24#include "of_private.h"
25
26/**
27 * struct of_overlay_info - Holds a single overlay info
28 * @target: target of the overlay operation
29 * @overlay: pointer to the overlay contents node
30 *
31 * Holds a single overlay state, including all the overlay logs &
32 * records.
33 */
34struct of_overlay_info {
35 struct device_node *target;
36 struct device_node *overlay;
37};
38
39/**
40 * struct of_overlay - Holds a complete overlay transaction
41 * @node: List on which we are located
42 * @count: Count of ovinfo structures
43 * @ovinfo_tab: Overlay info table (count sized)
44 * @cset: Changeset to be used
45 *
46 * Holds a complete overlay transaction
47 */
48struct of_overlay {
49 int id;
50 struct list_head node;
51 int count;
52 struct of_overlay_info *ovinfo_tab;
53 struct of_changeset cset;
54};
55
56static int of_overlay_apply_one(struct of_overlay *ov,
57 struct device_node *target, const struct device_node *overlay);
58
59static int of_overlay_apply_single_property(struct of_overlay *ov,
60 struct device_node *target, struct property *prop)
61{
62 struct property *propn, *tprop;
63
64 /* NOTE: Multiple changes of single properties not supported */
65 tprop = of_find_property(target, prop->name, NULL);
66
67 /* special properties are not meant to be updated (silent NOP) */
68 if (of_prop_cmp(prop->name, "name") == 0 ||
69 of_prop_cmp(prop->name, "phandle") == 0 ||
70 of_prop_cmp(prop->name, "linux,phandle") == 0)
71 return 0;
72
73 propn = __of_prop_dup(prop, GFP_KERNEL);
74 if (propn == NULL)
75 return -ENOMEM;
76
77 /* not found? add */
78 if (tprop == NULL)
79 return of_changeset_add_property(&ov->cset, target, propn);
80
81 /* found? update */
82 return of_changeset_update_property(&ov->cset, target, propn);
83}
84
85static int of_overlay_apply_single_device_node(struct of_overlay *ov,
86 struct device_node *target, struct device_node *child)
87{
88 const char *cname;
89 struct device_node *tchild;
90 int ret = 0;
91
92 cname = kbasename(child->full_name);
93 if (cname == NULL)
94 return -ENOMEM;
95
96 /* NOTE: Multiple mods of created nodes not supported */
97 tchild = of_get_child_by_name(target, cname);
98 if (tchild != NULL) {
99 /* apply overlay recursively */
100 ret = of_overlay_apply_one(ov, tchild, child);
101 of_node_put(tchild);
102 } else {
103 /* create empty tree as a target */
104 tchild = __of_node_dup(child, "%s/%s", target->full_name, cname);
105 if (!tchild)
106 return -ENOMEM;
107
108 /* point to parent */
109 tchild->parent = target;
110
111 ret = of_changeset_attach_node(&ov->cset, tchild);
112 if (ret)
113 return ret;
114
115 ret = of_overlay_apply_one(ov, tchild, child);
116 if (ret)
117 return ret;
118 }
119
120 return ret;
121}
122
123/*
124 * Apply a single overlay node recursively.
125 *
126 * Note that the in case of an error the target node is left
127 * in a inconsistent state. Error recovery should be performed
128 * by using the changeset.
129 */
130static int of_overlay_apply_one(struct of_overlay *ov,
131 struct device_node *target, const struct device_node *overlay)
132{
133 struct device_node *child;
134 struct property *prop;
135 int ret;
136
137 for_each_property_of_node(overlay, prop) {
138 ret = of_overlay_apply_single_property(ov, target, prop);
139 if (ret) {
140 pr_err("%s: Failed to apply prop @%s/%s\n",
141 __func__, target->full_name, prop->name);
142 return ret;
143 }
144 }
145
146 for_each_child_of_node(overlay, child) {
147 ret = of_overlay_apply_single_device_node(ov, target, child);
148 if (ret != 0) {
149 pr_err("%s: Failed to apply single node @%s/%s\n",
150 __func__, target->full_name,
151 child->name);
152 of_node_put(child);
153 return ret;
154 }
155 }
156
157 return 0;
158}
159
160/**
161 * of_overlay_apply() - Apply @count overlays pointed at by @ovinfo_tab
162 * @ov: Overlay to apply
163 *
164 * Applies the overlays given, while handling all error conditions
165 * appropriately. Either the operation succeeds, or if it fails the
166 * live tree is reverted to the state before the attempt.
167 * Returns 0, or an error if the overlay attempt failed.
168 */
169static int of_overlay_apply(struct of_overlay *ov)
170{
171 int i, err;
172
173 /* first we apply the overlays atomically */
174 for (i = 0; i < ov->count; i++) {
175 struct of_overlay_info *ovinfo = &ov->ovinfo_tab[i];
176
177 err = of_overlay_apply_one(ov, ovinfo->target, ovinfo->overlay);
178 if (err != 0) {
179 pr_err("%s: overlay failed '%s'\n",
180 __func__, ovinfo->target->full_name);
181 return err;
182 }
183 }
184
185 return 0;
186}
187
188/*
189 * Find the target node using a number of different strategies
190 * in order of preference
191 *
192 * "target" property containing the phandle of the target
193 * "target-path" property containing the path of the target
194 */
195static struct device_node *find_target_node(struct device_node *info_node)
196{
197 const char *path;
198 u32 val;
199 int ret;
200
201 /* first try to go by using the target as a phandle */
202 ret = of_property_read_u32(info_node, "target", &val);
203 if (ret == 0)
204 return of_find_node_by_phandle(val);
205
206 /* now try to locate by path */
207 ret = of_property_read_string(info_node, "target-path", &path);
208 if (ret == 0)
209 return of_find_node_by_path(path);
210
211 pr_err("%s: Failed to find target for node %p (%s)\n", __func__,
212 info_node, info_node->name);
213
214 return NULL;
215}
216
217/**
218 * of_fill_overlay_info() - Fill an overlay info structure
219 * @ov Overlay to fill
220 * @info_node: Device node containing the overlay
221 * @ovinfo: Pointer to the overlay info structure to fill
222 *
223 * Fills an overlay info structure with the overlay information
224 * from a device node. This device node must have a target property
225 * which contains a phandle of the overlay target node, and an
226 * __overlay__ child node which has the overlay contents.
227 * Both ovinfo->target & ovinfo->overlay have their references taken.
228 *
229 * Returns 0 on success, or a negative error value.
230 */
231static int of_fill_overlay_info(struct of_overlay *ov,
232 struct device_node *info_node, struct of_overlay_info *ovinfo)
233{
234 ovinfo->overlay = of_get_child_by_name(info_node, "__overlay__");
235 if (ovinfo->overlay == NULL)
236 goto err_fail;
237
238 ovinfo->target = find_target_node(info_node);
239 if (ovinfo->target == NULL)
240 goto err_fail;
241
242 return 0;
243
244err_fail:
245 of_node_put(ovinfo->target);
246 of_node_put(ovinfo->overlay);
247
248 memset(ovinfo, 0, sizeof(*ovinfo));
249 return -EINVAL;
250}
251
252/**
253 * of_build_overlay_info() - Build an overlay info array
254 * @ov Overlay to build
255 * @tree: Device node containing all the overlays
256 *
257 * Helper function that given a tree containing overlay information,
258 * allocates and builds an overlay info array containing it, ready
259 * for use using of_overlay_apply.
260 *
261 * Returns 0 on success with the @cntp @ovinfop pointers valid,
262 * while on error a negative error value is returned.
263 */
264static int of_build_overlay_info(struct of_overlay *ov,
265 struct device_node *tree)
266{
267 struct device_node *node;
268 struct of_overlay_info *ovinfo;
269 int cnt, err;
270
271 /* worst case; every child is a node */
272 cnt = 0;
273 for_each_child_of_node(tree, node)
274 cnt++;
275
276 ovinfo = kcalloc(cnt, sizeof(*ovinfo), GFP_KERNEL);
277 if (ovinfo == NULL)
278 return -ENOMEM;
279
280 cnt = 0;
281 for_each_child_of_node(tree, node) {
282 memset(&ovinfo[cnt], 0, sizeof(*ovinfo));
283 err = of_fill_overlay_info(ov, node, &ovinfo[cnt]);
284 if (err == 0)
285 cnt++;
286 }
287
288 /* if nothing filled, return error */
289 if (cnt == 0) {
290 kfree(ovinfo);
291 return -ENODEV;
292 }
293
294 ov->count = cnt;
295 ov->ovinfo_tab = ovinfo;
296
297 return 0;
298}
299
300/**
301 * of_free_overlay_info() - Free an overlay info array
302 * @ov Overlay to free the overlay info from
303 * @ovinfo_tab: Array of overlay_info's to free
304 *
305 * Releases the memory of a previously allocated ovinfo array
306 * by of_build_overlay_info.
307 * Returns 0, or an error if the arguments are bogus.
308 */
309static int of_free_overlay_info(struct of_overlay *ov)
310{
311 struct of_overlay_info *ovinfo;
312 int i;
313
314 /* do it in reverse */
315 for (i = ov->count - 1; i >= 0; i--) {
316 ovinfo = &ov->ovinfo_tab[i];
317
318 of_node_put(ovinfo->target);
319 of_node_put(ovinfo->overlay);
320 }
321 kfree(ov->ovinfo_tab);
322
323 return 0;
324}
325
326static LIST_HEAD(ov_list);
327static DEFINE_IDR(ov_idr);
328
329/**
330 * of_overlay_create() - Create and apply an overlay
331 * @tree: Device node containing all the overlays
332 *
333 * Creates and applies an overlay while also keeping track
334 * of the overlay in a list. This list can be used to prevent
335 * illegal overlay removals.
336 *
337 * Returns the id of the created overlay, or a negative error number
338 */
339int of_overlay_create(struct device_node *tree)
340{
341 struct of_overlay *ov;
342 int err, id;
343
344 /* allocate the overlay structure */
345 ov = kzalloc(sizeof(*ov), GFP_KERNEL);
346 if (ov == NULL)
347 return -ENOMEM;
348 ov->id = -1;
349
350 INIT_LIST_HEAD(&ov->node);
351
352 of_changeset_init(&ov->cset);
353
354 mutex_lock(&of_mutex);
355
356 id = idr_alloc(&ov_idr, ov, 0, 0, GFP_KERNEL);
357 if (id < 0) {
358 pr_err("%s: idr_alloc() failed for tree@%s\n",
359 __func__, tree->full_name);
360 err = id;
361 goto err_destroy_trans;
362 }
363 ov->id = id;
364
365 /* build the overlay info structures */
366 err = of_build_overlay_info(ov, tree);
367 if (err) {
368 pr_err("%s: of_build_overlay_info() failed for tree@%s\n",
369 __func__, tree->full_name);
370 goto err_free_idr;
371 }
372
373 /* apply the overlay */
374 err = of_overlay_apply(ov);
375 if (err) {
376 pr_err("%s: of_overlay_apply() failed for tree@%s\n",
377 __func__, tree->full_name);
378 goto err_abort_trans;
379 }
380
381 /* apply the changeset */
382 err = __of_changeset_apply(&ov->cset);
383 if (err) {
384 pr_err("%s: __of_changeset_apply() failed for tree@%s\n",
385 __func__, tree->full_name);
386 goto err_revert_overlay;
387 }
388
389 /* add to the tail of the overlay list */
390 list_add_tail(&ov->node, &ov_list);
391
392 mutex_unlock(&of_mutex);
393
394 return id;
395
396err_revert_overlay:
397err_abort_trans:
398 of_free_overlay_info(ov);
399err_free_idr:
400 idr_remove(&ov_idr, ov->id);
401err_destroy_trans:
402 of_changeset_destroy(&ov->cset);
403 kfree(ov);
404 mutex_unlock(&of_mutex);
405
406 return err;
407}
408EXPORT_SYMBOL_GPL(of_overlay_create);
409
410/* check whether the given node, lies under the given tree */
411static int overlay_subtree_check(struct device_node *tree,
412 struct device_node *dn)
413{
414 struct device_node *child;
415
416 /* match? */
417 if (tree == dn)
418 return 1;
419
420 for_each_child_of_node(tree, child) {
421 if (overlay_subtree_check(child, dn)) {
422 of_node_put(child);
423 return 1;
424 }
425 }
426
427 return 0;
428}
429
430/* check whether this overlay is the topmost */
431static int overlay_is_topmost(struct of_overlay *ov, struct device_node *dn)
432{
433 struct of_overlay *ovt;
434 struct of_changeset_entry *ce;
435
436 list_for_each_entry_reverse(ovt, &ov_list, node) {
437 /* if we hit ourselves, we're done */
438 if (ovt == ov)
439 break;
440
441 /* check against each subtree affected by this overlay */
442 list_for_each_entry(ce, &ovt->cset.entries, node) {
443 if (overlay_subtree_check(ce->np, dn)) {
444 pr_err("%s: #%d clashes #%d @%s\n",
445 __func__, ov->id, ovt->id,
446 dn->full_name);
447 return 0;
448 }
449 }
450 }
451
452 /* overlay is topmost */
453 return 1;
454}
455
456/*
457 * We can safely remove the overlay only if it's the top-most one.
458 * Newly applied overlays are inserted at the tail of the overlay list,
459 * so a top most overlay is the one that is closest to the tail.
460 *
461 * The topmost check is done by exploiting this property. For each
462 * affected device node in the log list we check if this overlay is
463 * the one closest to the tail. If another overlay has affected this
464 * device node and is closest to the tail, then removal is not permited.
465 */
466static int overlay_removal_is_ok(struct of_overlay *ov)
467{
468 struct of_changeset_entry *ce;
469
470 list_for_each_entry(ce, &ov->cset.entries, node) {
471 if (!overlay_is_topmost(ov, ce->np)) {
472 pr_err("%s: overlay #%d is not topmost\n",
473 __func__, ov->id);
474 return 0;
475 }
476 }
477
478 return 1;
479}
480
481/**
482 * of_overlay_destroy() - Removes an overlay
483 * @id: Overlay id number returned by a previous call to of_overlay_create
484 *
485 * Removes an overlay if it is permissible.
486 *
487 * Returns 0 on success, or a negative error number
488 */
489int of_overlay_destroy(int id)
490{
491 struct of_overlay *ov;
492 int err;
493
494 mutex_lock(&of_mutex);
495
496 ov = idr_find(&ov_idr, id);
497 if (ov == NULL) {
498 err = -ENODEV;
499 pr_err("%s: Could not find overlay #%d\n",
500 __func__, id);
501 goto out;
502 }
503
504 /* check whether the overlay is safe to remove */
505 if (!overlay_removal_is_ok(ov)) {
506 err = -EBUSY;
507 pr_err("%s: removal check failed for overlay #%d\n",
508 __func__, id);
509 goto out;
510 }
511
512
513 list_del(&ov->node);
514 __of_changeset_revert(&ov->cset);
515 of_free_overlay_info(ov);
516 idr_remove(&ov_idr, id);
517 of_changeset_destroy(&ov->cset);
518 kfree(ov);
519
520 err = 0;
521
522out:
523 mutex_unlock(&of_mutex);
524
525 return err;
526}
527EXPORT_SYMBOL_GPL(of_overlay_destroy);
528
529/**
530 * of_overlay_destroy_all() - Removes all overlays from the system
531 *
532 * Removes all overlays from the system in the correct order.
533 *
534 * Returns 0 on success, or a negative error number
535 */
536int of_overlay_destroy_all(void)
537{
538 struct of_overlay *ov, *ovn;
539
540 mutex_lock(&of_mutex);
541
542 /* the tail of list is guaranteed to be safe to remove */
543 list_for_each_entry_safe_reverse(ov, ovn, &ov_list, node) {
544 list_del(&ov->node);
545 __of_changeset_revert(&ov->cset);
546 of_free_overlay_info(ov);
547 idr_remove(&ov_idr, ov->id);
548 kfree(ov);
549 }
550
551 mutex_unlock(&of_mutex);
552
553 return 0;
554}
555EXPORT_SYMBOL_GPL(of_overlay_destroy_all);
1// SPDX-License-Identifier: GPL-2.0
2/*
3 * Functions for working with device tree overlays
4 *
5 * Copyright (C) 2012 Pantelis Antoniou <panto@antoniou-consulting.com>
6 * Copyright (C) 2012 Texas Instruments Inc.
7 */
8
9#define pr_fmt(fmt) "OF: overlay: " fmt
10
11#include <linux/kernel.h>
12#include <linux/module.h>
13#include <linux/of.h>
14#include <linux/of_device.h>
15#include <linux/of_fdt.h>
16#include <linux/string.h>
17#include <linux/ctype.h>
18#include <linux/errno.h>
19#include <linux/slab.h>
20#include <linux/libfdt.h>
21#include <linux/err.h>
22#include <linux/idr.h>
23
24#include "of_private.h"
25
26/**
27 * struct target - info about current target node as recursing through overlay
28 * @np: node where current level of overlay will be applied
29 * @in_livetree: @np is a node in the live devicetree
30 *
31 * Used in the algorithm to create the portion of a changeset that describes
32 * an overlay fragment, which is a devicetree subtree. Initially @np is a node
33 * in the live devicetree where the overlay subtree is targeted to be grafted
34 * into. When recursing to the next level of the overlay subtree, the target
35 * also recurses to the next level of the live devicetree, as long as overlay
36 * subtree node also exists in the live devicetree. When a node in the overlay
37 * subtree does not exist at the same level in the live devicetree, target->np
38 * points to a newly allocated node, and all subsequent targets in the subtree
39 * will be newly allocated nodes.
40 */
41struct target {
42 struct device_node *np;
43 bool in_livetree;
44};
45
46/**
47 * struct fragment - info about fragment nodes in overlay expanded device tree
48 * @target: target of the overlay operation
49 * @overlay: pointer to the __overlay__ node
50 */
51struct fragment {
52 struct device_node *overlay;
53 struct device_node *target;
54};
55
56/**
57 * struct overlay_changeset
58 * @id: changeset identifier
59 * @ovcs_list: list on which we are located
60 * @new_fdt: Memory allocated to hold unflattened aligned FDT
61 * @overlay_mem: the memory chunk that contains @overlay_root
62 * @overlay_root: expanded device tree that contains the fragment nodes
63 * @notify_state: most recent notify action used on overlay
64 * @count: count of fragment structures
65 * @fragments: fragment nodes in the overlay expanded device tree
66 * @symbols_fragment: last element of @fragments[] is the __symbols__ node
67 * @cset: changeset to apply fragments to live device tree
68 */
69struct overlay_changeset {
70 int id;
71 struct list_head ovcs_list;
72 const void *new_fdt;
73 const void *overlay_mem;
74 struct device_node *overlay_root;
75 enum of_overlay_notify_action notify_state;
76 int count;
77 struct fragment *fragments;
78 bool symbols_fragment;
79 struct of_changeset cset;
80};
81
82/* flags are sticky - once set, do not reset */
83static int devicetree_state_flags;
84#define DTSF_APPLY_FAIL 0x01
85#define DTSF_REVERT_FAIL 0x02
86
87/*
88 * If a changeset apply or revert encounters an error, an attempt will
89 * be made to undo partial changes, but may fail. If the undo fails
90 * we do not know the state of the devicetree.
91 */
92static int devicetree_corrupt(void)
93{
94 return devicetree_state_flags &
95 (DTSF_APPLY_FAIL | DTSF_REVERT_FAIL);
96}
97
98static int build_changeset_next_level(struct overlay_changeset *ovcs,
99 struct target *target, const struct device_node *overlay_node);
100
101/*
102 * of_resolve_phandles() finds the largest phandle in the live tree.
103 * of_overlay_apply() may add a larger phandle to the live tree.
104 * Do not allow race between two overlays being applied simultaneously:
105 * mutex_lock(&of_overlay_phandle_mutex)
106 * of_resolve_phandles()
107 * of_overlay_apply()
108 * mutex_unlock(&of_overlay_phandle_mutex)
109 */
110static DEFINE_MUTEX(of_overlay_phandle_mutex);
111
112void of_overlay_mutex_lock(void)
113{
114 mutex_lock(&of_overlay_phandle_mutex);
115}
116
117void of_overlay_mutex_unlock(void)
118{
119 mutex_unlock(&of_overlay_phandle_mutex);
120}
121
122static LIST_HEAD(ovcs_list);
123static DEFINE_IDR(ovcs_idr);
124
125static BLOCKING_NOTIFIER_HEAD(overlay_notify_chain);
126
127/**
128 * of_overlay_notifier_register() - Register notifier for overlay operations
129 * @nb: Notifier block to register
130 *
131 * Register for notification on overlay operations on device tree nodes. The
132 * reported actions definied by @of_reconfig_change. The notifier callback
133 * furthermore receives a pointer to the affected device tree node.
134 *
135 * Note that a notifier callback is not supposed to store pointers to a device
136 * tree node or its content beyond @OF_OVERLAY_POST_REMOVE corresponding to the
137 * respective node it received.
138 */
139int of_overlay_notifier_register(struct notifier_block *nb)
140{
141 return blocking_notifier_chain_register(&overlay_notify_chain, nb);
142}
143EXPORT_SYMBOL_GPL(of_overlay_notifier_register);
144
145/**
146 * of_overlay_notifier_unregister() - Unregister notifier for overlay operations
147 * @nb: Notifier block to unregister
148 */
149int of_overlay_notifier_unregister(struct notifier_block *nb)
150{
151 return blocking_notifier_chain_unregister(&overlay_notify_chain, nb);
152}
153EXPORT_SYMBOL_GPL(of_overlay_notifier_unregister);
154
155static int overlay_notify(struct overlay_changeset *ovcs,
156 enum of_overlay_notify_action action)
157{
158 struct of_overlay_notify_data nd;
159 int i, ret;
160
161 ovcs->notify_state = action;
162
163 for (i = 0; i < ovcs->count; i++) {
164 struct fragment *fragment = &ovcs->fragments[i];
165
166 nd.target = fragment->target;
167 nd.overlay = fragment->overlay;
168
169 ret = blocking_notifier_call_chain(&overlay_notify_chain,
170 action, &nd);
171 if (notifier_to_errno(ret)) {
172 ret = notifier_to_errno(ret);
173 pr_err("overlay changeset %s notifier error %d, target: %pOF\n",
174 of_overlay_action_name(action), ret, nd.target);
175 return ret;
176 }
177 }
178
179 return 0;
180}
181
182/*
183 * The values of properties in the "/__symbols__" node are paths in
184 * the ovcs->overlay_root. When duplicating the properties, the paths
185 * need to be adjusted to be the correct path for the live device tree.
186 *
187 * The paths refer to a node in the subtree of a fragment node's "__overlay__"
188 * node, for example "/fragment@0/__overlay__/symbol_path_tail",
189 * where symbol_path_tail can be a single node or it may be a multi-node path.
190 *
191 * The duplicated property value will be modified by replacing the
192 * "/fragment_name/__overlay/" portion of the value with the target
193 * path from the fragment node.
194 */
195static struct property *dup_and_fixup_symbol_prop(
196 struct overlay_changeset *ovcs, const struct property *prop)
197{
198 struct fragment *fragment;
199 struct property *new_prop;
200 struct device_node *fragment_node;
201 struct device_node *overlay_node;
202 const char *path;
203 const char *path_tail;
204 const char *target_path;
205 int k;
206 int overlay_name_len;
207 int path_len;
208 int path_tail_len;
209 int target_path_len;
210
211 if (!prop->value)
212 return NULL;
213 if (strnlen(prop->value, prop->length) >= prop->length)
214 return NULL;
215 path = prop->value;
216 path_len = strlen(path);
217
218 if (path_len < 1)
219 return NULL;
220 fragment_node = __of_find_node_by_path(ovcs->overlay_root, path + 1);
221 overlay_node = __of_find_node_by_path(fragment_node, "__overlay__/");
222 of_node_put(fragment_node);
223 of_node_put(overlay_node);
224
225 for (k = 0; k < ovcs->count; k++) {
226 fragment = &ovcs->fragments[k];
227 if (fragment->overlay == overlay_node)
228 break;
229 }
230 if (k >= ovcs->count)
231 return NULL;
232
233 overlay_name_len = snprintf(NULL, 0, "%pOF", fragment->overlay);
234
235 if (overlay_name_len > path_len)
236 return NULL;
237 path_tail = path + overlay_name_len;
238 path_tail_len = strlen(path_tail);
239
240 target_path = kasprintf(GFP_KERNEL, "%pOF", fragment->target);
241 if (!target_path)
242 return NULL;
243 target_path_len = strlen(target_path);
244
245 new_prop = kzalloc(sizeof(*new_prop), GFP_KERNEL);
246 if (!new_prop)
247 goto err_free_target_path;
248
249 new_prop->name = kstrdup(prop->name, GFP_KERNEL);
250 new_prop->length = target_path_len + path_tail_len + 1;
251 new_prop->value = kzalloc(new_prop->length, GFP_KERNEL);
252 if (!new_prop->name || !new_prop->value)
253 goto err_free_new_prop;
254
255 strcpy(new_prop->value, target_path);
256 strcpy(new_prop->value + target_path_len, path_tail);
257
258 of_property_set_flag(new_prop, OF_DYNAMIC);
259
260 kfree(target_path);
261
262 return new_prop;
263
264err_free_new_prop:
265 kfree(new_prop->name);
266 kfree(new_prop->value);
267 kfree(new_prop);
268err_free_target_path:
269 kfree(target_path);
270
271 return NULL;
272}
273
274/**
275 * add_changeset_property() - add @overlay_prop to overlay changeset
276 * @ovcs: overlay changeset
277 * @target: where @overlay_prop will be placed
278 * @overlay_prop: property to add or update, from overlay tree
279 * @is_symbols_prop: 1 if @overlay_prop is from node "/__symbols__"
280 *
281 * If @overlay_prop does not already exist in live devicetree, add changeset
282 * entry to add @overlay_prop in @target, else add changeset entry to update
283 * value of @overlay_prop.
284 *
285 * @target may be either in the live devicetree or in a new subtree that
286 * is contained in the changeset.
287 *
288 * Some special properties are not added or updated (no error returned):
289 * "name", "phandle", "linux,phandle".
290 *
291 * Properties "#address-cells" and "#size-cells" are not updated if they
292 * are already in the live tree, but if present in the live tree, the values
293 * in the overlay must match the values in the live tree.
294 *
295 * Update of property in symbols node is not allowed.
296 *
297 * Return: 0 on success, -ENOMEM if memory allocation failure, or -EINVAL if
298 * invalid @overlay.
299 */
300static int add_changeset_property(struct overlay_changeset *ovcs,
301 struct target *target, struct property *overlay_prop,
302 bool is_symbols_prop)
303{
304 struct property *new_prop = NULL, *prop;
305 int ret = 0;
306
307 if (target->in_livetree)
308 if (!of_prop_cmp(overlay_prop->name, "name") ||
309 !of_prop_cmp(overlay_prop->name, "phandle") ||
310 !of_prop_cmp(overlay_prop->name, "linux,phandle"))
311 return 0;
312
313 if (target->in_livetree)
314 prop = of_find_property(target->np, overlay_prop->name, NULL);
315 else
316 prop = NULL;
317
318 if (prop) {
319 if (!of_prop_cmp(prop->name, "#address-cells")) {
320 if (!of_prop_val_eq(prop, overlay_prop)) {
321 pr_err("ERROR: changing value of #address-cells is not allowed in %pOF\n",
322 target->np);
323 ret = -EINVAL;
324 }
325 return ret;
326
327 } else if (!of_prop_cmp(prop->name, "#size-cells")) {
328 if (!of_prop_val_eq(prop, overlay_prop)) {
329 pr_err("ERROR: changing value of #size-cells is not allowed in %pOF\n",
330 target->np);
331 ret = -EINVAL;
332 }
333 return ret;
334 }
335 }
336
337 if (is_symbols_prop) {
338 if (prop)
339 return -EINVAL;
340 new_prop = dup_and_fixup_symbol_prop(ovcs, overlay_prop);
341 } else {
342 new_prop = __of_prop_dup(overlay_prop, GFP_KERNEL);
343 }
344
345 if (!new_prop)
346 return -ENOMEM;
347
348 if (!prop) {
349 if (!target->in_livetree) {
350 new_prop->next = target->np->deadprops;
351 target->np->deadprops = new_prop;
352 }
353 ret = of_changeset_add_property(&ovcs->cset, target->np,
354 new_prop);
355 } else {
356 ret = of_changeset_update_property(&ovcs->cset, target->np,
357 new_prop);
358 }
359
360 if (!of_node_check_flag(target->np, OF_OVERLAY))
361 pr_err("WARNING: memory leak will occur if overlay removed, property: %pOF/%s\n",
362 target->np, new_prop->name);
363
364 if (ret) {
365 kfree(new_prop->name);
366 kfree(new_prop->value);
367 kfree(new_prop);
368 }
369 return ret;
370}
371
372/**
373 * add_changeset_node() - add @node (and children) to overlay changeset
374 * @ovcs: overlay changeset
375 * @target: where @node will be placed in live tree or changeset
376 * @node: node from within overlay device tree fragment
377 *
378 * If @node does not already exist in @target, add changeset entry
379 * to add @node in @target.
380 *
381 * If @node already exists in @target, and the existing node has
382 * a phandle, the overlay node is not allowed to have a phandle.
383 *
384 * If @node has child nodes, add the children recursively via
385 * build_changeset_next_level().
386 *
387 * NOTE_1: A live devicetree created from a flattened device tree (FDT) will
388 * not contain the full path in node->full_name. Thus an overlay
389 * created from an FDT also will not contain the full path in
390 * node->full_name. However, a live devicetree created from Open
391 * Firmware may have the full path in node->full_name.
392 *
393 * add_changeset_node() follows the FDT convention and does not include
394 * the full path in node->full_name. Even though it expects the overlay
395 * to not contain the full path, it uses kbasename() to remove the
396 * full path should it exist. It also uses kbasename() in comparisons
397 * to nodes in the live devicetree so that it can apply an overlay to
398 * a live devicetree created from Open Firmware.
399 *
400 * NOTE_2: Multiple mods of created nodes not supported.
401 *
402 * Return: 0 on success, -ENOMEM if memory allocation failure, or -EINVAL if
403 * invalid @overlay.
404 */
405static int add_changeset_node(struct overlay_changeset *ovcs,
406 struct target *target, struct device_node *node)
407{
408 const char *node_kbasename;
409 const __be32 *phandle;
410 struct device_node *tchild;
411 struct target target_child;
412 int ret = 0, size;
413
414 node_kbasename = kbasename(node->full_name);
415
416 for_each_child_of_node(target->np, tchild)
417 if (!of_node_cmp(node_kbasename, kbasename(tchild->full_name)))
418 break;
419
420 if (!tchild) {
421 tchild = __of_node_dup(NULL, node_kbasename);
422 if (!tchild)
423 return -ENOMEM;
424
425 tchild->parent = target->np;
426 tchild->name = __of_get_property(node, "name", NULL);
427
428 if (!tchild->name)
429 tchild->name = "<NULL>";
430
431 /* ignore obsolete "linux,phandle" */
432 phandle = __of_get_property(node, "phandle", &size);
433 if (phandle && (size == 4))
434 tchild->phandle = be32_to_cpup(phandle);
435
436 of_node_set_flag(tchild, OF_OVERLAY);
437
438 ret = of_changeset_attach_node(&ovcs->cset, tchild);
439 if (ret)
440 return ret;
441
442 target_child.np = tchild;
443 target_child.in_livetree = false;
444
445 ret = build_changeset_next_level(ovcs, &target_child, node);
446 of_node_put(tchild);
447 return ret;
448 }
449
450 if (node->phandle && tchild->phandle) {
451 ret = -EINVAL;
452 } else {
453 target_child.np = tchild;
454 target_child.in_livetree = target->in_livetree;
455 ret = build_changeset_next_level(ovcs, &target_child, node);
456 }
457 of_node_put(tchild);
458
459 return ret;
460}
461
462/**
463 * build_changeset_next_level() - add level of overlay changeset
464 * @ovcs: overlay changeset
465 * @target: where to place @overlay_node in live tree
466 * @overlay_node: node from within an overlay device tree fragment
467 *
468 * Add the properties (if any) and nodes (if any) from @overlay_node to the
469 * @ovcs->cset changeset. If an added node has child nodes, they will
470 * be added recursively.
471 *
472 * Do not allow symbols node to have any children.
473 *
474 * Return: 0 on success, -ENOMEM if memory allocation failure, or -EINVAL if
475 * invalid @overlay_node.
476 */
477static int build_changeset_next_level(struct overlay_changeset *ovcs,
478 struct target *target, const struct device_node *overlay_node)
479{
480 struct device_node *child;
481 struct property *prop;
482 int ret;
483
484 for_each_property_of_node(overlay_node, prop) {
485 ret = add_changeset_property(ovcs, target, prop, 0);
486 if (ret) {
487 pr_debug("Failed to apply prop @%pOF/%s, err=%d\n",
488 target->np, prop->name, ret);
489 return ret;
490 }
491 }
492
493 for_each_child_of_node(overlay_node, child) {
494 ret = add_changeset_node(ovcs, target, child);
495 if (ret) {
496 pr_debug("Failed to apply node @%pOF/%pOFn, err=%d\n",
497 target->np, child, ret);
498 of_node_put(child);
499 return ret;
500 }
501 }
502
503 return 0;
504}
505
506/*
507 * Add the properties from __overlay__ node to the @ovcs->cset changeset.
508 */
509static int build_changeset_symbols_node(struct overlay_changeset *ovcs,
510 struct target *target,
511 const struct device_node *overlay_symbols_node)
512{
513 struct property *prop;
514 int ret;
515
516 for_each_property_of_node(overlay_symbols_node, prop) {
517 ret = add_changeset_property(ovcs, target, prop, 1);
518 if (ret) {
519 pr_debug("Failed to apply symbols prop @%pOF/%s, err=%d\n",
520 target->np, prop->name, ret);
521 return ret;
522 }
523 }
524
525 return 0;
526}
527
528static int find_dup_cset_node_entry(struct overlay_changeset *ovcs,
529 struct of_changeset_entry *ce_1)
530{
531 struct of_changeset_entry *ce_2;
532 char *fn_1, *fn_2;
533 int node_path_match;
534
535 if (ce_1->action != OF_RECONFIG_ATTACH_NODE &&
536 ce_1->action != OF_RECONFIG_DETACH_NODE)
537 return 0;
538
539 ce_2 = ce_1;
540 list_for_each_entry_continue(ce_2, &ovcs->cset.entries, node) {
541 if ((ce_2->action != OF_RECONFIG_ATTACH_NODE &&
542 ce_2->action != OF_RECONFIG_DETACH_NODE) ||
543 of_node_cmp(ce_1->np->full_name, ce_2->np->full_name))
544 continue;
545
546 fn_1 = kasprintf(GFP_KERNEL, "%pOF", ce_1->np);
547 fn_2 = kasprintf(GFP_KERNEL, "%pOF", ce_2->np);
548 node_path_match = !fn_1 || !fn_2 || !strcmp(fn_1, fn_2);
549 kfree(fn_1);
550 kfree(fn_2);
551 if (node_path_match) {
552 pr_err("ERROR: multiple fragments add and/or delete node %pOF\n",
553 ce_1->np);
554 return -EINVAL;
555 }
556 }
557
558 return 0;
559}
560
561static int find_dup_cset_prop(struct overlay_changeset *ovcs,
562 struct of_changeset_entry *ce_1)
563{
564 struct of_changeset_entry *ce_2;
565 char *fn_1, *fn_2;
566 int node_path_match;
567
568 if (ce_1->action != OF_RECONFIG_ADD_PROPERTY &&
569 ce_1->action != OF_RECONFIG_REMOVE_PROPERTY &&
570 ce_1->action != OF_RECONFIG_UPDATE_PROPERTY)
571 return 0;
572
573 ce_2 = ce_1;
574 list_for_each_entry_continue(ce_2, &ovcs->cset.entries, node) {
575 if ((ce_2->action != OF_RECONFIG_ADD_PROPERTY &&
576 ce_2->action != OF_RECONFIG_REMOVE_PROPERTY &&
577 ce_2->action != OF_RECONFIG_UPDATE_PROPERTY) ||
578 of_node_cmp(ce_1->np->full_name, ce_2->np->full_name))
579 continue;
580
581 fn_1 = kasprintf(GFP_KERNEL, "%pOF", ce_1->np);
582 fn_2 = kasprintf(GFP_KERNEL, "%pOF", ce_2->np);
583 node_path_match = !fn_1 || !fn_2 || !strcmp(fn_1, fn_2);
584 kfree(fn_1);
585 kfree(fn_2);
586 if (node_path_match &&
587 !of_prop_cmp(ce_1->prop->name, ce_2->prop->name)) {
588 pr_err("ERROR: multiple fragments add, update, and/or delete property %pOF/%s\n",
589 ce_1->np, ce_1->prop->name);
590 return -EINVAL;
591 }
592 }
593
594 return 0;
595}
596
597/**
598 * changeset_dup_entry_check() - check for duplicate entries
599 * @ovcs: Overlay changeset
600 *
601 * Check changeset @ovcs->cset for multiple {add or delete} node entries for
602 * the same node or duplicate {add, delete, or update} properties entries
603 * for the same property.
604 *
605 * Return: 0 on success, or -EINVAL if duplicate changeset entry found.
606 */
607static int changeset_dup_entry_check(struct overlay_changeset *ovcs)
608{
609 struct of_changeset_entry *ce_1;
610 int dup_entry = 0;
611
612 list_for_each_entry(ce_1, &ovcs->cset.entries, node) {
613 dup_entry |= find_dup_cset_node_entry(ovcs, ce_1);
614 dup_entry |= find_dup_cset_prop(ovcs, ce_1);
615 }
616
617 return dup_entry ? -EINVAL : 0;
618}
619
620/**
621 * build_changeset() - populate overlay changeset in @ovcs from @ovcs->fragments
622 * @ovcs: Overlay changeset
623 *
624 * Create changeset @ovcs->cset to contain the nodes and properties of the
625 * overlay device tree fragments in @ovcs->fragments[]. If an error occurs,
626 * any portions of the changeset that were successfully created will remain
627 * in @ovcs->cset.
628 *
629 * Return: 0 on success, -ENOMEM if memory allocation failure, or -EINVAL if
630 * invalid overlay in @ovcs->fragments[].
631 */
632static int build_changeset(struct overlay_changeset *ovcs)
633{
634 struct fragment *fragment;
635 struct target target;
636 int fragments_count, i, ret;
637
638 /*
639 * if there is a symbols fragment in ovcs->fragments[i] it is
640 * the final element in the array
641 */
642 if (ovcs->symbols_fragment)
643 fragments_count = ovcs->count - 1;
644 else
645 fragments_count = ovcs->count;
646
647 for (i = 0; i < fragments_count; i++) {
648 fragment = &ovcs->fragments[i];
649
650 target.np = fragment->target;
651 target.in_livetree = true;
652 ret = build_changeset_next_level(ovcs, &target,
653 fragment->overlay);
654 if (ret) {
655 pr_debug("fragment apply failed '%pOF'\n",
656 fragment->target);
657 return ret;
658 }
659 }
660
661 if (ovcs->symbols_fragment) {
662 fragment = &ovcs->fragments[ovcs->count - 1];
663
664 target.np = fragment->target;
665 target.in_livetree = true;
666 ret = build_changeset_symbols_node(ovcs, &target,
667 fragment->overlay);
668 if (ret) {
669 pr_debug("symbols fragment apply failed '%pOF'\n",
670 fragment->target);
671 return ret;
672 }
673 }
674
675 return changeset_dup_entry_check(ovcs);
676}
677
678/*
679 * Find the target node using a number of different strategies
680 * in order of preference:
681 *
682 * 1) "target" property containing the phandle of the target
683 * 2) "target-path" property containing the path of the target
684 */
685static struct device_node *find_target(struct device_node *info_node)
686{
687 struct device_node *node;
688 const char *path;
689 u32 val;
690 int ret;
691
692 ret = of_property_read_u32(info_node, "target", &val);
693 if (!ret) {
694 node = of_find_node_by_phandle(val);
695 if (!node)
696 pr_err("find target, node: %pOF, phandle 0x%x not found\n",
697 info_node, val);
698 return node;
699 }
700
701 ret = of_property_read_string(info_node, "target-path", &path);
702 if (!ret) {
703 node = of_find_node_by_path(path);
704 if (!node)
705 pr_err("find target, node: %pOF, path '%s' not found\n",
706 info_node, path);
707 return node;
708 }
709
710 pr_err("find target, node: %pOF, no target property\n", info_node);
711
712 return NULL;
713}
714
715/**
716 * init_overlay_changeset() - initialize overlay changeset from overlay tree
717 * @ovcs: Overlay changeset to build
718 *
719 * Initialize @ovcs. Populate @ovcs->fragments with node information from
720 * the top level of @overlay_root. The relevant top level nodes are the
721 * fragment nodes and the __symbols__ node. Any other top level node will
722 * be ignored. Populate other @ovcs fields.
723 *
724 * Return: 0 on success, -ENOMEM if memory allocation failure, -EINVAL if error
725 * detected in @overlay_root. On error return, the caller of
726 * init_overlay_changeset() must call free_overlay_changeset().
727 */
728static int init_overlay_changeset(struct overlay_changeset *ovcs)
729{
730 struct device_node *node, *overlay_node;
731 struct fragment *fragment;
732 struct fragment *fragments;
733 int cnt, ret;
734
735 /*
736 * None of the resources allocated by this function will be freed in
737 * the error paths. Instead the caller of this function is required
738 * to call free_overlay_changeset() (which will free the resources)
739 * if error return.
740 */
741
742 /*
743 * Warn for some issues. Can not return -EINVAL for these until
744 * of_unittest_apply_overlay() is fixed to pass these checks.
745 */
746 if (!of_node_check_flag(ovcs->overlay_root, OF_DYNAMIC))
747 pr_debug("%s() ovcs->overlay_root is not dynamic\n", __func__);
748
749 if (!of_node_check_flag(ovcs->overlay_root, OF_DETACHED))
750 pr_debug("%s() ovcs->overlay_root is not detached\n", __func__);
751
752 if (!of_node_is_root(ovcs->overlay_root))
753 pr_debug("%s() ovcs->overlay_root is not root\n", __func__);
754
755 of_changeset_init(&ovcs->cset);
756
757 cnt = 0;
758
759 /* fragment nodes */
760 for_each_child_of_node(ovcs->overlay_root, node) {
761 overlay_node = of_get_child_by_name(node, "__overlay__");
762 if (overlay_node) {
763 cnt++;
764 of_node_put(overlay_node);
765 }
766 }
767
768 node = of_get_child_by_name(ovcs->overlay_root, "__symbols__");
769 if (node) {
770 cnt++;
771 of_node_put(node);
772 }
773
774 fragments = kcalloc(cnt, sizeof(*fragments), GFP_KERNEL);
775 if (!fragments) {
776 ret = -ENOMEM;
777 goto err_out;
778 }
779 ovcs->fragments = fragments;
780
781 cnt = 0;
782 for_each_child_of_node(ovcs->overlay_root, node) {
783 overlay_node = of_get_child_by_name(node, "__overlay__");
784 if (!overlay_node)
785 continue;
786
787 fragment = &fragments[cnt];
788 fragment->overlay = overlay_node;
789 fragment->target = find_target(node);
790 if (!fragment->target) {
791 of_node_put(fragment->overlay);
792 ret = -EINVAL;
793 of_node_put(node);
794 goto err_out;
795 }
796
797 cnt++;
798 }
799
800 /*
801 * if there is a symbols fragment in ovcs->fragments[i] it is
802 * the final element in the array
803 */
804 node = of_get_child_by_name(ovcs->overlay_root, "__symbols__");
805 if (node) {
806 ovcs->symbols_fragment = 1;
807 fragment = &fragments[cnt];
808 fragment->overlay = node;
809 fragment->target = of_find_node_by_path("/__symbols__");
810
811 if (!fragment->target) {
812 pr_err("symbols in overlay, but not in live tree\n");
813 ret = -EINVAL;
814 goto err_out;
815 }
816
817 cnt++;
818 }
819
820 if (!cnt) {
821 pr_err("no fragments or symbols in overlay\n");
822 ret = -EINVAL;
823 goto err_out;
824 }
825
826 ovcs->count = cnt;
827
828 return 0;
829
830err_out:
831 pr_err("%s() failed, ret = %d\n", __func__, ret);
832
833 return ret;
834}
835
836static void free_overlay_changeset(struct overlay_changeset *ovcs)
837{
838 int i;
839
840 if (ovcs->cset.entries.next)
841 of_changeset_destroy(&ovcs->cset);
842
843 if (ovcs->id) {
844 idr_remove(&ovcs_idr, ovcs->id);
845 list_del(&ovcs->ovcs_list);
846 ovcs->id = 0;
847 }
848
849
850 for (i = 0; i < ovcs->count; i++) {
851 of_node_put(ovcs->fragments[i].target);
852 of_node_put(ovcs->fragments[i].overlay);
853 }
854 kfree(ovcs->fragments);
855
856 /*
857 * There should be no live pointers into ovcs->overlay_mem and
858 * ovcs->new_fdt due to the policy that overlay notifiers are not
859 * allowed to retain pointers into the overlay devicetree other
860 * than during the window from OF_OVERLAY_PRE_APPLY overlay
861 * notifiers until the OF_OVERLAY_POST_REMOVE overlay notifiers.
862 *
863 * A memory leak will occur here if within the window.
864 */
865
866 if (ovcs->notify_state == OF_OVERLAY_INIT ||
867 ovcs->notify_state == OF_OVERLAY_POST_REMOVE) {
868 kfree(ovcs->overlay_mem);
869 kfree(ovcs->new_fdt);
870 }
871 kfree(ovcs);
872}
873
874/*
875 * internal documentation
876 *
877 * of_overlay_apply() - Create and apply an overlay changeset
878 * @ovcs: overlay changeset
879 *
880 * Creates and applies an overlay changeset.
881 *
882 * If an error is returned by an overlay changeset pre-apply notifier
883 * then no further overlay changeset pre-apply notifier will be called.
884 *
885 * If an error is returned by an overlay changeset post-apply notifier
886 * then no further overlay changeset post-apply notifier will be called.
887 *
888 * If more than one notifier returns an error, then the last notifier
889 * error to occur is returned.
890 *
891 * If an error occurred while applying the overlay changeset, then an
892 * attempt is made to revert any changes that were made to the
893 * device tree. If there were any errors during the revert attempt
894 * then the state of the device tree can not be determined, and any
895 * following attempt to apply or remove an overlay changeset will be
896 * refused.
897 *
898 * Returns 0 on success, or a negative error number. On error return,
899 * the caller of of_overlay_apply() must call free_overlay_changeset().
900 */
901
902static int of_overlay_apply(struct overlay_changeset *ovcs)
903{
904 int ret = 0, ret_revert, ret_tmp;
905
906 ret = of_resolve_phandles(ovcs->overlay_root);
907 if (ret)
908 goto out;
909
910 ret = init_overlay_changeset(ovcs);
911 if (ret)
912 goto out;
913
914 ret = overlay_notify(ovcs, OF_OVERLAY_PRE_APPLY);
915 if (ret)
916 goto out;
917
918 ret = build_changeset(ovcs);
919 if (ret)
920 goto out;
921
922 ret_revert = 0;
923 ret = __of_changeset_apply_entries(&ovcs->cset, &ret_revert);
924 if (ret) {
925 if (ret_revert) {
926 pr_debug("overlay changeset revert error %d\n",
927 ret_revert);
928 devicetree_state_flags |= DTSF_APPLY_FAIL;
929 }
930 goto out;
931 }
932
933 ret = __of_changeset_apply_notify(&ovcs->cset);
934 if (ret)
935 pr_err("overlay apply changeset entry notify error %d\n", ret);
936 /* notify failure is not fatal, continue */
937
938 ret_tmp = overlay_notify(ovcs, OF_OVERLAY_POST_APPLY);
939 if (ret_tmp)
940 if (!ret)
941 ret = ret_tmp;
942
943out:
944 pr_debug("%s() err=%d\n", __func__, ret);
945
946 return ret;
947}
948
949/*
950 * of_overlay_fdt_apply() - Create and apply an overlay changeset
951 * @overlay_fdt: pointer to overlay FDT
952 * @overlay_fdt_size: number of bytes in @overlay_fdt
953 * @ret_ovcs_id: pointer for returning created changeset id
954 *
955 * Creates and applies an overlay changeset.
956 *
957 * See of_overlay_apply() for important behavior information.
958 *
959 * Return: 0 on success, or a negative error number. *@ret_ovcs_id is set to
960 * the value of overlay changeset id, which can be passed to of_overlay_remove()
961 * to remove the overlay.
962 *
963 * On error return, the changeset may be partially applied. This is especially
964 * likely if an OF_OVERLAY_POST_APPLY notifier returns an error. In this case
965 * the caller should call of_overlay_remove() with the value in *@ret_ovcs_id.
966 */
967
968int of_overlay_fdt_apply(const void *overlay_fdt, u32 overlay_fdt_size,
969 int *ret_ovcs_id)
970{
971 void *new_fdt;
972 void *new_fdt_align;
973 void *overlay_mem;
974 int ret;
975 u32 size;
976 struct overlay_changeset *ovcs;
977
978 *ret_ovcs_id = 0;
979
980 if (devicetree_corrupt()) {
981 pr_err("devicetree state suspect, refuse to apply overlay\n");
982 return -EBUSY;
983 }
984
985 if (overlay_fdt_size < sizeof(struct fdt_header) ||
986 fdt_check_header(overlay_fdt)) {
987 pr_err("Invalid overlay_fdt header\n");
988 return -EINVAL;
989 }
990
991 size = fdt_totalsize(overlay_fdt);
992 if (overlay_fdt_size < size)
993 return -EINVAL;
994
995 ovcs = kzalloc(sizeof(*ovcs), GFP_KERNEL);
996 if (!ovcs)
997 return -ENOMEM;
998
999 of_overlay_mutex_lock();
1000 mutex_lock(&of_mutex);
1001
1002 /*
1003 * ovcs->notify_state must be set to OF_OVERLAY_INIT before allocating
1004 * ovcs resources, implicitly set by kzalloc() of ovcs
1005 */
1006
1007 ovcs->id = idr_alloc(&ovcs_idr, ovcs, 1, 0, GFP_KERNEL);
1008 if (ovcs->id <= 0) {
1009 ret = ovcs->id;
1010 goto err_free_ovcs;
1011 }
1012
1013 INIT_LIST_HEAD(&ovcs->ovcs_list);
1014 list_add_tail(&ovcs->ovcs_list, &ovcs_list);
1015
1016 /*
1017 * Must create permanent copy of FDT because of_fdt_unflatten_tree()
1018 * will create pointers to the passed in FDT in the unflattened tree.
1019 */
1020 new_fdt = kmalloc(size + FDT_ALIGN_SIZE, GFP_KERNEL);
1021 if (!new_fdt) {
1022 ret = -ENOMEM;
1023 goto err_free_ovcs;
1024 }
1025 ovcs->new_fdt = new_fdt;
1026
1027 new_fdt_align = PTR_ALIGN(new_fdt, FDT_ALIGN_SIZE);
1028 memcpy(new_fdt_align, overlay_fdt, size);
1029
1030 overlay_mem = of_fdt_unflatten_tree(new_fdt_align, NULL,
1031 &ovcs->overlay_root);
1032 if (!overlay_mem) {
1033 pr_err("unable to unflatten overlay_fdt\n");
1034 ret = -EINVAL;
1035 goto err_free_ovcs;
1036 }
1037 ovcs->overlay_mem = overlay_mem;
1038
1039 ret = of_overlay_apply(ovcs);
1040 /*
1041 * If of_overlay_apply() error, calling free_overlay_changeset() may
1042 * result in a memory leak if the apply partly succeeded, so do NOT
1043 * goto err_free_ovcs. Instead, the caller of of_overlay_fdt_apply()
1044 * can call of_overlay_remove();
1045 */
1046 *ret_ovcs_id = ovcs->id;
1047 goto out_unlock;
1048
1049err_free_ovcs:
1050 free_overlay_changeset(ovcs);
1051
1052out_unlock:
1053 mutex_unlock(&of_mutex);
1054 of_overlay_mutex_unlock();
1055 return ret;
1056}
1057EXPORT_SYMBOL_GPL(of_overlay_fdt_apply);
1058
1059/*
1060 * Find @np in @tree.
1061 *
1062 * Returns 1 if @np is @tree or is contained in @tree, else 0
1063 */
1064static int find_node(struct device_node *tree, struct device_node *np)
1065{
1066 struct device_node *child;
1067
1068 if (tree == np)
1069 return 1;
1070
1071 for_each_child_of_node(tree, child) {
1072 if (find_node(child, np)) {
1073 of_node_put(child);
1074 return 1;
1075 }
1076 }
1077
1078 return 0;
1079}
1080
1081/*
1082 * Is @remove_ce_node a child of, a parent of, or the same as any
1083 * node in an overlay changeset more topmost than @remove_ovcs?
1084 *
1085 * Returns 1 if found, else 0
1086 */
1087static int node_overlaps_later_cs(struct overlay_changeset *remove_ovcs,
1088 struct device_node *remove_ce_node)
1089{
1090 struct overlay_changeset *ovcs;
1091 struct of_changeset_entry *ce;
1092
1093 list_for_each_entry_reverse(ovcs, &ovcs_list, ovcs_list) {
1094 if (ovcs == remove_ovcs)
1095 break;
1096
1097 list_for_each_entry(ce, &ovcs->cset.entries, node) {
1098 if (find_node(ce->np, remove_ce_node)) {
1099 pr_err("%s: #%d overlaps with #%d @%pOF\n",
1100 __func__, remove_ovcs->id, ovcs->id,
1101 remove_ce_node);
1102 return 1;
1103 }
1104 if (find_node(remove_ce_node, ce->np)) {
1105 pr_err("%s: #%d overlaps with #%d @%pOF\n",
1106 __func__, remove_ovcs->id, ovcs->id,
1107 remove_ce_node);
1108 return 1;
1109 }
1110 }
1111 }
1112
1113 return 0;
1114}
1115
1116/*
1117 * We can safely remove the overlay only if it's the top-most one.
1118 * Newly applied overlays are inserted at the tail of the overlay list,
1119 * so a top most overlay is the one that is closest to the tail.
1120 *
1121 * The topmost check is done by exploiting this property. For each
1122 * affected device node in the log list we check if this overlay is
1123 * the one closest to the tail. If another overlay has affected this
1124 * device node and is closest to the tail, then removal is not permited.
1125 */
1126static int overlay_removal_is_ok(struct overlay_changeset *remove_ovcs)
1127{
1128 struct of_changeset_entry *remove_ce;
1129
1130 list_for_each_entry(remove_ce, &remove_ovcs->cset.entries, node) {
1131 if (node_overlaps_later_cs(remove_ovcs, remove_ce->np)) {
1132 pr_err("overlay #%d is not topmost\n", remove_ovcs->id);
1133 return 0;
1134 }
1135 }
1136
1137 return 1;
1138}
1139
1140/**
1141 * of_overlay_remove() - Revert and free an overlay changeset
1142 * @ovcs_id: Pointer to overlay changeset id
1143 *
1144 * Removes an overlay if it is permissible. @ovcs_id was previously returned
1145 * by of_overlay_fdt_apply().
1146 *
1147 * If an error occurred while attempting to revert the overlay changeset,
1148 * then an attempt is made to re-apply any changeset entry that was
1149 * reverted. If an error occurs on re-apply then the state of the device
1150 * tree can not be determined, and any following attempt to apply or remove
1151 * an overlay changeset will be refused.
1152 *
1153 * A non-zero return value will not revert the changeset if error is from:
1154 * - parameter checks
1155 * - overlay changeset pre-remove notifier
1156 * - overlay changeset entry revert
1157 *
1158 * If an error is returned by an overlay changeset pre-remove notifier
1159 * then no further overlay changeset pre-remove notifier will be called.
1160 *
1161 * If more than one notifier returns an error, then the last notifier
1162 * error to occur is returned.
1163 *
1164 * A non-zero return value will revert the changeset if error is from:
1165 * - overlay changeset entry notifier
1166 * - overlay changeset post-remove notifier
1167 *
1168 * If an error is returned by an overlay changeset post-remove notifier
1169 * then no further overlay changeset post-remove notifier will be called.
1170 *
1171 * Return: 0 on success, or a negative error number. *@ovcs_id is set to
1172 * zero after reverting the changeset, even if a subsequent error occurs.
1173 */
1174int of_overlay_remove(int *ovcs_id)
1175{
1176 struct overlay_changeset *ovcs;
1177 int ret, ret_apply, ret_tmp;
1178
1179 if (devicetree_corrupt()) {
1180 pr_err("suspect devicetree state, refuse to remove overlay\n");
1181 ret = -EBUSY;
1182 goto out;
1183 }
1184
1185 mutex_lock(&of_mutex);
1186
1187 ovcs = idr_find(&ovcs_idr, *ovcs_id);
1188 if (!ovcs) {
1189 ret = -ENODEV;
1190 pr_err("remove: Could not find overlay #%d\n", *ovcs_id);
1191 goto err_unlock;
1192 }
1193
1194 if (!overlay_removal_is_ok(ovcs)) {
1195 ret = -EBUSY;
1196 goto err_unlock;
1197 }
1198
1199 ret = overlay_notify(ovcs, OF_OVERLAY_PRE_REMOVE);
1200 if (ret)
1201 goto err_unlock;
1202
1203 ret_apply = 0;
1204 ret = __of_changeset_revert_entries(&ovcs->cset, &ret_apply);
1205 if (ret) {
1206 if (ret_apply)
1207 devicetree_state_flags |= DTSF_REVERT_FAIL;
1208 goto err_unlock;
1209 }
1210
1211 ret = __of_changeset_revert_notify(&ovcs->cset);
1212 if (ret)
1213 pr_err("overlay remove changeset entry notify error %d\n", ret);
1214 /* notify failure is not fatal, continue */
1215
1216 *ovcs_id = 0;
1217
1218 /*
1219 * Note that the overlay memory will be kfree()ed by
1220 * free_overlay_changeset() even if the notifier for
1221 * OF_OVERLAY_POST_REMOVE returns an error.
1222 */
1223 ret_tmp = overlay_notify(ovcs, OF_OVERLAY_POST_REMOVE);
1224 if (ret_tmp)
1225 if (!ret)
1226 ret = ret_tmp;
1227
1228 free_overlay_changeset(ovcs);
1229
1230err_unlock:
1231 /*
1232 * If jumped over free_overlay_changeset(), then did not kfree()
1233 * overlay related memory. This is a memory leak unless a subsequent
1234 * of_overlay_remove() of this overlay is successful.
1235 */
1236 mutex_unlock(&of_mutex);
1237
1238out:
1239 pr_debug("%s() err=%d\n", __func__, ret);
1240
1241 return ret;
1242}
1243EXPORT_SYMBOL_GPL(of_overlay_remove);
1244
1245/**
1246 * of_overlay_remove_all() - Reverts and frees all overlay changesets
1247 *
1248 * Removes all overlays from the system in the correct order.
1249 *
1250 * Return: 0 on success, or a negative error number
1251 */
1252int of_overlay_remove_all(void)
1253{
1254 struct overlay_changeset *ovcs, *ovcs_n;
1255 int ret;
1256
1257 /* the tail of list is guaranteed to be safe to remove */
1258 list_for_each_entry_safe_reverse(ovcs, ovcs_n, &ovcs_list, ovcs_list) {
1259 ret = of_overlay_remove(&ovcs->id);
1260 if (ret)
1261 return ret;
1262 }
1263
1264 return 0;
1265}
1266EXPORT_SYMBOL_GPL(of_overlay_remove_all);