3 files changed, 20 insertions, 11 deletions

diff --git a/drivers/gpu/drm/nouveau/core/core/client.c b/drivers/gpu/drm/nouveau/core/core/client.c
index 10598dede9e9..68bf06768123 100644
--- a/drivers/gpu/drm/nouveau/core/core/client.c
+++ b/drivers/gpu/drm/nouveau/core/core/client.c
@@ -132,12 +132,12 @@ nvkm_client_notify_new(struct nouveau_client *client,
                 if (ret == 0) {
                         client->notify[index] = notify;
                         notify->client = client;
-                        return 0;
+                        return index;
                 }
         }
 
         kfree(notify);
-        return 0;
+        return ret;
 }
 
 static int
diff --git a/drivers/gpu/drm/nouveau/core/include/core/client.h b/drivers/gpu/drm/nouveau/core/include/core/client.h
index 4fc6ab12382d..1794a05205d8 100644
--- a/drivers/gpu/drm/nouveau/core/include/core/client.h
+++ b/drivers/gpu/drm/nouveau/core/include/core/client.h
@@ -14,7 +14,7 @@ struct nouveau_client {
         void *data;
 
         int (*ntfy)(const void *, u32, const void *, u32);
-        struct nvkm_client_notify *notify[8];
+        struct nvkm_client_notify *notify[16];
 };
 
 static inline struct nouveau_client *
diff --git a/drivers/gpu/drm/nouveau/nvif/notify.c b/drivers/gpu/drm/nouveau/nvif/notify.c
index 7c06123a559c..7e03cdd17827 100644
--- a/drivers/gpu/drm/nouveau/nvif/notify.c
+++ b/drivers/gpu/drm/nouveau/nvif/notify.c
@@ -87,12 +87,25 @@ nvif_notify_get(struct nvif_notify *notify)
         return 0;
 }
 
+static inline int
+nvif_notify_func(struct nvif_notify *notify, bool keep)
+{
+        int ret = notify->func(notify);
+        if (ret == NVIF_NOTIFY_KEEP ||
+            !test_and_clear_bit(NVKM_NOTIFY_USER, &notify->flags)) {
+                if (!keep)
+                        atomic_dec(&notify->putcnt);
+                else
+                        nvif_notify_get_(notify);
+        }
+        return ret;
+}
+
 static void
 nvif_notify_work(struct work_struct *work)
 {
         struct nvif_notify *notify = container_of(work, typeof(*notify), work);
-        if (notify->func(notify) == NVIF_NOTIFY_KEEP)
-                nvif_notify_get_(notify);
+        nvif_notify_func(notify, true);
 }
 
 int
@@ -113,19 +126,15 @@ nvif_notify(const void *header, u32 length, const void *data, u32 size)
         if (!WARN_ON(notify == NULL)) {
                 struct nvif_client *client = nvif_client(notify->object);
                 if (!WARN_ON(notify->size != size)) {
+                        atomic_inc(&notify->putcnt);
                         if (test_bit(NVIF_NOTIFY_WORK, &notify->flags)) {
-                                atomic_inc(&notify->putcnt);
                                 memcpy((void *)notify->data, data, size);
                                 schedule_work(&notify->work);
                                 return NVIF_NOTIFY_DROP;
                         }
                         notify->data = data;
-                        ret = notify->func(notify);
+                        ret = nvif_notify_func(notify, client->driver->keep);
                         notify->data = NULL;
-                        if (ret != NVIF_NOTIFY_DROP && client->driver->keep) {
-                                atomic_inc(&notify->putcnt);
-                                nvif_notify_get_(notify);
-                        }
                 }
         }