Branch data Line data Source code
1 : : /* SPDX-License-Identifier: BSD-3-Clause
2 : : * Copyright (C) 2018 Intel Corporation.
3 : : * All rights reserved.
4 : : * Copyright (c) 2021, 2023 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
5 : : */
6 : :
7 : : #include "spdk/stdinc.h"
8 : : #include "spdk/nvmf_spec.h"
9 : : #include "spdk_internal/cunit.h"
10 : : #include "spdk/bdev_zone.h"
11 : :
12 : : #include "common/lib/test_env.c"
13 : : #include "common/lib/test_sock.c"
14 : :
15 : : #include "nvmf/ctrlr.c"
16 : : #include "nvmf/tcp.c"
17 : : #include "spdk/sock.h"
18 : : #include "spdk/hexlify.h"
19 : :
20 : : #define UT_IPV4_ADDR "192.168.0.1"
21 : : #define UT_PORT "4420"
22 : : #define UT_NVMF_ADRFAM_INVALID 0xf
23 : : #define UT_MAX_QUEUE_DEPTH 128
24 : : #define UT_MAX_QPAIRS_PER_CTRLR 128
25 : : #define UT_IN_CAPSULE_DATA_SIZE 1024
26 : : #define UT_MAX_IO_SIZE 4096
27 : : #define UT_IO_UNIT_SIZE 1024
28 : : #define UT_MAX_AQ_DEPTH 64
29 : : #define UT_SQ_HEAD_MAX 128
30 : : #define UT_NUM_SHARED_BUFFERS 128
31 : :
32 : : static void *g_accel_p = (void *)0xdeadbeaf;
33 : :
34 : 4 : SPDK_LOG_REGISTER_COMPONENT(nvmf)
35 : :
36 : 0 : DEFINE_STUB(spdk_nvmf_qpair_get_listen_trid,
37 : : int,
38 : : (struct spdk_nvmf_qpair *qpair, struct spdk_nvme_transport_id *trid),
39 : : 0);
40 : :
41 : 0 : DEFINE_STUB(nvmf_subsystem_add_ctrlr,
42 : : int,
43 : : (struct spdk_nvmf_subsystem *subsystem, struct spdk_nvmf_ctrlr *ctrlr),
44 : : 0);
45 : :
46 : 0 : DEFINE_STUB(nvmf_subsystem_get_ctrlr,
47 : : struct spdk_nvmf_ctrlr *,
48 : : (struct spdk_nvmf_subsystem *subsystem, uint16_t cntlid),
49 : : NULL);
50 : :
51 : 0 : DEFINE_STUB(spdk_nvmf_tgt_find_subsystem,
52 : : struct spdk_nvmf_subsystem *,
53 : : (struct spdk_nvmf_tgt *tgt, const char *subnqn),
54 : : NULL);
55 : :
56 [ # # ]: 0 : DEFINE_STUB(spdk_nvmf_subsystem_listener_allowed,
57 : : bool,
58 : : (struct spdk_nvmf_subsystem *subsystem, const struct spdk_nvme_transport_id *trid),
59 : : true);
60 : :
61 : 0 : DEFINE_STUB(nvmf_subsystem_find_listener,
62 : : struct spdk_nvmf_subsystem_listener *,
63 : : (struct spdk_nvmf_subsystem *subsystem,
64 : : const struct spdk_nvme_transport_id *trid),
65 : : (void *)0x1);
66 : :
67 : 0 : DEFINE_STUB_V(nvmf_get_discovery_log_page,
68 : : (struct spdk_nvmf_tgt *tgt, const char *hostnqn, struct iovec *iov,
69 : : uint32_t iovcnt, uint64_t offset, uint32_t length, struct spdk_nvme_transport_id *cmd_src_trid));
70 : :
71 : 0 : DEFINE_STUB_V(nvmf_subsystem_remove_ctrlr,
72 : : (struct spdk_nvmf_subsystem *subsystem, struct spdk_nvmf_ctrlr *ctrlr));
73 : :
74 : 0 : DEFINE_STUB(spdk_nvmf_subsystem_get_first_ns,
75 : : struct spdk_nvmf_ns *,
76 : : (struct spdk_nvmf_subsystem *subsystem),
77 : : NULL);
78 : :
79 : 0 : DEFINE_STUB(spdk_nvmf_subsystem_get_next_ns,
80 : : struct spdk_nvmf_ns *,
81 : : (struct spdk_nvmf_subsystem *subsystem, struct spdk_nvmf_ns *prev_ns),
82 : : NULL);
83 : :
84 [ # # ]: 0 : DEFINE_STUB(spdk_nvmf_subsystem_host_allowed,
85 : : bool,
86 : : (struct spdk_nvmf_subsystem *subsystem, const char *hostnqn),
87 : : true);
88 : :
89 [ # # ]: 0 : DEFINE_STUB(nvmf_ctrlr_dsm_supported,
90 : : bool,
91 : : (struct spdk_nvmf_ctrlr *ctrlr),
92 : : false);
93 : :
94 [ # # ]: 0 : DEFINE_STUB(nvmf_ctrlr_write_zeroes_supported,
95 : : bool,
96 : : (struct spdk_nvmf_ctrlr *ctrlr),
97 : : false);
98 : :
99 [ # # ]: 0 : DEFINE_STUB(nvmf_ctrlr_copy_supported,
100 : : bool,
101 : : (struct spdk_nvmf_ctrlr *ctrlr),
102 : : false);
103 : :
104 : 0 : DEFINE_STUB(nvmf_bdev_ctrlr_read_cmd,
105 : : int,
106 : : (struct spdk_bdev *bdev, struct spdk_bdev_desc *desc, struct spdk_io_channel *ch,
107 : : struct spdk_nvmf_request *req),
108 : : 0);
109 : :
110 : 0 : DEFINE_STUB(nvmf_bdev_ctrlr_write_cmd,
111 : : int,
112 : : (struct spdk_bdev *bdev, struct spdk_bdev_desc *desc, struct spdk_io_channel *ch,
113 : : struct spdk_nvmf_request *req),
114 : : 0);
115 : :
116 : 0 : DEFINE_STUB(nvmf_bdev_ctrlr_compare_cmd,
117 : : int,
118 : : (struct spdk_bdev *bdev, struct spdk_bdev_desc *desc, struct spdk_io_channel *ch,
119 : : struct spdk_nvmf_request *req),
120 : : 0);
121 : :
122 : 0 : DEFINE_STUB(nvmf_bdev_ctrlr_compare_and_write_cmd,
123 : : int,
124 : : (struct spdk_bdev *bdev, struct spdk_bdev_desc *desc, struct spdk_io_channel *ch,
125 : : struct spdk_nvmf_request *cmp_req, struct spdk_nvmf_request *write_req),
126 : : 0);
127 : :
128 : 0 : DEFINE_STUB(nvmf_bdev_ctrlr_write_zeroes_cmd,
129 : : int,
130 : : (struct spdk_bdev *bdev, struct spdk_bdev_desc *desc, struct spdk_io_channel *ch,
131 : : struct spdk_nvmf_request *req),
132 : : 0);
133 : :
134 : 0 : DEFINE_STUB(nvmf_bdev_ctrlr_flush_cmd,
135 : : int,
136 : : (struct spdk_bdev *bdev, struct spdk_bdev_desc *desc, struct spdk_io_channel *ch,
137 : : struct spdk_nvmf_request *req),
138 : : 0);
139 : :
140 : 0 : DEFINE_STUB(nvmf_bdev_ctrlr_dsm_cmd,
141 : : int,
142 : : (struct spdk_bdev *bdev, struct spdk_bdev_desc *desc, struct spdk_io_channel *ch,
143 : : struct spdk_nvmf_request *req),
144 : : 0);
145 : :
146 : 0 : DEFINE_STUB(nvmf_bdev_ctrlr_copy_cmd,
147 : : int,
148 : : (struct spdk_bdev *bdev, struct spdk_bdev_desc *desc, struct spdk_io_channel *ch,
149 : : struct spdk_nvmf_request *req),
150 : : 0);
151 : :
152 : 0 : DEFINE_STUB(nvmf_bdev_ctrlr_nvme_passthru_io,
153 : : int,
154 : : (struct spdk_bdev *bdev, struct spdk_bdev_desc *desc, struct spdk_io_channel *ch,
155 : : struct spdk_nvmf_request *req),
156 : : 0);
157 : :
158 : 0 : DEFINE_STUB(spdk_nvmf_bdev_ctrlr_abort_cmd,
159 : : int,
160 : : (struct spdk_bdev *bdev, struct spdk_bdev_desc *desc, struct spdk_io_channel *ch,
161 : : struct spdk_nvmf_request *req, struct spdk_nvmf_request *req_to_abort),
162 : : 0);
163 : :
164 [ # # ]: 0 : DEFINE_STUB(nvmf_bdev_ctrlr_get_dif_ctx,
165 : : bool,
166 : : (struct spdk_bdev *bdev, struct spdk_nvme_cmd *cmd, struct spdk_dif_ctx *dif_ctx),
167 : : false);
168 : :
169 : 0 : DEFINE_STUB(nvmf_transport_req_complete,
170 : : int,
171 : : (struct spdk_nvmf_request *req),
172 : : 0);
173 : :
174 [ # # ]: 0 : DEFINE_STUB(nvmf_bdev_zcopy_enabled,
175 : : bool,
176 : : (struct spdk_bdev *bdev),
177 : : false);
178 : :
179 : 0 : DEFINE_STUB(nvmf_bdev_ctrlr_zcopy_start,
180 : : int,
181 : : (struct spdk_bdev *bdev, struct spdk_bdev_desc *desc, struct spdk_io_channel *ch,
182 : : struct spdk_nvmf_request *req),
183 : : 0);
184 : :
185 : 0 : DEFINE_STUB_V(nvmf_bdev_ctrlr_zcopy_end, (struct spdk_nvmf_request *req, bool commit));
186 : :
187 : 0 : DEFINE_STUB_V(spdk_nvmf_request_free_buffers,
188 : : (struct spdk_nvmf_request *req, struct spdk_nvmf_transport_poll_group *group,
189 : : struct spdk_nvmf_transport *transport));
190 : :
191 : 0 : DEFINE_STUB(spdk_sock_get_optimal_sock_group,
192 : : int,
193 : : (struct spdk_sock *sock, struct spdk_sock_group **group, struct spdk_sock_group *hint),
194 : : 0);
195 : :
196 : 0 : DEFINE_STUB(spdk_sock_group_get_ctx,
197 : : void *,
198 : : (struct spdk_sock_group *group),
199 : : NULL);
200 : :
201 : 0 : DEFINE_STUB_V(nvmf_ns_reservation_request, (void *ctx));
202 : :
203 : 0 : DEFINE_STUB_V(spdk_nvme_trid_populate_transport, (struct spdk_nvme_transport_id *trid,
204 : : enum spdk_nvme_transport_type trtype));
205 : 4 : DEFINE_STUB_V(spdk_nvmf_transport_register, (const struct spdk_nvmf_transport_ops *ops));
206 : :
207 : 0 : DEFINE_STUB_V(spdk_nvmf_tgt_new_qpair, (struct spdk_nvmf_tgt *tgt, struct spdk_nvmf_qpair *qpair));
208 : :
209 : 0 : DEFINE_STUB_V(nvmf_transport_qpair_abort_request,
210 : : (struct spdk_nvmf_qpair *qpair, struct spdk_nvmf_request *req));
211 : :
212 : 0 : DEFINE_STUB_V(spdk_nvme_print_command, (uint16_t qid, struct spdk_nvme_cmd *cmd));
213 : 0 : DEFINE_STUB_V(spdk_nvme_print_completion, (uint16_t qid, struct spdk_nvme_cpl *cpl));
214 : :
215 : 0 : DEFINE_STUB(nvmf_transport_req_free,
216 : : int,
217 : : (struct spdk_nvmf_request *req),
218 : : 0);
219 : :
220 : 4 : DEFINE_STUB(accel_channel_create, int, (void *io_device, void *ctx_buf), 0);
221 : 4 : DEFINE_STUB_V(accel_channel_destroy, (void *io_device, void *ctx_buf));
222 : 0 : DEFINE_STUB(spdk_bdev_reset, int, (struct spdk_bdev_desc *desc, struct spdk_io_channel *ch,
223 : : spdk_bdev_io_completion_cb cb, void *cb_arg), 0);
224 : 0 : DEFINE_STUB_V(spdk_bdev_free_io, (struct spdk_bdev_io *bdev_io));
225 : :
226 : 0 : DEFINE_STUB(spdk_bdev_get_max_active_zones, uint32_t,
227 : : (const struct spdk_bdev *bdev), 0);
228 : 0 : DEFINE_STUB(spdk_bdev_get_max_open_zones, uint32_t,
229 : : (const struct spdk_bdev *bdev), 0);
230 [ # # ]: 0 : DEFINE_STUB(spdk_bdev_is_zoned, bool, (const struct spdk_bdev *bdev), false);
231 : 0 : DEFINE_STUB(spdk_bdev_get_zone_size, uint64_t, (const struct spdk_bdev *bdev), 0);
232 : :
233 : 0 : DEFINE_STUB(spdk_nvme_ns_get_format_index, uint32_t,
234 : : (const struct spdk_nvme_ns_data *nsdata), 0);
235 : :
236 : 0 : DEFINE_STUB(spdk_sock_get_impl_name, const char *, (struct spdk_sock *sock), "");
237 : :
238 [ # # ]: 0 : DEFINE_STUB(spdk_nvmf_subsystem_is_discovery, bool, (struct spdk_nvmf_subsystem *subsystem), false);
239 : :
240 [ # # ]: 0 : DEFINE_STUB(nvmf_ns_is_ptpl_capable, bool, (const struct spdk_nvmf_ns *ns), false);
241 : :
242 : : struct spdk_io_channel *
243 : 4 : spdk_accel_get_io_channel(void)
244 : : {
245 : 4 : return spdk_get_io_channel(g_accel_p);
246 : : }
247 : :
248 : 0 : DEFINE_STUB(spdk_accel_submit_crc32cv,
249 : : int,
250 : : (struct spdk_io_channel *ch, uint32_t *dst, struct iovec *iovs,
251 : : uint32_t iovcnt, uint32_t seed, spdk_accel_completion_cb cb_fn, void *cb_arg),
252 : : 0);
253 : :
254 : 0 : DEFINE_STUB(spdk_nvmf_bdev_ctrlr_nvme_passthru_admin,
255 : : int,
256 : : (struct spdk_bdev *bdev, struct spdk_bdev_desc *desc,
257 : : struct spdk_io_channel *ch, struct spdk_nvmf_request *req,
258 : : spdk_nvmf_nvme_passthru_cmd_cb cb_fn),
259 : : 0)
260 : :
261 : : struct spdk_bdev {
262 : : int ut_mock;
263 : : uint64_t blockcnt;
264 : : };
265 : :
266 : : int
267 : 0 : spdk_nvme_transport_id_compare(const struct spdk_nvme_transport_id *trid1,
268 : : const struct spdk_nvme_transport_id *trid2)
269 : : {
270 : 0 : return 0;
271 : : }
272 : :
273 : : const char *
274 : 0 : spdk_nvme_transport_id_trtype_str(enum spdk_nvme_transport_type trtype)
275 : : {
276 [ # # # # ]: 0 : switch (trtype) {
277 : 0 : case SPDK_NVME_TRANSPORT_PCIE:
278 : 0 : return "PCIe";
279 : 0 : case SPDK_NVME_TRANSPORT_RDMA:
280 : 0 : return "RDMA";
281 : 0 : case SPDK_NVME_TRANSPORT_FC:
282 : 0 : return "FC";
283 : 0 : default:
284 : 0 : return NULL;
285 : : }
286 : : }
287 : :
288 : : int
289 : 0 : spdk_nvme_transport_id_populate_trstring(struct spdk_nvme_transport_id *trid, const char *trstring)
290 : : {
291 : : int len, i;
292 : :
293 [ # # ]: 0 : if (trstring == NULL) {
294 : 0 : return -EINVAL;
295 : : }
296 : :
297 [ # # ]: 0 : len = strnlen(trstring, SPDK_NVMF_TRSTRING_MAX_LEN);
298 [ # # ]: 0 : if (len == SPDK_NVMF_TRSTRING_MAX_LEN) {
299 : 0 : return -EINVAL;
300 : : }
301 : :
302 : : /* cast official trstring to uppercase version of input. */
303 [ # # ]: 0 : for (i = 0; i < len; i++) {
304 : 0 : trid->trstring[i] = toupper(trstring[i]);
305 : : }
306 : 0 : return 0;
307 : : }
308 : :
309 : : int
310 : 0 : spdk_nvmf_qpair_disconnect(struct spdk_nvmf_qpair *qpair, nvmf_qpair_disconnect_cb cb_fn, void *ctx)
311 : : {
312 : 0 : return 0;
313 : : }
314 : :
315 : : int
316 : 8 : spdk_nvmf_request_get_buffers(struct spdk_nvmf_request *req,
317 : : struct spdk_nvmf_transport_poll_group *group,
318 : : struct spdk_nvmf_transport *transport,
319 : : uint32_t length)
320 : : {
321 : : /* length more than 1 io unit length will fail. */
322 [ + + ]: 8 : if (length >= transport->opts.io_unit_size) {
323 : 4 : return -EINVAL;
324 : : }
325 : :
326 : 4 : req->iovcnt = 1;
327 : 4 : req->iov[0].iov_base = (void *)0xDEADBEEF;
328 : :
329 : 4 : return 0;
330 : : }
331 : :
332 : :
333 : : void
334 : 0 : nvmf_bdev_ctrlr_identify_ns(struct spdk_nvmf_ns *ns, struct spdk_nvme_ns_data *nsdata,
335 : : bool dif_insert_or_strip)
336 : : {
337 : : uint64_t num_blocks;
338 : :
339 [ # # ]: 0 : SPDK_CU_ASSERT_FATAL(ns->bdev != NULL);
340 : 0 : num_blocks = ns->bdev->blockcnt;
341 : 0 : nsdata->nsze = num_blocks;
342 : 0 : nsdata->ncap = num_blocks;
343 : 0 : nsdata->nuse = num_blocks;
344 : 0 : nsdata->nlbaf = 0;
345 : 0 : nsdata->flbas.format = 0;
346 : 0 : nsdata->flbas.msb_format = 0;
347 : 0 : nsdata->lbaf[0].lbads = spdk_u32log2(512);
348 : 0 : }
349 : :
350 : : const char *
351 : 0 : spdk_nvmf_subsystem_get_sn(const struct spdk_nvmf_subsystem *subsystem)
352 : : {
353 : 0 : return subsystem->sn;
354 : : }
355 : :
356 : : const char *
357 : 0 : spdk_nvmf_subsystem_get_mn(const struct spdk_nvmf_subsystem *subsystem)
358 : : {
359 : 0 : return subsystem->mn;
360 : : }
361 : :
362 : : static void
363 : 4 : test_nvmf_tcp_create(void)
364 : : {
365 : : struct spdk_thread *thread;
366 : : struct spdk_nvmf_transport *transport;
367 : : struct spdk_nvmf_tcp_transport *ttransport;
368 : 4 : struct spdk_nvmf_transport_opts opts;
369 : :
370 : 4 : thread = spdk_thread_create(NULL, NULL);
371 [ - + ]: 4 : SPDK_CU_ASSERT_FATAL(thread != NULL);
372 : 4 : spdk_set_thread(thread);
373 : :
374 : : /* case 1 */
375 : 4 : memset(&opts, 0, sizeof(opts));
376 : 4 : opts.max_queue_depth = UT_MAX_QUEUE_DEPTH;
377 : 4 : opts.max_qpairs_per_ctrlr = UT_MAX_QPAIRS_PER_CTRLR;
378 : 4 : opts.in_capsule_data_size = UT_IN_CAPSULE_DATA_SIZE;
379 : 4 : opts.max_io_size = UT_MAX_IO_SIZE;
380 : 4 : opts.io_unit_size = UT_IO_UNIT_SIZE;
381 : 4 : opts.max_aq_depth = UT_MAX_AQ_DEPTH;
382 : 4 : opts.num_shared_buffers = UT_NUM_SHARED_BUFFERS;
383 : : /* expect success */
384 : 4 : transport = nvmf_tcp_create(&opts);
385 : 4 : CU_ASSERT_PTR_NOT_NULL(transport);
386 : 4 : ttransport = SPDK_CONTAINEROF(transport, struct spdk_nvmf_tcp_transport, transport);
387 [ - + ]: 4 : SPDK_CU_ASSERT_FATAL(ttransport != NULL);
388 : 4 : transport->opts = opts;
389 : 4 : CU_ASSERT(transport->opts.max_queue_depth == UT_MAX_QUEUE_DEPTH);
390 : 4 : CU_ASSERT(transport->opts.max_io_size == UT_MAX_IO_SIZE);
391 : 4 : CU_ASSERT(transport->opts.in_capsule_data_size == UT_IN_CAPSULE_DATA_SIZE);
392 : 4 : CU_ASSERT(transport->opts.io_unit_size == UT_IO_UNIT_SIZE);
393 : : /* destroy transport */
394 : 4 : CU_ASSERT(nvmf_tcp_destroy(transport, NULL, NULL) == 0);
395 : :
396 : : /* case 2 */
397 : 4 : memset(&opts, 0, sizeof(opts));
398 : 4 : opts.max_queue_depth = UT_MAX_QUEUE_DEPTH;
399 : 4 : opts.max_qpairs_per_ctrlr = UT_MAX_QPAIRS_PER_CTRLR;
400 : 4 : opts.in_capsule_data_size = UT_IN_CAPSULE_DATA_SIZE;
401 : 4 : opts.max_io_size = UT_MAX_IO_SIZE;
402 : 4 : opts.io_unit_size = UT_MAX_IO_SIZE + 1;
403 : 4 : opts.max_aq_depth = UT_MAX_AQ_DEPTH;
404 : 4 : opts.num_shared_buffers = UT_NUM_SHARED_BUFFERS;
405 : : /* expect success */
406 : 4 : transport = nvmf_tcp_create(&opts);
407 : 4 : CU_ASSERT_PTR_NOT_NULL(transport);
408 : 4 : ttransport = SPDK_CONTAINEROF(transport, struct spdk_nvmf_tcp_transport, transport);
409 [ - + ]: 4 : SPDK_CU_ASSERT_FATAL(ttransport != NULL);
410 : 4 : transport->opts = opts;
411 : 4 : CU_ASSERT(transport->opts.max_queue_depth == UT_MAX_QUEUE_DEPTH);
412 : 4 : CU_ASSERT(transport->opts.max_io_size == UT_MAX_IO_SIZE);
413 : 4 : CU_ASSERT(transport->opts.in_capsule_data_size == UT_IN_CAPSULE_DATA_SIZE);
414 : 4 : CU_ASSERT(transport->opts.io_unit_size == UT_MAX_IO_SIZE);
415 : : /* destroy transport */
416 : 4 : CU_ASSERT(nvmf_tcp_destroy(transport, NULL, NULL) == 0);
417 : :
418 : : /* case 3 */
419 : 4 : memset(&opts, 0, sizeof(opts));
420 : 4 : opts.max_queue_depth = UT_MAX_QUEUE_DEPTH;
421 : 4 : opts.max_qpairs_per_ctrlr = UT_MAX_QPAIRS_PER_CTRLR;
422 : 4 : opts.in_capsule_data_size = UT_IN_CAPSULE_DATA_SIZE;
423 : 4 : opts.max_io_size = UT_MAX_IO_SIZE;
424 : 4 : opts.io_unit_size = 16;
425 : 4 : opts.max_aq_depth = UT_MAX_AQ_DEPTH;
426 : : /* expect fails */
427 : 4 : transport = nvmf_tcp_create(&opts);
428 : 4 : CU_ASSERT_PTR_NULL(transport);
429 : :
430 : 4 : spdk_thread_exit(thread);
431 [ + + ]: 8 : while (!spdk_thread_is_exited(thread)) {
432 : 4 : spdk_thread_poll(thread, 0, 0);
433 : : }
434 : 4 : spdk_thread_destroy(thread);
435 : 4 : }
436 : :
437 : : static void
438 : 4 : test_nvmf_tcp_destroy(void)
439 : : {
440 : : struct spdk_thread *thread;
441 : : struct spdk_nvmf_transport *transport;
442 : 4 : struct spdk_nvmf_transport_opts opts;
443 : :
444 : 4 : thread = spdk_thread_create(NULL, NULL);
445 [ - + ]: 4 : SPDK_CU_ASSERT_FATAL(thread != NULL);
446 : 4 : spdk_set_thread(thread);
447 : :
448 : : /* case 1 */
449 : 4 : memset(&opts, 0, sizeof(opts));
450 : 4 : opts.max_queue_depth = UT_MAX_QUEUE_DEPTH;
451 : 4 : opts.max_qpairs_per_ctrlr = UT_MAX_QPAIRS_PER_CTRLR;
452 : 4 : opts.in_capsule_data_size = UT_IN_CAPSULE_DATA_SIZE;
453 : 4 : opts.max_io_size = UT_MAX_IO_SIZE;
454 : 4 : opts.io_unit_size = UT_IO_UNIT_SIZE;
455 : 4 : opts.max_aq_depth = UT_MAX_AQ_DEPTH;
456 : 4 : opts.num_shared_buffers = UT_NUM_SHARED_BUFFERS;
457 : 4 : transport = nvmf_tcp_create(&opts);
458 : 4 : CU_ASSERT_PTR_NOT_NULL(transport);
459 : 4 : transport->opts = opts;
460 : : /* destroy transport */
461 : 4 : CU_ASSERT(nvmf_tcp_destroy(transport, NULL, NULL) == 0);
462 : :
463 : 4 : spdk_thread_exit(thread);
464 [ + + ]: 8 : while (!spdk_thread_is_exited(thread)) {
465 : 4 : spdk_thread_poll(thread, 0, 0);
466 : : }
467 : 4 : spdk_thread_destroy(thread);
468 : 4 : }
469 : :
470 : : static void
471 : 4 : init_accel(void)
472 : : {
473 : 4 : spdk_io_device_register(g_accel_p, accel_channel_create, accel_channel_destroy,
474 : : sizeof(int), "accel_p");
475 : 4 : }
476 : :
477 : : static void
478 : 4 : fini_accel(void)
479 : : {
480 : 4 : spdk_io_device_unregister(g_accel_p, NULL);
481 : 4 : }
482 : :
483 : : static void
484 : 4 : test_nvmf_tcp_poll_group_create(void)
485 : : {
486 : : struct spdk_nvmf_transport *transport;
487 : : struct spdk_nvmf_transport_poll_group *group;
488 : : struct spdk_nvmf_tcp_poll_group *tgroup;
489 : : struct spdk_thread *thread;
490 : 4 : struct spdk_nvmf_transport_opts opts;
491 : 4 : struct spdk_sock_group grp = {};
492 : :
493 : 4 : thread = spdk_thread_create(NULL, NULL);
494 [ - + ]: 4 : SPDK_CU_ASSERT_FATAL(thread != NULL);
495 : 4 : spdk_set_thread(thread);
496 : :
497 : 4 : init_accel();
498 : :
499 : 4 : memset(&opts, 0, sizeof(opts));
500 : 4 : opts.max_queue_depth = UT_MAX_QUEUE_DEPTH;
501 : 4 : opts.max_qpairs_per_ctrlr = UT_MAX_QPAIRS_PER_CTRLR;
502 : 4 : opts.in_capsule_data_size = UT_IN_CAPSULE_DATA_SIZE;
503 : 4 : opts.max_io_size = UT_MAX_IO_SIZE;
504 : 4 : opts.io_unit_size = UT_IO_UNIT_SIZE;
505 : 4 : opts.max_aq_depth = UT_MAX_AQ_DEPTH;
506 : 4 : opts.num_shared_buffers = UT_NUM_SHARED_BUFFERS;
507 : 4 : transport = nvmf_tcp_create(&opts);
508 : 4 : CU_ASSERT_PTR_NOT_NULL(transport);
509 : 4 : transport->opts = opts;
510 : 4 : MOCK_SET(spdk_sock_group_create, &grp);
511 : 4 : group = nvmf_tcp_poll_group_create(transport, NULL);
512 : 4 : MOCK_CLEAR_P(spdk_sock_group_create);
513 [ - + ]: 4 : SPDK_CU_ASSERT_FATAL(group);
514 [ + - ]: 4 : if (opts.in_capsule_data_size < SPDK_NVME_TCP_IN_CAPSULE_DATA_MAX_SIZE) {
515 : 4 : tgroup = SPDK_CONTAINEROF(group, struct spdk_nvmf_tcp_poll_group, group);
516 [ - + ]: 4 : SPDK_CU_ASSERT_FATAL(tgroup->control_msg_list);
517 : : }
518 : 4 : group->transport = transport;
519 : 4 : nvmf_tcp_poll_group_destroy(group);
520 : 4 : nvmf_tcp_destroy(transport, NULL, NULL);
521 : :
522 : 4 : fini_accel();
523 : 4 : spdk_thread_exit(thread);
524 [ + + ]: 8 : while (!spdk_thread_is_exited(thread)) {
525 : 4 : spdk_thread_poll(thread, 0, 0);
526 : : }
527 : 4 : spdk_thread_destroy(thread);
528 : 4 : }
529 : :
530 : : static void
531 : 4 : test_nvmf_tcp_send_c2h_data(void)
532 : : {
533 : : struct spdk_thread *thread;
534 : 4 : struct spdk_nvmf_tcp_transport ttransport = {};
535 : 4 : struct spdk_nvmf_tcp_qpair tqpair = {};
536 : 4 : struct spdk_nvmf_tcp_req tcp_req = {};
537 : 4 : struct nvme_tcp_pdu pdu = {};
538 : : struct spdk_nvme_tcp_c2h_data_hdr *c2h_data;
539 : :
540 : 4 : ttransport.tcp_opts.c2h_success = true;
541 : 4 : thread = spdk_thread_create(NULL, NULL);
542 [ - + ]: 4 : SPDK_CU_ASSERT_FATAL(thread != NULL);
543 : 4 : spdk_set_thread(thread);
544 : :
545 : 4 : tcp_req.pdu = &pdu;
546 : 4 : tcp_req.req.length = 300;
547 : 4 : tcp_req.req.qpair = &tqpair.qpair;
548 : :
549 : 4 : tqpair.qpair.transport = &ttransport.transport;
550 : :
551 : : /* Set qpair state to make unrelated operations NOP */
552 : 4 : tqpair.state = NVME_TCP_QPAIR_STATE_RUNNING;
553 : 4 : tqpair.recv_state = NVME_TCP_PDU_RECV_STATE_ERROR;
554 : :
555 : 4 : tcp_req.req.cmd = (union nvmf_h2c_msg *)&tcp_req.cmd;
556 : :
557 : 4 : tcp_req.req.iov[0].iov_base = (void *)0xDEADBEEF;
558 : 4 : tcp_req.req.iov[0].iov_len = 101;
559 : 4 : tcp_req.req.iov[1].iov_base = (void *)0xFEEDBEEF;
560 : 4 : tcp_req.req.iov[1].iov_len = 100;
561 : 4 : tcp_req.req.iov[2].iov_base = (void *)0xC0FFEE;
562 : 4 : tcp_req.req.iov[2].iov_len = 99;
563 : 4 : tcp_req.req.iovcnt = 3;
564 : 4 : tcp_req.req.length = 300;
565 : :
566 : 4 : nvmf_tcp_send_c2h_data(&tqpair, &tcp_req);
567 : :
568 : 4 : c2h_data = &pdu.hdr.c2h_data;
569 : 4 : CU_ASSERT(c2h_data->datao == 0);
570 : 4 : CU_ASSERT(c2h_data->datal = 300);
571 : 4 : CU_ASSERT(c2h_data->common.plen == sizeof(*c2h_data) + 300);
572 : 4 : CU_ASSERT(c2h_data->common.flags & SPDK_NVME_TCP_C2H_DATA_FLAGS_LAST_PDU);
573 : 4 : CU_ASSERT(c2h_data->common.flags & SPDK_NVME_TCP_C2H_DATA_FLAGS_SUCCESS);
574 : :
575 : 4 : CU_ASSERT(pdu.data_iovcnt == 3);
576 : 4 : CU_ASSERT((uint64_t)pdu.data_iov[0].iov_base == 0xDEADBEEF);
577 : 4 : CU_ASSERT(pdu.data_iov[0].iov_len == 101);
578 : 4 : CU_ASSERT((uint64_t)pdu.data_iov[1].iov_base == 0xFEEDBEEF);
579 : 4 : CU_ASSERT(pdu.data_iov[1].iov_len == 100);
580 : 4 : CU_ASSERT((uint64_t)pdu.data_iov[2].iov_base == 0xC0FFEE);
581 : 4 : CU_ASSERT(pdu.data_iov[2].iov_len == 99);
582 : :
583 : 4 : tcp_req.pdu_in_use = false;
584 : 4 : tcp_req.rsp.cdw0 = 1;
585 : 4 : nvmf_tcp_send_c2h_data(&tqpair, &tcp_req);
586 : :
587 : 4 : CU_ASSERT(c2h_data->common.flags & SPDK_NVME_TCP_C2H_DATA_FLAGS_LAST_PDU);
588 : 4 : CU_ASSERT((c2h_data->common.flags & SPDK_NVME_TCP_C2H_DATA_FLAGS_SUCCESS) == 0);
589 : :
590 : 4 : ttransport.tcp_opts.c2h_success = false;
591 : 4 : tcp_req.pdu_in_use = false;
592 : 4 : tcp_req.rsp.cdw0 = 0;
593 : 4 : nvmf_tcp_send_c2h_data(&tqpair, &tcp_req);
594 : :
595 : 4 : CU_ASSERT(c2h_data->common.flags & SPDK_NVME_TCP_C2H_DATA_FLAGS_LAST_PDU);
596 : 4 : CU_ASSERT((c2h_data->common.flags & SPDK_NVME_TCP_C2H_DATA_FLAGS_SUCCESS) == 0);
597 : :
598 : 4 : tcp_req.pdu_in_use = false;
599 : 4 : tcp_req.rsp.cdw0 = 1;
600 : 4 : nvmf_tcp_send_c2h_data(&tqpair, &tcp_req);
601 : :
602 : 4 : CU_ASSERT(c2h_data->common.flags & SPDK_NVME_TCP_C2H_DATA_FLAGS_LAST_PDU);
603 : 4 : CU_ASSERT((c2h_data->common.flags & SPDK_NVME_TCP_C2H_DATA_FLAGS_SUCCESS) == 0);
604 : :
605 : 4 : spdk_thread_exit(thread);
606 [ + + ]: 8 : while (!spdk_thread_is_exited(thread)) {
607 : 4 : spdk_thread_poll(thread, 0, 0);
608 : : }
609 : 4 : spdk_thread_destroy(thread);
610 : 4 : }
611 : :
612 : : #define NVMF_TCP_PDU_MAX_H2C_DATA_SIZE (128 * 1024)
613 : :
614 : : static void
615 : 4 : test_nvmf_tcp_h2c_data_hdr_handle(void)
616 : : {
617 : 4 : struct spdk_nvmf_tcp_transport ttransport = {};
618 : 4 : struct spdk_nvmf_tcp_qpair tqpair = {};
619 : 4 : struct nvme_tcp_pdu pdu = {};
620 : 4 : struct spdk_nvmf_tcp_req tcp_req = {};
621 : : struct spdk_nvme_tcp_h2c_data_hdr *h2c_data;
622 : :
623 : : /* Set qpair state to make unrelated operations NOP */
624 : 4 : tqpair.state = NVME_TCP_QPAIR_STATE_RUNNING;
625 : 4 : tqpair.recv_state = NVME_TCP_PDU_RECV_STATE_ERROR;
626 : 4 : tqpair.resource_count = 1;
627 : 4 : tqpair.reqs = &tcp_req;
628 : :
629 : 4 : tcp_req.req.iov[0].iov_base = (void *)0xDEADBEEF;
630 : 4 : tcp_req.req.iov[0].iov_len = 101;
631 : 4 : tcp_req.req.iov[1].iov_base = (void *)0xFEEDBEEF;
632 : 4 : tcp_req.req.iov[1].iov_len = 99;
633 : 4 : tcp_req.req.iovcnt = 2;
634 : 4 : tcp_req.req.length = 200;
635 : 4 : tcp_req.state = TCP_REQUEST_STATE_TRANSFERRING_HOST_TO_CONTROLLER;
636 : :
637 : 4 : tcp_req.req.cmd = (union nvmf_h2c_msg *)&tcp_req.cmd;
638 : 4 : tcp_req.req.cmd->nvme_cmd.cid = 1;
639 : 4 : tcp_req.ttag = 1;
640 : :
641 : 4 : h2c_data = &pdu.hdr.h2c_data;
642 : 4 : h2c_data->cccid = 1;
643 : 4 : h2c_data->ttag = 1;
644 : 4 : h2c_data->datao = 0;
645 : 4 : h2c_data->datal = 200;
646 : :
647 : 4 : nvmf_tcp_h2c_data_hdr_handle(&ttransport, &tqpair, &pdu);
648 : :
649 : 4 : CU_ASSERT(pdu.data_iovcnt == 2);
650 : 4 : CU_ASSERT((uint64_t)pdu.data_iov[0].iov_base == 0xDEADBEEF);
651 : 4 : CU_ASSERT(pdu.data_iov[0].iov_len == 101);
652 : 4 : CU_ASSERT((uint64_t)pdu.data_iov[1].iov_base == 0xFEEDBEEF);
653 : 4 : CU_ASSERT(pdu.data_iov[1].iov_len == 99);
654 : 4 : }
655 : :
656 : :
657 : : static void
658 : 4 : test_nvmf_tcp_in_capsule_data_handle(void)
659 : : {
660 : 4 : struct spdk_nvmf_tcp_transport ttransport = {};
661 : 4 : struct spdk_nvmf_tcp_qpair tqpair = {};
662 : 4 : struct nvme_tcp_pdu *pdu, pdu_in_progress = {};
663 : 4 : union nvmf_c2h_msg rsp0 = {};
664 : 4 : union nvmf_c2h_msg rsp = {};
665 : :
666 : 4 : struct spdk_nvmf_request *req_temp = NULL;
667 : 4 : struct spdk_nvmf_tcp_req tcp_req2 = {};
668 : 4 : struct spdk_nvmf_tcp_req tcp_req1 = {};
669 : :
670 : : struct spdk_nvme_tcp_cmd *capsule_data;
671 : : struct spdk_nvmf_capsule_cmd *nvmf_capsule_data;
672 : : struct spdk_nvme_sgl_descriptor *sgl;
673 : :
674 : : struct spdk_nvmf_transport_poll_group *group;
675 : 4 : struct spdk_nvmf_tcp_poll_group tcp_group = {};
676 : 4 : struct spdk_sock_group grp = {};
677 : :
678 : 4 : tqpair.pdu_in_progress = &pdu_in_progress;
679 : 4 : ttransport.transport.opts.max_io_size = UT_MAX_IO_SIZE;
680 : 4 : ttransport.transport.opts.io_unit_size = UT_IO_UNIT_SIZE;
681 : :
682 : 4 : tcp_group.sock_group = &grp;
683 : 4 : TAILQ_INIT(&tcp_group.qpairs);
684 : 4 : group = &tcp_group.group;
685 : 4 : group->transport = &ttransport.transport;
686 : 4 : STAILQ_INIT(&group->pending_buf_queue);
687 : 4 : tqpair.group = &tcp_group;
688 : :
689 : 4 : TAILQ_INIT(&tqpair.tcp_req_free_queue);
690 : 4 : TAILQ_INIT(&tqpair.tcp_req_working_queue);
691 : :
692 : 4 : TAILQ_INSERT_TAIL(&tqpair.tcp_req_free_queue, &tcp_req2, state_link);
693 : 4 : tqpair.state_cntr[TCP_REQUEST_STATE_FREE]++;
694 : 4 : tqpair.qpair.transport = &ttransport.transport;
695 : 4 : tqpair.state = NVME_TCP_QPAIR_STATE_RUNNING;
696 : 4 : tqpair.recv_state = NVME_TCP_PDU_RECV_STATE_AWAIT_PDU_PSH;
697 : 4 : tqpair.qpair.state = SPDK_NVMF_QPAIR_ACTIVE;
698 : :
699 : : /* init a null tcp_req into tqpair TCP_REQUEST_STATE_FREE queue */
700 : 4 : tcp_req2.req.qpair = &tqpair.qpair;
701 : 4 : tcp_req2.req.cmd = (union nvmf_h2c_msg *)&tcp_req2.cmd;
702 : 4 : tcp_req2.req.rsp = &rsp;
703 : :
704 : : /* init tcp_req1 */
705 : 4 : tcp_req1.req.qpair = &tqpair.qpair;
706 : 4 : tcp_req1.req.cmd = (union nvmf_h2c_msg *)&tcp_req1.cmd;
707 : 4 : tcp_req1.req.rsp = &rsp0;
708 : 4 : tcp_req1.state = TCP_REQUEST_STATE_NEW;
709 : :
710 : 4 : TAILQ_INSERT_TAIL(&tqpair.tcp_req_working_queue, &tcp_req1, state_link);
711 : 4 : tqpair.state_cntr[TCP_REQUEST_STATE_NEW]++;
712 : :
713 : : /* init pdu, make pdu need sgl buff */
714 : 4 : pdu = tqpair.pdu_in_progress;
715 : 4 : capsule_data = &pdu->hdr.capsule_cmd;
716 : 4 : nvmf_capsule_data = (struct spdk_nvmf_capsule_cmd *)&pdu->hdr.capsule_cmd.ccsqe;
717 : 4 : sgl = &capsule_data->ccsqe.dptr.sgl1;
718 : :
719 : 4 : capsule_data->common.pdu_type = SPDK_NVME_TCP_PDU_TYPE_CAPSULE_CMD;
720 : 4 : capsule_data->common.hlen = sizeof(*capsule_data);
721 : 4 : capsule_data->common.plen = 1096;
722 : 4 : capsule_data->ccsqe.opc = SPDK_NVME_OPC_FABRIC;
723 : :
724 : 4 : sgl->unkeyed.subtype = SPDK_NVME_SGL_SUBTYPE_TRANSPORT;
725 : 4 : sgl->generic.type = SPDK_NVME_SGL_TYPE_TRANSPORT_DATA_BLOCK;
726 : 4 : sgl->unkeyed.length = UT_IO_UNIT_SIZE;
727 : :
728 : 4 : nvmf_capsule_data->fctype = SPDK_NVMF_FABRIC_COMMAND_CONNECT;
729 : :
730 : : /* insert tcp_req1 to pending_buf_queue, And this req takes precedence over the next req. */
731 : 4 : nvmf_tcp_req_process(&ttransport, &tcp_req1);
732 : 4 : CU_ASSERT(STAILQ_FIRST(&group->pending_buf_queue) == &tcp_req1.req);
733 : :
734 : 4 : sgl->unkeyed.length = UT_IO_UNIT_SIZE - 1;
735 : :
736 : : /* process tqpair capsule req. but we still remain req in pending_buff. */
737 : 4 : nvmf_tcp_capsule_cmd_hdr_handle(&ttransport, &tqpair, tqpair.pdu_in_progress);
738 : 4 : CU_ASSERT(tqpair.recv_state == NVME_TCP_PDU_RECV_STATE_AWAIT_PDU_PAYLOAD);
739 : 4 : CU_ASSERT(STAILQ_FIRST(&group->pending_buf_queue) == &tcp_req1.req);
740 [ + + ]: 8 : STAILQ_FOREACH(req_temp, &group->pending_buf_queue, buf_link) {
741 [ - + ]: 4 : if (req_temp == &tcp_req2.req) {
742 : 0 : break;
743 : : }
744 : : }
745 : 4 : CU_ASSERT(req_temp == NULL);
746 : 4 : CU_ASSERT(tqpair.pdu_in_progress->req == (void *)&tcp_req2);
747 : 4 : }
748 : :
749 : : static void
750 : 4 : test_nvmf_tcp_qpair_init_mem_resource(void)
751 : : {
752 : : int rc;
753 : 4 : struct spdk_nvmf_tcp_qpair *tqpair = NULL;
754 : 4 : struct spdk_nvmf_transport transport = {};
755 : : struct spdk_thread *thread;
756 : :
757 : 4 : thread = spdk_thread_create(NULL, NULL);
758 [ - + ]: 4 : SPDK_CU_ASSERT_FATAL(thread != NULL);
759 : 4 : spdk_set_thread(thread);
760 : :
761 : 4 : tqpair = calloc(1, sizeof(*tqpair));
762 : 4 : tqpair->qpair.transport = &transport;
763 : :
764 : 4 : nvmf_tcp_opts_init(&transport.opts);
765 : 4 : CU_ASSERT(transport.opts.max_queue_depth == SPDK_NVMF_TCP_DEFAULT_MAX_IO_QUEUE_DEPTH);
766 : 4 : CU_ASSERT(transport.opts.max_qpairs_per_ctrlr == SPDK_NVMF_TCP_DEFAULT_MAX_QPAIRS_PER_CTRLR);
767 : 4 : CU_ASSERT(transport.opts.in_capsule_data_size == SPDK_NVMF_TCP_DEFAULT_IN_CAPSULE_DATA_SIZE);
768 : 4 : CU_ASSERT(transport.opts.max_io_size == SPDK_NVMF_TCP_DEFAULT_MAX_IO_SIZE);
769 : 4 : CU_ASSERT(transport.opts.io_unit_size == SPDK_NVMF_TCP_DEFAULT_IO_UNIT_SIZE);
770 : 4 : CU_ASSERT(transport.opts.max_aq_depth == SPDK_NVMF_TCP_DEFAULT_MAX_ADMIN_QUEUE_DEPTH);
771 : 4 : CU_ASSERT(transport.opts.num_shared_buffers == SPDK_NVMF_TCP_DEFAULT_NUM_SHARED_BUFFERS);
772 : 4 : CU_ASSERT(transport.opts.buf_cache_size == SPDK_NVMF_TCP_DEFAULT_BUFFER_CACHE_SIZE);
773 [ - + ]: 4 : CU_ASSERT(transport.opts.dif_insert_or_strip == SPDK_NVMF_TCP_DEFAULT_DIF_INSERT_OR_STRIP);
774 : 4 : CU_ASSERT(transport.opts.abort_timeout_sec == SPDK_NVMF_TCP_DEFAULT_ABORT_TIMEOUT_SEC);
775 : 4 : CU_ASSERT(transport.opts.transport_specific == NULL);
776 : :
777 : 4 : rc = nvmf_tcp_qpair_init(&tqpair->qpair);
778 : 4 : CU_ASSERT(rc == 0);
779 [ - + ]: 4 : CU_ASSERT(tqpair->host_hdgst_enable == true);
780 [ - + ]: 4 : CU_ASSERT(tqpair->host_ddgst_enable == true);
781 : :
782 : 4 : rc = nvmf_tcp_qpair_init_mem_resource(tqpair);
783 : 4 : CU_ASSERT(rc == 0);
784 : 4 : CU_ASSERT(tqpair->resource_count == SPDK_NVMF_TCP_DEFAULT_MAX_IO_QUEUE_DEPTH);
785 : 4 : CU_ASSERT(tqpair->reqs != NULL);
786 : 4 : CU_ASSERT(tqpair->bufs != NULL);
787 : 4 : CU_ASSERT(tqpair->pdus != NULL);
788 : : /* Just to check the first and last entry */
789 : 4 : CU_ASSERT(tqpair->reqs[0].ttag == 1);
790 : 4 : CU_ASSERT(tqpair->reqs[0].req.qpair == &tqpair->qpair);
791 : 4 : CU_ASSERT(tqpair->reqs[0].pdu == &tqpair->pdus[0]);
792 : 4 : CU_ASSERT(tqpair->reqs[0].pdu->qpair == &tqpair->qpair);
793 : 4 : CU_ASSERT(tqpair->reqs[0].buf == (void *)((uintptr_t)tqpair->bufs));
794 : 4 : CU_ASSERT(tqpair->reqs[0].req.rsp == (void *)&tqpair->reqs[0].rsp);
795 : 4 : CU_ASSERT(tqpair->reqs[0].req.cmd == (void *)&tqpair->reqs[0].cmd);
796 : 4 : CU_ASSERT(tqpair->reqs[0].state == TCP_REQUEST_STATE_FREE);
797 : 4 : CU_ASSERT(tqpair->reqs[127].ttag == 128);
798 : 4 : CU_ASSERT(tqpair->reqs[127].req.qpair == &tqpair->qpair);
799 : 4 : CU_ASSERT(tqpair->reqs[127].pdu == &tqpair->pdus[127]);
800 : 4 : CU_ASSERT(tqpair->reqs[127].pdu->qpair == &tqpair->qpair);
801 : 4 : CU_ASSERT(tqpair->reqs[127].buf == (void *)((uintptr_t)tqpair->bufs) + 127 * 4096);
802 : 4 : CU_ASSERT(tqpair->reqs[127].req.rsp == (void *)&tqpair->reqs[127].rsp);
803 : 4 : CU_ASSERT(tqpair->reqs[127].req.cmd == (void *)&tqpair->reqs[127].cmd);
804 : 4 : CU_ASSERT(tqpair->reqs[127].state == TCP_REQUEST_STATE_FREE);
805 : 4 : CU_ASSERT(tqpair->state_cntr[TCP_REQUEST_STATE_FREE] == SPDK_NVMF_TCP_DEFAULT_MAX_IO_QUEUE_DEPTH);
806 : 4 : CU_ASSERT(tqpair->mgmt_pdu == &tqpair->pdus[2 * SPDK_NVMF_TCP_DEFAULT_MAX_IO_QUEUE_DEPTH]);
807 : 4 : CU_ASSERT(tqpair->mgmt_pdu->qpair == tqpair);
808 : 4 : CU_ASSERT(tqpair->pdu_in_progress ==
809 : : &tqpair->pdus[2 * SPDK_NVMF_TCP_DEFAULT_MAX_IO_QUEUE_DEPTH - 1]);
810 : 4 : CU_ASSERT(tqpair->recv_buf_size == (4096 + sizeof(struct spdk_nvme_tcp_cmd) + 2 *
811 : : SPDK_NVME_TCP_DIGEST_LEN) * SPDK_NVMF_TCP_RECV_BUF_SIZE_FACTOR);
812 : :
813 : : /* Free all of tqpair resource */
814 : 4 : nvmf_tcp_qpair_destroy(tqpair);
815 : :
816 : 4 : spdk_thread_exit(thread);
817 [ + + ]: 8 : while (!spdk_thread_is_exited(thread)) {
818 : 4 : spdk_thread_poll(thread, 0, 0);
819 : : }
820 : 4 : spdk_thread_destroy(thread);
821 : 4 : }
822 : :
823 : : static void
824 : 4 : test_nvmf_tcp_send_c2h_term_req(void)
825 : : {
826 : 4 : struct spdk_nvmf_tcp_qpair tqpair = {};
827 : 4 : struct nvme_tcp_pdu pdu = {}, mgmt_pdu = {}, pdu_in_progress = {};
828 : 4 : enum spdk_nvme_tcp_term_req_fes fes = SPDK_NVME_TCP_TERM_REQ_FES_INVALID_HEADER_FIELD;
829 : 4 : uint32_t error_offset = 1;
830 : :
831 : 4 : mgmt_pdu.qpair = &tqpair;
832 : 4 : tqpair.mgmt_pdu = &mgmt_pdu;
833 : 4 : tqpair.pdu_in_progress = &pdu_in_progress;
834 : 4 : tqpair.tcp_pdu_working_count = 1;
835 : :
836 : : /* case1: hlen < SPDK_NVME_TCP_TERM_REQ_ERROR_DATA_MAX_SIZE, Expect: copy_len == hlen */
837 : 4 : pdu.hdr.common.hlen = 64;
838 : 4 : nvmf_tcp_send_c2h_term_req(&tqpair, &pdu, fes, error_offset);
839 : 4 : CU_ASSERT(tqpair.recv_state == NVME_TCP_PDU_RECV_STATE_QUIESCING);
840 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.hlen == sizeof(struct spdk_nvme_tcp_term_req_hdr));
841 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.plen == tqpair.mgmt_pdu->hdr.term_req.common.hlen +
842 : : pdu.hdr.common.hlen);
843 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.pdu_type == SPDK_NVME_TCP_PDU_TYPE_C2H_TERM_REQ);
844 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.fes == SPDK_NVME_TCP_TERM_REQ_FES_INVALID_HEADER_FIELD);
845 : :
846 : : /* case2: hlen > SPDK_NVME_TCP_TERM_REQ_ERROR_DATA_MAX_SIZE, Expect: copy_len == SPDK_NVME_TCP_TERM_REQ_ERROR_DATA_MAX_SIZE */
847 : 4 : pdu.hdr.common.hlen = 255;
848 : 4 : nvmf_tcp_send_c2h_term_req(&tqpair, &pdu, fes, error_offset);
849 : 4 : CU_ASSERT(tqpair.recv_state == NVME_TCP_PDU_RECV_STATE_QUIESCING);
850 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.hlen == sizeof(struct spdk_nvme_tcp_term_req_hdr));
851 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.plen == (unsigned)
852 : : tqpair.mgmt_pdu->hdr.term_req.common.hlen + SPDK_NVME_TCP_TERM_REQ_ERROR_DATA_MAX_SIZE);
853 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.pdu_type == SPDK_NVME_TCP_PDU_TYPE_C2H_TERM_REQ);
854 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.fes == SPDK_NVME_TCP_TERM_REQ_FES_INVALID_HEADER_FIELD);
855 : 4 : }
856 : :
857 : : static void
858 : 4 : test_nvmf_tcp_send_capsule_resp_pdu(void)
859 : : {
860 : 4 : struct spdk_nvmf_tcp_req tcp_req = {};
861 : 4 : struct spdk_nvmf_tcp_qpair tqpair = {};
862 : 4 : struct nvme_tcp_pdu pdu = {};
863 : :
864 : 4 : tcp_req.pdu_in_use = false;
865 : 4 : tcp_req.req.qpair = &tqpair.qpair;
866 : 4 : tcp_req.pdu = &pdu;
867 : 4 : tcp_req.req.rsp = (union nvmf_c2h_msg *)&tcp_req.rsp;
868 : 4 : tcp_req.req.cmd = (union nvmf_h2c_msg *)&tcp_req.cmd;
869 : 4 : tqpair.host_hdgst_enable = true;
870 : :
871 : 4 : nvmf_tcp_send_capsule_resp_pdu(&tcp_req, &tqpair);
872 : 4 : CU_ASSERT(pdu.hdr.capsule_resp.common.pdu_type == SPDK_NVME_TCP_PDU_TYPE_CAPSULE_RESP);
873 : 4 : CU_ASSERT(pdu.hdr.capsule_resp.common.plen == sizeof(struct spdk_nvme_tcp_rsp) +
874 : : SPDK_NVME_TCP_DIGEST_LEN);
875 : 4 : CU_ASSERT(pdu.hdr.capsule_resp.common.hlen == sizeof(struct spdk_nvme_tcp_rsp));
876 [ - + ]: 4 : CU_ASSERT(!memcmp(&pdu.hdr.capsule_resp.rccqe, &tcp_req.req.rsp->nvme_cpl,
877 : : sizeof(struct spdk_nvme_cpl)));
878 : 4 : CU_ASSERT(pdu.hdr.capsule_resp.common.flags & SPDK_NVME_TCP_CH_FLAGS_HDGSTF);
879 : 4 : CU_ASSERT(pdu.cb_fn == nvmf_tcp_request_free);
880 : 4 : CU_ASSERT(pdu.cb_arg == &tcp_req);
881 : 4 : CU_ASSERT(pdu.iov[0].iov_base == &pdu.hdr.raw);
882 : 4 : CU_ASSERT(pdu.iov[0].iov_len == sizeof(struct spdk_nvme_tcp_rsp) + SPDK_NVME_TCP_DIGEST_LEN);
883 : :
884 : : /* hdgst disable */
885 : 4 : tqpair.host_hdgst_enable = false;
886 : 4 : tcp_req.pdu_in_use = false;
887 : 4 : memset(&pdu, 0, sizeof(pdu));
888 : :
889 : 4 : nvmf_tcp_send_capsule_resp_pdu(&tcp_req, &tqpair);
890 : 4 : CU_ASSERT(pdu.hdr.capsule_resp.common.pdu_type == SPDK_NVME_TCP_PDU_TYPE_CAPSULE_RESP);
891 : 4 : CU_ASSERT(pdu.hdr.capsule_resp.common.plen == sizeof(struct spdk_nvme_tcp_rsp));
892 : 4 : CU_ASSERT(pdu.hdr.capsule_resp.common.hlen == sizeof(struct spdk_nvme_tcp_rsp));
893 [ - + ]: 4 : CU_ASSERT(!memcmp(&pdu.hdr.capsule_resp.rccqe, &tcp_req.req.rsp->nvme_cpl,
894 : : sizeof(struct spdk_nvme_cpl)));
895 : 4 : CU_ASSERT(!(pdu.hdr.capsule_resp.common.flags & SPDK_NVME_TCP_CH_FLAGS_HDGSTF));
896 : 4 : CU_ASSERT(pdu.cb_fn == nvmf_tcp_request_free);
897 : 4 : CU_ASSERT(pdu.cb_arg == &tcp_req);
898 : 4 : CU_ASSERT(pdu.iov[0].iov_base == &pdu.hdr.raw);
899 : 4 : CU_ASSERT(pdu.iov[0].iov_len == sizeof(struct spdk_nvme_tcp_rsp));
900 : 4 : }
901 : :
902 : : static void
903 : 4 : test_nvmf_tcp_icreq_handle(void)
904 : : {
905 : 4 : struct spdk_nvmf_tcp_transport ttransport = {};
906 : 4 : struct spdk_nvmf_tcp_qpair tqpair = {};
907 : 4 : struct nvme_tcp_pdu pdu = {};
908 : 4 : struct nvme_tcp_pdu mgmt_pdu = {};
909 : 4 : struct nvme_tcp_pdu pdu_in_progress = {};
910 : : struct spdk_nvme_tcp_ic_resp *ic_resp;
911 : :
912 : 4 : mgmt_pdu.qpair = &tqpair;
913 : 4 : tqpair.mgmt_pdu = &mgmt_pdu;
914 : 4 : tqpair.pdu_in_progress = &pdu_in_progress;
915 : 4 : tqpair.tcp_pdu_working_count = 1;
916 : :
917 : : /* case 1: Expected ICReq PFV 0 and got are different. */
918 : 4 : pdu.hdr.ic_req.pfv = 1;
919 : :
920 : 4 : nvmf_tcp_icreq_handle(&ttransport, &tqpair, &pdu);
921 : :
922 : 4 : CU_ASSERT(tqpair.recv_state == NVME_TCP_PDU_RECV_STATE_QUIESCING);
923 : :
924 : : /* case 2: Expected ICReq HPDA in range 0-31 and got are different. */
925 : 4 : pdu.hdr.ic_req.hpda = SPDK_NVME_TCP_HPDA_MAX + 1;
926 : :
927 : 4 : nvmf_tcp_icreq_handle(&ttransport, &tqpair, &pdu);
928 : :
929 : 4 : CU_ASSERT(tqpair.recv_state == NVME_TCP_PDU_RECV_STATE_QUIESCING);
930 : :
931 : : /* case 3: Expect: PASS. */
932 : 4 : ttransport.transport.opts.max_io_size = 32;
933 : 4 : pdu.hdr.ic_req.pfv = 0;
934 : 4 : tqpair.host_hdgst_enable = false;
935 : 4 : tqpair.host_ddgst_enable = false;
936 : 4 : tqpair.recv_buf_size = 64;
937 : 4 : pdu.hdr.ic_req.hpda = 16;
938 : :
939 : 4 : nvmf_tcp_icreq_handle(&ttransport, &tqpair, &pdu);
940 : :
941 : 4 : ic_resp = &tqpair.mgmt_pdu->hdr.ic_resp;
942 : 4 : CU_ASSERT(tqpair.recv_buf_size == MIN_SOCK_PIPE_SIZE);
943 : 4 : CU_ASSERT(tqpair.cpda == pdu.hdr.ic_req.hpda);
944 : 4 : CU_ASSERT(ic_resp->common.pdu_type == SPDK_NVME_TCP_PDU_TYPE_IC_RESP);
945 : 4 : CU_ASSERT(ic_resp->common.hlen == sizeof(struct spdk_nvme_tcp_ic_resp));
946 : 4 : CU_ASSERT(ic_resp->common.plen == sizeof(struct spdk_nvme_tcp_ic_resp));
947 : 4 : CU_ASSERT(ic_resp->pfv == 0);
948 : 4 : CU_ASSERT(ic_resp->cpda == tqpair.cpda);
949 : 4 : CU_ASSERT(ic_resp->maxh2cdata == ttransport.transport.opts.max_io_size);
950 : 4 : CU_ASSERT(ic_resp->dgst.bits.hdgst_enable == 0);
951 : 4 : CU_ASSERT(ic_resp->dgst.bits.ddgst_enable == 0);
952 : 4 : CU_ASSERT(tqpair.recv_state == NVME_TCP_PDU_RECV_STATE_AWAIT_PDU_READY);
953 : 4 : }
954 : :
955 : : static void
956 : 4 : test_nvmf_tcp_check_xfer_type(void)
957 : : {
958 : 4 : const uint16_t cid = 0xAA;
959 : 4 : struct spdk_nvmf_tcp_transport ttransport = {};
960 : 4 : struct spdk_nvmf_tcp_qpair tqpair = {};
961 : 4 : struct nvme_tcp_pdu pdu_in_progress = {};
962 : 4 : union nvmf_c2h_msg rsp0 = {};
963 : :
964 : 4 : struct spdk_nvmf_tcp_req tcp_req = {};
965 : 4 : struct nvme_tcp_pdu rsp_pdu = {};
966 : :
967 : : struct spdk_nvme_tcp_cmd *capsule_data;
968 : : struct spdk_nvme_sgl_descriptor *sgl;
969 : :
970 : : struct spdk_nvmf_transport_poll_group *group;
971 : 4 : struct spdk_nvmf_tcp_poll_group tcp_group = {};
972 : 4 : struct spdk_sock_group grp = {};
973 : :
974 : 4 : tqpair.pdu_in_progress = &pdu_in_progress;
975 : 4 : ttransport.transport.opts.max_io_size = UT_MAX_IO_SIZE;
976 : 4 : ttransport.transport.opts.io_unit_size = UT_IO_UNIT_SIZE;
977 : :
978 : 4 : tcp_group.sock_group = &grp;
979 : 4 : TAILQ_INIT(&tcp_group.qpairs);
980 : 4 : group = &tcp_group.group;
981 : 4 : group->transport = &ttransport.transport;
982 : 4 : STAILQ_INIT(&group->pending_buf_queue);
983 : 4 : tqpair.group = &tcp_group;
984 : :
985 : 4 : TAILQ_INIT(&tqpair.tcp_req_free_queue);
986 : 4 : TAILQ_INIT(&tqpair.tcp_req_working_queue);
987 : :
988 : 4 : tqpair.qpair.transport = &ttransport.transport;
989 : 4 : tqpair.state = NVME_TCP_QPAIR_STATE_RUNNING;
990 : 4 : tqpair.recv_state = NVME_TCP_PDU_RECV_STATE_AWAIT_PDU_PSH;
991 : 4 : tqpair.qpair.state = SPDK_NVMF_QPAIR_ACTIVE;
992 : :
993 : : /* init tcp_req */
994 : 4 : tcp_req.req.qpair = &tqpair.qpair;
995 : 4 : tcp_req.pdu = &rsp_pdu;
996 : 4 : tcp_req.req.cmd = (union nvmf_h2c_msg *)&tcp_req.cmd;
997 : 4 : tcp_req.req.rsp = &rsp0;
998 : 4 : tcp_req.state = TCP_REQUEST_STATE_NEW;
999 : :
1000 : 4 : TAILQ_INSERT_TAIL(&tqpair.tcp_req_working_queue, &tcp_req, state_link);
1001 : 4 : tqpair.state_cntr[TCP_REQUEST_STATE_NEW]++;
1002 : :
1003 : : /* init pdu, make pdu need sgl buff */
1004 : 4 : capsule_data = &tqpair.pdu_in_progress->hdr.capsule_cmd;
1005 : 4 : sgl = &capsule_data->ccsqe.dptr.sgl1;
1006 : :
1007 : 4 : capsule_data->common.pdu_type = SPDK_NVME_TCP_PDU_TYPE_CAPSULE_CMD;
1008 : 4 : capsule_data->common.hlen = sizeof(*capsule_data);
1009 : 4 : capsule_data->common.plen = 1096;
1010 : 4 : capsule_data->ccsqe.opc = 0x10 | SPDK_NVME_DATA_BIDIRECTIONAL;
1011 : : /* Need to set to a non zero valid to check it gets copied to the response */
1012 : 4 : capsule_data->ccsqe.cid = cid;
1013 : :
1014 : : /* Set up SGL to ensure nvmf_tcp_req_parse_sgl returns an error */
1015 : 4 : sgl->unkeyed.subtype = SPDK_NVME_SGL_SUBTYPE_TRANSPORT;
1016 : 4 : sgl->generic.type = SPDK_NVME_SGL_TYPE_TRANSPORT_DATA_BLOCK;
1017 : 4 : sgl->unkeyed.length = UT_IO_UNIT_SIZE;
1018 : :
1019 : : /* Process a command and ensure that it fails and the request is set up to return an error */
1020 : 4 : nvmf_tcp_req_process(&ttransport, &tcp_req);
1021 : 4 : CU_ASSERT(STAILQ_EMPTY(&group->pending_buf_queue));
1022 : 4 : CU_ASSERT(tcp_req.state == TCP_REQUEST_STATE_TRANSFERRING_CONTROLLER_TO_HOST);
1023 : 4 : CU_ASSERT(tqpair.recv_state == NVME_TCP_PDU_RECV_STATE_AWAIT_PDU_READY);
1024 : 4 : CU_ASSERT(tcp_req.req.rsp->nvme_cpl.cid == cid);
1025 : 4 : CU_ASSERT(tcp_req.req.rsp->nvme_cpl.status.sct == SPDK_NVME_SCT_GENERIC);
1026 : 4 : CU_ASSERT(tcp_req.req.rsp->nvme_cpl.status.sc == SPDK_NVME_SC_INVALID_OPCODE);
1027 : 4 : }
1028 : :
1029 : : static void
1030 : 4 : test_nvmf_tcp_invalid_sgl(void)
1031 : : {
1032 : 4 : const uint16_t cid = 0xAABB;
1033 : 4 : struct spdk_nvmf_tcp_transport ttransport = {};
1034 : 4 : struct spdk_nvmf_tcp_qpair tqpair = {};
1035 : 4 : struct nvme_tcp_pdu pdu_in_progress = {};
1036 : 4 : union nvmf_c2h_msg rsp0 = {};
1037 : :
1038 : 4 : struct spdk_nvmf_tcp_req tcp_req = {};
1039 : 4 : struct nvme_tcp_pdu rsp_pdu = {};
1040 : 4 : struct nvme_tcp_pdu mgmt_pdu = {};
1041 : :
1042 : : struct spdk_nvme_tcp_cmd *capsule_data;
1043 : : struct spdk_nvme_sgl_descriptor *sgl;
1044 : :
1045 : : struct spdk_nvmf_transport_poll_group *group;
1046 : 4 : struct spdk_nvmf_tcp_poll_group tcp_group = {};
1047 : 4 : struct spdk_sock_group grp = {};
1048 : :
1049 : 4 : tqpair.pdu_in_progress = &pdu_in_progress;
1050 : 4 : ttransport.transport.opts.max_io_size = UT_MAX_IO_SIZE;
1051 : 4 : ttransport.transport.opts.io_unit_size = UT_IO_UNIT_SIZE;
1052 : :
1053 : 4 : tcp_group.sock_group = &grp;
1054 : 4 : TAILQ_INIT(&tcp_group.qpairs);
1055 : 4 : group = &tcp_group.group;
1056 : 4 : group->transport = &ttransport.transport;
1057 : 4 : STAILQ_INIT(&group->pending_buf_queue);
1058 : 4 : tqpair.group = &tcp_group;
1059 : :
1060 : 4 : TAILQ_INIT(&tqpair.tcp_req_free_queue);
1061 : 4 : TAILQ_INIT(&tqpair.tcp_req_working_queue);
1062 : :
1063 : 4 : tqpair.qpair.transport = &ttransport.transport;
1064 : 4 : tqpair.state = NVME_TCP_QPAIR_STATE_RUNNING;
1065 : 4 : tqpair.recv_state = NVME_TCP_PDU_RECV_STATE_AWAIT_PDU_PSH;
1066 : 4 : tqpair.qpair.state = SPDK_NVMF_QPAIR_ACTIVE;
1067 : :
1068 : : /* init tcp_req */
1069 : 4 : tcp_req.req.qpair = &tqpair.qpair;
1070 : 4 : tcp_req.pdu = &rsp_pdu;
1071 : 4 : tcp_req.pdu->qpair = &tqpair;
1072 : 4 : tqpair.mgmt_pdu = &mgmt_pdu;
1073 : 4 : tqpair.mgmt_pdu->qpair = &tqpair;
1074 : 4 : tcp_req.req.cmd = (union nvmf_h2c_msg *)&tcp_req.cmd;
1075 : 4 : tcp_req.req.rsp = &rsp0;
1076 : 4 : tcp_req.state = TCP_REQUEST_STATE_NEW;
1077 : :
1078 : 4 : TAILQ_INSERT_TAIL(&tqpair.tcp_req_working_queue, &tcp_req, state_link);
1079 : 4 : tqpair.state_cntr[TCP_REQUEST_STATE_NEW]++;
1080 : :
1081 : : /* init pdu, make pdu need sgl buff */
1082 : 4 : capsule_data = &tqpair.pdu_in_progress->hdr.capsule_cmd;
1083 : 4 : sgl = &capsule_data->ccsqe.dptr.sgl1;
1084 : :
1085 : 4 : capsule_data->common.pdu_type = SPDK_NVME_TCP_PDU_TYPE_CAPSULE_CMD;
1086 : 4 : capsule_data->common.hlen = sizeof(*capsule_data);
1087 : 4 : capsule_data->common.plen = 1096;
1088 : 4 : capsule_data->ccsqe.opc = SPDK_NVME_OPC_WRITE;
1089 : : /* Need to set to a non zero valid to check it gets copied to the response */
1090 : 4 : capsule_data->ccsqe.cid = cid;
1091 : :
1092 : : /* Set up SGL to ensure nvmf_tcp_req_parse_sgl returns an error */
1093 : 4 : sgl->unkeyed.subtype = SPDK_NVME_SGL_SUBTYPE_TRANSPORT;
1094 : 4 : sgl->generic.type = SPDK_NVME_SGL_TYPE_TRANSPORT_DATA_BLOCK;
1095 : 4 : sgl->unkeyed.length = UT_MAX_IO_SIZE + 1;
1096 : :
1097 : : /* Process a command and ensure that it fails and the request is set up to return an error */
1098 : 4 : nvmf_tcp_req_process(&ttransport, &tcp_req);
1099 : 4 : CU_ASSERT(!STAILQ_EMPTY(&group->pending_buf_queue));
1100 : 4 : CU_ASSERT(tcp_req.state == TCP_REQUEST_STATE_NEED_BUFFER);
1101 : 4 : CU_ASSERT(tqpair.recv_state == NVME_TCP_PDU_RECV_STATE_QUIESCING);
1102 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.pdu_type == SPDK_NVME_TCP_PDU_TYPE_C2H_TERM_REQ);
1103 : 4 : }
1104 : :
1105 : : static void
1106 : 4 : test_nvmf_tcp_pdu_ch_handle(void)
1107 : : {
1108 : 4 : struct spdk_nvmf_tcp_qpair tqpair = {};
1109 : 4 : struct nvme_tcp_pdu mgmt_pdu = {}, pdu_in_progress = {};
1110 : :
1111 : 4 : mgmt_pdu.qpair = &tqpair;
1112 : 4 : tqpair.mgmt_pdu = &mgmt_pdu;
1113 : 4 : tqpair.pdu_in_progress = &pdu_in_progress;
1114 : 4 : tqpair.recv_state = NVME_TCP_PDU_RECV_STATE_AWAIT_PDU_CH;
1115 : 4 : tqpair.cpda = 0;
1116 : :
1117 : : /* Test case: Already received ICreq PDU. Expect: fail */
1118 : 4 : tqpair.pdu_in_progress->hdr.common.pdu_type = SPDK_NVME_TCP_PDU_TYPE_IC_REQ;
1119 : 4 : tqpair.state = NVME_TCP_QPAIR_STATE_INITIALIZING;
1120 : 4 : nvmf_tcp_pdu_ch_handle(&tqpair);
1121 : 4 : CU_ASSERT(tqpair.recv_state == NVME_TCP_PDU_RECV_STATE_QUIESCING);
1122 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.pdu_type == SPDK_NVME_TCP_PDU_TYPE_C2H_TERM_REQ);
1123 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.hlen == sizeof(struct spdk_nvme_tcp_term_req_hdr));
1124 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.plen == tqpair.mgmt_pdu->hdr.term_req.common.hlen);
1125 : :
1126 : : /* Test case: Expected PDU header length and received are different. Expect: fail */
1127 : 4 : tqpair.recv_state = NVME_TCP_PDU_RECV_STATE_AWAIT_PDU_CH;
1128 : 4 : tqpair.pdu_in_progress->hdr.common.pdu_type = SPDK_NVME_TCP_PDU_TYPE_IC_REQ;
1129 : 4 : tqpair.state = NVME_TCP_QPAIR_STATE_INVALID;
1130 : 4 : tqpair.pdu_in_progress->hdr.common.plen = sizeof(struct spdk_nvme_tcp_ic_req);
1131 : 4 : tqpair.pdu_in_progress->hdr.common.hlen = 0;
1132 : 4 : nvmf_tcp_pdu_ch_handle(&tqpair);
1133 : 4 : CU_ASSERT(tqpair.recv_state == NVME_TCP_PDU_RECV_STATE_QUIESCING);
1134 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.pdu_type == SPDK_NVME_TCP_PDU_TYPE_C2H_TERM_REQ);
1135 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.hlen == sizeof(struct spdk_nvme_tcp_term_req_hdr));
1136 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.plen == tqpair.mgmt_pdu->hdr.term_req.common.hlen);
1137 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.fei[0] == 2);
1138 : :
1139 : : /* Test case: The TCP/IP tqpair connection is not negotitated. Expect: fail */
1140 : 4 : tqpair.recv_state = NVME_TCP_PDU_RECV_STATE_AWAIT_PDU_CH;
1141 : 4 : tqpair.pdu_in_progress->hdr.common.pdu_type = SPDK_NVME_TCP_PDU_TYPE_IC_RESP;
1142 : 4 : tqpair.state = NVME_TCP_QPAIR_STATE_INVALID;
1143 : 4 : tqpair.pdu_in_progress->hdr.common.plen = sizeof(struct spdk_nvme_tcp_ic_req);
1144 : 4 : tqpair.pdu_in_progress->hdr.common.hlen = 0;
1145 : 4 : nvmf_tcp_pdu_ch_handle(&tqpair);
1146 : 4 : CU_ASSERT(tqpair.recv_state == NVME_TCP_PDU_RECV_STATE_QUIESCING);
1147 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.pdu_type == SPDK_NVME_TCP_PDU_TYPE_C2H_TERM_REQ);
1148 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.hlen == sizeof(struct spdk_nvme_tcp_term_req_hdr));
1149 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.plen == tqpair.mgmt_pdu->hdr.term_req.common.hlen);
1150 : :
1151 : : /* Test case: Unexpected PDU type. Expect: fail */
1152 : 4 : tqpair.recv_state = NVME_TCP_PDU_RECV_STATE_AWAIT_PDU_CH;
1153 : 4 : tqpair.pdu_in_progress->hdr.common.pdu_type = SPDK_NVME_TCP_PDU_TYPE_CAPSULE_RESP;
1154 : 4 : tqpair.state = NVME_TCP_QPAIR_STATE_RUNNING;
1155 : 4 : tqpair.pdu_in_progress->hdr.common.plen = 0;
1156 : 4 : tqpair.pdu_in_progress->hdr.common.hlen = sizeof(struct spdk_nvme_tcp_ic_req);
1157 : 4 : nvmf_tcp_pdu_ch_handle(&tqpair);
1158 : 4 : CU_ASSERT(tqpair.recv_state == NVME_TCP_PDU_RECV_STATE_QUIESCING);
1159 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.pdu_type == SPDK_NVME_TCP_PDU_TYPE_C2H_TERM_REQ);
1160 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.hlen == sizeof(struct spdk_nvme_tcp_term_req_hdr));
1161 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.plen == tqpair.mgmt_pdu->hdr.term_req.common.hlen +
1162 : : (unsigned)SPDK_NVME_TCP_TERM_REQ_ERROR_DATA_MAX_SIZE);
1163 : :
1164 : : /* Test case: PDU type is SPDK_NVME_TCP_PDU_TYPE_IC_REQ, let plen error. Expect: fail */
1165 : 4 : tqpair.recv_state = NVME_TCP_PDU_RECV_STATE_AWAIT_PDU_CH;
1166 : 4 : tqpair.pdu_in_progress->hdr.common.pdu_type = SPDK_NVME_TCP_PDU_TYPE_IC_REQ;
1167 : 4 : tqpair.state = NVME_TCP_QPAIR_STATE_INVALID;
1168 : 4 : tqpair.pdu_in_progress->hdr.common.plen = 0;
1169 : 4 : tqpair.pdu_in_progress->hdr.common.hlen = sizeof(struct spdk_nvme_tcp_ic_req);
1170 : 4 : nvmf_tcp_pdu_ch_handle(&tqpair);
1171 : 4 : CU_ASSERT(tqpair.recv_state == NVME_TCP_PDU_RECV_STATE_QUIESCING);
1172 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.pdu_type == SPDK_NVME_TCP_PDU_TYPE_C2H_TERM_REQ);
1173 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.hlen == sizeof(struct spdk_nvme_tcp_term_req_hdr));
1174 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.plen == tqpair.mgmt_pdu->hdr.term_req.common.hlen +
1175 : : (unsigned)SPDK_NVME_TCP_TERM_REQ_ERROR_DATA_MAX_SIZE);
1176 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.fei[0] == 4);
1177 : :
1178 : : /* Test case: PDU type is SPDK_NVME_TCP_PDU_TYPE_CAPSULE_CMD, let plen error. Expect: fail */
1179 : 4 : tqpair.recv_state = NVME_TCP_PDU_RECV_STATE_AWAIT_PDU_CH;
1180 : 4 : tqpair.pdu_in_progress->hdr.common.pdu_type = SPDK_NVME_TCP_PDU_TYPE_CAPSULE_CMD;
1181 : 4 : tqpair.state = NVME_TCP_QPAIR_STATE_RUNNING;
1182 : 4 : tqpair.pdu_in_progress->hdr.common.flags = SPDK_NVME_TCP_CH_FLAGS_HDGSTF;
1183 : 4 : tqpair.pdu_in_progress->hdr.common.plen = 0;
1184 : 4 : tqpair.pdu_in_progress->hdr.common.hlen = sizeof(struct spdk_nvme_tcp_cmd);
1185 : 4 : nvmf_tcp_pdu_ch_handle(&tqpair);
1186 : 4 : CU_ASSERT(tqpair.recv_state == NVME_TCP_PDU_RECV_STATE_QUIESCING);
1187 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.pdu_type == SPDK_NVME_TCP_PDU_TYPE_C2H_TERM_REQ);
1188 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.hlen == sizeof(struct spdk_nvme_tcp_term_req_hdr));
1189 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.plen == sizeof(struct spdk_nvme_tcp_cmd) + sizeof(
1190 : : struct spdk_nvme_tcp_term_req_hdr));
1191 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.fei[0] == 4);
1192 : :
1193 : : /* Test case: PDU type is SPDK_NVME_TCP_PDU_TYPE_H2C_DATA, let plen error. Expect: fail */
1194 : 4 : tqpair.recv_state = NVME_TCP_PDU_RECV_STATE_AWAIT_PDU_CH;
1195 : 4 : tqpair.pdu_in_progress->hdr.common.pdu_type = SPDK_NVME_TCP_PDU_TYPE_H2C_DATA;
1196 : 4 : tqpair.state = NVME_TCP_QPAIR_STATE_RUNNING;
1197 : 4 : tqpair.pdu_in_progress->hdr.common.plen = 0;
1198 : 4 : tqpair.pdu_in_progress->hdr.common.pdo = 64;
1199 : 4 : tqpair.pdu_in_progress->hdr.common.hlen = sizeof(struct spdk_nvme_tcp_h2c_data_hdr);
1200 : 4 : nvmf_tcp_pdu_ch_handle(&tqpair);
1201 : 4 : CU_ASSERT(tqpair.recv_state == NVME_TCP_PDU_RECV_STATE_QUIESCING);
1202 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.pdu_type == SPDK_NVME_TCP_PDU_TYPE_C2H_TERM_REQ);
1203 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.hlen == sizeof(struct spdk_nvme_tcp_term_req_hdr));
1204 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.plen == tqpair.mgmt_pdu->hdr.term_req.common.hlen +
1205 : : (unsigned)sizeof(struct spdk_nvme_tcp_term_req_hdr));
1206 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.fei[0] == 4);
1207 : :
1208 : : /* Test case: PDU type is SPDK_NVME_TCP_PDU_TYPE_H2C_TERM_REQ, let plen error. Expect: fail */
1209 : 4 : tqpair.recv_state = NVME_TCP_PDU_RECV_STATE_AWAIT_PDU_CH;
1210 : 4 : tqpair.pdu_in_progress->hdr.common.pdu_type = SPDK_NVME_TCP_PDU_TYPE_H2C_TERM_REQ;
1211 : 4 : tqpair.state = NVME_TCP_QPAIR_STATE_RUNNING;
1212 : 4 : tqpair.pdu_in_progress->hdr.common.plen = 0;
1213 : 4 : tqpair.pdu_in_progress->hdr.common.hlen = sizeof(struct spdk_nvme_tcp_term_req_hdr);
1214 : 4 : nvmf_tcp_pdu_ch_handle(&tqpair);
1215 : 4 : CU_ASSERT(tqpair.recv_state == NVME_TCP_PDU_RECV_STATE_QUIESCING);
1216 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.pdu_type == SPDK_NVME_TCP_PDU_TYPE_C2H_TERM_REQ);
1217 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.hlen == sizeof(struct spdk_nvme_tcp_term_req_hdr));
1218 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.plen == tqpair.mgmt_pdu->hdr.term_req.common.hlen +
1219 : : (unsigned)sizeof(struct spdk_nvme_tcp_term_req_hdr));
1220 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.fei[0] == 4);
1221 : :
1222 : : /* Test case: PDU type is SPDK_NVME_TCP_PDU_TYPE_CAPSULE_CMD, let pdo error. Expect: fail */
1223 : 4 : tqpair.recv_state = NVME_TCP_PDU_RECV_STATE_AWAIT_PDU_CH;
1224 : 4 : tqpair.pdu_in_progress->hdr.common.pdu_type = SPDK_NVME_TCP_PDU_TYPE_CAPSULE_CMD;
1225 : 4 : tqpair.state = NVME_TCP_QPAIR_STATE_RUNNING;
1226 : 4 : tqpair.cpda = 1;
1227 : 4 : tqpair.pdu_in_progress->hdr.common.flags = SPDK_NVME_TCP_CH_FLAGS_HDGSTF;
1228 : 4 : tqpair.pdu_in_progress->hdr.common.plen = 0;
1229 : 4 : tqpair.pdu_in_progress->hdr.common.pdo = 63;
1230 : 4 : tqpair.pdu_in_progress->hdr.common.hlen = sizeof(struct spdk_nvme_tcp_cmd);
1231 : 4 : nvmf_tcp_pdu_ch_handle(&tqpair);
1232 : 4 : CU_ASSERT(tqpair.recv_state == NVME_TCP_PDU_RECV_STATE_QUIESCING);
1233 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.pdu_type == SPDK_NVME_TCP_PDU_TYPE_C2H_TERM_REQ);
1234 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.hlen == sizeof(struct spdk_nvme_tcp_term_req_hdr));
1235 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.plen == sizeof(struct spdk_nvme_tcp_cmd) + sizeof(
1236 : : struct spdk_nvme_tcp_term_req_hdr));
1237 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.fei[0] == 3);
1238 : :
1239 : : /* Test case: PDU type is SPDK_NVME_TCP_PDU_TYPE_H2C_DATA, let pdo error. Expect: fail */
1240 : 4 : tqpair.recv_state = NVME_TCP_PDU_RECV_STATE_AWAIT_PDU_CH;
1241 : 4 : tqpair.pdu_in_progress->hdr.common.pdu_type = SPDK_NVME_TCP_PDU_TYPE_H2C_DATA;
1242 : 4 : tqpair.state = NVME_TCP_QPAIR_STATE_RUNNING;
1243 : 4 : tqpair.cpda = 1;
1244 : 4 : tqpair.pdu_in_progress->hdr.common.plen = 0;
1245 : 4 : tqpair.pdu_in_progress->hdr.common.pdo = 63;
1246 : 4 : tqpair.pdu_in_progress->hdr.common.hlen = sizeof(struct spdk_nvme_tcp_h2c_data_hdr);
1247 : 4 : nvmf_tcp_pdu_ch_handle(&tqpair);
1248 : 4 : CU_ASSERT(tqpair.recv_state == NVME_TCP_PDU_RECV_STATE_QUIESCING);
1249 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.pdu_type == SPDK_NVME_TCP_PDU_TYPE_C2H_TERM_REQ);
1250 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.hlen == sizeof(struct spdk_nvme_tcp_term_req_hdr));
1251 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.common.plen == tqpair.mgmt_pdu->hdr.term_req.common.hlen +
1252 : : (unsigned)sizeof(struct spdk_nvme_tcp_term_req_hdr));
1253 : 4 : CU_ASSERT(tqpair.mgmt_pdu->hdr.term_req.fei[0] == 3);
1254 : :
1255 : : /* Test case: All parameters is conformed to the function. Expect: PASS */
1256 : 4 : tqpair.recv_state = NVME_TCP_PDU_RECV_STATE_AWAIT_PDU_CH;
1257 : 4 : tqpair.pdu_in_progress->hdr.common.pdu_type = SPDK_NVME_TCP_PDU_TYPE_IC_REQ;
1258 : 4 : tqpair.state = NVME_TCP_QPAIR_STATE_INVALID;
1259 : 4 : tqpair.pdu_in_progress->hdr.common.plen = sizeof(struct spdk_nvme_tcp_ic_req);
1260 : 4 : tqpair.pdu_in_progress->hdr.common.hlen = sizeof(struct spdk_nvme_tcp_ic_req);
1261 : 4 : nvmf_tcp_pdu_ch_handle(&tqpair);
1262 : 4 : CU_ASSERT(tqpair.recv_state == NVME_TCP_PDU_RECV_STATE_AWAIT_PDU_PSH);
1263 : 4 : CU_ASSERT(tqpair.pdu_in_progress->psh_len == tqpair.pdu_in_progress->hdr.common.hlen - sizeof(
1264 : : struct spdk_nvme_tcp_common_pdu_hdr));
1265 : 4 : }
1266 : :
1267 : : static void
1268 : 4 : test_nvmf_tcp_tls_add_remove_credentials(void)
1269 : : {
1270 : : struct spdk_thread *thread;
1271 : : struct spdk_nvmf_transport *transport;
1272 : : struct spdk_nvmf_tcp_transport *ttransport;
1273 : 4 : struct spdk_nvmf_transport_opts opts;
1274 : 4 : struct spdk_nvmf_subsystem subsystem;
1275 : : struct tcp_psk_entry *entry;
1276 : 4 : const char subnqn[] = {"nqn.2016-06.io.spdk:cnode1"};
1277 : 4 : const char hostnqn[] = {"nqn.2016-06.io.spdk:host1"};
1278 : 4 : const char *psk = "NVMeTLSkey-1:01:VRLbtnN9AQb2WXW3c9+wEf/DRLz0QuLdbYvEhwtdWwNf9LrZ:";
1279 : 4 : char *psk_file_path = "/tmp/psk.txt";
1280 : 4 : bool found = false;
1281 : 4 : FILE *psk_file = NULL;
1282 : : mode_t oldmask;
1283 : :
1284 : 4 : thread = spdk_thread_create(NULL, NULL);
1285 [ - + ]: 4 : SPDK_CU_ASSERT_FATAL(thread != NULL);
1286 : 4 : spdk_set_thread(thread);
1287 : :
1288 : 4 : memset(&opts, 0, sizeof(opts));
1289 : 4 : opts.max_queue_depth = UT_MAX_QUEUE_DEPTH;
1290 : 4 : opts.max_qpairs_per_ctrlr = UT_MAX_QPAIRS_PER_CTRLR;
1291 : 4 : opts.in_capsule_data_size = UT_IN_CAPSULE_DATA_SIZE;
1292 : 4 : opts.max_io_size = UT_MAX_IO_SIZE;
1293 : 4 : opts.io_unit_size = UT_IO_UNIT_SIZE;
1294 : 4 : opts.max_aq_depth = UT_MAX_AQ_DEPTH;
1295 : 4 : opts.num_shared_buffers = UT_NUM_SHARED_BUFFERS;
1296 : 4 : transport = nvmf_tcp_create(&opts);
1297 : :
1298 : 4 : memset(&subsystem, 0, sizeof(subsystem));
1299 : 4 : snprintf(subsystem.subnqn, sizeof(subsystem.subnqn), "%s", subnqn);
1300 : :
1301 : : /* Create a text file containing PSK in interchange format. */
1302 : 4 : oldmask = umask(S_IXUSR | S_IRWXG | S_IRWXO);
1303 : 4 : psk_file = fopen(psk_file_path, "w");
1304 : 4 : CU_ASSERT(psk_file != NULL);
1305 [ - + ]: 4 : CU_ASSERT(fprintf(psk_file, "%s", psk) > 0);
1306 : 4 : CU_ASSERT(fclose(psk_file) == 0);
1307 : 4 : umask(oldmask);
1308 : :
1309 : 4 : struct spdk_json_val psk_json[] = {
1310 : : {"", 2, SPDK_JSON_VAL_OBJECT_BEGIN},
1311 : : {"psk", 3, SPDK_JSON_VAL_NAME},
1312 [ - + ]: 4 : {psk_file_path, strlen(psk_file_path), SPDK_JSON_VAL_STRING},
1313 : : {"", 0, SPDK_JSON_VAL_OBJECT_END},
1314 : : };
1315 : :
1316 : 4 : nvmf_tcp_subsystem_add_host(transport, &subsystem, hostnqn, psk_json);
1317 : :
1318 : 4 : ttransport = SPDK_CONTAINEROF(transport, struct spdk_nvmf_tcp_transport, transport);
1319 [ + + ]: 8 : TAILQ_FOREACH(entry, &ttransport->psks, link) {
1320 [ + + + - ]: 4 : if ((strcmp(subnqn, entry->subnqn) == 0) &&
1321 [ + + + - ]: 4 : (strcmp(hostnqn, entry->hostnqn) == 0)) {
1322 : 4 : found = true;
1323 : : }
1324 : : }
1325 : :
1326 : 4 : CU_ASSERT(found == true);
1327 : 4 : found = false;
1328 : :
1329 : 4 : nvmf_tcp_subsystem_remove_host(transport, &subsystem, hostnqn);
1330 : :
1331 : 4 : ttransport = SPDK_CONTAINEROF(transport, struct spdk_nvmf_tcp_transport, transport);
1332 [ - + ]: 4 : TAILQ_FOREACH(entry, &ttransport->psks, link) {
1333 [ # # # # ]: 0 : if ((strcmp(subnqn, entry->subnqn) == 0) &&
1334 [ # # # # ]: 0 : (strcmp(hostnqn, entry->hostnqn) == 0)) {
1335 : 0 : found = true;
1336 : : }
1337 : : }
1338 : :
1339 : 4 : CU_ASSERT(found == false);
1340 : :
1341 : 4 : CU_ASSERT(remove(psk_file_path) == 0);
1342 : :
1343 : 4 : CU_ASSERT(nvmf_tcp_destroy(transport, NULL, NULL) == 0);
1344 : :
1345 : 4 : spdk_thread_exit(thread);
1346 [ + + ]: 8 : while (!spdk_thread_is_exited(thread)) {
1347 : 4 : spdk_thread_poll(thread, 0, 0);
1348 : : }
1349 : 4 : spdk_thread_destroy(thread);
1350 : 4 : }
1351 : :
1352 : : static void
1353 : 4 : test_nvmf_tcp_tls_generate_psk_id(void)
1354 : : {
1355 : 4 : const char psk_id_reference[] = {"NVMe0R01 nqn.2016-06.io.spdk:host1 nqn.2016-06.io.spdk:cnode1"};
1356 : 4 : const char subnqn[] = {"nqn.2016-06.io.spdk:cnode1"};
1357 : 4 : const char hostnqn[] = {"nqn.2016-06.io.spdk:host1"};
1358 : 4 : char psk_id[NVMF_PSK_IDENTITY_LEN] = {};
1359 : 4 : char too_small_psk_id[5] = {};
1360 : :
1361 : : /* Check if we can generate expected PSK id. */
1362 : 4 : CU_ASSERT(nvme_tcp_generate_psk_identity(psk_id, NVMF_PSK_IDENTITY_LEN, hostnqn,
1363 : : subnqn, NVME_TCP_CIPHER_AES_128_GCM_SHA256) == 0);
1364 : 4 : CU_ASSERT(strcmp(psk_id, psk_id_reference) == 0);
1365 : :
1366 : : /* Test with a buffer that is too small to fit PSK id. */
1367 : 4 : CU_ASSERT(nvme_tcp_generate_psk_identity(too_small_psk_id, sizeof(too_small_psk_id), hostnqn,
1368 : : subnqn, NVME_TCP_CIPHER_AES_128_GCM_SHA256) != 0);
1369 : :
1370 : : /* Test with unknown cipher suite. */
1371 : 4 : CU_ASSERT(nvme_tcp_generate_psk_identity(psk_id, NVMF_PSK_IDENTITY_LEN, hostnqn,
1372 : : subnqn, UINT8_MAX) != 0);
1373 : 4 : }
1374 : :
1375 : : static void
1376 : 4 : test_nvmf_tcp_tls_generate_retained_psk(void)
1377 : : {
1378 : 4 : const char hostnqn[] = {"nqn.2016-06.io.spdk:host1"};
1379 : 4 : const char psk_reference1[] = {"1234567890ABCDEF"};
1380 : 4 : const char psk_reference2[] = {"FEDCBA0987654321"};
1381 : 4 : uint8_t unhexlified_str1[SPDK_TLS_PSK_MAX_LEN] = {};
1382 : 4 : uint8_t unhexlified_str2[SPDK_TLS_PSK_MAX_LEN] = {};
1383 : : char *unhexlified1;
1384 : : char *unhexlified2;
1385 : 4 : uint8_t psk_retained1[SPDK_TLS_PSK_MAX_LEN] = {};
1386 : 4 : uint8_t psk_retained2[SPDK_TLS_PSK_MAX_LEN] = {};
1387 : 4 : uint8_t too_small_psk_retained[5] = {};
1388 : : int psk_retained_len1, psk_retained_len2;
1389 : : int retained_size;
1390 : :
1391 : 4 : unhexlified1 = spdk_unhexlify(psk_reference1);
1392 [ - + ]: 4 : SPDK_CU_ASSERT_FATAL(unhexlified1 != NULL);
1393 : 4 : unhexlified2 = spdk_unhexlify(psk_reference2);
1394 [ - + ]: 4 : SPDK_CU_ASSERT_FATAL(unhexlified2 != NULL);
1395 : :
1396 [ - + ]: 4 : memcpy(unhexlified_str1, unhexlified1, strlen(psk_reference1) / 2);
1397 [ - + ]: 4 : memcpy(unhexlified_str2, unhexlified2, strlen(psk_reference2) / 2);
1398 : 4 : free(unhexlified1);
1399 : 4 : free(unhexlified2);
1400 : :
1401 : : /* Make sure that retained PSKs are different with different input PSKs and the same hash. */
1402 : 4 : retained_size = nvme_tcp_derive_retained_psk(unhexlified_str1, strlen(psk_reference1) / 2, hostnqn,
1403 : : psk_retained1, SPDK_TLS_PSK_MAX_LEN, NVME_TCP_HASH_ALGORITHM_SHA256);
1404 : 4 : CU_ASSERT(retained_size > 0);
1405 : :
1406 : 4 : CU_ASSERT(nvme_tcp_derive_retained_psk(unhexlified_str2, strlen(psk_reference2) / 2, hostnqn,
1407 : : psk_retained2,
1408 : : SPDK_TLS_PSK_MAX_LEN, NVME_TCP_HASH_ALGORITHM_SHA256) > 0);
1409 : 4 : CU_ASSERT(memcmp(psk_retained1, psk_retained2, retained_size) != 0);
1410 : :
1411 : : /* Make sure that retained PSKs are different with different hash and the same input PSKs. */
1412 : 4 : psk_retained_len1 = nvme_tcp_derive_retained_psk(unhexlified_str1, strlen(psk_reference1) / 2,
1413 : : hostnqn, psk_retained1, SPDK_TLS_PSK_MAX_LEN, NVME_TCP_HASH_ALGORITHM_SHA256);
1414 : 4 : CU_ASSERT(psk_retained_len1 > 0);
1415 : 4 : psk_retained_len2 = nvme_tcp_derive_retained_psk(unhexlified_str1, strlen(psk_reference1) / 2,
1416 : : hostnqn, psk_retained2, SPDK_TLS_PSK_MAX_LEN, NVME_TCP_HASH_ALGORITHM_SHA384);
1417 : 4 : CU_ASSERT(psk_retained_len2 > 0);
1418 : 4 : CU_ASSERT(psk_retained_len1 < psk_retained_len2);
1419 : :
1420 : : /* Make sure that passing unknown value as hash errors out the function. */
1421 : 4 : CU_ASSERT(nvme_tcp_derive_retained_psk(unhexlified_str1, strlen(psk_reference1) / 2, hostnqn,
1422 : : psk_retained1, SPDK_TLS_PSK_MAX_LEN, -1) < 0);
1423 : :
1424 : : /* Make sure that passing buffer insufficient in size errors out the function. */
1425 : 4 : CU_ASSERT(nvme_tcp_derive_retained_psk(unhexlified_str1, strlen(psk_reference1) / 2, hostnqn,
1426 : : too_small_psk_retained, sizeof(too_small_psk_retained), NVME_TCP_HASH_ALGORITHM_SHA256) < 0);
1427 : 4 : }
1428 : :
1429 : : static void
1430 : 4 : test_nvmf_tcp_tls_generate_tls_psk(void)
1431 : : {
1432 : 4 : const char psk_id_reference[] = {"NVMe0R01 nqn.2016-06.io.spdk:host1 nqn.2016-06.io.spdk:cnode1"};
1433 : 4 : const char hostnqn[] = {"nqn.2016-06.io.spdk:host1"};
1434 : 4 : const char psk_reference[] = {"1234567890ABCDEF"};
1435 : : char *unhexlified;
1436 : 4 : uint8_t unhexlified_str[SPDK_TLS_PSK_MAX_LEN] = {};
1437 : 4 : uint8_t psk_retained[SPDK_TLS_PSK_MAX_LEN] = {};
1438 : 4 : uint8_t psk_key1[SPDK_TLS_PSK_MAX_LEN] = {}, psk_key2[SPDK_TLS_PSK_MAX_LEN] = {};
1439 : 4 : uint8_t too_small_psk_tls[5] = {};
1440 : : int retained_size, tls_size;
1441 : :
1442 : 4 : unhexlified = spdk_unhexlify(psk_reference);
1443 : 4 : CU_ASSERT(unhexlified != NULL);
1444 : :
1445 [ - + ]: 4 : memcpy(unhexlified_str, unhexlified, strlen(psk_reference) / 2);
1446 : 4 : free(unhexlified);
1447 : :
1448 : 4 : retained_size = nvme_tcp_derive_retained_psk(unhexlified_str, strlen(psk_reference) / 2, hostnqn,
1449 : : psk_retained, SPDK_TLS_PSK_MAX_LEN, NVME_TCP_HASH_ALGORITHM_SHA256);
1450 : 4 : CU_ASSERT(retained_size > 0);
1451 : :
1452 : : /* Make sure that different cipher suites produce different TLS PSKs. */
1453 : 4 : tls_size = nvme_tcp_derive_tls_psk(psk_retained, retained_size, psk_id_reference, psk_key1,
1454 : : SPDK_TLS_PSK_MAX_LEN, NVME_TCP_CIPHER_AES_128_GCM_SHA256);
1455 : 4 : CU_ASSERT(tls_size > 0);
1456 : 4 : CU_ASSERT(nvme_tcp_derive_tls_psk(psk_retained, retained_size, psk_id_reference, psk_key2,
1457 : : SPDK_TLS_PSK_MAX_LEN, NVME_TCP_CIPHER_AES_256_GCM_SHA384) > 0);
1458 : 4 : CU_ASSERT(memcmp(psk_key1, psk_key2, tls_size) != 0);
1459 : :
1460 : : /* Make sure that passing unknown value as hash errors out the function. */
1461 : 4 : CU_ASSERT(nvme_tcp_derive_tls_psk(psk_retained, retained_size, psk_id_reference,
1462 : : psk_key1, SPDK_TLS_PSK_MAX_LEN, UINT8_MAX) < 0);
1463 : :
1464 : : /* Make sure that passing buffer insufficient in size errors out the function. */
1465 : 4 : CU_ASSERT(nvme_tcp_derive_tls_psk(psk_retained, retained_size, psk_id_reference,
1466 : : too_small_psk_tls, sizeof(too_small_psk_tls),
1467 : : NVME_TCP_CIPHER_AES_128_GCM_SHA256) < 0);
1468 : 4 : }
1469 : :
1470 : : int
1471 : 4 : main(int argc, char **argv)
1472 : : {
1473 : 4 : CU_pSuite suite = NULL;
1474 : : unsigned int num_failures;
1475 : :
1476 : 4 : CU_initialize_registry();
1477 : :
1478 : 4 : suite = CU_add_suite("nvmf", NULL, NULL);
1479 : :
1480 : 4 : CU_ADD_TEST(suite, test_nvmf_tcp_create);
1481 : 4 : CU_ADD_TEST(suite, test_nvmf_tcp_destroy);
1482 : 4 : CU_ADD_TEST(suite, test_nvmf_tcp_poll_group_create);
1483 : 4 : CU_ADD_TEST(suite, test_nvmf_tcp_send_c2h_data);
1484 : 4 : CU_ADD_TEST(suite, test_nvmf_tcp_h2c_data_hdr_handle);
1485 : 4 : CU_ADD_TEST(suite, test_nvmf_tcp_in_capsule_data_handle);
1486 : 4 : CU_ADD_TEST(suite, test_nvmf_tcp_qpair_init_mem_resource);
1487 : 4 : CU_ADD_TEST(suite, test_nvmf_tcp_send_c2h_term_req);
1488 : 4 : CU_ADD_TEST(suite, test_nvmf_tcp_send_capsule_resp_pdu);
1489 : 4 : CU_ADD_TEST(suite, test_nvmf_tcp_icreq_handle);
1490 : 4 : CU_ADD_TEST(suite, test_nvmf_tcp_check_xfer_type);
1491 : 4 : CU_ADD_TEST(suite, test_nvmf_tcp_invalid_sgl);
1492 : 4 : CU_ADD_TEST(suite, test_nvmf_tcp_pdu_ch_handle);
1493 : 4 : CU_ADD_TEST(suite, test_nvmf_tcp_tls_add_remove_credentials);
1494 : 4 : CU_ADD_TEST(suite, test_nvmf_tcp_tls_generate_psk_id);
1495 : 4 : CU_ADD_TEST(suite, test_nvmf_tcp_tls_generate_retained_psk);
1496 : 4 : CU_ADD_TEST(suite, test_nvmf_tcp_tls_generate_tls_psk);
1497 : :
1498 : 4 : num_failures = spdk_ut_run_tests(argc, argv, NULL);
1499 : 4 : CU_cleanup_registry();
1500 : 4 : return num_failures;
1501 : : }
|