1 files changed, 6 insertions, 0 deletions

diff --git a/lib/radix-tree.c b/lib/radix-tree.c
index d64815651e90..3291a8e37490 100644
--- a/lib/radix-tree.c
+++ b/lib/radix-tree.c
@@ -27,6 +27,7 @@
 #include <linux/radix-tree.h>
 #include <linux/percpu.h>
 #include <linux/slab.h>
+#include <linux/kmemleak.h>
 #include <linux/notifier.h>
 #include <linux/cpu.h>
 #include <linux/string.h>
@@ -200,6 +201,11 @@ radix_tree_node_alloc(struct radix_tree_root *root)
                         rtp->nodes[rtp->nr - 1] = NULL;
                         rtp->nr--;
                 }
+                /*
+                 * Update the allocation stack trace as this is more useful
+                 * for debugging.
+                 */
+                kmemleak_update_trace(ret);
         }
         if (ret == NULL)
                 ret = kmem_cache_alloc(radix_tree_node_cachep, gfp_mask);