sys/kern/kern_descrip.c

   1 /*
   2  * Copyright (c) 1982, 1986, 1989, 1991, 1993
   3  *      The Regents of the University of California.  All rights reserved.
   4  * (c) UNIX System Laboratories, Inc.
   5  * All or some portions of this file are derived from material licensed
   6  * to the University of California by American Telephone and Telegraph
   7  * Co. or Unix System Laboratories, Inc. and are reproduced herein with
   8  * the permission of UNIX System Laboratories, Inc.
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  * 3. All advertising materials mentioning features or use of this software
  19  *    must display the following acknowledgement:
  20  *      This product includes software developed by the University of
  21  *      California, Berkeley and its contributors.
  22  * 4. Neither the name of the University nor the names of its contributors
  23  *    may be used to endorse or promote products derived from this software
  24  *    without specific prior written permission.
  25  *
  26  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  27  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  28  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  29  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  30  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  31  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  32  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  33  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  34  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  35  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  36  * SUCH DAMAGE.
  37  *
  38  *      @(#)kern_descrip.c      8.6 (Berkeley) 4/19/94
  39  * $FreeBSD: src/sys/kern/kern_descrip.c,v 1.81.2.19 2004/02/28 00:43:31 tegge Exp $
  40  * $DragonFly: src/sys/kern/kern_descrip.c,v 1.32 2004/11/18 13:09:30 dillon Exp $
  41  */
  42
  43 #include "opt_compat.h"
  44 #include <sys/param.h>
  45 #include <sys/systm.h>
  46 #include <sys/malloc.h>
  47 #include <sys/sysproto.h>
  48 #include <sys/conf.h>
  49 #include <sys/filedesc.h>
  50 #include <sys/kernel.h>
  51 #include <sys/sysctl.h>
  52 #include <sys/vnode.h>
  53 #include <sys/proc.h>
  54 #include <sys/nlookup.h>
  55 #include <sys/file.h>
  56 #include <sys/stat.h>
  57 #include <sys/filio.h>
  58 #include <sys/fcntl.h>
  59 #include <sys/unistd.h>
  60 #include <sys/resourcevar.h>
  61 #include <sys/event.h>
  62 #include <sys/kern_syscall.h>
  63
  64 #include <vm/vm.h>
  65 #include <vm/vm_extern.h>
  66
  67 #include <sys/file2.h>
  68
  69 static MALLOC_DEFINE(M_FILEDESC, "file desc", "Open file descriptor table");
  70 static MALLOC_DEFINE(M_FILEDESC_TO_LEADER, "file desc to leader",
  71                      "file desc to leader structures");
  72 MALLOC_DEFINE(M_FILE, "file", "Open file structure");
  73 static MALLOC_DEFINE(M_SIGIO, "sigio", "sigio structures");
  74
  75 static   d_open_t  fdopen;
  76 #define NUMFDESC 64
  77
  78 #define CDEV_MAJOR 22
  79 static struct cdevsw fildesc_cdevsw = {
  80         /* name */      "FD",
  81         /* maj */       CDEV_MAJOR,
  82         /* flags */     0,
  83         /* port */      NULL,
  84         /* clone */     NULL,
  85
  86         /* open */      fdopen,
  87         /* close */     noclose,
  88         /* read */      noread,
  89         /* write */     nowrite,
  90         /* ioctl */     noioctl,
  91         /* poll */      nopoll,
  92         /* mmap */      nommap,
  93         /* strategy */  nostrategy,
  94         /* dump */      nodump,
  95         /* psize */     nopsize
  96 };
  97
  98 static int badfo_readwrite (struct file *fp, struct uio *uio,
  99     struct ucred *cred, int flags, struct thread *td);
 100 static int badfo_ioctl (struct file *fp, u_long com, caddr_t data,
 101     struct thread *td);
 102 static int badfo_poll (struct file *fp, int events,
 103     struct ucred *cred, struct thread *td);
 104 static int badfo_kqfilter (struct file *fp, struct knote *kn);
 105 static int badfo_stat (struct file *fp, struct stat *sb, struct thread *td);
 106 static int badfo_close (struct file *fp, struct thread *td);
 107
 108 /*
 109  * Descriptor management.
 110  */
 111 struct filelist filehead;       /* head of list of open files */
 112 int nfiles;                     /* actual number of open files */
 113 extern int cmask;
 114
 115 /*
 116  * System calls on descriptors.
 117  */
 118 /* ARGSUSED */
 119 int
 120 getdtablesize(struct getdtablesize_args *uap)
 121 {
 122         struct proc *p = curproc;
 123
 124         uap->sysmsg_result =
 125             min((int)p->p_rlimit[RLIMIT_NOFILE].rlim_cur, maxfilesperproc);
 126         return (0);
 127 }
 128
 129 /*
 130  * Duplicate a file descriptor to a particular value.
 131  *
 132  * note: keep in mind that a potential race condition exists when closing
 133  * descriptors from a shared descriptor table (via rfork).
 134  */
 135 /* ARGSUSED */
 136 int
 137 dup2(struct dup2_args *uap)
 138 {
 139         int error;
 140
 141         error = kern_dup(DUP_FIXED, uap->from, uap->to, uap->sysmsg_fds);
 142
 143         return (error);
 144 }
 145
 146 /*
 147  * Duplicate a file descriptor.
 148  */
 149 /* ARGSUSED */
 150 int
 151 dup(struct dup_args *uap)
 152 {
 153         int error;
 154
 155         error = kern_dup(DUP_VARIABLE, uap->fd, 0, uap->sysmsg_fds);
 156
 157         return (error);
 158 }
 159
 160 int
 161 kern_fcntl(int fd, int cmd, union fcntl_dat *dat)
 162 {
 163         struct thread *td = curthread;
 164         struct proc *p = td->td_proc;
 165         struct filedesc *fdp = p->p_fd;
 166         struct file *fp;
 167         char *pop;
 168         struct vnode *vp;
 169         u_int newmin;
 170         int tmp, error, flg = F_POSIX;
 171
 172         KKASSERT(p);
 173
 174         if ((unsigned)fd >= fdp->fd_nfiles ||
 175             (fp = fdp->fd_ofiles[fd]) == NULL)
 176                 return (EBADF);
 177         pop = &fdp->fd_ofileflags[fd];
 178
 179         switch (cmd) {
 180         case F_DUPFD:
 181                 newmin = dat->fc_fd;
 182                 if (newmin >= p->p_rlimit[RLIMIT_NOFILE].rlim_cur ||
 183                     newmin > maxfilesperproc)
 184                         return (EINVAL);
 185                 error = kern_dup(DUP_VARIABLE, fd, newmin, &dat->fc_fd);
 186                 return (error);
 187
 188         case F_GETFD:
 189                 dat->fc_cloexec = (*pop & UF_EXCLOSE) ? FD_CLOEXEC : 0;
 190                 return (0);
 191
 192         case F_SETFD:
 193                 *pop = (*pop &~ UF_EXCLOSE) |
 194                     (dat->fc_cloexec & FD_CLOEXEC ? UF_EXCLOSE : 0);
 195                 return (0);
 196
 197         case F_GETFL:
 198                 dat->fc_flags = OFLAGS(fp->f_flag);
 199                 return (0);
 200
 201         case F_SETFL:
 202                 fhold(fp);
 203                 fp->f_flag &= ~FCNTLFLAGS;
 204                 fp->f_flag |= FFLAGS(dat->fc_flags & ~O_ACCMODE) & FCNTLFLAGS;
 205                 tmp = fp->f_flag & FNONBLOCK;
 206                 error = fo_ioctl(fp, FIONBIO, (caddr_t)&tmp, td);
 207                 if (error) {
 208                         fdrop(fp, td);
 209                         return (error);
 210                 }
 211                 tmp = fp->f_flag & FASYNC;
 212                 error = fo_ioctl(fp, FIOASYNC, (caddr_t)&tmp, td);
 213                 if (!error) {
 214                         fdrop(fp, td);
 215                         return (0);
 216                 }
 217                 fp->f_flag &= ~FNONBLOCK;
 218                 tmp = 0;
 219                 fo_ioctl(fp, FIONBIO, (caddr_t)&tmp, td);
 220                 fdrop(fp, td);
 221                 return (error);
 222
 223         case F_GETOWN:
 224                 fhold(fp);
 225                 error = fo_ioctl(fp, FIOGETOWN, (caddr_t)&dat->fc_owner, td);
 226                 fdrop(fp, td);
 227                 return(error);
 228
 229         case F_SETOWN:
 230                 fhold(fp);
 231                 error = fo_ioctl(fp, FIOSETOWN, (caddr_t)&dat->fc_owner, td);
 232                 fdrop(fp, td);
 233                 return(error);
 234
 235         case F_SETLKW:
 236                 flg |= F_WAIT;
 237                 /* Fall into F_SETLK */
 238
 239         case F_SETLK:
 240                 if (fp->f_type != DTYPE_VNODE)
 241                         return (EBADF);
 242                 vp = (struct vnode *)fp->f_data;
 243
 244                 /*
 245                  * copyin/lockop may block
 246                  */
 247                 fhold(fp);
 248                 if (dat->fc_flock.l_whence == SEEK_CUR)
 249                         dat->fc_flock.l_start += fp->f_offset;
 250
 251                 switch (dat->fc_flock.l_type) {
 252                 case F_RDLCK:
 253                         if ((fp->f_flag & FREAD) == 0) {
 254                                 error = EBADF;
 255                                 break;
 256                         }
 257                         p->p_leader->p_flag |= P_ADVLOCK;
 258                         error = VOP_ADVLOCK(vp, (caddr_t)p->p_leader, F_SETLK,
 259                             &dat->fc_flock, flg);
 260                         break;
 261                 case F_WRLCK:
 262                         if ((fp->f_flag & FWRITE) == 0) {
 263                                 error = EBADF;
 264                                 break;
 265                         }
 266                         p->p_leader->p_flag |= P_ADVLOCK;
 267                         error = VOP_ADVLOCK(vp, (caddr_t)p->p_leader, F_SETLK,
 268                             &dat->fc_flock, flg);
 269                         break;
 270                 case F_UNLCK:
 271                         error = VOP_ADVLOCK(vp, (caddr_t)p->p_leader, F_UNLCK,
 272                                 &dat->fc_flock, F_POSIX);
 273                         break;
 274                 default:
 275                         error = EINVAL;
 276                         break;
 277                 }
 278                 /* Check for race with close */
 279                 if ((unsigned) fd >= fdp->fd_nfiles ||
 280                     fp != fdp->fd_ofiles[fd]) {
 281                         dat->fc_flock.l_whence = SEEK_SET;
 282                         dat->fc_flock.l_start = 0;
 283                         dat->fc_flock.l_len = 0;
 284                         dat->fc_flock.l_type = F_UNLCK;
 285                         (void) VOP_ADVLOCK(vp, (caddr_t)p->p_leader,
 286                                            F_UNLCK, &dat->fc_flock, F_POSIX);
 287                 }
 288                 fdrop(fp, td);
 289                 return(error);
 290
 291         case F_GETLK:
 292                 if (fp->f_type != DTYPE_VNODE)
 293                         return (EBADF);
 294                 vp = (struct vnode *)fp->f_data;
 295                 /*
 296                  * copyin/lockop may block
 297                  */
 298                 fhold(fp);
 299                 if (dat->fc_flock.l_type != F_RDLCK &&
 300                     dat->fc_flock.l_type != F_WRLCK &&
 301                     dat->fc_flock.l_type != F_UNLCK) {
 302                         fdrop(fp, td);
 303                         return (EINVAL);
 304                 }
 305                 if (dat->fc_flock.l_whence == SEEK_CUR)
 306                         dat->fc_flock.l_start += fp->f_offset;
 307                 error = VOP_ADVLOCK(vp, (caddr_t)p->p_leader, F_GETLK,
 308                             &dat->fc_flock, F_POSIX);
 309                 fdrop(fp, td);
 310                 return(error);
 311         default:
 312                 return (EINVAL);
 313         }
 314         /* NOTREACHED */
 315 }
 316
 317 /*
 318  * The file control system call.
 319  */
 320 int
 321 fcntl(struct fcntl_args *uap)
 322 {
 323         union fcntl_dat dat;
 324         int error;
 325
 326         switch (uap->cmd) {
 327         case F_DUPFD:
 328                 dat.fc_fd = uap->arg;
 329                 break;
 330         case F_SETFD:
 331                 dat.fc_cloexec = uap->arg;
 332                 break;
 333         case F_SETFL:
 334                 dat.fc_flags = uap->arg;
 335                 break;
 336         case F_SETOWN:
 337                 dat.fc_owner = uap->arg;
 338                 break;
 339         case F_SETLKW:
 340         case F_SETLK:
 341         case F_GETLK:
 342                 error = copyin((caddr_t)uap->arg, &dat.fc_flock,
 343                     sizeof(struct flock));
 344                 if (error)
 345                         return (error);
 346                 break;
 347         }
 348
 349         error = kern_fcntl(uap->fd, uap->cmd, &dat);
 350
 351         if (error == 0) {
 352                 switch (uap->cmd) {
 353                 case F_DUPFD:
 354                         uap->sysmsg_result = dat.fc_fd;
 355                         break;
 356                 case F_GETFD:
 357                         uap->sysmsg_result = dat.fc_cloexec;
 358                         break;
 359                 case F_GETFL:
 360                         uap->sysmsg_result = dat.fc_flags;
 361                         break;
 362                 case F_GETOWN:
 363                         uap->sysmsg_result = dat.fc_owner;
 364                 case F_GETLK:
 365                         error = copyout(&dat.fc_flock, (caddr_t)uap->arg,
 366                             sizeof(struct flock));
 367                         break;
 368                 }
 369         }
 370
 371         return (error);
 372 }
 373
 374 /*
 375  * Common code for dup, dup2, and fcntl(F_DUPFD).
 376  *
 377  * The type flag can be either DUP_FIXED or DUP_VARIABLE.  DUP_FIXED tells
 378  * kern_dup() to destructively dup over an existing file descriptor if new
 379  * is already open.  DUP_VARIABLE tells kern_dup() to find the lowest
 380  * unused file descriptor that is greater than or equal to new.
 381  */
 382 int
 383 kern_dup(enum dup_type type, int old, int new, int *res)
 384 {
 385         struct thread *td = curthread;
 386         struct proc *p = td->td_proc;
 387         struct filedesc *fdp = p->p_fd;
 388         struct file *fp;
 389         struct file *delfp;
 390         int holdleaders;
 391         int error, newfd;
 392
 393         /*
 394          * Verify that we have a valid descriptor to dup from and
 395          * possibly to dup to.
 396          */
 397         if (old < 0 || new < 0 || new > p->p_rlimit[RLIMIT_NOFILE].rlim_cur ||
 398             new >= maxfilesperproc)
 399                 return (EBADF);
 400         if (old >= fdp->fd_nfiles || fdp->fd_ofiles[old] == NULL)
 401                 return (EBADF);
 402         if (type == DUP_FIXED && old == new) {
 403                 *res = new;
 404                 return (0);
 405         }
 406         fp = fdp->fd_ofiles[old];
 407         fhold(fp);
 408
 409         /*
 410          * Expand the table for the new descriptor if needed.  This may
 411          * block and drop and reacquire the fidedesc lock.
 412          */
 413         if (type == DUP_VARIABLE || new >= fdp->fd_nfiles) {
 414                 error = fdalloc(p, new, &newfd);
 415                 if (error) {
 416                         fdrop(fp, td);
 417                         return (error);
 418                 }
 419         }
 420         if (type == DUP_VARIABLE)
 421                 new = newfd;
 422
 423         /*
 424          * If the old file changed out from under us then treat it as a
 425          * bad file descriptor.  Userland should do its own locking to
 426          * avoid this case.
 427          */
 428         if (fdp->fd_ofiles[old] != fp) {
 429                 if (fdp->fd_ofiles[new] == NULL) {
 430                         if (new < fdp->fd_freefile)
 431                                 fdp->fd_freefile = new;
 432                         while (fdp->fd_lastfile > 0 &&
 433                             fdp->fd_ofiles[fdp->fd_lastfile] == NULL)
 434                                 fdp->fd_lastfile--;
 435                 }
 436                 fdrop(fp, td);
 437                 return (EBADF);
 438         }
 439         KASSERT(old != new, ("new fd is same as old"));
 440
 441         /*
 442          * Save info on the descriptor being overwritten.  We have
 443          * to do the unmap now, but we cannot close it without
 444          * introducing an ownership race for the slot.
 445          */
 446         delfp = fdp->fd_ofiles[new];
 447         if (delfp != NULL && p->p_fdtol != NULL) {
 448                 /*
 449                  * Ask fdfree() to sleep to ensure that all relevant
 450                  * process leaders can be traversed in closef().
 451                  */
 452                 fdp->fd_holdleaderscount++;
 453                 holdleaders = 1;
 454         } else
 455                 holdleaders = 0;
 456         KASSERT(delfp == NULL || type == DUP_FIXED,
 457             ("dup() picked an open file"));
 458 #if 0
 459         if (delfp && (fdp->fd_ofileflags[new] & UF_MAPPED))
 460                 (void) munmapfd(p, new);
 461 #endif
 462
 463         /*
 464          * Duplicate the source descriptor, update lastfile
 465          */
 466         fdp->fd_ofiles[new] = fp;
 467         fdp->fd_ofileflags[new] = fdp->fd_ofileflags[old] &~ UF_EXCLOSE;
 468         if (new > fdp->fd_lastfile)
 469                 fdp->fd_lastfile = new;
 470         *res = new;
 471
 472         /*
 473          * If we dup'd over a valid file, we now own the reference to it
 474          * and must dispose of it using closef() semantics (as if a
 475          * close() were performed on it).
 476          */
 477         if (delfp) {
 478                 (void) closef(delfp, td);
 479                 if (holdleaders) {
 480                         fdp->fd_holdleaderscount--;
 481                         if (fdp->fd_holdleaderscount == 0 &&
 482                             fdp->fd_holdleaderswakeup != 0) {
 483                                 fdp->fd_holdleaderswakeup = 0;
 484                                 wakeup(&fdp->fd_holdleaderscount);
 485                         }
 486                 }
 487         }
 488         return (0);
 489 }
 490
 491 /*
 492  * If sigio is on the list associated with a process or process group,
 493  * disable signalling from the device, remove sigio from the list and
 494  * free sigio.
 495  */
 496 void
 497 funsetown(struct sigio *sigio)
 498 {
 499         int s;
 500
 501         if (sigio == NULL)
 502                 return;
 503         s = splhigh();
 504         *(sigio->sio_myref) = NULL;
 505         splx(s);
 506         if (sigio->sio_pgid < 0) {
 507                 SLIST_REMOVE(&sigio->sio_pgrp->pg_sigiolst, sigio,
 508                              sigio, sio_pgsigio);
 509         } else /* if ((*sigiop)->sio_pgid > 0) */ {
 510                 SLIST_REMOVE(&sigio->sio_proc->p_sigiolst, sigio,
 511                              sigio, sio_pgsigio);
 512         }
 513         crfree(sigio->sio_ucred);
 514         free(sigio, M_SIGIO);
 515 }
 516
 517 /* Free a list of sigio structures. */
 518 void
 519 funsetownlst(struct sigiolst *sigiolst)
 520 {
 521         struct sigio *sigio;
 522
 523         while ((sigio = SLIST_FIRST(sigiolst)) != NULL)
 524                 funsetown(sigio);
 525 }
 526
 527 /*
 528  * This is common code for FIOSETOWN ioctl called by fcntl(fd, F_SETOWN, arg).
 529  *
 530  * After permission checking, add a sigio structure to the sigio list for
 531  * the process or process group.
 532  */
 533 int
 534 fsetown(pid_t pgid, struct sigio **sigiop)
 535 {
 536         struct proc *proc;
 537         struct pgrp *pgrp;
 538         struct sigio *sigio;
 539         int s;
 540
 541         if (pgid == 0) {
 542                 funsetown(*sigiop);
 543                 return (0);
 544         }
 545         if (pgid > 0) {
 546                 proc = pfind(pgid);
 547                 if (proc == NULL)
 548                         return (ESRCH);
 549
 550                 /*
 551                  * Policy - Don't allow a process to FSETOWN a process
 552                  * in another session.
 553                  *
 554                  * Remove this test to allow maximum flexibility or
 555                  * restrict FSETOWN to the current process or process
 556                  * group for maximum safety.
 557                  */
 558                 if (proc->p_session != curproc->p_session)
 559                         return (EPERM);
 560
 561                 pgrp = NULL;
 562         } else /* if (pgid < 0) */ {
 563                 pgrp = pgfind(-pgid);
 564                 if (pgrp == NULL)
 565                         return (ESRCH);
 566
 567                 /*
 568                  * Policy - Don't allow a process to FSETOWN a process
 569                  * in another session.
 570                  *
 571                  * Remove this test to allow maximum flexibility or
 572                  * restrict FSETOWN to the current process or process
 573                  * group for maximum safety.
 574                  */
 575                 if (pgrp->pg_session != curproc->p_session)
 576                         return (EPERM);
 577
 578                 proc = NULL;
 579         }
 580         funsetown(*sigiop);
 581         sigio = malloc(sizeof(struct sigio), M_SIGIO, M_WAITOK);
 582         if (pgid > 0) {
 583                 SLIST_INSERT_HEAD(&proc->p_sigiolst, sigio, sio_pgsigio);
 584                 sigio->sio_proc = proc;
 585         } else {
 586                 SLIST_INSERT_HEAD(&pgrp->pg_sigiolst, sigio, sio_pgsigio);
 587                 sigio->sio_pgrp = pgrp;
 588         }
 589         sigio->sio_pgid = pgid;
 590         sigio->sio_ucred = crhold(curproc->p_ucred);
 591         /* It would be convenient if p_ruid was in ucred. */
 592         sigio->sio_ruid = curproc->p_ucred->cr_ruid;
 593         sigio->sio_myref = sigiop;
 594         s = splhigh();
 595         *sigiop = sigio;
 596         splx(s);
 597         return (0);
 598 }
 599
 600 /*
 601  * This is common code for FIOGETOWN ioctl called by fcntl(fd, F_GETOWN, arg).
 602  */
 603 pid_t
 604 fgetown(struct sigio *sigio)
 605 {
 606         return (sigio != NULL ? sigio->sio_pgid : 0);
 607 }
 608
 609 /*
 610  * Close a file descriptor.
 611  */
 612 /* ARGSUSED */
 613 int
 614 close(struct close_args *uap)
 615 {
 616         return(kern_close(uap->fd));
 617 }
 618
 619 int
 620 kern_close(int fd)
 621 {
 622         struct thread *td = curthread;
 623         struct proc *p = td->td_proc;
 624         struct filedesc *fdp;
 625         struct file *fp;
 626         int error;
 627         int holdleaders;
 628
 629         KKASSERT(p);
 630         fdp = p->p_fd;
 631
 632         if ((unsigned)fd >= fdp->fd_nfiles ||
 633             (fp = fdp->fd_ofiles[fd]) == NULL)
 634                 return (EBADF);
 635 #if 0
 636         if (fdp->fd_ofileflags[fd] & UF_MAPPED)
 637                 (void) munmapfd(p, fd);
 638 #endif
 639         fdp->fd_ofiles[fd] = NULL;
 640         fdp->fd_ofileflags[fd] = 0;
 641         holdleaders = 0;
 642         if (p->p_fdtol != NULL) {
 643                 /*
 644                  * Ask fdfree() to sleep to ensure that all relevant
 645                  * process leaders can be traversed in closef().
 646                  */
 647                 fdp->fd_holdleaderscount++;
 648                 holdleaders = 1;
 649         }
 650
 651         /*
 652          * we now hold the fp reference that used to be owned by the descriptor
 653          * array.
 654          */
 655         while (fdp->fd_lastfile > 0 && fdp->fd_ofiles[fdp->fd_lastfile] == NULL)
 656                 fdp->fd_lastfile--;
 657         if (fd < fdp->fd_freefile)
 658                 fdp->fd_freefile = fd;
 659         if (fd < fdp->fd_knlistsize)
 660                 knote_fdclose(p, fd);
 661         error = closef(fp, td);
 662         if (holdleaders) {
 663                 fdp->fd_holdleaderscount--;
 664                 if (fdp->fd_holdleaderscount == 0 &&
 665                     fdp->fd_holdleaderswakeup != 0) {
 666                         fdp->fd_holdleaderswakeup = 0;
 667                         wakeup(&fdp->fd_holdleaderscount);
 668                 }
 669         }
 670         return (error);
 671 }
 672
 673 int
 674 kern_fstat(int fd, struct stat *ub)
 675 {
 676         struct thread *td = curthread;
 677         struct proc *p = td->td_proc;
 678         struct filedesc *fdp;
 679         struct file *fp;
 680         int error;
 681
 682         KKASSERT(p);
 683
 684         fdp = p->p_fd;
 685         if ((unsigned)fd >= fdp->fd_nfiles ||
 686             (fp = fdp->fd_ofiles[fd]) == NULL)
 687                 return (EBADF);
 688         fhold(fp);
 689         error = fo_stat(fp, ub, td);
 690         fdrop(fp, td);
 691
 692         return (error);
 693 }
 694
 695 /*
 696  * Return status information about a file descriptor.
 697  */
 698 int
 699 fstat(struct fstat_args *uap)
 700 {
 701         struct stat st;
 702         int error;
 703
 704         error = kern_fstat(uap->fd, &st);
 705
 706         if (error == 0)
 707                 error = copyout(&st, uap->sb, sizeof(st));
 708         return (error);
 709 }
 710
 711 /*
 712  * XXX: This is for source compatibility with NetBSD.  Probably doesn't
 713  * belong here.
 714  */
 715 int
 716 nfstat(struct nfstat_args *uap)
 717 {
 718         struct stat st;
 719         struct nstat nst;
 720         int error;
 721
 722         error = kern_fstat(uap->fd, &st);
 723
 724         if (error == 0) {
 725                 cvtnstat(&st, &nst);
 726                 error = copyout(&nst, uap->sb, sizeof (nst));
 727         }
 728         return (error);
 729 }
 730
 731 /*
 732  * Return pathconf information about a file descriptor.
 733  */
 734 /* ARGSUSED */
 735 int
 736 fpathconf(struct fpathconf_args *uap)
 737 {
 738         struct thread *td = curthread;
 739         struct proc *p = td->td_proc;
 740         struct filedesc *fdp;
 741         struct file *fp;
 742         struct vnode *vp;
 743         int error = 0;
 744
 745         KKASSERT(p);
 746         fdp = p->p_fd;
 747         if ((unsigned)uap->fd >= fdp->fd_nfiles ||
 748             (fp = fdp->fd_ofiles[uap->fd]) == NULL)
 749                 return (EBADF);
 750
 751         fhold(fp);
 752
 753         switch (fp->f_type) {
 754         case DTYPE_PIPE:
 755         case DTYPE_SOCKET:
 756                 if (uap->name != _PC_PIPE_BUF) {
 757                         error = EINVAL;
 758                 } else {
 759                         uap->sysmsg_result = PIPE_BUF;
 760                         error = 0;
 761                 }
 762                 break;
 763         case DTYPE_FIFO:
 764         case DTYPE_VNODE:
 765                 vp = (struct vnode *)fp->f_data;
 766                 error = VOP_PATHCONF(vp, uap->name, uap->sysmsg_fds);
 767                 break;
 768         default:
 769                 error = EOPNOTSUPP;
 770                 break;
 771         }
 772         fdrop(fp, td);
 773         return(error);
 774 }
 775
 776 /*
 777  * Allocate a file descriptor for the process.
 778  */
 779 static int fdexpand;
 780 SYSCTL_INT(_debug, OID_AUTO, fdexpand, CTLFLAG_RD, &fdexpand, 0, "");
 781
 782 int
 783 fdalloc(struct proc *p, int want, int *result)
 784 {
 785         struct filedesc *fdp = p->p_fd;
 786         int i;
 787         int lim, last, nfiles;
 788         struct file **newofile;
 789         char *newofileflags;
 790
 791         /*
 792          * Search for a free descriptor starting at the higher
 793          * of want or fd_freefile.  If that fails, consider
 794          * expanding the ofile array.
 795          */
 796         lim = min((int)p->p_rlimit[RLIMIT_NOFILE].rlim_cur, maxfilesperproc);
 797         for (;;) {
 798                 last = min(fdp->fd_nfiles, lim);
 799                 if ((i = want) < fdp->fd_freefile)
 800                         i = fdp->fd_freefile;
 801                 for (; i < last; i++) {
 802                         if (fdp->fd_ofiles[i] == NULL) {
 803                                 fdp->fd_ofileflags[i] = 0;
 804                                 if (i > fdp->fd_lastfile)
 805                                         fdp->fd_lastfile = i;
 806                                 if (want <= fdp->fd_freefile)
 807                                         fdp->fd_freefile = i;
 808                                 *result = i;
 809                                 return (0);
 810                         }
 811                 }
 812
 813                 /*
 814                  * No space in current array.  Expand?
 815                  */
 816                 if (fdp->fd_nfiles >= lim)
 817                         return (EMFILE);
 818                 if (fdp->fd_nfiles < NDEXTENT)
 819                         nfiles = NDEXTENT;
 820                 else
 821                         nfiles = 2 * fdp->fd_nfiles;
 822                 newofile = malloc(nfiles * OFILESIZE, M_FILEDESC, M_WAITOK);
 823
 824                 /*
 825                  * deal with file-table extend race that might have occured
 826                  * when malloc was blocked.
 827                  */
 828                 if (fdp->fd_nfiles >= nfiles) {
 829                         free(newofile, M_FILEDESC);
 830                         continue;
 831                 }
 832                 newofileflags = (char *) &newofile[nfiles];
 833                 /*
 834                  * Copy the existing ofile and ofileflags arrays
 835                  * and zero the new portion of each array.
 836                  */
 837                 bcopy(fdp->fd_ofiles, newofile,
 838                         (i = sizeof(struct file *) * fdp->fd_nfiles));
 839                 bzero((char *)newofile + i, nfiles * sizeof(struct file *) - i);
 840                 bcopy(fdp->fd_ofileflags, newofileflags,
 841                         (i = sizeof(char) * fdp->fd_nfiles));
 842                 bzero(newofileflags + i, nfiles * sizeof(char) - i);
 843                 if (fdp->fd_nfiles > NDFILE)
 844                         free(fdp->fd_ofiles, M_FILEDESC);
 845                 fdp->fd_ofiles = newofile;
 846                 fdp->fd_ofileflags = newofileflags;
 847                 fdp->fd_nfiles = nfiles;
 848                 fdexpand++;
 849         }
 850         return (0);
 851 }
 852
 853 /*
 854  * Check to see whether n user file descriptors
 855  * are available to the process p.
 856  */
 857 int
 858 fdavail(struct proc *p, int n)
 859 {
 860         struct filedesc *fdp = p->p_fd;
 861         struct file **fpp;
 862         int i, lim, last;
 863
 864         lim = min((int)p->p_rlimit[RLIMIT_NOFILE].rlim_cur, maxfilesperproc);
 865         if ((i = lim - fdp->fd_nfiles) > 0 && (n -= i) <= 0)
 866                 return (1);
 867
 868         last = min(fdp->fd_nfiles, lim);
 869         fpp = &fdp->fd_ofiles[fdp->fd_freefile];
 870         for (i = last - fdp->fd_freefile; --i >= 0; fpp++) {
 871                 if (*fpp == NULL && --n <= 0)
 872                         return (1);
 873         }
 874         return (0);
 875 }
 876
 877 /*
 878  * falloc:
 879  *      Create a new open file structure and allocate a file decriptor
 880  *      for the process that refers to it.  If p is NULL, no descriptor
 881  *      is allocated and the file pointer is returned unassociated with
 882  *      any process.  resultfd is only used if p is not NULL and may
 883  *      separately be NULL indicating that you don't need the returned fd.
 884  *
 885  *      A held file pointer is returned.  If a descriptor has been allocated
 886  *      an additional hold on the fp will be made due to the fd_ofiles[]
 887  *      reference.
 888  */
 889 int
 890 falloc(struct proc *p, struct file **resultfp, int *resultfd)
 891 {
 892         static struct timeval lastfail;
 893         static int curfail;
 894         struct file *fp;
 895         int error;
 896
 897         fp = NULL;
 898
 899         /*
 900          * Handle filetable full issues and root overfill.
 901          */
 902         if (nfiles >= maxfiles - maxfilesrootres &&
 903             ((p && p->p_ucred->cr_ruid != 0) || nfiles >= maxfiles)) {
 904                 if (ppsratecheck(&lastfail, &curfail, 1)) {
 905                         printf("kern.maxfiles limit exceeded by uid %d, please see tuning(7).\n",
 906                                 (p ? p->p_ucred->cr_ruid : -1));
 907                 }
 908                 error = ENFILE;
 909                 goto done;
 910         }
 911
 912         /*
 913          * Allocate a new file descriptor.
 914          */
 915         nfiles++;
 916         fp = malloc(sizeof(struct file), M_FILE, M_WAITOK | M_ZERO);
 917         fp->f_count = 1;
 918         fp->f_ops = &badfileops;
 919         fp->f_seqcount = 1;
 920         if (p)
 921                 fp->f_cred = crhold(p->p_ucred);
 922         else
 923                 fp->f_cred = crhold(proc0.p_ucred);
 924         LIST_INSERT_HEAD(&filehead, fp, f_list);
 925         if (resultfd) {
 926                 if ((error = fsetfd(p, fp, resultfd)) != 0) {
 927                         fdrop(fp, p->p_thread);
 928                         fp = NULL;
 929                 }
 930         } else {
 931                 error = 0;
 932         }
 933 done:
 934         *resultfp = fp;
 935         return (0);
 936 }
 937
 938 /*
 939  * Associate a file pointer with a file descriptor.  On success the fp
 940  * will have an additional ref representing the fd_ofiles[] association.
 941  */
 942 int
 943 fsetfd(struct proc *p, struct file *fp, int *resultfd)
 944 {
 945         int i;
 946         int error;
 947
 948         KKASSERT(p);
 949
 950         i = -1;
 951         if ((error = fdalloc(p, 0, &i)) == 0) {
 952                 fhold(fp);
 953                 p->p_fd->fd_ofiles[i] = fp;
 954         }
 955         *resultfd = i;
 956         return (0);
 957 }
 958
 959 void
 960 fsetcred(struct file *fp, struct ucred *cr)
 961 {
 962         crhold(cr);
 963         crfree(fp->f_cred);
 964         fp->f_cred = cr;
 965 }
 966
 967 /*
 968  * Free a file descriptor.
 969  */
 970 void
 971 ffree(struct file *fp)
 972 {
 973         KASSERT((fp->f_count == 0), ("ffree: fp_fcount not 0!"));
 974         LIST_REMOVE(fp, f_list);
 975         crfree(fp->f_cred);
 976         if (fp->f_ncp) {
 977             cache_drop(fp->f_ncp);
 978             fp->f_ncp = NULL;
 979         }
 980         nfiles--;
 981         free(fp, M_FILE);
 982 }
 983
 984 /*
 985  * Build a new filedesc structure.
 986  */
 987 struct filedesc *
 988 fdinit(struct proc *p)
 989 {
 990         struct filedesc0 *newfdp;
 991         struct filedesc *fdp = p->p_fd;
 992
 993         newfdp = malloc(sizeof(struct filedesc0), M_FILEDESC, M_WAITOK|M_ZERO);
 994         newfdp->fd_fd.fd_cdir = fdp->fd_cdir;
 995         if (newfdp->fd_fd.fd_cdir) {
 996                 vref(newfdp->fd_fd.fd_cdir);
 997                 newfdp->fd_fd.fd_ncdir = cache_hold(fdp->fd_ncdir);
 998         }
 999         newfdp->fd_fd.fd_rdir = fdp->fd_rdir;
1000         newfdp->fd_fd.fd_nrdir = cache_hold(fdp->fd_nrdir);
1001         vref(newfdp->fd_fd.fd_rdir);
1002         newfdp->fd_fd.fd_jdir = fdp->fd_jdir;
1003         if (newfdp->fd_fd.fd_jdir) {
1004                 vref(newfdp->fd_fd.fd_jdir);
1005                 newfdp->fd_fd.fd_njdir = cache_hold(fdp->fd_njdir);
1006         }
1007
1008
1009         /* Create the file descriptor table. */
1010         newfdp->fd_fd.fd_refcnt = 1;
1011         newfdp->fd_fd.fd_cmask = cmask;
1012         newfdp->fd_fd.fd_ofiles = newfdp->fd_dfiles;
1013         newfdp->fd_fd.fd_ofileflags = newfdp->fd_dfileflags;
1014         newfdp->fd_fd.fd_nfiles = NDFILE;
1015         newfdp->fd_fd.fd_knlistsize = -1;
1016
1017         return (&newfdp->fd_fd);
1018 }
1019
1020 /*
1021  * Share a filedesc structure.
1022  */
1023 struct filedesc *
1024 fdshare(struct proc *p)
1025 {
1026         p->p_fd->fd_refcnt++;
1027         return (p->p_fd);
1028 }
1029
1030 /*
1031  * Copy a filedesc structure.
1032  */
1033 struct filedesc *
1034 fdcopy(struct proc *p)
1035 {
1036         struct filedesc *newfdp, *fdp = p->p_fd;
1037         struct file **fpp;
1038         int i;
1039
1040         /* Certain daemons might not have file descriptors. */
1041         if (fdp == NULL)
1042                 return (NULL);
1043
1044         newfdp = malloc(sizeof(struct filedesc0), M_FILEDESC, M_WAITOK);
1045         bcopy(fdp, newfdp, sizeof(struct filedesc));
1046         if (newfdp->fd_cdir) {
1047                 vref(newfdp->fd_cdir);
1048                 newfdp->fd_ncdir = cache_hold(newfdp->fd_ncdir);
1049         }
1050         /*
1051          * We must check for fd_rdir here, at least for now because
1052          * the init process is created before we have access to the
1053          * rootvode to take a reference to it.
1054          */
1055         if (newfdp->fd_rdir) {
1056                 vref(newfdp->fd_rdir);
1057                 newfdp->fd_nrdir = cache_hold(newfdp->fd_nrdir);
1058         }
1059         if (newfdp->fd_jdir) {
1060                 vref(newfdp->fd_jdir);
1061                 newfdp->fd_njdir = cache_hold(newfdp->fd_njdir);
1062         }
1063         newfdp->fd_refcnt = 1;
1064
1065         /*
1066          * If the number of open files fits in the internal arrays
1067          * of the open file structure, use them, otherwise allocate
1068          * additional memory for the number of descriptors currently
1069          * in use.
1070          */
1071         if (newfdp->fd_lastfile < NDFILE) {
1072                 newfdp->fd_ofiles = ((struct filedesc0 *) newfdp)->fd_dfiles;
1073                 newfdp->fd_ofileflags =
1074                     ((struct filedesc0 *) newfdp)->fd_dfileflags;
1075                 i = NDFILE;
1076         } else {
1077                 /*
1078                  * Compute the smallest multiple of NDEXTENT needed
1079                  * for the file descriptors currently in use,
1080                  * allowing the table to shrink.
1081                  */
1082                 i = newfdp->fd_nfiles;
1083                 while (i > 2 * NDEXTENT && i > newfdp->fd_lastfile * 2)
1084                         i /= 2;
1085                 newfdp->fd_ofiles = malloc(i * OFILESIZE, M_FILEDESC, M_WAITOK);
1086                 newfdp->fd_ofileflags = (char *) &newfdp->fd_ofiles[i];
1087         }
1088         newfdp->fd_nfiles = i;
1089         bcopy(fdp->fd_ofiles, newfdp->fd_ofiles, i * sizeof(struct file **));
1090         bcopy(fdp->fd_ofileflags, newfdp->fd_ofileflags, i * sizeof(char));
1091
1092         /*
1093          * kq descriptors cannot be copied.
1094          */
1095         if (newfdp->fd_knlistsize != -1) {
1096                 fpp = &newfdp->fd_ofiles[newfdp->fd_lastfile];
1097                 for (i = newfdp->fd_lastfile; i >= 0; i--, fpp--) {
1098                         if (*fpp != NULL && (*fpp)->f_type == DTYPE_KQUEUE) {
1099                                 *fpp = NULL;
1100                                 if (i < newfdp->fd_freefile)
1101                                         newfdp->fd_freefile = i;
1102                         }
1103                         if (*fpp == NULL && i == newfdp->fd_lastfile && i > 0)
1104                                 newfdp->fd_lastfile--;
1105                 }
1106                 newfdp->fd_knlist = NULL;
1107                 newfdp->fd_knlistsize = -1;
1108                 newfdp->fd_knhash = NULL;
1109                 newfdp->fd_knhashmask = 0;
1110         }
1111
1112         fpp = newfdp->fd_ofiles;
1113         for (i = newfdp->fd_lastfile; i-- >= 0; fpp++) {
1114                 if (*fpp != NULL)
1115                         fhold(*fpp);
1116         }
1117         return (newfdp);
1118 }
1119
1120 /*
1121  * Release a filedesc structure.
1122  */
1123 void
1124 fdfree(struct proc *p)
1125 {
1126         struct thread *td = p->p_thread;
1127         struct filedesc *fdp = p->p_fd;
1128         struct file **fpp;
1129         int i;
1130         struct filedesc_to_leader *fdtol;
1131         struct file *fp;
1132         struct vnode *vp;
1133         struct flock lf;
1134
1135         /* Certain daemons might not have file descriptors. */
1136         if (fdp == NULL)
1137                 return;
1138
1139         /* Check for special need to clear POSIX style locks */
1140         fdtol = p->p_fdtol;
1141         if (fdtol != NULL) {
1142                 KASSERT(fdtol->fdl_refcount > 0,
1143                         ("filedesc_to_refcount botch: fdl_refcount=%d",
1144                          fdtol->fdl_refcount));
1145                 if (fdtol->fdl_refcount == 1 &&
1146                     (p->p_leader->p_flag & P_ADVLOCK) != 0) {
1147                         i = 0;
1148                         fpp = fdp->fd_ofiles;
1149                         for (i = 0, fpp = fdp->fd_ofiles;
1150                              i <= fdp->fd_lastfile;
1151                              i++, fpp++) {
1152                                 if (*fpp == NULL ||
1153                                     (*fpp)->f_type != DTYPE_VNODE)
1154                                         continue;
1155                                 fp = *fpp;
1156                                 fhold(fp);
1157                                 lf.l_whence = SEEK_SET;
1158                                 lf.l_start = 0;
1159                                 lf.l_len = 0;
1160                                 lf.l_type = F_UNLCK;
1161                                 vp = (struct vnode *)fp->f_data;
1162                                 (void) VOP_ADVLOCK(vp,
1163                                                    (caddr_t)p->p_leader,
1164                                                    F_UNLCK,
1165                                                    &lf,
1166                                                    F_POSIX);
1167                                 fdrop(fp, p->p_thread);
1168                                 fpp = fdp->fd_ofiles + i;
1169                         }
1170                 }
1171         retry:
1172                 if (fdtol->fdl_refcount == 1) {
1173                         if (fdp->fd_holdleaderscount > 0 &&
1174                             (p->p_leader->p_flag & P_ADVLOCK) != 0) {
1175                                 /*
1176                                  * close() or do_dup() has cleared a reference
1177                                  * in a shared file descriptor table.
1178                                  */
1179                                 fdp->fd_holdleaderswakeup = 1;
1180                                 tsleep(&fdp->fd_holdleaderscount,
1181                                        0, "fdlhold", 0);
1182                                 goto retry;
1183                         }
1184                         if (fdtol->fdl_holdcount > 0) {
1185                                 /*
1186                                  * Ensure that fdtol->fdl_leader
1187                                  * remains valid in closef().
1188                                  */
1189                                 fdtol->fdl_wakeup = 1;
1190                                 tsleep(fdtol, 0, "fdlhold", 0);
1191                                 goto retry;
1192                         }
1193                 }
1194                 fdtol->fdl_refcount--;
1195                 if (fdtol->fdl_refcount == 0 &&
1196                     fdtol->fdl_holdcount == 0) {
1197                         fdtol->fdl_next->fdl_prev = fdtol->fdl_prev;
1198                         fdtol->fdl_prev->fdl_next = fdtol->fdl_next;
1199                 } else
1200                         fdtol = NULL;
1201                 p->p_fdtol = NULL;
1202                 if (fdtol != NULL)
1203                         free(fdtol, M_FILEDESC_TO_LEADER);
1204         }
1205         if (--fdp->fd_refcnt > 0)
1206                 return;
1207         /*
1208          * we are the last reference to the structure, we can
1209          * safely assume it will not change out from under us.
1210          */
1211         fpp = fdp->fd_ofiles;
1212         for (i = fdp->fd_lastfile; i-- >= 0; fpp++) {
1213                 if (*fpp)
1214                         (void) closef(*fpp, td);
1215         }
1216         if (fdp->fd_nfiles > NDFILE)
1217                 free(fdp->fd_ofiles, M_FILEDESC);
1218         if (fdp->fd_cdir) {
1219                 cache_drop(fdp->fd_ncdir);
1220                 vrele(fdp->fd_cdir);
1221         }
1222         cache_drop(fdp->fd_nrdir);
1223         vrele(fdp->fd_rdir);
1224         if (fdp->fd_jdir) {
1225                 cache_drop(fdp->fd_njdir);
1226                 vrele(fdp->fd_jdir);
1227         }
1228         if (fdp->fd_knlist)
1229                 free(fdp->fd_knlist, M_KQUEUE);
1230         if (fdp->fd_knhash)
1231                 free(fdp->fd_knhash, M_KQUEUE);
1232         free(fdp, M_FILEDESC);
1233 }
1234
1235 /*
1236  * For setugid programs, we don't want to people to use that setugidness
1237  * to generate error messages which write to a file which otherwise would
1238  * otherwise be off-limits to the process.
1239  *
1240  * This is a gross hack to plug the hole.  A better solution would involve
1241  * a special vop or other form of generalized access control mechanism.  We
1242  * go ahead and just reject all procfs file systems accesses as dangerous.
1243  *
1244  * Since setugidsafety calls this only for fd 0, 1 and 2, this check is
1245  * sufficient.  We also don't for check setugidness since we know we are.
1246  */
1247 static int
1248 is_unsafe(struct file *fp)
1249 {
1250         if (fp->f_type == DTYPE_VNODE &&
1251             ((struct vnode *)(fp->f_data))->v_tag == VT_PROCFS)
1252                 return (1);
1253         return (0);
1254 }
1255
1256 /*
1257  * Make this setguid thing safe, if at all possible.
1258  */
1259 void
1260 setugidsafety(struct proc *p)
1261 {
1262         struct thread *td = p->p_thread;
1263         struct filedesc *fdp = p->p_fd;
1264         int i;
1265
1266         /* Certain daemons might not have file descriptors. */
1267         if (fdp == NULL)
1268                 return;
1269
1270         /*
1271          * note: fdp->fd_ofiles may be reallocated out from under us while
1272          * we are blocked in a close.  Be careful!
1273          */
1274         for (i = 0; i <= fdp->fd_lastfile; i++) {
1275                 if (i > 2)
1276                         break;
1277                 if (fdp->fd_ofiles[i] && is_unsafe(fdp->fd_ofiles[i])) {
1278                         struct file *fp;
1279
1280 #if 0
1281                         if ((fdp->fd_ofileflags[i] & UF_MAPPED) != 0)
1282                                 (void) munmapfd(p, i);
1283 #endif
1284                         if (i < fdp->fd_knlistsize)
1285                                 knote_fdclose(p, i);
1286                         /*
1287                          * NULL-out descriptor prior to close to avoid
1288                          * a race while close blocks.
1289                          */
1290                         fp = fdp->fd_ofiles[i];
1291                         fdp->fd_ofiles[i] = NULL;
1292                         fdp->fd_ofileflags[i] = 0;
1293                         if (i < fdp->fd_freefile)
1294                                 fdp->fd_freefile = i;
1295                         (void) closef(fp, td);
1296                 }
1297         }
1298         while (fdp->fd_lastfile > 0 && fdp->fd_ofiles[fdp->fd_lastfile] == NULL)
1299                 fdp->fd_lastfile--;
1300 }
1301
1302 /*
1303  * Close any files on exec?
1304  */
1305 void
1306 fdcloseexec(struct proc *p)
1307 {
1308         struct thread *td = p->p_thread;
1309         struct filedesc *fdp = p->p_fd;
1310         int i;
1311
1312         /* Certain daemons might not have file descriptors. */
1313         if (fdp == NULL)
1314                 return;
1315
1316         /*
1317          * We cannot cache fd_ofiles or fd_ofileflags since operations
1318          * may block and rip them out from under us.
1319          */
1320         for (i = 0; i <= fdp->fd_lastfile; i++) {
1321                 if (fdp->fd_ofiles[i] != NULL &&
1322                     (fdp->fd_ofileflags[i] & UF_EXCLOSE)) {
1323                         struct file *fp;
1324
1325 #if 0
1326                         if (fdp->fd_ofileflags[i] & UF_MAPPED)
1327                                 (void) munmapfd(p, i);
1328 #endif
1329                         if (i < fdp->fd_knlistsize)
1330                                 knote_fdclose(p, i);
1331                         /*
1332                          * NULL-out descriptor prior to close to avoid
1333                          * a race while close blocks.
1334                          */
1335                         fp = fdp->fd_ofiles[i];
1336                         fdp->fd_ofiles[i] = NULL;
1337                         fdp->fd_ofileflags[i] = 0;
1338                         if (i < fdp->fd_freefile)
1339                                 fdp->fd_freefile = i;
1340                         (void) closef(fp, td);
1341                 }
1342         }
1343         while (fdp->fd_lastfile > 0 && fdp->fd_ofiles[fdp->fd_lastfile] == NULL)
1344                 fdp->fd_lastfile--;
1345 }
1346
1347 /*
1348  * It is unsafe for set[ug]id processes to be started with file
1349  * descriptors 0..2 closed, as these descriptors are given implicit
1350  * significance in the Standard C library.  fdcheckstd() will create a
1351  * descriptor referencing /dev/null for each of stdin, stdout, and
1352  * stderr that is not already open.
1353  */
1354 int
1355 fdcheckstd(struct proc *p)
1356 {
1357         struct thread *td = p->p_thread;
1358         struct nlookupdata nd;
1359         struct filedesc *fdp;
1360         struct file *fp;
1361         register_t retval;
1362         int fd, i, error, flags, devnull;
1363
1364        fdp = p->p_fd;
1365        if (fdp == NULL)
1366                return (0);
1367        devnull = -1;
1368        error = 0;
1369        for (i = 0; i < 3; i++) {
1370                 if (fdp->fd_ofiles[i] != NULL)
1371                         continue;
1372                 if (devnull < 0) {
1373                         if ((error = falloc(p, &fp, NULL)) != 0)
1374                                 break;
1375
1376                         error = nlookup_init(&nd, "/dev/null", UIO_SYSSPACE,
1377                                                 NLC_FOLLOW|NLC_LOCKVP);
1378                         flags = FREAD | FWRITE;
1379                         if (error == 0)
1380                                 error = vn_open(&nd, fp, flags, 0);
1381                         if (error == 0)
1382                                 error = fsetfd(p, fp, &fd);
1383                         fdrop(fp, td);
1384                         nlookup_done(&nd);
1385                         if (error)
1386                                 break;
1387                         KKASSERT(i == fd);
1388                         devnull = fd;
1389                 } else {
1390                         error = kern_dup(DUP_FIXED, devnull, i, &retval);
1391                         if (error != 0)
1392                                 break;
1393                 }
1394        }
1395        return (error);
1396 }
1397
1398 /*
1399  * Internal form of close.
1400  * Decrement reference count on file structure.
1401  * Note: td and/or p may be NULL when closing a file
1402  * that was being passed in a message.
1403  */
1404 int
1405 closef(struct file *fp, struct thread *td)
1406 {
1407         struct vnode *vp;
1408         struct flock lf;
1409         struct filedesc_to_leader *fdtol;
1410         struct proc *p;
1411
1412         if (fp == NULL)
1413                 return (0);
1414         if (td == NULL) {
1415                 td = curthread;
1416                 p = NULL;               /* allow no proc association */
1417         } else {
1418                 p = td->td_proc;        /* can also be NULL */
1419         }
1420         /*
1421          * POSIX record locking dictates that any close releases ALL
1422          * locks owned by this process.  This is handled by setting
1423          * a flag in the unlock to free ONLY locks obeying POSIX
1424          * semantics, and not to free BSD-style file locks.
1425          * If the descriptor was in a message, POSIX-style locks
1426          * aren't passed with the descriptor.
1427          */
1428         if (p != NULL &&
1429             fp->f_type == DTYPE_VNODE) {
1430                 if ((p->p_leader->p_flag & P_ADVLOCK) != 0) {
1431                         lf.l_whence = SEEK_SET;
1432                         lf.l_start = 0;
1433                         lf.l_len = 0;
1434                         lf.l_type = F_UNLCK;
1435                         vp = (struct vnode *)fp->f_data;
1436                         (void) VOP_ADVLOCK(vp, (caddr_t)p->p_leader, F_UNLCK,
1437                                            &lf, F_POSIX);
1438                 }
1439                 fdtol = p->p_fdtol;
1440                 if (fdtol != NULL) {
1441                         /*
1442                          * Handle special case where file descriptor table
1443                          * is shared between multiple process leaders.
1444                          */
1445                         for (fdtol = fdtol->fdl_next;
1446                              fdtol != p->p_fdtol;
1447                              fdtol = fdtol->fdl_next) {
1448                                 if ((fdtol->fdl_leader->p_flag &
1449                                      P_ADVLOCK) == 0)
1450                                         continue;
1451                                 fdtol->fdl_holdcount++;
1452                                 lf.l_whence = SEEK_SET;
1453                                 lf.l_start = 0;
1454                                 lf.l_len = 0;
1455                                 lf.l_type = F_UNLCK;
1456                                 vp = (struct vnode *)fp->f_data;
1457                                 (void) VOP_ADVLOCK(vp,
1458                                                    (caddr_t)p->p_leader,
1459                                                    F_UNLCK, &lf, F_POSIX);
1460                                 fdtol->fdl_holdcount--;
1461                                 if (fdtol->fdl_holdcount == 0 &&
1462                                     fdtol->fdl_wakeup != 0) {
1463                                         fdtol->fdl_wakeup = 0;
1464                                         wakeup(fdtol);
1465                                 }
1466                         }
1467                 }
1468         }
1469         return (fdrop(fp, td));
1470 }
1471
1472 int
1473 fdrop(struct file *fp, struct thread *td)
1474 {
1475         struct flock lf;
1476         struct vnode *vp;
1477         int error;
1478
1479         if (--fp->f_count > 0)
1480                 return (0);
1481         if (fp->f_count < 0)
1482                 panic("fdrop: count < 0");
1483         if ((fp->f_flag & FHASLOCK) && fp->f_type == DTYPE_VNODE) {
1484                 lf.l_whence = SEEK_SET;
1485                 lf.l_start = 0;
1486                 lf.l_len = 0;
1487                 lf.l_type = F_UNLCK;
1488                 vp = (struct vnode *)fp->f_data;
1489                 (void) VOP_ADVLOCK(vp, (caddr_t)fp, F_UNLCK, &lf, F_FLOCK);
1490         }
1491         if (fp->f_ops != &badfileops)
1492                 error = fo_close(fp, td);
1493         else
1494                 error = 0;
1495         ffree(fp);
1496         return (error);
1497 }
1498
1499 /*
1500  * Apply an advisory lock on a file descriptor.
1501  *
1502  * Just attempt to get a record lock of the requested type on
1503  * the entire file (l_whence = SEEK_SET, l_start = 0, l_len = 0).
1504  */
1505 /* ARGSUSED */
1506 int
1507 flock(struct flock_args *uap)
1508 {
1509         struct proc *p = curproc;
1510         struct filedesc *fdp = p->p_fd;
1511         struct file *fp;
1512         struct vnode *vp;
1513         struct flock lf;
1514
1515         if ((unsigned)uap->fd >= fdp->fd_nfiles ||
1516             (fp = fdp->fd_ofiles[uap->fd]) == NULL)
1517                 return (EBADF);
1518         if (fp->f_type != DTYPE_VNODE)
1519                 return (EOPNOTSUPP);
1520         vp = (struct vnode *)fp->f_data;
1521         lf.l_whence = SEEK_SET;
1522         lf.l_start = 0;
1523         lf.l_len = 0;
1524         if (uap->how & LOCK_UN) {
1525                 lf.l_type = F_UNLCK;
1526                 fp->f_flag &= ~FHASLOCK;
1527                 return (VOP_ADVLOCK(vp, (caddr_t)fp, F_UNLCK, &lf, F_FLOCK));
1528         }
1529         if (uap->how & LOCK_EX)
1530                 lf.l_type = F_WRLCK;
1531         else if (uap->how & LOCK_SH)
1532                 lf.l_type = F_RDLCK;
1533         else
1534                 return (EBADF);
1535         fp->f_flag |= FHASLOCK;
1536         if (uap->how & LOCK_NB)
1537                 return (VOP_ADVLOCK(vp, (caddr_t)fp, F_SETLK, &lf, F_FLOCK));
1538         return (VOP_ADVLOCK(vp, (caddr_t)fp, F_SETLK, &lf, F_FLOCK|F_WAIT));
1539 }
1540
1541 /*
1542  * File Descriptor pseudo-device driver (/dev/fd/).
1543  *
1544  * Opening minor device N dup()s the file (if any) connected to file
1545  * descriptor N belonging to the calling process.  Note that this driver
1546  * consists of only the ``open()'' routine, because all subsequent
1547  * references to this file will be direct to the other driver.
1548  */
1549 /* ARGSUSED */
1550 static int
1551 fdopen(dev_t dev, int mode, int type, struct thread *td)
1552 {
1553         KKASSERT(td->td_proc != NULL);
1554
1555         /*
1556          * XXX Kludge: set curproc->p_dupfd to contain the value of the
1557          * the file descriptor being sought for duplication. The error
1558          * return ensures that the vnode for this device will be released
1559          * by vn_open. Open will detect this special error and take the
1560          * actions in dupfdopen below. Other callers of vn_open or VOP_OPEN
1561          * will simply report the error.
1562          */
1563         td->td_proc->p_dupfd = minor(dev);
1564         return (ENODEV);
1565 }
1566
1567 /*
1568  * Duplicate the specified descriptor to a free descriptor.
1569  */
1570 int
1571 dupfdopen(struct filedesc *fdp, int indx, int dfd, int mode, int error)
1572 {
1573         struct file *wfp;
1574         struct file *fp;
1575
1576         /*
1577          * If the to-be-dup'd fd number is greater than the allowed number
1578          * of file descriptors, or the fd to be dup'd has already been
1579          * closed, then reject.
1580          */
1581         if ((u_int)dfd >= fdp->fd_nfiles ||
1582             (wfp = fdp->fd_ofiles[dfd]) == NULL) {
1583                 return (EBADF);
1584         }
1585
1586         /*
1587          * There are two cases of interest here.
1588          *
1589          * For ENODEV simply dup (dfd) to file descriptor
1590          * (indx) and return.
1591          *
1592          * For ENXIO steal away the file structure from (dfd) and
1593          * store it in (indx).  (dfd) is effectively closed by
1594          * this operation.
1595          *
1596          * Any other error code is just returned.
1597          */
1598         switch (error) {
1599         case ENODEV:
1600                 /*
1601                  * Check that the mode the file is being opened for is a
1602                  * subset of the mode of the existing descriptor.
1603                  */
1604                 if (((mode & (FREAD|FWRITE)) | wfp->f_flag) != wfp->f_flag)
1605                         return (EACCES);
1606                 fp = fdp->fd_ofiles[indx];
1607 #if 0
1608                 if (fp && fdp->fd_ofileflags[indx] & UF_MAPPED)
1609                         (void) munmapfd(p, indx);
1610 #endif
1611                 fdp->fd_ofiles[indx] = wfp;
1612                 fdp->fd_ofileflags[indx] = fdp->fd_ofileflags[dfd];
1613                 fhold(wfp);
1614                 if (indx > fdp->fd_lastfile)
1615                         fdp->fd_lastfile = indx;
1616                 /*
1617                  * we now own the reference to fp that the ofiles[] array
1618                  * used to own.  Release it.
1619                  */
1620                 if (fp)
1621                         fdrop(fp, curthread);
1622                 return (0);
1623
1624         case ENXIO:
1625                 /*
1626                  * Steal away the file pointer from dfd, and stuff it into indx.
1627                  */
1628                 fp = fdp->fd_ofiles[indx];
1629 #if 0
1630                 if (fp && fdp->fd_ofileflags[indx] & UF_MAPPED)
1631                         (void) munmapfd(p, indx);
1632 #endif
1633                 fdp->fd_ofiles[indx] = fdp->fd_ofiles[dfd];
1634                 fdp->fd_ofiles[dfd] = NULL;
1635                 fdp->fd_ofileflags[indx] = fdp->fd_ofileflags[dfd];
1636                 fdp->fd_ofileflags[dfd] = 0;
1637
1638                 /*
1639                  * we now own the reference to fp that the ofiles[] array
1640                  * used to own.  Release it.
1641                  */
1642                 if (fp)
1643                         fdrop(fp, curthread);
1644                 /*
1645                  * Complete the clean up of the filedesc structure by
1646                  * recomputing the various hints.
1647                  */
1648                 if (indx > fdp->fd_lastfile) {
1649                         fdp->fd_lastfile = indx;
1650                 } else {
1651                         while (fdp->fd_lastfile > 0 &&
1652                            fdp->fd_ofiles[fdp->fd_lastfile] == NULL) {
1653                                 fdp->fd_lastfile--;
1654                         }
1655                         if (dfd < fdp->fd_freefile)
1656                                 fdp->fd_freefile = dfd;
1657                 }
1658                 return (0);
1659
1660         default:
1661                 return (error);
1662         }
1663         /* NOTREACHED */
1664 }
1665
1666
1667 struct filedesc_to_leader *
1668 filedesc_to_leader_alloc(struct filedesc_to_leader *old,
1669                          struct proc *leader)
1670 {
1671         struct filedesc_to_leader *fdtol;
1672
1673         fdtol = malloc(sizeof(struct filedesc_to_leader),
1674                         M_FILEDESC_TO_LEADER, M_WAITOK);
1675         fdtol->fdl_refcount = 1;
1676         fdtol->fdl_holdcount = 0;
1677         fdtol->fdl_wakeup = 0;
1678         fdtol->fdl_leader = leader;
1679         if (old != NULL) {
1680                 fdtol->fdl_next = old->fdl_next;
1681                 fdtol->fdl_prev = old;
1682                 old->fdl_next = fdtol;
1683                 fdtol->fdl_next->fdl_prev = fdtol;
1684         } else {
1685                 fdtol->fdl_next = fdtol;
1686                 fdtol->fdl_prev = fdtol;
1687         }
1688         return fdtol;
1689 }
1690
1691 /*
1692  * Get file structures.
1693  */
1694 static int
1695 sysctl_kern_file(SYSCTL_HANDLER_ARGS)
1696 {
1697         int error;
1698         struct file *fp;
1699
1700         if (!req->oldptr) {
1701                 /*
1702                  * overestimate by 10 files
1703                  */
1704                 return (SYSCTL_OUT(req, 0, sizeof(filehead) +
1705                                 (nfiles + 10) * sizeof(struct file)));
1706         }
1707
1708         error = SYSCTL_OUT(req, (caddr_t)&filehead, sizeof(filehead));
1709         if (error)
1710                 return (error);
1711
1712         /*
1713          * followed by an array of file structures
1714          */
1715         LIST_FOREACH(fp, &filehead, f_list) {
1716                 error = SYSCTL_OUT(req, (caddr_t)fp, sizeof (struct file));
1717                 if (error)
1718                         return (error);
1719         }
1720         return (0);
1721 }
1722
1723 SYSCTL_PROC(_kern, KERN_FILE, file, CTLTYPE_OPAQUE|CTLFLAG_RD,
1724     0, 0, sysctl_kern_file, "S,file", "Entire file table");
1725
1726 SYSCTL_INT(_kern, KERN_MAXFILESPERPROC, maxfilesperproc, CTLFLAG_RW,
1727     &maxfilesperproc, 0, "Maximum files allowed open per process");
1728
1729 SYSCTL_INT(_kern, KERN_MAXFILES, maxfiles, CTLFLAG_RW,
1730     &maxfiles, 0, "Maximum number of files");
1731
1732 SYSCTL_INT(_kern, OID_AUTO, maxfilesrootres, CTLFLAG_RW,
1733     &maxfilesrootres, 0, "Descriptors reserved for root use");
1734
1735 SYSCTL_INT(_kern, OID_AUTO, openfiles, CTLFLAG_RD,
1736         &nfiles, 0, "System-wide number of open files");
1737
1738 static void
1739 fildesc_drvinit(void *unused)
1740 {
1741         int fd;
1742
1743         cdevsw_add(&fildesc_cdevsw, 0, 0);
1744         for (fd = 0; fd < NUMFDESC; fd++) {
1745                 make_dev(&fildesc_cdevsw, fd,
1746                     UID_BIN, GID_BIN, 0666, "fd/%d", fd);
1747         }
1748         make_dev(&fildesc_cdevsw, 0, UID_ROOT, GID_WHEEL, 0666, "stdin");
1749         make_dev(&fildesc_cdevsw, 1, UID_ROOT, GID_WHEEL, 0666, "stdout");
1750         make_dev(&fildesc_cdevsw, 2, UID_ROOT, GID_WHEEL, 0666, "stderr");
1751 }
1752
1753 struct fileops badfileops = {
1754         NULL,   /* port */
1755         NULL,   /* clone */
1756         badfo_readwrite,
1757         badfo_readwrite,
1758         badfo_ioctl,
1759         badfo_poll,
1760         badfo_kqfilter,
1761         badfo_stat,
1762         badfo_close
1763 };
1764
1765 static int
1766 badfo_readwrite(
1767         struct file *fp,
1768         struct uio *uio,
1769         struct ucred *cred,
1770         int flags,
1771         struct thread *td
1772 ) {
1773         return (EBADF);
1774 }
1775
1776 static int
1777 badfo_ioctl(struct file *fp, u_long com, caddr_t data, struct thread *td)
1778 {
1779         return (EBADF);
1780 }
1781
1782 static int
1783 badfo_poll(struct file *fp, int events, struct ucred *cred, struct thread *td)
1784 {
1785         return (0);
1786 }
1787
1788 static int
1789 badfo_kqfilter(struct file *fp, struct knote *kn)
1790 {
1791         return (0);
1792 }
1793
1794 static int
1795 badfo_stat(struct file *fp, struct stat *sb, struct thread *td)
1796 {
1797         return (EBADF);
1798 }
1799
1800 static int
1801 badfo_close(struct file *fp, struct thread *td)
1802 {
1803         return (EBADF);
1804 }
1805
1806 SYSINIT(fildescdev,SI_SUB_DRIVERS,SI_ORDER_MIDDLE+CDEV_MAJOR,
1807                                         fildesc_drvinit,NULL)