X-Git-Url: https://git.saurik.com/apple/xnu.git/blobdiff_plain/490019cf9519204c5fb36b2fba54ceb983bb6b72..c3c9b80d004dbbfdf763edeb97968c6997e3b45b:/bsd/kern/kern_descrip.c diff --git a/bsd/kern/kern_descrip.c b/bsd/kern/kern_descrip.c index e089c5363..8952d0220 100644 --- a/bsd/kern/kern_descrip.c +++ b/bsd/kern/kern_descrip.c @@ -1,8 +1,8 @@ /* - * Copyright (c) 2000-2015 Apple Inc. All rights reserved. + * Copyright (c) 2000-2016 Apple Inc. All rights reserved. * * @APPLE_OSREFERENCE_LICENSE_HEADER_START@ - * + * * This file contains Original Code and/or Modifications of Original Code * as defined in and that are subject to the Apple Public Source License * Version 2.0 (the 'License'). You may not use this file except in @@ -11,10 +11,10 @@ * unlawful or unlicensed copies of an Apple operating system, or to * circumvent, violate, or enable the circumvention or violation of, any * terms of an Apple operating system software license agreement. - * + * * Please obtain a copy of the License at * http://www.opensource.apple.com/apsl/ and read it before using this file. - * + * * The Original Code and all software distributed under the License are * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, @@ -22,7 +22,7 @@ * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. * Please see the License for the specific language governing rights and * limitations under the License. - * + * * @APPLE_OSREFERENCE_LICENSE_HEADER_END@ */ /* Copyright (c) 1995, 1997 Apple Computer, Inc. All Rights Reserved */ @@ -87,6 +87,7 @@ #include #include #include +#include #include #include #include @@ -98,46 +99,43 @@ #include #include #include - -#include - #include #include #include #include #include +#include +#include + #include #include #include -#include - -#include - #include -#include +#include #include -#include -#include +#include +#if CONFIG_MACF +#include +#endif + +#include +#include +#include +#define IPC_OBJECT_COPYIN_FLAGS_ALLOW_IMMOVABLE_SEND 0x1 kern_return_t ipc_object_copyin(ipc_space_t, mach_port_name_t, - mach_msg_type_name_t, ipc_port_t *); + mach_msg_type_name_t, ipc_port_t *, mach_port_context_t, mach_msg_guard_flags_t *, uint32_t); void ipc_port_release_send(ipc_port_t); -struct psemnode; -struct pshmnode; - +static void fileproc_drain(proc_t, struct fileproc *); static int finishdup(proc_t p, struct filedesc *fdp, int old, int new, int flags, int32_t *retval); -int falloc_locked(proc_t p, struct fileproc **resultfp, int *resultfd, vfs_context_t ctx, int locked); -void fg_drop(struct fileproc * fp); -void fg_free(struct fileglob *fg); -void fg_ref(struct fileproc * fp); void fileport_releasefg(struct fileglob *fg); -/* flags for close_internal_locked */ +/* flags for fp_close_and_unlock */ #define FD_DUP2RESV 1 /* We don't want these exported */ @@ -145,41 +143,201 @@ void fileport_releasefg(struct fileglob *fg); __private_extern__ int unlink1(vfs_context_t, vnode_t, user_addr_t, enum uio_seg, int); -static void _fdrelse(struct proc * p, int fd); - +static void fdrelse(struct proc * p, int fd); -extern void file_lock_init(void); - -extern kauth_scope_t kauth_scope_fileop; +extern kauth_scope_t kauth_scope_fileop; /* Conflict wait queue for when selects collide (opaque type) */ extern struct waitq select_conflict_queue; -#define f_flag f_fglob->fg_flag -#define f_type f_fglob->fg_ops->fo_type -#define f_msgcount f_fglob->fg_msgcount -#define f_cred f_fglob->fg_cred -#define f_ops f_fglob->fg_ops -#define f_offset f_fglob->fg_offset -#define f_data f_fglob->fg_data +#ifndef HFS_GET_BOOT_INFO +#define HFS_GET_BOOT_INFO (FCNTL_FS_SPECIFIC_BASE + 0x00004) +#endif + +#ifndef HFS_SET_BOOT_INFO +#define HFS_SET_BOOT_INFO (FCNTL_FS_SPECIFIC_BASE + 0x00005) +#endif + +#ifndef APFSIOC_REVERT_TO_SNAPSHOT +#define APFSIOC_REVERT_TO_SNAPSHOT _IOW('J', 1, u_int64_t) +#endif + +#define f_flag fp_glob->fg_flag +#define f_type fp_glob->fg_ops->fo_type +#define f_cred fp_glob->fg_cred +#define f_ops fp_glob->fg_ops +#define f_offset fp_glob->fg_offset +#define f_data fp_glob->fg_data #define CHECK_ADD_OVERFLOW_INT64L(x, y) \ - (((((x) > 0) && ((y) > 0) && ((x) > LLONG_MAX - (y))) || \ - (((x) < 0) && ((y) < 0) && ((x) < LLONG_MIN - (y)))) \ - ? 1 : 0) + (((((x) > 0) && ((y) > 0) && ((x) > LLONG_MAX - (y))) || \ + (((x) < 0) && ((y) < 0) && ((x) < LLONG_MIN - (y)))) \ + ? 1 : 0) + +ZONE_DECLARE(fg_zone, "fileglob", + sizeof(struct fileglob), ZC_NOENCRYPT | ZC_ZFREE_CLEARMEM); +ZONE_DECLARE(fp_zone, "fileproc", + sizeof(struct fileproc), ZC_NOENCRYPT | ZC_ZFREE_CLEARMEM); +ZONE_DECLARE(fdp_zone, "filedesc", + sizeof(struct filedesc), ZC_NOENCRYPT | ZC_ZFREE_CLEARMEM); +/* + * If you need accounting for KM_OFILETABL consider using + * KALLOC_HEAP_DEFINE to define a view. + */ +#define KM_OFILETABL KHEAP_DEFAULT + /* * Descriptor management. */ -struct fmsglist fmsghead; /* head of list of open files */ -struct fmsglist fmsg_ithead; /* head of list of open files */ -int nfiles; /* actual number of open files */ +int nfiles; /* actual number of open files */ +/* + * "uninitialized" ops -- ensure FILEGLOB_DTYPE(fg) always exists + */ +static const struct fileops uninitops; + +os_refgrp_decl(, f_refgrp, "files refcounts", NULL); +static LCK_GRP_DECLARE(file_lck_grp, "file"); + +#pragma mark fileglobs + +/*! + * @function fg_free + * + * @brief + * Free a file structure. + */ +static void +fg_free(struct fileglob *fg) +{ + os_atomic_dec(&nfiles, relaxed); + + if (fg->fg_vn_data) { + fg_vn_data_free(fg->fg_vn_data); + fg->fg_vn_data = NULL; + } + + if (IS_VALID_CRED(fg->fg_cred)) { + kauth_cred_unref(&fg->fg_cred); + } + lck_mtx_destroy(&fg->fg_lock, &file_lck_grp); + +#if CONFIG_MACF + mac_file_label_destroy(fg); +#endif + zfree(fg_zone, fg); +} + +OS_ALWAYS_INLINE +void +fg_ref(proc_t p, struct fileglob *fg) +{ +#if DEBUG || DEVELOPMENT + proc_fdlock_assert(p, LCK_MTX_ASSERT_OWNED); +#else + (void)p; +#endif + os_ref_retain_raw(&fg->fg_count, &f_refgrp); +} + +void +fg_drop_live(struct fileglob *fg) +{ + os_ref_release_live_raw(&fg->fg_count, &f_refgrp); +} + +int +fg_drop(proc_t p, struct fileglob *fg) +{ + struct vnode *vp; + struct vfs_context context; + int error = 0; + + if (fg == NULL) { + return 0; + } + + /* Set up context with cred stashed in fg */ + if (p == current_proc()) { + context.vc_thread = current_thread(); + } else { + context.vc_thread = NULL; + } + context.vc_ucred = fg->fg_cred; + + /* + * POSIX record locking dictates that any close releases ALL + * locks owned by this process. This is handled by setting + * a flag in the unlock to free ONLY locks obeying POSIX + * semantics, and not to free BSD-style file locks. + * If the descriptor was in a message, POSIX-style locks + * aren't passed with the descriptor. + */ + if (p && DTYPE_VNODE == FILEGLOB_DTYPE(fg) && + (p->p_ladvflag & P_LADVLOCK)) { + struct flock lf = { + .l_whence = SEEK_SET, + .l_type = F_UNLCK, + }; + + vp = (struct vnode *)fg->fg_data; + if ((error = vnode_getwithref(vp)) == 0) { + (void)VNOP_ADVLOCK(vp, (caddr_t)p, F_UNLCK, &lf, F_POSIX, &context, NULL); + (void)vnode_put(vp); + } + } + + if (os_ref_release_raw(&fg->fg_count, &f_refgrp) == 0) { + /* + * Since we ensure that fg->fg_ops is always initialized, + * it is safe to invoke fo_close on the fg + */ + error = fo_close(fg, &context); + + fg_free(fg); + } + + return error; +} + +/* + * fg_get_vnode + * + * Description: Return vnode associated with the file structure, if + * any. The lifetime of the returned vnode is bound to + * the lifetime of the file structure. + * + * Parameters: fg Pointer to fileglob to + * inspect + * + * Returns: vnode_t + */ +vnode_t +fg_get_vnode(struct fileglob *fg) +{ + if (FILEGLOB_DTYPE(fg) == DTYPE_VNODE) { + return (vnode_t)fg->fg_data; + } else { + return NULL; + } +} +bool +fg_sendable(struct fileglob *fg) +{ + switch (FILEGLOB_DTYPE(fg)) { + case DTYPE_VNODE: + case DTYPE_SOCKET: + case DTYPE_PIPE: + case DTYPE_PSXSHM: + case DTYPE_NETPOLICY: + return (fg->fg_lflags & FG_CONFINED) == 0; -lck_grp_attr_t * file_lck_grp_attr; -lck_grp_t * file_lck_grp; -lck_attr_t * file_lck_attr; + default: + return false; + } +} -lck_mtx_t * uipc_lock; +#pragma mark fileprocs /* * check_file_seek_range @@ -189,9 +347,9 @@ lck_mtx_t * uipc_lock; * Parameters: fl Flock structure. * cur_file_offset Current offset in the file. * - * Returns: 0 on Success. + * Returns: 0 on Success. * EOVERFLOW on overflow. - * EINVAL on offset less than zero. + * EINVAL on offset less than zero. */ static int @@ -211,13 +369,13 @@ check_file_seek_range(struct flock *fl, off_t cur_file_offset) return EINVAL; } /* Check if end marker is beyond LLONG_MAX. */ - if ((fl->l_len > 0) && (CHECK_ADD_OVERFLOW_INT64L(fl->l_start + - cur_file_offset, fl->l_len - 1))) { + if ((fl->l_len > 0) && (CHECK_ADD_OVERFLOW_INT64L(fl->l_start + + cur_file_offset, fl->l_len - 1))) { return EOVERFLOW; } /* Check if the end marker is negative. */ if ((fl->l_len <= 0) && (fl->l_start + cur_file_offset + - fl->l_len < 0)) { + fl->l_len < 0)) { return EINVAL; } } else if (fl->l_whence == SEEK_SET) { @@ -226,12 +384,12 @@ check_file_seek_range(struct flock *fl, off_t cur_file_offset) return EINVAL; } /* Check if the end marker is beyond LLONG_MAX. */ - if ((fl->l_len > 0) && + if ((fl->l_len > 0) && CHECK_ADD_OVERFLOW_INT64L(fl->l_start, fl->l_len - 1)) { return EOVERFLOW; } /* Check if the end marker is negative. */ - if ((fl->l_len < 0) && fl->l_start + fl->l_len < 0) { + if ((fl->l_len < 0) && fl->l_start + fl->l_len < 0) { return EINVAL; } } @@ -239,31 +397,29 @@ check_file_seek_range(struct flock *fl, off_t cur_file_offset) } -/* - * file_lock_init - * - * Description: Initialize the file lock group and the uipc and flist locks - * - * Parameters: (void) - * - * Returns: void - * - * Notes: Called at system startup from bsd_init(). - */ void -file_lock_init(void) +proc_dirs_lock_shared(proc_t p) { - /* allocate file lock group attribute and group */ - file_lck_grp_attr= lck_grp_attr_alloc_init(); - - file_lck_grp = lck_grp_alloc_init("file", file_lck_grp_attr); + lck_rw_lock_shared(&p->p_dirs_lock); +} - /* Allocate file lock attribute */ - file_lck_attr = lck_attr_alloc_init(); +void +proc_dirs_unlock_shared(proc_t p) +{ + lck_rw_unlock_shared(&p->p_dirs_lock); +} - uipc_lock = lck_mtx_alloc_init(file_lck_grp, file_lck_attr); +void +proc_dirs_lock_exclusive(proc_t p) +{ + lck_rw_lock_exclusive(&p->p_dirs_lock); } +void +proc_dirs_unlock_exclusive(proc_t p) +{ + lck_rw_unlock_exclusive(&p->p_dirs_lock); +} /* * proc_fdlock, proc_fdlock_spin @@ -312,6 +468,56 @@ proc_fdunlock(proc_t p) lck_mtx_unlock(&p->p_fdmlock); } +struct fdt_iterator +fdt_next(proc_t p, int fd, bool only_settled) +{ + struct fdt_iterator it; + struct filedesc *fdp = p->p_fd; + struct fileproc *fp; + int nfds = min(fdp->fd_lastfile + 1, fdp->fd_nfiles); + + while (++fd < nfds) { + fp = fdp->fd_ofiles[fd]; + if (fp == NULL || fp->fp_glob == NULL) { + continue; + } + if (only_settled && (fdp->fd_ofileflags[fd] & UF_RESERVED)) { + continue; + } + it.fdti_fd = fd; + it.fdti_fp = fp; + return it; + } + + it.fdti_fd = nfds; + it.fdti_fp = NULL; + return it; +} + +struct fdt_iterator +fdt_prev(proc_t p, int fd, bool only_settled) +{ + struct fdt_iterator it; + struct filedesc *fdp = p->p_fd; + struct fileproc *fp; + + while (--fd >= 0) { + fp = fdp->fd_ofiles[fd]; + if (fp == NULL || fp->fp_glob == NULL) { + continue; + } + if (only_settled && (fdp->fd_ofileflags[fd] & UF_RESERVED)) { + continue; + } + it.fdti_fd = fd; + it.fdti_fp = fp; + return it; + } + + it.fdti_fd = -1; + it.fdti_fp = NULL; + return it; +} /* * System calls on descriptors. @@ -319,7 +525,7 @@ proc_fdunlock(proc_t p) /* - * getdtablesize + * sys_getdtablesize * * Description: Returns the per process maximum size of the descriptor table * @@ -332,64 +538,56 @@ proc_fdunlock(proc_t p) * *retval (modified) Size of dtable */ int -getdtablesize(proc_t p, __unused struct getdtablesize_args *uap, int32_t *retval) +sys_getdtablesize(proc_t p, __unused struct getdtablesize_args *uap, int32_t *retval) { - proc_fdlock_spin(p); - *retval = min((int)p->p_rlimit[RLIMIT_NOFILE].rlim_cur, maxfiles); - proc_fdunlock(p); + *retval = (int32_t)MIN(proc_limitgetcur(p, RLIMIT_NOFILE, TRUE), maxfilesperproc); - return (0); + return 0; } -void +static void procfdtbl_reservefd(struct proc * p, int fd) { p->p_fd->fd_ofiles[fd] = NULL; - p->p_fd->fd_ofileflags[fd] |= UF_RESERVED; -} - -void -procfdtbl_markclosefd(struct proc * p, int fd) -{ - p->p_fd->fd_ofileflags[fd] |= (UF_RESERVED | UF_CLOSING); + p->p_fd->fd_ofileflags[fd] |= UF_RESERVED; } void procfdtbl_releasefd(struct proc * p, int fd, struct fileproc * fp) { - if (fp != NULL) - p->p_fd->fd_ofiles[fd] = fp; - p->p_fd->fd_ofileflags[fd] &= ~UF_RESERVED; + if (fp != NULL) { + p->p_fd->fd_ofiles[fd] = fp; + } + p->p_fd->fd_ofileflags[fd] &= ~UF_RESERVED; if ((p->p_fd->fd_ofileflags[fd] & UF_RESVWAIT) == UF_RESVWAIT) { p->p_fd->fd_ofileflags[fd] &= ~UF_RESVWAIT; wakeup(&p->p_fd); } } -void +static void procfdtbl_waitfd(struct proc * p, int fd) { - p->p_fd->fd_ofileflags[fd] |= UF_RESVWAIT; + p->p_fd->fd_ofileflags[fd] |= UF_RESVWAIT; msleep(&p->p_fd, &p->p_fdmlock, PRIBIO, "ftbl_waitfd", NULL); } - -void +static void procfdtbl_clearfd(struct proc * p, int fd) { int waiting; waiting = (p->p_fd->fd_ofileflags[fd] & UF_RESVWAIT); - p->p_fd->fd_ofiles[fd] = NULL; + p->p_fd->fd_ofiles[fd] = NULL; p->p_fd->fd_ofileflags[fd] = 0; - if ( waiting == UF_RESVWAIT) { + if (waiting == UF_RESVWAIT) { wakeup(&p->p_fd); } } /* - * _fdrelse + * fdrelse * * Description: Inline utility function to free an fd in a filedesc * @@ -403,23 +601,27 @@ procfdtbl_clearfd(struct proc * p, int fd) * the caller */ static void -_fdrelse(struct proc * p, int fd) +fdrelse(struct proc * p, int fd) { struct filedesc *fdp = p->p_fd; int nfd = 0; - if (fd < fdp->fd_freefile) + if (fd < fdp->fd_freefile) { fdp->fd_freefile = fd; + } #if DIAGNOSTIC - if (fd > fdp->fd_lastfile) - panic("fdrelse: fd_lastfile inconsistent"); + if (fd > fdp->fd_lastfile) { + panic("fdrelse: fd_lastfile inconsistent"); + } #endif procfdtbl_clearfd(p, fd); while ((nfd = fdp->fd_lastfile) > 0 && - fdp->fd_ofiles[nfd] == NULL && - !(fdp->fd_ofileflags[nfd] & UF_RESERVED)) + fdp->fd_ofiles[nfd] == NULL && + !(fdp->fd_ofileflags[nfd] & UF_RESERVED)) { + /* JMM - What about files with lingering EV_VANISHED knotes? */ fdp->fd_lastfile--; + } } @@ -430,80 +632,85 @@ fd_rdwr( uint64_t base, int64_t len, enum uio_seg segflg, - off_t offset, - int io_flg, + off_t offset, + int io_flg, int64_t *aresid) { - struct fileproc *fp; - proc_t p; - int error = 0; + struct fileproc *fp; + proc_t p; + int error = 0; int flags = 0; int spacetype; uio_t auio = NULL; - char uio_buf[ UIO_SIZEOF(1) ]; + char uio_buf[UIO_SIZEOF(1)]; struct vfs_context context = *(vfs_context_current()); - bool wrote_some = false; p = current_proc(); - error = fp_lookup(p, fd, &fp, 0); - if (error) - return(error); + error = fp_lookup(p, fd, &fp, 0); + if (error) { + return error; + } - if (fp->f_type != DTYPE_VNODE && fp->f_type != DTYPE_PIPE && fp->f_type != DTYPE_SOCKET) { + switch (FILEGLOB_DTYPE(fp->fp_glob)) { + case DTYPE_VNODE: + case DTYPE_PIPE: + case DTYPE_SOCKET: + break; + default: error = EINVAL; goto out; } if (rw == UIO_WRITE && !(fp->f_flag & FWRITE)) { - error = EBADF; + error = EBADF; goto out; } - + if (rw == UIO_READ && !(fp->f_flag & FREAD)) { - error = EBADF; - goto out; + error = EBADF; + goto out; } - - context.vc_ucred = fp->f_fglob->fg_cred; - if (UIO_SEG_IS_USER_SPACE(segflg)) + context.vc_ucred = fp->fp_glob->fg_cred; + + if (UIO_SEG_IS_USER_SPACE(segflg)) { spacetype = proc_is64bit(p) ? UIO_USERSPACE64 : UIO_USERSPACE32; - else + } else { spacetype = UIO_SYSSPACE; + } auio = uio_createwithbuffer(1, offset, spacetype, rw, &uio_buf[0], sizeof(uio_buf)); - uio_addiov(auio, base, len); + uio_addiov(auio, (user_addr_t)base, (user_size_t)len); - if ( !(io_flg & IO_APPEND)) + if (!(io_flg & IO_APPEND)) { flags = FOF_OFFSET; + } if (rw == UIO_WRITE) { user_ssize_t orig_resid = uio_resid(auio); error = fo_write(fp, auio, flags, &context); - wrote_some = uio_resid(auio) < orig_resid; - } else + if (uio_resid(auio) < orig_resid) { + os_atomic_or(&fp->fp_glob->fg_flag, FWASWRITTEN, relaxed); + } + } else { error = fo_read(fp, auio, flags, &context); + } - if (aresid) + if (aresid) { *aresid = uio_resid(auio); - else { - if (uio_resid(auio) && error == 0) - error = EIO; + } else if (uio_resid(auio) && error == 0) { + error = EIO; } out: - if (wrote_some) - fp_drop_written(p, fd, fp); - else - fp_drop(p, fd, fp, 0); - + fp_drop(p, fd, fp, 0); return error; } /* - * dup + * sys_dup * * Description: Duplicate a file descriptor. * @@ -518,7 +725,7 @@ out: * *retval (modified) The new descriptor */ int -dup(proc_t p, struct dup_args *uap, int32_t *retval) +sys_dup(proc_t p, struct dup_args *uap, int32_t *retval) { struct filedesc *fdp = p->p_fd; int old = uap->fd; @@ -526,35 +733,35 @@ dup(proc_t p, struct dup_args *uap, int32_t *retval) struct fileproc *fp; proc_fdlock(p); - if ( (error = fp_lookup(p, old, &fp, 1)) ) { + if ((error = fp_lookup(p, old, &fp, 1))) { proc_fdunlock(p); - return(error); + return error; } - if (FP_ISGUARDED(fp, GUARD_DUP)) { + if (fp_isguarded(fp, GUARD_DUP)) { error = fp_guard_exception(p, old, fp, kGUARD_EXC_DUP); (void) fp_drop(p, old, fp, 1); proc_fdunlock(p); - return (error); + return error; } - if ( (error = fdalloc(p, 0, &new)) ) { + if ((error = fdalloc(p, 0, &new))) { fp_drop(p, old, fp, 1); proc_fdunlock(p); - return (error); + return error; } error = finishdup(p, fdp, old, new, 0, retval); fp_drop(p, old, fp, 1); proc_fdunlock(p); - if (ENTR_SHOULDTRACE && fp->f_type == DTYPE_SOCKET) { + if (ENTR_SHOULDTRACE && FILEGLOB_DTYPE(fp->fp_glob) == DTYPE_SOCKET) { KERNEL_ENERGYTRACE(kEnTrActKernSocket, DBG_FUNC_START, new, 0, (int64_t)VM_KERNEL_ADDRPERM(fp->f_data)); } - return (error); + return error; } /* - * dup2 + * sys_dup2 * * Description: Duplicate a file descriptor to a particular value. * @@ -570,44 +777,50 @@ dup(proc_t p, struct dup_args *uap, int32_t *retval) * *retval (modified) The new descriptor */ int -dup2(proc_t p, struct dup2_args *uap, int32_t *retval) +sys_dup2(proc_t p, struct dup2_args *uap, int32_t *retval) +{ + return dup2(p, uap->from, uap->to, retval); +} + +int +dup2(proc_t p, int old, int new, int *retval) { struct filedesc *fdp = p->p_fd; - int old = uap->from, new = uap->to; - int i, error; struct fileproc *fp, *nfp; + int i, error; + rlim_t nofile = proc_limitgetcur(p, RLIMIT_NOFILE, TRUE); proc_fdlock(p); startover: - if ( (error = fp_lookup(p, old, &fp, 1)) ) { + if ((error = fp_lookup(p, old, &fp, 1))) { proc_fdunlock(p); - return(error); + return error; } - if (FP_ISGUARDED(fp, GUARD_DUP)) { + if (fp_isguarded(fp, GUARD_DUP)) { error = fp_guard_exception(p, old, fp, kGUARD_EXC_DUP); (void) fp_drop(p, old, fp, 1); proc_fdunlock(p); - return (error); + return error; } if (new < 0 || - (rlim_t)new >= p->p_rlimit[RLIMIT_NOFILE].rlim_cur || - new >= maxfiles) { + (rlim_t)new >= nofile || + new >= maxfilesperproc) { fp_drop(p, old, fp, 1); proc_fdunlock(p); - return (EBADF); + return EBADF; } if (old == new) { fp_drop(p, old, fp, 1); *retval = new; proc_fdunlock(p); - return (0); + return 0; } if (new < 0 || new >= fdp->fd_nfiles) { - if ( (error = fdalloc(p, new, &i)) ) { + if ((error = fdalloc(p, new, &i))) { fp_drop(p, old, fp, 1); proc_fdunlock(p); - return (error); + return error; } if (new != i) { fdrelse(p, i); @@ -615,55 +828,48 @@ startover: } } else { closeit: - while ((fdp->fd_ofileflags[new] & UF_RESERVED) == UF_RESERVED) { - fp_drop(p, old, fp, 1); - procfdtbl_waitfd(p, new); + if ((fdp->fd_ofileflags[new] & UF_RESERVED) == UF_RESERVED) { + fp_drop(p, old, fp, 1); + procfdtbl_waitfd(p, new); #if DIAGNOSTIC - proc_fdlock_assert(p, LCK_MTX_ASSERT_OWNED); + proc_fdlock_assert(p, LCK_MTX_ASSERT_OWNED); #endif - goto startover; + goto startover; } - if ((fdp->fd_ofiles[new] != NULL) && - ((error = fp_lookup(p, new, &nfp, 1)) == 0)) { - fp_drop(p, old, fp, 1); - if (FP_ISGUARDED(nfp, GUARD_CLOSE)) { + if ((nfp = fdp->fd_ofiles[new]) != NULL) { + if (fp_isguarded(nfp, GUARD_CLOSE)) { + fp_drop(p, old, fp, 1); error = fp_guard_exception(p, new, nfp, kGUARD_EXC_CLOSE); - (void) fp_drop(p, new, nfp, 1); proc_fdunlock(p); - return (error); + return error; } - (void)close_internal_locked(p, new, nfp, FD_DUP2RESV); -#if DIAGNOSTIC - proc_fdlock_assert(p, LCK_MTX_ASSERT_OWNED); -#endif - procfdtbl_clearfd(p, new); - goto startover; - } else { + (void)fp_close_and_unlock(p, new, nfp, FD_DUP2RESV); + proc_fdlock(p); + assert(fdp->fd_ofileflags[new] & UF_RESERVED); + } else { #if DIAGNOSTIC - if (fdp->fd_ofiles[new] != NULL) + if (fdp->fd_ofiles[new] != NULL) { panic("dup2: no ref on fileproc %d", new); + } #endif procfdtbl_reservefd(p, new); } - -#if DIAGNOSTIC - proc_fdlock_assert(p, LCK_MTX_ASSERT_OWNED); -#endif - } #if DIAGNOSTIC - if (fdp->fd_ofiles[new] != 0) + if (fdp->fd_ofiles[new] != 0) { panic("dup2: overwriting fd_ofiles with new %d", new); - if ((fdp->fd_ofileflags[new] & UF_RESERVED) == 0) + } + if ((fdp->fd_ofileflags[new] & UF_RESERVED) == 0) { panic("dup2: unreserved fileflags with new %d", new); + } #endif error = finishdup(p, fdp, old, new, 0, retval); fp_drop(p, old, fp, 1); proc_fdunlock(p); - return(error); + return error; } @@ -689,15 +895,17 @@ closeit: * blocking operation. */ int -fcntl(proc_t p, struct fcntl_args *uap, int32_t *retval) +sys_fcntl(proc_t p, struct fcntl_args *uap, int32_t *retval) { __pthread_testcancel(1); - return(fcntl_nocancel(p, (struct fcntl_nocancel_args *)uap, retval)); + return sys_fcntl_nocancel(p, (struct fcntl_nocancel_args *)uap, retval); } +#define ACCOUNT_OPENFROM_ENTITLEMENT \ + "com.apple.private.vfs.role-account-openfrom" /* - * fcntl_nocancel + * sys_fcntl_nocancel * * Description: A non-cancel-testing file control system call. * @@ -740,6 +948,7 @@ fcntl(proc_t p, struct fcntl_args *uap, int32_t *retval) * VNOP_ALLOCATE:??? * [F_SETSIZE,F_RDADVISE] * EBADF + * EINVAL * copyin:EFAULT * vnode_getwithref:??? * [F_RDAHEAD,F_NOCACHE] @@ -751,15 +960,16 @@ fcntl(proc_t p, struct fcntl_args *uap, int32_t *retval) * *retval (modified) fcntl return value (if any) */ int -fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) +sys_fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) { int fd = uap->fd; struct filedesc *fdp = p->p_fd; struct fileproc *fp; char *pop; - struct vnode *vp = NULLVP; /* for AUDIT_ARG() at end */ + struct vnode *vp = NULLVP; /* for AUDIT_ARG() at end */ + unsigned int oflags, nflags; int i, tmp, error, error2, flg = 0; - struct flock fl; + struct flock fl = {}; struct flocktimeout fltimeout; struct timespec *timeout = NULL; struct vfs_context context; @@ -769,14 +979,18 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) unsigned int fflag; user_addr_t argp; boolean_t is64bit; + rlim_t nofile; + int has_entitlement = 0; AUDIT_ARG(fd, uap->fd); AUDIT_ARG(cmd, uap->cmd); + nofile = proc_limitgetcur(p, RLIMIT_NOFILE, TRUE); + proc_fdlock(p); - if ( (error = fp_lookup(p, fd, &fp, 1)) ) { + if ((error = fp_lookup(p, fd, &fp, 1))) { proc_fdunlock(p); - return(error); + return error; } context.vc_thread = current_thread(); context.vc_ucred = fp->f_cred; @@ -784,8 +998,7 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) is64bit = proc_is64bit(p); if (is64bit) { argp = uap->arg; - } - else { + } else { /* * Since the arg parameter is defined as a long but may be * either a long or a pointer we must take care to handle @@ -798,32 +1011,33 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) argp = CAST_USER_ADDR_T((uint32_t)uap->arg); } - pop = &fdp->fd_ofileflags[fd]; - #if CONFIG_MACF - error = mac_file_check_fcntl(proc_ucred(p), fp->f_fglob, uap->cmd, + error = mac_file_check_fcntl(proc_ucred(p), fp->fp_glob, uap->cmd, uap->arg); - if (error) + if (error) { goto out; + } #endif - switch (uap->cmd) { + pop = &fdp->fd_ofileflags[fd]; + switch (uap->cmd) { case F_DUPFD: case F_DUPFD_CLOEXEC: - if (FP_ISGUARDED(fp, GUARD_DUP)) { + if (fp_isguarded(fp, GUARD_DUP)) { error = fp_guard_exception(p, fd, fp, kGUARD_EXC_DUP); goto out; } newmin = CAST_DOWN_EXPLICIT(int, uap->arg); /* arg is an int, so we won't lose bits */ AUDIT_ARG(value32, newmin); - if ((u_int)newmin >= p->p_rlimit[RLIMIT_NOFILE].rlim_cur || - newmin >= maxfiles) { + if ((rlim_t)newmin >= nofile || + newmin >= maxfilesperproc) { error = EINVAL; goto out; } - if ( (error = fdalloc(p, newmin, &i)) ) + if ((error = fdalloc(p, newmin, &i))) { goto out; + } error = finishdup(p, fdp, fd, i, uap->cmd == F_DUPFD_CLOEXEC ? UF_EXCLOSE : 0, retval); goto out; @@ -834,11 +1048,11 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) goto out; case F_SETFD: - AUDIT_ARG(value32, uap->arg); - if (uap->arg & FD_CLOEXEC) + AUDIT_ARG(value32, (uint32_t)uap->arg); + if (uap->arg & FD_CLOEXEC) { *pop |= UF_EXCLOSE; - else { - if (FILEPROC_TYPE(fp) == FTYPE_GUARDED) { + } else { + if (fp_isguarded(fp, 0)) { error = fp_guard_exception(p, fd, fp, kGUARD_EXC_NOCLOEXEC); goto out; @@ -854,19 +1068,29 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) goto out; case F_SETFL: - fp->f_flag &= ~FCNTLFLAGS; + // FIXME (rdar://54898652) + // + // this code is broken if fnctl(F_SETFL), ioctl() are + // called concurrently for the same fileglob. + tmp = CAST_DOWN_EXPLICIT(int, uap->arg); /* arg is an int, so we won't lose bits */ AUDIT_ARG(value32, tmp); - fp->f_flag |= FFLAGS(tmp) & FCNTLFLAGS; - tmp = fp->f_flag & FNONBLOCK; + + os_atomic_rmw_loop(&fp->f_flag, oflags, nflags, relaxed, { + nflags = oflags & ~FCNTLFLAGS; + nflags |= FFLAGS(tmp) & FCNTLFLAGS; + }); + tmp = nflags & FNONBLOCK; error = fo_ioctl(fp, FIONBIO, (caddr_t)&tmp, &context); - if (error) + if (error) { goto out; - tmp = fp->f_flag & FASYNC; + } + tmp = nflags & FASYNC; error = fo_ioctl(fp, FIOASYNC, (caddr_t)&tmp, &context); - if (!error) + if (!error) { goto out; - fp->f_flag &= ~FNONBLOCK; + } + os_atomic_andnot(&fp->f_flag, FNONBLOCK, relaxed); tmp = 0; (void)fo_ioctl(fp, FIONBIO, (caddr_t)&tmp, &context); goto out; @@ -877,7 +1101,7 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) error = 0; goto out; } - error = fo_ioctl(fp, (int)TIOCGPGRP, (caddr_t)retval, &context); + error = fo_ioctl(fp, TIOCGPGRP, (caddr_t)retval, &context); *retval = -*retval; goto out; @@ -886,7 +1110,7 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) AUDIT_ARG(value32, tmp); if (fp->f_type == DTYPE_SOCKET) { ((struct socket *)fp->f_data)->so_pgid = tmp; - error =0; + error = 0; goto out; } if (fp->f_type == DTYPE_PIPE) { @@ -905,7 +1129,7 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) tmp = (int)p1->p_pgrpid; proc_rele(p1); } - error = fo_ioctl(fp, (int)TIOCSPGRP, (caddr_t)&tmp, &context); + error = fo_ioctl(fp, TIOCSPGRP, (caddr_t)&tmp, &context); goto out; case F_SETNOSIGPIPE: @@ -913,18 +1137,19 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) if (fp->f_type == DTYPE_SOCKET) { #if SOCKETS error = sock_setsockopt((struct socket *)fp->f_data, - SOL_SOCKET, SO_NOSIGPIPE, &tmp, sizeof (tmp)); + SOL_SOCKET, SO_NOSIGPIPE, &tmp, sizeof(tmp)); #else error = EINVAL; #endif } else { - struct fileglob *fg = fp->f_fglob; + struct fileglob *fg = fp->fp_glob; lck_mtx_lock_spin(&fg->fg_lock); - if (tmp) + if (tmp) { fg->fg_lflags |= FG_NOSIGPIPE; - else - fg->fg_lflags &= FG_NOSIGPIPE; + } else { + fg->fg_lflags &= ~FG_NOSIGPIPE; + } lck_mtx_unlock(&fg->fg_lock); error = 0; } @@ -933,15 +1158,15 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) case F_GETNOSIGPIPE: if (fp->f_type == DTYPE_SOCKET) { #if SOCKETS - int retsize = sizeof (*retval); + int retsize = sizeof(*retval); error = sock_getsockopt((struct socket *)fp->f_data, SOL_SOCKET, SO_NOSIGPIPE, retval, &retsize); #else error = EINVAL; #endif } else { - *retval = (fp->f_fglob->fg_lflags & FG_NOSIGPIPE) ? - 1 : 0; + *retval = (fp->fp_glob->fg_lflags & FG_NOSIGPIPE) ? + 1 : 0; error = 0; } goto out; @@ -955,30 +1180,31 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) * mechanism to move the descriptor elsewhere will fail. */ if (CAST_DOWN_EXPLICIT(int, uap->arg)) { - struct fileglob *fg = fp->f_fglob; + struct fileglob *fg = fp->fp_glob; lck_mtx_lock_spin(&fg->fg_lock); - if (fg->fg_lflags & FG_CONFINED) + if (fg->fg_lflags & FG_CONFINED) { error = 0; - else if (1 != fg->fg_count) - error = EAGAIN; /* go close the dup .. */ - else if (UF_FORKCLOSE == (*pop & UF_FORKCLOSE)) { + } else if (1 != os_ref_get_count_raw(&fg->fg_count)) { + error = EAGAIN; /* go close the dup .. */ + } else if (UF_FORKCLOSE == (*pop & UF_FORKCLOSE)) { fg->fg_lflags |= FG_CONFINED; error = 0; - } else - error = EBADF; /* open without O_CLOFORK? */ - lck_mtx_unlock(&fg->fg_lock); + } else { + error = EBADF; /* open without O_CLOFORK? */ + } + lck_mtx_unlock(&fg->fg_lock); } else { /* * Other subsystems may have built on the immutability * of FG_CONFINED; clearing it may be tricky. */ - error = EPERM; /* immutable */ + error = EPERM; /* immutable */ } goto out; case F_GETCONFINED: - *retval = (fp->f_fglob->fg_lflags & FG_CONFINED) ? 1 : 0; + *retval = (fp->fp_glob->fg_lflags & FG_CONFINED) ? 1 : 0; error = 0; goto out; @@ -987,7 +1213,7 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) case F_OFD_SETLKWTIMEOUT: case F_OFD_SETLKW: flg |= F_WAIT; - /* Fall into F_SETLK */ + OS_FALLTHROUGH; case F_SETLK: case F_OFD_SETLK: @@ -1024,14 +1250,15 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) goto outdrop; } - if ( (error = vnode_getwithref(vp)) ) { + if ((error = vnode_getwithref(vp))) { goto outdrop; } - if (fl.l_whence == SEEK_CUR) + if (fl.l_whence == SEEK_CUR) { fl.l_start += offset; + } #if CONFIG_MACF - error = mac_file_check_lock(proc_ucred(p), fp->f_fglob, + error = mac_file_check_lock(proc_ucred(p), fp->fp_glob, F_SETLK, &fl); if (error) { (void)vnode_put(vp); @@ -1049,7 +1276,7 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) error = EBADF; break; } - error = VNOP_ADVLOCK(vp, (caddr_t)fp->f_fglob, + error = VNOP_ADVLOCK(vp, (caddr_t)fp->fp_glob, F_SETLK, &fl, flg, &context, timeout); break; case F_WRLCK: @@ -1057,11 +1284,11 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) error = EBADF; break; } - error = VNOP_ADVLOCK(vp, (caddr_t)fp->f_fglob, + error = VNOP_ADVLOCK(vp, (caddr_t)fp->fp_glob, F_SETLK, &fl, flg, &context, timeout); break; case F_UNLCK: - error = VNOP_ADVLOCK(vp, (caddr_t)fp->f_fglob, + error = VNOP_ADVLOCK(vp, (caddr_t)fp->fp_glob, F_UNLCK, &fl, F_OFD_LOCK, &context, timeout); break; @@ -1071,7 +1298,7 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) } if (0 == error && (F_RDLCK == fl.l_type || F_WRLCK == fl.l_type)) { - struct fileglob *fg = fp->f_fglob; + struct fileglob *fg = fp->fp_glob; /* * arrange F_UNLCK on last close (once @@ -1093,7 +1320,7 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) break; } // XXX UInt32 unsafe for LP64 kernel - OSBitOrAtomic(P_LADVLOCK, &p->p_ladvflag); + os_atomic_or(&p->p_ladvflag, P_LADVLOCK, relaxed); error = VNOP_ADVLOCK(vp, (caddr_t)p, F_SETLK, &fl, flg, &context, timeout); break; @@ -1103,7 +1330,7 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) break; } // XXX UInt32 unsafe for LP64 kernel - OSBitOrAtomic(P_LADVLOCK, &p->p_ladvflag); + os_atomic_or(&p->p_ladvflag, P_LADVLOCK, relaxed); error = VNOP_ADVLOCK(vp, (caddr_t)p, F_SETLK, &fl, flg, &context, timeout); break; @@ -1122,6 +1349,8 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) case F_GETLK: case F_OFD_GETLK: + case F_GETLKPID: + case F_OFD_GETLKPID: if (fp->f_type != DTYPE_VNODE) { error = EBADF; goto out; @@ -1133,8 +1362,9 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) /* Copy in the lock structure */ error = copyin(argp, (caddr_t)&fl, sizeof(fl)); - if (error) + if (error) { goto outdrop; + } /* Check starting byte and ending byte for EOVERFLOW in SEEK_CUR */ /* and ending byte for EOVERFLOW in SEEK_SET */ @@ -1168,22 +1398,23 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) goto outdrop; } - if ( (error = vnode_getwithref(vp)) == 0 ) { - if (fl.l_whence == SEEK_CUR) - fl.l_start += offset; + if ((error = vnode_getwithref(vp)) == 0) { + if (fl.l_whence == SEEK_CUR) { + fl.l_start += offset; + } #if CONFIG_MACF - error = mac_file_check_lock(proc_ucred(p), fp->f_fglob, + error = mac_file_check_lock(proc_ucred(p), fp->fp_glob, uap->cmd, &fl); if (error == 0) #endif switch (uap->cmd) { case F_OFD_GETLK: - error = VNOP_ADVLOCK(vp, (caddr_t)fp->f_fglob, + error = VNOP_ADVLOCK(vp, (caddr_t)fp->fp_glob, F_GETLK, &fl, F_OFD_LOCK, &context, NULL); break; case F_OFD_GETLKPID: - error = VNOP_ADVLOCK(vp, (caddr_t)fp->f_fglob, + error = VNOP_ADVLOCK(vp, (caddr_t)fp->fp_glob, F_GETLKPID, &fl, F_OFD_LOCK, &context, NULL); break; default: @@ -1194,8 +1425,9 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) (void)vnode_put(vp); - if (error == 0) + if (error == 0) { error = copyout((caddr_t)&fl, argp, sizeof(fl)); + } } goto outdrop; @@ -1218,12 +1450,13 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) } error = copyin(argp, (caddr_t)&alloc_struct, sizeof(alloc_struct)); - if (error) + if (error) { goto outdrop; + } /* now set the space allocated to 0 */ alloc_struct.fst_bytesalloc = 0; - + /* * Do some simple parameter checking */ @@ -1231,12 +1464,14 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) /* set up the flags */ alloc_flags |= PREALLOCATE; - - if (alloc_struct.fst_flags & F_ALLOCATECONTIG) + + if (alloc_struct.fst_flags & F_ALLOCATECONTIG) { alloc_flags |= ALLOCATECONTIG; + } - if (alloc_struct.fst_flags & F_ALLOCATEALL) + if (alloc_struct.fst_flags & F_ALLOCATEALL) { alloc_flags |= ALLOCATEALL; + } /* * Do any position mode specific stuff. The only @@ -1244,7 +1479,6 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) */ switch (alloc_struct.fst_posmode) { - case F_PEOFPOSMODE: if (alloc_struct.fst_offset != 0) { error = EINVAL; @@ -1266,25 +1500,141 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) default: { error = EINVAL; goto outdrop; - } } - if ( (error = vnode_getwithref(vp)) == 0 ) { - /* + } + if ((error = vnode_getwithref(vp)) == 0) { + /* * call allocate to get the space */ - error = VNOP_ALLOCATE(vp,alloc_struct.fst_length,alloc_flags, - &alloc_struct.fst_bytesalloc, alloc_struct.fst_offset, - &context); + error = VNOP_ALLOCATE(vp, alloc_struct.fst_length, alloc_flags, + &alloc_struct.fst_bytesalloc, alloc_struct.fst_offset, + &context); (void)vnode_put(vp); error2 = copyout((caddr_t)&alloc_struct, argp, sizeof(alloc_struct)); - if (error == 0) + if (error == 0) { error = error2; + } + } + goto outdrop; + } + case F_PUNCHHOLE: { + fpunchhole_t args; + + if (fp->f_type != DTYPE_VNODE) { + error = EBADF; + goto out; + } + + vp = (struct vnode *)fp->f_data; + proc_fdunlock(p); + + /* need write permissions */ + if ((fp->f_flag & FWRITE) == 0) { + error = EPERM; + goto outdrop; + } + + if ((error = copyin(argp, (caddr_t)&args, sizeof(args)))) { + goto outdrop; + } + + if ((error = vnode_getwithref(vp))) { + goto outdrop; + } + +#if CONFIG_MACF + if ((error = mac_vnode_check_write(&context, fp->fp_glob->fg_cred, vp))) { + (void)vnode_put(vp); + goto outdrop; + } +#endif + + error = VNOP_IOCTL(vp, F_PUNCHHOLE, (caddr_t)&args, 0, &context); + (void)vnode_put(vp); + + goto outdrop; + } + case F_TRIM_ACTIVE_FILE: { + ftrimactivefile_t args; + + if (priv_check_cred(kauth_cred_get(), PRIV_TRIM_ACTIVE_FILE, 0)) { + error = EACCES; + goto out; + } + + if (fp->f_type != DTYPE_VNODE) { + error = EBADF; + goto out; + } + + vp = (struct vnode *)fp->f_data; + proc_fdunlock(p); + + /* need write permissions */ + if ((fp->f_flag & FWRITE) == 0) { + error = EPERM; + goto outdrop; + } + + if ((error = copyin(argp, (caddr_t)&args, sizeof(args)))) { + goto outdrop; + } + + if ((error = vnode_getwithref(vp))) { + goto outdrop; } + + error = VNOP_IOCTL(vp, F_TRIM_ACTIVE_FILE, (caddr_t)&args, 0, &context); + (void)vnode_put(vp); + goto outdrop; - + } + case F_SPECULATIVE_READ: { + fspecread_t args; + + if (fp->f_type != DTYPE_VNODE) { + error = EBADF; + goto out; + } + + vp = (struct vnode *)fp->f_data; + proc_fdunlock(p); + + if ((error = copyin(argp, (caddr_t)&args, sizeof(args)))) { + goto outdrop; + } + + /* Discard invalid offsets or lengths */ + if ((args.fsr_offset < 0) || (args.fsr_length < 0)) { + error = EINVAL; + goto outdrop; + } + + /* + * Round the file offset down to a page-size boundary (or to 0). + * The filesystem will need to round the length up to the end of the page boundary + * or to the EOF of the file. + */ + uint64_t foff = (((uint64_t)args.fsr_offset) & ~((uint64_t)PAGE_MASK)); + uint64_t foff_delta = args.fsr_offset - foff; + args.fsr_offset = (off_t) foff; + + /* + * Now add in the delta to the supplied length. Since we may have adjusted the + * offset, increase it by the amount that we adjusted. + */ + args.fsr_length += foff_delta; + + if ((error = vnode_getwithref(vp))) { + goto outdrop; } + error = VNOP_IOCTL(vp, F_SPECULATIVE_READ, (caddr_t)&args, 0, &context); + (void)vnode_put(vp); + + goto outdrop; + } case F_SETSIZE: if (fp->f_type != DTYPE_VNODE) { error = EBADF; @@ -1293,18 +1643,20 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) vp = (struct vnode *)fp->f_data; proc_fdunlock(p); - error = copyin(argp, (caddr_t)&offset, sizeof (off_t)); - if (error) + error = copyin(argp, (caddr_t)&offset, sizeof(off_t)); + if (error) { goto outdrop; + } AUDIT_ARG(value64, offset); error = vnode_getwithref(vp); - if (error) + if (error) { goto outdrop; + } #if CONFIG_MACF error = mac_vnode_check_truncate(&context, - fp->f_fglob->fg_cred, vp); + fp->fp_glob->fg_cred, vp); if (error) { (void)vnode_put(vp); goto outdrop; @@ -1312,17 +1664,27 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) #endif /* * Make sure that we are root. Growing a file - * without zero filling the data is a security hole - * root would have access anyway so we'll allow it + * without zero filling the data is a security hole. */ if (!kauth_cred_issuser(kauth_cred_get())) { error = EACCES; } else { /* - * set the file size + * Require privilege to change file size without zerofill, + * else will change the file size and zerofill it. */ - error = vnode_setsize(vp, offset, IO_NOZEROFILL, - &context); + error = priv_check_cred(kauth_cred_get(), PRIV_VFS_SETSIZE, 0); + if (error == 0) { + error = vnode_setsize(vp, offset, IO_NOZEROFILL, &context); + } else { + error = vnode_setsize(vp, offset, 0, &context); + } + +#if CONFIG_MACF + if (error == 0) { + mac_vnode_notify_truncate(&context, fp->fp_glob->fg_cred, vp); + } +#endif } (void)vnode_put(vp); @@ -1333,11 +1695,11 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) error = EBADF; goto out; } - if (uap->arg) - fp->f_fglob->fg_flag &= ~FNORDAHEAD; - else - fp->f_fglob->fg_flag |= FNORDAHEAD; - + if (uap->arg) { + os_atomic_andnot(&fp->fp_glob->fg_flag, FNORDAHEAD, relaxed); + } else { + os_atomic_or(&fp->fp_glob->fg_flag, FNORDAHEAD, relaxed); + } goto out; case F_NOCACHE: @@ -1345,11 +1707,11 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) error = EBADF; goto out; } - if (uap->arg) - fp->f_fglob->fg_flag |= FNOCACHE; - else - fp->f_fglob->fg_flag &= ~FNOCACHE; - + if (uap->arg) { + os_atomic_or(&fp->fp_glob->fg_flag, FNOCACHE, relaxed); + } else { + os_atomic_andnot(&fp->fp_glob->fg_flag, FNOCACHE, relaxed); + } goto out; case F_NODIRECT: @@ -1357,11 +1719,11 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) error = EBADF; goto out; } - if (uap->arg) - fp->f_fglob->fg_flag |= FNODIRECT; - else - fp->f_fglob->fg_flag &= ~FNODIRECT; - + if (uap->arg) { + os_atomic_or(&fp->fp_glob->fg_flag, FNODIRECT, relaxed); + } else { + os_atomic_andnot(&fp->fp_glob->fg_flag, FNODIRECT, relaxed); + } goto out; case F_SINGLE_WRITER: @@ -1369,50 +1731,50 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) error = EBADF; goto out; } - if (uap->arg) - fp->f_fglob->fg_flag |= FSINGLE_WRITER; - else - fp->f_fglob->fg_flag &= ~FSINGLE_WRITER; - + if (uap->arg) { + os_atomic_or(&fp->fp_glob->fg_flag, FSINGLE_WRITER, relaxed); + } else { + os_atomic_andnot(&fp->fp_glob->fg_flag, FSINGLE_WRITER, relaxed); + } goto out; case F_GLOBAL_NOCACHE: - if (fp->f_type != DTYPE_VNODE) { - error = EBADF; + if (fp->f_type != DTYPE_VNODE) { + error = EBADF; goto out; } vp = (struct vnode *)fp->f_data; proc_fdunlock(p); - if ( (error = vnode_getwithref(vp)) == 0 ) { + if ((error = vnode_getwithref(vp)) == 0) { + *retval = vnode_isnocache(vp); - *retval = vnode_isnocache(vp); - - if (uap->arg) - vnode_setnocache(vp); - else - vnode_clearnocache(vp); + if (uap->arg) { + vnode_setnocache(vp); + } else { + vnode_clearnocache(vp); + } (void)vnode_put(vp); } goto outdrop; case F_CHECK_OPENEVT: - if (fp->f_type != DTYPE_VNODE) { - error = EBADF; + if (fp->f_type != DTYPE_VNODE) { + error = EBADF; goto out; } vp = (struct vnode *)fp->f_data; proc_fdunlock(p); - if ( (error = vnode_getwithref(vp)) == 0 ) { - - *retval = vnode_is_openevt(vp); + if ((error = vnode_getwithref(vp)) == 0) { + *retval = vnode_is_openevt(vp); - if (uap->arg) - vnode_set_openevt(vp); - else - vnode_clear_openevt(vp); + if (uap->arg) { + vnode_set_openevt(vp); + } else { + vnode_clear_openevt(vp); + } (void)vnode_put(vp); } @@ -1428,35 +1790,40 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) vp = (struct vnode *)fp->f_data; proc_fdunlock(p); - if ( (error = copyin(argp, (caddr_t)&ra_struct, sizeof(ra_struct))) ) + if ((error = copyin(argp, (caddr_t)&ra_struct, sizeof(ra_struct)))) { + goto outdrop; + } + if (ra_struct.ra_offset < 0 || ra_struct.ra_count < 0) { + error = EINVAL; goto outdrop; - if ( (error = vnode_getwithref(vp)) == 0 ) { - error = VNOP_IOCTL(vp, F_RDADVISE, (caddr_t)&ra_struct, 0, &context); + } + if ((error = vnode_getwithref(vp)) == 0) { + error = VNOP_IOCTL(vp, F_RDADVISE, (caddr_t)&ra_struct, 0, &context); (void)vnode_put(vp); } goto outdrop; - } + } - case F_FLUSH_DATA: + case F_FLUSH_DATA: - if (fp->f_type != DTYPE_VNODE) { - error = EBADF; - goto out; - } - vp = (struct vnode *)fp->f_data; - proc_fdunlock(p); + if (fp->f_type != DTYPE_VNODE) { + error = EBADF; + goto out; + } + vp = (struct vnode *)fp->f_data; + proc_fdunlock(p); - if ( (error = vnode_getwithref(vp)) == 0 ) { - error = cluster_push(vp, 0); + if ((error = vnode_getwithref(vp)) == 0) { + error = VNOP_FSYNC(vp, MNT_NOWAIT, &context); - (void)vnode_put(vp); - } - goto outdrop; + (void)vnode_put(vp); + } + goto outdrop; case F_LOG2PHYS: case F_LOG2PHYS_EXT: { - struct log2phys l2p_struct; /* structure for allocate command */ + struct log2phys l2p_struct = {}; /* structure for allocate command */ int devBlockSize; off_t file_offset = 0; @@ -1465,8 +1832,9 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) if (uap->cmd == F_LOG2PHYS_EXT) { error = copyin(argp, (caddr_t)&l2p_struct, sizeof(l2p_struct)); - if (error) + if (error) { goto out; + } file_offset = l2p_struct.l2p_devoffset; } else { file_offset = fp->f_offset; @@ -1477,7 +1845,7 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) } vp = (struct vnode *)fp->f_data; proc_fdunlock(p); - if ( (error = vnode_getwithref(vp)) ) { + if ((error = vnode_getwithref(vp))) { goto outdrop; } error = VNOP_OFFTOBLK(vp, file_offset, &lbn); @@ -1498,21 +1866,21 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) goto outdrop; } - a_size = MIN((uint64_t)l2p_struct.l2p_contigbytes, SIZE_MAX); + a_size = (size_t)MIN((uint64_t)l2p_struct.l2p_contigbytes, SIZE_MAX); } else { a_size = devBlockSize; } - + error = VNOP_BLOCKMAP(vp, offset, a_size, &bn, &run, NULL, 0, &context); (void)vnode_put(vp); if (!error) { - l2p_struct.l2p_flags = 0; /* for now */ + l2p_struct.l2p_flags = 0; /* for now */ if (uap->cmd == F_LOG2PHYS_EXT) { l2p_struct.l2p_contigbytes = run - (file_offset - offset); } else { - l2p_struct.l2p_contigbytes = 0; /* for now */ + l2p_struct.l2p_contigbytes = 0; /* for now */ } /* @@ -1522,16 +1890,16 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) if (bn == -1) { /* Don't multiply it by the block size */ l2p_struct.l2p_devoffset = bn; - } - else { + } else { l2p_struct.l2p_devoffset = bn * devBlockSize; l2p_struct.l2p_devoffset += file_offset - offset; } error = copyout((caddr_t)&l2p_struct, argp, sizeof(l2p_struct)); } goto outdrop; - } - case F_GETPATH: { + } + case F_GETPATH: + case F_GETPATH_NOFIRMLINK: { char *pathbufp; int pathlen; @@ -1543,19 +1911,21 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) proc_fdunlock(p); pathlen = MAXPATHLEN; - MALLOC(pathbufp, char *, pathlen, M_TEMP, M_WAITOK); - if (pathbufp == NULL) { - error = ENOMEM; - goto outdrop; - } - if ( (error = vnode_getwithref(vp)) == 0 ) { - error = vn_getpath(vp, pathbufp, &pathlen); - (void)vnode_put(vp); + pathbufp = zalloc(ZV_NAMEI); - if (error == 0) - error = copyout((caddr_t)pathbufp, argp, pathlen); + if ((error = vnode_getwithref(vp)) == 0) { + if (uap->cmd == F_GETPATH_NOFIRMLINK) { + error = vn_getpath_ext(vp, NULL, pathbufp, &pathlen, VN_GETPATH_NO_FIRMLINK); + } else { + error = vn_getpath(vp, pathbufp, &pathlen); + } + (void)vnode_put(vp); + + if (error == 0) { + error = copyout((caddr_t)pathbufp, argp, pathlen); + } } - FREE(pathbufp, M_TEMP); + zfree(ZV_NAMEI, pathbufp); goto outdrop; } @@ -1564,24 +1934,24 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) size_t pathlen; if (fp->f_type != DTYPE_VNODE) { - error = EBADF; + error = EBADF; goto out; } vp = (struct vnode *)fp->f_data; proc_fdunlock(p); pathlen = MAXPATHLEN; - pathbufp = kalloc(MAXPATHLEN); + pathbufp = zalloc(ZV_NAMEI); - if ( (error = copyinstr(argp, pathbufp, MAXPATHLEN, &pathlen)) == 0 ) { - if ( (error = vnode_getwithref(vp)) == 0 ) { + if ((error = copyinstr(argp, pathbufp, MAXPATHLEN, &pathlen)) == 0) { + if ((error = vnode_getwithref(vp)) == 0) { AUDIT_ARG(text, pathbufp); - error = vn_path_package_check(vp, pathbufp, pathlen, retval); + error = vn_path_package_check(vp, pathbufp, (int)pathlen, retval); (void)vnode_put(vp); } } - kfree(pathbufp, MAXPATHLEN); + zfree(ZV_NAMEI, pathbufp); goto outdrop; } @@ -1597,8 +1967,8 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) vp = (struct vnode *)fp->f_data; proc_fdunlock(p); - if ( (error = vnode_getwithref(vp)) == 0 ) { - error = VNOP_IOCTL(vp, uap->cmd, (caddr_t)NULL, 0, &context); + if ((error = vnode_getwithref(vp)) == 0) { + error = VNOP_IOCTL(vp, uap->cmd, (caddr_t)NULL, 0, &context); (void)vnode_put(vp); } @@ -1627,7 +1997,7 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) error = ENOENT; goto outdrop; } - + /* Only valid for directories */ if (vp->v_type != VDIR) { vnode_put(vp); @@ -1635,6 +2005,15 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) goto outdrop; } + /* + * Only entitled apps may use the credentials of the thread + * that opened the file descriptor. + * Non-entitled threads will use their own context. + */ + if (IOTaskHasEntitlement(current_task(), ACCOUNT_OPENFROM_ENTITLEMENT)) { + has_entitlement = 1; + } + /* Get flags, mode and pathname arguments. */ if (IS_64BIT_PROCESS(p)) { error = copyin(argp, &fopen, sizeof(fopen)); @@ -1654,16 +2033,16 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) AUDIT_ARG(mode, fopen.o_mode); VATTR_INIT(&va); /* Mask off all but regular access permissions */ - cmode = ((fopen.o_mode &~ fdp->fd_cmask) & ALLPERMS) & ~S_ISTXT; + cmode = ((fopen.o_mode & ~fdp->fd_cmask) & ALLPERMS) & ~S_ISTXT; VATTR_SET(&va, va_mode, cmode & ACCESSPERMS); /* Start the lookup relative to the file descriptor's vnode. */ NDINIT(&nd, LOOKUP, OP_OPEN, USEDVP | FOLLOW | AUDITVNPATH1, UIO_USERSPACE, - fopen.o_pathname, &context); + fopen.o_pathname, has_entitlement ? &context : vfs_context_current()); nd.ni_dvp = vp; - error = open1(&context, &nd, fopen.o_flags, &va, - fileproc_alloc_init, NULL, retval); + error = open1(has_entitlement ? &context : vfs_context_current(), + &nd, fopen.o_flags, &va, fileproc_alloc_init, NULL, retval); vnode_put(vp); break; @@ -1687,7 +2066,7 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) error = ENOENT; goto outdrop; } - + /* Only valid for directories */ if (vp->v_type != VDIR) { vnode_put(vp); @@ -1695,6 +2074,15 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) goto outdrop; } + /* + * Only entitled apps may use the credentials of the thread + * that opened the file descriptor. + * Non-entitled threads will use their own context. + */ + if (IOTaskHasEntitlement(current_task(), ACCOUNT_OPENFROM_ENTITLEMENT)) { + has_entitlement = 1; + } + /* Get flags, mode and pathname arguments. */ if (IS_64BIT_PROCESS(p)) { pathname = (user_addr_t)argp; @@ -1703,17 +2091,18 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) } /* Start the lookup relative to the file descriptor's vnode. */ - error = unlink1(&context, vp, pathname, UIO_USERSPACE, 0); - + error = unlink1(has_entitlement ? &context : vfs_context_current(), + vp, pathname, UIO_USERSPACE, 0); + vnode_put(vp); break; - } case F_ADDSIGS: case F_ADDFILESIGS: case F_ADDFILESIGS_FOR_DYLD_SIM: case F_ADDFILESIGS_RETURN: + case F_ADDFILESIGS_INFO: { struct cs_blob *blob = NULL; struct user_fsignatures fs; @@ -1721,6 +2110,9 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) vm_offset_t kernel_blob_addr; vm_size_t kernel_blob_size; int blob_add_flags = 0; + const size_t sizeof_fs = (uap->cmd == F_ADDFILESIGS_INFO ? + offsetof(struct user_fsignatures, fs_cdhash /* first output element */) : + offsetof(struct user_fsignatures, fs_fsignatures_size /* compat */)); if (fp->f_type != DTYPE_VNODE) { error = EBADF; @@ -1739,15 +2131,22 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) } error = vnode_getwithref(vp); - if (error) + if (error) { goto outdrop; + } if (IS_64BIT_PROCESS(p)) { - error = copyin(argp, &fs, sizeof (fs)); + error = copyin(argp, &fs, sizeof_fs); } else { + if (uap->cmd == F_ADDFILESIGS_INFO) { + error = EINVAL; + vnode_put(vp); + goto outdrop; + } + struct user32_fsignatures fs32; - error = copyin(argp, &fs32, sizeof (fs32)); + error = copyin(argp, &fs32, sizeof(fs32)); fs.fs_file_start = fs32.fs_file_start; fs.fs_blob_start = CAST_USER_ADDR_T(fs32.fs_blob_start); fs.fs_blob_size = fs32.fs_blob_size; @@ -1761,15 +2160,22 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) /* * First check if we have something loaded a this offset */ - blob = ubc_cs_blob_get(vp, CPU_TYPE_ANY, fs.fs_file_start); - if (blob != NULL) - { + blob = ubc_cs_blob_get(vp, CPU_TYPE_ANY, CPU_SUBTYPE_ANY, fs.fs_file_start); + if (blob != NULL) { /* If this is for dyld_sim revalidate the blob */ if (uap->cmd == F_ADDFILESIGS_FOR_DYLD_SIM) { - error = ubc_cs_blob_revalidate(vp, blob, blob_add_flags); + error = ubc_cs_blob_revalidate(vp, blob, NULL, blob_add_flags, proc_platform(p)); + if (error) { + blob = NULL; + if (error != EAGAIN) { + vnode_put(vp); + goto outdrop; + } + } } + } - } else { + if (blob == NULL) { /* * An arbitrary limit, to prevent someone from mapping in a 20GB blob. This should cover * our use cases for the immediate future, but note that at the time of this commit, some @@ -1788,210 +2194,445 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) kernel_blob_size = CAST_DOWN(vm_size_t, fs.fs_blob_size); kr = ubc_cs_blob_allocate(&kernel_blob_addr, &kernel_blob_size); - if (kr != KERN_SUCCESS) { + if (kr != KERN_SUCCESS || kernel_blob_size < fs.fs_blob_size) { error = ENOMEM; vnode_put(vp); goto outdrop; } - if(uap->cmd == F_ADDSIGS) { + if (uap->cmd == F_ADDSIGS) { error = copyin(fs.fs_blob_start, - (void *) kernel_blob_addr, - kernel_blob_size); - } else /* F_ADDFILESIGS || F_ADDFILESIGS_RETURN || F_ADDFILESIGS_FOR_DYLD_SIM */ { + (void *) kernel_blob_addr, + fs.fs_blob_size); + } else { /* F_ADDFILESIGS || F_ADDFILESIGS_RETURN || F_ADDFILESIGS_FOR_DYLD_SIM || F_ADDFILESIGS_INFO */ int resid; error = vn_rdwr(UIO_READ, - vp, - (caddr_t) kernel_blob_addr, - kernel_blob_size, - fs.fs_file_start + fs.fs_blob_start, - UIO_SYSSPACE, - 0, - kauth_cred_get(), - &resid, - p); + vp, + (caddr_t) kernel_blob_addr, + (int)kernel_blob_size, + fs.fs_file_start + fs.fs_blob_start, + UIO_SYSSPACE, + 0, + kauth_cred_get(), + &resid, + p); if ((error == 0) && resid) { /* kernel_blob_size rounded to a page size, but signature may be at end of file */ memset((void *)(kernel_blob_addr + (kernel_blob_size - resid)), 0x0, resid); } } - + if (error) { ubc_cs_blob_deallocate(kernel_blob_addr, - kernel_blob_size); + kernel_blob_size); vnode_put(vp); goto outdrop; } blob = NULL; error = ubc_cs_blob_add(vp, - CPU_TYPE_ANY, /* not for a specific architecture */ - fs.fs_file_start, - kernel_blob_addr, - kernel_blob_size, - blob_add_flags, - &blob); + proc_platform(p), + CPU_TYPE_ANY, /* not for a specific architecture */ + CPU_SUBTYPE_ANY, + fs.fs_file_start, + &kernel_blob_addr, + kernel_blob_size, + NULL, + blob_add_flags, + &blob); + + /* ubc_blob_add() has consumed "kernel_blob_addr" if it is zeroed */ if (error) { - ubc_cs_blob_deallocate(kernel_blob_addr, - kernel_blob_size); + if (kernel_blob_addr) { + ubc_cs_blob_deallocate(kernel_blob_addr, + kernel_blob_size); + } + vnode_put(vp); + goto outdrop; } else { - /* ubc_blob_add() has consumed "kernel_blob_addr" */ #if CHECK_CS_VALIDATION_BITMAP ubc_cs_validation_bitmap_allocate( vp ); #endif } } - if (uap->cmd == F_ADDFILESIGS_RETURN || uap->cmd == F_ADDFILESIGS_FOR_DYLD_SIM) { + if (uap->cmd == F_ADDFILESIGS_RETURN || uap->cmd == F_ADDFILESIGS_FOR_DYLD_SIM || + uap->cmd == F_ADDFILESIGS_INFO) { /* * The first element of the structure is a * off_t that happen to have the same size for * all archs. Lets overwrite that. */ off_t end_offset = 0; - if (blob) + if (blob) { end_offset = blob->csb_end_offset; - error = copyout(&end_offset, argp, sizeof (end_offset)); + } + error = copyout(&end_offset, argp, sizeof(end_offset)); + + if (error) { + vnode_put(vp); + goto outdrop; + } + } + + if (uap->cmd == F_ADDFILESIGS_INFO) { + /* Return information. What we copy out depends on the size of the + * passed in structure, to keep binary compatibility. */ + + if (fs.fs_fsignatures_size >= sizeof(struct user_fsignatures)) { + // enough room for fs_cdhash[20]+fs_hash_type + + if (blob != NULL) { + error = copyout(blob->csb_cdhash, + (vm_address_t)argp + offsetof(struct user_fsignatures, fs_cdhash), + USER_FSIGNATURES_CDHASH_LEN); + if (error) { + vnode_put(vp); + goto outdrop; + } + int hashtype = cs_hash_type(blob->csb_hashtype); + error = copyout(&hashtype, + (vm_address_t)argp + offsetof(struct user_fsignatures, fs_hash_type), + sizeof(int)); + if (error) { + vnode_put(vp); + goto outdrop; + } + } + } } (void) vnode_put(vp); break; } - case F_FINDSIGS: { - error = ENOTSUP; - goto out; - } -#if CONFIG_PROTECT - case F_GETPROTECTIONCLASS: { - int class = 0; - +#if CONFIG_SUPPLEMENTAL_SIGNATURES + case F_ADDFILESUPPL: + { + struct vnode *ivp; + struct cs_blob *blob = NULL; + struct user_fsupplement fs; + int orig_fd; + struct fileproc* orig_fp = NULL; + kern_return_t kr; + vm_offset_t kernel_blob_addr; + vm_size_t kernel_blob_size; + + if (!IS_64BIT_PROCESS(p)) { + error = EINVAL; + goto out; // drop fp and unlock fds + } + if (fp->f_type != DTYPE_VNODE) { error = EBADF; goto out; } - vp = (struct vnode *)fp->f_data; - - proc_fdunlock(p); - if (vnode_getwithref(vp)) { - error = ENOENT; - goto outdrop; + error = copyin(argp, &fs, sizeof(fs)); + if (error) { + goto out; } - - error = cp_vnode_getclass (vp, &class); - if (error == 0) { - *retval = class; + + orig_fd = fs.fs_orig_fd; + if ((error = fp_lookup(p, orig_fd, &orig_fp, 1))) { + printf("CODE SIGNING: Failed to find original file for supplemental signature attachment\n"); + goto out; } - vnode_put(vp); - break; - } - - case F_SETPROTECTIONCLASS: { - /* tmp must be a valid PROTECTION_CLASS_* */ - tmp = CAST_DOWN_EXPLICIT(uint32_t, uap->arg); - - if (fp->f_type != DTYPE_VNODE) { + if (orig_fp->f_type != DTYPE_VNODE) { error = EBADF; + fp_drop(p, orig_fd, orig_fp, 1); goto out; } + + ivp = (struct vnode *)orig_fp->f_data; + vp = (struct vnode *)fp->f_data; proc_fdunlock(p); - + + error = vnode_getwithref(ivp); + if (error) { + fp_drop(p, orig_fd, orig_fp, 0); + goto outdrop; //drop fp + } + + error = vnode_getwithref(vp); + if (error) { + vnode_put(ivp); + fp_drop(p, orig_fd, orig_fp, 0); + goto outdrop; + } + + if (fs.fs_blob_size > CS_MAX_BLOB_SIZE) { + error = E2BIG; + goto dropboth; // drop iocounts on vp and ivp, drop orig_fp then drop fp via outdrop + } + + kernel_blob_size = CAST_DOWN(vm_size_t, fs.fs_blob_size); + kr = ubc_cs_blob_allocate(&kernel_blob_addr, &kernel_blob_size); + if (kr != KERN_SUCCESS) { + error = ENOMEM; + goto dropboth; + } + + int resid; + error = vn_rdwr(UIO_READ, vp, + (caddr_t)kernel_blob_addr, (int)kernel_blob_size, + fs.fs_file_start + fs.fs_blob_start, + UIO_SYSSPACE, 0, + kauth_cred_get(), &resid, p); + if ((error == 0) && resid) { + /* kernel_blob_size rounded to a page size, but signature may be at end of file */ + memset((void *)(kernel_blob_addr + (kernel_blob_size - resid)), 0x0, resid); + } + + if (error) { + ubc_cs_blob_deallocate(kernel_blob_addr, + kernel_blob_size); + goto dropboth; + } + + error = ubc_cs_blob_add_supplement(vp, ivp, fs.fs_file_start, + &kernel_blob_addr, kernel_blob_size, &blob); + + /* ubc_blob_add_supplement() has consumed kernel_blob_addr if it is zeroed */ + if (error) { + if (kernel_blob_addr) { + ubc_cs_blob_deallocate(kernel_blob_addr, + kernel_blob_size); + } + goto dropboth; + } + vnode_put(ivp); + vnode_put(vp); + fp_drop(p, orig_fd, orig_fp, 0); + break; + +dropboth: + vnode_put(ivp); + vnode_put(vp); + fp_drop(p, orig_fd, orig_fp, 0); + goto outdrop; + } +#endif + case F_GETCODEDIR: + case F_FINDSIGS: { + error = ENOTSUP; + goto out; + } + case F_CHECK_LV: { + struct fileglob *fg; + fchecklv_t lv = {}; + + if (fp->f_type != DTYPE_VNODE) { + error = EBADF; + goto out; + } + fg = fp->fp_glob; + proc_fdunlock(p); + + if (IS_64BIT_PROCESS(p)) { + error = copyin(argp, &lv, sizeof(lv)); + } else { + struct user32_fchecklv lv32 = {}; + + error = copyin(argp, &lv32, sizeof(lv32)); + lv.lv_file_start = lv32.lv_file_start; + lv.lv_error_message = (void *)(uintptr_t)lv32.lv_error_message; + lv.lv_error_message_size = lv32.lv_error_message_size; + } + if (error) { + goto outdrop; + } + +#if CONFIG_MACF + error = mac_file_check_library_validation(p, fg, lv.lv_file_start, + (user_long_t)lv.lv_error_message, lv.lv_error_message_size); +#endif + + break; + } + case F_GETSIGSINFO: { + struct cs_blob *blob = NULL; + fgetsigsinfo_t sigsinfo = {}; + + if (fp->f_type != DTYPE_VNODE) { + error = EBADF; + goto out; + } + vp = (struct vnode *)fp->f_data; + proc_fdunlock(p); + + error = vnode_getwithref(vp); + if (error) { + goto outdrop; + } + + error = copyin(argp, &sigsinfo, sizeof(sigsinfo)); + if (error) { + vnode_put(vp); + goto outdrop; + } + + blob = ubc_cs_blob_get(vp, CPU_TYPE_ANY, CPU_SUBTYPE_ANY, sigsinfo.fg_file_start); + if (blob == NULL) { + error = ENOENT; + vnode_put(vp); + goto outdrop; + } + switch (sigsinfo.fg_info_request) { + case GETSIGSINFO_PLATFORM_BINARY: + sigsinfo.fg_sig_is_platform = blob->csb_platform_binary; + error = copyout(&sigsinfo.fg_sig_is_platform, + (vm_address_t)argp + offsetof(struct fgetsigsinfo, fg_sig_is_platform), + sizeof(sigsinfo.fg_sig_is_platform)); + if (error) { + vnode_put(vp); + goto outdrop; + } + break; + default: + error = EINVAL; + vnode_put(vp); + goto outdrop; + } + vnode_put(vp); + break; + } +#if CONFIG_PROTECT + case F_GETPROTECTIONCLASS: { + if (fp->f_type != DTYPE_VNODE) { + error = EBADF; + goto out; + } + vp = (struct vnode *)fp->f_data; + + proc_fdunlock(p); + + if (vnode_getwithref(vp)) { + error = ENOENT; + goto outdrop; + } + + struct vnode_attr va; + + VATTR_INIT(&va); + VATTR_WANTED(&va, va_dataprotect_class); + error = VNOP_GETATTR(vp, &va, &context); + if (!error) { + if (VATTR_IS_SUPPORTED(&va, va_dataprotect_class)) { + *retval = va.va_dataprotect_class; + } else { + error = ENOTSUP; + } + } + + vnode_put(vp); + break; + } + + case F_SETPROTECTIONCLASS: { + /* tmp must be a valid PROTECTION_CLASS_* */ + tmp = CAST_DOWN_EXPLICIT(uint32_t, uap->arg); + + if (fp->f_type != DTYPE_VNODE) { + error = EBADF; + goto out; + } + vp = (struct vnode *)fp->f_data; + + proc_fdunlock(p); + if (vnode_getwithref(vp)) { error = ENOENT; goto outdrop; - } - + } + /* Only go forward if you have write access */ vfs_context_t ctx = vfs_context_current(); - if(vnode_authorize(vp, NULLVP, (KAUTH_VNODE_ACCESS | KAUTH_VNODE_WRITE_DATA), ctx) != 0) { + if (vnode_authorize(vp, NULLVP, (KAUTH_VNODE_ACCESS | KAUTH_VNODE_WRITE_DATA), ctx) != 0) { vnode_put(vp); error = EBADF; goto outdrop; } - error = cp_vnode_setclass (vp, tmp); + + struct vnode_attr va; + + VATTR_INIT(&va); + VATTR_SET(&va, va_dataprotect_class, tmp); + + error = VNOP_SETATTR(vp, &va, ctx); + vnode_put(vp); break; - } + } case F_TRANSCODEKEY: { - - char *backup_keyp = NULL; - unsigned backup_key_len = CP_MAX_WRAPPEDKEYSIZE; - if (fp->f_type != DTYPE_VNODE) { error = EBADF; goto out; } - + vp = (struct vnode *)fp->f_data; proc_fdunlock(p); if (vnode_getwithref(vp)) { error = ENOENT; goto outdrop; - } + } - MALLOC(backup_keyp, char *, backup_key_len, M_TEMP, M_WAITOK); - if (backup_keyp == NULL) { + cp_key_t k = { + .len = CP_MAX_WRAPPEDKEYSIZE, + }; + + k.key = kheap_alloc(KHEAP_TEMP, CP_MAX_WRAPPEDKEYSIZE, Z_WAITOK | Z_ZERO); + if (k.key == NULL) { error = ENOMEM; - goto outdrop; + } else { + error = VNOP_IOCTL(vp, F_TRANSCODEKEY, (caddr_t)&k, 1, &context); } - error = cp_vnode_transcode (vp, backup_keyp, &backup_key_len); vnode_put(vp); if (error == 0) { - error = copyout((caddr_t)backup_keyp, argp, backup_key_len); - *retval = backup_key_len; + error = copyout(k.key, argp, k.len); + *retval = k.len; } - FREE(backup_keyp, M_TEMP); + kheap_free(KHEAP_TEMP, k.key, CP_MAX_WRAPPEDKEYSIZE); break; - } + } case F_GETPROTECTIONLEVEL: { - uint32_t cp_version = 0; - if (fp->f_type != DTYPE_VNODE) { - error = EBADF; + error = EBADF; goto out; } vp = (struct vnode*) fp->f_data; - proc_fdunlock (p); + proc_fdunlock(p); if (vnode_getwithref(vp)) { error = ENOENT; goto outdrop; } - /* - * if cp_get_major_vers fails, error will be set to proper errno - * and cp_version will still be 0. - */ - - error = cp_get_root_major_vers (vp, &cp_version); - *retval = cp_version; + error = VNOP_IOCTL(vp, F_GETPROTECTIONLEVEL, (caddr_t)retval, 0, &context); - vnode_put (vp); + vnode_put(vp); break; } case F_GETDEFAULTPROTLEVEL: { - uint32_t cp_default = 0; - if (fp->f_type != DTYPE_VNODE) { - error = EBADF; + error = EBADF; goto out; } vp = (struct vnode*) fp->f_data; - proc_fdunlock (p); + proc_fdunlock(p); if (vnode_getwithref(vp)) { error = ENOENT; @@ -1999,18 +2640,16 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) } /* - * if cp_get_major_vers fails, error will be set to proper errno + * if cp_get_major_vers fails, error will be set to proper errno * and cp_version will still be 0. */ - error = cp_get_default_level(vp, &cp_default); - *retval = cp_default; + error = VNOP_IOCTL(vp, F_GETDEFAULTPROTLEVEL, (caddr_t)retval, 0, &context); - vnode_put (vp); + vnode_put(vp); break; } - #endif /* CONFIG_PROTECT */ case F_MOVEDATAEXTENTS: { @@ -2021,26 +2660,30 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) int fd2 = CAST_DOWN_EXPLICIT(int32_t, uap->arg); error = priv_check_cred(kauth_cred_get(), PRIV_VFS_MOVE_DATA_EXTENTS, 0); - if (error) + if (error) { goto out; + } if (fp->f_type != DTYPE_VNODE) { error = EBADF; goto out; } - /* For now, special case HFS+ only, since this is SPI. */ + /* + * For now, special case HFS+ and APFS only, since this + * is SPI. + */ src_vp = (struct vnode *)fp->f_data; - if (src_vp->v_tag != VT_HFS) { - error = EINVAL; + if (src_vp->v_tag != VT_HFS && src_vp->v_tag != VT_APFS) { + error = ENOTSUP; goto out; } /* - * Get the references before we start acquiring iocounts on the vnodes, + * Get the references before we start acquiring iocounts on the vnodes, * while we still hold the proc fd lock */ - if ( (error = fp_lookup(p, fd2, &fp2, 1)) ) { + if ((error = fp_lookup(p, fd2, &fp2, 1))) { error = EBADF; goto out; } @@ -2050,15 +2693,15 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) goto out; } dst_vp = (struct vnode *)fp2->f_data; - if (dst_vp->v_tag != VT_HFS) { + if (dst_vp->v_tag != VT_HFS && dst_vp->v_tag != VT_APFS) { fp_drop(p, fd2, fp2, 1); - error = EINVAL; + error = ENOTSUP; goto out; } #if CONFIG_MACF /* Re-do MAC checks against the new FD, pass in a fake argument */ - error = mac_file_check_fcntl(proc_ucred(p), fp2->f_fglob, uap->cmd, 0); + error = mac_file_check_fcntl(proc_ucred(p), fp2->fp_glob, uap->cmd, 0); if (error) { fp_drop(p, fd2, fp2, 1); goto out; @@ -2073,30 +2716,30 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) fp_drop(p, fd2, fp2, 0); error = ENOENT; goto outdrop; - } + } if (vnode_getwithref(dst_vp)) { - vnode_put (src_vp); + vnode_put(src_vp); fp_drop(p, fd2, fp2, 0); error = ENOENT; goto outdrop; - } - - /* + } + + /* * Basic asserts; validate they are not the same and that * both live on the same filesystem. */ if (dst_vp == src_vp) { - vnode_put (src_vp); - vnode_put (dst_vp); - fp_drop (p, fd2, fp2, 0); + vnode_put(src_vp); + vnode_put(dst_vp); + fp_drop(p, fd2, fp2, 0); error = EINVAL; goto outdrop; - } + } if (dst_vp->v_mount != src_vp->v_mount) { - vnode_put (src_vp); - vnode_put (dst_vp); - fp_drop (p, fd2, fp2, 0); + vnode_put(src_vp); + vnode_put(dst_vp); + fp_drop(p, fd2, fp2, 0); error = EXDEV; goto outdrop; } @@ -2104,59 +2747,59 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) /* Now we have a legit pair of FDs. Go to work */ /* Now check for write access to the target files */ - if(vnode_authorize(src_vp, NULLVP, - (KAUTH_VNODE_ACCESS | KAUTH_VNODE_WRITE_DATA), &context) != 0) { + if (vnode_authorize(src_vp, NULLVP, + (KAUTH_VNODE_ACCESS | KAUTH_VNODE_WRITE_DATA), &context) != 0) { vnode_put(src_vp); vnode_put(dst_vp); fp_drop(p, fd2, fp2, 0); error = EBADF; goto outdrop; } - - if(vnode_authorize(dst_vp, NULLVP, - (KAUTH_VNODE_ACCESS | KAUTH_VNODE_WRITE_DATA), &context) != 0) { + + if (vnode_authorize(dst_vp, NULLVP, + (KAUTH_VNODE_ACCESS | KAUTH_VNODE_WRITE_DATA), &context) != 0) { vnode_put(src_vp); vnode_put(dst_vp); fp_drop(p, fd2, fp2, 0); error = EBADF; goto outdrop; } - + /* Verify that both vps point to files and not directories */ - if ( !vnode_isreg(src_vp) || !vnode_isreg(dst_vp)) { + if (!vnode_isreg(src_vp) || !vnode_isreg(dst_vp)) { error = EINVAL; - vnode_put (src_vp); - vnode_put (dst_vp); - fp_drop (p, fd2, fp2, 0); + vnode_put(src_vp); + vnode_put(dst_vp); + fp_drop(p, fd2, fp2, 0); goto outdrop; } - /* + /* * The exchangedata syscall handler passes in 0 for the flags to VNOP_EXCHANGE. * We'll pass in our special bit indicating that the new behavior is expected */ - + error = VNOP_EXCHANGE(src_vp, dst_vp, FSOPT_EXCHANGE_DATA_ONLY, &context); - - vnode_put (src_vp); - vnode_put (dst_vp); + + vnode_put(src_vp); + vnode_put(dst_vp); fp_drop(p, fd2, fp2, 0); break; } - - /* + + /* * SPI for making a file compressed. */ case F_MAKECOMPRESSED: { uint32_t gcounter = CAST_DOWN_EXPLICIT(uint32_t, uap->arg); if (fp->f_type != DTYPE_VNODE) { - error = EBADF; + error = EBADF; goto out; } vp = (struct vnode*) fp->f_data; - proc_fdunlock (p); + proc_fdunlock(p); /* get the vnode */ if (vnode_getwithref(vp)) { @@ -2172,9 +2815,9 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) } /* invoke ioctl to pass off to FS */ - /* Only go forward if you have write access */ + /* Only go forward if you have write access */ vfs_context_t ctx = vfs_context_current(); - if(vnode_authorize(vp, NULLVP, (KAUTH_VNODE_ACCESS | KAUTH_VNODE_WRITE_DATA), ctx) != 0) { + if (vnode_authorize(vp, NULLVP, (KAUTH_VNODE_ACCESS | KAUTH_VNODE_WRITE_DATA), ctx) != 0) { vnode_put(vp); error = EBADF; goto outdrop; @@ -2182,18 +2825,18 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) error = VNOP_IOCTL(vp, uap->cmd, (caddr_t)&gcounter, 0, &context); - vnode_put (vp); - break; + vnode_put(vp); + break; } - + /* * SPI (private) for indicating to a filesystem that subsequent writes to * the open FD will written to the Fastflow. */ case F_SET_GREEDY_MODE: - /* intentionally drop through to the same handler as F_SETSTATIC. - * both fcntls should pass the argument and their selector into VNOP_IOCTL. - */ + /* intentionally drop through to the same handler as F_SETSTATIC. + * both fcntls should pass the argument and their selector into VNOP_IOCTL. + */ /* * SPI (private) for indicating to a filesystem that subsequent writes to @@ -2221,7 +2864,7 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) /* Only go forward if you have write access */ vfs_context_t ctx = vfs_context_current(); - if(vnode_authorize(vp, NULLVP, (KAUTH_VNODE_ACCESS | KAUTH_VNODE_WRITE_DATA), ctx) != 0) { + if (vnode_authorize(vp, NULLVP, (KAUTH_VNODE_ACCESS | KAUTH_VNODE_WRITE_DATA), ctx) != 0) { vnode_put(vp); error = EBADF; goto outdrop; @@ -2229,7 +2872,7 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) error = VNOP_IOCTL(vp, uap->cmd, ioctl_arg, 0, &context); (void)vnode_put(vp); - + break; } @@ -2239,31 +2882,30 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) * or other flavors that may be necessary. */ case F_SETIOTYPE: { - caddr_t param_ptr; + caddr_t param_ptr; uint32_t param; if (uap->arg) { /* extract 32 bits of flags from userland */ param_ptr = (caddr_t) uap->arg; param = (uint32_t) param_ptr; - } - else { + } else { /* If no argument is specified, error out */ error = EINVAL; goto out; } - - /* - * Validate the different types of flags that can be specified: + + /* + * Validate the different types of flags that can be specified: * all of them are mutually exclusive for now. */ switch (param) { - case F_IOTYPE_ISOCHRONOUS: - break; + case F_IOTYPE_ISOCHRONOUS: + break; - default: - error = EINVAL; - goto out; + default: + error = EINVAL; + goto out; } @@ -2282,7 +2924,7 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) /* Only go forward if you have write access */ vfs_context_t ctx = vfs_context_current(); - if(vnode_authorize(vp, NULLVP, (KAUTH_VNODE_ACCESS | KAUTH_VNODE_WRITE_DATA), ctx) != 0) { + if (vnode_authorize(vp, NULLVP, (KAUTH_VNODE_ACCESS | KAUTH_VNODE_WRITE_DATA), ctx) != 0) { vnode_put(vp); error = EBADF; goto outdrop; @@ -2294,168 +2936,51 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) break; } - /* - * Extract the CodeDirectory of the vnode associated with - * the file descriptor and copy it back to user space + * Set the vnode pointed to by 'fd' + * and tag it as the (potentially future) backing store + * for another filesystem */ - case F_GETCODEDIR: { - struct user_fcodeblobs args; - + case F_SETBACKINGSTORE: { if (fp->f_type != DTYPE_VNODE) { error = EBADF; goto out; } vp = (struct vnode *)fp->f_data; - proc_fdunlock(p); - - if ((fp->f_flag & FREAD) == 0) { - error = EBADF; - goto outdrop; - } - - if (IS_64BIT_PROCESS(p)) { - struct user64_fcodeblobs args64; - - error = copyin(argp, &args64, sizeof(args64)); - if (error) - goto outdrop; - - args.f_cd_hash = args64.f_cd_hash; - args.f_hash_size = args64.f_hash_size; - args.f_cd_buffer = args64.f_cd_buffer; - args.f_cd_size = args64.f_cd_size; - args.f_out_size = args64.f_out_size; - args.f_arch = args64.f_arch; - } else { - struct user32_fcodeblobs args32; - - error = copyin(argp, &args32, sizeof(args32)); - if (error) - goto outdrop; - - args.f_cd_hash = CAST_USER_ADDR_T(args32.f_cd_hash); - args.f_hash_size = args32.f_hash_size; - args.f_cd_buffer = CAST_USER_ADDR_T(args32.f_cd_buffer); - args.f_cd_size = args32.f_cd_size; - args.f_out_size = CAST_USER_ADDR_T(args32.f_out_size); - args.f_arch = args32.f_arch; - } - if (vp->v_ubcinfo == NULL) { + if (vp->v_tag != VT_HFS) { error = EINVAL; - goto outdrop; - } - - struct cs_blob *t_blob = vp->v_ubcinfo->cs_blobs; - - /* - * This call fails if there is no cs_blob corresponding to the - * vnode, or if there are multiple cs_blobs present, and the caller - * did not specify which cpu_type they want the cs_blob for - */ - if (t_blob == NULL) { - error = ENOENT; /* there is no codesigning blob for this process */ - goto outdrop; - } else if (args.f_arch == 0 && t_blob->csb_next != NULL) { - error = ENOENT; /* too many architectures and none specified */ - goto outdrop; - } - - /* If the user specified an architecture, find the right blob */ - if (args.f_arch != 0) { - while (t_blob) { - if (t_blob->csb_cpu_type == args.f_arch) - break; - t_blob = t_blob->csb_next; - } - /* The cpu_type the user requested could not be found */ - if (t_blob == NULL) { - error = ENOENT; - goto outdrop; - } + goto out; } + proc_fdunlock(p); - const CS_CodeDirectory *cd = t_blob->csb_cd; - if (cd == NULL) { + if (vnode_getwithref(vp)) { error = ENOENT; goto outdrop; } - uint64_t buffer_size = ntohl(cd->length); - - if (buffer_size > UINT_MAX) { - error = ERANGE; + /* only proceed if you have write access */ + vfs_context_t ctx = vfs_context_current(); + if (vnode_authorize(vp, NULLVP, (KAUTH_VNODE_ACCESS | KAUTH_VNODE_WRITE_DATA), ctx) != 0) { + vnode_put(vp); + error = EBADF; goto outdrop; } - error = copyout(&buffer_size, args.f_out_size, sizeof(unsigned int)); - if (error) - goto outdrop; - if (sizeof(t_blob->csb_cdhash) > args.f_hash_size || - buffer_size > args.f_cd_size) { - error = ERANGE; - goto outdrop; + /* If arg != 0, set, otherwise unset */ + if (uap->arg) { + error = VNOP_IOCTL(vp, uap->cmd, (caddr_t)1, 0, &context); + } else { + error = VNOP_IOCTL(vp, uap->cmd, (caddr_t)NULL, 0, &context); } - error = copyout(t_blob->csb_cdhash, args.f_cd_hash, sizeof(t_blob->csb_cdhash)); - if (error) - goto outdrop; - error = copyout(cd, args.f_cd_buffer, buffer_size); - if (error) - goto outdrop; - - break; - } - - /* - * Set the vnode pointed to by 'fd' - * and tag it as the (potentially future) backing store - * for another filesystem - */ - case F_SETBACKINGSTORE: { - if (fp->f_type != DTYPE_VNODE) { - error = EBADF; - goto out; - } - - vp = (struct vnode *)fp->f_data; - - if (vp->v_tag != VT_HFS) { - error = EINVAL; - goto out; - } - proc_fdunlock(p); - - if (vnode_getwithref(vp)) { - error = ENOENT; - goto outdrop; - } - - /* only proceed if you have write access */ - vfs_context_t ctx = vfs_context_current(); - if(vnode_authorize(vp, NULLVP, (KAUTH_VNODE_ACCESS | KAUTH_VNODE_WRITE_DATA), ctx) != 0) { - vnode_put(vp); - error = EBADF; - goto outdrop; - } - - - /* If arg != 0, set, otherwise unset */ - if (uap->arg) { - error = VNOP_IOCTL (vp, uap->cmd, (caddr_t)1, 0, &context); - } - else { - error = VNOP_IOCTL (vp, uap->cmd, (caddr_t)NULL, 0, &context); - } - vnode_put(vp); break; } - /* + /* * like F_GETPATH, but special semantics for * the mobile time machine handler. */ @@ -2471,18 +2996,15 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) proc_fdunlock(p); pathlen = MAXPATHLEN; - MALLOC(pathbufp, char *, pathlen, M_TEMP, M_WAITOK); - if (pathbufp == NULL) { - error = ENOMEM; - goto outdrop; - } - if ( (error = vnode_getwithref(vp)) == 0 ) { + pathbufp = zalloc(ZV_NAMEI); + + if ((error = vnode_getwithref(vp)) == 0) { int backingstore = 0; - + /* Check for error from vn_getpath before moving on */ if ((error = vn_getpath(vp, pathbufp, &pathlen)) == 0) { if (vp->v_tag == VT_HFS) { - error = VNOP_IOCTL (vp, uap->cmd, (caddr_t) &backingstore, 0, &context); + error = VNOP_IOCTL(vp, uap->cmd, (caddr_t) &backingstore, 0, &context); } (void)vnode_put(vp); @@ -2490,7 +3012,7 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) error = copyout((caddr_t)pathbufp, argp, pathlen); } if (error == 0) { - /* + /* * If the copyout was successful, now check to ensure * that this vnode is not a BACKINGSTORE vnode. mtmd * wants the path regardless. @@ -2499,10 +3021,12 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) error = EBUSY; } } - } else + } else { (void)vnode_put(vp); + } } - FREE(pathbufp, M_TEMP); + + zfree(ZV_NAMEI, pathbufp); goto outdrop; } @@ -2526,18 +3050,27 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) * for this vnode; this can include special devices, and will * effectively overload fcntl() to send ioctl()'s. */ - if((uap->cmd & IOC_VOID) && (uap->cmd & IOC_INOUT)){ - error = EINVAL; + if ((uap->cmd & IOC_VOID) && (uap->cmd & IOC_INOUT)) { + error = EINVAL; goto out; } - + /* Catch any now-invalid fcntl() selectors */ switch (uap->cmd) { - case F_MARKDEPENDENCY: - error = EINVAL; - goto out; - default: - break; + case (int)APFSIOC_REVERT_TO_SNAPSHOT: + case (int)FSIOC_FIOSEEKHOLE: + case (int)FSIOC_FIOSEEKDATA: + case (int)FSIOC_CAS_BSDFLAGS: + case HFS_GET_BOOT_INFO: + case HFS_SET_BOOT_INFO: + case FIOPINSWAP: + case F_MARKDEPENDENCY: + case TIOCREVOKE: + case TIOCREVOKECLEAR: + error = EINVAL; + goto out; + default: + break; } if (fp->f_type != DTYPE_VNODE) { @@ -2547,9 +3080,9 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) vp = (struct vnode *)fp->f_data; proc_fdunlock(p); - if ( (error = vnode_getwithref(vp)) == 0 ) { + if ((error = vnode_getwithref(vp)) == 0) { #define STK_PARAMS 128 - char stkbuf[STK_PARAMS]; + char stkbuf[STK_PARAMS] = {0}; unsigned int size; caddr_t data, memp; /* @@ -2570,8 +3103,9 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) } memp = NULL; - if (size > sizeof (stkbuf)) { - if ((memp = (caddr_t)kalloc(size)) == 0) { + if (size > sizeof(stkbuf)) { + memp = (caddr_t)kheap_alloc(KHEAP_TEMP, size, Z_WAITOK); + if (memp == 0) { (void)vnode_put(vp); error = ENOMEM; goto outdrop; @@ -2580,21 +3114,22 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) } else { data = &stkbuf[0]; } - + if (uap->cmd & IOC_IN) { if (size) { /* structure */ error = copyin(argp, data, size); if (error) { (void)vnode_put(vp); - if (memp) - kfree(memp, size); + if (memp) { + kheap_free(KHEAP_TEMP, memp, size); + } goto outdrop; } /* Bzero the section beyond that which was needed */ if (size <= sizeof(stkbuf)) { - bzero ( (((uint8_t*)data) + size), (sizeof(stkbuf) - size)); + bzero((((uint8_t*)data) + size), (sizeof(stkbuf) - size)); } } else { /* int */ @@ -2612,9 +3147,9 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) bzero(data, size); } else if (uap->cmd & IOC_VOID) { if (is64bit) { - *(user_addr_t *)data = argp; + *(user_addr_t *)data = argp; } else { - *(uint32_t *)data = (uint32_t)argp; + *(uint32_t *)data = (uint32_t)argp; } } @@ -2623,10 +3158,12 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) (void)vnode_put(vp); /* Copy any output data to user */ - if (error == 0 && (uap->cmd & IOC_OUT) && size) + if (error == 0 && (uap->cmd & IOC_OUT) && size) { error = copyout(data, argp, size); - if (memp) - kfree(memp, size); + } + if (memp) { + kheap_free(KHEAP_TEMP, memp, size); + } } break; } @@ -2634,11 +3171,11 @@ fcntl_nocancel(proc_t p, struct fcntl_nocancel_args *uap, int32_t *retval) outdrop: AUDIT_ARG(vnpath_withref, vp, ARG_VNODE1); fp_drop(p, fd, fp, 0); - return(error); + return error; out: fp_drop(p, fd, fp, 1); proc_fdunlock(p); - return(error); + return error; } @@ -2683,16 +3220,14 @@ finishdup(proc_t p, if ((ofp = fdp->fd_ofiles[old]) == NULL || (fdp->fd_ofileflags[old] & UF_RESERVED)) { fdrelse(p, new); - return (EBADF); + return EBADF; } - fg_ref(ofp); #if CONFIG_MACF - error = mac_file_check_dup(proc_ucred(p), ofp->f_fglob, new); + error = mac_file_check_dup(proc_ucred(p), ofp->fp_glob, new); if (error) { - fg_drop(ofp); fdrelse(p, new); - return (error); + return error; } #endif @@ -2703,31 +3238,34 @@ finishdup(proc_t p, proc_fdlock(p); if (nfp == NULL) { - fg_drop(ofp); fdrelse(p, new); - return (ENOMEM); + return ENOMEM; } - nfp->f_fglob = ofp->f_fglob; + fg_ref(p, ofp->fp_glob); + nfp->fp_glob = ofp->fp_glob; #if DIAGNOSTIC - if (fdp->fd_ofiles[new] != 0) + if (fdp->fd_ofiles[new] != 0) { panic("finishdup: overwriting fd_ofiles with new %d", new); - if ((fdp->fd_ofileflags[new] & UF_RESERVED) == 0) + } + if ((fdp->fd_ofileflags[new] & UF_RESERVED) == 0) { panic("finishdup: unreserved fileflags with new %d", new); + } #endif - if (new > fdp->fd_lastfile) + if (new > fdp->fd_lastfile) { fdp->fd_lastfile = new; + } *fdflags(p, new) |= fd_flags; procfdtbl_releasefd(p, new, nfp); *retval = new; - return (0); + return 0; } /* - * close + * sys_close * * Description: The implementation of the close(2) system call * @@ -2740,167 +3278,141 @@ finishdup(proc_t p, * fp_lookup:EBADF Bad file descriptor * fp_guard_exception:??? Guarded file descriptor * close_internal:EBADF - * close_internal:??? Anything returnable by a per-fileops + * close_internal:??? Anything returnable by a per-fileops * close function */ int -close(proc_t p, struct close_args *uap, int32_t *retval) +sys_close(proc_t p, struct close_args *uap, __unused int32_t *retval) { __pthread_testcancel(1); - return(close_nocancel(p, (struct close_nocancel_args *)uap, retval)); + return close_nocancel(p, uap->fd); } +int +sys_close_nocancel(proc_t p, struct close_nocancel_args *uap, __unused int32_t *retval) +{ + return close_nocancel(p, uap->fd); +} int -close_nocancel(proc_t p, struct close_nocancel_args *uap, __unused int32_t *retval) +close_nocancel(proc_t p, int fd) { struct fileproc *fp; - int fd = uap->fd; - int error; AUDIT_SYSCLOSE(p, fd); proc_fdlock(p); - - if ( (error = fp_lookup(p,fd,&fp, 1)) ) { + if ((fp = fp_get_noref_locked(p, fd)) == NULL) { proc_fdunlock(p); - return(error); + return EBADF; } - if (FP_ISGUARDED(fp, GUARD_CLOSE)) { - error = fp_guard_exception(p, fd, fp, kGUARD_EXC_CLOSE); - (void) fp_drop(p, fd, fp, 1); + if (fp_isguarded(fp, GUARD_CLOSE)) { + int error = fp_guard_exception(p, fd, fp, kGUARD_EXC_CLOSE); proc_fdunlock(p); - return (error); + return error; } - error = close_internal_locked(p, fd, fp, 0); - - proc_fdunlock(p); - - return (error); + return fp_close_and_unlock(p, fd, fp, 0); } -/* - * close_internal_locked - * - * Close a file descriptor. - * - * Parameters: p Process in whose per process file table - * the close is to occur - * fd fd to be closed - * fp fileproc associated with the fd - * - * Returns: 0 Success - * EBADF fd already in close wait state - * closef_locked:??? Anything returnable by a per-fileops - * close function - * - * Locks: Assumes proc_fdlock for process is held by the caller and returns - * with lock held - * - * Notes: This function may drop and reacquire this lock; it is unsafe - * for a caller to assume that other state protected by the lock - * has not been subsequently changed out from under it. - */ int -close_internal_locked(proc_t p, int fd, struct fileproc *fp, int flags) +fp_close_and_unlock(proc_t p, int fd, struct fileproc *fp, int flags) { struct filedesc *fdp = p->p_fd; - int error =0; - int resvfd = flags & FD_DUP2RESV; - + struct fileglob *fg = fp->fp_glob; #if DIAGNOSTIC proc_fdlock_assert(p, LCK_MTX_ASSERT_OWNED); #endif - /* Keep people from using the filedesc while we are closing it */ - procfdtbl_markclosefd(p, fd); - - - if ((fp->f_flags & FP_CLOSING) == FP_CLOSING) { - panic("close_internal_locked: being called on already closing fd"); + /* + * Keep most people from finding the filedesc while we are closing it. + * + * Callers are: + * + * - dup2() which always waits for UF_RESERVED to clear + * + * - close/guarded_close/... who will fail the fileproc lookup if + * UF_RESERVED is set, + * + * - fdexec()/fdfree() who only run once all threads in the proc + * are properly canceled, hence no fileproc in this proc should + * be in flux. + * + * Which means that neither UF_RESERVED nor UF_CLOSING should be set. + * + * Callers of fp_get_noref_locked_with_iocount() can still find + * this entry so that they can drop their I/O reference despite + * not having remembered the fileproc pointer (namely select() and + * file_drop()). + */ + if (p->p_fd->fd_ofileflags[fd] & (UF_RESERVED | UF_CLOSING)) { + panic("%s: called with fileproc in flux (%d/:%p)", + __func__, fd, fp); } + p->p_fd->fd_ofileflags[fd] |= (UF_RESERVED | UF_CLOSING); + if ((fp->fp_flags & FP_AIOISSUED) || kauth_authorize_fileop_has_listeners()) { + proc_fdunlock(p); -#if DIAGNOSTIC - if ((fdp->fd_ofileflags[fd] & UF_RESERVED) == 0) - panic("close_internal: unreserved fileflags with fd %d", fd); -#endif - - fp->f_flags |= FP_CLOSING; - - if ( (fp->f_flags & FP_AIOISSUED) || kauth_authorize_fileop_has_listeners() ) { - - proc_fdunlock(p); - - if ( (fp->f_type == DTYPE_VNODE) && kauth_authorize_fileop_has_listeners() ) { - /* - * call out to allow 3rd party notification of close. + if ((FILEGLOB_DTYPE(fg) == DTYPE_VNODE) && kauth_authorize_fileop_has_listeners()) { + /* + * call out to allow 3rd party notification of close. * Ignore result of kauth_authorize_fileop call. */ - if (vnode_getwithref((vnode_t)fp->f_data) == 0) { - u_int fileop_flags = 0; - if ((fp->f_flags & FP_WRITTEN) != 0) - fileop_flags |= KAUTH_FILEOP_CLOSE_MODIFIED; - kauth_authorize_fileop(fp->f_fglob->fg_cred, KAUTH_FILEOP_CLOSE, - (uintptr_t)fp->f_data, (uintptr_t)fileop_flags); - vnode_put((vnode_t)fp->f_data); + if (vnode_getwithref((vnode_t)fg->fg_data) == 0) { + u_int fileop_flags = 0; + if (fg->fg_flag & FWASWRITTEN) { + fileop_flags |= KAUTH_FILEOP_CLOSE_MODIFIED; + } + kauth_authorize_fileop(fg->fg_cred, KAUTH_FILEOP_CLOSE, + (uintptr_t)fg->fg_data, (uintptr_t)fileop_flags); +#if CONFIG_MACF + mac_file_notify_close(proc_ucred(p), fp->fp_glob); +#endif + vnode_put((vnode_t)fg->fg_data); } } - if (fp->f_flags & FP_AIOISSUED) - /* + if (fp->fp_flags & FP_AIOISSUED) { + /* * cancel all async IO requests that can be cancelled. */ - _aio_close( p, fd ); + _aio_close( p, fd ); + } proc_fdlock(p); } - if (fd < fdp->fd_knlistsize) + if (fd < fdp->fd_knlistsize) { knote_fdclose(p, fd); - - if (fp->f_flags & FP_WAITEVENT) - (void)waitevent_close(p, fp); + } fileproc_drain(p, fp); - if (resvfd == 0) { - _fdrelse(p, fd); + if (flags & FD_DUP2RESV) { + fdp->fd_ofiles[fd] = NULL; + fdp->fd_ofileflags[fd] &= ~(UF_CLOSING | UF_EXCLOSE | UF_FORKCLOSE); } else { - procfdtbl_reservefd(p, fd); + fdrelse(p, fd); } - if (ENTR_SHOULDTRACE && fp->f_type == DTYPE_SOCKET) - KERNEL_ENERGYTRACE(kEnTrActKernSocket, DBG_FUNC_END, - fd, 0, (int64_t)VM_KERNEL_ADDRPERM(fp->f_data)); - - error = closef_locked(fp, fp->f_fglob, p); - if ((fp->f_flags & FP_WAITCLOSE) == FP_WAITCLOSE) - wakeup(&fp->f_flags); - fp->f_flags &= ~(FP_WAITCLOSE | FP_CLOSING); - proc_fdunlock(p); - fileproc_free(fp); - - proc_fdlock(p); - -#if DIAGNOSTIC - if (resvfd != 0) { - if ((fdp->fd_ofileflags[fd] & UF_RESERVED) == 0) - panic("close with reserved fd returns with freed fd:%d: proc: %p", fd, p); + if (ENTR_SHOULDTRACE && FILEGLOB_DTYPE(fg) == DTYPE_SOCKET) { + KERNEL_ENERGYTRACE(kEnTrActKernSocket, DBG_FUNC_END, + fd, 0, (int64_t)VM_KERNEL_ADDRPERM(fg->fg_data)); } -#endif - return(error); + fileproc_free(fp); + + return fg_drop(p, fg); } /* - * fstat1 + * fstat * * Description: Return status information about a file descriptor. * @@ -2934,7 +3446,7 @@ close_internal_locked(proc_t p, int fd, struct fileproc *fp, int flags) * XXX fileops instead. */ static int -fstat1(proc_t p, int fd, user_addr_t ub, user_addr_t xsecurity, user_addr_t xsecurity_size, int isstat64) +fstat(proc_t p, int fd, user_addr_t ub, user_addr_t xsecurity, user_addr_t xsecurity_size, int isstat64) { struct fileproc *fp; union { @@ -2959,7 +3471,7 @@ fstat1(proc_t p, int fd, user_addr_t ub, user_addr_t xsecurity, user_addr_t xsec AUDIT_ARG(fd, fd); if ((error = fp_lookup(p, fd, &fp, 0)) != 0) { - return(error); + return error; } type = fp->f_type; data = fp->f_data; @@ -2968,7 +3480,6 @@ fstat1(proc_t p, int fd, user_addr_t ub, user_addr_t xsecurity, user_addr_t xsec sbptr = (void *)&source; switch (type) { - case DTYPE_VNODE: if ((error = vnode_getwithref((vnode_t)data)) == 0) { /* @@ -2977,9 +3488,10 @@ fstat1(proc_t p, int fd, user_addr_t ub, user_addr_t xsecurity, user_addr_t xsec * going to let them get the basic stat information. */ if (xsecurity == USER_ADDR_NULL) { - error = vn_stat_noauth((vnode_t)data, sbptr, NULL, isstat64, ctx); + error = vn_stat_noauth((vnode_t)data, sbptr, NULL, isstat64, 0, ctx, + fp->fp_glob->fg_cred); } else { - error = vn_stat((vnode_t)data, sbptr, &fsec, isstat64, ctx); + error = vn_stat((vnode_t)data, sbptr, &fsec, isstat64, 0, ctx); } AUDIT_ARG(vnpath, (struct vnode *)data, ARG_VNODE1); @@ -3018,11 +3530,11 @@ fstat1(proc_t p, int fd, user_addr_t ub, user_addr_t xsecurity, user_addr_t xsec source.sb64.st_qspare[1] = 0LL; if (IS_64BIT_PROCESS(current_proc())) { - munge_user64_stat64(&source.sb64, &dest.user64_sb64); + munge_user64_stat64(&source.sb64, &dest.user64_sb64); my_size = sizeof(dest.user64_sb64); sbp = (caddr_t)&dest.user64_sb64; } else { - munge_user32_stat64(&source.sb64, &dest.user32_sb64); + munge_user32_stat64(&source.sb64, &dest.user32_sb64); my_size = sizeof(dest.user32_sb64); sbp = (caddr_t)&dest.user32_sb64; } @@ -3031,11 +3543,11 @@ fstat1(proc_t p, int fd, user_addr_t ub, user_addr_t xsecurity, user_addr_t xsec source.sb.st_qspare[0] = 0LL; source.sb.st_qspare[1] = 0LL; if (IS_64BIT_PROCESS(current_proc())) { - munge_user64_stat(&source.sb, &dest.user64_sb); + munge_user64_stat(&source.sb, &dest.user64_sb); my_size = sizeof(dest.user64_sb); sbp = (caddr_t)&dest.user64_sb; } else { - munge_user32_stat(&source.sb, &dest.user32_sb); + munge_user32_stat(&source.sb, &dest.user32_sb); my_size = sizeof(dest.user32_sb); sbp = (caddr_t)&dest.user32_sb; } @@ -3046,9 +3558,8 @@ fstat1(proc_t p, int fd, user_addr_t ub, user_addr_t xsecurity, user_addr_t xsec /* caller wants extended security information? */ if (xsecurity != USER_ADDR_NULL) { - /* did we get any? */ - if (fsec == KAUTH_FILESEC_NONE) { + if (fsec == KAUTH_FILESEC_NONE) { if (susize(xsecurity_size, 0) != 0) { error = EFAULT; goto out; @@ -3064,20 +3575,22 @@ fstat1(proc_t p, int fd, user_addr_t ub, user_addr_t xsecurity, user_addr_t xsec } /* if the caller supplied enough room, copy out to it */ - if (xsecurity_bufsize >= KAUTH_FILESEC_COPYSIZE(fsec)) + if (xsecurity_bufsize >= KAUTH_FILESEC_COPYSIZE(fsec)) { error = copyout(fsec, xsecurity, KAUTH_FILESEC_COPYSIZE(fsec)); + } } } out: fp_drop(p, fd, fp, 0); - if (fsec != NULL) + if (fsec != NULL) { kauth_filesec_free(fsec); - return (error); + } + return error; } /* - * fstat_extended + * sys_fstat_extended * * Description: Extended version of fstat supporting returning extended * security information @@ -3090,17 +3603,17 @@ out: * uap->xsecurity_size The size of xsecurity, or 0 * * Returns: 0 Success - * !0 Errno (see fstat1) + * !0 Errno (see fstat) */ int -fstat_extended(proc_t p, struct fstat_extended_args *uap, __unused int32_t *retval) +sys_fstat_extended(proc_t p, struct fstat_extended_args *uap, __unused int32_t *retval) { - return(fstat1(p, uap->fd, uap->ub, uap->xsecurity, uap->xsecurity_size, 0)); + return fstat(p, uap->fd, uap->ub, uap->xsecurity, uap->xsecurity_size, 0); } - + /* - * fstat + * sys_fstat * * Description: Get file status for the file associated with fd * @@ -3109,17 +3622,17 @@ fstat_extended(proc_t p, struct fstat_extended_args *uap, __unused int32_t *retv * uap->ub The user stat buffer * * Returns: 0 Success - * !0 Errno (see fstat1) + * !0 Errno (see fstat) */ int -fstat(proc_t p, register struct fstat_args *uap, __unused int32_t *retval) +sys_fstat(proc_t p, struct fstat_args *uap, __unused int32_t *retval) { - return(fstat1(p, uap->fd, uap->ub, 0, 0, 0)); + return fstat(p, uap->fd, uap->ub, 0, 0, 0); } /* - * fstat64_extended + * sys_fstat64_extended * * Description: Extended version of fstat64 supporting returning extended * security information @@ -3132,17 +3645,17 @@ fstat(proc_t p, register struct fstat_args *uap, __unused int32_t *retval) * uap->xsecurity_size The size of xsecurity, or 0 * * Returns: 0 Success - * !0 Errno (see fstat1) + * !0 Errno (see fstat) */ int -fstat64_extended(proc_t p, struct fstat64_extended_args *uap, __unused int32_t *retval) +sys_fstat64_extended(proc_t p, struct fstat64_extended_args *uap, __unused int32_t *retval) { - return(fstat1(p, uap->fd, uap->ub, uap->xsecurity, uap->xsecurity_size, 1)); + return fstat(p, uap->fd, uap->ub, uap->xsecurity, uap->xsecurity_size, 1); } - + /* - * fstat64 + * sys_fstat64 * * Description: Get 64 bit version of the file status for the file associated * with fd @@ -3152,17 +3665,17 @@ fstat64_extended(proc_t p, struct fstat64_extended_args *uap, __unused int32_t * * uap->ub The user stat buffer * * Returns: 0 Success - * !0 Errno (see fstat1) + * !0 Errno (see fstat) */ int -fstat64(proc_t p, register struct fstat64_args *uap, __unused int32_t *retval) +sys_fstat64(proc_t p, struct fstat64_args *uap, __unused int32_t *retval) { - return(fstat1(p, uap->fd, uap->ub, 0, 0, 1)); + return fstat(p, uap->fd, uap->ub, 0, 0, 1); } /* - * fpathconf + * sys_fpathconf * * Description: Return pathconf information about a file descriptor. * @@ -3181,7 +3694,7 @@ fstat64(proc_t p, register struct fstat64_args *uap, __unused int32_t *retval) * *retval (modified) Returned information (numeric) */ int -fpathconf(proc_t p, struct fpathconf_args *uap, int32_t *retval) +sys_fpathconf(proc_t p, struct fpathconf_args *uap, int32_t *retval) { int fd = uap->fd; struct fileproc *fp; @@ -3192,13 +3705,13 @@ fpathconf(proc_t p, struct fpathconf_args *uap, int32_t *retval) AUDIT_ARG(fd, uap->fd); - if ( (error = fp_lookup(p, fd, &fp, 0)) ) - return(error); + if ((error = fp_lookup(p, fd, &fp, 0))) { + return error; + } type = fp->f_type; data = fp->f_data; switch (type) { - case DTYPE_SOCKET: if (uap->name != _PC_PIPE_BUF) { error = EINVAL; @@ -3220,8 +3733,8 @@ fpathconf(proc_t p, struct fpathconf_args *uap, int32_t *retval) case DTYPE_VNODE: vp = (struct vnode *)data; - if ( (error = vnode_getwithref(vp)) == 0) { - AUDIT_ARG(vnpath, vp, ARG_VNODE1); + if ((error = vnode_getwithref(vp)) == 0) { + AUDIT_ARG(vnpath, vp, ARG_VNODE1); error = vn_pathconf(vp, uap->name, retval, vfs_context_current()); @@ -3232,12 +3745,11 @@ fpathconf(proc_t p, struct fpathconf_args *uap, int32_t *retval) default: error = EINVAL; goto out; - } /*NOTREACHED*/ out: fp_drop(p, fd, fp, 0); - return(error); + return error; } /* @@ -3270,9 +3782,13 @@ fdalloc(proc_t p, int want, int *result) { struct filedesc *fdp = p->p_fd; int i; - int lim, last, numfiles, oldnfiles; + int last, numfiles, oldnfiles; struct fileproc **newofiles, **ofiles; char *newofileflags; + rlim_t lim; + rlim_t nofile = proc_limitgetcur(p, RLIMIT_NOFILE, TRUE); + + nofile = MIN(nofile, INT_MAX); /* * Search for a free descriptor starting at the higher @@ -3283,44 +3799,50 @@ fdalloc(proc_t p, int want, int *result) proc_fdlock_assert(p, LCK_MTX_ASSERT_OWNED); #endif - lim = min((int)p->p_rlimit[RLIMIT_NOFILE].rlim_cur, maxfiles); + lim = MIN(nofile, maxfilesperproc); for (;;) { - last = min(fdp->fd_nfiles, lim); - if ((i = want) < fdp->fd_freefile) + last = (int)MIN((unsigned int)fdp->fd_nfiles, (unsigned int)lim); + if ((i = want) < fdp->fd_freefile) { i = fdp->fd_freefile; + } for (; i < last; i++) { if (fdp->fd_ofiles[i] == NULL && !(fdp->fd_ofileflags[i] & UF_RESERVED)) { procfdtbl_reservefd(p, i); - if (i > fdp->fd_lastfile) + if (i > fdp->fd_lastfile) { fdp->fd_lastfile = i; - if (want <= fdp->fd_freefile) + } + if (want <= fdp->fd_freefile) { fdp->fd_freefile = i; + } *result = i; - return (0); + return 0; } } /* * No space in current array. Expand? */ - if (fdp->fd_nfiles >= lim) - return (EMFILE); - if (fdp->fd_nfiles < NDEXTENT) + if ((rlim_t)fdp->fd_nfiles >= lim) { + return EMFILE; + } + if (fdp->fd_nfiles < NDEXTENT) { numfiles = NDEXTENT; - else + } else { numfiles = 2 * fdp->fd_nfiles; + } /* Enforce lim */ - if (numfiles > lim) - numfiles = lim; + if ((rlim_t)numfiles > lim) { + numfiles = (int)lim; + } proc_fdunlock(p); - MALLOC_ZONE(newofiles, struct fileproc **, - numfiles * OFILESIZE, M_OFILETABL, M_WAITOK); + newofiles = kheap_alloc(KM_OFILETABL, numfiles * OFILESIZE, + Z_WAITOK); proc_fdlock(p); if (newofiles == NULL) { - return (ENOMEM); + return ENOMEM; } if (fdp->fd_nfiles >= numfiles) { - FREE_ZONE(newofiles, numfiles * OFILESIZE, M_OFILETABL); + kheap_free(KM_OFILETABL, newofiles, numfiles * OFILESIZE); continue; } newofileflags = (char *) &newofiles[numfiles]; @@ -3330,20 +3852,20 @@ fdalloc(proc_t p, int want, int *result) */ oldnfiles = fdp->fd_nfiles; (void) memcpy(newofiles, fdp->fd_ofiles, - oldnfiles * sizeof(*fdp->fd_ofiles)); + oldnfiles * sizeof(*fdp->fd_ofiles)); (void) memset(&newofiles[oldnfiles], 0, - (numfiles - oldnfiles) * sizeof(*fdp->fd_ofiles)); + (numfiles - oldnfiles) * sizeof(*fdp->fd_ofiles)); (void) memcpy(newofileflags, fdp->fd_ofileflags, - oldnfiles * sizeof(*fdp->fd_ofileflags)); + oldnfiles * sizeof(*fdp->fd_ofileflags)); (void) memset(&newofileflags[oldnfiles], 0, - (numfiles - oldnfiles) * - sizeof(*fdp->fd_ofileflags)); + (numfiles - oldnfiles) * + sizeof(*fdp->fd_ofileflags)); ofiles = fdp->fd_ofiles; fdp->fd_ofiles = newofiles; fdp->fd_ofileflags = newofileflags; fdp->fd_nfiles = numfiles; - FREE_ZONE(ofiles, oldnfiles * OFILESIZE, M_OFILETABL); + kheap_free(KM_OFILETABL, ofiles, oldnfiles * OFILESIZE); fdexpand++; } } @@ -3372,140 +3894,90 @@ fdavail(proc_t p, int n) struct filedesc *fdp = p->p_fd; struct fileproc **fpp; char *flags; - int i, lim; + int i; + int lim; + rlim_t nofile = proc_limitgetcur(p, RLIMIT_NOFILE, TRUE); - lim = min((int)p->p_rlimit[RLIMIT_NOFILE].rlim_cur, maxfiles); - if ((i = lim - fdp->fd_nfiles) > 0 && (n -= i) <= 0) - return (1); + lim = (int)MIN(nofile, maxfilesperproc); + if ((i = lim - fdp->fd_nfiles) > 0 && (n -= i) <= 0) { + return 1; + } fpp = &fdp->fd_ofiles[fdp->fd_freefile]; flags = &fdp->fd_ofileflags[fdp->fd_freefile]; - for (i = fdp->fd_nfiles - fdp->fd_freefile; --i >= 0; fpp++, flags++) - if (*fpp == NULL && !(*flags & UF_RESERVED) && --n <= 0) - return (1); - return (0); + for (i = fdp->fd_nfiles - fdp->fd_freefile; --i >= 0; fpp++, flags++) { + if (*fpp == NULL && !(*flags & UF_RESERVED) && --n <= 0) { + return 1; + } + } + return 0; } -/* - * fdrelse - * - * Description: Legacy KPI wrapper function for _fdrelse - * - * Parameters: p Process in which fd lives - * fd fd to free - * - * Returns: void - * - * Locks: Assumes proc_fdlock for process is held by the caller - */ -void -fdrelse(proc_t p, int fd) +struct fileproc * +fp_get_noref_locked(proc_t p, int fd) { - _fdrelse(p, fd); -} + struct filedesc *fdp = p->p_fd; + struct fileproc *fp; + if (fd < 0 || fd >= fdp->fd_nfiles || + (fp = fdp->fd_ofiles[fd]) == NULL || + (fdp->fd_ofileflags[fd] & UF_RESERVED)) { + return NULL; + } + return fp; +} -/* - * fdgetf_noref - * - * Description: Get the fileproc pointer for the given fd from the per process - * open file table without taking an explicit reference on it. - * - * Parameters: p Process containing fd - * fd fd to obtain fileproc for - * resultfp Pointer to pointer return area - * - * Returns: 0 Success - * EBADF - * - * Implicit returns: - * *resultfp (modified) Pointer to fileproc pointer - * - * Locks: Assumes proc_fdlock for process is held by the caller - * - * Notes: Because there is no reference explicitly taken, the returned - * fileproc pointer is only valid so long as the proc_fdlock - * remains held by the caller. - */ -int -fdgetf_noref(proc_t p, int fd, struct fileproc **resultfp) +struct fileproc * +fp_get_noref_locked_with_iocount(proc_t p, int fd) { struct filedesc *fdp = p->p_fd; - struct fileproc *fp; + struct fileproc *fp = NULL; if (fd < 0 || fd >= fdp->fd_nfiles || - (fp = fdp->fd_ofiles[fd]) == NULL || - (fdp->fd_ofileflags[fd] & UF_RESERVED)) { - return (EBADF); + (fp = fdp->fd_ofiles[fd]) == NULL || + os_ref_get_count(&fp->fp_iocount) <= 1 || + ((fdp->fd_ofileflags[fd] & UF_RESERVED) && + !(fdp->fd_ofileflags[fd] & UF_CLOSING))) { + panic("%s: caller without an ioccount on fileproc (%d/:%p)", + __func__, fd, fp); } - if (resultfp) - *resultfp = fp; - return (0); -} + return fp; +} -/* - * fp_getfvp - * - * Description: Get fileproc and vnode pointer for a given fd from the per - * process open file table of the specified process, and if - * successful, increment the f_iocount - * - * Parameters: p Process in which fd lives - * fd fd to get information for - * resultfp Pointer to result fileproc - * pointer area, or 0 if none - * resultvp Pointer to result vnode pointer - * area, or 0 if none - * - * Returns: 0 Success - * EBADF Bad file descriptor - * ENOTSUP fd does not refer to a vnode - * - * Implicit returns: - * *resultfp (modified) Fileproc pointer - * *resultvp (modified) vnode pointer - * - * Notes: The resultfp and resultvp fields are optional, and may be - * independently specified as NULL to skip returning information - * - * Locks: Internally takes and releases proc_fdlock - */ int -fp_getfvp(proc_t p, int fd, struct fileproc **resultfp, struct vnode **resultvp) +fp_get_ftype(proc_t p, int fd, file_type_t ftype, int err, struct fileproc **fpp) { struct filedesc *fdp = p->p_fd; struct fileproc *fp; proc_fdlock_spin(p); if (fd < 0 || fd >= fdp->fd_nfiles || - (fp = fdp->fd_ofiles[fd]) == NULL || - (fdp->fd_ofileflags[fd] & UF_RESERVED)) { + (fp = fdp->fd_ofiles[fd]) == NULL || + (fdp->fd_ofileflags[fd] & UF_RESERVED)) { proc_fdunlock(p); - return (EBADF); + return EBADF; } - if (fp->f_type != DTYPE_VNODE) { + + if (fp->f_type != ftype) { proc_fdunlock(p); - return(ENOTSUP); + return err; } - fp->f_iocount++; - if (resultfp) - *resultfp = fp; - if (resultvp) - *resultvp = (struct vnode *)fp->f_data; + os_ref_retain_locked(&fp->fp_iocount); proc_fdunlock(p); - return (0); + *fpp = fp; + return 0; } /* - * fp_getfvpandvid + * fp_getfvp * - * Description: Get fileproc, vnode pointer, and vid for a given fd from the - * per process open file table of the specified process, and if - * successful, increment the f_iocount + * Description: Get fileproc and vnode pointer for a given fd from the per + * process open file table of the specified process, and if + * successful, increment the fp_iocount * * Parameters: p Process in which fd lives * fd fd to get information for @@ -3513,7 +3985,6 @@ fp_getfvp(proc_t p, int fd, struct fileproc **resultfp, struct vnode **resultvp) * pointer area, or 0 if none * resultvp Pointer to result vnode pointer * area, or 0 if none - * vidp Pointer to resuld vid area * * Returns: 0 Success * EBADF Bad file descriptor @@ -3522,7 +3993,6 @@ fp_getfvp(proc_t p, int fd, struct fileproc **resultfp, struct vnode **resultvp) * Implicit returns: * *resultfp (modified) Fileproc pointer * *resultvp (modified) vnode pointer - * *vidp vid value * * Notes: The resultfp and resultvp fields are optional, and may be * independently specified as NULL to skip returning information @@ -3530,331 +4000,81 @@ fp_getfvp(proc_t p, int fd, struct fileproc **resultfp, struct vnode **resultvp) * Locks: Internally takes and releases proc_fdlock */ int -fp_getfvpandvid(proc_t p, int fd, struct fileproc **resultfp, - struct vnode **resultvp, uint32_t *vidp) -{ - struct filedesc *fdp = p->p_fd; - struct fileproc *fp; - - proc_fdlock_spin(p); - if (fd < 0 || fd >= fdp->fd_nfiles || - (fp = fdp->fd_ofiles[fd]) == NULL || - (fdp->fd_ofileflags[fd] & UF_RESERVED)) { - proc_fdunlock(p); - return (EBADF); - } - if (fp->f_type != DTYPE_VNODE) { - proc_fdunlock(p); - return(ENOTSUP); - } - fp->f_iocount++; - - if (resultfp) - *resultfp = fp; - if (resultvp) - *resultvp = (struct vnode *)fp->f_data; - if (vidp) - *vidp = (uint32_t)vnode_vid((struct vnode *)fp->f_data); - proc_fdunlock(p); - - return (0); -} - - -/* - * fp_getfsock - * - * Description: Get fileproc and socket pointer for a given fd from the - * per process open file table of the specified process, and if - * successful, increment the f_iocount - * - * Parameters: p Process in which fd lives - * fd fd to get information for - * resultfp Pointer to result fileproc - * pointer area, or 0 if none - * results Pointer to result socket - * pointer area, or 0 if none - * - * Returns: EBADF The file descriptor is invalid - * EOPNOTSUPP The file descriptor is not a socket - * 0 Success - * - * Implicit returns: - * *resultfp (modified) Fileproc pointer - * *results (modified) socket pointer - * - * Notes: EOPNOTSUPP should probably be ENOTSOCK; this function is only - * ever called from accept1(). - */ -int -fp_getfsock(proc_t p, int fd, struct fileproc **resultfp, - struct socket **results) -{ - struct filedesc *fdp = p->p_fd; - struct fileproc *fp; - - proc_fdlock_spin(p); - if (fd < 0 || fd >= fdp->fd_nfiles || - (fp = fdp->fd_ofiles[fd]) == NULL || - (fdp->fd_ofileflags[fd] & UF_RESERVED)) { - proc_fdunlock(p); - return (EBADF); - } - if (fp->f_type != DTYPE_SOCKET) { - proc_fdunlock(p); - return(EOPNOTSUPP); - } - fp->f_iocount++; - - if (resultfp) - *resultfp = fp; - if (results) - *results = (struct socket *)fp->f_data; - proc_fdunlock(p); - - return (0); -} - - -/* - * fp_getfkq - * - * Description: Get fileproc and kqueue pointer for a given fd from the - * per process open file table of the specified process, and if - * successful, increment the f_iocount - * - * Parameters: p Process in which fd lives - * fd fd to get information for - * resultfp Pointer to result fileproc - * pointer area, or 0 if none - * resultkq Pointer to result kqueue - * pointer area, or 0 if none - * - * Returns: EBADF The file descriptor is invalid - * EBADF The file descriptor is not a socket - * 0 Success - * - * Implicit returns: - * *resultfp (modified) Fileproc pointer - * *resultkq (modified) kqueue pointer - * - * Notes: The second EBADF should probably be something else to make - * the error condition distinct. - */ -int -fp_getfkq(proc_t p, int fd, struct fileproc **resultfp, - struct kqueue **resultkq) +fp_getfvp(proc_t p, int fd, struct fileproc **resultfp, struct vnode **resultvp) { - struct filedesc *fdp = p->p_fd; struct fileproc *fp; + int error; - proc_fdlock_spin(p); - if ( fd < 0 || fd >= fdp->fd_nfiles || - (fp = fdp->fd_ofiles[fd]) == NULL || - (fdp->fd_ofileflags[fd] & UF_RESERVED)) { - proc_fdunlock(p); - return (EBADF); - } - if (fp->f_type != DTYPE_KQUEUE) { - proc_fdunlock(p); - return(EBADF); + error = fp_get_ftype(p, fd, DTYPE_VNODE, ENOTSUP, &fp); + if (error == 0) { + if (resultfp) { + *resultfp = fp; + } + if (resultvp) { + *resultvp = (struct vnode *)fp->f_data; + } } - fp->f_iocount++; - - if (resultfp) - *resultfp = fp; - if (resultkq) - *resultkq = (struct kqueue *)fp->f_data; - proc_fdunlock(p); - return (0); + return error; } /* - * fp_getfpshm + * fp_get_pipe_id * - * Description: Get fileproc and POSIX shared memory pointer for a given fd - * from the per process open file table of the specified process - * and if successful, increment the f_iocount + * Description: Get pipe id for a given fd from the per process open file table + * of the specified process. * * Parameters: p Process in which fd lives * fd fd to get information for - * resultfp Pointer to result fileproc - * pointer area, or 0 if none - * resultpshm Pointer to result POSIX - * shared memory pointer - * pointer area, or 0 if none + * result_pipe_id Pointer to result pipe id * - * Returns: EBADF The file descriptor is invalid - * EBADF The file descriptor is not a POSIX - * shared memory area - * 0 Success + * Returns: 0 Success + * EIVAL NULL pointer arguments passed + * fp_lookup:EBADF Bad file descriptor + * ENOTSUP fd does not refer to a pipe * * Implicit returns: - * *resultfp (modified) Fileproc pointer - * *resultpshm (modified) POSIX shared memory pointer + * *result_pipe_id (modified) pipe id * - * Notes: The second EBADF should probably be something else to make - * the error condition distinct. + * Locks: Internally takes and releases proc_fdlock */ int -fp_getfpshm(proc_t p, int fd, struct fileproc **resultfp, - struct pshmnode **resultpshm) +fp_get_pipe_id(proc_t p, int fd, uint64_t *result_pipe_id) { - struct filedesc *fdp = p->p_fd; - struct fileproc *fp; + struct fileproc *fp = FILEPROC_NULL; + struct fileglob *fg = NULL; + int error = 0; - proc_fdlock_spin(p); - if (fd < 0 || fd >= fdp->fd_nfiles || - (fp = fdp->fd_ofiles[fd]) == NULL || - (fdp->fd_ofileflags[fd] & UF_RESERVED)) { - proc_fdunlock(p); - return (EBADF); + if (p == NULL || result_pipe_id == NULL) { + return EINVAL; } - if (fp->f_type != DTYPE_PSXSHM) { + proc_fdlock(p); + if ((error = fp_lookup(p, fd, &fp, 1))) { proc_fdunlock(p); - return(EBADF); + return error; } - fp->f_iocount++; + fg = fp->fp_glob; - if (resultfp) - *resultfp = fp; - if (resultpshm) - *resultpshm = (struct pshmnode *)fp->f_data; - proc_fdunlock(p); - - return (0); -} - - -/* - * fp_getfsem - * - * Description: Get fileproc and POSIX semaphore pointer for a given fd from - * the per process open file table of the specified process - * and if successful, increment the f_iocount - * - * Parameters: p Process in which fd lives - * fd fd to get information for - * resultfp Pointer to result fileproc - * pointer area, or 0 if none - * resultpsem Pointer to result POSIX - * semaphore pointer area, or - * 0 if none - * - * Returns: EBADF The file descriptor is invalid - * EBADF The file descriptor is not a POSIX - * semaphore - * 0 Success - * - * Implicit returns: - * *resultfp (modified) Fileproc pointer - * *resultpsem (modified) POSIX semaphore pointer - * - * Notes: The second EBADF should probably be something else to make - * the error condition distinct. - * - * In order to support unnamed POSIX semaphores, the named - * POSIX semaphores will have to move out of the per-process - * open filetable, and into a global table that is shared with - * unnamed POSIX semaphores, since unnamed POSIX semaphores - * are typically used by declaring instances in shared memory, - * and there's no other way to do this without changing the - * underlying type, which would introduce binary compatibility - * issues. - */ -int -fp_getfpsem(proc_t p, int fd, struct fileproc **resultfp, - struct psemnode **resultpsem) -{ - struct filedesc *fdp = p->p_fd; - struct fileproc *fp; - - proc_fdlock_spin(p); - if (fd < 0 || fd >= fdp->fd_nfiles || - (fp = fdp->fd_ofiles[fd]) == NULL || - (fdp->fd_ofileflags[fd] & UF_RESERVED)) { - proc_fdunlock(p); - return (EBADF); - } - if (fp->f_type != DTYPE_PSXSEM) { - proc_fdunlock(p); - return(EBADF); + if (FILEGLOB_DTYPE(fg) == DTYPE_PIPE) { + *result_pipe_id = pipe_id((struct pipe*)fg->fg_data); + } else { + error = ENOTSUP; } - fp->f_iocount++; - if (resultfp) - *resultfp = fp; - if (resultpsem) - *resultpsem = (struct psemnode *)fp->f_data; + fp_drop(p, fd, fp, 1); proc_fdunlock(p); - - return (0); + return error; } -/* - * fp_getfpipe - * - * Description: Get fileproc and pipe pointer for a given fd from the - * per process open file table of the specified process - * and if successful, increment the f_iocount - * - * Parameters: p Process in which fd lives - * fd fd to get information for - * resultfp Pointer to result fileproc - * pointer area, or 0 if none - * resultpipe Pointer to result pipe - * pointer area, or 0 if none - * - * Returns: EBADF The file descriptor is invalid - * EBADF The file descriptor is not a socket - * 0 Success - * - * Implicit returns: - * *resultfp (modified) Fileproc pointer - * *resultpipe (modified) pipe pointer - * - * Notes: The second EBADF should probably be something else to make - * the error condition distinct. - */ -int -fp_getfpipe(proc_t p, int fd, struct fileproc **resultfp, - struct pipe **resultpipe) -{ - struct filedesc *fdp = p->p_fd; - struct fileproc *fp; - - proc_fdlock_spin(p); - if (fd < 0 || fd >= fdp->fd_nfiles || - (fp = fdp->fd_ofiles[fd]) == NULL || - (fdp->fd_ofileflags[fd] & UF_RESERVED)) { - proc_fdunlock(p); - return (EBADF); - } - if (fp->f_type != DTYPE_PIPE) { - proc_fdunlock(p); - return(EBADF); - } - fp->f_iocount++; - - if (resultfp) - *resultfp = fp; - if (resultpipe) - *resultpipe = (struct pipe *)fp->f_data; - proc_fdunlock(p); - - return (0); -} - /* * fp_lookup * * Description: Get fileproc pointer for a given fd from the per process * open file table of the specified process and if successful, - * increment the f_iocount + * increment the fp_iocount * * Parameters: p Process in which fd lives * fd fd to get information for @@ -3879,32 +4099,37 @@ fp_lookup(proc_t p, int fd, struct fileproc **resultfp, int locked) struct filedesc *fdp = p->p_fd; struct fileproc *fp; - if (!locked) + if (!locked) { proc_fdlock_spin(p); + } if (fd < 0 || fdp == NULL || fd >= fdp->fd_nfiles || - (fp = fdp->fd_ofiles[fd]) == NULL || - (fdp->fd_ofileflags[fd] & UF_RESERVED)) { - if (!locked) + (fp = fdp->fd_ofiles[fd]) == NULL || + (fdp->fd_ofileflags[fd] & UF_RESERVED)) { + if (!locked) { proc_fdunlock(p); - return (EBADF); + } + return EBADF; } - fp->f_iocount++; + os_ref_retain_locked(&fp->fp_iocount); - if (resultfp) + if (resultfp) { *resultfp = fp; - if (!locked) + } + if (!locked) { proc_fdunlock(p); - - return (0); + } + + return 0; } /* * fp_tryswap - * + * * Description: Swap the fileproc pointer for a given fd with a new * fileproc pointer in the per-process open file table of * the specified process. The fdlock must be held at entry. + * Iff the swap is successful, the old fileproc pointer is freed. * * Parameters: p Process containing the fd * fd The fd of interest @@ -3913,7 +4138,7 @@ fp_lookup(proc_t p, int fd, struct fileproc **resultfp, int locked) * Returns: 0 Success * EBADF Bad file descriptor * EINTR Interrupted - * EKEEPLOOKING f_iocount changed while lock was dropped. + * EKEEPLOOKING Other references were active, try again. */ int fp_tryswap(proc_t p, int fd, struct fileproc *nfp) @@ -3923,27 +4148,35 @@ fp_tryswap(proc_t p, int fd, struct fileproc *nfp) proc_fdlock_assert(p, LCK_MTX_ASSERT_OWNED); - if (0 != (error = fp_lookup(p, fd, &fp, 1))) - return (error); + if (0 != (error = fp_lookup(p, fd, &fp, 1))) { + return error; + } /* * At this point, our caller (change_guardedfd_np) has - * one f_iocount reference, and we just took another + * one fp_iocount reference, and we just took another * one to begin the replacement. + * fp and nfp have a +1 reference from allocation. + * Thus if no-one else is looking, fp_iocount should be 3. */ - if (fp->f_iocount < 2) { - panic("f_iocount too small %d", fp->f_iocount); - } else if (2 == fp->f_iocount) { - + if (os_ref_get_count(&fp->fp_iocount) < 3 || + 1 != os_ref_get_count(&nfp->fp_iocount)) { + panic("%s: fp_iocount", __func__); + } else if (3 == os_ref_get_count(&fp->fp_iocount)) { /* Copy the contents of *fp, preserving the "type" of *nfp */ - nfp->f_flags = (nfp->f_flags & FP_TYPEMASK) | - (fp->f_flags & ~FP_TYPEMASK); - nfp->f_iocount = fp->f_iocount; - nfp->f_fglob = fp->f_fglob; - nfp->f_wset = fp->f_wset; + nfp->fp_flags = (nfp->fp_flags & FP_TYPEMASK) | + (fp->fp_flags & ~FP_TYPEMASK); + os_ref_retain_locked(&nfp->fp_iocount); + os_ref_retain_locked(&nfp->fp_iocount); + nfp->fp_glob = fp->fp_glob; + nfp->fp_wset = fp->fp_wset; p->p_fd->fd_ofiles[fd] = nfp; - (void) fp_drop(p, fd, nfp, 1); + fp_drop(p, fd, nfp, 1); + + os_ref_release_live(&fp->fp_iocount); + os_ref_release_live(&fp->fp_iocount); + fileproc_free(fp); } else { /* * Wait for all other active references to evaporate. @@ -3957,83 +4190,10 @@ fp_tryswap(proc_t p, int fd, struct fileproc *nfp) * reevaluation of the change-guard attempt. */ error = EKEEPLOOKING; - printf("%s: lookup collision fd %d\n", __func__, fd); } (void) fp_drop(p, fd, fp, 1); } - return (error); -} - - -/* - * fp_drop_written - * - * Description: Set the FP_WRITTEN flag on the fileproc and drop the I/O - * reference previously taken by calling fp_lookup et. al. - * - * Parameters: p Process in which the fd lives - * fd fd associated with the fileproc - * fp fileproc on which to set the - * flag and drop the reference - * - * Returns: 0 Success - * fp_drop:EBADF Bad file descriptor - * - * Locks: This function internally takes and drops the proc_fdlock for - * the supplied process - * - * Notes: The fileproc must correspond to the fd in the supplied proc - */ -int -fp_drop_written(proc_t p, int fd, struct fileproc *fp) -{ - int error; - - proc_fdlock_spin(p); - - fp->f_flags |= FP_WRITTEN; - - error = fp_drop(p, fd, fp, 1); - - proc_fdunlock(p); - - return (error); -} - - -/* - * fp_drop_event - * - * Description: Set the FP_WAITEVENT flag on the fileproc and drop the I/O - * reference previously taken by calling fp_lookup et. al. - * - * Parameters: p Process in which the fd lives - * fd fd associated with the fileproc - * fp fileproc on which to set the - * flag and drop the reference - * - * Returns: 0 Success - * fp_drop:EBADF Bad file descriptor - * - * Locks: This function internally takes and drops the proc_fdlock for - * the supplied process - * - * Notes: The fileproc must correspond to the fd in the supplied proc - */ -int -fp_drop_event(proc_t p, int fd, struct fileproc *fp) -{ - int error; - - proc_fdlock_spin(p); - - fp->f_flags |= FP_WAITEVENT; - - error = fp_drop(p, fd, fp, 1); - - proc_fdunlock(p); - - return (error); + return error; } @@ -4064,35 +4224,39 @@ int fp_drop(proc_t p, int fd, struct fileproc *fp, int locked) { struct filedesc *fdp = p->p_fd; - int needwakeup = 0; + int needwakeup = 0; - if (!locked) + if (!locked) { proc_fdlock_spin(p); - if ((fp == FILEPROC_NULL) && (fd < 0 || fd >= fdp->fd_nfiles || - (fp = fdp->fd_ofiles[fd]) == NULL || - ((fdp->fd_ofileflags[fd] & UF_RESERVED) && - !(fdp->fd_ofileflags[fd] & UF_CLOSING)))) { - if (!locked) + } + if ((fp == FILEPROC_NULL) && (fd < 0 || fd >= fdp->fd_nfiles || + (fp = fdp->fd_ofiles[fd]) == NULL || + ((fdp->fd_ofileflags[fd] & UF_RESERVED) && + !(fdp->fd_ofileflags[fd] & UF_CLOSING)))) { + if (!locked) { proc_fdunlock(p); - return (EBADF); + } + return EBADF; } - fp->f_iocount--; - if (fp->f_iocount == 0) { - if (fp->f_flags & FP_SELCONFLICT) - fp->f_flags &= ~FP_SELCONFLICT; + if (1 == os_ref_release_locked(&fp->fp_iocount)) { + if (fp->fp_flags & FP_SELCONFLICT) { + fp->fp_flags &= ~FP_SELCONFLICT; + } if (p->p_fpdrainwait) { p->p_fpdrainwait = 0; needwakeup = 1; } } - if (!locked) + if (!locked) { proc_fdunlock(p); - if (needwakeup) - wakeup(&p->p_fpdrainwait); - - return (0); + } + if (needwakeup) { + wakeup(&p->p_fpdrainwait); + } + + return 0; } @@ -4116,7 +4280,7 @@ fp_drop(proc_t p, int fd, struct fileproc *fp, int locked) * Locks: This function internally takes and drops the proc_fdlock for * the current process * - * Notes: If successful, this function increments the f_iocount on the + * Notes: If successful, this function increments the fp_iocount on the * fd's corresponding fileproc. * * The fileproc referenced is not returned; because of this, care @@ -4125,7 +4289,7 @@ fp_drop(proc_t p, int fd, struct fileproc *fp, int locked) * not be recoverable from the vnode, if there is a subsequent * close that destroys the associate fileproc. The caller should * therefore retain their own reference on the fileproc so that - * the f_iocount can be dropped subsequently. Failure to do this + * the fp_iocount can be dropped subsequently. Failure to do this * can result in the returned pointer immediately becoming invalid * following the call. * @@ -4134,28 +4298,9 @@ fp_drop(proc_t p, int fd, struct fileproc *fp, int locked) int file_vnode(int fd, struct vnode **vpp) { - proc_t p = current_proc(); - struct fileproc *fp; - int error; - - proc_fdlock_spin(p); - if ( (error = fp_lookup(p, fd, &fp, 1)) ) { - proc_fdunlock(p); - return(error); - } - if (fp->f_type != DTYPE_VNODE) { - fp_drop(p, fd, fp,1); - proc_fdunlock(p); - return(EINVAL); - } - if (vpp != NULL) - *vpp = (struct vnode *)fp->f_data; - proc_fdunlock(p); - - return(0); + return file_vnode_withvid(fd, vpp, NULL); } - /* * file_vnode_withvid * @@ -4177,7 +4322,7 @@ file_vnode(int fd, struct vnode **vpp) * Locks: This function internally takes and drops the proc_fdlock for * the current process * - * Notes: If successful, this function increments the f_iocount on the + * Notes: If successful, this function increments the fp_iocount on the * fd's corresponding fileproc. * * The fileproc referenced is not returned; because of this, care @@ -4186,43 +4331,30 @@ file_vnode(int fd, struct vnode **vpp) * not be recoverable from the vnode, if there is a subsequent * close that destroys the associate fileproc. The caller should * therefore retain their own reference on the fileproc so that - * the f_iocount can be dropped subsequently. Failure to do this + * the fp_iocount can be dropped subsequently. Failure to do this * can result in the returned pointer immediately becoming invalid * following the call. * * Use of this function is discouraged. */ -int -file_vnode_withvid(int fd, struct vnode **vpp, uint32_t * vidp) -{ - proc_t p = current_proc(); - struct fileproc *fp; - vnode_t vp; - int error; - - proc_fdlock_spin(p); - if ( (error = fp_lookup(p, fd, &fp, 1)) ) { - proc_fdunlock(p); - return(error); - } - if (fp->f_type != DTYPE_VNODE) { - fp_drop(p, fd, fp,1); - proc_fdunlock(p); - return(EINVAL); - } - vp = (struct vnode *)fp->f_data; - if (vpp != NULL) - *vpp = vp; - - if ((vidp != NULL) && (vp != NULLVP)) - *vidp = (uint32_t)vp->v_id; - - proc_fdunlock(p); +int +file_vnode_withvid(int fd, struct vnode **vpp, uint32_t *vidp) +{ + struct fileproc *fp; + int error; - return(0); + error = fp_get_ftype(current_proc(), fd, DTYPE_VNODE, EINVAL, &fp); + if (error == 0) { + if (vpp) { + *vpp = fp->f_data; + } + if (vidp) { + *vidp = vnode_vid(fp->f_data); + } + } + return error; } - /* * file_socket * @@ -4242,7 +4374,7 @@ file_vnode_withvid(int fd, struct vnode **vpp, uint32_t * vidp) * Locks: This function internally takes and drops the proc_fdlock for * the current process * - * Notes: If successful, this function increments the f_iocount on the + * Notes: If successful, this function increments the fp_iocount on the * fd's corresponding fileproc. * * The fileproc referenced is not returned; because of this, care @@ -4251,7 +4383,7 @@ file_vnode_withvid(int fd, struct vnode **vpp, uint32_t * vidp) * not be recoverable from the socket, if there is a subsequent * close that destroys the associate fileproc. The caller should * therefore retain their own reference on the fileproc so that - * the f_iocount can be dropped subsequently. Failure to do this + * the fp_iocount can be dropped subsequently. Failure to do this * can result in the returned pointer immediately becoming invalid * following the call. * @@ -4260,24 +4392,16 @@ file_vnode_withvid(int fd, struct vnode **vpp, uint32_t * vidp) int file_socket(int fd, struct socket **sp) { - proc_t p = current_proc(); struct fileproc *fp; int error; - - proc_fdlock_spin(p); - if ( (error = fp_lookup(p, fd, &fp, 1)) ) { - proc_fdunlock(p); - return(error); - } - if (fp->f_type != DTYPE_SOCKET) { - fp_drop(p, fd, fp,1); - proc_fdunlock(p); - return(ENOTSOCK); - } - *sp = (struct socket *)fp->f_data; - proc_fdunlock(p); - return(0); + error = fp_get_ftype(current_proc(), fd, DTYPE_SOCKET, ENOTSOCK, &fp); + if (error == 0) { + if (sp) { + *sp = (struct socket *)fp->f_data; + } + } + return error; } @@ -4300,28 +4424,23 @@ file_socket(int fd, struct socket **sp) * * Locks: This function internally takes and drops the proc_fdlock for * the current process - * - * Notes: This function will internally increment and decrement the - * f_iocount of the fileproc as part of its operation. */ int file_flags(int fd, int *flags) { - proc_t p = current_proc(); struct fileproc *fp; - int error; - + int error = EBADF; + proc_fdlock_spin(p); - if ( (error = fp_lookup(p, fd, &fp, 1)) ) { - proc_fdunlock(p); - return(error); + fp = fp_get_noref_locked(p, fd); + if (fp) { + *flags = (int)fp->f_flag; + error = 0; } - *flags = (int)fp->f_flag; - fp_drop(p, fd, fp,1); proc_fdunlock(p); - return(0); + return error; } @@ -4336,51 +4455,34 @@ file_flags(int fd, int *flags) * to be dropped * * Returns: 0 Success - * EBADF Bad file descriptor * * Description: Given an fd, look it up in the current process's per process - * open file table, and drop it's fileproc's f_iocount by one + * open file table, and drop it's fileproc's fp_iocount by one * * Notes: This is intended as a corresponding operation to the functions * file_vnode() and file_socket() operations. * - * Technically, the close reference is supposed to be protected - * by a fileproc_drain(), however, a drain will only block if - * the fd refers to a character device, and that device has had - * preparefileread() called on it. If it refers to something - * other than a character device, then the drain will occur and - * block each close attempt, rather than merely the last close. - * - * Since it's possible for an fd that refers to a character - * device to have an intermediate close followed by an open to - * cause a different file to correspond to that descriptor, - * unless there was a cautionary reference taken on the fileproc, - * this is an inherently unsafe function. This happens in the - * case where multiple fd's in a process refer to the same - * character device (e.g. stdin/out/err pointing to a tty, etc.). + * If the caller can't possibly hold an I/O reference, + * this function will panic the kernel rather than allowing + * for memory corruption. Callers should always call this + * because they acquired an I/O reference on this file before. * * Use of this function is discouraged. */ -int +int file_drop(int fd) { struct fileproc *fp; proc_t p = current_proc(); - int needwakeup = 0; + int needwakeup = 0; proc_fdlock_spin(p); - if (fd < 0 || fd >= p->p_fd->fd_nfiles || - (fp = p->p_fd->fd_ofiles[fd]) == NULL || - ((p->p_fd->fd_ofileflags[fd] & UF_RESERVED) && - !(p->p_fd->fd_ofileflags[fd] & UF_CLOSING))) { - proc_fdunlock(p); - return (EBADF); - } - fp->f_iocount --; + fp = fp_get_noref_locked_with_iocount(p, fd); - if (fp->f_iocount == 0) { - if (fp->f_flags & FP_SELCONFLICT) - fp->f_flags &= ~FP_SELCONFLICT; + if (1 == os_ref_release_locked(&fp->fp_iocount)) { + if (fp->fp_flags & FP_SELCONFLICT) { + fp->fp_flags &= ~FP_SELCONFLICT; + } if (p->p_fpdrainwait) { p->p_fpdrainwait = 0; @@ -4389,77 +4491,16 @@ file_drop(int fd) } proc_fdunlock(p); - if (needwakeup) - wakeup(&p->p_fpdrainwait); - return(0); -} - - -static int falloc_withalloc_locked(proc_t, struct fileproc **, int *, - vfs_context_t, struct fileproc * (*)(void *), void *, int); - -/* - * falloc - * - * Description: Allocate an entry in the per process open file table and - * return the corresponding fileproc and fd. - * - * Parameters: p The process in whose open file - * table the fd is to be allocated - * resultfp Pointer to fileproc pointer - * return area - * resultfd Pointer to fd return area - * ctx VFS context - * - * Returns: 0 Success - * falloc:ENFILE Too many open files in system - * falloc:EMFILE Too many open files in process - * falloc:ENOMEM M_FILEPROC or M_FILEGLOB zone - * exhausted - * - * Implicit returns: - * *resultfd (modified) Returned fileproc pointer - * *resultfd (modified) Returned fd - * - * Locks: This function takes and drops the proc_fdlock; if this lock - * is already held, use falloc_locked() instead. - * - * Notes: This function takes separate process and context arguments - * solely to support kern_exec.c; otherwise, it would take - * neither, and expect falloc_locked() to use the - * vfs_context_current() routine internally. - */ -int -falloc(proc_t p, struct fileproc **resultfp, int *resultfd, vfs_context_t ctx) -{ - return (falloc_withalloc(p, resultfp, resultfd, ctx, - fileproc_alloc_init, NULL)); + if (needwakeup) { + wakeup(&p->p_fpdrainwait); + } + return 0; } -/* - * Like falloc, but including the fileproc allocator and create-args - */ -int -falloc_withalloc(proc_t p, struct fileproc **resultfp, int *resultfd, - vfs_context_t ctx, fp_allocfn_t fp_zalloc, void *arg) -{ - int error; - - proc_fdlock(p); - error = falloc_withalloc_locked(p, - resultfp, resultfd, ctx, fp_zalloc, arg, 1); - proc_fdunlock(p); - - return (error); -} -/* - * "uninitialized" ops -- ensure fg->fg_ops->fo_type always exists - */ -static const struct fileops uninitops; /* - * falloc_locked + * falloc_withalloc * * Create a new open file structure and allocate * a file descriptor for the process that refers to it. @@ -4475,67 +4516,51 @@ static const struct fileops uninitops; * return area * resultfd Pointer to fd return area * ctx VFS context - * locked Flag to indicate whether the - * caller holds proc_fdlock + * fp_zalloc fileproc allocator to use + * crarg allocator args * * Returns: 0 Success * ENFILE Too many open files in system * fdalloc:EMFILE Too many open files in process - * ENOMEM M_FILEPROC or M_FILEGLOB zone + * fdalloc:ENOMEM M_OFILETABL zone exhausted + * ENOMEM fp_zone or fg_zone zone * exhausted - * fdalloc:ENOMEM * * Implicit returns: * *resultfd (modified) Returned fileproc pointer * *resultfd (modified) Returned fd * - * Locks: If the parameter 'locked' is zero, this function takes and - * drops the proc_fdlock; if non-zero, the caller must hold the - * lock. - * - * Notes: If you intend to use a non-zero 'locked' parameter, use the - * utility function falloc() instead. - * - * This function takes separate process and context arguments + * Notes: This function takes separate process and context arguments * solely to support kern_exec.c; otherwise, it would take * neither, and use the vfs_context_current() routine internally. */ int -falloc_locked(proc_t p, struct fileproc **resultfp, int *resultfd, - vfs_context_t ctx, int locked) -{ - return (falloc_withalloc_locked(p, resultfp, resultfd, ctx, - fileproc_alloc_init, NULL, locked)); -} - -static int -falloc_withalloc_locked(proc_t p, struct fileproc **resultfp, int *resultfd, - vfs_context_t ctx, fp_allocfn_t fp_zalloc, void *crarg, - int locked) +falloc_withalloc(proc_t p, struct fileproc **resultfp, int *resultfd, + vfs_context_t ctx, fp_allocfn_t fp_zalloc, void *crarg) { struct fileproc *fp; struct fileglob *fg; int error, nfd; - if (!locked) - proc_fdlock(p); - if ( (error = fdalloc(p, 0, &nfd)) ) { - if (!locked) - proc_fdunlock(p); - return (error); - } + /* Make sure we don't go beyond the system-wide limit */ if (nfiles >= maxfiles) { - if (!locked) - proc_fdunlock(p); tablefull("file"); - return (ENFILE); + return ENFILE; + } + + proc_fdlock(p); + + /* fdalloc will make sure the process stays below per-process limit */ + if ((error = fdalloc(p, 0, &nfd))) { + proc_fdunlock(p); + return error; } + #if CONFIG_MACF error = mac_file_check_create(proc_ucred(p)); if (error) { - if (!locked) - proc_fdunlock(p); - return (error); + proc_fdunlock(p); + return error; } #endif @@ -4549,88 +4574,52 @@ falloc_withalloc_locked(proc_t p, struct fileproc **resultfp, int *resultfd, fp = (*fp_zalloc)(crarg); if (fp == NULL) { - if (locked) - proc_fdlock(p); - return (ENOMEM); - } - MALLOC_ZONE(fg, struct fileglob *, sizeof(struct fileglob), M_FILEGLOB, M_WAITOK); - if (fg == NULL) { - fileproc_free(fp); - if (locked) - proc_fdlock(p); - return (ENOMEM); + return ENOMEM; } - bzero(fg, sizeof(struct fileglob)); - lck_mtx_init(&fg->fg_lock, file_lck_grp, file_lck_attr); + fg = zalloc_flags(fg_zone, Z_WAITOK | Z_ZERO); + lck_mtx_init(&fg->fg_lock, &file_lck_grp, LCK_ATTR_NULL); - fp->f_iocount = 1; - fg->fg_count = 1; + os_ref_retain_locked(&fp->fp_iocount); + os_ref_init_raw(&fg->fg_count, &f_refgrp); fg->fg_ops = &uninitops; - fp->f_fglob = fg; + fp->fp_glob = fg; #if CONFIG_MACF mac_file_label_init(fg); #endif kauth_cred_ref(ctx->vc_ucred); - proc_fdlock(p); - fp->f_cred = ctx->vc_ucred; #if CONFIG_MACF mac_file_label_associate(fp->f_cred, fg); #endif - OSAddAtomic(1, &nfiles); + os_atomic_inc(&nfiles, relaxed); + + proc_fdlock(p); p->p_fd->fd_ofiles[nfd] = fp; - if (!locked) - proc_fdunlock(p); + proc_fdunlock(p); - if (resultfp) + if (resultfp) { *resultfp = fp; - if (resultfd) + } + if (resultfd) { *resultfd = nfd; + } - return (0); + return 0; } - -/* - * fg_free - * - * Description: Free a file structure; drop the global open file count, and - * drop the credential reference, if the fileglob has one, and - * destroy the instance mutex before freeing - * - * Parameters: fg Pointer to fileglob to be - * freed - * - * Returns: void - */ -void -fg_free(struct fileglob *fg) +int +falloc(proc_t p, struct fileproc **resultfp, int *resultfd, vfs_context_t ctx) { - OSAddAtomic(-1, &nfiles); - - if (fg->fg_vn_data) { - fg_vn_data_free(fg->fg_vn_data); - fg->fg_vn_data = NULL; - } - - if (IS_VALID_CRED(fg->fg_cred)) { - kauth_cred_unref(&fg->fg_cred); - } - lck_mtx_destroy(&fg->fg_lock, file_lck_grp); - -#if CONFIG_MACF - mac_file_label_destroy(fg); -#endif - FREE_ZONE(fg, sizeof *fg, M_FILEGLOB); + return falloc_withalloc(p, resultfp, resultfd, ctx, + fileproc_alloc_init, NULL); } - /* * fdexec * @@ -4648,18 +4637,39 @@ fg_free(struct fileglob *fg) * Returns: void * * Locks: This function internally takes and drops proc_fdlock() + * But assumes tables don't grow/change while unlocked. * */ void -fdexec(proc_t p, short flags) +fdexec(proc_t p, short flags, int self_exec) { struct filedesc *fdp = p->p_fd; int i; boolean_t cloexec_default = (flags & POSIX_SPAWN_CLOEXEC_DEFAULT) != 0; + thread_t self = current_thread(); + struct uthread *ut = get_bsdthread_info(self); + struct kqworkq *dealloc_kqwq = NULL; + + /* + * If the current thread is bound as a workq/workloop + * servicing thread, we need to unbind it first. + */ + if (ut->uu_kqr_bound && self_exec) { + kqueue_threadreq_unbind(p, ut->uu_kqr_bound); + } proc_fdlock(p); - for (i = fdp->fd_lastfile; i >= 0; i--) { + /* + * Deallocate the knotes for this process + * and mark the tables non-existent so + * subsequent kqueue closes go faster. + */ + knotes_dealloc(p); + assert(fdp->fd_knlistsize == 0); + assert(fdp->fd_knhashmask == 0); + + for (i = fdp->fd_lastfile; i >= 0; i--) { struct fileproc *fp = fdp->fd_ofiles[i]; char *flagp = &fdp->fd_ofileflags[i]; @@ -4670,41 +4680,35 @@ fdexec(proc_t p, short flags) * except files marked explicitly as "inherit" and * not marked close-on-exec. */ - if ((*flagp & (UF_EXCLOSE|UF_INHERIT)) != UF_INHERIT) + if ((*flagp & (UF_EXCLOSE | UF_INHERIT)) != UF_INHERIT) { *flagp |= UF_EXCLOSE; + } *flagp &= ~UF_INHERIT; } if ( - ((*flagp & (UF_RESERVED|UF_EXCLOSE)) == UF_EXCLOSE) + ((*flagp & (UF_RESERVED | UF_EXCLOSE)) == UF_EXCLOSE) #if CONFIG_MACF - || (fp && mac_file_check_inherit(proc_ucred(p), fp->f_fglob)) + || (fp && mac_file_check_inherit(proc_ucred(p), fp->fp_glob)) #endif - ) { - if (i < fdp->fd_knlistsize) - knote_fdclose(p, i); - procfdtbl_clearfd(p, i); - if (i == fdp->fd_lastfile && i > 0) - fdp->fd_lastfile--; - if (i < fdp->fd_freefile) - fdp->fd_freefile = i; + ) { + fp_close_and_unlock(p, i, fp, 0); + proc_fdlock(p); + } + } - /* - * Wait for any third party viewers (e.g., lsof) - * to release their references to this fileproc. - */ - while (fp->f_iocount > 0) { - p->p_fpdrainwait = 1; - msleep(&p->p_fpdrainwait, &p->p_fdmlock, PRIBIO, - "fpdrain", NULL); - } + /* release the per-process workq kq */ + if (fdp->fd_wqkqueue) { + dealloc_kqwq = fdp->fd_wqkqueue; + fdp->fd_wqkqueue = NULL; + } - closef_locked(fp, fp->f_fglob, p); + proc_fdunlock(p); - fileproc_free(fp); - } + /* Anything to free? */ + if (dealloc_kqwq) { + kqworkq_dealloc(dealloc_kqwq); } - proc_fdunlock(p); } @@ -4750,12 +4754,9 @@ fdcopy(proc_t p, vnode_t uth_cdir) struct filedesc *newfdp, *fdp = p->p_fd; int i; struct fileproc *ofp, *fp; - vnode_t v_dir; + vnode_t v_dir; - MALLOC_ZONE(newfdp, struct filedesc *, - sizeof(*newfdp), M_FILEDESC, M_WAITOK); - if (newfdp == NULL) - return(NULL); + newfdp = zalloc(fdp_zone); proc_fdlock(p); @@ -4769,8 +4770,9 @@ fdcopy(proc_t p, vnode_t uth_cdir) * inherit the new current working directory from the current thread * instead, before we take our references. */ - if (uth_cdir != NULLVP) + if (uth_cdir != NULLVP) { newfdp->fd_cdir = uth_cdir; + } /* * For both fd_cdir and fd_rdir make sure we get @@ -4780,16 +4782,18 @@ fdcopy(proc_t p, vnode_t uth_cdir) * and allows us to do the vnode_rele only on * a properly referenced vp */ - if ( (v_dir = newfdp->fd_cdir) ) { - if (vnode_getwithref(v_dir) == 0) { - if ( (vnode_ref(v_dir)) ) - newfdp->fd_cdir = NULL; + if ((v_dir = newfdp->fd_cdir)) { + if (vnode_getwithref(v_dir) == 0) { + if ((vnode_ref(v_dir))) { + newfdp->fd_cdir = NULL; + } vnode_put(v_dir); - } else - newfdp->fd_cdir = NULL; + } else { + newfdp->fd_cdir = NULL; + } } if (newfdp->fd_cdir == NULL && fdp->fd_cdir) { - /* + /* * we couldn't get a new reference on * the current working directory being * inherited... we might as well drop @@ -4798,22 +4802,24 @@ fdcopy(proc_t p, vnode_t uth_cdir) * it useless... by dropping it we'll * be that much closer to recycling it */ - vnode_rele(fdp->fd_cdir); + vnode_rele(fdp->fd_cdir); fdp->fd_cdir = NULL; } - if ( (v_dir = newfdp->fd_rdir) ) { + if ((v_dir = newfdp->fd_rdir)) { if (vnode_getwithref(v_dir) == 0) { - if ( (vnode_ref(v_dir)) ) - newfdp->fd_rdir = NULL; + if ((vnode_ref(v_dir))) { + newfdp->fd_rdir = NULL; + } vnode_put(v_dir); } else { - newfdp->fd_rdir = NULL; + newfdp->fd_rdir = NULL; } } /* Coming from a chroot environment and unable to get a reference... */ if (newfdp->fd_rdir == NULL && fdp->fd_rdir) { - /* + proc_fdunlock(p); + /* * We couldn't get a new reference on * the chroot directory being * inherited... this is fatal, since @@ -4821,10 +4827,11 @@ fdcopy(proc_t p, vnode_t uth_cdir) * escape from a chroot environment by * the new process. */ - if (newfdp->fd_cdir) - vnode_rele(newfdp->fd_cdir); - FREE_ZONE(newfdp, sizeof *newfdp, M_FILEDESC); - return(NULL); + if (newfdp->fd_cdir) { + vnode_rele(newfdp->fd_cdir); + } + zfree(fdp_zone, newfdp); + return NULL; } /* @@ -4833,32 +4840,34 @@ fdcopy(proc_t p, vnode_t uth_cdir) * additional memory for the number of descriptors currently * in use. */ - if (newfdp->fd_lastfile < NDFILE) + if (newfdp->fd_lastfile < NDFILE) { i = NDFILE; - else { + } else { /* * Compute the smallest multiple of NDEXTENT needed * for the file descriptors currently in use, * allowing the table to shrink. */ i = newfdp->fd_nfiles; - while (i > 1 + 2 * NDEXTENT && i > 1 + newfdp->fd_lastfile * 2) + while (i > 1 + 2 * NDEXTENT && i > 1 + newfdp->fd_lastfile * 2) { i /= 2; + } } proc_fdunlock(p); - MALLOC_ZONE(newfdp->fd_ofiles, struct fileproc **, - i * OFILESIZE, M_OFILETABL, M_WAITOK); + newfdp->fd_ofiles = kheap_alloc(KM_OFILETABL, i * OFILESIZE, + Z_WAITOK | Z_ZERO); if (newfdp->fd_ofiles == NULL) { - if (newfdp->fd_cdir) - vnode_rele(newfdp->fd_cdir); - if (newfdp->fd_rdir) + if (newfdp->fd_cdir) { + vnode_rele(newfdp->fd_cdir); + } + if (newfdp->fd_rdir) { vnode_rele(newfdp->fd_rdir); + } - FREE_ZONE(newfdp, sizeof(*newfdp), M_FILEDESC); - return(NULL); + zfree(fdp_zone, newfdp); + return NULL; } - (void) memset(newfdp->fd_ofiles, 0, i * OFILESIZE); proc_fdlock(p); newfdp->fd_ofileflags = (char *) &newfdp->fd_ofiles[i]; @@ -4869,44 +4878,20 @@ fdcopy(proc_t p, vnode_t uth_cdir) char *flags; (void) memcpy(newfdp->fd_ofiles, fdp->fd_ofiles, - (newfdp->fd_lastfile + 1) * sizeof(*fdp->fd_ofiles)); + (newfdp->fd_lastfile + 1) * sizeof(*fdp->fd_ofiles)); (void) memcpy(newfdp->fd_ofileflags, fdp->fd_ofileflags, - (newfdp->fd_lastfile + 1) * sizeof(*fdp->fd_ofileflags)); - - /* - * kq descriptors cannot be copied. - */ - if (newfdp->fd_knlistsize != -1) { - fpp = &newfdp->fd_ofiles[newfdp->fd_lastfile]; - flags = &newfdp->fd_ofileflags[newfdp->fd_lastfile]; - for (i = newfdp->fd_lastfile; - i >= 0; i--, fpp--, flags--) { - if (*flags & UF_RESERVED) - continue; /* (removed below) */ - if (*fpp != NULL && (*fpp)->f_type == DTYPE_KQUEUE) { - *fpp = NULL; - *flags = 0; - if (i < newfdp->fd_freefile) - newfdp->fd_freefile = i; - } - if (*fpp == NULL && i == newfdp->fd_lastfile && i > 0) - newfdp->fd_lastfile--; - } - newfdp->fd_knlist = NULL; - newfdp->fd_knlistsize = -1; - newfdp->fd_knhash = NULL; - newfdp->fd_knhashmask = 0; - } - fpp = newfdp->fd_ofiles; - flags = newfdp->fd_ofileflags; + (newfdp->fd_lastfile + 1) * sizeof(*fdp->fd_ofileflags)); - for (i = newfdp->fd_lastfile + 1; --i >= 0; fpp++, flags++) + fpp = &newfdp->fd_ofiles[newfdp->fd_lastfile]; + flags = &newfdp->fd_ofileflags[newfdp->fd_lastfile]; + for (i = newfdp->fd_lastfile; i >= 0; i--, fpp--, flags--) { if ((ofp = *fpp) != NULL && - 0 == (ofp->f_fglob->fg_lflags & FG_CONFINED) && - 0 == (*flags & (UF_FORKCLOSE|UF_RESERVED))) { + 0 == (ofp->fp_glob->fg_lflags & FG_CONFINED) && + 0 == (*flags & (UF_FORKCLOSE | UF_RESERVED))) { #if DEBUG - if (FILEPROC_TYPE(ofp) != FTYPE_SIMPLE) + if (FILEPROC_TYPE(ofp) != FTYPE_SIMPLE) { panic("complex fileproc"); + } #endif fp = fileproc_alloc_init(NULL); if (fp == NULL) { @@ -4916,22 +4901,43 @@ fdcopy(proc_t p, vnode_t uth_cdir) */ *fpp = NULL; } else { - fp->f_flags |= - (ofp->f_flags & ~FP_TYPEMASK); - fp->f_fglob = ofp->f_fglob; - (void)fg_ref(fp); + fp->fp_flags |= + (ofp->fp_flags & ~FP_TYPEMASK); + fp->fp_glob = ofp->fp_glob; + fg_ref(p, fp->fp_glob); *fpp = fp; } } else { - if (i < newfdp->fd_freefile) - newfdp->fd_freefile = i; *fpp = NULL; *flags = 0; } + if (*fpp == NULL) { + if (i == newfdp->fd_lastfile && i > 0) { + newfdp->fd_lastfile--; + } + if (i < newfdp->fd_freefile) { + newfdp->fd_freefile = i; + } + } + } } proc_fdunlock(p); - return (newfdp); + + /* + * Initialize knote and kqueue tracking structs + */ + newfdp->fd_knlist = NULL; + newfdp->fd_knlistsize = 0; + newfdp->fd_knhash = NULL; + newfdp->fd_knhashmask = 0; + newfdp->fd_kqhash = NULL; + newfdp->fd_kqhashmask = 0; + newfdp->fd_wqkqueue = NULL; + lck_mtx_init(&newfdp->fd_kqhashlock, &proc_kqhashlock_grp, &proc_lck_attr); + lck_mtx_init(&newfdp->fd_knhashlock, &proc_knhashlock_grp, &proc_lck_attr); + + return newfdp; } @@ -4954,158 +4960,86 @@ fdfree(proc_t p) { struct filedesc *fdp; struct fileproc *fp; + struct kqworkq *dealloc_kqwq = NULL; int i; proc_fdlock(p); if (p == kernproc || NULL == (fdp = p->p_fd)) { - proc_fdunlock(p); + proc_fdunlock(p); return; } extern struct filedesc filedesc0; - if (&filedesc0 == fdp) + if (&filedesc0 == fdp) { panic("filedesc0"); - - if (fdp->fd_nfiles > 0 && fdp->fd_ofiles) { - for (i = fdp->fd_lastfile; i >= 0; i--) { - if ((fp = fdp->fd_ofiles[i]) != NULL) { - - if (fdp->fd_ofileflags[i] & UF_RESERVED) - panic("fdfree: found fp with UF_RESERVED"); - - procfdtbl_reservefd(p, i); - - if (i < fdp->fd_knlistsize) - knote_fdclose(p, i); - if (fp->f_flags & FP_WAITEVENT) - (void)waitevent_close(p, fp); - (void) closef_locked(fp, fp->f_fglob, p); - fileproc_free(fp); - } - } - FREE_ZONE(fdp->fd_ofiles, fdp->fd_nfiles * OFILESIZE, M_OFILETABL); - fdp->fd_ofiles = NULL; - fdp->fd_nfiles = 0; - } - - proc_fdunlock(p); - - if (fdp->fd_cdir) - vnode_rele(fdp->fd_cdir); - if (fdp->fd_rdir) - vnode_rele(fdp->fd_rdir); - - proc_fdlock_spin(p); - p->p_fd = NULL; - proc_fdunlock(p); - - if (fdp->fd_knlist) - FREE(fdp->fd_knlist, M_KQUEUE); - if (fdp->fd_knhash) - FREE(fdp->fd_knhash, M_KQUEUE); - - FREE_ZONE(fdp, sizeof(*fdp), M_FILEDESC); -} - -/* - * closef_locked - * - * Description: Internal form of closef; called with proc_fdlock held - * - * Parameters: fp Pointer to fileproc for fd - * fg Pointer to fileglob for fd - * p Pointer to proc structure - * - * Returns: 0 Success - * closef_finish:??? Anything returnable by a per-fileops - * close function - * - * Note: Decrements reference count on file structure; if this was the - * last reference, then closef_finish() is called - * - * p and fp are allowed to be NULL when closing a file that was - * being passed in a message (but only if we are called when this - * is NOT the last reference). - */ -int -closef_locked(struct fileproc *fp, struct fileglob *fg, proc_t p) -{ - struct vnode *vp; - struct flock lf; - struct vfs_context context; - int error; - - if (fg == NULL) { - return (0); } - /* Set up context with cred stashed in fg */ - if (p == current_proc()) - context.vc_thread = current_thread(); - else - context.vc_thread = NULL; - context.vc_ucred = fg->fg_cred; - /* - * POSIX record locking dictates that any close releases ALL - * locks owned by this process. This is handled by setting - * a flag in the unlock to free ONLY locks obeying POSIX - * semantics, and not to free BSD-style file locks. - * If the descriptor was in a message, POSIX-style locks - * aren't passed with the descriptor. + * deallocate all the knotes up front and claim empty + * tables to make any subsequent kqueue closes faster. */ - if (p && (p->p_ladvflag & P_LADVLOCK) && - DTYPE_VNODE == FILEGLOB_DTYPE(fg)) { - proc_fdunlock(p); + knotes_dealloc(p); + assert(fdp->fd_knlistsize == 0); + assert(fdp->fd_knhashmask == 0); - lf.l_whence = SEEK_SET; - lf.l_start = 0; - lf.l_len = 0; - lf.l_type = F_UNLCK; - vp = (struct vnode *)fg->fg_data; + /* + * dealloc all workloops that have outstanding retains + * when created with scheduling parameters. + */ + kqworkloops_dealloc(p); - if ( (error = vnode_getwithref(vp)) == 0 ) { - (void) VNOP_ADVLOCK(vp, (caddr_t)p, F_UNLCK, &lf, F_POSIX, &context, NULL); - (void)vnode_put(vp); + /* close file descriptors */ + if (fdp->fd_nfiles > 0 && fdp->fd_ofiles) { + for (i = fdp->fd_lastfile; i >= 0; i--) { + if ((fp = fdp->fd_ofiles[i]) != NULL) { + if (fdp->fd_ofileflags[i] & UF_RESERVED) { + panic("fdfree: found fp with UF_RESERVED"); + } + fp_close_and_unlock(p, i, fp, 0); + proc_fdlock(p); + } } - proc_fdlock(p); + kheap_free(KM_OFILETABL, fdp->fd_ofiles, fdp->fd_nfiles * OFILESIZE); + fdp->fd_ofiles = NULL; + fdp->fd_nfiles = 0; } - lck_mtx_lock_spin(&fg->fg_lock); - fg->fg_count--; - if (fg->fg_count > 0) { - lck_mtx_unlock(&fg->fg_lock); - return (0); + if (fdp->fd_wqkqueue) { + dealloc_kqwq = fdp->fd_wqkqueue; + fdp->fd_wqkqueue = NULL; } -#if DIAGNOSTIC - if (fg->fg_count != 0) - panic("fg %p: being freed with bad fg_count (%d)", fg, fg->fg_count); -#endif - if (fp && (fp->f_flags & FP_WRITTEN)) - fg->fg_flag |= FWASWRITTEN; + proc_fdunlock(p); - fg->fg_lflags |= FG_TERM; - lck_mtx_unlock(&fg->fg_lock); + if (dealloc_kqwq) { + kqworkq_dealloc(dealloc_kqwq); + } + if (fdp->fd_cdir) { + vnode_rele(fdp->fd_cdir); + } + if (fdp->fd_rdir) { + vnode_rele(fdp->fd_rdir); + } - if (p) - proc_fdunlock(p); + proc_fdlock_spin(p); + p->p_fd = NULL; + proc_fdunlock(p); - /* Since we ensure that fg->fg_ops is always initialized, - * it is safe to invoke fo_close on the fg */ - error = fo_close(fg, &context); + if (fdp->fd_kqhash) { + for (uint32_t j = 0; j <= fdp->fd_kqhashmask; j++) { + assert(LIST_EMPTY(&fdp->fd_kqhash[j])); + } + hashdestroy(fdp->fd_kqhash, M_KQUEUE, fdp->fd_kqhashmask); + } - fg_free(fg); - - if (p) - proc_fdlock(p); + lck_mtx_destroy(&fdp->fd_kqhashlock, &proc_kqhashlock_grp); + lck_mtx_destroy(&fdp->fd_knhashlock, &proc_knhashlock_grp); - return(error); + zfree(fdp_zone, fdp); } - /* * fileproc_drain * @@ -5124,47 +5058,65 @@ closef_locked(struct fileproc *fp, struct fileglob *fg, proc_t p) * close to prevent fd's from being closed out from under * operations currently in progress and blocked * - * See Also: file_vnode(), file_socket(), file_drop(), and the cautions + * See Also: file_vnode(), file_socket(), file_drop(), and the cautions * regarding their use and interaction with this function. */ void fileproc_drain(proc_t p, struct fileproc * fp) { struct vfs_context context; + thread_t thread; + bool is_current_proc; - context.vc_thread = proc_thread(p); /* XXX */ - context.vc_ucred = fp->f_fglob->fg_cred; + is_current_proc = (p == current_proc()); - fp->f_iocount-- ; /* (the one the close holds) */ + if (!is_current_proc) { + proc_lock(p); + thread = proc_thread(p); /* XXX */ + thread_reference(thread); + proc_unlock(p); + } else { + thread = current_thread(); + } - while (fp->f_iocount) { + context.vc_thread = thread; + context.vc_ucred = fp->fp_glob->fg_cred; - lck_mtx_convert_spin(&p->p_fdmlock); + /* Set the vflag for drain */ + fileproc_modify_vflags(fp, FPV_DRAIN, FALSE); - if (fp->f_fglob->fg_ops->fo_drain) { - (*fp->f_fglob->fg_ops->fo_drain)(fp, &context); - } - if ((fp->f_flags & FP_INSELECT) == FP_INSELECT) { - if (waitq_wakeup64_all((struct waitq *)fp->f_wset, NO_EVENT64, - THREAD_INTERRUPTED, WAITQ_ALL_PRIORITIES) == KERN_INVALID_ARGUMENT) - panic("bad wait queue for waitq_wakeup64_all %p (fp:%p)", fp->f_wset, fp); + while (os_ref_get_count(&fp->fp_iocount) > 1) { + lck_mtx_convert_spin(&p->p_fdmlock); + + fo_drain(fp, &context); + if ((fp->fp_flags & FP_INSELECT) == FP_INSELECT) { + if (waitq_wakeup64_all((struct waitq *)fp->fp_wset, NO_EVENT64, + THREAD_INTERRUPTED, WAITQ_ALL_PRIORITIES) == KERN_INVALID_ARGUMENT) { + panic("bad wait queue for waitq_wakeup64_all %p (fp:%p)", fp->fp_wset, fp); + } } - if ((fp->f_flags & FP_SELCONFLICT) == FP_SELCONFLICT) { + if ((fp->fp_flags & FP_SELCONFLICT) == FP_SELCONFLICT) { if (waitq_wakeup64_all(&select_conflict_queue, NO_EVENT64, - THREAD_INTERRUPTED, WAITQ_ALL_PRIORITIES) == KERN_INVALID_ARGUMENT) + THREAD_INTERRUPTED, WAITQ_ALL_PRIORITIES) == KERN_INVALID_ARGUMENT) { panic("bad select_conflict_queue"); + } } p->p_fpdrainwait = 1; msleep(&p->p_fpdrainwait, &p->p_fdmlock, PRIBIO, "fpdrain", NULL); - } #if DIAGNOSTIC - if ((fp->f_flags & FP_INSELECT) != 0) + if ((fp->fp_flags & FP_INSELECT) != 0) { panic("FP_INSELECT set on drained fp"); + } #endif - if ((fp->f_flags & FP_SELCONFLICT) == FP_SELCONFLICT) - fp->f_flags &= ~FP_SELCONFLICT; + if ((fp->fp_flags & FP_SELCONFLICT) == FP_SELCONFLICT) { + fp->fp_flags &= ~FP_SELCONFLICT; + } + + if (!is_current_proc) { + thread_deallocate(thread); + } } @@ -5178,27 +5130,22 @@ fileproc_drain(proc_t p, struct fileproc * fp) * Parameters: p Process containing fd * fd fd to be released * fp fileproc to be freed - * - * Returns: 0 Success - * - * Notes: XXX function should be void - no one interprets the returns - * XXX code */ -int +void fp_free(proc_t p, int fd, struct fileproc * fp) { - proc_fdlock_spin(p); + proc_fdlock_spin(p); fdrelse(p, fd); - proc_fdunlock(p); + proc_fdunlock(p); - fg_free(fp->f_fglob); + fg_free(fp->fp_glob); + os_ref_release_live(&fp->fp_iocount); fileproc_free(fp); - return(0); } /* - * flock + * sys_flock * * Description: Apply an advisory lock on a file descriptor. * @@ -5207,7 +5154,7 @@ fp_free(proc_t p, int fd, struct fileproc * fp) * attempted * uap->how (Un)Lock bits, including type * retval Pointer to the call return area - * + * * Returns: 0 Success * fp_getfvp:EBADF Bad file descriptor * fp_getfvp:ENOTSUP fd does not refer to a vnode @@ -5221,7 +5168,7 @@ fp_free(proc_t p, int fd, struct fileproc * fp) * the entire file (l_whence = SEEK_SET, l_start = 0, l_len = 0). */ int -flock(proc_t p, struct flock_args *uap, __unused int32_t *retval) +sys_flock(proc_t p, struct flock_args *uap, __unused int32_t *retval) { int fd = uap->fd; int how = uap->how; @@ -5229,13 +5176,13 @@ flock(proc_t p, struct flock_args *uap, __unused int32_t *retval) struct vnode *vp; struct flock lf; vfs_context_t ctx = vfs_context_current(); - int error=0; + int error = 0; AUDIT_ARG(fd, uap->fd); - if ( (error = fp_getfvp(p, fd, &fp, &vp)) ) { - return(error); + if ((error = fp_getfvp(p, fd, &fp, &vp))) { + return error; } - if ( (error = vnode_getwithref(vp)) ) { + if ((error = vnode_getwithref(vp))) { goto out1; } AUDIT_ARG(vnpath, vp, ARG_VNODE1); @@ -5245,57 +5192,56 @@ flock(proc_t p, struct flock_args *uap, __unused int32_t *retval) lf.l_len = 0; if (how & LOCK_UN) { lf.l_type = F_UNLCK; - fp->f_flag &= ~FHASLOCK; - error = VNOP_ADVLOCK(vp, (caddr_t)fp->f_fglob, F_UNLCK, &lf, F_FLOCK, ctx, NULL); + error = VNOP_ADVLOCK(vp, (caddr_t)fp->fp_glob, F_UNLCK, &lf, F_FLOCK, ctx, NULL); goto out; } - if (how & LOCK_EX) + if (how & LOCK_EX) { lf.l_type = F_WRLCK; - else if (how & LOCK_SH) + } else if (how & LOCK_SH) { lf.l_type = F_RDLCK; - else { - error = EBADF; + } else { + error = EBADF; goto out; } #if CONFIG_MACF - error = mac_file_check_lock(proc_ucred(p), fp->f_fglob, F_SETLK, &lf); - if (error) + error = mac_file_check_lock(proc_ucred(p), fp->fp_glob, F_SETLK, &lf); + if (error) { goto out; + } #endif - fp->f_flag |= FHASLOCK; - if (how & LOCK_NB) { - error = VNOP_ADVLOCK(vp, (caddr_t)fp->f_fglob, F_SETLK, &lf, F_FLOCK, ctx, NULL); - goto out; + error = VNOP_ADVLOCK(vp, (caddr_t)fp->fp_glob, F_SETLK, &lf, + (how & LOCK_NB ? F_FLOCK : F_FLOCK | F_WAIT), + ctx, NULL); + if (!error) { + os_atomic_or(&fp->fp_glob->fg_flag, FWASLOCKED, relaxed); } - error = VNOP_ADVLOCK(vp, (caddr_t)fp->f_fglob, F_SETLK, &lf, F_FLOCK|F_WAIT, ctx, NULL); out: (void)vnode_put(vp); out1: fp_drop(p, fd, fp, 0); - return(error); - + return error; } /* - * fileport_makeport + * sys_fileport_makeport * * Description: Obtain a Mach send right for a given file descriptor. * * Parameters: p Process calling fileport - * uap->fd The fd to reference - * uap->portnamep User address at which to place port name. + * uap->fd The fd to reference + * uap->portnamep User address at which to place port name. * * Returns: 0 Success. - * EBADF Bad file descriptor. - * EINVAL File descriptor had type that cannot be sent, misc. other errors. - * EFAULT Address at which to store port name is not valid. - * EAGAIN Resource shortage. + * EBADF Bad file descriptor. + * EINVAL File descriptor had type that cannot be sent, misc. other errors. + * EFAULT Address at which to store port name is not valid. + * EAGAIN Resource shortage. * * Implicit returns: - * On success, name of send right is stored at user-specified address. + * On success, name of send right is stored at user-specified address. */ int -fileport_makeport(proc_t p, struct fileport_makeport_args *uap, +sys_fileport_makeport(proc_t p, struct fileport_makeport_args *uap, __unused int *retval) { int err; @@ -5312,37 +5258,37 @@ fileport_makeport(proc_t p, struct fileport_makeport_args *uap, goto out_unlock; } - if (!file_issendable(p, fp)) { + fg = fp->fp_glob; + if (!fg_sendable(fg)) { err = EINVAL; goto out_unlock; } - if (FP_ISGUARDED(fp, GUARD_FILEPORT)) { + if (fp_isguarded(fp, GUARD_FILEPORT)) { err = fp_guard_exception(p, fd, fp, kGUARD_EXC_FILEPORT); goto out_unlock; } /* Dropped when port is deallocated */ - fg = fp->f_fglob; - fg_ref(fp); + fg_ref(p, fg); proc_fdunlock(p); /* Allocate and initialize a port */ fileport = fileport_alloc(fg); if (fileport == IPC_PORT_NULL) { + fg_drop_live(fg); err = EAGAIN; - fg_drop(fp); goto out; } - + /* Add an entry. Deallocates port on failure. */ name = ipc_port_copyout_send(fileport, get_task_ipcspace(p->task)); if (!MACH_PORT_VALID(name)) { err = EINVAL; goto out; - } - + } + err = copyout(&name, user_portaddr, sizeof(mach_port_name_t)); if (err != 0) { goto out; @@ -5375,20 +5321,14 @@ out: void fileport_releasefg(struct fileglob *fg) { - (void)closef_locked(NULL, fg, PROC_NULL); - - return; + (void)fg_drop(PROC_NULL, fg); } - /* * fileport_makefd * * Description: Obtain the file descriptor for a given Mach send right. * - * Parameters: p Process calling fileport - * uap->port Name of send right to file port. - * * Returns: 0 Success * EINVAL Invalid Mach port name, or port is not for a file. * fdalloc:EMFILE @@ -5398,24 +5338,13 @@ fileport_releasefg(struct fileglob *fg) * *retval (modified) The new descriptor */ int -fileport_makefd(proc_t p, struct fileport_makefd_args *uap, int32_t *retval) +fileport_makefd(proc_t p, ipc_port_t port, int uf_flags, int *retval) { struct fileglob *fg; - struct fileproc *fp = FILEPROC_NULL; - ipc_port_t port = IPC_PORT_NULL; - mach_port_name_t send = uap->port; - kern_return_t res; + struct fileproc *fp = FILEPROC_NULL; int fd; int err; - res = ipc_object_copyin(get_task_ipcspace(p->task), - send, MACH_MSG_TYPE_COPY_SEND, &port); - - if (res != KERN_SUCCESS) { - err = EINVAL; - goto out; - } - fg = fileport_port_to_fileglob(port); if (fg == NULL) { err = EINVAL; @@ -5428,16 +5357,18 @@ fileport_makefd(proc_t p, struct fileport_makefd_args *uap, int32_t *retval) goto out; } - fp->f_fglob = fg; - fg_ref(fp); - - proc_fdlock(p); + proc_fdlock(p); err = fdalloc(p, 0, &fd); if (err != 0) { proc_fdunlock(p); goto out; } - *fdflags(p, fd) |= UF_EXCLOSE; + if (uf_flags) { + *fdflags(p, fd) |= uf_flags; + } + + fp->fp_glob = fg; + fg_ref(p, fg); procfdtbl_releasefd(p, fd, fp); proc_fdunlock(p); @@ -5447,7 +5378,43 @@ fileport_makefd(proc_t p, struct fileport_makefd_args *uap, int32_t *retval) out: if ((fp != NULL) && (0 != err)) { fileproc_free(fp); - } + } + + return err; +} + +/* + * sys_fileport_makefd + * + * Description: Obtain the file descriptor for a given Mach send right. + * + * Parameters: p Process calling fileport + * uap->port Name of send right to file port. + * + * Returns: 0 Success + * EINVAL Invalid Mach port name, or port is not for a file. + * fdalloc:EMFILE + * fdalloc:ENOMEM Unable to allocate fileproc or extend file table. + * + * Implicit returns: + * *retval (modified) The new descriptor + */ +int +sys_fileport_makefd(proc_t p, struct fileport_makefd_args *uap, int32_t *retval) +{ + ipc_port_t port = IPC_PORT_NULL; + mach_port_name_t send = uap->port; + kern_return_t res; + int err; + + res = ipc_object_copyin(get_task_ipcspace(p->task), + send, MACH_MSG_TYPE_COPY_SEND, &port, 0, NULL, IPC_OBJECT_COPYIN_FLAGS_ALLOW_IMMOVABLE_SEND); + + if (res == KERN_SUCCESS) { + err = fileport_makefd(p, port, UF_EXCLOSE, retval); + } else { + err = EINVAL; + } if (IPC_PORT_NULL != port) { ipc_port_release_send(port); @@ -5498,17 +5465,16 @@ dupfdopen(struct filedesc *fdp, int indx, int dfd, int flags, int error) fp = fdp->fd_ofiles[indx]; if (dfd < 0 || dfd >= fdp->fd_nfiles || - (wfp = fdp->fd_ofiles[dfd]) == NULL || wfp == fp || - (fdp->fd_ofileflags[dfd] & UF_RESERVED)) { - - proc_fdunlock(p); - return (EBADF); + (wfp = fdp->fd_ofiles[dfd]) == NULL || wfp == fp || + (fdp->fd_ofileflags[dfd] & UF_RESERVED)) { + proc_fdunlock(p); + return EBADF; } #if CONFIG_MACF - myerror = mac_file_check_dup(proc_ucred(p), wfp->f_fglob, dfd); + myerror = mac_file_check_dup(proc_ucred(p), wfp->fp_glob, dfd); if (myerror) { proc_fdunlock(p); - return (myerror); + return myerror; } #endif /* @@ -5525,232 +5491,43 @@ dupfdopen(struct filedesc *fdp, int indx, int dfd, int flags, int error) */ switch (error) { case ENODEV: - if (FP_ISGUARDED(wfp, GUARD_DUP)) { + if (fp_isguarded(wfp, GUARD_DUP)) { proc_fdunlock(p); - return (EPERM); + return EPERM; } /* * Check that the mode the file is being opened for is a * subset of the mode of the existing descriptor. */ - if (((flags & (FREAD|FWRITE)) | wfp->f_flag) != wfp->f_flag) { - proc_fdunlock(p); - return (EACCES); + if (((flags & (FREAD | FWRITE)) | wfp->f_flag) != wfp->f_flag) { + proc_fdunlock(p); + return EACCES; } - if (indx > fdp->fd_lastfile) + if (indx > fdp->fd_lastfile) { fdp->fd_lastfile = indx; - (void)fg_ref(wfp); + } - if (fp->f_fglob) - fg_free(fp->f_fglob); - fp->f_fglob = wfp->f_fglob; + if (fp->fp_glob) { + fg_free(fp->fp_glob); + } + fg_ref(p, wfp->fp_glob); + fp->fp_glob = wfp->fp_glob; fdp->fd_ofileflags[indx] = fdp->fd_ofileflags[dfd] | - (flags & O_CLOEXEC) ? UF_EXCLOSE : 0; + (flags & O_CLOEXEC) ? UF_EXCLOSE : 0; - proc_fdunlock(p); - return (0); + proc_fdunlock(p); + return 0; default: - proc_fdunlock(p); - return (error); + proc_fdunlock(p); + return error; } /* NOTREACHED */ } -/* - * fg_ref - * - * Description: Add a reference to a fileglob by fileproc - * - * Parameters: fp fileproc containing fileglob - * pointer - * - * Returns: void - * - * Notes: XXX Should use OSAddAtomic? - */ -void -fg_ref(struct fileproc * fp) -{ - struct fileglob *fg; - - fg = fp->f_fglob; - - lck_mtx_lock_spin(&fg->fg_lock); - -#if DIAGNOSTIC - if ((fp->f_flags & ~((unsigned int)FP_VALID_FLAGS)) != 0) - panic("fg_ref: invalid bits on fp %p", fp); - - if (fg->fg_count == 0) - panic("fg_ref: adding fgcount to zeroed fg: fp %p fg %p", - fp, fg); -#endif - fg->fg_count++; - lck_mtx_unlock(&fg->fg_lock); -} - - -/* - * fg_drop - * - * Description: Remove a reference to a fileglob by fileproc - * - * Parameters: fp fileproc containing fileglob - * pointer - * - * Returns: void - * - * Notes: XXX Should use OSAddAtomic? - */ -void -fg_drop(struct fileproc * fp) -{ - struct fileglob *fg; - - fg = fp->f_fglob; - lck_mtx_lock_spin(&fg->fg_lock); - fg->fg_count--; - lck_mtx_unlock(&fg->fg_lock); -} - -#if SOCKETS -/* - * fg_insertuipc_mark - * - * Description: Mark fileglob for insertion onto message queue if needed - * Also takes fileglob reference - * - * Parameters: fg Fileglob pointer to insert - * - * Returns: true, if the fileglob needs to be inserted onto msg queue - * - * Locks: Takes and drops fg_lock, potentially many times - */ -boolean_t -fg_insertuipc_mark(struct fileglob * fg) -{ - boolean_t insert = FALSE; - - lck_mtx_lock_spin(&fg->fg_lock); - while (fg->fg_lflags & FG_RMMSGQ) { - lck_mtx_convert_spin(&fg->fg_lock); - - fg->fg_lflags |= FG_WRMMSGQ; - msleep(&fg->fg_lflags, &fg->fg_lock, 0, "fg_insertuipc", NULL); - } - - fg->fg_count++; - fg->fg_msgcount++; - if (fg->fg_msgcount == 1) { - fg->fg_lflags |= FG_INSMSGQ; - insert = TRUE; - } - lck_mtx_unlock(&fg->fg_lock); - return (insert); -} - -/* - * fg_insertuipc - * - * Description: Insert marked fileglob onto message queue - * - * Parameters: fg Fileglob pointer to insert - * - * Returns: void - * - * Locks: Takes and drops fg_lock & uipc_lock - * DO NOT call this function with proc_fdlock held as unp_gc() - * can potentially try to acquire proc_fdlock, which can result - * in a deadlock if this function is in unp_gc_wait(). - */ -void -fg_insertuipc(struct fileglob * fg) -{ - if (fg->fg_lflags & FG_INSMSGQ) { - lck_mtx_lock_spin(uipc_lock); - unp_gc_wait(); - LIST_INSERT_HEAD(&fmsghead, fg, f_msglist); - lck_mtx_unlock(uipc_lock); - lck_mtx_lock(&fg->fg_lock); - fg->fg_lflags &= ~FG_INSMSGQ; - if (fg->fg_lflags & FG_WINSMSGQ) { - fg->fg_lflags &= ~FG_WINSMSGQ; - wakeup(&fg->fg_lflags); - } - lck_mtx_unlock(&fg->fg_lock); - } -} - -/* - * fg_removeuipc_mark - * - * Description: Mark the fileglob for removal from message queue if needed - * Also releases fileglob message queue reference - * - * Parameters: fg Fileglob pointer to remove - * - * Returns: true, if the fileglob needs to be removed from msg queue - * - * Locks: Takes and drops fg_lock, potentially many times - */ -boolean_t -fg_removeuipc_mark(struct fileglob * fg) -{ - boolean_t remove = FALSE; - - lck_mtx_lock_spin(&fg->fg_lock); - while (fg->fg_lflags & FG_INSMSGQ) { - lck_mtx_convert_spin(&fg->fg_lock); - - fg->fg_lflags |= FG_WINSMSGQ; - msleep(&fg->fg_lflags, &fg->fg_lock, 0, "fg_removeuipc", NULL); - } - fg->fg_msgcount--; - if (fg->fg_msgcount == 0) { - fg->fg_lflags |= FG_RMMSGQ; - remove = TRUE; - } - lck_mtx_unlock(&fg->fg_lock); - return (remove); -} - -/* - * fg_removeuipc - * - * Description: Remove marked fileglob from message queue - * - * Parameters: fg Fileglob pointer to remove - * - * Returns: void - * - * Locks: Takes and drops fg_lock & uipc_lock - * DO NOT call this function with proc_fdlock held as unp_gc() - * can potentially try to acquire proc_fdlock, which can result - * in a deadlock if this function is in unp_gc_wait(). - */ -void -fg_removeuipc(struct fileglob * fg) -{ - if (fg->fg_lflags & FG_RMMSGQ) { - lck_mtx_lock_spin(uipc_lock); - unp_gc_wait(); - LIST_REMOVE(fg, f_msglist); - lck_mtx_unlock(uipc_lock); - lck_mtx_lock(&fg->fg_lock); - fg->fg_lflags &= ~FG_RMMSGQ; - if (fg->fg_lflags & FG_WRMMSGQ) { - fg->fg_lflags &= ~FG_WRMMSGQ; - wakeup(&fg->fg_lflags); - } - lck_mtx_unlock(&fg->fg_lock); - } -} -#endif /* SOCKETS */ - /* * fo_read * @@ -5768,7 +5545,14 @@ fg_removeuipc(struct fileglob * fg) int fo_read(struct fileproc *fp, struct uio *uio, int flags, vfs_context_t ctx) { - return ((*fp->f_ops->fo_read)(fp, uio, flags, ctx)); + return (*fp->f_ops->fo_read)(fp, uio, flags, ctx); +} + +int +fo_no_read(struct fileproc *fp, struct uio *uio, int flags, vfs_context_t ctx) +{ +#pragma unused(fp, uio, flags, ctx) + return ENXIO; } @@ -5789,7 +5573,14 @@ fo_read(struct fileproc *fp, struct uio *uio, int flags, vfs_context_t ctx) int fo_write(struct fileproc *fp, struct uio *uio, int flags, vfs_context_t ctx) { - return((*fp->f_ops->fo_write)(fp, uio, flags, ctx)); + return (*fp->f_ops->fo_write)(fp, uio, flags, ctx); +} + +int +fo_no_write(struct fileproc *fp, struct uio *uio, int flags, vfs_context_t ctx) +{ +#pragma unused(fp, uio, flags, ctx) + return ENXIO; } @@ -5815,7 +5606,7 @@ fo_write(struct fileproc *fp, struct uio *uio, int flags, vfs_context_t ctx) * function, it will need to revalidate/reacquire any cached * protected data obtained prior to the call. */ -int +int fo_ioctl(struct fileproc *fp, u_long com, caddr_t data, vfs_context_t ctx) { int error; @@ -5823,8 +5614,15 @@ fo_ioctl(struct fileproc *fp, u_long com, caddr_t data, vfs_context_t ctx) proc_fdunlock(vfs_context_proc(ctx)); error = (*fp->f_ops->fo_ioctl)(fp, com, data, ctx); proc_fdlock(vfs_context_proc(ctx)); - return(error); -} + return error; +} + +int +fo_no_ioctl(struct fileproc *fp, u_long com, caddr_t data, vfs_context_t ctx) +{ +#pragma unused(fp, com, data, ctx) + return ENOTTY; +} /* @@ -5843,8 +5641,15 @@ fo_ioctl(struct fileproc *fp, u_long com, caddr_t data, vfs_context_t ctx) */ int fo_select(struct fileproc *fp, int which, void *wql, vfs_context_t ctx) -{ - return((*fp->f_ops->fo_select)(fp, which, wql, ctx)); +{ + return (*fp->f_ops->fo_select)(fp, which, wql, ctx); +} + +int +fo_no_select(struct fileproc *fp, int which, void *wql, vfs_context_t ctx) +{ +#pragma unused(fp, which, wql, ctx) + return ENOTSUP; } @@ -5863,75 +5668,109 @@ fo_select(struct fileproc *fp, int which, void *wql, vfs_context_t ctx) */ int fo_close(struct fileglob *fg, vfs_context_t ctx) -{ - return((*fg->fg_ops->fo_close)(fg, ctx)); +{ + return (*fg->fg_ops->fo_close)(fg, ctx); } /* - * fo_kqfilter + * fo_drain * * Description: Generic fileops kqueue filter indirected through the fileops * pointer in the fileproc structure * * Parameters: fp fileproc structure pointer - * kn pointer to knote to filter on * ctx VFS context for operation * * Returns: 0 Success - * !0 Errno from kqueue filter + * !0 errno from drain */ int -fo_kqfilter(struct fileproc *fp, struct knote *kn, vfs_context_t ctx) +fo_drain(struct fileproc *fp, vfs_context_t ctx) +{ + return (*fp->f_ops->fo_drain)(fp, ctx); +} + +int +fo_no_drain(struct fileproc *fp, vfs_context_t ctx) { - return ((*fp->f_ops->fo_kqfilter)(fp, kn, ctx)); +#pragma unused(fp, ctx) + return ENOTSUP; } + /* - * The ability to send a file descriptor to another - * process is opt-in by file type. + * fo_kqfilter + * + * Description: Generic fileops kqueue filter indirected through the fileops + * pointer in the fileproc structure + * + * Parameters: fp fileproc structure pointer + * kn pointer to knote to filter on + * + * Returns: (kn->kn_flags & EV_ERROR) error in kn->kn_data + * 0 Filter is not active + * !0 Filter is active */ -boolean_t -file_issendable(proc_t p, struct fileproc *fp) +int +fo_kqfilter(struct fileproc *fp, struct knote *kn, struct kevent_qos_s *kev) { - proc_fdlock_assert(p, LCK_MTX_ASSERT_OWNED); + return (*fp->f_ops->fo_kqfilter)(fp, kn, kev); +} - switch (fp->f_type) { - case DTYPE_VNODE: - case DTYPE_SOCKET: - case DTYPE_PIPE: - case DTYPE_PSXSHM: - return (0 == (fp->f_fglob->fg_lflags & FG_CONFINED)); - default: - /* DTYPE_KQUEUE, DTYPE_FSEVENTS, DTYPE_PSXSEM */ - return FALSE; - } +int +fo_no_kqfilter(struct fileproc *fp, struct knote *kn, struct kevent_qos_s *kev) +{ +#pragma unused(fp, kev) + knote_set_error(kn, ENOTSUP); + return 0; } struct fileproc * fileproc_alloc_init(__unused void *arg) { - struct fileproc *fp; - - MALLOC_ZONE(fp, struct fileproc *, sizeof (*fp), M_FILEPROC, M_WAITOK); - if (fp) - bzero(fp, sizeof (*fp)); + struct fileproc *fp = zalloc_flags(fp_zone, Z_WAITOK | Z_ZERO); - return (fp); + os_ref_init(&fp->fp_iocount, &f_refgrp); + return fp; } + void fileproc_free(struct fileproc *fp) { + os_ref_count_t __unused refc = os_ref_release(&fp->fp_iocount); +#if DEVELOPMENT || DEBUG + if (0 != refc) { + panic("%s: pid %d refc: %u != 0", + __func__, proc_pid(current_proc()), refc); + } +#endif switch (FILEPROC_TYPE(fp)) { case FTYPE_SIMPLE: - FREE_ZONE(fp, sizeof (*fp), M_FILEPROC); + zfree(fp_zone, fp); break; case FTYPE_GUARDED: guarded_fileproc_free(fp); break; default: - panic("%s: corrupt fp %p flags %x", __func__, fp, fp->f_flags); + panic("%s: corrupt fp %p flags %x", __func__, fp, fp->fp_flags); + } +} + +void +fileproc_modify_vflags(struct fileproc *fp, fileproc_vflags_t vflags, boolean_t clearflags) +{ + if (clearflags) { + os_atomic_andnot(&fp->fp_vflags, vflags, relaxed); + } else { + os_atomic_or(&fp->fp_vflags, vflags, relaxed); } } + +fileproc_vflags_t +fileproc_get_vflags(struct fileproc *fp) +{ + return os_atomic_load(&fp->fp_vflags, relaxed); +}