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 | Z_SPRAYQTN | Z_SET_NOTEARLY, \ 61 VM_KERN_MEMORY_LIBKERN), NULL); \ 62 *__countp = (uint32_t)MIN(__kar.size / sizeof(ty), UINT32_MAX); \ 63 (ty *)__kar.addr; \ 64 }) 65 66 #define kreallocp_type_container(ty, ptr, old_count, countp, flags) ({ \ 67 uint32_t *__countp = (countp); \ 68 struct kalloc_result __kar; \ 69 static KALLOC_TYPE_VAR_DEFINE_3(kt_view_var, ty, KT_SHARED_ACCT); \ 70 __kar = krealloc_ext(kt_mangle_var_view(kt_view_var), ptr, \ 71 kt_size(0, sizeof(ty), old_count), \ 72 kt_size(0, sizeof(ty), *__countp), \ 73 Z_VM_TAG_BT(flags | Z_FULLSIZE | Z_SPRAYQTN | Z_SET_NOTEARLY, \ 74 VM_KERN_MEMORY_LIBKERN), NULL); \ 75 *__countp = (uint32_t)MIN(__kar.size / sizeof(ty), UINT32_MAX); \ 76 (ty *)__kar.addr; \ 77 }) 78 79 #if OSALLOCDEBUG 80 81 #if IOTRACKING 82 #define OSCONTAINER_ACCUMSIZE(s) do { OSAddAtomicLong((s), &debug_container_malloc_size); trackingAccumSize(s); } while(0) 83 #else 84 #define OSCONTAINER_ACCUMSIZE(s) do { OSAddAtomicLong((s), &debug_container_malloc_size); } while(0) 85 #endif 86 #define OSMETA_ACCUMSIZE(s) do { OSAddAtomicLong((s), &debug_container_malloc_size); } while(0) 87 #define OSIVAR_ACCUMSIZE(s) do { OSAddAtomicLong((s), &debug_ivars_size); } while(0) 88 89 #else /* OSALLOCDEBUG */ 90 91 #define OSCONTAINER_ACCUMSIZE(s) 92 #define OSMETA_ACCUMSIZE(s) 93 #define OSIVAR_ACCUMSIZE(s) 94 95 #endif /* !OSALLOCDEBUG */ 96 #endif /* XNU_KERNEL_PRIVATE */ 97 98 #ifndef NULL 99 #if defined (__cplusplus) 100 #if __cplusplus >= 201103L 101 #define NULL nullptr 102 #else 103 #define NULL 0 104 #endif 105 #else 106 #define NULL ((void *)0) 107 #endif 108 #endif 109 110 #endif /* _OS_OSLIB_H */ 111