xref: /freebsd/sys/kern/sys_generic.c (revision f28526e9466cd60ed33053e922238ba1c9040341)
19454b2d8SWarner Losh /*-
251369649SPedro F. Giffuni  * SPDX-License-Identifier: BSD-3-Clause
351369649SPedro F. Giffuni  *
4df8bae1dSRodney W. Grimes  * Copyright (c) 1982, 1986, 1989, 1993
5df8bae1dSRodney W. Grimes  *	The Regents of the University of California.  All rights reserved.
6df8bae1dSRodney W. Grimes  * (c) UNIX System Laboratories, Inc.
7df8bae1dSRodney W. Grimes  * All or some portions of this file are derived from material licensed
8df8bae1dSRodney W. Grimes  * to the University of California by American Telephone and Telegraph
9df8bae1dSRodney W. Grimes  * Co. or Unix System Laboratories, Inc. and are reproduced herein with
10df8bae1dSRodney W. Grimes  * the permission of UNIX System Laboratories, Inc.
11df8bae1dSRodney W. Grimes  *
12df8bae1dSRodney W. Grimes  * Redistribution and use in source and binary forms, with or without
13df8bae1dSRodney W. Grimes  * modification, are permitted provided that the following conditions
14df8bae1dSRodney W. Grimes  * are met:
15df8bae1dSRodney W. Grimes  * 1. Redistributions of source code must retain the above copyright
16df8bae1dSRodney W. Grimes  *    notice, this list of conditions and the following disclaimer.
17df8bae1dSRodney W. Grimes  * 2. Redistributions in binary form must reproduce the above copyright
18df8bae1dSRodney W. Grimes  *    notice, this list of conditions and the following disclaimer in the
19df8bae1dSRodney W. Grimes  *    documentation and/or other materials provided with the distribution.
2069a28758SEd Maste  * 3. Neither the name of the University nor the names of its contributors
21df8bae1dSRodney W. Grimes  *    may be used to endorse or promote products derived from this software
22df8bae1dSRodney W. Grimes  *    without specific prior written permission.
23df8bae1dSRodney W. Grimes  *
24df8bae1dSRodney W. Grimes  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
25df8bae1dSRodney W. Grimes  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26df8bae1dSRodney W. Grimes  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27df8bae1dSRodney W. Grimes  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28df8bae1dSRodney W. Grimes  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29df8bae1dSRodney W. Grimes  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30df8bae1dSRodney W. Grimes  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31df8bae1dSRodney W. Grimes  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32df8bae1dSRodney W. Grimes  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33df8bae1dSRodney W. Grimes  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34df8bae1dSRodney W. Grimes  * SUCH DAMAGE.
35df8bae1dSRodney W. Grimes  */
36df8bae1dSRodney W. Grimes 
37677b542eSDavid E. O'Brien #include <sys/cdefs.h>
38a9d2f8d8SRobert Watson #include "opt_capsicum.h"
39db6a20e2SGarrett Wollman #include "opt_ktrace.h"
40db6a20e2SGarrett Wollman 
41df8bae1dSRodney W. Grimes #include <sys/param.h>
42df8bae1dSRodney W. Grimes #include <sys/systm.h>
43d2d3e875SBruce Evans #include <sys/sysproto.h>
444a144410SRobert Watson #include <sys/capsicum.h>
45df8bae1dSRodney W. Grimes #include <sys/filedesc.h>
4620982410SBruce Evans #include <sys/filio.h>
473ac4d1efSBruce Evans #include <sys/fcntl.h>
48df8bae1dSRodney W. Grimes #include <sys/file.h>
492609222aSPawel Jakub Dawidek #include <sys/lock.h>
50df8bae1dSRodney W. Grimes #include <sys/proc.h>
51797f2d22SPoul-Henning Kamp #include <sys/signalvar.h>
52df8bae1dSRodney W. Grimes #include <sys/socketvar.h>
53df8bae1dSRodney W. Grimes #include <sys/uio.h>
547a202823SKonstantin Belousov #include <sys/eventfd.h>
55df8bae1dSRodney W. Grimes #include <sys/kernel.h>
56e4650294SJohn Baldwin #include <sys/ktr.h>
57104a9b7eSAlexander Kabaev #include <sys/limits.h>
58df8bae1dSRodney W. Grimes #include <sys/malloc.h>
5942d11757SPeter Wemm #include <sys/poll.h>
6089b71647SPeter Wemm #include <sys/resourcevar.h>
610a2c3d48SGarrett Wollman #include <sys/selinfo.h>
6244f3b092SJohn Baldwin #include <sys/sleepqueue.h>
637a202823SKonstantin Belousov #include <sys/specialfd.h>
648f19eb88SIan Dowse #include <sys/syscallsubr.h>
658cb96f20SPeter Wemm #include <sys/sysctl.h>
6642d11757SPeter Wemm #include <sys/sysent.h>
679bbee259SAndrey A. Chernov #include <sys/vnode.h>
68d8decc9aSKonstantin Belousov #include <sys/unistd.h>
69279d7226SMatthew Dillon #include <sys/bio.h>
70279d7226SMatthew Dillon #include <sys/buf.h>
71265fc98fSSeigo Tanimura #include <sys/condvar.h>
72df8bae1dSRodney W. Grimes #ifdef KTRACE
73df8bae1dSRodney W. Grimes #include <sys/ktrace.h>
74df8bae1dSRodney W. Grimes #endif
75df8bae1dSRodney W. Grimes 
76e4650294SJohn Baldwin #include <security/audit/audit.h>
77ace8398dSJeff Roberson 
7850ae6690SHans Petter Selasky /*
7950ae6690SHans Petter Selasky  * The following macro defines how many bytes will be allocated from
8050ae6690SHans Petter Selasky  * the stack instead of memory allocated when passing the IOCTL data
8150ae6690SHans Petter Selasky  * structures from userspace and to the kernel. Some IOCTLs having
8250ae6690SHans Petter Selasky  * small data structures are used very frequently and this small
8350ae6690SHans Petter Selasky  * buffer on the stack gives a significant speedup improvement for
8450ae6690SHans Petter Selasky  * those requests. The value of this define should be greater or equal
8550ae6690SHans Petter Selasky  * to 64 bytes and should also be power of two. The data structure is
8650ae6690SHans Petter Selasky  * currently hard-aligned to a 8-byte boundary on the stack. This
8750ae6690SHans Petter Selasky  * should currently be sufficient for all supported platforms.
8850ae6690SHans Petter Selasky  */
8950ae6690SHans Petter Selasky #define	SYS_IOCTL_SMALL_SIZE	128	/* bytes */
9050ae6690SHans Petter Selasky #define	SYS_IOCTL_SMALL_ALIGN	8	/* bytes */
9150ae6690SHans Petter Selasky 
92611fcff9SJohn Baldwin #ifdef __LP64__
93611fcff9SJohn Baldwin static int iosize_max_clamp = 0;
948bb9a904SKonstantin Belousov SYSCTL_INT(_debug, OID_AUTO, iosize_max_clamp, CTLFLAG_RW,
958bb9a904SKonstantin Belousov     &iosize_max_clamp, 0, "Clamp max i/o size to INT_MAX");
96611fcff9SJohn Baldwin static int devfs_iosize_max_clamp = 1;
97bf3e483bSKonstantin Belousov SYSCTL_INT(_debug, OID_AUTO, devfs_iosize_max_clamp, CTLFLAG_RW,
98bf3e483bSKonstantin Belousov     &devfs_iosize_max_clamp, 0, "Clamp max i/o size to INT_MAX for devices");
99611fcff9SJohn Baldwin #endif
100bf3e483bSKonstantin Belousov 
1018bb9a904SKonstantin Belousov /*
1028bb9a904SKonstantin Belousov  * Assert that the return value of read(2) and write(2) syscalls fits
1038bb9a904SKonstantin Belousov  * into a register.  If not, an architecture will need to provide the
1048bb9a904SKonstantin Belousov  * usermode wrappers to reconstruct the result.
1058bb9a904SKonstantin Belousov  */
1068bb9a904SKonstantin Belousov CTASSERT(sizeof(register_t) >= sizeof(size_t));
107526d0bd5SKonstantin Belousov 
108a1c995b6SPoul-Henning Kamp static MALLOC_DEFINE(M_IOCTLOPS, "ioctlops", "ioctl data buffer");
109a1c995b6SPoul-Henning Kamp static MALLOC_DEFINE(M_SELECT, "select", "select() buffer");
110a1c995b6SPoul-Henning Kamp MALLOC_DEFINE(M_IOV, "iov", "large iov's");
11155166637SPoul-Henning Kamp 
1126d8feddaSKonstantin Belousov static int	pollout(struct thread *, struct pollfd *, struct pollfd *,
1136d8feddaSKonstantin Belousov 		    u_int);
114bbbb04ceSAlfred Perlstein static int	pollscan(struct thread *, struct pollfd *, u_int);
115ace8398dSJeff Roberson static int	pollrescan(struct thread *);
116bbbb04ceSAlfred Perlstein static int	selscan(struct thread *, fd_mask **, fd_mask **, int);
117ace8398dSJeff Roberson static int	selrescan(struct thread *, fd_mask **, fd_mask **);
118ace8398dSJeff Roberson static void	selfdalloc(struct thread *, void *);
119ace8398dSJeff Roberson static void	selfdfree(struct seltd *, struct selfd *);
120bcd9e0ddSJohn Baldwin static int	dofileread(struct thread *, int, struct file *, struct uio *,
121bcd9e0ddSJohn Baldwin 		    off_t, int);
122bcd9e0ddSJohn Baldwin static int	dofilewrite(struct thread *, int, struct file *, struct uio *,
123bcd9e0ddSJohn Baldwin 		    off_t, int);
124512824f8SSeigo Tanimura static void	doselwakeup(struct selinfo *, int);
125ace8398dSJeff Roberson static void	seltdinit(struct thread *);
126cf5e4fe6SDavide Italiano static int	seltdwait(struct thread *, sbintime_t, sbintime_t);
127ace8398dSJeff Roberson static void	seltdclear(struct thread *);
128ace8398dSJeff Roberson 
129ace8398dSJeff Roberson /*
130ace8398dSJeff Roberson  * One seltd per-thread allocated on demand as needed.
131ace8398dSJeff Roberson  *
132ace8398dSJeff Roberson  *	t - protected by st_mtx
133ace8398dSJeff Roberson  * 	k - Only accessed by curthread or read-only
134ace8398dSJeff Roberson  */
135ace8398dSJeff Roberson struct seltd {
136ace8398dSJeff Roberson 	STAILQ_HEAD(, selfd)	st_selq;	/* (k) List of selfds. */
137ace8398dSJeff Roberson 	struct selfd		*st_free1;	/* (k) free fd for read set. */
138ace8398dSJeff Roberson 	struct selfd		*st_free2;	/* (k) free fd for write set. */
139ace8398dSJeff Roberson 	struct mtx		st_mtx;		/* Protects struct seltd */
140ace8398dSJeff Roberson 	struct cv		st_wait;	/* (t) Wait channel. */
141ace8398dSJeff Roberson 	int			st_flags;	/* (t) SELTD_ flags. */
142ace8398dSJeff Roberson };
143ace8398dSJeff Roberson 
144ace8398dSJeff Roberson #define	SELTD_PENDING	0x0001			/* We have pending events. */
145ace8398dSJeff Roberson #define	SELTD_RESCAN	0x0002			/* Doing a rescan. */
146ace8398dSJeff Roberson 
147ace8398dSJeff Roberson /*
148ace8398dSJeff Roberson  * One selfd allocated per-thread per-file-descriptor.
149ace8398dSJeff Roberson  *	f - protected by sf_mtx
150ace8398dSJeff Roberson  */
151ace8398dSJeff Roberson struct selfd {
152ace8398dSJeff Roberson 	STAILQ_ENTRY(selfd)	sf_link;	/* (k) fds owned by this td. */
153ace8398dSJeff Roberson 	TAILQ_ENTRY(selfd)	sf_threads;	/* (f) fds on this selinfo. */
154ace8398dSJeff Roberson 	struct selinfo		*sf_si;		/* (f) selinfo when linked. */
155ace8398dSJeff Roberson 	struct mtx		*sf_mtx;	/* Pointer to selinfo mtx. */
156ace8398dSJeff Roberson 	struct seltd		*sf_td;		/* (k) owning seltd. */
157ace8398dSJeff Roberson 	void			*sf_cookie;	/* (k) fd or pollfd. */
158ace8398dSJeff Roberson };
159ace8398dSJeff Roberson 
160ea33cca9SMateusz Guzik MALLOC_DEFINE(M_SELFD, "selfd", "selfd");
1612141453eSJeff Roberson static struct mtx_pool *mtxpool_select;
1628fe387abSDmitrij Tejblum 
163611fcff9SJohn Baldwin #ifdef __LP64__
164611fcff9SJohn Baldwin size_t
165611fcff9SJohn Baldwin devfs_iosize_max(void)
166611fcff9SJohn Baldwin {
167611fcff9SJohn Baldwin 
168611fcff9SJohn Baldwin 	return (devfs_iosize_max_clamp || SV_CURPROC_FLAG(SV_ILP32) ?
169611fcff9SJohn Baldwin 	    INT_MAX : SSIZE_MAX);
170611fcff9SJohn Baldwin }
171611fcff9SJohn Baldwin 
172611fcff9SJohn Baldwin size_t
173611fcff9SJohn Baldwin iosize_max(void)
174611fcff9SJohn Baldwin {
175611fcff9SJohn Baldwin 
176611fcff9SJohn Baldwin 	return (iosize_max_clamp || SV_CURPROC_FLAG(SV_ILP32) ?
177611fcff9SJohn Baldwin 	    INT_MAX : SSIZE_MAX);
178611fcff9SJohn Baldwin }
179611fcff9SJohn Baldwin #endif
180611fcff9SJohn Baldwin 
181d2d3e875SBruce Evans #ifndef _SYS_SYSPROTO_H_
182df8bae1dSRodney W. Grimes struct read_args {
183df8bae1dSRodney W. Grimes 	int	fd;
184134e06feSBruce Evans 	void	*buf;
185134e06feSBruce Evans 	size_t	nbyte;
186df8bae1dSRodney W. Grimes };
187d2d3e875SBruce Evans #endif
18826f9a767SRodney W. Grimes int
189cc3c9df8SEd Maste sys_read(struct thread *td, struct read_args *uap)
190df8bae1dSRodney W. Grimes {
191bcd9e0ddSJohn Baldwin 	struct uio auio;
192bcd9e0ddSJohn Baldwin 	struct iovec aiov;
193279d7226SMatthew Dillon 	int error;
194df8bae1dSRodney W. Grimes 
195526d0bd5SKonstantin Belousov 	if (uap->nbyte > IOSIZE_MAX)
196bcd9e0ddSJohn Baldwin 		return (EINVAL);
197bcd9e0ddSJohn Baldwin 	aiov.iov_base = uap->buf;
198bcd9e0ddSJohn Baldwin 	aiov.iov_len = uap->nbyte;
199bcd9e0ddSJohn Baldwin 	auio.uio_iov = &aiov;
200bcd9e0ddSJohn Baldwin 	auio.uio_iovcnt = 1;
201bcd9e0ddSJohn Baldwin 	auio.uio_resid = uap->nbyte;
202bcd9e0ddSJohn Baldwin 	auio.uio_segflg = UIO_USERSPACE;
203bcd9e0ddSJohn Baldwin 	error = kern_readv(td, uap->fd, &auio);
204279d7226SMatthew Dillon 	return (error);
205df8bae1dSRodney W. Grimes }
206df8bae1dSRodney W. Grimes 
207df8bae1dSRodney W. Grimes /*
208bcd9e0ddSJohn Baldwin  * Positioned read system call
2094160ccd9SAlan Cox  */
2104160ccd9SAlan Cox #ifndef _SYS_SYSPROTO_H_
2114160ccd9SAlan Cox struct pread_args {
2124160ccd9SAlan Cox 	int	fd;
2134160ccd9SAlan Cox 	void	*buf;
2144160ccd9SAlan Cox 	size_t	nbyte;
2158fe387abSDmitrij Tejblum 	int	pad;
2164160ccd9SAlan Cox 	off_t	offset;
2174160ccd9SAlan Cox };
2184160ccd9SAlan Cox #endif
2194160ccd9SAlan Cox int
220b38b22b0SEdward Tomasz Napierala sys_pread(struct thread *td, struct pread_args *uap)
221b38b22b0SEdward Tomasz Napierala {
222b38b22b0SEdward Tomasz Napierala 
223b38b22b0SEdward Tomasz Napierala 	return (kern_pread(td, uap->fd, uap->buf, uap->nbyte, uap->offset));
224b38b22b0SEdward Tomasz Napierala }
225b38b22b0SEdward Tomasz Napierala 
226b38b22b0SEdward Tomasz Napierala int
227b38b22b0SEdward Tomasz Napierala kern_pread(struct thread *td, int fd, void *buf, size_t nbyte, off_t offset)
2284160ccd9SAlan Cox {
2294160ccd9SAlan Cox 	struct uio auio;
2304160ccd9SAlan Cox 	struct iovec aiov;
231bcd9e0ddSJohn Baldwin 	int error;
2324160ccd9SAlan Cox 
233b38b22b0SEdward Tomasz Napierala 	if (nbyte > IOSIZE_MAX)
234bcd9e0ddSJohn Baldwin 		return (EINVAL);
235b38b22b0SEdward Tomasz Napierala 	aiov.iov_base = buf;
236b38b22b0SEdward Tomasz Napierala 	aiov.iov_len = nbyte;
2374160ccd9SAlan Cox 	auio.uio_iov = &aiov;
2384160ccd9SAlan Cox 	auio.uio_iovcnt = 1;
239b38b22b0SEdward Tomasz Napierala 	auio.uio_resid = nbyte;
2404160ccd9SAlan Cox 	auio.uio_segflg = UIO_USERSPACE;
241b38b22b0SEdward Tomasz Napierala 	error = kern_preadv(td, fd, &auio, offset);
2424160ccd9SAlan Cox 	return (error);
2434160ccd9SAlan Cox }
2444160ccd9SAlan Cox 
2450538aafcSKonstantin Belousov #if defined(COMPAT_FREEBSD6)
246c2815ad5SPeter Wemm int
247b38b22b0SEdward Tomasz Napierala freebsd6_pread(struct thread *td, struct freebsd6_pread_args *uap)
248c2815ad5SPeter Wemm {
249c2815ad5SPeter Wemm 
250b38b22b0SEdward Tomasz Napierala 	return (kern_pread(td, uap->fd, uap->buf, uap->nbyte, uap->offset));
251c2815ad5SPeter Wemm }
2520538aafcSKonstantin Belousov #endif
253c2815ad5SPeter Wemm 
2544160ccd9SAlan Cox /*
255df8bae1dSRodney W. Grimes  * Scatter read system call.
256df8bae1dSRodney W. Grimes  */
257d2d3e875SBruce Evans #ifndef _SYS_SYSPROTO_H_
258df8bae1dSRodney W. Grimes struct readv_args {
2597147b19dSBruce Evans 	int	fd;
260df8bae1dSRodney W. Grimes 	struct	iovec *iovp;
261df8bae1dSRodney W. Grimes 	u_int	iovcnt;
262df8bae1dSRodney W. Grimes };
263d2d3e875SBruce Evans #endif
26426f9a767SRodney W. Grimes int
2658451d0ddSKip Macy sys_readv(struct thread *td, struct readv_args *uap)
266df8bae1dSRodney W. Grimes {
267b88ec951SJohn Baldwin 	struct uio *auio;
268b88ec951SJohn Baldwin 	int error;
269b88ec951SJohn Baldwin 
270b88ec951SJohn Baldwin 	error = copyinuio(uap->iovp, uap->iovcnt, &auio);
271b88ec951SJohn Baldwin 	if (error)
272b88ec951SJohn Baldwin 		return (error);
273b88ec951SJohn Baldwin 	error = kern_readv(td, uap->fd, auio);
274b88ec951SJohn Baldwin 	free(auio, M_IOV);
275b88ec951SJohn Baldwin 	return (error);
276b88ec951SJohn Baldwin }
277b88ec951SJohn Baldwin 
278b88ec951SJohn Baldwin int
279b88ec951SJohn Baldwin kern_readv(struct thread *td, int fd, struct uio *auio)
280b88ec951SJohn Baldwin {
281b064d43dSMatthew Dillon 	struct file *fp;
282bcd9e0ddSJohn Baldwin 	int error;
283bcd9e0ddSJohn Baldwin 
284cbd92ce6SMatt Macy 	error = fget_read(td, fd, &cap_read_rights, &fp);
285bcd9e0ddSJohn Baldwin 	if (error)
286bcd9e0ddSJohn Baldwin 		return (error);
287bcd9e0ddSJohn Baldwin 	error = dofileread(td, fd, fp, auio, (off_t)-1, 0);
288bcd9e0ddSJohn Baldwin 	fdrop(fp, td);
289bcd9e0ddSJohn Baldwin 	return (error);
290bcd9e0ddSJohn Baldwin }
291bcd9e0ddSJohn Baldwin 
292bcd9e0ddSJohn Baldwin /*
293bcd9e0ddSJohn Baldwin  * Scatter positioned read system call.
294bcd9e0ddSJohn Baldwin  */
295bcd9e0ddSJohn Baldwin #ifndef _SYS_SYSPROTO_H_
296bcd9e0ddSJohn Baldwin struct preadv_args {
297bcd9e0ddSJohn Baldwin 	int	fd;
298bcd9e0ddSJohn Baldwin 	struct	iovec *iovp;
299bcd9e0ddSJohn Baldwin 	u_int	iovcnt;
300bcd9e0ddSJohn Baldwin 	off_t	offset;
301bcd9e0ddSJohn Baldwin };
302bcd9e0ddSJohn Baldwin #endif
303bcd9e0ddSJohn Baldwin int
3048451d0ddSKip Macy sys_preadv(struct thread *td, struct preadv_args *uap)
305bcd9e0ddSJohn Baldwin {
306bcd9e0ddSJohn Baldwin 	struct uio *auio;
307bcd9e0ddSJohn Baldwin 	int error;
308bcd9e0ddSJohn Baldwin 
309bcd9e0ddSJohn Baldwin 	error = copyinuio(uap->iovp, uap->iovcnt, &auio);
310bcd9e0ddSJohn Baldwin 	if (error)
311bcd9e0ddSJohn Baldwin 		return (error);
312bcd9e0ddSJohn Baldwin 	error = kern_preadv(td, uap->fd, auio, uap->offset);
313bcd9e0ddSJohn Baldwin 	free(auio, M_IOV);
314bcd9e0ddSJohn Baldwin 	return (error);
315bcd9e0ddSJohn Baldwin }
316bcd9e0ddSJohn Baldwin 
317bcd9e0ddSJohn Baldwin int
318cc3c9df8SEd Maste kern_preadv(struct thread *td, int fd, struct uio *auio, off_t offset)
319bcd9e0ddSJohn Baldwin {
320bcd9e0ddSJohn Baldwin 	struct file *fp;
321bcd9e0ddSJohn Baldwin 	int error;
322bcd9e0ddSJohn Baldwin 
323cbd92ce6SMatt Macy 	error = fget_read(td, fd, &cap_pread_rights, &fp);
324bcd9e0ddSJohn Baldwin 	if (error)
325bcd9e0ddSJohn Baldwin 		return (error);
326bcd9e0ddSJohn Baldwin 	if (!(fp->f_ops->fo_flags & DFLAG_SEEKABLE))
327bcd9e0ddSJohn Baldwin 		error = ESPIPE;
32832a1fb0dSMahdi Mokhtari 	else if (offset < 0 &&
32932a1fb0dSMahdi Mokhtari 	    (fp->f_vnode == NULL || fp->f_vnode->v_type != VCHR))
330bcd9e0ddSJohn Baldwin 		error = EINVAL;
331bcd9e0ddSJohn Baldwin 	else
332bcd9e0ddSJohn Baldwin 		error = dofileread(td, fd, fp, auio, offset, FOF_OFFSET);
333bcd9e0ddSJohn Baldwin 	fdrop(fp, td);
334bcd9e0ddSJohn Baldwin 	return (error);
335bcd9e0ddSJohn Baldwin }
336bcd9e0ddSJohn Baldwin 
337bcd9e0ddSJohn Baldwin /*
338bcd9e0ddSJohn Baldwin  * Common code for readv and preadv that reads data in
339bcd9e0ddSJohn Baldwin  * from a file using the passed in uio, offset, and flags.
340bcd9e0ddSJohn Baldwin  */
341bcd9e0ddSJohn Baldwin static int
342cc3c9df8SEd Maste dofileread(struct thread *td, int fd, struct file *fp, struct uio *auio,
343cc3c9df8SEd Maste     off_t offset, int flags)
344bcd9e0ddSJohn Baldwin {
345bcd9e0ddSJohn Baldwin 	ssize_t cnt;
34682641acdSAlan Cox 	int error;
347df8bae1dSRodney W. Grimes #ifdef KTRACE
348552afd9cSPoul-Henning Kamp 	struct uio *ktruio = NULL;
349df8bae1dSRodney W. Grimes #endif
350df8bae1dSRodney W. Grimes 
35151d1f690SRobert Watson 	AUDIT_ARG_FD(fd);
35251d1f690SRobert Watson 
3534f8d23d6SPoul-Henning Kamp 	/* Finish zero length reads right here */
3544f8d23d6SPoul-Henning Kamp 	if (auio->uio_resid == 0) {
3554f8d23d6SPoul-Henning Kamp 		td->td_retval[0] = 0;
3564f8d23d6SPoul-Henning Kamp 		return (0);
3574f8d23d6SPoul-Henning Kamp 	}
358552afd9cSPoul-Henning Kamp 	auio->uio_rw = UIO_READ;
359bcd9e0ddSJohn Baldwin 	auio->uio_offset = offset;
360552afd9cSPoul-Henning Kamp 	auio->uio_td = td;
361df8bae1dSRodney W. Grimes #ifdef KTRACE
362552afd9cSPoul-Henning Kamp 	if (KTRPOINT(td, KTR_GENIO))
363552afd9cSPoul-Henning Kamp 		ktruio = cloneuio(auio);
364df8bae1dSRodney W. Grimes #endif
365552afd9cSPoul-Henning Kamp 	cnt = auio->uio_resid;
366bcd9e0ddSJohn Baldwin 	if ((error = fo_read(fp, auio, td->td_ucred, flags, td))) {
367552afd9cSPoul-Henning Kamp 		if (auio->uio_resid != cnt && (error == ERESTART ||
368df8bae1dSRodney W. Grimes 		    error == EINTR || error == EWOULDBLOCK))
369df8bae1dSRodney W. Grimes 			error = 0;
370279d7226SMatthew Dillon 	}
371552afd9cSPoul-Henning Kamp 	cnt -= auio->uio_resid;
372df8bae1dSRodney W. Grimes #ifdef KTRACE
373552afd9cSPoul-Henning Kamp 	if (ktruio != NULL) {
374552afd9cSPoul-Henning Kamp 		ktruio->uio_resid = cnt;
375b88ec951SJohn Baldwin 		ktrgenio(fd, UIO_READ, ktruio, error);
376df8bae1dSRodney W. Grimes 	}
377df8bae1dSRodney W. Grimes #endif
378b40ce416SJulian Elischer 	td->td_retval[0] = cnt;
379df8bae1dSRodney W. Grimes 	return (error);
380df8bae1dSRodney W. Grimes }
381df8bae1dSRodney W. Grimes 
382d2d3e875SBruce Evans #ifndef _SYS_SYSPROTO_H_
383df8bae1dSRodney W. Grimes struct write_args {
384df8bae1dSRodney W. Grimes 	int	fd;
385134e06feSBruce Evans 	const void *buf;
386134e06feSBruce Evans 	size_t	nbyte;
387df8bae1dSRodney W. Grimes };
388d2d3e875SBruce Evans #endif
38926f9a767SRodney W. Grimes int
390cc3c9df8SEd Maste sys_write(struct thread *td, struct write_args *uap)
391df8bae1dSRodney W. Grimes {
392bcd9e0ddSJohn Baldwin 	struct uio auio;
393bcd9e0ddSJohn Baldwin 	struct iovec aiov;
394279d7226SMatthew Dillon 	int error;
395df8bae1dSRodney W. Grimes 
396526d0bd5SKonstantin Belousov 	if (uap->nbyte > IOSIZE_MAX)
397bcd9e0ddSJohn Baldwin 		return (EINVAL);
398bcd9e0ddSJohn Baldwin 	aiov.iov_base = (void *)(uintptr_t)uap->buf;
399bcd9e0ddSJohn Baldwin 	aiov.iov_len = uap->nbyte;
400bcd9e0ddSJohn Baldwin 	auio.uio_iov = &aiov;
401bcd9e0ddSJohn Baldwin 	auio.uio_iovcnt = 1;
402bcd9e0ddSJohn Baldwin 	auio.uio_resid = uap->nbyte;
403bcd9e0ddSJohn Baldwin 	auio.uio_segflg = UIO_USERSPACE;
404bcd9e0ddSJohn Baldwin 	error = kern_writev(td, uap->fd, &auio);
405279d7226SMatthew Dillon 	return (error);
406df8bae1dSRodney W. Grimes }
407df8bae1dSRodney W. Grimes 
408df8bae1dSRodney W. Grimes /*
4090c14ff0eSRobert Watson  * Positioned write system call.
4104160ccd9SAlan Cox  */
4114160ccd9SAlan Cox #ifndef _SYS_SYSPROTO_H_
4124160ccd9SAlan Cox struct pwrite_args {
4134160ccd9SAlan Cox 	int	fd;
4144160ccd9SAlan Cox 	const void *buf;
4154160ccd9SAlan Cox 	size_t	nbyte;
4168fe387abSDmitrij Tejblum 	int	pad;
4174160ccd9SAlan Cox 	off_t	offset;
4184160ccd9SAlan Cox };
4194160ccd9SAlan Cox #endif
4204160ccd9SAlan Cox int
421b38b22b0SEdward Tomasz Napierala sys_pwrite(struct thread *td, struct pwrite_args *uap)
422b38b22b0SEdward Tomasz Napierala {
423b38b22b0SEdward Tomasz Napierala 
424b38b22b0SEdward Tomasz Napierala 	return (kern_pwrite(td, uap->fd, uap->buf, uap->nbyte, uap->offset));
425b38b22b0SEdward Tomasz Napierala }
426b38b22b0SEdward Tomasz Napierala 
427b38b22b0SEdward Tomasz Napierala int
428b38b22b0SEdward Tomasz Napierala kern_pwrite(struct thread *td, int fd, const void *buf, size_t nbyte,
429b38b22b0SEdward Tomasz Napierala     off_t offset)
4304160ccd9SAlan Cox {
4314160ccd9SAlan Cox 	struct uio auio;
4324160ccd9SAlan Cox 	struct iovec aiov;
433bcd9e0ddSJohn Baldwin 	int error;
4344160ccd9SAlan Cox 
435b38b22b0SEdward Tomasz Napierala 	if (nbyte > IOSIZE_MAX)
436bcd9e0ddSJohn Baldwin 		return (EINVAL);
437b38b22b0SEdward Tomasz Napierala 	aiov.iov_base = (void *)(uintptr_t)buf;
438b38b22b0SEdward Tomasz Napierala 	aiov.iov_len = nbyte;
4394160ccd9SAlan Cox 	auio.uio_iov = &aiov;
4404160ccd9SAlan Cox 	auio.uio_iovcnt = 1;
441b38b22b0SEdward Tomasz Napierala 	auio.uio_resid = nbyte;
4424160ccd9SAlan Cox 	auio.uio_segflg = UIO_USERSPACE;
443b38b22b0SEdward Tomasz Napierala 	error = kern_pwritev(td, fd, &auio, offset);
4444160ccd9SAlan Cox 	return (error);
4454160ccd9SAlan Cox }
4464160ccd9SAlan Cox 
4470538aafcSKonstantin Belousov #if defined(COMPAT_FREEBSD6)
448c2815ad5SPeter Wemm int
449b38b22b0SEdward Tomasz Napierala freebsd6_pwrite(struct thread *td, struct freebsd6_pwrite_args *uap)
450c2815ad5SPeter Wemm {
451c2815ad5SPeter Wemm 
452b38b22b0SEdward Tomasz Napierala 	return (kern_pwrite(td, uap->fd, uap->buf, uap->nbyte, uap->offset));
453c2815ad5SPeter Wemm }
4540538aafcSKonstantin Belousov #endif
455c2815ad5SPeter Wemm 
4564160ccd9SAlan Cox /*
4570c14ff0eSRobert Watson  * Gather write system call.
458df8bae1dSRodney W. Grimes  */
459d2d3e875SBruce Evans #ifndef _SYS_SYSPROTO_H_
460df8bae1dSRodney W. Grimes struct writev_args {
461df8bae1dSRodney W. Grimes 	int	fd;
462df8bae1dSRodney W. Grimes 	struct	iovec *iovp;
463df8bae1dSRodney W. Grimes 	u_int	iovcnt;
464df8bae1dSRodney W. Grimes };
465d2d3e875SBruce Evans #endif
46626f9a767SRodney W. Grimes int
4678451d0ddSKip Macy sys_writev(struct thread *td, struct writev_args *uap)
468df8bae1dSRodney W. Grimes {
469b88ec951SJohn Baldwin 	struct uio *auio;
470b88ec951SJohn Baldwin 	int error;
471b88ec951SJohn Baldwin 
472b88ec951SJohn Baldwin 	error = copyinuio(uap->iovp, uap->iovcnt, &auio);
473b88ec951SJohn Baldwin 	if (error)
474b88ec951SJohn Baldwin 		return (error);
475b88ec951SJohn Baldwin 	error = kern_writev(td, uap->fd, auio);
476b88ec951SJohn Baldwin 	free(auio, M_IOV);
477b88ec951SJohn Baldwin 	return (error);
478b88ec951SJohn Baldwin }
479b88ec951SJohn Baldwin 
480b88ec951SJohn Baldwin int
481b88ec951SJohn Baldwin kern_writev(struct thread *td, int fd, struct uio *auio)
482b88ec951SJohn Baldwin {
483b064d43dSMatthew Dillon 	struct file *fp;
484bcd9e0ddSJohn Baldwin 	int error;
485bcd9e0ddSJohn Baldwin 
486cbd92ce6SMatt Macy 	error = fget_write(td, fd, &cap_write_rights, &fp);
487bcd9e0ddSJohn Baldwin 	if (error)
488af56abaaSJohn Baldwin 		return (error);
489bcd9e0ddSJohn Baldwin 	error = dofilewrite(td, fd, fp, auio, (off_t)-1, 0);
490bcd9e0ddSJohn Baldwin 	fdrop(fp, td);
491bcd9e0ddSJohn Baldwin 	return (error);
492bcd9e0ddSJohn Baldwin }
493bcd9e0ddSJohn Baldwin 
494bcd9e0ddSJohn Baldwin /*
4950c14ff0eSRobert Watson  * Gather positioned write system call.
496bcd9e0ddSJohn Baldwin  */
497bcd9e0ddSJohn Baldwin #ifndef _SYS_SYSPROTO_H_
498bcd9e0ddSJohn Baldwin struct pwritev_args {
499bcd9e0ddSJohn Baldwin 	int	fd;
500bcd9e0ddSJohn Baldwin 	struct	iovec *iovp;
501bcd9e0ddSJohn Baldwin 	u_int	iovcnt;
502bcd9e0ddSJohn Baldwin 	off_t	offset;
503bcd9e0ddSJohn Baldwin };
504bcd9e0ddSJohn Baldwin #endif
505bcd9e0ddSJohn Baldwin int
5068451d0ddSKip Macy sys_pwritev(struct thread *td, struct pwritev_args *uap)
507bcd9e0ddSJohn Baldwin {
508bcd9e0ddSJohn Baldwin 	struct uio *auio;
509bcd9e0ddSJohn Baldwin 	int error;
510bcd9e0ddSJohn Baldwin 
511bcd9e0ddSJohn Baldwin 	error = copyinuio(uap->iovp, uap->iovcnt, &auio);
512bcd9e0ddSJohn Baldwin 	if (error)
513bcd9e0ddSJohn Baldwin 		return (error);
514bcd9e0ddSJohn Baldwin 	error = kern_pwritev(td, uap->fd, auio, uap->offset);
515bcd9e0ddSJohn Baldwin 	free(auio, M_IOV);
516bcd9e0ddSJohn Baldwin 	return (error);
517bcd9e0ddSJohn Baldwin }
518bcd9e0ddSJohn Baldwin 
519bcd9e0ddSJohn Baldwin int
520d5cdcc3aSAndrew Gallatin kern_pwritev(struct thread *td, int fd, struct uio *auio, off_t offset)
521bcd9e0ddSJohn Baldwin {
522bcd9e0ddSJohn Baldwin 	struct file *fp;
523bcd9e0ddSJohn Baldwin 	int error;
524bcd9e0ddSJohn Baldwin 
525cbd92ce6SMatt Macy 	error = fget_write(td, fd, &cap_pwrite_rights, &fp);
526bcd9e0ddSJohn Baldwin 	if (error)
527af56abaaSJohn Baldwin 		return (error);
528bcd9e0ddSJohn Baldwin 	if (!(fp->f_ops->fo_flags & DFLAG_SEEKABLE))
529bcd9e0ddSJohn Baldwin 		error = ESPIPE;
53032a1fb0dSMahdi Mokhtari 	else if (offset < 0 &&
53132a1fb0dSMahdi Mokhtari 	    (fp->f_vnode == NULL || fp->f_vnode->v_type != VCHR))
532bcd9e0ddSJohn Baldwin 		error = EINVAL;
533bcd9e0ddSJohn Baldwin 	else
534bcd9e0ddSJohn Baldwin 		error = dofilewrite(td, fd, fp, auio, offset, FOF_OFFSET);
535bcd9e0ddSJohn Baldwin 	fdrop(fp, td);
536bcd9e0ddSJohn Baldwin 	return (error);
537bcd9e0ddSJohn Baldwin }
538bcd9e0ddSJohn Baldwin 
539bcd9e0ddSJohn Baldwin /*
540bcd9e0ddSJohn Baldwin  * Common code for writev and pwritev that writes data to
541bcd9e0ddSJohn Baldwin  * a file using the passed in uio, offset, and flags.
542bcd9e0ddSJohn Baldwin  */
543bcd9e0ddSJohn Baldwin static int
544cc3c9df8SEd Maste dofilewrite(struct thread *td, int fd, struct file *fp, struct uio *auio,
545cc3c9df8SEd Maste     off_t offset, int flags)
546bcd9e0ddSJohn Baldwin {
547bcd9e0ddSJohn Baldwin 	ssize_t cnt;
548552afd9cSPoul-Henning Kamp 	int error;
549df8bae1dSRodney W. Grimes #ifdef KTRACE
550552afd9cSPoul-Henning Kamp 	struct uio *ktruio = NULL;
551df8bae1dSRodney W. Grimes #endif
552df8bae1dSRodney W. Grimes 
55351d1f690SRobert Watson 	AUDIT_ARG_FD(fd);
554552afd9cSPoul-Henning Kamp 	auio->uio_rw = UIO_WRITE;
555552afd9cSPoul-Henning Kamp 	auio->uio_td = td;
556bcd9e0ddSJohn Baldwin 	auio->uio_offset = offset;
557df8bae1dSRodney W. Grimes #ifdef KTRACE
558552afd9cSPoul-Henning Kamp 	if (KTRPOINT(td, KTR_GENIO))
559552afd9cSPoul-Henning Kamp 		ktruio = cloneuio(auio);
560df8bae1dSRodney W. Grimes #endif
561552afd9cSPoul-Henning Kamp 	cnt = auio->uio_resid;
5627a2c93b8SGleb Smirnoff 	error = fo_write(fp, auio, td->td_ucred, flags, td);
5637a2c93b8SGleb Smirnoff 	/*
5647a2c93b8SGleb Smirnoff 	 * Socket layer is responsible for special error handling,
5657a2c93b8SGleb Smirnoff 	 * see sousrsend().
5667a2c93b8SGleb Smirnoff 	 */
5677a2c93b8SGleb Smirnoff 	if (error != 0 && fp->f_type != DTYPE_SOCKET) {
568552afd9cSPoul-Henning Kamp 		if (auio->uio_resid != cnt && (error == ERESTART ||
569df8bae1dSRodney W. Grimes 		    error == EINTR || error == EWOULDBLOCK))
570df8bae1dSRodney W. Grimes 			error = 0;
5717a2c93b8SGleb Smirnoff 		if (error == EPIPE) {
572b40ce416SJulian Elischer 			PROC_LOCK(td->td_proc);
5737a6f3d78SJohn Baldwin 			tdsignal(td, SIGPIPE);
574b40ce416SJulian Elischer 			PROC_UNLOCK(td->td_proc);
57519eb87d2SJohn Baldwin 		}
576df8bae1dSRodney W. Grimes 	}
577552afd9cSPoul-Henning Kamp 	cnt -= auio->uio_resid;
578df8bae1dSRodney W. Grimes #ifdef KTRACE
579552afd9cSPoul-Henning Kamp 	if (ktruio != NULL) {
580552afd9cSPoul-Henning Kamp 		ktruio->uio_resid = cnt;
581b88ec951SJohn Baldwin 		ktrgenio(fd, UIO_WRITE, ktruio, error);
582df8bae1dSRodney W. Grimes 	}
583df8bae1dSRodney W. Grimes #endif
584b40ce416SJulian Elischer 	td->td_retval[0] = cnt;
585df8bae1dSRodney W. Grimes 	return (error);
586df8bae1dSRodney W. Grimes }
587df8bae1dSRodney W. Grimes 
588e4650294SJohn Baldwin /*
589e4650294SJohn Baldwin  * Truncate a file given a file descriptor.
590e4650294SJohn Baldwin  *
591e4650294SJohn Baldwin  * Can't use fget_write() here, since must return EINVAL and not EBADF if the
592e4650294SJohn Baldwin  * descriptor isn't writable.
593e4650294SJohn Baldwin  */
594e4650294SJohn Baldwin int
595cc3c9df8SEd Maste kern_ftruncate(struct thread *td, int fd, off_t length)
596e4650294SJohn Baldwin {
597e4650294SJohn Baldwin 	struct file *fp;
598e4650294SJohn Baldwin 	int error;
599e4650294SJohn Baldwin 
60014961ba7SRobert Watson 	AUDIT_ARG_FD(fd);
601e4650294SJohn Baldwin 	if (length < 0)
602e4650294SJohn Baldwin 		return (EINVAL);
603cbd92ce6SMatt Macy 	error = fget(td, fd, &cap_ftruncate_rights, &fp);
604e4650294SJohn Baldwin 	if (error)
605e4650294SJohn Baldwin 		return (error);
60614961ba7SRobert Watson 	AUDIT_ARG_FILE(td->td_proc, fp);
607e4650294SJohn Baldwin 	if (!(fp->f_flag & FWRITE)) {
608e4650294SJohn Baldwin 		fdrop(fp, td);
609e4650294SJohn Baldwin 		return (EINVAL);
610e4650294SJohn Baldwin 	}
611e4650294SJohn Baldwin 	error = fo_truncate(fp, length, td->td_ucred, td);
612e4650294SJohn Baldwin 	fdrop(fp, td);
613e4650294SJohn Baldwin 	return (error);
614e4650294SJohn Baldwin }
615e4650294SJohn Baldwin 
616e4650294SJohn Baldwin #ifndef _SYS_SYSPROTO_H_
617e4650294SJohn Baldwin struct ftruncate_args {
618e4650294SJohn Baldwin 	int	fd;
619e4650294SJohn Baldwin 	int	pad;
620e4650294SJohn Baldwin 	off_t	length;
621e4650294SJohn Baldwin };
622e4650294SJohn Baldwin #endif
623e4650294SJohn Baldwin int
624cc3c9df8SEd Maste sys_ftruncate(struct thread *td, struct ftruncate_args *uap)
625e4650294SJohn Baldwin {
626e4650294SJohn Baldwin 
627e4650294SJohn Baldwin 	return (kern_ftruncate(td, uap->fd, uap->length));
628e4650294SJohn Baldwin }
629e4650294SJohn Baldwin 
630e4650294SJohn Baldwin #if defined(COMPAT_43)
631e4650294SJohn Baldwin #ifndef _SYS_SYSPROTO_H_
632e4650294SJohn Baldwin struct oftruncate_args {
633e4650294SJohn Baldwin 	int	fd;
634e4650294SJohn Baldwin 	long	length;
635e4650294SJohn Baldwin };
636e4650294SJohn Baldwin #endif
637e4650294SJohn Baldwin int
638cc3c9df8SEd Maste oftruncate(struct thread *td, struct oftruncate_args *uap)
639e4650294SJohn Baldwin {
640e4650294SJohn Baldwin 
641e4650294SJohn Baldwin 	return (kern_ftruncate(td, uap->fd, uap->length));
642e4650294SJohn Baldwin }
643e4650294SJohn Baldwin #endif /* COMPAT_43 */
644e4650294SJohn Baldwin 
645d2d3e875SBruce Evans #ifndef _SYS_SYSPROTO_H_
646df8bae1dSRodney W. Grimes struct ioctl_args {
647df8bae1dSRodney W. Grimes 	int	fd;
648069e9bc1SDoug Rabson 	u_long	com;
649df8bae1dSRodney W. Grimes 	caddr_t	data;
650df8bae1dSRodney W. Grimes };
651d2d3e875SBruce Evans #endif
652df8bae1dSRodney W. Grimes /* ARGSUSED */
65326f9a767SRodney W. Grimes int
6548451d0ddSKip Macy sys_ioctl(struct thread *td, struct ioctl_args *uap)
655df8bae1dSRodney W. Grimes {
65650ae6690SHans Petter Selasky 	u_char smalldata[SYS_IOCTL_SMALL_SIZE] __aligned(SYS_IOCTL_SMALL_ALIGN);
657a90fb6cfSHans Petter Selasky 	uint32_t com;
6589fddcc66SRuslan Ermilov 	int arg, error;
6593e15c66fSPoul-Henning Kamp 	u_int size;
6609fddcc66SRuslan Ermilov 	caddr_t data;
661df8bae1dSRodney W. Grimes 
662a90fb6cfSHans Petter Selasky #ifdef INVARIANTS
6639fc6aa06SPoul-Henning Kamp 	if (uap->com > 0xffffffff) {
6649fc6aa06SPoul-Henning Kamp 		printf(
6659fc6aa06SPoul-Henning Kamp 		    "WARNING pid %d (%s): ioctl sign-extension ioctl %lx\n",
666431f8906SJulian Elischer 		    td->td_proc->p_pid, td->td_name, uap->com);
6679fc6aa06SPoul-Henning Kamp 	}
668a90fb6cfSHans Petter Selasky #endif
669a90fb6cfSHans Petter Selasky 	com = (uint32_t)uap->com;
670df8bae1dSRodney W. Grimes 
671df8bae1dSRodney W. Grimes 	/*
672df8bae1dSRodney W. Grimes 	 * Interpret high order word to find amount of data to be
673df8bae1dSRodney W. Grimes 	 * copied to/from the user's address space.
674df8bae1dSRodney W. Grimes 	 */
675df8bae1dSRodney W. Grimes 	size = IOCPARM_LEN(com);
676ca51b19bSPoul-Henning Kamp 	if ((size > IOCPARM_MAX) ||
677ca51b19bSPoul-Henning Kamp 	    ((com & (IOC_VOID  | IOC_IN | IOC_OUT)) == 0) ||
6782de92a38SPeter Wemm #if defined(COMPAT_FREEBSD5) || defined(COMPAT_FREEBSD4) || defined(COMPAT_43)
6792de92a38SPeter Wemm 	    ((com & IOC_OUT) && size == 0) ||
6802de92a38SPeter Wemm #else
6812de92a38SPeter Wemm 	    ((com & (IOC_IN | IOC_OUT)) && size == 0) ||
6822de92a38SPeter Wemm #endif
6839fddcc66SRuslan Ermilov 	    ((com & IOC_VOID) && size > 0 && size != sizeof(int)))
684426da3bcSAlfred Perlstein 		return (ENOTTY);
685279d7226SMatthew Dillon 
686ca51b19bSPoul-Henning Kamp 	if (size > 0) {
687715457f6SDavid E. O'Brien 		if (com & IOC_VOID) {
6889fddcc66SRuslan Ermilov 			/* Integer argument. */
6899fddcc66SRuslan Ermilov 			arg = (intptr_t)uap->data;
6909fddcc66SRuslan Ermilov 			data = (void *)&arg;
6919fddcc66SRuslan Ermilov 			size = 0;
6925cbf44bfSMateusz Guzik 		} else {
6930ecd606bSHans Petter Selasky 			if (size > SYS_IOCTL_SMALL_SIZE)
694715457f6SDavid E. O'Brien 				data = malloc((u_long)size, M_IOCTLOPS, M_WAITOK);
6950ecd606bSHans Petter Selasky 			else
6960ecd606bSHans Petter Selasky 				data = smalldata;
6975cbf44bfSMateusz Guzik 		}
6989fddcc66SRuslan Ermilov 	} else
6999fddcc66SRuslan Ermilov 		data = (void *)&uap->data;
700df8bae1dSRodney W. Grimes 	if (com & IOC_IN) {
701df8bae1dSRodney W. Grimes 		error = copyin(uap->data, data, (u_int)size);
7025cbf44bfSMateusz Guzik 		if (error != 0)
7035cbf44bfSMateusz Guzik 			goto out;
704ca51b19bSPoul-Henning Kamp 	} else if (com & IOC_OUT) {
705df8bae1dSRodney W. Grimes 		/*
706df8bae1dSRodney W. Grimes 		 * Zero the buffer so the user always
707df8bae1dSRodney W. Grimes 		 * gets back something deterministic.
708df8bae1dSRodney W. Grimes 		 */
709df8bae1dSRodney W. Grimes 		bzero(data, size);
710279d7226SMatthew Dillon 	}
711df8bae1dSRodney W. Grimes 
712d9f46233SJohn Baldwin 	error = kern_ioctl(td, uap->fd, com, data);
713d9f46233SJohn Baldwin 
714d9f46233SJohn Baldwin 	if (error == 0 && (com & IOC_OUT))
715d9f46233SJohn Baldwin 		error = copyout(data, uap->data, (u_int)size);
716d9f46233SJohn Baldwin 
7175cbf44bfSMateusz Guzik out:
7180ecd606bSHans Petter Selasky 	if (size > SYS_IOCTL_SMALL_SIZE)
7199fddcc66SRuslan Ermilov 		free(data, M_IOCTLOPS);
720d9f46233SJohn Baldwin 	return (error);
721d9f46233SJohn Baldwin }
722d9f46233SJohn Baldwin 
723d9f46233SJohn Baldwin int
724d9f46233SJohn Baldwin kern_ioctl(struct thread *td, int fd, u_long com, caddr_t data)
725d9f46233SJohn Baldwin {
726d9f46233SJohn Baldwin 	struct file *fp;
727d9f46233SJohn Baldwin 	struct filedesc *fdp;
7282609222aSPawel Jakub Dawidek 	int error, tmp, locked;
729d9f46233SJohn Baldwin 
73064c9a4d9SRobert Watson 	AUDIT_ARG_FD(fd);
73164c9a4d9SRobert Watson 	AUDIT_ARG_CMD(com);
7322609222aSPawel Jakub Dawidek 
733d9f46233SJohn Baldwin 	fdp = td->td_proc->p_fd;
7342609222aSPawel Jakub Dawidek 
735d9f46233SJohn Baldwin 	switch (com) {
736d9f46233SJohn Baldwin 	case FIONCLEX:
737d9f46233SJohn Baldwin 	case FIOCLEX:
7385e3f7694SRobert Watson 		FILEDESC_XLOCK(fdp);
7392609222aSPawel Jakub Dawidek 		locked = LA_XLOCKED;
7402609222aSPawel Jakub Dawidek 		break;
7412609222aSPawel Jakub Dawidek 	default:
7422609222aSPawel Jakub Dawidek #ifdef CAPABILITIES
7432609222aSPawel Jakub Dawidek 		FILEDESC_SLOCK(fdp);
7442609222aSPawel Jakub Dawidek 		locked = LA_SLOCKED;
7452609222aSPawel Jakub Dawidek #else
7462609222aSPawel Jakub Dawidek 		locked = LA_UNLOCKED;
7472609222aSPawel Jakub Dawidek #endif
7482609222aSPawel Jakub Dawidek 		break;
7492609222aSPawel Jakub Dawidek 	}
7502609222aSPawel Jakub Dawidek 
7512609222aSPawel Jakub Dawidek #ifdef CAPABILITIES
752f17ef286SMateusz Guzik 	if ((fp = fget_noref(fdp, fd)) == NULL) {
7532609222aSPawel Jakub Dawidek 		error = EBADF;
7542609222aSPawel Jakub Dawidek 		goto out;
7552609222aSPawel Jakub Dawidek 	}
7562609222aSPawel Jakub Dawidek 	if ((error = cap_ioctl_check(fdp, fd, com)) != 0) {
7572609222aSPawel Jakub Dawidek 		fp = NULL;	/* fhold() was not called yet */
7582609222aSPawel Jakub Dawidek 		goto out;
7592609222aSPawel Jakub Dawidek 	}
760f1cf2b9dSKonstantin Belousov 	if (!fhold(fp)) {
761f1cf2b9dSKonstantin Belousov 		error = EBADF;
762f1cf2b9dSKonstantin Belousov 		fp = NULL;
763f1cf2b9dSKonstantin Belousov 		goto out;
764f1cf2b9dSKonstantin Belousov 	}
7652609222aSPawel Jakub Dawidek 	if (locked == LA_SLOCKED) {
7662609222aSPawel Jakub Dawidek 		FILEDESC_SUNLOCK(fdp);
7672609222aSPawel Jakub Dawidek 		locked = LA_UNLOCKED;
7682609222aSPawel Jakub Dawidek 	}
7692609222aSPawel Jakub Dawidek #else
770cbd92ce6SMatt Macy 	error = fget(td, fd, &cap_ioctl_rights, &fp);
7717008be5bSPawel Jakub Dawidek 	if (error != 0) {
7722609222aSPawel Jakub Dawidek 		fp = NULL;
7732609222aSPawel Jakub Dawidek 		goto out;
7742609222aSPawel Jakub Dawidek 	}
7752609222aSPawel Jakub Dawidek #endif
7762609222aSPawel Jakub Dawidek 	if ((fp->f_flag & (FREAD | FWRITE)) == 0) {
7772609222aSPawel Jakub Dawidek 		error = EBADF;
7782609222aSPawel Jakub Dawidek 		goto out;
7792609222aSPawel Jakub Dawidek 	}
7802609222aSPawel Jakub Dawidek 
7812609222aSPawel Jakub Dawidek 	switch (com) {
7822609222aSPawel Jakub Dawidek 	case FIONCLEX:
7832609222aSPawel Jakub Dawidek 		fdp->fd_ofiles[fd].fde_flags &= ~UF_EXCLOSE;
7842609222aSPawel Jakub Dawidek 		goto out;
7852609222aSPawel Jakub Dawidek 	case FIOCLEX:
7862609222aSPawel Jakub Dawidek 		fdp->fd_ofiles[fd].fde_flags |= UF_EXCLOSE;
787d9f46233SJohn Baldwin 		goto out;
788d9f46233SJohn Baldwin 	case FIONBIO:
789bb56ec4aSPoul-Henning Kamp 		if ((tmp = *(int *)data))
790397c19d1SJeff Roberson 			atomic_set_int(&fp->f_flag, FNONBLOCK);
791df8bae1dSRodney W. Grimes 		else
792397c19d1SJeff Roberson 			atomic_clear_int(&fp->f_flag, FNONBLOCK);
7938ccf264fSPoul-Henning Kamp 		data = (void *)&tmp;
794d9f46233SJohn Baldwin 		break;
795d9f46233SJohn Baldwin 	case FIOASYNC:
796bb56ec4aSPoul-Henning Kamp 		if ((tmp = *(int *)data))
797397c19d1SJeff Roberson 			atomic_set_int(&fp->f_flag, FASYNC);
798df8bae1dSRodney W. Grimes 		else
799397c19d1SJeff Roberson 			atomic_clear_int(&fp->f_flag, FASYNC);
8008ccf264fSPoul-Henning Kamp 		data = (void *)&tmp;
801d9f46233SJohn Baldwin 		break;
802df8bae1dSRodney W. Grimes 	}
8038ccf264fSPoul-Henning Kamp 
8048ccf264fSPoul-Henning Kamp 	error = fo_ioctl(fp, com, data, td->td_ucred, td);
805d9f46233SJohn Baldwin out:
8062609222aSPawel Jakub Dawidek 	switch (locked) {
8072609222aSPawel Jakub Dawidek 	case LA_XLOCKED:
8082609222aSPawel Jakub Dawidek 		FILEDESC_XUNLOCK(fdp);
8092609222aSPawel Jakub Dawidek 		break;
8102609222aSPawel Jakub Dawidek #ifdef CAPABILITIES
8112609222aSPawel Jakub Dawidek 	case LA_SLOCKED:
8122609222aSPawel Jakub Dawidek 		FILEDESC_SUNLOCK(fdp);
8132609222aSPawel Jakub Dawidek 		break;
8142609222aSPawel Jakub Dawidek #endif
8152609222aSPawel Jakub Dawidek 	default:
8162609222aSPawel Jakub Dawidek 		FILEDESC_UNLOCK_ASSERT(fdp);
8172609222aSPawel Jakub Dawidek 		break;
8182609222aSPawel Jakub Dawidek 	}
8192609222aSPawel Jakub Dawidek 	if (fp != NULL)
820b40ce416SJulian Elischer 		fdrop(fp, td);
821df8bae1dSRodney W. Grimes 	return (error);
822df8bae1dSRodney W. Grimes }
823df8bae1dSRodney W. Grimes 
824125dcf8cSKonstantin Belousov int
8252856d85eSKyle Evans sys_posix_fallocate(struct thread *td, struct posix_fallocate_args *uap)
8262856d85eSKyle Evans {
8272856d85eSKyle Evans 	int error;
8282856d85eSKyle Evans 
8292856d85eSKyle Evans 	error = kern_posix_fallocate(td, uap->fd, uap->offset, uap->len);
8302856d85eSKyle Evans 	return (kern_posix_error(td, error));
8312856d85eSKyle Evans }
8322856d85eSKyle Evans 
8332856d85eSKyle Evans int
8342856d85eSKyle Evans kern_posix_fallocate(struct thread *td, int fd, off_t offset, off_t len)
8352856d85eSKyle Evans {
8362856d85eSKyle Evans 	struct file *fp;
8372856d85eSKyle Evans 	int error;
8382856d85eSKyle Evans 
8392856d85eSKyle Evans 	AUDIT_ARG_FD(fd);
8402856d85eSKyle Evans 	if (offset < 0 || len <= 0)
8412856d85eSKyle Evans 		return (EINVAL);
8422856d85eSKyle Evans 	/* Check for wrap. */
8432856d85eSKyle Evans 	if (offset > OFF_MAX - len)
8442856d85eSKyle Evans 		return (EFBIG);
8452856d85eSKyle Evans 	AUDIT_ARG_FD(fd);
8462856d85eSKyle Evans 	error = fget(td, fd, &cap_pwrite_rights, &fp);
8472856d85eSKyle Evans 	if (error != 0)
8482856d85eSKyle Evans 		return (error);
8492856d85eSKyle Evans 	AUDIT_ARG_FILE(td->td_proc, fp);
8502856d85eSKyle Evans 	if ((fp->f_ops->fo_flags & DFLAG_SEEKABLE) == 0) {
8512856d85eSKyle Evans 		error = ESPIPE;
8522856d85eSKyle Evans 		goto out;
8532856d85eSKyle Evans 	}
8542856d85eSKyle Evans 	if ((fp->f_flag & FWRITE) == 0) {
8552856d85eSKyle Evans 		error = EBADF;
8562856d85eSKyle Evans 		goto out;
8572856d85eSKyle Evans 	}
8582856d85eSKyle Evans 
8592856d85eSKyle Evans 	error = fo_fallocate(fp, offset, len, td);
8602856d85eSKyle Evans  out:
8612856d85eSKyle Evans 	fdrop(fp, td);
8622856d85eSKyle Evans 	return (error);
8632856d85eSKyle Evans }
8642856d85eSKyle Evans 
8652856d85eSKyle Evans int
8660dc332bfSKa Ho Ng sys_fspacectl(struct thread *td, struct fspacectl_args *uap)
8670dc332bfSKa Ho Ng {
8680dc332bfSKa Ho Ng 	struct spacectl_range rqsr, rmsr;
8690dc332bfSKa Ho Ng 	int error, cerror;
8700dc332bfSKa Ho Ng 
8710dc332bfSKa Ho Ng 	error = copyin(uap->rqsr, &rqsr, sizeof(rqsr));
8720dc332bfSKa Ho Ng 	if (error != 0)
8730dc332bfSKa Ho Ng 		return (error);
8740dc332bfSKa Ho Ng 
8750dc332bfSKa Ho Ng 	error = kern_fspacectl(td, uap->fd, uap->cmd, &rqsr, uap->flags,
8760dc332bfSKa Ho Ng 	    &rmsr);
8770dc332bfSKa Ho Ng 	if (uap->rmsr != NULL) {
8780dc332bfSKa Ho Ng 		cerror = copyout(&rmsr, uap->rmsr, sizeof(rmsr));
8790dc332bfSKa Ho Ng 		if (error == 0)
8800dc332bfSKa Ho Ng 			error = cerror;
8810dc332bfSKa Ho Ng 	}
8820dc332bfSKa Ho Ng 	return (error);
8830dc332bfSKa Ho Ng }
8840dc332bfSKa Ho Ng 
8850dc332bfSKa Ho Ng int
8860dc332bfSKa Ho Ng kern_fspacectl(struct thread *td, int fd, int cmd,
8870dc332bfSKa Ho Ng     const struct spacectl_range *rqsr, int flags, struct spacectl_range *rmsrp)
8880dc332bfSKa Ho Ng {
8890dc332bfSKa Ho Ng 	struct file *fp;
8900dc332bfSKa Ho Ng 	struct spacectl_range rmsr;
8910dc332bfSKa Ho Ng 	int error;
8920dc332bfSKa Ho Ng 
8930dc332bfSKa Ho Ng 	AUDIT_ARG_FD(fd);
8940dc332bfSKa Ho Ng 	AUDIT_ARG_CMD(cmd);
8950dc332bfSKa Ho Ng 	AUDIT_ARG_FFLAGS(flags);
8960dc332bfSKa Ho Ng 
8970dc332bfSKa Ho Ng 	if (rqsr == NULL)
8980dc332bfSKa Ho Ng 		return (EINVAL);
8990dc332bfSKa Ho Ng 	rmsr = *rqsr;
9000dc332bfSKa Ho Ng 	if (rmsrp != NULL)
9010dc332bfSKa Ho Ng 		*rmsrp = rmsr;
9020dc332bfSKa Ho Ng 
9030dc332bfSKa Ho Ng 	if (cmd != SPACECTL_DEALLOC ||
9040dc332bfSKa Ho Ng 	    rqsr->r_offset < 0 || rqsr->r_len <= 0 ||
9050dc332bfSKa Ho Ng 	    rqsr->r_offset > OFF_MAX - rqsr->r_len ||
9060dc332bfSKa Ho Ng 	    (flags & ~SPACECTL_F_SUPPORTED) != 0)
9070dc332bfSKa Ho Ng 		return (EINVAL);
9080dc332bfSKa Ho Ng 
9090dc332bfSKa Ho Ng 	error = fget_write(td, fd, &cap_pwrite_rights, &fp);
9100dc332bfSKa Ho Ng 	if (error != 0)
9110dc332bfSKa Ho Ng 		return (error);
9120dc332bfSKa Ho Ng 	AUDIT_ARG_FILE(td->td_proc, fp);
9130dc332bfSKa Ho Ng 	if ((fp->f_ops->fo_flags & DFLAG_SEEKABLE) == 0) {
9140dc332bfSKa Ho Ng 		error = ESPIPE;
9150dc332bfSKa Ho Ng 		goto out;
9160dc332bfSKa Ho Ng 	}
9170dc332bfSKa Ho Ng 	if ((fp->f_flag & FWRITE) == 0) {
9180dc332bfSKa Ho Ng 		error = EBADF;
9190dc332bfSKa Ho Ng 		goto out;
9200dc332bfSKa Ho Ng 	}
9210dc332bfSKa Ho Ng 
9220dc332bfSKa Ho Ng 	error = fo_fspacectl(fp, cmd, &rmsr.r_offset, &rmsr.r_len, flags,
9230dc332bfSKa Ho Ng 	    td->td_ucred, td);
9240dc332bfSKa Ho Ng 	/* fspacectl is not restarted after signals if the file is modified. */
9250dc332bfSKa Ho Ng 	if (rmsr.r_len != rqsr->r_len && (error == ERESTART ||
9260dc332bfSKa Ho Ng 	    error == EINTR || error == EWOULDBLOCK))
9270dc332bfSKa Ho Ng 		error = 0;
9280dc332bfSKa Ho Ng 	if (rmsrp != NULL)
9290dc332bfSKa Ho Ng 		*rmsrp = rmsr;
9300dc332bfSKa Ho Ng out:
9310dc332bfSKa Ho Ng 	fdrop(fp, td);
9320dc332bfSKa Ho Ng 	return (error);
9330dc332bfSKa Ho Ng }
9340dc332bfSKa Ho Ng 
9350dc332bfSKa Ho Ng int
9367a202823SKonstantin Belousov kern_specialfd(struct thread *td, int type, void *arg)
9377a202823SKonstantin Belousov {
9387a202823SKonstantin Belousov 	struct file *fp;
9397a202823SKonstantin Belousov 	struct specialfd_eventfd *ae;
9407a202823SKonstantin Belousov 	int error, fd, fflags;
9417a202823SKonstantin Belousov 
9427a202823SKonstantin Belousov 	fflags = 0;
9437a202823SKonstantin Belousov 	error = falloc_noinstall(td, &fp);
9447a202823SKonstantin Belousov 	if (error != 0)
9457a202823SKonstantin Belousov 		return (error);
9467a202823SKonstantin Belousov 
9477a202823SKonstantin Belousov 	switch (type) {
9487a202823SKonstantin Belousov 	case SPECIALFD_EVENTFD:
9497a202823SKonstantin Belousov 		ae = arg;
9507a202823SKonstantin Belousov 		if ((ae->flags & EFD_CLOEXEC) != 0)
9517a202823SKonstantin Belousov 			fflags |= O_CLOEXEC;
9527a202823SKonstantin Belousov 		error = eventfd_create_file(td, fp, ae->initval, ae->flags);
9537a202823SKonstantin Belousov 		break;
9547a202823SKonstantin Belousov 	default:
9557a202823SKonstantin Belousov 		error = EINVAL;
9567a202823SKonstantin Belousov 		break;
9577a202823SKonstantin Belousov 	}
9587a202823SKonstantin Belousov 
9597a202823SKonstantin Belousov 	if (error == 0)
9607a202823SKonstantin Belousov 		error = finstall(td, fp, &fd, fflags, NULL);
9617a202823SKonstantin Belousov 	fdrop(fp, td);
9627a202823SKonstantin Belousov 	if (error == 0)
9637a202823SKonstantin Belousov 		td->td_retval[0] = fd;
9647a202823SKonstantin Belousov 	return (error);
9657a202823SKonstantin Belousov }
9667a202823SKonstantin Belousov 
9677a202823SKonstantin Belousov int
9687a202823SKonstantin Belousov sys___specialfd(struct thread *td, struct __specialfd_args *args)
9697a202823SKonstantin Belousov {
9707a202823SKonstantin Belousov 	struct specialfd_eventfd ae;
9717a202823SKonstantin Belousov 	int error;
9727a202823SKonstantin Belousov 
9737a202823SKonstantin Belousov 	switch (args->type) {
9747a202823SKonstantin Belousov 	case SPECIALFD_EVENTFD:
9757a202823SKonstantin Belousov 		if (args->len != sizeof(struct specialfd_eventfd)) {
9767a202823SKonstantin Belousov 			error = EINVAL;
9777a202823SKonstantin Belousov 			break;
9787a202823SKonstantin Belousov 		}
9797a202823SKonstantin Belousov 		error = copyin(args->req, &ae, sizeof(ae));
9807a202823SKonstantin Belousov 		if (error != 0)
9817a202823SKonstantin Belousov 			break;
9827a202823SKonstantin Belousov 		if ((ae.flags & ~(EFD_CLOEXEC | EFD_NONBLOCK |
9837a202823SKonstantin Belousov 		    EFD_SEMAPHORE)) != 0) {
9847a202823SKonstantin Belousov 			error = EINVAL;
9857a202823SKonstantin Belousov 			break;
9867a202823SKonstantin Belousov 		}
9877a202823SKonstantin Belousov 		error = kern_specialfd(td, args->type, &ae);
9887a202823SKonstantin Belousov 		break;
9897a202823SKonstantin Belousov 	default:
9907a202823SKonstantin Belousov 		error = EINVAL;
9917a202823SKonstantin Belousov 		break;
9927a202823SKonstantin Belousov 	}
9937a202823SKonstantin Belousov 	return (error);
9947a202823SKonstantin Belousov }
9957a202823SKonstantin Belousov 
9967a202823SKonstantin Belousov int
997125dcf8cSKonstantin Belousov poll_no_poll(int events)
998125dcf8cSKonstantin Belousov {
999125dcf8cSKonstantin Belousov 	/*
1000125dcf8cSKonstantin Belousov 	 * Return true for read/write.  If the user asked for something
1001125dcf8cSKonstantin Belousov 	 * special, return POLLNVAL, so that clients have a way of
1002125dcf8cSKonstantin Belousov 	 * determining reliably whether or not the extended
1003125dcf8cSKonstantin Belousov 	 * functionality is present without hard-coding knowledge
1004125dcf8cSKonstantin Belousov 	 * of specific filesystem implementations.
1005125dcf8cSKonstantin Belousov 	 */
1006125dcf8cSKonstantin Belousov 	if (events & ~POLLSTANDARD)
1007125dcf8cSKonstantin Belousov 		return (POLLNVAL);
1008125dcf8cSKonstantin Belousov 
1009125dcf8cSKonstantin Belousov 	return (events & (POLLIN | POLLOUT | POLLRDNORM | POLLWRNORM));
1010125dcf8cSKonstantin Belousov }
1011125dcf8cSKonstantin Belousov 
1012066d836bSKonstantin Belousov int
10138451d0ddSKip Macy sys_pselect(struct thread *td, struct pselect_args *uap)
1014066d836bSKonstantin Belousov {
1015066d836bSKonstantin Belousov 	struct timespec ts;
1016066d836bSKonstantin Belousov 	struct timeval tv, *tvp;
1017066d836bSKonstantin Belousov 	sigset_t set, *uset;
1018066d836bSKonstantin Belousov 	int error;
1019066d836bSKonstantin Belousov 
1020066d836bSKonstantin Belousov 	if (uap->ts != NULL) {
1021066d836bSKonstantin Belousov 		error = copyin(uap->ts, &ts, sizeof(ts));
1022066d836bSKonstantin Belousov 		if (error != 0)
1023066d836bSKonstantin Belousov 		    return (error);
1024066d836bSKonstantin Belousov 		TIMESPEC_TO_TIMEVAL(&tv, &ts);
1025066d836bSKonstantin Belousov 		tvp = &tv;
1026066d836bSKonstantin Belousov 	} else
1027066d836bSKonstantin Belousov 		tvp = NULL;
1028066d836bSKonstantin Belousov 	if (uap->sm != NULL) {
1029066d836bSKonstantin Belousov 		error = copyin(uap->sm, &set, sizeof(set));
1030066d836bSKonstantin Belousov 		if (error != 0)
1031066d836bSKonstantin Belousov 			return (error);
1032066d836bSKonstantin Belousov 		uset = &set;
1033066d836bSKonstantin Belousov 	} else
1034066d836bSKonstantin Belousov 		uset = NULL;
1035066d836bSKonstantin Belousov 	return (kern_pselect(td, uap->nd, uap->in, uap->ou, uap->ex, tvp,
1036066d836bSKonstantin Belousov 	    uset, NFDBITS));
1037066d836bSKonstantin Belousov }
1038066d836bSKonstantin Belousov 
1039066d836bSKonstantin Belousov int
1040066d836bSKonstantin Belousov kern_pselect(struct thread *td, int nd, fd_set *in, fd_set *ou, fd_set *ex,
1041066d836bSKonstantin Belousov     struct timeval *tvp, sigset_t *uset, int abi_nfdbits)
1042066d836bSKonstantin Belousov {
1043066d836bSKonstantin Belousov 	int error;
1044066d836bSKonstantin Belousov 
1045066d836bSKonstantin Belousov 	if (uset != NULL) {
1046066d836bSKonstantin Belousov 		error = kern_sigprocmask(td, SIG_SETMASK, uset,
1047066d836bSKonstantin Belousov 		    &td->td_oldsigmask, 0);
1048066d836bSKonstantin Belousov 		if (error != 0)
1049066d836bSKonstantin Belousov 			return (error);
1050066d836bSKonstantin Belousov 		td->td_pflags |= TDP_OLDMASK;
1051066d836bSKonstantin Belousov 		/*
1052066d836bSKonstantin Belousov 		 * Make sure that ast() is called on return to
1053066d836bSKonstantin Belousov 		 * usermode and TDP_OLDMASK is cleared, restoring old
1054066d836bSKonstantin Belousov 		 * sigmask.
1055066d836bSKonstantin Belousov 		 */
1056c6d31b83SKonstantin Belousov 		ast_sched(td, TDA_SIGSUSPEND);
1057066d836bSKonstantin Belousov 	}
1058066d836bSKonstantin Belousov 	error = kern_select(td, nd, in, ou, ex, tvp, abi_nfdbits);
1059066d836bSKonstantin Belousov 	return (error);
1060066d836bSKonstantin Belousov }
1061066d836bSKonstantin Belousov 
1062d2d3e875SBruce Evans #ifndef _SYS_SYSPROTO_H_
1063df8bae1dSRodney W. Grimes struct select_args {
1064b08f7993SSujal Patel 	int	nd;
1065df8bae1dSRodney W. Grimes 	fd_set	*in, *ou, *ex;
1066df8bae1dSRodney W. Grimes 	struct	timeval *tv;
1067df8bae1dSRodney W. Grimes };
1068d2d3e875SBruce Evans #endif
106926f9a767SRodney W. Grimes int
10708451d0ddSKip Macy sys_select(struct thread *td, struct select_args *uap)
1071df8bae1dSRodney W. Grimes {
10728f19eb88SIan Dowse 	struct timeval tv, *tvp;
10738f19eb88SIan Dowse 	int error;
10748f19eb88SIan Dowse 
10758f19eb88SIan Dowse 	if (uap->tv != NULL) {
10768f19eb88SIan Dowse 		error = copyin(uap->tv, &tv, sizeof(tv));
10778f19eb88SIan Dowse 		if (error)
10788f19eb88SIan Dowse 			return (error);
10798f19eb88SIan Dowse 		tvp = &tv;
10808f19eb88SIan Dowse 	} else
10818f19eb88SIan Dowse 		tvp = NULL;
10828f19eb88SIan Dowse 
1083b55ef216SKonstantin Belousov 	return (kern_select(td, uap->nd, uap->in, uap->ou, uap->ex, tvp,
1084b55ef216SKonstantin Belousov 	    NFDBITS));
10858f19eb88SIan Dowse }
10868f19eb88SIan Dowse 
108756be1b9aSKonstantin Belousov /*
108856be1b9aSKonstantin Belousov  * In the unlikely case when user specified n greater then the last
108956be1b9aSKonstantin Belousov  * open file descriptor, check that no bits are set after the last
109056be1b9aSKonstantin Belousov  * valid fd.  We must return EBADF if any is set.
109156be1b9aSKonstantin Belousov  *
109256be1b9aSKonstantin Belousov  * There are applications that rely on the behaviour.
109356be1b9aSKonstantin Belousov  *
1094d8bc2a17SMateusz Guzik  * nd is fd_nfiles.
109556be1b9aSKonstantin Belousov  */
109656be1b9aSKonstantin Belousov static int
109756be1b9aSKonstantin Belousov select_check_badfd(fd_set *fd_in, int nd, int ndu, int abi_nfdbits)
109856be1b9aSKonstantin Belousov {
109956be1b9aSKonstantin Belousov 	char *addr, *oaddr;
110056be1b9aSKonstantin Belousov 	int b, i, res;
110156be1b9aSKonstantin Belousov 	uint8_t bits;
110256be1b9aSKonstantin Belousov 
110356be1b9aSKonstantin Belousov 	if (nd >= ndu || fd_in == NULL)
110456be1b9aSKonstantin Belousov 		return (0);
110556be1b9aSKonstantin Belousov 
110656be1b9aSKonstantin Belousov 	oaddr = NULL;
110756be1b9aSKonstantin Belousov 	bits = 0; /* silence gcc */
110856be1b9aSKonstantin Belousov 	for (i = nd; i < ndu; i++) {
110956be1b9aSKonstantin Belousov 		b = i / NBBY;
111056be1b9aSKonstantin Belousov #if BYTE_ORDER == LITTLE_ENDIAN
111156be1b9aSKonstantin Belousov 		addr = (char *)fd_in + b;
111256be1b9aSKonstantin Belousov #else
111356be1b9aSKonstantin Belousov 		addr = (char *)fd_in;
111456be1b9aSKonstantin Belousov 		if (abi_nfdbits == NFDBITS) {
111556be1b9aSKonstantin Belousov 			addr += rounddown(b, sizeof(fd_mask)) +
111656be1b9aSKonstantin Belousov 			    sizeof(fd_mask) - 1 - b % sizeof(fd_mask);
111756be1b9aSKonstantin Belousov 		} else {
111856be1b9aSKonstantin Belousov 			addr += rounddown(b, sizeof(uint32_t)) +
111956be1b9aSKonstantin Belousov 			    sizeof(uint32_t) - 1 - b % sizeof(uint32_t);
112056be1b9aSKonstantin Belousov 		}
112156be1b9aSKonstantin Belousov #endif
112256be1b9aSKonstantin Belousov 		if (addr != oaddr) {
112356be1b9aSKonstantin Belousov 			res = fubyte(addr);
112456be1b9aSKonstantin Belousov 			if (res == -1)
112556be1b9aSKonstantin Belousov 				return (EFAULT);
112656be1b9aSKonstantin Belousov 			oaddr = addr;
112756be1b9aSKonstantin Belousov 			bits = res;
112856be1b9aSKonstantin Belousov 		}
112956be1b9aSKonstantin Belousov 		if ((bits & (1 << (i % NBBY))) != 0)
113056be1b9aSKonstantin Belousov 			return (EBADF);
113156be1b9aSKonstantin Belousov 	}
113256be1b9aSKonstantin Belousov 	return (0);
113356be1b9aSKonstantin Belousov }
113456be1b9aSKonstantin Belousov 
11358f19eb88SIan Dowse int
11368f19eb88SIan Dowse kern_select(struct thread *td, int nd, fd_set *fd_in, fd_set *fd_ou,
1137b55ef216SKonstantin Belousov     fd_set *fd_ex, struct timeval *tvp, int abi_nfdbits)
11388f19eb88SIan Dowse {
1139426da3bcSAlfred Perlstein 	struct filedesc *fdp;
1140d5e4d7e1SBruce Evans 	/*
1141d5e4d7e1SBruce Evans 	 * The magic 2048 here is chosen to be just enough for FD_SETSIZE
1142d5e4d7e1SBruce Evans 	 * infds with the new FD_SETSIZE of 1024, and more than enough for
1143d5e4d7e1SBruce Evans 	 * FD_SETSIZE infds, outfds and exceptfds with the old FD_SETSIZE
1144d5e4d7e1SBruce Evans 	 * of 256.
1145d5e4d7e1SBruce Evans 	 */
1146d5e4d7e1SBruce Evans 	fd_mask s_selbits[howmany(2048, NFDBITS)];
1147eb209311SAlfred Perlstein 	fd_mask *ibits[3], *obits[3], *selbits, *sbp;
1148cf5e4fe6SDavide Italiano 	struct timeval rtv;
1149cf5e4fe6SDavide Italiano 	sbintime_t asbt, precision, rsbt;
1150b55ef216SKonstantin Belousov 	u_int nbufbytes, ncpbytes, ncpubytes, nfdbits;
1151cf5e4fe6SDavide Italiano 	int error, lf, ndu;
1152df8bae1dSRodney W. Grimes 
11538f19eb88SIan Dowse 	if (nd < 0)
1154acbfbfeaSSujal Patel 		return (EINVAL);
1155426da3bcSAlfred Perlstein 	fdp = td->td_proc->p_fd;
115656be1b9aSKonstantin Belousov 	ndu = nd;
1157d8bc2a17SMateusz Guzik 	lf = fdp->fd_nfiles;
1158d8bc2a17SMateusz Guzik 	if (nd > lf)
1159d8bc2a17SMateusz Guzik 		nd = lf;
116056be1b9aSKonstantin Belousov 
116156be1b9aSKonstantin Belousov 	error = select_check_badfd(fd_in, nd, ndu, abi_nfdbits);
116256be1b9aSKonstantin Belousov 	if (error != 0)
116356be1b9aSKonstantin Belousov 		return (error);
116456be1b9aSKonstantin Belousov 	error = select_check_badfd(fd_ou, nd, ndu, abi_nfdbits);
116556be1b9aSKonstantin Belousov 	if (error != 0)
116656be1b9aSKonstantin Belousov 		return (error);
116756be1b9aSKonstantin Belousov 	error = select_check_badfd(fd_ex, nd, ndu, abi_nfdbits);
116856be1b9aSKonstantin Belousov 	if (error != 0)
116956be1b9aSKonstantin Belousov 		return (error);
1170b08f7993SSujal Patel 
1171d5e4d7e1SBruce Evans 	/*
1172d5e4d7e1SBruce Evans 	 * Allocate just enough bits for the non-null fd_sets.  Use the
1173d5e4d7e1SBruce Evans 	 * preallocated auto buffer if possible.
1174d5e4d7e1SBruce Evans 	 */
11758f19eb88SIan Dowse 	nfdbits = roundup(nd, NFDBITS);
1176d5e4d7e1SBruce Evans 	ncpbytes = nfdbits / NBBY;
1177b55ef216SKonstantin Belousov 	ncpubytes = roundup(nd, abi_nfdbits) / NBBY;
1178d5e4d7e1SBruce Evans 	nbufbytes = 0;
11798f19eb88SIan Dowse 	if (fd_in != NULL)
1180d5e4d7e1SBruce Evans 		nbufbytes += 2 * ncpbytes;
11818f19eb88SIan Dowse 	if (fd_ou != NULL)
1182d5e4d7e1SBruce Evans 		nbufbytes += 2 * ncpbytes;
11838f19eb88SIan Dowse 	if (fd_ex != NULL)
1184d5e4d7e1SBruce Evans 		nbufbytes += 2 * ncpbytes;
1185d5e4d7e1SBruce Evans 	if (nbufbytes <= sizeof s_selbits)
1186d5e4d7e1SBruce Evans 		selbits = &s_selbits[0];
1187d5e4d7e1SBruce Evans 	else
1188a163d034SWarner Losh 		selbits = malloc(nbufbytes, M_SELECT, M_WAITOK);
1189b08f7993SSujal Patel 
1190b08f7993SSujal Patel 	/*
1191d5e4d7e1SBruce Evans 	 * Assign pointers into the bit buffers and fetch the input bits.
1192d5e4d7e1SBruce Evans 	 * Put the output buffers together so that they can be bzeroed
1193d5e4d7e1SBruce Evans 	 * together.
1194b08f7993SSujal Patel 	 */
1195d5e4d7e1SBruce Evans 	sbp = selbits;
1196df8bae1dSRodney W. Grimes #define	getbits(name, x) \
1197d5e4d7e1SBruce Evans 	do {								\
1198841c0c7eSNathan Whitehorn 		if (name == NULL) {					\
1199d5e4d7e1SBruce Evans 			ibits[x] = NULL;				\
1200841c0c7eSNathan Whitehorn 			obits[x] = NULL;				\
1201841c0c7eSNathan Whitehorn 		} else {						\
1202d5e4d7e1SBruce Evans 			ibits[x] = sbp + nbufbytes / 2 / sizeof *sbp;	\
1203d5e4d7e1SBruce Evans 			obits[x] = sbp;					\
1204d5e4d7e1SBruce Evans 			sbp += ncpbytes / sizeof *sbp;			\
1205b55ef216SKonstantin Belousov 			error = copyin(name, ibits[x], ncpubytes);	\
1206265fc98fSSeigo Tanimura 			if (error != 0)					\
1207ace8398dSJeff Roberson 				goto done;				\
1208d6fda03aSMateusz Guzik 			if (ncpbytes != ncpubytes)			\
1209b55ef216SKonstantin Belousov 				bzero((char *)ibits[x] + ncpubytes,	\
1210b55ef216SKonstantin Belousov 				    ncpbytes - ncpubytes);		\
1211e04ac2feSJohn Baldwin 		}							\
1212d5e4d7e1SBruce Evans 	} while (0)
12138f19eb88SIan Dowse 	getbits(fd_in, 0);
12148f19eb88SIan Dowse 	getbits(fd_ou, 1);
12158f19eb88SIan Dowse 	getbits(fd_ex, 2);
1216df8bae1dSRodney W. Grimes #undef	getbits
1217841c0c7eSNathan Whitehorn 
1218841c0c7eSNathan Whitehorn #if BYTE_ORDER == BIG_ENDIAN && defined(__LP64__)
1219841c0c7eSNathan Whitehorn 	/*
1220841c0c7eSNathan Whitehorn 	 * XXX: swizzle_fdset assumes that if abi_nfdbits != NFDBITS,
1221841c0c7eSNathan Whitehorn 	 * we are running under 32-bit emulation. This should be more
1222841c0c7eSNathan Whitehorn 	 * generic.
1223841c0c7eSNathan Whitehorn 	 */
1224841c0c7eSNathan Whitehorn #define swizzle_fdset(bits)						\
1225841c0c7eSNathan Whitehorn 	if (abi_nfdbits != NFDBITS && bits != NULL) {			\
1226841c0c7eSNathan Whitehorn 		int i;							\
1227841c0c7eSNathan Whitehorn 		for (i = 0; i < ncpbytes / sizeof *sbp; i++)		\
1228841c0c7eSNathan Whitehorn 			bits[i] = (bits[i] >> 32) | (bits[i] << 32);	\
1229841c0c7eSNathan Whitehorn 	}
1230841c0c7eSNathan Whitehorn #else
1231841c0c7eSNathan Whitehorn #define swizzle_fdset(bits)
1232841c0c7eSNathan Whitehorn #endif
1233841c0c7eSNathan Whitehorn 
1234841c0c7eSNathan Whitehorn 	/* Make sure the bit order makes it through an ABI transition */
1235841c0c7eSNathan Whitehorn 	swizzle_fdset(ibits[0]);
1236841c0c7eSNathan Whitehorn 	swizzle_fdset(ibits[1]);
1237841c0c7eSNathan Whitehorn 	swizzle_fdset(ibits[2]);
1238841c0c7eSNathan Whitehorn 
1239d5e4d7e1SBruce Evans 	if (nbufbytes != 0)
1240d5e4d7e1SBruce Evans 		bzero(selbits, nbufbytes / 2);
1241df8bae1dSRodney W. Grimes 
1242cf5e4fe6SDavide Italiano 	precision = 0;
12438f19eb88SIan Dowse 	if (tvp != NULL) {
1244cf5e4fe6SDavide Italiano 		rtv = *tvp;
1245cf5e4fe6SDavide Italiano 		if (rtv.tv_sec < 0 || rtv.tv_usec < 0 ||
1246cf5e4fe6SDavide Italiano 		    rtv.tv_usec >= 1000000) {
1247df8bae1dSRodney W. Grimes 			error = EINVAL;
1248ace8398dSJeff Roberson 			goto done;
1249df8bae1dSRodney W. Grimes 		}
125021a37a71SAlexander Motin 		if (!timevalisset(&rtv))
1251980c545dSAlexander Motin 			asbt = 0;
125221a37a71SAlexander Motin 		else if (rtv.tv_sec <= INT32_MAX) {
1253cf5e4fe6SDavide Italiano 			rsbt = tvtosbt(rtv);
1254cf5e4fe6SDavide Italiano 			precision = rsbt;
1255cf5e4fe6SDavide Italiano 			precision >>= tc_precexp;
1256cf5e4fe6SDavide Italiano 			if (TIMESEL(&asbt, rsbt))
1257cf5e4fe6SDavide Italiano 				asbt += tc_tick_sbt;
12584bc38a5aSDavide Italiano 			if (asbt <= SBT_MAX - rsbt)
1259cf5e4fe6SDavide Italiano 				asbt += rsbt;
126021a37a71SAlexander Motin 			else
1261980c545dSAlexander Motin 				asbt = -1;
1262980c545dSAlexander Motin 		} else
1263980c545dSAlexander Motin 			asbt = -1;
1264cf5e4fe6SDavide Italiano 	} else
1265cf5e4fe6SDavide Italiano 		asbt = -1;
1266ace8398dSJeff Roberson 	seltdinit(td);
1267ace8398dSJeff Roberson 	/* Iterate until the timeout expires or descriptors become ready. */
1268ace8398dSJeff Roberson 	for (;;) {
12698f19eb88SIan Dowse 		error = selscan(td, ibits, obits, nd);
1270ace8398dSJeff Roberson 		if (error || td->td_retval[0] != 0)
1271ace8398dSJeff Roberson 			break;
1272cf5e4fe6SDavide Italiano 		error = seltdwait(td, asbt, precision);
1273ace8398dSJeff Roberson 		if (error)
1274ace8398dSJeff Roberson 			break;
1275ace8398dSJeff Roberson 		error = selrescan(td, ibits, obits);
1276ace8398dSJeff Roberson 		if (error || td->td_retval[0] != 0)
1277ace8398dSJeff Roberson 			break;
127885f190e4SAlfred Perlstein 	}
1279ace8398dSJeff Roberson 	seltdclear(td);
1280265fc98fSSeigo Tanimura 
1281df8bae1dSRodney W. Grimes done:
1282df8bae1dSRodney W. Grimes 	/* select is not restarted after signals... */
1283df8bae1dSRodney W. Grimes 	if (error == ERESTART)
1284df8bae1dSRodney W. Grimes 		error = EINTR;
1285df8bae1dSRodney W. Grimes 	if (error == EWOULDBLOCK)
1286df8bae1dSRodney W. Grimes 		error = 0;
1287841c0c7eSNathan Whitehorn 
1288841c0c7eSNathan Whitehorn 	/* swizzle bit order back, if necessary */
1289841c0c7eSNathan Whitehorn 	swizzle_fdset(obits[0]);
1290841c0c7eSNathan Whitehorn 	swizzle_fdset(obits[1]);
1291841c0c7eSNathan Whitehorn 	swizzle_fdset(obits[2]);
1292841c0c7eSNathan Whitehorn #undef swizzle_fdset
1293841c0c7eSNathan Whitehorn 
1294df8bae1dSRodney W. Grimes #define	putbits(name, x) \
1295b55ef216SKonstantin Belousov 	if (name && (error2 = copyout(obits[x], name, ncpubytes))) \
1296df8bae1dSRodney W. Grimes 		error = error2;
1297df8bae1dSRodney W. Grimes 	if (error == 0) {
1298df8bae1dSRodney W. Grimes 		int error2;
1299df8bae1dSRodney W. Grimes 
13008f19eb88SIan Dowse 		putbits(fd_in, 0);
13018f19eb88SIan Dowse 		putbits(fd_ou, 1);
13028f19eb88SIan Dowse 		putbits(fd_ex, 2);
1303df8bae1dSRodney W. Grimes #undef putbits
1304df8bae1dSRodney W. Grimes 	}
1305d5e4d7e1SBruce Evans 	if (selbits != &s_selbits[0])
1306d5e4d7e1SBruce Evans 		free(selbits, M_SELECT);
1307ad2edad9SMatthew Dillon 
1308df8bae1dSRodney W. Grimes 	return (error);
1309df8bae1dSRodney W. Grimes }
131011b763dfSJeff Roberson /*
131111b763dfSJeff Roberson  * Convert a select bit set to poll flags.
1312748b9df6SJeff Roberson  *
131311b763dfSJeff Roberson  * The backend always returns POLLHUP/POLLERR if appropriate and we
131411b763dfSJeff Roberson  * return this as a set bit in any set.
131511b763dfSJeff Roberson  */
1316cae3f9ddSMark Johnston static const int select_flags[3] = {
131711b763dfSJeff Roberson     POLLRDNORM | POLLHUP | POLLERR,
131811b763dfSJeff Roberson     POLLWRNORM | POLLHUP | POLLERR,
131961e53a38SKonstantin Belousov     POLLRDBAND | POLLERR
132011b763dfSJeff Roberson };
132111b763dfSJeff Roberson 
132211b763dfSJeff Roberson /*
132311b763dfSJeff Roberson  * Compute the fo_poll flags required for a fd given by the index and
132411b763dfSJeff Roberson  * bit position in the fd_mask array.
132511b763dfSJeff Roberson  */
132611b763dfSJeff Roberson static __inline int
132760b7f468SStephane E. Potvin selflags(fd_mask **ibits, int idx, fd_mask bit)
132811b763dfSJeff Roberson {
132911b763dfSJeff Roberson 	int flags;
133011b763dfSJeff Roberson 	int msk;
133111b763dfSJeff Roberson 
133211b763dfSJeff Roberson 	flags = 0;
133311b763dfSJeff Roberson 	for (msk = 0; msk < 3; msk++) {
133411b763dfSJeff Roberson 		if (ibits[msk] == NULL)
133511b763dfSJeff Roberson 			continue;
133660b7f468SStephane E. Potvin 		if ((ibits[msk][idx] & bit) == 0)
133711b763dfSJeff Roberson 			continue;
133811b763dfSJeff Roberson 		flags |= select_flags[msk];
133911b763dfSJeff Roberson 	}
134011b763dfSJeff Roberson 	return (flags);
134111b763dfSJeff Roberson }
134211b763dfSJeff Roberson 
134311b763dfSJeff Roberson /*
134411b763dfSJeff Roberson  * Set the appropriate output bits given a mask of fired events and the
134511b763dfSJeff Roberson  * input bits originally requested.
134611b763dfSJeff Roberson  */
134711b763dfSJeff Roberson static __inline int
134811b763dfSJeff Roberson selsetbits(fd_mask **ibits, fd_mask **obits, int idx, fd_mask bit, int events)
134911b763dfSJeff Roberson {
135011b763dfSJeff Roberson 	int msk;
135111b763dfSJeff Roberson 	int n;
135211b763dfSJeff Roberson 
135311b763dfSJeff Roberson 	n = 0;
135411b763dfSJeff Roberson 	for (msk = 0; msk < 3; msk++) {
135511b763dfSJeff Roberson 		if ((events & select_flags[msk]) == 0)
135611b763dfSJeff Roberson 			continue;
135711b763dfSJeff Roberson 		if (ibits[msk] == NULL)
135811b763dfSJeff Roberson 			continue;
135911b763dfSJeff Roberson 		if ((ibits[msk][idx] & bit) == 0)
136011b763dfSJeff Roberson 			continue;
136111b763dfSJeff Roberson 		/*
136211b763dfSJeff Roberson 		 * XXX Check for a duplicate set.  This can occur because a
136311b763dfSJeff Roberson 		 * socket calls selrecord() twice for each poll() call
136411b763dfSJeff Roberson 		 * resulting in two selfds per real fd.  selrescan() will
136511b763dfSJeff Roberson 		 * call selsetbits twice as a result.
136611b763dfSJeff Roberson 		 */
136711b763dfSJeff Roberson 		if ((obits[msk][idx] & bit) != 0)
136811b763dfSJeff Roberson 			continue;
136911b763dfSJeff Roberson 		obits[msk][idx] |= bit;
137011b763dfSJeff Roberson 		n++;
137111b763dfSJeff Roberson 	}
137211b763dfSJeff Roberson 
137311b763dfSJeff Roberson 	return (n);
137411b763dfSJeff Roberson }
1375df8bae1dSRodney W. Grimes 
1376ace8398dSJeff Roberson /*
1377ace8398dSJeff Roberson  * Traverse the list of fds attached to this thread's seltd and check for
1378ace8398dSJeff Roberson  * completion.
1379ace8398dSJeff Roberson  */
1380ace8398dSJeff Roberson static int
1381ace8398dSJeff Roberson selrescan(struct thread *td, fd_mask **ibits, fd_mask **obits)
1382ace8398dSJeff Roberson {
138311b763dfSJeff Roberson 	struct filedesc *fdp;
138411b763dfSJeff Roberson 	struct selinfo *si;
1385ace8398dSJeff Roberson 	struct seltd *stp;
1386ace8398dSJeff Roberson 	struct selfd *sfp;
1387ace8398dSJeff Roberson 	struct selfd *sfn;
1388ace8398dSJeff Roberson 	struct file *fp;
13899cdacff1SJeff Roberson 	fd_mask bit;
13909cdacff1SJeff Roberson 	int fd, ev, n, idx;
1391a9d2f8d8SRobert Watson 	int error;
13926affe1b7SMateusz Guzik 	bool only_user;
1393ace8398dSJeff Roberson 
139411b763dfSJeff Roberson 	fdp = td->td_proc->p_fd;
1395ace8398dSJeff Roberson 	stp = td->td_sel;
139611b763dfSJeff Roberson 	n = 0;
13976affe1b7SMateusz Guzik 	only_user = FILEDESC_IS_ONLY_USER(fdp);
1398ace8398dSJeff Roberson 	STAILQ_FOREACH_SAFE(sfp, &stp->st_selq, sf_link, sfn) {
1399ace8398dSJeff Roberson 		fd = (int)(uintptr_t)sfp->sf_cookie;
1400ace8398dSJeff Roberson 		si = sfp->sf_si;
1401ace8398dSJeff Roberson 		selfdfree(stp, sfp);
1402ace8398dSJeff Roberson 		/* If the selinfo wasn't cleared the event didn't fire. */
1403ace8398dSJeff Roberson 		if (si != NULL)
1404ace8398dSJeff Roberson 			continue;
14056affe1b7SMateusz Guzik 		if (only_user)
14066affe1b7SMateusz Guzik 			error = fget_only_user(fdp, fd, &cap_event_rights, &fp);
14076affe1b7SMateusz Guzik 		else
1408513c7a6eSMateusz Guzik 			error = fget_unlocked(td, fd, &cap_event_rights, &fp);
14096affe1b7SMateusz Guzik 		if (__predict_false(error != 0))
1410a9d2f8d8SRobert Watson 			return (error);
141111b763dfSJeff Roberson 		idx = fd / NFDBITS;
14129cdacff1SJeff Roberson 		bit = (fd_mask)1 << (fd % NFDBITS);
141311b763dfSJeff Roberson 		ev = fo_poll(fp, selflags(ibits, idx, bit), td->td_ucred, td);
14146affe1b7SMateusz Guzik 		if (only_user)
14156affe1b7SMateusz Guzik 			fput_only_user(fdp, fp);
14166affe1b7SMateusz Guzik 		else
1417bf422e5fSJeff Roberson 			fdrop(fp, td);
141811b763dfSJeff Roberson 		if (ev != 0)
141911b763dfSJeff Roberson 			n += selsetbits(ibits, obits, idx, bit, ev);
1420ace8398dSJeff Roberson 	}
1421ace8398dSJeff Roberson 	stp->st_flags = 0;
1422ace8398dSJeff Roberson 	td->td_retval[0] = n;
1423ace8398dSJeff Roberson 	return (0);
1424ace8398dSJeff Roberson }
1425ace8398dSJeff Roberson 
1426ace8398dSJeff Roberson /*
1427ace8398dSJeff Roberson  * Perform the initial filedescriptor scan and register ourselves with
1428ace8398dSJeff Roberson  * each selinfo.
1429ace8398dSJeff Roberson  */
1430265fc98fSSeigo Tanimura static int
1431cc3c9df8SEd Maste selscan(struct thread *td, fd_mask **ibits, fd_mask **obits, int nfd)
1432df8bae1dSRodney W. Grimes {
143311b763dfSJeff Roberson 	struct filedesc *fdp;
1434df8bae1dSRodney W. Grimes 	struct file *fp;
14359cdacff1SJeff Roberson 	fd_mask bit;
143611b763dfSJeff Roberson 	int ev, flags, end, fd;
14379cdacff1SJeff Roberson 	int n, idx;
1438a9d2f8d8SRobert Watson 	int error;
14396affe1b7SMateusz Guzik 	bool only_user;
1440df8bae1dSRodney W. Grimes 
144111b763dfSJeff Roberson 	fdp = td->td_proc->p_fd;
144211b763dfSJeff Roberson 	n = 0;
14436affe1b7SMateusz Guzik 	only_user = FILEDESC_IS_ONLY_USER(fdp);
14449cdacff1SJeff Roberson 	for (idx = 0, fd = 0; fd < nfd; idx++) {
144511b763dfSJeff Roberson 		end = imin(fd + NFDBITS, nfd);
144611b763dfSJeff Roberson 		for (bit = 1; fd < end; bit <<= 1, fd++) {
144711b763dfSJeff Roberson 			/* Compute the list of events we're interested in. */
144811b763dfSJeff Roberson 			flags = selflags(ibits, idx, bit);
144911b763dfSJeff Roberson 			if (flags == 0)
1450f082218cSPeter Wemm 				continue;
14516affe1b7SMateusz Guzik 			if (only_user)
14526affe1b7SMateusz Guzik 				error = fget_only_user(fdp, fd, &cap_event_rights, &fp);
14536affe1b7SMateusz Guzik 			else
1454513c7a6eSMateusz Guzik 				error = fget_unlocked(td, fd, &cap_event_rights, &fp);
14556affe1b7SMateusz Guzik 			if (__predict_false(error != 0))
1456a9d2f8d8SRobert Watson 				return (error);
1457ace8398dSJeff Roberson 			selfdalloc(td, (void *)(uintptr_t)fd);
145811b763dfSJeff Roberson 			ev = fo_poll(fp, flags, td->td_ucred, td);
14596affe1b7SMateusz Guzik 			if (only_user)
14606affe1b7SMateusz Guzik 				fput_only_user(fdp, fp);
14616affe1b7SMateusz Guzik 			else
1462bf422e5fSJeff Roberson 				fdrop(fp, td);
146311b763dfSJeff Roberson 			if (ev != 0)
146411b763dfSJeff Roberson 				n += selsetbits(ibits, obits, idx, bit, ev);
1465df8bae1dSRodney W. Grimes 		}
1466df8bae1dSRodney W. Grimes 	}
146711b763dfSJeff Roberson 
1468b40ce416SJulian Elischer 	td->td_retval[0] = n;
1469df8bae1dSRodney W. Grimes 	return (0);
1470df8bae1dSRodney W. Grimes }
1471df8bae1dSRodney W. Grimes 
147242d11757SPeter Wemm int
1473186d9c34SDmitry Chagin sys_poll(struct thread *td, struct poll_args *uap)
1474186d9c34SDmitry Chagin {
1475186d9c34SDmitry Chagin 	struct timespec ts, *tsp;
1476186d9c34SDmitry Chagin 
1477186d9c34SDmitry Chagin 	if (uap->timeout != INFTIM) {
1478186d9c34SDmitry Chagin 		if (uap->timeout < 0)
1479186d9c34SDmitry Chagin 			return (EINVAL);
1480186d9c34SDmitry Chagin 		ts.tv_sec = uap->timeout / 1000;
1481186d9c34SDmitry Chagin 		ts.tv_nsec = (uap->timeout % 1000) * 1000000;
1482186d9c34SDmitry Chagin 		tsp = &ts;
1483186d9c34SDmitry Chagin 	} else
1484186d9c34SDmitry Chagin 		tsp = NULL;
1485186d9c34SDmitry Chagin 
1486186d9c34SDmitry Chagin 	return (kern_poll(td, uap->fds, uap->nfds, tsp, NULL));
1487186d9c34SDmitry Chagin }
1488186d9c34SDmitry Chagin 
1489e884512aSDmitry Chagin /*
1490e884512aSDmitry Chagin  * kfds points to an array in the kernel.
1491e884512aSDmitry Chagin  */
1492186d9c34SDmitry Chagin int
1493e884512aSDmitry Chagin kern_poll_kfds(struct thread *td, struct pollfd *kfds, u_int nfds,
1494186d9c34SDmitry Chagin     struct timespec *tsp, sigset_t *uset)
149542d11757SPeter Wemm {
1496186d9c34SDmitry Chagin 	sbintime_t sbt, precision, tmp;
1497186d9c34SDmitry Chagin 	time_t over;
1498186d9c34SDmitry Chagin 	struct timespec ts;
1499cf5e4fe6SDavide Italiano 	int error;
150042d11757SPeter Wemm 
1501186d9c34SDmitry Chagin 	precision = 0;
1502186d9c34SDmitry Chagin 	if (tsp != NULL) {
150391e7bdcdSDmitry Chagin 		if (!timespecvalid_interval(tsp))
1504186d9c34SDmitry Chagin 			return (EINVAL);
1505186d9c34SDmitry Chagin 		if (tsp->tv_sec == 0 && tsp->tv_nsec == 0)
1506186d9c34SDmitry Chagin 			sbt = 0;
1507186d9c34SDmitry Chagin 		else {
1508186d9c34SDmitry Chagin 			ts = *tsp;
1509186d9c34SDmitry Chagin 			if (ts.tv_sec > INT32_MAX / 2) {
1510186d9c34SDmitry Chagin 				over = ts.tv_sec - INT32_MAX / 2;
1511186d9c34SDmitry Chagin 				ts.tv_sec -= over;
1512186d9c34SDmitry Chagin 			} else
1513186d9c34SDmitry Chagin 				over = 0;
1514186d9c34SDmitry Chagin 			tmp = tstosbt(ts);
1515186d9c34SDmitry Chagin 			precision = tmp;
1516186d9c34SDmitry Chagin 			precision >>= tc_precexp;
1517186d9c34SDmitry Chagin 			if (TIMESEL(&sbt, tmp))
1518186d9c34SDmitry Chagin 				sbt += tc_tick_sbt;
1519186d9c34SDmitry Chagin 			sbt += tmp;
1520186d9c34SDmitry Chagin 		}
1521186d9c34SDmitry Chagin 	} else
1522186d9c34SDmitry Chagin 		sbt = -1;
1523186d9c34SDmitry Chagin 
1524186d9c34SDmitry Chagin 	if (uset != NULL) {
1525186d9c34SDmitry Chagin 		error = kern_sigprocmask(td, SIG_SETMASK, uset,
1526186d9c34SDmitry Chagin 		    &td->td_oldsigmask, 0);
1527186d9c34SDmitry Chagin 		if (error)
1528e884512aSDmitry Chagin 			return (error);
1529186d9c34SDmitry Chagin 		td->td_pflags |= TDP_OLDMASK;
1530186d9c34SDmitry Chagin 		/*
1531186d9c34SDmitry Chagin 		 * Make sure that ast() is called on return to
1532186d9c34SDmitry Chagin 		 * usermode and TDP_OLDMASK is cleared, restoring old
1533186d9c34SDmitry Chagin 		 * sigmask.
1534186d9c34SDmitry Chagin 		 */
1535c6d31b83SKonstantin Belousov 		ast_sched(td, TDA_SIGSUSPEND);
153642d11757SPeter Wemm 	}
1537186d9c34SDmitry Chagin 
1538ace8398dSJeff Roberson 	seltdinit(td);
1539ace8398dSJeff Roberson 	/* Iterate until the timeout expires or descriptors become ready. */
1540ace8398dSJeff Roberson 	for (;;) {
15412384981bSConrad Meyer 		error = pollscan(td, kfds, nfds);
1542ace8398dSJeff Roberson 		if (error || td->td_retval[0] != 0)
1543ace8398dSJeff Roberson 			break;
1544186d9c34SDmitry Chagin 		error = seltdwait(td, sbt, precision);
1545ace8398dSJeff Roberson 		if (error)
1546ace8398dSJeff Roberson 			break;
1547ace8398dSJeff Roberson 		error = pollrescan(td);
1548ace8398dSJeff Roberson 		if (error || td->td_retval[0] != 0)
1549ace8398dSJeff Roberson 			break;
155085f190e4SAlfred Perlstein 	}
1551ace8398dSJeff Roberson 	seltdclear(td);
1552265fc98fSSeigo Tanimura 
155342d11757SPeter Wemm 	/* poll is not restarted after signals... */
155442d11757SPeter Wemm 	if (error == ERESTART)
155542d11757SPeter Wemm 		error = EINTR;
155642d11757SPeter Wemm 	if (error == EWOULDBLOCK)
155742d11757SPeter Wemm 		error = 0;
155842d11757SPeter Wemm 	return (error);
155942d11757SPeter Wemm }
156042d11757SPeter Wemm 
1561186d9c34SDmitry Chagin int
1562186d9c34SDmitry Chagin sys_ppoll(struct thread *td, struct ppoll_args *uap)
1563186d9c34SDmitry Chagin {
1564186d9c34SDmitry Chagin 	struct timespec ts, *tsp;
1565186d9c34SDmitry Chagin 	sigset_t set, *ssp;
1566186d9c34SDmitry Chagin 	int error;
1567186d9c34SDmitry Chagin 
1568186d9c34SDmitry Chagin 	if (uap->ts != NULL) {
1569186d9c34SDmitry Chagin 		error = copyin(uap->ts, &ts, sizeof(ts));
1570186d9c34SDmitry Chagin 		if (error)
1571186d9c34SDmitry Chagin 			return (error);
1572186d9c34SDmitry Chagin 		tsp = &ts;
1573186d9c34SDmitry Chagin 	} else
1574186d9c34SDmitry Chagin 		tsp = NULL;
1575186d9c34SDmitry Chagin 	if (uap->set != NULL) {
1576186d9c34SDmitry Chagin 		error = copyin(uap->set, &set, sizeof(set));
1577186d9c34SDmitry Chagin 		if (error)
1578186d9c34SDmitry Chagin 			return (error);
1579186d9c34SDmitry Chagin 		ssp = &set;
1580186d9c34SDmitry Chagin 	} else
1581186d9c34SDmitry Chagin 		ssp = NULL;
1582186d9c34SDmitry Chagin 	return (kern_poll(td, uap->fds, uap->nfds, tsp, ssp));
1583186d9c34SDmitry Chagin }
1584186d9c34SDmitry Chagin 
1585e884512aSDmitry Chagin /*
1586e884512aSDmitry Chagin  * ufds points to an array in user space.
1587e884512aSDmitry Chagin  */
1588e884512aSDmitry Chagin int
1589e884512aSDmitry Chagin kern_poll(struct thread *td, struct pollfd *ufds, u_int nfds,
1590e884512aSDmitry Chagin     struct timespec *tsp, sigset_t *set)
1591e884512aSDmitry Chagin {
1592e884512aSDmitry Chagin 	struct pollfd *kfds;
1593e884512aSDmitry Chagin 	struct pollfd stackfds[32];
1594e884512aSDmitry Chagin 	int error;
1595e884512aSDmitry Chagin 
1596e884512aSDmitry Chagin 	if (kern_poll_maxfds(nfds))
1597e884512aSDmitry Chagin 		return (EINVAL);
1598e884512aSDmitry Chagin 	if (nfds > nitems(stackfds))
1599e884512aSDmitry Chagin 		kfds = mallocarray(nfds, sizeof(*kfds), M_TEMP, M_WAITOK);
1600e884512aSDmitry Chagin 	else
1601e884512aSDmitry Chagin 		kfds = stackfds;
1602e884512aSDmitry Chagin 	error = copyin(ufds, kfds, nfds * sizeof(*kfds));
1603e884512aSDmitry Chagin 	if (error != 0)
1604e884512aSDmitry Chagin 		goto out;
1605e884512aSDmitry Chagin 
1606e884512aSDmitry Chagin 	error = kern_poll_kfds(td, kfds, nfds, tsp, set);
1607e884512aSDmitry Chagin 	if (error == 0)
1608e884512aSDmitry Chagin 		error = pollout(td, kfds, ufds, nfds);
1609e884512aSDmitry Chagin 
1610e884512aSDmitry Chagin out:
1611e884512aSDmitry Chagin 	if (nfds > nitems(stackfds))
1612e884512aSDmitry Chagin 		free(kfds, M_TEMP);
1613e884512aSDmitry Chagin 	return (error);
1614e884512aSDmitry Chagin }
1615e884512aSDmitry Chagin 
1616e884512aSDmitry Chagin bool
1617e884512aSDmitry Chagin kern_poll_maxfds(u_int nfds)
1618e884512aSDmitry Chagin {
1619e884512aSDmitry Chagin 
1620e884512aSDmitry Chagin 	/*
1621e884512aSDmitry Chagin 	 * This is kinda bogus.  We have fd limits, but that is not
1622e884512aSDmitry Chagin 	 * really related to the size of the pollfd array.  Make sure
1623e884512aSDmitry Chagin 	 * we let the process use at least FD_SETSIZE entries and at
1624e884512aSDmitry Chagin 	 * least enough for the system-wide limits.  We want to be reasonably
1625e884512aSDmitry Chagin 	 * safe, but not overly restrictive.
1626e884512aSDmitry Chagin 	 */
1627e884512aSDmitry Chagin 	return (nfds > maxfilesperproc && nfds > FD_SETSIZE);
1628e884512aSDmitry Chagin }
1629e884512aSDmitry Chagin 
163042d11757SPeter Wemm static int
1631ace8398dSJeff Roberson pollrescan(struct thread *td)
1632ace8398dSJeff Roberson {
1633ace8398dSJeff Roberson 	struct seltd *stp;
1634ace8398dSJeff Roberson 	struct selfd *sfp;
1635ace8398dSJeff Roberson 	struct selfd *sfn;
1636ace8398dSJeff Roberson 	struct selinfo *si;
1637ace8398dSJeff Roberson 	struct filedesc *fdp;
1638ace8398dSJeff Roberson 	struct file *fp;
1639ace8398dSJeff Roberson 	struct pollfd *fd;
164045e1f854SMateusz Guzik 	int n, error;
164145e1f854SMateusz Guzik 	bool only_user;
1642ace8398dSJeff Roberson 
1643ace8398dSJeff Roberson 	n = 0;
1644ace8398dSJeff Roberson 	fdp = td->td_proc->p_fd;
1645ace8398dSJeff Roberson 	stp = td->td_sel;
164645e1f854SMateusz Guzik 	only_user = FILEDESC_IS_ONLY_USER(fdp);
1647ace8398dSJeff Roberson 	STAILQ_FOREACH_SAFE(sfp, &stp->st_selq, sf_link, sfn) {
1648ace8398dSJeff Roberson 		fd = (struct pollfd *)sfp->sf_cookie;
1649ace8398dSJeff Roberson 		si = sfp->sf_si;
1650ace8398dSJeff Roberson 		selfdfree(stp, sfp);
1651ace8398dSJeff Roberson 		/* If the selinfo wasn't cleared the event didn't fire. */
1652ace8398dSJeff Roberson 		if (si != NULL)
1653ace8398dSJeff Roberson 			continue;
165445e1f854SMateusz Guzik 		if (only_user)
165545e1f854SMateusz Guzik 			error = fget_only_user(fdp, fd->fd, &cap_event_rights, &fp);
165645e1f854SMateusz Guzik 		else
1657513c7a6eSMateusz Guzik 			error = fget_unlocked(td, fd->fd, &cap_event_rights, &fp);
165845e1f854SMateusz Guzik 		if (__predict_false(error != 0)) {
1659ace8398dSJeff Roberson 			fd->revents = POLLNVAL;
1660ace8398dSJeff Roberson 			n++;
1661ace8398dSJeff Roberson 			continue;
1662ace8398dSJeff Roberson 		}
1663ace8398dSJeff Roberson 		/*
1664ace8398dSJeff Roberson 		 * Note: backend also returns POLLHUP and
1665ace8398dSJeff Roberson 		 * POLLERR if appropriate.
1666ace8398dSJeff Roberson 		 */
1667ace8398dSJeff Roberson 		fd->revents = fo_poll(fp, fd->events, td->td_ucred, td);
166845e1f854SMateusz Guzik 		if (only_user)
166945e1f854SMateusz Guzik 			fput_only_user(fdp, fp);
167045e1f854SMateusz Guzik 		else
167145e1f854SMateusz Guzik 			fdrop(fp, td);
1672ace8398dSJeff Roberson 		if (fd->revents != 0)
1673ace8398dSJeff Roberson 			n++;
1674ace8398dSJeff Roberson 	}
1675ace8398dSJeff Roberson 	stp->st_flags = 0;
1676ace8398dSJeff Roberson 	td->td_retval[0] = n;
1677ace8398dSJeff Roberson 	return (0);
1678ace8398dSJeff Roberson }
1679ace8398dSJeff Roberson 
1680ace8398dSJeff Roberson static int
1681cc3c9df8SEd Maste pollout(struct thread *td, struct pollfd *fds, struct pollfd *ufds, u_int nfd)
1682ae81968fSRobert Watson {
1683ae81968fSRobert Watson 	int error = 0;
1684ae81968fSRobert Watson 	u_int i = 0;
16856d8feddaSKonstantin Belousov 	u_int n = 0;
1686ae81968fSRobert Watson 
1687ae81968fSRobert Watson 	for (i = 0; i < nfd; i++) {
1688ae81968fSRobert Watson 		error = copyout(&fds->revents, &ufds->revents,
1689ae81968fSRobert Watson 		    sizeof(ufds->revents));
1690ae81968fSRobert Watson 		if (error)
1691ae81968fSRobert Watson 			return (error);
16926d8feddaSKonstantin Belousov 		if (fds->revents != 0)
16936d8feddaSKonstantin Belousov 			n++;
1694ae81968fSRobert Watson 		fds++;
1695ae81968fSRobert Watson 		ufds++;
1696ae81968fSRobert Watson 	}
16976d8feddaSKonstantin Belousov 	td->td_retval[0] = n;
1698ae81968fSRobert Watson 	return (0);
1699ae81968fSRobert Watson }
1700ae81968fSRobert Watson 
1701ae81968fSRobert Watson static int
1702cc3c9df8SEd Maste pollscan(struct thread *td, struct pollfd *fds, u_int nfd)
170342d11757SPeter Wemm {
1704b1607c87SMateusz Guzik 	struct filedesc *fdp;
170542d11757SPeter Wemm 	struct file *fp;
170645e1f854SMateusz Guzik 	int i, n, error;
170745e1f854SMateusz Guzik 	bool only_user;
170842d11757SPeter Wemm 
1709b1607c87SMateusz Guzik 	n = 0;
1710b1607c87SMateusz Guzik 	fdp = td->td_proc->p_fd;
171145e1f854SMateusz Guzik 	only_user = FILEDESC_IS_ONLY_USER(fdp);
1712eb209311SAlfred Perlstein 	for (i = 0; i < nfd; i++, fds++) {
1713b1607c87SMateusz Guzik 		if (fds->fd < 0) {
1714337c9691SJordan K. Hubbard 			fds->revents = 0;
1715b1607c87SMateusz Guzik 			continue;
1716b1607c87SMateusz Guzik 		}
171745e1f854SMateusz Guzik 		if (only_user)
171845e1f854SMateusz Guzik 			error = fget_only_user(fdp, fds->fd, &cap_event_rights, &fp);
171945e1f854SMateusz Guzik 		else
1720513c7a6eSMateusz Guzik 			error = fget_unlocked(td, fds->fd, &cap_event_rights, &fp);
172145e1f854SMateusz Guzik 		if (__predict_false(error != 0)) {
172242d11757SPeter Wemm 			fds->revents = POLLNVAL;
172342d11757SPeter Wemm 			n++;
1724b1607c87SMateusz Guzik 			continue;
1725b1607c87SMateusz Guzik 		}
17262087c896SBruce Evans 		/*
17272087c896SBruce Evans 		 * Note: backend also returns POLLHUP and
17282087c896SBruce Evans 		 * POLLERR if appropriate.
17292087c896SBruce Evans 		 */
1730ace8398dSJeff Roberson 		selfdalloc(td, fds);
173113ccadd4SBrian Feldman 		fds->revents = fo_poll(fp, fds->events,
1732ea6027a8SRobert Watson 		    td->td_ucred, td);
173345e1f854SMateusz Guzik 		if (only_user)
173445e1f854SMateusz Guzik 			fput_only_user(fdp, fp);
173545e1f854SMateusz Guzik 		else
173645e1f854SMateusz Guzik 			fdrop(fp, td);
1737f2159cc7SKonstantin Belousov 		/*
1738f2159cc7SKonstantin Belousov 		 * POSIX requires POLLOUT to be never
1739f2159cc7SKonstantin Belousov 		 * set simultaneously with POLLHUP.
1740f2159cc7SKonstantin Belousov 		 */
1741f2159cc7SKonstantin Belousov 		if ((fds->revents & POLLHUP) != 0)
1742f2159cc7SKonstantin Belousov 			fds->revents &= ~POLLOUT;
1743f2159cc7SKonstantin Belousov 
174442d11757SPeter Wemm 		if (fds->revents != 0)
174542d11757SPeter Wemm 			n++;
174642d11757SPeter Wemm 	}
1747b40ce416SJulian Elischer 	td->td_retval[0] = n;
174842d11757SPeter Wemm 	return (0);
174942d11757SPeter Wemm }
175042d11757SPeter Wemm 
175142d11757SPeter Wemm /*
1752237abf0cSDavide Italiano  * XXX This was created specifically to support netncp and netsmb.  This
1753237abf0cSDavide Italiano  * allows the caller to specify a socket to wait for events on.  It returns
1754237abf0cSDavide Italiano  * 0 if any events matched and an error otherwise.  There is no way to
1755237abf0cSDavide Italiano  * determine which events fired.
1756237abf0cSDavide Italiano  */
1757237abf0cSDavide Italiano int
1758237abf0cSDavide Italiano selsocket(struct socket *so, int events, struct timeval *tvp, struct thread *td)
1759237abf0cSDavide Italiano {
1760237abf0cSDavide Italiano 	struct timeval rtv;
1761237abf0cSDavide Italiano 	sbintime_t asbt, precision, rsbt;
1762237abf0cSDavide Italiano 	int error;
1763237abf0cSDavide Italiano 
176476665df9SPeter Wemm 	precision = 0;	/* stupid gcc! */
1765237abf0cSDavide Italiano 	if (tvp != NULL) {
1766237abf0cSDavide Italiano 		rtv = *tvp;
1767237abf0cSDavide Italiano 		if (rtv.tv_sec < 0 || rtv.tv_usec < 0 ||
1768237abf0cSDavide Italiano 		    rtv.tv_usec >= 1000000)
1769237abf0cSDavide Italiano 			return (EINVAL);
1770237abf0cSDavide Italiano 		if (!timevalisset(&rtv))
1771237abf0cSDavide Italiano 			asbt = 0;
1772237abf0cSDavide Italiano 		else if (rtv.tv_sec <= INT32_MAX) {
1773237abf0cSDavide Italiano 			rsbt = tvtosbt(rtv);
1774237abf0cSDavide Italiano 			precision = rsbt;
1775237abf0cSDavide Italiano 			precision >>= tc_precexp;
1776237abf0cSDavide Italiano 			if (TIMESEL(&asbt, rsbt))
1777237abf0cSDavide Italiano 				asbt += tc_tick_sbt;
17784bc38a5aSDavide Italiano 			if (asbt <= SBT_MAX - rsbt)
1779237abf0cSDavide Italiano 				asbt += rsbt;
1780237abf0cSDavide Italiano 			else
1781237abf0cSDavide Italiano 				asbt = -1;
1782237abf0cSDavide Italiano 		} else
1783237abf0cSDavide Italiano 			asbt = -1;
1784237abf0cSDavide Italiano 	} else
1785237abf0cSDavide Italiano 		asbt = -1;
1786237abf0cSDavide Italiano 	seltdinit(td);
1787237abf0cSDavide Italiano 	/*
1788237abf0cSDavide Italiano 	 * Iterate until the timeout expires or the socket becomes ready.
1789237abf0cSDavide Italiano 	 */
1790237abf0cSDavide Italiano 	for (;;) {
1791237abf0cSDavide Italiano 		selfdalloc(td, NULL);
179204c91ac4SBrooks Davis 		if (sopoll(so, events, NULL, td) != 0) {
179304c91ac4SBrooks Davis 			error = 0;
179404c91ac4SBrooks Davis 			break;
179504c91ac4SBrooks Davis 		}
1796237abf0cSDavide Italiano 		error = seltdwait(td, asbt, precision);
1797237abf0cSDavide Italiano 		if (error)
1798237abf0cSDavide Italiano 			break;
1799237abf0cSDavide Italiano 	}
1800237abf0cSDavide Italiano 	seltdclear(td);
1801237abf0cSDavide Italiano 	/* XXX Duplicates ncp/smb behavior. */
1802237abf0cSDavide Italiano 	if (error == ERESTART)
1803237abf0cSDavide Italiano 		error = 0;
1804237abf0cSDavide Italiano 	return (error);
1805237abf0cSDavide Italiano }
1806237abf0cSDavide Italiano 
1807237abf0cSDavide Italiano /*
1808ace8398dSJeff Roberson  * Preallocate two selfds associated with 'cookie'.  Some fo_poll routines
1809ace8398dSJeff Roberson  * have two select sets, one for read and another for write.
1810ace8398dSJeff Roberson  */
1811ace8398dSJeff Roberson static void
1812ace8398dSJeff Roberson selfdalloc(struct thread *td, void *cookie)
1813ace8398dSJeff Roberson {
1814ace8398dSJeff Roberson 	struct seltd *stp;
1815ace8398dSJeff Roberson 
1816ace8398dSJeff Roberson 	stp = td->td_sel;
1817ace8398dSJeff Roberson 	if (stp->st_free1 == NULL)
1818ea33cca9SMateusz Guzik 		stp->st_free1 = malloc(sizeof(*stp->st_free1), M_SELFD, M_WAITOK|M_ZERO);
1819ace8398dSJeff Roberson 	stp->st_free1->sf_td = stp;
1820ace8398dSJeff Roberson 	stp->st_free1->sf_cookie = cookie;
1821ace8398dSJeff Roberson 	if (stp->st_free2 == NULL)
1822ea33cca9SMateusz Guzik 		stp->st_free2 = malloc(sizeof(*stp->st_free2), M_SELFD, M_WAITOK|M_ZERO);
1823ace8398dSJeff Roberson 	stp->st_free2->sf_td = stp;
1824ace8398dSJeff Roberson 	stp->st_free2->sf_cookie = cookie;
1825ace8398dSJeff Roberson }
1826ace8398dSJeff Roberson 
1827ace8398dSJeff Roberson static void
1828ace8398dSJeff Roberson selfdfree(struct seltd *stp, struct selfd *sfp)
1829ace8398dSJeff Roberson {
1830ace8398dSJeff Roberson 	STAILQ_REMOVE(&stp->st_selq, sfp, selfd, sf_link);
183131b2ac4bSMateusz Guzik 	/*
183231b2ac4bSMateusz Guzik 	 * Paired with doselwakeup.
183331b2ac4bSMateusz Guzik 	 */
183431b2ac4bSMateusz Guzik 	if (atomic_load_acq_ptr((uintptr_t *)&sfp->sf_si) != (uintptr_t)NULL) {
1835ace8398dSJeff Roberson 		mtx_lock(sfp->sf_mtx);
1836fcb5b3a4SKonstantin Belousov 		if (sfp->sf_si != NULL) {
1837ace8398dSJeff Roberson 			TAILQ_REMOVE(&sfp->sf_si->si_tdlist, sfp, sf_threads);
1838fcb5b3a4SKonstantin Belousov 		}
1839ace8398dSJeff Roberson 		mtx_unlock(sfp->sf_mtx);
184073f2e5f7SMateusz Guzik 	}
1841ea33cca9SMateusz Guzik 	free(sfp, M_SELFD);
184285f190e4SAlfred Perlstein }
184385f190e4SAlfred Perlstein 
18446aba400aSAttilio Rao /* Drain the waiters tied to all the selfd belonging the specified selinfo. */
18456aba400aSAttilio Rao void
1846cc3c9df8SEd Maste seldrain(struct selinfo *sip)
18476aba400aSAttilio Rao {
18486aba400aSAttilio Rao 
18496aba400aSAttilio Rao 	/*
18506aba400aSAttilio Rao 	 * This feature is already provided by doselwakeup(), thus it is
18516aba400aSAttilio Rao 	 * enough to go for it.
18526aba400aSAttilio Rao 	 * Eventually, the context, should take care to avoid races
18536aba400aSAttilio Rao 	 * between thread calling select()/poll() and file descriptor
18546aba400aSAttilio Rao 	 * detaching, but, again, the races are just the same as
18556aba400aSAttilio Rao 	 * selwakeup().
18566aba400aSAttilio Rao 	 */
18576aba400aSAttilio Rao         doselwakeup(sip, -1);
18586aba400aSAttilio Rao }
18596aba400aSAttilio Rao 
1860df8bae1dSRodney W. Grimes /*
1861df8bae1dSRodney W. Grimes  * Record a select request.
1862df8bae1dSRodney W. Grimes  */
1863df8bae1dSRodney W. Grimes void
1864cc3c9df8SEd Maste selrecord(struct thread *selector, struct selinfo *sip)
1865df8bae1dSRodney W. Grimes {
1866ace8398dSJeff Roberson 	struct selfd *sfp;
1867ace8398dSJeff Roberson 	struct seltd *stp;
1868ace8398dSJeff Roberson 	struct mtx *mtxp;
1869df8bae1dSRodney W. Grimes 
1870ace8398dSJeff Roberson 	stp = selector->td_sel;
187185f190e4SAlfred Perlstein 	/*
1872ace8398dSJeff Roberson 	 * Don't record when doing a rescan.
187385f190e4SAlfred Perlstein 	 */
1874ace8398dSJeff Roberson 	if (stp->st_flags & SELTD_RESCAN)
1875ace8398dSJeff Roberson 		return;
1876ace8398dSJeff Roberson 	/*
1877ace8398dSJeff Roberson 	 * Grab one of the preallocated descriptors.
1878ace8398dSJeff Roberson 	 */
1879ace8398dSJeff Roberson 	sfp = NULL;
1880ace8398dSJeff Roberson 	if ((sfp = stp->st_free1) != NULL)
1881ace8398dSJeff Roberson 		stp->st_free1 = NULL;
1882ace8398dSJeff Roberson 	else if ((sfp = stp->st_free2) != NULL)
1883ace8398dSJeff Roberson 		stp->st_free2 = NULL;
1884ace8398dSJeff Roberson 	else
1885ace8398dSJeff Roberson 		panic("selrecord: No free selfd on selq");
18862141453eSJeff Roberson 	mtxp = sip->si_mtx;
18872141453eSJeff Roberson 	if (mtxp == NULL)
18882141453eSJeff Roberson 		mtxp = mtx_pool_find(mtxpool_select, sip);
1889ace8398dSJeff Roberson 	/*
1890ace8398dSJeff Roberson 	 * Initialize the sfp and queue it in the thread.
1891ace8398dSJeff Roberson 	 */
1892ace8398dSJeff Roberson 	sfp->sf_si = sip;
1893ace8398dSJeff Roberson 	sfp->sf_mtx = mtxp;
1894ace8398dSJeff Roberson 	STAILQ_INSERT_TAIL(&stp->st_selq, sfp, sf_link);
1895ace8398dSJeff Roberson 	/*
1896ace8398dSJeff Roberson 	 * Now that we've locked the sip, check for initialization.
1897ace8398dSJeff Roberson 	 */
1898ace8398dSJeff Roberson 	mtx_lock(mtxp);
1899ace8398dSJeff Roberson 	if (sip->si_mtx == NULL) {
1900ace8398dSJeff Roberson 		sip->si_mtx = mtxp;
1901ace8398dSJeff Roberson 		TAILQ_INIT(&sip->si_tdlist);
190285f190e4SAlfred Perlstein 	}
1903ace8398dSJeff Roberson 	/*
1904ace8398dSJeff Roberson 	 * Add this thread to the list of selfds listening on this selinfo.
1905ace8398dSJeff Roberson 	 */
1906ace8398dSJeff Roberson 	TAILQ_INSERT_TAIL(&sip->si_tdlist, sfp, sf_threads);
1907ace8398dSJeff Roberson 	mtx_unlock(sip->si_mtx);
1908df8bae1dSRodney W. Grimes }
1909df8bae1dSRodney W. Grimes 
1910512824f8SSeigo Tanimura /* Wake up a selecting thread. */
1911df8bae1dSRodney W. Grimes void
1912cc3c9df8SEd Maste selwakeup(struct selinfo *sip)
1913df8bae1dSRodney W. Grimes {
1914512824f8SSeigo Tanimura 	doselwakeup(sip, -1);
1915512824f8SSeigo Tanimura }
1916512824f8SSeigo Tanimura 
1917512824f8SSeigo Tanimura /* Wake up a selecting thread, and set its priority. */
1918512824f8SSeigo Tanimura void
1919cc3c9df8SEd Maste selwakeuppri(struct selinfo *sip, int pri)
1920512824f8SSeigo Tanimura {
1921512824f8SSeigo Tanimura 	doselwakeup(sip, pri);
1922512824f8SSeigo Tanimura }
1923512824f8SSeigo Tanimura 
1924512824f8SSeigo Tanimura /*
1925512824f8SSeigo Tanimura  * Do a wakeup when a selectable event occurs.
1926512824f8SSeigo Tanimura  */
1927512824f8SSeigo Tanimura static void
1928cc3c9df8SEd Maste doselwakeup(struct selinfo *sip, int pri)
1929512824f8SSeigo Tanimura {
1930ace8398dSJeff Roberson 	struct selfd *sfp;
1931ace8398dSJeff Roberson 	struct selfd *sfn;
1932ace8398dSJeff Roberson 	struct seltd *stp;
1933df8bae1dSRodney W. Grimes 
1934ace8398dSJeff Roberson 	/* If it's not initialized there can't be any waiters. */
1935ace8398dSJeff Roberson 	if (sip->si_mtx == NULL)
1936b40ce416SJulian Elischer 		return;
1937ace8398dSJeff Roberson 	/*
1938ace8398dSJeff Roberson 	 * Locking the selinfo locks all selfds associated with it.
1939ace8398dSJeff Roberson 	 */
1940ace8398dSJeff Roberson 	mtx_lock(sip->si_mtx);
1941ace8398dSJeff Roberson 	TAILQ_FOREACH_SAFE(sfp, &sip->si_tdlist, sf_threads, sfn) {
1942ace8398dSJeff Roberson 		/*
1943ace8398dSJeff Roberson 		 * Once we remove this sfp from the list and clear the
1944ace8398dSJeff Roberson 		 * sf_si seltdclear will know to ignore this si.
1945ace8398dSJeff Roberson 		 */
1946ace8398dSJeff Roberson 		TAILQ_REMOVE(&sip->si_tdlist, sfp, sf_threads);
1947ace8398dSJeff Roberson 		stp = sfp->sf_td;
1948ace8398dSJeff Roberson 		mtx_lock(&stp->st_mtx);
1949ace8398dSJeff Roberson 		stp->st_flags |= SELTD_PENDING;
1950ace8398dSJeff Roberson 		cv_broadcastpri(&stp->st_wait, pri);
1951ace8398dSJeff Roberson 		mtx_unlock(&stp->st_mtx);
195210e64782SMateusz Guzik 		/*
195310e64782SMateusz Guzik 		 * Paired with selfdfree.
195410e64782SMateusz Guzik 		 *
195510e64782SMateusz Guzik 		 * Storing this only after the wakeup provides an invariant that
195610e64782SMateusz Guzik 		 * stp is not used after selfdfree returns.
195710e64782SMateusz Guzik 		 */
195810e64782SMateusz Guzik 		atomic_store_rel_ptr((uintptr_t *)&sfp->sf_si, (uintptr_t)NULL);
1959b40ce416SJulian Elischer 	}
1960ace8398dSJeff Roberson 	mtx_unlock(sip->si_mtx);
1961ace8398dSJeff Roberson }
1962ace8398dSJeff Roberson 
1963ace8398dSJeff Roberson static void
1964ace8398dSJeff Roberson seltdinit(struct thread *td)
1965ace8398dSJeff Roberson {
1966ace8398dSJeff Roberson 	struct seltd *stp;
1967ace8398dSJeff Roberson 
196810e64782SMateusz Guzik 	stp = td->td_sel;
196910e64782SMateusz Guzik 	if (stp != NULL) {
197010e64782SMateusz Guzik 		MPASS(stp->st_flags == 0);
197110e64782SMateusz Guzik 		MPASS(STAILQ_EMPTY(&stp->st_selq));
197210e64782SMateusz Guzik 		return;
197310e64782SMateusz Guzik 	}
197410e64782SMateusz Guzik 	stp = malloc(sizeof(*stp), M_SELECT, M_WAITOK|M_ZERO);
1975ace8398dSJeff Roberson 	mtx_init(&stp->st_mtx, "sellck", NULL, MTX_DEF);
1976ace8398dSJeff Roberson 	cv_init(&stp->st_wait, "select");
1977ace8398dSJeff Roberson 	stp->st_flags = 0;
1978ace8398dSJeff Roberson 	STAILQ_INIT(&stp->st_selq);
197910e64782SMateusz Guzik 	td->td_sel = stp;
1980ace8398dSJeff Roberson }
1981ace8398dSJeff Roberson 
1982ace8398dSJeff Roberson static int
1983cf5e4fe6SDavide Italiano seltdwait(struct thread *td, sbintime_t sbt, sbintime_t precision)
1984ace8398dSJeff Roberson {
1985ace8398dSJeff Roberson 	struct seltd *stp;
1986ace8398dSJeff Roberson 	int error;
1987ace8398dSJeff Roberson 
1988ace8398dSJeff Roberson 	stp = td->td_sel;
1989ace8398dSJeff Roberson 	/*
1990ace8398dSJeff Roberson 	 * An event of interest may occur while we do not hold the seltd
1991ace8398dSJeff Roberson 	 * locked so check the pending flag before we sleep.
1992ace8398dSJeff Roberson 	 */
1993ace8398dSJeff Roberson 	mtx_lock(&stp->st_mtx);
1994ace8398dSJeff Roberson 	/*
1995ace8398dSJeff Roberson 	 * Any further calls to selrecord will be a rescan.
1996ace8398dSJeff Roberson 	 */
1997ace8398dSJeff Roberson 	stp->st_flags |= SELTD_RESCAN;
1998ace8398dSJeff Roberson 	if (stp->st_flags & SELTD_PENDING) {
1999ace8398dSJeff Roberson 		mtx_unlock(&stp->st_mtx);
2000ace8398dSJeff Roberson 		return (0);
2001ace8398dSJeff Roberson 	}
2002cf5e4fe6SDavide Italiano 	if (sbt == 0)
2003cf5e4fe6SDavide Italiano 		error = EWOULDBLOCK;
2004cf5e4fe6SDavide Italiano 	else if (sbt != -1)
2005cf5e4fe6SDavide Italiano 		error = cv_timedwait_sig_sbt(&stp->st_wait, &stp->st_mtx,
2006cf5e4fe6SDavide Italiano 		    sbt, precision, C_ABSOLUTE);
2007ace8398dSJeff Roberson 	else
2008ace8398dSJeff Roberson 		error = cv_wait_sig(&stp->st_wait, &stp->st_mtx);
2009ace8398dSJeff Roberson 	mtx_unlock(&stp->st_mtx);
2010ace8398dSJeff Roberson 
2011ace8398dSJeff Roberson 	return (error);
2012ace8398dSJeff Roberson }
2013ace8398dSJeff Roberson 
2014ace8398dSJeff Roberson void
2015ace8398dSJeff Roberson seltdfini(struct thread *td)
2016ace8398dSJeff Roberson {
2017ace8398dSJeff Roberson 	struct seltd *stp;
2018ace8398dSJeff Roberson 
2019ace8398dSJeff Roberson 	stp = td->td_sel;
2020ace8398dSJeff Roberson 	if (stp == NULL)
2021ace8398dSJeff Roberson 		return;
202210e64782SMateusz Guzik 	MPASS(stp->st_flags == 0);
202310e64782SMateusz Guzik 	MPASS(STAILQ_EMPTY(&stp->st_selq));
2024ace8398dSJeff Roberson 	if (stp->st_free1)
2025ea33cca9SMateusz Guzik 		free(stp->st_free1, M_SELFD);
2026ace8398dSJeff Roberson 	if (stp->st_free2)
2027ea33cca9SMateusz Guzik 		free(stp->st_free2, M_SELFD);
2028ace8398dSJeff Roberson 	td->td_sel = NULL;
202936bce27bSKonstantin Belousov 	cv_destroy(&stp->st_wait);
203036bce27bSKonstantin Belousov 	mtx_destroy(&stp->st_mtx);
2031ace8398dSJeff Roberson 	free(stp, M_SELECT);
2032ace8398dSJeff Roberson }
2033ace8398dSJeff Roberson 
2034ace8398dSJeff Roberson /*
2035ace8398dSJeff Roberson  * Remove the references to the thread from all of the objects we were
2036ace8398dSJeff Roberson  * polling.
2037ace8398dSJeff Roberson  */
2038ace8398dSJeff Roberson static void
2039ace8398dSJeff Roberson seltdclear(struct thread *td)
2040ace8398dSJeff Roberson {
2041ace8398dSJeff Roberson 	struct seltd *stp;
2042ace8398dSJeff Roberson 	struct selfd *sfp;
2043ace8398dSJeff Roberson 	struct selfd *sfn;
2044ace8398dSJeff Roberson 
2045ace8398dSJeff Roberson 	stp = td->td_sel;
2046ace8398dSJeff Roberson 	STAILQ_FOREACH_SAFE(sfp, &stp->st_selq, sf_link, sfn)
2047ace8398dSJeff Roberson 		selfdfree(stp, sfp);
2048ace8398dSJeff Roberson 	stp->st_flags = 0;
2049df8bae1dSRodney W. Grimes }
2050265fc98fSSeigo Tanimura 
20514d77a549SAlfred Perlstein static void selectinit(void *);
2052ace8398dSJeff Roberson SYSINIT(select, SI_SUB_SYSCALLS, SI_ORDER_ANY, selectinit, NULL);
2053265fc98fSSeigo Tanimura static void
2054ace8398dSJeff Roberson selectinit(void *dummy __unused)
2055265fc98fSSeigo Tanimura {
20562141453eSJeff Roberson 
20572141453eSJeff Roberson 	mtxpool_select = mtx_pool_create("select mtxpool", 128, MTX_DEF);
2058265fc98fSSeigo Tanimura }
20590acf5d0bSMark Johnston 
20600acf5d0bSMark Johnston /*
20610acf5d0bSMark Johnston  * Set up a syscall return value that follows the convention specified for
20620acf5d0bSMark Johnston  * posix_* functions.
20630acf5d0bSMark Johnston  */
20640acf5d0bSMark Johnston int
20650acf5d0bSMark Johnston kern_posix_error(struct thread *td, int error)
20660acf5d0bSMark Johnston {
20670acf5d0bSMark Johnston 
20680acf5d0bSMark Johnston 	if (error <= 0)
20690acf5d0bSMark Johnston 		return (error);
20700acf5d0bSMark Johnston 	td->td_errno = error;
20710acf5d0bSMark Johnston 	td->td_pflags |= TDP_NERRNO;
20720acf5d0bSMark Johnston 	td->td_retval[0] = error;
20730acf5d0bSMark Johnston 	return (0);
20740acf5d0bSMark Johnston }
2075d8decc9aSKonstantin Belousov 
2076d8decc9aSKonstantin Belousov int
2077d8decc9aSKonstantin Belousov kcmp_cmp(uintptr_t a, uintptr_t b)
2078d8decc9aSKonstantin Belousov {
2079d8decc9aSKonstantin Belousov 	if (a == b)
2080d8decc9aSKonstantin Belousov 		return (0);
2081d8decc9aSKonstantin Belousov 	else if (a < b)
2082d8decc9aSKonstantin Belousov 		return (1);
2083d8decc9aSKonstantin Belousov 	return (2);
2084d8decc9aSKonstantin Belousov }
2085d8decc9aSKonstantin Belousov 
2086d8decc9aSKonstantin Belousov static int
2087d8decc9aSKonstantin Belousov kcmp_pget(struct thread *td, pid_t pid, struct proc **pp)
2088d8decc9aSKonstantin Belousov {
2089d8decc9aSKonstantin Belousov 	if (pid == td->td_proc->p_pid) {
2090d8decc9aSKonstantin Belousov 		*pp = td->td_proc;
2091d8decc9aSKonstantin Belousov 		return (0);
2092d8decc9aSKonstantin Belousov 	}
2093d8decc9aSKonstantin Belousov 	return (pget(pid, PGET_CANDEBUG | PGET_NOTWEXIT | PGET_HOLD, pp));
2094d8decc9aSKonstantin Belousov }
2095d8decc9aSKonstantin Belousov 
2096d8decc9aSKonstantin Belousov int
2097d8decc9aSKonstantin Belousov kern_kcmp(struct thread *td, pid_t pid1, pid_t pid2, int type,
2098d8decc9aSKonstantin Belousov     uintptr_t idx1, uintptr_t idx2)
2099d8decc9aSKonstantin Belousov {
2100d8decc9aSKonstantin Belousov 	struct proc *p1, *p2;
2101d8decc9aSKonstantin Belousov 	struct file *fp1, *fp2;
2102d8decc9aSKonstantin Belousov 	int error, res;
2103d8decc9aSKonstantin Belousov 
2104d8decc9aSKonstantin Belousov 	res = -1;
2105d8decc9aSKonstantin Belousov 	p1 = p2 = NULL;
2106d8decc9aSKonstantin Belousov 	error = kcmp_pget(td, pid1, &p1);
2107d8decc9aSKonstantin Belousov 	if (error == 0)
2108d8decc9aSKonstantin Belousov 		error = kcmp_pget(td, pid2, &p2);
2109d8decc9aSKonstantin Belousov 	if (error != 0)
2110d8decc9aSKonstantin Belousov 		goto out;
2111d8decc9aSKonstantin Belousov 
2112d8decc9aSKonstantin Belousov 	switch (type) {
2113d8decc9aSKonstantin Belousov 	case KCMP_FILE:
2114d8decc9aSKonstantin Belousov 	case KCMP_FILEOBJ:
2115d8decc9aSKonstantin Belousov 		error = fget_remote(td, p1, idx1, &fp1);
2116d8decc9aSKonstantin Belousov 		if (error == 0) {
2117d8decc9aSKonstantin Belousov 			error = fget_remote(td, p2, idx2, &fp2);
2118d8decc9aSKonstantin Belousov 			if (error == 0) {
2119d8decc9aSKonstantin Belousov 				if (type == KCMP_FILEOBJ)
2120d8decc9aSKonstantin Belousov 					res = fo_cmp(fp1, fp2, td);
2121d8decc9aSKonstantin Belousov 				else
2122d8decc9aSKonstantin Belousov 					res = kcmp_cmp((uintptr_t)fp1,
2123d8decc9aSKonstantin Belousov 					    (uintptr_t)fp2);
2124d8decc9aSKonstantin Belousov 				fdrop(fp2, td);
2125d8decc9aSKonstantin Belousov 			}
2126d8decc9aSKonstantin Belousov 			fdrop(fp1, td);
2127d8decc9aSKonstantin Belousov 		}
2128d8decc9aSKonstantin Belousov 		break;
2129d8decc9aSKonstantin Belousov 	case KCMP_FILES:
2130d8decc9aSKonstantin Belousov 		res = kcmp_cmp((uintptr_t)p1->p_fd, (uintptr_t)p2->p_fd);
2131d8decc9aSKonstantin Belousov 		break;
2132d8decc9aSKonstantin Belousov 	case KCMP_SIGHAND:
2133d8decc9aSKonstantin Belousov 		res = kcmp_cmp((uintptr_t)p1->p_sigacts,
2134d8decc9aSKonstantin Belousov 		    (uintptr_t)p2->p_sigacts);
2135d8decc9aSKonstantin Belousov 		break;
2136d8decc9aSKonstantin Belousov 	case KCMP_VM:
2137d8decc9aSKonstantin Belousov 		res = kcmp_cmp((uintptr_t)p1->p_vmspace,
2138d8decc9aSKonstantin Belousov 		    (uintptr_t)p2->p_vmspace);
2139d8decc9aSKonstantin Belousov 		break;
2140d8decc9aSKonstantin Belousov 	default:
2141d8decc9aSKonstantin Belousov 		error = EINVAL;
2142d8decc9aSKonstantin Belousov 		break;
2143d8decc9aSKonstantin Belousov 	}
2144d8decc9aSKonstantin Belousov 
2145d8decc9aSKonstantin Belousov out:
2146d8decc9aSKonstantin Belousov 	if (p1 != NULL && p1 != td->td_proc)
2147d8decc9aSKonstantin Belousov 		PRELE(p1);
2148d8decc9aSKonstantin Belousov 	if (p2 != NULL && p2 != td->td_proc)
2149d8decc9aSKonstantin Belousov 		PRELE(p2);
2150d8decc9aSKonstantin Belousov 
2151d8decc9aSKonstantin Belousov 	td->td_retval[0] = res;
2152d8decc9aSKonstantin Belousov 	return (error);
2153d8decc9aSKonstantin Belousov }
2154d8decc9aSKonstantin Belousov 
2155d8decc9aSKonstantin Belousov int
2156d8decc9aSKonstantin Belousov sys_kcmp(struct thread *td, struct kcmp_args *uap)
2157d8decc9aSKonstantin Belousov {
2158d8decc9aSKonstantin Belousov 	return (kern_kcmp(td, uap->pid1, uap->pid2, uap->type,
2159d8decc9aSKonstantin Belousov 	    uap->idx1, uap->idx2));
2160d8decc9aSKonstantin Belousov }
2161*f28526e9SKonstantin Belousov 
2162*f28526e9SKonstantin Belousov int
2163*f28526e9SKonstantin Belousov file_kcmp_generic(struct file *fp1, struct file *fp2, struct thread *td)
2164*f28526e9SKonstantin Belousov {
2165*f28526e9SKonstantin Belousov 	if (fp1->f_type != fp2->f_type)
2166*f28526e9SKonstantin Belousov 		return (3);
2167*f28526e9SKonstantin Belousov 	return (kcmp_cmp((uintptr_t)fp1->f_data, (uintptr_t)fp2->f_data));
2168*f28526e9SKonstantin Belousov }
2169