1 /*
2 * Copyright (c) 1982, 1986, 1990, 1993
3 * The Regents of the University of California. All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. Neither the name of the University nor the names of its contributors
14 * may be used to endorse or promote products derived from this software
15 * without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * SUCH DAMAGE.
28 *
29 * @(#)sys_socket.c 8.1 (Berkeley) 6/10/93
30 * $FreeBSD: src/sys/kern/sys_socket.c,v 1.28.2.2 2001/02/26 04:23:16 jlemon Exp $
31 */
32
33 #include <sys/param.h>
34 #include <sys/systm.h>
35 #include <sys/file.h>
36 #include <sys/lock.h>
37 #include <sys/protosw.h>
38 #include <sys/signalvar.h>
39 #include <sys/socket.h>
40 #include <sys/socketvar.h>
41 #include <sys/socketops.h>
42 #include <sys/filio.h> /* XXX */
43 #include <sys/sockio.h>
44 #include <sys/vnode.h>
45 #include <sys/stat.h>
46 #include <sys/uio.h>
47 #include <sys/filedesc.h>
48 #include <sys/ucred.h>
49
50 #include <sys/socketvar2.h>
51
52 #include <net/if.h>
53 #include <net/if_var.h>
54 #include <net/route.h>
55
56 struct fileops socketops = {
57 .fo_read = soo_read,
58 .fo_write = soo_write,
59 .fo_ioctl = soo_ioctl,
60 .fo_kqfilter = sokqfilter,
61 .fo_stat = soo_stat,
62 .fo_close = soo_close,
63 .fo_shutdown = soo_shutdown,
64 .fo_seek = badfo_seek
65 };
66
67 /*
68 * MPSAFE
69 */
70 int
soo_read(struct file * fp,struct uio * uio,struct ucred * cred,int fflags)71 soo_read(struct file *fp, struct uio *uio, struct ucred *cred, int fflags)
72 {
73 struct socket *so;
74 int error;
75 int msgflags;
76
77 atomic_set_int(&curthread->td_mpflags, TDF_MP_BATCH_DEMARC);
78
79 so = (struct socket *)fp->f_data;
80
81 if (fflags & O_FBLOCKING)
82 msgflags = 0;
83 else if (fflags & O_FNONBLOCKING)
84 msgflags = MSG_FNONBLOCKING;
85 else if (fp->f_flag & FNONBLOCK)
86 msgflags = MSG_FNONBLOCKING;
87 else
88 msgflags = 0;
89
90 error = so_pru_soreceive(so, NULL, uio, NULL, NULL, &msgflags);
91 return (error);
92 }
93
94 /*
95 * MPSAFE
96 */
97 int
soo_write(struct file * fp,struct uio * uio,struct ucred * cred,int fflags)98 soo_write(struct file *fp, struct uio *uio, struct ucred *cred, int fflags)
99 {
100 struct socket *so;
101 int error;
102 int msgflags;
103
104 so = (struct socket *)fp->f_data;
105
106 if (fflags & O_FBLOCKING)
107 msgflags = 0;
108 else if (fflags & O_FNONBLOCKING)
109 msgflags = MSG_FNONBLOCKING;
110 else if (fp->f_flag & FNONBLOCK)
111 msgflags = MSG_FNONBLOCKING;
112 else
113 msgflags = 0;
114
115 error = so_pru_sosend(so, NULL, uio, NULL, NULL, msgflags, uio->uio_td);
116 if (error == EPIPE && !(fflags & MSG_NOSIGNAL) &&
117 !(so->so_options & SO_NOSIGPIPE) &&
118 uio->uio_td->td_lwp) {
119 lwpsignal(uio->uio_td->td_proc, uio->uio_td->td_lwp, SIGPIPE);
120 }
121 return (error);
122 }
123
124 /*
125 * MPSAFE
126 */
127 int
soo_ioctl(struct file * fp,u_long cmd,caddr_t data,struct ucred * cred,struct sysmsg * msg)128 soo_ioctl(struct file *fp, u_long cmd, caddr_t data,
129 struct ucred *cred, struct sysmsg *msg)
130 {
131 struct socket *so;
132 int error;
133
134 so = (struct socket *)fp->f_data;
135
136 switch (cmd) {
137 case FIONBIO:
138 error = 0;
139 break;
140 case FIOASYNC:
141 if (*(int *)data) {
142 sosetstate(so, SS_ASYNC);
143 atomic_set_int(&so->so_rcv.ssb_flags, SSB_ASYNC);
144 atomic_set_int(&so->so_snd.ssb_flags, SSB_ASYNC);
145 } else {
146 soclrstate(so, SS_ASYNC);
147 atomic_clear_int(&so->so_rcv.ssb_flags, SSB_ASYNC);
148 atomic_clear_int(&so->so_snd.ssb_flags, SSB_ASYNC);
149 }
150 error = 0;
151 break;
152 case FIONREAD:
153 *(int *)data = so->so_rcv.ssb_cc;
154 error = 0;
155 break;
156 case FIOSETOWN:
157 error = fsetown(*(int *)data, &so->so_sigio);
158 break;
159 case FIOGETOWN:
160 *(int *)data = fgetown(&so->so_sigio);
161 error = 0;
162 break;
163 case SIOCSPGRP:
164 error = fsetown(-(*(int *)data), &so->so_sigio);
165 break;
166 case SIOCGPGRP:
167 *(int *)data = -fgetown(&so->so_sigio);
168 error = 0;
169 break;
170 case SIOCATMARK:
171 *(int *)data = (so->so_state&SS_RCVATMARK) != 0;
172 error = 0;
173 break;
174 default:
175 /*
176 * Interface/routing/protocol specific ioctls:
177 * interface and routing ioctls should have a
178 * different entry since a socket's unnecessary
179 */
180 if (IOCGROUP(cmd) == 'i') {
181 error = ifioctl(so, cmd, data, cred);
182 } else if (IOCGROUP(cmd) == 'r') {
183 error = rtioctl(cmd, data, cred);
184 } else {
185 error = so_pru_control_direct(so, cmd, data, NULL);
186 }
187 break;
188 }
189 return (error);
190 }
191
192 /*
193 * MPSAFE
194 */
195 int
soo_stat(struct file * fp,struct stat * ub,struct ucred * cred)196 soo_stat(struct file *fp, struct stat *ub, struct ucred *cred)
197 {
198 struct socket *so;
199 int error;
200
201 bzero((caddr_t)ub, sizeof (*ub));
202 ub->st_mode = S_IFSOCK;
203 so = (struct socket *)fp->f_data;
204
205 /*
206 * If SS_CANTRCVMORE is set, but there's still data left in the
207 * receive buffer, the socket is still readable.
208 */
209 if ((so->so_state & SS_CANTRCVMORE) == 0 ||
210 so->so_rcv.ssb_cc != 0)
211 ub->st_mode |= S_IRUSR | S_IRGRP | S_IROTH;
212 if ((so->so_state & SS_CANTSENDMORE) == 0)
213 ub->st_mode |= S_IWUSR | S_IWGRP | S_IWOTH;
214 ub->st_size = so->so_rcv.ssb_cc;
215 ub->st_uid = so->so_cred->cr_uid;
216 ub->st_gid = so->so_cred->cr_gid;
217 ub->st_ino = so->so_inum;
218 error = so_pru_sense(so, ub);
219
220 return (error);
221 }
222
223 /*
224 * MPSAFE
225 */
226 int
soo_close(struct file * fp)227 soo_close(struct file *fp)
228 {
229 int error;
230
231 fp->f_ops = &badfileops;
232 if (fp->f_data)
233 error = soclose((struct socket *)fp->f_data, fp->f_flag);
234 else
235 error = 0;
236 fp->f_data = NULL;
237 return (error);
238 }
239
240 /*
241 * MPSAFE
242 */
243 int
soo_shutdown(struct file * fp,int how)244 soo_shutdown(struct file *fp, int how)
245 {
246 int error;
247
248 if (fp->f_data)
249 error = soshutdown((struct socket *)fp->f_data, how);
250 else
251 error = 0;
252 return (error);
253 }
254
255