mirror of
https://github.com/NLnetLabs/unbound.git
synced 2025-12-20 23:00:56 -05:00
iterator and dns cache work.
git-svn-id: file:///svn/unbound/trunk@342 be551aaa-1e26-0410-a405-d3ace91eadb9
This commit is contained in:
parent
265993017d
commit
5def8556c6
10 changed files with 580 additions and 8 deletions
|
|
@ -5,6 +5,8 @@
|
||||||
- packed rrset key has type and class as easily accessable struct
|
- packed rrset key has type and class as easily accessable struct
|
||||||
members. They are still kept in network format for fast msg encode.
|
members. They are still kept in network format for fast msg encode.
|
||||||
- dns cache find_delegation routine.
|
- dns cache find_delegation routine.
|
||||||
|
- iterator main functions setup.
|
||||||
|
- dns cache lookup setup.
|
||||||
|
|
||||||
24 May 2007: Wouter
|
24 May 2007: Wouter
|
||||||
- small changes to prepare for subqueries.
|
- small changes to prepare for subqueries.
|
||||||
|
|
|
||||||
|
|
@ -82,7 +82,7 @@ iter_apply_cfg(struct iter_env* iter_env, struct config_file* cfg)
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
verbose(VERB_ALGO, "iterator: fwd queries to: %s %d",
|
verbose(VERB_ALGO, "iterator: fwd queries to: %s %d",
|
||||||
cfg->fwd_address, cfg->fwd_port);
|
cfg->fwd_address, cfg->fwd_port);
|
||||||
}
|
}
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -85,6 +85,31 @@ iter_deinit(struct module_env* env, int id)
|
||||||
/** new query for iterator */
|
/** new query for iterator */
|
||||||
static int
|
static int
|
||||||
iter_new(struct module_qstate* qstate, int id)
|
iter_new(struct module_qstate* qstate, int id)
|
||||||
|
{
|
||||||
|
struct iter_qstate* iq = (struct iter_qstate*)region_alloc(
|
||||||
|
qstate->region, sizeof(struct iter_qstate));
|
||||||
|
qstate->minfo[id] = iq;
|
||||||
|
if(!iq)
|
||||||
|
return 0;
|
||||||
|
memset(iq, 0, sizeof(*iq));
|
||||||
|
iq->state = INIT_REQUEST_STATE;
|
||||||
|
iq->final_state = FINISHED_STATE;
|
||||||
|
iq->prepend_list = NULL;
|
||||||
|
iq->prepend_last = NULL;
|
||||||
|
iq->dp = NULL;
|
||||||
|
iq->current_target = NULL;
|
||||||
|
iq->num_target_queries = -1; /* default our targetQueries counter. */
|
||||||
|
iq->num_current_queries = 0;
|
||||||
|
iq->query_restart_count = 0;
|
||||||
|
iq->referral_count = 0;
|
||||||
|
iq->priming_stub = 0;
|
||||||
|
outbound_list_init(&iq->outlist);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** new query for iterator in forward mode */
|
||||||
|
static int
|
||||||
|
fwd_new(struct module_qstate* qstate, int id)
|
||||||
{
|
{
|
||||||
struct iter_qstate* iq = (struct iter_qstate*)region_alloc(
|
struct iter_qstate* iq = (struct iter_qstate*)region_alloc(
|
||||||
qstate->region, sizeof(struct iter_qstate));
|
qstate->region, sizeof(struct iter_qstate));
|
||||||
|
|
@ -98,7 +123,6 @@ iter_new(struct module_qstate* qstate, int id)
|
||||||
return 0;
|
return 0;
|
||||||
memset(iq, 0, sizeof(*iq));
|
memset(iq, 0, sizeof(*iq));
|
||||||
outbound_list_init(&iq->outlist);
|
outbound_list_init(&iq->outlist);
|
||||||
iq->num_target_queries = -1; /* default our targetQueries counter. */
|
|
||||||
if(qstate->qinfo.has_cd)
|
if(qstate->qinfo.has_cd)
|
||||||
flags |= BIT_CD;
|
flags |= BIT_CD;
|
||||||
e = (*env->send_query)(qstate->qinfo.qname, qstate->qinfo.qnamesize,
|
e = (*env->send_query)(qstate->qinfo.qname, qstate->qinfo.qnamesize,
|
||||||
|
|
@ -147,6 +171,11 @@ perform_forward(struct module_qstate* qstate, enum module_ev event, int id,
|
||||||
struct outbound_entry* outbound)
|
struct outbound_entry* outbound)
|
||||||
{
|
{
|
||||||
verbose(VERB_ALGO, "iterator: forwarding");
|
verbose(VERB_ALGO, "iterator: forwarding");
|
||||||
|
if(event == module_event_new) {
|
||||||
|
if(!fwd_new(qstate, id))
|
||||||
|
qstate->ext_state[id] = module_error;
|
||||||
|
return;
|
||||||
|
}
|
||||||
/* it must be a query reply */
|
/* it must be a query reply */
|
||||||
if(!outbound) {
|
if(!outbound) {
|
||||||
verbose(VERB_ALGO, "query reply was not serviced");
|
verbose(VERB_ALGO, "query reply was not serviced");
|
||||||
|
|
@ -166,23 +195,355 @@ perform_forward(struct module_qstate* qstate, enum module_ev event, int id,
|
||||||
qstate->ext_state[id] = module_error;
|
qstate->ext_state[id] = module_error;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Transition to the next state. This can be used to advance a currently
|
||||||
|
* processing event. It cannot be used to reactivate a forEvent.
|
||||||
|
*
|
||||||
|
* @param qstate: query state
|
||||||
|
* @param iq: iterator query state
|
||||||
|
* @param nextstate The state to transition to.
|
||||||
|
* @return true. This is so this can be called as the return value for the
|
||||||
|
* actual process*State() methods. (Transitioning to the next state
|
||||||
|
* implies further processing).
|
||||||
|
*/
|
||||||
|
static int
|
||||||
|
next_state(struct module_qstate* qstate, struct iter_qstate* iq,
|
||||||
|
enum iter_state nextstate)
|
||||||
|
{
|
||||||
|
/* If transitioning to a "response" state, make sure that there is a
|
||||||
|
* response */
|
||||||
|
if(iter_state_is_responsestate(nextstate)) {
|
||||||
|
if(qstate->reply == NULL) {
|
||||||
|
log_err("transitioning to response state sans "
|
||||||
|
"response.");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
iq->state = nextstate;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Transition an event to its final state. Final states always either return
|
||||||
|
* a result up the module chain, or reactivate a dependent event. Which
|
||||||
|
* final state to transtion to is set in the module state for the event when
|
||||||
|
* it was created, and depends on the original purpose of the event.
|
||||||
|
*
|
||||||
|
* The response is stored in the qstate->buf buffer.
|
||||||
|
*
|
||||||
|
* @param qstate: query state
|
||||||
|
* @param iq: iterator query state
|
||||||
|
* @return false. This is so this method can be used as the return value for
|
||||||
|
* the processState methods. (Transitioning to the final state
|
||||||
|
*/
|
||||||
|
static int
|
||||||
|
final_state(struct module_qstate* qstate, struct iter_qstate* iq)
|
||||||
|
{
|
||||||
|
return next_state(qstate, iq, iq->final_state);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Return an error to the client
|
||||||
|
*/
|
||||||
|
static int
|
||||||
|
error_response(struct module_qstate* qstate, struct iter_qstate* iq, int rcode)
|
||||||
|
{
|
||||||
|
log_info("err response %s", ldns_lookup_by_id(ldns_rcodes, rcode)?
|
||||||
|
ldns_lookup_by_id(ldns_rcodes, rcode)->name:"??");
|
||||||
|
qinfo_query_encode(qstate->buf, &qstate->qinfo);
|
||||||
|
LDNS_RCODE_SET(ldns_buffer_begin(qstate->buf), rcode);
|
||||||
|
LDNS_QR_SET(ldns_buffer_begin(qstate->buf));
|
||||||
|
return final_state(qstate, iq);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Process the initial part of the request handling. This state roughly
|
||||||
|
* corresponds to resolver algorithms steps 1 (find answer in cache) and 2
|
||||||
|
* (find the best servers to ask).
|
||||||
|
*
|
||||||
|
* Note that all requests start here, and query restarts revisit this state.
|
||||||
|
*
|
||||||
|
* This state either generates: 1) a response, from cache or error, 2) a
|
||||||
|
* priming event, or 3) forwards the request to the next state (init2,
|
||||||
|
* generally).
|
||||||
|
*
|
||||||
|
* @param qstate: query state.
|
||||||
|
* @param iq: iterator query state.
|
||||||
|
* @param ie: iterator shared global environment.
|
||||||
|
* @return true if the event needs more request processing immediately,
|
||||||
|
* false if not.
|
||||||
|
*/
|
||||||
|
static int
|
||||||
|
processInitRequest(struct module_qstate* qstate, struct iter_qstate* iq,
|
||||||
|
struct iter_env* ie)
|
||||||
|
{
|
||||||
|
int d;
|
||||||
|
uint8_t* delname;
|
||||||
|
size_t delnamelen;
|
||||||
|
log_nametypeclass("resolving", qstate->qinfo.qname,
|
||||||
|
qstate->qinfo.qtype, qstate->qinfo.qclass);
|
||||||
|
/* check effort */
|
||||||
|
|
||||||
|
/* We enforce a maximum number of query restarts. This is primarily a
|
||||||
|
* cheap way to prevent CNAME loops. */
|
||||||
|
if(iq->query_restart_count > MAX_RESTART_COUNT) {
|
||||||
|
verbose(VERB_DETAIL, "request has exceeded the maximum number"
|
||||||
|
" of query restarts with %d", iq->query_restart_count);
|
||||||
|
return error_response(qstate, iq, LDNS_RCODE_SERVFAIL);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* We enforce a maximum recursion/dependency depth -- in general,
|
||||||
|
* this is unnecessary for dependency loops (although it will
|
||||||
|
* catch those), but it provides a sensible limit to the amount
|
||||||
|
* of work required to answer a given query. */
|
||||||
|
d = module_subreq_depth(qstate);
|
||||||
|
verbose(VERB_ALGO, "request has dependency depth of %d", d);
|
||||||
|
if(d > ie->max_dependency_depth) {
|
||||||
|
verbose(VERB_DETAIL, "request has exceeded the maximum "
|
||||||
|
"dependency depth with depth of %d", d);
|
||||||
|
return error_response(qstate, iq, LDNS_RCODE_SERVFAIL);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Resolver Algorithm Step 1 -- Look for the answer in local data. */
|
||||||
|
|
||||||
|
/* This either results in a query restart (CNAME cache response), a
|
||||||
|
* terminating response (ANSWER), or a cache miss (null). */
|
||||||
|
|
||||||
|
/* TODO: cache lookup */
|
||||||
|
/* lookup qname, qtype qclass */
|
||||||
|
|
||||||
|
/* TODO: handle positive cache response */
|
||||||
|
/* calc response type (from cache msg) */
|
||||||
|
/* if cname */
|
||||||
|
/* handle cname, overwrite qname, restart */
|
||||||
|
/* it is an answer, response, to final state */
|
||||||
|
|
||||||
|
/* TODO attempt to forward the request */
|
||||||
|
|
||||||
|
/* TODO attempt to find a covering DNAME in the cache */
|
||||||
|
|
||||||
|
/* Resolver Algorithm Step 2 -- find the "best" servers. */
|
||||||
|
|
||||||
|
/* first, adjust for DS queries. To avoid the grandparent problem,
|
||||||
|
* we just look for the closest set of server to the parent of qname.
|
||||||
|
*/
|
||||||
|
delname = qstate->qinfo.qname;
|
||||||
|
delnamelen = qstate->qinfo.qnamesize;
|
||||||
|
if(qstate->qinfo.qtype == LDNS_RR_TYPE_DS && delname[0] != 0) {
|
||||||
|
/* do not adjust root label */
|
||||||
|
size_t lablen = delname[0] + 1;
|
||||||
|
delname += lablen;
|
||||||
|
delnamelen -= lablen;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Lookup the delegation in the cache. If null, then the cache needs
|
||||||
|
* to be primed for the qclass. */
|
||||||
|
iq->dp = dns_cache_find_delegation(qstate->env, delname, delnamelen,
|
||||||
|
qstate->qinfo.qclass, qstate->region);
|
||||||
|
|
||||||
|
/* If the cache has returned nothing, then we have a root priming
|
||||||
|
* situation. */
|
||||||
|
if(iq->dp == NULL) {
|
||||||
|
/* Note that the result of this will set a new
|
||||||
|
* DelegationPoint based on the result of priming. */
|
||||||
|
|
||||||
|
/* TODO
|
||||||
|
if(!prime_root(qstate, iq, ie, qstate->qinfo.qclass))
|
||||||
|
return error_response(qstate, iq, LDNS_RCODE_SERVFAIL);
|
||||||
|
*/
|
||||||
|
|
||||||
|
/* priming creates an sends a subordinate query, with
|
||||||
|
* this query as the parent. So further processing for
|
||||||
|
* this event will stop until reactivated by the results
|
||||||
|
* of priming. */
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Reset the RD flag. If this is a query restart, then the RD
|
||||||
|
* will have been turned off. */
|
||||||
|
/*
|
||||||
|
TODO store original flags and original qinfo
|
||||||
|
qstate->query_flags |= (qstate->orig_query_flags & BIT_RD);
|
||||||
|
*/
|
||||||
|
|
||||||
|
/* Otherwise, set the current delegation point and move on to the
|
||||||
|
* next state. */
|
||||||
|
return next_state(qstate, iq, INIT_REQUEST_2_STATE);
|
||||||
|
}
|
||||||
|
|
||||||
|
#if 0
|
||||||
|
/** TODO */
|
||||||
|
static int
|
||||||
|
processInitRequest2(struct module_qstate* qstate, struct iter_qstate* iq,
|
||||||
|
struct iter_env* ie)
|
||||||
|
{
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** TODO */
|
||||||
|
static int
|
||||||
|
processInitRequest3(struct module_qstate* qstate, struct iter_qstate* iq,
|
||||||
|
struct iter_env* ie)
|
||||||
|
{
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** TODO */
|
||||||
|
static int
|
||||||
|
processQueryTargets(struct module_qstate* qstate, struct iter_qstate* iq,
|
||||||
|
struct iter_env* ie)
|
||||||
|
{
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** TODO */
|
||||||
|
static int
|
||||||
|
processQueryResponse(struct module_qstate* qstate, struct iter_qstate* iq,
|
||||||
|
struct iter_env* ie)
|
||||||
|
{
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** TODO */
|
||||||
|
static int
|
||||||
|
processPrimeResponse(struct module_qstate* qstate, struct iter_qstate* iq,
|
||||||
|
struct iter_env* ie)
|
||||||
|
{
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** TODO */
|
||||||
|
static int
|
||||||
|
processTargetResponse(struct module_qstate* qstate, struct iter_qstate* iq,
|
||||||
|
struct iter_env* ie)
|
||||||
|
{
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** TODO */
|
||||||
|
static int
|
||||||
|
processFinished(struct module_qstate* qstate, struct iter_qstate* iq,
|
||||||
|
struct iter_env* ie)
|
||||||
|
{
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Handle iterator state.
|
||||||
|
* Handle events. This is the real processing loop for events, responsible
|
||||||
|
* for moving events through the various states. If a processing method
|
||||||
|
* returns true, then it will be advanced to the next state. If false, then
|
||||||
|
* processing will stop.
|
||||||
|
*
|
||||||
|
* @param qstate: query state.
|
||||||
|
* @param ie: iterator shared global environment.
|
||||||
|
* @param iq: iterator query state.
|
||||||
|
*/
|
||||||
|
static void
|
||||||
|
iter_handle(struct module_qstate* qstate, struct iter_qstate* iq,
|
||||||
|
struct iter_env* ie)
|
||||||
|
{
|
||||||
|
int cont = 1;
|
||||||
|
while(cont) {
|
||||||
|
verbose(VERB_ALGO, "iter_handle processing q with state %s",
|
||||||
|
iter_state_to_string(iq->state));
|
||||||
|
switch(iq->state) {
|
||||||
|
case INIT_REQUEST_STATE:
|
||||||
|
cont = processInitRequest(qstate, iq, ie);
|
||||||
|
break;
|
||||||
|
#if 0
|
||||||
|
case INIT_REQUEST_2_STATE:
|
||||||
|
cont = processInitRequest2(qstate, iq, ie);
|
||||||
|
break;
|
||||||
|
case INIT_REQUEST_3_STATE:
|
||||||
|
cont = processInitRequest3(qstate, iq, ie);
|
||||||
|
break;
|
||||||
|
case QUERYTARGETS_STATE:
|
||||||
|
cont = processQueryTargets(qstate, iq, ie);
|
||||||
|
break;
|
||||||
|
case QUERY_RESP_STATE:
|
||||||
|
cont = processQueryResponse(qstate, iq, ie);
|
||||||
|
break;
|
||||||
|
case PRIME_RESP_STATE:
|
||||||
|
cont = processPrimeResponse(qstate, iq, ie);
|
||||||
|
break;
|
||||||
|
case TARGET_RESP_STATE:
|
||||||
|
cont = processTargetResponse(qstate, iq, ie);
|
||||||
|
break;
|
||||||
|
case FINISHED_STATE:
|
||||||
|
cont = processFinished(qstate, iq, ie);
|
||||||
|
break;
|
||||||
|
#endif
|
||||||
|
default:
|
||||||
|
log_warn("iterator: invalid state: %d",
|
||||||
|
iq->state);
|
||||||
|
cont = 0;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* This is the primary entry point for processing request events. Note that
|
||||||
|
* this method should only be used by external modules.
|
||||||
|
* @param qstate: query state.
|
||||||
|
* @param ie: iterator shared global environment.
|
||||||
|
* @param iq: iterator query state.
|
||||||
|
*/
|
||||||
|
static void
|
||||||
|
process_request(struct module_qstate* qstate, struct iter_qstate* iq,
|
||||||
|
struct iter_env* ie)
|
||||||
|
{
|
||||||
|
/* external requests start in the INIT state, and finish using the
|
||||||
|
* FINISHED state. */
|
||||||
|
iq->state = INIT_REQUEST_STATE;
|
||||||
|
iq->final_state = FINISHED_STATE;
|
||||||
|
verbose(VERB_ALGO, "process_request: new external request event");
|
||||||
|
iter_handle(qstate, iq, ie);
|
||||||
|
}
|
||||||
|
|
||||||
|
/** process authoritative server reply */
|
||||||
|
static void
|
||||||
|
process_response(struct module_qstate* qstate, struct iter_qstate* iq,
|
||||||
|
struct iter_env* ie, struct outbound_entry* outbound)
|
||||||
|
{
|
||||||
|
verbose(VERB_ALGO, "process_response: new external response event");
|
||||||
|
/* TODO outbound: use it for scrubbing and so on */
|
||||||
|
iq->state = QUERY_RESP_STATE;
|
||||||
|
iter_handle(qstate, iq, ie);
|
||||||
|
}
|
||||||
|
|
||||||
/** iterator operate on a query */
|
/** iterator operate on a query */
|
||||||
static void
|
static void
|
||||||
iter_operate(struct module_qstate* qstate, enum module_ev event, int id,
|
iter_operate(struct module_qstate* qstate, enum module_ev event, int id,
|
||||||
struct outbound_entry* outbound)
|
struct outbound_entry* outbound)
|
||||||
{
|
{
|
||||||
struct iter_env* ie = (struct iter_env*)qstate->env->modinfo[id];
|
struct iter_env* ie = (struct iter_env*)qstate->env->modinfo[id];
|
||||||
|
struct iter_qstate* iq;
|
||||||
verbose(VERB_ALGO, "iterator[module %d] operate: extstate:%s event:%s",
|
verbose(VERB_ALGO, "iterator[module %d] operate: extstate:%s event:%s",
|
||||||
id, strextstate(qstate->ext_state[id]), strmodulevent(event));
|
id, strextstate(qstate->ext_state[id]), strmodulevent(event));
|
||||||
if(event == module_event_new) {
|
|
||||||
if(!iter_new(qstate, id))
|
|
||||||
qstate->ext_state[id] = module_error;
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if(ie->fwd_addrlen != 0) {
|
if(ie->fwd_addrlen != 0) {
|
||||||
perform_forward(qstate, event, id, outbound);
|
perform_forward(qstate, event, id, outbound);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
/* perform iterator state machine */
|
||||||
|
if(event == module_event_new) {
|
||||||
|
log_info("iter state machine");
|
||||||
|
if(!iter_new(qstate, id)) {
|
||||||
|
qstate->ext_state[id] = module_error;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
iq = (struct iter_qstate*)qstate->minfo[id];
|
||||||
|
process_request(qstate, iq, ie);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
iq = (struct iter_qstate*)qstate->minfo[id];
|
||||||
|
if(event == module_event_reply) {
|
||||||
|
process_response(qstate, iq, ie, outbound);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
/* TODO: uhh */
|
||||||
|
|
||||||
log_err("bad event for iterator");
|
log_err("bad event for iterator");
|
||||||
qstate->ext_state[id] = module_error;
|
qstate->ext_state[id] = module_error;
|
||||||
}
|
}
|
||||||
|
|
@ -238,3 +599,18 @@ iter_state_to_string(enum iter_state state)
|
||||||
return "UNKNOWN ITER STATE";
|
return "UNKNOWN ITER STATE";
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int
|
||||||
|
iter_state_is_responsestate(enum iter_state s)
|
||||||
|
{
|
||||||
|
switch(s) {
|
||||||
|
case INIT_REQUEST_STATE :
|
||||||
|
case INIT_REQUEST_2_STATE :
|
||||||
|
case INIT_REQUEST_3_STATE :
|
||||||
|
case QUERYTARGETS_STATE :
|
||||||
|
return 0;
|
||||||
|
default:
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
|
||||||
|
|
@ -48,6 +48,9 @@ struct delegpt;
|
||||||
struct packed_rrset_list;
|
struct packed_rrset_list;
|
||||||
struct iter_hints;
|
struct iter_hints;
|
||||||
|
|
||||||
|
/** max number of query restarts. Determines max number of CNAME chain. */
|
||||||
|
#define MAX_RESTART_COUNT 8
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Global state for the iterator.
|
* Global state for the iterator.
|
||||||
*/
|
*/
|
||||||
|
|
@ -212,4 +215,11 @@ struct module_func_block* iter_get_funcblock();
|
||||||
*/
|
*/
|
||||||
const char* iter_state_to_string(enum iter_state state);
|
const char* iter_state_to_string(enum iter_state state);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* See if iterator state is a response state
|
||||||
|
* @param s: to inspect
|
||||||
|
* @return true if response state.
|
||||||
|
*/
|
||||||
|
int iter_state_is_responsestate(enum iter_state s);
|
||||||
|
|
||||||
#endif /* ITERATOR_ITERATOR_H */
|
#endif /* ITERATOR_ITERATOR_H */
|
||||||
|
|
|
||||||
117
services/cache/dns.c
vendored
117
services/cache/dns.c
vendored
|
|
@ -46,6 +46,7 @@
|
||||||
#include "util/data/packed_rrset.h"
|
#include "util/data/packed_rrset.h"
|
||||||
#include "util/module.h"
|
#include "util/module.h"
|
||||||
#include "util/net_help.h"
|
#include "util/net_help.h"
|
||||||
|
#include "util/region-allocator.h"
|
||||||
|
|
||||||
/** store rrsets in the rrset cache. */
|
/** store rrsets in the rrset cache. */
|
||||||
static void
|
static void
|
||||||
|
|
@ -229,3 +230,119 @@ dns_cache_find_delegation(struct module_env* env, uint8_t* qname,
|
||||||
delegpt_log(dp);
|
delegpt_log(dp);
|
||||||
return dp;
|
return dp;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/** allocate rrset in region - no more locks needed */
|
||||||
|
static struct ub_packed_rrset_key*
|
||||||
|
copy_rrset(struct ub_packed_rrset_key* key, struct region* region)
|
||||||
|
{
|
||||||
|
/* lock, lrutouch rrset in cache */
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** allocate dns_msg from query_info and reply_info */
|
||||||
|
static struct dns_msg*
|
||||||
|
tomsg(struct msgreply_entry* e, struct reply_info* r, struct region* region)
|
||||||
|
{
|
||||||
|
struct dns_msg* msg = (struct dns_msg*)region_alloc(region,
|
||||||
|
sizeof(struct dns_msg));
|
||||||
|
size_t i;
|
||||||
|
if(!msg)
|
||||||
|
return NULL;
|
||||||
|
memcpy(&msg->qinfo, &e->key, sizeof(struct query_info));
|
||||||
|
msg->qinfo.qname = region_alloc_init(region, e->key.qname,
|
||||||
|
e->key.qnamesize);
|
||||||
|
if(!msg->qinfo.qname)
|
||||||
|
return NULL;
|
||||||
|
/* allocate replyinfo struct and rrset key array separately */
|
||||||
|
msg->rep = (struct reply_info*)region_alloc(region,
|
||||||
|
sizeof(struct reply_info) - sizeof(struct rrset_ref));
|
||||||
|
if(!msg->rep)
|
||||||
|
return NULL;
|
||||||
|
memcpy(msg->rep, r,
|
||||||
|
sizeof(struct reply_info) - sizeof(struct rrset_ref));
|
||||||
|
msg->rep->rrsets = (struct ub_packed_rrset_key**)region_alloc(region,
|
||||||
|
msg->rep->rrset_count * sizeof(struct ub_packed_rrset_key*));
|
||||||
|
if(!msg->rep->rrsets)
|
||||||
|
return NULL;
|
||||||
|
/* try to lock all of the rrsets we need */
|
||||||
|
for(i=0; i<msg->rep->rrset_count; i++) {
|
||||||
|
msg->rep->rrsets[i] = copy_rrset(r->rrsets[i], region);
|
||||||
|
if(!msg->rep->rrsets[i])
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
return msg;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** allocate dns_msg from CNAME record */
|
||||||
|
static struct dns_msg*
|
||||||
|
cnamemsg(uint8_t* qname, size_t qnamelen, struct ub_packed_rrset_key* rrset,
|
||||||
|
struct packed_rrset_data* d, struct region* region)
|
||||||
|
{
|
||||||
|
struct dns_msg* msg = (struct dns_msg*)region_alloc(region,
|
||||||
|
sizeof(struct dns_msg));
|
||||||
|
if(!msg)
|
||||||
|
return NULL;
|
||||||
|
msg->qinfo.qnamesize = rrset->rk.dname_len;
|
||||||
|
msg->qinfo.qname = region_alloc_init(region, rrset->rk.dname,
|
||||||
|
rrset->rk.dname_len);
|
||||||
|
if(!msg->qinfo.qname)
|
||||||
|
return NULL;
|
||||||
|
msg->qinfo.has_cd = (rrset->rk.flags&PACKED_RRSET_CD)?1:0;
|
||||||
|
msg->qinfo.qtype = LDNS_RR_TYPE_CNAME;
|
||||||
|
msg->qinfo.qclass = ntohs(rrset->rk.rrset_class);
|
||||||
|
/* TODO create reply info with the CNAME */
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
struct dns_msg*
|
||||||
|
dns_cache_lookup(struct module_env* env,
|
||||||
|
uint8_t* qname, size_t qnamelen, uint16_t qtype, uint16_t qclass,
|
||||||
|
int has_cd, struct region* region)
|
||||||
|
{
|
||||||
|
struct lruhash_entry* e;
|
||||||
|
struct query_info k;
|
||||||
|
hashvalue_t h;
|
||||||
|
uint32_t now = (uint32_t)time(NULL);
|
||||||
|
struct ub_packed_rrset_key* rrset;
|
||||||
|
|
||||||
|
/* lookup first, this has both NXdomains and ANSWER responses */
|
||||||
|
k.qname = qname;
|
||||||
|
k.qnamesize = qnamelen;
|
||||||
|
k.qtype = qtype;
|
||||||
|
k.qclass = qclass;
|
||||||
|
k.has_cd = has_cd;
|
||||||
|
h = query_info_hash(&k);
|
||||||
|
e = slabhash_lookup(env->msg_cache, h, &k, 0);
|
||||||
|
if(e) {
|
||||||
|
/* check ttl */
|
||||||
|
struct msgreply_entry* key = (struct msgreply_entry*)e->key;
|
||||||
|
struct reply_info* data = (struct reply_info*)e->data;
|
||||||
|
if(now <= data->ttl) {
|
||||||
|
struct dns_msg* msg = tomsg(key, data, region);
|
||||||
|
lock_rw_unlock(&e->lock);
|
||||||
|
return msg;
|
||||||
|
}
|
||||||
|
lock_rw_unlock(&e->lock);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* see if we have a CNAME for this domain */
|
||||||
|
rrset = rrset_cache_lookup(env->rrset_cache, qname, qnamelen,
|
||||||
|
LDNS_RR_TYPE_CNAME, qclass,
|
||||||
|
(uint32_t)(has_cd?PACKED_RRSET_CD:0), now, 0);
|
||||||
|
if(rrset) {
|
||||||
|
struct packed_rrset_data* d = (struct packed_rrset_data*)
|
||||||
|
rrset->entry.data;
|
||||||
|
if(now <= d->ttl) {
|
||||||
|
/* construct CNAME response */
|
||||||
|
struct dns_msg* msg = cnamemsg(qname, qnamelen, rrset,
|
||||||
|
d, region);
|
||||||
|
lock_rw_unlock(&rrset->entry.lock);
|
||||||
|
return msg;
|
||||||
|
}
|
||||||
|
lock_rw_unlock(&rrset->entry.lock);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* construct DS, DNSKEY messages from rrset cache. TODO */
|
||||||
|
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
|
||||||
28
services/cache/dns.h
vendored
28
services/cache/dns.h
vendored
|
|
@ -42,12 +42,23 @@
|
||||||
#ifndef SERVICES_CACHE_DNS_H
|
#ifndef SERVICES_CACHE_DNS_H
|
||||||
#define SERVICES_CACHE_DNS_H
|
#define SERVICES_CACHE_DNS_H
|
||||||
#include "util/storage/lruhash.h"
|
#include "util/storage/lruhash.h"
|
||||||
|
#include "util/data/msgreply.h"
|
||||||
struct module_env;
|
struct module_env;
|
||||||
struct query_info;
|
struct query_info;
|
||||||
struct reply_info;
|
struct reply_info;
|
||||||
struct region;
|
struct region;
|
||||||
struct delegpt;
|
struct delegpt;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Region allocated message reply
|
||||||
|
*/
|
||||||
|
struct dns_msg {
|
||||||
|
/** query info */
|
||||||
|
struct query_info qinfo;
|
||||||
|
/** reply info - ptr to packed repinfo structure */
|
||||||
|
struct reply_info *rep;
|
||||||
|
};
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Store message in the cache. Stores in message cache and rrset cache.
|
* Store message in the cache. Stores in message cache and rrset cache.
|
||||||
* Both qinfo and rep should be malloced and are put in the cache.
|
* Both qinfo and rep should be malloced and are put in the cache.
|
||||||
|
|
@ -76,7 +87,22 @@ struct delegpt* dns_cache_find_delegation(struct module_env* env,
|
||||||
uint8_t* qname, size_t qnamelen, uint16_t qclass,
|
uint8_t* qname, size_t qnamelen, uint16_t qclass,
|
||||||
struct region* region);
|
struct region* region);
|
||||||
|
|
||||||
/** Find cached message */
|
/**
|
||||||
|
* Find cached message
|
||||||
|
* @param env: module environment with the DNS cache.
|
||||||
|
* @param qname: query name.
|
||||||
|
* @param qnamelen: length of qname.
|
||||||
|
* @param qtype: query type.
|
||||||
|
* @param qclass: query class.
|
||||||
|
* @param has_cd: if true, CD flag is turned on for lookup.
|
||||||
|
* @param region: where to allocate result.
|
||||||
|
* @return new response message (alloced in region, rrsets do not have IDs).
|
||||||
|
* or NULL on error or if not found in cache.
|
||||||
|
*/
|
||||||
|
struct dns_msg* dns_cache_lookup(struct module_env* env,
|
||||||
|
uint8_t* qname, size_t qnamelen, uint16_t qtype, uint16_t qclass,
|
||||||
|
int has_cd, struct region* region);
|
||||||
|
|
||||||
/** Find covering DNAME */
|
/** Find covering DNAME */
|
||||||
|
|
||||||
#endif /* SERVICES_CACHE_DNS_H */
|
#endif /* SERVICES_CACHE_DNS_H */
|
||||||
|
|
|
||||||
|
|
@ -87,3 +87,14 @@ module_subreq_remove(struct module_qstate* sub)
|
||||||
p = p->subquery_next;
|
p = p->subquery_next;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int
|
||||||
|
module_subreq_depth(struct module_qstate* sub)
|
||||||
|
{
|
||||||
|
int d = 0;
|
||||||
|
while(sub->parent) {
|
||||||
|
d++;
|
||||||
|
sub = sub->parent;
|
||||||
|
}
|
||||||
|
return d;
|
||||||
|
}
|
||||||
|
|
|
||||||
|
|
@ -275,4 +275,11 @@ const char* strmodulevent(enum module_ev e);
|
||||||
*/
|
*/
|
||||||
void module_subreq_remove(struct module_qstate* sub);
|
void module_subreq_remove(struct module_qstate* sub);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Calculate depth of subrequest
|
||||||
|
* @param sub: the subrequest. parent point is used.
|
||||||
|
* @return: depth > 0 for subrequests.
|
||||||
|
*/
|
||||||
|
int module_subreq_depth(struct module_qstate* sub);
|
||||||
|
|
||||||
#endif /* UTIL_MODULE_H */
|
#endif /* UTIL_MODULE_H */
|
||||||
|
|
|
||||||
|
|
@ -40,6 +40,7 @@
|
||||||
#include "config.h"
|
#include "config.h"
|
||||||
#include "util/net_help.h"
|
#include "util/net_help.h"
|
||||||
#include "util/log.h"
|
#include "util/log.h"
|
||||||
|
#include "util/data/dname.h"
|
||||||
#include <fcntl.h>
|
#include <fcntl.h>
|
||||||
|
|
||||||
/** returns true is string addr is an ip6 specced address. */
|
/** returns true is string addr is an ip6 specced address. */
|
||||||
|
|
@ -176,3 +177,15 @@ ipstrtoaddr(const char* ip, int port, struct sockaddr_storage* addr,
|
||||||
}
|
}
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
log_nametypeclass(const char* str, uint8_t* name, uint16_t type,
|
||||||
|
uint16_t dclass)
|
||||||
|
{
|
||||||
|
char buf[LDNS_MAX_DOMAINLEN+1];
|
||||||
|
dname_str(name, buf);
|
||||||
|
log_info("%s <%s %s %s>", str, buf,
|
||||||
|
ldns_rr_descript(type)?ldns_rr_descript(type)->_name: "??",
|
||||||
|
ldns_lookup_by_id(ldns_rr_classes, (int)dclass)?
|
||||||
|
ldns_lookup_by_id(ldns_rr_classes, (int)dclass)->name:"??");
|
||||||
|
}
|
||||||
|
|
|
||||||
|
|
@ -134,4 +134,14 @@ void log_addr(const char* str, struct sockaddr_storage* addr,
|
||||||
int ipstrtoaddr(const char* ip, int port, struct sockaddr_storage* addr,
|
int ipstrtoaddr(const char* ip, int port, struct sockaddr_storage* addr,
|
||||||
socklen_t* addrlen);
|
socklen_t* addrlen);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Print string with neat domain name, type and class.
|
||||||
|
* @param str: string of message.
|
||||||
|
* @param name: domain name uncompressed wireformat.
|
||||||
|
* @param type: host format RR type.
|
||||||
|
* @param dclass: host format RR class.
|
||||||
|
*/
|
||||||
|
void log_nametypeclass(const char* str, uint8_t* name, uint16_t type,
|
||||||
|
uint16_t dclass);
|
||||||
|
|
||||||
#endif /* NET_HELP_H */
|
#endif /* NET_HELP_H */
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue