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 };
65
66 /*
67 * MPSAFE
68 */
69 int
soo_read(struct file * fp,struct uio * uio,struct ucred * cred,int fflags)70 soo_read(struct file *fp, struct uio *uio, struct ucred *cred, int fflags)
71 {
72 struct socket *so;
73 int error;
74 int msgflags;
75
76 atomic_set_int(&curthread->td_mpflags, TDF_MP_BATCH_DEMARC);
77
78 so = (struct socket *)fp->f_data;
79
80 if (fflags & O_FBLOCKING)
81 msgflags = 0;
82 else if (fflags & O_FNONBLOCKING)
83 msgflags = MSG_FNONBLOCKING;
84 else if (fp->f_flag & FNONBLOCK)
85 msgflags = MSG_FNONBLOCKING;
86 else
87 msgflags = 0;
88
89 error = so_pru_soreceive(so, NULL, uio, NULL, NULL, &msgflags);
90 return (error);
91 }
92
93 /*
94 * MPSAFE
95 */
96 int
soo_write(struct file * fp,struct uio * uio,struct ucred * cred,int fflags)97 soo_write(struct file *fp, struct uio *uio, struct ucred *cred, int fflags)
98 {
99 struct socket *so;
100 int error;
101 int msgflags;
102
103 so = (struct socket *)fp->f_data;
104
105 if (fflags & O_FBLOCKING)
106 msgflags = 0;
107 else if (fflags & O_FNONBLOCKING)
108 msgflags = MSG_FNONBLOCKING;
109 else if (fp->f_flag & FNONBLOCK)
110 msgflags = MSG_FNONBLOCKING;
111 else
112 msgflags = 0;
113
114 error = so_pru_sosend(so, NULL, uio, NULL, NULL, msgflags, uio->uio_td);
115 if (error == EPIPE && !(fflags & MSG_NOSIGNAL) &&
116 !(so->so_options & SO_NOSIGPIPE) &&
117 uio->uio_td->td_lwp) {
118 lwpsignal(uio->uio_td->td_proc, uio->uio_td->td_lwp, SIGPIPE);
119 }
120 return (error);
121 }
122
123 /*
124 * MPSAFE
125 */
126 int
soo_ioctl(struct file * fp,u_long cmd,caddr_t data,struct ucred * cred,struct sysmsg * msg)127 soo_ioctl(struct file *fp, u_long cmd, caddr_t data,
128 struct ucred *cred, struct sysmsg *msg)
129 {
130 struct socket *so;
131 int error;
132
133 so = (struct socket *)fp->f_data;
134
135 switch (cmd) {
136 case FIONBIO:
137 error = 0;
138 break;
139 case FIOASYNC:
140 if (*(int *)data) {
141 sosetstate(so, SS_ASYNC);
142 atomic_set_int(&so->so_rcv.ssb_flags, SSB_ASYNC);
143 atomic_set_int(&so->so_snd.ssb_flags, SSB_ASYNC);
144 } else {
145 soclrstate(so, SS_ASYNC);
146 atomic_clear_int(&so->so_rcv.ssb_flags, SSB_ASYNC);
147 atomic_clear_int(&so->so_snd.ssb_flags, SSB_ASYNC);
148 }
149 error = 0;
150 break;
151 case FIONREAD:
152 *(int *)data = so->so_rcv.ssb_cc;
153 error = 0;
154 break;
155 case FIOSETOWN:
156 error = fsetown(*(int *)data, &so->so_sigio);
157 break;
158 case FIOGETOWN:
159 *(int *)data = fgetown(&so->so_sigio);
160 error = 0;
161 break;
162 case SIOCSPGRP:
163 error = fsetown(-(*(int *)data), &so->so_sigio);
164 break;
165 case SIOCGPGRP:
166 *(int *)data = -fgetown(&so->so_sigio);
167 error = 0;
168 break;
169 case SIOCATMARK:
170 *(int *)data = (so->so_state&SS_RCVATMARK) != 0;
171 error = 0;
172 break;
173 default:
174 /*
175 * Interface/routing/protocol specific ioctls:
176 * interface and routing ioctls should have a
177 * different entry since a socket's unnecessary
178 */
179 if (IOCGROUP(cmd) == 'i') {
180 error = ifioctl(so, cmd, data, cred);
181 } else if (IOCGROUP(cmd) == 'r') {
182 error = rtioctl(cmd, data, cred);
183 } else {
184 error = so_pru_control_direct(so, cmd, data, NULL);
185 }
186 break;
187 }
188 return (error);
189 }
190
191 /*
192 * MPSAFE
193 */
194 int
soo_stat(struct file * fp,struct stat * ub,struct ucred * cred)195 soo_stat(struct file *fp, struct stat *ub, struct ucred *cred)
196 {
197 struct socket *so;
198 int error;
199
200 bzero((caddr_t)ub, sizeof (*ub));
201 ub->st_mode = S_IFSOCK;
202 so = (struct socket *)fp->f_data;
203
204 /*
205 * If SS_CANTRCVMORE is set, but there's still data left in the
206 * receive buffer, the socket is still readable.
207 */
208 if ((so->so_state & SS_CANTRCVMORE) == 0 ||
209 so->so_rcv.ssb_cc != 0)
210 ub->st_mode |= S_IRUSR | S_IRGRP | S_IROTH;
211 if ((so->so_state & SS_CANTSENDMORE) == 0)
212 ub->st_mode |= S_IWUSR | S_IWGRP | S_IWOTH;
213 ub->st_size = so->so_rcv.ssb_cc;
214 ub->st_uid = so->so_cred->cr_uid;
215 ub->st_gid = so->so_cred->cr_gid;
216 ub->st_ino = so->so_inum;
217 error = so_pru_sense(so, ub);
218
219 return (error);
220 }
221
222 /*
223 * MPSAFE
224 */
225 int
soo_close(struct file * fp)226 soo_close(struct file *fp)
227 {
228 int error;
229
230 fp->f_ops = &badfileops;
231 if (fp->f_data)
232 error = soclose((struct socket *)fp->f_data, fp->f_flag);
233 else
234 error = 0;
235 fp->f_data = NULL;
236 return (error);
237 }
238
239 /*
240 * MPSAFE
241 */
242 int
soo_shutdown(struct file * fp,int how)243 soo_shutdown(struct file *fp, int how)
244 {
245 int error;
246
247 if (fp->f_data)
248 error = soshutdown((struct socket *)fp->f_data, how);
249 else
250 error = 0;
251 return (error);
252 }
253
254