ip_pool.c revision 1.2.4.4 1 1.2.4.4 yamt /* $NetBSD: ip_pool.c,v 1.2.4.4 2014/05/22 11:40:58 yamt Exp $ */
2 1.2.4.2 yamt
3 1.2.4.2 yamt /*
4 1.2.4.2 yamt * Copyright (C) 2012 by Darren Reed.
5 1.2.4.2 yamt *
6 1.2.4.2 yamt * See the IPFILTER.LICENCE file for details on licencing.
7 1.2.4.2 yamt */
8 1.2.4.2 yamt #if defined(KERNEL) || defined(_KERNEL)
9 1.2.4.2 yamt # undef KERNEL
10 1.2.4.2 yamt # undef _KERNEL
11 1.2.4.2 yamt # define KERNEL 1
12 1.2.4.2 yamt # define _KERNEL 1
13 1.2.4.2 yamt #endif
14 1.2.4.2 yamt #if defined(__osf__)
15 1.2.4.2 yamt # define _PROTO_NET_H_
16 1.2.4.2 yamt #endif
17 1.2.4.2 yamt #include <sys/errno.h>
18 1.2.4.2 yamt #include <sys/types.h>
19 1.2.4.2 yamt #include <sys/param.h>
20 1.2.4.2 yamt #if defined(__NetBSD__)
21 1.2.4.2 yamt # if (NetBSD >= 199905) && !defined(IPFILTER_LKM) && defined(_KERNEL)
22 1.2.4.2 yamt # include "opt_ipfilter.h"
23 1.2.4.2 yamt # endif
24 1.2.4.2 yamt #endif
25 1.2.4.2 yamt #include <sys/file.h>
26 1.2.4.2 yamt #if !defined(_KERNEL) && !defined(__KERNEL__)
27 1.2.4.2 yamt # include <stdio.h>
28 1.2.4.2 yamt # include <stdlib.h>
29 1.2.4.2 yamt # include <string.h>
30 1.2.4.2 yamt # define _KERNEL
31 1.2.4.2 yamt # ifdef __OpenBSD__
32 1.2.4.2 yamt struct file;
33 1.2.4.2 yamt # endif
34 1.2.4.2 yamt # include <sys/uio.h>
35 1.2.4.2 yamt # undef _KERNEL
36 1.2.4.2 yamt #else
37 1.2.4.2 yamt # include <sys/systm.h>
38 1.2.4.2 yamt # if defined(NetBSD) && (__NetBSD_Version__ >= 104000000)
39 1.2.4.2 yamt # include <sys/proc.h>
40 1.2.4.2 yamt # endif
41 1.2.4.2 yamt #endif
42 1.2.4.2 yamt #include <sys/time.h>
43 1.2.4.2 yamt #if defined(_KERNEL) && !defined(SOLARIS2)
44 1.2.4.2 yamt # include <sys/mbuf.h>
45 1.2.4.2 yamt #endif
46 1.2.4.2 yamt #if defined(__SVR4) || defined(__svr4__)
47 1.2.4.2 yamt # include <sys/byteorder.h>
48 1.2.4.2 yamt # ifdef _KERNEL
49 1.2.4.2 yamt # include <sys/dditypes.h>
50 1.2.4.2 yamt # endif
51 1.2.4.2 yamt # include <sys/stream.h>
52 1.2.4.2 yamt # include <sys/kmem.h>
53 1.2.4.2 yamt #endif
54 1.2.4.2 yamt #if defined(__FreeBSD_version) && (__FreeBSD_version >= 300000)
55 1.2.4.2 yamt # include <sys/malloc.h>
56 1.2.4.2 yamt #endif
57 1.2.4.2 yamt
58 1.2.4.2 yamt #include <sys/socket.h>
59 1.2.4.2 yamt #include <net/if.h>
60 1.2.4.2 yamt #include <netinet/in.h>
61 1.2.4.2 yamt #if !defined(_KERNEL)
62 1.2.4.2 yamt # include "ipf.h"
63 1.2.4.2 yamt #endif
64 1.2.4.2 yamt
65 1.2.4.2 yamt #include "netinet/ip_compat.h"
66 1.2.4.2 yamt #include "netinet/ip_fil.h"
67 1.2.4.2 yamt #include "netinet/ip_pool.h"
68 1.2.4.2 yamt #include "netinet/radix_ipf.h"
69 1.2.4.2 yamt
70 1.2.4.2 yamt /* END OF INCLUDES */
71 1.2.4.2 yamt
72 1.2.4.2 yamt #if !defined(lint)
73 1.2.4.2 yamt #if defined(__NetBSD__)
74 1.2.4.2 yamt #include <sys/cdefs.h>
75 1.2.4.4 yamt __KERNEL_RCSID(0, "$NetBSD: ip_pool.c,v 1.2.4.4 2014/05/22 11:40:58 yamt Exp $");
76 1.2.4.2 yamt #else
77 1.2.4.2 yamt static const char sccsid[] = "@(#)ip_fil.c 2.41 6/5/96 (C) 1993-2000 Darren Reed";
78 1.2.4.3 yamt static const char rcsid[] = "@(#)Id: ip_pool.c,v 1.1.1.2 2012/07/22 13:45:31 darrenr Exp";
79 1.2.4.2 yamt #endif
80 1.2.4.2 yamt #endif
81 1.2.4.2 yamt
82 1.2.4.2 yamt typedef struct ipf_pool_softc_s {
83 1.2.4.2 yamt void *ipf_radix;
84 1.2.4.2 yamt ip_pool_t *ipf_pool_list[LOOKUP_POOL_SZ];
85 1.2.4.2 yamt ipf_pool_stat_t ipf_pool_stats;
86 1.2.4.2 yamt ip_pool_node_t *ipf_node_explist;
87 1.2.4.2 yamt } ipf_pool_softc_t;
88 1.2.4.2 yamt
89 1.2.4.2 yamt
90 1.2.4.3 yamt static void ipf_pool_clearnodes(ipf_main_softc_t *, ipf_pool_softc_t *,
91 1.2.4.3 yamt ip_pool_t *);
92 1.2.4.2 yamt static int ipf_pool_create(ipf_main_softc_t *, ipf_pool_softc_t *, iplookupop_t *);
93 1.2.4.2 yamt static int ipf_pool_deref(ipf_main_softc_t *, void *, void *);
94 1.2.4.2 yamt static int ipf_pool_destroy(ipf_main_softc_t *, ipf_pool_softc_t *, int, char *);
95 1.2.4.2 yamt static void *ipf_pool_exists(ipf_pool_softc_t *, int, char *);
96 1.2.4.2 yamt static void *ipf_pool_find(void *, int, char *);
97 1.2.4.2 yamt static ip_pool_node_t *ipf_pool_findeq(ipf_pool_softc_t *, ip_pool_t *,
98 1.2.4.2 yamt addrfamily_t *, addrfamily_t *);
99 1.2.4.3 yamt static void ipf_pool_free(ipf_main_softc_t *, ipf_pool_softc_t *, ip_pool_t *);
100 1.2.4.2 yamt static int ipf_pool_insert_node(ipf_main_softc_t *, ipf_pool_softc_t *,
101 1.2.4.2 yamt ip_pool_t *, struct ip_pool_node *);
102 1.2.4.2 yamt static int ipf_pool_iter_deref(ipf_main_softc_t *, void *, int, int, void *);
103 1.2.4.2 yamt static int ipf_pool_iter_next(ipf_main_softc_t *, void *, ipftoken_t *,
104 1.2.4.2 yamt ipflookupiter_t *);
105 1.2.4.2 yamt static size_t ipf_pool_flush(ipf_main_softc_t *, void *, iplookupflush_t *);
106 1.2.4.2 yamt static int ipf_pool_node_add(ipf_main_softc_t *, void *, iplookupop_t *,
107 1.2.4.2 yamt int);
108 1.2.4.2 yamt static int ipf_pool_node_del(ipf_main_softc_t *, void *, iplookupop_t *,
109 1.2.4.2 yamt int);
110 1.2.4.2 yamt static void ipf_pool_node_deref(ipf_pool_softc_t *, ip_pool_node_t *);
111 1.2.4.3 yamt static int ipf_pool_remove_node(ipf_main_softc_t *, ipf_pool_softc_t *,
112 1.2.4.3 yamt ip_pool_t *, ip_pool_node_t *);
113 1.2.4.2 yamt static int ipf_pool_search(ipf_main_softc_t *, void *, int,
114 1.2.4.2 yamt void *, u_int);
115 1.2.4.2 yamt static void *ipf_pool_soft_create(ipf_main_softc_t *);
116 1.2.4.2 yamt static void ipf_pool_soft_destroy(ipf_main_softc_t *, void *);
117 1.2.4.2 yamt static void ipf_pool_soft_fini(ipf_main_softc_t *, void *);
118 1.2.4.2 yamt static int ipf_pool_soft_init(ipf_main_softc_t *, void *);
119 1.2.4.2 yamt static int ipf_pool_stats_get(ipf_main_softc_t *, void *, iplookupop_t *);
120 1.2.4.2 yamt static int ipf_pool_table_add(ipf_main_softc_t *, void *, iplookupop_t *);
121 1.2.4.2 yamt static int ipf_pool_table_del(ipf_main_softc_t *, void *, iplookupop_t *);
122 1.2.4.2 yamt static void *ipf_pool_select_add_ref(void *, int, char *);
123 1.2.4.2 yamt static void ipf_pool_expire(ipf_main_softc_t *, void *);
124 1.2.4.2 yamt
125 1.2.4.2 yamt ipf_lookup_t ipf_pool_backend = {
126 1.2.4.2 yamt IPLT_POOL,
127 1.2.4.2 yamt ipf_pool_soft_create,
128 1.2.4.2 yamt ipf_pool_soft_destroy,
129 1.2.4.2 yamt ipf_pool_soft_init,
130 1.2.4.2 yamt ipf_pool_soft_fini,
131 1.2.4.2 yamt ipf_pool_search,
132 1.2.4.2 yamt ipf_pool_flush,
133 1.2.4.2 yamt ipf_pool_iter_deref,
134 1.2.4.2 yamt ipf_pool_iter_next,
135 1.2.4.2 yamt ipf_pool_node_add,
136 1.2.4.2 yamt ipf_pool_node_del,
137 1.2.4.2 yamt ipf_pool_stats_get,
138 1.2.4.2 yamt ipf_pool_table_add,
139 1.2.4.2 yamt ipf_pool_table_del,
140 1.2.4.2 yamt ipf_pool_deref,
141 1.2.4.2 yamt ipf_pool_find,
142 1.2.4.2 yamt ipf_pool_select_add_ref,
143 1.2.4.2 yamt NULL,
144 1.2.4.2 yamt ipf_pool_expire,
145 1.2.4.2 yamt NULL
146 1.2.4.2 yamt };
147 1.2.4.2 yamt
148 1.2.4.2 yamt
149 1.2.4.2 yamt #ifdef TEST_POOL
150 1.2.4.2 yamt void treeprint(ip_pool_t *);
151 1.2.4.2 yamt
152 1.2.4.2 yamt int
153 1.2.4.2 yamt main(argc, argv)
154 1.2.4.2 yamt int argc;
155 1.2.4.2 yamt char *argv[];
156 1.2.4.2 yamt {
157 1.2.4.2 yamt ip_pool_node_t node;
158 1.2.4.2 yamt addrfamily_t a, b;
159 1.2.4.2 yamt iplookupop_t op;
160 1.2.4.2 yamt ip_pool_t *ipo;
161 1.2.4.2 yamt i6addr_t ip;
162 1.2.4.2 yamt
163 1.2.4.2 yamt RWLOCK_INIT(softc->ipf_poolrw, "poolrw");
164 1.2.4.2 yamt ipf_pool_init();
165 1.2.4.2 yamt
166 1.2.4.2 yamt bzero((char *)&ip, sizeof(ip));
167 1.2.4.2 yamt bzero((char *)&op, sizeof(op));
168 1.2.4.2 yamt bzero((char *)&node, sizeof(node));
169 1.2.4.2 yamt strlcpy(op.iplo_name, "0", sizeof(op.iplo_name));
170 1.2.4.2 yamt
171 1.2.4.2 yamt if (ipf_pool_create(&op) == 0)
172 1.2.4.2 yamt ipo = ipf_pool_exists(0, "0");
173 1.2.4.2 yamt
174 1.2.4.2 yamt node.ipn_addr.adf_family = AF_INET;
175 1.2.4.2 yamt
176 1.2.4.2 yamt node.ipn_addr.adf_addr.in4.s_addr = 0x0a010203;
177 1.2.4.2 yamt node.ipn_mask.adf_addr.in4.s_addr = 0xffffffff;
178 1.2.4.2 yamt node.ipn_info = 1;
179 1.2.4.2 yamt ipf_pool_insert_node(ipo, &node);
180 1.2.4.2 yamt
181 1.2.4.2 yamt node.ipn_addr.adf_addr.in4.s_addr = 0x0a000000;
182 1.2.4.2 yamt node.ipn_mask.adf_addr.in4.s_addr = 0xff000000;
183 1.2.4.2 yamt node.ipn_info = 0;
184 1.2.4.2 yamt ipf_pool_insert_node(ipo, &node);
185 1.2.4.2 yamt
186 1.2.4.2 yamt node.ipn_addr.adf_addr.in4.s_addr = 0x0a010100;
187 1.2.4.2 yamt node.ipn_mask.adf_addr.in4.s_addr = 0xffffff00;
188 1.2.4.2 yamt node.ipn_info = 1;
189 1.2.4.2 yamt ipf_pool_insert_node(ipo, &node);
190 1.2.4.2 yamt
191 1.2.4.2 yamt node.ipn_addr.adf_addr.in4.s_addr = 0x0a010200;
192 1.2.4.2 yamt node.ipn_mask.adf_addr.in4.s_addr = 0xffffff00;
193 1.2.4.2 yamt node.ipn_info = 0;
194 1.2.4.2 yamt ipf_pool_insert_node(ipo, &node);
195 1.2.4.2 yamt
196 1.2.4.2 yamt node.ipn_addr.adf_addr.in4.s_addr = 0x0a010000;
197 1.2.4.2 yamt node.ipn_mask.adf_addr.in4.s_addr = 0xffff0000;
198 1.2.4.2 yamt node.ipn_info = 1;
199 1.2.4.2 yamt ipf_pool_insert_node(ipo, &node);
200 1.2.4.2 yamt
201 1.2.4.2 yamt node.ipn_addr.adf_addr.in4.s_addr = 0x0a01020f;
202 1.2.4.2 yamt node.ipn_mask.adf_addr.in4.s_addr = 0xffffffff;
203 1.2.4.2 yamt node.ipn_info = 1;
204 1.2.4.2 yamt ipf_pool_insert_node(ipo, &node);
205 1.2.4.2 yamt #ifdef DEBUG_POOL
206 1.2.4.2 yamt treeprint(ipo);
207 1.2.4.2 yamt #endif
208 1.2.4.2 yamt ip.in4.s_addr = 0x0a00aabb;
209 1.2.4.2 yamt printf("search(%#x) = %d (0)\n", ip.in4.s_addr,
210 1.2.4.2 yamt ipf_pool_search(ipo, 4, &ip, 1));
211 1.2.4.2 yamt
212 1.2.4.2 yamt ip.in4.s_addr = 0x0a000001;
213 1.2.4.2 yamt printf("search(%#x) = %d (0)\n", ip.in4.s_addr,
214 1.2.4.2 yamt ipf_pool_search(ipo, 4, &ip, 1));
215 1.2.4.2 yamt
216 1.2.4.2 yamt ip.in4.s_addr = 0x0a000101;
217 1.2.4.2 yamt printf("search(%#x) = %d (0)\n", ip.in4.s_addr,
218 1.2.4.2 yamt ipf_pool_search(ipo, 4, &ip, 1));
219 1.2.4.2 yamt
220 1.2.4.2 yamt ip.in4.s_addr = 0x0a010001;
221 1.2.4.2 yamt printf("search(%#x) = %d (1)\n", ip.in4.s_addr,
222 1.2.4.2 yamt ipf_pool_search(ipo, 4, &ip, 1));
223 1.2.4.2 yamt
224 1.2.4.2 yamt ip.in4.s_addr = 0x0a010101;
225 1.2.4.2 yamt printf("search(%#x) = %d (1)\n", ip.in4.s_addr,
226 1.2.4.2 yamt ipf_pool_search(ipo, 4, &ip, 1));
227 1.2.4.2 yamt
228 1.2.4.2 yamt ip.in4.s_addr = 0x0a010201;
229 1.2.4.2 yamt printf("search(%#x) = %d (0)\n", ip.in4.s_addr,
230 1.2.4.2 yamt ipf_pool_search(ipo, 4, &ip, 1));
231 1.2.4.2 yamt
232 1.2.4.2 yamt ip.in4.s_addr = 0x0a010203;
233 1.2.4.2 yamt printf("search(%#x) = %d (1)\n", ip.in4.s_addr,
234 1.2.4.2 yamt ipf_pool_search(ipo, 4, &ip, 1));
235 1.2.4.2 yamt
236 1.2.4.2 yamt ip.in4.s_addr = 0x0a01020f;
237 1.2.4.2 yamt printf("search(%#x) = %d (1)\n", ip.in4.s_addr,
238 1.2.4.2 yamt ipf_pool_search(ipo, 4, &ip, 1));
239 1.2.4.2 yamt
240 1.2.4.2 yamt ip.in4.s_addr = 0x0b00aabb;
241 1.2.4.2 yamt printf("search(%#x) = %d (-1)\n", ip.in4.s_addr,
242 1.2.4.2 yamt ipf_pool_search(ipo, 4, &ip, 1));
243 1.2.4.2 yamt
244 1.2.4.2 yamt #ifdef DEBUG_POOL
245 1.2.4.2 yamt treeprint(ipo);
246 1.2.4.2 yamt #endif
247 1.2.4.2 yamt
248 1.2.4.2 yamt ipf_pool_fini();
249 1.2.4.2 yamt
250 1.2.4.2 yamt return 0;
251 1.2.4.2 yamt }
252 1.2.4.2 yamt
253 1.2.4.2 yamt
254 1.2.4.2 yamt void
255 1.2.4.2 yamt treeprint(ipo)
256 1.2.4.2 yamt ip_pool_t *ipo;
257 1.2.4.2 yamt {
258 1.2.4.2 yamt ip_pool_node_t *c;
259 1.2.4.2 yamt
260 1.2.4.2 yamt for (c = ipo->ipo_list; c != NULL; c = c->ipn_next)
261 1.2.4.2 yamt printf("Node %p(%s) (%#x/%#x) = %d hits %lu\n",
262 1.2.4.2 yamt c, c->ipn_name, c->ipn_addr.adf_addr.in4.s_addr,
263 1.2.4.2 yamt c->ipn_mask.adf_addr.in4.s_addr,
264 1.2.4.2 yamt c->ipn_info, c->ipn_hits);
265 1.2.4.2 yamt }
266 1.2.4.2 yamt #endif /* TEST_POOL */
267 1.2.4.2 yamt
268 1.2.4.2 yamt
269 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
270 1.2.4.2 yamt /* Function: ipf_pool_soft_create */
271 1.2.4.2 yamt /* Returns: void * - NULL = failure, else pointer to local context */
272 1.2.4.2 yamt /* Parameters: softc(I) - pointer to soft context main structure */
273 1.2.4.2 yamt /* */
274 1.2.4.2 yamt /* Initialise the routing table data structures where required. */
275 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
276 1.2.4.2 yamt static void *
277 1.2.4.2 yamt ipf_pool_soft_create(ipf_main_softc_t *softc)
278 1.2.4.2 yamt {
279 1.2.4.2 yamt ipf_pool_softc_t *softp;
280 1.2.4.2 yamt
281 1.2.4.2 yamt KMALLOC(softp, ipf_pool_softc_t *);
282 1.2.4.3 yamt if (softp == NULL) {
283 1.2.4.3 yamt IPFERROR(70032);
284 1.2.4.2 yamt return NULL;
285 1.2.4.3 yamt }
286 1.2.4.2 yamt
287 1.2.4.2 yamt bzero((char *)softp, sizeof(*softp));
288 1.2.4.2 yamt
289 1.2.4.2 yamt softp->ipf_radix = ipf_rx_create();
290 1.2.4.2 yamt if (softp->ipf_radix == NULL) {
291 1.2.4.3 yamt IPFERROR(70033);
292 1.2.4.2 yamt KFREE(softp);
293 1.2.4.2 yamt return NULL;
294 1.2.4.2 yamt }
295 1.2.4.2 yamt
296 1.2.4.2 yamt return softp;
297 1.2.4.2 yamt }
298 1.2.4.2 yamt
299 1.2.4.2 yamt
300 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
301 1.2.4.2 yamt /* Function: ipf_pool_soft_init */
302 1.2.4.2 yamt /* Returns: int - 0 = success, else error */
303 1.2.4.2 yamt /* Parameters: softc(I) - pointer to soft context main structure */
304 1.2.4.2 yamt /* arg(I) - pointer to local context to use */
305 1.2.4.2 yamt /* */
306 1.2.4.2 yamt /* Initialise the routing table data structures where required. */
307 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
308 1.2.4.2 yamt static int
309 1.2.4.2 yamt ipf_pool_soft_init(ipf_main_softc_t *softc, void *arg)
310 1.2.4.2 yamt {
311 1.2.4.2 yamt ipf_pool_softc_t *softp = arg;
312 1.2.4.2 yamt
313 1.2.4.2 yamt ipf_rx_init(softp->ipf_radix);
314 1.2.4.2 yamt
315 1.2.4.2 yamt return 0;
316 1.2.4.2 yamt }
317 1.2.4.2 yamt
318 1.2.4.2 yamt
319 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
320 1.2.4.2 yamt /* Function: ipf_pool_soft_fini */
321 1.2.4.2 yamt /* Returns: Nil */
322 1.2.4.2 yamt /* Parameters: softc(I) - pointer to soft context main structure */
323 1.2.4.2 yamt /* arg(I) - pointer to local context to use */
324 1.2.4.2 yamt /* Locks: WRITE(ipf_global) */
325 1.2.4.2 yamt /* */
326 1.2.4.2 yamt /* Clean up all the pool data structures allocated and call the cleanup */
327 1.2.4.2 yamt /* function for the radix tree that supports the pools. ipf_pool_destroy is */
328 1.2.4.2 yamt /* used to delete the pools one by one to ensure they're properly freed up. */
329 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
330 1.2.4.2 yamt static void
331 1.2.4.2 yamt ipf_pool_soft_fini(ipf_main_softc_t *softc, void *arg)
332 1.2.4.2 yamt {
333 1.2.4.2 yamt ipf_pool_softc_t *softp = arg;
334 1.2.4.2 yamt ip_pool_t *p, *q;
335 1.2.4.2 yamt int i;
336 1.2.4.2 yamt
337 1.2.4.2 yamt softc = arg;
338 1.2.4.2 yamt
339 1.2.4.2 yamt for (i = -1; i <= IPL_LOGMAX; i++) {
340 1.2.4.2 yamt for (q = softp->ipf_pool_list[i + 1]; (p = q) != NULL; ) {
341 1.2.4.2 yamt q = p->ipo_next;
342 1.2.4.2 yamt (void) ipf_pool_destroy(softc, arg, i, p->ipo_name);
343 1.2.4.2 yamt }
344 1.2.4.2 yamt }
345 1.2.4.2 yamt }
346 1.2.4.2 yamt
347 1.2.4.2 yamt
348 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
349 1.2.4.2 yamt /* Function: ipf_pool_soft_destroy */
350 1.2.4.2 yamt /* Returns: Nil */
351 1.2.4.2 yamt /* Parameters: softc(I) - pointer to soft context main structure */
352 1.2.4.2 yamt /* arg(I) - pointer to local context to use */
353 1.2.4.2 yamt /* */
354 1.2.4.2 yamt /* Clean up the pool by free'ing the radix tree associated with it and free */
355 1.2.4.2 yamt /* up the pool context too. */
356 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
357 1.2.4.2 yamt static void
358 1.2.4.2 yamt ipf_pool_soft_destroy(ipf_main_softc_t *softc, void *arg)
359 1.2.4.2 yamt {
360 1.2.4.2 yamt ipf_pool_softc_t *softp = arg;
361 1.2.4.2 yamt
362 1.2.4.2 yamt ipf_rx_destroy(softp->ipf_radix);
363 1.2.4.2 yamt
364 1.2.4.2 yamt KFREE(softp);
365 1.2.4.2 yamt }
366 1.2.4.2 yamt
367 1.2.4.2 yamt
368 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
369 1.2.4.2 yamt /* Function: ipf_pool_node_add */
370 1.2.4.2 yamt /* Returns: int - 0 = success, else error */
371 1.2.4.2 yamt /* Parameters: softc(I) - pointer to soft context main structure */
372 1.2.4.2 yamt /* arg(I) - pointer to local context to use */
373 1.2.4.2 yamt /* op(I) - pointer to lookup operatin data */
374 1.2.4.2 yamt /* */
375 1.2.4.3 yamt /* When adding a new node, a check is made to ensure that the address/mask */
376 1.2.4.3 yamt /* pair supplied has been appropriately prepared by applying the mask to */
377 1.2.4.3 yamt /* the address prior to calling for the pair to be added. */
378 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
379 1.2.4.2 yamt static int
380 1.2.4.2 yamt ipf_pool_node_add(ipf_main_softc_t *softc, void *arg, iplookupop_t *op, int uid)
381 1.2.4.2 yamt {
382 1.2.4.2 yamt ip_pool_node_t node, *m;
383 1.2.4.2 yamt ip_pool_t *p;
384 1.2.4.2 yamt int err;
385 1.2.4.2 yamt
386 1.2.4.2 yamt if (op->iplo_size != sizeof(node)) {
387 1.2.4.2 yamt IPFERROR(70014);
388 1.2.4.2 yamt return EINVAL;
389 1.2.4.2 yamt }
390 1.2.4.2 yamt
391 1.2.4.2 yamt err = COPYIN(op->iplo_struct, &node, sizeof(node));
392 1.2.4.2 yamt if (err != 0) {
393 1.2.4.2 yamt IPFERROR(70015);
394 1.2.4.2 yamt return EFAULT;
395 1.2.4.2 yamt }
396 1.2.4.2 yamt
397 1.2.4.2 yamt p = ipf_pool_find(arg, op->iplo_unit, op->iplo_name);
398 1.2.4.2 yamt if (p == NULL) {
399 1.2.4.2 yamt IPFERROR(70017);
400 1.2.4.2 yamt return ESRCH;
401 1.2.4.2 yamt }
402 1.2.4.2 yamt
403 1.2.4.3 yamt if (node.ipn_addr.adf_family == AF_INET) {
404 1.2.4.3 yamt if (node.ipn_addr.adf_len != offsetof(addrfamily_t, adf_addr) +
405 1.2.4.3 yamt sizeof(struct in_addr)) {
406 1.2.4.3 yamt IPFERROR(70028);
407 1.2.4.3 yamt return EINVAL;
408 1.2.4.3 yamt }
409 1.2.4.3 yamt }
410 1.2.4.3 yamt #ifdef USE_INET6
411 1.2.4.3 yamt else if (node.ipn_addr.adf_family == AF_INET6) {
412 1.2.4.3 yamt if (node.ipn_addr.adf_len != offsetof(addrfamily_t, adf_addr) +
413 1.2.4.3 yamt sizeof(struct in6_addr)) {
414 1.2.4.3 yamt IPFERROR(70034);
415 1.2.4.3 yamt return EINVAL;
416 1.2.4.3 yamt }
417 1.2.4.3 yamt }
418 1.2.4.3 yamt #endif
419 1.2.4.3 yamt if (node.ipn_mask.adf_len != node.ipn_addr.adf_len) {
420 1.2.4.3 yamt IPFERROR(70029);
421 1.2.4.3 yamt return EINVAL;
422 1.2.4.3 yamt }
423 1.2.4.3 yamt
424 1.2.4.3 yamt /*
425 1.2.4.3 yamt * Check that the address/mask pair works.
426 1.2.4.3 yamt */
427 1.2.4.3 yamt if (node.ipn_addr.adf_family == AF_INET) {
428 1.2.4.3 yamt if ((node.ipn_addr.adf_addr.in4.s_addr &
429 1.2.4.3 yamt node.ipn_mask.adf_addr.in4.s_addr) !=
430 1.2.4.3 yamt node.ipn_addr.adf_addr.in4.s_addr) {
431 1.2.4.3 yamt IPFERROR(70035);
432 1.2.4.3 yamt return EINVAL;
433 1.2.4.3 yamt }
434 1.2.4.3 yamt }
435 1.2.4.3 yamt #ifdef USE_INET6
436 1.2.4.3 yamt else if (node.ipn_addr.adf_family == AF_INET6) {
437 1.2.4.3 yamt if (IP6_MASKNEQ(&node.ipn_addr.adf_addr.in6,
438 1.2.4.3 yamt &node.ipn_mask.adf_addr.in6,
439 1.2.4.3 yamt &node.ipn_addr.adf_addr.in6)) {
440 1.2.4.3 yamt IPFERROR(70036);
441 1.2.4.3 yamt return EINVAL;
442 1.2.4.3 yamt }
443 1.2.4.3 yamt }
444 1.2.4.3 yamt #endif
445 1.2.4.3 yamt
446 1.2.4.2 yamt /*
447 1.2.4.2 yamt * add an entry to a pool - return an error if it already
448 1.2.4.2 yamt * exists remove an entry from a pool - if it exists
449 1.2.4.2 yamt * - in both cases, the pool *must* exist!
450 1.2.4.2 yamt */
451 1.2.4.2 yamt m = ipf_pool_findeq(arg, p, &node.ipn_addr, &node.ipn_mask);
452 1.2.4.2 yamt if (m != NULL) {
453 1.2.4.2 yamt IPFERROR(70018);
454 1.2.4.2 yamt return EEXIST;
455 1.2.4.2 yamt }
456 1.2.4.2 yamt err = ipf_pool_insert_node(softc, arg, p, &node);
457 1.2.4.2 yamt
458 1.2.4.2 yamt return err;
459 1.2.4.2 yamt }
460 1.2.4.2 yamt
461 1.2.4.2 yamt
462 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
463 1.2.4.2 yamt /* Function: ipf_pool_node_del */
464 1.2.4.2 yamt /* Returns: int - 0 = success, else error */
465 1.2.4.2 yamt /* Parameters: softc(I) - pointer to soft context main structure */
466 1.2.4.2 yamt /* arg(I) - pointer to local context to use */
467 1.2.4.2 yamt /* op(I) - pointer to lookup operatin data */
468 1.2.4.2 yamt /* */
469 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
470 1.2.4.2 yamt static int
471 1.2.4.2 yamt ipf_pool_node_del(ipf_main_softc_t *softc, void *arg, iplookupop_t *op, int uid)
472 1.2.4.2 yamt {
473 1.2.4.2 yamt ip_pool_node_t node, *m;
474 1.2.4.2 yamt ip_pool_t *p;
475 1.2.4.2 yamt int err;
476 1.2.4.2 yamt
477 1.2.4.2 yamt
478 1.2.4.2 yamt if (op->iplo_size != sizeof(node)) {
479 1.2.4.2 yamt IPFERROR(70019);
480 1.2.4.2 yamt return EINVAL;
481 1.2.4.2 yamt }
482 1.2.4.2 yamt node.ipn_uid = uid;
483 1.2.4.2 yamt
484 1.2.4.2 yamt err = COPYIN(op->iplo_struct, &node, sizeof(node));
485 1.2.4.2 yamt if (err != 0) {
486 1.2.4.2 yamt IPFERROR(70020);
487 1.2.4.2 yamt return EFAULT;
488 1.2.4.2 yamt }
489 1.2.4.2 yamt
490 1.2.4.3 yamt if (node.ipn_addr.adf_family == AF_INET) {
491 1.2.4.3 yamt if (node.ipn_addr.adf_len != offsetof(addrfamily_t, adf_addr) +
492 1.2.4.3 yamt sizeof(struct in_addr)) {
493 1.2.4.3 yamt IPFERROR(70030);
494 1.2.4.3 yamt return EINVAL;
495 1.2.4.3 yamt }
496 1.2.4.3 yamt }
497 1.2.4.3 yamt #ifdef USE_INET6
498 1.2.4.3 yamt else if (node.ipn_addr.adf_family == AF_INET6) {
499 1.2.4.3 yamt if (node.ipn_addr.adf_len != offsetof(addrfamily_t, adf_addr) +
500 1.2.4.3 yamt sizeof(struct in6_addr)) {
501 1.2.4.3 yamt IPFERROR(70037);
502 1.2.4.3 yamt return EINVAL;
503 1.2.4.3 yamt }
504 1.2.4.3 yamt }
505 1.2.4.3 yamt #endif
506 1.2.4.3 yamt if (node.ipn_mask.adf_len != node.ipn_addr.adf_len) {
507 1.2.4.3 yamt IPFERROR(70031);
508 1.2.4.3 yamt return EINVAL;
509 1.2.4.3 yamt }
510 1.2.4.3 yamt
511 1.2.4.2 yamt p = ipf_pool_find(arg, op->iplo_unit, op->iplo_name);
512 1.2.4.2 yamt if (p == NULL) {
513 1.2.4.2 yamt IPFERROR(70021);
514 1.2.4.2 yamt return ESRCH;
515 1.2.4.2 yamt }
516 1.2.4.2 yamt
517 1.2.4.2 yamt m = ipf_pool_findeq(arg, p, &node.ipn_addr, &node.ipn_mask);
518 1.2.4.2 yamt if (m == NULL) {
519 1.2.4.2 yamt IPFERROR(70022);
520 1.2.4.2 yamt return ENOENT;
521 1.2.4.2 yamt }
522 1.2.4.2 yamt
523 1.2.4.2 yamt if ((uid != 0) && (uid != m->ipn_uid)) {
524 1.2.4.2 yamt IPFERROR(70024);
525 1.2.4.2 yamt return EACCES;
526 1.2.4.2 yamt }
527 1.2.4.2 yamt
528 1.2.4.3 yamt err = ipf_pool_remove_node(softc, arg, p, m);
529 1.2.4.2 yamt
530 1.2.4.2 yamt return err;
531 1.2.4.2 yamt }
532 1.2.4.2 yamt
533 1.2.4.2 yamt
534 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
535 1.2.4.2 yamt /* Function: ipf_pool_table_add */
536 1.2.4.2 yamt /* Returns: int - 0 = success, else error */
537 1.2.4.2 yamt /* Parameters: softc(I) - pointer to soft context main structure */
538 1.2.4.2 yamt /* arg(I) - pointer to local context to use */
539 1.2.4.2 yamt /* op(I) - pointer to lookup operatin data */
540 1.2.4.2 yamt /* */
541 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
542 1.2.4.2 yamt static int
543 1.2.4.2 yamt ipf_pool_table_add(ipf_main_softc_t *softc, void *arg, iplookupop_t *op)
544 1.2.4.2 yamt {
545 1.2.4.2 yamt int err;
546 1.2.4.2 yamt
547 1.2.4.2 yamt if (((op->iplo_arg & LOOKUP_ANON) == 0) &&
548 1.2.4.2 yamt (ipf_pool_find(arg, op->iplo_unit, op->iplo_name) != NULL)) {
549 1.2.4.2 yamt IPFERROR(70023);
550 1.2.4.2 yamt err = EEXIST;
551 1.2.4.2 yamt } else {
552 1.2.4.2 yamt err = ipf_pool_create(softc, arg, op);
553 1.2.4.2 yamt }
554 1.2.4.2 yamt
555 1.2.4.2 yamt return err;
556 1.2.4.2 yamt }
557 1.2.4.2 yamt
558 1.2.4.2 yamt
559 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
560 1.2.4.2 yamt /* Function: ipf_pool_table_del */
561 1.2.4.2 yamt /* Returns: int - 0 = success, else error */
562 1.2.4.2 yamt /* Parameters: softc(I) - pointer to soft context main structure */
563 1.2.4.2 yamt /* arg(I) - pointer to local context to use */
564 1.2.4.2 yamt /* op(I) - pointer to lookup operatin data */
565 1.2.4.2 yamt /* */
566 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
567 1.2.4.2 yamt static int
568 1.2.4.2 yamt ipf_pool_table_del(ipf_main_softc_t *softc, void *arg, iplookupop_t *op)
569 1.2.4.2 yamt {
570 1.2.4.2 yamt return ipf_pool_destroy(softc, arg, op->iplo_unit, op->iplo_name);
571 1.2.4.2 yamt }
572 1.2.4.2 yamt
573 1.2.4.2 yamt
574 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
575 1.2.4.2 yamt /* Function: ipf_pool_statistics */
576 1.2.4.2 yamt /* Returns: int - 0 = success, else error */
577 1.2.4.2 yamt /* Parameters: softc(I) - pointer to soft context main structure */
578 1.2.4.2 yamt /* arg(I) - pointer to local context to use */
579 1.2.4.2 yamt /* op(I) - pointer to lookup operatin data */
580 1.2.4.2 yamt /* */
581 1.2.4.2 yamt /* Copy the current statistics out into user space, collecting pool list */
582 1.2.4.2 yamt /* pointers as appropriate for later use. */
583 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
584 1.2.4.2 yamt static int
585 1.2.4.2 yamt ipf_pool_stats_get(ipf_main_softc_t *softc, void *arg, iplookupop_t *op)
586 1.2.4.2 yamt {
587 1.2.4.2 yamt ipf_pool_softc_t *softp = arg;
588 1.2.4.2 yamt ipf_pool_stat_t stats;
589 1.2.4.2 yamt int unit, i, err = 0;
590 1.2.4.2 yamt
591 1.2.4.2 yamt if (op->iplo_size != sizeof(ipf_pool_stat_t)) {
592 1.2.4.2 yamt IPFERROR(70001);
593 1.2.4.2 yamt return EINVAL;
594 1.2.4.2 yamt }
595 1.2.4.2 yamt
596 1.2.4.2 yamt bcopy((char *)&softp->ipf_pool_stats, (char *)&stats, sizeof(stats));
597 1.2.4.2 yamt unit = op->iplo_unit;
598 1.2.4.2 yamt if (unit == IPL_LOGALL) {
599 1.2.4.2 yamt for (i = 0; i <= LOOKUP_POOL_MAX; i++)
600 1.2.4.2 yamt stats.ipls_list[i] = softp->ipf_pool_list[i];
601 1.2.4.2 yamt } else if (unit >= 0 && unit <= IPL_LOGMAX) {
602 1.2.4.2 yamt unit++; /* -1 => 0 */
603 1.2.4.2 yamt if (op->iplo_name[0] != '\0')
604 1.2.4.2 yamt stats.ipls_list[unit] = ipf_pool_exists(softp, unit - 1,
605 1.2.4.2 yamt op->iplo_name);
606 1.2.4.2 yamt else
607 1.2.4.2 yamt stats.ipls_list[unit] = softp->ipf_pool_list[unit];
608 1.2.4.2 yamt } else {
609 1.2.4.2 yamt IPFERROR(70025);
610 1.2.4.2 yamt err = EINVAL;
611 1.2.4.2 yamt }
612 1.2.4.2 yamt if (err == 0) {
613 1.2.4.2 yamt err = COPYOUT(&stats, op->iplo_struct, sizeof(stats));
614 1.2.4.2 yamt if (err != 0) {
615 1.2.4.2 yamt IPFERROR(70026);
616 1.2.4.2 yamt return EFAULT;
617 1.2.4.2 yamt }
618 1.2.4.2 yamt }
619 1.2.4.2 yamt return 0;
620 1.2.4.2 yamt }
621 1.2.4.2 yamt
622 1.2.4.2 yamt
623 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
624 1.2.4.2 yamt /* Function: ipf_pool_exists */
625 1.2.4.2 yamt /* Returns: int - 0 = success, else error */
626 1.2.4.2 yamt /* Parameters: softp(I) - pointer to soft context pool information */
627 1.2.4.2 yamt /* unit(I) - ipfilter device to which we are working on */
628 1.2.4.2 yamt /* name(I) - name of the pool */
629 1.2.4.2 yamt /* */
630 1.2.4.2 yamt /* Find a matching pool inside the collection of pools for a particular */
631 1.2.4.2 yamt /* device, indicated by the unit number. */
632 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
633 1.2.4.2 yamt static void *
634 1.2.4.2 yamt ipf_pool_exists(ipf_pool_softc_t *softp, int unit, char *name)
635 1.2.4.2 yamt {
636 1.2.4.2 yamt ip_pool_t *p;
637 1.2.4.2 yamt int i;
638 1.2.4.2 yamt
639 1.2.4.2 yamt if (unit == IPL_LOGALL) {
640 1.2.4.2 yamt for (i = 0; i <= LOOKUP_POOL_MAX; i++) {
641 1.2.4.2 yamt for (p = softp->ipf_pool_list[i]; p != NULL;
642 1.2.4.2 yamt p = p->ipo_next) {
643 1.2.4.2 yamt if (strncmp(p->ipo_name, name,
644 1.2.4.2 yamt sizeof(p->ipo_name)) == 0)
645 1.2.4.2 yamt break;
646 1.2.4.2 yamt }
647 1.2.4.2 yamt if (p != NULL)
648 1.2.4.2 yamt break;
649 1.2.4.2 yamt }
650 1.2.4.2 yamt } else {
651 1.2.4.2 yamt for (p = softp->ipf_pool_list[unit + 1]; p != NULL;
652 1.2.4.2 yamt p = p->ipo_next)
653 1.2.4.2 yamt if (strncmp(p->ipo_name, name,
654 1.2.4.2 yamt sizeof(p->ipo_name)) == 0)
655 1.2.4.2 yamt break;
656 1.2.4.2 yamt }
657 1.2.4.2 yamt return p;
658 1.2.4.2 yamt }
659 1.2.4.2 yamt
660 1.2.4.2 yamt
661 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
662 1.2.4.2 yamt /* Function: ipf_pool_find */
663 1.2.4.2 yamt /* Returns: int - 0 = success, else error */
664 1.2.4.2 yamt /* Parameters: arg(I) - pointer to local context to use */
665 1.2.4.2 yamt /* unit(I) - ipfilter device to which we are working on */
666 1.2.4.2 yamt /* name(I) - name of the pool */
667 1.2.4.2 yamt /* */
668 1.2.4.2 yamt /* Find a matching pool inside the collection of pools for a particular */
669 1.2.4.2 yamt /* device, indicated by the unit number. If it is marked for deletion then */
670 1.2.4.2 yamt /* pretend it does not exist. */
671 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
672 1.2.4.2 yamt static void *
673 1.2.4.2 yamt ipf_pool_find(void *arg, int unit, char *name)
674 1.2.4.2 yamt {
675 1.2.4.2 yamt ipf_pool_softc_t *softp = arg;
676 1.2.4.2 yamt ip_pool_t *p;
677 1.2.4.2 yamt
678 1.2.4.2 yamt p = ipf_pool_exists(softp, unit, name);
679 1.2.4.2 yamt if ((p != NULL) && (p->ipo_flags & IPOOL_DELETE))
680 1.2.4.2 yamt return NULL;
681 1.2.4.2 yamt
682 1.2.4.2 yamt return p;
683 1.2.4.2 yamt }
684 1.2.4.2 yamt
685 1.2.4.2 yamt
686 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
687 1.2.4.2 yamt /* Function: ipf_pool_select_add_ref */
688 1.2.4.2 yamt /* Returns: int - 0 = success, else error */
689 1.2.4.2 yamt /* Parameters: arg(I) - pointer to local context to use */
690 1.2.4.2 yamt /* unit(I) - ipfilter device to which we are working on */
691 1.2.4.2 yamt /* name(I) - name of the pool */
692 1.2.4.2 yamt /* */
693 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
694 1.2.4.2 yamt static void *
695 1.2.4.2 yamt ipf_pool_select_add_ref(void *arg, int unit, char *name)
696 1.2.4.2 yamt {
697 1.2.4.2 yamt ip_pool_t *p;
698 1.2.4.2 yamt
699 1.2.4.2 yamt p = ipf_pool_find(arg, -1, name);
700 1.2.4.2 yamt if (p == NULL)
701 1.2.4.2 yamt p = ipf_pool_find(arg, unit, name);
702 1.2.4.2 yamt if (p != NULL) {
703 1.2.4.2 yamt ATOMIC_INC32(p->ipo_ref);
704 1.2.4.2 yamt }
705 1.2.4.2 yamt return p;
706 1.2.4.2 yamt }
707 1.2.4.2 yamt
708 1.2.4.2 yamt
709 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
710 1.2.4.2 yamt /* Function: ipf_pool_findeq */
711 1.2.4.2 yamt /* Returns: int - 0 = success, else error */
712 1.2.4.2 yamt /* Parameters: softp(I) - pointer to soft context pool information */
713 1.2.4.2 yamt /* ipo(I) - pointer to the pool getting the new node. */
714 1.2.4.2 yamt /* addr(I) - pointer to address information to match on */
715 1.2.4.2 yamt /* mask(I) - pointer to the address mask to match */
716 1.2.4.2 yamt /* */
717 1.2.4.2 yamt /* Searches for an exact match of an entry in the pool. */
718 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
719 1.2.4.2 yamt extern void printhostmask(int, u_32_t *, u_32_t *);
720 1.2.4.2 yamt static ip_pool_node_t *
721 1.2.4.2 yamt ipf_pool_findeq(ipf_pool_softc_t *softp, ip_pool_t *ipo, addrfamily_t *addr,
722 1.2.4.2 yamt addrfamily_t *mask)
723 1.2.4.2 yamt {
724 1.2.4.2 yamt ipf_rdx_node_t *n;
725 1.2.4.2 yamt
726 1.2.4.2 yamt n = ipo->ipo_head->lookup(ipo->ipo_head, addr, mask);
727 1.2.4.2 yamt return (ip_pool_node_t *)n;
728 1.2.4.2 yamt }
729 1.2.4.2 yamt
730 1.2.4.2 yamt
731 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
732 1.2.4.2 yamt /* Function: ipf_pool_search */
733 1.2.4.2 yamt /* Returns: int - 0 == +ve match, -1 == error, 1 == -ve/no match */
734 1.2.4.2 yamt /* Parameters: softc(I) - pointer to soft context main structure */
735 1.2.4.2 yamt /* tptr(I) - pointer to the pool to search */
736 1.2.4.2 yamt /* version(I) - IP protocol version (4 or 6) */
737 1.2.4.2 yamt /* dptr(I) - pointer to address information */
738 1.2.4.2 yamt /* bytes(I) - length of packet */
739 1.2.4.2 yamt /* */
740 1.2.4.2 yamt /* Search the pool for a given address and return a search result. */
741 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
742 1.2.4.2 yamt static int
743 1.2.4.2 yamt ipf_pool_search(ipf_main_softc_t *softc, void *tptr, int ipversion, void *dptr,
744 1.2.4.2 yamt u_int bytes)
745 1.2.4.2 yamt {
746 1.2.4.2 yamt ipf_rdx_node_t *rn;
747 1.2.4.2 yamt ip_pool_node_t *m;
748 1.2.4.2 yamt i6addr_t *addr;
749 1.2.4.2 yamt addrfamily_t v;
750 1.2.4.2 yamt ip_pool_t *ipo;
751 1.2.4.2 yamt int rv;
752 1.2.4.2 yamt
753 1.2.4.2 yamt ipo = tptr;
754 1.2.4.2 yamt if (ipo == NULL)
755 1.2.4.2 yamt return -1;
756 1.2.4.2 yamt
757 1.2.4.2 yamt rv = 1;
758 1.2.4.2 yamt m = NULL;
759 1.2.4.2 yamt addr = (i6addr_t *)dptr;
760 1.2.4.2 yamt bzero(&v, sizeof(v));
761 1.2.4.2 yamt
762 1.2.4.2 yamt if (ipversion == 4) {
763 1.2.4.2 yamt v.adf_family = AF_INET;
764 1.2.4.3 yamt v.adf_len = offsetof(addrfamily_t, adf_addr) +
765 1.2.4.3 yamt sizeof(struct in_addr);
766 1.2.4.2 yamt v.adf_addr.in4 = addr->in4;
767 1.2.4.2 yamt #ifdef USE_INET6
768 1.2.4.2 yamt } else if (ipversion == 6) {
769 1.2.4.2 yamt v.adf_family = AF_INET6;
770 1.2.4.3 yamt v.adf_len = offsetof(addrfamily_t, adf_addr) +
771 1.2.4.3 yamt sizeof(struct in6_addr);
772 1.2.4.2 yamt v.adf_addr.in6 = addr->in6;
773 1.2.4.2 yamt #endif
774 1.2.4.2 yamt } else
775 1.2.4.2 yamt return -1;
776 1.2.4.2 yamt
777 1.2.4.2 yamt READ_ENTER(&softc->ipf_poolrw);
778 1.2.4.2 yamt
779 1.2.4.2 yamt rn = ipo->ipo_head->matchaddr(ipo->ipo_head, &v);
780 1.2.4.2 yamt
781 1.2.4.2 yamt if ((rn != NULL) && (rn->root == 0)) {
782 1.2.4.2 yamt m = (ip_pool_node_t *)rn;
783 1.2.4.2 yamt ipo->ipo_hits++;
784 1.2.4.2 yamt m->ipn_bytes += bytes;
785 1.2.4.2 yamt m->ipn_hits++;
786 1.2.4.2 yamt rv = m->ipn_info;
787 1.2.4.2 yamt }
788 1.2.4.2 yamt RWLOCK_EXIT(&softc->ipf_poolrw);
789 1.2.4.2 yamt return rv;
790 1.2.4.2 yamt }
791 1.2.4.2 yamt
792 1.2.4.2 yamt
793 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
794 1.2.4.2 yamt /* Function: ipf_pool_insert_node */
795 1.2.4.2 yamt /* Returns: int - 0 = success, else error */
796 1.2.4.2 yamt /* Parameters: softc(I) - pointer to soft context main structure */
797 1.2.4.2 yamt /* softp(I) - pointer to soft context pool information */
798 1.2.4.2 yamt /* ipo(I) - pointer to the pool getting the new node. */
799 1.2.4.2 yamt /* node(I) - structure with address/mask to add */
800 1.2.4.2 yamt /* Locks: WRITE(ipf_poolrw) */
801 1.2.4.2 yamt /* */
802 1.2.4.2 yamt /* Add another node to the pool given by ipo. The three parameters passed */
803 1.2.4.2 yamt /* in (addr, mask, info) shold all be stored in the node. */
804 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
805 1.2.4.2 yamt static int
806 1.2.4.2 yamt ipf_pool_insert_node(ipf_main_softc_t *softc, ipf_pool_softc_t *softp,
807 1.2.4.2 yamt ip_pool_t *ipo, struct ip_pool_node *node)
808 1.2.4.2 yamt {
809 1.2.4.2 yamt ipf_rdx_node_t *rn;
810 1.2.4.2 yamt ip_pool_node_t *x;
811 1.2.4.2 yamt
812 1.2.4.2 yamt if ((node->ipn_addr.adf_len > sizeof(*rn)) ||
813 1.2.4.2 yamt (node->ipn_addr.adf_len < 4)) {
814 1.2.4.2 yamt IPFERROR(70003);
815 1.2.4.2 yamt return EINVAL;
816 1.2.4.2 yamt }
817 1.2.4.2 yamt
818 1.2.4.2 yamt if ((node->ipn_mask.adf_len > sizeof(*rn)) ||
819 1.2.4.2 yamt (node->ipn_mask.adf_len < 4)) {
820 1.2.4.2 yamt IPFERROR(70004);
821 1.2.4.2 yamt return EINVAL;
822 1.2.4.2 yamt }
823 1.2.4.2 yamt
824 1.2.4.2 yamt KMALLOC(x, ip_pool_node_t *);
825 1.2.4.2 yamt if (x == NULL) {
826 1.2.4.2 yamt IPFERROR(70002);
827 1.2.4.2 yamt return ENOMEM;
828 1.2.4.2 yamt }
829 1.2.4.2 yamt
830 1.2.4.2 yamt *x = *node;
831 1.2.4.2 yamt bzero((char *)x->ipn_nodes, sizeof(x->ipn_nodes));
832 1.2.4.2 yamt x->ipn_owner = ipo;
833 1.2.4.2 yamt x->ipn_hits = 0;
834 1.2.4.2 yamt x->ipn_next = NULL;
835 1.2.4.2 yamt x->ipn_pnext = NULL;
836 1.2.4.2 yamt x->ipn_dnext = NULL;
837 1.2.4.2 yamt x->ipn_pdnext = NULL;
838 1.2.4.2 yamt
839 1.2.4.2 yamt if (x->ipn_die != 0) {
840 1.2.4.2 yamt /*
841 1.2.4.2 yamt * If the new node has a given expiration time, insert it
842 1.2.4.2 yamt * into the list of expiring nodes with the ones to be
843 1.2.4.2 yamt * removed first added to the front of the list. The
844 1.2.4.2 yamt * insertion is O(n) but it is kept sorted for quick scans
845 1.2.4.2 yamt * at expiration interval checks.
846 1.2.4.2 yamt */
847 1.2.4.2 yamt ip_pool_node_t *n;
848 1.2.4.2 yamt
849 1.2.4.2 yamt x->ipn_die = softc->ipf_ticks + IPF_TTLVAL(x->ipn_die);
850 1.2.4.2 yamt for (n = softp->ipf_node_explist; n != NULL; n = n->ipn_dnext) {
851 1.2.4.2 yamt if (x->ipn_die < n->ipn_die)
852 1.2.4.2 yamt break;
853 1.2.4.2 yamt if (n->ipn_dnext == NULL) {
854 1.2.4.2 yamt /*
855 1.2.4.2 yamt * We've got to the last node and everything
856 1.2.4.2 yamt * wanted to be expired before this new node,
857 1.2.4.2 yamt * so we have to tack it on the end...
858 1.2.4.2 yamt */
859 1.2.4.2 yamt n->ipn_dnext = x;
860 1.2.4.2 yamt x->ipn_pdnext = &n->ipn_dnext;
861 1.2.4.2 yamt n = NULL;
862 1.2.4.2 yamt break;
863 1.2.4.2 yamt }
864 1.2.4.2 yamt }
865 1.2.4.2 yamt
866 1.2.4.2 yamt if (softp->ipf_node_explist == NULL) {
867 1.2.4.2 yamt softp->ipf_node_explist = x;
868 1.2.4.2 yamt x->ipn_pdnext = &softp->ipf_node_explist;
869 1.2.4.2 yamt } else if (n != NULL) {
870 1.2.4.2 yamt x->ipn_dnext = n;
871 1.2.4.2 yamt x->ipn_pdnext = n->ipn_pdnext;
872 1.2.4.2 yamt n->ipn_pdnext = &x->ipn_dnext;
873 1.2.4.2 yamt }
874 1.2.4.2 yamt }
875 1.2.4.2 yamt
876 1.2.4.2 yamt rn = ipo->ipo_head->addaddr(ipo->ipo_head, &x->ipn_addr, &x->ipn_mask,
877 1.2.4.2 yamt x->ipn_nodes);
878 1.2.4.2 yamt #ifdef DEBUG_POOL
879 1.2.4.2 yamt printf("Added %p at %p\n", x, rn);
880 1.2.4.2 yamt #endif
881 1.2.4.2 yamt
882 1.2.4.2 yamt if (rn == NULL) {
883 1.2.4.2 yamt KFREE(x);
884 1.2.4.2 yamt IPFERROR(70005);
885 1.2.4.2 yamt return ENOMEM;
886 1.2.4.2 yamt }
887 1.2.4.2 yamt
888 1.2.4.2 yamt x->ipn_ref = 1;
889 1.2.4.3 yamt x->ipn_pnext = ipo->ipo_tail;
890 1.2.4.3 yamt *ipo->ipo_tail = x;
891 1.2.4.3 yamt ipo->ipo_tail = &x->ipn_next;
892 1.2.4.2 yamt
893 1.2.4.2 yamt softp->ipf_pool_stats.ipls_nodes++;
894 1.2.4.2 yamt
895 1.2.4.2 yamt return 0;
896 1.2.4.2 yamt }
897 1.2.4.2 yamt
898 1.2.4.2 yamt
899 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
900 1.2.4.2 yamt /* Function: ipf_pool_create */
901 1.2.4.2 yamt /* Returns: int - 0 = success, else error */
902 1.2.4.2 yamt /* Parameters: softc(I) - pointer to soft context main structure */
903 1.2.4.2 yamt /* softp(I) - pointer to soft context pool information */
904 1.2.4.2 yamt /* op(I) - pointer to iplookup struct with call details */
905 1.2.4.2 yamt /* Locks: WRITE(ipf_poolrw) */
906 1.2.4.2 yamt /* */
907 1.2.4.2 yamt /* Creates a new group according to the paramters passed in via the */
908 1.2.4.2 yamt /* iplookupop structure. Does not check to see if the group already exists */
909 1.2.4.2 yamt /* when being inserted - assume this has already been done. If the pool is */
910 1.2.4.2 yamt /* marked as being anonymous, give it a new, unique, identifier. Call any */
911 1.2.4.2 yamt /* other functions required to initialise the structure. */
912 1.2.4.2 yamt /* */
913 1.2.4.2 yamt /* If the structure is flagged for deletion then reset the flag and return, */
914 1.2.4.2 yamt /* as this likely means we've tried to free a pool that is in use (flush) */
915 1.2.4.2 yamt /* and now want to repopulate it with "new" data. */
916 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
917 1.2.4.2 yamt static int
918 1.2.4.2 yamt ipf_pool_create(ipf_main_softc_t *softc, ipf_pool_softc_t *softp,
919 1.2.4.2 yamt iplookupop_t *op)
920 1.2.4.2 yamt {
921 1.2.4.2 yamt char name[FR_GROUPLEN];
922 1.2.4.2 yamt int poolnum, unit;
923 1.2.4.2 yamt ip_pool_t *h;
924 1.2.4.2 yamt
925 1.2.4.2 yamt unit = op->iplo_unit;
926 1.2.4.2 yamt
927 1.2.4.2 yamt if ((op->iplo_arg & LOOKUP_ANON) == 0) {
928 1.2.4.2 yamt h = ipf_pool_exists(softp, unit, op->iplo_name);
929 1.2.4.2 yamt if (h != NULL) {
930 1.2.4.2 yamt if ((h->ipo_flags & IPOOL_DELETE) == 0) {
931 1.2.4.2 yamt IPFERROR(70006);
932 1.2.4.2 yamt return EEXIST;
933 1.2.4.2 yamt }
934 1.2.4.2 yamt h->ipo_flags &= ~IPOOL_DELETE;
935 1.2.4.2 yamt return 0;
936 1.2.4.2 yamt }
937 1.2.4.2 yamt }
938 1.2.4.2 yamt
939 1.2.4.2 yamt KMALLOC(h, ip_pool_t *);
940 1.2.4.2 yamt if (h == NULL) {
941 1.2.4.2 yamt IPFERROR(70007);
942 1.2.4.2 yamt return ENOMEM;
943 1.2.4.2 yamt }
944 1.2.4.2 yamt bzero(h, sizeof(*h));
945 1.2.4.2 yamt
946 1.2.4.2 yamt if (ipf_rx_inithead(softp->ipf_radix, &h->ipo_head) != 0) {
947 1.2.4.2 yamt KFREE(h);
948 1.2.4.2 yamt IPFERROR(70008);
949 1.2.4.2 yamt return ENOMEM;
950 1.2.4.2 yamt }
951 1.2.4.2 yamt
952 1.2.4.2 yamt if ((op->iplo_arg & LOOKUP_ANON) != 0) {
953 1.2.4.2 yamt ip_pool_t *p;
954 1.2.4.2 yamt
955 1.2.4.2 yamt h->ipo_flags |= IPOOL_ANON;
956 1.2.4.2 yamt poolnum = LOOKUP_ANON;
957 1.2.4.2 yamt
958 1.2.4.4 yamt snprintf(name, sizeof(name), "%x", poolnum);
959 1.2.4.2 yamt
960 1.2.4.2 yamt for (p = softp->ipf_pool_list[unit + 1]; p != NULL; ) {
961 1.2.4.2 yamt if (strncmp(name, p->ipo_name,
962 1.2.4.2 yamt sizeof(p->ipo_name)) == 0) {
963 1.2.4.2 yamt poolnum++;
964 1.2.4.4 yamt snprintf(name, sizeof(name), "%x", poolnum);
965 1.2.4.2 yamt p = softp->ipf_pool_list[unit + 1];
966 1.2.4.2 yamt } else
967 1.2.4.2 yamt p = p->ipo_next;
968 1.2.4.2 yamt }
969 1.2.4.2 yamt
970 1.2.4.2 yamt (void)strncpy(h->ipo_name, name, sizeof(h->ipo_name));
971 1.2.4.2 yamt (void)strncpy(op->iplo_name, name, sizeof(op->iplo_name));
972 1.2.4.2 yamt } else {
973 1.2.4.2 yamt (void)strncpy(h->ipo_name, op->iplo_name, sizeof(h->ipo_name));
974 1.2.4.2 yamt }
975 1.2.4.2 yamt
976 1.2.4.2 yamt h->ipo_radix = softp->ipf_radix;
977 1.2.4.2 yamt h->ipo_ref = 1;
978 1.2.4.2 yamt h->ipo_list = NULL;
979 1.2.4.3 yamt h->ipo_tail = &h->ipo_list;
980 1.2.4.2 yamt h->ipo_unit = unit;
981 1.2.4.2 yamt h->ipo_next = softp->ipf_pool_list[unit + 1];
982 1.2.4.2 yamt if (softp->ipf_pool_list[unit + 1] != NULL)
983 1.2.4.2 yamt softp->ipf_pool_list[unit + 1]->ipo_pnext = &h->ipo_next;
984 1.2.4.2 yamt h->ipo_pnext = &softp->ipf_pool_list[unit + 1];
985 1.2.4.2 yamt softp->ipf_pool_list[unit + 1] = h;
986 1.2.4.2 yamt
987 1.2.4.2 yamt softp->ipf_pool_stats.ipls_pools++;
988 1.2.4.2 yamt
989 1.2.4.2 yamt return 0;
990 1.2.4.2 yamt }
991 1.2.4.2 yamt
992 1.2.4.2 yamt
993 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
994 1.2.4.2 yamt /* Function: ipf_pool_remove_node */
995 1.2.4.2 yamt /* Returns: int - 0 = success, else error */
996 1.2.4.2 yamt /* Parameters: softc(I) - pointer to soft context main structure */
997 1.2.4.2 yamt /* ipo(I) - pointer to the pool to remove the node from. */
998 1.2.4.2 yamt /* ipe(I) - address being deleted as a node */
999 1.2.4.2 yamt /* Locks: WRITE(ipf_poolrw) */
1000 1.2.4.2 yamt /* */
1001 1.2.4.2 yamt /* Remove a node from the pool given by ipo. */
1002 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
1003 1.2.4.2 yamt static int
1004 1.2.4.3 yamt ipf_pool_remove_node(ipf_main_softc_t *softc, ipf_pool_softc_t *softp,
1005 1.2.4.3 yamt ip_pool_t *ipo, ip_pool_node_t *ipe)
1006 1.2.4.2 yamt {
1007 1.2.4.3 yamt void *ptr;
1008 1.2.4.3 yamt
1009 1.2.4.3 yamt if (ipo->ipo_tail == &ipe->ipn_next)
1010 1.2.4.3 yamt ipo->ipo_tail = ipe->ipn_pnext;
1011 1.2.4.2 yamt
1012 1.2.4.2 yamt if (ipe->ipn_pnext != NULL)
1013 1.2.4.2 yamt *ipe->ipn_pnext = ipe->ipn_next;
1014 1.2.4.2 yamt if (ipe->ipn_next != NULL)
1015 1.2.4.2 yamt ipe->ipn_next->ipn_pnext = ipe->ipn_pnext;
1016 1.2.4.2 yamt
1017 1.2.4.2 yamt if (ipe->ipn_pdnext != NULL)
1018 1.2.4.2 yamt *ipe->ipn_pdnext = ipe->ipn_dnext;
1019 1.2.4.2 yamt if (ipe->ipn_dnext != NULL)
1020 1.2.4.2 yamt ipe->ipn_dnext->ipn_pdnext = ipe->ipn_pdnext;
1021 1.2.4.2 yamt
1022 1.2.4.3 yamt ptr = ipo->ipo_head->deladdr(ipo->ipo_head, &ipe->ipn_addr,
1023 1.2.4.3 yamt &ipe->ipn_mask);
1024 1.2.4.2 yamt
1025 1.2.4.3 yamt if (ptr != NULL) {
1026 1.2.4.3 yamt ipf_pool_node_deref(softp, ipe);
1027 1.2.4.3 yamt return 0;
1028 1.2.4.3 yamt }
1029 1.2.4.3 yamt IPFERROR(70027);
1030 1.2.4.3 yamt return ESRCH;
1031 1.2.4.2 yamt }
1032 1.2.4.2 yamt
1033 1.2.4.2 yamt
1034 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
1035 1.2.4.2 yamt /* Function: ipf_pool_destroy */
1036 1.2.4.2 yamt /* Returns: int - 0 = success, else error */
1037 1.2.4.2 yamt /* Parameters: softc(I) - pointer to soft context main structure */
1038 1.2.4.2 yamt /* softp(I) - pointer to soft context pool information */
1039 1.2.4.2 yamt /* unit(I) - ipfilter device to which we are working on */
1040 1.2.4.2 yamt /* name(I) - name of the pool */
1041 1.2.4.2 yamt /* Locks: WRITE(ipf_poolrw) or WRITE(ipf_global) */
1042 1.2.4.2 yamt /* */
1043 1.2.4.2 yamt /* Search for a pool using paramters passed in and if it's not otherwise */
1044 1.2.4.2 yamt /* busy, free it. If it is busy, clear all of its nodes, mark it for being */
1045 1.2.4.2 yamt /* deleted and return an error saying it is busy. */
1046 1.2.4.2 yamt /* */
1047 1.2.4.2 yamt /* NOTE: Because this function is called out of ipfdetach() where ipf_poolrw*/
1048 1.2.4.2 yamt /* may not be initialised, we can't use an ASSERT to enforce the locking */
1049 1.2.4.2 yamt /* assertion that one of the two (ipf_poolrw,ipf_global) is held. */
1050 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
1051 1.2.4.2 yamt static int
1052 1.2.4.2 yamt ipf_pool_destroy(ipf_main_softc_t *softc, ipf_pool_softc_t *softp, int unit,
1053 1.2.4.2 yamt char *name)
1054 1.2.4.2 yamt {
1055 1.2.4.2 yamt ip_pool_t *ipo;
1056 1.2.4.2 yamt
1057 1.2.4.2 yamt ipo = ipf_pool_exists(softp, unit, name);
1058 1.2.4.2 yamt if (ipo == NULL) {
1059 1.2.4.2 yamt IPFERROR(70009);
1060 1.2.4.2 yamt return ESRCH;
1061 1.2.4.2 yamt }
1062 1.2.4.2 yamt
1063 1.2.4.2 yamt if (ipo->ipo_ref != 1) {
1064 1.2.4.3 yamt ipf_pool_clearnodes(softc, softp, ipo);
1065 1.2.4.2 yamt ipo->ipo_flags |= IPOOL_DELETE;
1066 1.2.4.2 yamt return 0;
1067 1.2.4.2 yamt }
1068 1.2.4.2 yamt
1069 1.2.4.3 yamt ipf_pool_free(softc, softp, ipo);
1070 1.2.4.2 yamt return 0;
1071 1.2.4.2 yamt }
1072 1.2.4.2 yamt
1073 1.2.4.2 yamt
1074 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
1075 1.2.4.2 yamt /* Function: ipf_pool_flush */
1076 1.2.4.2 yamt /* Returns: int - number of pools deleted */
1077 1.2.4.2 yamt /* Parameters: softc(I) - pointer to soft context main structure */
1078 1.2.4.2 yamt /* arg(I) - pointer to local context to use */
1079 1.2.4.2 yamt /* fp(I) - which pool(s) to flush */
1080 1.2.4.2 yamt /* Locks: WRITE(ipf_poolrw) or WRITE(ipf_global) */
1081 1.2.4.2 yamt /* */
1082 1.2.4.2 yamt /* Free all pools associated with the device that matches the unit number */
1083 1.2.4.2 yamt /* passed in with operation. */
1084 1.2.4.2 yamt /* */
1085 1.2.4.2 yamt /* NOTE: Because this function is called out of ipfdetach() where ipf_poolrw*/
1086 1.2.4.2 yamt /* may not be initialised, we can't use an ASSERT to enforce the locking */
1087 1.2.4.2 yamt /* assertion that one of the two (ipf_poolrw,ipf_global) is held. */
1088 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
1089 1.2.4.2 yamt static size_t
1090 1.2.4.2 yamt ipf_pool_flush(ipf_main_softc_t *softc, void *arg, iplookupflush_t *fp)
1091 1.2.4.2 yamt {
1092 1.2.4.2 yamt ipf_pool_softc_t *softp = arg;
1093 1.2.4.2 yamt int i, num = 0, unit, err;
1094 1.2.4.2 yamt ip_pool_t *p, *q;
1095 1.2.4.2 yamt
1096 1.2.4.2 yamt unit = fp->iplf_unit;
1097 1.2.4.2 yamt for (i = -1; i <= IPL_LOGMAX; i++) {
1098 1.2.4.2 yamt if (unit != IPLT_ALL && i != unit)
1099 1.2.4.2 yamt continue;
1100 1.2.4.2 yamt for (q = softp->ipf_pool_list[i + 1]; (p = q) != NULL; ) {
1101 1.2.4.2 yamt q = p->ipo_next;
1102 1.2.4.2 yamt err = ipf_pool_destroy(softc, softp, i, p->ipo_name);
1103 1.2.4.2 yamt if (err == 0)
1104 1.2.4.2 yamt num++;
1105 1.2.4.2 yamt }
1106 1.2.4.2 yamt }
1107 1.2.4.2 yamt return num;
1108 1.2.4.2 yamt }
1109 1.2.4.2 yamt
1110 1.2.4.2 yamt
1111 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
1112 1.2.4.2 yamt /* Function: ipf_pool_free */
1113 1.2.4.2 yamt /* Returns: void */
1114 1.2.4.3 yamt /* Parameters: softc(I) - pointer to soft context main structure */
1115 1.2.4.3 yamt /* softp(I) - pointer to soft context pool information */
1116 1.2.4.2 yamt /* ipo(I) - pointer to pool structure */
1117 1.2.4.2 yamt /* Locks: WRITE(ipf_poolrw) or WRITE(ipf_global) */
1118 1.2.4.2 yamt /* */
1119 1.2.4.2 yamt /* Deletes the pool strucutre passed in from the list of pools and deletes */
1120 1.2.4.2 yamt /* all of the address information stored in it, including any tree data */
1121 1.2.4.2 yamt /* structures also allocated. */
1122 1.2.4.2 yamt /* */
1123 1.2.4.2 yamt /* NOTE: Because this function is called out of ipfdetach() where ipf_poolrw*/
1124 1.2.4.2 yamt /* may not be initialised, we can't use an ASSERT to enforce the locking */
1125 1.2.4.2 yamt /* assertion that one of the two (ipf_poolrw,ipf_global) is held. */
1126 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
1127 1.2.4.2 yamt static void
1128 1.2.4.3 yamt ipf_pool_free(ipf_main_softc_t *softc, ipf_pool_softc_t *softp, ip_pool_t *ipo)
1129 1.2.4.2 yamt {
1130 1.2.4.2 yamt
1131 1.2.4.3 yamt ipf_pool_clearnodes(softc, softp, ipo);
1132 1.2.4.2 yamt
1133 1.2.4.2 yamt if (ipo->ipo_next != NULL)
1134 1.2.4.2 yamt ipo->ipo_next->ipo_pnext = ipo->ipo_pnext;
1135 1.2.4.2 yamt *ipo->ipo_pnext = ipo->ipo_next;
1136 1.2.4.2 yamt ipf_rx_freehead(ipo->ipo_head);
1137 1.2.4.2 yamt KFREE(ipo);
1138 1.2.4.2 yamt
1139 1.2.4.2 yamt softp->ipf_pool_stats.ipls_pools--;
1140 1.2.4.2 yamt }
1141 1.2.4.2 yamt
1142 1.2.4.2 yamt
1143 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
1144 1.2.4.2 yamt /* Function: ipf_pool_clearnodes */
1145 1.2.4.2 yamt /* Returns: void */
1146 1.2.4.3 yamt /* Parameters: softc(I) - pointer to soft context main structure */
1147 1.2.4.3 yamt /* softp(I) - pointer to soft context pool information */
1148 1.2.4.2 yamt /* ipo(I) - pointer to pool structure */
1149 1.2.4.2 yamt /* Locks: WRITE(ipf_poolrw) or WRITE(ipf_global) */
1150 1.2.4.2 yamt /* */
1151 1.2.4.2 yamt /* Deletes all nodes stored in a pool structure. */
1152 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
1153 1.2.4.2 yamt static void
1154 1.2.4.3 yamt ipf_pool_clearnodes(ipf_main_softc_t *softc, ipf_pool_softc_t *softp,
1155 1.2.4.3 yamt ip_pool_t *ipo)
1156 1.2.4.2 yamt {
1157 1.2.4.2 yamt ip_pool_node_t *n, **next;
1158 1.2.4.2 yamt
1159 1.2.4.3 yamt for (next = &ipo->ipo_list; (n = *next) != NULL; )
1160 1.2.4.3 yamt ipf_pool_remove_node(softc, softp, ipo, n);
1161 1.2.4.2 yamt
1162 1.2.4.2 yamt ipo->ipo_list = NULL;
1163 1.2.4.2 yamt }
1164 1.2.4.2 yamt
1165 1.2.4.2 yamt
1166 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
1167 1.2.4.2 yamt /* Function: ipf_pool_deref */
1168 1.2.4.2 yamt /* Returns: void */
1169 1.2.4.2 yamt /* Parameters: softc(I) - pointer to soft context main structure */
1170 1.2.4.2 yamt /* arg(I) - pointer to local context to use */
1171 1.2.4.2 yamt /* pool(I) - pointer to pool structure */
1172 1.2.4.2 yamt /* Locks: WRITE(ipf_poolrw) */
1173 1.2.4.2 yamt /* */
1174 1.2.4.2 yamt /* Drop the number of known references to this pool structure by one and if */
1175 1.2.4.2 yamt /* we arrive at zero known references, free it. */
1176 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
1177 1.2.4.2 yamt static int
1178 1.2.4.2 yamt ipf_pool_deref(ipf_main_softc_t *softc, void *arg, void *pool)
1179 1.2.4.2 yamt {
1180 1.2.4.2 yamt ip_pool_t *ipo = pool;
1181 1.2.4.2 yamt
1182 1.2.4.2 yamt ipo->ipo_ref--;
1183 1.2.4.2 yamt
1184 1.2.4.2 yamt if (ipo->ipo_ref == 0)
1185 1.2.4.3 yamt ipf_pool_free(softc, arg, ipo);
1186 1.2.4.2 yamt
1187 1.2.4.2 yamt else if ((ipo->ipo_ref == 1) && (ipo->ipo_flags & IPOOL_DELETE))
1188 1.2.4.2 yamt ipf_pool_destroy(softc, arg, ipo->ipo_unit, ipo->ipo_name);
1189 1.2.4.2 yamt
1190 1.2.4.2 yamt return 0;
1191 1.2.4.2 yamt }
1192 1.2.4.2 yamt
1193 1.2.4.2 yamt
1194 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
1195 1.2.4.2 yamt /* Function: ipf_pool_node_deref */
1196 1.2.4.2 yamt /* Returns: void */
1197 1.2.4.2 yamt /* Parameters: softp(I) - pointer to soft context pool information */
1198 1.2.4.2 yamt /* ipn(I) - pointer to pool structure */
1199 1.2.4.2 yamt /* Locks: WRITE(ipf_poolrw) */
1200 1.2.4.2 yamt /* */
1201 1.2.4.2 yamt /* Drop a reference to the pool node passed in and if we're the last, free */
1202 1.2.4.2 yamt /* it all up and adjust the stats accordingly. */
1203 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
1204 1.2.4.2 yamt static void
1205 1.2.4.2 yamt ipf_pool_node_deref(ipf_pool_softc_t *softp, ip_pool_node_t *ipn)
1206 1.2.4.2 yamt {
1207 1.2.4.2 yamt
1208 1.2.4.2 yamt ipn->ipn_ref--;
1209 1.2.4.2 yamt
1210 1.2.4.2 yamt if (ipn->ipn_ref == 0) {
1211 1.2.4.2 yamt KFREE(ipn);
1212 1.2.4.2 yamt softp->ipf_pool_stats.ipls_nodes--;
1213 1.2.4.2 yamt }
1214 1.2.4.2 yamt }
1215 1.2.4.2 yamt
1216 1.2.4.2 yamt
1217 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
1218 1.2.4.2 yamt /* Function: ipf_pool_iter_next */
1219 1.2.4.2 yamt /* Returns: void */
1220 1.2.4.2 yamt /* Parameters: softc(I) - pointer to soft context main structure */
1221 1.2.4.2 yamt /* arg(I) - pointer to local context to use */
1222 1.2.4.2 yamt /* token(I) - pointer to pool structure */
1223 1.2.4.2 yamt /* ilp(IO) - pointer to pool iterating structure */
1224 1.2.4.2 yamt /* */
1225 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
1226 1.2.4.2 yamt static int
1227 1.2.4.2 yamt ipf_pool_iter_next(ipf_main_softc_t *softc, void *arg, ipftoken_t *token,
1228 1.2.4.2 yamt ipflookupiter_t *ilp)
1229 1.2.4.2 yamt {
1230 1.2.4.2 yamt ipf_pool_softc_t *softp = arg;
1231 1.2.4.2 yamt ip_pool_node_t *node, zn, *nextnode;
1232 1.2.4.2 yamt ip_pool_t *ipo, zp, *nextipo;
1233 1.2.4.2 yamt void *pnext;
1234 1.2.4.2 yamt int err;
1235 1.2.4.2 yamt
1236 1.2.4.2 yamt err = 0;
1237 1.2.4.2 yamt node = NULL;
1238 1.2.4.2 yamt nextnode = NULL;
1239 1.2.4.2 yamt ipo = NULL;
1240 1.2.4.2 yamt nextipo = NULL;
1241 1.2.4.2 yamt
1242 1.2.4.2 yamt READ_ENTER(&softc->ipf_poolrw);
1243 1.2.4.2 yamt
1244 1.2.4.2 yamt switch (ilp->ili_otype)
1245 1.2.4.2 yamt {
1246 1.2.4.2 yamt case IPFLOOKUPITER_LIST :
1247 1.2.4.2 yamt ipo = token->ipt_data;
1248 1.2.4.2 yamt if (ipo == NULL) {
1249 1.2.4.2 yamt nextipo = softp->ipf_pool_list[(int)ilp->ili_unit + 1];
1250 1.2.4.2 yamt } else {
1251 1.2.4.2 yamt nextipo = ipo->ipo_next;
1252 1.2.4.2 yamt }
1253 1.2.4.2 yamt
1254 1.2.4.2 yamt if (nextipo != NULL) {
1255 1.2.4.2 yamt ATOMIC_INC32(nextipo->ipo_ref);
1256 1.2.4.2 yamt token->ipt_data = nextipo;
1257 1.2.4.2 yamt } else {
1258 1.2.4.2 yamt bzero((char *)&zp, sizeof(zp));
1259 1.2.4.2 yamt nextipo = &zp;
1260 1.2.4.2 yamt token->ipt_data = NULL;
1261 1.2.4.2 yamt }
1262 1.2.4.2 yamt pnext = nextipo->ipo_next;
1263 1.2.4.2 yamt break;
1264 1.2.4.2 yamt
1265 1.2.4.2 yamt case IPFLOOKUPITER_NODE :
1266 1.2.4.2 yamt node = token->ipt_data;
1267 1.2.4.2 yamt if (node == NULL) {
1268 1.2.4.2 yamt ipo = ipf_pool_exists(arg, ilp->ili_unit,
1269 1.2.4.2 yamt ilp->ili_name);
1270 1.2.4.2 yamt if (ipo == NULL) {
1271 1.2.4.2 yamt IPFERROR(70010);
1272 1.2.4.2 yamt err = ESRCH;
1273 1.2.4.2 yamt } else {
1274 1.2.4.2 yamt nextnode = ipo->ipo_list;
1275 1.2.4.2 yamt ipo = NULL;
1276 1.2.4.2 yamt }
1277 1.2.4.2 yamt } else {
1278 1.2.4.2 yamt nextnode = node->ipn_next;
1279 1.2.4.2 yamt }
1280 1.2.4.2 yamt
1281 1.2.4.2 yamt if (nextnode != NULL) {
1282 1.2.4.2 yamt ATOMIC_INC32(nextnode->ipn_ref);
1283 1.2.4.2 yamt token->ipt_data = nextnode;
1284 1.2.4.2 yamt } else {
1285 1.2.4.2 yamt bzero((char *)&zn, sizeof(zn));
1286 1.2.4.2 yamt nextnode = &zn;
1287 1.2.4.2 yamt token->ipt_data = NULL;
1288 1.2.4.2 yamt }
1289 1.2.4.2 yamt pnext = nextnode->ipn_next;
1290 1.2.4.2 yamt break;
1291 1.2.4.2 yamt
1292 1.2.4.2 yamt default :
1293 1.2.4.2 yamt IPFERROR(70011);
1294 1.2.4.2 yamt pnext = NULL;
1295 1.2.4.2 yamt err = EINVAL;
1296 1.2.4.2 yamt break;
1297 1.2.4.2 yamt }
1298 1.2.4.2 yamt
1299 1.2.4.2 yamt RWLOCK_EXIT(&softc->ipf_poolrw);
1300 1.2.4.2 yamt if (err != 0)
1301 1.2.4.2 yamt return err;
1302 1.2.4.2 yamt
1303 1.2.4.2 yamt switch (ilp->ili_otype)
1304 1.2.4.2 yamt {
1305 1.2.4.2 yamt case IPFLOOKUPITER_LIST :
1306 1.2.4.2 yamt err = COPYOUT(nextipo, ilp->ili_data, sizeof(*nextipo));
1307 1.2.4.2 yamt if (err != 0) {
1308 1.2.4.2 yamt IPFERROR(70012);
1309 1.2.4.2 yamt err = EFAULT;
1310 1.2.4.2 yamt }
1311 1.2.4.2 yamt if (ipo != NULL) {
1312 1.2.4.2 yamt WRITE_ENTER(&softc->ipf_poolrw);
1313 1.2.4.2 yamt ipf_pool_deref(softc, softp, ipo);
1314 1.2.4.2 yamt RWLOCK_EXIT(&softc->ipf_poolrw);
1315 1.2.4.2 yamt }
1316 1.2.4.2 yamt break;
1317 1.2.4.2 yamt
1318 1.2.4.2 yamt case IPFLOOKUPITER_NODE :
1319 1.2.4.2 yamt err = COPYOUT(nextnode, ilp->ili_data, sizeof(*nextnode));
1320 1.2.4.2 yamt if (err != 0) {
1321 1.2.4.2 yamt IPFERROR(70013);
1322 1.2.4.2 yamt err = EFAULT;
1323 1.2.4.2 yamt }
1324 1.2.4.2 yamt if (node != NULL) {
1325 1.2.4.2 yamt WRITE_ENTER(&softc->ipf_poolrw);
1326 1.2.4.2 yamt ipf_pool_node_deref(softp, node);
1327 1.2.4.2 yamt RWLOCK_EXIT(&softc->ipf_poolrw);
1328 1.2.4.2 yamt }
1329 1.2.4.2 yamt break;
1330 1.2.4.2 yamt }
1331 1.2.4.2 yamt if (pnext == NULL)
1332 1.2.4.2 yamt ipf_token_mark_complete(token);
1333 1.2.4.2 yamt
1334 1.2.4.2 yamt return err;
1335 1.2.4.2 yamt }
1336 1.2.4.2 yamt
1337 1.2.4.2 yamt
1338 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
1339 1.2.4.2 yamt /* Function: ipf_pool_iterderef */
1340 1.2.4.2 yamt /* Returns: void */
1341 1.2.4.2 yamt /* Parameters: softc(I) - pointer to soft context main structure */
1342 1.2.4.2 yamt /* arg(I) - pointer to local context to use */
1343 1.2.4.2 yamt /* unit(I) - ipfilter device to which we are working on */
1344 1.2.4.2 yamt /* Locks: WRITE(ipf_poolrw) */
1345 1.2.4.2 yamt /* */
1346 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
1347 1.2.4.2 yamt static int
1348 1.2.4.2 yamt ipf_pool_iter_deref(ipf_main_softc_t *softc, void *arg, int otype, int unit,
1349 1.2.4.2 yamt void *data)
1350 1.2.4.2 yamt {
1351 1.2.4.2 yamt ipf_pool_softc_t *softp = arg;
1352 1.2.4.2 yamt
1353 1.2.4.2 yamt if (data == NULL)
1354 1.2.4.2 yamt return EINVAL;
1355 1.2.4.2 yamt
1356 1.2.4.2 yamt if (unit < 0 || unit > IPL_LOGMAX)
1357 1.2.4.2 yamt return EINVAL;
1358 1.2.4.2 yamt
1359 1.2.4.2 yamt switch (otype)
1360 1.2.4.2 yamt {
1361 1.2.4.2 yamt case IPFLOOKUPITER_LIST :
1362 1.2.4.2 yamt ipf_pool_deref(softc, softp, (ip_pool_t *)data);
1363 1.2.4.2 yamt break;
1364 1.2.4.2 yamt
1365 1.2.4.2 yamt case IPFLOOKUPITER_NODE :
1366 1.2.4.2 yamt ipf_pool_node_deref(softp, (ip_pool_node_t *)data);
1367 1.2.4.2 yamt break;
1368 1.2.4.2 yamt default :
1369 1.2.4.2 yamt break;
1370 1.2.4.2 yamt }
1371 1.2.4.2 yamt
1372 1.2.4.2 yamt return 0;
1373 1.2.4.2 yamt }
1374 1.2.4.2 yamt
1375 1.2.4.2 yamt
1376 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
1377 1.2.4.2 yamt /* Function: ipf_pool_expire */
1378 1.2.4.2 yamt /* Returns: Nil */
1379 1.2.4.2 yamt /* Parameters: softc(I) - pointer to soft context main structure */
1380 1.2.4.2 yamt /* arg(I) - pointer to local context to use */
1381 1.2.4.2 yamt /* */
1382 1.2.4.2 yamt /* At present this function exists just to support temporary addition of */
1383 1.2.4.2 yamt /* nodes to the address pool. */
1384 1.2.4.2 yamt /* ------------------------------------------------------------------------ */
1385 1.2.4.2 yamt static void
1386 1.2.4.2 yamt ipf_pool_expire(ipf_main_softc_t *softc, void *arg)
1387 1.2.4.2 yamt {
1388 1.2.4.2 yamt ipf_pool_softc_t *softp = arg;
1389 1.2.4.2 yamt ip_pool_node_t *n;
1390 1.2.4.2 yamt
1391 1.2.4.2 yamt while ((n = softp->ipf_node_explist) != NULL) {
1392 1.2.4.2 yamt /*
1393 1.2.4.2 yamt * Because the list is kept sorted on insertion, the fist
1394 1.2.4.2 yamt * one that dies in the future means no more work to do.
1395 1.2.4.2 yamt */
1396 1.2.4.2 yamt if (n->ipn_die > softc->ipf_ticks)
1397 1.2.4.2 yamt break;
1398 1.2.4.3 yamt ipf_pool_remove_node(softc, softp, n->ipn_owner, n);
1399 1.2.4.2 yamt }
1400 1.2.4.2 yamt }
1401 1.2.4.2 yamt
1402 1.2.4.2 yamt
1403 1.2.4.2 yamt
1404 1.2.4.2 yamt
1405 1.2.4.2 yamt #ifndef _KERNEL
1406 1.2.4.2 yamt void
1407 1.2.4.2 yamt ipf_pool_dump(softc, arg)
1408 1.2.4.2 yamt ipf_main_softc_t *softc;
1409 1.2.4.2 yamt void *arg;
1410 1.2.4.2 yamt {
1411 1.2.4.2 yamt ipf_pool_softc_t *softp = arg;
1412 1.2.4.2 yamt ip_pool_t *ipl;
1413 1.2.4.2 yamt int i;
1414 1.2.4.2 yamt
1415 1.2.4.2 yamt printf("List of configured pools\n");
1416 1.2.4.2 yamt for (i = 0; i <= LOOKUP_POOL_MAX; i++)
1417 1.2.4.2 yamt for (ipl = softp->ipf_pool_list[i]; ipl != NULL;
1418 1.2.4.2 yamt ipl = ipl->ipo_next)
1419 1.2.4.2 yamt printpool(ipl, bcopywrap, NULL, opts, NULL);
1420 1.2.4.2 yamt }
1421 1.2.4.2 yamt #endif
1422