[BACK]Return to uipc_socket.c CVS log [TXT][DIR] Up to [cvs.NetBSD.org] / src / sys / kern

Please note that diffs are not public domain; they are subject to the copyright notices on the relevant files.

Diff for /src/sys/kern/uipc_socket.c between version 1.140.6.6 and 1.141

version 1.140.6.6, 2007/12/09 19:38:27 version 1.141, 2007/08/06 11:41:52
Line 79  __KERNEL_RCSID(0, "$NetBSD$");
Line 79  __KERNEL_RCSID(0, "$NetBSD$");
 #include <sys/systm.h>  #include <sys/systm.h>
 #include <sys/proc.h>  #include <sys/proc.h>
 #include <sys/file.h>  #include <sys/file.h>
 #include <sys/filedesc.h>  
 #include <sys/malloc.h>  #include <sys/malloc.h>
 #include <sys/mbuf.h>  #include <sys/mbuf.h>
 #include <sys/domain.h>  #include <sys/domain.h>
Line 104  POOL_INIT(socket_pool, sizeof(struct soc
Line 103  POOL_INIT(socket_pool, sizeof(struct soc
 MALLOC_DEFINE(M_SOOPTS, "soopts", "socket options");  MALLOC_DEFINE(M_SOOPTS, "soopts", "socket options");
 MALLOC_DEFINE(M_SONAME, "soname", "socket name");  MALLOC_DEFINE(M_SONAME, "soname", "socket name");
   
 extern const struct fileops socketops;  
   
 extern int      somaxconn;                      /* patchable (XXX sysctl) */  extern int      somaxconn;                      /* patchable (XXX sysctl) */
 int             somaxconn = SOMAXCONN;  int             somaxconn = SOMAXCONN;
   
Line 303  sodopendfreel()
Line 300  sodopendfreel()
 {  {
         struct mbuf *m, *next;          struct mbuf *m, *next;
         size_t rv = 0;          size_t rv = 0;
           int s;
   
         KASSERT(mutex_owned(&so_pendfree_lock));          KASSERT(mutex_owned(&so_pendfree_lock));
   
Line 318  sodopendfreel()
Line 316  sodopendfreel()
                         sodoloanfree((m->m_flags & M_EXT_PAGES) ?                          sodoloanfree((m->m_flags & M_EXT_PAGES) ?
                             m->m_ext.ext_pgs : NULL, m->m_ext.ext_buf,                              m->m_ext.ext_pgs : NULL, m->m_ext.ext_buf,
                             m->m_ext.ext_size);                              m->m_ext.ext_size);
                         pool_cache_put(mb_cache, m);                          s = splvm();
                           pool_cache_put(&mbpool_cache, m);
                           splx(s);
                 }                  }
   
                 mutex_enter(&so_pendfree_lock);                  mutex_enter(&so_pendfree_lock);
Line 427  sokva_reclaim_callback(struct callback_e
Line 427  sokva_reclaim_callback(struct callback_e
         return CALLBACK_CHAIN_CONTINUE;          return CALLBACK_CHAIN_CONTINUE;
 }  }
   
 struct mbuf *  
 getsombuf(struct socket *so, int type)  
 {  
         struct mbuf *m;  
   
         m = m_get(M_WAIT, type);  
         MCLAIM(m, so->so_mowner);  
         return m;  
 }  
   
 struct mbuf *  
 m_intopt(struct socket *so, int val)  
 {  
         struct mbuf *m;  
   
         m = getsombuf(so, MT_SOOPTS);  
         m->m_len = sizeof(int);  
         *mtod(m, int *) = val;  
         return m;  
 }  
   
 void  void
 soinit(void)  soinit(void)
 {  {
   
         mutex_init(&so_pendfree_lock, MUTEX_DEFAULT, IPL_VM);          mutex_init(&so_pendfree_lock, MUTEX_DRIVER, IPL_VM);
         cv_init(&socurkva_cv, "sokva");          cv_init(&socurkva_cv, "sokva");
   
         /* Set the initial adjusted socket buffer size. */          /* Set the initial adjusted socket buffer size. */
Line 516  socreate(int dom, struct socket **aso, i
Line 495  socreate(int dom, struct socket **aso, i
         so->so_snd.sb_mowner = &prp->pr_domain->dom_mowner;          so->so_snd.sb_mowner = &prp->pr_domain->dom_mowner;
         so->so_mowner = &prp->pr_domain->dom_mowner;          so->so_mowner = &prp->pr_domain->dom_mowner;
 #endif  #endif
         selinit(&so->so_rcv.sb_sel);  
         selinit(&so->so_snd.sb_sel);  
         uid = kauth_cred_geteuid(l->l_cred);          uid = kauth_cred_geteuid(l->l_cred);
         so->so_uidinfo = uid_find(uid);          so->so_uidinfo = uid_find(uid);
         error = (*prp->pr_usrreq)(so, PRU_ATTACH, NULL,          error = (*prp->pr_usrreq)(so, PRU_ATTACH, NULL,
Line 533  socreate(int dom, struct socket **aso, i
Line 510  socreate(int dom, struct socket **aso, i
         return 0;          return 0;
 }  }
   
 /* On success, write file descriptor to fdout and return zero.  On  
  * failure, return non-zero; *fdout will be undefined.  
  */  
 int  
 fsocreate(int domain, struct socket **sop, int type, int protocol,  
     struct lwp *l, int *fdout)  
 {  
         struct filedesc *fdp;  
         struct socket   *so;  
         struct file     *fp;  
         int             fd, error;  
   
         fdp = l->l_proc->p_fd;  
         /* falloc() will use the desciptor for us */  
         if ((error = falloc(l, &fp, &fd)) != 0)  
                 return (error);  
         fp->f_flag = FREAD|FWRITE;  
         fp->f_type = DTYPE_SOCKET;  
         fp->f_ops = &socketops;  
         error = socreate(domain, &so, type, protocol, l);  
         if (error != 0) {  
                 FILE_UNUSE(fp, l);  
                 fdremove(fdp, fd);  
                 ffree(fp);  
         } else {  
                 if (sop != NULL)  
                         *sop = so;  
                 fp->f_data = so;  
                 FILE_SET_MATURE(fp);  
                 FILE_UNUSE(fp, l);  
                 *fdout = fd;  
         }  
         return error;  
 }  
   
 int  int
 sobind(struct socket *so, struct mbuf *nam, struct lwp *l)  sobind(struct socket *so, struct mbuf *nam, struct lwp *l)
 {  {
Line 623  sofree(struct socket *so)
Line 565  sofree(struct socket *so)
                     RLIM_INFINITY);                      RLIM_INFINITY);
         sbrelease(&so->so_snd, so);          sbrelease(&so->so_snd, so);
         sorflush(so);          sorflush(so);
         seldestroy(&so->so_rcv.sb_sel);  
         seldestroy(&so->so_snd.sb_sel);  
         pool_put(&socket_pool, so);          pool_put(&socket_pool, so);
 }  }
   
Line 891  sosend(struct socket *so, struct mbuf *a
Line 831  sosend(struct socket *so, struct mbuf *a
                                 if (flags & MSG_EOR)                                  if (flags & MSG_EOR)
                                         top->m_flags |= M_EOR;                                          top->m_flags |= M_EOR;
                         } else do {                          } else do {
                                 if (top == NULL) {                                  if (top == 0) {
                                         m = m_gethdr(M_WAIT, MT_DATA);                                          m = m_gethdr(M_WAIT, MT_DATA);
                                         mlen = MHLEN;                                          mlen = MHLEN;
                                         m->m_pkthdr.len = 0;                                          m->m_pkthdr.len = 0;
Line 935  sosend(struct socket *so, struct mbuf *a
Line 875  sosend(struct socket *so, struct mbuf *a
                                         if (atomic && top == 0 && len < mlen)                                          if (atomic && top == 0 && len < mlen)
                                                 MH_ALIGN(m, len);                                                  MH_ALIGN(m, len);
                                 }                                  }
                                 error = uiomove(mtod(m, void *), (int)len, uio);                                  error = uiomove(mtod(m, void *), (int)len,
                                       uio);
  have_data:   have_data:
                                 resid = uio->uio_resid;                                  resid = uio->uio_resid;
                                 m->m_len = len;                                  m->m_len = len;
                                 *mp = m;                                  *mp = m;
                                 top->m_pkthdr.len += len;                                  top->m_pkthdr.len += len;
                                 if (error != 0)                                  if (error)
                                         goto release;                                          goto release;
                                 mp = &m->m_next;                                  mp = &m->m_next;
                                 if (resid <= 0) {                                  if (resid <= 0) {
Line 970  sosend(struct socket *so, struct mbuf *a
Line 911  sosend(struct socket *so, struct mbuf *a
                         splx(s);                          splx(s);
   
                         clen = 0;                          clen = 0;
                         control = NULL;                          control = 0;
                         top = NULL;                          top = 0;
                         mp = &top;                          mp = &top;
                         if (error != 0)                          if (error)
                                 goto release;                                  goto release;
                 } while (resid && space > 0);                  } while (resid && space > 0);
         } while (resid);          } while (resid);
Line 1010  soreceive(struct socket *so, struct mbuf
Line 951  soreceive(struct socket *so, struct mbuf
 {  {
         struct lwp *l = curlwp;          struct lwp *l = curlwp;
         struct mbuf     *m, **mp;          struct mbuf     *m, **mp;
         int atomic, flags, len, error, s, offset, moff, type, orig_resid;          int             flags, len, error, s, offset, moff, type, orig_resid;
         const struct protosw    *pr;          const struct protosw    *pr;
         struct mbuf     *nextrecord;          struct mbuf     *nextrecord;
         int             mbuf_removed = 0;          int             mbuf_removed = 0;
         const struct domain *dom;  
   
         pr = so->so_proto;          pr = so->so_proto;
         atomic = pr->pr_flags & PR_ATOMIC;  
         dom = pr->pr_domain;  
         mp = mp0;          mp = mp0;
         type = 0;          type = 0;
         orig_resid = uio->uio_resid;          orig_resid = uio->uio_resid;
   
         if (paddr != NULL)          if (paddr)
                 *paddr = NULL;                  *paddr = 0;
         if (controlp != NULL)          if (controlp)
                 *controlp = NULL;                  *controlp = 0;
         if (flagsp != NULL)          if (flagsp)
                 flags = *flagsp &~ MSG_EOR;                  flags = *flagsp &~ MSG_EOR;
         else          else
                 flags = 0;                  flags = 0;
Line 1045  soreceive(struct socket *so, struct mbuf
Line 983  soreceive(struct socket *so, struct mbuf
                         error = uiomove(mtod(m, void *),                          error = uiomove(mtod(m, void *),
                             (int) min(uio->uio_resid, m->m_len), uio);                              (int) min(uio->uio_resid, m->m_len), uio);
                         m = m_free(m);                          m = m_free(m);
                 } while (uio->uio_resid > 0 && error == 0 && m);                  } while (uio->uio_resid && error == 0 && m);
  bad:   bad:
                 if (m != NULL)                  if (m)
                         m_freem(m);                          m_freem(m);
                 return error;                  return (error);
         }          }
         if (mp != NULL)          if (mp)
                 *mp = NULL;                  *mp = NULL;
         if (so->so_state & SS_ISCONFIRMING && uio->uio_resid)          if (so->so_state & SS_ISCONFIRMING && uio->uio_resid)
                 (*pr->pr_usrreq)(so, PRU_RCVD, NULL, NULL, NULL, l);                  (*pr->pr_usrreq)(so, PRU_RCVD, NULL, NULL, NULL, l);
   
  restart:   restart:
         if ((error = sblock(&so->so_rcv, SBLOCKWAIT(flags))) != 0)          if ((error = sblock(&so->so_rcv, SBLOCKWAIT(flags))) != 0)
                 return error;                  return (error);
         s = splsoftnet();          s = splsoftnet();
   
         m = so->so_rcv.sb_mb;          m = so->so_rcv.sb_mb;
Line 1073  soreceive(struct socket *so, struct mbuf
Line 1011  soreceive(struct socket *so, struct mbuf
          * we have to do the receive in sections, and thus risk returning           * we have to do the receive in sections, and thus risk returning
          * a short count if a timeout or signal occurs after we start.           * a short count if a timeout or signal occurs after we start.
          */           */
         if (m == NULL ||          if (m == 0 || (((flags & MSG_DONTWAIT) == 0 &&
             ((flags & MSG_DONTWAIT) == 0 &&              so->so_rcv.sb_cc < uio->uio_resid) &&
              so->so_rcv.sb_cc < uio->uio_resid &&              (so->so_rcv.sb_cc < so->so_rcv.sb_lowat ||
              (so->so_rcv.sb_cc < so->so_rcv.sb_lowat ||              ((flags & MSG_WAITALL) && uio->uio_resid <= so->so_rcv.sb_hiwat)) &&
               ((flags & MSG_WAITALL) &&              m->m_nextpkt == 0 && (pr->pr_flags & PR_ATOMIC) == 0)) {
                uio->uio_resid <= so->so_rcv.sb_hiwat)) &&  
              m->m_nextpkt == NULL && !atomic)) {  
 #ifdef DIAGNOSTIC  #ifdef DIAGNOSTIC
                 if (m == NULL && so->so_rcv.sb_cc)                  if (m == 0 && so->so_rcv.sb_cc)
                         panic("receive 1");                          panic("receive 1");
 #endif  #endif
                 if (so->so_error) {                  if (so->so_error) {
                         if (m != NULL)                          if (m)
                                 goto dontblock;                                  goto dontblock;
                         error = so->so_error;                          error = so->so_error;
                         if ((flags & MSG_PEEK) == 0)                          if ((flags & MSG_PEEK) == 0)
Line 1093  soreceive(struct socket *so, struct mbuf
Line 1029  soreceive(struct socket *so, struct mbuf
                         goto release;                          goto release;
                 }                  }
                 if (so->so_state & SS_CANTRCVMORE) {                  if (so->so_state & SS_CANTRCVMORE) {
                         if (m != NULL)                          if (m)
                                 goto dontblock;                                  goto dontblock;
                         else                          else
                                 goto release;                                  goto release;
                 }                  }
                 for (; m != NULL; m = m->m_next)                  for (; m; m = m->m_next)
                         if (m->m_type == MT_OOBDATA  || (m->m_flags & M_EOR)) {                          if (m->m_type == MT_OOBDATA  || (m->m_flags & M_EOR)) {
                                 m = so->so_rcv.sb_mb;                                  m = so->so_rcv.sb_mb;
                                 goto dontblock;                                  goto dontblock;
Line 1119  soreceive(struct socket *so, struct mbuf
Line 1055  soreceive(struct socket *so, struct mbuf
                 sbunlock(&so->so_rcv);                  sbunlock(&so->so_rcv);
                 error = sbwait(&so->so_rcv);                  error = sbwait(&so->so_rcv);
                 splx(s);                  splx(s);
                 if (error != 0)                  if (error)
                         return error;                          return (error);
                 goto restart;                  goto restart;
         }          }
  dontblock:   dontblock:
Line 1129  soreceive(struct socket *so, struct mbuf
Line 1065  soreceive(struct socket *so, struct mbuf
          * While we process the initial mbufs containing address and control           * While we process the initial mbufs containing address and control
          * info, we save a copy of m->m_nextpkt into nextrecord.           * info, we save a copy of m->m_nextpkt into nextrecord.
          */           */
         if (l != NULL)          if (l)
                 l->l_proc->p_stats->p_ru.ru_msgrcv++;                  l->l_proc->p_stats->p_ru.ru_msgrcv++;
         KASSERT(m == so->so_rcv.sb_mb);          KASSERT(m == so->so_rcv.sb_mb);
         SBLASTRECORDCHK(&so->so_rcv, "soreceive 1");          SBLASTRECORDCHK(&so->so_rcv, "soreceive 1");
Line 1148  soreceive(struct socket *so, struct mbuf
Line 1084  soreceive(struct socket *so, struct mbuf
                 } else {                  } else {
                         sbfree(&so->so_rcv, m);                          sbfree(&so->so_rcv, m);
                         mbuf_removed = 1;                          mbuf_removed = 1;
                         if (paddr != NULL) {                          if (paddr) {
                                 *paddr = m;                                  *paddr = m;
                                 so->so_rcv.sb_mb = m->m_next;                                  so->so_rcv.sb_mb = m->m_next;
                                 m->m_next = NULL;                                  m->m_next = 0;
                                 m = so->so_rcv.sb_mb;                                  m = so->so_rcv.sb_mb;
                         } else {                          } else {
                                 MFREE(m, so->so_rcv.sb_mb);                                  MFREE(m, so->so_rcv.sb_mb);
Line 1159  soreceive(struct socket *so, struct mbuf
Line 1095  soreceive(struct socket *so, struct mbuf
                         }                          }
                 }                  }
         }          }
         while (m != NULL && m->m_type == MT_CONTROL && error == 0) {          while (m && m->m_type == MT_CONTROL && error == 0) {
                 if (flags & MSG_PEEK) {                  if (flags & MSG_PEEK) {
                         if (controlp != NULL)                          if (controlp)
                                 *controlp = m_copy(m, 0, m->m_len);                                  *controlp = m_copy(m, 0, m->m_len);
                         m = m->m_next;                          m = m->m_next;
                 } else {                  } else {
                         sbfree(&so->so_rcv, m);                          sbfree(&so->so_rcv, m);
                         mbuf_removed = 1;                          mbuf_removed = 1;
                         if (controlp != NULL) {                          if (controlp) {
                                   struct domain *dom = pr->pr_domain;
                                 if (dom->dom_externalize && l &&                                  if (dom->dom_externalize && l &&
                                     mtod(m, struct cmsghdr *)->cmsg_type ==                                      mtod(m, struct cmsghdr *)->cmsg_type ==
                                     SCM_RIGHTS)                                      SCM_RIGHTS)
                                         error = (*dom->dom_externalize)(m, l);                                          error = (*dom->dom_externalize)(m, l);
                                 *controlp = m;                                  *controlp = m;
                                 so->so_rcv.sb_mb = m->m_next;                                  so->so_rcv.sb_mb = m->m_next;
                                 m->m_next = NULL;                                  m->m_next = 0;
                                 m = so->so_rcv.sb_mb;                                  m = so->so_rcv.sb_mb;
                         } else {                          } else {
                                 /*                                  /*
                                  * Dispose of any SCM_RIGHTS message that went                                   * Dispose of any SCM_RIGHTS message that went
                                  * through the read path rather than recv.                                   * through the read path rather than recv.
                                  */                                   */
                                 if (dom->dom_dispose &&                                  if (pr->pr_domain->dom_dispose &&
                                     mtod(m, struct cmsghdr *)->cmsg_type == SCM_RIGHTS)                                      mtod(m, struct cmsghdr *)->cmsg_type == SCM_RIGHTS)
                                         (*dom->dom_dispose)(m);                                          (*pr->pr_domain->dom_dispose)(m);
                                 MFREE(m, so->so_rcv.sb_mb);                                  MFREE(m, so->so_rcv.sb_mb);
                                 m = so->so_rcv.sb_mb;                                  m = so->so_rcv.sb_mb;
                         }                          }
                 }                  }
                 if (controlp != NULL) {                  if (controlp) {
                         orig_resid = 0;                          orig_resid = 0;
                         controlp = &(*controlp)->m_next;                          controlp = &(*controlp)->m_next;
                 }                  }
Line 1200  soreceive(struct socket *so, struct mbuf
Line 1137  soreceive(struct socket *so, struct mbuf
          * the last packet on the chain (nextrecord == NULL) and we           * the last packet on the chain (nextrecord == NULL) and we
          * change m->m_nextpkt.           * change m->m_nextpkt.
          */           */
         if (m != NULL) {          if (m) {
                 if ((flags & MSG_PEEK) == 0) {                  if ((flags & MSG_PEEK) == 0) {
                         m->m_nextpkt = nextrecord;                          m->m_nextpkt = nextrecord;
                         /*                          /*
Line 1228  soreceive(struct socket *so, struct mbuf
Line 1165  soreceive(struct socket *so, struct mbuf
   
         moff = 0;          moff = 0;
         offset = 0;          offset = 0;
         while (m != NULL && uio->uio_resid > 0 && error == 0) {          while (m && uio->uio_resid > 0 && error == 0) {
                 if (m->m_type == MT_OOBDATA) {                  if (m->m_type == MT_OOBDATA) {
                         if (type != MT_OOBDATA)                          if (type != MT_OOBDATA)
                                 break;                                  break;
Line 1252  soreceive(struct socket *so, struct mbuf
Line 1189  soreceive(struct socket *so, struct mbuf
                  * we must note any additions to the sockbuf when we                   * we must note any additions to the sockbuf when we
                  * block interrupts again.                   * block interrupts again.
                  */                   */
                 if (mp == NULL) {                  if (mp == 0) {
                         SBLASTRECORDCHK(&so->so_rcv, "soreceive uiomove");                          SBLASTRECORDCHK(&so->so_rcv, "soreceive uiomove");
                         SBLASTMBUFCHK(&so->so_rcv, "soreceive uiomove");                          SBLASTMBUFCHK(&so->so_rcv, "soreceive uiomove");
                         splx(s);                          splx(s);
                         error = uiomove(mtod(m, char *) + moff, (int)len, uio);                          error = uiomove(mtod(m, char *) + moff, (int)len, uio);
                         s = splsoftnet();                          s = splsoftnet();
                         if (error != 0) {                          if (error) {
                                 /*                                  /*
                                  * If any part of the record has been removed                                   * If any part of the record has been removed
                                  * (such as the MT_SONAME mbuf, which will                                   * (such as the MT_SONAME mbuf, which will
Line 1270  soreceive(struct socket *so, struct mbuf
Line 1207  soreceive(struct socket *so, struct mbuf
                                  * This avoids a later panic("receive 1a")                                   * This avoids a later panic("receive 1a")
                                  * when compiled with DIAGNOSTIC.                                   * when compiled with DIAGNOSTIC.
                                  */                                   */
                                 if (m && mbuf_removed && atomic)                                  if (m && mbuf_removed
                                       && (pr->pr_flags & PR_ATOMIC))
                                         (void) sbdroprecord(&so->so_rcv);                                          (void) sbdroprecord(&so->so_rcv);
   
                                 goto release;                                  goto release;
Line 1311  soreceive(struct socket *so, struct mbuf
Line 1249  soreceive(struct socket *so, struct mbuf
                                 SBLASTRECORDCHK(&so->so_rcv, "soreceive 3");                                  SBLASTRECORDCHK(&so->so_rcv, "soreceive 3");
                                 SBLASTMBUFCHK(&so->so_rcv, "soreceive 3");                                  SBLASTMBUFCHK(&so->so_rcv, "soreceive 3");
                         }                          }
                 } else if (flags & MSG_PEEK)                  } else {
                         moff += len;                          if (flags & MSG_PEEK)
                 else {                                  moff += len;
                         if (mp != NULL)                          else {
                                 *mp = m_copym(m, 0, len, M_WAIT);                                  if (mp)
                         m->m_data += len;                                          *mp = m_copym(m, 0, len, M_WAIT);
                         m->m_len -= len;                                  m->m_data += len;
                         so->so_rcv.sb_cc -= len;                                  m->m_len -= len;
                                   so->so_rcv.sb_cc -= len;
                           }
                 }                  }
                 if (so->so_oobmark) {                  if (so->so_oobmark) {
                         if ((flags & MSG_PEEK) == 0) {                          if ((flags & MSG_PEEK) == 0) {
Line 1342  soreceive(struct socket *so, struct mbuf
Line 1282  soreceive(struct socket *so, struct mbuf
                  * with a short count but without error.                   * with a short count but without error.
                  * Keep sockbuf locked against other readers.                   * Keep sockbuf locked against other readers.
                  */                   */
                 while (flags & MSG_WAITALL && m == NULL && uio->uio_resid > 0 &&                  while (flags & MSG_WAITALL && m == 0 && uio->uio_resid > 0 &&
                     !sosendallatonce(so) && !nextrecord) {                      !sosendallatonce(so) && !nextrecord) {
                         if (so->so_error || so->so_state & SS_CANTRCVMORE)                          if (so->so_error || so->so_state & SS_CANTRCVMORE)
                                 break;                                  break;
Line 1363  soreceive(struct socket *so, struct mbuf
Line 1303  soreceive(struct socket *so, struct mbuf
                         SBLASTRECORDCHK(&so->so_rcv, "soreceive sbwait 2");                          SBLASTRECORDCHK(&so->so_rcv, "soreceive sbwait 2");
                         SBLASTMBUFCHK(&so->so_rcv, "soreceive sbwait 2");                          SBLASTMBUFCHK(&so->so_rcv, "soreceive sbwait 2");
                         error = sbwait(&so->so_rcv);                          error = sbwait(&so->so_rcv);
                         if (error != 0) {                          if (error) {
                                 sbunlock(&so->so_rcv);                                  sbunlock(&so->so_rcv);
                                 splx(s);                                  splx(s);
                                 return 0;                                  return (0);
                         }                          }
                         if ((m = so->so_rcv.sb_mb) != NULL)                          if ((m = so->so_rcv.sb_mb) != NULL)
                                 nextrecord = m->m_nextpkt;                                  nextrecord = m->m_nextpkt;
                 }                  }
         }          }
   
         if (m && atomic) {          if (m && pr->pr_flags & PR_ATOMIC) {
                 flags |= MSG_TRUNC;                  flags |= MSG_TRUNC;
                 if ((flags & MSG_PEEK) == 0)                  if ((flags & MSG_PEEK) == 0)
                         (void) sbdroprecord(&so->so_rcv);                          (void) sbdroprecord(&so->so_rcv);
         }          }
         if ((flags & MSG_PEEK) == 0) {          if ((flags & MSG_PEEK) == 0) {
                 if (m == NULL) {                  if (m == 0) {
                         /*                          /*
                          * First part is an inline SB_EMPTY_FIXUP().  Second                           * First part is an inline SB_EMPTY_FIXUP().  Second
                          * part makes sure sb_lastrecord is up-to-date if                           * part makes sure sb_lastrecord is up-to-date if
Line 1405  soreceive(struct socket *so, struct mbuf
Line 1345  soreceive(struct socket *so, struct mbuf
                 goto restart;                  goto restart;
         }          }
   
         if (flagsp != NULL)          if (flagsp)
                 *flagsp |= flags;                  *flagsp |= flags;
  release:   release:
         sbunlock(&so->so_rcv);          sbunlock(&so->so_rcv);
         splx(s);          splx(s);
         return error;          return (error);
 }  }
   
 int  int
Line 1427  soshutdown(struct socket *so, int how)
Line 1367  soshutdown(struct socket *so, int how)
         if (how == SHUT_WR || how == SHUT_RDWR)          if (how == SHUT_WR || how == SHUT_RDWR)
                 return (*pr->pr_usrreq)(so, PRU_SHUTDOWN, NULL,                  return (*pr->pr_usrreq)(so, PRU_SHUTDOWN, NULL,
                     NULL, NULL, NULL);                      NULL, NULL, NULL);
         return 0;          return (0);
 }  }
   
 void  void
Line 1457  sorflush(struct socket *so)
Line 1397  sorflush(struct socket *so)
         sbrelease(&asb, so);          sbrelease(&asb, so);
 }  }
   
 static int  int
 sosetopt1(struct socket *so, int level, int optname, struct mbuf *m)  sosetopt(struct socket *so, int level, int optname, struct mbuf *m0)
 {  {
         int optval, val;          int             error;
           struct mbuf     *m;
         struct linger   *l;          struct linger   *l;
         struct sockbuf  *sb;          struct sockbuf  *sb;
         struct timeval *tv;  
   
         switch (optname) {  
   
         case SO_LINGER:  
                 if (m == NULL || m->m_len != sizeof(struct linger))  
                         return EINVAL;  
                 l = mtod(m, struct linger *);  
                 if (l->l_linger < 0 || l->l_linger > USHRT_MAX ||  
                     l->l_linger > (INT_MAX / hz))  
                         return EDOM;  
                 so->so_linger = l->l_linger;  
                 if (l->l_onoff)  
                         so->so_options |= SO_LINGER;  
                 else  
                         so->so_options &= ~SO_LINGER;  
                 break;  
   
         case SO_DEBUG:  
         case SO_KEEPALIVE:  
         case SO_DONTROUTE:  
         case SO_USELOOPBACK:  
         case SO_BROADCAST:  
         case SO_REUSEADDR:  
         case SO_REUSEPORT:  
         case SO_OOBINLINE:  
         case SO_TIMESTAMP:  
                 if (m == NULL || m->m_len < sizeof(int))  
                         return EINVAL;  
                 if (*mtod(m, int *))  
                         so->so_options |= optname;  
                 else  
                         so->so_options &= ~optname;  
                 break;  
   
         case SO_SNDBUF:          error = 0;
         case SO_RCVBUF:          m = m0;
         case SO_SNDLOWAT:          if (level != SOL_SOCKET) {
         case SO_RCVLOWAT:                  if (so->so_proto && so->so_proto->pr_ctloutput)
                 if (m == NULL || m->m_len < sizeof(int))                          return ((*so->so_proto->pr_ctloutput)
                         return EINVAL;                                    (PRCO_SETOPT, so, level, optname, &m0));
                   error = ENOPROTOOPT;
           } else {
                   switch (optname) {
   
                 /*                  case SO_LINGER:
                  * Values < 1 make no sense for any of these                          if (m == NULL || m->m_len != sizeof(struct linger)) {
                  * options, so disallow them.                                  error = EINVAL;
                  */                                  goto bad;
                 optval = *mtod(m, int *);                          }
                 if (optval < 1)                          l = mtod(m, struct linger *);
                         return EINVAL;                          if (l->l_linger < 0 || l->l_linger > USHRT_MAX ||
                               l->l_linger > (INT_MAX / hz)) {
                                   error = EDOM;
                                   goto bad;
                           }
                           so->so_linger = l->l_linger;
                           if (l->l_onoff)
                                   so->so_options |= SO_LINGER;
                           else
                                   so->so_options &= ~SO_LINGER;
                           break;
   
                 switch (optname) {                  case SO_DEBUG:
                   case SO_KEEPALIVE:
                   case SO_DONTROUTE:
                   case SO_USELOOPBACK:
                   case SO_BROADCAST:
                   case SO_REUSEADDR:
                   case SO_REUSEPORT:
                   case SO_OOBINLINE:
                   case SO_TIMESTAMP:
                           if (m == NULL || m->m_len < sizeof(int)) {
                                   error = EINVAL;
                                   goto bad;
                           }
                           if (*mtod(m, int *))
                                   so->so_options |= optname;
                           else
                                   so->so_options &= ~optname;
                           break;
   
                 case SO_SNDBUF:                  case SO_SNDBUF:
                 case SO_RCVBUF:                  case SO_RCVBUF:
                         sb = (optname == SO_SNDBUF) ?  
                             &so->so_snd : &so->so_rcv;  
                         if (sbreserve(sb, (u_long)optval, so) == 0)  
                                 return ENOBUFS;  
                         sb->sb_flags &= ~SB_AUTOSIZE;  
                         break;  
   
                 /*  
                  * Make sure the low-water is never greater than  
                  * the high-water.  
                  */  
                 case SO_SNDLOWAT:                  case SO_SNDLOWAT:
                         so->so_snd.sb_lowat =  
                             (optval > so->so_snd.sb_hiwat) ?  
                             so->so_snd.sb_hiwat : optval;  
                         break;  
                 case SO_RCVLOWAT:                  case SO_RCVLOWAT:
                         so->so_rcv.sb_lowat =                      {
                             (optval > so->so_rcv.sb_hiwat) ?                          int optval;
                             so->so_rcv.sb_hiwat : optval;  
                         break;  
                 }  
                 break;  
   
         case SO_SNDTIMEO:                          if (m == NULL || m->m_len < sizeof(int)) {
         case SO_RCVTIMEO:                                  error = EINVAL;
                 if (m == NULL || m->m_len < sizeof(*tv))                                  goto bad;
                         return EINVAL;                          }
                 tv = mtod(m, struct timeval *);  
                 if (tv->tv_sec > (INT_MAX - tv->tv_usec / tick) / hz)  
                         return EDOM;  
                 val = tv->tv_sec * hz + tv->tv_usec / tick;  
                 if (val == 0 && tv->tv_usec != 0)  
                         val = 1;  
   
                 switch (optname) {                          /*
                            * Values < 1 make no sense for any of these
                            * options, so disallow them.
                            */
                           optval = *mtod(m, int *);
                           if (optval < 1) {
                                   error = EINVAL;
                                   goto bad;
                           }
   
                 case SO_SNDTIMEO:                          switch (optname) {
                         so->so_snd.sb_timeo = val;  
                           case SO_SNDBUF:
                           case SO_RCVBUF:
                                   sb = (optname == SO_SNDBUF) ?
                                       &so->so_snd : &so->so_rcv;
                                   if (sbreserve(sb, (u_long)optval, so) == 0) {
                                           error = ENOBUFS;
                                           goto bad;
                                   }
                                   sb->sb_flags &= ~SB_AUTOSIZE;
                                   break;
   
                           /*
                            * Make sure the low-water is never greater than
                            * the high-water.
                            */
                           case SO_SNDLOWAT:
                                   so->so_snd.sb_lowat =
                                       (optval > so->so_snd.sb_hiwat) ?
                                       so->so_snd.sb_hiwat : optval;
                                   break;
                           case SO_RCVLOWAT:
                                   so->so_rcv.sb_lowat =
                                       (optval > so->so_rcv.sb_hiwat) ?
                                       so->so_rcv.sb_hiwat : optval;
                                   break;
                           }
                         break;                          break;
                       }
   
                   case SO_SNDTIMEO:
                 case SO_RCVTIMEO:                  case SO_RCVTIMEO:
                         so->so_rcv.sb_timeo = val;                      {
                         break;                          struct timeval *tv;
                 }                          int val;
                 break;  
   
         default:                          if (m == NULL || m->m_len < sizeof(*tv)) {
                 return ENOPROTOOPT;                                  error = EINVAL;
         }                                  goto bad;
         return 0;                          }
 }                          tv = mtod(m, struct timeval *);
                           if (tv->tv_sec > (INT_MAX - tv->tv_usec / tick) / hz) {
                                   error = EDOM;
                                   goto bad;
                           }
                           val = tv->tv_sec * hz + tv->tv_usec / tick;
                           if (val == 0 && tv->tv_usec != 0)
                                   val = 1;
   
 int                          switch (optname) {
 sosetopt(struct socket *so, int level, int optname, struct mbuf *m)  
 {  
         int error, prerr;  
   
         if (level == SOL_SOCKET)                          case SO_SNDTIMEO:
                 error = sosetopt1(so, level, optname, m);                                  so->so_snd.sb_timeo = val;
         else                                  break;
                 error = ENOPROTOOPT;                          case SO_RCVTIMEO:
                                   so->so_rcv.sb_timeo = val;
                                   break;
                           }
                           break;
                       }
   
         if ((error == 0 || error == ENOPROTOOPT) &&                  default:
             so->so_proto != NULL && so->so_proto->pr_ctloutput != NULL) {                          error = ENOPROTOOPT;
                 /* give the protocol stack a shot */                          break;
                 prerr = (*so->so_proto->pr_ctloutput)(PRCO_SETOPT, so, level,                  }
                     optname, &m);                  if (error == 0 && so->so_proto && so->so_proto->pr_ctloutput) {
                 if (prerr == 0)                          (void) ((*so->so_proto->pr_ctloutput)
                         error = 0;                                    (PRCO_SETOPT, so, level, optname, &m0));
                 else if (prerr != ENOPROTOOPT)                          m = NULL;       /* freed by protocol */
                         error = prerr;                  }
         } else if (m != NULL)          }
                 (void)m_free(m);   bad:
         return error;          if (m)
                   (void) m_free(m);
           return (error);
 }  }
   
 int  int
Line 1795  soo_kqfilter(struct file *fp, struct kno
Line 1754  soo_kqfilter(struct file *fp, struct kno
                 sb = &so->so_snd;                  sb = &so->so_snd;
                 break;                  break;
         default:          default:
                 return (EINVAL);                  return (1);
         }          }
         SLIST_INSERT_HEAD(&sb->sb_sel.sel_klist, kn, kn_selnext);          SLIST_INSERT_HEAD(&sb->sb_sel.sel_klist, kn, kn_selnext);
         sb->sb_flags |= SB_KNOTE;          sb->sb_flags |= SB_KNOTE;

Legend:
Removed from v.1.140.6.6  
changed lines
  Added in v.1.141

CVSweb <webmaster@jp.NetBSD.org>