X-Git-Url: https://git.distorted.org.uk/~mdw/adns/blobdiff_plain/ae8cc9775a6eb23901a89a2604f73525b0553db8..09d96df2bba590305e08703e232fd8bf780b199f:/src/transmit.c diff --git a/src/transmit.c b/src/transmit.c index 7afb90f..ecea891 100644 --- a/src/transmit.c +++ b/src/transmit.c @@ -74,11 +74,10 @@ static adns_status mkquery_footer(vbuf *vb, adns_rrtype type) { return adns_s_ok; } -adns_status adns__qdpl_normal(adns_state ads, - const char **p_io, const char *pe, int labelnum, - char label_r[], int *ll_io, - adns_queryflags flags, - const typeinfo *typei) { +static adns_status qdparselabel(adns_state ads, + const char **p_io, const char *pe, + char label_r[], int *ll_io, + adns_queryflags flags) { int ll, c; const char *p; @@ -102,13 +101,6 @@ adns_status adns__qdpl_normal(adns_state ads, return adns_s_querydomaininvalid; } } - if (!(flags & adns_qf_quoteok_query)) { - if (c == '-') { - if (!ll) return adns_s_querydomaininvalid; - } else if (!ctype_alpha(c) && !ctype_digit(c)) { - return adns_s_querydomaininvalid; - } - } if (ll == *ll_io) return adns_s_querydomaininvalid; label_r[ll++]= c; } @@ -122,22 +114,22 @@ adns_status adns__mkquery(adns_state ads, vbuf *vb, int *id_r, const char *owner, int ol, const typeinfo *typei, adns_rrtype type, adns_queryflags flags) { - int labelnum, ll, nbytes; + int ll, nbytes; byte label[255]; byte *rqp; const char *p, *pe; adns_status st; + if (!((type^adns_r_addr) & adns_rrt_reprmask)) ads->nextid++; /* bodge */ st= mkquery_header(ads,vb,id_r,ol+2); if (st) return st; MKQUERY_START(vb); p= owner; pe= owner+ol; nbytes= 0; - labelnum= 0; while (p!=pe) { ll= sizeof(label); - st= typei->qdparselabel(ads, &p,pe, labelnum++, label, &ll, flags, typei); + st= qdparselabel(ads, &p,pe, label, &ll, flags); if (st) return st; if (!ll) return adns_s_querydomaininvalid; if (ll > DNS_MAXLABEL) return adns_s_querydomaintoolong; @@ -164,6 +156,7 @@ adns_status adns__mkquery_frdgram(adns_state ads, vbuf *vb, int *id_r, int lablen, labstart; adns_status st; + if (!((type^adns_r_addr) & adns_rrt_reprmask)) ads->nextid++; /* bodge */ st= mkquery_header(ads,vb,id_r,qd_dglen); if (st) return st; MKQUERY_START(vb); @@ -250,9 +243,9 @@ static void query_usetcp(adns_query qu, struct timeval now) { } void adns__query_send(adns_query qu, struct timeval now) { - struct sockaddr_in servaddr; int serv, r; adns_state ads; + adns_rr_addr *addr; assert(qu->state == query_tosend); if ((qu->flags & adns_qf_usevc) || (qu->query_dglen > DNS_MAXUDP)) { @@ -265,16 +258,12 @@ void adns__query_send(adns_query qu, struct timeval now) { return; } - serv= qu->udpnextserver; - memset(&servaddr,0,sizeof(servaddr)); - ads= qu->ads; - servaddr.sin_family= AF_INET; - servaddr.sin_addr= ads->servers[serv].addr; - servaddr.sin_port= htons(DNS_PORT); + serv= qu->udpnextserver; + addr= &ads->servers[serv]; r= sendto(ads->udpsocket,qu->query_dgram,qu->query_dglen,0, - (const struct sockaddr*)&servaddr,sizeof(servaddr)); + &addr->addr.sa,addr->len); if (r<0 && errno == EMSGSIZE) { qu->retries= 0; query_usetcp(qu,now);