1/*
2 * Copyright (c) 2000-2012 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) 1980, 1986, 1993
30 *	The Regents of the University of California.  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 * 3. All advertising materials mentioning features or use of this software
41 *    must display the following acknowledgement:
42 *	This product includes software developed by the University of
43 *	California, Berkeley and its contributors.
44 * 4. Neither the name of the University nor the names of its contributors
45 *    may be used to endorse or promote products derived from this software
46 *    without specific prior written permission.
47 *
48 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
49 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
50 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
51 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
52 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
53 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
54 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
55 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
56 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
57 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
58 * SUCH DAMAGE.
59 *
60 *	@(#)raw_usrreq.c	8.1 (Berkeley) 6/10/93
61 * $FreeBSD: src/sys/net/raw_usrreq.c,v 1.18 1999/08/28 00:48:28 peter Exp $
62 */
63
64#include <sys/param.h>
65#include <sys/systm.h>
66#include <sys/mbuf.h>
67#include <sys/proc.h>
68#include <sys/domain.h>
69#include <sys/protosw.h>
70#include <sys/socket.h>
71#include <sys/socketvar.h>
72#include <kern/locks.h>
73
74#include <net/raw_cb.h>
75
76decl_lck_mtx_data(,raw_mtx_data);	/*### global raw cb mutex for now */
77lck_mtx_t 	*raw_mtx = &raw_mtx_data;
78lck_attr_t 	*raw_mtx_attr;
79lck_grp_t 	*raw_mtx_grp;
80lck_grp_attr_t 	*raw_mtx_grp_attr;
81/*
82 * Initialize raw connection block q.
83 */
84void
85raw_init(struct protosw *pp, struct domain *dp)
86{
87#pragma unused(pp, dp)
88	static int raw_initialized = 0;
89
90	/* This is called by key_init as well, so do it only once */
91	if (!raw_initialized) {
92		raw_initialized = 1;
93
94		raw_mtx_grp_attr = lck_grp_attr_alloc_init();
95		raw_mtx_grp = lck_grp_alloc_init("rawcb", raw_mtx_grp_attr);
96		raw_mtx_attr = lck_attr_alloc_init();
97
98		lck_mtx_init(raw_mtx, raw_mtx_grp, raw_mtx_attr);
99		LIST_INIT(&rawcb_list);
100	}
101}
102
103
104/*
105 * Raw protocol input routine.  Find the socket
106 * associated with the packet(s) and move them over.  If
107 * nothing exists for this packet, drop it.
108 */
109/*
110 * Raw protocol interface.
111 */
112void
113raw_input(struct mbuf *m0, struct sockproto *proto, struct sockaddr *src,
114	  struct sockaddr *dst)
115{
116	struct rawcb *rp;
117	struct mbuf *m = m0;
118	int sockets = 0;
119	struct socket *last;
120	int error;
121
122//####LD raw_input is called from many places, input & output path. We have to assume the
123//####LD socket we'll find and need to append to is unlocked.
124//####LD calls from the output (locked) path need to make sure the socket is not locked when
125//####LD we call in raw_input
126	last = NULL;
127	lck_mtx_lock(raw_mtx);
128	LIST_FOREACH(rp, &rawcb_list, list) {
129		if (rp->rcb_proto.sp_family != proto->sp_family)
130			continue;
131		if (rp->rcb_proto.sp_protocol  &&
132		    rp->rcb_proto.sp_protocol != proto->sp_protocol)
133			continue;
134		/*
135		 * We assume the lower level routines have
136		 * placed the address in a canonical format
137		 * suitable for a structure comparison.
138		 *
139		 * Note that if the lengths are not the same
140		 * the comparison will fail at the first byte.
141		 */
142#define	equal(a1, a2) \
143  (bcmp((caddr_t)(a1), (caddr_t)(a2), a1->sa_len) == 0)
144		if (rp->rcb_laddr && !equal(rp->rcb_laddr, dst))
145			continue;
146		if (rp->rcb_faddr && !equal(rp->rcb_faddr, src))
147			continue;
148		if (last) {
149			struct mbuf *n;
150			n = m_copy(m, 0, (int)M_COPYALL);
151			if (n) {
152				socket_lock(last, 1);
153				if (sbappendaddr(&last->so_rcv, src,
154				    n, (struct mbuf *)0, &error) != 0) {
155					sorwakeup(last);
156					sockets++;
157				}
158				socket_unlock(last, 1);
159			}
160		}
161		last = rp->rcb_socket;
162	}
163	if (last) {
164		socket_lock(last, 1);
165		if (sbappendaddr(&last->so_rcv, src,
166		    m, (struct mbuf *)0, &error) != 0) {
167			sorwakeup(last);
168			sockets++;
169		}
170		socket_unlock(last, 1);
171	} else
172		m_freem(m);
173	lck_mtx_unlock(raw_mtx);
174}
175
176/*ARGSUSED*/
177void
178raw_ctlinput(int cmd, __unused struct sockaddr *arg, __unused void *dummy)
179{
180
181	if (cmd < 0 || cmd > PRC_NCMDS)
182		return;
183	/* INCOMPLETE */
184}
185
186static int
187raw_uabort(struct socket *so)
188{
189	struct rawcb *rp = sotorawcb(so);
190
191	lck_mtx_t * mutex_held;
192	if (so->so_proto->pr_getlock != NULL)
193		mutex_held = (*so->so_proto->pr_getlock)(so, 0);
194	else
195		mutex_held = so->so_proto->pr_domain->dom_mtx;
196	lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
197
198	if (rp == 0)
199		return EINVAL;
200	raw_disconnect(rp);
201	sofree(so);
202	soisdisconnected(so);
203	return 0;
204}
205
206/* pru_accept is EOPNOTSUPP */
207
208static int
209raw_uattach(struct socket *so, int proto, __unused struct proc *p)
210{
211	struct rawcb *rp = sotorawcb(so);
212
213	if (rp == 0)
214		return EINVAL;
215		if ((so->so_state & SS_PRIV) == 0)
216			return (EPERM);
217	return raw_attach(so, proto);
218}
219
220static int
221raw_ubind(__unused struct socket *so, __unused struct sockaddr *nam, __unused struct proc *p)
222{
223	return EINVAL;
224}
225
226static int
227raw_uconnect(__unused struct socket *so, __unused struct sockaddr *nam, __unused struct proc *p)
228{
229	return EINVAL;
230}
231
232/* pru_connect2 is EOPNOTSUPP */
233/* pru_control is EOPNOTSUPP */
234
235static int
236raw_udetach(struct socket *so)
237{
238	struct rawcb *rp = sotorawcb(so);
239
240	lck_mtx_t * mutex_held;
241	if (so->so_proto->pr_getlock != NULL)
242		mutex_held = (*so->so_proto->pr_getlock)(so, 0);
243	else
244		mutex_held = so->so_proto->pr_domain->dom_mtx;
245	lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
246	if (rp == 0)
247		return EINVAL;
248
249	raw_detach(rp);
250	return 0;
251}
252
253static int
254raw_udisconnect(struct socket *so)
255{
256	struct rawcb *rp = sotorawcb(so);
257
258	if (rp == 0)
259		return EINVAL;
260	if (rp->rcb_faddr == 0) {
261		return ENOTCONN;
262	}
263	raw_disconnect(rp);
264	soisdisconnected(so);
265	return 0;
266}
267
268/* pru_listen is EOPNOTSUPP */
269
270static int
271raw_upeeraddr(struct socket *so, struct sockaddr **nam)
272{
273	struct rawcb *rp = sotorawcb(so);
274
275	if (rp == 0)
276		return EINVAL;
277	if (rp->rcb_faddr == 0) {
278		return ENOTCONN;
279	}
280	*nam = dup_sockaddr(rp->rcb_faddr, 1);
281	return 0;
282}
283
284/* pru_rcvd is EOPNOTSUPP */
285/* pru_rcvoob is EOPNOTSUPP */
286
287static int
288raw_usend(struct socket *so, int flags, struct mbuf *m,
289	  struct sockaddr *nam, struct mbuf *control, __unused struct proc *p)
290{
291	int error;
292	struct rawcb *rp = sotorawcb(so);
293
294	lck_mtx_t * mutex_held;
295	if (so->so_proto->pr_getlock != NULL)
296		mutex_held = (*so->so_proto->pr_getlock)(so, 0);
297	else
298		mutex_held = so->so_proto->pr_domain->dom_mtx;
299	lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
300
301	if (rp == 0) {
302		error = EINVAL;
303		goto release;
304	}
305
306	if (flags & PRUS_OOB) {
307		error = EOPNOTSUPP;
308		goto release;
309	}
310
311	if (so->so_proto->pr_output == NULL) {
312		error = EOPNOTSUPP;
313		goto release;
314	}
315
316	if (control && control->m_len) {
317		error = EOPNOTSUPP;
318		goto release;
319	}
320	if (nam) {
321		if (rp->rcb_faddr) {
322			error = EISCONN;
323			goto release;
324		}
325		rp->rcb_faddr = nam;
326	} else if (rp->rcb_faddr == 0) {
327		error = ENOTCONN;
328		goto release;
329	}
330	error = (*so->so_proto->pr_output)(m, so);
331	m = NULL;
332	if (nam)
333		rp->rcb_faddr = NULL;
334release:
335	if (m != NULL)
336		m_freem(m);
337	return (error);
338}
339
340/* pru_sense is null */
341
342static int
343raw_ushutdown(struct socket *so)
344{
345	struct rawcb *rp = sotorawcb(so);
346	lck_mtx_t * mutex_held;
347	if (so->so_proto->pr_getlock != NULL)
348		mutex_held = (*so->so_proto->pr_getlock)(so, 0);
349	else
350		mutex_held = so->so_proto->pr_domain->dom_mtx;
351	lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
352
353	if (rp == 0)
354		return EINVAL;
355	socantsendmore(so);
356	return 0;
357}
358
359static int
360raw_usockaddr(struct socket *so, struct sockaddr **nam)
361{
362	struct rawcb *rp = sotorawcb(so);
363
364	if (rp == 0)
365		return EINVAL;
366	if (rp->rcb_laddr == 0)
367		return EINVAL;
368	*nam = dup_sockaddr(rp->rcb_laddr, 1);
369	return 0;
370}
371
372struct pr_usrreqs raw_usrreqs = {
373	.pru_abort =		raw_uabort,
374	.pru_attach =		raw_uattach,
375	.pru_bind =		raw_ubind,
376	.pru_connect =		raw_uconnect,
377	.pru_detach =		raw_udetach,
378	.pru_disconnect =	raw_udisconnect,
379	.pru_peeraddr =		raw_upeeraddr,
380	.pru_send =		raw_usend,
381	.pru_shutdown =		raw_ushutdown,
382	.pru_sockaddr =		raw_usockaddr,
383	.pru_sosend =		sosend,
384	.pru_soreceive =	soreceive,
385};
386