9512fe850e98fdd448c638ca63fdd92a8a510255ahl/*
9512fe850e98fdd448c638ca63fdd92a8a510255ahl * CDDL HEADER START
9512fe850e98fdd448c638ca63fdd92a8a510255ahl *
9512fe850e98fdd448c638ca63fdd92a8a510255ahl * The contents of this file are subject to the terms of the
9512fe850e98fdd448c638ca63fdd92a8a510255ahl * Common Development and Distribution License (the "License").
9512fe850e98fdd448c638ca63fdd92a8a510255ahl * You may not use this file except in compliance with the License.
9512fe850e98fdd448c638ca63fdd92a8a510255ahl *
9512fe850e98fdd448c638ca63fdd92a8a510255ahl * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9512fe850e98fdd448c638ca63fdd92a8a510255ahl * or http://www.opensolaris.org/os/licensing.
9512fe850e98fdd448c638ca63fdd92a8a510255ahl * See the License for the specific language governing permissions
9512fe850e98fdd448c638ca63fdd92a8a510255ahl * and limitations under the License.
9512fe850e98fdd448c638ca63fdd92a8a510255ahl *
9512fe850e98fdd448c638ca63fdd92a8a510255ahl * When distributing Covered Code, include this CDDL HEADER in each
9512fe850e98fdd448c638ca63fdd92a8a510255ahl * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
9512fe850e98fdd448c638ca63fdd92a8a510255ahl * If applicable, add the following below this CDDL HEADER, with the
9512fe850e98fdd448c638ca63fdd92a8a510255ahl * fields enclosed by brackets "[]" replaced with your own identifying
9512fe850e98fdd448c638ca63fdd92a8a510255ahl * information: Portions Copyright [yyyy] [name of copyright owner]
9512fe850e98fdd448c638ca63fdd92a8a510255ahl *
9512fe850e98fdd448c638ca63fdd92a8a510255ahl * CDDL HEADER END
9512fe850e98fdd448c638ca63fdd92a8a510255ahl */
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahl/*
8cb74972a66bde0af7b1a957d01e0095b82a8b91Jonathan Haslam * Copyright 2009 Sun Microsystems, Inc. All rights reserved.
9512fe850e98fdd448c638ca63fdd92a8a510255ahl * Use is subject to license terms.
9512fe850e98fdd448c638ca63fdd92a8a510255ahl */
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahl#include <strings.h>
9512fe850e98fdd448c638ca63fdd92a8a510255ahl#include <unistd.h>
9512fe850e98fdd448c638ca63fdd92a8a510255ahl#include <dtrace.h>
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahlstatic int g_count;
9512fe850e98fdd448c638ca63fdd92a8a510255ahlstatic int g_errs;
9512fe850e98fdd448c638ca63fdd92a8a510255ahlstatic int g_fd;
9512fe850e98fdd448c638ca63fdd92a8a510255ahlstatic int g_verbose;
9512fe850e98fdd448c638ca63fdd92a8a510255ahlstatic int g_errexit;
8cb74972a66bde0af7b1a957d01e0095b82a8b91Jonathan Haslamstatic char *g_progname;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahlstatic int
9512fe850e98fdd448c638ca63fdd92a8a510255ahlprobe(dtrace_hdl_t *dtp, const dtrace_probedesc_t *pdp, void *data)
9512fe850e98fdd448c638ca63fdd92a8a510255ahl{
9512fe850e98fdd448c638ca63fdd92a8a510255ahl dtrace_probeinfo_t p;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl dtrace_argdesc_t arg;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl char buf[BUFSIZ];
9512fe850e98fdd448c638ca63fdd92a8a510255ahl int i;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahl (void) printf("\r%6d", ++g_count);
9512fe850e98fdd448c638ca63fdd92a8a510255ahl (void) fflush(stdout);
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahl if (dtrace_probe_info(dtp, pdp, &p) != 0) {
9512fe850e98fdd448c638ca63fdd92a8a510255ahl (void) printf(" failed to get probe info for "
9512fe850e98fdd448c638ca63fdd92a8a510255ahl "%s:%s:%s:%s [%d]\n", pdp->dtpd_provider, pdp->dtpd_mod,
9512fe850e98fdd448c638ca63fdd92a8a510255ahl pdp->dtpd_func, pdp->dtpd_name, pdp->dtpd_id);
9512fe850e98fdd448c638ca63fdd92a8a510255ahl g_errs++;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl return (0);
9512fe850e98fdd448c638ca63fdd92a8a510255ahl }
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahl for (i = 0; i < p.dtp_argc; i++) {
9512fe850e98fdd448c638ca63fdd92a8a510255ahl if (p.dtp_argv[i].dtt_type == CTF_ERR) {
9512fe850e98fdd448c638ca63fdd92a8a510255ahl bzero(&arg, sizeof (dtrace_argdesc_t));
9512fe850e98fdd448c638ca63fdd92a8a510255ahl arg.dtargd_id = pdp->dtpd_id;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl arg.dtargd_ndx = i;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl (void) ioctl(g_fd, DTRACEIOC_PROBEARG, &arg);
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahl (void) printf(" failed to get types for args[%d] "
9512fe850e98fdd448c638ca63fdd92a8a510255ahl "of %s:%s:%s:%s [%d]: <%s> -> <%s>\n", i,
9512fe850e98fdd448c638ca63fdd92a8a510255ahl pdp->dtpd_provider, pdp->dtpd_mod, pdp->dtpd_func,
9512fe850e98fdd448c638ca63fdd92a8a510255ahl pdp->dtpd_name, pdp->dtpd_id,
9512fe850e98fdd448c638ca63fdd92a8a510255ahl arg.dtargd_native, arg.dtargd_xlate);
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahl g_errs++;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahl if (g_errexit)
9512fe850e98fdd448c638ca63fdd92a8a510255ahl return (-1);
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahl } else if (g_verbose) {
9512fe850e98fdd448c638ca63fdd92a8a510255ahl (void) printf("%d args[%d] : %s\n", pdp->dtpd_id, i,
9512fe850e98fdd448c638ca63fdd92a8a510255ahl ctf_type_name(p.dtp_argv[i].dtt_ctfp,
9512fe850e98fdd448c638ca63fdd92a8a510255ahl p.dtp_argv[i].dtt_type, buf, sizeof (buf)));
9512fe850e98fdd448c638ca63fdd92a8a510255ahl }
9512fe850e98fdd448c638ca63fdd92a8a510255ahl }
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahl return (0);
9512fe850e98fdd448c638ca63fdd92a8a510255ahl}
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahlint
9512fe850e98fdd448c638ca63fdd92a8a510255ahlmain(int argc, char *argv[])
9512fe850e98fdd448c638ca63fdd92a8a510255ahl{
9512fe850e98fdd448c638ca63fdd92a8a510255ahl dtrace_probedesc_t pd, *pdp = NULL;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl dtrace_hdl_t *dtp;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl int err, c;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl char *p;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
8cb74972a66bde0af7b1a957d01e0095b82a8b91Jonathan Haslam g_progname = argv[0];
8cb74972a66bde0af7b1a957d01e0095b82a8b91Jonathan Haslam
9512fe850e98fdd448c638ca63fdd92a8a510255ahl if ((dtp = dtrace_open(DTRACE_VERSION, 0, &err)) == NULL) {
9512fe850e98fdd448c638ca63fdd92a8a510255ahl (void) fprintf(stderr, "%s: failed to open dtrace: %s\n",
8cb74972a66bde0af7b1a957d01e0095b82a8b91Jonathan Haslam g_progname, dtrace_errmsg(dtp, err));
9512fe850e98fdd448c638ca63fdd92a8a510255ahl return (1);
9512fe850e98fdd448c638ca63fdd92a8a510255ahl }
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahl while ((c = getopt(argc, argv, "evx:")) != -1) {
9512fe850e98fdd448c638ca63fdd92a8a510255ahl switch (c) {
9512fe850e98fdd448c638ca63fdd92a8a510255ahl case 'e':
9512fe850e98fdd448c638ca63fdd92a8a510255ahl g_errexit++;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl break;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl case 'v':
9512fe850e98fdd448c638ca63fdd92a8a510255ahl g_verbose++;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl break;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl case 'x':
9512fe850e98fdd448c638ca63fdd92a8a510255ahl if ((p = strchr(optarg, '=')) != NULL)
9512fe850e98fdd448c638ca63fdd92a8a510255ahl *p++ = '\0';
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahl if (dtrace_setopt(dtp, optarg, p) != 0) {
9512fe850e98fdd448c638ca63fdd92a8a510255ahl (void) fprintf(stderr, "%s: failed to set "
8cb74972a66bde0af7b1a957d01e0095b82a8b91Jonathan Haslam "option -x %s: %s\n", g_progname, optarg,
9512fe850e98fdd448c638ca63fdd92a8a510255ahl dtrace_errmsg(dtp, dtrace_errno(dtp)));
9512fe850e98fdd448c638ca63fdd92a8a510255ahl return (2);
9512fe850e98fdd448c638ca63fdd92a8a510255ahl }
9512fe850e98fdd448c638ca63fdd92a8a510255ahl break;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahl default:
9512fe850e98fdd448c638ca63fdd92a8a510255ahl (void) fprintf(stderr, "Usage: %s [-ev] "
8cb74972a66bde0af7b1a957d01e0095b82a8b91Jonathan Haslam "[-x opt[=arg]] [probedesc]\n", g_progname);
9512fe850e98fdd448c638ca63fdd92a8a510255ahl return (2);
9512fe850e98fdd448c638ca63fdd92a8a510255ahl }
9512fe850e98fdd448c638ca63fdd92a8a510255ahl }
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahl argv += optind;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl argc -= optind;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahl if (argc > 0) {
8cb74972a66bde0af7b1a957d01e0095b82a8b91Jonathan Haslam if (dtrace_str2desc(dtp, DTRACE_PROBESPEC_NAME, argv[0], &pd)) {
9512fe850e98fdd448c638ca63fdd92a8a510255ahl (void) fprintf(stderr, "%s: invalid probe description "
8cb74972a66bde0af7b1a957d01e0095b82a8b91Jonathan Haslam "%s: %s\n", g_progname, argv[0],
9512fe850e98fdd448c638ca63fdd92a8a510255ahl dtrace_errmsg(dtp, dtrace_errno(dtp)));
9512fe850e98fdd448c638ca63fdd92a8a510255ahl return (2);
9512fe850e98fdd448c638ca63fdd92a8a510255ahl }
9512fe850e98fdd448c638ca63fdd92a8a510255ahl pdp = &pd;
9512fe850e98fdd448c638ca63fdd92a8a510255ahl }
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahl g_fd = dtrace_ctlfd(dtp);
9512fe850e98fdd448c638ca63fdd92a8a510255ahl (void) dtrace_probe_iter(dtp, pdp, probe, NULL);
9512fe850e98fdd448c638ca63fdd92a8a510255ahl dtrace_close(dtp);
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahl (void) printf("\nTotal probes: %d\n", g_count);
9512fe850e98fdd448c638ca63fdd92a8a510255ahl (void) printf("Total errors: %d\n\n", g_errs);
9512fe850e98fdd448c638ca63fdd92a8a510255ahl
9512fe850e98fdd448c638ca63fdd92a8a510255ahl return (g_errs != 0);
9512fe850e98fdd448c638ca63fdd92a8a510255ahl}