1 /*- 2 * BSD LICENSE 3 * 4 * Copyright (c) Intel Corporation. 5 * All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 11 * * Redistributions of source code must retain the above copyright 12 * notice, this list of conditions and the following disclaimer. 13 * * Redistributions in binary form must reproduce the above copyright 14 * notice, this list of conditions and the following disclaimer in 15 * the documentation and/or other materials provided with the 16 * distribution. 17 * * Neither the name of Intel Corporation nor the names of its 18 * contributors may be used to endorse or promote products derived 19 * from this software without specific prior written permission. 20 * 21 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 22 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 23 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 24 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 25 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 26 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 27 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 28 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 29 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 30 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 31 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 32 */ 33 34 #include "spdk/stdinc.h" 35 #include "spdk_cunit.h" 36 #include "common/lib/test_env.c" 37 #include "nvmf/rdma.c" 38 39 uint64_t g_mr_size; 40 struct ibv_mr g_rdma_mr; 41 42 #define RDMA_UT_UNITS_IN_MAX_IO 16 43 44 struct spdk_nvmf_transport_opts g_rdma_ut_transport_opts = { 45 .max_queue_depth = SPDK_NVMF_RDMA_DEFAULT_MAX_QUEUE_DEPTH, 46 .max_qpairs_per_ctrlr = SPDK_NVMF_RDMA_DEFAULT_MAX_QPAIRS_PER_CTRLR, 47 .in_capsule_data_size = SPDK_NVMF_RDMA_DEFAULT_IN_CAPSULE_DATA_SIZE, 48 .max_io_size = (SPDK_NVMF_RDMA_MIN_IO_BUFFER_SIZE * RDMA_UT_UNITS_IN_MAX_IO), 49 .io_unit_size = SPDK_NVMF_RDMA_MIN_IO_BUFFER_SIZE, 50 .max_aq_depth = SPDK_NVMF_RDMA_DEFAULT_AQ_DEPTH, 51 .num_shared_buffers = SPDK_NVMF_RDMA_DEFAULT_NUM_SHARED_BUFFERS, 52 }; 53 54 SPDK_LOG_REGISTER_COMPONENT("nvmf", SPDK_LOG_NVMF) 55 DEFINE_STUB(spdk_mem_map_set_translation, int, (struct spdk_mem_map *map, uint64_t vaddr, 56 uint64_t size, uint64_t translation), 0); 57 DEFINE_STUB(spdk_mem_map_clear_translation, int, (struct spdk_mem_map *map, uint64_t vaddr, 58 uint64_t size), 0); 59 DEFINE_STUB(spdk_mem_map_alloc, struct spdk_mem_map *, (uint64_t default_translation, 60 const struct spdk_mem_map_ops *ops, void *cb_ctx), NULL); 61 DEFINE_STUB(spdk_nvmf_qpair_disconnect, int, (struct spdk_nvmf_qpair *qpair, 62 nvmf_qpair_disconnect_cb cb_fn, void *ctx), 0); 63 DEFINE_STUB_V(spdk_mem_map_free, (struct spdk_mem_map **pmap)); 64 65 struct spdk_trace_histories *g_trace_histories; 66 DEFINE_STUB_V(spdk_trace_add_register_fn, (struct spdk_trace_register_fn *reg_fn)); 67 DEFINE_STUB_V(spdk_trace_register_object, (uint8_t type, char id_prefix)); 68 DEFINE_STUB_V(spdk_trace_register_description, (const char *name, const char *short_name, 69 uint16_t tpoint_id, uint8_t owner_type, uint8_t object_type, uint8_t new_object, 70 uint8_t arg1_is_ptr, const char *arg1_name)); 71 DEFINE_STUB_V(_spdk_trace_record, (uint64_t tsc, uint16_t tpoint_id, uint16_t poller_id, 72 uint32_t size, uint64_t object_id, uint64_t arg1)); 73 74 DEFINE_STUB_V(spdk_nvmf_request_exec, (struct spdk_nvmf_request *req)); 75 DEFINE_STUB(spdk_nvme_transport_id_compare, int, (const struct spdk_nvme_transport_id *trid1, 76 const struct spdk_nvme_transport_id *trid2), 0); 77 DEFINE_STUB_V(spdk_nvmf_ctrlr_abort_aer, (struct spdk_nvmf_ctrlr *ctrlr)); 78 79 uint64_t 80 spdk_mem_map_translate(const struct spdk_mem_map *map, uint64_t vaddr, uint64_t *size) 81 { 82 if (g_mr_size != 0) { 83 *(uint32_t *)size = g_mr_size; 84 } 85 86 return (uint64_t)&g_rdma_mr; 87 } 88 89 static void reset_nvmf_rdma_request(struct spdk_nvmf_rdma_request *rdma_req) 90 { 91 int i; 92 93 rdma_req->req.length = 0; 94 rdma_req->data_from_pool = false; 95 rdma_req->req.data = NULL; 96 rdma_req->data.wr.num_sge = 0; 97 rdma_req->data.wr.wr.rdma.remote_addr = 0; 98 rdma_req->data.wr.wr.rdma.rkey = 0; 99 100 for (i = 0; i < SPDK_NVMF_MAX_SGL_ENTRIES; i++) { 101 rdma_req->req.iov[i].iov_base = 0; 102 rdma_req->req.iov[i].iov_len = 0; 103 rdma_req->buffers[i] = 0; 104 rdma_req->data.wr.sg_list[i].addr = 0; 105 rdma_req->data.wr.sg_list[i].length = 0; 106 rdma_req->data.wr.sg_list[i].lkey = 0; 107 } 108 } 109 110 static void 111 test_spdk_nvmf_rdma_request_parse_sgl(void) 112 { 113 struct spdk_nvmf_rdma_transport rtransport; 114 struct spdk_nvmf_rdma_device device; 115 struct spdk_nvmf_rdma_request rdma_req; 116 struct spdk_nvmf_rdma_recv recv; 117 struct spdk_nvmf_rdma_poll_group group; 118 struct spdk_nvmf_rdma_qpair rqpair; 119 struct spdk_nvmf_rdma_poller poller; 120 union nvmf_c2h_msg cpl; 121 union nvmf_h2c_msg cmd; 122 struct spdk_nvme_sgl_descriptor *sgl; 123 struct spdk_nvmf_transport_pg_cache_buf bufs[4]; 124 int rc, i; 125 126 STAILQ_INIT(&group.group.buf_cache); 127 group.group.buf_cache_size = 0; 128 group.group.buf_cache_count = 0; 129 poller.group = &group; 130 rqpair.poller = &poller; 131 rqpair.max_send_sge = SPDK_NVMF_MAX_SGL_ENTRIES; 132 133 sgl = &cmd.nvme_cmd.dptr.sgl1; 134 rdma_req.recv = &recv; 135 rdma_req.req.cmd = &cmd; 136 rdma_req.req.rsp = &cpl; 137 rdma_req.data.wr.sg_list = rdma_req.data.sgl; 138 rdma_req.req.qpair = &rqpair.qpair; 139 rdma_req.req.xfer = SPDK_NVME_DATA_CONTROLLER_TO_HOST; 140 141 rtransport.transport.opts = g_rdma_ut_transport_opts; 142 rtransport.data_wr_pool = NULL; 143 rtransport.transport.data_buf_pool = NULL; 144 145 device.attr.device_cap_flags = 0; 146 g_rdma_mr.lkey = 0xABCD; 147 sgl->keyed.key = 0xEEEE; 148 sgl->address = 0xFFFF; 149 rdma_req.recv->buf = (void *)0xDDDD; 150 151 /* Test 1: sgl type: keyed data block subtype: address */ 152 sgl->generic.type = SPDK_NVME_SGL_TYPE_KEYED_DATA_BLOCK; 153 sgl->keyed.subtype = SPDK_NVME_SGL_SUBTYPE_ADDRESS; 154 155 /* Part 1: simple I/O, one SGL smaller than the transport io unit size */ 156 MOCK_SET(spdk_mempool_get, (void *)0x2000); 157 reset_nvmf_rdma_request(&rdma_req); 158 sgl->keyed.length = rtransport.transport.opts.io_unit_size / 2; 159 160 rc = spdk_nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req); 161 CU_ASSERT(rc == 0); 162 CU_ASSERT(rdma_req.data_from_pool == true); 163 CU_ASSERT(rdma_req.req.length == rtransport.transport.opts.io_unit_size / 2); 164 CU_ASSERT((uint64_t)rdma_req.req.data == 0x2000); 165 CU_ASSERT(rdma_req.data.wr.num_sge == 1); 166 CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0xEEEE); 167 CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0xFFFF); 168 CU_ASSERT((uint64_t)rdma_req.buffers[0] == 0x2000); 169 CU_ASSERT(rdma_req.data.wr.sg_list[0].addr == 0x2000); 170 CU_ASSERT(rdma_req.data.wr.sg_list[0].length == rtransport.transport.opts.io_unit_size / 2); 171 CU_ASSERT(rdma_req.data.wr.sg_list[0].lkey == g_rdma_mr.lkey); 172 173 /* Part 2: simple I/O, one SGL larger than the transport io unit size (equal to the max io size) */ 174 reset_nvmf_rdma_request(&rdma_req); 175 sgl->keyed.length = rtransport.transport.opts.io_unit_size * RDMA_UT_UNITS_IN_MAX_IO; 176 rc = spdk_nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req); 177 178 CU_ASSERT(rc == 0); 179 CU_ASSERT(rdma_req.data_from_pool == true); 180 CU_ASSERT(rdma_req.req.length == rtransport.transport.opts.io_unit_size * RDMA_UT_UNITS_IN_MAX_IO); 181 CU_ASSERT(rdma_req.data.wr.num_sge == RDMA_UT_UNITS_IN_MAX_IO); 182 CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0xEEEE); 183 CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0xFFFF); 184 for (i = 0; i < RDMA_UT_UNITS_IN_MAX_IO; i++) { 185 CU_ASSERT((uint64_t)rdma_req.buffers[i] == 0x2000); 186 CU_ASSERT(rdma_req.data.wr.sg_list[i].addr == 0x2000); 187 CU_ASSERT(rdma_req.data.wr.sg_list[i].length == rtransport.transport.opts.io_unit_size); 188 CU_ASSERT(rdma_req.data.wr.sg_list[i].lkey == g_rdma_mr.lkey); 189 } 190 191 /* Part 3: simple I/O one SGL larger than the transport max io size */ 192 reset_nvmf_rdma_request(&rdma_req); 193 sgl->keyed.length = rtransport.transport.opts.max_io_size * 2; 194 rc = spdk_nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req); 195 196 CU_ASSERT(rc == -1); 197 198 /* Part 4: Pretend there are no buffer pools */ 199 MOCK_SET(spdk_mempool_get, NULL); 200 reset_nvmf_rdma_request(&rdma_req); 201 sgl->keyed.length = rtransport.transport.opts.io_unit_size * RDMA_UT_UNITS_IN_MAX_IO; 202 rc = spdk_nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req); 203 204 CU_ASSERT(rc == 0); 205 CU_ASSERT(rdma_req.data_from_pool == false); 206 CU_ASSERT(rdma_req.req.data == NULL); 207 CU_ASSERT(rdma_req.data.wr.num_sge == 0); 208 CU_ASSERT(rdma_req.buffers[0] == NULL); 209 CU_ASSERT(rdma_req.data.wr.sg_list[0].addr == 0); 210 CU_ASSERT(rdma_req.data.wr.sg_list[0].length == 0); 211 CU_ASSERT(rdma_req.data.wr.sg_list[0].lkey == 0); 212 213 214 rdma_req.recv->buf = (void *)0xDDDD; 215 /* Test 2: sgl type: keyed data block subtype: offset (in capsule data) */ 216 sgl->generic.type = SPDK_NVME_SGL_TYPE_DATA_BLOCK; 217 sgl->unkeyed.subtype = SPDK_NVME_SGL_SUBTYPE_OFFSET; 218 219 /* Part 1: Normal I/O smaller than in capsule data size no offset */ 220 reset_nvmf_rdma_request(&rdma_req); 221 sgl->address = 0; 222 sgl->unkeyed.length = rtransport.transport.opts.in_capsule_data_size; 223 rc = spdk_nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req); 224 225 CU_ASSERT(rc == 0); 226 CU_ASSERT(rdma_req.req.data == (void *)0xDDDD); 227 CU_ASSERT(rdma_req.req.length == rtransport.transport.opts.in_capsule_data_size); 228 CU_ASSERT(rdma_req.data_from_pool == false); 229 230 /* Part 2: I/O offset + length too large */ 231 reset_nvmf_rdma_request(&rdma_req); 232 sgl->address = rtransport.transport.opts.in_capsule_data_size; 233 sgl->unkeyed.length = rtransport.transport.opts.in_capsule_data_size; 234 rc = spdk_nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req); 235 236 CU_ASSERT(rc == -1); 237 238 /* Part 3: I/O too large */ 239 reset_nvmf_rdma_request(&rdma_req); 240 sgl->address = 0; 241 sgl->unkeyed.length = rtransport.transport.opts.in_capsule_data_size * 2; 242 rc = spdk_nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req); 243 244 CU_ASSERT(rc == -1); 245 /* Test 3: use PG buffer cache */ 246 sgl->generic.type = SPDK_NVME_SGL_TYPE_KEYED_DATA_BLOCK; 247 sgl->keyed.subtype = SPDK_NVME_SGL_SUBTYPE_ADDRESS; 248 sgl->address = 0xFFFF; 249 rdma_req.recv->buf = (void *)0xDDDD; 250 g_rdma_mr.lkey = 0xABCD; 251 sgl->keyed.key = 0xEEEE; 252 253 for (i = 0; i < 4; i++) { 254 STAILQ_INSERT_TAIL(&group.group.buf_cache, &bufs[i], link); 255 } 256 257 /* part 1: use the four buffers from the pg cache */ 258 259 group.group.buf_cache_size = 4; 260 group.group.buf_cache_count = 4; 261 MOCK_SET(spdk_mempool_get, (void *)0x2000); 262 reset_nvmf_rdma_request(&rdma_req); 263 sgl->keyed.length = rtransport.transport.opts.io_unit_size * 4; 264 rc = spdk_nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req); 265 266 SPDK_CU_ASSERT_FATAL(rc == 0); 267 CU_ASSERT(rdma_req.data_from_pool == true); 268 CU_ASSERT(rdma_req.req.length == rtransport.transport.opts.io_unit_size * 4); 269 CU_ASSERT((uint64_t)rdma_req.req.data == (((uint64_t)&bufs[0] + NVMF_DATA_BUFFER_MASK) & 270 ~NVMF_DATA_BUFFER_MASK)); 271 CU_ASSERT(rdma_req.data.wr.num_sge == 4); 272 CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0xEEEE); 273 CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0xFFFF); 274 CU_ASSERT(group.group.buf_cache_count == 0); 275 CU_ASSERT(STAILQ_EMPTY(&group.group.buf_cache)); 276 for (i = 0; i < 4; i++) { 277 CU_ASSERT((uint64_t)rdma_req.buffers[i] == (uint64_t)&bufs[i]); 278 CU_ASSERT(rdma_req.data.wr.sg_list[i].addr == (((uint64_t)&bufs[i] + NVMF_DATA_BUFFER_MASK) & 279 ~NVMF_DATA_BUFFER_MASK)); 280 CU_ASSERT(rdma_req.data.wr.sg_list[i].length == rtransport.transport.opts.io_unit_size); 281 } 282 /* part 2: now that we have used the buffers from the cache, try again. We should get mempool buffers. */ 283 284 reset_nvmf_rdma_request(&rdma_req); 285 rc = spdk_nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req); 286 287 SPDK_CU_ASSERT_FATAL(rc == 0); 288 CU_ASSERT(rdma_req.data_from_pool == true); 289 CU_ASSERT(rdma_req.req.length == rtransport.transport.opts.io_unit_size * 4); 290 CU_ASSERT((uint64_t)rdma_req.req.data == 0x2000); 291 CU_ASSERT(rdma_req.data.wr.num_sge == 4); 292 CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0xEEEE); 293 CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0xFFFF); 294 CU_ASSERT(group.group.buf_cache_count == 0); 295 CU_ASSERT(STAILQ_EMPTY(&group.group.buf_cache)); 296 for (i = 0; i < 4; i++) { 297 CU_ASSERT((uint64_t)rdma_req.buffers[i] == 0x2000); 298 CU_ASSERT(rdma_req.data.wr.sg_list[i].addr == 0x2000); 299 CU_ASSERT(rdma_req.data.wr.sg_list[i].length == rtransport.transport.opts.io_unit_size); 300 CU_ASSERT(group.group.buf_cache_count == 0); 301 } 302 303 /* part 3: half and half */ 304 group.group.buf_cache_count = 2; 305 306 for (i = 0; i < 2; i++) { 307 STAILQ_INSERT_TAIL(&group.group.buf_cache, &bufs[i], link); 308 } 309 reset_nvmf_rdma_request(&rdma_req); 310 rc = spdk_nvmf_rdma_request_parse_sgl(&rtransport, &device, &rdma_req); 311 312 SPDK_CU_ASSERT_FATAL(rc == 0); 313 CU_ASSERT(rdma_req.data_from_pool == true); 314 CU_ASSERT(rdma_req.req.length == rtransport.transport.opts.io_unit_size * 4); 315 CU_ASSERT((uint64_t)rdma_req.req.data == (((uint64_t)&bufs[0] + NVMF_DATA_BUFFER_MASK) & 316 ~NVMF_DATA_BUFFER_MASK)); 317 CU_ASSERT(rdma_req.data.wr.num_sge == 4); 318 CU_ASSERT(rdma_req.data.wr.wr.rdma.rkey == 0xEEEE); 319 CU_ASSERT(rdma_req.data.wr.wr.rdma.remote_addr == 0xFFFF); 320 CU_ASSERT(group.group.buf_cache_count == 0); 321 for (i = 0; i < 2; i++) { 322 CU_ASSERT((uint64_t)rdma_req.buffers[i] == (uint64_t)&bufs[i]); 323 CU_ASSERT(rdma_req.data.wr.sg_list[i].addr == (((uint64_t)&bufs[i] + NVMF_DATA_BUFFER_MASK) & 324 ~NVMF_DATA_BUFFER_MASK)); 325 CU_ASSERT(rdma_req.data.wr.sg_list[i].length == rtransport.transport.opts.io_unit_size); 326 } 327 for (i = 2; i < 4; i++) { 328 CU_ASSERT((uint64_t)rdma_req.buffers[i] == 0x2000); 329 CU_ASSERT(rdma_req.data.wr.sg_list[i].addr == 0x2000); 330 CU_ASSERT(rdma_req.data.wr.sg_list[i].length == rtransport.transport.opts.io_unit_size); 331 } 332 } 333 334 int main(int argc, char **argv) 335 { 336 CU_pSuite suite = NULL; 337 unsigned int num_failures; 338 339 if (CU_initialize_registry() != CUE_SUCCESS) { 340 return CU_get_error(); 341 } 342 343 suite = CU_add_suite("nvmf", NULL, NULL); 344 if (suite == NULL) { 345 CU_cleanup_registry(); 346 return CU_get_error(); 347 } 348 349 if ( 350 CU_add_test(suite, "test_parse_sgl", test_spdk_nvmf_rdma_request_parse_sgl) == NULL) { 351 CU_cleanup_registry(); 352 return CU_get_error(); 353 } 354 355 CU_basic_set_mode(CU_BRM_VERBOSE); 356 CU_basic_run_tests(); 357 num_failures = CU_get_number_of_failures(); 358 CU_cleanup_registry(); 359 return num_failures; 360 } 361