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