lib/dns/zt.c

	zt.c revision dafcb997e390efa4423883dafd100c975c4095d6
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington/*
0c27b3fe77ac1d5094ba3521e8142d9e7973133fMark Andrews * Copyright (C) 2004  Internet Systems Consortium, Inc. ("ISC")
ec5347e2c775f027573ce5648b910361aa926c01Automatic Updater * Copyright (C) 1999-2002  Internet Software Consortium.
0c27b3fe77ac1d5094ba3521e8142d9e7973133fMark Andrews *
0c27b3fe77ac1d5094ba3521e8142d9e7973133fMark Andrews * Permission to use, copy, modify, and distribute this software for any
0c27b3fe77ac1d5094ba3521e8142d9e7973133fMark Andrews * purpose with or without fee is hereby granted, provided that the above
ec5347e2c775f027573ce5648b910361aa926c01Automatic Updater * copyright notice and this permission notice appear in all copies.
cb3e854e181373807f7f011e5050c1a8013b4841Brian Wellington *
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
271c4c7ffafeb0bda21278af6cac4535c0193f18Automatic Updater * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington * AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
dafcb997e390efa4423883dafd100c975c4095d6Mark Andrews * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
dafcb997e390efa4423883dafd100c975c4095d6Mark Andrews * PERFORMANCE OF THIS SOFTWARE.
dafcb997e390efa4423883dafd100c975c4095d6Mark Andrews */
cb3e854e181373807f7f011e5050c1a8013b4841Brian Wellington
cb3e854e181373807f7f011e5050c1a8013b4841Brian Wellington/* $Id: zt.c,v 1.38 2004/03/05 05:09:27 marka Exp $ */
dafcb997e390efa4423883dafd100c975c4095d6Mark Andrews
dafcb997e390efa4423883dafd100c975c4095d6Mark Andrews#include <config.h>
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington#include <isc/magic.h>
271c4c7ffafeb0bda21278af6cac4535c0193f18Automatic Updater#include <isc/mem.h>
ab023a65562e62b85a824509d829b6fad87e00b1Rob Austein#include <isc/util.h>
ab023a65562e62b85a824509d829b6fad87e00b1Rob Austein
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington#include <dns/rbt.h>
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington#include <dns/result.h>
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington#include <dns/zone.h>
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington#include <dns/zt.h>
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellingtonstruct dns_zt {
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington    /* Unlocked. */
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington    unsigned int        magic;
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington    isc_mem_t       *mctx;
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington    dns_rdataclass_t    rdclass;
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington    isc_rwlock_t        rwlock;
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington    /* Locked by lock. */
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington    isc_uint32_t        references;
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington    dns_rbt_t       *table;
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington};
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington#define ZTMAGIC         ISC_MAGIC('Z', 'T', 'b', 'l')
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington#define VALID_ZT(zt)        ISC_MAGIC_VALID(zt, ZTMAGIC)
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellingtonstatic void
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellingtonauto_detach(void *, void *);
c3c8823fed039b3a2b8e5ca8bc2f3301d1dd840eMark Andrews
c3c8823fed039b3a2b8e5ca8bc2f3301d1dd840eMark Andrewsstatic isc_result_t
c3c8823fed039b3a2b8e5ca8bc2f3301d1dd840eMark Andrewsload(dns_zone_t *zone, void *uap);
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellingtonstatic isc_result_t
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellingtonloadnew(dns_zone_t *zone, void *uap);
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellingtonisc_result_t
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellingtondns_zt_create(isc_mem_t *mctx, dns_rdataclass_t rdclass, dns_zt_t **ztp) {
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington    dns_zt_t *zt;
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington    isc_result_t result;
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington    REQUIRE(ztp != NULL && *ztp == NULL);
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington    zt = isc_mem_get(mctx, sizeof(*zt));
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington    if (zt == NULL)
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington        return (ISC_R_NOMEMORY);
c3c8823fed039b3a2b8e5ca8bc2f3301d1dd840eMark Andrews
c3c8823fed039b3a2b8e5ca8bc2f3301d1dd840eMark Andrews    zt->table = NULL;
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington    result = dns_rbt_create(mctx, auto_detach, zt, &zt->table);
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington    if (result != ISC_R_SUCCESS)
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington        goto cleanup_zt;
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington    result = isc_rwlock_init(&zt->rwlock, 0, 0);
1d92d8a2456b23842a649b6104c60a9d6ea25333Brian Wellington    if (result != ISC_R_SUCCESS) {
        UNEXPECTED_ERROR(__FILE__, __LINE__,
                 "isc_rwlock_init() failed: %s",
                 isc_result_totext(result));
        result = ISC_R_UNEXPECTED;
        goto cleanup_rbt;
    }

    zt->mctx = mctx;
    zt->references = 1;
    zt->rdclass = rdclass;
    zt->magic = ZTMAGIC;
    *ztp = zt;

    return (ISC_R_SUCCESS);

   cleanup_rbt:
    dns_rbt_destroy(&zt->table);

   cleanup_zt:
    isc_mem_put(mctx, zt, sizeof(*zt));

    return (result);
}

isc_result_t
dns_zt_mount(dns_zt_t *zt, dns_zone_t *zone) {
    isc_result_t result;
    dns_zone_t *dummy = NULL;
    dns_name_t *name;

    REQUIRE(VALID_ZT(zt));

    name = dns_zone_getorigin(zone);

    RWLOCK(&zt->rwlock, isc_rwlocktype_write);

    result = dns_rbt_addname(zt->table, name, zone);
    if (result == ISC_R_SUCCESS)
        dns_zone_attach(zone, &dummy);

    RWUNLOCK(&zt->rwlock, isc_rwlocktype_write);

    return (result);
}

isc_result_t
dns_zt_unmount(dns_zt_t *zt, dns_zone_t *zone) {
    isc_result_t result;
    dns_name_t *name;

    REQUIRE(VALID_ZT(zt));

    name = dns_zone_getorigin(zone);

    RWLOCK(&zt->rwlock, isc_rwlocktype_write);

    result = dns_rbt_deletename(zt->table, name, ISC_FALSE);

    RWUNLOCK(&zt->rwlock, isc_rwlocktype_write);

    return (result);
}

isc_result_t
dns_zt_find(dns_zt_t *zt, dns_name_t *name, unsigned int options,
        dns_name_t *foundname, dns_zone_t **zonep)
{
    isc_result_t result;
    dns_zone_t *dummy = NULL;
    unsigned int rbtoptions = 0;

    REQUIRE(VALID_ZT(zt));

    if ((options & DNS_ZTFIND_NOEXACT) != 0)
        rbtoptions |= DNS_RBTFIND_NOEXACT;

    RWLOCK(&zt->rwlock, isc_rwlocktype_read);

    result = dns_rbt_findname(zt->table, name, rbtoptions, foundname,
                  (void **) (void*)&dummy);
    if (result == ISC_R_SUCCESS || result == DNS_R_PARTIALMATCH)
        dns_zone_attach(dummy, zonep);

    RWUNLOCK(&zt->rwlock, isc_rwlocktype_read);

    return (result);
}

void
dns_zt_attach(dns_zt_t *zt, dns_zt_t **ztp) {

    REQUIRE(VALID_ZT(zt));
    REQUIRE(ztp != NULL && *ztp == NULL);

    RWLOCK(&zt->rwlock, isc_rwlocktype_write);

    INSIST(zt->references > 0);
    zt->references++;
    INSIST(zt->references != 0);

    RWUNLOCK(&zt->rwlock, isc_rwlocktype_write);

    *ztp = zt;
}

static isc_result_t
flush(dns_zone_t *zone, void *uap) {
    UNUSED(uap);
    return (dns_zone_flush(zone));
}

static void
zt_flushanddetach(dns_zt_t **ztp, isc_boolean_t need_flush) {
    isc_boolean_t destroy = ISC_FALSE;
    dns_zt_t *zt;

    REQUIRE(ztp != NULL && VALID_ZT(*ztp));

    zt = *ztp;

    RWLOCK(&zt->rwlock, isc_rwlocktype_write);

    INSIST(zt->references > 0);
    zt->references--;
    if (zt->references == 0)
        destroy = ISC_TRUE;

    RWUNLOCK(&zt->rwlock, isc_rwlocktype_write);

    if (destroy) {
        if (need_flush)
            (void)dns_zt_apply(zt, ISC_FALSE, flush, NULL);
        dns_rbt_destroy(&zt->table);
        isc_rwlock_destroy(&zt->rwlock);
        zt->magic = 0;
        isc_mem_put(zt->mctx, zt, sizeof(*zt));
    }

    *ztp = NULL;
}

void
dns_zt_flushanddetach(dns_zt_t **ztp) {
    zt_flushanddetach(ztp, ISC_TRUE);
}

void
dns_zt_detach(dns_zt_t **ztp) {
    zt_flushanddetach(ztp, ISC_FALSE);
}

isc_result_t
dns_zt_load(dns_zt_t *zt, isc_boolean_t stop) {
    isc_result_t result;

    REQUIRE(VALID_ZT(zt));

    RWLOCK(&zt->rwlock, isc_rwlocktype_read);
    result = dns_zt_apply(zt, stop, load, NULL);
    RWUNLOCK(&zt->rwlock, isc_rwlocktype_read);
    return (result);
}

static isc_result_t
load(dns_zone_t *zone, void *uap) {
    isc_result_t result;
    UNUSED(uap);
    result = dns_zone_load(zone);
    if (result == DNS_R_CONTINUE || result == DNS_R_UPTODATE)
        result = ISC_R_SUCCESS;
    return (result);
}

isc_result_t
dns_zt_loadnew(dns_zt_t *zt, isc_boolean_t stop) {
    isc_result_t result;

    REQUIRE(VALID_ZT(zt));

    RWLOCK(&zt->rwlock, isc_rwlocktype_read);
    result = dns_zt_apply(zt, stop, loadnew, NULL);
    RWUNLOCK(&zt->rwlock, isc_rwlocktype_read);
    return (result);
}

static isc_result_t
loadnew(dns_zone_t *zone, void *uap) {
    isc_result_t result;
    UNUSED(uap);
    result = dns_zone_loadnew(zone);
    if (result == DNS_R_CONTINUE || result == DNS_R_UPTODATE ||
        result == DNS_R_DYNAMIC)
        result = ISC_R_SUCCESS;
    return (result);
}

isc_result_t
dns_zt_apply(dns_zt_t *zt, isc_boolean_t stop,
         isc_result_t (*action)(dns_zone_t *, void *), void *uap)
{
    dns_rbtnode_t *node;
    dns_rbtnodechain_t chain;
    isc_result_t result;
    dns_zone_t *zone;

    REQUIRE(VALID_ZT(zt));
    REQUIRE(action != NULL);

    dns_rbtnodechain_init(&chain, zt->mctx);
    result = dns_rbtnodechain_first(&chain, zt->table, NULL, NULL);
    if (result == ISC_R_NOTFOUND) {
        /*
         * The tree is empty.
         */
        result = ISC_R_NOMORE;
    }
    while (result == DNS_R_NEWORIGIN || result == ISC_R_SUCCESS) {
        result = dns_rbtnodechain_current(&chain, NULL, NULL,
                          &node);
        if (result == ISC_R_SUCCESS) {
            zone = node->data;
            if (zone != NULL)
                result = (action)(zone, uap);
            if (result != ISC_R_SUCCESS && stop)
                goto cleanup;   /* don't break */
        }
        result = dns_rbtnodechain_next(&chain, NULL, NULL);
    }
    if (result == ISC_R_NOMORE)
        result = ISC_R_SUCCESS;

 cleanup:
    dns_rbtnodechain_invalidate(&chain);

    return (result);
}

/***
 *** Private
 ***/

static void
auto_detach(void *data, void *arg) {
    dns_zone_t *zone = data;

    UNUSED(arg);

    dns_zone_detach(&zone);
}