In order to make retrieval of a vfio_ap_queue struct more
efficient when we already have a pointer to the ap_matrix_mdev to which the
queue's APQN is assigned, let's go ahead and add a link from the
ap_matrix_mdev struct to the vfio_ap_queue struct.

Signed-off-by: Tony Krowiak <akrow...@linux.ibm.com>
---
 drivers/s390/crypto/vfio_ap_ops.c     | 102 ++++++++++++++++++--------
 drivers/s390/crypto/vfio_ap_private.h |   2 +
 2 files changed, 72 insertions(+), 32 deletions(-)

diff --git a/drivers/s390/crypto/vfio_ap_ops.c 
b/drivers/s390/crypto/vfio_ap_ops.c
index add442977b9a..9a019b2b86f8 100644
--- a/drivers/s390/crypto/vfio_ap_ops.c
+++ b/drivers/s390/crypto/vfio_ap_ops.c
@@ -50,6 +50,19 @@ static struct vfio_ap_queue *vfio_ap_get_queue(unsigned long 
apqn)
        return q;
 }
 
+struct vfio_ap_queue *vfio_ap_get_mdev_queue(struct ap_matrix_mdev 
*matrix_mdev,
+                                            unsigned long apqn)
+{
+       struct vfio_ap_queue *q;
+
+       hash_for_each_possible(matrix_mdev->qtable, q, mdev_qnode, apqn) {
+               if (q && (q->apqn == apqn))
+                       return q;
+       }
+
+       return NULL;
+}
+
 /**
  * vfio_ap_wait_for_irqclear
  * @apqn: The AP Queue number
@@ -337,6 +350,7 @@ static int vfio_ap_mdev_create(struct kobject *kobj, struct 
mdev_device *mdev)
        matrix_mdev->mdev = mdev;
        vfio_ap_matrix_init(&matrix_dev->info, &matrix_mdev->matrix);
        vfio_ap_matrix_init(&matrix_dev->info, &matrix_mdev->shadow_apcb);
+       hash_init(matrix_mdev->qtable);
        mdev_set_drvdata(mdev, matrix_mdev);
        matrix_mdev->pqap_hook.hook = handle_pqap;
        matrix_mdev->pqap_hook.owner = THIS_MODULE;
@@ -639,7 +653,7 @@ static int vfio_ap_mdev_filter_matrix(struct ap_matrix_mdev 
*matrix_mdev,
                         * filter the APQI.
                         */
                        apqn = AP_MKQID(apid, apqi);
-                       if (!vfio_ap_get_queue(apqn)) {
+                       if (!vfio_ap_get_mdev_queue(matrix_mdev, apqn)) {
                                if (filter_apids)
                                        clear_bit_inv(apid, shadow_apcb->apm);
                                else
@@ -682,7 +696,6 @@ static bool vfio_ap_mdev_config_shadow_apcb(struct 
ap_matrix_mdev *matrix_mdev)
        vfio_ap_matrix_init(&matrix_dev->info, &shadow_apcb);
        napm = bitmap_weight(matrix_mdev->matrix.apm, AP_DEVICES);
        naqm = bitmap_weight(matrix_mdev->matrix.aqm, AP_DOMAINS);
-
        /*
         * If there are no APIDs or no APQIs assigned to the matrix mdev,
         * then no APQNs shall be assigned to the guest CRYCB.
@@ -694,6 +707,7 @@ static bool vfio_ap_mdev_config_shadow_apcb(struct 
ap_matrix_mdev *matrix_mdev)
                 */
                napm = vfio_ap_mdev_filter_matrix(matrix_mdev, &shadow_apcb,
                                                  true);
+
                /*
                 * If there are no APQNs that can be assigned to the guest's
                 * CRYCB after filtering, then try filtering the APQIs.
@@ -742,56 +756,75 @@ enum qlink_type {
        UNLINK_APQI,
 };
 
+static void vfio_ap_mdev_link_queue(struct ap_matrix_mdev *matrix_mdev,
+                                   unsigned long apid, unsigned long apqi)
+{
+       struct vfio_ap_queue *q;
+
+       q = vfio_ap_get_queue(AP_MKQID(apid, apqi));
+       if (q) {
+               q->matrix_mdev = matrix_mdev;
+               hash_add(matrix_mdev->qtable,
+                        &q->mdev_qnode, q->apqn);
+       }
+}
+
+static void vfio_ap_mdev_unlink_queue(unsigned long apid, unsigned long apqi)
+{
+       struct vfio_ap_queue *q;
+
+       q = vfio_ap_get_queue(AP_MKQID(apid, apqi));
+       if (q) {
+               q->matrix_mdev = NULL;
+               hash_del(&q->mdev_qnode);
+       }
+}
+
 /**
  * vfio_ap_mdev_link_queues
  *
  * @matrix_mdev: The matrix mdev to link.
- * @type:       The type of link.
+ * @type:       The type of @qlink_id.
  * @qlink_id:   The APID or APQI of the queues to link.
  *
- * Sets the link from the queues with the specified @qlink_id (i.e., APID or
- * APQI) to @matrix_mdev:
- *     qlink_id == LINK_APID: Link @matrix_mdev to the queues with the
- *                 specified APID>
- *     qlink_id == UNLINK_APID: Unlink @matrix_mdev from the queues with the
- *                 specified APID>
- *     qlink_id == LINK_APQI: Link @matrix_mdev to the queues with the
- *                 specified APQI>
- *     qlink_id == UNLINK_APQI: Unlink @matrix_mdev from the queues with the
- *                 specified APQI>
+ * Sets or clears the links between the queues with the specified @qlink_id
+ * and the @matrix_mdev:
+ *     @type == LINK_APID: Set the links between the @matrix_mdev and the
+ *                         queues with the specified @qlink_id (APID)
+ *     @type == LINK_APQI: Set the links between the @matrix_mdev and the
+ *                         queues with the specified @qlink_id (APQI)
+ *     @type == UNLINK_APID: Clear the links between the @matrix_mdev and the
+ *                           queues with the specified @qlink_id (APID)
+ *     @type == UNLINK_APQI: Clear the links between the @matrix_mdev and the
+ *                           queues with the specified @qlink_id (APQI)
  */
 static void vfio_ap_mdev_link_queues(struct ap_matrix_mdev *matrix_mdev,
                                     enum qlink_type type,
                                     unsigned long qlink_id)
 {
        unsigned long id;
-       struct vfio_ap_queue *q;
+
 
        switch (type) {
        case LINK_APID:
+               for_each_set_bit_inv(id, matrix_mdev->matrix.aqm,
+                                    matrix_mdev->matrix.aqm_max + 1)
+                       vfio_ap_mdev_link_queue(matrix_mdev, qlink_id, id);
+               break;
        case UNLINK_APID:
                for_each_set_bit_inv(id, matrix_mdev->matrix.aqm,
-                                    matrix_mdev->matrix.aqm_max + 1) {
-                       q = vfio_ap_get_queue(AP_MKQID(qlink_id, id));
-                       if (q) {
-                               if (type == LINK_APID)
-                                       q->matrix_mdev = matrix_mdev;
-                               else
-                                       q->matrix_mdev = NULL;
-                       }
-               }
+                                    matrix_mdev->matrix.aqm_max + 1)
+                       vfio_ap_mdev_unlink_queue(qlink_id, id);
+               break;
+       case LINK_APQI:
+               for_each_set_bit_inv(id, matrix_mdev->matrix.apm,
+                                    matrix_mdev->matrix.apm_max + 1)
+                       vfio_ap_mdev_link_queue(matrix_mdev, id, qlink_id);
                break;
        default:
                for_each_set_bit_inv(id, matrix_mdev->matrix.apm,
-                                    matrix_mdev->matrix.apm_max + 1) {
-                       q = vfio_ap_get_queue(AP_MKQID(id, qlink_id));
-                       if (q) {
-                               if (type == LINK_APQI)
-                                       q->matrix_mdev = matrix_mdev;
-                               else
-                                       q->matrix_mdev = NULL;
-                       }
-               }
+                                    matrix_mdev->matrix.apm_max + 1)
+                       vfio_ap_mdev_unlink_queue(id, qlink_id);
                break;
        }
 }
@@ -1612,6 +1645,7 @@ static void vfio_ap_queue_link_mdev(struct vfio_ap_queue 
*q)
                if (test_bit_inv(apid, matrix_mdev->matrix.apm) &&
                    test_bit_inv(apqi, matrix_mdev->matrix.aqm)) {
                        q->matrix_mdev = matrix_mdev;
+                       hash_add(matrix_mdev->qtable, &q->mdev_qnode, q->apqn);
                        break;
                }
        }
@@ -1647,6 +1681,10 @@ void vfio_ap_mdev_remove_queue(struct ap_queue *queue)
        apqi = AP_QID_QUEUE(q->apqn);
        vfio_ap_mdev_reset_queue(apid, apqi, 1);
        vfio_ap_irq_disable(q);
+
+       if (q->matrix_mdev)
+               hash_del(&q->mdev_qnode);
+
        kfree(q);
        mutex_unlock(&matrix_dev->lock);
 }
diff --git a/drivers/s390/crypto/vfio_ap_private.h 
b/drivers/s390/crypto/vfio_ap_private.h
index 8e24a073166b..055bce6d45db 100644
--- a/drivers/s390/crypto/vfio_ap_private.h
+++ b/drivers/s390/crypto/vfio_ap_private.h
@@ -89,6 +89,7 @@ struct ap_matrix_mdev {
        struct kvm *kvm;
        struct kvm_s390_module_hook pqap_hook;
        struct mdev_device *mdev;
+       DECLARE_HASHTABLE(qtable, 8);
 };
 
 extern int vfio_ap_mdev_register(void);
@@ -100,6 +101,7 @@ struct vfio_ap_queue {
        int     apqn;
 #define VFIO_AP_ISC_INVALID 0xff
        unsigned char saved_isc;
+       struct hlist_node mdev_qnode;
 };
 
 int vfio_ap_mdev_probe_queue(struct ap_queue *queue);
-- 
2.21.1

Reply via email to