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