sys/compat/linuxkpi/common/include/linux/slab.h

   1 /*-
   2  * Copyright (c) 2010 Isilon Systems, Inc.
   3  * Copyright (c) 2010 iX Systems, Inc.
   4  * Copyright (c) 2010 Panasas, Inc.
   5  * Copyright (c) 2013-2017 Mellanox Technologies, Ltd.
   6  * All rights reserved.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions
  10  * are met:
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice unmodified, this list of conditions, and the following
  13  *    disclaimer.
  14  * 2. Redistributions in binary form must reproduce the above copyright
  15  *    notice, this list of conditions and the following disclaimer in the
  16  *    documentation and/or other materials provided with the distribution.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  19  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  20  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  21  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  22  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  23  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  24  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  25  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  27  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28  *
  29  * $FreeBSD$
  30  */
  31 #ifndef _LINUX_SLAB_H_
  32 #define _LINUX_SLAB_H_
  33
  34 #include <sys/param.h>
  35 #include <sys/systm.h>
  36 #include <sys/malloc.h>
  37 #include <sys/limits.h>
  38 #include <vm/uma.h>
  39
  40 #include <linux/types.h>
  41 #include <linux/gfp.h>
  42
  43 MALLOC_DECLARE(M_KMALLOC);
  44
  45 #define kvmalloc(size)                  kmalloc(size, 0)
  46 #define kzalloc(size, flags)            kmalloc(size, (flags) | __GFP_ZERO)
  47 #define kzalloc_node(size, flags, node) kmalloc(size, (flags) | __GFP_ZERO)
  48 #define kfree_const(ptr)                kfree(ptr)
  49 #define vzalloc(size)                   __vmalloc(size, GFP_KERNEL | __GFP_NOWARN | __GFP_ZERO, 0)
  50 #define vfree(arg)                      kfree(arg)
  51 #define kvfree(arg)                     kfree(arg)
  52 #define vmalloc_node(size, node)        __vmalloc(size, GFP_KERNEL, 0)
  53 #define vmalloc_user(size)              __vmalloc(size, GFP_KERNEL | __GFP_ZERO, 0)
  54 #define vmalloc(size)                   __vmalloc(size, GFP_KERNEL, 0)
  55 #define __kmalloc(...)                  kmalloc(__VA_ARGS__)
  56 #define kmalloc_node(chunk, flags, n)   kmalloc(chunk, flags)
  57
  58 /*
  59  * Prefix some functions with linux_ to avoid namespace conflict
  60  * with the OpenSolaris code in the kernel.
  61  */
  62 #define kmem_cache              linux_kmem_cache
  63 #define kmem_cache_create(...)  linux_kmem_cache_create(__VA_ARGS__)
  64 #define kmem_cache_alloc(...)   linux_kmem_cache_alloc(__VA_ARGS__)
  65 #define kmem_cache_free(...)    linux_kmem_cache_free(__VA_ARGS__)
  66 #define kmem_cache_destroy(...) linux_kmem_cache_destroy(__VA_ARGS__)
  67
  68 #define KMEM_CACHE(__struct, flags)                                     \
  69         linux_kmem_cache_create(#__struct, sizeof(struct __struct),     \
  70         __alignof(struct __struct), (flags), NULL)
  71
  72 typedef void linux_kmem_ctor_t (void *);
  73
  74 struct linux_kmem_cache {
  75         uma_zone_t cache_zone;
  76         linux_kmem_ctor_t *cache_ctor;
  77         unsigned cache_flags;
  78         unsigned cache_size;
  79 };
  80
  81 #define SLAB_HWCACHE_ALIGN      (1 << 0)
  82 #define SLAB_DESTROY_BY_RCU     (1 << 1)
  83 #define SLAB_RECLAIM_ACCOUNT    (1 << 2)
  84
  85 static inline gfp_t
  86 linux_check_m_flags(gfp_t flags)
  87 {
  88         const gfp_t m = M_NOWAIT | M_WAITOK;
  89
  90         /* make sure either M_NOWAIT or M_WAITOK is set */
  91         if ((flags & m) == 0)
  92                 flags |= M_NOWAIT;
  93         else if ((flags & m) == m)
  94                 flags &= ~M_WAITOK;
  95
  96         /* mask away LinuxKPI specific flags */
  97         return (flags & GFP_NATIVE_MASK);
  98 }
  99
 100 static inline void *
 101 kmalloc(size_t size, gfp_t flags)
 102 {
 103         return (malloc(size, M_KMALLOC, linux_check_m_flags(flags)));
 104 }
 105
 106 static inline void *
 107 kcalloc(size_t n, size_t size, gfp_t flags)
 108 {
 109         flags |= __GFP_ZERO;
 110         return (mallocarray(n, size, M_KMALLOC, linux_check_m_flags(flags)));
 111 }
 112
 113 static inline void *
 114 __vmalloc(size_t size, gfp_t flags, int other)
 115 {
 116         return (malloc(size, M_KMALLOC, linux_check_m_flags(flags)));
 117 }
 118
 119 static inline void *
 120 vmalloc_32(size_t size)
 121 {
 122         return (contigmalloc(size, M_KMALLOC, M_WAITOK, 0, UINT_MAX, 1, 1));
 123 }
 124
 125 static inline void *
 126 kmalloc_array(size_t n, size_t size, gfp_t flags)
 127 {
 128         return (mallocarray(n, size, M_KMALLOC, linux_check_m_flags(flags)));
 129 }
 130
 131 static inline void *
 132 krealloc(void *ptr, size_t size, gfp_t flags)
 133 {
 134         return (realloc(ptr, size, M_KMALLOC, linux_check_m_flags(flags)));
 135 }
 136
 137 static inline void
 138 kfree(const void *ptr)
 139 {
 140         free(__DECONST(void *, ptr), M_KMALLOC);
 141 }
 142
 143 extern struct linux_kmem_cache *linux_kmem_cache_create(const char *name,
 144     size_t size, size_t align, unsigned flags, linux_kmem_ctor_t *ctor);
 145
 146 static inline void *
 147 linux_kmem_cache_alloc(struct linux_kmem_cache *c, gfp_t flags)
 148 {
 149         return (uma_zalloc_arg(c->cache_zone, c,
 150             linux_check_m_flags(flags)));
 151 }
 152
 153 static inline void *
 154 kmem_cache_zalloc(struct linux_kmem_cache *c, gfp_t flags)
 155 {
 156         return (uma_zalloc_arg(c->cache_zone, c,
 157             linux_check_m_flags(flags | M_ZERO)));
 158 }
 159
 160 extern void linux_kmem_cache_free_rcu(struct linux_kmem_cache *, void *);
 161
 162 static inline void
 163 linux_kmem_cache_free(struct linux_kmem_cache *c, void *m)
 164 {
 165         if (unlikely(c->cache_flags & SLAB_DESTROY_BY_RCU))
 166                 linux_kmem_cache_free_rcu(c, m);
 167         else
 168                 uma_zfree(c->cache_zone, m);
 169 }
 170
 171 extern void linux_kmem_cache_destroy(struct linux_kmem_cache *);
 172
 173 #endif                                  /* _LINUX_SLAB_H_ */