Rename do_dup() to kern_dup() and pull in some changes from FreeBSD-CURRENT.
[dragonfly.git] / 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.17 2003/06/06 20:21:32 tegge Exp $
40  * $DragonFly: src/sys/kern/kern_descrip.c,v 1.16 2003/10/15 06:38:46 daver 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/namei.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         /* autoq */     0,
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(sigio)
498         struct sigio *sigio;
499 {
500         int s;
501
502         if (sigio == NULL)
503                 return;
504         s = splhigh();
505         *(sigio->sio_myref) = NULL;
506         splx(s);
507         if (sigio->sio_pgid < 0) {
508                 SLIST_REMOVE(&sigio->sio_pgrp->pg_sigiolst, sigio,
509                              sigio, sio_pgsigio);
510         } else /* if ((*sigiop)->sio_pgid > 0) */ {
511                 SLIST_REMOVE(&sigio->sio_proc->p_sigiolst, sigio,
512                              sigio, sio_pgsigio);
513         }
514         crfree(sigio->sio_ucred);
515         FREE(sigio, M_SIGIO);
516 }
517
518 /* Free a list of sigio structures. */
519 void
520 funsetownlst(sigiolst)
521         struct sigiolst *sigiolst;
522 {
523         struct sigio *sigio;
524
525         while ((sigio = SLIST_FIRST(sigiolst)) != NULL)
526                 funsetown(sigio);
527 }
528
529 /*
530  * This is common code for FIOSETOWN ioctl called by fcntl(fd, F_SETOWN, arg).
531  *
532  * After permission checking, add a sigio structure to the sigio list for
533  * the process or process group.
534  */
535 int
536 fsetown(pgid, sigiop)
537         pid_t pgid;
538         struct sigio **sigiop;
539 {
540         struct proc *proc;
541         struct pgrp *pgrp;
542         struct sigio *sigio;
543         int s;
544
545         if (pgid == 0) {
546                 funsetown(*sigiop);
547                 return (0);
548         }
549         if (pgid > 0) {
550                 proc = pfind(pgid);
551                 if (proc == NULL)
552                         return (ESRCH);
553
554                 /*
555                  * Policy - Don't allow a process to FSETOWN a process
556                  * in another session.
557                  *
558                  * Remove this test to allow maximum flexibility or
559                  * restrict FSETOWN to the current process or process
560                  * group for maximum safety.
561                  */
562                 if (proc->p_session != curproc->p_session)
563                         return (EPERM);
564
565                 pgrp = NULL;
566         } else /* if (pgid < 0) */ {
567                 pgrp = pgfind(-pgid);
568                 if (pgrp == NULL)
569                         return (ESRCH);
570
571                 /*
572                  * Policy - Don't allow a process to FSETOWN a process
573                  * in another session.
574                  *
575                  * Remove this test to allow maximum flexibility or
576                  * restrict FSETOWN to the current process or process
577                  * group for maximum safety.
578                  */
579                 if (pgrp->pg_session != curproc->p_session)
580                         return (EPERM);
581
582                 proc = NULL;
583         }
584         funsetown(*sigiop);
585         MALLOC(sigio, struct sigio *, sizeof(struct sigio), M_SIGIO, M_WAITOK);
586         if (pgid > 0) {
587                 SLIST_INSERT_HEAD(&proc->p_sigiolst, sigio, sio_pgsigio);
588                 sigio->sio_proc = proc;
589         } else {
590                 SLIST_INSERT_HEAD(&pgrp->pg_sigiolst, sigio, sio_pgsigio);
591                 sigio->sio_pgrp = pgrp;
592         }
593         sigio->sio_pgid = pgid;
594         sigio->sio_ucred = crhold(curproc->p_ucred);
595         /* It would be convenient if p_ruid was in ucred. */
596         sigio->sio_ruid = curproc->p_ucred->cr_ruid;
597         sigio->sio_myref = sigiop;
598         s = splhigh();
599         *sigiop = sigio;
600         splx(s);
601         return (0);
602 }
603
604 /*
605  * This is common code for FIOGETOWN ioctl called by fcntl(fd, F_GETOWN, arg).
606  */
607 pid_t
608 fgetown(sigio)
609         struct sigio *sigio;
610 {
611         return (sigio != NULL ? sigio->sio_pgid : 0);
612 }
613
614 /*
615  * Close a file descriptor.
616  */
617 /* ARGSUSED */
618 int
619 close(struct close_args *uap)
620 {
621         struct thread *td = curthread;
622         struct proc *p = td->td_proc;
623         struct filedesc *fdp;
624         struct file *fp;
625         int fd = uap->fd;
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 #if defined(COMPAT_43) || defined(COMPAT_SUNOS)
674 /*
675  * Return status information about a file descriptor.
676  */
677 /* ARGSUSED */
678 int
679 ofstat(struct ofstat_args *uap)
680 {
681         struct thread *td = curthread;
682         struct proc *p = td->td_proc;
683         struct filedesc *fdp;
684         struct file *fp;
685         struct stat ub;
686         struct ostat oub;
687         int error;
688
689         KKASSERT(p);
690         fdp = p->p_fd;
691         if ((unsigned)uap->fd >= fdp->fd_nfiles ||
692             (fp = fdp->fd_ofiles[uap->fd]) == NULL)
693                 return (EBADF);
694         fhold(fp);
695         error = fo_stat(fp, &ub, td);
696         if (error == 0) {
697                 cvtstat(&ub, &oub);
698                 error = copyout((caddr_t)&oub, (caddr_t)uap->sb, sizeof (oub));
699         }
700         fdrop(fp, td);
701         return (error);
702 }
703 #endif /* COMPAT_43 || COMPAT_SUNOS */
704
705 /*
706  * Return status information about a file descriptor.
707  */
708 /* ARGSUSED */
709 int
710 fstat(struct fstat_args *uap)
711 {
712         struct thread *td = curthread;
713         struct proc *p = td->td_proc;
714         struct filedesc *fdp = p->p_fd;
715         struct file *fp;
716         struct stat ub;
717         int error;
718
719         KKASSERT(p);
720         fdp = p->p_fd;
721         if ((unsigned)uap->fd >= fdp->fd_nfiles ||
722             (fp = fdp->fd_ofiles[uap->fd]) == NULL)
723                 return (EBADF);
724         fhold(fp);
725         error = fo_stat(fp, &ub, td);
726         if (error == 0)
727                 error = copyout((caddr_t)&ub, (caddr_t)uap->sb, sizeof (ub));
728         fdrop(fp, td);
729         return (error);
730 }
731
732 /*
733  * Return status information about a file descriptor.
734  */
735 /* ARGSUSED */
736 int
737 nfstat(struct nfstat_args *uap)
738 {
739         struct thread *td = curthread;
740         struct proc *p = td->td_proc;
741         struct filedesc *fdp;
742         struct file *fp;
743         struct stat ub;
744         struct nstat nub;
745         int error;
746
747         KKASSERT(p);
748         fdp = p->p_fd;
749         if ((unsigned)uap->fd >= fdp->fd_nfiles ||
750             (fp = fdp->fd_ofiles[uap->fd]) == NULL)
751                 return (EBADF);
752         fhold(fp);
753         error = fo_stat(fp, &ub, td);
754         if (error == 0) {
755                 cvtnstat(&ub, &nub);
756                 error = copyout((caddr_t)&nub, (caddr_t)uap->sb, sizeof (nub));
757         }
758         fdrop(fp, td);
759         return (error);
760 }
761
762 /*
763  * Return pathconf information about a file descriptor.
764  */
765 /* ARGSUSED */
766 int
767 fpathconf(struct fpathconf_args *uap)
768 {
769         struct thread *td = curthread;
770         struct proc *p = td->td_proc;
771         struct filedesc *fdp;
772         struct file *fp;
773         struct vnode *vp;
774         int error = 0;
775
776         KKASSERT(p);
777         fdp = p->p_fd;
778         if ((unsigned)uap->fd >= fdp->fd_nfiles ||
779             (fp = fdp->fd_ofiles[uap->fd]) == NULL)
780                 return (EBADF);
781
782         fhold(fp);
783
784         switch (fp->f_type) {
785         case DTYPE_PIPE:
786         case DTYPE_SOCKET:
787                 if (uap->name != _PC_PIPE_BUF) {
788                         error = EINVAL;
789                 } else {
790                         uap->sysmsg_result = PIPE_BUF;
791                         error = 0;
792                 }
793                 break;
794         case DTYPE_FIFO:
795         case DTYPE_VNODE:
796                 vp = (struct vnode *)fp->f_data;
797                 error = VOP_PATHCONF(vp, uap->name, uap->sysmsg_fds);
798                 break;
799         default:
800                 error = EOPNOTSUPP;
801                 break;
802         }
803         fdrop(fp, td);
804         return(error);
805 }
806
807 /*
808  * Allocate a file descriptor for the process.
809  */
810 static int fdexpand;
811 SYSCTL_INT(_debug, OID_AUTO, fdexpand, CTLFLAG_RD, &fdexpand, 0, "");
812
813 int
814 fdalloc(struct proc *p, int want, int *result)
815 {
816         struct filedesc *fdp = p->p_fd;
817         int i;
818         int lim, last, nfiles;
819         struct file **newofile;
820         char *newofileflags;
821
822         /*
823          * Search for a free descriptor starting at the higher
824          * of want or fd_freefile.  If that fails, consider
825          * expanding the ofile array.
826          */
827         lim = min((int)p->p_rlimit[RLIMIT_NOFILE].rlim_cur, maxfilesperproc);
828         for (;;) {
829                 last = min(fdp->fd_nfiles, lim);
830                 if ((i = want) < fdp->fd_freefile)
831                         i = fdp->fd_freefile;
832                 for (; i < last; i++) {
833                         if (fdp->fd_ofiles[i] == NULL) {
834                                 fdp->fd_ofileflags[i] = 0;
835                                 if (i > fdp->fd_lastfile)
836                                         fdp->fd_lastfile = i;
837                                 if (want <= fdp->fd_freefile)
838                                         fdp->fd_freefile = i;
839                                 *result = i;
840                                 return (0);
841                         }
842                 }
843
844                 /*
845                  * No space in current array.  Expand?
846                  */
847                 if (fdp->fd_nfiles >= lim)
848                         return (EMFILE);
849                 if (fdp->fd_nfiles < NDEXTENT)
850                         nfiles = NDEXTENT;
851                 else
852                         nfiles = 2 * fdp->fd_nfiles;
853                 MALLOC(newofile, struct file **, nfiles * OFILESIZE,
854                     M_FILEDESC, M_WAITOK);
855
856                 /*
857                  * deal with file-table extend race that might have occured
858                  * when malloc was blocked.
859                  */
860                 if (fdp->fd_nfiles >= nfiles) {
861                         FREE(newofile, M_FILEDESC);
862                         continue;
863                 }
864                 newofileflags = (char *) &newofile[nfiles];
865                 /*
866                  * Copy the existing ofile and ofileflags arrays
867                  * and zero the new portion of each array.
868                  */
869                 bcopy(fdp->fd_ofiles, newofile,
870                         (i = sizeof(struct file *) * fdp->fd_nfiles));
871                 bzero((char *)newofile + i, nfiles * sizeof(struct file *) - i);
872                 bcopy(fdp->fd_ofileflags, newofileflags,
873                         (i = sizeof(char) * fdp->fd_nfiles));
874                 bzero(newofileflags + i, nfiles * sizeof(char) - i);
875                 if (fdp->fd_nfiles > NDFILE)
876                         FREE(fdp->fd_ofiles, M_FILEDESC);
877                 fdp->fd_ofiles = newofile;
878                 fdp->fd_ofileflags = newofileflags;
879                 fdp->fd_nfiles = nfiles;
880                 fdexpand++;
881         }
882         return (0);
883 }
884
885 /*
886  * Check to see whether n user file descriptors
887  * are available to the process p.
888  */
889 int
890 fdavail(p, n)
891         struct proc *p;
892         int n;
893 {
894         struct filedesc *fdp = p->p_fd;
895         struct file **fpp;
896         int i, lim, last;
897
898         lim = min((int)p->p_rlimit[RLIMIT_NOFILE].rlim_cur, maxfilesperproc);
899         if ((i = lim - fdp->fd_nfiles) > 0 && (n -= i) <= 0)
900                 return (1);
901
902         last = min(fdp->fd_nfiles, lim);
903         fpp = &fdp->fd_ofiles[fdp->fd_freefile];
904         for (i = last - fdp->fd_freefile; --i >= 0; fpp++) {
905                 if (*fpp == NULL && --n <= 0)
906                         return (1);
907         }
908         return (0);
909 }
910
911 /*
912  * falloc:
913  *      Create a new open file structure and allocate a file decriptor
914  *      for the process that refers to it.  If p is NULL, no descriptor
915  *      is allocated and the file pointer is returned unassociated with
916  *      any process.  resultfd is only used if p is not NULL and may
917  *      separately be NULL indicating that you don't need the returned fd.
918  */
919 int
920 falloc(struct proc *p, struct file **resultfp, int *resultfd)
921 {
922         struct file *fp, *fq;
923         int error, i;
924
925         if (nfiles >= maxfiles) {
926                 tablefull("file");
927                 return (ENFILE);
928         }
929         /*
930          * Allocate a new file descriptor.
931          * If the process has file descriptor zero open, add to the list
932          * of open files at that point, otherwise put it at the front of
933          * the list of open files.
934          */
935         nfiles++;
936         MALLOC(fp, struct file *, sizeof(struct file), M_FILE, M_WAITOK);
937         bzero(fp, sizeof(struct file));
938
939         /*
940          * wait until after malloc (which may have blocked) returns before
941          * allocating the slot, else a race might have shrunk it if we had
942          * allocated it before the malloc.
943          */
944         i = -1;
945         if (p && (error = fdalloc(p, 0, &i))) {
946                 nfiles--;
947                 FREE(fp, M_FILE);
948                 return (error);
949         }
950         fp->f_count = 1;
951         fp->f_ops = &badfileops;
952         fp->f_seqcount = 1;
953         if (p) {
954                 fp->f_cred = crhold(p->p_ucred);
955                 if ((fq = p->p_fd->fd_ofiles[0]) != NULL) {
956                         LIST_INSERT_AFTER(fq, fp, f_list);
957                 } else {
958                         LIST_INSERT_HEAD(&filehead, fp, f_list);
959                 }
960                 p->p_fd->fd_ofiles[i] = fp;
961         } else {
962                 fp->f_cred = crhold(proc0.p_ucred);
963                 LIST_INSERT_HEAD(&filehead, fp, f_list);
964         }
965         if (resultfp)
966                 *resultfp = fp;
967         if (resultfd)
968                 *resultfd = i;
969         return (0);
970 }
971
972 void
973 fsetcred(struct file *fp, struct ucred *cr)
974 {
975         crhold(cr);
976         crfree(fp->f_cred);
977         fp->f_cred = cr;
978 }
979
980 /*
981  * Free a file descriptor.
982  */
983 void
984 ffree(fp)
985         struct file *fp;
986 {
987         KASSERT((fp->f_count == 0), ("ffree: fp_fcount not 0!"));
988         LIST_REMOVE(fp, f_list);
989         crfree(fp->f_cred);
990         nfiles--;
991         FREE(fp, M_FILE);
992 }
993
994 /*
995  * Build a new filedesc structure.
996  */
997 struct filedesc *
998 fdinit(p)
999         struct proc *p;
1000 {
1001         struct filedesc0 *newfdp;
1002         struct filedesc *fdp = p->p_fd;
1003
1004         MALLOC(newfdp, struct filedesc0 *, sizeof(struct filedesc0),
1005             M_FILEDESC, M_WAITOK);
1006         bzero(newfdp, sizeof(struct filedesc0));
1007         newfdp->fd_fd.fd_cdir = fdp->fd_cdir;
1008         if (newfdp->fd_fd.fd_cdir)
1009                 VREF(newfdp->fd_fd.fd_cdir);
1010         newfdp->fd_fd.fd_rdir = fdp->fd_rdir;
1011         if (newfdp->fd_fd.fd_rdir)
1012                 VREF(newfdp->fd_fd.fd_rdir);
1013         newfdp->fd_fd.fd_jdir = fdp->fd_jdir;
1014         if (newfdp->fd_fd.fd_jdir)
1015                 VREF(newfdp->fd_fd.fd_jdir);
1016
1017         /* Create the file descriptor table. */
1018         newfdp->fd_fd.fd_refcnt = 1;
1019         newfdp->fd_fd.fd_cmask = cmask;
1020         newfdp->fd_fd.fd_ofiles = newfdp->fd_dfiles;
1021         newfdp->fd_fd.fd_ofileflags = newfdp->fd_dfileflags;
1022         newfdp->fd_fd.fd_nfiles = NDFILE;
1023         newfdp->fd_fd.fd_knlistsize = -1;
1024
1025         return (&newfdp->fd_fd);
1026 }
1027
1028 /*
1029  * Share a filedesc structure.
1030  */
1031 struct filedesc *
1032 fdshare(p)
1033         struct proc *p;
1034 {
1035         p->p_fd->fd_refcnt++;
1036         return (p->p_fd);
1037 }
1038
1039 /*
1040  * Copy a filedesc structure.
1041  */
1042 struct filedesc *
1043 fdcopy(p)
1044         struct proc *p;
1045 {
1046         struct filedesc *newfdp, *fdp = p->p_fd;
1047         struct file **fpp;
1048         int i;
1049
1050         /* Certain daemons might not have file descriptors. */
1051         if (fdp == NULL)
1052                 return (NULL);
1053
1054         MALLOC(newfdp, struct filedesc *, sizeof(struct filedesc0),
1055             M_FILEDESC, M_WAITOK);
1056         bcopy(fdp, newfdp, sizeof(struct filedesc));
1057         if (newfdp->fd_cdir)
1058                 VREF(newfdp->fd_cdir);
1059         if (newfdp->fd_rdir)
1060                 VREF(newfdp->fd_rdir);
1061         if (newfdp->fd_jdir)
1062                 VREF(newfdp->fd_jdir);
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                 MALLOC(newfdp->fd_ofiles, struct file **, i * OFILESIZE,
1086                     M_FILEDESC, M_WAITOK);
1087                 newfdp->fd_ofileflags = (char *) &newfdp->fd_ofiles[i];
1088         }
1089         newfdp->fd_nfiles = i;
1090         bcopy(fdp->fd_ofiles, newfdp->fd_ofiles, i * sizeof(struct file **));
1091         bcopy(fdp->fd_ofileflags, newfdp->fd_ofileflags, i * sizeof(char));
1092
1093         /*
1094          * kq descriptors cannot be copied.
1095          */
1096         if (newfdp->fd_knlistsize != -1) {
1097                 fpp = &newfdp->fd_ofiles[newfdp->fd_lastfile];
1098                 for (i = newfdp->fd_lastfile; i >= 0; i--, fpp--) {
1099                         if (*fpp != NULL && (*fpp)->f_type == DTYPE_KQUEUE) {
1100                                 *fpp = NULL;
1101                                 if (i < newfdp->fd_freefile)
1102                                         newfdp->fd_freefile = i;
1103                         }
1104                         if (*fpp == NULL && i == newfdp->fd_lastfile && i > 0)
1105                                 newfdp->fd_lastfile--;
1106                 }
1107                 newfdp->fd_knlist = NULL;
1108                 newfdp->fd_knlistsize = -1;
1109                 newfdp->fd_knhash = NULL;
1110                 newfdp->fd_knhashmask = 0;
1111         }
1112
1113         fpp = newfdp->fd_ofiles;
1114         for (i = newfdp->fd_lastfile; i-- >= 0; fpp++) {
1115                 if (*fpp != NULL)
1116                         fhold(*fpp);
1117         }
1118         return (newfdp);
1119 }
1120
1121 /*
1122  * Release a filedesc structure.
1123  */
1124 void
1125 fdfree(struct proc *p)
1126 {
1127         struct thread *td = p->p_thread;
1128         struct filedesc *fdp = p->p_fd;
1129         struct file **fpp;
1130         int i;
1131         struct filedesc_to_leader *fdtol;
1132         struct file *fp;
1133         struct vnode *vp;
1134         struct flock lf;
1135
1136         /* Certain daemons might not have file descriptors. */
1137         if (fdp == NULL)
1138                 return;
1139
1140         /* Check for special need to clear POSIX style locks */
1141         fdtol = p->p_fdtol;
1142         if (fdtol != NULL) {
1143                 KASSERT(fdtol->fdl_refcount > 0,
1144                         ("filedesc_to_refcount botch: fdl_refcount=%d",
1145                          fdtol->fdl_refcount));
1146                 if (fdtol->fdl_refcount == 1 &&
1147                     (p->p_leader->p_flag & P_ADVLOCK) != 0) {
1148                         i = 0;
1149                         fpp = fdp->fd_ofiles;
1150                         for (i = 0, fpp = fdp->fd_ofiles;
1151                              i < fdp->fd_lastfile;
1152                              i++, fpp++) {
1153                                 if (*fpp == NULL ||
1154                                     (*fpp)->f_type != DTYPE_VNODE)
1155                                         continue;
1156                                 fp = *fpp;
1157                                 fhold(fp);
1158                                 lf.l_whence = SEEK_SET;
1159                                 lf.l_start = 0;
1160                                 lf.l_len = 0;
1161                                 lf.l_type = F_UNLCK;
1162                                 vp = (struct vnode *)fp->f_data;
1163                                 (void) VOP_ADVLOCK(vp,
1164                                                    (caddr_t)p->p_leader,
1165                                                    F_UNLCK,
1166                                                    &lf,
1167                                                    F_POSIX);
1168                                 fdrop(fp, p->p_thread);
1169                                 fpp = fdp->fd_ofiles + i;
1170                         }
1171                 }
1172         retry:
1173                 if (fdtol->fdl_refcount == 1) {
1174                         if (fdp->fd_holdleaderscount > 0 &&
1175                             (p->p_leader->p_flag & P_ADVLOCK) != 0) {
1176                                 /*
1177                                  * close() or do_dup() has cleared a reference
1178                                  * in a shared file descriptor table.
1179                                  */
1180                                 fdp->fd_holdleaderswakeup = 1;
1181                                 tsleep(&fdp->fd_holdleaderscount,
1182                                        0, "fdlhold", 0);
1183                                 goto retry;
1184                         }
1185                         if (fdtol->fdl_holdcount > 0) {
1186                                 /* 
1187                                  * Ensure that fdtol->fdl_leader
1188                                  * remains valid in closef().
1189                                  */
1190                                 fdtol->fdl_wakeup = 1;
1191                                 tsleep(fdtol, 0, "fdlhold", 0);
1192                                 goto retry;
1193                         }
1194                 }
1195                 fdtol->fdl_refcount--;
1196                 if (fdtol->fdl_refcount == 0 &&
1197                     fdtol->fdl_holdcount == 0) {
1198                         fdtol->fdl_next->fdl_prev = fdtol->fdl_prev;
1199                         fdtol->fdl_prev->fdl_next = fdtol->fdl_next;
1200                 } else
1201                         fdtol = NULL;
1202                 p->p_fdtol = NULL;
1203                 if (fdtol != NULL)
1204                         FREE(fdtol, M_FILEDESC_TO_LEADER);
1205         }
1206         if (--fdp->fd_refcnt > 0)
1207                 return;
1208         /*
1209          * we are the last reference to the structure, we can
1210          * safely assume it will not change out from under us.
1211          */
1212         fpp = fdp->fd_ofiles;
1213         for (i = fdp->fd_lastfile; i-- >= 0; fpp++) {
1214                 if (*fpp)
1215                         (void) closef(*fpp, td);
1216         }
1217         if (fdp->fd_nfiles > NDFILE)
1218                 FREE(fdp->fd_ofiles, M_FILEDESC);
1219         if (fdp->fd_cdir)
1220                 vrele(fdp->fd_cdir);
1221         if (fdp->fd_rdir)
1222                 vrele(fdp->fd_rdir);
1223         if (fdp->fd_jdir)
1224                 vrele(fdp->fd_jdir);
1225         if (fdp->fd_knlist)
1226                 FREE(fdp->fd_knlist, M_KQUEUE);
1227         if (fdp->fd_knhash)
1228                 FREE(fdp->fd_knhash, M_KQUEUE);
1229         FREE(fdp, M_FILEDESC);
1230 }
1231
1232 /*
1233  * For setugid programs, we don't want to people to use that setugidness
1234  * to generate error messages which write to a file which otherwise would
1235  * otherwise be off-limits to the process.
1236  *
1237  * This is a gross hack to plug the hole.  A better solution would involve
1238  * a special vop or other form of generalized access control mechanism.  We
1239  * go ahead and just reject all procfs file systems accesses as dangerous.
1240  *
1241  * Since setugidsafety calls this only for fd 0, 1 and 2, this check is
1242  * sufficient.  We also don't for check setugidness since we know we are.
1243  */
1244 static int
1245 is_unsafe(struct file *fp)
1246 {
1247         if (fp->f_type == DTYPE_VNODE && 
1248             ((struct vnode *)(fp->f_data))->v_tag == VT_PROCFS)
1249                 return (1);
1250         return (0);
1251 }
1252
1253 /*
1254  * Make this setguid thing safe, if at all possible.
1255  */
1256 void
1257 setugidsafety(struct proc *p)
1258 {
1259         struct thread *td = p->p_thread;
1260         struct filedesc *fdp = p->p_fd;
1261         int i;
1262
1263         /* Certain daemons might not have file descriptors. */
1264         if (fdp == NULL)
1265                 return;
1266
1267         /*
1268          * note: fdp->fd_ofiles may be reallocated out from under us while
1269          * we are blocked in a close.  Be careful!
1270          */
1271         for (i = 0; i <= fdp->fd_lastfile; i++) {
1272                 if (i > 2)
1273                         break;
1274                 if (fdp->fd_ofiles[i] && is_unsafe(fdp->fd_ofiles[i])) {
1275                         struct file *fp;
1276
1277 #if 0
1278                         if ((fdp->fd_ofileflags[i] & UF_MAPPED) != 0)
1279                                 (void) munmapfd(p, i);
1280 #endif
1281                         if (i < fdp->fd_knlistsize)
1282                                 knote_fdclose(p, i);
1283                         /*
1284                          * NULL-out descriptor prior to close to avoid
1285                          * a race while close blocks.
1286                          */
1287                         fp = fdp->fd_ofiles[i];
1288                         fdp->fd_ofiles[i] = NULL;
1289                         fdp->fd_ofileflags[i] = 0;
1290                         if (i < fdp->fd_freefile)
1291                                 fdp->fd_freefile = i;
1292                         (void) closef(fp, td);
1293                 }
1294         }
1295         while (fdp->fd_lastfile > 0 && fdp->fd_ofiles[fdp->fd_lastfile] == NULL)
1296                 fdp->fd_lastfile--;
1297 }
1298
1299 /*
1300  * Close any files on exec?
1301  */
1302 void
1303 fdcloseexec(struct proc *p)
1304 {
1305         struct thread *td = p->p_thread;
1306         struct filedesc *fdp = p->p_fd;
1307         int i;
1308
1309         /* Certain daemons might not have file descriptors. */
1310         if (fdp == NULL)
1311                 return;
1312
1313         /*
1314          * We cannot cache fd_ofiles or fd_ofileflags since operations
1315          * may block and rip them out from under us.
1316          */
1317         for (i = 0; i <= fdp->fd_lastfile; i++) {
1318                 if (fdp->fd_ofiles[i] != NULL &&
1319                     (fdp->fd_ofileflags[i] & UF_EXCLOSE)) {
1320                         struct file *fp;
1321
1322 #if 0
1323                         if (fdp->fd_ofileflags[i] & UF_MAPPED)
1324                                 (void) munmapfd(p, i);
1325 #endif
1326                         if (i < fdp->fd_knlistsize)
1327                                 knote_fdclose(p, i);
1328                         /*
1329                          * NULL-out descriptor prior to close to avoid
1330                          * a race while close blocks.
1331                          */
1332                         fp = fdp->fd_ofiles[i];
1333                         fdp->fd_ofiles[i] = NULL;
1334                         fdp->fd_ofileflags[i] = 0;
1335                         if (i < fdp->fd_freefile)
1336                                 fdp->fd_freefile = i;
1337                         (void) closef(fp, td);
1338                 }
1339         }
1340         while (fdp->fd_lastfile > 0 && fdp->fd_ofiles[fdp->fd_lastfile] == NULL)
1341                 fdp->fd_lastfile--;
1342 }
1343
1344 /*
1345  * It is unsafe for set[ug]id processes to be started with file
1346  * descriptors 0..2 closed, as these descriptors are given implicit
1347  * significance in the Standard C library.  fdcheckstd() will create a
1348  * descriptor referencing /dev/null for each of stdin, stdout, and
1349  * stderr that is not already open.
1350  */
1351 int
1352 fdcheckstd(struct proc *p)
1353 {
1354         struct thread *td = p->p_thread;
1355         struct nameidata nd;
1356         struct filedesc *fdp;
1357         struct file *fp;
1358         register_t retval;
1359         int fd, i, error, flags, devnull;
1360
1361        fdp = p->p_fd;
1362        if (fdp == NULL)
1363                return (0);
1364        devnull = -1;
1365        error = 0;
1366        for (i = 0; i < 3; i++) {
1367                if (fdp->fd_ofiles[i] != NULL)
1368                        continue;
1369                if (devnull < 0) {
1370                        error = falloc(p, &fp, &fd);
1371                        if (error != 0)
1372                                break;
1373                        NDINIT(&nd, NAMEI_LOOKUP, CNP_FOLLOW, UIO_SYSSPACE, 
1374                            "/dev/null", td);
1375                        flags = FREAD | FWRITE;
1376                        error = vn_open(&nd, flags, 0);
1377                        if (error != 0) {
1378                                fdp->fd_ofiles[i] = NULL;
1379                                fdrop(fp, td);
1380                                break;
1381                        }
1382                        NDFREE(&nd, NDF_ONLY_PNBUF);
1383                        fp->f_data = (caddr_t)nd.ni_vp;
1384                        fp->f_flag = flags;
1385                        fp->f_ops = &vnops;
1386                        fp->f_type = DTYPE_VNODE;
1387                        VOP_UNLOCK(nd.ni_vp, 0, td);
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         MALLOC(fdtol, struct filedesc_to_leader *,
1674                sizeof(struct filedesc_to_leader),
1675                M_FILEDESC_TO_LEADER,
1676                M_WAITOK);
1677         fdtol->fdl_refcount = 1;
1678         fdtol->fdl_holdcount = 0;
1679         fdtol->fdl_wakeup = 0;
1680         fdtol->fdl_leader = leader;
1681         if (old != NULL) {
1682                 fdtol->fdl_next = old->fdl_next;
1683                 fdtol->fdl_prev = old;
1684                 old->fdl_next = fdtol;
1685                 fdtol->fdl_next->fdl_prev = fdtol;
1686         } else {
1687                 fdtol->fdl_next = fdtol;
1688                 fdtol->fdl_prev = fdtol;
1689         }
1690         return fdtol;
1691 }
1692
1693 /*
1694  * Get file structures.
1695  */
1696 static int
1697 sysctl_kern_file(SYSCTL_HANDLER_ARGS)
1698 {
1699         int error;
1700         struct file *fp;
1701
1702         if (!req->oldptr) {
1703                 /*
1704                  * overestimate by 10 files
1705                  */
1706                 return (SYSCTL_OUT(req, 0, sizeof(filehead) + 
1707                                 (nfiles + 10) * sizeof(struct file)));
1708         }
1709
1710         error = SYSCTL_OUT(req, (caddr_t)&filehead, sizeof(filehead));
1711         if (error)
1712                 return (error);
1713
1714         /*
1715          * followed by an array of file structures
1716          */
1717         LIST_FOREACH(fp, &filehead, f_list) {
1718                 error = SYSCTL_OUT(req, (caddr_t)fp, sizeof (struct file));
1719                 if (error)
1720                         return (error);
1721         }
1722         return (0);
1723 }
1724
1725 SYSCTL_PROC(_kern, KERN_FILE, file, CTLTYPE_OPAQUE|CTLFLAG_RD,
1726     0, 0, sysctl_kern_file, "S,file", "Entire file table");
1727
1728 SYSCTL_INT(_kern, KERN_MAXFILESPERPROC, maxfilesperproc, CTLFLAG_RW, 
1729     &maxfilesperproc, 0, "Maximum files allowed open per process");
1730
1731 SYSCTL_INT(_kern, KERN_MAXFILES, maxfiles, CTLFLAG_RW, 
1732     &maxfiles, 0, "Maximum number of files");
1733
1734 SYSCTL_INT(_kern, OID_AUTO, openfiles, CTLFLAG_RD, 
1735         &nfiles, 0, "System-wide number of open files");
1736
1737 static void
1738 fildesc_drvinit(void *unused)
1739 {
1740         int fd;
1741
1742         for (fd = 0; fd < NUMFDESC; fd++)
1743                 make_dev(&fildesc_cdevsw, fd,
1744                     UID_BIN, GID_BIN, 0666, "fd/%d", fd);
1745         make_dev(&fildesc_cdevsw, 0, UID_ROOT, GID_WHEEL, 0666, "stdin");
1746         make_dev(&fildesc_cdevsw, 1, UID_ROOT, GID_WHEEL, 0666, "stdout");
1747         make_dev(&fildesc_cdevsw, 2, UID_ROOT, GID_WHEEL, 0666, "stderr");
1748 }
1749
1750 struct fileops badfileops = {
1751         NULL,   /* port */
1752         0,      /* autoq */
1753         badfo_readwrite,
1754         badfo_readwrite,
1755         badfo_ioctl,
1756         badfo_poll,
1757         badfo_kqfilter,
1758         badfo_stat,
1759         badfo_close
1760 };
1761
1762 static int
1763 badfo_readwrite(
1764         struct file *fp,
1765         struct uio *uio,
1766         struct ucred *cred,
1767         int flags,
1768         struct thread *td
1769 ) {
1770         return (EBADF);
1771 }
1772
1773 static int
1774 badfo_ioctl(struct file *fp, u_long com, caddr_t data, struct thread *td)
1775 {
1776         return (EBADF);
1777 }
1778
1779 static int
1780 badfo_poll(struct file *fp, int events, struct ucred *cred, struct thread *td)
1781 {
1782         return (0);
1783 }
1784
1785 static int
1786 badfo_kqfilter(struct file *fp, struct knote *kn)
1787 {
1788         return (0);
1789 }
1790
1791 static int
1792 badfo_stat(struct file *fp, struct stat *sb, struct thread *td)
1793 {
1794         return (EBADF);
1795 }
1796
1797 static int
1798 badfo_close(struct file *fp, struct thread *td)
1799 {
1800         return (EBADF);
1801 }
1802
1803 SYSINIT(fildescdev,SI_SUB_DRIVERS,SI_ORDER_MIDDLE+CDEV_MAJOR,
1804                                         fildesc_drvinit,NULL)