14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * CDDL HEADER START
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync *
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * The contents of this file are subject to the terms of the
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * Common Development and Distribution License (the "License").
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * You may not use this file except in compliance with the License.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync *
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * or http://www.opensolaris.org/os/licensing.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * See the License for the specific language governing permissions
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * and limitations under the License.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync *
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * When distributing Covered Code, include this CDDL HEADER in each
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * If applicable, add the following below this CDDL HEADER, with the
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * fields enclosed by brackets "[]" replaced with your own identifying
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * information: Portions Copyright [yyyy] [name of copyright owner]
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync *
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * CDDL HEADER END
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * Copyright 2006 Sun Microsystems, Inc. All rights reserved.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * Use is subject to license terms.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifndef VBOX
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#pragma ident "%Z%%M% %I% %E% SMI"
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#include <sys/types.h>
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#include <sys/stat.h>
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifndef _MSC_VER
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync# include <sys/wait.h>
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#include <dtrace.h>
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#include <stdlib.h>
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#include <stdarg.h>
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#include <stdio.h>
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifndef VBOX
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync# include <strings.h>
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifndef _MSC_VER
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync# include <unistd.h>
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#else
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync# include <direct.h>
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync# include <io.h>
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#include <limits.h>
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#include <fcntl.h>
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#include <errno.h>
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#include <signal.h>
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifndef VBOX
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#include <alloca.h>
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#include <libgen.h>
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#include <libproc.h>
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifdef VBOX
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync# include <stdio.h>
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync# include <iprt/alloca.h>
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync# include <iprt/getopt.h>
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync# include <iprt/initterm.h>
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync# include <iprt/path.h>
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync# include <iprt/message.h>
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync# include <iprt/process.h>
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync# include <iprt/string.h>
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync
77b490c77af277218928facc8eca6ad1a58de9e9vboxsync# include "VBoxDTraceLibCWrappers.h"
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync# ifdef _MSC_VER
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync# pragma warning(disable:4267) /* size_t conversion warnings */
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync# pragma warning(disable:4018) /* signed/unsigned mismatch */
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync# endif
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsynctypedef struct dtrace_cmd {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync void (*dc_func)(struct dtrace_cmd *); /* function to compile arg */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_probespec_t dc_spec; /* probe specifier context */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char *dc_arg; /* argument from main argv */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync const char *dc_name; /* name for error messages */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync const char *dc_desc; /* desc for error messages */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_prog_t *dc_prog; /* program compiled from arg */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char dc_ofile[PATH_MAX]; /* derived output file name */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync} dtrace_cmd_t;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#define DMODE_VERS 0 /* display version information and exit (-V) */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#define DMODE_EXEC 1 /* compile program for enabling (-a/e/E) */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#define DMODE_ANON 2 /* compile program for anonymous tracing (-A) */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#define DMODE_LINK 3 /* compile program for linking with ELF (-G) */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#define DMODE_LIST 4 /* compile program and list probes (-l) */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#define DMODE_HEADER 5 /* compile program for headergen (-h) */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#define E_SUCCESS 0
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#define E_ERROR 1
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#define E_USAGE 2
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifndef VBOX
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic const char DTRACE_OPTSTR[] =
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "3:6:aAb:Bc:CD:ef:FGhHi:I:lL:m:n:o:p:P:qs:SU:vVwx:X:Z";
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#else
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsyncstatic const RTGETOPTDEF g_aOptions[] =
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync{
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { "-32", 10064, RTGETOPT_REQ_NOTHING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { "-64", 10032, RTGETOPT_REQ_NOTHING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'a', RTGETOPT_REQ_NOTHING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'A', RTGETOPT_REQ_NOTHING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'b', RTGETOPT_REQ_STRING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'B', RTGETOPT_REQ_NOTHING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'c', RTGETOPT_REQ_STRING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'C', RTGETOPT_REQ_NOTHING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'D', RTGETOPT_REQ_STRING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'e', RTGETOPT_REQ_NOTHING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'f', RTGETOPT_REQ_STRING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'F', RTGETOPT_REQ_NOTHING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'G', RTGETOPT_REQ_NOTHING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'h', RTGETOPT_REQ_NOTHING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'H', RTGETOPT_REQ_NOTHING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'i', RTGETOPT_REQ_STRING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'I', RTGETOPT_REQ_STRING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'l', RTGETOPT_REQ_NOTHING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'L', RTGETOPT_REQ_STRING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'M', RTGETOPT_REQ_STRING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'n', RTGETOPT_REQ_STRING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'o', RTGETOPT_REQ_STRING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'p', RTGETOPT_REQ_STRING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'P', RTGETOPT_REQ_STRING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'q', RTGETOPT_REQ_NOTHING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 's', RTGETOPT_REQ_STRING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'S', RTGETOPT_REQ_NOTHING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'U', RTGETOPT_REQ_STRING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'v', RTGETOPT_REQ_NOTHING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'V', RTGETOPT_REQ_NOTHING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'w', RTGETOPT_REQ_NOTHING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'x', RTGETOPT_REQ_STRING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'X', RTGETOPT_REQ_STRING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync { NULL, 'Z', RTGETOPT_REQ_NOTHING },
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync};
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif /* VBOX */
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic char **g_argv;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int g_argc;
e3c0207b236896a903ef328bfa74ee97e61bd38bvboxsync#ifndef VBOX /* No linking. */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic char **g_objv;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int g_objc;
e3c0207b236896a903ef328bfa74ee97e61bd38bvboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic dtrace_cmd_t *g_cmdv;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int g_cmdc;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic struct ps_prochandle **g_psv;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int g_psc;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int g_pslive;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic char *g_pname;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int g_quiet;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int g_flowindent;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int g_intr;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int g_impatient;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int g_newline;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int g_total;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int g_cflags;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int g_oflags;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int g_verbose;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int g_exec = 1;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int g_mode = DMODE_EXEC;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int g_status = E_SUCCESS;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int g_grabanon = 0;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic const char *g_ofile = NULL;
e3c0207b236896a903ef328bfa74ee97e61bd38bvboxsync#ifndef VBOX /* stdout isn't a necessarily constant usable like this in C code. */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic FILE *g_ofp = stdout;
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#else
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsyncstatic FILE *g_ofp = NULL;
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic dtrace_hdl_t *g_dtp;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic char *g_etcfile = "/etc/system";
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic const char *g_etcbegin = "* vvvv Added by DTrace";
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic const char *g_etcend = "* ^^^^ Added by DTrace";
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic const char *g_etc[] = {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync"*",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync"* The following forceload directives were added by dtrace(1M) to allow for",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync"* tracing during boot. If these directives are removed, the system will",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync"* continue to function, but tracing will not occur during boot as desired.",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync"* To remove these directives (and this block comment) automatically, run",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync"* \"dtrace -A\" without additional arguments. See the \"Anonymous Tracing\"",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync"* chapter of the Solaris Dynamic Tracing Guide for details.",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync"*",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncNULL };
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncusage(FILE *fp)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync static const char predact[] = "[[ predicate ] action ]";
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) fprintf(fp, "Usage: %s [-32|-64] [-aACeFGhHlqSvVwZ] "
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "[-b bufsz] [-c cmd] [-D name[=def]]\n\t[-I path] [-L path] "
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "[-o output] [-p pid] [-s script] [-U name]\n\t"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "[-x opt[=val]] [-X a|c|s|t]\n\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t[-P provider %s]\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t[-m [ provider: ] module %s]\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t[-f [[ provider: ] module: ] func %s]\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t[-n [[[ provider: ] module: ] func: ] name %s]\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t[-i probe-id %s] [ args ... ]\n\n", g_pname,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync predact, predact, predact, predact, predact);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) fprintf(fp, "\tpredicate -> '/' D-expression '/'\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) fprintf(fp, "\t action -> '{' D-statements '}'\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) fprintf(fp, "\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-32 generate 32-bit D programs and ELF files\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-64 generate 64-bit D programs and ELF files\n\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-a claim anonymous tracing state\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-A generate driver.conf(4) directives for anonymous tracing\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-b set trace buffer size\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-c run specified command and exit upon its completion\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-C run cpp(1) preprocessor on script files\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-D define symbol when invoking preprocessor\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-e exit after compiling request but prior to enabling probes\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-f enable or list probes matching the specified function name\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-F coalesce trace output by function\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-G generate an ELF file containing embedded dtrace program\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-h generate a header file with definitions for static probes\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-H print included files when invoking preprocessor\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-i enable or list probes matching the specified probe id\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-I add include directory to preprocessor search path\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-l list probes matching specified criteria\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-L add library directory to library search path\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-m enable or list probes matching the specified module name\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-n enable or list probes matching the specified probe name\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-o set output file\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-p grab specified process-ID and cache its symbol tables\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-P enable or list probes matching the specified provider name\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-q set quiet mode (only output explicitly traced data)\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-s enable or list probes according to the specified D script\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-S print D compiler intermediate code\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-U undefine symbol when invoking preprocessor\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-v set verbose mode (report stability attributes, arguments)\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-V report DTrace API version\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-w permit destructive actions\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-x enable or modify compiler and tracing options\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-X specify ISO C conformance settings for preprocessor\n"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "\t-Z permit probe descriptions that match zero probes\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (E_USAGE);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic void
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncverror(const char *fmt, va_list ap)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync int error = errno;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) fprintf(stderr, "%s: ", g_pname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) vfprintf(stderr, fmt, ap);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (fmt[strlen(fmt) - 1] != '\n')
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) fprintf(stderr, ": %s\n", strerror(error));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*PRINTFLIKE1*/
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic void
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncfatal(const char *fmt, ...)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync va_list ap;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync va_start(ap, fmt);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync verror(fmt, ap);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync va_end(ap);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync exit(E_ERROR);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*PRINTFLIKE1*/
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic void
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncdfatal(const char *fmt, ...)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync va_list ap;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync va_start(ap, fmt);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) fprintf(stderr, "%s: ", g_pname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (fmt != NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) vfprintf(stderr, fmt, ap);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync va_end(ap);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (fmt != NULL && fmt[strlen(fmt) - 1] != '\n') {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) fprintf(stderr, ": %s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_errmsg(g_dtp, dtrace_errno(g_dtp)));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } else if (fmt == NULL) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) fprintf(stderr, "%s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_errmsg(g_dtp, dtrace_errno(g_dtp)));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * Close the DTrace handle to ensure that any controlled processes are
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * correctly restored and continued.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_close(g_dtp);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync exit(E_ERROR);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*PRINTFLIKE1*/
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic void
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncerror(const char *fmt, ...)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync va_list ap;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync va_start(ap, fmt);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync verror(fmt, ap);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync va_end(ap);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*PRINTFLIKE1*/
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic void
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncnotice(const char *fmt, ...)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync va_list ap;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_quiet)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return; /* -q or quiet pragma suppresses notice()s */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync va_start(ap, fmt);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync verror(fmt, ap);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync va_end(ap);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*PRINTFLIKE1*/
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic void
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncoprintf(const char *fmt, ...)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync va_list ap;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync int n;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_ofp == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync va_start(ap, fmt);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync n = vfprintf(g_ofp, fmt, ap);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync va_end(ap);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (n < 0) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (errno != EINTR) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to write to %s",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_ofile ? g_ofile : "<stdout>");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync clearerr(g_ofp);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic char **
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncmake_argv(char *s)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync const char *ws = "\f\n\r\t\v ";
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char **argv = malloc(sizeof (char *) * (strlen(s) / 2 + 1));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync int argc = 0;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char *p = s;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (argv == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (NULL);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (p = strtok(s, ws); p != NULL; p = strtok(NULL, ws))
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync argv[argc++] = p;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (argc == 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync argv[argc++] = s;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync argv[argc] = NULL;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (argv);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic void
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncdof_prune(const char *fname)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync struct stat sbuf;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync size_t sz, i, j, mark, len;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char *buf;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync int msg = 0, fd;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((fd = open(fname, O_RDONLY)) == -1) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * This is okay only if the file doesn't exist at all.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (errno != ENOENT)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to open %s", fname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (fstat(fd, &sbuf) == -1)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to fstat %s", fname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((buf = malloc((sz = sbuf.st_size) + 1)) == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to allocate memory for %s", fname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (read(fd, buf, sz) != sz)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to read %s", fname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync buf[sz] = '\0';
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) close(fd);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((fd = open(fname, O_WRONLY | O_TRUNC)) == -1)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to open %s for writing", fname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync len = strlen("dof-data-");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (mark = 0, i = 0; i < sz; i++) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (strncmp(&buf[i], "dof-data-", len) != 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync continue;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * This is only a match if it's in the 0th column.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (i != 0 && buf[i - 1] != '\n')
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync continue;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (msg++ == 0) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync error("cleaned up old anonymous "
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "enabling in %s\n", fname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * We have a match. First write out our data up until now.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (i != mark) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (write(fd, &buf[mark], i - mark) != i - mark)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to write to %s", fname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * Now scan forward until we scan past a newline.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (j = i; j < sz && buf[j] != '\n'; j++)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync continue;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * Reset our mark.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((mark = j + 1) >= sz)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync i = j;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (mark < sz) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (write(fd, &buf[mark], sz - mark) != sz - mark)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to write to %s", fname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) close(fd);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync free(buf);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic void
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncetcsystem_prune(void)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync struct stat sbuf;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync size_t sz;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char *buf, *start, *end;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync int fd;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char *fname = g_etcfile, *tmpname;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((fd = open(fname, O_RDONLY)) == -1)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to open %s", fname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (fstat(fd, &sbuf) == -1)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to fstat %s", fname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((buf = malloc((sz = sbuf.st_size) + 1)) == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to allocate memory for %s", fname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (read(fd, buf, sz) != sz)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to read %s", fname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync buf[sz] = '\0';
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) close(fd);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((start = strstr(buf, g_etcbegin)) == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync goto out;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (strlen(buf) != sz) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("embedded nul byte in %s; manual repair of %s "
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "required\n", fname, fname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (strstr(start + 1, g_etcbegin) != NULL) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("multiple start sentinels in %s; manual repair of %s "
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "required\n", fname, fname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((end = strstr(buf, g_etcend)) == NULL) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("missing end sentinel in %s; manual repair of %s "
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "required\n", fname, fname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (start > end) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("end sentinel preceeds start sentinel in %s; manual "
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "repair of %s required\n", fname, fname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync end += strlen(g_etcend) + 1;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync bcopy(end, start, strlen(end) + 1);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync tmpname = alloca(sz = strlen(fname) + 80);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) snprintf(tmpname, sz, "%s.dtrace.%d", fname, getpid());
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((fd = open(tmpname,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync O_WRONLY | O_CREAT | O_EXCL, sbuf.st_mode)) == -1)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to create %s", tmpname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (write(fd, buf, strlen(buf)) < strlen(buf)) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) unlink(tmpname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to write to %s", tmpname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) close(fd);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifndef _MSC_VER
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (chown(tmpname, sbuf.st_uid, sbuf.st_gid) != 0) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) unlink(tmpname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to chown(2) %s to uid %d, gid %d", tmpname,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (int)sbuf.st_uid, (int)sbuf.st_gid);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (rename(tmpname, fname) == -1)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("rename of %s to %s failed", tmpname, fname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync error("cleaned up forceload directives in %s\n", fname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncout:
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync free(buf);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic void
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncetcsystem_add(void)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync const char *mods[20];
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync int nmods, line;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((g_ofp = fopen(g_ofile = g_etcfile, "a")) == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to open output file '%s'", g_ofile);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("%s\n", g_etcbegin);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (line = 0; g_etc[line] != NULL; line++)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("%s\n", g_etc[line]);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync nmods = dtrace_provider_modules(g_dtp, mods,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync sizeof (mods) / sizeof (char *) - 1);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (nmods >= sizeof (mods) / sizeof (char *))
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("unexpectedly large number of modules!");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync mods[nmods++] = "dtrace";
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (line = 0; line < nmods; line++)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("forceload: drv/%s\n", mods[line]);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("%s\n", g_etcend);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (fclose(g_ofp) == EOF)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to close output file '%s'", g_ofile);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync error("added forceload directives to %s\n", g_ofile);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic void
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncprint_probe_info(const dtrace_probeinfo_t *p)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char buf[BUFSIZ];
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync int i;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\n\tProbe Description Attributes\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\t\tIdentifier Names: %s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_stability_name(p->dtp_attr.dtat_name));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\t\tData Semantics: %s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_stability_name(p->dtp_attr.dtat_data));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\t\tDependency Class: %s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_class_name(p->dtp_attr.dtat_class));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\n\tArgument Attributes\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\t\tIdentifier Names: %s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_stability_name(p->dtp_arga.dtat_name));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\t\tData Semantics: %s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_stability_name(p->dtp_arga.dtat_data));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\t\tDependency Class: %s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_class_name(p->dtp_arga.dtat_class));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\n\tArgument Types\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (i = 0; i < p->dtp_argc; i++) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (ctf_type_name(p->dtp_argv[i].dtt_ctfp,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync p->dtp_argv[i].dtt_type, buf, sizeof (buf)) == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) strlcpy(buf, "(unknown)", sizeof (buf));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\t\targs[%d]: %s\n", i, buf);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (p->dtp_argc == 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\t\tNone\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*ARGSUSED*/
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncinfo_stmt(dtrace_hdl_t *dtp, dtrace_prog_t *pgp,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_stmtdesc_t *stp, dtrace_ecbdesc_t **last)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_ecbdesc_t *edp = stp->dtsd_ecbdesc;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_probedesc_t *pdp = &edp->dted_probe;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_probeinfo_t p;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (edp == *last)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (0);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\n%s:%s:%s:%s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync pdp->dtpd_provider, pdp->dtpd_mod, pdp->dtpd_func, pdp->dtpd_name);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_probe_info(dtp, pdp, &p) == 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync print_probe_info(&p);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync *last = edp;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (0);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * Execute the specified program by enabling the corresponding instrumentation.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * If -e has been specified, we get the program info but do not enable it. If
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * -v has been specified, we print a stability report for the program.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic void
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncexec_prog(const dtrace_cmd_t *dcp)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_ecbdesc_t *last = NULL;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_proginfo_t dpi;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (!g_exec) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_program_info(g_dtp, dcp->dc_prog, &dpi);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } else if (dtrace_program_exec(g_dtp, dcp->dc_prog, &dpi) == -1) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to enable '%s'", dcp->dc_name);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } else {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync notice("%s '%s' matched %u probe%s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_desc, dcp->dc_name,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dpi.dpi_matches, dpi.dpi_matches == 1 ? "" : "s");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_verbose) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\nStability attributes for %s %s:\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_desc, dcp->dc_name);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\n\tMinimum Probe Description Attributes\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\t\tIdentifier Names: %s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_stability_name(dpi.dpi_descattr.dtat_name));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\t\tData Semantics: %s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_stability_name(dpi.dpi_descattr.dtat_data));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\t\tDependency Class: %s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_class_name(dpi.dpi_descattr.dtat_class));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\n\tMinimum Statement Attributes\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\t\tIdentifier Names: %s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_stability_name(dpi.dpi_stmtattr.dtat_name));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\t\tData Semantics: %s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_stability_name(dpi.dpi_stmtattr.dtat_data));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\t\tDependency Class: %s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_class_name(dpi.dpi_stmtattr.dtat_class));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (!g_exec) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) dtrace_stmt_iter(g_dtp, dcp->dc_prog,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (dtrace_stmt_f *)info_stmt, &last);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } else
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_total += dpi.dpi_matches;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * Print out the specified DOF buffer as a set of ASCII bytes appropriate for
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * storing in a driver.conf(4) file associated with the dtrace driver.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic void
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncanon_prog(const dtrace_cmd_t *dcp, dof_hdr_t *dof, int n)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync const uchar_t *p, *q;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dof == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to create DOF image for '%s'", dcp->dc_name);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync p = (uchar_t *)dof;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync q = p + dof->dofh_loadsz;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("dof-data-%d=0x%x", n, *p++);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync while (p < q)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf(",0x%x", *p++);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf(";\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_dof_destroy(g_dtp, dof);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
e3c0207b236896a903ef328bfa74ee97e61bd38bvboxsync#ifndef VBOX
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * Link the specified D program in DOF form into an ELF file for use in either
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * helpers, userland provider definitions, or both. If -o was specified, that
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * path is used as the output file name. If -o wasn't specified and the input
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * program is from a script whose name is %.d, use basename(%.o) as the output
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * file name. Otherwise we use "d.out" as the default output file name.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic void
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsynclink_prog(dtrace_cmd_t *dcp)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char *p;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_cmdc == 1 && g_ofile != NULL) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) strlcpy(dcp->dc_ofile, g_ofile, sizeof (dcp->dc_ofile));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } else if ((p = strrchr(dcp->dc_arg, '.')) != NULL &&
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync strcmp(p, ".d") == 0) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync p[0] = '\0'; /* strip .d suffix */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) snprintf(dcp->dc_ofile, sizeof (dcp->dc_ofile),
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "%s.o", basename(dcp->dc_arg));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } else {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) snprintf(dcp->dc_ofile, sizeof (dcp->dc_ofile),
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_cmdc > 1 ? "%s.%d" : "%s", "d.out", (int)(dcp - g_cmdv));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_program_link(g_dtp, dcp->dc_prog, DTRACE_D_PROBES,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_ofile, g_objc, g_objv) != 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to link %s %s", dcp->dc_desc, dcp->dc_name);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
e3c0207b236896a903ef328bfa74ee97e61bd38bvboxsync#endif /* !VBOX */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*ARGSUSED*/
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsynclist_probe(dtrace_hdl_t *dtp, const dtrace_probedesc_t *pdp, void *arg)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_probeinfo_t p;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("%5d %10s %17s %33s %s\n", pdp->dtpd_id,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync pdp->dtpd_provider, pdp->dtpd_mod, pdp->dtpd_func, pdp->dtpd_name);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_verbose && dtrace_probe_info(dtp, pdp, &p) == 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync print_probe_info(&p);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (0);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*ARGSUSED*/
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsynclist_stmt(dtrace_hdl_t *dtp, dtrace_prog_t *pgp,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_stmtdesc_t *stp, dtrace_ecbdesc_t **last)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_ecbdesc_t *edp = stp->dtsd_ecbdesc;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (edp == *last)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (0);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_probe_iter(g_dtp, &edp->dted_probe, list_probe, NULL) != 0) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync error("failed to match %s:%s:%s:%s: %s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync edp->dted_probe.dtpd_provider, edp->dted_probe.dtpd_mod,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync edp->dted_probe.dtpd_func, edp->dted_probe.dtpd_name,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_errmsg(dtp, dtrace_errno(dtp)));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync *last = edp;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (0);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * List the probes corresponding to the specified program by iterating over
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * each statement and then matching probes to the statement probe descriptions.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic void
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsynclist_prog(const dtrace_cmd_t *dcp)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_ecbdesc_t *last = NULL;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) dtrace_stmt_iter(g_dtp, dcp->dc_prog,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (dtrace_stmt_f *)list_stmt, &last);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic void
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsynccompile_file(dtrace_cmd_t *dcp)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char *arg0;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync FILE *fp;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((fp = fopen(dcp->dc_arg, "r")) == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to open %s", dcp->dc_arg);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync arg0 = g_argv[0];
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_argv[0] = dcp->dc_arg;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((dcp->dc_prog = dtrace_program_fcompile(g_dtp, fp,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_cflags, g_argc, g_argv)) == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to compile script %s", dcp->dc_arg);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_argv[0] = arg0;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) fclose(fp);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_desc = "script";
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_name = dcp->dc_arg;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic void
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsynccompile_str(dtrace_cmd_t *dcp)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char *p;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((dcp->dc_prog = dtrace_program_strcompile(g_dtp, dcp->dc_arg,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_spec, g_cflags | DTRACE_C_PSPEC, g_argc, g_argv)) == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("invalid probe specifier %s", dcp->dc_arg);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((p = strpbrk(dcp->dc_arg, "{/;")) != NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync *p = '\0'; /* crop name for reporting */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_desc = "description";
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_name = dcp->dc_arg;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*ARGSUSED*/
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic void
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncprochandler(struct ps_prochandle *P, const char *msg, void *arg)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifndef VBOX
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync const psinfo_t *prp = Ppsinfo(P);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync int pid = Pstatus(P)->pr_pid;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char name[SIG2STR_MAX];
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (msg != NULL) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync notice("pid %d: %s\n", pid, msg);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync switch (Pstate(P)) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case PS_UNDEAD:
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * Ideally we would like to always report pr_wstat here, but it
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * isn't possible given current /proc semantics. If we grabbed
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * the process, Ppsinfo() will either fail or return a zeroed
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * psinfo_t depending on how far the parent is in reaping it.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * When /proc provides a stable pr_wstat in the status file,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * this code can be improved by examining this new pr_wstat.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (prp != NULL && WIFSIGNALED(prp->pr_wstat)) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync notice("pid %d terminated by %s\n", pid,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync proc_signame(WTERMSIG(prp->pr_wstat),
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync name, sizeof (name)));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } else if (prp != NULL && WEXITSTATUS(prp->pr_wstat) != 0) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync notice("pid %d exited with status %d\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync pid, WEXITSTATUS(prp->pr_wstat));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } else {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync notice("pid %d has exited\n", pid);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_pslive--;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case PS_LOST:
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync notice("pid %d exec'd a set-id or unobservable program\n", pid);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_pslive--;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif /* !VBOX */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*ARGSUSED*/
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncerrhandler(const dtrace_errdata_t *data, void *arg)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync error(data->dteda_msg);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (DTRACE_HANDLE_OK);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*ARGSUSED*/
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncdrophandler(const dtrace_dropdata_t *data, void *arg)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync error(data->dtdda_msg);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (DTRACE_HANDLE_OK);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*ARGSUSED*/
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncsetopthandler(const dtrace_setoptdata_t *data, void *arg)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (strcmp(data->dtsda_option, "quiet") == 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_quiet = data->dtsda_newval != DTRACEOPT_UNSET;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (strcmp(data->dtsda_option, "flowindent") == 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_flowindent = data->dtsda_newval != DTRACEOPT_UNSET;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (DTRACE_HANDLE_OK);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#define BUFDUMPHDR(hdr) \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) printf("%s: %s%s\n", g_pname, hdr, strlen(hdr) > 0 ? ":" : "");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#define BUFDUMPSTR(ptr, field) \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) printf("%s: %20s => ", g_pname, #field); \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((ptr)->field != NULL) { \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync const char *c = (ptr)->field; \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) printf("\""); \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync do { \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (*c == '\n') { \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) printf("\\n"); \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync continue; \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) printf("%c", *c); \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } while (*c++ != '\0'); \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) printf("\"\n"); \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } else { \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) printf("<NULL>\n"); \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#define BUFDUMPASSTR(ptr, field, str) \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) printf("%s: %20s => %s\n", g_pname, #field, str);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#define BUFDUMP(ptr, field) \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) printf("%s: %20s => %lld\n", g_pname, #field, \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (long long)(ptr)->field);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync#define BUFDUMPPTR(ptr, field) \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) printf("%s: %20s => %s\n", g_pname, #field, \
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (ptr)->field != NULL ? "<non-NULL>" : "<NULL>");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*ARGSUSED*/
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncbufhandler(const dtrace_bufdata_t *bufdata, void *arg)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync const dtrace_aggdata_t *agg = bufdata->dtbda_aggdata;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync const dtrace_recdesc_t *rec = bufdata->dtbda_recdesc;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync const dtrace_probedesc_t *pd;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync uint32_t flags = bufdata->dtbda_flags;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char buf[512], *c = buf, *end = c + sizeof (buf);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync int i, printed;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync struct {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync const char *name;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync uint32_t value;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } flagnames[] = {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync { "AGGVAL", DTRACE_BUFDATA_AGGVAL },
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync { "AGGKEY", DTRACE_BUFDATA_AGGKEY },
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync { "AGGFORMAT", DTRACE_BUFDATA_AGGFORMAT },
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync { "AGGLAST", DTRACE_BUFDATA_AGGLAST },
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync { "???", UINT32_MAX },
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync { NULL }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync };
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (bufdata->dtbda_probe != NULL) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync pd = bufdata->dtbda_probe->dtpda_pdesc;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } else if (agg != NULL) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync pd = agg->dtada_pdesc;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } else {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync pd = NULL;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPHDR(">>> Called buffer handler");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPHDR("");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPHDR(" dtrace_bufdata");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPSTR(bufdata, dtbda_buffered);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPPTR(bufdata, dtbda_probe);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPPTR(bufdata, dtbda_aggdata);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPPTR(bufdata, dtbda_recdesc);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) snprintf(c, end - c, "0x%x ", bufdata->dtbda_flags);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync c += strlen(c);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (i = 0, printed = 0; flagnames[i].name != NULL; i++) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (!(flags & flagnames[i].value))
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync continue;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) snprintf(c, end - c,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "%s%s", printed++ ? " | " : "(", flagnames[i].name);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync c += strlen(c);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync flags &= ~flagnames[i].value;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (printed)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) snprintf(c, end - c, ")");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPASSTR(bufdata, dtbda_flags, buf);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPHDR("");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (pd != NULL) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPHDR(" dtrace_probedesc");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPSTR(pd, dtpd_provider);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPSTR(pd, dtpd_mod);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPSTR(pd, dtpd_func);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPSTR(pd, dtpd_name);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPHDR("");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (rec != NULL) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPHDR(" dtrace_recdesc");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMP(rec, dtrd_action);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMP(rec, dtrd_size);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (agg != NULL) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync uint8_t *data;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync int lim = rec->dtrd_size;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) sprintf(buf, "%d (data: ", rec->dtrd_offset);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync c = buf + strlen(buf);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (lim > sizeof (uint64_t))
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync lim = sizeof (uint64_t);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync data = (uint8_t *)agg->dtada_data + rec->dtrd_offset;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (i = 0; i < lim; i++) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) snprintf(c, end - c, "%s%02x",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync i == 0 ? "" : " ", *data++);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync c += strlen(c);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) snprintf(c, end - c,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "%s)", lim < rec->dtrd_size ? " ..." : "");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPASSTR(rec, dtrd_offset, buf);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } else {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMP(rec, dtrd_offset);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPHDR("");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (agg != NULL) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_aggdesc_t *desc = agg->dtada_desc;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPHDR(" dtrace_aggdesc");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPSTR(desc, dtagd_name);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMP(desc, dtagd_varid);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMP(desc, dtagd_id);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMP(desc, dtagd_nrecs);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync BUFDUMPHDR("");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (DTRACE_HANDLE_OK);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*ARGSUSED*/
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncchewrec(const dtrace_probedata_t *data, const dtrace_recdesc_t *rec, void *arg)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_actkind_t act;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync uintptr_t addr;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (rec == NULL) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * We have processed the final record; output the newline if
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * we're not in quiet mode.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (!g_quiet)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (DTRACE_CONSUME_NEXT);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync act = rec->dtrd_action;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync addr = (uintptr_t)data->dtpda_data;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (act == DTRACEACT_EXIT) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_status = *((uint32_t *)addr);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (DTRACE_CONSUME_NEXT);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (DTRACE_CONSUME_THIS);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*ARGSUSED*/
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic int
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncchew(const dtrace_probedata_t *data, void *arg)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_probedesc_t *pd = data->dtpda_pdesc;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync processorid_t cpu = data->dtpda_cpu;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync static int heading;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_impatient) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_newline = 0;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (DTRACE_CONSUME_ABORT);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (heading == 0) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (!g_flowindent) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (!g_quiet) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("%3s %6s %32s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "CPU", "ID", "FUNCTION:NAME");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } else {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("%3s %-41s\n", "CPU", "FUNCTION");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync heading = 1;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (!g_flowindent) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (!g_quiet) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char name[DTRACE_FUNCNAMELEN + DTRACE_NAMELEN + 2];
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) snprintf(name, sizeof (name), "%s:%s",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync pd->dtpd_func, pd->dtpd_name);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("%3d %6d %32s ", cpu, pd->dtpd_id, name);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } else {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync int indent = data->dtpda_indent;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char *name;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync size_t len;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (data->dtpda_flow == DTRACEFLOW_NONE) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync len = indent + DTRACE_FUNCNAMELEN + DTRACE_NAMELEN + 5;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync name = alloca(len);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) snprintf(name, len, "%*s%s%s:%s", indent, "",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync data->dtpda_prefix, pd->dtpd_func,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync pd->dtpd_name);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } else {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync len = indent + DTRACE_FUNCNAMELEN + 5;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync name = alloca(len);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) snprintf(name, len, "%*s%s%s", indent, "",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync data->dtpda_prefix, pd->dtpd_func);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("%3d %-41s ", cpu, name);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (DTRACE_CONSUME_THIS);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic void
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncgo(void)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync int i;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync struct {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char *name;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char *optname;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_optval_t val;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } bufs[] = {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync { "buffer size", "bufsize" },
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync { "aggregation size", "aggsize" },
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync { "speculation size", "specsize" },
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync { "dynamic variable size", "dynvarsize" },
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync { NULL }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }, rates[] = {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync { "cleaning rate", "cleanrate" },
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync { "status rate", "statusrate" },
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync { NULL }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync };
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (i = 0; bufs[i].name != NULL; i++) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_getopt(g_dtp, bufs[i].optname, &bufs[i].val) == -1)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("couldn't get option %s", bufs[i].optname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (i = 0; rates[i].name != NULL; i++) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_getopt(g_dtp, rates[i].optname, &rates[i].val) == -1)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("couldn't get option %s", rates[i].optname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_go(g_dtp) == -1)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("could not enable tracing");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (i = 0; bufs[i].name != NULL; i++) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_optval_t j = 0, mul = 10;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_optval_t nsize;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (bufs[i].val == DTRACEOPT_UNSET)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync continue;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) dtrace_getopt(g_dtp, bufs[i].optname, &nsize);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (nsize == DTRACEOPT_UNSET || nsize == 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync continue;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (nsize >= bufs[i].val - sizeof (uint64_t))
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync continue;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (; (INT64_C(1) << mul) <= nsize; j++, mul += 10)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync continue;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (!(nsize & ((INT64_C(1) << (mul - 10)) - 1))) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync error("%s lowered to %lld%c\n", bufs[i].name,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (long long)nsize >> (mul - 10), " kmgtpe"[j]);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } else {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync error("%s lowered to %lld bytes\n", bufs[i].name,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (long long)nsize);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (i = 0; rates[i].name != NULL; i++) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_optval_t nval;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char *dir;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (rates[i].val == DTRACEOPT_UNSET)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync continue;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) dtrace_getopt(g_dtp, rates[i].optname, &nval);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (nval == DTRACEOPT_UNSET || nval == 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync continue;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (rates[i].val == nval)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync continue;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dir = nval > rates[i].val ? "reduced" : "increased";
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (nval <= NANOSEC && (NANOSEC % nval) == 0) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync error("%s %s to %lld hz\n", rates[i].name, dir,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (long long)NANOSEC / (long long)nval);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync continue;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((nval % NANOSEC) == 0) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync error("%s %s to once every %lld seconds\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync rates[i].name, dir,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (long long)nval / (long long)NANOSEC);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync continue;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync error("%s %s to once every %lld nanoseconds\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync rates[i].name, dir, (long long)nval);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync/*ARGSUSED*/
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncstatic void
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncintr(int signo)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (!g_intr)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_newline = 1;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_intr++)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_impatient = 1;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
57a7aeb014cdd1900c8bfecb0f9dcc7b96a61cc9vboxsync#ifdef VBOX
57a7aeb014cdd1900c8bfecb0f9dcc7b96a61cc9vboxsyncDECLEXPORT(int) RTCALL VBoxDTraceMain(int argc, char **argv)
57a7aeb014cdd1900c8bfecb0f9dcc7b96a61cc9vboxsync#else
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncint
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsyncmain(int argc, char *argv[])
57a7aeb014cdd1900c8bfecb0f9dcc7b96a61cc9vboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync{
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_bufdesc_t buf;
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifndef _MSC_VER
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync struct sigaction act, oact;
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_status_t status[2];
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_optval_t opt;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_cmd_t *dcp;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync int done = 0, mode = 0;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync int err, i;
e3c0207b236896a903ef328bfa74ee97e61bd38bvboxsync#ifndef VBOX
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char c, *p, **v;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync struct ps_prochandle *P;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync pid_t pid;
57a7aeb014cdd1900c8bfecb0f9dcc7b96a61cc9vboxsync
57a7aeb014cdd1900c8bfecb0f9dcc7b96a61cc9vboxsync g_pname = basename(argv[0]);
0fea9bc4af28518049902cca68d209af223a8e13vboxsync#else
0fea9bc4af28518049902cca68d209af223a8e13vboxsync int c;
0fea9bc4af28518049902cca68d209af223a8e13vboxsync char *p;
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync RTGETOPTUNION ValueUnion;
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync RTGETOPTSTATE GetState;
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync
57a7aeb014cdd1900c8bfecb0f9dcc7b96a61cc9vboxsync err = RTR3InitDll(0);
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync if (RT_FAILURE(err))
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync return RTMsgInitFailure(err);
dfd08267d2958ae1cd559dd7dc2f36bf5461648dvboxsync dtrace_init();
e3c0207b236896a903ef328bfa74ee97e61bd38bvboxsync
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync g_ofp = stdout;
57a7aeb014cdd1900c8bfecb0f9dcc7b96a61cc9vboxsync g_pname = RTProcShortName();
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (argc == 1)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (usage(stderr));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((g_argv = malloc(sizeof (char *) * argc)) == NULL ||
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (g_cmdv = malloc(sizeof (dtrace_cmd_t) * argc)) == NULL ||
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (g_psv = malloc(sizeof (struct ps_prochandle *) * argc)) == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to allocate memory for arguments");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_argv[g_argc++] = argv[0]; /* propagate argv[0] to D as $0/$$0 */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync argv[0] = g_pname; /* rewrite argv[0] for getopt errors */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync bzero(status, sizeof (status));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync bzero(&buf, sizeof (buf));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * Make an initial pass through argv[] processing any arguments that
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * affect our behavior mode (g_mode) and flags used for dtrace_open().
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * We also accumulate arguments that are not affiliated with getopt
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * options into g_argv[], and abort if any invalid options are found.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifndef VBOX
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (optind = 1; optind < argc; optind++) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync while ((c = getopt(argc, argv, DTRACE_OPTSTR)) != EOF) {
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#else
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync RTGetOptInit(&GetState, argc, argv, g_aOptions, RT_ELEMENTS(g_aOptions), 1, 0);
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync while ((c = RTGetOpt(&GetState, &ValueUnion))) {
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync {
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync const char *optarg = ValueUnion.psz;
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync switch (c) {
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifndef VBOX
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case '3':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (strcmp(optarg, "2") != 0) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) fprintf(stderr,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "%s: illegal option -- 3%s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync argv[0], optarg);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (usage(stderr));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#else
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync case 10032:
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_oflags &= ~DTRACE_O_LP64;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_oflags |= DTRACE_O_ILP32;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifndef VBOX
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case '6':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (strcmp(optarg, "4") != 0) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) fprintf(stderr,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "%s: illegal option -- 6%s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync argv[0], optarg);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (usage(stderr));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#else
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync case 10064:
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_oflags &= ~DTRACE_O_ILP32;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_oflags |= DTRACE_O_LP64;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'a':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_grabanon++; /* also checked in pass 2 below */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'A':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_mode = DMODE_ANON;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_exec = 0;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync mode++;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'e':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_exec = 0;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync done = 1;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'h':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_mode = DMODE_HEADER;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_oflags |= DTRACE_O_NODEV;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_cflags |= DTRACE_C_ZDEFS; /* -h implies -Z */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_exec = 0;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync mode++;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
e3c0207b236896a903ef328bfa74ee97e61bd38bvboxsync#ifndef VBOX
dfd08267d2958ae1cd559dd7dc2f36bf5461648dvboxsync case 'G':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_mode = DMODE_LINK;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_oflags |= DTRACE_O_NODEV;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_cflags |= DTRACE_C_ZDEFS; /* -G implies -Z */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_exec = 0;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync mode++;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
e3c0207b236896a903ef328bfa74ee97e61bd38bvboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'l':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_mode = DMODE_LIST;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_cflags |= DTRACE_C_ZDEFS; /* -l implies -Z */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync mode++;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'V':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_mode = DMODE_VERS;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync mode++;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifndef VBOX
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync default:
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (strchr(DTRACE_OPTSTR, c) == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (usage(stderr));
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#else
dfd08267d2958ae1cd559dd7dc2f36bf5461648dvboxsync case 'c':
dfd08267d2958ae1cd559dd7dc2f36bf5461648dvboxsync case 'p':
dfd08267d2958ae1cd559dd7dc2f36bf5461648dvboxsync case 'G':
dfd08267d2958ae1cd559dd7dc2f36bf5461648dvboxsync fprintf(stderr, "%s: -%c is not supported\n", g_pname, c);
dfd08267d2958ae1cd559dd7dc2f36bf5461648dvboxsync return (E_USAGE);
dfd08267d2958ae1cd559dd7dc2f36bf5461648dvboxsync
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync case VINF_GETOPT_NOT_OPTION:
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync g_argv[g_argc++] = (char *)ValueUnion.psz;
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync break;
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync default:
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync if (c < 0) { /* Note: Not all options are handled. */
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync RTGetOptPrintError(c, &ValueUnion);
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync return (usage(stderr));
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync }
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifndef VBOX
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (optind < argc)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_argv[g_argc++] = argv[optind];
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (mode > 1) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) fprintf(stderr, "%s: only one of the [-AGhlV] options "
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "can be specified at a time\n", g_pname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (E_USAGE);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_mode == DMODE_VERS)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (printf("%s: %s\n", g_pname, _dtrace_version) <= 0);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifndef VBOX
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * If we're in linker mode and the data model hasn't been specified,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * we try to guess the appropriate setting by examining the object
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * files. We ignore certain errors since we'll catch them later when
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * we actually process the object files.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_mode == DMODE_LINK &&
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (g_oflags & (DTRACE_O_ILP32 | DTRACE_O_LP64)) == 0 &&
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync elf_version(EV_CURRENT) != EV_NONE) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync int fd;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync Elf *elf;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync GElf_Ehdr ehdr;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (i = 1; i < g_argc; i++) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((fd = open64(g_argv[i], O_RDONLY)) == -1)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((elf = elf_begin(fd, ELF_C_READ, NULL)) == NULL) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) close(fd);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (elf_kind(elf) != ELF_K_ELF ||
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync gelf_getehdr(elf, &ehdr) == NULL) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) close(fd);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) elf_end(elf);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) close(fd);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) elf_end(elf);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (ehdr.e_ident[EI_CLASS] == ELFCLASS64) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_oflags & DTRACE_O_ILP32) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("can't mix 32-bit and 64-bit "
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "object files\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_oflags |= DTRACE_O_LP64;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } else if (ehdr.e_ident[EI_CLASS] == ELFCLASS32) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_oflags & DTRACE_O_LP64) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("can't mix 32-bit and 64-bit "
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "object files\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_oflags |= DTRACE_O_ILP32;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } else {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif /* !VBOX */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * Open libdtrace. If we are not actually going to be enabling any
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * instrumentation attempt to reopen libdtrace using DTRACE_O_NODEV.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync while ((g_dtp = dtrace_open(DTRACE_VERSION, g_oflags, &err)) == NULL) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (!(g_oflags & DTRACE_O_NODEV) && !g_exec && !g_grabanon) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_oflags |= DTRACE_O_NODEV;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync continue;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to initialize dtrace: %s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_errmsg(NULL, err));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) dtrace_setopt(g_dtp, "bufsize", "4m");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) dtrace_setopt(g_dtp, "aggsize", "4m");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * If -G is specified, enable -xlink=dynamic and -xunodefs to permit
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * references to undefined symbols to remain as unresolved relocations.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * If -A is specified, enable -xlink=primary to permit static linking
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * only to kernel symbols that are defined in a primary kernel module.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_mode == DMODE_LINK) {
e3c0207b236896a903ef328bfa74ee97e61bd38bvboxsync#ifndef VBOX /* No link mode. */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) dtrace_setopt(g_dtp, "linkmode", "dynamic");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) dtrace_setopt(g_dtp, "unodefs", NULL);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * Use the remaining arguments as the list of object files
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * when in linker mode.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_objc = g_argc - 1;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_objv = g_argv + 1;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * We still use g_argv[0], the name of the executable.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_argc = 1;
e3c0207b236896a903ef328bfa74ee97e61bd38bvboxsync#else /* VBOX */
e3c0207b236896a903ef328bfa74ee97e61bd38bvboxsync AssertFailed();
e3c0207b236896a903ef328bfa74ee97e61bd38bvboxsync#endif /* VBOX */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } else if (g_mode == DMODE_ANON)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) dtrace_setopt(g_dtp, "linkmode", "primary");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * Now that we have libdtrace open, make a second pass through argv[]
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * to perform any dtrace_setopt() calls and change any compiler flags.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * We also accumulate any program specifications into our g_cmdv[] at
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * this time; these will compiled as part of the fourth processing pass.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifndef VBOX
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (optind = 1; optind < argc; optind++) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync while ((c = getopt(argc, argv, DTRACE_OPTSTR)) != EOF) {
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#else
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync RTGetOptInit(&GetState, argc, argv, g_aOptions, RT_ELEMENTS(g_aOptions), 1, 0);
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync while ((c = RTGetOpt(&GetState, &ValueUnion))) {
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync {
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync char *optarg = (char *)ValueUnion.psz;
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync switch (c) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'a':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_setopt(g_dtp, "grabanon", 0) != 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to set -a");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'b':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_setopt(g_dtp,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "bufsize", optarg) != 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to set -b %s", optarg);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'B':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_ofp = NULL;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'C':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_cflags |= DTRACE_C_CPP;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'D':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_setopt(g_dtp, "define", optarg) != 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to set -D %s", optarg);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'f':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp = &g_cmdv[g_cmdc++];
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_func = compile_str;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_spec = DTRACE_PROBESPEC_FUNC;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_arg = optarg;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'F':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_setopt(g_dtp, "flowindent", 0) != 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to set -F");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'H':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_setopt(g_dtp, "cpphdrs", 0) != 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to set -H");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'i':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp = &g_cmdv[g_cmdc++];
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_func = compile_str;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_spec = DTRACE_PROBESPEC_NAME;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_arg = optarg;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'I':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_setopt(g_dtp, "incdir", optarg) != 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to set -I %s", optarg);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'L':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_setopt(g_dtp, "libdir", optarg) != 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to set -L %s", optarg);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'm':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp = &g_cmdv[g_cmdc++];
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_func = compile_str;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_spec = DTRACE_PROBESPEC_MOD;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_arg = optarg;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'n':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp = &g_cmdv[g_cmdc++];
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_func = compile_str;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_spec = DTRACE_PROBESPEC_NAME;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_arg = optarg;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'P':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp = &g_cmdv[g_cmdc++];
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_func = compile_str;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_spec = DTRACE_PROBESPEC_PROVIDER;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_arg = optarg;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'q':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_setopt(g_dtp, "quiet", 0) != 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to set -q");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'o':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_ofile = optarg;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 's':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp = &g_cmdv[g_cmdc++];
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_func = compile_file;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_spec = DTRACE_PROBESPEC_NONE;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dcp->dc_arg = optarg;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'S':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_cflags |= DTRACE_C_DIFV;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'U':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_setopt(g_dtp, "undef", optarg) != 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to set -U %s", optarg);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'v':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_verbose++;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'w':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_setopt(g_dtp, "destructive", 0) != 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to set -w");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'x':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((p = strchr(optarg, '=')) != NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync *p++ = '\0';
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_setopt(g_dtp, optarg, p) != 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to set -x %s", optarg);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'X':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_setopt(g_dtp, "stdc", optarg) != 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to set -X %s", optarg);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'Z':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_cflags |= DTRACE_C_ZDEFS;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifndef VBOX
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync default:
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (strchr(DTRACE_OPTSTR, c) == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (usage(stderr));
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#else
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync default:
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync if (c < 0) { /* Note: Not all options are handled. */
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync RTGetOptPrintError(c, &ValueUnion);
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync return (usage(stderr));
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync }
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_ofp == NULL && g_mode != DMODE_EXEC) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) fprintf(stderr, "%s: -B not valid in combination"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync " with [-AGl] options\n", g_pname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (E_USAGE);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_ofp == NULL && g_ofile != NULL) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) fprintf(stderr, "%s: -B not valid in combination"
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync " with -o option\n", g_pname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (E_USAGE);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
dfd08267d2958ae1cd559dd7dc2f36bf5461648dvboxsync#ifndef VBOX
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * In our third pass we handle any command-line options related to
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * grabbing or creating victim processes. The behavior of these calls
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * may been affected by any library options set by the second pass.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifndef VBOX
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (optind = 1; optind < argc; optind++) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync while ((c = getopt(argc, argv, DTRACE_OPTSTR)) != EOF) {
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#else
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync RTGetOptInit(&GetState, argc, argv, g_aOptions, RT_ELEMENTS(g_aOptions), 1, 0);
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync while ((c = RTGetOpt(&GetState, &ValueUnion))) {
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync {
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync char *optarg = (char *)ValueUnion.psz;
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync switch (c) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'c':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((v = make_argv(optarg)) == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to allocate memory");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync P = dtrace_proc_create(g_dtp, v[0], v);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (P == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal(NULL); /* dtrace_errmsg() only */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_psv[g_psc++] = P;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync free(v);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case 'p':
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync errno = 0;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync pid = strtol(optarg, &p, 10);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (errno != 0 || p == optarg || p[0] != '\0')
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("invalid pid: %s\n", optarg);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync P = dtrace_proc_grab(g_dtp, pid, 0);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (P == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal(NULL); /* dtrace_errmsg() only */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_psv[g_psc++] = P;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
dfd08267d2958ae1cd559dd7dc2f36bf5461648dvboxsync#endif /* !VBOX */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * In our fourth pass we finish g_cmdv[] by calling dc_func to convert
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * each string or file specification into a compiled program structure.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (i = 0; i < g_cmdc; i++)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_cmdv[i].dc_func(&g_cmdv[i]);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_mode != DMODE_LIST) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_handle_err(g_dtp, &errhandler, NULL) == -1)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to establish error handler");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_handle_drop(g_dtp, &drophandler, NULL) == -1)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to establish drop handler");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_handle_proc(g_dtp, &prochandler, NULL) == -1)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to establish proc handler");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_handle_setopt(g_dtp, &setopthandler, NULL) == -1)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to establish setopt handler");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_ofp == NULL &&
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_handle_buffered(g_dtp, &bufhandler, NULL) == -1)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to establish buffered handler");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) dtrace_getopt(g_dtp, "flowindent", &opt);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_flowindent = opt != DTRACEOPT_UNSET;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) dtrace_getopt(g_dtp, "grabanon", &opt);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_grabanon = opt != DTRACEOPT_UNSET;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) dtrace_getopt(g_dtp, "quiet", &opt);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_quiet = opt != DTRACEOPT_UNSET;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * Now make a fifth and final pass over the options that have been
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * turned into programs and saved in g_cmdv[], performing any mode-
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * specific processing. If g_mode is DMODE_EXEC, we will break out
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * of the switch() and continue on to the data processing loop. For
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * other modes, we will exit dtrace once mode-specific work is done.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync switch (g_mode) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case DMODE_EXEC:
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_ofile != NULL && (g_ofp = fopen(g_ofile, "a")) == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to open output file '%s'", g_ofile);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (i = 0; i < g_cmdc; i++)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync exec_prog(&g_cmdv[i]);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (done && !g_grabanon) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_close(g_dtp);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (g_status);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case DMODE_ANON:
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_ofile == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_ofile = "/kernel/drv/dtrace.conf";
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dof_prune(g_ofile); /* strip out any old DOF directives */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync etcsystem_prune(); /* string out any forceload directives */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_cmdc == 0) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_close(g_dtp);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (g_status);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((g_ofp = fopen(g_ofile, "a")) == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to open output file '%s'", g_ofile);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (i = 0; i < g_cmdc; i++) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync anon_prog(&g_cmdv[i],
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_dof_create(g_dtp, g_cmdv[i].dc_prog, 0), i);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * Dump out the DOF corresponding to the error handler and the
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * current options as the final DOF property in the .conf file.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync anon_prog(NULL, dtrace_geterr_dof(g_dtp), i++);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync anon_prog(NULL, dtrace_getopt_dof(g_dtp), i++);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (fclose(g_ofp) == EOF)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to close output file '%s'", g_ofile);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * These messages would use notice() rather than error(), but
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * we don't want them suppressed when -A is run on a D program
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * that itself contains a #pragma D option quiet.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync error("saved anonymous enabling in %s\n", g_ofile);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync etcsystem_add();
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync error("run update_drv(1M) or reboot to enable changes\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_close(g_dtp);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (g_status);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case DMODE_LINK:
e3c0207b236896a903ef328bfa74ee97e61bd38bvboxsync#ifndef VBOX /* No link mode. */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_cmdc == 0) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) fprintf(stderr, "%s: -G requires one or more "
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "scripts or enabling options\n", g_pname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_close(g_dtp);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (E_USAGE);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (i = 0; i < g_cmdc; i++)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync link_prog(&g_cmdv[i]);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_cmdc > 1 && g_ofile != NULL) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char **objv = alloca(g_cmdc * sizeof (char *));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (i = 0; i < g_cmdc; i++)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync objv[i] = g_cmdv[i].dc_ofile;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_program_link(g_dtp, NULL, DTRACE_D_PROBES,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_ofile, g_cmdc, objv) != 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal(NULL); /* dtrace_errmsg() only */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_close(g_dtp);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (g_status);
e3c0207b236896a903ef328bfa74ee97e61bd38bvboxsync#else /* VBOX */
e3c0207b236896a903ef328bfa74ee97e61bd38bvboxsync AssertFailed();
e3c0207b236896a903ef328bfa74ee97e61bd38bvboxsync dtrace_close(g_dtp);
e3c0207b236896a903ef328bfa74ee97e61bd38bvboxsync return (E_USAGE);
e3c0207b236896a903ef328bfa74ee97e61bd38bvboxsync#endif /* VBOX */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case DMODE_LIST:
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_ofile != NULL && (g_ofp = fopen(g_ofile, "a")) == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to open output file '%s'", g_ofile);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("%5s %10s %17s %33s %s\n",
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "ID", "PROVIDER", "MODULE", "FUNCTION", "NAME");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (i = 0; i < g_cmdc; i++)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync list_prog(&g_cmdv[i]);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_cmdc == 0)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) dtrace_probe_iter(g_dtp, NULL, list_probe, NULL);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_close(g_dtp);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (g_status);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case DMODE_HEADER:
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_cmdc == 0) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) fprintf(stderr, "%s: -h requires one or more "
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "scripts or enabling options\n", g_pname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_close(g_dtp);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (E_USAGE);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_ofile == NULL) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync char *p;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_cmdc > 1) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) fprintf(stderr, "%s: -h requires an "
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "output file if multiple scripts are "
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "specified\n", g_pname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_close(g_dtp);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (E_USAGE);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((p = strrchr(g_cmdv[0].dc_arg, '.')) == NULL ||
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync strcmp(p, ".d") != 0) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) fprintf(stderr, "%s: -h requires an "
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "output file if no scripts are "
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "specified\n", g_pname);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_close(g_dtp);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (E_USAGE);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync p[0] = '\0'; /* strip .d suffix */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_ofile = p = g_cmdv[0].dc_ofile;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) snprintf(p, sizeof (g_cmdv[0].dc_ofile),
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync "%s.h", basename(g_cmdv[0].dc_arg));
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if ((g_ofp = fopen(g_ofile, "w")) == NULL)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fatal("failed to open header file '%s'", g_ofile);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("/*\n * Generated by dtrace(1M).\n */\n\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_program_header(g_dtp, g_ofp, g_ofile) != 0 ||
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync fclose(g_ofp) == EOF)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to create header file %s", g_ofile);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_close(g_dtp);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (g_status);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * If -a and -Z were not specified and no probes have been matched, no
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * probe criteria was specified on the command line and we abort.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_total == 0 && !g_grabanon && !(g_cflags & DTRACE_C_ZDEFS))
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("no probes %s\n", g_cmdc ? "matched" : "specified");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * Start tracing. Once we dtrace_go(), reload any options that affect
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * our globals in case consuming anonymous state has changed them.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync go();
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) dtrace_getopt(g_dtp, "flowindent", &opt);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_flowindent = opt != DTRACEOPT_UNSET;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) dtrace_getopt(g_dtp, "grabanon", &opt);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_grabanon = opt != DTRACEOPT_UNSET;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) dtrace_getopt(g_dtp, "quiet", &opt);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_quiet = opt != DTRACEOPT_UNSET;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) dtrace_getopt(g_dtp, "destructive", &opt);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (opt != DTRACEOPT_UNSET)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync notice("allowing destructive actions\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#ifndef _MSC_VER
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) sigemptyset(&act.sa_mask);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync act.sa_flags = 0;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync act.sa_handler = intr;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (sigaction(SIGINT, NULL, &oact) == 0 && oact.sa_handler != SIG_IGN)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) sigaction(SIGINT, &act, NULL);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (sigaction(SIGTERM, NULL, &oact) == 0 && oact.sa_handler != SIG_IGN)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync (void) sigaction(SIGTERM, &act, NULL);
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#else
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync signal(SIGINT, intr);
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync signal(SIGTERM, intr);
ed2a9def43e38be9ed1ee10cdc35307de992a165vboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * Now that tracing is active and we are ready to consume trace data,
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * continue any grabbed or created processes, setting them running
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * using the /proc control mechanism inside of libdtrace.
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
dfd08267d2958ae1cd559dd7dc2f36bf5461648dvboxsync#ifndef VBOX
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync for (i = 0; i < g_psc; i++)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_proc_continue(g_dtp, g_psv[i]);
dfd08267d2958ae1cd559dd7dc2f36bf5461648dvboxsync#endif
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_pslive = g_psc; /* count for prochandler() */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync do {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (!g_intr && !done)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_sleep(g_dtp);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_newline) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync /*
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * Output a newline just to make the output look
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * slightly cleaner. Note that we do this even in
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync * "quiet" mode...
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync */
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync g_newline = 0;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (done || g_intr || (g_psc != 0 && g_pslive == 0)) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync done = 1;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_stop(g_dtp) == -1)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("couldn't stop tracing");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync switch (dtrace_work(g_dtp, g_ofp, chew, chewrec, NULL)) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case DTRACE_WORKSTATUS_DONE:
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync done = 1;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync case DTRACE_WORKSTATUS_OKAY:
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync break;
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync default:
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (!g_impatient && dtrace_errno(g_dtp) != EINTR)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("processing aborted");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (g_ofp != NULL && fflush(g_ofp) == EOF)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync clearerr(g_ofp);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync } while (!done);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync oprintf("\n");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (!g_impatient) {
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync if (dtrace_aggregate_print(g_dtp, g_ofp, NULL) == -1 &&
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_errno(g_dtp) != EINTR)
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dfatal("failed to print aggregations");
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync }
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync dtrace_close(g_dtp);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync return (g_status);
14ea49401f3c8c61422aefbda43809e275f60c6cvboxsync}