dnssec_return_validation_chain without ldns

This commit is contained in:
Willem Toorop 2015-03-16 17:05:03 +01:00
parent 70cb26bb00
commit 9942550748
3 changed files with 144 additions and 98 deletions

View File

@ -48,6 +48,7 @@
#include "dnssec.h" #include "dnssec.h"
#include "rr-dict.h" #include "rr-dict.h"
#include "gldns/str2wire.h" #include "gldns/str2wire.h"
#include "gldns/wire2str.h"
void priv_getdns_call_user_callback(getdns_dns_req *, struct getdns_dict *); void priv_getdns_call_user_callback(getdns_dns_req *, struct getdns_dict *);
@ -61,7 +62,7 @@ struct validation_chain {
struct chain_response { struct chain_response {
int err; int err;
ldns_rr_list *result; getdns_list *result;
int sec; int sec;
char *bogus; char *bogus;
struct validation_chain *chain; struct validation_chain *chain;
@ -74,17 +75,19 @@ struct chain_link {
struct chain_response DS; struct chain_response DS;
}; };
static void launch_chain_link_lookup(struct validation_chain *chain, char *name); static void launch_chain_link_lookup(struct validation_chain *chain,
priv_getdns_rdf_iter *rdf_dname);
static void destroy_chain(struct validation_chain *chain); static void destroy_chain(struct validation_chain *chain);
static void callback_on_complete_chain(struct validation_chain *chain) static void callback_on_complete_chain(struct validation_chain *chain)
{ {
struct getdns_context *context = chain->dns_req->context; getdns_context *context = chain->dns_req->context;
struct getdns_dict *response; getdns_dict *response;
struct chain_link *link; struct chain_link *link;
size_t ongoing = chain->lock; size_t ongoing = chain->lock;
ldns_rr_list *keys; getdns_list *keys;
struct getdns_list *getdns_keys; size_t i;
getdns_dict *rr_dict;
RBTREE_FOR(link, struct chain_link *, RBTREE_FOR(link, struct chain_link *,
(getdns_rbtree_t *)&(chain->root)) { (getdns_rbtree_t *)&(chain->root)) {
@ -95,24 +98,29 @@ static void callback_on_complete_chain(struct validation_chain *chain)
((const char *)link->node.key)[1] != '\0' )) ((const char *)link->node.key)[1] != '\0' ))
ongoing++; ongoing++;
} }
if (ongoing == 0) { if (ongoing > 0)
getdns_dns_req *dns_req = chain->dns_req; return;
response = create_getdns_response(chain->dns_req);
keys = ldns_rr_list_new(); if (!(response = create_getdns_response(chain->dns_req)) ||
RBTREE_FOR(link, struct chain_link *, !(keys = getdns_list_create_with_context(context))) {
(getdns_rbtree_t *)&(chain->root)) {
(void) ldns_rr_list_cat(keys, link->DNSKEY.result); priv_getdns_call_user_callback(chain->dns_req, response);
(void) ldns_rr_list_cat(keys, link->DS.result);
}
getdns_keys = create_list_from_rr_list(context, keys);
(void) getdns_dict_set_list(response, "validation_chain",
getdns_keys);
getdns_list_destroy(getdns_keys);
ldns_rr_list_free(keys);
priv_getdns_call_user_callback(dns_req, response);
destroy_chain(chain); destroy_chain(chain);
return;
} }
RBTREE_FOR(link, struct chain_link *,
(getdns_rbtree_t *)&(chain->root)) {
for (i = 0; !getdns_list_get_dict( link->DNSKEY.result
, i, &rr_dict); i++)
(void) getdns_list_append_dict(keys, rr_dict);
for (i = 0; !getdns_list_get_dict( link->DS.result
, i, &rr_dict); i++)
(void) getdns_list_append_dict(keys, rr_dict);
}
(void) getdns_dict_set_list(response, "validation_chain", keys);
getdns_list_destroy(keys);
priv_getdns_call_user_callback(chain->dns_req, response);
destroy_chain(chain);
} }
@ -120,58 +128,71 @@ static void
ub_chain_response_callback(void *arg, int err, struct ub_result* ub_res) ub_chain_response_callback(void *arg, int err, struct ub_result* ub_res)
{ {
struct chain_response *response = (struct chain_response *) arg; struct chain_response *response = (struct chain_response *) arg;
ldns_status r; getdns_context *context = response->chain->dns_req->context;
ldns_pkt *p; priv_getdns_rr_iter rr_iter_storage, *rr_iter;
ldns_rr_list *answer; priv_getdns_rdf_iter rdf_storage, *rdf;
ldns_rr_list *keys; gldns_pkt_section section;
size_t i; uint16_t rr_type, type_covered;
getdns_dict *rr_dict;
getdns_list *keys;
size_t nkeys;
response->err = err; response->err = err;
response->sec = ub_res ? ub_res->secure : 0; response->sec = ub_res ? ub_res->secure : 0;
response->bogus = ub_res ? ub_res->why_bogus : NULL; response->bogus = ub_res ? ub_res->why_bogus : NULL;
if (ub_res == NULL) if (ub_res == NULL || ub_res->answer_packet == NULL ||
goto done; !(keys = getdns_list_create_with_context(context)))
goto done;
r = ldns_wire2pkt(&p, ub_res->answer_packet, ub_res->answer_len); for ( rr_iter = priv_getdns_rr_iter_init(&rr_iter_storage
if (r != LDNS_STATUS_OK) { , ub_res->answer_packet
if (err == 0) , ub_res->answer_len)
response->err = r; ; rr_iter
goto done; ; rr_iter = priv_getdns_rr_iter_next(rr_iter)
} ) {
section = priv_getdns_rr_iter_section(rr_iter);
keys = ldns_rr_list_new(); if (section != GLDNS_SECTION_ANSWER)
answer = ldns_pkt_answer(p);
for (i = 0; i < ldns_rr_list_rr_count(answer); i++) {
ldns_rr *rr = ldns_rr_list_rr(answer, i);
if (ldns_rr_get_type(rr) == LDNS_RR_TYPE_DNSKEY ||
ldns_rr_get_type(rr) == LDNS_RR_TYPE_DS) {
(void) ldns_rr_list_push_rr(keys, ldns_rr_clone(rr));
continue; continue;
rr_type = gldns_read_uint16(rr_iter->rr_type);
if (rr_type == GETDNS_RRTYPE_DS ||
rr_type == GETDNS_RRTYPE_DNSKEY) {
if (!(rr_dict = priv_getdns_rr_iter2rr_dict(context, rr_iter)))
continue;
if (getdns_list_append_dict(keys, rr_dict))
break;
} }
if (ldns_rr_get_type(rr) != LDNS_RR_TYPE_RRSIG) if (rr_type != GETDNS_RRTYPE_RRSIG)
continue; continue;
if (ldns_read_uint16(ldns_rdf_data(ldns_rr_rdf(rr, 0))) == if (!(rdf = priv_getdns_rdf_iter_init(&rdf_storage, rr_iter)))
LDNS_RR_TYPE_DS)
launch_chain_link_lookup(response->chain,
ldns_rdf2str(ldns_rr_rdf(rr, 7)));
else if (ldns_read_uint16(ldns_rdf_data(ldns_rr_rdf(rr, 0))) !=
LDNS_RR_TYPE_DNSKEY)
continue; continue;
(void) ldns_rr_list_push_rr(keys, ldns_rr_clone(rr)); type_covered = gldns_read_uint16(rdf->pos);
if (type_covered == GETDNS_RRTYPE_DS) {
if ((rdf = priv_getdns_rdf_iter_init_at(
&rdf_storage, rr_iter, 7)))
launch_chain_link_lookup(response->chain, rdf);
} else if (type_covered != GETDNS_RRTYPE_DNSKEY)
continue;
if (!(rr_dict = priv_getdns_rr_iter2rr_dict(context, rr_iter)))
continue;
if (getdns_list_append_dict(keys, rr_dict))
break;
} }
if (ldns_rr_list_rr_count(keys)) if (getdns_list_get_length(keys, &nkeys))
response->result = keys; getdns_list_destroy(keys);
else
ldns_rr_list_free(keys);
ldns_pkt_free(p); else if (!nkeys)
ub_resolve_free(ub_res); getdns_list_destroy(keys);
else
response->result = keys;
ub_resolve_free(ub_res);
done: if (response->err == 0 && response->result == NULL) done: if (response->err == 0 && response->result == NULL)
response->err = -1; response->err = -1;
@ -195,23 +216,35 @@ resolve(char* name, int rrtype, struct chain_response *response)
{ {
return ub_resolve_async( return ub_resolve_async(
response->chain->dns_req->context->unbound_ctx, response->chain->dns_req->context->unbound_ctx,
name, rrtype, LDNS_RR_CLASS_IN, response, name, rrtype, GETDNS_RRCLASS_IN, response,
ub_chain_response_callback, &response->unbound_id); ub_chain_response_callback, &response->unbound_id);
} }
static void static void
launch_chain_link_lookup(struct validation_chain *chain, char *name) launch_chain_link_lookup(
struct validation_chain *chain, priv_getdns_rdf_iter *rdf_dname)
{ {
int r; int r;
struct chain_link *link = (struct chain_link *) struct chain_link *link;
getdns_rbtree_search((getdns_rbtree_t *)&(chain->root), name); uint8_t dname_spc[256], *dname;
char name[1024];
if (link) { size_t dname_spc_sz = sizeof(dname_spc);
free(name);
if (!(dname = priv_getdns_rdf_if_or_as_decompressed(
rdf_dname, dname_spc, &dname_spc_sz)))
return; return;
}
if (!gldns_wire2str_dname_buf(dname, (dname == dname_spc ?
sizeof(dname_spc) : rdf_dname->nxt - rdf_dname->pos),
name, sizeof(name)))
return;
if ((link = (struct chain_link *)
getdns_rbtree_search((getdns_rbtree_t *)&(chain->root), name)))
return;
link = GETDNS_MALLOC(chain->mf, struct chain_link); link = GETDNS_MALLOC(chain->mf, struct chain_link);
link->node.key = name; link->node.key = getdns_strdup(&chain->mf, name);
chain_response_init(chain, &link->DNSKEY); chain_response_init(chain, &link->DNSKEY);
chain_response_init(chain, &link->DS); chain_response_init(chain, &link->DS);
@ -219,12 +252,12 @@ launch_chain_link_lookup(struct validation_chain *chain, char *name)
getdns_rbtree_insert(&(chain->root), (getdns_rbnode_t *)link); getdns_rbtree_insert(&(chain->root), (getdns_rbnode_t *)link);
chain->lock++; chain->lock++;
r = resolve(name, LDNS_RR_TYPE_DNSKEY, &link->DNSKEY); r = resolve(name, GETDNS_RRTYPE_DNSKEY, &link->DNSKEY);
if (r != 0) if (r != 0)
link->DNSKEY.err = r; link->DNSKEY.err = r;
if (name[0] != '.' || name[1] != '\0') { if (name[0] != '.' || name[1] != '\0') {
r = resolve(name, LDNS_RR_TYPE_DS, &link->DS); r = resolve(name, GETDNS_RRTYPE_DS, &link->DS);
if (r != 0) if (r != 0)
link->DS.err = r; link->DS.err = r;
} }
@ -258,8 +291,8 @@ static void destroy_chain_link(getdns_rbnode_t * node, void *arg)
struct validation_chain *chain = (struct validation_chain*) arg; struct validation_chain *chain = (struct validation_chain*) arg;
free((void *)link->node.key); free((void *)link->node.key);
ldns_rr_list_deep_free(link->DNSKEY.result); getdns_list_destroy(link->DNSKEY.result);
ldns_rr_list_deep_free(link->DS.result); getdns_list_destroy(link->DS.result);
GETDNS_FREE(chain->mf, link); GETDNS_FREE(chain->mf, link);
} }
@ -275,6 +308,10 @@ getdns_get_validation_chain(getdns_dns_req *dns_req, uint64_t *timeout)
{ {
getdns_network_req **netreq_p, *netreq; getdns_network_req **netreq_p, *netreq;
struct validation_chain *chain = create_chain(dns_req, timeout); struct validation_chain *chain = create_chain(dns_req, timeout);
priv_getdns_rr_iter rr_iter_storage, *rr_iter;
priv_getdns_rdf_iter rdf_storage, *rdf;
gldns_pkt_section section;
uint16_t rr_type;
if (! chain) { if (! chain) {
priv_getdns_call_user_callback( priv_getdns_call_user_callback(
@ -282,32 +319,28 @@ getdns_get_validation_chain(getdns_dns_req *dns_req, uint64_t *timeout)
return; return;
} }
for (netreq_p = dns_req->netreqs; (netreq = *netreq_p); netreq_p++) { for (netreq_p = dns_req->netreqs; (netreq = *netreq_p); netreq_p++) {
size_t i;
ldns_status r;
ldns_pkt *pkt;
ldns_rr_list *answer;
ldns_rr_list *authority;
if ((r = ldns_wire2pkt(&pkt, for ( rr_iter = priv_getdns_rr_iter_init(&rr_iter_storage
netreq->response, netreq->response_len))) , netreq->response
continue; , netreq->response_len)
; rr_iter
answer = ldns_pkt_answer(pkt); ; rr_iter = priv_getdns_rr_iter_next(rr_iter)
authority = ldns_pkt_authority(pkt); ) {
section = priv_getdns_rr_iter_section(rr_iter);
if (section != GLDNS_SECTION_ANSWER &&
section != GLDNS_SECTION_AUTHORITY)
continue;
for (i = 0; i < ldns_rr_list_rr_count(answer); i++) { rr_type = gldns_read_uint16(rr_iter->rr_type);
ldns_rr *rr = ldns_rr_list_rr(answer, i); if (rr_type != GETDNS_RRTYPE_RRSIG)
if (ldns_rr_get_type(rr) == LDNS_RR_TYPE_RRSIG) continue;
launch_chain_link_lookup(chain,
ldns_rdf2str(ldns_rr_rdf(rr, 7))); if (!(rdf = priv_getdns_rdf_iter_init_at(
&rdf_storage , rr_iter, 7)))
continue;
launch_chain_link_lookup(chain, rdf);
} }
for (i = 0; i < ldns_rr_list_rr_count(authority); i++) {
ldns_rr *rr = ldns_rr_list_rr(authority, i);
if (ldns_rr_get_type(rr) == LDNS_RR_TYPE_RRSIG)
launch_chain_link_lookup(chain,
ldns_rdf2str(ldns_rr_rdf(rr, 7)));
}
ldns_pkt_free(pkt);
} }
callback_on_complete_chain(chain); callback_on_complete_chain(chain);
} }

View File

@ -310,6 +310,16 @@ done:
return NULL; return NULL;
} }
priv_getdns_rdf_iter *
priv_getdns_rdf_iter_init_at(
priv_getdns_rdf_iter *i, priv_getdns_rr_iter *rr, size_t pos)
{
for ( i = priv_getdns_rdf_iter_init(i, rr)
; i && pos
; i = priv_getdns_rdf_iter_next(i), pos--);
return i;
}
uint8_t * uint8_t *
priv_getdns_rdf_if_or_as_decompressed( priv_getdns_rdf_if_or_as_decompressed(
priv_getdns_rdf_iter *i, uint8_t *ff_bytes, size_t *len) priv_getdns_rdf_iter *i, uint8_t *ff_bytes, size_t *len)

View File

@ -99,6 +99,9 @@ priv_getdns_rdf_iter *priv_getdns_rdf_iter_init(priv_getdns_rdf_iter *i,
priv_getdns_rdf_iter *priv_getdns_rdf_iter_next(priv_getdns_rdf_iter *i); priv_getdns_rdf_iter *priv_getdns_rdf_iter_next(priv_getdns_rdf_iter *i);
priv_getdns_rdf_iter *priv_getdns_rdf_iter_init_at(priv_getdns_rdf_iter *i,
priv_getdns_rr_iter *rr, size_t pos);
uint8_t *priv_getdns_rdf_if_or_as_decompressed( uint8_t *priv_getdns_rdf_if_or_as_decompressed(
priv_getdns_rdf_iter *i, uint8_t *ff_bytes, size_t *len); priv_getdns_rdf_iter *i, uint8_t *ff_bytes, size_t *len);