blob: 008791bbdfb322bab68ecff4ca3f66bce04a6b67 [file] [log] [blame]
Christoph Hellwig71102302016-07-06 21:55:52 +09001/*
2 * NVMe over Fabrics RDMA host code.
3 * Copyright (c) 2015-2016 HGST, a Western Digital Company.
4 *
5 * This program is free software; you can redistribute it and/or modify it
6 * under the terms and conditions of the GNU General Public License,
7 * version 2, as published by the Free Software Foundation.
8 *
9 * This program is distributed in the hope it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
12 * more details.
13 */
14#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
Christoph Hellwig71102302016-07-06 21:55:52 +090015#include <linux/module.h>
16#include <linux/init.h>
17#include <linux/slab.h>
18#include <linux/err.h>
19#include <linux/string.h>
Christoph Hellwig71102302016-07-06 21:55:52 +090020#include <linux/atomic.h>
21#include <linux/blk-mq.h>
Sagi Grimberg0b366582017-07-13 11:09:44 +030022#include <linux/blk-mq-rdma.h>
Christoph Hellwig71102302016-07-06 21:55:52 +090023#include <linux/types.h>
24#include <linux/list.h>
25#include <linux/mutex.h>
26#include <linux/scatterlist.h>
27#include <linux/nvme.h>
Christoph Hellwig71102302016-07-06 21:55:52 +090028#include <asm/unaligned.h>
29
30#include <rdma/ib_verbs.h>
31#include <rdma/rdma_cm.h>
Christoph Hellwig71102302016-07-06 21:55:52 +090032#include <linux/nvme-rdma.h>
33
34#include "nvme.h"
35#include "fabrics.h"
36
37
Sagi Grimberg782d8202017-03-21 16:32:38 +020038#define NVME_RDMA_CONNECT_TIMEOUT_MS 3000 /* 3 second */
Christoph Hellwig71102302016-07-06 21:55:52 +090039
Christoph Hellwig71102302016-07-06 21:55:52 +090040#define NVME_RDMA_MAX_SEGMENTS 256
41
42#define NVME_RDMA_MAX_INLINE_SEGMENTS 1
43
Christoph Hellwig71102302016-07-06 21:55:52 +090044struct nvme_rdma_device {
Max Gurtovoyf87c89a2017-10-23 12:59:27 +030045 struct ib_device *dev;
46 struct ib_pd *pd;
Christoph Hellwig71102302016-07-06 21:55:52 +090047 struct kref ref;
48 struct list_head entry;
49};
50
51struct nvme_rdma_qe {
52 struct ib_cqe cqe;
53 void *data;
54 u64 dma;
55};
56
57struct nvme_rdma_queue;
58struct nvme_rdma_request {
Christoph Hellwigd49187e2016-11-10 07:32:33 -080059 struct nvme_request req;
Christoph Hellwig71102302016-07-06 21:55:52 +090060 struct ib_mr *mr;
61 struct nvme_rdma_qe sqe;
62 struct ib_sge sge[1 + NVME_RDMA_MAX_INLINE_SEGMENTS];
63 u32 num_sge;
64 int nents;
65 bool inline_data;
Christoph Hellwig71102302016-07-06 21:55:52 +090066 struct ib_reg_wr reg_wr;
67 struct ib_cqe reg_cqe;
68 struct nvme_rdma_queue *queue;
69 struct sg_table sg_table;
70 struct scatterlist first_sgl[];
71};
72
73enum nvme_rdma_queue_flags {
Sagi Grimberg5013e982017-10-11 15:29:12 +030074 NVME_RDMA_Q_ALLOCATED = 0,
75 NVME_RDMA_Q_LIVE = 1,
Christoph Hellwig71102302016-07-06 21:55:52 +090076};
77
78struct nvme_rdma_queue {
79 struct nvme_rdma_qe *rsp_ring;
Marta Rybczynska5e599d72017-06-06 13:27:21 +020080 atomic_t sig_count;
Christoph Hellwig71102302016-07-06 21:55:52 +090081 int queue_size;
82 size_t cmnd_capsule_len;
83 struct nvme_rdma_ctrl *ctrl;
84 struct nvme_rdma_device *device;
85 struct ib_cq *ib_cq;
86 struct ib_qp *qp;
87
88 unsigned long flags;
89 struct rdma_cm_id *cm_id;
90 int cm_error;
91 struct completion cm_done;
92};
93
94struct nvme_rdma_ctrl {
Christoph Hellwig71102302016-07-06 21:55:52 +090095 /* read only in the hot path */
96 struct nvme_rdma_queue *queues;
Christoph Hellwig71102302016-07-06 21:55:52 +090097
98 /* other member variables */
Christoph Hellwig71102302016-07-06 21:55:52 +090099 struct blk_mq_tag_set tag_set;
Christoph Hellwig71102302016-07-06 21:55:52 +0900100 struct work_struct err_work;
101
102 struct nvme_rdma_qe async_event_sqe;
103
Christoph Hellwig71102302016-07-06 21:55:52 +0900104 struct delayed_work reconnect_work;
105
106 struct list_head list;
107
108 struct blk_mq_tag_set admin_tag_set;
109 struct nvme_rdma_device *device;
110
Christoph Hellwig71102302016-07-06 21:55:52 +0900111 u32 max_fr_pages;
112
Sagi Grimberg0928f9b2017-02-05 21:49:32 +0200113 struct sockaddr_storage addr;
114 struct sockaddr_storage src_addr;
Christoph Hellwig71102302016-07-06 21:55:52 +0900115
116 struct nvme_ctrl ctrl;
117};
118
119static inline struct nvme_rdma_ctrl *to_rdma_ctrl(struct nvme_ctrl *ctrl)
120{
121 return container_of(ctrl, struct nvme_rdma_ctrl, ctrl);
122}
123
124static LIST_HEAD(device_list);
125static DEFINE_MUTEX(device_list_mutex);
126
127static LIST_HEAD(nvme_rdma_ctrl_list);
128static DEFINE_MUTEX(nvme_rdma_ctrl_mutex);
129
Christoph Hellwig71102302016-07-06 21:55:52 +0900130/*
131 * Disabling this option makes small I/O goes faster, but is fundamentally
132 * unsafe. With it turned off we will have to register a global rkey that
133 * allows read and write access to all physical memory.
134 */
135static bool register_always = true;
136module_param(register_always, bool, 0444);
137MODULE_PARM_DESC(register_always,
138 "Use memory registration even for contiguous memory regions");
139
140static int nvme_rdma_cm_handler(struct rdma_cm_id *cm_id,
141 struct rdma_cm_event *event);
142static void nvme_rdma_recv_done(struct ib_cq *cq, struct ib_wc *wc);
Christoph Hellwig71102302016-07-06 21:55:52 +0900143
Sagi Grimberg90af3512017-07-10 09:22:28 +0300144static const struct blk_mq_ops nvme_rdma_mq_ops;
145static const struct blk_mq_ops nvme_rdma_admin_mq_ops;
146
Christoph Hellwig71102302016-07-06 21:55:52 +0900147/* XXX: really should move to a generic header sooner or later.. */
148static inline void put_unaligned_le24(u32 val, u8 *p)
149{
150 *p++ = val;
151 *p++ = val >> 8;
152 *p++ = val >> 16;
153}
154
155static inline int nvme_rdma_queue_idx(struct nvme_rdma_queue *queue)
156{
157 return queue - queue->ctrl->queues;
158}
159
160static inline size_t nvme_rdma_inline_data_size(struct nvme_rdma_queue *queue)
161{
162 return queue->cmnd_capsule_len - sizeof(struct nvme_command);
163}
164
165static void nvme_rdma_free_qe(struct ib_device *ibdev, struct nvme_rdma_qe *qe,
166 size_t capsule_size, enum dma_data_direction dir)
167{
168 ib_dma_unmap_single(ibdev, qe->dma, capsule_size, dir);
169 kfree(qe->data);
170}
171
172static int nvme_rdma_alloc_qe(struct ib_device *ibdev, struct nvme_rdma_qe *qe,
173 size_t capsule_size, enum dma_data_direction dir)
174{
175 qe->data = kzalloc(capsule_size, GFP_KERNEL);
176 if (!qe->data)
177 return -ENOMEM;
178
179 qe->dma = ib_dma_map_single(ibdev, qe->data, capsule_size, dir);
180 if (ib_dma_mapping_error(ibdev, qe->dma)) {
181 kfree(qe->data);
182 return -ENOMEM;
183 }
184
185 return 0;
186}
187
188static void nvme_rdma_free_ring(struct ib_device *ibdev,
189 struct nvme_rdma_qe *ring, size_t ib_queue_size,
190 size_t capsule_size, enum dma_data_direction dir)
191{
192 int i;
193
194 for (i = 0; i < ib_queue_size; i++)
195 nvme_rdma_free_qe(ibdev, &ring[i], capsule_size, dir);
196 kfree(ring);
197}
198
199static struct nvme_rdma_qe *nvme_rdma_alloc_ring(struct ib_device *ibdev,
200 size_t ib_queue_size, size_t capsule_size,
201 enum dma_data_direction dir)
202{
203 struct nvme_rdma_qe *ring;
204 int i;
205
206 ring = kcalloc(ib_queue_size, sizeof(struct nvme_rdma_qe), GFP_KERNEL);
207 if (!ring)
208 return NULL;
209
210 for (i = 0; i < ib_queue_size; i++) {
211 if (nvme_rdma_alloc_qe(ibdev, &ring[i], capsule_size, dir))
212 goto out_free_ring;
213 }
214
215 return ring;
216
217out_free_ring:
218 nvme_rdma_free_ring(ibdev, ring, i, capsule_size, dir);
219 return NULL;
220}
221
222static void nvme_rdma_qp_event(struct ib_event *event, void *context)
223{
Max Gurtovoy27a4bee2016-11-23 11:38:48 +0200224 pr_debug("QP event %s (%d)\n",
225 ib_event_msg(event->event), event->event);
226
Christoph Hellwig71102302016-07-06 21:55:52 +0900227}
228
229static int nvme_rdma_wait_for_cm(struct nvme_rdma_queue *queue)
230{
231 wait_for_completion_interruptible_timeout(&queue->cm_done,
232 msecs_to_jiffies(NVME_RDMA_CONNECT_TIMEOUT_MS) + 1);
233 return queue->cm_error;
234}
235
236static int nvme_rdma_create_qp(struct nvme_rdma_queue *queue, const int factor)
237{
238 struct nvme_rdma_device *dev = queue->device;
239 struct ib_qp_init_attr init_attr;
240 int ret;
241
242 memset(&init_attr, 0, sizeof(init_attr));
243 init_attr.event_handler = nvme_rdma_qp_event;
244 /* +1 for drain */
245 init_attr.cap.max_send_wr = factor * queue->queue_size + 1;
246 /* +1 for drain */
247 init_attr.cap.max_recv_wr = queue->queue_size + 1;
248 init_attr.cap.max_recv_sge = 1;
249 init_attr.cap.max_send_sge = 1 + NVME_RDMA_MAX_INLINE_SEGMENTS;
250 init_attr.sq_sig_type = IB_SIGNAL_REQ_WR;
251 init_attr.qp_type = IB_QPT_RC;
252 init_attr.send_cq = queue->ib_cq;
253 init_attr.recv_cq = queue->ib_cq;
254
255 ret = rdma_create_qp(queue->cm_id, dev->pd, &init_attr);
256
257 queue->qp = queue->cm_id->qp;
258 return ret;
259}
260
261static int nvme_rdma_reinit_request(void *data, struct request *rq)
262{
263 struct nvme_rdma_ctrl *ctrl = data;
264 struct nvme_rdma_device *dev = ctrl->device;
265 struct nvme_rdma_request *req = blk_mq_rq_to_pdu(rq);
266 int ret = 0;
267
Sagi Grimberg0fc176d2017-10-11 15:29:09 +0300268 if (WARN_ON_ONCE(!req->mr))
269 return 0;
270
Christoph Hellwig71102302016-07-06 21:55:52 +0900271 ib_dereg_mr(req->mr);
272
273 req->mr = ib_alloc_mr(dev->pd, IB_MR_TYPE_MEM_REG,
274 ctrl->max_fr_pages);
275 if (IS_ERR(req->mr)) {
Christoph Hellwig71102302016-07-06 21:55:52 +0900276 ret = PTR_ERR(req->mr);
Wei Yongjun458a9632016-07-12 11:06:17 +0000277 req->mr = NULL;
Colin Ian King1bda18d2016-09-05 16:24:38 +0100278 goto out;
Christoph Hellwig71102302016-07-06 21:55:52 +0900279 }
280
Sagi Grimbergf5b7b552016-08-24 12:25:56 +0300281 req->mr->need_inval = false;
Christoph Hellwig71102302016-07-06 21:55:52 +0900282
283out:
284 return ret;
285}
286
Christoph Hellwig385475e2017-06-13 09:15:19 +0200287static void nvme_rdma_exit_request(struct blk_mq_tag_set *set,
288 struct request *rq, unsigned int hctx_idx)
Christoph Hellwig71102302016-07-06 21:55:52 +0900289{
Christoph Hellwig385475e2017-06-13 09:15:19 +0200290 struct nvme_rdma_ctrl *ctrl = set->driver_data;
Christoph Hellwig71102302016-07-06 21:55:52 +0900291 struct nvme_rdma_request *req = blk_mq_rq_to_pdu(rq);
Christoph Hellwig385475e2017-06-13 09:15:19 +0200292 int queue_idx = (set == &ctrl->tag_set) ? hctx_idx + 1 : 0;
Christoph Hellwig71102302016-07-06 21:55:52 +0900293 struct nvme_rdma_queue *queue = &ctrl->queues[queue_idx];
294 struct nvme_rdma_device *dev = queue->device;
295
296 if (req->mr)
297 ib_dereg_mr(req->mr);
298
299 nvme_rdma_free_qe(dev->dev, &req->sqe, sizeof(struct nvme_command),
300 DMA_TO_DEVICE);
301}
302
Christoph Hellwig385475e2017-06-13 09:15:19 +0200303static int nvme_rdma_init_request(struct blk_mq_tag_set *set,
304 struct request *rq, unsigned int hctx_idx,
305 unsigned int numa_node)
Christoph Hellwig71102302016-07-06 21:55:52 +0900306{
Christoph Hellwig385475e2017-06-13 09:15:19 +0200307 struct nvme_rdma_ctrl *ctrl = set->driver_data;
Christoph Hellwig71102302016-07-06 21:55:52 +0900308 struct nvme_rdma_request *req = blk_mq_rq_to_pdu(rq);
Christoph Hellwig385475e2017-06-13 09:15:19 +0200309 int queue_idx = (set == &ctrl->tag_set) ? hctx_idx + 1 : 0;
Christoph Hellwig71102302016-07-06 21:55:52 +0900310 struct nvme_rdma_queue *queue = &ctrl->queues[queue_idx];
311 struct nvme_rdma_device *dev = queue->device;
312 struct ib_device *ibdev = dev->dev;
313 int ret;
314
Christoph Hellwig71102302016-07-06 21:55:52 +0900315 ret = nvme_rdma_alloc_qe(ibdev, &req->sqe, sizeof(struct nvme_command),
316 DMA_TO_DEVICE);
317 if (ret)
318 return ret;
319
320 req->mr = ib_alloc_mr(dev->pd, IB_MR_TYPE_MEM_REG,
321 ctrl->max_fr_pages);
322 if (IS_ERR(req->mr)) {
323 ret = PTR_ERR(req->mr);
324 goto out_free_qe;
325 }
326
327 req->queue = queue;
328
329 return 0;
330
331out_free_qe:
332 nvme_rdma_free_qe(dev->dev, &req->sqe, sizeof(struct nvme_command),
333 DMA_TO_DEVICE);
334 return -ENOMEM;
335}
336
Christoph Hellwig71102302016-07-06 21:55:52 +0900337static int nvme_rdma_init_hctx(struct blk_mq_hw_ctx *hctx, void *data,
338 unsigned int hctx_idx)
339{
340 struct nvme_rdma_ctrl *ctrl = data;
341 struct nvme_rdma_queue *queue = &ctrl->queues[hctx_idx + 1];
342
Sagi Grimbergd858e5f2017-04-24 10:58:29 +0300343 BUG_ON(hctx_idx >= ctrl->ctrl.queue_count);
Christoph Hellwig71102302016-07-06 21:55:52 +0900344
345 hctx->driver_data = queue;
346 return 0;
347}
348
349static int nvme_rdma_init_admin_hctx(struct blk_mq_hw_ctx *hctx, void *data,
350 unsigned int hctx_idx)
351{
352 struct nvme_rdma_ctrl *ctrl = data;
353 struct nvme_rdma_queue *queue = &ctrl->queues[0];
354
355 BUG_ON(hctx_idx != 0);
356
357 hctx->driver_data = queue;
358 return 0;
359}
360
361static void nvme_rdma_free_dev(struct kref *ref)
362{
363 struct nvme_rdma_device *ndev =
364 container_of(ref, struct nvme_rdma_device, ref);
365
366 mutex_lock(&device_list_mutex);
367 list_del(&ndev->entry);
368 mutex_unlock(&device_list_mutex);
369
Christoph Hellwig71102302016-07-06 21:55:52 +0900370 ib_dealloc_pd(ndev->pd);
Christoph Hellwig71102302016-07-06 21:55:52 +0900371 kfree(ndev);
372}
373
374static void nvme_rdma_dev_put(struct nvme_rdma_device *dev)
375{
376 kref_put(&dev->ref, nvme_rdma_free_dev);
377}
378
379static int nvme_rdma_dev_get(struct nvme_rdma_device *dev)
380{
381 return kref_get_unless_zero(&dev->ref);
382}
383
384static struct nvme_rdma_device *
385nvme_rdma_find_get_device(struct rdma_cm_id *cm_id)
386{
387 struct nvme_rdma_device *ndev;
388
389 mutex_lock(&device_list_mutex);
390 list_for_each_entry(ndev, &device_list, entry) {
391 if (ndev->dev->node_guid == cm_id->device->node_guid &&
392 nvme_rdma_dev_get(ndev))
393 goto out_unlock;
394 }
395
396 ndev = kzalloc(sizeof(*ndev), GFP_KERNEL);
397 if (!ndev)
398 goto out_err;
399
400 ndev->dev = cm_id->device;
401 kref_init(&ndev->ref);
402
Christoph Hellwig11975e02016-09-05 12:56:20 +0200403 ndev->pd = ib_alloc_pd(ndev->dev,
404 register_always ? 0 : IB_PD_UNSAFE_GLOBAL_RKEY);
Christoph Hellwig71102302016-07-06 21:55:52 +0900405 if (IS_ERR(ndev->pd))
406 goto out_free_dev;
407
Christoph Hellwig71102302016-07-06 21:55:52 +0900408 if (!(ndev->dev->attrs.device_cap_flags &
409 IB_DEVICE_MEM_MGT_EXTENSIONS)) {
410 dev_err(&ndev->dev->dev,
411 "Memory registrations not supported.\n");
Christoph Hellwig11975e02016-09-05 12:56:20 +0200412 goto out_free_pd;
Christoph Hellwig71102302016-07-06 21:55:52 +0900413 }
414
415 list_add(&ndev->entry, &device_list);
416out_unlock:
417 mutex_unlock(&device_list_mutex);
418 return ndev;
419
Christoph Hellwig71102302016-07-06 21:55:52 +0900420out_free_pd:
421 ib_dealloc_pd(ndev->pd);
422out_free_dev:
423 kfree(ndev);
424out_err:
425 mutex_unlock(&device_list_mutex);
426 return NULL;
427}
428
429static void nvme_rdma_destroy_queue_ib(struct nvme_rdma_queue *queue)
430{
Sagi Grimberg0c5b43b2017-10-11 15:29:08 +0300431 struct nvme_rdma_device *dev = queue->device;
432 struct ib_device *ibdev = dev->dev;
Christoph Hellwig71102302016-07-06 21:55:52 +0900433
434 rdma_destroy_qp(queue->cm_id);
435 ib_free_cq(queue->ib_cq);
436
437 nvme_rdma_free_ring(ibdev, queue->rsp_ring, queue->queue_size,
438 sizeof(struct nvme_completion), DMA_FROM_DEVICE);
439
440 nvme_rdma_dev_put(dev);
441}
442
Sagi Grimbergca6e95b2017-05-04 13:33:09 +0300443static int nvme_rdma_create_queue_ib(struct nvme_rdma_queue *queue)
Christoph Hellwig71102302016-07-06 21:55:52 +0900444{
Sagi Grimbergca6e95b2017-05-04 13:33:09 +0300445 struct ib_device *ibdev;
Christoph Hellwig71102302016-07-06 21:55:52 +0900446 const int send_wr_factor = 3; /* MR, SEND, INV */
447 const int cq_factor = send_wr_factor + 1; /* + RECV */
448 int comp_vector, idx = nvme_rdma_queue_idx(queue);
Christoph Hellwig71102302016-07-06 21:55:52 +0900449 int ret;
450
Sagi Grimbergca6e95b2017-05-04 13:33:09 +0300451 queue->device = nvme_rdma_find_get_device(queue->cm_id);
452 if (!queue->device) {
453 dev_err(queue->cm_id->device->dev.parent,
454 "no client data found!\n");
455 return -ECONNREFUSED;
456 }
457 ibdev = queue->device->dev;
Christoph Hellwig71102302016-07-06 21:55:52 +0900458
459 /*
Sagi Grimberg0b366582017-07-13 11:09:44 +0300460 * Spread I/O queues completion vectors according their queue index.
461 * Admin queues can always go on completion vector 0.
Christoph Hellwig71102302016-07-06 21:55:52 +0900462 */
Sagi Grimberg0b366582017-07-13 11:09:44 +0300463 comp_vector = idx == 0 ? idx : idx - 1;
Christoph Hellwig71102302016-07-06 21:55:52 +0900464
465 /* +1 for ib_stop_cq */
Sagi Grimbergca6e95b2017-05-04 13:33:09 +0300466 queue->ib_cq = ib_alloc_cq(ibdev, queue,
467 cq_factor * queue->queue_size + 1,
468 comp_vector, IB_POLL_SOFTIRQ);
Christoph Hellwig71102302016-07-06 21:55:52 +0900469 if (IS_ERR(queue->ib_cq)) {
470 ret = PTR_ERR(queue->ib_cq);
Sagi Grimbergca6e95b2017-05-04 13:33:09 +0300471 goto out_put_dev;
Christoph Hellwig71102302016-07-06 21:55:52 +0900472 }
473
474 ret = nvme_rdma_create_qp(queue, send_wr_factor);
475 if (ret)
476 goto out_destroy_ib_cq;
477
478 queue->rsp_ring = nvme_rdma_alloc_ring(ibdev, queue->queue_size,
479 sizeof(struct nvme_completion), DMA_FROM_DEVICE);
480 if (!queue->rsp_ring) {
481 ret = -ENOMEM;
482 goto out_destroy_qp;
483 }
484
485 return 0;
486
487out_destroy_qp:
Max Gurtovoy1f61def2017-11-06 16:18:51 +0200488 rdma_destroy_qp(queue->cm_id);
Christoph Hellwig71102302016-07-06 21:55:52 +0900489out_destroy_ib_cq:
490 ib_free_cq(queue->ib_cq);
Sagi Grimbergca6e95b2017-05-04 13:33:09 +0300491out_put_dev:
492 nvme_rdma_dev_put(queue->device);
Christoph Hellwig71102302016-07-06 21:55:52 +0900493 return ret;
494}
495
Sagi Grimberg41e8cfa2017-07-10 09:22:36 +0300496static int nvme_rdma_alloc_queue(struct nvme_rdma_ctrl *ctrl,
Christoph Hellwig71102302016-07-06 21:55:52 +0900497 int idx, size_t queue_size)
498{
499 struct nvme_rdma_queue *queue;
Max Gurtovoy8f4e8da2017-02-19 20:08:03 +0200500 struct sockaddr *src_addr = NULL;
Christoph Hellwig71102302016-07-06 21:55:52 +0900501 int ret;
502
503 queue = &ctrl->queues[idx];
504 queue->ctrl = ctrl;
505 init_completion(&queue->cm_done);
506
507 if (idx > 0)
508 queue->cmnd_capsule_len = ctrl->ctrl.ioccsz * 16;
509 else
510 queue->cmnd_capsule_len = sizeof(struct nvme_command);
511
512 queue->queue_size = queue_size;
Marta Rybczynska5e599d72017-06-06 13:27:21 +0200513 atomic_set(&queue->sig_count, 0);
Christoph Hellwig71102302016-07-06 21:55:52 +0900514
515 queue->cm_id = rdma_create_id(&init_net, nvme_rdma_cm_handler, queue,
516 RDMA_PS_TCP, IB_QPT_RC);
517 if (IS_ERR(queue->cm_id)) {
518 dev_info(ctrl->ctrl.device,
519 "failed to create CM ID: %ld\n", PTR_ERR(queue->cm_id));
520 return PTR_ERR(queue->cm_id);
521 }
522
Max Gurtovoy8f4e8da2017-02-19 20:08:03 +0200523 if (ctrl->ctrl.opts->mask & NVMF_OPT_HOST_TRADDR)
Sagi Grimberg0928f9b2017-02-05 21:49:32 +0200524 src_addr = (struct sockaddr *)&ctrl->src_addr;
Max Gurtovoy8f4e8da2017-02-19 20:08:03 +0200525
Sagi Grimberg0928f9b2017-02-05 21:49:32 +0200526 queue->cm_error = -ETIMEDOUT;
527 ret = rdma_resolve_addr(queue->cm_id, src_addr,
528 (struct sockaddr *)&ctrl->addr,
Christoph Hellwig71102302016-07-06 21:55:52 +0900529 NVME_RDMA_CONNECT_TIMEOUT_MS);
530 if (ret) {
531 dev_info(ctrl->ctrl.device,
532 "rdma_resolve_addr failed (%d).\n", ret);
533 goto out_destroy_cm_id;
534 }
535
536 ret = nvme_rdma_wait_for_cm(queue);
537 if (ret) {
538 dev_info(ctrl->ctrl.device,
Sagi Grimbergd8bfcee2017-10-11 15:29:07 +0300539 "rdma connection establishment failed (%d)\n", ret);
Christoph Hellwig71102302016-07-06 21:55:52 +0900540 goto out_destroy_cm_id;
541 }
542
Sagi Grimberg5013e982017-10-11 15:29:12 +0300543 set_bit(NVME_RDMA_Q_ALLOCATED, &queue->flags);
Christoph Hellwig71102302016-07-06 21:55:52 +0900544
545 return 0;
546
547out_destroy_cm_id:
548 rdma_destroy_id(queue->cm_id);
549 return ret;
550}
551
552static void nvme_rdma_stop_queue(struct nvme_rdma_queue *queue)
553{
Sagi Grimberga57bd542017-08-28 21:41:10 +0200554 if (!test_and_clear_bit(NVME_RDMA_Q_LIVE, &queue->flags))
555 return;
556
Christoph Hellwig71102302016-07-06 21:55:52 +0900557 rdma_disconnect(queue->cm_id);
558 ib_drain_qp(queue->qp);
559}
560
561static void nvme_rdma_free_queue(struct nvme_rdma_queue *queue)
562{
Sagi Grimberg5013e982017-10-11 15:29:12 +0300563 if (!test_and_clear_bit(NVME_RDMA_Q_ALLOCATED, &queue->flags))
Christoph Hellwig71102302016-07-06 21:55:52 +0900564 return;
Sagi Grimberga57bd542017-08-28 21:41:10 +0200565
566 nvme_rdma_destroy_queue_ib(queue);
567 rdma_destroy_id(queue->cm_id);
Christoph Hellwig71102302016-07-06 21:55:52 +0900568}
569
570static void nvme_rdma_free_io_queues(struct nvme_rdma_ctrl *ctrl)
571{
572 int i;
573
Sagi Grimbergd858e5f2017-04-24 10:58:29 +0300574 for (i = 1; i < ctrl->ctrl.queue_count; i++)
Sagi Grimberga57bd542017-08-28 21:41:10 +0200575 nvme_rdma_free_queue(&ctrl->queues[i]);
Christoph Hellwig71102302016-07-06 21:55:52 +0900576}
577
Sagi Grimberga57bd542017-08-28 21:41:10 +0200578static void nvme_rdma_stop_io_queues(struct nvme_rdma_ctrl *ctrl)
579{
580 int i;
581
582 for (i = 1; i < ctrl->ctrl.queue_count; i++)
583 nvme_rdma_stop_queue(&ctrl->queues[i]);
Christoph Hellwig71102302016-07-06 21:55:52 +0900584}
585
Sagi Grimberg68e16fc2017-07-10 09:22:37 +0300586static int nvme_rdma_start_queue(struct nvme_rdma_ctrl *ctrl, int idx)
587{
588 int ret;
589
590 if (idx)
591 ret = nvmf_connect_io_queue(&ctrl->ctrl, idx);
592 else
593 ret = nvmf_connect_admin_queue(&ctrl->ctrl);
594
595 if (!ret)
596 set_bit(NVME_RDMA_Q_LIVE, &ctrl->queues[idx].flags);
597 else
598 dev_info(ctrl->ctrl.device,
599 "failed to connect queue: %d ret=%d\n", idx, ret);
600 return ret;
601}
602
603static int nvme_rdma_start_io_queues(struct nvme_rdma_ctrl *ctrl)
Christoph Hellwig71102302016-07-06 21:55:52 +0900604{
605 int i, ret = 0;
606
Sagi Grimbergd858e5f2017-04-24 10:58:29 +0300607 for (i = 1; i < ctrl->ctrl.queue_count; i++) {
Sagi Grimberg68e16fc2017-07-10 09:22:37 +0300608 ret = nvme_rdma_start_queue(ctrl, i);
609 if (ret)
Sagi Grimberga57bd542017-08-28 21:41:10 +0200610 goto out_stop_queues;
Christoph Hellwig71102302016-07-06 21:55:52 +0900611 }
612
Steve Wisec8dbc372016-11-08 09:16:02 -0800613 return 0;
614
Sagi Grimberga57bd542017-08-28 21:41:10 +0200615out_stop_queues:
Sagi Grimberg68e16fc2017-07-10 09:22:37 +0300616 for (i--; i >= 1; i--)
617 nvme_rdma_stop_queue(&ctrl->queues[i]);
Christoph Hellwig71102302016-07-06 21:55:52 +0900618 return ret;
619}
620
Sagi Grimberg41e8cfa2017-07-10 09:22:36 +0300621static int nvme_rdma_alloc_io_queues(struct nvme_rdma_ctrl *ctrl)
Christoph Hellwig71102302016-07-06 21:55:52 +0900622{
Sagi Grimbergc248c642017-03-09 13:26:07 +0200623 struct nvmf_ctrl_options *opts = ctrl->ctrl.opts;
Sagi Grimberg0b366582017-07-13 11:09:44 +0300624 struct ib_device *ibdev = ctrl->device->dev;
Sagi Grimbergc248c642017-03-09 13:26:07 +0200625 unsigned int nr_io_queues;
Christoph Hellwig71102302016-07-06 21:55:52 +0900626 int i, ret;
627
Sagi Grimbergc248c642017-03-09 13:26:07 +0200628 nr_io_queues = min(opts->nr_io_queues, num_online_cpus());
Sagi Grimberg0b366582017-07-13 11:09:44 +0300629
630 /*
631 * we map queues according to the device irq vectors for
632 * optimal locality so we don't need more queues than
633 * completion vectors.
634 */
635 nr_io_queues = min_t(unsigned int, nr_io_queues,
636 ibdev->num_comp_vectors);
637
Sagi Grimbergc248c642017-03-09 13:26:07 +0200638 ret = nvme_set_queue_count(&ctrl->ctrl, &nr_io_queues);
639 if (ret)
640 return ret;
641
Sagi Grimbergd858e5f2017-04-24 10:58:29 +0300642 ctrl->ctrl.queue_count = nr_io_queues + 1;
643 if (ctrl->ctrl.queue_count < 2)
Sagi Grimbergc248c642017-03-09 13:26:07 +0200644 return 0;
645
646 dev_info(ctrl->ctrl.device,
647 "creating %d I/O queues.\n", nr_io_queues);
648
Sagi Grimbergd858e5f2017-04-24 10:58:29 +0300649 for (i = 1; i < ctrl->ctrl.queue_count; i++) {
Sagi Grimberg41e8cfa2017-07-10 09:22:36 +0300650 ret = nvme_rdma_alloc_queue(ctrl, i,
651 ctrl->ctrl.sqsize + 1);
652 if (ret)
Christoph Hellwig71102302016-07-06 21:55:52 +0900653 goto out_free_queues;
Christoph Hellwig71102302016-07-06 21:55:52 +0900654 }
655
656 return 0;
657
658out_free_queues:
Steve Wisef361e5a2016-09-02 09:01:27 -0700659 for (i--; i >= 1; i--)
Sagi Grimberga57bd542017-08-28 21:41:10 +0200660 nvme_rdma_free_queue(&ctrl->queues[i]);
Christoph Hellwig71102302016-07-06 21:55:52 +0900661
662 return ret;
663}
664
Sagi Grimberg60070c72017-10-11 15:29:06 +0300665static void nvme_rdma_free_tagset(struct nvme_ctrl *nctrl,
666 struct blk_mq_tag_set *set)
Sagi Grimbergb28a3082017-07-10 09:22:30 +0300667{
668 struct nvme_rdma_ctrl *ctrl = to_rdma_ctrl(nctrl);
Sagi Grimbergb28a3082017-07-10 09:22:30 +0300669
670 blk_mq_free_tag_set(set);
671 nvme_rdma_dev_put(ctrl->device);
672}
673
674static struct blk_mq_tag_set *nvme_rdma_alloc_tagset(struct nvme_ctrl *nctrl,
675 bool admin)
676{
677 struct nvme_rdma_ctrl *ctrl = to_rdma_ctrl(nctrl);
678 struct blk_mq_tag_set *set;
679 int ret;
680
681 if (admin) {
682 set = &ctrl->admin_tag_set;
683 memset(set, 0, sizeof(*set));
684 set->ops = &nvme_rdma_admin_mq_ops;
Keith Busch38dabe22017-11-07 15:13:10 -0700685 set->queue_depth = NVME_AQ_MQ_TAG_DEPTH;
Sagi Grimbergb28a3082017-07-10 09:22:30 +0300686 set->reserved_tags = 2; /* connect + keep-alive */
687 set->numa_node = NUMA_NO_NODE;
688 set->cmd_size = sizeof(struct nvme_rdma_request) +
689 SG_CHUNK_SIZE * sizeof(struct scatterlist);
690 set->driver_data = ctrl;
691 set->nr_hw_queues = 1;
692 set->timeout = ADMIN_TIMEOUT;
Israel Rukshin94f29d42017-10-18 12:38:24 +0000693 set->flags = BLK_MQ_F_NO_SCHED;
Sagi Grimbergb28a3082017-07-10 09:22:30 +0300694 } else {
695 set = &ctrl->tag_set;
696 memset(set, 0, sizeof(*set));
697 set->ops = &nvme_rdma_mq_ops;
698 set->queue_depth = nctrl->opts->queue_size;
699 set->reserved_tags = 1; /* fabric connect */
700 set->numa_node = NUMA_NO_NODE;
701 set->flags = BLK_MQ_F_SHOULD_MERGE;
702 set->cmd_size = sizeof(struct nvme_rdma_request) +
703 SG_CHUNK_SIZE * sizeof(struct scatterlist);
704 set->driver_data = ctrl;
705 set->nr_hw_queues = nctrl->queue_count - 1;
706 set->timeout = NVME_IO_TIMEOUT;
707 }
708
709 ret = blk_mq_alloc_tag_set(set);
710 if (ret)
711 goto out;
712
713 /*
714 * We need a reference on the device as long as the tag_set is alive,
715 * as the MRs in the request structures need a valid ib_device.
716 */
717 ret = nvme_rdma_dev_get(ctrl->device);
718 if (!ret) {
719 ret = -EINVAL;
720 goto out_free_tagset;
721 }
722
723 return set;
724
725out_free_tagset:
726 blk_mq_free_tag_set(set);
727out:
728 return ERR_PTR(ret);
729}
730
Sagi Grimberg3f02fff2017-07-10 09:22:32 +0300731static void nvme_rdma_destroy_admin_queue(struct nvme_rdma_ctrl *ctrl,
732 bool remove)
Christoph Hellwig71102302016-07-06 21:55:52 +0900733{
734 nvme_rdma_free_qe(ctrl->queues[0].device->dev, &ctrl->async_event_sqe,
735 sizeof(struct nvme_command), DMA_TO_DEVICE);
Sagi Grimberga57bd542017-08-28 21:41:10 +0200736 nvme_rdma_stop_queue(&ctrl->queues[0]);
Sagi Grimberg3f02fff2017-07-10 09:22:32 +0300737 if (remove) {
738 blk_cleanup_queue(ctrl->ctrl.admin_q);
Sagi Grimberg60070c72017-10-11 15:29:06 +0300739 nvme_rdma_free_tagset(&ctrl->ctrl, ctrl->ctrl.admin_tagset);
Sagi Grimberg3f02fff2017-07-10 09:22:32 +0300740 }
Sagi Grimberga57bd542017-08-28 21:41:10 +0200741 nvme_rdma_free_queue(&ctrl->queues[0]);
Christoph Hellwig71102302016-07-06 21:55:52 +0900742}
743
Sagi Grimberg3f02fff2017-07-10 09:22:32 +0300744static int nvme_rdma_configure_admin_queue(struct nvme_rdma_ctrl *ctrl,
745 bool new)
Sagi Grimberg90af3512017-07-10 09:22:28 +0300746{
747 int error;
748
Sagi Grimberg41e8cfa2017-07-10 09:22:36 +0300749 error = nvme_rdma_alloc_queue(ctrl, 0, NVME_AQ_DEPTH);
Sagi Grimberg90af3512017-07-10 09:22:28 +0300750 if (error)
751 return error;
752
753 ctrl->device = ctrl->queues[0].device;
754
Sagi Grimberg90af3512017-07-10 09:22:28 +0300755 ctrl->max_fr_pages = min_t(u32, NVME_RDMA_MAX_SEGMENTS,
756 ctrl->device->dev->attrs.max_fast_reg_page_list_len);
757
Sagi Grimberg3f02fff2017-07-10 09:22:32 +0300758 if (new) {
759 ctrl->ctrl.admin_tagset = nvme_rdma_alloc_tagset(&ctrl->ctrl, true);
760 if (IS_ERR(ctrl->ctrl.admin_tagset))
761 goto out_free_queue;
Sagi Grimberg90af3512017-07-10 09:22:28 +0300762
Sagi Grimberg3f02fff2017-07-10 09:22:32 +0300763 ctrl->ctrl.admin_q = blk_mq_init_queue(&ctrl->admin_tag_set);
764 if (IS_ERR(ctrl->ctrl.admin_q)) {
765 error = PTR_ERR(ctrl->ctrl.admin_q);
766 goto out_free_tagset;
767 }
768 } else {
Sagi Grimberg31b84462017-10-11 12:53:07 +0300769 error = nvme_reinit_tagset(&ctrl->ctrl, ctrl->ctrl.admin_tagset);
Sagi Grimberg3f02fff2017-07-10 09:22:32 +0300770 if (error)
771 goto out_free_queue;
Sagi Grimberg90af3512017-07-10 09:22:28 +0300772 }
773
Sagi Grimberg68e16fc2017-07-10 09:22:37 +0300774 error = nvme_rdma_start_queue(ctrl, 0);
Sagi Grimberg90af3512017-07-10 09:22:28 +0300775 if (error)
776 goto out_cleanup_queue;
777
Sagi Grimberg09fdc232017-07-10 09:22:39 +0300778 error = ctrl->ctrl.ops->reg_read64(&ctrl->ctrl, NVME_REG_CAP,
Sagi Grimberg90af3512017-07-10 09:22:28 +0300779 &ctrl->ctrl.cap);
780 if (error) {
781 dev_err(ctrl->ctrl.device,
782 "prop_get NVME_REG_CAP failed\n");
783 goto out_cleanup_queue;
784 }
785
786 ctrl->ctrl.sqsize =
787 min_t(int, NVME_CAP_MQES(ctrl->ctrl.cap), ctrl->ctrl.sqsize);
788
789 error = nvme_enable_ctrl(&ctrl->ctrl, ctrl->ctrl.cap);
790 if (error)
791 goto out_cleanup_queue;
792
793 ctrl->ctrl.max_hw_sectors =
Linus Torvalds126e76f2017-09-09 12:49:01 -0700794 (ctrl->max_fr_pages - 1) << (ilog2(SZ_4K) - 9);
Sagi Grimberg90af3512017-07-10 09:22:28 +0300795
796 error = nvme_init_identify(&ctrl->ctrl);
797 if (error)
798 goto out_cleanup_queue;
799
800 error = nvme_rdma_alloc_qe(ctrl->queues[0].device->dev,
801 &ctrl->async_event_sqe, sizeof(struct nvme_command),
802 DMA_TO_DEVICE);
803 if (error)
804 goto out_cleanup_queue;
805
806 return 0;
807
808out_cleanup_queue:
Sagi Grimberg3f02fff2017-07-10 09:22:32 +0300809 if (new)
810 blk_cleanup_queue(ctrl->ctrl.admin_q);
Sagi Grimberg90af3512017-07-10 09:22:28 +0300811out_free_tagset:
Sagi Grimberg3f02fff2017-07-10 09:22:32 +0300812 if (new)
Sagi Grimberg60070c72017-10-11 15:29:06 +0300813 nvme_rdma_free_tagset(&ctrl->ctrl, ctrl->ctrl.admin_tagset);
Sagi Grimberg90af3512017-07-10 09:22:28 +0300814out_free_queue:
815 nvme_rdma_free_queue(&ctrl->queues[0]);
816 return error;
817}
818
Sagi Grimberga57bd542017-08-28 21:41:10 +0200819static void nvme_rdma_destroy_io_queues(struct nvme_rdma_ctrl *ctrl,
820 bool remove)
821{
822 nvme_rdma_stop_io_queues(ctrl);
823 if (remove) {
824 blk_cleanup_queue(ctrl->ctrl.connect_q);
Sagi Grimberg60070c72017-10-11 15:29:06 +0300825 nvme_rdma_free_tagset(&ctrl->ctrl, ctrl->ctrl.tagset);
Sagi Grimberga57bd542017-08-28 21:41:10 +0200826 }
827 nvme_rdma_free_io_queues(ctrl);
828}
829
830static int nvme_rdma_configure_io_queues(struct nvme_rdma_ctrl *ctrl, bool new)
831{
832 int ret;
833
Sagi Grimberg41e8cfa2017-07-10 09:22:36 +0300834 ret = nvme_rdma_alloc_io_queues(ctrl);
Sagi Grimberga57bd542017-08-28 21:41:10 +0200835 if (ret)
836 return ret;
837
838 if (new) {
839 ctrl->ctrl.tagset = nvme_rdma_alloc_tagset(&ctrl->ctrl, false);
840 if (IS_ERR(ctrl->ctrl.tagset))
841 goto out_free_io_queues;
842
843 ctrl->ctrl.connect_q = blk_mq_init_queue(&ctrl->tag_set);
844 if (IS_ERR(ctrl->ctrl.connect_q)) {
845 ret = PTR_ERR(ctrl->ctrl.connect_q);
846 goto out_free_tag_set;
847 }
848 } else {
Sagi Grimberg31b84462017-10-11 12:53:07 +0300849 ret = nvme_reinit_tagset(&ctrl->ctrl, ctrl->ctrl.tagset);
Sagi Grimberga57bd542017-08-28 21:41:10 +0200850 if (ret)
851 goto out_free_io_queues;
852
853 blk_mq_update_nr_hw_queues(&ctrl->tag_set,
854 ctrl->ctrl.queue_count - 1);
855 }
856
Sagi Grimberg68e16fc2017-07-10 09:22:37 +0300857 ret = nvme_rdma_start_io_queues(ctrl);
Sagi Grimberga57bd542017-08-28 21:41:10 +0200858 if (ret)
859 goto out_cleanup_connect_q;
860
861 return 0;
862
863out_cleanup_connect_q:
864 if (new)
865 blk_cleanup_queue(ctrl->ctrl.connect_q);
866out_free_tag_set:
867 if (new)
Sagi Grimberg60070c72017-10-11 15:29:06 +0300868 nvme_rdma_free_tagset(&ctrl->ctrl, ctrl->ctrl.tagset);
Sagi Grimberga57bd542017-08-28 21:41:10 +0200869out_free_io_queues:
870 nvme_rdma_free_io_queues(ctrl);
871 return ret;
Christoph Hellwig71102302016-07-06 21:55:52 +0900872}
873
874static void nvme_rdma_free_ctrl(struct nvme_ctrl *nctrl)
875{
876 struct nvme_rdma_ctrl *ctrl = to_rdma_ctrl(nctrl);
877
878 if (list_empty(&ctrl->list))
879 goto free_ctrl;
880
881 mutex_lock(&nvme_rdma_ctrl_mutex);
882 list_del(&ctrl->list);
883 mutex_unlock(&nvme_rdma_ctrl_mutex);
884
Christoph Hellwig71102302016-07-06 21:55:52 +0900885 kfree(ctrl->queues);
886 nvmf_free_options(nctrl->opts);
887free_ctrl:
888 kfree(ctrl);
889}
890
Sagi Grimbergfd8563c2017-03-18 20:58:29 +0200891static void nvme_rdma_reconnect_or_remove(struct nvme_rdma_ctrl *ctrl)
892{
893 /* If we are resetting/deleting then do nothing */
894 if (ctrl->ctrl.state != NVME_CTRL_RECONNECTING) {
895 WARN_ON_ONCE(ctrl->ctrl.state == NVME_CTRL_NEW ||
896 ctrl->ctrl.state == NVME_CTRL_LIVE);
897 return;
898 }
899
900 if (nvmf_should_reconnect(&ctrl->ctrl)) {
901 dev_info(ctrl->ctrl.device, "Reconnecting in %d seconds...\n",
902 ctrl->ctrl.opts->reconnect_delay);
Sagi Grimberg9a6327d2017-06-07 20:31:55 +0200903 queue_delayed_work(nvme_wq, &ctrl->reconnect_work,
Sagi Grimbergfd8563c2017-03-18 20:58:29 +0200904 ctrl->ctrl.opts->reconnect_delay * HZ);
905 } else {
906 dev_info(ctrl->ctrl.device, "Removing controller...\n");
Sagi Grimberg12fa1302017-10-29 14:21:01 +0200907 nvme_delete_ctrl(&ctrl->ctrl);
Sagi Grimbergfd8563c2017-03-18 20:58:29 +0200908 }
909}
910
Christoph Hellwig71102302016-07-06 21:55:52 +0900911static void nvme_rdma_reconnect_ctrl_work(struct work_struct *work)
912{
913 struct nvme_rdma_ctrl *ctrl = container_of(to_delayed_work(work),
914 struct nvme_rdma_ctrl, reconnect_work);
915 bool changed;
916 int ret;
917
Sagi Grimbergfdf9dfa2017-05-04 13:33:15 +0300918 ++ctrl->ctrl.nr_reconnects;
Sagi Grimbergfd8563c2017-03-18 20:58:29 +0200919
Sagi Grimberg31fdf182017-08-28 21:40:06 +0200920 ret = nvme_rdma_configure_admin_queue(ctrl, false);
Christoph Hellwig71102302016-07-06 21:55:52 +0900921 if (ret)
Sagi Grimberge818a5b2017-06-05 20:35:56 +0300922 goto requeue;
Christoph Hellwig71102302016-07-06 21:55:52 +0900923
Sagi Grimbergd858e5f2017-04-24 10:58:29 +0300924 if (ctrl->ctrl.queue_count > 1) {
Sagi Grimberga57bd542017-08-28 21:41:10 +0200925 ret = nvme_rdma_configure_io_queues(ctrl, false);
Christoph Hellwig71102302016-07-06 21:55:52 +0900926 if (ret)
Sagi Grimberg5e1fe612017-10-11 15:29:11 +0300927 goto destroy_admin;
Christoph Hellwig71102302016-07-06 21:55:52 +0900928 }
929
930 changed = nvme_change_ctrl_state(&ctrl->ctrl, NVME_CTRL_LIVE);
Sagi Grimberg0a960afd2017-09-21 17:01:37 +0300931 if (!changed) {
932 /* state change failure is ok if we're in DELETING state */
933 WARN_ON_ONCE(ctrl->ctrl.state != NVME_CTRL_DELETING);
934 return;
935 }
936
Sagi Grimbergd09f2b42017-07-02 10:56:43 +0300937 nvme_start_ctrl(&ctrl->ctrl);
Christoph Hellwig71102302016-07-06 21:55:52 +0900938
Sagi Grimberg5e1fe612017-10-11 15:29:11 +0300939 dev_info(ctrl->ctrl.device, "Successfully reconnected (%d attempts)\n",
940 ctrl->ctrl.nr_reconnects);
941
942 ctrl->ctrl.nr_reconnects = 0;
Christoph Hellwig71102302016-07-06 21:55:52 +0900943
944 return;
945
Sagi Grimberg5e1fe612017-10-11 15:29:11 +0300946destroy_admin:
947 nvme_rdma_destroy_admin_queue(ctrl, false);
Christoph Hellwig71102302016-07-06 21:55:52 +0900948requeue:
Sagi Grimbergfd8563c2017-03-18 20:58:29 +0200949 dev_info(ctrl->ctrl.device, "Failed reconnect attempt %d\n",
Sagi Grimbergfdf9dfa2017-05-04 13:33:15 +0300950 ctrl->ctrl.nr_reconnects);
Sagi Grimbergfd8563c2017-03-18 20:58:29 +0200951 nvme_rdma_reconnect_or_remove(ctrl);
Christoph Hellwig71102302016-07-06 21:55:52 +0900952}
953
954static void nvme_rdma_error_recovery_work(struct work_struct *work)
955{
956 struct nvme_rdma_ctrl *ctrl = container_of(work,
957 struct nvme_rdma_ctrl, err_work);
958
Sagi Grimberge4d753d2017-09-21 17:01:38 +0300959 nvme_stop_keep_alive(&ctrl->ctrl);
Sagi Grimberge89ca582016-09-02 09:01:54 -0700960
Sagi Grimberg148b4e72017-07-10 09:22:35 +0300961 if (ctrl->ctrl.queue_count > 1) {
Christoph Hellwig71102302016-07-06 21:55:52 +0900962 nvme_stop_queues(&ctrl->ctrl);
Christoph Hellwig71102302016-07-06 21:55:52 +0900963 blk_mq_tagset_busy_iter(&ctrl->tag_set,
964 nvme_cancel_request, &ctrl->ctrl);
Sagi Grimberg5e1fe612017-10-11 15:29:11 +0300965 nvme_rdma_destroy_io_queues(ctrl, false);
966 }
967
968 blk_mq_quiesce_queue(ctrl->ctrl.admin_q);
Christoph Hellwig71102302016-07-06 21:55:52 +0900969 blk_mq_tagset_busy_iter(&ctrl->admin_tag_set,
970 nvme_cancel_request, &ctrl->ctrl);
Sagi Grimberg5e1fe612017-10-11 15:29:11 +0300971 nvme_rdma_destroy_admin_queue(ctrl, false);
Christoph Hellwig71102302016-07-06 21:55:52 +0900972
Sagi Grimberge818a5b2017-06-05 20:35:56 +0300973 /*
974 * queues are not a live anymore, so restart the queues to fail fast
975 * new IO
976 */
Sagi Grimbergfb051332017-07-02 15:33:32 +0300977 blk_mq_unquiesce_queue(ctrl->ctrl.admin_q);
Sagi Grimberge818a5b2017-06-05 20:35:56 +0300978 nvme_start_queues(&ctrl->ctrl);
979
Sagi Grimbergfd8563c2017-03-18 20:58:29 +0200980 nvme_rdma_reconnect_or_remove(ctrl);
Christoph Hellwig71102302016-07-06 21:55:52 +0900981}
982
983static void nvme_rdma_error_recovery(struct nvme_rdma_ctrl *ctrl)
984{
985 if (!nvme_change_ctrl_state(&ctrl->ctrl, NVME_CTRL_RECONNECTING))
986 return;
987
Sagi Grimberg9a6327d2017-06-07 20:31:55 +0200988 queue_work(nvme_wq, &ctrl->err_work);
Christoph Hellwig71102302016-07-06 21:55:52 +0900989}
990
991static void nvme_rdma_wr_error(struct ib_cq *cq, struct ib_wc *wc,
992 const char *op)
993{
994 struct nvme_rdma_queue *queue = cq->cq_context;
995 struct nvme_rdma_ctrl *ctrl = queue->ctrl;
996
997 if (ctrl->ctrl.state == NVME_CTRL_LIVE)
998 dev_info(ctrl->ctrl.device,
999 "%s for CQE 0x%p failed with status %s (%d)\n",
1000 op, wc->wr_cqe,
1001 ib_wc_status_msg(wc->status), wc->status);
1002 nvme_rdma_error_recovery(ctrl);
1003}
1004
1005static void nvme_rdma_memreg_done(struct ib_cq *cq, struct ib_wc *wc)
1006{
1007 if (unlikely(wc->status != IB_WC_SUCCESS))
1008 nvme_rdma_wr_error(cq, wc, "MEMREG");
1009}
1010
1011static void nvme_rdma_inv_rkey_done(struct ib_cq *cq, struct ib_wc *wc)
1012{
1013 if (unlikely(wc->status != IB_WC_SUCCESS))
1014 nvme_rdma_wr_error(cq, wc, "LOCAL_INV");
1015}
1016
1017static int nvme_rdma_inv_rkey(struct nvme_rdma_queue *queue,
1018 struct nvme_rdma_request *req)
1019{
1020 struct ib_send_wr *bad_wr;
1021 struct ib_send_wr wr = {
1022 .opcode = IB_WR_LOCAL_INV,
1023 .next = NULL,
1024 .num_sge = 0,
1025 .send_flags = 0,
1026 .ex.invalidate_rkey = req->mr->rkey,
1027 };
1028
1029 req->reg_cqe.done = nvme_rdma_inv_rkey_done;
1030 wr.wr_cqe = &req->reg_cqe;
1031
1032 return ib_post_send(queue->qp, &wr, &bad_wr);
1033}
1034
1035static void nvme_rdma_unmap_data(struct nvme_rdma_queue *queue,
1036 struct request *rq)
1037{
1038 struct nvme_rdma_request *req = blk_mq_rq_to_pdu(rq);
1039 struct nvme_rdma_ctrl *ctrl = queue->ctrl;
1040 struct nvme_rdma_device *dev = queue->device;
1041 struct ib_device *ibdev = dev->dev;
1042 int res;
1043
1044 if (!blk_rq_bytes(rq))
1045 return;
1046
Sagi Grimberg60a51882017-10-11 15:29:10 +03001047 if (req->mr->need_inval && test_bit(NVME_RDMA_Q_LIVE, &req->queue->flags)) {
Christoph Hellwig71102302016-07-06 21:55:52 +09001048 res = nvme_rdma_inv_rkey(queue, req);
Max Gurtovoya7b7c7a2017-08-14 15:29:26 +03001049 if (unlikely(res < 0)) {
Christoph Hellwig71102302016-07-06 21:55:52 +09001050 dev_err(ctrl->ctrl.device,
1051 "Queueing INV WR for rkey %#x failed (%d)\n",
1052 req->mr->rkey, res);
1053 nvme_rdma_error_recovery(queue->ctrl);
1054 }
1055 }
1056
1057 ib_dma_unmap_sg(ibdev, req->sg_table.sgl,
1058 req->nents, rq_data_dir(rq) ==
1059 WRITE ? DMA_TO_DEVICE : DMA_FROM_DEVICE);
1060
1061 nvme_cleanup_cmd(rq);
1062 sg_free_table_chained(&req->sg_table, true);
1063}
1064
1065static int nvme_rdma_set_sg_null(struct nvme_command *c)
1066{
1067 struct nvme_keyed_sgl_desc *sg = &c->common.dptr.ksgl;
1068
1069 sg->addr = 0;
1070 put_unaligned_le24(0, sg->length);
1071 put_unaligned_le32(0, sg->key);
1072 sg->type = NVME_KEY_SGL_FMT_DATA_DESC << 4;
1073 return 0;
1074}
1075
1076static int nvme_rdma_map_sg_inline(struct nvme_rdma_queue *queue,
1077 struct nvme_rdma_request *req, struct nvme_command *c)
1078{
1079 struct nvme_sgl_desc *sg = &c->common.dptr.sgl;
1080
1081 req->sge[1].addr = sg_dma_address(req->sg_table.sgl);
1082 req->sge[1].length = sg_dma_len(req->sg_table.sgl);
1083 req->sge[1].lkey = queue->device->pd->local_dma_lkey;
1084
1085 sg->addr = cpu_to_le64(queue->ctrl->ctrl.icdoff);
1086 sg->length = cpu_to_le32(sg_dma_len(req->sg_table.sgl));
1087 sg->type = (NVME_SGL_FMT_DATA_DESC << 4) | NVME_SGL_FMT_OFFSET;
1088
1089 req->inline_data = true;
1090 req->num_sge++;
1091 return 0;
1092}
1093
1094static int nvme_rdma_map_sg_single(struct nvme_rdma_queue *queue,
1095 struct nvme_rdma_request *req, struct nvme_command *c)
1096{
1097 struct nvme_keyed_sgl_desc *sg = &c->common.dptr.ksgl;
1098
1099 sg->addr = cpu_to_le64(sg_dma_address(req->sg_table.sgl));
1100 put_unaligned_le24(sg_dma_len(req->sg_table.sgl), sg->length);
Christoph Hellwig11975e02016-09-05 12:56:20 +02001101 put_unaligned_le32(queue->device->pd->unsafe_global_rkey, sg->key);
Christoph Hellwig71102302016-07-06 21:55:52 +09001102 sg->type = NVME_KEY_SGL_FMT_DATA_DESC << 4;
1103 return 0;
1104}
1105
1106static int nvme_rdma_map_sg_fr(struct nvme_rdma_queue *queue,
1107 struct nvme_rdma_request *req, struct nvme_command *c,
1108 int count)
1109{
1110 struct nvme_keyed_sgl_desc *sg = &c->common.dptr.ksgl;
1111 int nr;
1112
Max Gurtovoyb925a2d2017-08-28 12:52:27 +03001113 /*
1114 * Align the MR to a 4K page size to match the ctrl page size and
1115 * the block virtual boundary.
1116 */
1117 nr = ib_map_mr_sg(req->mr, req->sg_table.sgl, count, NULL, SZ_4K);
Max Gurtovoya7b7c7a2017-08-14 15:29:26 +03001118 if (unlikely(nr < count)) {
Christoph Hellwig71102302016-07-06 21:55:52 +09001119 if (nr < 0)
1120 return nr;
1121 return -EINVAL;
1122 }
1123
1124 ib_update_fast_reg_key(req->mr, ib_inc_rkey(req->mr->rkey));
1125
1126 req->reg_cqe.done = nvme_rdma_memreg_done;
1127 memset(&req->reg_wr, 0, sizeof(req->reg_wr));
1128 req->reg_wr.wr.opcode = IB_WR_REG_MR;
1129 req->reg_wr.wr.wr_cqe = &req->reg_cqe;
1130 req->reg_wr.wr.num_sge = 0;
1131 req->reg_wr.mr = req->mr;
1132 req->reg_wr.key = req->mr->rkey;
1133 req->reg_wr.access = IB_ACCESS_LOCAL_WRITE |
1134 IB_ACCESS_REMOTE_READ |
1135 IB_ACCESS_REMOTE_WRITE;
1136
Sagi Grimbergf5b7b552016-08-24 12:25:56 +03001137 req->mr->need_inval = true;
Christoph Hellwig71102302016-07-06 21:55:52 +09001138
1139 sg->addr = cpu_to_le64(req->mr->iova);
1140 put_unaligned_le24(req->mr->length, sg->length);
1141 put_unaligned_le32(req->mr->rkey, sg->key);
1142 sg->type = (NVME_KEY_SGL_FMT_DATA_DESC << 4) |
1143 NVME_SGL_FMT_INVALIDATE;
1144
1145 return 0;
1146}
1147
1148static int nvme_rdma_map_data(struct nvme_rdma_queue *queue,
Christoph Hellwigb131c612017-01-13 12:29:12 +01001149 struct request *rq, struct nvme_command *c)
Christoph Hellwig71102302016-07-06 21:55:52 +09001150{
1151 struct nvme_rdma_request *req = blk_mq_rq_to_pdu(rq);
1152 struct nvme_rdma_device *dev = queue->device;
1153 struct ib_device *ibdev = dev->dev;
Christoph Hellwigf9d03f92016-12-08 15:20:32 -07001154 int count, ret;
Christoph Hellwig71102302016-07-06 21:55:52 +09001155
1156 req->num_sge = 1;
1157 req->inline_data = false;
Sagi Grimbergf5b7b552016-08-24 12:25:56 +03001158 req->mr->need_inval = false;
Christoph Hellwig71102302016-07-06 21:55:52 +09001159
1160 c->common.flags |= NVME_CMD_SGL_METABUF;
1161
1162 if (!blk_rq_bytes(rq))
1163 return nvme_rdma_set_sg_null(c);
1164
1165 req->sg_table.sgl = req->first_sgl;
Christoph Hellwigf9d03f92016-12-08 15:20:32 -07001166 ret = sg_alloc_table_chained(&req->sg_table,
1167 blk_rq_nr_phys_segments(rq), req->sg_table.sgl);
Christoph Hellwig71102302016-07-06 21:55:52 +09001168 if (ret)
1169 return -ENOMEM;
1170
Christoph Hellwigf9d03f92016-12-08 15:20:32 -07001171 req->nents = blk_rq_map_sg(rq->q, rq, req->sg_table.sgl);
Christoph Hellwig71102302016-07-06 21:55:52 +09001172
Christoph Hellwigf9d03f92016-12-08 15:20:32 -07001173 count = ib_dma_map_sg(ibdev, req->sg_table.sgl, req->nents,
Christoph Hellwig71102302016-07-06 21:55:52 +09001174 rq_data_dir(rq) == WRITE ? DMA_TO_DEVICE : DMA_FROM_DEVICE);
1175 if (unlikely(count <= 0)) {
1176 sg_free_table_chained(&req->sg_table, true);
1177 return -EIO;
1178 }
1179
1180 if (count == 1) {
Christoph Hellwigb131c612017-01-13 12:29:12 +01001181 if (rq_data_dir(rq) == WRITE && nvme_rdma_queue_idx(queue) &&
1182 blk_rq_payload_bytes(rq) <=
1183 nvme_rdma_inline_data_size(queue))
Christoph Hellwig71102302016-07-06 21:55:52 +09001184 return nvme_rdma_map_sg_inline(queue, req, c);
1185
Christoph Hellwig11975e02016-09-05 12:56:20 +02001186 if (dev->pd->flags & IB_PD_UNSAFE_GLOBAL_RKEY)
Christoph Hellwig71102302016-07-06 21:55:52 +09001187 return nvme_rdma_map_sg_single(queue, req, c);
1188 }
1189
1190 return nvme_rdma_map_sg_fr(queue, req, c, count);
1191}
1192
1193static void nvme_rdma_send_done(struct ib_cq *cq, struct ib_wc *wc)
1194{
1195 if (unlikely(wc->status != IB_WC_SUCCESS))
1196 nvme_rdma_wr_error(cq, wc, "SEND");
1197}
1198
Marta Rybczynska5e599d72017-06-06 13:27:21 +02001199/*
1200 * We want to signal completion at least every queue depth/2. This returns the
1201 * largest power of two that is not above half of (queue size + 1) to optimize
1202 * (avoid divisions).
1203 */
1204static inline bool nvme_rdma_queue_sig_limit(struct nvme_rdma_queue *queue)
Marta Rybczynska0544f542017-04-10 17:12:34 +02001205{
Marta Rybczynska5e599d72017-06-06 13:27:21 +02001206 int limit = 1 << ilog2((queue->queue_size + 1) / 2);
Marta Rybczynska0544f542017-04-10 17:12:34 +02001207
Marta Rybczynska5e599d72017-06-06 13:27:21 +02001208 return (atomic_inc_return(&queue->sig_count) & (limit - 1)) == 0;
Marta Rybczynska0544f542017-04-10 17:12:34 +02001209}
1210
Christoph Hellwig71102302016-07-06 21:55:52 +09001211static int nvme_rdma_post_send(struct nvme_rdma_queue *queue,
1212 struct nvme_rdma_qe *qe, struct ib_sge *sge, u32 num_sge,
1213 struct ib_send_wr *first, bool flush)
1214{
1215 struct ib_send_wr wr, *bad_wr;
1216 int ret;
1217
1218 sge->addr = qe->dma;
1219 sge->length = sizeof(struct nvme_command),
1220 sge->lkey = queue->device->pd->local_dma_lkey;
1221
1222 qe->cqe.done = nvme_rdma_send_done;
1223
1224 wr.next = NULL;
1225 wr.wr_cqe = &qe->cqe;
1226 wr.sg_list = sge;
1227 wr.num_sge = num_sge;
1228 wr.opcode = IB_WR_SEND;
1229 wr.send_flags = 0;
1230
1231 /*
1232 * Unsignalled send completions are another giant desaster in the
1233 * IB Verbs spec: If we don't regularly post signalled sends
1234 * the send queue will fill up and only a QP reset will rescue us.
1235 * Would have been way to obvious to handle this in hardware or
1236 * at least the RDMA stack..
1237 *
Christoph Hellwig71102302016-07-06 21:55:52 +09001238 * Always signal the flushes. The magic request used for the flush
1239 * sequencer is not allocated in our driver's tagset and it's
1240 * triggered to be freed by blk_cleanup_queue(). So we need to
1241 * always mark it as signaled to ensure that the "wr_cqe", which is
Masahiro Yamadab43daed2017-02-27 14:29:09 -08001242 * embedded in request's payload, is not freed when __ib_process_cq()
Christoph Hellwig71102302016-07-06 21:55:52 +09001243 * calls wr_cqe->done().
1244 */
Marta Rybczynska0544f542017-04-10 17:12:34 +02001245 if (nvme_rdma_queue_sig_limit(queue) || flush)
Christoph Hellwig71102302016-07-06 21:55:52 +09001246 wr.send_flags |= IB_SEND_SIGNALED;
1247
1248 if (first)
1249 first->next = &wr;
1250 else
1251 first = &wr;
1252
1253 ret = ib_post_send(queue->qp, first, &bad_wr);
Max Gurtovoya7b7c7a2017-08-14 15:29:26 +03001254 if (unlikely(ret)) {
Christoph Hellwig71102302016-07-06 21:55:52 +09001255 dev_err(queue->ctrl->ctrl.device,
1256 "%s failed with error code %d\n", __func__, ret);
1257 }
1258 return ret;
1259}
1260
1261static int nvme_rdma_post_recv(struct nvme_rdma_queue *queue,
1262 struct nvme_rdma_qe *qe)
1263{
1264 struct ib_recv_wr wr, *bad_wr;
1265 struct ib_sge list;
1266 int ret;
1267
1268 list.addr = qe->dma;
1269 list.length = sizeof(struct nvme_completion);
1270 list.lkey = queue->device->pd->local_dma_lkey;
1271
1272 qe->cqe.done = nvme_rdma_recv_done;
1273
1274 wr.next = NULL;
1275 wr.wr_cqe = &qe->cqe;
1276 wr.sg_list = &list;
1277 wr.num_sge = 1;
1278
1279 ret = ib_post_recv(queue->qp, &wr, &bad_wr);
Max Gurtovoya7b7c7a2017-08-14 15:29:26 +03001280 if (unlikely(ret)) {
Christoph Hellwig71102302016-07-06 21:55:52 +09001281 dev_err(queue->ctrl->ctrl.device,
1282 "%s failed with error code %d\n", __func__, ret);
1283 }
1284 return ret;
1285}
1286
1287static struct blk_mq_tags *nvme_rdma_tagset(struct nvme_rdma_queue *queue)
1288{
1289 u32 queue_idx = nvme_rdma_queue_idx(queue);
1290
1291 if (queue_idx == 0)
1292 return queue->ctrl->admin_tag_set.tags[queue_idx];
1293 return queue->ctrl->tag_set.tags[queue_idx - 1];
1294}
1295
1296static void nvme_rdma_submit_async_event(struct nvme_ctrl *arg, int aer_idx)
1297{
1298 struct nvme_rdma_ctrl *ctrl = to_rdma_ctrl(arg);
1299 struct nvme_rdma_queue *queue = &ctrl->queues[0];
1300 struct ib_device *dev = queue->device->dev;
1301 struct nvme_rdma_qe *sqe = &ctrl->async_event_sqe;
1302 struct nvme_command *cmd = sqe->data;
1303 struct ib_sge sge;
1304 int ret;
1305
1306 if (WARN_ON_ONCE(aer_idx != 0))
1307 return;
1308
1309 ib_dma_sync_single_for_cpu(dev, sqe->dma, sizeof(*cmd), DMA_TO_DEVICE);
1310
1311 memset(cmd, 0, sizeof(*cmd));
1312 cmd->common.opcode = nvme_admin_async_event;
Keith Busch38dabe22017-11-07 15:13:10 -07001313 cmd->common.command_id = NVME_AQ_BLK_MQ_DEPTH;
Christoph Hellwig71102302016-07-06 21:55:52 +09001314 cmd->common.flags |= NVME_CMD_SGL_METABUF;
1315 nvme_rdma_set_sg_null(cmd);
1316
1317 ib_dma_sync_single_for_device(dev, sqe->dma, sizeof(*cmd),
1318 DMA_TO_DEVICE);
1319
1320 ret = nvme_rdma_post_send(queue, sqe, &sge, 1, NULL, false);
1321 WARN_ON_ONCE(ret);
1322}
1323
1324static int nvme_rdma_process_nvme_rsp(struct nvme_rdma_queue *queue,
1325 struct nvme_completion *cqe, struct ib_wc *wc, int tag)
1326{
Christoph Hellwig71102302016-07-06 21:55:52 +09001327 struct request *rq;
1328 struct nvme_rdma_request *req;
1329 int ret = 0;
1330
Christoph Hellwig71102302016-07-06 21:55:52 +09001331 rq = blk_mq_tag_to_rq(nvme_rdma_tagset(queue), cqe->command_id);
1332 if (!rq) {
1333 dev_err(queue->ctrl->ctrl.device,
1334 "tag 0x%x on QP %#x not found\n",
1335 cqe->command_id, queue->qp->qp_num);
1336 nvme_rdma_error_recovery(queue->ctrl);
1337 return ret;
1338 }
1339 req = blk_mq_rq_to_pdu(rq);
1340
Christoph Hellwig71102302016-07-06 21:55:52 +09001341 if (rq->tag == tag)
1342 ret = 1;
1343
1344 if ((wc->wc_flags & IB_WC_WITH_INVALIDATE) &&
1345 wc->ex.invalidate_rkey == req->mr->rkey)
Sagi Grimbergf5b7b552016-08-24 12:25:56 +03001346 req->mr->need_inval = false;
Christoph Hellwig71102302016-07-06 21:55:52 +09001347
Christoph Hellwig27fa9bc2017-04-20 16:02:57 +02001348 nvme_end_request(rq, cqe->status, cqe->result);
Christoph Hellwig71102302016-07-06 21:55:52 +09001349 return ret;
1350}
1351
1352static int __nvme_rdma_recv_done(struct ib_cq *cq, struct ib_wc *wc, int tag)
1353{
1354 struct nvme_rdma_qe *qe =
1355 container_of(wc->wr_cqe, struct nvme_rdma_qe, cqe);
1356 struct nvme_rdma_queue *queue = cq->cq_context;
1357 struct ib_device *ibdev = queue->device->dev;
1358 struct nvme_completion *cqe = qe->data;
1359 const size_t len = sizeof(struct nvme_completion);
1360 int ret = 0;
1361
1362 if (unlikely(wc->status != IB_WC_SUCCESS)) {
1363 nvme_rdma_wr_error(cq, wc, "RECV");
1364 return 0;
1365 }
1366
1367 ib_dma_sync_single_for_cpu(ibdev, qe->dma, len, DMA_FROM_DEVICE);
1368 /*
1369 * AEN requests are special as they don't time out and can
1370 * survive any kind of queue freeze and often don't respond to
1371 * aborts. We don't even bother to allocate a struct request
1372 * for them but rather special case them here.
1373 */
1374 if (unlikely(nvme_rdma_queue_idx(queue) == 0 &&
Keith Busch38dabe22017-11-07 15:13:10 -07001375 cqe->command_id >= NVME_AQ_BLK_MQ_DEPTH))
Christoph Hellwig7bf58532016-11-10 07:32:34 -08001376 nvme_complete_async_event(&queue->ctrl->ctrl, cqe->status,
1377 &cqe->result);
Christoph Hellwig71102302016-07-06 21:55:52 +09001378 else
1379 ret = nvme_rdma_process_nvme_rsp(queue, cqe, wc, tag);
1380 ib_dma_sync_single_for_device(ibdev, qe->dma, len, DMA_FROM_DEVICE);
1381
1382 nvme_rdma_post_recv(queue, qe);
1383 return ret;
1384}
1385
1386static void nvme_rdma_recv_done(struct ib_cq *cq, struct ib_wc *wc)
1387{
1388 __nvme_rdma_recv_done(cq, wc, -1);
1389}
1390
1391static int nvme_rdma_conn_established(struct nvme_rdma_queue *queue)
1392{
1393 int ret, i;
1394
1395 for (i = 0; i < queue->queue_size; i++) {
1396 ret = nvme_rdma_post_recv(queue, &queue->rsp_ring[i]);
1397 if (ret)
1398 goto out_destroy_queue_ib;
1399 }
1400
1401 return 0;
1402
1403out_destroy_queue_ib:
1404 nvme_rdma_destroy_queue_ib(queue);
1405 return ret;
1406}
1407
1408static int nvme_rdma_conn_rejected(struct nvme_rdma_queue *queue,
1409 struct rdma_cm_event *ev)
1410{
Steve Wise7f039532016-10-26 12:36:47 -07001411 struct rdma_cm_id *cm_id = queue->cm_id;
1412 int status = ev->status;
1413 const char *rej_msg;
1414 const struct nvme_rdma_cm_rej *rej_data;
1415 u8 rej_data_len;
1416
1417 rej_msg = rdma_reject_msg(cm_id, status);
1418 rej_data = rdma_consumer_reject_data(cm_id, ev, &rej_data_len);
1419
1420 if (rej_data && rej_data_len >= sizeof(u16)) {
1421 u16 sts = le16_to_cpu(rej_data->sts);
Christoph Hellwig71102302016-07-06 21:55:52 +09001422
1423 dev_err(queue->ctrl->ctrl.device,
Steve Wise7f039532016-10-26 12:36:47 -07001424 "Connect rejected: status %d (%s) nvme status %d (%s).\n",
1425 status, rej_msg, sts, nvme_rdma_cm_msg(sts));
Christoph Hellwig71102302016-07-06 21:55:52 +09001426 } else {
1427 dev_err(queue->ctrl->ctrl.device,
Steve Wise7f039532016-10-26 12:36:47 -07001428 "Connect rejected: status %d (%s).\n", status, rej_msg);
Christoph Hellwig71102302016-07-06 21:55:52 +09001429 }
1430
1431 return -ECONNRESET;
1432}
1433
1434static int nvme_rdma_addr_resolved(struct nvme_rdma_queue *queue)
1435{
Christoph Hellwig71102302016-07-06 21:55:52 +09001436 int ret;
1437
Sagi Grimbergca6e95b2017-05-04 13:33:09 +03001438 ret = nvme_rdma_create_queue_ib(queue);
1439 if (ret)
1440 return ret;
Christoph Hellwig71102302016-07-06 21:55:52 +09001441
1442 ret = rdma_resolve_route(queue->cm_id, NVME_RDMA_CONNECT_TIMEOUT_MS);
1443 if (ret) {
1444 dev_err(queue->ctrl->ctrl.device,
1445 "rdma_resolve_route failed (%d).\n",
1446 queue->cm_error);
1447 goto out_destroy_queue;
1448 }
1449
1450 return 0;
1451
1452out_destroy_queue:
1453 nvme_rdma_destroy_queue_ib(queue);
Christoph Hellwig71102302016-07-06 21:55:52 +09001454 return ret;
1455}
1456
1457static int nvme_rdma_route_resolved(struct nvme_rdma_queue *queue)
1458{
1459 struct nvme_rdma_ctrl *ctrl = queue->ctrl;
1460 struct rdma_conn_param param = { };
Roland Dreier0b857b42016-07-31 00:27:39 -07001461 struct nvme_rdma_cm_req priv = { };
Christoph Hellwig71102302016-07-06 21:55:52 +09001462 int ret;
1463
1464 param.qp_num = queue->qp->qp_num;
1465 param.flow_control = 1;
1466
1467 param.responder_resources = queue->device->dev->attrs.max_qp_rd_atom;
Sagi Grimberg2ac17c22016-06-22 15:06:00 +03001468 /* maximum retry count */
1469 param.retry_count = 7;
Christoph Hellwig71102302016-07-06 21:55:52 +09001470 param.rnr_retry_count = 7;
1471 param.private_data = &priv;
1472 param.private_data_len = sizeof(priv);
1473
1474 priv.recfmt = cpu_to_le16(NVME_RDMA_CM_FMT_1_0);
1475 priv.qid = cpu_to_le16(nvme_rdma_queue_idx(queue));
Jay Freyenseef994d9d2016-08-17 15:00:26 -07001476 /*
1477 * set the admin queue depth to the minimum size
1478 * specified by the Fabrics standard.
1479 */
1480 if (priv.qid == 0) {
Sagi Grimberg7aa1f422017-06-18 16:15:59 +03001481 priv.hrqsize = cpu_to_le16(NVME_AQ_DEPTH);
1482 priv.hsqsize = cpu_to_le16(NVME_AQ_DEPTH - 1);
Jay Freyenseef994d9d2016-08-17 15:00:26 -07001483 } else {
Jay Freyenseec5af8652016-08-17 15:00:27 -07001484 /*
1485 * current interpretation of the fabrics spec
1486 * is at minimum you make hrqsize sqsize+1, or a
1487 * 1's based representation of sqsize.
1488 */
Jay Freyenseef994d9d2016-08-17 15:00:26 -07001489 priv.hrqsize = cpu_to_le16(queue->queue_size);
Jay Freyenseec5af8652016-08-17 15:00:27 -07001490 priv.hsqsize = cpu_to_le16(queue->ctrl->ctrl.sqsize);
Jay Freyenseef994d9d2016-08-17 15:00:26 -07001491 }
Christoph Hellwig71102302016-07-06 21:55:52 +09001492
1493 ret = rdma_connect(queue->cm_id, &param);
1494 if (ret) {
1495 dev_err(ctrl->ctrl.device,
1496 "rdma_connect failed (%d).\n", ret);
1497 goto out_destroy_queue_ib;
1498 }
1499
1500 return 0;
1501
1502out_destroy_queue_ib:
1503 nvme_rdma_destroy_queue_ib(queue);
1504 return ret;
1505}
1506
Christoph Hellwig71102302016-07-06 21:55:52 +09001507static int nvme_rdma_cm_handler(struct rdma_cm_id *cm_id,
1508 struct rdma_cm_event *ev)
1509{
1510 struct nvme_rdma_queue *queue = cm_id->context;
1511 int cm_error = 0;
1512
1513 dev_dbg(queue->ctrl->ctrl.device, "%s (%d): status %d id %p\n",
1514 rdma_event_msg(ev->event), ev->event,
1515 ev->status, cm_id);
1516
1517 switch (ev->event) {
1518 case RDMA_CM_EVENT_ADDR_RESOLVED:
1519 cm_error = nvme_rdma_addr_resolved(queue);
1520 break;
1521 case RDMA_CM_EVENT_ROUTE_RESOLVED:
1522 cm_error = nvme_rdma_route_resolved(queue);
1523 break;
1524 case RDMA_CM_EVENT_ESTABLISHED:
1525 queue->cm_error = nvme_rdma_conn_established(queue);
1526 /* complete cm_done regardless of success/failure */
1527 complete(&queue->cm_done);
1528 return 0;
1529 case RDMA_CM_EVENT_REJECTED:
Sagi Grimbergabf87d52017-05-04 13:33:10 +03001530 nvme_rdma_destroy_queue_ib(queue);
Christoph Hellwig71102302016-07-06 21:55:52 +09001531 cm_error = nvme_rdma_conn_rejected(queue, ev);
1532 break;
Christoph Hellwig71102302016-07-06 21:55:52 +09001533 case RDMA_CM_EVENT_ROUTE_ERROR:
1534 case RDMA_CM_EVENT_CONNECT_ERROR:
1535 case RDMA_CM_EVENT_UNREACHABLE:
Sagi Grimbergabf87d52017-05-04 13:33:10 +03001536 nvme_rdma_destroy_queue_ib(queue);
1537 case RDMA_CM_EVENT_ADDR_ERROR:
Christoph Hellwig71102302016-07-06 21:55:52 +09001538 dev_dbg(queue->ctrl->ctrl.device,
1539 "CM error event %d\n", ev->event);
1540 cm_error = -ECONNRESET;
1541 break;
1542 case RDMA_CM_EVENT_DISCONNECTED:
1543 case RDMA_CM_EVENT_ADDR_CHANGE:
1544 case RDMA_CM_EVENT_TIMEWAIT_EXIT:
1545 dev_dbg(queue->ctrl->ctrl.device,
1546 "disconnect received - connection closed\n");
1547 nvme_rdma_error_recovery(queue->ctrl);
1548 break;
1549 case RDMA_CM_EVENT_DEVICE_REMOVAL:
Steve Wisee87a9112016-09-02 09:01:54 -07001550 /* device removal is handled via the ib_client API */
1551 break;
Christoph Hellwig71102302016-07-06 21:55:52 +09001552 default:
1553 dev_err(queue->ctrl->ctrl.device,
1554 "Unexpected RDMA CM event (%d)\n", ev->event);
1555 nvme_rdma_error_recovery(queue->ctrl);
1556 break;
1557 }
1558
1559 if (cm_error) {
1560 queue->cm_error = cm_error;
1561 complete(&queue->cm_done);
1562 }
1563
1564 return 0;
1565}
1566
1567static enum blk_eh_timer_return
1568nvme_rdma_timeout(struct request *rq, bool reserved)
1569{
1570 struct nvme_rdma_request *req = blk_mq_rq_to_pdu(rq);
1571
Nitzan Carmie62a5382017-10-22 09:37:04 +00001572 dev_warn(req->queue->ctrl->ctrl.device,
1573 "I/O %d QID %d timeout, reset controller\n",
1574 rq->tag, nvme_rdma_queue_idx(req->queue));
1575
Christoph Hellwig71102302016-07-06 21:55:52 +09001576 /* queue error recovery */
1577 nvme_rdma_error_recovery(req->queue->ctrl);
1578
1579 /* fail with DNR on cmd timeout */
Christoph Hellwig27fa9bc2017-04-20 16:02:57 +02001580 nvme_req(rq)->status = NVME_SC_ABORT_REQ | NVME_SC_DNR;
Christoph Hellwig71102302016-07-06 21:55:52 +09001581
1582 return BLK_EH_HANDLED;
1583}
1584
Christoph Hellwig553cd9e2016-11-02 08:49:18 -06001585/*
1586 * We cannot accept any other command until the Connect command has completed.
1587 */
Christoph Hellwiga104c9f2017-06-12 18:26:06 +02001588static inline blk_status_t
1589nvme_rdma_queue_is_ready(struct nvme_rdma_queue *queue, struct request *rq)
Christoph Hellwig553cd9e2016-11-02 08:49:18 -06001590{
1591 if (unlikely(!test_bit(NVME_RDMA_Q_LIVE, &queue->flags))) {
Christoph Hellwig1392370e2017-01-03 14:29:02 +03001592 struct nvme_command *cmd = nvme_req(rq)->cmd;
Christoph Hellwig553cd9e2016-11-02 08:49:18 -06001593
Christoph Hellwig57292b52017-01-31 16:57:29 +01001594 if (!blk_rq_is_passthrough(rq) ||
Christoph Hellwig553cd9e2016-11-02 08:49:18 -06001595 cmd->common.opcode != nvme_fabrics_command ||
Sagi Grimberge818a5b2017-06-05 20:35:56 +03001596 cmd->fabrics.fctype != nvme_fabrics_type_connect) {
1597 /*
1598 * reconnecting state means transport disruption, which
1599 * can take a long time and even might fail permanently,
1600 * so we can't let incoming I/O be requeued forever.
1601 * fail it fast to allow upper layers a chance to
1602 * failover.
1603 */
1604 if (queue->ctrl->ctrl.state == NVME_CTRL_RECONNECTING)
Christoph Hellwiga104c9f2017-06-12 18:26:06 +02001605 return BLK_STS_IOERR;
1606 return BLK_STS_RESOURCE; /* try again later */
Sagi Grimberge818a5b2017-06-05 20:35:56 +03001607 }
Christoph Hellwig553cd9e2016-11-02 08:49:18 -06001608 }
1609
Sagi Grimberge818a5b2017-06-05 20:35:56 +03001610 return 0;
Christoph Hellwig553cd9e2016-11-02 08:49:18 -06001611}
1612
Christoph Hellwigfc17b652017-06-03 09:38:05 +02001613static blk_status_t nvme_rdma_queue_rq(struct blk_mq_hw_ctx *hctx,
Christoph Hellwig71102302016-07-06 21:55:52 +09001614 const struct blk_mq_queue_data *bd)
1615{
1616 struct nvme_ns *ns = hctx->queue->queuedata;
1617 struct nvme_rdma_queue *queue = hctx->driver_data;
1618 struct request *rq = bd->rq;
1619 struct nvme_rdma_request *req = blk_mq_rq_to_pdu(rq);
1620 struct nvme_rdma_qe *sqe = &req->sqe;
1621 struct nvme_command *c = sqe->data;
1622 bool flush = false;
1623 struct ib_device *dev;
Christoph Hellwigfc17b652017-06-03 09:38:05 +02001624 blk_status_t ret;
1625 int err;
Christoph Hellwig71102302016-07-06 21:55:52 +09001626
1627 WARN_ON_ONCE(rq->tag < 0);
1628
Sagi Grimberge818a5b2017-06-05 20:35:56 +03001629 ret = nvme_rdma_queue_is_ready(queue, rq);
1630 if (unlikely(ret))
Christoph Hellwiga104c9f2017-06-12 18:26:06 +02001631 return ret;
Christoph Hellwig553cd9e2016-11-02 08:49:18 -06001632
Christoph Hellwig71102302016-07-06 21:55:52 +09001633 dev = queue->device->dev;
1634 ib_dma_sync_single_for_cpu(dev, sqe->dma,
1635 sizeof(struct nvme_command), DMA_TO_DEVICE);
1636
1637 ret = nvme_setup_cmd(ns, rq, c);
Christoph Hellwigfc17b652017-06-03 09:38:05 +02001638 if (ret)
Christoph Hellwig71102302016-07-06 21:55:52 +09001639 return ret;
1640
Christoph Hellwig71102302016-07-06 21:55:52 +09001641 blk_mq_start_request(rq);
1642
Christoph Hellwigfc17b652017-06-03 09:38:05 +02001643 err = nvme_rdma_map_data(queue, rq, c);
Max Gurtovoya7b7c7a2017-08-14 15:29:26 +03001644 if (unlikely(err < 0)) {
Christoph Hellwig71102302016-07-06 21:55:52 +09001645 dev_err(queue->ctrl->ctrl.device,
Christoph Hellwigfc17b652017-06-03 09:38:05 +02001646 "Failed to map data (%d)\n", err);
Christoph Hellwig71102302016-07-06 21:55:52 +09001647 nvme_cleanup_cmd(rq);
1648 goto err;
1649 }
1650
1651 ib_dma_sync_single_for_device(dev, sqe->dma,
1652 sizeof(struct nvme_command), DMA_TO_DEVICE);
1653
Christoph Hellwigaebf5262017-01-31 16:57:31 +01001654 if (req_op(rq) == REQ_OP_FLUSH)
Christoph Hellwig71102302016-07-06 21:55:52 +09001655 flush = true;
Christoph Hellwigfc17b652017-06-03 09:38:05 +02001656 err = nvme_rdma_post_send(queue, sqe, req->sge, req->num_sge,
Sagi Grimbergf5b7b552016-08-24 12:25:56 +03001657 req->mr->need_inval ? &req->reg_wr.wr : NULL, flush);
Max Gurtovoya7b7c7a2017-08-14 15:29:26 +03001658 if (unlikely(err)) {
Christoph Hellwig71102302016-07-06 21:55:52 +09001659 nvme_rdma_unmap_data(queue, rq);
1660 goto err;
1661 }
1662
Christoph Hellwigfc17b652017-06-03 09:38:05 +02001663 return BLK_STS_OK;
Christoph Hellwig71102302016-07-06 21:55:52 +09001664err:
Christoph Hellwigfc17b652017-06-03 09:38:05 +02001665 if (err == -ENOMEM || err == -EAGAIN)
1666 return BLK_STS_RESOURCE;
1667 return BLK_STS_IOERR;
Christoph Hellwig71102302016-07-06 21:55:52 +09001668}
1669
1670static int nvme_rdma_poll(struct blk_mq_hw_ctx *hctx, unsigned int tag)
1671{
1672 struct nvme_rdma_queue *queue = hctx->driver_data;
1673 struct ib_cq *cq = queue->ib_cq;
1674 struct ib_wc wc;
1675 int found = 0;
1676
Christoph Hellwig71102302016-07-06 21:55:52 +09001677 while (ib_poll_cq(cq, 1, &wc) > 0) {
1678 struct ib_cqe *cqe = wc.wr_cqe;
1679
1680 if (cqe) {
1681 if (cqe->done == nvme_rdma_recv_done)
1682 found |= __nvme_rdma_recv_done(cq, &wc, tag);
1683 else
1684 cqe->done(cq, &wc);
1685 }
1686 }
1687
1688 return found;
1689}
1690
1691static void nvme_rdma_complete_rq(struct request *rq)
1692{
1693 struct nvme_rdma_request *req = blk_mq_rq_to_pdu(rq);
Christoph Hellwig71102302016-07-06 21:55:52 +09001694
Christoph Hellwig77f02a72017-03-30 13:41:32 +02001695 nvme_rdma_unmap_data(req->queue, rq);
1696 nvme_complete_rq(rq);
Christoph Hellwig71102302016-07-06 21:55:52 +09001697}
1698
Sagi Grimberg0b366582017-07-13 11:09:44 +03001699static int nvme_rdma_map_queues(struct blk_mq_tag_set *set)
1700{
1701 struct nvme_rdma_ctrl *ctrl = set->driver_data;
1702
1703 return blk_mq_rdma_map_queues(set, ctrl->device->dev, 0);
1704}
1705
Eric Biggersf363b082017-03-30 13:39:16 -07001706static const struct blk_mq_ops nvme_rdma_mq_ops = {
Christoph Hellwig71102302016-07-06 21:55:52 +09001707 .queue_rq = nvme_rdma_queue_rq,
1708 .complete = nvme_rdma_complete_rq,
Christoph Hellwig71102302016-07-06 21:55:52 +09001709 .init_request = nvme_rdma_init_request,
1710 .exit_request = nvme_rdma_exit_request,
Christoph Hellwig71102302016-07-06 21:55:52 +09001711 .init_hctx = nvme_rdma_init_hctx,
1712 .poll = nvme_rdma_poll,
1713 .timeout = nvme_rdma_timeout,
Sagi Grimberg0b366582017-07-13 11:09:44 +03001714 .map_queues = nvme_rdma_map_queues,
Christoph Hellwig71102302016-07-06 21:55:52 +09001715};
1716
Eric Biggersf363b082017-03-30 13:39:16 -07001717static const struct blk_mq_ops nvme_rdma_admin_mq_ops = {
Christoph Hellwig71102302016-07-06 21:55:52 +09001718 .queue_rq = nvme_rdma_queue_rq,
1719 .complete = nvme_rdma_complete_rq,
Christoph Hellwig385475e2017-06-13 09:15:19 +02001720 .init_request = nvme_rdma_init_request,
1721 .exit_request = nvme_rdma_exit_request,
Christoph Hellwig71102302016-07-06 21:55:52 +09001722 .init_hctx = nvme_rdma_init_admin_hctx,
1723 .timeout = nvme_rdma_timeout,
1724};
1725
Sagi Grimberg18398af2017-07-10 09:22:31 +03001726static void nvme_rdma_shutdown_ctrl(struct nvme_rdma_ctrl *ctrl, bool shutdown)
Christoph Hellwig71102302016-07-06 21:55:52 +09001727{
Christoph Hellwig71102302016-07-06 21:55:52 +09001728 cancel_work_sync(&ctrl->err_work);
1729 cancel_delayed_work_sync(&ctrl->reconnect_work);
1730
Sagi Grimbergd858e5f2017-04-24 10:58:29 +03001731 if (ctrl->ctrl.queue_count > 1) {
Christoph Hellwig71102302016-07-06 21:55:52 +09001732 nvme_stop_queues(&ctrl->ctrl);
1733 blk_mq_tagset_busy_iter(&ctrl->tag_set,
1734 nvme_cancel_request, &ctrl->ctrl);
Sagi Grimberga57bd542017-08-28 21:41:10 +02001735 nvme_rdma_destroy_io_queues(ctrl, shutdown);
Christoph Hellwig71102302016-07-06 21:55:52 +09001736 }
1737
Sagi Grimberg18398af2017-07-10 09:22:31 +03001738 if (shutdown)
Christoph Hellwig71102302016-07-06 21:55:52 +09001739 nvme_shutdown_ctrl(&ctrl->ctrl);
Sagi Grimberg18398af2017-07-10 09:22:31 +03001740 else
1741 nvme_disable_ctrl(&ctrl->ctrl, ctrl->ctrl.cap);
Christoph Hellwig71102302016-07-06 21:55:52 +09001742
Sagi Grimbergfb051332017-07-02 15:33:32 +03001743 blk_mq_quiesce_queue(ctrl->ctrl.admin_q);
Christoph Hellwig71102302016-07-06 21:55:52 +09001744 blk_mq_tagset_busy_iter(&ctrl->admin_tag_set,
1745 nvme_cancel_request, &ctrl->ctrl);
Sagi Grimbergfb051332017-07-02 15:33:32 +03001746 blk_mq_unquiesce_queue(ctrl->ctrl.admin_q);
Sagi Grimberg3f02fff2017-07-10 09:22:32 +03001747 nvme_rdma_destroy_admin_queue(ctrl, shutdown);
Christoph Hellwig71102302016-07-06 21:55:52 +09001748}
1749
Christoph Hellwigc5017e82017-10-29 10:44:29 +02001750static void nvme_rdma_delete_ctrl(struct nvme_ctrl *ctrl)
Christoph Hellwig71102302016-07-06 21:55:52 +09001751{
Christoph Hellwige9bc2582017-10-29 10:44:30 +02001752 nvme_rdma_shutdown_ctrl(to_rdma_ctrl(ctrl), true);
Christoph Hellwig71102302016-07-06 21:55:52 +09001753}
1754
Christoph Hellwig71102302016-07-06 21:55:52 +09001755static void nvme_rdma_reset_ctrl_work(struct work_struct *work)
1756{
Christoph Hellwigd86c4d82017-06-15 15:41:08 +02001757 struct nvme_rdma_ctrl *ctrl =
1758 container_of(work, struct nvme_rdma_ctrl, ctrl.reset_work);
Christoph Hellwig71102302016-07-06 21:55:52 +09001759 int ret;
1760 bool changed;
1761
Sagi Grimbergd09f2b42017-07-02 10:56:43 +03001762 nvme_stop_ctrl(&ctrl->ctrl);
Sagi Grimberg18398af2017-07-10 09:22:31 +03001763 nvme_rdma_shutdown_ctrl(ctrl, false);
Christoph Hellwig71102302016-07-06 21:55:52 +09001764
Sagi Grimberg3f02fff2017-07-10 09:22:32 +03001765 ret = nvme_rdma_configure_admin_queue(ctrl, false);
Sagi Grimberg370ae6e2017-07-10 09:22:38 +03001766 if (ret)
1767 goto out_fail;
Christoph Hellwig71102302016-07-06 21:55:52 +09001768
Sagi Grimbergd858e5f2017-04-24 10:58:29 +03001769 if (ctrl->ctrl.queue_count > 1) {
Sagi Grimberga57bd542017-08-28 21:41:10 +02001770 ret = nvme_rdma_configure_io_queues(ctrl, false);
Christoph Hellwig71102302016-07-06 21:55:52 +09001771 if (ret)
Sagi Grimberg370ae6e2017-07-10 09:22:38 +03001772 goto out_fail;
Christoph Hellwig71102302016-07-06 21:55:52 +09001773 }
1774
1775 changed = nvme_change_ctrl_state(&ctrl->ctrl, NVME_CTRL_LIVE);
Sagi Grimberg0ad0bfa2017-10-11 12:49:51 +03001776 if (!changed) {
1777 /* state change failure is ok if we're in DELETING state */
1778 WARN_ON_ONCE(ctrl->ctrl.state != NVME_CTRL_DELETING);
1779 return;
1780 }
Christoph Hellwig71102302016-07-06 21:55:52 +09001781
Sagi Grimbergd09f2b42017-07-02 10:56:43 +03001782 nvme_start_ctrl(&ctrl->ctrl);
Christoph Hellwig71102302016-07-06 21:55:52 +09001783
1784 return;
1785
Sagi Grimberg370ae6e2017-07-10 09:22:38 +03001786out_fail:
Christoph Hellwig71102302016-07-06 21:55:52 +09001787 dev_warn(ctrl->ctrl.device, "Removing after reset failure\n");
Christoph Hellwige9bc2582017-10-29 10:44:30 +02001788 nvme_remove_namespaces(&ctrl->ctrl);
1789 nvme_rdma_shutdown_ctrl(ctrl, true);
1790 nvme_uninit_ctrl(&ctrl->ctrl);
1791 nvme_put_ctrl(&ctrl->ctrl);
Christoph Hellwig71102302016-07-06 21:55:52 +09001792}
1793
Christoph Hellwig71102302016-07-06 21:55:52 +09001794static const struct nvme_ctrl_ops nvme_rdma_ctrl_ops = {
1795 .name = "rdma",
1796 .module = THIS_MODULE,
Christoph Hellwigd3d5b872017-05-20 15:14:44 +02001797 .flags = NVME_F_FABRICS,
Christoph Hellwig71102302016-07-06 21:55:52 +09001798 .reg_read32 = nvmf_reg_read32,
1799 .reg_read64 = nvmf_reg_read64,
1800 .reg_write32 = nvmf_reg_write32,
Christoph Hellwig71102302016-07-06 21:55:52 +09001801 .free_ctrl = nvme_rdma_free_ctrl,
1802 .submit_async_event = nvme_rdma_submit_async_event,
Christoph Hellwigc5017e82017-10-29 10:44:29 +02001803 .delete_ctrl = nvme_rdma_delete_ctrl,
Christoph Hellwig71102302016-07-06 21:55:52 +09001804 .get_address = nvmf_get_address,
Sagi Grimberg31b84462017-10-11 12:53:07 +03001805 .reinit_request = nvme_rdma_reinit_request,
Christoph Hellwig71102302016-07-06 21:55:52 +09001806};
1807
James Smart36e835f2017-10-20 16:17:09 -07001808static inline bool
1809__nvme_rdma_options_match(struct nvme_rdma_ctrl *ctrl,
1810 struct nvmf_ctrl_options *opts)
1811{
1812 char *stdport = __stringify(NVME_RDMA_IP_PORT);
1813
1814
1815 if (!nvmf_ctlr_matches_baseopts(&ctrl->ctrl, opts) ||
1816 strcmp(opts->traddr, ctrl->ctrl.opts->traddr))
1817 return false;
1818
1819 if (opts->mask & NVMF_OPT_TRSVCID &&
1820 ctrl->ctrl.opts->mask & NVMF_OPT_TRSVCID) {
1821 if (strcmp(opts->trsvcid, ctrl->ctrl.opts->trsvcid))
1822 return false;
1823 } else if (opts->mask & NVMF_OPT_TRSVCID) {
1824 if (strcmp(opts->trsvcid, stdport))
1825 return false;
1826 } else if (ctrl->ctrl.opts->mask & NVMF_OPT_TRSVCID) {
1827 if (strcmp(stdport, ctrl->ctrl.opts->trsvcid))
1828 return false;
1829 }
1830 /* else, it's a match as both have stdport. Fall to next checks */
1831
1832 /*
1833 * checking the local address is rough. In most cases, one
1834 * is not specified and the host port is selected by the stack.
1835 *
1836 * Assume no match if:
1837 * local address is specified and address is not the same
1838 * local address is not specified but remote is, or vice versa
1839 * (admin using specific host_traddr when it matters).
1840 */
1841 if (opts->mask & NVMF_OPT_HOST_TRADDR &&
1842 ctrl->ctrl.opts->mask & NVMF_OPT_HOST_TRADDR) {
1843 if (strcmp(opts->host_traddr, ctrl->ctrl.opts->host_traddr))
1844 return false;
1845 } else if (opts->mask & NVMF_OPT_HOST_TRADDR ||
1846 ctrl->ctrl.opts->mask & NVMF_OPT_HOST_TRADDR)
1847 return false;
1848 /*
1849 * if neither controller had an host port specified, assume it's
1850 * a match as everything else matched.
1851 */
1852
1853 return true;
1854}
1855
1856/*
1857 * Fails a connection request if it matches an existing controller
1858 * (association) with the same tuple:
1859 * <Host NQN, Host ID, local address, remote address, remote port, SUBSYS NQN>
1860 *
1861 * if local address is not specified in the request, it will match an
1862 * existing controller with all the other parameters the same and no
1863 * local port address specified as well.
1864 *
1865 * The ports don't need to be compared as they are intrinsically
1866 * already matched by the port pointers supplied.
1867 */
1868static bool
1869nvme_rdma_existing_controller(struct nvmf_ctrl_options *opts)
1870{
1871 struct nvme_rdma_ctrl *ctrl;
1872 bool found = false;
1873
1874 mutex_lock(&nvme_rdma_ctrl_mutex);
1875 list_for_each_entry(ctrl, &nvme_rdma_ctrl_list, list) {
1876 found = __nvme_rdma_options_match(ctrl, opts);
1877 if (found)
1878 break;
1879 }
1880 mutex_unlock(&nvme_rdma_ctrl_mutex);
1881
1882 return found;
1883}
1884
Christoph Hellwig71102302016-07-06 21:55:52 +09001885static struct nvme_ctrl *nvme_rdma_create_ctrl(struct device *dev,
1886 struct nvmf_ctrl_options *opts)
1887{
1888 struct nvme_rdma_ctrl *ctrl;
1889 int ret;
1890 bool changed;
Sagi Grimberg0928f9b2017-02-05 21:49:32 +02001891 char *port;
Christoph Hellwig71102302016-07-06 21:55:52 +09001892
1893 ctrl = kzalloc(sizeof(*ctrl), GFP_KERNEL);
1894 if (!ctrl)
1895 return ERR_PTR(-ENOMEM);
1896 ctrl->ctrl.opts = opts;
1897 INIT_LIST_HEAD(&ctrl->list);
1898
Sagi Grimberg0928f9b2017-02-05 21:49:32 +02001899 if (opts->mask & NVMF_OPT_TRSVCID)
1900 port = opts->trsvcid;
1901 else
1902 port = __stringify(NVME_RDMA_IP_PORT);
1903
1904 ret = inet_pton_with_scope(&init_net, AF_UNSPEC,
1905 opts->traddr, port, &ctrl->addr);
Christoph Hellwig71102302016-07-06 21:55:52 +09001906 if (ret) {
Sagi Grimberg0928f9b2017-02-05 21:49:32 +02001907 pr_err("malformed address passed: %s:%s\n", opts->traddr, port);
Christoph Hellwig71102302016-07-06 21:55:52 +09001908 goto out_free_ctrl;
1909 }
1910
Max Gurtovoy8f4e8da2017-02-19 20:08:03 +02001911 if (opts->mask & NVMF_OPT_HOST_TRADDR) {
Sagi Grimberg0928f9b2017-02-05 21:49:32 +02001912 ret = inet_pton_with_scope(&init_net, AF_UNSPEC,
1913 opts->host_traddr, NULL, &ctrl->src_addr);
Max Gurtovoy8f4e8da2017-02-19 20:08:03 +02001914 if (ret) {
Sagi Grimberg0928f9b2017-02-05 21:49:32 +02001915 pr_err("malformed src address passed: %s\n",
Max Gurtovoy8f4e8da2017-02-19 20:08:03 +02001916 opts->host_traddr);
1917 goto out_free_ctrl;
1918 }
1919 }
1920
James Smart36e835f2017-10-20 16:17:09 -07001921 if (!opts->duplicate_connect && nvme_rdma_existing_controller(opts)) {
1922 ret = -EALREADY;
1923 goto out_free_ctrl;
1924 }
1925
Christoph Hellwig71102302016-07-06 21:55:52 +09001926 ret = nvme_init_ctrl(&ctrl->ctrl, dev, &nvme_rdma_ctrl_ops,
1927 0 /* no quirks, we're perfect! */);
1928 if (ret)
1929 goto out_free_ctrl;
1930
Christoph Hellwig71102302016-07-06 21:55:52 +09001931 INIT_DELAYED_WORK(&ctrl->reconnect_work,
1932 nvme_rdma_reconnect_ctrl_work);
1933 INIT_WORK(&ctrl->err_work, nvme_rdma_error_recovery_work);
Christoph Hellwigd86c4d82017-06-15 15:41:08 +02001934 INIT_WORK(&ctrl->ctrl.reset_work, nvme_rdma_reset_ctrl_work);
Christoph Hellwig71102302016-07-06 21:55:52 +09001935
Sagi Grimbergd858e5f2017-04-24 10:58:29 +03001936 ctrl->ctrl.queue_count = opts->nr_io_queues + 1; /* +1 for admin queue */
Jay Freyenseec5af8652016-08-17 15:00:27 -07001937 ctrl->ctrl.sqsize = opts->queue_size - 1;
Christoph Hellwig71102302016-07-06 21:55:52 +09001938 ctrl->ctrl.kato = opts->kato;
1939
1940 ret = -ENOMEM;
Sagi Grimbergd858e5f2017-04-24 10:58:29 +03001941 ctrl->queues = kcalloc(ctrl->ctrl.queue_count, sizeof(*ctrl->queues),
Christoph Hellwig71102302016-07-06 21:55:52 +09001942 GFP_KERNEL);
1943 if (!ctrl->queues)
1944 goto out_uninit_ctrl;
1945
Sagi Grimberg3f02fff2017-07-10 09:22:32 +03001946 ret = nvme_rdma_configure_admin_queue(ctrl, true);
Christoph Hellwig71102302016-07-06 21:55:52 +09001947 if (ret)
1948 goto out_kfree_queues;
1949
1950 /* sanity check icdoff */
1951 if (ctrl->ctrl.icdoff) {
1952 dev_err(ctrl->ctrl.device, "icdoff is not supported!\n");
Dan Carpenterbb472ba2017-06-14 13:46:45 +03001953 ret = -EINVAL;
Christoph Hellwig71102302016-07-06 21:55:52 +09001954 goto out_remove_admin_queue;
1955 }
1956
1957 /* sanity check keyed sgls */
1958 if (!(ctrl->ctrl.sgls & (1 << 20))) {
1959 dev_err(ctrl->ctrl.device, "Mandatory keyed sgls are not support\n");
Dan Carpenterbb472ba2017-06-14 13:46:45 +03001960 ret = -EINVAL;
Christoph Hellwig71102302016-07-06 21:55:52 +09001961 goto out_remove_admin_queue;
1962 }
1963
1964 if (opts->queue_size > ctrl->ctrl.maxcmd) {
1965 /* warn if maxcmd is lower than queue_size */
1966 dev_warn(ctrl->ctrl.device,
1967 "queue_size %zu > ctrl maxcmd %u, clamping down\n",
1968 opts->queue_size, ctrl->ctrl.maxcmd);
1969 opts->queue_size = ctrl->ctrl.maxcmd;
1970 }
1971
Samuel Jones76c08bf2016-10-25 09:22:34 +02001972 if (opts->queue_size > ctrl->ctrl.sqsize + 1) {
1973 /* warn if sqsize is lower than queue_size */
1974 dev_warn(ctrl->ctrl.device,
1975 "queue_size %zu > ctrl sqsize %u, clamping down\n",
1976 opts->queue_size, ctrl->ctrl.sqsize + 1);
1977 opts->queue_size = ctrl->ctrl.sqsize + 1;
1978 }
1979
Christoph Hellwig71102302016-07-06 21:55:52 +09001980 if (opts->nr_io_queues) {
Sagi Grimberga57bd542017-08-28 21:41:10 +02001981 ret = nvme_rdma_configure_io_queues(ctrl, true);
Christoph Hellwig71102302016-07-06 21:55:52 +09001982 if (ret)
1983 goto out_remove_admin_queue;
1984 }
1985
1986 changed = nvme_change_ctrl_state(&ctrl->ctrl, NVME_CTRL_LIVE);
1987 WARN_ON_ONCE(!changed);
1988
Sagi Grimberg0928f9b2017-02-05 21:49:32 +02001989 dev_info(ctrl->ctrl.device, "new ctrl: NQN \"%s\", addr %pISpcs\n",
Christoph Hellwig71102302016-07-06 21:55:52 +09001990 ctrl->ctrl.opts->subsysnqn, &ctrl->addr);
1991
Christoph Hellwigd22524a2017-10-18 13:25:42 +02001992 nvme_get_ctrl(&ctrl->ctrl);
Christoph Hellwig71102302016-07-06 21:55:52 +09001993
1994 mutex_lock(&nvme_rdma_ctrl_mutex);
1995 list_add_tail(&ctrl->list, &nvme_rdma_ctrl_list);
1996 mutex_unlock(&nvme_rdma_ctrl_mutex);
1997
Sagi Grimbergd09f2b42017-07-02 10:56:43 +03001998 nvme_start_ctrl(&ctrl->ctrl);
Christoph Hellwig71102302016-07-06 21:55:52 +09001999
2000 return &ctrl->ctrl;
2001
2002out_remove_admin_queue:
Sagi Grimberg3f02fff2017-07-10 09:22:32 +03002003 nvme_rdma_destroy_admin_queue(ctrl, true);
Christoph Hellwig71102302016-07-06 21:55:52 +09002004out_kfree_queues:
2005 kfree(ctrl->queues);
2006out_uninit_ctrl:
2007 nvme_uninit_ctrl(&ctrl->ctrl);
2008 nvme_put_ctrl(&ctrl->ctrl);
2009 if (ret > 0)
2010 ret = -EIO;
2011 return ERR_PTR(ret);
2012out_free_ctrl:
2013 kfree(ctrl);
2014 return ERR_PTR(ret);
2015}
2016
2017static struct nvmf_transport_ops nvme_rdma_transport = {
2018 .name = "rdma",
2019 .required_opts = NVMF_OPT_TRADDR,
Max Gurtovoy8f4e8da2017-02-19 20:08:03 +02002020 .allowed_opts = NVMF_OPT_TRSVCID | NVMF_OPT_RECONNECT_DELAY |
Sagi Grimbergfd8563c2017-03-18 20:58:29 +02002021 NVMF_OPT_HOST_TRADDR | NVMF_OPT_CTRL_LOSS_TMO,
Christoph Hellwig71102302016-07-06 21:55:52 +09002022 .create_ctrl = nvme_rdma_create_ctrl,
2023};
2024
Steve Wisee87a9112016-09-02 09:01:54 -07002025static void nvme_rdma_remove_one(struct ib_device *ib_device, void *client_data)
2026{
2027 struct nvme_rdma_ctrl *ctrl;
2028
2029 /* Delete all controllers using this device */
2030 mutex_lock(&nvme_rdma_ctrl_mutex);
2031 list_for_each_entry(ctrl, &nvme_rdma_ctrl_list, list) {
2032 if (ctrl->device->dev != ib_device)
2033 continue;
2034 dev_info(ctrl->ctrl.device,
2035 "Removing ctrl: NQN \"%s\", addr %pISp\n",
2036 ctrl->ctrl.opts->subsysnqn, &ctrl->addr);
Christoph Hellwigc5017e82017-10-29 10:44:29 +02002037 nvme_delete_ctrl(&ctrl->ctrl);
Steve Wisee87a9112016-09-02 09:01:54 -07002038 }
2039 mutex_unlock(&nvme_rdma_ctrl_mutex);
2040
Sagi Grimberg9a6327d2017-06-07 20:31:55 +02002041 flush_workqueue(nvme_wq);
Steve Wisee87a9112016-09-02 09:01:54 -07002042}
2043
2044static struct ib_client nvme_rdma_ib_client = {
2045 .name = "nvme_rdma",
Steve Wisee87a9112016-09-02 09:01:54 -07002046 .remove = nvme_rdma_remove_one
2047};
2048
Christoph Hellwig71102302016-07-06 21:55:52 +09002049static int __init nvme_rdma_init_module(void)
2050{
Steve Wisee87a9112016-09-02 09:01:54 -07002051 int ret;
2052
Steve Wisee87a9112016-09-02 09:01:54 -07002053 ret = ib_register_client(&nvme_rdma_ib_client);
Sagi Grimberga56c79c2017-03-19 06:21:42 +02002054 if (ret)
Sagi Grimberg9a6327d2017-06-07 20:31:55 +02002055 return ret;
Steve Wisee87a9112016-09-02 09:01:54 -07002056
Sagi Grimberga56c79c2017-03-19 06:21:42 +02002057 ret = nvmf_register_transport(&nvme_rdma_transport);
2058 if (ret)
2059 goto err_unreg_client;
2060
2061 return 0;
2062
2063err_unreg_client:
2064 ib_unregister_client(&nvme_rdma_ib_client);
Sagi Grimberga56c79c2017-03-19 06:21:42 +02002065 return ret;
Christoph Hellwig71102302016-07-06 21:55:52 +09002066}
2067
2068static void __exit nvme_rdma_cleanup_module(void)
2069{
Christoph Hellwig71102302016-07-06 21:55:52 +09002070 nvmf_unregister_transport(&nvme_rdma_transport);
Steve Wisee87a9112016-09-02 09:01:54 -07002071 ib_unregister_client(&nvme_rdma_ib_client);
Christoph Hellwig71102302016-07-06 21:55:52 +09002072}
2073
2074module_init(nvme_rdma_init_module);
2075module_exit(nvme_rdma_cleanup_module);
2076
2077MODULE_LICENSE("GPL v2");