Cleanups for 4.4BSD-Lite

[unix-history] / usr / src / sys / kern / sys_generic.c

diff --git a/usr/src/sys/kern/sys_generic.c b/usr/src/sys/kern/sys_generic.c
index 272d399..a5db90f 100644 (file)
--- a/usr/src/sys/kern/sys_generic.c
+++ b/usr/src/sys/kern/sys_generic.c
@@ -1,187 +1,363 @@
-/*     sys_generic.c   6.7     85/02/08        */
-
-#include "param.h"
-#include "systm.h"
-#include "dir.h"
-#include "user.h"
-#include "ioctl.h"
-#include "file.h"
-#include "proc.h"
-#include "uio.h"
-#include "kernel.h"
-#include "stat.h"
+/*
+ * Copyright (c) 1982, 1986, 1989, 1993
+ *     The Regents of the University of California.  All rights reserved.
+ *
+ * %sccs.include.redist.c%
+ *
+ *     @(#)sys_generic.c       8.2 (Berkeley) %G%
+ */
+
+#include <sys/param.h>
+#include <sys/systm.h>
+#include <sys/filedesc.h>
+#include <sys/ioctl.h>
+#include <sys/file.h>
+#include <sys/proc.h>
+#include <sys/socketvar.h>
+#include <sys/uio.h>
+#include <sys/kernel.h>
+#include <sys/stat.h>
+#include <sys/malloc.h>
+#ifdef KTRACE
+#include <sys/ktrace.h>
+#endif

 
 /*
  * Read system call.
  */
 
 /*
  * Read system call.
  */

-read()
+struct read_args {
+       int     fd;
+       char    *buf;
+       u_int   nbyte;
+};
+/* ARGSUSED */
+read(p, uap, retval)
+       struct proc *p;
+       register struct read_args *uap;
+       int *retval;

 {
 {

-       register struct a {
-               int     fdes;
-               char    *cbuf;
-               unsigned count;
-       } *uap = (struct a *)u.u_ap;
+       register struct file *fp;
+       register struct filedesc *fdp = p->p_fd;

        struct uio auio;
        struct iovec aiov;
        struct uio auio;
        struct iovec aiov;

+       long cnt, error = 0;
+#ifdef KTRACE
+       struct iovec ktriov;
+#endif

 
 

-       aiov.iov_base = (caddr_t)uap->cbuf;
-       aiov.iov_len = uap->count;
+       if (((u_int)uap->fd) >= fdp->fd_nfiles ||
+           (fp = fdp->fd_ofiles[uap->fd]) == NULL ||
+           (fp->f_flag & FREAD) == 0)
+               return (EBADF);
+       aiov.iov_base = (caddr_t)uap->buf;
+       aiov.iov_len = uap->nbyte;

        auio.uio_iov = &aiov;
        auio.uio_iovcnt = 1;
        auio.uio_iov = &aiov;
        auio.uio_iovcnt = 1;

-       rwuio(&auio, UIO_READ);
+       auio.uio_resid = uap->nbyte;
+       auio.uio_rw = UIO_READ;
+       auio.uio_segflg = UIO_USERSPACE;
+       auio.uio_procp = p;
+#ifdef KTRACE
+       /*
+        * if tracing, save a copy of iovec
+        */
+       if (KTRPOINT(p, KTR_GENIO))
+               ktriov = aiov;
+#endif
+       cnt = uap->nbyte;
+       if (error = (*fp->f_ops->fo_read)(fp, &auio, fp->f_cred))
+               if (auio.uio_resid != cnt && (error == ERESTART ||
+                   error == EINTR || error == EWOULDBLOCK))
+                       error = 0;
+       cnt -= auio.uio_resid;
+#ifdef KTRACE
+       if (KTRPOINT(p, KTR_GENIO) && error == 0)
+               ktrgenio(p->p_tracep, uap->fd, UIO_READ, &ktriov, cnt, error);
+#endif
+       *retval = cnt;
+       return (error);

 }
 
 }
 

-readv()
+/*
+ * Scatter read system call.
+ */
+struct readv_args {
+       int     fdes;
+       struct  iovec *iovp;
+       u_int   iovcnt;
+};
+readv(p, uap, retval)
+       struct proc *p;
+       register struct readv_args *uap;
+       int *retval;

 {
 {

-       register struct a {
-               int     fdes;
-               struct  iovec *iovp;
-               int     iovcnt;
-       } *uap = (struct a *)u.u_ap;
+       register struct file *fp;
+       register struct filedesc *fdp = p->p_fd;

        struct uio auio;
        struct uio auio;

-       struct iovec aiov[16];          /* XXX */
+       register struct iovec *iov;
+       struct iovec *needfree;
+       struct iovec aiov[UIO_SMALLIOV];
+       long i, cnt, error = 0;
+       u_int iovlen;
+#ifdef KTRACE
+       struct iovec *ktriov = NULL;
+#endif

 
 

-       if (uap->iovcnt <= 0 || uap->iovcnt > sizeof(aiov)/sizeof(aiov[0])) {
-               u.u_error = EINVAL;
-               return;
+       if (((u_int)uap->fdes) >= fdp->fd_nfiles ||
+           (fp = fdp->fd_ofiles[uap->fdes]) == NULL ||
+           (fp->f_flag & FREAD) == 0)
+               return (EBADF);
+       /* note: can't use iovlen until iovcnt is validated */
+       iovlen = uap->iovcnt * sizeof (struct iovec);
+       if (uap->iovcnt > UIO_SMALLIOV) {
+               if (uap->iovcnt > UIO_MAXIOV)
+                       return (EINVAL);
+               MALLOC(iov, struct iovec *, iovlen, M_IOV, M_WAITOK);
+               needfree = iov;
+       } else {
+               iov = aiov;
+               needfree = NULL;

        }
        }

-       auio.uio_iov = aiov;
+       auio.uio_iov = iov;

        auio.uio_iovcnt = uap->iovcnt;
        auio.uio_iovcnt = uap->iovcnt;

-       u.u_error = copyin((caddr_t)uap->iovp, (caddr_t)aiov,
-           (unsigned)(uap->iovcnt * sizeof (struct iovec)));
-       if (u.u_error)
-               return;
-       rwuio(&auio, UIO_READ);
+       auio.uio_rw = UIO_READ;
+       auio.uio_segflg = UIO_USERSPACE;
+       auio.uio_procp = p;
+       if (error = copyin((caddr_t)uap->iovp, (caddr_t)iov, iovlen))
+               goto done;
+       auio.uio_resid = 0;
+       for (i = 0; i < uap->iovcnt; i++) {
+               if (iov->iov_len < 0) {
+                       error = EINVAL;
+                       goto done;
+               }
+               auio.uio_resid += iov->iov_len;
+               if (auio.uio_resid < 0) {
+                       error = EINVAL;
+                       goto done;
+               }
+               iov++;
+       }
+#ifdef KTRACE
+       /*
+        * if tracing, save a copy of iovec
+        */
+       if (KTRPOINT(p, KTR_GENIO))  {
+               MALLOC(ktriov, struct iovec *, iovlen, M_TEMP, M_WAITOK);
+               bcopy((caddr_t)auio.uio_iov, (caddr_t)ktriov, iovlen);
+       }
+#endif
+       cnt = auio.uio_resid;
+       if (error = (*fp->f_ops->fo_read)(fp, &auio, fp->f_cred))
+               if (auio.uio_resid != cnt && (error == ERESTART ||
+                   error == EINTR || error == EWOULDBLOCK))
+                       error = 0;
+       cnt -= auio.uio_resid;
+#ifdef KTRACE
+       if (ktriov != NULL) {
+               if (error == 0)
+                       ktrgenio(p->p_tracep, uap->fdes, UIO_READ, ktriov,
+                           cnt, error);
+               FREE(ktriov, M_TEMP);
+       }
+#endif
+       *retval = cnt;
+done:
+       if (needfree)
+               FREE(needfree, M_IOV);
+       return (error);

 }
 
 /*
  * Write system call
  */
 }
 
 /*
  * Write system call
  */

-write()
+struct write_args {
+       int     fd;
+       char    *buf;
+       u_int   nbyte;
+};
+write(p, uap, retval)
+       struct proc *p;
+       register struct write_args *uap;
+       int *retval;

 {
 {

-       register struct a {
-               int     fdes;
-               char    *cbuf;
-               int     count;
-       } *uap = (struct a *)u.u_ap;
+       register struct file *fp;
+       register struct filedesc *fdp = p->p_fd;

        struct uio auio;
        struct iovec aiov;
        struct uio auio;
        struct iovec aiov;

+       long cnt, error = 0;
+#ifdef KTRACE
+       struct iovec ktriov;
+#endif

 
 

+       if (((u_int)uap->fd) >= fdp->fd_nfiles ||
+           (fp = fdp->fd_ofiles[uap->fd]) == NULL ||
+           (fp->f_flag & FWRITE) == 0)
+               return (EBADF);
+       aiov.iov_base = (caddr_t)uap->buf;
+       aiov.iov_len = uap->nbyte;

        auio.uio_iov = &aiov;
        auio.uio_iovcnt = 1;
        auio.uio_iov = &aiov;
        auio.uio_iovcnt = 1;

-       aiov.iov_base = uap->cbuf;
-       aiov.iov_len = uap->count;
-       rwuio(&auio, UIO_WRITE);
-}
-
-writev()
-{
-       register struct a {
-               int     fdes;
-               struct  iovec *iovp;
-               int     iovcnt;
-       } *uap = (struct a *)u.u_ap;
-       struct uio auio;
-       struct iovec aiov[16];          /* XXX */
-
-       if (uap->iovcnt <= 0 || uap->iovcnt > sizeof(aiov)/sizeof(aiov[0])) {
-               u.u_error = EINVAL;
-               return;
+       auio.uio_resid = uap->nbyte;
+       auio.uio_rw = UIO_WRITE;
+       auio.uio_segflg = UIO_USERSPACE;
+       auio.uio_procp = p;
+#ifdef KTRACE
+       /*
+        * if tracing, save a copy of iovec
+        */
+       if (KTRPOINT(p, KTR_GENIO))
+               ktriov = aiov;
+#endif
+       cnt = uap->nbyte;
+       if (error = (*fp->f_ops->fo_write)(fp, &auio, fp->f_cred)) {
+               if (auio.uio_resid != cnt && (error == ERESTART ||
+                   error == EINTR || error == EWOULDBLOCK))
+                       error = 0;
+               if (error == EPIPE)
+                       psignal(p, SIGPIPE);

        }
        }

-       auio.uio_iov = aiov;
-       auio.uio_iovcnt = uap->iovcnt;
-       u.u_error = copyin((caddr_t)uap->iovp, (caddr_t)aiov,
-           (unsigned)(uap->iovcnt * sizeof (struct iovec)));
-       if (u.u_error)
-               return;
-       rwuio(&auio, UIO_WRITE);
+       cnt -= auio.uio_resid;
+#ifdef KTRACE
+       if (KTRPOINT(p, KTR_GENIO) && error == 0)
+               ktrgenio(p->p_tracep, uap->fd, UIO_WRITE,
+                   &ktriov, cnt, error);
+#endif
+       *retval = cnt;
+       return (error);

 }
 
 }
 

-rwuio(uio, rw)
-       register struct uio *uio;
-       enum uio_rw rw;
+/*
+ * Gather write system call
+ */
+struct writev_args {
+       int     fd;
+       struct  iovec *iovp;
+       u_int   iovcnt;
+};
+writev(p, uap, retval)
+       struct proc *p;
+       register struct writev_args *uap;
+       int *retval;

 {
 {

-       struct a {
-               int     fdes;
-       };

        register struct file *fp;
        register struct file *fp;

+       register struct filedesc *fdp = p->p_fd;
+       struct uio auio;

        register struct iovec *iov;
        register struct iovec *iov;

-       int i, count;
+       struct iovec *needfree;
+       struct iovec aiov[UIO_SMALLIOV];
+       long i, cnt, error = 0;
+       u_int iovlen;
+#ifdef KTRACE
+       struct iovec *ktriov = NULL;
+#endif

 
 

-       GETF(fp, ((struct a *)u.u_ap)->fdes);
-       if ((fp->f_flag&(rw==UIO_READ ? FREAD : FWRITE)) == 0) {
-               u.u_error = EBADF;
-               return;
+       if (((u_int)uap->fd) >= fdp->fd_nfiles ||
+           (fp = fdp->fd_ofiles[uap->fd]) == NULL ||
+           (fp->f_flag & FWRITE) == 0)
+               return (EBADF);
+       /* note: can't use iovlen until iovcnt is validated */
+       iovlen = uap->iovcnt * sizeof (struct iovec);
+       if (uap->iovcnt > UIO_SMALLIOV) {
+               if (uap->iovcnt > UIO_MAXIOV)
+                       return (EINVAL);
+               MALLOC(iov, struct iovec *, iovlen, M_IOV, M_WAITOK);
+               needfree = iov;
+       } else {
+               iov = aiov;
+               needfree = NULL;

        }
        }

-       uio->uio_resid = 0;
-       uio->uio_segflg = UIO_USERSPACE;
-       iov = uio->uio_iov;
-       for (i = 0; i < uio->uio_iovcnt; i++) {
+       auio.uio_iov = iov;
+       auio.uio_iovcnt = uap->iovcnt;
+       auio.uio_rw = UIO_WRITE;
+       auio.uio_segflg = UIO_USERSPACE;
+       auio.uio_procp = p;
+       if (error = copyin((caddr_t)uap->iovp, (caddr_t)iov, iovlen))
+               goto done;
+       auio.uio_resid = 0;
+       for (i = 0; i < uap->iovcnt; i++) {

                if (iov->iov_len < 0) {
                if (iov->iov_len < 0) {

-                       u.u_error = EINVAL;
-                       return;
+                       error = EINVAL;
+                       goto done;

                }
                }

-               uio->uio_resid += iov->iov_len;
-               if (uio->uio_resid < 0) {
-                       u.u_error = EINVAL;
-                       return;
+               auio.uio_resid += iov->iov_len;
+               if (auio.uio_resid < 0) {
+                       error = EINVAL;
+                       goto done;

                }
                iov++;
        }
                }
                iov++;
        }

-       count = uio->uio_resid;
-       uio->uio_offset = fp->f_offset;
-       if ((u.u_procp->p_flag&SOUSIG) == 0 && setjmp(&u.u_qsave)) {
-               if (uio->uio_resid == count)
-                       u.u_eosys = RESTARTSYS;
-       } else
-               u.u_error = (*fp->f_ops->fo_rw)(fp, rw, uio);
-       u.u_r.r_val1 = count - uio->uio_resid;
-       fp->f_offset += u.u_r.r_val1;
+#ifdef KTRACE
+       /*
+        * if tracing, save a copy of iovec
+        */
+       if (KTRPOINT(p, KTR_GENIO))  {
+               MALLOC(ktriov, struct iovec *, iovlen, M_TEMP, M_WAITOK);
+               bcopy((caddr_t)auio.uio_iov, (caddr_t)ktriov, iovlen);
+       }
+#endif
+       cnt = auio.uio_resid;
+       if (error = (*fp->f_ops->fo_write)(fp, &auio, fp->f_cred)) {
+               if (auio.uio_resid != cnt && (error == ERESTART ||
+                   error == EINTR || error == EWOULDBLOCK))
+                       error = 0;
+               if (error == EPIPE)
+                       psignal(p, SIGPIPE);
+       }
+       cnt -= auio.uio_resid;
+#ifdef KTRACE
+       if (ktriov != NULL) {
+               if (error == 0)
+                       ktrgenio(p->p_tracep, uap->fd, UIO_WRITE,
+                               ktriov, cnt, error);
+               FREE(ktriov, M_TEMP);
+       }
+#endif
+       *retval = cnt;
+done:
+       if (needfree)
+               FREE(needfree, M_IOV);
+       return (error);

 }
 
 /*
  * Ioctl system call
  */
 }
 
 /*
  * Ioctl system call
  */

-ioctl()
+struct ioctl_args {
+       int     fd;
+       int     com;
+       caddr_t data;
+};
+/* ARGSUSED */
+ioctl(p, uap, retval)
+       struct proc *p;
+       register struct ioctl_args *uap;
+       int *retval;

 {
        register struct file *fp;
 {
        register struct file *fp;

-       struct a {
-               int     fdes;
-               int     cmd;
-               caddr_t cmarg;
-       } *uap;
-       register int com;
+       register struct filedesc *fdp = p->p_fd;
+       register int com, error;

        register u_int size;
        register u_int size;

-       char data[IOCPARM_MASK+1];
-
-       uap = (struct a *)u.u_ap;
-       GETF(fp, uap->fdes);
-       if ((fp->f_flag & (FREAD|FWRITE)) == 0) {
-               u.u_error = EBADF;
-               return;
-       }
-       com = uap->cmd;
+       caddr_t memp = 0;
+#define STK_PARAMS     128
+       char stkbuf[STK_PARAMS];
+       caddr_t data = stkbuf;
+       int tmp;
+
+       if ((u_int)uap->fd >= fdp->fd_nfiles ||
+           (fp = fdp->fd_ofiles[uap->fd]) == NULL)
+               return (EBADF);
+       if ((fp->f_flag & (FREAD|FWRITE)) == 0)
+               return (EBADF);
+       com = uap->com;

 
 

-#if defined(vax) && defined(COMPAT)
-       /*
-        * Map old style ioctl's into new for the
-        * sake of backwards compatibility (sigh).
-        */
-       if ((com&~0xffff) == 0) {
-               com = mapioctl(com);
-               if (com == 0) {
-                       u.u_error = EINVAL;
-                       return;
-               }
-       }
-#endif

        if (com == FIOCLEX) {
        if (com == FIOCLEX) {

-               u.u_pofile[uap->fdes] |= UF_EXCLOSE;
-               return;
+               fdp->fd_ofileflags[uap->fd] |= UF_EXCLOSE;
+               return (0);

        }
        if (com == FIONCLEX) {
        }
        if (com == FIONCLEX) {

-               u.u_pofile[uap->fdes] &= ~UF_EXCLOSE;
-               return;
+               fdp->fd_ofileflags[uap->fd] &= ~UF_EXCLOSE;
+               return (0);

        }
 
        /*
        }
 
        /*


@@ -189,117 +365,158 @@ ioctl()
         * amount of data to be copied to/from the
         * user's address space.
         */
         * amount of data to be copied to/from the
         * user's address space.
         */

-       size = (com &~ (IOC_INOUT|IOC_VOID)) >> 16;
-       if (size > sizeof (data)) {
-               u.u_error = EFAULT;
-               return;
+       size = IOCPARM_LEN(com);
+       if (size > IOCPARM_MAX)
+               return (ENOTTY);
+       if (size > sizeof (stkbuf)) {
+               memp = (caddr_t)malloc((u_long)size, M_IOCTLOPS, M_WAITOK);
+               data = memp;

        }
        if (com&IOC_IN) {
                if (size) {
        }
        if (com&IOC_IN) {
                if (size) {

-                       u.u_error =
-                           copyin(uap->cmarg, (caddr_t)data, (u_int)size);
-                       if (u.u_error)
-                               return;
+                       error = copyin(uap->data, data, (u_int)size);
+                       if (error) {
+                               if (memp)
+                                       free(memp, M_IOCTLOPS);
+                               return (error);
+                       }

                } else
                } else

-                       *(caddr_t *)data = uap->cmarg;
+                       *(caddr_t *)data = uap->data;

        } else if ((com&IOC_OUT) && size)
                /*
        } else if ((com&IOC_OUT) && size)
                /*

-                * Zero the buffer on the stack so the user
-                * always gets back something deterministic.
+                * Zero the buffer so the user always
+                * gets back something deterministic.

                 */
                 */

-               bzero((caddr_t)data, size);
+               bzero(data, size);

        else if (com&IOC_VOID)
        else if (com&IOC_VOID)

-               *(caddr_t *)data = uap->cmarg;
+               *(caddr_t *)data = uap->data;

 
        switch (com) {
 
        case FIONBIO:
 
        switch (com) {
 
        case FIONBIO:

-               u.u_error = fset(fp, FNDELAY, *(int *)data);
-               return;
+               if (tmp = *(int *)data)
+                       fp->f_flag |= FNONBLOCK;
+               else
+                       fp->f_flag &= ~FNONBLOCK;
+               error = (*fp->f_ops->fo_ioctl)(fp, FIONBIO, (caddr_t)&tmp, p);
+               break;

 
        case FIOASYNC:
 
        case FIOASYNC:

-               u.u_error = fset(fp, FASYNC, *(int *)data);
-               return;
+               if (tmp = *(int *)data)
+                       fp->f_flag |= FASYNC;
+               else
+                       fp->f_flag &= ~FASYNC;
+               error = (*fp->f_ops->fo_ioctl)(fp, FIOASYNC, (caddr_t)&tmp, p);
+               break;

 
        case FIOSETOWN:
 
        case FIOSETOWN:

-               u.u_error = fsetown(fp, *(int *)data);
-               return;
+               tmp = *(int *)data;
+               if (fp->f_type == DTYPE_SOCKET) {
+                       ((struct socket *)fp->f_data)->so_pgid = tmp;
+                       error = 0;
+                       break;
+               }
+               if (tmp <= 0) {
+                       tmp = -tmp;
+               } else {
+                       struct proc *p1 = pfind(tmp);
+                       if (p1 == 0) {
+                               error = ESRCH;
+                               break;
+                       }
+                       tmp = p1->p_pgrp->pg_id;
+               }
+               error = (*fp->f_ops->fo_ioctl)
+                       (fp, (int)TIOCSPGRP, (caddr_t)&tmp, p);
+               break;

 
        case FIOGETOWN:
 
        case FIOGETOWN:

-               u.u_error = fgetown(fp, (int *)data);
-               return;
+               if (fp->f_type == DTYPE_SOCKET) {
+                       error = 0;
+                       *(int *)data = ((struct socket *)fp->f_data)->so_pgid;
+                       break;
+               }
+               error = (*fp->f_ops->fo_ioctl)(fp, (int)TIOCGPGRP, data, p);
+               *(int *)data = -*(int *)data;
+               break;
+
+       default:
+               error = (*fp->f_ops->fo_ioctl)(fp, com, data, p);
+               /*
+                * Copy any data to user, size was
+                * already set and checked above.
+                */
+               if (error == 0 && (com&IOC_OUT) && size)
+                       error = copyout(data, uap->data, (u_int)size);
+               break;

        }
        }

-       u.u_error = (*fp->f_ops->fo_ioctl)(fp, com, data);
-       /*
-        * Copy any data to user, size was
-        * already set and checked above.
-        */
-       if (u.u_error == 0 && (com&IOC_OUT) && size)
-               u.u_error = copyout(data, uap->cmarg, (u_int)size);
+       if (memp)
+               free(memp, M_IOCTLOPS);
+       return (error);

 }
 
 }
 

-int    unselect();
-int    nselcoll;
-
-/*
- * Select uses bit masks of file descriptors in ints.
- * These macros manipulate such bit fields (the filesystem macros use chars).
- */
-#define NBI            (sizeof(int) * NBBY)            /* bits per int */
-#define        NI              howmany(NOFILE, NBI)
-#define        tbit(p, n)      ((p)[(n)/NBI] & (1 << ((n) % NBI)))
-#define        sbit(p, n)      ((p)[(n)/NBI] |= (1 << ((n) % NBI)))
-#define        cbit(p, n)      ((p)[(n)/NBI] &= ~(1 << ((n) % NBI)))
+int    selwait, nselcoll;

 
 /*
  * Select system call.
  */
 
 /*
  * Select system call.
  */

-select()
+struct select_args {
+       u_int   nd;
+       fd_set  *in, *ou, *ex;
+       struct  timeval *tv;
+};
+select(p, uap, retval)
+       register struct proc *p;
+       register struct select_args *uap;
+       int *retval;

 {
 {

-       register struct uap  {
-               int     nd;
-               int     *in, *ou, *ex;
-               struct  timeval *tv;
-       } *uap = (struct uap *)u.u_ap;
-       int ibits[3][NI], obits[3][NI];
+       fd_set ibits[3], obits[3];

        struct timeval atv;
        struct timeval atv;

-       int s, ncoll, ni;
-       label_t lqsave;
+       int s, ncoll, error = 0, timo;
+       u_int ni;

 
 

-       bzero(ibits, sizeof(ibits));
-       bzero(obits, sizeof(obits));
-       if (uap->nd > NOFILE)
-               uap->nd = NOFILE;       /* forgiving, if slightly wrong */
-       ni = howmany(uap->nd, NBI);
+       bzero((caddr_t)ibits, sizeof(ibits));
+       bzero((caddr_t)obits, sizeof(obits));
+       if (uap->nd > FD_SETSIZE)
+               return (EINVAL);
+       if (uap->nd > p->p_fd->fd_nfiles)
+               uap->nd = p->p_fd->fd_nfiles;   /* forgiving; slightly wrong */
+       ni = howmany(uap->nd, NFDBITS) * sizeof(fd_mask);

 
 #define        getbits(name, x) \
 
 #define        getbits(name, x) \

-       if (uap->name) { \
-               u.u_error = copyin((caddr_t)uap->name, (caddr_t)ibits[x], \
-                   ni * sizeof(int)); \
-               if (u.u_error) \
-                       goto done; \
-       }
+       if (uap->name && \
+           (error = copyin((caddr_t)uap->name, (caddr_t)&ibits[x], ni))) \
+               goto done;

        getbits(in, 0);
        getbits(ou, 1);
        getbits(ex, 2);
 #undef getbits
 
        if (uap->tv) {
        getbits(in, 0);
        getbits(ou, 1);
        getbits(ex, 2);
 #undef getbits
 
        if (uap->tv) {

-               u.u_error = copyin((caddr_t)uap->tv, (caddr_t)&atv,
+               error = copyin((caddr_t)uap->tv, (caddr_t)&atv,

                        sizeof (atv));
                        sizeof (atv));

-               if (u.u_error)
+               if (error)

                        goto done;
                if (itimerfix(&atv)) {
                        goto done;
                if (itimerfix(&atv)) {

-                       u.u_error = EINVAL;
+                       error = EINVAL;

                        goto done;
                }
                        goto done;
                }

-               s = splhigh(); timevaladd(&atv, &time); splx(s);
-       }
+               s = splclock();
+               timevaladd(&atv, (struct timeval *)&time);
+               timo = hzto(&atv);
+               /*
+                * Avoid inadvertently sleeping forever.
+                */
+               if (timo == 0)
+                       timo = 1;
+               splx(s);
+       } else
+               timo = 0;

 retry:
        ncoll = nselcoll;
 retry:
        ncoll = nselcoll;

-       u.u_procp->p_flag |= SSEL;
-       u.u_r.r_val1 = selscan(ibits, obits, uap->nd);
-       if (u.u_error || u.u_r.r_val1)
+       p->p_flag |= SSEL;
+       error = selscan(p, ibits, obits, uap->nd, retval);
+       if (error || *retval)

                goto done;
        s = splhigh();
        /* this should be timercmp(&time, &atv, >=) */
                goto done;
        s = splhigh();
        /* this should be timercmp(&time, &atv, >=) */


@@ -308,120 +525,120 @@ retry:
                splx(s);
                goto done;
        }
                splx(s);
                goto done;
        }

-       if ((u.u_procp->p_flag & SSEL) == 0 || nselcoll != ncoll) {
-               u.u_procp->p_flag &= ~SSEL;
+       if ((p->p_flag & SSEL) == 0 || nselcoll != ncoll) {

                splx(s);
                goto retry;
        }
                splx(s);
                goto retry;
        }

-       u.u_procp->p_flag &= ~SSEL;
-       if (uap->tv) {
-               lqsave = u.u_qsave;
-               if (setjmp(&u.u_qsave)) {
-                       untimeout(unselect, (caddr_t)u.u_procp);
-                       u.u_error = EINTR;
-                       splx(s);
-                       goto done;
-               }
-               timeout(unselect, (caddr_t)u.u_procp, hzto(&atv));
-       }
-       sleep((caddr_t)&selwait, PZERO+1);
-       if (uap->tv) {
-               u.u_qsave = lqsave;
-               untimeout(unselect, (caddr_t)u.u_procp);
-       }
+       p->p_flag &= ~SSEL;
+       error = tsleep((caddr_t)&selwait, PSOCK | PCATCH, "select", timo);

        splx(s);
        splx(s);

-       goto retry;
+       if (error == 0)
+               goto retry;

 done:
 done:

+       p->p_flag &= ~SSEL;
+       /* select is not restarted after signals... */
+       if (error == ERESTART)
+               error = EINTR;
+       if (error == EWOULDBLOCK)
+               error = 0;

 #define        putbits(name, x) \
 #define        putbits(name, x) \

-       if (uap->name) { \
-               int error = copyout((caddr_t)obits[x], (caddr_t)uap->name, \
-                   ni * sizeof(int)); \
-               if (error) \
-                       u.u_error = error; \
-       }
-       putbits(in, 0);
-       putbits(ou, 1);
-       putbits(ex, 2);
+       if (uap->name && \
+           (error2 = copyout((caddr_t)&obits[x], (caddr_t)uap->name, ni))) \
+               error = error2;
+       if (error == 0) {
+               int error2;
+
+               putbits(in, 0);
+               putbits(ou, 1);
+               putbits(ex, 2);

 #undef putbits
 #undef putbits

-}
-
-unselect(p)
-       register struct proc *p;
-{
-       register int s = splhigh();
-
-       switch (p->p_stat) {
-
-       case SSLEEP:
-               setrun(p);
-               break;
-
-       case SSTOP:
-               unsleep(p);
-               break;

        }
        }

-       splx(s);
+       return (error);

 }
 
 }
 

-selscan(ibits, obits, nfd)
-       int (*ibits)[NI], (*obits)[NI];
+selscan(p, ibits, obits, nfd, retval)
+       struct proc *p;
+       fd_set *ibits, *obits;
+       int nfd, *retval;

 {
 {

-       register int which, bits, i, j;
-       int flag;
+       register struct filedesc *fdp = p->p_fd;
+       register int msk, i, j, fd;
+       register fd_mask bits;

        struct file *fp;
        int n = 0;
        struct file *fp;
        int n = 0;

+       static int flag[3] = { FREAD, FWRITE, 0 };

 
 

-       for (which = 0; which < 3; which++) {
-               switch (which) {
-
-               case 0:
-                       flag = FREAD; break;
-
-               case 1:
-                       flag = FWRITE; break;
-
-               case 2:
-                       flag = 0; break;
-               }
-               for (i = 0; i < nfd; i += NBI) {
-                       bits = ibits[which][i/NBI];
-                       while ((j = ffs(bits)) && i + --j < nfd) {
+       for (msk = 0; msk < 3; msk++) {
+               for (i = 0; i < nfd; i += NFDBITS) {
+                       bits = ibits[msk].fds_bits[i/NFDBITS];
+                       while ((j = ffs(bits)) && (fd = i + --j) < nfd) {

                                bits &= ~(1 << j);
                                bits &= ~(1 << j);

-                               fp = u.u_ofile[i + j];
-                               if (fp == NULL) {
-                                       u.u_error = EBADF;
-                                       break;
-                               }
-                               if ((*fp->f_ops->fo_select)(fp, flag)) {
-                                       sbit(obits[which], i + j);
+                               fp = fdp->fd_ofiles[fd];
+                               if (fp == NULL)
+                                       return (EBADF);
+                               if ((*fp->f_ops->fo_select)(fp, flag[msk], p)) {
+                                       FD_SET(fd, &obits[msk]);

                                        n++;
                                }
                        }
                }
        }
                                        n++;
                                }
                        }
                }
        }

-       return (n);
+       *retval = n;
+       return (0);

 }
 
 /*ARGSUSED*/
 }
 
 /*ARGSUSED*/

-seltrue(dev, flag)
+seltrue(dev, flag, p)

        dev_t dev;
        int flag;
        dev_t dev;
        int flag;

+       struct proc *p;

 {
 
        return (1);
 }
 
 {
 
        return (1);
 }
 

-selwakeup(p, coll)
-       register struct proc *p;
-       int coll;
+/*
+ * Record a select request.
+ */
+void
+selrecord(selector, sip)
+       struct proc *selector;
+       struct selinfo *sip;

 {
 {

+       struct proc *p;
+       pid_t mypid;

 
 

-       if (coll) {
+       mypid = selector->p_pid;
+       if (sip->si_pid == mypid)
+               return;
+       if (sip->si_pid && (p = pfind(sip->si_pid)) &&
+           p->p_wchan == (caddr_t)&selwait)
+               sip->si_flags |= SI_COLL;
+       else
+               sip->si_pid = mypid;
+}
+
+/*
+ * Do a wakeup when a selectable event occurs.
+ */
+void
+selwakeup(sip)
+       register struct selinfo *sip;
+{
+       register struct proc *p;
+       int s;
+
+       if (sip->si_pid == 0)
+               return;
+       if (sip->si_flags & SI_COLL) {

                nselcoll++;
                nselcoll++;

+               sip->si_flags &= ~SI_COLL;

                wakeup((caddr_t)&selwait);
        }
                wakeup((caddr_t)&selwait);
        }

-       if (p) {
-               int s = splhigh();
+       p = pfind(sip->si_pid);
+       sip->si_pid = 0;
+       if (p != NULL) {
+               s = splhigh();

                if (p->p_wchan == (caddr_t)&selwait) {
                        if (p->p_stat == SSLEEP)
                                setrun(p);
                if (p->p_wchan == (caddr_t)&selwait) {
                        if (p->p_stat == SSLEEP)
                                setrun(p);