17c478bd9Sstevel@tonic-gate /*
27c478bd9Sstevel@tonic-gate  * CDDL HEADER START
37c478bd9Sstevel@tonic-gate  *
47c478bd9Sstevel@tonic-gate  * The contents of this file are subject to the terms of the
5b8fac8e1Sjhaslam  * Common Development and Distribution License (the "License").
6b8fac8e1Sjhaslam  * You may not use this file except in compliance with the License.
77c478bd9Sstevel@tonic-gate  *
87c478bd9Sstevel@tonic-gate  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
97c478bd9Sstevel@tonic-gate  * or http://www.opensolaris.org/os/licensing.
107c478bd9Sstevel@tonic-gate  * See the License for the specific language governing permissions
117c478bd9Sstevel@tonic-gate  * and limitations under the License.
127c478bd9Sstevel@tonic-gate  *
137c478bd9Sstevel@tonic-gate  * When distributing Covered Code, include this CDDL HEADER in each
147c478bd9Sstevel@tonic-gate  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
157c478bd9Sstevel@tonic-gate  * If applicable, add the following below this CDDL HEADER, with the
167c478bd9Sstevel@tonic-gate  * fields enclosed by brackets "[]" replaced with your own identifying
177c478bd9Sstevel@tonic-gate  * information: Portions Copyright [yyyy] [name of copyright owner]
187c478bd9Sstevel@tonic-gate  *
197c478bd9Sstevel@tonic-gate  * CDDL HEADER END
207c478bd9Sstevel@tonic-gate  */
217c478bd9Sstevel@tonic-gate /*
22b8fac8e1Sjhaslam  * Copyright 2007 Sun Microsystems, Inc.  All rights reserved.
237c478bd9Sstevel@tonic-gate  * Use is subject to license terms.
247c478bd9Sstevel@tonic-gate  */
257c478bd9Sstevel@tonic-gate 
267c478bd9Sstevel@tonic-gate #include <stddef.h>
277c478bd9Sstevel@tonic-gate #include <stdlib.h>
287c478bd9Sstevel@tonic-gate #include <strings.h>
297c478bd9Sstevel@tonic-gate #include <errno.h>
307c478bd9Sstevel@tonic-gate #include <unistd.h>
317c478bd9Sstevel@tonic-gate #include <assert.h>
327c478bd9Sstevel@tonic-gate #include <alloca.h>
337c478bd9Sstevel@tonic-gate 
341a7c1b72Smws #include <dt_impl.h>
351a7c1b72Smws #include <dt_program.h>
361a7c1b72Smws 
377c478bd9Sstevel@tonic-gate static const char _dt_errprog[] =
387c478bd9Sstevel@tonic-gate "dtrace:::ERROR"
397c478bd9Sstevel@tonic-gate "{"
407c478bd9Sstevel@tonic-gate "	trace(arg1);"
417c478bd9Sstevel@tonic-gate "	trace(arg2);"
427c478bd9Sstevel@tonic-gate "	trace(arg3);"
437c478bd9Sstevel@tonic-gate "	trace(arg4);"
447c478bd9Sstevel@tonic-gate "	trace(arg5);"
457c478bd9Sstevel@tonic-gate "}";
467c478bd9Sstevel@tonic-gate 
477c478bd9Sstevel@tonic-gate int
dtrace_handle_err(dtrace_hdl_t * dtp,dtrace_handle_err_f * hdlr,void * arg)487c478bd9Sstevel@tonic-gate dtrace_handle_err(dtrace_hdl_t *dtp, dtrace_handle_err_f *hdlr, void *arg)
497c478bd9Sstevel@tonic-gate {
507c478bd9Sstevel@tonic-gate 	dtrace_prog_t *pgp = NULL;
517c478bd9Sstevel@tonic-gate 	dt_stmt_t *stp;
527c478bd9Sstevel@tonic-gate 	dtrace_ecbdesc_t *edp;
537c478bd9Sstevel@tonic-gate 
547c478bd9Sstevel@tonic-gate 	/*
557c478bd9Sstevel@tonic-gate 	 * We don't currently support multiple error handlers.
567c478bd9Sstevel@tonic-gate 	 */
577c478bd9Sstevel@tonic-gate 	if (dtp->dt_errhdlr != NULL)
587c478bd9Sstevel@tonic-gate 		return (dt_set_errno(dtp, EALREADY));
597c478bd9Sstevel@tonic-gate 
607c478bd9Sstevel@tonic-gate 	/*
617c478bd9Sstevel@tonic-gate 	 * If the DTRACEOPT_GRABANON is enabled, the anonymous enabling will
627c478bd9Sstevel@tonic-gate 	 * already have a dtrace:::ERROR probe enabled; save 'hdlr' and 'arg'
637c478bd9Sstevel@tonic-gate 	 * but do not bother compiling and enabling _dt_errprog.
647c478bd9Sstevel@tonic-gate 	 */
657c478bd9Sstevel@tonic-gate 	if (dtp->dt_options[DTRACEOPT_GRABANON] != DTRACEOPT_UNSET)
667c478bd9Sstevel@tonic-gate 		goto out;
677c478bd9Sstevel@tonic-gate 
687c478bd9Sstevel@tonic-gate 	if ((pgp = dtrace_program_strcompile(dtp, _dt_errprog,
697c478bd9Sstevel@tonic-gate 	    DTRACE_PROBESPEC_NAME, DTRACE_C_ZDEFS, 0, NULL)) == NULL)
707c478bd9Sstevel@tonic-gate 		return (dt_set_errno(dtp, dtrace_errno(dtp)));
717c478bd9Sstevel@tonic-gate 
727c478bd9Sstevel@tonic-gate 	stp = dt_list_next(&pgp->dp_stmts);
737c478bd9Sstevel@tonic-gate 	assert(stp != NULL);
747c478bd9Sstevel@tonic-gate 
757c478bd9Sstevel@tonic-gate 	edp = stp->ds_desc->dtsd_ecbdesc;
767c478bd9Sstevel@tonic-gate 	assert(edp != NULL);
777c478bd9Sstevel@tonic-gate 	edp->dted_uarg = DT_ECB_ERROR;
787c478bd9Sstevel@tonic-gate 
797c478bd9Sstevel@tonic-gate out:
807c478bd9Sstevel@tonic-gate 	dtp->dt_errhdlr = hdlr;
817c478bd9Sstevel@tonic-gate 	dtp->dt_errarg = arg;
827c478bd9Sstevel@tonic-gate 	dtp->dt_errprog = pgp;
837c478bd9Sstevel@tonic-gate 
847c478bd9Sstevel@tonic-gate 	return (0);
857c478bd9Sstevel@tonic-gate }
867c478bd9Sstevel@tonic-gate 
877c478bd9Sstevel@tonic-gate int
dtrace_handle_drop(dtrace_hdl_t * dtp,dtrace_handle_drop_f * hdlr,void * arg)887c478bd9Sstevel@tonic-gate dtrace_handle_drop(dtrace_hdl_t *dtp, dtrace_handle_drop_f *hdlr, void *arg)
897c478bd9Sstevel@tonic-gate {
907c478bd9Sstevel@tonic-gate 	if (dtp->dt_drophdlr != NULL)
917c478bd9Sstevel@tonic-gate 		return (dt_set_errno(dtp, EALREADY));
927c478bd9Sstevel@tonic-gate 
937c478bd9Sstevel@tonic-gate 	dtp->dt_drophdlr = hdlr;
947c478bd9Sstevel@tonic-gate 	dtp->dt_droparg = arg;
957c478bd9Sstevel@tonic-gate 
967c478bd9Sstevel@tonic-gate 	return (0);
977c478bd9Sstevel@tonic-gate }
987c478bd9Sstevel@tonic-gate 
997c478bd9Sstevel@tonic-gate int
dtrace_handle_proc(dtrace_hdl_t * dtp,dtrace_handle_proc_f * hdlr,void * arg)1007c478bd9Sstevel@tonic-gate dtrace_handle_proc(dtrace_hdl_t *dtp, dtrace_handle_proc_f *hdlr, void *arg)
1017c478bd9Sstevel@tonic-gate {
1027c478bd9Sstevel@tonic-gate 	if (dtp->dt_prochdlr != NULL)
1037c478bd9Sstevel@tonic-gate 		return (dt_set_errno(dtp, EALREADY));
1047c478bd9Sstevel@tonic-gate 
1057c478bd9Sstevel@tonic-gate 	dtp->dt_prochdlr = hdlr;
1067c478bd9Sstevel@tonic-gate 	dtp->dt_procarg = arg;
1077c478bd9Sstevel@tonic-gate 
1087c478bd9Sstevel@tonic-gate 	return (0);
1097c478bd9Sstevel@tonic-gate }
1107c478bd9Sstevel@tonic-gate 
1117c478bd9Sstevel@tonic-gate int
dtrace_handle_buffered(dtrace_hdl_t * dtp,dtrace_handle_buffered_f * hdlr,void * arg)1127c478bd9Sstevel@tonic-gate dtrace_handle_buffered(dtrace_hdl_t *dtp, dtrace_handle_buffered_f *hdlr,
1137c478bd9Sstevel@tonic-gate     void *arg)
1147c478bd9Sstevel@tonic-gate {
1157c478bd9Sstevel@tonic-gate 	if (dtp->dt_bufhdlr != NULL)
1167c478bd9Sstevel@tonic-gate 		return (dt_set_errno(dtp, EALREADY));
1177c478bd9Sstevel@tonic-gate 
1187c478bd9Sstevel@tonic-gate 	if (hdlr == NULL)
1197c478bd9Sstevel@tonic-gate 		return (dt_set_errno(dtp, EINVAL));
1207c478bd9Sstevel@tonic-gate 
1217c478bd9Sstevel@tonic-gate 	dtp->dt_bufhdlr = hdlr;
1227c478bd9Sstevel@tonic-gate 	dtp->dt_bufarg = arg;
1237c478bd9Sstevel@tonic-gate 
1247c478bd9Sstevel@tonic-gate 	return (0);
1257c478bd9Sstevel@tonic-gate }
1267c478bd9Sstevel@tonic-gate 
127a1b5e537Sbmc int
dtrace_handle_setopt(dtrace_hdl_t * dtp,dtrace_handle_setopt_f * hdlr,void * arg)128a1b5e537Sbmc dtrace_handle_setopt(dtrace_hdl_t *dtp, dtrace_handle_setopt_f *hdlr,
129a1b5e537Sbmc     void *arg)
130a1b5e537Sbmc {
131a1b5e537Sbmc 	if (hdlr == NULL)
132a1b5e537Sbmc 		return (dt_set_errno(dtp, EINVAL));
133a1b5e537Sbmc 
134a1b5e537Sbmc 	dtp->dt_setopthdlr = hdlr;
135a1b5e537Sbmc 	dtp->dt_setoptarg = arg;
136a1b5e537Sbmc 
137a1b5e537Sbmc 	return (0);
138a1b5e537Sbmc }
139a1b5e537Sbmc 
1407c478bd9Sstevel@tonic-gate #define	DT_REC(type, ndx) *((type *)((uintptr_t)data->dtpda_data + \
1417c478bd9Sstevel@tonic-gate     epd->dtepd_rec[(ndx)].dtrd_offset))
1427c478bd9Sstevel@tonic-gate 
1437c478bd9Sstevel@tonic-gate static int
dt_handle_err(dtrace_hdl_t * dtp,dtrace_probedata_t * data)1447c478bd9Sstevel@tonic-gate dt_handle_err(dtrace_hdl_t *dtp, dtrace_probedata_t *data)
1457c478bd9Sstevel@tonic-gate {
1467c478bd9Sstevel@tonic-gate 	dtrace_eprobedesc_t *epd = data->dtpda_edesc, *errepd;
1477c478bd9Sstevel@tonic-gate 	dtrace_probedesc_t *pd = data->dtpda_pdesc, *errpd;
1487c478bd9Sstevel@tonic-gate 	dtrace_errdata_t err;
1497c478bd9Sstevel@tonic-gate 	dtrace_epid_t epid;
1507c478bd9Sstevel@tonic-gate 
1517c478bd9Sstevel@tonic-gate 	char where[30];
1527c478bd9Sstevel@tonic-gate 	char details[30];
1537c478bd9Sstevel@tonic-gate 	char offinfo[30];
1547c478bd9Sstevel@tonic-gate 	const int slop = 80;
1557c478bd9Sstevel@tonic-gate 	const char *faultstr;
1567c478bd9Sstevel@tonic-gate 	char *str;
1577c478bd9Sstevel@tonic-gate 	int len;
1587c478bd9Sstevel@tonic-gate 
1597c478bd9Sstevel@tonic-gate 	assert(epd->dtepd_uarg == DT_ECB_ERROR);
1607c478bd9Sstevel@tonic-gate 
1617c478bd9Sstevel@tonic-gate 	if (epd->dtepd_nrecs != 5 || strcmp(pd->dtpd_provider, "dtrace") != 0 ||
1627c478bd9Sstevel@tonic-gate 	    strcmp(pd->dtpd_name, "ERROR") != 0)
1637c478bd9Sstevel@tonic-gate 		return (dt_set_errno(dtp, EDT_BADERROR));
1647c478bd9Sstevel@tonic-gate 
1657c478bd9Sstevel@tonic-gate 	/*
1667c478bd9Sstevel@tonic-gate 	 * This is an error.  We have the following items here:  EPID,
1677c478bd9Sstevel@tonic-gate 	 * faulting action, DIF offset, fault code and faulting address.
1687c478bd9Sstevel@tonic-gate 	 */
1697c478bd9Sstevel@tonic-gate 	epid = (uint32_t)DT_REC(uint64_t, 0);
1707c478bd9Sstevel@tonic-gate 
1717c478bd9Sstevel@tonic-gate 	if (dt_epid_lookup(dtp, epid, &errepd, &errpd) != 0)
1727c478bd9Sstevel@tonic-gate 		return (dt_set_errno(dtp, EDT_BADERROR));
1737c478bd9Sstevel@tonic-gate 
1747c478bd9Sstevel@tonic-gate 	err.dteda_edesc = errepd;
1757c478bd9Sstevel@tonic-gate 	err.dteda_pdesc = errpd;
1767c478bd9Sstevel@tonic-gate 	err.dteda_cpu = data->dtpda_cpu;
1777c478bd9Sstevel@tonic-gate 	err.dteda_action = (int)DT_REC(uint64_t, 1);
1787c478bd9Sstevel@tonic-gate 	err.dteda_offset = (int)DT_REC(uint64_t, 2);
1797c478bd9Sstevel@tonic-gate 	err.dteda_fault = (int)DT_REC(uint64_t, 3);
1807c478bd9Sstevel@tonic-gate 	err.dteda_addr = DT_REC(uint64_t, 4);
1817c478bd9Sstevel@tonic-gate 
1827c478bd9Sstevel@tonic-gate 	faultstr = dtrace_faultstr(dtp, err.dteda_fault);
1837c478bd9Sstevel@tonic-gate 	len = sizeof (where) + sizeof (offinfo) + strlen(faultstr) +
1847c478bd9Sstevel@tonic-gate 	    strlen(errpd->dtpd_provider) + strlen(errpd->dtpd_mod) +
1857c478bd9Sstevel@tonic-gate 	    strlen(errpd->dtpd_name) + strlen(errpd->dtpd_func) +
1867c478bd9Sstevel@tonic-gate 	    slop;
1877c478bd9Sstevel@tonic-gate 
1887c478bd9Sstevel@tonic-gate 	str = (char *)alloca(len);
1897c478bd9Sstevel@tonic-gate 
1907c478bd9Sstevel@tonic-gate 	if (err.dteda_action == 0) {
1917c478bd9Sstevel@tonic-gate 		(void) sprintf(where, "predicate");
1927c478bd9Sstevel@tonic-gate 	} else {
1937c478bd9Sstevel@tonic-gate 		(void) sprintf(where, "action #%d", err.dteda_action);
1947c478bd9Sstevel@tonic-gate 	}
1957c478bd9Sstevel@tonic-gate 
1967c478bd9Sstevel@tonic-gate 	if (err.dteda_offset != -1) {
1977c478bd9Sstevel@tonic-gate 		(void) sprintf(offinfo, " at DIF offset %d", err.dteda_offset);
1987c478bd9Sstevel@tonic-gate 	} else {
1997c478bd9Sstevel@tonic-gate 		offinfo[0] = 0;
2007c478bd9Sstevel@tonic-gate 	}
2017c478bd9Sstevel@tonic-gate 
2027c478bd9Sstevel@tonic-gate 	switch (err.dteda_fault) {
2037c478bd9Sstevel@tonic-gate 	case DTRACEFLT_BADADDR:
2047c478bd9Sstevel@tonic-gate 	case DTRACEFLT_BADALIGN:
205b8fac8e1Sjhaslam 	case DTRACEFLT_BADSTACK:
2067c478bd9Sstevel@tonic-gate 		(void) sprintf(details, " (0x%llx)",
2077c478bd9Sstevel@tonic-gate 		    (u_longlong_t)err.dteda_addr);
2087c478bd9Sstevel@tonic-gate 		break;
2097c478bd9Sstevel@tonic-gate 
2107c478bd9Sstevel@tonic-gate 	default:
2117c478bd9Sstevel@tonic-gate 		details[0] = 0;
2127c478bd9Sstevel@tonic-gate 	}
2137c478bd9Sstevel@tonic-gate 
2147c478bd9Sstevel@tonic-gate 	(void) snprintf(str, len, "error on enabled probe ID %u "
2157c478bd9Sstevel@tonic-gate 	    "(ID %u: %s:%s:%s:%s): %s%s in %s%s\n",
2167c478bd9Sstevel@tonic-gate 	    epid, errpd->dtpd_id, errpd->dtpd_provider,
2177c478bd9Sstevel@tonic-gate 	    errpd->dtpd_mod, errpd->dtpd_func,
2187c478bd9Sstevel@tonic-gate 	    errpd->dtpd_name, dtrace_faultstr(dtp, err.dteda_fault),
2197c478bd9Sstevel@tonic-gate 	    details, where, offinfo);
2207c478bd9Sstevel@tonic-gate 
2217c478bd9Sstevel@tonic-gate 	err.dteda_msg = str;
2227c478bd9Sstevel@tonic-gate 
2237c478bd9Sstevel@tonic-gate 	if (dtp->dt_errhdlr == NULL)
2247c478bd9Sstevel@tonic-gate 		return (dt_set_errno(dtp, EDT_ERRABORT));
2257c478bd9Sstevel@tonic-gate 
2267c478bd9Sstevel@tonic-gate 	if ((*dtp->dt_errhdlr)(&err, dtp->dt_errarg) == DTRACE_HANDLE_ABORT)
2277c478bd9Sstevel@tonic-gate 		return (dt_set_errno(dtp, EDT_ERRABORT));
2287c478bd9Sstevel@tonic-gate 
2297c478bd9Sstevel@tonic-gate 	return (0);
2307c478bd9Sstevel@tonic-gate }
2317c478bd9Sstevel@tonic-gate 
2327c478bd9Sstevel@tonic-gate int
dt_handle_liberr(dtrace_hdl_t * dtp,const dtrace_probedata_t * data,const char * faultstr)2337c478bd9Sstevel@tonic-gate dt_handle_liberr(dtrace_hdl_t *dtp, const dtrace_probedata_t *data,
2347c478bd9Sstevel@tonic-gate     const char *faultstr)
2357c478bd9Sstevel@tonic-gate {
2367c478bd9Sstevel@tonic-gate 	dtrace_probedesc_t *errpd = data->dtpda_pdesc;
2377c478bd9Sstevel@tonic-gate 	dtrace_errdata_t err;
2387c478bd9Sstevel@tonic-gate 	const int slop = 80;
2397c478bd9Sstevel@tonic-gate 	char *str;
2407c478bd9Sstevel@tonic-gate 	int len;
2417c478bd9Sstevel@tonic-gate 
2427c478bd9Sstevel@tonic-gate 	err.dteda_edesc = data->dtpda_edesc;
2437c478bd9Sstevel@tonic-gate 	err.dteda_pdesc = errpd;
2447c478bd9Sstevel@tonic-gate 	err.dteda_cpu = data->dtpda_cpu;
2457c478bd9Sstevel@tonic-gate 	err.dteda_action = -1;
2467c478bd9Sstevel@tonic-gate 	err.dteda_offset = -1;
2477c478bd9Sstevel@tonic-gate 	err.dteda_fault = DTRACEFLT_LIBRARY;
248*d60ff336SToomas Soome 	err.dteda_addr = 0;
2497c478bd9Sstevel@tonic-gate 
2507c478bd9Sstevel@tonic-gate 	len = strlen(faultstr) +
2517c478bd9Sstevel@tonic-gate 	    strlen(errpd->dtpd_provider) + strlen(errpd->dtpd_mod) +
2527c478bd9Sstevel@tonic-gate 	    strlen(errpd->dtpd_name) + strlen(errpd->dtpd_func) +
2537c478bd9Sstevel@tonic-gate 	    slop;
2547c478bd9Sstevel@tonic-gate 
2557c478bd9Sstevel@tonic-gate 	str = alloca(len);
2567c478bd9Sstevel@tonic-gate 
2577c478bd9Sstevel@tonic-gate 	(void) snprintf(str, len, "error on enabled probe ID %u "
2587c478bd9Sstevel@tonic-gate 	    "(ID %u: %s:%s:%s:%s): %s\n",
2597c478bd9Sstevel@tonic-gate 	    data->dtpda_edesc->dtepd_epid,
2607c478bd9Sstevel@tonic-gate 	    errpd->dtpd_id, errpd->dtpd_provider,
2617c478bd9Sstevel@tonic-gate 	    errpd->dtpd_mod, errpd->dtpd_func,
2627c478bd9Sstevel@tonic-gate 	    errpd->dtpd_name, faultstr);
2637c478bd9Sstevel@tonic-gate 
2647c478bd9Sstevel@tonic-gate 	err.dteda_msg = str;
2657c478bd9Sstevel@tonic-gate 
2667c478bd9Sstevel@tonic-gate 	if (dtp->dt_errhdlr == NULL)
2677c478bd9Sstevel@tonic-gate 		return (dt_set_errno(dtp, EDT_ERRABORT));
2687c478bd9Sstevel@tonic-gate 
2697c478bd9Sstevel@tonic-gate 	if ((*dtp->dt_errhdlr)(&err, dtp->dt_errarg) == DTRACE_HANDLE_ABORT)
2707c478bd9Sstevel@tonic-gate 		return (dt_set_errno(dtp, EDT_ERRABORT));
2717c478bd9Sstevel@tonic-gate 
2727c478bd9Sstevel@tonic-gate 	return (0);
2737c478bd9Sstevel@tonic-gate }
2747c478bd9Sstevel@tonic-gate 
275a1b5e537Sbmc #define	DROPTAG(x)	x, #x
276a1b5e537Sbmc 
277a1b5e537Sbmc static const struct {
278a1b5e537Sbmc 	dtrace_dropkind_t dtdrg_kind;
279a1b5e537Sbmc 	char *dtdrg_tag;
280a1b5e537Sbmc } _dt_droptags[] = {
281a1b5e537Sbmc 	{ DROPTAG(DTRACEDROP_PRINCIPAL) },
282a1b5e537Sbmc 	{ DROPTAG(DTRACEDROP_AGGREGATION) },
283a1b5e537Sbmc 	{ DROPTAG(DTRACEDROP_DYNAMIC) },
284a1b5e537Sbmc 	{ DROPTAG(DTRACEDROP_DYNRINSE) },
285a1b5e537Sbmc 	{ DROPTAG(DTRACEDROP_DYNDIRTY) },
286a1b5e537Sbmc 	{ DROPTAG(DTRACEDROP_SPEC) },
287a1b5e537Sbmc 	{ DROPTAG(DTRACEDROP_SPECBUSY) },
288a1b5e537Sbmc 	{ DROPTAG(DTRACEDROP_SPECUNAVAIL) },
289a1b5e537Sbmc 	{ DROPTAG(DTRACEDROP_DBLERROR) },
290a1b5e537Sbmc 	{ DROPTAG(DTRACEDROP_STKSTROVERFLOW) },
291a1b5e537Sbmc 	{ 0, NULL }
292a1b5e537Sbmc };
293a1b5e537Sbmc 
294a1b5e537Sbmc static const char *
dt_droptag(dtrace_dropkind_t kind)295a1b5e537Sbmc dt_droptag(dtrace_dropkind_t kind)
296a1b5e537Sbmc {
297a1b5e537Sbmc 	int i;
298a1b5e537Sbmc 
299a1b5e537Sbmc 	for (i = 0; _dt_droptags[i].dtdrg_tag != NULL; i++) {
300a1b5e537Sbmc 		if (_dt_droptags[i].dtdrg_kind == kind)
301a1b5e537Sbmc 			return (_dt_droptags[i].dtdrg_tag);
302a1b5e537Sbmc 	}
303a1b5e537Sbmc 
304a1b5e537Sbmc 	return ("DTRACEDROP_UNKNOWN");
305a1b5e537Sbmc }
306a1b5e537Sbmc 
3077c478bd9Sstevel@tonic-gate int
dt_handle_cpudrop(dtrace_hdl_t * dtp,processorid_t cpu,dtrace_dropkind_t what,uint64_t howmany)3087c478bd9Sstevel@tonic-gate dt_handle_cpudrop(dtrace_hdl_t *dtp, processorid_t cpu,
3097c478bd9Sstevel@tonic-gate     dtrace_dropkind_t what, uint64_t howmany)
3107c478bd9Sstevel@tonic-gate {
3117c478bd9Sstevel@tonic-gate 	dtrace_dropdata_t drop;
312a1b5e537Sbmc 	char str[80], *s;
313a1b5e537Sbmc 	int size;
3147c478bd9Sstevel@tonic-gate 
3157c478bd9Sstevel@tonic-gate 	assert(what == DTRACEDROP_PRINCIPAL || what == DTRACEDROP_AGGREGATION);
3167c478bd9Sstevel@tonic-gate 
3177c478bd9Sstevel@tonic-gate 	bzero(&drop, sizeof (drop));
3187c478bd9Sstevel@tonic-gate 	drop.dtdda_handle = dtp;
3197c478bd9Sstevel@tonic-gate 	drop.dtdda_cpu = cpu;
3207c478bd9Sstevel@tonic-gate 	drop.dtdda_kind = what;
3217c478bd9Sstevel@tonic-gate 	drop.dtdda_drops = howmany;
3227c478bd9Sstevel@tonic-gate 	drop.dtdda_msg = str;
3237c478bd9Sstevel@tonic-gate 
324a1b5e537Sbmc 	if (dtp->dt_droptags) {
325a1b5e537Sbmc 		(void) snprintf(str, sizeof (str), "[%s] ", dt_droptag(what));
326a1b5e537Sbmc 		s = &str[strlen(str)];
327a1b5e537Sbmc 		size = sizeof (str) - (s - str);
328a1b5e537Sbmc 	} else {
329a1b5e537Sbmc 		s = str;
330a1b5e537Sbmc 		size = sizeof (str);
331a1b5e537Sbmc 	}
332a1b5e537Sbmc 
333a1b5e537Sbmc 	(void) snprintf(s, size, "%llu %sdrop%s on CPU %d\n",
3347c478bd9Sstevel@tonic-gate 	    howmany, what == DTRACEDROP_PRINCIPAL ? "" : "aggregation ",
3357c478bd9Sstevel@tonic-gate 	    howmany > 1 ? "s" : "", cpu);
3367c478bd9Sstevel@tonic-gate 
3377c478bd9Sstevel@tonic-gate 	if (dtp->dt_drophdlr == NULL)
3387c478bd9Sstevel@tonic-gate 		return (dt_set_errno(dtp, EDT_DROPABORT));
3397c478bd9Sstevel@tonic-gate 
3407c478bd9Sstevel@tonic-gate 	if ((*dtp->dt_drophdlr)(&drop, dtp->dt_droparg) == DTRACE_HANDLE_ABORT)
3417c478bd9Sstevel@tonic-gate 		return (dt_set_errno(dtp, EDT_DROPABORT));
3427c478bd9Sstevel@tonic-gate 
3437c478bd9Sstevel@tonic-gate 	return (0);
3447c478bd9Sstevel@tonic-gate }
3457c478bd9Sstevel@tonic-gate 
3467c478bd9Sstevel@tonic-gate static const struct {
3477c478bd9Sstevel@tonic-gate 	dtrace_dropkind_t dtdrt_kind;
3487c478bd9Sstevel@tonic-gate 	uintptr_t dtdrt_offset;
3497c478bd9Sstevel@tonic-gate 	const char *dtdrt_str;
3507c478bd9Sstevel@tonic-gate 	const char *dtdrt_msg;
3517c478bd9Sstevel@tonic-gate } _dt_droptab[] = {
3527c478bd9Sstevel@tonic-gate 	{ DTRACEDROP_DYNAMIC,
3537c478bd9Sstevel@tonic-gate 	    offsetof(dtrace_status_t, dtst_dyndrops),
3547c478bd9Sstevel@tonic-gate 	    "dynamic variable drop" },
3557c478bd9Sstevel@tonic-gate 
3567c478bd9Sstevel@tonic-gate 	{ DTRACEDROP_DYNRINSE,
3577c478bd9Sstevel@tonic-gate 	    offsetof(dtrace_status_t, dtst_dyndrops_rinsing),
3587c478bd9Sstevel@tonic-gate 	    "dynamic variable drop", " with non-empty rinsing list" },
3597c478bd9Sstevel@tonic-gate 
3607c478bd9Sstevel@tonic-gate 	{ DTRACEDROP_DYNDIRTY,
3617c478bd9Sstevel@tonic-gate 	    offsetof(dtrace_status_t, dtst_dyndrops_dirty),
3627c478bd9Sstevel@tonic-gate 	    "dynamic variable drop", " with non-empty dirty list" },
3637c478bd9Sstevel@tonic-gate 
3647c478bd9Sstevel@tonic-gate 	{ DTRACEDROP_SPEC,
3657c478bd9Sstevel@tonic-gate 	    offsetof(dtrace_status_t, dtst_specdrops),
3667c478bd9Sstevel@tonic-gate 	    "speculative drop" },
3677c478bd9Sstevel@tonic-gate 
3687c478bd9Sstevel@tonic-gate 	{ DTRACEDROP_SPECBUSY,
3697c478bd9Sstevel@tonic-gate 	    offsetof(dtrace_status_t, dtst_specdrops_busy),
3707c478bd9Sstevel@tonic-gate 	    "failed speculation", " (available buffer(s) still busy)" },
3717c478bd9Sstevel@tonic-gate 
3727c478bd9Sstevel@tonic-gate 	{ DTRACEDROP_SPECUNAVAIL,
3737c478bd9Sstevel@tonic-gate 	    offsetof(dtrace_status_t, dtst_specdrops_unavail),
3747c478bd9Sstevel@tonic-gate 	    "failed speculation", " (no speculative buffer available)" },
3757c478bd9Sstevel@tonic-gate 
376a1b5e537Sbmc 	{ DTRACEDROP_STKSTROVERFLOW,
377a1b5e537Sbmc 	    offsetof(dtrace_status_t, dtst_stkstroverflows),
378a1b5e537Sbmc 	    "jstack()/ustack() string table overflow" },
379a1b5e537Sbmc 
380a1b5e537Sbmc 	{ DTRACEDROP_DBLERROR,
381a1b5e537Sbmc 	    offsetof(dtrace_status_t, dtst_dblerrors),
382a1b5e537Sbmc 	    "error", " in ERROR probe enabling" },
383a1b5e537Sbmc 
3847c478bd9Sstevel@tonic-gate 	{ 0, 0, NULL }
3857c478bd9Sstevel@tonic-gate };
3867c478bd9Sstevel@tonic-gate 
3877c478bd9Sstevel@tonic-gate int
dt_handle_status(dtrace_hdl_t * dtp,dtrace_status_t * old,dtrace_status_t * new)3887c478bd9Sstevel@tonic-gate dt_handle_status(dtrace_hdl_t *dtp, dtrace_status_t *old, dtrace_status_t *new)
3897c478bd9Sstevel@tonic-gate {
3907c478bd9Sstevel@tonic-gate 	dtrace_dropdata_t drop;
391a1b5e537Sbmc 	char str[80], *s;
3927c478bd9Sstevel@tonic-gate 	uintptr_t base = (uintptr_t)new, obase = (uintptr_t)old;
393a1b5e537Sbmc 	int i, size;
3947c478bd9Sstevel@tonic-gate 
3957c478bd9Sstevel@tonic-gate 	bzero(&drop, sizeof (drop));
3967c478bd9Sstevel@tonic-gate 	drop.dtdda_handle = dtp;
3977c478bd9Sstevel@tonic-gate 	drop.dtdda_cpu = DTRACE_CPUALL;
3987c478bd9Sstevel@tonic-gate 	drop.dtdda_msg = str;
3997c478bd9Sstevel@tonic-gate 
4007c478bd9Sstevel@tonic-gate 	/*
4017c478bd9Sstevel@tonic-gate 	 * First, check to see if we've been killed -- in which case we abort.
4027c478bd9Sstevel@tonic-gate 	 */
4037c478bd9Sstevel@tonic-gate 	if (new->dtst_killed && !old->dtst_killed)
4047c478bd9Sstevel@tonic-gate 		return (dt_set_errno(dtp, EDT_BRICKED));
4057c478bd9Sstevel@tonic-gate 
4067c478bd9Sstevel@tonic-gate 	for (i = 0; _dt_droptab[i].dtdrt_str != NULL; i++) {
4077c478bd9Sstevel@tonic-gate 		uintptr_t naddr = base + _dt_droptab[i].dtdrt_offset;
4087c478bd9Sstevel@tonic-gate 		uintptr_t oaddr = obase + _dt_droptab[i].dtdrt_offset;
4097c478bd9Sstevel@tonic-gate 
4107c478bd9Sstevel@tonic-gate 		uint64_t nval = *((uint64_t *)naddr);
4117c478bd9Sstevel@tonic-gate 		uint64_t oval = *((uint64_t *)oaddr);
4127c478bd9Sstevel@tonic-gate 
4137c478bd9Sstevel@tonic-gate 		if (nval == oval)
4147c478bd9Sstevel@tonic-gate 			continue;
4157c478bd9Sstevel@tonic-gate 
416a1b5e537Sbmc 		if (dtp->dt_droptags) {
417a1b5e537Sbmc 			(void) snprintf(str, sizeof (str), "[%s] ",
418a1b5e537Sbmc 			    dt_droptag(_dt_droptab[i].dtdrt_kind));
419a1b5e537Sbmc 			s = &str[strlen(str)];
420a1b5e537Sbmc 			size = sizeof (str) - (s - str);
421a1b5e537Sbmc 		} else {
422a1b5e537Sbmc 			s = str;
423a1b5e537Sbmc 			size = sizeof (str);
424a1b5e537Sbmc 		}
425a1b5e537Sbmc 
426a1b5e537Sbmc 		(void) snprintf(s, size, "%llu %s%s%s\n", nval - oval,
4277c478bd9Sstevel@tonic-gate 		    _dt_droptab[i].dtdrt_str, (nval - oval > 1) ? "s" : "",
4287c478bd9Sstevel@tonic-gate 		    _dt_droptab[i].dtdrt_msg != NULL ?
4297c478bd9Sstevel@tonic-gate 		    _dt_droptab[i].dtdrt_msg : "");
4307c478bd9Sstevel@tonic-gate 
4317c478bd9Sstevel@tonic-gate 		drop.dtdda_kind = _dt_droptab[i].dtdrt_kind;
4327c478bd9Sstevel@tonic-gate 		drop.dtdda_total = nval;
4337c478bd9Sstevel@tonic-gate 		drop.dtdda_drops = nval - oval;
4347c478bd9Sstevel@tonic-gate 
4357c478bd9Sstevel@tonic-gate 		if (dtp->dt_drophdlr == NULL)
4367c478bd9Sstevel@tonic-gate 			return (dt_set_errno(dtp, EDT_DROPABORT));
4377c478bd9Sstevel@tonic-gate 
4387c478bd9Sstevel@tonic-gate 		if ((*dtp->dt_drophdlr)(&drop,
4397c478bd9Sstevel@tonic-gate 		    dtp->dt_droparg) == DTRACE_HANDLE_ABORT)
4407c478bd9Sstevel@tonic-gate 			return (dt_set_errno(dtp, EDT_DROPABORT));
4417c478bd9Sstevel@tonic-gate 	}
4427c478bd9Sstevel@tonic-gate 
4437c478bd9Sstevel@tonic-gate 	return (0);
4447c478bd9Sstevel@tonic-gate }
4457c478bd9Sstevel@tonic-gate 
446a1b5e537Sbmc int
dt_handle_setopt(dtrace_hdl_t * dtp,dtrace_setoptdata_t * data)447a1b5e537Sbmc dt_handle_setopt(dtrace_hdl_t *dtp, dtrace_setoptdata_t *data)
448a1b5e537Sbmc {
449a1b5e537Sbmc 	void *arg = dtp->dt_setoptarg;
450a1b5e537Sbmc 
451a1b5e537Sbmc 	if (dtp->dt_setopthdlr == NULL)
452a1b5e537Sbmc 		return (0);
453a1b5e537Sbmc 
454a1b5e537Sbmc 	if ((*dtp->dt_setopthdlr)(data, arg) == DTRACE_HANDLE_ABORT)
455a1b5e537Sbmc 		return (dt_set_errno(dtp, EDT_DIRABORT));
456a1b5e537Sbmc 
457a1b5e537Sbmc 	return (0);
458a1b5e537Sbmc }
459a1b5e537Sbmc 
4607c478bd9Sstevel@tonic-gate int
dt_handle(dtrace_hdl_t * dtp,dtrace_probedata_t * data)4617c478bd9Sstevel@tonic-gate dt_handle(dtrace_hdl_t *dtp, dtrace_probedata_t *data)
4627c478bd9Sstevel@tonic-gate {
4637c478bd9Sstevel@tonic-gate 	dtrace_eprobedesc_t *epd = data->dtpda_edesc;
4647c478bd9Sstevel@tonic-gate 	int rval;
4657c478bd9Sstevel@tonic-gate 
4667c478bd9Sstevel@tonic-gate 	switch (epd->dtepd_uarg) {
4677c478bd9Sstevel@tonic-gate 	case DT_ECB_ERROR:
4687c478bd9Sstevel@tonic-gate 		rval = dt_handle_err(dtp, data);
4697c478bd9Sstevel@tonic-gate 		break;
4707c478bd9Sstevel@tonic-gate 
4717c478bd9Sstevel@tonic-gate 	default:
4727c478bd9Sstevel@tonic-gate 		return (DTRACE_CONSUME_THIS);
4737c478bd9Sstevel@tonic-gate 	}
4747c478bd9Sstevel@tonic-gate 
4757c478bd9Sstevel@tonic-gate 	if (rval == 0)
4767c478bd9Sstevel@tonic-gate 		return (DTRACE_CONSUME_NEXT);
4777c478bd9Sstevel@tonic-gate 
4787c478bd9Sstevel@tonic-gate 	return (DTRACE_CONSUME_ERROR);
4797c478bd9Sstevel@tonic-gate }
480