3 * - useful general-purpose resolver client program
4 * main program and useful subroutines
8 * Copyright (C) 1997-2000 Ian Jackson <ian@davenant.greenend.org.uk>
10 * It is part of adns, which is
11 * Copyright (C) 1997-2000 Ian Jackson <ian@davenant.greenend.org.uk>
12 * Copyright (C) 1999 Tony Finch <dot@dotat.at>
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)
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.
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.
32 const char *config_text
;
34 static int used
, avail
;
37 void quitnow(int rc
) {
38 if (ads
) adns_finish(ads
);
44 void sysfail(const char *what
, int errnoval
) {
45 fprintf(stderr
,"adnshost failed: %s: %s\n",what
,strerror(errnoval
));
49 void usageerr(const char *fmt
, ...) {
51 fputs("adnshost usage error: ",stderr
);
53 vfprintf(stderr
,fmt
,al
);
60 sysfail("write to stdout",errno
);
63 void *xmalloc(size_t sz
) {
66 p
= malloc(sz
); if (!p
) sysfail("malloc",sz
);
70 char *xstrsave(const char *str
) {
73 p
= xmalloc(strlen(str
)+1);
78 void of_config(const struct optioninfo
*oi
, const char *arg
, const char *arg2
) {
82 void of_type(const struct optioninfo
*oi
, const char *arg
, const char *arg2
) {
83 static const struct typename
{
87 /* enhanced versions */
89 { adns_r_soa
, "soa" },
90 { adns_r_ptr
, "ptr" },
93 { adns_r_addr
, "addr" },
95 /* types with only one version */
96 { adns_r_cname
, "cname" },
97 { adns_r_hinfo
, "hinfo" },
98 { adns_r_txt
, "txt" },
102 { adns_r_ns_raw
, "ns-" },
103 { adns_r_soa_raw
, "soa-" },
104 { adns_r_ptr_raw
, "ptr-" },
105 { adns_r_mx_raw
, "mx-" },
106 { adns_r_rp_raw
, "rp-" },
111 const struct typename
*tnp
;
114 tnp
->type
&& strcmp(arg
,tnp
->desc
);
116 if (!tnp
->type
) usageerr("unknown RR type %s",arg
);
120 static void process_optarg(const char *arg
,
121 const char *const **argv_p
,
123 const struct optioninfo
*oip
;
127 if (arg
[0] == '-' || arg
[0] == '+') {
128 if (arg
[0] == '-' && arg
[1] == '-') {
129 if (!strncmp(arg
,"--no-",5)) {
131 oip
= opt_findl(arg
+5);
134 oip
= opt_findl(arg
+2);
136 if (oip
->type
== ot_funcarg
) {
137 arg
= argv_p ?
*++(*argv_p
) : value
;
138 if (!arg
) usageerr("option --%s requires a value argument",oip
->lopt
);
140 } else if (oip
->type
== ot_funcarg2
) {
143 arg2
= arg ?
*++(*argv_p
) : 0;
145 usageerr("option --%s requires two more arguments", oip
->lopt
);
147 if (value
) usageerr("option --%s does not take a value",oip
->lopt
);
151 opt_do(oip
,invert
,arg
,arg2
);
152 } else if (arg
[0] == '-' && arg
[1] == 0) {
153 arg
= argv_p ?
*++(*argv_p
) : value
;
154 if (!arg
) usageerr("option `-' must be followed by a domain");
156 } else { /* arg[1] != '-', != '\0' */
157 invert
= (arg
[0] == '+');
160 oip
= opt_finds(&arg
);
161 if (oip
->type
== ot_funcarg
) {
163 arg
= argv_p ?
*++(*argv_p
) : value
;
164 if (!arg
) usageerr("option -%s requires a value argument",oip
->sopt
);
166 if (value
) usageerr("two values for option -%s given !",oip
->sopt
);
168 opt_do(oip
,invert
,arg
,0);
171 if (value
) usageerr("option -%s does not take a value",oip
->sopt
);
172 opt_do(oip
,invert
,0,0);
176 } else { /* arg[0] != '-' */
181 static void read_stdin(void) {
183 char *newline
, *space
;
186 while (!anydone
|| used
) {
187 while (!(newline
= memchr(buf
,'\n',used
))) {
189 avail
+= 20; avail
<<= 1;
190 buf
= realloc(buf
,avail
);
191 if (!buf
) sysfail("realloc stdin buffer",errno
);
194 r
= read(0,buf
+used
,avail
-used
);
195 } while (r
< 0 && errno
== EINTR
);
198 /* fake up final newline */
206 if (r
< 0) sysfail("read stdin",errno
);
210 space
= strchr(buf
,' ');
211 if (space
) *space
++= 0;
212 process_optarg(buf
,0,space
);
213 used
-= (newline
-buf
);
214 memmove(buf
,newline
,used
);
219 int main(int argc
, const char *const *argv
) {
220 struct timeval
*tv
, tvbuf
;
225 fd_set readfds
, writefds
, exceptfds
;
228 while ((arg
= *++argv
)) process_optarg(arg
,&argv
,0);
230 if (!ov_pipe
&& !ads
) usageerr("no domains given, and -f/--pipe not used; try --help");
236 qu
= ov_asynch ?
0 : outstanding
.head ? outstanding
.head
->qu
: 0;
237 r
= adns_check(ads
,&qu
,&answer
,&qun_v
);
238 if (r
== EAGAIN
) break;
239 if (r
== ESRCH
) { if (!ov_pipe
) goto x_quit
; else break; }
241 query_done(qun_v
,answer
);
252 adns_beforeselect(ads
, &maxfd
, &readfds
,&writefds
,&exceptfds
, &tv
,&tvbuf
,0);
253 r
= select(maxfd
, &readfds
,&writefds
,&exceptfds
, tv
);
255 if (errno
== EINTR
) continue;
256 sysfail("select",errno
);
258 adns_afterselect(ads
, maxfd
, &readfds
,&writefds
,&exceptfds
, 0);
259 if (ov_pipe
&& FD_ISSET(0,&readfds
)) read_stdin();
262 if (fclose(stdout
)) outerr();