Adjust the function structure, one for the success path, the other for the failure path. Signed-off-by: Xishi Qiu <qiuxishi@xxxxxxxxxx> --- drivers/base/dma-contiguous.c | 16 +++++++++------- 1 files changed, 9 insertions(+), 7 deletions(-) diff --git a/drivers/base/dma-contiguous.c b/drivers/base/dma-contiguous.c index 1bcfaed..aa72f93 100644 --- a/drivers/base/dma-contiguous.c +++ b/drivers/base/dma-contiguous.c @@ -167,26 +167,28 @@ static __init struct cma *cma_create_area(unsigned long base_pfn, cma = kmalloc(sizeof *cma, GFP_KERNEL); if (!cma) - return ERR_PTR(-ENOMEM); + goto err; cma->base_pfn = base_pfn; cma->count = count; cma->bitmap = kzalloc(bitmap_size, GFP_KERNEL); if (!cma->bitmap) - goto no_mem; + goto err; ret = cma_activate_area(base_pfn, count); if (ret) - goto error; + goto err; pr_debug("%s: returned %p\n", __func__, (void *)cma); return cma; -error: - kfree(cma->bitmap); -no_mem: - kfree(cma); +err: + if (cma) { + if (cma->bitmap) + kfree(cma->bitmap); + kfree(cma); + } return ERR_PTR(ret); } -- 1.7.1 -- To unsubscribe, send a message with 'unsubscribe linux-mm' in the body to majordomo@xxxxxxxxx. For more info on Linux MM, see: http://www.linux-mm.org/ . Don't email: <a href=mailto:"dont@xxxxxxxxx"> email@xxxxxxxxx </a>