int dns_parse_srv(dnscc_t *qdn, dnscc_t *pkt, dnscc_t *cur, dnscc_t *end, void **result) { struct dns_rr_srv *ret; struct dns_parse p; struct dns_rr rr; int r, l; char *sp; dnsc_t srv[DNS_MAXDN]; assert(dns_get16(cur+2) == DNS_C_IN && dns_get16(cur+0) == DNS_T_SRV); /* first, validate the answer and count size of the result */ l = 0; dns_initparse(&p, qdn, pkt, cur, end); while((r = dns_nextrr(&p, &rr)) > 0) { cur = rr.dnsrr_dptr + 6; r = dns_getdn(pkt, &cur, end, srv, sizeof(srv)); if (r <= 0 || cur != rr.dnsrr_dend) return DNS_E_PROTOCOL; l += dns_dntop_size(srv); } if (r < 0) return DNS_E_PROTOCOL; if (!p.dnsp_nrr) return DNS_E_NODATA; /* next, allocate and set up result */ l += dns_stdrr_size(&p); ret = malloc(sizeof(*ret) + sizeof(struct dns_srv) * p.dnsp_nrr + l); if (!ret) return DNS_E_NOMEM; ret->dnssrv_nrr = p.dnsp_nrr; ret->dnssrv_srv = (struct dns_srv *)(ret+1); /* and 3rd, fill in result, finally */ sp = (char*)(ret->dnssrv_srv + p.dnsp_nrr); for (dns_rewind(&p, qdn), r = 0; dns_nextrr(&p, &rr); ++r) { ret->dnssrv_srv[r].name = sp; cur = rr.dnsrr_dptr; ret->dnssrv_srv[r].priority = dns_get16(cur); ret->dnssrv_srv[r].weight = dns_get16(cur+2); ret->dnssrv_srv[r].port = dns_get16(cur+4); cur += 6; dns_getdn(pkt, &cur, end, srv, sizeof(srv)); sp += dns_dntop(srv, sp, DNS_MAXNAME); } dns_stdrr_finish((struct dns_rr_null *)ret, sp, &p); *result = ret; return 0; }
int main(int argc, char **argv) { int i; int sz; dnsc_t dn[DNS_MAXDN+10]; dnsc_t *dl, *dp; int isabs; sz = (argc > 1) ? atoi(argv[1]) : 0; for(i = 2; i < argc; ++i) { int r = dns_ptodn(argv[i], 0, dn, sz, &isabs); printf("%s: ", argv[i]); if (r < 0) printf("error\n"); else if (!r) printf("buffer too small\n"); else { printf("len=%d dnlen=%d size=%d name:", r, dns_dnlen(dn), dns_dntop_size(dn)); dl = dn; while(*dl) { printf(" %d=", *dl); dp = dl + 1; dl = dp + *dl; while(dp < dl) { if (*dp <= ' ' || *dp >= 0x7f) printf("\\%03d", *dp); else if (*dp == '.' || *dp == '\\') printf("\\%c", *dp); else putchar(*dp); ++dp; } } if (isabs) putchar('.'); putchar('\n'); } } return 0; }