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