/*
 * services/localzone.c - local zones authority service.
 *
 * Copyright (c) 2007, 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 COPYRIGHT
 * HOLDER 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 enable local zone authority service.
 */
#include "config.h"
#include "services/localzone.h"
#include "sldns/str2wire.h"
#include "util/regional.h"
#include "util/config_file.h"
#include "util/data/dname.h"
#include "util/data/packed_rrset.h"
#include "util/data/msgencode.h"
#include "util/net_help.h"
#include "util/netevent.h"
#include "util/data/msgreply.h"
#include "util/data/msgparse.h"
#include "util/as112.h"

/* maximum RRs in an RRset, to cap possible 'endless' list RRs.
 * with 16 bytes for an A record, a 64K packet has about 4000 max */
#define LOCALZONE_RRSET_COUNT_MAX 4096

/** print all RRsets in local zone */
static void
local_zone_out(struct local_zone* z)
{
        struct local_data* d;
        struct local_rrset* p;
        RBTREE_FOR(d, struct local_data*, &z->data) {
                for(p = d->rrsets; p; p = p->next) {
                        log_nametypeclass(NO_VERBOSE, "rrset", d->name,
                                ntohs(p->rrset->rk.type),
                                ntohs(p->rrset->rk.rrset_class));
                }
        }
}

static void
local_zone_print(struct local_zone* z)
{
        char buf[64];
        lock_rw_rdlock(&z->lock);
        snprintf(buf, sizeof(buf), "%s zone",
                local_zone_type2str(z->type));
        log_nametypeclass(NO_VERBOSE, buf, z->name, 0, z->dclass);
        local_zone_out(z);
        lock_rw_unlock(&z->lock);
}

void local_zones_print(struct local_zones* zones)
{
        struct local_zone* z;
        lock_rw_rdlock(&zones->lock);
        log_info("number of auth zones %u", (unsigned)zones->ztree.count);
        RBTREE_FOR(z, struct local_zone*, &zones->ztree) {
                local_zone_print(z);
        }
        lock_rw_unlock(&zones->lock);
}

struct local_zones* 
local_zones_create(void)
{
        struct local_zones* zones = (struct local_zones*)calloc(1, 
                sizeof(*zones));
        if(!zones)
                return NULL;
        rbtree_init(&zones->ztree, &local_zone_cmp);
        lock_rw_init(&zones->lock);
        lock_protect(&zones->lock, &zones->ztree, sizeof(zones->ztree));
        /* also lock protects the rbnode's in struct local_zone */
        return zones;
}

/** helper traverse to delete zones */
static void 
lzdel(rbnode_type* n, void* ATTR_UNUSED(arg))
{
        struct local_zone* z = (struct local_zone*)n->key;
        local_zone_delete(z);
}

void 
local_zones_delete(struct local_zones* zones)
{
        if(!zones)
                return;
        lock_rw_destroy(&zones->lock);
        /* walk through zones and delete them all */
        traverse_postorder(&zones->ztree, lzdel, NULL);
        free(zones);
}

void 
local_zone_delete(struct local_zone* z)
{
        if(!z)
                return;
        lock_rw_destroy(&z->lock);
        regional_destroy(z->region);
        free(z->name);
        free(z->taglist);
        free(z);
}

int 
local_zone_cmp(const void* z1, const void* z2)
{
        /* first sort on class, so that hierarchy can be maintained within
         * a class */
        struct local_zone* a = (struct local_zone*)z1;
        struct local_zone* b = (struct local_zone*)z2;
        int m;
        if(a->dclass != b->dclass) {
                if(a->dclass < b->dclass)
                        return -1;
                return 1;
        }
        return dname_lab_cmp(a->name, a->namelabs, b->name, b->namelabs, &m);
}

int 
local_data_cmp(const void* d1, const void* d2)
{
        struct local_data* a = (struct local_data*)d1;
        struct local_data* b = (struct local_data*)d2;
        int m;
        return dname_canon_lab_cmp(a->name, a->namelabs, b->name, 
                b->namelabs, &m);
}

/* form wireformat from text format domain name */
int
parse_dname(const char* str, uint8_t** res, size_t* len, int* labs)
{
        *res = sldns_str2wire_dname(str, len);
        *labs = 0;
        if(!*res) {
                log_err("cannot parse name %s", str);
                return 0;
        }
        *labs = dname_count_size_labels(*res, len);
        return 1;
}

/** create a new localzone */
static struct local_zone*
local_zone_create(uint8_t* nm, size_t len, int labs, 
        enum localzone_type t, uint16_t dclass)
{
        struct local_zone* z = (struct local_zone*)calloc(1, sizeof(*z));
        if(!z) {
                return NULL;
        }
        z->node.key = z;
        z->dclass = dclass;
        z->type = t;
        z->name = nm;
        z->namelen = len;
        z->namelabs = labs;
        lock_rw_init(&z->lock);
        z->region = regional_create_nochunk(sizeof(struct regional));
        if(!z->region) {
                free(z);
                return NULL;
        }
        rbtree_init(&z->data, &local_data_cmp);
        lock_protect(&z->lock, &z->parent, sizeof(*z)-sizeof(rbnode_type));
        /* also the zones->lock protects node, parent, name*, class */
        return z;
}

/** enter a new zone with allocated dname returns with WRlock */
static struct local_zone*
lz_enter_zone_dname(struct local_zones* zones, uint8_t* nm, size_t len, 
        int labs, enum localzone_type t, uint16_t c)
{
        struct local_zone* z = local_zone_create(nm, len, labs, t, c);
        if(!z) {
                free(nm);
                log_err("out of memory");
                return NULL;
        }

        /* add to rbtree */
        lock_rw_wrlock(&zones->lock);
        lock_rw_wrlock(&z->lock);
        if(!rbtree_insert(&zones->ztree, &z->node)) {
                struct local_zone* oldz;
                char str[LDNS_MAX_DOMAINLEN];
                dname_str(nm, str);
                log_warn("duplicate local-zone %s", str);
                lock_rw_unlock(&z->lock);
                /* save zone name locally before deallocation,
                 * otherwise, nm is gone if we zone_delete now. */
                oldz = z;
                /* find the correct zone, so not an error for duplicate */
                z = local_zones_find(zones, nm, len, labs, c);
                lock_rw_wrlock(&z->lock);
                lock_rw_unlock(&zones->lock);
                local_zone_delete(oldz);
                return z;
        }
        lock_rw_unlock(&zones->lock);
        return z;
}

/** enter a new zone */
struct local_zone*
lz_enter_zone(struct local_zones* zones, const char* name, const char* type,
        uint16_t dclass)
{
        struct local_zone* z;
        enum localzone_type t;
        uint8_t* nm;
        size_t len;
        int labs;
        if(!parse_dname(name, &nm, &len, &labs)) {
                log_err("bad zone name %s %s", name, type);
                return NULL;
        }
        if(!local_zone_str2type(type, &t)) {
                log_err("bad lz_enter_zone type %s %s", name, type);
                free(nm);
                return NULL;
        }
        if(!(z=lz_enter_zone_dname(zones, nm, len, labs, t, dclass))) {
                log_err("could not enter zone %s %s", name, type);
                return NULL;
        }
        return z;
}

int
rrstr_get_rr_content(const char* str, uint8_t** nm, uint16_t* type,
        uint16_t* dclass, time_t* ttl, uint8_t* rr, size_t len,
        uint8_t** rdata, size_t* rdata_len)
{
        size_t dname_len = 0;
        int e = sldns_str2wire_rr_buf(str, rr, &len, &dname_len, 3600,
                NULL, 0, NULL, 0);
        if(e) {
                log_err("error parsing local-data at %d: '%s': %s",
                        LDNS_WIREPARSE_OFFSET(e), str,
                        sldns_get_errorstr_parse(e));
                return 0;
        }
        *nm = memdup(rr, dname_len);
        if(!*nm) {
                log_err("out of memory");
                return 0;
        }
        *dclass = sldns_wirerr_get_class(rr, len, dname_len);
        *type = sldns_wirerr_get_type(rr, len, dname_len);
        *ttl = (time_t)sldns_wirerr_get_ttl(rr, len, dname_len);
        *rdata = sldns_wirerr_get_rdatawl(rr, len, dname_len);
        *rdata_len = sldns_wirerr_get_rdatalen(rr, len, dname_len)+2;
        return 1;
}

/** return name and class of rr; parses string */
static int
get_rr_nameclass(const char* str, uint8_t** nm, uint16_t* dclass,
        uint16_t* dtype)
{
        uint8_t rr[LDNS_RR_BUF_SIZE];
        size_t len = sizeof(rr), dname_len = 0;
        int s = sldns_str2wire_rr_buf(str, rr, &len, &dname_len, 3600,
                NULL, 0, NULL, 0);
        if(s != 0) {
                log_err("error parsing local-data at %d '%s': %s",
                        LDNS_WIREPARSE_OFFSET(s), str,
                        sldns_get_errorstr_parse(s));
                return 0;
        }
        *nm = memdup(rr, dname_len);
        *dclass = sldns_wirerr_get_class(rr, len, dname_len);
        *dtype = sldns_wirerr_get_type(rr, len, dname_len);
        if(!*nm) {
                log_err("out of memory");
                return 0;
        }
        return 1;
}

/**
 * Find an rrset in local data structure.
 * @param data: local data domain name structure.
 * @param type: type to look for (host order).
 * @param alias_ok: 1 if matching a non-exact, alias type such as CNAME is
 * allowed.  otherwise 0.
 * @return rrset pointer or NULL if not found.
 */
static struct local_rrset*
local_data_find_type(struct local_data* data, uint16_t type, int alias_ok)
{
        struct local_rrset* p, *cname = NULL;
        type = htons(type);
        for(p = data->rrsets; p; p = p->next) {
                if(p->rrset->rk.type == type)
                        return p;
                if(alias_ok && p->rrset->rk.type == htons(LDNS_RR_TYPE_CNAME))
                        cname = p;
        }
        if(alias_ok)
                return cname;
        return NULL;
}

/** check for RR duplicates */
static int
rr_is_duplicate(struct packed_rrset_data* pd, uint8_t* rdata, size_t rdata_len)
{
        size_t i;
        for(i=0; i<pd->count; i++) {
                if(pd->rr_len[i] == rdata_len &&
                        memcmp(pd->rr_data[i], rdata, rdata_len) == 0)
                        return 1;
        }
        return 0;
}

/** new local_rrset */
static struct local_rrset*
new_local_rrset(struct regional* region, struct local_data* node,
        uint16_t rrtype, uint16_t rrclass)
{
        struct packed_rrset_data* pd;
        struct local_rrset* rrset = (struct local_rrset*)
                regional_alloc_zero(region, sizeof(*rrset));
        if(!rrset) {
                log_err("out of memory");
                return NULL;
        }
        rrset->next = node->rrsets;
        node->rrsets = rrset;
        rrset->rrset = (struct ub_packed_rrset_key*)
                regional_alloc_zero(region, sizeof(*rrset->rrset));
        if(!rrset->rrset) {
                log_err("out of memory");
                return NULL;
        }
        rrset->rrset->entry.key = rrset->rrset;
        pd = (struct packed_rrset_data*)regional_alloc_zero(region,
                sizeof(*pd));
        if(!pd) {
                log_err("out of memory");
                return NULL;
        }
        pd->trust = rrset_trust_prim_noglue;
        pd->security = sec_status_insecure;
        rrset->rrset->entry.data = pd;
        rrset->rrset->rk.dname = node->name;
        rrset->rrset->rk.dname_len = node->namelen;
        rrset->rrset->rk.type = htons(rrtype);
        rrset->rrset->rk.rrset_class = htons(rrclass);
        return rrset;
}

/** insert RR into RRset data structure; Wastes a couple of bytes */
int
rrset_insert_rr(struct regional* region, struct packed_rrset_data* pd,
        uint8_t* rdata, size_t rdata_len, time_t ttl, const char* rrstr)
{
        size_t* oldlen = pd->rr_len;
        time_t* oldttl = pd->rr_ttl;
        uint8_t** olddata = pd->rr_data;

        /* add RR to rrset */
        if(pd->count > LOCALZONE_RRSET_COUNT_MAX) {
                log_warn("RRset '%s' has more than %d records, record ignored",
                        rrstr, LOCALZONE_RRSET_COUNT_MAX);
                return 1;
        }
        pd->count++;
        pd->rr_len = regional_alloc(region, sizeof(*pd->rr_len)*pd->count);
        pd->rr_ttl = regional_alloc(region, sizeof(*pd->rr_ttl)*pd->count);
        pd->rr_data = regional_alloc(region, sizeof(*pd->rr_data)*pd->count);
        if(!pd->rr_len || !pd->rr_ttl || !pd->rr_data) {
                log_err("out of memory");
                return 0;
        }
        if(pd->count > 1) {
                memcpy(pd->rr_len+1, oldlen, 
                        sizeof(*pd->rr_len)*(pd->count-1));
                memcpy(pd->rr_ttl+1, oldttl, 
                        sizeof(*pd->rr_ttl)*(pd->count-1));
                memcpy(pd->rr_data+1, olddata, 
                        sizeof(*pd->rr_data)*(pd->count-1));
        }
        pd->rr_len[0] = rdata_len;
        pd->rr_ttl[0] = ttl;
        pd->rr_data[0] = regional_alloc_init(region, rdata, rdata_len);
        if(!pd->rr_data[0]) {
                log_err("out of memory");
                return 0;
        }
        return 1;
}

/** Delete RR from local-zone RRset, wastes memory as the deleted RRs cannot be
 * free'd (regionally alloc'd) */
int
local_rrset_remove_rr(struct packed_rrset_data* pd, size_t index)
{
        log_assert(pd->count > 0);
        if(index >= pd->count) {
                log_warn("Trying to remove RR with out of bound index");
                return 0;
        }
        if(index + 1 < pd->count) {
                /* not removing last element */
                size_t nexti = index + 1;
                size_t num = pd->count - nexti;
                memmove(pd->rr_len+index, pd->rr_len+nexti, sizeof(*pd->rr_len)*num);
                memmove(pd->rr_ttl+index, pd->rr_ttl+nexti, sizeof(*pd->rr_ttl)*num);
                memmove(pd->rr_data+index, pd->rr_data+nexti, sizeof(*pd->rr_data)*num);
        }
        pd->count--;
        return 1;
}

struct local_data* 
local_zone_find_data(struct local_zone* z, uint8_t* nm, size_t nmlen, int nmlabs)
{
        struct local_data key;
        key.node.key = &key;
        key.name = nm;
        key.namelen = nmlen;
        key.namelabs = nmlabs;
        return (struct local_data*)rbtree_search(&z->data, &key.node);
}

/** find a node, create it if not and all its empty nonterminal parents */
static int
lz_find_create_node(struct local_zone* z, uint8_t* nm, size_t nmlen, 
        int nmlabs, struct local_data** res)
{
        struct local_data* ld = local_zone_find_data(z, nm, nmlen, nmlabs);
        if(!ld) {
                /* create a domain name to store rr. */
                ld = (struct local_data*)regional_alloc_zero(z->region,
                        sizeof(*ld));
                if(!ld) {
                        log_err("out of memory adding local data");
                        return 0;
                }
                ld->node.key = ld;
                ld->name = regional_alloc_init(z->region, nm, nmlen);
                if(!ld->name) {
                        log_err("out of memory");
                        return 0;
                }
                ld->namelen = nmlen;
                ld->namelabs = nmlabs;
                if(!rbtree_insert(&z->data, &ld->node)) {
                        log_assert(0); /* duplicate name */
                }
                /* see if empty nonterminals need to be created */
                if(nmlabs > z->namelabs) {
                        dname_remove_label(&nm, &nmlen);
                        if(!lz_find_create_node(z, nm, nmlen, nmlabs-1, res))
                                return 0;
                }
        }
        *res = ld;
        return 1;
}

/* Mark the SOA record for the zone. This only marks the SOA rrset; the data
 * for the RR is entered later on local_zone_enter_rr() as with the other
 * records. An artificial soa_negative record with a modified TTL (minimum of
 * the TTL and the SOA.MINIMUM) is also created and marked for usage with
 * negative answers and to avoid allocations during those answers. */
static int
lz_mark_soa_for_zone(struct local_zone* z, struct ub_packed_rrset_key* soa_rrset,
        uint8_t* rdata, size_t rdata_len, time_t ttl, const char* rrstr)
{
        struct packed_rrset_data* pd = (struct packed_rrset_data*)
                regional_alloc_zero(z->region, sizeof(*pd));
        struct ub_packed_rrset_key* rrset_negative = (struct ub_packed_rrset_key*)
                regional_alloc_zero(z->region, sizeof(*rrset_negative));
        time_t minimum;
        if(!rrset_negative||!pd) {
                log_err("out of memory");
                return 0;
        }
        /* Mark the original SOA record and then continue with the negative one. */
        z->soa = soa_rrset;
        rrset_negative->entry.key = rrset_negative;
        pd->trust = rrset_trust_prim_noglue;
        pd->security = sec_status_insecure;
        rrset_negative->entry.data = pd;
        rrset_negative->rk.dname = soa_rrset->rk.dname;
        rrset_negative->rk.dname_len = soa_rrset->rk.dname_len;
        rrset_negative->rk.type = soa_rrset->rk.type;
        rrset_negative->rk.rrset_class = soa_rrset->rk.rrset_class;
        if(!rrset_insert_rr(z->region, pd, rdata, rdata_len, ttl, rrstr))
                return 0;
        /* last 4 bytes are minimum ttl in network format */
        if(pd->count == 0 || pd->rr_len[0] < 2+4)
                return 0;
        minimum = (time_t)sldns_read_uint32(pd->rr_data[0]+(pd->rr_len[0]-4));
        minimum = ttl<minimum?ttl:minimum;
        pd->ttl = minimum;
        pd->rr_ttl[0] = minimum;

        z->soa_negative = rrset_negative;
        return 1;
}

int
local_zone_enter_rr(struct local_zone* z, uint8_t* nm, size_t nmlen,
        int nmlabs, uint16_t rrtype, uint16_t rrclass, time_t ttl,
        uint8_t* rdata, size_t rdata_len, const char* rrstr)
{
        struct local_data* node;
        struct local_rrset* rrset;
        struct packed_rrset_data* pd;

        if(!lz_find_create_node(z, nm, nmlen, nmlabs, &node)) {
                return 0;
        }
        log_assert(node);

        /* Reject it if we would end up having CNAME and other data (including
         * another CNAME) for a redirect zone. */
        if((z->type == local_zone_redirect ||
                z->type == local_zone_inform_redirect) && node->rrsets) {
                const char* othertype = NULL;
                if (rrtype == LDNS_RR_TYPE_CNAME)
                        othertype = "other";
                else if (node->rrsets->rrset->rk.type ==
                         htons(LDNS_RR_TYPE_CNAME)) {
                        othertype = "CNAME";
                }
                if(othertype) {
                        log_err("local-data '%s' in redirect zone must not "
                                "coexist with %s local-data", rrstr, othertype);
                        return 0;
                }
        }
        rrset = local_data_find_type(node, rrtype, 0);
        if(!rrset) {
                rrset = new_local_rrset(z->region, node, rrtype, rrclass);
                if(!rrset)
                        return 0;
                if(query_dname_compare(node->name, z->name) == 0) {
                        if(rrtype == LDNS_RR_TYPE_NSEC)
                          rrset->rrset->rk.flags = PACKED_RRSET_NSEC_AT_APEX;
                        if(rrtype == LDNS_RR_TYPE_SOA &&
                                !lz_mark_soa_for_zone(z, rrset->rrset, rdata, rdata_len, ttl,
                                        rrstr))
                                return 0;
                }
        } 
        pd = (struct packed_rrset_data*)rrset->rrset->entry.data;
        log_assert(rrset && pd);

        /* check for duplicate RR */
        if(rr_is_duplicate(pd, rdata, rdata_len)) {
                verbose(VERB_ALGO, "ignoring duplicate RR: %s", rrstr);
                return 1;
        } 
        return rrset_insert_rr(z->region, pd, rdata, rdata_len, ttl, rrstr);
}

/** enter data RR into auth zone */
static int
lz_enter_rr_into_zone(struct local_zone* z, const char* rrstr)
{
        uint8_t* nm;
        size_t nmlen;
        int nmlabs, ret;
        uint16_t rrtype = 0, rrclass = 0;
        time_t ttl = 0;
        uint8_t rr[LDNS_RR_BUF_SIZE];
        uint8_t* rdata;
        size_t rdata_len;
        if(!rrstr_get_rr_content(rrstr, &nm, &rrtype, &rrclass, &ttl, rr,
                sizeof(rr), &rdata, &rdata_len)) {
                log_err("bad local-data: %s", rrstr);
                return 0;
        }
        log_assert(z->dclass == rrclass);
        if((z->type == local_zone_redirect ||
                z->type == local_zone_inform_redirect) &&
                query_dname_compare(z->name, nm) != 0) {
                log_err("local-data in redirect zone must reside at top of zone"
                        ", not at %s", rrstr);
                free(nm);
                return 0;
        }
        nmlabs = dname_count_size_labels(nm, &nmlen);
        ret = local_zone_enter_rr(z, nm, nmlen, nmlabs, rrtype, rrclass, ttl,
                rdata, rdata_len, rrstr);
        free(nm);
        return ret;
}

/** enter a data RR into auth data; a zone for it must exist */
static int
lz_enter_rr_str(struct local_zones* zones, const char* rr)
{
        uint8_t* rr_name;
        uint16_t rr_class, rr_type;
        size_t len;
        int labs;
        struct local_zone* z;
        int r;
        if(!get_rr_nameclass(rr, &rr_name, &rr_class, &rr_type)) {
                log_err("bad rr %s", rr);
                return 0;
        }
        labs = dname_count_size_labels(rr_name, &len);
        lock_rw_rdlock(&zones->lock);
        z = local_zones_lookup(zones, rr_name, len, labs, rr_class, rr_type);
        if(!z) {
                lock_rw_unlock(&zones->lock);
                fatal_exit("internal error: no zone for rr %s", rr);
        }
        lock_rw_wrlock(&z->lock);
        lock_rw_unlock(&zones->lock);
        free(rr_name);
        r = lz_enter_rr_into_zone(z, rr);
        lock_rw_unlock(&z->lock);
        return r;
}

/** enter tagstring into zone */
static int
lz_enter_zone_tag(struct local_zones* zones, char* zname, uint8_t* list,
        size_t len, uint16_t rr_class)
{
        uint8_t dname[LDNS_MAX_DOMAINLEN+1];
        size_t dname_len = sizeof(dname);
        int dname_labs, r = 0;
        struct local_zone* z;

        if(sldns_str2wire_dname_buf(zname, dname, &dname_len) != 0) {
                log_err("cannot parse zone name in local-zone-tag: %s", zname);
                return 0;
        }
        dname_labs = dname_count_labels(dname);
        
        lock_rw_rdlock(&zones->lock);
        z = local_zones_find(zones, dname, dname_len, dname_labs, rr_class);
        if(!z) {
                lock_rw_unlock(&zones->lock);
                log_err("no local-zone for tag %s", zname);
                return 0;
        }
        lock_rw_wrlock(&z->lock);
        lock_rw_unlock(&zones->lock);
        free(z->taglist);
        z->taglist = memdup(list, len);
        z->taglen = len;
        if(z->taglist)
                r = 1;
        lock_rw_unlock(&z->lock);
        return r;
}

/** enter override into zone */
static int
lz_enter_override(struct local_zones* zones, char* zname, char* netblock,
        char* type, uint16_t rr_class)
{
        uint8_t dname[LDNS_MAX_DOMAINLEN+1];
        size_t dname_len = sizeof(dname);
        int dname_labs;
        struct sockaddr_storage addr;
        int net;
        socklen_t addrlen;
        struct local_zone* z;
        enum localzone_type t;

        /* parse zone name */
        if(sldns_str2wire_dname_buf(zname, dname, &dname_len) != 0) {
                log_err("cannot parse zone name in local-zone-override: %s %s",
                        zname, netblock);
                return 0;
        }
        dname_labs = dname_count_labels(dname);

        /* parse netblock */
        if(!netblockstrtoaddr(netblock, UNBOUND_DNS_PORT, &addr, &addrlen,
                &net)) {
                log_err("cannot parse netblock in local-zone-override: %s %s",
                        zname, netblock);
                return 0;
        }

        /* parse zone type */
        if(!local_zone_str2type(type, &t)) {
                log_err("cannot parse type in local-zone-override: %s %s %s",
                        zname, netblock, type);
                return 0;
        }

        /* find localzone entry */
        lock_rw_rdlock(&zones->lock);
        z = local_zones_find(zones, dname, dname_len, dname_labs, rr_class);
        if(!z) {
                lock_rw_unlock(&zones->lock);
                log_err("no local-zone for local-zone-override %s", zname);
                return 0;
        }
        lock_rw_wrlock(&z->lock);
        lock_rw_unlock(&zones->lock);

        /* create netblock addr_tree if not present yet */
        if(!z->override_tree) {
                z->override_tree = (struct rbtree_type*)regional_alloc_zero(
                        z->region, sizeof(*z->override_tree));
                if(!z->override_tree) {
                        lock_rw_unlock(&z->lock);
                        log_err("out of memory");
                        return 0;
                }
                addr_tree_init(z->override_tree);
        }
        /* add new elem to tree */
        if(z->override_tree) {
                struct local_zone_override* n;
                n = (struct local_zone_override*)regional_alloc_zero(
                        z->region, sizeof(*n));
                if(!n) {
                        lock_rw_unlock(&z->lock);
                        log_err("out of memory");
                        return 0;
                }
                n->type = t;
                if(!addr_tree_insert(z->override_tree,
                        (struct addr_tree_node*)n, &addr, addrlen, net)) {
                        lock_rw_unlock(&z->lock);
                        log_err("duplicate local-zone-override %s %s",
                                zname, netblock);
                        return 1;
                }
        }

        lock_rw_unlock(&z->lock);
        return 1;
}

/** parse local-zone: statements */
static int
lz_enter_zones(struct local_zones* zones, struct config_file* cfg)
{
        struct config_str2list* p;
#ifndef THREADS_DISABLED
        struct local_zone* z;
#endif
        for(p = cfg->local_zones; p; p = p->next) {
                if(!(
#ifndef THREADS_DISABLED
                        z=
#endif
                        lz_enter_zone(zones, p->str, p->str2,
                        LDNS_RR_CLASS_IN)))
                        return 0;
                lock_rw_unlock(&z->lock);
        }
        return 1;
}

/** lookup a zone in rbtree; exact match only; SLOW due to parse */
static int
lz_exists(struct local_zones* zones, const char* name)
{
        struct local_zone z;
        z.node.key = &z;
        z.dclass = LDNS_RR_CLASS_IN;
        if(!parse_dname(name, &z.name, &z.namelen, &z.namelabs)) {
                log_err("bad name %s", name);
                return 0;
        }
        lock_rw_rdlock(&zones->lock);
        if(rbtree_search(&zones->ztree, &z.node)) {
                lock_rw_unlock(&zones->lock);
                free(z.name);
                return 1;
        }
        lock_rw_unlock(&zones->lock);
        free(z.name);
        return 0;
}

/** lookup a zone in cfg->nodefault list */
static int
lz_nodefault(struct config_file* cfg, const char* name)
{
        struct config_strlist* p;
        size_t len = strlen(name);
        if(len == 0) return 0;
        if(name[len-1] == '.') len--;

        for(p = cfg->local_zones_nodefault; p; p = p->next) {
                /* compare zone name, lowercase, compare without ending . */
                if(strncasecmp(p->str, name, len) == 0 && 
                        (strlen(p->str) == len || (strlen(p->str)==len+1 &&
                        p->str[len] == '.')))
                        return 1;
        }
        return 0;
}

/** enter (AS112) empty default zone */
static int
add_empty_default(struct local_zones* zones, struct config_file* cfg,
        const char* name)
{
        struct local_zone* z;
        char str[1024]; /* known long enough */
        if(lz_exists(zones, name) || lz_nodefault(cfg, name))
                return 1; /* do not enter default content */
        if(!(z=lz_enter_zone(zones, name, "static", LDNS_RR_CLASS_IN)))
                return 0;
        snprintf(str, sizeof(str), "%s 10800 IN SOA localhost. "
                "nobody.invalid. 1 3600 1200 604800 10800", name);
        if(!lz_enter_rr_into_zone(z, str)) {
                lock_rw_unlock(&z->lock);
                return 0;
        }
        snprintf(str, sizeof(str), "%s 10800 IN NS localhost. ", name);
        if(!lz_enter_rr_into_zone(z, str)) {
                lock_rw_unlock(&z->lock);
                return 0;
        }
        lock_rw_unlock(&z->lock);
        return 1;
}

/** enter default zones */
int local_zone_enter_defaults(struct local_zones* zones, struct config_file* cfg)
{
        struct local_zone* z;
        const char** zstr;

        /* Do not add any default */
        if(cfg->local_zones_disable_default)
                return 1;

        /* this list of zones is from RFC 6303 and RFC 7686 */

        /* block localhost level zones first, then onion and later the LAN zones */

        /* localhost. zone */
        if(!lz_exists(zones, "localhost.") &&
                !lz_nodefault(cfg, "localhost.")) {
                if(!(z=lz_enter_zone(zones, "localhost.", "redirect", 
                        LDNS_RR_CLASS_IN)) ||
                   !lz_enter_rr_into_zone(z,
                        "localhost. 10800 IN NS localhost.") ||
                   !lz_enter_rr_into_zone(z,
                        "localhost. 10800 IN SOA localhost. nobody.invalid. "
                        "1 3600 1200 604800 10800") ||
                   !lz_enter_rr_into_zone(z,
                        "localhost. 10800 IN A 127.0.0.1") ||
                   !lz_enter_rr_into_zone(z,
                        "localhost. 10800 IN AAAA ::1")) {
                        log_err("out of memory adding default zone");
                        if(z) { lock_rw_unlock(&z->lock); }
                        return 0;
                }
                lock_rw_unlock(&z->lock);
        }
        /* reverse ip4 zone */
        if(!lz_exists(zones, "127.in-addr.arpa.") &&
                !lz_nodefault(cfg, "127.in-addr.arpa.")) {
                if(!(z=lz_enter_zone(zones, "127.in-addr.arpa.", "static", 
                        LDNS_RR_CLASS_IN)) ||
                   !lz_enter_rr_into_zone(z,
                        "127.in-addr.arpa. 10800 IN NS localhost.") ||
                   !lz_enter_rr_into_zone(z,
                        "127.in-addr.arpa. 10800 IN SOA localhost. "
                        "nobody.invalid. 1 3600 1200 604800 10800") ||
                   !lz_enter_rr_into_zone(z,
                        "1.0.0.127.in-addr.arpa. 10800 IN PTR localhost.")) {
                        log_err("out of memory adding default zone");
                        if(z) { lock_rw_unlock(&z->lock); }
                        return 0;
                }
                lock_rw_unlock(&z->lock);
        }
        /* reverse ip6 zone */
        if(!lz_exists(zones, "1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa.") &&
                !lz_nodefault(cfg, "1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa.")) {
                if(!(z=lz_enter_zone(zones, "1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa.", "static", 
                        LDNS_RR_CLASS_IN)) ||
                   !lz_enter_rr_into_zone(z,
                        "1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa. 10800 IN NS localhost.") ||
                   !lz_enter_rr_into_zone(z,
                        "1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa. 10800 IN SOA localhost. "
                        "nobody.invalid. 1 3600 1200 604800 10800") ||
                   !lz_enter_rr_into_zone(z,
                        "1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa. 10800 IN PTR localhost.")) {
                        log_err("out of memory adding default zone");
                        if(z) { lock_rw_unlock(&z->lock); }
                        return 0;
                }
                lock_rw_unlock(&z->lock);
        }
        /* home.arpa. zone (RFC 8375) */
        if(!add_empty_default(zones, cfg, "home.arpa.")) {
                log_err("out of memory adding default zone");
                return 0;
        }
        /* resolver.arpa. zone (RFC 9462) */
        if(!add_empty_default(zones, cfg, "resolver.arpa.")) {
                log_err("out of memory adding default zone");
                return 0;
        }
        /* service.arpa. zone (draft-ietf-dnssd-srp-25) */
        if(!add_empty_default(zones, cfg, "service.arpa.")) {
                log_err("out of memory adding default zone");
                return 0;
        }
        /* onion. zone (RFC 7686) */
        if(!add_empty_default(zones, cfg, "onion.")) {
                log_err("out of memory adding default zone");
                return 0;
        }
        /* test. zone (RFC 6761) */
        if(!add_empty_default(zones, cfg, "test.")) {
                log_err("out of memory adding default zone");
                return 0;
        }
        /* invalid. zone (RFC 6761) */
        if(!add_empty_default(zones, cfg, "invalid.")) {
                log_err("out of memory adding default zone");
                return 0;
        }
        /* block AS112 zones, unless asked not to */
        if(!cfg->unblock_lan_zones) {
                for(zstr = as112_zones; *zstr; zstr++) {
                        if(!add_empty_default(zones, cfg, *zstr)) {
                                log_err("out of memory adding default zone");
                                return 0;
                        }
                }
        }
        return 1;
}

/** parse local-zone-override: statements */
static int
lz_enter_overrides(struct local_zones* zones, struct config_file* cfg)
{
        struct config_str3list* p;
        for(p = cfg->local_zone_overrides; p; p = p->next) {
                if(!lz_enter_override(zones, p->str, p->str2, p->str3,
                        LDNS_RR_CLASS_IN))
                        return 0;
        }
        return 1;
}

/* return closest parent in the tree, NULL if none */
static struct local_zone* find_closest_parent(struct local_zone* curr,
        struct local_zone* prev)
{
        struct local_zone* p;
        int m;
        if(!prev || prev->dclass != curr->dclass) return NULL;
        (void)dname_lab_cmp(prev->name, prev->namelabs, curr->name,
                curr->namelabs, &m); /* we know prev is smaller */
                /* sort order like: . com. bla.com. zwb.com. net. */
                /* find the previous, or parent-parent-parent */
        for(p = prev; p; p = p->parent) {
                        /* looking for name with few labels, a parent */
                        if(p->namelabs <= m) {
                                /* ==: since prev matched m, this is closest*/
                                /* <: prev matches more, but is not a parent,
                                 * this one is a (grand)parent */
                        return p;
                }
        }
        return NULL;
                        }

/** setup parent pointers, so that a lookup can be done for closest match */
void
lz_init_parents(struct local_zones* zones)
{
        struct local_zone* node, *prev = NULL;
        lock_rw_wrlock(&zones->lock);
        RBTREE_FOR(node, struct local_zone*, &zones->ztree) {
                lock_rw_wrlock(&node->lock);
                node->parent = find_closest_parent(node, prev);
                prev = node;
                if(node->override_tree)
                        addr_tree_init_parents(node->override_tree);
                lock_rw_unlock(&node->lock);
        }
        lock_rw_unlock(&zones->lock);
}

/** enter implicit transparent zone for local-data: without local-zone: */
static int
lz_setup_implicit(struct local_zones* zones, struct config_file* cfg)
{
        /* walk over all items that have no parent zone and find
         * the name that covers them all (could be the root) and
         * add that as a transparent zone */
        struct config_strlist* p;
        int have_name = 0;
        int have_other_classes = 0;
        uint16_t dclass = 0;
        uint8_t* nm = 0;
        size_t nmlen = 0;
        int nmlabs = 0;
        int match = 0; /* number of labels match count */

        lz_init_parents(zones); /* to enable local_zones_lookup() */
        for(p = cfg->local_data; p; p = p->next) {
                uint8_t* rr_name;
                uint16_t rr_class, rr_type;
                size_t len;
                int labs;
                if(!get_rr_nameclass(p->str, &rr_name, &rr_class, &rr_type)) {
                        log_err("Bad local-data RR %s", p->str);
                        return 0;
                }
                labs = dname_count_size_labels(rr_name, &len);
                lock_rw_rdlock(&zones->lock);
                if(!local_zones_lookup(zones, rr_name, len, labs, rr_class,
                        rr_type)) {
                        /* Check if there is a zone that this could go
                         * under but for different class; created zones are
                         * always for LDNS_RR_CLASS_IN. Create the zone with
                         * a different class but the same configured
                         * local_zone_type. */
                        struct local_zone* z = local_zones_lookup(zones,
                                rr_name, len, labs, LDNS_RR_CLASS_IN, rr_type);
                        if(z) {
                                uint8_t* name = memdup(z->name, z->namelen);
                                size_t znamelen = z->namelen;
                                int znamelabs = z->namelabs;
                                enum localzone_type ztype = z->type;
                                lock_rw_unlock(&zones->lock);
                                if(!name) {
                                        log_err("out of memory");
                                        free(rr_name);
                                        return 0;
                                }
                                if(!(
#ifndef THREADS_DISABLED
                                        z =
#endif
                                        lz_enter_zone_dname(zones, name,
                                                znamelen, znamelabs,
                                                ztype, rr_class))) {
                                        free(rr_name);
                                        return 0;
                                }
                                lock_rw_unlock(&z->lock);
                                free(rr_name);
                                continue;
                        }
                        if(!have_name) {
                                dclass = rr_class;
                                nm = rr_name;
                                nmlen = len;
                                nmlabs = labs;
                                match = labs;
                                have_name = 1;
                        } else {
                                int m;
                                if(rr_class != dclass) {
                                        /* process other classes later */
                                        free(rr_name);
                                        have_other_classes = 1;
                                        lock_rw_unlock(&zones->lock);
                                        continue;
                                }
                                /* find smallest shared topdomain */
                                (void)dname_lab_cmp(nm, nmlabs, 
                                        rr_name, labs, &m);
                                free(rr_name);
                                if(m < match)
                                        match = m;
                        }
                } else free(rr_name);
                lock_rw_unlock(&zones->lock);
        }
        if(have_name) {
                uint8_t* n2;
#ifndef THREADS_DISABLED
                struct local_zone* z;
#endif
                /* allocate zone of smallest shared topdomain to contain em */
                n2 = nm;
                dname_remove_labels(&n2, &nmlen, nmlabs - match);
                n2 = memdup(n2, nmlen);
                free(nm);
                if(!n2) {
                        log_err("out of memory");
                        return 0;
                }
                log_nametypeclass(VERB_ALGO, "implicit transparent local-zone", 
                        n2, 0, dclass);
                if(!(
#ifndef THREADS_DISABLED
                        z=
#endif
                        lz_enter_zone_dname(zones, n2, nmlen, match,
                        local_zone_transparent, dclass))) {
                        return 0;
                }
                lock_rw_unlock(&z->lock);
        }
        if(have_other_classes) { 
                /* restart to setup other class */
                return lz_setup_implicit(zones, cfg);
        }
        return 1;
}

/** enter local-zone-tag info */
static int
lz_enter_zone_tags(struct local_zones* zones, struct config_file* cfg)
{
        struct config_strbytelist* p;
        int c = 0;
        for(p = cfg->local_zone_tags; p; p = p->next) {
                if(!lz_enter_zone_tag(zones, p->str, p->str2, p->str2len,
                        LDNS_RR_CLASS_IN))
                        return 0;
                c++;
        }
        if(c) verbose(VERB_ALGO, "applied tags to %d local zones", c);
        return 1;
}
        
/** enter auth data */
static int
lz_enter_data(struct local_zones* zones, struct config_file* cfg)
{
        struct config_strlist* p;
        for(p = cfg->local_data; p; p = p->next) {
                if(!lz_enter_rr_str(zones, p->str))
                        return 0;
        }
        return 1;
}

/** free memory from config */
static void
lz_freeup_cfg(struct config_file* cfg)
{
        config_deldblstrlist(cfg->local_zones);
        cfg->local_zones = NULL;
        config_delstrlist(cfg->local_zones_nodefault);
        cfg->local_zones_nodefault = NULL;
        config_delstrlist(cfg->local_data);
        cfg->local_data = NULL;
}

int 
local_zones_apply_cfg(struct local_zones* zones, struct config_file* cfg)
{
        /* create zones from zone statements. */
        if(!lz_enter_zones(zones, cfg)) {
                return 0;
        }
        /* apply default zones+content (unless disabled, or overridden) */
        if(!local_zone_enter_defaults(zones, cfg)) {
                return 0;
        }
        /* enter local zone overrides */
        if(!lz_enter_overrides(zones, cfg)) {
                return 0;
        }
        /* create implicit transparent zone from data. */
        if(!lz_setup_implicit(zones, cfg)) {
                return 0;
        }

        /* setup parent ptrs for lookup during data entry */
        lz_init_parents(zones);
        /* insert local zone tags */
        if(!lz_enter_zone_tags(zones, cfg)) {
                return 0;
        }
        /* insert local data */
        if(!lz_enter_data(zones, cfg)) {
                return 0;
        }
        /* freeup memory from cfg struct. */
        lz_freeup_cfg(cfg);
        return 1;
}

struct local_zone* 
local_zones_lookup(struct local_zones* zones,
        uint8_t* name, size_t len, int labs, uint16_t dclass, uint16_t dtype)
{
        return local_zones_tags_lookup(zones, name, len, labs,
                dclass, dtype, NULL, 0, 1);
}

struct local_zone* 
local_zones_tags_lookup(struct local_zones* zones,
        uint8_t* name, size_t len, int labs, uint16_t dclass, uint16_t dtype,
        uint8_t* taglist, size_t taglen, int ignoretags)
{
        rbnode_type* res = NULL;
        struct local_zone *result;
        struct local_zone key;
        int m;
        /* for type DS use a zone higher when on a zonecut */
        if(dtype == LDNS_RR_TYPE_DS && !dname_is_root(name)) {
                dname_remove_label(&name, &len);
                labs--;
        }
        key.node.key = &key;
        key.dclass = dclass;
        key.name = name;
        key.namelen = len;
        key.namelabs = labs;
        rbtree_find_less_equal(&zones->ztree, &key, &res);
        result = (struct local_zone*)res;
        /* exact or smaller element (or no element) */
        if(!result || result->dclass != dclass)
                return NULL;
        /* count number of labels matched */
        (void)dname_lab_cmp(result->name, result->namelabs, key.name,
                key.namelabs, &m);
        while(result) { /* go up until qname is zone or subdomain of zone */
                if(result->namelabs <= m)
                        if(ignoretags || !result->taglist ||
                                taglist_intersect(result->taglist, 
                                result->taglen, taglist, taglen))
                                break;
                result = result->parent;
        }
        return result;
}

struct local_zone* 
local_zones_find(struct local_zones* zones,
        uint8_t* name, size_t len, int labs, uint16_t dclass)
{
        struct local_zone key;
        key.node.key = &key;
        key.dclass = dclass;
        key.name = name;
        key.namelen = len;
        key.namelabs = labs;
        /* exact */
        return (struct local_zone*)rbtree_search(&zones->ztree, &key);
}

struct local_zone*
local_zones_find_le(struct local_zones* zones,
        uint8_t* name, size_t len, int labs, uint16_t dclass,
        int* exact)
{
        struct local_zone key;
        rbnode_type *node;
        key.node.key = &key;
        key.dclass = dclass;
        key.name = name;
        key.namelen = len;
        key.namelabs = labs;
        *exact = rbtree_find_less_equal(&zones->ztree, &key, &node);
        return (struct local_zone*)node;
}

/** encode answer consisting of 1 rrset */
static int
local_encode(struct query_info* qinfo, struct module_env* env,
        struct edns_data* edns, struct comm_reply* repinfo, sldns_buffer* buf,
        struct regional* temp, struct ub_packed_rrset_key* rrset, int ansec,
        int rcode)
{
        struct reply_info rep;
        uint16_t udpsize;
        /* make answer with time=0 for fixed TTL values */
        memset(&rep, 0, sizeof(rep));
        rep.flags = (uint16_t)((BIT_QR | BIT_AA | BIT_RA) | rcode);
        rep.qdcount = 1;
        if(ansec)
                rep.an_numrrsets = 1;
        else    rep.ns_numrrsets = 1;
        rep.rrset_count = 1;
        rep.rrsets = &rrset;
        rep.reason_bogus = LDNS_EDE_NONE;
        udpsize = edns->udp_size;
        edns->edns_version = EDNS_ADVERTISED_VERSION;
        edns->udp_size = EDNS_ADVERTISED_SIZE;
        edns->ext_rcode = 0;
        edns->bits &= EDNS_DO;
        if(!inplace_cb_reply_local_call(env, qinfo, NULL, &rep, rcode, edns,
                repinfo, temp, env->now_tv) || !reply_info_answer_encode(qinfo, &rep,
                *(uint16_t*)sldns_buffer_begin(buf), sldns_buffer_read_u16_at(buf, 2),
                buf, 0, 0, temp, udpsize, edns, (int)(edns->bits&EDNS_DO), 0)) {
                error_encode(buf, (LDNS_RCODE_SERVFAIL|BIT_AA), qinfo,
                        *(uint16_t*)sldns_buffer_begin(buf),
                        sldns_buffer_read_u16_at(buf, 2), edns);
        }
        return 1;
}

/** encode local error answer */
static void
local_error_encode(struct query_info* qinfo, struct module_env* env,
        struct edns_data* edns, struct comm_reply* repinfo, sldns_buffer* buf,
        struct regional* temp, int rcode, int r, int ede_code,
        const char* ede_txt)
{
        edns->edns_version = EDNS_ADVERTISED_VERSION;
        edns->udp_size = EDNS_ADVERTISED_SIZE;
        edns->ext_rcode = 0;
        edns->bits &= EDNS_DO;

        if(!inplace_cb_reply_local_call(env, qinfo, NULL, NULL,
                rcode, edns, repinfo, temp, env->now_tv))
                edns->opt_list_inplace_cb_out = NULL;

        if(ede_code != LDNS_EDE_NONE && env->cfg->ede) {
                edns_opt_list_append_ede(&edns->opt_list_out, temp,
                        ede_code, ede_txt);
        }

        error_encode(buf, r, qinfo, *(uint16_t*)sldns_buffer_begin(buf),
                sldns_buffer_read_u16_at(buf, 2), edns);
}

/** find local data tag string match for the given type in the list */
int
local_data_find_tag_datas(const struct query_info* qinfo,
        struct config_strlist* list, struct ub_packed_rrset_key* r,
        struct regional* temp)
{
        struct config_strlist* p;
        char buf[65536];
        uint8_t rr[LDNS_RR_BUF_SIZE];
        size_t len;
        int res;
        struct packed_rrset_data* d;
        for(p=list; p; p=p->next) {
                uint16_t rdr_type;

                len = sizeof(rr);
                /* does this element match the type? */
                snprintf(buf, sizeof(buf), ". %s", p->str);
                res = sldns_str2wire_rr_buf(buf, rr, &len, NULL, 3600,
                        NULL, 0, NULL, 0);
                if(res != 0)
                        /* parse errors are already checked before, in
                         * acllist check_data, skip this for robustness */
                        continue;
                if(len < 1 /* . */ + 8 /* typeclassttl*/ + 2 /*rdatalen*/)
                        continue;
                rdr_type = sldns_wirerr_get_type(rr, len, 1);
                if(rdr_type != qinfo->qtype && rdr_type != LDNS_RR_TYPE_CNAME)
                        continue;
                
                /* do we have entries already? if not setup key */
                if(r->rk.dname == NULL) {
                        r->entry.key = r;
                        r->rk.dname = qinfo->qname;
                        r->rk.dname_len = qinfo->qname_len;
                        r->rk.type = htons(rdr_type);
                        r->rk.rrset_class = htons(qinfo->qclass);
                        r->rk.flags = 0;
                        d = (struct packed_rrset_data*)regional_alloc_zero(
                                temp, sizeof(struct packed_rrset_data)
                                + sizeof(size_t) + sizeof(uint8_t*) +
                                sizeof(time_t));
                        if(!d) return 0; /* out of memory */
                        r->entry.data = d;
                        d->ttl = sldns_wirerr_get_ttl(rr, len, 1);
                        d->rr_len = (size_t*)((uint8_t*)d +
                                sizeof(struct packed_rrset_data));
                        d->rr_data = (uint8_t**)&(d->rr_len[1]);
                        d->rr_ttl = (time_t*)&(d->rr_data[1]);
                }
                d = (struct packed_rrset_data*)r->entry.data;
                /* add entry to the data */
                if(d->count != 0) {
                        size_t* oldlen = d->rr_len;
                        uint8_t** olddata = d->rr_data;
                        time_t* oldttl = d->rr_ttl;
                        /* increase arrays for lookup */
                        /* this is of course slow for very many records,
                         * but most redirects are expected with few records */
                        d->rr_len = (size_t*)regional_alloc_zero(temp,
                                (d->count+1)*sizeof(size_t));
                        d->rr_data = (uint8_t**)regional_alloc_zero(temp,
                                (d->count+1)*sizeof(uint8_t*));
                        d->rr_ttl = (time_t*)regional_alloc_zero(temp,
                                (d->count+1)*sizeof(time_t));
                        if(!d->rr_len || !d->rr_data || !d->rr_ttl)
                                return 0; /* out of memory */
                        /* first one was allocated after struct d, but new
                         * ones get their own array increment alloc, so
                         * copy old content */
                        memmove(d->rr_len, oldlen, d->count*sizeof(size_t));
                        memmove(d->rr_data, olddata, d->count*sizeof(uint8_t*));
                        memmove(d->rr_ttl, oldttl, d->count*sizeof(time_t));
                }

                d->rr_len[d->count] = sldns_wirerr_get_rdatalen(rr, len, 1)+2;
                d->rr_ttl[d->count] = sldns_wirerr_get_ttl(rr, len, 1);
                d->rr_data[d->count] = regional_alloc_init(temp,
                        sldns_wirerr_get_rdatawl(rr, len, 1),
                        d->rr_len[d->count]);
                if(!d->rr_data[d->count])
                        return 0; /* out of memory */
                d->count++;
        }
        if(r->rk.dname)
                return 1;
        return 0;
}

static int
find_tag_datas(struct query_info* qinfo, struct config_strlist* list,
        struct ub_packed_rrset_key* r, struct regional* temp)
{
        int result = local_data_find_tag_datas(qinfo, list, r, temp);

        /* If we've found a non-exact alias type of local data, make a shallow
         * copy of the RRset and remember it in qinfo to complete the alias
         * chain later. */
        if(result && qinfo->qtype != LDNS_RR_TYPE_CNAME &&
                r->rk.type == htons(LDNS_RR_TYPE_CNAME)) {
                qinfo->local_alias =
                        regional_alloc_zero(temp, sizeof(struct local_rrset));
                if(!qinfo->local_alias)
                        return 0; /* out of memory */
                qinfo->local_alias->rrset =
                        regional_alloc_init(temp, r, sizeof(*r));
                if(!qinfo->local_alias->rrset)
                        return 0; /* out of memory */
        }
        return result;
}

int
local_data_answer(struct local_zone* z, struct module_env* env,
        struct query_info* qinfo, struct edns_data* edns,
        struct comm_reply* repinfo, sldns_buffer* buf,
        struct regional* temp, int labs, struct local_data** ldp,
        enum localzone_type lz_type, int tag, struct config_strlist** tag_datas,
        size_t tag_datas_size, char** tagname, int num_tags)
{
        struct local_data key;
        struct local_data* ld;
        struct local_rrset* lr;
        key.node.key = &key;
        key.name = qinfo->qname;
        key.namelen = qinfo->qname_len;
        key.namelabs = labs;
        if(lz_type == local_zone_redirect ||
                lz_type == local_zone_inform_redirect) {
                key.name = z->name;
                key.namelen = z->namelen;
                key.namelabs = z->namelabs;
                if(tag != -1 && (size_t)tag<tag_datas_size && tag_datas[tag]) {
                        struct ub_packed_rrset_key r;
                        memset(&r, 0, sizeof(r));
                        if(find_tag_datas(qinfo, tag_datas[tag], &r, temp)) {
                                verbose(VERB_ALGO, "redirect with tag data [%d] %s",
                                        tag, (tag<num_tags?tagname[tag]:"null"));

                                /* If we found a matching alias, we should
                                 * use it as part of the answer, but we can't
                                 * encode it until we complete the alias
                                 * chain. */
                                if(qinfo->local_alias)
                                        return 1;
                                return local_encode(qinfo, env, edns, repinfo, buf, temp,
                                        &r, 1, LDNS_RCODE_NOERROR);
                        }
                }
        }
        ld = (struct local_data*)rbtree_search(&z->data, &key.node);
        *ldp = ld;
        if(!ld) {
                return 0;
        }
        lr = local_data_find_type(ld, qinfo->qtype, 1);
        if(!lr)
                return 0;

        /* Special case for alias matching.  See local_data_answer(). */
        if((lz_type == local_zone_redirect ||
                lz_type == local_zone_inform_redirect) &&
                qinfo->qtype != LDNS_RR_TYPE_CNAME &&
                lr->rrset->rk.type == htons(LDNS_RR_TYPE_CNAME)) {
                uint8_t* ctarget;
                size_t ctargetlen = 0;

                qinfo->local_alias =
                        regional_alloc_zero(temp, sizeof(struct local_rrset));
                if(!qinfo->local_alias)
                        return 0; /* out of memory */
                qinfo->local_alias->rrset = regional_alloc_init(
                        temp, lr->rrset, sizeof(*lr->rrset));
                if(!qinfo->local_alias->rrset)
                        return 0; /* out of memory */
                qinfo->local_alias->rrset->rk.dname = qinfo->qname;
                qinfo->local_alias->rrset->rk.dname_len = qinfo->qname_len;
                get_cname_target(lr->rrset, &ctarget, &ctargetlen);
                if(!ctargetlen)
                        return 0; /* invalid cname */
                if(dname_is_wild(ctarget)) {
                        /* synthesize cname target */
                        struct packed_rrset_data* d, *lr_d;
                        /* -3 for wildcard label and root label from qname */
                        size_t newtargetlen = qinfo->qname_len + ctargetlen - 3;

                        log_assert(ctargetlen >= 3);
                        log_assert(qinfo->qname_len >= 1);

                        if(newtargetlen > LDNS_MAX_DOMAINLEN) {
                                qinfo->local_alias = NULL;
                                local_error_encode(qinfo, env, edns, repinfo,
                                        buf, temp, LDNS_RCODE_YXDOMAIN,
                                        (LDNS_RCODE_YXDOMAIN|BIT_AA),
                                        LDNS_EDE_OTHER,
                                        "DNAME expansion became too large");
                                return 1;
                        }
                        memset(&qinfo->local_alias->rrset->entry, 0,
                                sizeof(qinfo->local_alias->rrset->entry));
                        qinfo->local_alias->rrset->entry.key =
                                qinfo->local_alias->rrset;
                        qinfo->local_alias->rrset->entry.hash =
                                rrset_key_hash(&qinfo->local_alias->rrset->rk);
                        d = (struct packed_rrset_data*)regional_alloc_zero(temp,
                                sizeof(struct packed_rrset_data) + sizeof(size_t) +
                                sizeof(uint8_t*) + sizeof(time_t) + sizeof(uint16_t)
                                + newtargetlen);
                        if(!d)
                                return 0; /* out of memory */
                        lr_d = (struct packed_rrset_data*)lr->rrset->entry.data;
                        qinfo->local_alias->rrset->entry.data = d;
                        d->ttl = lr_d->rr_ttl[0]; /* RFC6672-like behavior:
                                            synth CNAME TTL uses original TTL*/
                        d->count = 1;
                        d->rrsig_count = 0;
                        d->trust = rrset_trust_ans_noAA;
                        d->rr_len = (size_t*)((uint8_t*)d +
                                sizeof(struct packed_rrset_data));
                        d->rr_len[0] = newtargetlen + sizeof(uint16_t);
                        packed_rrset_ptr_fixup(d);
                        d->rr_ttl[0] = d->ttl;
                        sldns_write_uint16(d->rr_data[0], newtargetlen);
                        /* write qname */
                        memmove(d->rr_data[0] + sizeof(uint16_t), qinfo->qname,
                                qinfo->qname_len - 1);
                        /* write cname target wildcard label */
                        memmove(d->rr_data[0] + sizeof(uint16_t) +
                                qinfo->qname_len - 1, ctarget + 2,
                                ctargetlen - 2);
                }
                return 1;
        }
        if(lz_type == local_zone_redirect ||
                lz_type == local_zone_inform_redirect) {
                /* convert rrset name to query name; like a wildcard */
                struct ub_packed_rrset_key r = *lr->rrset;
                r.rk.dname = qinfo->qname;
                r.rk.dname_len = qinfo->qname_len;
                return local_encode(qinfo, env, edns, repinfo, buf, temp, &r, 1,
                        LDNS_RCODE_NOERROR);
        }
        return local_encode(qinfo, env, edns, repinfo, buf, temp, lr->rrset, 1,
                LDNS_RCODE_NOERROR);
}

/**
 * See if the local zone does not cover the name, eg. the name is not
 * in the zone and the zone is transparent */
static int
local_zone_does_not_cover(struct local_zone* z, struct query_info* qinfo,
        int labs)
{
        struct local_data key;
        struct local_data* ld = NULL;
        struct local_rrset* lr = NULL;
        if(z->type == local_zone_always_transparent || z->type == local_zone_block_a)
                return 1;
        if(z->type != local_zone_transparent
                && z->type != local_zone_typetransparent
                && z->type != local_zone_inform)
                return 0;
        key.node.key = &key;
        key.name = qinfo->qname;
        key.namelen = qinfo->qname_len;
        key.namelabs = labs;
        ld = (struct local_data*)rbtree_search(&z->data, &key.node);
        if(z->type == local_zone_transparent || z->type == local_zone_inform)
                return (ld == NULL);
        if(ld)
                lr = local_data_find_type(ld, qinfo->qtype, 1);
        /* local_zone_typetransparent */
        return (lr == NULL);
}

static inline int
local_zone_is_udp_query(struct comm_reply* repinfo) {
        return repinfo != NULL
                        ? (repinfo->c != NULL
                                ? repinfo->c->type == comm_udp
                                : 0)
                        : 0;
}

int
local_zones_zone_answer(struct local_zone* z, struct module_env* env,
        struct query_info* qinfo, struct edns_data* edns,
        struct comm_reply* repinfo, sldns_buffer* buf, struct regional* temp,
        struct local_data* ld, enum localzone_type lz_type)
{
        if(lz_type == local_zone_deny ||
                lz_type == local_zone_always_deny ||
                lz_type == local_zone_inform_deny) {
                /** no reply at all, signal caller by clearing buffer. */
                sldns_buffer_clear(buf);
                sldns_buffer_flip(buf);
                return 1;
        } else if(lz_type == local_zone_refuse
                || lz_type == local_zone_always_refuse) {
                local_error_encode(qinfo, env, edns, repinfo, buf, temp,
                        LDNS_RCODE_REFUSED, (LDNS_RCODE_REFUSED|BIT_AA),
                        LDNS_EDE_NONE, NULL);
                return 1;
        } else if(lz_type == local_zone_static ||
                lz_type == local_zone_redirect ||
                lz_type == local_zone_inform_redirect ||
                lz_type == local_zone_always_nxdomain ||
                lz_type == local_zone_always_nodata ||
                (lz_type == local_zone_truncate
                        && local_zone_is_udp_query(repinfo))) {
                /* for static, reply nodata or nxdomain
                 * for redirect, reply nodata */
                /* no additional section processing,
                 * cname, dname or wildcard processing,
                 * or using closest match for NSEC.
                 * or using closest match for returning delegation downwards
                 */
                int rcode = (ld || lz_type == local_zone_redirect ||
                        lz_type == local_zone_inform_redirect ||
                        lz_type == local_zone_always_nodata ||
                        lz_type == local_zone_truncate)?
                        LDNS_RCODE_NOERROR:LDNS_RCODE_NXDOMAIN;
                rcode = (lz_type == local_zone_truncate ? (rcode|BIT_TC) : rcode);
                if(z != NULL && z->soa && z->soa_negative)
                        return local_encode(qinfo, env, edns, repinfo, buf, temp,
                                z->soa_negative, 0, rcode);
                local_error_encode(qinfo, env, edns, repinfo, buf, temp,
                        rcode, (rcode|BIT_AA), LDNS_EDE_NONE, NULL);
                return 1;
        } else if(lz_type == local_zone_typetransparent
                || lz_type == local_zone_always_transparent) {
                /* no NODATA or NXDOMAINS for this zone type */
                return 0;
        } else if(lz_type == local_zone_block_a) {
                /* Return NODATA for all A queries */
                if(qinfo->qtype == LDNS_RR_TYPE_A) {
                        local_error_encode(qinfo, env, edns, repinfo, buf, temp,
                                LDNS_RCODE_NOERROR, (LDNS_RCODE_NOERROR|BIT_AA),
                                LDNS_EDE_NONE, NULL);
                                return 1;
                }

                return 0;
        } else if(lz_type == local_zone_always_null) {
                /* 0.0.0.0 or ::0 or noerror/nodata for this zone type,
                 * used for blocklists. */
                if(qinfo->qtype == LDNS_RR_TYPE_A ||
                        qinfo->qtype == LDNS_RR_TYPE_AAAA) {
                        struct ub_packed_rrset_key lrr;
                        struct packed_rrset_data d;
                        time_t rr_ttl = 3600;
                        size_t rr_len = 0;
                        uint8_t rr_data[2+16] = {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
                        uint8_t* rr_datas = rr_data;
                        memset(&lrr, 0, sizeof(lrr));
                        memset(&d, 0, sizeof(d));
                        lrr.entry.data = &d;
                        lrr.rk.dname = qinfo->qname;
                        lrr.rk.dname_len = qinfo->qname_len;
                        lrr.rk.type = htons(qinfo->qtype);
                        lrr.rk.rrset_class = htons(qinfo->qclass);
                        if(qinfo->qtype == LDNS_RR_TYPE_A) {
                                rr_len = 4;
                                sldns_write_uint16(rr_data, rr_len);
                                rr_len += 2;
                        } else {
                                rr_len = 16;
                                sldns_write_uint16(rr_data, rr_len);
                                rr_len += 2;
                        }
                        d.ttl = rr_ttl;
                        d.count = 1;
                        d.rr_len = &rr_len;
                        d.rr_data = &rr_datas;
                        d.rr_ttl = &rr_ttl;
                        return local_encode(qinfo, env, edns, repinfo, buf, temp,
                                &lrr, 1, LDNS_RCODE_NOERROR);
                } else {
                        /* NODATA: No EDE needed */
                        local_error_encode(qinfo, env, edns, repinfo, buf,
                                temp, LDNS_RCODE_NOERROR,
                                (LDNS_RCODE_NOERROR|BIT_AA), -1, NULL);
                }
                return 1;
        }
        /* else lz_type == local_zone_transparent */

        /* if the zone is transparent and the name exists, but the type
         * does not, then we should make this noerror/nodata */
        if(ld && ld->rrsets) {
                int rcode = LDNS_RCODE_NOERROR;
                if(z != NULL && z->soa && z->soa_negative)
                        return local_encode(qinfo, env, edns, repinfo, buf, temp,
                                z->soa_negative, 0, rcode);
                /* NODATA: No EDE needed */
                local_error_encode(qinfo, env, edns, repinfo, buf, temp, rcode,
                        (rcode|BIT_AA), LDNS_EDE_NONE, NULL);
                return 1;
        }

        /* stop here, and resolve further on */
        return 0;
}

/** print log information for an inform zone query */
static void
lz_inform_print(struct local_zone* z, struct query_info* qinfo,
        struct sockaddr_storage* addr, socklen_t addrlen)
{
        char ip[128], txt[512];
        char zname[LDNS_MAX_DOMAINLEN];
        uint16_t port = ntohs(((struct sockaddr_in*)addr)->sin_port);
        dname_str(z->name, zname);
        addr_to_str(addr, addrlen, ip, sizeof(ip));
        snprintf(txt, sizeof(txt), "%s %s %s@%u", zname, local_zone_type2str(z->type), ip,
                (unsigned)port);
        log_nametypeclass(NO_VERBOSE, txt, qinfo->qname, qinfo->qtype, qinfo->qclass);
}

static enum localzone_type
lz_type(uint8_t *taglist, size_t taglen, uint8_t *taglist2, size_t taglen2,
        uint8_t *tagactions, size_t tagactionssize, enum localzone_type lzt,
        struct comm_reply* repinfo, struct rbtree_type* override_tree,
        int* tag, char** tagname, int num_tags)
{
        struct local_zone_override* lzo;        
        if(repinfo && override_tree) {
                lzo = (struct local_zone_override*)addr_tree_lookup(
                        override_tree, &repinfo->client_addr,
                        repinfo->client_addrlen);
                if(lzo && lzo->type) {
                        verbose(VERB_ALGO, "local zone override to type %s",
                                local_zone_type2str(lzo->type));
                        return lzo->type;
                }
        }
        if(!taglist || !taglist2)
                return lzt;
        return local_data_find_tag_action(taglist, taglen, taglist2, taglen2,
                tagactions, tagactionssize, lzt, tag, tagname, num_tags);
}

enum localzone_type
local_data_find_tag_action(const uint8_t* taglist, size_t taglen,
        const uint8_t* taglist2, size_t taglen2, const uint8_t* tagactions,
        size_t tagactionssize, enum localzone_type lzt, int* tag,
        char* const* tagname, int num_tags)
{
        size_t i, j;
        uint8_t tagmatch;

        for(i=0; i<taglen && i<taglen2; i++) {
                tagmatch = (taglist[i] & taglist2[i]);
                for(j=0; j<8 && tagmatch>0; j++) {
                        if((tagmatch & 0x1)) {
                                *tag = (int)(i*8+j);
                                verbose(VERB_ALGO, "matched tag [%d] %s",
                                        *tag, (*tag<num_tags?tagname[*tag]:"null"));
                                /* does this tag have a tag action? */
                                if(i*8+j < tagactionssize && tagactions
                                   && tagactions[i*8+j] != 0) {
                                  verbose(VERB_ALGO, "tag action [%d] %s to type %s",
                                        *tag, (*tag<num_tags?tagname[*tag]:"null"),
                                        local_zone_type2str(
                                        (enum localzone_type)
                                        tagactions[i*8+j]));
                                  return (enum localzone_type)tagactions[i*8+j];
                                }
                                return lzt;
                        }
                        tagmatch >>= 1; 
                }
        }
        return lzt;
}

int 
local_zones_answer(struct local_zones* zones, struct module_env* env,
        struct query_info* qinfo, struct edns_data* edns, sldns_buffer* buf,
        struct regional* temp, struct comm_reply* repinfo, uint8_t* taglist,
        size_t taglen, uint8_t* tagactions, size_t tagactionssize,
        struct config_strlist** tag_datas, size_t tag_datas_size,
        char** tagname, int num_tags, struct view* view)
{
        /* see if query is covered by a zone,
         *      if so:  - try to match (exact) local data 
         *              - look at zone type for negative response. */
        int labs = dname_count_labels(qinfo->qname);
        struct local_data* ld = NULL;
        struct local_zone* z = NULL;
        enum localzone_type lzt = local_zone_transparent;
        int r, tag = -1;

        if(view) {
                lock_rw_rdlock(&view->lock);
                if(view->local_zones &&
                        (z = local_zones_lookup(view->local_zones,
                        qinfo->qname, qinfo->qname_len, labs,
                        qinfo->qclass, qinfo->qtype))) {
                        lock_rw_rdlock(&z->lock);
                        lzt = z->type;
                }
                if(lzt == local_zone_noview) {
                        lock_rw_unlock(&z->lock);
                        z = NULL;
                }
                if(z && (lzt == local_zone_transparent ||
                        lzt == local_zone_typetransparent ||
                        lzt == local_zone_inform ||
                        lzt == local_zone_always_transparent ||
                        lzt == local_zone_block_a) &&
                        local_zone_does_not_cover(z, qinfo, labs)) {
                        lock_rw_unlock(&z->lock);
                        z = NULL;
                }
                if(view->local_zones && !z && !view->isfirst){
                        lock_rw_unlock(&view->lock);
                        return 0;
                }
                if(z && verbosity >= VERB_ALGO) {
                        char zname[LDNS_MAX_DOMAINLEN];
                        dname_str(z->name, zname);
                        verbose(VERB_ALGO, "using localzone %s %s from view %s", 
                                zname, local_zone_type2str(lzt), view->name);
                }
                lock_rw_unlock(&view->lock);
        }
        if(!z) {
                /* try global local_zones tree */
                lock_rw_rdlock(&zones->lock);
                if(!(z = local_zones_tags_lookup(zones, qinfo->qname,
                        qinfo->qname_len, labs, qinfo->qclass, qinfo->qtype,
                        taglist, taglen, 0))) {
                        lock_rw_unlock(&zones->lock);
                        return 0;
                }
                lock_rw_rdlock(&z->lock);
                lzt = lz_type(taglist, taglen, z->taglist, z->taglen,
                        tagactions, tagactionssize, z->type, repinfo,
                        z->override_tree, &tag, tagname, num_tags);
                lock_rw_unlock(&zones->lock);
                if(z && verbosity >= VERB_ALGO) {
                        char zname[LDNS_MAX_DOMAINLEN];
                        dname_str(z->name, zname);
                        verbose(VERB_ALGO, "using localzone %s %s", zname,
                                local_zone_type2str(lzt));
                }
        }
        if((env->cfg->log_local_actions ||
                        lzt == local_zone_inform ||
                        lzt == local_zone_inform_deny ||
                        lzt == local_zone_inform_redirect)
                        && repinfo)
                lz_inform_print(z, qinfo, &repinfo->client_addr,
                        repinfo->client_addrlen);

        if(lzt != local_zone_always_refuse
                && lzt != local_zone_always_transparent
                && lzt != local_zone_block_a
                && lzt != local_zone_always_nxdomain
                && lzt != local_zone_always_nodata
                && lzt != local_zone_always_deny
                && local_data_answer(z, env, qinfo, edns, repinfo, buf, temp, labs,
                        &ld, lzt, tag, tag_datas, tag_datas_size, tagname, num_tags)) {
                lock_rw_unlock(&z->lock);
                /* We should tell the caller that encode is deferred if we found
                 * a local alias. */
                return !qinfo->local_alias;
        }
        r = local_zones_zone_answer(z, env, qinfo, edns, repinfo, buf, temp, ld, lzt);
        lock_rw_unlock(&z->lock);
        return r && !qinfo->local_alias; /* see above */
}

const char* local_zone_type2str(enum localzone_type t)
{
        switch(t) {
                case local_zone_unset: return "unset";
                case local_zone_deny: return "deny";
                case local_zone_refuse: return "refuse";
                case local_zone_redirect: return "redirect";
                case local_zone_transparent: return "transparent";
                case local_zone_typetransparent: return "typetransparent";
                case local_zone_static: return "static";
                case local_zone_nodefault: return "nodefault";
                case local_zone_inform: return "inform";
                case local_zone_inform_deny: return "inform_deny";
                case local_zone_inform_redirect: return "inform_redirect";
                case local_zone_always_transparent: return "always_transparent";
                case local_zone_block_a: return "block_a";
                case local_zone_always_refuse: return "always_refuse";
                case local_zone_always_nxdomain: return "always_nxdomain";
                case local_zone_always_nodata: return "always_nodata";
                case local_zone_always_deny: return "always_deny";
                case local_zone_always_null: return "always_null";
                case local_zone_noview: return "noview";
                case local_zone_truncate: return "truncate";
                case local_zone_invalid: return "invalid";
        }
        return "badtyped"; 
}

int local_zone_str2type(const char* type, enum localzone_type* t)
{
        if(strcmp(type, "deny") == 0)
                *t = local_zone_deny;
        else if(strcmp(type, "refuse") == 0)
                *t = local_zone_refuse;
        else if(strcmp(type, "static") == 0)
                *t = local_zone_static;
        else if(strcmp(type, "transparent") == 0)
                *t = local_zone_transparent;
        else if(strcmp(type, "typetransparent") == 0)
                *t = local_zone_typetransparent;
        else if(strcmp(type, "redirect") == 0)
                *t = local_zone_redirect;
        else if(strcmp(type, "inform") == 0)
                *t = local_zone_inform;
        else if(strcmp(type, "inform_deny") == 0)
                *t = local_zone_inform_deny;
        else if(strcmp(type, "inform_redirect") == 0)
                *t = local_zone_inform_redirect;
        else if(strcmp(type, "always_transparent") == 0)
                *t = local_zone_always_transparent;
        else if(strcmp(type, "block_a") == 0)
                *t = local_zone_block_a;
        else if(strcmp(type, "always_refuse") == 0)
                *t = local_zone_always_refuse;
        else if(strcmp(type, "always_nxdomain") == 0)
                *t = local_zone_always_nxdomain;
        else if(strcmp(type, "always_nodata") == 0)
                *t = local_zone_always_nodata;
        else if(strcmp(type, "always_deny") == 0)
                *t = local_zone_always_deny;
        else if(strcmp(type, "always_null") == 0)
                *t = local_zone_always_null;
        else if(strcmp(type, "noview") == 0)
                *t = local_zone_noview;
        else if(strcmp(type, "truncate") == 0)
                *t = local_zone_truncate;
        else if(strcmp(type, "nodefault") == 0)
                *t = local_zone_nodefault;
        else return 0;
        return 1;
}

/** iterate over the kiddies of the given name and set their parent ptr */
static void
set_kiddo_parents(struct local_zone* z, struct local_zone* match, 
        struct local_zone* newp)
{
        /* both zones and z are locked already */
        /* in the sorted rbtree, the kiddies of z are located after z */
        /* z must be present in the tree */
        struct local_zone* p = z;
        p = (struct local_zone*)rbtree_next(&p->node);
        while(p!=(struct local_zone*)RBTREE_NULL &&
                p->dclass == z->dclass && dname_strict_subdomain(p->name,
                p->namelabs, z->name, z->namelabs)) {
                /* update parent ptr */
                /* only when matches with existing parent pointer, so that
                 * deeper child structures are not touched, i.e.
                 * update of x, and a.x, b.x, f.b.x, g.b.x, c.x, y
                 * gets to update a.x, b.x and c.x */
                lock_rw_wrlock(&p->lock);
                if(p->parent == match)
                        p->parent = newp;
                lock_rw_unlock(&p->lock);
                p = (struct local_zone*)rbtree_next(&p->node);
        }
}

struct local_zone* local_zones_add_zone(struct local_zones* zones,
        uint8_t* name, size_t len, int labs, uint16_t dclass,
        enum localzone_type tp)
{
        int exact;
        /* create */
        struct local_zone *prev;
        struct local_zone* z = local_zone_create(name, len, labs, tp, dclass);
        if(!z) {
                free(name);
                return NULL;
        }
        lock_rw_wrlock(&z->lock);

        /* find the closest parent */
        prev = local_zones_find_le(zones, name, len, labs, dclass, &exact);
        if(!exact)
                z->parent = find_closest_parent(z, prev);

        /* insert into the tree */
        if(exact||!rbtree_insert(&zones->ztree, &z->node)) {
                /* duplicate entry! */
                lock_rw_unlock(&z->lock);
                local_zone_delete(z);
                log_err("internal: duplicate entry in local_zones_add_zone");
                return NULL;
        }

        /* set parent pointers right */
        set_kiddo_parents(z, z->parent, z);

        lock_rw_unlock(&z->lock);
        return z;
}

void local_zones_del_zone(struct local_zones* zones, struct local_zone* z)
{
        /* fix up parents in tree */
        lock_rw_wrlock(&z->lock);
        set_kiddo_parents(z, z, z->parent);

        /* remove from tree */
        (void)rbtree_delete(&zones->ztree, z);

        /* delete the zone */
        lock_rw_unlock(&z->lock);
        local_zone_delete(z);
}

int
local_zones_add_RR(struct local_zones* zones, const char* rr)
{
        uint8_t* rr_name;
        uint16_t rr_class, rr_type;
        size_t len;
        int labs;
        struct local_zone* z;
        int r;
        if(!get_rr_nameclass(rr, &rr_name, &rr_class, &rr_type)) {
                return 0;
        }
        labs = dname_count_size_labels(rr_name, &len);
        /* could first try readlock then get writelock if zone does not exist,
         * but we do not add enough RRs (from multiple threads) to optimize */
        lock_rw_wrlock(&zones->lock);
        z = local_zones_lookup(zones, rr_name, len, labs, rr_class, rr_type);
        if(!z) {
                z = local_zones_add_zone(zones, rr_name, len, labs, rr_class,
                        local_zone_transparent);
                if(!z) {
                        lock_rw_unlock(&zones->lock);
                        return 0;
                }
        } else {
                free(rr_name);
        }
        lock_rw_wrlock(&z->lock);
        lock_rw_unlock(&zones->lock);
        r = lz_enter_rr_into_zone(z, rr);
        lock_rw_unlock(&z->lock);
        return r;
}

/** returns true if the node is terminal so no deeper domain names exist */
static int
is_terminal(struct local_data* d)
{
        /* for empty nonterminals, the deeper domain names are sorted
         * right after them, so simply check the next name in the tree 
         */
        struct local_data* n = (struct local_data*)rbtree_next(&d->node);
        if(n == (struct local_data*)RBTREE_NULL)
                return 1; /* last in tree, no deeper node */
        if(dname_strict_subdomain(n->name, n->namelabs, d->name, d->namelabs))
                return 0; /* there is a deeper node */
        return 1;
}

/** delete empty terminals from tree when final data is deleted */
static void 
del_empty_term(struct local_zone* z, struct local_data* d, 
        uint8_t* name, size_t len, int labs)
{
        while(d && d->rrsets == NULL && is_terminal(d)) {
                /* is this empty nonterminal? delete */
                /* note, no memory recycling in zone region */
                (void)rbtree_delete(&z->data, d);

                /* go up and to the next label */
                if(dname_is_root(name))
                        return;
                dname_remove_label(&name, &len);
                labs--;
                d = local_zone_find_data(z, name, len, labs);
        }
}

/** find and remove type from list in domain struct */
static void
del_local_rrset(struct local_data* d, uint16_t dtype)
{
        struct local_rrset* prev=NULL, *p=d->rrsets;
        while(p && ntohs(p->rrset->rk.type) != dtype) {
                prev = p;
                p = p->next;
        }
        if(!p) 
                return; /* rrset type not found */
        /* unlink it */
        if(prev) prev->next = p->next;
        else d->rrsets = p->next;
        /* no memory recycling for zone deletions ... */
}

void local_zones_del_data(struct local_zones* zones, 
        uint8_t* name, size_t len, int labs, uint16_t dclass)
{
        /* find zone */
        struct local_zone* z;
        struct local_data* d;

        /* remove DS */
        lock_rw_rdlock(&zones->lock);
        z = local_zones_lookup(zones, name, len, labs, dclass, LDNS_RR_TYPE_DS);
        if(z) {
                lock_rw_wrlock(&z->lock);
                d = local_zone_find_data(z, name, len, labs);
                if(d) {
                        del_local_rrset(d, LDNS_RR_TYPE_DS);
                        del_empty_term(z, d, name, len, labs);
                }
                lock_rw_unlock(&z->lock);
        }
        lock_rw_unlock(&zones->lock);

        /* remove other types */
        lock_rw_rdlock(&zones->lock);
        z = local_zones_lookup(zones, name, len, labs, dclass, 0);
        if(!z) {
                /* no such zone, we're done */
                lock_rw_unlock(&zones->lock);
                return;
        }
        lock_rw_wrlock(&z->lock);
        lock_rw_unlock(&zones->lock);

        /* find the domain */
        d = local_zone_find_data(z, name, len, labs);
        if(d) {
                /* no memory recycling for zone deletions ... */
                d->rrsets = NULL;
                /* did we delete the soa record ? */
                if(query_dname_compare(d->name, z->name) == 0) {
                        z->soa = NULL;
                        z->soa_negative = NULL;
                }

                /* cleanup the empty nonterminals for this name */
                del_empty_term(z, d, name, len, labs);
        }

        lock_rw_unlock(&z->lock);
}

/** Get memory usage for local_zone */
static size_t
local_zone_get_mem(struct local_zone* z)
{
        size_t m = sizeof(*z);
        lock_rw_rdlock(&z->lock);
        m += z->namelen + z->taglen + regional_get_mem(z->region);
        lock_rw_unlock(&z->lock);
        return m;
}

size_t local_zones_get_mem(struct local_zones* zones)
{
        struct local_zone* z;
        size_t m;
        if(!zones) return 0;
        m = sizeof(*zones);
        lock_rw_rdlock(&zones->lock);
        RBTREE_FOR(z, struct local_zone*, &zones->ztree) {
                m += local_zone_get_mem(z);
        }
        lock_rw_unlock(&zones->lock);
        return m;
}

void local_zones_swap_tree(struct local_zones* zones, struct local_zones* data)
{
        rbtree_type oldtree = zones->ztree;
        zones->ztree = data->ztree;
        data->ztree = oldtree;
}