bsd/sys/eventhandler.h

   1 /*
   2  * Copyright (c) 2016-2018 Apple Inc. All rights reserved.
   3  *
   4  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
   5  *
   6  * This file contains Original Code and/or Modifications of Original Code
   7  * as defined in and that are subject to the Apple Public Source License
   8  * Version 2.0 (the 'License'). You may not use this file except in
   9  * compliance with the License. The rights granted to you under the License
  10  * may not be used to create, or enable the creation or redistribution of,
  11  * unlawful or unlicensed copies of an Apple operating system, or to
  12  * circumvent, violate, or enable the circumvention or violation of, any
  13  * terms of an Apple operating system software license agreement.
  14  *
  15  * Please obtain a copy of the License at
  16  * http://www.opensource.apple.com/apsl/ and read it before using this file.
  17  *
  18  * The Original Code and all software distributed under the License are
  19  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  20  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  21  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  22  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  23  * Please see the License for the specific language governing rights and
  24  * limitations under the License.
  25  *
  26  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
  27  */
  28 /*-
  29  * Copyright (c) 1999 Michael Smith <msmith@freebsd.org>
  30  * All rights reserved.
  31  *
  32  * Redistribution and use in source and binary forms, with or without
  33  * modification, are permitted provided that the following conditions
  34  * are met:
  35  * 1. Redistributions of source code must retain the above copyright
  36  *    notice, this list of conditions and the following disclaimer.
  37  * 2. Redistributions in binary form must reproduce the above copyright
  38  *    notice, this list of conditions and the following disclaimer in the
  39  *    documentation and/or other materials provided with the distribution.
  40  *
  41  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  42  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  43  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  44  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  45  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  46  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  47  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  48  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  49  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  50  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  51  * SUCH DAMAGE.
  52  *
  53  * $FreeBSD$
  54  */
  55
  56 #ifndef _SYS_EVENTHANDLER_H_
  57 #define _SYS_EVENTHANDLER_H_
  58
  59 #include <ptrauth.h>
  60
  61 #include <kern/locks.h>
  62 #include <sys/queue.h>
  63 #include <sys/cdefs.h>
  64 #include <sys/syslog.h>
  65 #include <uuid/uuid.h>
  66
  67 extern int evh_debug;
  68 extern lck_grp_t        *el_lock_grp;
  69 extern lck_attr_t       *el_lock_attr;
  70 extern struct eventhandler_entry_arg eventhandler_entry_dummy_arg;
  71
  72 struct eventhandler_lists_ctxt {
  73         TAILQ_HEAD(, eventhandler_list)  eventhandler_lists;
  74         int eventhandler_lists_initted;
  75         decl_lck_mtx_data(, eventhandler_mutex);
  76 };
  77
  78 struct eventhandler_entry_arg {
  79         uuid_t ee_fm_uuid;      /* Flow manager UUID */
  80         uuid_t ee_fr_uuid;      /* Flow route UUID */
  81 };
  82
  83 struct eventhandler_entry {
  84         TAILQ_ENTRY(eventhandler_entry) ee_link;
  85         int                             ee_priority;
  86 #define EHE_DEAD_PRIORITY       (-1)
  87         struct eventhandler_entry_arg   ee_arg;
  88 };
  89
  90 #define EVENTHANDLER_MAX_NAME   32
  91
  92 struct eventhandler_list {
  93         char                            el_name[EVENTHANDLER_MAX_NAME];
  94         int                             el_flags;
  95 #define EHL_INITTED     (1<<0)
  96         u_int                           el_runcount;
  97         decl_lck_mtx_data(, el_lock);
  98         TAILQ_ENTRY(eventhandler_list)  el_link;
  99         TAILQ_HEAD(, eventhandler_entry) el_entries;
 100 };
 101
 102 typedef struct eventhandler_entry       *eventhandler_tag;
 103
 104 #define EHL_LOCK_INIT(p)        lck_mtx_init(&(p)->el_lock, el_lock_grp, el_lock_attr)
 105 #define EHL_LOCK(p)             lck_mtx_lock(&(p)->el_lock)
 106 #define EHL_LOCK_SPIN(p)        lck_mtx_lock_spin(&(p)->el_lock)
 107 #define EHL_LOCK_CONVERT(p)     lck_mtx_convert_spin(&(p)->el_lock)
 108 #define EHL_UNLOCK(p)           lck_mtx_unlock(&(p)->el_lock)
 109 #define EHL_LOCK_ASSERT(p, x)   LCK_MTX_ASSERT(&(p)->el_lock, x)
 110 #define EHL_LOCK_DESTROY(p)     lck_mtx_destroy(&(p)->el_lock, el_lock_grp)
 111
 112 #define evhlog(x)       do { if (evh_debug >= 1) log x; } while (0)
 113
 114 /*
 115  * Macro to invoke the handlers for a given event.
 116  */
 117 #define _EVENTHANDLER_INVOKE(name, list, ...) do {                      \
 118         struct eventhandler_entry *_ep;                                 \
 119         struct eventhandler_entry_ ## name *_t;                         \
 120                                                                         \
 121         VERIFY((list)->el_flags & EHL_INITTED);                         \
 122         EHL_LOCK_ASSERT((list), LCK_MTX_ASSERT_OWNED);                  \
 123         (list)->el_runcount++;                                          \
 124         VERIFY((list)->el_runcount > 0);                                \
 125         evhlog((LOG_DEBUG, "eventhandler_invoke(\"" __STRING(name) "\")"));     \
 126         TAILQ_FOREACH(_ep, &((list)->el_entries), ee_link) {            \
 127                 if (_ep->ee_priority != EHE_DEAD_PRIORITY) {            \
 128                         EHL_UNLOCK((list));                             \
 129                         _t = (struct eventhandler_entry_ ## name *)_ep; \
 130                         evhlog((LOG_DEBUG, "eventhandler_invoke: executing %p", \
 131                             (void *)VM_KERNEL_UNSLIDE((void *)_t->eh_func)));   \
 132                         _t->eh_func(_ep->ee_arg , ## __VA_ARGS__);      \
 133                         EHL_LOCK_SPIN((list));                          \
 134                 }                                                       \
 135         }                                                               \
 136         VERIFY((list)->el_runcount > 0);                                \
 137         (list)->el_runcount--;                                          \
 138         if ((list)->el_runcount == 0) {                                 \
 139                 EHL_LOCK_CONVERT((list));                               \
 140                 eventhandler_prune_list(list);                          \
 141         }                                                               \
 142         EHL_UNLOCK((list));                                             \
 143 } while (0)
 144
 145 /*
 146  * Slow handlers are entirely dynamic; lists are created
 147  * when entries are added to them, and thus have no concept of "owner",
 148  *
 149  * Slow handlers need to be declared, but do not need to be defined. The
 150  * declaration must be in scope wherever the handler is to be invoked.
 151  */
 152 #define EVENTHANDLER_DECLARE(name, type)                                \
 153 struct eventhandler_entry_ ## name                                      \
 154 {                                                                       \
 155         struct eventhandler_entry       ee;                             \
 156         type                            eh_func;                        \
 157 };                                                                      \
 158 struct __hack
 159
 160 /*
 161  * XXX EVENTHANDLER_DEFINE by itself doesn't do much on XNU
 162  * All it does is that it declares the static eventhandler_tag
 163  * and defines an init routine that still needs to called to put the
 164  * event and callback on the list.
 165  */
 166 #define EVENTHANDLER_DEFINE(evthdlr_ref, name, func, arg, priority)                     \
 167         static eventhandler_tag name ## _tag;                           \
 168         static void name ## _evh_init(void *ctx)                        \
 169         {                                                               \
 170                 name ## _tag = EVENTHANDLER_REGISTER(evthdlr_ref, name, func, ctx,      \
 171                     priority);                                          \
 172         }                                                               \
 173         SYSINIT(name ## _evh_init, SI_SUB_CONFIGURE, SI_ORDER_ANY,      \
 174             name ## _evh_init, arg);                                    \
 175         struct __hack
 176
 177 #define EVENTHANDLER_INVOKE(evthdlr_ref, name, ...)                                     \
 178 do {                                                                    \
 179         struct eventhandler_list *_el;                                  \
 180                                                                         \
 181         if ((_el = eventhandler_find_list(evthdlr_ref, #name)) != NULL)                 \
 182                 _EVENTHANDLER_INVOKE(name, _el , ## __VA_ARGS__);       \
 183 } while (0)
 184
 185 #define EVENTHANDLER_REGISTER(evthdlr_ref, name, func, arg, priority)           \
 186         eventhandler_register(evthdlr_ref, NULL, #name, ptrauth_nop_cast(void *, &func), arg, priority)
 187
 188 #define EVENTHANDLER_DEREGISTER(evthdlr_ref, name, tag)                                 \
 189 do {                                                                    \
 190         struct eventhandler_list *_el;                                  \
 191                                                                         \
 192         if ((_el = eventhandler_find_list(evthdlr_ref, #name)) != NULL)         \
 193                 eventhandler_deregister(_el, tag);                      \
 194 } while(0)
 195
 196 void eventhandler_init(void);
 197 extern void eventhandler_reap_caches(boolean_t);
 198 eventhandler_tag eventhandler_register(struct eventhandler_lists_ctxt *evthdlr_lists_ctxt,
 199     struct eventhandler_list *list, const char *name, void *func, struct eventhandler_entry_arg arg, int priority);
 200 void eventhandler_deregister(struct eventhandler_list *list,
 201     eventhandler_tag tag);
 202 struct eventhandler_list *eventhandler_find_list(
 203         struct eventhandler_lists_ctxt *evthdlr_lists_ctxt, const char *name);
 204 void eventhandler_prune_list(struct eventhandler_list *list);
 205 void eventhandler_lists_ctxt_init(struct eventhandler_lists_ctxt *evthdlr_lists_ctxt);
 206 void eventhandler_lists_ctxt_destroy(struct eventhandler_lists_ctxt *evthdlr_lists_ctxt);
 207
 208 /* Generic priority levels */
 209 #define EVENTHANDLER_PRI_FIRST  0
 210 #define EVENTHANDLER_PRI_ANY    10000
 211 #define EVENTHANDLER_PRI_LAST   20000
 212
 213 #endif /* _SYS_EVENTHANDLER_H_ */