drivers/compress/isal/isal_compress_pmd.c

   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 /* Process compression/decompression operation */
 270 static int
 271 process_op(struct isal_comp_qp *qp, struct rte_comp_op *op,
 272                 struct isal_priv_xform *priv_xform)
 273 {
 274         switch (priv_xform->type) {
 275         case RTE_COMP_COMPRESS:
 276                 process_isal_deflate(op, qp, priv_xform);
 277                 break;
 278         case RTE_COMP_DECOMPRESS:
 279                 break;
 280         default:
 281                 ISAL_PMD_LOG(ERR, "Operation Not Supported\n");
 282                 return -ENOTSUP;
 283         }
 284         return 0;
 285 }
 286
 287 /* Enqueue burst */
 288 static uint16_t
 289 isal_comp_pmd_enqueue_burst(void *queue_pair, struct rte_comp_op **ops,
 290                         uint16_t nb_ops)
 291 {
 292         struct isal_comp_qp *qp = queue_pair;
 293         uint16_t i;
 294         int retval;
 295         int16_t num_enq = RTE_MIN(qp->num_free_elements, nb_ops);
 296
 297         for (i = 0; i < num_enq; i++) {
 298                 if (unlikely(ops[i]->op_type != RTE_COMP_OP_STATELESS)) {
 299                         ops[i]->status = RTE_COMP_OP_STATUS_INVALID_ARGS;
 300                         ISAL_PMD_LOG(ERR, "Stateful operation not Supported\n");
 301                         qp->qp_stats.enqueue_err_count++;
 302                         continue;
 303                 }
 304                 retval = process_op(qp, ops[i], ops[i]->private_xform);
 305                 if (unlikely(retval < 0) ||
 306                                 ops[i]->status != RTE_COMP_OP_STATUS_SUCCESS) {
 307                         qp->qp_stats.enqueue_err_count++;
 308                 }
 309         }
 310
 311         retval = rte_ring_enqueue_burst(qp->processed_pkts, (void *)ops,
 312                         num_enq, NULL);
 313         qp->num_free_elements -= retval;
 314         qp->qp_stats.enqueued_count += retval;
 315
 316         return retval;
 317 }
 318
 319 /* Dequeue burst */
 320 static uint16_t
 321 isal_comp_pmd_dequeue_burst(void *queue_pair, struct rte_comp_op **ops,
 322                 uint16_t nb_ops)
 323 {
 324         struct isal_comp_qp *qp = queue_pair;
 325         uint16_t nb_dequeued;
 326
 327         nb_dequeued = rte_ring_dequeue_burst(qp->processed_pkts, (void **)ops,
 328                         nb_ops, NULL);
 329         qp->num_free_elements += nb_dequeued;
 330         qp->qp_stats.dequeued_count += nb_dequeued;
 331
 332         return nb_dequeued;
 333 }
 334
 335 /* Create ISA-L compression device */
 336 static int
 337 compdev_isal_create(const char *name, struct rte_vdev_device *vdev,
 338                 struct rte_compressdev_pmd_init_params *init_params)
 339 {
 340         struct rte_compressdev *dev;
 341
 342         dev = rte_compressdev_pmd_create(name, &vdev->device,
 343                         sizeof(struct isal_comp_private), init_params);
 344         if (dev == NULL) {
 345                 ISAL_PMD_LOG(ERR, "failed to create compressdev vdev");
 346                 return -EFAULT;
 347         }
 348
 349         dev->dev_ops = isal_compress_pmd_ops;
 350
 351         /* register rx/tx burst functions for data path */
 352         dev->dequeue_burst = isal_comp_pmd_dequeue_burst;
 353         dev->enqueue_burst = isal_comp_pmd_enqueue_burst;
 354
 355         return 0;
 356 }
 357
 358 /** Remove compression device */
 359 static int
 360 compdev_isal_remove_dev(struct rte_vdev_device *vdev)
 361 {
 362         struct rte_compressdev *compdev;
 363         const char *name;
 364
 365         name = rte_vdev_device_name(vdev);
 366         if (name == NULL)
 367                 return -EINVAL;
 368
 369         compdev = rte_compressdev_pmd_get_named_dev(name);
 370         if (compdev == NULL)
 371                 return -ENODEV;
 372
 373         return rte_compressdev_pmd_destroy(compdev);
 374 }
 375
 376 /** Initialise ISA-L compression device */
 377 static int
 378 compdev_isal_probe(struct rte_vdev_device *dev)
 379 {
 380         struct rte_compressdev_pmd_init_params init_params = {
 381                 "",
 382                 rte_socket_id(),
 383         };
 384         const char *name, *args;
 385         int retval;
 386
 387         name = rte_vdev_device_name(dev);
 388         if (name == NULL)
 389                 return -EINVAL;
 390
 391         args = rte_vdev_device_args(dev);
 392
 393         retval = rte_compressdev_pmd_parse_input_args(&init_params, args);
 394         if (retval) {
 395                 ISAL_PMD_LOG(ERR,
 396                         "Failed to parse initialisation arguments[%s]\n", args);
 397                 return -EINVAL;
 398         }
 399
 400         return compdev_isal_create(name, dev, &init_params);
 401 }
 402
 403 static struct rte_vdev_driver compdev_isal_pmd_drv = {
 404         .probe = compdev_isal_probe,
 405         .remove = compdev_isal_remove_dev,
 406 };
 407
 408 RTE_PMD_REGISTER_VDEV(COMPDEV_NAME_ISAL_PMD, compdev_isal_pmd_drv);
 409 RTE_PMD_REGISTER_PARAM_STRING(COMPDEV_NAME_ISAL_PMD,
 410         "socket_id=<int>");
 411
 412 RTE_INIT(isal_init_log);
 413
 414 static void
 415 isal_init_log(void)
 416 {
 417         isal_logtype_driver = rte_log_register("comp_isal");
 418         if (isal_logtype_driver >= 0)
 419                 rte_log_set_level(isal_logtype_driver, RTE_LOG_INFO);
 420 }