[PATCH 1/6] crypto: ccp - Apply appropriate gfp_t type to memory allocations

[Tom Lendacky]

Fix some memory allocations to use the appropriate gfp_t type based
on the CRYPTO_TFM_REQ_MAY_SLEEP flag.

Signed-off-by: Tom Lendacky <thomas.lenda...@amd.com>
---
 drivers/crypto/ccp/ccp-crypto-aes-cmac.c |    5 ++++-
 drivers/crypto/ccp/ccp-crypto-sha.c      |    5 ++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/drivers/crypto/ccp/ccp-crypto-aes-cmac.c 
b/drivers/crypto/ccp/ccp-crypto-aes-cmac.c
index 64dd35e..398832c 100644
--- a/drivers/crypto/ccp/ccp-crypto-aes-cmac.c
+++ b/drivers/crypto/ccp/ccp-crypto-aes-cmac.c
@@ -61,6 +61,7 @@ static int ccp_do_cmac_update(struct ahash_request *req, 
unsigned int nbytes,
        unsigned int block_size =
                crypto_tfm_alg_blocksize(crypto_ahash_tfm(tfm));
        unsigned int len, need_pad, sg_count;
+       gfp_t gfp;
        int ret;
 
        if (!ctx->u.aes.key_len)
@@ -99,7 +100,9 @@ static int ccp_do_cmac_update(struct ahash_request *req, 
unsigned int nbytes,
         * possible data pieces (buffer, input data, padding)
         */
        sg_count = (nbytes) ? sg_nents(req->src) + 2 : 2;
-       ret = sg_alloc_table(&rctx->data_sg, sg_count, GFP_KERNEL);
+       gfp = req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP ?
+               GFP_KERNEL : GFP_ATOMIC;
+       ret = sg_alloc_table(&rctx->data_sg, sg_count, gfp);
        if (ret)
                return ret;
 
diff --git a/drivers/crypto/ccp/ccp-crypto-sha.c 
b/drivers/crypto/ccp/ccp-crypto-sha.c
index b0881df..0571940 100644
--- a/drivers/crypto/ccp/ccp-crypto-sha.c
+++ b/drivers/crypto/ccp/ccp-crypto-sha.c
@@ -128,6 +128,7 @@ static int ccp_do_sha_update(struct ahash_request *req, 
unsigned int nbytes,
        unsigned int block_size =
                crypto_tfm_alg_blocksize(crypto_ahash_tfm(tfm));
        unsigned int len, sg_count;
+       gfp_t gfp;
        int ret;
 
        if (!final && ((nbytes + rctx->buf_count) <= block_size)) {
@@ -156,7 +157,9 @@ static int ccp_do_sha_update(struct ahash_request *req, 
unsigned int nbytes,
         * possible data pieces (hmac ipad, buffer, input data)
         */
        sg_count = (nbytes) ? sg_nents(req->src) + 2 : 2;
-       ret = sg_alloc_table(&rctx->data_sg, sg_count, GFP_KERNEL);
+       gfp = req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP ?
+               GFP_KERNEL : GFP_ATOMIC;
+       ret = sg_alloc_table(&rctx->data_sg, sg_count, gfp);
        if (ret)
                return ret;
 


--
To unsubscribe from this list: send the line "unsubscribe linux-crypto" in
the body of a message to majord...@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html