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