Make all block allocations use the same func [2/2]
[akaros.git] / kern / include / kmalloc.h
1 /* Copyright (c) 2009 The Regents of the University of California. 
2  * See the COPYRIGHT files at the top of this source tree for full 
3  * license information.
4  * 
5  * Kevin Klues <klueska@cs.berkeley.edu>    
6  */
7
8 #pragma once
9
10 #include <ros/common.h>
11 #include <kref.h>
12
13 #define NUM_KMALLOC_CACHES 13
14 #define KMALLOC_ALIGNMENT 16
15 #define KMALLOC_SMALLEST (sizeof(struct kmalloc_tag) << 1)
16 #define KMALLOC_LARGEST KMALLOC_SMALLEST << NUM_KMALLOC_CACHES
17
18 void kmalloc_init(void);
19 void *kmalloc(size_t size, int flags);
20 void *kmalloc_array(size_t nmemb, size_t size, int flags);
21 void *kzmalloc(size_t size, int flags);
22 void *kmalloc_align(size_t size, int flags, size_t align);
23 void *kzmalloc_align(size_t size, int flags, size_t align);
24 void *krealloc(void *buf, size_t size, int flags);
25 void *kreallocarray(void *buf, size_t nmemb, size_t size, int flags);
26 int kmalloc_refcnt(void *buf);
27 void kmalloc_incref(void *buf);
28 void kfree(void *buf);
29 void kmalloc_canary_check(char *str);
30 void *debug_canary;
31
32 #define MEM_ATOMIC                              (1 << 1)
33 #define MEM_WAIT                                (1 << 2)
34 #define MEM_ERROR                               (1 << 3)
35
36 /* Kmalloc tag flags looks like this:
37  *
38  * +--------------28---------------+-----4------+
39  * |       Flag specific data      |    Flags   |
40  * +-------------------------------+------------+
41  */
42 #define KMALLOC_TAG_CACHE               1       /* memory came from slabs */
43 #define KMALLOC_TAG_PAGES               2       /* memory came from page allocator */
44 #define KMALLOC_TAG_UNALIGN             3       /* not a real tag, jump back by offset */
45 #define KMALLOC_ALIGN_SHIFT             4       /* max flag is 16 */
46 #define KMALLOC_FLAG_MASK               ((1 << KMALLOC_ALIGN_SHIFT) - 1)
47
48 #define KMALLOC_CANARY 0xdeadbabe
49
50 /* The kmalloc align/free paths require that flags is at the end of this
51  * struct, and that it is not padded. */
52 struct kmalloc_tag {
53         union {
54                 struct kmem_cache *my_cache;
55                 size_t num_pages;
56                 uint64_t unused_force_align;
57         };
58         struct kref kref;
59         uint32_t canary;
60         int flags;
61 };