/ duct-tape / xnu / bsd / sys / eventhandler.h
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_ */