kern_ktrace.c revision 1.156.2.1 1 /* $NetBSD: kern_ktrace.c,v 1.156.2.1 2011/06/23 14:20:19 cherry Exp $ */
2
3 /*-
4 * Copyright (c) 2006, 2007, 2008 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * This code is derived from software contributed to The NetBSD Foundation
8 * by Andrew Doran.
9 *
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
13 * 1. Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 * POSSIBILITY OF SUCH DAMAGE.
30 */
31
32 /*
33 * Copyright (c) 1989, 1993
34 * The Regents of the University of California. All rights reserved.
35 *
36 * Redistribution and use in source and binary forms, with or without
37 * modification, are permitted provided that the following conditions
38 * are met:
39 * 1. Redistributions of source code must retain the above copyright
40 * notice, this list of conditions and the following disclaimer.
41 * 2. Redistributions in binary form must reproduce the above copyright
42 * notice, this list of conditions and the following disclaimer in the
43 * documentation and/or other materials provided with the distribution.
44 * 3. Neither the name of the University nor the names of its contributors
45 * may be used to endorse or promote products derived from this software
46 * without specific prior written permission.
47 *
48 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
49 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
50 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
51 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
52 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
53 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
54 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
55 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
56 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
57 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
58 * SUCH DAMAGE.
59 *
60 * @(#)kern_ktrace.c 8.5 (Berkeley) 5/14/95
61 */
62
63 #include <sys/cdefs.h>
64 __KERNEL_RCSID(0, "$NetBSD: kern_ktrace.c,v 1.156.2.1 2011/06/23 14:20:19 cherry Exp $");
65
66 #include <sys/param.h>
67 #include <sys/systm.h>
68 #include <sys/proc.h>
69 #include <sys/file.h>
70 #include <sys/namei.h>
71 #include <sys/vnode.h>
72 #include <sys/kernel.h>
73 #include <sys/kthread.h>
74 #include <sys/ktrace.h>
75 #include <sys/kmem.h>
76 #include <sys/syslog.h>
77 #include <sys/filedesc.h>
78 #include <sys/ioctl.h>
79 #include <sys/callout.h>
80 #include <sys/kauth.h>
81
82 #include <sys/mount.h>
83 #include <sys/sa.h>
84 #include <sys/syscallargs.h>
85
86 /*
87 * TODO:
88 * - need better error reporting?
89 * - userland utility to sort ktrace.out by timestamp.
90 * - keep minimum information in ktrace_entry when rest of alloc failed.
91 * - per trace control of configurable parameters.
92 */
93
94 struct ktrace_entry {
95 TAILQ_ENTRY(ktrace_entry) kte_list;
96 struct ktr_header kte_kth;
97 void *kte_buf;
98 size_t kte_bufsz;
99 #define KTE_SPACE 32
100 uint8_t kte_space[KTE_SPACE];
101 };
102
103 struct ktr_desc {
104 TAILQ_ENTRY(ktr_desc) ktd_list;
105 int ktd_flags;
106 #define KTDF_WAIT 0x0001
107 #define KTDF_DONE 0x0002
108 #define KTDF_BLOCKING 0x0004
109 #define KTDF_INTERACTIVE 0x0008
110 int ktd_error;
111 #define KTDE_ENOMEM 0x0001
112 #define KTDE_ENOSPC 0x0002
113 int ktd_errcnt;
114 int ktd_ref; /* # of reference */
115 int ktd_qcount; /* # of entry in the queue */
116
117 /*
118 * Params to control behaviour.
119 */
120 int ktd_delayqcnt; /* # of entry allowed to delay */
121 int ktd_wakedelay; /* delay of wakeup in *tick* */
122 int ktd_intrwakdl; /* ditto, but when interactive */
123
124 file_t *ktd_fp; /* trace output file */
125 lwp_t *ktd_lwp; /* our kernel thread */
126 TAILQ_HEAD(, ktrace_entry) ktd_queue;
127 callout_t ktd_wakch; /* delayed wakeup */
128 kcondvar_t ktd_sync_cv;
129 kcondvar_t ktd_cv;
130 };
131
132 static int ktealloc(struct ktrace_entry **, void **, lwp_t *, int,
133 size_t);
134 static void ktrwrite(struct ktr_desc *, struct ktrace_entry *);
135 static int ktrace_common(lwp_t *, int, int, int, file_t *);
136 static int ktrops(lwp_t *, struct proc *, int, int,
137 struct ktr_desc *);
138 static int ktrsetchildren(lwp_t *, struct proc *, int, int,
139 struct ktr_desc *);
140 static int ktrcanset(lwp_t *, struct proc *);
141 static int ktrsamefile(file_t *, file_t *);
142 static void ktr_kmem(lwp_t *, int, const void *, size_t);
143 static void ktr_io(lwp_t *, int, enum uio_rw, struct iovec *, size_t);
144
145 static struct ktr_desc *
146 ktd_lookup(file_t *);
147 static void ktdrel(struct ktr_desc *);
148 static void ktdref(struct ktr_desc *);
149 static void ktraddentry(lwp_t *, struct ktrace_entry *, int);
150 /* Flags for ktraddentry (3rd arg) */
151 #define KTA_NOWAIT 0x0000
152 #define KTA_WAITOK 0x0001
153 #define KTA_LARGE 0x0002
154 static void ktefree(struct ktrace_entry *);
155 static void ktd_logerrl(struct ktr_desc *, int);
156 static void ktrace_thread(void *);
157 static int ktrderefall(struct ktr_desc *, int);
158
159 /*
160 * Default vaules.
161 */
162 #define KTD_MAXENTRY 1000 /* XXX: tune */
163 #define KTD_TIMEOUT 5 /* XXX: tune */
164 #define KTD_DELAYQCNT 100 /* XXX: tune */
165 #define KTD_WAKEDELAY 5000 /* XXX: tune */
166 #define KTD_INTRWAKDL 100 /* XXX: tune */
167
168 /*
169 * Patchable variables.
170 */
171 int ktd_maxentry = KTD_MAXENTRY; /* max # of entry in the queue */
172 int ktd_timeout = KTD_TIMEOUT; /* timeout in seconds */
173 int ktd_delayqcnt = KTD_DELAYQCNT; /* # of entry allowed to delay */
174 int ktd_wakedelay = KTD_WAKEDELAY; /* delay of wakeup in *ms* */
175 int ktd_intrwakdl = KTD_INTRWAKDL; /* ditto, but when interactive */
176
177 kmutex_t ktrace_lock;
178 int ktrace_on;
179 static TAILQ_HEAD(, ktr_desc) ktdq = TAILQ_HEAD_INITIALIZER(ktdq);
180 static pool_cache_t kte_cache;
181
182 static kauth_listener_t ktrace_listener;
183
184 static void
185 ktd_wakeup(struct ktr_desc *ktd)
186 {
187
188 callout_stop(&ktd->ktd_wakch);
189 cv_signal(&ktd->ktd_cv);
190 }
191
192 static void
193 ktd_callout(void *arg)
194 {
195
196 mutex_enter(&ktrace_lock);
197 ktd_wakeup(arg);
198 mutex_exit(&ktrace_lock);
199 }
200
201 static void
202 ktd_logerrl(struct ktr_desc *ktd, int error)
203 {
204
205 ktd->ktd_error |= error;
206 ktd->ktd_errcnt++;
207 }
208
209 #if 0
210 static void
211 ktd_logerr(struct proc *p, int error)
212 {
213 struct ktr_desc *ktd;
214
215 KASSERT(mutex_owned(&ktrace_lock));
216
217 ktd = p->p_tracep;
218 if (ktd == NULL)
219 return;
220
221 ktd_logerrl(ktd, error);
222 }
223 #endif
224
225 static inline int
226 ktrenter(lwp_t *l)
227 {
228
229 if ((l->l_pflag & LP_KTRACTIVE) != 0)
230 return 1;
231 l->l_pflag |= LP_KTRACTIVE;
232 return 0;
233 }
234
235 static inline void
236 ktrexit(lwp_t *l)
237 {
238
239 l->l_pflag &= ~LP_KTRACTIVE;
240 }
241
242 static int
243 ktrace_listener_cb(kauth_cred_t cred, kauth_action_t action, void *cookie,
244 void *arg0, void *arg1, void *arg2, void *arg3)
245 {
246 struct proc *p;
247 int result;
248 enum kauth_process_req req;
249
250 result = KAUTH_RESULT_DEFER;
251 p = arg0;
252
253 if (action != KAUTH_PROCESS_KTRACE)
254 return result;
255
256 req = (enum kauth_process_req)(unsigned long)arg1;
257
258 /* Privileged; secmodel should handle these. */
259 if (req == KAUTH_REQ_PROCESS_KTRACE_PERSISTENT)
260 return result;
261
262 if ((p->p_traceflag & KTRFAC_PERSISTENT) ||
263 (p->p_flag & PK_SUGID))
264 return result;
265
266 if (kauth_cred_geteuid(cred) == kauth_cred_getuid(p->p_cred) &&
267 kauth_cred_getuid(cred) == kauth_cred_getsvuid(p->p_cred) &&
268 kauth_cred_getgid(cred) == kauth_cred_getgid(p->p_cred) &&
269 kauth_cred_getgid(cred) == kauth_cred_getsvgid(p->p_cred))
270 result = KAUTH_RESULT_ALLOW;
271
272 return result;
273 }
274
275 /*
276 * Initialise the ktrace system.
277 */
278 void
279 ktrinit(void)
280 {
281
282 mutex_init(&ktrace_lock, MUTEX_DEFAULT, IPL_NONE);
283 kte_cache = pool_cache_init(sizeof(struct ktrace_entry), 0, 0, 0,
284 "ktrace", &pool_allocator_nointr, IPL_NONE, NULL, NULL, NULL);
285
286 ktrace_listener = kauth_listen_scope(KAUTH_SCOPE_PROCESS,
287 ktrace_listener_cb, NULL);
288 }
289
290 /*
291 * Release a reference. Called with ktrace_lock held.
292 */
293 void
294 ktdrel(struct ktr_desc *ktd)
295 {
296
297 KASSERT(mutex_owned(&ktrace_lock));
298
299 KDASSERT(ktd->ktd_ref != 0);
300 KASSERT(ktd->ktd_ref > 0);
301 KASSERT(ktrace_on > 0);
302 ktrace_on--;
303 if (--ktd->ktd_ref <= 0) {
304 ktd->ktd_flags |= KTDF_DONE;
305 cv_signal(&ktd->ktd_cv);
306 }
307 }
308
309 void
310 ktdref(struct ktr_desc *ktd)
311 {
312
313 KASSERT(mutex_owned(&ktrace_lock));
314
315 ktd->ktd_ref++;
316 ktrace_on++;
317 }
318
319 struct ktr_desc *
320 ktd_lookup(file_t *fp)
321 {
322 struct ktr_desc *ktd;
323
324 KASSERT(mutex_owned(&ktrace_lock));
325
326 for (ktd = TAILQ_FIRST(&ktdq); ktd != NULL;
327 ktd = TAILQ_NEXT(ktd, ktd_list)) {
328 if (ktrsamefile(ktd->ktd_fp, fp)) {
329 ktdref(ktd);
330 break;
331 }
332 }
333
334 return (ktd);
335 }
336
337 void
338 ktraddentry(lwp_t *l, struct ktrace_entry *kte, int flags)
339 {
340 struct proc *p = l->l_proc;
341 struct ktr_desc *ktd;
342 #ifdef DEBUG
343 struct timeval t1, t2;
344 #endif
345
346 mutex_enter(&ktrace_lock);
347
348 if (p->p_traceflag & KTRFAC_TRC_EMUL) {
349 /* Add emulation trace before first entry for this process */
350 p->p_traceflag &= ~KTRFAC_TRC_EMUL;
351 mutex_exit(&ktrace_lock);
352 ktrexit(l);
353 ktremul();
354 (void)ktrenter(l);
355 mutex_enter(&ktrace_lock);
356 }
357
358 /* Tracing may have been cancelled. */
359 ktd = p->p_tracep;
360 if (ktd == NULL)
361 goto freekte;
362
363 /*
364 * Bump reference count so that the object will remain while
365 * we are here. Note that the trace is controlled by other
366 * process.
367 */
368 ktdref(ktd);
369
370 if (ktd->ktd_flags & KTDF_DONE)
371 goto relktd;
372
373 if (ktd->ktd_qcount > ktd_maxentry) {
374 ktd_logerrl(ktd, KTDE_ENOSPC);
375 goto relktd;
376 }
377 TAILQ_INSERT_TAIL(&ktd->ktd_queue, kte, kte_list);
378 ktd->ktd_qcount++;
379 if (ktd->ktd_flags & KTDF_BLOCKING)
380 goto skip_sync;
381
382 if (flags & KTA_WAITOK &&
383 (/* flags & KTA_LARGE */0 || ktd->ktd_flags & KTDF_WAIT ||
384 ktd->ktd_qcount > ktd_maxentry >> 1))
385 /*
386 * Sync with writer thread since we're requesting rather
387 * big one or many requests are pending.
388 */
389 do {
390 ktd->ktd_flags |= KTDF_WAIT;
391 ktd_wakeup(ktd);
392 #ifdef DEBUG
393 getmicrouptime(&t1);
394 #endif
395 if (cv_timedwait(&ktd->ktd_sync_cv, &ktrace_lock,
396 ktd_timeout * hz) != 0) {
397 ktd->ktd_flags |= KTDF_BLOCKING;
398 /*
399 * Maybe the writer thread is blocking
400 * completely for some reason, but
401 * don't stop target process forever.
402 */
403 log(LOG_NOTICE, "ktrace timeout\n");
404 break;
405 }
406 #ifdef DEBUG
407 getmicrouptime(&t2);
408 timersub(&t2, &t1, &t2);
409 if (t2.tv_sec > 0)
410 log(LOG_NOTICE,
411 "ktrace long wait: %lld.%06ld\n",
412 (long long)t2.tv_sec, (long)t2.tv_usec);
413 #endif
414 } while (p->p_tracep == ktd &&
415 (ktd->ktd_flags & (KTDF_WAIT | KTDF_DONE)) == KTDF_WAIT);
416 else {
417 /* Schedule delayed wakeup */
418 if (ktd->ktd_qcount > ktd->ktd_delayqcnt)
419 ktd_wakeup(ktd); /* Wakeup now */
420 else if (!callout_pending(&ktd->ktd_wakch))
421 callout_reset(&ktd->ktd_wakch,
422 ktd->ktd_flags & KTDF_INTERACTIVE ?
423 ktd->ktd_intrwakdl : ktd->ktd_wakedelay,
424 ktd_callout, ktd);
425 }
426
427 skip_sync:
428 ktdrel(ktd);
429 mutex_exit(&ktrace_lock);
430 ktrexit(l);
431 return;
432
433 relktd:
434 ktdrel(ktd);
435
436 freekte:
437 mutex_exit(&ktrace_lock);
438 ktefree(kte);
439 ktrexit(l);
440 }
441
442 void
443 ktefree(struct ktrace_entry *kte)
444 {
445
446 if (kte->kte_buf != kte->kte_space)
447 kmem_free(kte->kte_buf, kte->kte_bufsz);
448 pool_cache_put(kte_cache, kte);
449 }
450
451 /*
452 * "deep" compare of two files for the purposes of clearing a trace.
453 * Returns true if they're the same open file, or if they point at the
454 * same underlying vnode/socket.
455 */
456
457 int
458 ktrsamefile(file_t *f1, file_t *f2)
459 {
460
461 return ((f1 == f2) ||
462 ((f1 != NULL) && (f2 != NULL) &&
463 (f1->f_type == f2->f_type) &&
464 (f1->f_data == f2->f_data)));
465 }
466
467 void
468 ktrderef(struct proc *p)
469 {
470 struct ktr_desc *ktd = p->p_tracep;
471
472 KASSERT(mutex_owned(&ktrace_lock));
473
474 p->p_traceflag = 0;
475 if (ktd == NULL)
476 return;
477 p->p_tracep = NULL;
478
479 cv_broadcast(&ktd->ktd_sync_cv);
480 ktdrel(ktd);
481 }
482
483 void
484 ktradref(struct proc *p)
485 {
486 struct ktr_desc *ktd = p->p_tracep;
487
488 KASSERT(mutex_owned(&ktrace_lock));
489
490 ktdref(ktd);
491 }
492
493 int
494 ktrderefall(struct ktr_desc *ktd, int auth)
495 {
496 lwp_t *curl = curlwp;
497 struct proc *p;
498 int error = 0;
499
500 mutex_enter(proc_lock);
501 PROCLIST_FOREACH(p, &allproc) {
502 if (p->p_tracep != ktd)
503 continue;
504 mutex_enter(p->p_lock);
505 mutex_enter(&ktrace_lock);
506 if (p->p_tracep == ktd) {
507 if (!auth || ktrcanset(curl, p))
508 ktrderef(p);
509 else
510 error = EPERM;
511 }
512 mutex_exit(&ktrace_lock);
513 mutex_exit(p->p_lock);
514 }
515 mutex_exit(proc_lock);
516
517 return error;
518 }
519
520 int
521 ktealloc(struct ktrace_entry **ktep, void **bufp, lwp_t *l, int type,
522 size_t sz)
523 {
524 struct proc *p = l->l_proc;
525 struct ktrace_entry *kte;
526 struct ktr_header *kth;
527 void *buf;
528
529 if (ktrenter(l))
530 return EAGAIN;
531
532 kte = pool_cache_get(kte_cache, PR_WAITOK);
533 if (sz > sizeof(kte->kte_space)) {
534 if ((buf = kmem_alloc(sz, KM_SLEEP)) == NULL) {
535 pool_cache_put(kte_cache, kte);
536 ktrexit(l);
537 return ENOMEM;
538 }
539 } else
540 buf = kte->kte_space;
541
542 kte->kte_bufsz = sz;
543 kte->kte_buf = buf;
544
545 kth = &kte->kte_kth;
546 (void)memset(kth, 0, sizeof(*kth));
547 kth->ktr_len = sz;
548 kth->ktr_type = type;
549 kth->ktr_pid = p->p_pid;
550 memcpy(kth->ktr_comm, p->p_comm, MAXCOMLEN);
551 kth->ktr_version = KTRFAC_VERSION(p->p_traceflag);
552 kth->ktr_lid = l->l_lid;
553 nanotime(&kth->ktr_ts);
554
555 *ktep = kte;
556 *bufp = buf;
557
558 return 0;
559 }
560
561 void
562 ktr_syscall(register_t code, const register_t args[], int narg)
563 {
564 lwp_t *l = curlwp;
565 struct proc *p = l->l_proc;
566 struct ktrace_entry *kte;
567 struct ktr_syscall *ktp;
568 register_t *argp;
569 size_t len;
570 u_int i;
571
572 if (!KTRPOINT(p, KTR_SYSCALL))
573 return;
574
575 len = sizeof(struct ktr_syscall) + narg * sizeof argp[0];
576
577 if (ktealloc(&kte, (void *)&ktp, l, KTR_SYSCALL, len))
578 return;
579
580 ktp->ktr_code = code;
581 ktp->ktr_argsize = narg * sizeof argp[0];
582 argp = (register_t *)(ktp + 1);
583 for (i = 0; i < narg; i++)
584 *argp++ = args[i];
585
586 ktraddentry(l, kte, KTA_WAITOK);
587 }
588
589 void
590 ktr_sysret(register_t code, int error, register_t *retval)
591 {
592 lwp_t *l = curlwp;
593 struct ktrace_entry *kte;
594 struct ktr_sysret *ktp;
595
596 if (!KTRPOINT(l->l_proc, KTR_SYSRET))
597 return;
598
599 if (ktealloc(&kte, (void *)&ktp, l, KTR_SYSRET,
600 sizeof(struct ktr_sysret)))
601 return;
602
603 ktp->ktr_code = code;
604 ktp->ktr_eosys = 0; /* XXX unused */
605 ktp->ktr_error = error;
606 ktp->ktr_retval = retval ? retval[0] : 0;
607 ktp->ktr_retval_1 = retval ? retval[1] : 0;
608
609 ktraddentry(l, kte, KTA_WAITOK);
610 }
611
612 void
613 ktr_namei(const char *path, size_t pathlen)
614 {
615 lwp_t *l = curlwp;
616
617 if (!KTRPOINT(l->l_proc, KTR_NAMEI))
618 return;
619
620 ktr_kmem(l, KTR_NAMEI, path, pathlen);
621 }
622
623 void
624 ktr_namei2(const char *eroot, size_t erootlen,
625 const char *path, size_t pathlen)
626 {
627 lwp_t *l = curlwp;
628 struct ktrace_entry *kte;
629 void *buf;
630
631 if (!KTRPOINT(l->l_proc, KTR_NAMEI))
632 return;
633
634 if (ktealloc(&kte, &buf, l, KTR_NAMEI, erootlen + pathlen))
635 return;
636 memcpy(buf, eroot, erootlen);
637 buf = (char *)buf + erootlen;
638 memcpy(buf, path, pathlen);
639 ktraddentry(l, kte, KTA_WAITOK);
640 }
641
642 void
643 ktr_emul(void)
644 {
645 lwp_t *l = curlwp;
646 const char *emul = l->l_proc->p_emul->e_name;
647
648 if (!KTRPOINT(l->l_proc, KTR_EMUL))
649 return;
650
651 ktr_kmem(l, KTR_EMUL, emul, strlen(emul));
652 }
653
654 void
655 ktr_execarg(const void *bf, size_t len)
656 {
657 lwp_t *l = curlwp;
658
659 if (!KTRPOINT(l->l_proc, KTR_EXEC_ARG))
660 return;
661
662 ktr_kmem(l, KTR_EXEC_ARG, bf, len);
663 }
664
665 void
666 ktr_execenv(const void *bf, size_t len)
667 {
668 lwp_t *l = curlwp;
669
670 if (!KTRPOINT(l->l_proc, KTR_EXEC_ENV))
671 return;
672
673 ktr_kmem(l, KTR_EXEC_ENV, bf, len);
674 }
675
676 void
677 ktr_execfd(int fd, u_int dtype)
678 {
679 struct ktrace_entry *kte;
680 struct ktr_execfd* ktp;
681
682 lwp_t *l = curlwp;
683
684 if (!KTRPOINT(l->l_proc, KTR_EXEC_FD))
685 return;
686
687 if (ktealloc(&kte, (void *)&ktp, l, KTR_EXEC_FD, sizeof(*ktp)))
688 return;
689
690 ktp->ktr_fd = fd;
691 ktp->ktr_dtype = dtype;
692 ktraddentry(l, kte, KTA_WAITOK);
693 }
694
695 static void
696 ktr_kmem(lwp_t *l, int type, const void *bf, size_t len)
697 {
698 struct ktrace_entry *kte;
699 void *buf;
700
701 if (ktealloc(&kte, &buf, l, type, len))
702 return;
703 memcpy(buf, bf, len);
704 ktraddentry(l, kte, KTA_WAITOK);
705 }
706
707 static void
708 ktr_io(lwp_t *l, int fd, enum uio_rw rw, struct iovec *iov, size_t len)
709 {
710 struct ktrace_entry *kte;
711 struct ktr_genio *ktp;
712 size_t resid = len, cnt, buflen;
713 char *cp;
714
715 next:
716 buflen = min(PAGE_SIZE, resid + sizeof(struct ktr_genio));
717
718 if (ktealloc(&kte, (void *)&ktp, l, KTR_GENIO, buflen))
719 return;
720
721 ktp->ktr_fd = fd;
722 ktp->ktr_rw = rw;
723
724 cp = (void *)(ktp + 1);
725 buflen -= sizeof(struct ktr_genio);
726 kte->kte_kth.ktr_len = sizeof(struct ktr_genio);
727
728 while (buflen > 0) {
729 cnt = min(iov->iov_len, buflen);
730 if (copyin(iov->iov_base, cp, cnt) != 0)
731 goto out;
732 kte->kte_kth.ktr_len += cnt;
733 cp += cnt;
734 buflen -= cnt;
735 resid -= cnt;
736 iov->iov_len -= cnt;
737 if (iov->iov_len == 0)
738 iov++;
739 else
740 iov->iov_base = (char *)iov->iov_base + cnt;
741 }
742
743 /*
744 * Don't push so many entry at once. It will cause kmem map
745 * shortage.
746 */
747 ktraddentry(l, kte, KTA_WAITOK | KTA_LARGE);
748 if (resid > 0) {
749 if (curcpu()->ci_schedstate.spc_flags & SPCF_SHOULDYIELD) {
750 (void)ktrenter(l);
751 preempt();
752 ktrexit(l);
753 }
754
755 goto next;
756 }
757
758 return;
759
760 out:
761 ktefree(kte);
762 ktrexit(l);
763 }
764
765 void
766 ktr_genio(int fd, enum uio_rw rw, const void *addr, size_t len, int error)
767 {
768 lwp_t *l = curlwp;
769 struct iovec iov;
770
771 if (!KTRPOINT(l->l_proc, KTR_GENIO) || error != 0)
772 return;
773 iov.iov_base = __UNCONST(addr);
774 iov.iov_len = len;
775 ktr_io(l, fd, rw, &iov, len);
776 }
777
778 void
779 ktr_geniov(int fd, enum uio_rw rw, struct iovec *iov, size_t len, int error)
780 {
781 lwp_t *l = curlwp;
782
783 if (!KTRPOINT(l->l_proc, KTR_GENIO) || error != 0)
784 return;
785 ktr_io(l, fd, rw, iov, len);
786 }
787
788 void
789 ktr_mibio(int fd, enum uio_rw rw, const void *addr, size_t len, int error)
790 {
791 lwp_t *l = curlwp;
792 struct iovec iov;
793
794 if (!KTRPOINT(l->l_proc, KTR_MIB) || error != 0)
795 return;
796 iov.iov_base = __UNCONST(addr);
797 iov.iov_len = len;
798 ktr_io(l, fd, rw, &iov, len);
799 }
800
801 void
802 ktr_psig(int sig, sig_t action, const sigset_t *mask,
803 const ksiginfo_t *ksi)
804 {
805 struct ktrace_entry *kte;
806 lwp_t *l = curlwp;
807 struct {
808 struct ktr_psig kp;
809 siginfo_t si;
810 } *kbuf;
811
812 if (!KTRPOINT(l->l_proc, KTR_PSIG))
813 return;
814
815 if (ktealloc(&kte, (void *)&kbuf, l, KTR_PSIG, sizeof(*kbuf)))
816 return;
817
818 kbuf->kp.signo = (char)sig;
819 kbuf->kp.action = action;
820 kbuf->kp.mask = *mask;
821
822 if (ksi) {
823 kbuf->kp.code = KSI_TRAPCODE(ksi);
824 (void)memset(&kbuf->si, 0, sizeof(kbuf->si));
825 kbuf->si._info = ksi->ksi_info;
826 kte->kte_kth.ktr_len = sizeof(*kbuf);
827 } else {
828 kbuf->kp.code = 0;
829 kte->kte_kth.ktr_len = sizeof(struct ktr_psig);
830 }
831
832 ktraddentry(l, kte, KTA_WAITOK);
833 }
834
835 void
836 ktr_csw(int out, int user)
837 {
838 lwp_t *l = curlwp;
839 struct proc *p = l->l_proc;
840 struct ktrace_entry *kte;
841 struct ktr_csw *kc;
842
843 if (!KTRPOINT(p, KTR_CSW))
844 return;
845
846 /*
847 * Don't record context switches resulting from blocking on
848 * locks; it's too easy to get duff results.
849 */
850 if (l->l_syncobj == &mutex_syncobj || l->l_syncobj == &rw_syncobj)
851 return;
852
853 /*
854 * We can't sleep if we're already going to sleep (if original
855 * condition is met during sleep, we hang up).
856 *
857 * XXX This is not ideal: it would be better to maintain a pool
858 * of ktes and actually push this to the kthread when context
859 * switch happens, however given the points where we are called
860 * from that is difficult to do.
861 */
862 if (out) {
863 if (ktrenter(l))
864 return;
865
866 nanotime(&l->l_ktrcsw);
867 l->l_pflag |= LP_KTRCSW;
868 if (user)
869 l->l_pflag |= LP_KTRCSWUSER;
870 else
871 l->l_pflag &= ~LP_KTRCSWUSER;
872
873 ktrexit(l);
874 return;
875 }
876
877 /*
878 * On the way back in, we need to record twice: once for entry, and
879 * once for exit.
880 */
881 if ((l->l_pflag & LP_KTRCSW) != 0) {
882 struct timespec *ts;
883 l->l_pflag &= ~LP_KTRCSW;
884
885 if (ktealloc(&kte, (void *)&kc, l, KTR_CSW, sizeof(*kc)))
886 return;
887
888 kc->out = 1;
889 kc->user = ((l->l_pflag & LP_KTRCSWUSER) != 0);
890
891 ts = &l->l_ktrcsw;
892 switch (KTRFAC_VERSION(p->p_traceflag)) {
893 case 0:
894 kte->kte_kth.ktr_otv.tv_sec = ts->tv_sec;
895 kte->kte_kth.ktr_otv.tv_usec = ts->tv_nsec / 1000;
896 break;
897 case 1:
898 kte->kte_kth.ktr_ots.tv_sec = ts->tv_sec;
899 kte->kte_kth.ktr_ots.tv_nsec = ts->tv_nsec;
900 break;
901 case 2:
902 kte->kte_kth.ktr_ts.tv_sec = ts->tv_sec;
903 kte->kte_kth.ktr_ts.tv_nsec = ts->tv_nsec;
904 break;
905 default:
906 break;
907 }
908
909 ktraddentry(l, kte, KTA_WAITOK);
910 }
911
912 if (ktealloc(&kte, (void *)&kc, l, KTR_CSW, sizeof(*kc)))
913 return;
914
915 kc->out = 0;
916 kc->user = user;
917
918 ktraddentry(l, kte, KTA_WAITOK);
919 }
920
921 bool
922 ktr_point(int fac_bit)
923 {
924 return curlwp->l_proc->p_traceflag & fac_bit;
925 }
926
927 int
928 ktruser(const char *id, void *addr, size_t len, int ustr)
929 {
930 struct ktrace_entry *kte;
931 struct ktr_user *ktp;
932 lwp_t *l = curlwp;
933 void *user_dta;
934 int error;
935
936 if (!KTRPOINT(l->l_proc, KTR_USER))
937 return 0;
938
939 if (len > KTR_USER_MAXLEN)
940 return ENOSPC;
941
942 error = ktealloc(&kte, (void *)&ktp, l, KTR_USER, sizeof(*ktp) + len);
943 if (error != 0)
944 return error;
945
946 if (ustr) {
947 if (copyinstr(id, ktp->ktr_id, KTR_USER_MAXIDLEN, NULL) != 0)
948 ktp->ktr_id[0] = '\0';
949 } else
950 strncpy(ktp->ktr_id, id, KTR_USER_MAXIDLEN);
951 ktp->ktr_id[KTR_USER_MAXIDLEN-1] = '\0';
952
953 user_dta = (void *)(ktp + 1);
954 if ((error = copyin(addr, (void *)user_dta, len)) != 0)
955 len = 0;
956
957 ktraddentry(l, kte, KTA_WAITOK);
958 return error;
959 }
960
961 void
962 ktr_kuser(const char *id, void *addr, size_t len)
963 {
964 struct ktrace_entry *kte;
965 struct ktr_user *ktp;
966 lwp_t *l = curlwp;
967 int error;
968
969 if (!KTRPOINT(l->l_proc, KTR_USER))
970 return;
971
972 if (len > KTR_USER_MAXLEN)
973 return;
974
975 error = ktealloc(&kte, (void *)&ktp, l, KTR_USER, sizeof(*ktp) + len);
976 if (error != 0)
977 return;
978
979 strlcpy(ktp->ktr_id, id, KTR_USER_MAXIDLEN);
980
981 memcpy(ktp + 1, addr, len);
982
983 ktraddentry(l, kte, KTA_WAITOK);
984 }
985
986 void
987 ktr_saupcall(struct lwp *l, int type, int nevent, int nint, void *sas,
988 void *ap, void *ksas)
989 {
990 struct ktrace_entry *kte;
991 struct ktr_saupcall *ktp;
992 size_t len, sz;
993 struct sa_t **sapp;
994 int i;
995
996 if (!KTRPOINT(l->l_proc, KTR_SAUPCALL))
997 return;
998
999 len = sizeof(struct ktr_saupcall);
1000 sz = len + sizeof(struct sa_t) * (nevent + nint + 1);
1001
1002 if (ktealloc(&kte, (void *)&ktp, l, KTR_SAUPCALL, sz))
1003 return;
1004
1005 ktp->ktr_type = type;
1006 ktp->ktr_nevent = nevent;
1007 ktp->ktr_nint = nint;
1008 ktp->ktr_sas = sas;
1009 ktp->ktr_ap = ap;
1010
1011 /* Copy the sa_t's */
1012 sapp = (struct sa_t **) ksas;
1013
1014 for (i = nevent + nint; i >= 0; i--) {
1015 memcpy((char *)ktp + len, *sapp, sizeof(struct sa_t));
1016 len += sizeof(struct sa_t);
1017 sapp++;
1018 }
1019
1020 kte->kte_kth.ktr_len = len;
1021 ktraddentry(l, kte, KTA_WAITOK);
1022 }
1023
1024 void
1025 ktr_mib(const int *name, u_int namelen)
1026 {
1027 struct ktrace_entry *kte;
1028 int *namep;
1029 size_t size;
1030 lwp_t *l = curlwp;
1031
1032 if (!KTRPOINT(l->l_proc, KTR_MIB))
1033 return;
1034
1035 size = namelen * sizeof(*name);
1036
1037 if (ktealloc(&kte, (void *)&namep, l, KTR_MIB, size))
1038 return;
1039
1040 (void)memcpy(namep, name, namelen * sizeof(*name));
1041
1042 ktraddentry(l, kte, KTA_WAITOK);
1043 }
1044
1045 /* Interface and common routines */
1046
1047 int
1048 ktrace_common(lwp_t *curl, int ops, int facs, int pid, file_t *fp)
1049 {
1050 struct proc *curp;
1051 struct proc *p;
1052 struct pgrp *pg;
1053 struct ktr_desc *ktd = NULL;
1054 int ret = 0;
1055 int error = 0;
1056 int descend;
1057
1058 curp = curl->l_proc;
1059 descend = ops & KTRFLAG_DESCEND;
1060 facs = facs & ~((unsigned) KTRFAC_PERSISTENT);
1061
1062 (void)ktrenter(curl);
1063
1064 switch (KTROP(ops)) {
1065
1066 case KTROP_CLEARFILE:
1067 /*
1068 * Clear all uses of the tracefile
1069 */
1070 mutex_enter(&ktrace_lock);
1071 ktd = ktd_lookup(fp);
1072 mutex_exit(&ktrace_lock);
1073 if (ktd == NULL)
1074 goto done;
1075 error = ktrderefall(ktd, 1);
1076 goto done;
1077
1078 case KTROP_SET:
1079 mutex_enter(&ktrace_lock);
1080 ktd = ktd_lookup(fp);
1081 mutex_exit(&ktrace_lock);
1082 if (ktd == NULL) {
1083 ktd = kmem_alloc(sizeof(*ktd), KM_SLEEP);
1084 TAILQ_INIT(&ktd->ktd_queue);
1085 callout_init(&ktd->ktd_wakch, CALLOUT_MPSAFE);
1086 cv_init(&ktd->ktd_cv, "ktrwait");
1087 cv_init(&ktd->ktd_sync_cv, "ktrsync");
1088 ktd->ktd_flags = 0;
1089 ktd->ktd_qcount = 0;
1090 ktd->ktd_error = 0;
1091 ktd->ktd_errcnt = 0;
1092 ktd->ktd_delayqcnt = ktd_delayqcnt;
1093 ktd->ktd_wakedelay = mstohz(ktd_wakedelay);
1094 ktd->ktd_intrwakdl = mstohz(ktd_intrwakdl);
1095 ktd->ktd_ref = 0;
1096 ktd->ktd_fp = fp;
1097 mutex_enter(&ktrace_lock);
1098 ktdref(ktd);
1099 mutex_exit(&ktrace_lock);
1100
1101 /*
1102 * XXX: not correct. needs an way to detect
1103 * whether ktruss or ktrace.
1104 */
1105 if (fp->f_type == DTYPE_PIPE)
1106 ktd->ktd_flags |= KTDF_INTERACTIVE;
1107
1108 mutex_enter(&fp->f_lock);
1109 fp->f_count++;
1110 mutex_exit(&fp->f_lock);
1111 error = kthread_create(PRI_NONE, KTHREAD_MPSAFE, NULL,
1112 ktrace_thread, ktd, &ktd->ktd_lwp, "ktrace");
1113 if (error != 0) {
1114 kmem_free(ktd, sizeof(*ktd));
1115 mutex_enter(&fp->f_lock);
1116 fp->f_count--;
1117 mutex_exit(&fp->f_lock);
1118 goto done;
1119 }
1120
1121 mutex_enter(&ktrace_lock);
1122 if (ktd_lookup(fp) != NULL) {
1123 ktdrel(ktd);
1124 ktd = NULL;
1125 } else
1126 TAILQ_INSERT_TAIL(&ktdq, ktd, ktd_list);
1127 if (ktd == NULL)
1128 cv_wait(&lbolt, &ktrace_lock);
1129 mutex_exit(&ktrace_lock);
1130 if (ktd == NULL)
1131 goto done;
1132 }
1133 break;
1134
1135 case KTROP_CLEAR:
1136 break;
1137 }
1138
1139 /*
1140 * need something to (un)trace (XXX - why is this here?)
1141 */
1142 if (!facs) {
1143 error = EINVAL;
1144 goto done;
1145 }
1146
1147 /*
1148 * do it
1149 */
1150 mutex_enter(proc_lock);
1151 if (pid < 0) {
1152 /*
1153 * by process group
1154 */
1155 pg = pgrp_find(-pid);
1156 if (pg == NULL)
1157 error = ESRCH;
1158 else {
1159 LIST_FOREACH(p, &pg->pg_members, p_pglist) {
1160 if (descend)
1161 ret |= ktrsetchildren(curl, p, ops,
1162 facs, ktd);
1163 else
1164 ret |= ktrops(curl, p, ops, facs,
1165 ktd);
1166 }
1167 }
1168
1169 } else {
1170 /*
1171 * by pid
1172 */
1173 p = proc_find(pid);
1174 if (p == NULL)
1175 error = ESRCH;
1176 else if (descend)
1177 ret |= ktrsetchildren(curl, p, ops, facs, ktd);
1178 else
1179 ret |= ktrops(curl, p, ops, facs, ktd);
1180 }
1181 mutex_exit(proc_lock);
1182 if (error == 0 && !ret)
1183 error = EPERM;
1184 done:
1185 if (ktd != NULL) {
1186 mutex_enter(&ktrace_lock);
1187 if (error != 0) {
1188 /*
1189 * Wakeup the thread so that it can be die if we
1190 * can't trace any process.
1191 */
1192 ktd_wakeup(ktd);
1193 }
1194 if (KTROP(ops) == KTROP_SET || KTROP(ops) == KTROP_CLEARFILE)
1195 ktdrel(ktd);
1196 mutex_exit(&ktrace_lock);
1197 }
1198 ktrexit(curl);
1199 return (error);
1200 }
1201
1202 /*
1203 * fktrace system call
1204 */
1205 /* ARGSUSED */
1206 int
1207 sys_fktrace(struct lwp *l, const struct sys_fktrace_args *uap, register_t *retval)
1208 {
1209 /* {
1210 syscallarg(int) fd;
1211 syscallarg(int) ops;
1212 syscallarg(int) facs;
1213 syscallarg(int) pid;
1214 } */
1215 file_t *fp;
1216 int error, fd;
1217
1218 fd = SCARG(uap, fd);
1219 if ((fp = fd_getfile(fd)) == NULL)
1220 return (EBADF);
1221 if ((fp->f_flag & FWRITE) == 0)
1222 error = EBADF;
1223 else
1224 error = ktrace_common(l, SCARG(uap, ops),
1225 SCARG(uap, facs), SCARG(uap, pid), fp);
1226 fd_putfile(fd);
1227 return error;
1228 }
1229
1230 /*
1231 * ktrace system call
1232 */
1233 /* ARGSUSED */
1234 int
1235 sys_ktrace(struct lwp *l, const struct sys_ktrace_args *uap, register_t *retval)
1236 {
1237 /* {
1238 syscallarg(const char *) fname;
1239 syscallarg(int) ops;
1240 syscallarg(int) facs;
1241 syscallarg(int) pid;
1242 } */
1243 struct vnode *vp = NULL;
1244 file_t *fp = NULL;
1245 struct pathbuf *pb;
1246 struct nameidata nd;
1247 int error = 0;
1248 int fd;
1249
1250 if (ktrenter(l))
1251 return EAGAIN;
1252
1253 if (KTROP(SCARG(uap, ops)) != KTROP_CLEAR) {
1254 /*
1255 * an operation which requires a file argument.
1256 */
1257 error = pathbuf_copyin(SCARG(uap, fname), &pb);
1258 if (error) {
1259 ktrexit(l);
1260 return (error);
1261 }
1262 NDINIT(&nd, LOOKUP, FOLLOW, pb);
1263 if ((error = vn_open(&nd, FREAD|FWRITE, 0)) != 0) {
1264 pathbuf_destroy(pb);
1265 ktrexit(l);
1266 return (error);
1267 }
1268 vp = nd.ni_vp;
1269 pathbuf_destroy(pb);
1270 VOP_UNLOCK(vp);
1271 if (vp->v_type != VREG) {
1272 vn_close(vp, FREAD|FWRITE, l->l_cred);
1273 ktrexit(l);
1274 return (EACCES);
1275 }
1276 /*
1277 * This uses up a file descriptor slot in the
1278 * tracing process for the duration of this syscall.
1279 * This is not expected to be a problem.
1280 */
1281 if ((error = fd_allocfile(&fp, &fd)) != 0) {
1282 vn_close(vp, FWRITE, l->l_cred);
1283 ktrexit(l);
1284 return error;
1285 }
1286 fp->f_flag = FWRITE;
1287 fp->f_type = DTYPE_VNODE;
1288 fp->f_ops = &vnops;
1289 fp->f_data = (void *)vp;
1290 vp = NULL;
1291 }
1292 error = ktrace_common(l, SCARG(uap, ops), SCARG(uap, facs),
1293 SCARG(uap, pid), fp);
1294 if (fp != NULL) {
1295 if (error != 0) {
1296 /* File unused. */
1297 fd_abort(curproc, fp, fd);
1298 } else {
1299 /* File was used. */
1300 fd_abort(curproc, NULL, fd);
1301 }
1302 }
1303 return (error);
1304 }
1305
1306 int
1307 ktrops(lwp_t *curl, struct proc *p, int ops, int facs,
1308 struct ktr_desc *ktd)
1309 {
1310 int vers = ops & KTRFAC_VER_MASK;
1311 int error = 0;
1312
1313 mutex_enter(p->p_lock);
1314 mutex_enter(&ktrace_lock);
1315
1316 if (!ktrcanset(curl, p))
1317 goto out;
1318
1319 switch (vers) {
1320 case KTRFACv0:
1321 case KTRFACv1:
1322 case KTRFACv2:
1323 break;
1324 default:
1325 error = EINVAL;
1326 goto out;
1327 }
1328
1329 if (KTROP(ops) == KTROP_SET) {
1330 if (p->p_tracep != ktd) {
1331 /*
1332 * if trace file already in use, relinquish
1333 */
1334 ktrderef(p);
1335 p->p_tracep = ktd;
1336 ktradref(p);
1337 }
1338 p->p_traceflag |= facs;
1339 if (kauth_authorize_process(curl->l_cred, KAUTH_PROCESS_KTRACE,
1340 p, KAUTH_ARG(KAUTH_REQ_PROCESS_KTRACE_PERSISTENT), NULL,
1341 NULL) == 0)
1342 p->p_traceflag |= KTRFAC_PERSISTENT;
1343 } else {
1344 /* KTROP_CLEAR */
1345 if (((p->p_traceflag &= ~facs) & KTRFAC_MASK) == 0) {
1346 /* no more tracing */
1347 ktrderef(p);
1348 }
1349 }
1350
1351 if (p->p_traceflag)
1352 p->p_traceflag |= vers;
1353 /*
1354 * Emit an emulation record, every time there is a ktrace
1355 * change/attach request.
1356 */
1357 if (KTRPOINT(p, KTR_EMUL))
1358 p->p_traceflag |= KTRFAC_TRC_EMUL;
1359
1360 p->p_trace_enabled = trace_is_enabled(p);
1361 #ifdef __HAVE_SYSCALL_INTERN
1362 (*p->p_emul->e_syscall_intern)(p);
1363 #endif
1364
1365 out:
1366 mutex_exit(&ktrace_lock);
1367 mutex_exit(p->p_lock);
1368
1369 return (1);
1370 }
1371
1372 int
1373 ktrsetchildren(lwp_t *curl, struct proc *top, int ops, int facs,
1374 struct ktr_desc *ktd)
1375 {
1376 struct proc *p;
1377 int ret = 0;
1378
1379 KASSERT(mutex_owned(proc_lock));
1380
1381 p = top;
1382 for (;;) {
1383 ret |= ktrops(curl, p, ops, facs, ktd);
1384 /*
1385 * If this process has children, descend to them next,
1386 * otherwise do any siblings, and if done with this level,
1387 * follow back up the tree (but not past top).
1388 */
1389 if (LIST_FIRST(&p->p_children) != NULL) {
1390 p = LIST_FIRST(&p->p_children);
1391 continue;
1392 }
1393 for (;;) {
1394 if (p == top)
1395 return (ret);
1396 if (LIST_NEXT(p, p_sibling) != NULL) {
1397 p = LIST_NEXT(p, p_sibling);
1398 break;
1399 }
1400 p = p->p_pptr;
1401 }
1402 }
1403 /*NOTREACHED*/
1404 }
1405
1406 void
1407 ktrwrite(struct ktr_desc *ktd, struct ktrace_entry *kte)
1408 {
1409 size_t hlen;
1410 struct uio auio;
1411 struct iovec aiov[64], *iov;
1412 struct ktrace_entry *top = kte;
1413 struct ktr_header *kth;
1414 file_t *fp = ktd->ktd_fp;
1415 int error;
1416 next:
1417 auio.uio_iov = iov = &aiov[0];
1418 auio.uio_offset = 0;
1419 auio.uio_rw = UIO_WRITE;
1420 auio.uio_resid = 0;
1421 auio.uio_iovcnt = 0;
1422 UIO_SETUP_SYSSPACE(&auio);
1423 do {
1424 struct timespec ts;
1425 lwpid_t lid;
1426 kth = &kte->kte_kth;
1427
1428 hlen = sizeof(struct ktr_header);
1429 switch (kth->ktr_version) {
1430 case 0:
1431 ts = kth->ktr_time;
1432
1433 kth->ktr_otv.tv_sec = ts.tv_sec;
1434 kth->ktr_otv.tv_usec = ts.tv_nsec / 1000;
1435 kth->ktr_unused = NULL;
1436 hlen -= sizeof(kth->_v) -
1437 MAX(sizeof(kth->_v._v0), sizeof(kth->_v._v1));
1438 break;
1439 case 1:
1440 ts = kth->ktr_time;
1441 lid = kth->ktr_lid;
1442
1443 kth->ktr_ots.tv_sec = ts.tv_sec;
1444 kth->ktr_ots.tv_nsec = ts.tv_nsec;
1445 kth->ktr_olid = lid;
1446 hlen -= sizeof(kth->_v) -
1447 MAX(sizeof(kth->_v._v0), sizeof(kth->_v._v1));
1448 break;
1449 }
1450 iov->iov_base = (void *)kth;
1451 iov++->iov_len = hlen;
1452 auio.uio_resid += hlen;
1453 auio.uio_iovcnt++;
1454 if (kth->ktr_len > 0) {
1455 iov->iov_base = kte->kte_buf;
1456 iov++->iov_len = kth->ktr_len;
1457 auio.uio_resid += kth->ktr_len;
1458 auio.uio_iovcnt++;
1459 }
1460 } while ((kte = TAILQ_NEXT(kte, kte_list)) != NULL &&
1461 auio.uio_iovcnt < sizeof(aiov) / sizeof(aiov[0]) - 1);
1462
1463 again:
1464 error = (*fp->f_ops->fo_write)(fp, &fp->f_offset, &auio,
1465 fp->f_cred, FOF_UPDATE_OFFSET);
1466 switch (error) {
1467
1468 case 0:
1469 if (auio.uio_resid > 0)
1470 goto again;
1471 if (kte != NULL)
1472 goto next;
1473 break;
1474
1475 case EWOULDBLOCK:
1476 kpause("ktrzzz", false, 1, NULL);
1477 goto again;
1478
1479 default:
1480 /*
1481 * If error encountered, give up tracing on this
1482 * vnode. Don't report EPIPE as this can easily
1483 * happen with fktrace()/ktruss.
1484 */
1485 #ifndef DEBUG
1486 if (error != EPIPE)
1487 #endif
1488 log(LOG_NOTICE,
1489 "ktrace write failed, errno %d, tracing stopped\n",
1490 error);
1491 (void)ktrderefall(ktd, 0);
1492 }
1493
1494 while ((kte = top) != NULL) {
1495 top = TAILQ_NEXT(top, kte_list);
1496 ktefree(kte);
1497 }
1498 }
1499
1500 void
1501 ktrace_thread(void *arg)
1502 {
1503 struct ktr_desc *ktd = arg;
1504 file_t *fp = ktd->ktd_fp;
1505 struct ktrace_entry *kte;
1506 int ktrerr, errcnt;
1507
1508 mutex_enter(&ktrace_lock);
1509 for (;;) {
1510 kte = TAILQ_FIRST(&ktd->ktd_queue);
1511 if (kte == NULL) {
1512 if (ktd->ktd_flags & KTDF_WAIT) {
1513 ktd->ktd_flags &= ~(KTDF_WAIT | KTDF_BLOCKING);
1514 cv_broadcast(&ktd->ktd_sync_cv);
1515 }
1516 if (ktd->ktd_ref == 0)
1517 break;
1518 cv_wait(&ktd->ktd_cv, &ktrace_lock);
1519 continue;
1520 }
1521 TAILQ_INIT(&ktd->ktd_queue);
1522 ktd->ktd_qcount = 0;
1523 ktrerr = ktd->ktd_error;
1524 errcnt = ktd->ktd_errcnt;
1525 ktd->ktd_error = ktd->ktd_errcnt = 0;
1526 mutex_exit(&ktrace_lock);
1527
1528 if (ktrerr) {
1529 log(LOG_NOTICE,
1530 "ktrace failed, fp %p, error 0x%x, total %d\n",
1531 fp, ktrerr, errcnt);
1532 }
1533 ktrwrite(ktd, kte);
1534 mutex_enter(&ktrace_lock);
1535 }
1536
1537 TAILQ_REMOVE(&ktdq, ktd, ktd_list);
1538 mutex_exit(&ktrace_lock);
1539
1540 /*
1541 * ktrace file descriptor can't be watched (are not visible to
1542 * userspace), so no kqueue stuff here
1543 * XXX: The above comment is wrong, because the fktrace file
1544 * descriptor is available in userland.
1545 */
1546 closef(fp);
1547
1548 cv_destroy(&ktd->ktd_sync_cv);
1549 cv_destroy(&ktd->ktd_cv);
1550
1551 callout_stop(&ktd->ktd_wakch);
1552 callout_destroy(&ktd->ktd_wakch);
1553 kmem_free(ktd, sizeof(*ktd));
1554
1555 kthread_exit(0);
1556 }
1557
1558 /*
1559 * Return true if caller has permission to set the ktracing state
1560 * of target. Essentially, the target can't possess any
1561 * more permissions than the caller. KTRFAC_PERSISTENT signifies that
1562 * the tracing will persist on sugid processes during exec; it is only
1563 * settable by a process with appropriate credentials.
1564 *
1565 * TODO: check groups. use caller effective gid.
1566 */
1567 int
1568 ktrcanset(lwp_t *calll, struct proc *targetp)
1569 {
1570 KASSERT(mutex_owned(targetp->p_lock));
1571 KASSERT(mutex_owned(&ktrace_lock));
1572
1573 if (kauth_authorize_process(calll->l_cred, KAUTH_PROCESS_KTRACE,
1574 targetp, NULL, NULL, NULL) == 0)
1575 return (1);
1576
1577 return (0);
1578 }
1579
1580 /*
1581 * Put user defined entry to ktrace records.
1582 */
1583 int
1584 sys_utrace(struct lwp *l, const struct sys_utrace_args *uap, register_t *retval)
1585 {
1586 /* {
1587 syscallarg(const char *) label;
1588 syscallarg(void *) addr;
1589 syscallarg(size_t) len;
1590 } */
1591
1592 return ktruser(SCARG(uap, label), SCARG(uap, addr),
1593 SCARG(uap, len), 1);
1594 }
1595