1 /*
2 * Copyright (c) 2015-2020 François Tigeot <ftigeot@wolfpond.org>
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice unmodified, this list of conditions, and the following
10 * disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
16 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
19 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25 */
26
27 #ifndef _LINUX_SLAB_H_
28 #define _LINUX_SLAB_H_
29
30 #include <linux/gfp.h>
31 #include <linux/types.h>
32
33 MALLOC_DECLARE(M_DRM);
34
35 #define kzalloc(size, flags) kmalloc(size, M_DRM, flags | M_ZERO)
36
37 #undef kfree
38 #define kfree(ptr) do { \
39 if (ptr != NULL) \
40 _kfree((void *)ptr, M_DRM); \
41 } while (0)
42
43 #define kcalloc(n, size, flags) kzalloc((n) * (size), flags)
44
45 static inline void *
kmalloc_array(size_t n,size_t size,gfp_t flags)46 kmalloc_array(size_t n, size_t size, gfp_t flags)
47 {
48 return kmalloc(n * size, M_DRM, flags);
49 }
50
51 #include <linux/kasan.h>
52
53 struct kmem_cache {
54 unsigned int size;
55 };
56
57 #define SLAB_HWCACHE_ALIGN 0x01UL
58 #define SLAB_RECLAIM_ACCOUNT 0x02UL
59 #define SLAB_TYPESAFE_BY_RCU 0x04UL
60
61 #define KMEM_CACHE(__struct, flags) \
62 kmem_cache_create(#__struct, sizeof(struct __struct), \
63 __alignof(struct __struct), (flags), NULL)
64
65 static inline struct kmem_cache *
kmem_cache_create(const char * name,size_t size,size_t align,unsigned long flags,void (* ctor)(void *))66 kmem_cache_create(const char *name, size_t size, size_t align,
67 unsigned long flags, void (*ctor)(void *))
68 {
69 struct kmem_cache *kc;
70
71 kc = kzalloc(sizeof(struct kmem_cache), flags);
72 kc->size = size;
73
74 return kc;
75 }
76
77 static inline void
kmem_cache_destroy(struct kmem_cache * kc)78 kmem_cache_destroy(struct kmem_cache *kc)
79 {
80 kfree(kc);
81 }
82
83 static inline void *
kmem_cache_alloc(struct kmem_cache * kc,gfp_t flags)84 kmem_cache_alloc(struct kmem_cache *kc, gfp_t flags)
85 {
86 return kmalloc(kc->size, M_DRM, flags);
87 }
88
89 static inline void *
kmem_cache_zalloc(struct kmem_cache * kc,gfp_t flags)90 kmem_cache_zalloc(struct kmem_cache *kc, gfp_t flags)
91 {
92 return kzalloc(kc->size, flags);
93 }
94
95 static inline void
kmem_cache_free(struct kmem_cache * kc,void * ptr)96 kmem_cache_free(struct kmem_cache *kc, void *ptr)
97 {
98 kfree(ptr);
99 }
100
101 #endif /* _LINUX_SLAB_H_ */
102