[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.170 and 1.171

version 1.170, 2008/08/04 03:55:47 version 1.171, 2008/08/06 15:01:23
Line 419  getsombuf(struct socket *so, int type)
Line 419  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 1575  sorflush(struct socket *so)
Line 1564  sorflush(struct socket *so)
         sbrelease(&asb, so);          sbrelease(&asb, so);
 }  }
   
   /*
    * internal set SOL_SOCKET options
    */
 static int  static int
 sosetopt1(struct socket *so, int level, int optname, struct mbuf *m)  sosetopt1(struct socket *so, const struct sockopt *sopt)
 {  {
 #ifdef INET          int error, optval;
         int error, optval, val;          struct linger l;
 #else          struct timeval tv;
         int optval, val;  
 #endif  
         struct linger   *l;  
         struct sockbuf  *sb;  
         struct timeval *tv;  
   
         switch (optname) {          switch (sopt->sopt_name) {
   
 #ifdef INET  #ifdef INET
         case SO_ACCEPTFILTER:          case SO_ACCEPTFILTER:
                 error = do_setopt_accept_filter(so, m);                  error = do_setopt_accept_filter(so, sopt);
                 if (error)                  if (error)
                         return error;                          return error;
                 break;                  break;
 #endif  #endif
   
         case SO_LINGER:          case SO_LINGER:
                 if (m == NULL || m->m_len != sizeof(struct linger))                  error = sockopt_get(sopt, &l, sizeof(l));
                         return EINVAL;                  if (error)
                 l = mtod(m, struct linger *);                          return (error);
                 if (l->l_linger < 0 || l->l_linger > USHRT_MAX ||  
                     l->l_linger > (INT_MAX / hz))                  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 1620  sosetopt1(struct socket *so, int level, 
Line 1609  sosetopt1(struct socket *so, int level, 
         case SO_REUSEPORT:          case SO_REUSEPORT:
         case SO_OOBINLINE:          case SO_OOBINLINE:
         case SO_TIMESTAMP:          case SO_TIMESTAMP:
                 if (m == NULL || m->m_len < sizeof(int))                  error = sockopt_getint(sopt, &optval);
                         return EINVAL;                  if (error)
                 if (*mtod(m, int *))                          return (error);
                         so->so_options |= optname;  
                   if (optval)
                           so->so_options |= sopt->sopt_name;
                 else                  else
                         so->so_options &= ~optname;                          so->so_options &= ~sopt->sopt_name;
                 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:
                 if (m == NULL || m->m_len < sizeof(int))                  error = sockopt_getint(sopt, &optval);
                         return EINVAL;                  if (error)
                           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 (optname) {                  switch (sopt->sopt_name) {
   
                 case SO_SNDBUF:                  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_RCVBUF:                  case SO_RCVBUF:
                         sb = (optname == SO_SNDBUF) ?                          if (sbreserve(&so->so_rcv, (u_long)optval, so) == 0)
                             &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 1659  sosetopt1(struct socket *so, int level, 
Line 1654  sosetopt1(struct socket *so, int level, 
                  * the high-water.                   * the high-water.
                  */                   */
                 case SO_SNDLOWAT:                  case SO_SNDLOWAT:
                         so->so_snd.sb_lowat =                          if (optval > so->so_snd.sb_hiwat)
                             (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:
                         so->so_rcv.sb_lowat =                          if (optval > so->so_rcv.sb_hiwat)
                             (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:
                 if (m == NULL || m->m_len < sizeof(*tv))                  error = sockopt_get(sopt, &tv, sizeof(tv));
                         return EINVAL;                  if (error)
                 tv = mtod(m, struct timeval *);                          return (error);
                 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;  
   
                 switch (optname) {                  optval = tv.tv_sec * hz + tv.tv_usec / tick;
                   if (optval == 0 && tv.tv_usec != 0)
                           optval = 1;
   
                   switch (sopt->sopt_name) {
                 case SO_SNDTIMEO:                  case SO_SNDTIMEO:
                         so->so_snd.sb_timeo = val;                          so->so_snd.sb_timeo = optval;
                         break;                          break;
                 case SO_RCVTIMEO:                  case SO_RCVTIMEO:
                         so->so_rcv.sb_timeo = val;                          so->so_rcv.sb_timeo = optval;
                         break;                          break;
                 }                  }
                 break;                  break;
Line 1700  sosetopt1(struct socket *so, int level, 
Line 1699  sosetopt1(struct socket *so, int level, 
 }  }
   
 int  int
 sosetopt(struct socket *so, int level, int optname, struct mbuf *m)  sosetopt(struct socket *so, struct sockopt *sopt)
 {  {
         int error, prerr;          int error, prerr;
   
         solock(so);          solock(so);
         if (level == SOL_SOCKET)          if (sopt->sopt_level == SOL_SOCKET)
                 error = sosetopt1(so, level, optname, m);                  error = sosetopt1(so, sopt);
         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, level,                  prerr = (*so->so_proto->pr_ctloutput)(PRCO_SETOPT, so, sopt);
                     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
 sogetopt(struct socket *so, int level, int optname, struct mbuf **mp)  so_setsockopt(struct lwp *l, struct socket *so, int level, int name,
       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 (level != SOL_SOCKET) {          if (sopt->sopt_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, level, optname, mp));                              (PRCO_GETOPT, so, sopt));
                 } else                  } else
                         error = (ENOPROTOOPT);                          error = (ENOPROTOOPT);
         } else {          } else {
                 m = m_get(M_WAIT, MT_SOOPTS);                  error = sogetopt1(so, sopt);
                 m->m_len = sizeof(int);          }
           sounlock(so);
           return (error);
   }
   
                 switch (optname) {  /*
    * alloc sockopt data buffer buffer
    *      - will be released at destroy
    */
   static void
   sockopt_alloc(struct sockopt *sopt, size_t len)
   {
   
 #ifdef INET          KASSERT(sopt->sopt_size == 0);
                 case SO_ACCEPTFILTER:  
                         error = do_getopt_accept_filter(so, m);  
                         break;  
 #endif  
   
                 case SO_LINGER:          if (len > sizeof(sopt->sopt_buf))
                         m->m_len = sizeof(struct linger);                  sopt->sopt_data = malloc(len, M_SOOPTS, M_WAITOK | M_ZERO);
                         mtod(m, struct linger *)->l_onoff =          else
                             (so->so_options & SO_LINGER) ? 1 : 0;                  sopt->sopt_data = sopt->sopt_buf;
                         mtod(m, struct linger *)->l_linger = so->so_linger;  
                         break;  
   
                 case SO_USELOOPBACK:          sopt->sopt_size = len;
                 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:  /*
                         *mtod(m, int *) = so->so_type;   * initialise sockopt storage
                         break;   */
   void
   sockopt_init(struct sockopt *sopt, int level, int name, size_t size)
   {
   
                 case SO_ERROR:          memset(sopt, 0, sizeof(*sopt));
                         *mtod(m, int *) = so->so_error;  
                         so->so_error = 0;  
                         break;  
   
                 case SO_SNDBUF:          sopt->sopt_level = level;
                         *mtod(m, int *) = so->so_snd.sb_hiwat;          sopt->sopt_name = name;
                         break;          sockopt_alloc(sopt, size);
   }
   
                 case SO_RCVBUF:  /*
                         *mtod(m, int *) = so->so_rcv.sb_hiwat;   * destroy sockopt storage
                         break;   *      - will release any held memory references
    */
   void
   sockopt_destroy(struct sockopt *sopt)
   {
   
                 case SO_SNDLOWAT:          if (sopt->sopt_data != sopt->sopt_buf)
                         *mtod(m, int *) = so->so_snd.sb_lowat;                  free(sopt->sopt_data, M_SOOPTS);
                         break;  
   
                 case SO_RCVLOWAT:          memset(sopt, 0, sizeof(*sopt));
                         *mtod(m, int *) = so->so_rcv.sb_lowat;  }
                         break;  
   
                 case SO_SNDTIMEO:  /*
                 case SO_RCVTIMEO:   * set sockopt value
                     {   *      - value is copied into sockopt
                         int val = (optname == SO_SNDTIMEO ?   *      - memory is allocated when necessary
                              so->so_snd.sb_timeo : so->so_rcv.sb_timeo);   */
   int
                         m->m_len = sizeof(struct timeval);  sockopt_set(struct sockopt *sopt, const void *buf, size_t len)
                         mtod(m, struct timeval *)->tv_sec = val / hz;  {
                         mtod(m, struct timeval *)->tv_usec =  
                             (val % hz) * tick;  
                         break;  
                     }  
   
                 case SO_OVERFLOWED:          if (sopt->sopt_size == 0)
                         *mtod(m, int *) = so->so_rcv.sb_overflowed;                  sockopt_alloc(sopt, len);
                         break;  
   
                 default:          KASSERT(sopt->sopt_size == len);
                         sounlock(so);          memcpy(sopt->sopt_data, buf, len);
                         (void)m_free(m);          return 0;
                         return (ENOPROTOOPT);  }
                 }  
                 *mp = m;  /*
                 error = 0;   * common case of set sockopt integer value
    */
   int
   sockopt_setint(struct sockopt *sopt, int val)
   {
   
           return sockopt_set(sopt, &val, sizeof(int));
   }
   
   /*
    * get sockopt value
    *      - correct size must be given
    */
   int
   sockopt_get(const struct sockopt *sopt, void *buf, size_t len)
   {
   
           if (sopt->sopt_size != len)
                   return EINVAL;
   
           memcpy(buf, sopt->sopt_data, len);
           return 0;
   }
   
   /*
    * common case of get sockopt integer value
    */
   int
   sockopt_getint(const struct sockopt *sopt, int *valp)
   {
   
           return sockopt_get(sopt, valp, sizeof(int));
   }
   
   /*
    * set sockopt value from mbuf
    *      - ONLY for legacy code
    *      - mbuf is released by sockopt
    */
   int
   sockopt_setmbuf(struct sockopt *sopt, struct mbuf *m)
   {
           size_t len;
   
           len = m_length(m);
   
           if (sopt->sopt_size == 0)
                   sockopt_alloc(sopt, len);
   
           KASSERT(sopt->sopt_size == len);
           m_copydata(m, 0, len, sopt->sopt_data);
           m_freem(m);
   
           return 0;
   }
   
   /*
    * get sockopt value into mbuf
    *      - ONLY for legacy code
    *      - 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);
           if (m == NULL)
                   return NULL;
   
           m->m_len = MLEN;
           m_copyback(m, 0, sopt->sopt_size, sopt->sopt_data);
           if (m_length(m) != max(sopt->sopt_size, MLEN)) {
                   m_freem(m);
                   return NULL;
         }          }
           m->m_len = min(sopt->sopt_size, MLEN);
   
         sounlock(so);          return m;
         return (error);  
 }  }
   
 void  void

Legend:
Removed from v.1.170  
changed lines
  Added in v.1.171

CVSweb <webmaster@jp.NetBSD.org>