X-Git-Url: https://git.distorted.org.uk/~mdw/adns/blobdiff_plain/fc6a52aea07419d4eb1a42458ba2459faf52e4b9..02b47faadfe03c3b302dadb78422e14648f07b50:/client/adnstest.c diff --git a/client/adnstest.c b/client/adnstest.c index a293d3b..cbb896a 100644 --- a/client/adnstest.c +++ b/client/adnstest.c @@ -98,8 +98,16 @@ static int consistsof(const char *string, const char *accept) { } int main(int argc, char *const *argv) { + struct myctx { + adns_query qu; + int doneyet, found; + const char *fdom; + }; + adns_state ads; - adns_query *qus, qu; + adns_query qu; + struct myctx *mcs, *mc, *mcw; + void *mcr; adns_answer *ans; const char *initstring, *rrtn, *fmtn; const char *const *fdomlist, *domain; @@ -130,7 +138,7 @@ int main(int argc, char *const *argv) { initflagsnum= strtoul(initflags,&ep,0); if (*ep == ',') { owninitflags= ep+1; - if (!consistsof(owninitflags,"p")) { + if (!consistsof(owninitflags,"ps")) { fputs("unknown owninitflag\n",stderr); exit(4); } @@ -154,6 +162,7 @@ int main(int argc, char *const *argv) { " [ :,... ]\n" " [ [[,]/] ... ]\n" "initflags: p use poll(2) instead of select(2)\n" + " s use adns_wait with specified query, instead of 0\n" "queryflags: a print status abbrevs instead of strings\n", stderr); exit(4); @@ -173,8 +182,8 @@ int main(int argc, char *const *argv) { for (qc=0; fdomlist[qc]; qc++); for (tc=0; types[tc] != adns_r_none; tc++); - qus= malloc(sizeof(qus)*qc*tc); - if (!qus) { perror("malloc qus"); exit(3); } + mcs= malloc(sizeof(*mcs)*qc*tc); + if (!mcs) { perror("malloc mcs"); exit(3); } if (initstring) { r= adns_init_strcfg(&ads, @@ -197,11 +206,16 @@ int main(int argc, char *const *argv) { exit(4); } for (ti=0; tidoneyet= 0; + mc->fdom= fdomlist[qi]; + fprintf(stdout,"%s flags %d type %d",domain,qflags,types[ti]); - r= adns_submit(ads,domain,types[ti],qflags,0,&qus[qi*tc+ti]); + r= adns_submit(ads,domain,types[ti],qflags,mc,&mc->qu); if (r == adns_s_unknownrrtype) { fprintf(stdout," not implemented\n"); - qus[qi*tc+ti]= 0; + mc->qu= 0; + mc->doneyet= 1; } else if (r) { failure_errno("submit",r); } else { @@ -213,64 +227,99 @@ int main(int argc, char *const *argv) { } } - for (qi=0; qifound= 0; + } + } + for (adns_forallqueries_begin(ads); + (qu= adns_forallqueries_next(ads,&mcr)); + ) { + mc= mcr; + assert(qu == mc->qu); + assert(!mc->doneyet); + mc->found= 1; + } + mcw= 0; + for (qi=0; qidoneyet) continue; + assert(mc->found); + if (!mcw) mcw= mc; + } + } + if (!mcw) break; + + if (strchr(owninitflags,'s')) { + qu= mcw->qu; + mc= mcw; + } else { + qu= 0; + mc= 0; + } - if (strchr(owninitflags,'p')) { + if (strchr(owninitflags,'p')) { + for (;;) { + r= adns_check(ads,&qu,&ans,&mcr); + if (r != EWOULDBLOCK) break; for (;;) { - r= adns_check(ads,&qu,&ans,0); - if (r != EWOULDBLOCK) break; - for (;;) { - npollfds= npollfdsavail; - timeout= -1; - r= adns_beforepoll(ads, pollfds, &npollfds, &timeout, 0); - if (r != ERANGE) break; - pollfds= realloc(pollfds,sizeof(*pollfds)*npollfds); - if (!pollfds) failure_errno("realloc pollfds",errno); - npollfdsavail= npollfds; - } - if (r) failure_errno("beforepoll",r); - r= poll(pollfds,npollfds,timeout); - if (r == -1) failure_errno("poll",errno); - adns_afterpoll(ads,pollfds, r?npollfds:0, 0); + npollfds= npollfdsavail; + timeout= -1; + r= adns_beforepoll(ads, pollfds, &npollfds, &timeout, 0); + if (r != ERANGE) break; + pollfds= realloc(pollfds,sizeof(*pollfds)*npollfds); + if (!pollfds) failure_errno("realloc pollfds",errno); + npollfdsavail= npollfds; } - } else { - r= adns_wait(ads,&qu,&ans,0); + if (r) failure_errno("beforepoll",r); + r= poll(pollfds,npollfds,timeout); + if (r == -1) failure_errno("poll",errno); + adns_afterpoll(ads,pollfds, r?npollfds:0, 0); } - if (r) failure_errno("wait/check",r); + } else { + r= adns_wait(ads,&qu,&ans,&mcr); + } + if (r) failure_errno("wait/check",r); + + if (mc) assert(mcr==mc); + else mc= mcr; + assert(qu==mc->qu); + assert(!mc->doneyet); + + fdom_split(mc->fdom,&domain,&qflags,ownflags,sizeof(ownflags)); - if (gettimeofday(&now,0)) { perror("gettimeofday"); exit(3); } + if (gettimeofday(&now,0)) { perror("gettimeofday"); exit(3); } - ri= adns_rr_info(ans->type, &rrtn,&fmtn,&len, 0,0); - fprintf(stdout, "%s flags %d type ",domain,qflags); - dumptype(ri,rrtn,fmtn); - fprintf(stdout, "%s%s: %s; nrrs=%d; cname=%s; owner=%s; ttl=%ld\n", - ownflags[0] ? " ownflags=" : "", ownflags, - strchr(ownflags,'a') - ? adns_errabbrev(ans->status) - : adns_strerror(ans->status), - ans->nrrs, - ans->cname ? ans->cname : "$", - ans->owner ? ans->owner : "$", - (long)ans->expires - (long)now.tv_sec); - if (ans->nrrs) { - assert(!ri); - for (i=0; inrrs; i++) { - r= adns_rr_info(ans->type, 0,0,0, ans->rrs.bytes + i*len, &show); - if (r) failure_status("info",r); - fprintf(stdout," %s\n",show); - free(show); - } + ri= adns_rr_info(ans->type, &rrtn,&fmtn,&len, 0,0); + fprintf(stdout, "%s flags %d type ",domain,qflags); + dumptype(ri,rrtn,fmtn); + fprintf(stdout, "%s%s: %s; nrrs=%d; cname=%s; owner=%s; ttl=%ld\n", + ownflags[0] ? " ownflags=" : "", ownflags, + strchr(ownflags,'a') + ? adns_errabbrev(ans->status) + : adns_strerror(ans->status), + ans->nrrs, + ans->cname ? ans->cname : "$", + ans->owner ? ans->owner : "$", + (long)ans->expires - (long)now.tv_sec); + if (ans->nrrs) { + assert(!ri); + for (i=0; inrrs; i++) { + r= adns_rr_info(ans->type, 0,0,0, ans->rrs.bytes + i*len, &show); + if (r) failure_status("info",r); + fprintf(stdout," %s\n",show); + free(show); } - free(ans); } + free(ans); + + mc->doneyet= 1; } - free(qus); + free(mcs); adns_finish(ads); exit(0);