1/*
2 * Copyright (c) 2005-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
29#ifndef _DTRACE_GLUE_H
30#define _DTRACE_GLUE_H
31
32#ifdef KERNEL_BUILD
33
34#include <libkern/libkern.h>
35#include <kern/locks.h>
36#include <kern/debug.h>
37#include <kern/thread_call.h>
38#include <kern/thread.h>
39#include <machine/machine_routines.h>
40#include <sys/syslog.h>
41#include <sys/ucred.h>
42#include <stdarg.h>
43#include <mach/kmod.h>
44#include <libkern/OSAtomic.h>
45
46#if defined(__i386__) || defined(__x86_64__)
47#include <i386/mp.h>
48#endif
49
50/*
51 * cmn_err
52 */
53#define CE_NOTE 1 /* notice */
54#define CE_WARN 2 /* warning */
55
56extern void cmn_err( int, const char *, ... );
57
58/*
59 * pid/proc
60 */
61
62/* Solaris proc_t is the struct. Darwin's proc_t is a pointer to it. */
63#define proc_t struct proc /* Steer clear of the Darwin typedef for proc_t */
64#define curproc ((struct proc *)current_proc()) /* Called from probe context, must blacklist */
65
66proc_t* sprlock(pid_t pid);
67void sprunlock(proc_t *p);
68
69void dtrace_sprlock(proc_t *p);
70void dtrace_sprunlock(proc_t *p);
71
72/*
73 * uread/uwrite
74 */
75
76int uread(proc_t *p, void *buf, user_size_t len, user_addr_t a);
77int uwrite(proc_t *p, void *buf, user_size_t len, user_addr_t a);
78
79/*
80 * fuword / suword
81 */
82
83int fuword8(user_addr_t, uint8_t *);
84int fuword16(user_addr_t, uint16_t *);
85int fuword32(user_addr_t, uint32_t *);
86int fuword64(user_addr_t, uint64_t *);
87
88void fuword32_noerr(user_addr_t, uint32_t *);
89void fuword64_noerr(user_addr_t, uint64_t *);
90
91int suword64(user_addr_t, uint64_t value);
92int suword32(user_addr_t, uint32_t value);
93
94/*
95 * cpuvar
96 */
97extern lck_mtx_t cpu_lock;
98extern lck_mtx_t cyc_lock;
99extern lck_mtx_t mod_lock;
100
101/*
102 * wrap_timer_call: wrapper of timer_call for cyclic timers.
103 */
104struct wrap_timer_call;
105
106/*
107 * Per-CPU data.
108 */
109typedef struct dtrace_cpu {
110 processorid_t cpu_id; /* CPU number */
111 struct dtrace_cpu *cpu_next; /* next existing CPU */
112 lck_rw_t cpu_ft_lock; /* DTrace: fasttrap lock */
113 uintptr_t cpu_dtrace_caller; /* DTrace: caller, if any */
114 hrtime_t cpu_dtrace_chillmark; /* DTrace: chill mark time */
115 hrtime_t cpu_dtrace_chilled; /* DTrace: total chill time */
116 boolean_t cpu_dtrace_invop_underway; /* DTrace gaurds against invalid op re-entrancy */
117
118 /* Local cyclic timers on this CPU */
119 LIST_HEAD(cyc_list_head, wrap_timer_call) cpu_cyc_list;
120} dtrace_cpu_t;
121
122extern dtrace_cpu_t *cpu_list;
123
124/*
125 * The cpu_core structure consists of per-CPU state available in any context.
126 * On some architectures, this may mean that the page(s) containing the
127 * NCPU-sized array of cpu_core structures must be locked in the TLB -- it
128 * is up to the platform to assure that this is performed properly. Note that
129 * the structure is sized to avoid false sharing.
130 */
131#define CPU_CACHE_COHERENCE_SIZE 64
132
133typedef struct cpu_core {
134 uint64_t cpuc_dtrace_illval; /* DTrace illegal value */
135 lck_mtx_t cpuc_pid_lock; /* DTrace pid provider lock */
136 uint16_t cpuc_dtrace_flags; /* DTrace flags */
137 uint64_t cpuc_missing_tos; /* Addr. of top most stack frame if missing */
138 uint8_t cpuc_pad[CPU_CACHE_COHERENCE_SIZE - sizeof(uint64_t) - sizeof(lck_mtx_t) - sizeof(uint16_t) - sizeof(uint64_t) ]; /* padding */
139} cpu_core_t;
140
141extern cpu_core_t *cpu_core;
142
143extern unsigned int dtrace_max_cpus; /* max number of enabled cpus */
144#define NCPU dtrace_max_cpus
145
146extern int cpu_number(void); /* From #include <kern/cpu_number.h>. Called from probe context, must blacklist. */
147
148#define CPU (&(cpu_list[cpu_number()])) /* Pointer to current CPU */
149#define CPU_ON_INTR(cpup) ml_at_interrupt_context() /* always invoked on current cpu */
150
151/*
152 * Routines used to register interest in cpu's being added to or removed
153 * from the system.
154 */
155typedef enum {
156 CPU_INIT,
157 CPU_CONFIG,
158 CPU_UNCONFIG,
159 CPU_ON,
160 CPU_OFF,
161 CPU_CPUPART_IN,
162 CPU_CPUPART_OUT
163} cpu_setup_t;
164
165typedef int cpu_setup_func_t(cpu_setup_t, int, void *);
166
167extern void register_cpu_setup_func(cpu_setup_func_t *, void *);
168extern void unregister_cpu_setup_func(cpu_setup_func_t *, void *);
169
170/*
171 * CPU_DTRACE
172 */
173
174/*
175 * DTrace flags.
176 */
177#define CPU_DTRACE_NOFAULT 0x0001 /* Don't fault */
178#define CPU_DTRACE_DROP 0x0002 /* Drop this ECB */
179#define CPU_DTRACE_BADADDR 0x0004 /* DTrace fault: bad address */
180#define CPU_DTRACE_BADALIGN 0x0008 /* DTrace fault: bad alignment */
181#define CPU_DTRACE_DIVZERO 0x0010 /* DTrace fault: divide by zero */
182#define CPU_DTRACE_ILLOP 0x0020 /* DTrace fault: illegal operation */
183#define CPU_DTRACE_NOSCRATCH 0x0040 /* DTrace fault: out of scratch */
184#define CPU_DTRACE_KPRIV 0x0080 /* DTrace fault: bad kernel access */
185#define CPU_DTRACE_UPRIV 0x0100 /* DTrace fault: bad user access */
186#define CPU_DTRACE_TUPOFLOW 0x0200 /* DTrace fault: tuple stack overflow */
187#define CPU_DTRACE_USTACK_FP 0x0400 /* pid provider hint to ustack() */
188#define CPU_DTRACE_ENTRY 0x0800 /* pid provider hint to ustack() */
189#define CPU_DTRACE_BADSTACK 0x1000 /* DTrace fault: bad stack */
190
191#define CPU_DTRACE_FAULT (CPU_DTRACE_BADADDR | CPU_DTRACE_BADALIGN | \
192 CPU_DTRACE_DIVZERO | CPU_DTRACE_ILLOP | \
193 CPU_DTRACE_NOSCRATCH | CPU_DTRACE_KPRIV | \
194 CPU_DTRACE_UPRIV | CPU_DTRACE_TUPOFLOW | \
195 CPU_DTRACE_BADSTACK)
196#define CPU_DTRACE_ERROR (CPU_DTRACE_FAULT | CPU_DTRACE_DROP)
197
198/*
199 * Loadable Modules
200 */
201
202/* Keep the compiler happy */
203struct dtrace_module_symbols;
204
205/* Solaris' modctl structure, greatly simplified, shadowing parts of xnu kmod structure. */
206typedef struct modctl {
207 struct modctl *mod_next;
208 struct modctl *mod_stale; // stale module chain
209 uint32_t mod_id; // the kext unique identifier
210 char mod_modname[KMOD_MAX_NAME];
211 int mod_loadcnt;
212 char mod_loaded;
213 uint16_t mod_flags; // See flags below
214 int mod_nenabled; // # of enabled DTrace probes in module
215 vm_address_t mod_address; // starting address (of Mach-o header blob)
216 vm_size_t mod_size; // total size (of blob)
217 UUID mod_uuid;
218 struct dtrace_module_symbols* mod_user_symbols;
219} modctl_t;
220
221/* Definitions for mod_flags */
222#define MODCTL_IS_MACH_KERNEL 0x01 // This module represents /mach_kernel
223#define MODCTL_HAS_KERNEL_SYMBOLS 0x02 // Kernel symbols (nlist) are available
224#define MODCTL_FBT_PROBES_PROVIDED 0x04 // fbt probes have been provided
225#define MODCTL_FBT_INVALID 0x08 // Module is invalid for fbt probes
226#define MODCTL_SDT_PROBES_PROVIDED 0x10 // sdt probes have been provided
227#define MODCTL_SDT_INVALID 0x20 // Module is invalid for sdt probes
228#define MODCTL_HAS_UUID 0x40 // Module has UUID
229#define MODCTL_FBT_PRIVATE_PROBES_PROVIDED 0x80 // fbt private probes have been provided
230#define MODCTL_FBT_PROVIDE_PRIVATE_PROBES 0x100 // fbt provider must provide private probes
231#define MODCTL_FBT_PROVIDE_BLACKLISTED_PROBES 0x200 // fbt provider must provide blacklisted probes
232#define MODCTL_FBT_BLACKLISTED_PROBES_PROVIDED 0x400 // fbt blacklisted probes have been provided
233#define MODCTL_IS_STATIC_KEXT 0x800 // module is a static kext
234
235/* Simple/singular mod_flags accessors */
236#define MOD_IS_MACH_KERNEL(mod) (mod->mod_flags & MODCTL_IS_MACH_KERNEL)
237#define MOD_HAS_KERNEL_SYMBOLS(mod) (mod->mod_flags & MODCTL_HAS_KERNEL_SYMBOLS)
238#define MOD_HAS_USERSPACE_SYMBOLS(mod) (mod->mod_user_symbols) /* No point in duplicating state in the flags bits */
239#define MOD_FBT_PROBES_PROVIDED(mod) (mod->mod_flags & MODCTL_FBT_PROBES_PROVIDED)
240#define MOD_FBT_INVALID(mod) (mod->mod_flags & MODCTL_FBT_INVALID)
241#define MOD_SDT_PROBES_PROVIDED(mod) (mod->mod_flags & MODCTL_SDT_PROBES_PROVIDED)
242#define MOD_SDT_INVALID(mod) (mod->mod_flags & MODCTL_SDT_INVALID)
243#define MOD_HAS_UUID(mod) (mod->mod_flags & MODCTL_HAS_UUID)
244#define MOD_FBT_PRIVATE_PROBES_PROVIDED(mod) (mod->mod_flags & MODCTL_FBT_PRIVATE_PROBES_PROVIDED)
245#define MOD_FBT_PROVIDE_PRIVATE_PROBES(mod) (mod->mod_flags & MODCTL_FBT_PROVIDE_PRIVATE_PROBES)
246#define MOD_FBT_BLACKLISTED_PROBES_PROVIDED(mod) (mod->mod_flags & MODCTL_FBT_BLACKLISTED_PROBES_PROVIDED)
247#define MOD_FBT_PROVIDE_BLACKLISTED_PROBES(mod) (mod->mod_flags & MODCTL_FBT_PROVIDE_BLACKLISTED_PROBES)
248#define MOD_IS_STATIC_KEXT(mod) (mod->mod_flags & MODCTL_IS_STATIC_KEXT)
249
250/* Compound accessors */
251#define MOD_FBT_PRIVATE_PROBES_DONE(mod) (MOD_FBT_PRIVATE_PROBES_PROVIDED(mod) || !MOD_FBT_PROVIDE_PRIVATE_PROBES(mod))
252#define MOD_FBT_BLACKLISTED_PROBES_DONE(mod) (MOD_FBT_BLACKLISTED_PROBES_PROVIDED(mod) || !MOD_FBT_PROVIDE_BLACKLISTED_PROBES(mod))
253#define MOD_FBT_DONE(mod) ((MOD_FBT_PROBES_PROVIDED(mod) && MOD_FBT_PRIVATE_PROBES_DONE(mod) && MOD_FBT_BLACKLISTED_PROBES_DONE(mod)) || MOD_FBT_INVALID(mod))
254#define MOD_SDT_DONE(mod) (MOD_SDT_PROBES_PROVIDED(mod) || MOD_SDT_INVALID(mod))
255#define MOD_SYMBOLS_DONE(mod) (MOD_FBT_DONE(mod) && MOD_SDT_DONE(mod))
256
257extern modctl_t *dtrace_modctl_list;
258
259extern int dtrace_addr_in_module(void*, struct modctl*);
260
261/*
262 * cred_t
263 */
264/* Privileges */
265#define PRIV_DTRACE_KERNEL 3
266#define PRIV_DTRACE_PROC 4
267#define PRIV_DTRACE_USER 5
268#define PRIV_PROC_OWNER 30
269#define PRIV_PROC_ZONE 35
270#define PRIV_ALL (-1) /* All privileges required */
271
272/* Privilege sets */
273#define PRIV_EFFECTIVE 0
274
275typedef struct ucred cred_t;
276#define cr_suid cr_svuid
277#define cr_sgid cr_svgid
278
279extern cred_t *dtrace_CRED(void); /* Safe to call from probe context. */
280#define CRED() kauth_cred_get() /* Can't be called from probe context! */
281extern int PRIV_POLICY_CHOICE(void *, int, int);
282extern int PRIV_POLICY_ONLY(void *, int, int);
283extern uid_t crgetuid(const cred_t *);
284#define crgetzoneid(x) ((zoneid_t)0)
285
286/*
287 * "cyclic"
288 */
289#define CY_LOW_LEVEL 0
290#define CY_HIGH_LEVEL 2
291#define CY_LEVELS 3
292
293typedef uintptr_t cyclic_id_t;
294typedef cyclic_id_t *cyclic_id_list_t;
295typedef uint16_t cyc_level_t;
296typedef void (*cyc_func_t)(void *);
297
298#define CYCLIC_NONE ((cyclic_id_t)0)
299
300typedef struct cyc_time {
301 hrtime_t cyt_when;
302 hrtime_t cyt_interval;
303} cyc_time_t;
304
305typedef struct cyc_handler {
306 cyc_func_t cyh_func;
307 void *cyh_arg;
308 cyc_level_t cyh_level;
309} cyc_handler_t;
310
311typedef struct cyc_omni_handler {
312 void (*cyo_online)(void *, dtrace_cpu_t *, cyc_handler_t *, cyc_time_t *);
313 void (*cyo_offline)(void *, dtrace_cpu_t *, void *);
314 void *cyo_arg;
315} cyc_omni_handler_t;
316
317extern void dtrace_install_cpu_hooks(void);
318
319extern cyclic_id_t cyclic_add(cyc_handler_t *, cyc_time_t *);
320extern void cyclic_remove(cyclic_id_t);
321
322extern cyclic_id_list_t cyclic_add_omni(cyc_omni_handler_t *);
323extern void cyclic_remove_omni(cyclic_id_list_t);
324
325extern cyclic_id_t cyclic_timer_add(cyc_handler_t *, cyc_time_t *);
326extern void cyclic_timer_remove(cyclic_id_t);
327
328/*
329 * ddi
330 */
331
332#define DDI_SUCCESS 0
333#define DDI_FAILURE -1
334
335#define DDI_PSEUDO "ddi_pseudo"
336
337typedef enum {
338 DDI_DETACH = 0,
339 DDI_SUSPEND = 1,
340 DDI_PM_SUSPEND = 2,
341 DDI_HOTPLUG_DETACH = 3 /* detach, don't try to auto-unconfig */
342} ddi_detach_cmd_t;
343
344#define DDI_PROP_SUCCESS 0
345
346#define DDI_PROP_DONTPASS 1
347typedef uint_t major_t;
348typedef uint_t minor_t;
349
350typedef struct __dev_info *dev_info_t;
351
352extern int ddi_driver_major(dev_info_t *);
353
354extern int ddi_create_minor_node(dev_info_t *, const char *, int, minor_t, const char *, int);
355extern void ddi_remove_minor_node(dev_info_t *, char *);
356
357extern major_t getemajor(dev_t);
358extern minor_t getminor(dev_t);
359
360/*
361 * Kernel Debug Interface
362 */
363extern void debug_enter(char *);
364
365/*
366 * DTrace specific zone allocation
367 */
368
369/*
370 * kmem
371 */
372
373#define KM_SLEEP 0x00000000
374#define KM_NOSLEEP 0x00000001
375
376typedef struct vmem vmem_t;
377typedef struct kmem_cache kmem_cache_t;
378
379#define kmem_free dt_kmem_free /* Avoid clash with Darwin's kmem_free */
380#define kmem_free_aligned dt_kmem_free_aligned
381
382#define kmem_alloc(size, kmflag) \
383 ({ VM_ALLOC_SITE_STATIC(0, 0); \
384 dt_kmem_alloc_site(size, kmflag, &site); })
385
386extern void *dt_kmem_alloc_site(size_t, int, vm_allocation_site_t*);
387extern void dt_kmem_free(void *, size_t);
388
389#define kmem_zalloc(size, kmflag) \
390 ({ VM_ALLOC_SITE_STATIC(0, 0); \
391 dt_kmem_zalloc_site(size, kmflag, &site); })
392
393extern void *dt_kmem_zalloc_site(size_t, int, vm_allocation_site_t*);
394
395#define kmem_alloc_aligned(size, align, kmflag) \
396 ({ VM_ALLOC_SITE_STATIC(0, 0); \
397 dt_kmem_alloc_aligned_site(size, align, kmflag, &site); })
398extern void *dt_kmem_alloc_aligned_site(size_t, size_t, int, vm_allocation_site_t*);
399
400#define kmem_zalloc_aligned(size, align, kmflag) \
401 ({ VM_ALLOC_SITE_STATIC(0, 0); \
402 dt_kmem_zalloc_aligned_site(size, align, kmflag, &site); })
403extern void *dt_kmem_zalloc_aligned_site(size_t, size_t, int, vm_allocation_site_t*);
404
405extern void dt_kmem_free_aligned(void*, size_t);
406
407extern kmem_cache_t *
408kmem_cache_create(const char *, size_t, size_t, int (*)(void *, void *, int),
409 void (*)(void *, void *), void (*)(void *), void *, vmem_t *, int);
410extern void *kmem_cache_alloc(kmem_cache_t *, int);
411extern void kmem_cache_free(kmem_cache_t *, void *);
412extern void kmem_cache_destroy(kmem_cache_t *);
413
414/*
415 * kthread
416 */
417
418typedef struct _kthread kthread_t; /* For dtrace_vtime_switch(), dtrace_panicked and dtrace_errthread */
419
420/*
421 * proc
422 */
423
424
425#define DATAMODEL_ILP32 0x00100000
426#define DATAMODEL_LP64 0x00200000
427
428#define DATAMODEL_NONE 0
429
430#if defined(__LP64__)
431#define DATAMODEL_NATIVE DATAMODEL_LP64
432#else
433#define DATAMODEL_NATIVE DATAMODEL_ILP32
434#endif /* __LP64__ */
435
436typedef unsigned int model_t; /* For dtrace_instr_size_isa() prototype in <sys/dtrace.h> */
437
438/*
439 * vmem
440 */
441
442#define VMC_IDENTIFIER 0x00040000 /* not backed by memory */
443#define VM_SLEEP 0x00000000 /* same as KM_SLEEP */
444#define VM_BESTFIT 0x00000100
445
446extern void *vmem_alloc(vmem_t *, size_t, int);
447extern vmem_t *vmem_create(const char *, void *, size_t, size_t, void *,
448 void *, vmem_t *, size_t, int);
449extern void vmem_destroy(vmem_t *);
450extern void vmem_free(vmem_t *vmp, void *vaddr, size_t size);
451
452/*
453 * Atomic
454 */
455
456static inline uint8_t atomic_or_8(uint8_t *addr, uint8_t mask)
457{
458 return OSBitOrAtomic8(mask, addr);
459}
460
461static inline uint32_t atomic_and_32( uint32_t *addr, int32_t mask)
462{
463 return OSBitAndAtomic(mask, addr);
464}
465
466static inline uint32_t atomic_add_32( uint32_t *theAddress, int32_t theAmount )
467{
468 return OSAddAtomic( theAmount, theAddress );
469}
470
471#if defined(__i386__) || defined(__x86_64__)
472static inline void atomic_add_64( uint64_t *theAddress, int64_t theAmount )
473{
474 (void)OSAddAtomic64( theAmount, (SInt64 *)theAddress );
475}
476#elif defined(__arm__)
477static inline void atomic_add_64( uint64_t *theAddress, int64_t theAmount )
478{
479 // FIXME
480 // atomic_add_64() is at present only called from fasttrap.c to increment
481 // or decrement a 64bit counter. Narrow to 32bits since arm has
482 // no convenient 64bit atomic op.
483
484 (void)OSAddAtomic( (int32_t)theAmount, &(((SInt32 *)theAddress)[0]));
485}
486#elif defined (__arm64__)
487static inline void atomic_add_64( uint64_t *theAddress, int64_t theAmount )
488{
489 (void)OSAddAtomic64( theAmount, (SInt64 *)theAddress );
490}
491#endif
492
493static inline uint32_t atomic_or_32(uint32_t *addr, uint32_t mask)
494{
495 return OSBitOrAtomic(mask, addr);
496}
497
498
499/*
500 * Miscellaneous
501 */
502
503typedef uintptr_t pc_t;
504typedef uintptr_t greg_t; /* For dtrace_impl.h prototype of dtrace_getfp() */
505#if defined(__arm__) || defined(__arm64__)
506#define regs arm_saved_state
507#endif
508extern struct regs *find_user_regs( thread_t thread);
509extern vm_offset_t dtrace_get_cpu_int_stack_top(void);
510extern vm_offset_t max_valid_stack_address(void); /* kern/thread.h */
511
512#define panic_quiesce (panic_active())
513
514#define IS_P2ALIGNED(v, a) ((((uintptr_t)(v)) & ((uintptr_t)(a) - 1)) == 0)
515
516extern int vuprintf(const char *, va_list);
517
518extern hrtime_t dtrace_abs_to_nano(uint64_t);
519
520__private_extern__ const char * strstr(const char *, const char *);
521const void* bsearch(const void*, const void*, size_t, size_t, int (*compar)(const void *, const void *));
522
523int dtrace_copy_maxsize(void);
524int dtrace_buffer_copyout(const void*, user_addr_t, vm_size_t);
525
526
527#define DTRACE_NCLIENTS 32
528
529#undef proc_t
530
531/*
532 * Safe counted string compare against a literal string. The sizeof() intentionally
533 * counts the trailing NUL, and so ensures that all the characters in the literal
534 * can participate in the comparison.
535 */
536#define LIT_STRNEQL(s1, lit_s2) (0 == strncmp( (s1), (lit_s2), sizeof((lit_s2)) ))
537
538/*
539 * Safe counted string compare of a literal against the beginning of a string. Here
540 * the sizeof() is reduced by 1 so that the trailing null of the literal does not
541 * participate in the comparison.
542 */
543#define LIT_STRNSTART(s1, lit_s2) (0 == strncmp( (s1), (lit_s2), sizeof((lit_s2)) - 1 ))
544
545#define KERNELBASE VM_MIN_KERNEL_ADDRESS
546#endif /* KERNEL_BUILD */
547#endif /* _DTRACE_GLUE_H */
548
549