drm/i915: Add ability for tracking buffer objects per client

In order to show per client memory usage lets add some infrastructure
which enables tracking buffer objects owned by clients.

We add a per client list protected by a new per client lock and to support
delayed destruction (post client exit) we make tracked objects hold
references to the owning client.

Also, object memory region teardown is moved to the existing RCU free
callback to allow safe dereference from the fdinfo RCU read section.

Signed-off-by: Tvrtko Ursulin <tvrtko.ursulin@intel.com>
Reviewed-by: Aravind Iddamsetty <aravind.iddamsetty@intel.com>
Link: https://patchwork.freedesktop.org/patch/msgid/20231107101806.608990-1-tvrtko.ursulin@linux.intel.com

diff --git a/drivers/gpu/drm/i915/gem/i915_gem_object.c b/drivers/gpu/drm/i915/gem/i915_gem_object.c
index c26d8755582506af5b25d4f56f5dd3b521dc1c2b..25eeeb863209eece6a3123ba45f3a1c605f4bb78 100644 (file)
--- a/drivers/gpu/drm/i915/gem/i915_gem_object.c
+++ b/drivers/gpu/drm/i915/gem/i915_gem_object.c
@@ -106,6 +106,10 @@ void i915_gem_object_init(struct drm_i915_gem_object *obj,
 
        INIT_LIST_HEAD(&obj->mm.link);
 
+#ifdef CONFIG_PROC_FS
+       INIT_LIST_HEAD(&obj->client_link);
+#endif
+
        INIT_LIST_HEAD(&obj->lut_list);
        spin_lock_init(&obj->lut_lock);
 
@@ -293,6 +297,10 @@ void __i915_gem_free_object_rcu(struct rcu_head *head)
                container_of(head, typeof(*obj), rcu);
        struct drm_i915_private *i915 = to_i915(obj->base.dev);
 
+       /* We need to keep this alive for RCU read access from fdinfo. */
+       if (obj->mm.n_placements > 1)
+               kfree(obj->mm.placements);
+
        i915_gem_object_free(obj);
 
        GEM_BUG_ON(!atomic_read(&i915->mm.free_count));
@@ -389,9 +397,6 @@ void __i915_gem_free_object(struct drm_i915_gem_object *obj)
        if (obj->ops->release)
                obj->ops->release(obj);
 
-       if (obj->mm.n_placements > 1)
-               kfree(obj->mm.placements);
-
        if (obj->shares_resv_from)
                i915_vm_resv_put(obj->shares_resv_from);
 
@@ -442,6 +447,8 @@ static void i915_gem_free_object(struct drm_gem_object *gem_obj)
 
        GEM_BUG_ON(i915_gem_object_is_framebuffer(obj));
 
+       i915_drm_client_remove_object(obj);
+
        /*
         * Before we free the object, make sure any pure RCU-only
         * read-side critical sections are complete, e.g.

diff --git a/drivers/gpu/drm/i915/gem/i915_gem_object_types.h b/drivers/gpu/drm/i915/gem/i915_gem_object_types.h
index 2292404007c8152ab99bc709d7b43ff0b88bd355..0c5cdab278b69eb0a56c1c08695e7e311a614970 100644 (file)
--- a/drivers/gpu/drm/i915/gem/i915_gem_object_types.h
+++ b/drivers/gpu/drm/i915/gem/i915_gem_object_types.h
@@ -302,6 +302,18 @@ struct drm_i915_gem_object {
         */
        struct i915_address_space *shares_resv_from;
 
+#ifdef CONFIG_PROC_FS
+       /**
+        * @client: @i915_drm_client which created the object
+        */
+       struct i915_drm_client *client;
+
+       /**
+        * @client_link: Link into @i915_drm_client.objects_list
+        */
+       struct list_head client_link;
+#endif
+
        union {
                struct rcu_head rcu;
                struct llist_node freed;

diff --git a/drivers/gpu/drm/i915/i915_drm_client.c b/drivers/gpu/drm/i915/i915_drm_client.c
index 2a44b3876cb510a825d38fe04b7f1d5259622998..2e5e69edc0f9ae7b90923655fd41dfb846f727f7 100644 (file)
--- a/drivers/gpu/drm/i915/i915_drm_client.c
+++ b/drivers/gpu/drm/i915/i915_drm_client.c
@@ -28,6 +28,10 @@ struct i915_drm_client *i915_drm_client_alloc(void)
        kref_init(&client->kref);
        spin_lock_init(&client->ctx_lock);
        INIT_LIST_HEAD(&client->ctx_list);
+#ifdef CONFIG_PROC_FS
+       spin_lock_init(&client->objects_lock);
+       INIT_LIST_HEAD(&client->objects_list);
+#endif
 
        return client;
 }
@@ -108,4 +112,36 @@ void i915_drm_client_fdinfo(struct drm_printer *p, struct drm_file *file)
        for (i = 0; i < ARRAY_SIZE(uabi_class_names); i++)
                show_client_class(p, i915, file_priv->client, i);
 }
+
+void i915_drm_client_add_object(struct i915_drm_client *client,
+                               struct drm_i915_gem_object *obj)
+{
+       unsigned long flags;
+
+       GEM_WARN_ON(obj->client);
+       GEM_WARN_ON(!list_empty(&obj->client_link));
+
+       spin_lock_irqsave(&client->objects_lock, flags);
+       obj->client = i915_drm_client_get(client);
+       list_add_tail_rcu(&obj->client_link, &client->objects_list);
+       spin_unlock_irqrestore(&client->objects_lock, flags);
+}
+
+bool i915_drm_client_remove_object(struct drm_i915_gem_object *obj)
+{
+       struct i915_drm_client *client = fetch_and_zero(&obj->client);
+       unsigned long flags;
+
+       /* Object may not be associated with a client. */
+       if (!client)
+               return false;
+
+       spin_lock_irqsave(&client->objects_lock, flags);
+       list_del_rcu(&obj->client_link);
+       spin_unlock_irqrestore(&client->objects_lock, flags);
+
+       i915_drm_client_put(client);
+
+       return true;
+}
 #endif

diff --git a/drivers/gpu/drm/i915/i915_drm_client.h b/drivers/gpu/drm/i915/i915_drm_client.h
index 67816c912bca1377a65049bfeb3892688b46a8d1..5f58fdf7dcb8c59794d9d4870949a017c5c23b68 100644 (file)
--- a/drivers/gpu/drm/i915/i915_drm_client.h
+++ b/drivers/gpu/drm/i915/i915_drm_client.h
@@ -12,6 +12,9 @@
 
 #include <uapi/drm/i915_drm.h>
 
+#include "i915_file_private.h"
+#include "gem/i915_gem_object_types.h"
+
 #define I915_LAST_UABI_ENGINE_CLASS I915_ENGINE_CLASS_COMPUTE
 
 struct drm_file;
@@ -25,6 +28,20 @@ struct i915_drm_client {
        spinlock_t ctx_lock; /* For add/remove from ctx_list. */
        struct list_head ctx_list; /* List of contexts belonging to client. */
 
+#ifdef CONFIG_PROC_FS
+       /**
+        * @objects_lock: lock protecting @objects_list
+        */
+       spinlock_t objects_lock;
+
+       /**
+        * @objects_list: list of objects created by this client
+        *
+        * Protected by @objects_lock.
+        */
+       struct list_head objects_list;
+#endif
+
        /**
         * @past_runtime: Accumulation of pphwsp runtimes from closed contexts.
         */
@@ -49,4 +66,19 @@ struct i915_drm_client *i915_drm_client_alloc(void);
 
 void i915_drm_client_fdinfo(struct drm_printer *p, struct drm_file *file);
 
+#ifdef CONFIG_PROC_FS
+void i915_drm_client_add_object(struct i915_drm_client *client,
+                               struct drm_i915_gem_object *obj);
+bool i915_drm_client_remove_object(struct drm_i915_gem_object *obj);
+#else
+static inline void i915_drm_client_add_object(struct i915_drm_client *client,
+                                             struct drm_i915_gem_object *obj)
+{
+}
+
+static inline bool i915_drm_client_remove_object(struct drm_i915_gem_object *obj)
+{
+}
+#endif
+
 #endif /* !__I915_DRM_CLIENT_H__ */