chiark / gitweb /
90413a0bd4a2d98849a0094b97d7e16ae6a439b8
[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 <stdlib.h>
28
29 #include <sys/types.h>
30 #include <sys/socket.h>
31 #include <netinet/in.h>
32 #include <arpa/inet.h>
33
34 #include "internal.h"
35
36 #define R_NOMEM       return adns_s_nomemory
37 #define CSP_ADDSTR(s) do {                      \
38     if (!adns__vbuf_appendstr(vb,(s))) R_NOMEM; \
39   } while (0)
40
41 /*
42  * order of sections:
43  *
44  * _string                    (pap)
45  * _textdata, _qstring        (csp)
46  * _str                       (mf,cs)
47  * _intstr                    (mf,csp,cs)
48  * _manyistr                  (mf,cs)
49  * _txt                       (pa)
50  * _inaddr                    (pa,dip,di,cs +search_sortlist)
51  * _addr                      (pa,di,div,csp,cs)
52  * _domain                    (pap,csp,cs)
53  * _dom_raw                   (pa)
54  * _host_raw                  (pa)
55  * _hostaddr                  (pap,pa,dip,di,mfp,mf,csp,cs
56  *                              +pap_findaddrs, icb_hostaddr)
57  * _mx_raw                    (pa,di)
58  * _mx                        (pa,di)
59  * _inthostaddr               (mf,cs)
60  * _inthost                   (cs)
61  * _ptr                       (ckl,pa +icb_ptr)
62  * _strpair                   (mf)
63  * _intstrpair                (mf)
64  * _hinfo                     (pa)
65  * _mailbox                   (pap,csp +pap_mailbox822)
66  * _rp                        (pa,cs)
67  * _soa                       (pa,mf,cs)
68  * _srv*                      (ckl,(pap),pa*2,mf*2,di,(csp),cs*2,postsort)
69  * _byteblock                 (mf)
70  * _opaque                    (pa,cs)
71  * _flat                      (mf)
72  *
73  * within each section:
74  *    ckl_*
75  *    pap_*
76  *    pa_*
77  *    dip_*
78  *    di_*
79  *    mfp_*
80  *    mf_*
81  *    csp_*
82  *    cs_*
83  *    postsort_*
84  */
85
86 /*
87  * _qstring               (pap,csp)
88  */
89
90 static adns_status pap_qstring(const parseinfo *pai, int *cbyte_io, int max,
91                               int *len_r, char **str_r) {
92   /* Neither len_r nor str_r may be null.
93    * End of datagram (overrun) is indicated by returning adns_s_invaliddata;
94    */
95   const byte *dgram= pai->dgram;
96   int l, cbyte;
97   char *str;
98
99   cbyte= *cbyte_io;
100
101   if (cbyte >= max) return adns_s_invaliddata;
102   GET_B(cbyte,l);
103   if (cbyte+l > max) return adns_s_invaliddata;
104   
105   str= adns__alloc_interim(pai->qu, l+1);
106   if (!str) R_NOMEM;
107   
108   str[l]= 0;
109   memcpy(str,dgram+cbyte,l);
110
111   *len_r= l;
112   *str_r= str;
113   *cbyte_io= cbyte+l;
114   
115   return adns_s_ok;
116 }
117
118 static adns_status csp_qstring(vbuf *vb, const char *dp, int len) {
119   unsigned char ch;
120   char buf[10];
121   int cn;
122
123   CSP_ADDSTR("\"");
124   for (cn=0; cn<len; cn++) {
125     ch= *dp++;
126     if (ch == '\\') {
127       CSP_ADDSTR("\\\\");
128     } else if (ch == '"') {
129       CSP_ADDSTR("\\\"");
130     } else if (ch >= 32 && ch <= 126) {
131       if (!adns__vbuf_append(vb,&ch,1)) R_NOMEM;
132     } else {
133       sprintf(buf,"\\x%02x",ch);
134       CSP_ADDSTR(buf);
135     }
136   }
137   CSP_ADDSTR("\"");
138   
139   return adns_s_ok;
140 }
141
142 /*
143  * _str  (mf)
144  */
145
146 static void mf_str(adns_query qu, void *datap) {
147   char **rrp= datap;
148
149   adns__makefinal_str(qu,rrp);
150 }
151
152 /*
153  * _intstr  (mf)
154  */
155
156 static void mf_intstr(adns_query qu, void *datap) {
157   adns_rr_intstr *rrp= datap;
158
159   adns__makefinal_str(qu,&rrp->str);
160 }
161
162 /*
163  * _manyistr   (mf)
164  */
165
166 static void mf_manyistr(adns_query qu, void *datap) {
167   adns_rr_intstr **rrp= datap;
168   adns_rr_intstr *te, *table;
169   void *tablev;
170   int tc;
171
172   for (tc=0, te= *rrp; te->i >= 0; te++, tc++);
173   tablev= *rrp;
174   adns__makefinal_block(qu,&tablev,sizeof(*te)*(tc+1));
175   *rrp= table= tablev;
176   for (te= *rrp; te->i >= 0; te++)
177     adns__makefinal_str(qu,&te->str);
178 }
179
180 /*
181  * _txt   (pa,cs)
182  */
183
184 static adns_status pa_txt(const parseinfo *pai, int cbyte,
185                           int max, void *datap) {
186   adns_rr_intstr **rrp= datap, *table, *te;
187   const byte *dgram= pai->dgram;
188   int ti, tc, l, startbyte;
189   adns_status st;
190
191   startbyte= cbyte;
192   if (cbyte >= max) return adns_s_invaliddata;
193   tc= 0;
194   while (cbyte < max) {
195     GET_B(cbyte,l);
196     cbyte+= l;
197     tc++;
198   }
199   if (cbyte != max || !tc) return adns_s_invaliddata;
200
201   table= adns__alloc_interim(pai->qu,sizeof(*table)*(tc+1));
202   if (!table) R_NOMEM;
203
204   for (cbyte=startbyte, ti=0, te=table; ti<tc; ti++, te++) {
205     st= pap_qstring(pai, &cbyte, max, &te->i, &te->str);
206     if (st) return st;
207   }
208   assert(cbyte == max);
209
210   te->i= -1;
211   te->str= 0;
212   
213   *rrp= table;
214   return adns_s_ok;
215 }
216
217 static adns_status cs_txt(vbuf *vb, const void *datap) {
218   const adns_rr_intstr *const *rrp= datap;
219   const adns_rr_intstr *current;
220   adns_status st;
221   int spc;
222
223   for (current= *rrp, spc=0;  current->i >= 0;  current++, spc=1) {
224     if (spc) CSP_ADDSTR(" ");
225     st= csp_qstring(vb,current->str,current->i); if (st) return st;
226   }
227   return adns_s_ok;
228 }
229
230 /*
231  * _hinfo   (cs)
232  */
233
234 static adns_status cs_hinfo(vbuf *vb, const void *datap) {
235   const adns_rr_intstrpair *rrp= datap;
236   adns_status st;
237
238   st= csp_qstring(vb,rrp->array[0].str,rrp->array[0].i);  if (st) return st;
239   CSP_ADDSTR(" ");
240   st= csp_qstring(vb,rrp->array[1].str,rrp->array[1].i);  if (st) return st;
241   return adns_s_ok;
242 }
243
244 /*
245  * _inaddr   (pa,dip,di,cs +search_sortlist)
246  */
247
248 static adns_status pa_inaddr(const parseinfo *pai, int cbyte,
249                              int max, void *datap) {
250   struct in_addr *storeto= datap;
251   
252   if (max-cbyte != 4) return adns_s_invaliddata;
253   memcpy(storeto, pai->dgram + cbyte, 4);
254   return adns_s_ok;
255 }
256
257 static int search_sortlist(adns_state ads, struct in_addr ad) {
258   const struct sortlist *slp;
259   int i;
260   
261   for (i=0, slp=ads->sortlist;
262        i<ads->nsortlist &&
263          !((ad.s_addr & slp->mask.s_addr) == slp->base.s_addr);
264        i++, slp++);
265   return i;
266 }
267
268 static int dip_inaddr(adns_state ads, struct in_addr a, struct in_addr b) {
269   int ai, bi;
270   
271   if (!ads->nsortlist) return 0;
272
273   ai= search_sortlist(ads,a);
274   bi= search_sortlist(ads,b);
275   return bi<ai;
276 }
277
278 static int di_inaddr(adns_state ads,
279                      const void *datap_a, const void *datap_b) {
280   const struct in_addr *ap= datap_a, *bp= datap_b;
281
282   return dip_inaddr(ads,*ap,*bp);
283 }
284
285 static adns_status cs_inaddr(vbuf *vb, const void *datap) {
286   const struct in_addr *rrp= datap, rr= *rrp;
287   const char *ia;
288
289   ia= inet_ntoa(rr); assert(ia);
290   CSP_ADDSTR(ia);
291   return adns_s_ok;
292 }
293
294 /*
295  * _addr   (pa,di,div,csp,cs)
296  */
297
298 static adns_status pa_addr(const parseinfo *pai, int cbyte,
299                            int max, void *datap) {
300   adns_rr_addr *storeto= datap;
301   const byte *dgram= pai->dgram;
302
303   if (max-cbyte != 4) return adns_s_invaliddata;
304   storeto->len= sizeof(storeto->addr.inet);
305   memset(&storeto->addr,0,sizeof(storeto->addr.inet));
306   storeto->addr.inet.sin_family= AF_INET;
307   memcpy(&storeto->addr.inet.sin_addr,dgram+cbyte,4);
308   return adns_s_ok;
309 }
310
311 static int di_addr(adns_state ads, const void *datap_a, const void *datap_b) {
312   const adns_rr_addr *ap= datap_a, *bp= datap_b;
313
314   assert(ap->addr.sa.sa_family == AF_INET);
315   return dip_inaddr(ads, ap->addr.inet.sin_addr, bp->addr.inet.sin_addr);
316 }
317
318 static int div_addr(void *context, const void *datap_a, const void *datap_b) {
319   const adns_state ads= context;
320
321   return di_addr(ads, datap_a, datap_b);
322 }                    
323
324 static adns_status csp_addr(vbuf *vb, const adns_rr_addr *rrp) {
325   const char *ia;
326   char buf[30];
327
328   switch (rrp->addr.inet.sin_family) {
329   case AF_INET:
330     CSP_ADDSTR("INET ");
331     ia= inet_ntoa(rrp->addr.inet.sin_addr); assert(ia);
332     CSP_ADDSTR(ia);
333     break;
334   default:
335     sprintf(buf,"AF=%u",rrp->addr.sa.sa_family);
336     CSP_ADDSTR(buf);
337     break;
338   }
339   return adns_s_ok;
340 }
341
342 static adns_status cs_addr(vbuf *vb, const void *datap) {
343   const adns_rr_addr *rrp= datap;
344
345   return csp_addr(vb,rrp);
346 }
347
348 /*
349  * _domain      (pap,csp,cs)
350  * _dom_raw     (pa)
351  */
352
353 static adns_status pap_domain(const parseinfo *pai, int *cbyte_io, int max,
354                               char **domain_r, parsedomain_flags flags) {
355   adns_status st;
356   char *dm;
357   
358   st= adns__parse_domain(pai->qu->ads, pai->serv, pai->qu, &pai->qu->vb, flags,
359                          pai->dgram,pai->dglen, cbyte_io, max);
360   if (st) return st;
361   if (!pai->qu->vb.used) return adns_s_invaliddata;
362
363   dm= adns__alloc_interim(pai->qu, pai->qu->vb.used+1);
364   if (!dm) R_NOMEM;
365
366   dm[pai->qu->vb.used]= 0;
367   memcpy(dm,pai->qu->vb.buf,pai->qu->vb.used);
368   
369   *domain_r= dm;
370   return adns_s_ok;
371 }
372
373 static adns_status csp_domain(vbuf *vb, const char *domain) {
374   CSP_ADDSTR(domain);
375   if (!*domain) CSP_ADDSTR(".");
376   return adns_s_ok;
377 }
378
379 static adns_status cs_domain(vbuf *vb, const void *datap) {
380   const char *const *domainp= datap;
381   return csp_domain(vb,*domainp);
382 }
383
384 static adns_status pa_dom_raw(const parseinfo *pai, int cbyte,
385                               int max, void *datap) {
386   char **rrp= datap;
387   adns_status st;
388
389   st= pap_domain(pai, &cbyte, max, rrp, pdf_quoteok);
390   if (st) return st;
391   
392   if (cbyte != max) return adns_s_invaliddata;
393   return adns_s_ok;
394 }
395
396 /*
397  * _host_raw   (pa)
398  */
399
400 static adns_status pa_host_raw(const parseinfo *pai, int cbyte,
401                                int max, void *datap) {
402   char **rrp= datap;
403   adns_status st;
404
405   st= pap_domain(pai, &cbyte, max, rrp,
406                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
407   if (st) return st;
408   
409   if (cbyte != max) return adns_s_invaliddata;
410   return adns_s_ok;
411 }
412
413 /*
414  * _hostaddr   (pap,pa,dip,di,mfp,mf,csp,cs +pap_findaddrs, icb_hostaddr)
415  */
416
417 static adns_status pap_findaddrs(const parseinfo *pai, adns_rr_hostaddr *ha,
418                                  int *cbyte_io, int count, int dmstart) {
419   int rri, naddrs;
420   int type, class, rdlen, rdstart, ownermatched;
421   unsigned long ttl;
422   adns_status st;
423   
424   for (rri=0, naddrs=-1; rri<count; rri++) {
425     st= adns__findrr_anychk(pai->qu, pai->serv, pai->dgram,
426                             pai->dglen, cbyte_io,
427                             &type, &class, &ttl, &rdlen, &rdstart,
428                             pai->dgram, pai->dglen, dmstart, &ownermatched);
429     if (st) return st;
430     if (!ownermatched || class != DNS_CLASS_IN || type != adns_r_a) {
431       if (naddrs>0) break; else continue;
432     }
433     if (naddrs == -1) {
434       naddrs= 0;
435     }
436     if (!adns__vbuf_ensure(&pai->qu->vb, (naddrs+1)*sizeof(adns_rr_addr)))
437       R_NOMEM;
438     adns__update_expires(pai->qu,ttl,pai->now);
439     st= pa_addr(pai, rdstart,rdstart+rdlen,
440                 pai->qu->vb.buf + naddrs*sizeof(adns_rr_addr));
441     if (st) return st;
442     naddrs++;
443   }
444   if (naddrs >= 0) {
445     ha->addrs= adns__alloc_interim(pai->qu, naddrs*sizeof(adns_rr_addr));
446     if (!ha->addrs) R_NOMEM;
447     memcpy(ha->addrs, pai->qu->vb.buf, naddrs*sizeof(adns_rr_addr));
448     ha->naddrs= naddrs;
449     ha->astatus= adns_s_ok;
450
451     adns__isort(ha->addrs, naddrs, sizeof(adns_rr_addr), pai->qu->vb.buf,
452                 div_addr, pai->ads);
453   }
454   return adns_s_ok;
455 }
456
457 static void icb_hostaddr(adns_query parent, adns_query child) {
458   adns_answer *cans= child->answer;
459   adns_rr_hostaddr *rrp= child->ctx.pinfo.hostaddr;
460   adns_state ads= parent->ads;
461   adns_status st;
462
463   st= cans->status;
464   rrp->astatus= st;
465   rrp->naddrs= (st>0 && st<=adns_s_max_tempfail) ? -1 : cans->nrrs;
466   rrp->addrs= cans->rrs.addr;
467   adns__transfer_interim(child, parent, rrp->addrs,
468                          rrp->naddrs*sizeof(adns_rr_addr));
469
470   if (parent->children.head) {
471     LIST_LINK_TAIL(ads->childw,parent);
472   } else {
473     adns__query_done(parent);
474   }
475 }
476
477 static adns_status pap_hostaddr(const parseinfo *pai, int *cbyte_io,
478                                 int max, adns_rr_hostaddr *rrp) {
479   adns_status st;
480   int dmstart, cbyte;
481   qcontext ctx;
482   int id;
483   adns_query nqu;
484   adns_queryflags nflags;
485
486   dmstart= cbyte= *cbyte_io;
487   st= pap_domain(pai, &cbyte, max, &rrp->host,
488                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
489   if (st) return st;
490   *cbyte_io= cbyte;
491
492   rrp->astatus= adns_s_ok;
493   rrp->naddrs= -1;
494   rrp->addrs= 0;
495
496   cbyte= pai->nsstart;
497
498   st= pap_findaddrs(pai, rrp, &cbyte, pai->nscount, dmstart);
499   if (st) return st;
500   if (rrp->naddrs != -1) return adns_s_ok;
501
502   st= pap_findaddrs(pai, rrp, &cbyte, pai->arcount, dmstart);
503   if (st) return st;
504   if (rrp->naddrs != -1) return adns_s_ok;
505
506   st= adns__mkquery_frdgram(pai->ads, &pai->qu->vb, &id,
507                             pai->dgram, pai->dglen, dmstart,
508                             adns_r_addr, adns_qf_quoteok_query);
509   if (st) return st;
510
511   ctx.ext= 0;
512   ctx.callback= icb_hostaddr;
513   ctx.pinfo.hostaddr= rrp;
514   memset(&ctx.tinfo, 0, sizeof(ctx.tinfo));
515   
516   nflags= adns_qf_quoteok_query;
517   if (!(pai->qu->flags & adns_qf_cname_loose)) nflags |= adns_qf_cname_forbid;
518   
519   st= adns__internal_submit(pai->ads, &nqu, adns__findtype(adns_r_addr),
520                             ((adns_r_addr & adns_rrt_reprmask) |
521                              (pai->qu->answer->type & ~adns_rrt_reprmask)),
522                             &pai->qu->vb, id, nflags, pai->now, &ctx);
523   if (st) return st;
524
525   nqu->parent= pai->qu;
526   LIST_LINK_TAIL_PART(pai->qu->children,nqu,siblings.);
527
528   return adns_s_ok;
529 }
530
531 static adns_status pa_hostaddr(const parseinfo *pai, int cbyte,
532                                int max, void *datap) {
533   adns_rr_hostaddr *rrp= datap;
534   adns_status st;
535
536   st= pap_hostaddr(pai, &cbyte, max, rrp);
537   if (st) return st;
538   if (cbyte != max) return adns_s_invaliddata;
539
540   return adns_s_ok;
541 }
542
543 static int dip_hostaddr(adns_state ads,
544                         const adns_rr_hostaddr *ap, const adns_rr_hostaddr *bp) {
545   if (ap->astatus != bp->astatus) return ap->astatus;
546   if (ap->astatus) return 0;
547
548   assert(ap->addrs[0].addr.sa.sa_family == AF_INET);
549   assert(bp->addrs[0].addr.sa.sa_family == AF_INET);
550   return dip_inaddr(ads,
551                     ap->addrs[0].addr.inet.sin_addr,
552                     bp->addrs[0].addr.inet.sin_addr);
553 }
554
555 static int di_hostaddr(adns_state ads,
556                        const void *datap_a, const void *datap_b) {
557   const adns_rr_hostaddr *ap= datap_a, *bp= datap_b;
558
559   return dip_hostaddr(ads, ap,bp);
560 }
561
562 static void mfp_hostaddr(adns_query qu, adns_rr_hostaddr *rrp) {
563   void *tablev;
564
565   adns__makefinal_str(qu,&rrp->host);
566   tablev= rrp->addrs;
567   adns__makefinal_block(qu, &tablev, rrp->naddrs*sizeof(*rrp->addrs));
568   rrp->addrs= tablev;
569 }
570
571 static void mf_hostaddr(adns_query qu, void *datap) {
572   adns_rr_hostaddr *rrp= datap;
573
574   mfp_hostaddr(qu,rrp);
575 }
576
577 static adns_status csp_hostaddr(vbuf *vb, const adns_rr_hostaddr *rrp) {
578   const char *errstr;
579   adns_status st;
580   char buf[20];
581   int i;
582
583   st= csp_domain(vb,rrp->host);  if (st) return st;
584
585   CSP_ADDSTR(" ");
586   CSP_ADDSTR(adns_errtypeabbrev(rrp->astatus));
587
588   sprintf(buf," %d ",rrp->astatus);
589   CSP_ADDSTR(buf);
590
591   CSP_ADDSTR(adns_errabbrev(rrp->astatus));
592   CSP_ADDSTR(" ");
593
594   errstr= adns_strerror(rrp->astatus);
595   st= csp_qstring(vb,errstr,strlen(errstr));  if (st) return st;
596   
597   if (rrp->naddrs >= 0) {
598     CSP_ADDSTR(" (");
599     for (i=0; i<rrp->naddrs; i++) {
600       CSP_ADDSTR(" ");
601       st= csp_addr(vb,&rrp->addrs[i]);
602     }
603     CSP_ADDSTR(" )");
604   } else {
605     CSP_ADDSTR(" ?");
606   }
607   return adns_s_ok;
608 }
609
610 static adns_status cs_hostaddr(vbuf *vb, const void *datap) {
611   const adns_rr_hostaddr *rrp= datap;
612
613   return csp_hostaddr(vb,rrp);
614 }
615
616 /*
617  * _mx_raw   (pa,di)
618  */
619
620 static adns_status pa_mx_raw(const parseinfo *pai, int cbyte,
621                              int max, void *datap) {
622   const byte *dgram= pai->dgram;
623   adns_rr_intstr *rrp= datap;
624   adns_status st;
625   int pref;
626
627   if (cbyte+2 > max) return adns_s_invaliddata;
628   GET_W(cbyte,pref);
629   rrp->i= pref;
630   st= pap_domain(pai, &cbyte, max, &rrp->str,
631                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
632   if (st) return st;
633   
634   if (cbyte != max) return adns_s_invaliddata;
635   return adns_s_ok;
636 }
637
638 static int di_mx_raw(adns_state ads, const void *datap_a, const void *datap_b) {
639   const adns_rr_intstr *ap= datap_a, *bp= datap_b;
640
641   if (ap->i < bp->i) return 0;
642   if (ap->i > bp->i) return 1;
643   return 0;
644 }
645
646 /*
647  * _mx   (pa,di)
648  */
649
650 static adns_status pa_mx(const parseinfo *pai, int cbyte,
651                          int max, void *datap) {
652   const byte *dgram= pai->dgram;
653   adns_rr_inthostaddr *rrp= datap;
654   adns_status st;
655   int pref;
656
657   if (cbyte+2 > max) return adns_s_invaliddata;
658   GET_W(cbyte,pref);
659   rrp->i= pref;
660   st= pap_hostaddr(pai, &cbyte, max, &rrp->ha);
661   if (st) return st;
662   
663   if (cbyte != max) return adns_s_invaliddata;
664   return adns_s_ok;
665 }
666
667 static int di_mx(adns_state ads, const void *datap_a, const void *datap_b) {
668   const adns_rr_inthostaddr *ap= datap_a, *bp= datap_b;
669
670   if (ap->i < bp->i) return 0;
671   if (ap->i > bp->i) return 1;
672   return dip_hostaddr(ads, &ap->ha, &bp->ha);
673 }
674
675 /*
676  * _inthostaddr  (mf,cs)
677  */
678
679 static void mf_inthostaddr(adns_query qu, void *datap) {
680   adns_rr_inthostaddr *rrp= datap;
681
682   mfp_hostaddr(qu,&rrp->ha);
683 }
684
685 static adns_status cs_inthostaddr(vbuf *vb, const void *datap) {
686   const adns_rr_inthostaddr *rrp= datap;
687   char buf[10];
688
689   sprintf(buf,"%u ",rrp->i);
690   CSP_ADDSTR(buf);
691
692   return csp_hostaddr(vb,&rrp->ha);
693 }
694
695 /*
696  * _inthost  (cs)
697  */
698
699 static adns_status cs_inthost(vbuf *vb, const void *datap) {
700   const adns_rr_intstr *rrp= datap;
701   char buf[10];
702
703   sprintf(buf,"%u ",rrp->i);
704   CSP_ADDSTR(buf);
705   return csp_domain(vb,rrp->str);
706 }
707
708 /*
709  * _ptr   (ckl,pa +icb_ptr)
710  */
711
712 static adns_status ckl_ptr(adns_state ads, adns_queryflags flags,
713                            union checklabel_state *cls, qcontext *ctx,
714                            int labnum, const char *label, int lablen) {
715   static const char *const (expectdomain[])= { DNS_INADDR_ARPA };
716   char *ep;
717   const char *ed;
718   char labbuf[4];
719   int l;
720
721   if (labnum < 4) {
722     if (lablen<=0 || lablen>3) return adns_s_querydomainwrong;
723     memcpy(labbuf, label, lablen);
724     labbuf[lablen]= 0;
725     cls->ptr.ipv[3-labnum]= strtoul(labbuf,&ep,10);
726     if (*ep) return adns_s_querydomainwrong;
727     if (lablen>1 && *label=='0') return adns_s_querydomainwrong;
728   } else if (labnum < 4 + sizeof(expectdomain)/sizeof(*expectdomain)) {
729     ed= expectdomain[labnum-4];
730     l= strlen(ed);
731     if (lablen != l || memcmp(label, ed, l)) return adns_s_querydomainwrong;
732   } else {
733     if (lablen) return adns_s_querydomainwrong;
734     ctx->tinfo.ptr.addr.s_addr=
735       htonl((cls->ptr.ipv[0]<<24) | (cls->ptr.ipv[1]<<16) |
736             (cls->ptr.ipv[2]<< 8) | (cls->ptr.ipv[3]));
737   }
738   return adns_s_ok;
739 }
740
741 static void icb_ptr(adns_query parent, adns_query child) {
742   adns_answer *cans= child->answer;
743   const struct in_addr *queried;
744   const unsigned char *found;
745   adns_state ads= parent->ads;
746   int i;
747
748   if (cans->status == adns_s_nxdomain || cans->status == adns_s_nodata) {
749     adns__query_fail(parent,adns_s_inconsistent);
750     return;
751   } else if (cans->status) {
752     adns__query_fail(parent,cans->status);
753     return;
754   }
755
756   queried= &parent->ctx.tinfo.ptr.addr;
757   for (i=0, found=cans->rrs.bytes; i<cans->nrrs; i++, found+=cans->rrsz) {
758     if (!memcmp(queried,found,cans->rrsz)) {
759       if (!parent->children.head) {
760         adns__query_done(parent);
761         return;
762       } else {
763         LIST_LINK_TAIL(ads->childw,parent);
764         return;
765       }
766     }
767   }
768
769   adns__query_fail(parent,adns_s_inconsistent);
770 }
771
772 static adns_status pa_ptr(const parseinfo *pai, int dmstart,
773                           int max, void *datap) {
774   char **rrp= datap;
775   adns_status st;
776   int cbyte, id;
777   adns_query nqu;
778   qcontext ctx;
779
780   cbyte= dmstart;
781   st= pap_domain(pai, &cbyte, max, rrp,
782                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
783   if (st) return st;
784   if (cbyte != max) return adns_s_invaliddata;
785
786   st= adns__mkquery_frdgram(pai->ads, &pai->qu->vb, &id,
787                             pai->dgram, pai->dglen, dmstart,
788                             adns_r_a, adns_qf_quoteok_query);
789   if (st) return st;
790
791   ctx.ext= 0;
792   ctx.callback= icb_ptr;
793   memset(&ctx.pinfo,0,sizeof(ctx.pinfo));
794   memset(&ctx.tinfo,0,sizeof(ctx.tinfo));
795   st= adns__internal_submit(pai->ads, &nqu, adns__findtype(adns_r_a),
796                             adns_r_a, &pai->qu->vb, id, adns_qf_quoteok_query,
797                             pai->now, &ctx);
798   if (st) return st;
799
800   nqu->parent= pai->qu;
801   LIST_LINK_TAIL_PART(pai->qu->children,nqu,siblings.);
802   return adns_s_ok;
803 }
804
805 /*
806  * _strpair   (mf)
807  */
808
809 static void mf_strpair(adns_query qu, void *datap) {
810   adns_rr_strpair *rrp= datap;
811
812   adns__makefinal_str(qu,&rrp->array[0]);
813   adns__makefinal_str(qu,&rrp->array[1]);
814 }
815
816 /*
817  * _intstrpair   (mf)
818  */
819
820 static void mf_intstrpair(adns_query qu, void *datap) {
821   adns_rr_intstrpair *rrp= datap;
822
823   adns__makefinal_str(qu,&rrp->array[0].str);
824   adns__makefinal_str(qu,&rrp->array[1].str);
825 }
826
827 /*
828  * _hinfo   (pa)
829  */
830
831 static adns_status pa_hinfo(const parseinfo *pai, int cbyte,
832                             int max, void *datap) {
833   adns_rr_intstrpair *rrp= datap;
834   adns_status st;
835   int i;
836
837   for (i=0; i<2; i++) {
838     st= pap_qstring(pai, &cbyte, max, &rrp->array[i].i, &rrp->array[i].str);
839     if (st) return st;
840   }
841
842   if (cbyte != max) return adns_s_invaliddata;
843   
844   return adns_s_ok;
845 }
846
847 /*
848  * _mailbox   (pap,cs +pap_mailbox822)
849  */
850
851 static adns_status pap_mailbox822(const parseinfo *pai,
852                                   int *cbyte_io, int max, char **mb_r) {
853   int lablen, labstart, i, needquote, c, r, neednorm;
854   const unsigned char *p;
855   char *str;
856   findlabel_state fls;
857   adns_status st;
858   vbuf *vb;
859
860   vb= &pai->qu->vb;
861   vb->used= 0;
862   adns__findlabel_start(&fls, pai->ads,
863                         -1, pai->qu,
864                         pai->dgram, pai->dglen, max,
865                         *cbyte_io, cbyte_io);
866   st= adns__findlabel_next(&fls,&lablen,&labstart);
867   if (!lablen) {
868     adns__vbuf_appendstr(vb,".");
869     goto x_ok;
870   }
871
872   neednorm= 1;
873   for (i=0, needquote=0, p= pai->dgram+labstart; i<lablen; i++) {
874     c= *p++;
875     if ((c&~128) < 32 || (c&~128) == 127) return adns_s_invaliddata;
876     if (c == '.' && !neednorm) neednorm= 1;
877     else if (c==' ' || c>=127 || ctype_822special(c)) needquote++;
878     else neednorm= 0;
879   }
880
881   if (needquote || neednorm) {
882     r= adns__vbuf_ensure(vb, lablen+needquote+4); if (!r) R_NOMEM;
883     adns__vbuf_appendq(vb,"\"",1);
884     for (i=0, needquote=0, p= pai->dgram+labstart; i<lablen; i++, p++) {
885       c= *p;
886       if (c == '"' || c=='\\') adns__vbuf_appendq(vb,"\\",1);
887       adns__vbuf_appendq(vb,p,1);
888     }
889     adns__vbuf_appendq(vb,"\"",1);
890   } else {
891     r= adns__vbuf_append(vb, pai->dgram+labstart, lablen); if (!r) R_NOMEM;
892   }
893
894   r= adns__vbuf_appendstr(vb,"@"); if (!r) R_NOMEM;
895
896   st= adns__parse_domain_more(&fls,pai->ads, pai->qu,vb,0, pai->dgram);
897   if (st) return st;
898
899  x_ok:
900   str= adns__alloc_interim(pai->qu, vb->used+1); if (!str) R_NOMEM;
901   memcpy(str,vb->buf,vb->used);
902   str[vb->used]= 0;
903   *mb_r= str;
904   return adns_s_ok;
905 }
906
907 static adns_status pap_mailbox(const parseinfo *pai, int *cbyte_io, int max,
908                                char **mb_r) {
909   if (pai->qu->typei->typekey & adns__qtf_mail822) {
910     return pap_mailbox822(pai, cbyte_io, max, mb_r);
911   } else {
912     return pap_domain(pai, cbyte_io, max, mb_r, pdf_quoteok);
913   }
914 }
915
916 static adns_status csp_mailbox(vbuf *vb, const char *mailbox) {
917   return csp_domain(vb,mailbox);
918 }
919
920 /*
921  * _rp   (pa,cs)
922  */
923
924 static adns_status pa_rp(const parseinfo *pai, int cbyte,
925                          int max, void *datap) {
926   adns_rr_strpair *rrp= datap;
927   adns_status st;
928
929   st= pap_mailbox(pai, &cbyte, max, &rrp->array[0]);
930   if (st) return st;
931
932   st= pap_domain(pai, &cbyte, max, &rrp->array[1], pdf_quoteok);
933   if (st) return st;
934
935   if (cbyte != max) return adns_s_invaliddata;
936   return adns_s_ok;
937 }
938
939 static adns_status cs_rp(vbuf *vb, const void *datap) {
940   const adns_rr_strpair *rrp= datap;
941   adns_status st;
942
943   st= csp_mailbox(vb,rrp->array[0]);  if (st) return st;
944   CSP_ADDSTR(" ");
945   st= csp_domain(vb,rrp->array[1]);  if (st) return st;
946
947   return adns_s_ok;
948 }  
949
950 /*
951  * _soa   (pa,mf,cs)
952  */
953
954 static adns_status pa_soa(const parseinfo *pai, int cbyte,
955                           int max, void *datap) {
956   adns_rr_soa *rrp= datap;
957   const byte *dgram= pai->dgram;
958   adns_status st;
959   int msw, lsw, i;
960
961   st= pap_domain(pai, &cbyte, max, &rrp->mname,
962                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
963   if (st) return st;
964
965   st= pap_mailbox(pai, &cbyte, max, &rrp->rname);
966   if (st) return st;
967
968   if (cbyte+20 != max) return adns_s_invaliddata;
969   
970   for (i=0; i<5; i++) {
971     GET_W(cbyte,msw);
972     GET_W(cbyte,lsw);
973     (&rrp->serial)[i]= (msw<<16) | lsw;
974   }
975
976   return adns_s_ok;
977 }
978
979 static void mf_soa(adns_query qu, void *datap) {
980   adns_rr_soa *rrp= datap;
981
982   adns__makefinal_str(qu,&rrp->mname);
983   adns__makefinal_str(qu,&rrp->rname);
984 }
985
986 static adns_status cs_soa(vbuf *vb, const void *datap) {
987   const adns_rr_soa *rrp= datap;
988   char buf[20];
989   int i;
990   adns_status st;
991   
992   st= csp_domain(vb,rrp->mname);  if (st) return st;
993   CSP_ADDSTR(" ");
994   st= csp_mailbox(vb,rrp->rname);  if (st) return st;
995
996   for (i=0; i<5; i++) {
997     sprintf(buf," %lu",(&rrp->serial)[i]);
998     CSP_ADDSTR(buf);
999   }
1000
1001   return adns_s_ok;
1002 }
1003
1004 /*
1005  * _srv*  (ckl,(pap),pa*2,mf*2,di,(csp),cs*2,postsort)
1006  */
1007
1008 static adns_status ckl_srv(adns_state ads, adns_queryflags flags,
1009                            union checklabel_state *cls, qcontext *ctx,
1010                            int labnum, const char *label, int lablen) {
1011   if (labnum < 2 && !(flags & adns_qf_quoteok_query)) {
1012     if (!lablen || label[0] != '_') return adns_s_querydomaininvalid;
1013     return adns_s_ok;
1014   }
1015   return adns__ckl_hostname(ads, flags, cls, ctx, labnum, label, lablen);
1016 }
1017
1018 static adns_status pap_srv_begin(const parseinfo *pai, int *cbyte_io, int max,
1019                                  adns_rr_srvha *rrp
1020                                    /* might be adns_rr_srvraw* */) {
1021   const byte *dgram= pai->dgram;
1022   int ti, cbyte;
1023
1024   cbyte= *cbyte_io;
1025   if ((*cbyte_io += 6) > max) return adns_s_invaliddata;
1026   
1027   rrp->priority= GET_W(cbyte, ti);
1028   rrp->weight=   GET_W(cbyte, ti);
1029   rrp->port=     GET_W(cbyte, ti);
1030   return adns_s_ok;
1031 }
1032
1033 static adns_status pa_srvraw(const parseinfo *pai, int cbyte,
1034                              int max, void *datap) {
1035   adns_rr_srvraw *rrp= datap;
1036   adns_status st;
1037
1038   st= pap_srv_begin(pai,&cbyte,max,datap);
1039   if (st) return st;
1040   
1041   st= pap_domain(pai, &cbyte, max, &rrp->host,
1042                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
1043   if (st) return st;
1044   
1045   if (cbyte != max) return adns_s_invaliddata;
1046   return adns_s_ok;
1047 }
1048
1049 static adns_status pa_srvha(const parseinfo *pai, int cbyte,
1050                             int max, void *datap) {
1051   adns_rr_srvha *rrp= datap;
1052   adns_status st;
1053
1054   st= pap_srv_begin(pai,&cbyte,max,datap);       if (st) return st;
1055   st= pap_hostaddr(pai, &cbyte, max, &rrp->ha);  if (st) return st;
1056   if (cbyte != max) return adns_s_invaliddata;
1057   return adns_s_ok;
1058 }
1059
1060 static void mf_srvraw(adns_query qu, void *datap) {
1061   adns_rr_srvraw *rrp= datap;
1062   adns__makefinal_str(qu, &rrp->host);
1063 }
1064
1065 static void mf_srvha(adns_query qu, void *datap) {
1066   adns_rr_srvha *rrp= datap;
1067   mfp_hostaddr(qu,&rrp->ha);
1068 }
1069
1070 static int di_srv(adns_state ads, const void *datap_a, const void *datap_b) {
1071   const adns_rr_srvraw *ap= datap_a, *bp= datap_b;
1072     /* might be const adns_rr_svhostaddr* */
1073
1074   if (ap->priority < bp->priority) return 0;
1075   if (ap->priority > bp->priority) return 1;
1076   return 0;
1077 }
1078
1079 static adns_status csp_srv_begin(vbuf *vb, const adns_rr_srvha *rrp
1080                                    /* might be adns_rr_srvraw* */) {
1081   char buf[30];
1082   sprintf(buf,"%u %u %u ", rrp->priority, rrp->weight, rrp->port);
1083   CSP_ADDSTR(buf);
1084   return adns_s_ok;
1085 }
1086
1087 static adns_status cs_srvraw(vbuf *vb, const void *datap) {
1088   const adns_rr_srvraw *rrp= datap;
1089   adns_status st;
1090   
1091   st= csp_srv_begin(vb,(const void*)rrp);  if (st) return st;
1092   return csp_domain(vb,rrp->host);
1093 }
1094
1095 static adns_status cs_srvha(vbuf *vb, const void *datap) {
1096   const adns_rr_srvha *rrp= datap;
1097   adns_status st;
1098
1099   st= csp_srv_begin(vb,(const void*)datap);  if (st) return st;
1100   return csp_hostaddr(vb,&rrp->ha);
1101 }
1102
1103 static void postsort_srv(adns_state ads, void *array, int nrrs,
1104                          const struct typeinfo *typei) {
1105   /* we treat everything in the array as if it were an adns_rr_srvha
1106    * even though the array might be of adns_rr_srvraw.  That's OK
1107    * because they have the same prefix, which is all we access.
1108    * We use typei->rrsz, too, rather than naive array indexing, of course.
1109    */
1110   char *workbegin, *workend, *search, *arrayend;
1111   const adns_rr_srvha *rr;
1112   union { adns_rr_srvha ha; adns_rr_srvraw raw; } rrtmp;
1113   int cpriority, totalweight, runtotal;
1114   long randval;
1115
1116   for (workbegin= array, arrayend= workbegin + typei->rrsz * nrrs;
1117        workbegin < arrayend;
1118        workbegin= workend) {
1119     cpriority= (rr=(void*)workbegin)->priority;
1120     
1121     for (workend= workbegin, totalweight= 0;
1122          workend < arrayend && (rr=(void*)workend)->priority == cpriority;
1123          workend += typei->rrsz) {
1124       totalweight += rr->weight;
1125     }
1126
1127     /* Now workbegin..(workend-1) incl. are exactly all of the RRs of
1128      * cpriority.  From now on, workbegin points to the `remaining'
1129      * records: we select one record at a time (RFC2782 `Usage rules'
1130      * and `Format of the SRV RR' subsection `Weight') to place at
1131      * workbegin (swapping with the one that was there, and then
1132      * advance workbegin. */
1133     for (;
1134          workbegin + typei->rrsz < workend; /* don't bother if just one */
1135          workbegin += typei->rrsz) {
1136       
1137       randval= nrand48(ads->rand48xsubi);
1138       randval %= (totalweight + 1);
1139         /* makes it into 0..totalweight inclusive; with 2^10 RRs,
1140          * totalweight must be <= 2^26 so probability nonuniformity is
1141          * no worse than 1 in 2^(31-26) ie 1 in 2^5, ie
1142          *  abs(log(P_intended(RR_i) / P_actual(RR_i)) <= log(2^-5).
1143          */
1144
1145       for (search=workbegin, runtotal=0;
1146            (runtotal += (rr=(void*)search)->weight) < randval;
1147            search += typei->rrsz);
1148       assert(search < arrayend);
1149       totalweight -= rr->weight;
1150       if (search != workbegin) {
1151         memcpy(&rrtmp, workbegin, typei->rrsz);
1152         memcpy(workbegin, search, typei->rrsz);
1153         memcpy(search, &rrtmp, typei->rrsz);
1154       }
1155     }
1156   }
1157   /* tests:
1158    *  dig -t srv _srv._tcp.test.iwj.relativity.greenend.org.uk.
1159    *   ./adnshost_s -t srv- _sip._udp.voip.net.cam.ac.uk.
1160    *   ./adnshost_s -t srv- _jabber._tcp.jabber.org
1161    */
1162 }
1163
1164 /*
1165  * _byteblock   (mf)
1166  */
1167
1168 static void mf_byteblock(adns_query qu, void *datap) {
1169   adns_rr_byteblock *rrp= datap;
1170   void *bytes= rrp->data;
1171   adns__makefinal_block(qu,&bytes,rrp->len);
1172   rrp->data= bytes;
1173 }
1174
1175 /*
1176  * _opaque   (pa,cs)
1177  */
1178
1179 static adns_status pa_opaque(const parseinfo *pai, int cbyte,
1180                              int max, void *datap) {
1181   adns_rr_byteblock *rrp= datap;
1182
1183   rrp->len= max - cbyte;
1184   rrp->data= adns__alloc_interim(pai->qu, rrp->len);
1185   if (!rrp->data) R_NOMEM;
1186   memcpy(rrp->data, pai->dgram + cbyte, rrp->len);
1187   return adns_s_ok;
1188 }
1189
1190 static adns_status cs_opaque(vbuf *vb, const void *datap) {
1191   const adns_rr_byteblock *rrp= datap;
1192   char buf[10];
1193   int l;
1194   unsigned char *p;
1195
1196   sprintf(buf,"\\# %d",rrp->len);
1197   CSP_ADDSTR(buf);
1198   
1199   for (l= rrp->len, p= rrp->data;
1200        l>=4;
1201        l -= 4, p += 4) {
1202     sprintf(buf," %02x%02x%02x%02x",p[0],p[1],p[2],p[3]);
1203     CSP_ADDSTR(buf);
1204   }
1205   for (;
1206        l>0;
1207        l--, p++) {
1208     sprintf(buf," %02x",*p);
1209     CSP_ADDSTR(buf);
1210   }
1211   return adns_s_ok;
1212 }
1213   
1214 /*
1215  * _flat   (mf)
1216  */
1217
1218 static void mf_flat(adns_query qu, void *data) { }
1219
1220 /*
1221  * Now the table.
1222  */
1223
1224 #define TYPESZ_M(member)           (sizeof(*((adns_answer*)0)->rrs.member))
1225
1226 #define DEEP_TYPE(code,rrt,fmt,memb,parser,comparer,/*printer*/...)     \
1227  { adns_r_##code&adns_rrt_reprmask, rrt,fmt,TYPESZ_M(memb), mf_##memb,  \
1228      GLUE(cs_, CAR(__VA_ARGS__)),pa_##parser,di_##comparer,             \
1229      adns__ckl_hostname, CDR(__VA_ARGS__) }
1230 #define FLAT_TYPE(code,rrt,fmt,memb,parser,comparer,/*printer*/...)     \
1231  { adns_r_##code&adns_rrt_reprmask, rrt,fmt,TYPESZ_M(memb), mf_flat,    \
1232      GLUE(cs_, CAR(__VA_ARGS__)),pa_##parser,di_##comparer,             \
1233      adns__ckl_hostname, CDR(__VA_ARGS__) }
1234
1235 #define di_0 0
1236
1237 static const typeinfo typeinfos[] = {
1238 /* Must be in ascending order of rrtype ! */
1239 /* mem-mgmt code  rrt     fmt   member   parser      comparer  printer */
1240
1241 FLAT_TYPE(a,      "A",     0,   inaddr,    inaddr,  inaddr,inaddr          ),
1242 DEEP_TYPE(ns_raw, "NS",   "raw",str,       host_raw,0,     domain          ),
1243 DEEP_TYPE(cname,  "CNAME", 0,   str,       dom_raw, 0,     domain          ),
1244 DEEP_TYPE(soa_raw,"SOA",  "raw",soa,       soa,     0,     soa             ),
1245 DEEP_TYPE(ptr_raw,"PTR",  "raw",str,       host_raw,0,     domain          ),
1246 DEEP_TYPE(hinfo,  "HINFO", 0,   intstrpair,hinfo,   0,     hinfo           ),
1247 DEEP_TYPE(mx_raw, "MX",   "raw",intstr,    mx_raw,  mx_raw,inthost         ),
1248 DEEP_TYPE(txt,    "TXT",   0,   manyistr,  txt,     0,     txt             ),
1249 DEEP_TYPE(rp_raw, "RP",   "raw",strpair,   rp,      0,     rp              ),
1250 DEEP_TYPE(srv_raw,"SRV",  "raw",srvraw ,   srvraw,  srv,   srvraw,
1251                               .checklabel= ckl_srv, .postsort= postsort_srv),
1252
1253 FLAT_TYPE(addr,   "A",  "addr", addr,      addr,    addr,  addr            ),
1254 DEEP_TYPE(ns,     "NS", "+addr",hostaddr,  hostaddr,hostaddr,hostaddr      ),
1255 DEEP_TYPE(ptr,    "PTR","checked",str,     ptr,     0,     domain,
1256                                                        .checklabel= ckl_ptr),
1257 DEEP_TYPE(mx,     "MX", "+addr",inthostaddr,mx,     mx,    inthostaddr,    ),
1258 DEEP_TYPE(srv,    "SRV","+addr",srvha,     srvha,   srv,   srvha,
1259                               .checklabel= ckl_srv, .postsort= postsort_srv),
1260
1261 DEEP_TYPE(soa,    "SOA","822",  soa,       soa,     0,     soa             ),
1262 DEEP_TYPE(rp,     "RP", "822",  strpair,   rp,      0,     rp              ),
1263 };
1264
1265 static const typeinfo typeinfo_unknown=
1266 DEEP_TYPE(unknown,0, "unknown",byteblock,opaque,  0,     opaque            );
1267
1268 const typeinfo *adns__findtype(adns_rrtype type) {
1269   const typeinfo *begin, *end, *mid;
1270
1271   if (type & adns_r_unknown) return &typeinfo_unknown;
1272   type &= adns_rrt_reprmask;
1273
1274   begin= typeinfos;  end= typeinfos+(sizeof(typeinfos)/sizeof(typeinfo));
1275
1276   while (begin < end) {
1277     mid= begin + ((end-begin)>>1);
1278     if (mid->typekey == type) return mid;
1279     if (type > mid->typekey) begin= mid+1;
1280     else end= mid;
1281   }
1282   return 0;
1283 }