kern_ktrace.c revision 1.33 1 1.33 mycroft /* $NetBSD: kern_ktrace.c,v 1.33 1998/09/11 12:50:10 mycroft Exp $ */
2 1.11 cgd
3 1.1 cgd /*
4 1.9 cgd * Copyright (c) 1989, 1993
5 1.9 cgd * The Regents of the University of California. All rights reserved.
6 1.1 cgd *
7 1.1 cgd * Redistribution and use in source and binary forms, with or without
8 1.1 cgd * modification, are permitted provided that the following conditions
9 1.1 cgd * are met:
10 1.1 cgd * 1. Redistributions of source code must retain the above copyright
11 1.1 cgd * notice, this list of conditions and the following disclaimer.
12 1.1 cgd * 2. Redistributions in binary form must reproduce the above copyright
13 1.1 cgd * notice, this list of conditions and the following disclaimer in the
14 1.1 cgd * documentation and/or other materials provided with the distribution.
15 1.1 cgd * 3. All advertising materials mentioning features or use of this software
16 1.1 cgd * must display the following acknowledgement:
17 1.1 cgd * This product includes software developed by the University of
18 1.1 cgd * California, Berkeley and its contributors.
19 1.1 cgd * 4. Neither the name of the University nor the names of its contributors
20 1.1 cgd * may be used to endorse or promote products derived from this software
21 1.1 cgd * without specific prior written permission.
22 1.1 cgd *
23 1.1 cgd * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
24 1.1 cgd * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25 1.1 cgd * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26 1.1 cgd * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
27 1.1 cgd * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28 1.1 cgd * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29 1.1 cgd * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30 1.1 cgd * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31 1.1 cgd * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32 1.1 cgd * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33 1.1 cgd * SUCH DAMAGE.
34 1.1 cgd *
35 1.25 fvdl * @(#)kern_ktrace.c 8.5 (Berkeley) 5/14/95
36 1.1 cgd */
37 1.29 thorpej
38 1.29 thorpej #include "opt_ktrace.h"
39 1.1 cgd
40 1.9 cgd #ifdef KTRACE
41 1.9 cgd
42 1.7 mycroft #include <sys/param.h>
43 1.13 cgd #include <sys/systm.h>
44 1.7 mycroft #include <sys/proc.h>
45 1.7 mycroft #include <sys/file.h>
46 1.7 mycroft #include <sys/namei.h>
47 1.7 mycroft #include <sys/vnode.h>
48 1.7 mycroft #include <sys/ktrace.h>
49 1.7 mycroft #include <sys/malloc.h>
50 1.7 mycroft #include <sys/syslog.h>
51 1.28 christos #include <sys/filedesc.h>
52 1.1 cgd
53 1.13 cgd #include <sys/mount.h>
54 1.13 cgd #include <sys/syscallargs.h>
55 1.22 christos
56 1.22 christos struct ktr_header *ktrgetheader __P((int));
57 1.28 christos int ktrops __P((struct proc *, struct proc *, int, int, void *));
58 1.28 christos int ktrsetchildren __P((struct proc *, struct proc *, int, int, void *));
59 1.28 christos void ktrwrite __P((struct proc *, void *, struct ktr_header *));
60 1.22 christos int ktrcanset __P((struct proc *, struct proc *));
61 1.22 christos
62 1.28 christos void
63 1.28 christos ktrderef(p)
64 1.28 christos struct proc *p;
65 1.28 christos {
66 1.28 christos if (p->p_tracep == NULL)
67 1.28 christos return;
68 1.28 christos
69 1.28 christos if (p->p_traceflag & KTRFAC_FD) {
70 1.28 christos struct file *fp = p->p_tracep;
71 1.28 christos
72 1.28 christos closef(fp, NULL);
73 1.28 christos } else {
74 1.28 christos struct vnode *vp = p->p_tracep;
75 1.28 christos
76 1.28 christos vrele(vp);
77 1.28 christos }
78 1.28 christos p->p_tracep = NULL;
79 1.28 christos p->p_traceflag = 0;
80 1.28 christos }
81 1.28 christos
82 1.28 christos void
83 1.28 christos ktradref(p)
84 1.28 christos struct proc *p;
85 1.28 christos {
86 1.28 christos if (p->p_traceflag & KTRFAC_FD) {
87 1.28 christos struct file *fp = p->p_tracep;
88 1.28 christos
89 1.28 christos fp->f_count++;
90 1.28 christos } else {
91 1.28 christos struct vnode *vp = p->p_tracep;
92 1.28 christos
93 1.28 christos VREF(vp);
94 1.28 christos }
95 1.28 christos }
96 1.28 christos
97 1.1 cgd struct ktr_header *
98 1.1 cgd ktrgetheader(type)
99 1.4 andrew int type;
100 1.1 cgd {
101 1.28 christos struct ktr_header *kth;
102 1.1 cgd struct proc *p = curproc; /* XXX */
103 1.1 cgd
104 1.31 perry MALLOC(kth, struct ktr_header *, sizeof(struct ktr_header),
105 1.1 cgd M_TEMP, M_WAITOK);
106 1.1 cgd kth->ktr_type = type;
107 1.1 cgd microtime(&kth->ktr_time);
108 1.1 cgd kth->ktr_pid = p->p_pid;
109 1.32 perry memcpy(kth->ktr_comm, p->p_comm, MAXCOMLEN);
110 1.1 cgd return (kth);
111 1.1 cgd }
112 1.1 cgd
113 1.17 cgd void
114 1.28 christos ktrsyscall(v, code, argsize, args)
115 1.28 christos void *v;
116 1.16 mycroft register_t code;
117 1.16 mycroft size_t argsize;
118 1.16 mycroft register_t args[];
119 1.1 cgd {
120 1.9 cgd struct ktr_header *kth;
121 1.1 cgd struct ktr_syscall *ktp;
122 1.9 cgd struct proc *p = curproc; /* XXX */
123 1.17 cgd register_t *argp;
124 1.26 mrg int len = sizeof(struct ktr_syscall) + argsize;
125 1.17 cgd int i;
126 1.1 cgd
127 1.9 cgd p->p_traceflag |= KTRFAC_ACTIVE;
128 1.9 cgd kth = ktrgetheader(KTR_SYSCALL);
129 1.1 cgd MALLOC(ktp, struct ktr_syscall *, len, M_TEMP, M_WAITOK);
130 1.1 cgd ktp->ktr_code = code;
131 1.17 cgd ktp->ktr_argsize = argsize;
132 1.17 cgd argp = (register_t *)((char *)ktp + sizeof(struct ktr_syscall));
133 1.31 perry for (i = 0; i < (argsize / sizeof(*argp)); i++)
134 1.1 cgd *argp++ = args[i];
135 1.1 cgd kth->ktr_buf = (caddr_t)ktp;
136 1.1 cgd kth->ktr_len = len;
137 1.28 christos ktrwrite(p, v, kth);
138 1.1 cgd FREE(ktp, M_TEMP);
139 1.1 cgd FREE(kth, M_TEMP);
140 1.9 cgd p->p_traceflag &= ~KTRFAC_ACTIVE;
141 1.1 cgd }
142 1.1 cgd
143 1.17 cgd void
144 1.28 christos ktrsysret(v, code, error, retval)
145 1.28 christos void *v;
146 1.16 mycroft register_t code;
147 1.16 mycroft int error;
148 1.16 mycroft register_t retval;
149 1.1 cgd {
150 1.9 cgd struct ktr_header *kth;
151 1.1 cgd struct ktr_sysret ktp;
152 1.9 cgd struct proc *p = curproc; /* XXX */
153 1.1 cgd
154 1.9 cgd p->p_traceflag |= KTRFAC_ACTIVE;
155 1.9 cgd kth = ktrgetheader(KTR_SYSRET);
156 1.1 cgd ktp.ktr_code = code;
157 1.1 cgd ktp.ktr_error = error;
158 1.1 cgd ktp.ktr_retval = retval; /* what about val2 ? */
159 1.1 cgd
160 1.1 cgd kth->ktr_buf = (caddr_t)&ktp;
161 1.1 cgd kth->ktr_len = sizeof(struct ktr_sysret);
162 1.1 cgd
163 1.28 christos ktrwrite(p, v, kth);
164 1.1 cgd FREE(kth, M_TEMP);
165 1.9 cgd p->p_traceflag &= ~KTRFAC_ACTIVE;
166 1.1 cgd }
167 1.1 cgd
168 1.17 cgd void
169 1.28 christos ktrnamei(v, path)
170 1.28 christos void *v;
171 1.1 cgd char *path;
172 1.1 cgd {
173 1.9 cgd struct ktr_header *kth;
174 1.9 cgd struct proc *p = curproc; /* XXX */
175 1.1 cgd
176 1.9 cgd p->p_traceflag |= KTRFAC_ACTIVE;
177 1.9 cgd kth = ktrgetheader(KTR_NAMEI);
178 1.1 cgd kth->ktr_len = strlen(path);
179 1.1 cgd kth->ktr_buf = path;
180 1.18 christos
181 1.28 christos ktrwrite(p, v, kth);
182 1.18 christos FREE(kth, M_TEMP);
183 1.18 christos p->p_traceflag &= ~KTRFAC_ACTIVE;
184 1.18 christos }
185 1.18 christos
186 1.18 christos void
187 1.28 christos ktremul(v, p, emul)
188 1.28 christos void *v;
189 1.28 christos struct proc *p;
190 1.18 christos char *emul;
191 1.18 christos {
192 1.18 christos struct ktr_header *kth;
193 1.18 christos
194 1.18 christos p->p_traceflag |= KTRFAC_ACTIVE;
195 1.18 christos kth = ktrgetheader(KTR_EMUL);
196 1.18 christos kth->ktr_len = strlen(emul);
197 1.18 christos kth->ktr_buf = emul;
198 1.1 cgd
199 1.28 christos ktrwrite(p, v, kth);
200 1.1 cgd FREE(kth, M_TEMP);
201 1.9 cgd p->p_traceflag &= ~KTRFAC_ACTIVE;
202 1.1 cgd }
203 1.1 cgd
204 1.17 cgd void
205 1.28 christos ktrgenio(v, fd, rw, iov, len, error)
206 1.28 christos void *v;
207 1.1 cgd int fd;
208 1.1 cgd enum uio_rw rw;
209 1.28 christos struct iovec *iov;
210 1.4 andrew int len, error;
211 1.1 cgd {
212 1.9 cgd struct ktr_header *kth;
213 1.28 christos struct ktr_genio *ktp;
214 1.28 christos caddr_t cp;
215 1.28 christos int resid = len, cnt;
216 1.9 cgd struct proc *p = curproc; /* XXX */
217 1.1 cgd
218 1.1 cgd if (error)
219 1.1 cgd return;
220 1.9 cgd p->p_traceflag |= KTRFAC_ACTIVE;
221 1.9 cgd kth = ktrgetheader(KTR_GENIO);
222 1.1 cgd MALLOC(ktp, struct ktr_genio *, sizeof(struct ktr_genio) + len,
223 1.1 cgd M_TEMP, M_WAITOK);
224 1.1 cgd ktp->ktr_fd = fd;
225 1.1 cgd ktp->ktr_rw = rw;
226 1.31 perry cp = (caddr_t)((char *)ktp + sizeof(struct ktr_genio));
227 1.1 cgd while (resid > 0) {
228 1.1 cgd if ((cnt = iov->iov_len) > resid)
229 1.1 cgd cnt = resid;
230 1.1 cgd if (copyin(iov->iov_base, cp, (unsigned)cnt))
231 1.1 cgd goto done;
232 1.1 cgd cp += cnt;
233 1.1 cgd resid -= cnt;
234 1.1 cgd iov++;
235 1.1 cgd }
236 1.1 cgd kth->ktr_buf = (caddr_t)ktp;
237 1.31 perry kth->ktr_len = sizeof(struct ktr_genio) + len;
238 1.1 cgd
239 1.28 christos ktrwrite(p, v, kth);
240 1.1 cgd done:
241 1.1 cgd FREE(kth, M_TEMP);
242 1.1 cgd FREE(ktp, M_TEMP);
243 1.9 cgd p->p_traceflag &= ~KTRFAC_ACTIVE;
244 1.1 cgd }
245 1.1 cgd
246 1.17 cgd void
247 1.28 christos ktrpsig(v, sig, action, mask, code)
248 1.28 christos void *v;
249 1.9 cgd int sig;
250 1.9 cgd sig_t action;
251 1.33 mycroft sigset_t *mask;
252 1.33 mycroft int code;
253 1.1 cgd {
254 1.9 cgd struct ktr_header *kth;
255 1.1 cgd struct ktr_psig kp;
256 1.9 cgd struct proc *p = curproc; /* XXX */
257 1.1 cgd
258 1.9 cgd p->p_traceflag |= KTRFAC_ACTIVE;
259 1.9 cgd kth = ktrgetheader(KTR_PSIG);
260 1.1 cgd kp.signo = (char)sig;
261 1.1 cgd kp.action = action;
262 1.33 mycroft kp.mask = *mask;
263 1.1 cgd kp.code = code;
264 1.1 cgd kth->ktr_buf = (caddr_t)&kp;
265 1.31 perry kth->ktr_len = sizeof(struct ktr_psig);
266 1.1 cgd
267 1.28 christos ktrwrite(p, v, kth);
268 1.1 cgd FREE(kth, M_TEMP);
269 1.9 cgd p->p_traceflag &= ~KTRFAC_ACTIVE;
270 1.9 cgd }
271 1.9 cgd
272 1.17 cgd void
273 1.28 christos ktrcsw(v, out, user)
274 1.28 christos void *v;
275 1.9 cgd int out, user;
276 1.9 cgd {
277 1.9 cgd struct ktr_header *kth;
278 1.9 cgd struct ktr_csw kc;
279 1.9 cgd struct proc *p = curproc; /* XXX */
280 1.9 cgd
281 1.9 cgd p->p_traceflag |= KTRFAC_ACTIVE;
282 1.9 cgd kth = ktrgetheader(KTR_CSW);
283 1.9 cgd kc.out = out;
284 1.9 cgd kc.user = user;
285 1.9 cgd kth->ktr_buf = (caddr_t)&kc;
286 1.31 perry kth->ktr_len = sizeof(struct ktr_csw);
287 1.9 cgd
288 1.28 christos ktrwrite(p, v, kth);
289 1.9 cgd FREE(kth, M_TEMP);
290 1.9 cgd p->p_traceflag &= ~KTRFAC_ACTIVE;
291 1.1 cgd }
292 1.1 cgd
293 1.1 cgd /* Interface and common routines */
294 1.1 cgd
295 1.1 cgd /*
296 1.1 cgd * ktrace system call
297 1.1 cgd */
298 1.1 cgd /* ARGSUSED */
299 1.17 cgd int
300 1.28 christos sys_fktrace(curp, v, retval)
301 1.28 christos struct proc *curp;
302 1.28 christos void *v;
303 1.28 christos register_t *retval;
304 1.28 christos {
305 1.28 christos struct sys_fktrace_args /* {
306 1.28 christos syscallarg(int) fd;
307 1.28 christos syscallarg(int) ops;
308 1.28 christos syscallarg(int) facs;
309 1.28 christos syscallarg(int) pid;
310 1.28 christos } */ *uap = v;
311 1.28 christos struct file *fp = NULL;
312 1.28 christos struct proc *p;
313 1.28 christos struct filedesc *fdp = curp->p_fd;
314 1.28 christos struct pgrp *pg;
315 1.28 christos int facs;
316 1.28 christos int ops;
317 1.28 christos int descend;
318 1.28 christos int ret = 0;
319 1.28 christos int error = 0;
320 1.28 christos
321 1.28 christos if (((u_int)SCARG(uap, fd)) >= fdp->fd_nfiles ||
322 1.28 christos (fp = fdp->fd_ofiles[SCARG(uap, fd)]) == NULL ||
323 1.28 christos (fp->f_flag & FWRITE) == 0)
324 1.28 christos return (EBADF);
325 1.28 christos
326 1.28 christos ops = KTROP(SCARG(uap, ops)) | KTRFLAG_FD;
327 1.28 christos descend = SCARG(uap, ops) & KTRFLAG_DESCEND;
328 1.28 christos facs = SCARG(uap, facs) & ~((unsigned) KTRFAC_ROOT);
329 1.28 christos curp->p_traceflag |= KTRFAC_ACTIVE;
330 1.28 christos
331 1.28 christos /*
332 1.28 christos * Clear all uses of the tracefile
333 1.28 christos */
334 1.28 christos if (KTROP(ops) == KTROP_CLEARFILE) {
335 1.28 christos for (p = allproc.lh_first; p != 0; p = p->p_list.le_next) {
336 1.28 christos if (p->p_tracep == fp) {
337 1.28 christos if (ktrcanset(curp, p))
338 1.28 christos ktrderef(p);
339 1.28 christos else
340 1.28 christos error = EPERM;
341 1.28 christos }
342 1.28 christos }
343 1.28 christos goto done;
344 1.28 christos }
345 1.28 christos /*
346 1.28 christos * need something to (un)trace (XXX - why is this here?)
347 1.28 christos */
348 1.28 christos if (!facs) {
349 1.28 christos error = EINVAL;
350 1.28 christos goto done;
351 1.28 christos }
352 1.28 christos /*
353 1.28 christos * do it
354 1.28 christos */
355 1.28 christos if (SCARG(uap, pid) < 0) {
356 1.28 christos /*
357 1.28 christos * by process group
358 1.28 christos */
359 1.28 christos pg = pgfind(-SCARG(uap, pid));
360 1.28 christos if (pg == NULL) {
361 1.28 christos error = ESRCH;
362 1.28 christos goto done;
363 1.28 christos }
364 1.28 christos for (p = pg->pg_members.lh_first; p != 0; p = p->p_pglist.le_next)
365 1.28 christos if (descend)
366 1.28 christos ret |= ktrsetchildren(curp, p, ops, facs, fp);
367 1.28 christos else
368 1.28 christos ret |= ktrops(curp, p, ops, facs, fp);
369 1.28 christos
370 1.28 christos } else {
371 1.28 christos /*
372 1.28 christos * by pid
373 1.28 christos */
374 1.28 christos p = pfind(SCARG(uap, pid));
375 1.28 christos if (p == NULL) {
376 1.28 christos error = ESRCH;
377 1.28 christos goto done;
378 1.28 christos }
379 1.28 christos if (descend)
380 1.28 christos ret |= ktrsetchildren(curp, p, ops, facs, fp);
381 1.28 christos else
382 1.28 christos ret |= ktrops(curp, p, ops, facs, fp);
383 1.28 christos }
384 1.28 christos if (!ret)
385 1.28 christos error = EPERM;
386 1.28 christos done:
387 1.28 christos curp->p_traceflag &= ~KTRFAC_ACTIVE;
388 1.28 christos return (error);
389 1.28 christos }
390 1.28 christos
391 1.28 christos /*
392 1.28 christos * ktrace system call
393 1.28 christos */
394 1.28 christos /* ARGSUSED */
395 1.28 christos int
396 1.20 mycroft sys_ktrace(curp, v, retval)
397 1.1 cgd struct proc *curp;
398 1.19 thorpej void *v;
399 1.19 thorpej register_t *retval;
400 1.19 thorpej {
401 1.28 christos struct sys_ktrace_args /* {
402 1.24 mycroft syscallarg(const char *) fname;
403 1.13 cgd syscallarg(int) ops;
404 1.13 cgd syscallarg(int) facs;
405 1.13 cgd syscallarg(int) pid;
406 1.19 thorpej } */ *uap = v;
407 1.28 christos struct vnode *vp = NULL;
408 1.28 christos struct proc *p;
409 1.1 cgd struct pgrp *pg;
410 1.22 christos int facs = SCARG(uap, facs) & ~((unsigned) KTRFAC_ROOT);
411 1.13 cgd int ops = KTROP(SCARG(uap, ops));
412 1.13 cgd int descend = SCARG(uap, ops) & KTRFLAG_DESCEND;
413 1.1 cgd int ret = 0;
414 1.1 cgd int error = 0;
415 1.1 cgd struct nameidata nd;
416 1.1 cgd
417 1.9 cgd curp->p_traceflag |= KTRFAC_ACTIVE;
418 1.1 cgd if (ops != KTROP_CLEAR) {
419 1.1 cgd /*
420 1.1 cgd * an operation which requires a file argument.
421 1.1 cgd */
422 1.13 cgd NDINIT(&nd, LOOKUP, FOLLOW, UIO_USERSPACE, SCARG(uap, fname),
423 1.13 cgd curp);
424 1.22 christos if ((error = vn_open(&nd, FREAD|FWRITE, 0)) != 0) {
425 1.9 cgd curp->p_traceflag &= ~KTRFAC_ACTIVE;
426 1.1 cgd return (error);
427 1.9 cgd }
428 1.1 cgd vp = nd.ni_vp;
429 1.25 fvdl VOP_UNLOCK(vp, 0);
430 1.1 cgd if (vp->v_type != VREG) {
431 1.1 cgd (void) vn_close(vp, FREAD|FWRITE, curp->p_ucred, curp);
432 1.9 cgd curp->p_traceflag &= ~KTRFAC_ACTIVE;
433 1.1 cgd return (EACCES);
434 1.1 cgd }
435 1.1 cgd }
436 1.1 cgd /*
437 1.1 cgd * Clear all uses of the tracefile
438 1.1 cgd */
439 1.28 christos if (KTROP(ops) == KTROP_CLEARFILE) {
440 1.12 mycroft for (p = allproc.lh_first; p != 0; p = p->p_list.le_next) {
441 1.27 mycroft if (p->p_tracep == vp &&
442 1.27 mycroft !ktrops(curp, p, KTROP_CLEAR, ~0, vp))
443 1.27 mycroft error = EPERM;
444 1.1 cgd }
445 1.1 cgd goto done;
446 1.1 cgd }
447 1.1 cgd /*
448 1.1 cgd * need something to (un)trace (XXX - why is this here?)
449 1.1 cgd */
450 1.1 cgd if (!facs) {
451 1.1 cgd error = EINVAL;
452 1.1 cgd goto done;
453 1.1 cgd }
454 1.1 cgd /*
455 1.1 cgd * do it
456 1.1 cgd */
457 1.13 cgd if (SCARG(uap, pid) < 0) {
458 1.1 cgd /*
459 1.1 cgd * by process group
460 1.1 cgd */
461 1.13 cgd pg = pgfind(-SCARG(uap, pid));
462 1.1 cgd if (pg == NULL) {
463 1.1 cgd error = ESRCH;
464 1.1 cgd goto done;
465 1.1 cgd }
466 1.12 mycroft for (p = pg->pg_members.lh_first; p != 0; p = p->p_pglist.le_next)
467 1.1 cgd if (descend)
468 1.1 cgd ret |= ktrsetchildren(curp, p, ops, facs, vp);
469 1.1 cgd else
470 1.1 cgd ret |= ktrops(curp, p, ops, facs, vp);
471 1.1 cgd
472 1.1 cgd } else {
473 1.1 cgd /*
474 1.1 cgd * by pid
475 1.1 cgd */
476 1.13 cgd p = pfind(SCARG(uap, pid));
477 1.1 cgd if (p == NULL) {
478 1.1 cgd error = ESRCH;
479 1.1 cgd goto done;
480 1.1 cgd }
481 1.1 cgd if (descend)
482 1.1 cgd ret |= ktrsetchildren(curp, p, ops, facs, vp);
483 1.1 cgd else
484 1.1 cgd ret |= ktrops(curp, p, ops, facs, vp);
485 1.1 cgd }
486 1.1 cgd if (!ret)
487 1.1 cgd error = EPERM;
488 1.1 cgd done:
489 1.1 cgd if (vp != NULL)
490 1.1 cgd (void) vn_close(vp, FWRITE, curp->p_ucred, curp);
491 1.9 cgd curp->p_traceflag &= ~KTRFAC_ACTIVE;
492 1.1 cgd return (error);
493 1.1 cgd }
494 1.1 cgd
495 1.4 andrew int
496 1.28 christos ktrops(curp, p, ops, facs, v)
497 1.9 cgd struct proc *p, *curp;
498 1.4 andrew int ops, facs;
499 1.28 christos void *v;
500 1.1 cgd {
501 1.1 cgd
502 1.1 cgd if (!ktrcanset(curp, p))
503 1.1 cgd return (0);
504 1.28 christos if (KTROP(ops) == KTROP_SET) {
505 1.28 christos if (p->p_tracep != v) {
506 1.1 cgd /*
507 1.1 cgd * if trace file already in use, relinquish
508 1.1 cgd */
509 1.28 christos ktrderef(p);
510 1.28 christos if (ops & KTRFLAG_FD)
511 1.28 christos p->p_traceflag = KTRFAC_FD;
512 1.28 christos p->p_tracep = v;
513 1.28 christos ktradref(p);
514 1.1 cgd }
515 1.1 cgd p->p_traceflag |= facs;
516 1.1 cgd if (curp->p_ucred->cr_uid == 0)
517 1.1 cgd p->p_traceflag |= KTRFAC_ROOT;
518 1.1 cgd } else {
519 1.1 cgd /* KTROP_CLEAR */
520 1.1 cgd if (((p->p_traceflag &= ~facs) & KTRFAC_MASK) == 0) {
521 1.1 cgd /* no more tracing */
522 1.28 christos ktrderef(p);
523 1.1 cgd }
524 1.1 cgd }
525 1.21 christos
526 1.21 christos /*
527 1.21 christos * Emit an emulation record, every time there is a ktrace
528 1.21 christos * change/attach request.
529 1.21 christos */
530 1.21 christos if (KTRPOINT(p, KTR_EMUL))
531 1.28 christos ktremul(p->p_tracep, p, p->p_emul->e_name);
532 1.1 cgd
533 1.1 cgd return (1);
534 1.1 cgd }
535 1.1 cgd
536 1.22 christos int
537 1.28 christos ktrsetchildren(curp, top, ops, facs, v)
538 1.1 cgd struct proc *curp, *top;
539 1.4 andrew int ops, facs;
540 1.28 christos void *v;
541 1.1 cgd {
542 1.28 christos struct proc *p;
543 1.28 christos int ret = 0;
544 1.1 cgd
545 1.1 cgd p = top;
546 1.1 cgd for (;;) {
547 1.28 christos ret |= ktrops(curp, p, ops, facs, v);
548 1.1 cgd /*
549 1.1 cgd * If this process has children, descend to them next,
550 1.1 cgd * otherwise do any siblings, and if done with this level,
551 1.1 cgd * follow back up the tree (but not past top).
552 1.1 cgd */
553 1.12 mycroft if (p->p_children.lh_first)
554 1.12 mycroft p = p->p_children.lh_first;
555 1.1 cgd else for (;;) {
556 1.1 cgd if (p == top)
557 1.1 cgd return (ret);
558 1.12 mycroft if (p->p_sibling.le_next) {
559 1.12 mycroft p = p->p_sibling.le_next;
560 1.1 cgd break;
561 1.1 cgd }
562 1.12 mycroft p = p->p_pptr;
563 1.1 cgd }
564 1.1 cgd }
565 1.1 cgd /*NOTREACHED*/
566 1.1 cgd }
567 1.1 cgd
568 1.22 christos void
569 1.28 christos ktrwrite(p, v, kth)
570 1.28 christos struct proc *p;
571 1.28 christos void *v;
572 1.28 christos struct ktr_header *kth;
573 1.1 cgd {
574 1.1 cgd struct uio auio;
575 1.1 cgd struct iovec aiov[2];
576 1.1 cgd int error;
577 1.1 cgd
578 1.28 christos if (v == NULL)
579 1.1 cgd return;
580 1.1 cgd auio.uio_iov = &aiov[0];
581 1.1 cgd auio.uio_offset = 0;
582 1.1 cgd auio.uio_segflg = UIO_SYSSPACE;
583 1.1 cgd auio.uio_rw = UIO_WRITE;
584 1.1 cgd aiov[0].iov_base = (caddr_t)kth;
585 1.1 cgd aiov[0].iov_len = sizeof(struct ktr_header);
586 1.1 cgd auio.uio_resid = sizeof(struct ktr_header);
587 1.1 cgd auio.uio_iovcnt = 1;
588 1.1 cgd auio.uio_procp = (struct proc *)0;
589 1.1 cgd if (kth->ktr_len > 0) {
590 1.1 cgd auio.uio_iovcnt++;
591 1.1 cgd aiov[1].iov_base = kth->ktr_buf;
592 1.1 cgd aiov[1].iov_len = kth->ktr_len;
593 1.1 cgd auio.uio_resid += kth->ktr_len;
594 1.1 cgd }
595 1.28 christos if (p->p_traceflag & KTRFAC_FD) {
596 1.28 christos struct file *fp = v;
597 1.28 christos
598 1.30 thorpej error = (*fp->f_ops->fo_write)(fp, &fp->f_offset, &auio,
599 1.30 thorpej fp->f_cred, FOF_UPDATE_OFFSET);
600 1.28 christos }
601 1.28 christos else {
602 1.28 christos struct vnode *vp = v;
603 1.28 christos
604 1.28 christos vn_lock(vp, LK_EXCLUSIVE | LK_RETRY);
605 1.28 christos error = VOP_WRITE(vp, &auio, IO_UNIT|IO_APPEND, p->p_ucred);
606 1.28 christos VOP_UNLOCK(vp, 0);
607 1.28 christos }
608 1.1 cgd if (!error)
609 1.1 cgd return;
610 1.1 cgd /*
611 1.1 cgd * If error encountered, give up tracing on this vnode.
612 1.1 cgd */
613 1.1 cgd log(LOG_NOTICE, "ktrace write failed, errno %d, tracing stopped\n",
614 1.1 cgd error);
615 1.12 mycroft for (p = allproc.lh_first; p != 0; p = p->p_list.le_next) {
616 1.28 christos if (p->p_tracep == v)
617 1.28 christos ktrderef(p);
618 1.1 cgd }
619 1.1 cgd }
620 1.1 cgd
621 1.1 cgd /*
622 1.1 cgd * Return true if caller has permission to set the ktracing state
623 1.1 cgd * of target. Essentially, the target can't possess any
624 1.1 cgd * more permissions than the caller. KTRFAC_ROOT signifies that
625 1.1 cgd * root previously set the tracing status on the target process, and
626 1.1 cgd * so, only root may further change it.
627 1.1 cgd *
628 1.1 cgd * TODO: check groups. use caller effective gid.
629 1.1 cgd */
630 1.22 christos int
631 1.1 cgd ktrcanset(callp, targetp)
632 1.1 cgd struct proc *callp, *targetp;
633 1.1 cgd {
634 1.28 christos struct pcred *caller = callp->p_cred;
635 1.28 christos struct pcred *target = targetp->p_cred;
636 1.1 cgd
637 1.1 cgd if ((caller->pc_ucred->cr_uid == target->p_ruid &&
638 1.1 cgd target->p_ruid == target->p_svuid &&
639 1.1 cgd caller->p_rgid == target->p_rgid && /* XXX */
640 1.1 cgd target->p_rgid == target->p_svgid &&
641 1.1 cgd (targetp->p_traceflag & KTRFAC_ROOT) == 0) ||
642 1.1 cgd caller->pc_ucred->cr_uid == 0)
643 1.1 cgd return (1);
644 1.1 cgd
645 1.1 cgd return (0);
646 1.1 cgd }
647 1.9 cgd
648 1.9 cgd #endif
649