use rdwri()

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

diff --git a/usr/src/sys/kern/uipc_socket.c b/usr/src/sys/kern/uipc_socket.c
index 93b9a71..1193d69 100644 (file)
--- a/usr/src/sys/kern/uipc_socket.c
+++ b/usr/src/sys/kern/uipc_socket.c
@@ -1,4 +1,4 @@
-/*     uipc_socket.c   4.23    82/01/13        */
+/*     uipc_socket.c   4.47    82/08/14        */

 
 #include "../h/param.h"
 #include "../h/systm.h"
 
 #include "../h/param.h"
 #include "../h/systm.h"


@@ -16,6 +16,8 @@
 #include "../h/ioctl.h"
 #include "../net/in.h"
 #include "../net/in_systm.h"
 #include "../h/ioctl.h"
 #include "../net/in.h"
 #include "../net/in_systm.h"

+#include "../net/route.h"
+#include "../h/uio.h"

 
 /*
  * Socket support routines.
 
 /*
  * Socket support routines.


@@ -37,7 +39,6 @@ socreate(aso, type, asp, asa, options)
        register struct socket *so;
        struct mbuf *m;
        int pf, proto, error;
        register struct socket *so;
        struct mbuf *m;
        int pf, proto, error;

-COUNT(SOCREATE);

 
        /*
         * Use process standard protocol/protocol family if none
 
        /*
         * Use process standard protocol/protocol family if none


@@ -70,6 +71,14 @@ COUNT(SOCREATE);
                return (ENOBUFS);
        so = mtod(m, struct socket *);
        so->so_options = options;
                return (ENOBUFS);
        so = mtod(m, struct socket *);
        so->so_options = options;

+       if (options & SO_ACCEPTCONN) {
+               so->so_q = so;
+               so->so_q0 = so;
+               so->so_qlimit = (so->so_options & SO_NEWFDONCONN) ? 5 : 1;
+       }
+       so->so_state = 0;
+       if (u.u_uid == 0)
+               so->so_state = SS_PRIV;

 
        /*
         * Attach protocol to socket, initializing
 
        /*
         * Attach protocol to socket, initializing


@@ -78,7 +87,8 @@ COUNT(SOCREATE);
        so->so_proto = prp;
        error = (*prp->pr_usrreq)(so, PRU_ATTACH, 0, asa);
        if (error) {
        so->so_proto = prp;
        error = (*prp->pr_usrreq)(so, PRU_ATTACH, 0, asa);
        if (error) {

-               (void) m_free(dtom(so));
+               so->so_state |= SS_NOFDREF;
+               sofree(so);

                return (error);
        }
        *aso = so;
                return (error);
        }
        *aso = so;


@@ -89,8 +99,12 @@ sofree(so)
        struct socket *so;
 {
 
        struct socket *so;
 {
 

-COUNT(SOFREE);
-       if (so->so_pcb || (so->so_state & SS_USERGONE) == 0)
+       if (so->so_head) {
+               if (!soqremque(so, 0) && !soqremque(so, 1))
+                       panic("sofree dq");
+               so->so_head = 0;
+       }
+       if (so->so_pcb || (so->so_state & SS_NOFDREF) == 0)

                return;
        sbrelease(&so->so_snd);
        sbrelease(&so->so_rcv);
                return;
        sbrelease(&so->so_snd);
        sbrelease(&so->so_rcv);


@@ -102,64 +116,66 @@ COUNT(SOFREE);
  * Initiate disconnect if connected.
  * Free socket when disconnect complete.
  */
  * Initiate disconnect if connected.
  * Free socket when disconnect complete.
  */

-soclose(so)
+soclose(so, exiting)

        register struct socket *so;
        register struct socket *so;

+       int exiting;

 {
        int s = splnet();               /* conservative */
 {
        int s = splnet();               /* conservative */

+       register struct socket *so2;

 
 

-COUNT(SOCLOSE);
+       if (so->so_options & SO_ACCEPTCONN) {
+               while (so->so_q0 != so)
+                       soclose(so->so_q0, 1);
+               while (so->so_q != so)
+                       soclose(so->so_q, 1);
+       }

        if (so->so_pcb == 0)
                goto discard;
        if (so->so_pcb == 0)
                goto discard;

+       if (exiting)
+               so->so_options |= SO_KEEPALIVE;

        if (so->so_state & SS_ISCONNECTED) {
                if ((so->so_state & SS_ISDISCONNECTING) == 0) {
                        u.u_error = sodisconnect(so, (struct sockaddr *)0);
                        if (u.u_error) {
        if (so->so_state & SS_ISCONNECTED) {
                if ((so->so_state & SS_ISDISCONNECTING) == 0) {
                        u.u_error = sodisconnect(so, (struct sockaddr *)0);
                        if (u.u_error) {

+                               if (exiting)
+                                       goto drop;

                                splx(s);
                                return;
                        }
                }
                if ((so->so_options & SO_DONTLINGER) == 0) {
                        if ((so->so_state & SS_ISDISCONNECTING) &&
                                splx(s);
                                return;
                        }
                }
                if ((so->so_options & SO_DONTLINGER) == 0) {
                        if ((so->so_state & SS_ISDISCONNECTING) &&

-                           (so->so_options & SO_NONBLOCKING)) {
+                           (so->so_state & SS_NBIO) &&
+                           exiting == 0) {

                                u.u_error = EINPROGRESS;
                                splx(s);
                                return;
                        }
                                u.u_error = EINPROGRESS;
                                splx(s);
                                return;
                        }

-                       /* should use tsleep here */
+                       /* should use tsleep here, for at most linger */

                        while (so->so_state & SS_ISCONNECTED)
                                sleep((caddr_t)&so->so_timeo, PZERO+1);
                }
        }
                        while (so->so_state & SS_ISCONNECTED)
                                sleep((caddr_t)&so->so_timeo, PZERO+1);
                }
        }

-       u.u_error = (*so->so_proto->pr_usrreq)(so, PRU_DETACH, 0, 0);
+drop:
+       if (so->so_pcb) {
+               u.u_error = (*so->so_proto->pr_usrreq)(so, PRU_DETACH, 0, 0);
+               if (exiting == 0 && u.u_error) {
+                       splx(s);
+                       return;
+               }
+       }

 discard:
 discard:

-       so->so_state |= SS_USERGONE;
+       so->so_state |= SS_NOFDREF;

        sofree(so);
        splx(s);
 }
 
        sofree(so);
        splx(s);
 }
 

-sosplice(pso, so)
-       struct socket *pso, *so;
-{
-
-COUNT(SOSPLICE);
-       if (pso->so_proto->pr_family != PF_UNIX) {
-               struct socket *tso;
-               tso = pso; pso = so; so = tso;
-       }
-       if (pso->so_proto->pr_family != PF_UNIX)
-               return (EOPNOTSUPP);
-       /* check types and buffer space */
-       /* merge buffers */
-       return (0);
-}
-

 /*ARGSUSED*/
 sostat(so, sb)
        struct socket *so;
        struct stat *sb;
 {
 
 /*ARGSUSED*/
 sostat(so, sb)
        struct socket *so;
        struct stat *sb;
 {
 

-COUNT(SOSTAT);

        bzero((caddr_t)sb, sizeof (*sb));               /* XXX */
        return (0);                                     /* XXX */
 }
        bzero((caddr_t)sb, sizeof (*sb));               /* XXX */
        return (0);                                     /* XXX */
 }


@@ -174,18 +190,7 @@ soaccept(so, asa)
        int s = splnet();
        int error;
 
        int s = splnet();
        int error;
 

-COUNT(SOACCEPT);
-       if ((so->so_options & SO_ACCEPTCONN) == 0) {
-               error = EINVAL;                 /* XXX */
-               goto bad;
-       }
-       if ((so->so_state & SS_CONNAWAITING) == 0) {
-               error = ENOTCONN;
-               goto bad;
-       }
-       so->so_state &= ~SS_CONNAWAITING;

        error = (*so->so_proto->pr_usrreq)(so, PRU_ACCEPT, 0, (caddr_t)asa);
        error = (*so->so_proto->pr_usrreq)(so, PRU_ACCEPT, 0, (caddr_t)asa);

-bad:

        splx(s);
        return (error);
 }
        splx(s);
        return (error);
 }


@@ -202,7 +207,6 @@ soconnect(so, asa)
        int s = splnet();
        int error;
 
        int s = splnet();
        int error;
 

-COUNT(SOCONNECT);

        if (so->so_state & (SS_ISCONNECTED|SS_ISCONNECTING)) {
                error = EISCONN;
                goto bad;
        if (so->so_state & (SS_ISCONNECTED|SS_ISCONNECTING)) {
                error = EISCONN;
                goto bad;


@@ -226,7 +230,6 @@ sodisconnect(so, asa)
        int s = splnet();
        int error;
 
        int s = splnet();
        int error;
 

-COUNT(SODISCONNECT);

        if ((so->so_state & SS_ISCONNECTED) == 0) {
                error = ENOTCONN;
                goto bad;
        if ((so->so_state & SS_ISCONNECTED) == 0) {
                error = ENOTCONN;
                goto bad;


@@ -258,21 +261,26 @@ sosend(so, asa)
        register u_int len;
        int error = 0, space, s;
 
        register u_int len;
        int error = 0, space, s;
 

-COUNT(SOSEND);
-       if (so->so_state & SS_CANTSENDMORE)
-               return (EPIPE);

        if (sosendallatonce(so) && u.u_count > so->so_snd.sb_hiwat)
                return (EMSGSIZE);
        if (sosendallatonce(so) && u.u_count > so->so_snd.sb_hiwat)
                return (EMSGSIZE);

-       if ((so->so_snd.sb_flags & SB_LOCK) && (so->so_options & SO_NONBLOCKING))
+#ifdef notdef
+       /* NEED TO PREVENT BUSY WAITING IN SELECT FOR WRITING */
+       if ((so->so_snd.sb_flags & SB_LOCK) && (so->so_state & SS_NBIO))

                return (EWOULDBLOCK);
                return (EWOULDBLOCK);

+#endif
+restart:

        sblock(&so->so_snd);
 #define        snderr(errno)   { error = errno; splx(s); goto release; }
 
        sblock(&so->so_snd);
 #define        snderr(errno)   { error = errno; splx(s); goto release; }
 

-       s = splnet();

 again:
 again:

+       s = splnet();
+       if (so->so_state & SS_CANTSENDMORE) {
+               psignal(u.u_procp, SIGPIPE);
+               snderr(EPIPE);
+       }

        if (so->so_error) {
                error = so->so_error;
        if (so->so_error) {
                error = so->so_error;

-               so->so_error = 0;
+               so->so_error = 0;                               /* ??? */

                splx(s);
                goto release;
        }
                splx(s);
                goto release;
        }


@@ -284,11 +292,11 @@ again:
        }
        if (top) {
                error = (*so->so_proto->pr_usrreq)(so, PRU_SEND, top, asa);
        }
        if (top) {
                error = (*so->so_proto->pr_usrreq)(so, PRU_SEND, top, asa);

+               top = 0;

                if (error) {
                        splx(s);
                        goto release;
                }
                if (error) {
                        splx(s);
                        goto release;
                }

-               top = 0;

                mp = ⊤
        }
        if (u.u_count == 0) {
                mp = ⊤
        }
        if (u.u_count == 0) {


@@ -296,20 +304,19 @@ again:
                goto release;
        }
        space = sbspace(&so->so_snd);
                goto release;
        }
        space = sbspace(&so->so_snd);

-       if (space == 0 || sosendallatonce(so) && space < u.u_count) {
-               if (so->so_options & SO_NONBLOCKING)
+       if (space <= 0 || sosendallatonce(so) && space < u.u_count) {
+               if (so->so_state & SS_NBIO)

                        snderr(EWOULDBLOCK);
                sbunlock(&so->so_snd);
                sbwait(&so->so_snd);
                splx(s);
                        snderr(EWOULDBLOCK);
                sbunlock(&so->so_snd);
                sbwait(&so->so_snd);
                splx(s);

-               goto again;
+               goto restart;

        }
        splx(s);
        while (u.u_count && space > 0) {
                MGET(m, 1);
                if (m == NULL) {
        }
        splx(s);
        while (u.u_count && space > 0) {
                MGET(m, 1);
                if (m == NULL) {

-                       error = ENOBUFS;
-                       m_freem(top);
+                       error = ENOBUFS;                        /* SIGPIPE? */

                        goto release;
                }
                if (u.u_count >= CLBYTES && space >= CLBYTES) {
                        goto release;
                }
                if (u.u_count >= CLBYTES && space >= CLBYTES) {


@@ -330,23 +337,26 @@ nopages:
                mp = &m->m_next;
                space = sbspace(&so->so_snd);
        }
                mp = &m->m_next;
                space = sbspace(&so->so_snd);
        }

-       s = splnet();

        goto again;
 
 release:
        sbunlock(&so->so_snd);
        goto again;
 
 release:
        sbunlock(&so->so_snd);

+       if (top)
+               m_freem(top);

        return (error);
 }
 
        return (error);
 }
 

-soreceive(so, asa)
+soreceive(so, asa, uio)

        register struct socket *so;
        struct sockaddr *asa;
        register struct socket *so;
        struct sockaddr *asa;

+       struct uio *uio;

 {
 {

+       register struct iovec *iov;

        register struct mbuf *m, *n;
        u_int len;
        register struct mbuf *m, *n;
        u_int len;

-       int eor, s, error = 0;
+       int eor, s, error = 0, resid = uio->uio_resid;
+       int cnt;

 
 

-COUNT(SORECEIVE);

 restart:
        sblock(&so->so_rcv);
        s = splnet();
 restart:
        sblock(&so->so_rcv);
        s = splnet();


@@ -366,7 +376,7 @@ restart:
                if ((so->so_state & SS_ISCONNECTED) == 0 &&
                    (so->so_proto->pr_flags & PR_CONNREQUIRED))
                        rcverr(ENOTCONN);
                if ((so->so_state & SS_ISCONNECTED) == 0 &&
                    (so->so_proto->pr_flags & PR_CONNREQUIRED))
                        rcverr(ENOTCONN);

-               if (so->so_options & SO_NONBLOCKING)
+               if (so->so_state & SS_NBIO)

                        rcverr(EWOULDBLOCK);
                sbunlock(&so->so_rcv);
                sbwait(&so->so_rcv);
                        rcverr(EWOULDBLOCK);
                sbunlock(&so->so_rcv);
                sbwait(&so->so_rcv);


@@ -390,23 +400,36 @@ restart:
        }
        eor = 0;
        do {
        }
        eor = 0;
        do {

-               len = MIN(m->m_len, u.u_count);
+               if (uio->uio_iovcnt == 0)
+                       break;
+               iov = uio->uio_iov;
+               len = iov->iov_len;
+               so->so_state &= ~SS_RCVATMARK;
+               if (so->so_oobmark && len > so->so_oobmark)
+                       len = so->so_oobmark;
+               if (len > m->m_len)
+                       len = m->m_len;
+               splx(s);
+               uiomove(mtod(m, caddr_t), len, UIO_WRITETO, uio);
+               s = splnet();

                if (len == m->m_len) {
                        eor = (int)m->m_act;
                        sbfree(&so->so_rcv, m);
                        so->so_rcv.sb_mb = m->m_next;
                if (len == m->m_len) {
                        eor = (int)m->m_act;
                        sbfree(&so->so_rcv, m);
                        so->so_rcv.sb_mb = m->m_next;

-               }
-               splx(s);
-               iomove(mtod(m, caddr_t), len, B_READ);
-               s = splnet();
-               if (len == m->m_len) {

                        MFREE(m, n);
                } else {
                        m->m_off += len;
                        m->m_len -= len;
                        so->so_rcv.sb_cc -= len;
                }
                        MFREE(m, n);
                } else {
                        m->m_off += len;
                        m->m_len -= len;
                        so->so_rcv.sb_cc -= len;
                }

-       } while ((m = so->so_rcv.sb_mb) && u.u_count && !eor);
+               if (so->so_oobmark) {
+                       so->so_oobmark -= len;
+                       if (so->so_oobmark == 0) {
+                               so->so_state |= SS_RCVATMARK;
+                               break;
+                       }
+               }
+       } while ((m = so->so_rcv.sb_mb) && !eor);

        if ((so->so_proto->pr_flags & PR_ATOMIC) && eor == 0)
                do {
                        if (m == 0)
        if ((so->so_proto->pr_flags & PR_ATOMIC) && eor == 0)
                do {
                        if (m == 0)


@@ -425,114 +448,134 @@ release:
        return (error);
 }
 
        return (error);
 }
 

+sohasoutofband(so)
+       struct socket *so;
+{
+
+       if (so->so_pgrp == 0)
+               return;
+       if (so->so_pgrp > 0)
+               gsignal(so->so_pgrp, SIGURG);
+       else {
+               struct proc *p = pfind(-so->so_pgrp);
+
+               if (p)
+                       psignal(p, SIGURG);
+       }
+}
+

 /*ARGSUSED*/
 /*ARGSUSED*/

-soioctl(so, cmd, cmdp)
+soioctl(so, cmd, data)

        register struct socket *so;
        int cmd;
        register struct socket *so;
        int cmd;

-       register caddr_t cmdp;
+       register char *data;

 {
 
 {
 

-COUNT(SOIOCTL);

        switch (cmd) {
 
        switch (cmd) {
 

-       case FIONBIO: {
-               int nbio;
-               if (copyin(cmdp, (caddr_t)&nbio, sizeof (nbio))) {
-                       u.u_error = EFAULT;
-                       return;
-               }
-               if (nbio)
-                       so->so_options |= SO_NONBLOCKING;
+       case FIONBIO:
+               if (*(int *)data)
+                       so->so_state |= SS_NBIO;

                else
                else

-                       so->so_options &= ~SO_NONBLOCKING;
+                       so->so_state &= ~SS_NBIO;

                return;
                return;

-       }

 
 

-       case FIOASYNC: {
-               int async;
-               if (copyin(cmdp, (caddr_t)&async, sizeof (async))) {
-                       u.u_error = EFAULT;
-                       return;
-               }
-               if (async)
-                       ;
+       case FIOASYNC:
+               if (*(int *)data)
+                       so->so_state |= SS_ASYNC;

                else
                else

-                       ;
+                       so->so_state &= ~SS_ASYNC;

                return;
                return;

-       }

 
 

-       case SIOCSKEEP: {
-               int keep;
-               if (copyin(cmdp, (caddr_t)&keep, sizeof (keep))) {
-                       u.u_error = EFAULT;
-                       return;
-               }
-               if (keep)
-                       so->so_options &= ~SO_NOKEEPALIVE;
+       case SIOCSKEEP:
+               if (*(int *)data)
+                       so->so_options &= ~SO_KEEPALIVE;

                else
                else

-                       so->so_options |= SO_NOKEEPALIVE;
+                       so->so_options |= SO_KEEPALIVE;

                return;
                return;

-       }

 
 

-       case SIOCGKEEP: {
-               int keep = (so->so_options & SO_NOKEEPALIVE) == 0;
-               if (copyout((caddr_t)&keep, cmdp, sizeof (keep)))
-                       u.u_error = EFAULT;
+       case SIOCGKEEP:
+               *(int *)data = (so->so_options & SO_KEEPALIVE) != 0;

                return;
                return;

-       }

 
 

-       case SIOCSLINGER: {
-               int linger;
-               if (copyin(cmdp, (caddr_t)&linger, sizeof (linger))) {
-                       u.u_error = EFAULT;
-                       return;
-               }
-               so->so_linger = linger;
+       case SIOCSLINGER:
+               so->so_linger = *(int *)data;

                if (so->so_linger)
                        so->so_options &= ~SO_DONTLINGER;
                else
                        so->so_options |= SO_DONTLINGER;
                return;
                if (so->so_linger)
                        so->so_options &= ~SO_DONTLINGER;
                else
                        so->so_options |= SO_DONTLINGER;
                return;

-       }

 
 

-       case SIOCGLINGER: {
-               int linger = so->so_linger;
-               if (copyout((caddr_t)&linger, cmdp, sizeof (linger))) {
-                       u.u_error = EFAULT;
-                       return;
-               }
-       }
+       case SIOCGLINGER:
+               *(int *)data = so->so_linger;
+               return;
+
+       case SIOCSPGRP:
+               so->so_pgrp = *(int *)data;
+               return;
+
+       case SIOCGPGRP:
+               *(int *)data = so->so_pgrp;
+               return;

 
        case SIOCDONE: {
 
        case SIOCDONE: {

-               int flags;
-               if (copyin(cmdp, (caddr_t)&flags, sizeof (flags))) {
-                       u.u_error = EFAULT;
-                       return;
-               }
+               int flags = *(int *)data;
+

                flags++;
                if (flags & FREAD) {
                        int s = splimp();
                        socantrcvmore(so);
                        sbflush(&so->so_rcv);
                flags++;
                if (flags & FREAD) {
                        int s = splimp();
                        socantrcvmore(so);
                        sbflush(&so->so_rcv);

+                       splx(s);

                }
                if (flags & FWRITE)
                }
                if (flags & FWRITE)

-                       u.u_error = (*so->so_proto->pr_usrreq)(so, PRU_DISCONNECT, (struct mbuf *)0, 0);
+                       u.u_error = (*so->so_proto->pr_usrreq)(so, PRU_SHUTDOWN, (struct mbuf *)0, 0);

                return;
        }
 
                return;
        }
 

+       case SIOCSENDOOB: {
+               char oob = *(char *)data;
+               struct mbuf *m;
+
+               m = m_get(M_DONTWAIT);
+               if (m == 0) {
+                       u.u_error = ENOBUFS;
+                       return;
+               }
+               m->m_off = MMINOFF;
+               m->m_len = sizeof (char);
+               *mtod(m, char *) = oob;
+               (*so->so_proto->pr_usrreq)(so, PRU_SENDOOB, m, 0);
+               return;

        }
        }

-       switch (so->so_type) {

 
 

-       case SOCK_STREAM:
-               break;
+       case SIOCRCVOOB: {
+               struct mbuf *m = m_get(M_DONTWAIT);
+
+               if (m == 0) {
+                       u.u_error = ENOBUFS;
+                       return;
+               }
+               m->m_off = MMINOFF; *mtod(m, caddr_t) = 0;
+               (*so->so_proto->pr_usrreq)(so, PRU_RCVOOB, m, 0);
+               *(char *)data = *mtod(m, char *);
+               (void) m_free(m);
+               return;
+       }

 
 

-       case SOCK_DGRAM:
-               break;
+       case SIOCATMARK:
+               *(int *)data = (so->so_state&SS_RCVATMARK) != 0;
+               return;

 
 

-       case SOCK_RDM:
-               break;
+       /* routing table update calls */
+       case SIOCADDRT:
+       case SIOCDELRT:
+               if (!suser())
+                       return;
+               u.u_error = rtrequest(cmd, (struct rtentry *)data);
+               return;

 
 

-       case SOCK_RAW:
-               break;
+       /* type/protocol specific ioctls */

        }
        }

+       u.u_error = EOPNOTSUPP;

 }
 }