]> git.saurik.com Git - apple/xnu.git/blobdiff - bsd/net/raw_usrreq.c
xnu-6153.11.26.tar.gz
[apple/xnu.git] / bsd / net / raw_usrreq.c
index 1284ca8c775229956c1f7030fb458f0b224fa343..eb8521ed7738ae2270fd42ac5e97b64b8fda542b 100644 (file)
@@ -1,8 +1,8 @@
 /*
- * Copyright (c) 2000-2010 Apple Inc. All rights reserved.
+ * Copyright (c) 2000-2012 Apple Inc. All rights reserved.
  *
  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
- * 
+ *
  * This file contains Original Code and/or Modifications of Original Code
  * as defined in and that are subject to the Apple Public Source License
  * Version 2.0 (the 'License'). You may not use this file except in
  * unlawful or unlicensed copies of an Apple operating system, or to
  * circumvent, violate, or enable the circumvention or violation of, any
  * terms of an Apple operating system software license agreement.
- * 
+ *
  * Please obtain a copy of the License at
  * http://www.opensource.apple.com/apsl/ and read it before using this file.
- * 
+ *
  * The Original Code and all software distributed under the License are
  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
@@ -22,7 +22,7 @@
  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  * Please see the License for the specific language governing rights and
  * limitations under the License.
- * 
+ *
  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
  */
 /*
 
 #include <net/raw_cb.h>
 
-decl_lck_mtx_data(,raw_mtx_data);      /*### global raw cb mutex for now */
-lck_mtx_t      *raw_mtx = &raw_mtx_data;
-lck_attr_t     *raw_mtx_attr;
-lck_grp_t      *raw_mtx_grp;
-lck_grp_attr_t         *raw_mtx_grp_attr;
+decl_lck_mtx_data(, raw_mtx_data);       /*### global raw cb mutex for now */
+lck_mtx_t       *raw_mtx = &raw_mtx_data;
+lck_attr_t      *raw_mtx_attr;
+lck_grp_t       *raw_mtx_grp;
+lck_grp_attr_t  *raw_mtx_grp_attr;
 /*
  * Initialize raw connection block q.
  */
 void
-raw_init(void)
+raw_init(struct protosw *pp, struct domain *dp)
 {
-       raw_mtx_grp_attr = lck_grp_attr_alloc_init();
+#pragma unused(pp, dp)
+       static int raw_initialized = 0;
 
-       raw_mtx_grp = lck_grp_alloc_init("rawcb", raw_mtx_grp_attr);
+       /* This is called by key_init as well, so do it only once */
+       if (!raw_initialized) {
+               raw_initialized = 1;
 
-       raw_mtx_attr = lck_attr_alloc_init();
+               raw_mtx_grp_attr = lck_grp_attr_alloc_init();
+               raw_mtx_grp = lck_grp_alloc_init("rawcb", raw_mtx_grp_attr);
+               raw_mtx_attr = lck_attr_alloc_init();
 
-       lck_mtx_init(raw_mtx, raw_mtx_grp, raw_mtx_attr);
-       LIST_INIT(&rawcb_list);
+               lck_mtx_init(raw_mtx, raw_mtx_grp, raw_mtx_attr);
+               LIST_INIT(&rawcb_list);
+       }
 }
 
 
@@ -105,7 +111,7 @@ raw_init(void)
  */
 void
 raw_input(struct mbuf *m0, struct sockproto *proto, struct sockaddr *src,
-         struct sockaddr *dst)
+    struct sockaddr *dst)
 {
        struct rawcb *rp;
        struct mbuf *m = m0;
@@ -113,18 +119,20 @@ raw_input(struct mbuf *m0, struct sockproto *proto, struct sockaddr *src,
        struct socket *last;
        int error;
 
-//####LD raw_input is called from many places, input & output path. We have to assume the 
+//####LD raw_input is called from many places, input & output path. We have to assume the
 //####LD socket we'll find and need to append to is unlocked.
 //####LD calls from the output (locked) path need to make sure the socket is not locked when
 //####LD we call in raw_input
        last = NULL;
        lck_mtx_lock(raw_mtx);
        LIST_FOREACH(rp, &rawcb_list, list) {
-               if (rp->rcb_proto.sp_family != proto->sp_family)
+               if (rp->rcb_proto.sp_family != proto->sp_family) {
                        continue;
-               if (rp->rcb_proto.sp_protocol  &&
-                   rp->rcb_proto.sp_protocol != proto->sp_protocol)
+               }
+               if (rp->rcb_proto.sp_protocol &&
+                   rp->rcb_proto.sp_protocol != proto->sp_protocol) {
                        continue;
+               }
                /*
                 * We assume the lower level routines have
                 * placed the address in a canonical format
@@ -133,12 +141,14 @@ raw_input(struct mbuf *m0, struct sockproto *proto, struct sockaddr *src,
                 * Note that if the lengths are not the same
                 * the comparison will fail at the first byte.
                 */
-#define        equal(a1, a2) \
+#define equal(a1, a2) \
   (bcmp((caddr_t)(a1), (caddr_t)(a2), a1->sa_len) == 0)
-               if (rp->rcb_laddr && !equal(rp->rcb_laddr, dst))
+               if (rp->rcb_laddr && !equal(rp->rcb_laddr, dst)) {
                        continue;
-               if (rp->rcb_faddr && !equal(rp->rcb_faddr, src))
+               }
+               if (rp->rcb_faddr && !equal(rp->rcb_faddr, src)) {
                        continue;
+               }
                if (last) {
                        struct mbuf *n;
                        n = m_copy(m, 0, (int)M_COPYALL);
@@ -162,18 +172,20 @@ raw_input(struct mbuf *m0, struct sockproto *proto, struct sockaddr *src,
                        sockets++;
                }
                socket_unlock(last, 1);
-       } else
+       } else {
                m_freem(m);
+       }
        lck_mtx_unlock(raw_mtx);
 }
 
 /*ARGSUSED*/
 void
-raw_ctlinput(int cmd, __unused struct sockaddr *arg, __unused void *dummy)
+raw_ctlinput(int cmd, __unused struct sockaddr *arg, __unused void *dummy,
+    __unused struct ifnet *ifp)
 {
-
-       if (cmd < 0 || cmd > PRC_NCMDS)
+       if (cmd < 0 || cmd >= PRC_NCMDS) {
                return;
+       }
        /* INCOMPLETE */
 }
 
@@ -183,14 +195,16 @@ raw_uabort(struct socket *so)
        struct rawcb *rp = sotorawcb(so);
 
        lck_mtx_t * mutex_held;
-       if (so->so_proto->pr_getlock != NULL)
+       if (so->so_proto->pr_getlock != NULL) {
                mutex_held = (*so->so_proto->pr_getlock)(so, 0);
-       else
+       } else {
                mutex_held = so->so_proto->pr_domain->dom_mtx;
-       lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
+       }
+       LCK_MTX_ASSERT(mutex_held, LCK_MTX_ASSERT_OWNED);
 
-       if (rp == 0)
+       if (rp == 0) {
                return EINVAL;
+       }
        raw_disconnect(rp);
        sofree(so);
        soisdisconnected(so);
@@ -204,10 +218,12 @@ raw_uattach(struct socket *so, int proto, __unused struct proc *p)
 {
        struct rawcb *rp = sotorawcb(so);
 
-       if (rp == 0)
+       if (rp == 0) {
                return EINVAL;
-               if ((so->so_state & SS_PRIV) == 0)
-                       return (EPERM);
+       }
+       if ((so->so_state & SS_PRIV) == 0) {
+               return EPERM;
+       }
        return raw_attach(so, proto);
 }
 
@@ -232,13 +248,15 @@ raw_udetach(struct socket *so)
        struct rawcb *rp = sotorawcb(so);
 
        lck_mtx_t * mutex_held;
-       if (so->so_proto->pr_getlock != NULL)
+       if (so->so_proto->pr_getlock != NULL) {
                mutex_held = (*so->so_proto->pr_getlock)(so, 0);
-       else
+       } else {
                mutex_held = so->so_proto->pr_domain->dom_mtx;
-       lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
-       if (rp == 0)
+       }
+       LCK_MTX_ASSERT(mutex_held, LCK_MTX_ASSERT_OWNED);
+       if (rp == 0) {
                return EINVAL;
+       }
 
        raw_detach(rp);
        return 0;
@@ -249,8 +267,9 @@ raw_udisconnect(struct socket *so)
 {
        struct rawcb *rp = sotorawcb(so);
 
-       if (rp == 0)
+       if (rp == 0) {
                return EINVAL;
+       }
        if (rp->rcb_faddr == 0) {
                return ENOTCONN;
        }
@@ -266,8 +285,9 @@ raw_upeeraddr(struct socket *so, struct sockaddr **nam)
 {
        struct rawcb *rp = sotorawcb(so);
 
-       if (rp == 0)
+       if (rp == 0) {
                return EINVAL;
+       }
        if (rp->rcb_faddr == 0) {
                return ENOTCONN;
        }
@@ -280,17 +300,18 @@ raw_upeeraddr(struct socket *so, struct sockaddr **nam)
 
 static int
 raw_usend(struct socket *so, int flags, struct mbuf *m,
-         struct sockaddr *nam, struct mbuf *control, __unused struct proc *p)
+    struct sockaddr *nam, struct mbuf *control, __unused struct proc *p)
 {
        int error;
        struct rawcb *rp = sotorawcb(so);
 
        lck_mtx_t * mutex_held;
-       if (so->so_proto->pr_getlock != NULL)
+       if (so->so_proto->pr_getlock != NULL) {
                mutex_held = (*so->so_proto->pr_getlock)(so, 0);
-       else
+       } else {
                mutex_held = so->so_proto->pr_domain->dom_mtx;
-       lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
+       }
+       LCK_MTX_ASSERT(mutex_held, LCK_MTX_ASSERT_OWNED);
 
        if (rp == 0) {
                error = EINVAL;
@@ -323,12 +344,14 @@ raw_usend(struct socket *so, int flags, struct mbuf *m,
        }
        error = (*so->so_proto->pr_output)(m, so);
        m = NULL;
-       if (nam)
+       if (nam) {
                rp->rcb_faddr = NULL;
+       }
 release:
-       if (m != NULL)
+       if (m != NULL) {
                m_freem(m);
-       return (error);
+       }
+       return error;
 }
 
 /* pru_sense is null */
@@ -338,14 +361,16 @@ raw_ushutdown(struct socket *so)
 {
        struct rawcb *rp = sotorawcb(so);
        lck_mtx_t * mutex_held;
-       if (so->so_proto->pr_getlock != NULL)
+       if (so->so_proto->pr_getlock != NULL) {
                mutex_held = (*so->so_proto->pr_getlock)(so, 0);
-       else
+       } else {
                mutex_held = so->so_proto->pr_domain->dom_mtx;
-       lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
+       }
+       LCK_MTX_ASSERT(mutex_held, LCK_MTX_ASSERT_OWNED);
 
-       if (rp == 0)
+       if (rp == 0) {
                return EINVAL;
+       }
        socantsendmore(so);
        return 0;
 }
@@ -355,18 +380,27 @@ raw_usockaddr(struct socket *so, struct sockaddr **nam)
 {
        struct rawcb *rp = sotorawcb(so);
 
-       if (rp == 0)
+       if (rp == 0) {
                return EINVAL;
-       if (rp->rcb_laddr == 0)
+       }
+       if (rp->rcb_laddr == 0) {
                return EINVAL;
+       }
        *nam = dup_sockaddr(rp->rcb_laddr, 1);
        return 0;
 }
 
 struct pr_usrreqs raw_usrreqs = {
-       raw_uabort, pru_accept_notsupp, raw_uattach, raw_ubind, raw_uconnect,
-       pru_connect2_notsupp, pru_control_notsupp, raw_udetach, 
-       raw_udisconnect, pru_listen_notsupp, raw_upeeraddr, pru_rcvd_notsupp,
-       pru_rcvoob_notsupp, raw_usend, pru_sense_null, raw_ushutdown,
-       raw_usockaddr, sosend, soreceive, pru_sopoll_notsupp
+       .pru_abort =            raw_uabort,
+       .pru_attach =           raw_uattach,
+       .pru_bind =             raw_ubind,
+       .pru_connect =          raw_uconnect,
+       .pru_detach =           raw_udetach,
+       .pru_disconnect =       raw_udisconnect,
+       .pru_peeraddr =         raw_upeeraddr,
+       .pru_send =             raw_usend,
+       .pru_shutdown =         raw_ushutdown,
+       .pru_sockaddr =         raw_usockaddr,
+       .pru_sosend =           sosend,
+       .pru_soreceive =        soreceive,
 };