app/test-compress-perf/comp_perf_test_verify.c

   1 /* SPDX-License-Identifier: BSD-3-Clause
   2  * Copyright(c) 2018 Intel Corporation
   3  */
   4
   5 #include <rte_malloc.h>
   6 #include <rte_eal.h>
   7 #include <rte_log.h>
   8 #include <rte_compressdev.h>
   9
  10 #include "comp_perf_test_verify.h"
  11 #include "comp_perf_test_common.h"
  12
  13 void
  14 cperf_verify_test_destructor(void *arg)
  15 {
  16         if (arg) {
  17                 comp_perf_free_memory(&((struct cperf_verify_ctx *)arg)->mem);
  18                 rte_free(arg);
  19         }
  20 }
  21
  22 void *
  23 cperf_verify_test_constructor(uint8_t dev_id, uint16_t qp_id,
  24                 struct comp_test_data *options)
  25 {
  26         struct cperf_verify_ctx *ctx = NULL;
  27
  28         ctx = rte_malloc(NULL, sizeof(struct cperf_verify_ctx), 0);
  29
  30         if (ctx == NULL)
  31                 return NULL;
  32
  33         ctx->mem.dev_id = dev_id;
  34         ctx->mem.qp_id = qp_id;
  35         ctx->options = options;
  36
  37         if (!comp_perf_allocate_memory(ctx->options, &ctx->mem) &&
  38                         !prepare_bufs(ctx->options, &ctx->mem))
  39                 return ctx;
  40
  41         cperf_verify_test_destructor(ctx);
  42         return NULL;
  43 }
  44
  45 static int
  46 main_loop(struct cperf_verify_ctx *ctx, enum rte_comp_xform_type type)
  47 {
  48         struct comp_test_data *test_data = ctx->options;
  49         uint8_t *output_data_ptr;
  50         size_t *output_data_sz;
  51         struct cperf_mem_resources *mem = &ctx->mem;
  52
  53         uint8_t dev_id = mem->dev_id;
  54         uint32_t i, iter, num_iter;
  55         struct rte_comp_op **ops, **deq_ops;
  56         void *priv_xform = NULL;
  57         struct rte_comp_xform xform;
  58         size_t output_size = 0;
  59         struct rte_mbuf **input_bufs, **output_bufs;
  60         int res = 0;
  61         int allocated = 0;
  62         uint32_t out_seg_sz;
  63
  64         if (test_data == NULL || !test_data->burst_sz) {
  65                 RTE_LOG(ERR, USER1,
  66                         "Unknown burst size\n");
  67                 return -1;
  68         }
  69
  70         ops = rte_zmalloc_socket(NULL,
  71                 2 * mem->total_bufs * sizeof(struct rte_comp_op *),
  72                 0, rte_socket_id());
  73
  74         if (ops == NULL) {
  75                 RTE_LOG(ERR, USER1,
  76                         "Can't allocate memory for ops strucures\n");
  77                 return -1;
  78         }
  79
  80         deq_ops = &ops[mem->total_bufs];
  81
  82         if (type == RTE_COMP_COMPRESS) {
  83                 xform = (struct rte_comp_xform) {
  84                         .type = RTE_COMP_COMPRESS,
  85                         .compress = {
  86                                 .algo = RTE_COMP_ALGO_DEFLATE,
  87                                 .deflate.huffman = test_data->huffman_enc,
  88                                 .level = test_data->level,
  89                                 .window_size = test_data->window_sz,
  90                                 .chksum = RTE_COMP_CHECKSUM_NONE,
  91                                 .hash_algo = RTE_COMP_HASH_ALGO_NONE
  92                         }
  93                 };
  94                 output_data_ptr = ctx->mem.compressed_data;
  95                 output_data_sz = &ctx->comp_data_sz;
  96                 input_bufs = mem->decomp_bufs;
  97                 output_bufs = mem->comp_bufs;
  98                 out_seg_sz = test_data->out_seg_sz;
  99         } else {
 100                 xform = (struct rte_comp_xform) {
 101                         .type = RTE_COMP_DECOMPRESS,
 102                         .decompress = {
 103                                 .algo = RTE_COMP_ALGO_DEFLATE,
 104                                 .chksum = RTE_COMP_CHECKSUM_NONE,
 105                                 .window_size = test_data->window_sz,
 106                                 .hash_algo = RTE_COMP_HASH_ALGO_NONE
 107                         }
 108                 };
 109                 output_data_ptr = ctx->mem.decompressed_data;
 110                 output_data_sz = &ctx->decomp_data_sz;
 111                 input_bufs = mem->comp_bufs;
 112                 output_bufs = mem->decomp_bufs;
 113                 out_seg_sz = test_data->seg_sz;
 114         }
 115
 116         /* Create private xform */
 117         if (rte_compressdev_private_xform_create(dev_id, &xform,
 118                         &priv_xform) < 0) {
 119                 RTE_LOG(ERR, USER1, "Private xform could not be created\n");
 120                 res = -1;
 121                 goto end;
 122         }
 123
 124         num_iter = 1;
 125
 126         for (iter = 0; iter < num_iter; iter++) {
 127                 uint32_t total_ops = mem->total_bufs;
 128                 uint32_t remaining_ops = mem->total_bufs;
 129                 uint32_t total_deq_ops = 0;
 130                 uint32_t total_enq_ops = 0;
 131                 uint16_t ops_unused = 0;
 132                 uint16_t num_enq = 0;
 133                 uint16_t num_deq = 0;
 134
 135                 output_size = 0;
 136
 137                 while (remaining_ops > 0) {
 138                         uint16_t num_ops = RTE_MIN(remaining_ops,
 139                                                    test_data->burst_sz);
 140                         uint16_t ops_needed = num_ops - ops_unused;
 141
 142                         /*
 143                          * Move the unused operations from the previous
 144                          * enqueue_burst call to the front, to maintain order
 145                          */
 146                         if ((ops_unused > 0) && (num_enq > 0)) {
 147                                 size_t nb_b_to_mov =
 148                                       ops_unused * sizeof(struct rte_comp_op *);
 149
 150                                 memmove(ops, &ops[num_enq], nb_b_to_mov);
 151                         }
 152
 153                         /* Allocate compression operations */
 154                         if (ops_needed && !rte_comp_op_bulk_alloc(
 155                                                 mem->op_pool,
 156                                                 &ops[ops_unused],
 157                                                 ops_needed)) {
 158                                 RTE_LOG(ERR, USER1,
 159                                       "Could not allocate enough operations\n");
 160                                 res = -1;
 161                                 goto end;
 162                         }
 163                         allocated += ops_needed;
 164
 165                         for (i = 0; i < ops_needed; i++) {
 166                                 /*
 167                                  * Calculate next buffer to attach to operation
 168                                  */
 169                                 uint32_t buf_id = total_enq_ops + i +
 170                                                 ops_unused;
 171                                 uint16_t op_id = ops_unused + i;
 172                                 /* Reset all data in output buffers */
 173                                 struct rte_mbuf *m = output_bufs[buf_id];
 174
 175                                 m->pkt_len = out_seg_sz * m->nb_segs;
 176                                 while (m) {
 177                                         m->data_len = m->buf_len - m->data_off;
 178                                         m = m->next;
 179                                 }
 180                                 ops[op_id]->m_src = input_bufs[buf_id];
 181                                 ops[op_id]->m_dst = output_bufs[buf_id];
 182                                 ops[op_id]->src.offset = 0;
 183                                 ops[op_id]->src.length =
 184                                         rte_pktmbuf_pkt_len(input_bufs[buf_id]);
 185                                 ops[op_id]->dst.offset = 0;
 186                                 ops[op_id]->flush_flag = RTE_COMP_FLUSH_FINAL;
 187                                 ops[op_id]->input_chksum = buf_id;
 188                                 ops[op_id]->private_xform = priv_xform;
 189                         }
 190
 191                         if (unlikely(test_data->perf_comp_force_stop))
 192                                 goto end;
 193
 194                         num_enq = rte_compressdev_enqueue_burst(dev_id,
 195                                                                 mem->qp_id, ops,
 196                                                                 num_ops);
 197                         if (num_enq == 0) {
 198                                 struct rte_compressdev_stats stats;
 199
 200                                 rte_compressdev_stats_get(dev_id, &stats);
 201                                 if (stats.enqueue_err_count) {
 202                                         res = -1;
 203                                         goto end;
 204                                 }
 205                         }
 206
 207                         ops_unused = num_ops - num_enq;
 208                         remaining_ops -= num_enq;
 209                         total_enq_ops += num_enq;
 210
 211                         num_deq = rte_compressdev_dequeue_burst(dev_id,
 212                                                            mem->qp_id,
 213                                                            deq_ops,
 214                                                            test_data->burst_sz);
 215                         total_deq_ops += num_deq;
 216
 217                         for (i = 0; i < num_deq; i++) {
 218                                 struct rte_comp_op *op = deq_ops[i];
 219
 220                                 if (op->status != RTE_COMP_OP_STATUS_SUCCESS) {
 221                                         RTE_LOG(ERR, USER1,
 222                                                 "Some operations were not successful\n");
 223                                         goto end;
 224                                 }
 225
 226                                 const void *read_data_addr =
 227                                                 rte_pktmbuf_read(op->m_dst, 0,
 228                                                 op->produced, output_data_ptr);
 229                                 if (read_data_addr == NULL) {
 230                                         RTE_LOG(ERR, USER1,
 231                                                 "Could not copy buffer in destination\n");
 232                                         res = -1;
 233                                         goto end;
 234                                 }
 235
 236                                 if (read_data_addr != output_data_ptr)
 237                                         rte_memcpy(output_data_ptr,
 238                                                    rte_pktmbuf_mtod(op->m_dst,
 239                                                                     uint8_t *),
 240                                                    op->produced);
 241                                 output_data_ptr += op->produced;
 242                                 output_size += op->produced;
 243
 244                         }
 245
 246
 247                         if (iter == num_iter - 1) {
 248                                 for (i = 0; i < num_deq; i++) {
 249                                         struct rte_comp_op *op = deq_ops[i];
 250                                         struct rte_mbuf *m = op->m_dst;
 251
 252                                         m->pkt_len = op->produced;
 253                                         uint32_t remaining_data = op->produced;
 254                                         uint16_t data_to_append;
 255
 256                                         while (remaining_data > 0) {
 257                                                 data_to_append =
 258                                                         RTE_MIN(remaining_data,
 259                                                         out_seg_sz);
 260                                                 m->data_len = data_to_append;
 261                                                 remaining_data -=
 262                                                                 data_to_append;
 263                                                 m = m->next;
 264                                         }
 265                                 }
 266                         }
 267                         rte_mempool_put_bulk(mem->op_pool,
 268                                              (void **)deq_ops, num_deq);
 269                         allocated -= num_deq;
 270                 }
 271
 272                 /* Dequeue the last operations */
 273                 while (total_deq_ops < total_ops) {
 274                         if (unlikely(test_data->perf_comp_force_stop))
 275                                 goto end;
 276
 277                         num_deq = rte_compressdev_dequeue_burst(dev_id,
 278                                                         mem->qp_id,
 279                                                         deq_ops,
 280                                                         test_data->burst_sz);
 281                         if (num_deq == 0) {
 282                                 struct rte_compressdev_stats stats;
 283
 284                                 rte_compressdev_stats_get(dev_id, &stats);
 285                                 if (stats.dequeue_err_count) {
 286                                         res = -1;
 287                                         goto end;
 288                                 }
 289                         }
 290
 291                         total_deq_ops += num_deq;
 292
 293                         for (i = 0; i < num_deq; i++) {
 294                                 struct rte_comp_op *op = deq_ops[i];
 295
 296                                 if (op->status != RTE_COMP_OP_STATUS_SUCCESS) {
 297                                         RTE_LOG(ERR, USER1,
 298                                                 "Some operations were not successful\n");
 299                                         goto end;
 300                                 }
 301
 302                                 const void *read_data_addr =
 303                                                 rte_pktmbuf_read(op->m_dst,
 304                                                                  op->dst.offset,
 305                                                 op->produced, output_data_ptr);
 306                                 if (read_data_addr == NULL) {
 307                                         RTE_LOG(ERR, USER1,
 308                                                 "Could not copy buffer in destination\n");
 309                                         res = -1;
 310                                         goto end;
 311                                 }
 312
 313                                 if (read_data_addr != output_data_ptr)
 314                                         rte_memcpy(output_data_ptr,
 315                                                    rte_pktmbuf_mtod(
 316                                                         op->m_dst, uint8_t *),
 317                                                    op->produced);
 318                                 output_data_ptr += op->produced;
 319                                 output_size += op->produced;
 320
 321                         }
 322
 323                         if (iter == num_iter - 1) {
 324                                 for (i = 0; i < num_deq; i++) {
 325                                         struct rte_comp_op *op = deq_ops[i];
 326                                         struct rte_mbuf *m = op->m_dst;
 327
 328                                         m->pkt_len = op->produced;
 329                                         uint32_t remaining_data = op->produced;
 330                                         uint16_t data_to_append;
 331
 332                                         while (remaining_data > 0) {
 333                                                 data_to_append =
 334                                                 RTE_MIN(remaining_data,
 335                                                         out_seg_sz);
 336                                                 m->data_len = data_to_append;
 337                                                 remaining_data -=
 338                                                                 data_to_append;
 339                                                 m = m->next;
 340                                         }
 341                                 }
 342                         }
 343                         rte_mempool_put_bulk(mem->op_pool,
 344                                              (void **)deq_ops, num_deq);
 345                         allocated -= num_deq;
 346                 }
 347         }
 348
 349         if (output_data_sz)
 350                 *output_data_sz = output_size;
 351 end:
 352         rte_mempool_put_bulk(mem->op_pool, (void **)ops, allocated);
 353         rte_compressdev_private_xform_free(dev_id, priv_xform);
 354         rte_free(ops);
 355
 356         if (test_data->perf_comp_force_stop) {
 357                 RTE_LOG(ERR, USER1,
 358                       "lcore: %d Perf. test has been aborted by user\n",
 359                         mem->lcore_id);
 360                 res = -1;
 361         }
 362
 363         return res;
 364 }
 365
 366 int
 367 cperf_verify_test_runner(void *test_ctx)
 368 {
 369         struct cperf_verify_ctx *ctx = test_ctx;
 370         struct comp_test_data *test_data = ctx->options;
 371         int ret = EXIT_SUCCESS;
 372         static rte_atomic16_t display_once = RTE_ATOMIC16_INIT(0);
 373         uint32_t lcore = rte_lcore_id();
 374
 375         ctx->mem.lcore_id = lcore;
 376
 377         test_data->ratio = 0;
 378
 379         if (main_loop(ctx, RTE_COMP_COMPRESS) < 0) {
 380                 ret = EXIT_FAILURE;
 381                 goto end;
 382         }
 383
 384         if (main_loop(ctx, RTE_COMP_DECOMPRESS) < 0) {
 385                 ret = EXIT_FAILURE;
 386                 goto end;
 387         }
 388
 389         if (ctx->decomp_data_sz != test_data->input_data_sz) {
 390                 RTE_LOG(ERR, USER1,
 391            "Decompressed data length not equal to input data length\n");
 392                 RTE_LOG(ERR, USER1,
 393                         "Decompressed size = %zu, expected = %zu\n",
 394                         ctx->decomp_data_sz, test_data->input_data_sz);
 395                 ret = EXIT_FAILURE;
 396                 goto end;
 397         } else {
 398                 if (memcmp(ctx->mem.decompressed_data,
 399                                 test_data->input_data,
 400                                 test_data->input_data_sz) != 0) {
 401                         RTE_LOG(ERR, USER1,
 402                     "Decompressed data is not the same as file data\n");
 403                         ret = EXIT_FAILURE;
 404                         goto end;
 405                 }
 406         }
 407
 408         ctx->ratio = (double) ctx->comp_data_sz /
 409                         test_data->input_data_sz * 100;
 410
 411         if (!ctx->silent) {
 412                 if (rte_atomic16_test_and_set(&display_once)) {
 413                         printf("%12s%6s%12s%17s\n",
 414                             "lcore id", "Level", "Comp size", "Comp ratio [%]");
 415                 }
 416                 printf("%12u%6u%12zu%17.2f\n",
 417                        ctx->mem.lcore_id,
 418                        test_data->level, ctx->comp_data_sz, ctx->ratio);
 419         }
 420
 421 end:
 422         return ret;
 423 }