[adns] / src / setup.c

/*
 * setup.c
 * - configuration file parsing
 * - management of global state
 */
/*
 *  This file is part of adns, which is Copyright (C) 1997-1999 Ian Jackson
 *  
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2, or (at your option)
 *  any later version.
 *  
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *  
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software Foundation,
 *  Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. 
 */

#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>

#include <netdb.h>
#include <arpa/inet.h>

#include "internal.h"

static void readconfig(adns_state ads, const char *filename);

static void addserver(adns_state ads, struct in_addr addr) {
  int i;
  struct server *ss;
  
  for (i=0; i<ads->nservers; i++) {
    if (ads->servers[i].addr.s_addr == addr.s_addr) {
      adns__debug(ads,-1,0,"duplicate nameserver %s ignored",inet_ntoa(addr));
      return;
    }
  }
  
  if (ads->nservers>=MAXSERVERS) {
    adns__diag(ads,-1,0,"too many nameservers, ignoring %s",inet_ntoa(addr));
    return;
  }

  ss= ads->servers+ads->nservers;
  ss->addr= addr;
  ads->nservers++;
}

static void saveerr(adns_state ads, int en) {
  if (!ads->configerrno) ads->configerrno= en;
}

static void configparseerr(adns_state ads, const char *fn, int lno,
			   const char *fmt, ...) {
  va_list al;

  saveerr(ads,EINVAL);
  if (!ads->diagfile || (ads->iflags & adns_if_noerrprint)) return;

  if (lno==-1) fprintf(ads->diagfile,"adns: %s: ",fn);
  else fprintf(ads->diagfile,"adns: %s:%d: ",fn,lno);
  va_start(al,fmt);
  vfprintf(ads->diagfile,fmt,al);
  va_end(al);
  fputc('\n',ads->diagfile);
}

static int nextword(const char **bufp_io, const char **word_r, int *l_r) {
  const char *p, *q;

  p= *bufp_io;
  while (ctype_whitespace(*p)) p++;
  if (!*p) return 0;

  q= p;
  while (*q && !ctype_whitespace(*q)) q++;

  *l_r= q-p;
  *word_r= p;
  *bufp_io= q;

  return 1;
}

static void ccf_nameserver(adns_state ads, const char *fn, int lno, const char *buf) {
  struct in_addr ia;
  
  if (!inet_aton(buf,&ia)) {
    configparseerr(ads,fn,lno,"invalid nameserver address `%s'",buf);
    return;
  }
  adns__debug(ads,-1,0,"using nameserver %s",inet_ntoa(ia));
  addserver(ads,ia);
}

static void ccf_search(adns_state ads, const char *fn, int lno, const char *buf) {
  const char *bufp, *word;
  char *newchars, **newptrs, **pp;
  int count, tl, l;

  if (!buf) return;

  bufp= buf;
  count= 0;
  tl= 0;
  while (nextword(&bufp,&word,&l)) { count++; tl += l+1; }

  newptrs= malloc(sizeof(char*)*count);  if (!newptrs) { saveerr(ads,errno); return; }
  newchars= malloc(tl);  if (!newchars) { saveerr(ads,errno); free(newptrs); return; }

  bufp= buf;
  pp= newptrs;
  while (nextword(&bufp,&word,&l)) {
    *pp++= newchars;
    memcpy(newchars,word,l);
    newchars += l;
    *newchars++ = 0;
  }

  free(ads->searchlist);
  ads->nsearchlist= count;
  ads->searchlist= newptrs;
  /* fixme: actually pay attention */
}

static void ccf_sortlist(adns_state ads, const char *fn, int lno, const char *buf) {
  const char *word;
  char tbuf[200], *slash, *ep;
  struct in_addr base, mask;
  int l;
  unsigned long initial, baselocal;

  if (!buf) return;
  
  ads->nsortlist= 0;
  while (nextword(&buf,&word,&l)) {
    if (ads->nsortlist >= MAXSORTLIST) {
      adns__diag(ads,-1,0,"too many sortlist entries, ignoring %.*s onwards",l,word);
      return;
    }

    if (l >= sizeof(tbuf)) {
      configparseerr(ads,fn,lno,"sortlist entry `%.*s' too long",l,word);
      continue;
    }
    
    memcpy(tbuf,word,l);
    slash= strchr(tbuf,'/');
    if (slash) *slash++= 0;
    
    if (!inet_aton(tbuf,&base)) {
      configparseerr(ads,fn,lno,"invalid address `%s' in sortlist",tbuf);
      continue;
    }

    if (slash) {
      if (strchr(slash,'.')) {
	if (!inet_aton(slash,&mask)) {
	  configparseerr(ads,fn,lno,"invalid mask `%s' in sortlist",slash);
	  continue;
	}
	if (base.s_addr & ~mask.s_addr) {
	  configparseerr(ads,fn,lno,
			 "mask `%s' in sortlist overlaps address `%s'",slash,tbuf);
	  continue;
	}
      } else {
	initial= strtoul(slash,&ep,10);
	if (*ep || initial>32) {
	  configparseerr(ads,fn,lno,"mask length `%s' invalid",slash);
	  continue;
	}
	mask.s_addr= htonl((0x0ffffffffUL) << (32-initial));
      }
    } else {
      baselocal= ntohl(base.s_addr);
      if (!baselocal & 0x080000000UL) /* class A */
	mask.s_addr= htonl(0x0ff000000UL);
      else if ((baselocal & 0x0c0000000UL) == 0x080000000UL)
	mask.s_addr= htonl(0x0ffff0000UL); /* class B */
      else if ((baselocal & 0x0f0000000UL) == 0x0e0000000UL)
	mask.s_addr= htonl(0x0ff000000UL); /* class C */
      else {
	configparseerr(ads,fn,lno,
		       "network address `%s' in sortlist is not in classed ranges,"
		       " must specify mask explicitly", tbuf);
	continue;
      }
    }

    ads->sortlist[ads->nsortlist].base= base;
    ads->sortlist[ads->nsortlist].mask= mask;
    ads->nsortlist++;
  }
}

static void ccf_options(adns_state ads, const char *fn, int lno, const char *buf) {
  if (!buf) return;
  adns__diag(ads,-1,0,"warning - `options' ignored fixme");
}

static void ccf_clearnss(adns_state ads, const char *fn, int lno, const char *buf) {
  ads->nservers= 0;
}

static void ccf_include(adns_state ads, const char *fn, int lno, const char *buf) {
  if (!*buf) {
    configparseerr(ads,fn,lno,"`include' directive with no filename");
    return;
  }
  readconfig(ads,buf);
}

static const struct configcommandinfo {
  const char *name;
  void (*fn)(adns_state ads, const char *fn, int lno, const char *buf);
} configcommandinfos[]= {
  { "nameserver",        ccf_nameserver  },
  { "domain",            ccf_search      },
  { "search",            ccf_search      },
  { "sortlist",          ccf_sortlist    },
  { "options",           ccf_options     },
  { "clearnameservers",  ccf_clearnss    },
  { "include",           ccf_include     },
  {  0                                   }
};

typedef union {
  FILE *file;
  const char *text;
} getline_ctx;

static int gl_file(adns_state ads, getline_ctx *src_io, const char *filename,
		   int lno, char *buf, int buflen) {
  FILE *file= src_io->file;
  int c, i;
  char *p;

  p= buf;
  buflen--;
  i= 0;
    
  for (;;) { /* loop over chars */
    if (i == buflen) {
      adns__diag(ads,-1,0,"%s:%d: line too long, ignored",filename,lno);
      goto x_badline;
    }
    c= getc(file);
    if (!c) {
      adns__diag(ads,-1,0,"%s:%d: line contains nul, ignored",filename,lno);
      goto x_badline;
    } else if (c == '\n') {
      break;
    } else if (c == EOF) {
      if (ferror(file)) {
	saveerr(ads,errno);
	adns__diag(ads,-1,0,"%s:%d: read error: %s",filename,lno,strerror(errno));
	return -1;
      }
      if (!i) return -1;
      break;
    } else {
      *p++= c;
      i++;
    }
  }

  *p++= 0;
  return i;

 x_badline:
  saveerr(ads,EINVAL);
  while ((c= getc(file)) != EOF && c != '\n');
  return -2;
}

static int gl_text(adns_state ads, getline_ctx *src_io, const char *filename,
		   int lno, char *buf, int buflen) {
  const char *cp= src_io->text;
  int l;

  if (!cp || !*cp) return -1;

  if (*cp == ';' || *cp == '\n') cp++;
  l= strcspn(cp,";\n");
  src_io->text = cp+l;

  if (l >= buflen) {
    adns__diag(ads,-1,0,"%s:%d: line too long, ignored",filename,lno);
    saveerr(ads,EINVAL);
    return -2;
  }
    
  memcpy(buf,cp,l);
  buf[l]= 0;
  return l;
}

static void readconfiggeneric(adns_state ads, const char *filename,
			      int (*getline)(adns_state ads, getline_ctx*,
					     const char *filename, int lno,
					     char *buf, int buflen),
			      /* Returns >=0 for success, -1 for EOF or error
			       * (error will have been reported), or -2 for
			       * bad line was encountered, try again.
			       */
			      getline_ctx gl_ctx) {
  char linebuf[2000], *p, *q;
  int lno, l, dirl;
  const struct configcommandinfo *ccip;

  for (lno=1;
       (l= getline(ads,&gl_ctx, filename,lno, linebuf,sizeof(linebuf))) != -1;
       lno++) {
    if (l == -2) continue;
    while (l>0 && ctype_whitespace(linebuf[l-1])) l--;
    linebuf[l]= 0;
    p= linebuf;
    while (ctype_whitespace(*p)) p++;
    if (*p == '#' || !*p) continue;
    q= p;
    while (*q && !ctype_whitespace(*q)) q++;
    dirl= q-p;
    for (ccip=configcommandinfos;
	 ccip->name && !(strlen(ccip->name)==dirl && !memcmp(ccip->name,p,q-p));
	 ccip++);
    if (!ccip->name) {
      adns__diag(ads,-1,0,"%s:%d: unknown configuration directive `%.*s'",
		 filename,lno,q-p,p);
      continue;
    }
    while (ctype_whitespace(*q)) q++;
    ccip->fn(ads,filename,lno,q);
  }
}

static const char *instrum_getenv(adns_state ads, const char *envvar) {
  const char *value;

  value= getenv(envvar);
  if (!value) adns__debug(ads,-1,0,"environment variable %s not set",envvar);
  else adns__debug(ads,-1,0,"environment variable %s set to `%s'",envvar,value);
  return value;
}

static void readconfig(adns_state ads, const char *filename) {
  getline_ctx gl_ctx;
  
  gl_ctx.file= fopen(filename,"r");
  if (!gl_ctx.file) {
    if (errno == ENOENT) {
      adns__debug(ads,-1,0,"configuration file `%s' does not exist",filename);
      return;
    }
    saveerr(ads,errno);
    adns__diag(ads,-1,0,"cannot open configuration file `%s': %s",
	       filename,strerror(errno));
    return;
  }

  readconfiggeneric(ads,filename,gl_file,gl_ctx);
  
  fclose(gl_ctx.file);
}

static void readconfigtext(adns_state ads, const char *text, const char *showname) {
  getline_ctx gl_ctx;
  
  gl_ctx.text= text;
  readconfiggeneric(ads,showname,gl_text,gl_ctx);
}
  
static void readconfigenv(adns_state ads, const char *envvar) {
  const char *filename;

  if (ads->iflags & adns_if_noenv) {
    adns__debug(ads,-1,0,"not checking environment variable `%s'",envvar);
    return;
  }
  filename= instrum_getenv(ads,envvar);
  if (filename) readconfig(ads,filename);
}

static void readconfigenvtext(adns_state ads, const char *envvar) {
  const char *textdata;

  if (ads->iflags & adns_if_noenv) {
    adns__debug(ads,-1,0,"not checking environment variable `%s'",envvar);
    return;
  }
  textdata= instrum_getenv(ads,envvar);
  if (textdata) readconfigtext(ads,textdata,envvar);
}


int adns__setnonblock(adns_state ads, int fd) {
  int r;
  
  r= fcntl(fd,F_GETFL,0); if (r<0) return errno;
  r |= O_NONBLOCK;
  r= fcntl(fd,F_SETFL,r); if (r<0) return errno;
  return 0;
}

static int init_begin(adns_state *ads_r, adns_initflags flags, FILE *diagfile) {
  adns_state ads;
  
  ads= malloc(sizeof(*ads)); if (!ads) return errno;

  ads->iflags= flags;
  ads->diagfile= diagfile;
  LIST_INIT(ads->timew);
  LIST_INIT(ads->childw);
  LIST_INIT(ads->output);
  ads->nextid= 0x311f;
  ads->udpsocket= ads->tcpsocket= -1;
  adns__vbuf_init(&ads->tcpsend);
  adns__vbuf_init(&ads->tcprecv);
  ads->nservers= ads->nsortlist= ads->nsearchlist= ads->tcpserver= 0;
  ads->tcpstate= server_disconnected;
  ads->searchlist= 0;
  ads->searchndots= 1;
  timerclear(&ads->tcptimeout);

  *ads_r= ads;
  return 0;
}

static int init_finish(adns_state ads) {
  struct in_addr ia;
  struct protoent *proto;
  int r;
  
  if (!ads->nservers) {
    if (ads->diagfile && ads->iflags & adns_if_debug)
      fprintf(ads->diagfile,"adns: no nameservers, using localhost\n");
    ia.s_addr= htonl(INADDR_LOOPBACK);
    addserver(ads,ia);
  }

  proto= getprotobyname("udp"); if (!proto) { r= ENOPROTOOPT; goto x_free; }
  ads->udpsocket= socket(AF_INET,SOCK_DGRAM,proto->p_proto);
  if (ads->udpsocket<0) { r= errno; goto x_free; }

  r= adns__setnonblock(ads,ads->udpsocket);
  if (r) { r= errno; goto x_closeudp; }
  
  return 0;

 x_closeudp:
  close(ads->udpsocket);
 x_free:
  free(ads);
  return r;
}

static void init_abort(adns_state ads) {
  if (ads->nsearchlist) {
    free(ads->searchlist[0]);
    free(ads->searchlist);
  }
  free(ads);
}

int adns_init(adns_state *ads_r, adns_initflags flags, FILE *diagfile) {
  adns_state ads;
  const char *res_options, *adns_res_options;
  int r;
  
  r= init_begin(&ads, flags, diagfile ? diagfile : stderr);
  if (r) return r;
  
  res_options= instrum_getenv(ads,"RES_OPTIONS");
  adns_res_options= instrum_getenv(ads,"ADNS_RES_OPTIONS");
  ccf_options(ads,"RES_OPTIONS",-1,res_options);
  ccf_options(ads,"ADNS_RES_OPTIONS",-1,adns_res_options);

  readconfig(ads,"/etc/resolv.conf");
  readconfigenv(ads,"RES_CONF");
  readconfigenv(ads,"ADNS_RES_CONF");

  readconfigenvtext(ads,"RES_CONF_TEXT");
  readconfigenvtext(ads,"ADNS_RES_CONF_TEXT");

  ccf_options(ads,"RES_OPTIONS",-1,res_options);
  ccf_options(ads,"ADNS_RES_OPTIONS",-1,adns_res_options);

  ccf_search(ads,"LOCALDOMAIN",-1,instrum_getenv(ads,"LOCALDOMAIN"));
  ccf_search(ads,"ADNS_LOCALDOMAIN",-1,instrum_getenv(ads,"ADNS_LOCALDOMAIN"));

  if (ads->configerrno && ads->configerrno != EINVAL) {
    r= ads->configerrno;
    init_abort(ads);
    return r;
  }

  r= init_finish(ads);
  if (r) return r;

  *ads_r= ads;
  return 0;
}

int adns_init_strcfg(adns_state *ads_r, adns_initflags flags,
		     FILE *diagfile, const char *configtext) {
  adns_state ads;
  int r;

  r= init_begin(&ads, flags, diagfile);  if (r) return r;

  readconfigtext(ads,configtext,"<supplied configuration text>");
  if (ads->configerrno) {
    r= ads->configerrno;
    init_abort(ads);
    return r;
  }

  r= init_finish(ads);  if (r) return r;
  *ads_r= ads;
  return 0;
}

void adns_finish(adns_state ads) {
  for (;;) {
    if (ads->timew.head) adns_cancel(ads->timew.head);
    else if (ads->childw.head) adns_cancel(ads->childw.head);
    else if (ads->output.head) adns_cancel(ads->output.head);
    else break;
  }
  close(ads->udpsocket);
  if (ads->tcpsocket >= 0) close(ads->tcpsocket);
  adns__vbuf_free(&ads->tcpsend);
  adns__vbuf_free(&ads->tcprecv);
  free(ads);
}
Commit	Line	Data
e576be50	1	/*
	2	* setup.c
	3	* - configuration file parsing
	4	* - management of global state
	5	*/
	6	/*
a719a4be	7	* This file is part of adns, which is Copyright (C) 1997-1999 Ian Jackson
e576be50	8	*
	9	* This program is free software; you can redistribute it and/or modify
	10	* it under the terms of the GNU General Public License as published by
	11	* the Free Software Foundation; either version 2, or (at your option)
	12	* any later version.
	13	*
	14	* This program is distributed in the hope that it will be useful,
	15	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	16	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	17	* GNU General Public License for more details.
	18	*
	19	* You should have received a copy of the GNU General Public License
	20	* along with this program; if not, write to the Free Software Foundation,
	21	* Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
	22	*/
656b2da9	23
4353a5c4	24	#include <stdlib.h>
	25	#include <errno.h>
	26	#include <string.h>
	27	#include <unistd.h>
	28	#include <fcntl.h>
656b2da9	29
4353a5c4	30	#include <netdb.h>
	31	#include <arpa/inet.h>
	32
	33	#include "internal.h"
	34
36369543	35	static void readconfig(adns_state ads, const char *filename);
36369543	36
656b2da9	37	static void addserver(adns_state ads, struct in_addr addr) {
	38	int i;
	39	struct server *ss;
	40
	41	for (i=0; i<ads->nservers; i++) {
	42	if (ads->servers[i].addr.s_addr == addr.s_addr) {
3955725c	43	adns__debug(ads,-1,0,"duplicate nameserver %s ignored",inet_ntoa(addr));
656b2da9	44	return;
	45	}
	46	}
	47
	48	if (ads->nservers>=MAXSERVERS) {
3955725c	49	adns__diag(ads,-1,0,"too many nameservers, ignoring %s",inet_ntoa(addr));
656b2da9	50	return;
	51	}
	52
	53	ss= ads->servers+ads->nservers;
	54	ss->addr= addr;
656b2da9	55	ads->nservers++;
	56	}
	57
36369543	58	static void saveerr(adns_state ads, int en) {
	59	if (!ads->configerrno) ads->configerrno= en;
	60	}
	61
656b2da9	62	static void configparseerr(adns_state ads, const char *fn, int lno,
	63	const char *fmt, ...) {
	64	va_list al;
36369543	65
	66	saveerr(ads,EINVAL);
	67	if (!ads->diagfile \|\| (ads->iflags & adns_if_noerrprint)) return;
	68
	69	if (lno==-1) fprintf(ads->diagfile,"adns: %s: ",fn);
	70	else fprintf(ads->diagfile,"adns: %s:%d: ",fn,lno);
656b2da9	71	va_start(al,fmt);
36369543	72	vfprintf(ads->diagfile,fmt,al);
656b2da9	73	va_end(al);
36369543	74	fputc('\n',ads->diagfile);
656b2da9	75	}
656b2da9	76
32af6b2a	77	static int nextword(const char bufp_io, const char word_r, int *l_r) {
	78	const char p, q;
	79
	80	p= *bufp_io;
	81	while (ctype_whitespace(*p)) p++;
	82	if (!*p) return 0;
	83
	84	q= p;
	85	while (q && !ctype_whitespace(q)) q++;
	86
	87	*l_r= q-p;
	88	*word_r= p;
	89	*bufp_io= q;
	90
	91	return 1;
	92	}
	93
656b2da9	94	static void ccf_nameserver(adns_state ads, const char fn, int lno, const char buf) {
	95	struct in_addr ia;
	96
	97	if (!inet_aton(buf,&ia)) {
	98	configparseerr(ads,fn,lno,"invalid nameserver address `%s'",buf);
	99	return;
	100	}
3955725c	101	adns__debug(ads,-1,0,"using nameserver %s",inet_ntoa(ia));
656b2da9	102	addserver(ads,ia);
	103	}
	104
	105	static void ccf_search(adns_state ads, const char fn, int lno, const char buf) {
32af6b2a	106	const char bufp, word;
	107	char newchars, newptrs, *pp;
	108	int count, tl, l;
	109
656b2da9	110	if (!buf) return;
32af6b2a	111
	112	bufp= buf;
	113	count= 0;
	114	tl= 0;
	115	while (nextword(&bufp,&word,&l)) { count++; tl += l+1; }
	116
	117	newptrs= malloc(sizeof(char)count); if (!newptrs) { saveerr(ads,errno); return; }
	118	newchars= malloc(tl); if (!newchars) { saveerr(ads,errno); free(newptrs); return; }
	119
	120	bufp= buf;
	121	pp= newptrs;
	122	while (nextword(&bufp,&word,&l)) {
	123	*pp++= newchars;
	124	memcpy(newchars,word,l);
	125	newchars += l;
	126	*newchars++ = 0;
	127	}
	128
	129	free(ads->searchlist);
	130	ads->nsearchlist= count;
	131	ads->searchlist= newptrs;
	132	/* fixme: actually pay attention */
656b2da9	133	}
656b2da9	134
32af6b2a	135	static void ccf_sortlist(adns_state ads, const char fn, int lno, const char buf) {
32af6b2a	136	const char *word;
09957b1c	137	char tbuf[200], slash, ep;
	138	struct in_addr base, mask;
	139	int l;
	140	unsigned long initial, baselocal;
	141
32af6b2a	142	if (!buf) return;
32af6b2a	143
09957b1c	144	ads->nsortlist= 0;
32af6b2a	145	while (nextword(&buf,&word,&l)) {
09957b1c	146	if (ads->nsortlist >= MAXSORTLIST) {
32af6b2a	147	adns__diag(ads,-1,0,"too many sortlist entries, ignoring %.*s onwards",l,word);
09957b1c	148	return;
	149	}
	150
	151	if (l >= sizeof(tbuf)) {
32af6b2a	152	configparseerr(ads,fn,lno,"sortlist entry `%.*s' too long",l,word);
09957b1c	153	continue;
	154	}
	155
32af6b2a	156	memcpy(tbuf,word,l);
09957b1c	157	slash= strchr(tbuf,'/');
	158	if (slash) *slash++= 0;
	159
	160	if (!inet_aton(tbuf,&base)) {
	161	configparseerr(ads,fn,lno,"invalid address `%s' in sortlist",tbuf);
	162	continue;
	163	}
	164
	165	if (slash) {
	166	if (strchr(slash,'.')) {
	167	if (!inet_aton(slash,&mask)) {
	168	configparseerr(ads,fn,lno,"invalid mask `%s' in sortlist",slash);
	169	continue;
	170	}
	171	if (base.s_addr & ~mask.s_addr) {
	172	configparseerr(ads,fn,lno,
	173	"mask `%s' in sortlist overlaps address `%s'",slash,tbuf);
	174	continue;
	175	}
	176	} else {
	177	initial= strtoul(slash,&ep,10);
	178	if (*ep \|\| initial>32) {
	179	configparseerr(ads,fn,lno,"mask length `%s' invalid",slash);
	180	continue;
	181	}
	182	mask.s_addr= htonl((0x0ffffffffUL) << (32-initial));
	183	}
	184	} else {
	185	baselocal= ntohl(base.s_addr);
	186	if (!baselocal & 0x080000000UL) /* class A */
	187	mask.s_addr= htonl(0x0ff000000UL);
	188	else if ((baselocal & 0x0c0000000UL) == 0x080000000UL)
	189	mask.s_addr= htonl(0x0ffff0000UL); /* class B */
	190	else if ((baselocal & 0x0f0000000UL) == 0x0e0000000UL)
	191	mask.s_addr= htonl(0x0ff000000UL); /* class C */
	192	else {
	193	configparseerr(ads,fn,lno,
	194	"network address `%s' in sortlist is not in classed ranges,"
	195	" must specify mask explicitly", tbuf);
	196	continue;
	197	}
	198	}
	199
	200	ads->sortlist[ads->nsortlist].base= base;
	201	ads->sortlist[ads->nsortlist].mask= mask;
	202	ads->nsortlist++;
	203	}
656b2da9	204	}
	205
	206	static void ccf_options(adns_state ads, const char fn, int lno, const char buf) {
	207	if (!buf) return;
3955725c	208	adns__diag(ads,-1,0,"warning - `options' ignored fixme");
656b2da9	209	}
	210
	211	static void ccf_clearnss(adns_state ads, const char fn, int lno, const char buf) {
	212	ads->nservers= 0;
	213	}
	214
36369543	215	static void ccf_include(adns_state ads, const char fn, int lno, const char buf) {
	216	if (!*buf) {
	217	configparseerr(ads,fn,lno,"`include' directive with no filename");
	218	return;
	219	}
	220	readconfig(ads,buf);
	221	}
	222
656b2da9	223	static const struct configcommandinfo {
	224	const char *name;
	225	void (fn)(adns_state ads, const char fn, int lno, const char *buf);
	226	} configcommandinfos[]= {
	227	{ "nameserver", ccf_nameserver },
	228	{ "domain", ccf_search },
	229	{ "search", ccf_search },
	230	{ "sortlist", ccf_sortlist },
	231	{ "options", ccf_options },
	232	{ "clearnameservers", ccf_clearnss },
36369543	233	{ "include", ccf_include },
656b2da9	234	{ 0 }
	235	};
	236
36369543	237	typedef union {
656b2da9	238	FILE *file;
36369543	239	const char *text;
36369543	240	} getline_ctx;
656b2da9	241
36369543	242	static int gl_file(adns_state ads, getline_ctx src_io, const char filename,
	243	int lno, char *buf, int buflen) {
	244	FILE *file= src_io->file;
	245	int c, i;
	246	char *p;
	247
	248	p= buf;
	249	buflen--;
	250	i= 0;
	251
	252	for (;;) { /* loop over chars */
	253	if (i == buflen) {
	254	adns__diag(ads,-1,0,"%s:%d: line too long, ignored",filename,lno);
	255	goto x_badline;
	256	}
	257	c= getc(file);
	258	if (!c) {
	259	adns__diag(ads,-1,0,"%s:%d: line contains nul, ignored",filename,lno);
	260	goto x_badline;
	261	} else if (c == '\n') {
	262	break;
	263	} else if (c == EOF) {
	264	if (ferror(file)) {
	265	saveerr(ads,errno);
	266	adns__diag(ads,-1,0,"%s:%d: read error: %s",filename,lno,strerror(errno));
	267	return -1;
	268	}
	269	if (!i) return -1;
	270	break;
	271	} else {
	272	*p++= c;
	273	i++;
656b2da9	274	}
656b2da9	275	}
656b2da9	276
36369543	277	*p++= 0;
	278	return i;
	279
	280	x_badline:
	281	saveerr(ads,EINVAL);
	282	while ((c= getc(file)) != EOF && c != '\n');
	283	return -2;
	284	}
	285
	286	static int gl_text(adns_state ads, getline_ctx src_io, const char filename,
	287	int lno, char *buf, int buflen) {
09957b1c	288	const char *cp= src_io->text;
36369543	289	int l;
36369543	290
09957b1c	291	if (!cp \|\| !*cp) return -1;
36369543	292
09957b1c	293	if (cp == ';' \|\| cp == '\n') cp++;
	294	l= strcspn(cp,";\n");
	295	src_io->text = cp+l;
36369543	296
	297	if (l >= buflen) {
	298	adns__diag(ads,-1,0,"%s:%d: line too long, ignored",filename,lno);
	299	saveerr(ads,EINVAL);
	300	return -2;
	301	}
	302
	303	memcpy(buf,cp,l);
	304	buf[l]= 0;
	305	return l;
	306	}
	307
	308	static void readconfiggeneric(adns_state ads, const char *filename,
	309	int (getline)(adns_state ads, getline_ctx,
	310	const char *filename, int lno,
	311	char *buf, int buflen),
	312	/* Returns >=0 for success, -1 for EOF or error
	313	* (error will have been reported), or -2 for
	314	* bad line was encountered, try again.
	315	*/
	316	getline_ctx gl_ctx) {
	317	char linebuf[2000], p, q;
	318	int lno, l, dirl;
	319	const struct configcommandinfo *ccip;
	320
	321	for (lno=1;
	322	(l= getline(ads,&gl_ctx, filename,lno, linebuf,sizeof(linebuf))) != -1;
	323	lno++) {
	324	if (l == -2) continue;
656b2da9	325	while (l>0 && ctype_whitespace(linebuf[l-1])) l--;
	326	linebuf[l]= 0;
	327	p= linebuf;
	328	while (ctype_whitespace(*p)) p++;
36369543	329	if (p == '#' \|\| !p) continue;
656b2da9	330	q= p;
656b2da9	331	while (q && !ctype_whitespace(q)) q++;
36369543	332	dirl= q-p;
656b2da9	333	for (ccip=configcommandinfos;
36369543	334	ccip->name && !(strlen(ccip->name)==dirl && !memcmp(ccip->name,p,q-p));
656b2da9	335	ccip++);
656b2da9	336	if (!ccip->name) {
3955725c	337	adns__diag(ads,-1,0,"%s:%d: unknown configuration directive `%.*s'",
4353a5c4	338	filename,lno,q-p,p);
656b2da9	339	continue;
	340	}
	341	while (ctype_whitespace(*q)) q++;
	342	ccip->fn(ads,filename,lno,q);
	343	}
656b2da9	344	}
	345
	346	static const char instrum_getenv(adns_state ads, const char envvar) {
	347	const char *value;
	348
	349	value= getenv(envvar);
3955725c	350	if (!value) adns__debug(ads,-1,0,"environment variable %s not set",envvar);
3955725c	351	else adns__debug(ads,-1,0,"environment variable %s set to `%s'",envvar,value);
656b2da9	352	return value;
	353	}
	354
36369543	355	static void readconfig(adns_state ads, const char *filename) {
	356	getline_ctx gl_ctx;
	357
	358	gl_ctx.file= fopen(filename,"r");
	359	if (!gl_ctx.file) {
	360	if (errno == ENOENT) {
	361	adns__debug(ads,-1,0,"configuration file `%s' does not exist",filename);
	362	return;
	363	}
	364	saveerr(ads,errno);
	365	adns__diag(ads,-1,0,"cannot open configuration file `%s': %s",
	366	filename,strerror(errno));
	367	return;
	368	}
	369
	370	readconfiggeneric(ads,filename,gl_file,gl_ctx);
	371
	372	fclose(gl_ctx.file);
	373	}
	374
	375	static void readconfigtext(adns_state ads, const char text, const char showname) {
	376	getline_ctx gl_ctx;
	377
	378	gl_ctx.text= text;
	379	readconfiggeneric(ads,showname,gl_text,gl_ctx);
	380	}
	381
656b2da9	382	static void readconfigenv(adns_state ads, const char *envvar) {
	383	const char *filename;
	384
	385	if (ads->iflags & adns_if_noenv) {
3955725c	386	adns__debug(ads,-1,0,"not checking environment variable `%s'",envvar);
656b2da9	387	return;
	388	}
	389	filename= instrum_getenv(ads,envvar);
	390	if (filename) readconfig(ads,filename);
	391	}
4353a5c4	392
09957b1c	393	static void readconfigenvtext(adns_state ads, const char *envvar) {
	394	const char *textdata;
	395
	396	if (ads->iflags & adns_if_noenv) {
	397	adns__debug(ads,-1,0,"not checking environment variable `%s'",envvar);
	398	return;
	399	}
	400	textdata= instrum_getenv(ads,envvar);
	401	if (textdata) readconfigtext(ads,textdata,envvar);
	402	}
	403
4353a5c4	404
	405	int adns__setnonblock(adns_state ads, int fd) {
	406	int r;
656b2da9	407
4353a5c4	408	r= fcntl(fd,F_GETFL,0); if (r<0) return errno;
	409	r \|= O_NONBLOCK;
	410	r= fcntl(fd,F_SETFL,r); if (r<0) return errno;
	411	return 0;
	412	}
	413
36369543	414	static int init_begin(adns_state ads_r, adns_initflags flags, FILE diagfile) {
656b2da9	415	adns_state ads;
656b2da9	416
656b2da9	417	ads= malloc(sizeof(*ads)); if (!ads) return errno;
36369543	418
656b2da9	419	ads->iflags= flags;
36369543	420	ads->diagfile= diagfile;
4353a5c4	421	LIST_INIT(ads->timew);
	422	LIST_INIT(ads->childw);
	423	LIST_INIT(ads->output);
	424	ads->nextid= 0x311f;
	425	ads->udpsocket= ads->tcpsocket= -1;
4353a5c4	426	adns__vbuf_init(&ads->tcpsend);
4353a5c4	427	adns__vbuf_init(&ads->tcprecv);
32af6b2a	428	ads->nservers= ads->nsortlist= ads->nsearchlist= ads->tcpserver= 0;
4353a5c4	429	ads->tcpstate= server_disconnected;
32af6b2a	430	ads->searchlist= 0;
660d7d3b	431	ads->searchndots= 1;
4353a5c4	432	timerclear(&ads->tcptimeout);
656b2da9	433
36369543	434	*ads_r= ads;
	435	return 0;
	436	}
656b2da9	437
36369543	438	static int init_finish(adns_state ads) {
	439	struct in_addr ia;
	440	struct protoent *proto;
	441	int r;
	442
656b2da9	443	if (!ads->nservers) {
36369543	444	if (ads->diagfile && ads->iflags & adns_if_debug)
36369543	445	fprintf(ads->diagfile,"adns: no nameservers, using localhost\n");
09957b1c	446	ia.s_addr= htonl(INADDR_LOOPBACK);
656b2da9	447	addserver(ads,ia);
	448	}
	449
	450	proto= getprotobyname("udp"); if (!proto) { r= ENOPROTOOPT; goto x_free; }
	451	ads->udpsocket= socket(AF_INET,SOCK_DGRAM,proto->p_proto);
8402e34c	452	if (ads->udpsocket<0) { r= errno; goto x_free; }
4bec51a4	453
94436798	454	r= adns__setnonblock(ads,ads->udpsocket);
94436798	455	if (r) { r= errno; goto x_closeudp; }
656b2da9	456
656b2da9	457	return 0;
656b2da9	458
94436798	459	x_closeudp:
94436798	460	close(ads->udpsocket);
656b2da9	461	x_free:
	462	free(ads);
	463	return r;
	464	}
	465
32af6b2a	466	static void init_abort(adns_state ads) {
	467	if (ads->nsearchlist) {
	468	free(ads->searchlist[0]);
	469	free(ads->searchlist);
	470	}
	471	free(ads);
	472	}
	473
36369543	474	int adns_init(adns_state ads_r, adns_initflags flags, FILE diagfile) {
	475	adns_state ads;
	476	const char res_options, adns_res_options;
	477	int r;
	478
	479	r= init_begin(&ads, flags, diagfile ? diagfile : stderr);
	480	if (r) return r;
	481
	482	res_options= instrum_getenv(ads,"RES_OPTIONS");
	483	adns_res_options= instrum_getenv(ads,"ADNS_RES_OPTIONS");
	484	ccf_options(ads,"RES_OPTIONS",-1,res_options);
	485	ccf_options(ads,"ADNS_RES_OPTIONS",-1,adns_res_options);
	486
	487	readconfig(ads,"/etc/resolv.conf");
	488	readconfigenv(ads,"RES_CONF");
	489	readconfigenv(ads,"ADNS_RES_CONF");
	490
09957b1c	491	readconfigenvtext(ads,"RES_CONF_TEXT");
	492	readconfigenvtext(ads,"ADNS_RES_CONF_TEXT");
	493
36369543	494	ccf_options(ads,"RES_OPTIONS",-1,res_options);
	495	ccf_options(ads,"ADNS_RES_OPTIONS",-1,adns_res_options);
	496
	497	ccf_search(ads,"LOCALDOMAIN",-1,instrum_getenv(ads,"LOCALDOMAIN"));
	498	ccf_search(ads,"ADNS_LOCALDOMAIN",-1,instrum_getenv(ads,"ADNS_LOCALDOMAIN"));
	499
32af6b2a	500	if (ads->configerrno && ads->configerrno != EINVAL) {
	501	r= ads->configerrno;
	502	init_abort(ads);
	503	return r;
	504	}
	505
36369543	506	r= init_finish(ads);
	507	if (r) return r;
	508
	509	*ads_r= ads;
	510	return 0;
	511	}
	512
	513	int adns_init_strcfg(adns_state *ads_r, adns_initflags flags,
	514	FILE diagfile, const char configtext) {
	515	adns_state ads;
	516	int r;
	517
	518	r= init_begin(&ads, flags, diagfile); if (r) return r;
	519
	520	readconfigtext(ads,configtext,"<supplied configuration text>");
	521	if (ads->configerrno) {
	522	r= ads->configerrno;
32af6b2a	523	init_abort(ads);
36369543	524	return r;
	525	}
	526
	527	r= init_finish(ads); if (r) return r;
	528	*ads_r= ads;
	529	return 0;
	530	}
	531
9ec44266	532	void adns_finish(adns_state ads) {
	533	for (;;) {
	534	if (ads->timew.head) adns_cancel(ads->timew.head);
	535	else if (ads->childw.head) adns_cancel(ads->childw.head);
	536	else if (ads->output.head) adns_cancel(ads->output.head);
	537	else break;
	538	}
	539	close(ads->udpsocket);
	540	if (ads->tcpsocket >= 0) close(ads->tcpsocket);
	541	adns__vbuf_free(&ads->tcpsend);
	542	adns__vbuf_free(&ads->tcprecv);
	543	free(ads);
656b2da9	544	}