npf.c revision 1.15 1 1.15 rmind /* $NetBSD: npf.c,v 1.15 2012/12/23 21:01:05 rmind Exp $ */
2 1.1 rmind
3 1.1 rmind /*-
4 1.6 rmind * Copyright (c) 2010-2012 The NetBSD Foundation, Inc.
5 1.1 rmind * All rights reserved.
6 1.1 rmind *
7 1.1 rmind * This material is based upon work partially supported by The
8 1.1 rmind * NetBSD Foundation under a contract with Mindaugas Rasiukevicius.
9 1.1 rmind *
10 1.1 rmind * Redistribution and use in source and binary forms, with or without
11 1.1 rmind * modification, are permitted provided that the following conditions
12 1.1 rmind * are met:
13 1.1 rmind * 1. Redistributions of source code must retain the above copyright
14 1.1 rmind * notice, this list of conditions and the following disclaimer.
15 1.1 rmind * 2. Redistributions in binary form must reproduce the above copyright
16 1.1 rmind * notice, this list of conditions and the following disclaimer in the
17 1.1 rmind * documentation and/or other materials provided with the distribution.
18 1.1 rmind *
19 1.1 rmind * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 1.1 rmind * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 1.1 rmind * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 1.1 rmind * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 1.1 rmind * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 1.1 rmind * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 1.1 rmind * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 1.1 rmind * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 1.1 rmind * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 1.1 rmind * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 1.1 rmind * POSSIBILITY OF SUCH DAMAGE.
30 1.1 rmind */
31 1.1 rmind
32 1.1 rmind #include <sys/cdefs.h>
33 1.15 rmind __KERNEL_RCSID(0, "$NetBSD: npf.c,v 1.15 2012/12/23 21:01:05 rmind Exp $");
34 1.1 rmind
35 1.1 rmind #include <sys/types.h>
36 1.1 rmind #include <netinet/in_systm.h>
37 1.1 rmind #include <netinet/in.h>
38 1.11 rmind #include <net/if.h>
39 1.1 rmind #include <prop/proplib.h>
40 1.1 rmind
41 1.1 rmind #include <stdlib.h>
42 1.1 rmind #include <string.h>
43 1.7 rmind #include <assert.h>
44 1.1 rmind #include <errno.h>
45 1.1 rmind #include <err.h>
46 1.1 rmind
47 1.1 rmind #define _NPF_PRIVATE
48 1.1 rmind #include "npf.h"
49 1.1 rmind
50 1.1 rmind struct nl_config {
51 1.1 rmind /* Rules, translations, tables, procedures. */
52 1.8 rmind prop_dictionary_t ncf_dict;
53 1.1 rmind prop_array_t ncf_rules_list;
54 1.1 rmind prop_array_t ncf_rproc_list;
55 1.1 rmind prop_array_t ncf_table_list;
56 1.1 rmind prop_array_t ncf_nat_list;
57 1.1 rmind /* Priority counters. */
58 1.1 rmind pri_t ncf_rule_pri;
59 1.1 rmind pri_t ncf_nat_pri;
60 1.11 rmind /* Debug information. */
61 1.11 rmind prop_dictionary_t ncf_debug;
62 1.7 rmind /* Error report. */
63 1.7 rmind prop_dictionary_t ncf_err;
64 1.4 rmind /* Custom file to externalise property-list. */
65 1.4 rmind const char * ncf_plist;
66 1.6 rmind bool ncf_flush;
67 1.1 rmind };
68 1.1 rmind
69 1.1 rmind struct nl_rule {
70 1.1 rmind prop_dictionary_t nrl_dict;
71 1.1 rmind };
72 1.1 rmind
73 1.1 rmind struct nl_rproc {
74 1.1 rmind prop_dictionary_t nrp_dict;
75 1.1 rmind };
76 1.1 rmind
77 1.1 rmind struct nl_table {
78 1.1 rmind prop_dictionary_t ntl_dict;
79 1.1 rmind };
80 1.1 rmind
81 1.13 rmind struct nl_ext {
82 1.13 rmind const char * nxt_name;
83 1.13 rmind prop_dictionary_t nxt_dict;
84 1.13 rmind };
85 1.13 rmind
86 1.1 rmind /*
87 1.1 rmind * CONFIGURATION INTERFACE.
88 1.1 rmind */
89 1.1 rmind
90 1.1 rmind nl_config_t *
91 1.1 rmind npf_config_create(void)
92 1.1 rmind {
93 1.1 rmind nl_config_t *ncf;
94 1.1 rmind
95 1.7 rmind ncf = calloc(1, sizeof(*ncf));
96 1.1 rmind if (ncf == NULL) {
97 1.1 rmind return NULL;
98 1.1 rmind }
99 1.1 rmind ncf->ncf_rules_list = prop_array_create();
100 1.1 rmind ncf->ncf_rproc_list = prop_array_create();
101 1.1 rmind ncf->ncf_table_list = prop_array_create();
102 1.1 rmind ncf->ncf_nat_list = prop_array_create();
103 1.1 rmind
104 1.1 rmind ncf->ncf_rule_pri = 1;
105 1.1 rmind ncf->ncf_nat_pri = 1;
106 1.1 rmind
107 1.4 rmind ncf->ncf_plist = NULL;
108 1.6 rmind ncf->ncf_flush = false;
109 1.4 rmind
110 1.1 rmind return ncf;
111 1.1 rmind }
112 1.1 rmind
113 1.1 rmind int
114 1.1 rmind npf_config_submit(nl_config_t *ncf, int fd)
115 1.1 rmind {
116 1.7 rmind const char *plist = ncf->ncf_plist;
117 1.1 rmind prop_dictionary_t npf_dict;
118 1.1 rmind int error = 0;
119 1.1 rmind
120 1.1 rmind npf_dict = prop_dictionary_create();
121 1.1 rmind if (npf_dict == NULL) {
122 1.1 rmind return ENOMEM;
123 1.1 rmind }
124 1.15 rmind prop_dictionary_set_uint32(npf_dict, "version", NPF_VERSION);
125 1.1 rmind prop_dictionary_set(npf_dict, "rules", ncf->ncf_rules_list);
126 1.1 rmind prop_dictionary_set(npf_dict, "rprocs", ncf->ncf_rproc_list);
127 1.1 rmind prop_dictionary_set(npf_dict, "tables", ncf->ncf_table_list);
128 1.1 rmind prop_dictionary_set(npf_dict, "translation", ncf->ncf_nat_list);
129 1.6 rmind prop_dictionary_set_bool(npf_dict, "flush", ncf->ncf_flush);
130 1.15 rmind if (ncf->ncf_debug) {
131 1.15 rmind prop_dictionary_set(npf_dict, "debug", ncf->ncf_debug);
132 1.15 rmind }
133 1.1 rmind
134 1.4 rmind if (plist) {
135 1.4 rmind if (!prop_dictionary_externalize_to_file(npf_dict, plist)) {
136 1.4 rmind error = errno;
137 1.4 rmind }
138 1.7 rmind prop_object_release(npf_dict);
139 1.7 rmind return error;
140 1.7 rmind }
141 1.7 rmind
142 1.7 rmind error = prop_dictionary_sendrecv_ioctl(npf_dict, fd,
143 1.7 rmind IOC_NPF_RELOAD, &ncf->ncf_err);
144 1.7 rmind if (error) {
145 1.7 rmind prop_object_release(npf_dict);
146 1.7 rmind assert(ncf->ncf_err == NULL);
147 1.7 rmind return error;
148 1.1 rmind }
149 1.7 rmind
150 1.7 rmind prop_dictionary_get_int32(ncf->ncf_err, "errno", &error);
151 1.1 rmind prop_object_release(npf_dict);
152 1.1 rmind return error;
153 1.1 rmind }
154 1.1 rmind
155 1.8 rmind nl_config_t *
156 1.8 rmind npf_config_retrieve(int fd, bool *active, bool *loaded)
157 1.8 rmind {
158 1.8 rmind prop_dictionary_t npf_dict;
159 1.8 rmind nl_config_t *ncf;
160 1.8 rmind int error;
161 1.8 rmind
162 1.8 rmind error = prop_dictionary_recv_ioctl(fd, IOC_NPF_GETCONF, &npf_dict);
163 1.8 rmind if (error) {
164 1.8 rmind return NULL;
165 1.8 rmind }
166 1.8 rmind ncf = calloc(1, sizeof(*ncf));
167 1.8 rmind if (ncf == NULL) {
168 1.8 rmind prop_object_release(npf_dict);
169 1.8 rmind return NULL;
170 1.8 rmind }
171 1.8 rmind ncf->ncf_dict = npf_dict;
172 1.8 rmind ncf->ncf_rules_list = prop_dictionary_get(npf_dict, "rules");
173 1.8 rmind ncf->ncf_rproc_list = prop_dictionary_get(npf_dict, "rprocs");
174 1.8 rmind ncf->ncf_table_list = prop_dictionary_get(npf_dict, "tables");
175 1.8 rmind ncf->ncf_nat_list = prop_dictionary_get(npf_dict, "translation");
176 1.8 rmind
177 1.8 rmind prop_dictionary_get_bool(npf_dict, "active", active);
178 1.8 rmind *loaded = (ncf->ncf_rules_list != NULL);
179 1.8 rmind return ncf;
180 1.8 rmind }
181 1.8 rmind
182 1.6 rmind int
183 1.6 rmind npf_config_flush(int fd)
184 1.6 rmind {
185 1.6 rmind nl_config_t *ncf;
186 1.6 rmind int error;
187 1.6 rmind
188 1.6 rmind ncf = npf_config_create();
189 1.6 rmind if (ncf == NULL) {
190 1.6 rmind return ENOMEM;
191 1.6 rmind }
192 1.6 rmind ncf->ncf_flush = true;
193 1.6 rmind error = npf_config_submit(ncf, fd);
194 1.6 rmind npf_config_destroy(ncf);
195 1.6 rmind return error;
196 1.6 rmind }
197 1.6 rmind
198 1.1 rmind void
199 1.7 rmind _npf_config_error(nl_config_t *ncf, nl_error_t *ne)
200 1.7 rmind {
201 1.7 rmind memset(ne, 0, sizeof(*ne));
202 1.7 rmind prop_dictionary_get_int32(ncf->ncf_err, "id", &ne->ne_id);
203 1.7 rmind prop_dictionary_get_cstring(ncf->ncf_err,
204 1.7 rmind "source-file", &ne->ne_source_file);
205 1.7 rmind prop_dictionary_get_uint32(ncf->ncf_err,
206 1.7 rmind "source-line", &ne->ne_source_line);
207 1.7 rmind prop_dictionary_get_int32(ncf->ncf_err,
208 1.7 rmind "ncode-error", &ne->ne_ncode_error);
209 1.7 rmind prop_dictionary_get_int32(ncf->ncf_err,
210 1.7 rmind "ncode-errat", &ne->ne_ncode_errat);
211 1.7 rmind }
212 1.7 rmind
213 1.7 rmind void
214 1.1 rmind npf_config_destroy(nl_config_t *ncf)
215 1.1 rmind {
216 1.1 rmind
217 1.14 rmind if (!ncf->ncf_dict) {
218 1.8 rmind prop_object_release(ncf->ncf_rules_list);
219 1.8 rmind prop_object_release(ncf->ncf_rproc_list);
220 1.8 rmind prop_object_release(ncf->ncf_table_list);
221 1.8 rmind prop_object_release(ncf->ncf_nat_list);
222 1.8 rmind }
223 1.7 rmind if (ncf->ncf_err) {
224 1.7 rmind prop_object_release(ncf->ncf_err);
225 1.7 rmind }
226 1.11 rmind if (ncf->ncf_debug) {
227 1.11 rmind prop_object_release(ncf->ncf_debug);
228 1.11 rmind }
229 1.1 rmind free(ncf);
230 1.1 rmind }
231 1.1 rmind
232 1.4 rmind void
233 1.4 rmind _npf_config_setsubmit(nl_config_t *ncf, const char *plist_file)
234 1.4 rmind {
235 1.4 rmind
236 1.4 rmind ncf->ncf_plist = plist_file;
237 1.4 rmind }
238 1.4 rmind
239 1.1 rmind static bool
240 1.1 rmind _npf_prop_array_lookup(prop_array_t array, const char *key, const char *name)
241 1.1 rmind {
242 1.1 rmind prop_dictionary_t dict;
243 1.1 rmind prop_object_iterator_t it;
244 1.1 rmind
245 1.1 rmind it = prop_array_iterator(array);
246 1.1 rmind while ((dict = prop_object_iterator_next(it)) != NULL) {
247 1.1 rmind const char *lname;
248 1.1 rmind prop_dictionary_get_cstring_nocopy(dict, key, &lname);
249 1.1 rmind if (strcmp(name, lname) == 0)
250 1.1 rmind break;
251 1.1 rmind }
252 1.1 rmind prop_object_iterator_release(it);
253 1.1 rmind return dict ? true : false;
254 1.1 rmind }
255 1.1 rmind
256 1.1 rmind /*
257 1.13 rmind * NPF EXTENSION INTERFACE.
258 1.13 rmind */
259 1.13 rmind
260 1.13 rmind nl_ext_t *
261 1.13 rmind npf_ext_construct(const char *name)
262 1.13 rmind {
263 1.13 rmind nl_ext_t *ext;
264 1.13 rmind
265 1.13 rmind ext = malloc(sizeof(*ext));
266 1.13 rmind if (ext == NULL) {
267 1.13 rmind return NULL;
268 1.13 rmind }
269 1.13 rmind ext->nxt_name = strdup(name);
270 1.13 rmind if (ext->nxt_name == NULL) {
271 1.13 rmind free(ext);
272 1.13 rmind return NULL;
273 1.13 rmind }
274 1.13 rmind ext->nxt_dict = prop_dictionary_create();
275 1.13 rmind
276 1.13 rmind return ext;
277 1.13 rmind }
278 1.13 rmind
279 1.13 rmind void
280 1.13 rmind npf_ext_param_u32(nl_ext_t *ext, const char *key, uint32_t val)
281 1.13 rmind {
282 1.13 rmind prop_dictionary_t extdict = ext->nxt_dict;
283 1.13 rmind prop_dictionary_set_uint32(extdict, key, val);
284 1.13 rmind }
285 1.13 rmind
286 1.13 rmind void
287 1.13 rmind npf_ext_param_bool(nl_ext_t *ext, const char *key, bool val)
288 1.13 rmind {
289 1.13 rmind prop_dictionary_t extdict = ext->nxt_dict;
290 1.13 rmind prop_dictionary_set_bool(extdict, key, val);
291 1.13 rmind }
292 1.13 rmind
293 1.13 rmind /*
294 1.1 rmind * RULE INTERFACE.
295 1.1 rmind */
296 1.1 rmind
297 1.1 rmind nl_rule_t *
298 1.1 rmind npf_rule_create(const char *name, uint32_t attr, u_int if_idx)
299 1.1 rmind {
300 1.1 rmind prop_dictionary_t rldict;
301 1.1 rmind nl_rule_t *rl;
302 1.1 rmind
303 1.5 christos rl = malloc(sizeof(*rl));
304 1.1 rmind if (rl == NULL) {
305 1.1 rmind return NULL;
306 1.1 rmind }
307 1.1 rmind rldict = prop_dictionary_create();
308 1.1 rmind if (rldict == NULL) {
309 1.1 rmind free(rl);
310 1.1 rmind return NULL;
311 1.1 rmind }
312 1.1 rmind if (name) {
313 1.1 rmind prop_dictionary_set_cstring(rldict, "name", name);
314 1.1 rmind }
315 1.1 rmind prop_dictionary_set_uint32(rldict, "attributes", attr);
316 1.1 rmind
317 1.1 rmind if (if_idx) {
318 1.1 rmind prop_dictionary_set_uint32(rldict, "interface", if_idx);
319 1.1 rmind }
320 1.1 rmind rl->nrl_dict = rldict;
321 1.1 rmind return rl;
322 1.1 rmind }
323 1.1 rmind
324 1.1 rmind int
325 1.1 rmind npf_rule_setcode(nl_rule_t *rl, int type, const void *code, size_t sz)
326 1.1 rmind {
327 1.1 rmind prop_dictionary_t rldict = rl->nrl_dict;
328 1.1 rmind prop_data_t cdata;
329 1.1 rmind
330 1.1 rmind if (type != NPF_CODE_NCODE) {
331 1.1 rmind return ENOTSUP;
332 1.1 rmind }
333 1.1 rmind cdata = prop_data_create_data(code, sz);
334 1.1 rmind if (cdata == NULL) {
335 1.1 rmind return ENOMEM;
336 1.1 rmind }
337 1.1 rmind prop_dictionary_set(rldict, "ncode", cdata);
338 1.1 rmind prop_object_release(cdata);
339 1.1 rmind return 0;
340 1.1 rmind }
341 1.1 rmind
342 1.1 rmind int
343 1.1 rmind npf_rule_setproc(nl_config_t *ncf, nl_rule_t *rl, const char *name)
344 1.1 rmind {
345 1.1 rmind prop_dictionary_t rldict = rl->nrl_dict;
346 1.1 rmind
347 1.1 rmind if (!npf_rproc_exists_p(ncf, name)) {
348 1.1 rmind return ENOENT;
349 1.1 rmind }
350 1.1 rmind prop_dictionary_set_cstring(rldict, "rproc", name);
351 1.1 rmind return 0;
352 1.1 rmind }
353 1.1 rmind
354 1.1 rmind bool
355 1.1 rmind npf_rule_exists_p(nl_config_t *ncf, const char *name)
356 1.1 rmind {
357 1.1 rmind
358 1.1 rmind return _npf_prop_array_lookup(ncf->ncf_rules_list, "name", name);
359 1.1 rmind }
360 1.1 rmind
361 1.1 rmind int
362 1.1 rmind npf_rule_insert(nl_config_t *ncf, nl_rule_t *parent, nl_rule_t *rl, pri_t pri)
363 1.1 rmind {
364 1.1 rmind prop_dictionary_t rldict = rl->nrl_dict;
365 1.1 rmind prop_array_t rlset;
366 1.1 rmind
367 1.1 rmind if (pri == NPF_PRI_NEXT) {
368 1.1 rmind pri = ncf->ncf_rule_pri++;
369 1.1 rmind } else if (ncf) {
370 1.1 rmind ncf->ncf_rule_pri = pri + 1;
371 1.1 rmind }
372 1.1 rmind prop_dictionary_set_int32(rldict, "priority", pri);
373 1.1 rmind
374 1.1 rmind if (parent) {
375 1.1 rmind prop_dictionary_t pdict = parent->nrl_dict;
376 1.1 rmind rlset = prop_dictionary_get(pdict, "subrules");
377 1.1 rmind if (rlset == NULL) {
378 1.1 rmind rlset = prop_array_create();
379 1.1 rmind prop_dictionary_set(pdict, "subrules", rlset);
380 1.1 rmind prop_object_release(rlset);
381 1.1 rmind }
382 1.1 rmind } else {
383 1.1 rmind rlset = ncf->ncf_rules_list;
384 1.1 rmind }
385 1.1 rmind prop_array_add(rlset, rldict);
386 1.1 rmind return 0;
387 1.1 rmind }
388 1.1 rmind
389 1.8 rmind static int
390 1.8 rmind _npf_rule_foreach1(prop_array_t rules, unsigned nlevel, nl_rule_callback_t func)
391 1.8 rmind {
392 1.8 rmind prop_dictionary_t rldict;
393 1.8 rmind prop_object_iterator_t it;
394 1.8 rmind
395 1.8 rmind if (!rules || prop_object_type(rules) != PROP_TYPE_ARRAY) {
396 1.8 rmind return ENOENT;
397 1.8 rmind }
398 1.8 rmind it = prop_array_iterator(rules);
399 1.8 rmind if (it == NULL) {
400 1.8 rmind return ENOMEM;
401 1.8 rmind }
402 1.8 rmind while ((rldict = prop_object_iterator_next(it)) != NULL) {
403 1.8 rmind prop_array_t subrules;
404 1.8 rmind nl_rule_t nrl;
405 1.8 rmind
406 1.8 rmind nrl.nrl_dict = rldict;
407 1.8 rmind (*func)(&nrl, nlevel);
408 1.8 rmind
409 1.8 rmind subrules = prop_dictionary_get(rldict, "subrules");
410 1.14 rmind if (!subrules) {
411 1.14 rmind continue;
412 1.14 rmind }
413 1.8 rmind (void)_npf_rule_foreach1(subrules, nlevel + 1, func);
414 1.13 rmind prop_object_release(subrules);
415 1.8 rmind }
416 1.8 rmind prop_object_iterator_release(it);
417 1.8 rmind return 0;
418 1.8 rmind }
419 1.8 rmind
420 1.8 rmind int
421 1.8 rmind _npf_rule_foreach(nl_config_t *ncf, nl_rule_callback_t func)
422 1.8 rmind {
423 1.8 rmind
424 1.8 rmind return _npf_rule_foreach1(ncf->ncf_rules_list, 0, func);
425 1.8 rmind }
426 1.8 rmind
427 1.8 rmind pri_t
428 1.8 rmind _npf_rule_getinfo(nl_rule_t *nrl, const char **rname, uint32_t *attr,
429 1.8 rmind u_int *if_idx)
430 1.8 rmind {
431 1.8 rmind prop_dictionary_t rldict = nrl->nrl_dict;
432 1.8 rmind pri_t prio;
433 1.8 rmind
434 1.8 rmind prop_dictionary_get_cstring_nocopy(rldict, "name", rname);
435 1.8 rmind prop_dictionary_get_uint32(rldict, "attributes", attr);
436 1.8 rmind prop_dictionary_get_int32(rldict, "priority", &prio);
437 1.8 rmind prop_dictionary_get_uint32(rldict, "interface", if_idx);
438 1.8 rmind return prio;
439 1.8 rmind }
440 1.8 rmind
441 1.8 rmind const void *
442 1.8 rmind _npf_rule_ncode(nl_rule_t *nrl, size_t *size)
443 1.8 rmind {
444 1.8 rmind prop_dictionary_t rldict = nrl->nrl_dict;
445 1.8 rmind prop_object_t obj = prop_dictionary_get(rldict, "ncode");
446 1.8 rmind *size = prop_data_size(obj);
447 1.8 rmind return prop_data_data_nocopy(obj);
448 1.8 rmind }
449 1.8 rmind
450 1.8 rmind const char *
451 1.8 rmind _npf_rule_rproc(nl_rule_t *nrl)
452 1.8 rmind {
453 1.8 rmind prop_dictionary_t rldict = nrl->nrl_dict;
454 1.8 rmind const char *rpname = NULL;
455 1.8 rmind
456 1.8 rmind prop_dictionary_get_cstring_nocopy(rldict, "rproc", &rpname);
457 1.8 rmind return rpname;
458 1.8 rmind }
459 1.8 rmind
460 1.1 rmind void
461 1.1 rmind npf_rule_destroy(nl_rule_t *rl)
462 1.1 rmind {
463 1.1 rmind
464 1.1 rmind prop_object_release(rl->nrl_dict);
465 1.1 rmind free(rl);
466 1.1 rmind }
467 1.1 rmind
468 1.1 rmind /*
469 1.1 rmind * RULE PROCEDURE INTERFACE.
470 1.1 rmind */
471 1.1 rmind
472 1.1 rmind nl_rproc_t *
473 1.1 rmind npf_rproc_create(const char *name)
474 1.1 rmind {
475 1.1 rmind prop_dictionary_t rpdict;
476 1.13 rmind prop_array_t extcalls;
477 1.1 rmind nl_rproc_t *nrp;
478 1.1 rmind
479 1.1 rmind nrp = malloc(sizeof(nl_rproc_t));
480 1.1 rmind if (nrp == NULL) {
481 1.1 rmind return NULL;
482 1.1 rmind }
483 1.1 rmind rpdict = prop_dictionary_create();
484 1.1 rmind if (rpdict == NULL) {
485 1.1 rmind free(nrp);
486 1.1 rmind return NULL;
487 1.1 rmind }
488 1.1 rmind prop_dictionary_set_cstring(rpdict, "name", name);
489 1.13 rmind
490 1.13 rmind extcalls = prop_array_create();
491 1.13 rmind if (extcalls == NULL) {
492 1.13 rmind prop_object_release(rpdict);
493 1.13 rmind free(nrp);
494 1.13 rmind return NULL;
495 1.13 rmind }
496 1.13 rmind prop_dictionary_set(rpdict, "extcalls", extcalls);
497 1.13 rmind prop_object_release(extcalls);
498 1.13 rmind
499 1.1 rmind nrp->nrp_dict = rpdict;
500 1.1 rmind return nrp;
501 1.1 rmind }
502 1.1 rmind
503 1.1 rmind int
504 1.13 rmind npf_rproc_extcall(nl_rproc_t *rp, nl_ext_t *ext)
505 1.1 rmind {
506 1.1 rmind prop_dictionary_t rpdict = rp->nrp_dict;
507 1.13 rmind prop_dictionary_t extdict = ext->nxt_dict;
508 1.13 rmind prop_array_t extcalls;
509 1.1 rmind
510 1.13 rmind extcalls = prop_dictionary_get(rpdict, "extcalls");
511 1.13 rmind if (_npf_prop_array_lookup(extcalls, "name", ext->nxt_name)) {
512 1.13 rmind return EEXIST;
513 1.13 rmind }
514 1.13 rmind prop_dictionary_set_cstring(extdict, "name", ext->nxt_name);
515 1.13 rmind prop_array_add(extcalls, extdict);
516 1.1 rmind return 0;
517 1.1 rmind }
518 1.1 rmind
519 1.13 rmind bool
520 1.13 rmind npf_rproc_exists_p(nl_config_t *ncf, const char *name)
521 1.1 rmind {
522 1.1 rmind
523 1.13 rmind return _npf_prop_array_lookup(ncf->ncf_rproc_list, "name", name);
524 1.1 rmind }
525 1.1 rmind
526 1.1 rmind int
527 1.1 rmind npf_rproc_insert(nl_config_t *ncf, nl_rproc_t *rp)
528 1.1 rmind {
529 1.1 rmind prop_dictionary_t rpdict = rp->nrp_dict;
530 1.1 rmind const char *name;
531 1.1 rmind
532 1.1 rmind if (!prop_dictionary_get_cstring_nocopy(rpdict, "name", &name)) {
533 1.1 rmind return EINVAL;
534 1.1 rmind }
535 1.1 rmind if (npf_rproc_exists_p(ncf, name)) {
536 1.1 rmind return EEXIST;
537 1.1 rmind }
538 1.1 rmind prop_array_add(ncf->ncf_rproc_list, rpdict);
539 1.1 rmind return 0;
540 1.1 rmind }
541 1.1 rmind
542 1.1 rmind /*
543 1.1 rmind * TRANSLATION INTERFACE.
544 1.1 rmind */
545 1.1 rmind
546 1.1 rmind nl_nat_t *
547 1.5 christos npf_nat_create(int type, u_int flags, u_int if_idx,
548 1.1 rmind npf_addr_t *addr, int af, in_port_t port)
549 1.1 rmind {
550 1.1 rmind nl_rule_t *rl;
551 1.1 rmind prop_dictionary_t rldict;
552 1.1 rmind prop_data_t addrdat;
553 1.1 rmind uint32_t attr;
554 1.1 rmind size_t sz;
555 1.1 rmind
556 1.1 rmind if (af == AF_INET) {
557 1.1 rmind sz = sizeof(struct in_addr);
558 1.1 rmind } else if (af == AF_INET6) {
559 1.1 rmind sz = sizeof(struct in6_addr);
560 1.1 rmind } else {
561 1.1 rmind return NULL;
562 1.1 rmind }
563 1.1 rmind
564 1.1 rmind attr = NPF_RULE_PASS | NPF_RULE_FINAL |
565 1.2 rmind (type == NPF_NATOUT ? NPF_RULE_OUT : NPF_RULE_IN);
566 1.1 rmind
567 1.1 rmind /* Create a rule for NAT policy. Next, will add translation data. */
568 1.1 rmind rl = npf_rule_create(NULL, attr, if_idx);
569 1.1 rmind if (rl == NULL) {
570 1.1 rmind return NULL;
571 1.1 rmind }
572 1.1 rmind rldict = rl->nrl_dict;
573 1.1 rmind
574 1.1 rmind /* Translation type and flags. */
575 1.1 rmind prop_dictionary_set_int32(rldict, "type", type);
576 1.1 rmind prop_dictionary_set_uint32(rldict, "flags", flags);
577 1.1 rmind
578 1.1 rmind /* Translation IP. */
579 1.1 rmind addrdat = prop_data_create_data(addr, sz);
580 1.1 rmind if (addrdat == NULL) {
581 1.1 rmind npf_rule_destroy(rl);
582 1.1 rmind return NULL;
583 1.1 rmind }
584 1.1 rmind prop_dictionary_set(rldict, "translation-ip", addrdat);
585 1.1 rmind prop_object_release(addrdat);
586 1.1 rmind
587 1.1 rmind /* Translation port (for redirect case). */
588 1.1 rmind prop_dictionary_set_uint16(rldict, "translation-port", port);
589 1.1 rmind
590 1.1 rmind return (nl_nat_t *)rl;
591 1.1 rmind }
592 1.1 rmind
593 1.1 rmind int
594 1.1 rmind npf_nat_insert(nl_config_t *ncf, nl_nat_t *nt, pri_t pri)
595 1.1 rmind {
596 1.1 rmind prop_dictionary_t rldict = nt->nrl_dict;
597 1.1 rmind
598 1.1 rmind if (pri == NPF_PRI_NEXT) {
599 1.1 rmind pri = ncf->ncf_nat_pri++;
600 1.1 rmind } else {
601 1.1 rmind ncf->ncf_nat_pri = pri + 1;
602 1.1 rmind }
603 1.1 rmind prop_dictionary_set_int32(rldict, "priority", pri);
604 1.1 rmind prop_array_add(ncf->ncf_nat_list, rldict);
605 1.1 rmind return 0;
606 1.1 rmind }
607 1.1 rmind
608 1.9 rmind int
609 1.9 rmind _npf_nat_foreach(nl_config_t *ncf, nl_rule_callback_t func)
610 1.9 rmind {
611 1.9 rmind
612 1.9 rmind return _npf_rule_foreach1(ncf->ncf_nat_list, 0, func);
613 1.9 rmind }
614 1.9 rmind
615 1.9 rmind void
616 1.9 rmind _npf_nat_getinfo(nl_nat_t *nt, int *type, u_int *flags, npf_addr_t *addr,
617 1.9 rmind size_t *alen, in_port_t *port)
618 1.9 rmind {
619 1.9 rmind prop_dictionary_t rldict = nt->nrl_dict;
620 1.9 rmind
621 1.9 rmind prop_dictionary_get_int32(rldict, "type", type);
622 1.9 rmind prop_dictionary_get_uint32(rldict, "flags", flags);
623 1.9 rmind
624 1.9 rmind prop_object_t obj = prop_dictionary_get(rldict, "translation-ip");
625 1.9 rmind *alen = prop_data_size(obj);
626 1.9 rmind memcpy(addr, prop_data_data_nocopy(obj), *alen);
627 1.9 rmind
628 1.9 rmind prop_dictionary_get_uint16(rldict, "translation-port", port);
629 1.9 rmind }
630 1.9 rmind
631 1.1 rmind /*
632 1.1 rmind * TABLE INTERFACE.
633 1.1 rmind */
634 1.1 rmind
635 1.1 rmind nl_table_t *
636 1.5 christos npf_table_create(u_int id, int type)
637 1.1 rmind {
638 1.1 rmind prop_dictionary_t tldict;
639 1.1 rmind prop_array_t tblents;
640 1.1 rmind nl_table_t *tl;
641 1.1 rmind
642 1.5 christos tl = malloc(sizeof(*tl));
643 1.1 rmind if (tl == NULL) {
644 1.1 rmind return NULL;
645 1.1 rmind }
646 1.1 rmind tldict = prop_dictionary_create();
647 1.1 rmind if (tldict == NULL) {
648 1.1 rmind free(tl);
649 1.1 rmind return NULL;
650 1.1 rmind }
651 1.1 rmind prop_dictionary_set_uint32(tldict, "id", id);
652 1.1 rmind prop_dictionary_set_int32(tldict, "type", type);
653 1.1 rmind
654 1.1 rmind tblents = prop_array_create();
655 1.1 rmind if (tblents == NULL) {
656 1.1 rmind prop_object_release(tldict);
657 1.1 rmind free(tl);
658 1.1 rmind return NULL;
659 1.1 rmind }
660 1.1 rmind prop_dictionary_set(tldict, "entries", tblents);
661 1.1 rmind prop_object_release(tblents);
662 1.1 rmind
663 1.1 rmind tl->ntl_dict = tldict;
664 1.1 rmind return tl;
665 1.1 rmind }
666 1.1 rmind
667 1.1 rmind int
668 1.15 rmind npf_table_add_entry(nl_table_t *tl, int af, const npf_addr_t *addr,
669 1.15 rmind const npf_netmask_t mask)
670 1.1 rmind {
671 1.1 rmind prop_dictionary_t tldict = tl->ntl_dict, entdict;
672 1.1 rmind prop_array_t tblents;
673 1.3 zoltan prop_data_t addrdata;
674 1.15 rmind unsigned alen;
675 1.1 rmind
676 1.1 rmind /* Create the table entry. */
677 1.1 rmind entdict = prop_dictionary_create();
678 1.10 rmind if (entdict == NULL) {
679 1.1 rmind return ENOMEM;
680 1.1 rmind }
681 1.15 rmind
682 1.15 rmind switch (af) {
683 1.15 rmind case AF_INET:
684 1.15 rmind alen = sizeof(struct in_addr);
685 1.15 rmind break;
686 1.15 rmind case AF_INET6:
687 1.15 rmind alen = sizeof(struct in6_addr);
688 1.15 rmind break;
689 1.15 rmind default:
690 1.15 rmind return EINVAL;
691 1.15 rmind }
692 1.15 rmind
693 1.10 rmind addrdata = prop_data_create_data(addr, alen);
694 1.3 zoltan prop_dictionary_set(entdict, "addr", addrdata);
695 1.3 zoltan prop_dictionary_set_uint8(entdict, "mask", mask);
696 1.3 zoltan prop_object_release(addrdata);
697 1.1 rmind
698 1.1 rmind tblents = prop_dictionary_get(tldict, "entries");
699 1.1 rmind prop_array_add(tblents, entdict);
700 1.1 rmind prop_object_release(entdict);
701 1.1 rmind return 0;
702 1.1 rmind }
703 1.1 rmind
704 1.1 rmind bool
705 1.1 rmind npf_table_exists_p(nl_config_t *ncf, u_int tid)
706 1.1 rmind {
707 1.1 rmind prop_dictionary_t tldict;
708 1.1 rmind prop_object_iterator_t it;
709 1.1 rmind
710 1.1 rmind it = prop_array_iterator(ncf->ncf_table_list);
711 1.1 rmind while ((tldict = prop_object_iterator_next(it)) != NULL) {
712 1.1 rmind u_int i;
713 1.1 rmind if (prop_dictionary_get_uint32(tldict, "id", &i) && tid == i)
714 1.1 rmind break;
715 1.1 rmind }
716 1.1 rmind prop_object_iterator_release(it);
717 1.1 rmind return tldict ? true : false;
718 1.1 rmind }
719 1.1 rmind
720 1.1 rmind int
721 1.1 rmind npf_table_insert(nl_config_t *ncf, nl_table_t *tl)
722 1.1 rmind {
723 1.1 rmind prop_dictionary_t tldict = tl->ntl_dict;
724 1.1 rmind u_int tid;
725 1.1 rmind
726 1.1 rmind if (!prop_dictionary_get_uint32(tldict, "id", &tid)) {
727 1.1 rmind return EINVAL;
728 1.1 rmind }
729 1.1 rmind if (npf_table_exists_p(ncf, tid)) {
730 1.1 rmind return EEXIST;
731 1.1 rmind }
732 1.1 rmind prop_array_add(ncf->ncf_table_list, tldict);
733 1.1 rmind return 0;
734 1.1 rmind }
735 1.1 rmind
736 1.1 rmind void
737 1.1 rmind npf_table_destroy(nl_table_t *tl)
738 1.1 rmind {
739 1.1 rmind
740 1.1 rmind prop_object_release(tl->ntl_dict);
741 1.1 rmind free(tl);
742 1.1 rmind }
743 1.1 rmind
744 1.9 rmind void
745 1.9 rmind _npf_table_foreach(nl_config_t *ncf, nl_table_callback_t func)
746 1.9 rmind {
747 1.9 rmind prop_dictionary_t tldict;
748 1.9 rmind prop_object_iterator_t it;
749 1.9 rmind
750 1.9 rmind it = prop_array_iterator(ncf->ncf_table_list);
751 1.9 rmind while ((tldict = prop_object_iterator_next(it)) != NULL) {
752 1.9 rmind u_int id;
753 1.9 rmind int type;
754 1.9 rmind
755 1.9 rmind prop_dictionary_get_uint32(tldict, "id", &id);
756 1.9 rmind prop_dictionary_get_int32(tldict, "type", &type);
757 1.9 rmind (*func)(id, type);
758 1.9 rmind }
759 1.9 rmind prop_object_iterator_release(it);
760 1.9 rmind }
761 1.9 rmind
762 1.1 rmind /*
763 1.1 rmind * MISC.
764 1.1 rmind */
765 1.1 rmind
766 1.1 rmind int
767 1.5 christos npf_update_rule(int fd, const char *rname __unused, nl_rule_t *rl)
768 1.1 rmind {
769 1.7 rmind prop_dictionary_t rldict = rl->nrl_dict, errdict = NULL;
770 1.1 rmind int error;
771 1.1 rmind
772 1.7 rmind error = prop_dictionary_sendrecv_ioctl(rldict, fd,
773 1.7 rmind IOC_NPF_UPDATE_RULE, &errdict);
774 1.7 rmind if (errdict) {
775 1.7 rmind prop_object_release(errdict);
776 1.7 rmind }
777 1.1 rmind return error;
778 1.1 rmind }
779 1.1 rmind
780 1.1 rmind int
781 1.1 rmind npf_sessions_recv(int fd, const char *fpath)
782 1.1 rmind {
783 1.1 rmind prop_dictionary_t sdict;
784 1.1 rmind int error;
785 1.1 rmind
786 1.1 rmind error = prop_dictionary_recv_ioctl(fd, IOC_NPF_SESSIONS_SAVE, &sdict);
787 1.1 rmind if (error) {
788 1.1 rmind return error;
789 1.1 rmind }
790 1.1 rmind if (!prop_dictionary_externalize_to_file(sdict, fpath)) {
791 1.1 rmind error = errno;
792 1.1 rmind }
793 1.1 rmind prop_object_release(sdict);
794 1.1 rmind return error;
795 1.1 rmind }
796 1.1 rmind
797 1.1 rmind int
798 1.1 rmind npf_sessions_send(int fd, const char *fpath)
799 1.1 rmind {
800 1.1 rmind prop_dictionary_t sdict;
801 1.1 rmind int error;
802 1.1 rmind
803 1.1 rmind if (fpath) {
804 1.1 rmind sdict = prop_dictionary_internalize_from_file(fpath);
805 1.1 rmind if (sdict == NULL) {
806 1.1 rmind return errno;
807 1.1 rmind }
808 1.1 rmind } else {
809 1.1 rmind /* Empty: will flush the sessions. */
810 1.1 rmind prop_array_t selist = prop_array_create();
811 1.1 rmind sdict = prop_dictionary_create();
812 1.1 rmind prop_dictionary_set(sdict, "session-list", selist);
813 1.1 rmind prop_object_release(selist);
814 1.1 rmind }
815 1.1 rmind error = prop_dictionary_send_ioctl(sdict, fd, IOC_NPF_SESSIONS_LOAD);
816 1.1 rmind prop_object_release(sdict);
817 1.1 rmind return error;
818 1.1 rmind }
819 1.11 rmind
820 1.11 rmind static prop_dictionary_t
821 1.11 rmind _npf_debug_initonce(nl_config_t *ncf)
822 1.11 rmind {
823 1.11 rmind if (!ncf->ncf_debug) {
824 1.11 rmind prop_array_t iflist = prop_array_create();
825 1.11 rmind ncf->ncf_debug = prop_dictionary_create();
826 1.11 rmind prop_dictionary_set(ncf->ncf_debug, "interfaces", iflist);
827 1.11 rmind prop_object_release(iflist);
828 1.11 rmind }
829 1.11 rmind return ncf->ncf_debug;
830 1.11 rmind }
831 1.11 rmind
832 1.11 rmind void
833 1.11 rmind _npf_debug_addif(nl_config_t *ncf, struct ifaddrs *ifa, u_int if_idx)
834 1.11 rmind {
835 1.11 rmind prop_dictionary_t ifdict, dbg = _npf_debug_initonce(ncf);
836 1.11 rmind prop_array_t iflist = prop_dictionary_get(dbg, "interfaces");
837 1.11 rmind
838 1.11 rmind if (_npf_prop_array_lookup(iflist, "name", ifa->ifa_name)) {
839 1.11 rmind return;
840 1.11 rmind }
841 1.11 rmind
842 1.11 rmind ifdict = prop_dictionary_create();
843 1.11 rmind prop_dictionary_set_cstring(ifdict, "name", ifa->ifa_name);
844 1.11 rmind prop_dictionary_set_uint32(ifdict, "flags", ifa->ifa_flags);
845 1.11 rmind if (!if_idx) {
846 1.11 rmind if_idx = if_nametoindex(ifa->ifa_name);
847 1.11 rmind }
848 1.11 rmind prop_dictionary_set_uint32(ifdict, "idx", if_idx);
849 1.11 rmind
850 1.11 rmind const struct sockaddr *sa = ifa->ifa_addr;
851 1.11 rmind npf_addr_t addr;
852 1.11 rmind size_t alen = 0;
853 1.11 rmind
854 1.11 rmind switch (sa ? sa->sa_family : -1) {
855 1.11 rmind case AF_INET: {
856 1.11 rmind const struct sockaddr_in *sin = (const void *)sa;
857 1.11 rmind alen = sizeof(sin->sin_addr);
858 1.11 rmind memcpy(&addr, &sin->sin_addr, alen);
859 1.11 rmind break;
860 1.11 rmind }
861 1.11 rmind case AF_INET6: {
862 1.11 rmind const struct sockaddr_in6 *sin6 = (const void *)sa;
863 1.11 rmind alen = sizeof(sin6->sin6_addr);
864 1.11 rmind memcpy(&addr, &sin6->sin6_addr, alen);
865 1.11 rmind break;
866 1.11 rmind }
867 1.11 rmind default:
868 1.11 rmind break;
869 1.11 rmind }
870 1.11 rmind
871 1.11 rmind if (alen) {
872 1.11 rmind prop_data_t addrdata = prop_data_create_data(&addr, alen);
873 1.11 rmind prop_dictionary_set(ifdict, "addr", addrdata);
874 1.11 rmind prop_object_release(addrdata);
875 1.11 rmind }
876 1.11 rmind prop_array_add(iflist, ifdict);
877 1.11 rmind prop_object_release(ifdict);
878 1.11 rmind }
879