chiark / gitweb /
Don't use adns__must_gettimeofday in callbacks.
[adns] / src / types.c
index 11f9fd4d7f10fb1033dbcee1eddb78fc5bf627a5..e69e2f4782aeff160da6a87fb0400bf4d4652db3 100644 (file)
@@ -51,7 +51,7 @@
  * _txt                       (pa)
  * _inaddr                    (pa,cs,di, +search_sortlist, dip_genaddr)
  * _in6addr                   (pa,cs,di)
- * _addr                      (pap,pa,di,csp,cs,qs,  +search_sortlist_sa,
+ * _addr                      (pap,pa,di,csp,cs,gsz,qs,  +search_sortlist_sa,
  *                             dip_sockaddr, rrtypes)
  * _domain                    (pap)
  * _host_raw                  (pa)
@@ -335,8 +335,8 @@ static adns_status cs_in6addr(vbuf *vb, const void *datap) {
 }
 
 /*
- * _addr   (pap,pa,di,csp,cs,qs, +search_sortlist_sa, dip_sockaddr,
- *             addr_rrtypes, addr_rrsz)
+ * _addr   (pap,pa,di,csp,cs,gsz,qs, +search_sortlist_sa, dip_sockaddr,
+ *             addr_rrtypes)
  */
 
 /* About CNAME handling in addr queries.
@@ -487,6 +487,8 @@ static void addr_rrtypes(adns_state ads, adns_rrtype type,
   adns_rrtype qtf = type & adns__qtf_deref;
   adns_queryflags permitaf = 0, hackaf = 0;
 
+  if (!(qf & (adns_qf_ipv4_only | adns_qf_ipv6_only)))
+    qf |= adns_qf_ipv4_only | adns_qf_ipv6_only;
   if (!(type & adns__qtf_bigaddr) || !(type & adns__qtf_manyaf))
     qf = (qf & adns__qf_afmask) | adns_qf_ipv4_only;
   else if (ads->iflags & adns_if_afmask) {
@@ -500,16 +502,15 @@ static void addr_rrtypes(adns_state ads, adns_rrtype type,
       qf &= hackaf | permitaf | ~adns__qf_afmask;
   }
 
-
   if (qf & adns_qf_ipv4_only) rrty[n++] = adns_r_a | qtf;
   if (qf & adns_qf_ipv6_only) rrty[n++] = adns_r_aaaa | qtf;
 
   *nrrty = n;
 }
 
-static size_t addr_rrsz(adns_query qu)
+static int gsz_addr(adns_rrtype type)
 {
-  return qu->answer->type & adns__qtf_bigaddr ?
+  return type & adns__qtf_bigaddr ?
     sizeof(adns_rr_addr) : sizeof(adns_rr_addr_v4only);
 }
 
@@ -518,7 +519,10 @@ static adns_status append_addrs(adns_query qu, adns_query from, size_t rrsz,
                                const adns_rr_addr *sp, int slen)
 {
   size_t drrsz = *dlen*rrsz, srrsz = slen*rrsz;
-  byte *p = adns__alloc_interim(qu, drrsz + srrsz);
+  byte *p;
+
+  /* if (!slen) return adns_s_ok; */
+  p = adns__alloc_interim(qu, drrsz + srrsz);
   if (!p) R_NOMEM;
   if (*dlen) {
     memcpy(p, *dp, drrsz);
@@ -543,9 +547,6 @@ static void addr_subqueries(adns_query qu, struct timeval now,
     ~(adns_qf_search);
   qcontext ctx;
 
-  if (!(qu->answer->type & adns__qtf_bigaddr))
-    qu->answer->rrsz = sizeof(adns_rr_addr_v4only);
-
   /* This always makes child queries, even if there's only the one.  This
    * seems wasteful, but there's only one case where it'd be safe -- namely
    * IPv4-only -- and that's not the case I want to optimize.
@@ -605,7 +606,7 @@ static adns_status addr_submit(adns_query parent, adns_query *query_r,
 
   memcpy(qu->t.addr.rrty, rrty, nrrty*sizeof(*rrty));
   qu->t.addr.nrrty = nrrty;
-  addr_subqueries(qu, now, parent->query_dgram, parent->query_dglen);
+  addr_subqueries(qu, now, qu->query_dgram, qu->query_dglen);
   *query_r = qu;
   return adns_s_ok;
 }
@@ -635,9 +636,8 @@ static void icb_addr(adns_query parent, adns_query child)
 {
   adns_state ads = parent->ads;
   adns_answer *pans = parent->answer, *cans = child->answer;
-  struct timeval tvbuf;
+  struct timeval now;
   adns_status err;
-  const struct timeval *now = 0;
   int id;
 
   if (!(child->flags & adns__qf_addr_cname) &&
@@ -677,8 +677,8 @@ static void icb_addr(adns_query parent, adns_query child)
      * settled on.
      */
     adns__cancel_children(parent);
-    adns__must_gettimeofday(ads, &now, &tvbuf);
-    if (now) addr_subqueries(parent, *now, child->vb.buf, child->vb.used);
+    if (gettimeofday(&now, 0)) goto x_gtod;
+    addr_subqueries(parent, now, child->vb.buf, child->vb.used);
     return;
   }
 
@@ -697,8 +697,8 @@ static void icb_addr(adns_query parent, adns_query child)
     adns__cancel_children(parent);
     adns__free_interim(parent, pans->rrs.bytes);
     pans->rrs.bytes = 0; pans->nrrs = 0;
-    adns__must_gettimeofday(ads, &now, &tvbuf);
-    if (now) adns__search_next(ads, parent, *now);
+    if (gettimeofday(&now, 0)) goto x_gtod;
+    adns__search_next(ads, parent, now);
     return;
   }
 
@@ -718,6 +718,11 @@ static void icb_addr(adns_query parent, adns_query child)
   parent->flags |= adns__qf_addr_answer;
   return;
 
+x_gtod:
+  adns__diag(ads, -1, parent, "gettimeofday failed: %s", strerror(errno));
+  err = adns_s_systemfail;
+  goto x_err;
+
 x_err:
   adns__query_fail(parent, err);
 }
@@ -851,10 +856,9 @@ static void icb_hostaddr(adns_query parent, adns_query child) {
   adns_rr_hostaddr *rrp= child->ctx.info.hostaddr;
   adns_state ads= parent->ads;
   adns_status st;
-  size_t addrsz = addr_rrsz(parent);
+  size_t addrsz = gsz_addr(parent->answer->type);
 
   st= cans->status == adns_s_nodata ? adns_s_ok : cans->status;
-  rrp->astatus= st;
 
   if (st) goto done;
   assert(addrsz == cans->rrsz);
@@ -872,9 +876,10 @@ static void icb_hostaddr(adns_query parent, adns_query child) {
 done:
   if (st) {
     adns__free_interim(parent, rrp->addrs);
-    rrp->naddrs= (st>0 && st<=adns_s_max_tempfail) ? -1 : cans->nrrs;
+    rrp->naddrs= (st>0 && st<=adns_s_max_tempfail) ? -1 : 0;
   }
 
+  rrp->astatus= st;
   if (parent->children.head) {
     LIST_LINK_TAIL(ads->childw,parent);
   } else {
@@ -892,7 +897,7 @@ static adns_status pap_hostaddr(const parseinfo *pai, int *cbyte_io,
   adns_queryflags nflags;
   adns_rrtype rrty[ADDR_MAXRRTYPES];
   size_t nrrty;
-  size_t addrsz = addr_rrsz(pai->qu);
+  size_t addrsz = gsz_addr(pai->qu->answer->type);
 
   dmstart= cbyte= *cbyte_io;
   st= pap_domain(pai, &cbyte, max, &rrp->host,
@@ -1685,14 +1690,14 @@ static void mf_flat(adns_query qu, void *data) { }
 
 #define DEEP_TYPE(code,rrt,fmt,memb,parser,comparer,printer)   \
 { adns_r_##code & adns_rrt_reprmask, rrt,fmt,TYPESZ_M(memb),   \
-    mf_##memb, printer,parser,comparer, adns__qdpl_normal,0,0 }
+    mf_##memb, printer,parser,comparer, adns__qdpl_normal,0,0,0 }
 #define FLAT_TYPE(code,rrt,fmt,memb,parser,comparer,printer)   \
 { adns_r_##code & adns_rrt_reprmask, rrt,fmt,TYPESZ_M(memb),   \
-     mf_flat, printer,parser,comparer, adns__qdpl_normal,0,0 }
+    mf_flat, printer,parser,comparer, adns__qdpl_normal,0,0,0 }
 #define XTRA_TYPE(code,rrt,fmt,memb,parser,comparer,printer,              \
-                 makefinal,qdpl,postsort,sender)                          \
+                 makefinal,qdpl,postsort,getrrsz,sender)                  \
 { adns_r_##code & adns_rrt_reprmask, rrt,fmt,TYPESZ_M(memb), makefinal,           \
-    printer,parser,comparer,qdpl,postsort,sender }
+    printer,parser,comparer,qdpl,postsort,getrrsz,sender }
 
 static const typeinfo typeinfos[] = {
 /* Must be in ascending order of rrtype ! */
@@ -1709,15 +1714,15 @@ DEEP_TYPE(txt,    "TXT",   0,   manyistr,pa_txt,     0,        cs_txt        ),
 DEEP_TYPE(rp_raw, "RP",   "raw",strpair, pa_rp,      0,        cs_rp         ),
 FLAT_TYPE(aaaa,   "AAAA",  0,   in6addr, pa_in6addr, di_in6addr,cs_in6addr   ),
 XTRA_TYPE(srv_raw,"SRV",  "raw",srvraw , pa_srvraw,  di_srv,   cs_srvraw,
-                                        mf_srvraw, qdpl_srv, postsort_srv, 0),
+                                     mf_srvraw, qdpl_srv, postsort_srv, 0, 0),
 
 XTRA_TYPE(addr,   "A",  "addr", addr,    pa_addr,    di_addr,  cs_addr,
-                                      mf_flat, adns__qdpl_normal, 0, qs_addr),
+                            mf_flat, adns__qdpl_normal, 0, gsz_addr, qs_addr),
 DEEP_TYPE(ns,     "NS", "+addr",hostaddr,pa_hostaddr,di_hostaddr,cs_hostaddr ),
 DEEP_TYPE(ptr,    "PTR","checked",str,   pa_ptr,     0,        cs_domain     ),
 DEEP_TYPE(mx,     "MX", "+addr",inthostaddr,pa_mx,   di_mx,    cs_inthostaddr),
 XTRA_TYPE(srv,    "SRV","+addr",srvha,   pa_srvha,   di_srv,   cs_srvha,
-                                         mf_srvha, qdpl_srv, postsort_srv, 0),
+                                      mf_srvha, qdpl_srv, postsort_srv, 0, 0),
 
 DEEP_TYPE(soa,    "SOA","822",  soa,     pa_soa,     0,        cs_soa        ),
 DEEP_TYPE(rp,     "RP", "822",  strpair, pa_rp,      0,        cs_rp         ),