+#include <mach/boolean.h>
+#include <sys/cdefs.h>
+#include <mach/vm_types.h>
+
+__BEGIN_DECLS
+
+#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 kfree(void *data,
+ vm_size_t size);
+
+#define kfree(data, size) \
+_Pragma("clang diagnostic push") \
+_Pragma("clang diagnostic ignored \"-Wshadow\"") \
+ do { \
+ _Static_assert(sizeof (data) == sizeof (void *) || sizeof (data) == sizeof (mach_vm_address_t), "data is not a pointer"); \
+ void *__tmp_addr = (void *) data; \
+ vm_size_t __tmp_size = size; \
+ data = (__typeof__(data)) NULL; \
+ (kfree)(__tmp_addr, __tmp_size); \
+ } while (0) \
+_Pragma("clang diagnostic pop")
+
+#define kfree_addr(addr) \
+_Pragma("clang diagnostic push") \
+_Pragma("clang diagnostic ignored \"-Wshadow\"") \
+ do { \
+ _Static_assert(sizeof (addr) == sizeof (void *) || sizeof (addr) == sizeof (mach_vm_address_t), "addr is not a pointer"); \
+ void *__tmp_addr = (void *) addr; \
+ addr = (__typeof__(addr)) NULL; \
+ (kfree_addr)(__tmp_addr); \
+ } while (0) \
+_Pragma("clang diagnostic pop")
+
+#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);