xref: /onnv-gate/usr/src/lib/libdtrace/common/dt_cc.c (revision 457:d8f2995c64aa)
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  */
220Sstevel@tonic-gate /*
230Sstevel@tonic-gate  * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
240Sstevel@tonic-gate  * Use is subject to license terms.
250Sstevel@tonic-gate  */
260Sstevel@tonic-gate 
270Sstevel@tonic-gate #pragma ident	"%Z%%M%	%I%	%E% SMI"
280Sstevel@tonic-gate 
290Sstevel@tonic-gate /*
300Sstevel@tonic-gate  * DTrace D Language Compiler
310Sstevel@tonic-gate  *
320Sstevel@tonic-gate  * The code in this source file implements the main engine for the D language
330Sstevel@tonic-gate  * compiler.  The driver routine for the compiler is dt_compile(), below.  The
340Sstevel@tonic-gate  * compiler operates on either stdio FILEs or in-memory strings as its input
350Sstevel@tonic-gate  * and can produce either dtrace_prog_t structures from a D program or a single
360Sstevel@tonic-gate  * dtrace_difo_t structure from a D expression.  Multiple entry points are
370Sstevel@tonic-gate  * provided as wrappers around dt_compile() for the various input/output pairs.
380Sstevel@tonic-gate  * The compiler itself is implemented across the following source files:
390Sstevel@tonic-gate  *
400Sstevel@tonic-gate  * dt_lex.l - lex scanner
410Sstevel@tonic-gate  * dt_grammar.y - yacc grammar
420Sstevel@tonic-gate  * dt_parser.c - parse tree creation and semantic checking
430Sstevel@tonic-gate  * dt_decl.c - declaration stack processing
440Sstevel@tonic-gate  * dt_xlator.c - D translator lookup and creation
450Sstevel@tonic-gate  * dt_ident.c - identifier and symbol table routines
460Sstevel@tonic-gate  * dt_pragma.c - #pragma processing and D pragmas
470Sstevel@tonic-gate  * dt_printf.c - D printf() and printa() argument checking and processing
480Sstevel@tonic-gate  * dt_cc.c - compiler driver and dtrace_prog_t construction
490Sstevel@tonic-gate  * dt_cg.c - DIF code generator
500Sstevel@tonic-gate  * dt_as.c - DIF assembler
510Sstevel@tonic-gate  * dt_dof.c - dtrace_prog_t -> DOF conversion
520Sstevel@tonic-gate  *
530Sstevel@tonic-gate  * Several other source files provide collections of utility routines used by
540Sstevel@tonic-gate  * these major files.  The compiler itself is implemented in multiple passes:
550Sstevel@tonic-gate  *
560Sstevel@tonic-gate  * (1) The input program is scanned and parsed by dt_lex.l and dt_grammar.y
570Sstevel@tonic-gate  *     and parse tree nodes are constructed using the routines in dt_parser.c.
580Sstevel@tonic-gate  *     This node construction pass is described further in dt_parser.c.
590Sstevel@tonic-gate  *
600Sstevel@tonic-gate  * (2) The parse tree is "cooked" by assigning each clause a context (see the
610Sstevel@tonic-gate  *     routine dt_setcontext(), below) based on its probe description and then
620Sstevel@tonic-gate  *     recursively descending the tree performing semantic checking.  The cook
630Sstevel@tonic-gate  *     routines are also implemented in dt_parser.c and described there.
640Sstevel@tonic-gate  *
650Sstevel@tonic-gate  * (3) For actions that are DIF expression statements, the DIF code generator
660Sstevel@tonic-gate  *     and assembler are invoked to create a finished DIFO for the statement.
670Sstevel@tonic-gate  *
680Sstevel@tonic-gate  * (4) The dtrace_prog_t data structures for the program clauses and actions
690Sstevel@tonic-gate  *     are built, containing pointers to any DIFOs created in step (3).
700Sstevel@tonic-gate  *
710Sstevel@tonic-gate  * (5) The caller invokes a routine in dt_dof.c to convert the finished program
720Sstevel@tonic-gate  *     into DOF format for use in anonymous tracing or enabling in the kernel.
730Sstevel@tonic-gate  *
740Sstevel@tonic-gate  * In the implementation, steps 2-4 are intertwined in that they are performed
750Sstevel@tonic-gate  * in order for each clause as part of a loop that executes over the clauses.
760Sstevel@tonic-gate  *
770Sstevel@tonic-gate  * The D compiler currently implements nearly no optimization.  The compiler
780Sstevel@tonic-gate  * implements integer constant folding as part of pass (1), and a set of very
790Sstevel@tonic-gate  * simple peephole optimizations as part of pass (3).  As with any C compiler,
800Sstevel@tonic-gate  * a large number of optimizations are possible on both the intermediate data
810Sstevel@tonic-gate  * structures and the generated DIF code.  These possibilities should be
820Sstevel@tonic-gate  * investigated in the context of whether they will have any substantive effect
830Sstevel@tonic-gate  * on the overall DTrace probe effect before they are undertaken.
840Sstevel@tonic-gate  */
850Sstevel@tonic-gate 
860Sstevel@tonic-gate #include <sys/types.h>
870Sstevel@tonic-gate #include <sys/wait.h>
880Sstevel@tonic-gate 
890Sstevel@tonic-gate #include <assert.h>
900Sstevel@tonic-gate #include <strings.h>
910Sstevel@tonic-gate #include <signal.h>
920Sstevel@tonic-gate #include <unistd.h>
930Sstevel@tonic-gate #include <stdlib.h>
940Sstevel@tonic-gate #include <stdio.h>
950Sstevel@tonic-gate #include <errno.h>
960Sstevel@tonic-gate #include <ucontext.h>
970Sstevel@tonic-gate #include <limits.h>
980Sstevel@tonic-gate #include <alloca.h>
990Sstevel@tonic-gate #include <ctype.h>
1000Sstevel@tonic-gate 
1010Sstevel@tonic-gate #define	_POSIX_PTHREAD_SEMANTICS
1020Sstevel@tonic-gate #include <dirent.h>
1030Sstevel@tonic-gate #undef	_POSIX_PTHREAD_SEMANTICS
1040Sstevel@tonic-gate 
1050Sstevel@tonic-gate #include <dt_module.h>
106265Smws #include <dt_program.h>
1070Sstevel@tonic-gate #include <dt_provider.h>
1080Sstevel@tonic-gate #include <dt_printf.h>
1090Sstevel@tonic-gate #include <dt_pid.h>
1100Sstevel@tonic-gate #include <dt_grammar.h>
1110Sstevel@tonic-gate #include <dt_ident.h>
1120Sstevel@tonic-gate #include <dt_string.h>
1130Sstevel@tonic-gate #include <dt_impl.h>
1140Sstevel@tonic-gate 
1150Sstevel@tonic-gate static const dtrace_diftype_t dt_void_rtype = {
1160Sstevel@tonic-gate 	DIF_TYPE_CTF, CTF_K_INTEGER, 0, 0, 0
1170Sstevel@tonic-gate };
1180Sstevel@tonic-gate 
1190Sstevel@tonic-gate static const dtrace_diftype_t dt_int_rtype = {
1200Sstevel@tonic-gate 	DIF_TYPE_CTF, CTF_K_INTEGER, 0, 0, sizeof (uint64_t)
1210Sstevel@tonic-gate };
1220Sstevel@tonic-gate 
1230Sstevel@tonic-gate /*ARGSUSED*/
1240Sstevel@tonic-gate static int
1250Sstevel@tonic-gate dt_idreset(dt_idhash_t *dhp, dt_ident_t *idp, void *ignored)
1260Sstevel@tonic-gate {
1270Sstevel@tonic-gate 	idp->di_flags &= ~(DT_IDFLG_REF | DT_IDFLG_MOD |
1280Sstevel@tonic-gate 	    DT_IDFLG_DIFR | DT_IDFLG_DIFW);
1290Sstevel@tonic-gate 	return (0);
1300Sstevel@tonic-gate }
1310Sstevel@tonic-gate 
1320Sstevel@tonic-gate /*ARGSUSED*/
1330Sstevel@tonic-gate static int
1340Sstevel@tonic-gate dt_idpragma(dt_idhash_t *dhp, dt_ident_t *idp, void *ignored)
1350Sstevel@tonic-gate {
1360Sstevel@tonic-gate 	yylineno = idp->di_lineno;
1370Sstevel@tonic-gate 	xyerror(D_PRAGMA_UNUSED, "unused #pragma %s\n", (char *)idp->di_iarg);
1380Sstevel@tonic-gate 	return (0);
1390Sstevel@tonic-gate }
1400Sstevel@tonic-gate 
1410Sstevel@tonic-gate static dtrace_stmtdesc_t *
1420Sstevel@tonic-gate dt_stmt_create(dtrace_hdl_t *dtp, dtrace_ecbdesc_t *edp,
1430Sstevel@tonic-gate     dtrace_attribute_t descattr, dtrace_attribute_t stmtattr)
1440Sstevel@tonic-gate {
1450Sstevel@tonic-gate 	dtrace_stmtdesc_t *sdp = dtrace_stmt_create(dtp, edp);
1460Sstevel@tonic-gate 
1470Sstevel@tonic-gate 	if (sdp == NULL)
1480Sstevel@tonic-gate 		longjmp(yypcb->pcb_jmpbuf, EDT_NOMEM);
1490Sstevel@tonic-gate 
1500Sstevel@tonic-gate 	assert(yypcb->pcb_stmt == NULL);
1510Sstevel@tonic-gate 	yypcb->pcb_stmt = sdp;
1520Sstevel@tonic-gate 
1530Sstevel@tonic-gate 	sdp->dtsd_descattr = descattr;
1540Sstevel@tonic-gate 	sdp->dtsd_stmtattr = stmtattr;
1550Sstevel@tonic-gate 
1560Sstevel@tonic-gate 	return (sdp);
1570Sstevel@tonic-gate }
1580Sstevel@tonic-gate 
1590Sstevel@tonic-gate static dtrace_actdesc_t *
1600Sstevel@tonic-gate dt_stmt_action(dtrace_hdl_t *dtp, dtrace_stmtdesc_t *sdp)
1610Sstevel@tonic-gate {
1620Sstevel@tonic-gate 	dtrace_actdesc_t *new;
1630Sstevel@tonic-gate 
1640Sstevel@tonic-gate 	if ((new = dtrace_stmt_action(dtp, sdp)) == NULL)
1650Sstevel@tonic-gate 		longjmp(yypcb->pcb_jmpbuf, EDT_NOMEM);
1660Sstevel@tonic-gate 
1670Sstevel@tonic-gate 	return (new);
1680Sstevel@tonic-gate }
1690Sstevel@tonic-gate 
1700Sstevel@tonic-gate /*
1710Sstevel@tonic-gate  * Utility function to determine if a given action description is destructive.
1720Sstevel@tonic-gate  * The dtdo_destructive bit is set for us by the DIF assembler (see dt_as.c).
1730Sstevel@tonic-gate  */
1740Sstevel@tonic-gate static int
1750Sstevel@tonic-gate dt_action_destructive(const dtrace_actdesc_t *ap)
1760Sstevel@tonic-gate {
1770Sstevel@tonic-gate 	return (DTRACEACT_ISDESTRUCTIVE(ap->dtad_kind) || (ap->dtad_kind ==
1780Sstevel@tonic-gate 	    DTRACEACT_DIFEXPR && ap->dtad_difo->dtdo_destructive));
1790Sstevel@tonic-gate }
1800Sstevel@tonic-gate 
1810Sstevel@tonic-gate static void
1820Sstevel@tonic-gate dt_stmt_append(dtrace_stmtdesc_t *sdp, const dt_node_t *dnp)
1830Sstevel@tonic-gate {
1840Sstevel@tonic-gate 	dtrace_ecbdesc_t *edp = sdp->dtsd_ecbdesc;
1850Sstevel@tonic-gate 	dtrace_actdesc_t *ap, *tap;
1860Sstevel@tonic-gate 	int commit = 0;
1870Sstevel@tonic-gate 	int speculate = 0;
1880Sstevel@tonic-gate 	int datarec = 0;
1890Sstevel@tonic-gate 
1900Sstevel@tonic-gate 	/*
1910Sstevel@tonic-gate 	 * Make sure that the new statement jibes with the rest of the ECB.
1920Sstevel@tonic-gate 	 */
1930Sstevel@tonic-gate 	for (ap = edp->dted_action; ap != NULL; ap = ap->dtad_next) {
1940Sstevel@tonic-gate 		if (ap->dtad_kind == DTRACEACT_COMMIT) {
1950Sstevel@tonic-gate 			if (commit) {
1960Sstevel@tonic-gate 				dnerror(dnp, D_COMM_COMM, "commit( ) may "
1970Sstevel@tonic-gate 				    "not follow commit( )\n");
1980Sstevel@tonic-gate 			}
1990Sstevel@tonic-gate 
2000Sstevel@tonic-gate 			if (datarec) {
2010Sstevel@tonic-gate 				dnerror(dnp, D_COMM_DREC, "commit( ) may "
2020Sstevel@tonic-gate 				    "not follow data-recording action(s)\n");
2030Sstevel@tonic-gate 			}
2040Sstevel@tonic-gate 
2050Sstevel@tonic-gate 			for (tap = ap; tap != NULL; tap = tap->dtad_next) {
2060Sstevel@tonic-gate 				if (!DTRACEACT_ISAGG(tap->dtad_kind))
2070Sstevel@tonic-gate 					continue;
2080Sstevel@tonic-gate 
2090Sstevel@tonic-gate 				dnerror(dnp, D_AGG_COMM, "aggregating actions "
2100Sstevel@tonic-gate 				    "may not follow commit( )\n");
2110Sstevel@tonic-gate 			}
2120Sstevel@tonic-gate 
2130Sstevel@tonic-gate 			commit = 1;
2140Sstevel@tonic-gate 			continue;
2150Sstevel@tonic-gate 		}
2160Sstevel@tonic-gate 
2170Sstevel@tonic-gate 		if (ap->dtad_kind == DTRACEACT_SPECULATE) {
2180Sstevel@tonic-gate 			if (speculate) {
2190Sstevel@tonic-gate 				dnerror(dnp, D_SPEC_SPEC, "speculate( ) may "
2200Sstevel@tonic-gate 				    "not follow speculate( )\n");
2210Sstevel@tonic-gate 			}
2220Sstevel@tonic-gate 
2230Sstevel@tonic-gate 			if (commit) {
2240Sstevel@tonic-gate 				dnerror(dnp, D_SPEC_COMM, "speculate( ) may "
2250Sstevel@tonic-gate 				    "not follow commit( )\n");
2260Sstevel@tonic-gate 			}
2270Sstevel@tonic-gate 
2280Sstevel@tonic-gate 			if (datarec) {
2290Sstevel@tonic-gate 				dnerror(dnp, D_SPEC_DREC, "speculate( ) may "
2300Sstevel@tonic-gate 				    "not follow data-recording action(s)\n");
2310Sstevel@tonic-gate 			}
2320Sstevel@tonic-gate 
2330Sstevel@tonic-gate 			speculate = 1;
2340Sstevel@tonic-gate 			continue;
2350Sstevel@tonic-gate 		}
2360Sstevel@tonic-gate 
2370Sstevel@tonic-gate 		if (DTRACEACT_ISAGG(ap->dtad_kind)) {
2380Sstevel@tonic-gate 			if (speculate) {
2390Sstevel@tonic-gate 				dnerror(dnp, D_AGG_SPEC, "aggregating actions "
2400Sstevel@tonic-gate 				    "may not follow speculate( )\n");
2410Sstevel@tonic-gate 			}
2420Sstevel@tonic-gate 
2430Sstevel@tonic-gate 			datarec = 1;
2440Sstevel@tonic-gate 			continue;
2450Sstevel@tonic-gate 		}
2460Sstevel@tonic-gate 
2470Sstevel@tonic-gate 		if (speculate) {
2480Sstevel@tonic-gate 			if (dt_action_destructive(ap)) {
2490Sstevel@tonic-gate 				dnerror(dnp, D_ACT_SPEC, "destructive actions "
2500Sstevel@tonic-gate 				    "may not follow speculate( )\n");
2510Sstevel@tonic-gate 			}
2520Sstevel@tonic-gate 
2530Sstevel@tonic-gate 			if (ap->dtad_kind == DTRACEACT_EXIT) {
2540Sstevel@tonic-gate 				dnerror(dnp, D_EXIT_SPEC, "exit( ) may not "
2550Sstevel@tonic-gate 				    "follow speculate( )\n");
2560Sstevel@tonic-gate 			}
2570Sstevel@tonic-gate 		}
2580Sstevel@tonic-gate 
2590Sstevel@tonic-gate 		/*
2600Sstevel@tonic-gate 		 * Exclude all non data-recording actions.
2610Sstevel@tonic-gate 		 */
2620Sstevel@tonic-gate 		if (dt_action_destructive(ap) ||
2630Sstevel@tonic-gate 		    ap->dtad_kind == DTRACEACT_DISCARD)
2640Sstevel@tonic-gate 			continue;
2650Sstevel@tonic-gate 
2660Sstevel@tonic-gate 		if (ap->dtad_kind == DTRACEACT_DIFEXPR &&
2670Sstevel@tonic-gate 		    ap->dtad_difo->dtdo_rtype.dtdt_kind == DIF_TYPE_CTF &&
2680Sstevel@tonic-gate 		    ap->dtad_difo->dtdo_rtype.dtdt_size == 0)
2690Sstevel@tonic-gate 			continue;
2700Sstevel@tonic-gate 
2710Sstevel@tonic-gate 		if (commit) {
2720Sstevel@tonic-gate 			dnerror(dnp, D_DREC_COMM, "data-recording actions "
2730Sstevel@tonic-gate 			    "may not follow commit( )\n");
2740Sstevel@tonic-gate 		}
2750Sstevel@tonic-gate 
2760Sstevel@tonic-gate 		if (!speculate)
2770Sstevel@tonic-gate 			datarec = 1;
2780Sstevel@tonic-gate 	}
2790Sstevel@tonic-gate 
2800Sstevel@tonic-gate 	if (dtrace_stmt_add(yypcb->pcb_hdl, yypcb->pcb_prog, sdp) != 0)
2810Sstevel@tonic-gate 		longjmp(yypcb->pcb_jmpbuf, dtrace_errno(yypcb->pcb_hdl));
2820Sstevel@tonic-gate 
2830Sstevel@tonic-gate 	if (yypcb->pcb_stmt == sdp)
2840Sstevel@tonic-gate 		yypcb->pcb_stmt = NULL;
2850Sstevel@tonic-gate }
2860Sstevel@tonic-gate 
2870Sstevel@tonic-gate /*
2880Sstevel@tonic-gate  * For the first element of an aggregation tuple or for printa(), we create a
2890Sstevel@tonic-gate  * simple DIF program that simply returns the immediate value that is the ID
2900Sstevel@tonic-gate  * of the aggregation itself.  This could be optimized in the future by
2910Sstevel@tonic-gate  * creating a new in-kernel dtad_kind that just returns an integer.
2920Sstevel@tonic-gate  */
2930Sstevel@tonic-gate static void
2940Sstevel@tonic-gate dt_action_difconst(dtrace_actdesc_t *ap, uint_t id, dtrace_actkind_t kind)
2950Sstevel@tonic-gate {
296265Smws 	dtrace_hdl_t *dtp = yypcb->pcb_hdl;
297265Smws 	dtrace_difo_t *dp = dt_zalloc(dtp, sizeof (dtrace_difo_t));
2980Sstevel@tonic-gate 
2990Sstevel@tonic-gate 	if (dp == NULL)
3000Sstevel@tonic-gate 		longjmp(yypcb->pcb_jmpbuf, EDT_NOMEM);
3010Sstevel@tonic-gate 
302265Smws 	dp->dtdo_buf = dt_alloc(dtp, sizeof (dif_instr_t) * 2);
303265Smws 	dp->dtdo_inttab = dt_alloc(dtp, sizeof (uint64_t));
3040Sstevel@tonic-gate 
3050Sstevel@tonic-gate 	if (dp->dtdo_buf == NULL || dp->dtdo_inttab == NULL) {
306265Smws 		dt_difo_free(dtp, dp);
3070Sstevel@tonic-gate 		longjmp(yypcb->pcb_jmpbuf, EDT_NOMEM);
3080Sstevel@tonic-gate 	}
3090Sstevel@tonic-gate 
3100Sstevel@tonic-gate 	dp->dtdo_buf[0] = DIF_INSTR_SETX(0, 1); /* setx	DIF_INTEGER[0], %r1 */
3110Sstevel@tonic-gate 	dp->dtdo_buf[1] = DIF_INSTR_RET(1);	/* ret	%r1 */
3120Sstevel@tonic-gate 	dp->dtdo_len = 2;
3130Sstevel@tonic-gate 	dp->dtdo_inttab[0] = id;
3140Sstevel@tonic-gate 	dp->dtdo_intlen = 1;
3150Sstevel@tonic-gate 	dp->dtdo_rtype = dt_int_rtype;
3160Sstevel@tonic-gate 
3170Sstevel@tonic-gate 	ap->dtad_difo = dp;
3180Sstevel@tonic-gate 	ap->dtad_kind = kind;
3190Sstevel@tonic-gate }
3200Sstevel@tonic-gate 
3210Sstevel@tonic-gate static void
3220Sstevel@tonic-gate dt_action_clear(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
3230Sstevel@tonic-gate {
3240Sstevel@tonic-gate 	dt_ident_t *aid;
3250Sstevel@tonic-gate 	dtrace_actdesc_t *ap;
3260Sstevel@tonic-gate 	dt_node_t *anp;
3270Sstevel@tonic-gate 
3280Sstevel@tonic-gate 	char n[DT_TYPE_NAMELEN];
3290Sstevel@tonic-gate 	int argc = 0;
3300Sstevel@tonic-gate 
3310Sstevel@tonic-gate 	for (anp = dnp->dn_args; anp != NULL; anp = anp->dn_list)
3320Sstevel@tonic-gate 		argc++; /* count up arguments for error messages below */
3330Sstevel@tonic-gate 
3340Sstevel@tonic-gate 	if (argc != 1) {
3350Sstevel@tonic-gate 		dnerror(dnp, D_CLEAR_PROTO,
3360Sstevel@tonic-gate 		    "%s( ) prototype mismatch: %d args passed, 1 expected\n",
3370Sstevel@tonic-gate 		    dnp->dn_ident->di_name, argc);
3380Sstevel@tonic-gate 	}
3390Sstevel@tonic-gate 
3400Sstevel@tonic-gate 	anp = dnp->dn_args;
3410Sstevel@tonic-gate 	assert(anp != NULL);
3420Sstevel@tonic-gate 
3430Sstevel@tonic-gate 	if (anp->dn_kind != DT_NODE_AGG) {
3440Sstevel@tonic-gate 		dnerror(dnp, D_CLEAR_AGGARG,
3450Sstevel@tonic-gate 		    "%s( ) argument #1 is incompatible with prototype:\n"
3460Sstevel@tonic-gate 		    "\tprototype: aggregation\n\t argument: %s\n",
3470Sstevel@tonic-gate 		    dnp->dn_ident->di_name,
3480Sstevel@tonic-gate 		    dt_node_type_name(anp, n, sizeof (n)));
3490Sstevel@tonic-gate 	}
3500Sstevel@tonic-gate 
3510Sstevel@tonic-gate 	aid = anp->dn_ident;
3520Sstevel@tonic-gate 
3530Sstevel@tonic-gate 	if (aid->di_gen == dtp->dt_gen && !(aid->di_flags & DT_IDFLG_MOD)) {
3540Sstevel@tonic-gate 		dnerror(dnp, D_CLEAR_AGGBAD,
3550Sstevel@tonic-gate 		    "undefined aggregation: @%s\n", aid->di_name);
3560Sstevel@tonic-gate 	}
3570Sstevel@tonic-gate 
3580Sstevel@tonic-gate 	ap = dt_stmt_action(dtp, sdp);
3590Sstevel@tonic-gate 	dt_action_difconst(ap, anp->dn_ident->di_id, DTRACEACT_LIBACT);
3600Sstevel@tonic-gate 	ap->dtad_arg = DT_ACT_CLEAR;
3610Sstevel@tonic-gate }
3620Sstevel@tonic-gate 
3630Sstevel@tonic-gate static void
3640Sstevel@tonic-gate dt_action_normalize(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
3650Sstevel@tonic-gate {
3660Sstevel@tonic-gate 	dt_ident_t *aid;
3670Sstevel@tonic-gate 	dtrace_actdesc_t *ap;
3680Sstevel@tonic-gate 	dt_node_t *anp, *normal;
3690Sstevel@tonic-gate 	int denormal = (strcmp(dnp->dn_ident->di_name, "denormalize") == 0);
3700Sstevel@tonic-gate 
3710Sstevel@tonic-gate 	char n[DT_TYPE_NAMELEN];
3720Sstevel@tonic-gate 	int argc = 0;
3730Sstevel@tonic-gate 
3740Sstevel@tonic-gate 	for (anp = dnp->dn_args; anp != NULL; anp = anp->dn_list)
3750Sstevel@tonic-gate 		argc++; /* count up arguments for error messages below */
3760Sstevel@tonic-gate 
3770Sstevel@tonic-gate 	if ((denormal && argc != 1) || (!denormal && argc != 2)) {
3780Sstevel@tonic-gate 		dnerror(dnp, D_NORMALIZE_PROTO,
3790Sstevel@tonic-gate 		    "%s( ) prototype mismatch: %d args passed, %d expected\n",
3800Sstevel@tonic-gate 		    dnp->dn_ident->di_name, argc, denormal ? 1 : 2);
3810Sstevel@tonic-gate 	}
3820Sstevel@tonic-gate 
3830Sstevel@tonic-gate 	anp = dnp->dn_args;
3840Sstevel@tonic-gate 	assert(anp != NULL);
3850Sstevel@tonic-gate 
3860Sstevel@tonic-gate 	if (anp->dn_kind != DT_NODE_AGG) {
3870Sstevel@tonic-gate 		dnerror(dnp, D_NORMALIZE_AGGARG,
3880Sstevel@tonic-gate 		    "%s( ) argument #1 is incompatible with prototype:\n"
3890Sstevel@tonic-gate 		    "\tprototype: aggregation\n\t argument: %s\n",
3900Sstevel@tonic-gate 		    dnp->dn_ident->di_name,
3910Sstevel@tonic-gate 		    dt_node_type_name(anp, n, sizeof (n)));
3920Sstevel@tonic-gate 	}
3930Sstevel@tonic-gate 
3940Sstevel@tonic-gate 	if ((normal = anp->dn_list) != NULL && !dt_node_is_scalar(normal)) {
3950Sstevel@tonic-gate 		dnerror(dnp, D_NORMALIZE_SCALAR,
3960Sstevel@tonic-gate 		    "%s( ) argument #2 must be of scalar type\n",
3970Sstevel@tonic-gate 		    dnp->dn_ident->di_name);
3980Sstevel@tonic-gate 	}
3990Sstevel@tonic-gate 
4000Sstevel@tonic-gate 	aid = anp->dn_ident;
4010Sstevel@tonic-gate 
4020Sstevel@tonic-gate 	if (aid->di_gen == dtp->dt_gen && !(aid->di_flags & DT_IDFLG_MOD)) {
4030Sstevel@tonic-gate 		dnerror(dnp, D_NORMALIZE_AGGBAD,
4040Sstevel@tonic-gate 		    "undefined aggregation: @%s\n", aid->di_name);
4050Sstevel@tonic-gate 	}
4060Sstevel@tonic-gate 
4070Sstevel@tonic-gate 	ap = dt_stmt_action(dtp, sdp);
4080Sstevel@tonic-gate 	dt_action_difconst(ap, anp->dn_ident->di_id, DTRACEACT_LIBACT);
4090Sstevel@tonic-gate 
4100Sstevel@tonic-gate 	if (denormal) {
4110Sstevel@tonic-gate 		ap->dtad_arg = DT_ACT_DENORMALIZE;
4120Sstevel@tonic-gate 		return;
4130Sstevel@tonic-gate 	}
4140Sstevel@tonic-gate 
4150Sstevel@tonic-gate 	ap->dtad_arg = DT_ACT_NORMALIZE;
4160Sstevel@tonic-gate 
4170Sstevel@tonic-gate 	assert(normal != NULL);
4180Sstevel@tonic-gate 	ap = dt_stmt_action(dtp, sdp);
4190Sstevel@tonic-gate 	dt_cg(yypcb, normal);
4200Sstevel@tonic-gate 
4210Sstevel@tonic-gate 	ap->dtad_difo = dt_as(yypcb);
4220Sstevel@tonic-gate 	ap->dtad_kind = DTRACEACT_LIBACT;
4230Sstevel@tonic-gate 	ap->dtad_arg = DT_ACT_NORMALIZE;
4240Sstevel@tonic-gate }
4250Sstevel@tonic-gate 
4260Sstevel@tonic-gate static void
4270Sstevel@tonic-gate dt_action_trunc(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
4280Sstevel@tonic-gate {
4290Sstevel@tonic-gate 	dt_ident_t *aid;
4300Sstevel@tonic-gate 	dtrace_actdesc_t *ap;
4310Sstevel@tonic-gate 	dt_node_t *anp, *trunc;
4320Sstevel@tonic-gate 
4330Sstevel@tonic-gate 	char n[DT_TYPE_NAMELEN];
4340Sstevel@tonic-gate 	int argc = 0;
4350Sstevel@tonic-gate 
4360Sstevel@tonic-gate 	for (anp = dnp->dn_args; anp != NULL; anp = anp->dn_list)
4370Sstevel@tonic-gate 		argc++; /* count up arguments for error messages below */
4380Sstevel@tonic-gate 
4390Sstevel@tonic-gate 	if (argc > 2 || argc < 1) {
4400Sstevel@tonic-gate 		dnerror(dnp, D_TRUNC_PROTO,
4410Sstevel@tonic-gate 		    "%s( ) prototype mismatch: %d args passed, %s expected\n",
4420Sstevel@tonic-gate 		    dnp->dn_ident->di_name, argc,
4430Sstevel@tonic-gate 		    argc < 1 ? "at least 1" : "no more than 2");
4440Sstevel@tonic-gate 	}
4450Sstevel@tonic-gate 
4460Sstevel@tonic-gate 	anp = dnp->dn_args;
4470Sstevel@tonic-gate 	assert(anp != NULL);
4480Sstevel@tonic-gate 	trunc = anp->dn_list;
4490Sstevel@tonic-gate 
4500Sstevel@tonic-gate 	if (anp->dn_kind != DT_NODE_AGG) {
4510Sstevel@tonic-gate 		dnerror(dnp, D_TRUNC_AGGARG,
4520Sstevel@tonic-gate 		    "%s( ) argument #1 is incompatible with prototype:\n"
4530Sstevel@tonic-gate 		    "\tprototype: aggregation\n\t argument: %s\n",
4540Sstevel@tonic-gate 		    dnp->dn_ident->di_name,
4550Sstevel@tonic-gate 		    dt_node_type_name(anp, n, sizeof (n)));
4560Sstevel@tonic-gate 	}
4570Sstevel@tonic-gate 
4580Sstevel@tonic-gate 	if (argc == 2) {
4590Sstevel@tonic-gate 		assert(trunc != NULL);
4600Sstevel@tonic-gate 		if (!dt_node_is_scalar(trunc)) {
4610Sstevel@tonic-gate 			dnerror(dnp, D_TRUNC_SCALAR,
4620Sstevel@tonic-gate 			    "%s( ) argument #2 must be of scalar type\n",
4630Sstevel@tonic-gate 			    dnp->dn_ident->di_name);
4640Sstevel@tonic-gate 		}
4650Sstevel@tonic-gate 	}
4660Sstevel@tonic-gate 
4670Sstevel@tonic-gate 	aid = anp->dn_ident;
4680Sstevel@tonic-gate 
4690Sstevel@tonic-gate 	if (aid->di_gen == dtp->dt_gen && !(aid->di_flags & DT_IDFLG_MOD)) {
4700Sstevel@tonic-gate 		dnerror(dnp, D_TRUNC_AGGBAD,
4710Sstevel@tonic-gate 		    "undefined aggregation: @%s\n", aid->di_name);
4720Sstevel@tonic-gate 	}
4730Sstevel@tonic-gate 
4740Sstevel@tonic-gate 	ap = dt_stmt_action(dtp, sdp);
4750Sstevel@tonic-gate 	dt_action_difconst(ap, anp->dn_ident->di_id, DTRACEACT_LIBACT);
4760Sstevel@tonic-gate 	ap->dtad_arg = DT_ACT_TRUNC;
4770Sstevel@tonic-gate 
4780Sstevel@tonic-gate 	ap = dt_stmt_action(dtp, sdp);
4790Sstevel@tonic-gate 
4800Sstevel@tonic-gate 	if (argc == 1) {
4810Sstevel@tonic-gate 		dt_action_difconst(ap, 0, DTRACEACT_LIBACT);
4820Sstevel@tonic-gate 	} else {
4830Sstevel@tonic-gate 		assert(trunc != NULL);
4840Sstevel@tonic-gate 		dt_cg(yypcb, trunc);
4850Sstevel@tonic-gate 		ap->dtad_difo = dt_as(yypcb);
4860Sstevel@tonic-gate 		ap->dtad_kind = DTRACEACT_LIBACT;
4870Sstevel@tonic-gate 	}
4880Sstevel@tonic-gate 
4890Sstevel@tonic-gate 	ap->dtad_arg = DT_ACT_TRUNC;
4900Sstevel@tonic-gate }
4910Sstevel@tonic-gate 
4920Sstevel@tonic-gate static void
4930Sstevel@tonic-gate dt_action_printa(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
4940Sstevel@tonic-gate {
4950Sstevel@tonic-gate 	dt_ident_t *aid, *fid;
4960Sstevel@tonic-gate 	dtrace_actdesc_t *ap;
4970Sstevel@tonic-gate 	const char *format;
4980Sstevel@tonic-gate 	dt_node_t *anp;
4990Sstevel@tonic-gate 
5000Sstevel@tonic-gate 	char n[DT_TYPE_NAMELEN];
5010Sstevel@tonic-gate 	int argc = 0, argr = 0;
5020Sstevel@tonic-gate 
5030Sstevel@tonic-gate 	for (anp = dnp->dn_args; anp != NULL; anp = anp->dn_list)
5040Sstevel@tonic-gate 		argc++; /* count up arguments for error messages below */
5050Sstevel@tonic-gate 
5060Sstevel@tonic-gate 	switch (dnp->dn_args->dn_kind) {
5070Sstevel@tonic-gate 	case DT_NODE_STRING:
5080Sstevel@tonic-gate 		format = dnp->dn_args->dn_string;
5090Sstevel@tonic-gate 		anp = dnp->dn_args->dn_list;
5100Sstevel@tonic-gate 		argr = 2;
5110Sstevel@tonic-gate 		break;
5120Sstevel@tonic-gate 	case DT_NODE_AGG:
5130Sstevel@tonic-gate 		format = NULL;
5140Sstevel@tonic-gate 		anp = dnp->dn_args;
5150Sstevel@tonic-gate 		argr = 1;
5160Sstevel@tonic-gate 		break;
5170Sstevel@tonic-gate 	default:
5180Sstevel@tonic-gate 		format = NULL;
5190Sstevel@tonic-gate 		anp = dnp->dn_args;
5200Sstevel@tonic-gate 		argr = 1;
5210Sstevel@tonic-gate 	}
5220Sstevel@tonic-gate 
5230Sstevel@tonic-gate 	if (argc != argr) {
5240Sstevel@tonic-gate 		dnerror(dnp, D_PRINTA_PROTO,
5250Sstevel@tonic-gate 		    "%s( ) prototype mismatch: %d args passed, %d expected\n",
5260Sstevel@tonic-gate 		    dnp->dn_ident->di_name, argc, argr);
5270Sstevel@tonic-gate 	}
5280Sstevel@tonic-gate 
5290Sstevel@tonic-gate 	if (anp->dn_kind != DT_NODE_AGG) {
5300Sstevel@tonic-gate 		dnerror(dnp, D_PRINTA_AGGARG,
5310Sstevel@tonic-gate 		    "%s( ) argument #%d is incompatible with prototype:\n"
5320Sstevel@tonic-gate 		    "\tprototype: aggregation\n\t argument: %s\n",
5330Sstevel@tonic-gate 		    dnp->dn_ident->di_name, argr,
5340Sstevel@tonic-gate 		    dt_node_type_name(anp, n, sizeof (n)));
5350Sstevel@tonic-gate 	}
5360Sstevel@tonic-gate 
5370Sstevel@tonic-gate 	aid = anp->dn_ident;
5380Sstevel@tonic-gate 	fid = aid->di_iarg;
5390Sstevel@tonic-gate 
5400Sstevel@tonic-gate 	if (aid->di_gen == dtp->dt_gen && !(aid->di_flags & DT_IDFLG_MOD)) {
5410Sstevel@tonic-gate 		dnerror(dnp, D_PRINTA_AGGBAD,
5420Sstevel@tonic-gate 		    "undefined aggregation: @%s\n", aid->di_name);
5430Sstevel@tonic-gate 	}
5440Sstevel@tonic-gate 
5450Sstevel@tonic-gate 	if (format != NULL) {
5460Sstevel@tonic-gate 		yylineno = dnp->dn_line;
5470Sstevel@tonic-gate 
5480Sstevel@tonic-gate 		sdp->dtsd_fmtdata = dt_printf_create(yypcb->pcb_hdl, format);
5490Sstevel@tonic-gate 		dt_printf_validate(sdp->dtsd_fmtdata,
5500Sstevel@tonic-gate 		    DT_PRINTF_AGGREGATION, dnp->dn_ident, 1,
5510Sstevel@tonic-gate 		    fid->di_id, ((dt_idsig_t *)aid->di_data)->dis_args);
5520Sstevel@tonic-gate 	}
5530Sstevel@tonic-gate 
5540Sstevel@tonic-gate 	ap = dt_stmt_action(dtp, sdp);
5550Sstevel@tonic-gate 	dt_action_difconst(ap, anp->dn_ident->di_id, DTRACEACT_PRINTA);
5560Sstevel@tonic-gate }
5570Sstevel@tonic-gate 
5580Sstevel@tonic-gate static void
5590Sstevel@tonic-gate dt_action_printflike(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp,
5600Sstevel@tonic-gate     dtrace_actkind_t kind)
5610Sstevel@tonic-gate {
5620Sstevel@tonic-gate 	dt_node_t *anp, *arg1;
5630Sstevel@tonic-gate 	dtrace_actdesc_t *ap = NULL;
5640Sstevel@tonic-gate 	char n[DT_TYPE_NAMELEN], *str;
5650Sstevel@tonic-gate 
5660Sstevel@tonic-gate 	assert(DTRACEACT_ISPRINTFLIKE(kind));
5670Sstevel@tonic-gate 
5680Sstevel@tonic-gate 	if (dnp->dn_args->dn_kind != DT_NODE_STRING) {
5690Sstevel@tonic-gate 		dnerror(dnp, D_PRINTF_ARG_FMT,
5700Sstevel@tonic-gate 		    "%s( ) argument #1 is incompatible with prototype:\n"
5710Sstevel@tonic-gate 		    "\tprototype: string constant\n\t argument: %s\n",
5720Sstevel@tonic-gate 		    dnp->dn_ident->di_name,
5730Sstevel@tonic-gate 		    dt_node_type_name(dnp->dn_args, n, sizeof (n)));
5740Sstevel@tonic-gate 	}
5750Sstevel@tonic-gate 
5760Sstevel@tonic-gate 	arg1 = dnp->dn_args->dn_list;
5770Sstevel@tonic-gate 	yylineno = dnp->dn_line;
5780Sstevel@tonic-gate 	str = dnp->dn_args->dn_string;
5790Sstevel@tonic-gate 
5800Sstevel@tonic-gate 
5810Sstevel@tonic-gate 	/*
5820Sstevel@tonic-gate 	 * If this is an freopen(), we use an empty string to denote that
5830Sstevel@tonic-gate 	 * stdout should be restored.  For other printf()-like actions, an
5840Sstevel@tonic-gate 	 * empty format string is illegal:  an empty format string would
5850Sstevel@tonic-gate 	 * result in malformed DOF, and the compiler thus flags an empty
5860Sstevel@tonic-gate 	 * format string as a compile-time error.  To avoid propagating the
5870Sstevel@tonic-gate 	 * freopen() special case throughout the system, we simply transpose
5880Sstevel@tonic-gate 	 * an empty string into a sentinel string (DT_FREOPEN_RESTORE) that
5890Sstevel@tonic-gate 	 * denotes that stdout should be restored.
5900Sstevel@tonic-gate 	 */
5910Sstevel@tonic-gate 	if (kind == DTRACEACT_FREOPEN) {
5920Sstevel@tonic-gate 		if (strcmp(str, DT_FREOPEN_RESTORE) == 0) {
5930Sstevel@tonic-gate 			/*
5940Sstevel@tonic-gate 			 * Our sentinel is always an invalid argument to
5950Sstevel@tonic-gate 			 * freopen(), but if it's been manually specified, we
5960Sstevel@tonic-gate 			 * must fail now instead of when the freopen() is
5970Sstevel@tonic-gate 			 * actually evaluated.
5980Sstevel@tonic-gate 			 */
5990Sstevel@tonic-gate 			dnerror(dnp, D_FREOPEN_INVALID,
6000Sstevel@tonic-gate 			    "%s( ) argument #1 cannot be \"%s\"\n",
6010Sstevel@tonic-gate 			    dnp->dn_ident->di_name, DT_FREOPEN_RESTORE);
6020Sstevel@tonic-gate 		}
6030Sstevel@tonic-gate 
6040Sstevel@tonic-gate 		if (str[0] == '\0')
6050Sstevel@tonic-gate 			str = DT_FREOPEN_RESTORE;
6060Sstevel@tonic-gate 	}
6070Sstevel@tonic-gate 
608265Smws 	sdp->dtsd_fmtdata = dt_printf_create(dtp, str);
6090Sstevel@tonic-gate 
6100Sstevel@tonic-gate 	dt_printf_validate(sdp->dtsd_fmtdata, DT_PRINTF_EXACTLEN,
6110Sstevel@tonic-gate 	    dnp->dn_ident, 1, DTRACEACT_AGGREGATION, arg1);
6120Sstevel@tonic-gate 
6130Sstevel@tonic-gate 	if (arg1 == NULL) {
6140Sstevel@tonic-gate 		dif_instr_t *dbuf;
6150Sstevel@tonic-gate 		dtrace_difo_t *dp;
6160Sstevel@tonic-gate 
617265Smws 		if ((dbuf = dt_alloc(dtp, sizeof (dif_instr_t))) == NULL ||
618265Smws 		    (dp = dt_zalloc(dtp, sizeof (dtrace_difo_t))) == NULL) {
619265Smws 			dt_free(dtp, dbuf);
6200Sstevel@tonic-gate 			longjmp(yypcb->pcb_jmpbuf, EDT_NOMEM);
6210Sstevel@tonic-gate 		}
6220Sstevel@tonic-gate 
6230Sstevel@tonic-gate 		dbuf[0] = DIF_INSTR_RET(DIF_REG_R0); /* ret %r0 */
6240Sstevel@tonic-gate 
6250Sstevel@tonic-gate 		dp->dtdo_buf = dbuf;
6260Sstevel@tonic-gate 		dp->dtdo_len = 1;
6270Sstevel@tonic-gate 		dp->dtdo_rtype = dt_int_rtype;
6280Sstevel@tonic-gate 
6290Sstevel@tonic-gate 		ap = dt_stmt_action(dtp, sdp);
6300Sstevel@tonic-gate 		ap->dtad_difo = dp;
6310Sstevel@tonic-gate 		ap->dtad_kind = kind;
6320Sstevel@tonic-gate 		return;
6330Sstevel@tonic-gate 	}
6340Sstevel@tonic-gate 
6350Sstevel@tonic-gate 	for (anp = arg1; anp != NULL; anp = anp->dn_list) {
6360Sstevel@tonic-gate 		ap = dt_stmt_action(dtp, sdp);
6370Sstevel@tonic-gate 		dt_cg(yypcb, anp);
6380Sstevel@tonic-gate 		ap->dtad_difo = dt_as(yypcb);
6390Sstevel@tonic-gate 		ap->dtad_kind = kind;
6400Sstevel@tonic-gate 	}
6410Sstevel@tonic-gate }
6420Sstevel@tonic-gate 
6430Sstevel@tonic-gate static void
6440Sstevel@tonic-gate dt_action_trace(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
6450Sstevel@tonic-gate {
6460Sstevel@tonic-gate 	dtrace_actdesc_t *ap = dt_stmt_action(dtp, sdp);
6470Sstevel@tonic-gate 
6480Sstevel@tonic-gate 	if (dt_node_is_void(dnp->dn_args)) {
6490Sstevel@tonic-gate 		dnerror(dnp->dn_args, D_TRACE_VOID,
6500Sstevel@tonic-gate 		    "trace( ) may not be applied to a void expression\n");
6510Sstevel@tonic-gate 	}
6520Sstevel@tonic-gate 
6530Sstevel@tonic-gate 	if (dt_node_is_dynamic(dnp->dn_args)) {
6540Sstevel@tonic-gate 		dnerror(dnp->dn_args, D_TRACE_DYN,
6550Sstevel@tonic-gate 		    "trace( ) may not be applied to a dynamic expression\n");
6560Sstevel@tonic-gate 	}
6570Sstevel@tonic-gate 
6580Sstevel@tonic-gate 	dt_cg(yypcb, dnp->dn_args);
6590Sstevel@tonic-gate 	ap->dtad_difo = dt_as(yypcb);
6600Sstevel@tonic-gate 	ap->dtad_kind = DTRACEACT_DIFEXPR;
6610Sstevel@tonic-gate }
6620Sstevel@tonic-gate 
6630Sstevel@tonic-gate static void
6640Sstevel@tonic-gate dt_action_tracemem(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
6650Sstevel@tonic-gate {
6660Sstevel@tonic-gate 	dtrace_actdesc_t *ap = dt_stmt_action(dtp, sdp);
6670Sstevel@tonic-gate 
6680Sstevel@tonic-gate 	dt_node_t *addr = dnp->dn_args;
6690Sstevel@tonic-gate 	dt_node_t *size = dnp->dn_args->dn_list;
6700Sstevel@tonic-gate 
6710Sstevel@tonic-gate 	char n[DT_TYPE_NAMELEN];
6720Sstevel@tonic-gate 
6730Sstevel@tonic-gate 	if (dt_node_is_integer(addr) == 0 && dt_node_is_pointer(addr) == 0) {
6740Sstevel@tonic-gate 		dnerror(addr, D_TRACEMEM_ADDR,
6750Sstevel@tonic-gate 		    "tracemem( ) argument #1 is incompatible with "
6760Sstevel@tonic-gate 		    "prototype:\n\tprototype: pointer or integer\n"
6770Sstevel@tonic-gate 		    "\t argument: %s\n",
6780Sstevel@tonic-gate 		    dt_node_type_name(addr, n, sizeof (n)));
6790Sstevel@tonic-gate 	}
6800Sstevel@tonic-gate 
6810Sstevel@tonic-gate 	if (dt_node_is_posconst(size) == 0) {
6820Sstevel@tonic-gate 		dnerror(size, D_TRACEMEM_SIZE, "tracemem( ) argument #2 must "
6830Sstevel@tonic-gate 		    "be a non-zero positive integral constant expression\n");
6840Sstevel@tonic-gate 	}
6850Sstevel@tonic-gate 
6860Sstevel@tonic-gate 	dt_cg(yypcb, addr);
6870Sstevel@tonic-gate 	ap->dtad_difo = dt_as(yypcb);
6880Sstevel@tonic-gate 	ap->dtad_kind = DTRACEACT_DIFEXPR;
6890Sstevel@tonic-gate 
6900Sstevel@tonic-gate 	ap->dtad_difo->dtdo_rtype.dtdt_flags |= DIF_TF_BYREF;
6910Sstevel@tonic-gate 	ap->dtad_difo->dtdo_rtype.dtdt_size = size->dn_value;
6920Sstevel@tonic-gate }
6930Sstevel@tonic-gate 
6940Sstevel@tonic-gate static void
6950Sstevel@tonic-gate dt_action_stack_args(dtrace_hdl_t *dtp, dtrace_actdesc_t *ap, dt_node_t *arg0)
6960Sstevel@tonic-gate {
6970Sstevel@tonic-gate 	ap->dtad_kind = DTRACEACT_STACK;
6980Sstevel@tonic-gate 
6990Sstevel@tonic-gate 	if (dtp->dt_options[DTRACEOPT_STACKFRAMES] != DTRACEOPT_UNSET) {
7000Sstevel@tonic-gate 		ap->dtad_arg = dtp->dt_options[DTRACEOPT_STACKFRAMES];
7010Sstevel@tonic-gate 	} else {
7020Sstevel@tonic-gate 		ap->dtad_arg = 0;
7030Sstevel@tonic-gate 	}
7040Sstevel@tonic-gate 
7050Sstevel@tonic-gate 	if (arg0 != NULL) {
7060Sstevel@tonic-gate 		if (arg0->dn_list != NULL) {
7070Sstevel@tonic-gate 			dnerror(arg0, D_STACK_PROTO, "stack( ) prototype "
7080Sstevel@tonic-gate 			    "mismatch: too many arguments\n");
7090Sstevel@tonic-gate 		}
7100Sstevel@tonic-gate 
7110Sstevel@tonic-gate 		if (dt_node_is_posconst(arg0) == 0) {
7120Sstevel@tonic-gate 			dnerror(arg0, D_STACK_SIZE, "stack( ) size must be a "
7130Sstevel@tonic-gate 			    "non-zero positive integral constant expression\n");
7140Sstevel@tonic-gate 		}
7150Sstevel@tonic-gate 
7160Sstevel@tonic-gate 		ap->dtad_arg = arg0->dn_value;
7170Sstevel@tonic-gate 	}
7180Sstevel@tonic-gate }
7190Sstevel@tonic-gate 
7200Sstevel@tonic-gate static void
7210Sstevel@tonic-gate dt_action_stack(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
7220Sstevel@tonic-gate {
7230Sstevel@tonic-gate 	dtrace_actdesc_t *ap = dt_stmt_action(dtp, sdp);
7240Sstevel@tonic-gate 	dt_action_stack_args(dtp, ap, dnp->dn_args);
7250Sstevel@tonic-gate }
7260Sstevel@tonic-gate 
7270Sstevel@tonic-gate static void
7280Sstevel@tonic-gate dt_action_ustack_args(dtrace_hdl_t *dtp, dtrace_actdesc_t *ap, dt_node_t *dnp)
7290Sstevel@tonic-gate {
7300Sstevel@tonic-gate 	uint32_t nframes = 0;
7310Sstevel@tonic-gate 	uint32_t strsize = 0;	/* default string table size */
7320Sstevel@tonic-gate 	dt_node_t *arg0 = dnp->dn_args;
7330Sstevel@tonic-gate 	dt_node_t *arg1 = arg0 != NULL ? arg0->dn_list : NULL;
7340Sstevel@tonic-gate 
7350Sstevel@tonic-gate 	assert(dnp->dn_ident->di_id == DT_ACT_JSTACK ||
7360Sstevel@tonic-gate 	    dnp->dn_ident->di_id == DT_ACT_USTACK);
7370Sstevel@tonic-gate 
7380Sstevel@tonic-gate 	if (dnp->dn_ident->di_id == DT_ACT_JSTACK) {
7390Sstevel@tonic-gate 		if (dtp->dt_options[DTRACEOPT_JSTACKFRAMES] != DTRACEOPT_UNSET)
7400Sstevel@tonic-gate 			nframes = dtp->dt_options[DTRACEOPT_JSTACKFRAMES];
7410Sstevel@tonic-gate 
7420Sstevel@tonic-gate 		if (dtp->dt_options[DTRACEOPT_JSTACKSTRSIZE] != DTRACEOPT_UNSET)
7430Sstevel@tonic-gate 			strsize = dtp->dt_options[DTRACEOPT_JSTACKSTRSIZE];
7440Sstevel@tonic-gate 
7450Sstevel@tonic-gate 		ap->dtad_kind = DTRACEACT_JSTACK;
7460Sstevel@tonic-gate 	} else {
7470Sstevel@tonic-gate 		assert(dnp->dn_ident->di_id == DT_ACT_USTACK);
7480Sstevel@tonic-gate 
7490Sstevel@tonic-gate 		if (dtp->dt_options[DTRACEOPT_USTACKFRAMES] != DTRACEOPT_UNSET)
7500Sstevel@tonic-gate 			nframes = dtp->dt_options[DTRACEOPT_USTACKFRAMES];
7510Sstevel@tonic-gate 
7520Sstevel@tonic-gate 		ap->dtad_kind = DTRACEACT_USTACK;
7530Sstevel@tonic-gate 	}
7540Sstevel@tonic-gate 
7550Sstevel@tonic-gate 	if (arg0 != NULL) {
7560Sstevel@tonic-gate 		if (!dt_node_is_posconst(arg0)) {
7570Sstevel@tonic-gate 			dnerror(arg0, D_USTACK_FRAMES, "ustack( ) argument #1 "
7580Sstevel@tonic-gate 			    "must be a non-zero positive integer constant\n");
7590Sstevel@tonic-gate 		}
7600Sstevel@tonic-gate 		nframes = (uint32_t)arg0->dn_value;
7610Sstevel@tonic-gate 	}
7620Sstevel@tonic-gate 
7630Sstevel@tonic-gate 	if (arg1 != NULL) {
7640Sstevel@tonic-gate 		if (arg1->dn_kind != DT_NODE_INT ||
7650Sstevel@tonic-gate 		    ((arg1->dn_flags & DT_NF_SIGNED) &&
7660Sstevel@tonic-gate 		    (int64_t)arg1->dn_value < 0)) {
7670Sstevel@tonic-gate 			dnerror(arg1, D_USTACK_STRSIZE, "ustack( ) argument #2 "
7680Sstevel@tonic-gate 			    "must be a positive integer constant\n");
7690Sstevel@tonic-gate 		}
7700Sstevel@tonic-gate 
7710Sstevel@tonic-gate 		if (arg1->dn_list != NULL) {
7720Sstevel@tonic-gate 			dnerror(arg1, D_USTACK_PROTO, "ustack( ) prototype "
7730Sstevel@tonic-gate 			    "mismatch: too many arguments\n");
7740Sstevel@tonic-gate 		}
7750Sstevel@tonic-gate 
7760Sstevel@tonic-gate 		strsize = (uint32_t)arg1->dn_value;
7770Sstevel@tonic-gate 	}
7780Sstevel@tonic-gate 
7790Sstevel@tonic-gate 	ap->dtad_arg = DTRACE_USTACK_ARG(nframes, strsize);
7800Sstevel@tonic-gate }
7810Sstevel@tonic-gate 
7820Sstevel@tonic-gate static void
7830Sstevel@tonic-gate dt_action_ustack(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
7840Sstevel@tonic-gate {
7850Sstevel@tonic-gate 	dtrace_actdesc_t *ap = dt_stmt_action(dtp, sdp);
7860Sstevel@tonic-gate 	dt_action_ustack_args(dtp, ap, dnp);
7870Sstevel@tonic-gate }
7880Sstevel@tonic-gate 
789*457Sbmc static void
790*457Sbmc dt_action_setopt(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
791*457Sbmc {
792*457Sbmc 	dtrace_actdesc_t *ap;
793*457Sbmc 	dt_node_t *arg0, *arg1;
794*457Sbmc 
795*457Sbmc 	/*
796*457Sbmc 	 * The prototype guarantees that we are called with either one or
797*457Sbmc 	 * two arguments, and that any arguments that are present are strings.
798*457Sbmc 	 */
799*457Sbmc 	arg0 = dnp->dn_args;
800*457Sbmc 	arg1 = arg0->dn_list;
801*457Sbmc 
802*457Sbmc 	ap = dt_stmt_action(dtp, sdp);
803*457Sbmc 	dt_cg(yypcb, arg0);
804*457Sbmc 	ap->dtad_difo = dt_as(yypcb);
805*457Sbmc 	ap->dtad_kind = DTRACEACT_LIBACT;
806*457Sbmc 	ap->dtad_arg = DT_ACT_SETOPT;
807*457Sbmc 
808*457Sbmc 	ap = dt_stmt_action(dtp, sdp);
809*457Sbmc 
810*457Sbmc 	if (arg1 == NULL) {
811*457Sbmc 		dt_action_difconst(ap, 0, DTRACEACT_LIBACT);
812*457Sbmc 	} else {
813*457Sbmc 		dt_cg(yypcb, arg1);
814*457Sbmc 		ap->dtad_difo = dt_as(yypcb);
815*457Sbmc 		ap->dtad_kind = DTRACEACT_LIBACT;
816*457Sbmc 	}
817*457Sbmc 
818*457Sbmc 	ap->dtad_arg = DT_ACT_SETOPT;
819*457Sbmc }
820*457Sbmc 
821*457Sbmc /*ARGSUSED*/
822*457Sbmc static void
823*457Sbmc dt_action_symmod_args(dtrace_hdl_t *dtp, dtrace_actdesc_t *ap,
824*457Sbmc     dt_node_t *dnp, dtrace_actkind_t kind)
825*457Sbmc {
826*457Sbmc 	assert(kind == DTRACEACT_SYM || kind == DTRACEACT_MOD ||
827*457Sbmc 	    kind == DTRACEACT_USYM || kind == DTRACEACT_UMOD ||
828*457Sbmc 	    kind == DTRACEACT_UADDR);
829*457Sbmc 
830*457Sbmc 	dt_cg(yypcb, dnp);
831*457Sbmc 	ap->dtad_difo = dt_as(yypcb);
832*457Sbmc 	ap->dtad_kind = kind;
833*457Sbmc 	ap->dtad_difo->dtdo_rtype.dtdt_size = sizeof (uint64_t);
834*457Sbmc }
835*457Sbmc 
836*457Sbmc static void
837*457Sbmc dt_action_symmod(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp,
838*457Sbmc     dtrace_actkind_t kind)
839*457Sbmc {
840*457Sbmc 	dtrace_actdesc_t *ap = dt_stmt_action(dtp, sdp);
841*457Sbmc 	dt_action_symmod_args(dtp, ap, dnp->dn_args, kind);
842*457Sbmc }
843*457Sbmc 
8440Sstevel@tonic-gate /*ARGSUSED*/
8450Sstevel@tonic-gate static void
8460Sstevel@tonic-gate dt_action_ftruncate(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
8470Sstevel@tonic-gate {
8480Sstevel@tonic-gate 	dtrace_actdesc_t *ap = dt_stmt_action(dtp, sdp);
8490Sstevel@tonic-gate 
8500Sstevel@tonic-gate 	/*
8510Sstevel@tonic-gate 	 * Library actions need a DIFO that serves as an argument.  As
8520Sstevel@tonic-gate 	 * ftruncate() doesn't take an argument, we generate the constant 0
8530Sstevel@tonic-gate 	 * in a DIFO; this constant will be ignored when the ftruncate() is
8540Sstevel@tonic-gate 	 * processed.
8550Sstevel@tonic-gate 	 */
8560Sstevel@tonic-gate 	dt_action_difconst(ap, 0, DTRACEACT_LIBACT);
8570Sstevel@tonic-gate 	ap->dtad_arg = DT_ACT_FTRUNCATE;
8580Sstevel@tonic-gate }
8590Sstevel@tonic-gate 
8600Sstevel@tonic-gate /*ARGSUSED*/
8610Sstevel@tonic-gate static void
8620Sstevel@tonic-gate dt_action_stop(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
8630Sstevel@tonic-gate {
8640Sstevel@tonic-gate 	dtrace_actdesc_t *ap = dt_stmt_action(dtp, sdp);
8650Sstevel@tonic-gate 
8660Sstevel@tonic-gate 	ap->dtad_kind = DTRACEACT_STOP;
8670Sstevel@tonic-gate 	ap->dtad_arg = 0;
8680Sstevel@tonic-gate }
8690Sstevel@tonic-gate 
8700Sstevel@tonic-gate /*ARGSUSED*/
8710Sstevel@tonic-gate static void
8720Sstevel@tonic-gate dt_action_breakpoint(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
8730Sstevel@tonic-gate {
8740Sstevel@tonic-gate 	dtrace_actdesc_t *ap = dt_stmt_action(dtp, sdp);
8750Sstevel@tonic-gate 
8760Sstevel@tonic-gate 	ap->dtad_kind = DTRACEACT_BREAKPOINT;
8770Sstevel@tonic-gate 	ap->dtad_arg = 0;
8780Sstevel@tonic-gate }
8790Sstevel@tonic-gate 
8800Sstevel@tonic-gate /*ARGSUSED*/
8810Sstevel@tonic-gate static void
8820Sstevel@tonic-gate dt_action_panic(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
8830Sstevel@tonic-gate {
8840Sstevel@tonic-gate 	dtrace_actdesc_t *ap = dt_stmt_action(dtp, sdp);
8850Sstevel@tonic-gate 
8860Sstevel@tonic-gate 	ap->dtad_kind = DTRACEACT_PANIC;
8870Sstevel@tonic-gate 	ap->dtad_arg = 0;
8880Sstevel@tonic-gate }
8890Sstevel@tonic-gate 
8900Sstevel@tonic-gate static void
8910Sstevel@tonic-gate dt_action_chill(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
8920Sstevel@tonic-gate {
8930Sstevel@tonic-gate 	dtrace_actdesc_t *ap = dt_stmt_action(dtp, sdp);
8940Sstevel@tonic-gate 
8950Sstevel@tonic-gate 	dt_cg(yypcb, dnp->dn_args);
8960Sstevel@tonic-gate 	ap->dtad_difo = dt_as(yypcb);
8970Sstevel@tonic-gate 	ap->dtad_kind = DTRACEACT_CHILL;
8980Sstevel@tonic-gate }
8990Sstevel@tonic-gate 
9000Sstevel@tonic-gate static void
9010Sstevel@tonic-gate dt_action_raise(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
9020Sstevel@tonic-gate {
9030Sstevel@tonic-gate 	dtrace_actdesc_t *ap = dt_stmt_action(dtp, sdp);
9040Sstevel@tonic-gate 
9050Sstevel@tonic-gate 	dt_cg(yypcb, dnp->dn_args);
9060Sstevel@tonic-gate 	ap->dtad_difo = dt_as(yypcb);
9070Sstevel@tonic-gate 	ap->dtad_kind = DTRACEACT_RAISE;
9080Sstevel@tonic-gate }
9090Sstevel@tonic-gate 
9100Sstevel@tonic-gate static void
9110Sstevel@tonic-gate dt_action_exit(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
9120Sstevel@tonic-gate {
9130Sstevel@tonic-gate 	dtrace_actdesc_t *ap = dt_stmt_action(dtp, sdp);
9140Sstevel@tonic-gate 
9150Sstevel@tonic-gate 	dt_cg(yypcb, dnp->dn_args);
9160Sstevel@tonic-gate 	ap->dtad_difo = dt_as(yypcb);
9170Sstevel@tonic-gate 	ap->dtad_kind = DTRACEACT_EXIT;
9180Sstevel@tonic-gate 	ap->dtad_difo->dtdo_rtype.dtdt_size = sizeof (int);
9190Sstevel@tonic-gate }
9200Sstevel@tonic-gate 
9210Sstevel@tonic-gate static void
9220Sstevel@tonic-gate dt_action_speculate(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
9230Sstevel@tonic-gate {
9240Sstevel@tonic-gate 	dtrace_actdesc_t *ap = dt_stmt_action(dtp, sdp);
9250Sstevel@tonic-gate 
9260Sstevel@tonic-gate 	dt_cg(yypcb, dnp->dn_args);
9270Sstevel@tonic-gate 	ap->dtad_difo = dt_as(yypcb);
9280Sstevel@tonic-gate 	ap->dtad_kind = DTRACEACT_SPECULATE;
9290Sstevel@tonic-gate }
9300Sstevel@tonic-gate 
9310Sstevel@tonic-gate static void
9320Sstevel@tonic-gate dt_action_commit(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
9330Sstevel@tonic-gate {
9340Sstevel@tonic-gate 	dtrace_actdesc_t *ap = dt_stmt_action(dtp, sdp);
9350Sstevel@tonic-gate 
9360Sstevel@tonic-gate 	dt_cg(yypcb, dnp->dn_args);
9370Sstevel@tonic-gate 	ap->dtad_difo = dt_as(yypcb);
9380Sstevel@tonic-gate 	ap->dtad_kind = DTRACEACT_COMMIT;
9390Sstevel@tonic-gate }
9400Sstevel@tonic-gate 
9410Sstevel@tonic-gate static void
9420Sstevel@tonic-gate dt_action_discard(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
9430Sstevel@tonic-gate {
9440Sstevel@tonic-gate 	dtrace_actdesc_t *ap = dt_stmt_action(dtp, sdp);
9450Sstevel@tonic-gate 
9460Sstevel@tonic-gate 	dt_cg(yypcb, dnp->dn_args);
9470Sstevel@tonic-gate 	ap->dtad_difo = dt_as(yypcb);
9480Sstevel@tonic-gate 	ap->dtad_kind = DTRACEACT_DISCARD;
9490Sstevel@tonic-gate }
9500Sstevel@tonic-gate 
9510Sstevel@tonic-gate static void
9520Sstevel@tonic-gate dt_compile_fun(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
9530Sstevel@tonic-gate {
9540Sstevel@tonic-gate 	switch (dnp->dn_expr->dn_ident->di_id) {
9550Sstevel@tonic-gate 	case DT_ACT_BREAKPOINT:
9560Sstevel@tonic-gate 		dt_action_breakpoint(dtp, dnp->dn_expr, sdp);
9570Sstevel@tonic-gate 		break;
9580Sstevel@tonic-gate 	case DT_ACT_CHILL:
9590Sstevel@tonic-gate 		dt_action_chill(dtp, dnp->dn_expr, sdp);
9600Sstevel@tonic-gate 		break;
9610Sstevel@tonic-gate 	case DT_ACT_CLEAR:
9620Sstevel@tonic-gate 		dt_action_clear(dtp, dnp->dn_expr, sdp);
9630Sstevel@tonic-gate 		break;
9640Sstevel@tonic-gate 	case DT_ACT_COMMIT:
9650Sstevel@tonic-gate 		dt_action_commit(dtp, dnp->dn_expr, sdp);
9660Sstevel@tonic-gate 		break;
9670Sstevel@tonic-gate 	case DT_ACT_DENORMALIZE:
9680Sstevel@tonic-gate 		dt_action_normalize(dtp, dnp->dn_expr, sdp);
9690Sstevel@tonic-gate 		break;
9700Sstevel@tonic-gate 	case DT_ACT_DISCARD:
9710Sstevel@tonic-gate 		dt_action_discard(dtp, dnp->dn_expr, sdp);
9720Sstevel@tonic-gate 		break;
9730Sstevel@tonic-gate 	case DT_ACT_EXIT:
9740Sstevel@tonic-gate 		dt_action_exit(dtp, dnp->dn_expr, sdp);
9750Sstevel@tonic-gate 		break;
9760Sstevel@tonic-gate 	case DT_ACT_FREOPEN:
9770Sstevel@tonic-gate 		dt_action_printflike(dtp, dnp->dn_expr, sdp, DTRACEACT_FREOPEN);
9780Sstevel@tonic-gate 		break;
9790Sstevel@tonic-gate 	case DT_ACT_FTRUNCATE:
9800Sstevel@tonic-gate 		dt_action_ftruncate(dtp, dnp->dn_expr, sdp);
9810Sstevel@tonic-gate 		break;
982*457Sbmc 	case DT_ACT_MOD:
983*457Sbmc 		dt_action_symmod(dtp, dnp->dn_expr, sdp, DTRACEACT_MOD);
984*457Sbmc 		break;
9850Sstevel@tonic-gate 	case DT_ACT_NORMALIZE:
9860Sstevel@tonic-gate 		dt_action_normalize(dtp, dnp->dn_expr, sdp);
9870Sstevel@tonic-gate 		break;
9880Sstevel@tonic-gate 	case DT_ACT_PANIC:
9890Sstevel@tonic-gate 		dt_action_panic(dtp, dnp->dn_expr, sdp);
9900Sstevel@tonic-gate 		break;
9910Sstevel@tonic-gate 	case DT_ACT_PRINTA:
9920Sstevel@tonic-gate 		dt_action_printa(dtp, dnp->dn_expr, sdp);
9930Sstevel@tonic-gate 		break;
9940Sstevel@tonic-gate 	case DT_ACT_PRINTF:
9950Sstevel@tonic-gate 		dt_action_printflike(dtp, dnp->dn_expr, sdp, DTRACEACT_PRINTF);
9960Sstevel@tonic-gate 		break;
9970Sstevel@tonic-gate 	case DT_ACT_RAISE:
9980Sstevel@tonic-gate 		dt_action_raise(dtp, dnp->dn_expr, sdp);
9990Sstevel@tonic-gate 		break;
1000*457Sbmc 	case DT_ACT_SETOPT:
1001*457Sbmc 		dt_action_setopt(dtp, dnp->dn_expr, sdp);
1002*457Sbmc 		break;
10030Sstevel@tonic-gate 	case DT_ACT_SPECULATE:
10040Sstevel@tonic-gate 		dt_action_speculate(dtp, dnp->dn_expr, sdp);
10050Sstevel@tonic-gate 		break;
10060Sstevel@tonic-gate 	case DT_ACT_STACK:
10070Sstevel@tonic-gate 		dt_action_stack(dtp, dnp->dn_expr, sdp);
10080Sstevel@tonic-gate 		break;
10090Sstevel@tonic-gate 	case DT_ACT_STOP:
10100Sstevel@tonic-gate 		dt_action_stop(dtp, dnp->dn_expr, sdp);
10110Sstevel@tonic-gate 		break;
1012*457Sbmc 	case DT_ACT_SYM:
1013*457Sbmc 		dt_action_symmod(dtp, dnp->dn_expr, sdp, DTRACEACT_SYM);
1014*457Sbmc 		break;
10150Sstevel@tonic-gate 	case DT_ACT_SYSTEM:
10160Sstevel@tonic-gate 		dt_action_printflike(dtp, dnp->dn_expr, sdp, DTRACEACT_SYSTEM);
10170Sstevel@tonic-gate 		break;
10180Sstevel@tonic-gate 	case DT_ACT_TRACE:
10190Sstevel@tonic-gate 		dt_action_trace(dtp, dnp->dn_expr, sdp);
10200Sstevel@tonic-gate 		break;
10210Sstevel@tonic-gate 	case DT_ACT_TRACEMEM:
10220Sstevel@tonic-gate 		dt_action_tracemem(dtp, dnp->dn_expr, sdp);
10230Sstevel@tonic-gate 		break;
10240Sstevel@tonic-gate 	case DT_ACT_TRUNC:
10250Sstevel@tonic-gate 		dt_action_trunc(dtp, dnp->dn_expr, sdp);
10260Sstevel@tonic-gate 		break;
1027*457Sbmc 	case DT_ACT_UADDR:
1028*457Sbmc 		dt_action_symmod(dtp, dnp->dn_expr, sdp, DTRACEACT_UADDR);
1029*457Sbmc 		break;
1030*457Sbmc 	case DT_ACT_UMOD:
1031*457Sbmc 		dt_action_symmod(dtp, dnp->dn_expr, sdp, DTRACEACT_UMOD);
1032*457Sbmc 		break;
1033*457Sbmc 	case DT_ACT_USYM:
1034*457Sbmc 		dt_action_symmod(dtp, dnp->dn_expr, sdp, DTRACEACT_USYM);
1035*457Sbmc 		break;
10360Sstevel@tonic-gate 	case DT_ACT_USTACK:
10370Sstevel@tonic-gate 	case DT_ACT_JSTACK:
10380Sstevel@tonic-gate 		dt_action_ustack(dtp, dnp->dn_expr, sdp);
10390Sstevel@tonic-gate 		break;
10400Sstevel@tonic-gate 	default:
10410Sstevel@tonic-gate 		dnerror(dnp->dn_expr, D_UNKNOWN, "tracing function %s( ) is "
10420Sstevel@tonic-gate 		    "not yet supported\n", dnp->dn_expr->dn_ident->di_name);
10430Sstevel@tonic-gate 	}
10440Sstevel@tonic-gate }
10450Sstevel@tonic-gate 
10460Sstevel@tonic-gate static void
10470Sstevel@tonic-gate dt_compile_exp(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
10480Sstevel@tonic-gate {
10490Sstevel@tonic-gate 	dtrace_actdesc_t *ap = dt_stmt_action(dtp, sdp);
10500Sstevel@tonic-gate 
10510Sstevel@tonic-gate 	dt_cg(yypcb, dnp->dn_expr);
10520Sstevel@tonic-gate 	ap->dtad_difo = dt_as(yypcb);
10530Sstevel@tonic-gate 	ap->dtad_difo->dtdo_rtype = dt_void_rtype;
10540Sstevel@tonic-gate 	ap->dtad_kind = DTRACEACT_DIFEXPR;
10550Sstevel@tonic-gate }
10560Sstevel@tonic-gate 
10570Sstevel@tonic-gate static void
10580Sstevel@tonic-gate dt_compile_agg(dtrace_hdl_t *dtp, dt_node_t *dnp, dtrace_stmtdesc_t *sdp)
10590Sstevel@tonic-gate {
10600Sstevel@tonic-gate 	dt_ident_t *aid, *fid;
1061*457Sbmc 	dt_node_t *anp, *incr = NULL;
10620Sstevel@tonic-gate 	dtrace_actdesc_t *ap;
1063*457Sbmc 	uint_t n = 1, argmax;
1064*457Sbmc 	uint64_t arg = 0;
10650Sstevel@tonic-gate 
10660Sstevel@tonic-gate 	/*
10670Sstevel@tonic-gate 	 * If the aggregation has no aggregating function applied to it, then
10680Sstevel@tonic-gate 	 * this statement has no effect.  Flag this as a programming error.
10690Sstevel@tonic-gate 	 */
10700Sstevel@tonic-gate 	if (dnp->dn_aggfun == NULL) {
10710Sstevel@tonic-gate 		dnerror(dnp, D_AGG_NULL, "expression has null effect: @%s\n",
10720Sstevel@tonic-gate 		    dnp->dn_ident->di_name);
10730Sstevel@tonic-gate 	}
10740Sstevel@tonic-gate 
10750Sstevel@tonic-gate 	aid = dnp->dn_ident;
10760Sstevel@tonic-gate 	fid = dnp->dn_aggfun->dn_ident;
10770Sstevel@tonic-gate 
10780Sstevel@tonic-gate 	if (dnp->dn_aggfun->dn_args != NULL &&
10790Sstevel@tonic-gate 	    dt_node_is_scalar(dnp->dn_aggfun->dn_args) == 0) {
10800Sstevel@tonic-gate 		dnerror(dnp->dn_aggfun, D_AGG_SCALAR, "%s( ) argument #1 must "
10810Sstevel@tonic-gate 		    "be of scalar type\n", fid->di_name);
10820Sstevel@tonic-gate 	}
10830Sstevel@tonic-gate 
10840Sstevel@tonic-gate 	/*
10850Sstevel@tonic-gate 	 * The ID of the aggregation itself is implicitly recorded as the first
10860Sstevel@tonic-gate 	 * member of each aggregation tuple so we can distinguish them later.
10870Sstevel@tonic-gate 	 */
10880Sstevel@tonic-gate 	ap = dt_stmt_action(dtp, sdp);
10890Sstevel@tonic-gate 	dt_action_difconst(ap, aid->di_id, DTRACEACT_DIFEXPR);
10900Sstevel@tonic-gate 
10910Sstevel@tonic-gate 	for (anp = dnp->dn_aggtup; anp != NULL; anp = anp->dn_list) {
10920Sstevel@tonic-gate 		ap = dt_stmt_action(dtp, sdp);
10930Sstevel@tonic-gate 		n++;
10940Sstevel@tonic-gate 
10950Sstevel@tonic-gate 		if (anp->dn_kind == DT_NODE_FUNC) {
10960Sstevel@tonic-gate 			if (anp->dn_ident->di_id == DT_ACT_STACK) {
10970Sstevel@tonic-gate 				dt_action_stack_args(dtp, ap, anp->dn_args);
10980Sstevel@tonic-gate 				continue;
10990Sstevel@tonic-gate 			}
11000Sstevel@tonic-gate 
11010Sstevel@tonic-gate 			if (anp->dn_ident->di_id == DT_ACT_USTACK ||
11020Sstevel@tonic-gate 			    anp->dn_ident->di_id == DT_ACT_JSTACK) {
11030Sstevel@tonic-gate 				dt_action_ustack_args(dtp, ap, anp);
11040Sstevel@tonic-gate 				continue;
11050Sstevel@tonic-gate 			}
1106*457Sbmc 
1107*457Sbmc 			switch (anp->dn_ident->di_id) {
1108*457Sbmc 			case DT_ACT_UADDR:
1109*457Sbmc 				dt_action_symmod_args(dtp, ap,
1110*457Sbmc 				    anp->dn_args, DTRACEACT_UADDR);
1111*457Sbmc 				continue;
1112*457Sbmc 
1113*457Sbmc 			case DT_ACT_USYM:
1114*457Sbmc 				dt_action_symmod_args(dtp, ap,
1115*457Sbmc 				    anp->dn_args, DTRACEACT_USYM);
1116*457Sbmc 				continue;
1117*457Sbmc 
1118*457Sbmc 			case DT_ACT_UMOD:
1119*457Sbmc 				dt_action_symmod_args(dtp, ap,
1120*457Sbmc 				    anp->dn_args, DTRACEACT_UMOD);
1121*457Sbmc 				continue;
1122*457Sbmc 
1123*457Sbmc 			case DT_ACT_SYM:
1124*457Sbmc 				dt_action_symmod_args(dtp, ap,
1125*457Sbmc 				    anp->dn_args, DTRACEACT_SYM);
1126*457Sbmc 				continue;
1127*457Sbmc 
1128*457Sbmc 			case DT_ACT_MOD:
1129*457Sbmc 				dt_action_symmod_args(dtp, ap,
1130*457Sbmc 				    anp->dn_args, DTRACEACT_MOD);
1131*457Sbmc 				continue;
1132*457Sbmc 
1133*457Sbmc 			default:
1134*457Sbmc 				break;
1135*457Sbmc 			}
11360Sstevel@tonic-gate 		}
11370Sstevel@tonic-gate 
11380Sstevel@tonic-gate 		dt_cg(yypcb, anp);
11390Sstevel@tonic-gate 		ap->dtad_difo = dt_as(yypcb);
11400Sstevel@tonic-gate 		ap->dtad_kind = DTRACEACT_DIFEXPR;
11410Sstevel@tonic-gate 	}
11420Sstevel@tonic-gate 
11430Sstevel@tonic-gate 	if (fid->di_id == DTRACEAGG_LQUANTIZE) {
11440Sstevel@tonic-gate 		/*
1145*457Sbmc 		 * For linear quantization, we have between two and four
1146*457Sbmc 		 * arguments in addition to the expression:
11470Sstevel@tonic-gate 		 *
11480Sstevel@tonic-gate 		 *    arg1 => Base value
11490Sstevel@tonic-gate 		 *    arg2 => Limit value
11500Sstevel@tonic-gate 		 *    arg3 => Quantization level step size (defaults to 1)
1151*457Sbmc 		 *    arg4 => Quantization increment value (defaults to 1)
11520Sstevel@tonic-gate 		 */
11530Sstevel@tonic-gate 		dt_node_t *arg1 = dnp->dn_aggfun->dn_args->dn_list;
11540Sstevel@tonic-gate 		dt_node_t *arg2 = arg1->dn_list;
11550Sstevel@tonic-gate 		dt_node_t *arg3 = arg2->dn_list;
11560Sstevel@tonic-gate 		uint64_t nlevels, step = 1;
11570Sstevel@tonic-gate 		int64_t baseval, limitval;
11580Sstevel@tonic-gate 
11590Sstevel@tonic-gate 		if (arg1->dn_kind != DT_NODE_INT) {
11600Sstevel@tonic-gate 			dnerror(arg1, D_LQUANT_BASETYPE, "lquantize( ) "
11610Sstevel@tonic-gate 			    "argument #1 must be an integer constant\n");
11620Sstevel@tonic-gate 		}
11630Sstevel@tonic-gate 
11640Sstevel@tonic-gate 		baseval = (int64_t)arg1->dn_value;
11650Sstevel@tonic-gate 
11660Sstevel@tonic-gate 		if (baseval < INT32_MIN || baseval > INT32_MAX) {
11670Sstevel@tonic-gate 			dnerror(arg1, D_LQUANT_BASEVAL, "lquantize( ) "
11680Sstevel@tonic-gate 			    "argument #1 must be a 32-bit quantity\n");
11690Sstevel@tonic-gate 		}
11700Sstevel@tonic-gate 
11710Sstevel@tonic-gate 		if (arg2->dn_kind != DT_NODE_INT) {
11720Sstevel@tonic-gate 			dnerror(arg2, D_LQUANT_LIMTYPE, "lquantize( ) "
11730Sstevel@tonic-gate 			    "argument #2 must be an integer constant\n");
11740Sstevel@tonic-gate 		}
11750Sstevel@tonic-gate 
11760Sstevel@tonic-gate 		limitval = (int64_t)arg2->dn_value;
11770Sstevel@tonic-gate 
11780Sstevel@tonic-gate 		if (limitval < INT32_MIN || limitval > INT32_MAX) {
11790Sstevel@tonic-gate 			dnerror(arg2, D_LQUANT_LIMVAL, "lquantize( ) "
11800Sstevel@tonic-gate 			    "argument #2 must be a 32-bit quantity\n");
11810Sstevel@tonic-gate 		}
11820Sstevel@tonic-gate 
11830Sstevel@tonic-gate 		if (limitval < baseval) {
11840Sstevel@tonic-gate 			dnerror(dnp, D_LQUANT_MISMATCH,
11850Sstevel@tonic-gate 			    "lquantize( ) base (argument #1) must be less "
11860Sstevel@tonic-gate 			    "than limit (argument #2)\n");
11870Sstevel@tonic-gate 		}
11880Sstevel@tonic-gate 
11890Sstevel@tonic-gate 		if (arg3 != NULL) {
11900Sstevel@tonic-gate 			if (!dt_node_is_posconst(arg3)) {
11910Sstevel@tonic-gate 				dnerror(arg3, D_LQUANT_STEPTYPE, "lquantize( ) "
11920Sstevel@tonic-gate 				    "argument #3 must be a non-zero positive "
11930Sstevel@tonic-gate 				    "integer constant\n");
11940Sstevel@tonic-gate 			}
11950Sstevel@tonic-gate 
11960Sstevel@tonic-gate 			if ((step = arg3->dn_value) > UINT16_MAX) {
11970Sstevel@tonic-gate 				dnerror(arg3, D_LQUANT_STEPVAL, "lquantize( ) "
11980Sstevel@tonic-gate 				    "argument #3 must be a 16-bit quantity\n");
11990Sstevel@tonic-gate 			}
12000Sstevel@tonic-gate 		}
12010Sstevel@tonic-gate 
12020Sstevel@tonic-gate 		nlevels = (limitval - baseval) / step;
12030Sstevel@tonic-gate 
12040Sstevel@tonic-gate 		if (nlevels == 0) {
12050Sstevel@tonic-gate 			dnerror(dnp, D_LQUANT_STEPLARGE,
12060Sstevel@tonic-gate 			    "lquantize( ) step (argument #3) too large: must "
12070Sstevel@tonic-gate 			    "have at least one quantization level\n");
12080Sstevel@tonic-gate 		}
12090Sstevel@tonic-gate 
12100Sstevel@tonic-gate 		if (nlevels > UINT16_MAX) {
12110Sstevel@tonic-gate 			dnerror(dnp, D_LQUANT_STEPSMALL, "lquantize( ) step "
12120Sstevel@tonic-gate 			    "(argument #3) too small: number of quantization "
12130Sstevel@tonic-gate 			    "levels must be a 16-bit quantity\n");
12140Sstevel@tonic-gate 		}
12150Sstevel@tonic-gate 
1216*457Sbmc 		arg = (step << DTRACE_LQUANTIZE_STEPSHIFT) |
12170Sstevel@tonic-gate 		    (nlevels << DTRACE_LQUANTIZE_LEVELSHIFT) |
12180Sstevel@tonic-gate 		    ((baseval << DTRACE_LQUANTIZE_BASESHIFT) &
12190Sstevel@tonic-gate 		    DTRACE_LQUANTIZE_BASEMASK);
1220*457Sbmc 
1221*457Sbmc 		incr = arg3 != NULL ? arg3->dn_list : NULL;
1222*457Sbmc 		argmax = 5;
1223*457Sbmc 	}
1224*457Sbmc 
1225*457Sbmc 	if (fid->di_id == DTRACEAGG_QUANTIZE) {
1226*457Sbmc 		incr = dnp->dn_aggfun->dn_args->dn_list;
1227*457Sbmc 		argmax = 2;
1228*457Sbmc 	}
1229*457Sbmc 
1230*457Sbmc 	if (incr != NULL) {
1231*457Sbmc 		if (!dt_node_is_scalar(incr)) {
1232*457Sbmc 			dnerror(dnp, D_PROTO_ARG, "%s( ) increment value "
1233*457Sbmc 			    "(argument #%d) must be of scalar type\n",
1234*457Sbmc 			    fid->di_name, argmax);
1235*457Sbmc 		}
1236*457Sbmc 
1237*457Sbmc 		if ((anp = incr->dn_list) != NULL) {
1238*457Sbmc 			int argc = argmax;
1239*457Sbmc 
1240*457Sbmc 			for (; anp != NULL; anp = anp->dn_list)
1241*457Sbmc 				argc++;
1242*457Sbmc 
1243*457Sbmc 			dnerror(incr, D_PROTO_LEN, "%s( ) prototype "
1244*457Sbmc 			    "mismatch: %d args passed, at most %d expected",
1245*457Sbmc 			    fid->di_name, argc, argmax);
1246*457Sbmc 		}
1247*457Sbmc 
1248*457Sbmc 		ap = dt_stmt_action(dtp, sdp);
1249*457Sbmc 		n++;
1250*457Sbmc 
1251*457Sbmc 		dt_cg(yypcb, incr);
1252*457Sbmc 		ap->dtad_difo = dt_as(yypcb);
1253*457Sbmc 		ap->dtad_difo->dtdo_rtype = dt_void_rtype;
1254*457Sbmc 		ap->dtad_kind = DTRACEACT_DIFEXPR;
1255*457Sbmc 	}
1256*457Sbmc 
1257*457Sbmc 	assert(sdp->dtsd_aggdata == NULL);
1258*457Sbmc 	sdp->dtsd_aggdata = aid;
1259*457Sbmc 
1260*457Sbmc 	ap = dt_stmt_action(dtp, sdp);
1261*457Sbmc 	assert(fid->di_kind == DT_IDENT_AGGFUNC);
1262*457Sbmc 	assert(DTRACEACT_ISAGG(fid->di_id));
1263*457Sbmc 	ap->dtad_kind = fid->di_id;
1264*457Sbmc 	ap->dtad_ntuple = n;
1265*457Sbmc 	ap->dtad_arg = arg;
1266*457Sbmc 
1267*457Sbmc 	if (dnp->dn_aggfun->dn_args != NULL) {
1268*457Sbmc 		dt_cg(yypcb, dnp->dn_aggfun->dn_args);
1269*457Sbmc 		ap->dtad_difo = dt_as(yypcb);
12700Sstevel@tonic-gate 	}
12710Sstevel@tonic-gate }
12720Sstevel@tonic-gate 
12730Sstevel@tonic-gate static void
12740Sstevel@tonic-gate dt_compile_one_clause(dtrace_hdl_t *dtp, dt_node_t *cnp, dt_node_t *pnp)
12750Sstevel@tonic-gate {
12760Sstevel@tonic-gate 	dtrace_ecbdesc_t *edp;
12770Sstevel@tonic-gate 	dtrace_stmtdesc_t *sdp;
12780Sstevel@tonic-gate 	dt_node_t *dnp;
12790Sstevel@tonic-gate 
12800Sstevel@tonic-gate 	yylineno = pnp->dn_line;
12810Sstevel@tonic-gate 	dt_setcontext(dtp, pnp->dn_desc);
12820Sstevel@tonic-gate 	(void) dt_node_cook(cnp, DT_IDFLG_REF);
12830Sstevel@tonic-gate 
12840Sstevel@tonic-gate 	if (DT_TREEDUMP_PASS(dtp, 2))
12850Sstevel@tonic-gate 		dt_node_printr(cnp, stderr, 0);
12860Sstevel@tonic-gate 
1287265Smws 	if ((edp = dt_ecbdesc_create(dtp, pnp->dn_desc)) == NULL)
12880Sstevel@tonic-gate 		longjmp(yypcb->pcb_jmpbuf, EDT_NOMEM);
12890Sstevel@tonic-gate 
12900Sstevel@tonic-gate 	assert(yypcb->pcb_ecbdesc == NULL);
12910Sstevel@tonic-gate 	yypcb->pcb_ecbdesc = edp;
12920Sstevel@tonic-gate 
12930Sstevel@tonic-gate 	if (cnp->dn_pred != NULL) {
12940Sstevel@tonic-gate 		dt_cg(yypcb, cnp->dn_pred);
12950Sstevel@tonic-gate 		edp->dted_pred.dtpdd_difo = dt_as(yypcb);
12960Sstevel@tonic-gate 	}
12970Sstevel@tonic-gate 
12980Sstevel@tonic-gate 	if (cnp->dn_acts == NULL) {
12990Sstevel@tonic-gate 		dt_stmt_append(dt_stmt_create(dtp, edp,
13000Sstevel@tonic-gate 		    cnp->dn_ctxattr, _dtrace_defattr), cnp);
13010Sstevel@tonic-gate 	}
13020Sstevel@tonic-gate 
13030Sstevel@tonic-gate 	for (dnp = cnp->dn_acts; dnp != NULL; dnp = dnp->dn_list) {
13040Sstevel@tonic-gate 		assert(yypcb->pcb_stmt == NULL);
13050Sstevel@tonic-gate 		sdp = dt_stmt_create(dtp, edp, cnp->dn_ctxattr, cnp->dn_attr);
13060Sstevel@tonic-gate 
13070Sstevel@tonic-gate 		switch (dnp->dn_kind) {
13080Sstevel@tonic-gate 		case DT_NODE_DEXPR:
13090Sstevel@tonic-gate 			if (dnp->dn_expr->dn_kind == DT_NODE_AGG)
13100Sstevel@tonic-gate 				dt_compile_agg(dtp, dnp->dn_expr, sdp);
13110Sstevel@tonic-gate 			else
13120Sstevel@tonic-gate 				dt_compile_exp(dtp, dnp, sdp);
13130Sstevel@tonic-gate 			break;
13140Sstevel@tonic-gate 		case DT_NODE_DFUNC:
13150Sstevel@tonic-gate 			dt_compile_fun(dtp, dnp, sdp);
13160Sstevel@tonic-gate 			break;
13170Sstevel@tonic-gate 		case DT_NODE_AGG:
13180Sstevel@tonic-gate 			dt_compile_agg(dtp, dnp, sdp);
13190Sstevel@tonic-gate 			break;
13200Sstevel@tonic-gate 		default:
13210Sstevel@tonic-gate 			dnerror(dnp, D_UNKNOWN, "internal error -- node kind "
13220Sstevel@tonic-gate 			    "%u is not a valid statement\n", dnp->dn_kind);
13230Sstevel@tonic-gate 		}
13240Sstevel@tonic-gate 
13250Sstevel@tonic-gate 		assert(yypcb->pcb_stmt == sdp);
13260Sstevel@tonic-gate 		dt_stmt_append(sdp, dnp);
13270Sstevel@tonic-gate 	}
13280Sstevel@tonic-gate 
13290Sstevel@tonic-gate 	assert(yypcb->pcb_ecbdesc == edp);
1330265Smws 	dt_ecbdesc_release(dtp, edp);
13310Sstevel@tonic-gate 	dt_endcontext(dtp);
13320Sstevel@tonic-gate 	yypcb->pcb_ecbdesc = NULL;
13330Sstevel@tonic-gate }
13340Sstevel@tonic-gate 
13350Sstevel@tonic-gate static void
13360Sstevel@tonic-gate dt_compile_clause(dtrace_hdl_t *dtp, dt_node_t *cnp)
13370Sstevel@tonic-gate {
13380Sstevel@tonic-gate 	dt_node_t *pnp;
13390Sstevel@tonic-gate 
13400Sstevel@tonic-gate 	for (pnp = cnp->dn_pdescs; pnp != NULL; pnp = pnp->dn_list)
13410Sstevel@tonic-gate 		dt_compile_one_clause(dtp, cnp, pnp);
13420Sstevel@tonic-gate }
13430Sstevel@tonic-gate 
1344265Smws static void
1345265Smws dt_compile_xlator(dt_node_t *dnp)
1346265Smws {
1347265Smws 	dt_xlator_t *dxp = dnp->dn_xlator;
1348265Smws 	dt_node_t *mnp;
1349265Smws 
1350265Smws 	for (mnp = dnp->dn_members; mnp != NULL; mnp = mnp->dn_list) {
1351265Smws 		assert(dxp->dx_membdif[mnp->dn_membid] == NULL);
1352265Smws 		dt_cg(yypcb, mnp);
1353265Smws 		dxp->dx_membdif[mnp->dn_membid] = dt_as(yypcb);
1354265Smws 	}
1355265Smws }
1356265Smws 
13570Sstevel@tonic-gate void
13580Sstevel@tonic-gate dt_setcontext(dtrace_hdl_t *dtp, dtrace_probedesc_t *pdp)
13590Sstevel@tonic-gate {
13600Sstevel@tonic-gate 	const dtrace_pattr_t *pap;
13610Sstevel@tonic-gate 	dt_probe_t *prp;
13620Sstevel@tonic-gate 	dt_ident_t *idp;
13630Sstevel@tonic-gate 	char attrstr[8];
13640Sstevel@tonic-gate 	int err;
13650Sstevel@tonic-gate 
13660Sstevel@tonic-gate 	/*
13670Sstevel@tonic-gate 	 * If the provider name ends with what could be interpreted as a
13680Sstevel@tonic-gate 	 * number, we assume that it's a pid and that we may need to
13690Sstevel@tonic-gate 	 * dynamically create those probes for that process.
13700Sstevel@tonic-gate 	 */
13710Sstevel@tonic-gate 	if (isdigit(pdp->dtpd_provider[strlen(pdp->dtpd_provider) - 1]))
13720Sstevel@tonic-gate 		dt_pid_create_probes(pdp, dtp);
13730Sstevel@tonic-gate 
13740Sstevel@tonic-gate 	/*
13750Sstevel@tonic-gate 	 * Call dt_probe_info() to get the probe arguments and attributes.  If
13760Sstevel@tonic-gate 	 * a representative probe is found, set 'pap' to the probe provider's
13770Sstevel@tonic-gate 	 * attributes.  Otherwise set 'pap' to default Unstable attributes.
13780Sstevel@tonic-gate 	 */
13790Sstevel@tonic-gate 	if ((prp = dt_probe_info(dtp, pdp, &yypcb->pcb_pinfo)) == NULL) {
13800Sstevel@tonic-gate 		pap = &_dtrace_prvdesc;
13810Sstevel@tonic-gate 		err = dtrace_errno(dtp);
13820Sstevel@tonic-gate 		bzero(&yypcb->pcb_pinfo, sizeof (dtrace_probeinfo_t));
13830Sstevel@tonic-gate 		yypcb->pcb_pinfo.dtp_attr = pap->dtpa_provider;
13840Sstevel@tonic-gate 		yypcb->pcb_pinfo.dtp_arga = pap->dtpa_args;
13850Sstevel@tonic-gate 	} else {
13860Sstevel@tonic-gate 		pap = &prp->pr_pvp->pv_desc.dtvd_attr;
13870Sstevel@tonic-gate 		err = 0;
13880Sstevel@tonic-gate 	}
13890Sstevel@tonic-gate 
13900Sstevel@tonic-gate 	if (err == EDT_NOPROBE && !(yypcb->pcb_cflags & DTRACE_C_ZDEFS)) {
13910Sstevel@tonic-gate 		xyerror(D_PDESC_ZERO, "probe description %s:%s:%s:%s does not "
13920Sstevel@tonic-gate 		    "match any probes\n", pdp->dtpd_provider, pdp->dtpd_mod,
13930Sstevel@tonic-gate 		    pdp->dtpd_func, pdp->dtpd_name);
13940Sstevel@tonic-gate 	}
13950Sstevel@tonic-gate 
13960Sstevel@tonic-gate 	if (err != EDT_NOPROBE && err != EDT_UNSTABLE && err != 0)
13970Sstevel@tonic-gate 		xyerror(D_PDESC_INVAL, "%s\n", dtrace_errmsg(dtp, err));
13980Sstevel@tonic-gate 
13990Sstevel@tonic-gate 	dt_dprintf("set context to %s:%s:%s:%s [%u] prp=%p attr=%s argc=%d\n",
14000Sstevel@tonic-gate 	    pdp->dtpd_provider, pdp->dtpd_mod, pdp->dtpd_func, pdp->dtpd_name,
14010Sstevel@tonic-gate 	    pdp->dtpd_id, (void *)prp, dt_attr_str(yypcb->pcb_pinfo.dtp_attr,
14020Sstevel@tonic-gate 	    attrstr, sizeof (attrstr)), yypcb->pcb_pinfo.dtp_argc);
14030Sstevel@tonic-gate 
14040Sstevel@tonic-gate 	/*
14050Sstevel@tonic-gate 	 * Reset the stability attributes of D global variables that vary
14060Sstevel@tonic-gate 	 * based on the attributes of the provider and context itself.
14070Sstevel@tonic-gate 	 */
14080Sstevel@tonic-gate 	if ((idp = dt_idhash_lookup(dtp->dt_globals, "probeprov")) != NULL)
14090Sstevel@tonic-gate 		idp->di_attr = pap->dtpa_provider;
14100Sstevel@tonic-gate 	if ((idp = dt_idhash_lookup(dtp->dt_globals, "probemod")) != NULL)
14110Sstevel@tonic-gate 		idp->di_attr = pap->dtpa_mod;
14120Sstevel@tonic-gate 	if ((idp = dt_idhash_lookup(dtp->dt_globals, "probefunc")) != NULL)
14130Sstevel@tonic-gate 		idp->di_attr = pap->dtpa_func;
14140Sstevel@tonic-gate 	if ((idp = dt_idhash_lookup(dtp->dt_globals, "probename")) != NULL)
14150Sstevel@tonic-gate 		idp->di_attr = pap->dtpa_name;
14160Sstevel@tonic-gate 	if ((idp = dt_idhash_lookup(dtp->dt_globals, "args")) != NULL)
14170Sstevel@tonic-gate 		idp->di_attr = pap->dtpa_args;
14180Sstevel@tonic-gate 
14190Sstevel@tonic-gate 	yypcb->pcb_pdesc = pdp;
14200Sstevel@tonic-gate 	yypcb->pcb_probe = prp;
14210Sstevel@tonic-gate }
14220Sstevel@tonic-gate 
14230Sstevel@tonic-gate /*
14240Sstevel@tonic-gate  * Reset context-dependent variables and state at the end of cooking a D probe
14250Sstevel@tonic-gate  * definition clause.  This ensures that external declarations between clauses
14260Sstevel@tonic-gate  * do not reference any stale context-dependent data from the previous clause.
14270Sstevel@tonic-gate  */
14280Sstevel@tonic-gate void
14290Sstevel@tonic-gate dt_endcontext(dtrace_hdl_t *dtp)
14300Sstevel@tonic-gate {
14310Sstevel@tonic-gate 	static const char *const cvars[] = {
14320Sstevel@tonic-gate 		"probeprov", "probemod", "probefunc", "probename", "args", NULL
14330Sstevel@tonic-gate 	};
14340Sstevel@tonic-gate 
14350Sstevel@tonic-gate 	dt_ident_t *idp;
14360Sstevel@tonic-gate 	int i;
14370Sstevel@tonic-gate 
14380Sstevel@tonic-gate 	for (i = 0; cvars[i] != NULL; i++) {
14390Sstevel@tonic-gate 		if ((idp = dt_idhash_lookup(dtp->dt_globals, cvars[i])) != NULL)
14400Sstevel@tonic-gate 			idp->di_attr = _dtrace_defattr;
14410Sstevel@tonic-gate 	}
14420Sstevel@tonic-gate 
14430Sstevel@tonic-gate 	yypcb->pcb_pdesc = NULL;
14440Sstevel@tonic-gate 	yypcb->pcb_probe = NULL;
14450Sstevel@tonic-gate }
14460Sstevel@tonic-gate 
14470Sstevel@tonic-gate static int
14480Sstevel@tonic-gate dt_reduceid(dt_idhash_t *dhp, dt_ident_t *idp, dtrace_hdl_t *dtp)
14490Sstevel@tonic-gate {
14500Sstevel@tonic-gate 	if (idp->di_vers != 0 && idp->di_vers > dtp->dt_vmax)
14510Sstevel@tonic-gate 		dt_idhash_delete(dhp, idp);
14520Sstevel@tonic-gate 
14530Sstevel@tonic-gate 	return (0);
14540Sstevel@tonic-gate }
14550Sstevel@tonic-gate 
14560Sstevel@tonic-gate /*
14570Sstevel@tonic-gate  * When dtrace_setopt() is called for "version", it calls dt_reduce() to remove
14580Sstevel@tonic-gate  * any identifiers or translators that have been previously defined as bound to
14590Sstevel@tonic-gate  * a version greater than the specified version.  Therefore, in our current
14600Sstevel@tonic-gate  * version implementation, establishing a binding is a one-way transformation.
14610Sstevel@tonic-gate  * In addition, no versioning is currently provided for types as our .d library
14620Sstevel@tonic-gate  * files do not define any types and we reserve prefixes DTRACE_ and dtrace_
14630Sstevel@tonic-gate  * for our exclusive use.  If required, type versioning will require more work.
14640Sstevel@tonic-gate  */
14650Sstevel@tonic-gate int
14660Sstevel@tonic-gate dt_reduce(dtrace_hdl_t *dtp, dt_version_t v)
14670Sstevel@tonic-gate {
14680Sstevel@tonic-gate 	char s[DT_VERSION_STRMAX];
14690Sstevel@tonic-gate 	dt_xlator_t *dxp, *nxp;
14700Sstevel@tonic-gate 
14710Sstevel@tonic-gate 	if (v > dtp->dt_vmax)
14720Sstevel@tonic-gate 		return (dt_set_errno(dtp, EDT_VERSREDUCED));
14730Sstevel@tonic-gate 	else if (v == dtp->dt_vmax)
14740Sstevel@tonic-gate 		return (0); /* no reduction necessary */
14750Sstevel@tonic-gate 
14760Sstevel@tonic-gate 	dt_dprintf("reducing api version to %s\n",
14770Sstevel@tonic-gate 	    dt_version_num2str(v, s, sizeof (s)));
14780Sstevel@tonic-gate 
14790Sstevel@tonic-gate 	dtp->dt_vmax = v;
14800Sstevel@tonic-gate 
14810Sstevel@tonic-gate 	for (dxp = dt_list_next(&dtp->dt_xlators); dxp != NULL; dxp = nxp) {
14820Sstevel@tonic-gate 		nxp = dt_list_next(dxp);
14830Sstevel@tonic-gate 		if ((dxp->dx_souid.di_vers != 0 && dxp->dx_souid.di_vers > v) ||
14840Sstevel@tonic-gate 		    (dxp->dx_ptrid.di_vers != 0 && dxp->dx_ptrid.di_vers > v))
14850Sstevel@tonic-gate 			dt_list_delete(&dtp->dt_xlators, dxp);
14860Sstevel@tonic-gate 	}
14870Sstevel@tonic-gate 
14880Sstevel@tonic-gate 	(void) dt_idhash_iter(dtp->dt_macros, (dt_idhash_f *)dt_reduceid, dtp);
14890Sstevel@tonic-gate 	(void) dt_idhash_iter(dtp->dt_aggs, (dt_idhash_f *)dt_reduceid, dtp);
14900Sstevel@tonic-gate 	(void) dt_idhash_iter(dtp->dt_globals, (dt_idhash_f *)dt_reduceid, dtp);
14910Sstevel@tonic-gate 	(void) dt_idhash_iter(dtp->dt_tls, (dt_idhash_f *)dt_reduceid, dtp);
14920Sstevel@tonic-gate 
14930Sstevel@tonic-gate 	return (0);
14940Sstevel@tonic-gate }
14950Sstevel@tonic-gate 
14960Sstevel@tonic-gate /*
14970Sstevel@tonic-gate  * Fork and exec the cpp(1) preprocessor to run over the specified input file,
14980Sstevel@tonic-gate  * and return a FILE handle for the cpp output.  We use the /dev/fd filesystem
14990Sstevel@tonic-gate  * here to simplify the code by leveraging file descriptor inheritance.
15000Sstevel@tonic-gate  */
15010Sstevel@tonic-gate static FILE *
15020Sstevel@tonic-gate dt_preproc(dtrace_hdl_t *dtp, FILE *ifp)
15030Sstevel@tonic-gate {
15040Sstevel@tonic-gate 	int argc = dtp->dt_cpp_argc;
15050Sstevel@tonic-gate 	char **argv = malloc(sizeof (char *) * (argc + 5));
15060Sstevel@tonic-gate 	FILE *ofp = tmpfile();
15070Sstevel@tonic-gate 
15080Sstevel@tonic-gate 	char ipath[20], opath[20]; /* big enough for /dev/fd/ + INT_MAX + \0 */
15090Sstevel@tonic-gate 	char verdef[32]; /* big enough for -D__SUNW_D_VERSION=0x%08x + \0 */
15100Sstevel@tonic-gate 
15110Sstevel@tonic-gate 	struct sigaction act, oact;
15120Sstevel@tonic-gate 	sigset_t mask, omask;
15130Sstevel@tonic-gate 
15140Sstevel@tonic-gate 	int wstat, estat;
15150Sstevel@tonic-gate 	pid_t pid;
15160Sstevel@tonic-gate 	off64_t off;
15170Sstevel@tonic-gate 	int c;
15180Sstevel@tonic-gate 
15190Sstevel@tonic-gate 	if (argv == NULL || ofp == NULL) {
15200Sstevel@tonic-gate 		(void) dt_set_errno(dtp, errno);
15210Sstevel@tonic-gate 		goto err;
15220Sstevel@tonic-gate 	}
15230Sstevel@tonic-gate 
15240Sstevel@tonic-gate 	/*
15250Sstevel@tonic-gate 	 * If the input is a seekable file, see if it is an interpreter file.
15260Sstevel@tonic-gate 	 * If we see #!, seek past the first line because cpp will choke on it.
15270Sstevel@tonic-gate 	 * We start cpp just prior to the \n at the end of this line so that
15280Sstevel@tonic-gate 	 * it still sees the newline, ensuring that #line values are correct.
15290Sstevel@tonic-gate 	 */
15300Sstevel@tonic-gate 	if (isatty(fileno(ifp)) == 0 && (off = ftello64(ifp)) != -1) {
15310Sstevel@tonic-gate 		if ((c = fgetc(ifp)) == '#' && (c = fgetc(ifp)) == '!') {
15320Sstevel@tonic-gate 			for (off += 2; c != '\n'; off++) {
15330Sstevel@tonic-gate 				if ((c = fgetc(ifp)) == EOF)
15340Sstevel@tonic-gate 					break;
15350Sstevel@tonic-gate 			}
15360Sstevel@tonic-gate 			if (c == '\n')
15370Sstevel@tonic-gate 				off--; /* start cpp just prior to \n */
15380Sstevel@tonic-gate 		}
15390Sstevel@tonic-gate 		(void) fflush(ifp);
15400Sstevel@tonic-gate 		(void) fseeko64(ifp, off, SEEK_SET);
15410Sstevel@tonic-gate 	}
15420Sstevel@tonic-gate 
15430Sstevel@tonic-gate 	(void) snprintf(ipath, sizeof (ipath), "/dev/fd/%d", fileno(ifp));
15440Sstevel@tonic-gate 	(void) snprintf(opath, sizeof (opath), "/dev/fd/%d", fileno(ofp));
15450Sstevel@tonic-gate 
15460Sstevel@tonic-gate 	bcopy(dtp->dt_cpp_argv, argv, sizeof (char *) * argc);
15470Sstevel@tonic-gate 
15480Sstevel@tonic-gate 	(void) snprintf(verdef, sizeof (verdef),
15490Sstevel@tonic-gate 	    "-D__SUNW_D_VERSION=0x%08x", dtp->dt_vmax);
15500Sstevel@tonic-gate 	argv[argc++] = verdef;
15510Sstevel@tonic-gate 
15520Sstevel@tonic-gate 	switch (dtp->dt_stdcmode) {
15530Sstevel@tonic-gate 	case DT_STDC_XA:
15540Sstevel@tonic-gate 	case DT_STDC_XT:
15550Sstevel@tonic-gate 		argv[argc++] = "-D__STDC__=0";
15560Sstevel@tonic-gate 		break;
15570Sstevel@tonic-gate 	case DT_STDC_XC:
15580Sstevel@tonic-gate 		argv[argc++] = "-D__STDC__=1";
15590Sstevel@tonic-gate 		break;
15600Sstevel@tonic-gate 	}
15610Sstevel@tonic-gate 
15620Sstevel@tonic-gate 	argv[argc++] = ipath;
15630Sstevel@tonic-gate 	argv[argc++] = opath;
15640Sstevel@tonic-gate 	argv[argc] = NULL;
15650Sstevel@tonic-gate 
15660Sstevel@tonic-gate 	/*
15670Sstevel@tonic-gate 	 * libdtrace must be able to be embedded in other programs that may
15680Sstevel@tonic-gate 	 * include application-specific signal handlers.  Therefore, if we
15690Sstevel@tonic-gate 	 * need to fork to run cpp(1), we must avoid generating a SIGCHLD
15700Sstevel@tonic-gate 	 * that could confuse the containing application.  To do this,
15710Sstevel@tonic-gate 	 * we block SIGCHLD and reset its disposition to SIG_DFL.
15720Sstevel@tonic-gate 	 * We restore our signal state once we are done.
15730Sstevel@tonic-gate 	 */
15740Sstevel@tonic-gate 	(void) sigemptyset(&mask);
15750Sstevel@tonic-gate 	(void) sigaddset(&mask, SIGCHLD);
15760Sstevel@tonic-gate 	(void) sigprocmask(SIG_BLOCK, &mask, &omask);
15770Sstevel@tonic-gate 
15780Sstevel@tonic-gate 	bzero(&act, sizeof (act));
15790Sstevel@tonic-gate 	act.sa_handler = SIG_DFL;
15800Sstevel@tonic-gate 	(void) sigaction(SIGCHLD, &act, &oact);
15810Sstevel@tonic-gate 
15820Sstevel@tonic-gate 	if ((pid = fork1()) == -1) {
15830Sstevel@tonic-gate 		(void) sigaction(SIGCHLD, &oact, NULL);
15840Sstevel@tonic-gate 		(void) sigprocmask(SIG_SETMASK, &omask, NULL);
15850Sstevel@tonic-gate 		(void) dt_set_errno(dtp, EDT_CPPFORK);
15860Sstevel@tonic-gate 		goto err;
15870Sstevel@tonic-gate 	}
15880Sstevel@tonic-gate 
15890Sstevel@tonic-gate 	if (pid == 0) {
15900Sstevel@tonic-gate 		(void) execvp(dtp->dt_cpp_path, argv);
15910Sstevel@tonic-gate 		_exit(errno == ENOENT ? 127 : 126);
15920Sstevel@tonic-gate 	}
15930Sstevel@tonic-gate 
15940Sstevel@tonic-gate 	do {
15950Sstevel@tonic-gate 		dt_dprintf("waiting for %s (PID %d)\n", dtp->dt_cpp_path,
15960Sstevel@tonic-gate 		    (int)pid);
15970Sstevel@tonic-gate 	} while (waitpid(pid, &wstat, 0) == -1 && errno == EINTR);
15980Sstevel@tonic-gate 
15990Sstevel@tonic-gate 	(void) sigaction(SIGCHLD, &oact, NULL);
16000Sstevel@tonic-gate 	(void) sigprocmask(SIG_SETMASK, &omask, NULL);
16010Sstevel@tonic-gate 
16020Sstevel@tonic-gate 	dt_dprintf("%s returned exit status 0x%x\n", dtp->dt_cpp_path, wstat);
16030Sstevel@tonic-gate 	estat = WIFEXITED(wstat) ? WEXITSTATUS(wstat) : -1;
16040Sstevel@tonic-gate 
16050Sstevel@tonic-gate 	if (estat != 0) {
16060Sstevel@tonic-gate 		switch (estat) {
16070Sstevel@tonic-gate 		case 126:
16080Sstevel@tonic-gate 			(void) dt_set_errno(dtp, EDT_CPPEXEC);
16090Sstevel@tonic-gate 			break;
16100Sstevel@tonic-gate 		case 127:
16110Sstevel@tonic-gate 			(void) dt_set_errno(dtp, EDT_CPPENT);
16120Sstevel@tonic-gate 			break;
16130Sstevel@tonic-gate 		default:
16140Sstevel@tonic-gate 			(void) dt_set_errno(dtp, EDT_CPPERR);
16150Sstevel@tonic-gate 		}
16160Sstevel@tonic-gate 		goto err;
16170Sstevel@tonic-gate 	}
16180Sstevel@tonic-gate 
16190Sstevel@tonic-gate 	free(argv);
16200Sstevel@tonic-gate 	(void) fflush(ofp);
16210Sstevel@tonic-gate 	(void) fseek(ofp, 0, SEEK_SET);
16220Sstevel@tonic-gate 	return (ofp);
16230Sstevel@tonic-gate 
16240Sstevel@tonic-gate err:
16250Sstevel@tonic-gate 	free(argv);
16260Sstevel@tonic-gate 	(void) fclose(ofp);
16270Sstevel@tonic-gate 	return (NULL);
16280Sstevel@tonic-gate }
16290Sstevel@tonic-gate 
16300Sstevel@tonic-gate /*
16310Sstevel@tonic-gate  * Open all of the .d library files found in the specified directory and try to
16320Sstevel@tonic-gate  * compile each one in order to cache its inlines and translators, etc.  We
16330Sstevel@tonic-gate  * silently ignore any missing directories and other files found therein.
16340Sstevel@tonic-gate  * We only fail (and thereby fail dt_load_libs()) if we fail to compile a
16350Sstevel@tonic-gate  * library and the error is something other than #pragma D depends_on.
16360Sstevel@tonic-gate  * Dependency errors are silently ignored to permit a library directory to
16370Sstevel@tonic-gate  * contain libraries which may not be accessible depending on our privileges.
16380Sstevel@tonic-gate  *
16390Sstevel@tonic-gate  * Note that at present, no ordering is defined between library files found in
16400Sstevel@tonic-gate  * the same directory: if cross-library dependencies are eventually required,
16410Sstevel@tonic-gate  * we will need to extend the #pragma D depends_on directive with an additional
16420Sstevel@tonic-gate  * class for libraries, and this function will need to create a graph of the
16430Sstevel@tonic-gate  * various library pathnames and then perform a topological ordering using the
16440Sstevel@tonic-gate  * dependency information before we attempt to compile any of them.
16450Sstevel@tonic-gate  */
16460Sstevel@tonic-gate static int
16470Sstevel@tonic-gate dt_load_libs_dir(dtrace_hdl_t *dtp, const char *path)
16480Sstevel@tonic-gate {
16490Sstevel@tonic-gate 	struct dirent *dp, *ep;
16500Sstevel@tonic-gate 	const char *p;
16510Sstevel@tonic-gate 	DIR *dirp;
16520Sstevel@tonic-gate 
16530Sstevel@tonic-gate 	char fname[PATH_MAX];
16540Sstevel@tonic-gate 	dtrace_prog_t *pgp;
16550Sstevel@tonic-gate 	FILE *fp;
16560Sstevel@tonic-gate 
16570Sstevel@tonic-gate 	if ((dirp = opendir(path)) == NULL) {
16580Sstevel@tonic-gate 		dt_dprintf("skipping lib dir %s: %s\n", path, strerror(errno));
16590Sstevel@tonic-gate 		return (0);
16600Sstevel@tonic-gate 	}
16610Sstevel@tonic-gate 
16620Sstevel@tonic-gate 	ep = alloca(sizeof (struct dirent) + PATH_MAX + 1);
16630Sstevel@tonic-gate 	bzero(ep, sizeof (struct dirent) + PATH_MAX + 1);
16640Sstevel@tonic-gate 
16650Sstevel@tonic-gate 	while (readdir_r(dirp, ep, &dp) == 0 && dp != NULL) {
16660Sstevel@tonic-gate 		if ((p = strrchr(dp->d_name, '.')) == NULL || strcmp(p, ".d"))
16670Sstevel@tonic-gate 			continue; /* skip any filename not ending in .d */
16680Sstevel@tonic-gate 
16690Sstevel@tonic-gate 		(void) snprintf(fname, sizeof (fname),
16700Sstevel@tonic-gate 		    "%s/%s", path, dp->d_name);
16710Sstevel@tonic-gate 
16720Sstevel@tonic-gate 		if ((fp = fopen(fname, "r")) == NULL) {
16730Sstevel@tonic-gate 			dt_dprintf("skipping library %s: %s\n",
16740Sstevel@tonic-gate 			    fname, strerror(errno));
16750Sstevel@tonic-gate 			continue;
16760Sstevel@tonic-gate 		}
16770Sstevel@tonic-gate 
16780Sstevel@tonic-gate 		dtp->dt_filetag = fname;
16790Sstevel@tonic-gate 		pgp = dtrace_program_fcompile(dtp, fp, DTRACE_C_EMPTY, 0, NULL);
16800Sstevel@tonic-gate 		(void) fclose(fp);
16810Sstevel@tonic-gate 		dtp->dt_filetag = NULL;
16820Sstevel@tonic-gate 
16830Sstevel@tonic-gate 		if (pgp == NULL && (dtp->dt_errno != EDT_COMPILER ||
16840Sstevel@tonic-gate 		    dtp->dt_errtag != dt_errtag(D_PRAGMA_DEPEND))) {
16850Sstevel@tonic-gate 			(void) closedir(dirp);
16860Sstevel@tonic-gate 			return (-1); /* preserve dt_errno */
16870Sstevel@tonic-gate 		}
16880Sstevel@tonic-gate 
16890Sstevel@tonic-gate 		if (pgp == NULL) {
16900Sstevel@tonic-gate 			dt_dprintf("skipping library: %s\n",
16910Sstevel@tonic-gate 			    dtrace_errmsg(dtp, dtrace_errno(dtp)));
16920Sstevel@tonic-gate 		} else
1693265Smws 			dt_program_destroy(dtp, pgp);
16940Sstevel@tonic-gate 	}
16950Sstevel@tonic-gate 
16960Sstevel@tonic-gate 	(void) closedir(dirp);
16970Sstevel@tonic-gate 	return (0);
16980Sstevel@tonic-gate }
16990Sstevel@tonic-gate 
17000Sstevel@tonic-gate /*
17010Sstevel@tonic-gate  * Load the contents of any appropriate DTrace .d library files.  These files
17020Sstevel@tonic-gate  * contain inlines and translators that will be cached by the compiler.  We
17030Sstevel@tonic-gate  * defer this activity until the first compile to permit libdtrace clients to
17040Sstevel@tonic-gate  * add their own library directories and so that we can properly report errors.
17050Sstevel@tonic-gate  */
17060Sstevel@tonic-gate static int
17070Sstevel@tonic-gate dt_load_libs(dtrace_hdl_t *dtp)
17080Sstevel@tonic-gate {
17090Sstevel@tonic-gate 	dt_dirpath_t *dirp;
17100Sstevel@tonic-gate 
17110Sstevel@tonic-gate 	if (dtp->dt_cflags & DTRACE_C_NOLIBS)
17120Sstevel@tonic-gate 		return (0); /* libraries already processed */
17130Sstevel@tonic-gate 
17140Sstevel@tonic-gate 	dtp->dt_cflags |= DTRACE_C_NOLIBS;
17150Sstevel@tonic-gate 
17160Sstevel@tonic-gate 	for (dirp = dt_list_next(&dtp->dt_lib_path);
17170Sstevel@tonic-gate 	    dirp != NULL; dirp = dt_list_next(dirp)) {
17180Sstevel@tonic-gate 		if (dt_load_libs_dir(dtp, dirp->dir_path) != 0) {
17190Sstevel@tonic-gate 			dtp->dt_cflags &= ~DTRACE_C_NOLIBS;
17200Sstevel@tonic-gate 			return (-1); /* errno is set for us */
17210Sstevel@tonic-gate 		}
17220Sstevel@tonic-gate 	}
17230Sstevel@tonic-gate 
17240Sstevel@tonic-gate 	return (0);
17250Sstevel@tonic-gate }
17260Sstevel@tonic-gate 
17270Sstevel@tonic-gate static void *
17280Sstevel@tonic-gate dt_compile(dtrace_hdl_t *dtp, int context, dtrace_probespec_t pspec, void *arg,
17290Sstevel@tonic-gate     uint_t cflags, int argc, char *const argv[], FILE *fp, const char *s)
17300Sstevel@tonic-gate {
17310Sstevel@tonic-gate 	dt_node_t *dnp;
17320Sstevel@tonic-gate 	dt_decl_t *ddp;
17330Sstevel@tonic-gate 	dt_pcb_t pcb;
17340Sstevel@tonic-gate 	void *rv;
17350Sstevel@tonic-gate 	int err;
17360Sstevel@tonic-gate 
17370Sstevel@tonic-gate 	if ((fp == NULL && s == NULL) || (cflags & ~DTRACE_C_MASK) != 0) {
17380Sstevel@tonic-gate 		(void) dt_set_errno(dtp, EINVAL);
17390Sstevel@tonic-gate 		return (NULL);
17400Sstevel@tonic-gate 	}
17410Sstevel@tonic-gate 
17420Sstevel@tonic-gate 	if (dt_list_next(&dtp->dt_lib_path) != NULL && dt_load_libs(dtp) != 0)
17430Sstevel@tonic-gate 		return (NULL); /* errno is set for us */
17440Sstevel@tonic-gate 
17450Sstevel@tonic-gate 	(void) ctf_discard(dtp->dt_cdefs->dm_ctfp);
17460Sstevel@tonic-gate 	(void) ctf_discard(dtp->dt_ddefs->dm_ctfp);
17470Sstevel@tonic-gate 
17480Sstevel@tonic-gate 	(void) dt_idhash_iter(dtp->dt_globals, dt_idreset, NULL);
17490Sstevel@tonic-gate 	(void) dt_idhash_iter(dtp->dt_tls, dt_idreset, NULL);
17500Sstevel@tonic-gate 
17510Sstevel@tonic-gate 	if (fp && (cflags & DTRACE_C_CPP) && (fp = dt_preproc(dtp, fp)) == NULL)
17520Sstevel@tonic-gate 		return (NULL); /* errno is set for us */
17530Sstevel@tonic-gate 
17540Sstevel@tonic-gate 	dt_pcb_push(dtp, &pcb);
17550Sstevel@tonic-gate 
17560Sstevel@tonic-gate 	pcb.pcb_fileptr = fp;
17570Sstevel@tonic-gate 	pcb.pcb_string = s;
17580Sstevel@tonic-gate 	pcb.pcb_strptr = s;
17590Sstevel@tonic-gate 	pcb.pcb_strlen = s ? strlen(s) : 0;
17600Sstevel@tonic-gate 	pcb.pcb_sargc = argc;
17610Sstevel@tonic-gate 	pcb.pcb_sargv = argv;
17620Sstevel@tonic-gate 	pcb.pcb_sflagv = argc ? calloc(argc, sizeof (ushort_t)) : NULL;
17630Sstevel@tonic-gate 	pcb.pcb_pspec = pspec;
17640Sstevel@tonic-gate 	pcb.pcb_cflags = dtp->dt_cflags | cflags;
17650Sstevel@tonic-gate 	pcb.pcb_amin = dtp->dt_amin;
17660Sstevel@tonic-gate 	pcb.pcb_yystate = -1;
17670Sstevel@tonic-gate 	pcb.pcb_context = context;
17680Sstevel@tonic-gate 	pcb.pcb_token = context;
17690Sstevel@tonic-gate 
17700Sstevel@tonic-gate 	if (context == DT_CTX_DPROG)
17710Sstevel@tonic-gate 		yybegin(YYS_CLAUSE);
17720Sstevel@tonic-gate 	else
17730Sstevel@tonic-gate 		yybegin(YYS_EXPR);
17740Sstevel@tonic-gate 
17750Sstevel@tonic-gate 	if ((err = setjmp(yypcb->pcb_jmpbuf)) != 0)
17760Sstevel@tonic-gate 		goto out;
17770Sstevel@tonic-gate 
17780Sstevel@tonic-gate 	if (yypcb->pcb_sargc != 0 && yypcb->pcb_sflagv == NULL)
17790Sstevel@tonic-gate 		longjmp(yypcb->pcb_jmpbuf, EDT_NOMEM);
17800Sstevel@tonic-gate 
17810Sstevel@tonic-gate 	yypcb->pcb_idents = dt_idhash_create("ambiguous", NULL, 0, 0);
17820Sstevel@tonic-gate 	yypcb->pcb_locals = dt_idhash_create("clause local", NULL,
17830Sstevel@tonic-gate 	    DIF_VAR_OTHER_UBASE, DIF_VAR_OTHER_MAX);
17840Sstevel@tonic-gate 
17850Sstevel@tonic-gate 	if (yypcb->pcb_idents == NULL || yypcb->pcb_locals == NULL)
17860Sstevel@tonic-gate 		longjmp(yypcb->pcb_jmpbuf, EDT_NOMEM);
17870Sstevel@tonic-gate 
17880Sstevel@tonic-gate 	/*
17890Sstevel@tonic-gate 	 * Invoke the parser to evaluate the D source code.  If any errors
17900Sstevel@tonic-gate 	 * occur during parsing, an error function will be called and we
17910Sstevel@tonic-gate 	 * will longjmp back to pcb_jmpbuf to abort.  If parsing succeeds,
17920Sstevel@tonic-gate 	 * we optionally display the parse tree if debugging is enabled.
17930Sstevel@tonic-gate 	 */
17940Sstevel@tonic-gate 	if (yyparse() != 0 || yypcb->pcb_root == NULL)
17950Sstevel@tonic-gate 		xyerror(D_EMPTY, "empty D program translation unit\n");
17960Sstevel@tonic-gate 
17970Sstevel@tonic-gate 	yybegin(YYS_DONE);
17980Sstevel@tonic-gate 
17990Sstevel@tonic-gate 	if (context != DT_CTX_DTYPE && DT_TREEDUMP_PASS(dtp, 1))
18000Sstevel@tonic-gate 		dt_node_printr(yypcb->pcb_root, stderr, 0);
18010Sstevel@tonic-gate 
18020Sstevel@tonic-gate 	if (yypcb->pcb_pragmas != NULL)
18030Sstevel@tonic-gate 		(void) dt_idhash_iter(yypcb->pcb_pragmas, dt_idpragma, NULL);
18040Sstevel@tonic-gate 
18050Sstevel@tonic-gate 	if (argc > 1 && !(yypcb->pcb_cflags & DTRACE_C_ARGREF) &&
18060Sstevel@tonic-gate 	    !(yypcb->pcb_sflagv[argc - 1] & DT_IDFLG_REF)) {
18070Sstevel@tonic-gate 		xyerror(D_MACRO_UNUSED, "extraneous argument '%s' ($%d is "
18080Sstevel@tonic-gate 		    "not referenced)\n", yypcb->pcb_sargv[argc - 1], argc - 1);
18090Sstevel@tonic-gate 	}
18100Sstevel@tonic-gate 
18110Sstevel@tonic-gate 	/*
18120Sstevel@tonic-gate 	 * If we have successfully created a parse tree for a D program, loop
18130Sstevel@tonic-gate 	 * over the clauses and actions and instantiate the corresponding
18140Sstevel@tonic-gate 	 * libdtrace program.  If we are parsing a D expression, then we
18150Sstevel@tonic-gate 	 * simply run the code generator and assembler on the resulting tree.
18160Sstevel@tonic-gate 	 */
18170Sstevel@tonic-gate 	switch (context) {
18180Sstevel@tonic-gate 	case DT_CTX_DPROG:
18190Sstevel@tonic-gate 		assert(yypcb->pcb_root->dn_kind == DT_NODE_PROG);
18200Sstevel@tonic-gate 
18210Sstevel@tonic-gate 		if ((dnp = yypcb->pcb_root->dn_list) == NULL &&
18220Sstevel@tonic-gate 		    !(yypcb->pcb_cflags & DTRACE_C_EMPTY))
18230Sstevel@tonic-gate 			xyerror(D_EMPTY, "empty D program translation unit\n");
18240Sstevel@tonic-gate 
1825265Smws 		if ((yypcb->pcb_prog = dt_program_create(dtp)) == NULL)
18260Sstevel@tonic-gate 			longjmp(yypcb->pcb_jmpbuf, dtrace_errno(dtp));
18270Sstevel@tonic-gate 
18280Sstevel@tonic-gate 		for (; dnp != NULL; dnp = dnp->dn_list) {
18290Sstevel@tonic-gate 			switch (dnp->dn_kind) {
18300Sstevel@tonic-gate 			case DT_NODE_CLAUSE:
18310Sstevel@tonic-gate 				dt_compile_clause(dtp, dnp);
18320Sstevel@tonic-gate 				break;
1833265Smws 			case DT_NODE_XLATOR:
1834265Smws 				if (dtp->dt_xlatemode == DT_XL_DYNAMIC)
1835265Smws 					dt_compile_xlator(dnp);
1836265Smws 				break;
18370Sstevel@tonic-gate 			case DT_NODE_PROVIDER:
18380Sstevel@tonic-gate 				(void) dt_node_cook(dnp, DT_IDFLG_REF);
18390Sstevel@tonic-gate 				break;
18400Sstevel@tonic-gate 			}
18410Sstevel@tonic-gate 		}
18420Sstevel@tonic-gate 
1843265Smws 		yypcb->pcb_prog->dp_xrefs = yypcb->pcb_asxrefs;
1844265Smws 		yypcb->pcb_prog->dp_xrefslen = yypcb->pcb_asxreflen;
1845265Smws 		yypcb->pcb_asxrefs = NULL;
1846265Smws 		yypcb->pcb_asxreflen = 0;
1847265Smws 
1848265Smws 		rv = yypcb->pcb_prog;
18490Sstevel@tonic-gate 		break;
18500Sstevel@tonic-gate 
18510Sstevel@tonic-gate 	case DT_CTX_DEXPR:
18520Sstevel@tonic-gate 		(void) dt_node_cook(yypcb->pcb_root, DT_IDFLG_REF);
18530Sstevel@tonic-gate 		dt_cg(yypcb, yypcb->pcb_root);
18540Sstevel@tonic-gate 		rv = dt_as(yypcb);
18550Sstevel@tonic-gate 		break;
18560Sstevel@tonic-gate 
18570Sstevel@tonic-gate 	case DT_CTX_DTYPE:
18580Sstevel@tonic-gate 		ddp = (dt_decl_t *)yypcb->pcb_root; /* root is really a decl */
18590Sstevel@tonic-gate 		err = dt_decl_type(ddp, arg);
18600Sstevel@tonic-gate 		dt_decl_free(ddp);
18610Sstevel@tonic-gate 
18620Sstevel@tonic-gate 		if (err != 0)
18630Sstevel@tonic-gate 			longjmp(yypcb->pcb_jmpbuf, EDT_COMPILER);
18640Sstevel@tonic-gate 
18650Sstevel@tonic-gate 		rv = NULL;
18660Sstevel@tonic-gate 		break;
18670Sstevel@tonic-gate 	}
18680Sstevel@tonic-gate 
18690Sstevel@tonic-gate out:
18700Sstevel@tonic-gate 	if (context != DT_CTX_DTYPE && DT_TREEDUMP_PASS(dtp, 3))
18710Sstevel@tonic-gate 		dt_node_printr(yypcb->pcb_root, stderr, 0);
18720Sstevel@tonic-gate 
18730Sstevel@tonic-gate 	if (dtp->dt_cdefs_fd != -1 && (ftruncate64(dtp->dt_cdefs_fd, 0) == -1 ||
18740Sstevel@tonic-gate 	    lseek64(dtp->dt_cdefs_fd, 0, SEEK_SET) == -1 ||
18750Sstevel@tonic-gate 	    ctf_write(dtp->dt_cdefs->dm_ctfp, dtp->dt_cdefs_fd) == CTF_ERR))
18760Sstevel@tonic-gate 		dt_dprintf("failed to update CTF cache: %s\n", strerror(errno));
18770Sstevel@tonic-gate 
18780Sstevel@tonic-gate 	if (dtp->dt_ddefs_fd != -1 && (ftruncate64(dtp->dt_ddefs_fd, 0) == -1 ||
18790Sstevel@tonic-gate 	    lseek64(dtp->dt_ddefs_fd, 0, SEEK_SET) == -1 ||
18800Sstevel@tonic-gate 	    ctf_write(dtp->dt_ddefs->dm_ctfp, dtp->dt_ddefs_fd) == CTF_ERR))
18810Sstevel@tonic-gate 		dt_dprintf("failed to update CTF cache: %s\n", strerror(errno));
18820Sstevel@tonic-gate 
18830Sstevel@tonic-gate 	if (yypcb->pcb_fileptr && (cflags & DTRACE_C_CPP))
18840Sstevel@tonic-gate 		(void) fclose(yypcb->pcb_fileptr); /* close dt_preproc() file */
18850Sstevel@tonic-gate 
18860Sstevel@tonic-gate 	dt_pcb_pop(dtp, err);
18870Sstevel@tonic-gate 	(void) dt_set_errno(dtp, err);
18880Sstevel@tonic-gate 	return (err ? NULL : rv);
18890Sstevel@tonic-gate }
18900Sstevel@tonic-gate 
18910Sstevel@tonic-gate dtrace_prog_t *
18920Sstevel@tonic-gate dtrace_program_strcompile(dtrace_hdl_t *dtp, const char *s,
18930Sstevel@tonic-gate     dtrace_probespec_t spec, uint_t cflags, int argc, char *const argv[])
18940Sstevel@tonic-gate {
18950Sstevel@tonic-gate 	return (dt_compile(dtp, DT_CTX_DPROG,
18960Sstevel@tonic-gate 	    spec, NULL, cflags, argc, argv, NULL, s));
18970Sstevel@tonic-gate }
18980Sstevel@tonic-gate 
18990Sstevel@tonic-gate dtrace_prog_t *
19000Sstevel@tonic-gate dtrace_program_fcompile(dtrace_hdl_t *dtp, FILE *fp,
19010Sstevel@tonic-gate     uint_t cflags, int argc, char *const argv[])
19020Sstevel@tonic-gate {
19030Sstevel@tonic-gate 	return (dt_compile(dtp, DT_CTX_DPROG,
19040Sstevel@tonic-gate 	    DTRACE_PROBESPEC_NAME, NULL, cflags, argc, argv, fp, NULL));
19050Sstevel@tonic-gate }
19060Sstevel@tonic-gate 
19070Sstevel@tonic-gate int
19080Sstevel@tonic-gate dtrace_type_strcompile(dtrace_hdl_t *dtp, const char *s, dtrace_typeinfo_t *dtt)
19090Sstevel@tonic-gate {
19100Sstevel@tonic-gate 	(void) dt_compile(dtp, DT_CTX_DTYPE,
19110Sstevel@tonic-gate 	    DTRACE_PROBESPEC_NONE, dtt, 0, 0, NULL, NULL, s);
19120Sstevel@tonic-gate 	return (dtp->dt_errno ? -1 : 0);
19130Sstevel@tonic-gate }
19140Sstevel@tonic-gate 
19150Sstevel@tonic-gate int
19160Sstevel@tonic-gate dtrace_type_fcompile(dtrace_hdl_t *dtp, FILE *fp, dtrace_typeinfo_t *dtt)
19170Sstevel@tonic-gate {
19180Sstevel@tonic-gate 	(void) dt_compile(dtp, DT_CTX_DTYPE,
19190Sstevel@tonic-gate 	    DTRACE_PROBESPEC_NONE, dtt, 0, 0, NULL, fp, NULL);
19200Sstevel@tonic-gate 	return (dtp->dt_errno ? -1 : 0);
19210Sstevel@tonic-gate }
1922