X-Git-Url: https://git.saurik.com/apple/xnu.git/blobdiff_plain/316670eb35587141e969394ae8537d66b9211e80..cb3231590a3c94ab4375e2228bd5e86b0cf1ad7e:/bsd/net/raw_usrreq.c diff --git a/bsd/net/raw_usrreq.c b/bsd/net/raw_usrreq.c index 1284ca8c7..eb8521ed7 100644 --- a/bsd/net/raw_usrreq.c +++ b/bsd/net/raw_usrreq.c @@ -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 @@ -11,10 +11,10 @@ * 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@ */ /* @@ -73,25 +73,31 @@ #include -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, };