FreeBSD/Linux Kernel Cross Reference
sys/kern/sys_socket.c
1 /* $NetBSD: sys_socket.c,v 1.43 2005/02/26 21:34:55 perry Exp $ */
2
3 /*
4 * Copyright (c) 1982, 1986, 1990, 1993
5 * The Regents of the University of California. All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 * 3. Neither the name of the University nor the names of its contributors
16 * may be used to endorse or promote products derived from this software
17 * without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 * SUCH DAMAGE.
30 *
31 * @(#)sys_socket.c 8.3 (Berkeley) 2/14/95
32 */
33
34 #include <sys/cdefs.h>
35 __KERNEL_RCSID(0, "$NetBSD: sys_socket.c,v 1.43 2005/02/26 21:34:55 perry Exp $");
36
37 #include <sys/param.h>
38 #include <sys/systm.h>
39 #include <sys/file.h>
40 #include <sys/mbuf.h>
41 #include <sys/protosw.h>
42 #include <sys/socket.h>
43 #include <sys/socketvar.h>
44 #include <sys/ioctl.h>
45 #include <sys/stat.h>
46 #include <sys/poll.h>
47 #include <sys/proc.h>
48
49 #include <net/if.h>
50 #include <net/route.h>
51
52 struct fileops socketops = {
53 soo_read, soo_write, soo_ioctl, soo_fcntl, soo_poll,
54 soo_stat, soo_close, soo_kqfilter
55 };
56
57 /* ARGSUSED */
58 int
59 soo_read(fp, offset, uio, cred, flags)
60 struct file *fp;
61 off_t *offset;
62 struct uio *uio;
63 struct ucred *cred;
64 int flags;
65 {
66 struct socket *so = (struct socket *) fp->f_data;
67 return ((*so->so_receive)(so, (struct mbuf **)0,
68 uio, (struct mbuf **)0, (struct mbuf **)0, (int *)0));
69 }
70
71 /* ARGSUSED */
72 int
73 soo_write(fp, offset, uio, cred, flags)
74 struct file *fp;
75 off_t *offset;
76 struct uio *uio;
77 struct ucred *cred;
78 int flags;
79 {
80 struct socket *so = (struct socket *) fp->f_data;
81 return ((*so->so_send)(so, (struct mbuf *)0,
82 uio, (struct mbuf *)0, (struct mbuf *)0, 0, uio->uio_procp));
83 }
84
85 int
86 soo_ioctl(fp, cmd, data, p)
87 struct file *fp;
88 u_long cmd;
89 void *data;
90 struct proc *p;
91 {
92 struct socket *so = (struct socket *)fp->f_data;
93
94 switch (cmd) {
95
96 case FIONBIO:
97 if (*(int *)data)
98 so->so_state |= SS_NBIO;
99 else
100 so->so_state &= ~SS_NBIO;
101 return (0);
102
103 case FIOASYNC:
104 if (*(int *)data) {
105 so->so_state |= SS_ASYNC;
106 so->so_rcv.sb_flags |= SB_ASYNC;
107 so->so_snd.sb_flags |= SB_ASYNC;
108 } else {
109 so->so_state &= ~SS_ASYNC;
110 so->so_rcv.sb_flags &= ~SB_ASYNC;
111 so->so_snd.sb_flags &= ~SB_ASYNC;
112 }
113 return (0);
114
115 case FIONREAD:
116 *(int *)data = so->so_rcv.sb_cc;
117 return (0);
118
119 case FIONWRITE:
120 *(int *)data = so->so_snd.sb_cc;
121 return (0);
122
123 case FIONSPACE:
124 /*
125 * See the comment around sbspace()'s definition
126 * in sys/socketvar.h in face of counts about maximum
127 * to understand the following test. We detect overflow
128 * and return zero.
129 */
130 if ((so->so_snd.sb_hiwat < so->so_snd.sb_cc)
131 || (so->so_snd.sb_mbmax < so->so_snd.sb_mbcnt))
132 *(int *)data = 0;
133 else
134 *(int *)data = sbspace(&so->so_snd);
135 return (0);
136
137 case SIOCSPGRP:
138 case FIOSETOWN:
139 case TIOCSPGRP:
140 return fsetown(p, &so->so_pgid, cmd, data);
141
142 case SIOCGPGRP:
143 case FIOGETOWN:
144 case TIOCGPGRP:
145 return fgetown(p, so->so_pgid, cmd, data);
146
147 case SIOCATMARK:
148 *(int *)data = (so->so_state&SS_RCVATMARK) != 0;
149 return (0);
150 }
151 /*
152 * Interface/routing/protocol specific ioctls:
153 * interface and routing ioctls should have a
154 * different entry since a socket's unnecessary
155 */
156 if (IOCGROUP(cmd) == 'i')
157 return (ifioctl(so, cmd, data, p));
158 if (IOCGROUP(cmd) == 'r')
159 return (rtioctl(cmd, data, p));
160 return ((*so->so_proto->pr_usrreq)(so, PRU_CONTROL,
161 (struct mbuf *)cmd, (struct mbuf *)data, (struct mbuf *)0, p));
162 }
163
164 int
165 soo_fcntl(fp, cmd, data, p)
166 struct file *fp;
167 u_int cmd;
168 void *data;
169 struct proc *p;
170 {
171 if (cmd == F_SETFL)
172 return (0);
173 else
174 return (EOPNOTSUPP);
175 }
176
177 int
178 soo_poll(fp, events, p)
179 struct file *fp;
180 int events;
181 struct proc *p;
182 {
183 struct socket *so = (struct socket *)fp->f_data;
184 int revents = 0;
185 int s = splsoftnet();
186
187 if (events & (POLLIN | POLLRDNORM))
188 if (soreadable(so))
189 revents |= events & (POLLIN | POLLRDNORM);
190
191 if (events & (POLLOUT | POLLWRNORM))
192 if (sowritable(so))
193 revents |= events & (POLLOUT | POLLWRNORM);
194
195 if (events & (POLLPRI | POLLRDBAND))
196 if (so->so_oobmark || (so->so_state & SS_RCVATMARK))
197 revents |= events & (POLLPRI | POLLRDBAND);
198
199 if (revents == 0) {
200 if (events & (POLLIN | POLLPRI | POLLRDNORM | POLLRDBAND)) {
201 selrecord(p, &so->so_rcv.sb_sel);
202 so->so_rcv.sb_flags |= SB_SEL;
203 }
204
205 if (events & (POLLOUT | POLLWRNORM)) {
206 selrecord(p, &so->so_snd.sb_sel);
207 so->so_snd.sb_flags |= SB_SEL;
208 }
209 }
210
211 splx(s);
212 return (revents);
213 }
214
215 int
216 soo_stat(fp, ub, p)
217 struct file *fp;
218 struct stat *ub;
219 struct proc *p;
220 {
221 struct socket *so = (struct socket *)fp->f_data;
222
223 memset((caddr_t)ub, 0, sizeof(*ub));
224 ub->st_mode = S_IFSOCK;
225 return ((*so->so_proto->pr_usrreq)(so, PRU_SENSE,
226 (struct mbuf *)ub, (struct mbuf *)0, (struct mbuf *)0, p));
227 }
228
229 /* ARGSUSED */
230 int
231 soo_close(fp, p)
232 struct file *fp;
233 struct proc *p;
234 {
235 int error = 0;
236
237 if (fp->f_data)
238 error = soclose((struct socket *)fp->f_data);
239 fp->f_data = 0;
240 return (error);
241 }
Cache object: f8ec5f06b37163c21ae23219474d4623
|