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