Branch data Line data Source code
1 : : /* SPDX-License-Identifier: BSD-3-Clause
2 : : * Copyright (C) 2018 Intel Corporation.
3 : : * Copyright (c) 2023-2024, NVIDIA CORPORATION & AFFILIATES.
4 : : * All rights reserved.
5 : : */
6 : :
7 : : #include "spdk_internal/cunit.h"
8 : :
9 : : #include "common/lib/ut_multithread.c"
10 : : #include "spdk_internal/mock.h"
11 : : #include "thread/thread_internal.h"
12 : : #include "unit/lib/json_mock.c"
13 : :
14 : : #include <rte_crypto.h>
15 : : #include <rte_cryptodev.h>
16 : : #include <rte_version.h>
17 : :
18 : : unsigned ut_rte_crypto_op_bulk_alloc;
19 : : int ut_rte_crypto_op_attach_sym_session = 0;
20 : : #define MOCK_INFO_GET_1QP_AESNI 0
21 : : #define MOCK_INFO_GET_1QP_QAT 1
22 : : #define MOCK_INFO_GET_1QP_MLX5 2
23 : : #define MOCK_INFO_GET_1QP_BOGUS_PMD 3
24 : : int ut_rte_cryptodev_info_get = 0;
25 : : bool ut_rte_cryptodev_info_get_mocked = false;
26 : :
27 : : #include "bdev/crypto/vbdev_crypto.c"
28 : :
29 : : /* SPDK stubs */
30 [ - + ]: 4 : DEFINE_STUB(spdk_bdev_queue_io_wait, int, (struct spdk_bdev *bdev, struct spdk_io_channel *ch,
31 : : struct spdk_bdev_io_wait_entry *entry), 0);
32 : 2 : DEFINE_STUB_V(spdk_bdev_module_list_add, (struct spdk_bdev_module *bdev_module));
33 : 16 : DEFINE_STUB_V(spdk_bdev_free_io, (struct spdk_bdev_io *g_bdev_io));
34 : 0 : DEFINE_STUB_V(spdk_bdev_io_put_aux_buf, (struct spdk_bdev_io *bdev_io, void *aux_buf));
35 [ # # # # ]: 0 : DEFINE_STUB(spdk_bdev_io_type_supported, bool, (struct spdk_bdev *bdev,
36 : : enum spdk_bdev_io_type io_type), 0);
37 : 0 : DEFINE_STUB_V(spdk_bdev_module_release_bdev, (struct spdk_bdev *bdev));
38 : 0 : DEFINE_STUB_V(spdk_bdev_close, (struct spdk_bdev_desc *desc));
39 [ # # ]: 0 : DEFINE_STUB(spdk_bdev_get_name, const char *, (const struct spdk_bdev *bdev), 0);
40 [ - + ]: 12 : DEFINE_STUB(spdk_bdev_get_buf_align, size_t, (const struct spdk_bdev *bdev), 64);
41 [ # # ]: 0 : DEFINE_STUB(spdk_bdev_get_io_channel, struct spdk_io_channel *, (struct spdk_bdev_desc *desc), 0);
42 : 0 : DEFINE_STUB_V(spdk_bdev_unregister, (struct spdk_bdev *bdev, spdk_bdev_unregister_cb cb_fn,
43 : : void *cb_arg));
44 [ # # ]: 0 : DEFINE_STUB(spdk_bdev_unregister_by_name, int, (const char *bdev_name,
45 : : struct spdk_bdev_module *module,
46 : : spdk_bdev_unregister_cb cb_fn, void *cb_arg), 0);
47 [ # # ]: 0 : DEFINE_STUB(spdk_bdev_open_ext, int, (const char *bdev_name, bool write,
48 : : spdk_bdev_event_cb_t event_cb,
49 : : void *event_ctx, struct spdk_bdev_desc **_desc), 0);
50 [ # # ]: 0 : DEFINE_STUB(spdk_bdev_desc_get_bdev, struct spdk_bdev *, (struct spdk_bdev_desc *desc), NULL);
51 [ # # ]: 0 : DEFINE_STUB(spdk_bdev_module_claim_bdev, int, (struct spdk_bdev *bdev, struct spdk_bdev_desc *desc,
52 : : struct spdk_bdev_module *module), 0);
53 : 0 : DEFINE_STUB_V(spdk_bdev_module_examine_done, (struct spdk_bdev_module *module));
54 [ # # ]: 0 : DEFINE_STUB(spdk_bdev_register, int, (struct spdk_bdev *vbdev), 0);
55 : 0 : DEFINE_STUB_V(spdk_bdev_destruct_done, (struct spdk_bdev *bdev, int bdeverrno));
56 : :
57 [ # # ]: 0 : DEFINE_STUB(spdk_accel_crypto_key_destroy, int, (struct spdk_accel_crypto_key *key), 0);
58 [ - + ]: 10 : DEFINE_STUB(spdk_accel_append_decrypt, int,
59 : : (struct spdk_accel_sequence **seq, struct spdk_io_channel *ch,
60 : : struct spdk_accel_crypto_key *key, struct iovec *dst_iovs,
61 : : uint32_t dst_iovcnt, struct spdk_memory_domain *dst_domain, void *dst_domain_ctx,
62 : : struct iovec *src_iovs, uint32_t src_iovcnt, struct spdk_memory_domain *src_domain,
63 : : void *src_domain_ctx, uint64_t iv, uint32_t block_size,
64 : : spdk_accel_step_cb cb_fn, void *cb_arg), 0);
65 [ - + ]: 12 : DEFINE_STUB(spdk_accel_append_encrypt, int,
66 : : (struct spdk_accel_sequence **seq, struct spdk_io_channel *ch,
67 : : struct spdk_accel_crypto_key *key, struct iovec *dst_iovs,
68 : : uint32_t dst_iovcnt, struct spdk_memory_domain *dst_domain, void *dst_domain_ctx,
69 : : struct iovec *src_iovs, uint32_t src_iovcnt, struct spdk_memory_domain *src_domain,
70 : : void *src_domain_ctx, uint64_t iv, uint32_t block_size,
71 : : spdk_accel_step_cb cb_fn, void *cb_arg), 0);
72 : 14 : DEFINE_STUB_V(spdk_accel_sequence_abort, (struct spdk_accel_sequence *seq));
73 : 12 : DEFINE_STUB_V(spdk_accel_put_buf, (struct spdk_io_channel *ch, void *buf,
74 : : struct spdk_memory_domain *domain, void *domain_ctx));
75 [ # # ]: 0 : DEFINE_STUB(spdk_bdev_get_memory_domains, int,
76 : : (struct spdk_bdev *bdev, struct spdk_memory_domain **domains, int sz), 0);
77 [ # # ]: 0 : DEFINE_STUB(spdk_accel_get_memory_domain, struct spdk_memory_domain *, (void), (void *)0xdeadbeef);
78 [ # # ]: 0 : DEFINE_STUB(spdk_accel_get_buf_align, uint8_t,
79 : : (enum spdk_accel_opcode opcode, const struct spdk_accel_operation_exec_ctx *ctx), 0);
80 [ # # ]: 0 : DEFINE_STUB(spdk_accel_get_opc_memory_domains, int, (enum spdk_accel_opcode opcode,
81 : : struct spdk_memory_domain **domains,
82 : : int array_size), 0);
83 : :
84 : : /* global vars and setup/cleanup functions used for all test functions */
85 : : struct spdk_bdev_io *g_base_io;
86 : : struct crypto_bdev_io *g_io_ctx;
87 : : struct crypto_io_channel *g_crypto_ch;
88 : : struct spdk_io_channel *g_io_ch;
89 : : struct vbdev_crypto g_crypto_bdev;
90 : : struct vbdev_crypto_opts g_crypto_bdev_opts;
91 : :
92 : : int
93 : 12 : spdk_accel_get_buf(struct spdk_io_channel *ch, uint64_t len, void **buf,
94 : : struct spdk_memory_domain **domain, void **domain_ctx)
95 : : {
96 : 12 : *buf = (void *)0xdeadbeef;
97 : 12 : *domain = (void *)0xbeefdead;
98 : :
99 : 12 : return 0;
100 : : }
101 : :
102 : : void
103 : 10 : spdk_bdev_io_get_buf(struct spdk_bdev_io *bdev_io, spdk_bdev_io_get_buf_cb cb, uint64_t len)
104 : : {
105 : 10 : cb(g_io_ch, bdev_io, true);
106 : 10 : }
107 : :
108 : : struct ut_vbdev_crypto_bdev_cpl_args {
109 : : spdk_bdev_io_completion_cb cb_fn;
110 : : struct spdk_bdev_io *bdev_io;
111 : : void *cb_arg;
112 : : bool result;
113 : : };
114 : :
115 : : static void
116 : 14 : _ut_vbdev_crypto_bdev_cpl(void *arg)
117 : : {
118 : 14 : struct ut_vbdev_crypto_bdev_cpl_args *cpl_args = arg;
119 : :
120 [ - + ]: 14 : cpl_args->cb_fn(cpl_args->bdev_io, cpl_args->result, cpl_args->cb_arg);
121 : 14 : free(cpl_args);
122 : 14 : }
123 : :
124 : : static void
125 : 14 : ut_vbdev_crypto_bdev_cpl(spdk_bdev_io_completion_cb cb_fn, struct spdk_bdev_io *bdev_io,
126 : : bool result, void *cb_arg)
127 : : {
128 : 14 : struct ut_vbdev_crypto_bdev_cpl_args *cpl_args = calloc(1, sizeof(*cpl_args));
129 : :
130 [ - + ]: 14 : SPDK_CU_ASSERT_FATAL(cpl_args);
131 : 14 : cpl_args->cb_fn = cb_fn;
132 : 14 : cpl_args->bdev_io = bdev_io;
133 : 14 : cpl_args->result = result;
134 : 14 : cpl_args->cb_arg = cb_arg;
135 : :
136 : 14 : spdk_thread_send_msg(spdk_get_thread(), _ut_vbdev_crypto_bdev_cpl, cpl_args);
137 : 14 : }
138 : :
139 : : /* Mock these functions to call the callback and then return the value we require */
140 [ # # ]: 0 : DEFINE_RETURN_MOCK(spdk_bdev_readv_blocks_ext, int);
141 : : int
142 : 8 : spdk_bdev_readv_blocks_ext(struct spdk_bdev_desc *desc, struct spdk_io_channel *ch,
143 : : struct iovec *iov, int iovcnt,
144 : : uint64_t offset_blocks, uint64_t num_blocks,
145 : : spdk_bdev_io_completion_cb cb, void *cb_arg,
146 : : struct spdk_bdev_ext_io_opts *opts)
147 : : {
148 [ - + - + : 8 : HANDLE_RETURN_MOCK(spdk_bdev_readv_blocks_ext);
+ + ]
149 : 4 : ut_vbdev_crypto_bdev_cpl(cb, g_base_io,
150 : 4 : g_base_io->internal.status == SPDK_BDEV_IO_STATUS_SUCCESS, cb_arg);
151 : 4 : return 0;
152 : : }
153 : :
154 [ # # ]: 0 : DEFINE_RETURN_MOCK(spdk_bdev_writev_blocks_ext, int);
155 : : int
156 : 12 : spdk_bdev_writev_blocks_ext(struct spdk_bdev_desc *desc, struct spdk_io_channel *ch,
157 : : struct iovec *iov, int iovcnt,
158 : : uint64_t offset_blocks, uint64_t num_blocks,
159 : : spdk_bdev_io_completion_cb cb, void *cb_arg,
160 : : struct spdk_bdev_ext_io_opts *opts)
161 : : {
162 [ - + - + : 12 : HANDLE_RETURN_MOCK(spdk_bdev_writev_blocks_ext);
+ + ]
163 : 6 : ut_vbdev_crypto_bdev_cpl(cb, g_base_io,
164 : 6 : g_base_io->internal.status == SPDK_BDEV_IO_STATUS_SUCCESS, cb_arg);
165 : 6 : return 0;
166 : : }
167 : :
168 [ # # ]: 0 : DEFINE_RETURN_MOCK(spdk_bdev_unmap_blocks, int);
169 : : int
170 : 4 : spdk_bdev_unmap_blocks(struct spdk_bdev_desc *desc, struct spdk_io_channel *ch,
171 : : uint64_t offset_blocks, uint64_t num_blocks,
172 : : spdk_bdev_io_completion_cb cb, void *cb_arg)
173 : : {
174 [ - + - + : 4 : HANDLE_RETURN_MOCK(spdk_bdev_unmap_blocks);
+ + ]
175 : 2 : ut_vbdev_crypto_bdev_cpl(cb, g_base_io,
176 : 2 : g_base_io->internal.status == SPDK_BDEV_IO_STATUS_SUCCESS, cb_arg);
177 : :
178 : 2 : return 0;
179 : : }
180 : :
181 [ # # ]: 0 : DEFINE_RETURN_MOCK(spdk_bdev_flush_blocks, int);
182 : : int
183 : 4 : spdk_bdev_flush_blocks(struct spdk_bdev_desc *desc, struct spdk_io_channel *ch,
184 : : uint64_t offset_blocks, uint64_t num_blocks, spdk_bdev_io_completion_cb cb,
185 : : void *cb_arg)
186 : : {
187 [ - + - + : 4 : HANDLE_RETURN_MOCK(spdk_bdev_flush_blocks);
+ + ]
188 : 2 : ut_vbdev_crypto_bdev_cpl(cb, g_base_io,
189 : 2 : g_base_io->internal.status == SPDK_BDEV_IO_STATUS_SUCCESS, cb_arg);
190 : 2 : return 0;
191 : : }
192 : :
193 [ # # ]: 0 : DEFINE_RETURN_MOCK(spdk_bdev_reset, int);
194 : : int
195 : 0 : spdk_bdev_reset(struct spdk_bdev_desc *desc, struct spdk_io_channel *ch,
196 : : spdk_bdev_io_completion_cb cb, void *cb_arg)
197 : : {
198 [ # # # # : 0 : HANDLE_RETURN_MOCK(spdk_bdev_reset);
# # ]
199 : 0 : ut_vbdev_crypto_bdev_cpl(cb, g_base_io,
200 : 0 : g_base_io->internal.status == SPDK_BDEV_IO_STATUS_SUCCESS, cb_arg);
201 : 0 : return 0;
202 : : }
203 : :
204 : : bool g_completion_called = false;
205 : : void
206 : 34 : spdk_bdev_io_complete(struct spdk_bdev_io *bdev_io, enum spdk_bdev_io_status status)
207 : : {
208 : 34 : bdev_io->internal.status = status;
209 : 34 : g_completion_called = true;
210 : 34 : }
211 : :
212 : : void
213 : 16 : spdk_bdev_io_complete_base_io_status(struct spdk_bdev_io *bdev_io,
214 : : const struct spdk_bdev_io *base_io)
215 : : {
216 : 16 : spdk_bdev_io_complete(bdev_io, base_io->internal.status);
217 : 16 : }
218 : :
219 : : struct ut_vbdev_crypto_accel_cpl_args {
220 : : spdk_accel_completion_cb cb_fn;
221 : : void *cb_arg;
222 : : int rc;
223 : : };
224 : :
225 : 0 : struct spdk_io_channel *spdk_accel_get_io_channel(void)
226 : : {
227 : 0 : return (struct spdk_io_channel *)0xfeedbeef;
228 : : }
229 : :
230 : : /* Global setup for all tests that share a bunch of preparation... */
231 : : static int
232 : 2 : test_setup(void)
233 : : {
234 : : /* Prepare essential variables for test routines */
235 : 2 : g_base_io = calloc(1, sizeof(struct spdk_bdev_io) + sizeof(struct crypto_bdev_io));
236 : 2 : g_base_io->u.bdev.iovs = calloc(1, sizeof(struct iovec) * 128);
237 : 2 : g_base_io->bdev = &g_crypto_bdev.crypto_bdev;
238 : 2 : g_io_ch = calloc(1, sizeof(struct spdk_io_channel) + sizeof(struct crypto_io_channel));
239 : 2 : g_crypto_ch = (struct crypto_io_channel *)spdk_io_channel_get_ctx(g_io_ch);
240 : 2 : g_io_ctx = (struct crypto_bdev_io *)g_base_io->driver_ctx;
241 [ - + ]: 2 : memset(&g_crypto_bdev, 0, sizeof(struct vbdev_crypto));
242 [ - + ]: 2 : memset(&g_crypto_bdev_opts, 0, sizeof(struct vbdev_crypto_opts));
243 : 2 : g_crypto_bdev.crypto_bdev.blocklen = 512;
244 : 2 : g_io_ctx->crypto_ch = g_crypto_ch;
245 : 2 : g_io_ctx->crypto_bdev = &g_crypto_bdev;
246 : 2 : g_io_ctx->crypto_bdev->opts = &g_crypto_bdev_opts;
247 : :
248 : 2 : return 0;
249 : : }
250 : :
251 : : /* Global teardown for all tests */
252 : : static int
253 : 2 : test_cleanup(void)
254 : : {
255 : 2 : free(g_base_io->u.bdev.iovs);
256 : 2 : free(g_base_io);
257 : 2 : free(g_io_ch);
258 : 2 : return 0;
259 : : }
260 : :
261 : : struct ut_crypto_io {
262 : : struct spdk_bdev_io bdev_io;
263 : : struct crypto_bdev_io crypto_io;
264 : : };
265 : :
266 : : #define UT_IO_INIT(iov) \
267 : : { \
268 : : .bdev_io = { \
269 : : .bdev = &g_crypto_bdev.crypto_bdev, \
270 : : .u.bdev.iovs = (iov), \
271 : : } \
272 : : }
273 : :
274 : : static void
275 : 2 : test_error_paths(void)
276 : : {
277 : 1 : struct iovec iov;
278 : 2 : struct ut_crypto_io io = UT_IO_INIT(&iov);
279 : 2 : struct spdk_bdev_io *bdev_io = &io.bdev_io;
280 : 2 : struct crypto_bdev_io *crypto_io = &io.crypto_io;
281 : :
282 : 2 : g_crypto_bdev.crypto_bdev.blocklen = 512;
283 : :
284 : 2 : bdev_io->internal.status = SPDK_BDEV_IO_STATUS_PENDING;
285 : 2 : bdev_io->u.bdev.iovcnt = 1;
286 : 2 : bdev_io->u.bdev.num_blocks = 1;
287 : 2 : bdev_io->u.bdev.iovs[0].iov_len = 512;
288 : 2 : bdev_io->u.bdev.iovs[0].iov_base = (void *)0xDEADBEEF;
289 : 2 : bdev_io->type = SPDK_BDEV_IO_TYPE_WRITE;
290 : :
291 : : /* test error returned by accel fw */
292 : 2 : MOCK_SET(spdk_accel_append_encrypt, -ENOMEM);
293 : 2 : g_completion_called = false;
294 : 2 : vbdev_crypto_submit_request(g_io_ch, bdev_io);
295 : 2 : CU_ASSERT(bdev_io->internal.status == SPDK_BDEV_IO_STATUS_NOMEM);
296 [ - + ]: 2 : CU_ASSERT(g_completion_called);
297 : :
298 : 2 : MOCK_SET(spdk_accel_append_encrypt, -EINVAL);
299 : 2 : vbdev_crypto_submit_request(g_io_ch, bdev_io);
300 : 2 : CU_ASSERT(bdev_io->internal.status == SPDK_BDEV_IO_STATUS_FAILED);
301 : 2 : MOCK_SET(spdk_accel_append_encrypt, 0);
302 : :
303 : : /* Test error returned from bdev */
304 : 2 : bdev_io->internal.status = SPDK_BDEV_IO_STATUS_PENDING;
305 : 2 : MOCK_SET(spdk_bdev_writev_blocks_ext, -ENOMEM);
306 : 2 : vbdev_crypto_submit_request(g_io_ch, bdev_io);
307 : 2 : poll_threads();
308 : 2 : CU_ASSERT(bdev_io->internal.status == SPDK_BDEV_IO_STATUS_PENDING);
309 : 2 : CU_ASSERT(crypto_io->bdev_io_wait.bdev == &g_crypto_bdev.crypto_bdev);
310 : 2 : CU_ASSERT(crypto_io->bdev_io_wait.cb_fn == vbdev_crypto_resubmit_io);
311 : 2 : CU_ASSERT(crypto_io->bdev_io_wait.cb_arg == bdev_io);
312 : 2 : CU_ASSERT(crypto_io->resubmit_state == CRYPTO_IO_ENCRYPT_DONE);
313 [ - + ]: 2 : memset(&crypto_io->bdev_io_wait, 0, sizeof(crypto_io->bdev_io_wait));
314 [ - - - + ]: 2 : MOCK_CLEAR(spdk_bdev_readv_blocks_ext);
315 : :
316 : 2 : bdev_io->internal.status = SPDK_BDEV_IO_STATUS_PENDING;
317 : 2 : MOCK_SET(spdk_bdev_writev_blocks_ext, -EINVAL);
318 : 2 : vbdev_crypto_submit_request(g_io_ch, bdev_io);
319 : 2 : poll_threads();
320 : 2 : CU_ASSERT(bdev_io->internal.status == SPDK_BDEV_IO_STATUS_FAILED);
321 [ - - - + ]: 2 : MOCK_CLEAR(spdk_bdev_writev_blocks_ext);
322 : :
323 : : /* Test error returned in bdev cpl */
324 : 2 : bdev_io->internal.status = SPDK_BDEV_IO_STATUS_PENDING;
325 : 2 : g_base_io->internal.status = SPDK_BDEV_IO_STATUS_FAILED;
326 : 2 : vbdev_crypto_submit_request(g_io_ch, bdev_io);
327 : 2 : poll_threads();
328 : 2 : poll_threads();
329 : 2 : CU_ASSERT(bdev_io->internal.status == SPDK_BDEV_IO_STATUS_FAILED);
330 : 2 : g_base_io->internal.status = SPDK_BDEV_IO_STATUS_SUCCESS;
331 : :
332 : : /* the same for read path */
333 : : /* Test error returned from bdev */
334 : 2 : bdev_io->type = SPDK_BDEV_IO_TYPE_READ;
335 : 2 : bdev_io->internal.status = SPDK_BDEV_IO_STATUS_SUCCESS;
336 : :
337 : 2 : MOCK_SET(spdk_bdev_readv_blocks_ext, -ENOMEM);
338 : 2 : vbdev_crypto_submit_request(g_io_ch, bdev_io);
339 : 2 : CU_ASSERT(bdev_io->internal.status == SPDK_BDEV_IO_STATUS_SUCCESS);
340 : 2 : CU_ASSERT(crypto_io->bdev_io_wait.bdev == &g_crypto_bdev.crypto_bdev);
341 : 2 : CU_ASSERT(crypto_io->bdev_io_wait.cb_fn == vbdev_crypto_resubmit_io);
342 : 2 : CU_ASSERT(crypto_io->bdev_io_wait.cb_arg == bdev_io);
343 : 2 : CU_ASSERT(crypto_io->resubmit_state == CRYPTO_IO_DECRYPT_DONE);
344 [ - + ]: 2 : memset(&crypto_io->bdev_io_wait, 0, sizeof(crypto_io->bdev_io_wait));
345 [ - - - + ]: 2 : MOCK_CLEAR(spdk_bdev_readv_blocks_ext);
346 : :
347 : 2 : bdev_io->internal.status = SPDK_BDEV_IO_STATUS_PENDING;
348 : 2 : MOCK_SET(spdk_bdev_readv_blocks_ext, -EINVAL);
349 : 2 : vbdev_crypto_submit_request(g_io_ch, bdev_io);
350 : 2 : CU_ASSERT(bdev_io->internal.status == SPDK_BDEV_IO_STATUS_FAILED);
351 [ - - - + ]: 2 : MOCK_CLEAR(spdk_bdev_readv_blocks_ext);
352 : :
353 : : /* Test error returned in bdev cpl */
354 : 2 : bdev_io->internal.status = SPDK_BDEV_IO_STATUS_PENDING;
355 : 2 : g_base_io->internal.status = SPDK_BDEV_IO_STATUS_FAILED;
356 : 2 : vbdev_crypto_submit_request(g_io_ch, bdev_io);
357 : 2 : poll_threads();
358 : 2 : CU_ASSERT(bdev_io->internal.status == SPDK_BDEV_IO_STATUS_FAILED);
359 : 2 : g_base_io->internal.status = SPDK_BDEV_IO_STATUS_SUCCESS;
360 : :
361 : : /* test error returned by accel fw */
362 : 2 : bdev_io->internal.status = SPDK_BDEV_IO_STATUS_PENDING;
363 : 2 : MOCK_SET(spdk_accel_append_decrypt, -ENOMEM);
364 : 2 : g_completion_called = false;
365 : 2 : vbdev_crypto_submit_request(g_io_ch, bdev_io);
366 : 2 : CU_ASSERT(bdev_io->internal.status == SPDK_BDEV_IO_STATUS_NOMEM);
367 [ - + ]: 2 : CU_ASSERT(g_completion_called);
368 : 2 : MOCK_SET(spdk_accel_append_decrypt, 0);
369 : 2 : g_completion_called = false;
370 : 2 : }
371 : :
372 : : static void
373 : 2 : test_simple_write(void)
374 : : {
375 : 1 : struct iovec iov;
376 : 2 : struct ut_crypto_io io = UT_IO_INIT(&iov);
377 : 2 : struct spdk_bdev_io *bdev_io = &io.bdev_io;
378 : 2 : struct crypto_bdev_io *crypto_io = &io.crypto_io;
379 : :
380 : : /* Single element block size write */
381 : 2 : bdev_io->internal.status = SPDK_BDEV_IO_STATUS_PENDING;
382 : 2 : bdev_io->u.bdev.iovcnt = 1;
383 : 2 : bdev_io->u.bdev.num_blocks = 1;
384 : 2 : bdev_io->u.bdev.offset_blocks = 0;
385 : 2 : bdev_io->u.bdev.iovs[0].iov_len = 512;
386 : 2 : bdev_io->u.bdev.iovs[0].iov_base = &test_simple_write;
387 : 2 : g_crypto_bdev.crypto_bdev.blocklen = 512;
388 : 2 : bdev_io->type = SPDK_BDEV_IO_TYPE_WRITE;
389 : :
390 : 2 : vbdev_crypto_submit_request(g_io_ch, bdev_io);
391 : : /* 1st poll to trigger accel completions, 2nd for bdev */
392 : 2 : poll_threads();
393 : 2 : poll_threads();
394 : 2 : CU_ASSERT(bdev_io->internal.status == SPDK_BDEV_IO_STATUS_SUCCESS);
395 : 2 : CU_ASSERT(crypto_io->aux_buf_iov.iov_len == 512);
396 : 2 : CU_ASSERT(crypto_io->aux_buf_iov.iov_base != NULL);
397 : 2 : CU_ASSERT(crypto_io->aux_offset_blocks == 0);
398 : 2 : CU_ASSERT(crypto_io->aux_num_blocks == 1);
399 : 2 : }
400 : :
401 : : static void
402 : 2 : test_simple_read(void)
403 : : {
404 : 1 : struct iovec iov;
405 : 2 : struct ut_crypto_io io = UT_IO_INIT(&iov);
406 : 2 : struct spdk_bdev_io *bdev_io = &io.bdev_io;
407 : :
408 : : /* Single element block size read */
409 : 2 : bdev_io->internal.status = SPDK_BDEV_IO_STATUS_PENDING;
410 : 2 : bdev_io->u.bdev.iovcnt = 1;
411 : 2 : bdev_io->u.bdev.num_blocks = 1;
412 : 2 : bdev_io->u.bdev.iovs[0].iov_len = 512;
413 : 2 : bdev_io->u.bdev.iovs[0].iov_base = &test_simple_read;
414 : 2 : g_crypto_bdev.crypto_bdev.blocklen = 512;
415 : 2 : bdev_io->type = SPDK_BDEV_IO_TYPE_READ;
416 : :
417 : 2 : vbdev_crypto_submit_request(g_io_ch, bdev_io);
418 : : /* 1st poll to trigger dev completions, 2nd for accel */
419 : 2 : poll_threads();
420 : 2 : poll_threads();
421 : 2 : CU_ASSERT(bdev_io->internal.status == SPDK_BDEV_IO_STATUS_SUCCESS);
422 : 2 : }
423 : :
424 : : static void
425 : 2 : test_passthru(void)
426 : : {
427 : 2 : struct ut_crypto_io io = UT_IO_INIT(NULL);
428 : 2 : struct spdk_bdev_io *bdev_io = &io.bdev_io;
429 : :
430 : : /* Make sure these follow our completion callback, test success & fail. */
431 : 2 : bdev_io->type = SPDK_BDEV_IO_TYPE_UNMAP;
432 [ - - - + ]: 2 : MOCK_CLEAR(spdk_bdev_unmap_blocks);
433 : 2 : vbdev_crypto_submit_request(g_io_ch, bdev_io);
434 : 2 : poll_threads();
435 : 2 : CU_ASSERT(bdev_io->internal.status == SPDK_BDEV_IO_STATUS_SUCCESS);
436 : 2 : MOCK_SET(spdk_bdev_unmap_blocks, -EINVAL);
437 : 2 : vbdev_crypto_submit_request(g_io_ch, bdev_io);
438 : 2 : CU_ASSERT(bdev_io->internal.status == SPDK_BDEV_IO_STATUS_FAILED);
439 [ - - - + ]: 2 : MOCK_CLEAR(spdk_bdev_unmap_blocks);
440 : :
441 : 2 : bdev_io->type = SPDK_BDEV_IO_TYPE_FLUSH;
442 [ - - - + ]: 2 : MOCK_CLEAR(spdk_bdev_flush_blocks);
443 : 2 : vbdev_crypto_submit_request(g_io_ch, bdev_io);
444 : 2 : poll_threads();
445 : 2 : CU_ASSERT(bdev_io->internal.status == SPDK_BDEV_IO_STATUS_SUCCESS);
446 : 2 : MOCK_SET(spdk_bdev_flush_blocks, -EINVAL);
447 : 2 : vbdev_crypto_submit_request(g_io_ch, bdev_io);
448 : 2 : CU_ASSERT(bdev_io->internal.status == SPDK_BDEV_IO_STATUS_FAILED);
449 [ - - - + ]: 2 : MOCK_CLEAR(spdk_bdev_flush_blocks);
450 : :
451 : : /* We should never get a WZ command, we report that we don't support it. */
452 : 2 : bdev_io->type = SPDK_BDEV_IO_TYPE_WRITE_ZEROES;
453 : 2 : vbdev_crypto_submit_request(g_io_ch, bdev_io);
454 : 2 : CU_ASSERT(bdev_io->internal.status == SPDK_BDEV_IO_STATUS_FAILED);
455 : 2 : }
456 : :
457 : : static void
458 : 2 : test_reset(void)
459 : : {
460 : : /* TODO: There are a few different ways to do this given that
461 : : * the code uses spdk_for_each_channel() to implement reset
462 : : * handling. Submitting w/o UT for this function for now and
463 : : * will follow up with something shortly.
464 : : */
465 : 2 : }
466 : :
467 : : static void
468 : 2 : test_crypto_op_complete(void)
469 : : {
470 : 2 : struct ut_crypto_io io = UT_IO_INIT(NULL);
471 : 2 : struct spdk_bdev_io *bdev_io = &io.bdev_io;
472 : :
473 : : /* Test read completion. */
474 : 2 : g_base_io->internal.status = SPDK_BDEV_IO_STATUS_SUCCESS;
475 : 2 : bdev_io->internal.status = SPDK_BDEV_IO_STATUS_PENDING;
476 : 2 : bdev_io->type = SPDK_BDEV_IO_TYPE_READ;
477 : 2 : g_completion_called = false;
478 : 2 : _complete_internal_io(g_base_io, true, bdev_io);
479 : 2 : CU_ASSERT(g_base_io->internal.status == SPDK_BDEV_IO_STATUS_SUCCESS);
480 [ - + ]: 2 : CU_ASSERT(g_completion_called == true);
481 : :
482 : : /* Test write completion success. */
483 : 2 : g_base_io->internal.status = SPDK_BDEV_IO_STATUS_SUCCESS;
484 : 2 : bdev_io->internal.status = SPDK_BDEV_IO_STATUS_PENDING;
485 : 2 : bdev_io->type = SPDK_BDEV_IO_TYPE_WRITE;
486 : 2 : g_completion_called = false;
487 [ - - - + ]: 2 : MOCK_CLEAR(spdk_bdev_writev_blocks_ext);
488 : 2 : crypto_write(g_crypto_ch, bdev_io);
489 : 2 : poll_threads();
490 : 2 : CU_ASSERT(bdev_io->internal.status == SPDK_BDEV_IO_STATUS_SUCCESS);
491 [ - + ]: 2 : CU_ASSERT(g_completion_called == true);
492 : :
493 : : /* Test write completion failed. */
494 : 2 : bdev_io->internal.status = SPDK_BDEV_IO_STATUS_PENDING;
495 : 2 : bdev_io->type = SPDK_BDEV_IO_TYPE_WRITE;
496 : 2 : g_completion_called = false;
497 : 2 : MOCK_SET(spdk_bdev_writev_blocks_ext, -EINVAL);
498 : 2 : crypto_write(g_crypto_ch, bdev_io);
499 : 2 : CU_ASSERT(bdev_io->internal.status == SPDK_BDEV_IO_STATUS_FAILED);
500 [ - + ]: 2 : CU_ASSERT(g_completion_called == true);
501 [ - - - + ]: 2 : MOCK_CLEAR(spdk_bdev_writev_blocks_ext);
502 : 2 : }
503 : :
504 : : static void
505 : 2 : test_supported_io(void)
506 : : {
507 : 2 : void *ctx = NULL;
508 : 2 : bool rc = true;
509 : :
510 : : /* Make sure we always report false to WZ, we need the bdev layer to
511 : : * send real 0's so we can encrypt/decrypt them.
512 : : */
513 : 2 : rc = vbdev_crypto_io_type_supported(ctx, SPDK_BDEV_IO_TYPE_WRITE_ZEROES);
514 : 2 : CU_ASSERT(rc == false);
515 : 2 : }
516 : : int
517 : 2 : main(int argc, char **argv)
518 : : {
519 : 2 : CU_pSuite suite = NULL;
520 : : unsigned int num_failures;
521 : :
522 : 2 : CU_initialize_registry();
523 : :
524 : 2 : suite = CU_add_suite("crypto", test_setup, test_cleanup);
525 : 2 : CU_ADD_TEST(suite, test_error_paths);
526 : 2 : CU_ADD_TEST(suite, test_simple_write);
527 : 2 : CU_ADD_TEST(suite, test_simple_read);
528 : 2 : CU_ADD_TEST(suite, test_passthru);
529 : 2 : CU_ADD_TEST(suite, test_crypto_op_complete);
530 : 2 : CU_ADD_TEST(suite, test_supported_io);
531 : 2 : CU_ADD_TEST(suite, test_reset);
532 : :
533 : 2 : allocate_threads(1);
534 : 2 : set_thread(0);
535 : :
536 : 2 : num_failures = spdk_ut_run_tests(argc, argv, NULL);
537 : :
538 : 2 : free_threads();
539 : :
540 : 2 : CU_cleanup_registry();
541 : 2 : return num_failures;
542 : : }
|