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 kcalloc(n, size, flags)         kmalloc((n) * (size), (flags) | __GFP_ZERO)
  50 #define vzalloc(size)                   __vmalloc(size, GFP_KERNEL | __GFP_NOWARN | __GFP_ZERO, 0)
  51 #define vfree(arg)                      kfree(arg)
  52 #define kvfree(arg)                     kfree(arg)
  53 #define vmalloc_node(size, node)        __vmalloc(size, GFP_KERNEL, 0)
  54 #define vmalloc_user(size)              __vmalloc(size, GFP_KERNEL | __GFP_ZERO, 0)
  55 #define vmalloc(size)                   __vmalloc(size, GFP_KERNEL, 0)
  56 #define __kmalloc(...)                  kmalloc(__VA_ARGS__)
  57 #define kmalloc_node(chunk, flags, n)   kmalloc(chunk, flags)
  58
  59 /*
  60  * Prefix some functions with linux_ to avoid namespace conflict
  61  * with the OpenSolaris code in the kernel.
  62  */
  63 #define kmem_cache              linux_kmem_cache
  64 #define kmem_cache_create(...)  linux_kmem_cache_create(__VA_ARGS__)
  65 #define kmem_cache_alloc(...)   linux_kmem_cache_alloc(__VA_ARGS__)
  66 #define kmem_cache_free(...)    linux_kmem_cache_free(__VA_ARGS__)
  67 #define kmem_cache_destroy(...) linux_kmem_cache_destroy(__VA_ARGS__)
  68
  69 typedef void linux_kmem_ctor_t (void *);
  70
  71 struct linux_kmem_cache {
  72         uma_zone_t cache_zone;
  73         linux_kmem_ctor_t *cache_ctor;
  74         unsigned cache_flags;
  75         unsigned cache_size;
  76 };
  77
  78 #define SLAB_HWCACHE_ALIGN      (1 << 0)
  79 #define SLAB_DESTROY_BY_RCU     (1 << 1)
  80 #define SLAB_RECLAIM_ACCOUNT    (1 << 2)
  81
  82 static inline gfp_t
  83 linux_check_m_flags(gfp_t flags)
  84 {
  85         const gfp_t m = M_NOWAIT | M_WAITOK;
  86
  87         /* make sure either M_NOWAIT or M_WAITOK is set */
  88         if ((flags & m) == 0)
  89                 flags |= M_NOWAIT;
  90         else if ((flags & m) == m)
  91                 flags &= ~M_WAITOK;
  92
  93         /* mask away LinuxKPI specific flags */
  94         return (flags & GFP_NATIVE_MASK);
  95 }
  96
  97 static inline void *
  98 kmalloc(size_t size, gfp_t flags)
  99 {
 100         return (malloc(size, M_KMALLOC, linux_check_m_flags(flags)));
 101 }
 102
 103 static inline void *
 104 __vmalloc(size_t size, gfp_t flags, int other)
 105 {
 106         return (malloc(size, M_KMALLOC, linux_check_m_flags(flags)));
 107 }
 108
 109 static inline void *
 110 vmalloc_32(size_t size)
 111 {
 112         return (contigmalloc(size, M_KMALLOC, M_WAITOK, 0, UINT_MAX, 1, 1));
 113 }
 114
 115 static inline void *
 116 kmalloc_array(size_t n, size_t size, gfp_t flags)
 117 {
 118         if (size != 0 && n > (SIZE_MAX / size))
 119                 return (NULL);
 120         return (malloc(n * size, M_KMALLOC, linux_check_m_flags(flags)));
 121 }
 122
 123 static inline void *
 124 krealloc(void *ptr, size_t size, gfp_t flags)
 125 {
 126         return (realloc(ptr, size, M_KMALLOC, linux_check_m_flags(flags)));
 127 }
 128
 129 static inline void
 130 kfree(const void *ptr)
 131 {
 132         free(__DECONST(void *, ptr), M_KMALLOC);
 133 }
 134
 135 extern struct linux_kmem_cache *linux_kmem_cache_create(const char *name,
 136     size_t size, size_t align, unsigned flags, linux_kmem_ctor_t *ctor);
 137
 138 static inline void *
 139 linux_kmem_cache_alloc(struct linux_kmem_cache *c, gfp_t flags)
 140 {
 141         return (uma_zalloc_arg(c->cache_zone, c,
 142             linux_check_m_flags(flags)));
 143 }
 144
 145 static inline void *
 146 kmem_cache_zalloc(struct linux_kmem_cache *c, gfp_t flags)
 147 {
 148         return (uma_zalloc_arg(c->cache_zone, c,
 149             linux_check_m_flags(flags | M_ZERO)));
 150 }
 151
 152 extern void linux_kmem_cache_free_rcu(struct linux_kmem_cache *, void *);
 153
 154 static inline void
 155 linux_kmem_cache_free(struct linux_kmem_cache *c, void *m)
 156 {
 157         if (unlikely(c->cache_flags & SLAB_DESTROY_BY_RCU))
 158                 linux_kmem_cache_free_rcu(c, m);
 159         else
 160                 uma_zfree(c->cache_zone, m);
 161 }
 162
 163 extern void linux_kmem_cache_destroy(struct linux_kmem_cache *);
 164
 165 #endif                                  /* _LINUX_SLAB_H_ */