xref: /netbsd-src/sys/kern/sys_socket.c (revision 288bb96063654ec504ca8732afc683d3ebc514b5)
1 /*	$NetBSD: sys_socket.c,v 1.64 2011/06/30 22:38:50 dyoung Exp $	*/
2 
3 /*-
4  * Copyright (c) 2008, 2009 The NetBSD Foundation, Inc.
5  * All rights reserved.
6  *
7  * This code is derived from software contributed to The NetBSD Foundation
8  * by Andrew Doran.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted provided that the following conditions
12  * are met:
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29  * POSSIBILITY OF SUCH DAMAGE.
30  */
31 
32 /*
33  * Copyright (c) 1982, 1986, 1990, 1993
34  *	The Regents of the University of California.  All rights reserved.
35  *
36  * Redistribution and use in source and binary forms, with or without
37  * modification, are permitted provided that the following conditions
38  * are met:
39  * 1. Redistributions of source code must retain the above copyright
40  *    notice, this list of conditions and the following disclaimer.
41  * 2. Redistributions in binary form must reproduce the above copyright
42  *    notice, this list of conditions and the following disclaimer in the
43  *    documentation and/or other materials provided with the distribution.
44  * 3. 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  *	@(#)sys_socket.c	8.3 (Berkeley) 2/14/95
61  */
62 
63 #include <sys/cdefs.h>
64 __KERNEL_RCSID(0, "$NetBSD: sys_socket.c,v 1.64 2011/06/30 22:38:50 dyoung Exp $");
65 
66 #include <sys/param.h>
67 #include <sys/systm.h>
68 #include <sys/systm.h>
69 #include <sys/file.h>
70 #include <sys/mbuf.h>
71 #include <sys/protosw.h>
72 #include <sys/socket.h>
73 #include <sys/socketvar.h>
74 #include <sys/ioctl.h>
75 #include <sys/stat.h>
76 #include <sys/poll.h>
77 #include <sys/proc.h>
78 #include <sys/kauth.h>
79 
80 #include <net/if.h>
81 #include <net/route.h>
82 
83 const struct fileops socketops = {
84 	.fo_read = soo_read,
85 	.fo_write = soo_write,
86 	.fo_ioctl = soo_ioctl,
87 	.fo_fcntl = soo_fcntl,
88 	.fo_poll = soo_poll,
89 	.fo_stat = soo_stat,
90 	.fo_close = soo_close,
91 	.fo_kqfilter = soo_kqfilter,
92 	.fo_restart = soo_restart,
93 };
94 
95 /* ARGSUSED */
96 int
97 soo_read(file_t *fp, off_t *offset, struct uio *uio, kauth_cred_t cred,
98 	 int flags)
99 {
100 	struct socket *so = fp->f_data;
101 	int error;
102 
103 	error = (*so->so_receive)(so, NULL, uio, NULL, NULL, NULL);
104 
105 	return error;
106 }
107 
108 /* ARGSUSED */
109 int
110 soo_write(file_t *fp, off_t *offset, struct uio *uio, kauth_cred_t cred,
111 	  int flags)
112 {
113 	struct socket *so = fp->f_data;
114 	int error;
115 
116 	error = (*so->so_send)(so, NULL, uio, NULL, NULL, 0, curlwp);
117 
118 	return error;
119 }
120 
121 int
122 soo_ioctl(file_t *fp, u_long cmd, void *data)
123 {
124 	struct socket *so = fp->f_data;
125 	int error = 0;
126 
127 	switch (cmd) {
128 
129 	case FIONBIO:
130 		/* No reason to lock and this call is made very often. */
131 		so->so_nbio = *(int *)data;
132 		break;
133 
134 	case FIOASYNC:
135 		solock(so);
136 		if (*(int *)data) {
137 			so->so_state |= SS_ASYNC;
138 			so->so_rcv.sb_flags |= SB_ASYNC;
139 			so->so_snd.sb_flags |= SB_ASYNC;
140 		} else {
141 			so->so_state &= ~SS_ASYNC;
142 			so->so_rcv.sb_flags &= ~SB_ASYNC;
143 			so->so_snd.sb_flags &= ~SB_ASYNC;
144 		}
145 		sounlock(so);
146 		break;
147 
148 	case FIONREAD:
149 		*(int *)data = so->so_rcv.sb_cc;
150 		break;
151 
152 	case FIONWRITE:
153 		*(int *)data = so->so_snd.sb_cc;
154 		break;
155 
156 	case FIONSPACE:
157 		/*
158 		 * See the comment around sbspace()'s definition
159 		 * in sys/socketvar.h in face of counts about maximum
160 		 * to understand the following test. We detect overflow
161 		 * and return zero.
162 		 */
163 		solock(so);
164 		if ((so->so_snd.sb_hiwat < so->so_snd.sb_cc)
165 		    || (so->so_snd.sb_mbmax < so->so_snd.sb_mbcnt))
166 			*(int *)data = 0;
167 		else
168 			*(int *)data = sbspace(&so->so_snd);
169 		sounlock(so);
170 		break;
171 
172 	case SIOCSPGRP:
173 	case FIOSETOWN:
174 	case TIOCSPGRP:
175 		error = fsetown(&so->so_pgid, cmd, data);
176 		break;
177 
178 	case SIOCGPGRP:
179 	case FIOGETOWN:
180 	case TIOCGPGRP:
181 		error = fgetown(so->so_pgid, cmd, data);
182 		break;
183 
184 	case SIOCATMARK:
185 		*(int *)data = (so->so_state&SS_RCVATMARK) != 0;
186 		break;
187 
188 	default:
189 		/*
190 		 * Interface/routing/protocol specific ioctls:
191 		 * interface and routing ioctls should have a
192 		 * different entry since a socket's unnecessary
193 		 */
194 		KERNEL_LOCK(1, NULL);
195 		if (IOCGROUP(cmd) == 'i')
196 			error = ifioctl(so, cmd, data, curlwp);
197 		else if (IOCGROUP(cmd) == 'r')
198 			error = rtioctl(cmd, data, curlwp);
199 		else {
200 			error = (*so->so_proto->pr_usrreq)(so, PRU_CONTROL,
201 			    (struct mbuf *)cmd, (struct mbuf *)data, NULL,
202 			     curlwp);
203 		}
204 		KERNEL_UNLOCK_ONE(NULL);
205 		break;
206 	}
207 
208 
209 	return error;
210 }
211 
212 int
213 soo_fcntl(file_t *fp, u_int cmd, void *data)
214 {
215 
216 	if (cmd == F_SETFL)
217 		return 0;
218 	else
219 		return EOPNOTSUPP;
220 }
221 
222 int
223 soo_poll(file_t *fp, int events)
224 {
225 
226 	return sopoll(fp->f_data, events);
227 }
228 
229 int
230 soo_stat(file_t *fp, struct stat *ub)
231 {
232 	struct socket *so = fp->f_data;
233 	int error;
234 
235 	memset(ub, 0, sizeof(*ub));
236 	ub->st_mode = S_IFSOCK;
237 
238 	solock(so);
239 	error = (*so->so_proto->pr_usrreq)(so, PRU_SENSE,
240 	    (struct mbuf *)ub, NULL, NULL, curlwp);
241 	sounlock(so);
242 
243 	return error;
244 }
245 
246 /* ARGSUSED */
247 int
248 soo_close(file_t *fp)
249 {
250 	int error = 0;
251 
252 	if (fp->f_data)
253 		error = soclose(fp->f_data);
254 	fp->f_data = 0;
255 
256 	return error;
257 }
258 
259 void
260 soo_restart(file_t *fp)
261 {
262 
263 	sorestart(fp->f_data);
264 }
265