radix.c revision 1.41 1 1.41 dsl /* $NetBSD: radix.c,v 1.41 2009/03/14 15:36:23 dsl Exp $ */
2 1.7 cgd
3 1.1 cgd /*
4 1.6 mycroft * Copyright (c) 1988, 1989, 1993
5 1.6 mycroft * The Regents of the University of California. All rights reserved.
6 1.1 cgd *
7 1.1 cgd * Redistribution and use in source and binary forms, with or without
8 1.1 cgd * modification, are permitted provided that the following conditions
9 1.1 cgd * are met:
10 1.1 cgd * 1. Redistributions of source code must retain the above copyright
11 1.1 cgd * notice, this list of conditions and the following disclaimer.
12 1.1 cgd * 2. Redistributions in binary form must reproduce the above copyright
13 1.1 cgd * notice, this list of conditions and the following disclaimer in the
14 1.1 cgd * documentation and/or other materials provided with the distribution.
15 1.20 agc * 3. Neither the name of the University nor the names of its contributors
16 1.1 cgd * may be used to endorse or promote products derived from this software
17 1.1 cgd * without specific prior written permission.
18 1.1 cgd *
19 1.1 cgd * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20 1.1 cgd * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 1.1 cgd * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 1.1 cgd * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23 1.1 cgd * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 1.1 cgd * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 1.1 cgd * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 1.1 cgd * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 1.1 cgd * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 1.1 cgd * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 1.1 cgd * SUCH DAMAGE.
30 1.1 cgd *
31 1.13 fvdl * @(#)radix.c 8.6 (Berkeley) 10/17/95
32 1.1 cgd */
33 1.1 cgd
34 1.1 cgd /*
35 1.1 cgd * Routines to build and maintain radix trees for routing lookups.
36 1.1 cgd */
37 1.18 lukem
38 1.18 lukem #include <sys/cdefs.h>
39 1.41 dsl __KERNEL_RCSID(0, "$NetBSD: radix.c,v 1.41 2009/03/14 15:36:23 dsl Exp $");
40 1.18 lukem
41 1.12 christos #ifndef _NET_RADIX_H_
42 1.4 mycroft #include <sys/param.h>
43 1.12 christos #ifdef _KERNEL
44 1.27 enami #include "opt_inet.h"
45 1.27 enami
46 1.4 mycroft #include <sys/systm.h>
47 1.4 mycroft #include <sys/malloc.h>
48 1.1 cgd #define M_DONTWAIT M_NOWAIT
49 1.6 mycroft #include <sys/domain.h>
50 1.9 mycroft #else
51 1.9 mycroft #include <stdlib.h>
52 1.6 mycroft #endif
53 1.32 dyoung #include <machine/stdarg.h>
54 1.9 mycroft #include <sys/syslog.h>
55 1.4 mycroft #include <net/radix.h>
56 1.12 christos #endif
57 1.6 mycroft
58 1.32 dyoung typedef void (*rn_printer_t)(void *, const char *fmt, ...);
59 1.32 dyoung
60 1.6 mycroft int max_keylen;
61 1.6 mycroft struct radix_mask *rn_mkfreelist;
62 1.1 cgd struct radix_node_head *mask_rnhead;
63 1.9 mycroft static char *addmask_key;
64 1.21 matt static const char normal_chars[] =
65 1.21 matt {0, 0x80, 0xc0, 0xe0, 0xf0, 0xf8, 0xfc, 0xfe, -1};
66 1.6 mycroft static char *rn_zeros, *rn_ones;
67 1.6 mycroft
68 1.6 mycroft #define rn_masktop (mask_rnhead->rnh_treetop)
69 1.10 christos
70 1.23 matt static int rn_satisfies_leaf(const char *, struct radix_node *, int);
71 1.23 matt static int rn_lexobetter(const void *, const void *);
72 1.23 matt static struct radix_mask *rn_new_radix_mask(struct radix_node *,
73 1.23 matt struct radix_mask *);
74 1.32 dyoung static struct radix_node *rn_walknext(struct radix_node *, rn_printer_t,
75 1.32 dyoung void *);
76 1.32 dyoung static struct radix_node *rn_walkfirst(struct radix_node *, rn_printer_t,
77 1.32 dyoung void *);
78 1.32 dyoung static void rn_nodeprint(struct radix_node *, rn_printer_t, void *,
79 1.32 dyoung const char *);
80 1.32 dyoung
81 1.32 dyoung #define SUBTREE_OPEN "[ "
82 1.32 dyoung #define SUBTREE_CLOSE " ]"
83 1.32 dyoung
84 1.32 dyoung #ifdef RN_DEBUG
85 1.32 dyoung static void rn_treeprint(struct radix_node_head *, rn_printer_t, void *);
86 1.32 dyoung #endif /* RN_DEBUG */
87 1.12 christos
88 1.1 cgd /*
89 1.1 cgd * The data structure for the keys is a radix tree with one way
90 1.1 cgd * branching removed. The index rn_b at an internal node n represents a bit
91 1.1 cgd * position to be tested. The tree is arranged so that all descendants
92 1.1 cgd * of a node n have keys whose bits all agree up to position rn_b - 1.
93 1.1 cgd * (We say the index of n is rn_b.)
94 1.1 cgd *
95 1.1 cgd * There is at least one descendant which has a one bit at position rn_b,
96 1.1 cgd * and at least one with a zero there.
97 1.1 cgd *
98 1.1 cgd * A route is determined by a pair of key and mask. We require that the
99 1.1 cgd * bit-wise logical and of the key and mask to be the key.
100 1.1 cgd * We define the index of a route to associated with the mask to be
101 1.1 cgd * the first bit number in the mask where 0 occurs (with bit number 0
102 1.1 cgd * representing the highest order bit).
103 1.28 perry *
104 1.1 cgd * We say a mask is normal if every bit is 0, past the index of the mask.
105 1.1 cgd * If a node n has a descendant (k, m) with index(m) == index(n) == rn_b,
106 1.1 cgd * and m is a normal mask, then the route applies to every descendant of n.
107 1.1 cgd * If the index(m) < rn_b, this implies the trailing last few bits of k
108 1.1 cgd * before bit b are all 0, (and hence consequently true of every descendant
109 1.1 cgd * of n), so the route applies to all descendants of the node as well.
110 1.28 perry *
111 1.9 mycroft * Similar logic shows that a non-normal mask m such that
112 1.1 cgd * index(m) <= index(n) could potentially apply to many children of n.
113 1.1 cgd * Thus, for each non-host route, we attach its mask to a list at an internal
114 1.28 perry * node as high in the tree as we can go.
115 1.9 mycroft *
116 1.9 mycroft * The present version of the code makes use of normal routes in short-
117 1.31 wiz * circuiting an explicit mask and compare operation when testing whether
118 1.9 mycroft * a key satisfies a normal route, and also in remembering the unique leaf
119 1.9 mycroft * that governs a subtree.
120 1.1 cgd */
121 1.1 cgd
122 1.1 cgd struct radix_node *
123 1.23 matt rn_search(
124 1.23 matt const void *v_arg,
125 1.23 matt struct radix_node *head)
126 1.1 cgd {
127 1.21 matt const u_char * const v = v_arg;
128 1.14 augustss struct radix_node *x;
129 1.1 cgd
130 1.21 matt for (x = head; x->rn_b >= 0;) {
131 1.1 cgd if (x->rn_bmask & v[x->rn_off])
132 1.1 cgd x = x->rn_r;
133 1.1 cgd else
134 1.1 cgd x = x->rn_l;
135 1.1 cgd }
136 1.37 dyoung return x;
137 1.13 fvdl }
138 1.1 cgd
139 1.1 cgd struct radix_node *
140 1.23 matt rn_search_m(
141 1.23 matt const void *v_arg,
142 1.23 matt struct radix_node *head,
143 1.23 matt const void *m_arg)
144 1.1 cgd {
145 1.14 augustss struct radix_node *x;
146 1.21 matt const u_char * const v = v_arg;
147 1.21 matt const u_char * const m = m_arg;
148 1.1 cgd
149 1.1 cgd for (x = head; x->rn_b >= 0;) {
150 1.1 cgd if ((x->rn_bmask & m[x->rn_off]) &&
151 1.1 cgd (x->rn_bmask & v[x->rn_off]))
152 1.1 cgd x = x->rn_r;
153 1.1 cgd else
154 1.1 cgd x = x->rn_l;
155 1.1 cgd }
156 1.1 cgd return x;
157 1.13 fvdl }
158 1.1 cgd
159 1.6 mycroft int
160 1.23 matt rn_refines(
161 1.23 matt const void *m_arg,
162 1.23 matt const void *n_arg)
163 1.6 mycroft {
164 1.21 matt const char *m = m_arg;
165 1.21 matt const char *n = n_arg;
166 1.29 christos const char *lim = n + *(const u_char *)n;
167 1.21 matt const char *lim2 = lim;
168 1.29 christos int longer = (*(const u_char *)n++) - (int)(*(const u_char *)m++);
169 1.6 mycroft int masks_are_equal = 1;
170 1.6 mycroft
171 1.6 mycroft if (longer > 0)
172 1.6 mycroft lim -= longer;
173 1.6 mycroft while (n < lim) {
174 1.6 mycroft if (*n & ~(*m))
175 1.6 mycroft return 0;
176 1.6 mycroft if (*n++ != *m++)
177 1.6 mycroft masks_are_equal = 0;
178 1.6 mycroft }
179 1.6 mycroft while (n < lim2)
180 1.6 mycroft if (*n++)
181 1.6 mycroft return 0;
182 1.6 mycroft if (masks_are_equal && (longer < 0))
183 1.6 mycroft for (lim2 = m - longer; m < lim2; )
184 1.6 mycroft if (*m++)
185 1.6 mycroft return 1;
186 1.37 dyoung return !masks_are_equal;
187 1.6 mycroft }
188 1.1 cgd
189 1.9 mycroft struct radix_node *
190 1.23 matt rn_lookup(
191 1.23 matt const void *v_arg,
192 1.23 matt const void *m_arg,
193 1.23 matt struct radix_node_head *head)
194 1.9 mycroft {
195 1.14 augustss struct radix_node *x;
196 1.21 matt const char *netmask = NULL;
197 1.9 mycroft
198 1.9 mycroft if (m_arg) {
199 1.9 mycroft if ((x = rn_addmask(m_arg, 1, head->rnh_treetop->rn_off)) == 0)
200 1.37 dyoung return NULL;
201 1.9 mycroft netmask = x->rn_key;
202 1.9 mycroft }
203 1.9 mycroft x = rn_match(v_arg, head);
204 1.37 dyoung if (x != NULL && netmask != NULL) {
205 1.37 dyoung while (x != NULL && x->rn_mask != netmask)
206 1.9 mycroft x = x->rn_dupedkey;
207 1.9 mycroft }
208 1.9 mycroft return x;
209 1.9 mycroft }
210 1.9 mycroft
211 1.10 christos static int
212 1.23 matt rn_satisfies_leaf(
213 1.23 matt const char *trial,
214 1.23 matt struct radix_node *leaf,
215 1.23 matt int skip)
216 1.23 matt {
217 1.23 matt const char *cp = trial;
218 1.23 matt const char *cp2 = leaf->rn_key;
219 1.23 matt const char *cp3 = leaf->rn_mask;
220 1.21 matt const char *cplim;
221 1.29 christos int length = min(*(const u_char *)cp, *(const u_char *)cp2);
222 1.9 mycroft
223 1.9 mycroft if (cp3 == 0)
224 1.9 mycroft cp3 = rn_ones;
225 1.9 mycroft else
226 1.29 christos length = min(length, *(const u_char *)cp3);
227 1.9 mycroft cplim = cp + length; cp3 += skip; cp2 += skip;
228 1.9 mycroft for (cp += skip; cp < cplim; cp++, cp2++, cp3++)
229 1.9 mycroft if ((*cp ^ *cp2) & *cp3)
230 1.9 mycroft return 0;
231 1.9 mycroft return 1;
232 1.9 mycroft }
233 1.1 cgd
234 1.1 cgd struct radix_node *
235 1.23 matt rn_match(
236 1.23 matt const void *v_arg,
237 1.23 matt struct radix_node_head *head)
238 1.1 cgd {
239 1.21 matt const char * const v = v_arg;
240 1.23 matt struct radix_node *t = head->rnh_treetop;
241 1.23 matt struct radix_node *top = t;
242 1.23 matt struct radix_node *x;
243 1.23 matt struct radix_node *saved_t;
244 1.21 matt const char *cp = v;
245 1.21 matt const char *cp2;
246 1.21 matt const char *cplim;
247 1.23 matt int off = t->rn_off;
248 1.29 christos int vlen = *(const u_char *)cp;
249 1.23 matt int matched_off;
250 1.14 augustss int test, b, rn_b;
251 1.1 cgd
252 1.1 cgd /*
253 1.6 mycroft * Open code rn_search(v, top) to avoid overhead of extra
254 1.1 cgd * subroutine call.
255 1.1 cgd */
256 1.1 cgd for (; t->rn_b >= 0; ) {
257 1.1 cgd if (t->rn_bmask & cp[t->rn_off])
258 1.1 cgd t = t->rn_r;
259 1.1 cgd else
260 1.1 cgd t = t->rn_l;
261 1.1 cgd }
262 1.1 cgd /*
263 1.1 cgd * See if we match exactly as a host destination
264 1.9 mycroft * or at least learn how many bits match, for normal mask finesse.
265 1.9 mycroft *
266 1.9 mycroft * It doesn't hurt us to limit how many bytes to check
267 1.9 mycroft * to the length of the mask, since if it matches we had a genuine
268 1.9 mycroft * match and the leaf we have is the most specific one anyway;
269 1.9 mycroft * if it didn't match with a shorter length it would fail
270 1.9 mycroft * with a long one. This wins big for class B&C netmasks which
271 1.9 mycroft * are probably the most common case...
272 1.1 cgd */
273 1.9 mycroft if (t->rn_mask)
274 1.29 christos vlen = *(const u_char *)t->rn_mask;
275 1.1 cgd cp += off; cp2 = t->rn_key + off; cplim = v + vlen;
276 1.1 cgd for (; cp < cplim; cp++, cp2++)
277 1.1 cgd if (*cp != *cp2)
278 1.1 cgd goto on1;
279 1.1 cgd /*
280 1.1 cgd * This extra grot is in case we are explicitly asked
281 1.1 cgd * to look up the default. Ugh!
282 1.1 cgd */
283 1.1 cgd if ((t->rn_flags & RNF_ROOT) && t->rn_dupedkey)
284 1.1 cgd t = t->rn_dupedkey;
285 1.1 cgd return t;
286 1.1 cgd on1:
287 1.9 mycroft test = (*cp ^ *cp2) & 0xff; /* find first bit that differs */
288 1.9 mycroft for (b = 7; (test >>= 1) > 0;)
289 1.9 mycroft b--;
290 1.1 cgd matched_off = cp - v;
291 1.9 mycroft b += matched_off << 3;
292 1.9 mycroft rn_b = -1 - b;
293 1.9 mycroft /*
294 1.9 mycroft * If there is a host route in a duped-key chain, it will be first.
295 1.9 mycroft */
296 1.9 mycroft if ((saved_t = t)->rn_mask == 0)
297 1.9 mycroft t = t->rn_dupedkey;
298 1.9 mycroft for (; t; t = t->rn_dupedkey)
299 1.1 cgd /*
300 1.9 mycroft * Even if we don't match exactly as a host,
301 1.1 cgd * we may match if the leaf we wound up at is
302 1.1 cgd * a route to a net.
303 1.1 cgd */
304 1.9 mycroft if (t->rn_flags & RNF_NORMAL) {
305 1.9 mycroft if (rn_b <= t->rn_b)
306 1.9 mycroft return t;
307 1.15 itojun } else if (rn_satisfies_leaf(v, t, matched_off))
308 1.9 mycroft return t;
309 1.1 cgd t = saved_t;
310 1.1 cgd /* start searching up the tree */
311 1.1 cgd do {
312 1.14 augustss struct radix_mask *m;
313 1.1 cgd t = t->rn_p;
314 1.12 christos m = t->rn_mklist;
315 1.12 christos if (m) {
316 1.1 cgd /*
317 1.9 mycroft * If non-contiguous masks ever become important
318 1.9 mycroft * we can restore the masking and open coding of
319 1.9 mycroft * the search and satisfaction test and put the
320 1.9 mycroft * calculation of "off" back before the "do".
321 1.1 cgd */
322 1.1 cgd do {
323 1.9 mycroft if (m->rm_flags & RNF_NORMAL) {
324 1.9 mycroft if (rn_b <= m->rm_b)
325 1.37 dyoung return m->rm_leaf;
326 1.9 mycroft } else {
327 1.9 mycroft off = min(t->rn_off, matched_off);
328 1.9 mycroft x = rn_search_m(v, t, m->rm_mask);
329 1.9 mycroft while (x && x->rn_mask != m->rm_mask)
330 1.9 mycroft x = x->rn_dupedkey;
331 1.15 itojun if (x && rn_satisfies_leaf(v, x, off))
332 1.17 itojun return x;
333 1.9 mycroft }
334 1.15 itojun m = m->rm_mklist;
335 1.12 christos } while (m);
336 1.1 cgd }
337 1.6 mycroft } while (t != top);
338 1.37 dyoung return NULL;
339 1.13 fvdl }
340 1.28 perry
341 1.32 dyoung static void
342 1.32 dyoung rn_nodeprint(struct radix_node *rn, rn_printer_t printer, void *arg,
343 1.32 dyoung const char *delim)
344 1.32 dyoung {
345 1.32 dyoung (*printer)(arg, "%s(%s%p: p<%p> l<%p> r<%p>)",
346 1.32 dyoung delim, ((void *)rn == arg) ? "*" : "", rn, rn->rn_p,
347 1.32 dyoung rn->rn_l, rn->rn_r);
348 1.32 dyoung }
349 1.32 dyoung
350 1.1 cgd #ifdef RN_DEBUG
351 1.6 mycroft int rn_debug = 1;
352 1.32 dyoung
353 1.32 dyoung static void
354 1.32 dyoung rn_dbg_print(void *arg, const char *fmt, ...)
355 1.32 dyoung {
356 1.32 dyoung va_list ap;
357 1.32 dyoung
358 1.32 dyoung va_start(ap, fmt);
359 1.32 dyoung vlog(LOG_DEBUG, fmt, ap);
360 1.32 dyoung va_end(ap);
361 1.32 dyoung }
362 1.32 dyoung
363 1.32 dyoung static void
364 1.32 dyoung rn_treeprint(struct radix_node_head *h, rn_printer_t printer, void *arg)
365 1.32 dyoung {
366 1.32 dyoung struct radix_node *dup, *rn;
367 1.32 dyoung const char *delim;
368 1.32 dyoung
369 1.32 dyoung if (printer == NULL)
370 1.32 dyoung return;
371 1.32 dyoung
372 1.32 dyoung rn = rn_walkfirst(h->rnh_treetop, printer, arg);
373 1.32 dyoung for (;;) {
374 1.32 dyoung /* Process leaves */
375 1.32 dyoung delim = "";
376 1.32 dyoung for (dup = rn; dup != NULL; dup = dup->rn_dupedkey) {
377 1.32 dyoung if ((dup->rn_flags & RNF_ROOT) != 0)
378 1.32 dyoung continue;
379 1.32 dyoung rn_nodeprint(dup, printer, arg, delim);
380 1.32 dyoung delim = ", ";
381 1.32 dyoung }
382 1.32 dyoung rn = rn_walknext(rn, printer, arg);
383 1.32 dyoung if (rn->rn_flags & RNF_ROOT)
384 1.32 dyoung return;
385 1.32 dyoung }
386 1.32 dyoung /* NOTREACHED */
387 1.32 dyoung }
388 1.32 dyoung
389 1.32 dyoung #define traverse(__head, __rn) rn_treeprint((__head), rn_dbg_print, (__rn))
390 1.32 dyoung #endif /* RN_DEBUG */
391 1.1 cgd
392 1.1 cgd struct radix_node *
393 1.23 matt rn_newpair(
394 1.23 matt const void *v,
395 1.23 matt int b,
396 1.23 matt struct radix_node nodes[2])
397 1.1 cgd {
398 1.23 matt struct radix_node *tt = nodes;
399 1.23 matt struct radix_node *t = tt + 1;
400 1.1 cgd t->rn_b = b; t->rn_bmask = 0x80 >> (b & 7);
401 1.1 cgd t->rn_l = tt; t->rn_off = b >> 3;
402 1.21 matt tt->rn_b = -1; tt->rn_key = v; tt->rn_p = t;
403 1.1 cgd tt->rn_flags = t->rn_flags = RNF_ACTIVE;
404 1.1 cgd return t;
405 1.1 cgd }
406 1.1 cgd
407 1.1 cgd struct radix_node *
408 1.23 matt rn_insert(
409 1.23 matt const void *v_arg,
410 1.23 matt struct radix_node_head *head,
411 1.23 matt int *dupentry,
412 1.23 matt struct radix_node nodes[2])
413 1.1 cgd {
414 1.6 mycroft struct radix_node *top = head->rnh_treetop;
415 1.14 augustss struct radix_node *t = rn_search(v_arg, top);
416 1.23 matt struct radix_node *tt;
417 1.23 matt const char *v = v_arg;
418 1.21 matt int head_off = top->rn_off;
419 1.29 christos int vlen = *((const u_char *)v);
420 1.21 matt const char *cp = v + head_off;
421 1.14 augustss int b;
422 1.1 cgd /*
423 1.9 mycroft * Find first bit at which v and t->rn_key differ
424 1.1 cgd */
425 1.1 cgd {
426 1.21 matt const char *cp2 = t->rn_key + head_off;
427 1.21 matt const char *cplim = v + vlen;
428 1.14 augustss int cmp_res;
429 1.1 cgd
430 1.1 cgd while (cp < cplim)
431 1.1 cgd if (*cp2++ != *cp++)
432 1.1 cgd goto on1;
433 1.1 cgd *dupentry = 1;
434 1.1 cgd return t;
435 1.1 cgd on1:
436 1.1 cgd *dupentry = 0;
437 1.1 cgd cmp_res = (cp[-1] ^ cp2[-1]) & 0xff;
438 1.1 cgd for (b = (cp - v) << 3; cmp_res; b--)
439 1.1 cgd cmp_res >>= 1;
440 1.1 cgd }
441 1.1 cgd {
442 1.14 augustss struct radix_node *p, *x = top;
443 1.1 cgd cp = v;
444 1.1 cgd do {
445 1.1 cgd p = x;
446 1.28 perry if (cp[x->rn_off] & x->rn_bmask)
447 1.1 cgd x = x->rn_r;
448 1.1 cgd else x = x->rn_l;
449 1.1 cgd } while (b > (unsigned) x->rn_b); /* x->rn_b < b && x->rn_b >= 0 */
450 1.1 cgd #ifdef RN_DEBUG
451 1.1 cgd if (rn_debug)
452 1.32 dyoung log(LOG_DEBUG, "%s: Going In:\n", __func__), traverse(head, p);
453 1.1 cgd #endif
454 1.6 mycroft t = rn_newpair(v_arg, b, nodes); tt = t->rn_l;
455 1.1 cgd if ((cp[p->rn_off] & p->rn_bmask) == 0)
456 1.1 cgd p->rn_l = t;
457 1.1 cgd else
458 1.1 cgd p->rn_r = t;
459 1.1 cgd x->rn_p = t; t->rn_p = p; /* frees x, p as temp vars below */
460 1.1 cgd if ((cp[t->rn_off] & t->rn_bmask) == 0) {
461 1.1 cgd t->rn_r = x;
462 1.1 cgd } else {
463 1.1 cgd t->rn_r = tt; t->rn_l = x;
464 1.1 cgd }
465 1.1 cgd #ifdef RN_DEBUG
466 1.32 dyoung if (rn_debug) {
467 1.32 dyoung log(LOG_DEBUG, "%s: Coming Out:\n", __func__),
468 1.32 dyoung traverse(head, p);
469 1.32 dyoung }
470 1.32 dyoung #endif /* RN_DEBUG */
471 1.1 cgd }
472 1.37 dyoung return tt;
473 1.1 cgd }
474 1.1 cgd
475 1.1 cgd struct radix_node *
476 1.23 matt rn_addmask(
477 1.23 matt const void *n_arg,
478 1.23 matt int search,
479 1.23 matt int skip)
480 1.1 cgd {
481 1.21 matt const char *netmask = n_arg;
482 1.21 matt const char *cp;
483 1.21 matt const char *cplim;
484 1.23 matt struct radix_node *x;
485 1.23 matt struct radix_node *saved_x;
486 1.14 augustss int b = 0, mlen, j;
487 1.9 mycroft int maskduplicated, m0, isnormal;
488 1.9 mycroft static int last_zeroed = 0;
489 1.9 mycroft
490 1.29 christos if ((mlen = *(const u_char *)netmask) > max_keylen)
491 1.9 mycroft mlen = max_keylen;
492 1.9 mycroft if (skip == 0)
493 1.9 mycroft skip = 1;
494 1.9 mycroft if (mlen <= skip)
495 1.37 dyoung return mask_rnhead->rnh_nodes;
496 1.9 mycroft if (skip > 1)
497 1.39 dyoung memmove(addmask_key + 1, rn_ones + 1, skip - 1);
498 1.9 mycroft if ((m0 = mlen) > skip)
499 1.39 dyoung memmove(addmask_key + skip, netmask + skip, mlen - skip);
500 1.9 mycroft /*
501 1.9 mycroft * Trim trailing zeroes.
502 1.9 mycroft */
503 1.9 mycroft for (cp = addmask_key + mlen; (cp > addmask_key) && cp[-1] == 0;)
504 1.9 mycroft cp--;
505 1.9 mycroft mlen = cp - addmask_key;
506 1.9 mycroft if (mlen <= skip) {
507 1.9 mycroft if (m0 >= last_zeroed)
508 1.9 mycroft last_zeroed = mlen;
509 1.37 dyoung return mask_rnhead->rnh_nodes;
510 1.9 mycroft }
511 1.9 mycroft if (m0 < last_zeroed)
512 1.39 dyoung memset(addmask_key + m0, 0, last_zeroed - m0);
513 1.9 mycroft *addmask_key = last_zeroed = mlen;
514 1.9 mycroft x = rn_search(addmask_key, rn_masktop);
515 1.39 dyoung if (memcmp(addmask_key, x->rn_key, mlen) != 0)
516 1.9 mycroft x = 0;
517 1.9 mycroft if (x || search)
518 1.37 dyoung return x;
519 1.6 mycroft R_Malloc(x, struct radix_node *, max_keylen + 2 * sizeof (*x));
520 1.37 dyoung if ((saved_x = x) == NULL)
521 1.37 dyoung return NULL;
522 1.39 dyoung memset(x, 0, max_keylen + 2 * sizeof (*x));
523 1.34 christos cp = netmask = (void *)(x + 2);
524 1.39 dyoung memmove(x + 2, addmask_key, mlen);
525 1.9 mycroft x = rn_insert(cp, mask_rnhead, &maskduplicated, x);
526 1.9 mycroft if (maskduplicated) {
527 1.16 enami log(LOG_ERR, "rn_addmask: mask impossibly already in tree\n");
528 1.9 mycroft Free(saved_x);
529 1.37 dyoung return x;
530 1.9 mycroft }
531 1.9 mycroft /*
532 1.9 mycroft * Calculate index of mask, and check for normalcy.
533 1.9 mycroft */
534 1.9 mycroft cplim = netmask + mlen; isnormal = 1;
535 1.29 christos for (cp = netmask + skip; (cp < cplim) && *(const u_char *)cp == 0xff;)
536 1.9 mycroft cp++;
537 1.1 cgd if (cp != cplim) {
538 1.28 perry for (j = 0x80; (j & *cp) != 0; j >>= 1)
539 1.9 mycroft b++;
540 1.9 mycroft if (*cp != normal_chars[b] || cp != (cplim - 1))
541 1.9 mycroft isnormal = 0;
542 1.1 cgd }
543 1.9 mycroft b += (cp - netmask) << 3;
544 1.1 cgd x->rn_b = -1 - b;
545 1.9 mycroft if (isnormal)
546 1.9 mycroft x->rn_flags |= RNF_NORMAL;
547 1.37 dyoung return x;
548 1.1 cgd }
549 1.1 cgd
550 1.9 mycroft static int /* XXX: arbitrary ordering for non-contiguous masks */
551 1.23 matt rn_lexobetter(
552 1.23 matt const void *m_arg,
553 1.23 matt const void *n_arg)
554 1.23 matt {
555 1.23 matt const u_char *mp = m_arg;
556 1.23 matt const u_char *np = n_arg;
557 1.23 matt const u_char *lim;
558 1.9 mycroft
559 1.9 mycroft if (*mp > *np)
560 1.9 mycroft return 1; /* not really, but need to check longer one first */
561 1.28 perry if (*mp == *np)
562 1.9 mycroft for (lim = mp + *mp; mp < lim;)
563 1.9 mycroft if (*mp++ > *np++)
564 1.9 mycroft return 1;
565 1.9 mycroft return 0;
566 1.9 mycroft }
567 1.9 mycroft
568 1.9 mycroft static struct radix_mask *
569 1.23 matt rn_new_radix_mask(
570 1.23 matt struct radix_node *tt,
571 1.23 matt struct radix_mask *next)
572 1.9 mycroft {
573 1.14 augustss struct radix_mask *m;
574 1.9 mycroft
575 1.9 mycroft MKGet(m);
576 1.37 dyoung if (m == NULL) {
577 1.9 mycroft log(LOG_ERR, "Mask for route not entered\n");
578 1.37 dyoung return NULL;
579 1.9 mycroft }
580 1.39 dyoung memset(m, 0, sizeof(*m));
581 1.9 mycroft m->rm_b = tt->rn_b;
582 1.9 mycroft m->rm_flags = tt->rn_flags;
583 1.9 mycroft if (tt->rn_flags & RNF_NORMAL)
584 1.9 mycroft m->rm_leaf = tt;
585 1.9 mycroft else
586 1.9 mycroft m->rm_mask = tt->rn_mask;
587 1.9 mycroft m->rm_mklist = next;
588 1.9 mycroft tt->rn_mklist = m;
589 1.9 mycroft return m;
590 1.9 mycroft }
591 1.9 mycroft
592 1.1 cgd struct radix_node *
593 1.23 matt rn_addroute(
594 1.23 matt const void *v_arg,
595 1.23 matt const void *n_arg,
596 1.23 matt struct radix_node_head *head,
597 1.23 matt struct radix_node treenodes[2])
598 1.1 cgd {
599 1.33 dyoung const char *v = v_arg, *netmask = n_arg;
600 1.33 dyoung struct radix_node *t, *x = NULL, *tt;
601 1.33 dyoung struct radix_node *saved_tt, *top = head->rnh_treetop;
602 1.10 christos short b = 0, b_leaf = 0;
603 1.9 mycroft int keyduplicated;
604 1.21 matt const char *mmask;
605 1.1 cgd struct radix_mask *m, **mp;
606 1.1 cgd
607 1.1 cgd /*
608 1.1 cgd * In dealing with non-contiguous masks, there may be
609 1.1 cgd * many different routes which have the same mask.
610 1.1 cgd * We will find it useful to have a unique pointer to
611 1.1 cgd * the mask to speed avoiding duplicate references at
612 1.1 cgd * nodes and possibly save time in calculating indices.
613 1.1 cgd */
614 1.38 dyoung if (netmask != NULL) {
615 1.37 dyoung if ((x = rn_addmask(netmask, 0, top->rn_off)) == NULL)
616 1.37 dyoung return NULL;
617 1.9 mycroft b_leaf = x->rn_b;
618 1.9 mycroft b = -1 - x->rn_b;
619 1.1 cgd netmask = x->rn_key;
620 1.1 cgd }
621 1.1 cgd /*
622 1.1 cgd * Deal with duplicated keys: attach node to previous instance
623 1.1 cgd */
624 1.1 cgd saved_tt = tt = rn_insert(v, head, &keyduplicated, treenodes);
625 1.1 cgd if (keyduplicated) {
626 1.38 dyoung for (t = tt; tt != NULL; t = tt, tt = tt->rn_dupedkey) {
627 1.1 cgd if (tt->rn_mask == netmask)
628 1.37 dyoung return NULL;
629 1.38 dyoung if (netmask == NULL ||
630 1.38 dyoung (tt->rn_mask != NULL &&
631 1.38 dyoung (b_leaf < tt->rn_b || /* index(netmask) > node */
632 1.9 mycroft rn_refines(netmask, tt->rn_mask) ||
633 1.9 mycroft rn_lexobetter(netmask, tt->rn_mask))))
634 1.6 mycroft break;
635 1.9 mycroft }
636 1.1 cgd /*
637 1.1 cgd * If the mask is not duplicated, we wouldn't
638 1.1 cgd * find it among possible duplicate key entries
639 1.1 cgd * anyway, so the above test doesn't hurt.
640 1.1 cgd *
641 1.6 mycroft * We sort the masks for a duplicated key the same way as
642 1.6 mycroft * in a masklist -- most specific to least specific.
643 1.6 mycroft * This may require the unfortunate nuisance of relocating
644 1.1 cgd * the head of the list.
645 1.12 christos *
646 1.12 christos * We also reverse, or doubly link the list through the
647 1.12 christos * parent pointer.
648 1.1 cgd */
649 1.9 mycroft if (tt == saved_tt) {
650 1.6 mycroft struct radix_node *xx = x;
651 1.6 mycroft /* link in at head of list */
652 1.6 mycroft (tt = treenodes)->rn_dupedkey = t;
653 1.6 mycroft tt->rn_flags = t->rn_flags;
654 1.6 mycroft tt->rn_p = x = t->rn_p;
655 1.12 christos t->rn_p = tt;
656 1.33 dyoung if (x->rn_l == t)
657 1.33 dyoung x->rn_l = tt;
658 1.33 dyoung else
659 1.33 dyoung x->rn_r = tt;
660 1.33 dyoung saved_tt = tt;
661 1.33 dyoung x = xx;
662 1.6 mycroft } else {
663 1.6 mycroft (tt = treenodes)->rn_dupedkey = t->rn_dupedkey;
664 1.6 mycroft t->rn_dupedkey = tt;
665 1.12 christos tt->rn_p = t;
666 1.12 christos if (tt->rn_dupedkey)
667 1.12 christos tt->rn_dupedkey->rn_p = tt;
668 1.6 mycroft }
669 1.36 dyoung tt->rn_key = v;
670 1.1 cgd tt->rn_b = -1;
671 1.9 mycroft tt->rn_flags = RNF_ACTIVE;
672 1.1 cgd }
673 1.1 cgd /*
674 1.1 cgd * Put mask in tree.
675 1.1 cgd */
676 1.38 dyoung if (netmask != NULL) {
677 1.1 cgd tt->rn_mask = netmask;
678 1.1 cgd tt->rn_b = x->rn_b;
679 1.9 mycroft tt->rn_flags |= x->rn_flags & RNF_NORMAL;
680 1.1 cgd }
681 1.1 cgd t = saved_tt->rn_p;
682 1.9 mycroft if (keyduplicated)
683 1.9 mycroft goto on2;
684 1.1 cgd b_leaf = -1 - t->rn_b;
685 1.33 dyoung if (t->rn_r == saved_tt)
686 1.33 dyoung x = t->rn_l;
687 1.33 dyoung else
688 1.33 dyoung x = t->rn_r;
689 1.1 cgd /* Promote general routes from below */
690 1.28 perry if (x->rn_b < 0) {
691 1.38 dyoung for (mp = &t->rn_mklist; x != NULL; x = x->rn_dupedkey) {
692 1.38 dyoung if (x->rn_mask != NULL && x->rn_b >= b_leaf &&
693 1.38 dyoung x->rn_mklist == NULL) {
694 1.38 dyoung *mp = m = rn_new_radix_mask(x, NULL);
695 1.38 dyoung if (m != NULL)
696 1.38 dyoung mp = &m->rm_mklist;
697 1.38 dyoung }
698 1.1 cgd }
699 1.38 dyoung } else if (x->rn_mklist != NULL) {
700 1.1 cgd /*
701 1.1 cgd * Skip over masks whose index is > that of new node
702 1.1 cgd */
703 1.38 dyoung for (mp = &x->rn_mklist; (m = *mp) != NULL; mp = &m->rm_mklist)
704 1.1 cgd if (m->rm_b >= b_leaf)
705 1.1 cgd break;
706 1.33 dyoung t->rn_mklist = m;
707 1.38 dyoung *mp = NULL;
708 1.1 cgd }
709 1.9 mycroft on2:
710 1.1 cgd /* Add new route to highest possible ancestor's list */
711 1.38 dyoung if (netmask == NULL || b > t->rn_b)
712 1.1 cgd return tt; /* can't lift at all */
713 1.1 cgd b_leaf = tt->rn_b;
714 1.1 cgd do {
715 1.1 cgd x = t;
716 1.1 cgd t = t->rn_p;
717 1.6 mycroft } while (b <= t->rn_b && x != top);
718 1.1 cgd /*
719 1.1 cgd * Search through routes associated with node to
720 1.1 cgd * insert new route according to index.
721 1.9 mycroft * Need same criteria as when sorting dupedkeys to avoid
722 1.9 mycroft * double loop on deletion.
723 1.1 cgd */
724 1.38 dyoung for (mp = &x->rn_mklist; (m = *mp) != NULL; mp = &m->rm_mklist) {
725 1.1 cgd if (m->rm_b < b_leaf)
726 1.1 cgd continue;
727 1.1 cgd if (m->rm_b > b_leaf)
728 1.1 cgd break;
729 1.9 mycroft if (m->rm_flags & RNF_NORMAL) {
730 1.9 mycroft mmask = m->rm_leaf->rn_mask;
731 1.9 mycroft if (tt->rn_flags & RNF_NORMAL) {
732 1.16 enami log(LOG_ERR, "Non-unique normal route,"
733 1.16 enami " mask not entered\n");
734 1.9 mycroft return tt;
735 1.9 mycroft }
736 1.9 mycroft } else
737 1.9 mycroft mmask = m->rm_mask;
738 1.9 mycroft if (mmask == netmask) {
739 1.1 cgd m->rm_refs++;
740 1.1 cgd tt->rn_mklist = m;
741 1.1 cgd return tt;
742 1.1 cgd }
743 1.9 mycroft if (rn_refines(netmask, mmask) || rn_lexobetter(netmask, mmask))
744 1.6 mycroft break;
745 1.1 cgd }
746 1.9 mycroft *mp = rn_new_radix_mask(tt, *mp);
747 1.1 cgd return tt;
748 1.1 cgd }
749 1.1 cgd
750 1.1 cgd struct radix_node *
751 1.33 dyoung rn_delete1(
752 1.23 matt const void *v_arg,
753 1.23 matt const void *netmask_arg,
754 1.33 dyoung struct radix_node_head *head,
755 1.33 dyoung struct radix_node *rn)
756 1.23 matt {
757 1.33 dyoung struct radix_node *t, *p, *x, *tt;
758 1.33 dyoung struct radix_mask *m, *saved_m, **mp;
759 1.33 dyoung struct radix_node *dupedkey, *saved_tt, *top;
760 1.33 dyoung const char *v, *netmask;
761 1.6 mycroft int b, head_off, vlen;
762 1.1 cgd
763 1.33 dyoung v = v_arg;
764 1.33 dyoung netmask = netmask_arg;
765 1.6 mycroft x = head->rnh_treetop;
766 1.6 mycroft tt = rn_search(v, x);
767 1.6 mycroft head_off = x->rn_off;
768 1.29 christos vlen = *(const u_char *)v;
769 1.6 mycroft saved_tt = tt;
770 1.6 mycroft top = x;
771 1.38 dyoung if (tt == NULL ||
772 1.39 dyoung memcmp(v + head_off, tt->rn_key + head_off, vlen - head_off) != 0)
773 1.37 dyoung return NULL;
774 1.1 cgd /*
775 1.1 cgd * Delete our route from mask lists.
776 1.1 cgd */
777 1.38 dyoung if (netmask != NULL) {
778 1.38 dyoung if ((x = rn_addmask(netmask, 1, head_off)) == NULL)
779 1.37 dyoung return NULL;
780 1.9 mycroft netmask = x->rn_key;
781 1.1 cgd while (tt->rn_mask != netmask)
782 1.38 dyoung if ((tt = tt->rn_dupedkey) == NULL)
783 1.37 dyoung return NULL;
784 1.1 cgd }
785 1.38 dyoung if (tt->rn_mask == NULL || (saved_m = m = tt->rn_mklist) == NULL)
786 1.1 cgd goto on1;
787 1.9 mycroft if (tt->rn_flags & RNF_NORMAL) {
788 1.9 mycroft if (m->rm_leaf != tt || m->rm_refs > 0) {
789 1.9 mycroft log(LOG_ERR, "rn_delete: inconsistent annotation\n");
790 1.37 dyoung return NULL; /* dangling ref could cause disaster */
791 1.9 mycroft }
792 1.28 perry } else {
793 1.9 mycroft if (m->rm_mask != tt->rn_mask) {
794 1.9 mycroft log(LOG_ERR, "rn_delete: inconsistent annotation\n");
795 1.9 mycroft goto on1;
796 1.9 mycroft }
797 1.9 mycroft if (--m->rm_refs >= 0)
798 1.9 mycroft goto on1;
799 1.1 cgd }
800 1.1 cgd b = -1 - tt->rn_b;
801 1.1 cgd t = saved_tt->rn_p;
802 1.1 cgd if (b > t->rn_b)
803 1.1 cgd goto on1; /* Wasn't lifted at all */
804 1.1 cgd do {
805 1.1 cgd x = t;
806 1.1 cgd t = t->rn_p;
807 1.6 mycroft } while (b <= t->rn_b && x != top);
808 1.38 dyoung for (mp = &x->rn_mklist; (m = *mp) != NULL; mp = &m->rm_mklist) {
809 1.1 cgd if (m == saved_m) {
810 1.1 cgd *mp = m->rm_mklist;
811 1.1 cgd MKFree(m);
812 1.1 cgd break;
813 1.1 cgd }
814 1.38 dyoung }
815 1.38 dyoung if (m == NULL) {
816 1.9 mycroft log(LOG_ERR, "rn_delete: couldn't find our annotation\n");
817 1.9 mycroft if (tt->rn_flags & RNF_NORMAL)
818 1.37 dyoung return NULL; /* Dangling ref to us */
819 1.9 mycroft }
820 1.1 cgd on1:
821 1.1 cgd /*
822 1.1 cgd * Eliminate us from tree
823 1.1 cgd */
824 1.1 cgd if (tt->rn_flags & RNF_ROOT)
825 1.37 dyoung return NULL;
826 1.1 cgd #ifdef RN_DEBUG
827 1.32 dyoung if (rn_debug)
828 1.32 dyoung log(LOG_DEBUG, "%s: Going In:\n", __func__), traverse(head, tt);
829 1.6 mycroft #endif
830 1.1 cgd t = tt->rn_p;
831 1.12 christos dupedkey = saved_tt->rn_dupedkey;
832 1.38 dyoung if (dupedkey != NULL) {
833 1.12 christos /*
834 1.12 christos * Here, tt is the deletion target, and
835 1.12 christos * saved_tt is the head of the dupedkey chain.
836 1.12 christos */
837 1.1 cgd if (tt == saved_tt) {
838 1.33 dyoung x = dupedkey;
839 1.33 dyoung x->rn_p = t;
840 1.33 dyoung if (t->rn_l == tt)
841 1.33 dyoung t->rn_l = x;
842 1.33 dyoung else
843 1.33 dyoung t->rn_r = x;
844 1.6 mycroft } else {
845 1.12 christos /* find node in front of tt on the chain */
846 1.38 dyoung for (x = p = saved_tt;
847 1.38 dyoung p != NULL && p->rn_dupedkey != tt;)
848 1.6 mycroft p = p->rn_dupedkey;
849 1.38 dyoung if (p != NULL) {
850 1.12 christos p->rn_dupedkey = tt->rn_dupedkey;
851 1.38 dyoung if (tt->rn_dupedkey != NULL)
852 1.12 christos tt->rn_dupedkey->rn_p = p;
853 1.38 dyoung } else
854 1.38 dyoung log(LOG_ERR, "rn_delete: couldn't find us\n");
855 1.6 mycroft }
856 1.6 mycroft t = tt + 1;
857 1.6 mycroft if (t->rn_flags & RNF_ACTIVE) {
858 1.33 dyoung *++x = *t;
859 1.33 dyoung p = t->rn_p;
860 1.33 dyoung if (p->rn_l == t)
861 1.33 dyoung p->rn_l = x;
862 1.33 dyoung else
863 1.33 dyoung p->rn_r = x;
864 1.33 dyoung x->rn_l->rn_p = x;
865 1.33 dyoung x->rn_r->rn_p = x;
866 1.1 cgd }
867 1.1 cgd goto out;
868 1.1 cgd }
869 1.33 dyoung if (t->rn_l == tt)
870 1.33 dyoung x = t->rn_r;
871 1.33 dyoung else
872 1.33 dyoung x = t->rn_l;
873 1.1 cgd p = t->rn_p;
874 1.33 dyoung if (p->rn_r == t)
875 1.33 dyoung p->rn_r = x;
876 1.33 dyoung else
877 1.33 dyoung p->rn_l = x;
878 1.1 cgd x->rn_p = p;
879 1.1 cgd /*
880 1.1 cgd * Demote routes attached to us.
881 1.1 cgd */
882 1.38 dyoung if (t->rn_mklist == NULL)
883 1.38 dyoung ;
884 1.38 dyoung else if (x->rn_b >= 0) {
885 1.38 dyoung for (mp = &x->rn_mklist; (m = *mp) != NULL; mp = &m->rm_mklist)
886 1.38 dyoung ;
887 1.38 dyoung *mp = t->rn_mklist;
888 1.38 dyoung } else {
889 1.38 dyoung /* If there are any key,mask pairs in a sibling
890 1.38 dyoung duped-key chain, some subset will appear sorted
891 1.38 dyoung in the same order attached to our mklist */
892 1.38 dyoung for (m = t->rn_mklist;
893 1.38 dyoung m != NULL && x != NULL;
894 1.38 dyoung x = x->rn_dupedkey) {
895 1.38 dyoung if (m == x->rn_mklist) {
896 1.38 dyoung struct radix_mask *mm = m->rm_mklist;
897 1.38 dyoung x->rn_mklist = NULL;
898 1.38 dyoung if (--(m->rm_refs) < 0)
899 1.38 dyoung MKFree(m);
900 1.38 dyoung m = mm;
901 1.38 dyoung }
902 1.38 dyoung }
903 1.38 dyoung if (m != NULL) {
904 1.38 dyoung log(LOG_ERR, "rn_delete: Orphaned Mask %p at %p\n",
905 1.38 dyoung m, x);
906 1.1 cgd }
907 1.1 cgd }
908 1.1 cgd /*
909 1.1 cgd * We may be holding an active internal node in the tree.
910 1.1 cgd */
911 1.1 cgd x = tt + 1;
912 1.1 cgd if (t != x) {
913 1.1 cgd *t = *x;
914 1.33 dyoung t->rn_l->rn_p = t;
915 1.33 dyoung t->rn_r->rn_p = t;
916 1.1 cgd p = x->rn_p;
917 1.33 dyoung if (p->rn_l == x)
918 1.33 dyoung p->rn_l = t;
919 1.33 dyoung else
920 1.33 dyoung p->rn_r = t;
921 1.1 cgd }
922 1.1 cgd out:
923 1.32 dyoung #ifdef RN_DEBUG
924 1.32 dyoung if (rn_debug) {
925 1.32 dyoung log(LOG_DEBUG, "%s: Coming Out:\n", __func__),
926 1.32 dyoung traverse(head, tt);
927 1.32 dyoung }
928 1.32 dyoung #endif /* RN_DEBUG */
929 1.1 cgd tt->rn_flags &= ~RNF_ACTIVE;
930 1.1 cgd tt[1].rn_flags &= ~RNF_ACTIVE;
931 1.37 dyoung return tt;
932 1.1 cgd }
933 1.1 cgd
934 1.33 dyoung struct radix_node *
935 1.33 dyoung rn_delete(
936 1.33 dyoung const void *v_arg,
937 1.33 dyoung const void *netmask_arg,
938 1.33 dyoung struct radix_node_head *head)
939 1.33 dyoung {
940 1.33 dyoung return rn_delete1(v_arg, netmask_arg, head, NULL);
941 1.33 dyoung }
942 1.33 dyoung
943 1.32 dyoung static struct radix_node *
944 1.32 dyoung rn_walknext(struct radix_node *rn, rn_printer_t printer, void *arg)
945 1.32 dyoung {
946 1.32 dyoung /* If at right child go back up, otherwise, go right */
947 1.32 dyoung while (rn->rn_p->rn_r == rn && (rn->rn_flags & RNF_ROOT) == 0) {
948 1.32 dyoung if (printer != NULL)
949 1.32 dyoung (*printer)(arg, SUBTREE_CLOSE);
950 1.32 dyoung rn = rn->rn_p;
951 1.32 dyoung }
952 1.32 dyoung if (printer)
953 1.32 dyoung rn_nodeprint(rn->rn_p, printer, arg, "");
954 1.32 dyoung /* Find the next *leaf* since next node might vanish, too */
955 1.32 dyoung for (rn = rn->rn_p->rn_r; rn->rn_b >= 0;) {
956 1.32 dyoung if (printer != NULL)
957 1.32 dyoung (*printer)(arg, SUBTREE_OPEN);
958 1.32 dyoung rn = rn->rn_l;
959 1.32 dyoung }
960 1.32 dyoung return rn;
961 1.32 dyoung }
962 1.32 dyoung
963 1.32 dyoung static struct radix_node *
964 1.32 dyoung rn_walkfirst(struct radix_node *rn, rn_printer_t printer, void *arg)
965 1.32 dyoung {
966 1.32 dyoung /* First time through node, go left */
967 1.32 dyoung while (rn->rn_b >= 0) {
968 1.32 dyoung if (printer != NULL)
969 1.32 dyoung (*printer)(arg, SUBTREE_OPEN);
970 1.32 dyoung rn = rn->rn_l;
971 1.32 dyoung }
972 1.32 dyoung return rn;
973 1.32 dyoung }
974 1.32 dyoung
975 1.6 mycroft int
976 1.23 matt rn_walktree(
977 1.23 matt struct radix_node_head *h,
978 1.23 matt int (*f)(struct radix_node *, void *),
979 1.23 matt void *w)
980 1.6 mycroft {
981 1.6 mycroft int error;
982 1.32 dyoung struct radix_node *base, *next, *rn;
983 1.6 mycroft /*
984 1.6 mycroft * This gets complicated because we may delete the node
985 1.6 mycroft * while applying the function f to it, so we need to calculate
986 1.6 mycroft * the successor node in advance.
987 1.6 mycroft */
988 1.32 dyoung rn = rn_walkfirst(h->rnh_treetop, NULL, NULL);
989 1.6 mycroft for (;;) {
990 1.6 mycroft base = rn;
991 1.32 dyoung next = rn_walknext(rn, NULL, NULL);
992 1.6 mycroft /* Process leaves */
993 1.10 christos while ((rn = base) != NULL) {
994 1.6 mycroft base = rn->rn_dupedkey;
995 1.6 mycroft if (!(rn->rn_flags & RNF_ROOT) && (error = (*f)(rn, w)))
996 1.37 dyoung return error;
997 1.6 mycroft }
998 1.6 mycroft rn = next;
999 1.6 mycroft if (rn->rn_flags & RNF_ROOT)
1000 1.37 dyoung return 0;
1001 1.6 mycroft }
1002 1.6 mycroft /* NOTREACHED */
1003 1.6 mycroft }
1004 1.6 mycroft
1005 1.6 mycroft int
1006 1.41 dsl rn_inithead(void **head, int off)
1007 1.1 cgd {
1008 1.14 augustss struct radix_node_head *rnh;
1009 1.15 itojun
1010 1.37 dyoung if (*head != NULL)
1011 1.37 dyoung return 1;
1012 1.1 cgd R_Malloc(rnh, struct radix_node_head *, sizeof (*rnh));
1013 1.37 dyoung if (rnh == NULL)
1014 1.37 dyoung return 0;
1015 1.15 itojun *head = rnh;
1016 1.15 itojun return rn_inithead0(rnh, off);
1017 1.15 itojun }
1018 1.15 itojun
1019 1.15 itojun int
1020 1.41 dsl rn_inithead0(struct radix_node_head *rnh, int off)
1021 1.15 itojun {
1022 1.23 matt struct radix_node *t;
1023 1.23 matt struct radix_node *tt;
1024 1.23 matt struct radix_node *ttt;
1025 1.15 itojun
1026 1.39 dyoung memset(rnh, 0, sizeof(*rnh));
1027 1.1 cgd t = rn_newpair(rn_zeros, off, rnh->rnh_nodes);
1028 1.1 cgd ttt = rnh->rnh_nodes + 2;
1029 1.1 cgd t->rn_r = ttt;
1030 1.1 cgd t->rn_p = t;
1031 1.1 cgd tt = t->rn_l;
1032 1.1 cgd tt->rn_flags = t->rn_flags = RNF_ROOT | RNF_ACTIVE;
1033 1.1 cgd tt->rn_b = -1 - off;
1034 1.1 cgd *ttt = *tt;
1035 1.1 cgd ttt->rn_key = rn_ones;
1036 1.6 mycroft rnh->rnh_addaddr = rn_addroute;
1037 1.6 mycroft rnh->rnh_deladdr = rn_delete;
1038 1.6 mycroft rnh->rnh_matchaddr = rn_match;
1039 1.9 mycroft rnh->rnh_lookup = rn_lookup;
1040 1.1 cgd rnh->rnh_treetop = t;
1041 1.37 dyoung return 1;
1042 1.6 mycroft }
1043 1.6 mycroft
1044 1.6 mycroft void
1045 1.6 mycroft rn_init()
1046 1.6 mycroft {
1047 1.6 mycroft char *cp, *cplim;
1048 1.8 jtc #ifdef _KERNEL
1049 1.27 enami static int initialized;
1050 1.27 enami __link_set_decl(domains, struct domain);
1051 1.27 enami struct domain *const *dpp;
1052 1.27 enami
1053 1.27 enami if (initialized)
1054 1.27 enami return;
1055 1.27 enami initialized = 1;
1056 1.6 mycroft
1057 1.27 enami __link_set_foreach(dpp, domains) {
1058 1.27 enami if ((*dpp)->dom_maxrtkey > max_keylen)
1059 1.27 enami max_keylen = (*dpp)->dom_maxrtkey;
1060 1.27 enami }
1061 1.25 christos #endif
1062 1.6 mycroft if (max_keylen == 0) {
1063 1.9 mycroft log(LOG_ERR,
1064 1.9 mycroft "rn_init: radix functions require max_keylen be set\n");
1065 1.6 mycroft return;
1066 1.6 mycroft }
1067 1.6 mycroft R_Malloc(rn_zeros, char *, 3 * max_keylen);
1068 1.6 mycroft if (rn_zeros == NULL)
1069 1.6 mycroft panic("rn_init");
1070 1.39 dyoung memset(rn_zeros, 0, 3 * max_keylen);
1071 1.6 mycroft rn_ones = cp = rn_zeros + max_keylen;
1072 1.9 mycroft addmask_key = cplim = rn_ones + max_keylen;
1073 1.6 mycroft while (cp < cplim)
1074 1.6 mycroft *cp++ = -1;
1075 1.19 thorpej if (rn_inithead((void *)&mask_rnhead, 0) == 0)
1076 1.6 mycroft panic("rn_init 2");
1077 1.1 cgd }
1078