drivers/crypto/inside-secure/safexcel_cipher.c - pub/scm/linux/kernel/git/jic23/wrapdrive - Git at Google

 /*
  * Copyright (C) 2017 Marvell
  *
  * Antoine Tenart <antoine.tenart@free-electrons.com>
  *
  * This file is licensed under the terms of the GNU General Public
  * License version 2. This program is licensed "as is" without any
  * warranty of any kind, whether express or implied.
  */

 #include <linux/device.h>
 #include <linux/dma-mapping.h>
 #include <linux/dmapool.h>

 #include <crypto/aes.h>
 #include <crypto/skcipher.h>

 #include "safexcel.h"

 enum safexcel_cipher_direction {
 	SAFEXCEL_ENCRYPT,
 	SAFEXCEL_DECRYPT,
 };

 struct safexcel_cipher_ctx {
 	struct safexcel_context base;
 	struct safexcel_crypto_priv *priv;

 	enum safexcel_cipher_direction direction;
 	u32 mode;

 	__le32 key[8];
 	unsigned int key_len;
 };

 static void safexcel_cipher_token(struct safexcel_cipher_ctx *ctx,
 				  struct crypto_async_request *async,
 				  struct safexcel_command_desc *cdesc,
 				  u32 length)
 {
 	struct skcipher_request *req = skcipher_request_cast(async);
 	struct safexcel_token *token;
 	unsigned offset = 0;

 	if (ctx->mode == CONTEXT_CONTROL_CRYPTO_MODE_CBC) {
 		offset = AES_BLOCK_SIZE / sizeof(u32);
 		memcpy(cdesc->control_data.token, req->iv, AES_BLOCK_SIZE);

 		cdesc->control_data.options |= EIP197_OPTION_4_TOKEN_IV_CMD;
 	}

 	token = (struct safexcel_token *)(cdesc->control_data.token + offset);

 	token[0].opcode = EIP197_TOKEN_OPCODE_DIRECTION;
 	token[0].packet_length = length;
 	token[0].stat = EIP197_TOKEN_STAT_LAST_PACKET;
 	token[0].instructions = EIP197_TOKEN_INS_LAST |
 				EIP197_TOKEN_INS_TYPE_CRYTO |
 				EIP197_TOKEN_INS_TYPE_OUTPUT;
 }

 static int safexcel_aes_setkey(struct crypto_skcipher *ctfm, const u8 *key,
 			       unsigned int len)
 {
 	struct crypto_tfm *tfm = crypto_skcipher_tfm(ctfm);
 	struct safexcel_cipher_ctx *ctx = crypto_tfm_ctx(tfm);
 	struct crypto_aes_ctx aes;
 	int ret, i;

 	ret = crypto_aes_expand_key(&aes, key, len);
 	if (ret) {
 		crypto_skcipher_set_flags(ctfm, CRYPTO_TFM_RES_BAD_KEY_LEN);
 		return ret;
 	}

 	for (i = 0; i < len / sizeof(u32); i++) {
 		if (ctx->key[i] != cpu_to_le32(aes.key_enc[i])) {
 			ctx->base.needs_inv = true;
 			break;
 		}
 	}

 	for (i = 0; i < len / sizeof(u32); i++)
 		ctx->key[i] = cpu_to_le32(aes.key_enc[i]);

 	ctx->key_len = len;

 	memzero_explicit(&aes, sizeof(aes));
 	return 0;
 }

 static int safexcel_context_control(struct safexcel_cipher_ctx *ctx,
 				    struct safexcel_command_desc *cdesc)
 {
 	struct safexcel_crypto_priv *priv = ctx->priv;
 	int ctrl_size;

 	if (ctx->direction == SAFEXCEL_ENCRYPT)
 		cdesc->control_data.control0 |= CONTEXT_CONTROL_TYPE_CRYPTO_OUT;
 	else
 		cdesc->control_data.control0 |= CONTEXT_CONTROL_TYPE_CRYPTO_IN;

 	cdesc->control_data.control0 |= CONTEXT_CONTROL_KEY_EN;
 	cdesc->control_data.control1 |= ctx->mode;

 	switch (ctx->key_len) {
 	case AES_KEYSIZE_128:
 		cdesc->control_data.control0 |= CONTEXT_CONTROL_CRYPTO_ALG_AES128;
 		ctrl_size = 4;
 		break;
 	case AES_KEYSIZE_192:
 		cdesc->control_data.control0 |= CONTEXT_CONTROL_CRYPTO_ALG_AES192;
 		ctrl_size = 6;
 		break;
 	case AES_KEYSIZE_256:
 		cdesc->control_data.control0 |= CONTEXT_CONTROL_CRYPTO_ALG_AES256;
 		ctrl_size = 8;
 		break;
 	default:
 		dev_err(priv->dev, "aes keysize not supported: %u\n",
 			ctx->key_len);
 		return -EINVAL;
 	}
 	cdesc->control_data.control0 |= CONTEXT_CONTROL_SIZE(ctrl_size);

 	return 0;
 }

 static int safexcel_handle_result(struct safexcel_crypto_priv *priv, int ring,
 				  struct crypto_async_request *async,
 				  bool *should_complete, int *ret)
 {
 	struct skcipher_request *req = skcipher_request_cast(async);
 	struct safexcel_result_desc *rdesc;
 	int ndesc = 0;

 	*ret = 0;

 	spin_lock_bh(&priv->ring[ring].egress_lock);
 	do {
 		rdesc = safexcel_ring_next_rptr(priv, &priv->ring[ring].rdr);
 		if (IS_ERR(rdesc)) {
 			dev_err(priv->dev,
 				"cipher: result: could not retrieve the result descriptor\n");
 			*ret = PTR_ERR(rdesc);
 			break;
 		}

 		if (rdesc->result_data.error_code) {
 			dev_err(priv->dev,
 				"cipher: result: result descriptor error (%d)\n",
 				rdesc->result_data.error_code);
 			*ret = -EIO;
 		}

 		ndesc++;
 	} while (!rdesc->last_seg);

 	safexcel_complete(priv, ring);
 	spin_unlock_bh(&priv->ring[ring].egress_lock);

 	if (req->src == req->dst) {
 		dma_unmap_sg(priv->dev, req->src,
 			     sg_nents_for_len(req->src, req->cryptlen),
 			     DMA_BIDIRECTIONAL);
 	} else {
 		dma_unmap_sg(priv->dev, req->src,
 			     sg_nents_for_len(req->src, req->cryptlen),
 			     DMA_TO_DEVICE);
 		dma_unmap_sg(priv->dev, req->dst,
 			     sg_nents_for_len(req->dst, req->cryptlen),
 			     DMA_FROM_DEVICE);
 	}

 	*should_complete = true;

 	return ndesc;
 }

 static int safexcel_aes_send(struct crypto_async_request *async,
 			     int ring, struct safexcel_request *request,
 			     int *commands, int *results)
 {
 	struct skcipher_request *req = skcipher_request_cast(async);
 	struct safexcel_cipher_ctx *ctx = crypto_tfm_ctx(req->base.tfm);
 	struct safexcel_crypto_priv *priv = ctx->priv;
 	struct safexcel_command_desc *cdesc;
 	struct safexcel_result_desc *rdesc;
 	struct scatterlist *sg;
 	int nr_src, nr_dst, n_cdesc = 0, n_rdesc = 0, queued = req->cryptlen;
 	int i, ret = 0;

 	if (req->src == req->dst) {
 		nr_src = dma_map_sg(priv->dev, req->src,
 				    sg_nents_for_len(req->src, req->cryptlen),
 				    DMA_BIDIRECTIONAL);
 		nr_dst = nr_src;
 		if (!nr_src)
 			return -EINVAL;
 	} else {
 		nr_src = dma_map_sg(priv->dev, req->src,
 				    sg_nents_for_len(req->src, req->cryptlen),
 				    DMA_TO_DEVICE);
 		if (!nr_src)
 			return -EINVAL;

 		nr_dst = dma_map_sg(priv->dev, req->dst,
 				    sg_nents_for_len(req->dst, req->cryptlen),
 				    DMA_FROM_DEVICE);
 		if (!nr_dst) {
 			dma_unmap_sg(priv->dev, req->src,
 				     sg_nents_for_len(req->src, req->cryptlen),
 				     DMA_TO_DEVICE);
 			return -EINVAL;
 		}
 	}

 	memcpy(ctx->base.ctxr->data, ctx->key, ctx->key_len);

 	spin_lock_bh(&priv->ring[ring].egress_lock);

 	/* command descriptors */
 	for_each_sg(req->src, sg, nr_src, i) {
 		int len = sg_dma_len(sg);

 		/* Do not overflow the request */
 		if (queued - len < 0)
 			len = queued;

 		cdesc = safexcel_add_cdesc(priv, ring, !n_cdesc, !(queued - len),
 					   sg_dma_address(sg), len, req->cryptlen,
 					   ctx->base.ctxr_dma);
 		if (IS_ERR(cdesc)) {
 			/* No space left in the command descriptor ring */
 			ret = PTR_ERR(cdesc);
 			goto cdesc_rollback;
 		}
 		n_cdesc++;

 		if (n_cdesc == 1) {
 			safexcel_context_control(ctx, cdesc);
 			safexcel_cipher_token(ctx, async, cdesc, req->cryptlen);
 		}

 		queued -= len;
 		if (!queued)
 			break;
 	}

 	/* result descriptors */
 	for_each_sg(req->dst, sg, nr_dst, i) {
 		bool first = !i, last = (i == nr_dst - 1);
 		u32 len = sg_dma_len(sg);

 		rdesc = safexcel_add_rdesc(priv, ring, first, last,
 					   sg_dma_address(sg), len);
 		if (IS_ERR(rdesc)) {
 			/* No space left in the result descriptor ring */
 			ret = PTR_ERR(rdesc);
 			goto rdesc_rollback;
 		}
 		n_rdesc++;
 	}

 	spin_unlock_bh(&priv->ring[ring].egress_lock);

 	request->req = &req->base;
 	ctx->base.handle_result = safexcel_handle_result;

 	*commands = n_cdesc;
 	*results = n_rdesc;
 	return 0;

 rdesc_rollback:
 	for (i = 0; i < n_rdesc; i++)
 		safexcel_ring_rollback_wptr(priv, &priv->ring[ring].rdr);
 cdesc_rollback:
 	for (i = 0; i < n_cdesc; i++)
 		safexcel_ring_rollback_wptr(priv, &priv->ring[ring].cdr);

 	spin_unlock_bh(&priv->ring[ring].egress_lock);

 	if (req->src == req->dst) {
 		dma_unmap_sg(priv->dev, req->src,
 			     sg_nents_for_len(req->src, req->cryptlen),
 			     DMA_BIDIRECTIONAL);
 	} else {
 		dma_unmap_sg(priv->dev, req->src,
 			     sg_nents_for_len(req->src, req->cryptlen),
 			     DMA_TO_DEVICE);
 		dma_unmap_sg(priv->dev, req->dst,
 			     sg_nents_for_len(req->dst, req->cryptlen),
 			     DMA_FROM_DEVICE);
 	}

 	return ret;
 }

 static int safexcel_handle_inv_result(struct safexcel_crypto_priv *priv,
 				      int ring,
 				      struct crypto_async_request *async,
 				      bool *should_complete, int *ret)
 {
 	struct skcipher_request *req = skcipher_request_cast(async);
 	struct safexcel_cipher_ctx *ctx = crypto_tfm_ctx(req->base.tfm);
 	struct safexcel_result_desc *rdesc;
 	int ndesc = 0, enq_ret;

 	*ret = 0;

 	spin_lock_bh(&priv->ring[ring].egress_lock);
 	do {
 		rdesc = safexcel_ring_next_rptr(priv, &priv->ring[ring].rdr);
 		if (IS_ERR(rdesc)) {
 			dev_err(priv->dev,
 				"cipher: invalidate: could not retrieve the result descriptor\n");
 			*ret = PTR_ERR(rdesc);
 			break;
 		}

 		if (rdesc->result_data.error_code) {
 			dev_err(priv->dev, "cipher: invalidate: result descriptor error (%d)\n",
 				rdesc->result_data.error_code);
 			*ret = -EIO;
 		}

 		ndesc++;
 	} while (!rdesc->last_seg);

 	safexcel_complete(priv, ring);
 	spin_unlock_bh(&priv->ring[ring].egress_lock);

 	if (ctx->base.exit_inv) {
 		dma_pool_free(priv->context_pool, ctx->base.ctxr,
 			      ctx->base.ctxr_dma);

 		*should_complete = true;

 		return ndesc;
 	}

 	ring = safexcel_select_ring(priv);
 	ctx->base.ring = ring;
 	ctx->base.needs_inv = false;
 	ctx->base.send = safexcel_aes_send;

 	spin_lock_bh(&priv->ring[ring].queue_lock);
 	enq_ret = crypto_enqueue_request(&priv->ring[ring].queue, async);
 	spin_unlock_bh(&priv->ring[ring].queue_lock);

 	if (enq_ret != -EINPROGRESS)
 		*ret = enq_ret;

 	if (!priv->ring[ring].need_dequeue)
 		safexcel_dequeue(priv, ring);

 	*should_complete = false;

 	return ndesc;
 }

 static int safexcel_cipher_send_inv(struct crypto_async_request *async,
 				    int ring, struct safexcel_request *request,
 				    int *commands, int *results)
 {
 	struct skcipher_request *req = skcipher_request_cast(async);
 	struct safexcel_cipher_ctx *ctx = crypto_tfm_ctx(req->base.tfm);
 	struct safexcel_crypto_priv *priv = ctx->priv;
 	int ret;

 	ctx->base.handle_result = safexcel_handle_inv_result;

 	ret = safexcel_invalidate_cache(async, &ctx->base, priv,
 					ctx->base.ctxr_dma, ring, request);
 	if (unlikely(ret))
 		return ret;

 	*commands = 1;
 	*results = 1;

 	return 0;
 }

 static int safexcel_cipher_exit_inv(struct crypto_tfm *tfm)
 {
 	struct safexcel_cipher_ctx *ctx = crypto_tfm_ctx(tfm);
 	struct safexcel_crypto_priv *priv = ctx->priv;
 	struct skcipher_request req;
 	struct safexcel_inv_result result = { 0 };
 	int ring = ctx->base.ring;

 	memset(&req, 0, sizeof(struct skcipher_request));

 	/* create invalidation request */
 	init_completion(&result.completion);
 	skcipher_request_set_callback(&req, CRYPTO_TFM_REQ_MAY_BACKLOG,
 					safexcel_inv_complete, &result);

 	skcipher_request_set_tfm(&req, __crypto_skcipher_cast(tfm));
 	ctx = crypto_tfm_ctx(req.base.tfm);
 	ctx->base.exit_inv = true;
 	ctx->base.send = safexcel_cipher_send_inv;

 	spin_lock_bh(&priv->ring[ring].queue_lock);
 	crypto_enqueue_request(&priv->ring[ring].queue, &req.base);
 	spin_unlock_bh(&priv->ring[ring].queue_lock);

 	if (!priv->ring[ring].need_dequeue)
 		safexcel_dequeue(priv, ring);

 	wait_for_completion_interruptible(&result.completion);

 	if (result.error) {
 		dev_warn(priv->dev,
 			"cipher: sync: invalidate: completion error %d\n",
 			 result.error);
 		return result.error;
 	}

 	return 0;
 }

 static int safexcel_aes(struct skcipher_request *req,
 			enum safexcel_cipher_direction dir, u32 mode)
 {
 	struct safexcel_cipher_ctx *ctx = crypto_tfm_ctx(req->base.tfm);
 	struct safexcel_crypto_priv *priv = ctx->priv;
 	int ret, ring;

 	ctx->direction = dir;
 	ctx->mode = mode;

 	if (ctx->base.ctxr) {
 		if (ctx->base.needs_inv)
 			ctx->base.send = safexcel_cipher_send_inv;
 	} else {
 		ctx->base.ring = safexcel_select_ring(priv);
 		ctx->base.send = safexcel_aes_send;

 		ctx->base.ctxr = dma_pool_zalloc(priv->context_pool,
 						 EIP197_GFP_FLAGS(req->base),
 						 &ctx->base.ctxr_dma);
 		if (!ctx->base.ctxr)
 			return -ENOMEM;
 	}

 	ring = ctx->base.ring;

 	spin_lock_bh(&priv->ring[ring].queue_lock);
 	ret = crypto_enqueue_request(&priv->ring[ring].queue, &req->base);
 	spin_unlock_bh(&priv->ring[ring].queue_lock);

 	if (!priv->ring[ring].need_dequeue)
 		safexcel_dequeue(priv, ring);

 	return ret;
 }

 static int safexcel_ecb_aes_encrypt(struct skcipher_request *req)
 {
 	return safexcel_aes(req, SAFEXCEL_ENCRYPT,
 			    CONTEXT_CONTROL_CRYPTO_MODE_ECB);
 }

 static int safexcel_ecb_aes_decrypt(struct skcipher_request *req)
 {
 	return safexcel_aes(req, SAFEXCEL_DECRYPT,
 			    CONTEXT_CONTROL_CRYPTO_MODE_ECB);
 }

 static int safexcel_skcipher_cra_init(struct crypto_tfm *tfm)
 {
 	struct safexcel_cipher_ctx *ctx = crypto_tfm_ctx(tfm);
 	struct safexcel_alg_template *tmpl =
 		container_of(tfm->__crt_alg, struct safexcel_alg_template,
 			     alg.skcipher.base);

 	ctx->priv = tmpl->priv;

 	return 0;
 }

 static void safexcel_skcipher_cra_exit(struct crypto_tfm *tfm)
 {
 	struct safexcel_cipher_ctx *ctx = crypto_tfm_ctx(tfm);
 	struct safexcel_crypto_priv *priv = ctx->priv;
 	int ret;

 	memzero_explicit(ctx->key, 8 * sizeof(u32));

 	/* context not allocated, skip invalidation */
 	if (!ctx->base.ctxr)
 		return;

 	memzero_explicit(ctx->base.ctxr->data, 8 * sizeof(u32));

 	ret = safexcel_cipher_exit_inv(tfm);
 	if (ret)
 		dev_warn(priv->dev, "cipher: invalidation error %d\n", ret);
 }

 struct safexcel_alg_template safexcel_alg_ecb_aes = {
 	.type = SAFEXCEL_ALG_TYPE_SKCIPHER,
 	.alg.skcipher = {
 		.setkey = safexcel_aes_setkey,
 		.encrypt = safexcel_ecb_aes_encrypt,
 		.decrypt = safexcel_ecb_aes_decrypt,
 		.min_keysize = AES_MIN_KEY_SIZE,
 		.max_keysize = AES_MAX_KEY_SIZE,
 		.base = {
 			.cra_name = "ecb(aes)",
 			.cra_driver_name = "safexcel-ecb-aes",
 			.cra_priority = 300,
 			.cra_flags = CRYPTO_ALG_TYPE_SKCIPHER | CRYPTO_ALG_ASYNC |
 				     CRYPTO_ALG_KERN_DRIVER_ONLY,
 			.cra_blocksize = AES_BLOCK_SIZE,
 			.cra_ctxsize = sizeof(struct safexcel_cipher_ctx),
 			.cra_alignmask = 0,
 			.cra_init = safexcel_skcipher_cra_init,
 			.cra_exit = safexcel_skcipher_cra_exit,
 			.cra_module = THIS_MODULE,
 		},
 	},
 };

 static int safexcel_cbc_aes_encrypt(struct skcipher_request *req)
 {
 	return safexcel_aes(req, SAFEXCEL_ENCRYPT,
 			    CONTEXT_CONTROL_CRYPTO_MODE_CBC);
 }

 static int safexcel_cbc_aes_decrypt(struct skcipher_request *req)
 {
 	return safexcel_aes(req, SAFEXCEL_DECRYPT,
 			    CONTEXT_CONTROL_CRYPTO_MODE_CBC);
 }

 struct safexcel_alg_template safexcel_alg_cbc_aes = {
 	.type = SAFEXCEL_ALG_TYPE_SKCIPHER,
 	.alg.skcipher = {
 		.setkey = safexcel_aes_setkey,
 		.encrypt = safexcel_cbc_aes_encrypt,
 		.decrypt = safexcel_cbc_aes_decrypt,
 		.min_keysize = AES_MIN_KEY_SIZE,
 		.max_keysize = AES_MAX_KEY_SIZE,
 		.ivsize = AES_BLOCK_SIZE,
 		.base = {
 			.cra_name = "cbc(aes)",
 			.cra_driver_name = "safexcel-cbc-aes",
 			.cra_priority = 300,
 			.cra_flags = CRYPTO_ALG_TYPE_SKCIPHER | CRYPTO_ALG_ASYNC |
 				     CRYPTO_ALG_KERN_DRIVER_ONLY,
 			.cra_blocksize = AES_BLOCK_SIZE,
 			.cra_ctxsize = sizeof(struct safexcel_cipher_ctx),
 			.cra_alignmask = 0,
 			.cra_init = safexcel_skcipher_cra_init,
 			.cra_exit = safexcel_skcipher_cra_exit,
 			.cra_module = THIS_MODULE,
 		},
 	},
 };
	/*
	* Copyright (C) 2017 Marvell
	*
	* Antoine Tenart <antoine.tenart@free-electrons.com>
	*
	* This file is licensed under the terms of the GNU General Public
	* License version 2. This program is licensed "as is" without any
	* warranty of any kind, whether express or implied.
	*/

	#include <linux/device.h>
	#include <linux/dma-mapping.h>
	#include <linux/dmapool.h>

	#include <crypto/aes.h>
	#include <crypto/skcipher.h>

	#include "safexcel.h"

	enum safexcel_cipher_direction {
	SAFEXCEL_ENCRYPT,
	SAFEXCEL_DECRYPT,
	};

	struct safexcel_cipher_ctx {
	struct safexcel_context base;
	struct safexcel_crypto_priv *priv;

	enum safexcel_cipher_direction direction;
	u32 mode;

	__le32 key[8];
	unsigned int key_len;
	};

	static void safexcel_cipher_token(struct safexcel_cipher_ctx *ctx,
	struct crypto_async_request *async,
	struct safexcel_command_desc *cdesc,
	u32 length)
	{
	struct skcipher_request *req = skcipher_request_cast(async);
	struct safexcel_token *token;
	unsigned offset = 0;

	if (ctx->mode == CONTEXT_CONTROL_CRYPTO_MODE_CBC) {
	offset = AES_BLOCK_SIZE / sizeof(u32);
	memcpy(cdesc->control_data.token, req->iv, AES_BLOCK_SIZE);

	cdesc->control_data.options \|= EIP197_OPTION_4_TOKEN_IV_CMD;
	}

	token = (struct safexcel_token *)(cdesc->control_data.token + offset);

	token[0].opcode = EIP197_TOKEN_OPCODE_DIRECTION;
	token[0].packet_length = length;
	token[0].stat = EIP197_TOKEN_STAT_LAST_PACKET;
	token[0].instructions = EIP197_TOKEN_INS_LAST \|
	EIP197_TOKEN_INS_TYPE_CRYTO \|
	EIP197_TOKEN_INS_TYPE_OUTPUT;
	}

	static int safexcel_aes_setkey(struct crypto_skcipher ctfm, const u8 key,
	unsigned int len)
	{
	struct crypto_tfm *tfm = crypto_skcipher_tfm(ctfm);
	struct safexcel_cipher_ctx *ctx = crypto_tfm_ctx(tfm);
	struct crypto_aes_ctx aes;
	int ret, i;

	ret = crypto_aes_expand_key(&aes, key, len);
	if (ret) {
	crypto_skcipher_set_flags(ctfm, CRYPTO_TFM_RES_BAD_KEY_LEN);
	return ret;
	}

	for (i = 0; i < len / sizeof(u32); i++) {
	if (ctx->key[i] != cpu_to_le32(aes.key_enc[i])) {
	ctx->base.needs_inv = true;
	break;
	}
	}

	for (i = 0; i < len / sizeof(u32); i++)
	ctx->key[i] = cpu_to_le32(aes.key_enc[i]);

	ctx->key_len = len;

	memzero_explicit(&aes, sizeof(aes));
	return 0;
	}

	static int safexcel_context_control(struct safexcel_cipher_ctx *ctx,
	struct safexcel_command_desc *cdesc)
	{
	struct safexcel_crypto_priv *priv = ctx->priv;
	int ctrl_size;

	if (ctx->direction == SAFEXCEL_ENCRYPT)
	cdesc->control_data.control0 \|= CONTEXT_CONTROL_TYPE_CRYPTO_OUT;
	else
	cdesc->control_data.control0 \|= CONTEXT_CONTROL_TYPE_CRYPTO_IN;

	cdesc->control_data.control0 \|= CONTEXT_CONTROL_KEY_EN;
	cdesc->control_data.control1 \|= ctx->mode;

	switch (ctx->key_len) {
	case AES_KEYSIZE_128:
	cdesc->control_data.control0 \|= CONTEXT_CONTROL_CRYPTO_ALG_AES128;
	ctrl_size = 4;
	break;
	case AES_KEYSIZE_192:
	cdesc->control_data.control0 \|= CONTEXT_CONTROL_CRYPTO_ALG_AES192;
	ctrl_size = 6;
	break;
	case AES_KEYSIZE_256:
	cdesc->control_data.control0 \|= CONTEXT_CONTROL_CRYPTO_ALG_AES256;
	ctrl_size = 8;
	break;
	default:
	dev_err(priv->dev, "aes keysize not supported: %u\n",
	ctx->key_len);
	return -EINVAL;
	}
	cdesc->control_data.control0 \|= CONTEXT_CONTROL_SIZE(ctrl_size);

	return 0;
	}

	static int safexcel_handle_result(struct safexcel_crypto_priv *priv, int ring,
	struct crypto_async_request *async,
	bool should_complete, int ret)
	{
	struct skcipher_request *req = skcipher_request_cast(async);
	struct safexcel_result_desc *rdesc;
	int ndesc = 0;

	*ret = 0;

	spin_lock_bh(&priv->ring[ring].egress_lock);
	do {
	rdesc = safexcel_ring_next_rptr(priv, &priv->ring[ring].rdr);
	if (IS_ERR(rdesc)) {
	dev_err(priv->dev,
	"cipher: result: could not retrieve the result descriptor\n");
	*ret = PTR_ERR(rdesc);
	break;
	}

	if (rdesc->result_data.error_code) {
	dev_err(priv->dev,
	"cipher: result: result descriptor error (%d)\n",
	rdesc->result_data.error_code);
	*ret = -EIO;
	}

	ndesc++;
	} while (!rdesc->last_seg);

	safexcel_complete(priv, ring);
	spin_unlock_bh(&priv->ring[ring].egress_lock);

	if (req->src == req->dst) {
	dma_unmap_sg(priv->dev, req->src,
	sg_nents_for_len(req->src, req->cryptlen),
	DMA_BIDIRECTIONAL);
	} else {
	dma_unmap_sg(priv->dev, req->src,
	sg_nents_for_len(req->src, req->cryptlen),
	DMA_TO_DEVICE);
	dma_unmap_sg(priv->dev, req->dst,
	sg_nents_for_len(req->dst, req->cryptlen),
	DMA_FROM_DEVICE);
	}

	*should_complete = true;

	return ndesc;
	}

	static int safexcel_aes_send(struct crypto_async_request *async,
	int ring, struct safexcel_request *request,
	int commands, int results)
	{
	struct skcipher_request *req = skcipher_request_cast(async);
	struct safexcel_cipher_ctx *ctx = crypto_tfm_ctx(req->base.tfm);
	struct safexcel_crypto_priv *priv = ctx->priv;
	struct safexcel_command_desc *cdesc;
	struct safexcel_result_desc *rdesc;
	struct scatterlist *sg;
	int nr_src, nr_dst, n_cdesc = 0, n_rdesc = 0, queued = req->cryptlen;
	int i, ret = 0;

	if (req->src == req->dst) {
	nr_src = dma_map_sg(priv->dev, req->src,
	sg_nents_for_len(req->src, req->cryptlen),
	DMA_BIDIRECTIONAL);
	nr_dst = nr_src;
	if (!nr_src)
	return -EINVAL;
	} else {
	nr_src = dma_map_sg(priv->dev, req->src,
	sg_nents_for_len(req->src, req->cryptlen),
	DMA_TO_DEVICE);
	if (!nr_src)
	return -EINVAL;

	nr_dst = dma_map_sg(priv->dev, req->dst,
	sg_nents_for_len(req->dst, req->cryptlen),
	DMA_FROM_DEVICE);
	if (!nr_dst) {
	dma_unmap_sg(priv->dev, req->src,
	sg_nents_for_len(req->src, req->cryptlen),
	DMA_TO_DEVICE);
	return -EINVAL;
	}
	}

	memcpy(ctx->base.ctxr->data, ctx->key, ctx->key_len);

	spin_lock_bh(&priv->ring[ring].egress_lock);

	/* command descriptors */
	for_each_sg(req->src, sg, nr_src, i) {
	int len = sg_dma_len(sg);

	/* Do not overflow the request */
	if (queued - len < 0)
	len = queued;

	cdesc = safexcel_add_cdesc(priv, ring, !n_cdesc, !(queued - len),
	sg_dma_address(sg), len, req->cryptlen,
	ctx->base.ctxr_dma);
	if (IS_ERR(cdesc)) {
	/* No space left in the command descriptor ring */
	ret = PTR_ERR(cdesc);
	goto cdesc_rollback;
	}
	n_cdesc++;

	if (n_cdesc == 1) {
	safexcel_context_control(ctx, cdesc);
	safexcel_cipher_token(ctx, async, cdesc, req->cryptlen);
	}

	queued -= len;
	if (!queued)
	break;
	}

	/* result descriptors */
	for_each_sg(req->dst, sg, nr_dst, i) {
	bool first = !i, last = (i == nr_dst - 1);
	u32 len = sg_dma_len(sg);

	rdesc = safexcel_add_rdesc(priv, ring, first, last,
	sg_dma_address(sg), len);
	if (IS_ERR(rdesc)) {
	/* No space left in the result descriptor ring */
	ret = PTR_ERR(rdesc);
	goto rdesc_rollback;
	}
	n_rdesc++;
	}

	spin_unlock_bh(&priv->ring[ring].egress_lock);

	request->req = &req->base;
	ctx->base.handle_result = safexcel_handle_result;

	*commands = n_cdesc;
	*results = n_rdesc;
	return 0;

	rdesc_rollback:
	for (i = 0; i < n_rdesc; i++)
	safexcel_ring_rollback_wptr(priv, &priv->ring[ring].rdr);
	cdesc_rollback:
	for (i = 0; i < n_cdesc; i++)
	safexcel_ring_rollback_wptr(priv, &priv->ring[ring].cdr);

	spin_unlock_bh(&priv->ring[ring].egress_lock);

	if (req->src == req->dst) {
	dma_unmap_sg(priv->dev, req->src,
	sg_nents_for_len(req->src, req->cryptlen),
	DMA_BIDIRECTIONAL);
	} else {
	dma_unmap_sg(priv->dev, req->src,
	sg_nents_for_len(req->src, req->cryptlen),
	DMA_TO_DEVICE);
	dma_unmap_sg(priv->dev, req->dst,
	sg_nents_for_len(req->dst, req->cryptlen),
	DMA_FROM_DEVICE);
	}

	return ret;
	}

	static int safexcel_handle_inv_result(struct safexcel_crypto_priv *priv,
	int ring,
	struct crypto_async_request *async,
	bool should_complete, int ret)
	{
	struct skcipher_request *req = skcipher_request_cast(async);
	struct safexcel_cipher_ctx *ctx = crypto_tfm_ctx(req->base.tfm);
	struct safexcel_result_desc *rdesc;
	int ndesc = 0, enq_ret;

	*ret = 0;

	spin_lock_bh(&priv->ring[ring].egress_lock);
	do {
	rdesc = safexcel_ring_next_rptr(priv, &priv->ring[ring].rdr);
	if (IS_ERR(rdesc)) {
	dev_err(priv->dev,
	"cipher: invalidate: could not retrieve the result descriptor\n");
	*ret = PTR_ERR(rdesc);
	break;
	}

	if (rdesc->result_data.error_code) {
	dev_err(priv->dev, "cipher: invalidate: result descriptor error (%d)\n",
	rdesc->result_data.error_code);
	*ret = -EIO;
	}

	ndesc++;
	} while (!rdesc->last_seg);

	safexcel_complete(priv, ring);
	spin_unlock_bh(&priv->ring[ring].egress_lock);

	if (ctx->base.exit_inv) {
	dma_pool_free(priv->context_pool, ctx->base.ctxr,
	ctx->base.ctxr_dma);

	*should_complete = true;

	return ndesc;
	}

	ring = safexcel_select_ring(priv);
	ctx->base.ring = ring;
	ctx->base.needs_inv = false;
	ctx->base.send = safexcel_aes_send;

	spin_lock_bh(&priv->ring[ring].queue_lock);
	enq_ret = crypto_enqueue_request(&priv->ring[ring].queue, async);
	spin_unlock_bh(&priv->ring[ring].queue_lock);

	if (enq_ret != -EINPROGRESS)
	*ret = enq_ret;

	if (!priv->ring[ring].need_dequeue)
	safexcel_dequeue(priv, ring);

	*should_complete = false;

	return ndesc;
	}

	static int safexcel_cipher_send_inv(struct crypto_async_request *async,
	int ring, struct safexcel_request *request,
	int commands, int results)
	{
	struct skcipher_request *req = skcipher_request_cast(async);
	struct safexcel_cipher_ctx *ctx = crypto_tfm_ctx(req->base.tfm);
	struct safexcel_crypto_priv *priv = ctx->priv;
	int ret;

	ctx->base.handle_result = safexcel_handle_inv_result;

	ret = safexcel_invalidate_cache(async, &ctx->base, priv,
	ctx->base.ctxr_dma, ring, request);
	if (unlikely(ret))
	return ret;

	*commands = 1;
	*results = 1;

	return 0;
	}

	static int safexcel_cipher_exit_inv(struct crypto_tfm *tfm)
	{
	struct safexcel_cipher_ctx *ctx = crypto_tfm_ctx(tfm);
	struct safexcel_crypto_priv *priv = ctx->priv;
	struct skcipher_request req;
	struct safexcel_inv_result result = { 0 };
	int ring = ctx->base.ring;

	memset(&req, 0, sizeof(struct skcipher_request));

	/* create invalidation request */
	init_completion(&result.completion);
	skcipher_request_set_callback(&req, CRYPTO_TFM_REQ_MAY_BACKLOG,
	safexcel_inv_complete, &result);

	skcipher_request_set_tfm(&req, __crypto_skcipher_cast(tfm));
	ctx = crypto_tfm_ctx(req.base.tfm);
	ctx->base.exit_inv = true;
	ctx->base.send = safexcel_cipher_send_inv;

	spin_lock_bh(&priv->ring[ring].queue_lock);
	crypto_enqueue_request(&priv->ring[ring].queue, &req.base);
	spin_unlock_bh(&priv->ring[ring].queue_lock);

	if (!priv->ring[ring].need_dequeue)
	safexcel_dequeue(priv, ring);

	wait_for_completion_interruptible(&result.completion);

	if (result.error) {
	dev_warn(priv->dev,
	"cipher: sync: invalidate: completion error %d\n",
	result.error);
	return result.error;
	}

	return 0;
	}

	static int safexcel_aes(struct skcipher_request *req,
	enum safexcel_cipher_direction dir, u32 mode)
	{
	struct safexcel_cipher_ctx *ctx = crypto_tfm_ctx(req->base.tfm);
	struct safexcel_crypto_priv *priv = ctx->priv;
	int ret, ring;

	ctx->direction = dir;
	ctx->mode = mode;

	if (ctx->base.ctxr) {
	if (ctx->base.needs_inv)
	ctx->base.send = safexcel_cipher_send_inv;
	} else {
	ctx->base.ring = safexcel_select_ring(priv);
	ctx->base.send = safexcel_aes_send;

	ctx->base.ctxr = dma_pool_zalloc(priv->context_pool,
	EIP197_GFP_FLAGS(req->base),
	&ctx->base.ctxr_dma);
	if (!ctx->base.ctxr)
	return -ENOMEM;
	}

	ring = ctx->base.ring;

	spin_lock_bh(&priv->ring[ring].queue_lock);
	ret = crypto_enqueue_request(&priv->ring[ring].queue, &req->base);
	spin_unlock_bh(&priv->ring[ring].queue_lock);

	if (!priv->ring[ring].need_dequeue)
	safexcel_dequeue(priv, ring);

	return ret;
	}

	static int safexcel_ecb_aes_encrypt(struct skcipher_request *req)
	{
	return safexcel_aes(req, SAFEXCEL_ENCRYPT,
	CONTEXT_CONTROL_CRYPTO_MODE_ECB);
	}

	static int safexcel_ecb_aes_decrypt(struct skcipher_request *req)
	{
	return safexcel_aes(req, SAFEXCEL_DECRYPT,
	CONTEXT_CONTROL_CRYPTO_MODE_ECB);
	}

	static int safexcel_skcipher_cra_init(struct crypto_tfm *tfm)
	{
	struct safexcel_cipher_ctx *ctx = crypto_tfm_ctx(tfm);
	struct safexcel_alg_template *tmpl =
	container_of(tfm->__crt_alg, struct safexcel_alg_template,
	alg.skcipher.base);

	ctx->priv = tmpl->priv;

	return 0;
	}

	static void safexcel_skcipher_cra_exit(struct crypto_tfm *tfm)
	{
	struct safexcel_cipher_ctx *ctx = crypto_tfm_ctx(tfm);
	struct safexcel_crypto_priv *priv = ctx->priv;
	int ret;

	memzero_explicit(ctx->key, 8 * sizeof(u32));

	/* context not allocated, skip invalidation */
	if (!ctx->base.ctxr)
	return;

	memzero_explicit(ctx->base.ctxr->data, 8 * sizeof(u32));

	ret = safexcel_cipher_exit_inv(tfm);
	if (ret)
	dev_warn(priv->dev, "cipher: invalidation error %d\n", ret);
	}

	struct safexcel_alg_template safexcel_alg_ecb_aes = {
	.type = SAFEXCEL_ALG_TYPE_SKCIPHER,
	.alg.skcipher = {
	.setkey = safexcel_aes_setkey,
	.encrypt = safexcel_ecb_aes_encrypt,
	.decrypt = safexcel_ecb_aes_decrypt,
	.min_keysize = AES_MIN_KEY_SIZE,
	.max_keysize = AES_MAX_KEY_SIZE,
	.base = {
	.cra_name = "ecb(aes)",
	.cra_driver_name = "safexcel-ecb-aes",
	.cra_priority = 300,
	.cra_flags = CRYPTO_ALG_TYPE_SKCIPHER \| CRYPTO_ALG_ASYNC \|
	CRYPTO_ALG_KERN_DRIVER_ONLY,
	.cra_blocksize = AES_BLOCK_SIZE,
	.cra_ctxsize = sizeof(struct safexcel_cipher_ctx),
	.cra_alignmask = 0,
	.cra_init = safexcel_skcipher_cra_init,
	.cra_exit = safexcel_skcipher_cra_exit,
	.cra_module = THIS_MODULE,
	},
	},
	};

	static int safexcel_cbc_aes_encrypt(struct skcipher_request *req)
	{
	return safexcel_aes(req, SAFEXCEL_ENCRYPT,
	CONTEXT_CONTROL_CRYPTO_MODE_CBC);
	}

	static int safexcel_cbc_aes_decrypt(struct skcipher_request *req)
	{
	return safexcel_aes(req, SAFEXCEL_DECRYPT,
	CONTEXT_CONTROL_CRYPTO_MODE_CBC);
	}

	struct safexcel_alg_template safexcel_alg_cbc_aes = {
	.type = SAFEXCEL_ALG_TYPE_SKCIPHER,
	.alg.skcipher = {
	.setkey = safexcel_aes_setkey,
	.encrypt = safexcel_cbc_aes_encrypt,
	.decrypt = safexcel_cbc_aes_decrypt,
	.min_keysize = AES_MIN_KEY_SIZE,
	.max_keysize = AES_MAX_KEY_SIZE,
	.ivsize = AES_BLOCK_SIZE,
	.base = {
	.cra_name = "cbc(aes)",
	.cra_driver_name = "safexcel-cbc-aes",
	.cra_priority = 300,
	.cra_flags = CRYPTO_ALG_TYPE_SKCIPHER \| CRYPTO_ALG_ASYNC \|
	CRYPTO_ALG_KERN_DRIVER_ONLY,
	.cra_blocksize = AES_BLOCK_SIZE,
	.cra_ctxsize = sizeof(struct safexcel_cipher_ctx),
	.cra_alignmask = 0,
	.cra_init = safexcel_skcipher_cra_init,
	.cra_exit = safexcel_skcipher_cra_exit,
	.cra_module = THIS_MODULE,
	},
	},
	};