[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-2000 Ian Jackson <ian@davenant.greenend.org.uk>
 *
 *  It is part of adns, which is
 *    Copyright (C) 1997-2000 Ian Jackson <ian@davenant.greenend.org.uk>
 *    Copyright (C) 1999-2000 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) {
  adns_initflags initflags;
  int r;
  
  if (ads) return;

  if (signal(SIGPIPE,SIG_IGN) == SIG_ERR) sysfail("ignore SIGPIPE",errno);

  initflags= adns_if_noautosys|adns_if_nosigpipe|ov_verbose;
  if (!ov_env) initflags |= adns_if_noenv;

  if (config_text) {
    r= adns_init_strcfg(&ads, initflags, stderr, config_text);
  } else {
    r= adns_init(&ads, initflags, 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, const char *arg2) {
  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 of_reverse(const struct optioninfo *oi, const char *arg, const char *arg2) {
  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= xmalloc(strlen(arg) + strlen(arg2) + 2);
  sprintf(qun->owner, "%s %s", arg,arg2);
  r= adns_submit_reverse_any(ads,
			     (struct sockaddr*)&sa, arg2,
			     ov_type == adns_r_none ? adns_r_txt : 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->owner);
  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, const char *arg2) {
  free(ov_id);
  ov_id= xstrsave(arg);
}

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