LXR linux/include/linux/kasan.h

   1#ifndef _LINUX_KASAN_H
   2#define _LINUX_KASAN_H
   3
   4#include <linux/sched.h>
   5#include <linux/types.h>
   6
   7struct kmem_cache;
   8struct page;
   9struct vm_struct;
  10
  11#ifdef CONFIG_KASAN
  12
  13#define KASAN_SHADOW_SCALE_SHIFT 3
  14
  15#include <asm/kasan.h>
  16#include <asm/pgtable.h>
  17
  18extern unsigned char kasan_zero_page[PAGE_SIZE];
  19extern pte_t kasan_zero_pte[PTRS_PER_PTE];
  20extern pmd_t kasan_zero_pmd[PTRS_PER_PMD];
  21extern pud_t kasan_zero_pud[PTRS_PER_PUD];
  22
  23void kasan_populate_zero_shadow(const void *shadow_start,
  24                                const void *shadow_end);
  25
  26static inline void *kasan_mem_to_shadow(const void *addr)
  27{
  28        return (void *)((unsigned long)addr >> KASAN_SHADOW_SCALE_SHIFT)
  29                + KASAN_SHADOW_OFFSET;
  30}
  31
  32/* Enable reporting bugs after kasan_disable_current() */
  33static inline void kasan_enable_current(void)
  34{
  35        current->kasan_depth++;
  36}
  37
  38/* Disable reporting bugs for current task */
  39static inline void kasan_disable_current(void)
  40{
  41        current->kasan_depth--;
  42}
  43
  44void kasan_unpoison_shadow(const void *address, size_t size);
  45
  46void kasan_unpoison_task_stack(struct task_struct *task);
  47void kasan_unpoison_stack_above_sp_to(const void *watermark);
  48
  49void kasan_alloc_pages(struct page *page, unsigned int order);
  50void kasan_free_pages(struct page *page, unsigned int order);
  51
  52void kasan_cache_create(struct kmem_cache *cache, size_t *size,
  53                        unsigned long *flags);
  54void kasan_cache_shrink(struct kmem_cache *cache);
  55void kasan_cache_destroy(struct kmem_cache *cache);
  56
  57void kasan_poison_slab(struct page *page);
  58void kasan_unpoison_object_data(struct kmem_cache *cache, void *object);
  59void kasan_poison_object_data(struct kmem_cache *cache, void *object);
  60void kasan_init_slab_obj(struct kmem_cache *cache, const void *object);
  61
  62void kasan_kmalloc_large(const void *ptr, size_t size, gfp_t flags);
  63void kasan_kfree_large(const void *ptr);
  64void kasan_poison_kfree(void *ptr);
  65void kasan_kmalloc(struct kmem_cache *s, const void *object, size_t size,
  66                  gfp_t flags);
  67void kasan_krealloc(const void *object, size_t new_size, gfp_t flags);
  68
  69void kasan_slab_alloc(struct kmem_cache *s, void *object, gfp_t flags);
  70bool kasan_slab_free(struct kmem_cache *s, void *object);
  71
  72struct kasan_cache {
  73        int alloc_meta_offset;
  74        int free_meta_offset;
  75};
  76
  77int kasan_module_alloc(void *addr, size_t size);
  78void kasan_free_shadow(const struct vm_struct *vm);
  79
  80size_t ksize(const void *);
  81static inline void kasan_unpoison_slab(const void *ptr) { ksize(ptr); }
  82size_t kasan_metadata_size(struct kmem_cache *cache);
  83
  84#else /* CONFIG_KASAN */
  85
  86static inline void kasan_unpoison_shadow(const void *address, size_t size) {}
  87
  88static inline void kasan_unpoison_task_stack(struct task_struct *task) {}
  89static inline void kasan_unpoison_stack_above_sp_to(const void *watermark) {}
  90
  91static inline void kasan_enable_current(void) {}
  92static inline void kasan_disable_current(void) {}
  93
  94static inline void kasan_alloc_pages(struct page *page, unsigned int order) {}
  95static inline void kasan_free_pages(struct page *page, unsigned int order) {}
  96
  97static inline void kasan_cache_create(struct kmem_cache *cache,
  98                                      size_t *size,
  99                                      unsigned long *flags) {}
 100static inline void kasan_cache_shrink(struct kmem_cache *cache) {}
 101static inline void kasan_cache_destroy(struct kmem_cache *cache) {}
 102
 103static inline void kasan_poison_slab(struct page *page) {}
 104static inline void kasan_unpoison_object_data(struct kmem_cache *cache,
 105                                        void *object) {}
 106static inline void kasan_poison_object_data(struct kmem_cache *cache,
 107                                        void *object) {}
 108static inline void kasan_init_slab_obj(struct kmem_cache *cache,
 109                                const void *object) {}
 110
 111static inline void kasan_kmalloc_large(void *ptr, size_t size, gfp_t flags) {}
 112static inline void kasan_kfree_large(const void *ptr) {}
 113static inline void kasan_poison_kfree(void *ptr) {}
 114static inline void kasan_kmalloc(struct kmem_cache *s, const void *object,
 115                                size_t size, gfp_t flags) {}
 116static inline void kasan_krealloc(const void *object, size_t new_size,
 117                                 gfp_t flags) {}
 118
 119static inline void kasan_slab_alloc(struct kmem_cache *s, void *object,
 120                                   gfp_t flags) {}
 121static inline bool kasan_slab_free(struct kmem_cache *s, void *object)
 122{
 123        return false;
 124}
 125
 126static inline int kasan_module_alloc(void *addr, size_t size) { return 0; }
 127static inline void kasan_free_shadow(const struct vm_struct *vm) {}
 128
 129static inline void kasan_unpoison_slab(const void *ptr) { }
 130static inline size_t kasan_metadata_size(struct kmem_cache *cache) { return 0; }
 131
 132#endif /* CONFIG_KASAN */
 133
 134#endif /* LINUX_KASAN_H */
 135