contrib/bind-9.3/lib/dns/order.c

   1 /*
   2  * Copyright (C) 2004  Internet Systems Consortium, Inc. ("ISC")
   3  * Copyright (C) 2002  Internet Software Consortium.
   4  *
   5  * Permission to use, copy, modify, and distribute this software for any
   6  * purpose with or without fee is hereby granted, provided that the above
   7  * copyright notice and this permission notice appear in all copies.
   8  *
   9  * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
  10  * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
  11  * AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
  12  * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
  13  * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
  14  * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
  15  * PERFORMANCE OF THIS SOFTWARE.
  16  */
  17
  18 /* $Id: order.c,v 1.4.202.4 2004/03/08 09:04:30 marka Exp $ */
  19
  20 #include <config.h>
  21
  22 #include <isc/magic.h>
  23 #include <isc/mem.h>
  24 #include <isc/types.h>
  25 #include <isc/util.h>
  26 #include <isc/refcount.h>
  27
  28 #include <dns/fixedname.h>
  29 #include <dns/name.h>
  30 #include <dns/order.h>
  31 #include <dns/rdataset.h>
  32 #include <dns/types.h>
  33
  34 typedef struct dns_order_ent dns_order_ent_t;
  35 struct dns_order_ent {
  36         dns_fixedname_t                 name;
  37         dns_rdataclass_t                rdclass;
  38         dns_rdatatype_t                 rdtype;
  39         unsigned int                    mode;
  40         ISC_LINK(dns_order_ent_t)       link;
  41 };
  42
  43 struct dns_order {
  44         unsigned int                    magic;
  45         isc_refcount_t                  references;
  46         ISC_LIST(dns_order_ent_t)       ents;
  47         isc_mem_t                       *mctx;
  48 };
  49
  50 #define DNS_ORDER_MAGIC ISC_MAGIC('O','r','d','r')
  51 #define DNS_ORDER_VALID(order)  ISC_MAGIC_VALID(order, DNS_ORDER_MAGIC)
  52
  53 isc_result_t
  54 dns_order_create(isc_mem_t *mctx, dns_order_t **orderp) {
  55         dns_order_t *order;
  56         REQUIRE(orderp != NULL && *orderp == NULL);
  57
  58         order = isc_mem_get(mctx, sizeof(*order));
  59         if (order == NULL)
  60                 return (ISC_R_NOMEMORY);
  61
  62         ISC_LIST_INIT(order->ents);
  63         isc_refcount_init(&order->references, 1);     /* Implicit attach. */
  64
  65         order->mctx = NULL;
  66         isc_mem_attach(mctx, &order->mctx);
  67         order->magic = DNS_ORDER_MAGIC;
  68         *orderp = order;
  69         return (ISC_R_SUCCESS);
  70 }
  71
  72 isc_result_t
  73 dns_order_add(dns_order_t *order, dns_name_t *name,
  74               dns_rdatatype_t rdtype, dns_rdataclass_t rdclass,
  75               unsigned int mode)
  76 {
  77         dns_order_ent_t *ent;
  78
  79         REQUIRE(DNS_ORDER_VALID(order));
  80         REQUIRE(mode == DNS_RDATASETATTR_RANDOMIZE ||
  81                 mode == DNS_RDATASETATTR_FIXEDORDER ||
  82                 mode == 0 /* DNS_RDATASETATTR_CYCLIC */ );
  83
  84         ent = isc_mem_get(order->mctx, sizeof(*ent));
  85         if (ent == NULL)
  86                 return (ISC_R_NOMEMORY);
  87
  88         dns_fixedname_init(&ent->name);
  89         RUNTIME_CHECK(dns_name_copy(name, dns_fixedname_name(&ent->name), NULL)
  90                       == ISC_R_SUCCESS);
  91         ent->rdtype = rdtype;
  92         ent->rdclass = rdclass;
  93         ent->mode = mode;
  94         ISC_LINK_INIT(ent, link);
  95         ISC_LIST_INITANDAPPEND(order->ents, ent, link);
  96         return (ISC_R_SUCCESS);
  97 }
  98
  99 static inline isc_boolean_t
 100 match(dns_name_t *name1, dns_name_t *name2) {
 101
 102         if (dns_name_iswildcard(name2))
 103                 return(dns_name_matcheswildcard(name1, name2));
 104         return (dns_name_equal(name1, name2));
 105 }
 106
 107 unsigned int
 108 dns_order_find(dns_order_t *order, dns_name_t *name,
 109                dns_rdatatype_t rdtype, dns_rdataclass_t rdclass)
 110 {
 111         dns_order_ent_t *ent;
 112         REQUIRE(DNS_ORDER_VALID(order));
 113
 114         for (ent = ISC_LIST_HEAD(order->ents);
 115              ent != NULL;
 116              ent = ISC_LIST_NEXT(ent, link)) {
 117                 if (ent->rdtype != rdtype && ent->rdtype != dns_rdatatype_any)
 118                         continue;
 119                 if (ent->rdclass != rdclass &&
 120                     ent->rdclass != dns_rdataclass_any)
 121                         continue;
 122                 if (match(name, dns_fixedname_name(&ent->name)))
 123                         return (ent->mode);
 124         }
 125         return (0);
 126 }
 127
 128 void
 129 dns_order_attach(dns_order_t *source, dns_order_t **target) {
 130         REQUIRE(DNS_ORDER_VALID(source));
 131         REQUIRE(target != NULL && *target == NULL);
 132         isc_refcount_increment(&source->references, NULL);
 133         *target = source;
 134 }
 135
 136 void
 137 dns_order_detach(dns_order_t **orderp) {
 138         dns_order_t *order;
 139         dns_order_ent_t *ent;
 140         unsigned int references;
 141
 142         REQUIRE(orderp != NULL);
 143         order = *orderp;
 144         REQUIRE(DNS_ORDER_VALID(order));
 145         isc_refcount_decrement(&order->references, &references);
 146         *orderp = NULL;
 147         if (references != 0)
 148                 return;
 149
 150         order->magic = 0;
 151         while ((ent = ISC_LIST_HEAD(order->ents)) != NULL) {
 152                 ISC_LIST_UNLINK(order->ents, ent, link);
 153                 isc_mem_put(order->mctx, ent, sizeof(*ent));
 154         }
 155         isc_refcount_destroy(&order->references);
 156         isc_mem_putanddetach(&order->mctx, order, sizeof(*order));
 157 }