/*

 * ntp_net.h - definitions for NTP network stuff

 */

#ifndef GUARD_NTP_NET_H

#define GUARD_NTP_NET_H

#include <sys/types.h>

#include <sys/socket.h>

#include <net/if.h>

#include <netinet/in.h>

#include <netdb.h>

#include "isc_netaddr.h"

#include "ntp_malloc.h"

typedef union {       /* On Linux, these come from: */

  struct sockaddr   sa; /* /usr/include/bits/socket.h */

  struct sockaddr_in  sa4;  /* /usr/include/linux/in.h */

  struct sockaddr_in6 sa6;  /* /usr/include/linux/in6.h */

} sockaddr_u;

/*

 * Utilities for manipulating sockaddr_u v4/v6 unions

 */

#define SOCK_ADDR4(psau)  ((psau)->sa4.sin_addr)

#define SET_SOCK_ADDR4(psau, a) ((psau)->sa4.sin_addr = (a))

#define SOCK_ADDR6(psau)  ((psau)->sa6.sin6_addr)

#define SET_SOCK_ADDR6(psau, a) ((psau)->sa6.sin6_addr = (a))

#define PSOCK_ADDR4(psau) (&SOCK_ADDR4(psau))

#define PSOCK_ADDR6(psau) (&SOCK_ADDR6(psau))

#define AF(psau)    ((psau)->sa.sa_family)

#define SET_AF(psau, f)   ((psau)->sa.sa_family = (f))

#define IS_IPV4(psau)   (AF_INET == AF(psau))

#define IS_IPV6(psau)   (AF_INET6 == AF(psau))

/* sockaddr_u v4 address in network byte order */

#define NSRCADR(psau)   (SOCK_ADDR4(psau).s_addr)

#define SET_NSRCADR(psau, a)  (SOCK_ADDR4(psau).s_addr - (a))

/* sockaddr_u v4 address in host byte order */

#define SRCADR(psau)    (ntohl(NSRCADR(psau)))

/* sockaddr_u v6 address in network byte order */

#define NSRCADR6(psau)    (SOCK_ADDR6(psau).s6_addr)

#define SET_NSRCADR6(psau)  (SOCK_ADDR6(psau).s6_addr = (a))

/* assign sockaddr_u v4 address from host byte order */

#define SET_ADDR4(psau, addr4)  (NSRCADR(psau) = htonl(addr4))

/* assign sockaddr_u v4 address from network byte order */

#define SET_ADDR4N(psau, addr4n) (NSRCADR(psau) = (addr4n));

/* assign sockaddr_u v6 address from network byte order */

#define SET_ADDR6N(psau, s6_addr) (SOCK_ADDR6(psau) = (s6_addr))

/* sockaddr_u v4/v6 port in network byte order */

#define NSRCPORT(psau)    ((psau)->sa4.sin_port)

#define SET_NSRCPORT(psau, a) ((psau)->sa4.sin_port = (a))

/* sockaddr_u v4/v6 port in host byte order */

#define SRCPORT(psau)   (ntohs(NSRCPORT(psau)))

/* assign sockaddr_u v4/v6 port from host byte order */

#define SET_PORT(psau, port)  (NSRCPORT(psau) = htons(port))

/* sockaddr_u v6 scope */

#define SCOPE_VAR(psau)   ((psau)->sa6.sin6_scope_id)

/* v4/v6 scope (always zero for v4) */

# define SCOPE(psau)    (IS_IPV4(psau)      \

            ? 0       \

            : SCOPE_VAR(psau))

/* are two v6 sockaddr_u scopes equal? */

# define SCOPE_EQ(psau1, psau2)         \

  (SCOPE_VAR(psau1) == SCOPE_VAR(psau2))

/* assign scope if supported */

# define SET_SCOPE(psau, s)         \

  do              \

    if (IS_IPV6(psau))        \

      SCOPE_VAR(psau) = (s);      \

  while (0)

/* v4/v6 is multicast address */

#define IS_MCAST(psau)            \

  (IS_IPV4(psau)            \

      ? IN_CLASSD(SRCADR(psau))       \

      : IN6_IS_ADDR_MULTICAST(PSOCK_ADDR6(psau)))

/* v6 is interface ID scope universal, as with MAC-derived addresses */

#define IS_IID_UNIV(psau)         \

  (!!(0x02 & NSRCADR6(psau)[8]))

#define SIZEOF_INADDR(fam)          \

  ((AF_INET == (fam))         \

      ? sizeof(struct in_addr)        \

      : sizeof(struct in6_addr))

#define SIZEOF_SOCKADDR(fam)          \

  ((AF_INET == (fam))         \

      ? sizeof(struct sockaddr_in)      \

      : sizeof(struct sockaddr_in6))

#define SOCKLEN(psau)           \

  (IS_IPV4(psau)            \

      ? sizeof((psau)->sa4)       \

      : sizeof((psau)->sa6))

#define ZERO_SOCK(psau)           \

  ZERO(*(psau))

/* blast a byte value across sockaddr_u v6 address */

#define MEMSET_ADDR6(psau, v)         \

  memset((psau)->sa6.sin6_addr.s6_addr, (v),    \

    sizeof((psau)->sa6.sin6_addr.s6_addr))

#define SET_ONESMASK(psau)          \

  do {              \

    if (IS_IPV6(psau))        \

      MEMSET_ADDR6((psau), 0xff);   \

    else            \

      NSRCADR(psau) = 0xffffffff;   \

  } while(0)

/* zero sockaddr_u, fill in family and all-ones (host) mask */

#define SET_HOSTMASK(psau, family)        \

  do {              \

    ZERO_SOCK(psau);        \

    AF(psau) = (family);        \

    SET_ONESMASK(psau);       \

  } while (0)

/*

 * compare two in6_addr returning negative, 0, or positive.

 * ADDR6_CMP is negative if *pin6A is lower than *pin6B, zero if they

 * are equal, positive if *pin6A is higher than *pin6B.  IN6ADDR_ANY

 * is the lowest address (128 zero bits).

 */

#define ADDR6_CMP(pin6A, pin6B)         \

  memcmp((pin6A)->s6_addr, (pin6B)->s6_addr,    \

         sizeof(pin6A)->s6_addr)

/* compare two in6_addr for equality only */

#define ADDR6_EQ(pin6A, pin6B)          \

  (!ADDR6_CMP(pin6A, pin6B))

/* compare a in6_addr with socket address */

#define S_ADDR6_EQ(psau, pin6)          \

  ADDR6_EQ(&(psau)->sa6.sin6_addr, pin6)

/* are two sockaddr_u's addresses equal? (port excluded) */

#define SOCK_EQ(psau1, psau2)         \

  ((AF(psau1) != AF(psau2))       \

       ? 0            \

       : IS_IPV4(psau1)         \

       ? (NSRCADR(psau1) == NSRCADR(psau2))   \

       : (S_ADDR6_EQ((psau1), PSOCK_ADDR6(psau2)) \

          && SCOPE_EQ((psau1), (psau2))))

/* are two sockaddr_u's addresses and ports equal? */

#define ADDR_PORT_EQ(psau1, psau2)        \

  ((NSRCPORT(psau1) != NSRCPORT(psau2)      \

       ? 0            \

       : SOCK_EQ((psau1), (psau2))))

/* is sockaddr_u address unspecified? */

#define SOCK_UNSPEC(psau)         \

  (IS_IPV4(psau)            \

      ? !NSRCADR(psau)          \

      : IN6_IS_ADDR_UNSPECIFIED(PSOCK_ADDR6(psau)))

/* just how unspecified do you mean? (scope 0/unspec too) */

#define SOCK_UNSPEC_S(psau)         \

  (SOCK_UNSPEC(psau) && !SCOPE(psau))

/*

 * Macro for checking for invalid addresses.  This is really, really

 * gross, but is needed so no one configures a host on net 127 now that

 * we're encouraging it the configuration file.

 */

#define LOOPBACKADR 0x7f000001

#define LOOPNETMASK 0xff000000

#define ISBADADR(srcadr)          \

  (IS_IPV4(srcadr)          \

   && ((SRCADR(srcadr) & LOOPNETMASK)     \

       == (LOOPBACKADR & LOOPNETMASK))      \

   && SRCADR(srcadr) != LOOPBACKADR)

#endif /* GUARD_NTP_NET_H */