On Tue, Sep 15, 2015 at 08:14:08PM +0300, Denis Kirjanov wrote:
> diff --git a/arch/powerpc/include/asm/msi_bitmap.h 
> b/arch/powerpc/include/asm/msi_bitmap.h
> index 97ac3f4..9a1d2fb 100644
> --- a/arch/powerpc/include/asm/msi_bitmap.h
> +++ b/arch/powerpc/include/asm/msi_bitmap.h
> @@ -19,6 +19,7 @@ struct msi_bitmap {
>       unsigned long           *bitmap;
>       spinlock_t              lock;
>       unsigned int            irq_count;
> +     bool            bitmap_from_slab;

Nitpick: same alignment for bitmap_from_slab with irq_count etc. (unless
it's just my mail client).

>  };
>  
>  int msi_bitmap_alloc_hwirqs(struct msi_bitmap *bmp, int num);
> diff --git a/arch/powerpc/sysdev/msi_bitmap.c 
> b/arch/powerpc/sysdev/msi_bitmap.c
> index 73b64c7..305ebe3 100644
> --- a/arch/powerpc/sysdev/msi_bitmap.c
> +++ b/arch/powerpc/sysdev/msi_bitmap.c
> @@ -11,6 +11,7 @@
>  #include <linux/slab.h>
>  #include <linux/kernel.h>
>  #include <linux/bitmap.h>
> +#include <linux/bootmem.h>
>  #include <asm/msi_bitmap.h>
>  #include <asm/setup.h>
>  
> @@ -122,7 +123,12 @@ int msi_bitmap_alloc(struct msi_bitmap *bmp, unsigned 
> int irq_count,
>       size = BITS_TO_LONGS(irq_count) * sizeof(long);
>       pr_debug("msi_bitmap: allocator bitmap size is 0x%x bytes\n", size);
>  
> -     bmp->bitmap = zalloc_maybe_bootmem(size, GFP_KERNEL);
> +     if (slab_is_available()) {
> +             bmp->bitmap = kzalloc(size, GFP_KERNEL);
> +             bmp->bitmap_from_slab = true;
> +     } else
> +             bmp->bitmap = memblock_virt_alloc(size, 0);

I don't think bmp->bitmap_from_slab is always initialised, so you need
to explicitly set it to false here.

> +
>       if (!bmp->bitmap) {
>               pr_debug("msi_bitmap: ENOMEM allocating allocator bitmap!\n");
>               return -ENOMEM;
> @@ -138,7 +144,8 @@ int msi_bitmap_alloc(struct msi_bitmap *bmp, unsigned int 
> irq_count,
>  
>  void msi_bitmap_free(struct msi_bitmap *bmp)
>  {
> -     /* we can't free the bitmap we don't know if it's bootmem etc. */
> +     if (bmp->bitmap_from_slab)
> +             kfree(bmp->bitmap);
>       of_node_put(bmp->of_node);
>       bmp->bitmap = NULL;
>  }
> @@ -200,11 +207,11 @@ static void __init test_basics(void)
>       WARN_ON(rc < 0 && rc % 128 != 0);
>  
>       msi_bitmap_free(&bmp);
> +     if (!bmp.bitmap_from_slab)
> +             kmemleak_not_leak(bmp.bitmap);

As I mentioned in the other thread, I think you can call
kmemleak_not_leak() immediately after memblock_virt_alloc() (together
with a comment that this is never going to be freed). That would match
the other kmemleak API uses throughout the kernel.

-- 
Catalin

_______________________________________________
Linuxppc-dev mailing list
Linuxppc-dev@lists.ozlabs.org
https://lists.ozlabs.org/listinfo/linuxppc-dev

Reply via email to