dt_consume.c revision 249573
119304Speter/*
219304Speter * CDDL HEADER START
319304Speter *
419304Speter * The contents of this file are subject to the terms of the
519304Speter * Common Development and Distribution License (the "License").
619304Speter * You may not use this file except in compliance with the License.
719304Speter *
819304Speter * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
919304Speter * or http://www.opensolaris.org/os/licensing.
1019304Speter * See the License for the specific language governing permissions
1119304Speter * and limitations under the License.
1219304Speter *
13254225Speter * When distributing Covered Code, include this CDDL HEADER in each
1419304Speter * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
1519304Speter * If applicable, add the following below this CDDL HEADER, with the
1619304Speter * fields enclosed by brackets "[]" replaced with your own identifying
1719304Speter * information: Portions Copyright [yyyy] [name of copyright owner]
18254225Speter *
1919304Speter * CDDL HEADER END
2019304Speter */
2119304Speter/*
2219304Speter * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
2319304Speter * Use is subject to license terms.
2419304Speter */
25254225Speter
2619304Speter/*
2719304Speter * Copyright (c) 2011, Joyent, Inc. All rights reserved.
2819304Speter * Copyright (c) 2011 by Delphix. All rights reserved.
2919304Speter */
3019304Speter
3119304Speter#include <stdlib.h>
3219304Speter#include <strings.h>
3319304Speter#include <errno.h>
3419304Speter#include <unistd.h>
3519304Speter#include <limits.h>
36254225Speter#include <assert.h>
37254225Speter#include <ctype.h>
38254225Speter#if defined(sun)
3919304Speter#include <alloca.h>
4019304Speter#endif
4119304Speter#include <dt_impl.h>
4219304Speter#if !defined(sun)
4319304Speter#include <libproc_compat.h>
4419304Speter#endif
4519304Speter
4619304Speter#define	DT_MASK_LO 0x00000000FFFFFFFFULL
47254225Speter
4819304Speter/*
4919304Speter * We declare this here because (1) we need it and (2) we want to avoid a
5019304Speter * dependency on libm in libdtrace.
5119304Speter */
5219304Speterstatic long double
5319304Speterdt_fabsl(long double x)
5419304Speter{
5519304Speter	if (x < 0)
5619304Speter		return (-x);
5719304Speter
5819304Speter	return (x);
5919304Speter}
6019304Speter
6119304Speter/*
6219304Speter * 128-bit arithmetic functions needed to support the stddev() aggregating
6319304Speter * action.
64254225Speter */
6519304Speterstatic int
6619304Speterdt_gt_128(uint64_t *a, uint64_t *b)
67254225Speter{
6819304Speter	return (a[1] > b[1] || (a[1] == b[1] && a[0] > b[0]));
6919304Speter}
7019304Speter
7119304Speterstatic int
7219304Speterdt_ge_128(uint64_t *a, uint64_t *b)
73254225Speter{
7419304Speter	return (a[1] > b[1] || (a[1] == b[1] && a[0] >= b[0]));
7519304Speter}
7619304Speter
7719304Speterstatic int
7819304Speterdt_le_128(uint64_t *a, uint64_t *b)
7919304Speter{
8019304Speter	return (a[1] < b[1] || (a[1] == b[1] && a[0] <= b[0]));
8119304Speter}
8219304Speter
8319304Speter/*
8419304Speter * Shift the 128-bit value in a by b. If b is positive, shift left.
8519304Speter * If b is negative, shift right.
8619304Speter */
87254225Speterstatic void
8819304Speterdt_shift_128(uint64_t *a, int b)
8919304Speter{
90254225Speter	uint64_t mask;
9119304Speter
9219304Speter	if (b == 0)
9319304Speter		return;
94254225Speter
9519304Speter	if (b < 0) {
96254225Speter		b = -b;
9719304Speter		if (b >= 64) {
9819304Speter			a[0] = a[1] >> (b - 64);
9919304Speter			a[1] = 0;
10019304Speter		} else {
101254225Speter			a[0] >>= b;
10219304Speter			mask = 1LL << (64 - b);
10319304Speter			mask -= 1;
10419304Speter			a[0] |= ((a[1] & mask) << (64 - b));
10519304Speter			a[1] >>= b;
10619304Speter		}
10719304Speter	} else {
108254225Speter		if (b >= 64) {
10919304Speter			a[1] = a[0] << (b - 64);
11019304Speter			a[0] = 0;
11119304Speter		} else {
11219304Speter			a[1] <<= b;
11319304Speter			mask = a[0] >> (64 - b);
11419304Speter			a[1] |= mask;
11519304Speter			a[0] <<= b;
11619304Speter		}
117254225Speter	}
11819304Speter}
11919304Speter
12019304Speterstatic int
12119304Speterdt_nbits_128(uint64_t *a)
12219304Speter{
12319304Speter	int nbits = 0;
12419304Speter	uint64_t tmp[2];
12519304Speter	uint64_t zero[2] = { 0, 0 };
126254225Speter
12719304Speter	tmp[0] = a[0];
12819304Speter	tmp[1] = a[1];
129254225Speter
13019304Speter	dt_shift_128(tmp, -1);
13119304Speter	while (dt_gt_128(tmp, zero)) {
13219304Speter		dt_shift_128(tmp, -1);
133254225Speter		nbits++;
13419304Speter	}
135254225Speter
13619304Speter	return (nbits);
137254225Speter}
138254225Speter
139254225Speterstatic void
14019304Speterdt_subtract_128(uint64_t *minuend, uint64_t *subtrahend, uint64_t *difference)
14119304Speter{
14219304Speter	uint64_t result[2];
14319304Speter
14419304Speter	result[0] = minuend[0] - subtrahend[0];
14519304Speter	result[1] = minuend[1] - subtrahend[1] -
14619304Speter	    (minuend[0] < subtrahend[0] ? 1 : 0);
14719304Speter
14819304Speter	difference[0] = result[0];
14919304Speter	difference[1] = result[1];
15019304Speter}
15119304Speter
15219304Speterstatic void
15319304Speterdt_add_128(uint64_t *addend1, uint64_t *addend2, uint64_t *sum)
15419304Speter{
15519304Speter	uint64_t result[2];
15619304Speter
15719304Speter	result[0] = addend1[0] + addend2[0];
15819304Speter	result[1] = addend1[1] + addend2[1] +
15919304Speter	    (result[0] < addend1[0] || result[0] < addend2[0] ? 1 : 0);
16019304Speter
16119304Speter	sum[0] = result[0];
16219304Speter	sum[1] = result[1];
16319304Speter}
16419304Speter
16519304Speter/*
16619304Speter * The basic idea is to break the 2 64-bit values into 4 32-bit values,
16719304Speter * use native multiplication on those, and then re-combine into the
16819304Speter * resulting 128-bit value.
16919304Speter *
17019304Speter * (hi1 << 32 + lo1) * (hi2 << 32 + lo2) =
171254225Speter *     hi1 * hi2 << 64 +
172254225Speter *     hi1 * lo2 << 32 +
173254225Speter *     hi2 * lo1 << 32 +
17419304Speter *     lo1 * lo2
17519304Speter */
17619304Speterstatic void
17719304Speterdt_multiply_128(uint64_t factor1, uint64_t factor2, uint64_t *product)
17819304Speter{
17919304Speter	uint64_t hi1, hi2, lo1, lo2;
18019304Speter	uint64_t tmp[2];
181254225Speter
182254225Speter	hi1 = factor1 >> 32;
183254225Speter	hi2 = factor2 >> 32;
184254225Speter
185254225Speter	lo1 = factor1 & DT_MASK_LO;
18619304Speter	lo2 = factor2 & DT_MASK_LO;
18719304Speter
18819304Speter	product[0] = lo1 * lo2;
18919304Speter	product[1] = hi1 * hi2;
19019304Speter
19119304Speter	tmp[0] = hi1 * lo2;
19219304Speter	tmp[1] = 0;
19319304Speter	dt_shift_128(tmp, 32);
19419304Speter	dt_add_128(product, tmp, product);
19519304Speter
19619304Speter	tmp[0] = hi2 * lo1;
19719304Speter	tmp[1] = 0;
19819304Speter	dt_shift_128(tmp, 32);
19919304Speter	dt_add_128(product, tmp, product);
20019304Speter}
20119304Speter
20219304Speter/*
203254225Speter * This is long-hand division.
20419304Speter *
20519304Speter * We initialize subtrahend by shifting divisor left as far as possible. We
20619304Speter * loop, comparing subtrahend to dividend:  if subtrahend is smaller, we
20719304Speter * subtract and set the appropriate bit in the result.  We then shift
20819304Speter * subtrahend right by one bit for the next comparison.
20919304Speter */
21019304Speterstatic void
21119304Speterdt_divide_128(uint64_t *dividend, uint64_t divisor, uint64_t *quotient)
212254225Speter{
21319304Speter	uint64_t result[2] = { 0, 0 };
21419304Speter	uint64_t remainder[2];
215254225Speter	uint64_t subtrahend[2];
21619304Speter	uint64_t divisor_128[2];
21719304Speter	uint64_t mask[2] = { 1, 0 };
21819304Speter	int log = 0;
21919304Speter
220254225Speter	assert(divisor != 0);
22119304Speter
22219304Speter	divisor_128[0] = divisor;
22319304Speter	divisor_128[1] = 0;
22419304Speter
22519304Speter	remainder[0] = dividend[0];
226254225Speter	remainder[1] = dividend[1];
22719304Speter
22819304Speter	subtrahend[0] = divisor;
22919304Speter	subtrahend[1] = 0;
23019304Speter
23119304Speter	while (divisor > 0) {
23219304Speter		log++;
23319304Speter		divisor >>= 1;
23419304Speter	}
23519304Speter
23619304Speter	dt_shift_128(subtrahend, 128 - log);
23719304Speter	dt_shift_128(mask, 128 - log);
23819304Speter
23919304Speter	while (dt_ge_128(remainder, divisor_128)) {
24019304Speter		if (dt_ge_128(remainder, subtrahend)) {
24119304Speter			dt_subtract_128(remainder, subtrahend, remainder);
24219304Speter			result[0] |= mask[0];
24319304Speter			result[1] |= mask[1];
24419304Speter		}
24519304Speter
246254225Speter		dt_shift_128(subtrahend, -1);
24719304Speter		dt_shift_128(mask, -1);
24819304Speter	}
24919304Speter
25019304Speter	quotient[0] = result[0];
25119304Speter	quotient[1] = result[1];
25219304Speter}
25319304Speter
25419304Speter/*
25519304Speter * This is the long-hand method of calculating a square root.
25619304Speter * The algorithm is as follows:
25719304Speter *
25819304Speter * 1. Group the digits by 2 from the right.
25919304Speter * 2. Over the leftmost group, find the largest single-digit number
26019304Speter *    whose square is less than that group.
26119304Speter * 3. Subtract the result of the previous step (2 or 4, depending) and
26219304Speter *    bring down the next two-digit group.
26319304Speter * 4. For the result R we have so far, find the largest single-digit number
26419304Speter *    x such that 2 * R * 10 * x + x^2 is less than the result from step 3.
26519304Speter *    (Note that this is doubling R and performing a decimal left-shift by 1
26619304Speter *    and searching for the appropriate decimal to fill the one's place.)
26719304Speter *    The value x is the next digit in the square root.
26819304Speter * Repeat steps 3 and 4 until the desired precision is reached.  (We're
26919304Speter * dealing with integers, so the above is sufficient.)
27019304Speter *
27119304Speter * In decimal, the square root of 582,734 would be calculated as so:
27219304Speter *
27319304Speter *     __7__6__3
27419304Speter *    | 58 27 34
27519304Speter *     -49       (7^2 == 49 => 7 is the first digit in the square root)
276254225Speter *      --
277254225Speter *       9 27    (Subtract and bring down the next group.)
278254225Speter * 146   8 76    (2 * 7 * 10 * 6 + 6^2 == 876 => 6 is the next digit in
279254225Speter *      -----     the square root)
280254225Speter *         51 34 (Subtract and bring down the next group.)
281254225Speter * 1523    45 69 (2 * 76 * 10 * 3 + 3^2 == 4569 => 3 is the next digit in
282254225Speter *         -----  the square root)
283254225Speter *          5 65 (remainder)
284254225Speter *
285254225Speter * The above algorithm applies similarly in binary, but note that the
286254225Speter * only possible non-zero value for x in step 4 is 1, so step 4 becomes a
287254225Speter * simple decision: is 2 * R * 2 * 1 + 1^2 (aka R << 2 + 1) less than the
288254225Speter * preceding difference?
289254225Speter *
290254225Speter * In binary, the square root of 11011011 would be calculated as so:
291254225Speter *
292254225Speter *     __1__1__1__0
293254225Speter *    | 11 01 10 11
294254225Speter *      01          (0 << 2 + 1 == 1 < 11 => this bit is 1)
295254225Speter *      --
296254225Speter *      10 01 10 11
297254225Speter * 101   1 01       (1 << 2 + 1 == 101 < 1001 => next bit is 1)
298254225Speter *      -----
299254225Speter *       1 00 10 11
300254225Speter * 1101    11 01    (11 << 2 + 1 == 1101 < 10010 => next bit is 1)
301254225Speter *       -------
302254225Speter *          1 01 11
303254225Speter * 11101    1 11 01 (111 << 2 + 1 == 11101 > 10111 => last bit is 0)
304254225Speter *
305254225Speter */
306254225Speterstatic uint64_t
307254225Speterdt_sqrt_128(uint64_t *square)
308254225Speter{
309254225Speter	uint64_t result[2] = { 0, 0 };
310254225Speter	uint64_t diff[2] = { 0, 0 };
311254225Speter	uint64_t one[2] = { 1, 0 };
312254225Speter	uint64_t next_pair[2];
313254225Speter	uint64_t next_try[2];
314254225Speter	uint64_t bit_pairs, pair_shift;
315254225Speter	int i;
316254225Speter
317254225Speter	bit_pairs = dt_nbits_128(square) / 2;
318254225Speter	pair_shift = bit_pairs * 2;
319254225Speter
320254225Speter	for (i = 0; i <= bit_pairs; i++) {
321254225Speter		/*
322254225Speter		 * Bring down the next pair of bits.
323254225Speter		 */
324254225Speter		next_pair[0] = square[0];
325254225Speter		next_pair[1] = square[1];
326254225Speter		dt_shift_128(next_pair, -pair_shift);
327254225Speter		next_pair[0] &= 0x3;
328254225Speter		next_pair[1] = 0;
329254225Speter
330254225Speter		dt_shift_128(diff, 2);
331254225Speter		dt_add_128(diff, next_pair, diff);
332254225Speter
333254225Speter		/*
334254225Speter		 * next_try = R << 2 + 1
335254225Speter		 */
336254225Speter		next_try[0] = result[0];
337254225Speter		next_try[1] = result[1];
338254225Speter		dt_shift_128(next_try, 2);
339254225Speter		dt_add_128(next_try, one, next_try);
340254225Speter
341254225Speter		if (dt_le_128(next_try, diff)) {
342254225Speter			dt_subtract_128(diff, next_try, diff);
343254225Speter			dt_shift_128(result, 1);
344254225Speter			dt_add_128(result, one, result);
345254225Speter		} else {
346254225Speter			dt_shift_128(result, 1);
347254225Speter		}
348254225Speter
349254225Speter		pair_shift -= 2;
350254225Speter	}
351254225Speter
352254225Speter	assert(result[1] == 0);
353254225Speter
354254225Speter	return (result[0]);
355254225Speter}
356254225Speter
35719304Speteruint64_t
35819304Speterdt_stddev(uint64_t *data, uint64_t normal)
35919304Speter{
36019304Speter	uint64_t avg_of_squares[2];
361254225Speter	uint64_t square_of_avg[2];
36219304Speter	int64_t norm_avg;
36319304Speter	uint64_t diff[2];
364254225Speter
36519304Speter	/*
366254225Speter	 * The standard approximation for standard deviation is
367254225Speter	 * sqrt(average(x**2) - average(x)**2), i.e. the square root
368254225Speter	 * of the average of the squares minus the square of the average.
36919304Speter	 */
37019304Speter	dt_divide_128(data + 2, normal, avg_of_squares);
37119304Speter	dt_divide_128(avg_of_squares, data[0], avg_of_squares);
37219304Speter
37319304Speter	norm_avg = (int64_t)data[1] / (int64_t)normal / (int64_t)data[0];
37419304Speter
37519304Speter	if (norm_avg < 0)
37619304Speter		norm_avg = -norm_avg;
37719304Speter
37819304Speter	dt_multiply_128((uint64_t)norm_avg, (uint64_t)norm_avg, square_of_avg);
37919304Speter
38019304Speter	dt_subtract_128(avg_of_squares, square_of_avg, diff);
38119304Speter
382254225Speter	return (dt_sqrt_128(diff));
38319304Speter}
384254225Speter
38519304Speterstatic int
386254225Speterdt_flowindent(dtrace_hdl_t *dtp, dtrace_probedata_t *data, dtrace_epid_t last,
38719304Speter    dtrace_bufdesc_t *buf, size_t offs)
38819304Speter{
38919304Speter	dtrace_probedesc_t *pd = data->dtpda_pdesc, *npd;
39019304Speter	dtrace_eprobedesc_t *epd = data->dtpda_edesc, *nepd;
39119304Speter	char *p = pd->dtpd_provider, *n = pd->dtpd_name, *sub;
39219304Speter	dtrace_flowkind_t flow = DTRACEFLOW_NONE;
39319304Speter	const char *str = NULL;
39419304Speter	static const char *e_str[2] = { " -> ", " => " };
39519304Speter	static const char *r_str[2] = { " <- ", " <= " };
39619304Speter	static const char *ent = "entry", *ret = "return";
39719304Speter	static int entlen = 0, retlen = 0;
39819304Speter	dtrace_epid_t next, id = epd->dtepd_epid;
399254225Speter	int rval;
40019304Speter
401254225Speter	if (entlen == 0) {
402254225Speter		assert(retlen == 0);
403254225Speter		entlen = strlen(ent);
40419304Speter		retlen = strlen(ret);
40519304Speter	}
40619304Speter
40719304Speter	/*
40819304Speter	 * If the name of the probe is "entry" or ends with "-entry", we
40919304Speter	 * treat it as an entry; if it is "return" or ends with "-return",
41019304Speter	 * we treat it as a return.  (This allows application-provided probes
41119304Speter	 * like "method-entry" or "function-entry" to participate in flow
41219304Speter	 * indentation -- without accidentally misinterpreting popular probe
41319304Speter	 * names like "carpentry", "gentry" or "Coventry".)
414254225Speter	 */
41519304Speter	if ((sub = strstr(n, ent)) != NULL && sub[entlen] == '\0' &&
416254225Speter	    (sub == n || sub[-1] == '-')) {
417254225Speter		flow = DTRACEFLOW_ENTRY;
418254225Speter		str = e_str[strcmp(p, "syscall") == 0];
41919304Speter	} else if ((sub = strstr(n, ret)) != NULL && sub[retlen] == '\0' &&
42019304Speter	    (sub == n || sub[-1] == '-')) {
42119304Speter		flow = DTRACEFLOW_RETURN;
42219304Speter		str = r_str[strcmp(p, "syscall") == 0];
42319304Speter	}
42419304Speter
42519304Speter	/*
42619304Speter	 * If we're going to indent this, we need to check the ID of our last
42719304Speter	 * call.  If we're looking at the same probe ID but a different EPID,
42819304Speter	 * we _don't_ want to indent.  (Yes, there are some minor holes in
42919304Speter	 * this scheme -- it's a heuristic.)
43019304Speter	 */
43119304Speter	if (flow == DTRACEFLOW_ENTRY) {
43219304Speter		if ((last != DTRACE_EPIDNONE && id != last &&
43319304Speter		    pd->dtpd_id == dtp->dt_pdesc[last]->dtpd_id))
43419304Speter			flow = DTRACEFLOW_NONE;
43519304Speter	}
43619304Speter
437254225Speter	/*
43819304Speter	 * If we're going to unindent this, it's more difficult to see if
43919304Speter	 * we don't actually want to unindent it -- we need to look at the
44019304Speter	 * _next_ EPID.
44119304Speter	 */
44219304Speter	if (flow == DTRACEFLOW_RETURN) {
44319304Speter		offs += epd->dtepd_size;
44419304Speter
445254225Speter		do {
44619304Speter			if (offs >= buf->dtbd_size) {
44719304Speter				/*
44819304Speter				 * We're at the end -- maybe.  If the oldest
44919304Speter				 * record is non-zero, we need to wrap.
45019304Speter				 */
45119304Speter				if (buf->dtbd_oldest != 0) {
45219304Speter					offs = 0;
45319304Speter				} else {
45419304Speter					goto out;
45519304Speter				}
45619304Speter			}
45719304Speter
45819304Speter			next = *(uint32_t *)((uintptr_t)buf->dtbd_data + offs);
45919304Speter
46019304Speter			if (next == DTRACE_EPIDNONE)
461254225Speter				offs += sizeof (id);
46219304Speter		} while (next == DTRACE_EPIDNONE);
46319304Speter
46419304Speter		if ((rval = dt_epid_lookup(dtp, next, &nepd, &npd)) != 0)
46519304Speter			return (rval);
46619304Speter
46719304Speter		if (next != id && npd->dtpd_id == pd->dtpd_id)
46819304Speter			flow = DTRACEFLOW_NONE;
46919304Speter	}
47019304Speter
47119304Speterout:
47219304Speter	if (flow == DTRACEFLOW_ENTRY || flow == DTRACEFLOW_RETURN) {
47319304Speter		data->dtpda_prefix = str;
47419304Speter	} else {
47519304Speter		data->dtpda_prefix = "| ";
47619304Speter	}
47719304Speter
47819304Speter	if (flow == DTRACEFLOW_RETURN && data->dtpda_indent > 0)
47919304Speter		data->dtpda_indent -= 2;
48019304Speter
48119304Speter	data->dtpda_flow = flow;
48219304Speter
48319304Speter	return (0);
48419304Speter}
48519304Speter
48619304Speterstatic int
48719304Speterdt_nullprobe()
48819304Speter{
48919304Speter	return (DTRACE_CONSUME_THIS);
49019304Speter}
49119304Speter
49219304Speterstatic int
49319304Speterdt_nullrec()
49419304Speter{
49519304Speter	return (DTRACE_CONSUME_NEXT);
49619304Speter}
49719304Speter
49819304Speterint
49919304Speterdt_print_quantline(dtrace_hdl_t *dtp, FILE *fp, int64_t val,
50019304Speter    uint64_t normal, long double total, char positives, char negatives)
50119304Speter{
50219304Speter	long double f;
50319304Speter	uint_t depth, len = 40;
50419304Speter
50519304Speter	const char *ats = "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@";
50619304Speter	const char *spaces = "                                        ";
50719304Speter
50819304Speter	assert(strlen(ats) == len && strlen(spaces) == len);
50919304Speter	assert(!(total == 0 && (positives || negatives)));
51019304Speter	assert(!(val < 0 && !negatives));
51119304Speter	assert(!(val > 0 && !positives));
51219304Speter	assert(!(val != 0 && total == 0));
51319304Speter
51419304Speter	if (!negatives) {
51519304Speter		if (positives) {
51619304Speter			f = (dt_fabsl((long double)val) * len) / total;
51719304Speter			depth = (uint_t)(f + 0.5);
51819304Speter		} else {
51919304Speter			depth = 0;
52019304Speter		}
52119304Speter
52219304Speter		return (dt_printf(dtp, fp, "|%s%s %-9lld\n", ats + len - depth,
52319304Speter		    spaces + depth, (long long)val / normal));
52419304Speter	}
52519304Speter
526254225Speter	if (!positives) {
52719304Speter		f = (dt_fabsl((long double)val) * len) / total;
52819304Speter		depth = (uint_t)(f + 0.5);
52919304Speter
53019304Speter		return (dt_printf(dtp, fp, "%s%s| %-9lld\n", spaces + depth,
53119304Speter		    ats + len - depth, (long long)val / normal));
53219304Speter	}
53319304Speter
53419304Speter	/*
53519304Speter	 * If we're here, we have both positive and negative bucket values.
53619304Speter	 * To express this graphically, we're going to generate both positive
53719304Speter	 * and negative bars separated by a centerline.  These bars are half
53819304Speter	 * the size of normal quantize()/lquantize() bars, so we divide the
53919304Speter	 * length in half before calculating the bar length.
54019304Speter	 */
54119304Speter	len /= 2;
54219304Speter	ats = &ats[len];
54319304Speter	spaces = &spaces[len];
54419304Speter
54519304Speter	f = (dt_fabsl((long double)val) * len) / total;
54619304Speter	depth = (uint_t)(f + 0.5);
54719304Speter
54819304Speter	if (val <= 0) {
549254225Speter		return (dt_printf(dtp, fp, "%s%s|%*s %-9lld\n", spaces + depth,
55019304Speter		    ats + len - depth, len, "", (long long)val / normal));
551254225Speter	} else {
552254225Speter		return (dt_printf(dtp, fp, "%20s|%s%s %-9lld\n", "",
553254225Speter		    ats + len - depth, spaces + depth,
55419304Speter		    (long long)val / normal));
555254225Speter	}
556254225Speter}
55719304Speter
55819304Speterint
55919304Speterdt_print_quantize(dtrace_hdl_t *dtp, FILE *fp, const void *addr,
56019304Speter    size_t size, uint64_t normal)
56119304Speter{
56219304Speter	const int64_t *data = addr;
563254225Speter	int i, first_bin = 0, last_bin = DTRACE_QUANTIZE_NBUCKETS - 1;
564254225Speter	long double total = 0;
565254225Speter	char positives = 0, negatives = 0;
566254225Speter
567254225Speter	if (size != DTRACE_QUANTIZE_NBUCKETS * sizeof (uint64_t))
56819304Speter		return (dt_set_errno(dtp, EDT_DMISMATCH));
56919304Speter
57019304Speter	while (first_bin < DTRACE_QUANTIZE_NBUCKETS - 1 && data[first_bin] == 0)
57119304Speter		first_bin++;
572254225Speter
573254225Speter	if (first_bin == DTRACE_QUANTIZE_NBUCKETS - 1) {
574254225Speter		/*
575254225Speter		 * There isn't any data.  This is possible if (and only if)
576254225Speter		 * negative increment values have been used.  In this case,
577254225Speter		 * we'll print the buckets around 0.
578254225Speter		 */
579254225Speter		first_bin = DTRACE_QUANTIZE_ZEROBUCKET - 1;
580254225Speter		last_bin = DTRACE_QUANTIZE_ZEROBUCKET + 1;
581254225Speter	} else {
582254225Speter		if (first_bin > 0)
583254225Speter			first_bin--;
58419304Speter
585254225Speter		while (last_bin > 0 && data[last_bin] == 0)
586254225Speter			last_bin--;
58719304Speter
588254225Speter		if (last_bin < DTRACE_QUANTIZE_NBUCKETS - 1)
58919304Speter			last_bin++;
59019304Speter	}
59119304Speter
59219304Speter	for (i = first_bin; i <= last_bin; i++) {
593254225Speter		positives |= (data[i] > 0);
59419304Speter		negatives |= (data[i] < 0);
595254225Speter		total += dt_fabsl((long double)data[i]);
59619304Speter	}
59719304Speter
598254225Speter	if (dt_printf(dtp, fp, "\n%16s %41s %-9s\n", "value",
599254225Speter	    "------------- Distribution -------------", "count") < 0)
600254225Speter		return (-1);
601254225Speter
602254225Speter	for (i = first_bin; i <= last_bin; i++) {
603254225Speter		if (dt_printf(dtp, fp, "%16lld ",
60419304Speter		    (long long)DTRACE_QUANTIZE_BUCKETVAL(i)) < 0)
60519304Speter			return (-1);
606254225Speter
607254225Speter		if (dt_print_quantline(dtp, fp, data[i], normal, total,
608254225Speter		    positives, negatives) < 0)
609254225Speter			return (-1);
610254225Speter	}
611254225Speter
612254225Speter	return (0);
613254225Speter}
61419304Speter
61519304Speterint
61619304Speterdt_print_lquantize(dtrace_hdl_t *dtp, FILE *fp, const void *addr,
61719304Speter    size_t size, uint64_t normal)
618254225Speter{
61919304Speter	const int64_t *data = addr;
620254225Speter	int i, first_bin, last_bin, base;
62119304Speter	uint64_t arg;
62219304Speter	long double total = 0;
62319304Speter	uint16_t step, levels;
62419304Speter	char positives = 0, negatives = 0;
62519304Speter
626254225Speter	if (size < sizeof (uint64_t))
627254225Speter		return (dt_set_errno(dtp, EDT_DMISMATCH));
628254225Speter
62919304Speter	arg = *data++;
630254225Speter	size -= sizeof (uint64_t);
631254225Speter
63219304Speter	base = DTRACE_LQUANTIZE_BASE(arg);
633254225Speter	step = DTRACE_LQUANTIZE_STEP(arg);
634254225Speter	levels = DTRACE_LQUANTIZE_LEVELS(arg);
63519304Speter
636254225Speter	first_bin = 0;
637254225Speter	last_bin = levels + 1;
63819304Speter
63919304Speter	if (size != sizeof (uint64_t) * (levels + 2))
64019304Speter		return (dt_set_errno(dtp, EDT_DMISMATCH));
64119304Speter
64219304Speter	while (first_bin <= levels + 1 && data[first_bin] == 0)
643254225Speter		first_bin++;
644254225Speter
64519304Speter	if (first_bin > levels + 1) {
64619304Speter		first_bin = 0;
64719304Speter		last_bin = 2;
64819304Speter	} else {
64919304Speter		if (first_bin > 0)
65019304Speter			first_bin--;
65119304Speter
65219304Speter		while (last_bin > 0 && data[last_bin] == 0)
65319304Speter			last_bin--;
65419304Speter
655254225Speter		if (last_bin < levels + 1)
65619304Speter			last_bin++;
657254225Speter	}
65819304Speter
65919304Speter	for (i = first_bin; i <= last_bin; i++) {
66019304Speter		positives |= (data[i] > 0);
66119304Speter		negatives |= (data[i] < 0);
66219304Speter		total += dt_fabsl((long double)data[i]);
66319304Speter	}
66419304Speter
66519304Speter	if (dt_printf(dtp, fp, "\n%16s %41s %-9s\n", "value",
666254225Speter	    "------------- Distribution -------------", "count") < 0)
66719304Speter		return (-1);
66819304Speter
66919304Speter	for (i = first_bin; i <= last_bin; i++) {
67019304Speter		char c[32];
67119304Speter		int err;
67219304Speter
673254225Speter		if (i == 0) {
674254225Speter			(void) snprintf(c, sizeof (c), "< %d",
675254225Speter			    base / (uint32_t)normal);
676254225Speter			err = dt_printf(dtp, fp, "%16s ", c);
67719304Speter		} else if (i == levels + 1) {
67819304Speter			(void) snprintf(c, sizeof (c), ">= %d",
67919304Speter			    base + (levels * step));
68019304Speter			err = dt_printf(dtp, fp, "%16s ", c);
68119304Speter		} else {
68219304Speter			err = dt_printf(dtp, fp, "%16d ",
68319304Speter			    base + (i - 1) * step);
68419304Speter		}
68519304Speter
68619304Speter		if (err < 0 || dt_print_quantline(dtp, fp, data[i], normal,
68719304Speter		    total, positives, negatives) < 0)
68819304Speter			return (-1);
68919304Speter	}
69019304Speter
69119304Speter	return (0);
69219304Speter}
69319304Speter
69419304Speterint
69519304Speterdt_print_llquantize(dtrace_hdl_t *dtp, FILE *fp, const void *addr,
69619304Speter    size_t size, uint64_t normal)
69719304Speter{
69819304Speter	int i, first_bin, last_bin, bin = 1, order, levels;
69919304Speter	uint16_t factor, low, high, nsteps;
70019304Speter	const int64_t *data = addr;
70119304Speter	int64_t value = 1, next, step;
70219304Speter	char positives = 0, negatives = 0;
70319304Speter	long double total = 0;
70419304Speter	uint64_t arg;
70519304Speter	char c[32];
70619304Speter
70719304Speter	if (size < sizeof (uint64_t))
70819304Speter		return (dt_set_errno(dtp, EDT_DMISMATCH));
70919304Speter
71019304Speter	arg = *data++;
71119304Speter	size -= sizeof (uint64_t);
71219304Speter
71319304Speter	factor = DTRACE_LLQUANTIZE_FACTOR(arg);
71419304Speter	low = DTRACE_LLQUANTIZE_LOW(arg);
71519304Speter	high = DTRACE_LLQUANTIZE_HIGH(arg);
71619304Speter	nsteps = DTRACE_LLQUANTIZE_NSTEP(arg);
71719304Speter
71819304Speter	/*
71919304Speter	 * We don't expect to be handed invalid llquantize() parameters here,
72019304Speter	 * but sanity check them (to a degree) nonetheless.
72119304Speter	 */
72219304Speter	if (size > INT32_MAX || factor < 2 || low >= high ||
72319304Speter	    nsteps == 0 || factor > nsteps)
72419304Speter		return (dt_set_errno(dtp, EDT_DMISMATCH));
72519304Speter
72619304Speter	levels = (int)size / sizeof (uint64_t);
72719304Speter
72819304Speter	first_bin = 0;
72919304Speter	last_bin = levels - 1;
73019304Speter
73119304Speter	while (first_bin < levels && data[first_bin] == 0)
73219304Speter		first_bin++;
73319304Speter
73419304Speter	if (first_bin == levels) {
73519304Speter		first_bin = 0;
73619304Speter		last_bin = 1;
73719304Speter	} else {
73819304Speter		if (first_bin > 0)
73919304Speter			first_bin--;
74019304Speter
74119304Speter		while (last_bin > 0 && data[last_bin] == 0)
74219304Speter			last_bin--;
74319304Speter
744254225Speter		if (last_bin < levels - 1)
745254225Speter			last_bin++;
74619304Speter	}
74719304Speter
74819304Speter	for (i = first_bin; i <= last_bin; i++) {
74919304Speter		positives |= (data[i] > 0);
75019304Speter		negatives |= (data[i] < 0);
75119304Speter		total += dt_fabsl((long double)data[i]);
75219304Speter	}
75319304Speter
75419304Speter	if (dt_printf(dtp, fp, "\n%16s %41s %-9s\n", "value",
75519304Speter	    "------------- Distribution -------------", "count") < 0)
75619304Speter		return (-1);
75719304Speter
75819304Speter	for (order = 0; order < low; order++)
75919304Speter		value *= factor;
76019304Speter
76119304Speter	next = value * factor;
76219304Speter	step = next > nsteps ? next / nsteps : 1;
763254225Speter
76419304Speter	if (first_bin == 0) {
765254225Speter		(void) snprintf(c, sizeof (c), "< %lld", (long long)value);
76619304Speter
76719304Speter		if (dt_printf(dtp, fp, "%16s ", c) < 0)
76819304Speter			return (-1);
76919304Speter
77019304Speter		if (dt_print_quantline(dtp, fp, data[0], normal,
77119304Speter		    total, positives, negatives) < 0)
77219304Speter			return (-1);
77319304Speter	}
77419304Speter
77519304Speter	while (order <= high) {
77619304Speter		if (bin >= first_bin && bin <= last_bin) {
77719304Speter			if (dt_printf(dtp, fp, "%16lld ", (long long)value) < 0)
77819304Speter				return (-1);
77919304Speter
78019304Speter			if (dt_print_quantline(dtp, fp, data[bin],
78119304Speter			    normal, total, positives, negatives) < 0)
78219304Speter				return (-1);
78319304Speter		}
78419304Speter
78519304Speter		assert(value < next);
78619304Speter		bin++;
78719304Speter
78819304Speter		if ((value += step) != next)
78919304Speter			continue;
79019304Speter
79119304Speter		next = value * factor;
79219304Speter		step = next > nsteps ? next / nsteps : 1;
79319304Speter		order++;
79419304Speter	}
79519304Speter
79619304Speter	if (last_bin < bin)
79719304Speter		return (0);
79819304Speter
79919304Speter	assert(last_bin == bin);
800254225Speter	(void) snprintf(c, sizeof (c), ">= %lld", (long long)value);
80119304Speter
80219304Speter	if (dt_printf(dtp, fp, "%16s ", c) < 0)
80319304Speter		return (-1);
80419304Speter
80519304Speter	return (dt_print_quantline(dtp, fp, data[bin], normal,
80619304Speter	    total, positives, negatives));
80719304Speter}
80819304Speter
80919304Speter/*ARGSUSED*/
810254225Speterstatic int
811254225Speterdt_print_average(dtrace_hdl_t *dtp, FILE *fp, caddr_t addr,
812254225Speter    size_t size, uint64_t normal)
813254225Speter{
814254225Speter	/* LINTED - alignment */
815254225Speter	int64_t *data = (int64_t *)addr;
816254225Speter
817254225Speter	return (dt_printf(dtp, fp, " %16lld", data[0] ?
818254225Speter	    (long long)(data[1] / (int64_t)normal / data[0]) : 0));
819254225Speter}
820254225Speter
821254225Speter/*ARGSUSED*/
822254225Speterstatic int
823254225Speterdt_print_stddev(dtrace_hdl_t *dtp, FILE *fp, caddr_t addr,
824254225Speter    size_t size, uint64_t normal)
825254225Speter{
826254225Speter	/* LINTED - alignment */
827254225Speter	uint64_t *data = (uint64_t *)addr;
828254225Speter
829254225Speter	return (dt_printf(dtp, fp, " %16llu", data[0] ?
830254225Speter	    (unsigned long long) dt_stddev(data, normal) : 0));
831254225Speter}
832254225Speter
833254225Speter/*ARGSUSED*/
834254225Speterint
835254225Speterdt_print_bytes(dtrace_hdl_t *dtp, FILE *fp, caddr_t addr,
836254225Speter    size_t nbytes, int width, int quiet, int forceraw)
837254225Speter{
838254225Speter	/*
839254225Speter	 * If the byte stream is a series of printable characters, followed by
840254225Speter	 * a terminating byte, we print it out as a string.  Otherwise, we
841254225Speter	 * assume that it's something else and just print the bytes.
842254225Speter	 */
843254225Speter	int i, j, margin = 5;
844254225Speter	char *c = (char *)addr;
845254225Speter
846254225Speter	if (nbytes == 0)
847254225Speter		return (0);
848254225Speter
849254225Speter	if (forceraw)
850254225Speter		goto raw;
851254225Speter
852254225Speter	if (dtp->dt_options[DTRACEOPT_RAWBYTES] != DTRACEOPT_UNSET)
853254225Speter		goto raw;
854254225Speter
855254225Speter	for (i = 0; i < nbytes; i++) {
856254225Speter		/*
857254225Speter		 * We define a "printable character" to be one for which
858254225Speter		 * isprint(3C) returns non-zero, isspace(3C) returns non-zero,
859254225Speter		 * or a character which is either backspace or the bell.
860254225Speter		 * Backspace and the bell are regrettably special because
861254225Speter		 * they fail the first two tests -- and yet they are entirely
862254225Speter		 * printable.  These are the only two control characters that
863254225Speter		 * have meaning for the terminal and for which isprint(3C) and
864254225Speter		 * isspace(3C) return 0.
865254225Speter		 */
866254225Speter		if (isprint(c[i]) || isspace(c[i]) ||
867254225Speter		    c[i] == '\b' || c[i] == '\a')
868254225Speter			continue;
869254225Speter
870254225Speter		if (c[i] == '\0' && i > 0) {
871254225Speter			/*
872254225Speter			 * This looks like it might be a string.  Before we
873254225Speter			 * assume that it is indeed a string, check the
874254225Speter			 * remainder of the byte range; if it contains
875254225Speter			 * additional non-nul characters, we'll assume that
876254225Speter			 * it's a binary stream that just happens to look like
877254225Speter			 * a string, and we'll print out the individual bytes.
878254225Speter			 */
879254225Speter			for (j = i + 1; j < nbytes; j++) {
880254225Speter				if (c[j] != '\0')
881254225Speter					break;
882254225Speter			}
883254225Speter
884254225Speter			if (j != nbytes)
885254225Speter				break;
886254225Speter
887254225Speter			if (quiet)
888254225Speter				return (dt_printf(dtp, fp, "%s", c));
889254225Speter			else
890254225Speter				return (dt_printf(dtp, fp, "  %-*s", width, c));
891254225Speter		}
892254225Speter
893254225Speter		break;
894254225Speter	}
895254225Speter
896254225Speter	if (i == nbytes) {
897254225Speter		/*
898254225Speter		 * The byte range is all printable characters, but there is
899254225Speter		 * no trailing nul byte.  We'll assume that it's a string and
900254225Speter		 * print it as such.
901254225Speter		 */
902254225Speter		char *s = alloca(nbytes + 1);
903254225Speter		bcopy(c, s, nbytes);
904254225Speter		s[nbytes] = '\0';
905254225Speter		return (dt_printf(dtp, fp, "  %-*s", width, s));
906254225Speter	}
907254225Speter
908254225Speterraw:
909254225Speter	if (dt_printf(dtp, fp, "\n%*s      ", margin, "") < 0)
910254225Speter		return (-1);
911254225Speter
912254225Speter	for (i = 0; i < 16; i++)
913254225Speter		if (dt_printf(dtp, fp, "  %c", "0123456789abcdef"[i]) < 0)
914254225Speter			return (-1);
915
916	if (dt_printf(dtp, fp, "  0123456789abcdef\n") < 0)
917		return (-1);
918
919
920	for (i = 0; i < nbytes; i += 16) {
921		if (dt_printf(dtp, fp, "%*s%5x:", margin, "", i) < 0)
922			return (-1);
923
924		for (j = i; j < i + 16 && j < nbytes; j++) {
925			if (dt_printf(dtp, fp, " %02x", (uchar_t)c[j]) < 0)
926				return (-1);
927		}
928
929		while (j++ % 16) {
930			if (dt_printf(dtp, fp, "   ") < 0)
931				return (-1);
932		}
933
934		if (dt_printf(dtp, fp, "  ") < 0)
935			return (-1);
936
937		for (j = i; j < i + 16 && j < nbytes; j++) {
938			if (dt_printf(dtp, fp, "%c",
939			    c[j] < ' ' || c[j] > '~' ? '.' : c[j]) < 0)
940				return (-1);
941		}
942
943		if (dt_printf(dtp, fp, "\n") < 0)
944			return (-1);
945	}
946
947	return (0);
948}
949
950int
951dt_print_stack(dtrace_hdl_t *dtp, FILE *fp, const char *format,
952    caddr_t addr, int depth, int size)
953{
954	dtrace_syminfo_t dts;
955	GElf_Sym sym;
956	int i, indent;
957	char c[PATH_MAX * 2];
958	uint64_t pc;
959
960	if (dt_printf(dtp, fp, "\n") < 0)
961		return (-1);
962
963	if (format == NULL)
964		format = "%s";
965
966	if (dtp->dt_options[DTRACEOPT_STACKINDENT] != DTRACEOPT_UNSET)
967		indent = (int)dtp->dt_options[DTRACEOPT_STACKINDENT];
968	else
969		indent = _dtrace_stkindent;
970
971	for (i = 0; i < depth; i++) {
972		switch (size) {
973		case sizeof (uint32_t):
974			/* LINTED - alignment */
975			pc = *((uint32_t *)addr);
976			break;
977
978		case sizeof (uint64_t):
979			/* LINTED - alignment */
980			pc = *((uint64_t *)addr);
981			break;
982
983		default:
984			return (dt_set_errno(dtp, EDT_BADSTACKPC));
985		}
986
987		if (pc == 0)
988			break;
989
990		addr += size;
991
992		if (dt_printf(dtp, fp, "%*s", indent, "") < 0)
993			return (-1);
994
995		if (dtrace_lookup_by_addr(dtp, pc, &sym, &dts) == 0) {
996			if (pc > sym.st_value) {
997				(void) snprintf(c, sizeof (c), "%s`%s+0x%llx",
998				    dts.dts_object, dts.dts_name,
999				    (u_longlong_t)(pc - sym.st_value));
1000			} else {
1001				(void) snprintf(c, sizeof (c), "%s`%s",
1002				    dts.dts_object, dts.dts_name);
1003			}
1004		} else {
1005			/*
1006			 * We'll repeat the lookup, but this time we'll specify
1007			 * a NULL GElf_Sym -- indicating that we're only
1008			 * interested in the containing module.
1009			 */
1010			if (dtrace_lookup_by_addr(dtp, pc, NULL, &dts) == 0) {
1011				(void) snprintf(c, sizeof (c), "%s`0x%llx",
1012				    dts.dts_object, (u_longlong_t)pc);
1013			} else {
1014				(void) snprintf(c, sizeof (c), "0x%llx",
1015				    (u_longlong_t)pc);
1016			}
1017		}
1018
1019		if (dt_printf(dtp, fp, format, c) < 0)
1020			return (-1);
1021
1022		if (dt_printf(dtp, fp, "\n") < 0)
1023			return (-1);
1024	}
1025
1026	return (0);
1027}
1028
1029int
1030dt_print_ustack(dtrace_hdl_t *dtp, FILE *fp, const char *format,
1031    caddr_t addr, uint64_t arg)
1032{
1033	/* LINTED - alignment */
1034	uint64_t *pc = (uint64_t *)addr;
1035	uint32_t depth = DTRACE_USTACK_NFRAMES(arg);
1036	uint32_t strsize = DTRACE_USTACK_STRSIZE(arg);
1037	const char *strbase = addr + (depth + 1) * sizeof (uint64_t);
1038	const char *str = strsize ? strbase : NULL;
1039	int err = 0;
1040
1041	char name[PATH_MAX], objname[PATH_MAX], c[PATH_MAX * 2];
1042	struct ps_prochandle *P;
1043	GElf_Sym sym;
1044	int i, indent;
1045	pid_t pid;
1046
1047	if (depth == 0)
1048		return (0);
1049
1050	pid = (pid_t)*pc++;
1051
1052	if (dt_printf(dtp, fp, "\n") < 0)
1053		return (-1);
1054
1055	if (format == NULL)
1056		format = "%s";
1057
1058	if (dtp->dt_options[DTRACEOPT_STACKINDENT] != DTRACEOPT_UNSET)
1059		indent = (int)dtp->dt_options[DTRACEOPT_STACKINDENT];
1060	else
1061		indent = _dtrace_stkindent;
1062
1063	/*
1064	 * Ultimately, we need to add an entry point in the library vector for
1065	 * determining <symbol, offset> from <pid, address>.  For now, if
1066	 * this is a vector open, we just print the raw address or string.
1067	 */
1068	if (dtp->dt_vector == NULL)
1069		P = dt_proc_grab(dtp, pid, PGRAB_RDONLY | PGRAB_FORCE, 0);
1070	else
1071		P = NULL;
1072
1073	if (P != NULL)
1074		dt_proc_lock(dtp, P); /* lock handle while we perform lookups */
1075
1076	for (i = 0; i < depth && pc[i] != 0; i++) {
1077		const prmap_t *map;
1078
1079		if ((err = dt_printf(dtp, fp, "%*s", indent, "")) < 0)
1080			break;
1081
1082		if (P != NULL && Plookup_by_addr(P, pc[i],
1083		    name, sizeof (name), &sym) == 0) {
1084			(void) Pobjname(P, pc[i], objname, sizeof (objname));
1085
1086			if (pc[i] > sym.st_value) {
1087				(void) snprintf(c, sizeof (c),
1088				    "%s`%s+0x%llx", dt_basename(objname), name,
1089				    (u_longlong_t)(pc[i] - sym.st_value));
1090			} else {
1091				(void) snprintf(c, sizeof (c),
1092				    "%s`%s", dt_basename(objname), name);
1093			}
1094		} else if (str != NULL && str[0] != '\0' && str[0] != '@' &&
1095		    (P != NULL && ((map = Paddr_to_map(P, pc[i])) == NULL ||
1096		    (map->pr_mflags & MA_WRITE)))) {
1097			/*
1098			 * If the current string pointer in the string table
1099			 * does not point to an empty string _and_ the program
1100			 * counter falls in a writable region, we'll use the
1101			 * string from the string table instead of the raw
1102			 * address.  This last condition is necessary because
1103			 * some (broken) ustack helpers will return a string
1104			 * even for a program counter that they can't
1105			 * identify.  If we have a string for a program
1106			 * counter that falls in a segment that isn't
1107			 * writable, we assume that we have fallen into this
1108			 * case and we refuse to use the string.
1109			 */
1110			(void) snprintf(c, sizeof (c), "%s", str);
1111		} else {
1112			if (P != NULL && Pobjname(P, pc[i], objname,
1113			    sizeof (objname)) != 0) {
1114				(void) snprintf(c, sizeof (c), "%s`0x%llx",
1115				    dt_basename(objname), (u_longlong_t)pc[i]);
1116			} else {
1117				(void) snprintf(c, sizeof (c), "0x%llx",
1118				    (u_longlong_t)pc[i]);
1119			}
1120		}
1121
1122		if ((err = dt_printf(dtp, fp, format, c)) < 0)
1123			break;
1124
1125		if ((err = dt_printf(dtp, fp, "\n")) < 0)
1126			break;
1127
1128		if (str != NULL && str[0] == '@') {
1129			/*
1130			 * If the first character of the string is an "at" sign,
1131			 * then the string is inferred to be an annotation --
1132			 * and it is printed out beneath the frame and offset
1133			 * with brackets.
1134			 */
1135			if ((err = dt_printf(dtp, fp, "%*s", indent, "")) < 0)
1136				break;
1137
1138			(void) snprintf(c, sizeof (c), "  [ %s ]", &str[1]);
1139
1140			if ((err = dt_printf(dtp, fp, format, c)) < 0)
1141				break;
1142
1143			if ((err = dt_printf(dtp, fp, "\n")) < 0)
1144				break;
1145		}
1146
1147		if (str != NULL) {
1148			str += strlen(str) + 1;
1149			if (str - strbase >= strsize)
1150				str = NULL;
1151		}
1152	}
1153
1154	if (P != NULL) {
1155		dt_proc_unlock(dtp, P);
1156		dt_proc_release(dtp, P);
1157	}
1158
1159	return (err);
1160}
1161
1162static int
1163dt_print_usym(dtrace_hdl_t *dtp, FILE *fp, caddr_t addr, dtrace_actkind_t act)
1164{
1165	/* LINTED - alignment */
1166	uint64_t pid = ((uint64_t *)addr)[0];
1167	/* LINTED - alignment */
1168	uint64_t pc = ((uint64_t *)addr)[1];
1169	const char *format = "  %-50s";
1170	char *s;
1171	int n, len = 256;
1172
1173	if (act == DTRACEACT_USYM && dtp->dt_vector == NULL) {
1174		struct ps_prochandle *P;
1175
1176		if ((P = dt_proc_grab(dtp, pid,
1177		    PGRAB_RDONLY | PGRAB_FORCE, 0)) != NULL) {
1178			GElf_Sym sym;
1179
1180			dt_proc_lock(dtp, P);
1181
1182			if (Plookup_by_addr(P, pc, NULL, 0, &sym) == 0)
1183				pc = sym.st_value;
1184
1185			dt_proc_unlock(dtp, P);
1186			dt_proc_release(dtp, P);
1187		}
1188	}
1189
1190	do {
1191		n = len;
1192		s = alloca(n);
1193	} while ((len = dtrace_uaddr2str(dtp, pid, pc, s, n)) > n);
1194
1195	return (dt_printf(dtp, fp, format, s));
1196}
1197
1198int
1199dt_print_umod(dtrace_hdl_t *dtp, FILE *fp, const char *format, caddr_t addr)
1200{
1201	/* LINTED - alignment */
1202	uint64_t pid = ((uint64_t *)addr)[0];
1203	/* LINTED - alignment */
1204	uint64_t pc = ((uint64_t *)addr)[1];
1205	int err = 0;
1206
1207	char objname[PATH_MAX], c[PATH_MAX * 2];
1208	struct ps_prochandle *P;
1209
1210	if (format == NULL)
1211		format = "  %-50s";
1212
1213	/*
1214	 * See the comment in dt_print_ustack() for the rationale for
1215	 * printing raw addresses in the vectored case.
1216	 */
1217	if (dtp->dt_vector == NULL)
1218		P = dt_proc_grab(dtp, pid, PGRAB_RDONLY | PGRAB_FORCE, 0);
1219	else
1220		P = NULL;
1221
1222	if (P != NULL)
1223		dt_proc_lock(dtp, P); /* lock handle while we perform lookups */
1224
1225	if (P != NULL && Pobjname(P, pc, objname, sizeof (objname)) != 0) {
1226		(void) snprintf(c, sizeof (c), "%s", dt_basename(objname));
1227	} else {
1228		(void) snprintf(c, sizeof (c), "0x%llx", (u_longlong_t)pc);
1229	}
1230
1231	err = dt_printf(dtp, fp, format, c);
1232
1233	if (P != NULL) {
1234		dt_proc_unlock(dtp, P);
1235		dt_proc_release(dtp, P);
1236	}
1237
1238	return (err);
1239}
1240
1241int
1242dt_print_memory(dtrace_hdl_t *dtp, FILE *fp, caddr_t addr)
1243{
1244	int quiet = (dtp->dt_options[DTRACEOPT_QUIET] != DTRACEOPT_UNSET);
1245	size_t nbytes = *((uintptr_t *) addr);
1246
1247	return (dt_print_bytes(dtp, fp, addr + sizeof(uintptr_t),
1248	    nbytes, 50, quiet, 1));
1249}
1250
1251typedef struct dt_type_cbdata {
1252	dtrace_hdl_t		*dtp;
1253	dtrace_typeinfo_t	dtt;
1254	caddr_t			addr;
1255	caddr_t			addrend;
1256	const char		*name;
1257	int			f_type;
1258	int			indent;
1259	int			type_width;
1260	int			name_width;
1261	FILE			*fp;
1262} dt_type_cbdata_t;
1263
1264static int	dt_print_type_data(dt_type_cbdata_t *, ctf_id_t);
1265
1266static int
1267dt_print_type_member(const char *name, ctf_id_t type, ulong_t off, void *arg)
1268{
1269	dt_type_cbdata_t cbdata;
1270	dt_type_cbdata_t *cbdatap = arg;
1271	ssize_t ssz;
1272
1273	if ((ssz = ctf_type_size(cbdatap->dtt.dtt_ctfp, type)) <= 0)
1274		return (0);
1275
1276	off /= 8;
1277
1278	cbdata = *cbdatap;
1279	cbdata.name = name;
1280	cbdata.addr += off;
1281	cbdata.addrend = cbdata.addr + ssz;
1282
1283	return (dt_print_type_data(&cbdata, type));
1284}
1285
1286static int
1287dt_print_type_width(const char *name, ctf_id_t type, ulong_t off, void *arg)
1288{
1289	char buf[DT_TYPE_NAMELEN];
1290	char *p;
1291	dt_type_cbdata_t *cbdatap = arg;
1292	size_t sz = strlen(name);
1293
1294	ctf_type_name(cbdatap->dtt.dtt_ctfp, type, buf, sizeof (buf));
1295
1296	if ((p = strchr(buf, '[')) != NULL)
1297		p[-1] = '\0';
1298	else
1299		p = "";
1300
1301	sz += strlen(p);
1302
1303	if (sz > cbdatap->name_width)
1304		cbdatap->name_width = sz;
1305
1306	sz = strlen(buf);
1307
1308	if (sz > cbdatap->type_width)
1309		cbdatap->type_width = sz;
1310
1311	return (0);
1312}
1313
1314static int
1315dt_print_type_data(dt_type_cbdata_t *cbdatap, ctf_id_t type)
1316{
1317	caddr_t addr = cbdatap->addr;
1318	caddr_t addrend = cbdatap->addrend;
1319	char buf[DT_TYPE_NAMELEN];
1320	char *p;
1321	int cnt = 0;
1322	uint_t kind = ctf_type_kind(cbdatap->dtt.dtt_ctfp, type);
1323	ssize_t ssz = ctf_type_size(cbdatap->dtt.dtt_ctfp, type);
1324
1325	ctf_type_name(cbdatap->dtt.dtt_ctfp, type, buf, sizeof (buf));
1326
1327	if ((p = strchr(buf, '[')) != NULL)
1328		p[-1] = '\0';
1329	else
1330		p = "";
1331
1332	if (cbdatap->f_type) {
1333		int type_width = roundup(cbdatap->type_width + 1, 4);
1334		int name_width = roundup(cbdatap->name_width + 1, 4);
1335
1336		name_width -= strlen(cbdatap->name);
1337
1338		dt_printf(cbdatap->dtp, cbdatap->fp, "%*s%-*s%s%-*s	= ",cbdatap->indent * 4,"",type_width,buf,cbdatap->name,name_width,p);
1339	}
1340
1341	while (addr < addrend) {
1342		dt_type_cbdata_t cbdata;
1343		ctf_arinfo_t arinfo;
1344		ctf_encoding_t cte;
1345		uintptr_t *up;
1346		void *vp = addr;
1347		cbdata = *cbdatap;
1348		cbdata.name = "";
1349		cbdata.addr = addr;
1350		cbdata.addrend = addr + ssz;
1351		cbdata.f_type = 0;
1352		cbdata.indent++;
1353		cbdata.type_width = 0;
1354		cbdata.name_width = 0;
1355
1356		if (cnt > 0)
1357			dt_printf(cbdatap->dtp, cbdatap->fp, "%*s", cbdatap->indent * 4,"");
1358
1359		switch (kind) {
1360		case CTF_K_INTEGER:
1361			if (ctf_type_encoding(cbdatap->dtt.dtt_ctfp, type, &cte) != 0)
1362				return (-1);
1363			if ((cte.cte_format & CTF_INT_SIGNED) != 0)
1364				switch (cte.cte_bits) {
1365				case 8:
1366					if (isprint(*((char *) vp)))
1367						dt_printf(cbdatap->dtp, cbdatap->fp, "'%c', ", *((char *) vp));
1368					dt_printf(cbdatap->dtp, cbdatap->fp, "%d (0x%x);\n", *((char *) vp), *((char *) vp));
1369					break;
1370				case 16:
1371					dt_printf(cbdatap->dtp, cbdatap->fp, "%hd (0x%hx);\n", *((short *) vp), *((u_short *) vp));
1372					break;
1373				case 32:
1374					dt_printf(cbdatap->dtp, cbdatap->fp, "%d (0x%x);\n", *((int *) vp), *((u_int *) vp));
1375					break;
1376				case 64:
1377					dt_printf(cbdatap->dtp, cbdatap->fp, "%jd (0x%jx);\n", *((long long *) vp), *((unsigned long long *) vp));
1378					break;
1379				default:
1380					dt_printf(cbdatap->dtp, cbdatap->fp, "CTF_K_INTEGER: format %x offset %u bits %u\n",cte.cte_format,cte.cte_offset,cte.cte_bits);
1381					break;
1382				}
1383			else
1384				switch (cte.cte_bits) {
1385				case 8:
1386					dt_printf(cbdatap->dtp, cbdatap->fp, "%u (0x%x);\n", *((uint8_t *) vp) & 0xff, *((uint8_t *) vp) & 0xff);
1387					break;
1388				case 16:
1389					dt_printf(cbdatap->dtp, cbdatap->fp, "%hu (0x%hx);\n", *((u_short *) vp), *((u_short *) vp));
1390					break;
1391				case 32:
1392					dt_printf(cbdatap->dtp, cbdatap->fp, "%u (0x%x);\n", *((u_int *) vp), *((u_int *) vp));
1393					break;
1394				case 64:
1395					dt_printf(cbdatap->dtp, cbdatap->fp, "%ju (0x%jx);\n", *((unsigned long long *) vp), *((unsigned long long *) vp));
1396					break;
1397				default:
1398					dt_printf(cbdatap->dtp, cbdatap->fp, "CTF_K_INTEGER: format %x offset %u bits %u\n",cte.cte_format,cte.cte_offset,cte.cte_bits);
1399					break;
1400				}
1401			break;
1402		case CTF_K_FLOAT:
1403			dt_printf(cbdatap->dtp, cbdatap->fp, "CTF_K_FLOAT: format %x offset %u bits %u\n",cte.cte_format,cte.cte_offset,cte.cte_bits);
1404			break;
1405		case CTF_K_POINTER:
1406			dt_printf(cbdatap->dtp, cbdatap->fp, "%p;\n", *((void **) addr));
1407			break;
1408		case CTF_K_ARRAY:
1409			if (ctf_array_info(cbdatap->dtt.dtt_ctfp, type, &arinfo) != 0)
1410				return (-1);
1411			dt_printf(cbdatap->dtp, cbdatap->fp, "{\n%*s",cbdata.indent * 4,"");
1412			dt_print_type_data(&cbdata, arinfo.ctr_contents);
1413			dt_printf(cbdatap->dtp, cbdatap->fp, "%*s};\n",cbdatap->indent * 4,"");
1414			break;
1415		case CTF_K_FUNCTION:
1416			dt_printf(cbdatap->dtp, cbdatap->fp, "CTF_K_FUNCTION:\n");
1417			break;
1418		case CTF_K_STRUCT:
1419			cbdata.f_type = 1;
1420			if (ctf_member_iter(cbdatap->dtt.dtt_ctfp, type,
1421			    dt_print_type_width, &cbdata) != 0)
1422				return (-1);
1423			dt_printf(cbdatap->dtp, cbdatap->fp, "{\n");
1424			if (ctf_member_iter(cbdatap->dtt.dtt_ctfp, type,
1425			    dt_print_type_member, &cbdata) != 0)
1426				return (-1);
1427			dt_printf(cbdatap->dtp, cbdatap->fp, "%*s};\n",cbdatap->indent * 4,"");
1428			break;
1429		case CTF_K_UNION:
1430			cbdata.f_type = 1;
1431			if (ctf_member_iter(cbdatap->dtt.dtt_ctfp, type,
1432			    dt_print_type_width, &cbdata) != 0)
1433				return (-1);
1434			dt_printf(cbdatap->dtp, cbdatap->fp, "{\n");
1435			if (ctf_member_iter(cbdatap->dtt.dtt_ctfp, type,
1436			    dt_print_type_member, &cbdata) != 0)
1437				return (-1);
1438			dt_printf(cbdatap->dtp, cbdatap->fp, "%*s};\n",cbdatap->indent * 4,"");
1439			break;
1440		case CTF_K_ENUM:
1441			dt_printf(cbdatap->dtp, cbdatap->fp, "%s;\n", ctf_enum_name(cbdatap->dtt.dtt_ctfp, type, *((int *) vp)));
1442			break;
1443		case CTF_K_TYPEDEF:
1444			dt_print_type_data(&cbdata, ctf_type_reference(cbdatap->dtt.dtt_ctfp,type));
1445			break;
1446		case CTF_K_VOLATILE:
1447			if (cbdatap->f_type)
1448				dt_printf(cbdatap->dtp, cbdatap->fp, "volatile ");
1449			dt_print_type_data(&cbdata, ctf_type_reference(cbdatap->dtt.dtt_ctfp,type));
1450			break;
1451		case CTF_K_CONST:
1452			if (cbdatap->f_type)
1453				dt_printf(cbdatap->dtp, cbdatap->fp, "const ");
1454			dt_print_type_data(&cbdata, ctf_type_reference(cbdatap->dtt.dtt_ctfp,type));
1455			break;
1456		case CTF_K_RESTRICT:
1457			if (cbdatap->f_type)
1458				dt_printf(cbdatap->dtp, cbdatap->fp, "restrict ");
1459			dt_print_type_data(&cbdata, ctf_type_reference(cbdatap->dtt.dtt_ctfp,type));
1460			break;
1461		default:
1462			break;
1463		}
1464
1465		addr += ssz;
1466		cnt++;
1467	}
1468
1469	return (0);
1470}
1471
1472static int
1473dt_print_type(dtrace_hdl_t *dtp, FILE *fp, caddr_t addr)
1474{
1475	caddr_t addrend;
1476	char *p;
1477	dtrace_typeinfo_t dtt;
1478	dt_type_cbdata_t cbdata;
1479	int num = 0;
1480	int quiet = (dtp->dt_options[DTRACEOPT_QUIET] != DTRACEOPT_UNSET);
1481	ssize_t ssz;
1482
1483	if (!quiet)
1484		dt_printf(dtp, fp, "\n");
1485
1486	/* Get the total number of bytes of data buffered. */
1487	size_t nbytes = *((uintptr_t *) addr);
1488	addr += sizeof(uintptr_t);
1489
1490	/*
1491	 * Get the size of the type so that we can check that it matches
1492	 * the CTF data we look up and so that we can figure out how many
1493	 * type elements are buffered.
1494	 */
1495	size_t typs = *((uintptr_t *) addr);
1496	addr += sizeof(uintptr_t);
1497
1498	/*
1499	 * Point to the type string in the buffer. Get it's string
1500	 * length and round it up to become the offset to the start
1501	 * of the buffered type data which we would like to be aligned
1502	 * for easy access.
1503	 */
1504	char *strp = (char *) addr;
1505	int offset = roundup(strlen(strp) + 1, sizeof(uintptr_t));
1506
1507	/*
1508	 * The type string might have a format such as 'int [20]'.
1509	 * Check if there is an array dimension present.
1510	 */
1511	if ((p = strchr(strp, '[')) != NULL) {
1512		/* Strip off the array dimension. */
1513		*p++ = '\0';
1514
1515		for (; *p != '\0' && *p != ']'; p++)
1516			num = num * 10 + *p - '0';
1517	} else
1518		/* No array dimension, so default. */
1519		num = 1;
1520
1521	/* Lookup the CTF type from the type string. */
1522	if (dtrace_lookup_by_type(dtp,  DTRACE_OBJ_EVERY, strp, &dtt) < 0)
1523		return (-1);
1524
1525	/* Offset the buffer address to the start of the data... */
1526	addr += offset;
1527
1528	ssz = ctf_type_size(dtt.dtt_ctfp, dtt.dtt_type);
1529
1530	if (typs != ssz) {
1531		printf("Expected type size from buffer (%lu) to match type size looked up now (%ld)\n", (u_long) typs, (long) ssz);
1532		return (-1);
1533	}
1534
1535	cbdata.dtp = dtp;
1536	cbdata.dtt = dtt;
1537	cbdata.name = "";
1538	cbdata.addr = addr;
1539	cbdata.addrend = addr + nbytes;
1540	cbdata.indent = 1;
1541	cbdata.f_type = 1;
1542	cbdata.type_width = 0;
1543	cbdata.name_width = 0;
1544	cbdata.fp = fp;
1545
1546	return (dt_print_type_data(&cbdata, dtt.dtt_type));
1547}
1548
1549static int
1550dt_print_sym(dtrace_hdl_t *dtp, FILE *fp, const char *format, caddr_t addr)
1551{
1552	/* LINTED - alignment */
1553	uint64_t pc = *((uint64_t *)addr);
1554	dtrace_syminfo_t dts;
1555	GElf_Sym sym;
1556	char c[PATH_MAX * 2];
1557
1558	if (format == NULL)
1559		format = "  %-50s";
1560
1561	if (dtrace_lookup_by_addr(dtp, pc, &sym, &dts) == 0) {
1562		(void) snprintf(c, sizeof (c), "%s`%s",
1563		    dts.dts_object, dts.dts_name);
1564	} else {
1565		/*
1566		 * We'll repeat the lookup, but this time we'll specify a
1567		 * NULL GElf_Sym -- indicating that we're only interested in
1568		 * the containing module.
1569		 */
1570		if (dtrace_lookup_by_addr(dtp, pc, NULL, &dts) == 0) {
1571			(void) snprintf(c, sizeof (c), "%s`0x%llx",
1572			    dts.dts_object, (u_longlong_t)pc);
1573		} else {
1574			(void) snprintf(c, sizeof (c), "0x%llx",
1575			    (u_longlong_t)pc);
1576		}
1577	}
1578
1579	if (dt_printf(dtp, fp, format, c) < 0)
1580		return (-1);
1581
1582	return (0);
1583}
1584
1585int
1586dt_print_mod(dtrace_hdl_t *dtp, FILE *fp, const char *format, caddr_t addr)
1587{
1588	/* LINTED - alignment */
1589	uint64_t pc = *((uint64_t *)addr);
1590	dtrace_syminfo_t dts;
1591	char c[PATH_MAX * 2];
1592
1593	if (format == NULL)
1594		format = "  %-50s";
1595
1596	if (dtrace_lookup_by_addr(dtp, pc, NULL, &dts) == 0) {
1597		(void) snprintf(c, sizeof (c), "%s", dts.dts_object);
1598	} else {
1599		(void) snprintf(c, sizeof (c), "0x%llx", (u_longlong_t)pc);
1600	}
1601
1602	if (dt_printf(dtp, fp, format, c) < 0)
1603		return (-1);
1604
1605	return (0);
1606}
1607
1608typedef struct dt_normal {
1609	dtrace_aggvarid_t dtnd_id;
1610	uint64_t dtnd_normal;
1611} dt_normal_t;
1612
1613static int
1614dt_normalize_agg(const dtrace_aggdata_t *aggdata, void *arg)
1615{
1616	dt_normal_t *normal = arg;
1617	dtrace_aggdesc_t *agg = aggdata->dtada_desc;
1618	dtrace_aggvarid_t id = normal->dtnd_id;
1619
1620	if (agg->dtagd_nrecs == 0)
1621		return (DTRACE_AGGWALK_NEXT);
1622
1623	if (agg->dtagd_varid != id)
1624		return (DTRACE_AGGWALK_NEXT);
1625
1626	((dtrace_aggdata_t *)aggdata)->dtada_normal = normal->dtnd_normal;
1627	return (DTRACE_AGGWALK_NORMALIZE);
1628}
1629
1630static int
1631dt_normalize(dtrace_hdl_t *dtp, caddr_t base, dtrace_recdesc_t *rec)
1632{
1633	dt_normal_t normal;
1634	caddr_t addr;
1635
1636	/*
1637	 * We (should) have two records:  the aggregation ID followed by the
1638	 * normalization value.
1639	 */
1640	addr = base + rec->dtrd_offset;
1641
1642	if (rec->dtrd_size != sizeof (dtrace_aggvarid_t))
1643		return (dt_set_errno(dtp, EDT_BADNORMAL));
1644
1645	/* LINTED - alignment */
1646	normal.dtnd_id = *((dtrace_aggvarid_t *)addr);
1647	rec++;
1648
1649	if (rec->dtrd_action != DTRACEACT_LIBACT)
1650		return (dt_set_errno(dtp, EDT_BADNORMAL));
1651
1652	if (rec->dtrd_arg != DT_ACT_NORMALIZE)
1653		return (dt_set_errno(dtp, EDT_BADNORMAL));
1654
1655	addr = base + rec->dtrd_offset;
1656
1657	switch (rec->dtrd_size) {
1658	case sizeof (uint64_t):
1659		/* LINTED - alignment */
1660		normal.dtnd_normal = *((uint64_t *)addr);
1661		break;
1662	case sizeof (uint32_t):
1663		/* LINTED - alignment */
1664		normal.dtnd_normal = *((uint32_t *)addr);
1665		break;
1666	case sizeof (uint16_t):
1667		/* LINTED - alignment */
1668		normal.dtnd_normal = *((uint16_t *)addr);
1669		break;
1670	case sizeof (uint8_t):
1671		normal.dtnd_normal = *((uint8_t *)addr);
1672		break;
1673	default:
1674		return (dt_set_errno(dtp, EDT_BADNORMAL));
1675	}
1676
1677	(void) dtrace_aggregate_walk(dtp, dt_normalize_agg, &normal);
1678
1679	return (0);
1680}
1681
1682static int
1683dt_denormalize_agg(const dtrace_aggdata_t *aggdata, void *arg)
1684{
1685	dtrace_aggdesc_t *agg = aggdata->dtada_desc;
1686	dtrace_aggvarid_t id = *((dtrace_aggvarid_t *)arg);
1687
1688	if (agg->dtagd_nrecs == 0)
1689		return (DTRACE_AGGWALK_NEXT);
1690
1691	if (agg->dtagd_varid != id)
1692		return (DTRACE_AGGWALK_NEXT);
1693
1694	return (DTRACE_AGGWALK_DENORMALIZE);
1695}
1696
1697static int
1698dt_clear_agg(const dtrace_aggdata_t *aggdata, void *arg)
1699{
1700	dtrace_aggdesc_t *agg = aggdata->dtada_desc;
1701	dtrace_aggvarid_t id = *((dtrace_aggvarid_t *)arg);
1702
1703	if (agg->dtagd_nrecs == 0)
1704		return (DTRACE_AGGWALK_NEXT);
1705
1706	if (agg->dtagd_varid != id)
1707		return (DTRACE_AGGWALK_NEXT);
1708
1709	return (DTRACE_AGGWALK_CLEAR);
1710}
1711
1712typedef struct dt_trunc {
1713	dtrace_aggvarid_t dttd_id;
1714	uint64_t dttd_remaining;
1715} dt_trunc_t;
1716
1717static int
1718dt_trunc_agg(const dtrace_aggdata_t *aggdata, void *arg)
1719{
1720	dt_trunc_t *trunc = arg;
1721	dtrace_aggdesc_t *agg = aggdata->dtada_desc;
1722	dtrace_aggvarid_t id = trunc->dttd_id;
1723
1724	if (agg->dtagd_nrecs == 0)
1725		return (DTRACE_AGGWALK_NEXT);
1726
1727	if (agg->dtagd_varid != id)
1728		return (DTRACE_AGGWALK_NEXT);
1729
1730	if (trunc->dttd_remaining == 0)
1731		return (DTRACE_AGGWALK_REMOVE);
1732
1733	trunc->dttd_remaining--;
1734	return (DTRACE_AGGWALK_NEXT);
1735}
1736
1737static int
1738dt_trunc(dtrace_hdl_t *dtp, caddr_t base, dtrace_recdesc_t *rec)
1739{
1740	dt_trunc_t trunc;
1741	caddr_t addr;
1742	int64_t remaining;
1743	int (*func)(dtrace_hdl_t *, dtrace_aggregate_f *, void *);
1744
1745	/*
1746	 * We (should) have two records:  the aggregation ID followed by the
1747	 * number of aggregation entries after which the aggregation is to be
1748	 * truncated.
1749	 */
1750	addr = base + rec->dtrd_offset;
1751
1752	if (rec->dtrd_size != sizeof (dtrace_aggvarid_t))
1753		return (dt_set_errno(dtp, EDT_BADTRUNC));
1754
1755	/* LINTED - alignment */
1756	trunc.dttd_id = *((dtrace_aggvarid_t *)addr);
1757	rec++;
1758
1759	if (rec->dtrd_action != DTRACEACT_LIBACT)
1760		return (dt_set_errno(dtp, EDT_BADTRUNC));
1761
1762	if (rec->dtrd_arg != DT_ACT_TRUNC)
1763		return (dt_set_errno(dtp, EDT_BADTRUNC));
1764
1765	addr = base + rec->dtrd_offset;
1766
1767	switch (rec->dtrd_size) {
1768	case sizeof (uint64_t):
1769		/* LINTED - alignment */
1770		remaining = *((int64_t *)addr);
1771		break;
1772	case sizeof (uint32_t):
1773		/* LINTED - alignment */
1774		remaining = *((int32_t *)addr);
1775		break;
1776	case sizeof (uint16_t):
1777		/* LINTED - alignment */
1778		remaining = *((int16_t *)addr);
1779		break;
1780	case sizeof (uint8_t):
1781		remaining = *((int8_t *)addr);
1782		break;
1783	default:
1784		return (dt_set_errno(dtp, EDT_BADNORMAL));
1785	}
1786
1787	if (remaining < 0) {
1788		func = dtrace_aggregate_walk_valsorted;
1789		remaining = -remaining;
1790	} else {
1791		func = dtrace_aggregate_walk_valrevsorted;
1792	}
1793
1794	assert(remaining >= 0);
1795	trunc.dttd_remaining = remaining;
1796
1797	(void) func(dtp, dt_trunc_agg, &trunc);
1798
1799	return (0);
1800}
1801
1802static int
1803dt_print_datum(dtrace_hdl_t *dtp, FILE *fp, dtrace_recdesc_t *rec,
1804    caddr_t addr, size_t size, uint64_t normal)
1805{
1806	int err;
1807	dtrace_actkind_t act = rec->dtrd_action;
1808
1809	switch (act) {
1810	case DTRACEACT_STACK:
1811		return (dt_print_stack(dtp, fp, NULL, addr,
1812		    rec->dtrd_arg, rec->dtrd_size / rec->dtrd_arg));
1813
1814	case DTRACEACT_USTACK:
1815	case DTRACEACT_JSTACK:
1816		return (dt_print_ustack(dtp, fp, NULL, addr, rec->dtrd_arg));
1817
1818	case DTRACEACT_USYM:
1819	case DTRACEACT_UADDR:
1820		return (dt_print_usym(dtp, fp, addr, act));
1821
1822	case DTRACEACT_UMOD:
1823		return (dt_print_umod(dtp, fp, NULL, addr));
1824
1825	case DTRACEACT_SYM:
1826		return (dt_print_sym(dtp, fp, NULL, addr));
1827
1828	case DTRACEACT_MOD:
1829		return (dt_print_mod(dtp, fp, NULL, addr));
1830
1831	case DTRACEAGG_QUANTIZE:
1832		return (dt_print_quantize(dtp, fp, addr, size, normal));
1833
1834	case DTRACEAGG_LQUANTIZE:
1835		return (dt_print_lquantize(dtp, fp, addr, size, normal));
1836
1837	case DTRACEAGG_LLQUANTIZE:
1838		return (dt_print_llquantize(dtp, fp, addr, size, normal));
1839
1840	case DTRACEAGG_AVG:
1841		return (dt_print_average(dtp, fp, addr, size, normal));
1842
1843	case DTRACEAGG_STDDEV:
1844		return (dt_print_stddev(dtp, fp, addr, size, normal));
1845
1846	default:
1847		break;
1848	}
1849
1850	switch (size) {
1851	case sizeof (uint64_t):
1852		err = dt_printf(dtp, fp, " %16lld",
1853		    /* LINTED - alignment */
1854		    (long long)*((uint64_t *)addr) / normal);
1855		break;
1856	case sizeof (uint32_t):
1857		/* LINTED - alignment */
1858		err = dt_printf(dtp, fp, " %8d", *((uint32_t *)addr) /
1859		    (uint32_t)normal);
1860		break;
1861	case sizeof (uint16_t):
1862		/* LINTED - alignment */
1863		err = dt_printf(dtp, fp, " %5d", *((uint16_t *)addr) /
1864		    (uint32_t)normal);
1865		break;
1866	case sizeof (uint8_t):
1867		err = dt_printf(dtp, fp, " %3d", *((uint8_t *)addr) /
1868		    (uint32_t)normal);
1869		break;
1870	default:
1871		err = dt_print_bytes(dtp, fp, addr, size, 50, 0, 0);
1872		break;
1873	}
1874
1875	return (err);
1876}
1877
1878int
1879dt_print_aggs(const dtrace_aggdata_t **aggsdata, int naggvars, void *arg)
1880{
1881	int i, aggact = 0;
1882	dt_print_aggdata_t *pd = arg;
1883	const dtrace_aggdata_t *aggdata = aggsdata[0];
1884	dtrace_aggdesc_t *agg = aggdata->dtada_desc;
1885	FILE *fp = pd->dtpa_fp;
1886	dtrace_hdl_t *dtp = pd->dtpa_dtp;
1887	dtrace_recdesc_t *rec;
1888	dtrace_actkind_t act;
1889	caddr_t addr;
1890	size_t size;
1891
1892	/*
1893	 * Iterate over each record description in the key, printing the traced
1894	 * data, skipping the first datum (the tuple member created by the
1895	 * compiler).
1896	 */
1897	for (i = 1; i < agg->dtagd_nrecs; i++) {
1898		rec = &agg->dtagd_rec[i];
1899		act = rec->dtrd_action;
1900		addr = aggdata->dtada_data + rec->dtrd_offset;
1901		size = rec->dtrd_size;
1902
1903		if (DTRACEACT_ISAGG(act)) {
1904			aggact = i;
1905			break;
1906		}
1907
1908		if (dt_print_datum(dtp, fp, rec, addr, size, 1) < 0)
1909			return (-1);
1910
1911		if (dt_buffered_flush(dtp, NULL, rec, aggdata,
1912		    DTRACE_BUFDATA_AGGKEY) < 0)
1913			return (-1);
1914	}
1915
1916	assert(aggact != 0);
1917
1918	for (i = (naggvars == 1 ? 0 : 1); i < naggvars; i++) {
1919		uint64_t normal;
1920
1921		aggdata = aggsdata[i];
1922		agg = aggdata->dtada_desc;
1923		rec = &agg->dtagd_rec[aggact];
1924		act = rec->dtrd_action;
1925		addr = aggdata->dtada_data + rec->dtrd_offset;
1926		size = rec->dtrd_size;
1927
1928		assert(DTRACEACT_ISAGG(act));
1929		normal = aggdata->dtada_normal;
1930
1931		if (dt_print_datum(dtp, fp, rec, addr, size, normal) < 0)
1932			return (-1);
1933
1934		if (dt_buffered_flush(dtp, NULL, rec, aggdata,
1935		    DTRACE_BUFDATA_AGGVAL) < 0)
1936			return (-1);
1937
1938		if (!pd->dtpa_allunprint)
1939			agg->dtagd_flags |= DTRACE_AGD_PRINTED;
1940	}
1941
1942	if (dt_printf(dtp, fp, "\n") < 0)
1943		return (-1);
1944
1945	if (dt_buffered_flush(dtp, NULL, NULL, aggdata,
1946	    DTRACE_BUFDATA_AGGFORMAT | DTRACE_BUFDATA_AGGLAST) < 0)
1947		return (-1);
1948
1949	return (0);
1950}
1951
1952int
1953dt_print_agg(const dtrace_aggdata_t *aggdata, void *arg)
1954{
1955	dt_print_aggdata_t *pd = arg;
1956	dtrace_aggdesc_t *agg = aggdata->dtada_desc;
1957	dtrace_aggvarid_t aggvarid = pd->dtpa_id;
1958
1959	if (pd->dtpa_allunprint) {
1960		if (agg->dtagd_flags & DTRACE_AGD_PRINTED)
1961			return (0);
1962	} else {
1963		/*
1964		 * If we're not printing all unprinted aggregations, then the
1965		 * aggregation variable ID denotes a specific aggregation
1966		 * variable that we should print -- skip any other aggregations
1967		 * that we encounter.
1968		 */
1969		if (agg->dtagd_nrecs == 0)
1970			return (0);
1971
1972		if (aggvarid != agg->dtagd_varid)
1973			return (0);
1974	}
1975
1976	return (dt_print_aggs(&aggdata, 1, arg));
1977}
1978
1979int
1980dt_setopt(dtrace_hdl_t *dtp, const dtrace_probedata_t *data,
1981    const char *option, const char *value)
1982{
1983	int len, rval;
1984	char *msg;
1985	const char *errstr;
1986	dtrace_setoptdata_t optdata;
1987
1988	bzero(&optdata, sizeof (optdata));
1989	(void) dtrace_getopt(dtp, option, &optdata.dtsda_oldval);
1990
1991	if (dtrace_setopt(dtp, option, value) == 0) {
1992		(void) dtrace_getopt(dtp, option, &optdata.dtsda_newval);
1993		optdata.dtsda_probe = data;
1994		optdata.dtsda_option = option;
1995		optdata.dtsda_handle = dtp;
1996
1997		if ((rval = dt_handle_setopt(dtp, &optdata)) != 0)
1998			return (rval);
1999
2000		return (0);
2001	}
2002
2003	errstr = dtrace_errmsg(dtp, dtrace_errno(dtp));
2004	len = strlen(option) + strlen(value) + strlen(errstr) + 80;
2005	msg = alloca(len);
2006
2007	(void) snprintf(msg, len, "couldn't set option \"%s\" to \"%s\": %s\n",
2008	    option, value, errstr);
2009
2010	if ((rval = dt_handle_liberr(dtp, data, msg)) == 0)
2011		return (0);
2012
2013	return (rval);
2014}
2015
2016static int
2017dt_consume_cpu(dtrace_hdl_t *dtp, FILE *fp, int cpu, dtrace_bufdesc_t *buf,
2018    dtrace_consume_probe_f *efunc, dtrace_consume_rec_f *rfunc, void *arg)
2019{
2020	dtrace_epid_t id;
2021	size_t offs, start = buf->dtbd_oldest, end = buf->dtbd_size;
2022	int flow = (dtp->dt_options[DTRACEOPT_FLOWINDENT] != DTRACEOPT_UNSET);
2023	int quiet = (dtp->dt_options[DTRACEOPT_QUIET] != DTRACEOPT_UNSET);
2024	int rval, i, n;
2025	dtrace_epid_t last = DTRACE_EPIDNONE;
2026	uint64_t tracememsize = 0;
2027	dtrace_probedata_t data;
2028	uint64_t drops;
2029	caddr_t addr;
2030
2031	bzero(&data, sizeof (data));
2032	data.dtpda_handle = dtp;
2033	data.dtpda_cpu = cpu;
2034
2035again:
2036	for (offs = start; offs < end; ) {
2037		dtrace_eprobedesc_t *epd;
2038
2039		/*
2040		 * We're guaranteed to have an ID.
2041		 */
2042		id = *(uint32_t *)((uintptr_t)buf->dtbd_data + offs);
2043
2044		if (id == DTRACE_EPIDNONE) {
2045			/*
2046			 * This is filler to assure proper alignment of the
2047			 * next record; we simply ignore it.
2048			 */
2049			offs += sizeof (id);
2050			continue;
2051		}
2052
2053		if ((rval = dt_epid_lookup(dtp, id, &data.dtpda_edesc,
2054		    &data.dtpda_pdesc)) != 0)
2055			return (rval);
2056
2057		epd = data.dtpda_edesc;
2058		data.dtpda_data = buf->dtbd_data + offs;
2059
2060		if (data.dtpda_edesc->dtepd_uarg != DT_ECB_DEFAULT) {
2061			rval = dt_handle(dtp, &data);
2062
2063			if (rval == DTRACE_CONSUME_NEXT)
2064				goto nextepid;
2065
2066			if (rval == DTRACE_CONSUME_ERROR)
2067				return (-1);
2068		}
2069
2070		if (flow)
2071			(void) dt_flowindent(dtp, &data, last, buf, offs);
2072
2073		rval = (*efunc)(&data, arg);
2074
2075		if (flow) {
2076			if (data.dtpda_flow == DTRACEFLOW_ENTRY)
2077				data.dtpda_indent += 2;
2078		}
2079
2080		if (rval == DTRACE_CONSUME_NEXT)
2081			goto nextepid;
2082
2083		if (rval == DTRACE_CONSUME_ABORT)
2084			return (dt_set_errno(dtp, EDT_DIRABORT));
2085
2086		if (rval != DTRACE_CONSUME_THIS)
2087			return (dt_set_errno(dtp, EDT_BADRVAL));
2088
2089		for (i = 0; i < epd->dtepd_nrecs; i++) {
2090			dtrace_recdesc_t *rec = &epd->dtepd_rec[i];
2091			dtrace_actkind_t act = rec->dtrd_action;
2092
2093			data.dtpda_data = buf->dtbd_data + offs +
2094			    rec->dtrd_offset;
2095			addr = data.dtpda_data;
2096
2097			if (act == DTRACEACT_LIBACT) {
2098				uint64_t arg = rec->dtrd_arg;
2099				dtrace_aggvarid_t id;
2100
2101				switch (arg) {
2102				case DT_ACT_CLEAR:
2103					/* LINTED - alignment */
2104					id = *((dtrace_aggvarid_t *)addr);
2105					(void) dtrace_aggregate_walk(dtp,
2106					    dt_clear_agg, &id);
2107					continue;
2108
2109				case DT_ACT_DENORMALIZE:
2110					/* LINTED - alignment */
2111					id = *((dtrace_aggvarid_t *)addr);
2112					(void) dtrace_aggregate_walk(dtp,
2113					    dt_denormalize_agg, &id);
2114					continue;
2115
2116				case DT_ACT_FTRUNCATE:
2117					if (fp == NULL)
2118						continue;
2119
2120					(void) fflush(fp);
2121					(void) ftruncate(fileno(fp), 0);
2122					(void) fseeko(fp, 0, SEEK_SET);
2123					continue;
2124
2125				case DT_ACT_NORMALIZE:
2126					if (i == epd->dtepd_nrecs - 1)
2127						return (dt_set_errno(dtp,
2128						    EDT_BADNORMAL));
2129
2130					if (dt_normalize(dtp,
2131					    buf->dtbd_data + offs, rec) != 0)
2132						return (-1);
2133
2134					i++;
2135					continue;
2136
2137				case DT_ACT_SETOPT: {
2138					uint64_t *opts = dtp->dt_options;
2139					dtrace_recdesc_t *valrec;
2140					uint32_t valsize;
2141					caddr_t val;
2142					int rv;
2143
2144					if (i == epd->dtepd_nrecs - 1) {
2145						return (dt_set_errno(dtp,
2146						    EDT_BADSETOPT));
2147					}
2148
2149					valrec = &epd->dtepd_rec[++i];
2150					valsize = valrec->dtrd_size;
2151
2152					if (valrec->dtrd_action != act ||
2153					    valrec->dtrd_arg != arg) {
2154						return (dt_set_errno(dtp,
2155						    EDT_BADSETOPT));
2156					}
2157
2158					if (valsize > sizeof (uint64_t)) {
2159						val = buf->dtbd_data + offs +
2160						    valrec->dtrd_offset;
2161					} else {
2162						val = "1";
2163					}
2164
2165					rv = dt_setopt(dtp, &data, addr, val);
2166
2167					if (rv != 0)
2168						return (-1);
2169
2170					flow = (opts[DTRACEOPT_FLOWINDENT] !=
2171					    DTRACEOPT_UNSET);
2172					quiet = (opts[DTRACEOPT_QUIET] !=
2173					    DTRACEOPT_UNSET);
2174
2175					continue;
2176				}
2177
2178				case DT_ACT_TRUNC:
2179					if (i == epd->dtepd_nrecs - 1)
2180						return (dt_set_errno(dtp,
2181						    EDT_BADTRUNC));
2182
2183					if (dt_trunc(dtp,
2184					    buf->dtbd_data + offs, rec) != 0)
2185						return (-1);
2186
2187					i++;
2188					continue;
2189
2190				default:
2191					continue;
2192				}
2193			}
2194
2195			if (act == DTRACEACT_TRACEMEM_DYNSIZE &&
2196			    rec->dtrd_size == sizeof (uint64_t)) {
2197			    	/* LINTED - alignment */
2198				tracememsize = *((unsigned long long *)addr);
2199				continue;
2200			}
2201
2202			rval = (*rfunc)(&data, rec, arg);
2203
2204			if (rval == DTRACE_CONSUME_NEXT)
2205				continue;
2206
2207			if (rval == DTRACE_CONSUME_ABORT)
2208				return (dt_set_errno(dtp, EDT_DIRABORT));
2209
2210			if (rval != DTRACE_CONSUME_THIS)
2211				return (dt_set_errno(dtp, EDT_BADRVAL));
2212
2213			if (act == DTRACEACT_STACK) {
2214				int depth = rec->dtrd_arg;
2215
2216				if (dt_print_stack(dtp, fp, NULL, addr, depth,
2217				    rec->dtrd_size / depth) < 0)
2218					return (-1);
2219				goto nextrec;
2220			}
2221
2222			if (act == DTRACEACT_USTACK ||
2223			    act == DTRACEACT_JSTACK) {
2224				if (dt_print_ustack(dtp, fp, NULL,
2225				    addr, rec->dtrd_arg) < 0)
2226					return (-1);
2227				goto nextrec;
2228			}
2229
2230			if (act == DTRACEACT_SYM) {
2231				if (dt_print_sym(dtp, fp, NULL, addr) < 0)
2232					return (-1);
2233				goto nextrec;
2234			}
2235
2236			if (act == DTRACEACT_MOD) {
2237				if (dt_print_mod(dtp, fp, NULL, addr) < 0)
2238					return (-1);
2239				goto nextrec;
2240			}
2241
2242			if (act == DTRACEACT_USYM || act == DTRACEACT_UADDR) {
2243				if (dt_print_usym(dtp, fp, addr, act) < 0)
2244					return (-1);
2245				goto nextrec;
2246			}
2247
2248			if (act == DTRACEACT_UMOD) {
2249				if (dt_print_umod(dtp, fp, NULL, addr) < 0)
2250					return (-1);
2251				goto nextrec;
2252			}
2253
2254			if (act == DTRACEACT_PRINTM) {
2255				if (dt_print_memory(dtp, fp, addr) < 0)
2256					return (-1);
2257				goto nextrec;
2258			}
2259
2260			if (act == DTRACEACT_PRINTT) {
2261				if (dt_print_type(dtp, fp, addr) < 0)
2262					return (-1);
2263				goto nextrec;
2264			}
2265
2266			if (DTRACEACT_ISPRINTFLIKE(act)) {
2267				void *fmtdata;
2268				int (*func)(dtrace_hdl_t *, FILE *, void *,
2269				    const dtrace_probedata_t *,
2270				    const dtrace_recdesc_t *, uint_t,
2271				    const void *buf, size_t);
2272
2273				if ((fmtdata = dt_format_lookup(dtp,
2274				    rec->dtrd_format)) == NULL)
2275					goto nofmt;
2276
2277				switch (act) {
2278				case DTRACEACT_PRINTF:
2279					func = dtrace_fprintf;
2280					break;
2281				case DTRACEACT_PRINTA:
2282					func = dtrace_fprinta;
2283					break;
2284				case DTRACEACT_SYSTEM:
2285					func = dtrace_system;
2286					break;
2287				case DTRACEACT_FREOPEN:
2288					func = dtrace_freopen;
2289					break;
2290				}
2291
2292				n = (*func)(dtp, fp, fmtdata, &data,
2293				    rec, epd->dtepd_nrecs - i,
2294				    (uchar_t *)buf->dtbd_data + offs,
2295				    buf->dtbd_size - offs);
2296
2297				if (n < 0)
2298					return (-1); /* errno is set for us */
2299
2300				if (n > 0)
2301					i += n - 1;
2302				goto nextrec;
2303			}
2304
2305			/*
2306			 * If this is a DIF expression, and the record has a
2307			 * format set, this indicates we have a CTF type name
2308			 * associated with the data and we should try to print
2309			 * it out by type.
2310			 */
2311			if (act == DTRACEACT_DIFEXPR) {
2312				const char *strdata = dt_strdata_lookup(dtp,
2313				    rec->dtrd_format);
2314				if (strdata != NULL) {
2315					n = dtrace_print(dtp, fp, strdata,
2316					    addr, rec->dtrd_size);
2317
2318					/*
2319					 * dtrace_print() will return -1 on
2320					 * error, or return the number of bytes
2321					 * consumed.  It will return 0 if the
2322					 * type couldn't be determined, and we
2323					 * should fall through to the normal
2324					 * trace method.
2325					 */
2326					if (n < 0)
2327						return (-1);
2328
2329					if (n > 0)
2330						goto nextrec;
2331				}
2332			}
2333
2334nofmt:
2335			if (act == DTRACEACT_PRINTA) {
2336				dt_print_aggdata_t pd;
2337				dtrace_aggvarid_t *aggvars;
2338				int j, naggvars = 0;
2339				size_t size = ((epd->dtepd_nrecs - i) *
2340				    sizeof (dtrace_aggvarid_t));
2341
2342				if ((aggvars = dt_alloc(dtp, size)) == NULL)
2343					return (-1);
2344
2345				/*
2346				 * This might be a printa() with multiple
2347				 * aggregation variables.  We need to scan
2348				 * forward through the records until we find
2349				 * a record from a different statement.
2350				 */
2351				for (j = i; j < epd->dtepd_nrecs; j++) {
2352					dtrace_recdesc_t *nrec;
2353					caddr_t naddr;
2354
2355					nrec = &epd->dtepd_rec[j];
2356
2357					if (nrec->dtrd_uarg != rec->dtrd_uarg)
2358						break;
2359
2360					if (nrec->dtrd_action != act) {
2361						return (dt_set_errno(dtp,
2362						    EDT_BADAGG));
2363					}
2364
2365					naddr = buf->dtbd_data + offs +
2366					    nrec->dtrd_offset;
2367
2368					aggvars[naggvars++] =
2369					    /* LINTED - alignment */
2370					    *((dtrace_aggvarid_t *)naddr);
2371				}
2372
2373				i = j - 1;
2374				bzero(&pd, sizeof (pd));
2375				pd.dtpa_dtp = dtp;
2376				pd.dtpa_fp = fp;
2377
2378				assert(naggvars >= 1);
2379
2380				if (naggvars == 1) {
2381					pd.dtpa_id = aggvars[0];
2382					dt_free(dtp, aggvars);
2383
2384					if (dt_printf(dtp, fp, "\n") < 0 ||
2385					    dtrace_aggregate_walk_sorted(dtp,
2386					    dt_print_agg, &pd) < 0)
2387						return (-1);
2388					goto nextrec;
2389				}
2390
2391				if (dt_printf(dtp, fp, "\n") < 0 ||
2392				    dtrace_aggregate_walk_joined(dtp, aggvars,
2393				    naggvars, dt_print_aggs, &pd) < 0) {
2394					dt_free(dtp, aggvars);
2395					return (-1);
2396				}
2397
2398				dt_free(dtp, aggvars);
2399				goto nextrec;
2400			}
2401
2402			if (act == DTRACEACT_TRACEMEM) {
2403				if (tracememsize == 0 ||
2404				    tracememsize > rec->dtrd_size) {
2405					tracememsize = rec->dtrd_size;
2406				}
2407
2408				n = dt_print_bytes(dtp, fp, addr,
2409				    tracememsize, 33, quiet, 1);
2410
2411				tracememsize = 0;
2412
2413				if (n < 0)
2414					return (-1);
2415
2416				goto nextrec;
2417			}
2418
2419			switch (rec->dtrd_size) {
2420			case sizeof (uint64_t):
2421				n = dt_printf(dtp, fp,
2422				    quiet ? "%lld" : " %16lld",
2423				    /* LINTED - alignment */
2424				    *((unsigned long long *)addr));
2425				break;
2426			case sizeof (uint32_t):
2427				n = dt_printf(dtp, fp, quiet ? "%d" : " %8d",
2428				    /* LINTED - alignment */
2429				    *((uint32_t *)addr));
2430				break;
2431			case sizeof (uint16_t):
2432				n = dt_printf(dtp, fp, quiet ? "%d" : " %5d",
2433				    /* LINTED - alignment */
2434				    *((uint16_t *)addr));
2435				break;
2436			case sizeof (uint8_t):
2437				n = dt_printf(dtp, fp, quiet ? "%d" : " %3d",
2438				    *((uint8_t *)addr));
2439				break;
2440			default:
2441				n = dt_print_bytes(dtp, fp, addr,
2442				    rec->dtrd_size, 33, quiet, 0);
2443				break;
2444			}
2445
2446			if (n < 0)
2447				return (-1); /* errno is set for us */
2448
2449nextrec:
2450			if (dt_buffered_flush(dtp, &data, rec, NULL, 0) < 0)
2451				return (-1); /* errno is set for us */
2452		}
2453
2454		/*
2455		 * Call the record callback with a NULL record to indicate
2456		 * that we're done processing this EPID.
2457		 */
2458		rval = (*rfunc)(&data, NULL, arg);
2459nextepid:
2460		offs += epd->dtepd_size;
2461		last = id;
2462	}
2463
2464	if (buf->dtbd_oldest != 0 && start == buf->dtbd_oldest) {
2465		end = buf->dtbd_oldest;
2466		start = 0;
2467		goto again;
2468	}
2469
2470	if ((drops = buf->dtbd_drops) == 0)
2471		return (0);
2472
2473	/*
2474	 * Explicitly zero the drops to prevent us from processing them again.
2475	 */
2476	buf->dtbd_drops = 0;
2477
2478	return (dt_handle_cpudrop(dtp, cpu, DTRACEDROP_PRINCIPAL, drops));
2479}
2480
2481typedef struct dt_begin {
2482	dtrace_consume_probe_f *dtbgn_probefunc;
2483	dtrace_consume_rec_f *dtbgn_recfunc;
2484	void *dtbgn_arg;
2485	dtrace_handle_err_f *dtbgn_errhdlr;
2486	void *dtbgn_errarg;
2487	int dtbgn_beginonly;
2488} dt_begin_t;
2489
2490static int
2491dt_consume_begin_probe(const dtrace_probedata_t *data, void *arg)
2492{
2493	dt_begin_t *begin = (dt_begin_t *)arg;
2494	dtrace_probedesc_t *pd = data->dtpda_pdesc;
2495
2496	int r1 = (strcmp(pd->dtpd_provider, "dtrace") == 0);
2497	int r2 = (strcmp(pd->dtpd_name, "BEGIN") == 0);
2498
2499	if (begin->dtbgn_beginonly) {
2500		if (!(r1 && r2))
2501			return (DTRACE_CONSUME_NEXT);
2502	} else {
2503		if (r1 && r2)
2504			return (DTRACE_CONSUME_NEXT);
2505	}
2506
2507	/*
2508	 * We have a record that we're interested in.  Now call the underlying
2509	 * probe function...
2510	 */
2511	return (begin->dtbgn_probefunc(data, begin->dtbgn_arg));
2512}
2513
2514static int
2515dt_consume_begin_record(const dtrace_probedata_t *data,
2516    const dtrace_recdesc_t *rec, void *arg)
2517{
2518	dt_begin_t *begin = (dt_begin_t *)arg;
2519
2520	return (begin->dtbgn_recfunc(data, rec, begin->dtbgn_arg));
2521}
2522
2523static int
2524dt_consume_begin_error(const dtrace_errdata_t *data, void *arg)
2525{
2526	dt_begin_t *begin = (dt_begin_t *)arg;
2527	dtrace_probedesc_t *pd = data->dteda_pdesc;
2528
2529	int r1 = (strcmp(pd->dtpd_provider, "dtrace") == 0);
2530	int r2 = (strcmp(pd->dtpd_name, "BEGIN") == 0);
2531
2532	if (begin->dtbgn_beginonly) {
2533		if (!(r1 && r2))
2534			return (DTRACE_HANDLE_OK);
2535	} else {
2536		if (r1 && r2)
2537			return (DTRACE_HANDLE_OK);
2538	}
2539
2540	return (begin->dtbgn_errhdlr(data, begin->dtbgn_errarg));
2541}
2542
2543static int
2544dt_consume_begin(dtrace_hdl_t *dtp, FILE *fp, dtrace_bufdesc_t *buf,
2545    dtrace_consume_probe_f *pf, dtrace_consume_rec_f *rf, void *arg)
2546{
2547	/*
2548	 * There's this idea that the BEGIN probe should be processed before
2549	 * everything else, and that the END probe should be processed after
2550	 * anything else.  In the common case, this is pretty easy to deal
2551	 * with.  However, a situation may arise where the BEGIN enabling and
2552	 * END enabling are on the same CPU, and some enabling in the middle
2553	 * occurred on a different CPU.  To deal with this (blech!) we need to
2554	 * consume the BEGIN buffer up until the end of the BEGIN probe, and
2555	 * then set it aside.  We will then process every other CPU, and then
2556	 * we'll return to the BEGIN CPU and process the rest of the data
2557	 * (which will inevitably include the END probe, if any).  Making this
2558	 * even more complicated (!) is the library's ERROR enabling.  Because
2559	 * this enabling is processed before we even get into the consume call
2560	 * back, any ERROR firing would result in the library's ERROR enabling
2561	 * being processed twice -- once in our first pass (for BEGIN probes),
2562	 * and again in our second pass (for everything but BEGIN probes).  To
2563	 * deal with this, we interpose on the ERROR handler to assure that we
2564	 * only process ERROR enablings induced by BEGIN enablings in the
2565	 * first pass, and that we only process ERROR enablings _not_ induced
2566	 * by BEGIN enablings in the second pass.
2567	 */
2568	dt_begin_t begin;
2569	processorid_t cpu = dtp->dt_beganon;
2570	dtrace_bufdesc_t nbuf;
2571#if !defined(sun)
2572	dtrace_bufdesc_t *pbuf;
2573#endif
2574	int rval, i;
2575	static int max_ncpus;
2576	dtrace_optval_t size;
2577
2578	dtp->dt_beganon = -1;
2579
2580#if defined(sun)
2581	if (dt_ioctl(dtp, DTRACEIOC_BUFSNAP, buf) == -1) {
2582#else
2583	if (dt_ioctl(dtp, DTRACEIOC_BUFSNAP, &buf) == -1) {
2584#endif
2585		/*
2586		 * We really don't expect this to fail, but it is at least
2587		 * technically possible for this to fail with ENOENT.  In this
2588		 * case, we just drive on...
2589		 */
2590		if (errno == ENOENT)
2591			return (0);
2592
2593		return (dt_set_errno(dtp, errno));
2594	}
2595
2596	if (!dtp->dt_stopped || buf->dtbd_cpu != dtp->dt_endedon) {
2597		/*
2598		 * This is the simple case.  We're either not stopped, or if
2599		 * we are, we actually processed any END probes on another
2600		 * CPU.  We can simply consume this buffer and return.
2601		 */
2602		return (dt_consume_cpu(dtp, fp, cpu, buf, pf, rf, arg));
2603	}
2604
2605	begin.dtbgn_probefunc = pf;
2606	begin.dtbgn_recfunc = rf;
2607	begin.dtbgn_arg = arg;
2608	begin.dtbgn_beginonly = 1;
2609
2610	/*
2611	 * We need to interpose on the ERROR handler to be sure that we
2612	 * only process ERRORs induced by BEGIN.
2613	 */
2614	begin.dtbgn_errhdlr = dtp->dt_errhdlr;
2615	begin.dtbgn_errarg = dtp->dt_errarg;
2616	dtp->dt_errhdlr = dt_consume_begin_error;
2617	dtp->dt_errarg = &begin;
2618
2619	rval = dt_consume_cpu(dtp, fp, cpu, buf, dt_consume_begin_probe,
2620	    dt_consume_begin_record, &begin);
2621
2622	dtp->dt_errhdlr = begin.dtbgn_errhdlr;
2623	dtp->dt_errarg = begin.dtbgn_errarg;
2624
2625	if (rval != 0)
2626		return (rval);
2627
2628	/*
2629	 * Now allocate a new buffer.  We'll use this to deal with every other
2630	 * CPU.
2631	 */
2632	bzero(&nbuf, sizeof (dtrace_bufdesc_t));
2633	(void) dtrace_getopt(dtp, "bufsize", &size);
2634	if ((nbuf.dtbd_data = malloc(size)) == NULL)
2635		return (dt_set_errno(dtp, EDT_NOMEM));
2636
2637	if (max_ncpus == 0)
2638		max_ncpus = dt_sysconf(dtp, _SC_CPUID_MAX) + 1;
2639
2640	for (i = 0; i < max_ncpus; i++) {
2641		nbuf.dtbd_cpu = i;
2642
2643		if (i == cpu)
2644			continue;
2645
2646#if defined(sun)
2647		if (dt_ioctl(dtp, DTRACEIOC_BUFSNAP, &nbuf) == -1) {
2648#else
2649		pbuf = &nbuf;
2650		if (dt_ioctl(dtp, DTRACEIOC_BUFSNAP, &pbuf) == -1) {
2651#endif
2652			/*
2653			 * If we failed with ENOENT, it may be because the
2654			 * CPU was unconfigured -- this is okay.  Any other
2655			 * error, however, is unexpected.
2656			 */
2657			if (errno == ENOENT)
2658				continue;
2659
2660			free(nbuf.dtbd_data);
2661
2662			return (dt_set_errno(dtp, errno));
2663		}
2664
2665		if ((rval = dt_consume_cpu(dtp, fp,
2666		    i, &nbuf, pf, rf, arg)) != 0) {
2667			free(nbuf.dtbd_data);
2668			return (rval);
2669		}
2670	}
2671
2672	free(nbuf.dtbd_data);
2673
2674	/*
2675	 * Okay -- we're done with the other buffers.  Now we want to
2676	 * reconsume the first buffer -- but this time we're looking for
2677	 * everything _but_ BEGIN.  And of course, in order to only consume
2678	 * those ERRORs _not_ associated with BEGIN, we need to reinstall our
2679	 * ERROR interposition function...
2680	 */
2681	begin.dtbgn_beginonly = 0;
2682
2683	assert(begin.dtbgn_errhdlr == dtp->dt_errhdlr);
2684	assert(begin.dtbgn_errarg == dtp->dt_errarg);
2685	dtp->dt_errhdlr = dt_consume_begin_error;
2686	dtp->dt_errarg = &begin;
2687
2688	rval = dt_consume_cpu(dtp, fp, cpu, buf, dt_consume_begin_probe,
2689	    dt_consume_begin_record, &begin);
2690
2691	dtp->dt_errhdlr = begin.dtbgn_errhdlr;
2692	dtp->dt_errarg = begin.dtbgn_errarg;
2693
2694	return (rval);
2695}
2696
2697int
2698dtrace_consume(dtrace_hdl_t *dtp, FILE *fp,
2699    dtrace_consume_probe_f *pf, dtrace_consume_rec_f *rf, void *arg)
2700{
2701	dtrace_bufdesc_t *buf = &dtp->dt_buf;
2702	dtrace_optval_t size;
2703	static int max_ncpus;
2704	int i, rval;
2705	dtrace_optval_t interval = dtp->dt_options[DTRACEOPT_SWITCHRATE];
2706	hrtime_t now = gethrtime();
2707
2708	if (dtp->dt_lastswitch != 0) {
2709		if (now - dtp->dt_lastswitch < interval)
2710			return (0);
2711
2712		dtp->dt_lastswitch += interval;
2713	} else {
2714		dtp->dt_lastswitch = now;
2715	}
2716
2717	if (!dtp->dt_active)
2718		return (dt_set_errno(dtp, EINVAL));
2719
2720	if (max_ncpus == 0)
2721		max_ncpus = dt_sysconf(dtp, _SC_CPUID_MAX) + 1;
2722
2723	if (pf == NULL)
2724		pf = (dtrace_consume_probe_f *)dt_nullprobe;
2725
2726	if (rf == NULL)
2727		rf = (dtrace_consume_rec_f *)dt_nullrec;
2728
2729	if (buf->dtbd_data == NULL) {
2730		(void) dtrace_getopt(dtp, "bufsize", &size);
2731		if ((buf->dtbd_data = malloc(size)) == NULL)
2732			return (dt_set_errno(dtp, EDT_NOMEM));
2733
2734		buf->dtbd_size = size;
2735	}
2736
2737	/*
2738	 * If we have just begun, we want to first process the CPU that
2739	 * executed the BEGIN probe (if any).
2740	 */
2741	if (dtp->dt_active && dtp->dt_beganon != -1) {
2742		buf->dtbd_cpu = dtp->dt_beganon;
2743		if ((rval = dt_consume_begin(dtp, fp, buf, pf, rf, arg)) != 0)
2744			return (rval);
2745	}
2746
2747	for (i = 0; i < max_ncpus; i++) {
2748		buf->dtbd_cpu = i;
2749
2750		/*
2751		 * If we have stopped, we want to process the CPU on which the
2752		 * END probe was processed only _after_ we have processed
2753		 * everything else.
2754		 */
2755		if (dtp->dt_stopped && (i == dtp->dt_endedon))
2756			continue;
2757
2758#if defined(sun)
2759		if (dt_ioctl(dtp, DTRACEIOC_BUFSNAP, buf) == -1) {
2760#else
2761		if (dt_ioctl(dtp, DTRACEIOC_BUFSNAP, &buf) == -1) {
2762#endif
2763			/*
2764			 * If we failed with ENOENT, it may be because the
2765			 * CPU was unconfigured -- this is okay.  Any other
2766			 * error, however, is unexpected.
2767			 */
2768			if (errno == ENOENT)
2769				continue;
2770
2771			return (dt_set_errno(dtp, errno));
2772		}
2773
2774		if ((rval = dt_consume_cpu(dtp, fp, i, buf, pf, rf, arg)) != 0)
2775			return (rval);
2776	}
2777
2778	if (!dtp->dt_stopped)
2779		return (0);
2780
2781	buf->dtbd_cpu = dtp->dt_endedon;
2782
2783#if defined(sun)
2784	if (dt_ioctl(dtp, DTRACEIOC_BUFSNAP, buf) == -1) {
2785#else
2786	if (dt_ioctl(dtp, DTRACEIOC_BUFSNAP, &buf) == -1) {
2787#endif
2788		/*
2789		 * This _really_ shouldn't fail, but it is strictly speaking
2790		 * possible for this to return ENOENT if the CPU that called
2791		 * the END enabling somehow managed to become unconfigured.
2792		 * It's unclear how the user can possibly expect anything
2793		 * rational to happen in this case -- the state has been thrown
2794		 * out along with the unconfigured CPU -- so we'll just drive
2795		 * on...
2796		 */
2797		if (errno == ENOENT)
2798			return (0);
2799
2800		return (dt_set_errno(dtp, errno));
2801	}
2802
2803	return (dt_consume_cpu(dtp, fp, dtp->dt_endedon, buf, pf, rf, arg));
2804}
2805