npf_extmod.c revision 1.3.2.2 1 1.3.2.2 riz /* $NetBSD: npf_extmod.c,v 1.3.2.2 2012/11/18 22:38:28 riz Exp $ */
2 1.3.2.2 riz
3 1.3.2.2 riz /*-
4 1.3.2.2 riz * Copyright (c) 2012 The NetBSD Foundation, Inc.
5 1.3.2.2 riz * All rights reserved.
6 1.3.2.2 riz *
7 1.3.2.2 riz * This code is derived from software contributed to The NetBSD Foundation
8 1.3.2.2 riz * by Mindaugas Rasiukevicius.
9 1.3.2.2 riz *
10 1.3.2.2 riz * Redistribution and use in source and binary forms, with or without
11 1.3.2.2 riz * modification, are permitted provided that the following conditions
12 1.3.2.2 riz * are met:
13 1.3.2.2 riz * 1. Redistributions of source code must retain the above copyright
14 1.3.2.2 riz * notice, this list of conditions and the following disclaimer.
15 1.3.2.2 riz * 2. Redistributions in binary form must reproduce the above copyright
16 1.3.2.2 riz * notice, this list of conditions and the following disclaimer in the
17 1.3.2.2 riz * documentation and/or other materials provided with the distribution.
18 1.3.2.2 riz *
19 1.3.2.2 riz * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 1.3.2.2 riz * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 1.3.2.2 riz * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 1.3.2.2 riz * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 1.3.2.2 riz * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 1.3.2.2 riz * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 1.3.2.2 riz * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 1.3.2.2 riz * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 1.3.2.2 riz * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 1.3.2.2 riz * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 1.3.2.2 riz * POSSIBILITY OF SUCH DAMAGE.
30 1.3.2.2 riz */
31 1.3.2.2 riz
32 1.3.2.2 riz /*
33 1.3.2.2 riz * npfctl(8) extension loading interface.
34 1.3.2.2 riz */
35 1.3.2.2 riz
36 1.3.2.2 riz #include <sys/cdefs.h>
37 1.3.2.2 riz __RCSID("$NetBSD: npf_extmod.c,v 1.3.2.2 2012/11/18 22:38:28 riz Exp $");
38 1.3.2.2 riz
39 1.3.2.2 riz #include <stdlib.h>
40 1.3.2.2 riz #include <inttypes.h>
41 1.3.2.2 riz #include <string.h>
42 1.3.2.2 riz #include <err.h>
43 1.3.2.2 riz #include <dlfcn.h>
44 1.3.2.2 riz
45 1.3.2.2 riz #include "npfctl.h"
46 1.3.2.2 riz
47 1.3.2.2 riz struct npf_extmod {
48 1.3.2.2 riz char * name;
49 1.3.2.2 riz npfext_initfunc_t init;
50 1.3.2.2 riz npfext_consfunc_t cons;
51 1.3.2.2 riz npfext_paramfunc_t param;
52 1.3.2.2 riz struct npf_extmod * next;
53 1.3.2.2 riz };
54 1.3.2.2 riz
55 1.3.2.2 riz static npf_extmod_t * npf_extmod_list;
56 1.3.2.2 riz
57 1.3.2.2 riz static void *
58 1.3.2.2 riz npf_extmod_sym(void *handle, const char *name, const char *func)
59 1.3.2.2 riz {
60 1.3.2.2 riz char buf[64];
61 1.3.2.2 riz void *sym;
62 1.3.2.2 riz
63 1.3.2.2 riz snprintf(buf, sizeof(buf), "npfext_%s_%s", name, func);
64 1.3.2.2 riz sym = dlsym(handle, buf);
65 1.3.2.2 riz if (sym == NULL) {
66 1.3.2.2 riz errx(EXIT_FAILURE, "dlsym: %s", dlerror());
67 1.3.2.2 riz }
68 1.3.2.2 riz return sym;
69 1.3.2.2 riz }
70 1.3.2.2 riz
71 1.3.2.2 riz static npf_extmod_t *
72 1.3.2.2 riz npf_extmod_load(const char *name)
73 1.3.2.2 riz {
74 1.3.2.2 riz npf_extmod_t *ext;
75 1.3.2.2 riz void *handle;
76 1.3.2.2 riz char extlib[PATH_MAX];
77 1.3.2.2 riz
78 1.3.2.2 riz snprintf(extlib, sizeof(extlib), "/usr/lib/npf/ext_%s.so", name);
79 1.3.2.2 riz handle = dlopen(extlib, RTLD_LAZY | RTLD_LOCAL);
80 1.3.2.2 riz if (handle == NULL) {
81 1.3.2.2 riz errx(EXIT_FAILURE, "dlopen: %s", dlerror());
82 1.3.2.2 riz }
83 1.3.2.2 riz
84 1.3.2.2 riz ext = zalloc(sizeof(npf_extmod_t));
85 1.3.2.2 riz ext->name = xstrdup(name);
86 1.3.2.2 riz ext->init = npf_extmod_sym(handle, name, "init");
87 1.3.2.2 riz ext->cons = npf_extmod_sym(handle, name, "construct");
88 1.3.2.2 riz ext->param = npf_extmod_sym(handle, name, "param");
89 1.3.2.2 riz
90 1.3.2.2 riz /* Initialise the module. */
91 1.3.2.2 riz if (ext->init() != 0) {
92 1.3.2.2 riz free(ext);
93 1.3.2.2 riz return NULL;
94 1.3.2.2 riz }
95 1.3.2.2 riz
96 1.3.2.2 riz ext->next = npf_extmod_list;
97 1.3.2.2 riz npf_extmod_list = ext;
98 1.3.2.2 riz return ext;
99 1.3.2.2 riz }
100 1.3.2.2 riz
101 1.3.2.2 riz npf_extmod_t *
102 1.3.2.2 riz npf_extmod_get(const char *name, nl_ext_t **extcall)
103 1.3.2.2 riz {
104 1.3.2.2 riz npf_extmod_t *extmod = npf_extmod_list;
105 1.3.2.2 riz
106 1.3.2.2 riz while (extmod) {
107 1.3.2.2 riz if ((strcmp(extmod->name, name) == 0) &&
108 1.3.2.2 riz (*extcall = extmod->cons(name)) != NULL) {
109 1.3.2.2 riz return extmod;
110 1.3.2.2 riz }
111 1.3.2.2 riz extmod = extmod->next;
112 1.3.2.2 riz }
113 1.3.2.2 riz
114 1.3.2.2 riz extmod = npf_extmod_load(name);
115 1.3.2.2 riz if (extmod && (*extcall = extmod->cons(name)) != NULL) {
116 1.3.2.2 riz return extmod;
117 1.3.2.2 riz }
118 1.3.2.2 riz
119 1.3.2.2 riz return NULL;
120 1.3.2.2 riz }
121 1.3.2.2 riz
122 1.3.2.2 riz int
123 1.3.2.2 riz npf_extmod_param(npf_extmod_t *extmod, nl_ext_t *ext,
124 1.3.2.2 riz const char *param, const char *val)
125 1.3.2.2 riz {
126 1.3.2.2 riz return extmod->param(ext, param, val);
127 1.3.2.2 riz }
128