crypto: ccp - Set the AES size field for all modes

Ensure that the size field is correctly populated for
all AES modes.

Signed-off-by: Gary R Hook <gary.hook@amd.com>
Signed-off-by: Herbert Xu <herbert@gondor.apana.org.au>

1 files changed, 1 insertions, 2 deletions

diff --git a/drivers/crypto/ccp/ccp-dev-v5.c b/drivers/crypto/ccp/ccp-dev-v5.c
index 612898b4aaad..9c6ff8b813d9 100644
--- a/drivers/crypto/ccp/ccp-dev-v5.c
+++ b/drivers/crypto/ccp/ccp-dev-v5.c
@@ -284,8 +284,7 @@ static int ccp5_perform_aes(struct ccp_op *op)
         CCP_AES_ENCRYPT(&function) = op->u.aes.action;
         CCP_AES_MODE(&function) = op->u.aes.mode;
         CCP_AES_TYPE(&function) = op->u.aes.type;
-        if (op->u.aes.mode == CCP_AES_MODE_CFB)
-                CCP_AES_SIZE(&function) = 0x7f;
+        CCP_AES_SIZE(&function) = op->u.aes.size;
 
         CCP5_CMD_FUNCTION(&desc) = function.raw;