[PATCH 4/9] crypto: caam/jr - add support for more XTS key lengths

[Andrei Botila]

From: Andrei Botila <andrei.bot...@nxp.com>

CAAM accelerator only supports XTS-AES-128 and XTS-AES-256 since
it adheres strictly to the standard. All the other key lengths
are accepted and processed through a fallback as long as they pass
the xts_verify_key() checks.

Fixes: c6415a6016bf ("crypto: caam - add support for acipher xts(aes)")
Cc: <sta...@vger.kernel.org> # v4.4+
Signed-off-by: Andrei Botila <andrei.bot...@nxp.com>
---
 drivers/crypto/caam/caamalg.c | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/drivers/crypto/caam/caamalg.c b/drivers/crypto/caam/caamalg.c
index ebf4dc87ca2e..a5447ae430b0 100644
--- a/drivers/crypto/caam/caamalg.c
+++ b/drivers/crypto/caam/caamalg.c
@@ -57,6 +57,7 @@
 #include "key_gen.h"
 #include "caamalg_desc.h"
 #include <crypto/engine.h>
+#include <crypto/xts.h>
 #include <asm/unaligned.h>
 
 /*
@@ -835,9 +836,10 @@ static int xts_skcipher_setkey(struct crypto_skcipher 
*skcipher, const u8 *key,
        u32 *desc;
        int err;
 
-       if (keylen != 2 * AES_MIN_KEY_SIZE  && keylen != 2 * AES_MAX_KEY_SIZE) {
+       err = xts_verify_key(skcipher, key, keylen);
+       if (err) {
                dev_dbg(jrdev, "key size mismatch\n");
-               return -EINVAL;
+               return err;
        }
 
        err = crypto_skcipher_setkey(ctx->fallback, key, keylen);
@@ -1790,7 +1792,9 @@ static inline int skcipher_crypt(struct skcipher_request 
*req, bool encrypt)
        if (!req->cryptlen)
                return 0;
 
-       if (ctx->fallback && xts_skcipher_ivsize(req)) {
+       if (ctx->fallback && (xts_skcipher_ivsize(req) ||
+                             (ctx->cdata.keylen != 2 * AES_KEYSIZE_128 &&
+                              ctx->cdata.keylen != 2 * AES_KEYSIZE_256))) {
                struct caam_skcipher_req_ctx *rctx = skcipher_request_ctx(req);
 
                skcipher_request_set_tfm(&rctx->fallback_req, ctx->fallback);
-- 
2.17.1