1/* 2 * Copyright (C) 2004, 2005, 2007, 2009 Internet Systems Consortium, Inc. ("ISC") 3 * Copyright (C) 1999-2001 Internet Software Consortium. 4 * 5 * Permission to use, copy, modify, and/or 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: lib.c,v 1.19 2009/09/03 00:12:23 each Exp $ */ 19 20/*! \file */ 21 22#include <config.h> 23 24#include <stddef.h> 25 26#include <isc/hash.h> 27#include <isc/mem.h> 28#include <isc/msgcat.h> 29#include <isc/mutex.h> 30#include <isc/once.h> 31#include <isc/util.h> 32 33#include <dns/db.h> 34#include <dns/ecdb.h> 35#include <dns/lib.h> 36#include <dns/result.h> 37 38#include <dst/dst.h> 39 40 41/*** 42 *** Globals 43 ***/ 44 45LIBDNS_EXTERNAL_DATA unsigned int dns_pps = 0U; 46LIBDNS_EXTERNAL_DATA isc_msgcat_t * dns_msgcat = NULL; 47 48 49/*** 50 *** Private 51 ***/ 52 53static isc_once_t msgcat_once = ISC_ONCE_INIT; 54 55 56/*** 57 *** Functions 58 ***/ 59 60static void 61open_msgcat(void) { 62 isc_msgcat_open("libdns.cat", &dns_msgcat); 63} 64 65void 66dns_lib_initmsgcat(void) { 67 68 /* 69 * Initialize the DNS library's message catalog, dns_msgcat, if it 70 * has not already been initialized. 71 */ 72 73 RUNTIME_CHECK(isc_once_do(&msgcat_once, open_msgcat) == ISC_R_SUCCESS); 74} 75 76static isc_once_t init_once = ISC_ONCE_INIT; 77static isc_mem_t *dns_g_mctx = NULL; 78#ifndef BIND9 79static dns_dbimplementation_t *dbimp = NULL; 80#endif 81static isc_boolean_t initialize_done = ISC_FALSE; 82static isc_mutex_t reflock; 83static unsigned int references = 0; 84 85static void 86initialize(void) { 87 isc_result_t result; 88 89 REQUIRE(initialize_done == ISC_FALSE); 90 91 result = isc_mem_create(0, 0, &dns_g_mctx); 92 if (result != ISC_R_SUCCESS) 93 return; 94 dns_result_register(); 95#ifndef BIND9 96 result = dns_ecdb_register(dns_g_mctx, &dbimp); 97 if (result != ISC_R_SUCCESS) 98 goto cleanup_mctx; 99#endif 100 result = isc_hash_create(dns_g_mctx, NULL, DNS_NAME_MAXWIRE); 101 if (result != ISC_R_SUCCESS) 102 goto cleanup_db; 103 104 result = dst_lib_init(dns_g_mctx, NULL, 0); 105 if (result != ISC_R_SUCCESS) 106 goto cleanup_hash; 107 108 result = isc_mutex_init(&reflock); 109 if (result != ISC_R_SUCCESS) 110 goto cleanup_dst; 111 112 initialize_done = ISC_TRUE; 113 return; 114 115 cleanup_dst: 116 dst_lib_destroy(); 117 cleanup_hash: 118 isc_hash_destroy(); 119 cleanup_db: 120#ifndef BIND9 121 dns_ecdb_unregister(&dbimp); 122 cleanup_mctx: 123#endif 124 isc_mem_detach(&dns_g_mctx); 125} 126 127isc_result_t 128dns_lib_init(void) { 129 isc_result_t result; 130 131 /* 132 * Since this routine is expected to be used by a normal application, 133 * it should be better to return an error, instead of an emergency 134 * abort, on any failure. 135 */ 136 result = isc_once_do(&init_once, initialize); 137 if (result != ISC_R_SUCCESS) 138 return (result); 139 140 if (!initialize_done) 141 return (ISC_R_FAILURE); 142 143 LOCK(&reflock); 144 references++; 145 UNLOCK(&reflock); 146 147 return (ISC_R_SUCCESS); 148} 149 150void 151dns_lib_shutdown(void) { 152 isc_boolean_t cleanup_ok = ISC_FALSE; 153 154 LOCK(&reflock); 155 if (--references == 0) 156 cleanup_ok = ISC_TRUE; 157 UNLOCK(&reflock); 158 159 if (!cleanup_ok) 160 return; 161 162 dst_lib_destroy(); 163 isc_hash_destroy(); 164#ifndef BIND9 165 dns_ecdb_unregister(&dbimp); 166#endif 167 isc_mem_detach(&dns_g_mctx); 168} 169