sys/kern/kern_descrip.c

   1 /*
   2  * Copyright (c) 2005 The DragonFly Project.  All rights reserved.
   3  *
   4  * This code is derived from software contributed to The DragonFly Project
   5  * by Jeffrey Hsu.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  *
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in
  15  *    the documentation and/or other materials provided with the
  16  *    distribution.
  17  * 3. Neither the name of The DragonFly Project nor the names of its
  18  *    contributors may be used to endorse or promote products derived
  19  *    from this software without specific, prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  24  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
  25  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  26  * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
  27  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  28  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  29  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  30  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  31  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  32  * SUCH DAMAGE.
  33  *
  34  *
  35  * Copyright (c) 1982, 1986, 1989, 1991, 1993
  36  *      The Regents of the University of California.  All rights reserved.
  37  * (c) UNIX System Laboratories, Inc.
  38  * All or some portions of this file are derived from material licensed
  39  * to the University of California by American Telephone and Telegraph
  40  * Co. or Unix System Laboratories, Inc. and are reproduced herein with
  41  * the permission of UNIX System Laboratories, Inc.
  42  *
  43  * Redistribution and use in source and binary forms, with or without
  44  * modification, are permitted provided that the following conditions
  45  * are met:
  46  * 1. Redistributions of source code must retain the above copyright
  47  *    notice, this list of conditions and the following disclaimer.
  48  * 2. Redistributions in binary form must reproduce the above copyright
  49  *    notice, this list of conditions and the following disclaimer in the
  50  *    documentation and/or other materials provided with the distribution.
  51  * 3. All advertising materials mentioning features or use of this software
  52  *    must display the following acknowledgement:
  53  *      This product includes software developed by the University of
  54  *      California, Berkeley and its contributors.
  55  * 4. Neither the name of the University nor the names of its contributors
  56  *    may be used to endorse or promote products derived from this software
  57  *    without specific prior written permission.
  58  *
  59  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  60  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  61  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  62  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  63  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  64  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  65  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  66  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  67  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  68  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  69  * SUCH DAMAGE.
  70  *
  71  *      @(#)kern_descrip.c      8.6 (Berkeley) 4/19/94
  72  * $FreeBSD: src/sys/kern/kern_descrip.c,v 1.81.2.19 2004/02/28 00:43:31 tegge Exp $
  73  */
  74
  75 #include "opt_compat.h"
  76 #include <sys/param.h>
  77 #include <sys/systm.h>
  78 #include <sys/malloc.h>
  79 #include <sys/sysproto.h>
  80 #include <sys/conf.h>
  81 #include <sys/device.h>
  82 #include <sys/file.h>
  83 #include <sys/filedesc.h>
  84 #include <sys/kernel.h>
  85 #include <sys/sysctl.h>
  86 #include <sys/vnode.h>
  87 #include <sys/proc.h>
  88 #include <sys/nlookup.h>
  89 #include <sys/file.h>
  90 #include <sys/stat.h>
  91 #include <sys/filio.h>
  92 #include <sys/fcntl.h>
  93 #include <sys/unistd.h>
  94 #include <sys/resourcevar.h>
  95 #include <sys/event.h>
  96 #include <sys/kern_syscall.h>
  97 #include <sys/kcore.h>
  98 #include <sys/kinfo.h>
  99 #include <sys/un.h>
 100
 101 #include <vm/vm.h>
 102 #include <vm/vm_extern.h>
 103
 104 #include <sys/thread2.h>
 105 #include <sys/file2.h>
 106 #include <sys/spinlock2.h>
 107
 108 static void fsetfd_locked(struct filedesc *fdp, struct file *fp, int fd);
 109 static void fdreserve_locked (struct filedesc *fdp, int fd0, int incr);
 110 static struct file *funsetfd_locked (struct filedesc *fdp, int fd);
 111 static void ffree(struct file *fp);
 112
 113 static MALLOC_DEFINE(M_FILEDESC, "file desc", "Open file descriptor table");
 114 static MALLOC_DEFINE(M_FILEDESC_TO_LEADER, "file desc to leader",
 115                      "file desc to leader structures");
 116 MALLOC_DEFINE(M_FILE, "file", "Open file structure");
 117 static MALLOC_DEFINE(M_SIGIO, "sigio", "sigio structures");
 118
 119 static struct krate krate_uidinfo = { .freq = 1 };
 120
 121 static   d_open_t  fdopen;
 122 #define NUMFDESC 64
 123
 124 #define CDEV_MAJOR 22
 125 static struct dev_ops fildesc_ops = {
 126         { "FD", 0, 0 },
 127         .d_open =       fdopen,
 128 };
 129
 130 /*
 131  * Descriptor management.
 132  */
 133 static struct filelist filehead = LIST_HEAD_INITIALIZER(&filehead);
 134 static struct spinlock filehead_spin = SPINLOCK_INITIALIZER(&filehead_spin);
 135 static int nfiles;              /* actual number of open files */
 136 extern int cmask;
 137
 138 /*
 139  * Fixup fd_freefile and fd_lastfile after a descriptor has been cleared.
 140  *
 141  * MPSAFE - must be called with fdp->fd_spin exclusively held
 142  */
 143 static __inline
 144 void
 145 fdfixup_locked(struct filedesc *fdp, int fd)
 146 {
 147         if (fd < fdp->fd_freefile) {
 148                fdp->fd_freefile = fd;
 149         }
 150         while (fdp->fd_lastfile >= 0 &&
 151                fdp->fd_files[fdp->fd_lastfile].fp == NULL &&
 152                fdp->fd_files[fdp->fd_lastfile].reserved == 0
 153         ) {
 154                 --fdp->fd_lastfile;
 155         }
 156 }
 157
 158 /*
 159  * System calls on descriptors.
 160  *
 161  * MPSAFE
 162  */
 163 int
 164 sys_getdtablesize(struct getdtablesize_args *uap)
 165 {
 166         struct proc *p = curproc;
 167         struct plimit *limit = p->p_limit;
 168         int dtsize;
 169
 170         spin_lock(&limit->p_spin);
 171         if (limit->pl_rlimit[RLIMIT_NOFILE].rlim_cur > INT_MAX)
 172                 dtsize = INT_MAX;
 173         else
 174                 dtsize = (int)limit->pl_rlimit[RLIMIT_NOFILE].rlim_cur;
 175         spin_unlock(&limit->p_spin);
 176
 177         if (dtsize > maxfilesperproc)
 178                 dtsize = maxfilesperproc;
 179         if (dtsize < minfilesperproc)
 180                 dtsize = minfilesperproc;
 181         if (p->p_ucred->cr_uid && dtsize > maxfilesperuser)
 182                 dtsize = maxfilesperuser;
 183         uap->sysmsg_result = dtsize;
 184         return (0);
 185 }
 186
 187 /*
 188  * Duplicate a file descriptor to a particular value.
 189  *
 190  * note: keep in mind that a potential race condition exists when closing
 191  * descriptors from a shared descriptor table (via rfork).
 192  *
 193  * MPSAFE
 194  */
 195 int
 196 sys_dup2(struct dup2_args *uap)
 197 {
 198         int error;
 199         int fd = 0;
 200
 201         error = kern_dup(DUP_FIXED, uap->from, uap->to, &fd);
 202         uap->sysmsg_fds[0] = fd;
 203
 204         return (error);
 205 }
 206
 207 /*
 208  * Duplicate a file descriptor.
 209  *
 210  * MPSAFE
 211  */
 212 int
 213 sys_dup(struct dup_args *uap)
 214 {
 215         int error;
 216         int fd = 0;
 217
 218         error = kern_dup(DUP_VARIABLE, uap->fd, 0, &fd);
 219         uap->sysmsg_fds[0] = fd;
 220
 221         return (error);
 222 }
 223
 224 /*
 225  * MPALMOSTSAFE - acquires mplock for fp operations
 226  */
 227 int
 228 kern_fcntl(int fd, int cmd, union fcntl_dat *dat, struct ucred *cred)
 229 {
 230         struct thread *td = curthread;
 231         struct proc *p = td->td_proc;
 232         struct file *fp;
 233         struct vnode *vp;
 234         u_int newmin;
 235         u_int oflags;
 236         u_int nflags;
 237         int tmp, error, flg = F_POSIX;
 238
 239         KKASSERT(p);
 240
 241         /*
 242          * Operations on file descriptors that do not require a file pointer.
 243          */
 244         switch (cmd) {
 245         case F_GETFD:
 246                 error = fgetfdflags(p->p_fd, fd, &tmp);
 247                 if (error == 0)
 248                         dat->fc_cloexec = (tmp & UF_EXCLOSE) ? FD_CLOEXEC : 0;
 249                 return (error);
 250
 251         case F_SETFD:
 252                 if (dat->fc_cloexec & FD_CLOEXEC)
 253                         error = fsetfdflags(p->p_fd, fd, UF_EXCLOSE);
 254                 else
 255                         error = fclrfdflags(p->p_fd, fd, UF_EXCLOSE);
 256                 return (error);
 257         case F_DUPFD:
 258                 newmin = dat->fc_fd;
 259                 error = kern_dup(DUP_VARIABLE, fd, newmin, &dat->fc_fd);
 260                 return (error);
 261         default:
 262                 break;
 263         }
 264
 265         /*
 266          * Operations on file pointers
 267          */
 268         if ((fp = holdfp(p->p_fd, fd, -1)) == NULL)
 269                 return (EBADF);
 270
 271         switch (cmd) {
 272         case F_GETFL:
 273                 dat->fc_flags = OFLAGS(fp->f_flag);
 274                 error = 0;
 275                 break;
 276
 277         case F_SETFL:
 278                 oflags = fp->f_flag;
 279                 nflags = FFLAGS(dat->fc_flags & ~O_ACCMODE) & FCNTLFLAGS;
 280                 nflags |= oflags & ~FCNTLFLAGS;
 281
 282                 error = 0;
 283                 if (((nflags ^ oflags) & O_APPEND) && (oflags & FAPPENDONLY))
 284                         error = EINVAL;
 285                 if (error == 0 && ((nflags ^ oflags) & FASYNC)) {
 286                         tmp = nflags & FASYNC;
 287                         error = fo_ioctl(fp, FIOASYNC, (caddr_t)&tmp,
 288                                          cred, NULL);
 289                 }
 290                 if (error == 0)
 291                         fp->f_flag = nflags;
 292                 break;
 293
 294         case F_GETOWN:
 295                 error = fo_ioctl(fp, FIOGETOWN, (caddr_t)&dat->fc_owner,
 296                                  cred, NULL);
 297                 break;
 298
 299         case F_SETOWN:
 300                 error = fo_ioctl(fp, FIOSETOWN, (caddr_t)&dat->fc_owner,
 301                                  cred, NULL);
 302                 break;
 303
 304         case F_SETLKW:
 305                 flg |= F_WAIT;
 306                 /* Fall into F_SETLK */
 307
 308         case F_SETLK:
 309                 if (fp->f_type != DTYPE_VNODE) {
 310                         error = EBADF;
 311                         break;
 312                 }
 313                 vp = (struct vnode *)fp->f_data;
 314
 315                 /*
 316                  * copyin/lockop may block
 317                  */
 318                 if (dat->fc_flock.l_whence == SEEK_CUR)
 319                         dat->fc_flock.l_start += fp->f_offset;
 320
 321                 switch (dat->fc_flock.l_type) {
 322                 case F_RDLCK:
 323                         if ((fp->f_flag & FREAD) == 0) {
 324                                 error = EBADF;
 325                                 break;
 326                         }
 327                         p->p_leader->p_flag |= P_ADVLOCK;
 328                         error = VOP_ADVLOCK(vp, (caddr_t)p->p_leader, F_SETLK,
 329                             &dat->fc_flock, flg);
 330                         break;
 331                 case F_WRLCK:
 332                         if ((fp->f_flag & FWRITE) == 0) {
 333                                 error = EBADF;
 334                                 break;
 335                         }
 336                         p->p_leader->p_flag |= P_ADVLOCK;
 337                         error = VOP_ADVLOCK(vp, (caddr_t)p->p_leader, F_SETLK,
 338                             &dat->fc_flock, flg);
 339                         break;
 340                 case F_UNLCK:
 341                         error = VOP_ADVLOCK(vp, (caddr_t)p->p_leader, F_UNLCK,
 342                                 &dat->fc_flock, F_POSIX);
 343                         break;
 344                 default:
 345                         error = EINVAL;
 346                         break;
 347                 }
 348
 349                 /*
 350                  * It is possible to race a close() on the descriptor while
 351                  * we were blocked getting the lock.  If this occurs the
 352                  * close might not have caught the lock.
 353                  */
 354                 if (checkfdclosed(p->p_fd, fd, fp)) {
 355                         dat->fc_flock.l_whence = SEEK_SET;
 356                         dat->fc_flock.l_start = 0;
 357                         dat->fc_flock.l_len = 0;
 358                         dat->fc_flock.l_type = F_UNLCK;
 359                         (void) VOP_ADVLOCK(vp, (caddr_t)p->p_leader,
 360                                            F_UNLCK, &dat->fc_flock, F_POSIX);
 361                 }
 362                 break;
 363
 364         case F_GETLK:
 365                 if (fp->f_type != DTYPE_VNODE) {
 366                         error = EBADF;
 367                         break;
 368                 }
 369                 vp = (struct vnode *)fp->f_data;
 370                 /*
 371                  * copyin/lockop may block
 372                  */
 373                 if (dat->fc_flock.l_type != F_RDLCK &&
 374                     dat->fc_flock.l_type != F_WRLCK &&
 375                     dat->fc_flock.l_type != F_UNLCK) {
 376                         error = EINVAL;
 377                         break;
 378                 }
 379                 if (dat->fc_flock.l_whence == SEEK_CUR)
 380                         dat->fc_flock.l_start += fp->f_offset;
 381                 error = VOP_ADVLOCK(vp, (caddr_t)p->p_leader, F_GETLK,
 382                             &dat->fc_flock, F_POSIX);
 383                 break;
 384         default:
 385                 error = EINVAL;
 386                 break;
 387         }
 388
 389         fdrop(fp);
 390         return (error);
 391 }
 392
 393 /*
 394  * The file control system call.
 395  *
 396  * MPSAFE
 397  */
 398 int
 399 sys_fcntl(struct fcntl_args *uap)
 400 {
 401         union fcntl_dat dat;
 402         int error;
 403
 404         switch (uap->cmd) {
 405         case F_DUPFD:
 406                 dat.fc_fd = uap->arg;
 407                 break;
 408         case F_SETFD:
 409                 dat.fc_cloexec = uap->arg;
 410                 break;
 411         case F_SETFL:
 412                 dat.fc_flags = uap->arg;
 413                 break;
 414         case F_SETOWN:
 415                 dat.fc_owner = uap->arg;
 416                 break;
 417         case F_SETLKW:
 418         case F_SETLK:
 419         case F_GETLK:
 420                 error = copyin((caddr_t)uap->arg, &dat.fc_flock,
 421                                sizeof(struct flock));
 422                 if (error)
 423                         return (error);
 424                 break;
 425         }
 426
 427         error = kern_fcntl(uap->fd, uap->cmd, &dat, curthread->td_ucred);
 428
 429         if (error == 0) {
 430                 switch (uap->cmd) {
 431                 case F_DUPFD:
 432                         uap->sysmsg_result = dat.fc_fd;
 433                         break;
 434                 case F_GETFD:
 435                         uap->sysmsg_result = dat.fc_cloexec;
 436                         break;
 437                 case F_GETFL:
 438                         uap->sysmsg_result = dat.fc_flags;
 439                         break;
 440                 case F_GETOWN:
 441                         uap->sysmsg_result = dat.fc_owner;
 442                 case F_GETLK:
 443                         error = copyout(&dat.fc_flock, (caddr_t)uap->arg,
 444                             sizeof(struct flock));
 445                         break;
 446                 }
 447         }
 448
 449         return (error);
 450 }
 451
 452 /*
 453  * Common code for dup, dup2, and fcntl(F_DUPFD).
 454  *
 455  * The type flag can be either DUP_FIXED or DUP_VARIABLE.  DUP_FIXED tells
 456  * kern_dup() to destructively dup over an existing file descriptor if new
 457  * is already open.  DUP_VARIABLE tells kern_dup() to find the lowest
 458  * unused file descriptor that is greater than or equal to new.
 459  *
 460  * MPSAFE
 461  */
 462 int
 463 kern_dup(enum dup_type type, int old, int new, int *res)
 464 {
 465         struct thread *td = curthread;
 466         struct proc *p = td->td_proc;
 467         struct filedesc *fdp = p->p_fd;
 468         struct file *fp;
 469         struct file *delfp;
 470         int oldflags;
 471         int holdleaders;
 472         int dtsize;
 473         int error, newfd;
 474
 475         /*
 476          * Verify that we have a valid descriptor to dup from and
 477          * possibly to dup to.
 478          *
 479          * NOTE: maxfilesperuser is not applicable to dup()
 480          */
 481 retry:
 482         if (p->p_rlimit[RLIMIT_NOFILE].rlim_cur > INT_MAX)
 483                 dtsize = INT_MAX;
 484         else
 485                 dtsize = (int)p->p_rlimit[RLIMIT_NOFILE].rlim_cur;
 486         if (dtsize > maxfilesperproc)
 487                 dtsize = maxfilesperproc;
 488         if (dtsize < minfilesperproc)
 489                 dtsize = minfilesperproc;
 490
 491         if (new < 0 || new > dtsize)
 492                 return (EINVAL);
 493
 494         spin_lock(&fdp->fd_spin);
 495         if ((unsigned)old >= fdp->fd_nfiles || fdp->fd_files[old].fp == NULL) {
 496                 spin_unlock(&fdp->fd_spin);
 497                 return (EBADF);
 498         }
 499         if (type == DUP_FIXED && old == new) {
 500                 *res = new;
 501                 spin_unlock(&fdp->fd_spin);
 502                 return (0);
 503         }
 504         fp = fdp->fd_files[old].fp;
 505         oldflags = fdp->fd_files[old].fileflags;
 506         fhold(fp);      /* MPSAFE - can be called with a spinlock held */
 507
 508         /*
 509          * Allocate a new descriptor if DUP_VARIABLE, or expand the table
 510          * if the requested descriptor is beyond the current table size.
 511          *
 512          * This can block.  Retry if the source descriptor no longer matches
 513          * or if our expectation in the expansion case races.
 514          *
 515          * If we are not expanding or allocating a new decriptor, then reset
 516          * the target descriptor to a reserved state so we have a uniform
 517          * setup for the next code block.
 518          */
 519         if (type == DUP_VARIABLE || new >= fdp->fd_nfiles) {
 520                 spin_unlock(&fdp->fd_spin);
 521                 error = fdalloc(p, new, &newfd);
 522                 spin_lock(&fdp->fd_spin);
 523                 if (error) {
 524                         spin_unlock(&fdp->fd_spin);
 525                         fdrop(fp);
 526                         return (error);
 527                 }
 528                 /*
 529                  * Check for ripout
 530                  */
 531                 if (old >= fdp->fd_nfiles || fdp->fd_files[old].fp != fp) {
 532                         fsetfd_locked(fdp, NULL, newfd);
 533                         spin_unlock(&fdp->fd_spin);
 534                         fdrop(fp);
 535                         goto retry;
 536                 }
 537                 /*
 538                  * Check for expansion race
 539                  */
 540                 if (type != DUP_VARIABLE && new != newfd) {
 541                         fsetfd_locked(fdp, NULL, newfd);
 542                         spin_unlock(&fdp->fd_spin);
 543                         fdrop(fp);
 544                         goto retry;
 545                 }
 546                 /*
 547                  * Check for ripout, newfd reused old (this case probably
 548                  * can't occur).
 549                  */
 550                 if (old == newfd) {
 551                         fsetfd_locked(fdp, NULL, newfd);
 552                         spin_unlock(&fdp->fd_spin);
 553                         fdrop(fp);
 554                         goto retry;
 555                 }
 556                 new = newfd;
 557                 delfp = NULL;
 558         } else {
 559                 if (fdp->fd_files[new].reserved) {
 560                         spin_unlock(&fdp->fd_spin);
 561                         fdrop(fp);
 562                         kprintf("Warning: dup(): target descriptor %d is reserved, waiting for it to be resolved\n", new);
 563                         tsleep(fdp, 0, "fdres", hz);
 564                         goto retry;
 565                 }
 566
 567                 /*
 568                  * If the target descriptor was never allocated we have
 569                  * to allocate it.  If it was we have to clean out the
 570                  * old descriptor.  delfp inherits the ref from the
 571                  * descriptor table.
 572                  */
 573                 delfp = fdp->fd_files[new].fp;
 574                 fdp->fd_files[new].fp = NULL;
 575                 fdp->fd_files[new].reserved = 1;
 576                 if (delfp == NULL) {
 577                         fdreserve_locked(fdp, new, 1);
 578                         if (new > fdp->fd_lastfile)
 579                                 fdp->fd_lastfile = new;
 580                 }
 581
 582         }
 583
 584         /*
 585          * NOTE: still holding an exclusive spinlock
 586          */
 587
 588         /*
 589          * If a descriptor is being overwritten we may hve to tell
 590          * fdfree() to sleep to ensure that all relevant process
 591          * leaders can be traversed in closef().
 592          */
 593         if (delfp != NULL && p->p_fdtol != NULL) {
 594                 fdp->fd_holdleaderscount++;
 595                 holdleaders = 1;
 596         } else {
 597                 holdleaders = 0;
 598         }
 599         KASSERT(delfp == NULL || type == DUP_FIXED,
 600                 ("dup() picked an open file"));
 601
 602         /*
 603          * Duplicate the source descriptor, update lastfile.  If the new
 604          * descriptor was not allocated and we aren't replacing an existing
 605          * descriptor we have to mark the descriptor as being in use.
 606          *
 607          * The fd_files[] array inherits fp's hold reference.
 608          */
 609         fsetfd_locked(fdp, fp, new);
 610         fdp->fd_files[new].fileflags = oldflags & ~UF_EXCLOSE;
 611         spin_unlock(&fdp->fd_spin);
 612         fdrop(fp);
 613         *res = new;
 614
 615         /*
 616          * If we dup'd over a valid file, we now own the reference to it
 617          * and must dispose of it using closef() semantics (as if a
 618          * close() were performed on it).
 619          */
 620         if (delfp) {
 621                 if (SLIST_FIRST(&delfp->f_klist))
 622                         knote_fdclose(delfp, fdp, new);
 623                 closef(delfp, p);
 624                 if (holdleaders) {
 625                         spin_lock(&fdp->fd_spin);
 626                         fdp->fd_holdleaderscount--;
 627                         if (fdp->fd_holdleaderscount == 0 &&
 628                             fdp->fd_holdleaderswakeup != 0) {
 629                                 fdp->fd_holdleaderswakeup = 0;
 630                                 spin_unlock(&fdp->fd_spin);
 631                                 wakeup(&fdp->fd_holdleaderscount);
 632                         } else {
 633                                 spin_unlock(&fdp->fd_spin);
 634                         }
 635                 }
 636         }
 637         return (0);
 638 }
 639
 640 /*
 641  * If sigio is on the list associated with a process or process group,
 642  * disable signalling from the device, remove sigio from the list and
 643  * free sigio.
 644  *
 645  * MPSAFE
 646  */
 647 void
 648 funsetown(struct sigio *sigio)
 649 {
 650         if (sigio == NULL)
 651                 return;
 652         lwkt_gettoken(&proc_token);
 653         *(sigio->sio_myref) = NULL;
 654         if (sigio->sio_pgid < 0) {
 655                 SLIST_REMOVE(&sigio->sio_pgrp->pg_sigiolst, sigio,
 656                              sigio, sio_pgsigio);
 657         } else /* if ((*sigiop)->sio_pgid > 0) */ {
 658                 SLIST_REMOVE(&sigio->sio_proc->p_sigiolst, sigio,
 659                              sigio, sio_pgsigio);
 660         }
 661         lwkt_reltoken(&proc_token);
 662         crfree(sigio->sio_ucred);
 663         kfree(sigio, M_SIGIO);
 664 }
 665
 666 /*
 667  * Free a list of sigio structures.
 668  *
 669  * MPSAFE
 670  */
 671 void
 672 funsetownlst(struct sigiolst *sigiolst)
 673 {
 674         struct sigio *sigio;
 675
 676         lwkt_gettoken(&proc_token);
 677         while ((sigio = SLIST_FIRST(sigiolst)) != NULL)
 678                 funsetown(sigio);
 679         lwkt_reltoken(&proc_token);
 680 }
 681
 682 /*
 683  * This is common code for FIOSETOWN ioctl called by fcntl(fd, F_SETOWN, arg).
 684  *
 685  * After permission checking, add a sigio structure to the sigio list for
 686  * the process or process group.
 687  *
 688  * MPSAFE
 689  */
 690 int
 691 fsetown(pid_t pgid, struct sigio **sigiop)
 692 {
 693         struct proc *proc;
 694         struct pgrp *pgrp;
 695         struct sigio *sigio;
 696         int error;
 697
 698         if (pgid == 0) {
 699                 funsetown(*sigiop);
 700                 return (0);
 701         }
 702
 703         lwkt_gettoken(&proc_token);
 704         if (pgid > 0) {
 705                 proc = pfind(pgid);
 706                 if (proc == NULL) {
 707                         error = ESRCH;
 708                         goto done;
 709                 }
 710
 711                 /*
 712                  * Policy - Don't allow a process to FSETOWN a process
 713                  * in another session.
 714                  *
 715                  * Remove this test to allow maximum flexibility or
 716                  * restrict FSETOWN to the current process or process
 717                  * group for maximum safety.
 718                  */
 719                 if (proc->p_session != curproc->p_session) {
 720                         error = EPERM;
 721                         goto done;
 722                 }
 723
 724                 pgrp = NULL;
 725         } else /* if (pgid < 0) */ {
 726                 pgrp = pgfind(-pgid);
 727                 if (pgrp == NULL) {
 728                         error = ESRCH;
 729                         goto done;
 730                 }
 731
 732                 /*
 733                  * Policy - Don't allow a process to FSETOWN a process
 734                  * in another session.
 735                  *
 736                  * Remove this test to allow maximum flexibility or
 737                  * restrict FSETOWN to the current process or process
 738                  * group for maximum safety.
 739                  */
 740                 if (pgrp->pg_session != curproc->p_session) {
 741                         error = EPERM;
 742                         goto done;
 743                 }
 744
 745                 proc = NULL;
 746         }
 747         funsetown(*sigiop);
 748         sigio = kmalloc(sizeof(struct sigio), M_SIGIO, M_WAITOK);
 749         if (pgid > 0) {
 750                 SLIST_INSERT_HEAD(&proc->p_sigiolst, sigio, sio_pgsigio);
 751                 sigio->sio_proc = proc;
 752         } else {
 753                 SLIST_INSERT_HEAD(&pgrp->pg_sigiolst, sigio, sio_pgsigio);
 754                 sigio->sio_pgrp = pgrp;
 755         }
 756         sigio->sio_pgid = pgid;
 757         sigio->sio_ucred = crhold(curthread->td_ucred);
 758         /* It would be convenient if p_ruid was in ucred. */
 759         sigio->sio_ruid = sigio->sio_ucred->cr_ruid;
 760         sigio->sio_myref = sigiop;
 761         *sigiop = sigio;
 762         error = 0;
 763 done:
 764         lwkt_reltoken(&proc_token);
 765         return (error);
 766 }
 767
 768 /*
 769  * This is common code for FIOGETOWN ioctl called by fcntl(fd, F_GETOWN, arg).
 770  *
 771  * MPSAFE
 772  */
 773 pid_t
 774 fgetown(struct sigio **sigiop)
 775 {
 776         struct sigio *sigio;
 777         pid_t own;
 778
 779         lwkt_gettoken(&proc_token);
 780         sigio = *sigiop;
 781         own = (sigio != NULL ? sigio->sio_pgid : 0);
 782         lwkt_reltoken(&proc_token);
 783
 784         return (own);
 785 }
 786
 787 /*
 788  * Close many file descriptors.
 789  *
 790  * MPSAFE
 791  */
 792 int
 793 sys_closefrom(struct closefrom_args *uap)
 794 {
 795         return(kern_closefrom(uap->fd));
 796 }
 797
 798 /*
 799  * Close all file descriptors greater then or equal to fd
 800  *
 801  * MPSAFE
 802  */
 803 int
 804 kern_closefrom(int fd)
 805 {
 806         struct thread *td = curthread;
 807         struct proc *p = td->td_proc;
 808         struct filedesc *fdp;
 809
 810         KKASSERT(p);
 811         fdp = p->p_fd;
 812
 813         if (fd < 0)
 814                 return (EINVAL);
 815
 816         /*
 817          * NOTE: This function will skip unassociated descriptors and
 818          * reserved descriptors that have not yet been assigned.
 819          * fd_lastfile can change as a side effect of kern_close().
 820          */
 821         spin_lock(&fdp->fd_spin);
 822         while (fd <= fdp->fd_lastfile) {
 823                 if (fdp->fd_files[fd].fp != NULL) {
 824                         spin_unlock(&fdp->fd_spin);
 825                         /* ok if this races another close */
 826                         if (kern_close(fd) == EINTR)
 827                                 return (EINTR);
 828                         spin_lock(&fdp->fd_spin);
 829                 }
 830                 ++fd;
 831         }
 832         spin_unlock(&fdp->fd_spin);
 833         return (0);
 834 }
 835
 836 /*
 837  * Close a file descriptor.
 838  *
 839  * MPSAFE
 840  */
 841 int
 842 sys_close(struct close_args *uap)
 843 {
 844         return(kern_close(uap->fd));
 845 }
 846
 847 /*
 848  * MPSAFE
 849  */
 850 int
 851 kern_close(int fd)
 852 {
 853         struct thread *td = curthread;
 854         struct proc *p = td->td_proc;
 855         struct filedesc *fdp;
 856         struct file *fp;
 857         int error;
 858         int holdleaders;
 859
 860         KKASSERT(p);
 861         fdp = p->p_fd;
 862
 863         spin_lock(&fdp->fd_spin);
 864         if ((fp = funsetfd_locked(fdp, fd)) == NULL) {
 865                 spin_unlock(&fdp->fd_spin);
 866                 return (EBADF);
 867         }
 868         holdleaders = 0;
 869         if (p->p_fdtol != NULL) {
 870                 /*
 871                  * Ask fdfree() to sleep to ensure that all relevant
 872                  * process leaders can be traversed in closef().
 873                  */
 874                 fdp->fd_holdleaderscount++;
 875                 holdleaders = 1;
 876         }
 877
 878         /*
 879          * we now hold the fp reference that used to be owned by the descriptor
 880          * array.
 881          */
 882         spin_unlock(&fdp->fd_spin);
 883         if (SLIST_FIRST(&fp->f_klist))
 884                 knote_fdclose(fp, fdp, fd);
 885         error = closef(fp, p);
 886         if (holdleaders) {
 887                 spin_lock(&fdp->fd_spin);
 888                 fdp->fd_holdleaderscount--;
 889                 if (fdp->fd_holdleaderscount == 0 &&
 890                     fdp->fd_holdleaderswakeup != 0) {
 891                         fdp->fd_holdleaderswakeup = 0;
 892                         spin_unlock(&fdp->fd_spin);
 893                         wakeup(&fdp->fd_holdleaderscount);
 894                 } else {
 895                         spin_unlock(&fdp->fd_spin);
 896                 }
 897         }
 898         return (error);
 899 }
 900
 901 /*
 902  * shutdown_args(int fd, int how)
 903  */
 904 int
 905 kern_shutdown(int fd, int how)
 906 {
 907         struct thread *td = curthread;
 908         struct proc *p = td->td_proc;
 909         struct file *fp;
 910         int error;
 911
 912         KKASSERT(p);
 913
 914         if ((fp = holdfp(p->p_fd, fd, -1)) == NULL)
 915                 return (EBADF);
 916         error = fo_shutdown(fp, how);
 917         fdrop(fp);
 918
 919         return (error);
 920 }
 921
 922 /*
 923  * MPALMOSTSAFE
 924  */
 925 int
 926 sys_shutdown(struct shutdown_args *uap)
 927 {
 928         int error;
 929
 930         error = kern_shutdown(uap->s, uap->how);
 931
 932         return (error);
 933 }
 934
 935 /*
 936  * MPSAFE
 937  */
 938 int
 939 kern_fstat(int fd, struct stat *ub)
 940 {
 941         struct thread *td = curthread;
 942         struct proc *p = td->td_proc;
 943         struct file *fp;
 944         int error;
 945
 946         KKASSERT(p);
 947
 948         if ((fp = holdfp(p->p_fd, fd, -1)) == NULL)
 949                 return (EBADF);
 950         error = fo_stat(fp, ub, td->td_ucred);
 951         fdrop(fp);
 952
 953         return (error);
 954 }
 955
 956 /*
 957  * Return status information about a file descriptor.
 958  *
 959  * MPSAFE
 960  */
 961 int
 962 sys_fstat(struct fstat_args *uap)
 963 {
 964         struct stat st;
 965         int error;
 966
 967         error = kern_fstat(uap->fd, &st);
 968
 969         if (error == 0)
 970                 error = copyout(&st, uap->sb, sizeof(st));
 971         return (error);
 972 }
 973
 974 /*
 975  * Return pathconf information about a file descriptor.
 976  *
 977  * MPALMOSTSAFE
 978  */
 979 int
 980 sys_fpathconf(struct fpathconf_args *uap)
 981 {
 982         struct thread *td = curthread;
 983         struct proc *p = td->td_proc;
 984         struct file *fp;
 985         struct vnode *vp;
 986         int error = 0;
 987
 988         if ((fp = holdfp(p->p_fd, uap->fd, -1)) == NULL)
 989                 return (EBADF);
 990
 991         switch (fp->f_type) {
 992         case DTYPE_PIPE:
 993         case DTYPE_SOCKET:
 994                 if (uap->name != _PC_PIPE_BUF) {
 995                         error = EINVAL;
 996                 } else {
 997                         uap->sysmsg_result = PIPE_BUF;
 998                         error = 0;
 999                 }
1000                 break;
1001         case DTYPE_FIFO:
1002         case DTYPE_VNODE:
1003                 vp = (struct vnode *)fp->f_data;
1004                 error = VOP_PATHCONF(vp, uap->name, &uap->sysmsg_reg);
1005                 break;
1006         default:
1007                 error = EOPNOTSUPP;
1008                 break;
1009         }
1010         fdrop(fp);
1011         return(error);
1012 }
1013
1014 static int fdexpand;
1015 SYSCTL_INT(_debug, OID_AUTO, fdexpand, CTLFLAG_RD, &fdexpand, 0,
1016     "Number of times a file table has been expanded");
1017
1018 /*
1019  * Grow the file table so it can hold through descriptor (want).
1020  *
1021  * The fdp's spinlock must be held exclusively on entry and may be held
1022  * exclusively on return.  The spinlock may be cycled by the routine.
1023  *
1024  * MPSAFE
1025  */
1026 static void
1027 fdgrow_locked(struct filedesc *fdp, int want)
1028 {
1029         struct fdnode *newfiles;
1030         struct fdnode *oldfiles;
1031         int nf, extra;
1032
1033         nf = fdp->fd_nfiles;
1034         do {
1035                 /* nf has to be of the form 2^n - 1 */
1036                 nf = 2 * nf + 1;
1037         } while (nf <= want);
1038
1039         spin_unlock(&fdp->fd_spin);
1040         newfiles = kmalloc(nf * sizeof(struct fdnode), M_FILEDESC, M_WAITOK);
1041         spin_lock(&fdp->fd_spin);
1042
1043         /*
1044          * We could have raced another extend while we were not holding
1045          * the spinlock.
1046          */
1047         if (fdp->fd_nfiles >= nf) {
1048                 spin_unlock(&fdp->fd_spin);
1049                 kfree(newfiles, M_FILEDESC);
1050                 spin_lock(&fdp->fd_spin);
1051                 return;
1052         }
1053         /*
1054          * Copy the existing ofile and ofileflags arrays
1055          * and zero the new portion of each array.
1056          */
1057         extra = nf - fdp->fd_nfiles;
1058         bcopy(fdp->fd_files, newfiles, fdp->fd_nfiles * sizeof(struct fdnode));
1059         bzero(&newfiles[fdp->fd_nfiles], extra * sizeof(struct fdnode));
1060
1061         oldfiles = fdp->fd_files;
1062         fdp->fd_files = newfiles;
1063         fdp->fd_nfiles = nf;
1064
1065         if (oldfiles != fdp->fd_builtin_files) {
1066                 spin_unlock(&fdp->fd_spin);
1067                 kfree(oldfiles, M_FILEDESC);
1068                 spin_lock(&fdp->fd_spin);
1069         }
1070         fdexpand++;
1071 }
1072
1073 /*
1074  * Number of nodes in right subtree, including the root.
1075  */
1076 static __inline int
1077 right_subtree_size(int n)
1078 {
1079         return (n ^ (n | (n + 1)));
1080 }
1081
1082 /*
1083  * Bigger ancestor.
1084  */
1085 static __inline int
1086 right_ancestor(int n)
1087 {
1088         return (n | (n + 1));
1089 }
1090
1091 /*
1092  * Smaller ancestor.
1093  */
1094 static __inline int
1095 left_ancestor(int n)
1096 {
1097         return ((n & (n + 1)) - 1);
1098 }
1099
1100 /*
1101  * Traverse the in-place binary tree buttom-up adjusting the allocation
1102  * count so scans can determine where free descriptors are located.
1103  *
1104  * MPSAFE - caller must be holding an exclusive spinlock on fdp
1105  */
1106 static
1107 void
1108 fdreserve_locked(struct filedesc *fdp, int fd, int incr)
1109 {
1110         while (fd >= 0) {
1111                 fdp->fd_files[fd].allocated += incr;
1112                 KKASSERT(fdp->fd_files[fd].allocated >= 0);
1113                 fd = left_ancestor(fd);
1114         }
1115 }
1116
1117 /*
1118  * Reserve a file descriptor for the process.  If no error occurs, the
1119  * caller MUST at some point call fsetfd() or assign a file pointer
1120  * or dispose of the reservation.
1121  *
1122  * MPSAFE
1123  */
1124 int
1125 fdalloc(struct proc *p, int want, int *result)
1126 {
1127         struct filedesc *fdp = p->p_fd;
1128         struct uidinfo *uip;
1129         int fd, rsize, rsum, node, lim;
1130
1131         /*
1132          * Check dtable size limit
1133          */
1134         spin_lock(&p->p_limit->p_spin);
1135         if (p->p_rlimit[RLIMIT_NOFILE].rlim_cur > INT_MAX)
1136                 lim = INT_MAX;
1137         else
1138                 lim = (int)p->p_rlimit[RLIMIT_NOFILE].rlim_cur;
1139         spin_unlock(&p->p_limit->p_spin);
1140
1141         if (lim > maxfilesperproc)
1142                 lim = maxfilesperproc;
1143         if (lim < minfilesperproc)
1144                 lim = minfilesperproc;
1145         if (want >= lim)
1146                 return (EMFILE);
1147
1148         /*
1149          * Check that the user has not run out of descriptors (non-root only).
1150          * As a safety measure the dtable is allowed to have at least
1151          * minfilesperproc open fds regardless of the maxfilesperuser limit.
1152          */
1153         if (p->p_ucred->cr_uid && fdp->fd_nfiles >= minfilesperproc) {
1154                 uip = p->p_ucred->cr_uidinfo;
1155                 if (uip->ui_openfiles > maxfilesperuser) {
1156                         krateprintf(&krate_uidinfo,
1157                                     "Warning: user %d pid %d (%s) ran out of "
1158                                     "file descriptors (%d/%d)\n",
1159                                     p->p_ucred->cr_uid, (int)p->p_pid,
1160                                     p->p_comm,
1161                                     uip->ui_openfiles, maxfilesperuser);
1162                         return(ENFILE);
1163                 }
1164         }
1165
1166         /*
1167          * Grow the dtable if necessary
1168          */
1169         spin_lock(&fdp->fd_spin);
1170         if (want >= fdp->fd_nfiles)
1171                 fdgrow_locked(fdp, want);
1172
1173         /*
1174          * Search for a free descriptor starting at the higher
1175          * of want or fd_freefile.  If that fails, consider
1176          * expanding the ofile array.
1177          *
1178          * NOTE! the 'allocated' field is a cumulative recursive allocation
1179          * count.  If we happen to see a value of 0 then we can shortcut
1180          * our search.  Otherwise we run through through the tree going
1181          * down branches we know have free descriptor(s) until we hit a
1182          * leaf node.  The leaf node will be free but will not necessarily
1183          * have an allocated field of 0.
1184          */
1185 retry:
1186         /* move up the tree looking for a subtree with a free node */
1187         for (fd = max(want, fdp->fd_freefile); fd < min(fdp->fd_nfiles, lim);
1188              fd = right_ancestor(fd)) {
1189                 if (fdp->fd_files[fd].allocated == 0)
1190                         goto found;
1191
1192                 rsize = right_subtree_size(fd);
1193                 if (fdp->fd_files[fd].allocated == rsize)
1194                         continue;       /* right subtree full */
1195
1196                 /*
1197                  * Free fd is in the right subtree of the tree rooted at fd.
1198                  * Call that subtree R.  Look for the smallest (leftmost)
1199                  * subtree of R with an unallocated fd: continue moving
1200                  * down the left branch until encountering a full left
1201                  * subtree, then move to the right.
1202                  */
1203                 for (rsum = 0, rsize /= 2; rsize > 0; rsize /= 2) {
1204                         node = fd + rsize;
1205                         rsum += fdp->fd_files[node].allocated;
1206                         if (fdp->fd_files[fd].allocated == rsum + rsize) {
1207                                 fd = node;      /* move to the right */
1208                                 if (fdp->fd_files[node].allocated == 0)
1209                                         goto found;
1210                                 rsum = 0;
1211                         }
1212                 }
1213                 goto found;
1214         }
1215
1216         /*
1217          * No space in current array.  Expand?
1218          */
1219         if (fdp->fd_nfiles >= lim) {
1220                 spin_unlock(&fdp->fd_spin);
1221                 return (EMFILE);
1222         }
1223         fdgrow_locked(fdp, want);
1224         goto retry;
1225
1226 found:
1227         KKASSERT(fd < fdp->fd_nfiles);
1228         if (fd > fdp->fd_lastfile)
1229                 fdp->fd_lastfile = fd;
1230         if (want <= fdp->fd_freefile)
1231                 fdp->fd_freefile = fd;
1232         *result = fd;
1233         KKASSERT(fdp->fd_files[fd].fp == NULL);
1234         KKASSERT(fdp->fd_files[fd].reserved == 0);
1235         fdp->fd_files[fd].fileflags = 0;
1236         fdp->fd_files[fd].reserved = 1;
1237         fdreserve_locked(fdp, fd, 1);
1238         spin_unlock(&fdp->fd_spin);
1239         return (0);
1240 }
1241
1242 /*
1243  * Check to see whether n user file descriptors
1244  * are available to the process p.
1245  *
1246  * MPSAFE
1247  */
1248 int
1249 fdavail(struct proc *p, int n)
1250 {
1251         struct filedesc *fdp = p->p_fd;
1252         struct fdnode *fdnode;
1253         int i, lim, last;
1254
1255         spin_lock(&p->p_limit->p_spin);
1256         if (p->p_rlimit[RLIMIT_NOFILE].rlim_cur > INT_MAX)
1257                 lim = INT_MAX;
1258         else
1259                 lim = (int)p->p_rlimit[RLIMIT_NOFILE].rlim_cur;
1260         spin_unlock(&p->p_limit->p_spin);
1261
1262         if (lim > maxfilesperproc)
1263                 lim = maxfilesperproc;
1264         if (lim < minfilesperproc)
1265                 lim = minfilesperproc;
1266
1267         spin_lock(&fdp->fd_spin);
1268         if ((i = lim - fdp->fd_nfiles) > 0 && (n -= i) <= 0) {
1269                 spin_unlock(&fdp->fd_spin);
1270                 return (1);
1271         }
1272         last = min(fdp->fd_nfiles, lim);
1273         fdnode = &fdp->fd_files[fdp->fd_freefile];
1274         for (i = last - fdp->fd_freefile; --i >= 0; ++fdnode) {
1275                 if (fdnode->fp == NULL && --n <= 0) {
1276                         spin_unlock(&fdp->fd_spin);
1277                         return (1);
1278                 }
1279         }
1280         spin_unlock(&fdp->fd_spin);
1281         return (0);
1282 }
1283
1284 /*
1285  * Revoke open descriptors referencing (f_data, f_type)
1286  *
1287  * Any revoke executed within a prison is only able to
1288  * revoke descriptors for processes within that prison.
1289  *
1290  * Returns 0 on success or an error code.
1291  */
1292 struct fdrevoke_info {
1293         void *data;
1294         short type;
1295         short unused;
1296         int count;
1297         int intransit;
1298         struct ucred *cred;
1299         struct file *nfp;
1300 };
1301
1302 static int fdrevoke_check_callback(struct file *fp, void *vinfo);
1303 static int fdrevoke_proc_callback(struct proc *p, void *vinfo);
1304
1305 int
1306 fdrevoke(void *f_data, short f_type, struct ucred *cred)
1307 {
1308         struct fdrevoke_info info;
1309         int error;
1310
1311         bzero(&info, sizeof(info));
1312         info.data = f_data;
1313         info.type = f_type;
1314         info.cred = cred;
1315         error = falloc(NULL, &info.nfp, NULL);
1316         if (error)
1317                 return (error);
1318
1319         /*
1320          * Scan the file pointer table once.  dups do not dup file pointers,
1321          * only descriptors, so there is no leak.  Set FREVOKED on the fps
1322          * being revoked.
1323          */
1324         allfiles_scan_exclusive(fdrevoke_check_callback, &info);
1325
1326         /*
1327          * If any fps were marked track down the related descriptors
1328          * and close them.  Any dup()s at this point will notice
1329          * the FREVOKED already set in the fp and do the right thing.
1330          *
1331          * Any fps with non-zero msgcounts (aka sent over a unix-domain
1332          * socket) bumped the intransit counter and will require a
1333          * scan.  Races against fps leaving the socket are closed by
1334          * the socket code checking for FREVOKED.
1335          */
1336         if (info.count)
1337                 allproc_scan(fdrevoke_proc_callback, &info);
1338         if (info.intransit)
1339                 unp_revoke_gc(info.nfp);
1340         fdrop(info.nfp);
1341         return(0);
1342 }
1343
1344 /*
1345  * Locate matching file pointers directly.
1346  *
1347  * WARNING: allfiles_scan_exclusive() holds a spinlock through these calls!
1348  */
1349 static int
1350 fdrevoke_check_callback(struct file *fp, void *vinfo)
1351 {
1352         struct fdrevoke_info *info = vinfo;
1353
1354         /*
1355          * File pointers already flagged for revokation are skipped.
1356          */
1357         if (fp->f_flag & FREVOKED)
1358                 return(0);
1359
1360         /*
1361          * If revoking from a prison file pointers created outside of
1362          * that prison, or file pointers without creds, cannot be revoked.
1363          */
1364         if (info->cred->cr_prison &&
1365             (fp->f_cred == NULL ||
1366              info->cred->cr_prison != fp->f_cred->cr_prison)) {
1367                 return(0);
1368         }
1369
1370         /*
1371          * If the file pointer matches then mark it for revocation.  The
1372          * flag is currently only used by unp_revoke_gc().
1373          *
1374          * info->count is a heuristic and can race in a SMP environment.
1375          */
1376         if (info->data == fp->f_data && info->type == fp->f_type) {
1377                 atomic_set_int(&fp->f_flag, FREVOKED);
1378                 info->count += fp->f_count;
1379                 if (fp->f_msgcount)
1380                         ++info->intransit;
1381         }
1382         return(0);
1383 }
1384
1385 /*
1386  * Locate matching file pointers via process descriptor tables.
1387  */
1388 static int
1389 fdrevoke_proc_callback(struct proc *p, void *vinfo)
1390 {
1391         struct fdrevoke_info *info = vinfo;
1392         struct filedesc *fdp;
1393         struct file *fp;
1394         int n;
1395
1396         if (p->p_stat == SIDL || p->p_stat == SZOMB)
1397                 return(0);
1398         if (info->cred->cr_prison &&
1399             info->cred->cr_prison != p->p_ucred->cr_prison) {
1400                 return(0);
1401         }
1402
1403         /*
1404          * If the controlling terminal of the process matches the
1405          * vnode being revoked we clear the controlling terminal.
1406          *
1407          * The normal spec_close() may not catch this because it
1408          * uses curproc instead of p.
1409          */
1410         if (p->p_session && info->type == DTYPE_VNODE &&
1411             info->data == p->p_session->s_ttyvp) {
1412                 p->p_session->s_ttyvp = NULL;
1413                 vrele(info->data);
1414         }
1415
1416         /*
1417          * Softref the fdp to prevent it from being destroyed
1418          */
1419         spin_lock(&p->p_spin);
1420         if ((fdp = p->p_fd) == NULL) {
1421                 spin_unlock(&p->p_spin);
1422                 return(0);
1423         }
1424         atomic_add_int(&fdp->fd_softrefs, 1);
1425         spin_unlock(&p->p_spin);
1426
1427         /*
1428          * Locate and close any matching file descriptors.
1429          */
1430         spin_lock(&fdp->fd_spin);
1431         for (n = 0; n < fdp->fd_nfiles; ++n) {
1432                 if ((fp = fdp->fd_files[n].fp) == NULL)
1433                         continue;
1434                 if (fp->f_flag & FREVOKED) {
1435                         fhold(info->nfp);
1436                         fdp->fd_files[n].fp = info->nfp;
1437                         spin_unlock(&fdp->fd_spin);
1438                         knote_fdclose(fp, fdp, n);      /* XXX */
1439                         closef(fp, p);
1440                         spin_lock(&fdp->fd_spin);
1441                         --info->count;
1442                 }
1443         }
1444         spin_unlock(&fdp->fd_spin);
1445         atomic_subtract_int(&fdp->fd_softrefs, 1);
1446         return(0);
1447 }
1448
1449 /*
1450  * falloc:
1451  *      Create a new open file structure and reserve a file decriptor
1452  *      for the process that refers to it.
1453  *
1454  *      Root creds are checked using lp, or assumed if lp is NULL.  If
1455  *      resultfd is non-NULL then lp must also be non-NULL.  No file
1456  *      descriptor is reserved (and no process context is needed) if
1457  *      resultfd is NULL.
1458  *
1459  *      A file pointer with a refcount of 1 is returned.  Note that the
1460  *      file pointer is NOT associated with the descriptor.  If falloc
1461  *      returns success, fsetfd() MUST be called to either associate the
1462  *      file pointer or clear the reservation.
1463  *
1464  * MPSAFE
1465  */
1466 int
1467 falloc(struct lwp *lp, struct file **resultfp, int *resultfd)
1468 {
1469         static struct timeval lastfail;
1470         static int curfail;
1471         struct file *fp;
1472         struct ucred *cred = lp ? lp->lwp_thread->td_ucred : proc0.p_ucred;
1473         int error;
1474
1475         fp = NULL;
1476
1477         /*
1478          * Handle filetable full issues and root overfill.
1479          */
1480         if (nfiles >= maxfiles - maxfilesrootres &&
1481             (cred->cr_ruid != 0 || nfiles >= maxfiles)) {
1482                 if (ppsratecheck(&lastfail, &curfail, 1)) {
1483                         kprintf("kern.maxfiles limit exceeded by uid %d, "
1484                                 "please see tuning(7).\n",
1485                                 cred->cr_ruid);
1486                 }
1487                 error = ENFILE;
1488                 goto done;
1489         }
1490
1491         /*
1492          * Allocate a new file descriptor.
1493          */
1494         fp = kmalloc(sizeof(struct file), M_FILE, M_WAITOK | M_ZERO);
1495         spin_init(&fp->f_spin);
1496         SLIST_INIT(&fp->f_klist);
1497         fp->f_count = 1;
1498         fp->f_ops = &badfileops;
1499         fp->f_seqcount = 1;
1500         fsetcred(fp, cred);
1501         spin_lock(&filehead_spin);
1502         nfiles++;
1503         LIST_INSERT_HEAD(&filehead, fp, f_list);
1504         spin_unlock(&filehead_spin);
1505         if (resultfd) {
1506                 if ((error = fdalloc(lp->lwp_proc, 0, resultfd)) != 0) {
1507                         fdrop(fp);
1508                         fp = NULL;
1509                 }
1510         } else {
1511                 error = 0;
1512         }
1513 done:
1514         *resultfp = fp;
1515         return (error);
1516 }
1517
1518 /*
1519  * Check for races against a file descriptor by determining that the
1520  * file pointer is still associated with the specified file descriptor,
1521  * and a close is not currently in progress.
1522  *
1523  * MPSAFE
1524  */
1525 int
1526 checkfdclosed(struct filedesc *fdp, int fd, struct file *fp)
1527 {
1528         int error;
1529
1530         spin_lock(&fdp->fd_spin);
1531         if ((unsigned)fd >= fdp->fd_nfiles || fp != fdp->fd_files[fd].fp)
1532                 error = EBADF;
1533         else
1534                 error = 0;
1535         spin_unlock(&fdp->fd_spin);
1536         return (error);
1537 }
1538
1539 /*
1540  * Associate a file pointer with a previously reserved file descriptor.
1541  * This function always succeeds.
1542  *
1543  * If fp is NULL, the file descriptor is returned to the pool.
1544  */
1545
1546 /*
1547  * MPSAFE (exclusive spinlock must be held on call)
1548  */
1549 static void
1550 fsetfd_locked(struct filedesc *fdp, struct file *fp, int fd)
1551 {
1552         KKASSERT((unsigned)fd < fdp->fd_nfiles);
1553         KKASSERT(fdp->fd_files[fd].reserved != 0);
1554         if (fp) {
1555                 fhold(fp);
1556                 fdp->fd_files[fd].fp = fp;
1557                 fdp->fd_files[fd].reserved = 0;
1558         } else {
1559                 fdp->fd_files[fd].reserved = 0;
1560                 fdreserve_locked(fdp, fd, -1);
1561                 fdfixup_locked(fdp, fd);
1562         }
1563 }
1564
1565 /*
1566  * MPSAFE
1567  */
1568 void
1569 fsetfd(struct filedesc *fdp, struct file *fp, int fd)
1570 {
1571         spin_lock(&fdp->fd_spin);
1572         fsetfd_locked(fdp, fp, fd);
1573         spin_unlock(&fdp->fd_spin);
1574 }
1575
1576 /*
1577  * MPSAFE (exclusive spinlock must be held on call)
1578  */
1579 static
1580 struct file *
1581 funsetfd_locked(struct filedesc *fdp, int fd)
1582 {
1583         struct file *fp;
1584
1585         if ((unsigned)fd >= fdp->fd_nfiles)
1586                 return (NULL);
1587         if ((fp = fdp->fd_files[fd].fp) == NULL)
1588                 return (NULL);
1589         fdp->fd_files[fd].fp = NULL;
1590         fdp->fd_files[fd].fileflags = 0;
1591
1592         fdreserve_locked(fdp, fd, -1);
1593         fdfixup_locked(fdp, fd);
1594         return(fp);
1595 }
1596
1597 /*
1598  * MPSAFE
1599  */
1600 int
1601 fgetfdflags(struct filedesc *fdp, int fd, int *flagsp)
1602 {
1603         int error;
1604
1605         spin_lock(&fdp->fd_spin);
1606         if (((u_int)fd) >= fdp->fd_nfiles) {
1607                 error = EBADF;
1608         } else if (fdp->fd_files[fd].fp == NULL) {
1609                 error = EBADF;
1610         } else {
1611                 *flagsp = fdp->fd_files[fd].fileflags;
1612                 error = 0;
1613         }
1614         spin_unlock(&fdp->fd_spin);
1615         return (error);
1616 }
1617
1618 /*
1619  * MPSAFE
1620  */
1621 int
1622 fsetfdflags(struct filedesc *fdp, int fd, int add_flags)
1623 {
1624         int error;
1625
1626         spin_lock(&fdp->fd_spin);
1627         if (((u_int)fd) >= fdp->fd_nfiles) {
1628                 error = EBADF;
1629         } else if (fdp->fd_files[fd].fp == NULL) {
1630                 error = EBADF;
1631         } else {
1632                 fdp->fd_files[fd].fileflags |= add_flags;
1633                 error = 0;
1634         }
1635         spin_unlock(&fdp->fd_spin);
1636         return (error);
1637 }
1638
1639 /*
1640  * MPSAFE
1641  */
1642 int
1643 fclrfdflags(struct filedesc *fdp, int fd, int rem_flags)
1644 {
1645         int error;
1646
1647         spin_lock(&fdp->fd_spin);
1648         if (((u_int)fd) >= fdp->fd_nfiles) {
1649                 error = EBADF;
1650         } else if (fdp->fd_files[fd].fp == NULL) {
1651                 error = EBADF;
1652         } else {
1653                 fdp->fd_files[fd].fileflags &= ~rem_flags;
1654                 error = 0;
1655         }
1656         spin_unlock(&fdp->fd_spin);
1657         return (error);
1658 }
1659
1660 /*
1661  * Set/Change/Clear the creds for a fp and synchronize the uidinfo.
1662  */
1663 void
1664 fsetcred(struct file *fp, struct ucred *ncr)
1665 {
1666         struct ucred *ocr;
1667         struct uidinfo *uip;
1668
1669         ocr = fp->f_cred;
1670         if (ocr == NULL || ncr == NULL || ocr->cr_uidinfo != ncr->cr_uidinfo) {
1671                 if (ocr) {
1672                         uip = ocr->cr_uidinfo;
1673                         atomic_add_int(&uip->ui_openfiles, -1);
1674                 }
1675                 if (ncr) {
1676                         uip = ncr->cr_uidinfo;
1677                         atomic_add_int(&uip->ui_openfiles, 1);
1678                 }
1679         }
1680         if (ncr)
1681                 crhold(ncr);
1682         fp->f_cred = ncr;
1683         if (ocr)
1684                 crfree(ocr);
1685 }
1686
1687 /*
1688  * Free a file descriptor.
1689  */
1690 static
1691 void
1692 ffree(struct file *fp)
1693 {
1694         KASSERT((fp->f_count == 0), ("ffree: fp_fcount not 0!"));
1695         spin_lock(&filehead_spin);
1696         LIST_REMOVE(fp, f_list);
1697         nfiles--;
1698         spin_unlock(&filehead_spin);
1699         fsetcred(fp, NULL);
1700         if (fp->f_nchandle.ncp)
1701             cache_drop(&fp->f_nchandle);
1702         kfree(fp, M_FILE);
1703 }
1704
1705 /*
1706  * called from init_main, initialize filedesc0 for proc0.
1707  */
1708 void
1709 fdinit_bootstrap(struct proc *p0, struct filedesc *fdp0, int cmask)
1710 {
1711         p0->p_fd = fdp0;
1712         p0->p_fdtol = NULL;
1713         fdp0->fd_refcnt = 1;
1714         fdp0->fd_cmask = cmask;
1715         fdp0->fd_files = fdp0->fd_builtin_files;
1716         fdp0->fd_nfiles = NDFILE;
1717         fdp0->fd_lastfile = -1;
1718         spin_init(&fdp0->fd_spin);
1719 }
1720
1721 /*
1722  * Build a new filedesc structure.
1723  *
1724  * NOT MPSAFE (vref)
1725  */
1726 struct filedesc *
1727 fdinit(struct proc *p)
1728 {
1729         struct filedesc *newfdp;
1730         struct filedesc *fdp = p->p_fd;
1731
1732         newfdp = kmalloc(sizeof(struct filedesc), M_FILEDESC, M_WAITOK|M_ZERO);
1733         spin_lock(&fdp->fd_spin);
1734         if (fdp->fd_cdir) {
1735                 newfdp->fd_cdir = fdp->fd_cdir;
1736                 vref(newfdp->fd_cdir);
1737                 cache_copy(&fdp->fd_ncdir, &newfdp->fd_ncdir);
1738         }
1739
1740         /*
1741          * rdir may not be set in e.g. proc0 or anything vm_fork'd off of
1742          * proc0, but should unconditionally exist in other processes.
1743          */
1744         if (fdp->fd_rdir) {
1745                 newfdp->fd_rdir = fdp->fd_rdir;
1746                 vref(newfdp->fd_rdir);
1747                 cache_copy(&fdp->fd_nrdir, &newfdp->fd_nrdir);
1748         }
1749         if (fdp->fd_jdir) {
1750                 newfdp->fd_jdir = fdp->fd_jdir;
1751                 vref(newfdp->fd_jdir);
1752                 cache_copy(&fdp->fd_njdir, &newfdp->fd_njdir);
1753         }
1754         spin_unlock(&fdp->fd_spin);
1755
1756         /* Create the file descriptor table. */
1757         newfdp->fd_refcnt = 1;
1758         newfdp->fd_cmask = cmask;
1759         newfdp->fd_files = newfdp->fd_builtin_files;
1760         newfdp->fd_nfiles = NDFILE;
1761         newfdp->fd_lastfile = -1;
1762         spin_init(&newfdp->fd_spin);
1763
1764         return (newfdp);
1765 }
1766
1767 /*
1768  * Share a filedesc structure.
1769  *
1770  * MPSAFE
1771  */
1772 struct filedesc *
1773 fdshare(struct proc *p)
1774 {
1775         struct filedesc *fdp;
1776
1777         fdp = p->p_fd;
1778         spin_lock(&fdp->fd_spin);
1779         fdp->fd_refcnt++;
1780         spin_unlock(&fdp->fd_spin);
1781         return (fdp);
1782 }
1783
1784 /*
1785  * Copy a filedesc structure.
1786  *
1787  * MPSAFE
1788  */
1789 struct filedesc *
1790 fdcopy(struct proc *p)
1791 {
1792         struct filedesc *fdp = p->p_fd;
1793         struct filedesc *newfdp;
1794         struct fdnode *fdnode;
1795         int i;
1796         int ni;
1797
1798         /*
1799          * Certain daemons might not have file descriptors.
1800          */
1801         if (fdp == NULL)
1802                 return (NULL);
1803
1804         /*
1805          * Allocate the new filedesc and fd_files[] array.  This can race
1806          * with operations by other threads on the fdp so we have to be
1807          * careful.
1808          */
1809         newfdp = kmalloc(sizeof(struct filedesc), M_FILEDESC, M_WAITOK | M_ZERO);
1810 again:
1811         spin_lock(&fdp->fd_spin);
1812         if (fdp->fd_lastfile < NDFILE) {
1813                 newfdp->fd_files = newfdp->fd_builtin_files;
1814                 i = NDFILE;
1815         } else {
1816                 /*
1817                  * We have to allocate (N^2-1) entries for our in-place
1818                  * binary tree.  Allow the table to shrink.
1819                  */
1820                 i = fdp->fd_nfiles;
1821                 ni = (i - 1) / 2;
1822                 while (ni > fdp->fd_lastfile && ni > NDFILE) {
1823                         i = ni;
1824                         ni = (i - 1) / 2;
1825                 }
1826                 spin_unlock(&fdp->fd_spin);
1827                 newfdp->fd_files = kmalloc(i * sizeof(struct fdnode),
1828                                           M_FILEDESC, M_WAITOK | M_ZERO);
1829
1830                 /*
1831                  * Check for race, retry
1832                  */
1833                 spin_lock(&fdp->fd_spin);
1834                 if (i <= fdp->fd_lastfile) {
1835                         spin_unlock(&fdp->fd_spin);
1836                         kfree(newfdp->fd_files, M_FILEDESC);
1837                         goto again;
1838                 }
1839         }
1840
1841         /*
1842          * Dup the remaining fields. vref() and cache_hold() can be
1843          * safely called while holding the read spinlock on fdp.
1844          *
1845          * The read spinlock on fdp is still being held.
1846          *
1847          * NOTE: vref and cache_hold calls for the case where the vnode
1848          * or cache entry already has at least one ref may be called
1849          * while holding spin locks.
1850          */
1851         if ((newfdp->fd_cdir = fdp->fd_cdir) != NULL) {
1852                 vref(newfdp->fd_cdir);
1853                 cache_copy(&fdp->fd_ncdir, &newfdp->fd_ncdir);
1854         }
1855         /*
1856          * We must check for fd_rdir here, at least for now because
1857          * the init process is created before we have access to the
1858          * rootvode to take a reference to it.
1859          */
1860         if ((newfdp->fd_rdir = fdp->fd_rdir) != NULL) {
1861                 vref(newfdp->fd_rdir);
1862                 cache_copy(&fdp->fd_nrdir, &newfdp->fd_nrdir);
1863         }
1864         if ((newfdp->fd_jdir = fdp->fd_jdir) != NULL) {
1865                 vref(newfdp->fd_jdir);
1866                 cache_copy(&fdp->fd_njdir, &newfdp->fd_njdir);
1867         }
1868         newfdp->fd_refcnt = 1;
1869         newfdp->fd_nfiles = i;
1870         newfdp->fd_lastfile = fdp->fd_lastfile;
1871         newfdp->fd_freefile = fdp->fd_freefile;
1872         newfdp->fd_cmask = fdp->fd_cmask;
1873         spin_init(&newfdp->fd_spin);
1874
1875         /*
1876          * Copy the descriptor table through (i).  This also copies the
1877          * allocation state.   Then go through and ref the file pointers
1878          * and clean up any KQ descriptors.
1879          *
1880          * kq descriptors cannot be copied.  Since we haven't ref'd the
1881          * copied files yet we can ignore the return value from funsetfd().
1882          *
1883          * The read spinlock on fdp is still being held.
1884          */
1885         bcopy(fdp->fd_files, newfdp->fd_files, i * sizeof(struct fdnode));
1886         for (i = 0 ; i < newfdp->fd_nfiles; ++i) {
1887                 fdnode = &newfdp->fd_files[i];
1888                 if (fdnode->reserved) {
1889                         fdreserve_locked(newfdp, i, -1);
1890                         fdnode->reserved = 0;
1891                         fdfixup_locked(newfdp, i);
1892                 } else if (fdnode->fp) {
1893                         if (fdnode->fp->f_type == DTYPE_KQUEUE) {
1894                                 (void)funsetfd_locked(newfdp, i);
1895                         } else {
1896                                 fhold(fdnode->fp);
1897                         }
1898                 }
1899         }
1900         spin_unlock(&fdp->fd_spin);
1901         return (newfdp);
1902 }
1903
1904 /*
1905  * Release a filedesc structure.
1906  *
1907  * NOT MPSAFE (MPSAFE for refs > 1, but the final cleanup code is not MPSAFE)
1908  */
1909 void
1910 fdfree(struct proc *p, struct filedesc *repl)
1911 {
1912         struct filedesc *fdp;
1913         struct fdnode *fdnode;
1914         int i;
1915         struct filedesc_to_leader *fdtol;
1916         struct file *fp;
1917         struct vnode *vp;
1918         struct flock lf;
1919
1920         /*
1921          * Certain daemons might not have file descriptors.
1922          */
1923         fdp = p->p_fd;
1924         if (fdp == NULL) {
1925                 p->p_fd = repl;
1926                 return;
1927         }
1928
1929         /*
1930          * Severe messing around to follow.
1931          */
1932         spin_lock(&fdp->fd_spin);
1933
1934         /* Check for special need to clear POSIX style locks */
1935         fdtol = p->p_fdtol;
1936         if (fdtol != NULL) {
1937                 KASSERT(fdtol->fdl_refcount > 0,
1938                         ("filedesc_to_refcount botch: fdl_refcount=%d",
1939                          fdtol->fdl_refcount));
1940                 if (fdtol->fdl_refcount == 1 &&
1941                     (p->p_leader->p_flag & P_ADVLOCK) != 0) {
1942                         for (i = 0; i <= fdp->fd_lastfile; ++i) {
1943                                 fdnode = &fdp->fd_files[i];
1944                                 if (fdnode->fp == NULL ||
1945                                     fdnode->fp->f_type != DTYPE_VNODE) {
1946                                         continue;
1947                                 }
1948                                 fp = fdnode->fp;
1949                                 fhold(fp);
1950                                 spin_unlock(&fdp->fd_spin);
1951
1952                                 lf.l_whence = SEEK_SET;
1953                                 lf.l_start = 0;
1954                                 lf.l_len = 0;
1955                                 lf.l_type = F_UNLCK;
1956                                 vp = (struct vnode *)fp->f_data;
1957                                 (void) VOP_ADVLOCK(vp,
1958                                                    (caddr_t)p->p_leader,
1959                                                    F_UNLCK,
1960                                                    &lf,
1961                                                    F_POSIX);
1962                                 fdrop(fp);
1963                                 spin_lock(&fdp->fd_spin);
1964                         }
1965                 }
1966         retry:
1967                 if (fdtol->fdl_refcount == 1) {
1968                         if (fdp->fd_holdleaderscount > 0 &&
1969                             (p->p_leader->p_flag & P_ADVLOCK) != 0) {
1970                                 /*
1971                                  * close() or do_dup() has cleared a reference
1972                                  * in a shared file descriptor table.
1973                                  */
1974                                 fdp->fd_holdleaderswakeup = 1;
1975                                 ssleep(&fdp->fd_holdleaderscount,
1976                                        &fdp->fd_spin, 0, "fdlhold", 0);
1977                                 goto retry;
1978                         }
1979                         if (fdtol->fdl_holdcount > 0) {
1980                                 /*
1981                                  * Ensure that fdtol->fdl_leader
1982                                  * remains valid in closef().
1983                                  */
1984                                 fdtol->fdl_wakeup = 1;
1985                                 ssleep(fdtol, &fdp->fd_spin, 0, "fdlhold", 0);
1986                                 goto retry;
1987                         }
1988                 }
1989                 fdtol->fdl_refcount--;
1990                 if (fdtol->fdl_refcount == 0 &&
1991                     fdtol->fdl_holdcount == 0) {
1992                         fdtol->fdl_next->fdl_prev = fdtol->fdl_prev;
1993                         fdtol->fdl_prev->fdl_next = fdtol->fdl_next;
1994                 } else {
1995                         fdtol = NULL;
1996                 }
1997                 p->p_fdtol = NULL;
1998                 if (fdtol != NULL) {
1999                         spin_unlock(&fdp->fd_spin);
2000                         kfree(fdtol, M_FILEDESC_TO_LEADER);
2001                         spin_lock(&fdp->fd_spin);
2002                 }
2003         }
2004         if (--fdp->fd_refcnt > 0) {
2005                 spin_unlock(&fdp->fd_spin);
2006                 spin_lock(&p->p_spin);
2007                 p->p_fd = repl;
2008                 spin_unlock(&p->p_spin);
2009                 return;
2010         }
2011
2012         /*
2013          * Even though we are the last reference to the structure allproc
2014          * scans may still reference the structure.  Maintain proper
2015          * locks until we can replace p->p_fd.
2016          *
2017          * Also note that kqueue's closef still needs to reference the
2018          * fdp via p->p_fd, so we have to close the descriptors before
2019          * we replace p->p_fd.
2020          */
2021         for (i = 0; i <= fdp->fd_lastfile; ++i) {
2022                 if (fdp->fd_files[i].fp) {
2023                         fp = funsetfd_locked(fdp, i);
2024                         if (fp) {
2025                                 spin_unlock(&fdp->fd_spin);
2026                                 if (SLIST_FIRST(&fp->f_klist))
2027                                         knote_fdclose(fp, fdp, i);
2028                                 closef(fp, p);
2029                                 spin_lock(&fdp->fd_spin);
2030                         }
2031                 }
2032         }
2033         spin_unlock(&fdp->fd_spin);
2034
2035         /*
2036          * Interlock against an allproc scan operations (typically frevoke).
2037          */
2038         spin_lock(&p->p_spin);
2039         p->p_fd = repl;
2040         spin_unlock(&p->p_spin);
2041
2042         /*
2043          * Wait for any softrefs to go away.  This race rarely occurs so
2044          * we can use a non-critical-path style poll/sleep loop.  The
2045          * race only occurs against allproc scans.
2046          *
2047          * No new softrefs can occur with the fdp disconnected from the
2048          * process.
2049          */
2050         if (fdp->fd_softrefs) {
2051                 kprintf("pid %d: Warning, fdp race avoided\n", p->p_pid);
2052                 while (fdp->fd_softrefs)
2053                         tsleep(&fdp->fd_softrefs, 0, "fdsoft", 1);
2054         }
2055
2056         if (fdp->fd_files != fdp->fd_builtin_files)
2057                 kfree(fdp->fd_files, M_FILEDESC);
2058         if (fdp->fd_cdir) {
2059                 cache_drop(&fdp->fd_ncdir);
2060                 vrele(fdp->fd_cdir);
2061         }
2062         if (fdp->fd_rdir) {
2063                 cache_drop(&fdp->fd_nrdir);
2064                 vrele(fdp->fd_rdir);
2065         }
2066         if (fdp->fd_jdir) {
2067                 cache_drop(&fdp->fd_njdir);
2068                 vrele(fdp->fd_jdir);
2069         }
2070         kfree(fdp, M_FILEDESC);
2071 }
2072
2073 /*
2074  * Retrieve and reference the file pointer associated with a descriptor.
2075  *
2076  * MPSAFE
2077  */
2078 struct file *
2079 holdfp(struct filedesc *fdp, int fd, int flag)
2080 {
2081         struct file* fp;
2082
2083         spin_lock(&fdp->fd_spin);
2084         if (((u_int)fd) >= fdp->fd_nfiles) {
2085                 fp = NULL;
2086                 goto done;
2087         }
2088         if ((fp = fdp->fd_files[fd].fp) == NULL)
2089                 goto done;
2090         if ((fp->f_flag & flag) == 0 && flag != -1) {
2091                 fp = NULL;
2092                 goto done;
2093         }
2094         fhold(fp);
2095 done:
2096         spin_unlock(&fdp->fd_spin);
2097         return (fp);
2098 }
2099
2100 /*
2101  * holdsock() - load the struct file pointer associated
2102  * with a socket into *fpp.  If an error occurs, non-zero
2103  * will be returned and *fpp will be set to NULL.
2104  *
2105  * MPSAFE
2106  */
2107 int
2108 holdsock(struct filedesc *fdp, int fd, struct file **fpp)
2109 {
2110         struct file *fp;
2111         int error;
2112
2113         spin_lock(&fdp->fd_spin);
2114         if ((unsigned)fd >= fdp->fd_nfiles) {
2115                 error = EBADF;
2116                 fp = NULL;
2117                 goto done;
2118         }
2119         if ((fp = fdp->fd_files[fd].fp) == NULL) {
2120                 error = EBADF;
2121                 goto done;
2122         }
2123         if (fp->f_type != DTYPE_SOCKET) {
2124                 error = ENOTSOCK;
2125                 goto done;
2126         }
2127         fhold(fp);
2128         error = 0;
2129 done:
2130         spin_unlock(&fdp->fd_spin);
2131         *fpp = fp;
2132         return (error);
2133 }
2134
2135 /*
2136  * Convert a user file descriptor to a held file pointer.
2137  *
2138  * MPSAFE
2139  */
2140 int
2141 holdvnode(struct filedesc *fdp, int fd, struct file **fpp)
2142 {
2143         struct file *fp;
2144         int error;
2145
2146         spin_lock(&fdp->fd_spin);
2147         if ((unsigned)fd >= fdp->fd_nfiles) {
2148                 error = EBADF;
2149                 fp = NULL;
2150                 goto done;
2151         }
2152         if ((fp = fdp->fd_files[fd].fp) == NULL) {
2153                 error = EBADF;
2154                 goto done;
2155         }
2156         if (fp->f_type != DTYPE_VNODE && fp->f_type != DTYPE_FIFO) {
2157                 fp = NULL;
2158                 error = EINVAL;
2159                 goto done;
2160         }
2161         fhold(fp);
2162         error = 0;
2163 done:
2164         spin_unlock(&fdp->fd_spin);
2165         *fpp = fp;
2166         return (error);
2167 }
2168
2169 /*
2170  * For setugid programs, we don't want to people to use that setugidness
2171  * to generate error messages which write to a file which otherwise would
2172  * otherwise be off-limits to the process.
2173  *
2174  * This is a gross hack to plug the hole.  A better solution would involve
2175  * a special vop or other form of generalized access control mechanism.  We
2176  * go ahead and just reject all procfs file systems accesses as dangerous.
2177  *
2178  * Since setugidsafety calls this only for fd 0, 1 and 2, this check is
2179  * sufficient.  We also don't for check setugidness since we know we are.
2180  */
2181 static int
2182 is_unsafe(struct file *fp)
2183 {
2184         if (fp->f_type == DTYPE_VNODE &&
2185             ((struct vnode *)(fp->f_data))->v_tag == VT_PROCFS)
2186                 return (1);
2187         return (0);
2188 }
2189
2190 /*
2191  * Make this setguid thing safe, if at all possible.
2192  *
2193  * NOT MPSAFE - scans fdp without spinlocks, calls knote_fdclose()
2194  */
2195 void
2196 setugidsafety(struct proc *p)
2197 {
2198         struct filedesc *fdp = p->p_fd;
2199         int i;
2200
2201         /* Certain daemons might not have file descriptors. */
2202         if (fdp == NULL)
2203                 return;
2204
2205         /*
2206          * note: fdp->fd_files may be reallocated out from under us while
2207          * we are blocked in a close.  Be careful!
2208          */
2209         for (i = 0; i <= fdp->fd_lastfile; i++) {
2210                 if (i > 2)
2211                         break;
2212                 if (fdp->fd_files[i].fp && is_unsafe(fdp->fd_files[i].fp)) {
2213                         struct file *fp;
2214
2215                         /*
2216                          * NULL-out descriptor prior to close to avoid
2217                          * a race while close blocks.
2218                          */
2219                         if ((fp = funsetfd_locked(fdp, i)) != NULL) {
2220                                 knote_fdclose(fp, fdp, i);
2221                                 closef(fp, p);
2222                         }
2223                 }
2224         }
2225 }
2226
2227 /*
2228  * Close any files on exec?
2229  *
2230  * NOT MPSAFE - scans fdp without spinlocks, calls knote_fdclose()
2231  */
2232 void
2233 fdcloseexec(struct proc *p)
2234 {
2235         struct filedesc *fdp = p->p_fd;
2236         int i;
2237
2238         /* Certain daemons might not have file descriptors. */
2239         if (fdp == NULL)
2240                 return;
2241
2242         /*
2243          * We cannot cache fd_files since operations may block and rip
2244          * them out from under us.
2245          */
2246         for (i = 0; i <= fdp->fd_lastfile; i++) {
2247                 if (fdp->fd_files[i].fp != NULL &&
2248                     (fdp->fd_files[i].fileflags & UF_EXCLOSE)) {
2249                         struct file *fp;
2250
2251                         /*
2252                          * NULL-out descriptor prior to close to avoid
2253                          * a race while close blocks.
2254                          */
2255                         if ((fp = funsetfd_locked(fdp, i)) != NULL) {
2256                                 knote_fdclose(fp, fdp, i);
2257                                 closef(fp, p);
2258                         }
2259                 }
2260         }
2261 }
2262
2263 /*
2264  * It is unsafe for set[ug]id processes to be started with file
2265  * descriptors 0..2 closed, as these descriptors are given implicit
2266  * significance in the Standard C library.  fdcheckstd() will create a
2267  * descriptor referencing /dev/null for each of stdin, stdout, and
2268  * stderr that is not already open.
2269  *
2270  * NOT MPSAFE - calls falloc, vn_open, etc
2271  */
2272 int
2273 fdcheckstd(struct lwp *lp)
2274 {
2275         struct nlookupdata nd;
2276         struct filedesc *fdp;
2277         struct file *fp;
2278         int retval;
2279         int i, error, flags, devnull;
2280
2281         fdp = lp->lwp_proc->p_fd;
2282         if (fdp == NULL)
2283                 return (0);
2284         devnull = -1;
2285         error = 0;
2286         for (i = 0; i < 3; i++) {
2287                 if (fdp->fd_files[i].fp != NULL)
2288                         continue;
2289                 if (devnull < 0) {
2290                         if ((error = falloc(lp, &fp, &devnull)) != 0)
2291                                 break;
2292
2293                         error = nlookup_init(&nd, "/dev/null", UIO_SYSSPACE,
2294                                                 NLC_FOLLOW|NLC_LOCKVP);
2295                         flags = FREAD | FWRITE;
2296                         if (error == 0)
2297                                 error = vn_open(&nd, fp, flags, 0);
2298                         if (error == 0)
2299                                 fsetfd(fdp, fp, devnull);
2300                         else
2301                                 fsetfd(fdp, NULL, devnull);
2302                         fdrop(fp);
2303                         nlookup_done(&nd);
2304                         if (error)
2305                                 break;
2306                         KKASSERT(i == devnull);
2307                 } else {
2308                         error = kern_dup(DUP_FIXED, devnull, i, &retval);
2309                         if (error != 0)
2310                                 break;
2311                 }
2312         }
2313         return (error);
2314 }
2315
2316 /*
2317  * Internal form of close.
2318  * Decrement reference count on file structure.
2319  * Note: td and/or p may be NULL when closing a file
2320  * that was being passed in a message.
2321  *
2322  * MPALMOSTSAFE - acquires mplock for VOP operations
2323  */
2324 int
2325 closef(struct file *fp, struct proc *p)
2326 {
2327         struct vnode *vp;
2328         struct flock lf;
2329         struct filedesc_to_leader *fdtol;
2330
2331         if (fp == NULL)
2332                 return (0);
2333
2334         /*
2335          * POSIX record locking dictates that any close releases ALL
2336          * locks owned by this process.  This is handled by setting
2337          * a flag in the unlock to free ONLY locks obeying POSIX
2338          * semantics, and not to free BSD-style file locks.
2339          * If the descriptor was in a message, POSIX-style locks
2340          * aren't passed with the descriptor.
2341          */
2342         if (p != NULL && fp->f_type == DTYPE_VNODE &&
2343             (((struct vnode *)fp->f_data)->v_flag & VMAYHAVELOCKS)
2344         ) {
2345                 if ((p->p_leader->p_flag & P_ADVLOCK) != 0) {
2346                         lf.l_whence = SEEK_SET;
2347                         lf.l_start = 0;
2348                         lf.l_len = 0;
2349                         lf.l_type = F_UNLCK;
2350                         vp = (struct vnode *)fp->f_data;
2351                         (void) VOP_ADVLOCK(vp, (caddr_t)p->p_leader, F_UNLCK,
2352                                            &lf, F_POSIX);
2353                 }
2354                 fdtol = p->p_fdtol;
2355                 if (fdtol != NULL) {
2356                         lwkt_gettoken(&p->p_token);
2357                         /*
2358                          * Handle special case where file descriptor table
2359                          * is shared between multiple process leaders.
2360                          */
2361                         for (fdtol = fdtol->fdl_next;
2362                              fdtol != p->p_fdtol;
2363                              fdtol = fdtol->fdl_next) {
2364                                 if ((fdtol->fdl_leader->p_flag &
2365                                      P_ADVLOCK) == 0)
2366                                         continue;
2367                                 fdtol->fdl_holdcount++;
2368                                 lf.l_whence = SEEK_SET;
2369                                 lf.l_start = 0;
2370                                 lf.l_len = 0;
2371                                 lf.l_type = F_UNLCK;
2372                                 vp = (struct vnode *)fp->f_data;
2373                                 (void) VOP_ADVLOCK(vp,
2374                                                    (caddr_t)fdtol->fdl_leader,
2375                                                    F_UNLCK, &lf, F_POSIX);
2376                                 fdtol->fdl_holdcount--;
2377                                 if (fdtol->fdl_holdcount == 0 &&
2378                                     fdtol->fdl_wakeup != 0) {
2379                                         fdtol->fdl_wakeup = 0;
2380                                         wakeup(fdtol);
2381                                 }
2382                         }
2383                         lwkt_reltoken(&p->p_token);
2384                 }
2385         }
2386         return (fdrop(fp));
2387 }
2388
2389 /*
2390  * MPSAFE
2391  *
2392  * fhold() can only be called if f_count is already at least 1 (i.e. the
2393  * caller of fhold() already has a reference to the file pointer in some
2394  * manner or other).
2395  *
2396  * f_count is not spin-locked.  Instead, atomic ops are used for
2397  * incrementing, decrementing, and handling the 1->0 transition.
2398  */
2399 void
2400 fhold(struct file *fp)
2401 {
2402         atomic_add_int(&fp->f_count, 1);
2403 }
2404
2405 /*
2406  * fdrop() - drop a reference to a descriptor
2407  *
2408  * MPALMOSTSAFE - acquires mplock for final close sequence
2409  */
2410 int
2411 fdrop(struct file *fp)
2412 {
2413         struct flock lf;
2414         struct vnode *vp;
2415         int error;
2416
2417         /*
2418          * A combined fetch and subtract is needed to properly detect
2419          * 1->0 transitions, otherwise two cpus dropping from a ref
2420          * count of 2 might both try to run the 1->0 code.
2421          */
2422         if (atomic_fetchadd_int(&fp->f_count, -1) > 1)
2423                 return (0);
2424
2425         KKASSERT(SLIST_FIRST(&fp->f_klist) == NULL);
2426
2427         /*
2428          * The last reference has gone away, we own the fp structure free
2429          * and clear.
2430          */
2431         if (fp->f_count < 0)
2432                 panic("fdrop: count < 0");
2433         if ((fp->f_flag & FHASLOCK) && fp->f_type == DTYPE_VNODE &&
2434             (((struct vnode *)fp->f_data)->v_flag & VMAYHAVELOCKS)
2435         ) {
2436                 lf.l_whence = SEEK_SET;
2437                 lf.l_start = 0;
2438                 lf.l_len = 0;
2439                 lf.l_type = F_UNLCK;
2440                 vp = (struct vnode *)fp->f_data;
2441                 (void) VOP_ADVLOCK(vp, (caddr_t)fp, F_UNLCK, &lf, 0);
2442         }
2443         if (fp->f_ops != &badfileops)
2444                 error = fo_close(fp);
2445         else
2446                 error = 0;
2447         ffree(fp);
2448         return (error);
2449 }
2450
2451 /*
2452  * Apply an advisory lock on a file descriptor.
2453  *
2454  * Just attempt to get a record lock of the requested type on
2455  * the entire file (l_whence = SEEK_SET, l_start = 0, l_len = 0).
2456  *
2457  * MPALMOSTSAFE
2458  */
2459 int
2460 sys_flock(struct flock_args *uap)
2461 {
2462         struct proc *p = curproc;
2463         struct file *fp;
2464         struct vnode *vp;
2465         struct flock lf;
2466         int error;
2467
2468         if ((fp = holdfp(p->p_fd, uap->fd, -1)) == NULL)
2469                 return (EBADF);
2470         if (fp->f_type != DTYPE_VNODE) {
2471                 error = EOPNOTSUPP;
2472                 goto done;
2473         }
2474         vp = (struct vnode *)fp->f_data;
2475         lf.l_whence = SEEK_SET;
2476         lf.l_start = 0;
2477         lf.l_len = 0;
2478         if (uap->how & LOCK_UN) {
2479                 lf.l_type = F_UNLCK;
2480                 fp->f_flag &= ~FHASLOCK;
2481                 error = VOP_ADVLOCK(vp, (caddr_t)fp, F_UNLCK, &lf, 0);
2482                 goto done;
2483         }
2484         if (uap->how & LOCK_EX)
2485                 lf.l_type = F_WRLCK;
2486         else if (uap->how & LOCK_SH)
2487                 lf.l_type = F_RDLCK;
2488         else {
2489                 error = EBADF;
2490                 goto done;
2491         }
2492         fp->f_flag |= FHASLOCK;
2493         if (uap->how & LOCK_NB)
2494                 error = VOP_ADVLOCK(vp, (caddr_t)fp, F_SETLK, &lf, 0);
2495         else
2496                 error = VOP_ADVLOCK(vp, (caddr_t)fp, F_SETLK, &lf, F_WAIT);
2497 done:
2498         fdrop(fp);
2499         return (error);
2500 }
2501
2502 /*
2503  * File Descriptor pseudo-device driver (/dev/fd/).
2504  *
2505  * Opening minor device N dup()s the file (if any) connected to file
2506  * descriptor N belonging to the calling process.  Note that this driver
2507  * consists of only the ``open()'' routine, because all subsequent
2508  * references to this file will be direct to the other driver.
2509  */
2510 static int
2511 fdopen(struct dev_open_args *ap)
2512 {
2513         thread_t td = curthread;
2514
2515         KKASSERT(td->td_lwp != NULL);
2516
2517         /*
2518          * XXX Kludge: set curlwp->lwp_dupfd to contain the value of the
2519          * the file descriptor being sought for duplication. The error
2520          * return ensures that the vnode for this device will be released
2521          * by vn_open. Open will detect this special error and take the
2522          * actions in dupfdopen below. Other callers of vn_open or VOP_OPEN
2523          * will simply report the error.
2524          */
2525         td->td_lwp->lwp_dupfd = minor(ap->a_head.a_dev);
2526         return (ENODEV);
2527 }
2528
2529 /*
2530  * The caller has reserved the file descriptor dfd for us.  On success we
2531  * must fsetfd() it.  On failure the caller will clean it up.
2532  *
2533  * MPSAFE
2534  */
2535 int
2536 dupfdopen(struct filedesc *fdp, int dfd, int sfd, int mode, int error)
2537 {
2538         struct file *wfp;
2539         struct file *xfp;
2540         int werror;
2541
2542         if ((wfp = holdfp(fdp, sfd, -1)) == NULL)
2543                 return (EBADF);
2544
2545         /*
2546          * Close a revoke/dup race.  Duping a descriptor marked as revoked
2547          * will dup a dummy descriptor instead of the real one.
2548          */
2549         if (wfp->f_flag & FREVOKED) {
2550                 kprintf("Warning: attempt to dup() a revoked descriptor\n");
2551                 fdrop(wfp);
2552                 wfp = NULL;
2553                 werror = falloc(NULL, &wfp, NULL);
2554                 if (werror)
2555                         return (werror);
2556         }
2557
2558         /*
2559          * There are two cases of interest here.
2560          *
2561          * For ENODEV simply dup sfd to file descriptor dfd and return.
2562          *
2563          * For ENXIO steal away the file structure from sfd and store it
2564          * dfd.  sfd is effectively closed by this operation.
2565          *
2566          * Any other error code is just returned.
2567          */
2568         switch (error) {
2569         case ENODEV:
2570                 /*
2571                  * Check that the mode the file is being opened for is a
2572                  * subset of the mode of the existing descriptor.
2573                  */
2574                 if (((mode & (FREAD|FWRITE)) | wfp->f_flag) != wfp->f_flag) {
2575                         error = EACCES;
2576                         break;
2577                 }
2578                 spin_lock(&fdp->fd_spin);
2579                 fdp->fd_files[dfd].fileflags = fdp->fd_files[sfd].fileflags;
2580                 fsetfd_locked(fdp, wfp, dfd);
2581                 spin_unlock(&fdp->fd_spin);
2582                 error = 0;
2583                 break;
2584         case ENXIO:
2585                 /*
2586                  * Steal away the file pointer from dfd, and stuff it into indx.
2587                  */
2588                 spin_lock(&fdp->fd_spin);
2589                 fdp->fd_files[dfd].fileflags = fdp->fd_files[sfd].fileflags;
2590                 fsetfd(fdp, wfp, dfd);
2591                 if ((xfp = funsetfd_locked(fdp, sfd)) != NULL) {
2592                         spin_unlock(&fdp->fd_spin);
2593                         fdrop(xfp);
2594                 } else {
2595                         spin_unlock(&fdp->fd_spin);
2596                 }
2597                 error = 0;
2598                 break;
2599         default:
2600                 break;
2601         }
2602         fdrop(wfp);
2603         return (error);
2604 }
2605
2606 /*
2607  * NOT MPSAFE - I think these refer to a common file descriptor table
2608  * and we need to spinlock that to link fdtol in.
2609  */
2610 struct filedesc_to_leader *
2611 filedesc_to_leader_alloc(struct filedesc_to_leader *old,
2612                          struct proc *leader)
2613 {
2614         struct filedesc_to_leader *fdtol;
2615
2616         fdtol = kmalloc(sizeof(struct filedesc_to_leader),
2617                         M_FILEDESC_TO_LEADER, M_WAITOK | M_ZERO);
2618         fdtol->fdl_refcount = 1;
2619         fdtol->fdl_holdcount = 0;
2620         fdtol->fdl_wakeup = 0;
2621         fdtol->fdl_leader = leader;
2622         if (old != NULL) {
2623                 fdtol->fdl_next = old->fdl_next;
2624                 fdtol->fdl_prev = old;
2625                 old->fdl_next = fdtol;
2626                 fdtol->fdl_next->fdl_prev = fdtol;
2627         } else {
2628                 fdtol->fdl_next = fdtol;
2629                 fdtol->fdl_prev = fdtol;
2630         }
2631         return fdtol;
2632 }
2633
2634 /*
2635  * Scan all file pointers in the system.  The callback is made with
2636  * the master list spinlock held exclusively.
2637  *
2638  * MPSAFE
2639  */
2640 void
2641 allfiles_scan_exclusive(int (*callback)(struct file *, void *), void *data)
2642 {
2643         struct file *fp;
2644         int res;
2645
2646         spin_lock(&filehead_spin);
2647         LIST_FOREACH(fp, &filehead, f_list) {
2648                 res = callback(fp, data);
2649                 if (res < 0)
2650                         break;
2651         }
2652         spin_unlock(&filehead_spin);
2653 }
2654
2655 /*
2656  * Get file structures.
2657  *
2658  * NOT MPSAFE - process list scan, SYSCTL_OUT (probably not mpsafe)
2659  */
2660
2661 struct sysctl_kern_file_info {
2662         int count;
2663         int error;
2664         struct sysctl_req *req;
2665 };
2666
2667 static int sysctl_kern_file_callback(struct proc *p, void *data);
2668
2669 static int
2670 sysctl_kern_file(SYSCTL_HANDLER_ARGS)
2671 {
2672         struct sysctl_kern_file_info info;
2673
2674         /*
2675          * Note: because the number of file descriptors is calculated
2676          * in different ways for sizing vs returning the data,
2677          * there is information leakage from the first loop.  However,
2678          * it is of a similar order of magnitude to the leakage from
2679          * global system statistics such as kern.openfiles.
2680          *
2681          * When just doing a count, note that we cannot just count
2682          * the elements and add f_count via the filehead list because
2683          * threaded processes share their descriptor table and f_count might
2684          * still be '1' in that case.
2685          *
2686          * Since the SYSCTL op can block, we must hold the process to
2687          * prevent it being ripped out from under us either in the
2688          * file descriptor loop or in the greater LIST_FOREACH.  The
2689          * process may be in varying states of disrepair.  If the process
2690          * is in SZOMB we may have caught it just as it is being removed
2691          * from the allproc list, we must skip it in that case to maintain
2692          * an unbroken chain through the allproc list.
2693          */
2694         info.count = 0;
2695         info.error = 0;
2696         info.req = req;
2697         allproc_scan(sysctl_kern_file_callback, &info);
2698
2699         /*
2700          * When just calculating the size, overestimate a bit to try to
2701          * prevent system activity from causing the buffer-fill call
2702          * to fail later on.
2703          */
2704         if (req->oldptr == NULL) {
2705                 info.count = (info.count + 16) + (info.count / 10);
2706                 info.error = SYSCTL_OUT(req, NULL,
2707                                         info.count * sizeof(struct kinfo_file));
2708         }
2709         return (info.error);
2710 }
2711
2712 static int
2713 sysctl_kern_file_callback(struct proc *p, void *data)
2714 {
2715         struct sysctl_kern_file_info *info = data;
2716         struct kinfo_file kf;
2717         struct filedesc *fdp;
2718         struct file *fp;
2719         uid_t uid;
2720         int n;
2721
2722         if (p->p_stat == SIDL || p->p_stat == SZOMB)
2723                 return(0);
2724         if (!PRISON_CHECK(info->req->td->td_ucred, p->p_ucred) != 0)
2725                 return(0);
2726
2727         /*
2728          * Softref the fdp to prevent it from being destroyed
2729          */
2730         spin_lock(&p->p_spin);
2731         if ((fdp = p->p_fd) == NULL) {
2732                 spin_unlock(&p->p_spin);
2733                 return(0);
2734         }
2735         atomic_add_int(&fdp->fd_softrefs, 1);
2736         spin_unlock(&p->p_spin);
2737
2738         /*
2739          * The fdp's own spinlock prevents the contents from being
2740          * modified.
2741          */
2742         spin_lock(&fdp->fd_spin);
2743         for (n = 0; n < fdp->fd_nfiles; ++n) {
2744                 if ((fp = fdp->fd_files[n].fp) == NULL)
2745                         continue;
2746                 if (info->req->oldptr == NULL) {
2747                         ++info->count;
2748                 } else {
2749                         uid = p->p_ucred ? p->p_ucred->cr_uid : -1;
2750                         kcore_make_file(&kf, fp, p->p_pid, uid, n);
2751                         spin_unlock(&fdp->fd_spin);
2752                         info->error = SYSCTL_OUT(info->req, &kf, sizeof(kf));
2753                         spin_lock(&fdp->fd_spin);
2754                         if (info->error)
2755                                 break;
2756                 }
2757         }
2758         spin_unlock(&fdp->fd_spin);
2759         atomic_subtract_int(&fdp->fd_softrefs, 1);
2760         if (info->error)
2761                 return(-1);
2762         return(0);
2763 }
2764
2765 SYSCTL_PROC(_kern, KERN_FILE, file, CTLTYPE_OPAQUE|CTLFLAG_RD,
2766     0, 0, sysctl_kern_file, "S,file", "Entire file table");
2767
2768 SYSCTL_INT(_kern, OID_AUTO, minfilesperproc, CTLFLAG_RW,
2769     &minfilesperproc, 0, "Minimum files allowed open per process");
2770 SYSCTL_INT(_kern, KERN_MAXFILESPERPROC, maxfilesperproc, CTLFLAG_RW,
2771     &maxfilesperproc, 0, "Maximum files allowed open per process");
2772 SYSCTL_INT(_kern, OID_AUTO, maxfilesperuser, CTLFLAG_RW,
2773     &maxfilesperuser, 0, "Maximum files allowed open per user");
2774
2775 SYSCTL_INT(_kern, KERN_MAXFILES, maxfiles, CTLFLAG_RW,
2776     &maxfiles, 0, "Maximum number of files");
2777
2778 SYSCTL_INT(_kern, OID_AUTO, maxfilesrootres, CTLFLAG_RW,
2779     &maxfilesrootres, 0, "Descriptors reserved for root use");
2780
2781 SYSCTL_INT(_kern, OID_AUTO, openfiles, CTLFLAG_RD,
2782         &nfiles, 0, "System-wide number of open files");
2783
2784 static void
2785 fildesc_drvinit(void *unused)
2786 {
2787         int fd;
2788
2789         for (fd = 0; fd < NUMFDESC; fd++) {
2790                 make_dev(&fildesc_ops, fd,
2791                          UID_BIN, GID_BIN, 0666, "fd/%d", fd);
2792         }
2793
2794         make_dev(&fildesc_ops, 0, UID_ROOT, GID_WHEEL, 0666, "stdin");
2795         make_dev(&fildesc_ops, 1, UID_ROOT, GID_WHEEL, 0666, "stdout");
2796         make_dev(&fildesc_ops, 2, UID_ROOT, GID_WHEEL, 0666, "stderr");
2797 }
2798
2799 /*
2800  * MPSAFE
2801  */
2802 struct fileops badfileops = {
2803         .fo_read = badfo_readwrite,
2804         .fo_write = badfo_readwrite,
2805         .fo_ioctl = badfo_ioctl,
2806         .fo_kqfilter = badfo_kqfilter,
2807         .fo_stat = badfo_stat,
2808         .fo_close = badfo_close,
2809         .fo_shutdown = badfo_shutdown
2810 };
2811
2812 int
2813 badfo_readwrite(
2814         struct file *fp,
2815         struct uio *uio,
2816         struct ucred *cred,
2817         int flags
2818 ) {
2819         return (EBADF);
2820 }
2821
2822 int
2823 badfo_ioctl(struct file *fp, u_long com, caddr_t data,
2824             struct ucred *cred, struct sysmsg *msgv)
2825 {
2826         return (EBADF);
2827 }
2828
2829 /*
2830  * Must return an error to prevent registration, typically
2831  * due to a revoked descriptor (file_filtops assigned).
2832  */
2833 int
2834 badfo_kqfilter(struct file *fp, struct knote *kn)
2835 {
2836         return (EOPNOTSUPP);
2837 }
2838
2839 /*
2840  * MPSAFE
2841  */
2842 int
2843 badfo_stat(struct file *fp, struct stat *sb, struct ucred *cred)
2844 {
2845         return (EBADF);
2846 }
2847
2848 /*
2849  * MPSAFE
2850  */
2851 int
2852 badfo_close(struct file *fp)
2853 {
2854         return (EBADF);
2855 }
2856
2857 /*
2858  * MPSAFE
2859  */
2860 int
2861 badfo_shutdown(struct file *fp, int how)
2862 {
2863         return (EBADF);
2864 }
2865
2866 /*
2867  * MPSAFE
2868  */
2869 int
2870 nofo_shutdown(struct file *fp, int how)
2871 {
2872         return (EOPNOTSUPP);
2873 }
2874
2875 SYSINIT(fildescdev,SI_SUB_DRIVERS,SI_ORDER_MIDDLE+CDEV_MAJOR,
2876                                         fildesc_drvinit,NULL)