chiark / gitweb /
Does further A lookups and uses 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 Copyright (C) 1997, 1998 Ian Jackson
7  *  
8  *  This program is free software; you can redistribute it and/or modify
9  *  it under the terms of the GNU General Public License as published by
10  *  the Free Software Foundation; either version 2, or (at your option)
11  *  any later version.
12  *  
13  *  This program is distributed in the hope that it will be useful,
14  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
15  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  *  GNU General Public License for more details.
17  *  
18  *  You should have received a copy of the GNU General Public License
19  *  along with this program; if not, write to the Free Software Foundation,
20  *  Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. 
21  */
22
23 #include <stdlib.h>
24 #include <string.h>
25
26 #include <arpa/inet.h>
27
28 #include "internal.h"
29
30 #define R_NOMEM           return adns_s_nolocalmem
31 #define CSP_ADDSTR(s)     if (!adns__vbuf_appendstr(vb,(s))) R_NOMEM; else;
32
33 /*
34  * order of sections:
35  *
36  * _textdata, _qstring        (csp)
37  * _str                       (mf,cs)
38  * _intstr                    (mf,cs)
39  * _manyistr                  (mf,cs)
40  * _txt                       (pa)
41  * _inaddr                    (pa,dip,di)
42  * _addr                      (pa,di,csp,cs)
43  * _domain                    (pap)
44  * _host_raw                  (pa)
45  * _hostaddr                  (pap,pa,dip,di,mfp,mf,csp,cs +pap_findaddrs)
46  * _mx_raw                    (pa,di)
47  * _mx                        (pa,di)
48  * _inthostaddr               (mf,cs)
49  * _flat                      (mf)
50  *
51  * within each section:
52  *    pap_*
53  *    pa_*
54  *    dip_*
55  *    di_*
56  *    mfp_*
57  *    mf_*
58  *    csp_*
59  *    cs_*
60  */
61
62 /*
63  * _textdata, _qstring   (csp)
64  */
65
66 static adns_status csp_textdata(vbuf *vb, const char *dp, int len) {
67   unsigned char ch;
68   char buf[10];
69   int cn;
70
71   CSP_ADDSTR("\"");
72   for (cn=0; cn<len; cn++) {
73     ch= *dp++;
74     if (ch >= 32 && ch <= 126 && ch != '"' && ch != '\\') {
75       if (!adns__vbuf_append(vb,&ch,1)) R_NOMEM;
76     } else {
77       sprintf(buf,"\\%02x",ch);
78       CSP_ADDSTR(buf);
79     }
80   }
81   CSP_ADDSTR("\"");
82   
83   return adns_s_ok;
84 }
85
86 static adns_status csp_qstring(vbuf *vb, const char *dp) {
87   return csp_textdata(vb, dp, strlen(dp));
88 }
89
90 /*
91  * _str  (mf,cs)
92  */
93
94 static void mf_str(adns_query qu, void *datap) {
95   char **rrp= datap;
96
97   adns__makefinal_str(qu,rrp);
98 }
99
100 static adns_status cs_str(vbuf *vb, const void *datap) {
101   const char *const *rrp= datap;
102
103   return csp_qstring(vb,*rrp);
104 }
105
106 /*
107  * _intstr  (mf,cs)
108  */
109
110 static void mf_intstr(adns_query qu, void *datap) {
111   adns_rr_intstr *rrp= datap;
112
113   adns__makefinal_str(qu,&rrp->str);
114 }
115
116 static adns_status cs_intstr(vbuf *vb, const void *datap) {
117   const adns_rr_intstr *rrp= datap;
118   char buf[10];
119
120   sprintf(buf,"%u ",rrp->i);
121   CSP_ADDSTR(buf);
122   return csp_qstring(vb,rrp->str);
123 }
124
125 /*
126  * _manyistr   (mf,cs)
127  */
128
129 static void mf_manyistr(adns_query qu, void *datap) {
130   adns_rr_intstr **rrp= datap;
131   adns_rr_intstr *te, *table;
132   void *tablev;
133   int tc;
134
135   for (tc=0, te= *rrp; te->i >= 0; te++, tc++);
136   tablev= *rrp;
137   adns__makefinal_block(qu,&tablev,sizeof(*te)*(tc+1));
138   *rrp= table= tablev;
139   for (te= *rrp; te->i >= 0; te++)
140     adns__makefinal_str(qu,&te->str);
141 }
142
143 static adns_status cs_manyistr(vbuf *vb, const void *datap) {
144   const adns_rr_intstr *const *rrp= datap;
145   const adns_rr_intstr *current;
146   adns_status st;
147   int spc;
148
149   for (spc=0, current= *rrp; current->i >= 0; current++) {
150     if (spc) CSP_ADDSTR(" ");
151     st= csp_textdata(vb,current->str,current->i); if (st) return st;
152   }
153   return adns_s_ok;
154 }
155
156 /*
157  * _txt   (pa)
158  */
159
160 static adns_status pa_txt(const parseinfo *pai, int cbyte, int max, void *datap) {
161   adns_rr_intstr **rrp= datap, *table, *te;
162   const byte *dgram= pai->dgram;
163   int ti, tc, l, startbyte;
164
165   startbyte= cbyte;
166   if (cbyte >= max) return adns_s_invaliddata;
167   tc= 0;
168   while (cbyte < max) {
169     GET_B(cbyte,l);
170     cbyte+= l;
171   }
172   if (cbyte != max) return adns_s_invaliddata;
173
174   table= adns__alloc_interim(pai->qu,sizeof(*table)*(tc+1));
175   if (!table) R_NOMEM;
176
177   for (cbyte=startbyte, ti=0, te=table; ti<tc; ti++, te++) {
178     GET_B(cbyte,l);
179     te->str= adns__alloc_interim(pai->qu, l+1);
180     if (!te->str) R_NOMEM;
181     te->str[l]= 0;
182     memcpy(te->str,dgram+cbyte,l);
183     te->i= l;
184   }
185   assert(cbyte == max);
186
187   te->i= -1;
188   te->str= 0;
189   
190   *rrp= table;
191   return adns_s_ok;
192 }
193
194 /*
195  * _inaddr   (pa,dip,di)
196  */
197
198 static adns_status pa_inaddr(const parseinfo *pai, int cbyte, int max, void *datap) {
199   struct in_addr *storeto= datap;
200   
201   if (max-cbyte != 4) return adns_s_invaliddata;
202   memcpy(storeto, pai->dgram + cbyte, 4);
203   return adns_s_ok;
204 }
205
206 static int dip_inaddr(struct in_addr a, struct in_addr b) {
207   /* fixme implement sortlist */
208   return 0;
209 }
210
211 static int di_inaddr(const void *datap_a, const void *datap_b) {
212   const struct in_addr *ap= datap_a, *bp= datap_b;
213
214   return dip_inaddr(*ap,*bp);
215 }
216
217 static adns_status cs_inaddr(vbuf *vb, const void *datap) {
218   const struct in_addr *rrp= datap, rr= *rrp;
219   const char *ia;
220
221   ia= inet_ntoa(rr); assert(ia);
222   CSP_ADDSTR(ia);
223   return adns_s_ok;
224 }
225
226 /*
227  * _addr   (pa,di,csp,cs)
228  */
229
230 static adns_status pa_addr(const parseinfo *pai, int cbyte, int max, void *datap) {
231   adns_rr_addr *storeto= datap;
232   const byte *dgram= pai->dgram;
233
234   if (max-cbyte != 4) return adns_s_invaliddata;
235   storeto->len= sizeof(storeto->addr.inet);
236   memset(&storeto->addr,0,sizeof(storeto->addr.inet));
237   storeto->addr.inet.sin_family= AF_INET;
238   storeto->addr.inet.sin_port= 0;
239   memcpy(&storeto->addr.inet.sin_addr,dgram+cbyte,4);
240   return adns_s_ok;
241 }
242
243 static int di_addr(const void *datap_a, const void *datap_b) {
244   const adns_rr_addr *ap= datap_a, *bp= datap_b;
245
246   assert(ap->addr.sa.sa_family == AF_INET);
247   return dip_inaddr(ap->addr.inet.sin_addr,bp->addr.inet.sin_addr);
248 }
249
250 static adns_status csp_addr(vbuf *vb, const adns_rr_addr *rrp) {
251   const char *ia;
252   static char buf[30];
253
254   switch (rrp->addr.inet.sin_family) {
255   case AF_INET:
256     CSP_ADDSTR("AF_INET ");
257     ia= inet_ntoa(rrp->addr.inet.sin_addr); assert(ia);
258     CSP_ADDSTR(ia);
259     break;
260   default:
261     sprintf(buf,"AF=%u",rrp->addr.sa.sa_family);
262     CSP_ADDSTR(buf);
263     break;
264   }
265   return adns_s_ok;
266 }
267
268 static adns_status cs_addr(vbuf *vb, const void *datap) {
269   const adns_rr_addr *rrp= datap;
270
271   return csp_addr(vb,rrp);
272 }
273
274 /*
275  * _domain  (pap)
276  */
277
278 static adns_status pap_domain(const parseinfo *pai, int *cbyte_io, int max,
279                               char **domain_r, parsedomain_flags flags) {
280   adns_status st;
281   char *dm;
282   
283   st= adns__parse_domain(pai->qu->ads, pai->serv, pai->qu, &pai->qu->vb, flags,
284                          pai->dgram,pai->dglen, cbyte_io, max);
285   if (st) return st;
286   if (!pai->qu->vb.used) return adns_s_invaliddata;
287
288   dm= adns__alloc_interim(pai->qu, pai->qu->vb.used+1);
289   if (!dm) R_NOMEM;
290
291   dm[pai->qu->vb.used]= 0;
292   memcpy(dm,pai->qu->vb.buf,pai->qu->vb.used);
293   
294   *domain_r= dm;
295   return adns_s_ok;
296 }
297
298 /*
299  * _host_raw   (pa)
300  */
301
302 static adns_status pa_host_raw(const parseinfo *pai, int cbyte, int max, void *datap) {
303   char **rrp= datap;
304   adns_status st;
305
306   st= pap_domain(pai, &cbyte, max, rrp,
307                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
308   if (st) return st;
309   
310   if (cbyte != max) return adns_s_invaliddata;
311   return adns_s_ok;
312 }
313
314 /*
315  * _hostaddr   (pap,pa,dip,di,mfp,mf,csp,cs +icb_hostaddr, pap_findaddrs)
316  */
317
318 static adns_status pap_findaddrs(const parseinfo *pai, adns_rr_hostaddr *ha,
319                                  int *cbyte_io, int count, int dmstart) {
320   int rri, naddrs;
321   int type, class, rdlen, rdstart, ownermatched;
322   adns_status st;
323   
324   for (rri=0, naddrs=-1; rri<count; rri++) {
325     st= adns__findrr_anychk(pai->qu, pai->serv, pai->dgram, pai->dglen, cbyte_io,
326                             &type, &class, &rdlen, &rdstart,
327                             pai->dgram, pai->dglen, dmstart, &ownermatched);
328     if (st) return st;
329     if (!ownermatched || class != DNS_CLASS_IN || type != adns_r_a) {
330       if (naddrs>0) break; else continue;
331     }
332     if (naddrs == -1) {
333       naddrs= 0;
334     }
335     if (!adns__vbuf_ensure(&pai->qu->vb, (naddrs+1)*sizeof(adns_rr_addr))) R_NOMEM;
336     st= pa_addr(pai, rdstart,rdstart+rdlen,
337                 pai->qu->vb.buf + naddrs*sizeof(adns_rr_addr));
338     if (st) return st;
339     naddrs++;
340   }
341   if (naddrs >= 0) {
342     ha->addrs= adns__alloc_interim(pai->qu, naddrs*sizeof(adns_rr_addr));
343     if (!ha->addrs) R_NOMEM;
344     memcpy(ha->addrs, pai->qu->vb.buf, naddrs*sizeof(adns_rr_addr));
345     ha->naddrs= naddrs;
346     ha->astatus= adns_s_ok;
347
348     adns__isort(ha->addrs, naddrs, sizeof(adns_rr_addr), pai->qu->vb.buf, di_addr);
349   }
350   return adns_s_ok;
351 }
352
353 static void icb_hostaddr(adns_query parent, adns_query child) {
354   adns_rr_hostaddr *rrp= child->context.intern.info.hostaddr;
355   adns_answer *cans= child->answer;
356
357   rrp->astatus= cans->status;
358   rrp->naddrs= cans->nrrs;
359   rrp->addrs= cans->rrs.addr;
360   adns__transfer_interim(child, parent, rrp->addrs, rrp->naddrs*sizeof(adns_rr_addr));
361 }
362
363 static adns_status pap_hostaddr(const parseinfo *pai, int *cbyte_io,
364                                 int max, adns_rr_hostaddr *rrp) {
365   adns_status st;
366   int dmstart, cbyte;
367   qcontext ctx;
368   int id;
369   adns_query nqu;
370
371   dmstart= cbyte= *cbyte_io;
372   st= pap_domain(pai, &cbyte, max, &rrp->host,
373                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
374   if (st) return st;
375   *cbyte_io= cbyte;
376
377   rrp->astatus= adns_s_ok;
378   rrp->naddrs= -1;
379   rrp->addrs= 0;
380
381   cbyte= pai->nsstart;
382
383   st= pap_findaddrs(pai, rrp, &cbyte, pai->nscount, dmstart);
384   if (st) return st;
385   if (rrp->naddrs != -1) return adns_s_ok;
386
387   st= pap_findaddrs(pai, rrp, &cbyte, pai->arcount, dmstart);
388   if (st) return st;
389   if (rrp->naddrs != -1) return adns_s_ok;
390
391   st= adns__mkquery_frdgram(pai->ads, &pai->qu->vb, &id,
392                             pai->dgram, pai->dglen, dmstart,
393                             adns_r_addr, adns_qf_quoteok_query);
394   if (st) return st;
395
396   ctx.intern.callback= icb_hostaddr;
397   ctx.intern.info.hostaddr= rrp;
398   st= adns__internal_submit(pai->ads, &nqu, adns__findtype(adns_r_addr),
399                             &pai->qu->vb, id,
400                             adns_qf_quoteok_query, pai->now, 0, &ctx);
401   if (st) return st;
402
403   nqu->parent= pai->qu;
404   LIST_LINK_TAIL_PART(pai->qu->children,nqu,siblings.);
405
406   return adns_s_ok;
407 }
408
409 static adns_status pa_hostaddr(const parseinfo *pai, int cbyte, int max, void *datap) {
410   adns_rr_hostaddr *rrp= datap;
411   adns_status st;
412
413   st= pap_hostaddr(pai, &cbyte, max, rrp);
414   if (st) return st;
415   if (cbyte != max) return adns_s_invaliddata;
416
417   return adns_s_ok;
418 }
419
420 static int dip_hostaddr(const adns_rr_hostaddr *ap, const adns_rr_hostaddr *bp) {
421   if (ap->astatus != bp->astatus) return ap->astatus;
422   if (ap->astatus) return 0;
423
424   assert(ap->addrs[0].addr.sa.sa_family == AF_INET);
425   assert(bp->addrs[0].addr.sa.sa_family == AF_INET);
426   return dip_inaddr(ap->addrs[0].addr.inet.sin_addr, bp->addrs[0].addr.inet.sin_addr);
427 }
428
429 static int di_hostaddr(const void *datap_a, const void *datap_b) {
430   const adns_rr_hostaddr *ap= datap_a, *bp= datap_b;
431
432   return dip_hostaddr(ap,bp);
433 }
434
435 static void mfp_hostaddr(adns_query qu, adns_rr_hostaddr *rrp) {
436   void *tablev;
437
438   adns__makefinal_str(qu,&rrp->host);
439   tablev= rrp->addrs;
440   adns__makefinal_block(qu, &tablev, rrp->naddrs*sizeof(*rrp->addrs));
441   rrp->addrs= tablev;
442 }
443
444 static void mf_hostaddr(adns_query qu, void *datap) {
445   adns_rr_hostaddr *rrp= datap;
446
447   mfp_hostaddr(qu,rrp);
448 }
449
450 static adns_status csp_hostaddr(vbuf *vb, const adns_rr_hostaddr *rrp) {
451   const char *to_add;
452   adns_status st;
453   int i;
454
455   st= csp_qstring(vb,rrp->host); if (st) return st;
456
457   if (rrp->astatus) {
458     CSP_ADDSTR(" - ");
459     CSP_ADDSTR(adns_strerror(rrp->astatus));
460   } else {
461     assert(rrp->naddrs > 0);
462     for (i=0, to_add= ": "; i<rrp->naddrs; i++, to_add= ", ") {
463       CSP_ADDSTR(to_add);
464       st= csp_addr(vb,&rrp->addrs[i]);
465     }
466   }
467   return adns_s_ok;
468 }
469
470 static adns_status cs_hostaddr(vbuf *vb, const void *datap) {
471   const adns_rr_hostaddr *rrp= datap;
472
473   return csp_hostaddr(vb,rrp);
474 }
475
476 /*
477  * _mx_raw   (pa,di)
478  */
479
480 static adns_status pa_mx_raw(const parseinfo *pai, int cbyte, int max, void *datap) {
481   const byte *dgram= pai->dgram;
482   adns_rr_intstr *rrp= datap;
483   adns_status st;
484   int pref;
485
486   if (cbyte+2 > max) return adns_s_invaliddata;
487   GET_W(cbyte,pref);
488   rrp->i= pref;
489   st= pap_domain(pai, &cbyte, max, &rrp->str,
490                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
491   if (st) return st;
492   
493   if (cbyte != max) return adns_s_invaliddata;
494   return adns_s_ok;
495 }
496
497 static int di_mx_raw(const void *datap_a, const void *datap_b) {
498   const adns_rr_intstr *ap= datap_a, *bp= datap_b;
499
500   if (ap->i < bp->i) return 0;
501   if (ap->i > bp->i) return 1;
502   return 0;
503 }
504
505 /*
506  * _mx   (pa,di)
507  */
508
509 static adns_status pa_mx(const parseinfo *pai, int cbyte, int max, void *datap) {
510   const byte *dgram= pai->dgram;
511   adns_rr_inthostaddr *rrp= datap;
512   adns_status st;
513   int pref;
514
515   if (cbyte+2 > max) return adns_s_invaliddata;
516   GET_W(cbyte,pref);
517   rrp->i= pref;
518   st= pap_hostaddr(pai, &cbyte, max, &rrp->ha);
519   if (st) return st;
520   
521   if (cbyte != max) return adns_s_invaliddata;
522   return adns_s_ok;
523 }
524
525 static int di_mx(const void *datap_a, const void *datap_b) {
526   const adns_rr_inthostaddr *ap= datap_a, *bp= datap_b;
527
528   if (ap->i < bp->i) return 0;
529   if (ap->i > bp->i) return 1;
530   return dip_hostaddr(&ap->ha,&bp->ha);
531 }
532
533 /*
534  * _inthostaddr  (mf,cs)
535  */
536
537 static void mf_inthostaddr(adns_query qu, void *datap) {
538   adns_rr_inthostaddr *rrp= datap;
539
540   mfp_hostaddr(qu,&rrp->ha);
541 }
542
543 static adns_status cs_inthostaddr(vbuf *vb, const void *datap) {
544   const adns_rr_inthostaddr *rrp= datap;
545   char buf[10];
546
547   sprintf(buf,"%u ",rrp->i);
548   CSP_ADDSTR(buf);
549
550   return csp_hostaddr(vb,&rrp->ha);
551 }
552
553 /*
554  * _flat   (mf)
555  */
556
557 static void mf_flat(adns_query qu, void *data) { }
558
559 /*
560  * Now the table.
561  */
562
563 #define TYPESZ_M(member)           (sizeof(*((adns_answer*)0)->rrs.member))
564
565 #define DEEP_MEMB(memb) TYPESZ_M(memb), mf_##memb, cs_##memb
566 #define FLAT_MEMB(memb) TYPESZ_M(memb), mf_flat, cs_##memb
567
568 #define DEEP_TYPE(code,rrt,fmt,memb,parser,comparer) \
569  { adns_r_##code, rrt, fmt, TYPESZ_M(memb), mf_##memb, cs_##memb, parser, comparer }
570 #define FLAT_TYPE(code,rrt,fmt,memb,parser,comparer) \
571  { adns_r_##code, rrt, fmt, TYPESZ_M(memb), mf_flat, cs_##memb, parser, comparer }
572
573 static const typeinfo typeinfos[] = {
574 /* Must be in ascending order of rrtype ! */
575 /* mem-mgmt code     rrt     fmt      member       parser        comparer    */
576                                                              
577   FLAT_TYPE(a,       "A",     0,      inaddr,      pa_inaddr,    di_inaddr   ),
578   DEEP_TYPE(ns_raw,  "NS",   "raw",   str,         pa_host_raw,  0           ),
579   DEEP_TYPE(cname,   "CNAME", 0,      str,         pa_host_raw,  0           ),
580 #if 0
581   DEEP_TYPE(soa_raw, "SOA",  "raw",   soa,         pa_soa,       0           ),
582 #endif                                                                               
583   DEEP_TYPE(ptr_raw, "PTR",  "raw",   str,         pa_host_raw,  0           ),
584 #if 0
585   DEEP_TYPE(hinfo,   "HINFO", 0,      strpair,     pa_hinfo,     0           ),
586 #endif                                                                               
587   DEEP_TYPE(mx_raw,  "MX",   "raw",   intstr,      pa_mx_raw,    di_mx_raw   ),
588   DEEP_TYPE(txt,     "TXT",   0,      manyistr,    pa_txt,       0           ),
589 #if 0
590   DEEP_TYPE(rp_raw,  "RP",   "raw",   strpair,     pa_rp,        0           ),
591 #endif                                                                               
592                                                                                      
593   FLAT_TYPE(addr,    "A",  "addr",    addr,        pa_addr,      di_addr     ),
594   DEEP_TYPE(ns,      "NS", "+addr",   hostaddr,    pa_hostaddr,  di_hostaddr ),
595 #if 0
596   DEEP_TYPE(ptr,     "PTR","checked", str,         pa_ptr,       0           ),
597 #endif
598   DEEP_TYPE(mx,      "MX", "+addr",   inthostaddr, pa_mx,        di_mx       ),
599                                                                            
600 #if 0
601   DEEP_TYPE(soa,     "SOA","822",     soa,         pa_soa,        0          ),
602   DEEP_TYPE(rp,      "RP", "822",     strpair,     pa_rp,         0          ),
603 #endif /* fixme - implement all types */
604 };
605
606 const typeinfo *adns__findtype(adns_rrtype type) {
607   const typeinfo *begin, *end, *mid;
608
609   begin= typeinfos;  end= typeinfos+(sizeof(typeinfos)/sizeof(typeinfo));
610
611   while (begin < end) {
612     mid= begin + ((end-begin)>>1);
613     if (mid->type == type) return mid;
614     if (type > mid->type) begin= mid+1;
615     else end= mid;
616   }
617   return 0;
618 }