+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;
+ qu->type= type;
+ qu->answer= 0;
+ qu->flags= flags;
+ qu->context= context;
+ qu->udpretries= 0;
+ qu->sentudp= qu->senttcp= 0;
+ qu->nextserver= 0;
+ memcpy(qu->owner,owner,ol); qu->owner[ol]= 0;
+ qu->querymsg= qm= qu->owner+ol+1;
+ memcpy(qm,ads->qbuf,qml);
+ qu->querylen= qml;
+ return qu;
+}
+
+static int failsubmit(adns_state ads, void *context, adns_query *query_r,
+ adns_rrtype type, adns_queryflags flags,
+ int id, adns_status stat) {
+ adns_query qu;
+
+ qu= allocquery(ads,0,0,0,id,type,flags,context); if (!qu) return errno;
+ query_fail(ads,qu,stat);
+ *query_r= qu;
+ return 0;
+}
+
+static void quproc_tosend(adns_state ads, adns_query qu, struct timeval now) {
+ /* Query must be on the `tosend' queue, and guarantees to remove it. */
+ struct sockaddr_in servaddr;
+ int serv;
+
+ if (qu->nextudpserver != -1) {
+ if (qu->udpretries >= UDPMAXRETRIES) {
+ DLIST_UNLINK(ads->tosend,qu);
+ query_fail(ads,qu,adns_s_notresponding);
+ return;
+ }
+ serv= qu->nextudpserver;
+ memset(&servaddr,0,sizeof(servaddr));
+ servaddr.sin_family= AF_INET;
+ servaddr.sin_addr= ads->servers[serv].addr;
+ servaddr.sin_port= htons(53);
+ r= sendto(ads->udpsocket,qu->querymsg,qu->querylen,0,&servaddr,sizeof(servaddr));
+ if (r<0 && errno == EMSGSIZE) {
+ qu->nextudpserver= -1;
+ } else {
+ if (r<0) {
+ diag("sendto %s failed: %s",inet_ntoa(servaddr.sin_addr),strerror(errno));
+ }
+ DLIST_UNLINK(ads->tosend,qu);
+ timevaladd(&now,UDPRETRYMS);
+ qu->timeout= now;
+ qu->sentudp |= (1<<serv);
+ qu->nextudpserver= (serv+1)%ads->nservers;
+ qu->udpretries++;
+ DLIST_LINKTAIL(ads->timew,qu);
+ return;
+ }
+ }
+
+ for (;;) {
+ serv= tcpserver_get(ads);
+ if (serv<0) { r=0; break; }
+ if (ads->opbufused) { r=0; break; }
+ r= write(ads->tcpsocket,qu->querymsg,qu->querylen);
+ if (r >= 0) break;
+ if (errno == EAGAIN || errno == EINTR || errno == ENOSPC ||
+ errno == ENOBUFS || errno == ENOMEM) {
+ r= 0; break;
+ }
+ tcpserver_broken(serv);
+ }
+ if (r < qu->querylen) {
+ newopbufused= qu->opbufused + (qu->querylen-r);
+ if (newopbufused > ads->opbufavail) {
+ newopbufavail= ads->newopbufused<<1;
+ newopbuf= realloc(newopbufavail);
+ if (!newopbuf) {
+ DLIST_UNLINK(ads->tosend,qu);
+ query_fail(ads,qu,adns_s_nolocalmem);
+ return;
+ }
+ ads->opbuf= newopbuf;
+ ads->opbufavail= newopbufavail;
+ }
+ memcpy(ads->opbuf+ads->opbufused,qu->querymsg+r,qu->querylen-r);
+ ads->opbufused= newopbufused;
+ }
+ DLIST_UNLINK(ads->tosend,qu);
+ timevaladd(&now,TCPMS);
+ qu->timeout= now;
+ qu->senttcp |= (1<<qu->nextserver);
+ DLIST_LINKTAIL(ads->timew,qu);
~mdw / adns / blobdiff