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