crypto: zynqmp-sha - Fix partial block implementation

The zynqmp-sha partial block was based on an old design of the
partial block API where the leftover calculation was done in the
Crypto API.  As the leftover calculation is now done by the
algorithm, fix this by passing the partial blocks to the fallback.

Also zero the stack descriptors.

Fixes: 201e9ec3b6 ("crypto: zynqmp-sha - Use API partial block handling")
Signed-off-by: Herbert Xu <herbert@gondor.apana.org.au>
This commit is contained in:
Herbert Xu 2025-05-04 14:13:50 +08:00
parent 7c234e138c
commit ecd71c95a6
2 changed files with 23 additions and 11 deletions

View File

@ -59,7 +59,7 @@ static int zynqmp_sha_init_tfm(struct crypto_shash *hash)
return PTR_ERR(fallback_tfm); return PTR_ERR(fallback_tfm);
if (crypto_shash_descsize(hash) < if (crypto_shash_descsize(hash) <
crypto_shash_descsize(tfm_ctx->fbk_tfm)) { crypto_shash_statesize(tfm_ctx->fbk_tfm)) {
crypto_free_shash(fallback_tfm); crypto_free_shash(fallback_tfm);
return -EINVAL; return -EINVAL;
} }
@ -76,15 +76,24 @@ static void zynqmp_sha_exit_tfm(struct crypto_shash *hash)
crypto_free_shash(tfm_ctx->fbk_tfm); crypto_free_shash(tfm_ctx->fbk_tfm);
} }
static int zynqmp_sha_continue(struct shash_desc *desc,
struct shash_desc *fbdesc, int err)
{
err = err ?: crypto_shash_export(fbdesc, shash_desc_ctx(desc));
shash_desc_zero(fbdesc);
return err;
}
static int zynqmp_sha_init(struct shash_desc *desc) static int zynqmp_sha_init(struct shash_desc *desc)
{ {
struct zynqmp_sha_tfm_ctx *tctx = crypto_shash_ctx(desc->tfm); struct zynqmp_sha_tfm_ctx *tctx = crypto_shash_ctx(desc->tfm);
struct crypto_shash *fbtfm = tctx->fbk_tfm; struct crypto_shash *fbtfm = tctx->fbk_tfm;
SHASH_DESC_ON_STACK(fbdesc, fbtfm); SHASH_DESC_ON_STACK(fbdesc, fbtfm);
int err;
fbdesc->tfm = fbtfm; fbdesc->tfm = fbtfm;
return crypto_shash_init(fbdesc) ?: err = crypto_shash_init(fbdesc);
crypto_shash_export_core(fbdesc, shash_desc_ctx(desc)); return zynqmp_sha_continue(desc, fbdesc, err);
} }
static int zynqmp_sha_update(struct shash_desc *desc, const u8 *data, unsigned int length) static int zynqmp_sha_update(struct shash_desc *desc, const u8 *data, unsigned int length)
@ -92,11 +101,12 @@ static int zynqmp_sha_update(struct shash_desc *desc, const u8 *data, unsigned i
struct zynqmp_sha_tfm_ctx *tctx = crypto_shash_ctx(desc->tfm); struct zynqmp_sha_tfm_ctx *tctx = crypto_shash_ctx(desc->tfm);
struct crypto_shash *fbtfm = tctx->fbk_tfm; struct crypto_shash *fbtfm = tctx->fbk_tfm;
SHASH_DESC_ON_STACK(fbdesc, fbtfm); SHASH_DESC_ON_STACK(fbdesc, fbtfm);
int err;
fbdesc->tfm = fbtfm; fbdesc->tfm = fbtfm;
return crypto_shash_import_core(fbdesc, shash_desc_ctx(desc)) ?: err = crypto_shash_import(fbdesc, shash_desc_ctx(desc)) ?:
crypto_shash_update(fbdesc, data, length) ?: crypto_shash_update(fbdesc, data, length);
crypto_shash_export_core(fbdesc, shash_desc_ctx(desc)); return zynqmp_sha_continue(desc, fbdesc, err);
} }
static int zynqmp_sha_finup(struct shash_desc *desc, const u8 *data, unsigned int length, u8 *out) static int zynqmp_sha_finup(struct shash_desc *desc, const u8 *data, unsigned int length, u8 *out)
@ -106,7 +116,7 @@ static int zynqmp_sha_finup(struct shash_desc *desc, const u8 *data, unsigned in
SHASH_DESC_ON_STACK(fbdesc, fbtfm); SHASH_DESC_ON_STACK(fbdesc, fbtfm);
fbdesc->tfm = fbtfm; fbdesc->tfm = fbtfm;
return crypto_shash_import_core(fbdesc, shash_desc_ctx(desc)) ?: return crypto_shash_import(fbdesc, shash_desc_ctx(desc)) ?:
crypto_shash_finup(fbdesc, data, length, out); crypto_shash_finup(fbdesc, data, length, out);
} }
@ -160,16 +170,14 @@ static struct zynqmp_sha_drv_ctx sha3_drv_ctx = {
.digest = zynqmp_sha_digest, .digest = zynqmp_sha_digest,
.init_tfm = zynqmp_sha_init_tfm, .init_tfm = zynqmp_sha_init_tfm,
.exit_tfm = zynqmp_sha_exit_tfm, .exit_tfm = zynqmp_sha_exit_tfm,
.descsize = sizeof(struct sha3_state), .descsize = SHA3_384_EXPORT_SIZE,
.digestsize = SHA3_384_DIGEST_SIZE, .digestsize = SHA3_384_DIGEST_SIZE,
.base = { .base = {
.cra_name = "sha3-384", .cra_name = "sha3-384",
.cra_driver_name = "zynqmp-sha3-384", .cra_driver_name = "zynqmp-sha3-384",
.cra_priority = 300, .cra_priority = 300,
.cra_flags = CRYPTO_ALG_KERN_DRIVER_ONLY | .cra_flags = CRYPTO_ALG_KERN_DRIVER_ONLY |
CRYPTO_ALG_NEED_FALLBACK | CRYPTO_ALG_NEED_FALLBACK,
CRYPTO_AHASH_ALG_BLOCK_ONLY |
CRYPTO_AHASH_ALG_FINUP_MAX,
.cra_blocksize = SHA3_384_BLOCK_SIZE, .cra_blocksize = SHA3_384_BLOCK_SIZE,
.cra_ctxsize = sizeof(struct zynqmp_sha_tfm_ctx), .cra_ctxsize = sizeof(struct zynqmp_sha_tfm_ctx),
.cra_module = THIS_MODULE, .cra_module = THIS_MODULE,

View File

@ -9,15 +9,19 @@
#define SHA3_224_DIGEST_SIZE (224 / 8) #define SHA3_224_DIGEST_SIZE (224 / 8)
#define SHA3_224_BLOCK_SIZE (200 - 2 * SHA3_224_DIGEST_SIZE) #define SHA3_224_BLOCK_SIZE (200 - 2 * SHA3_224_DIGEST_SIZE)
#define SHA3_224_EXPORT_SIZE SHA3_STATE_SIZE + SHA3_224_BLOCK_SIZE + 1
#define SHA3_256_DIGEST_SIZE (256 / 8) #define SHA3_256_DIGEST_SIZE (256 / 8)
#define SHA3_256_BLOCK_SIZE (200 - 2 * SHA3_256_DIGEST_SIZE) #define SHA3_256_BLOCK_SIZE (200 - 2 * SHA3_256_DIGEST_SIZE)
#define SHA3_256_EXPORT_SIZE SHA3_STATE_SIZE + SHA3_256_BLOCK_SIZE + 1
#define SHA3_384_DIGEST_SIZE (384 / 8) #define SHA3_384_DIGEST_SIZE (384 / 8)
#define SHA3_384_BLOCK_SIZE (200 - 2 * SHA3_384_DIGEST_SIZE) #define SHA3_384_BLOCK_SIZE (200 - 2 * SHA3_384_DIGEST_SIZE)
#define SHA3_384_EXPORT_SIZE SHA3_STATE_SIZE + SHA3_384_BLOCK_SIZE + 1
#define SHA3_512_DIGEST_SIZE (512 / 8) #define SHA3_512_DIGEST_SIZE (512 / 8)
#define SHA3_512_BLOCK_SIZE (200 - 2 * SHA3_512_DIGEST_SIZE) #define SHA3_512_BLOCK_SIZE (200 - 2 * SHA3_512_DIGEST_SIZE)
#define SHA3_512_EXPORT_SIZE SHA3_STATE_SIZE + SHA3_512_BLOCK_SIZE + 1
#define SHA3_STATE_SIZE 200 #define SHA3_STATE_SIZE 200