2 * Copyright (c) 2008-2009, Apple Inc.
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 * 3. Neither the name of Apple Inc. ("Apple") nor the names of
15 * its contributors may be used to endorse or promote products derived
16 * from this software without specific prior written permission.
18 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
19 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
20 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
21 * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
22 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
23 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
24 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
25 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
27 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30 #ifndef _SECURITY_AUDIT_AUDIT_BSD_H
31 #define _SECURITY_AUDIT_AUDIT_BSD_H
33 #include <sys/cdefs.h>
34 #include <machine/endian.h>
36 #if defined(_KERNEL) || defined(KERNEL)
37 #include <kern/kalloc.h>
43 #ifdef AUDIT_KASSERT_DEBUG
44 #define KASSERT(exp, msg) do { \
45 if (__builtin_expect(!(exp), 0)) { \
46 printf("%s:%d KASSERT failed: ", __FILE__, __LINE__); \
52 #define KASSERT(exp, msg) do { \
53 if (__builtin_expect(!(exp), 0)) \
57 #endif /* DIAGNOSTIC */
59 #define AU_MAX_LCK_NAME 32
61 #if __DARWIN_BYTE_ORDER == __DARWIN_BIG_ENDIAN
62 #define be16enc(p, d) *(p) = (d)
63 #define be32enc(p, d) *(p) = (d)
64 #define be64enc(p, d) *(p) = (d)
66 #else /* __DARWIN_BYTE_ORDER == __DARWIN_LITTLE_ENDIAN */
68 #include <libkern/OSByteOrder.h>
70 #define be16enc(p, d) OSWriteSwapInt16(p, 0, d)
71 #define be32enc(p, d) OSWriteSwapInt32(p, 0, d)
72 #define be64enc(p, d) OSWriteSwapInt64(p, 0, d)
73 #endif /* __DARWIN_BYTE_ORDER == __DARWIN_LITTLE_ENDIAN */
76 * BSD kernel memory allocation.
78 #define AUDIT_MALLOC_DEBUG 0 /* Change to 1 for malloc debugging. */
80 #define M_AUDITUNKNOWN 0
85 #define M_AUDITEVCLASS 5
86 #define M_AUDIT_PIPE 6
87 #define M_AUDIT_PIPE_ENTRY 7
88 #define M_AUDIT_PIPE_PRESELECT 8
89 #define M_AU_SESSION 9
90 #define M_AU_EV_PLIST 10
92 #define NUM_MALLOC_TYPES 11
96 #define M_WAITOK 0x0000 /* ok to block */
101 #define M_NOWAIT 0x0001 /* do not block */
105 #define M_ZERO 0x0004 /* bzero the allocation */
110 #define M_MAGIC 877983977
115 #if AUDIT_MALLOC_DEBUG
116 struct au_malloc_type
{
123 const char *mt_shortdesc
;
124 const char *mt_lastcaller
;
126 typedef struct au_malloc_type au_malloc_type_t
;
128 #define MALLOC_DEFINE(type, shortdesc, longdesc) \
129 au_malloc_type_t audit_##type[1] = { \
130 { 0, 0, 0, 0, (type < NUM_MALLOC_TYPES) ? type :\
131 M_AUDITUNKNOWN, M_MAGIC, shortdesc, NULL } \
134 extern au_malloc_type_t
*audit_malloc_types
[];
138 struct au_malloc_type
{
140 const char *mt_shortdesc
;
142 typedef struct au_malloc_type au_malloc_type_t
;
144 #define MALLOC_DEFINE(type, shortdesc, longdesc) \
145 au_malloc_type_t audit_##type[1] = { \
146 {M_MAGIC, shortdesc } \
149 #endif /* AUDIT_MALLOC_DEBUG */
151 #ifdef MALLOC_DECLARE
152 #undef MALLOC_DECLARE
154 #define MALLOC_DECLARE(type) \
155 extern au_malloc_type_t audit_##type[]
157 #if AUDIT_MALLOC_DEBUG
158 #define malloc(sz, tp, fl) _audit_malloc(sz, audit_##tp, fl, __FUNCTION__)
159 void *_audit_malloc(size_t size
, au_malloc_type_t
*type
, int flags
,
162 #define malloc(sz, tp, fl) _audit_malloc(sz, audit_##tp, fl)
163 void *_audit_malloc(size_t size
, au_malloc_type_t
*type
, int flags
);
166 #define free(ad, tp) _audit_free(ad, audit_##tp)
167 void _audit_free(void *addr
, au_malloc_type_t
*type
);
170 * BSD condition variable.
173 const char *cv_description
;
183 char mtx_name
[AU_MAX_LCK_NAME
];
193 char rw_name
[AU_MAX_LCK_NAME
];
205 char sl_name
[AU_MAX_LCK_NAME
];
217 char rl_name
[AU_MAX_LCK_NAME
];
222 * BSD condition variables functions.
224 void _audit_cv_init(struct cv
*cvp
, const char *desc
);
225 void _audit_cv_destroy(struct cv
*cvp
);
226 void _audit_cv_signal(struct cv
*cvp
);
227 void _audit_cv_broadcast(struct cv
*cvp
);
228 void _audit_cv_wait(struct cv
*cvp
, lck_mtx_t
*mp
, const char *desc
);
229 int _audit_cv_wait_sig(struct cv
*cvp
, lck_mtx_t
*mp
, const char *desc
);
230 int _audit_cv_wait_continuation(struct cv
*cvp
, lck_mtx_t
*mp
,
231 thread_continue_t function
);
232 #define cv_init(cvp, desc) _audit_cv_init(cvp, desc)
233 #define cv_destroy(cvp) _audit_cv_destroy(cvp)
234 #define cv_signal(cvp) _audit_cv_signal(cvp)
235 #define cv_broadcast(cvp) _audit_cv_broadcast(cvp)
236 #define cv_broadcastpri(cvp, pri) _audit_cv_broadcast(cvp)
237 #define cv_wait(cvp, mp) _audit_cv_wait(cvp, (mp)->mtx_lock, #cvp)
238 #define cv_wait_sig(cvp, mp) _audit_cv_wait_sig(cvp, (mp)->mtx_lock, #cvp)
239 #define cv_wait_continuation(cvp, mp, f) \
240 _audit_cv_wait_continuation(cvp, (mp)->mtx_lock, f)
245 void _audit_mtx_init(struct mtx
*mp
, const char *name
);
246 void _audit_mtx_destroy(struct mtx
*mp
);
247 #define mtx_init(mp, name, type, opts) \
248 _audit_mtx_init(mp, name)
249 #define mtx_lock(mp) lck_mtx_lock((mp)->mtx_lock)
250 #define mtx_unlock(mp) lck_mtx_unlock((mp)->mtx_lock)
251 #define mtx_destroy(mp) _audit_mtx_destroy(mp)
252 #define mtx_yield(mp) lck_mtx_yield((mp)->mtx_lock)
255 * Sleep lock functions.
257 void _audit_slck_init(struct slck
*lp
, const char *grpname
);
258 wait_result_t
_audit_slck_lock(struct slck
*lp
, int intr
);
259 void _audit_slck_unlock(struct slck
*lp
);
260 int _audit_slck_trylock(struct slck
*lp
);
261 void _audit_slck_assert(struct slck
*lp
, u_int
assert);
262 void _audit_slck_destroy(struct slck
*lp
);
263 #define slck_init(lp, name) _audit_slck_init((lp), (name))
264 #define slck_lock(lp) _audit_slck_lock((lp), 0)
265 #define slck_lock_sig(lp) (_audit_slck_lock((lp), 1) != THREAD_AWAKENED)
266 #define slck_unlock(lp) _audit_slck_unlock((lp))
267 #define slck_destroy(lp) _audit_slck_destroy((lp))
270 * Recursive lock functions.
272 void _audit_rlck_init(struct rlck
*lp
, const char *grpname
);
273 void _audit_rlck_lock(struct rlck
*lp
);
274 void _audit_rlck_unlock(struct rlck
*lp
);
275 void _audit_rlck_assert(struct rlck
*lp
, u_int
assert);
276 void _audit_rlck_destroy(struct rlck
*lp
);
277 #define rlck_init(lp, name) _audit_rlck_init((lp), (name))
278 #define rlck_lock(lp) _audit_rlck_lock((lp))
279 #define rlck_unlock(lp) _audit_rlck_unlock((lp))
280 #define rlck_destroy(lp) _audit_rlck_destroy((lp))
285 void _audit_rw_init(struct rwlock
*lp
, const char *name
);
286 void _audit_rw_destroy(struct rwlock
*lp
);
287 #define rw_init(lp, name) _audit_rw_init(lp, name)
288 #define rw_rlock(lp) lck_rw_lock_shared((lp)->rw_lock)
289 #define rw_runlock(lp) lck_rw_unlock_shared((lp)->rw_lock)
290 #define rw_wlock(lp) lck_rw_lock_exclusive((lp)->rw_lock)
291 #define rw_wunlock(lp) lck_rw_unlock_exclusive((lp)->rw_lock)
292 #define rw_destroy(lp) _audit_rw_destroy(lp)
294 #define MA_OWNED LCK_MTX_ASSERT_OWNED
295 #define RA_LOCKED LCK_RW_ASSERT_HELD
296 #define RA_RLOCKED LCK_RW_ASSERT_SHARED
297 #define RA_WLOCKED LCK_RW_ASSERT_EXCLUSIVE
298 #define SA_LOCKED LCK_RW_ASSERT_HELD
299 #define SA_XLOCKED LCK_RW_ASSERT_EXCLUSIVE
300 #define SL_OWNED LCK_MTX_ASSERT_OWNED
301 #define SL_NOTOWNED LCK_MTX_ASSERT_NOTOWNED
303 #define mtx_assert(mp, wht) lck_mtx_assert((mp)->mtx_lock, wht)
304 #define rw_assert(lp, wht) lck_rw_assert((lp)->rw_lock, wht)
305 #define sx_assert(lp, wht) lck_rw_assert((lp)->sx_lock, wht)
306 #define rlck_assert(lp, wht) _audit_rlck_assert((lp), wht)
307 #define slck_assert(lp, wht) _audit_slck_assert((lp), wht)
309 #define mtx_assert(mp, wht)
310 #define rw_assert(lp, wht)
311 #define sx_assert(lp, wht)
312 #define rlck_assert(lp, wht)
313 #define slck_assert(lp, wht)
314 #endif /* DIAGNOSTIC */
317 * BSD (IPv6) event rate limiter.
319 int _audit_ppsratecheck(struct timeval
*lasttime
, int *curpps
, int maxpps
);
320 #define ppsratecheck(tv, cr, mr) _audit_ppsratecheck(tv, cr, mr)
322 #endif /* defined(_KERNEL) || defined(KERNEL) */
323 #endif /* _SECURITY_AUDIT_AUDIT_BSD_H */