cmd/chkargs/chkargs.c

	chkargs.c revision 9512fe850e98fdd448c638ca63fdd92a8a510255
/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#pragma ident   "%Z%%M% %I% %E% SMI"

#include <strings.h>
#include <unistd.h>
#include <dtrace.h>

static int g_count;
static int g_errs;
static int g_fd;
static int g_verbose;
static int g_errexit;

static int
probe(dtrace_hdl_t *dtp, const dtrace_probedesc_t *pdp, void *data)
{
    dtrace_probeinfo_t p;
    dtrace_argdesc_t arg;
    char buf[BUFSIZ];
    int i;

    (void) printf("\r%6d", ++g_count);
    (void) fflush(stdout);

    if (dtrace_probe_info(dtp, pdp, &p) != 0) {
        (void) printf(" failed to get probe info for "
            "%s:%s:%s:%s [%d]\n", pdp->dtpd_provider, pdp->dtpd_mod,
            pdp->dtpd_func, pdp->dtpd_name, pdp->dtpd_id);
        g_errs++;
        return (0);
    }

    for (i = 0; i < p.dtp_argc; i++) {
        if (p.dtp_argv[i].dtt_type == CTF_ERR) {
            bzero(&arg, sizeof (dtrace_argdesc_t));
            arg.dtargd_id = pdp->dtpd_id;
            arg.dtargd_ndx = i;
            (void) ioctl(g_fd, DTRACEIOC_PROBEARG, &arg);

            (void) printf(" failed to get types for args[%d] "
                "of %s:%s:%s:%s [%d]: <%s> -> <%s>\n", i,
                pdp->dtpd_provider, pdp->dtpd_mod, pdp->dtpd_func,
                pdp->dtpd_name, pdp->dtpd_id,
                arg.dtargd_native, arg.dtargd_xlate);

            g_errs++;

            if (g_errexit)
                return (-1);

        } else if (g_verbose) {
            (void) printf("%d args[%d] : %s\n", pdp->dtpd_id, i,
                ctf_type_name(p.dtp_argv[i].dtt_ctfp,
                p.dtp_argv[i].dtt_type, buf, sizeof (buf)));
        }
    }

    return (0);
}

int
main(int argc, char *argv[])
{
    dtrace_probedesc_t pd, *pdp = NULL;
    dtrace_hdl_t *dtp;
    int err, c;
    char *p;

    if ((dtp = dtrace_open(DTRACE_VERSION, 0, &err)) == NULL) {
        (void) fprintf(stderr, "%s: failed to open dtrace: %s\n",
            argv[0], dtrace_errmsg(dtp, err));
        return (1);
    }

    while ((c = getopt(argc, argv, "evx:")) != -1) {
        switch (c) {
        case 'e':
            g_errexit++;
            break;
        case 'v':
            g_verbose++;
            break;
        case 'x':
            if ((p = strchr(optarg, '=')) != NULL)
                *p++ = '\0';

            if (dtrace_setopt(dtp, optarg, p) != 0) {
                (void) fprintf(stderr, "%s: failed to set "
                    "option -x %s: %s\n", argv[0], optarg,
                    dtrace_errmsg(dtp, dtrace_errno(dtp)));
                return (2);
            }
            break;

        default:
            (void) fprintf(stderr, "Usage: %s [-ev] "
                "[-x opt[=arg]] [probedesc]\n", argv[0]);
            return (2);
        }
    }

    argv += optind;
    argc -= optind;

    if (argc > 0) {
        if (dtrace_str2desc(dtp, DTRACE_PROBESPEC_NAME, argv[1], &pd)) {
            (void) fprintf(stderr, "%s: invalid probe description "
                "%s: %s\n", argv[0], argv[1],
                dtrace_errmsg(dtp, dtrace_errno(dtp)));
            return (2);
        }
        pdp = &pd;
    }

    g_fd = dtrace_ctlfd(dtp);
    (void) dtrace_probe_iter(dtp, pdp, probe, NULL);
    dtrace_close(dtp);

    (void) printf("\nTotal probes: %d\n", g_count);
    (void) printf("Total errors: %d\n\n", g_errs);

    return (g_errs != 0);
}