Since drm_mm_node are intended to be embedded into larger structs, we
can use the DEBUG_OBJECTS facility to help track the lifetime of the
drm_mm_node and ensure that they are not being used after the containing
object has been freed, along with the usual verification that the
drm_mm_nodes are being used correctly.

Signed-off-by: Chris Wilson <ch...@chris-wilson.co.uk>
---
And now switch on for CI. Expect failure from unmarked drm_mm_nodes.
---
 drivers/gpu/drm/Kconfig              |  13 +++
 drivers/gpu/drm/drm_mm.c             | 113 +++++++++++++++++++++++----
 drivers/gpu/drm/i915/Kconfig.debug   |   1 +
 drivers/gpu/drm/i915/gt/gen6_ppgtt.c |   2 -
 drivers/gpu/drm/i915/i915_vma.c      |  17 +++-
 include/drm/drm_mm.h                 |  31 ++++++++
 6 files changed, 158 insertions(+), 19 deletions(-)

diff --git a/drivers/gpu/drm/Kconfig b/drivers/gpu/drm/Kconfig
index 43594978958e..33ac38be55b0 100644
--- a/drivers/gpu/drm/Kconfig
+++ b/drivers/gpu/drm/Kconfig
@@ -54,6 +54,19 @@ config DRM_DEBUG_MM
 
          If in doubt, say "N".
 
+config DRM_DEBUG_MM_OBJECTS
+       bool "Use debugobjects to track live drm_mm_nodes"
+       default n
+       depends on DRM_DEBUG_MM
+       depends on DEBUG_OBJECTS
+       help
+         Enable allocation tracking of memory manager and leak detection on
+         shutdown.
+
+         Recommended for driver developers only.
+
+         If in doubt, say "N".
+
 config DRM_DEBUG_SELFTEST
        tristate "kselftests for DRM"
        depends on DRM
diff --git a/drivers/gpu/drm/drm_mm.c b/drivers/gpu/drm/drm_mm.c
index bc6e208949e8..3cbee92f5832 100644
--- a/drivers/gpu/drm/drm_mm.c
+++ b/drivers/gpu/drm/drm_mm.c
@@ -116,38 +116,99 @@ static noinline void save_stack(struct drm_mm_node *node)
        node->stack = stack_depot_save(entries, n, GFP_NOWAIT);
 }
 
-static void show_leaks(struct drm_mm *mm)
+static void __print_node(struct drm_mm_node *node,
+                        char *buf, int sz,
+                        const char *reason)
 {
-       struct drm_mm_node *node;
        unsigned long *entries;
        unsigned int nr_entries;
+
+       if (!node->stack) {
+               DRM_ERROR("node [%08llx + %08llx]: %s, unknown owner\n",
+                         node->start, node->size, reason);
+               return;
+       }
+
+       nr_entries = stack_depot_fetch(node->stack, &entries);
+       stack_trace_snprint(buf, sz, entries, nr_entries, 0);
+       DRM_ERROR("node [%08llx + %08llx]: %s, inserted at\n%s",
+                 node->start, node->size, reason, buf);
+}
+
+static void show_leaks(struct drm_mm *mm)
+{
+       struct drm_mm_node *node;
        char *buf;
 
        buf = kmalloc(BUFSZ, GFP_KERNEL);
        if (!buf)
                return;
 
-       list_for_each_entry(node, drm_mm_nodes(mm), node_list) {
-               if (!node->stack) {
-                       DRM_ERROR("node [%08llx + %08llx]: unknown owner\n",
-                                 node->start, node->size);
-                       continue;
-               }
-
-               nr_entries = stack_depot_fetch(node->stack, &entries);
-               stack_trace_snprint(buf, BUFSZ, entries, nr_entries, 0);
-               DRM_ERROR("node [%08llx + %08llx]: inserted at\n%s",
-                         node->start, node->size, buf);
-       }
+       list_for_each_entry(node, drm_mm_nodes(mm), node_list)
+               __print_node(node, buf, BUFSZ, "leaked");
 
        kfree(buf);
 }
 
+static void show_node(struct drm_mm_node *node, const char *reason)
+{
+       char buf[128];
+
+       __print_node(node, buf, sizeof(buf), reason);
+}
+
 #undef STACKDEPTH
 #undef BUFSZ
 #else
 static void save_stack(struct drm_mm_node *node) { }
 static void show_leaks(struct drm_mm *mm) { }
+
+static void show_node(struct drm_mm_node *node, const char *reason)
+{
+       DRM_ERROR("node [%08llx + %08llx]: %s\n",
+                 node->start, node->size, reason);
+}
+#endif
+
+#ifdef CONFIG_DRM_DEBUG_MM_OBJECTS
+
+static struct debug_obj_descr drm_mm_debug_descr = {
+       .name = "drm_mm_node",
+};
+
+static inline void debug_node_init(struct drm_mm_node *node)
+{
+       debug_object_init(node, &drm_mm_debug_descr);
+}
+
+static inline void debug_node_activate(struct drm_mm_node *node)
+{
+       debug_object_activate(node, &drm_mm_debug_descr);
+}
+
+static inline void debug_node_deactivate(struct drm_mm_node *node)
+{
+       debug_object_deactivate(node, &drm_mm_debug_descr);
+}
+
+static inline void debug_node_free(struct drm_mm_node *node)
+{
+       debug_object_free(node, &drm_mm_debug_descr);
+}
+
+static inline void debug_node_assert(struct drm_mm_node *node)
+{
+       debug_object_assert_init(node, &drm_mm_debug_descr);
+}
+
+#else
+
+static inline void debug_node_init(struct drm_mm_node *node) { }
+static inline void debug_node_activate(struct drm_mm_node *node) { }
+static inline void debug_node_deactivate(struct drm_mm_node *node) { }
+static inline void debug_node_free(struct drm_mm_node *node) { }
+static inline void debug_node_assert(struct drm_mm_node *node) { }
+
 #endif
 
 #define START(node) ((node)->start)
@@ -428,6 +489,8 @@ int drm_mm_reserve_node(struct drm_mm *mm, struct 
drm_mm_node *node)
        if (adj_start > node->start || adj_end < end)
                return -ENOSPC;
 
+       debug_node_activate(node);
+
        node->mm = mm;
 
        __set_bit(DRM_MM_NODE_ALLOCATED_BIT, &node->flags);
@@ -543,6 +606,8 @@ int drm_mm_insert_node_in_range(struct drm_mm * const mm,
                        }
                }
 
+               debug_node_activate(node);
+
                node->mm = mm;
                node->size = size;
                node->start = adj_start;
@@ -587,6 +652,7 @@ void drm_mm_remove_node(struct drm_mm_node *node)
 
        DRM_MM_BUG_ON(!drm_mm_node_allocated(node));
        DRM_MM_BUG_ON(drm_mm_node_scanned_block(node));
+       debug_node_deactivate(node);
 
        prev_node = list_prev_entry(node, node_list);
 
@@ -618,6 +684,8 @@ void drm_mm_replace_node(struct drm_mm_node *old, struct 
drm_mm_node *new)
        struct drm_mm *mm = old->mm;
 
        DRM_MM_BUG_ON(!drm_mm_node_allocated(old));
+       debug_node_deactivate(old);
+       debug_node_activate(new);
 
        *new = *old;
 
@@ -639,6 +707,21 @@ void drm_mm_replace_node(struct drm_mm_node *old, struct 
drm_mm_node *new)
 }
 EXPORT_SYMBOL(drm_mm_replace_node);
 
+void __drm_mm_node_init(struct drm_mm_node *node)
+{
+       debug_node_init(node);
+}
+EXPORT_SYMBOL(__drm_mm_node_init);
+
+void __drm_mm_node_fini(struct drm_mm_node *node)
+{
+       if (unlikely(drm_mm_node_allocated(node)))
+               show_node(node, "use-after-free");
+
+       debug_node_free(node);
+}
+EXPORT_SYMBOL(__drm_mm_node_fini);
+
 /**
  * DOC: lru scan roster
  *
@@ -742,6 +825,7 @@ bool drm_mm_scan_add_block(struct drm_mm_scan *scan,
        u64 col_start, col_end;
        u64 adj_start, adj_end;
 
+       debug_node_assert(node);
        DRM_MM_BUG_ON(node->mm != mm);
        DRM_MM_BUG_ON(!drm_mm_node_allocated(node));
        DRM_MM_BUG_ON(drm_mm_node_scanned_block(node));
@@ -829,6 +913,7 @@ bool drm_mm_scan_remove_block(struct drm_mm_scan *scan,
 {
        struct drm_mm_node *prev_node;
 
+       debug_node_assert(node);
        DRM_MM_BUG_ON(node->mm != scan->mm);
        DRM_MM_BUG_ON(!drm_mm_node_scanned_block(node));
        __clear_bit(DRM_MM_NODE_SCANNED_BIT, &node->flags);
diff --git a/drivers/gpu/drm/i915/Kconfig.debug 
b/drivers/gpu/drm/i915/Kconfig.debug
index 206882e154bc..21ecc4cb3e8b 100644
--- a/drivers/gpu/drm/i915/Kconfig.debug
+++ b/drivers/gpu/drm/i915/Kconfig.debug
@@ -35,6 +35,7 @@ config DRM_I915_DEBUG
        select X86_MSR # used by igt/pm_rpm
        select DRM_VGEM # used by igt/prime_vgem (dmabuf interop checks)
        select DRM_DEBUG_MM if DRM=y
+       select DRM_DEBUG_MM_OBJECTS
        select DRM_EXPORT_FOR_TESTS if m
        select DRM_DEBUG_SELFTEST
        select DMABUF_SELFTESTS
diff --git a/drivers/gpu/drm/i915/gt/gen6_ppgtt.c 
b/drivers/gpu/drm/i915/gt/gen6_ppgtt.c
index f4fec7eb4064..0c34d4dd6458 100644
--- a/drivers/gpu/drm/i915/gt/gen6_ppgtt.c
+++ b/drivers/gpu/drm/i915/gt/gen6_ppgtt.c
@@ -359,8 +359,6 @@ static struct i915_vma *pd_vma_create(struct gen6_ppgtt 
*ppgtt, int size)
 
        i915_active_init(&vma->active, NULL, NULL);
 
-       kref_init(&vma->ref);
-       mutex_init(&vma->pages_mutex);
        vma->vm = i915_vm_get(&ggtt->vm);
        vma->ops = &pd_vma_ops;
        vma->private = ppgtt;
diff --git a/drivers/gpu/drm/i915/i915_vma.c b/drivers/gpu/drm/i915/i915_vma.c
index 5b3efb43a8ef..64140d51ae2c 100644
--- a/drivers/gpu/drm/i915/i915_vma.c
+++ b/drivers/gpu/drm/i915/i915_vma.c
@@ -45,11 +45,24 @@ static struct i915_global_vma {
 
 struct i915_vma *i915_vma_alloc(void)
 {
-       return kmem_cache_zalloc(global.slab_vmas, GFP_KERNEL);
+       struct i915_vma *vma;
+
+       vma = kmem_cache_zalloc(global.slab_vmas, GFP_KERNEL);
+       if (!vma)
+               return NULL;
+
+       kref_init(&vma->ref);
+       mutex_init(&vma->pages_mutex);
+       drm_mm_node_init(&vma->node);
+
+       return vma;
 }
 
 void i915_vma_free(struct i915_vma *vma)
 {
+       drm_mm_node_fini(&vma->node);
+       mutex_destroy(&vma->pages_mutex);
+
        return kmem_cache_free(global.slab_vmas, vma);
 }
 
@@ -114,8 +127,6 @@ vma_create(struct drm_i915_gem_object *obj,
        if (vma == NULL)
                return ERR_PTR(-ENOMEM);
 
-       kref_init(&vma->ref);
-       mutex_init(&vma->pages_mutex);
        vma->vm = i915_vm_get(vm);
        vma->ops = &vm->vma_ops;
        vma->obj = obj;
diff --git a/include/drm/drm_mm.h b/include/drm/drm_mm.h
index d7939c054259..514685590603 100644
--- a/include/drm/drm_mm.h
+++ b/include/drm/drm_mm.h
@@ -275,6 +275,37 @@ static inline bool drm_mm_initialized(const struct drm_mm 
*mm)
        return mm->hole_stack.next;
 }
 
+/* stubs for external module compatiblity */
+void __drm_mm_node_init(struct drm_mm_node *node);
+void __drm_mm_node_fini(struct drm_mm_node *node);
+
+/**
+ * drm_mm_node_init - Prepare a node for use.
+ *
+ * Drivers should clear the drm_mm_node prior to use. If debug is enabled,
+ * the lifetime of the embedded drm_mm_node will be tracked.
+ */
+static inline void drm_mm_node_init(struct drm_mm_node *node)
+{
+#ifdef CONFIG_DRM_DEBUG_MM_OBJECTS
+       __drm_mm_node_init(node);
+#endif
+       memset(node, 0, sizeof(*node));
+}
+
+/**
+ * drm_mm_node_fini - Finalize a node for use.
+ *
+ * Drivers should indicate that the drm_mm_node is finished, and its
+ * state will then be verified by drm_mm.
+ */
+static inline void drm_mm_node_fini(struct drm_mm_node *node)
+{
+#ifdef CONFIG_DRM_DEBUG_MM_OBJECTS
+       __drm_mm_node_fini(node);
+#endif
+}
+
 /**
  * drm_mm_hole_follows - checks whether a hole follows this node
  * @node: drm_mm_node to check
-- 
2.20.1

_______________________________________________
dri-devel mailing list
dri-devel@lists.freedesktop.org
https://lists.freedesktop.org/mailman/listinfo/dri-devel

Reply via email to