Branch data Line data Source code
1 : : /* SPDX-License-Identifier: BSD-3-Clause
2 : : *
3 : : * Copyright (C) 2020 Intel Corporation.
4 : : * All rights reserved.
5 : : */
6 : :
7 : : #include "spdk_internal/cunit.h"
8 : :
9 : : #include "common/lib/test_env.c"
10 : : #include "nvme/nvme_cuse.c"
11 : :
12 [ # # ]: 0 : DEFINE_STUB(nvme_io_msg_send, int, (struct spdk_nvme_ctrlr *ctrlr, uint32_t nsid,
13 : : spdk_nvme_io_msg_fn fn, void *arg), 0);
14 : :
15 [ # # ]: 0 : DEFINE_STUB(spdk_nvme_ctrlr_cmd_admin_raw, int, (struct spdk_nvme_ctrlr *ctrlr,
16 : : struct spdk_nvme_cmd *cmd, void *buf, uint32_t len,
17 : : spdk_nvme_cmd_cb cb_fn, void *cb_arg), 0);
18 : :
19 [ # # ]: 0 : DEFINE_STUB(spdk_nvme_ctrlr_cmd_io_raw_with_md, int, (struct spdk_nvme_ctrlr *ctrlr,
20 : : struct spdk_nvme_qpair *qpair, struct spdk_nvme_cmd *cmd, void *buf, uint32_t len, void *md_buf,
21 : : spdk_nvme_cmd_cb cb_fn, void *cb_arg), 0);
22 : :
23 [ - + ]: 11 : DEFINE_STUB(spdk_nvme_ctrlr_get_num_ns, uint32_t, (struct spdk_nvme_ctrlr *ctrlr), 128);
24 : :
25 : : static uint32_t g_active_num_ns = 4;
26 : : static uint32_t g_active_nsid_min = 1;
27 : :
28 : : bool
29 : 1386 : spdk_nvme_ctrlr_is_active_ns(struct spdk_nvme_ctrlr *ctrlr, uint32_t nsid)
30 : : {
31 [ + + ]: 1386 : if (g_active_num_ns == 0) {
32 : 132 : return false;
33 : : }
34 [ + - + + : 1254 : return nsid && nsid >= g_active_nsid_min && nsid < g_active_num_ns + g_active_nsid_min;
+ + ]
35 : : }
36 : :
37 : : uint32_t
38 : 11 : spdk_nvme_ctrlr_get_first_active_ns(struct spdk_nvme_ctrlr *ctrlr)
39 : : {
40 [ + + ]: 11 : if (g_active_num_ns > 0) {
41 : 10 : return g_active_nsid_min;
42 : : } else {
43 : 1 : return 0;
44 : : }
45 : : }
46 : :
47 : : uint32_t
48 : 58 : spdk_nvme_ctrlr_get_next_active_ns(struct spdk_nvme_ctrlr *ctrlr, uint32_t nsid)
49 : : {
50 : 58 : nsid = nsid + 1;
51 : :
52 [ + + ]: 58 : if (spdk_nvme_ctrlr_is_active_ns(ctrlr, nsid)) {
53 : 48 : return nsid;
54 : : }
55 : :
56 : 10 : return 0;
57 : : }
58 : :
59 [ # # ]: 0 : DEFINE_STUB(spdk_nvme_ctrlr_reset, int, (struct spdk_nvme_ctrlr *ctrlr), 0);
60 : :
61 [ # # ]: 0 : DEFINE_STUB(spdk_nvme_ctrlr_reset_subsystem, int, (struct spdk_nvme_ctrlr *ctrlr), 0);
62 : :
63 [ # # ]: 0 : DEFINE_STUB(spdk_nvme_ns_cmd_read_with_md, int, (struct spdk_nvme_ns *ns,
64 : : struct spdk_nvme_qpair *qpair,
65 : : void *payload, void *metadata,
66 : : uint64_t lba, uint32_t lba_count, spdk_nvme_cmd_cb cb_fn, void *cb_arg,
67 : : uint32_t io_flags, uint16_t apptag_mask, uint16_t apptag), 0);
68 : :
69 [ # # ]: 0 : DEFINE_STUB(spdk_nvme_ns_cmd_write_with_md, int, (struct spdk_nvme_ns *ns,
70 : : struct spdk_nvme_qpair *qpair,
71 : : void *payload, void *metadata,
72 : : uint64_t lba, uint32_t lba_count, spdk_nvme_cmd_cb cb_fn, void *cb_arg,
73 : : uint32_t io_flags, uint16_t apptag_mask, uint16_t apptag), 0);
74 : :
75 [ # # ]: 0 : DEFINE_STUB(spdk_nvme_ns_get_num_sectors, uint64_t, (struct spdk_nvme_ns *ns), 0);
76 : :
77 [ # # ]: 0 : DEFINE_STUB(spdk_nvme_ns_get_sector_size, uint32_t, (struct spdk_nvme_ns *ns), 0);
78 : :
79 [ # # ]: 0 : DEFINE_STUB(spdk_nvme_ns_get_md_size, uint32_t, (struct spdk_nvme_ns *ns), 0);
80 : :
81 : 1 : DEFINE_STUB_V(spdk_unaffinitize_thread, (void));
82 : :
83 [ # # ]: 0 : DEFINE_STUB(spdk_nvme_ctrlr_get_ns, struct spdk_nvme_ns *, (struct spdk_nvme_ctrlr *ctrlr,
84 : : uint32_t nsid), NULL);
85 : :
86 [ - + ]: 1 : DEFINE_STUB(nvme_io_msg_ctrlr_register, int,
87 : : (struct spdk_nvme_ctrlr *ctrlr, struct nvme_io_msg_producer *io_msg_producer), 0);
88 : :
89 : 0 : DEFINE_STUB_V(nvme_io_msg_ctrlr_unregister,
90 : : (struct spdk_nvme_ctrlr *ctrlr, struct nvme_io_msg_producer *io_msg_producer));
91 : :
92 : 0 : DEFINE_STUB_V(nvme_ctrlr_update_namespaces, (struct spdk_nvme_ctrlr *ctrlr));
93 : :
94 : : static bool
95 : 1300 : wait_for_file(char *filename, bool exists)
96 : : {
97 : : int i;
98 : :
99 [ + - ]: 7569 : for (i = 0; i < 10000; i++) {
100 [ - + + + ]: 7569 : if ((access(filename, F_OK) != -1) ^ (!exists)) {
101 : 1300 : return true;
102 : : }
103 : 6269 : usleep(100);
104 : : }
105 : 0 : return false;
106 : : }
107 : :
108 : : static void
109 : 10 : verify_devices(struct spdk_nvme_ctrlr *ctrlr)
110 : : {
111 : 0 : char ctrlr_name[256];
112 : 0 : size_t ctrlr_name_size;
113 : 0 : char ctrlr_dev[256];
114 : 0 : char ns_dev[256 + 1 + 10]; /* sizeof ctrl_dev + 'n' + string size of UINT32_MAX */
115 : : uint32_t nsid, num_ns;
116 : : int rv;
117 : :
118 : 10 : ctrlr_name_size = sizeof(ctrlr_name);
119 : 10 : rv = spdk_nvme_cuse_get_ctrlr_name(ctrlr, ctrlr_name, &ctrlr_name_size);
120 [ - + ]: 10 : SPDK_CU_ASSERT_FATAL(rv == 0);
121 : :
122 [ - + ]: 10 : rv = snprintf(ctrlr_dev, sizeof(ctrlr_dev), "/dev/%s", ctrlr_name);
123 : 10 : CU_ASSERT(rv > 0);
124 : 10 : CU_ASSERT(wait_for_file(ctrlr_dev, true));
125 : :
126 : 10 : num_ns = spdk_nvme_ctrlr_get_num_ns(ctrlr);
127 : :
128 [ + + ]: 1290 : for (nsid = 1; nsid <= num_ns; nsid++) {
129 [ - + ]: 1280 : snprintf(ns_dev, sizeof(ns_dev), "%sn%" PRIu32, ctrlr_dev, nsid);
130 [ + + ]: 1280 : if (spdk_nvme_ctrlr_is_active_ns(ctrlr, nsid)) {
131 : 54 : CU_ASSERT(wait_for_file(ns_dev, true));
132 : : } else {
133 : 1226 : CU_ASSERT(wait_for_file(ns_dev, false));
134 : : }
135 : : }
136 : :
137 : : /* Next one should never exist */
138 [ - + ]: 10 : snprintf(ns_dev, sizeof(ns_dev), "%sn%" PRIu32, ctrlr_dev, nsid);
139 : 10 : CU_ASSERT(wait_for_file(ns_dev, false));
140 : 10 : }
141 : :
142 : : static void
143 : 1 : test_cuse_update(void)
144 : : {
145 : : int rc;
146 : 1 : struct spdk_nvme_ctrlr ctrlr = {};
147 : :
148 : 1 : rc = spdk_nvme_cuse_register(&ctrlr);
149 : 1 : CU_ASSERT(rc == 0);
150 : :
151 : 1 : g_active_num_ns = 4;
152 : 1 : g_active_nsid_min = 1;
153 : 1 : nvme_cuse_update(&ctrlr);
154 : 1 : verify_devices(&ctrlr);
155 : :
156 : 1 : g_active_num_ns = 0;
157 : 1 : g_active_nsid_min = 1;
158 : 1 : nvme_cuse_update(&ctrlr);
159 : 1 : verify_devices(&ctrlr);
160 : :
161 : 1 : g_active_num_ns = 4;
162 : 1 : g_active_nsid_min = spdk_nvme_ctrlr_get_num_ns(&ctrlr) - g_active_num_ns;
163 : 1 : nvme_cuse_update(&ctrlr);
164 : 1 : verify_devices(&ctrlr);
165 : :
166 : 1 : g_active_num_ns = 2;
167 : 1 : g_active_nsid_min = 2;
168 : 1 : nvme_cuse_update(&ctrlr);
169 : 1 : verify_devices(&ctrlr);
170 : :
171 : 1 : g_active_num_ns = 10;
172 : 1 : g_active_nsid_min = 5;
173 : 1 : nvme_cuse_update(&ctrlr);
174 : 1 : verify_devices(&ctrlr);
175 : :
176 : 1 : g_active_num_ns = 5;
177 : 1 : g_active_nsid_min = 3;
178 : 1 : nvme_cuse_update(&ctrlr);
179 : 1 : verify_devices(&ctrlr);
180 : :
181 : 1 : g_active_num_ns = 6;
182 : 1 : g_active_nsid_min = 1;
183 : 1 : nvme_cuse_update(&ctrlr);
184 : 1 : verify_devices(&ctrlr);
185 : :
186 : 1 : g_active_num_ns = 10;
187 : 1 : g_active_nsid_min = 10;
188 : 1 : nvme_cuse_update(&ctrlr);
189 : 1 : verify_devices(&ctrlr);
190 : :
191 : 1 : g_active_num_ns = 3;
192 : 1 : g_active_nsid_min = 13;
193 : 1 : nvme_cuse_update(&ctrlr);
194 : 1 : verify_devices(&ctrlr);
195 : :
196 : 1 : g_active_num_ns = 10;
197 : 1 : g_active_nsid_min = 10;
198 : 1 : nvme_cuse_update(&ctrlr);
199 : 1 : verify_devices(&ctrlr);
200 : :
201 : 1 : nvme_cuse_stop(&ctrlr);
202 : 1 : }
203 : :
204 : : int
205 : 1 : main(int argc, char **argv)
206 : : {
207 : 1 : CU_pSuite suite = NULL;
208 : : unsigned int num_failures;
209 : :
210 : 1 : CU_initialize_registry();
211 : 1 : suite = CU_add_suite("nvme_cuse", NULL, NULL);
212 : 1 : CU_ADD_TEST(suite, test_cuse_update);
213 : :
214 : 1 : num_failures = spdk_ut_run_tests(argc, argv, NULL);
215 : 1 : CU_cleanup_registry();
216 : 1 : return num_failures;
217 : : }
|