sys_socket.c revision 1.38 1 /* $NetBSD: sys_socket.c,v 1.38 2003/08/07 16:31:55 agc 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.38 2003/08/07 16:31:55 agc 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));
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 SIOCSPGRP:
120 so->so_pgid = *(int *)data;
121 return (0);
122
123 case SIOCGPGRP:
124 *(int *)data = so->so_pgid;
125 return (0);
126
127 case SIOCATMARK:
128 *(int *)data = (so->so_state&SS_RCVATMARK) != 0;
129 return (0);
130 }
131 /*
132 * Interface/routing/protocol specific ioctls:
133 * interface and routing ioctls should have a
134 * different entry since a socket's unnecessary
135 */
136 if (IOCGROUP(cmd) == 'i')
137 return (ifioctl(so, cmd, data, p));
138 if (IOCGROUP(cmd) == 'r')
139 return (rtioctl(cmd, data, p));
140 return ((*so->so_proto->pr_usrreq)(so, PRU_CONTROL,
141 (struct mbuf *)cmd, (struct mbuf *)data, (struct mbuf *)0, p));
142 }
143
144 int
145 soo_fcntl(fp, cmd, data, p)
146 struct file *fp;
147 u_int cmd;
148 void *data;
149 struct proc *p;
150 {
151 if (cmd == F_SETFL)
152 return (0);
153 else
154 return (EOPNOTSUPP);
155 }
156
157 int
158 soo_poll(fp, events, p)
159 struct file *fp;
160 int events;
161 struct proc *p;
162 {
163 struct socket *so = (struct socket *)fp->f_data;
164 int revents = 0;
165 int s = splsoftnet();
166
167 if (events & (POLLIN | POLLRDNORM))
168 if (soreadable(so))
169 revents |= events & (POLLIN | POLLRDNORM);
170
171 if (events & (POLLOUT | POLLWRNORM))
172 if (sowritable(so))
173 revents |= events & (POLLOUT | POLLWRNORM);
174
175 if (events & (POLLPRI | POLLRDBAND))
176 if (so->so_oobmark || (so->so_state & SS_RCVATMARK))
177 revents |= events & (POLLPRI | POLLRDBAND);
178
179 if (revents == 0) {
180 if (events & (POLLIN | POLLPRI | POLLRDNORM | POLLRDBAND)) {
181 selrecord(p, &so->so_rcv.sb_sel);
182 so->so_rcv.sb_flags |= SB_SEL;
183 }
184
185 if (events & (POLLOUT | POLLWRNORM)) {
186 selrecord(p, &so->so_snd.sb_sel);
187 so->so_snd.sb_flags |= SB_SEL;
188 }
189 }
190
191 splx(s);
192 return (revents);
193 }
194
195 int
196 soo_stat(fp, ub, p)
197 struct file *fp;
198 struct stat *ub;
199 struct proc *p;
200 {
201 struct socket *so = (struct socket *)fp->f_data;
202
203 memset((caddr_t)ub, 0, sizeof(*ub));
204 ub->st_mode = S_IFSOCK;
205 return ((*so->so_proto->pr_usrreq)(so, PRU_SENSE,
206 (struct mbuf *)ub, (struct mbuf *)0, (struct mbuf *)0, p));
207 }
208
209 /* ARGSUSED */
210 int
211 soo_close(fp, p)
212 struct file *fp;
213 struct proc *p;
214 {
215 int error = 0;
216
217 if (fp->f_data)
218 error = soclose((struct socket *)fp->f_data);
219 fp->f_data = 0;
220 return (error);
221 }
222