Commit f051f95e authored by LABBE Corentin's avatar LABBE Corentin Committed by Herbert Xu

crypto: picoxcell - check return value of sg_nents_for_len

The sg_nents_for_len() function could fail, this patch add a check for
its return value.
In the same time, we remove sg_count() as it is used as an alias of
sg_nents_for_len.
Signed-off-by: default avatarLABBE Corentin <clabbe.montjoie@gmail.com>
Acked-by: default avatarJamie Iles <jamie@jamieiles.com>
Signed-off-by: default avatarHerbert Xu <herbert@gondor.apana.org.au>
parent 4fa9948c
...@@ -272,12 +272,6 @@ static unsigned spacc_load_ctx(struct spacc_generic_ctx *ctx, ...@@ -272,12 +272,6 @@ static unsigned spacc_load_ctx(struct spacc_generic_ctx *ctx,
return indx; return indx;
} }
/* Count the number of scatterlist entries in a scatterlist. */
static inline int sg_count(struct scatterlist *sg_list, int nbytes)
{
return sg_nents_for_len(sg_list, nbytes);
}
static inline void ddt_set(struct spacc_ddt *ddt, dma_addr_t phys, size_t len) static inline void ddt_set(struct spacc_ddt *ddt, dma_addr_t phys, size_t len)
{ {
ddt->p = phys; ddt->p = phys;
...@@ -300,7 +294,11 @@ static struct spacc_ddt *spacc_sg_to_ddt(struct spacc_engine *engine, ...@@ -300,7 +294,11 @@ static struct spacc_ddt *spacc_sg_to_ddt(struct spacc_engine *engine,
struct spacc_ddt *ddt; struct spacc_ddt *ddt;
int i; int i;
nents = sg_count(payload, nbytes); nents = sg_nents_for_len(payload, nbytes);
if (nents < 0) {
dev_err(engine->dev, "Invalid numbers of SG.\n");
return NULL;
}
mapped_ents = dma_map_sg(engine->dev, payload, nents, dir); mapped_ents = dma_map_sg(engine->dev, payload, nents, dir);
if (mapped_ents + 1 > MAX_DDT_LEN) if (mapped_ents + 1 > MAX_DDT_LEN)
...@@ -336,13 +334,21 @@ static int spacc_aead_make_ddts(struct aead_request *areq) ...@@ -336,13 +334,21 @@ static int spacc_aead_make_ddts(struct aead_request *areq)
if (req->is_encrypt) if (req->is_encrypt)
total += crypto_aead_authsize(aead); total += crypto_aead_authsize(aead);
src_nents = sg_count(areq->src, total); src_nents = sg_nents_for_len(areq->src, total);
if (src_nents < 0) {
dev_err(engine->dev, "Invalid numbers of src SG.\n");
return src_nents;
}
if (src_nents + 1 > MAX_DDT_LEN) if (src_nents + 1 > MAX_DDT_LEN)
return -E2BIG; return -E2BIG;
dst_nents = 0; dst_nents = 0;
if (areq->src != areq->dst) { if (areq->src != areq->dst) {
dst_nents = sg_count(areq->dst, total); dst_nents = sg_nents_for_len(areq->dst, total);
if (dst_nents < 0) {
dev_err(engine->dev, "Invalid numbers of dst SG.\n");
return dst_nents;
}
if (src_nents + 1 > MAX_DDT_LEN) if (src_nents + 1 > MAX_DDT_LEN)
return -E2BIG; return -E2BIG;
} }
...@@ -422,13 +428,22 @@ static void spacc_aead_free_ddts(struct spacc_req *req) ...@@ -422,13 +428,22 @@ static void spacc_aead_free_ddts(struct spacc_req *req)
(req->is_encrypt ? crypto_aead_authsize(aead) : 0); (req->is_encrypt ? crypto_aead_authsize(aead) : 0);
struct spacc_aead_ctx *aead_ctx = crypto_aead_ctx(aead); struct spacc_aead_ctx *aead_ctx = crypto_aead_ctx(aead);
struct spacc_engine *engine = aead_ctx->generic.engine; struct spacc_engine *engine = aead_ctx->generic.engine;
unsigned nents = sg_count(areq->src, total); int nents = sg_nents_for_len(areq->src, total);
/* sg_nents_for_len should not fail since it works when mapping sg */
if (unlikely(nents < 0)) {
dev_err(engine->dev, "Invalid numbers of src SG.\n");
return;
}
if (areq->src != areq->dst) { if (areq->src != areq->dst) {
dma_unmap_sg(engine->dev, areq->src, nents, DMA_TO_DEVICE); dma_unmap_sg(engine->dev, areq->src, nents, DMA_TO_DEVICE);
dma_unmap_sg(engine->dev, areq->dst, nents = sg_nents_for_len(areq->dst, total);
sg_count(areq->dst, total), if (unlikely(nents < 0)) {
DMA_FROM_DEVICE); dev_err(engine->dev, "Invalid numbers of dst SG.\n");
return;
}
dma_unmap_sg(engine->dev, areq->dst, nents, DMA_FROM_DEVICE);
} else } else
dma_unmap_sg(engine->dev, areq->src, nents, DMA_BIDIRECTIONAL); dma_unmap_sg(engine->dev, areq->src, nents, DMA_BIDIRECTIONAL);
...@@ -440,7 +455,12 @@ static void spacc_free_ddt(struct spacc_req *req, struct spacc_ddt *ddt, ...@@ -440,7 +455,12 @@ static void spacc_free_ddt(struct spacc_req *req, struct spacc_ddt *ddt,
dma_addr_t ddt_addr, struct scatterlist *payload, dma_addr_t ddt_addr, struct scatterlist *payload,
unsigned nbytes, enum dma_data_direction dir) unsigned nbytes, enum dma_data_direction dir)
{ {
unsigned nents = sg_count(payload, nbytes); int nents = sg_nents_for_len(payload, nbytes);
if (nents < 0) {
dev_err(req->engine->dev, "Invalid numbers of SG.\n");
return;
}
dma_unmap_sg(req->engine->dev, payload, nents, dir); dma_unmap_sg(req->engine->dev, payload, nents, dir);
dma_pool_free(req->engine->req_pool, ddt, ddt_addr); dma_pool_free(req->engine->req_pool, ddt, ddt_addr);
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment