subr_callback.c revision 1.2.2.2
1 1.2.2.2 kardel /* $NetBSD: subr_callback.c,v 1.2.2.2 2006/06/01 22:38:09 kardel Exp $ */ 2 1.2.2.2 kardel 3 1.2.2.2 kardel /*- 4 1.2.2.2 kardel * Copyright (c)2006 YAMAMOTO Takashi, 5 1.2.2.2 kardel * All rights reserved. 6 1.2.2.2 kardel * 7 1.2.2.2 kardel * Redistribution and use in source and binary forms, with or without 8 1.2.2.2 kardel * modification, are permitted provided that the following conditions 9 1.2.2.2 kardel * are met: 10 1.2.2.2 kardel * 1. Redistributions of source code must retain the above copyright 11 1.2.2.2 kardel * notice, this list of conditions and the following disclaimer. 12 1.2.2.2 kardel * 2. Redistributions in binary form must reproduce the above copyright 13 1.2.2.2 kardel * notice, this list of conditions and the following disclaimer in the 14 1.2.2.2 kardel * documentation and/or other materials provided with the distribution. 15 1.2.2.2 kardel * 16 1.2.2.2 kardel * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 17 1.2.2.2 kardel * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 18 1.2.2.2 kardel * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 19 1.2.2.2 kardel * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 20 1.2.2.2 kardel * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 21 1.2.2.2 kardel * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 22 1.2.2.2 kardel * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 23 1.2.2.2 kardel * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 24 1.2.2.2 kardel * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 25 1.2.2.2 kardel * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 26 1.2.2.2 kardel * SUCH DAMAGE. 27 1.2.2.2 kardel */ 28 1.2.2.2 kardel 29 1.2.2.2 kardel #include <sys/cdefs.h> 30 1.2.2.2 kardel __KERNEL_RCSID(0, "$NetBSD: subr_callback.c,v 1.2.2.2 2006/06/01 22:38:09 kardel Exp $"); 31 1.2.2.2 kardel 32 1.2.2.2 kardel #include <sys/param.h> 33 1.2.2.2 kardel #include <sys/systm.h> 34 1.2.2.2 kardel #include <sys/proc.h> 35 1.2.2.2 kardel #include <sys/callback.h> 36 1.2.2.2 kardel 37 1.2.2.2 kardel #define CH_WANT 1 38 1.2.2.2 kardel 39 1.2.2.2 kardel void 40 1.2.2.2 kardel callback_head_init(struct callback_head *ch) 41 1.2.2.2 kardel { 42 1.2.2.2 kardel 43 1.2.2.2 kardel simple_lock_init(&ch->ch_lock); 44 1.2.2.2 kardel TAILQ_INIT(&ch->ch_q); 45 1.2.2.2 kardel ch->ch_next = NULL; 46 1.2.2.2 kardel ch->ch_nentries = 0; 47 1.2.2.2 kardel ch->ch_running = 0; 48 1.2.2.2 kardel ch->ch_flags = 0; 49 1.2.2.2 kardel } 50 1.2.2.2 kardel 51 1.2.2.2 kardel void 52 1.2.2.2 kardel callback_register(struct callback_head *ch, struct callback_entry *ce, 53 1.2.2.2 kardel void *obj, int (*fn)(struct callback_entry *, void *, void *)) 54 1.2.2.2 kardel { 55 1.2.2.2 kardel 56 1.2.2.2 kardel ce->ce_func = fn; 57 1.2.2.2 kardel ce->ce_obj = obj; 58 1.2.2.2 kardel simple_lock(&ch->ch_lock); 59 1.2.2.2 kardel TAILQ_INSERT_TAIL(&ch->ch_q, ce, ce_q); 60 1.2.2.2 kardel ch->ch_nentries++; 61 1.2.2.2 kardel simple_unlock(&ch->ch_lock); 62 1.2.2.2 kardel } 63 1.2.2.2 kardel 64 1.2.2.2 kardel void 65 1.2.2.2 kardel callback_unregister(struct callback_head *ch, struct callback_entry *ce) 66 1.2.2.2 kardel { 67 1.2.2.2 kardel 68 1.2.2.2 kardel simple_lock(&ch->ch_lock); 69 1.2.2.2 kardel while (ch->ch_running > 0) { 70 1.2.2.2 kardel ch->ch_flags |= CH_WANT; 71 1.2.2.2 kardel ltsleep(&ch->ch_running, PVM, "recunreg", 0, &ch->ch_lock); 72 1.2.2.2 kardel } 73 1.2.2.2 kardel if (__predict_false(ch->ch_next == ce)) { 74 1.2.2.2 kardel ch->ch_next = TAILQ_NEXT(ce, ce_q); 75 1.2.2.2 kardel } 76 1.2.2.2 kardel TAILQ_REMOVE(&ch->ch_q, ce, ce_q); 77 1.2.2.2 kardel ch->ch_nentries--; 78 1.2.2.2 kardel simple_unlock(&ch->ch_lock); 79 1.2.2.2 kardel } 80 1.2.2.2 kardel 81 1.2.2.2 kardel static int 82 1.2.2.2 kardel callback_runone(struct callback_head *ch, void *arg) 83 1.2.2.2 kardel { 84 1.2.2.2 kardel struct callback_entry *ce; 85 1.2.2.2 kardel int result; 86 1.2.2.2 kardel 87 1.2.2.2 kardel KASSERT(ch->ch_nentries > 0); 88 1.2.2.2 kardel KASSERT(ch->ch_running > 0); 89 1.2.2.2 kardel 90 1.2.2.2 kardel ce = ch->ch_next; 91 1.2.2.2 kardel if (ce == NULL) { 92 1.2.2.2 kardel ce = TAILQ_FIRST(&ch->ch_q); 93 1.2.2.2 kardel } 94 1.2.2.2 kardel KASSERT(ce != NULL); 95 1.2.2.2 kardel result = (*ce->ce_func)(ce, ce->ce_obj, arg); 96 1.2.2.2 kardel ch->ch_next = TAILQ_NEXT(ce, ce_q); 97 1.2.2.2 kardel return result; 98 1.2.2.2 kardel } 99 1.2.2.2 kardel 100 1.2.2.2 kardel static void 101 1.2.2.2 kardel callback_run_enter(struct callback_head *ch) 102 1.2.2.2 kardel { 103 1.2.2.2 kardel 104 1.2.2.2 kardel simple_lock(&ch->ch_lock); 105 1.2.2.2 kardel ch->ch_running++; 106 1.2.2.2 kardel simple_unlock(&ch->ch_lock); 107 1.2.2.2 kardel } 108 1.2.2.2 kardel 109 1.2.2.2 kardel static void 110 1.2.2.2 kardel callback_run_leave(struct callback_head *ch) 111 1.2.2.2 kardel { 112 1.2.2.2 kardel 113 1.2.2.2 kardel simple_lock(&ch->ch_lock); 114 1.2.2.2 kardel KASSERT(ch->ch_running > 0); 115 1.2.2.2 kardel ch->ch_running--; 116 1.2.2.2 kardel if (ch->ch_running == 0 && (ch->ch_flags & CH_WANT) != 0) { 117 1.2.2.2 kardel ch->ch_flags &= ~CH_WANT; 118 1.2.2.2 kardel wakeup(&ch->ch_running); 119 1.2.2.2 kardel } 120 1.2.2.2 kardel simple_unlock(&ch->ch_lock); 121 1.2.2.2 kardel } 122 1.2.2.2 kardel 123 1.2.2.2 kardel int 124 1.2.2.2 kardel callback_run_roundrobin(struct callback_head *ch, void *arg) 125 1.2.2.2 kardel { 126 1.2.2.2 kardel int i; 127 1.2.2.2 kardel int n; 128 1.2.2.2 kardel int result = 0; 129 1.2.2.2 kardel 130 1.2.2.2 kardel callback_run_enter(ch); 131 1.2.2.2 kardel n = ch->ch_nentries; 132 1.2.2.2 kardel for (i = 0; i < n; i++) { 133 1.2.2.2 kardel result = callback_runone(ch, arg); 134 1.2.2.2 kardel if (result != CALLBACK_CHAIN_CONTINUE) { 135 1.2.2.2 kardel break; 136 1.2.2.2 kardel } 137 1.2.2.2 kardel } 138 1.2.2.2 kardel callback_run_leave(ch); 139 1.2.2.2 kardel 140 1.2.2.2 kardel return result; 141 1.2.2.2 kardel } 142
Indexes created Fri Sep 26 20:09:58 GMT 2025