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

   1 /*
   2  * setup.c
   3  * - configuration file parsing
   4  * - management of global state
   5  */
   6 /*
   7  *  This file is part of adns, which is
   8  *    Copyright (C) 1997-2000,2003,2006,2014  Ian Jackson
   9  *    Copyright (C) 2014  Mark Wooding
  10  *    Copyright (C) 1999-2000,2003,2006  Tony Finch
  11  *    Copyright (C) 1991 Massachusetts Institute of Technology
  12  *  (See the file INSTALL for full details.)
  13  *
  14  *  This program is free software; you can redistribute it and/or modify
  15  *  it under the terms of the GNU General Public License as published by
  16  *  the Free Software Foundation; either version 3, or (at your option)
  17  *  any later version.
  18  *
  19  *  This program is distributed in the hope that it will be useful,
  20  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  21  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  22  *  GNU General Public License for more details.
  23  *
  24  *  You should have received a copy of the GNU General Public License
  25  *  along with this program; if not, write to the Free Software Foundation.
  26  */
  27
  28 #include <stdlib.h>
  29 #include <errno.h>
  30 #include <limits.h>
  31 #include <unistd.h>
  32 #include <fcntl.h>
  33
  34 #include <sys/types.h>
  35 #include <netdb.h>
  36 #include <sys/socket.h>
  37 #include <netinet/in.h>
  38 #include <arpa/inet.h>
  39
  40 #include "internal.h"
  41
  42 static void readconfig(adns_state ads, const char *filename, int warnmissing);
  43
  44 static void addserver(adns_state ads, const struct sockaddr *sa, int salen) {
  45   int i;
  46   adns_rr_addr *ss;
  47   char buf[ADNS_ADDR2TEXT_BUFLEN];
  48
  49   for (i=0; i<ads->nservers; i++) {
  50     if (adns__sockaddrs_equal(sa, &ads->servers[i].addr.sa)) {
  51       adns__debug(ads,-1,0,"duplicate nameserver %s ignored",
  52                   adns__sockaddr_ntoa(sa, buf));
  53       return;
  54     }
  55   }
  56
  57   if (ads->nservers>=MAXSERVERS) {
  58     adns__diag(ads,-1,0,"too many nameservers, ignoring %s",
  59                adns__sockaddr_ntoa(sa, buf));
  60     return;
  61   }
  62
  63   ss= ads->servers+ads->nservers;
  64   assert(salen <= sizeof(ss->addr));
  65   ss->len = salen;
  66   memcpy(&ss->addr, sa, salen);
  67   ads->nservers++;
  68 }
  69
  70 static void freesearchlist(adns_state ads) {
  71   if (ads->nsearchlist) free(*ads->searchlist);
  72   free(ads->searchlist);
  73 }
  74
  75 static void saveerr(adns_state ads, int en) {
  76   if (!ads->configerrno) ads->configerrno= en;
  77 }
  78
  79 static void configparseerr(adns_state ads, const char *fn, int lno,
  80                            const char *fmt, ...) {
  81   va_list al;
  82
  83   saveerr(ads,EINVAL);
  84   if (!ads->logfn || (ads->iflags & adns_if_noerrprint)) return;
  85
  86   if (lno==-1) adns__lprintf(ads,"adns: %s: ",fn);
  87   else adns__lprintf(ads,"adns: %s:%d: ",fn,lno);
  88   va_start(al,fmt);
  89   adns__vlprintf(ads,fmt,al);
  90   va_end(al);
  91   adns__lprintf(ads,"\n");
  92 }
  93
  94 static int nextword(const char **bufp_io, const char **word_r, int *l_r) {
  95   const char *p, *q;
  96
  97   p= *bufp_io;
  98   while (ctype_whitespace(*p)) p++;
  99   if (!*p) return 0;
 100
 101   q= p;
 102   while (*q && !ctype_whitespace(*q)) q++;
 103
 104   *l_r= q-p;
 105   *word_r= p;
 106   *bufp_io= q;
 107
 108   return 1;
 109 }
 110
 111 static void ccf_nameserver(adns_state ads, const char *fn,
 112                            int lno, const char *buf) {
 113   adns_rr_addr a;
 114   char addrbuf[ADNS_ADDR2TEXT_BUFLEN];
 115   int err;
 116
 117   a.len= sizeof(a.addr);
 118   err= adns_text2addr(buf,DNS_PORT, 0, &a.addr.sa,&a.len);
 119   switch (err) {
 120   case 0:
 121     break;
 122   case EINVAL:
 123     configparseerr(ads,fn,lno,"invalid nameserver address `%s'",buf);
 124     return;
 125   default:
 126     configparseerr(ads,fn,lno,"failed to parse nameserver address `%s': %s",
 127                    buf,strerror(err));
 128     return;
 129   }
 130   adns__debug(ads,-1,0,"using nameserver %s",
 131               adns__sockaddr_ntoa(&a.addr.sa, addrbuf));
 132   addserver(ads,&a.addr.sa,a.len);
 133 }
 134
 135 static void ccf_search(adns_state ads, const char *fn,
 136                        int lno, const char *buf) {
 137   const char *bufp, *word;
 138   char *newchars, **newptrs, **pp;
 139   int count, tl, l;
 140
 141   if (!buf) return;
 142
 143   bufp= buf;
 144   count= 0;
 145   tl= 0;
 146   while (nextword(&bufp,&word,&l)) { count++; tl += l+1; }
 147
 148   newptrs= malloc(sizeof(char*)*count);
 149   if (!newptrs) { saveerr(ads,errno); return; }
 150
 151   newchars= malloc(tl);
 152   if (!newchars) { saveerr(ads,errno); free(newptrs); return; }
 153
 154   bufp= buf;
 155   pp= newptrs;
 156   while (nextword(&bufp,&word,&l)) {
 157     *pp++= newchars;
 158     memcpy(newchars,word,l);
 159     newchars += l;
 160     *newchars++ = 0;
 161   }
 162
 163   freesearchlist(ads);
 164   ads->nsearchlist= count;
 165   ads->searchlist= newptrs;
 166 }
 167
 168 static int gen_pton(const char *text, int want_af, adns_sockaddr *a) {
 169   int err;
 170   int len;
 171
 172   len= sizeof(*a);
 173   err= adns_text2addr(text,0, adns_qf_addrlit_scope_forbid,
 174                       &a->sa, &len);
 175   if (err) { assert(err == EINVAL); return 0; }
 176   if (want_af != AF_UNSPEC && a->sa.sa_family != want_af) return 0;
 177   return 1;
 178 }
 179
 180 static void ccf_sortlist(adns_state ads, const char *fn,
 181                          int lno, const char *buf) {
 182   const char *word;
 183   char tbuf[200], *slash, *ep;
 184   const char *maskwhat;
 185   struct sortlist *sl;
 186   int l;
 187   int initial= -1;
 188
 189   if (!buf) return;
 190
 191   ads->nsortlist= 0;
 192   while (nextword(&buf,&word,&l)) {
 193     if (ads->nsortlist >= MAXSORTLIST) {
 194       adns__diag(ads,-1,0,"too many sortlist entries,"
 195                  " ignoring %.*s onwards",l,word);
 196       return;
 197     }
 198
 199     if (l >= sizeof(tbuf)) {
 200       configparseerr(ads,fn,lno,"sortlist entry `%.*s' too long",l,word);
 201       continue;
 202     }
 203
 204     memcpy(tbuf,word,l); tbuf[l]= 0;
 205     slash= strchr(tbuf,'/');
 206     if (slash) *slash++= 0;
 207
 208     sl= &ads->sortlist[ads->nsortlist];
 209     if (!gen_pton(tbuf, AF_UNSPEC, &sl->base)) {
 210       configparseerr(ads,fn,lno,"invalid address `%s' in sortlist",tbuf);
 211       continue;
 212     }
 213
 214     if (slash) {
 215       if (slash[strspn(slash, "0123456789")]) {
 216         maskwhat = "mask";
 217         if (!gen_pton(slash, sl->base.sa.sa_family, &sl->mask)) {
 218           configparseerr(ads,fn,lno,"invalid mask `%s' in sortlist",slash);
 219           continue;
 220         }
 221       } else {
 222         maskwhat = "prefix length";
 223         initial= strtoul(slash,&ep,10);
 224         if (*ep || initial>adns__addr_width(sl->base.sa.sa_family)) {
 225           configparseerr(ads,fn,lno,"mask length `%s' invalid",slash);
 226           continue;
 227         }
 228         sl->mask.sa.sa_family= sl->base.sa.sa_family;
 229         adns__prefix_mask(&sl->mask, initial);
 230       }
 231     } else {
 232       maskwhat = "implied prefix length";
 233       initial= adns__guess_prefix_length(&sl->base);
 234       if (initial < 0) {
 235         configparseerr(ads,fn,lno, "network address `%s'"
 236                        " in sortlist is not in classed ranges,"
 237                        " must specify mask explicitly", tbuf);
 238         continue;
 239       }
 240       sl->mask.sa.sa_family= sl->base.sa.sa_family;
 241       adns__prefix_mask(&sl->mask, initial);
 242     }
 243
 244     if (!adns__addr_matches(sl->base.sa.sa_family,
 245                             adns__sockaddr_addr(&sl->base.sa),
 246                             &sl->base,&sl->mask)) {
 247       if (initial >= 0) {
 248         configparseerr(ads,fn,lno, "%s %d in sortlist"
 249                        " overlaps address `%s'",maskwhat,initial,tbuf);
 250       } else {
 251         configparseerr(ads,fn,lno, "%s `%s' in sortlist"
 252                        " overlaps address `%s'",maskwhat,slash,tbuf);
 253       }
 254       continue;
 255     }
 256
 257     ads->nsortlist++;
 258   }
 259 }
 260
 261 static void ccf_options(adns_state ads, const char *fn,
 262                         int lno, const char *buf) {
 263   const char *word;
 264   char *ep;
 265   unsigned long v;
 266   int i,l;
 267
 268   if (!buf) return;
 269
 270   while (nextword(&buf,&word,&l)) {
 271     if (l==5 && !memcmp(word,"debug",5)) {
 272       ads->iflags |= adns_if_debug;
 273       continue;
 274     }
 275     if (l>=6 && !memcmp(word,"ndots:",6)) {
 276       v= strtoul(word+6,&ep,10);
 277       if (l==6 || ep != word+l || v > INT_MAX) {
 278         configparseerr(ads,fn,lno,"option `%.*s' malformed"
 279                        " or has bad value",l,word);
 280         continue;
 281       }
 282       ads->searchndots= v;
 283       continue;
 284     }
 285     if (l>=12 && !memcmp(word,"adns_checkc:",12)) {
 286       if (!strcmp(word+12,"none")) {
 287         ads->iflags &= ~adns_if_checkc_freq;
 288         ads->iflags |= adns_if_checkc_entex;
 289       } else if (!strcmp(word+12,"entex")) {
 290         ads->iflags &= ~adns_if_checkc_freq;
 291         ads->iflags |= adns_if_checkc_entex;
 292       } else if (!strcmp(word+12,"freq")) {
 293         ads->iflags |= adns_if_checkc_freq;
 294       } else {
 295         configparseerr(ads,fn,lno, "option adns_checkc has bad value `%s' "
 296                        "(must be none, entex or freq", word+12);
 297       }
 298       continue;
 299     }
 300     if (l>=8 && !memcmp(word,"adns_af:",8)) {
 301       word += 8;
 302       ads->iflags &= ~adns_if_afmask;
 303       if (strcmp(word,"any")) for (;;) {
 304         i= strcspn(word,",");
 305         if (i>=4 && !memcmp(word,"ipv4",4))
 306           ads->iflags |= adns_if_permit_ipv4;
 307         else if (i>=4 && !memcmp(word,"ipv6",4))
 308           ads->iflags |= adns_if_permit_ipv6;
 309         else {
 310           configparseerr(ads,fn,lno, "option adns_af has bad value `%.*s' "
 311                          "(must be `any' or list {`ipv4',`ipv6'},...)",
 312                          i, word);
 313           break;
 314         }
 315         if (!word[i]) break;
 316         word= word + i + 1;
 317       }
 318       continue;
 319     }
 320     adns__diag(ads,-1,0,"%s:%d: unknown option `%.*s'", fn,lno, l,word);
 321   }
 322 }
 323
 324 static void ccf_clearnss(adns_state ads, const char *fn,
 325                          int lno, const char *buf) {
 326   ads->nservers= 0;
 327 }
 328
 329 static void ccf_include(adns_state ads, const char *fn,
 330                         int lno, const char *buf) {
 331   if (!*buf) {
 332     configparseerr(ads,fn,lno,"`include' directive with no filename");
 333     return;
 334   }
 335   readconfig(ads,buf,1);
 336 }
 337
 338 static void ccf_lookup(adns_state ads, const char *fn, int lno,
 339                        const char *buf) {
 340   int found_bind=0;
 341   const char *word;
 342   int l;
 343
 344   if (!*buf) {
 345     configparseerr(ads,fn,lno,"`lookup' directive with no databases");
 346     return;
 347   }
 348
 349   while (nextword(&buf,&word,&l)) {
 350     if (l==4 && !memcmp(word,"bind",4)) {
 351       found_bind=1;
 352     } else if (l==4 && !memcmp(word,"file",4)) {
 353       /* ignore this and hope /etc/hosts is not essential */
 354     } else if (l==2 && !memcmp(word,"yp",2)) {
 355       adns__diag(ads,-1,0,"%s:%d: yp lookups not supported by adns", fn,lno);
 356       found_bind=-1;
 357     } else {
 358       adns__diag(ads,-1,0,"%s:%d: unknown `lookup' database `%.*s'",
 359                  fn,lno, l,word);
 360       found_bind=-1;
 361     }
 362   }
 363   if (!found_bind)
 364     adns__diag(ads,-1,0,"%s:%d: `lookup' specified, but not `bind'", fn,lno);
 365 }
 366
 367 static const struct configcommandinfo {
 368   const char *name;
 369   void (*fn)(adns_state ads, const char *fn, int lno, const char *buf);
 370 } configcommandinfos[]= {
 371   { "nameserver",        ccf_nameserver  },
 372   { "domain",            ccf_search      },
 373   { "search",            ccf_search      },
 374   { "sortlist",          ccf_sortlist    },
 375   { "options",           ccf_options     },
 376   { "clearnameservers",  ccf_clearnss    },
 377   { "include",           ccf_include     },
 378   { "lookup",            ccf_lookup      }, /* OpenBSD */
 379   {  0                                   }
 380 };
 381
 382 typedef union {
 383   FILE *file;
 384   const char *text;
 385 } getline_ctx;
 386
 387 static int gl_file(adns_state ads, getline_ctx *src_io, const char *filename,
 388                    int lno, char *buf, int buflen) {
 389   FILE *file= src_io->file;
 390   int c, i;
 391   char *p;
 392
 393   p= buf;
 394   buflen--;
 395   i= 0;
 396
 397   for (;;) { /* loop over chars */
 398     if (i == buflen) {
 399       adns__diag(ads,-1,0,"%s:%d: line too long, ignored",filename,lno);
 400       goto x_badline;
 401     }
 402     c= getc(file);
 403     if (!c) {
 404       adns__diag(ads,-1,0,"%s:%d: line contains nul, ignored",filename,lno);
 405       goto x_badline;
 406     } else if (c == '\n') {
 407       break;
 408     } else if (c == EOF) {
 409       if (ferror(file)) {
 410         saveerr(ads,errno);
 411         adns__diag(ads,-1,0,"%s:%d: read error: %s",
 412                    filename,lno,strerror(errno));
 413         return -1;
 414       }
 415       if (!i) return -1;
 416       break;
 417     } else {
 418       *p++= c;
 419       i++;
 420     }
 421   }
 422
 423   *p++= 0;
 424   return i;
 425
 426  x_badline:
 427   saveerr(ads,EINVAL);
 428   while ((c= getc(file)) != EOF && c != '\n');
 429   return -2;
 430 }
 431
 432 static int gl_text(adns_state ads, getline_ctx *src_io, const char *filename,
 433                    int lno, char *buf, int buflen) {
 434   const char *cp= src_io->text;
 435   int l;
 436
 437   if (!cp || !*cp) return -1;
 438
 439   if (*cp == ';' || *cp == '\n') cp++;
 440   l= strcspn(cp,";\n");
 441   src_io->text = cp+l;
 442
 443   if (l >= buflen) {
 444     adns__diag(ads,-1,0,"%s:%d: line too long, ignored",filename,lno);
 445     saveerr(ads,EINVAL);
 446     return -2;
 447   }
 448
 449   memcpy(buf,cp,l);
 450   buf[l]= 0;
 451   return l;
 452 }
 453
 454 static void readconfiggeneric(adns_state ads, const char *filename,
 455                               int (*getline)(adns_state ads, getline_ctx*,
 456                                              const char *filename, int lno,
 457                                              char *buf, int buflen),
 458                               /* Returns >=0 for success, -1 for EOF or error
 459                                * (error will have been reported), or -2 for
 460                                * bad line was encountered, try again.
 461                                */
 462                               getline_ctx gl_ctx) {
 463   char linebuf[2000], *p, *q;
 464   int lno, l, dirl;
 465   const struct configcommandinfo *ccip;
 466
 467   for (lno=1;
 468        (l= getline(ads,&gl_ctx, filename,lno, linebuf,sizeof(linebuf))) != -1;
 469        lno++) {
 470     if (l == -2) continue;
 471     while (l>0 && ctype_whitespace(linebuf[l-1])) l--;
 472     linebuf[l]= 0;
 473     p= linebuf;
 474     while (ctype_whitespace(*p)) p++;
 475     if (*p == '#' || *p == ';' || !*p) continue;
 476     q= p;
 477     while (*q && !ctype_whitespace(*q)) q++;
 478     dirl= q-p;
 479     for (ccip=configcommandinfos;
 480          ccip->name &&
 481            !(strlen(ccip->name)==dirl && !memcmp(ccip->name,p,q-p));
 482          ccip++);
 483     if (!ccip->name) {
 484       adns__diag(ads,-1,0,"%s:%d: unknown configuration directive `%.*s'",
 485                  filename,lno,(int)(q-p),p);
 486       continue;
 487     }
 488     while (ctype_whitespace(*q)) q++;
 489     ccip->fn(ads,filename,lno,q);
 490   }
 491 }
 492
 493 static const char *instrum_getenv(adns_state ads, const char *envvar) {
 494   const char *value;
 495
 496   value= getenv(envvar);
 497   if (!value) adns__debug(ads,-1,0,"environment variable %s not set",envvar);
 498   else adns__debug(ads,-1,0,"environment variable %s"
 499                    " set to `%s'",envvar,value);
 500   return value;
 501 }
 502
 503 static void readconfig(adns_state ads, const char *filename, int warnmissing) {
 504   getline_ctx gl_ctx;
 505
 506   gl_ctx.file= fopen(filename,"r");
 507   if (!gl_ctx.file) {
 508     if (errno == ENOENT) {
 509       if (warnmissing)
 510         adns__debug(ads,-1,0, "configuration file"
 511                     " `%s' does not exist",filename);
 512       return;
 513     }
 514     saveerr(ads,errno);
 515     adns__diag(ads,-1,0,"cannot open configuration file `%s': %s",
 516                filename,strerror(errno));
 517     return;
 518   }
 519
 520   readconfiggeneric(ads,filename,gl_file,gl_ctx);
 521
 522   fclose(gl_ctx.file);
 523 }
 524
 525 static void readconfigtext(adns_state ads, const char *text,
 526                            const char *showname) {
 527   getline_ctx gl_ctx;
 528
 529   gl_ctx.text= text;
 530   readconfiggeneric(ads,showname,gl_text,gl_ctx);
 531 }
 532
 533 static void readconfigenv(adns_state ads, const char *envvar) {
 534   const char *filename;
 535
 536   if (ads->iflags & adns_if_noenv) {
 537     adns__debug(ads,-1,0,"not checking environment variable `%s'",envvar);
 538     return;
 539   }
 540   filename= instrum_getenv(ads,envvar);
 541   if (filename) readconfig(ads,filename,1);
 542 }
 543
 544 static void readconfigenvtext(adns_state ads, const char *envvar) {
 545   const char *textdata;
 546
 547   if (ads->iflags & adns_if_noenv) {
 548     adns__debug(ads,-1,0,"not checking environment variable `%s'",envvar);
 549     return;
 550   }
 551   textdata= instrum_getenv(ads,envvar);
 552   if (textdata) readconfigtext(ads,textdata,envvar);
 553 }
 554
 555
 556 int adns__setnonblock(adns_state ads, int fd) {
 557   int r;
 558
 559   r= fcntl(fd,F_GETFL,0); if (r<0) return errno;
 560   r |= O_NONBLOCK;
 561   r= fcntl(fd,F_SETFL,r); if (r<0) return errno;
 562   return 0;
 563 }
 564
 565 static int init_begin(adns_state *ads_r, adns_initflags flags,
 566                       adns_logcallbackfn *logfn, void *logfndata) {
 567   adns_state ads;
 568   pid_t pid;
 569
 570   if (flags & ~(adns_initflags)(0x4fff))
 571     /* 0x4000 is reserved for `harmless' future expansion */
 572     return ENOSYS;
 573
 574   ads= malloc(sizeof(*ads)); if (!ads) return errno;
 575
 576   ads->iflags= flags;
 577   ads->logfn= logfn;
 578   ads->logfndata= logfndata;
 579   ads->configerrno= 0;
 580   LIST_INIT(ads->udpw);
 581   LIST_INIT(ads->tcpw);
 582   LIST_INIT(ads->childw);
 583   LIST_INIT(ads->output);
 584   LIST_INIT(ads->intdone);
 585   ads->forallnext= 0;
 586   ads->nextid= 0x311f;
 587   ads->nudpsockets= 0;
 588   ads->tcpsocket= -1;
 589   adns__vbuf_init(&ads->tcpsend);
 590   adns__vbuf_init(&ads->tcprecv);
 591   ads->tcprecv_skip= 0;
 592   ads->nservers= ads->nsortlist= ads->nsearchlist= ads->tcpserver= 0;
 593   ads->searchndots= 1;
 594   ads->tcpstate= server_disconnected;
 595   timerclear(&ads->tcptimeout);
 596   ads->searchlist= 0;
 597
 598   pid= getpid();
 599   ads->rand48xsubi[0]= pid;
 600   ads->rand48xsubi[1]= (unsigned long)pid >> 16;
 601   ads->rand48xsubi[2]= pid ^ ((unsigned long)pid >> 16);
 602
 603   *ads_r= ads;
 604   return 0;
 605 }
 606
 607 static int init_finish(adns_state ads) {
 608   struct sockaddr_in sin;
 609   struct protoent *proto;
 610   struct udpsocket *udp;
 611   int i;
 612   int r;
 613
 614   if (!ads->nservers) {
 615     if (ads->logfn && ads->iflags & adns_if_debug)
 616       adns__lprintf(ads,"adns: no nameservers, using IPv4 localhost\n");
 617     memset(&sin, 0, sizeof(sin));
 618     sin.sin_family = AF_INET;
 619     sin.sin_port = htons(DNS_PORT);
 620     sin.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
 621     addserver(ads,(struct sockaddr *)&sin, sizeof(sin));
 622   }
 623
 624   proto= getprotobyname("udp"); if (!proto) { r= ENOPROTOOPT; goto x_free; }
 625   ads->nudpsockets= 0;
 626   for (i=0; i<ads->nservers; i++) {
 627     if (adns__udpsocket_by_af(ads, ads->servers[i].addr.sa.sa_family))
 628       continue;
 629     assert(ads->nudpsockets < MAXUDP);
 630     udp= &ads->udpsockets[ads->nudpsockets];
 631     udp->af= ads->servers[i].addr.sa.sa_family;
 632     udp->fd= socket(udp->af,SOCK_DGRAM,proto->p_proto);
 633     if (udp->fd < 0) { r= errno; goto x_free; }
 634     ads->nudpsockets++;
 635     r= adns__setnonblock(ads,udp->fd);
 636     if (r) { r= errno; goto x_closeudp; }
 637   }
 638
 639   return 0;
 640
 641  x_closeudp:
 642   for (i=0; i<ads->nudpsockets; i++) close(ads->udpsockets[i].fd);
 643  x_free:
 644   free(ads);
 645   return r;
 646 }
 647
 648 static void init_abort(adns_state ads) {
 649   if (ads->nsearchlist) {
 650     free(ads->searchlist[0]);
 651     free(ads->searchlist);
 652   }
 653   free(ads);
 654 }
 655
 656 static void logfn_file(adns_state ads, void *logfndata,
 657                        const char *fmt, va_list al) {
 658   vfprintf(logfndata,fmt,al);
 659 }
 660
 661 static int init_files(adns_state *ads_r, adns_initflags flags,
 662                       adns_logcallbackfn *logfn, void *logfndata) {
 663   adns_state ads;
 664   const char *res_options, *adns_res_options;
 665   int r;
 666
 667   r= init_begin(&ads, flags, logfn, logfndata);
 668   if (r) return r;
 669
 670   res_options= instrum_getenv(ads,"RES_OPTIONS");
 671   adns_res_options= instrum_getenv(ads,"ADNS_RES_OPTIONS");
 672   ccf_options(ads,"RES_OPTIONS",-1,res_options);
 673   ccf_options(ads,"ADNS_RES_OPTIONS",-1,adns_res_options);
 674
 675   readconfig(ads,"/etc/resolv.conf",1);
 676   readconfig(ads,"/etc/resolv-adns.conf",0);
 677   readconfigenv(ads,"RES_CONF");
 678   readconfigenv(ads,"ADNS_RES_CONF");
 679
 680   readconfigenvtext(ads,"RES_CONF_TEXT");
 681   readconfigenvtext(ads,"ADNS_RES_CONF_TEXT");
 682
 683   ccf_options(ads,"RES_OPTIONS",-1,res_options);
 684   ccf_options(ads,"ADNS_RES_OPTIONS",-1,adns_res_options);
 685
 686   ccf_search(ads,"LOCALDOMAIN",-1,instrum_getenv(ads,"LOCALDOMAIN"));
 687   ccf_search(ads,"ADNS_LOCALDOMAIN",-1,instrum_getenv(ads,"ADNS_LOCALDOMAIN"));
 688
 689   if (ads->configerrno && ads->configerrno != EINVAL) {
 690     r= ads->configerrno;
 691     init_abort(ads);
 692     return r;
 693   }
 694
 695   r= init_finish(ads);
 696   if (r) return r;
 697
 698   adns__consistency(ads,0,cc_entex);
 699   *ads_r= ads;
 700   return 0;
 701 }
 702
 703 int adns_init(adns_state *ads_r, adns_initflags flags, FILE *diagfile) {
 704   return init_files(ads_r, flags, logfn_file, diagfile ? diagfile : stderr);
 705 }
 706
 707 static int init_strcfg(adns_state *ads_r, adns_initflags flags,
 708                        adns_logcallbackfn *logfn, void *logfndata,
 709                        const char *configtext) {
 710   adns_state ads;
 711   int r;
 712
 713   r= init_begin(&ads, flags, logfn, logfndata);
 714   if (r) return r;
 715
 716   readconfigtext(ads,configtext,"<supplied configuration text>");
 717   if (ads->configerrno) {
 718     r= ads->configerrno;
 719     init_abort(ads);
 720     return r;
 721   }
 722
 723   r= init_finish(ads);  if (r) return r;
 724   adns__consistency(ads,0,cc_entex);
 725   *ads_r= ads;
 726   return 0;
 727 }
 728
 729 int adns_init_strcfg(adns_state *ads_r, adns_initflags flags,
 730                      FILE *diagfile, const char *configtext) {
 731   return init_strcfg(ads_r, flags,
 732                      diagfile ? logfn_file : 0, diagfile,
 733                      configtext);
 734 }
 735
 736 int adns_init_logfn(adns_state *newstate_r, adns_initflags flags,
 737                     const char *configtext /*0=>use default config files*/,
 738                     adns_logcallbackfn *logfn /*0=>logfndata is a FILE* */,
 739                     void *logfndata /*0 with logfn==0 => discard*/) {
 740   if (!logfn && logfndata)
 741     logfn= logfn_file;
 742   if (configtext)
 743     return init_strcfg(newstate_r, flags, logfn, logfndata, configtext);
 744   else
 745     return init_files(newstate_r, flags, logfn, logfndata);
 746 }
 747
 748 void adns_finish(adns_state ads) {
 749   int i;
 750   adns__consistency(ads,0,cc_entex);
 751   for (;;) {
 752     if (ads->udpw.head) adns__cancel(ads->udpw.head);
 753     else if (ads->tcpw.head) adns__cancel(ads->tcpw.head);
 754     else if (ads->childw.head) adns__cancel(ads->childw.head);
 755     else if (ads->output.head) adns__cancel(ads->output.head);
 756     else if (ads->intdone.head) adns__cancel(ads->output.head);
 757     else break;
 758   }
 759   for (i=0; i<ads->nudpsockets; i++) close(ads->udpsockets[i].fd);
 760   if (ads->tcpsocket >= 0) close(ads->tcpsocket);
 761   adns__vbuf_free(&ads->tcpsend);
 762   adns__vbuf_free(&ads->tcprecv);
 763   freesearchlist(ads);
 764   free(ads);
 765 }
 766
 767 void adns_forallqueries_begin(adns_state ads) {
 768   adns__consistency(ads,0,cc_entex);
 769   ads->forallnext=
 770     ads->udpw.head ? ads->udpw.head :
 771     ads->tcpw.head ? ads->tcpw.head :
 772     ads->childw.head ? ads->childw.head :
 773     ads->output.head;
 774 }
 775
 776 adns_query adns_forallqueries_next(adns_state ads, void **context_r) {
 777   adns_query qu, nqu;
 778
 779   adns__consistency(ads,0,cc_entex);
 780   nqu= ads->forallnext;
 781   for (;;) {
 782     qu= nqu;
 783     if (!qu) return 0;
 784     if (qu->next) {
 785       nqu= qu->next;
 786     } else if (qu == ads->udpw.tail) {
 787       nqu=
 788         ads->tcpw.head ? ads->tcpw.head :
 789         ads->childw.head ? ads->childw.head :
 790         ads->output.head;
 791     } else if (qu == ads->tcpw.tail) {
 792       nqu=
 793         ads->childw.head ? ads->childw.head :
 794         ads->output.head;
 795     } else if (qu == ads->childw.tail) {
 796       nqu= ads->output.head;
 797     } else {
 798       nqu= 0;
 799     }
 800     if (!qu->parent) break;
 801   }
 802   ads->forallnext= nqu;
 803   if (context_r) *context_r= qu->ctx.ext;
 804   return qu;
 805 }