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