Branch data Line data Source code
1 : : /* SPDX-License-Identifier: BSD-3-Clause
2 : : * Copyright (C) 2015 Intel Corporation. All rights reserved.
3 : : * Copyright (c) 2020 Mellanox Technologies LTD. All rights reserved.
4 : : */
5 : :
6 : : #include "spdk_internal/cunit.h"
7 : :
8 : : #include "spdk/env.h"
9 : :
10 : : #include "nvme/nvme.c"
11 : :
12 : : #include "spdk_internal/mock.h"
13 : :
14 : : #include "common/lib/test_env.c"
15 : :
16 : 32 : DEFINE_STUB_V(nvme_ctrlr_proc_get_ref, (struct spdk_nvme_ctrlr *ctrlr));
17 : 44 : DEFINE_STUB_V(nvme_ctrlr_proc_put_ref, (struct spdk_nvme_ctrlr *ctrlr));
18 : 4 : DEFINE_STUB_V(nvme_ctrlr_fail, (struct spdk_nvme_ctrlr *ctrlr, bool hotremove));
19 [ - + - + ]: 52 : DEFINE_STUB(spdk_nvme_transport_available_by_name, bool,
20 : : (const char *transport_name), true);
21 : : /* return anything non-NULL, this won't be dereferenced anywhere in this test */
22 [ - + - + ]: 24 : DEFINE_STUB(nvme_ctrlr_get_current_process, struct spdk_nvme_ctrlr_process *,
23 : : (struct spdk_nvme_ctrlr *ctrlr), (struct spdk_nvme_ctrlr_process *)(uintptr_t)0x1);
24 [ - + - + ]: 12 : DEFINE_STUB(nvme_ctrlr_process_init, int,
25 : : (struct spdk_nvme_ctrlr *ctrlr), 0);
26 [ - + - + ]: 44 : DEFINE_STUB(nvme_ctrlr_get_ref_count, int,
27 : : (struct spdk_nvme_ctrlr *ctrlr), 0);
28 [ - + - + ]: 24 : DEFINE_STUB(dummy_probe_cb, bool,
29 : : (void *cb_ctx, const struct spdk_nvme_transport_id *trid,
30 : : struct spdk_nvme_ctrlr_opts *opts), false);
31 [ - + # # ]: 16 : DEFINE_STUB(nvme_transport_ctrlr_construct, struct spdk_nvme_ctrlr *,
32 : : (const struct spdk_nvme_transport_id *trid,
33 : : const struct spdk_nvme_ctrlr_opts *opts,
34 : : void *devhandle), NULL);
35 : 40 : DEFINE_STUB_V(nvme_io_msg_ctrlr_detach, (struct spdk_nvme_ctrlr *ctrlr));
36 [ # # # # ]: 0 : DEFINE_STUB(spdk_nvme_transport_available, bool,
37 : : (enum spdk_nvme_transport_type trtype), true);
38 [ - + # # ]: 13 : DEFINE_STUB(spdk_pci_event_listen, int, (void), 0);
39 [ # # # # ]: 0 : DEFINE_STUB(spdk_nvme_poll_group_process_completions, int64_t, (struct spdk_nvme_poll_group *group,
40 : : uint32_t completions_per_qpair, spdk_nvme_disconnected_qpair_cb disconnected_qpair_cb), 0);
41 [ # # # # ]: 0 : DEFINE_STUB(nvme_transport_ctrlr_scan_attached, int, (struct spdk_nvme_probe_ctx *probe_ctx), 0);
42 : :
43 : : static bool ut_destruct_called = false;
44 : : void
45 : 4 : nvme_ctrlr_destruct(struct spdk_nvme_ctrlr *ctrlr)
46 : : {
47 : 4 : ut_destruct_called = true;
48 : 4 : }
49 : :
50 : : void
51 : 40 : nvme_ctrlr_destruct_async(struct spdk_nvme_ctrlr *ctrlr, struct nvme_ctrlr_detach_ctx *ctx)
52 : : {
53 : 40 : ut_destruct_called = true;
54 : 40 : ctrlr->is_destructed = true;
55 : :
56 : 40 : ctx->shutdown_complete = true;
57 : 40 : }
58 : :
59 : : int
60 : 44 : nvme_ctrlr_destruct_poll_async(struct spdk_nvme_ctrlr *ctrlr,
61 : : struct nvme_ctrlr_detach_ctx *ctx)
62 : : {
63 [ + + + + ]: 44 : if (!ctx->shutdown_complete) {
64 : 4 : return -EAGAIN;
65 : : }
66 : :
67 [ + - ]: 40 : if (ctx->cb_fn) {
68 : 40 : ctx->cb_fn(ctrlr);
69 : 10 : }
70 : :
71 : 40 : return 0;
72 : 11 : }
73 : :
74 : : union spdk_nvme_csts_register
75 : 16 : spdk_nvme_ctrlr_get_regs_csts(struct spdk_nvme_ctrlr *ctrlr)
76 : : {
77 : 16 : union spdk_nvme_csts_register csts = {};
78 : 16 : return csts;
79 : : }
80 : :
81 : : void
82 : 52 : spdk_nvme_ctrlr_get_default_ctrlr_opts(struct spdk_nvme_ctrlr_opts *opts, size_t opts_size)
83 : : {
84 [ - + ]: 52 : memset(opts, 0, opts_size);
85 : 52 : nvme_get_default_hostnqn(opts->hostnqn, sizeof(opts->hostnqn));
86 : 52 : opts->opts_size = opts_size;
87 : 52 : }
88 : :
89 : : static void
90 : 44 : memset_trid(struct spdk_nvme_transport_id *trid1, struct spdk_nvme_transport_id *trid2)
91 : : {
92 [ - + ]: 44 : memset(trid1, 0, sizeof(struct spdk_nvme_transport_id));
93 [ - + ]: 44 : memset(trid2, 0, sizeof(struct spdk_nvme_transport_id));
94 : 44 : }
95 : :
96 : : static bool ut_check_trtype = false;
97 : : static bool ut_test_probe_internal = false;
98 : :
99 : : static int
100 : 4 : ut_nvme_pcie_ctrlr_scan(struct spdk_nvme_probe_ctx *probe_ctx,
101 : : bool direct_connect)
102 : : {
103 : : struct spdk_nvme_ctrlr *ctrlr;
104 : 4 : struct spdk_nvme_qpair qpair = {};
105 : : int rc;
106 : :
107 [ - + ]: 4 : if (probe_ctx->trid.trtype != SPDK_NVME_TRANSPORT_PCIE) {
108 : 0 : return -1;
109 : : }
110 : :
111 : 4 : ctrlr = calloc(1, sizeof(*ctrlr));
112 : 4 : CU_ASSERT(ctrlr != NULL);
113 : 4 : ctrlr->adminq = &qpair;
114 : :
115 : : /* happy path with first controller */
116 : 4 : MOCK_SET(nvme_transport_ctrlr_construct, ctrlr);
117 : 4 : rc = nvme_ctrlr_probe(&probe_ctx->trid, probe_ctx, NULL);
118 : 4 : CU_ASSERT(rc == 0);
119 : :
120 : : /* failed with the second controller */
121 : 4 : MOCK_SET(nvme_transport_ctrlr_construct, NULL);
122 : 4 : rc = nvme_ctrlr_probe(&probe_ctx->trid, probe_ctx, NULL);
123 : 4 : CU_ASSERT(rc != 0);
124 [ - + - + ]: 4 : MOCK_CLEAR_P(nvme_transport_ctrlr_construct);
125 : :
126 : 4 : return -1;
127 : 1 : }
128 : :
129 : : int
130 : 4 : nvme_transport_ctrlr_destruct(struct spdk_nvme_ctrlr *ctrlr)
131 : : {
132 : 4 : free(ctrlr);
133 : 4 : return 0;
134 : : }
135 : :
136 : : int
137 : 48 : nvme_transport_ctrlr_scan(struct spdk_nvme_probe_ctx *probe_ctx,
138 : : bool direct_connect)
139 : : {
140 : 48 : struct spdk_nvme_ctrlr *ctrlr = NULL;
141 : :
142 [ + + + + ]: 48 : if (ut_check_trtype == true) {
143 : 8 : CU_ASSERT(probe_ctx->trid.trtype == SPDK_NVME_TRANSPORT_PCIE);
144 : 2 : }
145 : :
146 [ + + + + ]: 48 : if (ut_test_probe_internal) {
147 : 4 : return ut_nvme_pcie_ctrlr_scan(probe_ctx, direct_connect);
148 : : }
149 : :
150 [ + + + + ]: 44 : if (direct_connect == true && probe_ctx->probe_cb) {
151 : 16 : nvme_robust_mutex_unlock(&g_spdk_nvme_driver->lock);
152 : 16 : ctrlr = nvme_get_ctrlr_by_trid(&probe_ctx->trid, NULL);
153 : 16 : nvme_robust_mutex_lock(&g_spdk_nvme_driver->lock);
154 : 16 : probe_ctx->probe_cb(probe_ctx->cb_ctx, &probe_ctx->trid, &ctrlr->opts);
155 : 4 : }
156 : 44 : return 0;
157 : 12 : }
158 : :
159 : : static bool ut_attach_cb_called = false;
160 : : static void
161 : 4 : dummy_attach_cb(void *cb_ctx, const struct spdk_nvme_transport_id *trid,
162 : : struct spdk_nvme_ctrlr *ctrlr, const struct spdk_nvme_ctrlr_opts *opts)
163 : : {
164 : 4 : ut_attach_cb_called = true;
165 : 4 : }
166 : :
167 : : static int ut_attach_fail_cb_rc = 0;
168 : : static void
169 : 20 : dummy_attach_fail_cb(void *cb_ctx, const struct spdk_nvme_transport_id *trid, int rc)
170 : : {
171 : 20 : ut_attach_fail_cb_rc = rc;
172 : 20 : }
173 : :
174 : : static void
175 : 4 : test_spdk_nvme_probe_ext(void)
176 : : {
177 : 4 : int rc = 0;
178 : 4 : const struct spdk_nvme_transport_id *trid = NULL;
179 : 4 : void *cb_ctx = NULL;
180 : 4 : spdk_nvme_probe_cb probe_cb = NULL;
181 : 4 : spdk_nvme_attach_cb attach_cb = dummy_attach_cb;
182 : 4 : spdk_nvme_attach_fail_cb attach_fail_cb = dummy_attach_fail_cb;
183 : 4 : spdk_nvme_remove_cb remove_cb = NULL;
184 : 3 : struct spdk_nvme_ctrlr ctrlr;
185 : 3 : pthread_mutexattr_t attr;
186 : 4 : struct nvme_driver dummy = {};
187 : 4 : g_spdk_nvme_driver = &dummy;
188 : :
189 : : /* driver init fails */
190 : 4 : MOCK_SET(spdk_process_is_primary, false);
191 : 4 : MOCK_SET(spdk_memzone_lookup, NULL);
192 : 4 : rc = spdk_nvme_probe_ext(trid, cb_ctx, probe_cb, attach_cb, attach_fail_cb, remove_cb);
193 : 4 : CU_ASSERT(rc == -1);
194 : :
195 : : /*
196 : : * For secondary processes, the attach_cb should automatically get
197 : : * called for any controllers already initialized by the primary
198 : : * process.
199 : : */
200 : 4 : MOCK_SET(spdk_nvme_transport_available_by_name, false);
201 : 4 : MOCK_SET(spdk_process_is_primary, true);
202 : 4 : dummy.initialized = true;
203 : 4 : g_spdk_nvme_driver = &dummy;
204 : 4 : rc = spdk_nvme_probe_ext(trid, cb_ctx, probe_cb, attach_cb, attach_fail_cb, remove_cb);
205 : 4 : CU_ASSERT(rc == -1);
206 : :
207 : : /* driver init passes, transport available, secondary call attach_cb */
208 : 4 : MOCK_SET(spdk_nvme_transport_available_by_name, true);
209 : 4 : MOCK_SET(spdk_process_is_primary, false);
210 : 4 : MOCK_SET(spdk_memzone_lookup, g_spdk_nvme_driver);
211 : 4 : dummy.initialized = true;
212 [ - + ]: 4 : memset(&ctrlr, 0, sizeof(struct spdk_nvme_ctrlr));
213 [ - + ]: 4 : CU_ASSERT(pthread_mutexattr_init(&attr) == 0);
214 [ - + ]: 4 : CU_ASSERT(pthread_mutex_init(&dummy.lock, &attr) == 0);
215 : 4 : TAILQ_INIT(&dummy.shared_attached_ctrlrs);
216 : 4 : TAILQ_INSERT_TAIL(&dummy.shared_attached_ctrlrs, &ctrlr, tailq);
217 : 4 : ut_attach_cb_called = false;
218 : : /* setup nvme_transport_ctrlr_scan() stub to also check the trype */
219 : 4 : ut_check_trtype = true;
220 : 4 : rc = spdk_nvme_probe_ext(trid, cb_ctx, probe_cb, attach_cb, attach_fail_cb, remove_cb);
221 : 4 : CU_ASSERT(rc == 0);
222 [ - + ]: 4 : CU_ASSERT(ut_attach_cb_called == true);
223 : :
224 : : /* driver init passes, transport available, we are primary */
225 : 4 : MOCK_SET(spdk_process_is_primary, true);
226 : 4 : rc = spdk_nvme_probe_ext(trid, cb_ctx, probe_cb, attach_cb, attach_fail_cb, remove_cb);
227 : 4 : CU_ASSERT(rc == 0);
228 : :
229 : 4 : g_spdk_nvme_driver = NULL;
230 : : /* reset to pre-test values */
231 [ - + - + ]: 4 : MOCK_CLEAR(spdk_memzone_lookup);
232 : 4 : ut_check_trtype = false;
233 : :
234 [ - + ]: 4 : pthread_mutex_destroy(&dummy.lock);
235 [ - + ]: 4 : pthread_mutexattr_destroy(&attr);
236 : 4 : }
237 : :
238 : : static void
239 : 4 : test_spdk_nvme_connect(void)
240 : : {
241 : 4 : struct spdk_nvme_ctrlr *ret_ctrlr = NULL;
242 : 4 : struct spdk_nvme_transport_id trid = {};
243 : 4 : struct spdk_nvme_ctrlr_opts opts = {};
244 : 3 : struct spdk_nvme_ctrlr ctrlr;
245 : 3 : pthread_mutexattr_t attr;
246 : 4 : struct nvme_driver dummy = {};
247 : :
248 : : /* initialize the variable to prepare the test */
249 : 4 : dummy.initialized = true;
250 : 4 : TAILQ_INIT(&dummy.shared_attached_ctrlrs);
251 : 4 : g_spdk_nvme_driver = &dummy;
252 [ - + ]: 4 : CU_ASSERT(pthread_mutexattr_init(&attr) == 0);
253 [ - + ]: 4 : CU_ASSERT(pthread_mutex_init(&g_spdk_nvme_driver->lock, &attr) == 0);
254 : :
255 : : /* set NULL trid pointer to test immediate return */
256 : 4 : ret_ctrlr = spdk_nvme_connect(NULL, NULL, 0);
257 : 4 : CU_ASSERT(ret_ctrlr == NULL);
258 : :
259 : : /* driver init passes, transport available, secondary process connects ctrlr */
260 : 4 : MOCK_SET(spdk_process_is_primary, false);
261 : 4 : MOCK_SET(spdk_memzone_lookup, g_spdk_nvme_driver);
262 : 4 : MOCK_SET(spdk_nvme_transport_available_by_name, true);
263 [ - + ]: 4 : memset(&trid, 0, sizeof(trid));
264 : 4 : trid.trtype = SPDK_NVME_TRANSPORT_PCIE;
265 : 4 : ret_ctrlr = spdk_nvme_connect(&trid, NULL, 0);
266 : 4 : CU_ASSERT(ret_ctrlr == NULL);
267 : :
268 : : /* driver init passes, setup one ctrlr on the attached_list */
269 [ - + ]: 4 : memset(&ctrlr, 0, sizeof(struct spdk_nvme_ctrlr));
270 : 4 : spdk_nvme_ctrlr_get_default_ctrlr_opts(&ctrlr.opts, sizeof(ctrlr.opts));
271 [ - + ]: 4 : snprintf(ctrlr.trid.traddr, sizeof(ctrlr.trid.traddr), "0000:01:00.0");
272 : 4 : ctrlr.trid.trtype = SPDK_NVME_TRANSPORT_PCIE;
273 : 4 : TAILQ_INSERT_TAIL(&g_spdk_nvme_driver->shared_attached_ctrlrs, &ctrlr, tailq);
274 : : /* get the ctrlr from the attached list */
275 [ - + ]: 4 : snprintf(trid.traddr, sizeof(trid.traddr), "0000:01:00.0");
276 : 4 : ret_ctrlr = spdk_nvme_connect(&trid, NULL, 0);
277 : 4 : CU_ASSERT(ret_ctrlr == &ctrlr);
278 : : /* get the ctrlr from the attached list with default ctrlr opts */
279 : 4 : ctrlr.opts.num_io_queues = DEFAULT_MAX_IO_QUEUES;
280 : 4 : ret_ctrlr = spdk_nvme_connect(&trid, NULL, 0);
281 : 4 : CU_ASSERT(ret_ctrlr == &ctrlr);
282 : 4 : CU_ASSERT_EQUAL(ret_ctrlr->opts.num_io_queues, DEFAULT_MAX_IO_QUEUES);
283 : : /* get the ctrlr from the attached list with default ctrlr opts and consistent opts_size */
284 : 4 : spdk_nvme_ctrlr_get_default_ctrlr_opts(&opts, sizeof(opts));
285 : 4 : opts.num_io_queues = 1;
286 : 4 : ret_ctrlr = spdk_nvme_connect(&trid, &opts, sizeof(opts));
287 : 4 : CU_ASSERT(ret_ctrlr == &ctrlr);
288 : 4 : CU_ASSERT_EQUAL(ret_ctrlr->opts.num_io_queues, 1);
289 : 4 : CU_ASSERT_EQUAL(ret_ctrlr->opts.opts_size, sizeof(opts));
290 : :
291 : : /* opts_size is 0 */
292 : 4 : ret_ctrlr = spdk_nvme_connect(&trid, &opts, 0);
293 : 4 : CU_ASSERT(ret_ctrlr == &ctrlr);
294 : 4 : CU_ASSERT_EQUAL(ret_ctrlr->opts.opts_size, 0);
295 : :
296 : : /* opts_size is less than sizeof(*opts) if opts != NULL */
297 : 4 : ret_ctrlr = spdk_nvme_connect(&trid, &opts, 4);
298 : 4 : CU_ASSERT(ret_ctrlr == &ctrlr);
299 : 4 : CU_ASSERT_EQUAL(ret_ctrlr->opts.num_io_queues, 1);
300 : 4 : CU_ASSERT_EQUAL(ret_ctrlr->opts.opts_size, 4);
301 : : /* remove the attached ctrlr on the attached_list */
302 : 4 : MOCK_SET(nvme_ctrlr_get_ref_count, 1);
303 : 4 : CU_ASSERT(spdk_nvme_detach(&ctrlr) == 0);
304 : 4 : CU_ASSERT(TAILQ_EMPTY(&g_spdk_nvme_driver->shared_attached_ctrlrs));
305 : :
306 : : /* driver init passes, transport available, primary process connects ctrlr */
307 : 4 : MOCK_SET(spdk_process_is_primary, true);
308 : : /* setup one ctrlr on the attached_list */
309 [ - + ]: 4 : memset(&ctrlr, 0, sizeof(struct spdk_nvme_ctrlr));
310 [ - + ]: 4 : snprintf(ctrlr.trid.traddr, sizeof(ctrlr.trid.traddr), "0000:02:00.0");
311 : 4 : ctrlr.trid.trtype = SPDK_NVME_TRANSPORT_PCIE;
312 : 4 : spdk_nvme_ctrlr_get_default_ctrlr_opts(&ctrlr.opts, sizeof(ctrlr.opts));
313 : 4 : TAILQ_INSERT_TAIL(&g_spdk_nvme_driver->shared_attached_ctrlrs, &ctrlr, tailq);
314 : : /* get the ctrlr from the attached list */
315 [ - + ]: 4 : snprintf(trid.traddr, sizeof(trid.traddr), "0000:02:00.0");
316 : 4 : ret_ctrlr = spdk_nvme_connect(&trid, NULL, 0);
317 : 4 : CU_ASSERT(ret_ctrlr == &ctrlr);
318 : : /* get the ctrlr from the attached list with default ctrlr opts */
319 : 4 : ctrlr.opts.num_io_queues = DEFAULT_MAX_IO_QUEUES;
320 : 4 : ret_ctrlr = spdk_nvme_connect(&trid, NULL, 0);
321 : 4 : CU_ASSERT(ret_ctrlr == &ctrlr);
322 : 4 : CU_ASSERT_EQUAL(ret_ctrlr->opts.num_io_queues, DEFAULT_MAX_IO_QUEUES);
323 : : /* get the ctrlr from the attached list with default ctrlr opts and consistent opts_size */
324 : 4 : opts.num_io_queues = 2;
325 : 4 : ret_ctrlr = spdk_nvme_connect(&trid, &opts, sizeof(opts));
326 : 4 : CU_ASSERT(ret_ctrlr == &ctrlr);
327 : 4 : CU_ASSERT_EQUAL(ret_ctrlr->opts.num_io_queues, 2);
328 : : /* remove the attached ctrlr on the attached_list */
329 : 4 : CU_ASSERT(spdk_nvme_detach(ret_ctrlr) == 0);
330 : 4 : CU_ASSERT(TAILQ_EMPTY(&g_spdk_nvme_driver->shared_attached_ctrlrs));
331 : :
332 : : /* test driver init failure return */
333 : 4 : MOCK_SET(spdk_process_is_primary, false);
334 : 4 : MOCK_SET(spdk_memzone_lookup, NULL);
335 : 4 : ret_ctrlr = spdk_nvme_connect(&trid, NULL, 0);
336 : 4 : CU_ASSERT(ret_ctrlr == NULL);
337 : 4 : }
338 : :
339 : : static struct spdk_nvme_probe_ctx *
340 : 12 : test_nvme_init_get_probe_ctx(void)
341 : : {
342 : : struct spdk_nvme_probe_ctx *probe_ctx;
343 : :
344 : 12 : probe_ctx = calloc(1, sizeof(*probe_ctx));
345 [ + + ]: 12 : SPDK_CU_ASSERT_FATAL(probe_ctx != NULL);
346 : 12 : TAILQ_INIT(&probe_ctx->init_ctrlrs);
347 : :
348 : 12 : return probe_ctx;
349 : : }
350 : :
351 : : static void
352 : 4 : test_nvme_init_controllers(void)
353 : : {
354 : 4 : int rc = 0;
355 : 4 : struct nvme_driver test_driver = {};
356 : 4 : void *cb_ctx = NULL;
357 : 4 : spdk_nvme_attach_cb attach_cb = dummy_attach_cb;
358 : 4 : spdk_nvme_attach_fail_cb attach_fail_cb = dummy_attach_fail_cb;
359 : : struct spdk_nvme_probe_ctx *probe_ctx;
360 : : struct spdk_nvme_ctrlr *ctrlr;
361 : 3 : pthread_mutexattr_t attr;
362 : :
363 : 4 : g_spdk_nvme_driver = &test_driver;
364 : 4 : ctrlr = calloc(1, sizeof(*ctrlr));
365 [ + + ]: 4 : SPDK_CU_ASSERT_FATAL(ctrlr != NULL);
366 : 4 : ctrlr->trid.trtype = SPDK_NVME_TRANSPORT_PCIE;
367 [ - + ]: 4 : CU_ASSERT(pthread_mutexattr_init(&attr) == 0);
368 [ - + ]: 4 : CU_ASSERT(pthread_mutex_init(&ctrlr->ctrlr_lock, &attr) == 0);
369 [ - + ]: 4 : CU_ASSERT(pthread_mutex_init(&test_driver.lock, &attr) == 0);
370 : 4 : TAILQ_INIT(&test_driver.shared_attached_ctrlrs);
371 : :
372 : : /*
373 : : * Try to initialize, but nvme_ctrlr_process_init will fail.
374 : : * Verify correct behavior when it does.
375 : : */
376 : 4 : MOCK_SET(nvme_ctrlr_process_init, 1);
377 : 4 : MOCK_SET(spdk_process_is_primary, 1);
378 : 4 : g_spdk_nvme_driver->initialized = false;
379 : 4 : ut_destruct_called = false;
380 : 4 : ut_attach_fail_cb_rc = 0;
381 : 4 : probe_ctx = test_nvme_init_get_probe_ctx();
382 : 4 : TAILQ_INSERT_TAIL(&probe_ctx->init_ctrlrs, ctrlr, tailq);
383 : 4 : probe_ctx->cb_ctx = cb_ctx;
384 : 4 : probe_ctx->attach_cb = attach_cb;
385 : 4 : probe_ctx->attach_fail_cb = attach_fail_cb;
386 : 4 : probe_ctx->trid.trtype = SPDK_NVME_TRANSPORT_PCIE;
387 : 4 : rc = nvme_init_controllers(probe_ctx);
388 : 4 : CU_ASSERT(rc == 0);
389 [ - + ]: 4 : CU_ASSERT(g_spdk_nvme_driver->initialized == true);
390 [ - + ]: 4 : CU_ASSERT(ut_destruct_called == true);
391 : 4 : CU_ASSERT(ut_attach_fail_cb_rc == 1);
392 : :
393 : : /*
394 : : * Controller init OK, need to move the controller state machine
395 : : * forward by setting the ctrl state so that it can be moved
396 : : * the shared_attached_ctrlrs list.
397 : : */
398 : 4 : probe_ctx = test_nvme_init_get_probe_ctx();
399 : 4 : TAILQ_INSERT_TAIL(&probe_ctx->init_ctrlrs, ctrlr, tailq);
400 : 4 : ctrlr->state = NVME_CTRLR_STATE_READY;
401 : 4 : MOCK_SET(nvme_ctrlr_process_init, 0);
402 : 4 : rc = nvme_init_controllers(probe_ctx);
403 : 4 : CU_ASSERT(rc == 0);
404 [ - + ]: 4 : CU_ASSERT(ut_attach_cb_called == true);
405 : 4 : CU_ASSERT(TAILQ_EMPTY(&g_nvme_attached_ctrlrs));
406 : 4 : CU_ASSERT(TAILQ_FIRST(&g_spdk_nvme_driver->shared_attached_ctrlrs) == ctrlr);
407 [ - + ]: 4 : TAILQ_REMOVE(&g_spdk_nvme_driver->shared_attached_ctrlrs, ctrlr, tailq);
408 : :
409 : : /*
410 : : * Reset to initial state
411 : : */
412 [ - + ]: 4 : CU_ASSERT(pthread_mutex_destroy(&ctrlr->ctrlr_lock) == 0);
413 [ - + ]: 4 : memset(ctrlr, 0, sizeof(struct spdk_nvme_ctrlr));
414 [ - + ]: 4 : CU_ASSERT(pthread_mutex_init(&ctrlr->ctrlr_lock, &attr) == 0);
415 : :
416 : : /*
417 : : * Non-PCIe controllers should be added to the per-process list, not the shared list.
418 : : */
419 : 4 : ctrlr->trid.trtype = SPDK_NVME_TRANSPORT_RDMA;
420 : 4 : probe_ctx = test_nvme_init_get_probe_ctx();
421 : 4 : TAILQ_INSERT_TAIL(&probe_ctx->init_ctrlrs, ctrlr, tailq);
422 : 4 : ctrlr->state = NVME_CTRLR_STATE_READY;
423 : 4 : MOCK_SET(nvme_ctrlr_process_init, 0);
424 : 4 : rc = nvme_init_controllers(probe_ctx);
425 : 4 : CU_ASSERT(rc == 0);
426 [ - + ]: 4 : CU_ASSERT(ut_attach_cb_called == true);
427 : 4 : CU_ASSERT(TAILQ_EMPTY(&g_spdk_nvme_driver->shared_attached_ctrlrs));
428 : 4 : CU_ASSERT(TAILQ_FIRST(&g_nvme_attached_ctrlrs) == ctrlr);
429 [ - + ]: 4 : TAILQ_REMOVE(&g_nvme_attached_ctrlrs, ctrlr, tailq);
430 [ - + ]: 4 : CU_ASSERT(pthread_mutex_destroy(&ctrlr->ctrlr_lock) == 0);
431 : 4 : free(ctrlr);
432 : 4 : CU_ASSERT(TAILQ_EMPTY(&g_nvme_attached_ctrlrs));
433 : :
434 : 4 : g_spdk_nvme_driver = NULL;
435 [ - + ]: 4 : pthread_mutexattr_destroy(&attr);
436 [ - + ]: 4 : pthread_mutex_destroy(&test_driver.lock);
437 : 4 : }
438 : :
439 : : static void
440 : 4 : test_nvme_driver_init(void)
441 : : {
442 : : int rc;
443 : 4 : struct nvme_driver dummy = {};
444 : 4 : g_spdk_nvme_driver = &dummy;
445 : :
446 : : /* adjust this so testing doesn't take so long */
447 : 4 : g_nvme_driver_timeout_ms = 100;
448 : :
449 : : /* process is primary and mem already reserved */
450 : 4 : MOCK_SET(spdk_process_is_primary, true);
451 : 4 : dummy.initialized = true;
452 : 4 : rc = nvme_driver_init();
453 : 4 : CU_ASSERT(rc == 0);
454 : :
455 : : /*
456 : : * Process is primary and mem not yet reserved but the call
457 : : * to spdk_memzone_reserve() returns NULL.
458 : : */
459 : 4 : g_spdk_nvme_driver = NULL;
460 : 4 : MOCK_SET(spdk_process_is_primary, true);
461 : 4 : MOCK_SET(spdk_memzone_reserve, NULL);
462 : 4 : rc = nvme_driver_init();
463 : 4 : CU_ASSERT(rc == -1);
464 : :
465 : : /* process is not primary, no mem already reserved */
466 : 4 : MOCK_SET(spdk_process_is_primary, false);
467 : 4 : MOCK_SET(spdk_memzone_lookup, NULL);
468 : 4 : g_spdk_nvme_driver = NULL;
469 : 4 : rc = nvme_driver_init();
470 : 4 : CU_ASSERT(rc == -1);
471 : :
472 : : /* process is not primary, mem is already reserved & init'd */
473 : 4 : MOCK_SET(spdk_process_is_primary, false);
474 : 4 : MOCK_SET(spdk_memzone_lookup, (void *)&dummy);
475 : 4 : dummy.initialized = true;
476 : 4 : rc = nvme_driver_init();
477 : 4 : CU_ASSERT(rc == 0);
478 : :
479 : : /* process is not primary, mem is reserved but not initialized */
480 : : /* and times out */
481 : 4 : MOCK_SET(spdk_process_is_primary, false);
482 : 4 : MOCK_SET(spdk_memzone_reserve, (void *)&dummy);
483 : 4 : dummy.initialized = false;
484 : 4 : rc = nvme_driver_init();
485 : 4 : CU_ASSERT(rc == -1);
486 : :
487 : : /* process is primary, got mem but mutex won't init */
488 : 4 : MOCK_SET(spdk_process_is_primary, true);
489 : 4 : MOCK_SET(spdk_memzone_reserve, (void *)&dummy);
490 : 4 : MOCK_SET(pthread_mutexattr_init, -1);
491 : 4 : g_spdk_nvme_driver = NULL;
492 : 4 : dummy.initialized = true;
493 : 4 : rc = nvme_driver_init();
494 : : /* for FreeBSD we can't can't effectively mock this path */
495 : : #ifndef __FreeBSD__
496 : 3 : CU_ASSERT(rc != 0);
497 : : #else
498 : 1 : CU_ASSERT(rc == 0);
499 : : #endif
500 : :
501 : : /* process is primary, got mem, mutex OK */
502 : 4 : MOCK_SET(spdk_process_is_primary, true);
503 [ - + - + ]: 4 : MOCK_CLEAR(pthread_mutexattr_init);
504 : 4 : g_spdk_nvme_driver = NULL;
505 : 4 : rc = nvme_driver_init();
506 [ - + ]: 4 : CU_ASSERT(g_spdk_nvme_driver->initialized == false);
507 : 4 : CU_ASSERT(TAILQ_EMPTY(&g_spdk_nvme_driver->shared_attached_ctrlrs));
508 : 4 : CU_ASSERT(rc == 0);
509 : :
510 : 4 : g_spdk_nvme_driver = NULL;
511 [ - + - + ]: 4 : MOCK_CLEAR(spdk_memzone_reserve);
512 [ - + - + ]: 4 : MOCK_CLEAR(spdk_memzone_lookup);
513 : 4 : }
514 : :
515 : : static void
516 : 4 : test_spdk_nvme_detach(void)
517 : : {
518 : 4 : int rc = 1;
519 : 3 : struct spdk_nvme_ctrlr ctrlr;
520 : : struct spdk_nvme_ctrlr *ret_ctrlr;
521 : 4 : struct nvme_driver test_driver = {};
522 : :
523 [ - + ]: 4 : memset(&ctrlr, 0, sizeof(ctrlr));
524 : 4 : ctrlr.trid.trtype = SPDK_NVME_TRANSPORT_PCIE;
525 : :
526 : 4 : g_spdk_nvme_driver = &test_driver;
527 : 4 : TAILQ_INIT(&test_driver.shared_attached_ctrlrs);
528 : 4 : TAILQ_INSERT_TAIL(&test_driver.shared_attached_ctrlrs, &ctrlr, tailq);
529 [ - + ]: 4 : CU_ASSERT(pthread_mutex_init(&test_driver.lock, NULL) == 0);
530 : :
531 : : /*
532 : : * Controllers are ref counted so mock the function that returns
533 : : * the ref count so that detach will actually call the destruct
534 : : * function which we've mocked simply to verify that it gets
535 : : * called (we aren't testing what the real destruct function does
536 : : * here.)
537 : : */
538 : 4 : MOCK_SET(nvme_ctrlr_get_ref_count, 1);
539 : 4 : rc = spdk_nvme_detach(&ctrlr);
540 : 4 : ret_ctrlr = TAILQ_FIRST(&test_driver.shared_attached_ctrlrs);
541 : 4 : CU_ASSERT(ret_ctrlr == NULL);
542 [ - + ]: 4 : CU_ASSERT(ut_destruct_called == true);
543 : 4 : CU_ASSERT(rc == 0);
544 : :
545 : : /*
546 : : * Mock the ref count to 1 so we confirm that the destruct
547 : : * function is not called and that attached ctrl list is
548 : : * not empty.
549 : : */
550 : 4 : MOCK_SET(nvme_ctrlr_get_ref_count, 2);
551 : 4 : TAILQ_INSERT_TAIL(&test_driver.shared_attached_ctrlrs, &ctrlr, tailq);
552 : 4 : ut_destruct_called = false;
553 : 4 : rc = spdk_nvme_detach(&ctrlr);
554 : 4 : ret_ctrlr = TAILQ_FIRST(&test_driver.shared_attached_ctrlrs);
555 : 4 : CU_ASSERT(ret_ctrlr != NULL);
556 [ - + ]: 4 : CU_ASSERT(ut_destruct_called == false);
557 : 4 : CU_ASSERT(rc == 0);
558 : :
559 : : /*
560 : : * Non-PCIe controllers should be on the per-process attached_ctrlrs list, not the
561 : : * shared_attached_ctrlrs list. Test an RDMA controller and ensure it is removed
562 : : * from the correct list.
563 : : */
564 [ - + ]: 4 : memset(&ctrlr, 0, sizeof(ctrlr));
565 : 4 : ctrlr.trid.trtype = SPDK_NVME_TRANSPORT_RDMA;
566 : 4 : TAILQ_INIT(&g_nvme_attached_ctrlrs);
567 : 4 : TAILQ_INSERT_TAIL(&g_nvme_attached_ctrlrs, &ctrlr, tailq);
568 : 4 : MOCK_SET(nvme_ctrlr_get_ref_count, 1);
569 : 4 : rc = spdk_nvme_detach(&ctrlr);
570 : 4 : CU_ASSERT(TAILQ_EMPTY(&g_nvme_attached_ctrlrs));
571 [ - + ]: 4 : CU_ASSERT(ut_destruct_called == true);
572 : 4 : CU_ASSERT(rc == 0);
573 : :
574 : 4 : g_spdk_nvme_driver = NULL;
575 [ - + ]: 4 : pthread_mutex_destroy(&test_driver.lock);
576 : 4 : }
577 : :
578 : : static void
579 : 4 : test_nvme_completion_poll_cb(void)
580 : : {
581 : : struct nvme_completion_poll_status *status;
582 : 3 : struct spdk_nvme_cpl cpl;
583 : :
584 : 4 : status = calloc(1, sizeof(*status));
585 [ + + ]: 4 : SPDK_CU_ASSERT_FATAL(status != NULL);
586 : :
587 : 4 : memset(&cpl, 0xff, sizeof(cpl));
588 : :
589 : 4 : nvme_completion_poll_cb(status, &cpl);
590 [ - + ]: 4 : CU_ASSERT(status->done == true);
591 [ - + ]: 4 : CU_ASSERT(memcmp(&cpl, &status->cpl,
592 : : sizeof(struct spdk_nvme_cpl)) == 0);
593 : :
594 : 4 : free(status);
595 : 4 : }
596 : :
597 : : /* stub callback used by test_nvme_user_copy_cmd_complete() */
598 : : static struct spdk_nvme_cpl ut_spdk_nvme_cpl = {0};
599 : : static void
600 : 12 : dummy_cb(void *user_cb_arg, const struct spdk_nvme_cpl *cpl)
601 : : {
602 : 12 : ut_spdk_nvme_cpl = *cpl;
603 : 12 : }
604 : :
605 : : static void
606 : 4 : test_nvme_user_copy_cmd_complete(void)
607 : : {
608 : 3 : struct nvme_request req;
609 : 4 : int test_data = 0xdeadbeef;
610 : 4 : int buff_size = sizeof(int);
611 : : void *user_buffer, *buff;
612 : 4 : int user_cb_arg = 0x123;
613 : : static struct spdk_nvme_cpl cpl;
614 : :
615 : 4 : memset(&req, 0, sizeof(req));
616 : 4 : memset(&cpl, 0x5a, sizeof(cpl));
617 : :
618 : : /* test without a user buffer provided */
619 : 4 : req.user_cb_fn = (void *)dummy_cb;
620 : 4 : req.user_cb_arg = (void *)&user_cb_arg;
621 : 4 : nvme_user_copy_cmd_complete(&req, &cpl);
622 : 4 : CU_ASSERT(memcmp(&ut_spdk_nvme_cpl, &cpl, sizeof(cpl)) == 0);
623 : 4 : CU_ASSERT(req.user_cb_fn == NULL);
624 : 4 : CU_ASSERT(req.user_cb_arg == NULL);
625 : 4 : CU_ASSERT(req.user_buffer == NULL);
626 : :
627 : : /* test with a user buffer provided */
628 : 4 : user_buffer = malloc(buff_size);
629 : 4 : req.user_cb_fn = (void *)dummy_cb;
630 : 4 : req.user_cb_arg = (void *)&user_cb_arg;
631 : 4 : req.user_buffer = user_buffer;
632 [ + + ]: 4 : SPDK_CU_ASSERT_FATAL(req.user_buffer != NULL);
633 [ - + ]: 4 : memset(req.user_buffer, 0, buff_size);
634 : 4 : req.payload_size = buff_size;
635 : 4 : buff = spdk_zmalloc(buff_size, 0x100, NULL, SPDK_ENV_LCORE_ID_ANY, SPDK_MALLOC_DMA);
636 [ + + ]: 4 : SPDK_CU_ASSERT_FATAL(buff != NULL);
637 : 4 : req.payload = NVME_PAYLOAD_CONTIG(buff, NULL);
638 [ - + ]: 4 : memcpy(buff, &test_data, buff_size);
639 : 4 : req.cmd.opc = SPDK_NVME_OPC_GET_LOG_PAGE;
640 : 4 : req.pid = getpid();
641 : :
642 : : /* zero out the test value set in the callback */
643 : 4 : memset(&ut_spdk_nvme_cpl, 0, sizeof(ut_spdk_nvme_cpl));
644 : :
645 : 4 : nvme_user_copy_cmd_complete(&req, &cpl);
646 [ - + ]: 4 : CU_ASSERT(memcmp(user_buffer, &test_data, buff_size) == 0);
647 : 4 : CU_ASSERT(memcmp(&ut_spdk_nvme_cpl, &cpl, sizeof(cpl)) == 0);
648 : 4 : CU_ASSERT(req.user_cb_fn == NULL);
649 : 4 : CU_ASSERT(req.user_cb_arg == NULL);
650 : 4 : CU_ASSERT(req.user_buffer == NULL);
651 : :
652 : : /*
653 : : * Now test the same path as above but this time choose an opc
654 : : * that results in a different data transfer type.
655 : : */
656 : 4 : memset(&ut_spdk_nvme_cpl, 0, sizeof(ut_spdk_nvme_cpl));
657 : 4 : req.user_cb_fn = (void *)dummy_cb;
658 : 4 : req.user_cb_arg = (void *)&user_cb_arg;
659 : 4 : req.user_buffer = user_buffer;
660 [ - + ]: 4 : memset(req.user_buffer, 0, buff_size);
661 : 4 : buff = spdk_zmalloc(buff_size, 0x100, NULL, SPDK_ENV_LCORE_ID_ANY, SPDK_MALLOC_DMA);
662 [ - + ]: 4 : SPDK_CU_ASSERT_FATAL(buff != NULL);
663 : 4 : req.payload = NVME_PAYLOAD_CONTIG(buff, NULL);
664 [ - + ]: 4 : memcpy(buff, &test_data, buff_size);
665 : 4 : req.cmd.opc = SPDK_NVME_OPC_SET_FEATURES;
666 : 4 : nvme_user_copy_cmd_complete(&req, &cpl);
667 [ - + ]: 4 : CU_ASSERT(memcmp(user_buffer, &test_data, buff_size) != 0);
668 : 4 : CU_ASSERT(memcmp(&ut_spdk_nvme_cpl, &cpl, sizeof(cpl)) == 0);
669 : 4 : CU_ASSERT(req.user_cb_fn == NULL);
670 : 4 : CU_ASSERT(req.user_cb_arg == NULL);
671 : 4 : CU_ASSERT(req.user_buffer == NULL);
672 : :
673 : : /* clean up */
674 : 4 : free(user_buffer);
675 : 4 : }
676 : :
677 : : static void
678 : 4 : test_nvme_allocate_request_null(void)
679 : : {
680 : 4 : struct spdk_nvme_qpair qpair = {};
681 : 4 : spdk_nvme_cmd_cb cb_fn = (spdk_nvme_cmd_cb)0x1234;
682 : 4 : void *cb_arg = (void *)0x5678;
683 : 4 : struct nvme_request *req = NULL;
684 : 3 : struct nvme_request dummy_req;
685 : :
686 : 4 : STAILQ_INIT(&qpair.free_req);
687 : 4 : STAILQ_INIT(&qpair.queued_req);
688 : :
689 : : /*
690 : : * Put a dummy on the queue so we can make a request
691 : : * and confirm that what comes back is what we expect.
692 : : */
693 [ + + ]: 4 : STAILQ_INSERT_HEAD(&qpair.free_req, &dummy_req, stailq);
694 : :
695 : 4 : req = nvme_allocate_request_null(&qpair, cb_fn, cb_arg);
696 : :
697 : : /*
698 : : * Compare the req with the parameters that we passed in
699 : : * as well as what the function is supposed to update.
700 : : */
701 [ + + ]: 4 : SPDK_CU_ASSERT_FATAL(req != NULL);
702 : 4 : CU_ASSERT(req->cb_fn == cb_fn);
703 : 4 : CU_ASSERT(req->cb_arg == cb_arg);
704 : 4 : CU_ASSERT(req->pid == getpid());
705 : 4 : CU_ASSERT(nvme_payload_type(&req->payload) == NVME_PAYLOAD_TYPE_CONTIG);
706 : 4 : CU_ASSERT(req->payload.md == NULL);
707 : 4 : CU_ASSERT(req->payload.contig_or_cb_arg == NULL);
708 : 4 : }
709 : :
710 : : static void
711 : 4 : test_nvme_allocate_request(void)
712 : : {
713 : 3 : struct spdk_nvme_qpair qpair;
714 : 3 : struct nvme_payload payload;
715 : 4 : uint32_t payload_struct_size = sizeof(payload);
716 : 4 : spdk_nvme_cmd_cb cb_fn = (spdk_nvme_cmd_cb)0x1234;
717 : 4 : void *cb_arg = (void *)0x6789;
718 : 4 : struct nvme_request *req = NULL;
719 : 3 : struct nvme_request dummy_req;
720 : :
721 : : /* Fill the whole payload struct with a known pattern */
722 : 4 : memset(&payload, 0x5a, payload_struct_size);
723 : 4 : STAILQ_INIT(&qpair.free_req);
724 : 4 : STAILQ_INIT(&qpair.queued_req);
725 : 4 : qpair.num_outstanding_reqs = 0;
726 : :
727 : : /* Test trying to allocate a request when no requests are available */
728 : 5 : req = nvme_allocate_request(&qpair, &payload, payload_struct_size, 0,
729 : 1 : cb_fn, cb_arg);
730 : 4 : CU_ASSERT(req == NULL);
731 : 4 : CU_ASSERT(qpair.num_outstanding_reqs == 0);
732 : :
733 : : /* put a dummy on the queue, and then allocate one */
734 [ + + ]: 4 : STAILQ_INSERT_HEAD(&qpair.free_req, &dummy_req, stailq);
735 : 5 : req = nvme_allocate_request(&qpair, &payload, payload_struct_size, 0,
736 : 1 : cb_fn, cb_arg);
737 : :
738 : : /* all the req elements should now match the passed in parameters */
739 [ + + ]: 4 : SPDK_CU_ASSERT_FATAL(req != NULL);
740 : 4 : CU_ASSERT(qpair.num_outstanding_reqs == 1);
741 : 4 : CU_ASSERT(req->cb_fn == cb_fn);
742 : 4 : CU_ASSERT(req->cb_arg == cb_arg);
743 [ - + ]: 4 : CU_ASSERT(memcmp(&req->payload, &payload, payload_struct_size) == 0);
744 : 4 : CU_ASSERT(req->payload_size == payload_struct_size);
745 : 4 : CU_ASSERT(req->pid == getpid());
746 : 4 : }
747 : :
748 : : static void
749 : 4 : test_nvme_free_request(void)
750 : : {
751 : 3 : struct nvme_request match_req;
752 : 4 : struct spdk_nvme_qpair qpair = {0};
753 : : struct nvme_request *req;
754 : :
755 : : /* put a req on the Q, take it off and compare */
756 [ - + ]: 4 : memset(&match_req.cmd, 0x5a, sizeof(struct spdk_nvme_cmd));
757 : 4 : match_req.qpair = &qpair;
758 : 4 : qpair.num_outstanding_reqs = 1;
759 : : /* the code under tests asserts this condition */
760 : 4 : match_req.num_children = 0;
761 : 4 : STAILQ_INIT(&qpair.free_req);
762 : 4 : match_req.qpair->reserved_req = NULL;
763 : :
764 : 4 : nvme_free_request(&match_req);
765 : 4 : req = STAILQ_FIRST(&match_req.qpair->free_req);
766 : 4 : CU_ASSERT(req == &match_req);
767 : 4 : CU_ASSERT(qpair.num_outstanding_reqs == 0);
768 : 4 : }
769 : :
770 : : static void
771 : 4 : test_nvme_allocate_request_user_copy(void)
772 : : {
773 : 3 : struct spdk_nvme_qpair qpair;
774 : 4 : spdk_nvme_cmd_cb cb_fn = (spdk_nvme_cmd_cb)0x12345;
775 : 4 : void *cb_arg = (void *)0x12345;
776 : 4 : bool host_to_controller = true;
777 : : struct nvme_request *req;
778 : 3 : struct nvme_request dummy_req;
779 : 4 : int test_data = 0xdeadbeef;
780 : 4 : void *buffer = NULL;
781 : 4 : uint32_t payload_size = sizeof(int);
782 : :
783 : 4 : STAILQ_INIT(&qpair.free_req);
784 : 4 : STAILQ_INIT(&qpair.queued_req);
785 : :
786 : : /* no buffer or valid payload size, early NULL return */
787 : 5 : req = nvme_allocate_request_user_copy(&qpair, buffer, payload_size, cb_fn,
788 : 1 : cb_arg, host_to_controller);
789 : 4 : CU_ASSERT(req == NULL);
790 : :
791 : : /* good buffer and valid payload size */
792 : 4 : buffer = malloc(payload_size);
793 [ + + ]: 4 : SPDK_CU_ASSERT_FATAL(buffer != NULL);
794 [ - + ]: 4 : memcpy(buffer, &test_data, payload_size);
795 : :
796 : : /* put a dummy on the queue */
797 [ + + ]: 4 : STAILQ_INSERT_HEAD(&qpair.free_req, &dummy_req, stailq);
798 : :
799 [ - + - + ]: 4 : MOCK_CLEAR(spdk_malloc);
800 [ - + - + ]: 4 : MOCK_CLEAR(spdk_zmalloc);
801 : 5 : req = nvme_allocate_request_user_copy(&qpair, buffer, payload_size, cb_fn,
802 : 1 : cb_arg, host_to_controller);
803 [ + + ]: 4 : SPDK_CU_ASSERT_FATAL(req != NULL);
804 : 4 : CU_ASSERT(req->user_cb_fn == cb_fn);
805 : 4 : CU_ASSERT(req->user_cb_arg == cb_arg);
806 : 4 : CU_ASSERT(req->user_buffer == buffer);
807 : 4 : CU_ASSERT(req->cb_arg == req);
808 [ - + - + ]: 4 : CU_ASSERT(memcmp(req->payload.contig_or_cb_arg, buffer, payload_size) == 0);
809 : 4 : spdk_free(req->payload.contig_or_cb_arg);
810 : :
811 : : /* same thing but additional path coverage, no copy */
812 : 4 : host_to_controller = false;
813 [ + + ]: 4 : STAILQ_INSERT_HEAD(&qpair.free_req, &dummy_req, stailq);
814 : :
815 : 5 : req = nvme_allocate_request_user_copy(&qpair, buffer, payload_size, cb_fn,
816 : 1 : cb_arg, host_to_controller);
817 [ - + ]: 4 : SPDK_CU_ASSERT_FATAL(req != NULL);
818 : 4 : CU_ASSERT(req->user_cb_fn == cb_fn);
819 : 4 : CU_ASSERT(req->user_cb_arg == cb_arg);
820 : 4 : CU_ASSERT(req->user_buffer == buffer);
821 : 4 : CU_ASSERT(req->cb_arg == req);
822 [ - + - + ]: 4 : CU_ASSERT(memcmp(req->payload.contig_or_cb_arg, buffer, payload_size) != 0);
823 : 4 : spdk_free(req->payload.contig_or_cb_arg);
824 : :
825 : : /* good buffer and valid payload size but make spdk_zmalloc fail */
826 : : /* set the mock pointer to NULL for spdk_zmalloc */
827 : 4 : MOCK_SET(spdk_zmalloc, NULL);
828 : 5 : req = nvme_allocate_request_user_copy(&qpair, buffer, payload_size, cb_fn,
829 : 1 : cb_arg, host_to_controller);
830 : 4 : CU_ASSERT(req == NULL);
831 : 4 : free(buffer);
832 [ - + - + ]: 4 : MOCK_CLEAR(spdk_zmalloc);
833 : 4 : }
834 : :
835 : : static void
836 : 4 : test_nvme_ctrlr_probe(void)
837 : : {
838 : 4 : int rc = 0;
839 : 4 : struct spdk_nvme_ctrlr ctrlr = {};
840 : 4 : struct spdk_nvme_qpair qpair = {};
841 : 4 : const struct spdk_nvme_transport_id trid = {};
842 : 4 : struct spdk_nvme_probe_ctx probe_ctx = {};
843 : 4 : void *devhandle = NULL;
844 : 4 : void *cb_ctx = NULL;
845 : 4 : struct spdk_nvme_ctrlr *dummy = NULL;
846 : :
847 : 4 : nvme_driver_init();
848 : :
849 : 4 : ctrlr.adminq = &qpair;
850 : 4 : nvme_get_default_hostnqn(ctrlr.opts.hostnqn, sizeof(ctrlr.opts.hostnqn));
851 : :
852 : 4 : TAILQ_INIT(&probe_ctx.init_ctrlrs);
853 : :
854 : : /* test when probe_cb returns false */
855 : :
856 : 4 : MOCK_SET(dummy_probe_cb, false);
857 : 4 : nvme_probe_ctx_init(&probe_ctx, &trid, NULL, cb_ctx, dummy_probe_cb, NULL, NULL, NULL);
858 : 4 : rc = nvme_ctrlr_probe(&trid, &probe_ctx, devhandle);
859 : 4 : CU_ASSERT(rc == 1);
860 : :
861 : : /* probe_cb returns true but we find a destructing ctrlr */
862 : 4 : MOCK_SET(dummy_probe_cb, true);
863 : 4 : ut_attach_fail_cb_rc = 0;
864 : 4 : ctrlr.is_destructed = true;
865 : 4 : TAILQ_INSERT_TAIL(&g_nvme_attached_ctrlrs, &ctrlr, tailq);
866 : 4 : nvme_probe_ctx_init(&probe_ctx, &trid, NULL, cb_ctx, dummy_probe_cb, NULL,
867 : : dummy_attach_fail_cb, NULL);
868 : 4 : rc = nvme_ctrlr_probe(&trid, &probe_ctx, devhandle);
869 : 4 : CU_ASSERT(rc == -EBUSY);
870 : 4 : CU_ASSERT(ut_attach_fail_cb_rc == -EBUSY);
871 [ - + ]: 4 : TAILQ_REMOVE(&g_nvme_attached_ctrlrs, &ctrlr, tailq);
872 : 4 : ctrlr.is_destructed = false;
873 : :
874 : : /* probe_cb returns true but we can't construct a ctrl */
875 : 4 : MOCK_SET(dummy_probe_cb, true);
876 : 4 : MOCK_SET(nvme_transport_ctrlr_construct, NULL);
877 : 4 : ut_attach_fail_cb_rc = 0;
878 : 4 : nvme_probe_ctx_init(&probe_ctx, &trid, NULL, cb_ctx, dummy_probe_cb, NULL,
879 : : dummy_attach_fail_cb, NULL);
880 : 4 : rc = nvme_ctrlr_probe(&trid, &probe_ctx, devhandle);
881 : 4 : CU_ASSERT(rc == -1);
882 : 4 : CU_ASSERT(ut_attach_fail_cb_rc == -ENODEV);
883 : :
884 : : /* happy path */
885 : 4 : MOCK_SET(dummy_probe_cb, true);
886 : 4 : MOCK_SET(nvme_transport_ctrlr_construct, &ctrlr);
887 : 4 : nvme_probe_ctx_init(&probe_ctx, &trid, NULL, cb_ctx, dummy_probe_cb, NULL, NULL, NULL);
888 : 4 : rc = nvme_ctrlr_probe(&trid, &probe_ctx, devhandle);
889 : 4 : CU_ASSERT(rc == 0);
890 : 4 : dummy = TAILQ_FIRST(&probe_ctx.init_ctrlrs);
891 [ + + ]: 4 : SPDK_CU_ASSERT_FATAL(dummy != NULL);
892 : 4 : CU_ASSERT(dummy == ut_nvme_transport_ctrlr_construct);
893 [ - + ]: 4 : TAILQ_REMOVE(&probe_ctx.init_ctrlrs, dummy, tailq);
894 [ - + - + ]: 4 : MOCK_CLEAR_P(nvme_transport_ctrlr_construct);
895 : :
896 : 4 : free(g_spdk_nvme_driver);
897 : 4 : }
898 : :
899 : : static void
900 : 4 : test_nvme_robust_mutex_init_shared(void)
901 : : {
902 : 3 : pthread_mutex_t mtx;
903 : 4 : int rc = 0;
904 : :
905 : : /* test where both pthread calls succeed */
906 : 4 : MOCK_SET(pthread_mutexattr_init, 0);
907 : 4 : MOCK_SET(pthread_mutex_init, 0);
908 : 4 : rc = nvme_robust_mutex_init_shared(&mtx);
909 : 4 : CU_ASSERT(rc == 0);
910 : :
911 : : /* test where we can't init attr's but init mutex works */
912 : 4 : MOCK_SET(pthread_mutexattr_init, -1);
913 : 4 : MOCK_SET(pthread_mutex_init, 0);
914 : 4 : rc = nvme_robust_mutex_init_shared(&mtx);
915 : : /* for FreeBSD the only possible return value is 0 */
916 : : #ifndef __FreeBSD__
917 : 3 : CU_ASSERT(rc != 0);
918 : : #else
919 : 1 : CU_ASSERT(rc == 0);
920 : : #endif
921 : :
922 : : /* test where we can init attr's but the mutex init fails */
923 : 4 : MOCK_SET(pthread_mutexattr_init, 0);
924 : 4 : MOCK_SET(pthread_mutex_init, -1);
925 : 4 : rc = nvme_robust_mutex_init_shared(&mtx);
926 : : /* for FreeBSD the only possible return value is 0 */
927 : : #ifndef __FreeBSD__
928 : 3 : CU_ASSERT(rc != 0);
929 : : #else
930 : 1 : CU_ASSERT(rc == 0);
931 : : #endif
932 [ - + - + ]: 4 : MOCK_CLEAR(pthread_mutex_init);
933 : 4 : }
934 : :
935 : : static void
936 : 4 : test_opc_data_transfer(void)
937 : : {
938 : : enum spdk_nvme_data_transfer xfer;
939 : :
940 : 4 : xfer = spdk_nvme_opc_get_data_transfer(SPDK_NVME_OPC_FLUSH);
941 : 4 : CU_ASSERT(xfer == SPDK_NVME_DATA_NONE);
942 : :
943 : 4 : xfer = spdk_nvme_opc_get_data_transfer(SPDK_NVME_OPC_WRITE);
944 : 4 : CU_ASSERT(xfer == SPDK_NVME_DATA_HOST_TO_CONTROLLER);
945 : :
946 : 4 : xfer = spdk_nvme_opc_get_data_transfer(SPDK_NVME_OPC_READ);
947 : 4 : CU_ASSERT(xfer == SPDK_NVME_DATA_CONTROLLER_TO_HOST);
948 : :
949 : 4 : xfer = spdk_nvme_opc_get_data_transfer(SPDK_NVME_OPC_GET_LOG_PAGE);
950 : 4 : CU_ASSERT(xfer == SPDK_NVME_DATA_CONTROLLER_TO_HOST);
951 : 4 : }
952 : :
953 : : static void
954 : 4 : test_trid_parse_and_compare(void)
955 : : {
956 : 3 : struct spdk_nvme_transport_id trid1, trid2;
957 : : int ret;
958 : :
959 : : /* set trid1 trid2 value to id parse */
960 : 4 : ret = spdk_nvme_transport_id_parse(NULL, "trtype:PCIe traddr:0000:04:00.0");
961 : 4 : CU_ASSERT(ret == -EINVAL);
962 : 4 : memset(&trid1, 0, sizeof(trid1));
963 : 4 : ret = spdk_nvme_transport_id_parse(&trid1, NULL);
964 : 4 : CU_ASSERT(ret == -EINVAL);
965 : 4 : ret = spdk_nvme_transport_id_parse(NULL, NULL);
966 : 4 : CU_ASSERT(ret == -EINVAL);
967 : 4 : memset(&trid1, 0, sizeof(trid1));
968 : 4 : ret = spdk_nvme_transport_id_parse(&trid1, "trtype-PCIe traddr-0000-04-00.0");
969 : 4 : CU_ASSERT(ret == -EINVAL);
970 : 4 : memset(&trid1, 0, sizeof(trid1));
971 : 4 : ret = spdk_nvme_transport_id_parse(&trid1, "trtype-PCIe traddr-0000-04-00.0-:");
972 : 4 : CU_ASSERT(ret == -EINVAL);
973 : 4 : memset(&trid1, 0, sizeof(trid1));
974 : 4 : ret = spdk_nvme_transport_id_parse(&trid1, " \t\n:");
975 : 4 : CU_ASSERT(ret == -EINVAL);
976 : 4 : memset(&trid1, 0, sizeof(trid1));
977 : 4 : CU_ASSERT(spdk_nvme_transport_id_parse(&trid1,
978 : : "trtype:rdma\n"
979 : : "adrfam:ipv4\n"
980 : : "traddr:192.168.100.8\n"
981 : : "trsvcid:4420\n"
982 : : "subnqn:nqn.2014-08.org.nvmexpress.discovery") == 0);
983 : 4 : CU_ASSERT(trid1.trtype == SPDK_NVME_TRANSPORT_RDMA);
984 : 4 : CU_ASSERT(trid1.adrfam == SPDK_NVMF_ADRFAM_IPV4);
985 : 4 : CU_ASSERT(strcmp(trid1.traddr, "192.168.100.8") == 0);
986 : 4 : CU_ASSERT(strcmp(trid1.trsvcid, "4420") == 0);
987 : 4 : CU_ASSERT(strcmp(trid1.subnqn, "nqn.2014-08.org.nvmexpress.discovery") == 0);
988 : :
989 : 4 : memset(&trid2, 0, sizeof(trid2));
990 : 4 : CU_ASSERT(spdk_nvme_transport_id_parse(&trid2, "trtype:PCIe traddr:0000:04:00.0") == 0);
991 : 4 : CU_ASSERT(trid2.trtype == SPDK_NVME_TRANSPORT_PCIE);
992 : 4 : CU_ASSERT(strcmp(trid2.traddr, "0000:04:00.0") == 0);
993 : :
994 : 4 : CU_ASSERT(spdk_nvme_transport_id_compare(&trid1, &trid2) != 0);
995 : :
996 : : /* set trid1 trid2 and test id_compare */
997 : 4 : memset_trid(&trid1, &trid2);
998 : 4 : trid1.adrfam = SPDK_NVMF_ADRFAM_IPV6;
999 : 4 : trid2.adrfam = SPDK_NVMF_ADRFAM_IPV4;
1000 : 4 : ret = spdk_nvme_transport_id_compare(&trid1, &trid2);
1001 : 4 : CU_ASSERT(ret > 0);
1002 : :
1003 : 4 : memset_trid(&trid1, &trid2);
1004 : 4 : snprintf(trid1.traddr, sizeof(trid1.traddr), "192.168.100.8");
1005 : 4 : snprintf(trid2.traddr, sizeof(trid2.traddr), "192.168.100.9");
1006 : 4 : ret = spdk_nvme_transport_id_compare(&trid1, &trid2);
1007 : 4 : CU_ASSERT(ret < 0);
1008 : :
1009 : 4 : memset_trid(&trid1, &trid2);
1010 : 4 : snprintf(trid1.trsvcid, sizeof(trid1.trsvcid), "4420");
1011 : 4 : snprintf(trid2.trsvcid, sizeof(trid2.trsvcid), "4421");
1012 : 4 : ret = spdk_nvme_transport_id_compare(&trid1, &trid2);
1013 : 4 : CU_ASSERT(ret < 0);
1014 : :
1015 : 4 : memset_trid(&trid1, &trid2);
1016 : 4 : snprintf(trid1.subnqn, sizeof(trid1.subnqn), "subnqn:nqn.2016-08.org.nvmexpress.discovery");
1017 : 4 : snprintf(trid2.subnqn, sizeof(trid2.subnqn), "subnqn:nqn.2017-08.org.nvmexpress.discovery");
1018 : 4 : ret = spdk_nvme_transport_id_compare(&trid1, &trid2);
1019 : 4 : CU_ASSERT(ret < 0);
1020 : :
1021 : 4 : memset_trid(&trid1, &trid2);
1022 : 4 : snprintf(trid1.subnqn, sizeof(trid1.subnqn), "subnqn:nqn.2016-08.org.nvmexpress.discovery");
1023 : 4 : snprintf(trid2.subnqn, sizeof(trid2.subnqn), "subnqn:nqn.2016-08.org.nvmexpress.discovery");
1024 : 4 : ret = spdk_nvme_transport_id_compare(&trid1, &trid2);
1025 : 4 : CU_ASSERT(ret == 0);
1026 : :
1027 : 4 : memset_trid(&trid1, &trid2);
1028 : 4 : snprintf(trid1.subnqn, sizeof(trid1.subnqn), "subnqn:nqn.2016-08.org.nvmexpress.discovery");
1029 : 4 : snprintf(trid2.subnqn, sizeof(trid2.subnqn), "subnqn:nqn.2016-08.org.Nvmexpress.discovery");
1030 : 4 : ret = spdk_nvme_transport_id_compare(&trid1, &trid2);
1031 : 4 : CU_ASSERT(ret > 0);
1032 : :
1033 : 4 : memset_trid(&trid1, &trid2);
1034 : 4 : ret = spdk_nvme_transport_id_compare(&trid1, &trid2);
1035 : 4 : CU_ASSERT(ret == 0);
1036 : :
1037 : : /* Compare PCI addresses via spdk_pci_addr_compare (rather than as strings) */
1038 : 4 : memset_trid(&trid1, &trid2);
1039 : 4 : CU_ASSERT(spdk_nvme_transport_id_parse(&trid1, "trtype:PCIe traddr:0000:04:00.0") == 0);
1040 : 4 : CU_ASSERT(spdk_nvme_transport_id_parse(&trid2, "trtype:PCIe traddr:04:00.0") == 0);
1041 : 4 : CU_ASSERT(spdk_nvme_transport_id_compare(&trid1, &trid2) == 0);
1042 : :
1043 : 4 : memset_trid(&trid1, &trid2);
1044 : 4 : CU_ASSERT(spdk_nvme_transport_id_parse(&trid1, "trtype:PCIe traddr:0000:05:00.0") == 0);
1045 : 4 : CU_ASSERT(spdk_nvme_transport_id_parse(&trid2, "trtype:PCIe traddr:04:00.0") == 0);
1046 : 4 : CU_ASSERT(spdk_nvme_transport_id_compare(&trid1, &trid2) > 0);
1047 : :
1048 : 4 : memset_trid(&trid1, &trid2);
1049 : 4 : CU_ASSERT(spdk_nvme_transport_id_parse(&trid1, "trtype:PCIe traddr:0000:04:00.0") == 0);
1050 : 4 : CU_ASSERT(spdk_nvme_transport_id_parse(&trid2, "trtype:PCIe traddr:05:00.0") == 0);
1051 : 4 : CU_ASSERT(spdk_nvme_transport_id_compare(&trid1, &trid2) < 0);
1052 : :
1053 : 4 : memset_trid(&trid1, &trid2);
1054 : 4 : CU_ASSERT(spdk_nvme_transport_id_parse(&trid1, "trtype=PCIe traddr=0000:04:00.0") == 0);
1055 : 4 : CU_ASSERT(spdk_nvme_transport_id_parse(&trid2, "trtype=PCIe traddr=05:00.0") == 0);
1056 : 4 : CU_ASSERT(spdk_nvme_transport_id_compare(&trid1, &trid2) < 0);
1057 : :
1058 : 4 : CU_ASSERT(spdk_nvme_transport_id_parse(&trid1,
1059 : : "trtype:tcp\n"
1060 : : "adrfam:ipv4\n"
1061 : : "traddr:192.168.100.8\n"
1062 : : "trsvcid:4420\n"
1063 : : "priority:2\n"
1064 : : "subnqn:nqn.2014-08.org.nvmexpress.discovery") == 0);
1065 : 4 : CU_ASSERT(trid1.priority == 2);
1066 : 4 : }
1067 : :
1068 : : static void
1069 : 4 : test_spdk_nvme_transport_id_parse_trtype(void)
1070 : : {
1071 : :
1072 : : enum spdk_nvme_transport_type *trtype;
1073 : 3 : enum spdk_nvme_transport_type sct;
1074 : : char *str;
1075 : :
1076 : 4 : trtype = NULL;
1077 : 4 : str = "unit_test";
1078 : :
1079 : : /* test function returned value when trtype is NULL but str not NULL */
1080 : 4 : CU_ASSERT(spdk_nvme_transport_id_parse_trtype(trtype, str) == (-EINVAL));
1081 : :
1082 : : /* test function returned value when str is NULL but trtype not NULL */
1083 : 4 : trtype = &sct;
1084 : 4 : str = NULL;
1085 : 4 : CU_ASSERT(spdk_nvme_transport_id_parse_trtype(trtype, str) == (-EINVAL));
1086 : :
1087 : : /* test function returned value when str and strtype not NULL, but str value
1088 : : * not "PCIe" or "RDMA" */
1089 : 4 : str = "unit_test";
1090 : 4 : CU_ASSERT(spdk_nvme_transport_id_parse_trtype(trtype, str) == 0);
1091 : 4 : CU_ASSERT((*trtype) == SPDK_NVME_TRANSPORT_CUSTOM);
1092 : :
1093 : : /* test trtype value when use function "strcasecmp" to compare str and "PCIe",not case-sensitive */
1094 : 4 : str = "PCIe";
1095 : 4 : spdk_nvme_transport_id_parse_trtype(trtype, str);
1096 : 4 : CU_ASSERT((*trtype) == SPDK_NVME_TRANSPORT_PCIE);
1097 : :
1098 : 4 : str = "pciE";
1099 : 4 : spdk_nvme_transport_id_parse_trtype(trtype, str);
1100 : 4 : CU_ASSERT((*trtype) == SPDK_NVME_TRANSPORT_PCIE);
1101 : :
1102 : : /* test trtype value when use function "strcasecmp" to compare str and "RDMA",not case-sensitive */
1103 : 4 : str = "RDMA";
1104 : 4 : spdk_nvme_transport_id_parse_trtype(trtype, str);
1105 : 4 : CU_ASSERT((*trtype) == SPDK_NVME_TRANSPORT_RDMA);
1106 : :
1107 : 4 : str = "rdma";
1108 : 4 : spdk_nvme_transport_id_parse_trtype(trtype, str);
1109 : 4 : CU_ASSERT((*trtype) == SPDK_NVME_TRANSPORT_RDMA);
1110 : :
1111 : : /* test trtype value when use function "strcasecmp" to compare str and "FC",not case-sensitive */
1112 : 4 : str = "FC";
1113 : 4 : spdk_nvme_transport_id_parse_trtype(trtype, str);
1114 : 4 : CU_ASSERT((*trtype) == SPDK_NVME_TRANSPORT_FC);
1115 : :
1116 : 4 : str = "fc";
1117 : 4 : spdk_nvme_transport_id_parse_trtype(trtype, str);
1118 : 4 : CU_ASSERT((*trtype) == SPDK_NVME_TRANSPORT_FC);
1119 : :
1120 : : /* test trtype value when use function "strcasecmp" to compare str and "TCP",not case-sensitive */
1121 : 4 : str = "TCP";
1122 : 4 : spdk_nvme_transport_id_parse_trtype(trtype, str);
1123 : 4 : CU_ASSERT((*trtype) == SPDK_NVME_TRANSPORT_TCP);
1124 : :
1125 : 4 : str = "tcp";
1126 : 4 : spdk_nvme_transport_id_parse_trtype(trtype, str);
1127 : 4 : CU_ASSERT((*trtype) == SPDK_NVME_TRANSPORT_TCP);
1128 : 4 : }
1129 : :
1130 : : static void
1131 : 4 : test_spdk_nvme_transport_id_parse_adrfam(void)
1132 : : {
1133 : :
1134 : : enum spdk_nvmf_adrfam *adrfam;
1135 : 3 : enum spdk_nvmf_adrfam sct;
1136 : : char *str;
1137 : :
1138 : 4 : adrfam = NULL;
1139 : 4 : str = "unit_test";
1140 : :
1141 : : /* test function returned value when adrfam is NULL but str not NULL */
1142 : 4 : CU_ASSERT(spdk_nvme_transport_id_parse_adrfam(adrfam, str) == (-EINVAL));
1143 : :
1144 : : /* test function returned value when str is NULL but adrfam not NULL */
1145 : 4 : adrfam = &sct;
1146 : 4 : str = NULL;
1147 : 4 : CU_ASSERT(spdk_nvme_transport_id_parse_adrfam(adrfam, str) == (-EINVAL));
1148 : :
1149 : : /* test function returned value when str and adrfam not NULL, but str value
1150 : : * not "IPv4" or "IPv6" or "IB" or "FC" */
1151 : 4 : str = "unit_test";
1152 : 4 : CU_ASSERT(spdk_nvme_transport_id_parse_adrfam(adrfam, str) == (-ENOENT));
1153 : :
1154 : : /* test adrfam value when use function "strcasecmp" to compare str and "IPv4",not case-sensitive */
1155 : 4 : str = "IPv4";
1156 : 4 : spdk_nvme_transport_id_parse_adrfam(adrfam, str);
1157 : 4 : CU_ASSERT((*adrfam) == SPDK_NVMF_ADRFAM_IPV4);
1158 : :
1159 : 4 : str = "ipV4";
1160 : 4 : spdk_nvme_transport_id_parse_adrfam(adrfam, str);
1161 : 4 : CU_ASSERT((*adrfam) == SPDK_NVMF_ADRFAM_IPV4);
1162 : :
1163 : : /* test adrfam value when use function "strcasecmp" to compare str and "IPv6",not case-sensitive */
1164 : 4 : str = "IPv6";
1165 : 4 : spdk_nvme_transport_id_parse_adrfam(adrfam, str);
1166 : 4 : CU_ASSERT((*adrfam) == SPDK_NVMF_ADRFAM_IPV6);
1167 : :
1168 : 4 : str = "ipV6";
1169 : 4 : spdk_nvme_transport_id_parse_adrfam(adrfam, str);
1170 : 4 : CU_ASSERT((*adrfam) == SPDK_NVMF_ADRFAM_IPV6);
1171 : :
1172 : : /* test adrfam value when use function "strcasecmp" to compare str and "IB",not case-sensitive */
1173 : 4 : str = "IB";
1174 : 4 : spdk_nvme_transport_id_parse_adrfam(adrfam, str);
1175 : 4 : CU_ASSERT((*adrfam) == SPDK_NVMF_ADRFAM_IB);
1176 : :
1177 : 4 : str = "ib";
1178 : 4 : spdk_nvme_transport_id_parse_adrfam(adrfam, str);
1179 : 4 : CU_ASSERT((*adrfam) == SPDK_NVMF_ADRFAM_IB);
1180 : :
1181 : : /* test adrfam value when use function "strcasecmp" to compare str and "FC",not case-sensitive */
1182 : 4 : str = "FC";
1183 : 4 : spdk_nvme_transport_id_parse_adrfam(adrfam, str);
1184 : 4 : CU_ASSERT((*adrfam) == SPDK_NVMF_ADRFAM_FC);
1185 : :
1186 : 4 : str = "fc";
1187 : 4 : spdk_nvme_transport_id_parse_adrfam(adrfam, str);
1188 : 4 : CU_ASSERT((*adrfam) == SPDK_NVMF_ADRFAM_FC);
1189 : :
1190 : 4 : }
1191 : :
1192 : : static void
1193 : 4 : test_trid_trtype_str(void)
1194 : : {
1195 : : const char *s;
1196 : :
1197 : 4 : s = spdk_nvme_transport_id_trtype_str(-5);
1198 : 4 : CU_ASSERT(s == NULL);
1199 : :
1200 : 4 : s = spdk_nvme_transport_id_trtype_str(SPDK_NVME_TRANSPORT_PCIE);
1201 [ + + ]: 4 : SPDK_CU_ASSERT_FATAL(s != NULL);
1202 [ - + ]: 4 : CU_ASSERT(strcmp(s, "PCIe") == 0);
1203 : :
1204 : 4 : s = spdk_nvme_transport_id_trtype_str(SPDK_NVME_TRANSPORT_RDMA);
1205 [ + + ]: 4 : SPDK_CU_ASSERT_FATAL(s != NULL);
1206 [ - + ]: 4 : CU_ASSERT(strcmp(s, "RDMA") == 0);
1207 : :
1208 : 4 : s = spdk_nvme_transport_id_trtype_str(SPDK_NVME_TRANSPORT_FC);
1209 [ + + ]: 4 : SPDK_CU_ASSERT_FATAL(s != NULL);
1210 [ - + ]: 4 : CU_ASSERT(strcmp(s, "FC") == 0);
1211 : :
1212 : 4 : s = spdk_nvme_transport_id_trtype_str(SPDK_NVME_TRANSPORT_TCP);
1213 [ - + ]: 4 : SPDK_CU_ASSERT_FATAL(s != NULL);
1214 [ - + ]: 4 : CU_ASSERT(strcmp(s, "TCP") == 0);
1215 : 4 : }
1216 : :
1217 : : static void
1218 : 4 : test_trid_adrfam_str(void)
1219 : : {
1220 : : const char *s;
1221 : :
1222 : 4 : s = spdk_nvme_transport_id_adrfam_str(-5);
1223 : 4 : CU_ASSERT(s == NULL);
1224 : :
1225 : 4 : s = spdk_nvme_transport_id_adrfam_str(SPDK_NVMF_ADRFAM_IPV4);
1226 [ + + ]: 4 : SPDK_CU_ASSERT_FATAL(s != NULL);
1227 [ - + ]: 4 : CU_ASSERT(strcmp(s, "IPv4") == 0);
1228 : :
1229 : 4 : s = spdk_nvme_transport_id_adrfam_str(SPDK_NVMF_ADRFAM_IPV6);
1230 [ + + ]: 4 : SPDK_CU_ASSERT_FATAL(s != NULL);
1231 [ - + ]: 4 : CU_ASSERT(strcmp(s, "IPv6") == 0);
1232 : :
1233 : 4 : s = spdk_nvme_transport_id_adrfam_str(SPDK_NVMF_ADRFAM_IB);
1234 [ + + ]: 4 : SPDK_CU_ASSERT_FATAL(s != NULL);
1235 [ - + ]: 4 : CU_ASSERT(strcmp(s, "IB") == 0);
1236 : :
1237 : 4 : s = spdk_nvme_transport_id_adrfam_str(SPDK_NVMF_ADRFAM_FC);
1238 [ - + ]: 4 : SPDK_CU_ASSERT_FATAL(s != NULL);
1239 [ - + ]: 4 : CU_ASSERT(strcmp(s, "FC") == 0);
1240 : 4 : }
1241 : :
1242 : : /* stub callback used by the test_nvme_request_check_timeout */
1243 : : static bool ut_timeout_cb_call = false;
1244 : : static void
1245 : 4 : dummy_timeout_cb(void *cb_arg, struct spdk_nvme_ctrlr *ctrlr,
1246 : : struct spdk_nvme_qpair *qpair, uint16_t cid)
1247 : : {
1248 : 4 : ut_timeout_cb_call = true;
1249 : 4 : }
1250 : :
1251 : : static void
1252 : 4 : test_nvme_request_check_timeout(void)
1253 : : {
1254 : : int rc;
1255 : 3 : struct spdk_nvme_qpair qpair;
1256 : 3 : struct nvme_request req;
1257 : 3 : struct spdk_nvme_ctrlr_process active_proc;
1258 : 4 : uint16_t cid = 0;
1259 : 4 : uint64_t now_tick = 0;
1260 : :
1261 [ - + ]: 4 : memset(&qpair, 0x0, sizeof(qpair));
1262 [ - + ]: 4 : memset(&req, 0x0, sizeof(req));
1263 [ - + ]: 4 : memset(&active_proc, 0x0, sizeof(active_proc));
1264 : 4 : req.qpair = &qpair;
1265 : 4 : active_proc.timeout_cb_fn = dummy_timeout_cb;
1266 : :
1267 : : /* if have called timeout_cb_fn then return directly */
1268 : 4 : req.timed_out = true;
1269 : 4 : rc = nvme_request_check_timeout(&req, cid, &active_proc, now_tick);
1270 : 4 : CU_ASSERT(rc == 0);
1271 [ - + ]: 4 : CU_ASSERT(ut_timeout_cb_call == false);
1272 : :
1273 : : /* if timeout isn't enabled then return directly */
1274 : 4 : req.timed_out = false;
1275 : 4 : req.submit_tick = 0;
1276 : 4 : rc = nvme_request_check_timeout(&req, cid, &active_proc, now_tick);
1277 : 4 : CU_ASSERT(rc == 0);
1278 [ - + ]: 4 : CU_ASSERT(ut_timeout_cb_call == false);
1279 : :
1280 : : /* req->pid isn't right then return directly */
1281 : 4 : req.submit_tick = 1;
1282 : 4 : req.pid = g_spdk_nvme_pid + 1;
1283 : 4 : rc = nvme_request_check_timeout(&req, cid, &active_proc, now_tick);
1284 : 4 : CU_ASSERT(rc == 0);
1285 [ - + ]: 4 : CU_ASSERT(ut_timeout_cb_call == false);
1286 : :
1287 : : /* AER command has no timeout */
1288 : 4 : req.pid = g_spdk_nvme_pid;
1289 : 4 : req.cmd.opc = SPDK_NVME_OPC_ASYNC_EVENT_REQUEST;
1290 : 4 : rc = nvme_request_check_timeout(&req, cid, &active_proc, now_tick);
1291 : 4 : CU_ASSERT(rc == 0);
1292 [ - + ]: 4 : CU_ASSERT(ut_timeout_cb_call == false);
1293 : :
1294 : : /* time isn't out */
1295 : 4 : qpair.id = 1;
1296 : 4 : rc = nvme_request_check_timeout(&req, cid, &active_proc, now_tick);
1297 : 4 : CU_ASSERT(rc == 1);
1298 [ - + ]: 4 : CU_ASSERT(ut_timeout_cb_call == false);
1299 : :
1300 : 4 : now_tick = 2;
1301 : 4 : rc = nvme_request_check_timeout(&req, cid, &active_proc, now_tick);
1302 : 4 : CU_ASSERT(req.timed_out == true);
1303 [ - + ]: 4 : CU_ASSERT(ut_timeout_cb_call == true);
1304 : 4 : CU_ASSERT(rc == 0);
1305 : 4 : }
1306 : :
1307 : : struct nvme_completion_poll_status g_status;
1308 : : uint64_t completion_delay_us, timeout_in_usecs;
1309 : : int g_process_comp_result;
1310 : : pthread_mutex_t g_robust_lock = PTHREAD_MUTEX_INITIALIZER;
1311 : :
1312 : : int
1313 : 40 : spdk_nvme_qpair_process_completions(struct spdk_nvme_qpair *qpair, uint32_t max_completions)
1314 : : {
1315 : 40 : spdk_delay_us(completion_delay_us);
1316 : :
1317 [ + + + + ]: 40 : g_status.done = completion_delay_us < timeout_in_usecs && g_process_comp_result == 0 ? true : false;
1318 : :
1319 : 40 : return g_process_comp_result;
1320 : : }
1321 : :
1322 : : static void
1323 : 4 : test_nvme_wait_for_completion(void)
1324 : : {
1325 : 3 : struct spdk_nvme_qpair qpair;
1326 : 3 : struct spdk_nvme_ctrlr ctrlr;
1327 : 4 : int rc = 0;
1328 : :
1329 [ - + ]: 4 : memset(&ctrlr, 0, sizeof(ctrlr));
1330 : 4 : ctrlr.trid.trtype = SPDK_NVME_TRANSPORT_PCIE;
1331 [ - + ]: 4 : memset(&qpair, 0, sizeof(qpair));
1332 : 4 : qpair.ctrlr = &ctrlr;
1333 : :
1334 : : /* completion timeout */
1335 [ - + ]: 4 : memset(&g_status, 0, sizeof(g_status));
1336 : 4 : completion_delay_us = 2000000;
1337 : 4 : timeout_in_usecs = 1000000;
1338 : 4 : rc = nvme_wait_for_completion_timeout(&qpair, &g_status, timeout_in_usecs);
1339 [ - + ]: 4 : CU_ASSERT(g_status.timed_out == true);
1340 [ - + ]: 4 : CU_ASSERT(g_status.done == false);
1341 : 4 : CU_ASSERT(rc == -ECANCELED);
1342 : :
1343 : : /* spdk_nvme_qpair_process_completions returns error */
1344 [ - + ]: 4 : memset(&g_status, 0, sizeof(g_status));
1345 : 4 : g_process_comp_result = -1;
1346 : 4 : completion_delay_us = 1000000;
1347 : 4 : timeout_in_usecs = 2000000;
1348 : 4 : rc = nvme_wait_for_completion_timeout(&qpair, &g_status, timeout_in_usecs);
1349 : 4 : CU_ASSERT(rc == -ECANCELED);
1350 [ - + ]: 4 : CU_ASSERT(g_status.timed_out == true);
1351 [ - + ]: 4 : CU_ASSERT(g_status.done == false);
1352 : 4 : CU_ASSERT(g_status.cpl.status.sct == SPDK_NVME_SCT_GENERIC);
1353 : 4 : CU_ASSERT(g_status.cpl.status.sc == SPDK_NVME_SC_ABORTED_SQ_DELETION);
1354 : :
1355 : 4 : g_process_comp_result = 0;
1356 : :
1357 : : /* complete in time */
1358 [ - + ]: 4 : memset(&g_status, 0, sizeof(g_status));
1359 : 4 : completion_delay_us = 1000000;
1360 : 4 : timeout_in_usecs = 2000000;
1361 : 4 : rc = nvme_wait_for_completion_timeout(&qpair, &g_status, timeout_in_usecs);
1362 [ - + ]: 4 : CU_ASSERT(g_status.timed_out == false);
1363 [ - + ]: 4 : CU_ASSERT(g_status.done == true);
1364 : 4 : CU_ASSERT(rc == 0);
1365 : :
1366 : : /* nvme_wait_for_completion */
1367 : : /* spdk_nvme_qpair_process_completions returns error */
1368 [ - + ]: 4 : memset(&g_status, 0, sizeof(g_status));
1369 : 4 : g_process_comp_result = -1;
1370 : 4 : rc = nvme_wait_for_completion(&qpair, &g_status);
1371 : 4 : CU_ASSERT(rc == -ECANCELED);
1372 [ - + ]: 4 : CU_ASSERT(g_status.timed_out == true);
1373 [ - + ]: 4 : CU_ASSERT(g_status.done == false);
1374 : 4 : CU_ASSERT(g_status.cpl.status.sct == SPDK_NVME_SCT_GENERIC);
1375 : 4 : CU_ASSERT(g_status.cpl.status.sc == SPDK_NVME_SC_ABORTED_SQ_DELETION);
1376 : :
1377 : : /* successful completion */
1378 [ - + ]: 4 : memset(&g_status, 0, sizeof(g_status));
1379 : 4 : g_process_comp_result = 0;
1380 : 4 : rc = nvme_wait_for_completion(&qpair, &g_status);
1381 : 4 : CU_ASSERT(rc == 0);
1382 [ - + ]: 4 : CU_ASSERT(g_status.timed_out == false);
1383 [ - + ]: 4 : CU_ASSERT(g_status.done == true);
1384 : :
1385 : : /* completion timeout */
1386 [ - + ]: 4 : memset(&g_status, 0, sizeof(g_status));
1387 : 4 : completion_delay_us = 2000000;
1388 : 4 : timeout_in_usecs = 1000000;
1389 : 4 : rc = nvme_wait_for_completion_robust_lock_timeout(&qpair, &g_status, &g_robust_lock,
1390 : 1 : timeout_in_usecs);
1391 [ - + ]: 4 : CU_ASSERT(g_status.timed_out == true);
1392 [ - + ]: 4 : CU_ASSERT(g_status.done == false);
1393 : 4 : CU_ASSERT(rc == -ECANCELED);
1394 : :
1395 : : /* spdk_nvme_qpair_process_completions returns error */
1396 [ - + ]: 4 : memset(&g_status, 0, sizeof(g_status));
1397 : 4 : g_process_comp_result = -1;
1398 : 4 : completion_delay_us = 1000000;
1399 : 4 : timeout_in_usecs = 2000000;
1400 : 4 : rc = nvme_wait_for_completion_robust_lock_timeout(&qpair, &g_status, &g_robust_lock,
1401 : 1 : timeout_in_usecs);
1402 : 4 : CU_ASSERT(rc == -ECANCELED);
1403 [ - + ]: 4 : CU_ASSERT(g_status.timed_out == true);
1404 [ - + ]: 4 : CU_ASSERT(g_status.done == false);
1405 : 4 : CU_ASSERT(g_status.cpl.status.sct == SPDK_NVME_SCT_GENERIC);
1406 : 4 : CU_ASSERT(g_status.cpl.status.sc == SPDK_NVME_SC_ABORTED_SQ_DELETION);
1407 : :
1408 : 4 : g_process_comp_result = 0;
1409 : :
1410 : : /* complete in time */
1411 [ - + ]: 4 : memset(&g_status, 0, sizeof(g_status));
1412 : 4 : completion_delay_us = 1000000;
1413 : 4 : timeout_in_usecs = 2000000;
1414 : 4 : rc = nvme_wait_for_completion_robust_lock_timeout(&qpair, &g_status, &g_robust_lock,
1415 : 1 : timeout_in_usecs);
1416 [ - + ]: 4 : CU_ASSERT(g_status.timed_out == false);
1417 [ - + ]: 4 : CU_ASSERT(g_status.done == true);
1418 : 4 : CU_ASSERT(rc == 0);
1419 : :
1420 : : /* nvme_wait_for_completion */
1421 : : /* spdk_nvme_qpair_process_completions returns error */
1422 [ - + ]: 4 : memset(&g_status, 0, sizeof(g_status));
1423 : 4 : g_process_comp_result = -1;
1424 : 4 : rc = nvme_wait_for_completion_robust_lock(&qpair, &g_status, &g_robust_lock);
1425 : 4 : CU_ASSERT(rc == -ECANCELED);
1426 [ - + ]: 4 : CU_ASSERT(g_status.timed_out == true);
1427 [ - + ]: 4 : CU_ASSERT(g_status.done == false);
1428 : 4 : CU_ASSERT(g_status.cpl.status.sct == SPDK_NVME_SCT_GENERIC);
1429 : 4 : CU_ASSERT(g_status.cpl.status.sc == SPDK_NVME_SC_ABORTED_SQ_DELETION);
1430 : :
1431 : : /* successful completion */
1432 [ - + ]: 4 : memset(&g_status, 0, sizeof(g_status));
1433 : 4 : g_process_comp_result = 0;
1434 : 4 : rc = nvme_wait_for_completion_robust_lock(&qpair, &g_status, &g_robust_lock);
1435 : 4 : CU_ASSERT(rc == 0);
1436 [ - + ]: 4 : CU_ASSERT(g_status.timed_out == false);
1437 [ - + ]: 4 : CU_ASSERT(g_status.done == true);
1438 : 4 : }
1439 : :
1440 : : static void
1441 : 4 : test_nvme_ctrlr_probe_internal(void)
1442 : : {
1443 : : struct spdk_nvme_probe_ctx *probe_ctx;
1444 : 4 : struct spdk_nvme_transport_id trid = {};
1445 : 4 : struct nvme_driver dummy = {};
1446 : : int rc;
1447 : :
1448 : 4 : probe_ctx = calloc(1, sizeof(*probe_ctx));
1449 : 4 : CU_ASSERT(probe_ctx != NULL);
1450 : :
1451 : 4 : MOCK_SET(spdk_process_is_primary, true);
1452 : 4 : MOCK_SET(spdk_memzone_reserve, (void *)&dummy);
1453 : 4 : g_spdk_nvme_driver = NULL;
1454 : 4 : rc = nvme_driver_init();
1455 : 4 : CU_ASSERT(rc == 0);
1456 : :
1457 : 4 : ut_attach_fail_cb_rc = 0;
1458 : 4 : ut_test_probe_internal = true;
1459 : 4 : MOCK_SET(dummy_probe_cb, true);
1460 : 4 : trid.trtype = SPDK_NVME_TRANSPORT_PCIE;
1461 : 4 : nvme_probe_ctx_init(probe_ctx, &trid, NULL, NULL, dummy_probe_cb, NULL, dummy_attach_fail_cb,
1462 : : NULL);
1463 : 4 : rc = nvme_probe_internal(probe_ctx, false);
1464 : 4 : CU_ASSERT(rc < 0);
1465 : 4 : CU_ASSERT(TAILQ_EMPTY(&probe_ctx->init_ctrlrs));
1466 : 4 : CU_ASSERT(ut_attach_fail_cb_rc == -EFAULT);
1467 : :
1468 : 4 : free(probe_ctx);
1469 : 4 : ut_test_probe_internal = false;
1470 : 4 : }
1471 : :
1472 : : static void
1473 : 4 : test_spdk_nvme_parse_func(void)
1474 : : {
1475 : 4 : struct spdk_nvme_host_id hostid = {};
1476 : 4 : char str[64] = {};
1477 : 4 : const char *rt_str = NULL;
1478 : 3 : uint32_t prchk_flags;
1479 : : int rc;
1480 : :
1481 : : /* Parse prchk flags. */
1482 : 4 : prchk_flags = 0;
1483 : 4 : rt_str = spdk_nvme_prchk_flags_str(SPDK_NVME_IO_FLAGS_PRCHK_REFTAG);
1484 [ - + - + ]: 4 : memcpy(str, rt_str, strlen(rt_str));
1485 : :
1486 : 4 : rc = spdk_nvme_prchk_flags_parse(&prchk_flags, str);
1487 : 4 : CU_ASSERT(rc == 0);
1488 : 4 : CU_ASSERT(prchk_flags & SPDK_NVME_IO_FLAGS_PRCHK_REFTAG);
1489 : :
1490 : 4 : prchk_flags = 0;
1491 : 4 : rt_str = spdk_nvme_prchk_flags_str(SPDK_NVME_IO_FLAGS_PRCHK_GUARD);
1492 [ - + - + ]: 4 : memcpy(str, rt_str, strlen(rt_str));
1493 : :
1494 : 4 : rc = spdk_nvme_prchk_flags_parse(&prchk_flags, str);
1495 : 4 : CU_ASSERT(prchk_flags & SPDK_NVME_IO_FLAGS_PRCHK_GUARD);
1496 : 4 : CU_ASSERT(rc == 0);
1497 : :
1498 : 4 : prchk_flags = 0;
1499 : 4 : rt_str = spdk_nvme_prchk_flags_str(SPDK_NVME_IO_FLAGS_PRCHK_REFTAG |
1500 : : SPDK_NVME_IO_FLAGS_PRCHK_GUARD);
1501 [ - + - + ]: 4 : memcpy(str, rt_str, strlen(rt_str));
1502 : :
1503 : 4 : rc = spdk_nvme_prchk_flags_parse(&prchk_flags, str);
1504 : 4 : CU_ASSERT(rc == 0);
1505 : 4 : CU_ASSERT(prchk_flags & SPDK_NVME_IO_FLAGS_PRCHK_REFTAG);
1506 : 4 : CU_ASSERT(prchk_flags & SPDK_NVME_IO_FLAGS_PRCHK_GUARD);
1507 : :
1508 : 4 : rc = spdk_nvme_prchk_flags_parse(NULL, NULL);
1509 : 4 : CU_ASSERT(rc == -EINVAL);
1510 : :
1511 : : /* Parse host id. */
1512 : 4 : memcpy(str, "hostaddr:192.168.1.1", sizeof("hostaddr:192.168.1.1"));
1513 : 4 : rc = spdk_nvme_host_id_parse(&hostid, str);
1514 : 4 : CU_ASSERT(rc == 0);
1515 : 4 : CU_ASSERT(!strncmp(hostid.hostaddr, "192.168.1.1", sizeof("192.168.1.1")));
1516 : :
1517 : 4 : memset(&hostid, 0, sizeof(hostid));
1518 : 4 : memcpy(str, "hostsvcid:192.168.1.2", sizeof("hostsvcid:192.168.1.2"));
1519 : 4 : rc = spdk_nvme_host_id_parse(&hostid, str);
1520 : 4 : CU_ASSERT(rc == 0);
1521 : 4 : CU_ASSERT(!strncmp(hostid.hostsvcid, "192.168.1.2", sizeof("192.168.1.2")));
1522 : :
1523 : : /* Unknown transport ID key */
1524 : 4 : memset(&hostid, 0, sizeof(hostid));
1525 : 4 : memcpy(str, "trtype:xxx", sizeof("trtype:xxx"));
1526 : 4 : rc = spdk_nvme_host_id_parse(&hostid, str);
1527 : 4 : CU_ASSERT(rc == 0);
1528 [ + - + - ]: 4 : CU_ASSERT(hostid.hostaddr[0] == '\0' && hostid.hostsvcid[0] == '\0');
1529 : 4 : }
1530 : :
1531 : : static void
1532 : 4 : test_spdk_nvme_detach_async(void)
1533 : : {
1534 : 4 : int rc = 1;
1535 : 3 : struct spdk_nvme_ctrlr ctrlr1, ctrlr2;
1536 : 4 : struct nvme_driver test_driver = {};
1537 : 3 : struct spdk_nvme_detach_ctx *detach_ctx;
1538 : : struct nvme_ctrlr_detach_ctx *ctx;
1539 : :
1540 : 4 : detach_ctx = NULL;
1541 [ - + ]: 4 : memset(&ctrlr1, 0, sizeof(ctrlr1));
1542 : 4 : ctrlr1.trid.trtype = SPDK_NVME_TRANSPORT_PCIE;
1543 [ - + ]: 4 : memset(&ctrlr2, 0, sizeof(ctrlr2));
1544 : 4 : ctrlr2.trid.trtype = SPDK_NVME_TRANSPORT_PCIE;
1545 : :
1546 : 4 : g_spdk_nvme_driver = &test_driver;
1547 : 4 : TAILQ_INIT(&test_driver.shared_attached_ctrlrs);
1548 : 4 : TAILQ_INSERT_TAIL(&test_driver.shared_attached_ctrlrs, &ctrlr1, tailq);
1549 : 4 : TAILQ_INSERT_TAIL(&test_driver.shared_attached_ctrlrs, &ctrlr2, tailq);
1550 [ - + ]: 4 : CU_ASSERT(pthread_mutex_init(&test_driver.lock, NULL) == 0);
1551 : 4 : MOCK_SET(nvme_ctrlr_get_ref_count, 1);
1552 : :
1553 : 4 : rc = spdk_nvme_detach_async(&ctrlr1, &detach_ctx);
1554 : 4 : CU_ASSERT(rc == 0);
1555 [ - + ]: 4 : CU_ASSERT(ctrlr1.is_destructed == true);
1556 : 4 : CU_ASSERT(detach_ctx != NULL);
1557 : :
1558 : 4 : rc = spdk_nvme_detach_async(&ctrlr2, &detach_ctx);
1559 : 4 : CU_ASSERT(rc == 0);
1560 [ - + ]: 4 : CU_ASSERT(ctrlr2.is_destructed == true);
1561 : 4 : CU_ASSERT(detach_ctx != NULL);
1562 : :
1563 : 4 : CU_ASSERT(TAILQ_EMPTY(&test_driver.shared_attached_ctrlrs) == false);
1564 : :
1565 : 4 : rc = spdk_nvme_detach_poll_async(detach_ctx);
1566 : 4 : CU_ASSERT(rc == 0);
1567 : 4 : CU_ASSERT(TAILQ_EMPTY(&test_driver.shared_attached_ctrlrs) == true);
1568 : :
1569 : : /* ctrlr1 is a PCIe controller but ctrlr2 is an non-PCIe controller.
1570 : : * Even for this case, detachment should complete successfully.
1571 : : */
1572 : 4 : detach_ctx = NULL;
1573 [ - + ]: 4 : memset(&ctrlr1, 0, sizeof(ctrlr1));
1574 : 4 : ctrlr1.trid.trtype = SPDK_NVME_TRANSPORT_RDMA;
1575 [ - + ]: 4 : memset(&ctrlr2, 0, sizeof(ctrlr2));
1576 : 4 : ctrlr2.trid.trtype = SPDK_NVME_TRANSPORT_PCIE;
1577 : 4 : TAILQ_INIT(&g_nvme_attached_ctrlrs);
1578 : 4 : TAILQ_INSERT_TAIL(&g_nvme_attached_ctrlrs, &ctrlr1, tailq);
1579 : 4 : TAILQ_INSERT_TAIL(&test_driver.shared_attached_ctrlrs, &ctrlr2, tailq);
1580 : :
1581 : 4 : rc = spdk_nvme_detach_async(&ctrlr1, &detach_ctx);
1582 : 4 : CU_ASSERT(rc == 0);
1583 [ - + ]: 4 : CU_ASSERT(ctrlr1.is_destructed == true);
1584 : 4 : CU_ASSERT(detach_ctx != NULL);
1585 : :
1586 : 4 : rc = spdk_nvme_detach_async(&ctrlr2, &detach_ctx);
1587 : 4 : CU_ASSERT(rc == 0);
1588 [ - + ]: 4 : CU_ASSERT(ctrlr2.is_destructed == true);
1589 : 4 : CU_ASSERT(detach_ctx != NULL);
1590 : :
1591 : 4 : CU_ASSERT(TAILQ_EMPTY(&g_nvme_attached_ctrlrs) == false);
1592 : 4 : CU_ASSERT(TAILQ_EMPTY(&test_driver.shared_attached_ctrlrs) == false);
1593 : :
1594 : 4 : rc = spdk_nvme_detach_poll_async(detach_ctx);
1595 : 4 : CU_ASSERT(rc == 0);
1596 : 4 : CU_ASSERT(TAILQ_EMPTY(&g_nvme_attached_ctrlrs) == true);
1597 : 4 : CU_ASSERT(TAILQ_EMPTY(&test_driver.shared_attached_ctrlrs) == true);
1598 : :
1599 : : /* Test if ctrlr2 can be detached by using the same context that
1600 : : * ctrlr1 uses while ctrlr1 is being detached.
1601 : : */
1602 : 4 : detach_ctx = NULL;
1603 [ - + ]: 4 : memset(&ctrlr1, 0, sizeof(ctrlr1));
1604 : 4 : ctrlr1.trid.trtype = SPDK_NVME_TRANSPORT_PCIE;
1605 [ - + ]: 4 : memset(&ctrlr2, 0, sizeof(ctrlr2));
1606 : 4 : ctrlr2.trid.trtype = SPDK_NVME_TRANSPORT_PCIE;
1607 : 4 : TAILQ_INSERT_TAIL(&test_driver.shared_attached_ctrlrs, &ctrlr1, tailq);
1608 : 4 : TAILQ_INSERT_TAIL(&test_driver.shared_attached_ctrlrs, &ctrlr2, tailq);
1609 : :
1610 : 4 : rc = spdk_nvme_detach_async(&ctrlr1, &detach_ctx);
1611 : 4 : CU_ASSERT(rc == 0);
1612 [ - + ]: 4 : CU_ASSERT(ctrlr1.is_destructed == true);
1613 [ + + ]: 4 : SPDK_CU_ASSERT_FATAL(detach_ctx != NULL);
1614 : :
1615 : 4 : ctx = TAILQ_FIRST(&detach_ctx->head);
1616 [ - + ]: 4 : SPDK_CU_ASSERT_FATAL(ctx != NULL);
1617 : 4 : CU_ASSERT(ctx->ctrlr == &ctrlr1);
1618 [ - + ]: 4 : CU_ASSERT(ctx->shutdown_complete == true);
1619 : :
1620 : : /* Set ctx->shutdown_complete for ctrlr1 to false to allow ctrlr2 to
1621 : : * add to detach_ctx while spdk_nvme_detach_poll_async() is being
1622 : : * executed.
1623 : : */
1624 : 4 : ctx->shutdown_complete = false;
1625 : :
1626 : 4 : rc = spdk_nvme_detach_poll_async(detach_ctx);
1627 : 4 : CU_ASSERT(rc == -EAGAIN);
1628 : :
1629 : 4 : rc = spdk_nvme_detach_async(&ctrlr2, &detach_ctx);
1630 : 4 : CU_ASSERT(rc == 0);
1631 [ - + ]: 4 : CU_ASSERT(ctrlr2.is_destructed == true);
1632 : :
1633 : : /* After ctrlr2 is added to detach_ctx, set ctx->shutdown_complete for
1634 : : * ctrlr1 to true to complete spdk_nvme_detach_poll_async().
1635 : : */
1636 : 4 : ctx->shutdown_complete = true;
1637 : :
1638 : 4 : rc = spdk_nvme_detach_poll_async(detach_ctx);
1639 : 4 : CU_ASSERT(rc == 0);
1640 : 4 : CU_ASSERT(TAILQ_EMPTY(&test_driver.shared_attached_ctrlrs) == true);
1641 : :
1642 : 4 : g_spdk_nvme_driver = NULL;
1643 [ - + ]: 4 : pthread_mutex_destroy(&test_driver.lock);
1644 [ - + - + ]: 4 : MOCK_CLEAR(nvme_ctrlr_get_ref_count);
1645 : 4 : }
1646 : :
1647 : : static void
1648 : 4 : test_nvme_parse_addr(void)
1649 : : {
1650 : 3 : struct sockaddr_storage dst_addr;
1651 : 4 : int rc = 0;
1652 : 3 : long int port;
1653 : :
1654 [ - + ]: 4 : memset(&dst_addr, 0, sizeof(dst_addr));
1655 : : /* case1: getaddrinfo failed */
1656 : 4 : rc = nvme_parse_addr(&dst_addr, AF_INET, NULL, NULL, &port);
1657 : 4 : CU_ASSERT(rc != 0);
1658 : :
1659 : : /* case2: res->ai_addrlen < sizeof(*sa). Expect: Pass. */
1660 : 4 : rc = nvme_parse_addr(&dst_addr, AF_INET, "12.34.56.78", "23", &port);
1661 : 4 : CU_ASSERT(rc == 0);
1662 : 4 : CU_ASSERT(port == 23);
1663 : 4 : CU_ASSERT(dst_addr.ss_family == AF_INET);
1664 : 4 : }
1665 : :
1666 : : int
1667 : 4 : main(int argc, char **argv)
1668 : : {
1669 : 4 : CU_pSuite suite = NULL;
1670 : : unsigned int num_failures;
1671 : :
1672 : 4 : CU_initialize_registry();
1673 : :
1674 : 4 : suite = CU_add_suite("nvme", NULL, NULL);
1675 : :
1676 : 4 : CU_ADD_TEST(suite, test_opc_data_transfer);
1677 : 4 : CU_ADD_TEST(suite, test_spdk_nvme_transport_id_parse_trtype);
1678 : 4 : CU_ADD_TEST(suite, test_spdk_nvme_transport_id_parse_adrfam);
1679 : 4 : CU_ADD_TEST(suite, test_trid_parse_and_compare);
1680 : 4 : CU_ADD_TEST(suite, test_trid_trtype_str);
1681 : 4 : CU_ADD_TEST(suite, test_trid_adrfam_str);
1682 : 4 : CU_ADD_TEST(suite, test_nvme_ctrlr_probe);
1683 : 4 : CU_ADD_TEST(suite, test_spdk_nvme_probe_ext);
1684 : 4 : CU_ADD_TEST(suite, test_spdk_nvme_connect);
1685 : 4 : CU_ADD_TEST(suite, test_nvme_ctrlr_probe_internal);
1686 : 4 : CU_ADD_TEST(suite, test_nvme_init_controllers);
1687 : 4 : CU_ADD_TEST(suite, test_nvme_driver_init);
1688 : 4 : CU_ADD_TEST(suite, test_spdk_nvme_detach);
1689 : 4 : CU_ADD_TEST(suite, test_nvme_completion_poll_cb);
1690 : 4 : CU_ADD_TEST(suite, test_nvme_user_copy_cmd_complete);
1691 : 4 : CU_ADD_TEST(suite, test_nvme_allocate_request_null);
1692 : 4 : CU_ADD_TEST(suite, test_nvme_allocate_request);
1693 : 4 : CU_ADD_TEST(suite, test_nvme_free_request);
1694 : 4 : CU_ADD_TEST(suite, test_nvme_allocate_request_user_copy);
1695 : 4 : CU_ADD_TEST(suite, test_nvme_robust_mutex_init_shared);
1696 : 4 : CU_ADD_TEST(suite, test_nvme_request_check_timeout);
1697 : 4 : CU_ADD_TEST(suite, test_nvme_wait_for_completion);
1698 : 4 : CU_ADD_TEST(suite, test_spdk_nvme_parse_func);
1699 : 4 : CU_ADD_TEST(suite, test_spdk_nvme_detach_async);
1700 : 4 : CU_ADD_TEST(suite, test_nvme_parse_addr);
1701 : :
1702 : 4 : num_failures = spdk_ut_run_tests(argc, argv, NULL);
1703 : 4 : CU_cleanup_registry();
1704 : 4 : return num_failures;
1705 : : }
|