[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.164.2.2 and 1.166

version 1.164.2.2, 2008/09/18 04:31:44 version 1.166, 2008/05/26 17:21:18
Line 65 
Line 65 
 #include <sys/cdefs.h>  #include <sys/cdefs.h>
 __KERNEL_RCSID(0, "$NetBSD$");  __KERNEL_RCSID(0, "$NetBSD$");
   
 #include "opt_inet.h"  
 #include "opt_sock_counters.h"  #include "opt_sock_counters.h"
 #include "opt_sosend_loan.h"  #include "opt_sosend_loan.h"
 #include "opt_mbuftrace.h"  #include "opt_mbuftrace.h"
 #include "opt_somaxkva.h"  #include "opt_somaxkva.h"
 #include "opt_multiprocessor.h" /* XXX */  
   
 #include <sys/param.h>  #include <sys/param.h>
 #include <sys/systm.h>  #include <sys/systm.h>
Line 129  EVCNT_ATTACH_STATIC(sosend_kvalimit);
Line 127  EVCNT_ATTACH_STATIC(sosend_kvalimit);
   
 static struct callback_entry sokva_reclaimerentry;  static struct callback_entry sokva_reclaimerentry;
   
 #if defined(SOSEND_NO_LOAN) || defined(MULTIPROCESSOR)  #ifdef SOSEND_NO_LOAN
 int sock_loan_thresh = -1;  int sock_loan_thresh = -1;
 #else  #else
 int sock_loan_thresh = 4096;  int sock_loan_thresh = 4096;
Line 419  getsombuf(struct socket *so, int type)
Line 417  getsombuf(struct socket *so, int type)
         return m;          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)
 {  {
Line 490  socreate(int dom, struct socket **aso, i
Line 499  socreate(int dom, struct socket **aso, i
 #endif  #endif
         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);
         so->so_egid = kauth_cred_getegid(l->l_cred);  
         so->so_cpid = l->l_proc->p_pid;  
         if (lockso != NULL) {          if (lockso != NULL) {
                 /* Caller wants us to share a lock. */                  /* Caller wants us to share a lock. */
                 lock = lockso->so_lock;                  lock = lockso->so_lock;
Line 614  sofree(struct socket *so)
Line 621  sofree(struct socket *so)
         KASSERT(!cv_has_waiters(&so->so_snd.sb_cv));          KASSERT(!cv_has_waiters(&so->so_snd.sb_cv));
         sorflush(so);          sorflush(so);
         refs = so->so_aborting; /* XXX */          refs = so->so_aborting; /* XXX */
 #ifdef INET  
         /* remove acccept filter if one is present. */  
         if (so->so_accf != NULL)  
                 do_setopt_accept_filter(so, NULL);  
 #endif  
         sounlock(so);          sounlock(so);
         if (refs == 0)          /* XXX */          if (refs == 0)          /* XXX */
                 soput(so);                  soput(so);
Line 640  soclose(struct socket *so)
Line 642  soclose(struct socket *so)
         solock(so);          solock(so);
         if (so->so_options & SO_ACCEPTCONN) {          if (so->so_options & SO_ACCEPTCONN) {
                 do {                  do {
                         while ((so2 = TAILQ_FIRST(&so->so_q0)) != 0) {                          if ((so2 = TAILQ_FIRST(&so->so_q0)) != 0) {
                                 KASSERT(solocked2(so, so2));                                  KASSERT(solocked2(so, so2));
                                 (void) soqremque(so2, 0);                                  (void) soqremque(so2, 0);
                                 /* soabort drops the lock. */                                  /* soabort drops the lock. */
                                 (void) soabort(so2);                                  (void) soabort(so2);
                                 solock(so);                                  solock(so);
                                   continue;
                         }                          }
                         while ((so2 = TAILQ_FIRST(&so->so_q)) != 0) {                          if ((so2 = TAILQ_FIRST(&so->so_q)) != 0) {
                                 KASSERT(solocked2(so, so2));                                  KASSERT(solocked2(so, so2));
                                 (void) soqremque(so2, 1);                                  (void) soqremque(so2, 1);
                                 /* soabort drops the lock. */                                  /* soabort drops the lock. */
                                 (void) soabort(so2);                                  (void) soabort(so2);
                                 solock(so);                                  solock(so);
                                   continue;
                         }                          }
                 } while (!TAILQ_EMPTY(&so->so_q0));                  } while (0);
         }          }
         if (so->so_pcb == 0)          if (so->so_pcb == 0)
                 goto discard;                  goto discard;
Line 1564  sorflush(struct socket *so)
Line 1568  sorflush(struct socket *so)
         sbrelease(&asb, so);          sbrelease(&asb, so);
 }  }
   
 /*  
  * internal set SOL_SOCKET options  
  */  
 static int  static int
 sosetopt1(struct socket *so, const struct sockopt *sopt)  sosetopt1(struct socket *so, int level, int optname, struct mbuf *m)
 {  {
         int error, optval;          int optval, val;
         struct linger l;          struct linger   *l;
         struct timeval tv;          struct sockbuf  *sb;
           struct timeval *tv;
   
         switch (sopt->sopt_name) {          switch (optname) {
   
 #ifdef INET          case SO_LINGER:
         case SO_ACCEPTFILTER:                  if (m == NULL || m->m_len != sizeof(struct linger))
                 error = do_setopt_accept_filter(so, sopt);                          return EINVAL;
                 if (error)                  l = mtod(m, struct linger *);
                         return error;                  if (l->l_linger < 0 || l->l_linger > USHRT_MAX ||
                 break;                      l->l_linger > (INT_MAX / hz))
 #endif  
   
         case SO_LINGER:  
                 error = sockopt_get(sopt, &l, sizeof(l));  
                 if (error)  
                         return (error);  
   
                 if (l.l_linger < 0 || l.l_linger > USHRT_MAX ||  
                     l.l_linger > (INT_MAX / hz))  
                         return EDOM;                          return EDOM;
                 so->so_linger = l.l_linger;                  so->so_linger = l->l_linger;
                 if (l.l_onoff)                  if (l->l_onoff)
                         so->so_options |= SO_LINGER;                          so->so_options |= SO_LINGER;
                 else                  else
                         so->so_options &= ~SO_LINGER;                          so->so_options &= ~SO_LINGER;
                   break;
                 break;  
   
         case SO_DEBUG:          case SO_DEBUG:
         case SO_KEEPALIVE:          case SO_KEEPALIVE:
Line 1609  sosetopt1(struct socket *so, const struc
Line 1601  sosetopt1(struct socket *so, const struc
         case SO_REUSEPORT:          case SO_REUSEPORT:
         case SO_OOBINLINE:          case SO_OOBINLINE:
         case SO_TIMESTAMP:          case SO_TIMESTAMP:
                 error = sockopt_getint(sopt, &optval);                  if (m == NULL || m->m_len < sizeof(int))
                 if (error)                          return EINVAL;
                         return (error);                  if (*mtod(m, int *))
                           so->so_options |= optname;
                 if (optval)  
                         so->so_options |= sopt->sopt_name;  
                 else                  else
                         so->so_options &= ~sopt->sopt_name;                          so->so_options &= ~optname;
                 break;                  break;
   
         case SO_SNDBUF:          case SO_SNDBUF:
         case SO_RCVBUF:          case SO_RCVBUF:
         case SO_SNDLOWAT:          case SO_SNDLOWAT:
         case SO_RCVLOWAT:          case SO_RCVLOWAT:
                 error = sockopt_getint(sopt, &optval);                  if (m == NULL || m->m_len < sizeof(int))
                 if (error)                          return EINVAL;
                         return (error);  
   
                 /*                  /*
                  * Values < 1 make no sense for any of these                   * Values < 1 make no sense for any of these
                  * options, so disallow them.                   * options, so disallow them.
                  */                   */
                   optval = *mtod(m, int *);
                 if (optval < 1)                  if (optval < 1)
                         return EINVAL;                          return EINVAL;
   
                 switch (sopt->sopt_name) {                  switch (optname) {
                 case SO_SNDBUF:  
                         if (sbreserve(&so->so_snd, (u_long)optval, so) == 0)  
                                 return ENOBUFS;  
   
                         so->so_snd.sb_flags &= ~SB_AUTOSIZE;  
                         break;  
   
                   case SO_SNDBUF:
                 case SO_RCVBUF:                  case SO_RCVBUF:
                         if (sbreserve(&so->so_rcv, (u_long)optval, so) == 0)                          sb = (optname == SO_SNDBUF) ?
                               &so->so_snd : &so->so_rcv;
                           if (sbreserve(sb, (u_long)optval, so) == 0)
                                 return ENOBUFS;                                  return ENOBUFS;
                           sb->sb_flags &= ~SB_AUTOSIZE;
                         so->so_rcv.sb_flags &= ~SB_AUTOSIZE;  
                         break;                          break;
   
                 /*                  /*
Line 1654  sosetopt1(struct socket *so, const struc
Line 1640  sosetopt1(struct socket *so, const struc
                  * the high-water.                   * the high-water.
                  */                   */
                 case SO_SNDLOWAT:                  case SO_SNDLOWAT:
                         if (optval > so->so_snd.sb_hiwat)                          so->so_snd.sb_lowat =
                                 optval = so->so_snd.sb_hiwat;                              (optval > so->so_snd.sb_hiwat) ?
                               so->so_snd.sb_hiwat : optval;
                         so->so_snd.sb_lowat = optval;  
                         break;                          break;
   
                 case SO_RCVLOWAT:                  case SO_RCVLOWAT:
                         if (optval > so->so_rcv.sb_hiwat)                          so->so_rcv.sb_lowat =
                                 optval = so->so_rcv.sb_hiwat;                              (optval > so->so_rcv.sb_hiwat) ?
                               so->so_rcv.sb_hiwat : optval;
                         so->so_rcv.sb_lowat = optval;  
                         break;                          break;
                 }                  }
                 break;                  break;
   
         case SO_SNDTIMEO:          case SO_SNDTIMEO:
         case SO_RCVTIMEO:          case SO_RCVTIMEO:
                 error = sockopt_get(sopt, &tv, sizeof(tv));                  if (m == NULL || m->m_len < sizeof(*tv))
                 if (error)                          return EINVAL;
                         return (error);                  tv = mtod(m, struct timeval *);
                   if (tv->tv_sec > (INT_MAX - tv->tv_usec / tick) / hz)
                 if (tv.tv_sec > (INT_MAX - tv.tv_usec / tick) / hz)  
                         return EDOM;                          return EDOM;
                   val = tv->tv_sec * hz + tv->tv_usec / tick;
                   if (val == 0 && tv->tv_usec != 0)
                           val = 1;
   
                 optval = tv.tv_sec * hz + tv.tv_usec / tick;                  switch (optname) {
                 if (optval == 0 && tv.tv_usec != 0)  
                         optval = 1;  
   
                 switch (sopt->sopt_name) {  
                 case SO_SNDTIMEO:                  case SO_SNDTIMEO:
                         so->so_snd.sb_timeo = optval;                          so->so_snd.sb_timeo = val;
                         break;                          break;
                 case SO_RCVTIMEO:                  case SO_RCVTIMEO:
                         so->so_rcv.sb_timeo = optval;                          so->so_rcv.sb_timeo = val;
                         break;                          break;
                 }                  }
                 break;                  break;
Line 1699  sosetopt1(struct socket *so, const struc
Line 1681  sosetopt1(struct socket *so, const struc
 }  }
   
 int  int
 sosetopt(struct socket *so, struct sockopt *sopt)  sosetopt(struct socket *so, int level, int optname, struct mbuf *m)
 {  {
         int error, prerr;          int error, prerr;
   
         solock(so);          solock(so);
         if (sopt->sopt_level == SOL_SOCKET)          if (level == SOL_SOCKET)
                 error = sosetopt1(so, sopt);                  error = sosetopt1(so, level, optname, m);
         else          else
                 error = ENOPROTOOPT;                  error = ENOPROTOOPT;
   
         if ((error == 0 || error == ENOPROTOOPT) &&          if ((error == 0 || error == ENOPROTOOPT) &&
             so->so_proto != NULL && so->so_proto->pr_ctloutput != NULL) {              so->so_proto != NULL && so->so_proto->pr_ctloutput != NULL) {
                 /* give the protocol stack a shot */                  /* give the protocol stack a shot */
                 prerr = (*so->so_proto->pr_ctloutput)(PRCO_SETOPT, so, sopt);                  prerr = (*so->so_proto->pr_ctloutput)(PRCO_SETOPT, so, level,
                       optname, &m);
                 if (prerr == 0)                  if (prerr == 0)
                         error = 0;                          error = 0;
                 else if (prerr != ENOPROTOOPT)                  else if (prerr != ENOPROTOOPT)
                         error = prerr;                          error = prerr;
         }          } else if (m != NULL)
                   (void)m_free(m);
         sounlock(so);          sounlock(so);
         return error;          return error;
 }  }
   
 /*  
  * so_setsockopt() is a wrapper providing a sockopt structure for sosetopt()  
  */  
 int  int
 so_setsockopt(struct lwp *l, struct socket *so, int level, int name,  sogetopt(struct socket *so, int level, int optname, struct mbuf **mp)
     const void *val, size_t valsize)  
 {  
         struct sockopt sopt;  
         int error;  
   
         KASSERT(valsize == 0 || val != NULL);  
   
         sockopt_init(&sopt, level, name, valsize);  
         sockopt_set(&sopt, val, valsize);  
   
         error = sosetopt(so, &sopt);  
   
         sockopt_destroy(&sopt);  
   
         return error;  
 }  
   
 /*  
  * internal get SOL_SOCKET options  
  */  
 static int  
 sogetopt1(struct socket *so, struct sockopt *sopt)  
 {  
         int error, optval;  
         struct linger l;  
         struct timeval tv;  
   
         switch (sopt->sopt_name) {  
   
 #ifdef INET  
         case SO_ACCEPTFILTER:  
                 error = do_getopt_accept_filter(so, sopt);  
                 break;  
 #endif  
   
         case SO_LINGER:  
                 l.l_onoff = (so->so_options & SO_LINGER) ? 1 : 0;  
                 l.l_linger = so->so_linger;  
   
                 error = sockopt_set(sopt, &l, sizeof(l));  
                 break;  
   
         case SO_USELOOPBACK:  
         case SO_DONTROUTE:  
         case SO_DEBUG:  
         case SO_KEEPALIVE:  
         case SO_REUSEADDR:  
         case SO_REUSEPORT:  
         case SO_BROADCAST:  
         case SO_OOBINLINE:  
         case SO_TIMESTAMP:  
                 error = sockopt_setint(sopt,  
                     (so->so_options & sopt->sopt_name) ? 1 : 0);  
                 break;  
   
         case SO_TYPE:  
                 error = sockopt_setint(sopt, so->so_type);  
                 break;  
   
         case SO_ERROR:  
                 error = sockopt_setint(sopt, so->so_error);  
                 so->so_error = 0;  
                 break;  
   
         case SO_SNDBUF:  
                 error = sockopt_setint(sopt, so->so_snd.sb_hiwat);  
                 break;  
   
         case SO_RCVBUF:  
                 error = sockopt_setint(sopt, so->so_rcv.sb_hiwat);  
                 break;  
   
         case SO_SNDLOWAT:  
                 error = sockopt_setint(sopt, so->so_snd.sb_lowat);  
                 break;  
   
         case SO_RCVLOWAT:  
                 error = sockopt_setint(sopt, so->so_rcv.sb_lowat);  
                 break;  
   
         case SO_SNDTIMEO:  
         case SO_RCVTIMEO:  
                 optval = (sopt->sopt_name == SO_SNDTIMEO ?  
                      so->so_snd.sb_timeo : so->so_rcv.sb_timeo);  
   
                 tv.tv_sec = optval / hz;  
                 tv.tv_usec = (optval % hz) * tick;  
   
                 error = sockopt_set(sopt, &tv, sizeof(tv));  
                 break;  
   
         case SO_OVERFLOWED:  
                 error = sockopt_setint(sopt, so->so_rcv.sb_overflowed);  
                 break;  
   
         default:  
                 error = ENOPROTOOPT;  
                 break;  
         }  
   
         return (error);  
 }  
   
 int  
 sogetopt(struct socket *so, struct sockopt *sopt)  
 {  {
           struct mbuf     *m;
         int             error;          int             error;
   
         solock(so);          solock(so);
         if (sopt->sopt_level != SOL_SOCKET) {          if (level != SOL_SOCKET) {
                 if (so->so_proto && so->so_proto->pr_ctloutput) {                  if (so->so_proto && so->so_proto->pr_ctloutput) {
                         error = ((*so->so_proto->pr_ctloutput)                          error = ((*so->so_proto->pr_ctloutput)
                             (PRCO_GETOPT, so, sopt));                                    (PRCO_GETOPT, so, level, optname, mp));
                 } else                  } else
                         error = (ENOPROTOOPT);                          error = (ENOPROTOOPT);
         } else {          } else {
                 error = sogetopt1(so, sopt);                  m = m_get(M_WAIT, MT_SOOPTS);
         }                  m->m_len = sizeof(int);
         sounlock(so);  
         return (error);  
 }  
   
 /*  
  * alloc sockopt data buffer buffer  
  *      - will be released at destroy  
  */  
 static void  
 sockopt_alloc(struct sockopt *sopt, size_t len)  
 {  
   
         KASSERT(sopt->sopt_size == 0);  
   
         if (len > sizeof(sopt->sopt_buf))  
                 sopt->sopt_data = malloc(len, M_SOOPTS, M_WAITOK | M_ZERO);  
         else  
                 sopt->sopt_data = sopt->sopt_buf;  
   
         sopt->sopt_size = len;  
 }  
   
 /*  
  * initialise sockopt storage  
  */  
 void  
 sockopt_init(struct sockopt *sopt, int level, int name, size_t size)  
 {  
   
         memset(sopt, 0, sizeof(*sopt));  
   
         sopt->sopt_level = level;  
         sopt->sopt_name = name;  
         sockopt_alloc(sopt, size);  
 }  
   
 /*  
  * destroy sockopt storage  
  *      - will release any held memory references  
  */  
 void  
 sockopt_destroy(struct sockopt *sopt)  
 {  
   
         if (sopt->sopt_data != sopt->sopt_buf)  
                 free(sopt->sopt_data, M_SOOPTS);  
   
         memset(sopt, 0, sizeof(*sopt));  
 }  
   
 /*  
  * set sockopt value  
  *      - value is copied into sockopt  
  *      - memory is allocated when necessary  
  */  
 int  
 sockopt_set(struct sockopt *sopt, const void *buf, size_t len)  
 {  
   
         if (sopt->sopt_size == 0)  
                 sockopt_alloc(sopt, len);  
   
         KASSERT(sopt->sopt_size == len);  
         memcpy(sopt->sopt_data, buf, len);  
         return 0;  
 }  
   
 /*  
  * common case of set sockopt integer value  
  */  
 int  
 sockopt_setint(struct sockopt *sopt, int val)  
 {  
   
         return sockopt_set(sopt, &val, sizeof(int));                  switch (optname) {
 }  
   
 /*                  case SO_LINGER:
  * get sockopt value                          m->m_len = sizeof(struct linger);
  *      - correct size must be given                          mtod(m, struct linger *)->l_onoff =
  */                              (so->so_options & SO_LINGER) ? 1 : 0;
 int                          mtod(m, struct linger *)->l_linger = so->so_linger;
 sockopt_get(const struct sockopt *sopt, void *buf, size_t len)                          break;
 {  
   
         if (sopt->sopt_size != len)  
                 return EINVAL;  
   
         memcpy(buf, sopt->sopt_data, len);                  case SO_USELOOPBACK:
         return 0;                  case SO_DONTROUTE:
 }                  case SO_DEBUG:
                   case SO_KEEPALIVE:
                   case SO_REUSEADDR:
                   case SO_REUSEPORT:
                   case SO_BROADCAST:
                   case SO_OOBINLINE:
                   case SO_TIMESTAMP:
                           *mtod(m, int *) = (so->so_options & optname) ? 1 : 0;
                           break;
   
 /*                  case SO_TYPE:
  * common case of get sockopt integer value                          *mtod(m, int *) = so->so_type;
  */                          break;
 int  
 sockopt_getint(const struct sockopt *sopt, int *valp)  
 {  
   
         return sockopt_get(sopt, valp, sizeof(int));                  case SO_ERROR:
 }                          *mtod(m, int *) = so->so_error;
                           so->so_error = 0;
                           break;
   
 /*                  case SO_SNDBUF:
  * set sockopt value from mbuf                          *mtod(m, int *) = so->so_snd.sb_hiwat;
  *      - ONLY for legacy code                          break;
  *      - mbuf is released by sockopt  
  */  
 int  
 sockopt_setmbuf(struct sockopt *sopt, struct mbuf *m)  
 {  
         size_t len;  
   
         len = m_length(m);                  case SO_RCVBUF:
                           *mtod(m, int *) = so->so_rcv.sb_hiwat;
                           break;
   
         if (sopt->sopt_size == 0)                  case SO_SNDLOWAT:
                 sockopt_alloc(sopt, len);                          *mtod(m, int *) = so->so_snd.sb_lowat;
                           break;
   
         KASSERT(sopt->sopt_size == len);                  case SO_RCVLOWAT:
         m_copydata(m, 0, len, sopt->sopt_data);                          *mtod(m, int *) = so->so_rcv.sb_lowat;
         m_freem(m);                          break;
   
         return 0;                  case SO_SNDTIMEO:
 }                  case SO_RCVTIMEO:
                       {
                           int val = (optname == SO_SNDTIMEO ?
                                so->so_snd.sb_timeo : so->so_rcv.sb_timeo);
   
                           m->m_len = sizeof(struct timeval);
                           mtod(m, struct timeval *)->tv_sec = val / hz;
                           mtod(m, struct timeval *)->tv_usec =
                               (val % hz) * tick;
                           break;
                       }
   
 /*                  case SO_OVERFLOWED:
  * get sockopt value into mbuf                          *mtod(m, int *) = so->so_rcv.sb_overflowed;
  *      - ONLY for legacy code                          break;
  *      - mbuf to be released by the caller  
  */  
 struct mbuf *  
 sockopt_getmbuf(const struct sockopt *sopt)  
 {  
         struct mbuf *m;  
   
         m = m_get(M_WAIT, MT_SOOPTS);                  default:
         if (m == NULL)                          sounlock(so);
                 return NULL;                          (void)m_free(m);
                           return (ENOPROTOOPT);
         m->m_len = MLEN;                  }
         m_copyback(m, 0, sopt->sopt_size, sopt->sopt_data);                  *mp = m;
         if (m_length(m) != max(sopt->sopt_size, MLEN)) {                  error = 0;
                 m_freem(m);  
                 return NULL;  
         }          }
         m->m_len = min(sopt->sopt_size, MLEN);  
   
         return m;          sounlock(so);
           return (error);
 }  }
   
 void  void

Legend:
Removed from v.1.164.2.2  
changed lines
  Added in v.1.166

CVSweb <webmaster@jp.NetBSD.org>