kern_event.c revision 1.1.1.1.2.2 1 /* $NetBSD: kern_event.c,v 1.1.1.1.2.2 2001/09/07 15:54:17 thorpej Exp $ */
2 /*-
3 * Copyright (c) 1999,2000,2001 Jonathan Lemon <jlemon (at) FreeBSD.org>
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25 * SUCH DAMAGE.
26 *
27 * $FreeBSD: src/sys/kern/kern_event.c,v 1.27 2001/07/05 17:10:44 rwatson Exp $
28 */
29
30 #include <sys/param.h>
31 #include <sys/systm.h>
32 #include <sys/kernel.h>
33 #include <sys/proc.h>
34 #include <sys/malloc.h>
35 #include <sys/unistd.h>
36 #include <sys/file.h>
37 #include <sys/fcntl.h>
38 #include <sys/select.h>
39 #include <sys/queue.h>
40 #include <sys/event.h>
41 #include <sys/eventvar.h>
42 #include <sys/poll.h>
43 #include <sys/pool.h>
44 #include <sys/protosw.h>
45 #include <sys/socket.h>
46 #include <sys/socketvar.h>
47 #include <sys/stat.h>
48 #include <sys/uio.h>
49 #include <sys/mount.h>
50 #include <sys/filedesc.h>
51 #include <sys/syscallargs.h>
52
53 static int kqueue_scan(struct file *fp, int maxevents,
54 struct kevent *ulistp, const struct timespec *timeout,
55 struct proc *p, register_t *retval);
56 static void kqueue_wakeup(struct kqueue *kq);
57
58 static int kqueue_read(struct file *fp, off_t *offset, struct uio *uio,
59 struct ucred *cred, int flags);
60 static int kqueue_write(struct file *fp, off_t *offset, struct uio *uio,
61 struct ucred *cred, int flags);
62 static int kqueue_ioctl(struct file *fp, u_long com, caddr_t data,
63 struct proc *p);
64 static int kqueue_fcntl(struct file *fp, u_int com, caddr_t data,
65 struct proc *p);
66 static int kqueue_poll(struct file *fp, int events, struct proc *p);
67 static int kqueue_kqfilter(struct file *fp, struct knote *kn);
68 static int kqueue_stat(struct file *fp, struct stat *sp, struct proc *p);
69 static int kqueue_close(struct file *fp, struct proc *p);
70
71 static struct fileops kqueueops = {
72 kqueue_read, kqueue_write, kqueue_ioctl, kqueue_fcntl, kqueue_poll,
73 kqueue_stat, kqueue_close, kqueue_kqfilter
74 };
75
76 static void knote_attach(struct knote *kn, struct filedesc *fdp);
77 static void knote_drop(struct knote *kn, struct proc *p);
78 static void knote_enqueue(struct knote *kn);
79 static void knote_dequeue(struct knote *kn);
80 static struct knote *knote_alloc(void);
81 static void knote_free(struct knote *kn);
82
83 static void filt_kqdetach(struct knote *kn);
84 static int filt_kqueue(struct knote *kn, long hint);
85 static int filt_procattach(struct knote *kn);
86 static void filt_procdetach(struct knote *kn);
87 static int filt_proc(struct knote *kn, long hint);
88 static int filt_fileattach(struct knote *kn);
89
90 static struct filterops kqread_filtops =
91 { 1, NULL, filt_kqdetach, filt_kqueue };
92 static struct filterops proc_filtops =
93 { 0, filt_procattach, filt_procdetach, filt_proc };
94 static struct filterops file_filtops =
95 { 1, filt_fileattach, NULL, NULL };
96
97 struct pool kqueue_pool;
98 struct pool knote_pool;
99
100 #define KNOTE_ACTIVATE(kn) \
101 do { \
102 kn->kn_status |= KN_ACTIVE; \
103 if ((kn->kn_status & (KN_QUEUED | KN_DISABLED)) == 0) \
104 knote_enqueue(kn); \
105 } while(0)
106
107 #define KN_HASHSIZE 64 /* XXX should be tunable */
108 #define KN_HASH(val, mask) (((val) ^ (val >> 8)) & (mask))
109
110 extern struct filterops sig_filtops;
111
112 /*
113 * Table for for all system-defined filters.
114 * These should be listed in the numeric order of the EVFILT_* defines.
115 * If filtops is NULL, the filter isn't implemented in NetBSD.
116 * End of list is when name is NULL.
117 */
118 struct kfilter {
119 char *name; /* name of filter */
120 uint32_t filter; /* id of filter */
121 struct filterops *filtops; /* operations for filter */
122 };
123
124 /* System defined filters */
125 static struct kfilter sys_kfilters[] = {
126 { "EVFILT_READ", EVFILT_READ, &file_filtops },
127 { "EVFILT_WRITE", EVFILT_WRITE, &file_filtops },
128 { "EVFILT_AIO", EVFILT_AIO, NULL },
129 { "EVFILT_VNODE", EVFILT_VNODE, &file_filtops },
130 { "EVFILT_PROC", EVFILT_PROC, &proc_filtops },
131 { "EVFILT_SIGNAL", EVFILT_SIGNAL, &sig_filtops },
132 { NULL, 0, NULL }, /* end of list */
133 };
134
135 /* User defined kfilters */
136 static struct kfilter *user_kfilters; /* array */
137 static int user_kfilterc; /* current offset */
138 static int user_kfiltermaxc; /* max size so far */
139
140 static struct kfilter *kfilter_byname(const char *);
141 static struct kfilter *kfilter_byfilter(uint32_t);
142
143 /*
144 * kqueue_init:
145 *
146 * Initialize the kqueue/knote facility.
147 */
148 void
149 kqueue_init(void)
150 {
151
152 pool_init(&kqueue_pool, sizeof(struct kqueue), 0, 0, 0, "kqueuepl",
153 0, pool_page_alloc_nointr, pool_page_free_nointr, M_KEVENT);
154 pool_init(&knote_pool, sizeof(struct knote), 0, 0, 0, "knotepl",
155 0, pool_page_alloc_nointr, pool_page_free_nointr, M_KEVENT);
156 }
157
158 /*
159 * Find kfilter entry by name, or NULL if not found.
160 */
161 static struct kfilter *
162 kfilter_byname(const char *name)
163 {
164 struct kfilter *kfilter;
165 int i;
166
167 kfilter = sys_kfilters; /* first look in system kfilters */
168 while (kfilter != NULL) {
169 for (i = 0; kfilter[i].name != NULL; i++) {
170 /* search for matching name */
171 if (kfilter[i].name[0] != '\0' &&
172 (strcmp(name, kfilter[i].name) == 0))
173 return (&kfilter[i]);
174 }
175 /* swap to user kfilters */
176 if (kfilter == sys_kfilters)
177 kfilter = user_kfilters;
178 else
179 kfilter = NULL;
180 }
181 return (NULL);
182 }
183
184 /*
185 * Find kfilter entry by filter id, or NULL if not found.
186 * Assumes entries are indexed in filter id order, for speed.
187 */
188 static struct kfilter *
189 kfilter_byfilter(uint32_t filter)
190 {
191 struct kfilter *kfilter;
192
193 if (filter < EVFILT_SYSCOUNT) /* it's a system filter */
194 kfilter = &sys_kfilters[filter];
195 else if (user_kfilters != NULL &&
196 filter < EVFILT_SYSCOUNT + user_kfilterc)
197 /* it's a user filter */
198 kfilter = &user_kfilters[filter - EVFILT_SYSCOUNT];
199 else
200 return (NULL); /* out of range */
201 KASSERT(kfilter->filter == filter); /* sanity check! */
202 return (kfilter);
203 }
204
205 /*
206 * Register a new kfilter. Stores the entry in user_kfilters.
207 * Returns 0 if operation succeeded, or an appropriate errno(2) otherwise.
208 * If retfilter != NULL, the new filterid is returned in it.
209 */
210 int
211 kfilter_register(const char *name, struct filterops *filtops, int *retfilter)
212 {
213 struct kfilter *kfilter;
214 int len;
215
216 if (name == NULL || name[0] == '\0' || filtops == NULL)
217 return (EINVAL); /* invalid args */
218 kfilter = kfilter_byname(name);
219 if (kfilter != NULL) /* already exists */
220 return (EEXIST);
221 if (user_kfilterc > 0xffffffff - EVFILT_SYSCOUNT)
222 return (EINVAL); /* too many */
223
224 /* need to grow user_kfilters */
225 if (user_kfilterc + 1 > user_kfiltermaxc) {
226 /*
227 * grow in KFILTER_EXTENT chunks. use
228 * malloc(9), because we want to
229 * traverse user_kfilters as an array.
230 */
231 user_kfiltermaxc += KFILTER_EXTENT;
232 kfilter = malloc(user_kfiltermaxc * sizeof(struct filter *),
233 M_KEVENT, M_WAITOK);
234 /* copy existing user_kfilters */
235 if (user_kfilters != NULL)
236 memcpy((caddr_t)kfilter, (caddr_t)user_kfilters,
237 user_kfilterc * sizeof(struct kfilter *));
238 /* zero new sections */
239 memset((caddr_t)kfilter +
240 user_kfilterc * sizeof(struct kfilter *), 0,
241 (user_kfiltermaxc - user_kfilterc) *
242 sizeof(struct kfilter *));
243 /* switch to new kfilter */
244 if (user_kfilters != NULL)
245 FREE(user_kfilters, M_KEVENT);
246 user_kfilters = kfilter;
247 }
248 len = strlen(name) + 1; /* copy name */
249 user_kfilters[user_kfilterc].name = (char *)
250 malloc(len, M_KEVENT, M_WAITOK);
251 memcpy(user_kfilters[user_kfilterc].name, name, len);
252 user_kfilters[user_kfilterc].filter = user_kfilterc + EVFILT_SYSCOUNT;
253 len = sizeof(struct filterops); /* copy filtops */
254 user_kfilters[user_kfilterc].filtops = (struct filterops *)
255 malloc(len, M_KEVENT, M_WAITOK);
256 memcpy(user_kfilters[user_kfilterc].filtops, filtops, len);
257 if (retfilter != NULL)
258 *retfilter = user_kfilters[user_kfilterc].filter;
259 user_kfilterc++; /* finally, increment count */
260 return (0);
261 }
262
263 /*
264 * Unregister a kfilter previously registered with kfilter_register.
265 * This retains the filter id, but clears the name and frees filtops (filter
266 * operations), so that the number isn't reused during a boot.
267 * Returns 0 if operation succeeded, or an appropriate errno(2) otherwise.
268 */
269 int
270 kfilter_unregister(const char *name)
271 {
272 struct kfilter *kfilter;
273
274 if (name == NULL || name[0] == '\0')
275 return (EINVAL); /* invalid name */
276 kfilter = kfilter_byname(name);
277 if (kfilter == NULL) /* not found */
278 return (ENOENT);
279 if (kfilter->filter < EVFILT_SYSCOUNT)
280 return (EINVAL); /* can't detach system filters */
281
282 if (kfilter->name[0] != '\0') {
283 free(kfilter->name, M_KEVENT);
284 kfilter->name = ""; /* mark as `not implemented' */
285 }
286 if (kfilter->filtops != NULL) {
287 free(kfilter->filtops, M_KEVENT);
288 kfilter->filtops = NULL; /* mark as `not implemented' */
289 }
290 return (0);
291 }
292
293
294 /*
295 * Filter attach method for EVFILT_READ and EVFILT_WRITE on normal file
296 * descriptors. Calls struct fileops kqfilter method for given file descriptor.
297 */
298 static int
299 filt_fileattach(struct knote *kn)
300 {
301 struct file *fp;
302
303 fp = kn->kn_fp;
304 return ((*fp->f_ops->fo_kqfilter)(fp, kn));
305 }
306
307 /*
308 * Filter detach method for EVFILT_READ on kqueue descriptor.
309 */
310 static void
311 filt_kqdetach(struct knote *kn)
312 {
313 struct kqueue *kq;
314
315 kq = (struct kqueue *)kn->kn_fp->f_data;
316 SLIST_REMOVE(&kq->kq_sel.si_klist, kn, knote, kn_selnext);
317 }
318
319 /*
320 * Filter event method for EVFILT_READ on kqueue descriptor.
321 */
322 /*ARGSUSED*/
323 static int
324 filt_kqueue(struct knote *kn, long hint)
325 {
326 struct kqueue *kq;
327
328 kq = (struct kqueue *)kn->kn_fp->f_data;
329 kn->kn_data = kq->kq_count;
330 return (kn->kn_data > 0);
331 }
332
333 /*
334 * Filter attach method for EVFILT_PROC.
335 */
336 static int
337 filt_procattach(struct knote *kn)
338 {
339 struct proc *p;
340
341 p = pfind(kn->kn_id);
342 if (p == NULL)
343 return (ESRCH);
344
345 kn->kn_ptr.p_proc = p;
346 kn->kn_flags |= EV_CLEAR; /* automatically set */
347
348 /*
349 * internal flag indicating registration done by kernel
350 */
351 if (kn->kn_flags & EV_FLAG1) {
352 kn->kn_data = kn->kn_sdata; /* ppid */
353 kn->kn_fflags = NOTE_CHILD;
354 kn->kn_flags &= ~EV_FLAG1;
355 }
356
357 /* XXXLUKEM */
358 /* XXX lock the proc here while adding to the list? */
359 SLIST_INSERT_HEAD(&p->p_klist, kn, kn_selnext);
360
361 return (0);
362 }
363
364 /*
365 * Filter detach method for EVFILT_PROC.
366 *
367 * The knote may be attached to a different process, which may exit,
368 * leaving nothing for the knote to be attached to. So when the process
369 * exits, the knote is marked as DETACHED and also flagged as ONESHOT so
370 * it will be deleted when read out. However, as part of the knote deletion,
371 * this routine is called, so a check is needed to avoid actually performing
372 * a detach, because the original process does not exist any more.
373 */
374 static void
375 filt_procdetach(struct knote *kn)
376 {
377 struct proc *p;
378
379 p = kn->kn_ptr.p_proc;
380 if (kn->kn_status & KN_DETACHED)
381 return;
382
383 /* XXXLUKEM */
384 /* XXX locking? this might modify another process. */
385 SLIST_REMOVE(&p->p_klist, kn, knote, kn_selnext);
386 }
387
388 /*
389 * Filter event method for EVFILT_PROC.
390 */
391 static int
392 filt_proc(struct knote *kn, long hint)
393 {
394 u_int event;
395
396 /*
397 * mask off extra data
398 */
399 event = (u_int)hint & NOTE_PCTRLMASK;
400
401 /*
402 * if the user is interested in this event, record it.
403 */
404 if (kn->kn_sfflags & event)
405 kn->kn_fflags |= event;
406
407 /*
408 * process is gone, so flag the event as finished.
409 */
410 if (event == NOTE_EXIT) {
411 kn->kn_status |= KN_DETACHED;
412 kn->kn_flags |= (EV_EOF | EV_ONESHOT);
413 return (1);
414 }
415
416 /*
417 * process forked, and user wants to track the new process,
418 * so attach a new knote to it, and immediately report an
419 * event with the parent's pid.
420 */
421 if ((event == NOTE_FORK) && (kn->kn_sfflags & NOTE_TRACK)) {
422 struct kevent kev;
423 int error;
424
425 /*
426 * register knote with new process.
427 */
428 kev.ident = hint & NOTE_PDATAMASK; /* pid */
429 kev.filter = kn->kn_filter;
430 kev.flags = kn->kn_flags | EV_ADD | EV_ENABLE | EV_FLAG1;
431 kev.fflags = kn->kn_sfflags;
432 kev.data = kn->kn_id; /* parent */
433 kev.udata = kn->kn_kevent.udata; /* preserve udata */
434 error = kqueue_register(kn->kn_kq, &kev, NULL);
435 if (error)
436 kn->kn_fflags |= NOTE_TRACKERR;
437 }
438
439 return (kn->kn_fflags != 0);
440 }
441
442 /*
443 * kqueue(2) system call.
444 */
445 int
446 sys_kqueue(struct proc *p, void *v, register_t *retval)
447 {
448 struct filedesc *fdp;
449 struct kqueue *kq;
450 struct file *fp;
451 int fd, error;
452
453 fdp = p->p_fd;
454 error = falloc(p, &fp, &fd); /* setup a new file descriptor */
455 if (error)
456 return (error);
457 fp->f_flag = FREAD | FWRITE;
458 fp->f_type = DTYPE_KQUEUE;
459 fp->f_ops = &kqueueops;
460 kq = pool_get(&kqueue_pool, PR_WAITOK);
461 memset((char *)kq, 0, sizeof(struct kqueue));
462 TAILQ_INIT(&kq->kq_head);
463 fp->f_data = (caddr_t)kq; /* store the kqueue with the fp */
464 *retval = fd;
465 if (fdp->fd_knlistsize < 0)
466 fdp->fd_knlistsize = 0; /* this process has a kq */
467 kq->kq_fdp = fdp;
468 FILE_SET_MATURE(fp);
469 FILE_UNUSE(fp, p); /* falloc() does FILE_USE() */
470 return (error);
471 }
472
473 /*
474 * kevent(2) system call.
475 */
476 int
477 sys_kevent(struct proc *p, void *v, register_t *retval)
478 {
479 struct sys_kevent_args /* {
480 syscallarg(int) fd;
481 syscallarg(const struct kevent *) changelist;
482 syscallarg(int) nchanges;
483 syscallarg(struct kevent *) eventlist;
484 syscallarg(int) nevents;
485 syscallarg(const struct timespec *) timeout;
486 } */ *uap = v;
487 struct filedesc *fdp;
488 struct kevent *kevp;
489 struct kqueue *kq;
490 struct file *fp;
491 struct timespec ts;
492 int i, n, nerrors, error;
493
494 fdp = p->p_fd; /* check that we're dealing with a kq */
495 if ((u_int)SCARG(uap, fd) >= fdp->fd_nfiles ||
496 (fp = fdp->fd_ofiles[SCARG(uap, fd)]) == NULL ||
497 (fp->f_type != DTYPE_KQUEUE))
498 return (EBADF);
499
500 FILE_USE(fp);
501
502 if (SCARG(uap, timeout) != NULL) {
503 error = copyin(SCARG(uap, timeout), &ts, sizeof(ts));
504 if (error)
505 goto done;
506 SCARG(uap, timeout) = &ts;
507 }
508
509 kq = (struct kqueue *)fp->f_data;
510 nerrors = 0;
511
512 /* traverse list of events to register */
513 while (SCARG(uap, nchanges) > 0) {
514 /* copyin a maximum of KQ_EVENTS at each pass */
515 n = MIN(SCARG(uap, nchanges), KQ_NEVENTS);
516 error = copyin(SCARG(uap, changelist), kq->kq_kev,
517 n * sizeof(struct kevent));
518 if (error)
519 goto done;
520 for (i = 0; i < n; i++) {
521 kevp = &kq->kq_kev[i];
522 kevp->flags &= ~EV_SYSFLAGS;
523 /* register each knote */
524 error = kqueue_register(kq, kevp, p);
525 if (error) {
526 if (SCARG(uap, nevents) != 0) {
527 kevp->flags = EV_ERROR;
528 kevp->data = error;
529 error = copyout((caddr_t)kevp,
530 (caddr_t)SCARG(uap, eventlist),
531 sizeof(*kevp));
532 if (error)
533 goto done;
534 SCARG(uap, eventlist)++;
535 SCARG(uap, nevents)--;
536 nerrors++;
537 } else {
538 goto done;
539 }
540 }
541 }
542 SCARG(uap, nchanges) -= n; /* update the results */
543 SCARG(uap, changelist) += n;
544 }
545 if (nerrors) {
546 *retval = nerrors;
547 error = 0;
548 goto done;
549 }
550
551 /* actually scan through the events */
552 error = kqueue_scan(fp, SCARG(uap, nevents), SCARG(uap, eventlist),
553 SCARG(uap, timeout), p, retval);
554 done:
555 FILE_UNUSE(fp, p);
556 return (error);
557 }
558
559 /*
560 * Register a given kevent kev onto the kqueue
561 */
562 int
563 kqueue_register(struct kqueue *kq, struct kevent *kev, struct proc *p)
564 {
565 struct filedesc *fdp;
566 struct kfilter *kfilter;
567 struct file *fp;
568 struct knote *kn;
569 int s, error;
570
571 fdp = kq->kq_fdp;
572 fp = NULL;
573 kn = NULL;
574 error = 0;
575 kfilter = kfilter_byfilter(kev->filter);
576 if (kfilter == NULL || kfilter->filtops == NULL)
577 return (EINVAL); /* filter not found nor implemented */
578
579 /* search if knote already exists */
580 if (kfilter->filtops->f_isfd) { /* monitoring a file descriptor */
581 if ((u_int)kev->ident >= fdp->fd_nfiles ||
582 (fp = fdp->fd_ofiles[kev->ident]) == NULL)
583 return (EBADF); /* validate descriptor */
584 FILE_USE(fp);
585
586 if (kev->ident < fdp->fd_knlistsize) {
587 SLIST_FOREACH(kn, &fdp->fd_knlist[kev->ident], kn_link)
588 if (kq == kn->kn_kq &&
589 kev->filter == kn->kn_filter)
590 break;
591 }
592 } else {
593 /*
594 * not monitoring a file descriptor, so
595 * lookup knotes in internal hash table
596 */
597 if (fdp->fd_knhashmask != 0) {
598 struct klist *list;
599
600 list = &fdp->fd_knhash[
601 KN_HASH((u_long)kev->ident, fdp->fd_knhashmask)];
602 SLIST_FOREACH(kn, list, kn_link)
603 if (kev->ident == kn->kn_id &&
604 kq == kn->kn_kq &&
605 kev->filter == kn->kn_filter)
606 break;
607 }
608 }
609
610 if (kn == NULL && ((kev->flags & EV_ADD) == 0)) {
611 error = ENOENT; /* filter not found */
612 goto done;
613 }
614
615 /*
616 * kn now contains the matching knote, or NULL if no match
617 */
618 if (kev->flags & EV_ADD) { /* add knote */
619
620 if (kn == NULL) { /* create new knote */
621 kn = knote_alloc();
622 if (kn == NULL) {
623 error = ENOMEM;
624 goto done;
625 }
626 kn->kn_fp = fp;
627 kn->kn_kq = kq;
628 kn->kn_fop = kfilter->filtops;
629
630 /*
631 * apply reference count to knote structure, and
632 * do not release it at the end of this routine.
633 */
634 fp = NULL;
635
636 kn->kn_sfflags = kev->fflags;
637 kn->kn_sdata = kev->data;
638 kev->fflags = 0;
639 kev->data = 0;
640 kn->kn_kevent = *kev;
641
642 knote_attach(kn, fdp);
643 if ((error = kfilter->filtops->f_attach(kn)) != 0) {
644 knote_drop(kn, p);
645 goto done;
646 }
647 } else { /* modify existing knote */
648 /*
649 * The user may change some filter values after the
650 * initial EV_ADD, but doing so will not reset any
651 * filter which have already been triggered.
652 */
653 kn->kn_sfflags = kev->fflags;
654 kn->kn_sdata = kev->data;
655 kn->kn_kevent.udata = kev->udata;
656 }
657
658 s = splhigh();
659 if (kn->kn_fop->f_event(kn, 0))
660 KNOTE_ACTIVATE(kn);
661 splx(s);
662
663 } else if (kev->flags & EV_DELETE) { /* delete knote */
664 kn->kn_fop->f_detach(kn);
665 knote_drop(kn, p);
666 goto done;
667 }
668
669 /* disable knote */
670 if ((kev->flags & EV_DISABLE) &&
671 ((kn->kn_status & KN_DISABLED) == 0)) {
672 s = splhigh();
673 kn->kn_status |= KN_DISABLED;
674 splx(s);
675 }
676
677 /* enable knote */
678 if ((kev->flags & EV_ENABLE) && (kn->kn_status & KN_DISABLED)) {
679 s = splhigh();
680 kn->kn_status &= ~KN_DISABLED;
681 if ((kn->kn_status & KN_ACTIVE) &&
682 ((kn->kn_status & KN_QUEUED) == 0))
683 knote_enqueue(kn);
684 splx(s);
685 }
686
687 done:
688 if (fp != NULL)
689 FILE_UNUSE(fp, p);
690 return (error);
691 }
692
693 /*
694 * Scan through the list of events on fp (for a maximum of maxevents),
695 * returning the results in to ulistp. Timeout is determined by tsp; if
696 * NULL, wait indefinitely, if 0 valued, perform a poll, otherwise wait
697 * as appropriate.
698 */
699 static int
700 kqueue_scan(struct file *fp, int maxevents, struct kevent *ulistp,
701 const struct timespec *tsp, struct proc *p, register_t *retval)
702 {
703 struct kqueue *kq;
704 struct kevent *kevp;
705 struct timeval atv;
706 struct knote *kn, marker;
707 int s, count, timeout, nkev, error;
708
709 kq = (struct kqueue *)fp->f_data;
710 count = maxevents;
711 nkev = error = 0;
712 if (count == 0)
713 goto done;
714
715 if (tsp != NULL) { /* timeout supplied */
716 TIMESPEC_TO_TIMEVAL(&atv, tsp);
717 if (itimerfix(&atv)) {
718 error = EINVAL;
719 goto done;
720 }
721 s = splclock();
722 timeradd(&atv, &time, &atv); /* calc. time to wait until */
723 splx(s);
724 if (tsp->tv_sec == 0 && tsp->tv_nsec == 0)
725 timeout = -1; /* perform a poll */
726 else
727 timeout = hzto(&atv); /* calculate hz till timeout */
728 } else {
729 atv.tv_sec = 0; /* no timeout, wait forever */
730 atv.tv_usec = 0;
731 timeout = 0;
732 }
733 goto start;
734
735 retry:
736 if (atv.tv_sec || atv.tv_usec) { /* timeout requested */
737 s = splclock();
738 if (timercmp(&time, &atv, >=)) {
739 splx(s);
740 goto done; /* timeout reached */
741 }
742 splx(s);
743 timeout = hzto(&atv); /* recalc. timeout remaining */
744 }
745
746 start:
747 kevp = kq->kq_kev;
748 s = splhigh();
749 if (kq->kq_count == 0) {
750 if (timeout < 0) {
751 error = EWOULDBLOCK;
752 } else {
753 kq->kq_state |= KQ_SLEEP;
754 error = tsleep(kq, PSOCK | PCATCH, "kqread", timeout);
755 }
756 splx(s);
757 if (error == 0)
758 goto retry;
759 /* don't restart after signals... */
760 if (error == ERESTART)
761 error = EINTR;
762 else if (error == EWOULDBLOCK)
763 error = 0;
764 goto done;
765 }
766
767 TAILQ_INSERT_TAIL(&kq->kq_head, &marker, kn_tqe);
768 /* mark end of knote list */
769 while (count) { /* while user wants data ... */
770 kn = TAILQ_FIRST(&kq->kq_head); /* get next knote */
771 TAILQ_REMOVE(&kq->kq_head, kn, kn_tqe);
772 if (kn == &marker) { /* if it's our marker, stop */
773 splx(s);
774 if (count == maxevents)
775 goto retry;
776 goto done;
777 }
778 if (kn->kn_status & KN_DISABLED) {
779 /* don't want disabled events */
780 kn->kn_status &= ~KN_QUEUED;
781 kq->kq_count--;
782 continue;
783 }
784 if ((kn->kn_flags & EV_ONESHOT) == 0 &&
785 kn->kn_fop->f_event(kn, 0) == 0) {
786 /*
787 * non-ONESHOT event that hasn't
788 * triggered again, so de-queue.
789 */
790 kn->kn_status &= ~(KN_QUEUED | KN_ACTIVE);
791 kq->kq_count--;
792 continue;
793 }
794 *kevp = kn->kn_kevent;
795 kevp++;
796 nkev++;
797 if (kn->kn_flags & EV_ONESHOT) {
798 /* delete ONESHOT events after retrieval */
799 kn->kn_status &= ~KN_QUEUED;
800 kq->kq_count--;
801 splx(s);
802 kn->kn_fop->f_detach(kn);
803 knote_drop(kn, p);
804 s = splhigh();
805 } else if (kn->kn_flags & EV_CLEAR) {
806 /* clear state after retrieval */
807 kn->kn_data = 0;
808 kn->kn_fflags = 0;
809 kn->kn_status &= ~(KN_QUEUED | KN_ACTIVE);
810 kq->kq_count--;
811 } else {
812 /* add event back on list */
813 TAILQ_INSERT_TAIL(&kq->kq_head, kn, kn_tqe);
814 }
815 count--;
816 if (nkev == KQ_NEVENTS) {
817 /* do copyouts in KQ_NEVENTS chunks */
818 splx(s);
819 error = copyout((caddr_t)&kq->kq_kev, (caddr_t)ulistp,
820 sizeof(struct kevent) * nkev);
821 ulistp += nkev;
822 nkev = 0;
823 kevp = kq->kq_kev;
824 s = splhigh();
825 if (error)
826 break;
827 }
828 }
829 /* remove marker */
830 TAILQ_REMOVE(&kq->kq_head, &marker, kn_tqe);
831 splx(s);
832 done:
833 if (nkev != 0) /* copyout remaining events */
834 error = copyout((caddr_t)&kq->kq_kev, (caddr_t)ulistp,
835 sizeof(struct kevent) * nkev);
836 *retval = maxevents - count;
837 return (error);
838 }
839
840 /*
841 * struct fileops read method for a kqueue descriptor.
842 * Not implemented.
843 * XXX: This could be expanded to call kqueue_scan, if desired.
844 */
845 /*ARGSUSED*/
846 static int
847 kqueue_read(struct file *fp, off_t *offset, struct uio *uio,
848 struct ucred *cred, int flags)
849 {
850
851 return (ENXIO);
852 }
853
854 /*
855 * struct fileops write method for a kqueue descriptor.
856 * Not implemented.
857 */
858 /*ARGSUSED*/
859 static int
860 kqueue_write(struct file *fp, off_t *offset, struct uio *uio,
861 struct ucred *cred, int flags)
862 {
863
864 return (ENXIO);
865 }
866
867 /*
868 * struct fileops ioctl method for a kqueue descriptor.
869 *
870 * Two ioctls are currently supported. They both use struct kfilter_mapping:
871 * KFILTER_BYNAME find name for filter, and return result in
872 * name, which is of size len.
873 * KFILTER_BYFILTER find filter for name. len is ignored.
874 */
875 /*ARGSUSED*/
876 static int
877 kqueue_ioctl(struct file *fp, u_long com, caddr_t data, struct proc *p)
878 {
879 struct kfilter_mapping *km;
880 struct kfilter *kfilter;
881 char *name;
882 int error;
883
884 km = (struct kfilter_mapping *)data;
885 error = 0;
886
887 switch (com) {
888 case KFILTER_BYFILTER: /* convert filter -> name */
889 kfilter = kfilter_byfilter(km->filter);
890 if (kfilter != NULL)
891 error = copyoutstr(kfilter->name, km->name, km->len,
892 NULL);
893 else
894 error = ENOENT;
895 break;
896
897 case KFILTER_BYNAME: /* convert name -> filter */
898 MALLOC(name, char *, KFILTER_MAXNAME, M_KEVENT, M_WAITOK);
899 error = copyinstr(km->name, name, KFILTER_MAXNAME, NULL);
900 if (error) {
901 free(name, M_KEVENT);
902 break;
903 }
904 kfilter = kfilter_byname(name);
905 if (kfilter != NULL)
906 km->filter = kfilter->filter;
907 else
908 error = ENOENT;
909 free(name, M_KEVENT);
910 break;
911
912 #if 1 /* XXXLUKEM - test register & unregister */
913 case KFILTER_REGISTER:
914 case KFILTER_UNREGISTER:
915 MALLOC(name, char *, KFILTER_MAXNAME, M_KEVENT, M_WAITOK);
916 error = copyinstr(km->name, name, KFILTER_MAXNAME, NULL);
917 if (error) {
918 free(name, M_KEVENT);
919 break;
920 }
921 if (com == KFILTER_REGISTER) {
922 kfilter = kfilter_byfilter(km->filter);
923 if (kfilter != NULL) {
924 error = kfilter_register(name,
925 kfilter->filtops, &km->filter);
926 } else
927 error = ENOENT;
928 } else
929 error = kfilter_unregister(name);
930 free(name, M_KEVENT);
931 break;
932 #endif
933
934 default:
935 error = ENOTTY;
936
937 }
938 return (error);
939 }
940
941 /*
942 * struct fileops fcntl method for a kqueue descriptor.
943 * Not implemented.
944 */
945 /*ARGSUSED*/
946 static int
947 kqueue_fcntl(struct file *fp, u_int com, caddr_t data, struct proc *p)
948 {
949
950 return (ENOTTY);
951 }
952
953 /*
954 * struct fileops poll method for a kqueue descriptor.
955 * Determine if kqueue has events pending.
956 */
957 /*ARGSUSED*/
958 static int
959 kqueue_poll(struct file *fp, int events, struct proc *p)
960 {
961 struct kqueue *kq;
962 int revents, s;
963
964 kq = (struct kqueue *)fp->f_data;
965 revents = 0;
966 s = splnet(); /* XXXLUKEM: is this correct? */
967 if (events & (POLLIN | POLLRDNORM)) {
968 if (kq->kq_count) {
969 revents |= events & (POLLIN | POLLRDNORM);
970 } else {
971 /* XXXLUKEM: splsched() for next? */
972 selrecord(p, &kq->kq_sel);
973 kq->kq_state |= KQ_SEL;
974 }
975 }
976 splx(s);
977 return (revents);
978 }
979
980 /*
981 * struct fileops stat method for a kqueue descriptor.
982 * Returns dummy info, with st_size being number of events pending.
983 */
984 /*ARGSUSED*/
985 static int
986 kqueue_stat(struct file *fp, struct stat *st, struct proc *p)
987 {
988 struct kqueue *kq;
989
990 kq = (struct kqueue *)fp->f_data;
991 memset((void *)st, 0, sizeof(*st));
992 st->st_size = kq->kq_count;
993 st->st_blksize = sizeof(struct kevent);
994 st->st_mode = S_IFIFO;
995 return (0);
996 }
997
998 /*
999 * struct fileops close method for a kqueue descriptor.
1000 * Cleans up kqueue.
1001 */
1002 /*ARGSUSED*/
1003 static int
1004 kqueue_close(struct file *fp, struct proc *p)
1005 {
1006 struct kqueue *kq;
1007 struct filedesc *fdp;
1008 struct knote **knp, *kn, *kn0;
1009 int i;
1010
1011 kq = (struct kqueue *)fp->f_data;
1012 fdp = p->p_fd;
1013 for (i = 0; i < fdp->fd_knlistsize; i++) {
1014 knp = &SLIST_FIRST(&fdp->fd_knlist[i]);
1015 kn = *knp;
1016 while (kn != NULL) {
1017 kn0 = SLIST_NEXT(kn, kn_link);
1018 if (kq == kn->kn_kq) {
1019 kn->kn_fop->f_detach(kn);
1020 FILE_UNUSE(kn->kn_fp, p);
1021 knote_free(kn);
1022 *knp = kn0;
1023 } else {
1024 knp = &SLIST_NEXT(kn, kn_link);
1025 }
1026 kn = kn0;
1027 }
1028 }
1029 if (fdp->fd_knhashmask != 0) {
1030 for (i = 0; i < fdp->fd_knhashmask + 1; i++) {
1031 knp = &SLIST_FIRST(&fdp->fd_knhash[i]);
1032 kn = *knp;
1033 while (kn != NULL) {
1034 kn0 = SLIST_NEXT(kn, kn_link);
1035 if (kq == kn->kn_kq) {
1036 kn->kn_fop->f_detach(kn);
1037 /* XXX non-fd release of kn->kn_ptr */
1038 knote_free(kn);
1039 *knp = kn0;
1040 } else {
1041 knp = &SLIST_NEXT(kn, kn_link);
1042 }
1043 kn = kn0;
1044 }
1045 }
1046 }
1047 pool_put(&kqueue_pool, kq);
1048 fp->f_data = NULL;
1049
1050 return (0);
1051 }
1052
1053 /*
1054 * wakeup a kqueue
1055 */
1056 static void
1057 kqueue_wakeup(struct kqueue *kq)
1058 {
1059
1060 if (kq->kq_state & KQ_SLEEP) { /* if currently sleeping ... */
1061 kq->kq_state &= ~KQ_SLEEP;
1062 wakeup(kq); /* ... wakeup */
1063 }
1064 if (kq->kq_state & KQ_SEL) { /* if currently polling ... */
1065 kq->kq_state &= ~KQ_SEL;
1066 selwakeup(&kq->kq_sel); /* ... selwakeup */
1067 }
1068 KNOTE(&kq->kq_sel.si_klist, 0);
1069 }
1070
1071 /*
1072 * struct fileops kqfilter method for a kqueue descriptor.
1073 * Event triggered when monitored kqueue changes.
1074 */
1075 /*ARGSUSED*/
1076 static int
1077 kqueue_kqfilter(struct file *fp, struct knote *kn)
1078 {
1079 struct kqueue *kq;
1080
1081 kq = (struct kqueue *)kn->kn_fp->f_data;
1082 if (kn->kn_filter != EVFILT_READ)
1083 return (1);
1084 kn->kn_fop = &kqread_filtops;
1085 SLIST_INSERT_HEAD(&kq->kq_sel.si_klist, kn, kn_selnext);
1086 return (0);
1087 }
1088
1089
1090 /*
1091 * Walk down a list of knotes, activating them if their event has triggered.
1092 */
1093 void
1094 knote(struct klist *list, long hint)
1095 {
1096 struct knote *kn;
1097
1098 SLIST_FOREACH(kn, list, kn_selnext)
1099 if (kn->kn_fop->f_event(kn, hint))
1100 KNOTE_ACTIVATE(kn);
1101 }
1102
1103 /*
1104 * Remove all knotes from a specified klist
1105 */
1106 void
1107 knote_remove(struct proc *p, struct klist *list)
1108 {
1109 struct knote *kn;
1110
1111 while ((kn = SLIST_FIRST(list)) != NULL) {
1112 kn->kn_fop->f_detach(kn);
1113 knote_drop(kn, p);
1114 }
1115 }
1116
1117 /*
1118 * Remove all knotes referencing a specified fd
1119 */
1120 void
1121 knote_fdclose(struct proc *p, int fd)
1122 {
1123 struct filedesc *fdp;
1124 struct klist *list;
1125
1126 fdp = p->p_fd;
1127 list = &fdp->fd_knlist[fd];
1128 knote_remove(p, list);
1129 }
1130
1131 /*
1132 * Attach a new knote to a file descriptor
1133 */
1134 static void
1135 knote_attach(struct knote *kn, struct filedesc *fdp)
1136 {
1137 struct klist *list;
1138 int size;
1139
1140 if (! kn->kn_fop->f_isfd) {
1141 /*
1142 * if knote is not on an fd, store
1143 * on internal hash table.
1144 */
1145 if (fdp->fd_knhashmask == 0)
1146 fdp->fd_knhash = hashinit(KN_HASHSIZE, HASH_LIST,
1147 M_KEVENT, M_WAITOK, &fdp->fd_knhashmask);
1148 list = &fdp->fd_knhash[KN_HASH(kn->kn_id, fdp->fd_knhashmask)];
1149 goto done;
1150 }
1151
1152 /*
1153 * otherwise, knote is on an fd.
1154 * knotes are stored in fd_knlist
1155 * indexed by kn->kn_id.
1156 */
1157 if (fdp->fd_knlistsize <= kn->kn_id) {
1158 /* expand list if too small */
1159 size = fdp->fd_knlistsize;
1160 while (size <= kn->kn_id)
1161 size += KQ_EXTENT; /* grow in KQ_EXTENT chunks */
1162 list = malloc(size * sizeof(struct klist *), M_KEVENT,M_WAITOK);
1163 /* copy existing knlist */
1164 memcpy((caddr_t)list, (caddr_t)fdp->fd_knlist,
1165 fdp->fd_knlistsize * sizeof(struct klist *));
1166 /* zero new sections */
1167 memset((caddr_t)list +
1168 fdp->fd_knlistsize * sizeof(struct klist *), 0,
1169 (size - fdp->fd_knlistsize) * sizeof(struct klist *));
1170 if (fdp->fd_knlist != NULL) /* switch to new knlist */
1171 FREE(fdp->fd_knlist, M_KEVENT);
1172 fdp->fd_knlistsize = size;
1173 fdp->fd_knlist = list;
1174 }
1175 list = &fdp->fd_knlist[kn->kn_id]; /* get list head for this fd */
1176 done:
1177 SLIST_INSERT_HEAD(list, kn, kn_link); /* add new knote */
1178 kn->kn_status = 0;
1179 }
1180
1181 /*
1182 * Drop knote.
1183 * Should be called at spl == 0, since we don't want to hold spl
1184 * while calling FILE_UNUSE and free.
1185 */
1186 static void
1187 knote_drop(struct knote *kn, struct proc *p)
1188 {
1189 struct filedesc *fdp;
1190 struct klist *list;
1191
1192 fdp = p->p_fd;
1193 if (kn->kn_fop->f_isfd)
1194 list = &fdp->fd_knlist[kn->kn_id];
1195 else
1196 list = &fdp->fd_knhash[KN_HASH(kn->kn_id, fdp->fd_knhashmask)];
1197
1198 SLIST_REMOVE(list, kn, knote, kn_link);
1199 if (kn->kn_status & KN_QUEUED)
1200 knote_dequeue(kn);
1201 if (kn->kn_fop->f_isfd)
1202 FILE_UNUSE(kn->kn_fp, p);
1203 knote_free(kn);
1204 }
1205
1206
1207 /*
1208 * Queue new event for knote.
1209 */
1210 static void
1211 knote_enqueue(struct knote *kn)
1212 {
1213 struct kqueue *kq;
1214 int s;
1215
1216 kq = kn->kn_kq;
1217 s = splhigh();
1218 KASSERT((kn->kn_status & KN_QUEUED) == 0);
1219
1220 TAILQ_INSERT_TAIL(&kq->kq_head, kn, kn_tqe);
1221 kn->kn_status |= KN_QUEUED;
1222 kq->kq_count++;
1223 splx(s);
1224 kqueue_wakeup(kq);
1225 }
1226
1227 /*
1228 * Dequeue event for knote.
1229 */
1230 static void
1231 knote_dequeue(struct knote *kn)
1232 {
1233 struct kqueue *kq;
1234 int s;
1235
1236 kq = kn->kn_kq;
1237 s = splhigh();
1238 KASSERT(kn->kn_status & KN_QUEUED);
1239
1240 TAILQ_REMOVE(&kq->kq_head, kn, kn_tqe);
1241 kn->kn_status &= ~KN_QUEUED;
1242 kq->kq_count--;
1243 splx(s);
1244 }
1245
1246 /*
1247 * Create a new knote.
1248 */
1249 static struct knote *
1250 knote_alloc(void)
1251 {
1252
1253 return (pool_get(&knote_pool, PR_WAITOK));
1254 }
1255
1256 /*
1257 * Free a knote.
1258 */
1259 static void
1260 knote_free(struct knote *kn)
1261 {
1262
1263 pool_put(&knote_pool, kn);
1264 }
1265