kasan, vmalloc: add vmalloc tagging for SW_TAGS

[linux-2.6-microblaze.git] / mm / kasan / shadow.c

diff --git a/mm/kasan/shadow.c b/mm/kasan/shadow.c
index 39d0b32..5a866f6 100644 (file)
--- a/mm/kasan/shadow.c
+++ b/mm/kasan/shadow.c
@@ -475,12 +475,14 @@ void kasan_release_vmalloc(unsigned long start, unsigned long end,
        }
 }
 
-void __kasan_unpoison_vmalloc(const void *start, unsigned long size)
+void *__kasan_unpoison_vmalloc(const void *start, unsigned long size)
 {
        if (!is_vmalloc_or_module_addr(start))
-               return;
+               return (void *)start;
 
+       start = set_tag(start, kasan_random_tag());
        kasan_unpoison(start, size, false);
+       return (void *)start;
 }
 
 /*