firmware: arm_ffa: Stash ffa_device instead of notify_type in notifier_cb_info

Currently, we store the type of the notification in the notifier_cb_info
structure that is put into the hast list to identify if the notification
block is for the secure partition or the non secure VM.

In order to support framework notifications to reuse the hash list and
to avoid creating one for each time, we need store the ffa_device pointer
itself as the same notification ID in framework notifications can be
registered by multiple FF-A devices.

Tested-by: Viresh Kumar <viresh.kumar@linaro.org>
Message-Id: <20250217-ffa_updates-v3-15-bd1d9de615e7@arm.com>
Signed-off-by: Sudeep Holla <sudeep.holla@arm.com>

diff --git a/drivers/firmware/arm_ffa/driver.c b/drivers/firmware/arm_ffa/driver.c
index 0abbf08bc8d39c701947366bdf9c8fac8e64cf64..5863272f45d9c075d8e86d45a90ac6b1fb743c98 100644 (file)
--- a/drivers/firmware/arm_ffa/driver.c
+++ b/drivers/firmware/arm_ffa/driver.c
@@ -1116,9 +1116,9 @@ static int ffa_memory_lend(struct ffa_mem_ops_args *args)
 
 struct notifier_cb_info {
        struct hlist_node hnode;
+       struct ffa_device *dev;
        ffa_notifier_cb cb;
        void *cb_data;
-       enum notify_type type;
 };
 
 static int ffa_sched_recv_cb_update(u16 part_id, ffa_sched_recv_cb callback,
@@ -1187,17 +1187,18 @@ notifier_hash_node_get(u16 notify_id, enum notify_type type)
        struct notifier_cb_info *node;
 
        hash_for_each_possible(drv_info->notifier_hash, node, hnode, notify_id)
-               if (type == node->type)
+               if (type == ffa_notify_type_get(node->dev->vm_id))
                        return node;
 
        return NULL;
 }
 
 static int
-update_notifier_cb(int notify_id, enum notify_type type, ffa_notifier_cb cb,
+update_notifier_cb(struct ffa_device *dev, int notify_id, ffa_notifier_cb cb,
                   void *cb_data, bool is_registration)
 {
        struct notifier_cb_info *cb_info = NULL;
+       enum notify_type type = ffa_notify_type_get(dev->vm_id);
        bool cb_found;
 
        cb_info = notifier_hash_node_get(notify_id, type);
@@ -1211,7 +1212,7 @@ update_notifier_cb(int notify_id, enum notify_type type, ffa_notifier_cb cb,
                if (!cb_info)
                        return -ENOMEM;
 
-               cb_info->type = type;
+               cb_info->dev = dev;
                cb_info->cb = cb;
                cb_info->cb_data = cb_data;
 
@@ -1226,7 +1227,6 @@ update_notifier_cb(int notify_id, enum notify_type type, ffa_notifier_cb cb,
 static int ffa_notify_relinquish(struct ffa_device *dev, int notify_id)
 {
        int rc;
-       enum notify_type type = ffa_notify_type_get(dev->vm_id);
 
        if (ffa_notifications_disabled())
                return -EOPNOTSUPP;
@@ -1236,7 +1236,7 @@ static int ffa_notify_relinquish(struct ffa_device *dev, int notify_id)
 
        mutex_lock(&drv_info->notify_lock);
 
-       rc = update_notifier_cb(notify_id, type, NULL, NULL, false);
+       rc = update_notifier_cb(dev, notify_id, NULL, NULL, false);
        if (rc) {
                pr_err("Could not unregister notification callback\n");
                mutex_unlock(&drv_info->notify_lock);
@@ -1255,7 +1255,6 @@ static int ffa_notify_request(struct ffa_device *dev, bool is_per_vcpu,
 {
        int rc;
        u32 flags = 0;
-       enum notify_type type = ffa_notify_type_get(dev->vm_id);
 
        if (ffa_notifications_disabled())
                return -EOPNOTSUPP;
@@ -1274,7 +1273,7 @@ static int ffa_notify_request(struct ffa_device *dev, bool is_per_vcpu,
                return rc;
        }
 
-       rc = update_notifier_cb(notify_id, type, cb, cb_data, true);
+       rc = update_notifier_cb(dev, notify_id, cb, cb_data, true);
        if (rc) {
                pr_err("Failed to register callback for %d - %d\n",
                       notify_id, rc);