2 * adns lookup TYPE DOMAIN [QUERY-OPTIONS] => [list RDATA]
3 * if no or dontknow, throws an exception, with errorCode one of
4 * ADNS permfail 300 nxdomain {No such domain}
5 * ADNS permfail 301 nodata {No such data}
6 * ADNS tempfail ERROR-CODE ERROR-NAME ERROR-STRING
8 * ERROR-CODE is the numerical adns status value
9 * ERROR-NAME is the symbolic adns status value (in lowercase)
10 * ERROR-STRING is the result of adns_strstatus
12 * adns synch TYPE DOMAIN [QUERY-OPTIONS] => RESULTS
13 * RESULTS is [list ok|permfail|tempfail
14 * ERROR-CODE ERROR-NAME ERROR-STRING \
17 * OWNER is the RR owner
18 * CNAME is the empty string or the canonical name if we went
21 * adns asynch ON-YES ON-NO ON-DONTKNOW XARGS \
23 * [QUERY-OPTIONS] => QUERY-ID
25 * [concat ON-YES|ON-NO|ON-DONTKNOW XARGS RESULTS]
26 * adns asynch-cancel QUERY-ID
28 * QUERY-OPTIONS are zero or more of
29 * -resolver RESOLVER (see adns new-resolver)
30 * default is to use a default resolver
39 * -reverse-any ZONE-A-LIKE
41 * adns new-resolver [RES-OPTIONS...] => RESOLVER
43 * -errfile stdout|stderr (stderr is the default)
45 * -errcallback CALLBACK results in eval CALLBACK [list MESSAGE]
46 * -noenv|-debug|-logpid
49 * -config CONFIG-STRING
51 * adns set-default-resolver RESOLVER
52 * cancels any outstanding queries from a previous anonymous
55 * adns destroy-resolver RESOLVER
56 * cancels outstanding queries
59 /* ---8<--- end of documentation comment --8<-- */
62 * adns.c - adns binding for Tcl
63 * Copyright 2006 Ian Jackson
65 * This program is free software; you can redistribute it and/or
66 * modify it under the terms of the GNU General Public License as
67 * published by the Free Software Foundation; either version 2 of the
68 * License, or (at your option) any later version.
70 * This program is distributed in the hope that it will be useful, but
71 * WITHOUT ANY WARRANTY; without even the implied warranty of
72 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
73 * General Public License for more details.
75 * You should have received a copy of the GNU General Public License
76 * along with this library; if not, write to the Free Software
77 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
87 #include "chiark_tcl_adns.h"
89 /*---------- important types and forward declarations ----------*/
91 typedef struct Query Query;
92 typedef struct Resolver Resolver;
93 typedef struct OptionInfo OptionInfo;
95 static void asynch_sethandlers(Resolver *res);
96 static void asynch_cancelhandlers(Resolver *res);
97 static void asynch_perturbed(Resolver *res);
99 static void asynch_query_dispose(Tcl_Interp *interp, Query *query);
101 #define ASSOC_DEFAULTRES "adns-defaultresolver"
103 /*---------- common resolver/query option processing ----------*/
106 /* this struct type is used to hold both resolver and query options */
107 /* common to resolver and query: */
108 unsigned long aflags;
109 unsigned long sflags;
112 Tcl_Obj *errcallback;
113 const char *config_string;
116 const char *reverseany;
121 int (*fn)(Tcl_Interp *ip, const OptionInfo *oi, Tcl_Obj *arg,
124 unsigned long flags_add, flags_remove;
131 static int oiufn_f(const OptionInfo *oi, unsigned long *flags) {
132 *flags &= ~oi->flags_remove;
133 *flags |= oi->flags_add;
136 static int oifn_fa(Tcl_Interp *ip, const OptionInfo *oi, Tcl_Obj *arg,
137 OptionParse *op) { return oiufn_f(oi,&op->aflags); }
138 static int oifn_fs(Tcl_Interp *ip, const OptionInfo *oi, Tcl_Obj *arg,
139 OptionParse *op) { return oiufn_f(oi,&op->sflags); }
141 static int oifn_reverse_any(Tcl_Interp *ip, const OptionInfo *oi,
142 Tcl_Obj *arg, OptionParse *op) {
143 return cht_pat_string(ip,arg,&op->reverseany);
146 #define OIFA1(t,f,r) { "-" #f, oifn_fa, 0, adns_##t##_##f, r }
147 #define OIFA2(t,f,g) { "-" #f "-" #g, oifn_fa, 0, adns_##t##_##f##_##g, 0 }
148 #define OIFS(f) { "-" #f, oifn_fs, 0, oisf_##f, 0 }
149 #define OICA(o) { "-" #o, oifn_##o, 1 }
151 static void optparse_blank(OptionParse *op) {
152 memset(op,0,sizeof(*op));
155 op->config_string= 0;
158 static int parse_options(Tcl_Interp *ip, int objc, Tcl_Obj *const *objv,
159 const OptionInfo opttable[], OptionParse *op) {
160 const OptionInfo *oi;
168 rc= cht_pat_enum(ip, *objv++, &oi_v, opttable, sizeof(OptionInfo),
169 "query or resolver option");
175 cht_setstringresult(ip,"missing value for option");
182 rc= oi->fn(ip,oi,arg,op);
188 /*---------- resolver management ----------*/
194 Tcl_TimerToken timertoken;
197 ScriptToInvoke errcallback;
198 Tcl_Obj *errstring_accum;
201 /* The default resolver is recorded using Tcl_SetAssocData with key
202 * ASSOC_DEFAULTRES to record the Resolver*. If it was explicitly
203 * created with `adns new-resolver' then ix will be >=0, and the
204 * resolver will be destroyed - leaving no default - when explicitly
205 * requested. If it was implicitly created (by starting a query when
206 * there is no default) then ix will be -1. */
208 static int oifn_errfile(Tcl_Interp *ip, const OptionInfo *oi,
209 Tcl_Obj *arg, OptionParse *op) {
213 rc= cht_pat_string(ip,arg,&str); if (rc) return rc;
214 if (!strcmp(str,"stderr")) op->errfile= stderr;
215 else if (!strcmp(str,"stdout")) op->errfile= stdout;
216 else return cht_staticerr(ip,"-errfile argument must be stderr or stdout",0);
218 op->aflags &= ~adns_if_noerrprint;
223 static int oifn_errcallback(Tcl_Interp *ip, const OptionInfo *oi,
224 Tcl_Obj *arg, OptionParse *op) {
225 op->errcallback= arg;
226 op->aflags &= ~adns_if_noerrprint;
231 static int oifn_config(Tcl_Interp *ip, const OptionInfo *oi,
232 Tcl_Obj *arg, OptionParse *op) {
233 return cht_pat_string(ip,arg,&op->config_string);
236 static const OptionInfo resolver_optioninfos[]= {
238 OIFA1(if,debug, adns_if_noerrprint),
239 OIFA1(if,logpid, adns_if_noerrprint),
240 OIFA1(if,noerrprint, adns_if_debug),
241 OIFA2(if,checkc,entex),
242 OIFA2(if,checkc,freq),
249 static void adnslogfn_flushmessage(Resolver *res) {
250 cht_scriptinv_invoke(&res->errcallback, 1, &res->errstring_accum);
251 Tcl_SetObjLength(res->errstring_accum, 0);
254 static void adnslogfn_callback(adns_state ads, void *logfndata,
255 const char *fmt, va_list al) {
256 Resolver *res= logfndata;
260 l= vasprintf(&str,fmt,al);
262 cht_posixerr(res->interp,errno,"construct adns log callback string");
263 Tcl_BackgroundError(res->interp);
266 if (l==0) { free(str); return; }
267 if ((newline= l>0 && str[l-1]=='\n')) l--;
269 if (!res->errstring_accum) {
270 res->errstring_accum= Tcl_NewStringObj(str,l);
271 Tcl_IncrRefCount(res->errstring_accum);
273 Tcl_AppendToObj(res->errstring_accum,str,l);
278 adnslogfn_flushmessage(res);
281 static Resolver *default_resolver(Tcl_Interp *ip) {
282 return Tcl_GetAssocData(ip,ASSOC_DEFAULTRES,0);
285 static void destroy_resolver(Tcl_Interp *ip, Resolver *res) {
291 if (res == default_resolver(ip))
292 Tcl_DeleteAssocData(ip,ASSOC_DEFAULTRES);
294 if (res->errstring_accum) {
295 rstr= Tcl_GetStringFromObj(res->errstring_accum, &logstring_len);
298 adnslogfn_flushmessage(res);
302 /* although adns would throw these away for us, we need to
303 * destroy our own data too and only adns has a list of them */
305 adns_forallqueries_begin(res->ads);
306 aqu= adns_forallqueries_next(res->ads, &query_v);
308 asynch_query_dispose(ip, query_v);
310 adns_finish(res->ads);
313 asynch_cancelhandlers(res);
314 cht_scriptinv_cancel(&res->errcallback);
315 Tcl_EventuallyFree(res, Tcl_Free);
318 static void destroy_resolver_idtabcb(Tcl_Interp *ip, void *resolver_v) {
319 destroy_resolver(ip,resolver_v);
321 static void destroy_resolver_defcb(ClientData resolver_v, Tcl_Interp *ip) {
322 destroy_resolver(ip,resolver_v);
325 int cht_do_adns_destroy_resolver(ClientData cd, Tcl_Interp *ip, void *res_v) {
326 cht_tabledataid_disposing(ip,res_v,&adnstcl_resolvers);
327 destroy_resolver(ip,res_v);
331 static int create_resolver(Tcl_Interp *ip, const OptionParse *op,
336 res= TALLOC(sizeof(*res)); assert(res);
342 for (i=0; i<3; i++) FD_ZERO(&res->handling[i]);
343 cht_scriptinv_init(&res->errcallback);
344 res->errstring_accum= 0;
347 cht_scriptinv_set(&res->errcallback, ip, op->errcallback, 0);
349 ec= adns_init_logfn(&res->ads,
350 op->aflags | adns_if_noautosys,
352 op->errcallback ? adnslogfn_callback : 0,
353 op->errcallback ? (void*)res : (void*)op->errfile);
354 if (ec) { rc= cht_posixerr(ip,ec,"create adns resolver"); goto x_rc; }
361 if (res->ads) adns_finish(res->ads);
367 int cht_do_adns_new_resolver(ClientData cd, Tcl_Interp *ip,
368 int objc, Tcl_Obj *const *objv,
375 rc= parse_options(ip,objc,objv,resolver_optioninfos,&op);
378 if (op.aflags & adns_if_noerrprint) {
383 rc= create_resolver(ip, &op, &res);
390 int cht_do_adns_set_default_resolver(ClientData cd, Tcl_Interp *ip, void *res_v) {
391 Resolver *res= res_v;
392 Tcl_DeleteAssocData(ip,ASSOC_DEFAULTRES);
393 Tcl_SetAssocData(ip, ASSOC_DEFAULTRES, 0, res);
397 const IdDataSpec cht_adnstcl_resolvers= {
398 "adns-res", "adns-resolvers-table", destroy_resolver_idtabcb
401 /*---------- query, query option and answers - common stuff ----------*/
403 #define RRTYPE_EXACTLY(t) { #t, adns_r_##t }
404 #define RRTYPE_RAW(t) { #t, adns_r_##t##_raw }
405 #define RRTYPE_PLUS(t) { #t "+", adns_r_##t }
406 #define RRTYPE_MINUS(t) { #t "-", adns_r_##t##_raw }
408 const AdnsTclRRTypeInfo cht_adnstclrrtypeinfo_entries[]= {
410 RRTYPE_EXACTLY(cname),
411 RRTYPE_EXACTLY(hinfo),
412 RRTYPE_EXACTLY(addr),
427 static int oifn_resolver(Tcl_Interp *ip, const OptionInfo *oi,
428 Tcl_Obj *arg, OptionParse *op) {
432 rc= cht_pat_iddata(ip,arg,&val_v,&adnstcl_resolvers);
438 static const OptionInfo query_optioninfos[]= {
441 OIFA2(qf,quoteok,query),
442 OIFA2(qf,quoteok,anshost),
443 OIFA2(qf,quotefail,cname),
444 OIFA2(qf,cname,loose),
445 OIFA2(qf,cname,forbid),
448 { "-reverse-any", oifn_reverse_any, 1 },
452 static int query_submit(Tcl_Interp *ip,
453 const AdnsTclRRTypeInfo *type, const char *domain,
454 int queryopts_objc, Tcl_Obj *const *queryopts_objv,
455 adns_query *aqu_r, void *context, Resolver **res_r) {
457 static const int aftry[]= { AF_INET, AF_INET6 };
463 op.aflags= adns_qf_owner;
467 rc= parse_options(ip, queryopts_objc,queryopts_objv, query_optioninfos,&op);
471 op.resolver= default_resolver(ip);
473 optparse_blank(&res_op);
474 rc= create_resolver(ip, &res_op, &op.resolver);
477 Tcl_SetAssocData(ip, ASSOC_DEFAULTRES,
478 destroy_resolver_defcb, op.resolver);
484 if (op.reverseany || (op.sflags & oisf_reverse)) {
486 for (af=aftry; af < af + sizeof(af)/sizeof(*af); af++) {
487 memset(&sa,0,sizeof(sa));
489 r= inet_pton(*af,domain,&sa);
490 if (!r) goto af_found;
492 return cht_staticerr(ip,"invalid address for adns reverse submit",
493 "ADNS REVERSE INVALID");
497 ads= op.resolver->ads;
500 ec= adns_submit_reverse_any(ads, &sa, op.reverseany,
501 type->number, op.aflags, context, aqu_r);
502 } else if (op.sflags & oisf_reverse) {
503 ec= adns_submit_reverse(ads, &sa,
504 type->number, op.aflags, context, aqu_r);
506 ec= adns_submit(ads, domain,
507 type->number, op.aflags, context, aqu_r);
510 return cht_posixerr(ip,ec,"submit adns query");
515 #define RESULTSTATUS_LLEN 4
516 #define RESULTLIST_LLEN 7
518 static void make_resultstatus(Tcl_Interp *ip, adns_status status,
519 Tcl_Obj *results[RESULTSTATUS_LLEN]) {
520 results[0]= cht_ret_string(ip, adns_errtypeabbrev(status));
521 results[1]= cht_ret_int(ip, status);
522 results[2]= cht_ret_string(ip, adns_errabbrev(status));
523 results[3]= cht_ret_string(ip, adns_strerror(status));
526 static Tcl_Obj *make_resultrdata(Tcl_Interp *ip, adns_answer *answer) {
527 Tcl_Obj **rdata, *rl;
530 char *datap, *rdatastring;
532 rdata= TALLOC(sizeof(*rdata) * answer->nrrs);
533 for (i=0, datap=answer->rrs.untyped;
535 i++, datap += rrsz) {
536 st= adns_rr_info(answer->type, 0,0, &rrsz, datap, &rdatastring);
538 rdata[i]= cht_ret_string(ip, rdatastring);
541 rl= Tcl_NewListObj(answer->nrrs, rdata);
546 static void make_resultlist(Tcl_Interp *ip, adns_answer *answer,
547 Tcl_Obj *results[RESULTLIST_LLEN]) {
549 make_resultstatus(ip, answer->status, results);
550 assert(RESULTSTATUS_LLEN==4);
551 results[4]= cht_ret_string(ip, answer->owner);
552 results[5]= cht_ret_string(ip, answer->cname ? answer->cname : "");
553 results[6]= make_resultrdata(ip, answer);
556 /*---------- synchronous query handling ----------*/
558 static int synch(Tcl_Interp *ip, const AdnsTclRRTypeInfo *rrtype,
560 int objc, Tcl_Obj *const *objv, adns_answer **answer_r) {
565 rc= query_submit(ip,rrtype,domain,objc,objv,&aqu,0,&res);
568 ec= adns_wait(res->ads,&aqu,answer_r,0);
571 asynch_perturbed(res);
575 int cht_do_adns_lookup(ClientData cd, Tcl_Interp *ip,
576 const AdnsTclRRTypeInfo *rrtype,
578 int objc, Tcl_Obj *const *objv,
583 rc= synch(ip,rrtype,domain,objc,objv,&answer); if (rc) return rc;
585 if (answer->status) {
586 Tcl_Obj *problem[RESULTSTATUS_LLEN];
587 make_resultstatus(ip, answer->status, problem);
588 *result= Tcl_NewListObj(RESULTSTATUS_LLEN, problem);
590 *result= make_resultrdata(ip, answer);
596 int cht_do_adns_synch(ClientData cd, Tcl_Interp *ip,
597 const AdnsTclRRTypeInfo *rrtype,
599 int objc, Tcl_Obj *const *objv,
603 Tcl_Obj *results[RESULTLIST_LLEN];
605 rc= synch(ip,rrtype,domain,objc,objv,&answer); if (rc) return rc;
606 make_resultlist(ip,answer,results);
608 *result= Tcl_NewListObj(RESULTLIST_LLEN, results);
612 /*---------- asynchronous query handling ----------*/
618 ScriptToInvoke on_yes, on_no, on_fail;
622 static void asynch_check_now(Resolver *res);
624 static void asynch_timerhandler(void *res_v) {
625 Resolver *res= res_v;
627 adns_processtimeouts(res->ads,0);
628 asynch_check_now(res);
631 static void asynch_filehandler(void *res_v, int mask) {
632 Resolver *res= res_v;
635 ec= adns_processany(res->ads);
636 if (ec) adns_globalsystemfailure(res->ads);
637 asynch_check_now(res);
640 static void asynch_sethandlers_generic(Resolver *res,
641 int shutdown /*from _cancelhandlers*/,
642 int immediate /*from _perturbed*/) {
645 struct timeval tv_buf, *timeout;
650 for (i=0; i<3; i++) FD_ZERO(&want[i]);
653 adns_beforeselect(res->ads,&maxfd,&want[0],&want[1],&want[2],
656 for (fd= 0; fd < maxfd || fd < res->maxfd; fd++)
658 if (!!FD_ISSET(fd, &res->handling[i])
659 != !!FD_ISSET(fd, &want[i])) {
661 if (FD_ISSET(fd, &want[0])) mask |= TCL_READABLE;
662 if (FD_ISSET(fd, &want[1])) mask |= TCL_WRITABLE;
663 if (FD_ISSET(fd, &want[2])) mask |= TCL_EXCEPTION;
665 Tcl_CreateFileHandler(fd,mask,asynch_filehandler,res);
666 FD_SET(fd, &res->handling[i]);
668 Tcl_DeleteFileHandler(fd);
669 FD_CLR(fd, &res->handling[i]);
674 Tcl_DeleteTimerHandler(res->timertoken);
677 res->timertoken= Tcl_CreateTimerHandler(0,asynch_timerhandler,res);
678 } else if (timeout) {
681 if (timeout->tv_sec >= INT_MAX/1000 - 1)
682 milliseconds= INT_MAX;
684 milliseconds= timeout->tv_sec * 1000 +
685 (timeout->tv_usec + 999) / 1000;
688 Tcl_CreateTimerHandler(milliseconds,asynch_timerhandler,res);
692 static void asynch_sethandlers(Resolver *res) {
693 asynch_sethandlers_generic(res,0,0);
695 static void asynch_cancelhandlers(Resolver *res) {
696 asynch_sethandlers_generic(res,1,0);
698 static void asynch_perturbed(Resolver *res) {
699 asynch_sethandlers_generic(res,0,1);
702 static void asynch_check_now(Resolver *res) {
703 Tcl_Interp *interp= res->interp;
710 Tcl_Obj *results[RESULTLIST_LLEN];
715 if (!res->ads) { /* oh, it has been destroyed! */
721 ec= adns_check(res->ads, &aqu, &answer, &query_v);
722 if (ec==ESRCH || ec==EAGAIN) break;
727 cht_tabledataid_disposing(interp, query, &adnstcl_queries);
729 si= (!answer->status ? &query->on_yes
730 : answer->status > adns_s_max_tempfail ? &query->on_no
733 make_resultlist(interp, answer, results);
735 cht_scriptinv_invoke(si, RESULTLIST_LLEN, results);
736 asynch_query_dispose(interp, query);
739 asynch_sethandlers(res);
744 int cht_do_adns_asynch(ClientData cd, Tcl_Interp *ip,
745 Tcl_Obj *on_yes, Tcl_Obj *on_no,
746 Tcl_Obj *on_fail, Tcl_Obj *xargs,
747 const AdnsTclRRTypeInfo *rrtype, const char *domain,
748 int objc, Tcl_Obj *const *objv, void **result) {
753 query= TALLOC(sizeof(*query));
756 cht_scriptinv_init(&query->on_yes);
757 cht_scriptinv_init(&query->on_no);
758 cht_scriptinv_init(&query->on_fail);
761 rc= query_submit(ip,rrtype,domain,objc,objv,&query->aqu,query,&query->res);
766 rc= cht_scriptinv_set(&query->on_yes, ip,on_yes, xargs); if (rc) goto x_rc;
767 rc= cht_scriptinv_set(&query->on_no, ip,on_no, xargs); if (rc) goto x_rc;
768 rc= cht_scriptinv_set(&query->on_fail,ip,on_fail,xargs); if (rc) goto x_rc;
770 Tcl_IncrRefCount(xargs);
772 query= 0; /* do not dispose */
776 if (query) asynch_query_dispose(ip, query);
777 if (res) asynch_perturbed(res);
781 int cht_do_adns_asynch_cancel(ClientData cd, Tcl_Interp *ip, void *query_v) {
782 Query *query= query_v;
783 Resolver *res= query->res;
784 asynch_query_dispose(ip, query);
785 asynch_perturbed(res);
789 static void asynch_query_dispose(Tcl_Interp *interp, Query *query) {
790 cht_tabledataid_disposing(interp, query, &adnstcl_queries);
791 cht_scriptinv_cancel(&query->on_yes);
792 cht_scriptinv_cancel(&query->on_no);
793 cht_scriptinv_cancel(&query->on_fail);
794 if (query->xargs) Tcl_DecrRefCount(query->xargs);
795 if (query->aqu) adns_cancel(query->aqu);
799 static void destroy_query_idtabcb(Tcl_Interp *interp, void *query_v) {
800 asynch_query_dispose(interp, query_v);
803 const IdDataSpec adnstcl_queries= {
804 "adns", "adns-query-table", destroy_query_idtabcb
807 /*---------- main hooks for tcl ----------*/
809 int cht_do_adnstoplevel_adns(ClientData cd, Tcl_Interp *ip,
810 const Adns_SubCommand *subcmd,
811 int objc, Tcl_Obj *const *objv) {
812 return subcmd->func(0,ip,objc,objv);
815 CHT_INIT(adns, {}, CHTI_COMMANDS(cht_adnstoplevel_entries))