mdw@git.distorted.org.uk Git - adns/blob - src/types.c

   1 /*
   2  * types.c
   3  * - RR-type-specific code, and the machinery to call it
   4  */
   5 /*
   6  *  This file is
   7  *    Copyright (C) 1997-1999 Ian Jackson <ian@davenant.greenend.org.uk>
   8  *
   9  *  It is part of adns, which is
  10  *    Copyright (C) 1997-2000 Ian Jackson <ian@davenant.greenend.org.uk>
  11  *    Copyright (C) 1999-2000 Tony Finch <dot@dotat.at>
  12  *
  13  *  This program is free software; you can redistribute it and/or modify
  14  *  it under the terms of the GNU General Public License as published by
  15  *  the Free Software Foundation; either version 2, or (at your option)
  16  *  any later version.
  17  *
  18  *  This program is distributed in the hope that it will be useful,
  19  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  20  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21  *  GNU General Public License for more details.
  22  *
  23  *  You should have received a copy of the GNU General Public License
  24  *  along with this program; if not, write to the Free Software Foundation,
  25  *  Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  26  */
  27
  28 #include <stdlib.h>
  29
  30 #include <sys/types.h>
  31 #include <sys/socket.h>
  32 #include <netinet/in.h>
  33 #include <arpa/inet.h>
  34
  35 #include "internal.h"
  36
  37 #define R_NOMEM       return adns_s_nomemory
  38 #define CSP_ADDSTR(s) do {                      \
  39     if (!adns__vbuf_appendstr(vb,(s))) R_NOMEM; \
  40   } while (0)
  41
  42 /*
  43  * order of sections:
  44  *
  45  * _string                    (pap)
  46  * _textdata, _qstring        (csp)
  47  * _str                       (mf,cs)
  48  * _intstr                    (mf,csp,cs)
  49  * _manyistr                  (mf,cs)
  50  * _txt                       (pa)
  51  * _inaddr                    (pa,dip,di)
  52  * _addr                      (pa,di,csp,cs)
  53  * _domain                    (pap)
  54  * _host_raw                  (pa)
  55  * _hostaddr                  (pap,pa,dip,di,mfp,mf,csp,cs +pap_findaddrs)
  56  * _mx_raw                    (pa,di)
  57  * _mx                        (pa,di)
  58  * _inthostaddr               (mf,cs)
  59  * _ptr                       (pa)
  60  * _strpair                   (mf,cs)
  61  * _intstrpair                (mf,cs)
  62  * _hinfo                     (pa)
  63  * _mailbox                   (pap +pap_mailbox822)
  64  * _rp                        (pa)
  65  * _soa                       (pa,mf,cs)
  66  * _srv*                      (qdpl,(pap),pa,mf,di,(csp),cs,postsort)
  67  * _flat                      (mf)
  68  *
  69  * within each section:
  70  *    qdpl_*
  71  *    pap_*
  72  *    pa_*
  73  *    dip_*
  74  *    di_*
  75  *    mfp_*
  76  *    mf_*
  77  *    csp_*
  78  *    cs_*
  79  *    postsort_*
  80  */
  81
  82 /*
  83  * _qstring               (pap,csp)
  84  */
  85
  86 static adns_status pap_qstring(const parseinfo *pai, int *cbyte_io, int max,
  87                               int *len_r, char **str_r) {
  88   /* Neither len_r nor str_r may be null.
  89    * End of datagram (overrun) is indicated by returning adns_s_invaliddata;
  90    */
  91   const byte *dgram= pai->dgram;
  92   int l, cbyte;
  93   char *str;
  94
  95   cbyte= *cbyte_io;
  96
  97   if (cbyte >= max) return adns_s_invaliddata;
  98   GET_B(cbyte,l);
  99   if (cbyte+l > max) return adns_s_invaliddata;
 100
 101   str= adns__alloc_interim(pai->qu, l+1);
 102   if (!str) R_NOMEM;
 103
 104   str[l]= 0;
 105   memcpy(str,dgram+cbyte,l);
 106
 107   *len_r= l;
 108   *str_r= str;
 109   *cbyte_io= cbyte+l;
 110
 111   return adns_s_ok;
 112 }
 113
 114 static adns_status csp_qstring(vbuf *vb, const char *dp, int len) {
 115   unsigned char ch;
 116   char buf[10];
 117   int cn;
 118
 119   CSP_ADDSTR("\"");
 120   for (cn=0; cn<len; cn++) {
 121     ch= *dp++;
 122     if (ch == '\\') {
 123       CSP_ADDSTR("\\\\");
 124     } else if (ch == '"') {
 125       CSP_ADDSTR("\\\"");
 126     } else if (ch >= 32 && ch <= 126) {
 127       if (!adns__vbuf_append(vb,&ch,1)) R_NOMEM;
 128     } else {
 129       sprintf(buf,"\\x%02x",ch);
 130       CSP_ADDSTR(buf);
 131     }
 132   }
 133   CSP_ADDSTR("\"");
 134
 135   return adns_s_ok;
 136 }
 137
 138 /*
 139  * _str  (mf)
 140  */
 141
 142 static void mf_str(adns_query qu, void *datap) {
 143   char **rrp= datap;
 144
 145   adns__makefinal_str(qu,rrp);
 146 }
 147
 148 /*
 149  * _intstr  (mf)
 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
 158 /*
 159  * _manyistr   (mf)
 160  */
 161
 162 static void mf_manyistr(adns_query qu, void *datap) {
 163   adns_rr_intstr **rrp= datap;
 164   adns_rr_intstr *te, *table;
 165   void *tablev;
 166   int tc;
 167
 168   for (tc=0, te= *rrp; te->i >= 0; te++, tc++);
 169   tablev= *rrp;
 170   adns__makefinal_block(qu,&tablev,sizeof(*te)*(tc+1));
 171   *rrp= table= tablev;
 172   for (te= *rrp; te->i >= 0; te++)
 173     adns__makefinal_str(qu,&te->str);
 174 }
 175
 176 /*
 177  * _txt   (pa,cs)
 178  */
 179
 180 static adns_status pa_txt(const parseinfo *pai, int cbyte,
 181                           int max, void *datap) {
 182   adns_rr_intstr **rrp= datap, *table, *te;
 183   const byte *dgram= pai->dgram;
 184   int ti, tc, l, startbyte;
 185   adns_status st;
 186
 187   startbyte= cbyte;
 188   if (cbyte >= max) return adns_s_invaliddata;
 189   tc= 0;
 190   while (cbyte < max) {
 191     GET_B(cbyte,l);
 192     cbyte+= l;
 193     tc++;
 194   }
 195   if (cbyte != max || !tc) return adns_s_invaliddata;
 196
 197   table= adns__alloc_interim(pai->qu,sizeof(*table)*(tc+1));
 198   if (!table) R_NOMEM;
 199
 200   for (cbyte=startbyte, ti=0, te=table; ti<tc; ti++, te++) {
 201     st= pap_qstring(pai, &cbyte, max, &te->i, &te->str);
 202     if (st) return st;
 203   }
 204   assert(cbyte == max);
 205
 206   te->i= -1;
 207   te->str= 0;
 208
 209   *rrp= table;
 210   return adns_s_ok;
 211 }
 212
 213 static adns_status cs_txt(vbuf *vb, const void *datap) {
 214   const adns_rr_intstr *const *rrp= datap;
 215   const adns_rr_intstr *current;
 216   adns_status st;
 217   int spc;
 218
 219   for (current= *rrp, spc=0;  current->i >= 0;  current++, spc=1) {
 220     if (spc) CSP_ADDSTR(" ");
 221     st= csp_qstring(vb,current->str,current->i); if (st) return st;
 222   }
 223   return adns_s_ok;
 224 }
 225
 226 /*
 227  * _hinfo   (cs)
 228  */
 229
 230 static adns_status cs_hinfo(vbuf *vb, const void *datap) {
 231   const adns_rr_intstrpair *rrp= datap;
 232   adns_status st;
 233
 234   st= csp_qstring(vb,rrp->array[0].str,rrp->array[0].i);  if (st) return st;
 235   CSP_ADDSTR(" ");
 236   st= csp_qstring(vb,rrp->array[1].str,rrp->array[1].i);  if (st) return st;
 237   return adns_s_ok;
 238 }
 239
 240 /*
 241  * _inaddr   (pa,dip,di)
 242  */
 243
 244 static adns_status pa_inaddr(const parseinfo *pai, int cbyte,
 245                              int max, void *datap) {
 246   struct in_addr *storeto= datap;
 247
 248   if (max-cbyte != 4) return adns_s_invaliddata;
 249   memcpy(storeto, pai->dgram + cbyte, 4);
 250   return adns_s_ok;
 251 }
 252
 253 static int search_sortlist(adns_state ads, struct in_addr ad) {
 254   const struct sortlist *slp;
 255   int i;
 256
 257   for (i=0, slp=ads->sortlist;
 258        i<ads->nsortlist &&
 259          !((ad.s_addr & slp->mask.s_addr) == slp->base.s_addr);
 260        i++, slp++);
 261   return i;
 262 }
 263
 264 static int dip_inaddr(adns_state ads, struct in_addr a, struct in_addr b) {
 265   int ai, bi;
 266
 267   if (!ads->nsortlist) return 0;
 268
 269   ai= search_sortlist(ads,a);
 270   bi= search_sortlist(ads,b);
 271   return bi<ai;
 272 }
 273
 274 static int di_inaddr(adns_state ads,
 275                      const void *datap_a, const void *datap_b) {
 276   const struct in_addr *ap= datap_a, *bp= datap_b;
 277
 278   return dip_inaddr(ads,*ap,*bp);
 279 }
 280
 281 static adns_status cs_inaddr(vbuf *vb, const void *datap) {
 282   const struct in_addr *rrp= datap, rr= *rrp;
 283   const char *ia;
 284
 285   ia= inet_ntoa(rr); assert(ia);
 286   CSP_ADDSTR(ia);
 287   return adns_s_ok;
 288 }
 289
 290 /*
 291  * _addr   (pa,di,csp,cs)
 292  */
 293
 294 static adns_status pa_addr(const parseinfo *pai, int cbyte,
 295                            int max, void *datap) {
 296   adns_rr_addr *storeto= datap;
 297   const byte *dgram= pai->dgram;
 298
 299   if (max-cbyte != 4) return adns_s_invaliddata;
 300   storeto->len= sizeof(storeto->addr.inet);
 301   memset(&storeto->addr,0,sizeof(storeto->addr.inet));
 302   storeto->addr.inet.sin_family= AF_INET;
 303   memcpy(&storeto->addr.inet.sin_addr,dgram+cbyte,4);
 304   return adns_s_ok;
 305 }
 306
 307 static int di_addr(adns_state ads, const void *datap_a, const void *datap_b) {
 308   const adns_rr_addr *ap= datap_a, *bp= datap_b;
 309
 310   assert(ap->addr.sa.sa_family == AF_INET);
 311   return dip_inaddr(ads, ap->addr.inet.sin_addr, bp->addr.inet.sin_addr);
 312 }
 313
 314 static int div_addr(void *context, const void *datap_a, const void *datap_b) {
 315   const adns_state ads= context;
 316
 317   return di_addr(ads, datap_a, datap_b);
 318 }
 319
 320 static adns_status csp_addr(vbuf *vb, const adns_rr_addr *rrp) {
 321   const char *ia;
 322   static char buf[30];
 323
 324   switch (rrp->addr.inet.sin_family) {
 325   case AF_INET:
 326     CSP_ADDSTR("INET ");
 327     ia= inet_ntoa(rrp->addr.inet.sin_addr); assert(ia);
 328     CSP_ADDSTR(ia);
 329     break;
 330   default:
 331     sprintf(buf,"AF=%u",rrp->addr.sa.sa_family);
 332     CSP_ADDSTR(buf);
 333     break;
 334   }
 335   return adns_s_ok;
 336 }
 337
 338 static adns_status cs_addr(vbuf *vb, const void *datap) {
 339   const adns_rr_addr *rrp= datap;
 340
 341   return csp_addr(vb,rrp);
 342 }
 343
 344 /*
 345  * _domain      (pap,csp,cs)
 346  * _dom_raw     (pa)
 347  */
 348
 349 static adns_status pap_domain(const parseinfo *pai, int *cbyte_io, int max,
 350                               char **domain_r, parsedomain_flags flags) {
 351   adns_status st;
 352   char *dm;
 353
 354   st= adns__parse_domain(pai->qu->ads, pai->serv, pai->qu, &pai->qu->vb, flags,
 355                          pai->dgram,pai->dglen, cbyte_io, max);
 356   if (st) return st;
 357   if (!pai->qu->vb.used) return adns_s_invaliddata;
 358
 359   dm= adns__alloc_interim(pai->qu, pai->qu->vb.used+1);
 360   if (!dm) R_NOMEM;
 361
 362   dm[pai->qu->vb.used]= 0;
 363   memcpy(dm,pai->qu->vb.buf,pai->qu->vb.used);
 364
 365   *domain_r= dm;
 366   return adns_s_ok;
 367 }
 368
 369 static adns_status csp_domain(vbuf *vb, const char *domain) {
 370   CSP_ADDSTR(domain);
 371   if (!*domain) CSP_ADDSTR(".");
 372   return adns_s_ok;
 373 }
 374
 375 static adns_status cs_domain(vbuf *vb, const void *datap) {
 376   const char *const *domainp= datap;
 377   return csp_domain(vb,*domainp);
 378 }
 379
 380 static adns_status pa_dom_raw(const parseinfo *pai, int cbyte,
 381                               int max, void *datap) {
 382   char **rrp= datap;
 383   adns_status st;
 384
 385   st= pap_domain(pai, &cbyte, max, rrp, pdf_quoteok);
 386   if (st) return st;
 387
 388   if (cbyte != max) return adns_s_invaliddata;
 389   return adns_s_ok;
 390 }
 391
 392 /*
 393  * _host_raw   (pa)
 394  */
 395
 396 static adns_status pa_host_raw(const parseinfo *pai, int cbyte,
 397                                int max, void *datap) {
 398   char **rrp= datap;
 399   adns_status st;
 400
 401   st= pap_domain(pai, &cbyte, max, rrp,
 402                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
 403   if (st) return st;
 404
 405   if (cbyte != max) return adns_s_invaliddata;
 406   return adns_s_ok;
 407 }
 408
 409 /*
 410  * _hostaddr   (pap,pa,dip,di,mfp,mf,csp,cs +icb_hostaddr, pap_findaddrs)
 411  */
 412
 413 static adns_status pap_findaddrs(const parseinfo *pai, adns_rr_hostaddr *ha,
 414                                  int *cbyte_io, int count, int dmstart) {
 415   int rri, naddrs;
 416   int type, class, rdlen, rdstart, ownermatched;
 417   unsigned long ttl;
 418   adns_status st;
 419
 420   for (rri=0, naddrs=-1; rri<count; rri++) {
 421     st= adns__findrr_anychk(pai->qu, pai->serv, pai->dgram,
 422                             pai->dglen, cbyte_io,
 423                             &type, &class, &ttl, &rdlen, &rdstart,
 424                             pai->dgram, pai->dglen, dmstart, &ownermatched);
 425     if (st) return st;
 426     if (!ownermatched || class != DNS_CLASS_IN || type != adns_r_a) {
 427       if (naddrs>0) break; else continue;
 428     }
 429     if (naddrs == -1) {
 430       naddrs= 0;
 431     }
 432     if (!adns__vbuf_ensure(&pai->qu->vb, (naddrs+1)*sizeof(adns_rr_addr)))
 433       R_NOMEM;
 434     adns__update_expires(pai->qu,ttl,pai->now);
 435     st= pa_addr(pai, rdstart,rdstart+rdlen,
 436                 pai->qu->vb.buf + naddrs*sizeof(adns_rr_addr));
 437     if (st) return st;
 438     naddrs++;
 439   }
 440   if (naddrs >= 0) {
 441     ha->addrs= adns__alloc_interim(pai->qu, naddrs*sizeof(adns_rr_addr));
 442     if (!ha->addrs) R_NOMEM;
 443     memcpy(ha->addrs, pai->qu->vb.buf, naddrs*sizeof(adns_rr_addr));
 444     ha->naddrs= naddrs;
 445     ha->astatus= adns_s_ok;
 446
 447     adns__isort(ha->addrs, naddrs, sizeof(adns_rr_addr), pai->qu->vb.buf,
 448                 div_addr, pai->ads);
 449   }
 450   return adns_s_ok;
 451 }
 452
 453 static void icb_hostaddr(adns_query parent, adns_query child) {
 454   adns_answer *cans= child->answer;
 455   adns_rr_hostaddr *rrp= child->ctx.info.hostaddr;
 456   adns_state ads= parent->ads;
 457   adns_status st;
 458
 459   st= cans->status;
 460   rrp->astatus= st;
 461   rrp->naddrs= (st>0 && st<=adns_s_max_tempfail) ? -1 : cans->nrrs;
 462   rrp->addrs= cans->rrs.addr;
 463   adns__transfer_interim(child, parent, rrp->addrs,
 464                          rrp->naddrs*sizeof(adns_rr_addr));
 465
 466   if (parent->children.head) {
 467     LIST_LINK_TAIL(ads->childw,parent);
 468   } else {
 469     adns__query_done(parent);
 470   }
 471 }
 472
 473 static adns_status pap_hostaddr(const parseinfo *pai, int *cbyte_io,
 474                                 int max, adns_rr_hostaddr *rrp) {
 475   adns_status st;
 476   int dmstart, cbyte;
 477   qcontext ctx;
 478   int id;
 479   adns_query nqu;
 480   adns_queryflags nflags;
 481
 482   dmstart= cbyte= *cbyte_io;
 483   st= pap_domain(pai, &cbyte, max, &rrp->host,
 484                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
 485   if (st) return st;
 486   *cbyte_io= cbyte;
 487
 488   rrp->astatus= adns_s_ok;
 489   rrp->naddrs= -1;
 490   rrp->addrs= 0;
 491
 492   cbyte= pai->nsstart;
 493
 494   st= pap_findaddrs(pai, rrp, &cbyte, pai->nscount, dmstart);
 495   if (st) return st;
 496   if (rrp->naddrs != -1) return adns_s_ok;
 497
 498   st= pap_findaddrs(pai, rrp, &cbyte, pai->arcount, dmstart);
 499   if (st) return st;
 500   if (rrp->naddrs != -1) return adns_s_ok;
 501
 502   st= adns__mkquery_frdgram(pai->ads, &pai->qu->vb, &id,
 503                             pai->dgram, pai->dglen, dmstart,
 504                             adns_r_addr, adns_qf_quoteok_query);
 505   if (st) return st;
 506
 507   ctx.ext= 0;
 508   ctx.callback= icb_hostaddr;
 509   ctx.info.hostaddr= rrp;
 510
 511   nflags= adns_qf_quoteok_query;
 512   if (!(pai->qu->flags & adns_qf_cname_loose)) nflags |= adns_qf_cname_forbid;
 513
 514   st= adns__internal_submit(pai->ads, &nqu, adns__findtype(adns_r_addr),
 515                             &pai->qu->vb, id, nflags, pai->now, &ctx);
 516   if (st) return st;
 517
 518   nqu->parent= pai->qu;
 519   LIST_LINK_TAIL_PART(pai->qu->children,nqu,siblings.);
 520
 521   return adns_s_ok;
 522 }
 523
 524 static adns_status pa_hostaddr(const parseinfo *pai, int cbyte,
 525                                int max, void *datap) {
 526   adns_rr_hostaddr *rrp= datap;
 527   adns_status st;
 528
 529   st= pap_hostaddr(pai, &cbyte, max, rrp);
 530   if (st) return st;
 531   if (cbyte != max) return adns_s_invaliddata;
 532
 533   return adns_s_ok;
 534 }
 535
 536 static int dip_hostaddr(adns_state ads,
 537                         const adns_rr_hostaddr *ap, const adns_rr_hostaddr *bp) {
 538   if (ap->astatus != bp->astatus) return ap->astatus;
 539   if (ap->astatus) return 0;
 540
 541   assert(ap->addrs[0].addr.sa.sa_family == AF_INET);
 542   assert(bp->addrs[0].addr.sa.sa_family == AF_INET);
 543   return dip_inaddr(ads,
 544                     ap->addrs[0].addr.inet.sin_addr,
 545                     bp->addrs[0].addr.inet.sin_addr);
 546 }
 547
 548 static int di_hostaddr(adns_state ads,
 549                        const void *datap_a, const void *datap_b) {
 550   const adns_rr_hostaddr *ap= datap_a, *bp= datap_b;
 551
 552   return dip_hostaddr(ads, ap,bp);
 553 }
 554
 555 static void mfp_hostaddr(adns_query qu, adns_rr_hostaddr *rrp) {
 556   void *tablev;
 557
 558   adns__makefinal_str(qu,&rrp->host);
 559   tablev= rrp->addrs;
 560   adns__makefinal_block(qu, &tablev, rrp->naddrs*sizeof(*rrp->addrs));
 561   rrp->addrs= tablev;
 562 }
 563
 564 static void mf_hostaddr(adns_query qu, void *datap) {
 565   adns_rr_hostaddr *rrp= datap;
 566
 567   mfp_hostaddr(qu,rrp);
 568 }
 569
 570 static adns_status csp_hostaddr(vbuf *vb, const adns_rr_hostaddr *rrp) {
 571   const char *errstr;
 572   adns_status st;
 573   char buf[20];
 574   int i;
 575
 576   st= csp_domain(vb,rrp->host);  if (st) return st;
 577
 578   CSP_ADDSTR(" ");
 579   CSP_ADDSTR(adns_errtypeabbrev(rrp->astatus));
 580
 581   sprintf(buf," %d ",rrp->astatus);
 582   CSP_ADDSTR(buf);
 583
 584   CSP_ADDSTR(adns_errabbrev(rrp->astatus));
 585   CSP_ADDSTR(" ");
 586
 587   errstr= adns_strerror(rrp->astatus);
 588   st= csp_qstring(vb,errstr,strlen(errstr));  if (st) return st;
 589
 590   if (rrp->naddrs >= 0) {
 591     CSP_ADDSTR(" (");
 592     for (i=0; i<rrp->naddrs; i++) {
 593       CSP_ADDSTR(" ");
 594       st= csp_addr(vb,&rrp->addrs[i]);
 595     }
 596     CSP_ADDSTR(" )");
 597   } else {
 598     CSP_ADDSTR(" ?");
 599   }
 600   return adns_s_ok;
 601 }
 602
 603 static adns_status cs_hostaddr(vbuf *vb, const void *datap) {
 604   const adns_rr_hostaddr *rrp= datap;
 605
 606   return csp_hostaddr(vb,rrp);
 607 }
 608
 609 /*
 610  * _mx_raw   (pa,di)
 611  */
 612
 613 static adns_status pa_mx_raw(const parseinfo *pai, int cbyte,
 614                              int max, void *datap) {
 615   const byte *dgram= pai->dgram;
 616   adns_rr_intstr *rrp= datap;
 617   adns_status st;
 618   int pref;
 619
 620   if (cbyte+2 > max) return adns_s_invaliddata;
 621   GET_W(cbyte,pref);
 622   rrp->i= pref;
 623   st= pap_domain(pai, &cbyte, max, &rrp->str,
 624                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
 625   if (st) return st;
 626
 627   if (cbyte != max) return adns_s_invaliddata;
 628   return adns_s_ok;
 629 }
 630
 631 static int di_mx_raw(adns_state ads, const void *datap_a, const void *datap_b) {
 632   const adns_rr_intstr *ap= datap_a, *bp= datap_b;
 633
 634   if (ap->i < bp->i) return 0;
 635   if (ap->i > bp->i) return 1;
 636   return 0;
 637 }
 638
 639 /*
 640  * _mx   (pa,di)
 641  */
 642
 643 static adns_status pa_mx(const parseinfo *pai, int cbyte,
 644                          int max, void *datap) {
 645   const byte *dgram= pai->dgram;
 646   adns_rr_inthostaddr *rrp= datap;
 647   adns_status st;
 648   int pref;
 649
 650   if (cbyte+2 > max) return adns_s_invaliddata;
 651   GET_W(cbyte,pref);
 652   rrp->i= pref;
 653   st= pap_hostaddr(pai, &cbyte, max, &rrp->ha);
 654   if (st) return st;
 655
 656   if (cbyte != max) return adns_s_invaliddata;
 657   return adns_s_ok;
 658 }
 659
 660 static int di_mx(adns_state ads, const void *datap_a, const void *datap_b) {
 661   const adns_rr_inthostaddr *ap= datap_a, *bp= datap_b;
 662
 663   if (ap->i < bp->i) return 0;
 664   if (ap->i > bp->i) return 1;
 665   return dip_hostaddr(ads, &ap->ha, &bp->ha);
 666 }
 667
 668 /*
 669  * _inthostaddr  (mf,cs)
 670  */
 671
 672 static void mf_inthostaddr(adns_query qu, void *datap) {
 673   adns_rr_inthostaddr *rrp= datap;
 674
 675   mfp_hostaddr(qu,&rrp->ha);
 676 }
 677
 678 static adns_status cs_inthostaddr(vbuf *vb, const void *datap) {
 679   const adns_rr_inthostaddr *rrp= datap;
 680   char buf[10];
 681
 682   sprintf(buf,"%u ",rrp->i);
 683   CSP_ADDSTR(buf);
 684
 685   return csp_hostaddr(vb,&rrp->ha);
 686 }
 687
 688 /*
 689  * _inthost  (cs)
 690  */
 691
 692 static adns_status cs_inthost(vbuf *vb, const void *datap) {
 693   const adns_rr_intstr *rrp= datap;
 694   char buf[10];
 695
 696   sprintf(buf,"%u ",rrp->i);
 697   CSP_ADDSTR(buf);
 698   return csp_domain(vb,rrp->str);
 699 }
 700
 701 /*
 702  * _ptr   (pa, +icb_ptr)
 703  */
 704
 705 static void icb_ptr(adns_query parent, adns_query child) {
 706   adns_answer *cans= child->answer;
 707   const adns_rr_addr *queried, *found;
 708   adns_state ads= parent->ads;
 709   int i;
 710
 711   if (cans->status == adns_s_nxdomain || cans->status == adns_s_nodata) {
 712     adns__query_fail(parent,adns_s_inconsistent);
 713     return;
 714   } else if (cans->status) {
 715     adns__query_fail(parent,cans->status);
 716     return;
 717   }
 718
 719   queried= &parent->ctx.info.ptr_parent_addr;
 720   for (i=0, found=cans->rrs.addr; i<cans->nrrs; i++, found++) {
 721     if (queried->len == found->len &&
 722         !memcmp(&queried->addr,&found->addr,queried->len)) {
 723       if (!parent->children.head) {
 724         adns__query_done(parent);
 725         return;
 726       } else {
 727         LIST_LINK_TAIL(ads->childw,parent);
 728         return;
 729       }
 730     }
 731   }
 732
 733   adns__query_fail(parent,adns_s_inconsistent);
 734 }
 735
 736 static adns_status pa_ptr(const parseinfo *pai, int dmstart,
 737                           int max, void *datap) {
 738   static const char *(expectdomain[])= { DNS_INADDR_ARPA };
 739
 740   char **rrp= datap;
 741   adns_status st;
 742   adns_rr_addr *ap;
 743   findlabel_state fls;
 744   char *ep;
 745   byte ipv[4];
 746   char labbuf[4];
 747   int cbyte, i, lablen, labstart, l, id;
 748   adns_query nqu;
 749   qcontext ctx;
 750
 751   cbyte= dmstart;
 752   st= pap_domain(pai, &cbyte, max, rrp,
 753                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
 754   if (st) return st;
 755   if (cbyte != max) return adns_s_invaliddata;
 756
 757   ap= &pai->qu->ctx.info.ptr_parent_addr;
 758   if (!ap->len) {
 759     adns__findlabel_start(&fls, pai->ads, -1, pai->qu,
 760                           pai->qu->query_dgram, pai->qu->query_dglen,
 761                           pai->qu->query_dglen, DNS_HDRSIZE, 0);
 762     for (i=0; i<4; i++) {
 763       st= adns__findlabel_next(&fls,&lablen,&labstart); assert(!st);
 764       if (lablen<=0 || lablen>3) return adns_s_querydomainwrong;
 765       memcpy(labbuf, pai->qu->query_dgram + labstart, lablen);
 766       labbuf[lablen]= 0;
 767       ipv[3-i]= strtoul(labbuf,&ep,10);
 768       if (*ep) return adns_s_querydomainwrong;
 769       if (lablen>1 && pai->qu->query_dgram[labstart]=='0')
 770         return adns_s_querydomainwrong;
 771     }
 772     for (i=0; i<sizeof(expectdomain)/sizeof(*expectdomain); i++) {
 773       st= adns__findlabel_next(&fls,&lablen,&labstart); assert(!st);
 774       l= strlen(expectdomain[i]);
 775       if (lablen != l ||
 776           memcmp(pai->qu->query_dgram + labstart, expectdomain[i], l))
 777         return adns_s_querydomainwrong;
 778     }
 779     st= adns__findlabel_next(&fls,&lablen,0); assert(!st);
 780     if (lablen) return adns_s_querydomainwrong;
 781
 782     ap->len= sizeof(struct sockaddr_in);
 783     memset(&ap->addr,0,sizeof(ap->addr.inet));
 784     ap->addr.inet.sin_family= AF_INET;
 785     ap->addr.inet.sin_addr.s_addr=
 786       htonl((ipv[0]<<24) | (ipv[1]<<16) | (ipv[2]<<8) | (ipv[3]));
 787   }
 788
 789   st= adns__mkquery_frdgram(pai->ads, &pai->qu->vb, &id,
 790                             pai->dgram, pai->dglen, dmstart,
 791                             adns_r_addr, adns_qf_quoteok_query);
 792   if (st) return st;
 793
 794   ctx.ext= 0;
 795   ctx.callback= icb_ptr;
 796   memset(&ctx.info,0,sizeof(ctx.info));
 797   st= adns__internal_submit(pai->ads, &nqu, adns__findtype(adns_r_addr),
 798                             &pai->qu->vb, id,
 799                             adns_qf_quoteok_query, pai->now, &ctx);
 800   if (st) return st;
 801
 802   nqu->parent= pai->qu;
 803   LIST_LINK_TAIL_PART(pai->qu->children,nqu,siblings.);
 804   return adns_s_ok;
 805 }
 806
 807 /*
 808  * _strpair   (mf)
 809  */
 810
 811 static void mf_strpair(adns_query qu, void *datap) {
 812   adns_rr_strpair *rrp= datap;
 813
 814   adns__makefinal_str(qu,&rrp->array[0]);
 815   adns__makefinal_str(qu,&rrp->array[1]);
 816 }
 817
 818 /*
 819  * _intstrpair   (mf)
 820  */
 821
 822 static void mf_intstrpair(adns_query qu, void *datap) {
 823   adns_rr_intstrpair *rrp= datap;
 824
 825   adns__makefinal_str(qu,&rrp->array[0].str);
 826   adns__makefinal_str(qu,&rrp->array[1].str);
 827 }
 828
 829 /*
 830  * _hinfo   (pa)
 831  */
 832
 833 static adns_status pa_hinfo(const parseinfo *pai, int cbyte,
 834                             int max, void *datap) {
 835   adns_rr_intstrpair *rrp= datap;
 836   adns_status st;
 837   int i;
 838
 839   for (i=0; i<2; i++) {
 840     st= pap_qstring(pai, &cbyte, max, &rrp->array[i].i, &rrp->array[i].str);
 841     if (st) return st;
 842   }
 843
 844   if (cbyte != max) return adns_s_invaliddata;
 845
 846   return adns_s_ok;
 847 }
 848
 849 /*
 850  * _mailbox   (pap,cs)
 851  */
 852
 853 static adns_status pap_mailbox822(const parseinfo *pai,
 854                                   int *cbyte_io, int max, char **mb_r) {
 855   int lablen, labstart, i, needquote, c, r, neednorm;
 856   const unsigned char *p;
 857   char *str;
 858   findlabel_state fls;
 859   adns_status st;
 860   vbuf *vb;
 861
 862   vb= &pai->qu->vb;
 863   vb->used= 0;
 864   adns__findlabel_start(&fls, pai->ads,
 865                         -1, pai->qu,
 866                         pai->dgram, pai->dglen, max,
 867                         *cbyte_io, cbyte_io);
 868   st= adns__findlabel_next(&fls,&lablen,&labstart);
 869   if (!lablen) {
 870     adns__vbuf_appendstr(vb,".");
 871     goto x_ok;
 872   }
 873
 874   neednorm= 1;
 875   for (i=0, needquote=0, p= pai->dgram+labstart; i<lablen; i++) {
 876     c= *p++;
 877     if ((c&~128) < 32 || (c&~128) == 127) return adns_s_invaliddata;
 878     if (c == '.' && !neednorm) neednorm= 1;
 879     else if (c==' ' || c>=127 || ctype_822special(c)) needquote++;
 880     else neednorm= 0;
 881   }
 882
 883   if (needquote || neednorm) {
 884     r= adns__vbuf_ensure(vb, lablen+needquote+4); if (!r) R_NOMEM;
 885     adns__vbuf_appendq(vb,"\"",1);
 886     for (i=0, needquote=0, p= pai->dgram+labstart; i<lablen; i++, p++) {
 887       c= *p;
 888       if (c == '"' || c=='\\') adns__vbuf_appendq(vb,"\\",1);
 889       adns__vbuf_appendq(vb,p,1);
 890     }
 891     adns__vbuf_appendq(vb,"\"",1);
 892   } else {
 893     r= adns__vbuf_append(vb, pai->dgram+labstart, lablen); if (!r) R_NOMEM;
 894   }
 895
 896   r= adns__vbuf_appendstr(vb,"@"); if (!r) R_NOMEM;
 897
 898   st= adns__parse_domain_more(&fls,pai->ads, pai->qu,vb,0, pai->dgram);
 899   if (st) return st;
 900
 901  x_ok:
 902   str= adns__alloc_interim(pai->qu, vb->used+1); if (!str) R_NOMEM;
 903   memcpy(str,vb->buf,vb->used);
 904   str[vb->used]= 0;
 905   *mb_r= str;
 906   return adns_s_ok;
 907 }
 908
 909 static adns_status pap_mailbox(const parseinfo *pai, int *cbyte_io, int max,
 910                                char **mb_r) {
 911   if (pai->qu->typei->type & adns__qtf_mail822) {
 912     return pap_mailbox822(pai, cbyte_io, max, mb_r);
 913   } else {
 914     return pap_domain(pai, cbyte_io, max, mb_r, pdf_quoteok);
 915   }
 916 }
 917
 918 static adns_status csp_mailbox(vbuf *vb, const char *mailbox) {
 919   return csp_domain(vb,mailbox);
 920 }
 921
 922 /*
 923  * _rp   (pa,cs)
 924  */
 925
 926 static adns_status pa_rp(const parseinfo *pai, int cbyte,
 927                          int max, void *datap) {
 928   adns_rr_strpair *rrp= datap;
 929   adns_status st;
 930
 931   st= pap_mailbox(pai, &cbyte, max, &rrp->array[0]);
 932   if (st) return st;
 933
 934   st= pap_domain(pai, &cbyte, max, &rrp->array[1], pdf_quoteok);
 935   if (st) return st;
 936
 937   if (cbyte != max) return adns_s_invaliddata;
 938   return adns_s_ok;
 939 }
 940
 941 static adns_status cs_rp(vbuf *vb, const void *datap) {
 942   const adns_rr_strpair *rrp= datap;
 943   adns_status st;
 944
 945   st= csp_mailbox(vb,rrp->array[0]);  if (st) return st;
 946   CSP_ADDSTR(" ");
 947   st= csp_domain(vb,rrp->array[1]);  if (st) return st;
 948
 949   return adns_s_ok;
 950 }
 951
 952 /*
 953  * _soa   (pa,mf,cs)
 954  */
 955
 956 static adns_status pa_soa(const parseinfo *pai, int cbyte,
 957                           int max, void *datap) {
 958   adns_rr_soa *rrp= datap;
 959   const byte *dgram= pai->dgram;
 960   adns_status st;
 961   int msw, lsw, i;
 962
 963   st= pap_domain(pai, &cbyte, max, &rrp->mname,
 964                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
 965   if (st) return st;
 966
 967   st= pap_mailbox(pai, &cbyte, max, &rrp->rname);
 968   if (st) return st;
 969
 970   if (cbyte+20 != max) return adns_s_invaliddata;
 971
 972   for (i=0; i<5; i++) {
 973     GET_W(cbyte,msw);
 974     GET_W(cbyte,lsw);
 975     (&rrp->serial)[i]= (msw<<16) | lsw;
 976   }
 977
 978   return adns_s_ok;
 979 }
 980
 981 static void mf_soa(adns_query qu, void *datap) {
 982   adns_rr_soa *rrp= datap;
 983
 984   adns__makefinal_str(qu,&rrp->mname);
 985   adns__makefinal_str(qu,&rrp->rname);
 986 }
 987
 988 static adns_status cs_soa(vbuf *vb, const void *datap) {
 989   const adns_rr_soa *rrp= datap;
 990   char buf[20];
 991   int i;
 992   adns_status st;
 993
 994   st= csp_domain(vb,rrp->mname);  if (st) return st;
 995   CSP_ADDSTR(" ");
 996   st= csp_mailbox(vb,rrp->rname);  if (st) return st;
 997
 998   for (i=0; i<5; i++) {
 999     sprintf(buf," %lu",(&rrp->serial)[i]);
1000     CSP_ADDSTR(buf);
1001   }
1002
1003   return adns_s_ok;
1004 }
1005
1006 /*
1007  * _srv*  (pa*2,di,cs*2,qdpl,postsort)
1008  */
1009
1010 static adns_status qdpl_srv(adns_state ads,
1011                             const char **p_io, const char *pe, int labelnum,
1012                             char label_r[DNS_MAXDOMAIN], int *ll_io,
1013                             adns_queryflags flags,
1014                             const typeinfo *typei) {
1015   int useflags;
1016   const char *p_orig;
1017   adns_status st;
1018
1019   if (labelnum < 2 && !(flags & adns_qf_quoteok_query)) {
1020     useflags= adns_qf_quoteok_query;
1021     p_orig= *p_io;
1022   } else {
1023     useflags= flags;
1024     p_orig= 0;
1025   }
1026   st= adns__qdpl_normal(ads, p_io,pe, labelnum,label_r, ll_io, useflags,typei);
1027   if (st) return st;
1028
1029   if (p_orig) {
1030     int ll= *ll_io;
1031     if (!ll || label_r[0]!='_')
1032       return adns_s_querydomaininvalid;
1033     if (memchr(p_orig+1, '\\', pe - (p_orig+1)))
1034       return adns_s_querydomaininvalid;
1035   }
1036   return adns_s_ok;
1037 }
1038
1039 static adns_status pap_srv_begin(const parseinfo *pai, int *cbyte_io, int max,
1040                                  adns_rr_srvraw *rrp
1041                                    /* might be adns_rr_srvha* */) {
1042   const byte *dgram= pai->dgram;
1043   int ti, cbyte;
1044
1045   cbyte= *cbyte_io;
1046   if ((*cbyte_io += 6) > max) return adns_s_invaliddata;
1047
1048   rrp->priority= GET_W(cbyte, ti);
1049   rrp->weight=   GET_W(cbyte, ti);
1050   rrp->port=     GET_W(cbyte, ti);
1051   return adns_s_ok;
1052 }
1053
1054 static adns_status pa_srvraw(const parseinfo *pai, int cbyte,
1055                              int max, void *datap) {
1056   adns_rr_srvraw *rrp= datap;
1057   adns_status st;
1058
1059   st= pap_srv_begin(pai,&cbyte,max,datap);
1060   if (st) return st;
1061
1062   st= pap_domain(pai, &cbyte, max, &rrp->host,
1063                  pai->qu->flags & adns_qf_quoteok_anshost ? pdf_quoteok : 0);
1064   if (st) return st;
1065
1066   if (cbyte != max) return adns_s_invaliddata;
1067   return adns_s_ok;
1068 }
1069
1070 static adns_status pa_srvha(const parseinfo *pai, int cbyte,
1071                             int max, void *datap) {
1072   adns_rr_srvha *rrp= datap;
1073   adns_status st;
1074
1075   st= pap_srv_begin(pai,&cbyte,max,datap);       if (st) return st;
1076   st= pap_hostaddr(pai, &cbyte, max, &rrp->ha);  if (st) return st;
1077   if (cbyte != max) return adns_s_invaliddata;
1078   return adns_s_ok;
1079 }
1080
1081 static void mf_srvraw(adns_query qu, void *datap) {
1082   adns_rr_srvraw *rrp= datap;
1083   adns__makefinal_str(qu, &rrp->host);
1084 }
1085
1086 static void mf_srvha(adns_query qu, void *datap) {
1087   adns_rr_srvha *rrp= datap;
1088   mfp_hostaddr(qu,&rrp->ha);
1089 }
1090
1091 static int di_srv(adns_state ads, const void *datap_a, const void *datap_b) {
1092   const adns_rr_srvraw *ap= datap_a, *bp= datap_b;
1093     /* might be const adns_rr_svhostaddr* */
1094
1095   if (ap->priority < bp->priority) return 0;
1096   if (ap->priority > bp->priority) return 1;
1097   return 0;
1098 }
1099
1100 static adns_status csp_srv_begin(vbuf *vb, const adns_rr_srvraw *rrp
1101                                    /* might be adns_rr_srvha* */) {
1102   char buf[30];
1103   sprintf(buf,"%u %u %u ", rrp->priority, rrp->weight, rrp->port);
1104   CSP_ADDSTR(buf);
1105   return adns_s_ok;
1106 }
1107
1108 static adns_status cs_srvraw(vbuf *vb, const void *datap) {
1109   const adns_rr_srvraw *rrp= datap;
1110   adns_status st;
1111
1112   st= csp_srv_begin(vb,rrp);  if (st) return st;
1113   return csp_domain(vb,rrp->host);
1114 }
1115
1116 static adns_status cs_srvha(vbuf *vb, const void *datap) {
1117   const adns_rr_srvha *rrp= datap;
1118   adns_status st;
1119
1120   st= csp_srv_begin(vb,datap);  if (st) return st;
1121   return csp_hostaddr(vb,&rrp->ha);
1122 }
1123
1124 static void postsort_srv(adns_state ads, void *array, int nobjs,
1125                          const struct typeinfo *typei) {
1126   fprintf(stderr,"(postsort_srv)\n");
1127   /* tests:
1128    *  dig -t srv _srv._tcp.test.iwj.relativity.greenend.org.uk.
1129    *   ./adnshost_s -t srv- _sip._udp.voip.net.cam.ac.uk.
1130    *   ./adnshost_s -t srv- _jabber._tcp.jabber.org
1131    */
1132 }
1133
1134 /*
1135  * _flat   (mf)
1136  */
1137
1138 static void mf_flat(adns_query qu, void *data) { }
1139
1140 /*
1141  * Now the table.
1142  */
1143
1144 #define TYPESZ_M(member)           (sizeof(*((adns_answer*)0)->rrs.member))
1145
1146 #define DEEP_MEMB(memb) TYPESZ_M(memb), mf_##memb, cs_##memb
1147 #define FLAT_MEMB(memb) TYPESZ_M(memb), mf_flat, cs_##memb
1148
1149 #define DEEP_TYPE(code,rrt,fmt,memb,parser,comparer,printer)    \
1150  { adns_r_##code, rrt,fmt,TYPESZ_M(memb), mf_##memb,            \
1151       printer,parser,comparer, adns__qdpl_normal,0 }
1152 #define FLAT_TYPE(code,rrt,fmt,memb,parser,comparer,printer)    \
1153  { adns_r_##code, rrt,fmt,TYPESZ_M(memb), mf_flat,              \
1154      printer,parser,comparer, adns__qdpl_normal,0 }
1155 #define XTRA_TYPE(code,rrt,fmt,memb,parser,comparer,printer,qdpl,postsort) \
1156  { adns_r_##code, rrt,fmt,TYPESZ_M(memb), mf_##memb,                       \
1157     printer,parser,comparer,qdpl,postsort }
1158
1159 static const typeinfo typeinfos[] = {
1160 /* Must be in ascending order of rrtype ! */
1161 /* mem-mgmt code  rrt     fmt   member   parser      comparer  printer */
1162
1163 FLAT_TYPE(a,      "A",     0,   inaddr,  pa_inaddr,  di_inaddr,cs_inaddr     ),
1164 DEEP_TYPE(ns_raw, "NS",   "raw",str,     pa_host_raw,0,        cs_domain     ),
1165 DEEP_TYPE(cname,  "CNAME", 0,   str,     pa_dom_raw, 0,        cs_domain     ),
1166 DEEP_TYPE(soa_raw,"SOA",  "raw",soa,     pa_soa,     0,        cs_soa        ),
1167 DEEP_TYPE(ptr_raw,"PTR",  "raw",str,     pa_host_raw,0,        cs_domain     ),
1168 DEEP_TYPE(hinfo,  "HINFO", 0, intstrpair,pa_hinfo,   0,        cs_hinfo      ),
1169 DEEP_TYPE(mx_raw, "MX",   "raw",intstr,  pa_mx_raw,  di_mx_raw,cs_inthost    ),
1170 DEEP_TYPE(txt,    "TXT",   0,   manyistr,pa_txt,     0,        cs_txt        ),
1171 DEEP_TYPE(rp_raw, "RP",   "raw",strpair, pa_rp,      0,        cs_rp         ),
1172 XTRA_TYPE(srv_raw,"SRV",  "raw",srvraw , pa_srvraw,  di_srv,   cs_srvraw,
1173                                                        qdpl_srv, postsort_srv),
1174
1175 FLAT_TYPE(addr,   "A",  "addr", addr,    pa_addr,    di_addr,  cs_addr       ),
1176 DEEP_TYPE(ns,     "NS", "+addr",hostaddr,pa_hostaddr,di_hostaddr,cs_hostaddr ),
1177 DEEP_TYPE(ptr,    "PTR","checked",str,   pa_ptr,     0,        cs_domain     ),
1178 DEEP_TYPE(mx,     "MX", "+addr",inthostaddr,pa_mx,   di_mx,    cs_inthostaddr),
1179 XTRA_TYPE(srv,    "SRV","+addr",srvha,   pa_srvha,   di_srv,   cs_srvha,
1180                                                        qdpl_srv, postsort_srv),
1181
1182 DEEP_TYPE(soa,    "SOA","822",  soa,     pa_soa,     0,        cs_soa        ),
1183 DEEP_TYPE(rp,     "RP", "822",  strpair, pa_rp,      0,        cs_rp         ),
1184 };
1185
1186 const typeinfo *adns__findtype(adns_rrtype type) {
1187   const typeinfo *begin, *end, *mid;
1188
1189   begin= typeinfos;  end= typeinfos+(sizeof(typeinfos)/sizeof(typeinfo));
1190
1191   while (begin < end) {
1192     mid= begin + ((end-begin)>>1);
1193     if (mid->type == type) return mid;
1194     if (type > mid->type) begin= mid+1;
1195     else end= mid;
1196   }
1197   return 0;
1198 }