X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ian/git?p=secnet.git;a=blobdiff_plain;f=resolver.c;h=a39e9fcde510dea5e96bf81dcf05aad079ea982d;hp=6088c562ed5c3d603756fb793f88fa46b8cc6466;hb=4ac7fd3ff73c98100d8a8546668d0212e4af3ffa;hpb=3f2c70f6b6d5ba48ff9fbe7887eb2e77ea128220 diff --git a/resolver.c b/resolver.c index 6088c56..a39e9fc 100644 --- a/resolver.c +++ b/resolver.c @@ -20,6 +20,7 @@ struct adns { struct query { void *cst; + const char *name; int port; struct comm_if *comm; resolve_answer_fn *answer; @@ -59,17 +60,17 @@ static bool_t resolve_request(void *sst, cstring_t name, snprintf(msg,sizeof(msg),"invalid address literal: %s", strerror(rv)); msg[sizeof(msg)-1]=0; - cb(cst,0,0,0,msg); + cb(cst,0,0,0,name,msg); } else { - cb(cst,&ca,1,1,0); + cb(cst,&ca,1,1,name,0); } #else ca.ia.sin.sin_family=AF_INET; ca.ia.sin.sin_port=htons(port); if (inet_aton(trimmed,&ca.ia.sin.sin_addr)) - cb(cst,&ca,1,1,0); + cb(cst,&ca,1,1,name,0); else - cb(cst,0,0,0,"invalid IP address"); + cb(cst,0,0,0,name,"invalid IP address"); #endif return True; } @@ -78,6 +79,7 @@ static bool_t resolve_request(void *sst, cstring_t name, q->cst=cst; q->comm=comm; q->port=port; + q->name=name; q->answer=cb; rv=adns_submit(st->ast, name, adns_r_addr, 0, q, &q->query); @@ -116,7 +118,7 @@ static void resolver_afterpoll(void *sst, struct pollfd *fds, int nfds) if (rv==0) { q=qp; if (ans->status!=adns_s_ok) { - q->answer(q->cst,NULL,0,0,adns_strerror(ans->status)); + q->answer(q->cst,NULL,0,0,q->name,adns_strerror(ans->status)); free(q); free(ans); } else { @@ -148,7 +150,7 @@ static void resolver_afterpoll(void *sst, struct pollfd *fds, int nfds) memcpy(&ca->ia,&ra->addr,ra->len); wslot++; } - q->answer(q->cst,ca_buf,wslot,total,0); + q->answer(q->cst,ca_buf,wslot,total,q->name,0); free(q); free(ans); } @@ -198,7 +200,7 @@ static list_t *adnsresolver_apply(closure_t *self, struct cloc loc, } register_for_poll(st, resolver_beforepoll, resolver_afterpoll, - ADNS_POLLFDS_RECOMMENDED+5,"resolver"); + "resolver"); return new_closure(&st->cl); }