text: /*
   * Copyright (c) 2004-2006 Maxim Sobolev <sobomax@FreeBSD.org>
   * Copyright (c) 2006-2009 Sippy Software, Inc., http://www.sippysoft.com
   * All rights reserved.
   *
   * Redistribution and use in source and binary forms, with or without
   * modification, are permitted provided that the following conditions
   * are met:
   * 1. Redistributions of source code must retain the above copyright
   *    notice, this list of conditions and the following disclaimer.
   * 2. Redistributions in binary form must reproduce the above copyright
   *    notice, this list of conditions and the following disclaimer in the
   *    documentation and/or other materials provided with the distribution.
   *
   * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
   * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
   * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
   * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
   * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
   * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
   * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
   * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
   * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
   * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
   * SUCH DAMAGE.
   *
   */
  
  #include <sys/types.h>
  #include <sys/socket.h>
  #include <netinet/in.h>
  #include <arpa/inet.h>
  #include <assert.h>
  #include <ctype.h>
  #include <errno.h>
  #include <netdb.h>
  #include <pthread.h>
  #include <stdio.h>
  #include <stdlib.h>
  #include <string.h>
  #include <unistd.h>
  
  #include "rtpp_defines.h"
  #include "rtpp_network.h"
  #include "rtpp_util.h"
  
  int
  ishostseq(struct sockaddr *ia1, struct sockaddr *ia2)
  {
  
      if (ia1->sa_family != ia2->sa_family)
  	return 0;
  
      switch (ia1->sa_family) {
      case AF_INET:
  	return (satosin(ia1)->sin_addr.s_addr ==
  	  satosin(ia2)->sin_addr.s_addr);
  
      case AF_INET6:
  	return (memcmp(&satosin6(ia1)->sin6_addr.s6_addr[0],
  	  &satosin6(ia2)->sin6_addr.s6_addr[0],
  	  sizeof(struct in6_addr)) == 0);
  
      default:
  	break;
      }
      /* Can't happen */
      abort();
  }
  
  int
  ishostnull(struct sockaddr *ia)
  {
      struct in6_addr *ap;
  
      switch (ia->sa_family) {
      case AF_INET:
  	return (satosin(ia)->sin_addr.s_addr == INADDR_ANY);
  
      case AF_INET6:
  	ap = &satosin6(ia)->sin6_addr;
  	return ((*(const uint32_t *)(const void *)(&ap->s6_addr[0]) == 0) &&
  		(*(const uint32_t *)(const void *)(&ap->s6_addr[4]) == 0) &&
  		(*(const uint32_t *)(const void *)(&ap->s6_addr[8]) == 0) &&
  		(*(const uint32_t *)(const void *)(&ap->s6_addr[12]) == 0));
  
      default:
  	break;
      }
  
      abort();
  }
  
  uint16_t
  getport(struct sockaddr *ia)
  {
  
      return (ntohs(getnport(ia)));
  }
  
  uint16_t
  getnport(struct sockaddr *ia)
  {
  
      switch (ia->sa_family) {
      case AF_INET:
          return (satosin(ia)->sin_port);
  
      case AF_INET6:
          return (satosin6(ia)->sin6_port);
  
      default:
          break;
      }
      /* Can't happen */
      abort();
  }
  
  int
  isaddrseq(struct sockaddr *ia1, struct sockaddr *ia2)
  {
  
      if (ishostseq(ia1, ia2) == 0)
          return (0);
      return (getport(ia1) == getport(ia2));
  }
  
  void
  setport(struct sockaddr *ia, int portnum)
  {
  
      assert(IS_VALID_PORT(portnum));
  
      switch (ia->sa_family) {
      case AF_INET:
          satosin(ia)->sin_port = htons(portnum);
          return;
  
      case AF_INET6:
          satosin6(ia)->sin6_port = htons(portnum);
          return;
  
      default:
          break;
      }
      /* Can't happen */
      abort();
  }
  
  void
  setanyport(struct sockaddr *ia)
  {
  
      switch (ia->sa_family) {
      case AF_INET:
          satosin(ia)->sin_port = 0;
          return;
  
      case AF_INET6:
          satosin6(ia)->sin6_port = 0;
          return;
  
      default:
          break;
      }
      /* Can't happen */
      abort();
  }
  
  char *
  addr2char_r(struct sockaddr *ia, char *buf, int size)
  {
      void *addr;
  
      switch (ia->sa_family) {
      case AF_INET:
  	addr = &(satosin(ia)->sin_addr);
  	break;
  
      case AF_INET6:
  	addr = &(satosin6(ia)->sin6_addr);
  	break;
  
      default:
  	abort();
      }
  
      return (char *)((void *)inet_ntop(ia->sa_family, addr, buf, size));
  }
  
  char *
  addrport2char_r(struct sockaddr *ia, char *buf, int size)
  {
      char abuf[MAX_ADDR_STRLEN];
      const char *bs, *es;
  
      switch (ia->sa_family) {
      case AF_INET:
          bs = es = "";
          break;
  
      case AF_INET6:
          bs = "[";
          es = "]";
          break;
  
      default:
          abort();
      }
  
      if (addr2char_r(ia, abuf, MAX_ADDR_STRLEN) == NULL)
          return (NULL);
      snprintf(buf, size, "%s%s%s:%u", bs, abuf, es, getport(ia));
      return (buf);
  }
  
  int
  resolve(struct sockaddr *ia, int pf, const char *host,
    const char *servname, int flags)
  {
      int n;
      struct addrinfo hints, *res;
  
      memset(&hints, 0, sizeof(hints));
      hints.ai_flags = flags;	     /* We create listening sockets */
      hints.ai_family = pf;	       /* Protocol family */
      hints.ai_socktype = SOCK_DGRAM;     /* UDP */
  
      n = getaddrinfo(host, servname, &hints, &res);
      if (n == 0) {
  	/* Use the first socket address returned */
  	memcpy(ia, res->ai_addr, res->ai_addrlen);
  	freeaddrinfo(res);
      }
  
      return n;
  }
  
  #if !defined(BYTE_ORDER)
  # error "BYTE_ORDER needs to be defined"
  #endif
  
  /*
   * Checksum routine for Internet Protocol family headers.
   */
  uint16_t
  rtpp_in_cksum(void *p, int len)
  {
      int sum = 0, oddbyte = 0, v = 0;
      u_char *cp = p;
  
      /* we assume < 2^16 bytes being summed */
      while (len > 0) {
          if (oddbyte) {
              sum += v + *cp++;
              len--;
          }
          if (((long)cp & 1) == 0) {
              while ((len -= 2) >= 0) {
                  sum += *(u_short *)cp;
                  cp += 2;
              }
          } else {
              while ((len -= 2) >= 0) {
  #if BYTE_ORDER == BIG_ENDIAN
                  sum += *cp++ << 8;
                  sum += *cp++;
  #else
                  sum += *cp++;
                  sum += *cp++ << 8;
  #endif
              }
          }
          if ((oddbyte = len & 1) != 0) {
  #if BYTE_ORDER == BIG_ENDIAN
              v = *cp << 8;
  #else
              v = *cp;
  #endif
          }
      }
      if (oddbyte)
          sum += v;
      sum = (sum >> 16) + (sum & 0xffff); /* add in accumulated carries */
      sum += sum >> 16;               /* add potential last carry */
      return (0xffff & ~sum);
  }
  
  int
  local4remote(struct sockaddr *ra, struct sockaddr_storage *la)
  {
      int s, r;
      socklen_t llen;
  
      s = socket(ra->sa_family, SOCK_DGRAM, 0);
      if (s == -1) {
          return (-1);
      }
      if (connect(s, ra, SA_LEN(ra)) == -1) {
          close(s);
          return (-1);
      }
      llen = sizeof(*la);
      r = getsockname(s, sstosa(la), &llen);
      close(s);
      return (r);
  }
  
  int
  extractaddr(const char *str, char **begin, char **end, int *pf)
  {
      const char *t;
      int tpf;
  
      if (*str != '[') {
  	tpf = AF_INET;
  	for (t = str; *str != '\0'; str++) {
  	    if (!isdigit(*str) && *str != '.')
  		break;
  	}
      } else {
  	tpf = AF_INET6;
  	str++;
  	for (t = str; *str != '\0'; str++) {
  	    if (!isxdigit(*str) && *str != ':')
  		break;
  	}
  	if (*str != ']')
  	    return (-1);
      }
      if (t == str)
  	return (-1);
      if (tpf == AF_INET6)
  	*end = (char *)(str + 1);
      else
  	*end = (char *)str;
      *pf = tpf;
      *begin = (char *)t;
      return(str - t);
  }
  
  int
  setbindhost(struct sockaddr *ia, int pf, const char *bindhost,
    const char *servname)
  {
      int n;
  
      /*
       * If user specified * then change it to NULL,
       * that will make getaddrinfo to return addr_any socket
       */
      if (bindhost && (strcmp(bindhost, "*") == 0))
  	bindhost = NULL;
  
      if ((n = resolve(ia, pf, bindhost, servname, AI_PASSIVE)) != 0) {
  	warnx("setbindhost: %s for %s %s", gai_strerror(n), bindhost, servname);
  	return -1;
      }
      return 0;
  }
  
  ssize_t
  recvfromto(int s, void *buf, size_t len, struct sockaddr *from,
    socklen_t *fromlen, struct sockaddr *to, socklen_t *tolen,
    struct timeval *timeptr)
  {
      /* We use a union to make sure hdr is aligned */
      union {
          struct cmsghdr hdr;
          unsigned char buf[CMSG_SPACE(1024)];
      } cmsgbuf;
  #if !defined(__FreeBSD__)
      struct in_pktinfo *pktinfo;
  #endif
      struct cmsghdr *cmsg;
      struct msghdr msg;
      struct iovec iov;
      ssize_t rval;
  
      memset(&msg, '\0', sizeof(msg));
      iov.iov_base = buf;
      iov.iov_len = len;
      msg.msg_name = from;
      msg.msg_namelen = *fromlen;
      msg.msg_iov = &iov;
      msg.msg_iovlen = 1;
      msg.msg_control = cmsgbuf.buf;
      msg.msg_controllen = sizeof(cmsgbuf.buf);
  
      rval = recvmsg(s, &msg, 0);
      if (rval < 0)
          return (rval);
  
      *tolen = 0;
      for (cmsg = CMSG_FIRSTHDR(&msg); cmsg != NULL;
        cmsg = CMSG_NXTHDR(&msg, cmsg)) {
  #if defined(__FreeBSD__)
          if (cmsg->cmsg_level == IPPROTO_IP &&
            cmsg->cmsg_type == IP_RECVDSTADDR) {
              memcpy(&satosin(to)->sin_addr, CMSG_DATA(cmsg),
                sizeof(struct in_addr));
              to->sa_family = AF_INET;
              *tolen = sizeof(struct sockaddr_in);
              break;
          }
  #else
          if (cmsg->cmsg_level == SOL_IP &&
            cmsg->cmsg_type == IP_PKTINFO) {
              pktinfo = (struct in_pktinfo *)CMSG_DATA(cmsg);
              memcpy(&satosin(to)->sin_addr, &pktinfo->ipi_addr,
                sizeof(struct in_addr));
              to->sa_family = AF_INET;
              *tolen = sizeof(struct sockaddr_in);
              break;
          }
  #endif
          if ((cmsg->cmsg_level == SOL_SOCKET)
            && (cmsg->cmsg_type == SCM_TIMESTAMP)) {
              memcpy(timeptr, CMSG_DATA(cmsg), sizeof(struct timeval));
          }
      }
      *fromlen = msg.msg_namelen;
      return (rval);
  }
  