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-2000 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_srv
, "srv" },
94 { adns_r_addr
, "addr" },
96 /* types with only one version */
97 { adns_r_cname
, "cname" },
98 { adns_r_hinfo
, "hinfo" },
99 { adns_r_txt
, "txt" },
103 { adns_r_ns_raw
, "ns-" },
104 { adns_r_soa_raw
, "soa-" },
105 { adns_r_ptr_raw
, "ptr-" },
106 { adns_r_mx_raw
, "mx-" },
107 { adns_r_rp_raw
, "rp-" },
108 { adns_r_srv_raw
, "srv-" },
113 const struct typename
*tnp
;
116 tnp
->type
&& strcmp(arg
,tnp
->desc
);
118 if (!tnp
->type
) usageerr("unknown RR type %s",arg
);
122 static void process_optarg(const char *arg
,
123 const char *const **argv_p
,
125 const struct optioninfo
*oip
;
129 if (arg
[0] == '-' || arg
[0] == '+') {
130 if (arg
[0] == '-' && arg
[1] == '-') {
131 if (!strncmp(arg
,"--no-",5)) {
133 oip
= opt_findl(arg
+5);
136 oip
= opt_findl(arg
+2);
138 if (oip
->type
== ot_funcarg
) {
139 arg
= argv_p ?
*++(*argv_p
) : value
;
140 if (!arg
) usageerr("option --%s requires a value argument",oip
->lopt
);
142 } else if (oip
->type
== ot_funcarg2
) {
145 arg2
= arg ?
*++(*argv_p
) : 0;
147 usageerr("option --%s requires two more arguments", oip
->lopt
);
149 if (value
) usageerr("option --%s does not take a value",oip
->lopt
);
153 opt_do(oip
,invert
,arg
,arg2
);
154 } else if (arg
[0] == '-' && arg
[1] == 0) {
155 arg
= argv_p ?
*++(*argv_p
) : value
;
156 if (!arg
) usageerr("option `-' must be followed by a domain");
158 } else { /* arg[1] != '-', != '\0' */
159 invert
= (arg
[0] == '+');
162 oip
= opt_finds(&arg
);
163 if (oip
->type
== ot_funcarg
) {
165 arg
= argv_p ?
*++(*argv_p
) : value
;
166 if (!arg
) usageerr("option -%s requires a value argument",oip
->sopt
);
168 if (value
) usageerr("two values for option -%s given !",oip
->sopt
);
170 opt_do(oip
,invert
,arg
,0);
173 if (value
) usageerr("option -%s does not take a value",oip
->sopt
);
174 opt_do(oip
,invert
,0,0);
178 } else { /* arg[0] != '-' */
183 static void read_stdin(void) {
185 char *newline
, *space
;
188 while (!anydone
|| used
) {
189 while (!(newline
= memchr(buf
,'\n',used
))) {
191 avail
+= 20; avail
<<= 1;
192 buf
= realloc(buf
,avail
);
193 if (!buf
) sysfail("realloc stdin buffer",errno
);
196 r
= read(0,buf
+used
,avail
-used
);
197 } while (r
< 0 && errno
== EINTR
);
200 /* fake up final newline */
208 if (r
< 0) sysfail("read stdin",errno
);
212 space
= strchr(buf
,' ');
213 if (space
) *space
++= 0;
214 process_optarg(buf
,0,space
);
215 used
-= (newline
-buf
);
216 memmove(buf
,newline
,used
);
221 int main(int argc
, const char *const *argv
) {
222 struct timeval
*tv
, tvbuf
;
227 fd_set readfds
, writefds
, exceptfds
;
230 while ((arg
= *++argv
)) process_optarg(arg
,&argv
,0);
232 if (!ov_pipe
&& !ads
) usageerr("no domains given, and -f/--pipe not used; try --help");
238 qu
= ov_asynch ?
0 : outstanding
.head ? outstanding
.head
->qu
: 0;
239 r
= adns_check(ads
,&qu
,&answer
,&qun_v
);
240 if (r
== EAGAIN
) break;
241 if (r
== ESRCH
) { if (!ov_pipe
) goto x_quit
; else break; }
243 query_done(qun_v
,answer
);
254 adns_beforeselect(ads
, &maxfd
, &readfds
,&writefds
,&exceptfds
, &tv
,&tvbuf
,0);
255 r
= select(maxfd
, &readfds
,&writefds
,&exceptfds
, tv
);
257 if (errno
== EINTR
) continue;
258 sysfail("select",errno
);
260 adns_afterselect(ads
, maxfd
, &readfds
,&writefds
,&exceptfds
, 0);
261 if (ov_pipe
&& FD_ISSET(0,&readfds
)) read_stdin();
264 if (fclose(stdout
)) outerr();