[adns] / src / addrfam.c

/*
 * addrfam.c
 * - address-family specific code
 */
/*
 *  This file is part of adns, which is
 *    Copyright (C) 1997-2000,2003,2006,2014  Ian Jackson
 *    Copyright (C) 2014  Mark Wooding
 *    Copyright (C) 1999-2000,2003,2006  Tony Finch
 *    Copyright (C) 1991 Massachusetts Institute of Technology
 *  (See the file INSTALL for full details.)
 *  
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 3, or (at your option)
 *  any later version.
 *  
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *  
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software Foundation.
 */

#include <stdlib.h>
#include <errno.h>
#include <limits.h>
#include <unistd.h>
#include <inttypes.h>
#include <stddef.h>
#include <stdbool.h>

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <net/if.h>

#include "internal.h"

/*
 * General address-family operations.
 */

#define SIN(cnst, sa) ((void)(sa)->sa_family, (cnst struct sockaddr_in *)(sa))
#define SIN6(cnst, sa) ((void)(sa)->sa_family, (cnst struct sockaddr_in6 *)(sa))

static void unknown_af(int af) NONRETURNING;
static void unknown_af(int af) {
  fprintf(stderr, "ADNS INTERNAL: unknown address family %d\n", af);
  abort();
}

/*
 * SOCKADDR_IN_IN6(CNST, struct sockaddr *sa, SIN, {
 *     // struct sockaddr_in *const SIN; // implicitly
 *     code for inet;
 *   }, {
 *     // struct sockaddr_in6 *const SIN6; // implicitly
 *     code for inet6;
 *   })
 *
 * SOCKADDR_IN_IN6_PAIR(CNST, struct sockaddr *sa, SINA,
 *                            struct sockaddr *sb, SINB, {
 *     // struct sockaddr_in *const SINA; // implicitly
 *     // struct sockaddr_in *const SINB; // implicitly
 *     code for inet;
 *   },{
 *     // struct sockaddr_in6 *const SINA6; // implicitly
 *     // struct sockaddr_in6 *const SINB6; // implicitly
 *     code for inet6;
 *   });
 *
 * SOCKADDR_IN_IN6_OTHER(CNST, struct sockaddr *sa, SIN, { in }, { in6 }, {
 *     code for other address family
 *   })
 *
 * AF_IN_IN6_OTHER(af, { in }, { in6 }, { other })
 *
 * Executes the first or second block according to the AF in sa.  CNST
 * may be `const' or empty.  For _PAIR, sa and sb must be same AF.
 *
 * All except _OTHER handle unknown AFs with unknown_af.
 *
 * Code blocks may not contain , outside parens.
 */
#define AF_IN_IN6_OTHER(af, for_inet, for_inet6, other)	\
  if ((af) == AF_INET) {					\
    for_inet							\
  } else if ((af) == AF_INET6) {				\
    for_inet6							\
  } else {							\
    other							\
  }
#define SOCKADDR_IN_IN6_OTHER(cnst, sa, sin, for_inet, for_inet6, other) \
  AF_IN_IN6_OTHER((sa)->sa_family, {					\
      cnst struct sockaddr_in *const sin = SIN(cnst,(sa));		\
      for_inet								\
  }, {									\
      cnst struct sockaddr_in6 *const sin##6 = SIN6(cnst,(sa));		\
      for_inet6								\
  },									\
    other								\
  )
#define SOCKADDR_IN_IN6(cnst, sa, sin, for_inet, for_inet6)		\
  SOCKADDR_IN_IN6_OTHER(cnst, sa, sin, for_inet, for_inet6, {	\
      unknown_af((sa)->sa_family);					\
  })
#define SOCKADDR_IN_IN6_PAIR(cnst, sa, sina, sb, sinb, for_inet, for_inet6) \
  do{									\
    assert((sa)->sa_family == (sb)->sa_family);				\
    SOCKADDR_IN_IN6(cnst, sa, sina, {					\
        cnst struct sockaddr_in *const sinb = SIN(cnst,(sb));		\
	for_inet							\
      }, {								\
        cnst struct sockaddr_in6 *const sinb##6 = SIN6(cnst,(sb));	\
	for_inet6							\
    });									\
  }while(0)

int adns__addrs_equal_raw(const struct sockaddr *a,
			 int bf, const void *b) {
  if (a->sa_family != bf) return 0;

  SOCKADDR_IN_IN6(const, a, sin, {
    return sin->sin_addr.s_addr == ((const struct in_addr*)b)->s_addr;
  }, {
    return !memcmp(&sin6->sin6_addr, b, sizeof(struct in6_addr));
  });
}

int adns__addrs_equal(const adns_sockaddr *a, const adns_sockaddr *b) {
  return adns__addrs_equal_raw(&a->sa, b->sa.sa_family,
			       adns__sockaddr_addr(&b->sa));
}

int adns__sockaddrs_equal(const struct sockaddr *sa,
			  const struct sockaddr *sb) {
  if (!adns__addrs_equal_raw(sa, sb->sa_family, adns__sockaddr_addr(sb)))
    return 0;
  SOCKADDR_IN_IN6_PAIR(const, sa, sina, sb, sinb, {
      return sina->sin_port == sinb->sin_port;
    }, {
      return sina6->sin6_port == sinb6->sin6_port &&
             sina6->sin6_scope_id == sinb6->sin6_scope_id;
    });
}

int adns__addr_width(int af) {
  AF_IN_IN6_OTHER(af, {
      return 32;
    }, {
      return 128;
    }, {
      unknown_af(af);
    });
}

void adns__prefix_mask(adns_sockaddr *sa, int len) {
  SOCKADDR_IN_IN6(, &sa->sa, sin, {
      assert(len <= 32);
      sin->sin_addr.s_addr= htonl(!len ? 0 : 0xffffffff << (32-len));
    }, {
      int i= len/8;
      int j= len%8;
      unsigned char *m= sin6->sin6_addr.s6_addr;
      assert(len <= 128);
      memset(m, 0xff, i);
      if (j) m[i++]= (0xff << (8-j)) & 0xff;
      memset(m+i, 0, 16-i);
    });
}

int adns__guess_prefix_length(const adns_sockaddr *sa) {
  SOCKADDR_IN_IN6(const, &sa->sa, sin, {
      unsigned a= (ntohl(sin->sin_addr.s_addr) >> 24) & 0xff;
      if (a < 128) return 8;
      else if (a < 192) return 16;
      else if (a < 224) return 24;
      else return -1;
    }, {
      (void)sin6;
      return 64;
    });
}

int adns__addr_matches(int af, const void *addr,
		       const adns_sockaddr *base, const adns_sockaddr *mask)
{
  if (af != base->sa.sa_family) return 0;
  SOCKADDR_IN_IN6_PAIR(const, &base->sa, sbase, &mask->sa, smask, {
      const struct in_addr *v4 = addr;
      return (v4->s_addr & smask->sin_addr.s_addr)
	== sbase->sin_addr.s_addr;
    }, {
      int i;
      const char *a= addr;
      const char *b= sbase6->sin6_addr.s6_addr;
      const char *m= smask6->sin6_addr.s6_addr;
      for (i = 0; i < 16; i++)
	if ((a[i] & m[i]) != b[i]) return 0;
      return 1;
    });
}

const void *adns__sockaddr_addr(const struct sockaddr *sa) {
  SOCKADDR_IN_IN6(const, sa, sin, {
      return &sin->sin_addr;
    }, {
      return &sin6->sin6_addr;
    });
}

void adns__addr_inject(const void *a, adns_sockaddr *sa) {
  SOCKADDR_IN_IN6( , &sa->sa, sin, {
      memcpy(&sin->sin_addr, a, sizeof(sin->sin_addr));
    }, {
      memcpy(&sin6->sin6_addr, a, sizeof(sin6->sin6_addr));
    });
}

/*
 * addr2text and text2addr
 */

#define ADDRFAM_DEBUG
#ifdef ADDRFAM_DEBUG
static void af_debug_func(const char *fmt, ...) {
  int esave= errno;
  va_list al;
  va_start(al,fmt);
  vfprintf(stderr,fmt,al);
  va_end(al);
  errno= esave;
}
# define af_debug(fmt,...) \
  (af_debug_func("%s: " fmt "\n", __func__, __VA_ARGS__))
#else
# define af_debug(fmt,...) ((void)("" fmt "", __VA_ARGS__))
#endif

static bool addrtext_our_errno(int e) {
  return
    e==EAFNOSUPPORT ||
    e==EINVAL ||
    e==ENOSPC ||
    e==ENOSYS;
}

static bool addrtext_scope_use_ifname(const struct sockaddr *sa) {
  const struct in6_addr *in6= &SIN6(const,sa)->sin6_addr;
  return
    IN6_IS_ADDR_LINKLOCAL(in6) ||
    IN6_IS_ADDR_MC_LINKLOCAL(in6);
}

int adns_text2addr(const char *text, uint16_t port, adns_queryflags flags,
		   struct sockaddr *sa, socklen_t *salen_io) {
  int af;
  char copybuf[INET6_ADDRSTRLEN];
  const char *parse=text;
  const char *scopestr=0;
  socklen_t needlen;
  void *dst;
  uint16_t *portp;

#define INVAL(how) do{				\
  af_debug("invalid: %s: `%s'", how, text);	\
  return EINVAL;				\
}while(0)

#define AFCORE(INETx,SINx,sinx)			\
    af= AF_##INETx;				\
    dst = &SINx(,sa)->sinx##_addr;		\
    portp = &SINx(,sa)->sinx##_port;		\
    needlen= sizeof(*SINx(,sa));

  if (!strchr(text, ':')) { /* INET */

    AFCORE(INET,SIN,sin);

  } else { /* INET6 */

    AFCORE(INET6,SIN6,sin6);

    const char *percent= strchr(text, '%');
    if (percent) {
      ptrdiff_t lhslen = percent - text;
      if (lhslen >= INET6_ADDRSTRLEN) INVAL("scoped addr lhs too long");
      memcpy(copybuf, text, lhslen);
      copybuf[lhslen]= 0;

      parse= copybuf;
      scopestr= percent+1;

      af_debug("will parse scoped addr `%s' %% `%s'", parse, scopestr);
    }

  }

#undef AFCORE

  if (scopestr && (flags & adns_qf_addrlit_scope_forbid))
    INVAL("scoped addr but _scope_forbid");

  if (*salen_io < needlen) {
    *salen_io = needlen;
    return ENOSPC;
  }

  memset(sa, 0, needlen);

  sa->sa_family= af;
  *portp = htons(port);

  if (af == AF_INET && !(flags & adns_qf_addrlit_ipv4_quadonly)) {
    /* we have to use inet_aton to deal with non-dotted-quad literals */
    int r= inet_aton(parse,&SIN(,sa)->sin_addr);
    if (!r) INVAL("inet_aton rejected");
  } else {
    int r= inet_pton(af,parse,dst);
    if (!r) INVAL("inet_pton rejected");
    assert(r>0);
  }

  if (scopestr) {
    errno=0;
    char *ep;
    unsigned long scope= strtoul(scopestr,&ep,10);
    if (errno==ERANGE) INVAL("numeric scope id too large for unsigned long");
    assert(!errno);
    if (!*ep) {
      if (scope > ~(uint32_t)0)
	INVAL("numeric scope id too large for uint32_t");
    } else { /* !!*ep */
      if (flags & adns_qf_addrlit_scope_numeric)
	INVAL("non-numeric scope but _scope_numeric");
      if (!addrtext_scope_use_ifname(sa)) {
	af_debug("cannot convert non-numeric scope"
		 " in non-link-local addr `%s'", text);
	return ENOSYS;
      }
      errno= 0;
      scope= if_nametoindex(scopestr);
      if (!scope) {
	/* RFC3493 says "No errors are defined".  It's not clear
	 * whether that is supposed to mean if_nametoindex "can't
	 * fail" (other than by the supplied name not being that of an
	 * interface) which seems unrealistic, or that it conflates
	 * all its errors together by failing to set errno, or simply
	 * that they didn't bother to document the errors.
	 *
	 * glibc, FreeBSD and OpenBSD all set errno (to ENXIO when
	 * appropriate).  See Debian bug #749349.
	 *
	 * We attempt to deal with this by clearing errno to start
	 * with, and then perhaps mapping the results. */
	af_debug("if_nametoindex rejected scope name (errno=%s)",
		 strerror(errno));
	if (errno==0) {
	  return ENXIO;
	} else if (addrtext_our_errno(errno)) {
	  /* we use these for other purposes, urgh. */
	  perror("adns: adns_text2addr: if_nametoindex"
		 " failed with unexpected error");
	  return EIO;
	} else {
	  return errno;
	}
      } else { /* ix>0 */
	if (scope > ~(uint32_t)0) {
	  fprintf(stderr,"adns: adns_text2addr: if_nametoindex"
		  " returned an interface index >=2^32 which will not fit"
		  " in sockaddr_in6.sin6_scope_id");
	  return EIO;
	}
      }
    } /* else; !!*ep */

    SIN6(,sa)->sin6_scope_id= scope;
  } /* if (scopestr) */

  *salen_io = needlen;
  return 0;
}

int adns_addr2text(const struct sockaddr *sa, adns_queryflags flags,
		   char *buffer, int *buflen_io, int *port_r) {
  const void *src;
  int port;

  if (*buflen_io < ADNS_ADDR2TEXT_BUFLEN) {
    *buflen_io = ADNS_ADDR2TEXT_BUFLEN;
    return ENOSPC;
  }

  SOCKADDR_IN_IN6_OTHER(const, sa, sin, {
      src= &sin->sin_addr;    port= sin->sin_port;
    }, {
      src= &sin6->sin6_addr;  port= sin6->sin6_port;
    }, {
      return EAFNOSUPPORT;
    });

  const char *ok= inet_ntop(sa->sa_family, src, buffer, *buflen_io);
  assert(ok);

  if (sa->sa_family == AF_INET6) {
    uint32_t scope = SIN6(const,sa)->sin6_scope_id;
    if (scope) {
      if (flags & adns_qf_addrlit_scope_forbid)
	return EINVAL;
      int scopeoffset = strlen(buffer);
      int remain = *buflen_io - scopeoffset;
      char *scopeptr =  buffer + scopeoffset;
      assert(remain >= IF_NAMESIZE+1/*%*/);
      *scopeptr++= '%'; remain--;
      bool parsedname = 0;
      af_debug("will print scoped addr `%.*s' %% %"PRIu32"",
	       scopeoffset,buffer, scope);
      if (scope <= UINT_MAX /* so we can pass it to if_indextoname */
	  && !(flags & adns_qf_addrlit_scope_numeric)
	  && addrtext_scope_use_ifname(sa)) {
	parsedname = if_indextoname(scope, scopeptr);
	if (!parsedname) {
	  af_debug("if_indextoname rejected scope (errno=%s)",
		   strerror(errno));
	  if (errno==ENXIO) {
	    /* fair enough, show it as a number then */
	  } else if (addrtext_our_errno(errno)) {
	    /* we use these for other purposes, urgh. */
	    perror("adns: adns_addr2text: if_indextoname"
		   " failed with unexpected error");
	    return EIO;
	  } else {
	    return errno;
	  }
	}
      }
      if (!parsedname) {
	int r = snprintf(scopeptr, remain,
			 "%"PRIu32"", scope);
	assert(r < *buflen_io - scopeoffset);
      }
      af_debug("printed scoped addr `%s'", buffer);
    }
  }

  if (port_r) *port_r= ntohs(port);
  return 0;
}

char *adns__sockaddr_ntoa(const struct sockaddr *sa, char *buf) {
  int err;
  int len= ADNS_ADDR2TEXT_BUFLEN;

  err= adns_addr2text(sa, 0, buf, &len, 0);
  if (err == EIO)
    err= adns_addr2text(sa, adns_qf_addrlit_scope_numeric, buf, &len, 0);
  assert(!err);
  return buf;
}

/*
 * Reverse-domain parsing and construction.
 */

int adns__make_reverse_domain(const struct sockaddr *sa, const char *zone,
			      char **buf_io, size_t bufsz,
			      char **buf_free_r) {
  size_t req;
  char *p;
  unsigned c, y;
  unsigned long aa;
  const unsigned char *ap;
  int i, j;

  AF_IN_IN6_OTHER(sa->sa_family, {
      req= 4 * 4;
      if (!zone) zone= "in-addr.arpa";
    }, {
      req = 2 * 32;
      if (!zone) zone= "ip6.arpa";
    }, {
      return ENOSYS;
    });

  req += strlen(zone) + 1;
  if (req <= bufsz)
    p= *buf_io;
  else {
    p= malloc(req); if (!p) return errno;
    *buf_free_r = p;
  }

  *buf_io= p;
  SOCKADDR_IN_IN6(const, sa, sin, {
      aa= ntohl(sin->sin_addr.s_addr);
      for (i=0; i<4; i++) {
	p += sprintf(p, "%d", (int)(aa & 0xff));
	*p++= '.';
	aa >>= 8;
      }
    }, {
      ap= sin6->sin6_addr.s6_addr + 16;
      for (i=0; i<16; i++) {
	c= *--ap;
	for (j=0; j<2; j++) {
	  y= c & 0xf;
	  *p++= (y < 10) ? y + '0' : y - 10 + 'a';
	  c >>= 4;
	  *p++= '.';
	}
      }
    });

  strcpy(p, zone);
  return 0;
}


#define REVPARSE_P_L(labnum)			\
  const char *p= dgram + rps->labstart[labnum];	\
  int l= rps->lablen[labnum]
  /*
   * REVPARSE_P_L(int labnum);
   *   expects:
   *     const char *dgram;
   *     const struct revparse_state *rps;
   *   produces:
   *     const char *p; // start of label labnum in dgram
   *     int l; // length of label in dgram
   */

static bool revparse_check_tail(struct revparse_state *rps,
				const char *dgram, int nlabels,
				int bodylen, const char *inarpa) {
  int i;

  if (nlabels != bodylen+2) return 0;
  for (i=0; i<2; i++) {
    REVPARSE_P_L(bodylen+i);
    const char *want= !i ? inarpa : "arpa";
    if (!adns__labels_equal(p,l, want,strlen(want))) return 0;
  }
  return 1;
}

static bool revparse_atoi(const char *p, int l, int base,
			  unsigned max, unsigned *v_r) {
  if (l>3) return 0;
  if (l>1 && p[0]=='0') return 0;
  unsigned v=0;
  while (l-- > 0) {
    int tv;
    int c= ctype_toupper(*p++);
    if ('0'<=c && c<='9') tv = c-'0';
    else if ('A'<=c && c<='Z') tv = c-'A'+10;
    else return 0;
    if (tv >= base) return 0;
    v *= base;
    v += tv;
  }
  if (v>max) return 0;
  *v_r= v;
  return 1;
}

static bool revparse_inet(struct revparse_state *rps,
			  const char *dgram, int nlabels,
			  adns_rrtype *rrtype_r, adns_sockaddr *addr_r) {
  if (!revparse_check_tail(rps,dgram,nlabels,4,"in-addr")) return 0;

  uint32_t a=0;
  int i;
  for (i=3; i>=0; i--) {
    REVPARSE_P_L(i);
    unsigned v;
    if (!revparse_atoi(p,l,10,255,&v)) return 0;
    a <<= 8;
    a |= v;
  }
  *rrtype_r= adns_r_a;
  addr_r->inet.sin_family= AF_INET;
  addr_r->inet.sin_addr.s_addr= htonl(a);
  return 1;
}

static bool revparse_inet6(struct revparse_state *rps,
			   const char *dgram, int nlabels,
			   adns_rrtype *rrtype_r, adns_sockaddr *addr_r) {
  if (!revparse_check_tail(rps,dgram,nlabels,32,"ip6")) return 0;

  int i, j;
  memset(addr_r,0,sizeof(*addr_r));
  unsigned char *a= addr_r->inet6.sin6_addr.s6_addr+16;
  for (i=0; i<32; ) { /* i incremented in inner loop */
    unsigned b=0;
    for (j=0; j<2; j++, i++) {
      REVPARSE_P_L(i);
      unsigned v;
      if (!revparse_atoi(p,l,16,15,&v)) return 0;
      b >>= 4;
      b |= v << 4;
    }
    *--a= b;
  }
  *rrtype_r= adns_r_aaaa;
  addr_r->inet.sin_family= AF_INET6;
  return 1;
}

bool adns__revparse_label(struct revparse_state *rps, int labnum,
			  const char *dgram, int labstart, int lablen) {
  if (labnum >= MAXREVLABELS)
    return 0;

  assert(labstart <= 65535);
  assert(lablen <= 255);
  rps->labstart[labnum] = labstart;
  rps->lablen[labnum] = lablen;
  return 1;
}

bool adns__revparse_done(struct revparse_state *rps,
			 const char *dgram, int nlabels,
			 adns_rrtype *rrtype_r, adns_sockaddr *addr_r) {
  return
    revparse_inet(rps,dgram,nlabels,rrtype_r,addr_r) ||
    revparse_inet6(rps,dgram,nlabels,rrtype_r,addr_r);
}
Commit	Line	Data
07554ccd MW	1	/*
	2	* addrfam.c
	3	* - address-family specific code
	4	*/
	5	/*
	6	* This file is part of adns, which is
26e1c3d6	7	* Copyright (C) 1997-2000,2003,2006,2014 Ian Jackson
17cd4f48	8	* Copyright (C) 2014 Mark Wooding
07554ccd MW	9	* Copyright (C) 1999-2000,2003,2006 Tony Finch
	10	* Copyright (C) 1991 Massachusetts Institute of Technology
	11	* (See the file INSTALL for full details.)
	12	*
	13	* This program is free software; you can redistribute it and/or modify
	14	* it under the terms of the GNU General Public License as published by
7f8bbe29	15	* the Free Software Foundation; either version 3, or (at your option)
07554ccd MW	16	* any later version.
	17	*
	18	* This program is distributed in the hope that it will be useful,
	19	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	20	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	21	* GNU General Public License for more details.
	22	*
	23	* You should have received a copy of the GNU General Public License
8c09a4c6	24	* along with this program; if not, write to the Free Software Foundation.
07554ccd MW	25	*/
	26
	27	#include <stdlib.h>
	28	#include <errno.h>
	29	#include <limits.h>
	30	#include <unistd.h>
5806b745 IJ	31	#include <inttypes.h>
	32	#include <stddef.h>
	33	#include <stdbool.h>
07554ccd MW	34
	35	#include <sys/types.h>
	36	#include <sys/socket.h>
	37	#include <netinet/in.h>
	38	#include <arpa/inet.h>
5806b745	39	#include <net/if.h>
07554ccd MW	40
	41	#include "internal.h"
	42
	43	/*
	44	* General address-family operations.
	45	*/
	46
468a4293 IJ	47	#define SIN(cnst, sa) ((void)(sa)->sa_family, (cnst struct sockaddr_in *)(sa))
468a4293 IJ	48	#define SIN6(cnst, sa) ((void)(sa)->sa_family, (cnst struct sockaddr_in6 *)(sa))
07554ccd	49
6380363d	50	static void unknown_af(int af) NONRETURNING;
07554ccd MW	51	static void unknown_af(int af) {
	52	fprintf(stderr, "ADNS INTERNAL: unknown address family %d\n", af);
	53	abort();
	54	}
	55
e0fc85c4 IJ	56	/*
	57	* SOCKADDR_IN_IN6(CNST, struct sockaddr *sa, SIN, {
	58	* // struct sockaddr_in *const SIN; // implicitly
	59	* code for inet;
	60	* }, {
	61	* // struct sockaddr_in6 *const SIN6; // implicitly
	62	* code for inet6;
	63	* })
	64	*
	65	* SOCKADDR_IN_IN6_PAIR(CNST, struct sockaddr *sa, SINA,
	66	* struct sockaddr *sb, SINB, {
	67	* // struct sockaddr_in *const SINA; // implicitly
	68	* // struct sockaddr_in *const SINB; // implicitly
	69	* code for inet;
	70	* },{
	71	* // struct sockaddr_in6 *const SINA6; // implicitly
	72	* // struct sockaddr_in6 *const SINB6; // implicitly
	73	* code for inet6;
	74	* });
	75	*
	76	* SOCKADDR_IN_IN6_OTHER(CNST, struct sockaddr *sa, SIN, { in }, { in6 }, {
	77	* code for other address family
	78	* })
	79	*
	80	* AF_IN_IN6_OTHER(af, { in }, { in6 }, { other })
	81	*
	82	* Executes the first or second block according to the AF in sa. CNST
	83	* may be `const' or empty. For _PAIR, sa and sb must be same AF.
	84	*
	85	* All except _OTHER handle unknown AFs with unknown_af.
	86	*
	87	* Code blocks may not contain , outside parens.
	88	*/
	89	#define AF_IN_IN6_OTHER(af, for_inet, for_inet6, other) \
	90	if ((af) == AF_INET) { \
	91	for_inet \
	92	} else if ((af) == AF_INET6) { \
	93	for_inet6 \
	94	} else { \
	95	other \
	96	}
	97	#define SOCKADDR_IN_IN6_OTHER(cnst, sa, sin, for_inet, for_inet6, other) \
468a4293 IJ	98	AF_IN_IN6_OTHER((sa)->sa_family, { \
468a4293 IJ	99	cnst struct sockaddr_in *const sin = SIN(cnst,(sa)); \
e0fc85c4 IJ	100	for_inet \
e0fc85c4 IJ	101	}, { \
468a4293	102	cnst struct sockaddr_in6 *const sin##6 = SIN6(cnst,(sa)); \
e0fc85c4 IJ	103	for_inet6 \
	104	}, \
	105	other \
	106	)
	107	#define SOCKADDR_IN_IN6(cnst, sa, sin, for_inet, for_inet6) \
	108	SOCKADDR_IN_IN6_OTHER(cnst, sa, sin, for_inet, for_inet6, { \
	109	unknown_af((sa)->sa_family); \
	110	})
	111	#define SOCKADDR_IN_IN6_PAIR(cnst, sa, sina, sb, sinb, for_inet, for_inet6) \
	112	do{ \
	113	assert((sa)->sa_family == (sb)->sa_family); \
	114	SOCKADDR_IN_IN6(cnst, sa, sina, { \
468a4293	115	cnst struct sockaddr_in *const sinb = SIN(cnst,(sb)); \
e0fc85c4	116	for_inet \
468a4293 IJ	117	}, { \
468a4293 IJ	118	cnst struct sockaddr_in6 *const sinb##6 = SIN6(cnst,(sb)); \
e0fc85c4 IJ	119	for_inet6 \
	120	}); \
	121	}while(0)
	122
82ae109e IJ	123	int adns__addrs_equal_raw(const struct sockaddr *a,
	124	int bf, const void *b) {
	125	if (a->sa_family != bf) return 0;
e0fc85c4 IJ	126
	127	SOCKADDR_IN_IN6(const, a, sin, {
	128	return sin->sin_addr.s_addr == ((const struct in_addr*)b)->s_addr;
	129	}, {
	130	return !memcmp(&sin6->sin6_addr, b, sizeof(struct in6_addr));
	131	});
07554ccd MW	132	}
07554ccd MW	133
82ae109e IJ	134	int adns__addrs_equal(const adns_sockaddr a, const adns_sockaddr b) {
	135	return adns__addrs_equal_raw(&a->sa, b->sa.sa_family,
	136	adns__sockaddr_addr(&b->sa));
	137	}
	138
	139	int adns__sockaddrs_equal(const struct sockaddr *sa,
	140	const struct sockaddr *sb) {
	141	if (!adns__addrs_equal_raw(sa, sb->sa_family, adns__sockaddr_addr(sb)))
	142	return 0;
e0fc85c4 IJ	143	SOCKADDR_IN_IN6_PAIR(const, sa, sina, sb, sinb, {
	144	return sina->sin_port == sinb->sin_port;
	145	}, {
	146	return sina6->sin6_port == sinb6->sin6_port &&
	147	sina6->sin6_scope_id == sinb6->sin6_scope_id;
	148	});
07554ccd MW	149	}
	150
	151	int adns__addr_width(int af) {
e0fc85c4 IJ	152	AF_IN_IN6_OTHER(af, {
	153	return 32;
	154	}, {
	155	return 128;
	156	}, {
	157	unknown_af(af);
	158	});
07554ccd MW	159	}
07554ccd MW	160
82ae109e	161	void adns__prefix_mask(adns_sockaddr *sa, int len) {
e0fc85c4 IJ	162	SOCKADDR_IN_IN6(, &sa->sa, sin, {
	163	assert(len <= 32);
	164	sin->sin_addr.s_addr= htonl(!len ? 0 : 0xffffffff << (32-len));
	165	}, {
	166	int i= len/8;
	167	int j= len%8;
	168	unsigned char *m= sin6->sin6_addr.s6_addr;
	169	assert(len <= 128);
	170	memset(m, 0xff, i);
	171	if (j) m[i++]= (0xff << (8-j)) & 0xff;
	172	memset(m+i, 0, 16-i);
	173	});
07554ccd MW	174	}
07554ccd MW	175
82ae109e	176	int adns__guess_prefix_length(const adns_sockaddr *sa) {
e0fc85c4 IJ	177	SOCKADDR_IN_IN6(const, &sa->sa, sin, {
	178	unsigned a= (ntohl(sin->sin_addr.s_addr) >> 24) & 0xff;
	179	if (a < 128) return 8;
	180	else if (a < 192) return 16;
	181	else if (a < 224) return 24;
	182	else return -1;
	183	}, {
	184	(void)sin6;
	185	return 64;
	186	});
07554ccd MW	187	}
07554ccd MW	188
82ae109e IJ	189	int adns__addr_matches(int af, const void *addr,
82ae109e IJ	190	const adns_sockaddr base, const adns_sockaddr mask)
07554ccd	191	{
82ae109e	192	if (af != base->sa.sa_family) return 0;
e0fc85c4 IJ	193	SOCKADDR_IN_IN6_PAIR(const, &base->sa, sbase, &mask->sa, smask, {
	194	const struct in_addr *v4 = addr;
	195	return (v4->s_addr & smask->sin_addr.s_addr)
	196	== sbase->sin_addr.s_addr;
	197	}, {
	198	int i;
	199	const char *a= addr;
	200	const char *b= sbase6->sin6_addr.s6_addr;
	201	const char *m= smask6->sin6_addr.s6_addr;
	202	for (i = 0; i < 16; i++)
	203	if ((a[i] & m[i]) != b[i]) return 0;
	204	return 1;
	205	});
07554ccd MW	206	}
07554ccd MW	207
82ae109e	208	const void adns__sockaddr_addr(const struct sockaddr sa) {
e0fc85c4 IJ	209	SOCKADDR_IN_IN6(const, sa, sin, {
	210	return &sin->sin_addr;
	211	}, {
	212	return &sin6->sin6_addr;
	213	});
07554ccd MW	214	}
07554ccd MW	215
82ae109e	216	void adns__addr_inject(const void a, adns_sockaddr sa) {
e0fc85c4 IJ	217	SOCKADDR_IN_IN6( , &sa->sa, sin, {
	218	memcpy(&sin->sin_addr, a, sizeof(sin->sin_addr));
	219	}, {
	220	memcpy(&sin6->sin6_addr, a, sizeof(sin6->sin6_addr));
	221	});
07554ccd	222	}
5806b745 IJ	223
	224	/*
	225	* addr2text and text2addr
	226	*/
	227
	228	#define ADDRFAM_DEBUG
	229	#ifdef ADDRFAM_DEBUG
	230	static void af_debug_func(const char *fmt, ...) {
	231	int esave= errno;
	232	va_list al;
	233	va_start(al,fmt);
	234	vfprintf(stderr,fmt,al);
	235	va_end(al);
	236	errno= esave;
	237	}
	238	# define af_debug(fmt,...) \
	239	(af_debug_func("%s: " fmt "\n", __func__, __VA_ARGS__))
	240	#else
	241	# define af_debug(fmt,...) ((void)("" fmt "", __VA_ARGS__))
	242	#endif
	243
	244	static bool addrtext_our_errno(int e) {
	245	return
	246	e==EAFNOSUPPORT \|\|
	247	e==EINVAL \|\|
	248	e==ENOSPC \|\|
	249	e==ENOSYS;
	250	}
	251
	252	static bool addrtext_scope_use_ifname(const struct sockaddr *sa) {
468a4293	253	const struct in6_addr *in6= &SIN6(const,sa)->sin6_addr;
5806b745 IJ	254	return
	255	IN6_IS_ADDR_LINKLOCAL(in6) \|\|
	256	IN6_IS_ADDR_MC_LINKLOCAL(in6);
	257	}
	258
	259	int adns_text2addr(const char *text, uint16_t port, adns_queryflags flags,
	260	struct sockaddr sa, socklen_t salen_io) {
	261	int af;
	262	char copybuf[INET6_ADDRSTRLEN];
	263	const char *parse=text;
	264	const char *scopestr=0;
	265	socklen_t needlen;
	266	void *dst;
	267	uint16_t *portp;
	268
	269	#define INVAL(how) do{ \
	270	af_debug("invalid: %s: `%s'", how, text); \
	271	return EINVAL; \
	272	}while(0)
	273
	274	#define AFCORE(INETx,SINx,sinx) \
	275	af= AF_##INETx; \
468a4293 IJ	276	dst = &SINx(,sa)->sinx##_addr; \
	277	portp = &SINx(,sa)->sinx##_port; \
	278	needlen= sizeof(*SINx(,sa));
5806b745 IJ	279
	280	if (!strchr(text, ':')) { /* INET */
	281
	282	AFCORE(INET,SIN,sin);
	283
	284	} else { /* INET6 */
	285
	286	AFCORE(INET6,SIN6,sin6);
	287
	288	const char *percent= strchr(text, '%');
	289	if (percent) {
	290	ptrdiff_t lhslen = percent - text;
	291	if (lhslen >= INET6_ADDRSTRLEN) INVAL("scoped addr lhs too long");
	292	memcpy(copybuf, text, lhslen);
	293	copybuf[lhslen]= 0;
	294
	295	parse= copybuf;
	296	scopestr= percent+1;
	297
	298	af_debug("will parse scoped addr `%s' %% `%s'", parse, scopestr);
	299	}
	300
	301	}
	302
	303	#undef AFCORE
	304
	305	if (scopestr && (flags & adns_qf_addrlit_scope_forbid))
	306	INVAL("scoped addr but _scope_forbid");
	307
	308	if (*salen_io < needlen) {
	309	*salen_io = needlen;
	310	return ENOSPC;
	311	}
	312
	313	memset(sa, 0, needlen);
	314
	315	sa->sa_family= af;
	316	*portp = htons(port);
	317
	318	if (af == AF_INET && !(flags & adns_qf_addrlit_ipv4_quadonly)) {
	319	/* we have to use inet_aton to deal with non-dotted-quad literals */
468a4293	320	int r= inet_aton(parse,&SIN(,sa)->sin_addr);
5806b745 IJ	321	if (!r) INVAL("inet_aton rejected");
	322	} else {
	323	int r= inet_pton(af,parse,dst);
	324	if (!r) INVAL("inet_pton rejected");
	325	assert(r>0);
	326	}
	327
	328	if (scopestr) {
	329	errno=0;
	330	char *ep;
	331	unsigned long scope= strtoul(scopestr,&ep,10);
	332	if (errno==ERANGE) INVAL("numeric scope id too large for unsigned long");
	333	assert(!errno);
	334	if (!*ep) {
	335	if (scope > ~(uint32_t)0)
	336	INVAL("numeric scope id too large for uint32_t");
	337	} else { /* !!ep /
	338	if (flags & adns_qf_addrlit_scope_numeric)
	339	INVAL("non-numeric scope but _scope_numeric");
	340	if (!addrtext_scope_use_ifname(sa)) {
	341	af_debug("cannot convert non-numeric scope"
	342	" in non-link-local addr `%s'", text);
	343	return ENOSYS;
	344	}
	345	errno= 0;
	346	scope= if_nametoindex(scopestr);
	347	if (!scope) {
	348	/* RFC3493 says "No errors are defined". It's not clear
	349	* whether that is supposed to mean if_nametoindex "can't
	350	* fail" (other than by the supplied name not being that of an
	351	* interface) which seems unrealistic, or that it conflates
	352	* all its errors together by failing to set errno, or simply
	353	* that they didn't bother to document the errors.
	354	*
	355	* glibc, FreeBSD and OpenBSD all set errno (to ENXIO when
	356	* appropriate). See Debian bug #749349.
	357	*
	358	* We attempt to deal with this by clearing errno to start
	359	* with, and then perhaps mapping the results. */
	360	af_debug("if_nametoindex rejected scope name (errno=%s)",
	361	strerror(errno));
	362	if (errno==0) {
	363	return ENXIO;
	364	} else if (addrtext_our_errno(errno)) {
	365	/* we use these for other purposes, urgh. */
	366	perror("adns: adns_text2addr: if_nametoindex"
	367	" failed with unexpected error");
	368	return EIO;
	369	} else {
	370	return errno;
	371	}
	372	} else { /* ix>0 */
	373	if (scope > ~(uint32_t)0) {
	374	fprintf(stderr,"adns: adns_text2addr: if_nametoindex"
	375	" returned an interface index >=2^32 which will not fit"
	376	" in sockaddr_in6.sin6_scope_id");
	377	return EIO;
	378	}
	379	}
	380	} /* else; !!ep /
	381
468a4293	382	SIN6(,sa)->sin6_scope_id= scope;
5806b745 IJ	383	} /* if (scopestr) */
	384
	385	*salen_io = needlen;
	386	return 0;
	387	}
	388
	389	int adns_addr2text(const struct sockaddr *sa, adns_queryflags flags,
	390	char buffer, int buflen_io, int *port_r) {
	391	const void *src;
	392	int port;
	393
	394	if (*buflen_io < ADNS_ADDR2TEXT_BUFLEN) {
	395	*buflen_io = ADNS_ADDR2TEXT_BUFLEN;
	396	return ENOSPC;
	397	}
	398
e0fc85c4 IJ	399	SOCKADDR_IN_IN6_OTHER(const, sa, sin, {
	400	src= &sin->sin_addr; port= sin->sin_port;
	401	}, {
	402	src= &sin6->sin6_addr; port= sin6->sin6_port;
	403	}, {
	404	return EAFNOSUPPORT;
	405	});
5806b745 IJ	406
	407	const char ok= inet_ntop(sa->sa_family, src, buffer, buflen_io);
	408	assert(ok);
	409
	410	if (sa->sa_family == AF_INET6) {
468a4293	411	uint32_t scope = SIN6(const,sa)->sin6_scope_id;
5806b745 IJ	412	if (scope) {
	413	if (flags & adns_qf_addrlit_scope_forbid)
	414	return EINVAL;
	415	int scopeoffset = strlen(buffer);
	416	int remain = *buflen_io - scopeoffset;
	417	char *scopeptr = buffer + scopeoffset;
	418	assert(remain >= IF_NAMESIZE+1/%/);
	419	*scopeptr++= '%'; remain--;
	420	bool parsedname = 0;
c3a165f5 MW	421	af_debug("will print scoped addr `%.*s' %% %"PRIu32"",
c3a165f5 MW	422	scopeoffset,buffer, scope);
5806b745 IJ	423	if (scope <= UINT_MAX /* so we can pass it to if_indextoname */
	424	&& !(flags & adns_qf_addrlit_scope_numeric)
	425	&& addrtext_scope_use_ifname(sa)) {
	426	parsedname = if_indextoname(scope, scopeptr);
	427	if (!parsedname) {
	428	af_debug("if_indextoname rejected scope (errno=%s)",
	429	strerror(errno));
	430	if (errno==ENXIO) {
	431	/* fair enough, show it as a number then */
	432	} else if (addrtext_our_errno(errno)) {
	433	/* we use these for other purposes, urgh. */
	434	perror("adns: adns_addr2text: if_indextoname"
	435	" failed with unexpected error");
	436	return EIO;
	437	} else {
	438	return errno;
	439	}
	440	}
	441	}
	442	if (!parsedname) {
	443	int r = snprintf(scopeptr, remain,
	444	"%"PRIu32"", scope);
	445	assert(r < *buflen_io - scopeoffset);
	446	}
	447	af_debug("printed scoped addr `%s'", buffer);
	448	}
	449	}
	450
	451	if (port_r) *port_r= ntohs(port);
	452	return 0;
	453	}
b0e8338a MW	454
	455	char adns__sockaddr_ntoa(const struct sockaddr sa, char *buf) {
	456	int err;
	457	int len= ADNS_ADDR2TEXT_BUFLEN;
	458
	459	err= adns_addr2text(sa, 0, buf, &len, 0);
	460	if (err == EIO)
	461	err= adns_addr2text(sa, adns_qf_addrlit_scope_numeric, buf, &len, 0);
	462	assert(!err);
	463	return buf;
	464	}
51cb3de3 MW	465
	466	/*
	467	* Reverse-domain parsing and construction.
	468	*/
	469
	470	int adns__make_reverse_domain(const struct sockaddr sa, const char zone,
	471	char **buf_io, size_t bufsz,
	472	char **buf_free_r) {
	473	size_t req;
	474	char *p;
	475	unsigned c, y;
	476	unsigned long aa;
	477	const unsigned char *ap;
	478	int i, j;
	479
e0fc85c4 IJ	480	AF_IN_IN6_OTHER(sa->sa_family, {
	481	req= 4 * 4;
	482	if (!zone) zone= "in-addr.arpa";
	483	}, {
	484	req = 2 * 32;
	485	if (!zone) zone= "ip6.arpa";
	486	}, {
	487	return ENOSYS;
	488	});
51cb3de3 MW	489
	490	req += strlen(zone) + 1;
	491	if (req <= bufsz)
	492	p= *buf_io;
	493	else {
	494	p= malloc(req); if (!p) return errno;
	495	*buf_free_r = p;
	496	}
	497
	498	*buf_io= p;
e0fc85c4 IJ	499	SOCKADDR_IN_IN6(const, sa, sin, {
	500	aa= ntohl(sin->sin_addr.s_addr);
	501	for (i=0; i<4; i++) {
	502	p += sprintf(p, "%d", (int)(aa & 0xff));
51cb3de3	503	*p++= '.';
e0fc85c4	504	aa >>= 8;
51cb3de3	505	}
e0fc85c4 IJ	506	}, {
	507	ap= sin6->sin6_addr.s6_addr + 16;
	508	for (i=0; i<16; i++) {
	509	c= *--ap;
	510	for (j=0; j<2; j++) {
	511	y= c & 0xf;
	512	*p++= (y < 10) ? y + '0' : y - 10 + 'a';
	513	c >>= 4;
	514	*p++= '.';
	515	}
	516	}
	517	});
51cb3de3 MW	518
	519	strcpy(p, zone);
	520	return 0;
	521	}
	522
	523
c0af047b IJ	524	#define REVPARSE_P_L(labnum) \
	525	const char *p= dgram + rps->labstart[labnum]; \
	526	int l= rps->lablen[labnum]
	527	/*
	528	* REVPARSE_P_L(int labnum);
	529	* expects:
	530	* const char *dgram;
	531	* const struct revparse_state *rps;
	532	* produces:
	533	* const char *p; // start of label labnum in dgram
	534	* int l; // length of label in dgram
	535	*/
	536
	537	static bool revparse_check_tail(struct revparse_state *rps,
	538	const char *dgram, int nlabels,
	539	int bodylen, const char *inarpa) {
	540	int i;
51cb3de3	541
c0af047b IJ	542	if (nlabels != bodylen+2) return 0;
	543	for (i=0; i<2; i++) {
	544	REVPARSE_P_L(bodylen+i);
	545	const char *want= !i ? inarpa : "arpa";
	546	if (!adns__labels_equal(p,l, want,strlen(want))) return 0;
51cb3de3	547	}
c0af047b	548	return 1;
51cb3de3 MW	549	}
51cb3de3 MW	550
c0af047b IJ	551	static bool revparse_atoi(const char *p, int l, int base,
	552	unsigned max, unsigned *v_r) {
	553	if (l>3) return 0;
	554	if (l>1 && p[0]=='0') return 0;
	555	unsigned v=0;
	556	while (l-- > 0) {
	557	int tv;
	558	int c= ctype_toupper(*p++);
	559	if ('0'<=c && c<='9') tv = c-'0';
	560	else if ('A'<=c && c<='Z') tv = c-'A'+10;
	561	else return 0;
	562	if (tv >= base) return 0;
	563	v *= base;
	564	v += tv;
	565	}
	566	if (v>max) return 0;
	567	*v_r= v;
	568	return 1;
51cb3de3 MW	569	}
51cb3de3 MW	570
c0af047b IJ	571	static bool revparse_inet(struct revparse_state *rps,
	572	const char *dgram, int nlabels,
	573	adns_rrtype rrtype_r, adns_sockaddr addr_r) {
	574	if (!revparse_check_tail(rps,dgram,nlabels,4,"in-addr")) return 0;
51cb3de3	575
c0af047b	576	uint32_t a=0;
51cb3de3	577	int i;
c0af047b IJ	578	for (i=3; i>=0; i--) {
	579	REVPARSE_P_L(i);
	580	unsigned v;
	581	if (!revparse_atoi(p,l,10,255,&v)) return 0;
	582	a <<= 8;
	583	a \|= v;
	584	}
	585	*rrtype_r= adns_r_a;
	586	addr_r->inet.sin_family= AF_INET;
	587	addr_r->inet.sin_addr.s_addr= htonl(a);
	588	return 1;
51cb3de3 MW	589	}
51cb3de3 MW	590
c0af047b IJ	591	static bool revparse_inet6(struct revparse_state *rps,
	592	const char *dgram, int nlabels,
	593	adns_rrtype rrtype_r, adns_sockaddr addr_r) {
	594	if (!revparse_check_tail(rps,dgram,nlabels,32,"ip6")) return 0;
51cb3de3	595
c0af047b IJ	596	int i, j;
	597	memset(addr_r,0,sizeof(*addr_r));
	598	unsigned char *a= addr_r->inet6.sin6_addr.s6_addr+16;
	599	for (i=0; i<32; ) { /* i incremented in inner loop */
	600	unsigned b=0;
	601	for (j=0; j<2; j++, i++) {
	602	REVPARSE_P_L(i);
	603	unsigned v;
	604	if (!revparse_atoi(p,l,16,15,&v)) return 0;
	605	b >>= 4;
	606	b \|= v << 4;
	607	}
	608	*--a= b;
	609	}
	610	*rrtype_r= adns_r_aaaa;
	611	addr_r->inet.sin_family= AF_INET6;
	612	return 1;
	613	}
51cb3de3	614
e9fe1bdb IJ	615	bool adns__revparse_label(struct revparse_state *rps, int labnum,
e9fe1bdb IJ	616	const char *dgram, int labstart, int lablen) {
c0af047b IJ	617	if (labnum >= MAXREVLABELS)
c0af047b IJ	618	return 0;
51cb3de3	619
c0af047b IJ	620	assert(labstart <= 65535);
	621	assert(lablen <= 255);
	622	rps->labstart[labnum] = labstart;
	623	rps->lablen[labnum] = lablen;
e9fe1bdb	624	return 1;
51cb3de3 MW	625	}
51cb3de3 MW	626
e9fe1bdb IJ	627	bool adns__revparse_done(struct revparse_state *rps,
	628	const char *dgram, int nlabels,
	629	adns_rrtype rrtype_r, adns_sockaddr addr_r) {
c0af047b IJ	630	return
	631	revparse_inet(rps,dgram,nlabels,rrtype_r,addr_r) \|\|
	632	revparse_inet6(rps,dgram,nlabels,rrtype_r,addr_r);
51cb3de3	633	}