X-Git-Url: https://git.saurik.com/apple/xnu.git/blobdiff_plain/ff6e181ae92fc6f1e89841290f461d1f2f9badd9..d26ffc64f583ab2d29df48f13518685602bc8832:/osfmk/kern/kalloc.h

diff --git a/osfmk/kern/kalloc.h b/osfmk/kern/kalloc.h
index fefa9f69f..3c6f5804b 100644
--- a/osfmk/kern/kalloc.h
+++ b/osfmk/kern/kalloc.h
@@ -1,14 +1,19 @@
 /*
  * Copyright (c) 2000-2005 Apple Computer, Inc. All rights reserved.
  *
- * @APPLE_LICENSE_HEADER_START@
+ * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
  * 
  * This file contains Original Code and/or Modifications of Original Code
  * as defined in and that are subject to the Apple Public Source License
  * Version 2.0 (the 'License'). You may not use this file except in
- * compliance with the License. Please obtain a copy of the License at
- * http://www.opensource.apple.com/apsl/ and read it before using this
- * file.
+ * compliance with the License. The rights granted to you under the License
+ * may not be used to create, or enable the creation or redistribution of,
+ * unlawful or unlicensed copies of an Apple operating system, or to
+ * circumvent, violate, or enable the circumvention or violation of, any
+ * terms of an Apple operating system software license agreement.
+ * 
+ * Please obtain a copy of the License at
+ * http://www.opensource.apple.com/apsl/ and read it before using this file.
  * 
  * The Original Code and all software distributed under the License are
  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
@@ -18,7 +23,7 @@
  * Please see the License for the specific language governing rights and
  * limitations under the License.
  * 
- * @APPLE_LICENSE_HEADER_END@
+ * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
  */
 /*
  * @OSF_COPYRIGHT@
@@ -55,33 +60,108 @@
 #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 *kget(vm_size_t	size);
 
 extern void kfree(void		*data,
 		  vm_size_t	size);
 
-__END_DECLS
+#else /* XNU_KERNEL_PRIVATE */
 
-#ifdef	MACH_KERNEL_PRIVATE
+extern void *kalloc(vm_size_t	size) __attribute__((alloc_size(1)));
 
-#include <kern/lock.h>
+extern void *kalloc_noblock(vm_size_t	size) __attribute__((alloc_size(1)));
 
-#define KALLOC_MINSIZE		16
+extern void kfree(void		*data,
+		  vm_size_t	size);
+
+#endif /* !XNU_KERNEL_PRIVATE */
+
+__END_DECLS
+
+#ifdef	MACH_KERNEL_PRIVATE
 
 extern void		kalloc_init(void);
 
-extern void		krealloc(void		**addrp,
-				 vm_size_t	old_size,
-				 vm_size_t	new_size,
-				 simple_lock_t	lock);
+extern void		kalloc_fake_zone_init( int );
 
 extern void		kalloc_fake_zone_info(
 				int		*count,
@@ -89,10 +169,13 @@ extern void		kalloc_fake_zone_info(
 				vm_size_t	*max_size,
 				vm_size_t	*elem_size,
 				vm_size_t	*alloc_size,
+				uint64_t	*sum_size,
 				int		*collectable,
-				int		*exhaustable);
+				int		*exhaustable,
+				int		*caller_acct);
 
 extern vm_size_t kalloc_max_prerounded;
+extern vm_size_t kalloc_large_total;
 
 #endif	/* MACH_KERNEL_PRIVATE */