xref: /dpdk/drivers/compress/isal/isal_compress_pmd.c (revision e977e4199a8d6bab72cf94e154adcad1fb964e5e)
1 /* SPDX-License-Identifier: BSD-3-Clause
2  * Copyright(c) 2018 Intel Corporation
3  */
4 #include <isa-l.h>
5 
6 #include <rte_bus_vdev.h>
7 #include <rte_common.h>
8 #include <rte_malloc.h>
9 #include <rte_mbuf.h>
10 #include <rte_compressdev_pmd.h>
11 
12 #include "isal_compress_pmd_private.h"
13 
14 #define RTE_COMP_ISAL_WINDOW_SIZE 15
15 #define RTE_COMP_ISAL_LEVEL_ZERO 0 /* ISA-L Level 0 used for fixed Huffman */
16 #define RTE_COMP_ISAL_LEVEL_ONE 1
17 #define RTE_COMP_ISAL_LEVEL_TWO 2
18 #define RTE_COMP_ISAL_LEVEL_THREE 3 /* Optimised for AVX512 & AVX2 only */
19 
20 int isal_logtype_driver;
21 
22 /* Verify and set private xform parameters */
23 int
24 isal_comp_set_priv_xform_parameters(struct isal_priv_xform *priv_xform,
25 		const struct rte_comp_xform *xform)
26 {
27 	if (xform == NULL)
28 		return -EINVAL;
29 
30 	/* Set compression private xform variables */
31 	if (xform->type == RTE_COMP_COMPRESS) {
32 		/* Set private xform type - COMPRESS/DECOMPRESS */
33 		priv_xform->type = RTE_COMP_COMPRESS;
34 
35 		/* Set private xform algorithm */
36 		if (xform->compress.algo != RTE_COMP_ALGO_DEFLATE) {
37 			if (xform->compress.algo == RTE_COMP_ALGO_NULL) {
38 				ISAL_PMD_LOG(ERR, "By-pass not supported\n");
39 				return -ENOTSUP;
40 			}
41 			ISAL_PMD_LOG(ERR, "Algorithm not supported\n");
42 			return -ENOTSUP;
43 		}
44 		priv_xform->compress.algo = RTE_COMP_ALGO_DEFLATE;
45 
46 		/* Set private xform checksum - raw deflate by default */
47 		if (xform->compress.chksum != RTE_COMP_CHECKSUM_NONE) {
48 			ISAL_PMD_LOG(ERR, "Checksum not supported\n");
49 			return -ENOTSUP;
50 		}
51 
52 		/* Set private xform window size, 32K supported */
53 		if (xform->compress.window_size == RTE_COMP_ISAL_WINDOW_SIZE)
54 			priv_xform->compress.window_size =
55 					RTE_COMP_ISAL_WINDOW_SIZE;
56 		else {
57 			ISAL_PMD_LOG(ERR, "Window size not supported\n");
58 			return -ENOTSUP;
59 		}
60 
61 		/* Set private xform huffman type */
62 		switch (xform->compress.deflate.huffman) {
63 		case(RTE_COMP_HUFFMAN_DEFAULT):
64 			priv_xform->compress.deflate.huffman =
65 					RTE_COMP_HUFFMAN_DEFAULT;
66 			break;
67 		case(RTE_COMP_HUFFMAN_FIXED):
68 			priv_xform->compress.deflate.huffman =
69 					RTE_COMP_HUFFMAN_FIXED;
70 			break;
71 		case(RTE_COMP_HUFFMAN_DYNAMIC):
72 			priv_xform->compress.deflate.huffman =
73 					RTE_COMP_HUFFMAN_DYNAMIC;
74 			break;
75 		default:
76 			ISAL_PMD_LOG(ERR, "Huffman code not supported\n");
77 			return -ENOTSUP;
78 		}
79 
80 		/* Set private xform level.
81 		 * Checking compliance with compressdev API, -1 <= level => 9
82 		 */
83 		if (xform->compress.level < RTE_COMP_LEVEL_PMD_DEFAULT ||
84 				xform->compress.level > RTE_COMP_LEVEL_MAX) {
85 			ISAL_PMD_LOG(ERR, "Compression level out of range\n");
86 			return -EINVAL;
87 		}
88 		/* Check for Compressdev API level 0, No compression
89 		 * not supported in ISA-L
90 		 */
91 		else if (xform->compress.level == RTE_COMP_LEVEL_NONE) {
92 			ISAL_PMD_LOG(ERR, "No Compression not supported\n");
93 			return -ENOTSUP;
94 		}
95 		/* If using fixed huffman code, level must be 0 */
96 		else if (priv_xform->compress.deflate.huffman ==
97 				RTE_COMP_HUFFMAN_FIXED) {
98 			ISAL_PMD_LOG(DEBUG, "ISA-L level 0 used due to a"
99 					" fixed huffman code\n");
100 			priv_xform->compress.level = RTE_COMP_ISAL_LEVEL_ZERO;
101 			priv_xform->level_buffer_size =
102 					ISAL_DEF_LVL0_DEFAULT;
103 		} else {
104 			/* Mapping API levels to ISA-L levels 1,2 & 3 */
105 			switch (xform->compress.level) {
106 			case RTE_COMP_LEVEL_PMD_DEFAULT:
107 				/* Default is 1 if not using fixed huffman */
108 				priv_xform->compress.level =
109 						RTE_COMP_ISAL_LEVEL_ONE;
110 				priv_xform->level_buffer_size =
111 						ISAL_DEF_LVL1_DEFAULT;
112 				break;
113 			case RTE_COMP_LEVEL_MIN:
114 				priv_xform->compress.level =
115 						RTE_COMP_ISAL_LEVEL_ONE;
116 				priv_xform->level_buffer_size =
117 						ISAL_DEF_LVL1_DEFAULT;
118 				break;
119 			case RTE_COMP_ISAL_LEVEL_TWO:
120 				priv_xform->compress.level =
121 						RTE_COMP_ISAL_LEVEL_TWO;
122 				priv_xform->level_buffer_size =
123 						ISAL_DEF_LVL2_DEFAULT;
124 				break;
125 			/* Level 3 or higher requested */
126 			default:
127 				/* Check for AVX512, to use ISA-L level 3 */
128 				if (rte_cpu_get_flag_enabled(
129 						RTE_CPUFLAG_AVX512F)) {
130 					priv_xform->compress.level =
131 						RTE_COMP_ISAL_LEVEL_THREE;
132 					priv_xform->level_buffer_size =
133 						ISAL_DEF_LVL3_DEFAULT;
134 				}
135 				/* Check for AVX2, to use ISA-L level 3 */
136 				else if (rte_cpu_get_flag_enabled(
137 						RTE_CPUFLAG_AVX2)) {
138 					priv_xform->compress.level =
139 						RTE_COMP_ISAL_LEVEL_THREE;
140 					priv_xform->level_buffer_size =
141 						ISAL_DEF_LVL3_DEFAULT;
142 				} else {
143 					ISAL_PMD_LOG(DEBUG, "Requested ISA-L level"
144 						" 3 or above; Level 3 optimized"
145 						" for AVX512 & AVX2 only."
146 						" level changed to 2.\n");
147 					priv_xform->compress.level =
148 						RTE_COMP_ISAL_LEVEL_TWO;
149 					priv_xform->level_buffer_size =
150 						ISAL_DEF_LVL2_DEFAULT;
151 				}
152 			}
153 		}
154 	}
155 
156 	/* Set decompression private xform variables */
157 	else if (xform->type == RTE_COMP_DECOMPRESS) {
158 
159 		/* Set private xform type - COMPRESS/DECOMPRESS */
160 		priv_xform->type = RTE_COMP_DECOMPRESS;
161 
162 		/* Set private xform algorithm */
163 		if (xform->decompress.algo != RTE_COMP_ALGO_DEFLATE) {
164 			if (xform->decompress.algo == RTE_COMP_ALGO_NULL) {
165 				ISAL_PMD_LOG(ERR, "By pass not supported\n");
166 				return -ENOTSUP;
167 			}
168 			ISAL_PMD_LOG(ERR, "Algorithm not supported\n");
169 			return -ENOTSUP;
170 		}
171 		priv_xform->decompress.algo = RTE_COMP_ALGO_DEFLATE;
172 
173 		/* Set private xform checksum - raw deflate by default */
174 		if (xform->compress.chksum != RTE_COMP_CHECKSUM_NONE) {
175 			ISAL_PMD_LOG(ERR, "Checksum not supported\n");
176 			return -ENOTSUP;
177 		}
178 
179 		/* Set private xform window size, 32K supported */
180 		if (xform->decompress.window_size == RTE_COMP_ISAL_WINDOW_SIZE)
181 			priv_xform->decompress.window_size =
182 					RTE_COMP_ISAL_WINDOW_SIZE;
183 		else {
184 			ISAL_PMD_LOG(ERR, "Window size not supported\n");
185 			return -ENOTSUP;
186 		}
187 	}
188 	return 0;
189 }
190 
191 /* Stateless Compression Function */
192 static int
193 process_isal_deflate(struct rte_comp_op *op, struct isal_comp_qp *qp,
194 		struct isal_priv_xform *priv_xform)
195 {
196 	int ret = 0;
197 	op->status = RTE_COMP_OP_STATUS_SUCCESS;
198 
199 	/* Required due to init clearing level_buf */
200 	uint8_t *temp_level_buf = qp->stream->level_buf;
201 
202 	/* Initialize compression stream */
203 	isal_deflate_stateless_init(qp->stream);
204 
205 	qp->stream->level_buf = temp_level_buf;
206 
207 	/* Stateless operation, input will be consumed in one go */
208 	qp->stream->flush = NO_FLUSH;
209 
210 	/* set op level & intermediate level buffer */
211 	qp->stream->level = priv_xform->compress.level;
212 	qp->stream->level_buf_size = priv_xform->level_buffer_size;
213 
214 	/* Point compression stream structure to input/output buffers */
215 	qp->stream->avail_in = op->src.length;
216 	qp->stream->next_in = rte_pktmbuf_mtod(op->m_src, uint8_t *);
217 	qp->stream->avail_out = op->m_dst->data_len;
218 	qp->stream->next_out  = rte_pktmbuf_mtod(op->m_dst, uint8_t *);
219 	qp->stream->end_of_stream = 1; /* All input consumed in one go */
220 
221 	if (unlikely(!qp->stream->next_in || !qp->stream->next_out)) {
222 		ISAL_PMD_LOG(ERR, "Invalid source or destination buffers\n");
223 		op->status = RTE_COMP_OP_STATUS_INVALID_ARGS;
224 		return -1;
225 	}
226 
227 	/* Set op huffman code */
228 	if (priv_xform->compress.deflate.huffman == RTE_COMP_HUFFMAN_FIXED)
229 		isal_deflate_set_hufftables(qp->stream, NULL,
230 				IGZIP_HUFFTABLE_STATIC);
231 	else if (priv_xform->compress.deflate.huffman ==
232 			RTE_COMP_HUFFMAN_DEFAULT)
233 		isal_deflate_set_hufftables(qp->stream, NULL,
234 			IGZIP_HUFFTABLE_DEFAULT);
235 	/* Dynamically change the huffman code to suit the input data */
236 	else if (priv_xform->compress.deflate.huffman ==
237 			RTE_COMP_HUFFMAN_DYNAMIC)
238 		isal_deflate_set_hufftables(qp->stream, NULL,
239 				IGZIP_HUFFTABLE_DEFAULT);
240 
241 	/* Execute compression operation */
242 	ret =  isal_deflate_stateless(qp->stream);
243 
244 	/* Check that output buffer did not run out of space */
245 	if (ret == STATELESS_OVERFLOW) {
246 		ISAL_PMD_LOG(ERR, "Output buffer not big enough\n");
247 		op->status = RTE_COMP_OP_STATUS_OUT_OF_SPACE_TERMINATED;
248 		return ret;
249 	}
250 
251 	/* Check that input buffer has been fully consumed */
252 	if (qp->stream->avail_in != (uint32_t)0) {
253 		ISAL_PMD_LOG(ERR, "Input buffer could not be read entirely\n");
254 		op->status = RTE_COMP_OP_STATUS_ERROR;
255 		return -1;
256 	}
257 
258 	if (ret != COMP_OK) {
259 		op->status = RTE_COMP_OP_STATUS_ERROR;
260 		return ret;
261 	}
262 
263 	op->consumed = qp->stream->total_in;
264 	op->produced = qp->stream->total_out;
265 
266 	return ret;
267 }
268 
269 /* Stateless Decompression Function */
270 static int
271 process_isal_inflate(struct rte_comp_op *op, struct isal_comp_qp *qp)
272 {
273 	int ret = 0;
274 
275 	op->status = RTE_COMP_OP_STATUS_SUCCESS;
276 
277 	/* Initialize decompression state */
278 	isal_inflate_init(qp->state);
279 
280 	/* Point decompression state structure to input/output buffers */
281 	qp->state->avail_in = op->src.length;
282 	qp->state->next_in = rte_pktmbuf_mtod(op->m_src, uint8_t *);
283 	qp->state->avail_out = op->m_dst->data_len;
284 	qp->state->next_out  = rte_pktmbuf_mtod(op->m_dst, uint8_t *);
285 
286 	if (unlikely(!qp->state->next_in || !qp->state->next_out)) {
287 		ISAL_PMD_LOG(ERR, "Invalid source or destination buffers\n");
288 		op->status = RTE_COMP_OP_STATUS_INVALID_ARGS;
289 		return -1;
290 	}
291 
292 	/* Execute decompression operation */
293 	ret = isal_inflate_stateless(qp->state);
294 
295 	if (ret == ISAL_OUT_OVERFLOW) {
296 		ISAL_PMD_LOG(ERR, "Output buffer not big enough\n");
297 		op->status = RTE_COMP_OP_STATUS_OUT_OF_SPACE_TERMINATED;
298 		return ret;
299 	}
300 
301 	/* Check that input buffer has been fully consumed */
302 	if (qp->state->avail_in != (uint32_t)0) {
303 		ISAL_PMD_LOG(ERR, "Input buffer could not be read entirely\n");
304 		op->status = RTE_COMP_OP_STATUS_ERROR;
305 		return -1;
306 	}
307 
308 	if (ret != ISAL_DECOMP_OK) {
309 		op->status = RTE_COMP_OP_STATUS_ERROR;
310 		return ret;
311 	}
312 
313 	op->consumed = op->src.length - qp->state->avail_in;
314 	op->produced = qp->state->total_out;
315 
316 return ret;
317 }
318 
319 /* Process compression/decompression operation */
320 static int
321 process_op(struct isal_comp_qp *qp, struct rte_comp_op *op,
322 		struct isal_priv_xform *priv_xform)
323 {
324 	switch (priv_xform->type) {
325 	case RTE_COMP_COMPRESS:
326 		process_isal_deflate(op, qp, priv_xform);
327 		break;
328 	case RTE_COMP_DECOMPRESS:
329 		process_isal_inflate(op, qp);
330 		break;
331 	default:
332 		ISAL_PMD_LOG(ERR, "Operation Not Supported\n");
333 		return -ENOTSUP;
334 	}
335 	return 0;
336 }
337 
338 /* Enqueue burst */
339 static uint16_t
340 isal_comp_pmd_enqueue_burst(void *queue_pair, struct rte_comp_op **ops,
341 			uint16_t nb_ops)
342 {
343 	struct isal_comp_qp *qp = queue_pair;
344 	uint16_t i;
345 	int retval;
346 	int16_t num_enq = RTE_MIN(qp->num_free_elements, nb_ops);
347 
348 	for (i = 0; i < num_enq; i++) {
349 		if (unlikely(ops[i]->op_type != RTE_COMP_OP_STATELESS)) {
350 			ops[i]->status = RTE_COMP_OP_STATUS_INVALID_ARGS;
351 			ISAL_PMD_LOG(ERR, "Stateful operation not Supported\n");
352 			qp->qp_stats.enqueue_err_count++;
353 			continue;
354 		}
355 		retval = process_op(qp, ops[i], ops[i]->private_xform);
356 		if (unlikely(retval < 0) ||
357 				ops[i]->status != RTE_COMP_OP_STATUS_SUCCESS) {
358 			qp->qp_stats.enqueue_err_count++;
359 		}
360 	}
361 
362 	retval = rte_ring_enqueue_burst(qp->processed_pkts, (void *)ops,
363 			num_enq, NULL);
364 	qp->num_free_elements -= retval;
365 	qp->qp_stats.enqueued_count += retval;
366 
367 	return retval;
368 }
369 
370 /* Dequeue burst */
371 static uint16_t
372 isal_comp_pmd_dequeue_burst(void *queue_pair, struct rte_comp_op **ops,
373 		uint16_t nb_ops)
374 {
375 	struct isal_comp_qp *qp = queue_pair;
376 	uint16_t nb_dequeued;
377 
378 	nb_dequeued = rte_ring_dequeue_burst(qp->processed_pkts, (void **)ops,
379 			nb_ops, NULL);
380 	qp->num_free_elements += nb_dequeued;
381 	qp->qp_stats.dequeued_count += nb_dequeued;
382 
383 	return nb_dequeued;
384 }
385 
386 /* Create ISA-L compression device */
387 static int
388 compdev_isal_create(const char *name, struct rte_vdev_device *vdev,
389 		struct rte_compressdev_pmd_init_params *init_params)
390 {
391 	struct rte_compressdev *dev;
392 
393 	dev = rte_compressdev_pmd_create(name, &vdev->device,
394 			sizeof(struct isal_comp_private), init_params);
395 	if (dev == NULL) {
396 		ISAL_PMD_LOG(ERR, "failed to create compressdev vdev");
397 		return -EFAULT;
398 	}
399 
400 	dev->dev_ops = isal_compress_pmd_ops;
401 
402 	/* register rx/tx burst functions for data path */
403 	dev->dequeue_burst = isal_comp_pmd_dequeue_burst;
404 	dev->enqueue_burst = isal_comp_pmd_enqueue_burst;
405 
406 	return 0;
407 }
408 
409 /** Remove compression device */
410 static int
411 compdev_isal_remove_dev(struct rte_vdev_device *vdev)
412 {
413 	struct rte_compressdev *compdev;
414 	const char *name;
415 
416 	name = rte_vdev_device_name(vdev);
417 	if (name == NULL)
418 		return -EINVAL;
419 
420 	compdev = rte_compressdev_pmd_get_named_dev(name);
421 	if (compdev == NULL)
422 		return -ENODEV;
423 
424 	return rte_compressdev_pmd_destroy(compdev);
425 }
426 
427 /** Initialise ISA-L compression device */
428 static int
429 compdev_isal_probe(struct rte_vdev_device *dev)
430 {
431 	struct rte_compressdev_pmd_init_params init_params = {
432 		"",
433 		rte_socket_id(),
434 	};
435 	const char *name, *args;
436 	int retval;
437 
438 	name = rte_vdev_device_name(dev);
439 	if (name == NULL)
440 		return -EINVAL;
441 
442 	args = rte_vdev_device_args(dev);
443 
444 	retval = rte_compressdev_pmd_parse_input_args(&init_params, args);
445 	if (retval) {
446 		ISAL_PMD_LOG(ERR,
447 			"Failed to parse initialisation arguments[%s]\n", args);
448 		return -EINVAL;
449 	}
450 
451 	return compdev_isal_create(name, dev, &init_params);
452 }
453 
454 static struct rte_vdev_driver compdev_isal_pmd_drv = {
455 	.probe = compdev_isal_probe,
456 	.remove = compdev_isal_remove_dev,
457 };
458 
459 RTE_PMD_REGISTER_VDEV(COMPDEV_NAME_ISAL_PMD, compdev_isal_pmd_drv);
460 RTE_PMD_REGISTER_PARAM_STRING(COMPDEV_NAME_ISAL_PMD,
461 	"socket_id=<int>");
462 
463 RTE_INIT(isal_init_log);
464 
465 static void
466 isal_init_log(void)
467 {
468 	isal_logtype_driver = rte_log_register("comp_isal");
469 	if (isal_logtype_driver >= 0)
470 		rte_log_set_level(isal_logtype_driver, RTE_LOG_INFO);
471 }
472