[Accel-config] [PATCH v1 09/14] accel-config/test: Add test code of operation Compress

Add test code of operation Compress

Signed-off-by: Li Zhang <li4.zhang(a)intel.com>
---
 test/iaa.c      | 136 ++++++++++++++++++++++++++++++++++++++++++++++++
 test/iaa.h      |   4 ++
 test/iaa_prep.c |  14 +++++
 test/iaa_test.c |  65 +++++++++++++++++++++++
 4 files changed, 219 insertions(+)

diff --git a/test/iaa.c b/test/iaa.c
index 524446a..0c607ee 100644
--- a/test/iaa.c
+++ b/test/iaa.c
@@ -18,6 +18,7 @@
 #include "iaa.h"
 #include "algorithms/iaa_crc64.h"
 #include "algorithms/iaa_zcompress.h"
+#include "algorithms/iaa_compress.h"
 
 static int init_crc64(struct task *tsk, int tflags, int opcode, unsigned long src1_xfer_size)
 {
@@ -153,6 +154,36 @@ static int init_zdecompress32(struct task *tsk, int tflags, int opcode, unsigned
 	return ACCTEST_STATUS_OK;
 }
 
+static int init_compress(struct task *tsk, int tflags, int opcode, unsigned long src1_xfer_size)
+{
+	tsk->pattern = 0x98765432abcdef01;
+	tsk->opcode = opcode;
+	tsk->test_flags = tflags;
+	tsk->xfer_size = src1_xfer_size;
+
+	tsk->src1 = aligned_alloc(32, src1_xfer_size);
+	if (!tsk->src1)
+		return -ENOMEM;
+	memset_pattern(tsk->src1, tsk->pattern, src1_xfer_size);
+
+	tsk->src2 = aligned_alloc(32, IAA_COMPRESS_SRC2_SIZE);
+	if (!tsk->src2)
+		return -ENOMEM;
+	memset_pattern(tsk->src2, 0, IAA_COMPRESS_SRC2_SIZE);
+
+	tsk->dst1 = aligned_alloc(32, IAA_COMPRESS_MAX_DEST_SIZE);
+	if (!tsk->dst1)
+		return -ENOMEM;
+	memset_pattern(tsk->dst1, 0, IAA_COMPRESS_MAX_DEST_SIZE);
+
+	tsk->output = aligned_alloc(32, IAA_COMPRESS_MAX_DEST_SIZE);
+	if (!tsk->output)
+		return -ENOMEM;
+	memset_pattern(tsk->output, 0, IAA_COMPRESS_MAX_DEST_SIZE);
+
+	return ACCTEST_STATUS_OK;
+}
+
 int init_task(struct task *tsk, int tflags, int opcode, unsigned long src1_xfer_size)
 {
 	int rc = 0;
@@ -176,6 +207,9 @@ int init_task(struct task *tsk, int tflags, int opcode, unsigned long src1_xfer_
 	case IAX_OPCODE_ZDECOMPRESS32:
 		rc = init_zdecompress32(tsk, tflags, opcode, src1_xfer_size);
 		break;
+	case IAX_OPCODE_COMPRESS:
+		rc = init_compress(tsk, tflags, opcode, src1_xfer_size);
+		break;
 	}
 
 	if (rc != ACCTEST_STATUS_OK) {
@@ -468,6 +502,62 @@ int iaa_zdecompress32_multi_task_nodes(struct acctest_context *ctx)
 	return ret;
 }
 
+static int iaa_wait_compress(struct acctest_context *ctx, struct task *tsk)
+{
+	struct completion_record *comp = tsk->comp;
+	int rc;
+
+	rc = acctest_wait_on_desc_timeout(comp, ctx, ms_timeout);
+	if (rc < 0) {
+		err("compress desc timeout\n");
+		return ACCTEST_STATUS_TIMEOUT;
+	}
+
+	return ACCTEST_STATUS_OK;
+}
+
+int iaa_compress_multi_task_nodes(struct acctest_context *ctx)
+{
+	struct task_node *tsk_node = ctx->multi_task_node;
+	int ret = ACCTEST_STATUS_OK;
+
+	while (tsk_node) {
+		tsk_node->tsk->dflags |= (IDXD_OP_FLAG_CRAV | IDXD_OP_FLAG_RCR);
+		if ((tsk_node->tsk->test_flags & TEST_FLAGS_BOF) && ctx->bof)
+			tsk_node->tsk->dflags |= IDXD_OP_FLAG_BOF;
+
+		tsk_node->tsk->dflags |= (IDXD_OP_FLAG_WR_SRC2_CMPL | IDXD_OP_FLAG_RD_SRC2_AECS);
+		tsk_node->tsk->iaa_src2_xfer_size = IAA_COMPRESS_AECS_SIZE;
+
+		memcpy(tsk_node->tsk->src2, (void *)iaa_compress_aecs, IAA_COMPRESS_AECS_SIZE);
+
+		tsk_node->tsk->iaa_compr_flags = (IDXD_COMPRESS_FLAG_EOB_BFINAL |
+						  IDXD_COMPRESS_FLAG_FLUSH_OUTPUT);
+		tsk_node->tsk->iaa_max_dst_size = IAA_COMPRESS_MAX_DEST_SIZE;
+
+		iaa_prep_compress(tsk_node->tsk);
+		tsk_node = tsk_node->next;
+	}
+
+	info("Submitted all compress jobs\n");
+	tsk_node = ctx->multi_task_node;
+	while (tsk_node) {
+		acctest_desc_submit(ctx, tsk_node->tsk->desc);
+		tsk_node = tsk_node->next;
+	}
+
+	tsk_node = ctx->multi_task_node;
+	while (tsk_node) {
+		ret = iaa_wait_compress(ctx, tsk_node->tsk);
+		if (ret != ACCTEST_STATUS_OK)
+			info("Desc: %p failed with ret: %d\n",
+			     tsk_node->tsk->desc, tsk_node->tsk->comp->status);
+		tsk_node = tsk_node->next;
+	}
+
+	return ret;
+}
+
 /* mismatch_expected: expect mismatched buffer with success status 0x1 */
 int iaa_task_result_verify(struct task *tsk, int mismatch_expected)
 {
@@ -494,6 +584,9 @@ int iaa_task_result_verify(struct task *tsk, int mismatch_expected)
 	case IAX_OPCODE_ZDECOMPRESS32:
 		ret = task_result_verify_zdecompress32(tsk, mismatch_expected);
 		break;
+	case IAX_OPCODE_COMPRESS:
+		ret = task_result_verify_compress(tsk, mismatch_expected);
+		break;
 	}
 
 	if (ret == ACCTEST_STATUS_OK)
@@ -722,3 +815,46 @@ int task_result_verify_zdecompress32(struct task *tsk, int mismatch_expected)
 
 	return -ENXIO;
 }
+
+int task_result_verify_compress(struct task *tsk, int mismatch_expected)
+{
+	int i = 0;
+	int rc;
+	int expected_len;
+
+	if (mismatch_expected)
+		warn("invalid arg mismatch_expected for %d\n", tsk->opcode);
+
+	rc = iaa_do_decompress(tsk->output, tsk->dst1, tsk->comp->iax_output_size, &expected_len);
+	if (rc)
+		return -ENXIO;
+	rc = memcmp(tsk->src1, tsk->output, expected_len);
+
+	if (!mismatch_expected) {
+		if (expected_len - tsk->xfer_size) {
+			err("Compress mismatch, exp len %d, act len %d\n",
+			    expected_len, tsk->xfer_size);
+
+			return -ENXIO;
+		}
+		if (rc) {
+			err("Compress mismatch, memcmp rc %d\n", rc);
+			for (i = 0; i < (expected_len / 4); i++) {
+				printf("Exp[%d]=0x%04X, Act[%d]=0x%04X\n",
+				       i, ((uint32_t *)tsk->output)[i],
+				       i, ((uint32_t *)tsk->src1)[i]);
+			}
+
+			return -ENXIO;
+		}
+		return ACCTEST_STATUS_OK;
+	}
+
+	/* mismatch_expected */
+	if (rc) {
+		info("expected mismatch\n");
+		return ACCTEST_STATUS_OK;
+	}
+
+	return -ENXIO;
+}
diff --git a/test/iaa.h b/test/iaa.h
index 2a14781..1e6010f 100644
--- a/test/iaa.h
+++ b/test/iaa.h
@@ -15,12 +15,15 @@ int iaa_zcompress16_multi_task_nodes(struct acctest_context *ctx);
 int iaa_zdecompress16_multi_task_nodes(struct acctest_context *ctx);
 int iaa_zcompress32_multi_task_nodes(struct acctest_context *ctx);
 int iaa_zdecompress32_multi_task_nodes(struct acctest_context *ctx);
+int iaa_compress_multi_task_nodes(struct acctest_context *ctx);
+
 void iaa_prep_noop(struct task *tsk);
 void iaa_prep_crc64(struct task *tsk);
 void iaa_prep_zcompress16(struct task *tsk);
 void iaa_prep_zdecompress16(struct task *tsk);
 void iaa_prep_zcompress32(struct task *tsk);
 void iaa_prep_zdecompress32(struct task *tsk);
+void iaa_prep_compress(struct task *tsk);
 
 int iaa_task_result_verify(struct task *tsk, int mismatch_expected);
 int iaa_task_result_verify_task_nodes(struct acctest_context *ctx, int mismatch_expected);
@@ -29,5 +32,6 @@ int task_result_verify_zcompress16(struct task *tsk, int mismatch_expected);
 int task_result_verify_zdecompress16(struct task *tsk, int mismatch_expected);
 int task_result_verify_zcompress32(struct task *tsk, int mismatch_expected);
 int task_result_verify_zdecompress32(struct task *tsk, int mismatch_expected);
+int task_result_verify_compress(struct task *tsk, int mismatch_expected);
 
 #endif
diff --git a/test/iaa_prep.c b/test/iaa_prep.c
index 78668ac..0d454ff 100644
--- a/test/iaa_prep.c
+++ b/test/iaa_prep.c
@@ -75,3 +75,17 @@ void iaa_prep_zdecompress32(struct task *tsk)
 	tsk->desc->iax_max_dst_size = tsk->iaa_max_dst_size;
 	tsk->comp->status = 0;
 }
+
+void iaa_prep_compress(struct task *tsk)
+{
+	info("preparing descriptor for compress\n");
+
+	acctest_prep_desc_common(tsk->desc, tsk->opcode, (uint64_t)(tsk->dst1),
+				 (uint64_t)(tsk->src1), tsk->xfer_size, tsk->dflags);
+	tsk->desc->completion_addr = (uint64_t)(tsk->comp);
+	tsk->desc->iax_compr_flags = tsk->iaa_compr_flags;
+	tsk->desc->iax_src2_addr = (uint64_t)(tsk->src2);
+	tsk->desc->iax_src2_xfer_size = tsk->iaa_src2_xfer_size;
+	tsk->desc->iax_max_dst_size = tsk->iaa_max_dst_size;
+	tsk->comp->status = 0;
+}
diff --git a/test/iaa_test.c b/test/iaa_test.c
index ba3d412..16bf363 100644
--- a/test/iaa_test.c
+++ b/test/iaa_test.c
@@ -221,6 +221,65 @@ static int test_zcompress(struct acctest_context *ctx, size_t buf_size,
 	return rc;
 }
 
+static int test_compress(struct acctest_context *ctx, size_t buf_size, int tflags,
+			 int extra_flags, uint32_t opcode, int num_desc)
+{
+	struct task_node *tsk_node;
+	int rc = ACCTEST_STATUS_OK;
+	int itr = num_desc, i = 0, range = 0;
+
+	info("testcompress: opcode %d len %#lx tflags %#x num_desc %ld extra_flags %#lx\n",
+	     opcode, buf_size, tflags, num_desc, extra_flags);
+
+	ctx->is_batch = 0;
+
+	if (ctx->dedicated == ACCFG_WQ_SHARED)
+		range = ctx->threshold;
+	else
+		range = ctx->wq_size;
+
+	while (itr > 0 && rc == ACCTEST_STATUS_OK) {
+		i = (itr < range) ? itr : range;
+		/* Allocate memory to all the task nodes, desc, completion record*/
+		rc = acctest_alloc_multiple_tasks(ctx, i);
+		if (rc != ACCTEST_STATUS_OK)
+			return rc;
+
+		/* allocate memory to src and dest buffers and fill in the desc for all the nodes*/
+		tsk_node = ctx->multi_task_node;
+		while (tsk_node) {
+			tsk_node->tsk->iaa_compr_flags = extra_flags;
+
+			rc = init_task(tsk_node->tsk, tflags, opcode, buf_size);
+			if (rc != ACCTEST_STATUS_OK)
+				return rc;
+
+			tsk_node = tsk_node->next;
+		}
+
+		switch (opcode) {
+		case IAX_OPCODE_COMPRESS:
+			rc = iaa_compress_multi_task_nodes(ctx);
+			if (rc != ACCTEST_STATUS_OK)
+				return rc;
+
+			/* Verification of all the nodes*/
+			rc = iaa_task_result_verify_task_nodes(ctx, 0);
+			if (rc != ACCTEST_STATUS_OK)
+				return rc;
+			break;
+		default:
+			err("Unsupported op %#x\n", opcode);
+			return -EINVAL;
+		}
+
+		acctest_free_task(ctx);
+		itr = itr - range;
+	}
+
+	return rc;
+}
+
 int main(int argc, char *argv[])
 {
 	struct acctest_context *iaa;
@@ -315,6 +374,12 @@ int main(int argc, char *argv[])
 			goto error;
 		break;
 
+	case IAX_OPCODE_COMPRESS:
+		rc = test_compress(iaa, buf_size, tflags, extra_flags, opcode, num_desc);
+		if (rc != ACCTEST_STATUS_OK)
+			goto error;
+		break;
+
 	default:
 		rc = -EINVAL;
 		break;
-- 
2.25.1