bsd/net/nwk_wq.c

   1 /*
   2  * Copyright (c) 2016 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 #include <stddef.h>
  29 #include <kern/debug.h>
  30 #include <kern/locks.h>
  31 #include <kern/thread.h>
  32 #include <kern/thread_call.h>
  33 #include <net/nwk_wq.h>
  34 #include <sys/proc_internal.h>
  35 #include <sys/systm.h>
  36 #include <sys/mcache.h>
  37
  38 MALLOC_DEFINE(M_NWKWQ, "nwkwq", "Network work-queue");
  39
  40 static TAILQ_HEAD(, nwk_wq_entry) nwk_wq_head;
  41 decl_lck_mtx_data(static, nwk_wq_lock);
  42
  43 /* Lock group and attributes */
  44 static lck_grp_attr_t *nwk_wq_lock_grp_attributes = NULL;
  45 static lck_grp_t *nwk_wq_lock_group = NULL;
  46
  47 /* Lock and lock attributes */
  48 static lck_attr_t *nwk_wq_lock_attributes = NULL;
  49 decl_lck_mtx_data(static, nwk_wq_lock);
  50
  51 /* Wait channel for Network work queue */
  52 static void *nwk_wq_waitch = NULL;
  53 static void nwk_wq_thread_func(void *, wait_result_t);
  54
  55 static int nwk_wq_thread_cont(int err);
  56 static void nwk_wq_thread_func(void *v, wait_result_t w);
  57
  58 void
  59 nwk_wq_init (void)
  60 {
  61         thread_t nwk_wq_thread = THREAD_NULL;
  62
  63         TAILQ_INIT(&nwk_wq_head);
  64         nwk_wq_lock_grp_attributes = lck_grp_attr_alloc_init();
  65         nwk_wq_lock_group = lck_grp_alloc_init("Network work queue lock",
  66             nwk_wq_lock_grp_attributes);
  67
  68         nwk_wq_lock_attributes = lck_attr_alloc_init();
  69         lck_mtx_init(&nwk_wq_lock, nwk_wq_lock_group, nwk_wq_lock_attributes);
  70         if (kernel_thread_start(nwk_wq_thread_func,
  71             NULL, &nwk_wq_thread) != KERN_SUCCESS) {
  72                 panic_plain("%s: couldn't create network work queue thread", __func__);
  73                 /* NOTREACHED */
  74         }
  75         thread_deallocate(nwk_wq_thread);
  76 }
  77
  78 static int
  79 nwk_wq_thread_cont(int err)
  80 {
  81         TAILQ_HEAD(, nwk_wq_entry) temp_nwk_wq_head;
  82         struct nwk_wq_entry *nwk_item;
  83         struct nwk_wq_entry *nwk_item_next;
  84
  85 #pragma unused(err)
  86         for (;;) {
  87                 nwk_item = NULL;
  88                 nwk_item_next = NULL;
  89                 TAILQ_INIT(&temp_nwk_wq_head);
  90
  91                 LCK_MTX_ASSERT(&nwk_wq_lock, LCK_MTX_ASSERT_OWNED);
  92                 while (TAILQ_FIRST(&nwk_wq_head) == NULL) {
  93                         (void) msleep0(&nwk_wq_waitch, &nwk_wq_lock,
  94                             (PZERO - 1), "nwk_wq_thread_cont", 0,
  95                             nwk_wq_thread_cont);
  96                         /* NOTREACHED */
  97                 }
  98
  99                 TAILQ_SWAP(&temp_nwk_wq_head, &nwk_wq_head, nwk_wq_entry, nwk_wq_link);
 100                 VERIFY(TAILQ_EMPTY(&nwk_wq_head));
 101                 lck_mtx_unlock(&nwk_wq_lock);
 102
 103                 VERIFY(TAILQ_FIRST(&temp_nwk_wq_head) != NULL);
 104                 TAILQ_FOREACH_SAFE(nwk_item, &temp_nwk_wq_head, nwk_wq_link, nwk_item_next) {
 105                         nwk_item->func(nwk_item->arg);
 106                         if (nwk_item->is_arg_managed == FALSE)
 107                                 FREE(nwk_item->arg, M_NWKWQ);
 108                         FREE(nwk_item, M_NWKWQ);
 109                 }
 110                 lck_mtx_lock(&nwk_wq_lock);
 111         }
 112 }
 113
 114 static void
 115 nwk_wq_thread_func(void *v, wait_result_t w)
 116 {
 117 #pragma unused(v, w)
 118         lck_mtx_lock(&nwk_wq_lock);
 119         (void) msleep0(&nwk_wq_waitch, &nwk_wq_lock,
 120             (PZERO - 1), "nwk_wq_thread_func", 0, nwk_wq_thread_cont);
 121         /*
 122          * msleep0() shouldn't have returned as PCATCH was not set;
 123          * therefore assert in this case.
 124          */
 125         lck_mtx_unlock(&nwk_wq_lock);
 126         VERIFY(0);
 127 }
 128
 129 void
 130 nwk_wq_enqueue(struct nwk_wq_entry *nwk_item)
 131 {
 132         lck_mtx_lock(&nwk_wq_lock);
 133         TAILQ_INSERT_TAIL(&nwk_wq_head, nwk_item, nwk_wq_link);
 134         lck_mtx_unlock(&nwk_wq_lock);
 135         wakeup((caddr_t)&nwk_wq_waitch);
 136 }
 137