chiark / gitweb /
src/types.c: Handle inconsistent CNAME records between addr answers.
[adns.git] / src / types.c
1 /*
2  * types.c
3  * - RR-type-specific code, and the machinery to call it
4  */
5 /*
6  *  This file is part of adns, which is
7  *    Copyright (C) 1997-2000,2003,2006  Ian Jackson
8  *    Copyright (C) 1999-2000,2003,2006  Tony Finch
9  *    Copyright (C) 1991 Massachusetts Institute of Technology
10  *  (See the file INSTALL for full details.)
11  *  
12  *  This program is free software; you can redistribute it and/or modify
13  *  it under the terms of the GNU General Public License as published by
14  *  the Free Software Foundation; either version 2, or (at your option)
15  *  any later version.
16  *  
17  *  This program is distributed in the hope that it will be useful,
18  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
19  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
20  *  GNU General Public License for more details.
21  *  
22  *  You should have received a copy of the GNU General Public License
23  *  along with this program; if not, write to the Free Software Foundation,
24  *  Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. 
25  */
26
27 #include <stddef.h>
28 #include <stdlib.h>
29
30 #include <sys/types.h>
31 #include <sys/socket.h>
32 #include <netinet/in.h>
33 #include <arpa/inet.h>
34
35 #include "internal.h"
36
37 #define R_NOMEM       return adns_s_nomemory
38 #define CSP_ADDSTR(s) do {                      \
39     if (!adns__vbuf_appendstr(vb,(s))) R_NOMEM; \
40   } while (0)
41
42 /*
43  * order of sections:
44  *
45  * _string                    (pap)
46  * _textdata, _qstring        (csp)
47  * _str                       (mf,cs)
48  * _intstr                    (mf,csp,cs)
49  * _manyistr                  (mf,cs)
50  * _txt                       (pa)
51  * _inaddr                    (pa,di,cs
52  *                              +search_sortlist, dip_genaddr, csp_genaddr)
53  * _in6addr                   (pa,di,cs)
54  * _addr                      (pap,pa,di,div,csp,cs,gsz,qs
55  *                              +search_sortlist_sa, dip_sockaddr,
56  *                               addr_rrtypes, addr_submit, icb_addr)
57  * _domain                    (pap,csp,cs)
58  * _dom_raw                   (pa)
59  * _host_raw                  (pa)
60  * _hostaddr                  (pap,pa,dip,di,mfp,mf,csp,cs
61  *                              +pap_findaddrs, icb_hostaddr)
62  * _mx_raw                    (pa,di)
63  * _mx                        (pa,di)
64  * _inthostaddr               (mf,cs)
65  * _inthost                   (cs)
66  * _ptr                       (ckl,pa +icb_ptr)
67  * _strpair                   (mf)
68  * _intstrpair                (mf)
69  * _hinfo                     (pa)
70  * _mailbox                   (pap,csp +pap_mailbox822)
71  * _rp                        (pa,cs)
72  * _soa                       (pa,mf,cs)
73  * _srv*                      (ckl,(pap),pa*2,mf*2,di,(csp),cs*2,postsort)
74  * _byteblock                 (mf)
75  * _opaque                    (pa,cs)
76  * _flat                      (mf)
77  *
78  * within each section:
79  *    ckl_*
80  *    pap_*
81  *    pa_*
82  *    dip_*
83  *    di_*
84  *    mfp_*
85  *    mf_*
86  *    csp_*
87  *    cs_*
88  *    gsz_*
89  *    postsort_*
90  *    qs_*
91  */
92
93 /*
94  * _qstring               (pap,csp)
95  */
96
97 static adns_status pap_qstring(const parseinfo *pai, int *cbyte_io, int max,
98                               int *len_r, char **str_r) {
99   /* Neither len_r nor str_r may be null.
100    * End of datagram (overrun) is indicated by returning adns_s_invaliddata;
101    */
102   const byte *dgram= pai->dgram;
103   int l, cbyte;
104   char *str;
105
106   cbyte= *cbyte_io;
107
108   if (cbyte >= max) return adns_s_invaliddata;
109   GET_B(cbyte,l);
110   if (cbyte+l > max) return adns_s_invaliddata;
111   
112   str= adns__alloc_interim(pai->qu, l+1);
113   if (!str) R_NOMEM;
114   
115   str[l]= 0;
116   memcpy(str,dgram+cbyte,l);
117
118   *len_r= l;
119   *str_r= str;
120   *cbyte_io= cbyte+l;
121   
122   return adns_s_ok;
123 }
124
125 static adns_status csp_qstring(vbuf *vb, const char *dp, int len) {
126   unsigned char ch;
127   char buf[10];
128   int cn;
129
130   CSP_ADDSTR("\"");
131   for (cn=0; cn<len; cn++) {
132     ch= *dp++;
133     if (ch == '\\') {
134       CSP_ADDSTR("\\\\");
135     } else if (ch == '"') {
136       CSP_ADDSTR("\\\"");
137     } else if (ch >= 32 && ch <= 126) {
138       if (!adns__vbuf_append(vb,&ch,1)) R_NOMEM;
139     } else {
140       sprintf(buf,"\\x%02x",ch);
141       CSP_ADDSTR(buf);
142     }
143   }
144   CSP_ADDSTR("\"");
145   
146   return adns_s_ok;
147 }
148
149 /*
150  * _str  (mf)
151  */
152
153 static void mf_str(adns_query qu, void *datap) {
154   char **rrp= datap;
155
156   adns__makefinal_str(qu,rrp);
157 }
158
159 /*
160  * _intstr  (mf)
161  */
162
163 static void mf_intstr(adns_query qu, void *datap) {
164   adns_rr_intstr *rrp= datap;
165
166   adns__makefinal_str(qu,&rrp->str);
167 }
168
169 /*
170  * _manyistr   (mf)
171  */
172
173 static void mf_manyistr(adns_query qu, void *datap) {
174   adns_rr_intstr **rrp= datap;
175   adns_rr_intstr *te, *table;
176   void *tablev;
177   int tc;
178
179   for (tc=0, te= *rrp; te->i >= 0; te++, tc++);
180   tablev= *rrp;
181   adns__makefinal_block(qu,&tablev,sizeof(*te)*(tc+1));
182   *rrp= table= tablev;
183   for (te= *rrp; te->i >= 0; te++)
184     adns__makefinal_str(qu,&te->str);
185 }
186
187 /*
188  * _txt   (pa,cs)
189  */
190
191 static adns_status pa_txt(const parseinfo *pai, int cbyte,
192                           int max, void *datap) {
193   adns_rr_intstr **rrp= datap, *table, *te;
194   const byte *dgram= pai->dgram;
195   int ti, tc, l, startbyte;
196   adns_status st;
197
198   startbyte= cbyte;
199   if (cbyte >= max) return adns_s_invaliddata;
200   tc= 0;
201   while (cbyte < max) {
202     GET_B(cbyte,l);
203     cbyte+= l;
204     tc++;
205   }
206   if (cbyte != max || !tc) return adns_s_invaliddata;
207
208   table= adns__alloc_interim(pai->qu,sizeof(*table)*(tc+1));
209   if (!table) R_NOMEM;
210
211   for (cbyte=startbyte, ti=0, te=table; ti<tc; ti++, te++) {
212     st= pap_qstring(pai, &cbyte, max, &te->i, &te->str);
213     if (st) return st;
214   }
215   assert(cbyte == max);
216
217   te->i= -1;
218   te->str= 0;
219   
220   *rrp= table;
221   return adns_s_ok;
222 }
223
224 static adns_status cs_txt(vbuf *vb, const void *datap) {
225   const adns_rr_intstr *const *rrp= datap;
226   const adns_rr_intstr *current;
227   adns_status st;
228   int spc;
229
230   for (current= *rrp, spc=0;  current->i >= 0;  current++, spc=1) {
231     if (spc) CSP_ADDSTR(" ");
232     st= csp_qstring(vb,current->str,current->i); if (st) return st;
233   }
234   return adns_s_ok;
235 }
236
237 /*
238  * _hinfo   (cs)
239  */
240
241 static adns_status cs_hinfo(vbuf *vb, const void *datap) {
242   const adns_rr_intstrpair *rrp= datap;
243   adns_status st;
244
245   st= csp_qstring(vb,rrp->array[0].str,rrp->array[0].i);  if (st) return st;
246   CSP_ADDSTR(" ");
247   st= csp_qstring(vb,rrp->array[1].str,rrp->array[1].i);  if (st) return st;
248   return adns_s_ok;
249 }
250
251 /*
252  * _inaddr   (pa,di,cs +search_sortlist, dip_genaddr, csp_genaddr)
253  */
254
255 static adns_status pa_inaddr(const parseinfo *pai, int cbyte,
256                              int max, void *datap) {
257   struct in_addr *storeto= datap;
258   
259   if (max-cbyte != 4) return adns_s_invaliddata;
260   memcpy(storeto, pai->dgram + cbyte, 4);
261   return adns_s_ok;
262 }
263
264 static int search_sortlist(adns_state ads, int af, const void *ad) {
265   const struct sortlist *slp;
266   const struct in6_addr *a6;
267   union gen_addr a;
268   int i;
269   int v6mappedp= 0;
270
271   if (af == AF_INET6) {
272     a6= ad;
273     if (IN6_IS_ADDR_V4MAPPED(a6)) {
274       a.v4.s_addr= htonl(((unsigned long)a6->s6_addr[12] << 24) |
275                          ((unsigned long)a6->s6_addr[13] << 16) |
276                          ((unsigned long)a6->s6_addr[14] <<  8) |
277                          ((unsigned long)a6->s6_addr[15] <<  0));
278       v6mappedp= 1;
279     }
280   }
281
282   for (i=0, slp=ads->sortlist;
283        i<ads->nsortlist &&
284          !adns__addr_match_p(af,ad, slp->af,&slp->base,&slp->mask) &&
285          !(v6mappedp &&
286            adns__addr_match_p(AF_INET,&a, slp->af,&slp->base,&slp->mask));
287        i++, slp++);
288   return i;
289 }
290
291 static int dip_genaddr(adns_state ads, int af, const void *a, const void *b) {
292   int ai, bi;
293   
294   if (!ads->nsortlist) return 0;
295
296   ai= search_sortlist(ads,af,a);
297   bi= search_sortlist(ads,af,b);
298   return bi<ai;
299 }
300
301 static int di_inaddr(adns_state ads,
302                      const void *datap_a, const void *datap_b) {
303   return dip_genaddr(ads,AF_INET,datap_a,datap_b);
304 }
305
306 static adns_status csp_genaddr(vbuf *vb, int af, const void *p) {
307   char buf[ADNS_ADDR2TEXT_BUFLEN];
308   int len= sizeof(buf);
309   adns_rr_addr a;
310   int err;
311
312   memset(&a, 0, sizeof(a));
313   a.addr.sa.sa_family= af;
314   adns__sockaddr_inject(p, 0, &a.addr.sa);
315   err= adns_addr2text(&a.addr.sa,0, buf,&len, 0); assert(!err);
316   CSP_ADDSTR(buf);
317   return adns_s_ok;
318 }
319
320 static adns_status cs_inaddr(vbuf *vb, const void *datap) {
321   return csp_genaddr(vb, AF_INET,datap);
322 }
323
324 /*
325  * _in6addr   (pa,di,cs)
326  */
327
328 static adns_status pa_in6addr(const parseinfo *pai, int cbyte,
329                              int max, void *datap) {
330   struct in6_addr *storeto= datap;
331
332   if (max-cbyte != 16) return adns_s_invaliddata;
333   memcpy(storeto->s6_addr, pai->dgram + cbyte, 16);
334   return adns_s_ok;
335 }
336
337 static int di_in6addr(adns_state ads,
338                      const void *datap_a, const void *datap_b) {
339   return dip_genaddr(ads,AF_INET6,datap_a,datap_b);
340 }
341
342 static adns_status cs_in6addr(vbuf *vb, const void *datap) {
343   return csp_genaddr(vb,AF_INET6,datap);
344 }
345
346 /*
347  * _addr   (pap,pa,di,div,csp,cs,gsz,qs
348  *              +search_sortlist_sa, dip_sockaddr, addr_rrtypes,
349  *               addr_submit, icb_addr)
350  */
351
352 static const typeinfo tinfo_addrsub;
353
354 #define ADDR_RRTYPES(_) _(a) _(aaaa)
355
356 static const adns_rrtype addr_all_rrtypes[] = {
357 #define RRTY_CODE(ty) adns_r_##ty,
358   ADDR_RRTYPES(RRTY_CODE)
359 #undef RRTY_CODE
360 };
361
362 enum {
363 #define RRTY_INDEX(ty) addr__ri_##ty,
364   ADDR_RRTYPES(RRTY_INDEX)
365 #undef RRTY_INDEX
366   addr_nrrtypes,
367 #define RRTY_FLAG(ty) addr_rf_##ty = 1 << addr__ri_##ty,
368   ADDR_RRTYPES(RRTY_FLAG)
369   addr__rrty_hunoz
370 #undef RRTY_FLAG
371 };
372
373 static unsigned addr_rrtypeflag(adns_rrtype type) {
374   int i;
375
376   type &= adns_rrt_typemask;
377   for (i=0; i<addr_nrrtypes && type!=addr_all_rrtypes[i]; i++);
378   return i < addr_nrrtypes ? 1 << i : 0;
379 }
380
381 /* About CNAME handling in addr queries.
382  *
383  * A user-level addr query is translated into a number of protocol-level
384  * queries, and its job is to reassemble the results.  This gets tricky if
385  * the answers aren't consistent.  In particular, if the answers report
386  * inconsistent indirection via CNAME records (e.g., different CNAMEs, or
387  * some indirect via a CNAME, and some don't) then we have trouble.
388  *
389  * Once we've received an answer, even if it was NODATA, we set
390  * adns__qf_addr_answer on the parent query.  This will let us detect a
391  * conflict between a no-CNAME-with-NODATA reply and a subsequent CNAME.
392  *
393  * If we detect a conflict of any kind, then at least one answer came back
394  * with a CNAME record, so we pick the first such answer (somewhat
395  * arbitrarily) as being the `right' canonical name, and set this in the
396  * parent query's answer->cname slot.  We discard address records from the
397  * wrong name.  And finally we cancel the outstanding child queries, and
398  * resubmit address queries for the address families we don't yet have, with
399  * adns__qf_addr_cname set so that we know that we're in the fixup state.
400  */
401
402 static adns_status pap_addr(const parseinfo *pai, int rrty, size_t rrsz,
403                             int *cbyte_io, int max, adns_rr_addr *storeto) {
404   const byte *dgram= pai->dgram;
405   int af, addrlen, salen;
406   struct in6_addr v6map;
407   const void *oaddr= dgram + *cbyte_io;
408   int avail= max - *cbyte_io;
409   int step= -1;
410   void *addrp= 0;
411
412   switch (rrty) {
413   case adns_r_a:
414     if ((pai->qu->flags & adns_qf_ipv6_mapv4) &&
415         (pai->qu->answer->type & adns__qtf_bigaddr)) {
416       if (avail < 4) return adns_s_invaliddata;
417       memset(v6map.s6_addr +  0, 0x00, 10);
418       memset(v6map.s6_addr + 10, 0xff,  2);
419       memcpy(v6map.s6_addr + 12, oaddr, 4);
420       oaddr= v6map.s6_addr; avail= sizeof(v6map.s6_addr);
421       if (step < 0) step= 4;
422       goto aaaa;
423     }
424     af= AF_INET; addrlen= 4;
425     addrp= &storeto->addr.inet.sin_addr;
426     salen= sizeof(storeto->addr.inet);
427     break;
428   case adns_r_aaaa:
429   aaaa:
430     af= AF_INET6; addrlen= 16;
431     addrp= storeto->addr.inet6.sin6_addr.s6_addr;
432     salen= sizeof(storeto->addr.inet6);
433     break;
434   }
435   assert(addrp);
436
437   assert(offsetof(adns_rr_addr, addr) + salen <= rrsz);
438   if (addrlen < avail) return adns_s_invaliddata;
439   if (step < 0) step= addrlen;
440   *cbyte_io += step;
441   memset(&storeto->addr, 0, salen);
442   storeto->len= salen;
443   storeto->addr.sa.sa_family= af;
444   memcpy(addrp, oaddr, addrlen);
445
446   return adns_s_ok;
447 }
448
449 static adns_status pa_addr(const parseinfo *pai, int cbyte,
450                            int max, void *datap) {
451   int err= pap_addr(pai, pai->qu->answer->type & adns_rrt_typemask,
452                     pai->qu->answer->rrsz, &cbyte, max, datap);
453   if (err) return err;
454   if (cbyte != max) return adns_s_invaliddata;
455   return adns_s_ok;
456 }
457
458 static int search_sortlist_sa(adns_state ads, const struct sockaddr *sa) {
459   union gen_addr a;
460   adns__sockaddr_extract(sa, &a, 0);
461   return search_sortlist(ads, sa->sa_family, &a);
462 }
463
464 static int dip_sockaddr(adns_state ads,
465                         const struct sockaddr *sa,
466                         const struct sockaddr *sb) {
467   if (!ads->sortlist) return 0;
468   return search_sortlist_sa(ads, sa) > search_sortlist_sa(ads, sb);
469 }
470
471 static int di_addr(adns_state ads, const void *datap_a, const void *datap_b) {
472   const adns_rr_addr *ap= datap_a, *bp= datap_b;
473   return dip_sockaddr(ads, &ap->addr.sa, &bp->addr.sa);
474 }
475
476 static int div_addr(void *context, const void *datap_a, const void *datap_b) {
477   const adns_state ads= context;
478
479   return di_addr(ads, datap_a, datap_b);
480 }                    
481
482 static adns_status csp_addr(vbuf *vb, const adns_rr_addr *rrp) {
483   char buf[ADNS_ADDR2TEXT_BUFLEN];
484   int len= sizeof(buf);
485   int err;
486
487   switch (rrp->addr.inet.sin_family) {
488   case AF_INET:
489     CSP_ADDSTR("INET ");
490     goto a2t;
491   case AF_INET6:
492     CSP_ADDSTR("INET6 ");
493     goto a2t;
494   a2t:
495     err= adns_addr2text(&rrp->addr.sa,0, buf,&len, 0); assert(!err);
496     CSP_ADDSTR(buf);
497     break;
498   default:
499     sprintf(buf,"AF=%u",rrp->addr.sa.sa_family);
500     CSP_ADDSTR(buf);
501     break;
502   }
503   return adns_s_ok;
504 }
505
506 static adns_status cs_addr(vbuf *vb, const void *datap) {
507   const adns_rr_addr *rrp= datap;
508
509   return csp_addr(vb,rrp);
510 }
511
512 static int gsz_addr(const typeinfo *typei, adns_rrtype type) {
513   return type & adns__qtf_bigaddr ?
514     sizeof(adns_rr_addr) : sizeof(adns_rr_addr_v4only);
515 }
516
517 static unsigned addr_rrtypes(adns_state ads, adns_rrtype type,
518                              adns_queryflags qf) {
519   /* Return a mask of addr_rf_... flags indicating which address families are
520    * wanted, given a query type and flags.
521    */
522   return addr_rf_a;
523 }
524
525 static void icb_addr(adns_query parent, adns_query child);
526
527 static void addr_subqueries(adns_query qu, struct timeval now,
528                             adns_queryflags qf_extra,
529                             const byte *qd_dgram, int qd_dglen) {
530   int i, err, id;
531   adns_query cqu;
532   adns_queryflags qf= (qu->flags & ~adns_qf_search) | qf_extra;
533   adns_rrtype qtf= qu->answer->type & adns__qtf_deref;
534   unsigned which= qu->ctx.tinfo.addr.want & ~qu->ctx.tinfo.addr.have;
535   qcontext ctx;
536
537   memset(&ctx, 0, sizeof(ctx));
538   ctx.callback= icb_addr;
539   for (i=0; i<addr_nrrtypes; i++) {
540     if (!(which & (1 << i))) continue;
541     err= adns__mkquery_frdgram(qu->ads, &qu->vb, &id, qd_dgram,qd_dglen,
542                                DNS_HDRSIZE, addr_all_rrtypes[i], qf);
543     if (err) goto x_error;
544     err= adns__internal_submit(qu->ads, &cqu, &tinfo_addrsub,
545                                addr_all_rrtypes[i] | qtf,
546                                &qu->vb, id, qf, now, &ctx);
547     if (err) goto x_error;
548     cqu->answer->rrsz= qu->answer->rrsz;
549     cqu->parent= qu;
550     LIST_LINK_TAIL_PART(qu->children, cqu,siblings.);
551   }
552   qu->state= query_childw;
553   LIST_LINK_TAIL(qu->ads->childw, qu);
554   return;
555
556 x_error:
557   adns__query_fail(qu, err);
558 }
559
560 static adns_status addr_submit(adns_query parent, adns_query *query_r,
561                                vbuf *qumsg_vb, int id, unsigned want,
562                                adns_queryflags flags, struct timeval now,
563                                qcontext *ctx) {
564   /* This is effectively a substitute for adns__internal_submit, intended for
565    * the case where the caller (possibly) only wants a subset of the
566    * available record types.  The memory management and callback rules are
567    * the same as for adns__internal_submit.
568    *
569    * Some differences: the query is linked onto the parent's children list
570    * before exit (though the parent's state is not changed, and it is not
571    * linked into the childw list queue); and we fiddle with the `tinfo'
572    * portion of the context structure (yes, modifying *ctx), since this is,
573    * in fact, the main purpose of this function.
574    */
575
576   adns_state ads= parent->ads;
577   adns_query qu;
578   adns_status err;
579   adns_rrtype type= ((adns_r_addr & adns_rrt_reprmask) |
580                      (parent->answer->type & ~adns_rrt_reprmask));
581
582   ctx->tinfo.addr.want= want;
583   ctx->tinfo.addr.have= 0;
584   err= adns__internal_submit(ads, &qu, adns__findtype(adns_r_addr),
585                              type, qumsg_vb, id, flags, now, ctx);
586   if (err) return err;
587
588   qu->parent= parent;
589   LIST_LINK_TAIL_PART(parent->children, qu, siblings.);
590   *query_r= qu;
591   return adns_s_ok;
592 }
593
594 static adns_status append_addrs(adns_query qu, size_t rrsz,
595                                 adns_rr_addr **dp, int *dlen,
596                                 const adns_rr_addr *sp, int slen) {
597   /* Append a vector of slen addr records, each of size rrsz, starting at ap,
598    * to a vector starting at *dp, of length *dlen.  On successful completion,
599    * *dp and *dlen are updated.
600    */
601
602   size_t drrsz= *dlen*rrsz, srrsz= slen*rrsz;
603   byte *p;
604
605   if (!slen) return adns_s_ok;
606   p= adns__alloc_interim(qu, drrsz + srrsz);
607   if (!p) R_NOMEM;
608   if (*dlen) {
609     memcpy(p, *dp, drrsz);
610     adns__free_interim(qu, *dp);
611   }
612   memcpy(p + drrsz, sp, srrsz);
613   *dlen += slen;
614   *dp= (adns_rr_addr *)p;
615   return adns_s_ok;
616 }
617
618 static void propagate_ttl(adns_query to, adns_query from)
619   { if (to->expires > from->expires) to->expires= from->expires; }
620
621 static adns_status copy_cname_from_child(adns_query parent, adns_query child) {
622   adns_answer *pans= parent->answer, *cans= child->answer;
623   size_t n= strlen(cans->cname) + 1;
624
625   pans->cname= adns__alloc_preserved(parent, n);
626   if (!pans->cname) R_NOMEM;
627   memcpy(pans->cname, cans->cname, n);
628   return adns_s_ok;
629 }
630
631 static void done_addr_type(adns_query qu, adns_rrtype type) {
632   unsigned f= addr_rrtypeflag(type);
633   assert(f); qu->ctx.tinfo.addr.have |= f;
634 }
635
636 static void icb_addr(adns_query parent, adns_query child) {
637   adns_state ads= parent->ads;
638   adns_answer *pans= parent->answer, *cans= child->answer;
639   struct timeval now;
640   adns_status err;
641   adns_queryflags qf;
642   int id;
643
644   propagate_ttl(parent, child);
645
646   if (!(child->flags & adns__qf_addr_cname) &&
647       (parent->flags & adns__qf_addr_answer) &&
648       (!!pans->cname != !!cans->cname ||
649        (pans->cname && strcmp(pans->cname, cans->cname)))) {
650     /* We've detected an inconsistency in CNAME records, and must deploy
651      * countermeasures.
652      */
653
654     if (!pans->cname) {
655       /* The child has a CNAME record, but the parent doesn't.  We must
656        * discard all of the parent's addresses, and substitute the child's.
657        */
658
659       assert(pans->rrsz == cans->rrsz);
660       adns__free_interim(parent, pans->rrs.bytes);
661       adns__transfer_interim(child, parent, cans->rrs.bytes);
662       pans->rrs.bytes= cans->rrs.bytes;
663       pans->nrrs= cans->nrrs;
664       parent->ctx.tinfo.addr.have= 0;
665       done_addr_type(parent, cans->type);
666       err= copy_cname_from_child(parent, child); if (err) goto x_err;
667     }
668
669     /* We've settled on the CNAME (now) associated with the parent, which
670      * already has appropriate address records.  Build a query datagram for
671      * this name so that we can issue child queries for the missing address
672      * families.  The child's vbuf looks handy for this.
673      */
674     err= adns__mkquery(ads, &child->vb, &id, pans->cname,
675                        strlen(pans->cname), &tinfo_addrsub,
676                        adns_r_addr, parent->flags);
677     if (err) goto x_err;
678
679     /* Now cancel the remaining children, and try again with the CNAME we've
680      * settled on.
681      */
682     adns__cancel_children(parent);
683     if (gettimeofday(&now, 0)) goto x_gtod;
684     qf= adns__qf_addr_cname;
685     if (!(parent->flags & adns_qf_cname_loose)) qf |= adns_qf_cname_forbid;
686     addr_subqueries(parent, now, qf, child->vb.buf, child->vb.used);
687     return;
688   }
689
690   if (cans->cname && !pans->cname) {
691     err= copy_cname_from_child(parent, child);
692     if (err) goto x_err;
693   }
694
695   if ((parent->flags & adns_qf_search) &&
696       !pans->cname && cans->status == adns_s_nxdomain) {
697     /* We're searching a list of suffixes, and the name doesn't exist.  Try
698      * the next one.
699      */
700
701     adns__cancel_children(parent);
702     adns__free_interim(parent, pans->rrs.bytes);
703     pans->rrs.bytes= 0; pans->nrrs= 0;
704     if (gettimeofday(&now, 0)) goto x_gtod;
705     adns__search_next(ads, parent, now);
706     return;
707   }
708
709   if (cans->status && cans->status != adns_s_nodata)
710     { err= cans->status; goto x_err; }
711
712   assert(pans->rrsz == cans->rrsz);
713   err= append_addrs(parent, pans->rrsz,
714                     &pans->rrs.addr, &pans->nrrs,
715                     cans->rrs.addr, cans->nrrs);
716   if (err) goto x_err;
717   done_addr_type(parent, cans->type);
718
719   if (parent->children.head) LIST_LINK_TAIL(ads->childw, parent);
720   else if (!pans->nrrs) adns__query_fail(parent, adns_s_nodata);
721   else adns__query_done(parent);
722   parent->flags |= adns__qf_addr_answer;
723   return;
724
725 x_gtod:
726   adns__diag(ads, -1, parent, "gettimeofday failed: %s", strerror(errno));
727   err= adns_s_systemfail;
728   goto x_err;
729
730 x_err:
731   adns__query_fail(parent, err);
732 }
733
734 static void qs_addr(adns_query qu, struct timeval now) {
735   if (!qu->ctx.tinfo.addr.want) {
736     qu->ctx.tinfo.addr.want= addr_rrtypes(qu->ads, qu->answer->type,
737                                           qu->flags);
738     qu->ctx.tinfo.addr.have= 0;
739   }
740   addr_subqueries(qu, now, 0, qu->query_dgram, qu->query_dglen);
741 }
742
743 /*
744  * _domain      (pap,csp,cs)
745  * _dom_raw     (pa)
746  */
747
748 static adns_status pap_domain(const parseinfo *pai, int *cbyte_io, int max,
749                               char **domain_r, parsedomain_flags flags) {
750   adns_status st;
751   char *dm;
752   
753   st= adns__parse_domain(pai->qu->ads, pai->serv, pai->qu, &pai->qu->vb, flags,
754                          pai->dgram,pai->dglen, cbyte_io, max);
755   if (st) return st;
756   if (!pai->qu->vb.used) return adns_s_invaliddata;
757
758   dm= adns__alloc_interim(pai->qu, pai->qu->vb.used+1);
759   if (!dm) R_NOMEM;
760
761   dm[pai->qu->vb.used]= 0;
762   memcpy(dm,pai->qu->vb.buf,pai->qu->vb.used);
763   
764   *domain_r= dm;
765   return adns_s_ok;
766 }
767
768 static adns_status csp_domain(vbuf *vb, const char *domain) {
769   CSP_ADDSTR(domain);
770   if (!*domain) CSP_ADDSTR(".");
771   return adns_s_ok;
772 }
773
774 static adns_status cs_domain(vbuf *vb, const void *datap) {
775   const char *const *domainp= datap;
776   return csp_domain(vb,*domainp);
777 }
778
779 static adns_status pa_dom_raw(const parseinfo *pai, int cbyte,
780                               int max, void *datap) {
781   char **rrp= datap;
782   adns_status st;
783
784   st= pap_domain(pai, &cbyte, max, rrp, pdf_quoteok);
785   if (st) return st;
786   
787   if (cbyte != max) return adns_s_invaliddata;
788   return adns_s_ok;
789 }
790
791 /*
792  * _host_raw   (pa)
793  */
794
795 static adns_status pa_host_raw(const parseinfo *pai, int cbyte,
796                                int max, void *datap) {
797   char **rrp= datap;
798   adns_status st;
799
800   st= pap_domain(pai, &cbyte, max, rrp,
801                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
802   if (st) return st;
803   
804   if (cbyte != max) return adns_s_invaliddata;
805   return adns_s_ok;
806 }
807
808 /*
809  * _hostaddr   (pap,pa,dip,di,mfp,mf,csp,cs +pap_findaddrs, icb_hostaddr)
810  */
811
812 static adns_status pap_findaddrs(const parseinfo *pai, adns_rr_hostaddr *ha,
813                                  unsigned *want_io, size_t addrsz,
814                                  int *cbyte_io, int count, int dmstart) {
815   int rri, naddrs;
816   unsigned typef, want= *want_io, need= want;
817   int type, class, rdlen, rdend, rdstart, ownermatched;
818   unsigned long ttl;
819   adns_status st;
820   
821   for (rri=0, naddrs=0; rri<count; rri++) {
822     st= adns__findrr_anychk(pai->qu, pai->serv, pai->dgram,
823                             pai->dglen, cbyte_io,
824                             &type, &class, &ttl, &rdlen, &rdstart,
825                             pai->dgram, pai->dglen, dmstart, &ownermatched);
826     if (st) return st;
827     if (!ownermatched || class != DNS_CLASS_IN) continue;
828     typef= addr_rrtypeflag(type);
829     if (!(want & typef)) continue;
830     need &= ~typef;
831     if (!adns__vbuf_ensure(&pai->qu->vb, (naddrs+1)*addrsz)) R_NOMEM;
832     adns__update_expires(pai->qu,ttl,pai->now);
833     rdend= rdstart + rdlen;
834     st= pap_addr(pai, type, addrsz, &rdstart, rdend,
835                  (adns_rr_addr *)(pai->qu->vb.buf + naddrs*addrsz));
836     if (st) return st;
837     if (rdstart != rdend) return adns_s_invaliddata;
838     naddrs++;
839   }
840   if (naddrs > 0) {
841     st= append_addrs(pai->qu, addrsz, &ha->addrs, &ha->naddrs,
842                      (const adns_rr_addr *)pai->qu->vb.buf, naddrs);
843     if (st) return st;
844     ha->astatus= adns_s_ok;
845
846     if (!need) {
847       adns__isort(ha->addrs, naddrs, addrsz, pai->qu->vb.buf,
848                   div_addr, pai->ads);
849     }
850   }
851   *want_io= need;
852   return adns_s_ok;
853 }
854
855 static void icb_hostaddr(adns_query parent, adns_query child) {
856   adns_answer *cans= child->answer;
857   adns_rr_hostaddr *rrp= child->ctx.pinfo.hostaddr;
858   adns_state ads= parent->ads;
859   adns_status st;
860   size_t addrsz= gsz_addr(0, parent->answer->type);
861
862   st= cans->status == adns_s_nodata ? adns_s_ok : cans->status;
863   if (st) goto done;
864   propagate_ttl(parent, child);
865
866   assert(addrsz == cans->rrsz);
867   st= append_addrs(parent, addrsz,
868                    &rrp->addrs, &rrp->naddrs,
869                    cans->rrs.addr, cans->nrrs);
870   if (st) goto done;
871   if (!rrp->naddrs) { st= adns_s_nodata; goto done; }
872
873   if (!adns__vbuf_ensure(&parent->vb, addrsz))
874     { st= adns_s_nomemory; goto done; }
875   adns__isort(rrp->addrs, rrp->naddrs, addrsz, parent->vb.buf,
876               div_addr, ads);
877
878 done:
879   if (st) {
880     adns__free_interim(parent, rrp->addrs);
881     rrp->naddrs= (st>0 && st<=adns_s_max_tempfail) ? -1 : 0;
882   }
883
884   rrp->astatus= st;
885   if (parent->children.head) {
886     LIST_LINK_TAIL(ads->childw,parent);
887   } else {
888     adns__query_done(parent);
889   }
890 }
891
892 static adns_status pap_hostaddr(const parseinfo *pai, int *cbyte_io,
893                                 int max, adns_rr_hostaddr *rrp) {
894   adns_status st;
895   int dmstart, cbyte;
896   qcontext ctx;
897   int id;
898   adns_query nqu;
899   adns_queryflags nflags;
900   unsigned want;
901   size_t addrsz= gsz_addr(0, pai->qu->answer->type);
902
903   dmstart= cbyte= *cbyte_io;
904   st= pap_domain(pai, &cbyte, max, &rrp->host,
905                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
906   if (st) return st;
907   *cbyte_io= cbyte;
908
909   rrp->astatus= adns_s_ok;
910   rrp->naddrs= 0;
911   rrp->addrs= 0;
912
913   cbyte= pai->nsstart;
914
915   want= addr_rrtypes(pai->ads, pai->qu->answer->type, pai->qu->flags);
916
917   st= pap_findaddrs(pai, rrp, &want, addrsz, &cbyte, pai->nscount, dmstart);
918   if (st) return st;
919   if (!want) return adns_s_ok;
920
921   st= pap_findaddrs(pai, rrp, &want, addrsz, &cbyte, pai->arcount, dmstart);
922   if (st) return st;
923   if (!want) return adns_s_ok;
924
925   st= adns__mkquery_frdgram(pai->ads, &pai->qu->vb, &id,
926                             pai->dgram, pai->dglen, dmstart,
927                             adns_r_addr, adns_qf_quoteok_query);
928   if (st) return st;
929
930   ctx.ext= 0;
931   ctx.callback= icb_hostaddr;
932   ctx.pinfo.hostaddr= rrp;
933   
934   nflags= adns_qf_quoteok_query | (pai->qu->flags & adns_qf_ipv6_mapv4);
935   if (!(pai->qu->flags & adns_qf_cname_loose)) nflags |= adns_qf_cname_forbid;
936   
937   st= addr_submit(pai->qu, &nqu, &pai->qu->vb, id, want,
938                   nflags, pai->now, &ctx);
939   if (st) return st;
940
941   return adns_s_ok;
942 }
943
944 static adns_status pa_hostaddr(const parseinfo *pai, int cbyte,
945                                int max, void *datap) {
946   adns_rr_hostaddr *rrp= datap;
947   adns_status st;
948
949   st= pap_hostaddr(pai, &cbyte, max, rrp);
950   if (st) return st;
951   if (cbyte != max) return adns_s_invaliddata;
952
953   return adns_s_ok;
954 }
955
956 static int dip_hostaddr(adns_state ads,
957                         const adns_rr_hostaddr *ap, const adns_rr_hostaddr *bp) {
958   if (ap->astatus != bp->astatus) return ap->astatus;
959   if (ap->astatus) return 0;
960
961   return dip_sockaddr(ads, &ap->addrs[0].addr.sa, &bp->addrs[0].addr.sa);
962 }
963
964 static int di_hostaddr(adns_state ads,
965                        const void *datap_a, const void *datap_b) {
966   const adns_rr_hostaddr *ap= datap_a, *bp= datap_b;
967
968   return dip_hostaddr(ads, ap,bp);
969 }
970
971 static void mfp_hostaddr(adns_query qu, adns_rr_hostaddr *rrp) {
972   void *tablev;
973   size_t addrsz= gsz_addr(0, qu->answer->type);
974
975   adns__makefinal_str(qu,&rrp->host);
976   tablev= rrp->addrs;
977   adns__makefinal_block(qu, &tablev, rrp->naddrs*addrsz);
978   rrp->addrs= tablev;
979 }
980
981 static void mf_hostaddr(adns_query qu, void *datap) {
982   adns_rr_hostaddr *rrp= datap;
983
984   mfp_hostaddr(qu,rrp);
985 }
986
987 static adns_status csp_hostaddr(vbuf *vb, const adns_rr_hostaddr *rrp) {
988   const char *errstr;
989   adns_status st;
990   char buf[20];
991   int i;
992
993   st= csp_domain(vb,rrp->host);  if (st) return st;
994
995   CSP_ADDSTR(" ");
996   CSP_ADDSTR(adns_errtypeabbrev(rrp->astatus));
997
998   sprintf(buf," %d ",rrp->astatus);
999   CSP_ADDSTR(buf);
1000
1001   CSP_ADDSTR(adns_errabbrev(rrp->astatus));
1002   CSP_ADDSTR(" ");
1003
1004   errstr= adns_strerror(rrp->astatus);
1005   st= csp_qstring(vb,errstr,strlen(errstr));  if (st) return st;
1006   
1007   if (rrp->naddrs >= 0) {
1008     CSP_ADDSTR(" (");
1009     for (i=0; i<rrp->naddrs; i++) {
1010       CSP_ADDSTR(" ");
1011       st= csp_addr(vb,&rrp->addrs[i]);
1012     }
1013     CSP_ADDSTR(" )");
1014   } else {
1015     CSP_ADDSTR(" ?");
1016   }
1017   return adns_s_ok;
1018 }
1019
1020 static adns_status cs_hostaddr(vbuf *vb, const void *datap) {
1021   const adns_rr_hostaddr *rrp= datap;
1022
1023   return csp_hostaddr(vb,rrp);
1024 }
1025
1026 /*
1027  * _mx_raw   (pa,di)
1028  */
1029
1030 static adns_status pa_mx_raw(const parseinfo *pai, int cbyte,
1031                              int max, void *datap) {
1032   const byte *dgram= pai->dgram;
1033   adns_rr_intstr *rrp= datap;
1034   adns_status st;
1035   int pref;
1036
1037   if (cbyte+2 > max) return adns_s_invaliddata;
1038   GET_W(cbyte,pref);
1039   rrp->i= pref;
1040   st= pap_domain(pai, &cbyte, max, &rrp->str,
1041                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
1042   if (st) return st;
1043   
1044   if (cbyte != max) return adns_s_invaliddata;
1045   return adns_s_ok;
1046 }
1047
1048 static int di_mx_raw(adns_state ads, const void *datap_a, const void *datap_b) {
1049   const adns_rr_intstr *ap= datap_a, *bp= datap_b;
1050
1051   if (ap->i < bp->i) return 0;
1052   if (ap->i > bp->i) return 1;
1053   return 0;
1054 }
1055
1056 /*
1057  * _mx   (pa,di)
1058  */
1059
1060 static adns_status pa_mx(const parseinfo *pai, int cbyte,
1061                          int max, void *datap) {
1062   const byte *dgram= pai->dgram;
1063   adns_rr_inthostaddr *rrp= datap;
1064   adns_status st;
1065   int pref;
1066
1067   if (cbyte+2 > max) return adns_s_invaliddata;
1068   GET_W(cbyte,pref);
1069   rrp->i= pref;
1070   st= pap_hostaddr(pai, &cbyte, max, &rrp->ha);
1071   if (st) return st;
1072   
1073   if (cbyte != max) return adns_s_invaliddata;
1074   return adns_s_ok;
1075 }
1076
1077 static int di_mx(adns_state ads, const void *datap_a, const void *datap_b) {
1078   const adns_rr_inthostaddr *ap= datap_a, *bp= datap_b;
1079
1080   if (ap->i < bp->i) return 0;
1081   if (ap->i > bp->i) return 1;
1082   return dip_hostaddr(ads, &ap->ha, &bp->ha);
1083 }
1084
1085 /*
1086  * _inthostaddr  (mf,cs)
1087  */
1088
1089 static void mf_inthostaddr(adns_query qu, void *datap) {
1090   adns_rr_inthostaddr *rrp= datap;
1091
1092   mfp_hostaddr(qu,&rrp->ha);
1093 }
1094
1095 static adns_status cs_inthostaddr(vbuf *vb, const void *datap) {
1096   const adns_rr_inthostaddr *rrp= datap;
1097   char buf[10];
1098
1099   sprintf(buf,"%u ",rrp->i);
1100   CSP_ADDSTR(buf);
1101
1102   return csp_hostaddr(vb,&rrp->ha);
1103 }
1104
1105 /*
1106  * _inthost  (cs)
1107  */
1108
1109 static adns_status cs_inthost(vbuf *vb, const void *datap) {
1110   const adns_rr_intstr *rrp= datap;
1111   char buf[10];
1112
1113   sprintf(buf,"%u ",rrp->i);
1114   CSP_ADDSTR(buf);
1115   return csp_domain(vb,rrp->str);
1116 }
1117
1118 /*
1119  * _ptr   (ckl,pa +icb_ptr)
1120  */
1121
1122 static adns_status ckl_ptr(adns_state ads, adns_queryflags flags,
1123                            union checklabel_state *cls, qcontext *ctx,
1124                            int labnum, const char *label, int lablen) {
1125   if (lablen) {
1126     if (adns__revparse_label(&cls->ptr, labnum, label,lablen))
1127       return adns_s_querydomainwrong;
1128   } else {
1129     if (adns__revparse_done(&cls->ptr, labnum,
1130                             &ctx->tinfo.ptr.rev_rrtype,
1131                             &ctx->tinfo.ptr.addr))
1132       return adns_s_querydomainwrong;
1133   }
1134   return adns_s_ok;
1135 }
1136
1137 static void icb_ptr(adns_query parent, adns_query child) {
1138   adns_answer *cans= child->answer;
1139   const struct af_addr *queried;
1140   const unsigned char *found;
1141   adns_state ads= parent->ads;
1142   int i;
1143
1144   if (cans->status == adns_s_nxdomain || cans->status == adns_s_nodata) {
1145     adns__query_fail(parent,adns_s_inconsistent);
1146     return;
1147   } else if (cans->status) {
1148     adns__query_fail(parent,cans->status);
1149     return;
1150   }
1151
1152   queried= &parent->ctx.tinfo.ptr.addr;
1153   for (i=0, found=cans->rrs.bytes; i<cans->nrrs; i++, found+=cans->rrsz) {
1154     if (adns__genaddr_equal_p(queried->af,&queried->addr,
1155                               parent->ctx.tinfo.ptr.addr.af,found)) {
1156       if (!parent->children.head) {
1157         adns__query_done(parent);
1158         return;
1159       } else {
1160         LIST_LINK_TAIL(ads->childw,parent);
1161         return;
1162       }
1163     }
1164   }
1165
1166   adns__query_fail(parent,adns_s_inconsistent);
1167 }
1168
1169 static adns_status pa_ptr(const parseinfo *pai, int dmstart,
1170                           int max, void *datap) {
1171   char **rrp= datap;
1172   adns_status st;
1173   adns_rrtype rrtype= pai->qu->ctx.tinfo.ptr.rev_rrtype;
1174   int cbyte, id;
1175   adns_query nqu;
1176   qcontext ctx;
1177
1178   cbyte= dmstart;
1179   st= pap_domain(pai, &cbyte, max, rrp,
1180                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
1181   if (st) return st;
1182   if (cbyte != max) return adns_s_invaliddata;
1183
1184   st= adns__mkquery_frdgram(pai->ads, &pai->qu->vb, &id,
1185                             pai->dgram, pai->dglen, dmstart,
1186                             rrtype, adns_qf_quoteok_query);
1187   if (st) return st;
1188
1189   ctx.ext= 0;
1190   ctx.callback= icb_ptr;
1191   memset(&ctx.pinfo,0,sizeof(ctx.pinfo));
1192   memset(&ctx.tinfo,0,sizeof(ctx.tinfo));
1193   st= adns__internal_submit(pai->ads, &nqu, adns__findtype(rrtype),
1194                             rrtype, &pai->qu->vb, id,
1195                             adns_qf_quoteok_query, pai->now, &ctx);
1196   if (st) return st;
1197
1198   nqu->parent= pai->qu;
1199   LIST_LINK_TAIL_PART(pai->qu->children,nqu,siblings.);
1200   return adns_s_ok;
1201 }
1202
1203 /*
1204  * _strpair   (mf)
1205  */
1206
1207 static void mf_strpair(adns_query qu, void *datap) {
1208   adns_rr_strpair *rrp= datap;
1209
1210   adns__makefinal_str(qu,&rrp->array[0]);
1211   adns__makefinal_str(qu,&rrp->array[1]);
1212 }
1213
1214 /*
1215  * _intstrpair   (mf)
1216  */
1217
1218 static void mf_intstrpair(adns_query qu, void *datap) {
1219   adns_rr_intstrpair *rrp= datap;
1220
1221   adns__makefinal_str(qu,&rrp->array[0].str);
1222   adns__makefinal_str(qu,&rrp->array[1].str);
1223 }
1224
1225 /*
1226  * _hinfo   (pa)
1227  */
1228
1229 static adns_status pa_hinfo(const parseinfo *pai, int cbyte,
1230                             int max, void *datap) {
1231   adns_rr_intstrpair *rrp= datap;
1232   adns_status st;
1233   int i;
1234
1235   for (i=0; i<2; i++) {
1236     st= pap_qstring(pai, &cbyte, max, &rrp->array[i].i, &rrp->array[i].str);
1237     if (st) return st;
1238   }
1239
1240   if (cbyte != max) return adns_s_invaliddata;
1241   
1242   return adns_s_ok;
1243 }
1244
1245 /*
1246  * _mailbox   (pap,cs +pap_mailbox822)
1247  */
1248
1249 static adns_status pap_mailbox822(const parseinfo *pai,
1250                                   int *cbyte_io, int max, char **mb_r) {
1251   int lablen, labstart, i, needquote, c, r, neednorm;
1252   const unsigned char *p;
1253   char *str;
1254   findlabel_state fls;
1255   adns_status st;
1256   vbuf *vb;
1257
1258   vb= &pai->qu->vb;
1259   vb->used= 0;
1260   adns__findlabel_start(&fls, pai->ads,
1261                         -1, pai->qu,
1262                         pai->dgram, pai->dglen, max,
1263                         *cbyte_io, cbyte_io);
1264   st= adns__findlabel_next(&fls,&lablen,&labstart);
1265   if (!lablen) {
1266     adns__vbuf_appendstr(vb,".");
1267     goto x_ok;
1268   }
1269
1270   neednorm= 1;
1271   for (i=0, needquote=0, p= pai->dgram+labstart; i<lablen; i++) {
1272     c= *p++;
1273     if ((c&~128) < 32 || (c&~128) == 127) return adns_s_invaliddata;
1274     if (c == '.' && !neednorm) neednorm= 1;
1275     else if (c==' ' || c>=127 || ctype_822special(c)) needquote++;
1276     else neednorm= 0;
1277   }
1278
1279   if (needquote || neednorm) {
1280     r= adns__vbuf_ensure(vb, lablen+needquote+4); if (!r) R_NOMEM;
1281     adns__vbuf_appendq(vb,"\"",1);
1282     for (i=0, needquote=0, p= pai->dgram+labstart; i<lablen; i++, p++) {
1283       c= *p;
1284       if (c == '"' || c=='\\') adns__vbuf_appendq(vb,"\\",1);
1285       adns__vbuf_appendq(vb,p,1);
1286     }
1287     adns__vbuf_appendq(vb,"\"",1);
1288   } else {
1289     r= adns__vbuf_append(vb, pai->dgram+labstart, lablen); if (!r) R_NOMEM;
1290   }
1291
1292   r= adns__vbuf_appendstr(vb,"@"); if (!r) R_NOMEM;
1293
1294   st= adns__parse_domain_more(&fls,pai->ads, pai->qu,vb,0, pai->dgram);
1295   if (st) return st;
1296
1297  x_ok:
1298   str= adns__alloc_interim(pai->qu, vb->used+1); if (!str) R_NOMEM;
1299   memcpy(str,vb->buf,vb->used);
1300   str[vb->used]= 0;
1301   *mb_r= str;
1302   return adns_s_ok;
1303 }
1304
1305 static adns_status pap_mailbox(const parseinfo *pai, int *cbyte_io, int max,
1306                                char **mb_r) {
1307   if (pai->qu->typei->typekey & adns__qtf_mail822) {
1308     return pap_mailbox822(pai, cbyte_io, max, mb_r);
1309   } else {
1310     return pap_domain(pai, cbyte_io, max, mb_r, pdf_quoteok);
1311   }
1312 }
1313
1314 static adns_status csp_mailbox(vbuf *vb, const char *mailbox) {
1315   return csp_domain(vb,mailbox);
1316 }
1317
1318 /*
1319  * _rp   (pa,cs)
1320  */
1321
1322 static adns_status pa_rp(const parseinfo *pai, int cbyte,
1323                          int max, void *datap) {
1324   adns_rr_strpair *rrp= datap;
1325   adns_status st;
1326
1327   st= pap_mailbox(pai, &cbyte, max, &rrp->array[0]);
1328   if (st) return st;
1329
1330   st= pap_domain(pai, &cbyte, max, &rrp->array[1], pdf_quoteok);
1331   if (st) return st;
1332
1333   if (cbyte != max) return adns_s_invaliddata;
1334   return adns_s_ok;
1335 }
1336
1337 static adns_status cs_rp(vbuf *vb, const void *datap) {
1338   const adns_rr_strpair *rrp= datap;
1339   adns_status st;
1340
1341   st= csp_mailbox(vb,rrp->array[0]);  if (st) return st;
1342   CSP_ADDSTR(" ");
1343   st= csp_domain(vb,rrp->array[1]);  if (st) return st;
1344
1345   return adns_s_ok;
1346 }  
1347
1348 /*
1349  * _soa   (pa,mf,cs)
1350  */
1351
1352 static adns_status pa_soa(const parseinfo *pai, int cbyte,
1353                           int max, void *datap) {
1354   adns_rr_soa *rrp= datap;
1355   const byte *dgram= pai->dgram;
1356   adns_status st;
1357   int msw, lsw, i;
1358
1359   st= pap_domain(pai, &cbyte, max, &rrp->mname,
1360                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
1361   if (st) return st;
1362
1363   st= pap_mailbox(pai, &cbyte, max, &rrp->rname);
1364   if (st) return st;
1365
1366   if (cbyte+20 != max) return adns_s_invaliddata;
1367   
1368   for (i=0; i<5; i++) {
1369     GET_W(cbyte,msw);
1370     GET_W(cbyte,lsw);
1371     (&rrp->serial)[i]= (msw<<16) | lsw;
1372   }
1373
1374   return adns_s_ok;
1375 }
1376
1377 static void mf_soa(adns_query qu, void *datap) {
1378   adns_rr_soa *rrp= datap;
1379
1380   adns__makefinal_str(qu,&rrp->mname);
1381   adns__makefinal_str(qu,&rrp->rname);
1382 }
1383
1384 static adns_status cs_soa(vbuf *vb, const void *datap) {
1385   const adns_rr_soa *rrp= datap;
1386   char buf[20];
1387   int i;
1388   adns_status st;
1389   
1390   st= csp_domain(vb,rrp->mname);  if (st) return st;
1391   CSP_ADDSTR(" ");
1392   st= csp_mailbox(vb,rrp->rname);  if (st) return st;
1393
1394   for (i=0; i<5; i++) {
1395     sprintf(buf," %lu",(&rrp->serial)[i]);
1396     CSP_ADDSTR(buf);
1397   }
1398
1399   return adns_s_ok;
1400 }
1401
1402 /*
1403  * _srv*  (ckl,(pap),pa*2,mf*2,di,(csp),cs*2,postsort)
1404  */
1405
1406 static adns_status ckl_srv(adns_state ads, adns_queryflags flags,
1407                            union checklabel_state *cls, qcontext *ctx,
1408                            int labnum, const char *label, int lablen) {
1409   if (labnum < 2 && !(flags & adns_qf_quoteok_query)) {
1410     if (!lablen || label[0] != '_') return adns_s_querydomaininvalid;
1411     return adns_s_ok;
1412   }
1413   return adns__ckl_hostname(ads, flags, cls, ctx, labnum, label, lablen);
1414 }
1415
1416 static adns_status pap_srv_begin(const parseinfo *pai, int *cbyte_io, int max,
1417                                  adns_rr_srvha *rrp
1418                                    /* might be adns_rr_srvraw* */) {
1419   const byte *dgram= pai->dgram;
1420   int ti, cbyte;
1421
1422   cbyte= *cbyte_io;
1423   if ((*cbyte_io += 6) > max) return adns_s_invaliddata;
1424   
1425   rrp->priority= GET_W(cbyte, ti);
1426   rrp->weight=   GET_W(cbyte, ti);
1427   rrp->port=     GET_W(cbyte, ti);
1428   return adns_s_ok;
1429 }
1430
1431 static adns_status pa_srvraw(const parseinfo *pai, int cbyte,
1432                              int max, void *datap) {
1433   adns_rr_srvraw *rrp= datap;
1434   adns_status st;
1435
1436   st= pap_srv_begin(pai,&cbyte,max,datap);
1437   if (st) return st;
1438   
1439   st= pap_domain(pai, &cbyte, max, &rrp->host,
1440                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
1441   if (st) return st;
1442   
1443   if (cbyte != max) return adns_s_invaliddata;
1444   return adns_s_ok;
1445 }
1446
1447 static adns_status pa_srvha(const parseinfo *pai, int cbyte,
1448                             int max, void *datap) {
1449   adns_rr_srvha *rrp= datap;
1450   adns_status st;
1451
1452   st= pap_srv_begin(pai,&cbyte,max,datap);       if (st) return st;
1453   st= pap_hostaddr(pai, &cbyte, max, &rrp->ha);  if (st) return st;
1454   if (cbyte != max) return adns_s_invaliddata;
1455   return adns_s_ok;
1456 }
1457
1458 static void mf_srvraw(adns_query qu, void *datap) {
1459   adns_rr_srvraw *rrp= datap;
1460   adns__makefinal_str(qu, &rrp->host);
1461 }
1462
1463 static void mf_srvha(adns_query qu, void *datap) {
1464   adns_rr_srvha *rrp= datap;
1465   mfp_hostaddr(qu,&rrp->ha);
1466 }
1467
1468 static int di_srv(adns_state ads, const void *datap_a, const void *datap_b) {
1469   const adns_rr_srvraw *ap= datap_a, *bp= datap_b;
1470     /* might be const adns_rr_svhostaddr* */
1471
1472   if (ap->priority < bp->priority) return 0;
1473   if (ap->priority > bp->priority) return 1;
1474   return 0;
1475 }
1476
1477 static adns_status csp_srv_begin(vbuf *vb, const adns_rr_srvha *rrp
1478                                    /* might be adns_rr_srvraw* */) {
1479   char buf[30];
1480   sprintf(buf,"%u %u %u ", rrp->priority, rrp->weight, rrp->port);
1481   CSP_ADDSTR(buf);
1482   return adns_s_ok;
1483 }
1484
1485 static adns_status cs_srvraw(vbuf *vb, const void *datap) {
1486   const adns_rr_srvraw *rrp= datap;
1487   adns_status st;
1488   
1489   st= csp_srv_begin(vb,(const void*)rrp);  if (st) return st;
1490   return csp_domain(vb,rrp->host);
1491 }
1492
1493 static adns_status cs_srvha(vbuf *vb, const void *datap) {
1494   const adns_rr_srvha *rrp= datap;
1495   adns_status st;
1496
1497   st= csp_srv_begin(vb,(const void*)datap);  if (st) return st;
1498   return csp_hostaddr(vb,&rrp->ha);
1499 }
1500
1501 static void postsort_srv(adns_state ads, void *array, int nrrs,int rrsz,
1502                          const struct typeinfo *typei) {
1503   /* we treat everything in the array as if it were an adns_rr_srvha
1504    * even though the array might be of adns_rr_srvraw.  That's OK
1505    * because they have the same prefix, which is all we access.
1506    * We use rrsz, too, rather than naive array indexing, of course.
1507    */
1508   char *workbegin, *workend, *search, *arrayend;
1509   const adns_rr_srvha *rr;
1510   union { adns_rr_srvha ha; adns_rr_srvraw raw; } rrtmp;
1511   int cpriority, totalweight, runtotal;
1512   long randval;
1513
1514   assert(rrsz <= sizeof(rrtmp));
1515   for (workbegin= array, arrayend= workbegin + rrsz * nrrs;
1516        workbegin < arrayend;
1517        workbegin= workend) {
1518     cpriority= (rr=(void*)workbegin)->priority;
1519     
1520     for (workend= workbegin, totalweight= 0;
1521          workend < arrayend && (rr=(void*)workend)->priority == cpriority;
1522          workend += rrsz) {
1523       totalweight += rr->weight;
1524     }
1525
1526     /* Now workbegin..(workend-1) incl. are exactly all of the RRs of
1527      * cpriority.  From now on, workbegin points to the `remaining'
1528      * records: we select one record at a time (RFC2782 `Usage rules'
1529      * and `Format of the SRV RR' subsection `Weight') to place at
1530      * workbegin (swapping with the one that was there, and then
1531      * advance workbegin. */
1532     for (;
1533          workbegin + rrsz < workend; /* don't bother if just one */
1534          workbegin += rrsz) {
1535       
1536       randval= nrand48(ads->rand48xsubi);
1537       randval %= (totalweight + 1);
1538         /* makes it into 0..totalweight inclusive; with 2^10 RRs,
1539          * totalweight must be <= 2^26 so probability nonuniformity is
1540          * no worse than 1 in 2^(31-26) ie 1 in 2^5, ie
1541          *  abs(log(P_intended(RR_i) / P_actual(RR_i)) <= log(2^-5).
1542          */
1543
1544       for (search=workbegin, runtotal=0;
1545            (runtotal += (rr=(void*)search)->weight) < randval;
1546            search += rrsz);
1547       assert(search < arrayend);
1548       totalweight -= rr->weight;
1549       if (search != workbegin) {
1550         memcpy(&rrtmp, workbegin, rrsz);
1551         memcpy(workbegin, search, rrsz);
1552         memcpy(search, &rrtmp, rrsz);
1553       }
1554     }
1555   }
1556   /* tests:
1557    *  dig -t srv _srv._tcp.test.iwj.relativity.greenend.org.uk.
1558    *   ./adnshost_s -t srv- _sip._udp.voip.net.cam.ac.uk.
1559    *   ./adnshost_s -t srv- _jabber._tcp.jabber.org
1560    */
1561 }
1562
1563 /*
1564  * _byteblock   (mf)
1565  */
1566
1567 static void mf_byteblock(adns_query qu, void *datap) {
1568   adns_rr_byteblock *rrp= datap;
1569   void *bytes= rrp->data;
1570   adns__makefinal_block(qu,&bytes,rrp->len);
1571   rrp->data= bytes;
1572 }
1573
1574 /*
1575  * _opaque   (pa,cs)
1576  */
1577
1578 static adns_status pa_opaque(const parseinfo *pai, int cbyte,
1579                              int max, void *datap) {
1580   adns_rr_byteblock *rrp= datap;
1581
1582   rrp->len= max - cbyte;
1583   rrp->data= adns__alloc_interim(pai->qu, rrp->len);
1584   if (!rrp->data) R_NOMEM;
1585   memcpy(rrp->data, pai->dgram + cbyte, rrp->len);
1586   return adns_s_ok;
1587 }
1588
1589 static adns_status cs_opaque(vbuf *vb, const void *datap) {
1590   const adns_rr_byteblock *rrp= datap;
1591   char buf[10];
1592   int l;
1593   unsigned char *p;
1594
1595   sprintf(buf,"\\# %d",rrp->len);
1596   CSP_ADDSTR(buf);
1597   
1598   for (l= rrp->len, p= rrp->data;
1599        l>=4;
1600        l -= 4, p += 4) {
1601     sprintf(buf," %02x%02x%02x%02x",p[0],p[1],p[2],p[3]);
1602     CSP_ADDSTR(buf);
1603   }
1604   for (;
1605        l>0;
1606        l--, p++) {
1607     sprintf(buf," %02x",*p);
1608     CSP_ADDSTR(buf);
1609   }
1610   return adns_s_ok;
1611 }
1612   
1613 /*
1614  * _flat   (mf)
1615  */
1616
1617 static void mf_flat(adns_query qu, void *data) { }
1618
1619 /*
1620  * Now the table.
1621  */
1622
1623 #define TYPESZ_M(member)           (sizeof(*((adns_answer*)0)->rrs.member))
1624
1625 #define DEEP_TYPE(code,rrt,fmt,memb,parser,comparer,/*printer*/...)     \
1626  { adns_r_##code&adns_rrt_reprmask, rrt,fmt,TYPESZ_M(memb), mf_##memb,  \
1627      GLUE(cs_, CAR(__VA_ARGS__)),pa_##parser,di_##comparer,             \
1628      adns__ckl_hostname, 0, adns__getrrsz_default, adns__query_send,    \
1629      CDR(__VA_ARGS__) }
1630 #define FLAT_TYPE(code,rrt,fmt,memb,parser,comparer,/*printer*/...)     \
1631  { adns_r_##code&adns_rrt_reprmask, rrt,fmt,TYPESZ_M(memb), mf_flat,    \
1632      GLUE(cs_, CAR(__VA_ARGS__)),pa_##parser,di_##comparer,             \
1633      adns__ckl_hostname, 0, adns__getrrsz_default, adns__query_send,    \
1634      CDR(__VA_ARGS__) }
1635
1636 #define di_0 0
1637
1638 static const typeinfo typeinfos[] = {
1639 /* Must be in ascending order of rrtype ! */
1640 /* mem-mgmt code  rrt     fmt   member   parser      comparer  printer */
1641
1642 FLAT_TYPE(a,      "A",     0,   inaddr,    inaddr,  inaddr,inaddr          ),
1643 DEEP_TYPE(ns_raw, "NS",   "raw",str,       host_raw,0,     domain          ),
1644 DEEP_TYPE(cname,  "CNAME", 0,   str,       dom_raw, 0,     domain          ),
1645 DEEP_TYPE(soa_raw,"SOA",  "raw",soa,       soa,     0,     soa             ),
1646 DEEP_TYPE(ptr_raw,"PTR",  "raw",str,       host_raw,0,     domain          ),
1647 DEEP_TYPE(hinfo,  "HINFO", 0,   intstrpair,hinfo,   0,     hinfo           ),
1648 DEEP_TYPE(mx_raw, "MX",   "raw",intstr,    mx_raw,  mx_raw,inthost         ),
1649 DEEP_TYPE(txt,    "TXT",   0,   manyistr,  txt,     0,     txt             ),
1650 DEEP_TYPE(rp_raw, "RP",   "raw",strpair,   rp,      0,     rp              ),
1651 FLAT_TYPE(aaaa,   "AAAA",  0,   in6addr,   in6addr, in6addr,in6addr        ),
1652 DEEP_TYPE(srv_raw,"SRV",  "raw",srvraw ,   srvraw,  srv,   srvraw,
1653                               .checklabel= ckl_srv, .postsort= postsort_srv),
1654
1655 FLAT_TYPE(addr,   "A",  "addr", addr,      addr,    addr,  addr,
1656                                    .getrrsz= gsz_addr, .query_send= qs_addr),
1657 DEEP_TYPE(ns,     "NS", "+addr",hostaddr,  hostaddr,hostaddr,hostaddr      ),
1658 DEEP_TYPE(ptr,    "PTR","checked",str,     ptr,     0,     domain,
1659                                                        .checklabel= ckl_ptr),
1660 DEEP_TYPE(mx,     "MX", "+addr",inthostaddr,mx,     mx,    inthostaddr,    ),
1661 DEEP_TYPE(srv,    "SRV","+addr",srvha,     srvha,   srv,   srvha,
1662                               .checklabel= ckl_srv, .postsort= postsort_srv),
1663
1664 DEEP_TYPE(soa,    "SOA","822",  soa,       soa,     0,     soa             ),
1665 DEEP_TYPE(rp,     "RP", "822",  strpair,   rp,      0,     rp              ),
1666 };
1667
1668 static const typeinfo tinfo_addrsub =
1669 FLAT_TYPE(none,   "<addr>","sub",addr,     addr,    0,     addr,
1670                                                          .getrrsz= gsz_addr);
1671
1672 static const typeinfo typeinfo_unknown=
1673 DEEP_TYPE(unknown,0, "unknown",byteblock,opaque,  0,     opaque            );
1674
1675 const typeinfo *adns__findtype(adns_rrtype type) {
1676   const typeinfo *begin, *end, *mid;
1677
1678   if (type & adns_r_unknown) return &typeinfo_unknown;
1679   type &= adns_rrt_reprmask;
1680
1681   begin= typeinfos;  end= typeinfos+(sizeof(typeinfos)/sizeof(typeinfo));
1682
1683   while (begin < end) {
1684     mid= begin + ((end-begin)>>1);
1685     if (mid->typekey == type) return mid;
1686     if (type > mid->typekey) begin= mid+1;
1687     else end= mid;
1688   }
1689   return 0;
1690 }