uipc_syscalls.c revision 1.196 1 /* $NetBSD: uipc_syscalls.c,v 1.196 2018/08/01 23:35:32 rjs 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, 1989, 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 * @(#)uipc_syscalls.c 8.6 (Berkeley) 2/14/95
61 */
62
63 #include <sys/cdefs.h>
64 __KERNEL_RCSID(0, "$NetBSD: uipc_syscalls.c,v 1.196 2018/08/01 23:35:32 rjs Exp $");
65
66 #ifdef _KERNEL_OPT
67 #include "opt_pipe.h"
68 #include "opt_sctp.h"
69 #endif
70
71 #define MBUFTYPES
72 #include <sys/param.h>
73 #include <sys/systm.h>
74 #include <sys/filedesc.h>
75 #include <sys/proc.h>
76 #include <sys/file.h>
77 #include <sys/buf.h>
78 #include <sys/mbuf.h>
79 #include <sys/protosw.h>
80 #include <sys/socket.h>
81 #include <sys/socketvar.h>
82 #include <sys/signalvar.h>
83 #include <sys/un.h>
84 #include <sys/ktrace.h>
85 #include <sys/event.h>
86 #include <sys/atomic.h>
87 #include <sys/kauth.h>
88
89 #ifdef SCTP
90 #include <netinet/sctp_uio.h>
91 #include <netinet/sctp_peeloff.h>
92 #endif
93
94 #include <sys/mount.h>
95 #include <sys/syscallargs.h>
96
97 /*
98 * System call interface to the socket abstraction.
99 */
100 extern const struct fileops socketops;
101
102 static int sockargs_sb(struct sockaddr_big *, const void *, socklen_t);
103
104 int
105 sys___socket30(struct lwp *l, const struct sys___socket30_args *uap,
106 register_t *retval)
107 {
108 /* {
109 syscallarg(int) domain;
110 syscallarg(int) type;
111 syscallarg(int) protocol;
112 } */
113 int fd, error;
114
115 error = fsocreate(SCARG(uap, domain), NULL, SCARG(uap, type),
116 SCARG(uap, protocol), &fd);
117 if (error == 0) {
118 *retval = fd;
119 }
120 return error;
121 }
122
123 int
124 sys_bind(struct lwp *l, const struct sys_bind_args *uap, register_t *retval)
125 {
126 /* {
127 syscallarg(int) s;
128 syscallarg(const struct sockaddr *) name;
129 syscallarg(unsigned int) namelen;
130 } */
131 int error;
132 struct sockaddr_big sb;
133
134 error = sockargs_sb(&sb, SCARG(uap, name), SCARG(uap, namelen));
135 if (error)
136 return error;
137
138 return do_sys_bind(l, SCARG(uap, s), (struct sockaddr *)&sb);
139 }
140
141 int
142 do_sys_bind(struct lwp *l, int fd, struct sockaddr *nam)
143 {
144 struct socket *so;
145 int error;
146
147 if ((error = fd_getsock(fd, &so)) != 0)
148 return error;
149 error = sobind(so, nam, l);
150 fd_putfile(fd);
151 return error;
152 }
153
154 int
155 sys_listen(struct lwp *l, const struct sys_listen_args *uap, register_t *retval)
156 {
157 /* {
158 syscallarg(int) s;
159 syscallarg(int) backlog;
160 } */
161 struct socket *so;
162 int error;
163
164 if ((error = fd_getsock(SCARG(uap, s), &so)) != 0)
165 return (error);
166 error = solisten(so, SCARG(uap, backlog), l);
167 fd_putfile(SCARG(uap, s));
168 return error;
169 }
170
171 int
172 do_sys_accept(struct lwp *l, int sock, struct sockaddr *name,
173 register_t *new_sock, const sigset_t *mask, int flags, int clrflags)
174 {
175 file_t *fp, *fp2;
176 int error, fd;
177 struct socket *so, *so2;
178 short wakeup_state = 0;
179
180 if ((fp = fd_getfile(sock)) == NULL)
181 return EBADF;
182 if (fp->f_type != DTYPE_SOCKET) {
183 fd_putfile(sock);
184 return ENOTSOCK;
185 }
186 if ((error = fd_allocfile(&fp2, &fd)) != 0) {
187 fd_putfile(sock);
188 return error;
189 }
190 *new_sock = fd;
191 so = fp->f_socket;
192 solock(so);
193
194 if (__predict_false(mask))
195 sigsuspendsetup(l, mask);
196
197 if (!(so->so_proto->pr_flags & PR_LISTEN)) {
198 error = EOPNOTSUPP;
199 goto bad;
200 }
201 if ((so->so_options & SO_ACCEPTCONN) == 0) {
202 error = EINVAL;
203 goto bad;
204 }
205 if ((so->so_state & SS_NBIO) && so->so_qlen == 0) {
206 error = EWOULDBLOCK;
207 goto bad;
208 }
209 while (so->so_qlen == 0 && so->so_error == 0) {
210 if (so->so_state & SS_CANTRCVMORE) {
211 so->so_error = ECONNABORTED;
212 break;
213 }
214 if (wakeup_state & SS_RESTARTSYS) {
215 error = ERESTART;
216 goto bad;
217 }
218 error = sowait(so, true, 0);
219 if (error) {
220 goto bad;
221 }
222 wakeup_state = so->so_state;
223 }
224 if (so->so_error) {
225 error = so->so_error;
226 so->so_error = 0;
227 goto bad;
228 }
229 /* connection has been removed from the listen queue */
230 KNOTE(&so->so_rcv.sb_sel.sel_klist, NOTE_SUBMIT);
231 so2 = TAILQ_FIRST(&so->so_q);
232 if (soqremque(so2, 1) == 0)
233 panic("accept");
234 fp2->f_type = DTYPE_SOCKET;
235 fp2->f_flag = (fp->f_flag & ~clrflags) |
236 ((flags & SOCK_NONBLOCK) ? FNONBLOCK : 0)|
237 ((flags & SOCK_NOSIGPIPE) ? FNOSIGPIPE : 0);
238 fp2->f_ops = &socketops;
239 fp2->f_socket = so2;
240 if (fp2->f_flag & FNONBLOCK)
241 so2->so_state |= SS_NBIO;
242 else
243 so2->so_state &= ~SS_NBIO;
244 error = soaccept(so2, name);
245 so2->so_cred = kauth_cred_dup(so->so_cred);
246 sounlock(so);
247 if (error) {
248 /* an error occurred, free the file descriptor and mbuf */
249 mutex_enter(&fp2->f_lock);
250 fp2->f_count++;
251 mutex_exit(&fp2->f_lock);
252 closef(fp2);
253 fd_abort(curproc, NULL, fd);
254 } else {
255 fd_set_exclose(l, fd, (flags & SOCK_CLOEXEC) != 0);
256 fd_affix(curproc, fp2, fd);
257 }
258 fd_putfile(sock);
259 if (__predict_false(mask))
260 sigsuspendteardown(l);
261 return error;
262 bad:
263 sounlock(so);
264 fd_putfile(sock);
265 fd_abort(curproc, fp2, fd);
266 if (__predict_false(mask))
267 sigsuspendteardown(l);
268 return error;
269 }
270
271 int
272 sys_accept(struct lwp *l, const struct sys_accept_args *uap, register_t *retval)
273 {
274 /* {
275 syscallarg(int) s;
276 syscallarg(struct sockaddr *) name;
277 syscallarg(unsigned int *) anamelen;
278 } */
279 int error, fd;
280 struct sockaddr_big name;
281
282 name.sb_len = UCHAR_MAX;
283 error = do_sys_accept(l, SCARG(uap, s), (struct sockaddr *)&name,
284 retval, NULL, 0, 0);
285 if (error != 0)
286 return error;
287 error = copyout_sockname_sb(SCARG(uap, name), SCARG(uap, anamelen),
288 MSG_LENUSRSPACE, &name);
289 if (error != 0) {
290 fd = (int)*retval;
291 if (fd_getfile(fd) != NULL)
292 (void)fd_close(fd);
293 }
294 return error;
295 }
296
297 int
298 sys_paccept(struct lwp *l, const struct sys_paccept_args *uap,
299 register_t *retval)
300 {
301 /* {
302 syscallarg(int) s;
303 syscallarg(struct sockaddr *) name;
304 syscallarg(unsigned int *) anamelen;
305 syscallarg(const sigset_t *) mask;
306 syscallarg(int) flags;
307 } */
308 int error, fd;
309 struct sockaddr_big name;
310 sigset_t *mask, amask;
311
312 if (SCARG(uap, mask) != NULL) {
313 error = copyin(SCARG(uap, mask), &amask, sizeof(amask));
314 if (error)
315 return error;
316 mask = &amask;
317 } else
318 mask = NULL;
319
320 name.sb_len = UCHAR_MAX;
321 error = do_sys_accept(l, SCARG(uap, s), (struct sockaddr *)&name,
322 retval, mask, SCARG(uap, flags), FNONBLOCK);
323 if (error != 0)
324 return error;
325 error = copyout_sockname_sb(SCARG(uap, name), SCARG(uap, anamelen),
326 MSG_LENUSRSPACE, &name);
327 if (error != 0) {
328 fd = (int)*retval;
329 if (fd_getfile(fd) != NULL)
330 (void)fd_close(fd);
331 }
332 return error;
333 }
334
335 int
336 sys_connect(struct lwp *l, const struct sys_connect_args *uap,
337 register_t *retval)
338 {
339 /* {
340 syscallarg(int) s;
341 syscallarg(const struct sockaddr *) name;
342 syscallarg(unsigned int) namelen;
343 } */
344 int error;
345 struct sockaddr_big sbig;
346
347 error = sockargs_sb(&sbig, SCARG(uap, name), SCARG(uap, namelen));
348 if (error)
349 return error;
350 return do_sys_connect(l, SCARG(uap, s), (struct sockaddr *)&sbig);
351 }
352
353 int
354 do_sys_connect(struct lwp *l, int fd, struct sockaddr *nam)
355 {
356 struct socket *so;
357 int error;
358 int interrupted = 0;
359
360 if ((error = fd_getsock(fd, &so)) != 0) {
361 return (error);
362 }
363 solock(so);
364 if ((so->so_state & SS_ISCONNECTING) != 0) {
365 error = EALREADY;
366 goto out;
367 }
368
369 error = soconnect(so, nam, l);
370 if (error)
371 goto bad;
372 if ((so->so_state & (SS_NBIO|SS_ISCONNECTING)) ==
373 (SS_NBIO|SS_ISCONNECTING)) {
374 error = EINPROGRESS;
375 goto out;
376 }
377 while ((so->so_state & SS_ISCONNECTING) != 0 && so->so_error == 0) {
378 error = sowait(so, true, 0);
379 if (__predict_false((so->so_state & SS_ISABORTING) != 0)) {
380 error = EPIPE;
381 interrupted = 1;
382 break;
383 }
384 if (error) {
385 if (error == EINTR || error == ERESTART)
386 interrupted = 1;
387 break;
388 }
389 }
390 if (error == 0) {
391 error = so->so_error;
392 so->so_error = 0;
393 }
394 bad:
395 if (!interrupted)
396 so->so_state &= ~SS_ISCONNECTING;
397 if (error == ERESTART)
398 error = EINTR;
399 out:
400 sounlock(so);
401 fd_putfile(fd);
402 return error;
403 }
404
405 static int
406 makesocket(struct lwp *l, file_t **fp, int *fd, int flags, int type,
407 int domain, int proto, struct socket *soo)
408 {
409 struct socket *so;
410 int error;
411
412 if ((error = socreate(domain, &so, type, proto, l, soo)) != 0) {
413 return error;
414 }
415 if (flags & SOCK_NONBLOCK) {
416 so->so_state |= SS_NBIO;
417 }
418
419 if ((error = fd_allocfile(fp, fd)) != 0) {
420 soclose(so);
421 return error;
422 }
423 fd_set_exclose(l, *fd, (flags & SOCK_CLOEXEC) != 0);
424 (*fp)->f_flag = FREAD|FWRITE|
425 ((flags & SOCK_NONBLOCK) ? FNONBLOCK : 0)|
426 ((flags & SOCK_NOSIGPIPE) ? FNOSIGPIPE : 0);
427 (*fp)->f_type = DTYPE_SOCKET;
428 (*fp)->f_ops = &socketops;
429 (*fp)->f_socket = so;
430 return 0;
431 }
432
433 int
434 sys_socketpair(struct lwp *l, const struct sys_socketpair_args *uap,
435 register_t *retval)
436 {
437 /* {
438 syscallarg(int) domain;
439 syscallarg(int) type;
440 syscallarg(int) protocol;
441 syscallarg(int *) rsv;
442 } */
443 file_t *fp1, *fp2;
444 struct socket *so1, *so2;
445 int fd, error, sv[2];
446 proc_t *p = curproc;
447 int flags = SCARG(uap, type) & SOCK_FLAGS_MASK;
448 int type = SCARG(uap, type) & ~SOCK_FLAGS_MASK;
449 int domain = SCARG(uap, domain);
450 int proto = SCARG(uap, protocol);
451
452 error = makesocket(l, &fp1, &fd, flags, type, domain, proto, NULL);
453 if (error)
454 return error;
455 so1 = fp1->f_socket;
456 sv[0] = fd;
457
458 error = makesocket(l, &fp2, &fd, flags, type, domain, proto, so1);
459 if (error)
460 goto out;
461 so2 = fp2->f_socket;
462 sv[1] = fd;
463
464 solock(so1);
465 error = soconnect2(so1, so2);
466 if (error == 0 && type == SOCK_DGRAM) {
467 /*
468 * Datagram socket connection is asymmetric.
469 */
470 error = soconnect2(so2, so1);
471 }
472 sounlock(so1);
473
474 if (error == 0)
475 error = copyout(sv, SCARG(uap, rsv), sizeof(sv));
476 if (error == 0) {
477 fd_affix(p, fp2, sv[1]);
478 fd_affix(p, fp1, sv[0]);
479 return 0;
480 }
481 fd_abort(p, fp2, sv[1]);
482 (void)soclose(so2);
483 out:
484 fd_abort(p, fp1, sv[0]);
485 (void)soclose(so1);
486 return error;
487 }
488
489 int
490 sys_sendto(struct lwp *l, const struct sys_sendto_args *uap,
491 register_t *retval)
492 {
493 /* {
494 syscallarg(int) s;
495 syscallarg(const void *) buf;
496 syscallarg(size_t) len;
497 syscallarg(int) flags;
498 syscallarg(const struct sockaddr *) to;
499 syscallarg(unsigned int) tolen;
500 } */
501 struct msghdr msg;
502 struct iovec aiov;
503
504 msg.msg_name = __UNCONST(SCARG(uap, to)); /* XXXUNCONST kills const */
505 msg.msg_namelen = SCARG(uap, tolen);
506 msg.msg_iov = &aiov;
507 msg.msg_iovlen = 1;
508 msg.msg_control = NULL;
509 msg.msg_flags = 0;
510 aiov.iov_base = __UNCONST(SCARG(uap, buf)); /* XXXUNCONST kills const */
511 aiov.iov_len = SCARG(uap, len);
512 return do_sys_sendmsg(l, SCARG(uap, s), &msg, SCARG(uap, flags),
513 retval);
514 }
515
516 int
517 sys_sendmsg(struct lwp *l, const struct sys_sendmsg_args *uap,
518 register_t *retval)
519 {
520 /* {
521 syscallarg(int) s;
522 syscallarg(const struct msghdr *) msg;
523 syscallarg(int) flags;
524 } */
525 struct msghdr msg;
526 int error;
527
528 error = copyin(SCARG(uap, msg), &msg, sizeof(msg));
529 if (error)
530 return (error);
531
532 msg.msg_flags = MSG_IOVUSRSPACE;
533 return do_sys_sendmsg(l, SCARG(uap, s), &msg, SCARG(uap, flags),
534 retval);
535 }
536
537 int
538 do_sys_sendmsg_so(struct lwp *l, int s, struct socket *so, file_t *fp,
539 struct msghdr *mp, int flags, register_t *retsize)
540 {
541
542 struct iovec aiov[UIO_SMALLIOV], *iov = aiov, *tiov, *ktriov = NULL;
543 struct sockaddr *sa = NULL;
544 struct mbuf *to, *control;
545 struct uio auio;
546 size_t len, iovsz;
547 int i, error;
548
549 ktrkuser("msghdr", mp, sizeof(*mp));
550
551 /* If the caller passed us stuff in mbufs, we must free them. */
552 to = (mp->msg_flags & MSG_NAMEMBUF) ? mp->msg_name : NULL;
553 control = (mp->msg_flags & MSG_CONTROLMBUF) ? mp->msg_control : NULL;
554 iovsz = mp->msg_iovlen * sizeof(struct iovec);
555
556 if (mp->msg_flags & MSG_IOVUSRSPACE) {
557 if ((unsigned int)mp->msg_iovlen > UIO_SMALLIOV) {
558 if ((unsigned int)mp->msg_iovlen > IOV_MAX) {
559 error = EMSGSIZE;
560 goto bad;
561 }
562 iov = kmem_alloc(iovsz, KM_SLEEP);
563 }
564 if (mp->msg_iovlen != 0) {
565 error = copyin(mp->msg_iov, iov, iovsz);
566 if (error)
567 goto bad;
568 }
569 auio.uio_iov = iov;
570 } else
571 auio.uio_iov = mp->msg_iov;
572
573 auio.uio_iovcnt = mp->msg_iovlen;
574 auio.uio_rw = UIO_WRITE;
575 auio.uio_offset = 0; /* XXX */
576 auio.uio_resid = 0;
577 KASSERT(l == curlwp);
578 auio.uio_vmspace = l->l_proc->p_vmspace;
579
580 tiov = auio.uio_iov;
581 for (i = 0; i < auio.uio_iovcnt; i++, tiov++) {
582 /*
583 * Writes return ssize_t because -1 is returned on error.
584 * Therefore, we must restrict the length to SSIZE_MAX to
585 * avoid garbage return values.
586 */
587 auio.uio_resid += tiov->iov_len;
588 if (tiov->iov_len > SSIZE_MAX || auio.uio_resid > SSIZE_MAX) {
589 error = EINVAL;
590 goto bad;
591 }
592 }
593
594 if (mp->msg_name && to == NULL) {
595 error = sockargs(&to, mp->msg_name, mp->msg_namelen,
596 UIO_USERSPACE, MT_SONAME);
597 if (error)
598 goto bad;
599 }
600
601 if (mp->msg_control) {
602 if (mp->msg_controllen < CMSG_ALIGN(sizeof(struct cmsghdr))) {
603 error = EINVAL;
604 goto bad;
605 }
606 if (control == NULL) {
607 error = sockargs(&control, mp->msg_control,
608 mp->msg_controllen, UIO_USERSPACE, MT_CONTROL);
609 if (error)
610 goto bad;
611 }
612 }
613
614 if (ktrpoint(KTR_GENIO) && iovsz > 0) {
615 ktriov = kmem_alloc(iovsz, KM_SLEEP);
616 memcpy(ktriov, auio.uio_iov, iovsz);
617 }
618
619 if (mp->msg_name)
620 MCLAIM(to, so->so_mowner);
621 if (mp->msg_control)
622 MCLAIM(control, so->so_mowner);
623
624 if (to) {
625 sa = mtod(to, struct sockaddr *);
626 }
627
628 len = auio.uio_resid;
629 error = (*so->so_send)(so, sa, &auio, NULL, control, flags, l);
630 /* Protocol is responsible for freeing 'control' */
631 control = NULL;
632
633 if (error) {
634 if (auio.uio_resid != len && (error == ERESTART ||
635 error == EINTR || error == EWOULDBLOCK))
636 error = 0;
637 if (error == EPIPE && (fp->f_flag & FNOSIGPIPE) == 0 &&
638 (flags & MSG_NOSIGNAL) == 0) {
639 mutex_enter(proc_lock);
640 psignal(l->l_proc, SIGPIPE);
641 mutex_exit(proc_lock);
642 }
643 }
644 if (error == 0)
645 *retsize = len - auio.uio_resid;
646
647 bad:
648 if (ktriov != NULL) {
649 ktrgeniov(s, UIO_WRITE, ktriov, *retsize, error);
650 kmem_free(ktriov, iovsz);
651 }
652
653 if (iov != aiov)
654 kmem_free(iov, iovsz);
655 if (to)
656 m_freem(to);
657 if (control)
658 m_freem(control);
659
660 return error;
661 }
662
663 int
664 do_sys_sendmsg(struct lwp *l, int s, struct msghdr *mp, int flags,
665 register_t *retsize)
666 {
667 int error;
668 struct socket *so;
669 file_t *fp;
670
671 if ((error = fd_getsock1(s, &so, &fp)) != 0) {
672 /* We have to free msg_name and msg_control ourselves */
673 if (mp->msg_flags & MSG_NAMEMBUF)
674 m_freem(mp->msg_name);
675 if (mp->msg_flags & MSG_CONTROLMBUF)
676 m_freem(mp->msg_control);
677 return error;
678 }
679 error = do_sys_sendmsg_so(l, s, so, fp, mp, flags, retsize);
680 /* msg_name and msg_control freed */
681 fd_putfile(s);
682 return error;
683 }
684
685 int
686 sys_recvfrom(struct lwp *l, const struct sys_recvfrom_args *uap,
687 register_t *retval)
688 {
689 /* {
690 syscallarg(int) s;
691 syscallarg(void *) buf;
692 syscallarg(size_t) len;
693 syscallarg(int) flags;
694 syscallarg(struct sockaddr *) from;
695 syscallarg(unsigned int *) fromlenaddr;
696 } */
697 struct msghdr msg;
698 struct iovec aiov;
699 int error;
700 struct mbuf *from;
701
702 msg.msg_name = NULL;
703 msg.msg_iov = &aiov;
704 msg.msg_iovlen = 1;
705 aiov.iov_base = SCARG(uap, buf);
706 aiov.iov_len = SCARG(uap, len);
707 msg.msg_control = NULL;
708 msg.msg_flags = SCARG(uap, flags) & MSG_USERFLAGS;
709
710 error = do_sys_recvmsg(l, SCARG(uap, s), &msg, &from, NULL, retval);
711 if (error != 0)
712 return error;
713
714 error = copyout_sockname(SCARG(uap, from), SCARG(uap, fromlenaddr),
715 MSG_LENUSRSPACE, from);
716 if (from != NULL)
717 m_free(from);
718 return error;
719 }
720
721 int
722 sys_recvmsg(struct lwp *l, const struct sys_recvmsg_args *uap,
723 register_t *retval)
724 {
725 /* {
726 syscallarg(int) s;
727 syscallarg(struct msghdr *) msg;
728 syscallarg(int) flags;
729 } */
730 struct msghdr msg;
731 int error;
732 struct mbuf *from, *control;
733
734 error = copyin(SCARG(uap, msg), &msg, sizeof(msg));
735 if (error)
736 return error;
737
738 msg.msg_flags = (SCARG(uap, flags) & MSG_USERFLAGS) | MSG_IOVUSRSPACE;
739
740 error = do_sys_recvmsg(l, SCARG(uap, s), &msg, &from,
741 msg.msg_control != NULL ? &control : NULL, retval);
742 if (error != 0)
743 return error;
744
745 if (msg.msg_control != NULL)
746 error = copyout_msg_control(l, &msg, control);
747
748 if (error == 0)
749 error = copyout_sockname(msg.msg_name, &msg.msg_namelen, 0,
750 from);
751 if (from != NULL)
752 m_free(from);
753 if (error == 0) {
754 ktrkuser("msghdr", &msg, sizeof(msg));
755 error = copyout(&msg, SCARG(uap, msg), sizeof(msg));
756 }
757
758 return error;
759 }
760
761 int
762 sys_sendmmsg(struct lwp *l, const struct sys_sendmmsg_args *uap,
763 register_t *retval)
764 {
765 /* {
766 syscallarg(int) s;
767 syscallarg(struct mmsghdr *) mmsg;
768 syscallarg(unsigned int) vlen;
769 syscallarg(unsigned int) flags;
770 } */
771 struct mmsghdr mmsg;
772 struct socket *so;
773 file_t *fp;
774 struct msghdr *msg = &mmsg.msg_hdr;
775 int error, s;
776 unsigned int vlen, flags, dg;
777
778 s = SCARG(uap, s);
779 if ((error = fd_getsock1(s, &so, &fp)) != 0)
780 return error;
781
782 vlen = SCARG(uap, vlen);
783 if (vlen > 1024)
784 vlen = 1024;
785
786 flags = (SCARG(uap, flags) & MSG_USERFLAGS) | MSG_IOVUSRSPACE;
787
788 for (dg = 0; dg < vlen;) {
789 error = copyin(SCARG(uap, mmsg) + dg, &mmsg, sizeof(mmsg));
790 if (error)
791 break;
792
793 msg->msg_flags = flags;
794
795 error = do_sys_sendmsg_so(l, s, so, fp, msg, flags, retval);
796 if (error)
797 break;
798
799 ktrkuser("msghdr", msg, sizeof(*msg));
800 mmsg.msg_len = *retval;
801 error = copyout(&mmsg, SCARG(uap, mmsg) + dg, sizeof(mmsg));
802 if (error)
803 break;
804 dg++;
805
806 }
807
808 *retval = dg;
809 if (error)
810 so->so_error = error;
811
812 fd_putfile(s);
813
814 /*
815 * If we succeeded at least once, return 0, hopefully so->so_error
816 * will catch it next time.
817 */
818 if (dg)
819 return 0;
820 return error;
821 }
822
823 /*
824 * Adjust for a truncated SCM_RIGHTS control message.
825 * This means closing any file descriptors that aren't present
826 * in the returned buffer.
827 * m is the mbuf holding the (already externalized) SCM_RIGHTS message.
828 */
829 static void
830 free_rights(struct mbuf *m)
831 {
832 struct cmsghdr *cm;
833 int *fdv;
834 unsigned int nfds, i;
835
836 KASSERT(sizeof(*cm) <= m->m_len);
837 cm = mtod(m, struct cmsghdr *);
838
839 KASSERT(CMSG_ALIGN(sizeof(*cm)) <= cm->cmsg_len);
840 KASSERT(cm->cmsg_len <= m->m_len);
841 nfds = (cm->cmsg_len - CMSG_ALIGN(sizeof(*cm))) / sizeof(int);
842 fdv = (int *)CMSG_DATA(cm);
843
844 for (i = 0; i < nfds; i++)
845 if (fd_getfile(fdv[i]) != NULL)
846 (void)fd_close(fdv[i]);
847 }
848
849 void
850 free_control_mbuf(struct lwp *l, struct mbuf *control, struct mbuf *uncopied)
851 {
852 struct mbuf *next;
853 struct cmsghdr *cmsg;
854 bool do_free_rights = false;
855
856 while (control != NULL) {
857 cmsg = mtod(control, struct cmsghdr *);
858 if (control == uncopied)
859 do_free_rights = true;
860 if (do_free_rights && cmsg->cmsg_level == SOL_SOCKET
861 && cmsg->cmsg_type == SCM_RIGHTS)
862 free_rights(control);
863 next = control->m_next;
864 m_free(control);
865 control = next;
866 }
867 }
868
869 /* Copy socket control/CMSG data to user buffer, frees the mbuf */
870 int
871 copyout_msg_control(struct lwp *l, struct msghdr *mp, struct mbuf *control)
872 {
873 int i, len, error = 0;
874 struct cmsghdr *cmsg;
875 struct mbuf *m;
876 char *q;
877
878 len = mp->msg_controllen;
879 if (len <= 0 || control == 0) {
880 mp->msg_controllen = 0;
881 free_control_mbuf(l, control, control);
882 return 0;
883 }
884
885 q = (char *)mp->msg_control;
886
887 for (m = control; m != NULL; ) {
888 cmsg = mtod(m, struct cmsghdr *);
889 i = m->m_len;
890 if (len < i) {
891 mp->msg_flags |= MSG_CTRUNC;
892 if (cmsg->cmsg_level == SOL_SOCKET
893 && cmsg->cmsg_type == SCM_RIGHTS)
894 /* Do not truncate me ... */
895 break;
896 i = len;
897 }
898 error = copyout(mtod(m, void *), q, i);
899 ktrkuser(mbuftypes[MT_CONTROL], cmsg, cmsg->cmsg_len);
900 if (error != 0) {
901 /* We must free all the SCM_RIGHTS */
902 m = control;
903 break;
904 }
905 m = m->m_next;
906 if (m)
907 i = ALIGN(i);
908 q += i;
909 len -= i;
910 if (len <= 0)
911 break;
912 }
913
914 free_control_mbuf(l, control, m);
915
916 mp->msg_controllen = q - (char *)mp->msg_control;
917 return error;
918 }
919
920 int
921 do_sys_recvmsg_so(struct lwp *l, int s, struct socket *so, struct msghdr *mp,
922 struct mbuf **from, struct mbuf **control, register_t *retsize)
923 {
924 struct iovec aiov[UIO_SMALLIOV], *iov = aiov, *tiov, *ktriov = NULL;
925 struct uio auio;
926 size_t len, iovsz;
927 int i, error;
928
929 ktrkuser("msghdr", mp, sizeof(*mp));
930
931 *from = NULL;
932 if (control != NULL)
933 *control = NULL;
934
935 iovsz = mp->msg_iovlen * sizeof(struct iovec);
936
937 if (mp->msg_flags & MSG_IOVUSRSPACE) {
938 if ((unsigned int)mp->msg_iovlen > UIO_SMALLIOV) {
939 if ((unsigned int)mp->msg_iovlen > IOV_MAX) {
940 error = EMSGSIZE;
941 goto out;
942 }
943 iov = kmem_alloc(iovsz, KM_SLEEP);
944 }
945 if (mp->msg_iovlen != 0) {
946 error = copyin(mp->msg_iov, iov, iovsz);
947 if (error)
948 goto out;
949 }
950 auio.uio_iov = iov;
951 } else
952 auio.uio_iov = mp->msg_iov;
953 auio.uio_iovcnt = mp->msg_iovlen;
954 auio.uio_rw = UIO_READ;
955 auio.uio_offset = 0; /* XXX */
956 auio.uio_resid = 0;
957 KASSERT(l == curlwp);
958 auio.uio_vmspace = l->l_proc->p_vmspace;
959
960 tiov = auio.uio_iov;
961 for (i = 0; i < auio.uio_iovcnt; i++, tiov++) {
962 /*
963 * Reads return ssize_t because -1 is returned on error.
964 * Therefore we must restrict the length to SSIZE_MAX to
965 * avoid garbage return values.
966 */
967 auio.uio_resid += tiov->iov_len;
968 if (tiov->iov_len > SSIZE_MAX || auio.uio_resid > SSIZE_MAX) {
969 error = EINVAL;
970 goto out;
971 }
972 }
973
974 if (ktrpoint(KTR_GENIO) && iovsz > 0) {
975 ktriov = kmem_alloc(iovsz, KM_SLEEP);
976 memcpy(ktriov, auio.uio_iov, iovsz);
977 }
978
979 len = auio.uio_resid;
980 mp->msg_flags &= MSG_USERFLAGS;
981 error = (*so->so_receive)(so, from, &auio, NULL, control,
982 &mp->msg_flags);
983 KASSERT(*from == NULL || (*from)->m_next == NULL);
984 len -= auio.uio_resid;
985 *retsize = len;
986 if (error != 0 && len != 0
987 && (error == ERESTART || error == EINTR || error == EWOULDBLOCK))
988 /* Some data transferred */
989 error = 0;
990
991 if (ktriov != NULL) {
992 ktrgeniov(s, UIO_READ, ktriov, len, error);
993 kmem_free(ktriov, iovsz);
994 }
995
996 if (error != 0) {
997 m_freem(*from);
998 *from = NULL;
999 if (control != NULL) {
1000 free_control_mbuf(l, *control, *control);
1001 *control = NULL;
1002 }
1003 }
1004 out:
1005 if (iov != aiov)
1006 kmem_free(iov, iovsz);
1007 return error;
1008 }
1009
1010
1011 int
1012 do_sys_recvmsg(struct lwp *l, int s, struct msghdr *mp,
1013 struct mbuf **from, struct mbuf **control, register_t *retsize)
1014 {
1015 int error;
1016 struct socket *so;
1017
1018 if ((error = fd_getsock(s, &so)) != 0)
1019 return error;
1020 error = do_sys_recvmsg_so(l, s, so, mp, from, control, retsize);
1021 fd_putfile(s);
1022 return error;
1023 }
1024
1025 int
1026 sys_recvmmsg(struct lwp *l, const struct sys_recvmmsg_args *uap,
1027 register_t *retval)
1028 {
1029 /* {
1030 syscallarg(int) s;
1031 syscallarg(struct mmsghdr *) mmsg;
1032 syscallarg(unsigned int) vlen;
1033 syscallarg(unsigned int) flags;
1034 syscallarg(struct timespec *) timeout;
1035 } */
1036 struct mmsghdr mmsg;
1037 struct socket *so;
1038 struct msghdr *msg = &mmsg.msg_hdr;
1039 int error, s;
1040 struct mbuf *from, *control;
1041 struct timespec ts, now;
1042 unsigned int vlen, flags, dg;
1043
1044 if (SCARG(uap, timeout)) {
1045 if ((error = copyin(SCARG(uap, timeout), &ts, sizeof(ts))) != 0)
1046 return error;
1047 getnanotime(&now);
1048 timespecadd(&now, &ts, &ts);
1049 }
1050
1051 s = SCARG(uap, s);
1052 if ((error = fd_getsock(s, &so)) != 0)
1053 return error;
1054
1055 vlen = SCARG(uap, vlen);
1056 if (vlen > 1024)
1057 vlen = 1024;
1058
1059 from = NULL;
1060 flags = (SCARG(uap, flags) & MSG_USERFLAGS) | MSG_IOVUSRSPACE;
1061
1062 for (dg = 0; dg < vlen;) {
1063 error = copyin(SCARG(uap, mmsg) + dg, &mmsg, sizeof(mmsg));
1064 if (error)
1065 break;
1066
1067 msg->msg_flags = flags & ~MSG_WAITFORONE;
1068
1069 if (from != NULL) {
1070 m_free(from);
1071 from = NULL;
1072 }
1073
1074 error = do_sys_recvmsg_so(l, s, so, msg, &from,
1075 msg->msg_control != NULL ? &control : NULL, retval);
1076 if (error) {
1077 if (error == EAGAIN && dg > 0)
1078 error = 0;
1079 break;
1080 }
1081
1082 if (msg->msg_control != NULL)
1083 error = copyout_msg_control(l, msg, control);
1084 if (error)
1085 break;
1086
1087 error = copyout_sockname(msg->msg_name, &msg->msg_namelen, 0,
1088 from);
1089 if (error)
1090 break;
1091
1092 ktrkuser("msghdr", msg, sizeof *msg);
1093 mmsg.msg_len = *retval;
1094
1095 error = copyout(&mmsg, SCARG(uap, mmsg) + dg, sizeof(mmsg));
1096 if (error)
1097 break;
1098
1099 dg++;
1100 if (msg->msg_flags & MSG_OOB)
1101 break;
1102
1103 if (SCARG(uap, timeout)) {
1104 getnanotime(&now);
1105 timespecsub(&now, &ts, &now);
1106 if (now.tv_sec > 0)
1107 break;
1108 }
1109
1110 if (flags & MSG_WAITFORONE)
1111 flags |= MSG_DONTWAIT;
1112
1113 }
1114
1115 if (from != NULL)
1116 m_free(from);
1117
1118 *retval = dg;
1119 if (error)
1120 so->so_error = error;
1121
1122 fd_putfile(s);
1123
1124 /*
1125 * If we succeeded at least once, return 0, hopefully so->so_error
1126 * will catch it next time.
1127 */
1128 if (dg)
1129 return 0;
1130
1131 return error;
1132 }
1133
1134 int
1135 sys_shutdown(struct lwp *l, const struct sys_shutdown_args *uap,
1136 register_t *retval)
1137 {
1138 /* {
1139 syscallarg(int) s;
1140 syscallarg(int) how;
1141 } */
1142 struct socket *so;
1143 int error;
1144
1145 if ((error = fd_getsock(SCARG(uap, s), &so)) != 0)
1146 return error;
1147 solock(so);
1148 error = soshutdown(so, SCARG(uap, how));
1149 sounlock(so);
1150 fd_putfile(SCARG(uap, s));
1151 return error;
1152 }
1153
1154 int
1155 sys_setsockopt(struct lwp *l, const struct sys_setsockopt_args *uap,
1156 register_t *retval)
1157 {
1158 /* {
1159 syscallarg(int) s;
1160 syscallarg(int) level;
1161 syscallarg(int) name;
1162 syscallarg(const void *) val;
1163 syscallarg(unsigned int) valsize;
1164 } */
1165 struct sockopt sopt;
1166 struct socket *so;
1167 file_t *fp;
1168 int error;
1169 unsigned int len;
1170
1171 len = SCARG(uap, valsize);
1172 if (len > 0 && SCARG(uap, val) == NULL)
1173 return EINVAL;
1174
1175 if (len > MCLBYTES)
1176 return EINVAL;
1177
1178 if ((error = fd_getsock1(SCARG(uap, s), &so, &fp)) != 0)
1179 return (error);
1180
1181 sockopt_init(&sopt, SCARG(uap, level), SCARG(uap, name), len);
1182
1183 if (len > 0) {
1184 error = copyin(SCARG(uap, val), sopt.sopt_data, len);
1185 if (error)
1186 goto out;
1187 }
1188
1189 error = sosetopt(so, &sopt);
1190 if (so->so_options & SO_NOSIGPIPE)
1191 atomic_or_uint(&fp->f_flag, FNOSIGPIPE);
1192 else
1193 atomic_and_uint(&fp->f_flag, ~FNOSIGPIPE);
1194
1195 out:
1196 sockopt_destroy(&sopt);
1197 fd_putfile(SCARG(uap, s));
1198 return error;
1199 }
1200
1201 static int
1202 getsockopt(struct lwp *l, const struct sys_getsockopt_args *uap,
1203 register_t *retval, bool copyarg)
1204 {
1205 struct sockopt sopt;
1206 struct socket *so;
1207 file_t *fp;
1208 unsigned int valsize, len;
1209 int error;
1210
1211 if (SCARG(uap, val) != NULL) {
1212 error = copyin(SCARG(uap, avalsize), &valsize, sizeof(valsize));
1213 if (error)
1214 return error;
1215 } else
1216 valsize = 0;
1217
1218 if (valsize > MCLBYTES)
1219 return EINVAL;
1220
1221 if ((error = fd_getsock1(SCARG(uap, s), &so, &fp)) != 0)
1222 return error;
1223
1224 sockopt_init(&sopt, SCARG(uap, level), SCARG(uap, name), valsize);
1225 if (copyarg && valsize > 0) {
1226 error = copyin(SCARG(uap, val), sopt.sopt_data, valsize);
1227 if (error)
1228 goto out;
1229 }
1230
1231 if (fp->f_flag & FNOSIGPIPE)
1232 so->so_options |= SO_NOSIGPIPE;
1233 else
1234 so->so_options &= ~SO_NOSIGPIPE;
1235
1236 error = sogetopt(so, &sopt);
1237 if (error || valsize == 0)
1238 goto out;
1239
1240 len = min(valsize, sopt.sopt_retsize);
1241 error = copyout(sopt.sopt_data, SCARG(uap, val), len);
1242 if (error)
1243 goto out;
1244
1245 error = copyout(&len, SCARG(uap, avalsize), sizeof(len));
1246 out:
1247 sockopt_destroy(&sopt);
1248 fd_putfile(SCARG(uap, s));
1249 return error;
1250 }
1251
1252 int
1253 sys_getsockopt(struct lwp *l, const struct sys_getsockopt_args *uap,
1254 register_t *retval)
1255 {
1256 /* {
1257 syscallarg(int) s;
1258 syscallarg(int) level;
1259 syscallarg(int) name;
1260 syscallarg(void *) val;
1261 syscallarg(unsigned int *) avalsize;
1262 } */
1263 return getsockopt(l, uap, retval, false);
1264 }
1265
1266 int
1267 sys_getsockopt2(struct lwp *l, const struct sys_getsockopt2_args *uap,
1268 register_t *retval)
1269 {
1270 /* {
1271 syscallarg(int) s;
1272 syscallarg(int) level;
1273 syscallarg(int) name;
1274 syscallarg(void *) val;
1275 syscallarg(unsigned int *) avalsize;
1276 } */
1277 return getsockopt(l, (const struct sys_getsockopt_args *) uap, retval, true);
1278 }
1279
1280 #ifdef PIPE_SOCKETPAIR
1281
1282 int
1283 pipe1(struct lwp *l, int *fildes, int flags)
1284 {
1285 file_t *rf, *wf;
1286 struct socket *rso, *wso;
1287 int fd, error;
1288 proc_t *p;
1289
1290 if (flags & ~(O_CLOEXEC|O_NONBLOCK|O_NOSIGPIPE))
1291 return EINVAL;
1292 p = curproc;
1293 if ((error = socreate(AF_LOCAL, &rso, SOCK_STREAM, 0, l, NULL)) != 0)
1294 return error;
1295 if ((error = socreate(AF_LOCAL, &wso, SOCK_STREAM, 0, l, rso)) != 0)
1296 goto free1;
1297 /* remember this socket pair implements a pipe */
1298 wso->so_state |= SS_ISAPIPE;
1299 rso->so_state |= SS_ISAPIPE;
1300 if ((error = fd_allocfile(&rf, &fd)) != 0)
1301 goto free2;
1302 fildes[0] = fd;
1303 rf->f_flag = FREAD | flags;
1304 rf->f_type = DTYPE_SOCKET;
1305 rf->f_ops = &socketops;
1306 rf->f_socket = rso;
1307 if ((error = fd_allocfile(&wf, &fd)) != 0)
1308 goto free3;
1309 wf->f_flag = FWRITE | flags;
1310 wf->f_type = DTYPE_SOCKET;
1311 wf->f_ops = &socketops;
1312 wf->f_socket = wso;
1313 fildes[1] = fd;
1314 solock(wso);
1315 error = unp_connect2(wso, rso);
1316 sounlock(wso);
1317 if (error != 0)
1318 goto free4;
1319 fd_affix(p, wf, fildes[1]);
1320 fd_affix(p, rf, fildes[0]);
1321 return (0);
1322 free4:
1323 fd_abort(p, wf, fildes[1]);
1324 free3:
1325 fd_abort(p, rf, fildes[0]);
1326 free2:
1327 (void)soclose(wso);
1328 free1:
1329 (void)soclose(rso);
1330 return error;
1331 }
1332 #endif /* PIPE_SOCKETPAIR */
1333
1334 /*
1335 * Get peer socket name.
1336 */
1337 int
1338 do_sys_getpeername(int fd, struct sockaddr *nam)
1339 {
1340 struct socket *so;
1341 int error;
1342
1343 if ((error = fd_getsock(fd, &so)) != 0)
1344 return error;
1345
1346 solock(so);
1347 if ((so->so_state & SS_ISCONNECTED) == 0)
1348 error = ENOTCONN;
1349 else {
1350 error = (*so->so_proto->pr_usrreqs->pr_peeraddr)(so, nam);
1351 }
1352 sounlock(so);
1353 fd_putfile(fd);
1354 return error;
1355 }
1356
1357 /*
1358 * Get local socket name.
1359 */
1360 int
1361 do_sys_getsockname(int fd, struct sockaddr *nam)
1362 {
1363 struct socket *so;
1364 int error;
1365
1366 if ((error = fd_getsock(fd, &so)) != 0)
1367 return error;
1368
1369 solock(so);
1370 error = (*so->so_proto->pr_usrreqs->pr_sockaddr)(so, nam);
1371 sounlock(so);
1372 fd_putfile(fd);
1373 return error;
1374 }
1375
1376 int
1377 copyout_sockname_sb(struct sockaddr *asa, unsigned int *alen, int flags,
1378 struct sockaddr_big *addr)
1379 {
1380 unsigned int len;
1381 int error;
1382
1383 if (asa == NULL)
1384 /* Assume application not interested */
1385 return 0;
1386
1387 if (flags & MSG_LENUSRSPACE) {
1388 error = copyin(alen, &len, sizeof(len));
1389 if (error)
1390 return error;
1391 } else
1392 len = *alen;
1393
1394 if (addr == NULL) {
1395 len = 0;
1396 error = 0;
1397 } else {
1398 if (len > addr->sb_len)
1399 len = addr->sb_len;
1400 /* XXX addr isn't an mbuf... */
1401 ktrkuser(mbuftypes[MT_SONAME], addr, len);
1402 error = copyout(addr, asa, len);
1403 }
1404
1405 if (error == 0) {
1406 if (flags & MSG_LENUSRSPACE)
1407 error = copyout(&len, alen, sizeof(len));
1408 else
1409 *alen = len;
1410 }
1411
1412 return error;
1413 }
1414
1415 int
1416 copyout_sockname(struct sockaddr *asa, unsigned int *alen, int flags,
1417 struct mbuf *addr)
1418 {
1419 int len;
1420 int error;
1421
1422 if (asa == NULL)
1423 /* Assume application not interested */
1424 return 0;
1425
1426 if (flags & MSG_LENUSRSPACE) {
1427 error = copyin(alen, &len, sizeof(len));
1428 if (error)
1429 return error;
1430 } else
1431 len = *alen;
1432 if (len < 0)
1433 return EINVAL;
1434
1435 if (addr == NULL) {
1436 len = 0;
1437 error = 0;
1438 } else {
1439 if (len > addr->m_len)
1440 len = addr->m_len;
1441 /* Maybe this ought to copy a chain ? */
1442 ktrkuser(mbuftypes[MT_SONAME], mtod(addr, void *), len);
1443 error = copyout(mtod(addr, void *), asa, len);
1444 }
1445
1446 if (error == 0) {
1447 if (flags & MSG_LENUSRSPACE)
1448 error = copyout(&len, alen, sizeof(len));
1449 else
1450 *alen = len;
1451 }
1452
1453 return error;
1454 }
1455
1456 /*
1457 * Get socket name.
1458 */
1459 int
1460 sys_getsockname(struct lwp *l, const struct sys_getsockname_args *uap,
1461 register_t *retval)
1462 {
1463 /* {
1464 syscallarg(int) fdes;
1465 syscallarg(struct sockaddr *) asa;
1466 syscallarg(unsigned int *) alen;
1467 } */
1468 struct sockaddr_big sbig;
1469 int error;
1470
1471 sbig.sb_len = UCHAR_MAX;
1472 error = do_sys_getsockname(SCARG(uap, fdes), (struct sockaddr *)&sbig);
1473 if (error != 0)
1474 return error;
1475
1476 error = copyout_sockname_sb(SCARG(uap, asa), SCARG(uap, alen),
1477 MSG_LENUSRSPACE, &sbig);
1478 return error;
1479 }
1480
1481 /*
1482 * Get name of peer for connected socket.
1483 */
1484 int
1485 sys_getpeername(struct lwp *l, const struct sys_getpeername_args *uap,
1486 register_t *retval)
1487 {
1488 /* {
1489 syscallarg(int) fdes;
1490 syscallarg(struct sockaddr *) asa;
1491 syscallarg(unsigned int *) alen;
1492 } */
1493 struct sockaddr_big sbig;
1494 int error;
1495
1496 sbig.sb_len = UCHAR_MAX;
1497 error = do_sys_getpeername(SCARG(uap, fdes), (struct sockaddr *)&sbig);
1498 if (error != 0)
1499 return error;
1500
1501 error = copyout_sockname_sb(SCARG(uap, asa), SCARG(uap, alen),
1502 MSG_LENUSRSPACE, &sbig);
1503 return error;
1504 }
1505
1506 static int
1507 sockargs_sb(struct sockaddr_big *sb, const void *name, socklen_t buflen)
1508 {
1509 int error;
1510
1511 /*
1512 * We can't allow socket names > UCHAR_MAX in length, since that
1513 * will overflow sb_len. Further no reasonable buflen is <=
1514 * offsetof(sockaddr_big, sb_data) since it shall be at least
1515 * the size of the preamble sb_len and sb_family members.
1516 */
1517 if (buflen > UCHAR_MAX ||
1518 buflen <= offsetof(struct sockaddr_big, sb_data))
1519 return EINVAL;
1520
1521 error = copyin(name, (void *)sb, buflen);
1522 if (error)
1523 return error;
1524
1525 ktrkuser(mbuftypes[MT_SONAME], sb, buflen);
1526 #if BYTE_ORDER != BIG_ENDIAN
1527 /*
1528 * 4.3BSD compat thing - need to stay, since bind(2),
1529 * connect(2), sendto(2) were not versioned for COMPAT_43.
1530 */
1531 if (sb->sb_family == 0 && sb->sb_len < AF_MAX)
1532 sb->sb_family = sb->sb_len;
1533 #endif
1534 sb->sb_len = buflen;
1535 return 0;
1536 }
1537
1538 /*
1539 * XXX In a perfect world, we wouldn't pass around socket control
1540 * XXX arguments in mbufs, and this could go away.
1541 */
1542 int
1543 sockargs(struct mbuf **mp, const void *bf, size_t buflen, enum uio_seg seg,
1544 int type)
1545 {
1546 struct mbuf *m;
1547 int error;
1548
1549 /*
1550 * We can't allow socket names > UCHAR_MAX in length, since that
1551 * will overflow sa_len. Control data more than a page size in
1552 * length is just too much.
1553 */
1554 if (buflen > (type == MT_SONAME ? UCHAR_MAX : PAGE_SIZE))
1555 return EINVAL;
1556
1557 /*
1558 * length must greater than sizeof(sa_family) + sizeof(sa_len)
1559 */
1560 if (type == MT_SONAME && buflen <= 2)
1561 return EINVAL;
1562
1563 /* Allocate an mbuf to hold the arguments. */
1564 m = m_get(M_WAIT, type);
1565 /* can't claim. don't who to assign it to. */
1566 if (buflen > MLEN) {
1567 /*
1568 * Won't fit into a regular mbuf, so we allocate just
1569 * enough external storage to hold the argument.
1570 */
1571 MEXTMALLOC(m, buflen, M_WAITOK);
1572 }
1573 m->m_len = buflen;
1574 if (seg == UIO_USERSPACE) {
1575 error = copyin(bf, mtod(m, void *), buflen);
1576 if (error) {
1577 (void)m_free(m);
1578 return error;
1579 }
1580 } else {
1581 memcpy(mtod(m, void *), bf, buflen);
1582 }
1583 *mp = m;
1584 switch (type) {
1585 case MT_SONAME:
1586 ktrkuser(mbuftypes[type], mtod(m, void *), buflen);
1587
1588 struct sockaddr *sa = mtod(m, struct sockaddr *);
1589 #if BYTE_ORDER != BIG_ENDIAN
1590 /*
1591 * 4.3BSD compat thing - need to stay, since bind(2),
1592 * connect(2), sendto(2) were not versioned for COMPAT_43.
1593 */
1594 if (sa->sa_family == 0 && sa->sa_len < AF_MAX)
1595 sa->sa_family = sa->sa_len;
1596 #endif
1597 sa->sa_len = buflen;
1598 return 0;
1599 case MT_CONTROL:
1600 if (!KTRPOINT(curproc, KTR_USER))
1601 return 0;
1602
1603 struct msghdr mhdr;
1604 mhdr.msg_control = mtod(m, void *);
1605 mhdr.msg_controllen = buflen;
1606 for (struct cmsghdr *cmsg = CMSG_FIRSTHDR(&mhdr); cmsg;
1607 cmsg = CMSG_NXTHDR(&mhdr, cmsg)) {
1608 ktrkuser(mbuftypes[type], cmsg, cmsg->cmsg_len);
1609 }
1610 return 0;
1611 default:
1612 return EINVAL;
1613 }
1614 }
1615
1616 int
1617 do_sys_peeloff(struct socket *head, void *data)
1618 {
1619 #ifdef SCTP
1620 /*file_t *lfp = NULL;*/
1621 file_t *nfp = NULL;
1622 int error;
1623 struct socket *so;
1624 int fd;
1625 uint32_t name;
1626 /*short fflag;*/ /* type must match fp->f_flag */
1627
1628 name = *(uint32_t *) data;
1629 error = sctp_can_peel_off(head, name);
1630 if (error) {
1631 printf("peeloff failed\n");
1632 return error;
1633 }
1634 /*
1635 * At this point we know we do have a assoc to pull
1636 * we proceed to get the fd setup. This may block
1637 * but that is ok.
1638 */
1639 error = fd_allocfile(&nfp, &fd);
1640 if (error) {
1641 /*
1642 * Probably ran out of file descriptors. Put the
1643 * unaccepted connection back onto the queue and
1644 * do another wakeup so some other process might
1645 * have a chance at it.
1646 */
1647 return error;
1648 }
1649 *(int *) data = fd;
1650
1651 so = sctp_get_peeloff(head, name, &error);
1652 if (so == NULL) {
1653 /*
1654 * Either someone else peeled it off OR
1655 * we can't get a socket.
1656 * close the new descriptor, assuming someone hasn't ripped it
1657 * out from under us.
1658 */
1659 mutex_enter(&nfp->f_lock);
1660 nfp->f_count++;
1661 mutex_exit(&nfp->f_lock);
1662 fd_abort(curlwp->l_proc, nfp, fd);
1663 return error;
1664 }
1665 so->so_state &= ~SS_NOFDREF;
1666 so->so_state &= ~SS_ISCONNECTING;
1667 so->so_head = NULL;
1668 so->so_cred = kauth_cred_dup(head->so_cred);
1669 nfp->f_socket = so;
1670 nfp->f_flag = FREAD|FWRITE;
1671 nfp->f_ops = &socketops;
1672 nfp->f_type = DTYPE_SOCKET;
1673
1674 fd_affix(curlwp->l_proc, nfp, fd);
1675
1676 return error;
1677 #else
1678 return EOPNOTSUPP;
1679 #endif
1680 }
1681