xref: /onnv-gate/usr/src/uts/common/dtrace/fasttrap.c (revision 665:7e271f1ed7fa)
10Sstevel@tonic-gate /*
20Sstevel@tonic-gate  * CDDL HEADER START
30Sstevel@tonic-gate  *
40Sstevel@tonic-gate  * The contents of this file are subject to the terms of the
50Sstevel@tonic-gate  * Common Development and Distribution License, Version 1.0 only
60Sstevel@tonic-gate  * (the "License").  You may not use this file except in compliance
70Sstevel@tonic-gate  * with the License.
80Sstevel@tonic-gate  *
90Sstevel@tonic-gate  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
100Sstevel@tonic-gate  * or http://www.opensolaris.org/os/licensing.
110Sstevel@tonic-gate  * See the License for the specific language governing permissions
120Sstevel@tonic-gate  * and limitations under the License.
130Sstevel@tonic-gate  *
140Sstevel@tonic-gate  * When distributing Covered Code, include this CDDL HEADER in each
150Sstevel@tonic-gate  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
160Sstevel@tonic-gate  * If applicable, add the following below this CDDL HEADER, with the
170Sstevel@tonic-gate  * fields enclosed by brackets "[]" replaced with your own identifying
180Sstevel@tonic-gate  * information: Portions Copyright [yyyy] [name of copyright owner]
190Sstevel@tonic-gate  *
200Sstevel@tonic-gate  * CDDL HEADER END
210Sstevel@tonic-gate  */
22390Sraf 
230Sstevel@tonic-gate /*
240Sstevel@tonic-gate  * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
250Sstevel@tonic-gate  * Use is subject to license terms.
260Sstevel@tonic-gate  */
270Sstevel@tonic-gate 
280Sstevel@tonic-gate #pragma ident	"%Z%%M%	%I%	%E% SMI"
290Sstevel@tonic-gate 
300Sstevel@tonic-gate #include <sys/atomic.h>
310Sstevel@tonic-gate #include <sys/errno.h>
320Sstevel@tonic-gate #include <sys/stat.h>
330Sstevel@tonic-gate #include <sys/modctl.h>
340Sstevel@tonic-gate #include <sys/conf.h>
350Sstevel@tonic-gate #include <sys/systm.h>
360Sstevel@tonic-gate #include <sys/ddi.h>
370Sstevel@tonic-gate #include <sys/sunddi.h>
380Sstevel@tonic-gate #include <sys/cpuvar.h>
390Sstevel@tonic-gate #include <sys/kmem.h>
400Sstevel@tonic-gate #include <sys/strsubr.h>
410Sstevel@tonic-gate #include <sys/fasttrap.h>
420Sstevel@tonic-gate #include <sys/fasttrap_impl.h>
430Sstevel@tonic-gate #include <sys/fasttrap_isa.h>
440Sstevel@tonic-gate #include <sys/dtrace.h>
450Sstevel@tonic-gate #include <sys/dtrace_impl.h>
460Sstevel@tonic-gate #include <sys/sysmacros.h>
470Sstevel@tonic-gate #include <sys/frame.h>
480Sstevel@tonic-gate #include <sys/stack.h>
490Sstevel@tonic-gate #include <sys/proc.h>
500Sstevel@tonic-gate #include <sys/priv.h>
510Sstevel@tonic-gate #include <sys/policy.h>
520Sstevel@tonic-gate #include <sys/ontrap.h>
530Sstevel@tonic-gate #include <sys/vmsystm.h>
540Sstevel@tonic-gate #include <sys/prsystm.h>
550Sstevel@tonic-gate 
560Sstevel@tonic-gate 
570Sstevel@tonic-gate #include <vm/as.h>
580Sstevel@tonic-gate #include <vm/seg.h>
590Sstevel@tonic-gate #include <vm/seg_dev.h>
600Sstevel@tonic-gate #include <vm/seg_vn.h>
610Sstevel@tonic-gate #include <vm/seg_spt.h>
620Sstevel@tonic-gate #include <vm/seg_kmem.h>
630Sstevel@tonic-gate 
640Sstevel@tonic-gate /*
650Sstevel@tonic-gate  * User-Land Trap-Based Tracing
660Sstevel@tonic-gate  * ----------------------------
670Sstevel@tonic-gate  *
680Sstevel@tonic-gate  * The fasttrap provider allows DTrace consumers to instrument any user-level
690Sstevel@tonic-gate  * instruction to gather data; this includes probes with semantic
700Sstevel@tonic-gate  * signifigance like entry and return as well as simple offsets into the
710Sstevel@tonic-gate  * function. While the specific techniques used are very ISA specific, the
720Sstevel@tonic-gate  * methodology is generalizable to any architecture.
730Sstevel@tonic-gate  *
740Sstevel@tonic-gate  *
750Sstevel@tonic-gate  * The General Methodology
760Sstevel@tonic-gate  * -----------------------
770Sstevel@tonic-gate  *
780Sstevel@tonic-gate  * With the primary goal of tracing every user-land instruction and the
790Sstevel@tonic-gate  * limitation that we can't trust user space so don't want to rely on much
800Sstevel@tonic-gate  * information there, we begin by replacing the instructions we want to trace
810Sstevel@tonic-gate  * with trap instructions. Each instruction we overwrite is saved into a hash
820Sstevel@tonic-gate  * table keyed by process ID and pc address. When we enter the kernel due to
830Sstevel@tonic-gate  * this trap instruction, we need the effects of the replaced instruction to
840Sstevel@tonic-gate  * appear to have occurred before we proceed with the user thread's
850Sstevel@tonic-gate  * execution.
860Sstevel@tonic-gate  *
870Sstevel@tonic-gate  * Each user level thread is represented by a ulwp_t structure which is
880Sstevel@tonic-gate  * always easily accessible through a register. The most basic way to produce
890Sstevel@tonic-gate  * the effects of the instruction we replaced is to copy that instruction out
900Sstevel@tonic-gate  * to a bit of scratch space reserved in the user thread's ulwp_t structure
910Sstevel@tonic-gate  * (a sort of kernel-private thread local storage), set the PC to that
920Sstevel@tonic-gate  * scratch space and single step. When we reenter the kernel after single
930Sstevel@tonic-gate  * stepping the instruction we must then adjust the PC to point to what would
940Sstevel@tonic-gate  * normally be the next instruction. Of course, special care must be taken
950Sstevel@tonic-gate  * for branches and jumps, but these represent such a small fraction of any
960Sstevel@tonic-gate  * instruction set that writing the code to emulate these in the kernel is
970Sstevel@tonic-gate  * not too difficult.
980Sstevel@tonic-gate  *
990Sstevel@tonic-gate  * Return probes may require several tracepoints to trace every return site,
1000Sstevel@tonic-gate  * and, conversely, each tracepoint may activate several probes (the entry
1010Sstevel@tonic-gate  * and offset 0 probes, for example). To solve this muliplexing problem,
1020Sstevel@tonic-gate  * tracepoints contain lists of probes to activate and probes contain lists
1030Sstevel@tonic-gate  * of tracepoints to enable. If a probe is activated, it adds its ID to
1040Sstevel@tonic-gate  * existing tracepoints or creates new ones as necessary.
1050Sstevel@tonic-gate  *
1060Sstevel@tonic-gate  * Most probes are activated _before_ the instruction is executed, but return
1070Sstevel@tonic-gate  * probes are activated _after_ the effects of the last instruction of the
1080Sstevel@tonic-gate  * function are visible. Return probes must be fired _after_ we have
1090Sstevel@tonic-gate  * single-stepped the instruction whereas all other probes are fired
1100Sstevel@tonic-gate  * beforehand.
1110Sstevel@tonic-gate  */
1120Sstevel@tonic-gate 
1130Sstevel@tonic-gate static dev_info_t *fasttrap_devi;
1140Sstevel@tonic-gate static dtrace_provider_id_t fasttrap_id;
1150Sstevel@tonic-gate static dtrace_meta_provider_id_t fasttrap_meta_id;
1160Sstevel@tonic-gate 
1170Sstevel@tonic-gate static timeout_id_t fasttrap_timeout;
1180Sstevel@tonic-gate static kmutex_t fasttrap_cleanup_mtx;
1190Sstevel@tonic-gate static uint_t fasttrap_cleanup_work;
1200Sstevel@tonic-gate 
1210Sstevel@tonic-gate /*
1220Sstevel@tonic-gate  * Generation count on modifications to the global tracepoint lookup table.
1230Sstevel@tonic-gate  */
1240Sstevel@tonic-gate static volatile uint64_t fasttrap_mod_gen;
1250Sstevel@tonic-gate 
1260Sstevel@tonic-gate /*
1270Sstevel@tonic-gate  * When the fasttrap provider is loaded, fasttrap_max is set to either
1280Sstevel@tonic-gate  * FASTTRAP_MAX_DEFAULT or the value for fasttrap-max-probes in the
1290Sstevel@tonic-gate  * fasttrap.conf file. Each time a probe is created, fasttrap_total is
1300Sstevel@tonic-gate  * incremented by the number of tracepoints that may be associated with that
1310Sstevel@tonic-gate  * probe; fasttrap_total is capped at fasttrap_max.
1320Sstevel@tonic-gate  */
1330Sstevel@tonic-gate #define	FASTTRAP_MAX_DEFAULT		250000
1340Sstevel@tonic-gate static uint32_t fasttrap_max;
1350Sstevel@tonic-gate static uint32_t fasttrap_total;
1360Sstevel@tonic-gate 
1370Sstevel@tonic-gate 
1380Sstevel@tonic-gate #define	FASTTRAP_TPOINTS_DEFAULT_SIZE	0x4000
1390Sstevel@tonic-gate #define	FASTTRAP_PROVIDERS_DEFAULT_SIZE	0x100
140532Sahl #define	FASTTRAP_PROCS_DEFAULT_SIZE	0x100
1410Sstevel@tonic-gate 
1420Sstevel@tonic-gate #define	FASTTRAP_PID_NAME		"pid"
1430Sstevel@tonic-gate 
1440Sstevel@tonic-gate fasttrap_hash_t			fasttrap_tpoints;
1450Sstevel@tonic-gate static fasttrap_hash_t		fasttrap_provs;
146532Sahl static fasttrap_hash_t		fasttrap_procs;
1470Sstevel@tonic-gate 
1480Sstevel@tonic-gate dtrace_id_t			fasttrap_probe_id;
1490Sstevel@tonic-gate static int			fasttrap_count;		/* ref count */
1500Sstevel@tonic-gate static int			fasttrap_pid_count;	/* pid ref count */
1510Sstevel@tonic-gate static kmutex_t			fasttrap_count_mtx;	/* lock on ref count */
1520Sstevel@tonic-gate 
153315Sahl #define	FASTTRAP_ENABLE_FAIL	1
154315Sahl #define	FASTTRAP_ENABLE_PARTIAL	2
155315Sahl 
1560Sstevel@tonic-gate static int fasttrap_tracepoint_enable(proc_t *, fasttrap_probe_t *, uint_t);
1570Sstevel@tonic-gate static void fasttrap_tracepoint_disable(proc_t *, fasttrap_probe_t *, uint_t);
1580Sstevel@tonic-gate 
1590Sstevel@tonic-gate static fasttrap_provider_t *fasttrap_provider_lookup(pid_t, const char *,
1600Sstevel@tonic-gate     const dtrace_pattr_t *);
1610Sstevel@tonic-gate static void fasttrap_provider_free(fasttrap_provider_t *);
1620Sstevel@tonic-gate static void fasttrap_provider_retire(fasttrap_provider_t *);
1630Sstevel@tonic-gate 
164532Sahl static fasttrap_proc_t *fasttrap_proc_lookup(pid_t);
165532Sahl static void fasttrap_proc_release(fasttrap_proc_t *);
166532Sahl 
1670Sstevel@tonic-gate #define	FASTTRAP_PROVS_INDEX(pid, name) \
1680Sstevel@tonic-gate 	((fasttrap_hash_str(name) + (pid)) & fasttrap_provs.fth_mask)
1690Sstevel@tonic-gate 
170532Sahl #define	FASTTRAP_PROCS_INDEX(pid) ((pid) & fasttrap_procs.fth_mask)
171532Sahl 
1720Sstevel@tonic-gate static int
1730Sstevel@tonic-gate fasttrap_highbit(ulong_t i)
1740Sstevel@tonic-gate {
1750Sstevel@tonic-gate 	int h = 1;
1760Sstevel@tonic-gate 
1770Sstevel@tonic-gate 	if (i == 0)
1780Sstevel@tonic-gate 		return (0);
1790Sstevel@tonic-gate #ifdef _LP64
1800Sstevel@tonic-gate 	if (i & 0xffffffff00000000ul) {
1810Sstevel@tonic-gate 		h += 32; i >>= 32;
1820Sstevel@tonic-gate 	}
1830Sstevel@tonic-gate #endif
1840Sstevel@tonic-gate 	if (i & 0xffff0000) {
1850Sstevel@tonic-gate 		h += 16; i >>= 16;
1860Sstevel@tonic-gate 	}
1870Sstevel@tonic-gate 	if (i & 0xff00) {
1880Sstevel@tonic-gate 		h += 8; i >>= 8;
1890Sstevel@tonic-gate 	}
1900Sstevel@tonic-gate 	if (i & 0xf0) {
1910Sstevel@tonic-gate 		h += 4; i >>= 4;
1920Sstevel@tonic-gate 	}
1930Sstevel@tonic-gate 	if (i & 0xc) {
1940Sstevel@tonic-gate 		h += 2; i >>= 2;
1950Sstevel@tonic-gate 	}
1960Sstevel@tonic-gate 	if (i & 0x2) {
1970Sstevel@tonic-gate 		h += 1;
1980Sstevel@tonic-gate 	}
1990Sstevel@tonic-gate 	return (h);
2000Sstevel@tonic-gate }
2010Sstevel@tonic-gate 
2020Sstevel@tonic-gate static uint_t
2030Sstevel@tonic-gate fasttrap_hash_str(const char *p)
2040Sstevel@tonic-gate {
2050Sstevel@tonic-gate 	unsigned int g;
2060Sstevel@tonic-gate 	uint_t hval = 0;
2070Sstevel@tonic-gate 
2080Sstevel@tonic-gate 	while (*p) {
2090Sstevel@tonic-gate 		hval = (hval << 4) + *p++;
2100Sstevel@tonic-gate 		if ((g = (hval & 0xf0000000)) != 0)
2110Sstevel@tonic-gate 			hval ^= g >> 24;
2120Sstevel@tonic-gate 		hval &= ~g;
2130Sstevel@tonic-gate 	}
2140Sstevel@tonic-gate 	return (hval);
2150Sstevel@tonic-gate }
2160Sstevel@tonic-gate 
2170Sstevel@tonic-gate void
2180Sstevel@tonic-gate fasttrap_sigtrap(proc_t *p, kthread_t *t, uintptr_t pc)
2190Sstevel@tonic-gate {
2200Sstevel@tonic-gate 	sigqueue_t *sqp = kmem_zalloc(sizeof (sigqueue_t), KM_SLEEP);
2210Sstevel@tonic-gate 
2220Sstevel@tonic-gate 	sqp->sq_info.si_signo = SIGTRAP;
2230Sstevel@tonic-gate 	sqp->sq_info.si_code = TRAP_DTRACE;
2240Sstevel@tonic-gate 	sqp->sq_info.si_addr = (caddr_t)pc;
2250Sstevel@tonic-gate 
2260Sstevel@tonic-gate 	mutex_enter(&p->p_lock);
2270Sstevel@tonic-gate 	sigaddqa(p, t, sqp);
2280Sstevel@tonic-gate 	mutex_exit(&p->p_lock);
2290Sstevel@tonic-gate 
2300Sstevel@tonic-gate 	if (t != NULL)
2310Sstevel@tonic-gate 		aston(t);
2320Sstevel@tonic-gate }
2330Sstevel@tonic-gate 
2340Sstevel@tonic-gate /*
2350Sstevel@tonic-gate  * This function ensures that no threads are actively using the memory
2360Sstevel@tonic-gate  * associated with probes that were formerly live.
2370Sstevel@tonic-gate  */
2380Sstevel@tonic-gate static void
2390Sstevel@tonic-gate fasttrap_mod_barrier(uint64_t gen)
2400Sstevel@tonic-gate {
2410Sstevel@tonic-gate 	int i;
2420Sstevel@tonic-gate 
2430Sstevel@tonic-gate 	if (gen < fasttrap_mod_gen)
2440Sstevel@tonic-gate 		return;
2450Sstevel@tonic-gate 
2460Sstevel@tonic-gate 	fasttrap_mod_gen++;
2470Sstevel@tonic-gate 
2480Sstevel@tonic-gate 	for (i = 0; i < NCPU; i++) {
2490Sstevel@tonic-gate 		mutex_enter(&cpu_core[i].cpuc_pid_lock);
2500Sstevel@tonic-gate 		mutex_exit(&cpu_core[i].cpuc_pid_lock);
2510Sstevel@tonic-gate 	}
2520Sstevel@tonic-gate }
2530Sstevel@tonic-gate 
2540Sstevel@tonic-gate /*
2550Sstevel@tonic-gate  * This is the timeout's callback for cleaning up the providers and their
2560Sstevel@tonic-gate  * probes.
2570Sstevel@tonic-gate  */
2580Sstevel@tonic-gate /*ARGSUSED*/
2590Sstevel@tonic-gate static void
2600Sstevel@tonic-gate fasttrap_pid_cleanup_cb(void *data)
2610Sstevel@tonic-gate {
2620Sstevel@tonic-gate 	fasttrap_provider_t **fpp, *fp;
2630Sstevel@tonic-gate 	fasttrap_bucket_t *bucket;
2640Sstevel@tonic-gate 	dtrace_provider_id_t provid;
2650Sstevel@tonic-gate 	int i, later;
2660Sstevel@tonic-gate 
2670Sstevel@tonic-gate 	static volatile int in = 0;
2680Sstevel@tonic-gate 	ASSERT(in == 0);
2690Sstevel@tonic-gate 	in = 1;
2700Sstevel@tonic-gate 
2710Sstevel@tonic-gate 	mutex_enter(&fasttrap_cleanup_mtx);
2720Sstevel@tonic-gate 	while (fasttrap_cleanup_work) {
2730Sstevel@tonic-gate 		fasttrap_cleanup_work = 0;
2740Sstevel@tonic-gate 		mutex_exit(&fasttrap_cleanup_mtx);
2750Sstevel@tonic-gate 
2760Sstevel@tonic-gate 		later = 0;
2770Sstevel@tonic-gate 
2780Sstevel@tonic-gate 		/*
2790Sstevel@tonic-gate 		 * Iterate over all the providers trying to remove the marked
280532Sahl 		 * ones. If a provider is marked but not retired, we just
2810Sstevel@tonic-gate 		 * have to take a crack at removing it -- it's no big deal if
2820Sstevel@tonic-gate 		 * we can't.
2830Sstevel@tonic-gate 		 */
2840Sstevel@tonic-gate 		for (i = 0; i < fasttrap_provs.fth_nent; i++) {
2850Sstevel@tonic-gate 			bucket = &fasttrap_provs.fth_table[i];
2860Sstevel@tonic-gate 			mutex_enter(&bucket->ftb_mtx);
2870Sstevel@tonic-gate 			fpp = (fasttrap_provider_t **)&bucket->ftb_data;
2880Sstevel@tonic-gate 
2890Sstevel@tonic-gate 			while ((fp = *fpp) != NULL) {
2900Sstevel@tonic-gate 				if (!fp->ftp_marked) {
2910Sstevel@tonic-gate 					fpp = &fp->ftp_next;
2920Sstevel@tonic-gate 					continue;
2930Sstevel@tonic-gate 				}
2940Sstevel@tonic-gate 
2950Sstevel@tonic-gate 				mutex_enter(&fp->ftp_mtx);
2960Sstevel@tonic-gate 
2970Sstevel@tonic-gate 				/*
2980Sstevel@tonic-gate 				 * If this provider is referenced either
2990Sstevel@tonic-gate 				 * because it is a USDT provider or is being
3000Sstevel@tonic-gate 				 * modified, we can't unregister or even
3010Sstevel@tonic-gate 				 * condense.
3020Sstevel@tonic-gate 				 */
3030Sstevel@tonic-gate 				if (fp->ftp_ccount != 0) {
3040Sstevel@tonic-gate 					mutex_exit(&fp->ftp_mtx);
3050Sstevel@tonic-gate 					fp->ftp_marked = 0;
3060Sstevel@tonic-gate 					continue;
3070Sstevel@tonic-gate 				}
3080Sstevel@tonic-gate 
309532Sahl 				if (!fp->ftp_retired || fp->ftp_rcount != 0)
3100Sstevel@tonic-gate 					fp->ftp_marked = 0;
3110Sstevel@tonic-gate 
3120Sstevel@tonic-gate 				mutex_exit(&fp->ftp_mtx);
3130Sstevel@tonic-gate 
3140Sstevel@tonic-gate 				/*
3150Sstevel@tonic-gate 				 * If we successfully unregister this
3160Sstevel@tonic-gate 				 * provider we can remove it from the hash
3170Sstevel@tonic-gate 				 * chain and free the memory. If our attempt
318532Sahl 				 * to unregister fails and this is a retired
3190Sstevel@tonic-gate 				 * provider, increment our flag to try again
3200Sstevel@tonic-gate 				 * pretty soon. If we've consumed more than
3210Sstevel@tonic-gate 				 * half of our total permitted number of
3220Sstevel@tonic-gate 				 * probes call dtrace_condense() to try to
3230Sstevel@tonic-gate 				 * clean out the unenabled probes.
3240Sstevel@tonic-gate 				 */
3250Sstevel@tonic-gate 				provid = fp->ftp_provid;
3260Sstevel@tonic-gate 				if (dtrace_unregister(provid) != 0) {
3270Sstevel@tonic-gate 					if (fasttrap_total > fasttrap_max / 2)
3280Sstevel@tonic-gate 						(void) dtrace_condense(provid);
3290Sstevel@tonic-gate 					later += fp->ftp_marked;
3300Sstevel@tonic-gate 					fpp = &fp->ftp_next;
3310Sstevel@tonic-gate 				} else {
3320Sstevel@tonic-gate 					*fpp = fp->ftp_next;
3330Sstevel@tonic-gate 					fasttrap_provider_free(fp);
3340Sstevel@tonic-gate 				}
3350Sstevel@tonic-gate 			}
3360Sstevel@tonic-gate 			mutex_exit(&bucket->ftb_mtx);
3370Sstevel@tonic-gate 		}
3380Sstevel@tonic-gate 
3390Sstevel@tonic-gate 		mutex_enter(&fasttrap_cleanup_mtx);
3400Sstevel@tonic-gate 	}
3410Sstevel@tonic-gate 
3420Sstevel@tonic-gate 	ASSERT(fasttrap_timeout != 0);
3430Sstevel@tonic-gate 
3440Sstevel@tonic-gate 	/*
345532Sahl 	 * If we were unable to remove a retired provider, try again after
3460Sstevel@tonic-gate 	 * a second. This situation can occur in certain circumstances where
3470Sstevel@tonic-gate 	 * providers cannot be unregistered even though they have no probes
3480Sstevel@tonic-gate 	 * enabled because of an execution of dtrace -l or something similar.
3490Sstevel@tonic-gate 	 * If the timeout has been disabled (set to 1 because we're trying
3500Sstevel@tonic-gate 	 * to detach), we set fasttrap_cleanup_work to ensure that we'll
3510Sstevel@tonic-gate 	 * get a chance to do that work if and when the timeout is reenabled
3520Sstevel@tonic-gate 	 * (if detach fails).
3530Sstevel@tonic-gate 	 */
3540Sstevel@tonic-gate 	if (later > 0 && fasttrap_timeout != (timeout_id_t)1)
3550Sstevel@tonic-gate 		fasttrap_timeout = timeout(&fasttrap_pid_cleanup_cb, NULL, hz);
3560Sstevel@tonic-gate 	else if (later > 0)
3570Sstevel@tonic-gate 		fasttrap_cleanup_work = 1;
3580Sstevel@tonic-gate 	else
3590Sstevel@tonic-gate 		fasttrap_timeout = 0;
3600Sstevel@tonic-gate 
3610Sstevel@tonic-gate 	mutex_exit(&fasttrap_cleanup_mtx);
3620Sstevel@tonic-gate 	in = 0;
3630Sstevel@tonic-gate }
3640Sstevel@tonic-gate 
3650Sstevel@tonic-gate /*
3660Sstevel@tonic-gate  * Activates the asynchronous cleanup mechanism.
3670Sstevel@tonic-gate  */
3680Sstevel@tonic-gate static void
3690Sstevel@tonic-gate fasttrap_pid_cleanup(void)
3700Sstevel@tonic-gate {
3710Sstevel@tonic-gate 	mutex_enter(&fasttrap_cleanup_mtx);
3720Sstevel@tonic-gate 	fasttrap_cleanup_work = 1;
3730Sstevel@tonic-gate 	if (fasttrap_timeout == 0)
3740Sstevel@tonic-gate 		fasttrap_timeout = timeout(&fasttrap_pid_cleanup_cb, NULL, 1);
3750Sstevel@tonic-gate 	mutex_exit(&fasttrap_cleanup_mtx);
3760Sstevel@tonic-gate }
3770Sstevel@tonic-gate 
3780Sstevel@tonic-gate /*
3790Sstevel@tonic-gate  * This is called from cfork() via dtrace_fasttrap_fork(). The child
3800Sstevel@tonic-gate  * process's address space is a (roughly) a copy of the parent process's so
3810Sstevel@tonic-gate  * we have to remove all the instrumentation we had previously enabled in the
3820Sstevel@tonic-gate  * parent.
3830Sstevel@tonic-gate  */
3840Sstevel@tonic-gate static void
3850Sstevel@tonic-gate fasttrap_fork(proc_t *p, proc_t *cp)
3860Sstevel@tonic-gate {
3870Sstevel@tonic-gate 	pid_t ppid = p->p_pid;
3880Sstevel@tonic-gate 	int i;
3890Sstevel@tonic-gate 
3900Sstevel@tonic-gate 	ASSERT(curproc == p);
3910Sstevel@tonic-gate 	ASSERT(p->p_proc_flag & P_PR_LOCK);
3920Sstevel@tonic-gate 	ASSERT(p->p_dtrace_count > 0);
3930Sstevel@tonic-gate 	ASSERT(cp->p_dtrace_count == 0);
3940Sstevel@tonic-gate 
3950Sstevel@tonic-gate 	/*
3960Sstevel@tonic-gate 	 * This would be simpler and faster if we maintained per-process
3970Sstevel@tonic-gate 	 * hash tables of enabled tracepoints. It could, however, potentially
3980Sstevel@tonic-gate 	 * slow down execution of a tracepoint since we'd need to go
3990Sstevel@tonic-gate 	 * through two levels of indirection. In the future, we should
4000Sstevel@tonic-gate 	 * consider either maintaining per-process ancillary lists of
4010Sstevel@tonic-gate 	 * enabled tracepoints or hanging a pointer to a per-process hash
4020Sstevel@tonic-gate 	 * table of enabled tracepoints off the proc structure.
4030Sstevel@tonic-gate 	 */
4040Sstevel@tonic-gate 
4050Sstevel@tonic-gate 	/*
4060Sstevel@tonic-gate 	 * We don't have to worry about the child process disappearing
4070Sstevel@tonic-gate 	 * because we're in fork().
4080Sstevel@tonic-gate 	 */
4090Sstevel@tonic-gate 	mutex_enter(&cp->p_lock);
4100Sstevel@tonic-gate 	sprlock_proc(cp);
4110Sstevel@tonic-gate 	mutex_exit(&cp->p_lock);
4120Sstevel@tonic-gate 
4130Sstevel@tonic-gate 	/*
4140Sstevel@tonic-gate 	 * Iterate over every tracepoint looking for ones that belong to the
4150Sstevel@tonic-gate 	 * parent process, and remove each from the child process.
4160Sstevel@tonic-gate 	 */
4170Sstevel@tonic-gate 	for (i = 0; i < fasttrap_tpoints.fth_nent; i++) {
4180Sstevel@tonic-gate 		fasttrap_tracepoint_t *tp;
4190Sstevel@tonic-gate 		fasttrap_bucket_t *bucket = &fasttrap_tpoints.fth_table[i];
4200Sstevel@tonic-gate 
4210Sstevel@tonic-gate 		mutex_enter(&bucket->ftb_mtx);
4220Sstevel@tonic-gate 		for (tp = bucket->ftb_data; tp != NULL; tp = tp->ftt_next) {
423532Sahl 			if (tp->ftt_pid == ppid &&
424532Sahl 			    !tp->ftt_proc->ftpc_defunct) {
4250Sstevel@tonic-gate 				int ret = fasttrap_tracepoint_remove(cp, tp);
4260Sstevel@tonic-gate 				ASSERT(ret == 0);
4270Sstevel@tonic-gate 			}
4280Sstevel@tonic-gate 		}
4290Sstevel@tonic-gate 		mutex_exit(&bucket->ftb_mtx);
4300Sstevel@tonic-gate 	}
4310Sstevel@tonic-gate 
4320Sstevel@tonic-gate 	mutex_enter(&cp->p_lock);
4330Sstevel@tonic-gate 	sprunlock(cp);
4340Sstevel@tonic-gate }
4350Sstevel@tonic-gate 
4360Sstevel@tonic-gate /*
4370Sstevel@tonic-gate  * This is called from proc_exit() or from exec_common() if p_dtrace_probes
4380Sstevel@tonic-gate  * is set on the proc structure to indicate that there is a pid provider
4390Sstevel@tonic-gate  * associated with this process.
4400Sstevel@tonic-gate  */
4410Sstevel@tonic-gate static void
4420Sstevel@tonic-gate fasttrap_exec_exit(proc_t *p)
4430Sstevel@tonic-gate {
4440Sstevel@tonic-gate 	fasttrap_provider_t *provider;
4450Sstevel@tonic-gate 
4460Sstevel@tonic-gate 	ASSERT(p == curproc);
4470Sstevel@tonic-gate 	ASSERT(MUTEX_HELD(&p->p_lock));
4480Sstevel@tonic-gate 
4490Sstevel@tonic-gate 	mutex_exit(&p->p_lock);
4500Sstevel@tonic-gate 
4510Sstevel@tonic-gate 	/*
4520Sstevel@tonic-gate 	 * We clean up the pid provider for this process here; user-land
4530Sstevel@tonic-gate 	 * static probes are handled by the meta-provider remove entry point.
4540Sstevel@tonic-gate 	 */
4550Sstevel@tonic-gate 	if ((provider = fasttrap_provider_lookup(p->p_pid,
4560Sstevel@tonic-gate 	    FASTTRAP_PID_NAME, NULL)) != NULL)
4570Sstevel@tonic-gate 		fasttrap_provider_retire(provider);
4580Sstevel@tonic-gate 
4590Sstevel@tonic-gate 	mutex_enter(&p->p_lock);
4600Sstevel@tonic-gate }
4610Sstevel@tonic-gate 
4620Sstevel@tonic-gate 
4630Sstevel@tonic-gate /*ARGSUSED*/
4640Sstevel@tonic-gate static void
4650Sstevel@tonic-gate fasttrap_pid_provide(void *arg, const dtrace_probedesc_t *desc)
4660Sstevel@tonic-gate {
4670Sstevel@tonic-gate 	/*
4680Sstevel@tonic-gate 	 * There are no "default" pid probes.
4690Sstevel@tonic-gate 	 */
4700Sstevel@tonic-gate }
4710Sstevel@tonic-gate 
4720Sstevel@tonic-gate /*ARGSUSED*/
4730Sstevel@tonic-gate static void
4740Sstevel@tonic-gate fasttrap_provide(void *arg, const dtrace_probedesc_t *desc)
4750Sstevel@tonic-gate {
4760Sstevel@tonic-gate 	if (dtrace_probe_lookup(fasttrap_id, NULL, "fasttrap", "fasttrap") == 0)
4770Sstevel@tonic-gate 		fasttrap_probe_id = dtrace_probe_create(fasttrap_id, NULL,
4780Sstevel@tonic-gate 		    "fasttrap", "fasttrap", FASTTRAP_AFRAMES, NULL);
4790Sstevel@tonic-gate }
4800Sstevel@tonic-gate 
4810Sstevel@tonic-gate static int
4820Sstevel@tonic-gate fasttrap_tracepoint_enable(proc_t *p, fasttrap_probe_t *probe, uint_t index)
4830Sstevel@tonic-gate {
4840Sstevel@tonic-gate 	fasttrap_tracepoint_t *tp, *new_tp = NULL;
4850Sstevel@tonic-gate 	fasttrap_bucket_t *bucket;
4860Sstevel@tonic-gate 	fasttrap_id_t *id;
4870Sstevel@tonic-gate 	pid_t pid;
4880Sstevel@tonic-gate 	uintptr_t pc;
4890Sstevel@tonic-gate 
4900Sstevel@tonic-gate 	ASSERT(index < probe->ftp_ntps);
4910Sstevel@tonic-gate 
4920Sstevel@tonic-gate 	pid = probe->ftp_pid;
4930Sstevel@tonic-gate 	pc = probe->ftp_tps[index].fit_tp->ftt_pc;
4940Sstevel@tonic-gate 	id = &probe->ftp_tps[index].fit_id;
4950Sstevel@tonic-gate 
4960Sstevel@tonic-gate 	ASSERT(probe->ftp_tps[index].fit_tp->ftt_pid == pid);
4970Sstevel@tonic-gate 
4980Sstevel@tonic-gate 	ASSERT(!(p->p_flag & SVFORK));
4990Sstevel@tonic-gate 
5000Sstevel@tonic-gate 	/*
5010Sstevel@tonic-gate 	 * Before we make any modifications, make sure we've imposed a barrier
5020Sstevel@tonic-gate 	 * on the generation in which this probe was last modified.
5030Sstevel@tonic-gate 	 */
5040Sstevel@tonic-gate 	fasttrap_mod_barrier(probe->ftp_gen);
5050Sstevel@tonic-gate 
5060Sstevel@tonic-gate 	bucket = &fasttrap_tpoints.fth_table[FASTTRAP_TPOINTS_INDEX(pid, pc)];
5070Sstevel@tonic-gate 
5080Sstevel@tonic-gate 	/*
5090Sstevel@tonic-gate 	 * If the tracepoint has already been enabled, just add our id to the
5100Sstevel@tonic-gate 	 * list of interested probes. This may be our second time through
5110Sstevel@tonic-gate 	 * this path in which case we'll have constructed the tracepoint we'd
5120Sstevel@tonic-gate 	 * like to install. If we can't find a match, and have an allocated
5130Sstevel@tonic-gate 	 * tracepoint ready to go, enable that one now.
5140Sstevel@tonic-gate 	 *
515532Sahl 	 * A tracepoint whose proc is defunct is also considered defunct.
5160Sstevel@tonic-gate 	 */
5170Sstevel@tonic-gate again:
5180Sstevel@tonic-gate 	mutex_enter(&bucket->ftb_mtx);
5190Sstevel@tonic-gate 	for (tp = bucket->ftb_data; tp != NULL; tp = tp->ftt_next) {
5200Sstevel@tonic-gate 		if (tp->ftt_pid != pid || tp->ftt_pc != pc ||
521532Sahl 		    tp->ftt_proc->ftpc_defunct)
5220Sstevel@tonic-gate 			continue;
5230Sstevel@tonic-gate 
5240Sstevel@tonic-gate 		/*
5250Sstevel@tonic-gate 		 * Now that we've found a matching tracepoint, it would be
5260Sstevel@tonic-gate 		 * a decent idea to confirm that the tracepoint is still
5270Sstevel@tonic-gate 		 * enabled and the trap instruction hasn't been overwritten.
5280Sstevel@tonic-gate 		 * Since this is a little hairy, we'll punt for now.
5290Sstevel@tonic-gate 		 */
5300Sstevel@tonic-gate 
5310Sstevel@tonic-gate 		/*
5320Sstevel@tonic-gate 		 * This can't be the first interested probe. We don't have
5330Sstevel@tonic-gate 		 * to worry about another thread being in the midst of
5340Sstevel@tonic-gate 		 * deleting this tracepoint (which would be the only valid
5350Sstevel@tonic-gate 		 * reason for a tracepoint to have no interested probes)
5360Sstevel@tonic-gate 		 * since we're holding P_PR_LOCK for this process.
5370Sstevel@tonic-gate 		 */
5380Sstevel@tonic-gate 		ASSERT(tp->ftt_ids != NULL || tp->ftt_retids != NULL);
5390Sstevel@tonic-gate 
5400Sstevel@tonic-gate 		if (probe->ftp_type == DTFTP_RETURN ||
5410Sstevel@tonic-gate 		    probe->ftp_type == DTFTP_POST_OFFSETS) {
5420Sstevel@tonic-gate 			id->fti_next = tp->ftt_retids;
5430Sstevel@tonic-gate 			membar_producer();
5440Sstevel@tonic-gate 			tp->ftt_retids = id;
5450Sstevel@tonic-gate 			membar_producer();
5460Sstevel@tonic-gate 		} else {
5470Sstevel@tonic-gate 			id->fti_next = tp->ftt_ids;
5480Sstevel@tonic-gate 			membar_producer();
5490Sstevel@tonic-gate 			tp->ftt_ids = id;
5500Sstevel@tonic-gate 			membar_producer();
5510Sstevel@tonic-gate 		}
5520Sstevel@tonic-gate 
5530Sstevel@tonic-gate 		mutex_exit(&bucket->ftb_mtx);
5540Sstevel@tonic-gate 
5550Sstevel@tonic-gate 		if (new_tp != NULL) {
5560Sstevel@tonic-gate 			new_tp->ftt_ids = NULL;
5570Sstevel@tonic-gate 			new_tp->ftt_retids = NULL;
5580Sstevel@tonic-gate 		}
5590Sstevel@tonic-gate 
5600Sstevel@tonic-gate 		return (0);
5610Sstevel@tonic-gate 	}
5620Sstevel@tonic-gate 
5630Sstevel@tonic-gate 	/*
5640Sstevel@tonic-gate 	 * If we have a good tracepoint ready to go, install it now while
5650Sstevel@tonic-gate 	 * we have the lock held and no one can screw with us.
5660Sstevel@tonic-gate 	 */
5670Sstevel@tonic-gate 	if (new_tp != NULL) {
568315Sahl 		int rc = 0;
5690Sstevel@tonic-gate 
5700Sstevel@tonic-gate 		new_tp->ftt_next = bucket->ftb_data;
5710Sstevel@tonic-gate 		membar_producer();
5720Sstevel@tonic-gate 		bucket->ftb_data = new_tp;
5730Sstevel@tonic-gate 		membar_producer();
5740Sstevel@tonic-gate 		mutex_exit(&bucket->ftb_mtx);
5750Sstevel@tonic-gate 
5760Sstevel@tonic-gate 		/*
577315Sahl 		 * Activate the tracepoint in the ISA-specific manner.
578315Sahl 		 * If this fails, we need to report the failure, but
579315Sahl 		 * indicate that this tracepoint must still be disabled
580315Sahl 		 * by calling fasttrap_tracepoint_disable().
5810Sstevel@tonic-gate 		 */
582315Sahl 		if (fasttrap_tracepoint_install(p, new_tp) != 0)
583315Sahl 			rc = FASTTRAP_ENABLE_PARTIAL;
584315Sahl 
585315Sahl 		/*
586315Sahl 		 * Increment the count of the number of tracepoints active in
587315Sahl 		 * the victim process.
588315Sahl 		 */
589315Sahl 		ASSERT(p->p_proc_flag & P_PR_LOCK);
590315Sahl 		p->p_dtrace_count++;
5910Sstevel@tonic-gate 
5920Sstevel@tonic-gate 		return (rc);
5930Sstevel@tonic-gate 	}
5940Sstevel@tonic-gate 
5950Sstevel@tonic-gate 	mutex_exit(&bucket->ftb_mtx);
5960Sstevel@tonic-gate 
5970Sstevel@tonic-gate 	/*
5980Sstevel@tonic-gate 	 * Initialize the tracepoint that's been preallocated with the probe.
5990Sstevel@tonic-gate 	 */
6000Sstevel@tonic-gate 	new_tp = probe->ftp_tps[index].fit_tp;
6010Sstevel@tonic-gate 
6020Sstevel@tonic-gate 	ASSERT(new_tp->ftt_pid == pid);
6030Sstevel@tonic-gate 	ASSERT(new_tp->ftt_pc == pc);
604532Sahl 	ASSERT(new_tp->ftt_proc == probe->ftp_prov->ftp_proc);
6050Sstevel@tonic-gate 	ASSERT(new_tp->ftt_ids == NULL);
6060Sstevel@tonic-gate 	ASSERT(new_tp->ftt_retids == NULL);
6070Sstevel@tonic-gate 
6080Sstevel@tonic-gate 	if (probe->ftp_type == DTFTP_RETURN ||
6090Sstevel@tonic-gate 	    probe->ftp_type == DTFTP_POST_OFFSETS) {
6100Sstevel@tonic-gate 		id->fti_next = NULL;
6110Sstevel@tonic-gate 		new_tp->ftt_retids = id;
6120Sstevel@tonic-gate 	} else {
6130Sstevel@tonic-gate 		id->fti_next = NULL;
6140Sstevel@tonic-gate 		new_tp->ftt_ids = id;
6150Sstevel@tonic-gate 	}
6160Sstevel@tonic-gate 
6170Sstevel@tonic-gate 	/*
618315Sahl 	 * If the ISA-dependent initialization goes to plan, go back to the
6190Sstevel@tonic-gate 	 * beginning and try to install this freshly made tracepoint.
6200Sstevel@tonic-gate 	 */
6210Sstevel@tonic-gate 	if (fasttrap_tracepoint_init(p, probe, new_tp, pc) == 0)
6220Sstevel@tonic-gate 		goto again;
6230Sstevel@tonic-gate 
6240Sstevel@tonic-gate 	new_tp->ftt_ids = NULL;
6250Sstevel@tonic-gate 	new_tp->ftt_retids = NULL;
6260Sstevel@tonic-gate 
627315Sahl 	return (FASTTRAP_ENABLE_FAIL);
6280Sstevel@tonic-gate }
6290Sstevel@tonic-gate 
6300Sstevel@tonic-gate static void
6310Sstevel@tonic-gate fasttrap_tracepoint_disable(proc_t *p, fasttrap_probe_t *probe, uint_t index)
6320Sstevel@tonic-gate {
6330Sstevel@tonic-gate 	fasttrap_bucket_t *bucket;
6340Sstevel@tonic-gate 	fasttrap_provider_t *provider = probe->ftp_prov;
6350Sstevel@tonic-gate 	fasttrap_tracepoint_t **pp, *tp;
6360Sstevel@tonic-gate 	fasttrap_id_t *id, **idp;
6370Sstevel@tonic-gate 	pid_t pid;
6380Sstevel@tonic-gate 	uintptr_t pc;
6390Sstevel@tonic-gate 
6400Sstevel@tonic-gate 	ASSERT(index < probe->ftp_ntps);
6410Sstevel@tonic-gate 
6420Sstevel@tonic-gate 	pid = probe->ftp_pid;
6430Sstevel@tonic-gate 	pc = probe->ftp_tps[index].fit_tp->ftt_pc;
6440Sstevel@tonic-gate 
6450Sstevel@tonic-gate 	ASSERT(probe->ftp_tps[index].fit_tp->ftt_pid == pid);
6460Sstevel@tonic-gate 
6470Sstevel@tonic-gate 	/*
6480Sstevel@tonic-gate 	 * Find the tracepoint and make sure that our id is one of the
6490Sstevel@tonic-gate 	 * ones registered with it.
6500Sstevel@tonic-gate 	 */
6510Sstevel@tonic-gate 	bucket = &fasttrap_tpoints.fth_table[FASTTRAP_TPOINTS_INDEX(pid, pc)];
6520Sstevel@tonic-gate 	mutex_enter(&bucket->ftb_mtx);
6530Sstevel@tonic-gate 	for (tp = bucket->ftb_data; tp != NULL; tp = tp->ftt_next) {
6540Sstevel@tonic-gate 		if (tp->ftt_pid == pid && tp->ftt_pc == pc &&
655532Sahl 		    tp->ftt_proc == provider->ftp_proc)
6560Sstevel@tonic-gate 			break;
6570Sstevel@tonic-gate 	}
6580Sstevel@tonic-gate 
6590Sstevel@tonic-gate 	/*
6600Sstevel@tonic-gate 	 * If we somehow lost this tracepoint, we're in a world of hurt.
6610Sstevel@tonic-gate 	 */
6620Sstevel@tonic-gate 	ASSERT(tp != NULL);
6630Sstevel@tonic-gate 
6640Sstevel@tonic-gate 	if (probe->ftp_type == DTFTP_RETURN ||
6650Sstevel@tonic-gate 	    probe->ftp_type == DTFTP_POST_OFFSETS) {
6660Sstevel@tonic-gate 		ASSERT(tp->ftt_retids != NULL);
6670Sstevel@tonic-gate 		idp = &tp->ftt_retids;
6680Sstevel@tonic-gate 	} else {
6690Sstevel@tonic-gate 		ASSERT(tp->ftt_ids != NULL);
6700Sstevel@tonic-gate 		idp = &tp->ftt_ids;
6710Sstevel@tonic-gate 	}
6720Sstevel@tonic-gate 
6730Sstevel@tonic-gate 	while ((*idp)->fti_probe != probe) {
6740Sstevel@tonic-gate 		idp = &(*idp)->fti_next;
6750Sstevel@tonic-gate 		ASSERT(*idp != NULL);
6760Sstevel@tonic-gate 	}
6770Sstevel@tonic-gate 
6780Sstevel@tonic-gate 	id = *idp;
6790Sstevel@tonic-gate 	*idp = id->fti_next;
6800Sstevel@tonic-gate 	membar_producer();
6810Sstevel@tonic-gate 
6820Sstevel@tonic-gate 	ASSERT(id->fti_probe == probe);
6830Sstevel@tonic-gate 
6840Sstevel@tonic-gate 	/*
6850Sstevel@tonic-gate 	 * If there are other registered enablings of this tracepoint, we're
6860Sstevel@tonic-gate 	 * all done, but if this was the last probe assocated with this
6870Sstevel@tonic-gate 	 * this tracepoint, we need to remove and free it.
6880Sstevel@tonic-gate 	 */
6890Sstevel@tonic-gate 	if (tp->ftt_ids != NULL || tp->ftt_retids != NULL) {
6900Sstevel@tonic-gate 
6910Sstevel@tonic-gate 		/*
6920Sstevel@tonic-gate 		 * If the current probe's tracepoint is in use, swap it
6930Sstevel@tonic-gate 		 * for an unused tracepoint.
6940Sstevel@tonic-gate 		 */
6950Sstevel@tonic-gate 		if (tp == probe->ftp_tps[index].fit_tp) {
6960Sstevel@tonic-gate 			fasttrap_probe_t *tmp_probe;
6970Sstevel@tonic-gate 			fasttrap_tracepoint_t **tmp_tp;
6980Sstevel@tonic-gate 			uint_t tmp_index;
6990Sstevel@tonic-gate 
7000Sstevel@tonic-gate 			if (tp->ftt_ids != NULL) {
7010Sstevel@tonic-gate 				tmp_probe = tp->ftt_ids->fti_probe;
7020Sstevel@tonic-gate 				tmp_index = FASTTRAP_ID_INDEX(tp->ftt_ids);
7030Sstevel@tonic-gate 				tmp_tp = &tmp_probe->ftp_tps[tmp_index].fit_tp;
7040Sstevel@tonic-gate 			} else {
7050Sstevel@tonic-gate 				tmp_probe = tp->ftt_retids->fti_probe;
7060Sstevel@tonic-gate 				tmp_index = FASTTRAP_ID_INDEX(tp->ftt_retids);
7070Sstevel@tonic-gate 				tmp_tp = &tmp_probe->ftp_tps[tmp_index].fit_tp;
7080Sstevel@tonic-gate 			}
7090Sstevel@tonic-gate 
7100Sstevel@tonic-gate 			ASSERT(*tmp_tp != NULL);
7110Sstevel@tonic-gate 			ASSERT(*tmp_tp != probe->ftp_tps[index].fit_tp);
7120Sstevel@tonic-gate 			ASSERT((*tmp_tp)->ftt_ids == NULL);
7130Sstevel@tonic-gate 			ASSERT((*tmp_tp)->ftt_retids == NULL);
7140Sstevel@tonic-gate 
7150Sstevel@tonic-gate 			probe->ftp_tps[index].fit_tp = *tmp_tp;
7160Sstevel@tonic-gate 			*tmp_tp = tp;
7170Sstevel@tonic-gate 
7180Sstevel@tonic-gate 		}
7190Sstevel@tonic-gate 
7200Sstevel@tonic-gate 		mutex_exit(&bucket->ftb_mtx);
7210Sstevel@tonic-gate 
7220Sstevel@tonic-gate 		/*
7230Sstevel@tonic-gate 		 * Tag the modified probe with the generation in which it was
7240Sstevel@tonic-gate 		 * changed.
7250Sstevel@tonic-gate 		 */
7260Sstevel@tonic-gate 		probe->ftp_gen = fasttrap_mod_gen;
7270Sstevel@tonic-gate 		return;
7280Sstevel@tonic-gate 	}
7290Sstevel@tonic-gate 
7300Sstevel@tonic-gate 	mutex_exit(&bucket->ftb_mtx);
7310Sstevel@tonic-gate 
7320Sstevel@tonic-gate 	/*
7330Sstevel@tonic-gate 	 * We can't safely remove the tracepoint from the set of active
7340Sstevel@tonic-gate 	 * tracepoints until we've actually removed the fasttrap instruction
7350Sstevel@tonic-gate 	 * from the process's text. We can, however, operate on this
7360Sstevel@tonic-gate 	 * tracepoint secure in the knowledge that no other thread is going to
7370Sstevel@tonic-gate 	 * be looking at it since we hold P_PR_LOCK on the process if it's
7380Sstevel@tonic-gate 	 * live or we hold the provider lock on the process if it's dead and
7390Sstevel@tonic-gate 	 * gone.
7400Sstevel@tonic-gate 	 */
7410Sstevel@tonic-gate 
7420Sstevel@tonic-gate 	/*
7430Sstevel@tonic-gate 	 * We only need to remove the actual instruction if we're looking
7440Sstevel@tonic-gate 	 * at an existing process
7450Sstevel@tonic-gate 	 */
7460Sstevel@tonic-gate 	if (p != NULL) {
7470Sstevel@tonic-gate 		/*
7480Sstevel@tonic-gate 		 * If we fail to restore the instruction we need to kill
7490Sstevel@tonic-gate 		 * this process since it's in a completely unrecoverable
7500Sstevel@tonic-gate 		 * state.
7510Sstevel@tonic-gate 		 */
7520Sstevel@tonic-gate 		if (fasttrap_tracepoint_remove(p, tp) != 0)
7530Sstevel@tonic-gate 			fasttrap_sigtrap(p, NULL, pc);
7540Sstevel@tonic-gate 
7550Sstevel@tonic-gate 		/*
7560Sstevel@tonic-gate 		 * Decrement the count of the number of tracepoints active
7570Sstevel@tonic-gate 		 * in the victim process.
7580Sstevel@tonic-gate 		 */
7590Sstevel@tonic-gate 		ASSERT(p->p_proc_flag & P_PR_LOCK);
7600Sstevel@tonic-gate 		p->p_dtrace_count--;
7610Sstevel@tonic-gate 	}
7620Sstevel@tonic-gate 
7630Sstevel@tonic-gate 	/*
7640Sstevel@tonic-gate 	 * Remove the probe from the hash table of active tracepoints.
7650Sstevel@tonic-gate 	 */
7660Sstevel@tonic-gate 	mutex_enter(&bucket->ftb_mtx);
7670Sstevel@tonic-gate 	pp = (fasttrap_tracepoint_t **)&bucket->ftb_data;
7680Sstevel@tonic-gate 	ASSERT(*pp != NULL);
7690Sstevel@tonic-gate 	while (*pp != tp) {
7700Sstevel@tonic-gate 		pp = &(*pp)->ftt_next;
7710Sstevel@tonic-gate 		ASSERT(*pp != NULL);
7720Sstevel@tonic-gate 	}
7730Sstevel@tonic-gate 
7740Sstevel@tonic-gate 	*pp = tp->ftt_next;
7750Sstevel@tonic-gate 	membar_producer();
7760Sstevel@tonic-gate 
7770Sstevel@tonic-gate 	mutex_exit(&bucket->ftb_mtx);
7780Sstevel@tonic-gate 
7790Sstevel@tonic-gate 	/*
7800Sstevel@tonic-gate 	 * Tag the modified probe with the generation in which it was changed.
7810Sstevel@tonic-gate 	 */
7820Sstevel@tonic-gate 	probe->ftp_gen = fasttrap_mod_gen;
7830Sstevel@tonic-gate }
7840Sstevel@tonic-gate 
7850Sstevel@tonic-gate typedef int fasttrap_probe_f(struct regs *);
7860Sstevel@tonic-gate 
7870Sstevel@tonic-gate static void
7880Sstevel@tonic-gate fasttrap_enable_common(int *count, fasttrap_probe_f **fptr, fasttrap_probe_f *f,
7890Sstevel@tonic-gate     fasttrap_probe_f **fptr2, fasttrap_probe_f *f2)
7900Sstevel@tonic-gate {
7910Sstevel@tonic-gate 	/*
7920Sstevel@tonic-gate 	 * We don't have to play the rw lock game here because we're
7930Sstevel@tonic-gate 	 * providing something rather than taking something away --
7940Sstevel@tonic-gate 	 * we can be sure that no threads have tried to follow this
7950Sstevel@tonic-gate 	 * function pointer yet.
7960Sstevel@tonic-gate 	 */
7970Sstevel@tonic-gate 	mutex_enter(&fasttrap_count_mtx);
7980Sstevel@tonic-gate 	if (*count == 0) {
7990Sstevel@tonic-gate 		ASSERT(*fptr == NULL);
8000Sstevel@tonic-gate 		*fptr = f;
8010Sstevel@tonic-gate 		if (fptr2 != NULL)
8020Sstevel@tonic-gate 			*fptr2 = f2;
8030Sstevel@tonic-gate 	}
8040Sstevel@tonic-gate 	ASSERT(*fptr == f);
8050Sstevel@tonic-gate 	ASSERT(fptr2 == NULL || *fptr2 == f2);
8060Sstevel@tonic-gate 	(*count)++;
8070Sstevel@tonic-gate 	mutex_exit(&fasttrap_count_mtx);
8080Sstevel@tonic-gate }
8090Sstevel@tonic-gate 
8100Sstevel@tonic-gate static void
8110Sstevel@tonic-gate fasttrap_disable_common(int *count, fasttrap_probe_f **fptr,
8120Sstevel@tonic-gate     fasttrap_probe_f **fptr2)
8130Sstevel@tonic-gate {
8140Sstevel@tonic-gate 	ASSERT(MUTEX_HELD(&cpu_lock));
8150Sstevel@tonic-gate 
8160Sstevel@tonic-gate 	mutex_enter(&fasttrap_count_mtx);
8170Sstevel@tonic-gate 	(*count)--;
8180Sstevel@tonic-gate 	ASSERT(*count >= 0);
8190Sstevel@tonic-gate 	if (*count == 0) {
8200Sstevel@tonic-gate 		cpu_t *cur, *cpu = CPU;
8210Sstevel@tonic-gate 
8220Sstevel@tonic-gate 		for (cur = cpu->cpu_next_onln; cur != cpu;
8230Sstevel@tonic-gate 			cur = cur->cpu_next_onln) {
8240Sstevel@tonic-gate 			rw_enter(&cur->cpu_ft_lock, RW_WRITER);
8250Sstevel@tonic-gate 		}
8260Sstevel@tonic-gate 
8270Sstevel@tonic-gate 		*fptr = NULL;
8280Sstevel@tonic-gate 		if (fptr2 != NULL)
8290Sstevel@tonic-gate 			*fptr2 = NULL;
8300Sstevel@tonic-gate 
8310Sstevel@tonic-gate 		for (cur = cpu->cpu_next_onln; cur != cpu;
8320Sstevel@tonic-gate 			cur = cur->cpu_next_onln) {
8330Sstevel@tonic-gate 			rw_exit(&cur->cpu_ft_lock);
8340Sstevel@tonic-gate 		}
8350Sstevel@tonic-gate 	}
8360Sstevel@tonic-gate 	mutex_exit(&fasttrap_count_mtx);
8370Sstevel@tonic-gate }
8380Sstevel@tonic-gate 
8390Sstevel@tonic-gate /*ARGSUSED*/
8400Sstevel@tonic-gate static void
8410Sstevel@tonic-gate fasttrap_enable(void *arg, dtrace_id_t id, void *parg)
8420Sstevel@tonic-gate {
8430Sstevel@tonic-gate 	/*
8440Sstevel@tonic-gate 	 * Enable the probe that corresponds to statically placed trace
8450Sstevel@tonic-gate 	 * points which have not explicitly been placed in the process's text
8460Sstevel@tonic-gate 	 * by the fasttrap provider.
8470Sstevel@tonic-gate 	 */
8480Sstevel@tonic-gate 	ASSERT(arg == NULL);
8490Sstevel@tonic-gate 	ASSERT(id == fasttrap_probe_id);
8500Sstevel@tonic-gate 
8510Sstevel@tonic-gate 	fasttrap_enable_common(&fasttrap_count,
8520Sstevel@tonic-gate 	    &dtrace_fasttrap_probe_ptr, fasttrap_probe, NULL, NULL);
8530Sstevel@tonic-gate }
8540Sstevel@tonic-gate 
8550Sstevel@tonic-gate 
8560Sstevel@tonic-gate /*ARGSUSED*/
8570Sstevel@tonic-gate static void
8580Sstevel@tonic-gate fasttrap_pid_enable(void *arg, dtrace_id_t id, void *parg)
8590Sstevel@tonic-gate {
8600Sstevel@tonic-gate 	fasttrap_probe_t *probe = parg;
8610Sstevel@tonic-gate 	proc_t *p;
862315Sahl 	int i, rc;
8630Sstevel@tonic-gate 
8640Sstevel@tonic-gate 	ASSERT(probe != NULL);
8650Sstevel@tonic-gate 	ASSERT(!probe->ftp_enabled);
8660Sstevel@tonic-gate 	ASSERT(id == probe->ftp_id);
8670Sstevel@tonic-gate 	ASSERT(MUTEX_HELD(&cpu_lock));
8680Sstevel@tonic-gate 
8690Sstevel@tonic-gate 	/*
8700Sstevel@tonic-gate 	 * Increment the count of enabled probes on this probe's provider;
8710Sstevel@tonic-gate 	 * the provider can't go away while the probe still exists. We
8720Sstevel@tonic-gate 	 * must increment this even if we aren't able to properly enable
8730Sstevel@tonic-gate 	 * this probe.
8740Sstevel@tonic-gate 	 */
8750Sstevel@tonic-gate 	mutex_enter(&probe->ftp_prov->ftp_mtx);
8760Sstevel@tonic-gate 	probe->ftp_prov->ftp_rcount++;
8770Sstevel@tonic-gate 	mutex_exit(&probe->ftp_prov->ftp_mtx);
8780Sstevel@tonic-gate 
8790Sstevel@tonic-gate 	/*
8800Sstevel@tonic-gate 	 * Bail out if we can't find the process for this probe or its
881532Sahl 	 * provider is retired (meaning it was valid in a previously exec'ed
8820Sstevel@tonic-gate 	 * incarnation of this address space). The provider can't go away
8830Sstevel@tonic-gate 	 * while we're in this code path.
8840Sstevel@tonic-gate 	 */
885532Sahl 	if (probe->ftp_prov->ftp_retired ||
8860Sstevel@tonic-gate 	    (p = sprlock(probe->ftp_pid)) == NULL)
8870Sstevel@tonic-gate 		return;
8880Sstevel@tonic-gate 
8890Sstevel@tonic-gate 	ASSERT(!(p->p_flag & SVFORK));
8900Sstevel@tonic-gate 	mutex_exit(&p->p_lock);
8910Sstevel@tonic-gate 
8920Sstevel@tonic-gate 	/*
8930Sstevel@tonic-gate 	 * We have to enable the trap entry before any user threads have
8940Sstevel@tonic-gate 	 * the chance to execute the trap instruction we're about to place
8950Sstevel@tonic-gate 	 * in their process's text.
8960Sstevel@tonic-gate 	 */
8970Sstevel@tonic-gate 	fasttrap_enable_common(&fasttrap_pid_count,
8980Sstevel@tonic-gate 	    &dtrace_pid_probe_ptr, fasttrap_pid_probe,
8990Sstevel@tonic-gate 	    &dtrace_return_probe_ptr, fasttrap_return_probe);
9000Sstevel@tonic-gate 
9010Sstevel@tonic-gate 	/*
9020Sstevel@tonic-gate 	 * Enable all the tracepoints and add this probe's id to each
9030Sstevel@tonic-gate 	 * tracepoint's list of active probes.
9040Sstevel@tonic-gate 	 */
9050Sstevel@tonic-gate 	for (i = 0; i < probe->ftp_ntps; i++) {
906315Sahl 		if ((rc = fasttrap_tracepoint_enable(p, probe, i)) != 0) {
907315Sahl 			/*
908315Sahl 			 * If enabling the tracepoint failed completely,
909315Sahl 			 * we don't have to disable it; if the failure
910315Sahl 			 * was only partial we must disable it.
911315Sahl 			 */
912315Sahl 			if (rc == FASTTRAP_ENABLE_FAIL)
913315Sahl 				i--;
914315Sahl 			else
915315Sahl 				ASSERT(rc == FASTTRAP_ENABLE_PARTIAL);
916315Sahl 
9170Sstevel@tonic-gate 			/*
9180Sstevel@tonic-gate 			 * Back up and pull out all the tracepoints we've
9190Sstevel@tonic-gate 			 * created so far for this probe.
9200Sstevel@tonic-gate 			 */
921*665Sahl 			while (i >= 0) {
9220Sstevel@tonic-gate 				fasttrap_tracepoint_disable(p, probe, i);
923*665Sahl 				i--;
9240Sstevel@tonic-gate 			}
9250Sstevel@tonic-gate 
9260Sstevel@tonic-gate 			mutex_enter(&p->p_lock);
9270Sstevel@tonic-gate 			sprunlock(p);
9280Sstevel@tonic-gate 
9290Sstevel@tonic-gate 			/*
9300Sstevel@tonic-gate 			 * Since we're not actually enabling this probe,
9310Sstevel@tonic-gate 			 * drop our reference on the trap table entry.
9320Sstevel@tonic-gate 			 */
9330Sstevel@tonic-gate 			fasttrap_disable_common(&fasttrap_pid_count,
9340Sstevel@tonic-gate 			    &dtrace_pid_probe_ptr, &dtrace_return_probe_ptr);
9350Sstevel@tonic-gate 			return;
9360Sstevel@tonic-gate 		}
9370Sstevel@tonic-gate 	}
9380Sstevel@tonic-gate 
9390Sstevel@tonic-gate 	mutex_enter(&p->p_lock);
9400Sstevel@tonic-gate 	sprunlock(p);
9410Sstevel@tonic-gate 
9420Sstevel@tonic-gate 	probe->ftp_enabled = 1;
9430Sstevel@tonic-gate }
9440Sstevel@tonic-gate 
9450Sstevel@tonic-gate 
9460Sstevel@tonic-gate /*ARGSUSED*/
9470Sstevel@tonic-gate static void
9480Sstevel@tonic-gate fasttrap_disable(void *arg, dtrace_id_t id, void *parg)
9490Sstevel@tonic-gate {
9500Sstevel@tonic-gate 	/*
9510Sstevel@tonic-gate 	 * Disable the probe the corresponds to statically placed trace
9520Sstevel@tonic-gate 	 * points.
9530Sstevel@tonic-gate 	 */
9540Sstevel@tonic-gate 	ASSERT(arg == NULL);
9550Sstevel@tonic-gate 	ASSERT(id == fasttrap_probe_id);
9560Sstevel@tonic-gate 	ASSERT(MUTEX_HELD(&cpu_lock));
9570Sstevel@tonic-gate 	fasttrap_disable_common(&fasttrap_count, &dtrace_fasttrap_probe_ptr,
9580Sstevel@tonic-gate 	    NULL);
9590Sstevel@tonic-gate }
9600Sstevel@tonic-gate 
9610Sstevel@tonic-gate /*ARGSUSED*/
9620Sstevel@tonic-gate static void
9630Sstevel@tonic-gate fasttrap_pid_disable(void *arg, dtrace_id_t id, void *parg)
9640Sstevel@tonic-gate {
9650Sstevel@tonic-gate 	fasttrap_probe_t *probe = parg;
9660Sstevel@tonic-gate 	fasttrap_provider_t *provider = probe->ftp_prov;
9670Sstevel@tonic-gate 	proc_t *p;
9680Sstevel@tonic-gate 	int i, whack = 0;
9690Sstevel@tonic-gate 
9700Sstevel@tonic-gate 	if (!probe->ftp_enabled) {
9710Sstevel@tonic-gate 		mutex_enter(&provider->ftp_mtx);
9720Sstevel@tonic-gate 		provider->ftp_rcount--;
9730Sstevel@tonic-gate 		ASSERT(provider->ftp_rcount >= 0);
9740Sstevel@tonic-gate 		mutex_exit(&provider->ftp_mtx);
9750Sstevel@tonic-gate 		return;
9760Sstevel@tonic-gate 	}
9770Sstevel@tonic-gate 
9780Sstevel@tonic-gate 	ASSERT(id == probe->ftp_id);
9790Sstevel@tonic-gate 
9800Sstevel@tonic-gate 	/*
9810Sstevel@tonic-gate 	 * We won't be able to acquire a /proc-esque lock on the process
9820Sstevel@tonic-gate 	 * iff the process is dead and gone. In this case, we rely on the
9830Sstevel@tonic-gate 	 * provider lock as a point of mutual exclusion to prevent other
9840Sstevel@tonic-gate 	 * DTrace consumers from disabling this probe.
9850Sstevel@tonic-gate 	 */
9860Sstevel@tonic-gate 	if ((p = sprlock(probe->ftp_pid)) != NULL) {
9870Sstevel@tonic-gate 		ASSERT(!(p->p_flag & SVFORK));
9880Sstevel@tonic-gate 		mutex_exit(&p->p_lock);
9890Sstevel@tonic-gate 	}
9900Sstevel@tonic-gate 
9910Sstevel@tonic-gate 	mutex_enter(&provider->ftp_mtx);
9920Sstevel@tonic-gate 
9930Sstevel@tonic-gate 	/*
9940Sstevel@tonic-gate 	 * Disable all the associated tracepoints.
9950Sstevel@tonic-gate 	 */
9960Sstevel@tonic-gate 	for (i = 0; i < probe->ftp_ntps; i++) {
9970Sstevel@tonic-gate 		fasttrap_tracepoint_disable(p, probe, i);
9980Sstevel@tonic-gate 	}
9990Sstevel@tonic-gate 
10000Sstevel@tonic-gate 	ASSERT(provider->ftp_rcount > 0);
10010Sstevel@tonic-gate 	provider->ftp_rcount--;
10020Sstevel@tonic-gate 
10030Sstevel@tonic-gate 	if (p != NULL) {
10040Sstevel@tonic-gate 		/*
10050Sstevel@tonic-gate 		 * Even though we may not be able to remove it entirely, we
1006532Sahl 		 * mark this retired provider to get a chance to remove some
10070Sstevel@tonic-gate 		 * of the associated probes.
10080Sstevel@tonic-gate 		 */
1009532Sahl 		if (provider->ftp_retired && !provider->ftp_marked)
10100Sstevel@tonic-gate 			whack = provider->ftp_marked = 1;
10110Sstevel@tonic-gate 		mutex_exit(&provider->ftp_mtx);
10120Sstevel@tonic-gate 
10130Sstevel@tonic-gate 		mutex_enter(&p->p_lock);
10140Sstevel@tonic-gate 		sprunlock(p);
10150Sstevel@tonic-gate 	} else {
10160Sstevel@tonic-gate 		/*
10170Sstevel@tonic-gate 		 * If the process is dead, we're just waiting for the
10180Sstevel@tonic-gate 		 * last probe to be disabled to be able to free it.
10190Sstevel@tonic-gate 		 */
10200Sstevel@tonic-gate 		if (provider->ftp_rcount == 0 && !provider->ftp_marked)
10210Sstevel@tonic-gate 			whack = provider->ftp_marked = 1;
10220Sstevel@tonic-gate 		mutex_exit(&provider->ftp_mtx);
10230Sstevel@tonic-gate 	}
10240Sstevel@tonic-gate 
10250Sstevel@tonic-gate 	if (whack)
10260Sstevel@tonic-gate 		fasttrap_pid_cleanup();
10270Sstevel@tonic-gate 
10280Sstevel@tonic-gate 	probe->ftp_enabled = 0;
10290Sstevel@tonic-gate 
10300Sstevel@tonic-gate 	ASSERT(MUTEX_HELD(&cpu_lock));
10310Sstevel@tonic-gate 	fasttrap_disable_common(&fasttrap_pid_count, &dtrace_pid_probe_ptr,
10320Sstevel@tonic-gate 	    &dtrace_return_probe_ptr);
10330Sstevel@tonic-gate }
10340Sstevel@tonic-gate 
10350Sstevel@tonic-gate /*ARGSUSED*/
10360Sstevel@tonic-gate static void
10370Sstevel@tonic-gate fasttrap_pid_getargdesc(void *arg, dtrace_id_t id, void *parg,
10380Sstevel@tonic-gate     dtrace_argdesc_t *desc)
10390Sstevel@tonic-gate {
10400Sstevel@tonic-gate 	fasttrap_probe_t *probe = parg;
10410Sstevel@tonic-gate 	char *str;
10420Sstevel@tonic-gate 	int i;
10430Sstevel@tonic-gate 
10440Sstevel@tonic-gate 	desc->dtargd_native[0] = '\0';
10450Sstevel@tonic-gate 	desc->dtargd_xlate[0] = '\0';
10460Sstevel@tonic-gate 
1047532Sahl 	if (probe->ftp_prov->ftp_retired != 0 ||
10480Sstevel@tonic-gate 	    desc->dtargd_ndx >= probe->ftp_nargs) {
10490Sstevel@tonic-gate 		desc->dtargd_ndx = DTRACE_ARGNONE;
10500Sstevel@tonic-gate 		return;
10510Sstevel@tonic-gate 	}
10520Sstevel@tonic-gate 
10530Sstevel@tonic-gate 	/*
10540Sstevel@tonic-gate 	 * We only need to set this member if the argument is remapped.
10550Sstevel@tonic-gate 	 */
10560Sstevel@tonic-gate 	if (probe->ftp_argmap != NULL)
10570Sstevel@tonic-gate 		desc->dtargd_mapping = probe->ftp_argmap[desc->dtargd_ndx];
10580Sstevel@tonic-gate 
10590Sstevel@tonic-gate 	str = probe->ftp_ntypes;
10600Sstevel@tonic-gate 	for (i = 0; i < desc->dtargd_mapping; i++) {
10610Sstevel@tonic-gate 		str += strlen(str) + 1;
10620Sstevel@tonic-gate 	}
10630Sstevel@tonic-gate 
10640Sstevel@tonic-gate 	ASSERT(strlen(str + 1) < sizeof (desc->dtargd_native));
10650Sstevel@tonic-gate 	(void) strcpy(desc->dtargd_native, str);
10660Sstevel@tonic-gate 
10670Sstevel@tonic-gate 	if (probe->ftp_xtypes == NULL)
10680Sstevel@tonic-gate 		return;
10690Sstevel@tonic-gate 
10700Sstevel@tonic-gate 	str = probe->ftp_xtypes;
10710Sstevel@tonic-gate 	for (i = 0; i < desc->dtargd_ndx; i++) {
10720Sstevel@tonic-gate 		str += strlen(str) + 1;
10730Sstevel@tonic-gate 	}
10740Sstevel@tonic-gate 
10750Sstevel@tonic-gate 	ASSERT(strlen(str + 1) < sizeof (desc->dtargd_xlate));
10760Sstevel@tonic-gate 	(void) strcpy(desc->dtargd_xlate, str);
10770Sstevel@tonic-gate }
10780Sstevel@tonic-gate 
10790Sstevel@tonic-gate /*ARGSUSED*/
10800Sstevel@tonic-gate static void
10810Sstevel@tonic-gate fasttrap_destroy(void *arg, dtrace_id_t id, void *parg)
10820Sstevel@tonic-gate {
10830Sstevel@tonic-gate 	ASSERT(arg == NULL);
10840Sstevel@tonic-gate 	ASSERT(id == fasttrap_probe_id);
10850Sstevel@tonic-gate }
10860Sstevel@tonic-gate 
10870Sstevel@tonic-gate /*ARGSUSED*/
10880Sstevel@tonic-gate static void
10890Sstevel@tonic-gate fasttrap_pid_destroy(void *arg, dtrace_id_t id, void *parg)
10900Sstevel@tonic-gate {
10910Sstevel@tonic-gate 	fasttrap_probe_t *probe = parg;
10920Sstevel@tonic-gate 	int i;
10930Sstevel@tonic-gate 	size_t size;
10940Sstevel@tonic-gate 
10950Sstevel@tonic-gate 	ASSERT(probe != NULL);
10960Sstevel@tonic-gate 	ASSERT(!probe->ftp_enabled);
10970Sstevel@tonic-gate 	ASSERT(fasttrap_total >= probe->ftp_ntps);
10980Sstevel@tonic-gate 
10990Sstevel@tonic-gate 	atomic_add_32(&fasttrap_total, -probe->ftp_ntps);
11000Sstevel@tonic-gate 	size = sizeof (fasttrap_probe_t) +
11010Sstevel@tonic-gate 	    sizeof (probe->ftp_tps[0]) * (probe->ftp_ntps - 1);
11020Sstevel@tonic-gate 
11030Sstevel@tonic-gate 	if (probe->ftp_gen + 1 >= fasttrap_mod_gen)
11040Sstevel@tonic-gate 		fasttrap_mod_barrier(probe->ftp_gen);
11050Sstevel@tonic-gate 
11060Sstevel@tonic-gate 	for (i = 0; i < probe->ftp_ntps; i++) {
11070Sstevel@tonic-gate 		kmem_free(probe->ftp_tps[i].fit_tp,
11080Sstevel@tonic-gate 		    sizeof (fasttrap_tracepoint_t));
11090Sstevel@tonic-gate 	}
11100Sstevel@tonic-gate 
11110Sstevel@tonic-gate 	kmem_free(probe, size);
11120Sstevel@tonic-gate }
11130Sstevel@tonic-gate 
11140Sstevel@tonic-gate 
11150Sstevel@tonic-gate static const dtrace_pattr_t fasttrap_attr = {
11160Sstevel@tonic-gate { DTRACE_STABILITY_EVOLVING, DTRACE_STABILITY_EVOLVING, DTRACE_CLASS_ISA },
11170Sstevel@tonic-gate { DTRACE_STABILITY_PRIVATE, DTRACE_STABILITY_PRIVATE, DTRACE_CLASS_UNKNOWN },
11180Sstevel@tonic-gate { DTRACE_STABILITY_PRIVATE, DTRACE_STABILITY_PRIVATE, DTRACE_CLASS_UNKNOWN },
11190Sstevel@tonic-gate { DTRACE_STABILITY_EVOLVING, DTRACE_STABILITY_EVOLVING, DTRACE_CLASS_ISA },
11200Sstevel@tonic-gate { DTRACE_STABILITY_EVOLVING, DTRACE_STABILITY_EVOLVING, DTRACE_CLASS_ISA },
11210Sstevel@tonic-gate };
11220Sstevel@tonic-gate 
11230Sstevel@tonic-gate static dtrace_pops_t fasttrap_pops = {
11240Sstevel@tonic-gate 	fasttrap_provide,
11250Sstevel@tonic-gate 	NULL,
11260Sstevel@tonic-gate 	fasttrap_enable,
11270Sstevel@tonic-gate 	fasttrap_disable,
11280Sstevel@tonic-gate 	NULL,
11290Sstevel@tonic-gate 	NULL,
11300Sstevel@tonic-gate 	NULL,
11310Sstevel@tonic-gate 	fasttrap_getarg,
11320Sstevel@tonic-gate 	NULL,
11330Sstevel@tonic-gate 	fasttrap_destroy
11340Sstevel@tonic-gate };
11350Sstevel@tonic-gate 
11360Sstevel@tonic-gate static const dtrace_pattr_t pid_attr = {
11370Sstevel@tonic-gate { DTRACE_STABILITY_EVOLVING, DTRACE_STABILITY_EVOLVING, DTRACE_CLASS_ISA },
11380Sstevel@tonic-gate { DTRACE_STABILITY_PRIVATE, DTRACE_STABILITY_PRIVATE, DTRACE_CLASS_UNKNOWN },
11390Sstevel@tonic-gate { DTRACE_STABILITY_PRIVATE, DTRACE_STABILITY_PRIVATE, DTRACE_CLASS_UNKNOWN },
11400Sstevel@tonic-gate { DTRACE_STABILITY_EVOLVING, DTRACE_STABILITY_EVOLVING, DTRACE_CLASS_ISA },
11410Sstevel@tonic-gate { DTRACE_STABILITY_PRIVATE, DTRACE_STABILITY_PRIVATE, DTRACE_CLASS_UNKNOWN },
11420Sstevel@tonic-gate };
11430Sstevel@tonic-gate 
11440Sstevel@tonic-gate static dtrace_pops_t pid_pops = {
11450Sstevel@tonic-gate 	fasttrap_pid_provide,
11460Sstevel@tonic-gate 	NULL,
11470Sstevel@tonic-gate 	fasttrap_pid_enable,
11480Sstevel@tonic-gate 	fasttrap_pid_disable,
11490Sstevel@tonic-gate 	NULL,
11500Sstevel@tonic-gate 	NULL,
11510Sstevel@tonic-gate 	fasttrap_pid_getargdesc,
11520Sstevel@tonic-gate 	fasttrap_getarg,
11530Sstevel@tonic-gate 	NULL,
11540Sstevel@tonic-gate 	fasttrap_pid_destroy
11550Sstevel@tonic-gate };
11560Sstevel@tonic-gate 
11570Sstevel@tonic-gate static dtrace_pops_t usdt_pops = {
11580Sstevel@tonic-gate 	fasttrap_pid_provide,
11590Sstevel@tonic-gate 	NULL,
11600Sstevel@tonic-gate 	fasttrap_pid_enable,
11610Sstevel@tonic-gate 	fasttrap_pid_disable,
11620Sstevel@tonic-gate 	NULL,
11630Sstevel@tonic-gate 	NULL,
11640Sstevel@tonic-gate 	fasttrap_pid_getargdesc,
11650Sstevel@tonic-gate 	fasttrap_usdt_getarg,
11660Sstevel@tonic-gate 	NULL,
11670Sstevel@tonic-gate 	fasttrap_pid_destroy
11680Sstevel@tonic-gate };
11690Sstevel@tonic-gate 
1170532Sahl static fasttrap_proc_t *
1171532Sahl fasttrap_proc_lookup(pid_t pid)
1172532Sahl {
1173532Sahl 	fasttrap_bucket_t *bucket;
1174532Sahl 	fasttrap_proc_t *fprc, *new_fprc;
1175532Sahl 
1176532Sahl 	bucket = &fasttrap_procs.fth_table[FASTTRAP_PROCS_INDEX(pid)];
1177532Sahl 	mutex_enter(&bucket->ftb_mtx);
1178532Sahl 
1179532Sahl 	for (fprc = bucket->ftb_data; fprc != NULL; fprc = fprc->ftpc_next) {
1180532Sahl 		if (fprc->ftpc_pid == pid && !fprc->ftpc_defunct) {
1181532Sahl 			mutex_enter(&fprc->ftpc_mtx);
1182532Sahl 			mutex_exit(&bucket->ftb_mtx);
1183532Sahl 			fprc->ftpc_count++;
1184532Sahl 			mutex_exit(&fprc->ftpc_mtx);
1185532Sahl 
1186532Sahl 			return (fprc);
1187532Sahl 		}
1188532Sahl 	}
1189532Sahl 
1190532Sahl 	/*
1191532Sahl 	 * Drop the bucket lock so we don't try to perform a sleeping
1192532Sahl 	 * allocation under it.
1193532Sahl 	 */
1194532Sahl 	mutex_exit(&bucket->ftb_mtx);
1195532Sahl 
1196532Sahl 	new_fprc = kmem_zalloc(sizeof (fasttrap_proc_t), KM_SLEEP);
1197532Sahl 	new_fprc->ftpc_pid = pid;
1198532Sahl 	new_fprc->ftpc_count = 1;
1199532Sahl 
1200532Sahl 	mutex_enter(&bucket->ftb_mtx);
1201532Sahl 
1202532Sahl 	/*
1203532Sahl 	 * Take another lap through the list to make sure a proc hasn't
1204532Sahl 	 * been created for this pid while we weren't under the bucket lock.
1205532Sahl 	 */
1206532Sahl 	for (fprc = bucket->ftb_data; fprc != NULL; fprc = fprc->ftpc_next) {
1207532Sahl 		if (fprc->ftpc_pid == pid && !fprc->ftpc_defunct) {
1208532Sahl 			mutex_enter(&fprc->ftpc_mtx);
1209532Sahl 			mutex_exit(&bucket->ftb_mtx);
1210532Sahl 			fprc->ftpc_count++;
1211532Sahl 			mutex_exit(&fprc->ftpc_mtx);
1212532Sahl 
1213532Sahl 			kmem_free(new_fprc, sizeof (fasttrap_proc_t));
1214532Sahl 
1215532Sahl 			return (fprc);
1216532Sahl 		}
1217532Sahl 	}
1218532Sahl 
1219532Sahl 	new_fprc->ftpc_next = bucket->ftb_data;
1220532Sahl 	bucket->ftb_data = new_fprc;
1221532Sahl 
1222532Sahl 	mutex_exit(&bucket->ftb_mtx);
1223532Sahl 
1224532Sahl 	return (new_fprc);
1225532Sahl }
1226532Sahl 
1227532Sahl static void
1228532Sahl fasttrap_proc_release(fasttrap_proc_t *proc)
1229532Sahl {
1230532Sahl 	fasttrap_bucket_t *bucket;
1231532Sahl 	fasttrap_proc_t *fprc, **fprcp;
1232532Sahl 	pid_t pid = proc->ftpc_pid;
1233532Sahl 
1234532Sahl 	mutex_enter(&proc->ftpc_mtx);
1235532Sahl 
1236532Sahl 	ASSERT(proc->ftpc_count != 0);
1237532Sahl 
1238532Sahl 	if (--proc->ftpc_count != 0) {
1239532Sahl 		mutex_exit(&proc->ftpc_mtx);
1240532Sahl 		return;
1241532Sahl 	}
1242532Sahl 
1243532Sahl 	mutex_exit(&proc->ftpc_mtx);
1244532Sahl 
1245532Sahl 	bucket = &fasttrap_procs.fth_table[FASTTRAP_PROCS_INDEX(pid)];
1246532Sahl 	mutex_enter(&bucket->ftb_mtx);
1247532Sahl 
1248532Sahl 	fprcp = (fasttrap_proc_t **)&bucket->ftb_data;
1249532Sahl 	while ((fprc = *fprcp) != NULL) {
1250532Sahl 		if (fprc == proc)
1251532Sahl 			break;
1252532Sahl 
1253532Sahl 		fprcp = &fprc->ftpc_next;
1254532Sahl 	}
1255532Sahl 
1256532Sahl 	/*
1257532Sahl 	 * Something strange has happened if we can't find the proc.
1258532Sahl 	 */
1259532Sahl 	ASSERT(fprc != NULL);
1260532Sahl 
1261532Sahl 	*fprcp = fprc->ftpc_next;
1262532Sahl 
1263532Sahl 	mutex_exit(&bucket->ftb_mtx);
1264532Sahl 
1265532Sahl 	kmem_free(fprc, sizeof (fasttrap_proc_t));
1266532Sahl }
1267532Sahl 
12680Sstevel@tonic-gate /*
12690Sstevel@tonic-gate  * Lookup a fasttrap-managed provider based on its name and associated pid.
12700Sstevel@tonic-gate  * If the pattr argument is non-NULL, this function instantiates the provider
12710Sstevel@tonic-gate  * if it doesn't exist otherwise it returns NULL. The provider is returned
12720Sstevel@tonic-gate  * with its lock held.
12730Sstevel@tonic-gate  */
12740Sstevel@tonic-gate static fasttrap_provider_t *
12750Sstevel@tonic-gate fasttrap_provider_lookup(pid_t pid, const char *name,
12760Sstevel@tonic-gate     const dtrace_pattr_t *pattr)
12770Sstevel@tonic-gate {
12780Sstevel@tonic-gate 	fasttrap_provider_t *fp, *new_fp = NULL;
12790Sstevel@tonic-gate 	fasttrap_bucket_t *bucket;
12800Sstevel@tonic-gate 	char provname[DTRACE_PROVNAMELEN];
12810Sstevel@tonic-gate 	proc_t *p;
12820Sstevel@tonic-gate 	uid_t uid = (uid_t)-1;
12830Sstevel@tonic-gate 
12840Sstevel@tonic-gate 	ASSERT(strlen(name) < sizeof (fp->ftp_name));
12850Sstevel@tonic-gate 
12860Sstevel@tonic-gate 	bucket = &fasttrap_provs.fth_table[FASTTRAP_PROVS_INDEX(pid, name)];
12870Sstevel@tonic-gate 	mutex_enter(&bucket->ftb_mtx);
12880Sstevel@tonic-gate 
12890Sstevel@tonic-gate 	/*
12900Sstevel@tonic-gate 	 * Take a lap through the list and return the match if we find it.
12910Sstevel@tonic-gate 	 */
12920Sstevel@tonic-gate 	for (fp = bucket->ftb_data; fp != NULL; fp = fp->ftp_next) {
12930Sstevel@tonic-gate 		if (fp->ftp_pid == pid && strcmp(fp->ftp_name, name) == 0 &&
1294532Sahl 		    !fp->ftp_retired) {
12950Sstevel@tonic-gate 			mutex_enter(&fp->ftp_mtx);
12960Sstevel@tonic-gate 			mutex_exit(&bucket->ftb_mtx);
12970Sstevel@tonic-gate 			return (fp);
12980Sstevel@tonic-gate 		}
12990Sstevel@tonic-gate 	}
13000Sstevel@tonic-gate 
13010Sstevel@tonic-gate 	/*
13020Sstevel@tonic-gate 	 * Drop the bucket lock so we don't try to perform a sleeping
13030Sstevel@tonic-gate 	 * allocation under it.
13040Sstevel@tonic-gate 	 */
13050Sstevel@tonic-gate 	mutex_exit(&bucket->ftb_mtx);
13060Sstevel@tonic-gate 
1307532Sahl 	/*
1308532Sahl 	 * If we didn't want to create a new provider, just return failure.
1309532Sahl 	 */
13100Sstevel@tonic-gate 	if (pattr == NULL)
13110Sstevel@tonic-gate 		return (NULL);
13120Sstevel@tonic-gate 
13130Sstevel@tonic-gate 	/*
13140Sstevel@tonic-gate 	 * Make sure the process exists, isn't a child created as the result
13150Sstevel@tonic-gate 	 * of a vfork(2), and isn't a zombie (but may be in fork). Record the
13160Sstevel@tonic-gate 	 * process's uid to pass to dtrace_register().
13170Sstevel@tonic-gate 	 */
13180Sstevel@tonic-gate 	mutex_enter(&pidlock);
1319390Sraf 	if ((p = prfind(pid)) == NULL) {
13200Sstevel@tonic-gate 		mutex_exit(&pidlock);
13210Sstevel@tonic-gate 		return (NULL);
13220Sstevel@tonic-gate 	}
13230Sstevel@tonic-gate 	mutex_enter(&p->p_lock);
13240Sstevel@tonic-gate 	mutex_exit(&pidlock);
1325390Sraf 	if (p->p_flag & (SVFORK | SEXITING)) {
1326390Sraf 		mutex_exit(&p->p_lock);
1327390Sraf 		return (NULL);
1328390Sraf 	}
13290Sstevel@tonic-gate 
13300Sstevel@tonic-gate 	/*
13310Sstevel@tonic-gate 	 * Increment p_dtrace_probes so that the process knows to inform us
13320Sstevel@tonic-gate 	 * when it exits or execs. fasttrap_provider_free() decrements this
13330Sstevel@tonic-gate 	 * when we're done with this provider.
13340Sstevel@tonic-gate 	 */
13350Sstevel@tonic-gate 	p->p_dtrace_probes++;
13360Sstevel@tonic-gate 
13370Sstevel@tonic-gate 	mutex_enter(&p->p_crlock);
13380Sstevel@tonic-gate 	uid = crgetruid(p->p_cred);
13390Sstevel@tonic-gate 	mutex_exit(&p->p_crlock);
13400Sstevel@tonic-gate 	mutex_exit(&p->p_lock);
13410Sstevel@tonic-gate 
13420Sstevel@tonic-gate 	new_fp = kmem_zalloc(sizeof (fasttrap_provider_t), KM_SLEEP);
1343532Sahl 	new_fp->ftp_pid = pid;
1344532Sahl 	new_fp->ftp_proc = fasttrap_proc_lookup(pid);
1345532Sahl 
1346532Sahl 	ASSERT(new_fp->ftp_proc != NULL);
13470Sstevel@tonic-gate 
13480Sstevel@tonic-gate 	mutex_enter(&bucket->ftb_mtx);
13490Sstevel@tonic-gate 
13500Sstevel@tonic-gate 	/*
13510Sstevel@tonic-gate 	 * Take another lap through the list to make sure a provider hasn't
13520Sstevel@tonic-gate 	 * been created for this pid while we weren't under the bucket lock.
13530Sstevel@tonic-gate 	 */
13540Sstevel@tonic-gate 	for (fp = bucket->ftb_data; fp != NULL; fp = fp->ftp_next) {
13550Sstevel@tonic-gate 		if (fp->ftp_pid == pid && strcmp(fp->ftp_name, name) == 0 &&
1356532Sahl 		    !fp->ftp_retired) {
13570Sstevel@tonic-gate 			mutex_enter(&fp->ftp_mtx);
13580Sstevel@tonic-gate 			mutex_exit(&bucket->ftb_mtx);
13590Sstevel@tonic-gate 			fasttrap_provider_free(new_fp);
13600Sstevel@tonic-gate 			return (fp);
13610Sstevel@tonic-gate 		}
13620Sstevel@tonic-gate 	}
13630Sstevel@tonic-gate 
13640Sstevel@tonic-gate 	(void) strcpy(new_fp->ftp_name, name);
13650Sstevel@tonic-gate 
13660Sstevel@tonic-gate 	/*
13670Sstevel@tonic-gate 	 * Fail and return NULL if either the provider name is too long
13680Sstevel@tonic-gate 	 * or we fail to register this new provider with the DTrace
13690Sstevel@tonic-gate 	 * framework. Note that this is the only place we ever construct
13700Sstevel@tonic-gate 	 * the full provider name -- we keep it in pieces in the provider
13710Sstevel@tonic-gate 	 * structure.
13720Sstevel@tonic-gate 	 */
13730Sstevel@tonic-gate 	if (snprintf(provname, sizeof (provname), "%s%u", name, (uint_t)pid) >=
13740Sstevel@tonic-gate 	    sizeof (provname) ||
13750Sstevel@tonic-gate 	    dtrace_register(provname, pattr,
13760Sstevel@tonic-gate 	    DTRACE_PRIV_PROC | DTRACE_PRIV_OWNER, uid,
13770Sstevel@tonic-gate 	    pattr == &pid_attr ? &pid_pops : &usdt_pops, new_fp,
13780Sstevel@tonic-gate 	    &new_fp->ftp_provid) != 0) {
13790Sstevel@tonic-gate 		mutex_exit(&bucket->ftb_mtx);
13800Sstevel@tonic-gate 		fasttrap_provider_free(new_fp);
13810Sstevel@tonic-gate 		return (NULL);
13820Sstevel@tonic-gate 	}
13830Sstevel@tonic-gate 
13840Sstevel@tonic-gate 	new_fp->ftp_next = bucket->ftb_data;
13850Sstevel@tonic-gate 	bucket->ftb_data = new_fp;
13860Sstevel@tonic-gate 
13870Sstevel@tonic-gate 	mutex_enter(&new_fp->ftp_mtx);
13880Sstevel@tonic-gate 	mutex_exit(&bucket->ftb_mtx);
13890Sstevel@tonic-gate 
13900Sstevel@tonic-gate 	return (new_fp);
13910Sstevel@tonic-gate }
13920Sstevel@tonic-gate 
13930Sstevel@tonic-gate static void
13940Sstevel@tonic-gate fasttrap_provider_free(fasttrap_provider_t *provider)
13950Sstevel@tonic-gate {
13960Sstevel@tonic-gate 	pid_t pid = provider->ftp_pid;
13970Sstevel@tonic-gate 	proc_t *p;
13980Sstevel@tonic-gate 
13990Sstevel@tonic-gate 	/*
14000Sstevel@tonic-gate 	 * There need to be no consumers using this provider and no
14010Sstevel@tonic-gate 	 * associated enabled probes.
14020Sstevel@tonic-gate 	 */
14030Sstevel@tonic-gate 	ASSERT(provider->ftp_ccount == 0);
14040Sstevel@tonic-gate 	ASSERT(provider->ftp_rcount == 0);
14050Sstevel@tonic-gate 
1406532Sahl 	fasttrap_proc_release(provider->ftp_proc);
1407532Sahl 
14080Sstevel@tonic-gate 	kmem_free(provider, sizeof (fasttrap_provider_t));
14090Sstevel@tonic-gate 
14100Sstevel@tonic-gate 	/*
14110Sstevel@tonic-gate 	 * Decrement p_dtrace_probes on the process whose provider we're
14120Sstevel@tonic-gate 	 * freeing. We don't have to worry about clobbering somone else's
14130Sstevel@tonic-gate 	 * modifications to it because we have locked the bucket that
14140Sstevel@tonic-gate 	 * corresponds to this process's hash chain in the provider hash
14150Sstevel@tonic-gate 	 * table. Don't sweat it if we can't find the process.
14160Sstevel@tonic-gate 	 */
14170Sstevel@tonic-gate 	mutex_enter(&pidlock);
14180Sstevel@tonic-gate 	if ((p = prfind(pid)) == NULL) {
14190Sstevel@tonic-gate 		mutex_exit(&pidlock);
14200Sstevel@tonic-gate 		return;
14210Sstevel@tonic-gate 	}
14220Sstevel@tonic-gate 
14230Sstevel@tonic-gate 	mutex_enter(&p->p_lock);
14240Sstevel@tonic-gate 	mutex_exit(&pidlock);
14250Sstevel@tonic-gate 
14260Sstevel@tonic-gate 	p->p_dtrace_probes--;
14270Sstevel@tonic-gate 	mutex_exit(&p->p_lock);
14280Sstevel@tonic-gate }
14290Sstevel@tonic-gate 
14300Sstevel@tonic-gate static void
14310Sstevel@tonic-gate fasttrap_provider_retire(fasttrap_provider_t *provider)
14320Sstevel@tonic-gate {
14330Sstevel@tonic-gate 	dtrace_provider_id_t provid = provider->ftp_provid;
14340Sstevel@tonic-gate 
14350Sstevel@tonic-gate 	/*
1436532Sahl 	 * Mark the provider to be removed in our post-processing step,
1437532Sahl 	 * mark it retired, and mark its proc as defunct (though it may
1438532Sahl 	 * already be marked defunct by another provider that shares the
1439532Sahl 	 * same proc). Marking it indicates that we should try to remove it;
1440532Sahl 	 * setting the retired flag indicates that we're done with this
1441532Sahl 	 * provider; setting the proc to be defunct indicates that all
1442532Sahl 	 * tracepoints associated with the traced process should be ignored.
14430Sstevel@tonic-gate 	 */
1444532Sahl 	provider->ftp_proc->ftpc_defunct = 1;
1445532Sahl 	provider->ftp_retired = 1;
14460Sstevel@tonic-gate 	provider->ftp_marked = 1;
14470Sstevel@tonic-gate 	mutex_exit(&provider->ftp_mtx);
14480Sstevel@tonic-gate 
14490Sstevel@tonic-gate 	/*
14500Sstevel@tonic-gate 	 * We don't have to worry about invalidating the same provider twice
14510Sstevel@tonic-gate 	 * since fasttrap_provider_lookup() will ignore provider that have
1452532Sahl 	 * been marked as retired.
14530Sstevel@tonic-gate 	 */
14540Sstevel@tonic-gate 	dtrace_invalidate(provid);
14550Sstevel@tonic-gate 
14560Sstevel@tonic-gate 	fasttrap_pid_cleanup();
14570Sstevel@tonic-gate }
14580Sstevel@tonic-gate 
14590Sstevel@tonic-gate static int
14600Sstevel@tonic-gate fasttrap_add_probe(fasttrap_probe_spec_t *pdata)
14610Sstevel@tonic-gate {
14620Sstevel@tonic-gate 	fasttrap_provider_t *provider;
14630Sstevel@tonic-gate 	fasttrap_probe_t *pp;
14640Sstevel@tonic-gate 	fasttrap_tracepoint_t *tp;
14650Sstevel@tonic-gate 	char *name;
14660Sstevel@tonic-gate 	size_t size;
14670Sstevel@tonic-gate 	int i, aframes, whack;
14680Sstevel@tonic-gate 
14690Sstevel@tonic-gate 	switch (pdata->ftps_type) {
14700Sstevel@tonic-gate 	case DTFTP_ENTRY:
14710Sstevel@tonic-gate 		name = "entry";
14720Sstevel@tonic-gate 		aframes = FASTTRAP_ENTRY_AFRAMES;
14730Sstevel@tonic-gate 		break;
14740Sstevel@tonic-gate 	case DTFTP_RETURN:
14750Sstevel@tonic-gate 		name = "return";
14760Sstevel@tonic-gate 		aframes = FASTTRAP_RETURN_AFRAMES;
14770Sstevel@tonic-gate 		break;
14780Sstevel@tonic-gate 	case DTFTP_OFFSETS:
14790Sstevel@tonic-gate 		name = NULL;
14800Sstevel@tonic-gate 		break;
14810Sstevel@tonic-gate 	default:
14820Sstevel@tonic-gate 		return (EINVAL);
14830Sstevel@tonic-gate 	}
14840Sstevel@tonic-gate 
14850Sstevel@tonic-gate 	if ((provider = fasttrap_provider_lookup(pdata->ftps_pid,
14860Sstevel@tonic-gate 	    FASTTRAP_PID_NAME, &pid_attr)) == NULL)
14870Sstevel@tonic-gate 		return (ESRCH);
14880Sstevel@tonic-gate 
14890Sstevel@tonic-gate 	/*
14900Sstevel@tonic-gate 	 * Increment this reference count to indicate that a consumer is
14910Sstevel@tonic-gate 	 * actively adding a new probe associated with this provider.
14920Sstevel@tonic-gate 	 */
14930Sstevel@tonic-gate 	provider->ftp_ccount++;
14940Sstevel@tonic-gate 	mutex_exit(&provider->ftp_mtx);
14950Sstevel@tonic-gate 
14960Sstevel@tonic-gate 	if (name != NULL) {
14970Sstevel@tonic-gate 		if (dtrace_probe_lookup(provider->ftp_provid,
14980Sstevel@tonic-gate 		    pdata->ftps_mod, pdata->ftps_func, name) != 0)
14990Sstevel@tonic-gate 			goto done;
15000Sstevel@tonic-gate 
15010Sstevel@tonic-gate 		atomic_add_32(&fasttrap_total, pdata->ftps_noffs);
15020Sstevel@tonic-gate 
15030Sstevel@tonic-gate 		if (fasttrap_total > fasttrap_max) {
15040Sstevel@tonic-gate 			atomic_add_32(&fasttrap_total, -pdata->ftps_noffs);
15050Sstevel@tonic-gate 			goto no_mem;
15060Sstevel@tonic-gate 		}
15070Sstevel@tonic-gate 
15080Sstevel@tonic-gate 		ASSERT(pdata->ftps_noffs > 0);
15090Sstevel@tonic-gate 		size = sizeof (fasttrap_probe_t) +
15100Sstevel@tonic-gate 		    sizeof (pp->ftp_tps[0]) * (pdata->ftps_noffs - 1);
15110Sstevel@tonic-gate 
15120Sstevel@tonic-gate 		pp = kmem_zalloc(size, KM_SLEEP);
15130Sstevel@tonic-gate 
15140Sstevel@tonic-gate 		pp->ftp_prov = provider;
15150Sstevel@tonic-gate 		pp->ftp_faddr = pdata->ftps_pc;
15160Sstevel@tonic-gate 		pp->ftp_fsize = pdata->ftps_size;
15170Sstevel@tonic-gate 		pp->ftp_pid = pdata->ftps_pid;
15180Sstevel@tonic-gate 		pp->ftp_ntps = pdata->ftps_noffs;
15190Sstevel@tonic-gate 		pp->ftp_type = pdata->ftps_type;
15200Sstevel@tonic-gate 
15210Sstevel@tonic-gate 		for (i = 0; i < pdata->ftps_noffs; i++) {
15220Sstevel@tonic-gate 			tp = kmem_zalloc(sizeof (fasttrap_tracepoint_t),
15230Sstevel@tonic-gate 			    KM_SLEEP);
15240Sstevel@tonic-gate 
1525532Sahl 			tp->ftt_proc = provider->ftp_proc;
15260Sstevel@tonic-gate 			tp->ftt_pc = pdata->ftps_offs[i] + pdata->ftps_pc;
15270Sstevel@tonic-gate 			tp->ftt_pid = pdata->ftps_pid;
15280Sstevel@tonic-gate 
15290Sstevel@tonic-gate 			pp->ftp_tps[i].fit_tp = tp;
15300Sstevel@tonic-gate 			pp->ftp_tps[i].fit_id.fti_probe = pp;
15310Sstevel@tonic-gate 		}
15320Sstevel@tonic-gate 
15330Sstevel@tonic-gate 		pp->ftp_id = dtrace_probe_create(provider->ftp_provid,
15340Sstevel@tonic-gate 		    pdata->ftps_mod, pdata->ftps_func, name, aframes, pp);
15350Sstevel@tonic-gate 	} else {
15360Sstevel@tonic-gate 		for (i = 0; i < pdata->ftps_noffs; i++) {
15370Sstevel@tonic-gate 			char name_str[17];
15380Sstevel@tonic-gate 
15390Sstevel@tonic-gate 			(void) sprintf(name_str, "%llx",
15400Sstevel@tonic-gate 			    (unsigned long long)pdata->ftps_offs[i]);
15410Sstevel@tonic-gate 
15420Sstevel@tonic-gate 			if (dtrace_probe_lookup(provider->ftp_provid,
15430Sstevel@tonic-gate 			    pdata->ftps_mod, pdata->ftps_func, name_str) != 0)
15440Sstevel@tonic-gate 				continue;
15450Sstevel@tonic-gate 
15460Sstevel@tonic-gate 			atomic_add_32(&fasttrap_total, 1);
15470Sstevel@tonic-gate 
15480Sstevel@tonic-gate 			if (fasttrap_total > fasttrap_max) {
15490Sstevel@tonic-gate 				atomic_add_32(&fasttrap_total, -1);
15500Sstevel@tonic-gate 				goto no_mem;
15510Sstevel@tonic-gate 			}
15520Sstevel@tonic-gate 
15530Sstevel@tonic-gate 			pp = kmem_zalloc(sizeof (fasttrap_probe_t), KM_SLEEP);
15540Sstevel@tonic-gate 
15550Sstevel@tonic-gate 			pp->ftp_prov = provider;
15560Sstevel@tonic-gate 			pp->ftp_faddr = pdata->ftps_pc;
15570Sstevel@tonic-gate 			pp->ftp_fsize = pdata->ftps_size;
15580Sstevel@tonic-gate 			pp->ftp_pid = pdata->ftps_pid;
15590Sstevel@tonic-gate 			pp->ftp_ntps = 1;
15600Sstevel@tonic-gate 			pp->ftp_type = pdata->ftps_type;
15610Sstevel@tonic-gate 
15620Sstevel@tonic-gate 			tp = kmem_zalloc(sizeof (fasttrap_tracepoint_t),
15630Sstevel@tonic-gate 			    KM_SLEEP);
15640Sstevel@tonic-gate 
1565532Sahl 			tp->ftt_proc = provider->ftp_proc;
15660Sstevel@tonic-gate 			tp->ftt_pc = pdata->ftps_offs[i] + pdata->ftps_pc;
15670Sstevel@tonic-gate 			tp->ftt_pid = pdata->ftps_pid;
15680Sstevel@tonic-gate 
15690Sstevel@tonic-gate 			pp->ftp_tps[0].fit_tp = tp;
15700Sstevel@tonic-gate 			pp->ftp_tps[0].fit_id.fti_probe = pp;
15710Sstevel@tonic-gate 
15720Sstevel@tonic-gate 			pp->ftp_id = dtrace_probe_create(provider->ftp_provid,
15730Sstevel@tonic-gate 			    pdata->ftps_mod, pdata->ftps_func, name_str,
15740Sstevel@tonic-gate 			    FASTTRAP_OFFSET_AFRAMES, pp);
15750Sstevel@tonic-gate 		}
15760Sstevel@tonic-gate 	}
15770Sstevel@tonic-gate 
15780Sstevel@tonic-gate done:
15790Sstevel@tonic-gate 	/*
15800Sstevel@tonic-gate 	 * We know that the provider is still valid since we incremented the
15810Sstevel@tonic-gate 	 * reference count. If someone tried to free this provider while we
15820Sstevel@tonic-gate 	 * were using it (e.g. because the process called exec(2) or exit(2)),
15830Sstevel@tonic-gate 	 * take note of that and try to free it now.
15840Sstevel@tonic-gate 	 */
15850Sstevel@tonic-gate 	mutex_enter(&provider->ftp_mtx);
15860Sstevel@tonic-gate 	provider->ftp_ccount--;
1587532Sahl 	whack = provider->ftp_retired;
15880Sstevel@tonic-gate 	mutex_exit(&provider->ftp_mtx);
15890Sstevel@tonic-gate 
15900Sstevel@tonic-gate 	if (whack)
15910Sstevel@tonic-gate 		fasttrap_pid_cleanup();
15920Sstevel@tonic-gate 
15930Sstevel@tonic-gate 	return (0);
15940Sstevel@tonic-gate 
15950Sstevel@tonic-gate no_mem:
15960Sstevel@tonic-gate 	/*
15970Sstevel@tonic-gate 	 * If we've exhausted the allowable resources, we'll try to remove
15980Sstevel@tonic-gate 	 * this provider to free some up. This is to cover the case where
15990Sstevel@tonic-gate 	 * the user has accidentally created many more probes than was
16000Sstevel@tonic-gate 	 * intended (e.g. pid123:::).
16010Sstevel@tonic-gate 	 */
16020Sstevel@tonic-gate 	mutex_enter(&provider->ftp_mtx);
16030Sstevel@tonic-gate 	provider->ftp_ccount--;
16040Sstevel@tonic-gate 	provider->ftp_marked = 1;
16050Sstevel@tonic-gate 	mutex_exit(&provider->ftp_mtx);
16060Sstevel@tonic-gate 
16070Sstevel@tonic-gate 	fasttrap_pid_cleanup();
16080Sstevel@tonic-gate 
16090Sstevel@tonic-gate 	return (ENOMEM);
16100Sstevel@tonic-gate }
16110Sstevel@tonic-gate 
16120Sstevel@tonic-gate /*ARGSUSED*/
16130Sstevel@tonic-gate static void *
16140Sstevel@tonic-gate fasttrap_meta_provide(void *arg, dtrace_helper_provdesc_t *dhpv, pid_t pid)
16150Sstevel@tonic-gate {
16160Sstevel@tonic-gate 	fasttrap_provider_t *provider;
16170Sstevel@tonic-gate 
16180Sstevel@tonic-gate 	/*
16190Sstevel@tonic-gate 	 * A 32-bit unsigned integer (like a pid for example) can be
16200Sstevel@tonic-gate 	 * expressed in 10 or fewer decimal digits. Make sure that we'll
16210Sstevel@tonic-gate 	 * have enough space for the provider name.
16220Sstevel@tonic-gate 	 */
16230Sstevel@tonic-gate 	if (strlen(dhpv->dthpv_provname) + 10 >=
16240Sstevel@tonic-gate 	    sizeof (provider->ftp_name)) {
16250Sstevel@tonic-gate 		cmn_err(CE_WARN, "failed to instantiate provider %s: "
16260Sstevel@tonic-gate 		    "name too long to accomodate pid", dhpv->dthpv_provname);
16270Sstevel@tonic-gate 		return (NULL);
16280Sstevel@tonic-gate 	}
16290Sstevel@tonic-gate 
16300Sstevel@tonic-gate 	/*
16310Sstevel@tonic-gate 	 * Don't let folks spoof the true pid provider.
16320Sstevel@tonic-gate 	 */
16330Sstevel@tonic-gate 	if (strcmp(dhpv->dthpv_provname, FASTTRAP_PID_NAME) == 0) {
16340Sstevel@tonic-gate 		cmn_err(CE_WARN, "failed to instantiate provider %s: "
16350Sstevel@tonic-gate 		    "%s is an invalid name", dhpv->dthpv_provname,
16360Sstevel@tonic-gate 		    FASTTRAP_PID_NAME);
16370Sstevel@tonic-gate 		return (NULL);
16380Sstevel@tonic-gate 	}
16390Sstevel@tonic-gate 
16400Sstevel@tonic-gate 	/*
16410Sstevel@tonic-gate 	 * The highest stability class that fasttrap supports is ISA; cap
16420Sstevel@tonic-gate 	 * the stability of the new provider accordingly.
16430Sstevel@tonic-gate 	 */
16440Sstevel@tonic-gate 	if (dhpv->dthpv_pattr.dtpa_provider.dtat_class >= DTRACE_CLASS_COMMON)
16450Sstevel@tonic-gate 		dhpv->dthpv_pattr.dtpa_provider.dtat_class = DTRACE_CLASS_ISA;
16460Sstevel@tonic-gate 	if (dhpv->dthpv_pattr.dtpa_mod.dtat_class >= DTRACE_CLASS_COMMON)
16470Sstevel@tonic-gate 		dhpv->dthpv_pattr.dtpa_mod.dtat_class = DTRACE_CLASS_ISA;
16480Sstevel@tonic-gate 	if (dhpv->dthpv_pattr.dtpa_func.dtat_class >= DTRACE_CLASS_COMMON)
16490Sstevel@tonic-gate 		dhpv->dthpv_pattr.dtpa_func.dtat_class = DTRACE_CLASS_ISA;
16500Sstevel@tonic-gate 	if (dhpv->dthpv_pattr.dtpa_name.dtat_class >= DTRACE_CLASS_COMMON)
16510Sstevel@tonic-gate 		dhpv->dthpv_pattr.dtpa_name.dtat_class = DTRACE_CLASS_ISA;
16520Sstevel@tonic-gate 	if (dhpv->dthpv_pattr.dtpa_args.dtat_class >= DTRACE_CLASS_COMMON)
16530Sstevel@tonic-gate 		dhpv->dthpv_pattr.dtpa_args.dtat_class = DTRACE_CLASS_ISA;
16540Sstevel@tonic-gate 
16550Sstevel@tonic-gate 	if ((provider = fasttrap_provider_lookup(pid, dhpv->dthpv_provname,
16560Sstevel@tonic-gate 	    &dhpv->dthpv_pattr)) == NULL) {
16570Sstevel@tonic-gate 		cmn_err(CE_WARN, "failed to instantiate provider %s for "
16580Sstevel@tonic-gate 		    "process %u",  dhpv->dthpv_provname, (uint_t)pid);
16590Sstevel@tonic-gate 		return (NULL);
16600Sstevel@tonic-gate 	}
16610Sstevel@tonic-gate 
16620Sstevel@tonic-gate 	/*
16630Sstevel@tonic-gate 	 * We elevate the consumer count here to ensure that this provider
16640Sstevel@tonic-gate 	 * isn't removed until after the meta provider has been told to
16650Sstevel@tonic-gate 	 * remove it.
16660Sstevel@tonic-gate 	 */
16670Sstevel@tonic-gate 	provider->ftp_ccount++;
16680Sstevel@tonic-gate 
16690Sstevel@tonic-gate 	mutex_exit(&provider->ftp_mtx);
16700Sstevel@tonic-gate 
16710Sstevel@tonic-gate 	return (provider);
16720Sstevel@tonic-gate }
16730Sstevel@tonic-gate 
16740Sstevel@tonic-gate /*ARGSUSED*/
16750Sstevel@tonic-gate static void
16760Sstevel@tonic-gate fasttrap_meta_create_probe(void *arg, void *parg,
16770Sstevel@tonic-gate     dtrace_helper_probedesc_t *dhpb)
16780Sstevel@tonic-gate {
16790Sstevel@tonic-gate 	fasttrap_provider_t *provider = parg;
16800Sstevel@tonic-gate 	fasttrap_probe_t *pp;
16810Sstevel@tonic-gate 	fasttrap_tracepoint_t *tp;
16820Sstevel@tonic-gate 	size_t size;
16830Sstevel@tonic-gate 	int i;
16840Sstevel@tonic-gate 
16850Sstevel@tonic-gate 	mutex_enter(&provider->ftp_mtx);
16860Sstevel@tonic-gate 
16870Sstevel@tonic-gate 	if (dtrace_probe_lookup(provider->ftp_provid, dhpb->dthpb_mod,
16880Sstevel@tonic-gate 	    dhpb->dthpb_func, dhpb->dthpb_name) != 0) {
16890Sstevel@tonic-gate 		mutex_exit(&provider->ftp_mtx);
16900Sstevel@tonic-gate 		return;
16910Sstevel@tonic-gate 	}
16920Sstevel@tonic-gate 
16930Sstevel@tonic-gate 	atomic_add_32(&fasttrap_total, dhpb->dthpb_noffs);
16940Sstevel@tonic-gate 
16950Sstevel@tonic-gate 	if (fasttrap_total > fasttrap_max) {
16960Sstevel@tonic-gate 		atomic_add_32(&fasttrap_total, -dhpb->dthpb_noffs);
16970Sstevel@tonic-gate 		mutex_exit(&provider->ftp_mtx);
16980Sstevel@tonic-gate 		return;
16990Sstevel@tonic-gate 	}
17000Sstevel@tonic-gate 
17010Sstevel@tonic-gate 	size = sizeof (fasttrap_probe_t) +
17020Sstevel@tonic-gate 	    sizeof (pp->ftp_tps[0]) * (dhpb->dthpb_noffs - 1);
17030Sstevel@tonic-gate 	pp = kmem_zalloc(size, KM_SLEEP);
17040Sstevel@tonic-gate 
17050Sstevel@tonic-gate 	pp->ftp_prov = provider;
17060Sstevel@tonic-gate 	pp->ftp_pid = provider->ftp_pid;
17070Sstevel@tonic-gate 	pp->ftp_ntps = dhpb->dthpb_noffs;
17080Sstevel@tonic-gate #ifdef __sparc
17090Sstevel@tonic-gate 	pp->ftp_type = DTFTP_POST_OFFSETS;
17100Sstevel@tonic-gate #else
17110Sstevel@tonic-gate 	pp->ftp_type = DTFTP_OFFSETS;
17120Sstevel@tonic-gate #endif
17130Sstevel@tonic-gate 	pp->ftp_nargs = dhpb->dthpb_xargc;
17140Sstevel@tonic-gate 	pp->ftp_xtypes = dhpb->dthpb_xtypes;
17150Sstevel@tonic-gate 	pp->ftp_ntypes = dhpb->dthpb_ntypes;
17160Sstevel@tonic-gate 
17170Sstevel@tonic-gate 	for (i = 0; i < pp->ftp_ntps; i++) {
17180Sstevel@tonic-gate 		tp = kmem_zalloc(sizeof (fasttrap_tracepoint_t), KM_SLEEP);
17190Sstevel@tonic-gate 
1720532Sahl 		tp->ftt_proc = provider->ftp_proc;
17210Sstevel@tonic-gate 		tp->ftt_pc = dhpb->dthpb_base + dhpb->dthpb_offs[i];
17220Sstevel@tonic-gate 		tp->ftt_pid = provider->ftp_pid;
17230Sstevel@tonic-gate 
17240Sstevel@tonic-gate 		pp->ftp_tps[i].fit_tp = tp;
17250Sstevel@tonic-gate 		pp->ftp_tps[i].fit_id.fti_probe = pp;
17260Sstevel@tonic-gate 	}
17270Sstevel@tonic-gate 
17280Sstevel@tonic-gate 	/*
17290Sstevel@tonic-gate 	 * If the arguments are shuffled around we set the argument remapping
17300Sstevel@tonic-gate 	 * table. Later, when the probe fires, we only remap the arguments
17310Sstevel@tonic-gate 	 * if the table is non-NULL.
17320Sstevel@tonic-gate 	 */
17330Sstevel@tonic-gate 	for (i = 0; i < dhpb->dthpb_xargc; i++) {
17340Sstevel@tonic-gate 		if (dhpb->dthpb_args[i] != i) {
17350Sstevel@tonic-gate 			pp->ftp_argmap = dhpb->dthpb_args;
17360Sstevel@tonic-gate 			break;
17370Sstevel@tonic-gate 		}
17380Sstevel@tonic-gate 	}
17390Sstevel@tonic-gate 
17400Sstevel@tonic-gate 	/*
17410Sstevel@tonic-gate 	 * The probe is fully constructed -- register it with DTrace.
17420Sstevel@tonic-gate 	 */
17430Sstevel@tonic-gate 	pp->ftp_id = dtrace_probe_create(provider->ftp_provid, dhpb->dthpb_mod,
17440Sstevel@tonic-gate 	    dhpb->dthpb_func, dhpb->dthpb_name, FASTTRAP_OFFSET_AFRAMES, pp);
17450Sstevel@tonic-gate 
17460Sstevel@tonic-gate 	mutex_exit(&provider->ftp_mtx);
17470Sstevel@tonic-gate }
17480Sstevel@tonic-gate 
17490Sstevel@tonic-gate /*ARGSUSED*/
17500Sstevel@tonic-gate static void
17510Sstevel@tonic-gate fasttrap_meta_remove(void *arg, dtrace_helper_provdesc_t *dhpv, pid_t pid)
17520Sstevel@tonic-gate {
17530Sstevel@tonic-gate 	fasttrap_provider_t *provider;
17540Sstevel@tonic-gate 
17550Sstevel@tonic-gate 	if ((provider = fasttrap_provider_lookup(pid,
17560Sstevel@tonic-gate 	    dhpv->dthpv_provname, NULL)) != NULL) {
17570Sstevel@tonic-gate 		/*
17580Sstevel@tonic-gate 		 * Drop the consumer count now that we're done with this
17590Sstevel@tonic-gate 		 * provider. If there are no other consumers retire it now.
17600Sstevel@tonic-gate 		 */
17610Sstevel@tonic-gate 		if (--provider->ftp_ccount == 0)
17620Sstevel@tonic-gate 			fasttrap_provider_retire(provider);
17630Sstevel@tonic-gate 		else
17640Sstevel@tonic-gate 			mutex_exit(&provider->ftp_mtx);
17650Sstevel@tonic-gate 	}
17660Sstevel@tonic-gate }
17670Sstevel@tonic-gate 
17680Sstevel@tonic-gate static dtrace_mops_t fasttrap_mops = {
17690Sstevel@tonic-gate 	fasttrap_meta_create_probe,
17700Sstevel@tonic-gate 	fasttrap_meta_provide,
17710Sstevel@tonic-gate 	fasttrap_meta_remove
17720Sstevel@tonic-gate };
17730Sstevel@tonic-gate 
17740Sstevel@tonic-gate /*ARGSUSED*/
17750Sstevel@tonic-gate static int
17760Sstevel@tonic-gate fasttrap_open(dev_t *devp, int flag, int otyp, cred_t *cred_p)
17770Sstevel@tonic-gate {
17780Sstevel@tonic-gate 	return (0);
17790Sstevel@tonic-gate }
17800Sstevel@tonic-gate 
17810Sstevel@tonic-gate /*ARGSUSED*/
17820Sstevel@tonic-gate static int
17830Sstevel@tonic-gate fasttrap_ioctl(dev_t dev, int cmd, intptr_t arg, int md, cred_t *cr, int *rv)
17840Sstevel@tonic-gate {
17850Sstevel@tonic-gate 	if (!dtrace_attached())
17860Sstevel@tonic-gate 		return (EAGAIN);
17870Sstevel@tonic-gate 
17880Sstevel@tonic-gate 	if (cmd == FASTTRAPIOC_MAKEPROBE) {
17890Sstevel@tonic-gate 		fasttrap_probe_spec_t *uprobe = (void *)arg;
17900Sstevel@tonic-gate 		fasttrap_probe_spec_t *probe;
17910Sstevel@tonic-gate 		uint64_t noffs;
17920Sstevel@tonic-gate 		size_t size;
17930Sstevel@tonic-gate 		int ret;
17940Sstevel@tonic-gate 		char *c;
17950Sstevel@tonic-gate 
17960Sstevel@tonic-gate 		if (copyin(&uprobe->ftps_noffs, &noffs,
17970Sstevel@tonic-gate 		    sizeof (uprobe->ftps_noffs)))
17980Sstevel@tonic-gate 			return (EFAULT);
17990Sstevel@tonic-gate 
18000Sstevel@tonic-gate 		/*
18010Sstevel@tonic-gate 		 * Probes must have at least one tracepoint.
18020Sstevel@tonic-gate 		 */
18030Sstevel@tonic-gate 		if (noffs == 0)
18040Sstevel@tonic-gate 			return (EINVAL);
18050Sstevel@tonic-gate 
18060Sstevel@tonic-gate 		size = sizeof (fasttrap_probe_spec_t) +
18070Sstevel@tonic-gate 		    sizeof (probe->ftps_offs[0]) * (noffs - 1);
18080Sstevel@tonic-gate 
18090Sstevel@tonic-gate 		if (size > 1024 * 1024)
18100Sstevel@tonic-gate 			return (ENOMEM);
18110Sstevel@tonic-gate 
18120Sstevel@tonic-gate 		probe = kmem_alloc(size, KM_SLEEP);
18130Sstevel@tonic-gate 
18140Sstevel@tonic-gate 		if (copyin(uprobe, probe, size) != 0) {
18150Sstevel@tonic-gate 			kmem_free(probe, size);
18160Sstevel@tonic-gate 			return (EFAULT);
18170Sstevel@tonic-gate 		}
18180Sstevel@tonic-gate 
18190Sstevel@tonic-gate 		/*
18200Sstevel@tonic-gate 		 * Verify that the function and module strings contain no
18210Sstevel@tonic-gate 		 * funny characters.
18220Sstevel@tonic-gate 		 */
18230Sstevel@tonic-gate 		for (c = &probe->ftps_func[0]; *c != '\0'; c++) {
18240Sstevel@tonic-gate 			if (*c < 0x20 || 0x7f <= *c) {
18250Sstevel@tonic-gate 				ret = EINVAL;
18260Sstevel@tonic-gate 				goto err;
18270Sstevel@tonic-gate 			}
18280Sstevel@tonic-gate 		}
18290Sstevel@tonic-gate 
18300Sstevel@tonic-gate 		for (c = &probe->ftps_mod[0]; *c != '\0'; c++) {
18310Sstevel@tonic-gate 			if (*c < 0x20 || 0x7f <= *c) {
18320Sstevel@tonic-gate 				ret = EINVAL;
18330Sstevel@tonic-gate 				goto err;
18340Sstevel@tonic-gate 			}
18350Sstevel@tonic-gate 		}
18360Sstevel@tonic-gate 
18370Sstevel@tonic-gate 		if (!PRIV_POLICY_CHOICE(cr, PRIV_ALL, B_FALSE)) {
18380Sstevel@tonic-gate 			proc_t *p;
18390Sstevel@tonic-gate 			pid_t pid = probe->ftps_pid;
18400Sstevel@tonic-gate 
18410Sstevel@tonic-gate 			mutex_enter(&pidlock);
18420Sstevel@tonic-gate 			/*
18430Sstevel@tonic-gate 			 * Report an error if the process doesn't exist
18440Sstevel@tonic-gate 			 * or is actively being birthed.
18450Sstevel@tonic-gate 			 */
18460Sstevel@tonic-gate 			if ((p = prfind(pid)) == NULL || p->p_stat == SIDL) {
18470Sstevel@tonic-gate 				mutex_exit(&pidlock);
18480Sstevel@tonic-gate 				return (ESRCH);
18490Sstevel@tonic-gate 			}
18500Sstevel@tonic-gate 			mutex_enter(&p->p_lock);
18510Sstevel@tonic-gate 			mutex_exit(&pidlock);
18520Sstevel@tonic-gate 
18530Sstevel@tonic-gate 			if ((ret = priv_proc_cred_perm(cr, p, NULL,
18540Sstevel@tonic-gate 			    VREAD | VWRITE)) != 0) {
18550Sstevel@tonic-gate 				mutex_exit(&p->p_lock);
18560Sstevel@tonic-gate 				return (ret);
18570Sstevel@tonic-gate 			}
18580Sstevel@tonic-gate 
18590Sstevel@tonic-gate 			mutex_exit(&p->p_lock);
18600Sstevel@tonic-gate 		}
18610Sstevel@tonic-gate 
18620Sstevel@tonic-gate 		ret = fasttrap_add_probe(probe);
18630Sstevel@tonic-gate err:
18640Sstevel@tonic-gate 		kmem_free(probe, size);
18650Sstevel@tonic-gate 
18660Sstevel@tonic-gate 		return (ret);
18670Sstevel@tonic-gate 
18680Sstevel@tonic-gate 	} else if (cmd == FASTTRAPIOC_GETINSTR) {
18690Sstevel@tonic-gate 		fasttrap_instr_query_t instr;
18700Sstevel@tonic-gate 		fasttrap_tracepoint_t *tp;
18710Sstevel@tonic-gate 		uint_t index;
18720Sstevel@tonic-gate 		int ret;
18730Sstevel@tonic-gate 
18740Sstevel@tonic-gate 		if (copyin((void *)arg, &instr, sizeof (instr)) != 0)
18750Sstevel@tonic-gate 			return (EFAULT);
18760Sstevel@tonic-gate 
18770Sstevel@tonic-gate 		if (!PRIV_POLICY_CHOICE(cr, PRIV_ALL, B_FALSE)) {
18780Sstevel@tonic-gate 			proc_t *p;
18790Sstevel@tonic-gate 			pid_t pid = instr.ftiq_pid;
18800Sstevel@tonic-gate 
18810Sstevel@tonic-gate 			mutex_enter(&pidlock);
18820Sstevel@tonic-gate 			/*
18830Sstevel@tonic-gate 			 * Report an error if the process doesn't exist
18840Sstevel@tonic-gate 			 * or is actively being birthed.
18850Sstevel@tonic-gate 			 */
18860Sstevel@tonic-gate 			if ((p = prfind(pid)) == NULL || p->p_stat == SIDL) {
18870Sstevel@tonic-gate 				mutex_exit(&pidlock);
18880Sstevel@tonic-gate 				return (ESRCH);
18890Sstevel@tonic-gate 			}
18900Sstevel@tonic-gate 			mutex_enter(&p->p_lock);
18910Sstevel@tonic-gate 			mutex_exit(&pidlock);
18920Sstevel@tonic-gate 
18930Sstevel@tonic-gate 			if ((ret = priv_proc_cred_perm(cr, p, NULL,
18940Sstevel@tonic-gate 			    VREAD)) != 0) {
18950Sstevel@tonic-gate 				mutex_exit(&p->p_lock);
18960Sstevel@tonic-gate 				return (ret);
18970Sstevel@tonic-gate 			}
18980Sstevel@tonic-gate 
18990Sstevel@tonic-gate 			mutex_exit(&p->p_lock);
19000Sstevel@tonic-gate 		}
19010Sstevel@tonic-gate 
19020Sstevel@tonic-gate 		index = FASTTRAP_TPOINTS_INDEX(instr.ftiq_pid, instr.ftiq_pc);
19030Sstevel@tonic-gate 
19040Sstevel@tonic-gate 		mutex_enter(&fasttrap_tpoints.fth_table[index].ftb_mtx);
19050Sstevel@tonic-gate 		tp = fasttrap_tpoints.fth_table[index].ftb_data;
19060Sstevel@tonic-gate 		while (tp != NULL) {
19070Sstevel@tonic-gate 			if (instr.ftiq_pid == tp->ftt_pid &&
19080Sstevel@tonic-gate 			    instr.ftiq_pc == tp->ftt_pc &&
1909532Sahl 			    !tp->ftt_proc->ftpc_defunct)
19100Sstevel@tonic-gate 				break;
19110Sstevel@tonic-gate 
19120Sstevel@tonic-gate 			tp = tp->ftt_next;
19130Sstevel@tonic-gate 		}
19140Sstevel@tonic-gate 
19150Sstevel@tonic-gate 		if (tp == NULL) {
19160Sstevel@tonic-gate 			mutex_exit(&fasttrap_tpoints.fth_table[index].ftb_mtx);
19170Sstevel@tonic-gate 			return (ENOENT);
19180Sstevel@tonic-gate 		}
19190Sstevel@tonic-gate 
19200Sstevel@tonic-gate 		bcopy(&tp->ftt_instr, &instr.ftiq_instr,
19210Sstevel@tonic-gate 		    sizeof (instr.ftiq_instr));
19220Sstevel@tonic-gate 		mutex_exit(&fasttrap_tpoints.fth_table[index].ftb_mtx);
19230Sstevel@tonic-gate 
19240Sstevel@tonic-gate 		if (copyout(&instr, (void *)arg, sizeof (instr)) != 0)
19250Sstevel@tonic-gate 			return (EFAULT);
19260Sstevel@tonic-gate 
19270Sstevel@tonic-gate 		return (0);
19280Sstevel@tonic-gate 	}
19290Sstevel@tonic-gate 
19300Sstevel@tonic-gate 	return (EINVAL);
19310Sstevel@tonic-gate }
19320Sstevel@tonic-gate 
19330Sstevel@tonic-gate static struct cb_ops fasttrap_cb_ops = {
19340Sstevel@tonic-gate 	fasttrap_open,		/* open */
19350Sstevel@tonic-gate 	nodev,			/* close */
19360Sstevel@tonic-gate 	nulldev,		/* strategy */
19370Sstevel@tonic-gate 	nulldev,		/* print */
19380Sstevel@tonic-gate 	nodev,			/* dump */
19390Sstevel@tonic-gate 	nodev,			/* read */
19400Sstevel@tonic-gate 	nodev,			/* write */
19410Sstevel@tonic-gate 	fasttrap_ioctl,		/* ioctl */
19420Sstevel@tonic-gate 	nodev,			/* devmap */
19430Sstevel@tonic-gate 	nodev,			/* mmap */
19440Sstevel@tonic-gate 	nodev,			/* segmap */
19450Sstevel@tonic-gate 	nochpoll,		/* poll */
19460Sstevel@tonic-gate 	ddi_prop_op,		/* cb_prop_op */
19470Sstevel@tonic-gate 	0,			/* streamtab  */
19480Sstevel@tonic-gate 	D_NEW | D_MP		/* Driver compatibility flag */
19490Sstevel@tonic-gate };
19500Sstevel@tonic-gate 
19510Sstevel@tonic-gate /*ARGSUSED*/
19520Sstevel@tonic-gate static int
19530Sstevel@tonic-gate fasttrap_info(dev_info_t *dip, ddi_info_cmd_t infocmd, void *arg, void **result)
19540Sstevel@tonic-gate {
19550Sstevel@tonic-gate 	int error;
19560Sstevel@tonic-gate 
19570Sstevel@tonic-gate 	switch (infocmd) {
19580Sstevel@tonic-gate 	case DDI_INFO_DEVT2DEVINFO:
19590Sstevel@tonic-gate 		*result = (void *)fasttrap_devi;
19600Sstevel@tonic-gate 		error = DDI_SUCCESS;
19610Sstevel@tonic-gate 		break;
19620Sstevel@tonic-gate 	case DDI_INFO_DEVT2INSTANCE:
19630Sstevel@tonic-gate 		*result = (void *)0;
19640Sstevel@tonic-gate 		error = DDI_SUCCESS;
19650Sstevel@tonic-gate 		break;
19660Sstevel@tonic-gate 	default:
19670Sstevel@tonic-gate 		error = DDI_FAILURE;
19680Sstevel@tonic-gate 	}
19690Sstevel@tonic-gate 	return (error);
19700Sstevel@tonic-gate }
19710Sstevel@tonic-gate 
19720Sstevel@tonic-gate static int
19730Sstevel@tonic-gate fasttrap_attach(dev_info_t *devi, ddi_attach_cmd_t cmd)
19740Sstevel@tonic-gate {
19750Sstevel@tonic-gate 	ulong_t nent;
19760Sstevel@tonic-gate 
19770Sstevel@tonic-gate 	switch (cmd) {
19780Sstevel@tonic-gate 	case DDI_ATTACH:
19790Sstevel@tonic-gate 		break;
19800Sstevel@tonic-gate 	case DDI_RESUME:
19810Sstevel@tonic-gate 		return (DDI_SUCCESS);
19820Sstevel@tonic-gate 	default:
19830Sstevel@tonic-gate 		return (DDI_FAILURE);
19840Sstevel@tonic-gate 	}
19850Sstevel@tonic-gate 
19860Sstevel@tonic-gate 	if (ddi_create_minor_node(devi, "fasttrap", S_IFCHR, 0,
19870Sstevel@tonic-gate 	    DDI_PSEUDO, NULL) == DDI_FAILURE ||
19880Sstevel@tonic-gate 	    dtrace_register("fasttrap", &fasttrap_attr, DTRACE_PRIV_USER, 0,
19890Sstevel@tonic-gate 	    &fasttrap_pops, NULL, &fasttrap_id) != 0) {
19900Sstevel@tonic-gate 		ddi_remove_minor_node(devi, NULL);
19910Sstevel@tonic-gate 		return (DDI_FAILURE);
19920Sstevel@tonic-gate 	}
19930Sstevel@tonic-gate 
19940Sstevel@tonic-gate 	ddi_report_dev(devi);
19950Sstevel@tonic-gate 	fasttrap_devi = devi;
19960Sstevel@tonic-gate 
19970Sstevel@tonic-gate 	/*
19980Sstevel@tonic-gate 	 * Install our hooks into fork(2), exec(2), and exit(2).
19990Sstevel@tonic-gate 	 */
20000Sstevel@tonic-gate 	dtrace_fasttrap_fork_ptr = &fasttrap_fork;
20010Sstevel@tonic-gate 	dtrace_fasttrap_exit_ptr = &fasttrap_exec_exit;
20020Sstevel@tonic-gate 	dtrace_fasttrap_exec_ptr = &fasttrap_exec_exit;
20030Sstevel@tonic-gate 
20040Sstevel@tonic-gate 	fasttrap_max = ddi_getprop(DDI_DEV_T_ANY, devi, DDI_PROP_DONTPASS,
20050Sstevel@tonic-gate 	    "fasttrap-max-probes", FASTTRAP_MAX_DEFAULT);
20060Sstevel@tonic-gate 	fasttrap_total = 0;
20070Sstevel@tonic-gate 
20080Sstevel@tonic-gate 	/*
20090Sstevel@tonic-gate 	 * Conjure up the tracepoints hashtable...
20100Sstevel@tonic-gate 	 */
20110Sstevel@tonic-gate 	nent = ddi_getprop(DDI_DEV_T_ANY, devi, DDI_PROP_DONTPASS,
20120Sstevel@tonic-gate 	    "fasttrap-hash-size", FASTTRAP_TPOINTS_DEFAULT_SIZE);
20130Sstevel@tonic-gate 
20140Sstevel@tonic-gate 	if (nent <= 0 || nent > 0x1000000)
20150Sstevel@tonic-gate 		nent = FASTTRAP_TPOINTS_DEFAULT_SIZE;
20160Sstevel@tonic-gate 
20170Sstevel@tonic-gate 	if ((nent & (nent - 1)) == 0)
20180Sstevel@tonic-gate 		fasttrap_tpoints.fth_nent = nent;
20190Sstevel@tonic-gate 	else
20200Sstevel@tonic-gate 		fasttrap_tpoints.fth_nent = 1 << fasttrap_highbit(nent);
20210Sstevel@tonic-gate 	ASSERT(fasttrap_tpoints.fth_nent > 0);
20220Sstevel@tonic-gate 	fasttrap_tpoints.fth_mask = fasttrap_tpoints.fth_nent - 1;
20230Sstevel@tonic-gate 	fasttrap_tpoints.fth_table = kmem_zalloc(fasttrap_tpoints.fth_nent *
20240Sstevel@tonic-gate 	    sizeof (fasttrap_bucket_t), KM_SLEEP);
20250Sstevel@tonic-gate 
20260Sstevel@tonic-gate 	/*
2027532Sahl 	 * ... and the providers hash table...
20280Sstevel@tonic-gate 	 */
20290Sstevel@tonic-gate 	nent = FASTTRAP_PROVIDERS_DEFAULT_SIZE;
20300Sstevel@tonic-gate 	if ((nent & (nent - 1)) == 0)
20310Sstevel@tonic-gate 		fasttrap_provs.fth_nent = nent;
20320Sstevel@tonic-gate 	else
20330Sstevel@tonic-gate 		fasttrap_provs.fth_nent = 1 << fasttrap_highbit(nent);
20340Sstevel@tonic-gate 	ASSERT(fasttrap_provs.fth_nent > 0);
20350Sstevel@tonic-gate 	fasttrap_provs.fth_mask = fasttrap_provs.fth_nent - 1;
2036532Sahl 	fasttrap_provs.fth_table = kmem_zalloc(fasttrap_provs.fth_nent *
2037532Sahl 	    sizeof (fasttrap_bucket_t), KM_SLEEP);
20380Sstevel@tonic-gate 
2039532Sahl 	/*
2040532Sahl 	 * ... and the procs hash table.
2041532Sahl 	 */
2042532Sahl 	nent = FASTTRAP_PROCS_DEFAULT_SIZE;
2043532Sahl 	if ((nent & (nent - 1)) == 0)
2044532Sahl 		fasttrap_procs.fth_nent = nent;
2045532Sahl 	else
2046532Sahl 		fasttrap_procs.fth_nent = 1 << fasttrap_highbit(nent);
2047532Sahl 	ASSERT(fasttrap_procs.fth_nent > 0);
2048532Sahl 	fasttrap_procs.fth_mask = fasttrap_procs.fth_nent - 1;
2049532Sahl 	fasttrap_procs.fth_table = kmem_zalloc(fasttrap_procs.fth_nent *
20500Sstevel@tonic-gate 	    sizeof (fasttrap_bucket_t), KM_SLEEP);
20510Sstevel@tonic-gate 
20520Sstevel@tonic-gate 	(void) dtrace_meta_register("fasttrap", &fasttrap_mops, NULL,
20530Sstevel@tonic-gate 	    &fasttrap_meta_id);
20540Sstevel@tonic-gate 
20550Sstevel@tonic-gate 	return (DDI_SUCCESS);
20560Sstevel@tonic-gate }
20570Sstevel@tonic-gate 
20580Sstevel@tonic-gate static int
20590Sstevel@tonic-gate fasttrap_detach(dev_info_t *devi, ddi_detach_cmd_t cmd)
20600Sstevel@tonic-gate {
20610Sstevel@tonic-gate 	int i, fail = 0;
20620Sstevel@tonic-gate 	timeout_id_t tmp;
20630Sstevel@tonic-gate 
20640Sstevel@tonic-gate 	switch (cmd) {
20650Sstevel@tonic-gate 	case DDI_DETACH:
20660Sstevel@tonic-gate 		break;
20670Sstevel@tonic-gate 	case DDI_SUSPEND:
20680Sstevel@tonic-gate 		return (DDI_SUCCESS);
20690Sstevel@tonic-gate 	default:
20700Sstevel@tonic-gate 		return (DDI_FAILURE);
20710Sstevel@tonic-gate 	}
20720Sstevel@tonic-gate 
20730Sstevel@tonic-gate 	/*
20740Sstevel@tonic-gate 	 * Unregister the meta-provider to make sure no new fasttrap-
20750Sstevel@tonic-gate 	 * managed providers come along while we're trying to close up
20760Sstevel@tonic-gate 	 * shop. If we fail to detach, we'll need to re-register as a
20770Sstevel@tonic-gate 	 * meta-provider. We can fail to unregister as a meta-provider
20780Sstevel@tonic-gate 	 * if providers we manage still exist.
20790Sstevel@tonic-gate 	 */
20800Sstevel@tonic-gate 	if (fasttrap_meta_id != DTRACE_METAPROVNONE &&
20810Sstevel@tonic-gate 	    dtrace_meta_unregister(fasttrap_meta_id) != 0)
20820Sstevel@tonic-gate 		return (DDI_FAILURE);
20830Sstevel@tonic-gate 
20840Sstevel@tonic-gate 	/*
20850Sstevel@tonic-gate 	 * Prevent any new timeouts from running by setting fasttrap_timeout
20860Sstevel@tonic-gate 	 * to a non-zero value, and wait for the current timeout to complete.
20870Sstevel@tonic-gate 	 */
20880Sstevel@tonic-gate 	mutex_enter(&fasttrap_cleanup_mtx);
20890Sstevel@tonic-gate 	fasttrap_cleanup_work = 0;
20900Sstevel@tonic-gate 
20910Sstevel@tonic-gate 	while (fasttrap_timeout != (timeout_id_t)1) {
20920Sstevel@tonic-gate 		tmp = fasttrap_timeout;
20930Sstevel@tonic-gate 		fasttrap_timeout = (timeout_id_t)1;
20940Sstevel@tonic-gate 
20950Sstevel@tonic-gate 		if (tmp != 0) {
20960Sstevel@tonic-gate 			mutex_exit(&fasttrap_cleanup_mtx);
20970Sstevel@tonic-gate 			(void) untimeout(tmp);
20980Sstevel@tonic-gate 			mutex_enter(&fasttrap_cleanup_mtx);
20990Sstevel@tonic-gate 		}
21000Sstevel@tonic-gate 	}
21010Sstevel@tonic-gate 
21020Sstevel@tonic-gate 	fasttrap_cleanup_work = 0;
21030Sstevel@tonic-gate 	mutex_exit(&fasttrap_cleanup_mtx);
21040Sstevel@tonic-gate 
21050Sstevel@tonic-gate 	/*
21060Sstevel@tonic-gate 	 * Iterate over all of our providers. If there's still a process
21070Sstevel@tonic-gate 	 * that corresponds to that pid, fail to detach.
21080Sstevel@tonic-gate 	 */
21090Sstevel@tonic-gate 	for (i = 0; i < fasttrap_provs.fth_nent; i++) {
21100Sstevel@tonic-gate 		fasttrap_provider_t **fpp, *fp;
21110Sstevel@tonic-gate 		fasttrap_bucket_t *bucket = &fasttrap_provs.fth_table[i];
21120Sstevel@tonic-gate 
21130Sstevel@tonic-gate 		mutex_enter(&bucket->ftb_mtx);
21140Sstevel@tonic-gate 		fpp = (fasttrap_provider_t **)&bucket->ftb_data;
21150Sstevel@tonic-gate 		while ((fp = *fpp) != NULL) {
21160Sstevel@tonic-gate 			/*
21170Sstevel@tonic-gate 			 * Acquire and release the lock as a simple way of
21180Sstevel@tonic-gate 			 * waiting for any other consumer to finish with
21190Sstevel@tonic-gate 			 * this provider. A thread must first acquire the
21200Sstevel@tonic-gate 			 * bucket lock so there's no chance of another thread
21210Sstevel@tonic-gate 			 * blocking on the providers lock.
21220Sstevel@tonic-gate 			 */
21230Sstevel@tonic-gate 			mutex_enter(&fp->ftp_mtx);
21240Sstevel@tonic-gate 			mutex_exit(&fp->ftp_mtx);
21250Sstevel@tonic-gate 
21260Sstevel@tonic-gate 			if (dtrace_unregister(fp->ftp_provid) != 0) {
21270Sstevel@tonic-gate 				fail = 1;
21280Sstevel@tonic-gate 				fpp = &fp->ftp_next;
21290Sstevel@tonic-gate 			} else {
21300Sstevel@tonic-gate 				*fpp = fp->ftp_next;
21310Sstevel@tonic-gate 				fasttrap_provider_free(fp);
21320Sstevel@tonic-gate 			}
21330Sstevel@tonic-gate 		}
21340Sstevel@tonic-gate 
21350Sstevel@tonic-gate 		mutex_exit(&bucket->ftb_mtx);
21360Sstevel@tonic-gate 	}
21370Sstevel@tonic-gate 
21380Sstevel@tonic-gate 	if (fail || dtrace_unregister(fasttrap_id) != 0) {
21390Sstevel@tonic-gate 		uint_t work;
21400Sstevel@tonic-gate 		/*
21410Sstevel@tonic-gate 		 * If we're failing to detach, we need to unblock timeouts
21420Sstevel@tonic-gate 		 * and start a new timeout if any work has accumulated while
21430Sstevel@tonic-gate 		 * we've been unsuccessfully trying to detach.
21440Sstevel@tonic-gate 		 */
21450Sstevel@tonic-gate 		mutex_enter(&fasttrap_cleanup_mtx);
21460Sstevel@tonic-gate 		fasttrap_timeout = 0;
21470Sstevel@tonic-gate 		work = fasttrap_cleanup_work;
21480Sstevel@tonic-gate 		mutex_exit(&fasttrap_cleanup_mtx);
21490Sstevel@tonic-gate 
21500Sstevel@tonic-gate 		if (work)
21510Sstevel@tonic-gate 			fasttrap_pid_cleanup();
21520Sstevel@tonic-gate 
21530Sstevel@tonic-gate 		(void) dtrace_meta_register("fasttrap", &fasttrap_mops, NULL,
21540Sstevel@tonic-gate 		    &fasttrap_meta_id);
21550Sstevel@tonic-gate 
21560Sstevel@tonic-gate 		return (DDI_FAILURE);
21570Sstevel@tonic-gate 	}
21580Sstevel@tonic-gate 
21590Sstevel@tonic-gate #ifdef DEBUG
21600Sstevel@tonic-gate 	mutex_enter(&fasttrap_count_mtx);
21610Sstevel@tonic-gate 	ASSERT(fasttrap_count == 0);
21620Sstevel@tonic-gate 	mutex_exit(&fasttrap_count_mtx);
21630Sstevel@tonic-gate #endif
21640Sstevel@tonic-gate 
21650Sstevel@tonic-gate 	kmem_free(fasttrap_tpoints.fth_table,
21660Sstevel@tonic-gate 	    fasttrap_tpoints.fth_nent * sizeof (fasttrap_bucket_t));
21670Sstevel@tonic-gate 	fasttrap_tpoints.fth_nent = 0;
21680Sstevel@tonic-gate 
21690Sstevel@tonic-gate 	kmem_free(fasttrap_provs.fth_table,
21700Sstevel@tonic-gate 	    fasttrap_provs.fth_nent * sizeof (fasttrap_bucket_t));
21710Sstevel@tonic-gate 	fasttrap_provs.fth_nent = 0;
21720Sstevel@tonic-gate 
2173532Sahl 	kmem_free(fasttrap_procs.fth_table,
2174532Sahl 	    fasttrap_procs.fth_nent * sizeof (fasttrap_bucket_t));
2175532Sahl 	fasttrap_procs.fth_nent = 0;
2176532Sahl 
21770Sstevel@tonic-gate 	/*
21780Sstevel@tonic-gate 	 * We know there are no tracepoints in any process anywhere in
21790Sstevel@tonic-gate 	 * the system so there is no process which has its p_dtrace_count
21800Sstevel@tonic-gate 	 * greater than zero, therefore we know that no thread can actively
21810Sstevel@tonic-gate 	 * be executing code in fasttrap_fork(). Similarly for p_dtrace_probes
21820Sstevel@tonic-gate 	 * and fasttrap_exec() and fasttrap_exit().
21830Sstevel@tonic-gate 	 */
21840Sstevel@tonic-gate 	ASSERT(dtrace_fasttrap_fork_ptr == &fasttrap_fork);
21850Sstevel@tonic-gate 	dtrace_fasttrap_fork_ptr = NULL;
21860Sstevel@tonic-gate 
21870Sstevel@tonic-gate 	ASSERT(dtrace_fasttrap_exec_ptr == &fasttrap_exec_exit);
21880Sstevel@tonic-gate 	dtrace_fasttrap_exec_ptr = NULL;
21890Sstevel@tonic-gate 
21900Sstevel@tonic-gate 	ASSERT(dtrace_fasttrap_exit_ptr == &fasttrap_exec_exit);
21910Sstevel@tonic-gate 	dtrace_fasttrap_exit_ptr = NULL;
21920Sstevel@tonic-gate 
21930Sstevel@tonic-gate 	ddi_remove_minor_node(devi, NULL);
21940Sstevel@tonic-gate 
21950Sstevel@tonic-gate 	return (DDI_SUCCESS);
21960Sstevel@tonic-gate }
21970Sstevel@tonic-gate 
21980Sstevel@tonic-gate static struct dev_ops fasttrap_ops = {
21990Sstevel@tonic-gate 	DEVO_REV,		/* devo_rev */
22000Sstevel@tonic-gate 	0,			/* refcnt */
22010Sstevel@tonic-gate 	fasttrap_info,		/* get_dev_info */
22020Sstevel@tonic-gate 	nulldev,		/* identify */
22030Sstevel@tonic-gate 	nulldev,		/* probe */
22040Sstevel@tonic-gate 	fasttrap_attach,	/* attach */
22050Sstevel@tonic-gate 	fasttrap_detach,	/* detach */
22060Sstevel@tonic-gate 	nodev,			/* reset */
22070Sstevel@tonic-gate 	&fasttrap_cb_ops,	/* driver operations */
22080Sstevel@tonic-gate 	NULL,			/* bus operations */
22090Sstevel@tonic-gate 	nodev			/* dev power */
22100Sstevel@tonic-gate };
22110Sstevel@tonic-gate 
22120Sstevel@tonic-gate /*
22130Sstevel@tonic-gate  * Module linkage information for the kernel.
22140Sstevel@tonic-gate  */
22150Sstevel@tonic-gate static struct modldrv modldrv = {
22160Sstevel@tonic-gate 	&mod_driverops,		/* module type (this is a pseudo driver) */
22170Sstevel@tonic-gate 	"Fasttrap Tracing",	/* name of module */
22180Sstevel@tonic-gate 	&fasttrap_ops,		/* driver ops */
22190Sstevel@tonic-gate };
22200Sstevel@tonic-gate 
22210Sstevel@tonic-gate static struct modlinkage modlinkage = {
22220Sstevel@tonic-gate 	MODREV_1,
22230Sstevel@tonic-gate 	(void *)&modldrv,
22240Sstevel@tonic-gate 	NULL
22250Sstevel@tonic-gate };
22260Sstevel@tonic-gate 
22270Sstevel@tonic-gate int
22280Sstevel@tonic-gate _init(void)
22290Sstevel@tonic-gate {
22300Sstevel@tonic-gate 	return (mod_install(&modlinkage));
22310Sstevel@tonic-gate }
22320Sstevel@tonic-gate 
22330Sstevel@tonic-gate int
22340Sstevel@tonic-gate _info(struct modinfo *modinfop)
22350Sstevel@tonic-gate {
22360Sstevel@tonic-gate 	return (mod_info(&modlinkage, modinfop));
22370Sstevel@tonic-gate }
22380Sstevel@tonic-gate 
22390Sstevel@tonic-gate int
22400Sstevel@tonic-gate _fini(void)
22410Sstevel@tonic-gate {
22420Sstevel@tonic-gate 	return (mod_remove(&modlinkage));
22430Sstevel@tonic-gate }
2244