2018-08-03 14:16:41 -07:00
|
|
|
/*
|
|
|
|
* Copyright (C) Internet Systems Consortium, Inc. ("ISC")
|
|
|
|
*
|
|
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
*
|
|
|
|
* See the COPYRIGHT file distributed with this work for additional
|
|
|
|
* information regarding copyright ownership.
|
|
|
|
*/
|
|
|
|
|
|
|
|
/*! \file */
|
|
|
|
|
|
|
|
#include <config.h>
|
|
|
|
|
|
|
|
#include <string.h>
|
|
|
|
|
|
|
|
#if HAVE_DLFCN_H
|
|
|
|
#include <dlfcn.h>
|
|
|
|
#elif _WIN32
|
|
|
|
#include <windows.h>
|
|
|
|
#endif
|
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
#include <isc/list.h>
|
|
|
|
#include <isc/log.h>
|
2018-08-03 14:16:41 -07:00
|
|
|
#include <isc/mem.h>
|
|
|
|
#include <isc/mutex.h>
|
|
|
|
#include <isc/result.h>
|
2018-09-14 12:32:36 -07:00
|
|
|
#include <isc/platform.h>
|
2018-08-03 14:16:41 -07:00
|
|
|
#include <isc/util.h>
|
2018-09-14 12:32:36 -07:00
|
|
|
#include <isc/types.h>
|
2018-08-03 14:16:41 -07:00
|
|
|
|
|
|
|
#include <ns/hooks.h>
|
|
|
|
#include <ns/log.h>
|
2018-09-14 12:32:36 -07:00
|
|
|
#include <ns/query.h>
|
2018-08-03 14:16:41 -07:00
|
|
|
|
|
|
|
#define CHECK(op) \
|
2018-09-14 12:32:36 -07:00
|
|
|
do { \
|
|
|
|
result = (op); \
|
|
|
|
if (result != ISC_R_SUCCESS) { \
|
|
|
|
goto cleanup; \
|
|
|
|
} \
|
2018-08-03 14:16:41 -07:00
|
|
|
} while (0)
|
|
|
|
|
|
|
|
typedef struct ns_hook_module ns_hook_module_t;
|
|
|
|
struct ns_hook_module {
|
|
|
|
isc_mem_t *mctx;
|
|
|
|
void *handle;
|
2018-09-14 12:32:36 -07:00
|
|
|
char *modpath;
|
2018-08-03 14:16:41 -07:00
|
|
|
ns_hook_register_t *register_func;
|
|
|
|
ns_hook_destroy_t *destroy_func;
|
|
|
|
void *inst;
|
|
|
|
LINK(ns_hook_module_t) link;
|
|
|
|
};
|
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
static ns_hooklist_t default_hooktable[NS_HOOKPOINTS_COUNT];
|
|
|
|
LIBNS_EXTERNAL_DATA ns_hooktable_t *ns__hook_table = &default_hooktable;
|
2018-08-03 14:16:41 -07:00
|
|
|
|
|
|
|
/*
|
|
|
|
* List of hook modules.
|
|
|
|
*
|
|
|
|
* These are stored here so they can be cleaned up on shutdown.
|
|
|
|
* (The order in which they are stored is not important.)
|
|
|
|
*/
|
2018-09-14 12:32:36 -07:00
|
|
|
static ISC_LIST(ns_hook_module_t) hook_modules;
|
|
|
|
static bool hook_modules_initialized = false;
|
2018-08-03 14:16:41 -07:00
|
|
|
|
|
|
|
#if HAVE_DLFCN_H && HAVE_DLOPEN
|
|
|
|
static isc_result_t
|
2018-09-14 12:32:36 -07:00
|
|
|
load_symbol(void *handle, const char *modpath,
|
2018-08-03 14:16:41 -07:00
|
|
|
const char *symbol_name, void **symbolp)
|
|
|
|
{
|
|
|
|
void *symbol;
|
|
|
|
|
|
|
|
REQUIRE(handle != NULL);
|
|
|
|
REQUIRE(symbolp != NULL && *symbolp == NULL);
|
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
/*
|
|
|
|
* Clear any pre-existing error conditions before running dlsym().
|
|
|
|
* (In this case, we expect dlsym() to return non-NULL values
|
|
|
|
* and will always return an error if it returns NULL, but
|
|
|
|
* this ensures that we'll report the correct error condition
|
|
|
|
* if there is one.)
|
|
|
|
*/
|
|
|
|
dlerror();
|
2018-08-03 14:16:41 -07:00
|
|
|
symbol = dlsym(handle, symbol_name);
|
|
|
|
if (symbol == NULL) {
|
2018-09-14 12:32:36 -07:00
|
|
|
const char *errmsg = dlerror();
|
2018-08-03 14:16:41 -07:00
|
|
|
if (errmsg == NULL) {
|
|
|
|
errmsg = "returned function pointer is NULL";
|
|
|
|
}
|
|
|
|
isc_log_write(ns_lctx, NS_LOGCATEGORY_GENERAL,
|
|
|
|
NS_LOGMODULE_HOOKS, ISC_LOG_ERROR,
|
2018-09-14 12:32:36 -07:00
|
|
|
"failed to look up symbol %s in "
|
2018-08-03 14:16:41 -07:00
|
|
|
"hook module '%s': %s",
|
2018-09-14 12:32:36 -07:00
|
|
|
symbol_name, modpath, errmsg);
|
2018-08-03 14:16:41 -07:00
|
|
|
return (ISC_R_FAILURE);
|
|
|
|
}
|
|
|
|
|
|
|
|
*symbolp = symbol;
|
|
|
|
|
|
|
|
return (ISC_R_SUCCESS);
|
|
|
|
}
|
|
|
|
|
|
|
|
static isc_result_t
|
2018-09-14 12:32:36 -07:00
|
|
|
load_library(isc_mem_t *mctx, const char *modpath, ns_hook_module_t **hmodp) {
|
2018-08-03 14:16:41 -07:00
|
|
|
isc_result_t result;
|
|
|
|
void *handle = NULL;
|
|
|
|
ns_hook_module_t *hmod = NULL;
|
|
|
|
ns_hook_register_t *register_func = NULL;
|
|
|
|
ns_hook_destroy_t *destroy_func = NULL;
|
|
|
|
ns_hook_version_t *version_func = NULL;
|
|
|
|
int version, flags;
|
|
|
|
|
|
|
|
REQUIRE(hmodp != NULL && *hmodp == NULL);
|
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
flags = RTLD_NOW | RTLD_LOCAL;
|
2018-08-03 14:16:41 -07:00
|
|
|
#ifdef RTLD_DEEPBIND
|
|
|
|
flags |= RTLD_DEEPBIND;
|
|
|
|
#endif
|
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
handle = dlopen(modpath, flags);
|
2018-08-03 14:16:41 -07:00
|
|
|
if (handle == NULL) {
|
2018-09-14 12:32:36 -07:00
|
|
|
const char *errmsg = dlerror();
|
|
|
|
if (errmsg == NULL) {
|
|
|
|
errmsg = "unknown error";
|
|
|
|
}
|
|
|
|
isc_log_write(ns_lctx, NS_LOGCATEGORY_GENERAL,
|
|
|
|
NS_LOGMODULE_HOOKS, ISC_LOG_ERROR,
|
|
|
|
"failed to dlopen() hook module '%s': %s",
|
|
|
|
modpath, errmsg);
|
|
|
|
return (ISC_R_FAILURE);
|
2018-08-03 14:16:41 -07:00
|
|
|
}
|
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
CHECK(load_symbol(handle, modpath, "hook_version",
|
2018-08-03 14:16:41 -07:00
|
|
|
(void **)&version_func));
|
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
version = version_func();
|
2018-08-03 14:16:41 -07:00
|
|
|
if (version < (NS_HOOK_VERSION - NS_HOOK_AGE) ||
|
|
|
|
version > NS_HOOK_VERSION)
|
|
|
|
{
|
|
|
|
isc_log_write(ns_lctx, NS_LOGCATEGORY_GENERAL,
|
|
|
|
NS_LOGMODULE_HOOKS, ISC_LOG_ERROR,
|
2018-09-14 12:32:36 -07:00
|
|
|
"hook API version mismatch: %d/%d",
|
2018-08-03 14:16:41 -07:00
|
|
|
version, NS_HOOK_VERSION);
|
|
|
|
CHECK(ISC_R_FAILURE);
|
|
|
|
}
|
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
CHECK(load_symbol(handle, modpath, "hook_register",
|
2018-08-03 14:16:41 -07:00
|
|
|
(void **)®ister_func));
|
2018-09-14 12:32:36 -07:00
|
|
|
CHECK(load_symbol(handle, modpath, "hook_destroy",
|
2018-08-03 14:16:41 -07:00
|
|
|
(void **)&destroy_func));
|
|
|
|
|
|
|
|
hmod = isc_mem_get(mctx, sizeof(*hmod));
|
|
|
|
hmod->mctx = NULL;
|
|
|
|
isc_mem_attach(mctx, &hmod->mctx);
|
|
|
|
hmod->handle = handle;
|
2018-09-14 12:32:36 -07:00
|
|
|
hmod->modpath = isc_mem_strdup(hmod->mctx, modpath);
|
2018-08-03 14:16:41 -07:00
|
|
|
hmod->register_func = register_func;
|
|
|
|
hmod->destroy_func = destroy_func;
|
|
|
|
hmod->inst = NULL;
|
|
|
|
|
|
|
|
ISC_LINK_INIT(hmod, link);
|
|
|
|
|
|
|
|
*hmodp = hmod;
|
|
|
|
hmod = NULL;
|
|
|
|
|
|
|
|
cleanup:
|
|
|
|
if (result != ISC_R_SUCCESS) {
|
|
|
|
isc_log_write(ns_lctx, NS_LOGCATEGORY_GENERAL,
|
|
|
|
NS_LOGMODULE_HOOKS, ISC_LOG_ERROR,
|
|
|
|
"failed to dynamically load "
|
2018-09-14 12:32:36 -07:00
|
|
|
"module '%s': %s", modpath,
|
|
|
|
isc_result_totext(result));
|
2018-08-03 14:16:41 -07:00
|
|
|
|
|
|
|
if (hmod != NULL) {
|
2018-09-14 12:32:36 -07:00
|
|
|
isc_mem_putanddetach(&hmod->mctx, hmod, sizeof(*hmod));
|
2018-08-03 14:16:41 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
if (handle != NULL) {
|
2018-09-14 12:32:36 -07:00
|
|
|
(void) dlclose(handle);
|
2018-08-03 14:16:41 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return (result);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
unload_library(ns_hook_module_t **hmodp) {
|
|
|
|
ns_hook_module_t *hmod;
|
|
|
|
|
|
|
|
REQUIRE(hmodp != NULL && *hmodp != NULL);
|
|
|
|
|
|
|
|
hmod = *hmodp;
|
|
|
|
*hmodp = NULL;
|
|
|
|
|
|
|
|
if (hmod->handle != NULL) {
|
2018-09-14 12:32:36 -07:00
|
|
|
(void) dlclose(hmod->handle);
|
2018-08-03 14:16:41 -07:00
|
|
|
}
|
2018-09-14 12:32:36 -07:00
|
|
|
if (hmod->modpath != NULL) {
|
|
|
|
isc_mem_free(hmod->mctx, hmod->modpath);
|
2018-08-03 14:16:41 -07:00
|
|
|
}
|
|
|
|
|
2018-08-10 19:32:12 -07:00
|
|
|
isc_mem_putanddetach(&hmod->mctx, hmod, sizeof(*hmod));
|
2018-08-03 14:16:41 -07:00
|
|
|
}
|
|
|
|
#elif _WIN32
|
|
|
|
static isc_result_t
|
2018-09-14 12:32:36 -07:00
|
|
|
load_symbol(HMODULE handle, const char *modpath,
|
2018-08-03 14:16:41 -07:00
|
|
|
const char *symbol_name, void **symbolp)
|
|
|
|
{
|
|
|
|
void *symbol;
|
|
|
|
|
|
|
|
REQUIRE(handle != NULL);
|
|
|
|
REQUIRE(symbolp != NULL && *symbolp == NULL);
|
|
|
|
|
|
|
|
symbol = GetProcAddress(handle, symbol_name);
|
|
|
|
if (symbol == NULL) {
|
|
|
|
int errstatus = GetLastError();
|
|
|
|
isc_log_write(ns_lctx, NS_LOGCATEGORY_GENERAL,
|
|
|
|
NS_LOGMODULE_HOOKS, ISC_LOG_ERROR,
|
|
|
|
"failed to look up symbol %s in "
|
|
|
|
"module '%s': %d",
|
2018-09-14 12:32:36 -07:00
|
|
|
symbol_name, modpath, errstatus);
|
2018-08-03 14:16:41 -07:00
|
|
|
return (ISC_R_FAILURE);
|
|
|
|
}
|
|
|
|
|
|
|
|
*symbolp = symbol;
|
|
|
|
|
|
|
|
return (ISC_R_SUCCESS);
|
|
|
|
}
|
|
|
|
|
|
|
|
static isc_result_t
|
2018-09-14 12:32:36 -07:00
|
|
|
load_library(isc_mem_t *mctx, const char *modpath, ns_hook_module_t **hmodp) {
|
2018-08-03 14:16:41 -07:00
|
|
|
isc_result_t result;
|
|
|
|
HMODULE handle;
|
|
|
|
ns_hook_module_t *hmod = NULL;
|
|
|
|
ns_hook_register_t *register_func = NULL;
|
|
|
|
ns_hook_destroy_t *destroy_func = NULL;
|
|
|
|
ns_hook_version_t *version_func = NULL;
|
|
|
|
int version;
|
|
|
|
|
|
|
|
REQUIRE(hmodp != NULL && *hmodp == NULL);
|
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
handle = LoadLibraryA(modpath);
|
2018-08-03 14:16:41 -07:00
|
|
|
if (handle == NULL) {
|
|
|
|
CHECK(ISC_R_FAILURE);
|
|
|
|
}
|
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
CHECK(load_symbol(handle, modpath, "hook_version",
|
2018-08-03 14:16:41 -07:00
|
|
|
(void **)&version_func));
|
|
|
|
|
|
|
|
version = version_func(NULL);
|
|
|
|
if (version < (NS_HOOK_VERSION - NS_HOOK_AGE) ||
|
|
|
|
version > NS_HOOK_VERSION)
|
|
|
|
{
|
|
|
|
isc_log_write(ns_lctx, NS_LOGCATEGORY_GENERAL,
|
|
|
|
NS_LOGMODULE_HOOKS, ISC_LOG_ERROR,
|
2018-09-14 12:32:36 -07:00
|
|
|
"hook API version mismatch: %d/%d",
|
2018-08-03 14:16:41 -07:00
|
|
|
version, NS_HOOK_VERSION);
|
|
|
|
CHECK(ISC_R_FAILURE);
|
|
|
|
}
|
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
CHECK(load_symbol(handle, modpath, "hook_register",
|
2018-08-03 14:16:41 -07:00
|
|
|
(void **)®ister_func));
|
2018-09-14 12:32:36 -07:00
|
|
|
CHECK(load_symbol(handle, modpath, "hook_destroy",
|
2018-08-03 14:16:41 -07:00
|
|
|
(void **)&destroy_func));
|
|
|
|
|
|
|
|
hmod = isc_mem_get(mctx, sizeof(*hmod));
|
|
|
|
hmod->mctx = NULL;
|
|
|
|
isc_mem_attach(mctx, &hmod->mctx);
|
|
|
|
hmod->handle = handle;
|
2018-09-14 12:32:36 -07:00
|
|
|
hmod->modpath = isc_mem_strdup(hmod->mctx, modpath);
|
2018-08-03 14:16:41 -07:00
|
|
|
hmod->register_func = register_func;
|
|
|
|
hmod->destroy_func = destroy_func;
|
|
|
|
hmod->inst = NULL;
|
|
|
|
|
|
|
|
ISC_LINK_INIT(hmod, link);
|
|
|
|
|
|
|
|
*hmodp = hmod;
|
|
|
|
hmod = NULL;
|
|
|
|
|
|
|
|
cleanup:
|
|
|
|
if (result != ISC_R_SUCCESS) {
|
|
|
|
isc_log_write(ns_lctx, NS_LOGCATEGORY_GENERAL,
|
|
|
|
NS_LOGMODULE_HOOKS, ISC_LOG_ERROR,
|
|
|
|
"failed to dynamically load "
|
2018-09-14 12:32:36 -07:00
|
|
|
"hook module '%s': %d (%s)", modpath,
|
2018-08-03 14:16:41 -07:00
|
|
|
GetLastError(), isc_result_totext(result));
|
2018-09-14 12:32:36 -07:00
|
|
|
|
2018-08-03 14:16:41 -07:00
|
|
|
if (hmod != NULL) {
|
2018-09-14 12:32:36 -07:00
|
|
|
isc_mem_putanddetach(&hmod->mctx, hmod, sizeof(*hmod));
|
2018-08-03 14:16:41 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
if (handle != NULL) {
|
|
|
|
FreeLibrary(handle);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return (result);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
unload_library(ns_hook_module_t **hmodp) {
|
|
|
|
ns_hook_module_t *hmod;
|
|
|
|
|
|
|
|
REQUIRE(hmodp != NULL && *hmodp != NULL);
|
|
|
|
|
|
|
|
hmod = *hmodp;
|
|
|
|
*hmodp = NULL;
|
|
|
|
|
|
|
|
if (hmod->handle != NULL) {
|
|
|
|
FreeLibrary(hmod->handle);
|
|
|
|
}
|
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
if (hmod->modpath != NULL) {
|
|
|
|
isc_mem_free(hmod->mctx, hmod->modpath);
|
2018-08-03 14:16:41 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
isc_mem_putanddetach(&hmod->mctx, hmod, sizeof(*hmod));
|
|
|
|
}
|
|
|
|
#else /* HAVE_DLFCN_H || _WIN32 */
|
|
|
|
static isc_result_t
|
2018-09-14 12:32:36 -07:00
|
|
|
load_library(isc_mem_t *mctx, const char *modpath, ns_hook_module_t **hmodp) {
|
2018-08-03 14:16:41 -07:00
|
|
|
UNUSED(mctx);
|
2018-09-14 12:32:36 -07:00
|
|
|
UNUSED(modpath);
|
2018-08-03 14:16:41 -07:00
|
|
|
UNUSED(hmodp);
|
|
|
|
|
|
|
|
isc_log_write(ns_lctx, NS_LOGCATEGORY_GENERAL,
|
|
|
|
NS_LOGMODULE_HOOKS, ISC_LOG_ERROR,
|
2018-09-14 12:32:36 -07:00
|
|
|
"hook module support is not implemented");
|
2018-08-03 14:16:41 -07:00
|
|
|
|
|
|
|
return (ISC_R_NOTIMPLEMENTED);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
unload_library(ns_hook_module_t **hmodp) {
|
|
|
|
UNUSED(hmodp);
|
|
|
|
}
|
|
|
|
#endif /* HAVE_DLFCN_H */
|
|
|
|
|
|
|
|
isc_result_t
|
2018-09-19 16:25:50 -07:00
|
|
|
ns_hookmodule_load(const char *modpath, const char *parameters,
|
2018-09-14 12:32:36 -07:00
|
|
|
const char *cfg_file, unsigned long cfg_line,
|
2018-08-12 23:06:00 -07:00
|
|
|
const void *cfg, void *actx,
|
2018-08-10 19:32:12 -07:00
|
|
|
ns_hookctx_t *hctx, ns_hooktable_t *hooktable)
|
2018-08-03 14:16:41 -07:00
|
|
|
{
|
|
|
|
isc_result_t result;
|
2018-09-14 12:32:36 -07:00
|
|
|
ns_hook_module_t *hmod = NULL;
|
2018-08-03 14:16:41 -07:00
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
REQUIRE(hook_modules_initialized);
|
2018-08-10 19:32:12 -07:00
|
|
|
REQUIRE(NS_HOOKCTX_VALID(hctx));
|
2018-09-14 12:32:36 -07:00
|
|
|
REQUIRE(hooktable != NULL);
|
2018-08-03 14:16:41 -07:00
|
|
|
|
|
|
|
isc_log_write(ns_lctx, NS_LOGCATEGORY_GENERAL,
|
|
|
|
NS_LOGMODULE_HOOKS, ISC_LOG_INFO,
|
2018-09-14 12:32:36 -07:00
|
|
|
"loading module '%s'", modpath);
|
2018-08-03 14:16:41 -07:00
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
CHECK(load_library(hctx->mctx, modpath, &hmod));
|
2018-09-19 16:25:50 -07:00
|
|
|
CHECK(hmod->register_func(parameters, cfg_file, cfg_line,
|
|
|
|
cfg, actx, hctx, hooktable, &hmod->inst));
|
2018-08-03 14:16:41 -07:00
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
ISC_LIST_APPEND(hook_modules, hmod, link);
|
2018-08-03 14:16:41 -07:00
|
|
|
|
|
|
|
cleanup:
|
2018-09-14 12:32:36 -07:00
|
|
|
if (result != ISC_R_SUCCESS && hmod != NULL) {
|
|
|
|
unload_library(&hmod);
|
2018-08-03 14:16:41 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
return (result);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2018-09-14 12:32:36 -07:00
|
|
|
ns_hookmodule_unload_all(void) {
|
2018-08-03 14:16:41 -07:00
|
|
|
ns_hook_module_t *hmod, *prev;
|
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
if (!hook_modules_initialized) {
|
|
|
|
return;
|
|
|
|
}
|
2018-08-03 14:16:41 -07:00
|
|
|
|
|
|
|
hmod = ISC_LIST_TAIL(hook_modules);
|
|
|
|
while (hmod != NULL) {
|
2018-09-14 12:32:36 -07:00
|
|
|
prev = ISC_LIST_PREV(hmod, link);
|
|
|
|
ISC_LIST_UNLINK(hook_modules, hmod, link);
|
2018-08-03 14:16:41 -07:00
|
|
|
isc_log_write(ns_lctx, NS_LOGCATEGORY_GENERAL,
|
|
|
|
NS_LOGMODULE_HOOKS, ISC_LOG_INFO,
|
2018-09-14 12:32:36 -07:00
|
|
|
"unloading module '%s'", hmod->modpath);
|
2018-08-03 14:16:41 -07:00
|
|
|
hmod->destroy_func(&hmod->inst);
|
|
|
|
ENSURE(hmod->inst == NULL);
|
|
|
|
unload_library(&hmod);
|
|
|
|
hmod = prev;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-10 19:32:12 -07:00
|
|
|
isc_result_t
|
|
|
|
ns_hook_createctx(isc_mem_t *mctx, ns_hookctx_t **hctxp) {
|
|
|
|
ns_hookctx_t *hctx;
|
|
|
|
|
|
|
|
REQUIRE(hctxp != NULL && *hctxp == NULL);
|
|
|
|
|
|
|
|
hctx = isc_mem_get(mctx, sizeof(*hctx));
|
|
|
|
memset(hctx, 0, sizeof(*hctx));
|
|
|
|
hctx->lctx = ns_lctx;
|
|
|
|
|
|
|
|
isc_mem_attach(mctx, &hctx->mctx);
|
|
|
|
hctx->magic = NS_HOOKCTX_MAGIC;
|
|
|
|
|
|
|
|
*hctxp = hctx;
|
|
|
|
|
|
|
|
return (ISC_R_SUCCESS);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ns_hook_destroyctx(ns_hookctx_t **hctxp) {
|
|
|
|
ns_hookctx_t *hctx;
|
|
|
|
|
|
|
|
REQUIRE(hctxp != NULL && NS_HOOKCTX_VALID(*hctxp));
|
|
|
|
|
|
|
|
hctx = *hctxp;
|
|
|
|
*hctxp = NULL;
|
|
|
|
|
|
|
|
hctx->magic = 0;
|
|
|
|
|
|
|
|
isc_mem_putanddetach(&hctx->mctx, hctx, sizeof(*hctx));
|
|
|
|
}
|
|
|
|
|
2018-08-03 14:16:41 -07:00
|
|
|
void
|
|
|
|
ns_hooktable_init(ns_hooktable_t *hooktable) {
|
|
|
|
int i;
|
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
if (!hook_modules_initialized) {
|
|
|
|
ISC_LIST_INIT(hook_modules);
|
|
|
|
hook_modules_initialized = true;
|
|
|
|
}
|
2018-08-03 14:16:41 -07:00
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
for (i = 0; i < NS_HOOKPOINTS_COUNT; i++) {
|
2018-08-03 14:16:41 -07:00
|
|
|
ISC_LIST_INIT((*hooktable)[i]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-12 11:19:36 -07:00
|
|
|
isc_result_t
|
|
|
|
ns_hooktable_create(isc_mem_t *mctx, ns_hooktable_t **tablep) {
|
|
|
|
ns_hooktable_t *hooktable;
|
|
|
|
|
|
|
|
REQUIRE(tablep != NULL && *tablep == NULL);
|
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
hooktable = isc_mem_get(mctx, sizeof(*hooktable));
|
2018-08-12 11:19:36 -07:00
|
|
|
|
|
|
|
ns_hooktable_init(hooktable);
|
|
|
|
|
|
|
|
*tablep = hooktable;
|
|
|
|
|
|
|
|
return (ISC_R_SUCCESS);
|
2018-08-03 14:16:41 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2018-08-12 11:19:36 -07:00
|
|
|
ns_hooktable_free(isc_mem_t *mctx, void **tablep) {
|
2018-09-14 12:32:36 -07:00
|
|
|
ns_hooktable_t *table;
|
|
|
|
|
2018-08-12 11:19:36 -07:00
|
|
|
REQUIRE(tablep != NULL && *tablep != NULL);
|
|
|
|
|
2018-09-14 12:32:36 -07:00
|
|
|
table = *tablep;
|
2018-08-12 11:19:36 -07:00
|
|
|
*tablep = NULL;
|
2018-09-14 12:32:36 -07:00
|
|
|
isc_mem_put(mctx, table, sizeof(*table));
|
2018-08-03 14:16:41 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
ns_hook_add(ns_hooktable_t *hooktable, ns_hookpoint_t hookpoint,
|
|
|
|
ns_hook_t *hook)
|
|
|
|
{
|
2018-09-14 12:32:36 -07:00
|
|
|
REQUIRE(hookpoint < NS_HOOKPOINTS_COUNT);
|
2018-08-03 14:16:41 -07:00
|
|
|
REQUIRE(hook != NULL);
|
|
|
|
|
|
|
|
if (hooktable == NULL) {
|
|
|
|
hooktable = ns__hook_table;
|
|
|
|
}
|
|
|
|
|
|
|
|
ISC_LINK_INIT(hook, link);
|
|
|
|
ISC_LIST_APPEND((*hooktable)[hookpoint], hook, link);
|
|
|
|
}
|