1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://www.opensolaris.org/os/licensing.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 
22 /*
23  * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
24  * Use is subject to license terms.
25  */
26 
27 #include <strings.h>
28 #include <unistd.h>
29 #include <dtrace.h>
30 
31 static int g_count;
32 static int g_errs;
33 static int g_fd;
34 static int g_verbose;
35 static int g_errexit;
36 static char *g_progname;
37 
38 static int
probe(dtrace_hdl_t * dtp,const dtrace_probedesc_t * pdp,void * data)39 probe(dtrace_hdl_t *dtp, const dtrace_probedesc_t *pdp, void *data)
40 {
41 	dtrace_probeinfo_t p;
42 	dtrace_argdesc_t arg;
43 	char buf[BUFSIZ];
44 	int i;
45 
46 	(void) printf("\r%6d", ++g_count);
47 	(void) fflush(stdout);
48 
49 	if (dtrace_probe_info(dtp, pdp, &p) != 0) {
50 		(void) printf(" failed to get probe info for "
51 		    "%s:%s:%s:%s [%d]\n", pdp->dtpd_provider, pdp->dtpd_mod,
52 		    pdp->dtpd_func, pdp->dtpd_name, pdp->dtpd_id);
53 		g_errs++;
54 		return (0);
55 	}
56 
57 	for (i = 0; i < p.dtp_argc; i++) {
58 		if (p.dtp_argv[i].dtt_type == CTF_ERR) {
59 			bzero(&arg, sizeof (dtrace_argdesc_t));
60 			arg.dtargd_id = pdp->dtpd_id;
61 			arg.dtargd_ndx = i;
62 			(void) ioctl(g_fd, DTRACEIOC_PROBEARG, &arg);
63 
64 			(void) printf(" failed to get types for args[%d] "
65 			    "of %s:%s:%s:%s [%d]: <%s> -> <%s>\n", i,
66 			    pdp->dtpd_provider, pdp->dtpd_mod, pdp->dtpd_func,
67 			    pdp->dtpd_name, pdp->dtpd_id,
68 			    arg.dtargd_native, arg.dtargd_xlate);
69 
70 			g_errs++;
71 
72 			if (g_errexit)
73 				return (-1);
74 
75 		} else if (g_verbose) {
76 			(void) printf("%d args[%d] : %s\n", pdp->dtpd_id, i,
77 			    ctf_type_name(p.dtp_argv[i].dtt_ctfp,
78 			    p.dtp_argv[i].dtt_type, buf, sizeof (buf)));
79 		}
80 	}
81 
82 	return (0);
83 }
84 
85 int
main(int argc,char * argv[])86 main(int argc, char *argv[])
87 {
88 	dtrace_probedesc_t pd, *pdp = NULL;
89 	dtrace_hdl_t *dtp;
90 	int err, c;
91 	char *p;
92 
93 	g_progname = argv[0];
94 
95 	if ((dtp = dtrace_open(DTRACE_VERSION, 0, &err)) == NULL) {
96 		(void) fprintf(stderr, "%s: failed to open dtrace: %s\n",
97 		    g_progname, dtrace_errmsg(dtp, err));
98 		return (1);
99 	}
100 
101 	while ((c = getopt(argc, argv, "evx:")) != -1) {
102 		switch (c) {
103 		case 'e':
104 			g_errexit++;
105 			break;
106 		case 'v':
107 			g_verbose++;
108 			break;
109 		case 'x':
110 			if ((p = strchr(optarg, '=')) != NULL)
111 				*p++ = '\0';
112 
113 			if (dtrace_setopt(dtp, optarg, p) != 0) {
114 				(void) fprintf(stderr, "%s: failed to set "
115 				    "option -x %s: %s\n", g_progname, optarg,
116 				    dtrace_errmsg(dtp, dtrace_errno(dtp)));
117 				return (2);
118 			}
119 			break;
120 
121 		default:
122 			(void) fprintf(stderr, "Usage: %s [-ev] "
123 			    "[-x opt[=arg]] [probedesc]\n", g_progname);
124 			return (2);
125 		}
126 	}
127 
128 	argv += optind;
129 	argc -= optind;
130 
131 	if (argc > 0) {
132 		if (dtrace_str2desc(dtp, DTRACE_PROBESPEC_NAME, argv[0], &pd)) {
133 			(void) fprintf(stderr, "%s: invalid probe description "
134 			    "%s: %s\n", g_progname, argv[0],
135 			    dtrace_errmsg(dtp, dtrace_errno(dtp)));
136 			return (2);
137 		}
138 		pdp = &pd;
139 	}
140 
141 	g_fd = dtrace_ctlfd(dtp);
142 	(void) dtrace_probe_iter(dtp, pdp, probe, NULL);
143 	dtrace_close(dtp);
144 
145 	(void) printf("\nTotal probes: %d\n", g_count);
146 	(void) printf("Total errors: %d\n\n", g_errs);
147 
148 	return (g_errs != 0);
149 }
150