drivers/infiniband/core/cq.c

   1 /*
   2  * Copyright (c) 2015 HGST, a Western Digital Company.
   3  *
   4  * This program is free software; you can redistribute it and/or modify it
   5  * under the terms and conditions of the GNU General Public License,
   6  * version 2, as published by the Free Software Foundation.
   7  *
   8  * This program is distributed in the hope it will be useful, but WITHOUT
   9  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  10  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  11  * more details.
  12  */
  13 #include <linux/module.h>
  14 #include <linux/err.h>
  15 #include <linux/slab.h>
  16 #include <rdma/ib_verbs.h>
  17
  18 /* # of WCs to poll for with a single call to ib_poll_cq */
  19 #define IB_POLL_BATCH                   16
  20
  21 /* # of WCs to iterate over before yielding */
  22 #define IB_POLL_BUDGET_IRQ              256
  23 #define IB_POLL_BUDGET_WORKQUEUE        65536
  24
  25 #define IB_POLL_FLAGS \
  26         (IB_CQ_NEXT_COMP | IB_CQ_REPORT_MISSED_EVENTS)
  27
  28 static int __ib_process_cq(struct ib_cq *cq, int budget)
  29 {
  30         int i, n, completed = 0;
  31
  32         while ((n = ib_poll_cq(cq, IB_POLL_BATCH, cq->wc)) > 0) {
  33                 for (i = 0; i < n; i++) {
  34                         struct ib_wc *wc = &cq->wc[i];
  35
  36                         if (wc->wr_cqe)
  37                                 wc->wr_cqe->done(cq, wc);
  38                         else
  39                                 WARN_ON_ONCE(wc->status == IB_WC_SUCCESS);
  40                 }
  41
  42                 completed += n;
  43
  44                 if (n != IB_POLL_BATCH ||
  45                     (budget != -1 && completed >= budget))
  46                         break;
  47         }
  48
  49         return completed;
  50 }
  51
  52 /**
  53  * ib_process_direct_cq - process a CQ in caller context
  54  * @cq:         CQ to process
  55  * @budget:     number of CQEs to poll for
  56  *
  57  * This function is used to process all outstanding CQ entries on a
  58  * %IB_POLL_DIRECT CQ.  It does not offload CQ processing to a different
  59  * context and does not ask for completion interrupts from the HCA.
  60  *
  61  * Note: for compatibility reasons -1 can be passed in %budget for unlimited
  62  * polling.  Do not use this feature in new code, it will be removed soon.
  63  */
  64 int ib_process_cq_direct(struct ib_cq *cq, int budget)
  65 {
  66         WARN_ON_ONCE(cq->poll_ctx != IB_POLL_DIRECT);
  67
  68         return __ib_process_cq(cq, budget);
  69 }
  70 EXPORT_SYMBOL(ib_process_cq_direct);
  71
  72 static void ib_cq_completion_direct(struct ib_cq *cq, void *private)
  73 {
  74         WARN_ONCE(1, "got unsolicited completion for CQ 0x%p\n", cq);
  75 }
  76
  77 static int ib_poll_handler(struct irq_poll *iop, int budget)
  78 {
  79         struct ib_cq *cq = container_of(iop, struct ib_cq, iop);
  80         int completed;
  81
  82         completed = __ib_process_cq(cq, budget);
  83         if (completed < budget) {
  84                 irq_poll_complete(&cq->iop);
  85                 if (ib_req_notify_cq(cq, IB_POLL_FLAGS) > 0)
  86                         irq_poll_sched(&cq->iop);
  87         }
  88
  89         return completed;
  90 }
  91
  92 static void ib_cq_completion_softirq(struct ib_cq *cq, void *private)
  93 {
  94         irq_poll_sched(&cq->iop);
  95 }
  96
  97 static void ib_cq_poll_work(struct work_struct *work)
  98 {
  99         struct ib_cq *cq = container_of(work, struct ib_cq, work);
 100         int completed;
 101
 102         completed = __ib_process_cq(cq, IB_POLL_BUDGET_WORKQUEUE);
 103         if (completed >= IB_POLL_BUDGET_WORKQUEUE ||
 104             ib_req_notify_cq(cq, IB_POLL_FLAGS) > 0)
 105                 queue_work(ib_comp_wq, &cq->work);
 106 }
 107
 108 static void ib_cq_completion_workqueue(struct ib_cq *cq, void *private)
 109 {
 110         queue_work(ib_comp_wq, &cq->work);
 111 }
 112
 113 /**
 114  * ib_alloc_cq - allocate a completion queue
 115  * @dev:                device to allocate the CQ for
 116  * @private:            driver private data, accessible from cq->cq_context
 117  * @nr_cqe:             number of CQEs to allocate
 118  * @comp_vector:        HCA completion vectors for this CQ
 119  * @poll_ctx:           context to poll the CQ from.
 120  *
 121  * This is the proper interface to allocate a CQ for in-kernel users. A
 122  * CQ allocated with this interface will automatically be polled from the
 123  * specified context.  The ULP needs must use wr->wr_cqe instead of wr->wr_id
 124  * to use this CQ abstraction.
 125  */
 126 struct ib_cq *ib_alloc_cq(struct ib_device *dev, void *private,
 127                 int nr_cqe, int comp_vector, enum ib_poll_context poll_ctx)
 128 {
 129         struct ib_cq_init_attr cq_attr = {
 130                 .cqe            = nr_cqe,
 131                 .comp_vector    = comp_vector,
 132         };
 133         struct ib_cq *cq;
 134         int ret = -ENOMEM;
 135
 136         cq = dev->create_cq(dev, &cq_attr, NULL, NULL);
 137         if (IS_ERR(cq))
 138                 return cq;
 139
 140         cq->device = dev;
 141         cq->uobject = NULL;
 142         cq->event_handler = NULL;
 143         cq->cq_context = private;
 144         cq->poll_ctx = poll_ctx;
 145         atomic_set(&cq->usecnt, 0);
 146
 147         cq->wc = kmalloc_array(IB_POLL_BATCH, sizeof(*cq->wc), GFP_KERNEL);
 148         if (!cq->wc)
 149                 goto out_destroy_cq;
 150
 151         switch (cq->poll_ctx) {
 152         case IB_POLL_DIRECT:
 153                 cq->comp_handler = ib_cq_completion_direct;
 154                 break;
 155         case IB_POLL_SOFTIRQ:
 156                 cq->comp_handler = ib_cq_completion_softirq;
 157
 158                 irq_poll_init(&cq->iop, IB_POLL_BUDGET_IRQ, ib_poll_handler);
 159                 ib_req_notify_cq(cq, IB_CQ_NEXT_COMP);
 160                 break;
 161         case IB_POLL_WORKQUEUE:
 162                 cq->comp_handler = ib_cq_completion_workqueue;
 163                 INIT_WORK(&cq->work, ib_cq_poll_work);
 164                 ib_req_notify_cq(cq, IB_CQ_NEXT_COMP);
 165                 break;
 166         default:
 167                 ret = -EINVAL;
 168                 goto out_free_wc;
 169         }
 170
 171         return cq;
 172
 173 out_free_wc:
 174         kfree(cq->wc);
 175 out_destroy_cq:
 176         cq->device->destroy_cq(cq);
 177         return ERR_PTR(ret);
 178 }
 179 EXPORT_SYMBOL(ib_alloc_cq);
 180
 181 /**
 182  * ib_free_cq - free a completion queue
 183  * @cq:         completion queue to free.
 184  */
 185 void ib_free_cq(struct ib_cq *cq)
 186 {
 187         int ret;
 188
 189         if (WARN_ON_ONCE(atomic_read(&cq->usecnt)))
 190                 return;
 191
 192         switch (cq->poll_ctx) {
 193         case IB_POLL_DIRECT:
 194                 break;
 195         case IB_POLL_SOFTIRQ:
 196                 irq_poll_disable(&cq->iop);
 197                 break;
 198         case IB_POLL_WORKQUEUE:
 199                 flush_work(&cq->work);
 200                 break;
 201         default:
 202                 WARN_ON_ONCE(1);
 203         }
 204
 205         kfree(cq->wc);
 206         ret = cq->device->destroy_cq(cq);
 207         WARN_ON_ONCE(ret);
 208 }
 209 EXPORT_SYMBOL(ib_free_cq);