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