Search list implementation compiles, but not tested.
[adns] / 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-1999 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_nomemory
31 #define CSP_ADDSTR(s) if (!adns__vbuf_appendstr(vb,(s))) R_NOMEM; else;
32
33 /*
34 * order of sections:
35 *
36 * _string (pap)
37 * _textdata, _qstring (csp)
38 * _str (mf,cs)
39 * _intstr (mf,csp,cs)
40 * _manyistr (mf,cs)
41 * _txt (pa)
42 * _inaddr (pa,dip,di)
43 * _addr (pa,di,csp,cs)
44 * _domain (pap)
45 * _host_raw (pa)
46 * _hostaddr (pap,pa,dip,di,mfp,mf,csp,cs +pap_findaddrs)
47 * _mx_raw (pa,di)
48 * _mx (pa,di)
49 * _inthostaddr (mf,cs)
50 * _ptr (pa)
51 * _strpair (mf,cs)
52 * _intstrpair (mf,cs)
53 * _hinfo (pa)
54 * _mailbox (pap)
55 * _rp (pa)
56 * _soa (pa,mf,cs)
57 * _flat (mf)
58 *
59 * within each section:
60 * pap_*
61 * pa_*
62 * dip_*
63 * di_*
64 * mfp_*
65 * mf_*
66 * csp_*
67 * cs_*
68 */
69
70 /*
71 * _string (pap)
72 * _textdata, _qstring (csp)
73 */
74
75 static adns_status pap_string(const parseinfo *pai, int *cbyte_io, int max,
76 int *len_r, char **str_r) {
77 /* Neither len_r nor str_r may be null.
78 * End of datagram (overrun) is indicated by returning adns_s_invaliddata;
79 */
80 const byte *dgram= pai->dgram;
81 int l, cbyte;
82 char *str;
83
84 cbyte= *cbyte_io;
85
86 if (cbyte >= max) return adns_s_invaliddata;
87 GET_B(cbyte,l);
88 if (cbyte+l > max) return adns_s_invaliddata;
89
90 str= adns__alloc_interim(pai->qu, l+1);
91 if (!str) R_NOMEM;
92
93 str[l]= 0;
94 memcpy(str,dgram+cbyte,l);
95
96 *len_r= l;
97 *str_r= str;
98 *cbyte_io= cbyte+l;
99
100 return adns_s_ok;
101 }
102
103 static adns_status csp_textdata(vbuf *vb, const char *dp, int len) {
104 unsigned char ch;
105 char buf[10];
106 int cn;
107
108 CSP_ADDSTR("\"");
109 for (cn=0; cn<len; cn++) {
110 ch= *dp++;
111 if (ch >= 32 && ch <= 126 && ch != '"' && ch != '\\') {
112 if (!adns__vbuf_append(vb,&ch,1)) R_NOMEM;
113 } else if (ch == '\\') {
114 CSP_ADDSTR("\\\\");
115 } else {
116 sprintf(buf,"\\x%02x",ch);
117 CSP_ADDSTR(buf);
118 }
119 }
120 CSP_ADDSTR("\"");
121
122 return adns_s_ok;
123 }
124
125 static adns_status csp_qstring(vbuf *vb, const char *dp) {
126 return csp_textdata(vb, dp, strlen(dp));
127 }
128
129 /*
130 * _str (mf,cs)
131 */
132
133 static void mf_str(adns_query qu, void *datap) {
134 char **rrp= datap;
135
136 adns__makefinal_str(qu,rrp);
137 }
138
139 static adns_status cs_str(vbuf *vb, const void *datap) {
140 const char *const *rrp= datap;
141
142 return csp_qstring(vb,*rrp);
143 }
144
145 /*
146 * _intstr (mf,csp,cs)
147 */
148
149 static void mf_intstr(adns_query qu, void *datap) {
150 adns_rr_intstr *rrp= datap;
151
152 adns__makefinal_str(qu,&rrp->str);
153 }
154
155 static adns_status csp_intstr(vbuf *vb, const adns_rr_intstr *rrp) {
156 char buf[10];
157
158 sprintf(buf,"%u ",rrp->i);
159 CSP_ADDSTR(buf);
160 return csp_qstring(vb,rrp->str);
161 }
162
163 static adns_status cs_intstr(vbuf *vb, const void *datap) {
164 const adns_rr_intstr *rrp= datap;
165
166 return csp_intstr(vb,rrp);
167 }
168
169 /*
170 * _manyistr (mf,cs)
171 */
172
173 static void mf_manyistr(adns_query qu, void *datap) {
174 adns_rr_intstr **rrp= datap;
175 adns_rr_intstr *te, *table;
176 void *tablev;
177 int tc;
178
179 for (tc=0, te= *rrp; te->i >= 0; te++, tc++);
180 tablev= *rrp;
181 adns__makefinal_block(qu,&tablev,sizeof(*te)*(tc+1));
182 *rrp= table= tablev;
183 for (te= *rrp; te->i >= 0; te++)
184 adns__makefinal_str(qu,&te->str);
185 }
186
187 static adns_status cs_manyistr(vbuf *vb, const void *datap) {
188 const adns_rr_intstr *const *rrp= datap;
189 const adns_rr_intstr *current;
190 adns_status st;
191 int spc;
192
193 for (spc=0, current= *rrp; current->i >= 0; current++) {
194 if (spc) CSP_ADDSTR(" ");
195 st= csp_textdata(vb,current->str,current->i); if (st) return st;
196 }
197 return adns_s_ok;
198 }
199
200 /*
201 * _txt (pa)
202 */
203
204 static adns_status pa_txt(const parseinfo *pai, int cbyte, int max, void *datap) {
205 adns_rr_intstr **rrp= datap, *table, *te;
206 const byte *dgram= pai->dgram;
207 int ti, tc, l, startbyte;
208 adns_status st;
209
210 startbyte= cbyte;
211 if (cbyte >= max) return adns_s_invaliddata;
212 tc= 0;
213 while (cbyte < max) {
214 GET_B(cbyte,l);
215 cbyte+= l;
216 tc++;
217 }
218 if (cbyte != max) return adns_s_invaliddata;
219
220 table= adns__alloc_interim(pai->qu,sizeof(*table)*(tc+1));
221 if (!table) R_NOMEM;
222
223 for (cbyte=startbyte, ti=0, te=table; ti<tc; ti++, te++) {
224 st= pap_string(pai, &cbyte, max, &te->i, &te->str);
225 if (st) return st;
226 }
227 assert(cbyte == max);
228
229 te->i= -1;
230 te->str= 0;
231
232 *rrp= table;
233 return adns_s_ok;
234 }
235
236 /*
237 * _inaddr (pa,dip,di)
238 */
239
240 static adns_status pa_inaddr(const parseinfo *pai, int cbyte, int max, void *datap) {
241 struct in_addr *storeto= datap;
242
243 if (max-cbyte != 4) return adns_s_invaliddata;
244 memcpy(storeto, pai->dgram + cbyte, 4);
245 return adns_s_ok;
246 }
247
248 static int search_sortlist(adns_state ads, struct in_addr ad) {
249 const struct sortlist *slp;
250 int i;
251
252 for (i=0, slp=ads->sortlist;
253 i<ads->nsortlist && !((ad.s_addr & slp->mask.s_addr) == slp->base.s_addr);
254 i++, slp++);
255 return i;
256 }
257
258 static int dip_inaddr(adns_state ads, struct in_addr a, struct in_addr b) {
259 int ai, bi;
260
261 if (!ads->nsortlist) return 0;
262
263 ai= search_sortlist(ads,a);
264 bi= search_sortlist(ads,b);
265 return bi<ai;
266 }
267
268 static int di_inaddr(adns_state ads, const void *datap_a, const void *datap_b) {
269 const struct in_addr *ap= datap_a, *bp= datap_b;
270
271 return dip_inaddr(ads,*ap,*bp);
272 }
273
274 static adns_status cs_inaddr(vbuf *vb, const void *datap) {
275 const struct in_addr *rrp= datap, rr= *rrp;
276 const char *ia;
277
278 ia= inet_ntoa(rr); assert(ia);
279 CSP_ADDSTR(ia);
280 return adns_s_ok;
281 }
282
283 /*
284 * _addr (pa,di,csp,cs)
285 */
286
287 static adns_status pa_addr(const parseinfo *pai, int cbyte, int max, void *datap) {
288 adns_rr_addr *storeto= datap;
289 const byte *dgram= pai->dgram;
290
291 if (max-cbyte != 4) return adns_s_invaliddata;
292 storeto->len= sizeof(storeto->addr.inet);
293 memset(&storeto->addr,0,sizeof(storeto->addr.inet));
294 storeto->addr.inet.sin_family= AF_INET;
295 memcpy(&storeto->addr.inet.sin_addr,dgram+cbyte,4);
296 return adns_s_ok;
297 }
298
299 static int di_addr(adns_state ads, const void *datap_a, const void *datap_b) {
300 const adns_rr_addr *ap= datap_a, *bp= datap_b;
301
302 assert(ap->addr.sa.sa_family == AF_INET);
303 return dip_inaddr(ads, ap->addr.inet.sin_addr, bp->addr.inet.sin_addr);
304 }
305
306 static int div_addr(void *context, const void *datap_a, const void *datap_b) {
307 const adns_state ads= context;
308
309 return di_addr(ads, datap_a, datap_b);
310 }
311
312 static adns_status csp_addr(vbuf *vb, const adns_rr_addr *rrp) {
313 const char *ia;
314 static char buf[30];
315
316 switch (rrp->addr.inet.sin_family) {
317 case AF_INET:
318 CSP_ADDSTR("AF_INET ");
319 ia= inet_ntoa(rrp->addr.inet.sin_addr); assert(ia);
320 CSP_ADDSTR(ia);
321 break;
322 default:
323 sprintf(buf,"AF=%u",rrp->addr.sa.sa_family);
324 CSP_ADDSTR(buf);
325 break;
326 }
327 return adns_s_ok;
328 }
329
330 static adns_status cs_addr(vbuf *vb, const void *datap) {
331 const adns_rr_addr *rrp= datap;
332
333 return csp_addr(vb,rrp);
334 }
335
336 /*
337 * _domain (pap)
338 */
339
340 static adns_status pap_domain(const parseinfo *pai, int *cbyte_io, int max,
341 char **domain_r, parsedomain_flags flags) {
342 adns_status st;
343 char *dm;
344
345 st= adns__parse_domain(pai->qu->ads, pai->serv, pai->qu, &pai->qu->vb, flags,
346 pai->dgram,pai->dglen, cbyte_io, max);
347 if (st) return st;
348 if (!pai->qu->vb.used) return adns_s_invaliddata;
349
350 dm= adns__alloc_interim(pai->qu, pai->qu->vb.used+1);
351 if (!dm) R_NOMEM;
352
353 dm[pai->qu->vb.used]= 0;
354 memcpy(dm,pai->qu->vb.buf,pai->qu->vb.used);
355
356 *domain_r= dm;
357 return adns_s_ok;
358 }
359
360 /*
361 * _host_raw (pa)
362 */
363
364 static adns_status pa_host_raw(const parseinfo *pai, int cbyte, int max, void *datap) {
365 char **rrp= datap;
366 adns_status st;
367
368 st= pap_domain(pai, &cbyte, max, rrp,
369 pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
370 if (st) return st;
371
372 if (cbyte != max) return adns_s_invaliddata;
373 return adns_s_ok;
374 }
375
376 /*
377 * _hostaddr (pap,pa,dip,di,mfp,mf,csp,cs +icb_hostaddr, pap_findaddrs)
378 */
379
380 static adns_status pap_findaddrs(const parseinfo *pai, adns_rr_hostaddr *ha,
381 int *cbyte_io, int count, int dmstart) {
382 int rri, naddrs;
383 int type, class, rdlen, rdstart, ownermatched;
384 unsigned long ttl;
385 adns_status st;
386
387 for (rri=0, naddrs=-1; rri<count; rri++) {
388 st= adns__findrr_anychk(pai->qu, pai->serv, pai->dgram, pai->dglen, cbyte_io,
389 &type, &class, &ttl, &rdlen, &rdstart,
390 pai->dgram, pai->dglen, dmstart, &ownermatched);
391 if (st) return st;
392 if (!ownermatched || class != DNS_CLASS_IN || type != adns_r_a) {
393 if (naddrs>0) break; else continue;
394 }
395 if (naddrs == -1) {
396 naddrs= 0;
397 }
398 if (!adns__vbuf_ensure(&pai->qu->vb, (naddrs+1)*sizeof(adns_rr_addr))) R_NOMEM;
399 adns__update_expires(pai->qu,ttl,pai->now);
400 st= pa_addr(pai, rdstart,rdstart+rdlen,
401 pai->qu->vb.buf + naddrs*sizeof(adns_rr_addr));
402 if (st) return st;
403 naddrs++;
404 }
405 if (naddrs >= 0) {
406 ha->addrs= adns__alloc_interim(pai->qu, naddrs*sizeof(adns_rr_addr));
407 if (!ha->addrs) R_NOMEM;
408 memcpy(ha->addrs, pai->qu->vb.buf, naddrs*sizeof(adns_rr_addr));
409 ha->naddrs= naddrs;
410 ha->astatus= adns_s_ok;
411
412 adns__isort(ha->addrs, naddrs, sizeof(adns_rr_addr), pai->qu->vb.buf,
413 div_addr, pai->ads);
414 }
415 return adns_s_ok;
416 }
417
418 static void icb_hostaddr(adns_query parent, adns_query child) {
419 adns_answer *cans= child->answer;
420 adns_rr_hostaddr *rrp= child->ctx.info.hostaddr;
421
422 rrp->astatus= cans->status;
423 rrp->naddrs= cans->nrrs;
424 rrp->addrs= cans->rrs.addr;
425 adns__transfer_interim(child, parent, rrp->addrs, rrp->naddrs*sizeof(adns_rr_addr));
426
427 if (!parent->children.head) adns__query_done(parent);
428 }
429
430 static adns_status pap_hostaddr(const parseinfo *pai, int *cbyte_io,
431 int max, adns_rr_hostaddr *rrp) {
432 adns_status st;
433 int dmstart, cbyte;
434 qcontext ctx;
435 int id;
436 adns_query nqu;
437 adns_queryflags nflags;
438
439 dmstart= cbyte= *cbyte_io;
440 st= pap_domain(pai, &cbyte, max, &rrp->host,
441 pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
442 if (st) return st;
443 *cbyte_io= cbyte;
444
445 rrp->astatus= adns_s_ok;
446 rrp->naddrs= -1;
447 rrp->addrs= 0;
448
449 cbyte= pai->nsstart;
450
451 st= pap_findaddrs(pai, rrp, &cbyte, pai->nscount, dmstart);
452 if (st) return st;
453 if (rrp->naddrs != -1) return adns_s_ok;
454
455 st= pap_findaddrs(pai, rrp, &cbyte, pai->arcount, dmstart);
456 if (st) return st;
457 if (rrp->naddrs != -1) return adns_s_ok;
458
459 st= adns__mkquery_frdgram(pai->ads, &pai->qu->vb, &id,
460 pai->dgram, pai->dglen, dmstart,
461 adns_r_addr, adns_qf_quoteok_query);
462 if (st) return st;
463
464 ctx.ext= 0;
465 ctx.callback= icb_hostaddr;
466 ctx.info.hostaddr= rrp;
467
468 nflags= adns_qf_quoteok_query;
469 if (!(pai->qu->flags & adns_qf_cname_loose)) nflags |= adns_qf_cname_forbid;
470
471 st= adns__internal_submit(pai->ads, &nqu, adns__findtype(adns_r_addr),
472 &pai->qu->vb, id, nflags, pai->now, &ctx);
473 if (st) return st;
474
475 nqu->parent= pai->qu;
476 LIST_LINK_TAIL_PART(pai->qu->children,nqu,siblings.);
477
478 return adns_s_ok;
479 }
480
481 static adns_status pa_hostaddr(const parseinfo *pai, int cbyte, int max, void *datap) {
482 adns_rr_hostaddr *rrp= datap;
483 adns_status st;
484
485 st= pap_hostaddr(pai, &cbyte, max, rrp);
486 if (st) return st;
487 if (cbyte != max) return adns_s_invaliddata;
488
489 return adns_s_ok;
490 }
491
492 static int dip_hostaddr(adns_state ads, const adns_rr_hostaddr *ap, const adns_rr_hostaddr *bp) {
493 if (ap->astatus != bp->astatus) return ap->astatus;
494 if (ap->astatus) return 0;
495
496 assert(ap->addrs[0].addr.sa.sa_family == AF_INET);
497 assert(bp->addrs[0].addr.sa.sa_family == AF_INET);
498 return dip_inaddr(ads,
499 ap->addrs[0].addr.inet.sin_addr,
500 bp->addrs[0].addr.inet.sin_addr);
501 }
502
503 static int di_hostaddr(adns_state ads, const void *datap_a, const void *datap_b) {
504 const adns_rr_hostaddr *ap= datap_a, *bp= datap_b;
505
506 return dip_hostaddr(ads, ap,bp);
507 }
508
509 static void mfp_hostaddr(adns_query qu, adns_rr_hostaddr *rrp) {
510 void *tablev;
511
512 adns__makefinal_str(qu,&rrp->host);
513 tablev= rrp->addrs;
514 adns__makefinal_block(qu, &tablev, rrp->naddrs*sizeof(*rrp->addrs));
515 rrp->addrs= tablev;
516 }
517
518 static void mf_hostaddr(adns_query qu, void *datap) {
519 adns_rr_hostaddr *rrp= datap;
520
521 mfp_hostaddr(qu,rrp);
522 }
523
524 static adns_status csp_hostaddr(vbuf *vb, const adns_rr_hostaddr *rrp) {
525 const char *to_add;
526 adns_status st;
527 int i;
528
529 st= csp_qstring(vb,rrp->host); if (st) return st;
530
531 if (rrp->astatus) {
532 CSP_ADDSTR(" - ");
533 CSP_ADDSTR(adns_strerror(rrp->astatus));
534 } else {
535 assert(rrp->naddrs > 0);
536 for (i=0, to_add= ": "; i<rrp->naddrs; i++, to_add= ", ") {
537 CSP_ADDSTR(to_add);
538 st= csp_addr(vb,&rrp->addrs[i]);
539 }
540 }
541 return adns_s_ok;
542 }
543
544 static adns_status cs_hostaddr(vbuf *vb, const void *datap) {
545 const adns_rr_hostaddr *rrp= datap;
546
547 return csp_hostaddr(vb,rrp);
548 }
549
550 /*
551 * _mx_raw (pa,di)
552 */
553
554 static adns_status pa_mx_raw(const parseinfo *pai, int cbyte, int max, void *datap) {
555 const byte *dgram= pai->dgram;
556 adns_rr_intstr *rrp= datap;
557 adns_status st;
558 int pref;
559
560 if (cbyte+2 > max) return adns_s_invaliddata;
561 GET_W(cbyte,pref);
562 rrp->i= pref;
563 st= pap_domain(pai, &cbyte, max, &rrp->str,
564 pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
565 if (st) return st;
566
567 if (cbyte != max) return adns_s_invaliddata;
568 return adns_s_ok;
569 }
570
571 static int di_mx_raw(adns_state ads, const void *datap_a, const void *datap_b) {
572 const adns_rr_intstr *ap= datap_a, *bp= datap_b;
573
574 if (ap->i < bp->i) return 0;
575 if (ap->i > bp->i) return 1;
576 return 0;
577 }
578
579 /*
580 * _mx (pa,di)
581 */
582
583 static adns_status pa_mx(const parseinfo *pai, int cbyte, int max, void *datap) {
584 const byte *dgram= pai->dgram;
585 adns_rr_inthostaddr *rrp= datap;
586 adns_status st;
587 int pref;
588
589 if (cbyte+2 > max) return adns_s_invaliddata;
590 GET_W(cbyte,pref);
591 rrp->i= pref;
592 st= pap_hostaddr(pai, &cbyte, max, &rrp->ha);
593 if (st) return st;
594
595 if (cbyte != max) return adns_s_invaliddata;
596 return adns_s_ok;
597 }
598
599 static int di_mx(adns_state ads, const void *datap_a, const void *datap_b) {
600 const adns_rr_inthostaddr *ap= datap_a, *bp= datap_b;
601
602 if (ap->i < bp->i) return 0;
603 if (ap->i > bp->i) return 1;
604 return dip_hostaddr(ads, &ap->ha, &bp->ha);
605 }
606
607 /*
608 * _inthostaddr (mf,cs)
609 */
610
611 static void mf_inthostaddr(adns_query qu, void *datap) {
612 adns_rr_inthostaddr *rrp= datap;
613
614 mfp_hostaddr(qu,&rrp->ha);
615 }
616
617 static adns_status cs_inthostaddr(vbuf *vb, const void *datap) {
618 const adns_rr_inthostaddr *rrp= datap;
619 char buf[10];
620
621 sprintf(buf,"%u ",rrp->i);
622 CSP_ADDSTR(buf);
623
624 return csp_hostaddr(vb,&rrp->ha);
625 }
626
627 /*
628 * _ptr (pa, +icb_ptr)
629 */
630
631 static void icb_ptr(adns_query parent, adns_query child) {
632 adns_answer *cans= child->answer;
633 const adns_rr_addr *queried, *found;
634 int i;
635
636 if (cans->status == adns_s_nxdomain || cans->status == adns_s_nodata) {
637 adns__query_fail(parent,adns_s_inconsistent);
638 return;
639 } else if (cans->status) {
640 adns__query_fail(parent,cans->status);
641 return;
642 }
643
644 queried= &parent->ctx.info.ptr_parent_addr;
645 for (i=0, found=cans->rrs.addr; i<cans->nrrs; i++, found++) {
646 if (queried->len == found->len &&
647 !memcmp(&queried->addr,&found->addr,queried->len)) {
648 if (!parent->children.head) adns__query_done(parent);
649 return;
650 }
651 }
652
653 adns__query_fail(parent,adns_s_inconsistent);
654 }
655
656 static adns_status pa_ptr(const parseinfo *pai, int dmstart, int max, void *datap) {
657 static const char *(expectdomain[])= { DNS_INADDR_ARPA };
658
659 char **rrp= datap;
660 adns_status st;
661 adns_rr_addr *ap;
662 findlabel_state fls;
663 char *ep;
664 byte ipv[4];
665 char labbuf[4];
666 int cbyte, i, lablen, labstart, l, id;
667 adns_query nqu;
668 qcontext ctx;
669
670 cbyte= dmstart;
671 st= pap_domain(pai, &cbyte, max, rrp,
672 pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
673 if (st) return st;
674 if (cbyte != max) return adns_s_invaliddata;
675
676 ap= &pai->qu->ctx.info.ptr_parent_addr;
677 if (!ap->len) {
678 adns__findlabel_start(&fls, pai->ads, -1, pai->qu,
679 pai->qu->query_dgram, pai->qu->query_dglen,
680 pai->qu->query_dglen, DNS_HDRSIZE, 0);
681 for (i=0; i<4; i++) {
682 st= adns__findlabel_next(&fls,&lablen,&labstart); assert(!st);
683 if (lablen<=0 || lablen>3) return adns_s_querydomainwrong;
684 memcpy(labbuf, pai->qu->query_dgram + labstart, lablen); labbuf[lablen]= 0;
685 ipv[3-i]= strtoul(labbuf,&ep,10); if (*ep) return adns_s_querydomainwrong;
686 if (lablen>1 && pai->qu->query_dgram[labstart]=='0')
687 return adns_s_querydomainwrong;
688 }
689 for (i=0; i<sizeof(expectdomain)/sizeof(*expectdomain); i++) {
690 st= adns__findlabel_next(&fls,&lablen,&labstart); assert(!st);
691 l= strlen(expectdomain[i]);
692 if (lablen != l || memcmp(pai->qu->query_dgram + labstart, expectdomain[i], l))
693 return adns_s_querydomainwrong;
694 }
695 st= adns__findlabel_next(&fls,&lablen,0); assert(!st);
696 if (lablen) return adns_s_querydomainwrong;
697
698 ap->len= sizeof(struct sockaddr_in);
699 memset(&ap->addr,0,sizeof(ap->addr.inet));
700 ap->addr.inet.sin_family= AF_INET;
701 ap->addr.inet.sin_addr.s_addr=
702 htonl((ipv[0]<<24) | (ipv[1]<<16) | (ipv[2]<<8) | (ipv[3]));
703 }
704
705 st= adns__mkquery_frdgram(pai->ads, &pai->qu->vb, &id,
706 pai->dgram, pai->dglen, dmstart,
707 adns_r_addr, adns_qf_quoteok_query);
708 if (st) return st;
709
710 ctx.ext= 0;
711 ctx.callback= icb_ptr;
712 memset(&ctx.info,0,sizeof(ctx.info));
713 st= adns__internal_submit(pai->ads, &nqu, adns__findtype(adns_r_addr),
714 &pai->qu->vb, id,
715 adns_qf_quoteok_query, pai->now, &ctx);
716 if (st) return st;
717
718 nqu->parent= pai->qu;
719 LIST_LINK_TAIL_PART(pai->qu->children,nqu,siblings.);
720
721 return adns_s_ok;
722 }
723
724 /*
725 * _strpair (mf,cs)
726 */
727
728 static void mf_strpair(adns_query qu, void *datap) {
729 adns_rr_strpair *rrp= datap;
730
731 adns__makefinal_str(qu,&rrp->array[0]);
732 adns__makefinal_str(qu,&rrp->array[1]);
733 }
734
735 static adns_status cs_strpair(vbuf *vb, const void *datap) {
736 const adns_rr_strpair *rrp= datap;
737 adns_status st;
738
739 st= csp_qstring(vb,rrp->array[0]); if (st) return st;
740 CSP_ADDSTR(" ");
741 st= csp_qstring(vb,rrp->array[1]); if (st) return st;
742
743 return adns_s_ok;
744 }
745
746 /*
747 * _intstrpair (mf,cs)
748 */
749
750 static void mf_intstrpair(adns_query qu, void *datap) {
751 adns_rr_intstrpair *rrp= datap;
752
753 adns__makefinal_str(qu,&rrp->array[0].str);
754 adns__makefinal_str(qu,&rrp->array[1].str);
755 }
756
757 static adns_status cs_intstrpair(vbuf *vb, const void *datap) {
758 const adns_rr_intstrpair *rrp= datap;
759 adns_status st;
760
761 st= csp_intstr(vb,&rrp->array[0]); if (st) return st;
762 CSP_ADDSTR(" ");
763 st= csp_intstr(vb,&rrp->array[1]); if (st) return st;
764
765 return adns_s_ok;
766 }
767
768 /*
769 * _hinfo (pa)
770 */
771
772 static adns_status pa_hinfo(const parseinfo *pai, int cbyte, int max, void *datap) {
773 adns_rr_intstrpair *rrp= datap;
774 adns_status st;
775 int i;
776
777 for (i=0; i<2; i++) {
778 st= pap_string(pai, &cbyte, max, &rrp->array[i].i, &rrp->array[i].str);
779 if (st) return st;
780 }
781
782 if (cbyte != max) return adns_s_invaliddata;
783
784 return adns_s_ok;
785 }
786
787 /*
788 * _mailbox (pap)
789 */
790
791 static adns_status pap_mailbox(const parseinfo *pai, int *cbyte_io, int max,
792 char **mb_r) {
793 int lablen, labstart, i, needquote, c, r, neednorm;
794 const unsigned char *p;
795 char *str;
796 findlabel_state fls;
797 adns_status st;
798 vbuf *vb;
799
800 vb= &pai->qu->vb;
801 vb->used= 0;
802 adns__findlabel_start(&fls, pai->ads,
803 -1, pai->qu,
804 pai->dgram, pai->dglen, max,
805 *cbyte_io, cbyte_io);
806 st= adns__findlabel_next(&fls,&lablen,&labstart);
807 if (!lablen) {
808 adns__vbuf_appendstr(vb,"<>");
809 goto x_ok;
810 }
811
812 neednorm= 1;
813 for (i=0, needquote=0, p= pai->dgram+labstart; i<lablen; i++) {
814 c= *p++;
815 if ((c&~128) < 32 || (c&~128) == 127) return adns_s_invaliddata;
816 if (c == '.' && !neednorm) neednorm= 1;
817 else if (strchr("()<>@,;:\\\".[]",c)) needquote++;
818 else neednorm= 0;
819 }
820
821 if (needquote || neednorm) {
822 r= adns__vbuf_ensure(vb, lablen+needquote+4); if (!r) R_NOMEM;
823 adns__vbuf_appendq(vb,"\"",1);
824 for (i=0, needquote=0, p= pai->dgram+labstart; i<lablen; i++, p++) {
825 c= *p;
826 if (strchr("()<>@,;:\\\".[]",c)) adns__vbuf_appendq(vb,"\\",1);
827 adns__vbuf_appendq(vb,p,1);
828 }
829 adns__vbuf_appendq(vb,"\"",1);
830 } else {
831 r= adns__vbuf_append(vb, pai->dgram+labstart, lablen); if (!r) R_NOMEM;
832 }
833
834 r= adns__vbuf_appendstr(vb,"@"); if (!r) R_NOMEM;
835
836 st= adns__parse_domain_more(&fls,pai->ads, pai->qu,vb,0, pai->dgram);
837 if (st) return st;
838
839 x_ok:
840 str= adns__alloc_interim(pai->qu, vb->used+1); if (!str) R_NOMEM;
841 memcpy(str,vb->buf,vb->used);
842 str[vb->used]= 0;
843 *mb_r= str;
844 return adns_s_ok;
845 }
846
847 /*
848 * _rp (pa)
849 */
850
851 static adns_status pa_rp(const parseinfo *pai, int cbyte, int max, void *datap) {
852 adns_rr_strpair *rrp= datap;
853 adns_status st;
854
855 st= pap_mailbox(pai, &cbyte, max, &rrp->array[0]);
856 if (st) return st;
857
858 st= pap_domain(pai, &cbyte, max, &rrp->array[1], pdf_quoteok);
859 if (st) return st;
860
861 if (cbyte != max) return adns_s_invaliddata;
862 return adns_s_ok;
863 }
864
865 /*
866 * _soa (pa,mf,cs)
867 */
868
869 static adns_status pa_soa(const parseinfo *pai, int cbyte, int max, void *datap) {
870 adns_rr_soa *rrp= datap;
871 const byte *dgram= pai->dgram;
872 adns_status st;
873 int msw, lsw, i;
874
875 st= pap_domain(pai, &cbyte, max, &rrp->mname,
876 pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
877 if (st) return st;
878
879 st= pap_mailbox(pai, &cbyte, max, &rrp->rname);
880 if (st) return st;
881
882 if (cbyte+20 != max) return adns_s_invaliddata;
883
884 for (i=0; i<5; i++) {
885 GET_W(cbyte,msw);
886 GET_W(cbyte,lsw);
887 (&rrp->serial)[i]= (msw<<16) | lsw;
888 }
889
890 return adns_s_ok;
891 }
892
893 static void mf_soa(adns_query qu, void *datap) {
894 adns_rr_soa *rrp= datap;
895
896 adns__makefinal_str(qu,&rrp->mname);
897 adns__makefinal_str(qu,&rrp->rname);
898 }
899
900 static adns_status cs_soa(vbuf *vb, const void *datap) {
901 const adns_rr_soa *rrp= datap;
902 char buf[20];
903 int i;
904 adns_status st;
905
906 st= csp_qstring(vb,rrp->mname); if (st) return st;
907 CSP_ADDSTR(" ");
908 st= csp_qstring(vb,rrp->rname); if (st) return st;
909
910 for (i=0; i<5; i++) {
911 sprintf(buf," %lu",(&rrp->serial)[i]);
912 CSP_ADDSTR(buf);
913 }
914
915 return adns_s_ok;
916 }
917
918 /*
919 * _flat (mf)
920 */
921
922 static void mf_flat(adns_query qu, void *data) { }
923
924 /*
925 * Now the table.
926 */
927
928 #define TYPESZ_M(member) (sizeof(*((adns_answer*)0)->rrs.member))
929
930 #define DEEP_MEMB(memb) TYPESZ_M(memb), mf_##memb, cs_##memb
931 #define FLAT_MEMB(memb) TYPESZ_M(memb), mf_flat, cs_##memb
932
933 #define DEEP_TYPE(code,rrt,fmt,memb,parser,comparer) \
934 { adns_r_##code, rrt, fmt, TYPESZ_M(memb), mf_##memb, cs_##memb, parser, comparer }
935 #define FLAT_TYPE(code,rrt,fmt,memb,parser,comparer) \
936 { adns_r_##code, rrt, fmt, TYPESZ_M(memb), mf_flat, cs_##memb, parser, comparer }
937
938 static const typeinfo typeinfos[] = {
939 /* Must be in ascending order of rrtype ! */
940 /* mem-mgmt code rrt fmt member parser comparer */
941
942 FLAT_TYPE(a, "A", 0, inaddr, pa_inaddr, di_inaddr ),
943 DEEP_TYPE(ns_raw, "NS", "raw", str, pa_host_raw, 0 ),
944 DEEP_TYPE(cname, "CNAME", 0, str, pa_host_raw, 0 ),
945 DEEP_TYPE(soa_raw, "SOA", "raw", soa, pa_soa, 0 ),
946 DEEP_TYPE(ptr_raw, "PTR", "raw", str, pa_host_raw, 0 ),
947 DEEP_TYPE(hinfo, "HINFO", 0, intstrpair, pa_hinfo, 0 ),
948 DEEP_TYPE(mx_raw, "MX", "raw", intstr, pa_mx_raw, di_mx_raw ),
949 DEEP_TYPE(txt, "TXT", 0, manyistr, pa_txt, 0 ),
950 DEEP_TYPE(rp_raw, "RP", "raw", strpair, pa_rp, 0 ),
951
952 FLAT_TYPE(addr, "A", "addr", addr, pa_addr, di_addr ),
953 DEEP_TYPE(ns, "NS", "+addr", hostaddr, pa_hostaddr, di_hostaddr ),
954 DEEP_TYPE(ptr, "PTR","checked", str, pa_ptr, 0 ),
955 DEEP_TYPE(mx, "MX", "+addr", inthostaddr, pa_mx, di_mx ),
956
957 DEEP_TYPE(soa, "SOA","822", soa, pa_soa, 0 ),
958 DEEP_TYPE(rp, "RP", "822", strpair, pa_rp, 0 ),
959 };
960
961 const typeinfo *adns__findtype(adns_rrtype type) {
962 const typeinfo *begin, *end, *mid;
963
964 begin= typeinfos; end= typeinfos+(sizeof(typeinfos)/sizeof(typeinfo));
965
966 while (begin < end) {
967 mid= begin + ((end-begin)>>1);
968 if (mid->type == type) return mid;
969 if (type > mid->type) begin= mid+1;
970 else end= mid;
971 }
972 return 0;
973 }