chiark / gitweb /
addrfam: Introduce SOCKADDR_IN_IN6, etc.
[adns.git] / src / addrfam.c
index 28bf128996c4bc2f88b96334b6cb8cb981b5f377..3c49c02d6b40656f8706bf9a26187d5539534b91 100644 (file)
 #define SIN6(sa) ((struct sockaddr_in6 *)(sa))
 #define CSIN6(sa) ((const struct sockaddr_in6 *)(sa))
 
-/* This gadget (thanks, Richard Kettlewell) makes sure that we handle the
- * same set of address families in each switch. */
-#define AF_CASES(pre)                                                  \
-  case AF_INET: goto pre##_inet;                                       \
-  case AF_INET6: goto pre##_inet6
-
+static void unknown_af(int af) NONRETURNING;
 static void unknown_af(int af) {
   fprintf(stderr, "ADNS INTERNAL: unknown address family %d\n", af);
   abort();
 }
 
-#define IN6_ADDR_EQUALP(a, b)                                          \
-  (!memcmp((a).s6_addr, (b).s6_addr, sizeof((a).s6_addr)))
-
-int adns__genaddr_equal_p(int af, const union gen_addr *a,
-                         int bf, const void *b) {
-  const union gen_addr *bb= b;
-  if (af != bf) return 0;
-  switch (af) {
-  AF_CASES(af);
-  af_inet: return a->v4.s_addr == bb->v4.s_addr;
-  af_inet6: return IN6_ADDR_EQUALP(a->v6, bb->v6);
-  default: unknown_af(af); return -1;
+/*
+ * 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 = (cnst void*)(sa);           \
+      for_inet                                                         \
+  }, {                                                                 \
+      cnst struct sockaddr_in6 *const sin##6 = (cnst void*)(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 = (cnst void*)(sb);         \
+       for_inet                                                        \
+    }, {                                                               \
+       cnst struct sockaddr_in6 *const sinb##6 = (cnst void*)(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__sockaddr_equal_p(const struct sockaddr *sa,
-                          const struct sockaddr *sb) {
-  if (sa->sa_family != sb->sa_family) return 0;
-  switch (sa->sa_family) {
-  AF_CASES(af);
-  af_inet: {
-    const struct sockaddr_in *sina= CSIN(sa), *sinb= CSIN(sb);
-    return (sina->sin_addr.s_addr == sinb->sin_addr.s_addr &&
-           sina->sin_port == sinb->sin_port);
-  }
-  af_inet6: {
-    /* Don't check the flowlabel.  That's apparently useful for routing
-     * performance, but doesn't affect the address in any important
-     * respect. */
-    const struct sockaddr_in6 *sin6a= CSIN6(sa), *sin6b= CSIN6(sb);
-    return (IN6_ADDR_EQUALP(sin6a->sin6_addr, sin6b->sin6_addr) &&
-           sin6a->sin6_port == sin6b->sin6_port &&
-           sin6a->sin6_scope_id == sin6b->sin6_scope_id);
-  }
-  default:
-    unknown_af(sa->sa_family);
-    return -1;
-  }
+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) {
-  switch (af) {
-  AF_CASES(af);
-  af_inet: return 32;
-  af_inet6: return 128;
-  default: unknown_af(af); return -1;
-  }
+  AF_IN_IN6_OTHER(af, {
+      return 32;
+    }, {
+      return 128;
+    }, {
+      unknown_af(af);
+    });
 }
 
-void adns__prefix_mask(int af, int len, union gen_addr *mask_r) {
-  switch (af) {
-  AF_CASES(af);
-  af_inet:
-    assert(len <= 32);
-    mask_r->v4.s_addr= htonl(!len ? 0 : 0xffffffff << (32-len));
-    break;
-  af_inet6: {
-    int i= len/8, j= len%8;
-    unsigned char *m= mask_r->v6.s6_addr;
-    assert(len <= 128);
-    memset(m, 0xff, i);
-    if (j) m[i++]= (0xff << (8-j)) & 0xff;
-    memset(m+i, 0, 16-i);
-  } break;
-  default:
-    unknown_af(af);
-    break;
-  }
+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(int af, const union gen_addr *addr) {
-  switch (af) {
-  AF_CASES(af);
-  af_inet: {
-    unsigned a= (ntohl(addr->v4.s_addr) >> 24) & 0xff;
-    if (a < 128) return 8;
-    else if (a < 192) return 16;
-    else if (a < 224) return 24;
-    else return -1;
-  } break;
-  af_inet6:
-    return 64;
-  default:
-    unknown_af(af);
-    return -1;
-  }
+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_match_p(int addraf, const union gen_addr *addr,
-                      int netaf, const union gen_addr *base,
-                      const union gen_addr *mask)
+int adns__addr_matches(int af, const void *addr,
+                      const adns_sockaddr *base, const adns_sockaddr *mask)
 {
-  if (addraf != netaf) return 0;
-  switch (addraf) {
-  AF_CASES(af);
-  af_inet:
-    return (addr->v4.s_addr & mask->v4.s_addr) == base->v4.s_addr;
-  af_inet6: {
-    int i;
-    const char *a= addr->v6.s6_addr;
-    const char *b= base->v6.s6_addr;
-    const char *m= mask->v6.s6_addr;
-    for (i = 0; i < 16; i++)
-      if ((a[i] & m[i]) != b[i]) return 0;
-    return 1;
-  } break;
-  default:
-    unknown_af(addraf);
-    return -1;
-  }
+  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;
+    });
 }
 
-void adns__sockaddr_extract(const struct sockaddr *sa,
-                           union gen_addr *a_r, int *port_r) {
-  switch (sa->sa_family) {
-  AF_CASES(af);
-  af_inet: {
-    const struct sockaddr_in *sin = CSIN(sa);
-    if (port_r) *port_r= ntohs(sin->sin_port);
-    if (a_r) a_r->v4= sin->sin_addr;
-    break;
-  }
-  af_inet6: {
-    const struct sockaddr_in6 *sin6 = CSIN6(sa);
-    if (port_r) *port_r= ntohs(sin6->sin6_port);
-    if (a_r) a_r->v6= sin6->sin6_addr;
-    break;
-  }
-  default:
-    unknown_af(sa->sa_family);
-  }
+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__sockaddr_inject(const union gen_addr *a, int port,
-                          struct sockaddr *sa) {
-  switch (sa->sa_family) {
-  AF_CASES(af);
-  af_inet: {
-    struct sockaddr_in *sin = SIN(sa);
-    if (port != -1) sin->sin_port= htons(port);
-    if (a) sin->sin_addr= a->v4;
-    break;
-  }
-  af_inet6: {
-    struct sockaddr_in6 *sin6 = SIN6(sa);
-    if (port != -1) sin6->sin6_port= htons(port);
-    if (a) sin6->sin6_addr= a->v6;
-    break;
-  }
-  default:
-    unknown_af(sa->sa_family);
-  }
+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));
+    });
 }
 
 /*
@@ -390,12 +399,13 @@ int adns_addr2text(const struct sockaddr *sa, adns_queryflags flags,
     return ENOSPC;
   }
 
-  switch (sa->sa_family) {
-    AF_CASES(af);
-    af_inet:  src= &CSIN(sa)->sin_addr;    port= CSIN(sa)->sin_port;    break;
-    af_inet6: src= &CSIN6(sa)->sin6_addr;  port= CSIN6(sa)->sin6_port;  break;
-    default: return EAFNOSUPPORT;
-  }
+  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);
@@ -455,3 +465,173 @@ char *adns__sockaddr_ntoa(const struct sockaddr *sa, char *buf) {
   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;
+}
+
+
+static int inet_rev_parsecomp(const char *p, size_t n) {
+  int i= 0;
+  if (n > 3) return -1;
+
+  while (n--) {
+    if ('0' <= *p && *p <= '9') i= 10*i + *p++ - '0';
+    else return -1;
+  }
+  return i;
+}
+
+static void inet_rev_mkaddr(adns_sockaddr *addr, const byte *ipv) {
+  struct in_addr *v4 = &addr->inet.sin_addr;
+  v4->s_addr= htonl((ipv[3]<<24) | (ipv[2]<<16) |
+                   (ipv[1]<<8) | (ipv[0]));
+}
+
+static int inet6_rev_parsecomp(const char *p, size_t n) {
+  if (n != 1) return -1;
+  else if ('0' <= *p && *p <= '9') return *p - '0';
+  else if ('a' <= *p && *p <= 'f') return *p - 'a' + 10;
+  else if ('A' <= *p && *p <= 'F') return *p - 'a' + 10;
+  else return -1;
+}
+
+static void inet6_rev_mkaddr(adns_sockaddr *addr, const byte *ipv) {
+  struct in6_addr *v6 = &addr->inet6.sin6_addr;
+  unsigned char *a= v6->s6_addr;
+  int i;
+
+  for (i=0; i<16; i++)
+    a[i]= (ipv[31-2*i] << 4) | (ipv[30-2*i] << 0);
+}
+
+static const struct revparse_domain {
+  int af;                              /* address family */
+  int nrevlab;                         /* n of reverse-address labels */
+  adns_rrtype rrtype;                  /* forward-lookup type */
+
+  int (*rev_parsecomp)(const char *p, size_t n);
+  /* parse a single component from a label; return the integer value, or -1
+   * if it was unintelligible.
+   */
+
+  void (*rev_mkaddr)(adns_sockaddr *addr, const byte *ipv);
+  /* write out the parsed protocol address from a vector of parsed components */
+
+  const char *const tail[3];           /* tail label names */
+} revparse_domains[NREVDOMAINS] = {
+  { AF_INET, 4, adns_r_a, inet_rev_parsecomp, inet_rev_mkaddr,
+    { DNS_INADDR_ARPA, 0 } },
+  { AF_INET6, 32, adns_r_aaaa, inet6_rev_parsecomp, inet6_rev_mkaddr,
+    { DNS_IP6_ARPA, 0 } },
+};
+
+#define REVDOMAIN_MAP(rps, labnum)                                     \
+  ((labnum) ? (rps)->map : (1 << NREVDOMAINS) - 1)
+
+int adns__revparse_label(struct revparse_state *rps, int labnum,
+                        const char *label, int lablen) {
+  unsigned f= REVDOMAIN_MAP(rps, labnum);
+  const struct revparse_domain *rpd;
+  const char *tp;
+  unsigned d;
+  int i, ac;
+
+  for (rpd=revparse_domains, i=0, d=1; i<NREVDOMAINS; rpd++, i++, d <<= 1) {
+    if (!(f & d)) continue;
+    if (labnum >= rpd->nrevlab) {
+      tp = rpd->tail[labnum - rpd->nrevlab];
+      if (!tp || strncmp(label, tp, lablen) != 0 || tp[lablen])
+       goto mismatch;
+    } else {
+      ac= rpd->rev_parsecomp(label, lablen);
+      if (ac < 0) goto mismatch;
+      assert(labnum < sizeof(rps->ipv[i]));
+      rps->ipv[i][labnum]= ac;
+    }
+    continue;
+
+  mismatch:
+    f &= ~d;
+    if (!f) return -1;
+  }
+
+  rps->map= f;
+  return 0;
+}
+
+int adns__revparse_done(struct revparse_state *rps, int nlabels,
+                       adns_rrtype *rrtype_r, adns_sockaddr *addr_r) {
+  unsigned f= REVDOMAIN_MAP(rps, nlabels);
+  const struct revparse_domain *rpd;
+  unsigned d;
+  int i, found= -1;
+
+  for (rpd=revparse_domains, i=0, d=1; i<NREVDOMAINS; rpd++, i++, d <<= 1) {
+    if (!(f & d)) continue;
+    if (nlabels >= rpd->nrevlab && !rpd->tail[nlabels - rpd->nrevlab])
+      { found = i; continue; }
+    f &= ~d;
+    if (!f) return -1;
+  }
+  assert(found >= 0); assert(f == (1 << found));
+
+  rpd= &revparse_domains[found];
+  *rrtype_r= rpd->rrtype;
+  addr_r->sa.sa_family= rpd->af;
+  rpd->rev_mkaddr(addr_r, rps->ipv[found]);
+  return 0;
+}