xnu-4570.71.2.tar.gz

[apple/xnu.git] / osfmk / kern / kalloc.h

diff --git a/osfmk/kern/kalloc.h b/osfmk/kern/kalloc.h
index 41fc204a32446e313b9d1885965a10ace05f7e2b..3c6f5804bb72dfc7015a138dde9b05d272d3a319 100644 (file)
--- a/osfmk/kern/kalloc.h
+++ b/osfmk/kern/kalloc.h
@@ -60,23 +60,105 @@
 #define _KERN_KALLOC_H_
 
 #include <mach/machine/vm_types.h>
+#include <mach/boolean.h>
 #include <sys/cdefs.h>
+#include <mach/vm_types.h>
 
 __BEGIN_DECLS
 
-extern void *kalloc(vm_size_t  size);
+#if XNU_KERNEL_PRIVATE
+
+extern void *
+kalloc_canblock(
+               vm_size_t                * size,
+               boolean_t              canblock,
+               vm_allocation_site_t * site);
+
+extern vm_size_t
+kalloc_size(
+               void                             * addr);
+
+extern vm_size_t
+kfree_addr(
+               void                             * addr);
+
+extern vm_size_t
+kalloc_bucket_size(
+               vm_size_t                          size);
+
+#define kalloc(size)                                \
+       ({ VM_ALLOC_SITE_STATIC(0, 0);                  \
+       vm_size_t tsize = (size);                       \
+       kalloc_canblock(&tsize, TRUE, &site); })
+
+#define kalloc_tag(size, itag)                      \
+       ({ VM_ALLOC_SITE_STATIC(0, (itag));             \
+       vm_size_t tsize = (size);                       \
+       kalloc_canblock(&tsize, TRUE, &site); })
+
+#define kalloc_tag_bt(size, itag)                   \
+       ({ VM_ALLOC_SITE_STATIC(VM_TAG_BT, (itag));     \
+       vm_size_t tsize = (size);                       \
+       kalloc_canblock(&tsize, TRUE, &site); })
+
+#define kalloc_noblock(size)                        \
+       ({ VM_ALLOC_SITE_STATIC(0, 0);                  \
+       vm_size_t tsize = (size);                       \
+       kalloc_canblock(&tsize, FALSE, &site); })
+
+#define kalloc_noblock_tag(size, itag)              \
+       ({ VM_ALLOC_SITE_STATIC(0, (itag));             \
+       vm_size_t tsize = (size);                       \
+       kalloc_canblock(&tsize, FALSE, &site); })
+
+#define kalloc_noblock_tag_bt(size, itag)           \
+       ({ VM_ALLOC_SITE_STATIC(VM_TAG_BT, (itag));     \
+       vm_size_t tsize = (size);                       \
+       kalloc_canblock(&tsize, FALSE, &site); })
+
+
+/* these versions update the size reference with the actual size allocated */
+
+#define kallocp(size)                               \
+       ({ VM_ALLOC_SITE_STATIC(0, 0);                  \
+       kalloc_canblock((size), TRUE, &site); })
+
+#define kallocp_tag(size, itag)                     \
+       ({ VM_ALLOC_SITE_STATIC(0, (itag));             \
+       kalloc_canblock((size), TRUE, &site); })
+
+#define kallocp_tag_bt(size, itag)                  \
+       ({ VM_ALLOC_SITE_STATIC(VM_TAG_BT, (itag));     \
+       kalloc_canblock((size), TRUE, &site); })
+
+#define kallocp_noblock(size)                       \
+       ({ VM_ALLOC_SITE_STATIC(0, 0);                  \
+       kalloc_canblock((size), FALSE, &site); })
+
+#define kallocp_noblock_tag_bt(size, itag)          \
+       ({ VM_ALLOC_SITE_STATIC(VM_TAG_BT, (itag));     \
+       kalloc_canblock((size), FALSE, &site); })
+
 
-extern void *kalloc_noblock(vm_size_t  size);
 
 extern void kfree(void         *data,
                  vm_size_t     size);
 
+#else /* XNU_KERNEL_PRIVATE */
+
+extern void *kalloc(vm_size_t  size) __attribute__((alloc_size(1)));
+
+extern void *kalloc_noblock(vm_size_t  size) __attribute__((alloc_size(1)));
+
+extern void kfree(void         *data,
+                 vm_size_t     size);
+
+#endif /* !XNU_KERNEL_PRIVATE */
+
 __END_DECLS
 
 #ifdef MACH_KERNEL_PRIVATE
 
-#include <kern/lock.h>
-
 extern void            kalloc_init(void);
 
 extern void            kalloc_fake_zone_init( int );