2 * services/modstack.c - stack of modules
4 * Copyright (c) 2007, NLnet Labs. All rights reserved.
6 * This software is open source.
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
12 * Redistributions of source code must retain the above copyright notice,
13 * this list of conditions and the following disclaimer.
15 * Redistributions in binary form must reproduce the above copyright notice,
16 * this list of conditions and the following disclaimer in the documentation
17 * and/or other materials provided with the distribution.
19 * Neither the name of the NLNET LABS nor the names of its contributors may
20 * be used to endorse or promote products derived from this software without
21 * specific prior written permission.
23 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27 * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
29 * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
30 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
31 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
32 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
33 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
39 * This file contains functions to help maintain a stack of modules.
43 #include "services/modstack.h"
44 #include "util/module.h"
45 #include "util/fptr_wlist.h"
46 #include "dns64/dns64.h"
47 #include "iterator/iterator.h"
48 #include "validator/validator.h"
49 #include "respip/respip.h"
51 #ifdef WITH_PYTHONMODULE
52 #include "pythonmod/pythonmod.h"
55 #include "cachedb/cachedb.h"
58 #include "ipsecmod/ipsecmod.h"
61 #include "edns-subnet/subnetmod.h"
64 /** count number of modules (words) in the string */
66 count_modules(const char* s)
73 while(*s && isspace((unsigned char)*s))
75 if(*s && !isspace((unsigned char)*s)) {
78 while(*s && !isspace((unsigned char)*s))
86 modstack_init(struct module_stack* stack)
93 modstack_config(struct module_stack* stack, const char* module_conf)
96 verbose(VERB_QUERY, "module config: \"%s\"", module_conf);
97 stack->num = count_modules(module_conf);
99 log_err("error: no modules specified");
102 if(stack->num > MAX_MODULE) {
103 log_err("error: too many modules (%d max %d)",
104 stack->num, MAX_MODULE);
107 stack->mod = (struct module_func_block**)calloc((size_t)
108 stack->num, sizeof(struct module_func_block*));
110 log_err("out of memory");
113 for(i=0; i<stack->num; i++) {
114 stack->mod[i] = module_factory(&module_conf);
117 snprintf(md, sizeof(md), "%s", module_conf);
118 if(strchr(md, ' ')) *(strchr(md, ' ')) = 0;
119 if(strchr(md, '\t')) *(strchr(md, '\t')) = 0;
120 log_err("Unknown value in module-config, module: '%s'."
121 " This module is not present (not compiled in),"
122 " See the list of linked modules with unbound -h",
130 /** The list of module names */
132 module_list_avail(void)
134 /* these are the modules available */
135 static const char* names[] = {
137 #ifdef WITH_PYTHONMODULE
156 /** func block get function type */
157 typedef struct module_func_block* (*fbgetfunctype)(void);
159 /** The list of module func blocks */
160 static fbgetfunctype*
161 module_funcs_avail(void)
163 static struct module_func_block* (*fb[])(void) = {
164 &dns64_get_funcblock,
165 #ifdef WITH_PYTHONMODULE
166 &pythonmod_get_funcblock,
169 &cachedb_get_funcblock,
172 &ipsecmod_get_funcblock,
175 &subnetmod_get_funcblock,
177 &respip_get_funcblock,
185 module_func_block* module_factory(const char** str)
188 const char* s = *str;
189 const char** names = module_list_avail();
190 fbgetfunctype* fb = module_funcs_avail();
191 while(*s && isspace((unsigned char)*s))
194 if(strncmp(names[i], s, strlen(names[i])) == 0) {
195 s += strlen(names[i]);
205 modstack_setup(struct module_stack* stack, const char* module_conf,
206 struct module_env* env)
210 modstack_desetup(stack, env);
211 /* fixed setup of the modules */
212 if(!modstack_config(stack, module_conf)) {
215 env->need_to_validate = 0; /* set by module init below */
216 for(i=0; i<stack->num; i++) {
217 verbose(VERB_OPS, "init module %d: %s",
218 i, stack->mod[i]->name);
219 fptr_ok(fptr_whitelist_mod_init(stack->mod[i]->init));
220 if(!(*stack->mod[i]->init)(env, i)) {
221 log_err("module init for module %s failed",
222 stack->mod[i]->name);
230 modstack_desetup(struct module_stack* stack, struct module_env* env)
233 for(i=0; i<stack->num; i++) {
234 fptr_ok(fptr_whitelist_mod_deinit(stack->mod[i]->deinit));
235 (*stack->mod[i]->deinit)(env, i);
243 modstack_find(struct module_stack* stack, const char* name)
246 for(i=0; i<stack->num; i++) {
247 if(strcmp(stack->mod[i]->name, name) == 0)
254 mod_get_mem(struct module_env* env, const char* name)
256 int m = modstack_find(&env->mesh->mods, name);
258 fptr_ok(fptr_whitelist_mod_get_mem(env->mesh->
259 mods.mod[m]->get_mem));
260 return (*env->mesh->mods.mod[m]->get_mem)(env, m);