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 typedef void linux_kmem_ctor_t (void *);
  69
  70 struct linux_kmem_cache {
  71         uma_zone_t cache_zone;
  72         linux_kmem_ctor_t *cache_ctor;
  73         unsigned cache_flags;
  74         unsigned cache_size;
  75 };
  76
  77 #define SLAB_HWCACHE_ALIGN      (1 << 0)
  78 #define SLAB_DESTROY_BY_RCU     (1 << 1)
  79 #define SLAB_RECLAIM_ACCOUNT    (1 << 2)
  80
  81 static inline gfp_t
  82 linux_check_m_flags(gfp_t flags)
  83 {
  84         const gfp_t m = M_NOWAIT | M_WAITOK;
  85
  86         /* make sure either M_NOWAIT or M_WAITOK is set */
  87         if ((flags & m) == 0)
  88                 flags |= M_NOWAIT;
  89         else if ((flags & m) == m)
  90                 flags &= ~M_WAITOK;
  91
  92         /* mask away LinuxKPI specific flags */
  93         return (flags & GFP_NATIVE_MASK);
  94 }
  95
  96 static inline void *
  97 kmalloc(size_t size, gfp_t flags)
  98 {
  99         return (malloc(size, M_KMALLOC, linux_check_m_flags(flags)));
 100 }
 101
 102 static inline void *
 103 kcalloc(size_t n, size_t size, gfp_t flags)
 104 {
 105         flags |= __GFP_ZERO;
 106         return (mallocarray(n, size, M_KMALLOC, linux_check_m_flags(flags)));
 107 }
 108
 109 static inline void *
 110 __vmalloc(size_t size, gfp_t flags, int other)
 111 {
 112         return (malloc(size, M_KMALLOC, linux_check_m_flags(flags)));
 113 }
 114
 115 static inline void *
 116 vmalloc_32(size_t size)
 117 {
 118         return (contigmalloc(size, M_KMALLOC, M_WAITOK, 0, UINT_MAX, 1, 1));
 119 }
 120
 121 static inline void *
 122 kmalloc_array(size_t n, size_t size, gfp_t flags)
 123 {
 124         return (mallocarray(n, size, M_KMALLOC, linux_check_m_flags(flags)));
 125 }
 126
 127 static inline void *
 128 krealloc(void *ptr, size_t size, gfp_t flags)
 129 {
 130         return (realloc(ptr, size, M_KMALLOC, linux_check_m_flags(flags)));
 131 }
 132
 133 static inline void
 134 kfree(const void *ptr)
 135 {
 136         free(__DECONST(void *, ptr), M_KMALLOC);
 137 }
 138
 139 extern struct linux_kmem_cache *linux_kmem_cache_create(const char *name,
 140     size_t size, size_t align, unsigned flags, linux_kmem_ctor_t *ctor);
 141
 142 static inline void *
 143 linux_kmem_cache_alloc(struct linux_kmem_cache *c, gfp_t flags)
 144 {
 145         return (uma_zalloc_arg(c->cache_zone, c,
 146             linux_check_m_flags(flags)));
 147 }
 148
 149 static inline void *
 150 kmem_cache_zalloc(struct linux_kmem_cache *c, gfp_t flags)
 151 {
 152         return (uma_zalloc_arg(c->cache_zone, c,
 153             linux_check_m_flags(flags | M_ZERO)));
 154 }
 155
 156 extern void linux_kmem_cache_free_rcu(struct linux_kmem_cache *, void *);
 157
 158 static inline void
 159 linux_kmem_cache_free(struct linux_kmem_cache *c, void *m)
 160 {
 161         if (unlikely(c->cache_flags & SLAB_DESTROY_BY_RCU))
 162                 linux_kmem_cache_free_rcu(c, m);
 163         else
 164                 uma_zfree(c->cache_zone, m);
 165 }
 166
 167 extern void linux_kmem_cache_destroy(struct linux_kmem_cache *);
 168
 169 #endif                                  /* _LINUX_SLAB_H_ */