2012-02-17 13:09:00

by Ajeet Yadav

[permalink] [raw]
Subject: [PATCH 2/2] ARM: dma-mapping: fix leak in consistent_init

Although the error in this case is unlikely, but logically
if error occurs then we leak memory.

Signed-off-by: Ajeet Yadav <[email protected]>
---
arch/arm/mm/dma-mapping.c | 24 ++++++++++++------------
1 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/arch/arm/mm/dma-mapping.c b/arch/arm/mm/dma-mapping.c
index 04bfa76..b8cf062 100644
--- a/arch/arm/mm/dma-mapping.c
+++ b/arch/arm/mm/dma-mapping.c
@@ -161,7 +161,6 @@ static struct arm_vmregion_head consistent_head = {
*/
static int __init consistent_init(void)
{
- int ret = 0;
pgd_t *pgd;
pud_t *pud;
pmd_t *pmd;
@@ -171,7 +170,7 @@ static int __init consistent_init(void)
unsigned long num_ptes = (CONSISTENT_END - base) >> PMD_SHIFT;

consistent_pte = kmalloc(num_ptes * sizeof(pte_t *), GFP_KERNEL);
- if (!consistent_pte) {
+ if (unlikely(!consistent_pte)) {
pr_err("%s: no memory\n", __func__);
return -ENOMEM;
}
@@ -183,32 +182,33 @@ static int __init consistent_init(void)
pgd = pgd_offset(&init_mm, base);

pud = pud_alloc(&init_mm, pgd, base);
- if (!pud) {
+ if (unlikely(!pud)) {
printk(KERN_ERR "%s: no pud tables\n", __func__);
- ret = -ENOMEM;
- break;
+ goto err;
}

pmd = pmd_alloc(&init_mm, pud, base);
- if (!pmd) {
+ if (unlikely(!pmd)) {
printk(KERN_ERR "%s: no pmd tables\n", __func__);
- ret = -ENOMEM;
- break;
+ goto err;
}
WARN_ON(!pmd_none(*pmd));

pte = pte_alloc_kernel(pmd, base);
- if (!pte) {
+ if (unlikely(!pte)) {
printk(KERN_ERR "%s: no pte tables\n", __func__);
- ret = -ENOMEM;
- break;
+ goto err;
}

consistent_pte[i++] = pte;
base += PMD_SIZE;
} while (base < CONSISTENT_END);

- return ret;
+ return 0;
+err:
+ kfree(consistent_pte);
+ consistent_pte = NULL;
+ return -ENOMEM;
}

core_initcall(consistent_init);
--
1.7.8.4


2012-02-17 13:13:48

by Russell King - ARM Linux

[permalink] [raw]
Subject: Re: [PATCH 2/2] ARM: dma-mapping: fix leak in consistent_init

On Fri, Feb 17, 2012 at 06:39:55PM +0530, Ajeet Yadav wrote:
> Although the error in this case is unlikely, but logically
> if error occurs then we leak memory.
>
> Signed-off-by: Ajeet Yadav <[email protected]>
> ---
> arch/arm/mm/dma-mapping.c | 24 ++++++++++++------------
> 1 files changed, 12 insertions(+), 12 deletions(-)
>
> diff --git a/arch/arm/mm/dma-mapping.c b/arch/arm/mm/dma-mapping.c
> index 04bfa76..b8cf062 100644
> --- a/arch/arm/mm/dma-mapping.c
> +++ b/arch/arm/mm/dma-mapping.c
> @@ -161,7 +161,6 @@ static struct arm_vmregion_head consistent_head = {
> */
> static int __init consistent_init(void)
> {
> - int ret = 0;
> pgd_t *pgd;
> pud_t *pud;
> pmd_t *pmd;
> @@ -171,7 +170,7 @@ static int __init consistent_init(void)
> unsigned long num_ptes = (CONSISTENT_END - base) >> PMD_SHIFT;
>
> consistent_pte = kmalloc(num_ptes * sizeof(pte_t *), GFP_KERNEL);
> - if (!consistent_pte) {
> + if (unlikely(!consistent_pte)) {

Please get rid of these unlikelys. This really isn't a performance
critical path.

2012-02-17 13:28:36

by Ajeet Yadav

[permalink] [raw]
Subject: Re: [PATCH 2/2] ARM: dma-mapping: fix leak in consistent_init

thank you for kind review, but how can I resent the patch using same mail chain

git send-email --smtp-encryption=tls --smtp-server=smtp.gmail.com
[email protected] --smtp-server-port=587 --from
"Ajeet Yadav <[email protected]>" --to "Russell King
<[email protected]>" --to "Jon Medhurst <[email protected]>" --to
"Nicolas Pitre <[email protected]>" --to "Catalin Marinas
<[email protected]>" --to "Sumit Bhattacharya
<[email protected]>" --to "[email protected]" --to
"[email protected]"
0002-ARM-dma-mapping-fix-leak-in-consistent_init.patch

On Fri, Feb 17, 2012 at 6:42 PM, Russell King - ARM Linux
<[email protected]> wrote:
> On Fri, Feb 17, 2012 at 06:39:55PM +0530, Ajeet Yadav wrote:
>> Although the error in this case is unlikely, but logically
>> if error occurs then we leak memory.
>>
>> Signed-off-by: Ajeet Yadav <[email protected]>
>> ---
>>  arch/arm/mm/dma-mapping.c |   24 ++++++++++++------------
>>  1 files changed, 12 insertions(+), 12 deletions(-)
>>
>> diff --git a/arch/arm/mm/dma-mapping.c b/arch/arm/mm/dma-mapping.c
>> index 04bfa76..b8cf062 100644
>> --- a/arch/arm/mm/dma-mapping.c
>> +++ b/arch/arm/mm/dma-mapping.c
>> @@ -161,7 +161,6 @@ static struct arm_vmregion_head consistent_head = {
>>   */
>>  static int __init consistent_init(void)
>>  {
>> -     int ret = 0;
>>       pgd_t *pgd;
>>       pud_t *pud;
>>       pmd_t *pmd;
>> @@ -171,7 +170,7 @@ static int __init consistent_init(void)
>>       unsigned long num_ptes = (CONSISTENT_END - base) >> PMD_SHIFT;
>>
>>       consistent_pte = kmalloc(num_ptes * sizeof(pte_t *), GFP_KERNEL);
>> -     if (!consistent_pte) {
>> +     if (unlikely(!consistent_pte)) {
>
> Please get rid of these unlikelys.  This really isn't a performance
> critical path.

2012-02-18 15:58:43

by Tixy

[permalink] [raw]
Subject: Re: [PATCH 2/2] ARM: dma-mapping: fix leak in consistent_init

On Fri, 2012-02-17 at 18:39 +0530, Ajeet Yadav wrote:
> Although the error in this case is unlikely, but logically
> if error occurs then we leak memory.
>
> Signed-off-by: Ajeet Yadav <[email protected]>

If you want to fix all the memory leaks then the page tables allocated
by pte_alloc_kernel() need freeing as well, (and the pud and pmd
tables?).

However, if we run out of memory this early in boot, then the system is
unusable anyway and it doesn't seem worth adding the extra code
complexity to avoid any of these memory leaks.

--
Tixy

> ---
> arch/arm/mm/dma-mapping.c | 24 ++++++++++++------------
> 1 files changed, 12 insertions(+), 12 deletions(-)
>
> diff --git a/arch/arm/mm/dma-mapping.c b/arch/arm/mm/dma-mapping.c
> index 04bfa76..b8cf062 100644
> --- a/arch/arm/mm/dma-mapping.c
> +++ b/arch/arm/mm/dma-mapping.c
> @@ -161,7 +161,6 @@ static struct arm_vmregion_head consistent_head = {
> */
> static int __init consistent_init(void)
> {
> - int ret = 0;
> pgd_t *pgd;
> pud_t *pud;
> pmd_t *pmd;
> @@ -171,7 +170,7 @@ static int __init consistent_init(void)
> unsigned long num_ptes = (CONSISTENT_END - base) >> PMD_SHIFT;
>
> consistent_pte = kmalloc(num_ptes * sizeof(pte_t *), GFP_KERNEL);
> - if (!consistent_pte) {
> + if (unlikely(!consistent_pte)) {
> pr_err("%s: no memory\n", __func__);
> return -ENOMEM;
> }
> @@ -183,32 +182,33 @@ static int __init consistent_init(void)
> pgd = pgd_offset(&init_mm, base);
>
> pud = pud_alloc(&init_mm, pgd, base);
> - if (!pud) {
> + if (unlikely(!pud)) {
> printk(KERN_ERR "%s: no pud tables\n", __func__);
> - ret = -ENOMEM;
> - break;
> + goto err;
> }
>
> pmd = pmd_alloc(&init_mm, pud, base);
> - if (!pmd) {
> + if (unlikely(!pmd)) {
> printk(KERN_ERR "%s: no pmd tables\n", __func__);
> - ret = -ENOMEM;
> - break;
> + goto err;
> }
> WARN_ON(!pmd_none(*pmd));
>
> pte = pte_alloc_kernel(pmd, base);
> - if (!pte) {
> + if (unlikely(!pte)) {
> printk(KERN_ERR "%s: no pte tables\n", __func__);
> - ret = -ENOMEM;
> - break;
> + goto err;
> }
>
> consistent_pte[i++] = pte;
> base += PMD_SIZE;
> } while (base < CONSISTENT_END);
>
> - return ret;
> + return 0;
> +err:
> + kfree(consistent_pte);
> + consistent_pte = NULL;
> + return -ENOMEM;
> }
>
> core_initcall(consistent_init);