xnu-792.6.76.tar.gz

[apple/xnu.git] / bsd / net / raw_usrreq.c

diff --git a/bsd/net/raw_usrreq.c b/bsd/net/raw_usrreq.c
index 217d0c6d20423966043eb7695e9bb5fb6a4fc97f..d9bf972175106371628171686c16bb891ef727ca 100644 (file)
--- a/bsd/net/raw_usrreq.c
+++ b/bsd/net/raw_usrreq.c
@@ -3,22 +3,19 @@
  *
  * @APPLE_LICENSE_HEADER_START@
  * 
  *
  * @APPLE_LICENSE_HEADER_START@
  * 

- * Copyright (c) 1999-2003 Apple Computer, Inc.  All Rights Reserved.
+ * The contents of this file constitute Original Code as defined in and
+ * are subject to the Apple Public Source License Version 1.1 (the
+ * "License").  You may not use this file except in compliance with the
+ * License.  Please obtain a copy of the License at
+ * http://www.apple.com/publicsource and read it before using this file.

  * 
  * 

- * 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
- * compliance with the License. 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
+ * This 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,
  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,

- * 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.
+ * FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT.  Please see the
+ * License for the specific language governing rights and limitations
+ * under the License.

  * 
  * @APPLE_LICENSE_HEADER_END@
  */
  * 
  * @APPLE_LICENSE_HEADER_END@
  */


@@ -62,18 +59,38 @@
 #include <sys/systm.h>
 #include <sys/mbuf.h>
 #include <sys/proc.h>
 #include <sys/systm.h>
 #include <sys/mbuf.h>
 #include <sys/proc.h>

+#include <sys/domain.h>

 #include <sys/protosw.h>
 #include <sys/socket.h>
 #include <sys/socketvar.h>
 #include <sys/protosw.h>
 #include <sys/socket.h>
 #include <sys/socketvar.h>

+#include <kern/locks.h>

 
 #include <net/raw_cb.h>
 
 
 #include <net/raw_cb.h>
 

+lck_mtx_t      *raw_mtx;       /*### global raw cb mutex for now */
+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()
 {
 /*
  * Initialize raw connection block q.
  */
 void
 raw_init()
 {

+       raw_mtx_grp_attr = lck_grp_attr_alloc_init();
+
+       lck_grp_attr_setdefault(raw_mtx_grp_attr);
+
+       raw_mtx_grp = lck_grp_alloc_init("rawcb", raw_mtx_grp_attr);
+
+       raw_mtx_attr = lck_attr_alloc_init();
+
+       lck_attr_setdefault(raw_mtx_attr);
+
+       if ((raw_mtx = lck_mtx_alloc_init(raw_mtx_grp, raw_mtx_attr)) == NULL) {
+               printf("raw_init: can't alloc raw_mtx\n");
+               return;
+       }

        LIST_INIT(&rawcb_list);
 }
 
        LIST_INIT(&rawcb_list);
 }
 


@@ -96,8 +113,14 @@ raw_input(m0, proto, src, dst)
        register struct mbuf *m = m0;
        register int sockets = 0;
        struct socket *last;
        register struct mbuf *m = m0;
        register int sockets = 0;
        struct socket *last;

+       int error;

 
 

+//####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 = 0;
        last = 0;

+       lck_mtx_lock(raw_mtx);

        LIST_FOREACH(rp, &rawcb_list, list) {
                if (rp->rcb_proto.sp_family != proto->sp_family)
                        continue;
        LIST_FOREACH(rp, &rawcb_list, list) {
                if (rp->rcb_proto.sp_family != proto->sp_family)
                        continue;


@@ -122,28 +145,28 @@ raw_input(m0, proto, src, dst)
                        struct mbuf *n;
                        n = m_copy(m, 0, (int)M_COPYALL);
                        if (n) {
                        struct mbuf *n;
                        n = m_copy(m, 0, (int)M_COPYALL);
                        if (n) {

+                               socket_lock(last, 1);

                                if (sbappendaddr(&last->so_rcv, src,
                                if (sbappendaddr(&last->so_rcv, src,

-                                   n, (struct mbuf *)0) == 0)
-                                       /* should notify about lost packet */
-                                       m_freem(n);
-                               else {
+                                   n, (struct mbuf *)0, &error) != 0) {

                                        sorwakeup(last);
                                        sockets++;
                                }
                                        sorwakeup(last);
                                        sockets++;
                                }

+                               socket_unlock(last, 1);

                        }
                }
                last = rp->rcb_socket;
        }
        if (last) {
                        }
                }
                last = rp->rcb_socket;
        }
        if (last) {

+               socket_lock(last, 1);

                if (sbappendaddr(&last->so_rcv, src,
                if (sbappendaddr(&last->so_rcv, src,

-                   m, (struct mbuf *)0) == 0)
-                       m_freem(m);
-               else {
+                   m, (struct mbuf *)0, &error) != 0) {

                        sorwakeup(last);
                        sockets++;
                }
                        sorwakeup(last);
                        sockets++;
                }

+               socket_unlock(last, 1);

        } else
                m_freem(m);
        } else
                m_freem(m);

+       lck_mtx_unlock(raw_mtx);

 }
 
 /*ARGSUSED*/
 }
 
 /*ARGSUSED*/


@@ -164,6 +187,13 @@ raw_uabort(struct socket *so)
 {
        struct rawcb *rp = sotorawcb(so);
 
 {
        struct rawcb *rp = sotorawcb(so);
 

+       lck_mtx_t * mutex_held;
+       if (so->so_proto->pr_getlock != NULL)
+               mutex_held = (*so->so_proto->pr_getlock)(so, 0);
+       else
+               mutex_held = so->so_proto->pr_domain->dom_mtx;
+       lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
+

        if (rp == 0)
                return EINVAL;
        raw_disconnect(rp);
        if (rp == 0)
                return EINVAL;
        raw_disconnect(rp);


@@ -178,7 +208,9 @@ static int
 raw_uattach(struct socket *so, int proto, struct proc *p)
 {
        struct rawcb *rp = sotorawcb(so);
 raw_uattach(struct socket *so, int proto, struct proc *p)
 {
        struct rawcb *rp = sotorawcb(so);

+#ifndef __APPLE__

        int error;
        int error;

+#endif

 
        if (rp == 0)
                return EINVAL;
 
        if (rp == 0)
                return EINVAL;


@@ -212,6 +244,12 @@ raw_udetach(struct socket *so)
 {
        struct rawcb *rp = sotorawcb(so);
 
 {
        struct rawcb *rp = sotorawcb(so);
 

+       lck_mtx_t * mutex_held;
+       if (so->so_proto->pr_getlock != NULL)
+               mutex_held = (*so->so_proto->pr_getlock)(so, 0);
+       else
+               mutex_held = so->so_proto->pr_domain->dom_mtx;
+       lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);

        if (rp == 0)
                return EINVAL;
 
        if (rp == 0)
                return EINVAL;
 


@@ -260,6 +298,13 @@ raw_usend(struct socket *so, int flags, struct mbuf *m,
        int error;
        struct rawcb *rp = sotorawcb(so);
 
        int error;
        struct rawcb *rp = sotorawcb(so);
 

+       lck_mtx_t * mutex_held;
+       if (so->so_proto->pr_getlock != NULL)
+               mutex_held = (*so->so_proto->pr_getlock)(so, 0);
+       else
+               mutex_held = so->so_proto->pr_domain->dom_mtx;
+       lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
+

        if (rp == 0) {
                error = EINVAL;
                goto release;
        if (rp == 0) {
                error = EINVAL;
                goto release;


@@ -300,6 +345,12 @@ static int
 raw_ushutdown(struct socket *so)
 {
        struct rawcb *rp = sotorawcb(so);
 raw_ushutdown(struct socket *so)
 {
        struct rawcb *rp = sotorawcb(so);

+       lck_mtx_t * mutex_held;
+       if (so->so_proto->pr_getlock != NULL)
+               mutex_held = (*so->so_proto->pr_getlock)(so, 0);
+       else
+               mutex_held = so->so_proto->pr_domain->dom_mtx;
+       lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);

 
        if (rp == 0)
                return EINVAL;
 
        if (rp == 0)
                return EINVAL;


@@ -325,5 +376,5 @@ struct pr_usrreqs raw_usrreqs = {
        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,
        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, sopoll
+       raw_usockaddr, sosend, soreceive, pru_sopoll_notsupp

 };
 };