ktrace.tproj/ktrace.c

   1 /*
   2  * Copyright (c) 1999 Apple Computer, Inc. All rights reserved.
   3  *
   4  * @APPLE_LICENSE_HEADER_START@
   5  *
   6  * "Portions Copyright (c) 1999 Apple Computer, Inc.  All Rights
   7  * Reserved.  This file contains Original Code and/or Modifications of
   8  * Original Code as defined in and that are subject to the Apple Public
   9  * Source License Version 1.0 (the 'License').  You may not use this file
  10  * except in compliance with the License.  Please obtain a copy of the
  11  * License at http://www.apple.com/publicsource and read it before using
  12  * this file.
  13  *
  14  * The Original Code and all software distributed under the License are
  15  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  16  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  17  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT.  Please see the
  19  * License for the specific language governing rights and limitations
  20  * under the License."
  21  *
  22  * @APPLE_LICENSE_HEADER_END@
  23  */
  24 /*-
  25  * Copyright (c) 1988, 1993
  26  *      The Regents of the University of California.  All rights reserved.
  27  *
  28  * Redistribution and use in source and binary forms, with or without
  29  * modification, are permitted provided that the following conditions
  30  * are met:
  31  * 1. Redistributions of source code must retain the above copyright
  32  *    notice, this list of conditions and the following disclaimer.
  33  * 2. Redistributions in binary form must reproduce the above copyright
  34  *    notice, this list of conditions and the following disclaimer in the
  35  *    documentation and/or other materials provided with the distribution.
  36  * 3. All advertising materials mentioning features or use of this software
  37  *    must display the following acknowledgement:
  38  *      This product includes software developed by the University of
  39  *      California, Berkeley and its contributors.
  40  * 4. Neither the name of the University nor the names of its contributors
  41  *    may be used to endorse or promote products derived from this software
  42  *    without specific prior written permission.
  43  *
  44  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  45  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  46  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  47  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  48  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  49  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  50  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  51  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  52  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  53  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  54  * SUCH DAMAGE.
  55  */
  56
  57 #include <sys/param.h>
  58 #include <sys/stat.h>
  59 #include <sys/file.h>
  60 #include <sys/time.h>
  61 #include <sys/errno.h>
  62 #include <sys/uio.h>
  63 #include <sys/ktrace.h>
  64
  65 #include <err.h>
  66 #include <stdio.h>
  67 #include <stdlib.h>
  68 #include <unistd.h>
  69
  70 #include "ktrace.h"
  71
  72 void no_ktrace __P((int));
  73 void usage __P((void));
  74
  75 main(argc, argv)
  76         int argc;
  77         char **argv;
  78 {
  79         enum { NOTSET, CLEAR, CLEARALL } clear;
  80         int append, ch, fd, inherit, ops, pid, pidset, trpoints;
  81         char *tracefile;
  82
  83         clear = NOTSET;
  84         append = ops = pidset = inherit = 0;
  85         trpoints = DEF_POINTS;
  86         tracefile = DEF_TRACEFILE;
  87         while ((ch = getopt(argc,argv,"aCcdf:g:ip:t:")) != EOF)
  88                 switch((char)ch) {
  89                 case 'a':
  90                         append = 1;
  91                         break;
  92                 case 'C':
  93                         clear = CLEARALL;
  94                         pidset = 1;
  95                         break;
  96                 case 'c':
  97                         clear = CLEAR;
  98                         break;
  99                 case 'd':
 100                         ops |= KTRFLAG_DESCEND;
 101                         break;
 102                 case 'f':
 103                         tracefile = optarg;
 104                         break;
 105                 case 'g':
 106                         pid = -rpid(optarg);
 107                         pidset = 1;
 108                         break;
 109                 case 'i':
 110                         inherit = 1;
 111                         break;
 112                 case 'p':
 113                         pid = rpid(optarg);
 114                         pidset = 1;
 115                         break;
 116                 case 't':
 117                         trpoints = getpoints(optarg);
 118                         if (trpoints < 0) {
 119                                 warnx("unknown facility in %s", optarg);
 120                                 usage();
 121                         }
 122                         break;
 123                 default:
 124                         usage();
 125                 }
 126         argv += optind;
 127         argc -= optind;
 128
 129         if (pidset && *argv || !pidset && !*argv)
 130                 usage();
 131
 132         if (inherit)
 133                 trpoints |= KTRFAC_INHERIT;
 134
 135         (void)signal(SIGSYS, no_ktrace);
 136         if (clear != NOTSET) {
 137                 if (clear == CLEARALL) {
 138                         ops = KTROP_CLEAR | KTRFLAG_DESCEND;
 139                         trpoints = ALL_POINTS;
 140                         pid = 1;
 141                 } else
 142                         ops |= pid ? KTROP_CLEAR : KTROP_CLEARFILE;
 143
 144                 if (ktrace(tracefile, ops, trpoints, pid) < 0)
 145                         err(1, tracefile);
 146                 exit(0);
 147         }
 148
 149         if ((fd = open(tracefile, O_CREAT | O_WRONLY | (append ? 0 : O_TRUNC),
 150             DEFFILEMODE)) < 0)
 151                 err(1, tracefile);
 152         (void)close(fd);
 153
 154         if (*argv) {
 155                 if (ktrace(tracefile, ops, trpoints, getpid()) < 0)
 156                         err(1, tracefile);
 157                 execvp(argv[0], &argv[0]);
 158                 err(1, "exec of '%s' failed", argv[0]);
 159         }
 160         else if (ktrace(tracefile, ops, trpoints, pid) < 0)
 161                 err(1, tracefile);
 162         exit(0);
 163 }
 164
 165 rpid(p)
 166         char *p;
 167 {
 168         static int first;
 169
 170         if (first++) {
 171                 warnx("only one -g or -p flag is permitted.");
 172                 usage();
 173         }
 174         if (!*p) {
 175                 warnx("illegal process id.");
 176                 usage();
 177         }
 178         return(atoi(p));
 179 }
 180
 181 void
 182 usage()
 183 {
 184         (void)fprintf(stderr,
 185 "usage:\tktrace [-aCcid] [-f trfile] [-g pgid] [-p pid] [-t [acgn]\n\tktrace [-aCcid] [-f trfile] [-t [acgn] command\n");
 186         exit(1);
 187 }
 188
 189 void
 190 no_ktrace(sig)
 191         int sig;
 192 {
 193         (void)fprintf(stderr,
 194 "error:\tktrace() system call not supported in the running kernel\n\tre-compile kernel with 'options KTRACE'\n");
 195         exit(1);
 196 }