[adns] / client / adh-query.c

/*
 * adh-query.c
 * - useful general-purpose resolver client program
 *   make queries and print answers
 */
/*
 *  This file is
 *    Copyright (C) 1997-1999 Ian Jackson <ian@davenant.greenend.org.uk>
 *
 *  It is part of adns, which is
 *    Copyright (C) 1997-1999 Ian Jackson <ian@davenant.greenend.org.uk>
 *    Copyright (C) 1999 Tony Finch <dot@dotat.at>
 *  
 *  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 "adnshost.h"

adns_state ads;
struct outstanding_list outstanding;

static unsigned long idcounter;

void ensure_adns_init(void) {
  int r;
  
  if (ads) return;

  if (signal(SIGPIPE,SIG_IGN) == SIG_ERR) sysfail("ignore SIGPIPE",errno);
  r= adns_init(&ads,
	       adns_if_noautosys|adns_if_nosigpipe |
	       (ov_env ? 0 : adns_if_noenv) |
	       ov_verbose,
	       0);
  if (r) sysfail("adns_init",r);

  if (ov_format == fmt_default)
    ov_format= ov_asynch ? fmt_asynch : fmt_simple;
}

static void prep_query(struct query_node **qun_r, int *quflags_r) {
  struct query_node *qun;
  char idbuf[20];
  
  if (ov_pipe && !ads) usageerr("-f/--pipe not consistent with domains on command line");
  ensure_adns_init();
  
  qun= malloc(sizeof(*qun));
  qun->pqfr= ov_pqfr;
  if (ov_id) {
    qun->id= xstrsave(ov_id);
  } else {
    sprintf(idbuf,"%lu",idcounter++);
    idcounter &= 0x0fffffffflu;
    qun->id= xstrsave(idbuf);
  }

  *quflags_r=
    (ov_search ? adns_qf_search : 0) |
    (ov_tcp ? adns_qf_usevc : 0) |
    ((ov_pqfr.show_owner || ov_format == fmt_simple) ? adns_qf_owner : 0) |
    (ov_qc_query ? adns_qf_quoteok_query : 0) |
    (ov_qc_anshost ? adns_qf_quoteok_anshost : 0) |
    (ov_qc_cname ? 0 : adns_qf_quoteok_cname) |
    ov_cname,
    
  *qun_r= qun;
}
  
void of_ptr(const struct optioninfo *oi, const char *arg) {
  struct query_node *qun;
  int quflags, r;
  struct sockaddr_in sa;

  memset(&sa,0,sizeof(sa));
  sa.sin_family= AF_INET;
  if (!inet_aton(arg,&sa.sin_addr)) usageerr("invalid IP address %s",arg);

  prep_query(&qun,&quflags);
  qun->owner= xstrsave(arg);
  r= adns_submit_reverse(ads,
			 (struct sockaddr*)&sa,
			 ov_type == adns_r_none ? adns_r_ptr : ov_type,
			 quflags,
			 qun,
			 &qun->qu);
  if (r) sysfail("adns_submit_reverse",r);

  LIST_LINK_TAIL(outstanding,qun);
}

void query_do(const char *domain) {
  struct query_node *qun;
  int quflags, r;

  prep_query(&qun,&quflags);
  qun->owner= xstrsave(domain);
  r= adns_submit(ads, domain,
		 ov_type == adns_r_none ? adns_r_addr : ov_type,
		 quflags,
		 qun,
		 &qun->qu);
  if (r) sysfail("adns_submit",r);

  LIST_LINK_TAIL(outstanding,qun);
}

static void dequeue_query(struct query_node *qun) {
  LIST_UNLINK(outstanding,qun);
  free(qun->id);
  free(qun);
}

static void print_withspace(const char *str) {
  if (printf("%s ", str) == EOF) outerr();
}

static void print_ttl(struct query_node *qun, adns_answer *answer) {
  unsigned long ttl;
  time_t now;
  
  switch (qun->pqfr.ttl) {
  case tm_none:
    return;
  case tm_rel:
    if (time(&now) == (time_t)-1) sysfail("get current time",errno);
    ttl= answer->expires < now ? 0 : answer->expires - now;
    break;
  case tm_abs:
    ttl= answer->expires;
    break;
  default:
    abort();
  }
  if (printf("%lu ",ttl) == EOF) outerr();
}

static const char *owner_show(struct query_node *qun, adns_answer *answer) {
  return answer->owner ? answer->owner : qun->owner;
}

static void print_owner_ttl(struct query_node *qun, adns_answer *answer) {
  if (qun->pqfr.show_owner) print_withspace(owner_show(qun,answer));
  print_ttl(qun,answer);
}

static void check_status(adns_status st) {
  static const adns_status statuspoints[]= {
    adns_s_ok,
    adns_s_max_localfail, adns_s_max_remotefail, adns_s_max_tempfail,
    adns_s_max_misconfig, adns_s_max_misquery
  };

  const adns_status *spp;
  int minrcode;

  for (minrcode=0, spp=statuspoints;
       spp < statuspoints + (sizeof(statuspoints)/sizeof(statuspoints[0]));
       spp++)
    if (st > *spp) minrcode++;
  if (rcode < minrcode) rcode= minrcode;
}

static void print_status(adns_status st, struct query_node *qun, adns_answer *answer) {
  const char *statustypeabbrev, *statusabbrev, *statusstring;

  statustypeabbrev= adns_errtypeabbrev(st);
  statusabbrev= adns_errabbrev(st);
  statusstring= adns_strerror(st);
  assert(!strchr(statusstring,'"'));

  if (printf("%s %d %s ", statustypeabbrev, st, statusabbrev)
      == EOF) outerr();
  print_owner_ttl(qun,answer);
  if (qun->pqfr.show_cname)
    print_withspace(answer->cname ? answer->cname : "$");
  if (printf("\"%s\"\n", statusstring) == EOF) outerr();
}

static void print_dnsfail(adns_status st, struct query_node *qun, adns_answer *answer) {
  int r;
  const char *typename, *statusstring;
  adns_status ist;
  
  if (ov_format == fmt_inline) {
    if (fputs("; failed ",stdout) == EOF) outerr();
    print_status(st,qun,answer);
    return;
  }
  assert(ov_format == fmt_simple);
  if (st == adns_s_nxdomain) {
    r= fprintf(stderr,"%s does not exist\n", owner_show(qun,answer));
  } else {
    ist= adns_rr_info(answer->type, &typename, 0,0,0,0);
    if (st == adns_s_nodata) {
      r= fprintf(stderr,"%s has no %s record\n", owner_show(qun,answer), typename);
    } else {
      statusstring= adns_strerror(st);
      r= fprintf(stderr,"Error during DNS %s lookup for %s: %s\n",
		 typename, owner_show(qun,answer), statusstring);
    }
  }
  if (r == EOF) sysfail("write error message to stderr",errno);
}
    
void query_done(struct query_node *qun, adns_answer *answer) {
  adns_status st, ist;
  int rrn, nrrs;
  const char *rrp, *realowner, *typename;
  char *datastr;

  st= answer->status;
  nrrs= answer->nrrs;
  if (ov_format == fmt_asynch) {
    check_status(st);
    if (printf("%s %d ", qun->id, nrrs) == EOF) outerr();
    print_status(st,qun,answer);
  } else {
    if (qun->pqfr.show_cname && answer->cname) {
      print_owner_ttl(qun,answer);
      if (qun->pqfr.show_type) print_withspace("CNAME");
      if (printf("%s\n", answer->cname) == EOF) outerr();
    }
    if (st) {
      check_status(st);
      print_dnsfail(st,qun,answer);
    }
  }
  if (qun->pqfr.show_owner) {
    realowner= answer->cname ? answer->cname : owner_show(qun,answer);
    assert(realowner);
  } else {
    realowner= 0;
  }
  if (nrrs) {
    for (rrn=0, rrp = answer->rrs.untyped;
	 rrn < nrrs;
	 rrn++, rrp += answer->rrsz) {
      if (realowner) print_withspace(realowner);
      print_ttl(qun,answer);
      ist= adns_rr_info(answer->type, &typename, 0, 0, rrp, &datastr);
      if (ist == adns_s_nomemory) sysfail("adns_rr_info failed",ENOMEM);
      assert(!ist);
      if (qun->pqfr.show_type) print_withspace(typename);
      if (printf("%s\n",datastr) == EOF) outerr();
      free(datastr);
    }
  }
  if (fflush(stdout)) outerr();
  free(answer);
  dequeue_query(qun);
}

void of_asynch_id(const struct optioninfo *oi, const char *arg) {
  free(ov_id);
  ov_id= xstrsave(arg);
}

void of_cancel_id(const struct optioninfo *oi, const char *arg) {
  struct query_node *qun;

  for (qun= outstanding.head;
       qun && strcmp(qun->id,arg);
       qun= qun->next);
  if (!qun) return;
  adns_cancel(qun->qu);
  dequeue_query(qun);
}
Commit	Line	Data
d1cff511	1	/*
	2	* adh-query.c
	3	* - useful general-purpose resolver client program
	4	* make queries and print answers
	5	*/
	6	/*
	7	* This file is
	8	* Copyright (C) 1997-1999 Ian Jackson <ian@davenant.greenend.org.uk>
	9	*
	10	* It is part of adns, which is
	11	* Copyright (C) 1997-1999 Ian Jackson <ian@davenant.greenend.org.uk>
	12	* Copyright (C) 1999 Tony Finch <dot@dotat.at>
	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 2, 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	* Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
	27	*/
	28
e3f575ff	29	#include "adnshost.h"
	30
	31	adns_state ads;
57d68ed1	32	struct outstanding_list outstanding;
e3f575ff	33
57d68ed1	34	static unsigned long idcounter;
57d68ed1	35
57d68ed1	36	void ensure_adns_init(void) {
	37	int r;
	38
	39	if (ads) return;
d1cff511	40
57d68ed1	41	if (signal(SIGPIPE,SIG_IGN) == SIG_ERR) sysfail("ignore SIGPIPE",errno);
	42	r= adns_init(&ads,
	43	adns_if_noautosys\|adns_if_nosigpipe \|
	44	(ov_env ? 0 : adns_if_noenv) \|
	45	ov_verbose,
	46	0);
	47	if (r) sysfail("adns_init",r);
413b9ad6	48
	49	if (ov_format == fmt_default)
	50	ov_format= ov_asynch ? fmt_asynch : fmt_simple;
57d68ed1	51	}
d1cff511	52
d7449548	53	static void prep_query(struct query_node *qun_r, int quflags_r) {
d1cff511	54	struct query_node *qun;
d1cff511	55	char idbuf[20];
d7449548	56
	57	if (ov_pipe && !ads) usageerr("-f/--pipe not consistent with domains on command line");
	58	ensure_adns_init();
	59
d1cff511	60	qun= malloc(sizeof(*qun));
	61	qun->pqfr= ov_pqfr;
	62	if (ov_id) {
	63	qun->id= xstrsave(ov_id);
	64	} else {
	65	sprintf(idbuf,"%lu",idcounter++);
	66	idcounter &= 0x0fffffffflu;
	67	qun->id= xstrsave(idbuf);
	68	}
d7449548	69
	70	*quflags_r=
	71	(ov_search ? adns_qf_search : 0) \|
	72	(ov_tcp ? adns_qf_usevc : 0) \|
413b9ad6	73	((ov_pqfr.show_owner \|\| ov_format == fmt_simple) ? adns_qf_owner : 0) \|
d7449548	74	(ov_qc_query ? adns_qf_quoteok_query : 0) \|
	75	(ov_qc_anshost ? adns_qf_quoteok_anshost : 0) \|
	76	(ov_qc_cname ? 0 : adns_qf_quoteok_cname) \|
	77	ov_cname,
	78
	79	*qun_r= qun;
	80	}
d1cff511	81
d7449548	82	void of_ptr(const struct optioninfo oi, const char arg) {
	83	struct query_node *qun;
	84	int quflags, r;
	85	struct sockaddr_in sa;
	86
	87	memset(&sa,0,sizeof(sa));
	88	sa.sin_family= AF_INET;
	89	if (!inet_aton(arg,&sa.sin_addr)) usageerr("invalid IP address %s",arg);
	90
	91	prep_query(&qun,&quflags);
4b2c4f8a	92	qun->owner= xstrsave(arg);
d7449548	93	r= adns_submit_reverse(ads,
	94	(struct sockaddr*)&sa,
	95	ov_type == adns_r_none ? adns_r_ptr : ov_type,
	96	quflags,
	97	qun,
	98	&qun->qu);
	99	if (r) sysfail("adns_submit_reverse",r);
	100
	101	LIST_LINK_TAIL(outstanding,qun);
	102	}
	103
	104	void query_do(const char *domain) {
	105	struct query_node *qun;
	106	int quflags, r;
	107
	108	prep_query(&qun,&quflags);
4b2c4f8a	109	qun->owner= xstrsave(domain);
e3f575ff	110	r= adns_submit(ads, domain,
e3f575ff	111	ov_type == adns_r_none ? adns_r_addr : ov_type,
d7449548	112	quflags,
d1cff511	113	qun,
	114	&qun->qu);
	115	if (r) sysfail("adns_submit",r);
	116
e3f575ff	117	LIST_LINK_TAIL(outstanding,qun);
d1cff511	118	}
e3f575ff	119
9fa14499	120	static void dequeue_query(struct query_node *qun) {
	121	LIST_UNLINK(outstanding,qun);
	122	free(qun->id);
	123	free(qun);
	124	}
	125
57d68ed1	126	static void print_withspace(const char *str) {
	127	if (printf("%s ", str) == EOF) outerr();
	128	}
	129
	130	static void print_ttl(struct query_node qun, adns_answer answer) {
	131	unsigned long ttl;
	132	time_t now;
	133
	134	switch (qun->pqfr.ttl) {
	135	case tm_none:
	136	return;
	137	case tm_rel:
	138	if (time(&now) == (time_t)-1) sysfail("get current time",errno);
	139	ttl= answer->expires < now ? 0 : answer->expires - now;
	140	break;
	141	case tm_abs:
	142	ttl= answer->expires;
	143	break;
	144	default:
	145	abort();
	146	}
	147	if (printf("%lu ",ttl) == EOF) outerr();
	148	}
	149
4b2c4f8a	150	static const char owner_show(struct query_node qun, adns_answer *answer) {
	151	return answer->owner ? answer->owner : qun->owner;
	152	}
	153
57d68ed1	154	static void print_owner_ttl(struct query_node qun, adns_answer answer) {
4b2c4f8a	155	if (qun->pqfr.show_owner) print_withspace(owner_show(qun,answer));
57d68ed1	156	print_ttl(qun,answer);
	157	}
	158
413b9ad6	159	static void check_status(adns_status st) {
9fa14499	160	static const adns_status statuspoints[]= {
	161	adns_s_ok,
	162	adns_s_max_localfail, adns_s_max_remotefail, adns_s_max_tempfail,
	163	adns_s_max_misconfig, adns_s_max_misquery
	164	};
	165
	166	const adns_status *spp;
9fa14499	167	int minrcode;
9fa14499	168
9fa14499	169	for (minrcode=0, spp=statuspoints;
	170	spp < statuspoints + (sizeof(statuspoints)/sizeof(statuspoints[0]));
	171	spp++)
	172	if (st > *spp) minrcode++;
	173	if (rcode < minrcode) rcode= minrcode;
413b9ad6	174	}
57d68ed1	175
413b9ad6	176	static void print_status(adns_status st, struct query_node qun, adns_answer answer) {
	177	const char statustypeabbrev, statusabbrev, *statusstring;
	178
	179	statustypeabbrev= adns_errtypeabbrev(st);
57d68ed1	180	statusabbrev= adns_errabbrev(st);
	181	statusstring= adns_strerror(st);
	182	assert(!strchr(statusstring,'"'));
	183
9fa14499	184	if (printf("%s %d %s ", statustypeabbrev, st, statusabbrev)
57d68ed1	185	== EOF) outerr();
	186	print_owner_ttl(qun,answer);
	187	if (qun->pqfr.show_cname)
	188	print_withspace(answer->cname ? answer->cname : "$");
	189	if (printf("\"%s\"\n", statusstring) == EOF) outerr();
	190	}
	191
413b9ad6	192	static void print_dnsfail(adns_status st, struct query_node qun, adns_answer answer) {
	193	int r;
	194	const char typename, statusstring;
	195	adns_status ist;
	196
	197	if (ov_format == fmt_inline) {
	198	if (fputs("; failed ",stdout) == EOF) outerr();
	199	print_status(st,qun,answer);
	200	return;
	201	}
	202	assert(ov_format == fmt_simple);
	203	if (st == adns_s_nxdomain) {
4b2c4f8a	204	r= fprintf(stderr,"%s does not exist\n", owner_show(qun,answer));
413b9ad6	205	} else {
	206	ist= adns_rr_info(answer->type, &typename, 0,0,0,0);
	207	if (st == adns_s_nodata) {
4b2c4f8a	208	r= fprintf(stderr,"%s has no %s record\n", owner_show(qun,answer), typename);
413b9ad6	209	} else {
	210	statusstring= adns_strerror(st);
	211	r= fprintf(stderr,"Error during DNS %s lookup for %s: %s\n",
4b2c4f8a	212	typename, owner_show(qun,answer), statusstring);
413b9ad6	213	}
	214	}
	215	if (r == EOF) sysfail("write error message to stderr",errno);
	216	}
	217
57d68ed1	218	void query_done(struct query_node qun, adns_answer answer) {
	219	adns_status st, ist;
	220	int rrn, nrrs;
	221	const char rrp, realowner, *typename;
	222	char *datastr;
	223
57d68ed1	224	st= answer->status;
57d68ed1	225	nrrs= answer->nrrs;
413b9ad6	226	if (ov_format == fmt_asynch) {
413b9ad6	227	check_status(st);
0b6f56cc	228	if (printf("%s %d ", qun->id, nrrs) == EOF) outerr();
57d68ed1	229	print_status(st,qun,answer);
57d68ed1	230	} else {
413b9ad6	231	if (qun->pqfr.show_cname && answer->cname) {
57d68ed1	232	print_owner_ttl(qun,answer);
413b9ad6	233	if (qun->pqfr.show_type) print_withspace("CNAME");
	234	if (printf("%s\n", answer->cname) == EOF) outerr();
	235	}
	236	if (st) {
	237	check_status(st);
	238	print_dnsfail(st,qun,answer);
57d68ed1	239	}
	240	}
	241	if (qun->pqfr.show_owner) {
4b2c4f8a	242	realowner= answer->cname ? answer->cname : owner_show(qun,answer);
57d68ed1	243	assert(realowner);
	244	} else {
	245	realowner= 0;
	246	}
	247	if (nrrs) {
	248	for (rrn=0, rrp = answer->rrs.untyped;
	249	rrn < nrrs;
	250	rrn++, rrp += answer->rrsz) {
	251	if (realowner) print_withspace(realowner);
	252	print_ttl(qun,answer);
	253	ist= adns_rr_info(answer->type, &typename, 0, 0, rrp, &datastr);
	254	if (ist == adns_s_nomemory) sysfail("adns_rr_info failed",ENOMEM);
	255	assert(!ist);
	256	if (qun->pqfr.show_type) print_withspace(typename);
	257	if (printf("%s\n",datastr) == EOF) outerr();
	258	free(datastr);
	259	}
	260	}
	261	if (fflush(stdout)) outerr();
57d68ed1	262	free(answer);
9fa14499	263	dequeue_query(qun);
57d68ed1	264	}
	265
	266	void of_asynch_id(const struct optioninfo oi, const char arg) {
	267	free(ov_id);
	268	ov_id= xstrsave(arg);
	269	}
	270
49d8f5b4	271	void of_cancel_id(const struct optioninfo oi, const char arg) {
9fa14499	272	struct query_node *qun;
57d68ed1	273
9fa14499	274	for (qun= outstanding.head;
e0d855e1	275	qun && strcmp(qun->id,arg);
9fa14499	276	qun= qun->next);
	277	if (!qun) return;
	278	adns_cancel(qun->qu);
	279	dequeue_query(qun);
	280	}