--- /dev/null
+/*
+ * daemon/cachedump.c - dump the cache to text format.
+ *
+ * Copyright (c) 2008, NLnet Labs. All rights reserved.
+ *
+ * This software is open source.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * Redistributions of source code must retain the above copyright notice,
+ * this list of conditions and the following disclaimer.
+ *
+ * Redistributions in binary form must reproduce the above copyright notice,
+ * this list of conditions and the following disclaimer in the documentation
+ * and/or other materials provided with the distribution.
+ *
+ * Neither the name of the NLNET LABS nor the names of its contributors may
+ * be used to endorse or promote products derived from this software without
+ * specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
+ * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE
+ * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+ * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+ * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+ * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+ * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+ * POSSIBILITY OF SUCH DAMAGE.
+ */
+
+/**
+ * \file
+ *
+ * This file contains functions to read and write the cache(s)
+ * to text format.
+ */
+#include "config.h"
+#include "daemon/cachedump.h"
+#include "daemon/remote.h"
+#include "daemon/worker.h"
+#include "services/cache/rrset.h"
+#include "services/cache/dns.h"
+#include "util/data/msgreply.h"
+#include "util/regional.h"
+#include "util/net_help.h"
+
+/** convert to ldns rr */
+static ldns_rr*
+to_rr(struct ub_packed_rrset_key* k, struct packed_rrset_data* d,
+ uint32_t now, size_t i, uint16_t type)
+{
+ ldns_rr* rr = ldns_rr_new();
+ ldns_rdf* rdf;
+ ldns_status status;
+ size_t pos;
+ log_assert(i < d->count + d->rrsig_count);
+ if(!rr) {
+ return NULL;
+ }
+ ldns_rr_set_type(rr, type);
+ ldns_rr_set_class(rr, ntohs(k->rk.rrset_class));
+ ldns_rr_set_ttl(rr, d->rr_ttl[i] - now);
+ pos = 0;
+ status = ldns_wire2dname(&rdf, k->rk.dname, k->rk.dname_len, &pos);
+ if(status != LDNS_STATUS_OK) {
+ /* we have detailed error in status */
+ ldns_rr_free(rr);
+ return NULL;
+ }
+ ldns_rr_set_owner(rr, rdf);
+ pos = 0;
+ status = ldns_wire2rdf(rr, d->rr_data[i], d->rr_len[i], &pos);
+ if(status != LDNS_STATUS_OK) {
+ /* we have detailed error in status */
+ ldns_rr_free(rr);
+ return NULL;
+ }
+ return rr;
+}
+
+/** dump one rrset zonefile line */
+static int
+dump_rrset_line(SSL* ssl, struct ub_packed_rrset_key* k,
+ struct packed_rrset_data* d, uint32_t now, int i, uint16_t type)
+{
+ char* s;
+ ldns_rr* rr = to_rr(k, d, now, i, type);
+ if(!rr) {
+ return ssl_printf(ssl, "BADRR\n");
+ }
+ s = ldns_rr2str(rr);
+ if(!s) {
+ ldns_rr_free(rr);
+ return ssl_printf(ssl, "BADRR\n");
+ }
+ if(!ssl_printf(ssl, "%s", s)) {
+ free(s);
+ ldns_rr_free(rr);
+ return 0;
+ }
+ free(s);
+ ldns_rr_free(rr);
+ return 1;
+}
+
+/** dump rrset key and data info */
+static int
+dump_rrset(SSL* ssl, struct ub_packed_rrset_key* k,
+ struct packed_rrset_data* d, uint32_t now)
+{
+ size_t i;
+ /* rd lock held by caller */
+ if(!k || !d) return 1;
+ if(d->ttl < now) return 1; /* expired */
+
+ /* meta line */
+ if(!ssl_printf(ssl, ";rrset%s %u %u %u %d %d\n",
+ (k->rk.flags & PACKED_RRSET_NSEC_AT_APEX)?" nsec_apex":"",
+ (unsigned)(d->ttl - now),
+ (unsigned)d->count, (unsigned)d->rrsig_count,
+ (int)d->trust, (int)d->security
+ ))
+ return 0;
+ for(i=0; i<d->count; i++) {
+ if(!dump_rrset_line(ssl, k, d, now, i, ntohs(k->rk.type)))
+ return 0;
+ }
+ for(i=0; i<d->rrsig_count; i++) {
+ if(!dump_rrset_line(ssl, k, d, now, i+d->count,
+ LDNS_RR_TYPE_RRSIG))
+ return 0;
+ }
+
+ return 1;
+}
+
+/** dump lruhash rrset cache */
+static int
+dump_rrset_lruhash(SSL* ssl, struct lruhash* h, uint32_t now)
+{
+ struct lruhash_entry* e;
+ /* lruhash already locked by caller */
+ /* walk in order of lru; best first */
+ for(e=h->lru_start; e; e = e->lru_next) {
+ lock_rw_rdlock(&e->lock);
+ if(!dump_rrset(ssl, (struct ub_packed_rrset_key*)e->key,
+ (struct packed_rrset_data*)e->data, now)) {
+ lock_rw_unlock(&e->lock);
+ return 0;
+ }
+ lock_rw_unlock(&e->lock);
+ }
+ return 1;
+}
+
+/** dump rrset cache */
+static int
+dump_rrset_cache(SSL* ssl, struct worker* worker)
+{
+ struct rrset_cache* r = worker->env.rrset_cache;
+ size_t slab;
+ if(!ssl_printf(ssl, "START_RRSET_CACHE\n")) return 0;
+ for(slab=0; slab<r->table.size; slab++) {
+ lock_quick_lock(&r->table.array[slab]->lock);
+ if(!dump_rrset_lruhash(ssl, r->table.array[slab],
+ *worker->env.now)) {
+ lock_quick_unlock(&r->table.array[slab]->lock);
+ return 0;
+ }
+ lock_quick_unlock(&r->table.array[slab]->lock);
+ }
+ return ssl_printf(ssl, "END_RRSET_CACHE\n");
+}
+
+/** dump message to rrset reference */
+static int
+dump_msg_ref(SSL* ssl, struct ub_packed_rrset_key* k)
+{
+ ldns_rdf* rdf;
+ ldns_status status;
+ size_t pos;
+ char* nm, *tp, *cl;
+
+ pos = 0;
+ status = ldns_wire2dname(&rdf, k->rk.dname, k->rk.dname_len, &pos);
+ if(status != LDNS_STATUS_OK) {
+ return ssl_printf(ssl, "BADREF\n");
+ }
+ nm = ldns_rdf2str(rdf);
+ ldns_rdf_deep_free(rdf);
+ tp = ldns_rr_type2str(ntohs(k->rk.type));
+ cl = ldns_rr_class2str(ntohs(k->rk.rrset_class));
+ if(!nm || !cl || !tp) {
+ free(nm);
+ free(tp);
+ free(cl);
+ return ssl_printf(ssl, "BADREF\n");
+ }
+ if(!ssl_printf(ssl, "%s %s %s %d\n", nm, cl, tp, (int)k->rk.flags)) {
+ free(nm);
+ free(tp);
+ free(cl);
+ return 0;
+ }
+ free(nm);
+ free(tp);
+ free(cl);
+
+ return 1;
+}
+
+/** dump message entry */
+static int
+dump_msg(SSL* ssl, struct query_info* k, struct reply_info* d,
+ uint32_t now)
+{
+ size_t i;
+ char* nm, *tp, *cl;
+ ldns_rdf* rdf;
+ ldns_status status;
+ size_t pos;
+ if(!k || !d) return 1;
+ if(d->ttl < now) return 1; /* expired */
+
+ pos = 0;
+ status = ldns_wire2dname(&rdf, k->qname, k->qname_len, &pos);
+ if(status != LDNS_STATUS_OK) {
+ return 1; /* skip this entry */
+ }
+ nm = ldns_rdf2str(rdf);
+ ldns_rdf_deep_free(rdf);
+ tp = ldns_rr_type2str(k->qtype);
+ cl = ldns_rr_class2str(k->qclass);
+ if(!nm || !tp || !cl) {
+ free(nm);
+ free(tp);
+ free(cl);
+ return 1; /* skip this entry */
+ }
+ if(!rrset_array_lock(d->ref, d->rrset_count, now)) {
+ /* rrsets have timed out or do not exist */
+ free(nm);
+ free(tp);
+ free(cl);
+ return 1; /* skip this entry */
+ }
+
+ /* meta line */
+ if(!ssl_printf(ssl, "msg %s %s %s %d %d %u %d %u %u %u\n",
+ nm, cl, tp,
+ (int)d->flags, (int)d->qdcount,
+ (unsigned)(d->ttl-now), (int)d->security,
+ (unsigned)d->an_numrrsets,
+ (unsigned)d->ns_numrrsets,
+ (unsigned)d->ar_numrrsets)) {
+ free(nm);
+ free(tp);
+ free(cl);
+ rrset_array_unlock(d->ref, d->rrset_count);
+ return 0;
+ }
+ free(nm);
+ free(tp);
+ free(cl);
+
+ for(i=0; i<d->rrset_count; i++) {
+ if(!dump_msg_ref(ssl, d->rrsets[i])) {
+ rrset_array_unlock(d->ref, d->rrset_count);
+ return 0;
+ }
+ }
+ rrset_array_unlock(d->ref, d->rrset_count);
+
+ return 1;
+}
+
+/** copy msg to worker pad */
+static int
+copy_msg(struct regional* region, struct lruhash_entry* e,
+ struct query_info** k, struct reply_info** d)
+{
+ struct reply_info* rep = (struct reply_info*)e->data;
+ *d = (struct reply_info*)regional_alloc_init(region, e->data,
+ sizeof(struct reply_info) +
+ sizeof(struct rrset_ref) * (rep->rrset_count-1) +
+ sizeof(struct ub_packed_rrset_key*) * rep->rrset_count);
+ if(!*d)
+ return 0;
+ (*d)->rrsets = (struct ub_packed_rrset_key**)(
+ (uint8_t*)(&((*d)->ref[0])) +
+ sizeof(struct rrset_ref) * rep->rrset_count);
+ *k = (struct query_info*)regional_alloc_init(region,
+ e->key, sizeof(struct query_info));
+ if(!*k)
+ return 0;
+ (*k)->qname = regional_alloc_init(region,
+ (*k)->qname, (*k)->qname_len);
+ return (*k)->qname != NULL;
+}
+
+/** dump lruhash msg cache */
+static int
+dump_msg_lruhash(SSL* ssl, struct worker* worker, struct lruhash* h)
+{
+ struct lruhash_entry* e;
+ struct query_info* k;
+ struct reply_info* d;
+
+ /* lruhash already locked by caller */
+ /* walk in order of lru; best first */
+ for(e=h->lru_start; e; e = e->lru_next) {
+ regional_free_all(worker->scratchpad);
+ lock_rw_rdlock(&e->lock);
+ /* make copy of rrset in worker buffer */
+ if(!copy_msg(worker->scratchpad, e, &k, &d)) {
+ lock_rw_unlock(&e->lock);
+ return 0;
+ }
+ lock_rw_unlock(&e->lock);
+ /* release lock so we can lookup the rrset references
+ * in the rrset cache */
+ if(!dump_msg(ssl, k, d, *worker->env.now)) {
+ return 0;
+ }
+ }
+ return 1;
+}
+
+/** dump msg cache */
+static int
+dump_msg_cache(SSL* ssl, struct worker* worker)
+{
+ struct slabhash* sh = worker->env.msg_cache;
+ size_t slab;
+ if(!ssl_printf(ssl, "START_MSG_CACHE\n")) return 0;
+ for(slab=0; slab<sh->size; slab++) {
+ lock_quick_lock(&sh->array[slab]->lock);
+ if(!dump_msg_lruhash(ssl, worker, sh->array[slab])) {
+ lock_quick_unlock(&sh->array[slab]->lock);
+ return 0;
+ }
+ lock_quick_unlock(&sh->array[slab]->lock);
+ }
+ return ssl_printf(ssl, "END_MSG_CACHE\n");
+}
+
+int
+dump_cache(SSL* ssl, struct worker* worker)
+{
+ if(!dump_rrset_cache(ssl, worker))
+ return 0;
+ if(!dump_msg_cache(ssl, worker))
+ return 0;
+ return ssl_printf(ssl, "EOF\n");
+}
+
+/** read a line from ssl into buffer */
+static int
+ssl_read_buf(SSL* ssl, ldns_buffer* buf)
+{
+ return ssl_read_line(ssl, (char*)ldns_buffer_begin(buf),
+ ldns_buffer_capacity(buf));
+}
+
+/** check fixed text on line */
+static int
+read_fixed(SSL* ssl, ldns_buffer* buf, const char* str)
+{
+ if(!ssl_read_buf(ssl, buf)) return 0;
+ return (strcmp((char*)ldns_buffer_begin(buf), str) == 0);
+}
+
+/** load an RR into rrset */
+static int
+load_rr(SSL* ssl, ldns_buffer* buf, struct regional* region,
+ struct ub_packed_rrset_key* rk, struct packed_rrset_data* d,
+ unsigned int i, int is_rrsig, int* go_on, uint32_t now)
+{
+ ldns_rr* rr;
+ ldns_status status;
+
+ /* read the line */
+ if(!ssl_read_buf(ssl, buf))
+ return 0;
+ if(strncmp((char*)ldns_buffer_begin(buf), "BADRR\n", 6) == 0) {
+ *go_on = 0;
+ return 1;
+ }
+ log_info("rd %s", (char*)ldns_buffer_begin(buf));
+ status = ldns_rr_new_frm_str(&rr, (char*)ldns_buffer_begin(buf),
+ LDNS_DEFAULT_TTL, NULL, NULL);
+ if(status != LDNS_STATUS_OK) {
+ (void)ssl_printf(ssl, "error cannot parse rr :%s: %s\n",
+ ldns_get_errorstr_by_id(status),
+ (char*)ldns_buffer_begin(buf));
+ return 0;
+ }
+ if(is_rrsig && ldns_rr_get_type(rr) != LDNS_RR_TYPE_RRSIG) {
+ (void)ssl_printf(ssl, "error expected rrsig but got %s\n",
+ (char*)ldns_buffer_begin(buf));
+ return 0;
+ }
+
+ /* convert ldns rr into packed_rr */
+ d->rr_ttl[i] = ldns_rr_ttl(rr) + now;
+ ldns_buffer_clear(buf);
+ ldns_buffer_skip(buf, 2);
+ status = ldns_rr_rdata2buffer_wire(buf, rr);
+ if(status != LDNS_STATUS_OK) {
+ (void)ssl_printf(ssl, "error cannot rr2wire :%s\n",
+ ldns_get_errorstr_by_id(status));
+ ldns_rr_free(rr);
+ return 0;
+ }
+ ldns_buffer_flip(buf);
+ ldns_buffer_write_u16_at(buf, 0, ldns_buffer_limit(buf) - 2);
+
+ d->rr_len[i] = ldns_buffer_limit(buf);
+ d->rr_data[i] = (uint8_t*)regional_alloc_init(region,
+ ldns_buffer_begin(buf), ldns_buffer_limit(buf));
+ if(!d->rr_data[i]) {
+ ldns_rr_free(rr);
+ (void)ssl_printf(ssl, "error out of memory\n");
+ return 0;
+ }
+
+ /* if first entry, fill the key structure */
+ if(i==0) {
+ rk->rk.type = htons(ldns_rr_get_type(rr));
+ rk->rk.rrset_class = htons(ldns_rr_get_class(rr));
+ ldns_buffer_clear(buf);
+ status = ldns_dname2buffer_wire(buf, ldns_rr_owner(rr));
+ ldns_buffer_flip(buf);
+ rk->rk.dname_len = ldns_buffer_limit(buf);
+ rk->rk.dname = regional_alloc_init(region,
+ ldns_buffer_begin(buf), ldns_buffer_limit(buf));
+ if(!rk->rk.dname) {
+ (void)ssl_printf(ssl, "error out of memory\n");
+ ldns_rr_free(rr);
+ return 0;
+ }
+ }
+ ldns_rr_free(rr);
+
+ return 1;
+}
+
+/** move entry into cache */
+static int
+move_into_cache(SSL* ssl, struct ub_packed_rrset_key* k,
+ struct packed_rrset_data* d, struct worker* worker)
+{
+ struct ub_packed_rrset_key* ak;
+ struct packed_rrset_data* ad;
+ size_t s, i, num = d->count + d->rrsig_count;
+ struct rrset_ref ref;
+ uint8_t* p;
+
+ ak = alloc_special_obtain(&worker->alloc);
+ if(!ak) {
+ (void)ssl_printf(ssl, "error out of memory\n");
+ return 0;
+ }
+ ak->entry.data = NULL;
+ ak->rk = k->rk;
+ ak->entry.hash = rrset_key_hash(&k->rk);
+ ak->rk.dname = (uint8_t*)memdup(k->rk.dname, k->rk.dname_len);
+ if(!ak->rk.dname) {
+ (void)ssl_printf(ssl, "error out of memory\n");
+ ub_packed_rrset_parsedelete(ak, &worker->alloc);
+ return 0;
+ }
+ s = sizeof(*ad) + (sizeof(size_t) + sizeof(uint8_t*) +
+ sizeof(uint32_t))* num;
+ for(i=0; i<num; i++)
+ s += d->rr_len[i];
+ ad = (struct packed_rrset_data*)malloc(s);
+ if(!ad) {
+ ub_packed_rrset_parsedelete(ak, &worker->alloc);
+ return 0;
+ }
+ p = (uint8_t*)ad;
+ memmove(p, d, sizeof(*ad));
+ p += sizeof(*ad);
+ memmove(p, &d->rr_len[0], sizeof(size_t)*num);
+ p += sizeof(size_t)*num;
+ memmove(p, &d->rr_data[0], sizeof(uint8_t*)*num);
+ p += sizeof(uint8_t*)*num;
+ memmove(p, &d->rr_ttl[0], sizeof(uint32_t)*num);
+ p += sizeof(uint32_t)*num;
+ for(i=0; i<num; i++) {
+ memmove(p, d->rr_data[i], d->rr_len[i]);
+ p += d->rr_len[i];
+ }
+ packed_rrset_ptr_fixup(ad);
+
+ ak->entry.data = ad;
+
+ ref.key = ak;
+ ref.id = ak->id;
+ (void)rrset_cache_update(worker->env.rrset_cache, &ref,
+ &worker->alloc, *worker->env.now);
+ return 1;
+}
+
+/** load an rrset entry */
+static int
+load_rrset(SSL* ssl, ldns_buffer* buf, struct worker* worker)
+{
+ char* s = (char*)ldns_buffer_begin(buf);
+ struct regional* region = worker->scratchpad;
+ struct ub_packed_rrset_key* rk;
+ struct packed_rrset_data* d;
+ unsigned int ttl, rr_count, rrsig_count, trust, security;
+ unsigned int i;
+ int go_on = 1;
+ regional_free_all(region);
+
+ rk = (struct ub_packed_rrset_key*)regional_alloc_zero(region,
+ sizeof(*rk));
+ d = (struct packed_rrset_data*)regional_alloc_zero(region, sizeof(*d));
+ if(!rk || !d) {
+ (void) ssl_printf(ssl, "error out of memory\n");
+ return 0;
+ }
+
+ if(strncmp(s, ";rrset", 6) != 0) {
+ (void)ssl_printf(ssl, "error expected ';rrset' but got %s\n",
+ s);
+ return 0;
+ }
+ s += 6;
+ if(strncmp(s, " nsec_apex", 10) == 0) {
+ s += 10;
+ rk->rk.flags |= PACKED_RRSET_NSEC_AT_APEX;
+ }
+ if(sscanf(s, " %u %u %u %u %u", &ttl, &rr_count, &rrsig_count,
+ &trust, &security) != 5) {
+ (void)ssl_printf(ssl, "error bad rrset spec %s\n", s);
+ return 0;
+ }
+ if(rr_count == 0 && rrsig_count == 0)
+ return 0;
+ d->count = (size_t)rr_count;
+ d->rrsig_count = (size_t)rrsig_count;
+ d->security = (enum sec_status)security;
+ d->trust = (enum rrset_trust)trust;
+ d->ttl = (uint32_t)ttl + *worker->env.now;
+
+ d->rr_len = regional_alloc_zero(region,
+ sizeof(size_t)*(d->count+d->rrsig_count));
+ d->rr_ttl = regional_alloc_zero(region,
+ sizeof(uint32_t)*(d->count+d->rrsig_count));
+ d->rr_data = regional_alloc_zero(region,
+ sizeof(uint8_t*)*(d->count+d->rrsig_count));
+ if(!d->rr_len || !d->rr_ttl || !d->rr_data) {
+ (void) ssl_printf(ssl, "error out of memory\n");
+ return 0;
+ }
+
+ /* read the rr's themselves */
+ for(i=0; i<rr_count; i++) {
+ if(!load_rr(ssl, buf, region, rk, d, i, 0,
+ &go_on, *worker->env.now)) {
+ return 0;
+ }
+ }
+ for(i=0; i<rrsig_count; i++) {
+ if(!load_rr(ssl, buf, region, rk, d, i+rr_count, 1,
+ &go_on, *worker->env.now)) {
+ return 0;
+ }
+ }
+ if(!go_on) {
+ /* skip this entry */
+ return 1;
+ }
+
+ return move_into_cache(ssl, rk, d, worker);
+}
+
+/** load rrset cache */
+static int
+load_rrset_cache(SSL* ssl, struct worker* worker)
+{
+ ldns_buffer* buf = worker->env.scratch_buffer;
+ if(!read_fixed(ssl, buf, "START_RRSET_CACHE")) return 0;
+ while(ssl_read_buf(ssl, buf) &&
+ strcmp((char*)ldns_buffer_begin(buf), "END_RRSET_CACHE")!=0) {
+ if(!load_rrset(ssl, buf, worker))
+ return 0;
+ }
+ return 1;
+}
+
+/** load a msg rrset reference */
+static int
+load_ref(SSL* ssl, ldns_buffer* buf, struct worker* worker,
+ struct regional *region, struct ub_packed_rrset_key** rrset,
+ int* go_on)
+{
+ ldns_rr* rr;
+ ldns_status status;
+ char* s = (char*)ldns_buffer_begin(buf);
+
+ /* read line */
+ if(!ssl_read_buf(ssl, buf))
+ return 0;
+
+ s += 4;
+ /* skip three words */
+ s = strchr(s, ' ');
+ if(s) s = strchr(s+1, ' ');
+ if(s) s = strchr(s+1, ' ');
+ if(!s) {
+ (void)ssl_printf(ssl, "error ref line too short, %s\n", s);
+ return 0;
+ }
+ s[0] = 0;
+
+ status = ldns_rr_new_question_frm_str(&rr,
+ (char*)ldns_buffer_begin(buf), NULL, NULL);
+ if(status != LDNS_STATUS_OK) {
+ (void)ssl_printf(ssl, "error cannot parse ref: %s %s\n",
+ ldns_get_errorstr_by_id(status), s);
+ ldns_rr_free(rr);
+ return 0;
+ }
+ ldns_rr_free(rr);
+
+
+ /* lookup in cache */
+ /* store in result */
+ return 1;
+}
+
+/** load a msg entry */
+static int
+load_msg(SSL* ssl, ldns_buffer* buf, struct worker* worker)
+{
+ struct regional* region = worker->scratchpad;
+ struct query_info qinf;
+ struct reply_info rep;
+ char* s = (char*)ldns_buffer_begin(buf);
+ ldns_rr* rr;
+ ldns_status status;
+ unsigned int flags, qdcount, ttl, security, an, ns, ar;
+ size_t i;
+ int go_on = 1;
+
+ regional_free_all(region);
+
+ if(strncmp(s, "msg ", 4) != 0) {
+ (void)ssl_printf(ssl, "error expected msg but got %s\n", s);
+ return 0;
+ }
+ s += 4;
+ /* skip three words */
+ s = strchr(s, ' ');
+ if(s) s = strchr(s+1, ' ');
+ if(s) s = strchr(s+1, ' ');
+ if(!s) {
+ (void)ssl_printf(ssl, "error msg line too short, %s\n", s);
+ return 0;
+ }
+ s[0] = 0;
+
+ status = ldns_rr_new_question_frm_str(&rr,
+ (char*)ldns_buffer_at(buf, 4), NULL, NULL);
+ if(status != LDNS_STATUS_OK) {
+ (void)ssl_printf(ssl, "error cannot parse query: %s %s\n",
+ ldns_get_errorstr_by_id(status), s);
+ ldns_rr_free(rr);
+ return 0;
+ }
+
+ /* read remainder of line */
+ if(sscanf(s+1, " %u %u %u %u %u %u %u", &flags, &qdcount, &ttl,
+ &security, &an, &ns, &ar) != 7) {
+ (void)ssl_printf(ssl, "error cannot parse numbers: %s\n", s+1);
+ ldns_rr_free(rr);
+ return 0;
+ }
+
+ /* fill qinfo and repinfo */
+ qinf.qtype = ldns_rr_get_type(rr);
+ qinf.qclass = ldns_rr_get_class(rr);
+ ldns_buffer_clear(buf);
+ status = ldns_dname2buffer_wire(buf, ldns_rr_owner(rr));
+ if(status != LDNS_STATUS_OK) {
+ (void)ssl_printf(ssl, "error cannot dname2wire: %s\n",
+ ldns_get_errorstr_by_id(status));
+ ldns_rr_free(rr);
+ return 0;
+ }
+ ldns_rr_free(rr);
+ ldns_buffer_flip(buf);
+ qinf.qname_len = ldns_buffer_limit(buf);
+ qinf.qname = (uint8_t*)regional_alloc_init(region,
+ ldns_buffer_begin(buf), ldns_buffer_limit(buf));
+ if(!qinf.qname) {
+ (void)ssl_printf(ssl, "error out of memory\n");
+ return 0;
+ }
+ rep.flags = (uint16_t)flags;
+ rep.qdcount = (uint16_t)qdcount;
+ rep.ttl = (uint32_t)ttl;
+ rep.security = (enum sec_status)security;
+ rep.an_numrrsets = (size_t)an;
+ rep.ns_numrrsets = (size_t)ns;
+ rep.ar_numrrsets = (size_t)ar;
+ rep.rrset_count = (size_t)an+(size_t)ns+(size_t)ar;
+ rep.rrsets = (struct ub_packed_rrset_key**)regional_alloc_zero(
+ region, sizeof(struct ub_packed_rrset_key*)*rep.rrset_count);
+
+ /* fill repinfo with references */
+ for(i=0; i<rep.rrset_count; i++) {
+ if(!load_ref(ssl, buf, worker, region, &rep.rrsets[i],
+ &go_on)) {
+ return 0;
+ }
+ }
+
+ if(!go_on)
+ return 1; /* skip this one, not all references satisfied */
+
+ if(!dns_cache_store(&worker->env, &qinf, &rep, 0)) {
+ (void)ssl_printf(ssl, "error out of memory\n");
+ return 0;
+ }
+ return 1;
+}
+
+/** load msg cache */
+static int
+load_msg_cache(SSL* ssl, struct worker* worker)
+{
+ ldns_buffer* buf = worker->env.scratch_buffer;
+ if(!read_fixed(ssl, buf, "START_MSG_CACHE")) return 0;
+ while(ssl_read_buf(ssl, buf) &&
+ strcmp((char*)ldns_buffer_begin(buf), "END_MSG_CACHE")!=0) {
+ if(!load_msg(ssl, buf, worker))
+ return 0;
+ }
+ return 1;
+}
+
+int
+load_cache(SSL* ssl, struct worker* worker)
+{
+ if(!load_rrset_cache(ssl, worker))
+ return 0;
+ if(!load_msg_cache(ssl, worker))
+ return 0;
+ return read_fixed(ssl, worker->env.scratch_buffer, "EOF");
+}
projects / thirdparty / unbound.git / commitdiff
