chiark / gitweb /
d463b7f9664aa715f045fee67af76d2e829ec8df
[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                             &pai->qu->vb, id, nflags, pai->now, &ctx);
521   if (st) return st;
522
523   nqu->parent= pai->qu;
524   LIST_LINK_TAIL_PART(pai->qu->children,nqu,siblings.);
525
526   return adns_s_ok;
527 }
528
529 static adns_status pa_hostaddr(const parseinfo *pai, int cbyte,
530                                int max, void *datap) {
531   adns_rr_hostaddr *rrp= datap;
532   adns_status st;
533
534   st= pap_hostaddr(pai, &cbyte, max, rrp);
535   if (st) return st;
536   if (cbyte != max) return adns_s_invaliddata;
537
538   return adns_s_ok;
539 }
540
541 static int dip_hostaddr(adns_state ads,
542                         const adns_rr_hostaddr *ap, const adns_rr_hostaddr *bp) {
543   if (ap->astatus != bp->astatus) return ap->astatus;
544   if (ap->astatus) return 0;
545
546   assert(ap->addrs[0].addr.sa.sa_family == AF_INET);
547   assert(bp->addrs[0].addr.sa.sa_family == AF_INET);
548   return dip_inaddr(ads,
549                     ap->addrs[0].addr.inet.sin_addr,
550                     bp->addrs[0].addr.inet.sin_addr);
551 }
552
553 static int di_hostaddr(adns_state ads,
554                        const void *datap_a, const void *datap_b) {
555   const adns_rr_hostaddr *ap= datap_a, *bp= datap_b;
556
557   return dip_hostaddr(ads, ap,bp);
558 }
559
560 static void mfp_hostaddr(adns_query qu, adns_rr_hostaddr *rrp) {
561   void *tablev;
562
563   adns__makefinal_str(qu,&rrp->host);
564   tablev= rrp->addrs;
565   adns__makefinal_block(qu, &tablev, rrp->naddrs*sizeof(*rrp->addrs));
566   rrp->addrs= tablev;
567 }
568
569 static void mf_hostaddr(adns_query qu, void *datap) {
570   adns_rr_hostaddr *rrp= datap;
571
572   mfp_hostaddr(qu,rrp);
573 }
574
575 static adns_status csp_hostaddr(vbuf *vb, const adns_rr_hostaddr *rrp) {
576   const char *errstr;
577   adns_status st;
578   char buf[20];
579   int i;
580
581   st= csp_domain(vb,rrp->host);  if (st) return st;
582
583   CSP_ADDSTR(" ");
584   CSP_ADDSTR(adns_errtypeabbrev(rrp->astatus));
585
586   sprintf(buf," %d ",rrp->astatus);
587   CSP_ADDSTR(buf);
588
589   CSP_ADDSTR(adns_errabbrev(rrp->astatus));
590   CSP_ADDSTR(" ");
591
592   errstr= adns_strerror(rrp->astatus);
593   st= csp_qstring(vb,errstr,strlen(errstr));  if (st) return st;
594   
595   if (rrp->naddrs >= 0) {
596     CSP_ADDSTR(" (");
597     for (i=0; i<rrp->naddrs; i++) {
598       CSP_ADDSTR(" ");
599       st= csp_addr(vb,&rrp->addrs[i]);
600     }
601     CSP_ADDSTR(" )");
602   } else {
603     CSP_ADDSTR(" ?");
604   }
605   return adns_s_ok;
606 }
607
608 static adns_status cs_hostaddr(vbuf *vb, const void *datap) {
609   const adns_rr_hostaddr *rrp= datap;
610
611   return csp_hostaddr(vb,rrp);
612 }
613
614 /*
615  * _mx_raw   (pa,di)
616  */
617
618 static adns_status pa_mx_raw(const parseinfo *pai, int cbyte,
619                              int max, void *datap) {
620   const byte *dgram= pai->dgram;
621   adns_rr_intstr *rrp= datap;
622   adns_status st;
623   int pref;
624
625   if (cbyte+2 > max) return adns_s_invaliddata;
626   GET_W(cbyte,pref);
627   rrp->i= pref;
628   st= pap_domain(pai, &cbyte, max, &rrp->str,
629                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
630   if (st) return st;
631   
632   if (cbyte != max) return adns_s_invaliddata;
633   return adns_s_ok;
634 }
635
636 static int di_mx_raw(adns_state ads, const void *datap_a, const void *datap_b) {
637   const adns_rr_intstr *ap= datap_a, *bp= datap_b;
638
639   if (ap->i < bp->i) return 0;
640   if (ap->i > bp->i) return 1;
641   return 0;
642 }
643
644 /*
645  * _mx   (pa,di)
646  */
647
648 static adns_status pa_mx(const parseinfo *pai, int cbyte,
649                          int max, void *datap) {
650   const byte *dgram= pai->dgram;
651   adns_rr_inthostaddr *rrp= datap;
652   adns_status st;
653   int pref;
654
655   if (cbyte+2 > max) return adns_s_invaliddata;
656   GET_W(cbyte,pref);
657   rrp->i= pref;
658   st= pap_hostaddr(pai, &cbyte, max, &rrp->ha);
659   if (st) return st;
660   
661   if (cbyte != max) return adns_s_invaliddata;
662   return adns_s_ok;
663 }
664
665 static int di_mx(adns_state ads, const void *datap_a, const void *datap_b) {
666   const adns_rr_inthostaddr *ap= datap_a, *bp= datap_b;
667
668   if (ap->i < bp->i) return 0;
669   if (ap->i > bp->i) return 1;
670   return dip_hostaddr(ads, &ap->ha, &bp->ha);
671 }
672
673 /*
674  * _inthostaddr  (mf,cs)
675  */
676
677 static void mf_inthostaddr(adns_query qu, void *datap) {
678   adns_rr_inthostaddr *rrp= datap;
679
680   mfp_hostaddr(qu,&rrp->ha);
681 }
682
683 static adns_status cs_inthostaddr(vbuf *vb, const void *datap) {
684   const adns_rr_inthostaddr *rrp= datap;
685   char buf[10];
686
687   sprintf(buf,"%u ",rrp->i);
688   CSP_ADDSTR(buf);
689
690   return csp_hostaddr(vb,&rrp->ha);
691 }
692
693 /*
694  * _inthost  (cs)
695  */
696
697 static adns_status cs_inthost(vbuf *vb, const void *datap) {
698   const adns_rr_intstr *rrp= datap;
699   char buf[10];
700
701   sprintf(buf,"%u ",rrp->i);
702   CSP_ADDSTR(buf);
703   return csp_domain(vb,rrp->str);
704 }
705
706 /*
707  * _ptr   (ckl,pa +icb_ptr)
708  */
709
710 static adns_status ckl_ptr(adns_state ads, adns_queryflags flags,
711                            union checklabel_state *cls, qcontext *ctx,
712                            int labnum, const char *label, int lablen) {
713   static const char *const (expectdomain[])= { DNS_INADDR_ARPA };
714   adns_rr_addr *ap;
715   char *ep;
716   const char *ed;
717   char labbuf[4];
718   int l;
719
720   if (labnum < 4) {
721     if (lablen<=0 || lablen>3) return adns_s_querydomainwrong;
722     memcpy(labbuf, label, lablen);
723     labbuf[lablen]= 0;
724     cls->ptr.ipv[3-labnum]= strtoul(labbuf,&ep,10);
725     if (*ep) return adns_s_querydomainwrong;
726     if (lablen>1 && *label=='0') return adns_s_querydomainwrong;
727   } else if (labnum < 4 + sizeof(expectdomain)/sizeof(*expectdomain)) {
728     ed= expectdomain[labnum-4];
729     l= strlen(ed);
730     if (lablen != l || memcmp(label, ed, l)) return adns_s_querydomainwrong;
731   } else {
732     if (lablen) return adns_s_querydomainwrong;
733     ap= &ctx->tinfo.ptr_addr;
734     ap->len= sizeof(struct sockaddr_in);
735     memset(&ap->addr,0,sizeof(ap->addr.inet));
736     ap->addr.inet.sin_family= AF_INET;
737     ap->addr.inet.sin_addr.s_addr=
738       htonl((cls->ptr.ipv[0]<<24) | (cls->ptr.ipv[1]<<16) |
739             (cls->ptr.ipv[2]<< 8) | (cls->ptr.ipv[3]));
740   }
741   return adns_s_ok;
742 }
743
744 static void icb_ptr(adns_query parent, adns_query child) {
745   adns_answer *cans= child->answer;
746   const adns_rr_addr *queried, *found;
747   adns_state ads= parent->ads;
748   int i;
749
750   if (cans->status == adns_s_nxdomain || cans->status == adns_s_nodata) {
751     adns__query_fail(parent,adns_s_inconsistent);
752     return;
753   } else if (cans->status) {
754     adns__query_fail(parent,cans->status);
755     return;
756   }
757
758   queried= &parent->ctx.tinfo.ptr_addr;
759   for (i=0, found=cans->rrs.addr; i<cans->nrrs; i++, found++) {
760     if (queried->len == found->len &&
761         !memcmp(&queried->addr,&found->addr,queried->len)) {
762       if (!parent->children.head) {
763         adns__query_done(parent);
764         return;
765       } else {
766         LIST_LINK_TAIL(ads->childw,parent);
767         return;
768       }
769     }
770   }
771
772   adns__query_fail(parent,adns_s_inconsistent);
773 }
774
775 static adns_status pa_ptr(const parseinfo *pai, int dmstart,
776                           int max, void *datap) {
777   char **rrp= datap;
778   adns_status st;
779   int cbyte, id;
780   adns_query nqu;
781   qcontext ctx;
782
783   cbyte= dmstart;
784   st= pap_domain(pai, &cbyte, max, rrp,
785                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
786   if (st) return st;
787   if (cbyte != max) return adns_s_invaliddata;
788
789   st= adns__mkquery_frdgram(pai->ads, &pai->qu->vb, &id,
790                             pai->dgram, pai->dglen, dmstart,
791                             adns_r_addr, adns_qf_quoteok_query);
792   if (st) return st;
793
794   ctx.ext= 0;
795   ctx.callback= icb_ptr;
796   memset(&ctx.pinfo,0,sizeof(ctx.pinfo));
797   memset(&ctx.tinfo,0,sizeof(ctx.tinfo));
798   st= adns__internal_submit(pai->ads, &nqu, adns__findtype(adns_r_addr),
799                             &pai->qu->vb, id,
800                             adns_qf_quoteok_query, pai->now, &ctx);
801   if (st) return st;
802
803   nqu->parent= pai->qu;
804   LIST_LINK_TAIL_PART(pai->qu->children,nqu,siblings.);
805   return adns_s_ok;
806 }
807
808 /*
809  * _strpair   (mf)
810  */
811
812 static void mf_strpair(adns_query qu, void *datap) {
813   adns_rr_strpair *rrp= datap;
814
815   adns__makefinal_str(qu,&rrp->array[0]);
816   adns__makefinal_str(qu,&rrp->array[1]);
817 }
818
819 /*
820  * _intstrpair   (mf)
821  */
822
823 static void mf_intstrpair(adns_query qu, void *datap) {
824   adns_rr_intstrpair *rrp= datap;
825
826   adns__makefinal_str(qu,&rrp->array[0].str);
827   adns__makefinal_str(qu,&rrp->array[1].str);
828 }
829
830 /*
831  * _hinfo   (pa)
832  */
833
834 static adns_status pa_hinfo(const parseinfo *pai, int cbyte,
835                             int max, void *datap) {
836   adns_rr_intstrpair *rrp= datap;
837   adns_status st;
838   int i;
839
840   for (i=0; i<2; i++) {
841     st= pap_qstring(pai, &cbyte, max, &rrp->array[i].i, &rrp->array[i].str);
842     if (st) return st;
843   }
844
845   if (cbyte != max) return adns_s_invaliddata;
846   
847   return adns_s_ok;
848 }
849
850 /*
851  * _mailbox   (pap,cs +pap_mailbox822)
852  */
853
854 static adns_status pap_mailbox822(const parseinfo *pai,
855                                   int *cbyte_io, int max, char **mb_r) {
856   int lablen, labstart, i, needquote, c, r, neednorm;
857   const unsigned char *p;
858   char *str;
859   findlabel_state fls;
860   adns_status st;
861   vbuf *vb;
862
863   vb= &pai->qu->vb;
864   vb->used= 0;
865   adns__findlabel_start(&fls, pai->ads,
866                         -1, pai->qu,
867                         pai->dgram, pai->dglen, max,
868                         *cbyte_io, cbyte_io);
869   st= adns__findlabel_next(&fls,&lablen,&labstart);
870   if (!lablen) {
871     adns__vbuf_appendstr(vb,".");
872     goto x_ok;
873   }
874
875   neednorm= 1;
876   for (i=0, needquote=0, p= pai->dgram+labstart; i<lablen; i++) {
877     c= *p++;
878     if ((c&~128) < 32 || (c&~128) == 127) return adns_s_invaliddata;
879     if (c == '.' && !neednorm) neednorm= 1;
880     else if (c==' ' || c>=127 || ctype_822special(c)) needquote++;
881     else neednorm= 0;
882   }
883
884   if (needquote || neednorm) {
885     r= adns__vbuf_ensure(vb, lablen+needquote+4); if (!r) R_NOMEM;
886     adns__vbuf_appendq(vb,"\"",1);
887     for (i=0, needquote=0, p= pai->dgram+labstart; i<lablen; i++, p++) {
888       c= *p;
889       if (c == '"' || c=='\\') adns__vbuf_appendq(vb,"\\",1);
890       adns__vbuf_appendq(vb,p,1);
891     }
892     adns__vbuf_appendq(vb,"\"",1);
893   } else {
894     r= adns__vbuf_append(vb, pai->dgram+labstart, lablen); if (!r) R_NOMEM;
895   }
896
897   r= adns__vbuf_appendstr(vb,"@"); if (!r) R_NOMEM;
898
899   st= adns__parse_domain_more(&fls,pai->ads, pai->qu,vb,0, pai->dgram);
900   if (st) return st;
901
902  x_ok:
903   str= adns__alloc_interim(pai->qu, vb->used+1); if (!str) R_NOMEM;
904   memcpy(str,vb->buf,vb->used);
905   str[vb->used]= 0;
906   *mb_r= str;
907   return adns_s_ok;
908 }
909
910 static adns_status pap_mailbox(const parseinfo *pai, int *cbyte_io, int max,
911                                char **mb_r) {
912   if (pai->qu->typei->typekey & adns__qtf_mail822) {
913     return pap_mailbox822(pai, cbyte_io, max, mb_r);
914   } else {
915     return pap_domain(pai, cbyte_io, max, mb_r, pdf_quoteok);
916   }
917 }
918
919 static adns_status csp_mailbox(vbuf *vb, const char *mailbox) {
920   return csp_domain(vb,mailbox);
921 }
922
923 /*
924  * _rp   (pa,cs)
925  */
926
927 static adns_status pa_rp(const parseinfo *pai, int cbyte,
928                          int max, void *datap) {
929   adns_rr_strpair *rrp= datap;
930   adns_status st;
931
932   st= pap_mailbox(pai, &cbyte, max, &rrp->array[0]);
933   if (st) return st;
934
935   st= pap_domain(pai, &cbyte, max, &rrp->array[1], pdf_quoteok);
936   if (st) return st;
937
938   if (cbyte != max) return adns_s_invaliddata;
939   return adns_s_ok;
940 }
941
942 static adns_status cs_rp(vbuf *vb, const void *datap) {
943   const adns_rr_strpair *rrp= datap;
944   adns_status st;
945
946   st= csp_mailbox(vb,rrp->array[0]);  if (st) return st;
947   CSP_ADDSTR(" ");
948   st= csp_domain(vb,rrp->array[1]);  if (st) return st;
949
950   return adns_s_ok;
951 }  
952
953 /*
954  * _soa   (pa,mf,cs)
955  */
956
957 static adns_status pa_soa(const parseinfo *pai, int cbyte,
958                           int max, void *datap) {
959   adns_rr_soa *rrp= datap;
960   const byte *dgram= pai->dgram;
961   adns_status st;
962   int msw, lsw, i;
963
964   st= pap_domain(pai, &cbyte, max, &rrp->mname,
965                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
966   if (st) return st;
967
968   st= pap_mailbox(pai, &cbyte, max, &rrp->rname);
969   if (st) return st;
970
971   if (cbyte+20 != max) return adns_s_invaliddata;
972   
973   for (i=0; i<5; i++) {
974     GET_W(cbyte,msw);
975     GET_W(cbyte,lsw);
976     (&rrp->serial)[i]= (msw<<16) | lsw;
977   }
978
979   return adns_s_ok;
980 }
981
982 static void mf_soa(adns_query qu, void *datap) {
983   adns_rr_soa *rrp= datap;
984
985   adns__makefinal_str(qu,&rrp->mname);
986   adns__makefinal_str(qu,&rrp->rname);
987 }
988
989 static adns_status cs_soa(vbuf *vb, const void *datap) {
990   const adns_rr_soa *rrp= datap;
991   char buf[20];
992   int i;
993   adns_status st;
994   
995   st= csp_domain(vb,rrp->mname);  if (st) return st;
996   CSP_ADDSTR(" ");
997   st= csp_mailbox(vb,rrp->rname);  if (st) return st;
998
999   for (i=0; i<5; i++) {
1000     sprintf(buf," %lu",(&rrp->serial)[i]);
1001     CSP_ADDSTR(buf);
1002   }
1003
1004   return adns_s_ok;
1005 }
1006
1007 /*
1008  * _srv*  (ckl,(pap),pa*2,mf*2,di,(csp),cs*2,postsort)
1009  */
1010
1011 static adns_status ckl_srv(adns_state ads, adns_queryflags flags,
1012                            union checklabel_state *cls, qcontext *ctx,
1013                            int labnum, const char *label, int lablen) {
1014   if (labnum < 2 && !(flags & adns_qf_quoteok_query)) {
1015     if (!lablen || label[0] != '_') return adns_s_querydomaininvalid;
1016     return adns_s_ok;
1017   }
1018   return adns__ckl_hostname(ads, flags, cls, ctx, labnum, label, lablen);
1019 }
1020
1021 static adns_status pap_srv_begin(const parseinfo *pai, int *cbyte_io, int max,
1022                                  adns_rr_srvha *rrp
1023                                    /* might be adns_rr_srvraw* */) {
1024   const byte *dgram= pai->dgram;
1025   int ti, cbyte;
1026
1027   cbyte= *cbyte_io;
1028   if ((*cbyte_io += 6) > max) return adns_s_invaliddata;
1029   
1030   rrp->priority= GET_W(cbyte, ti);
1031   rrp->weight=   GET_W(cbyte, ti);
1032   rrp->port=     GET_W(cbyte, ti);
1033   return adns_s_ok;
1034 }
1035
1036 static adns_status pa_srvraw(const parseinfo *pai, int cbyte,
1037                              int max, void *datap) {
1038   adns_rr_srvraw *rrp= datap;
1039   adns_status st;
1040
1041   st= pap_srv_begin(pai,&cbyte,max,datap);
1042   if (st) return st;
1043   
1044   st= pap_domain(pai, &cbyte, max, &rrp->host,
1045                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
1046   if (st) return st;
1047   
1048   if (cbyte != max) return adns_s_invaliddata;
1049   return adns_s_ok;
1050 }
1051
1052 static adns_status pa_srvha(const parseinfo *pai, int cbyte,
1053                             int max, void *datap) {
1054   adns_rr_srvha *rrp= datap;
1055   adns_status st;
1056
1057   st= pap_srv_begin(pai,&cbyte,max,datap);       if (st) return st;
1058   st= pap_hostaddr(pai, &cbyte, max, &rrp->ha);  if (st) return st;
1059   if (cbyte != max) return adns_s_invaliddata;
1060   return adns_s_ok;
1061 }
1062
1063 static void mf_srvraw(adns_query qu, void *datap) {
1064   adns_rr_srvraw *rrp= datap;
1065   adns__makefinal_str(qu, &rrp->host);
1066 }
1067
1068 static void mf_srvha(adns_query qu, void *datap) {
1069   adns_rr_srvha *rrp= datap;
1070   mfp_hostaddr(qu,&rrp->ha);
1071 }
1072
1073 static int di_srv(adns_state ads, const void *datap_a, const void *datap_b) {
1074   const adns_rr_srvraw *ap= datap_a, *bp= datap_b;
1075     /* might be const adns_rr_svhostaddr* */
1076
1077   if (ap->priority < bp->priority) return 0;
1078   if (ap->priority > bp->priority) return 1;
1079   return 0;
1080 }
1081
1082 static adns_status csp_srv_begin(vbuf *vb, const adns_rr_srvha *rrp
1083                                    /* might be adns_rr_srvraw* */) {
1084   char buf[30];
1085   sprintf(buf,"%u %u %u ", rrp->priority, rrp->weight, rrp->port);
1086   CSP_ADDSTR(buf);
1087   return adns_s_ok;
1088 }
1089
1090 static adns_status cs_srvraw(vbuf *vb, const void *datap) {
1091   const adns_rr_srvraw *rrp= datap;
1092   adns_status st;
1093   
1094   st= csp_srv_begin(vb,(const void*)rrp);  if (st) return st;
1095   return csp_domain(vb,rrp->host);
1096 }
1097
1098 static adns_status cs_srvha(vbuf *vb, const void *datap) {
1099   const adns_rr_srvha *rrp= datap;
1100   adns_status st;
1101
1102   st= csp_srv_begin(vb,(const void*)datap);  if (st) return st;
1103   return csp_hostaddr(vb,&rrp->ha);
1104 }
1105
1106 static void postsort_srv(adns_state ads, void *array, int nrrs,
1107                          const struct typeinfo *typei) {
1108   /* we treat everything in the array as if it were an adns_rr_srvha
1109    * even though the array might be of adns_rr_srvraw.  That's OK
1110    * because they have the same prefix, which is all we access.
1111    * We use typei->rrsz, too, rather than naive array indexing, of course.
1112    */
1113   char *workbegin, *workend, *search, *arrayend;
1114   const adns_rr_srvha *rr;
1115   union { adns_rr_srvha ha; adns_rr_srvraw raw; } rrtmp;
1116   int cpriority, totalweight, runtotal;
1117   long randval;
1118
1119   for (workbegin= array, arrayend= workbegin + typei->rrsz * nrrs;
1120        workbegin < arrayend;
1121        workbegin= workend) {
1122     cpriority= (rr=(void*)workbegin)->priority;
1123     
1124     for (workend= workbegin, totalweight= 0;
1125          workend < arrayend && (rr=(void*)workend)->priority == cpriority;
1126          workend += typei->rrsz) {
1127       totalweight += rr->weight;
1128     }
1129
1130     /* Now workbegin..(workend-1) incl. are exactly all of the RRs of
1131      * cpriority.  From now on, workbegin points to the `remaining'
1132      * records: we select one record at a time (RFC2782 `Usage rules'
1133      * and `Format of the SRV RR' subsection `Weight') to place at
1134      * workbegin (swapping with the one that was there, and then
1135      * advance workbegin. */
1136     for (;
1137          workbegin + typei->rrsz < workend; /* don't bother if just one */
1138          workbegin += typei->rrsz) {
1139       
1140       randval= nrand48(ads->rand48xsubi);
1141       randval %= (totalweight + 1);
1142         /* makes it into 0..totalweight inclusive; with 2^10 RRs,
1143          * totalweight must be <= 2^26 so probability nonuniformity is
1144          * no worse than 1 in 2^(31-26) ie 1 in 2^5, ie
1145          *  abs(log(P_intended(RR_i) / P_actual(RR_i)) <= log(2^-5).
1146          */
1147
1148       for (search=workbegin, runtotal=0;
1149            (runtotal += (rr=(void*)search)->weight) < randval;
1150            search += typei->rrsz);
1151       assert(search < arrayend);
1152       totalweight -= rr->weight;
1153       if (search != workbegin) {
1154         memcpy(&rrtmp, workbegin, typei->rrsz);
1155         memcpy(workbegin, search, typei->rrsz);
1156         memcpy(search, &rrtmp, typei->rrsz);
1157       }
1158     }
1159   }
1160   /* tests:
1161    *  dig -t srv _srv._tcp.test.iwj.relativity.greenend.org.uk.
1162    *   ./adnshost_s -t srv- _sip._udp.voip.net.cam.ac.uk.
1163    *   ./adnshost_s -t srv- _jabber._tcp.jabber.org
1164    */
1165 }
1166
1167 /*
1168  * _byteblock   (mf)
1169  */
1170
1171 static void mf_byteblock(adns_query qu, void *datap) {
1172   adns_rr_byteblock *rrp= datap;
1173   void *bytes= rrp->data;
1174   adns__makefinal_block(qu,&bytes,rrp->len);
1175   rrp->data= bytes;
1176 }
1177
1178 /*
1179  * _opaque   (pa,cs)
1180  */
1181
1182 static adns_status pa_opaque(const parseinfo *pai, int cbyte,
1183                              int max, void *datap) {
1184   adns_rr_byteblock *rrp= datap;
1185
1186   rrp->len= max - cbyte;
1187   rrp->data= adns__alloc_interim(pai->qu, rrp->len);
1188   if (!rrp->data) R_NOMEM;
1189   memcpy(rrp->data, pai->dgram + cbyte, rrp->len);
1190   return adns_s_ok;
1191 }
1192
1193 static adns_status cs_opaque(vbuf *vb, const void *datap) {
1194   const adns_rr_byteblock *rrp= datap;
1195   char buf[10];
1196   int l;
1197   unsigned char *p;
1198
1199   sprintf(buf,"\\# %d",rrp->len);
1200   CSP_ADDSTR(buf);
1201   
1202   for (l= rrp->len, p= rrp->data;
1203        l>=4;
1204        l -= 4, p += 4) {
1205     sprintf(buf," %02x%02x%02x%02x",p[0],p[1],p[2],p[3]);
1206     CSP_ADDSTR(buf);
1207   }
1208   for (;
1209        l>0;
1210        l--, p++) {
1211     sprintf(buf," %02x",*p);
1212     CSP_ADDSTR(buf);
1213   }
1214   return adns_s_ok;
1215 }
1216   
1217 /*
1218  * _flat   (mf)
1219  */
1220
1221 static void mf_flat(adns_query qu, void *data) { }
1222
1223 /*
1224  * Now the table.
1225  */
1226
1227 #define TYPESZ_M(member)           (sizeof(*((adns_answer*)0)->rrs.member))
1228
1229 #define DEEP_TYPE(code,rrt,fmt,memb,parser,comparer,/*printer*/...)     \
1230  { adns_r_##code, rrt,fmt,TYPESZ_M(memb), mf_##memb,                    \
1231      GLUE(cs_, CAR(__VA_ARGS__)),pa_##parser,di_##comparer,             \
1232      adns__ckl_hostname, CDR(__VA_ARGS__) }
1233 #define FLAT_TYPE(code,rrt,fmt,memb,parser,comparer,/*printer*/...)     \
1234  { adns_r_##code, rrt,fmt,TYPESZ_M(memb), mf_flat,                      \
1235      GLUE(cs_, CAR(__VA_ARGS__)),pa_##parser,di_##comparer,             \
1236      adns__ckl_hostname, CDR(__VA_ARGS__) }
1237
1238 #define di_0 0
1239
1240 static const typeinfo typeinfos[] = {
1241 /* Must be in ascending order of rrtype ! */
1242 /* mem-mgmt code  rrt     fmt   member   parser      comparer  printer */
1243
1244 FLAT_TYPE(a,      "A",     0,   inaddr,    inaddr,  inaddr,inaddr          ),
1245 DEEP_TYPE(ns_raw, "NS",   "raw",str,       host_raw,0,     domain          ),
1246 DEEP_TYPE(cname,  "CNAME", 0,   str,       dom_raw, 0,     domain          ),
1247 DEEP_TYPE(soa_raw,"SOA",  "raw",soa,       soa,     0,     soa             ),
1248 DEEP_TYPE(ptr_raw,"PTR",  "raw",str,       host_raw,0,     domain          ),
1249 DEEP_TYPE(hinfo,  "HINFO", 0,   intstrpair,hinfo,   0,     hinfo           ),
1250 DEEP_TYPE(mx_raw, "MX",   "raw",intstr,    mx_raw,  mx_raw,inthost         ),
1251 DEEP_TYPE(txt,    "TXT",   0,   manyistr,  txt,     0,     txt             ),
1252 DEEP_TYPE(rp_raw, "RP",   "raw",strpair,   rp,      0,     rp              ),
1253 DEEP_TYPE(srv_raw,"SRV",  "raw",srvraw ,   srvraw,  srv,   srvraw,
1254                               .checklabel= ckl_srv, .postsort= postsort_srv),
1255
1256 FLAT_TYPE(addr,   "A",  "addr", addr,      addr,    addr,  addr            ),
1257 DEEP_TYPE(ns,     "NS", "+addr",hostaddr,  hostaddr,hostaddr,hostaddr      ),
1258 DEEP_TYPE(ptr,    "PTR","checked",str,     ptr,     0,     domain,
1259                                                        .checklabel= ckl_ptr),
1260 DEEP_TYPE(mx,     "MX", "+addr",inthostaddr,mx,     mx,    inthostaddr,    ),
1261 DEEP_TYPE(srv,    "SRV","+addr",srvha,     srvha,   srv,   srvha,
1262                               .checklabel= ckl_srv, .postsort= postsort_srv),
1263
1264 DEEP_TYPE(soa,    "SOA","822",  soa,       soa,     0,     soa             ),
1265 DEEP_TYPE(rp,     "RP", "822",  strpair,   rp,      0,     rp              ),
1266 };
1267
1268 static const typeinfo typeinfo_unknown=
1269 DEEP_TYPE(unknown,0, "unknown",byteblock,opaque,  0,     opaque            );
1270
1271 const typeinfo *adns__findtype(adns_rrtype type) {
1272   const typeinfo *begin, *end, *mid;
1273
1274   if (type & adns_r_unknown) return &typeinfo_unknown;
1275
1276   begin= typeinfos;  end= typeinfos+(sizeof(typeinfos)/sizeof(typeinfo));
1277
1278   while (begin < end) {
1279     mid= begin + ((end-begin)>>1);
1280     if (mid->typekey == type) return mid;
1281     if (type > mid->typekey) begin= mid+1;
1282     else end= mid;
1283   }
1284   return 0;
1285 }