mm: slub: add kernel address sanitizer support for slub allocator

[karo-tx-linux.git] / mm / slab_common.c

diff --git a/mm/slab_common.c b/mm/slab_common.c
index 429a4506b382bd473dcba59b2109551b306a0697..999bb3424d44df71eb9b92d3ae6da75287a391d4 100644 (file)
--- a/mm/slab_common.c
+++ b/mm/slab_common.c
@@ -898,6 +898,7 @@ void *kmalloc_order(size_t size, gfp_t flags, unsigned int order)
        page = alloc_kmem_pages(flags, order);
        ret = page ? page_address(page) : NULL;
        kmemleak_alloc(ret, size, 1, flags);
+       kasan_kmalloc_large(ret, size);
        return ret;
 }
 EXPORT_SYMBOL(kmalloc_order);
@@ -1077,8 +1078,10 @@ static __always_inline void *__do_krealloc(const void *p, size_t new_size,
        if (p)
                ks = ksize(p);
 
-       if (ks >= new_size)
+       if (ks >= new_size) {
+               kasan_krealloc((void *)p, new_size);
                return (void *)p;
+       }
 
        ret = kmalloc_track_caller(new_size, flags);
        if (ret && p)