+ for (qu= ads->timew; qu; qu= qu->next) {
+ if (qu->sentudp) {
+ inter_addfd(maxfd,readfds,ads->udpsocket);
+ break;
+ }
+ }
+ switch (ads->tcpstate) {
+ case server_disc:
+ break;
+ case server_connecting:
+ inter_addfd(maxfd,readfds,ads->tcpsocket);
+ inter_addfd(maxfd,writefds,ads->tcpsocket);
+ inter_addfd(maxfd,exceptfds,ads->tcpsocket);
+ break;
+ case server_connected:
+ inter_addfd(maxfd,readfds,ads->tcpsocket);
+ inter_addfd(maxfd,exceptfds,ads->tcpsocket);
+ if (ads->opbufused) inter_addfd(maxfd,writefds,ads->tcpsocket);
+ default:
+ abort();
+ }
+
+}
+
+static int internal_check(adns_state ads,
+ adns_query *query_io,
+ adns_answer **answer,
+ void **context_r) {
+ adns_query qu;
+
+ qu= *query_io;
+ if (!qu) {
+ if (!ads->output.head) return EWOULDBLOCK;
+ qu= ads->output.head;
+ } else {
+ if (qu->id>=0) return EWOULDBLOCK;
+ }
+ LIST_UNLINK(ads->output,qu);
+ *answer= qu->answer;
+ if (context_r) *context_r= qu->context;
+ free(qu);
+ return 0;
+}
+
+int adns_wait(adns_state ads,
+ adns_query *query_io,
+ adns_answer **answer_r,
+ void **context_r) {
+ int r, maxfd, rsel, rcb;
+ fd_set readfds, writefds, exceptfds;
+ struct timeval tvbuf, *tvp;
+
+ for (;;) {
+ r= internal_check(ads,query_io,answer_r,context_r);
+ if (r && r != EWOULDBLOCK) return r;
+ FD_ZERO(&readfds); FD_ZERO(&writefds); FD_ZERO(&exceptfds);
+ maxfd= 0; tvp= 0;
+ adns_interest(ads,&maxfd,&readfds,&writefds,&exceptfds,&tvp,&tvbuf);
+ rsel= select(maxfd,&readfds,&writefds,&exceptfds,tvp);
+ if (rsel==-1) return r;
+ rcb= adns_callback(ads,maxfd,&readfds,&writefds,&exceptfds);
+ assert(rcb==rsel);
+ }
+}
+
+int adns_check(adns_state ads,
+ adns_query *query_io,
+ adns_answer **answer_r,
+ void **context_r) {
+ autosys(ads);
+ return internal_check(ads,query_io,answer_r,context_r);
+}
+
+int adns_synchronous(adns_state ads,
+ const char *owner,
+ adns_rrtype type,
+ adns_queryflags flags,
+ adns_answer **answer_r) {
+ adns_query qu;
+ int r;
+
+ r= adns_submit(ads,owner,type,flags,0,&qu);
+ if (r) return r;
+
+ do {
+ r= adns_wait(ads,&qu,answer_r,0);
+ } while (r==EINTR);
+ if (r) adns_cancel(ads,qu);
+ return r;
+}
+
+static adns_status mkquery(adns_state ads, const char *owner, int ol, int id,
+ adns_rrtype type, adns_queryflags flags, int *qml_r) {
+ int ll, c, nlabs, qbufreq;
+ unsigned char label[255], *nqbuf;
+ const char *p, *pe;
+
+#define MKQUERY_ADDB(b) *nqbuf++= (b)
+#define MKQUERY_ADDW(w) (MKQUERY_ADDB(((w)>>8)&0x0ff), MKQUERY_ADDB((w)&0x0ff))
+
+ qbufreq= 12+strlen(owner)+3;
+ if (ads->qbufavail < qbufreq) {
+ nqbuf= realloc(ads->qbuf,qbufreq);
+ if (!nqbuf) return adns_s_nolocalmem;
+ ads->qbuf= nqbuf; ads->qbufavail= qbufreq;
+ }
+ nqbuf= ads->qbuf;
+
+ MKQUERY_ADDW(id);
+ MKQUERY_ADDB(0x01); /* QR=Q(0), OPCODE=QUERY(0000), !AA, !TC, RD */
+ MKQUERY_ADDB(0x00); /* !RA, Z=000, RCODE=NOERROR(0000) */
+ MKQUERY_ADDW(1); /* QDCOUNT=1 */
+ MKQUERY_ADDW(0); /* ANCOUNT=0 */
+ MKQUERY_ADDW(0); /* NSCOUNT=0 */
+ MKQUERY_ADDW(0); /* ARCOUNT=0 */
+ p= owner; pe= owner+ol;
+ nlabs= 0;
+ if (!*p) return adns_s_invaliddomain;
+ do {
+ ll= 0;
+ while (p!=pe && (c= *p++)!='.') {
+ if (c=='\\') {
+ if (!(flags & adns_f_anyquote)) return adns_s_invaliddomain;
+ if (ctype_digit(p[0])) {
+ if (ctype_digit(p[1]) && ctype_digit(p[2])) {
+ c= (*p++ - '0')*100 + (*p++ - '0')*10 + (*p++ - '0');
+ if (c >= 256) return adns_s_invaliddomain;
+ } else {
+ return adns_s_invaliddomain;
+ }
+ } else if (!(c= *p++)) {
+ return adns_s_invaliddomain;
+ }
+ }
+ if (!(flags & adns_f_anyquote)) {
+ if ((c >= '0' && c <= '9') || c == '-') {
+ if (!ll) return adns_s_invaliddomain;
+ } else if ((c < 'a' || c > 'z') && (c < 'A' && c > 'Z')) {
+ return adns_s_invaliddomain;
+ }
+ }
+ if (ll == sizeof(label)) return adns_s_invaliddomain;
+ label[ll++]= c;
+ }
+ if (!ll) return adns_s_invaliddomain;
+ if (nlabs++ > 63) return adns_s_invaliddomain;
+ MKQUERY_ADDB(ll);
+ memcpy(nqbuf,label,ll); nqbuf+= ll;
+ } while (p!=pe);
+
+ MKQUERY_ADDB(0);
+ MKQUERY_ADDW(type & adns__rrt_typemask); /* QTYPE */
+ MKQUERY_ADDW(1); /* QCLASS=IN */
+
+ *qml_r= nqbuf - ads->qbuf;
+
+ return adns_s_ok;
+}
+
+static adns_query allocquery(adns_state ads, const char *owner, int ol,
+ int qml, int id, adns_rrtype type,
+ adns_queryflags flags, void *context) {
+ adns_query qu;
+ unsigned char *qm;
+
+ qu= malloc(sizeof(*qu)+ol+1+qml); if (!qu) return 0;
+ qu->next= qu->back= qu->parent= 0;
+ qu->children.head= qu->children.tail= 0;
+ qu->siblings.next= qu->siblings.back= 0;
+ qu->id= id;
~mdw / adns / blobdiff