of: Fix NULL dereference in selftest removal code

The selftest code removes its testcase data from the live tree when
exiting, but if the testcases data tree contains an empty child of the
root, then it causes an oops due to a NULL dereference. The reason is
that the code tries to directly dereference the child pointer without
checking first if a child is actually there.

The solution is to pass the parent node into detach_node_and_children()
instead of trying to pass the child. This required removing the code
that attempts to remove all of the sibling nodes in
detach_node_and_children(), which was never sensible in the first place.

At the same time add a check to make sure the bounds of the nodes list
are not exceeded by the testdata tree. If they are then abort.

Signed-off-by: Grant Likely <grant.likely@linaro.org>
Cc: Gaurav Minocha <gaurav.minocha.os@gmail.com>

1 files changed, 3 insertions, 6 deletions

diff --git a/drivers/of/selftest.c b/drivers/of/selftest.c
index a737cb5974de..883e60b04eb5 100644
--- a/drivers/of/selftest.c
+++ b/drivers/of/selftest.c
@@ -637,6 +637,8 @@ static int attach_node_and_children(struct device_node *np)
         dup = np;
 
         while (dup) {
+                if (WARN_ON(last_node_index >= NO_OF_NODES))
+                        return -EINVAL;
                 nodes[last_node_index++] = dup;
                 dup = dup->sibling;
         }
@@ -717,10 +719,6 @@ static void detach_node_and_children(struct device_node *np)
 {
         while (np->child)
                 detach_node_and_children(np->child);
-
-        while (np->sibling)
-                detach_node_and_children(np->sibling);
-
         of_detach_node(np);
 }
 
@@ -749,8 +747,7 @@ static void selftest_data_remove(void)
                 if (nodes[last_node_index]) {
                         np = of_find_node_by_path(nodes[last_node_index]->full_name);
                         if (strcmp(np->full_name, "/aliases") != 0) {
-                                detach_node_and_children(np->child);
-                                of_detach_node(np);
+                                detach_node_and_children(np);
                         } else {
                                 for_each_property_of_node(np, prop) {
                                         if (strcmp(prop->name, "testcase-alias") == 0)