Branch data Line data Source code
1 : : /* SPDX-License-Identifier: BSD-3-Clause
2 : : * Copyright (C) 2018 Intel Corporation. All rights reserved.
3 : : * Copyright (c) 2019, 2021 Mellanox Technologies LTD. All rights reserved.
4 : : * Copyright (c) 2023 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
5 : : */
6 : :
7 : : #include "spdk/stdinc.h"
8 : : #include "spdk_internal/cunit.h"
9 : : #include "common/lib/test_env.c"
10 : : #include "common/lib/test_iobuf.c"
11 : : #include "common/lib/test_rdma.c"
12 : : #include "nvmf/rdma.c"
13 : : #include "nvmf/transport.c"
14 : :
15 : : #define RDMA_UT_UNITS_IN_MAX_IO 16
16 : :
17 : : struct spdk_nvmf_transport_opts g_rdma_ut_transport_opts = {
18 : : .max_queue_depth = SPDK_NVMF_RDMA_DEFAULT_MAX_QUEUE_DEPTH,
19 : : .max_qpairs_per_ctrlr = SPDK_NVMF_RDMA_DEFAULT_MAX_QPAIRS_PER_CTRLR,
20 : : .in_capsule_data_size = SPDK_NVMF_RDMA_DEFAULT_IN_CAPSULE_DATA_SIZE,
21 : : .max_io_size = (SPDK_NVMF_RDMA_MIN_IO_BUFFER_SIZE * RDMA_UT_UNITS_IN_MAX_IO),
22 : : .io_unit_size = SPDK_NVMF_RDMA_MIN_IO_BUFFER_SIZE,
23 : : .max_aq_depth = SPDK_NVMF_RDMA_DEFAULT_AQ_DEPTH,
24 : : .num_shared_buffers = SPDK_NVMF_RDMA_DEFAULT_NUM_SHARED_BUFFERS,
25 : : };
26 : :
27 : 5 : SPDK_LOG_REGISTER_COMPONENT(nvmf)
28 : 0 : DEFINE_STUB(spdk_mem_map_set_translation, int, (struct spdk_mem_map *map, uint64_t vaddr,
29 : : uint64_t size, uint64_t translation), 0);
30 : 0 : DEFINE_STUB(spdk_mem_map_clear_translation, int, (struct spdk_mem_map *map, uint64_t vaddr,
31 : : uint64_t size), 0);
32 : 0 : DEFINE_STUB(spdk_mem_map_alloc, struct spdk_mem_map *, (uint64_t default_translation,
33 : : const struct spdk_mem_map_ops *ops, void *cb_ctx), NULL);
34 : 0 : DEFINE_STUB(spdk_nvmf_qpair_disconnect, int, (struct spdk_nvmf_qpair *qpair,
35 : : nvmf_qpair_disconnect_cb cb_fn, void *ctx), 0);
36 : 0 : DEFINE_STUB(spdk_nvmf_qpair_get_listen_trid, int,
37 : : (struct spdk_nvmf_qpair *qpair, struct spdk_nvme_transport_id *trid), 0);
38 : 0 : DEFINE_STUB_V(spdk_mem_map_free, (struct spdk_mem_map **pmap));
39 : :
40 : 25 : DEFINE_STUB_V(spdk_nvmf_request_exec, (struct spdk_nvmf_request *req));
41 : 0 : DEFINE_STUB(spdk_nvmf_request_complete, int, (struct spdk_nvmf_request *req), 0);
42 : 0 : DEFINE_STUB(spdk_nvme_transport_id_compare, int, (const struct spdk_nvme_transport_id *trid1,
43 : : const struct spdk_nvme_transport_id *trid2), 0);
44 : 0 : DEFINE_STUB_V(spdk_nvmf_ctrlr_abort_aer, (struct spdk_nvmf_ctrlr *ctrlr));
45 [ - + ]: 30 : DEFINE_STUB(spdk_nvmf_request_get_dif_ctx, bool, (struct spdk_nvmf_request *req,
46 : : struct spdk_dif_ctx *dif_ctx), false);
47 : 0 : DEFINE_STUB_V(spdk_nvme_trid_populate_transport, (struct spdk_nvme_transport_id *trid,
48 : : enum spdk_nvme_transport_type trtype));
49 : 0 : DEFINE_STUB_V(spdk_nvmf_tgt_new_qpair, (struct spdk_nvmf_tgt *tgt, struct spdk_nvmf_qpair *qpair));
50 : 0 : DEFINE_STUB(nvmf_ctrlr_abort_request, int, (struct spdk_nvmf_request *req), 0);
51 : 0 : DEFINE_STUB(spdk_nvme_transport_id_adrfam_str, const char *, (enum spdk_nvmf_adrfam adrfam), NULL);
52 : 0 : DEFINE_STUB(ibv_dereg_mr, int, (struct ibv_mr *mr), 0);
53 : 10 : DEFINE_STUB(ibv_resize_cq, int, (struct ibv_cq *cq, int cqe), 0);
54 : 0 : DEFINE_STUB(spdk_mempool_lookup, struct spdk_mempool *, (const char *name), NULL);
55 : :
56 : : /* ibv_reg_mr can be a macro, need to undefine it */
57 : : #ifdef ibv_reg_mr
58 : : #undef ibv_reg_mr
59 : : #endif
60 : :
61 : : DEFINE_RETURN_MOCK(ibv_reg_mr, struct ibv_mr *);
62 : : struct ibv_mr *
63 : : ibv_reg_mr(struct ibv_pd *pd, void *addr, size_t length, int access)
64 : : {
65 [ # # # # ]: 0 : HANDLE_RETURN_MOCK(ibv_reg_mr);
66 [ # # ]: 0 : if (length > 0) {
67 : 0 : return &g_rdma_mr;
68 : : } else {
69 : 0 : return NULL;
70 : : }
71 : : }
72 : :
73 : : int
74 : : ibv_query_qp(struct ibv_qp *qp, struct ibv_qp_attr *attr,
75 : : int attr_mask, struct ibv_qp_init_attr *init_attr)
76 : : {
77 [ + + ]: 15 : if (qp == NULL) {
78 : 5 : return -1;
79 : : } else {
80 : 10 : attr->port_num = 80;
81 : :
82 [ + + ]: 10 : if (qp->state == IBV_QPS_ERR) {
83 : 5 : attr->qp_state = 10;
84 : : } else {
85 : 5 : attr->qp_state = IBV_QPS_INIT;
86 : : }
87 : :
88 : 10 : return 0;
89 : : }
90 : : }
91 : :
92 : : const char *
93 : 0 : spdk_nvme_transport_id_trtype_str(enum spdk_nvme_transport_type trtype)
94 : : {
95 [ # # # # ]: 0 : switch (trtype) {
96 : 0 : case SPDK_NVME_TRANSPORT_PCIE:
97 : 0 : return "PCIe";
98 : 0 : case SPDK_NVME_TRANSPORT_RDMA:
99 : 0 : return "RDMA";
100 : 0 : case SPDK_NVME_TRANSPORT_FC:
101 : 0 : return "FC";
102 : 0 : default:
103 : 0 : return NULL;
104 : : }
105 : : }
106 : :
107 : : int
108 : 0 : spdk_nvme_transport_id_populate_trstring(struct spdk_nvme_transport_id *trid, const char *trstring)
109 : : {
110 : : int len, i;
111 : :
112 [ # # ]: 0 : if (trstring == NULL) {
113 : 0 : return -EINVAL;
114 : : }
115 : :
116 [ # # ]: 0 : len = strnlen(trstring, SPDK_NVMF_TRSTRING_MAX_LEN);
117 [ # # ]: 0 : if (len == SPDK_NVMF_TRSTRING_MAX_LEN) {
118 : 0 : return -EINVAL;
119 : : }
120 : :
121 : : /* cast official trstring to uppercase version of input. */
122 [ # # ]: 0 : for (i = 0; i < len; i++) {
123 : 0 : trid->trstring[i] = toupper(trstring[i]);
124 : : }
125 : 0 : return 0;
126 : : }
127 : :
128 : : static void
129 : 110 : reset_nvmf_rdma_request(struct spdk_nvmf_rdma_request *rdma_req)
130 : : {
131 : : int i;
132 : :
133 : 110 : rdma_req->req.length = 0;
134 : 110 : rdma_req->req.data_from_pool = false;
135 : 110 : rdma_req->data.wr.num_sge = 0;
136 : 110 : rdma_req->data.wr.wr.rdma.remote_addr = 0;
137 : 110 : rdma_req->data.wr.wr.rdma.rkey = 0;
138 : 110 : rdma_req->offset = 0;
139 [ - + ]: 110 : memset(&rdma_req->req.dif, 0, sizeof(rdma_req->req.dif));
140 : :
141 [ + + ]: 1870 : for (i = 0; i < SPDK_NVMF_MAX_SGL_ENTRIES; i++) {
142 : 1760 : rdma_req->req.iov[i].iov_base = 0;
143 : 1760 : rdma_req->req.iov[i].iov_len = 0;
144 : 1760 : rdma_req->data.wr.sg_list[i].addr = 0;
145 : 1760 : rdma_req->data.wr.sg_list[i].length = 0;
146 : 1760 : rdma_req->data.wr.sg_list[i].lkey = 0;
147 : : }
148 : 110 : rdma_req->req.iovcnt = 0;
149 [ + + ]: 110 : if (rdma_req->req.stripped_data) {
150 : 20 : free(rdma_req->req.stripped_data);
151 : 20 : rdma_req->req.stripped_data = NULL;
152 : : }
153 : 110 : }
154 : :
155 : : static void
156 : 5 : test_spdk_nvmf_rdma_request_parse_sgl(void)
157 : : {
158 : 4 : struct spdk_nvmf_rdma_transport rtransport;
159 : 4 : struct spdk_nvmf_rdma_device device;
160 : 5 : struct spdk_nvmf_rdma_request rdma_req = {};
161 : 4 : struct spdk_nvmf_rdma_recv recv;
162 : 4 : struct spdk_nvmf_rdma_poll_group group;
163 : 4 : struct spdk_nvmf_rdma_qpair rqpair;
164 : 4 : struct spdk_nvmf_rdma_poller poller;
165 : 4 : union nvmf_c2h_msg cpl;
166 : 4 : union nvmf_h2c_msg cmd;
167 : : struct spdk_nvme_sgl_descriptor *sgl;
168 : 5 : struct spdk_nvme_sgl_descriptor sgl_desc[SPDK_NVMF_MAX_SGL_ENTRIES] = {{0}};
169 : 4 : struct spdk_nvmf_rdma_request_data data;
170 : : int rc, i;
171 : : uint32_t sgl_length;
172 : :
173 : 5 : data.wr.sg_list = data.sgl;
174 : 5 : group.group.transport = &rtransport.transport;
175 : 5 : poller.group = &group;
176 : 5 : rqpair.poller = &poller;
177 : 5 : rqpair.max_send_sge = SPDK_NVMF_MAX_SGL_ENTRIES;
178 : :
179 : 5 : sgl = &cmd.nvme_cmd.dptr.sgl1;
180 : 5 : rdma_req.recv = &recv;
181 : 5 : rdma_req.req.cmd = &cmd;
182 : 5 : rdma_req.req.rsp = &cpl;
183 : 5 : rdma_req.data.wr.sg_list = rdma_req.data.sgl;
184 : 5 : rdma_req.req.qpair = &rqpair.qpair;
185 : 5 : rdma_req.req.xfer = SPDK_NVME_DATA_CONTROLLER_TO_HOST;
186 : :
187 : 5 : rtransport.transport.opts = g_rdma_ut_transport_opts;
188 : 5 : rtransport.data_wr_pool = NULL;
189 : :
190 : 5 : device.attr.device_cap_flags = 0;
191 : 5 : sgl->keyed.key = 0xEEEE;
192 : 5 : sgl->address = 0xFFFF;
193 : 5 : rdma_req.recv->buf = (void *)0xDDDD;
194 : :
195 : : /* Test 1: sgl type: keyed data block subtype: address */
196 : 5 : sgl->generic.type = SPDK_NVME_SGL_TYPE_KEYED_DATA_BLOCK;
197 : 5 : sgl->keyed.subtype = SPDK_NVME_SGL_SUBTYPE_ADDRESS;
198 : :
199 : : /* Part 1: simple I/O, one SGL smaller than the transport io unit size */
200 : 5 : MOCK_SET(spdk_iobuf_get, (void *)0x2000);
201 : 5 : reset_nvmf_rdma_request(&rdma_req);
202 : 5 : sgl->keyed.length = rtransport.transport.opts.io_unit_size / 2;
203 : :
204 : 5 : device.map = (void *)0x0;
205 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
206 : 5 : CU_ASSERT(rc == 0);
207 : 5 : CU_ASSERT(rdma_req.req.data_from_pool == true);
208 : 5 : CU_ASSERT(rdma_req.req.length == rtransport.transport.opts.io_unit_size / 2);
209 : 5 : CU_ASSERT((uint64_t)rdma_req.req.iovcnt == 1);
210 : 5 : CU_ASSERT((uint64_t)rdma_req.req.iov[0].iov_base == 0x2000);
211 : 5 : CU_ASSERT(rdma_req.data.wr.num_sge == 1);
212 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0xEEEE);
213 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0xFFFF);
214 : 5 : CU_ASSERT((uint64_t)rdma_req.req.iov[0].iov_base == 0x2000);
215 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].addr == 0x2000);
216 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].length == rtransport.transport.opts.io_unit_size / 2);
217 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].lkey == RDMA_UT_LKEY);
218 : :
219 : : /* Part 2: simple I/O, one SGL larger than the transport io unit size (equal to the max io size) */
220 : 5 : reset_nvmf_rdma_request(&rdma_req);
221 : 5 : sgl->keyed.length = rtransport.transport.opts.io_unit_size * RDMA_UT_UNITS_IN_MAX_IO;
222 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
223 : :
224 : 5 : CU_ASSERT(rc == 0);
225 : 5 : CU_ASSERT(rdma_req.req.data_from_pool == true);
226 : 5 : CU_ASSERT(rdma_req.req.length == rtransport.transport.opts.io_unit_size * RDMA_UT_UNITS_IN_MAX_IO);
227 : 5 : CU_ASSERT(rdma_req.data.wr.num_sge == RDMA_UT_UNITS_IN_MAX_IO);
228 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0xEEEE);
229 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0xFFFF);
230 [ + + ]: 85 : for (i = 0; i < RDMA_UT_UNITS_IN_MAX_IO; i++) {
231 : 80 : CU_ASSERT((uint64_t)rdma_req.req.iov[i].iov_base == 0x2000);
232 : 80 : CU_ASSERT(rdma_req.data.wr.sg_list[i].addr == 0x2000);
233 : 80 : CU_ASSERT(rdma_req.data.wr.sg_list[i].length == rtransport.transport.opts.io_unit_size);
234 : 80 : CU_ASSERT(rdma_req.data.wr.sg_list[i].lkey == RDMA_UT_LKEY);
235 : : }
236 : :
237 : : /* Part 3: simple I/O one SGL larger than the transport max io size */
238 : 5 : reset_nvmf_rdma_request(&rdma_req);
239 : 5 : sgl->keyed.length = rtransport.transport.opts.max_io_size * 2;
240 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
241 : :
242 : 5 : CU_ASSERT(rc == -1);
243 : :
244 : : /* Part 4: Pretend there are no buffer pools */
245 : 5 : MOCK_SET(spdk_iobuf_get, NULL);
246 : 5 : reset_nvmf_rdma_request(&rdma_req);
247 : 5 : sgl->keyed.length = rtransport.transport.opts.io_unit_size * RDMA_UT_UNITS_IN_MAX_IO;
248 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
249 : :
250 : 5 : CU_ASSERT(rc == 0);
251 : 5 : CU_ASSERT(rdma_req.req.data_from_pool == false);
252 : 5 : CU_ASSERT(rdma_req.req.iovcnt == 0);
253 : 5 : CU_ASSERT(rdma_req.data.wr.num_sge == 0);
254 : 5 : CU_ASSERT(rdma_req.req.iov[0].iov_base == NULL);
255 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].addr == 0);
256 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].length == 0);
257 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].lkey == 0);
258 : :
259 : 5 : rdma_req.recv->buf = (void *)0xDDDD;
260 : : /* Test 2: sgl type: keyed data block subtype: offset (in capsule data) */
261 : 5 : sgl->generic.type = SPDK_NVME_SGL_TYPE_DATA_BLOCK;
262 : 5 : sgl->unkeyed.subtype = SPDK_NVME_SGL_SUBTYPE_OFFSET;
263 : :
264 : : /* Part 1: Normal I/O smaller than in capsule data size no offset */
265 : 5 : reset_nvmf_rdma_request(&rdma_req);
266 : 5 : sgl->address = 0;
267 : 5 : sgl->unkeyed.length = rtransport.transport.opts.in_capsule_data_size;
268 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
269 : :
270 : 5 : CU_ASSERT(rc == 0);
271 : 5 : CU_ASSERT(rdma_req.req.iovcnt == 1);
272 : 5 : CU_ASSERT(rdma_req.req.iov[0].iov_base == (void *)0xDDDD);
273 : 5 : CU_ASSERT(rdma_req.req.length == rtransport.transport.opts.in_capsule_data_size);
274 : 5 : CU_ASSERT(rdma_req.req.data_from_pool == false);
275 : :
276 : : /* Part 2: I/O offset + length too large */
277 : 5 : reset_nvmf_rdma_request(&rdma_req);
278 : 5 : sgl->address = rtransport.transport.opts.in_capsule_data_size;
279 : 5 : sgl->unkeyed.length = rtransport.transport.opts.in_capsule_data_size;
280 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
281 : :
282 : 5 : CU_ASSERT(rc == -1);
283 : :
284 : : /* Part 3: I/O too large */
285 : 5 : reset_nvmf_rdma_request(&rdma_req);
286 : 5 : sgl->address = 0;
287 : 5 : sgl->unkeyed.length = rtransport.transport.opts.in_capsule_data_size * 2;
288 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
289 : :
290 : 5 : CU_ASSERT(rc == -1);
291 : :
292 : : /* Test 3: Multi SGL */
293 : 5 : sgl->generic.type = SPDK_NVME_SGL_TYPE_LAST_SEGMENT;
294 : 5 : sgl->unkeyed.subtype = SPDK_NVME_SGL_SUBTYPE_OFFSET;
295 : 5 : sgl->address = 0;
296 : 5 : rdma_req.recv->buf = (void *)&sgl_desc;
297 : 5 : MOCK_SET(spdk_iobuf_get, &data);
298 : 5 : MOCK_SET(spdk_mempool_get, &data);
299 : :
300 : : /* part 1: 2 segments each with 1 wr. */
301 : 5 : reset_nvmf_rdma_request(&rdma_req);
302 : 5 : sgl->unkeyed.length = 2 * sizeof(struct spdk_nvme_sgl_descriptor);
303 [ + + ]: 15 : for (i = 0; i < 2; i++) {
304 : 10 : sgl_desc[i].keyed.type = SPDK_NVME_SGL_TYPE_KEYED_DATA_BLOCK;
305 : 10 : sgl_desc[i].keyed.subtype = SPDK_NVME_SGL_SUBTYPE_ADDRESS;
306 : 10 : sgl_desc[i].keyed.length = rtransport.transport.opts.io_unit_size;
307 : 10 : sgl_desc[i].address = 0x4000 + i * rtransport.transport.opts.io_unit_size;
308 : 10 : sgl_desc[i].keyed.key = 0x44;
309 : : }
310 : :
311 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
312 : :
313 : 5 : CU_ASSERT(rc == 0);
314 : 5 : CU_ASSERT(rdma_req.req.data_from_pool == true);
315 : 5 : CU_ASSERT(rdma_req.req.length == rtransport.transport.opts.io_unit_size * 2);
316 : 5 : CU_ASSERT(rdma_req.data.wr.num_sge == 1);
317 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0x44);
318 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0x4000);
319 : 5 : CU_ASSERT(rdma_req.data.wr.next == &data.wr);
320 : 5 : CU_ASSERT(data.wr.wr.rdma.rkey == 0x44);
321 : 5 : CU_ASSERT(data.wr.wr.rdma.remote_addr == 0x4000 + rtransport.transport.opts.io_unit_size);
322 : 5 : CU_ASSERT(data.wr.num_sge == 1);
323 : 5 : CU_ASSERT(data.wr.next == &rdma_req.rsp.wr);
324 : :
325 : : /* part 2: 2 segments, each with 1 wr containing 8 sge_elements */
326 : 5 : reset_nvmf_rdma_request(&rdma_req);
327 : 5 : sgl->unkeyed.length = 2 * sizeof(struct spdk_nvme_sgl_descriptor);
328 [ + + ]: 15 : for (i = 0; i < 2; i++) {
329 : 10 : sgl_desc[i].keyed.type = SPDK_NVME_SGL_TYPE_KEYED_DATA_BLOCK;
330 : 10 : sgl_desc[i].keyed.subtype = SPDK_NVME_SGL_SUBTYPE_ADDRESS;
331 : 10 : sgl_desc[i].keyed.length = rtransport.transport.opts.io_unit_size * 8;
332 : 10 : sgl_desc[i].address = 0x4000 + i * 8 * rtransport.transport.opts.io_unit_size;
333 : 10 : sgl_desc[i].keyed.key = 0x44;
334 : : }
335 : :
336 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
337 : :
338 : 5 : CU_ASSERT(rc == 0);
339 : 5 : CU_ASSERT(rdma_req.req.data_from_pool == true);
340 : 5 : CU_ASSERT(rdma_req.req.length == rtransport.transport.opts.io_unit_size * 16);
341 : 5 : CU_ASSERT(rdma_req.req.iovcnt == 16);
342 : 5 : CU_ASSERT(rdma_req.data.wr.num_sge == 8);
343 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0x44);
344 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0x4000);
345 : 5 : CU_ASSERT(rdma_req.data.wr.next == &data.wr);
346 : 5 : CU_ASSERT(data.wr.wr.rdma.rkey == 0x44);
347 : 5 : CU_ASSERT(data.wr.wr.rdma.remote_addr == 0x4000 + rtransport.transport.opts.io_unit_size * 8);
348 : 5 : CU_ASSERT(data.wr.num_sge == 8);
349 : 5 : CU_ASSERT(data.wr.next == &rdma_req.rsp.wr);
350 : :
351 : : /* part 3: 2 segments, one very large, one very small */
352 : 5 : reset_nvmf_rdma_request(&rdma_req);
353 [ + + ]: 15 : for (i = 0; i < 2; i++) {
354 : 10 : sgl_desc[i].keyed.type = SPDK_NVME_SGL_TYPE_KEYED_DATA_BLOCK;
355 : 10 : sgl_desc[i].keyed.subtype = SPDK_NVME_SGL_SUBTYPE_ADDRESS;
356 : 10 : sgl_desc[i].keyed.key = 0x44;
357 : : }
358 : :
359 : 5 : sgl_desc[0].keyed.length = rtransport.transport.opts.io_unit_size * 15 +
360 : 5 : rtransport.transport.opts.io_unit_size / 2;
361 : 5 : sgl_desc[0].address = 0x4000;
362 : 5 : sgl_desc[1].keyed.length = rtransport.transport.opts.io_unit_size / 2;
363 : 5 : sgl_desc[1].address = 0x4000 + rtransport.transport.opts.io_unit_size * 15 +
364 : 5 : rtransport.transport.opts.io_unit_size / 2;
365 : :
366 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
367 : :
368 : 5 : CU_ASSERT(rc == 0);
369 : 5 : CU_ASSERT(rdma_req.req.data_from_pool == true);
370 : 5 : CU_ASSERT(rdma_req.req.length == rtransport.transport.opts.io_unit_size * 16);
371 : 5 : CU_ASSERT(rdma_req.req.iovcnt == 16);
372 : 5 : CU_ASSERT(rdma_req.data.wr.num_sge == 16);
373 [ + + ]: 80 : for (i = 0; i < 15; i++) {
374 : 75 : CU_ASSERT(rdma_req.data.sgl[i].length == rtransport.transport.opts.io_unit_size);
375 : : }
376 : 5 : CU_ASSERT(rdma_req.data.sgl[15].length == rtransport.transport.opts.io_unit_size / 2);
377 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0x44);
378 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0x4000);
379 : 5 : CU_ASSERT(rdma_req.data.wr.next == &data.wr);
380 : 5 : CU_ASSERT(data.wr.wr.rdma.rkey == 0x44);
381 : 5 : CU_ASSERT(data.wr.wr.rdma.remote_addr == 0x4000 + rtransport.transport.opts.io_unit_size * 15 +
382 : : rtransport.transport.opts.io_unit_size / 2);
383 : 5 : CU_ASSERT(data.sgl[0].length == rtransport.transport.opts.io_unit_size / 2);
384 : 5 : CU_ASSERT(data.wr.num_sge == 1);
385 : 5 : CU_ASSERT(data.wr.next == &rdma_req.rsp.wr);
386 : :
387 : : /* part 4: 2 SGL descriptors, each length is transport buffer / 2
388 : : * 1 transport buffers should be allocated */
389 : 5 : reset_nvmf_rdma_request(&rdma_req);
390 : 5 : sgl->unkeyed.length = 2 * sizeof(struct spdk_nvme_sgl_descriptor);
391 : 5 : sgl_length = rtransport.transport.opts.io_unit_size / 2;
392 [ + + ]: 15 : for (i = 0; i < 2; i++) {
393 : 10 : sgl_desc[i].keyed.length = sgl_length;
394 : 10 : sgl_desc[i].address = 0x4000 + i * sgl_length;
395 : : }
396 : :
397 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
398 : :
399 : 5 : CU_ASSERT(rc == 0);
400 : 5 : CU_ASSERT(rdma_req.req.data_from_pool == true);
401 : 5 : CU_ASSERT(rdma_req.req.length == rtransport.transport.opts.io_unit_size);
402 : 5 : CU_ASSERT(rdma_req.req.iovcnt == 1);
403 : :
404 : 5 : CU_ASSERT(rdma_req.data.sgl[0].length == sgl_length);
405 : : /* We mocked mempool_get to return address of data variable. Mempool is used
406 : : * to get both additional WRs and data buffers, so data points to &data */
407 : 5 : CU_ASSERT(rdma_req.data.sgl[0].addr == (uint64_t)&data);
408 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0x44);
409 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0x4000);
410 : 5 : CU_ASSERT(rdma_req.data.wr.num_sge == 1);
411 : 5 : CU_ASSERT(rdma_req.data.wr.next == &data.wr);
412 : :
413 : 5 : CU_ASSERT(data.wr.wr.rdma.rkey == 0x44);
414 : 5 : CU_ASSERT(data.wr.wr.rdma.remote_addr == 0x4000 + sgl_length);
415 : 5 : CU_ASSERT(data.sgl[0].length == sgl_length);
416 : 5 : CU_ASSERT(data.sgl[0].addr == (uint64_t)&data + sgl_length);
417 : 5 : CU_ASSERT(data.wr.num_sge == 1);
418 : :
419 : 5 : MOCK_CLEAR(spdk_mempool_get);
420 : 5 : MOCK_CLEAR(spdk_iobuf_get);
421 : :
422 : 5 : reset_nvmf_rdma_request(&rdma_req);
423 : 5 : }
424 : :
425 : : static struct spdk_nvmf_rdma_recv *
426 : 30 : create_recv(struct spdk_nvmf_rdma_qpair *rqpair, enum spdk_nvme_nvm_opcode opc)
427 : : {
428 : : struct spdk_nvmf_rdma_recv *rdma_recv;
429 : : union nvmf_h2c_msg *cmd;
430 : : struct spdk_nvme_sgl_descriptor *sgl;
431 : :
432 : 30 : rdma_recv = calloc(1, sizeof(*rdma_recv));
433 : 30 : rdma_recv->qpair = rqpair;
434 : 30 : cmd = calloc(1, sizeof(*cmd));
435 : 30 : rdma_recv->sgl[0].addr = (uintptr_t)cmd;
436 : 30 : cmd->nvme_cmd.opc = opc;
437 : 30 : sgl = &cmd->nvme_cmd.dptr.sgl1;
438 : 30 : sgl->keyed.key = 0xEEEE;
439 : 30 : sgl->address = 0xFFFF;
440 : 30 : sgl->keyed.type = SPDK_NVME_SGL_TYPE_KEYED_DATA_BLOCK;
441 : 30 : sgl->keyed.subtype = SPDK_NVME_SGL_SUBTYPE_ADDRESS;
442 : 30 : sgl->keyed.length = 1;
443 : :
444 : 30 : return rdma_recv;
445 : : }
446 : :
447 : : static void
448 : 30 : free_recv(struct spdk_nvmf_rdma_recv *rdma_recv)
449 : : {
450 : 30 : free((void *)rdma_recv->sgl[0].addr);
451 : 30 : free(rdma_recv);
452 : 30 : }
453 : :
454 : : static struct spdk_nvmf_rdma_request *
455 : 30 : create_req(struct spdk_nvmf_rdma_qpair *rqpair,
456 : : struct spdk_nvmf_rdma_recv *rdma_recv)
457 : : {
458 : : struct spdk_nvmf_rdma_request *rdma_req;
459 : : union nvmf_c2h_msg *cpl;
460 : :
461 : 30 : rdma_req = calloc(1, sizeof(*rdma_req));
462 : 30 : rdma_req->recv = rdma_recv;
463 : 30 : rdma_req->req.qpair = &rqpair->qpair;
464 : 30 : rdma_req->state = RDMA_REQUEST_STATE_NEW;
465 : 30 : rdma_req->data.wr.wr_id = (uintptr_t)&rdma_req->data_wr;
466 : 30 : rdma_req->data.wr.sg_list = rdma_req->data.sgl;
467 : 30 : cpl = calloc(1, sizeof(*cpl));
468 : 30 : rdma_req->rsp.sgl[0].addr = (uintptr_t)cpl;
469 : 30 : rdma_req->req.rsp = cpl;
470 : :
471 : 30 : return rdma_req;
472 : : }
473 : :
474 : : static void
475 : 30 : free_req(struct spdk_nvmf_rdma_request *rdma_req)
476 : : {
477 : 30 : free((void *)rdma_req->rsp.sgl[0].addr);
478 : 30 : free(rdma_req);
479 : 30 : }
480 : :
481 : : static void
482 : 30 : qpair_reset(struct spdk_nvmf_rdma_qpair *rqpair,
483 : : struct spdk_nvmf_rdma_poller *poller,
484 : : struct spdk_nvmf_rdma_device *device,
485 : : struct spdk_nvmf_rdma_resources *resources,
486 : : struct spdk_nvmf_transport *transport)
487 : : {
488 [ - + ]: 30 : memset(rqpair, 0, sizeof(*rqpair));
489 : 30 : STAILQ_INIT(&rqpair->pending_rdma_write_queue);
490 : 30 : STAILQ_INIT(&rqpair->pending_rdma_read_queue);
491 : 30 : STAILQ_INIT(&rqpair->pending_rdma_send_queue);
492 : 30 : rqpair->poller = poller;
493 : 30 : rqpair->device = device;
494 : 30 : rqpair->resources = resources;
495 : 30 : rqpair->qpair.qid = 1;
496 : 30 : rqpair->ibv_state = IBV_QPS_RTS;
497 : 30 : rqpair->qpair.state = SPDK_NVMF_QPAIR_ACTIVE;
498 : 30 : rqpair->max_send_sge = SPDK_NVMF_MAX_SGL_ENTRIES;
499 : 30 : rqpair->max_send_depth = 16;
500 : 30 : rqpair->max_read_depth = 16;
501 : 30 : rqpair->qpair.transport = transport;
502 : 30 : }
503 : :
504 : : static void
505 : 30 : poller_reset(struct spdk_nvmf_rdma_poller *poller,
506 : : struct spdk_nvmf_rdma_poll_group *group)
507 : : {
508 [ - + ]: 30 : memset(poller, 0, sizeof(*poller));
509 : 30 : STAILQ_INIT(&poller->qpairs_pending_recv);
510 : 30 : STAILQ_INIT(&poller->qpairs_pending_send);
511 : 30 : poller->group = group;
512 : 30 : }
513 : :
514 : : static void
515 : 5 : test_spdk_nvmf_rdma_request_process(void)
516 : : {
517 : 5 : struct spdk_nvmf_rdma_transport rtransport = {};
518 : 5 : struct spdk_nvmf_rdma_poll_group group = {};
519 : 5 : struct spdk_nvmf_rdma_poller poller = {};
520 : 5 : struct spdk_nvmf_rdma_device device = {};
521 : 5 : struct spdk_nvmf_rdma_resources resources = {};
522 : 5 : struct spdk_nvmf_rdma_qpair rqpair = {};
523 : : struct spdk_nvmf_rdma_recv *rdma_recv;
524 : : struct spdk_nvmf_rdma_request *rdma_req;
525 : 5 : struct spdk_iobuf_channel ch = {};
526 : : bool progress;
527 : :
528 : 5 : group.group.buf_cache = &ch;
529 : :
530 : 5 : STAILQ_INIT(&group.group.pending_buf_queue);
531 : 5 : poller_reset(&poller, &group);
532 : 5 : qpair_reset(&rqpair, &poller, &device, &resources, &rtransport.transport);
533 : :
534 : 5 : rtransport.transport.opts = g_rdma_ut_transport_opts;
535 : 5 : rtransport.data_wr_pool = spdk_mempool_create("test_wr_pool", 128,
536 : : sizeof(struct spdk_nvmf_rdma_request_data),
537 : : 0, 0);
538 : 5 : MOCK_CLEAR(spdk_iobuf_get);
539 : :
540 : 5 : device.attr.device_cap_flags = 0;
541 : 5 : device.map = (void *)0x0;
542 : :
543 : : /* Test 1: single SGL READ request */
544 : 5 : rdma_recv = create_recv(&rqpair, SPDK_NVME_OPC_READ);
545 : 5 : rdma_req = create_req(&rqpair, rdma_recv);
546 : 5 : rqpair.current_recv_depth = 1;
547 : : /* NEW -> EXECUTING */
548 : 5 : progress = nvmf_rdma_request_process(&rtransport, rdma_req);
549 : 5 : CU_ASSERT(progress == true);
550 : 5 : CU_ASSERT(rdma_req->state == RDMA_REQUEST_STATE_EXECUTING);
551 : 5 : CU_ASSERT(rdma_req->req.xfer == SPDK_NVME_DATA_CONTROLLER_TO_HOST);
552 : : /* EXECUTED -> TRANSFERRING_C2H */
553 : 5 : rdma_req->state = RDMA_REQUEST_STATE_EXECUTED;
554 : 5 : progress = nvmf_rdma_request_process(&rtransport, rdma_req);
555 : 5 : CU_ASSERT(progress == true);
556 : 5 : CU_ASSERT(rdma_req->state == RDMA_REQUEST_STATE_TRANSFERRING_CONTROLLER_TO_HOST);
557 : 5 : CU_ASSERT(rdma_req->recv == NULL);
558 : : /* COMPLETED -> FREE */
559 : 5 : rdma_req->state = RDMA_REQUEST_STATE_COMPLETED;
560 : 5 : progress = nvmf_rdma_request_process(&rtransport, rdma_req);
561 : 5 : CU_ASSERT(progress == true);
562 : 5 : CU_ASSERT(rdma_req->state == RDMA_REQUEST_STATE_FREE);
563 : :
564 : 5 : free_recv(rdma_recv);
565 : 5 : free_req(rdma_req);
566 : 5 : poller_reset(&poller, &group);
567 : 5 : qpair_reset(&rqpair, &poller, &device, &resources, &rtransport.transport);
568 : :
569 : : /* Test 2: single SGL WRITE request */
570 : 5 : rdma_recv = create_recv(&rqpair, SPDK_NVME_OPC_WRITE);
571 : 5 : rdma_req = create_req(&rqpair, rdma_recv);
572 : 5 : rqpair.current_recv_depth = 1;
573 : : /* NEW -> TRANSFERRING_H2C */
574 : 5 : progress = nvmf_rdma_request_process(&rtransport, rdma_req);
575 : 5 : CU_ASSERT(progress == true);
576 : 5 : CU_ASSERT(rdma_req->state == RDMA_REQUEST_STATE_TRANSFERRING_HOST_TO_CONTROLLER);
577 : 5 : CU_ASSERT(rdma_req->req.xfer == SPDK_NVME_DATA_HOST_TO_CONTROLLER);
578 : 5 : STAILQ_INIT(&poller.qpairs_pending_send);
579 : : /* READY_TO_EXECUTE -> EXECUTING */
580 : 5 : rdma_req->state = RDMA_REQUEST_STATE_READY_TO_EXECUTE;
581 : 5 : progress = nvmf_rdma_request_process(&rtransport, rdma_req);
582 : 5 : CU_ASSERT(progress == true);
583 : 5 : CU_ASSERT(rdma_req->state == RDMA_REQUEST_STATE_EXECUTING);
584 : : /* EXECUTED -> COMPLETING */
585 : 5 : rdma_req->state = RDMA_REQUEST_STATE_EXECUTED;
586 : 5 : progress = nvmf_rdma_request_process(&rtransport, rdma_req);
587 : 5 : CU_ASSERT(progress == true);
588 : 5 : CU_ASSERT(rdma_req->state == RDMA_REQUEST_STATE_COMPLETING);
589 : 5 : CU_ASSERT(rdma_req->recv == NULL);
590 : : /* COMPLETED -> FREE */
591 : 5 : rdma_req->state = RDMA_REQUEST_STATE_COMPLETED;
592 : 5 : progress = nvmf_rdma_request_process(&rtransport, rdma_req);
593 : 5 : CU_ASSERT(progress == true);
594 : 5 : CU_ASSERT(rdma_req->state == RDMA_REQUEST_STATE_FREE);
595 : :
596 : 5 : free_recv(rdma_recv);
597 : 5 : free_req(rdma_req);
598 : 5 : poller_reset(&poller, &group);
599 : 5 : qpair_reset(&rqpair, &poller, &device, &resources, &rtransport.transport);
600 : :
601 : : /* Test 3: WRITE+WRITE ibv_send batching */
602 : : {
603 : : struct spdk_nvmf_rdma_recv *recv1, *recv2;
604 : : struct spdk_nvmf_rdma_request *req1, *req2;
605 : 5 : recv1 = create_recv(&rqpair, SPDK_NVME_OPC_WRITE);
606 : 5 : req1 = create_req(&rqpair, recv1);
607 : 5 : recv2 = create_recv(&rqpair, SPDK_NVME_OPC_WRITE);
608 : 5 : req2 = create_req(&rqpair, recv2);
609 : :
610 : : /* WRITE 1: NEW -> TRANSFERRING_H2C */
611 : 5 : rqpair.current_recv_depth = 1;
612 : 5 : nvmf_rdma_request_process(&rtransport, req1);
613 : 5 : CU_ASSERT(req1->state == RDMA_REQUEST_STATE_TRANSFERRING_HOST_TO_CONTROLLER);
614 : :
615 : : /* WRITE 2: NEW -> TRANSFERRING_H2C */
616 : 5 : rqpair.current_recv_depth = 2;
617 : 5 : nvmf_rdma_request_process(&rtransport, req2);
618 : 5 : CU_ASSERT(req2->state == RDMA_REQUEST_STATE_TRANSFERRING_HOST_TO_CONTROLLER);
619 : :
620 : 5 : STAILQ_INIT(&poller.qpairs_pending_send);
621 : :
622 : : /* WRITE 1 completes before WRITE 2 has finished RDMA reading */
623 : : /* WRITE 1: READY_TO_EXECUTE -> EXECUTING */
624 : 5 : req1->state = RDMA_REQUEST_STATE_READY_TO_EXECUTE;
625 : 5 : nvmf_rdma_request_process(&rtransport, req1);
626 : 5 : CU_ASSERT(req1->state == RDMA_REQUEST_STATE_EXECUTING);
627 : : /* WRITE 1: EXECUTED -> COMPLETING */
628 : 5 : req1->state = RDMA_REQUEST_STATE_EXECUTED;
629 : 5 : nvmf_rdma_request_process(&rtransport, req1);
630 : 5 : CU_ASSERT(req1->state == RDMA_REQUEST_STATE_COMPLETING);
631 : 5 : STAILQ_INIT(&poller.qpairs_pending_send);
632 : : /* WRITE 1: COMPLETED -> FREE */
633 : 5 : req1->state = RDMA_REQUEST_STATE_COMPLETED;
634 : 5 : nvmf_rdma_request_process(&rtransport, req1);
635 : 5 : CU_ASSERT(req1->state == RDMA_REQUEST_STATE_FREE);
636 : :
637 : : /* Now WRITE 2 has finished reading and completes */
638 : : /* WRITE 2: COMPLETED -> FREE */
639 : : /* WRITE 2: READY_TO_EXECUTE -> EXECUTING */
640 : 5 : req2->state = RDMA_REQUEST_STATE_READY_TO_EXECUTE;
641 : 5 : nvmf_rdma_request_process(&rtransport, req2);
642 : 5 : CU_ASSERT(req2->state == RDMA_REQUEST_STATE_EXECUTING);
643 : : /* WRITE 1: EXECUTED -> COMPLETING */
644 : 5 : req2->state = RDMA_REQUEST_STATE_EXECUTED;
645 : 5 : nvmf_rdma_request_process(&rtransport, req2);
646 : 5 : CU_ASSERT(req2->state == RDMA_REQUEST_STATE_COMPLETING);
647 : 5 : STAILQ_INIT(&poller.qpairs_pending_send);
648 : : /* WRITE 1: COMPLETED -> FREE */
649 : 5 : req2->state = RDMA_REQUEST_STATE_COMPLETED;
650 : 5 : nvmf_rdma_request_process(&rtransport, req2);
651 : 5 : CU_ASSERT(req2->state == RDMA_REQUEST_STATE_FREE);
652 : :
653 : 5 : free_recv(recv1);
654 : 5 : free_req(req1);
655 : 5 : free_recv(recv2);
656 : 5 : free_req(req2);
657 : 5 : poller_reset(&poller, &group);
658 : 5 : qpair_reset(&rqpair, &poller, &device, &resources, &rtransport.transport);
659 : : }
660 : :
661 : : /* Test 4, invalid command, check xfer type */
662 : : {
663 : : struct spdk_nvmf_rdma_recv *rdma_recv_inv;
664 : : struct spdk_nvmf_rdma_request *rdma_req_inv;
665 : : /* construct an opcode that specifies BIDIRECTIONAL transfer */
666 : 5 : uint8_t opc = 0x10 | SPDK_NVME_DATA_BIDIRECTIONAL;
667 : :
668 : 5 : rdma_recv_inv = create_recv(&rqpair, opc);
669 : 5 : rdma_req_inv = create_req(&rqpair, rdma_recv_inv);
670 : :
671 : : /* NEW -> RDMA_REQUEST_STATE_COMPLETING */
672 : 5 : rqpair.current_recv_depth = 1;
673 : 5 : progress = nvmf_rdma_request_process(&rtransport, rdma_req_inv);
674 : 5 : CU_ASSERT(progress == true);
675 : 5 : CU_ASSERT(rdma_req_inv->state == RDMA_REQUEST_STATE_COMPLETING);
676 : 5 : CU_ASSERT(rdma_req_inv->req.rsp->nvme_cpl.status.sct == SPDK_NVME_SCT_GENERIC);
677 : 5 : CU_ASSERT(rdma_req_inv->req.rsp->nvme_cpl.status.sc == SPDK_NVME_SC_INVALID_OPCODE);
678 : :
679 : : /* RDMA_REQUEST_STATE_COMPLETED -> FREE */
680 : 5 : rdma_req_inv->state = RDMA_REQUEST_STATE_COMPLETED;
681 : 5 : nvmf_rdma_request_process(&rtransport, rdma_req_inv);
682 : 5 : CU_ASSERT(rdma_req_inv->state == RDMA_REQUEST_STATE_FREE);
683 : :
684 : 5 : free_recv(rdma_recv_inv);
685 : 5 : free_req(rdma_req_inv);
686 : 5 : poller_reset(&poller, &group);
687 : 5 : qpair_reset(&rqpair, &poller, &device, &resources, &rtransport.transport);
688 : : }
689 : :
690 : : /* Test 5: Write response waits in queue */
691 : : {
692 : 5 : rdma_recv = create_recv(&rqpair, SPDK_NVME_OPC_WRITE);
693 : 5 : rdma_req = create_req(&rqpair, rdma_recv);
694 : 5 : rqpair.current_recv_depth = 1;
695 : : /* NEW -> TRANSFERRING_H2C */
696 : 5 : progress = nvmf_rdma_request_process(&rtransport, rdma_req);
697 : 5 : CU_ASSERT(progress == true);
698 : 5 : CU_ASSERT(rdma_req->state == RDMA_REQUEST_STATE_TRANSFERRING_HOST_TO_CONTROLLER);
699 : 5 : CU_ASSERT(rdma_req->req.xfer == SPDK_NVME_DATA_HOST_TO_CONTROLLER);
700 : 5 : STAILQ_INIT(&poller.qpairs_pending_send);
701 : : /* READY_TO_EXECUTE -> EXECUTING */
702 : 5 : rdma_req->state = RDMA_REQUEST_STATE_READY_TO_EXECUTE;
703 : 5 : progress = nvmf_rdma_request_process(&rtransport, rdma_req);
704 : 5 : CU_ASSERT(progress == true);
705 : 5 : CU_ASSERT(rdma_req->state == RDMA_REQUEST_STATE_EXECUTING);
706 : : /* EXECUTED -> COMPLETING */
707 : 5 : rdma_req->state = RDMA_REQUEST_STATE_EXECUTED;
708 : : /* Send queue is full */
709 : 5 : rqpair.current_send_depth = rqpair.max_send_depth;
710 : 5 : progress = nvmf_rdma_request_process(&rtransport, rdma_req);
711 : 5 : CU_ASSERT(progress == true);
712 : 5 : CU_ASSERT(rdma_req->state == RDMA_REQUEST_STATE_READY_TO_COMPLETE_PENDING);
713 : 5 : CU_ASSERT(rdma_req == STAILQ_FIRST(&rqpair.pending_rdma_send_queue));
714 : :
715 : : /* Send queue is still full */
716 : 5 : progress = nvmf_rdma_request_process(&rtransport, rdma_req);
717 : 5 : CU_ASSERT(progress == false);
718 : 5 : CU_ASSERT(rdma_req->state == RDMA_REQUEST_STATE_READY_TO_COMPLETE_PENDING);
719 : 5 : CU_ASSERT(rdma_req == STAILQ_FIRST(&rqpair.pending_rdma_send_queue));
720 : :
721 : : /* Slot is available */
722 : 5 : rqpair.current_send_depth = rqpair.max_send_depth - 1;
723 : 5 : progress = nvmf_rdma_request_process(&rtransport, rdma_req);
724 : 5 : CU_ASSERT(progress == true);
725 : 5 : CU_ASSERT(STAILQ_EMPTY(&rqpair.pending_rdma_send_queue));
726 : 5 : CU_ASSERT(rdma_req->state == RDMA_REQUEST_STATE_COMPLETING);
727 : 5 : CU_ASSERT(rdma_req->recv == NULL);
728 : : /* COMPLETED -> FREE */
729 : 5 : rdma_req->state = RDMA_REQUEST_STATE_COMPLETED;
730 : 5 : progress = nvmf_rdma_request_process(&rtransport, rdma_req);
731 : 5 : CU_ASSERT(progress == true);
732 : 5 : CU_ASSERT(rdma_req->state == RDMA_REQUEST_STATE_FREE);
733 : :
734 : 5 : free_recv(rdma_recv);
735 : 5 : free_req(rdma_req);
736 : 5 : poller_reset(&poller, &group);
737 : 5 : qpair_reset(&rqpair, &poller, &device, &resources, &rtransport.transport);
738 : :
739 : : }
740 : :
741 : 5 : spdk_mempool_free(rtransport.data_wr_pool);
742 : 5 : }
743 : :
744 : : #define TEST_GROUPS_COUNT 5
745 : : static void
746 : 5 : test_nvmf_rdma_get_optimal_poll_group(void)
747 : : {
748 : 5 : struct spdk_nvmf_rdma_transport rtransport = {};
749 : 5 : struct spdk_nvmf_transport *transport = &rtransport.transport;
750 : 5 : struct spdk_nvmf_rdma_qpair rqpair = {};
751 : 4 : struct spdk_nvmf_transport_poll_group *groups[TEST_GROUPS_COUNT];
752 : 4 : struct spdk_nvmf_rdma_poll_group *rgroups[TEST_GROUPS_COUNT];
753 : : struct spdk_nvmf_transport_poll_group *result;
754 : 5 : struct spdk_nvmf_poll_group group = {};
755 : : uint32_t i;
756 : :
757 : 5 : rqpair.qpair.transport = transport;
758 : 5 : TAILQ_INIT(&rtransport.poll_groups);
759 : :
760 [ + + ]: 30 : for (i = 0; i < TEST_GROUPS_COUNT; i++) {
761 : 25 : groups[i] = nvmf_rdma_poll_group_create(transport, NULL);
762 : 25 : CU_ASSERT(groups[i] != NULL);
763 : 25 : groups[i]->group = &group;
764 : 25 : rgroups[i] = SPDK_CONTAINEROF(groups[i], struct spdk_nvmf_rdma_poll_group, group);
765 : 25 : groups[i]->transport = transport;
766 : : }
767 : 5 : CU_ASSERT(rtransport.conn_sched.next_admin_pg == rgroups[0]);
768 : 5 : CU_ASSERT(rtransport.conn_sched.next_io_pg == rgroups[0]);
769 : :
770 : : /* Emulate connection of %TEST_GROUPS_COUNT% initiators - each creates 1 admin and 1 io qp */
771 [ + + ]: 30 : for (i = 0; i < TEST_GROUPS_COUNT; i++) {
772 : 25 : rqpair.qpair.qid = 0;
773 : 25 : result = nvmf_rdma_get_optimal_poll_group(&rqpair.qpair);
774 : 25 : CU_ASSERT(result == groups[i]);
775 : 25 : CU_ASSERT(rtransport.conn_sched.next_admin_pg == rgroups[(i + 1) % TEST_GROUPS_COUNT]);
776 : 25 : CU_ASSERT(rtransport.conn_sched.next_io_pg == rgroups[i]);
777 : :
778 : 25 : rqpair.qpair.qid = 1;
779 : 25 : result = nvmf_rdma_get_optimal_poll_group(&rqpair.qpair);
780 : 25 : CU_ASSERT(result == groups[i]);
781 : 25 : CU_ASSERT(rtransport.conn_sched.next_admin_pg == rgroups[(i + 1) % TEST_GROUPS_COUNT]);
782 : 25 : CU_ASSERT(rtransport.conn_sched.next_io_pg == rgroups[(i + 1) % TEST_GROUPS_COUNT]);
783 : : }
784 : : /* wrap around, admin/io pg point to the first pg
785 : : Destroy all poll groups except of the last one */
786 [ + + ]: 25 : for (i = 0; i < TEST_GROUPS_COUNT - 1; i++) {
787 : 20 : nvmf_rdma_poll_group_destroy(groups[i]);
788 : 20 : CU_ASSERT(rtransport.conn_sched.next_admin_pg == rgroups[i + 1]);
789 : 20 : CU_ASSERT(rtransport.conn_sched.next_io_pg == rgroups[i + 1]);
790 : : }
791 : :
792 : 5 : CU_ASSERT(rtransport.conn_sched.next_admin_pg == rgroups[TEST_GROUPS_COUNT - 1]);
793 : 5 : CU_ASSERT(rtransport.conn_sched.next_io_pg == rgroups[TEST_GROUPS_COUNT - 1]);
794 : :
795 : : /* Check that pointers to the next admin/io poll groups are not changed */
796 : 5 : rqpair.qpair.qid = 0;
797 : 5 : result = nvmf_rdma_get_optimal_poll_group(&rqpair.qpair);
798 : 5 : CU_ASSERT(result == groups[TEST_GROUPS_COUNT - 1]);
799 : 5 : CU_ASSERT(rtransport.conn_sched.next_admin_pg == rgroups[TEST_GROUPS_COUNT - 1]);
800 : 5 : CU_ASSERT(rtransport.conn_sched.next_io_pg == rgroups[TEST_GROUPS_COUNT - 1]);
801 : :
802 : 5 : rqpair.qpair.qid = 1;
803 : 5 : result = nvmf_rdma_get_optimal_poll_group(&rqpair.qpair);
804 : 5 : CU_ASSERT(result == groups[TEST_GROUPS_COUNT - 1]);
805 : 5 : CU_ASSERT(rtransport.conn_sched.next_admin_pg == rgroups[TEST_GROUPS_COUNT - 1]);
806 : 5 : CU_ASSERT(rtransport.conn_sched.next_io_pg == rgroups[TEST_GROUPS_COUNT - 1]);
807 : :
808 : : /* Remove the last poll group, check that pointers are NULL */
809 : 5 : nvmf_rdma_poll_group_destroy(groups[TEST_GROUPS_COUNT - 1]);
810 : 5 : CU_ASSERT(rtransport.conn_sched.next_admin_pg == NULL);
811 : 5 : CU_ASSERT(rtransport.conn_sched.next_io_pg == NULL);
812 : :
813 : : /* Request optimal poll group, result must be NULL */
814 : 5 : rqpair.qpair.qid = 0;
815 : 5 : result = nvmf_rdma_get_optimal_poll_group(&rqpair.qpair);
816 : 5 : CU_ASSERT(result == NULL);
817 : :
818 : 5 : rqpair.qpair.qid = 1;
819 : 5 : result = nvmf_rdma_get_optimal_poll_group(&rqpair.qpair);
820 : 5 : CU_ASSERT(result == NULL);
821 : 5 : }
822 : : #undef TEST_GROUPS_COUNT
823 : :
824 : : static void
825 : 5 : test_spdk_nvmf_rdma_request_parse_sgl_with_md(void)
826 : : {
827 : 4 : struct spdk_nvmf_rdma_transport rtransport;
828 : 4 : struct spdk_nvmf_rdma_device device;
829 : 5 : struct spdk_nvmf_rdma_request rdma_req = {};
830 : 4 : struct spdk_nvmf_rdma_recv recv;
831 : 4 : struct spdk_nvmf_rdma_poll_group group;
832 : 4 : struct spdk_nvmf_rdma_qpair rqpair;
833 : 4 : struct spdk_nvmf_rdma_poller poller;
834 : 4 : union nvmf_c2h_msg cpl;
835 : 4 : union nvmf_h2c_msg cmd;
836 : : struct spdk_nvme_sgl_descriptor *sgl;
837 : 5 : struct spdk_nvme_sgl_descriptor sgl_desc[SPDK_NVMF_MAX_SGL_ENTRIES] = {{0}};
838 : 4 : char data_buffer[8192];
839 : 5 : struct spdk_nvmf_rdma_request_data *data = (struct spdk_nvmf_rdma_request_data *)data_buffer;
840 : 4 : char data2_buffer[8192];
841 : 5 : struct spdk_nvmf_rdma_request_data *data2 = (struct spdk_nvmf_rdma_request_data *)data2_buffer;
842 : 5 : const uint32_t data_bs = 512;
843 : 5 : const uint32_t md_size = 8;
844 : : int rc, i;
845 : 4 : struct spdk_dif_ctx_init_ext_opts dif_opts;
846 : :
847 : 5 : MOCK_CLEAR(spdk_mempool_get);
848 : 5 : MOCK_CLEAR(spdk_iobuf_get);
849 : :
850 : 5 : data->wr.sg_list = data->sgl;
851 : 5 : group.group.transport = &rtransport.transport;
852 : 5 : poller.group = &group;
853 : 5 : rqpair.poller = &poller;
854 : 5 : rqpair.max_send_sge = SPDK_NVMF_MAX_SGL_ENTRIES;
855 : :
856 : 5 : sgl = &cmd.nvme_cmd.dptr.sgl1;
857 : 5 : rdma_req.recv = &recv;
858 : 5 : rdma_req.req.cmd = &cmd;
859 : 5 : rdma_req.req.rsp = &cpl;
860 : 5 : rdma_req.data.wr.sg_list = rdma_req.data.sgl;
861 : 5 : rdma_req.req.qpair = &rqpair.qpair;
862 : 5 : rdma_req.req.xfer = SPDK_NVME_DATA_CONTROLLER_TO_HOST;
863 : :
864 : 5 : rtransport.transport.opts = g_rdma_ut_transport_opts;
865 : 5 : rtransport.data_wr_pool = NULL;
866 : :
867 : 5 : device.attr.device_cap_flags = 0;
868 : 5 : device.map = NULL;
869 : 5 : sgl->keyed.key = 0xEEEE;
870 : 5 : sgl->address = 0xFFFF;
871 : 5 : rdma_req.recv->buf = (void *)0xDDDD;
872 : :
873 : : /* Test 1: sgl type: keyed data block subtype: address */
874 : 5 : sgl->generic.type = SPDK_NVME_SGL_TYPE_KEYED_DATA_BLOCK;
875 : 5 : sgl->keyed.subtype = SPDK_NVME_SGL_SUBTYPE_ADDRESS;
876 : :
877 : : /* Part 1: simple I/O, one SGL smaller than the transport io unit size, block size 512 */
878 : 5 : MOCK_SET(spdk_iobuf_get, (void *)0x2000);
879 : 5 : reset_nvmf_rdma_request(&rdma_req);
880 : 5 : dif_opts.size = SPDK_SIZEOF(&dif_opts, dif_pi_format);
881 : 5 : dif_opts.dif_pi_format = SPDK_DIF_PI_FORMAT_16;
882 : 5 : spdk_dif_ctx_init(&rdma_req.req.dif.dif_ctx, data_bs + md_size, md_size, true, false,
883 : : SPDK_DIF_TYPE1, SPDK_DIF_FLAGS_GUARD_CHECK | SPDK_DIF_FLAGS_REFTAG_CHECK,
884 : : 0, 0, 0, 0, 0, &dif_opts);
885 : 5 : rdma_req.req.dif_enabled = true;
886 : 5 : rtransport.transport.opts.io_unit_size = data_bs * 8;
887 : 5 : rdma_req.req.qpair->transport = &rtransport.transport;
888 : 5 : sgl->keyed.length = data_bs * 4;
889 : :
890 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
891 : :
892 : 5 : CU_ASSERT(rc == 0);
893 : 5 : CU_ASSERT(rdma_req.req.data_from_pool == true);
894 : 5 : CU_ASSERT(rdma_req.req.length == data_bs * 4);
895 : 5 : CU_ASSERT(rdma_req.req.dif.orig_length == rdma_req.req.length);
896 : 5 : CU_ASSERT(rdma_req.req.dif.elba_length == (data_bs + md_size) * 4);
897 : 5 : CU_ASSERT(rdma_req.req.iovcnt == 1);
898 : 5 : CU_ASSERT((uint64_t)rdma_req.req.iov[0].iov_base == 0x2000);
899 : 5 : CU_ASSERT(rdma_req.data.wr.num_sge == 1);
900 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0xEEEE);
901 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0xFFFF);
902 : 5 : CU_ASSERT((uint64_t)rdma_req.req.iov[0].iov_base == 0x2000);
903 : :
904 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].addr == 0x2000);
905 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].length == rdma_req.req.length);
906 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].lkey == RDMA_UT_LKEY);
907 : :
908 : : /* Part 2: simple I/O, one SGL equal to io unit size, io_unit_size is not aligned with md_size,
909 : : block size 512 */
910 : 5 : MOCK_SET(spdk_iobuf_get, (void *)0x2000);
911 : 5 : reset_nvmf_rdma_request(&rdma_req);
912 : 5 : spdk_dif_ctx_init(&rdma_req.req.dif.dif_ctx, data_bs + md_size, md_size, true, false,
913 : : SPDK_DIF_TYPE1, SPDK_DIF_FLAGS_GUARD_CHECK | SPDK_DIF_FLAGS_REFTAG_CHECK,
914 : : 0, 0, 0, 0, 0, &dif_opts);
915 : 5 : rdma_req.req.dif_enabled = true;
916 : 5 : rtransport.transport.opts.io_unit_size = data_bs * 4;
917 : 5 : sgl->keyed.length = data_bs * 4;
918 : :
919 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
920 : :
921 : 5 : CU_ASSERT(rc == 0);
922 : 5 : CU_ASSERT(rdma_req.req.data_from_pool == true);
923 : 5 : CU_ASSERT(rdma_req.req.length == data_bs * 4);
924 : 5 : CU_ASSERT(rdma_req.req.dif.orig_length == rdma_req.req.length);
925 : 5 : CU_ASSERT(rdma_req.req.dif.elba_length == (data_bs + md_size) * 4);
926 : 5 : CU_ASSERT(rdma_req.req.iovcnt == 2);
927 : 5 : CU_ASSERT((uint64_t)rdma_req.req.iov[0].iov_base == 0x2000);
928 : 5 : CU_ASSERT(rdma_req.data.wr.num_sge == 5);
929 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0xEEEE);
930 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0xFFFF);
931 : 5 : CU_ASSERT((uint64_t)rdma_req.req.iov[0].iov_base == 0x2000);
932 : :
933 [ + + ]: 20 : for (i = 0; i < 3; ++i) {
934 : 15 : CU_ASSERT(rdma_req.data.wr.sg_list[i].addr == 0x2000 + i * (data_bs + md_size));
935 : 15 : CU_ASSERT(rdma_req.data.wr.sg_list[i].length == data_bs);
936 : 15 : CU_ASSERT(rdma_req.data.wr.sg_list[i].lkey == RDMA_UT_LKEY);
937 : : }
938 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[3].addr == 0x2000 + 3 * (data_bs + md_size));
939 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[3].length == 488);
940 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[3].lkey == RDMA_UT_LKEY);
941 : :
942 : : /* 2nd buffer consumed */
943 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[4].addr == 0x2000);
944 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[4].length == 24);
945 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[4].lkey == RDMA_UT_LKEY);
946 : :
947 : : /* Part 3: simple I/O, one SGL equal io unit size, io_unit_size is equal to block size 512 bytes */
948 : 5 : MOCK_SET(spdk_iobuf_get, (void *)0x2000);
949 : 5 : reset_nvmf_rdma_request(&rdma_req);
950 : 5 : spdk_dif_ctx_init(&rdma_req.req.dif.dif_ctx, data_bs + md_size, md_size, true, false,
951 : : SPDK_DIF_TYPE1, SPDK_DIF_FLAGS_GUARD_CHECK | SPDK_DIF_FLAGS_REFTAG_CHECK,
952 : : 0, 0, 0, 0, 0, &dif_opts);
953 : 5 : rdma_req.req.dif_enabled = true;
954 : 5 : rtransport.transport.opts.io_unit_size = data_bs;
955 : 5 : sgl->keyed.length = data_bs;
956 : :
957 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
958 : :
959 : 5 : CU_ASSERT(rc == 0);
960 : 5 : CU_ASSERT(rdma_req.req.data_from_pool == true);
961 : 5 : CU_ASSERT(rdma_req.req.length == data_bs);
962 : 5 : CU_ASSERT(rdma_req.req.dif.orig_length == rdma_req.req.length);
963 : 5 : CU_ASSERT(rdma_req.req.dif.elba_length == data_bs + md_size);
964 : 5 : CU_ASSERT(rdma_req.req.iovcnt == 2);
965 : 5 : CU_ASSERT((uint64_t)rdma_req.req.iov[0].iov_base == 0x2000);
966 : 5 : CU_ASSERT(rdma_req.data.wr.num_sge == 1);
967 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0xEEEE);
968 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0xFFFF);
969 : 5 : CU_ASSERT((uint64_t)rdma_req.req.iov[0].iov_base == 0x2000);
970 : :
971 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].addr == 0x2000);
972 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].length == data_bs);
973 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].lkey == RDMA_UT_LKEY);
974 : :
975 : 5 : CU_ASSERT(rdma_req.req.iovcnt == 2);
976 : 5 : CU_ASSERT(rdma_req.req.iov[0].iov_base == (void *)((unsigned long)0x2000));
977 : 5 : CU_ASSERT(rdma_req.req.iov[0].iov_len == data_bs);
978 : : /* 2nd buffer consumed for metadata */
979 : 5 : CU_ASSERT(rdma_req.req.iov[1].iov_base == (void *)((unsigned long)0x2000));
980 : 5 : CU_ASSERT(rdma_req.req.iov[1].iov_len == md_size);
981 : :
982 : : /* Part 4: simple I/O, one SGL equal io unit size, io_unit_size is aligned with md_size,
983 : : block size 512 */
984 : 5 : MOCK_SET(spdk_iobuf_get, (void *)0x2000);
985 : 5 : reset_nvmf_rdma_request(&rdma_req);
986 : 5 : spdk_dif_ctx_init(&rdma_req.req.dif.dif_ctx, data_bs + md_size, md_size, true, false,
987 : : SPDK_DIF_TYPE1, SPDK_DIF_FLAGS_GUARD_CHECK | SPDK_DIF_FLAGS_REFTAG_CHECK,
988 : : 0, 0, 0, 0, 0, &dif_opts);
989 : 5 : rdma_req.req.dif_enabled = true;
990 : 5 : rtransport.transport.opts.io_unit_size = (data_bs + md_size) * 4;
991 : 5 : sgl->keyed.length = data_bs * 4;
992 : :
993 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
994 : :
995 : 5 : CU_ASSERT(rc == 0);
996 : 5 : CU_ASSERT(rdma_req.req.data_from_pool == true);
997 : 5 : CU_ASSERT(rdma_req.req.length == data_bs * 4);
998 : 5 : CU_ASSERT(rdma_req.req.dif.orig_length == rdma_req.req.length);
999 : 5 : CU_ASSERT(rdma_req.req.dif.elba_length == (data_bs + md_size) * 4);
1000 : 5 : CU_ASSERT(rdma_req.req.iovcnt == 1);
1001 : 5 : CU_ASSERT((uint64_t)rdma_req.req.iov[0].iov_base == 0x2000);
1002 : 5 : CU_ASSERT(rdma_req.data.wr.num_sge == 1);
1003 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0xEEEE);
1004 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0xFFFF);
1005 : 5 : CU_ASSERT((uint64_t)rdma_req.req.iov[0].iov_base == 0x2000);
1006 : :
1007 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].addr == 0x2000);
1008 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].length == rdma_req.req.length);
1009 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].lkey == RDMA_UT_LKEY);
1010 : :
1011 : : /* Part 5: simple I/O, one SGL equal to 2x io unit size, io_unit_size is aligned with md_size,
1012 : : block size 512 */
1013 : 5 : MOCK_SET(spdk_iobuf_get, (void *)0x2000);
1014 : 5 : reset_nvmf_rdma_request(&rdma_req);
1015 : 5 : spdk_dif_ctx_init(&rdma_req.req.dif.dif_ctx, data_bs + md_size, md_size, true, false,
1016 : : SPDK_DIF_TYPE1, SPDK_DIF_FLAGS_GUARD_CHECK | SPDK_DIF_FLAGS_REFTAG_CHECK,
1017 : : 0, 0, 0, 0, 0, &dif_opts);
1018 : 5 : rdma_req.req.dif_enabled = true;
1019 : 5 : rtransport.transport.opts.io_unit_size = (data_bs + md_size) * 2;
1020 : 5 : sgl->keyed.length = data_bs * 4;
1021 : :
1022 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
1023 : :
1024 : 5 : CU_ASSERT(rc == 0);
1025 : 5 : CU_ASSERT(rdma_req.req.data_from_pool == true);
1026 : 5 : CU_ASSERT(rdma_req.req.length == data_bs * 4);
1027 : 5 : CU_ASSERT(rdma_req.req.dif.orig_length == rdma_req.req.length);
1028 : 5 : CU_ASSERT(rdma_req.req.dif.elba_length == (data_bs + md_size) * 4);
1029 : 5 : CU_ASSERT(rdma_req.req.iovcnt == 2);
1030 : 5 : CU_ASSERT((uint64_t)rdma_req.req.iov[0].iov_base == 0x2000);
1031 : 5 : CU_ASSERT(rdma_req.data.wr.num_sge == 2);
1032 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0xEEEE);
1033 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0xFFFF);
1034 : 5 : CU_ASSERT((uint64_t)rdma_req.req.iov[0].iov_base == 0x2000);
1035 : :
1036 [ + + ]: 15 : for (i = 0; i < 2; ++i) {
1037 : 10 : CU_ASSERT(rdma_req.data.wr.sg_list[i].addr == 0x2000);
1038 : 10 : CU_ASSERT(rdma_req.data.wr.sg_list[i].length == data_bs * 2);
1039 : : }
1040 : :
1041 : : /* Part 6: simple I/O, one SGL larger than the transport io unit size, io_unit_size is not aligned to md_size,
1042 : : block size 512 */
1043 : 5 : MOCK_SET(spdk_iobuf_get, (void *)0x2000);
1044 : 5 : reset_nvmf_rdma_request(&rdma_req);
1045 : 5 : spdk_dif_ctx_init(&rdma_req.req.dif.dif_ctx, data_bs + md_size, md_size, true, false,
1046 : : SPDK_DIF_TYPE1, SPDK_DIF_FLAGS_GUARD_CHECK | SPDK_DIF_FLAGS_REFTAG_CHECK,
1047 : : 0, 0, 0, 0, 0, &dif_opts);
1048 : 5 : rdma_req.req.dif_enabled = true;
1049 : 5 : rtransport.transport.opts.io_unit_size = data_bs * 4;
1050 : 5 : sgl->keyed.length = data_bs * 6;
1051 : :
1052 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
1053 : :
1054 : 5 : CU_ASSERT(rc == 0);
1055 : 5 : CU_ASSERT(rdma_req.req.data_from_pool == true);
1056 : 5 : CU_ASSERT(rdma_req.req.length == data_bs * 6);
1057 : 5 : CU_ASSERT(rdma_req.req.dif.orig_length == rdma_req.req.length);
1058 : 5 : CU_ASSERT(rdma_req.req.dif.elba_length == (data_bs + md_size) * 6);
1059 : 5 : CU_ASSERT(rdma_req.req.iovcnt == 2);
1060 : 5 : CU_ASSERT((uint64_t)rdma_req.req.iov[0].iov_base == 0x2000);
1061 : 5 : CU_ASSERT(rdma_req.data.wr.num_sge == 7);
1062 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0xEEEE);
1063 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0xFFFF);
1064 : 5 : CU_ASSERT((uint64_t)rdma_req.req.iov[0].iov_base == 0x2000);
1065 : :
1066 [ + + ]: 20 : for (i = 0; i < 3; ++i) {
1067 : 15 : CU_ASSERT(rdma_req.data.wr.sg_list[i].addr == 0x2000 + i * (data_bs + md_size));
1068 : 15 : CU_ASSERT(rdma_req.data.wr.sg_list[i].length == data_bs);
1069 : 15 : CU_ASSERT(rdma_req.data.wr.sg_list[i].lkey == RDMA_UT_LKEY);
1070 : : }
1071 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[3].addr == 0x2000 + 3 * (data_bs + md_size));
1072 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[3].length == 488);
1073 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[3].lkey == RDMA_UT_LKEY);
1074 : :
1075 : : /* 2nd IO buffer consumed */
1076 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[4].addr == 0x2000);
1077 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[4].length == 24);
1078 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[4].lkey == RDMA_UT_LKEY);
1079 : :
1080 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[5].addr == 0x2000 + 24 + md_size);
1081 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[5].length == 512);
1082 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[5].lkey == RDMA_UT_LKEY);
1083 : :
1084 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[6].addr == 0x2000 + 24 + 512 + md_size * 2);
1085 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[6].length == 512);
1086 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[6].lkey == RDMA_UT_LKEY);
1087 : :
1088 : : /* Part 7: simple I/O, number of SGL entries exceeds the number of entries
1089 : : one WR can hold. Additional WR is chained */
1090 : 5 : MOCK_SET(spdk_iobuf_get, data2_buffer);
1091 : 5 : MOCK_SET(spdk_mempool_get, data2_buffer);
1092 : 5 : reset_nvmf_rdma_request(&rdma_req);
1093 : 5 : spdk_dif_ctx_init(&rdma_req.req.dif.dif_ctx, data_bs + md_size, md_size, true, false,
1094 : : SPDK_DIF_TYPE1, SPDK_DIF_FLAGS_GUARD_CHECK | SPDK_DIF_FLAGS_REFTAG_CHECK,
1095 : : 0, 0, 0, 0, 0, &dif_opts);
1096 : 5 : rdma_req.req.dif_enabled = true;
1097 : 5 : rtransport.transport.opts.io_unit_size = data_bs * 16;
1098 : 5 : sgl->keyed.length = data_bs * 16;
1099 : :
1100 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
1101 : :
1102 : 5 : CU_ASSERT(rc == 0);
1103 : 5 : CU_ASSERT(rdma_req.req.data_from_pool == true);
1104 : 5 : CU_ASSERT(rdma_req.req.length == data_bs * 16);
1105 : 5 : CU_ASSERT(rdma_req.req.iovcnt == 2);
1106 : 5 : CU_ASSERT(rdma_req.req.dif.orig_length == rdma_req.req.length);
1107 : 5 : CU_ASSERT(rdma_req.req.dif.elba_length == (data_bs + md_size) * 16);
1108 : 5 : CU_ASSERT(rdma_req.req.iov[0].iov_base == data2_buffer);
1109 : 5 : CU_ASSERT(rdma_req.data.wr.num_sge == 16);
1110 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0xEEEE);
1111 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0xFFFF);
1112 : :
1113 [ + + ]: 80 : for (i = 0; i < 15; ++i) {
1114 : 75 : CU_ASSERT(rdma_req.data.wr.sg_list[i].addr == (uintptr_t)data2_buffer + i * (data_bs + md_size));
1115 : 75 : CU_ASSERT(rdma_req.data.wr.sg_list[i].length == data_bs);
1116 : 75 : CU_ASSERT(rdma_req.data.wr.sg_list[i].lkey == RDMA_UT_LKEY);
1117 : : }
1118 : :
1119 : : /* 8192 - (512 + 8) * 15 = 392 */
1120 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[i].addr == (uintptr_t)data2_buffer + i * (data_bs + md_size));
1121 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[i].length == 392);
1122 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[i].lkey == RDMA_UT_LKEY);
1123 : :
1124 : : /* additional wr from pool */
1125 : 5 : CU_ASSERT(rdma_req.data.wr.next == (void *)&data2->wr);
1126 : 5 : CU_ASSERT(rdma_req.data.wr.next->num_sge == 1);
1127 : 5 : CU_ASSERT(rdma_req.data.wr.next->next == &rdma_req.rsp.wr);
1128 : : /* 2nd IO buffer */
1129 : 5 : CU_ASSERT(data2->wr.sg_list[0].addr == (uintptr_t)data2_buffer);
1130 : 5 : CU_ASSERT(data2->wr.sg_list[0].length == 120);
1131 : 5 : CU_ASSERT(data2->wr.sg_list[0].lkey == RDMA_UT_LKEY);
1132 : :
1133 : : /* Part 8: simple I/O, data with metadata do not fit to 1 io_buffer */
1134 : 5 : MOCK_SET(spdk_iobuf_get, (void *)0x2000);
1135 : 5 : reset_nvmf_rdma_request(&rdma_req);
1136 : 5 : spdk_dif_ctx_init(&rdma_req.req.dif.dif_ctx, data_bs + md_size, md_size, true, false,
1137 : : SPDK_DIF_TYPE1, SPDK_DIF_FLAGS_GUARD_CHECK | SPDK_DIF_FLAGS_REFTAG_CHECK,
1138 : : 0, 0, 0, 0, 0, &dif_opts);
1139 : 5 : rdma_req.req.dif_enabled = true;
1140 : 5 : rtransport.transport.opts.io_unit_size = 516;
1141 : 5 : sgl->keyed.length = data_bs * 2;
1142 : :
1143 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
1144 : :
1145 : 5 : CU_ASSERT(rc == 0);
1146 : 5 : CU_ASSERT(rdma_req.req.data_from_pool == true);
1147 : 5 : CU_ASSERT(rdma_req.req.length == data_bs * 2);
1148 : 5 : CU_ASSERT(rdma_req.req.iovcnt == 3);
1149 : 5 : CU_ASSERT(rdma_req.req.dif.orig_length == rdma_req.req.length);
1150 : 5 : CU_ASSERT(rdma_req.req.dif.elba_length == (data_bs + md_size) * 2);
1151 : 5 : CU_ASSERT(rdma_req.req.iov[0].iov_base == (void *)0x2000);
1152 : 5 : CU_ASSERT(rdma_req.data.wr.num_sge == 2);
1153 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0xEEEE);
1154 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0xFFFF);
1155 : :
1156 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].addr == 0x2000);
1157 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].length == 512);
1158 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].lkey == RDMA_UT_LKEY);
1159 : :
1160 : : /* 2nd IO buffer consumed, offset 4 bytes due to part of the metadata
1161 : : is located at the beginning of that buffer */
1162 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[1].addr == 0x2000 + 4);
1163 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[1].length == 512);
1164 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[1].lkey == RDMA_UT_LKEY);
1165 : :
1166 : : /* Test 2: Multi SGL */
1167 : 5 : sgl->generic.type = SPDK_NVME_SGL_TYPE_LAST_SEGMENT;
1168 : 5 : sgl->unkeyed.subtype = SPDK_NVME_SGL_SUBTYPE_OFFSET;
1169 : 5 : sgl->address = 0;
1170 : 5 : rdma_req.recv->buf = (void *)&sgl_desc;
1171 : 5 : MOCK_SET(spdk_mempool_get, data_buffer);
1172 : 5 : MOCK_SET(spdk_iobuf_get, data_buffer);
1173 : :
1174 : : /* part 1: 2 segments each with 1 wr. io_unit_size is aligned with data_bs + md_size */
1175 : 5 : reset_nvmf_rdma_request(&rdma_req);
1176 : 5 : spdk_dif_ctx_init(&rdma_req.req.dif.dif_ctx, data_bs + md_size, md_size, true, false,
1177 : : SPDK_DIF_TYPE1,
1178 : : SPDK_DIF_FLAGS_GUARD_CHECK | SPDK_DIF_FLAGS_REFTAG_CHECK,
1179 : : 0, 0, 0, 0, 0, &dif_opts);
1180 : 5 : rdma_req.req.dif_enabled = true;
1181 : 5 : rtransport.transport.opts.io_unit_size = (data_bs + md_size) * 4;
1182 : 5 : sgl->unkeyed.length = 2 * sizeof(struct spdk_nvme_sgl_descriptor);
1183 : :
1184 [ + + ]: 15 : for (i = 0; i < 2; i++) {
1185 : 10 : sgl_desc[i].keyed.type = SPDK_NVME_SGL_TYPE_KEYED_DATA_BLOCK;
1186 : 10 : sgl_desc[i].keyed.subtype = SPDK_NVME_SGL_SUBTYPE_ADDRESS;
1187 : 10 : sgl_desc[i].keyed.length = data_bs * 4;
1188 : 10 : sgl_desc[i].address = 0x4000 + i * data_bs * 4;
1189 : 10 : sgl_desc[i].keyed.key = 0x44;
1190 : : }
1191 : :
1192 : 5 : rc = nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req);
1193 : :
1194 : 5 : CU_ASSERT(rc == 0);
1195 : 5 : CU_ASSERT(rdma_req.req.data_from_pool == true);
1196 : 5 : CU_ASSERT(rdma_req.req.length == data_bs * 4 * 2);
1197 : 5 : CU_ASSERT(rdma_req.req.dif.orig_length == rdma_req.req.length);
1198 : 5 : CU_ASSERT(rdma_req.req.dif.elba_length == (data_bs + md_size) * 4 * 2);
1199 : 5 : CU_ASSERT(rdma_req.data.wr.num_sge == 1);
1200 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].addr == (uintptr_t)(data_buffer));
1201 : 5 : CU_ASSERT(rdma_req.data.wr.sg_list[0].length == data_bs * 4);
1202 : :
1203 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0x44);
1204 : 5 : CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0x4000);
1205 : 5 : CU_ASSERT(rdma_req.data.wr.next == &data->wr);
1206 : 5 : CU_ASSERT(data->wr.wr.rdma.rkey == 0x44);
1207 : 5 : CU_ASSERT(data->wr.wr.rdma.remote_addr == 0x4000 + data_bs * 4);
1208 : 5 : CU_ASSERT(data->wr.num_sge == 1);
1209 : 5 : CU_ASSERT(data->wr.sg_list[0].addr == (uintptr_t)(data_buffer));
1210 : 5 : CU_ASSERT(data->wr.sg_list[0].length == data_bs * 4);
1211 : :
1212 : 5 : CU_ASSERT(data->wr.next == &rdma_req.rsp.wr);
1213 : 5 : reset_nvmf_rdma_request(&rdma_req);
1214 : 5 : }
1215 : :
1216 : : static void
1217 : 5 : test_nvmf_rdma_opts_init(void)
1218 : : {
1219 : 5 : struct spdk_nvmf_transport_opts opts = {};
1220 : :
1221 : 5 : nvmf_rdma_opts_init(&opts);
1222 : 5 : CU_ASSERT(opts.max_queue_depth == SPDK_NVMF_RDMA_DEFAULT_MAX_QUEUE_DEPTH);
1223 : 5 : CU_ASSERT(opts.max_qpairs_per_ctrlr == SPDK_NVMF_RDMA_DEFAULT_MAX_QPAIRS_PER_CTRLR);
1224 : 5 : CU_ASSERT(opts.in_capsule_data_size == SPDK_NVMF_RDMA_DEFAULT_IN_CAPSULE_DATA_SIZE);
1225 : 5 : CU_ASSERT(opts.max_io_size == SPDK_NVMF_RDMA_DEFAULT_MAX_IO_SIZE);
1226 : 5 : CU_ASSERT(opts.io_unit_size == SPDK_NVMF_RDMA_MIN_IO_BUFFER_SIZE);
1227 : 5 : CU_ASSERT(opts.max_aq_depth == SPDK_NVMF_RDMA_DEFAULT_AQ_DEPTH);
1228 : 5 : CU_ASSERT(opts.num_shared_buffers == SPDK_NVMF_RDMA_DEFAULT_NUM_SHARED_BUFFERS);
1229 : 5 : CU_ASSERT(opts.buf_cache_size == SPDK_NVMF_RDMA_DEFAULT_BUFFER_CACHE_SIZE);
1230 [ - + ]: 5 : CU_ASSERT(opts.dif_insert_or_strip == SPDK_NVMF_RDMA_DIF_INSERT_OR_STRIP);
1231 : 5 : CU_ASSERT(opts.abort_timeout_sec == SPDK_NVMF_RDMA_DEFAULT_ABORT_TIMEOUT_SEC);
1232 : 5 : CU_ASSERT(opts.transport_specific == NULL);
1233 : 5 : }
1234 : :
1235 : : static void
1236 : 5 : test_nvmf_rdma_request_free_data(void)
1237 : : {
1238 : 5 : struct spdk_nvmf_rdma_request rdma_req = {};
1239 : 5 : struct spdk_nvmf_rdma_transport rtransport = {};
1240 : 5 : struct spdk_nvmf_rdma_request_data *next_request_data = NULL;
1241 : :
1242 : 5 : MOCK_CLEAR(spdk_mempool_get);
1243 : 5 : rtransport.data_wr_pool = spdk_mempool_create("spdk_nvmf_rdma_wr_data",
1244 : : SPDK_NVMF_MAX_SGL_ENTRIES,
1245 : : sizeof(struct spdk_nvmf_rdma_request_data),
1246 : : SPDK_MEMPOOL_DEFAULT_CACHE_SIZE,
1247 : : SPDK_ENV_SOCKET_ID_ANY);
1248 : 5 : next_request_data = spdk_mempool_get(rtransport.data_wr_pool);
1249 [ - + ]: 5 : SPDK_CU_ASSERT_FATAL(((struct test_mempool *)rtransport.data_wr_pool)->count ==
1250 : : SPDK_NVMF_MAX_SGL_ENTRIES - 1);
1251 : 5 : next_request_data->wr.wr_id = (uint64_t)&rdma_req.data_wr;
1252 : 5 : next_request_data->wr.num_sge = 2;
1253 : 5 : next_request_data->wr.next = NULL;
1254 : 5 : rdma_req.data.wr.next = &next_request_data->wr;
1255 : 5 : rdma_req.data.wr.wr_id = (uint64_t)&rdma_req.data_wr;
1256 : 5 : rdma_req.data.wr.num_sge = 2;
1257 : 5 : rdma_req.transfer_wr = &rdma_req.data.wr;
1258 : :
1259 : 5 : nvmf_rdma_request_free_data(&rdma_req, &rtransport);
1260 : : /* Check if next_request_data put into memory pool */
1261 : 5 : CU_ASSERT(((struct test_mempool *)rtransport.data_wr_pool)->count == SPDK_NVMF_MAX_SGL_ENTRIES);
1262 : 5 : CU_ASSERT(rdma_req.data.wr.num_sge == 0);
1263 : :
1264 : 5 : spdk_mempool_free(rtransport.data_wr_pool);
1265 : 5 : }
1266 : :
1267 : : static void
1268 : 5 : test_nvmf_rdma_update_ibv_state(void)
1269 : : {
1270 : 5 : struct spdk_nvmf_rdma_qpair rqpair = {};
1271 : 5 : struct spdk_rdma_qp rdma_qp = {};
1272 : 5 : struct ibv_qp qp = {};
1273 : 5 : int rc = 0;
1274 : :
1275 : 5 : rqpair.rdma_qp = &rdma_qp;
1276 : :
1277 : : /* Case 1: Failed to get updated RDMA queue pair state */
1278 : 5 : rqpair.ibv_state = IBV_QPS_INIT;
1279 : 5 : rqpair.rdma_qp->qp = NULL;
1280 : :
1281 : 5 : rc = nvmf_rdma_update_ibv_state(&rqpair);
1282 : 5 : CU_ASSERT(rc == IBV_QPS_ERR + 1);
1283 : :
1284 : : /* Case 2: Bad state updated */
1285 : 5 : rqpair.rdma_qp->qp = &qp;
1286 : 5 : qp.state = IBV_QPS_ERR;
1287 : 5 : rc = nvmf_rdma_update_ibv_state(&rqpair);
1288 : 5 : CU_ASSERT(rqpair.ibv_state == 10);
1289 : 5 : CU_ASSERT(rc == IBV_QPS_ERR + 1);
1290 : :
1291 : : /* Case 3: Pass */
1292 : 5 : qp.state = IBV_QPS_INIT;
1293 : 5 : rc = nvmf_rdma_update_ibv_state(&rqpair);
1294 : 5 : CU_ASSERT(rqpair.ibv_state == IBV_QPS_INIT);
1295 : 5 : CU_ASSERT(rc == IBV_QPS_INIT);
1296 : 5 : }
1297 : :
1298 : : static void
1299 : 5 : test_nvmf_rdma_resources_create(void)
1300 : : {
1301 : : static struct spdk_nvmf_rdma_resources *rdma_resource;
1302 : 5 : struct spdk_nvmf_rdma_resource_opts opts = {};
1303 : 5 : struct spdk_nvmf_rdma_qpair qpair = {};
1304 : 5 : struct spdk_nvmf_rdma_recv *recv = NULL;
1305 : 5 : struct spdk_nvmf_rdma_request *req = NULL;
1306 : 5 : const int DEPTH = 128;
1307 : :
1308 : 5 : opts.max_queue_depth = DEPTH;
1309 : 5 : opts.in_capsule_data_size = 4096;
1310 : 5 : opts.shared = true;
1311 : 5 : opts.qpair = &qpair;
1312 : :
1313 : 5 : rdma_resource = nvmf_rdma_resources_create(&opts);
1314 : 5 : CU_ASSERT(rdma_resource != NULL);
1315 : : /* Just check first and last entry */
1316 : 5 : recv = &rdma_resource->recvs[0];
1317 : 5 : req = &rdma_resource->reqs[0];
1318 : 5 : CU_ASSERT(recv->rdma_wr.type == RDMA_WR_TYPE_RECV);
1319 : 5 : CU_ASSERT((uintptr_t)recv->buf == (uintptr_t)(rdma_resource->bufs));
1320 : 5 : CU_ASSERT(recv->sgl[0].addr == (uintptr_t)&rdma_resource->cmds[0]);
1321 : 5 : CU_ASSERT(recv->sgl[0].length == sizeof(rdma_resource->cmds[0]));
1322 : 5 : CU_ASSERT(recv->sgl[0].lkey == RDMA_UT_LKEY);
1323 : 5 : CU_ASSERT(recv->wr.num_sge == 2);
1324 : 5 : CU_ASSERT(recv->wr.wr_id == (uintptr_t)&rdma_resource->recvs[0].rdma_wr);
1325 : 5 : CU_ASSERT(recv->wr.sg_list == rdma_resource->recvs[0].sgl);
1326 : 5 : CU_ASSERT(req->req.rsp == &rdma_resource->cpls[0]);
1327 : 5 : CU_ASSERT(req->rsp.sgl[0].addr == (uintptr_t)&rdma_resource->cpls[0]);
1328 : 5 : CU_ASSERT(req->rsp.sgl[0].length == sizeof(rdma_resource->cpls[0]));
1329 : 5 : CU_ASSERT(req->rsp.sgl[0].lkey == RDMA_UT_LKEY);
1330 : 5 : CU_ASSERT(req->rsp_wr.type == RDMA_WR_TYPE_SEND);
1331 : 5 : CU_ASSERT(req->rsp.wr.wr_id == (uintptr_t)&rdma_resource->reqs[0].rsp_wr);
1332 : 5 : CU_ASSERT(req->rsp.wr.next == NULL);
1333 : 5 : CU_ASSERT(req->rsp.wr.opcode == IBV_WR_SEND);
1334 : 5 : CU_ASSERT(req->rsp.wr.send_flags == IBV_SEND_SIGNALED);
1335 : 5 : CU_ASSERT(req->rsp.wr.sg_list == rdma_resource->reqs[0].rsp.sgl);
1336 : 5 : CU_ASSERT(req->rsp.wr.num_sge == NVMF_DEFAULT_RSP_SGE);
1337 : 5 : CU_ASSERT(req->data_wr.type == RDMA_WR_TYPE_DATA);
1338 : 5 : CU_ASSERT(req->data.wr.wr_id == (uintptr_t)&rdma_resource->reqs[0].data_wr);
1339 : 5 : CU_ASSERT(req->data.wr.next == NULL);
1340 : 5 : CU_ASSERT(req->data.wr.send_flags == IBV_SEND_SIGNALED);
1341 : 5 : CU_ASSERT(req->data.wr.sg_list == rdma_resource->reqs[0].data.sgl);
1342 : 5 : CU_ASSERT(req->data.wr.num_sge == SPDK_NVMF_MAX_SGL_ENTRIES);
1343 : 5 : CU_ASSERT(req->state == RDMA_REQUEST_STATE_FREE);
1344 : :
1345 : 5 : recv = &rdma_resource->recvs[DEPTH - 1];
1346 : 5 : req = &rdma_resource->reqs[DEPTH - 1];
1347 : 5 : CU_ASSERT(recv->rdma_wr.type == RDMA_WR_TYPE_RECV);
1348 : 5 : CU_ASSERT((uintptr_t)recv->buf == (uintptr_t)(rdma_resource->bufs +
1349 : : (DEPTH - 1) * 4096));
1350 : 5 : CU_ASSERT(recv->sgl[0].addr == (uintptr_t)&rdma_resource->cmds[DEPTH - 1]);
1351 : 5 : CU_ASSERT(recv->sgl[0].length == sizeof(rdma_resource->cmds[DEPTH - 1]));
1352 : 5 : CU_ASSERT(recv->sgl[0].lkey == RDMA_UT_LKEY);
1353 : 5 : CU_ASSERT(recv->wr.num_sge == 2);
1354 : 5 : CU_ASSERT(recv->wr.wr_id == (uintptr_t)&rdma_resource->recvs[DEPTH - 1].rdma_wr);
1355 : 5 : CU_ASSERT(recv->wr.sg_list == rdma_resource->recvs[DEPTH - 1].sgl);
1356 : 5 : CU_ASSERT(req->req.rsp == &rdma_resource->cpls[DEPTH - 1]);
1357 : 5 : CU_ASSERT(req->rsp.sgl[0].addr == (uintptr_t)&rdma_resource->cpls[DEPTH - 1]);
1358 : 5 : CU_ASSERT(req->rsp.sgl[0].length == sizeof(rdma_resource->cpls[DEPTH - 1]));
1359 : 5 : CU_ASSERT(req->rsp.sgl[0].lkey == RDMA_UT_LKEY);
1360 : 5 : CU_ASSERT(req->rsp_wr.type == RDMA_WR_TYPE_SEND);
1361 : 5 : CU_ASSERT(req->rsp.wr.wr_id == (uintptr_t)&req->rsp_wr);
1362 : 5 : CU_ASSERT(req->rsp.wr.next == NULL);
1363 : 5 : CU_ASSERT(req->rsp.wr.opcode == IBV_WR_SEND);
1364 : 5 : CU_ASSERT(req->rsp.wr.send_flags == IBV_SEND_SIGNALED);
1365 : 5 : CU_ASSERT(req->rsp.wr.sg_list == rdma_resource->reqs[DEPTH - 1].rsp.sgl);
1366 : 5 : CU_ASSERT(req->rsp.wr.num_sge == NVMF_DEFAULT_RSP_SGE);
1367 : 5 : CU_ASSERT(req->data_wr.type == RDMA_WR_TYPE_DATA);
1368 : 5 : CU_ASSERT(req->data.wr.wr_id == (uintptr_t)&req->data_wr);
1369 : 5 : CU_ASSERT(req->data.wr.next == NULL);
1370 : 5 : CU_ASSERT(req->data.wr.send_flags == IBV_SEND_SIGNALED);
1371 : 5 : CU_ASSERT(req->data.wr.sg_list == rdma_resource->reqs[DEPTH - 1].data.sgl);
1372 : 5 : CU_ASSERT(req->data.wr.num_sge == SPDK_NVMF_MAX_SGL_ENTRIES);
1373 : 5 : CU_ASSERT(req->state == RDMA_REQUEST_STATE_FREE);
1374 : :
1375 : 5 : nvmf_rdma_resources_destroy(rdma_resource);
1376 : 5 : }
1377 : :
1378 : : static void
1379 : 5 : test_nvmf_rdma_qpair_compare(void)
1380 : : {
1381 : 5 : struct spdk_nvmf_rdma_qpair rqpair1 = {}, rqpair2 = {};
1382 : :
1383 : 5 : rqpair1.qp_num = 0;
1384 : 5 : rqpair2.qp_num = UINT32_MAX;
1385 : :
1386 : 5 : CU_ASSERT(nvmf_rdma_qpair_compare(&rqpair1, &rqpair2) < 0);
1387 : 5 : CU_ASSERT(nvmf_rdma_qpair_compare(&rqpair2, &rqpair1) > 0);
1388 : 5 : }
1389 : :
1390 : : static void
1391 : 5 : test_nvmf_rdma_resize_cq(void)
1392 : : {
1393 : 5 : int rc = -1;
1394 : 5 : int tnum_wr = 0;
1395 : 5 : int tnum_cqe = 0;
1396 : 5 : struct spdk_nvmf_rdma_qpair rqpair = {};
1397 : 5 : struct spdk_nvmf_rdma_poller rpoller = {};
1398 : 5 : struct spdk_nvmf_rdma_device rdevice = {};
1399 : 5 : struct ibv_context ircontext = {};
1400 : 5 : struct ibv_device idevice = {};
1401 : :
1402 : 5 : rdevice.context = &ircontext;
1403 : 5 : rqpair.poller = &rpoller;
1404 : 5 : ircontext.device = &idevice;
1405 : :
1406 : : /* Test1: Current capacity support required size. */
1407 : 5 : rpoller.required_num_wr = 10;
1408 : 5 : rpoller.num_cqe = 20;
1409 : 5 : rqpair.max_queue_depth = 2;
1410 : 5 : tnum_wr = rpoller.required_num_wr;
1411 : 5 : tnum_cqe = rpoller.num_cqe;
1412 : :
1413 : 5 : rc = nvmf_rdma_resize_cq(&rqpair, &rdevice);
1414 : 5 : CU_ASSERT(rc == 0);
1415 : 5 : CU_ASSERT(rpoller.required_num_wr == 10 + MAX_WR_PER_QP(rqpair.max_queue_depth));
1416 : 5 : CU_ASSERT(rpoller.required_num_wr > tnum_wr);
1417 : 5 : CU_ASSERT(rpoller.num_cqe == tnum_cqe);
1418 : :
1419 : : /* Test2: iWARP doesn't support CQ resize. */
1420 : 5 : tnum_wr = rpoller.required_num_wr;
1421 : 5 : tnum_cqe = rpoller.num_cqe;
1422 : 5 : idevice.transport_type = IBV_TRANSPORT_IWARP;
1423 : :
1424 : 5 : rc = nvmf_rdma_resize_cq(&rqpair, &rdevice);
1425 : 5 : CU_ASSERT(rc == -1);
1426 : 5 : CU_ASSERT(rpoller.required_num_wr == tnum_wr);
1427 : 5 : CU_ASSERT(rpoller.num_cqe == tnum_cqe);
1428 : :
1429 : :
1430 : : /* Test3: RDMA CQE requirement exceeds device max_cqe limitation. */
1431 : 5 : tnum_wr = rpoller.required_num_wr;
1432 : 5 : tnum_cqe = rpoller.num_cqe;
1433 : 5 : idevice.transport_type = IBV_TRANSPORT_UNKNOWN;
1434 : 5 : rdevice.attr.max_cqe = 3;
1435 : :
1436 : 5 : rc = nvmf_rdma_resize_cq(&rqpair, &rdevice);
1437 : 5 : CU_ASSERT(rc == -1);
1438 : 5 : CU_ASSERT(rpoller.required_num_wr == tnum_wr);
1439 : 5 : CU_ASSERT(rpoller.num_cqe == tnum_cqe);
1440 : :
1441 : : /* Test4: RDMA CQ resize failed. */
1442 : 5 : tnum_wr = rpoller.required_num_wr;
1443 : 5 : tnum_cqe = rpoller.num_cqe;
1444 : 5 : idevice.transport_type = IBV_TRANSPORT_IB;
1445 : 5 : rdevice.attr.max_cqe = 30;
1446 : 5 : MOCK_SET(ibv_resize_cq, -1);
1447 : :
1448 : 5 : rc = nvmf_rdma_resize_cq(&rqpair, &rdevice);
1449 : 5 : CU_ASSERT(rc == -1);
1450 : 5 : CU_ASSERT(rpoller.required_num_wr == tnum_wr);
1451 : 5 : CU_ASSERT(rpoller.num_cqe == tnum_cqe);
1452 : :
1453 : : /* Test5: RDMA CQ resize success. rsize = MIN(MAX(num_cqe * 2, required_num_wr), device->attr.max_cqe). */
1454 : 5 : tnum_wr = rpoller.required_num_wr;
1455 : 5 : tnum_cqe = rpoller.num_cqe;
1456 : 5 : MOCK_SET(ibv_resize_cq, 0);
1457 : :
1458 : 5 : rc = nvmf_rdma_resize_cq(&rqpair, &rdevice);
1459 : 5 : CU_ASSERT(rc == 0);
1460 : 5 : CU_ASSERT(rpoller.num_cqe = 30);
1461 : 5 : CU_ASSERT(rpoller.required_num_wr == 18 + MAX_WR_PER_QP(rqpair.max_queue_depth));
1462 : 5 : CU_ASSERT(rpoller.required_num_wr > tnum_wr);
1463 : 5 : CU_ASSERT(rpoller.num_cqe > tnum_cqe);
1464 : 5 : }
1465 : :
1466 : : int
1467 : 5 : main(int argc, char **argv)
1468 : : {
1469 : 5 : CU_pSuite suite = NULL;
1470 : : unsigned int num_failures;
1471 : :
1472 : 5 : CU_initialize_registry();
1473 : :
1474 : 5 : suite = CU_add_suite("nvmf", NULL, NULL);
1475 : :
1476 : 5 : CU_ADD_TEST(suite, test_spdk_nvmf_rdma_request_parse_sgl);
1477 : 5 : CU_ADD_TEST(suite, test_spdk_nvmf_rdma_request_process);
1478 : 5 : CU_ADD_TEST(suite, test_nvmf_rdma_get_optimal_poll_group);
1479 : 5 : CU_ADD_TEST(suite, test_spdk_nvmf_rdma_request_parse_sgl_with_md);
1480 : 5 : CU_ADD_TEST(suite, test_nvmf_rdma_opts_init);
1481 : 5 : CU_ADD_TEST(suite, test_nvmf_rdma_request_free_data);
1482 : 5 : CU_ADD_TEST(suite, test_nvmf_rdma_update_ibv_state);
1483 : 5 : CU_ADD_TEST(suite, test_nvmf_rdma_resources_create);
1484 : 5 : CU_ADD_TEST(suite, test_nvmf_rdma_qpair_compare);
1485 : 5 : CU_ADD_TEST(suite, test_nvmf_rdma_resize_cq);
1486 : :
1487 : 5 : num_failures = spdk_ut_run_tests(argc, argv, NULL);
1488 : 5 : CU_cleanup_registry();
1489 : 5 : return num_failures;
1490 : : }
|