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