xref: /titanic_50/usr/src/uts/common/dtrace/fasttrap.c (revision b096140d07210b5420ce3b9a738ea89dfa24cb0e)
17c478bd9Sstevel@tonic-gate /*
27c478bd9Sstevel@tonic-gate  * CDDL HEADER START
37c478bd9Sstevel@tonic-gate  *
47c478bd9Sstevel@tonic-gate  * The contents of this file are subject to the terms of the
57c478bd9Sstevel@tonic-gate  * Common Development and Distribution License, Version 1.0 only
67c478bd9Sstevel@tonic-gate  * (the "License").  You may not use this file except in compliance
77c478bd9Sstevel@tonic-gate  * with the License.
87c478bd9Sstevel@tonic-gate  *
97c478bd9Sstevel@tonic-gate  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
107c478bd9Sstevel@tonic-gate  * or http://www.opensolaris.org/os/licensing.
117c478bd9Sstevel@tonic-gate  * See the License for the specific language governing permissions
127c478bd9Sstevel@tonic-gate  * and limitations under the License.
137c478bd9Sstevel@tonic-gate  *
147c478bd9Sstevel@tonic-gate  * When distributing Covered Code, include this CDDL HEADER in each
157c478bd9Sstevel@tonic-gate  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
167c478bd9Sstevel@tonic-gate  * If applicable, add the following below this CDDL HEADER, with the
177c478bd9Sstevel@tonic-gate  * fields enclosed by brackets "[]" replaced with your own identifying
187c478bd9Sstevel@tonic-gate  * information: Portions Copyright [yyyy] [name of copyright owner]
197c478bd9Sstevel@tonic-gate  *
207c478bd9Sstevel@tonic-gate  * CDDL HEADER END
217c478bd9Sstevel@tonic-gate  */
2297eda132Sraf 
237c478bd9Sstevel@tonic-gate /*
247c478bd9Sstevel@tonic-gate  * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
257c478bd9Sstevel@tonic-gate  * Use is subject to license terms.
267c478bd9Sstevel@tonic-gate  */
277c478bd9Sstevel@tonic-gate 
287c478bd9Sstevel@tonic-gate #pragma ident	"%Z%%M%	%I%	%E% SMI"
297c478bd9Sstevel@tonic-gate 
307c478bd9Sstevel@tonic-gate #include <sys/atomic.h>
317c478bd9Sstevel@tonic-gate #include <sys/errno.h>
327c478bd9Sstevel@tonic-gate #include <sys/stat.h>
337c478bd9Sstevel@tonic-gate #include <sys/modctl.h>
347c478bd9Sstevel@tonic-gate #include <sys/conf.h>
357c478bd9Sstevel@tonic-gate #include <sys/systm.h>
367c478bd9Sstevel@tonic-gate #include <sys/ddi.h>
377c478bd9Sstevel@tonic-gate #include <sys/sunddi.h>
387c478bd9Sstevel@tonic-gate #include <sys/cpuvar.h>
397c478bd9Sstevel@tonic-gate #include <sys/kmem.h>
407c478bd9Sstevel@tonic-gate #include <sys/strsubr.h>
417c478bd9Sstevel@tonic-gate #include <sys/fasttrap.h>
427c478bd9Sstevel@tonic-gate #include <sys/fasttrap_impl.h>
437c478bd9Sstevel@tonic-gate #include <sys/fasttrap_isa.h>
447c478bd9Sstevel@tonic-gate #include <sys/dtrace.h>
457c478bd9Sstevel@tonic-gate #include <sys/dtrace_impl.h>
467c478bd9Sstevel@tonic-gate #include <sys/sysmacros.h>
477c478bd9Sstevel@tonic-gate #include <sys/frame.h>
487c478bd9Sstevel@tonic-gate #include <sys/stack.h>
497c478bd9Sstevel@tonic-gate #include <sys/proc.h>
507c478bd9Sstevel@tonic-gate #include <sys/priv.h>
517c478bd9Sstevel@tonic-gate #include <sys/policy.h>
527c478bd9Sstevel@tonic-gate #include <sys/ontrap.h>
537c478bd9Sstevel@tonic-gate #include <sys/vmsystm.h>
547c478bd9Sstevel@tonic-gate #include <sys/prsystm.h>
557c478bd9Sstevel@tonic-gate 
567c478bd9Sstevel@tonic-gate 
577c478bd9Sstevel@tonic-gate #include <vm/as.h>
587c478bd9Sstevel@tonic-gate #include <vm/seg.h>
597c478bd9Sstevel@tonic-gate #include <vm/seg_dev.h>
607c478bd9Sstevel@tonic-gate #include <vm/seg_vn.h>
617c478bd9Sstevel@tonic-gate #include <vm/seg_spt.h>
627c478bd9Sstevel@tonic-gate #include <vm/seg_kmem.h>
637c478bd9Sstevel@tonic-gate 
647c478bd9Sstevel@tonic-gate /*
657c478bd9Sstevel@tonic-gate  * User-Land Trap-Based Tracing
667c478bd9Sstevel@tonic-gate  * ----------------------------
677c478bd9Sstevel@tonic-gate  *
687c478bd9Sstevel@tonic-gate  * The fasttrap provider allows DTrace consumers to instrument any user-level
697c478bd9Sstevel@tonic-gate  * instruction to gather data; this includes probes with semantic
707c478bd9Sstevel@tonic-gate  * signifigance like entry and return as well as simple offsets into the
717c478bd9Sstevel@tonic-gate  * function. While the specific techniques used are very ISA specific, the
727c478bd9Sstevel@tonic-gate  * methodology is generalizable to any architecture.
737c478bd9Sstevel@tonic-gate  *
747c478bd9Sstevel@tonic-gate  *
757c478bd9Sstevel@tonic-gate  * The General Methodology
767c478bd9Sstevel@tonic-gate  * -----------------------
777c478bd9Sstevel@tonic-gate  *
787c478bd9Sstevel@tonic-gate  * With the primary goal of tracing every user-land instruction and the
797c478bd9Sstevel@tonic-gate  * limitation that we can't trust user space so don't want to rely on much
807c478bd9Sstevel@tonic-gate  * information there, we begin by replacing the instructions we want to trace
817c478bd9Sstevel@tonic-gate  * with trap instructions. Each instruction we overwrite is saved into a hash
827c478bd9Sstevel@tonic-gate  * table keyed by process ID and pc address. When we enter the kernel due to
837c478bd9Sstevel@tonic-gate  * this trap instruction, we need the effects of the replaced instruction to
847c478bd9Sstevel@tonic-gate  * appear to have occurred before we proceed with the user thread's
857c478bd9Sstevel@tonic-gate  * execution.
867c478bd9Sstevel@tonic-gate  *
877c478bd9Sstevel@tonic-gate  * Each user level thread is represented by a ulwp_t structure which is
887c478bd9Sstevel@tonic-gate  * always easily accessible through a register. The most basic way to produce
897c478bd9Sstevel@tonic-gate  * the effects of the instruction we replaced is to copy that instruction out
907c478bd9Sstevel@tonic-gate  * to a bit of scratch space reserved in the user thread's ulwp_t structure
917c478bd9Sstevel@tonic-gate  * (a sort of kernel-private thread local storage), set the PC to that
927c478bd9Sstevel@tonic-gate  * scratch space and single step. When we reenter the kernel after single
937c478bd9Sstevel@tonic-gate  * stepping the instruction we must then adjust the PC to point to what would
947c478bd9Sstevel@tonic-gate  * normally be the next instruction. Of course, special care must be taken
957c478bd9Sstevel@tonic-gate  * for branches and jumps, but these represent such a small fraction of any
967c478bd9Sstevel@tonic-gate  * instruction set that writing the code to emulate these in the kernel is
977c478bd9Sstevel@tonic-gate  * not too difficult.
987c478bd9Sstevel@tonic-gate  *
997c478bd9Sstevel@tonic-gate  * Return probes may require several tracepoints to trace every return site,
1007c478bd9Sstevel@tonic-gate  * and, conversely, each tracepoint may activate several probes (the entry
1017c478bd9Sstevel@tonic-gate  * and offset 0 probes, for example). To solve this muliplexing problem,
1027c478bd9Sstevel@tonic-gate  * tracepoints contain lists of probes to activate and probes contain lists
1037c478bd9Sstevel@tonic-gate  * of tracepoints to enable. If a probe is activated, it adds its ID to
1047c478bd9Sstevel@tonic-gate  * existing tracepoints or creates new ones as necessary.
1057c478bd9Sstevel@tonic-gate  *
1067c478bd9Sstevel@tonic-gate  * Most probes are activated _before_ the instruction is executed, but return
1077c478bd9Sstevel@tonic-gate  * probes are activated _after_ the effects of the last instruction of the
1087c478bd9Sstevel@tonic-gate  * function are visible. Return probes must be fired _after_ we have
1097c478bd9Sstevel@tonic-gate  * single-stepped the instruction whereas all other probes are fired
1107c478bd9Sstevel@tonic-gate  * beforehand.
1117c478bd9Sstevel@tonic-gate  */
1127c478bd9Sstevel@tonic-gate 
1137c478bd9Sstevel@tonic-gate static dev_info_t *fasttrap_devi;
1147c478bd9Sstevel@tonic-gate static dtrace_provider_id_t fasttrap_id;
1157c478bd9Sstevel@tonic-gate static dtrace_meta_provider_id_t fasttrap_meta_id;
1167c478bd9Sstevel@tonic-gate 
1177c478bd9Sstevel@tonic-gate static timeout_id_t fasttrap_timeout;
1187c478bd9Sstevel@tonic-gate static kmutex_t fasttrap_cleanup_mtx;
1197c478bd9Sstevel@tonic-gate static uint_t fasttrap_cleanup_work;
1207c478bd9Sstevel@tonic-gate 
1217c478bd9Sstevel@tonic-gate /*
1227c478bd9Sstevel@tonic-gate  * Generation count on modifications to the global tracepoint lookup table.
1237c478bd9Sstevel@tonic-gate  */
1247c478bd9Sstevel@tonic-gate static volatile uint64_t fasttrap_mod_gen;
1257c478bd9Sstevel@tonic-gate 
1267c478bd9Sstevel@tonic-gate /*
1277c478bd9Sstevel@tonic-gate  * When the fasttrap provider is loaded, fasttrap_max is set to either
1287c478bd9Sstevel@tonic-gate  * FASTTRAP_MAX_DEFAULT or the value for fasttrap-max-probes in the
1297c478bd9Sstevel@tonic-gate  * fasttrap.conf file. Each time a probe is created, fasttrap_total is
1307c478bd9Sstevel@tonic-gate  * incremented by the number of tracepoints that may be associated with that
1317c478bd9Sstevel@tonic-gate  * probe; fasttrap_total is capped at fasttrap_max.
1327c478bd9Sstevel@tonic-gate  */
1337c478bd9Sstevel@tonic-gate #define	FASTTRAP_MAX_DEFAULT		250000
1347c478bd9Sstevel@tonic-gate static uint32_t fasttrap_max;
1357c478bd9Sstevel@tonic-gate static uint32_t fasttrap_total;
1367c478bd9Sstevel@tonic-gate 
1377c478bd9Sstevel@tonic-gate 
1387c478bd9Sstevel@tonic-gate #define	FASTTRAP_TPOINTS_DEFAULT_SIZE	0x4000
1397c478bd9Sstevel@tonic-gate #define	FASTTRAP_PROVIDERS_DEFAULT_SIZE	0x100
140*b096140dSahl #define	FASTTRAP_PROCS_DEFAULT_SIZE	0x100
1417c478bd9Sstevel@tonic-gate 
1427c478bd9Sstevel@tonic-gate #define	FASTTRAP_PID_NAME		"pid"
1437c478bd9Sstevel@tonic-gate 
1447c478bd9Sstevel@tonic-gate fasttrap_hash_t			fasttrap_tpoints;
1457c478bd9Sstevel@tonic-gate static fasttrap_hash_t		fasttrap_provs;
146*b096140dSahl static fasttrap_hash_t		fasttrap_procs;
1477c478bd9Sstevel@tonic-gate 
1487c478bd9Sstevel@tonic-gate dtrace_id_t			fasttrap_probe_id;
1497c478bd9Sstevel@tonic-gate static int			fasttrap_count;		/* ref count */
1507c478bd9Sstevel@tonic-gate static int			fasttrap_pid_count;	/* pid ref count */
1517c478bd9Sstevel@tonic-gate static kmutex_t			fasttrap_count_mtx;	/* lock on ref count */
1527c478bd9Sstevel@tonic-gate 
1538f5db3e7Sahl #define	FASTTRAP_ENABLE_FAIL	1
1548f5db3e7Sahl #define	FASTTRAP_ENABLE_PARTIAL	2
1558f5db3e7Sahl 
1567c478bd9Sstevel@tonic-gate static int fasttrap_tracepoint_enable(proc_t *, fasttrap_probe_t *, uint_t);
1577c478bd9Sstevel@tonic-gate static void fasttrap_tracepoint_disable(proc_t *, fasttrap_probe_t *, uint_t);
1587c478bd9Sstevel@tonic-gate 
1597c478bd9Sstevel@tonic-gate static fasttrap_provider_t *fasttrap_provider_lookup(pid_t, const char *,
1607c478bd9Sstevel@tonic-gate     const dtrace_pattr_t *);
1617c478bd9Sstevel@tonic-gate static void fasttrap_provider_free(fasttrap_provider_t *);
1627c478bd9Sstevel@tonic-gate static void fasttrap_provider_retire(fasttrap_provider_t *);
1637c478bd9Sstevel@tonic-gate 
164*b096140dSahl static fasttrap_proc_t *fasttrap_proc_lookup(pid_t);
165*b096140dSahl static void fasttrap_proc_release(fasttrap_proc_t *);
166*b096140dSahl 
1677c478bd9Sstevel@tonic-gate #define	FASTTRAP_PROVS_INDEX(pid, name) \
1687c478bd9Sstevel@tonic-gate 	((fasttrap_hash_str(name) + (pid)) & fasttrap_provs.fth_mask)
1697c478bd9Sstevel@tonic-gate 
170*b096140dSahl #define	FASTTRAP_PROCS_INDEX(pid) ((pid) & fasttrap_procs.fth_mask)
171*b096140dSahl 
1727c478bd9Sstevel@tonic-gate static int
1737c478bd9Sstevel@tonic-gate fasttrap_highbit(ulong_t i)
1747c478bd9Sstevel@tonic-gate {
1757c478bd9Sstevel@tonic-gate 	int h = 1;
1767c478bd9Sstevel@tonic-gate 
1777c478bd9Sstevel@tonic-gate 	if (i == 0)
1787c478bd9Sstevel@tonic-gate 		return (0);
1797c478bd9Sstevel@tonic-gate #ifdef _LP64
1807c478bd9Sstevel@tonic-gate 	if (i & 0xffffffff00000000ul) {
1817c478bd9Sstevel@tonic-gate 		h += 32; i >>= 32;
1827c478bd9Sstevel@tonic-gate 	}
1837c478bd9Sstevel@tonic-gate #endif
1847c478bd9Sstevel@tonic-gate 	if (i & 0xffff0000) {
1857c478bd9Sstevel@tonic-gate 		h += 16; i >>= 16;
1867c478bd9Sstevel@tonic-gate 	}
1877c478bd9Sstevel@tonic-gate 	if (i & 0xff00) {
1887c478bd9Sstevel@tonic-gate 		h += 8; i >>= 8;
1897c478bd9Sstevel@tonic-gate 	}
1907c478bd9Sstevel@tonic-gate 	if (i & 0xf0) {
1917c478bd9Sstevel@tonic-gate 		h += 4; i >>= 4;
1927c478bd9Sstevel@tonic-gate 	}
1937c478bd9Sstevel@tonic-gate 	if (i & 0xc) {
1947c478bd9Sstevel@tonic-gate 		h += 2; i >>= 2;
1957c478bd9Sstevel@tonic-gate 	}
1967c478bd9Sstevel@tonic-gate 	if (i & 0x2) {
1977c478bd9Sstevel@tonic-gate 		h += 1;
1987c478bd9Sstevel@tonic-gate 	}
1997c478bd9Sstevel@tonic-gate 	return (h);
2007c478bd9Sstevel@tonic-gate }
2017c478bd9Sstevel@tonic-gate 
2027c478bd9Sstevel@tonic-gate static uint_t
2037c478bd9Sstevel@tonic-gate fasttrap_hash_str(const char *p)
2047c478bd9Sstevel@tonic-gate {
2057c478bd9Sstevel@tonic-gate 	unsigned int g;
2067c478bd9Sstevel@tonic-gate 	uint_t hval = 0;
2077c478bd9Sstevel@tonic-gate 
2087c478bd9Sstevel@tonic-gate 	while (*p) {
2097c478bd9Sstevel@tonic-gate 		hval = (hval << 4) + *p++;
2107c478bd9Sstevel@tonic-gate 		if ((g = (hval & 0xf0000000)) != 0)
2117c478bd9Sstevel@tonic-gate 			hval ^= g >> 24;
2127c478bd9Sstevel@tonic-gate 		hval &= ~g;
2137c478bd9Sstevel@tonic-gate 	}
2147c478bd9Sstevel@tonic-gate 	return (hval);
2157c478bd9Sstevel@tonic-gate }
2167c478bd9Sstevel@tonic-gate 
2177c478bd9Sstevel@tonic-gate void
2187c478bd9Sstevel@tonic-gate fasttrap_sigtrap(proc_t *p, kthread_t *t, uintptr_t pc)
2197c478bd9Sstevel@tonic-gate {
2207c478bd9Sstevel@tonic-gate 	sigqueue_t *sqp = kmem_zalloc(sizeof (sigqueue_t), KM_SLEEP);
2217c478bd9Sstevel@tonic-gate 
2227c478bd9Sstevel@tonic-gate 	sqp->sq_info.si_signo = SIGTRAP;
2237c478bd9Sstevel@tonic-gate 	sqp->sq_info.si_code = TRAP_DTRACE;
2247c478bd9Sstevel@tonic-gate 	sqp->sq_info.si_addr = (caddr_t)pc;
2257c478bd9Sstevel@tonic-gate 
2267c478bd9Sstevel@tonic-gate 	mutex_enter(&p->p_lock);
2277c478bd9Sstevel@tonic-gate 	sigaddqa(p, t, sqp);
2287c478bd9Sstevel@tonic-gate 	mutex_exit(&p->p_lock);
2297c478bd9Sstevel@tonic-gate 
2307c478bd9Sstevel@tonic-gate 	if (t != NULL)
2317c478bd9Sstevel@tonic-gate 		aston(t);
2327c478bd9Sstevel@tonic-gate }
2337c478bd9Sstevel@tonic-gate 
2347c478bd9Sstevel@tonic-gate /*
2357c478bd9Sstevel@tonic-gate  * This function ensures that no threads are actively using the memory
2367c478bd9Sstevel@tonic-gate  * associated with probes that were formerly live.
2377c478bd9Sstevel@tonic-gate  */
2387c478bd9Sstevel@tonic-gate static void
2397c478bd9Sstevel@tonic-gate fasttrap_mod_barrier(uint64_t gen)
2407c478bd9Sstevel@tonic-gate {
2417c478bd9Sstevel@tonic-gate 	int i;
2427c478bd9Sstevel@tonic-gate 
2437c478bd9Sstevel@tonic-gate 	if (gen < fasttrap_mod_gen)
2447c478bd9Sstevel@tonic-gate 		return;
2457c478bd9Sstevel@tonic-gate 
2467c478bd9Sstevel@tonic-gate 	fasttrap_mod_gen++;
2477c478bd9Sstevel@tonic-gate 
2487c478bd9Sstevel@tonic-gate 	for (i = 0; i < NCPU; i++) {
2497c478bd9Sstevel@tonic-gate 		mutex_enter(&cpu_core[i].cpuc_pid_lock);
2507c478bd9Sstevel@tonic-gate 		mutex_exit(&cpu_core[i].cpuc_pid_lock);
2517c478bd9Sstevel@tonic-gate 	}
2527c478bd9Sstevel@tonic-gate }
2537c478bd9Sstevel@tonic-gate 
2547c478bd9Sstevel@tonic-gate /*
2557c478bd9Sstevel@tonic-gate  * This is the timeout's callback for cleaning up the providers and their
2567c478bd9Sstevel@tonic-gate  * probes.
2577c478bd9Sstevel@tonic-gate  */
2587c478bd9Sstevel@tonic-gate /*ARGSUSED*/
2597c478bd9Sstevel@tonic-gate static void
2607c478bd9Sstevel@tonic-gate fasttrap_pid_cleanup_cb(void *data)
2617c478bd9Sstevel@tonic-gate {
2627c478bd9Sstevel@tonic-gate 	fasttrap_provider_t **fpp, *fp;
2637c478bd9Sstevel@tonic-gate 	fasttrap_bucket_t *bucket;
2647c478bd9Sstevel@tonic-gate 	dtrace_provider_id_t provid;
2657c478bd9Sstevel@tonic-gate 	int i, later;
2667c478bd9Sstevel@tonic-gate 
2677c478bd9Sstevel@tonic-gate 	static volatile int in = 0;
2687c478bd9Sstevel@tonic-gate 	ASSERT(in == 0);
2697c478bd9Sstevel@tonic-gate 	in = 1;
2707c478bd9Sstevel@tonic-gate 
2717c478bd9Sstevel@tonic-gate 	mutex_enter(&fasttrap_cleanup_mtx);
2727c478bd9Sstevel@tonic-gate 	while (fasttrap_cleanup_work) {
2737c478bd9Sstevel@tonic-gate 		fasttrap_cleanup_work = 0;
2747c478bd9Sstevel@tonic-gate 		mutex_exit(&fasttrap_cleanup_mtx);
2757c478bd9Sstevel@tonic-gate 
2767c478bd9Sstevel@tonic-gate 		later = 0;
2777c478bd9Sstevel@tonic-gate 
2787c478bd9Sstevel@tonic-gate 		/*
2797c478bd9Sstevel@tonic-gate 		 * Iterate over all the providers trying to remove the marked
280*b096140dSahl 		 * ones. If a provider is marked but not retired, we just
2817c478bd9Sstevel@tonic-gate 		 * have to take a crack at removing it -- it's no big deal if
2827c478bd9Sstevel@tonic-gate 		 * we can't.
2837c478bd9Sstevel@tonic-gate 		 */
2847c478bd9Sstevel@tonic-gate 		for (i = 0; i < fasttrap_provs.fth_nent; i++) {
2857c478bd9Sstevel@tonic-gate 			bucket = &fasttrap_provs.fth_table[i];
2867c478bd9Sstevel@tonic-gate 			mutex_enter(&bucket->ftb_mtx);
2877c478bd9Sstevel@tonic-gate 			fpp = (fasttrap_provider_t **)&bucket->ftb_data;
2887c478bd9Sstevel@tonic-gate 
2897c478bd9Sstevel@tonic-gate 			while ((fp = *fpp) != NULL) {
2907c478bd9Sstevel@tonic-gate 				if (!fp->ftp_marked) {
2917c478bd9Sstevel@tonic-gate 					fpp = &fp->ftp_next;
2927c478bd9Sstevel@tonic-gate 					continue;
2937c478bd9Sstevel@tonic-gate 				}
2947c478bd9Sstevel@tonic-gate 
2957c478bd9Sstevel@tonic-gate 				mutex_enter(&fp->ftp_mtx);
2967c478bd9Sstevel@tonic-gate 
2977c478bd9Sstevel@tonic-gate 				/*
2987c478bd9Sstevel@tonic-gate 				 * If this provider is referenced either
2997c478bd9Sstevel@tonic-gate 				 * because it is a USDT provider or is being
3007c478bd9Sstevel@tonic-gate 				 * modified, we can't unregister or even
3017c478bd9Sstevel@tonic-gate 				 * condense.
3027c478bd9Sstevel@tonic-gate 				 */
3037c478bd9Sstevel@tonic-gate 				if (fp->ftp_ccount != 0) {
3047c478bd9Sstevel@tonic-gate 					mutex_exit(&fp->ftp_mtx);
3057c478bd9Sstevel@tonic-gate 					fp->ftp_marked = 0;
3067c478bd9Sstevel@tonic-gate 					continue;
3077c478bd9Sstevel@tonic-gate 				}
3087c478bd9Sstevel@tonic-gate 
309*b096140dSahl 				if (!fp->ftp_retired || fp->ftp_rcount != 0)
3107c478bd9Sstevel@tonic-gate 					fp->ftp_marked = 0;
3117c478bd9Sstevel@tonic-gate 
3127c478bd9Sstevel@tonic-gate 				mutex_exit(&fp->ftp_mtx);
3137c478bd9Sstevel@tonic-gate 
3147c478bd9Sstevel@tonic-gate 				/*
3157c478bd9Sstevel@tonic-gate 				 * If we successfully unregister this
3167c478bd9Sstevel@tonic-gate 				 * provider we can remove it from the hash
3177c478bd9Sstevel@tonic-gate 				 * chain and free the memory. If our attempt
318*b096140dSahl 				 * to unregister fails and this is a retired
3197c478bd9Sstevel@tonic-gate 				 * provider, increment our flag to try again
3207c478bd9Sstevel@tonic-gate 				 * pretty soon. If we've consumed more than
3217c478bd9Sstevel@tonic-gate 				 * half of our total permitted number of
3227c478bd9Sstevel@tonic-gate 				 * probes call dtrace_condense() to try to
3237c478bd9Sstevel@tonic-gate 				 * clean out the unenabled probes.
3247c478bd9Sstevel@tonic-gate 				 */
3257c478bd9Sstevel@tonic-gate 				provid = fp->ftp_provid;
3267c478bd9Sstevel@tonic-gate 				if (dtrace_unregister(provid) != 0) {
3277c478bd9Sstevel@tonic-gate 					if (fasttrap_total > fasttrap_max / 2)
3287c478bd9Sstevel@tonic-gate 						(void) dtrace_condense(provid);
3297c478bd9Sstevel@tonic-gate 					later += fp->ftp_marked;
3307c478bd9Sstevel@tonic-gate 					fpp = &fp->ftp_next;
3317c478bd9Sstevel@tonic-gate 				} else {
3327c478bd9Sstevel@tonic-gate 					*fpp = fp->ftp_next;
3337c478bd9Sstevel@tonic-gate 					fasttrap_provider_free(fp);
3347c478bd9Sstevel@tonic-gate 				}
3357c478bd9Sstevel@tonic-gate 			}
3367c478bd9Sstevel@tonic-gate 			mutex_exit(&bucket->ftb_mtx);
3377c478bd9Sstevel@tonic-gate 		}
3387c478bd9Sstevel@tonic-gate 
3397c478bd9Sstevel@tonic-gate 		mutex_enter(&fasttrap_cleanup_mtx);
3407c478bd9Sstevel@tonic-gate 	}
3417c478bd9Sstevel@tonic-gate 
3427c478bd9Sstevel@tonic-gate 	ASSERT(fasttrap_timeout != 0);
3437c478bd9Sstevel@tonic-gate 
3447c478bd9Sstevel@tonic-gate 	/*
345*b096140dSahl 	 * If we were unable to remove a retired provider, try again after
3467c478bd9Sstevel@tonic-gate 	 * a second. This situation can occur in certain circumstances where
3477c478bd9Sstevel@tonic-gate 	 * providers cannot be unregistered even though they have no probes
3487c478bd9Sstevel@tonic-gate 	 * enabled because of an execution of dtrace -l or something similar.
3497c478bd9Sstevel@tonic-gate 	 * If the timeout has been disabled (set to 1 because we're trying
3507c478bd9Sstevel@tonic-gate 	 * to detach), we set fasttrap_cleanup_work to ensure that we'll
3517c478bd9Sstevel@tonic-gate 	 * get a chance to do that work if and when the timeout is reenabled
3527c478bd9Sstevel@tonic-gate 	 * (if detach fails).
3537c478bd9Sstevel@tonic-gate 	 */
3547c478bd9Sstevel@tonic-gate 	if (later > 0 && fasttrap_timeout != (timeout_id_t)1)
3557c478bd9Sstevel@tonic-gate 		fasttrap_timeout = timeout(&fasttrap_pid_cleanup_cb, NULL, hz);
3567c478bd9Sstevel@tonic-gate 	else if (later > 0)
3577c478bd9Sstevel@tonic-gate 		fasttrap_cleanup_work = 1;
3587c478bd9Sstevel@tonic-gate 	else
3597c478bd9Sstevel@tonic-gate 		fasttrap_timeout = 0;
3607c478bd9Sstevel@tonic-gate 
3617c478bd9Sstevel@tonic-gate 	mutex_exit(&fasttrap_cleanup_mtx);
3627c478bd9Sstevel@tonic-gate 	in = 0;
3637c478bd9Sstevel@tonic-gate }
3647c478bd9Sstevel@tonic-gate 
3657c478bd9Sstevel@tonic-gate /*
3667c478bd9Sstevel@tonic-gate  * Activates the asynchronous cleanup mechanism.
3677c478bd9Sstevel@tonic-gate  */
3687c478bd9Sstevel@tonic-gate static void
3697c478bd9Sstevel@tonic-gate fasttrap_pid_cleanup(void)
3707c478bd9Sstevel@tonic-gate {
3717c478bd9Sstevel@tonic-gate 	mutex_enter(&fasttrap_cleanup_mtx);
3727c478bd9Sstevel@tonic-gate 	fasttrap_cleanup_work = 1;
3737c478bd9Sstevel@tonic-gate 	if (fasttrap_timeout == 0)
3747c478bd9Sstevel@tonic-gate 		fasttrap_timeout = timeout(&fasttrap_pid_cleanup_cb, NULL, 1);
3757c478bd9Sstevel@tonic-gate 	mutex_exit(&fasttrap_cleanup_mtx);
3767c478bd9Sstevel@tonic-gate }
3777c478bd9Sstevel@tonic-gate 
3787c478bd9Sstevel@tonic-gate /*
3797c478bd9Sstevel@tonic-gate  * This is called from cfork() via dtrace_fasttrap_fork(). The child
3807c478bd9Sstevel@tonic-gate  * process's address space is a (roughly) a copy of the parent process's so
3817c478bd9Sstevel@tonic-gate  * we have to remove all the instrumentation we had previously enabled in the
3827c478bd9Sstevel@tonic-gate  * parent.
3837c478bd9Sstevel@tonic-gate  */
3847c478bd9Sstevel@tonic-gate static void
3857c478bd9Sstevel@tonic-gate fasttrap_fork(proc_t *p, proc_t *cp)
3867c478bd9Sstevel@tonic-gate {
3877c478bd9Sstevel@tonic-gate 	pid_t ppid = p->p_pid;
3887c478bd9Sstevel@tonic-gate 	int i;
3897c478bd9Sstevel@tonic-gate 
3907c478bd9Sstevel@tonic-gate 	ASSERT(curproc == p);
3917c478bd9Sstevel@tonic-gate 	ASSERT(p->p_proc_flag & P_PR_LOCK);
3927c478bd9Sstevel@tonic-gate 	ASSERT(p->p_dtrace_count > 0);
3937c478bd9Sstevel@tonic-gate 	ASSERT(cp->p_dtrace_count == 0);
3947c478bd9Sstevel@tonic-gate 
3957c478bd9Sstevel@tonic-gate 	/*
3967c478bd9Sstevel@tonic-gate 	 * This would be simpler and faster if we maintained per-process
3977c478bd9Sstevel@tonic-gate 	 * hash tables of enabled tracepoints. It could, however, potentially
3987c478bd9Sstevel@tonic-gate 	 * slow down execution of a tracepoint since we'd need to go
3997c478bd9Sstevel@tonic-gate 	 * through two levels of indirection. In the future, we should
4007c478bd9Sstevel@tonic-gate 	 * consider either maintaining per-process ancillary lists of
4017c478bd9Sstevel@tonic-gate 	 * enabled tracepoints or hanging a pointer to a per-process hash
4027c478bd9Sstevel@tonic-gate 	 * table of enabled tracepoints off the proc structure.
4037c478bd9Sstevel@tonic-gate 	 */
4047c478bd9Sstevel@tonic-gate 
4057c478bd9Sstevel@tonic-gate 	/*
4067c478bd9Sstevel@tonic-gate 	 * We don't have to worry about the child process disappearing
4077c478bd9Sstevel@tonic-gate 	 * because we're in fork().
4087c478bd9Sstevel@tonic-gate 	 */
4097c478bd9Sstevel@tonic-gate 	mutex_enter(&cp->p_lock);
4107c478bd9Sstevel@tonic-gate 	sprlock_proc(cp);
4117c478bd9Sstevel@tonic-gate 	mutex_exit(&cp->p_lock);
4127c478bd9Sstevel@tonic-gate 
4137c478bd9Sstevel@tonic-gate 	/*
4147c478bd9Sstevel@tonic-gate 	 * Iterate over every tracepoint looking for ones that belong to the
4157c478bd9Sstevel@tonic-gate 	 * parent process, and remove each from the child process.
4167c478bd9Sstevel@tonic-gate 	 */
4177c478bd9Sstevel@tonic-gate 	for (i = 0; i < fasttrap_tpoints.fth_nent; i++) {
4187c478bd9Sstevel@tonic-gate 		fasttrap_tracepoint_t *tp;
4197c478bd9Sstevel@tonic-gate 		fasttrap_bucket_t *bucket = &fasttrap_tpoints.fth_table[i];
4207c478bd9Sstevel@tonic-gate 
4217c478bd9Sstevel@tonic-gate 		mutex_enter(&bucket->ftb_mtx);
4227c478bd9Sstevel@tonic-gate 		for (tp = bucket->ftb_data; tp != NULL; tp = tp->ftt_next) {
423*b096140dSahl 			if (tp->ftt_pid == ppid &&
424*b096140dSahl 			    !tp->ftt_proc->ftpc_defunct) {
4257c478bd9Sstevel@tonic-gate 				int ret = fasttrap_tracepoint_remove(cp, tp);
4267c478bd9Sstevel@tonic-gate 				ASSERT(ret == 0);
4277c478bd9Sstevel@tonic-gate 			}
4287c478bd9Sstevel@tonic-gate 		}
4297c478bd9Sstevel@tonic-gate 		mutex_exit(&bucket->ftb_mtx);
4307c478bd9Sstevel@tonic-gate 	}
4317c478bd9Sstevel@tonic-gate 
4327c478bd9Sstevel@tonic-gate 	mutex_enter(&cp->p_lock);
4337c478bd9Sstevel@tonic-gate 	sprunlock(cp);
4347c478bd9Sstevel@tonic-gate }
4357c478bd9Sstevel@tonic-gate 
4367c478bd9Sstevel@tonic-gate /*
4377c478bd9Sstevel@tonic-gate  * This is called from proc_exit() or from exec_common() if p_dtrace_probes
4387c478bd9Sstevel@tonic-gate  * is set on the proc structure to indicate that there is a pid provider
4397c478bd9Sstevel@tonic-gate  * associated with this process.
4407c478bd9Sstevel@tonic-gate  */
4417c478bd9Sstevel@tonic-gate static void
4427c478bd9Sstevel@tonic-gate fasttrap_exec_exit(proc_t *p)
4437c478bd9Sstevel@tonic-gate {
4447c478bd9Sstevel@tonic-gate 	fasttrap_provider_t *provider;
4457c478bd9Sstevel@tonic-gate 
4467c478bd9Sstevel@tonic-gate 	ASSERT(p == curproc);
4477c478bd9Sstevel@tonic-gate 	ASSERT(MUTEX_HELD(&p->p_lock));
4487c478bd9Sstevel@tonic-gate 
4497c478bd9Sstevel@tonic-gate 	mutex_exit(&p->p_lock);
4507c478bd9Sstevel@tonic-gate 
4517c478bd9Sstevel@tonic-gate 	/*
4527c478bd9Sstevel@tonic-gate 	 * We clean up the pid provider for this process here; user-land
4537c478bd9Sstevel@tonic-gate 	 * static probes are handled by the meta-provider remove entry point.
4547c478bd9Sstevel@tonic-gate 	 */
4557c478bd9Sstevel@tonic-gate 	if ((provider = fasttrap_provider_lookup(p->p_pid,
4567c478bd9Sstevel@tonic-gate 	    FASTTRAP_PID_NAME, NULL)) != NULL)
4577c478bd9Sstevel@tonic-gate 		fasttrap_provider_retire(provider);
4587c478bd9Sstevel@tonic-gate 
4597c478bd9Sstevel@tonic-gate 	mutex_enter(&p->p_lock);
4607c478bd9Sstevel@tonic-gate }
4617c478bd9Sstevel@tonic-gate 
4627c478bd9Sstevel@tonic-gate 
4637c478bd9Sstevel@tonic-gate /*ARGSUSED*/
4647c478bd9Sstevel@tonic-gate static void
4657c478bd9Sstevel@tonic-gate fasttrap_pid_provide(void *arg, const dtrace_probedesc_t *desc)
4667c478bd9Sstevel@tonic-gate {
4677c478bd9Sstevel@tonic-gate 	/*
4687c478bd9Sstevel@tonic-gate 	 * There are no "default" pid probes.
4697c478bd9Sstevel@tonic-gate 	 */
4707c478bd9Sstevel@tonic-gate }
4717c478bd9Sstevel@tonic-gate 
4727c478bd9Sstevel@tonic-gate /*ARGSUSED*/
4737c478bd9Sstevel@tonic-gate static void
4747c478bd9Sstevel@tonic-gate fasttrap_provide(void *arg, const dtrace_probedesc_t *desc)
4757c478bd9Sstevel@tonic-gate {
4767c478bd9Sstevel@tonic-gate 	if (dtrace_probe_lookup(fasttrap_id, NULL, "fasttrap", "fasttrap") == 0)
4777c478bd9Sstevel@tonic-gate 		fasttrap_probe_id = dtrace_probe_create(fasttrap_id, NULL,
4787c478bd9Sstevel@tonic-gate 		    "fasttrap", "fasttrap", FASTTRAP_AFRAMES, NULL);
4797c478bd9Sstevel@tonic-gate }
4807c478bd9Sstevel@tonic-gate 
4817c478bd9Sstevel@tonic-gate static int
4827c478bd9Sstevel@tonic-gate fasttrap_tracepoint_enable(proc_t *p, fasttrap_probe_t *probe, uint_t index)
4837c478bd9Sstevel@tonic-gate {
4847c478bd9Sstevel@tonic-gate 	fasttrap_tracepoint_t *tp, *new_tp = NULL;
4857c478bd9Sstevel@tonic-gate 	fasttrap_bucket_t *bucket;
4867c478bd9Sstevel@tonic-gate 	fasttrap_id_t *id;
4877c478bd9Sstevel@tonic-gate 	pid_t pid;
4887c478bd9Sstevel@tonic-gate 	uintptr_t pc;
4897c478bd9Sstevel@tonic-gate 
4907c478bd9Sstevel@tonic-gate 	ASSERT(index < probe->ftp_ntps);
4917c478bd9Sstevel@tonic-gate 
4927c478bd9Sstevel@tonic-gate 	pid = probe->ftp_pid;
4937c478bd9Sstevel@tonic-gate 	pc = probe->ftp_tps[index].fit_tp->ftt_pc;
4947c478bd9Sstevel@tonic-gate 	id = &probe->ftp_tps[index].fit_id;
4957c478bd9Sstevel@tonic-gate 
4967c478bd9Sstevel@tonic-gate 	ASSERT(probe->ftp_tps[index].fit_tp->ftt_pid == pid);
4977c478bd9Sstevel@tonic-gate 
4987c478bd9Sstevel@tonic-gate 	ASSERT(!(p->p_flag & SVFORK));
4997c478bd9Sstevel@tonic-gate 
5007c478bd9Sstevel@tonic-gate 	/*
5017c478bd9Sstevel@tonic-gate 	 * Before we make any modifications, make sure we've imposed a barrier
5027c478bd9Sstevel@tonic-gate 	 * on the generation in which this probe was last modified.
5037c478bd9Sstevel@tonic-gate 	 */
5047c478bd9Sstevel@tonic-gate 	fasttrap_mod_barrier(probe->ftp_gen);
5057c478bd9Sstevel@tonic-gate 
5067c478bd9Sstevel@tonic-gate 	bucket = &fasttrap_tpoints.fth_table[FASTTRAP_TPOINTS_INDEX(pid, pc)];
5077c478bd9Sstevel@tonic-gate 
5087c478bd9Sstevel@tonic-gate 	/*
5097c478bd9Sstevel@tonic-gate 	 * If the tracepoint has already been enabled, just add our id to the
5107c478bd9Sstevel@tonic-gate 	 * list of interested probes. This may be our second time through
5117c478bd9Sstevel@tonic-gate 	 * this path in which case we'll have constructed the tracepoint we'd
5127c478bd9Sstevel@tonic-gate 	 * like to install. If we can't find a match, and have an allocated
5137c478bd9Sstevel@tonic-gate 	 * tracepoint ready to go, enable that one now.
5147c478bd9Sstevel@tonic-gate 	 *
515*b096140dSahl 	 * A tracepoint whose proc is defunct is also considered defunct.
5167c478bd9Sstevel@tonic-gate 	 */
5177c478bd9Sstevel@tonic-gate again:
5187c478bd9Sstevel@tonic-gate 	mutex_enter(&bucket->ftb_mtx);
5197c478bd9Sstevel@tonic-gate 	for (tp = bucket->ftb_data; tp != NULL; tp = tp->ftt_next) {
5207c478bd9Sstevel@tonic-gate 		if (tp->ftt_pid != pid || tp->ftt_pc != pc ||
521*b096140dSahl 		    tp->ftt_proc->ftpc_defunct)
5227c478bd9Sstevel@tonic-gate 			continue;
5237c478bd9Sstevel@tonic-gate 
5247c478bd9Sstevel@tonic-gate 		/*
5257c478bd9Sstevel@tonic-gate 		 * Now that we've found a matching tracepoint, it would be
5267c478bd9Sstevel@tonic-gate 		 * a decent idea to confirm that the tracepoint is still
5277c478bd9Sstevel@tonic-gate 		 * enabled and the trap instruction hasn't been overwritten.
5287c478bd9Sstevel@tonic-gate 		 * Since this is a little hairy, we'll punt for now.
5297c478bd9Sstevel@tonic-gate 		 */
5307c478bd9Sstevel@tonic-gate 
5317c478bd9Sstevel@tonic-gate 		/*
5327c478bd9Sstevel@tonic-gate 		 * This can't be the first interested probe. We don't have
5337c478bd9Sstevel@tonic-gate 		 * to worry about another thread being in the midst of
5347c478bd9Sstevel@tonic-gate 		 * deleting this tracepoint (which would be the only valid
5357c478bd9Sstevel@tonic-gate 		 * reason for a tracepoint to have no interested probes)
5367c478bd9Sstevel@tonic-gate 		 * since we're holding P_PR_LOCK for this process.
5377c478bd9Sstevel@tonic-gate 		 */
5387c478bd9Sstevel@tonic-gate 		ASSERT(tp->ftt_ids != NULL || tp->ftt_retids != NULL);
5397c478bd9Sstevel@tonic-gate 
5407c478bd9Sstevel@tonic-gate 		if (probe->ftp_type == DTFTP_RETURN ||
5417c478bd9Sstevel@tonic-gate 		    probe->ftp_type == DTFTP_POST_OFFSETS) {
5427c478bd9Sstevel@tonic-gate 			id->fti_next = tp->ftt_retids;
5437c478bd9Sstevel@tonic-gate 			membar_producer();
5447c478bd9Sstevel@tonic-gate 			tp->ftt_retids = id;
5457c478bd9Sstevel@tonic-gate 			membar_producer();
5467c478bd9Sstevel@tonic-gate 		} else {
5477c478bd9Sstevel@tonic-gate 			id->fti_next = tp->ftt_ids;
5487c478bd9Sstevel@tonic-gate 			membar_producer();
5497c478bd9Sstevel@tonic-gate 			tp->ftt_ids = id;
5507c478bd9Sstevel@tonic-gate 			membar_producer();
5517c478bd9Sstevel@tonic-gate 		}
5527c478bd9Sstevel@tonic-gate 
5537c478bd9Sstevel@tonic-gate 		mutex_exit(&bucket->ftb_mtx);
5547c478bd9Sstevel@tonic-gate 
5557c478bd9Sstevel@tonic-gate 		if (new_tp != NULL) {
5567c478bd9Sstevel@tonic-gate 			new_tp->ftt_ids = NULL;
5577c478bd9Sstevel@tonic-gate 			new_tp->ftt_retids = NULL;
5587c478bd9Sstevel@tonic-gate 		}
5597c478bd9Sstevel@tonic-gate 
5607c478bd9Sstevel@tonic-gate 		return (0);
5617c478bd9Sstevel@tonic-gate 	}
5627c478bd9Sstevel@tonic-gate 
5637c478bd9Sstevel@tonic-gate 	/*
5647c478bd9Sstevel@tonic-gate 	 * If we have a good tracepoint ready to go, install it now while
5657c478bd9Sstevel@tonic-gate 	 * we have the lock held and no one can screw with us.
5667c478bd9Sstevel@tonic-gate 	 */
5677c478bd9Sstevel@tonic-gate 	if (new_tp != NULL) {
5688f5db3e7Sahl 		int rc = 0;
5697c478bd9Sstevel@tonic-gate 
5707c478bd9Sstevel@tonic-gate 		new_tp->ftt_next = bucket->ftb_data;
5717c478bd9Sstevel@tonic-gate 		membar_producer();
5727c478bd9Sstevel@tonic-gate 		bucket->ftb_data = new_tp;
5737c478bd9Sstevel@tonic-gate 		membar_producer();
5747c478bd9Sstevel@tonic-gate 		mutex_exit(&bucket->ftb_mtx);
5757c478bd9Sstevel@tonic-gate 
5767c478bd9Sstevel@tonic-gate 		/*
5778f5db3e7Sahl 		 * Activate the tracepoint in the ISA-specific manner.
5788f5db3e7Sahl 		 * If this fails, we need to report the failure, but
5798f5db3e7Sahl 		 * indicate that this tracepoint must still be disabled
5808f5db3e7Sahl 		 * by calling fasttrap_tracepoint_disable().
5817c478bd9Sstevel@tonic-gate 		 */
5828f5db3e7Sahl 		if (fasttrap_tracepoint_install(p, new_tp) != 0)
5838f5db3e7Sahl 			rc = FASTTRAP_ENABLE_PARTIAL;
5848f5db3e7Sahl 
5857c478bd9Sstevel@tonic-gate 		/*
5868f5db3e7Sahl 		 * Increment the count of the number of tracepoints active in
5878f5db3e7Sahl 		 * the victim process.
5887c478bd9Sstevel@tonic-gate 		 */
5897c478bd9Sstevel@tonic-gate 		ASSERT(p->p_proc_flag & P_PR_LOCK);
5907c478bd9Sstevel@tonic-gate 		p->p_dtrace_count++;
5917c478bd9Sstevel@tonic-gate 
5927c478bd9Sstevel@tonic-gate 		return (rc);
5937c478bd9Sstevel@tonic-gate 	}
5947c478bd9Sstevel@tonic-gate 
5957c478bd9Sstevel@tonic-gate 	mutex_exit(&bucket->ftb_mtx);
5967c478bd9Sstevel@tonic-gate 
5977c478bd9Sstevel@tonic-gate 	/*
5987c478bd9Sstevel@tonic-gate 	 * Initialize the tracepoint that's been preallocated with the probe.
5997c478bd9Sstevel@tonic-gate 	 */
6007c478bd9Sstevel@tonic-gate 	new_tp = probe->ftp_tps[index].fit_tp;
6017c478bd9Sstevel@tonic-gate 
6027c478bd9Sstevel@tonic-gate 	ASSERT(new_tp->ftt_pid == pid);
6037c478bd9Sstevel@tonic-gate 	ASSERT(new_tp->ftt_pc == pc);
604*b096140dSahl 	ASSERT(new_tp->ftt_proc == probe->ftp_prov->ftp_proc);
6057c478bd9Sstevel@tonic-gate 	ASSERT(new_tp->ftt_ids == NULL);
6067c478bd9Sstevel@tonic-gate 	ASSERT(new_tp->ftt_retids == NULL);
6077c478bd9Sstevel@tonic-gate 
6087c478bd9Sstevel@tonic-gate 	if (probe->ftp_type == DTFTP_RETURN ||
6097c478bd9Sstevel@tonic-gate 	    probe->ftp_type == DTFTP_POST_OFFSETS) {
6107c478bd9Sstevel@tonic-gate 		id->fti_next = NULL;
6117c478bd9Sstevel@tonic-gate 		new_tp->ftt_retids = id;
6127c478bd9Sstevel@tonic-gate 	} else {
6137c478bd9Sstevel@tonic-gate 		id->fti_next = NULL;
6147c478bd9Sstevel@tonic-gate 		new_tp->ftt_ids = id;
6157c478bd9Sstevel@tonic-gate 	}
6167c478bd9Sstevel@tonic-gate 
6177c478bd9Sstevel@tonic-gate 	/*
6188f5db3e7Sahl 	 * If the ISA-dependent initialization goes to plan, go back to the
6197c478bd9Sstevel@tonic-gate 	 * beginning and try to install this freshly made tracepoint.
6207c478bd9Sstevel@tonic-gate 	 */
6217c478bd9Sstevel@tonic-gate 	if (fasttrap_tracepoint_init(p, probe, new_tp, pc) == 0)
6227c478bd9Sstevel@tonic-gate 		goto again;
6237c478bd9Sstevel@tonic-gate 
6247c478bd9Sstevel@tonic-gate 	new_tp->ftt_ids = NULL;
6257c478bd9Sstevel@tonic-gate 	new_tp->ftt_retids = NULL;
6267c478bd9Sstevel@tonic-gate 
6278f5db3e7Sahl 	return (FASTTRAP_ENABLE_FAIL);
6287c478bd9Sstevel@tonic-gate }
6297c478bd9Sstevel@tonic-gate 
6307c478bd9Sstevel@tonic-gate static void
6317c478bd9Sstevel@tonic-gate fasttrap_tracepoint_disable(proc_t *p, fasttrap_probe_t *probe, uint_t index)
6327c478bd9Sstevel@tonic-gate {
6337c478bd9Sstevel@tonic-gate 	fasttrap_bucket_t *bucket;
6347c478bd9Sstevel@tonic-gate 	fasttrap_provider_t *provider = probe->ftp_prov;
6357c478bd9Sstevel@tonic-gate 	fasttrap_tracepoint_t **pp, *tp;
6367c478bd9Sstevel@tonic-gate 	fasttrap_id_t *id, **idp;
6377c478bd9Sstevel@tonic-gate 	pid_t pid;
6387c478bd9Sstevel@tonic-gate 	uintptr_t pc;
6397c478bd9Sstevel@tonic-gate 
6407c478bd9Sstevel@tonic-gate 	ASSERT(index < probe->ftp_ntps);
6417c478bd9Sstevel@tonic-gate 
6427c478bd9Sstevel@tonic-gate 	pid = probe->ftp_pid;
6437c478bd9Sstevel@tonic-gate 	pc = probe->ftp_tps[index].fit_tp->ftt_pc;
6447c478bd9Sstevel@tonic-gate 
6457c478bd9Sstevel@tonic-gate 	ASSERT(probe->ftp_tps[index].fit_tp->ftt_pid == pid);
6467c478bd9Sstevel@tonic-gate 
6477c478bd9Sstevel@tonic-gate 	/*
6487c478bd9Sstevel@tonic-gate 	 * Find the tracepoint and make sure that our id is one of the
6497c478bd9Sstevel@tonic-gate 	 * ones registered with it.
6507c478bd9Sstevel@tonic-gate 	 */
6517c478bd9Sstevel@tonic-gate 	bucket = &fasttrap_tpoints.fth_table[FASTTRAP_TPOINTS_INDEX(pid, pc)];
6527c478bd9Sstevel@tonic-gate 	mutex_enter(&bucket->ftb_mtx);
6537c478bd9Sstevel@tonic-gate 	for (tp = bucket->ftb_data; tp != NULL; tp = tp->ftt_next) {
6547c478bd9Sstevel@tonic-gate 		if (tp->ftt_pid == pid && tp->ftt_pc == pc &&
655*b096140dSahl 		    tp->ftt_proc == provider->ftp_proc)
6567c478bd9Sstevel@tonic-gate 			break;
6577c478bd9Sstevel@tonic-gate 	}
6587c478bd9Sstevel@tonic-gate 
6597c478bd9Sstevel@tonic-gate 	/*
6607c478bd9Sstevel@tonic-gate 	 * If we somehow lost this tracepoint, we're in a world of hurt.
6617c478bd9Sstevel@tonic-gate 	 */
6627c478bd9Sstevel@tonic-gate 	ASSERT(tp != NULL);
6637c478bd9Sstevel@tonic-gate 
6647c478bd9Sstevel@tonic-gate 	if (probe->ftp_type == DTFTP_RETURN ||
6657c478bd9Sstevel@tonic-gate 	    probe->ftp_type == DTFTP_POST_OFFSETS) {
6667c478bd9Sstevel@tonic-gate 		ASSERT(tp->ftt_retids != NULL);
6677c478bd9Sstevel@tonic-gate 		idp = &tp->ftt_retids;
6687c478bd9Sstevel@tonic-gate 	} else {
6697c478bd9Sstevel@tonic-gate 		ASSERT(tp->ftt_ids != NULL);
6707c478bd9Sstevel@tonic-gate 		idp = &tp->ftt_ids;
6717c478bd9Sstevel@tonic-gate 	}
6727c478bd9Sstevel@tonic-gate 
6737c478bd9Sstevel@tonic-gate 	while ((*idp)->fti_probe != probe) {
6747c478bd9Sstevel@tonic-gate 		idp = &(*idp)->fti_next;
6757c478bd9Sstevel@tonic-gate 		ASSERT(*idp != NULL);
6767c478bd9Sstevel@tonic-gate 	}
6777c478bd9Sstevel@tonic-gate 
6787c478bd9Sstevel@tonic-gate 	id = *idp;
6797c478bd9Sstevel@tonic-gate 	*idp = id->fti_next;
6807c478bd9Sstevel@tonic-gate 	membar_producer();
6817c478bd9Sstevel@tonic-gate 
6827c478bd9Sstevel@tonic-gate 	ASSERT(id->fti_probe == probe);
6837c478bd9Sstevel@tonic-gate 
6847c478bd9Sstevel@tonic-gate 	/*
6857c478bd9Sstevel@tonic-gate 	 * If there are other registered enablings of this tracepoint, we're
6867c478bd9Sstevel@tonic-gate 	 * all done, but if this was the last probe assocated with this
6877c478bd9Sstevel@tonic-gate 	 * this tracepoint, we need to remove and free it.
6887c478bd9Sstevel@tonic-gate 	 */
6897c478bd9Sstevel@tonic-gate 	if (tp->ftt_ids != NULL || tp->ftt_retids != NULL) {
6907c478bd9Sstevel@tonic-gate 
6917c478bd9Sstevel@tonic-gate 		/*
6927c478bd9Sstevel@tonic-gate 		 * If the current probe's tracepoint is in use, swap it
6937c478bd9Sstevel@tonic-gate 		 * for an unused tracepoint.
6947c478bd9Sstevel@tonic-gate 		 */
6957c478bd9Sstevel@tonic-gate 		if (tp == probe->ftp_tps[index].fit_tp) {
6967c478bd9Sstevel@tonic-gate 			fasttrap_probe_t *tmp_probe;
6977c478bd9Sstevel@tonic-gate 			fasttrap_tracepoint_t **tmp_tp;
6987c478bd9Sstevel@tonic-gate 			uint_t tmp_index;
6997c478bd9Sstevel@tonic-gate 
7007c478bd9Sstevel@tonic-gate 			if (tp->ftt_ids != NULL) {
7017c478bd9Sstevel@tonic-gate 				tmp_probe = tp->ftt_ids->fti_probe;
7027c478bd9Sstevel@tonic-gate 				tmp_index = FASTTRAP_ID_INDEX(tp->ftt_ids);
7037c478bd9Sstevel@tonic-gate 				tmp_tp = &tmp_probe->ftp_tps[tmp_index].fit_tp;
7047c478bd9Sstevel@tonic-gate 			} else {
7057c478bd9Sstevel@tonic-gate 				tmp_probe = tp->ftt_retids->fti_probe;
7067c478bd9Sstevel@tonic-gate 				tmp_index = FASTTRAP_ID_INDEX(tp->ftt_retids);
7077c478bd9Sstevel@tonic-gate 				tmp_tp = &tmp_probe->ftp_tps[tmp_index].fit_tp;
7087c478bd9Sstevel@tonic-gate 			}
7097c478bd9Sstevel@tonic-gate 
7107c478bd9Sstevel@tonic-gate 			ASSERT(*tmp_tp != NULL);
7117c478bd9Sstevel@tonic-gate 			ASSERT(*tmp_tp != probe->ftp_tps[index].fit_tp);
7127c478bd9Sstevel@tonic-gate 			ASSERT((*tmp_tp)->ftt_ids == NULL);
7137c478bd9Sstevel@tonic-gate 			ASSERT((*tmp_tp)->ftt_retids == NULL);
7147c478bd9Sstevel@tonic-gate 
7157c478bd9Sstevel@tonic-gate 			probe->ftp_tps[index].fit_tp = *tmp_tp;
7167c478bd9Sstevel@tonic-gate 			*tmp_tp = tp;
7177c478bd9Sstevel@tonic-gate 
7187c478bd9Sstevel@tonic-gate 		}
7197c478bd9Sstevel@tonic-gate 
7207c478bd9Sstevel@tonic-gate 		mutex_exit(&bucket->ftb_mtx);
7217c478bd9Sstevel@tonic-gate 
7227c478bd9Sstevel@tonic-gate 		/*
7237c478bd9Sstevel@tonic-gate 		 * Tag the modified probe with the generation in which it was
7247c478bd9Sstevel@tonic-gate 		 * changed.
7257c478bd9Sstevel@tonic-gate 		 */
7267c478bd9Sstevel@tonic-gate 		probe->ftp_gen = fasttrap_mod_gen;
7277c478bd9Sstevel@tonic-gate 		return;
7287c478bd9Sstevel@tonic-gate 	}
7297c478bd9Sstevel@tonic-gate 
7307c478bd9Sstevel@tonic-gate 	mutex_exit(&bucket->ftb_mtx);
7317c478bd9Sstevel@tonic-gate 
7327c478bd9Sstevel@tonic-gate 	/*
7337c478bd9Sstevel@tonic-gate 	 * We can't safely remove the tracepoint from the set of active
7347c478bd9Sstevel@tonic-gate 	 * tracepoints until we've actually removed the fasttrap instruction
7357c478bd9Sstevel@tonic-gate 	 * from the process's text. We can, however, operate on this
7367c478bd9Sstevel@tonic-gate 	 * tracepoint secure in the knowledge that no other thread is going to
7377c478bd9Sstevel@tonic-gate 	 * be looking at it since we hold P_PR_LOCK on the process if it's
7387c478bd9Sstevel@tonic-gate 	 * live or we hold the provider lock on the process if it's dead and
7397c478bd9Sstevel@tonic-gate 	 * gone.
7407c478bd9Sstevel@tonic-gate 	 */
7417c478bd9Sstevel@tonic-gate 
7427c478bd9Sstevel@tonic-gate 	/*
7437c478bd9Sstevel@tonic-gate 	 * We only need to remove the actual instruction if we're looking
7447c478bd9Sstevel@tonic-gate 	 * at an existing process
7457c478bd9Sstevel@tonic-gate 	 */
7467c478bd9Sstevel@tonic-gate 	if (p != NULL) {
7477c478bd9Sstevel@tonic-gate 		/*
7487c478bd9Sstevel@tonic-gate 		 * If we fail to restore the instruction we need to kill
7497c478bd9Sstevel@tonic-gate 		 * this process since it's in a completely unrecoverable
7507c478bd9Sstevel@tonic-gate 		 * state.
7517c478bd9Sstevel@tonic-gate 		 */
7527c478bd9Sstevel@tonic-gate 		if (fasttrap_tracepoint_remove(p, tp) != 0)
7537c478bd9Sstevel@tonic-gate 			fasttrap_sigtrap(p, NULL, pc);
7547c478bd9Sstevel@tonic-gate 
7557c478bd9Sstevel@tonic-gate 		/*
7567c478bd9Sstevel@tonic-gate 		 * Decrement the count of the number of tracepoints active
7577c478bd9Sstevel@tonic-gate 		 * in the victim process.
7587c478bd9Sstevel@tonic-gate 		 */
7597c478bd9Sstevel@tonic-gate 		ASSERT(p->p_proc_flag & P_PR_LOCK);
7607c478bd9Sstevel@tonic-gate 		p->p_dtrace_count--;
7617c478bd9Sstevel@tonic-gate 	}
7627c478bd9Sstevel@tonic-gate 
7637c478bd9Sstevel@tonic-gate 	/*
7647c478bd9Sstevel@tonic-gate 	 * Remove the probe from the hash table of active tracepoints.
7657c478bd9Sstevel@tonic-gate 	 */
7667c478bd9Sstevel@tonic-gate 	mutex_enter(&bucket->ftb_mtx);
7677c478bd9Sstevel@tonic-gate 	pp = (fasttrap_tracepoint_t **)&bucket->ftb_data;
7687c478bd9Sstevel@tonic-gate 	ASSERT(*pp != NULL);
7697c478bd9Sstevel@tonic-gate 	while (*pp != tp) {
7707c478bd9Sstevel@tonic-gate 		pp = &(*pp)->ftt_next;
7717c478bd9Sstevel@tonic-gate 		ASSERT(*pp != NULL);
7727c478bd9Sstevel@tonic-gate 	}
7737c478bd9Sstevel@tonic-gate 
7747c478bd9Sstevel@tonic-gate 	*pp = tp->ftt_next;
7757c478bd9Sstevel@tonic-gate 	membar_producer();
7767c478bd9Sstevel@tonic-gate 
7777c478bd9Sstevel@tonic-gate 	mutex_exit(&bucket->ftb_mtx);
7787c478bd9Sstevel@tonic-gate 
7797c478bd9Sstevel@tonic-gate 	/*
7807c478bd9Sstevel@tonic-gate 	 * Tag the modified probe with the generation in which it was changed.
7817c478bd9Sstevel@tonic-gate 	 */
7827c478bd9Sstevel@tonic-gate 	probe->ftp_gen = fasttrap_mod_gen;
7837c478bd9Sstevel@tonic-gate }
7847c478bd9Sstevel@tonic-gate 
7857c478bd9Sstevel@tonic-gate typedef int fasttrap_probe_f(struct regs *);
7867c478bd9Sstevel@tonic-gate 
7877c478bd9Sstevel@tonic-gate static void
7887c478bd9Sstevel@tonic-gate fasttrap_enable_common(int *count, fasttrap_probe_f **fptr, fasttrap_probe_f *f,
7897c478bd9Sstevel@tonic-gate     fasttrap_probe_f **fptr2, fasttrap_probe_f *f2)
7907c478bd9Sstevel@tonic-gate {
7917c478bd9Sstevel@tonic-gate 	/*
7927c478bd9Sstevel@tonic-gate 	 * We don't have to play the rw lock game here because we're
7937c478bd9Sstevel@tonic-gate 	 * providing something rather than taking something away --
7947c478bd9Sstevel@tonic-gate 	 * we can be sure that no threads have tried to follow this
7957c478bd9Sstevel@tonic-gate 	 * function pointer yet.
7967c478bd9Sstevel@tonic-gate 	 */
7977c478bd9Sstevel@tonic-gate 	mutex_enter(&fasttrap_count_mtx);
7987c478bd9Sstevel@tonic-gate 	if (*count == 0) {
7997c478bd9Sstevel@tonic-gate 		ASSERT(*fptr == NULL);
8007c478bd9Sstevel@tonic-gate 		*fptr = f;
8017c478bd9Sstevel@tonic-gate 		if (fptr2 != NULL)
8027c478bd9Sstevel@tonic-gate 			*fptr2 = f2;
8037c478bd9Sstevel@tonic-gate 	}
8047c478bd9Sstevel@tonic-gate 	ASSERT(*fptr == f);
8057c478bd9Sstevel@tonic-gate 	ASSERT(fptr2 == NULL || *fptr2 == f2);
8067c478bd9Sstevel@tonic-gate 	(*count)++;
8077c478bd9Sstevel@tonic-gate 	mutex_exit(&fasttrap_count_mtx);
8087c478bd9Sstevel@tonic-gate }
8097c478bd9Sstevel@tonic-gate 
8107c478bd9Sstevel@tonic-gate static void
8117c478bd9Sstevel@tonic-gate fasttrap_disable_common(int *count, fasttrap_probe_f **fptr,
8127c478bd9Sstevel@tonic-gate     fasttrap_probe_f **fptr2)
8137c478bd9Sstevel@tonic-gate {
8147c478bd9Sstevel@tonic-gate 	ASSERT(MUTEX_HELD(&cpu_lock));
8157c478bd9Sstevel@tonic-gate 
8167c478bd9Sstevel@tonic-gate 	mutex_enter(&fasttrap_count_mtx);
8177c478bd9Sstevel@tonic-gate 	(*count)--;
8187c478bd9Sstevel@tonic-gate 	ASSERT(*count >= 0);
8197c478bd9Sstevel@tonic-gate 	if (*count == 0) {
8207c478bd9Sstevel@tonic-gate 		cpu_t *cur, *cpu = CPU;
8217c478bd9Sstevel@tonic-gate 
8227c478bd9Sstevel@tonic-gate 		for (cur = cpu->cpu_next_onln; cur != cpu;
8237c478bd9Sstevel@tonic-gate 			cur = cur->cpu_next_onln) {
8247c478bd9Sstevel@tonic-gate 			rw_enter(&cur->cpu_ft_lock, RW_WRITER);
8257c478bd9Sstevel@tonic-gate 		}
8267c478bd9Sstevel@tonic-gate 
8277c478bd9Sstevel@tonic-gate 		*fptr = NULL;
8287c478bd9Sstevel@tonic-gate 		if (fptr2 != NULL)
8297c478bd9Sstevel@tonic-gate 			*fptr2 = NULL;
8307c478bd9Sstevel@tonic-gate 
8317c478bd9Sstevel@tonic-gate 		for (cur = cpu->cpu_next_onln; cur != cpu;
8327c478bd9Sstevel@tonic-gate 			cur = cur->cpu_next_onln) {
8337c478bd9Sstevel@tonic-gate 			rw_exit(&cur->cpu_ft_lock);
8347c478bd9Sstevel@tonic-gate 		}
8357c478bd9Sstevel@tonic-gate 	}
8367c478bd9Sstevel@tonic-gate 	mutex_exit(&fasttrap_count_mtx);
8377c478bd9Sstevel@tonic-gate }
8387c478bd9Sstevel@tonic-gate 
8397c478bd9Sstevel@tonic-gate /*ARGSUSED*/
8407c478bd9Sstevel@tonic-gate static void
8417c478bd9Sstevel@tonic-gate fasttrap_enable(void *arg, dtrace_id_t id, void *parg)
8427c478bd9Sstevel@tonic-gate {
8437c478bd9Sstevel@tonic-gate 	/*
8447c478bd9Sstevel@tonic-gate 	 * Enable the probe that corresponds to statically placed trace
8457c478bd9Sstevel@tonic-gate 	 * points which have not explicitly been placed in the process's text
8467c478bd9Sstevel@tonic-gate 	 * by the fasttrap provider.
8477c478bd9Sstevel@tonic-gate 	 */
8487c478bd9Sstevel@tonic-gate 	ASSERT(arg == NULL);
8497c478bd9Sstevel@tonic-gate 	ASSERT(id == fasttrap_probe_id);
8507c478bd9Sstevel@tonic-gate 
8517c478bd9Sstevel@tonic-gate 	fasttrap_enable_common(&fasttrap_count,
8527c478bd9Sstevel@tonic-gate 	    &dtrace_fasttrap_probe_ptr, fasttrap_probe, NULL, NULL);
8537c478bd9Sstevel@tonic-gate }
8547c478bd9Sstevel@tonic-gate 
8557c478bd9Sstevel@tonic-gate 
8567c478bd9Sstevel@tonic-gate /*ARGSUSED*/
8577c478bd9Sstevel@tonic-gate static void
8587c478bd9Sstevel@tonic-gate fasttrap_pid_enable(void *arg, dtrace_id_t id, void *parg)
8597c478bd9Sstevel@tonic-gate {
8607c478bd9Sstevel@tonic-gate 	fasttrap_probe_t *probe = parg;
8617c478bd9Sstevel@tonic-gate 	proc_t *p;
8628f5db3e7Sahl 	int i, rc;
8637c478bd9Sstevel@tonic-gate 
8647c478bd9Sstevel@tonic-gate 	ASSERT(probe != NULL);
8657c478bd9Sstevel@tonic-gate 	ASSERT(!probe->ftp_enabled);
8667c478bd9Sstevel@tonic-gate 	ASSERT(id == probe->ftp_id);
8677c478bd9Sstevel@tonic-gate 	ASSERT(MUTEX_HELD(&cpu_lock));
8687c478bd9Sstevel@tonic-gate 
8697c478bd9Sstevel@tonic-gate 	/*
8707c478bd9Sstevel@tonic-gate 	 * Increment the count of enabled probes on this probe's provider;
8717c478bd9Sstevel@tonic-gate 	 * the provider can't go away while the probe still exists. We
8727c478bd9Sstevel@tonic-gate 	 * must increment this even if we aren't able to properly enable
8737c478bd9Sstevel@tonic-gate 	 * this probe.
8747c478bd9Sstevel@tonic-gate 	 */
8757c478bd9Sstevel@tonic-gate 	mutex_enter(&probe->ftp_prov->ftp_mtx);
8767c478bd9Sstevel@tonic-gate 	probe->ftp_prov->ftp_rcount++;
8777c478bd9Sstevel@tonic-gate 	mutex_exit(&probe->ftp_prov->ftp_mtx);
8787c478bd9Sstevel@tonic-gate 
8797c478bd9Sstevel@tonic-gate 	/*
8807c478bd9Sstevel@tonic-gate 	 * Bail out if we can't find the process for this probe or its
881*b096140dSahl 	 * provider is retired (meaning it was valid in a previously exec'ed
8827c478bd9Sstevel@tonic-gate 	 * incarnation of this address space). The provider can't go away
8837c478bd9Sstevel@tonic-gate 	 * while we're in this code path.
8847c478bd9Sstevel@tonic-gate 	 */
885*b096140dSahl 	if (probe->ftp_prov->ftp_retired ||
8867c478bd9Sstevel@tonic-gate 	    (p = sprlock(probe->ftp_pid)) == NULL)
8877c478bd9Sstevel@tonic-gate 		return;
8887c478bd9Sstevel@tonic-gate 
8897c478bd9Sstevel@tonic-gate 	ASSERT(!(p->p_flag & SVFORK));
8907c478bd9Sstevel@tonic-gate 	mutex_exit(&p->p_lock);
8917c478bd9Sstevel@tonic-gate 
8927c478bd9Sstevel@tonic-gate 	/*
8937c478bd9Sstevel@tonic-gate 	 * We have to enable the trap entry before any user threads have
8947c478bd9Sstevel@tonic-gate 	 * the chance to execute the trap instruction we're about to place
8957c478bd9Sstevel@tonic-gate 	 * in their process's text.
8967c478bd9Sstevel@tonic-gate 	 */
8977c478bd9Sstevel@tonic-gate 	fasttrap_enable_common(&fasttrap_pid_count,
8987c478bd9Sstevel@tonic-gate 	    &dtrace_pid_probe_ptr, fasttrap_pid_probe,
8997c478bd9Sstevel@tonic-gate 	    &dtrace_return_probe_ptr, fasttrap_return_probe);
9007c478bd9Sstevel@tonic-gate 
9017c478bd9Sstevel@tonic-gate 	/*
9027c478bd9Sstevel@tonic-gate 	 * Enable all the tracepoints and add this probe's id to each
9037c478bd9Sstevel@tonic-gate 	 * tracepoint's list of active probes.
9047c478bd9Sstevel@tonic-gate 	 */
9057c478bd9Sstevel@tonic-gate 	for (i = 0; i < probe->ftp_ntps; i++) {
9068f5db3e7Sahl 		if ((rc = fasttrap_tracepoint_enable(p, probe, i)) != 0) {
9078f5db3e7Sahl 			/*
9088f5db3e7Sahl 			 * If enabling the tracepoint failed completely,
9098f5db3e7Sahl 			 * we don't have to disable it; if the failure
9108f5db3e7Sahl 			 * was only partial we must disable it.
9118f5db3e7Sahl 			 */
9128f5db3e7Sahl 			if (rc == FASTTRAP_ENABLE_FAIL)
9138f5db3e7Sahl 				i--;
9148f5db3e7Sahl 			else
9158f5db3e7Sahl 				ASSERT(rc == FASTTRAP_ENABLE_PARTIAL);
9168f5db3e7Sahl 
9177c478bd9Sstevel@tonic-gate 			/*
9187c478bd9Sstevel@tonic-gate 			 * Back up and pull out all the tracepoints we've
9197c478bd9Sstevel@tonic-gate 			 * created so far for this probe.
9207c478bd9Sstevel@tonic-gate 			 */
9218f5db3e7Sahl 			while (i-- >= 0) {
9227c478bd9Sstevel@tonic-gate 				fasttrap_tracepoint_disable(p, probe, i);
9237c478bd9Sstevel@tonic-gate 			}
9247c478bd9Sstevel@tonic-gate 
9257c478bd9Sstevel@tonic-gate 			mutex_enter(&p->p_lock);
9267c478bd9Sstevel@tonic-gate 			sprunlock(p);
9277c478bd9Sstevel@tonic-gate 
9287c478bd9Sstevel@tonic-gate 			/*
9297c478bd9Sstevel@tonic-gate 			 * Since we're not actually enabling this probe,
9307c478bd9Sstevel@tonic-gate 			 * drop our reference on the trap table entry.
9317c478bd9Sstevel@tonic-gate 			 */
9327c478bd9Sstevel@tonic-gate 			fasttrap_disable_common(&fasttrap_pid_count,
9337c478bd9Sstevel@tonic-gate 			    &dtrace_pid_probe_ptr, &dtrace_return_probe_ptr);
9347c478bd9Sstevel@tonic-gate 			return;
9357c478bd9Sstevel@tonic-gate 		}
9367c478bd9Sstevel@tonic-gate 	}
9377c478bd9Sstevel@tonic-gate 
9387c478bd9Sstevel@tonic-gate 	mutex_enter(&p->p_lock);
9397c478bd9Sstevel@tonic-gate 	sprunlock(p);
9407c478bd9Sstevel@tonic-gate 
9417c478bd9Sstevel@tonic-gate 	probe->ftp_enabled = 1;
9427c478bd9Sstevel@tonic-gate }
9437c478bd9Sstevel@tonic-gate 
9447c478bd9Sstevel@tonic-gate 
9457c478bd9Sstevel@tonic-gate /*ARGSUSED*/
9467c478bd9Sstevel@tonic-gate static void
9477c478bd9Sstevel@tonic-gate fasttrap_disable(void *arg, dtrace_id_t id, void *parg)
9487c478bd9Sstevel@tonic-gate {
9497c478bd9Sstevel@tonic-gate 	/*
9507c478bd9Sstevel@tonic-gate 	 * Disable the probe the corresponds to statically placed trace
9517c478bd9Sstevel@tonic-gate 	 * points.
9527c478bd9Sstevel@tonic-gate 	 */
9537c478bd9Sstevel@tonic-gate 	ASSERT(arg == NULL);
9547c478bd9Sstevel@tonic-gate 	ASSERT(id == fasttrap_probe_id);
9557c478bd9Sstevel@tonic-gate 	ASSERT(MUTEX_HELD(&cpu_lock));
9567c478bd9Sstevel@tonic-gate 	fasttrap_disable_common(&fasttrap_count, &dtrace_fasttrap_probe_ptr,
9577c478bd9Sstevel@tonic-gate 	    NULL);
9587c478bd9Sstevel@tonic-gate }
9597c478bd9Sstevel@tonic-gate 
9607c478bd9Sstevel@tonic-gate /*ARGSUSED*/
9617c478bd9Sstevel@tonic-gate static void
9627c478bd9Sstevel@tonic-gate fasttrap_pid_disable(void *arg, dtrace_id_t id, void *parg)
9637c478bd9Sstevel@tonic-gate {
9647c478bd9Sstevel@tonic-gate 	fasttrap_probe_t *probe = parg;
9657c478bd9Sstevel@tonic-gate 	fasttrap_provider_t *provider = probe->ftp_prov;
9667c478bd9Sstevel@tonic-gate 	proc_t *p;
9677c478bd9Sstevel@tonic-gate 	int i, whack = 0;
9687c478bd9Sstevel@tonic-gate 
9697c478bd9Sstevel@tonic-gate 	if (!probe->ftp_enabled) {
9707c478bd9Sstevel@tonic-gate 		mutex_enter(&provider->ftp_mtx);
9717c478bd9Sstevel@tonic-gate 		provider->ftp_rcount--;
9727c478bd9Sstevel@tonic-gate 		ASSERT(provider->ftp_rcount >= 0);
9737c478bd9Sstevel@tonic-gate 		mutex_exit(&provider->ftp_mtx);
9747c478bd9Sstevel@tonic-gate 		return;
9757c478bd9Sstevel@tonic-gate 	}
9767c478bd9Sstevel@tonic-gate 
9777c478bd9Sstevel@tonic-gate 	ASSERT(id == probe->ftp_id);
9787c478bd9Sstevel@tonic-gate 
9797c478bd9Sstevel@tonic-gate 	/*
9807c478bd9Sstevel@tonic-gate 	 * We won't be able to acquire a /proc-esque lock on the process
9817c478bd9Sstevel@tonic-gate 	 * iff the process is dead and gone. In this case, we rely on the
9827c478bd9Sstevel@tonic-gate 	 * provider lock as a point of mutual exclusion to prevent other
9837c478bd9Sstevel@tonic-gate 	 * DTrace consumers from disabling this probe.
9847c478bd9Sstevel@tonic-gate 	 */
9857c478bd9Sstevel@tonic-gate 	if ((p = sprlock(probe->ftp_pid)) != NULL) {
9867c478bd9Sstevel@tonic-gate 		ASSERT(!(p->p_flag & SVFORK));
9877c478bd9Sstevel@tonic-gate 		mutex_exit(&p->p_lock);
9887c478bd9Sstevel@tonic-gate 	}
9897c478bd9Sstevel@tonic-gate 
9907c478bd9Sstevel@tonic-gate 	mutex_enter(&provider->ftp_mtx);
9917c478bd9Sstevel@tonic-gate 
9927c478bd9Sstevel@tonic-gate 	/*
9937c478bd9Sstevel@tonic-gate 	 * Disable all the associated tracepoints.
9947c478bd9Sstevel@tonic-gate 	 */
9957c478bd9Sstevel@tonic-gate 	for (i = 0; i < probe->ftp_ntps; i++) {
9967c478bd9Sstevel@tonic-gate 		fasttrap_tracepoint_disable(p, probe, i);
9977c478bd9Sstevel@tonic-gate 	}
9987c478bd9Sstevel@tonic-gate 
9997c478bd9Sstevel@tonic-gate 	ASSERT(provider->ftp_rcount > 0);
10007c478bd9Sstevel@tonic-gate 	provider->ftp_rcount--;
10017c478bd9Sstevel@tonic-gate 
10027c478bd9Sstevel@tonic-gate 	if (p != NULL) {
10037c478bd9Sstevel@tonic-gate 		/*
10047c478bd9Sstevel@tonic-gate 		 * Even though we may not be able to remove it entirely, we
1005*b096140dSahl 		 * mark this retired provider to get a chance to remove some
10067c478bd9Sstevel@tonic-gate 		 * of the associated probes.
10077c478bd9Sstevel@tonic-gate 		 */
1008*b096140dSahl 		if (provider->ftp_retired && !provider->ftp_marked)
10097c478bd9Sstevel@tonic-gate 			whack = provider->ftp_marked = 1;
10107c478bd9Sstevel@tonic-gate 		mutex_exit(&provider->ftp_mtx);
10117c478bd9Sstevel@tonic-gate 
10127c478bd9Sstevel@tonic-gate 		mutex_enter(&p->p_lock);
10137c478bd9Sstevel@tonic-gate 		sprunlock(p);
10147c478bd9Sstevel@tonic-gate 	} else {
10157c478bd9Sstevel@tonic-gate 		/*
10167c478bd9Sstevel@tonic-gate 		 * If the process is dead, we're just waiting for the
10177c478bd9Sstevel@tonic-gate 		 * last probe to be disabled to be able to free it.
10187c478bd9Sstevel@tonic-gate 		 */
10197c478bd9Sstevel@tonic-gate 		if (provider->ftp_rcount == 0 && !provider->ftp_marked)
10207c478bd9Sstevel@tonic-gate 			whack = provider->ftp_marked = 1;
10217c478bd9Sstevel@tonic-gate 		mutex_exit(&provider->ftp_mtx);
10227c478bd9Sstevel@tonic-gate 	}
10237c478bd9Sstevel@tonic-gate 
10247c478bd9Sstevel@tonic-gate 	if (whack)
10257c478bd9Sstevel@tonic-gate 		fasttrap_pid_cleanup();
10267c478bd9Sstevel@tonic-gate 
10277c478bd9Sstevel@tonic-gate 	probe->ftp_enabled = 0;
10287c478bd9Sstevel@tonic-gate 
10297c478bd9Sstevel@tonic-gate 	ASSERT(MUTEX_HELD(&cpu_lock));
10307c478bd9Sstevel@tonic-gate 	fasttrap_disable_common(&fasttrap_pid_count, &dtrace_pid_probe_ptr,
10317c478bd9Sstevel@tonic-gate 	    &dtrace_return_probe_ptr);
10327c478bd9Sstevel@tonic-gate }
10337c478bd9Sstevel@tonic-gate 
10347c478bd9Sstevel@tonic-gate /*ARGSUSED*/
10357c478bd9Sstevel@tonic-gate static void
10367c478bd9Sstevel@tonic-gate fasttrap_pid_getargdesc(void *arg, dtrace_id_t id, void *parg,
10377c478bd9Sstevel@tonic-gate     dtrace_argdesc_t *desc)
10387c478bd9Sstevel@tonic-gate {
10397c478bd9Sstevel@tonic-gate 	fasttrap_probe_t *probe = parg;
10407c478bd9Sstevel@tonic-gate 	char *str;
10417c478bd9Sstevel@tonic-gate 	int i;
10427c478bd9Sstevel@tonic-gate 
10437c478bd9Sstevel@tonic-gate 	desc->dtargd_native[0] = '\0';
10447c478bd9Sstevel@tonic-gate 	desc->dtargd_xlate[0] = '\0';
10457c478bd9Sstevel@tonic-gate 
1046*b096140dSahl 	if (probe->ftp_prov->ftp_retired != 0 ||
10477c478bd9Sstevel@tonic-gate 	    desc->dtargd_ndx >= probe->ftp_nargs) {
10487c478bd9Sstevel@tonic-gate 		desc->dtargd_ndx = DTRACE_ARGNONE;
10497c478bd9Sstevel@tonic-gate 		return;
10507c478bd9Sstevel@tonic-gate 	}
10517c478bd9Sstevel@tonic-gate 
10527c478bd9Sstevel@tonic-gate 	/*
10537c478bd9Sstevel@tonic-gate 	 * We only need to set this member if the argument is remapped.
10547c478bd9Sstevel@tonic-gate 	 */
10557c478bd9Sstevel@tonic-gate 	if (probe->ftp_argmap != NULL)
10567c478bd9Sstevel@tonic-gate 		desc->dtargd_mapping = probe->ftp_argmap[desc->dtargd_ndx];
10577c478bd9Sstevel@tonic-gate 
10587c478bd9Sstevel@tonic-gate 	str = probe->ftp_ntypes;
10597c478bd9Sstevel@tonic-gate 	for (i = 0; i < desc->dtargd_mapping; i++) {
10607c478bd9Sstevel@tonic-gate 		str += strlen(str) + 1;
10617c478bd9Sstevel@tonic-gate 	}
10627c478bd9Sstevel@tonic-gate 
10637c478bd9Sstevel@tonic-gate 	ASSERT(strlen(str + 1) < sizeof (desc->dtargd_native));
10647c478bd9Sstevel@tonic-gate 	(void) strcpy(desc->dtargd_native, str);
10657c478bd9Sstevel@tonic-gate 
10667c478bd9Sstevel@tonic-gate 	if (probe->ftp_xtypes == NULL)
10677c478bd9Sstevel@tonic-gate 		return;
10687c478bd9Sstevel@tonic-gate 
10697c478bd9Sstevel@tonic-gate 	str = probe->ftp_xtypes;
10707c478bd9Sstevel@tonic-gate 	for (i = 0; i < desc->dtargd_ndx; i++) {
10717c478bd9Sstevel@tonic-gate 		str += strlen(str) + 1;
10727c478bd9Sstevel@tonic-gate 	}
10737c478bd9Sstevel@tonic-gate 
10747c478bd9Sstevel@tonic-gate 	ASSERT(strlen(str + 1) < sizeof (desc->dtargd_xlate));
10757c478bd9Sstevel@tonic-gate 	(void) strcpy(desc->dtargd_xlate, str);
10767c478bd9Sstevel@tonic-gate }
10777c478bd9Sstevel@tonic-gate 
10787c478bd9Sstevel@tonic-gate /*ARGSUSED*/
10797c478bd9Sstevel@tonic-gate static void
10807c478bd9Sstevel@tonic-gate fasttrap_destroy(void *arg, dtrace_id_t id, void *parg)
10817c478bd9Sstevel@tonic-gate {
10827c478bd9Sstevel@tonic-gate 	ASSERT(arg == NULL);
10837c478bd9Sstevel@tonic-gate 	ASSERT(id == fasttrap_probe_id);
10847c478bd9Sstevel@tonic-gate }
10857c478bd9Sstevel@tonic-gate 
10867c478bd9Sstevel@tonic-gate /*ARGSUSED*/
10877c478bd9Sstevel@tonic-gate static void
10887c478bd9Sstevel@tonic-gate fasttrap_pid_destroy(void *arg, dtrace_id_t id, void *parg)
10897c478bd9Sstevel@tonic-gate {
10907c478bd9Sstevel@tonic-gate 	fasttrap_probe_t *probe = parg;
10917c478bd9Sstevel@tonic-gate 	int i;
10927c478bd9Sstevel@tonic-gate 	size_t size;
10937c478bd9Sstevel@tonic-gate 
10947c478bd9Sstevel@tonic-gate 	ASSERT(probe != NULL);
10957c478bd9Sstevel@tonic-gate 	ASSERT(!probe->ftp_enabled);
10967c478bd9Sstevel@tonic-gate 	ASSERT(fasttrap_total >= probe->ftp_ntps);
10977c478bd9Sstevel@tonic-gate 
10987c478bd9Sstevel@tonic-gate 	atomic_add_32(&fasttrap_total, -probe->ftp_ntps);
10997c478bd9Sstevel@tonic-gate 	size = sizeof (fasttrap_probe_t) +
11007c478bd9Sstevel@tonic-gate 	    sizeof (probe->ftp_tps[0]) * (probe->ftp_ntps - 1);
11017c478bd9Sstevel@tonic-gate 
11027c478bd9Sstevel@tonic-gate 	if (probe->ftp_gen + 1 >= fasttrap_mod_gen)
11037c478bd9Sstevel@tonic-gate 		fasttrap_mod_barrier(probe->ftp_gen);
11047c478bd9Sstevel@tonic-gate 
11057c478bd9Sstevel@tonic-gate 	for (i = 0; i < probe->ftp_ntps; i++) {
11067c478bd9Sstevel@tonic-gate 		kmem_free(probe->ftp_tps[i].fit_tp,
11077c478bd9Sstevel@tonic-gate 		    sizeof (fasttrap_tracepoint_t));
11087c478bd9Sstevel@tonic-gate 	}
11097c478bd9Sstevel@tonic-gate 
11107c478bd9Sstevel@tonic-gate 	kmem_free(probe, size);
11117c478bd9Sstevel@tonic-gate }
11127c478bd9Sstevel@tonic-gate 
11137c478bd9Sstevel@tonic-gate 
11147c478bd9Sstevel@tonic-gate static const dtrace_pattr_t fasttrap_attr = {
11157c478bd9Sstevel@tonic-gate { DTRACE_STABILITY_EVOLVING, DTRACE_STABILITY_EVOLVING, DTRACE_CLASS_ISA },
11167c478bd9Sstevel@tonic-gate { DTRACE_STABILITY_PRIVATE, DTRACE_STABILITY_PRIVATE, DTRACE_CLASS_UNKNOWN },
11177c478bd9Sstevel@tonic-gate { DTRACE_STABILITY_PRIVATE, DTRACE_STABILITY_PRIVATE, DTRACE_CLASS_UNKNOWN },
11187c478bd9Sstevel@tonic-gate { DTRACE_STABILITY_EVOLVING, DTRACE_STABILITY_EVOLVING, DTRACE_CLASS_ISA },
11197c478bd9Sstevel@tonic-gate { DTRACE_STABILITY_EVOLVING, DTRACE_STABILITY_EVOLVING, DTRACE_CLASS_ISA },
11207c478bd9Sstevel@tonic-gate };
11217c478bd9Sstevel@tonic-gate 
11227c478bd9Sstevel@tonic-gate static dtrace_pops_t fasttrap_pops = {
11237c478bd9Sstevel@tonic-gate 	fasttrap_provide,
11247c478bd9Sstevel@tonic-gate 	NULL,
11257c478bd9Sstevel@tonic-gate 	fasttrap_enable,
11267c478bd9Sstevel@tonic-gate 	fasttrap_disable,
11277c478bd9Sstevel@tonic-gate 	NULL,
11287c478bd9Sstevel@tonic-gate 	NULL,
11297c478bd9Sstevel@tonic-gate 	NULL,
11307c478bd9Sstevel@tonic-gate 	fasttrap_getarg,
11317c478bd9Sstevel@tonic-gate 	NULL,
11327c478bd9Sstevel@tonic-gate 	fasttrap_destroy
11337c478bd9Sstevel@tonic-gate };
11347c478bd9Sstevel@tonic-gate 
11357c478bd9Sstevel@tonic-gate static const dtrace_pattr_t pid_attr = {
11367c478bd9Sstevel@tonic-gate { DTRACE_STABILITY_EVOLVING, DTRACE_STABILITY_EVOLVING, DTRACE_CLASS_ISA },
11377c478bd9Sstevel@tonic-gate { DTRACE_STABILITY_PRIVATE, DTRACE_STABILITY_PRIVATE, DTRACE_CLASS_UNKNOWN },
11387c478bd9Sstevel@tonic-gate { DTRACE_STABILITY_PRIVATE, DTRACE_STABILITY_PRIVATE, DTRACE_CLASS_UNKNOWN },
11397c478bd9Sstevel@tonic-gate { DTRACE_STABILITY_EVOLVING, DTRACE_STABILITY_EVOLVING, DTRACE_CLASS_ISA },
11407c478bd9Sstevel@tonic-gate { DTRACE_STABILITY_PRIVATE, DTRACE_STABILITY_PRIVATE, DTRACE_CLASS_UNKNOWN },
11417c478bd9Sstevel@tonic-gate };
11427c478bd9Sstevel@tonic-gate 
11437c478bd9Sstevel@tonic-gate static dtrace_pops_t pid_pops = {
11447c478bd9Sstevel@tonic-gate 	fasttrap_pid_provide,
11457c478bd9Sstevel@tonic-gate 	NULL,
11467c478bd9Sstevel@tonic-gate 	fasttrap_pid_enable,
11477c478bd9Sstevel@tonic-gate 	fasttrap_pid_disable,
11487c478bd9Sstevel@tonic-gate 	NULL,
11497c478bd9Sstevel@tonic-gate 	NULL,
11507c478bd9Sstevel@tonic-gate 	fasttrap_pid_getargdesc,
11517c478bd9Sstevel@tonic-gate 	fasttrap_getarg,
11527c478bd9Sstevel@tonic-gate 	NULL,
11537c478bd9Sstevel@tonic-gate 	fasttrap_pid_destroy
11547c478bd9Sstevel@tonic-gate };
11557c478bd9Sstevel@tonic-gate 
11567c478bd9Sstevel@tonic-gate static dtrace_pops_t usdt_pops = {
11577c478bd9Sstevel@tonic-gate 	fasttrap_pid_provide,
11587c478bd9Sstevel@tonic-gate 	NULL,
11597c478bd9Sstevel@tonic-gate 	fasttrap_pid_enable,
11607c478bd9Sstevel@tonic-gate 	fasttrap_pid_disable,
11617c478bd9Sstevel@tonic-gate 	NULL,
11627c478bd9Sstevel@tonic-gate 	NULL,
11637c478bd9Sstevel@tonic-gate 	fasttrap_pid_getargdesc,
11647c478bd9Sstevel@tonic-gate 	fasttrap_usdt_getarg,
11657c478bd9Sstevel@tonic-gate 	NULL,
11667c478bd9Sstevel@tonic-gate 	fasttrap_pid_destroy
11677c478bd9Sstevel@tonic-gate };
11687c478bd9Sstevel@tonic-gate 
1169*b096140dSahl static fasttrap_proc_t *
1170*b096140dSahl fasttrap_proc_lookup(pid_t pid)
1171*b096140dSahl {
1172*b096140dSahl 	fasttrap_bucket_t *bucket;
1173*b096140dSahl 	fasttrap_proc_t *fprc, *new_fprc;
1174*b096140dSahl 
1175*b096140dSahl 	bucket = &fasttrap_procs.fth_table[FASTTRAP_PROCS_INDEX(pid)];
1176*b096140dSahl 	mutex_enter(&bucket->ftb_mtx);
1177*b096140dSahl 
1178*b096140dSahl 	for (fprc = bucket->ftb_data; fprc != NULL; fprc = fprc->ftpc_next) {
1179*b096140dSahl 		if (fprc->ftpc_pid == pid && !fprc->ftpc_defunct) {
1180*b096140dSahl 			mutex_enter(&fprc->ftpc_mtx);
1181*b096140dSahl 			mutex_exit(&bucket->ftb_mtx);
1182*b096140dSahl 			fprc->ftpc_count++;
1183*b096140dSahl 			mutex_exit(&fprc->ftpc_mtx);
1184*b096140dSahl 
1185*b096140dSahl 			return (fprc);
1186*b096140dSahl 		}
1187*b096140dSahl 	}
1188*b096140dSahl 
1189*b096140dSahl 	/*
1190*b096140dSahl 	 * Drop the bucket lock so we don't try to perform a sleeping
1191*b096140dSahl 	 * allocation under it.
1192*b096140dSahl 	 */
1193*b096140dSahl 	mutex_exit(&bucket->ftb_mtx);
1194*b096140dSahl 
1195*b096140dSahl 	new_fprc = kmem_zalloc(sizeof (fasttrap_proc_t), KM_SLEEP);
1196*b096140dSahl 	new_fprc->ftpc_pid = pid;
1197*b096140dSahl 	new_fprc->ftpc_count = 1;
1198*b096140dSahl 
1199*b096140dSahl 	mutex_enter(&bucket->ftb_mtx);
1200*b096140dSahl 
1201*b096140dSahl 	/*
1202*b096140dSahl 	 * Take another lap through the list to make sure a proc hasn't
1203*b096140dSahl 	 * been created for this pid while we weren't under the bucket lock.
1204*b096140dSahl 	 */
1205*b096140dSahl 	for (fprc = bucket->ftb_data; fprc != NULL; fprc = fprc->ftpc_next) {
1206*b096140dSahl 		if (fprc->ftpc_pid == pid && !fprc->ftpc_defunct) {
1207*b096140dSahl 			mutex_enter(&fprc->ftpc_mtx);
1208*b096140dSahl 			mutex_exit(&bucket->ftb_mtx);
1209*b096140dSahl 			fprc->ftpc_count++;
1210*b096140dSahl 			mutex_exit(&fprc->ftpc_mtx);
1211*b096140dSahl 
1212*b096140dSahl 			kmem_free(new_fprc, sizeof (fasttrap_proc_t));
1213*b096140dSahl 
1214*b096140dSahl 			return (fprc);
1215*b096140dSahl 		}
1216*b096140dSahl 	}
1217*b096140dSahl 
1218*b096140dSahl 	new_fprc->ftpc_next = bucket->ftb_data;
1219*b096140dSahl 	bucket->ftb_data = new_fprc;
1220*b096140dSahl 
1221*b096140dSahl 	mutex_exit(&bucket->ftb_mtx);
1222*b096140dSahl 
1223*b096140dSahl 	return (new_fprc);
1224*b096140dSahl }
1225*b096140dSahl 
1226*b096140dSahl static void
1227*b096140dSahl fasttrap_proc_release(fasttrap_proc_t *proc)
1228*b096140dSahl {
1229*b096140dSahl 	fasttrap_bucket_t *bucket;
1230*b096140dSahl 	fasttrap_proc_t *fprc, **fprcp;
1231*b096140dSahl 	pid_t pid = proc->ftpc_pid;
1232*b096140dSahl 
1233*b096140dSahl 	mutex_enter(&proc->ftpc_mtx);
1234*b096140dSahl 
1235*b096140dSahl 	ASSERT(proc->ftpc_count != 0);
1236*b096140dSahl 
1237*b096140dSahl 	if (--proc->ftpc_count != 0) {
1238*b096140dSahl 		mutex_exit(&proc->ftpc_mtx);
1239*b096140dSahl 		return;
1240*b096140dSahl 	}
1241*b096140dSahl 
1242*b096140dSahl 	mutex_exit(&proc->ftpc_mtx);
1243*b096140dSahl 
1244*b096140dSahl 	bucket = &fasttrap_procs.fth_table[FASTTRAP_PROCS_INDEX(pid)];
1245*b096140dSahl 	mutex_enter(&bucket->ftb_mtx);
1246*b096140dSahl 
1247*b096140dSahl 	fprcp = (fasttrap_proc_t **)&bucket->ftb_data;
1248*b096140dSahl 	while ((fprc = *fprcp) != NULL) {
1249*b096140dSahl 		if (fprc == proc)
1250*b096140dSahl 			break;
1251*b096140dSahl 
1252*b096140dSahl 		fprcp = &fprc->ftpc_next;
1253*b096140dSahl 	}
1254*b096140dSahl 
1255*b096140dSahl 	/*
1256*b096140dSahl 	 * Something strange has happened if we can't find the proc.
1257*b096140dSahl 	 */
1258*b096140dSahl 	ASSERT(fprc != NULL);
1259*b096140dSahl 
1260*b096140dSahl 	*fprcp = fprc->ftpc_next;
1261*b096140dSahl 
1262*b096140dSahl 	mutex_exit(&bucket->ftb_mtx);
1263*b096140dSahl 
1264*b096140dSahl 	kmem_free(fprc, sizeof (fasttrap_proc_t));
1265*b096140dSahl }
1266*b096140dSahl 
12677c478bd9Sstevel@tonic-gate /*
12687c478bd9Sstevel@tonic-gate  * Lookup a fasttrap-managed provider based on its name and associated pid.
12697c478bd9Sstevel@tonic-gate  * If the pattr argument is non-NULL, this function instantiates the provider
12707c478bd9Sstevel@tonic-gate  * if it doesn't exist otherwise it returns NULL. The provider is returned
12717c478bd9Sstevel@tonic-gate  * with its lock held.
12727c478bd9Sstevel@tonic-gate  */
12737c478bd9Sstevel@tonic-gate static fasttrap_provider_t *
12747c478bd9Sstevel@tonic-gate fasttrap_provider_lookup(pid_t pid, const char *name,
12757c478bd9Sstevel@tonic-gate     const dtrace_pattr_t *pattr)
12767c478bd9Sstevel@tonic-gate {
12777c478bd9Sstevel@tonic-gate 	fasttrap_provider_t *fp, *new_fp = NULL;
12787c478bd9Sstevel@tonic-gate 	fasttrap_bucket_t *bucket;
12797c478bd9Sstevel@tonic-gate 	char provname[DTRACE_PROVNAMELEN];
12807c478bd9Sstevel@tonic-gate 	proc_t *p;
12817c478bd9Sstevel@tonic-gate 	uid_t uid = (uid_t)-1;
12827c478bd9Sstevel@tonic-gate 
12837c478bd9Sstevel@tonic-gate 	ASSERT(strlen(name) < sizeof (fp->ftp_name));
12847c478bd9Sstevel@tonic-gate 
12857c478bd9Sstevel@tonic-gate 	bucket = &fasttrap_provs.fth_table[FASTTRAP_PROVS_INDEX(pid, name)];
12867c478bd9Sstevel@tonic-gate 	mutex_enter(&bucket->ftb_mtx);
12877c478bd9Sstevel@tonic-gate 
12887c478bd9Sstevel@tonic-gate 	/*
12897c478bd9Sstevel@tonic-gate 	 * Take a lap through the list and return the match if we find it.
12907c478bd9Sstevel@tonic-gate 	 */
12917c478bd9Sstevel@tonic-gate 	for (fp = bucket->ftb_data; fp != NULL; fp = fp->ftp_next) {
12927c478bd9Sstevel@tonic-gate 		if (fp->ftp_pid == pid && strcmp(fp->ftp_name, name) == 0 &&
1293*b096140dSahl 		    !fp->ftp_retired) {
12947c478bd9Sstevel@tonic-gate 			mutex_enter(&fp->ftp_mtx);
12957c478bd9Sstevel@tonic-gate 			mutex_exit(&bucket->ftb_mtx);
12967c478bd9Sstevel@tonic-gate 			return (fp);
12977c478bd9Sstevel@tonic-gate 		}
12987c478bd9Sstevel@tonic-gate 	}
12997c478bd9Sstevel@tonic-gate 
13007c478bd9Sstevel@tonic-gate 	/*
13017c478bd9Sstevel@tonic-gate 	 * Drop the bucket lock so we don't try to perform a sleeping
13027c478bd9Sstevel@tonic-gate 	 * allocation under it.
13037c478bd9Sstevel@tonic-gate 	 */
13047c478bd9Sstevel@tonic-gate 	mutex_exit(&bucket->ftb_mtx);
13057c478bd9Sstevel@tonic-gate 
1306*b096140dSahl 	/*
1307*b096140dSahl 	 * If we didn't want to create a new provider, just return failure.
1308*b096140dSahl 	 */
13097c478bd9Sstevel@tonic-gate 	if (pattr == NULL)
13107c478bd9Sstevel@tonic-gate 		return (NULL);
13117c478bd9Sstevel@tonic-gate 
13127c478bd9Sstevel@tonic-gate 	/*
13137c478bd9Sstevel@tonic-gate 	 * Make sure the process exists, isn't a child created as the result
13147c478bd9Sstevel@tonic-gate 	 * of a vfork(2), and isn't a zombie (but may be in fork). Record the
13157c478bd9Sstevel@tonic-gate 	 * process's uid to pass to dtrace_register().
13167c478bd9Sstevel@tonic-gate 	 */
13177c478bd9Sstevel@tonic-gate 	mutex_enter(&pidlock);
131897eda132Sraf 	if ((p = prfind(pid)) == NULL) {
13197c478bd9Sstevel@tonic-gate 		mutex_exit(&pidlock);
13207c478bd9Sstevel@tonic-gate 		return (NULL);
13217c478bd9Sstevel@tonic-gate 	}
13227c478bd9Sstevel@tonic-gate 	mutex_enter(&p->p_lock);
13237c478bd9Sstevel@tonic-gate 	mutex_exit(&pidlock);
132497eda132Sraf 	if (p->p_flag & (SVFORK | SEXITING)) {
132597eda132Sraf 		mutex_exit(&p->p_lock);
132697eda132Sraf 		return (NULL);
132797eda132Sraf 	}
13287c478bd9Sstevel@tonic-gate 
13297c478bd9Sstevel@tonic-gate 	/*
13307c478bd9Sstevel@tonic-gate 	 * Increment p_dtrace_probes so that the process knows to inform us
13317c478bd9Sstevel@tonic-gate 	 * when it exits or execs. fasttrap_provider_free() decrements this
13327c478bd9Sstevel@tonic-gate 	 * when we're done with this provider.
13337c478bd9Sstevel@tonic-gate 	 */
13347c478bd9Sstevel@tonic-gate 	p->p_dtrace_probes++;
13357c478bd9Sstevel@tonic-gate 
13367c478bd9Sstevel@tonic-gate 	mutex_enter(&p->p_crlock);
13377c478bd9Sstevel@tonic-gate 	uid = crgetruid(p->p_cred);
13387c478bd9Sstevel@tonic-gate 	mutex_exit(&p->p_crlock);
13397c478bd9Sstevel@tonic-gate 	mutex_exit(&p->p_lock);
13407c478bd9Sstevel@tonic-gate 
13417c478bd9Sstevel@tonic-gate 	new_fp = kmem_zalloc(sizeof (fasttrap_provider_t), KM_SLEEP);
1342*b096140dSahl 	new_fp->ftp_pid = pid;
1343*b096140dSahl 	new_fp->ftp_proc = fasttrap_proc_lookup(pid);
1344*b096140dSahl 
1345*b096140dSahl 	ASSERT(new_fp->ftp_proc != NULL);
13467c478bd9Sstevel@tonic-gate 
13477c478bd9Sstevel@tonic-gate 	mutex_enter(&bucket->ftb_mtx);
13487c478bd9Sstevel@tonic-gate 
13497c478bd9Sstevel@tonic-gate 	/*
13507c478bd9Sstevel@tonic-gate 	 * Take another lap through the list to make sure a provider hasn't
13517c478bd9Sstevel@tonic-gate 	 * been created for this pid while we weren't under the bucket lock.
13527c478bd9Sstevel@tonic-gate 	 */
13537c478bd9Sstevel@tonic-gate 	for (fp = bucket->ftb_data; fp != NULL; fp = fp->ftp_next) {
13547c478bd9Sstevel@tonic-gate 		if (fp->ftp_pid == pid && strcmp(fp->ftp_name, name) == 0 &&
1355*b096140dSahl 		    !fp->ftp_retired) {
13567c478bd9Sstevel@tonic-gate 			mutex_enter(&fp->ftp_mtx);
13577c478bd9Sstevel@tonic-gate 			mutex_exit(&bucket->ftb_mtx);
13587c478bd9Sstevel@tonic-gate 			fasttrap_provider_free(new_fp);
13597c478bd9Sstevel@tonic-gate 			return (fp);
13607c478bd9Sstevel@tonic-gate 		}
13617c478bd9Sstevel@tonic-gate 	}
13627c478bd9Sstevel@tonic-gate 
13637c478bd9Sstevel@tonic-gate 	(void) strcpy(new_fp->ftp_name, name);
13647c478bd9Sstevel@tonic-gate 
13657c478bd9Sstevel@tonic-gate 	/*
13667c478bd9Sstevel@tonic-gate 	 * Fail and return NULL if either the provider name is too long
13677c478bd9Sstevel@tonic-gate 	 * or we fail to register this new provider with the DTrace
13687c478bd9Sstevel@tonic-gate 	 * framework. Note that this is the only place we ever construct
13697c478bd9Sstevel@tonic-gate 	 * the full provider name -- we keep it in pieces in the provider
13707c478bd9Sstevel@tonic-gate 	 * structure.
13717c478bd9Sstevel@tonic-gate 	 */
13727c478bd9Sstevel@tonic-gate 	if (snprintf(provname, sizeof (provname), "%s%u", name, (uint_t)pid) >=
13737c478bd9Sstevel@tonic-gate 	    sizeof (provname) ||
13747c478bd9Sstevel@tonic-gate 	    dtrace_register(provname, pattr,
13757c478bd9Sstevel@tonic-gate 	    DTRACE_PRIV_PROC | DTRACE_PRIV_OWNER, uid,
13767c478bd9Sstevel@tonic-gate 	    pattr == &pid_attr ? &pid_pops : &usdt_pops, new_fp,
13777c478bd9Sstevel@tonic-gate 	    &new_fp->ftp_provid) != 0) {
13787c478bd9Sstevel@tonic-gate 		mutex_exit(&bucket->ftb_mtx);
13797c478bd9Sstevel@tonic-gate 		fasttrap_provider_free(new_fp);
13807c478bd9Sstevel@tonic-gate 		return (NULL);
13817c478bd9Sstevel@tonic-gate 	}
13827c478bd9Sstevel@tonic-gate 
13837c478bd9Sstevel@tonic-gate 	new_fp->ftp_next = bucket->ftb_data;
13847c478bd9Sstevel@tonic-gate 	bucket->ftb_data = new_fp;
13857c478bd9Sstevel@tonic-gate 
13867c478bd9Sstevel@tonic-gate 	mutex_enter(&new_fp->ftp_mtx);
13877c478bd9Sstevel@tonic-gate 	mutex_exit(&bucket->ftb_mtx);
13887c478bd9Sstevel@tonic-gate 
13897c478bd9Sstevel@tonic-gate 	return (new_fp);
13907c478bd9Sstevel@tonic-gate }
13917c478bd9Sstevel@tonic-gate 
13927c478bd9Sstevel@tonic-gate static void
13937c478bd9Sstevel@tonic-gate fasttrap_provider_free(fasttrap_provider_t *provider)
13947c478bd9Sstevel@tonic-gate {
13957c478bd9Sstevel@tonic-gate 	pid_t pid = provider->ftp_pid;
13967c478bd9Sstevel@tonic-gate 	proc_t *p;
13977c478bd9Sstevel@tonic-gate 
13987c478bd9Sstevel@tonic-gate 	/*
13997c478bd9Sstevel@tonic-gate 	 * There need to be no consumers using this provider and no
14007c478bd9Sstevel@tonic-gate 	 * associated enabled probes.
14017c478bd9Sstevel@tonic-gate 	 */
14027c478bd9Sstevel@tonic-gate 	ASSERT(provider->ftp_ccount == 0);
14037c478bd9Sstevel@tonic-gate 	ASSERT(provider->ftp_rcount == 0);
14047c478bd9Sstevel@tonic-gate 
1405*b096140dSahl 	fasttrap_proc_release(provider->ftp_proc);
1406*b096140dSahl 
14077c478bd9Sstevel@tonic-gate 	kmem_free(provider, sizeof (fasttrap_provider_t));
14087c478bd9Sstevel@tonic-gate 
14097c478bd9Sstevel@tonic-gate 	/*
14107c478bd9Sstevel@tonic-gate 	 * Decrement p_dtrace_probes on the process whose provider we're
14117c478bd9Sstevel@tonic-gate 	 * freeing. We don't have to worry about clobbering somone else's
14127c478bd9Sstevel@tonic-gate 	 * modifications to it because we have locked the bucket that
14137c478bd9Sstevel@tonic-gate 	 * corresponds to this process's hash chain in the provider hash
14147c478bd9Sstevel@tonic-gate 	 * table. Don't sweat it if we can't find the process.
14157c478bd9Sstevel@tonic-gate 	 */
14167c478bd9Sstevel@tonic-gate 	mutex_enter(&pidlock);
14177c478bd9Sstevel@tonic-gate 	if ((p = prfind(pid)) == NULL) {
14187c478bd9Sstevel@tonic-gate 		mutex_exit(&pidlock);
14197c478bd9Sstevel@tonic-gate 		return;
14207c478bd9Sstevel@tonic-gate 	}
14217c478bd9Sstevel@tonic-gate 
14227c478bd9Sstevel@tonic-gate 	mutex_enter(&p->p_lock);
14237c478bd9Sstevel@tonic-gate 	mutex_exit(&pidlock);
14247c478bd9Sstevel@tonic-gate 
14257c478bd9Sstevel@tonic-gate 	p->p_dtrace_probes--;
14267c478bd9Sstevel@tonic-gate 	mutex_exit(&p->p_lock);
14277c478bd9Sstevel@tonic-gate }
14287c478bd9Sstevel@tonic-gate 
14297c478bd9Sstevel@tonic-gate static void
14307c478bd9Sstevel@tonic-gate fasttrap_provider_retire(fasttrap_provider_t *provider)
14317c478bd9Sstevel@tonic-gate {
14327c478bd9Sstevel@tonic-gate 	dtrace_provider_id_t provid = provider->ftp_provid;
14337c478bd9Sstevel@tonic-gate 
14347c478bd9Sstevel@tonic-gate 	/*
1435*b096140dSahl 	 * Mark the provider to be removed in our post-processing step,
1436*b096140dSahl 	 * mark it retired, and mark its proc as defunct (though it may
1437*b096140dSahl 	 * already be marked defunct by another provider that shares the
1438*b096140dSahl 	 * same proc). Marking it indicates that we should try to remove it;
1439*b096140dSahl 	 * setting the retired flag indicates that we're done with this
1440*b096140dSahl 	 * provider; setting the proc to be defunct indicates that all
1441*b096140dSahl 	 * tracepoints associated with the traced process should be ignored.
14427c478bd9Sstevel@tonic-gate 	 */
1443*b096140dSahl 	provider->ftp_proc->ftpc_defunct = 1;
1444*b096140dSahl 	provider->ftp_retired = 1;
14457c478bd9Sstevel@tonic-gate 	provider->ftp_marked = 1;
14467c478bd9Sstevel@tonic-gate 	mutex_exit(&provider->ftp_mtx);
14477c478bd9Sstevel@tonic-gate 
14487c478bd9Sstevel@tonic-gate 	/*
14497c478bd9Sstevel@tonic-gate 	 * We don't have to worry about invalidating the same provider twice
14507c478bd9Sstevel@tonic-gate 	 * since fasttrap_provider_lookup() will ignore provider that have
1451*b096140dSahl 	 * been marked as retired.
14527c478bd9Sstevel@tonic-gate 	 */
14537c478bd9Sstevel@tonic-gate 	dtrace_invalidate(provid);
14547c478bd9Sstevel@tonic-gate 
14557c478bd9Sstevel@tonic-gate 	fasttrap_pid_cleanup();
14567c478bd9Sstevel@tonic-gate }
14577c478bd9Sstevel@tonic-gate 
14587c478bd9Sstevel@tonic-gate static int
14597c478bd9Sstevel@tonic-gate fasttrap_add_probe(fasttrap_probe_spec_t *pdata)
14607c478bd9Sstevel@tonic-gate {
14617c478bd9Sstevel@tonic-gate 	fasttrap_provider_t *provider;
14627c478bd9Sstevel@tonic-gate 	fasttrap_probe_t *pp;
14637c478bd9Sstevel@tonic-gate 	fasttrap_tracepoint_t *tp;
14647c478bd9Sstevel@tonic-gate 	char *name;
14657c478bd9Sstevel@tonic-gate 	size_t size;
14667c478bd9Sstevel@tonic-gate 	int i, aframes, whack;
14677c478bd9Sstevel@tonic-gate 
14687c478bd9Sstevel@tonic-gate 	switch (pdata->ftps_type) {
14697c478bd9Sstevel@tonic-gate 	case DTFTP_ENTRY:
14707c478bd9Sstevel@tonic-gate 		name = "entry";
14717c478bd9Sstevel@tonic-gate 		aframes = FASTTRAP_ENTRY_AFRAMES;
14727c478bd9Sstevel@tonic-gate 		break;
14737c478bd9Sstevel@tonic-gate 	case DTFTP_RETURN:
14747c478bd9Sstevel@tonic-gate 		name = "return";
14757c478bd9Sstevel@tonic-gate 		aframes = FASTTRAP_RETURN_AFRAMES;
14767c478bd9Sstevel@tonic-gate 		break;
14777c478bd9Sstevel@tonic-gate 	case DTFTP_OFFSETS:
14787c478bd9Sstevel@tonic-gate 		name = NULL;
14797c478bd9Sstevel@tonic-gate 		break;
14807c478bd9Sstevel@tonic-gate 	default:
14817c478bd9Sstevel@tonic-gate 		return (EINVAL);
14827c478bd9Sstevel@tonic-gate 	}
14837c478bd9Sstevel@tonic-gate 
14847c478bd9Sstevel@tonic-gate 	if ((provider = fasttrap_provider_lookup(pdata->ftps_pid,
14857c478bd9Sstevel@tonic-gate 	    FASTTRAP_PID_NAME, &pid_attr)) == NULL)
14867c478bd9Sstevel@tonic-gate 		return (ESRCH);
14877c478bd9Sstevel@tonic-gate 
14887c478bd9Sstevel@tonic-gate 	/*
14897c478bd9Sstevel@tonic-gate 	 * Increment this reference count to indicate that a consumer is
14907c478bd9Sstevel@tonic-gate 	 * actively adding a new probe associated with this provider.
14917c478bd9Sstevel@tonic-gate 	 */
14927c478bd9Sstevel@tonic-gate 	provider->ftp_ccount++;
14937c478bd9Sstevel@tonic-gate 	mutex_exit(&provider->ftp_mtx);
14947c478bd9Sstevel@tonic-gate 
14957c478bd9Sstevel@tonic-gate 	if (name != NULL) {
14967c478bd9Sstevel@tonic-gate 		if (dtrace_probe_lookup(provider->ftp_provid,
14977c478bd9Sstevel@tonic-gate 		    pdata->ftps_mod, pdata->ftps_func, name) != 0)
14987c478bd9Sstevel@tonic-gate 			goto done;
14997c478bd9Sstevel@tonic-gate 
15007c478bd9Sstevel@tonic-gate 		atomic_add_32(&fasttrap_total, pdata->ftps_noffs);
15017c478bd9Sstevel@tonic-gate 
15027c478bd9Sstevel@tonic-gate 		if (fasttrap_total > fasttrap_max) {
15037c478bd9Sstevel@tonic-gate 			atomic_add_32(&fasttrap_total, -pdata->ftps_noffs);
15047c478bd9Sstevel@tonic-gate 			goto no_mem;
15057c478bd9Sstevel@tonic-gate 		}
15067c478bd9Sstevel@tonic-gate 
15077c478bd9Sstevel@tonic-gate 		ASSERT(pdata->ftps_noffs > 0);
15087c478bd9Sstevel@tonic-gate 		size = sizeof (fasttrap_probe_t) +
15097c478bd9Sstevel@tonic-gate 		    sizeof (pp->ftp_tps[0]) * (pdata->ftps_noffs - 1);
15107c478bd9Sstevel@tonic-gate 
15117c478bd9Sstevel@tonic-gate 		pp = kmem_zalloc(size, KM_SLEEP);
15127c478bd9Sstevel@tonic-gate 
15137c478bd9Sstevel@tonic-gate 		pp->ftp_prov = provider;
15147c478bd9Sstevel@tonic-gate 		pp->ftp_faddr = pdata->ftps_pc;
15157c478bd9Sstevel@tonic-gate 		pp->ftp_fsize = pdata->ftps_size;
15167c478bd9Sstevel@tonic-gate 		pp->ftp_pid = pdata->ftps_pid;
15177c478bd9Sstevel@tonic-gate 		pp->ftp_ntps = pdata->ftps_noffs;
15187c478bd9Sstevel@tonic-gate 		pp->ftp_type = pdata->ftps_type;
15197c478bd9Sstevel@tonic-gate 
15207c478bd9Sstevel@tonic-gate 		for (i = 0; i < pdata->ftps_noffs; i++) {
15217c478bd9Sstevel@tonic-gate 			tp = kmem_zalloc(sizeof (fasttrap_tracepoint_t),
15227c478bd9Sstevel@tonic-gate 			    KM_SLEEP);
15237c478bd9Sstevel@tonic-gate 
1524*b096140dSahl 			tp->ftt_proc = provider->ftp_proc;
15257c478bd9Sstevel@tonic-gate 			tp->ftt_pc = pdata->ftps_offs[i] + pdata->ftps_pc;
15267c478bd9Sstevel@tonic-gate 			tp->ftt_pid = pdata->ftps_pid;
15277c478bd9Sstevel@tonic-gate 
15287c478bd9Sstevel@tonic-gate 			pp->ftp_tps[i].fit_tp = tp;
15297c478bd9Sstevel@tonic-gate 			pp->ftp_tps[i].fit_id.fti_probe = pp;
15307c478bd9Sstevel@tonic-gate 		}
15317c478bd9Sstevel@tonic-gate 
15327c478bd9Sstevel@tonic-gate 		pp->ftp_id = dtrace_probe_create(provider->ftp_provid,
15337c478bd9Sstevel@tonic-gate 		    pdata->ftps_mod, pdata->ftps_func, name, aframes, pp);
15347c478bd9Sstevel@tonic-gate 	} else {
15357c478bd9Sstevel@tonic-gate 		for (i = 0; i < pdata->ftps_noffs; i++) {
15367c478bd9Sstevel@tonic-gate 			char name_str[17];
15377c478bd9Sstevel@tonic-gate 
15387c478bd9Sstevel@tonic-gate 			(void) sprintf(name_str, "%llx",
15397c478bd9Sstevel@tonic-gate 			    (unsigned long long)pdata->ftps_offs[i]);
15407c478bd9Sstevel@tonic-gate 
15417c478bd9Sstevel@tonic-gate 			if (dtrace_probe_lookup(provider->ftp_provid,
15427c478bd9Sstevel@tonic-gate 			    pdata->ftps_mod, pdata->ftps_func, name_str) != 0)
15437c478bd9Sstevel@tonic-gate 				continue;
15447c478bd9Sstevel@tonic-gate 
15457c478bd9Sstevel@tonic-gate 			atomic_add_32(&fasttrap_total, 1);
15467c478bd9Sstevel@tonic-gate 
15477c478bd9Sstevel@tonic-gate 			if (fasttrap_total > fasttrap_max) {
15487c478bd9Sstevel@tonic-gate 				atomic_add_32(&fasttrap_total, -1);
15497c478bd9Sstevel@tonic-gate 				goto no_mem;
15507c478bd9Sstevel@tonic-gate 			}
15517c478bd9Sstevel@tonic-gate 
15527c478bd9Sstevel@tonic-gate 			pp = kmem_zalloc(sizeof (fasttrap_probe_t), KM_SLEEP);
15537c478bd9Sstevel@tonic-gate 
15547c478bd9Sstevel@tonic-gate 			pp->ftp_prov = provider;
15557c478bd9Sstevel@tonic-gate 			pp->ftp_faddr = pdata->ftps_pc;
15567c478bd9Sstevel@tonic-gate 			pp->ftp_fsize = pdata->ftps_size;
15577c478bd9Sstevel@tonic-gate 			pp->ftp_pid = pdata->ftps_pid;
15587c478bd9Sstevel@tonic-gate 			pp->ftp_ntps = 1;
15597c478bd9Sstevel@tonic-gate 			pp->ftp_type = pdata->ftps_type;
15607c478bd9Sstevel@tonic-gate 
15617c478bd9Sstevel@tonic-gate 			tp = kmem_zalloc(sizeof (fasttrap_tracepoint_t),
15627c478bd9Sstevel@tonic-gate 			    KM_SLEEP);
15637c478bd9Sstevel@tonic-gate 
1564*b096140dSahl 			tp->ftt_proc = provider->ftp_proc;
15657c478bd9Sstevel@tonic-gate 			tp->ftt_pc = pdata->ftps_offs[i] + pdata->ftps_pc;
15667c478bd9Sstevel@tonic-gate 			tp->ftt_pid = pdata->ftps_pid;
15677c478bd9Sstevel@tonic-gate 
15687c478bd9Sstevel@tonic-gate 			pp->ftp_tps[0].fit_tp = tp;
15697c478bd9Sstevel@tonic-gate 			pp->ftp_tps[0].fit_id.fti_probe = pp;
15707c478bd9Sstevel@tonic-gate 
15717c478bd9Sstevel@tonic-gate 			pp->ftp_id = dtrace_probe_create(provider->ftp_provid,
15727c478bd9Sstevel@tonic-gate 			    pdata->ftps_mod, pdata->ftps_func, name_str,
15737c478bd9Sstevel@tonic-gate 			    FASTTRAP_OFFSET_AFRAMES, pp);
15747c478bd9Sstevel@tonic-gate 		}
15757c478bd9Sstevel@tonic-gate 	}
15767c478bd9Sstevel@tonic-gate 
15777c478bd9Sstevel@tonic-gate done:
15787c478bd9Sstevel@tonic-gate 	/*
15797c478bd9Sstevel@tonic-gate 	 * We know that the provider is still valid since we incremented the
15807c478bd9Sstevel@tonic-gate 	 * reference count. If someone tried to free this provider while we
15817c478bd9Sstevel@tonic-gate 	 * were using it (e.g. because the process called exec(2) or exit(2)),
15827c478bd9Sstevel@tonic-gate 	 * take note of that and try to free it now.
15837c478bd9Sstevel@tonic-gate 	 */
15847c478bd9Sstevel@tonic-gate 	mutex_enter(&provider->ftp_mtx);
15857c478bd9Sstevel@tonic-gate 	provider->ftp_ccount--;
1586*b096140dSahl 	whack = provider->ftp_retired;
15877c478bd9Sstevel@tonic-gate 	mutex_exit(&provider->ftp_mtx);
15887c478bd9Sstevel@tonic-gate 
15897c478bd9Sstevel@tonic-gate 	if (whack)
15907c478bd9Sstevel@tonic-gate 		fasttrap_pid_cleanup();
15917c478bd9Sstevel@tonic-gate 
15927c478bd9Sstevel@tonic-gate 	return (0);
15937c478bd9Sstevel@tonic-gate 
15947c478bd9Sstevel@tonic-gate no_mem:
15957c478bd9Sstevel@tonic-gate 	/*
15967c478bd9Sstevel@tonic-gate 	 * If we've exhausted the allowable resources, we'll try to remove
15977c478bd9Sstevel@tonic-gate 	 * this provider to free some up. This is to cover the case where
15987c478bd9Sstevel@tonic-gate 	 * the user has accidentally created many more probes than was
15997c478bd9Sstevel@tonic-gate 	 * intended (e.g. pid123:::).
16007c478bd9Sstevel@tonic-gate 	 */
16017c478bd9Sstevel@tonic-gate 	mutex_enter(&provider->ftp_mtx);
16027c478bd9Sstevel@tonic-gate 	provider->ftp_ccount--;
16037c478bd9Sstevel@tonic-gate 	provider->ftp_marked = 1;
16047c478bd9Sstevel@tonic-gate 	mutex_exit(&provider->ftp_mtx);
16057c478bd9Sstevel@tonic-gate 
16067c478bd9Sstevel@tonic-gate 	fasttrap_pid_cleanup();
16077c478bd9Sstevel@tonic-gate 
16087c478bd9Sstevel@tonic-gate 	return (ENOMEM);
16097c478bd9Sstevel@tonic-gate }
16107c478bd9Sstevel@tonic-gate 
16117c478bd9Sstevel@tonic-gate /*ARGSUSED*/
16127c478bd9Sstevel@tonic-gate static void *
16137c478bd9Sstevel@tonic-gate fasttrap_meta_provide(void *arg, dtrace_helper_provdesc_t *dhpv, pid_t pid)
16147c478bd9Sstevel@tonic-gate {
16157c478bd9Sstevel@tonic-gate 	fasttrap_provider_t *provider;
16167c478bd9Sstevel@tonic-gate 
16177c478bd9Sstevel@tonic-gate 	/*
16187c478bd9Sstevel@tonic-gate 	 * A 32-bit unsigned integer (like a pid for example) can be
16197c478bd9Sstevel@tonic-gate 	 * expressed in 10 or fewer decimal digits. Make sure that we'll
16207c478bd9Sstevel@tonic-gate 	 * have enough space for the provider name.
16217c478bd9Sstevel@tonic-gate 	 */
16227c478bd9Sstevel@tonic-gate 	if (strlen(dhpv->dthpv_provname) + 10 >=
16237c478bd9Sstevel@tonic-gate 	    sizeof (provider->ftp_name)) {
16247c478bd9Sstevel@tonic-gate 		cmn_err(CE_WARN, "failed to instantiate provider %s: "
16257c478bd9Sstevel@tonic-gate 		    "name too long to accomodate pid", dhpv->dthpv_provname);
16267c478bd9Sstevel@tonic-gate 		return (NULL);
16277c478bd9Sstevel@tonic-gate 	}
16287c478bd9Sstevel@tonic-gate 
16297c478bd9Sstevel@tonic-gate 	/*
16307c478bd9Sstevel@tonic-gate 	 * Don't let folks spoof the true pid provider.
16317c478bd9Sstevel@tonic-gate 	 */
16327c478bd9Sstevel@tonic-gate 	if (strcmp(dhpv->dthpv_provname, FASTTRAP_PID_NAME) == 0) {
16337c478bd9Sstevel@tonic-gate 		cmn_err(CE_WARN, "failed to instantiate provider %s: "
16347c478bd9Sstevel@tonic-gate 		    "%s is an invalid name", dhpv->dthpv_provname,
16357c478bd9Sstevel@tonic-gate 		    FASTTRAP_PID_NAME);
16367c478bd9Sstevel@tonic-gate 		return (NULL);
16377c478bd9Sstevel@tonic-gate 	}
16387c478bd9Sstevel@tonic-gate 
16397c478bd9Sstevel@tonic-gate 	/*
16407c478bd9Sstevel@tonic-gate 	 * The highest stability class that fasttrap supports is ISA; cap
16417c478bd9Sstevel@tonic-gate 	 * the stability of the new provider accordingly.
16427c478bd9Sstevel@tonic-gate 	 */
16437c478bd9Sstevel@tonic-gate 	if (dhpv->dthpv_pattr.dtpa_provider.dtat_class >= DTRACE_CLASS_COMMON)
16447c478bd9Sstevel@tonic-gate 		dhpv->dthpv_pattr.dtpa_provider.dtat_class = DTRACE_CLASS_ISA;
16457c478bd9Sstevel@tonic-gate 	if (dhpv->dthpv_pattr.dtpa_mod.dtat_class >= DTRACE_CLASS_COMMON)
16467c478bd9Sstevel@tonic-gate 		dhpv->dthpv_pattr.dtpa_mod.dtat_class = DTRACE_CLASS_ISA;
16477c478bd9Sstevel@tonic-gate 	if (dhpv->dthpv_pattr.dtpa_func.dtat_class >= DTRACE_CLASS_COMMON)
16487c478bd9Sstevel@tonic-gate 		dhpv->dthpv_pattr.dtpa_func.dtat_class = DTRACE_CLASS_ISA;
16497c478bd9Sstevel@tonic-gate 	if (dhpv->dthpv_pattr.dtpa_name.dtat_class >= DTRACE_CLASS_COMMON)
16507c478bd9Sstevel@tonic-gate 		dhpv->dthpv_pattr.dtpa_name.dtat_class = DTRACE_CLASS_ISA;
16517c478bd9Sstevel@tonic-gate 	if (dhpv->dthpv_pattr.dtpa_args.dtat_class >= DTRACE_CLASS_COMMON)
16527c478bd9Sstevel@tonic-gate 		dhpv->dthpv_pattr.dtpa_args.dtat_class = DTRACE_CLASS_ISA;
16537c478bd9Sstevel@tonic-gate 
16547c478bd9Sstevel@tonic-gate 	if ((provider = fasttrap_provider_lookup(pid, dhpv->dthpv_provname,
16557c478bd9Sstevel@tonic-gate 	    &dhpv->dthpv_pattr)) == NULL) {
16567c478bd9Sstevel@tonic-gate 		cmn_err(CE_WARN, "failed to instantiate provider %s for "
16577c478bd9Sstevel@tonic-gate 		    "process %u",  dhpv->dthpv_provname, (uint_t)pid);
16587c478bd9Sstevel@tonic-gate 		return (NULL);
16597c478bd9Sstevel@tonic-gate 	}
16607c478bd9Sstevel@tonic-gate 
16617c478bd9Sstevel@tonic-gate 	/*
16627c478bd9Sstevel@tonic-gate 	 * We elevate the consumer count here to ensure that this provider
16637c478bd9Sstevel@tonic-gate 	 * isn't removed until after the meta provider has been told to
16647c478bd9Sstevel@tonic-gate 	 * remove it.
16657c478bd9Sstevel@tonic-gate 	 */
16667c478bd9Sstevel@tonic-gate 	provider->ftp_ccount++;
16677c478bd9Sstevel@tonic-gate 
16687c478bd9Sstevel@tonic-gate 	mutex_exit(&provider->ftp_mtx);
16697c478bd9Sstevel@tonic-gate 
16707c478bd9Sstevel@tonic-gate 	return (provider);
16717c478bd9Sstevel@tonic-gate }
16727c478bd9Sstevel@tonic-gate 
16737c478bd9Sstevel@tonic-gate /*ARGSUSED*/
16747c478bd9Sstevel@tonic-gate static void
16757c478bd9Sstevel@tonic-gate fasttrap_meta_create_probe(void *arg, void *parg,
16767c478bd9Sstevel@tonic-gate     dtrace_helper_probedesc_t *dhpb)
16777c478bd9Sstevel@tonic-gate {
16787c478bd9Sstevel@tonic-gate 	fasttrap_provider_t *provider = parg;
16797c478bd9Sstevel@tonic-gate 	fasttrap_probe_t *pp;
16807c478bd9Sstevel@tonic-gate 	fasttrap_tracepoint_t *tp;
16817c478bd9Sstevel@tonic-gate 	size_t size;
16827c478bd9Sstevel@tonic-gate 	int i;
16837c478bd9Sstevel@tonic-gate 
16847c478bd9Sstevel@tonic-gate 	mutex_enter(&provider->ftp_mtx);
16857c478bd9Sstevel@tonic-gate 
16867c478bd9Sstevel@tonic-gate 	if (dtrace_probe_lookup(provider->ftp_provid, dhpb->dthpb_mod,
16877c478bd9Sstevel@tonic-gate 	    dhpb->dthpb_func, dhpb->dthpb_name) != 0) {
16887c478bd9Sstevel@tonic-gate 		mutex_exit(&provider->ftp_mtx);
16897c478bd9Sstevel@tonic-gate 		return;
16907c478bd9Sstevel@tonic-gate 	}
16917c478bd9Sstevel@tonic-gate 
16927c478bd9Sstevel@tonic-gate 	atomic_add_32(&fasttrap_total, dhpb->dthpb_noffs);
16937c478bd9Sstevel@tonic-gate 
16947c478bd9Sstevel@tonic-gate 	if (fasttrap_total > fasttrap_max) {
16957c478bd9Sstevel@tonic-gate 		atomic_add_32(&fasttrap_total, -dhpb->dthpb_noffs);
16967c478bd9Sstevel@tonic-gate 		mutex_exit(&provider->ftp_mtx);
16977c478bd9Sstevel@tonic-gate 		return;
16987c478bd9Sstevel@tonic-gate 	}
16997c478bd9Sstevel@tonic-gate 
17007c478bd9Sstevel@tonic-gate 	size = sizeof (fasttrap_probe_t) +
17017c478bd9Sstevel@tonic-gate 	    sizeof (pp->ftp_tps[0]) * (dhpb->dthpb_noffs - 1);
17027c478bd9Sstevel@tonic-gate 	pp = kmem_zalloc(size, KM_SLEEP);
17037c478bd9Sstevel@tonic-gate 
17047c478bd9Sstevel@tonic-gate 	pp->ftp_prov = provider;
17057c478bd9Sstevel@tonic-gate 	pp->ftp_pid = provider->ftp_pid;
17067c478bd9Sstevel@tonic-gate 	pp->ftp_ntps = dhpb->dthpb_noffs;
17077c478bd9Sstevel@tonic-gate #ifdef __sparc
17087c478bd9Sstevel@tonic-gate 	pp->ftp_type = DTFTP_POST_OFFSETS;
17097c478bd9Sstevel@tonic-gate #else
17107c478bd9Sstevel@tonic-gate 	pp->ftp_type = DTFTP_OFFSETS;
17117c478bd9Sstevel@tonic-gate #endif
17127c478bd9Sstevel@tonic-gate 	pp->ftp_nargs = dhpb->dthpb_xargc;
17137c478bd9Sstevel@tonic-gate 	pp->ftp_xtypes = dhpb->dthpb_xtypes;
17147c478bd9Sstevel@tonic-gate 	pp->ftp_ntypes = dhpb->dthpb_ntypes;
17157c478bd9Sstevel@tonic-gate 
17167c478bd9Sstevel@tonic-gate 	for (i = 0; i < pp->ftp_ntps; i++) {
17177c478bd9Sstevel@tonic-gate 		tp = kmem_zalloc(sizeof (fasttrap_tracepoint_t), KM_SLEEP);
17187c478bd9Sstevel@tonic-gate 
1719*b096140dSahl 		tp->ftt_proc = provider->ftp_proc;
17207c478bd9Sstevel@tonic-gate 		tp->ftt_pc = dhpb->dthpb_base + dhpb->dthpb_offs[i];
17217c478bd9Sstevel@tonic-gate 		tp->ftt_pid = provider->ftp_pid;
17227c478bd9Sstevel@tonic-gate 
17237c478bd9Sstevel@tonic-gate 		pp->ftp_tps[i].fit_tp = tp;
17247c478bd9Sstevel@tonic-gate 		pp->ftp_tps[i].fit_id.fti_probe = pp;
17257c478bd9Sstevel@tonic-gate 	}
17267c478bd9Sstevel@tonic-gate 
17277c478bd9Sstevel@tonic-gate 	/*
17287c478bd9Sstevel@tonic-gate 	 * If the arguments are shuffled around we set the argument remapping
17297c478bd9Sstevel@tonic-gate 	 * table. Later, when the probe fires, we only remap the arguments
17307c478bd9Sstevel@tonic-gate 	 * if the table is non-NULL.
17317c478bd9Sstevel@tonic-gate 	 */
17327c478bd9Sstevel@tonic-gate 	for (i = 0; i < dhpb->dthpb_xargc; i++) {
17337c478bd9Sstevel@tonic-gate 		if (dhpb->dthpb_args[i] != i) {
17347c478bd9Sstevel@tonic-gate 			pp->ftp_argmap = dhpb->dthpb_args;
17357c478bd9Sstevel@tonic-gate 			break;
17367c478bd9Sstevel@tonic-gate 		}
17377c478bd9Sstevel@tonic-gate 	}
17387c478bd9Sstevel@tonic-gate 
17397c478bd9Sstevel@tonic-gate 	/*
17407c478bd9Sstevel@tonic-gate 	 * The probe is fully constructed -- register it with DTrace.
17417c478bd9Sstevel@tonic-gate 	 */
17427c478bd9Sstevel@tonic-gate 	pp->ftp_id = dtrace_probe_create(provider->ftp_provid, dhpb->dthpb_mod,
17437c478bd9Sstevel@tonic-gate 	    dhpb->dthpb_func, dhpb->dthpb_name, FASTTRAP_OFFSET_AFRAMES, pp);
17447c478bd9Sstevel@tonic-gate 
17457c478bd9Sstevel@tonic-gate 	mutex_exit(&provider->ftp_mtx);
17467c478bd9Sstevel@tonic-gate }
17477c478bd9Sstevel@tonic-gate 
17487c478bd9Sstevel@tonic-gate /*ARGSUSED*/
17497c478bd9Sstevel@tonic-gate static void
17507c478bd9Sstevel@tonic-gate fasttrap_meta_remove(void *arg, dtrace_helper_provdesc_t *dhpv, pid_t pid)
17517c478bd9Sstevel@tonic-gate {
17527c478bd9Sstevel@tonic-gate 	fasttrap_provider_t *provider;
17537c478bd9Sstevel@tonic-gate 
17547c478bd9Sstevel@tonic-gate 	if ((provider = fasttrap_provider_lookup(pid,
17557c478bd9Sstevel@tonic-gate 	    dhpv->dthpv_provname, NULL)) != NULL) {
17567c478bd9Sstevel@tonic-gate 		/*
17577c478bd9Sstevel@tonic-gate 		 * Drop the consumer count now that we're done with this
17587c478bd9Sstevel@tonic-gate 		 * provider. If there are no other consumers retire it now.
17597c478bd9Sstevel@tonic-gate 		 */
17607c478bd9Sstevel@tonic-gate 		if (--provider->ftp_ccount == 0)
17617c478bd9Sstevel@tonic-gate 			fasttrap_provider_retire(provider);
17627c478bd9Sstevel@tonic-gate 		else
17637c478bd9Sstevel@tonic-gate 			mutex_exit(&provider->ftp_mtx);
17647c478bd9Sstevel@tonic-gate 	}
17657c478bd9Sstevel@tonic-gate }
17667c478bd9Sstevel@tonic-gate 
17677c478bd9Sstevel@tonic-gate static dtrace_mops_t fasttrap_mops = {
17687c478bd9Sstevel@tonic-gate 	fasttrap_meta_create_probe,
17697c478bd9Sstevel@tonic-gate 	fasttrap_meta_provide,
17707c478bd9Sstevel@tonic-gate 	fasttrap_meta_remove
17717c478bd9Sstevel@tonic-gate };
17727c478bd9Sstevel@tonic-gate 
17737c478bd9Sstevel@tonic-gate /*ARGSUSED*/
17747c478bd9Sstevel@tonic-gate static int
17757c478bd9Sstevel@tonic-gate fasttrap_open(dev_t *devp, int flag, int otyp, cred_t *cred_p)
17767c478bd9Sstevel@tonic-gate {
17777c478bd9Sstevel@tonic-gate 	return (0);
17787c478bd9Sstevel@tonic-gate }
17797c478bd9Sstevel@tonic-gate 
17807c478bd9Sstevel@tonic-gate /*ARGSUSED*/
17817c478bd9Sstevel@tonic-gate static int
17827c478bd9Sstevel@tonic-gate fasttrap_ioctl(dev_t dev, int cmd, intptr_t arg, int md, cred_t *cr, int *rv)
17837c478bd9Sstevel@tonic-gate {
17847c478bd9Sstevel@tonic-gate 	if (!dtrace_attached())
17857c478bd9Sstevel@tonic-gate 		return (EAGAIN);
17867c478bd9Sstevel@tonic-gate 
17877c478bd9Sstevel@tonic-gate 	if (cmd == FASTTRAPIOC_MAKEPROBE) {
17887c478bd9Sstevel@tonic-gate 		fasttrap_probe_spec_t *uprobe = (void *)arg;
17897c478bd9Sstevel@tonic-gate 		fasttrap_probe_spec_t *probe;
17907c478bd9Sstevel@tonic-gate 		uint64_t noffs;
17917c478bd9Sstevel@tonic-gate 		size_t size;
17927c478bd9Sstevel@tonic-gate 		int ret;
17937c478bd9Sstevel@tonic-gate 		char *c;
17947c478bd9Sstevel@tonic-gate 
17957c478bd9Sstevel@tonic-gate 		if (copyin(&uprobe->ftps_noffs, &noffs,
17967c478bd9Sstevel@tonic-gate 		    sizeof (uprobe->ftps_noffs)))
17977c478bd9Sstevel@tonic-gate 			return (EFAULT);
17987c478bd9Sstevel@tonic-gate 
17997c478bd9Sstevel@tonic-gate 		/*
18007c478bd9Sstevel@tonic-gate 		 * Probes must have at least one tracepoint.
18017c478bd9Sstevel@tonic-gate 		 */
18027c478bd9Sstevel@tonic-gate 		if (noffs == 0)
18037c478bd9Sstevel@tonic-gate 			return (EINVAL);
18047c478bd9Sstevel@tonic-gate 
18057c478bd9Sstevel@tonic-gate 		size = sizeof (fasttrap_probe_spec_t) +
18067c478bd9Sstevel@tonic-gate 		    sizeof (probe->ftps_offs[0]) * (noffs - 1);
18077c478bd9Sstevel@tonic-gate 
18087c478bd9Sstevel@tonic-gate 		if (size > 1024 * 1024)
18097c478bd9Sstevel@tonic-gate 			return (ENOMEM);
18107c478bd9Sstevel@tonic-gate 
18117c478bd9Sstevel@tonic-gate 		probe = kmem_alloc(size, KM_SLEEP);
18127c478bd9Sstevel@tonic-gate 
18137c478bd9Sstevel@tonic-gate 		if (copyin(uprobe, probe, size) != 0) {
18147c478bd9Sstevel@tonic-gate 			kmem_free(probe, size);
18157c478bd9Sstevel@tonic-gate 			return (EFAULT);
18167c478bd9Sstevel@tonic-gate 		}
18177c478bd9Sstevel@tonic-gate 
18187c478bd9Sstevel@tonic-gate 		/*
18197c478bd9Sstevel@tonic-gate 		 * Verify that the function and module strings contain no
18207c478bd9Sstevel@tonic-gate 		 * funny characters.
18217c478bd9Sstevel@tonic-gate 		 */
18227c478bd9Sstevel@tonic-gate 		for (c = &probe->ftps_func[0]; *c != '\0'; c++) {
18237c478bd9Sstevel@tonic-gate 			if (*c < 0x20 || 0x7f <= *c) {
18247c478bd9Sstevel@tonic-gate 				ret = EINVAL;
18257c478bd9Sstevel@tonic-gate 				goto err;
18267c478bd9Sstevel@tonic-gate 			}
18277c478bd9Sstevel@tonic-gate 		}
18287c478bd9Sstevel@tonic-gate 
18297c478bd9Sstevel@tonic-gate 		for (c = &probe->ftps_mod[0]; *c != '\0'; c++) {
18307c478bd9Sstevel@tonic-gate 			if (*c < 0x20 || 0x7f <= *c) {
18317c478bd9Sstevel@tonic-gate 				ret = EINVAL;
18327c478bd9Sstevel@tonic-gate 				goto err;
18337c478bd9Sstevel@tonic-gate 			}
18347c478bd9Sstevel@tonic-gate 		}
18357c478bd9Sstevel@tonic-gate 
18367c478bd9Sstevel@tonic-gate 		if (!PRIV_POLICY_CHOICE(cr, PRIV_ALL, B_FALSE)) {
18377c478bd9Sstevel@tonic-gate 			proc_t *p;
18387c478bd9Sstevel@tonic-gate 			pid_t pid = probe->ftps_pid;
18397c478bd9Sstevel@tonic-gate 
18407c478bd9Sstevel@tonic-gate 			mutex_enter(&pidlock);
18417c478bd9Sstevel@tonic-gate 			/*
18427c478bd9Sstevel@tonic-gate 			 * Report an error if the process doesn't exist
18437c478bd9Sstevel@tonic-gate 			 * or is actively being birthed.
18447c478bd9Sstevel@tonic-gate 			 */
18457c478bd9Sstevel@tonic-gate 			if ((p = prfind(pid)) == NULL || p->p_stat == SIDL) {
18467c478bd9Sstevel@tonic-gate 				mutex_exit(&pidlock);
18477c478bd9Sstevel@tonic-gate 				return (ESRCH);
18487c478bd9Sstevel@tonic-gate 			}
18497c478bd9Sstevel@tonic-gate 			mutex_enter(&p->p_lock);
18507c478bd9Sstevel@tonic-gate 			mutex_exit(&pidlock);
18517c478bd9Sstevel@tonic-gate 
18527c478bd9Sstevel@tonic-gate 			if ((ret = priv_proc_cred_perm(cr, p, NULL,
18537c478bd9Sstevel@tonic-gate 			    VREAD | VWRITE)) != 0) {
18547c478bd9Sstevel@tonic-gate 				mutex_exit(&p->p_lock);
18557c478bd9Sstevel@tonic-gate 				return (ret);
18567c478bd9Sstevel@tonic-gate 			}
18577c478bd9Sstevel@tonic-gate 
18587c478bd9Sstevel@tonic-gate 			mutex_exit(&p->p_lock);
18597c478bd9Sstevel@tonic-gate 		}
18607c478bd9Sstevel@tonic-gate 
18617c478bd9Sstevel@tonic-gate 		ret = fasttrap_add_probe(probe);
18627c478bd9Sstevel@tonic-gate err:
18637c478bd9Sstevel@tonic-gate 		kmem_free(probe, size);
18647c478bd9Sstevel@tonic-gate 
18657c478bd9Sstevel@tonic-gate 		return (ret);
18667c478bd9Sstevel@tonic-gate 
18677c478bd9Sstevel@tonic-gate 	} else if (cmd == FASTTRAPIOC_GETINSTR) {
18687c478bd9Sstevel@tonic-gate 		fasttrap_instr_query_t instr;
18697c478bd9Sstevel@tonic-gate 		fasttrap_tracepoint_t *tp;
18707c478bd9Sstevel@tonic-gate 		uint_t index;
18717c478bd9Sstevel@tonic-gate 		int ret;
18727c478bd9Sstevel@tonic-gate 
18737c478bd9Sstevel@tonic-gate 		if (copyin((void *)arg, &instr, sizeof (instr)) != 0)
18747c478bd9Sstevel@tonic-gate 			return (EFAULT);
18757c478bd9Sstevel@tonic-gate 
18767c478bd9Sstevel@tonic-gate 		if (!PRIV_POLICY_CHOICE(cr, PRIV_ALL, B_FALSE)) {
18777c478bd9Sstevel@tonic-gate 			proc_t *p;
18787c478bd9Sstevel@tonic-gate 			pid_t pid = instr.ftiq_pid;
18797c478bd9Sstevel@tonic-gate 
18807c478bd9Sstevel@tonic-gate 			mutex_enter(&pidlock);
18817c478bd9Sstevel@tonic-gate 			/*
18827c478bd9Sstevel@tonic-gate 			 * Report an error if the process doesn't exist
18837c478bd9Sstevel@tonic-gate 			 * or is actively being birthed.
18847c478bd9Sstevel@tonic-gate 			 */
18857c478bd9Sstevel@tonic-gate 			if ((p = prfind(pid)) == NULL || p->p_stat == SIDL) {
18867c478bd9Sstevel@tonic-gate 				mutex_exit(&pidlock);
18877c478bd9Sstevel@tonic-gate 				return (ESRCH);
18887c478bd9Sstevel@tonic-gate 			}
18897c478bd9Sstevel@tonic-gate 			mutex_enter(&p->p_lock);
18907c478bd9Sstevel@tonic-gate 			mutex_exit(&pidlock);
18917c478bd9Sstevel@tonic-gate 
18927c478bd9Sstevel@tonic-gate 			if ((ret = priv_proc_cred_perm(cr, p, NULL,
18937c478bd9Sstevel@tonic-gate 			    VREAD)) != 0) {
18947c478bd9Sstevel@tonic-gate 				mutex_exit(&p->p_lock);
18957c478bd9Sstevel@tonic-gate 				return (ret);
18967c478bd9Sstevel@tonic-gate 			}
18977c478bd9Sstevel@tonic-gate 
18987c478bd9Sstevel@tonic-gate 			mutex_exit(&p->p_lock);
18997c478bd9Sstevel@tonic-gate 		}
19007c478bd9Sstevel@tonic-gate 
19017c478bd9Sstevel@tonic-gate 		index = FASTTRAP_TPOINTS_INDEX(instr.ftiq_pid, instr.ftiq_pc);
19027c478bd9Sstevel@tonic-gate 
19037c478bd9Sstevel@tonic-gate 		mutex_enter(&fasttrap_tpoints.fth_table[index].ftb_mtx);
19047c478bd9Sstevel@tonic-gate 		tp = fasttrap_tpoints.fth_table[index].ftb_data;
19057c478bd9Sstevel@tonic-gate 		while (tp != NULL) {
19067c478bd9Sstevel@tonic-gate 			if (instr.ftiq_pid == tp->ftt_pid &&
19077c478bd9Sstevel@tonic-gate 			    instr.ftiq_pc == tp->ftt_pc &&
1908*b096140dSahl 			    !tp->ftt_proc->ftpc_defunct)
19097c478bd9Sstevel@tonic-gate 				break;
19107c478bd9Sstevel@tonic-gate 
19117c478bd9Sstevel@tonic-gate 			tp = tp->ftt_next;
19127c478bd9Sstevel@tonic-gate 		}
19137c478bd9Sstevel@tonic-gate 
19147c478bd9Sstevel@tonic-gate 		if (tp == NULL) {
19157c478bd9Sstevel@tonic-gate 			mutex_exit(&fasttrap_tpoints.fth_table[index].ftb_mtx);
19167c478bd9Sstevel@tonic-gate 			return (ENOENT);
19177c478bd9Sstevel@tonic-gate 		}
19187c478bd9Sstevel@tonic-gate 
19197c478bd9Sstevel@tonic-gate 		bcopy(&tp->ftt_instr, &instr.ftiq_instr,
19207c478bd9Sstevel@tonic-gate 		    sizeof (instr.ftiq_instr));
19217c478bd9Sstevel@tonic-gate 		mutex_exit(&fasttrap_tpoints.fth_table[index].ftb_mtx);
19227c478bd9Sstevel@tonic-gate 
19237c478bd9Sstevel@tonic-gate 		if (copyout(&instr, (void *)arg, sizeof (instr)) != 0)
19247c478bd9Sstevel@tonic-gate 			return (EFAULT);
19257c478bd9Sstevel@tonic-gate 
19267c478bd9Sstevel@tonic-gate 		return (0);
19277c478bd9Sstevel@tonic-gate 	}
19287c478bd9Sstevel@tonic-gate 
19297c478bd9Sstevel@tonic-gate 	return (EINVAL);
19307c478bd9Sstevel@tonic-gate }
19317c478bd9Sstevel@tonic-gate 
19327c478bd9Sstevel@tonic-gate static struct cb_ops fasttrap_cb_ops = {
19337c478bd9Sstevel@tonic-gate 	fasttrap_open,		/* open */
19347c478bd9Sstevel@tonic-gate 	nodev,			/* close */
19357c478bd9Sstevel@tonic-gate 	nulldev,		/* strategy */
19367c478bd9Sstevel@tonic-gate 	nulldev,		/* print */
19377c478bd9Sstevel@tonic-gate 	nodev,			/* dump */
19387c478bd9Sstevel@tonic-gate 	nodev,			/* read */
19397c478bd9Sstevel@tonic-gate 	nodev,			/* write */
19407c478bd9Sstevel@tonic-gate 	fasttrap_ioctl,		/* ioctl */
19417c478bd9Sstevel@tonic-gate 	nodev,			/* devmap */
19427c478bd9Sstevel@tonic-gate 	nodev,			/* mmap */
19437c478bd9Sstevel@tonic-gate 	nodev,			/* segmap */
19447c478bd9Sstevel@tonic-gate 	nochpoll,		/* poll */
19457c478bd9Sstevel@tonic-gate 	ddi_prop_op,		/* cb_prop_op */
19467c478bd9Sstevel@tonic-gate 	0,			/* streamtab  */
19477c478bd9Sstevel@tonic-gate 	D_NEW | D_MP		/* Driver compatibility flag */
19487c478bd9Sstevel@tonic-gate };
19497c478bd9Sstevel@tonic-gate 
19507c478bd9Sstevel@tonic-gate /*ARGSUSED*/
19517c478bd9Sstevel@tonic-gate static int
19527c478bd9Sstevel@tonic-gate fasttrap_info(dev_info_t *dip, ddi_info_cmd_t infocmd, void *arg, void **result)
19537c478bd9Sstevel@tonic-gate {
19547c478bd9Sstevel@tonic-gate 	int error;
19557c478bd9Sstevel@tonic-gate 
19567c478bd9Sstevel@tonic-gate 	switch (infocmd) {
19577c478bd9Sstevel@tonic-gate 	case DDI_INFO_DEVT2DEVINFO:
19587c478bd9Sstevel@tonic-gate 		*result = (void *)fasttrap_devi;
19597c478bd9Sstevel@tonic-gate 		error = DDI_SUCCESS;
19607c478bd9Sstevel@tonic-gate 		break;
19617c478bd9Sstevel@tonic-gate 	case DDI_INFO_DEVT2INSTANCE:
19627c478bd9Sstevel@tonic-gate 		*result = (void *)0;
19637c478bd9Sstevel@tonic-gate 		error = DDI_SUCCESS;
19647c478bd9Sstevel@tonic-gate 		break;
19657c478bd9Sstevel@tonic-gate 	default:
19667c478bd9Sstevel@tonic-gate 		error = DDI_FAILURE;
19677c478bd9Sstevel@tonic-gate 	}
19687c478bd9Sstevel@tonic-gate 	return (error);
19697c478bd9Sstevel@tonic-gate }
19707c478bd9Sstevel@tonic-gate 
19717c478bd9Sstevel@tonic-gate static int
19727c478bd9Sstevel@tonic-gate fasttrap_attach(dev_info_t *devi, ddi_attach_cmd_t cmd)
19737c478bd9Sstevel@tonic-gate {
19747c478bd9Sstevel@tonic-gate 	ulong_t nent;
19757c478bd9Sstevel@tonic-gate 
19767c478bd9Sstevel@tonic-gate 	switch (cmd) {
19777c478bd9Sstevel@tonic-gate 	case DDI_ATTACH:
19787c478bd9Sstevel@tonic-gate 		break;
19797c478bd9Sstevel@tonic-gate 	case DDI_RESUME:
19807c478bd9Sstevel@tonic-gate 		return (DDI_SUCCESS);
19817c478bd9Sstevel@tonic-gate 	default:
19827c478bd9Sstevel@tonic-gate 		return (DDI_FAILURE);
19837c478bd9Sstevel@tonic-gate 	}
19847c478bd9Sstevel@tonic-gate 
19857c478bd9Sstevel@tonic-gate 	if (ddi_create_minor_node(devi, "fasttrap", S_IFCHR, 0,
19867c478bd9Sstevel@tonic-gate 	    DDI_PSEUDO, NULL) == DDI_FAILURE ||
19877c478bd9Sstevel@tonic-gate 	    dtrace_register("fasttrap", &fasttrap_attr, DTRACE_PRIV_USER, 0,
19887c478bd9Sstevel@tonic-gate 	    &fasttrap_pops, NULL, &fasttrap_id) != 0) {
19897c478bd9Sstevel@tonic-gate 		ddi_remove_minor_node(devi, NULL);
19907c478bd9Sstevel@tonic-gate 		return (DDI_FAILURE);
19917c478bd9Sstevel@tonic-gate 	}
19927c478bd9Sstevel@tonic-gate 
19937c478bd9Sstevel@tonic-gate 	ddi_report_dev(devi);
19947c478bd9Sstevel@tonic-gate 	fasttrap_devi = devi;
19957c478bd9Sstevel@tonic-gate 
19967c478bd9Sstevel@tonic-gate 	/*
19977c478bd9Sstevel@tonic-gate 	 * Install our hooks into fork(2), exec(2), and exit(2).
19987c478bd9Sstevel@tonic-gate 	 */
19997c478bd9Sstevel@tonic-gate 	dtrace_fasttrap_fork_ptr = &fasttrap_fork;
20007c478bd9Sstevel@tonic-gate 	dtrace_fasttrap_exit_ptr = &fasttrap_exec_exit;
20017c478bd9Sstevel@tonic-gate 	dtrace_fasttrap_exec_ptr = &fasttrap_exec_exit;
20027c478bd9Sstevel@tonic-gate 
20037c478bd9Sstevel@tonic-gate 	fasttrap_max = ddi_getprop(DDI_DEV_T_ANY, devi, DDI_PROP_DONTPASS,
20047c478bd9Sstevel@tonic-gate 	    "fasttrap-max-probes", FASTTRAP_MAX_DEFAULT);
20057c478bd9Sstevel@tonic-gate 	fasttrap_total = 0;
20067c478bd9Sstevel@tonic-gate 
20077c478bd9Sstevel@tonic-gate 	/*
20087c478bd9Sstevel@tonic-gate 	 * Conjure up the tracepoints hashtable...
20097c478bd9Sstevel@tonic-gate 	 */
20107c478bd9Sstevel@tonic-gate 	nent = ddi_getprop(DDI_DEV_T_ANY, devi, DDI_PROP_DONTPASS,
20117c478bd9Sstevel@tonic-gate 	    "fasttrap-hash-size", FASTTRAP_TPOINTS_DEFAULT_SIZE);
20127c478bd9Sstevel@tonic-gate 
20137c478bd9Sstevel@tonic-gate 	if (nent <= 0 || nent > 0x1000000)
20147c478bd9Sstevel@tonic-gate 		nent = FASTTRAP_TPOINTS_DEFAULT_SIZE;
20157c478bd9Sstevel@tonic-gate 
20167c478bd9Sstevel@tonic-gate 	if ((nent & (nent - 1)) == 0)
20177c478bd9Sstevel@tonic-gate 		fasttrap_tpoints.fth_nent = nent;
20187c478bd9Sstevel@tonic-gate 	else
20197c478bd9Sstevel@tonic-gate 		fasttrap_tpoints.fth_nent = 1 << fasttrap_highbit(nent);
20207c478bd9Sstevel@tonic-gate 	ASSERT(fasttrap_tpoints.fth_nent > 0);
20217c478bd9Sstevel@tonic-gate 	fasttrap_tpoints.fth_mask = fasttrap_tpoints.fth_nent - 1;
20227c478bd9Sstevel@tonic-gate 	fasttrap_tpoints.fth_table = kmem_zalloc(fasttrap_tpoints.fth_nent *
20237c478bd9Sstevel@tonic-gate 	    sizeof (fasttrap_bucket_t), KM_SLEEP);
20247c478bd9Sstevel@tonic-gate 
20257c478bd9Sstevel@tonic-gate 	/*
2026*b096140dSahl 	 * ... and the providers hash table...
20277c478bd9Sstevel@tonic-gate 	 */
20287c478bd9Sstevel@tonic-gate 	nent = FASTTRAP_PROVIDERS_DEFAULT_SIZE;
20297c478bd9Sstevel@tonic-gate 	if ((nent & (nent - 1)) == 0)
20307c478bd9Sstevel@tonic-gate 		fasttrap_provs.fth_nent = nent;
20317c478bd9Sstevel@tonic-gate 	else
20327c478bd9Sstevel@tonic-gate 		fasttrap_provs.fth_nent = 1 << fasttrap_highbit(nent);
20337c478bd9Sstevel@tonic-gate 	ASSERT(fasttrap_provs.fth_nent > 0);
20347c478bd9Sstevel@tonic-gate 	fasttrap_provs.fth_mask = fasttrap_provs.fth_nent - 1;
20357c478bd9Sstevel@tonic-gate 	fasttrap_provs.fth_table = kmem_zalloc(fasttrap_provs.fth_nent *
20367c478bd9Sstevel@tonic-gate 	    sizeof (fasttrap_bucket_t), KM_SLEEP);
20377c478bd9Sstevel@tonic-gate 
2038*b096140dSahl 	/*
2039*b096140dSahl 	 * ... and the procs hash table.
2040*b096140dSahl 	 */
2041*b096140dSahl 	nent = FASTTRAP_PROCS_DEFAULT_SIZE;
2042*b096140dSahl 	if ((nent & (nent - 1)) == 0)
2043*b096140dSahl 		fasttrap_procs.fth_nent = nent;
2044*b096140dSahl 	else
2045*b096140dSahl 		fasttrap_procs.fth_nent = 1 << fasttrap_highbit(nent);
2046*b096140dSahl 	ASSERT(fasttrap_procs.fth_nent > 0);
2047*b096140dSahl 	fasttrap_procs.fth_mask = fasttrap_procs.fth_nent - 1;
2048*b096140dSahl 	fasttrap_procs.fth_table = kmem_zalloc(fasttrap_procs.fth_nent *
2049*b096140dSahl 	    sizeof (fasttrap_bucket_t), KM_SLEEP);
2050*b096140dSahl 
20517c478bd9Sstevel@tonic-gate 	(void) dtrace_meta_register("fasttrap", &fasttrap_mops, NULL,
20527c478bd9Sstevel@tonic-gate 	    &fasttrap_meta_id);
20537c478bd9Sstevel@tonic-gate 
20547c478bd9Sstevel@tonic-gate 	return (DDI_SUCCESS);
20557c478bd9Sstevel@tonic-gate }
20567c478bd9Sstevel@tonic-gate 
20577c478bd9Sstevel@tonic-gate static int
20587c478bd9Sstevel@tonic-gate fasttrap_detach(dev_info_t *devi, ddi_detach_cmd_t cmd)
20597c478bd9Sstevel@tonic-gate {
20607c478bd9Sstevel@tonic-gate 	int i, fail = 0;
20617c478bd9Sstevel@tonic-gate 	timeout_id_t tmp;
20627c478bd9Sstevel@tonic-gate 
20637c478bd9Sstevel@tonic-gate 	switch (cmd) {
20647c478bd9Sstevel@tonic-gate 	case DDI_DETACH:
20657c478bd9Sstevel@tonic-gate 		break;
20667c478bd9Sstevel@tonic-gate 	case DDI_SUSPEND:
20677c478bd9Sstevel@tonic-gate 		return (DDI_SUCCESS);
20687c478bd9Sstevel@tonic-gate 	default:
20697c478bd9Sstevel@tonic-gate 		return (DDI_FAILURE);
20707c478bd9Sstevel@tonic-gate 	}
20717c478bd9Sstevel@tonic-gate 
20727c478bd9Sstevel@tonic-gate 	/*
20737c478bd9Sstevel@tonic-gate 	 * Unregister the meta-provider to make sure no new fasttrap-
20747c478bd9Sstevel@tonic-gate 	 * managed providers come along while we're trying to close up
20757c478bd9Sstevel@tonic-gate 	 * shop. If we fail to detach, we'll need to re-register as a
20767c478bd9Sstevel@tonic-gate 	 * meta-provider. We can fail to unregister as a meta-provider
20777c478bd9Sstevel@tonic-gate 	 * if providers we manage still exist.
20787c478bd9Sstevel@tonic-gate 	 */
20797c478bd9Sstevel@tonic-gate 	if (fasttrap_meta_id != DTRACE_METAPROVNONE &&
20807c478bd9Sstevel@tonic-gate 	    dtrace_meta_unregister(fasttrap_meta_id) != 0)
20817c478bd9Sstevel@tonic-gate 		return (DDI_FAILURE);
20827c478bd9Sstevel@tonic-gate 
20837c478bd9Sstevel@tonic-gate 	/*
20847c478bd9Sstevel@tonic-gate 	 * Prevent any new timeouts from running by setting fasttrap_timeout
20857c478bd9Sstevel@tonic-gate 	 * to a non-zero value, and wait for the current timeout to complete.
20867c478bd9Sstevel@tonic-gate 	 */
20877c478bd9Sstevel@tonic-gate 	mutex_enter(&fasttrap_cleanup_mtx);
20887c478bd9Sstevel@tonic-gate 	fasttrap_cleanup_work = 0;
20897c478bd9Sstevel@tonic-gate 
20907c478bd9Sstevel@tonic-gate 	while (fasttrap_timeout != (timeout_id_t)1) {
20917c478bd9Sstevel@tonic-gate 		tmp = fasttrap_timeout;
20927c478bd9Sstevel@tonic-gate 		fasttrap_timeout = (timeout_id_t)1;
20937c478bd9Sstevel@tonic-gate 
20947c478bd9Sstevel@tonic-gate 		if (tmp != 0) {
20957c478bd9Sstevel@tonic-gate 			mutex_exit(&fasttrap_cleanup_mtx);
20967c478bd9Sstevel@tonic-gate 			(void) untimeout(tmp);
20977c478bd9Sstevel@tonic-gate 			mutex_enter(&fasttrap_cleanup_mtx);
20987c478bd9Sstevel@tonic-gate 		}
20997c478bd9Sstevel@tonic-gate 	}
21007c478bd9Sstevel@tonic-gate 
21017c478bd9Sstevel@tonic-gate 	fasttrap_cleanup_work = 0;
21027c478bd9Sstevel@tonic-gate 	mutex_exit(&fasttrap_cleanup_mtx);
21037c478bd9Sstevel@tonic-gate 
21047c478bd9Sstevel@tonic-gate 	/*
21057c478bd9Sstevel@tonic-gate 	 * Iterate over all of our providers. If there's still a process
21067c478bd9Sstevel@tonic-gate 	 * that corresponds to that pid, fail to detach.
21077c478bd9Sstevel@tonic-gate 	 */
21087c478bd9Sstevel@tonic-gate 	for (i = 0; i < fasttrap_provs.fth_nent; i++) {
21097c478bd9Sstevel@tonic-gate 		fasttrap_provider_t **fpp, *fp;
21107c478bd9Sstevel@tonic-gate 		fasttrap_bucket_t *bucket = &fasttrap_provs.fth_table[i];
21117c478bd9Sstevel@tonic-gate 
21127c478bd9Sstevel@tonic-gate 		mutex_enter(&bucket->ftb_mtx);
21137c478bd9Sstevel@tonic-gate 		fpp = (fasttrap_provider_t **)&bucket->ftb_data;
21147c478bd9Sstevel@tonic-gate 		while ((fp = *fpp) != NULL) {
21157c478bd9Sstevel@tonic-gate 			/*
21167c478bd9Sstevel@tonic-gate 			 * Acquire and release the lock as a simple way of
21177c478bd9Sstevel@tonic-gate 			 * waiting for any other consumer to finish with
21187c478bd9Sstevel@tonic-gate 			 * this provider. A thread must first acquire the
21197c478bd9Sstevel@tonic-gate 			 * bucket lock so there's no chance of another thread
21207c478bd9Sstevel@tonic-gate 			 * blocking on the providers lock.
21217c478bd9Sstevel@tonic-gate 			 */
21227c478bd9Sstevel@tonic-gate 			mutex_enter(&fp->ftp_mtx);
21237c478bd9Sstevel@tonic-gate 			mutex_exit(&fp->ftp_mtx);
21247c478bd9Sstevel@tonic-gate 
21257c478bd9Sstevel@tonic-gate 			if (dtrace_unregister(fp->ftp_provid) != 0) {
21267c478bd9Sstevel@tonic-gate 				fail = 1;
21277c478bd9Sstevel@tonic-gate 				fpp = &fp->ftp_next;
21287c478bd9Sstevel@tonic-gate 			} else {
21297c478bd9Sstevel@tonic-gate 				*fpp = fp->ftp_next;
21307c478bd9Sstevel@tonic-gate 				fasttrap_provider_free(fp);
21317c478bd9Sstevel@tonic-gate 			}
21327c478bd9Sstevel@tonic-gate 		}
21337c478bd9Sstevel@tonic-gate 
21347c478bd9Sstevel@tonic-gate 		mutex_exit(&bucket->ftb_mtx);
21357c478bd9Sstevel@tonic-gate 	}
21367c478bd9Sstevel@tonic-gate 
21377c478bd9Sstevel@tonic-gate 	if (fail || dtrace_unregister(fasttrap_id) != 0) {
21387c478bd9Sstevel@tonic-gate 		uint_t work;
21397c478bd9Sstevel@tonic-gate 		/*
21407c478bd9Sstevel@tonic-gate 		 * If we're failing to detach, we need to unblock timeouts
21417c478bd9Sstevel@tonic-gate 		 * and start a new timeout if any work has accumulated while
21427c478bd9Sstevel@tonic-gate 		 * we've been unsuccessfully trying to detach.
21437c478bd9Sstevel@tonic-gate 		 */
21447c478bd9Sstevel@tonic-gate 		mutex_enter(&fasttrap_cleanup_mtx);
21457c478bd9Sstevel@tonic-gate 		fasttrap_timeout = 0;
21467c478bd9Sstevel@tonic-gate 		work = fasttrap_cleanup_work;
21477c478bd9Sstevel@tonic-gate 		mutex_exit(&fasttrap_cleanup_mtx);
21487c478bd9Sstevel@tonic-gate 
21497c478bd9Sstevel@tonic-gate 		if (work)
21507c478bd9Sstevel@tonic-gate 			fasttrap_pid_cleanup();
21517c478bd9Sstevel@tonic-gate 
21527c478bd9Sstevel@tonic-gate 		(void) dtrace_meta_register("fasttrap", &fasttrap_mops, NULL,
21537c478bd9Sstevel@tonic-gate 		    &fasttrap_meta_id);
21547c478bd9Sstevel@tonic-gate 
21557c478bd9Sstevel@tonic-gate 		return (DDI_FAILURE);
21567c478bd9Sstevel@tonic-gate 	}
21577c478bd9Sstevel@tonic-gate 
21587c478bd9Sstevel@tonic-gate #ifdef DEBUG
21597c478bd9Sstevel@tonic-gate 	mutex_enter(&fasttrap_count_mtx);
21607c478bd9Sstevel@tonic-gate 	ASSERT(fasttrap_count == 0);
21617c478bd9Sstevel@tonic-gate 	mutex_exit(&fasttrap_count_mtx);
21627c478bd9Sstevel@tonic-gate #endif
21637c478bd9Sstevel@tonic-gate 
21647c478bd9Sstevel@tonic-gate 	kmem_free(fasttrap_tpoints.fth_table,
21657c478bd9Sstevel@tonic-gate 	    fasttrap_tpoints.fth_nent * sizeof (fasttrap_bucket_t));
21667c478bd9Sstevel@tonic-gate 	fasttrap_tpoints.fth_nent = 0;
21677c478bd9Sstevel@tonic-gate 
21687c478bd9Sstevel@tonic-gate 	kmem_free(fasttrap_provs.fth_table,
21697c478bd9Sstevel@tonic-gate 	    fasttrap_provs.fth_nent * sizeof (fasttrap_bucket_t));
21707c478bd9Sstevel@tonic-gate 	fasttrap_provs.fth_nent = 0;
21717c478bd9Sstevel@tonic-gate 
2172*b096140dSahl 	kmem_free(fasttrap_procs.fth_table,
2173*b096140dSahl 	    fasttrap_procs.fth_nent * sizeof (fasttrap_bucket_t));
2174*b096140dSahl 	fasttrap_procs.fth_nent = 0;
2175*b096140dSahl 
21767c478bd9Sstevel@tonic-gate 	/*
21777c478bd9Sstevel@tonic-gate 	 * We know there are no tracepoints in any process anywhere in
21787c478bd9Sstevel@tonic-gate 	 * the system so there is no process which has its p_dtrace_count
21797c478bd9Sstevel@tonic-gate 	 * greater than zero, therefore we know that no thread can actively
21807c478bd9Sstevel@tonic-gate 	 * be executing code in fasttrap_fork(). Similarly for p_dtrace_probes
21817c478bd9Sstevel@tonic-gate 	 * and fasttrap_exec() and fasttrap_exit().
21827c478bd9Sstevel@tonic-gate 	 */
21837c478bd9Sstevel@tonic-gate 	ASSERT(dtrace_fasttrap_fork_ptr == &fasttrap_fork);
21847c478bd9Sstevel@tonic-gate 	dtrace_fasttrap_fork_ptr = NULL;
21857c478bd9Sstevel@tonic-gate 
21867c478bd9Sstevel@tonic-gate 	ASSERT(dtrace_fasttrap_exec_ptr == &fasttrap_exec_exit);
21877c478bd9Sstevel@tonic-gate 	dtrace_fasttrap_exec_ptr = NULL;
21887c478bd9Sstevel@tonic-gate 
21897c478bd9Sstevel@tonic-gate 	ASSERT(dtrace_fasttrap_exit_ptr == &fasttrap_exec_exit);
21907c478bd9Sstevel@tonic-gate 	dtrace_fasttrap_exit_ptr = NULL;
21917c478bd9Sstevel@tonic-gate 
21927c478bd9Sstevel@tonic-gate 	ddi_remove_minor_node(devi, NULL);
21937c478bd9Sstevel@tonic-gate 
21947c478bd9Sstevel@tonic-gate 	return (DDI_SUCCESS);
21957c478bd9Sstevel@tonic-gate }
21967c478bd9Sstevel@tonic-gate 
21977c478bd9Sstevel@tonic-gate static struct dev_ops fasttrap_ops = {
21987c478bd9Sstevel@tonic-gate 	DEVO_REV,		/* devo_rev */
21997c478bd9Sstevel@tonic-gate 	0,			/* refcnt */
22007c478bd9Sstevel@tonic-gate 	fasttrap_info,		/* get_dev_info */
22017c478bd9Sstevel@tonic-gate 	nulldev,		/* identify */
22027c478bd9Sstevel@tonic-gate 	nulldev,		/* probe */
22037c478bd9Sstevel@tonic-gate 	fasttrap_attach,	/* attach */
22047c478bd9Sstevel@tonic-gate 	fasttrap_detach,	/* detach */
22057c478bd9Sstevel@tonic-gate 	nodev,			/* reset */
22067c478bd9Sstevel@tonic-gate 	&fasttrap_cb_ops,	/* driver operations */
22077c478bd9Sstevel@tonic-gate 	NULL,			/* bus operations */
22087c478bd9Sstevel@tonic-gate 	nodev			/* dev power */
22097c478bd9Sstevel@tonic-gate };
22107c478bd9Sstevel@tonic-gate 
22117c478bd9Sstevel@tonic-gate /*
22127c478bd9Sstevel@tonic-gate  * Module linkage information for the kernel.
22137c478bd9Sstevel@tonic-gate  */
22147c478bd9Sstevel@tonic-gate static struct modldrv modldrv = {
22157c478bd9Sstevel@tonic-gate 	&mod_driverops,		/* module type (this is a pseudo driver) */
22167c478bd9Sstevel@tonic-gate 	"Fasttrap Tracing",	/* name of module */
22177c478bd9Sstevel@tonic-gate 	&fasttrap_ops,		/* driver ops */
22187c478bd9Sstevel@tonic-gate };
22197c478bd9Sstevel@tonic-gate 
22207c478bd9Sstevel@tonic-gate static struct modlinkage modlinkage = {
22217c478bd9Sstevel@tonic-gate 	MODREV_1,
22227c478bd9Sstevel@tonic-gate 	(void *)&modldrv,
22237c478bd9Sstevel@tonic-gate 	NULL
22247c478bd9Sstevel@tonic-gate };
22257c478bd9Sstevel@tonic-gate 
22267c478bd9Sstevel@tonic-gate int
22277c478bd9Sstevel@tonic-gate _init(void)
22287c478bd9Sstevel@tonic-gate {
22297c478bd9Sstevel@tonic-gate 	return (mod_install(&modlinkage));
22307c478bd9Sstevel@tonic-gate }
22317c478bd9Sstevel@tonic-gate 
22327c478bd9Sstevel@tonic-gate int
22337c478bd9Sstevel@tonic-gate _info(struct modinfo *modinfop)
22347c478bd9Sstevel@tonic-gate {
22357c478bd9Sstevel@tonic-gate 	return (mod_info(&modlinkage, modinfop));
22367c478bd9Sstevel@tonic-gate }
22377c478bd9Sstevel@tonic-gate 
22387c478bd9Sstevel@tonic-gate int
22397c478bd9Sstevel@tonic-gate _fini(void)
22407c478bd9Sstevel@tonic-gate {
22417c478bd9Sstevel@tonic-gate 	return (mod_remove(&modlinkage));
22427c478bd9Sstevel@tonic-gate }
2243