xref: /xnu-8792.61.2/libkern/libkern/c++/OSLib.h (revision 42e220869062b56f8d7d0726fd4c88954f87902c)
1 /*
2  * Copyright (c) 1999-2006 Apple Computer, Inc. All rights reserved.
3  *
4  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
5  *
6  * This file contains Original Code and/or Modifications of Original Code
7  * as defined in and that are subject to the Apple Public Source License
8  * Version 2.0 (the 'License'). You may not use this file except in
9  * compliance with the License. The rights granted to you under the License
10  * may not be used to create, or enable the creation or redistribution of,
11  * unlawful or unlicensed copies of an Apple operating system, or to
12  * circumvent, violate, or enable the circumvention or violation of, any
13  * terms of an Apple operating system software license agreement.
14  *
15  * Please obtain a copy of the License at
16  * http://www.opensource.apple.com/apsl/ and read it before using this file.
17  *
18  * The Original Code and all software distributed under the License are
19  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
20  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
21  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
22  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
23  * Please see the License for the specific language governing rights and
24  * limitations under the License.
25  *
26  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
27  */
28 #ifndef _OS_OSLIB_H
29 #define _OS_OSLIB_H
30 
31 #include <libkern/OSBase.h>
32 
33 #ifdef KERNEL
34 #define MACH_ASSERT 1
35 #endif
36 
37 __BEGIN_DECLS
38 
39 #include <stdarg.h>
40 #include <sys/systm.h>
41 
42 #include <kern/assert.h>
43 #ifdef KERNEL_PRIVATE
44 #include <kern/kalloc.h>
45 #endif
46 
47 __END_DECLS
48 
49 
50 #if XNU_KERNEL_PRIVATE
51 #include <libkern/OSAtomic.h>
52 #include <libkern/c++/OSCPPDebug.h>
53 
54 #define kallocp_type_container(ty, countp, flags) ({                           \
55 	uint32_t *__countp = (countp);                                         \
56 	struct kalloc_result __kar;                                            \
57 	static KALLOC_TYPE_VAR_DEFINE_3(kt_view_var, ty, KT_SHARED_ACCT);      \
58 	__kar = kalloc_ext(kt_mangle_var_view(kt_view_var),                    \
59 	    kt_size(0, sizeof(ty), *__countp),                                 \
60 	    Z_VM_TAG_BT(flags | Z_FULLSIZE, VM_KERN_MEMORY_LIBKERN), NULL);    \
61 	*__countp = (uint32_t)MIN(__kar.size / sizeof(ty), UINT32_MAX);        \
62 	(ty *)__kar.addr;                                                      \
63 })
64 
65 #define kreallocp_type_container(ty, ptr, old_count, countp, flags) ({         \
66 	uint32_t *__countp = (countp);                                         \
67 	struct kalloc_result __kar;                                            \
68 	static KALLOC_TYPE_VAR_DEFINE_3(kt_view_var, ty, KT_SHARED_ACCT);      \
69 	__kar = krealloc_ext(kt_mangle_var_view(kt_view_var), ptr,             \
70 	    kt_size(0, sizeof(ty), old_count),                                 \
71 	    kt_size(0, sizeof(ty), *__countp),                                 \
72 	    Z_VM_TAG_BT(flags | Z_FULLSIZE, VM_KERN_MEMORY_LIBKERN), NULL);    \
73 	*__countp = (uint32_t)MIN(__kar.size / sizeof(ty), UINT32_MAX);        \
74 	(ty *)__kar.addr;                                                      \
75 })
76 
77 #if OSALLOCDEBUG
78 
79 #if IOTRACKING
80 #define OSCONTAINER_ACCUMSIZE(s) do { OSAddAtomicLong((s), &debug_container_malloc_size); trackingAccumSize(s); } while(0)
81 #else
82 #define OSCONTAINER_ACCUMSIZE(s) do { OSAddAtomicLong((s), &debug_container_malloc_size); } while(0)
83 #endif
84 #define OSMETA_ACCUMSIZE(s)      do { OSAddAtomicLong((s), &debug_container_malloc_size); } while(0)
85 #define OSIVAR_ACCUMSIZE(s)      do { OSAddAtomicLong((s), &debug_ivars_size);            } while(0)
86 
87 #else /* OSALLOCDEBUG */
88 
89 #define OSCONTAINER_ACCUMSIZE(s)
90 #define OSMETA_ACCUMSIZE(s)
91 #define OSIVAR_ACCUMSIZE(s)
92 
93 #endif  /* !OSALLOCDEBUG */
94 #endif  /* XNU_KERNEL_PRIVATE */
95 
96 #ifndef NULL
97 #if defined (__cplusplus)
98 #if __cplusplus >= 201103L
99 #define NULL nullptr
100 #else
101 #define NULL 0
102 #endif
103 #else
104 #define NULL ((void *)0)
105 #endif
106 #endif
107 
108 #endif /* _OS_OSLIB_H  */
109