xref: /dragonfly/sys/net/radix.h (revision 0db87cb7)
1 /*
2  * Copyright (c) 1988, 1989, 1993
3  *	The Regents of the University of California.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *	This product includes software developed by the University of
16  *	California, Berkeley and its contributors.
17  * 4. Neither the name of the University nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  *
33  *	@(#)radix.h	8.2 (Berkeley) 10/31/94
34  * $FreeBSD: src/sys/net/radix.h,v 1.16.2.1 2000/05/03 19:17:11 wollman Exp $
35  * $DragonFly: src/sys/net/radix.h,v 1.12 2006/09/05 03:48:12 dillon Exp $
36  */
37 
38 #ifndef _NET_RADIX_H_
39 #define	_NET_RADIX_H_
40 
41 #ifndef _SYS_TYPES_H_
42 #include <sys/types.h>
43 #endif
44 
45 #ifdef MALLOC_DECLARE
46 MALLOC_DECLARE(M_RTABLE);
47 #endif
48 
49 /*
50  * Radix search tree node layout.
51  */
52 
53 struct radix_node {
54 	struct	radix_mask *rn_mklist;	/* masks contained in subtree */
55 	struct	radix_node *rn_parent;	/* parent */
56 	short	rn_bit;			/* bit offset; -1-index(netmask) */
57 	char	rn_bmask;		/* node: mask for bit test*/
58 	u_char	rn_flags;		/* enumerated next */
59 #define RNF_NORMAL	1		/* leaf contains normal route */
60 #define RNF_ROOT	2		/* leaf is root leaf for tree */
61 #define RNF_ACTIVE	4		/* This node is alive (for rtfree) */
62 	union {
63 		struct {			/* leaf only data: */
64 			char   *rn_Key;		/* object of search */
65 			char   *rn_Mask;	/* netmask, if present */
66 			struct	radix_node *rn_Dupedkey;
67 		} rn_leaf;
68 		struct {			/* node only data: */
69 			int	rn_Off;		/* where to start compare */
70 			struct	radix_node *rn_L;/* progeny */
71 			struct	radix_node *rn_R;/* progeny */
72 		} rn_node;
73 	}		rn_u;
74 #ifdef RN_DEBUG
75 	int rn_info;
76 	struct radix_node *rn_twin;
77 	struct radix_node *rn_ybro;
78 #endif
79 };
80 
81 #define	rn_dupedkey	rn_u.rn_leaf.rn_Dupedkey
82 #define	rn_key		rn_u.rn_leaf.rn_Key
83 #define	rn_mask		rn_u.rn_leaf.rn_Mask
84 #define	rn_offset	rn_u.rn_node.rn_Off
85 #define	rn_left		rn_u.rn_node.rn_L
86 #define	rn_right	rn_u.rn_node.rn_R
87 
88 /*
89  * We do this statically now because the dynamic initialization
90  * occurs too late and has an ordering problem w/ pf preloads
91  * vs protocol domains.
92  */
93 #define RN_MAXKEYLEN	32
94 #define RN_MAXKEYONES	{ -1, -1, -1, -1, -1, -1, -1, -1,	\
95 			  -1, -1, -1, -1, -1, -1, -1, -1,	\
96 			  -1, -1, -1, -1, -1, -1, -1, -1,	\
97 			  -1, -1, -1, -1, -1, -1, -1, -1 }
98 
99 /*
100  * Annotations to tree concerning potential routes applying to subtrees.
101  */
102 
103 struct radix_mask {
104 	short	rm_bit;			/* bit offset; -1-index(netmask) */
105 	char	rm_unused;		/* cf. rn_bmask */
106 	u_char	rm_flags;		/* cf. rn_flags */
107 	struct	radix_mask *rm_next;	/* list of more masks to try */
108 	union	{
109 		caddr_t	rmu_mask;		/* the mask */
110 		struct	radix_node *rmu_leaf;	/* for normal routes */
111 	}	rm_rmu;
112 	int	rm_refs;		/* # of references to this struct */
113 };
114 
115 #define	rm_mask rm_rmu.rmu_mask
116 #define	rm_leaf rm_rmu.rmu_leaf		/* extra field would make 32 bytes */
117 
118 typedef int walktree_f_t (struct radix_node *, void *);
119 
120 struct radix_node_head {
121 	struct	radix_node *rnh_treetop;
122 
123 	/* add based on sockaddr */
124 	struct	radix_node *(*rnh_addaddr)
125 		    (char *key, char *mask,
126 		     struct radix_node_head *head, struct radix_node nodes[]);
127 
128 	/* remove based on sockaddr */
129 	struct	radix_node *(*rnh_deladdr)
130 		    (char *key, char *mask, struct radix_node_head *head);
131 
132 	/* locate based on sockaddr */
133 	struct	radix_node *(*rnh_matchaddr)
134 		    (char *key, struct radix_node_head *head);
135 
136 	/* locate based on sockaddr */
137 	struct	radix_node *(*rnh_lookup)
138 		    (char *key, char *mask, struct radix_node_head *head);
139 
140 	/* traverse tree */
141 	int	(*rnh_walktree)
142 		    (struct radix_node_head *head, walktree_f_t *f, void *w);
143 
144 	/* traverse tree below a */
145 	int	(*rnh_walktree_from)
146 		    (struct radix_node_head *head, char *a, char *m,
147 		     walktree_f_t *f, void *w);
148 
149 	/*
150 	 * Do something when the last ref drops.
151 	 * A (*rnh_close)() routine
152 	 *	can clear RTF_UP
153 	 *	can remove a route from the radix tree
154 	 *	cannot change the reference count
155 	 *	cannot deallocate the route
156 	 */
157 	void	(*rnh_close)
158 		    (struct radix_node *rn, struct radix_node_head *head);
159 
160 	struct	radix_node rnh_nodes[3];	/* empty tree for common case */
161 
162 	int	rnh_last_zeroed;		/* mask last zeroed length */
163 
164 	/* unused entries */
165 	int	rnh_pktsize;		/* permit, but not require fixed keys */
166 	struct	radix_node *(*rnh_addpkt)	/* add based on packet hdr */
167 		    (void *v, char *mask,
168 		     struct radix_node_head *head, struct radix_node nodes[]);
169 	struct	radix_node *(*rnh_delpkt)	/* remove based on packet hdr */
170 		    (void *v, char *mask, struct radix_node_head *head);
171 
172 	/* traverse tree starting from a */
173 	int	(*rnh_walktree_at)
174 		    (struct radix_node_head *head, const char *a, const char *m,
175 		     walktree_f_t *f, void *w);
176 
177 	struct radix_node_head *rnh_maskhead;
178 };
179 
180 #ifndef _KERNEL
181 #define R_Malloc(p, t, n) (p = (t) malloc((n)))
182 #define Free(p) free(p);
183 #else
184 #define R_Malloc(p, t, n) (p = (t) kmalloc((n), M_RTABLE, M_INTWAIT | M_NULLOK))
185 #define Free(p) kfree(p, M_RTABLE);
186 #endif
187 
188 void			 rn_init (void);
189 int			 rn_inithead (void **, struct radix_node_head *, int);
190 struct radix_node_head	*rn_cpumaskhead(int cpu);
191 __boolean_t		 rn_refines (char *, char *);
192 struct radix_node	*rn_addmask (char *, __boolean_t, int,
193 				     struct radix_node_head *),
194 			*rn_addroute (char *, char *, struct radix_node_head *,
195 				      struct radix_node [2]),
196 			*rn_delete (char *, char *, struct radix_node_head *),
197 			*rn_lookup (char *key, char *mask,
198 				    struct radix_node_head *head),
199 			*rn_match (char *, struct radix_node_head *);
200 
201 #endif /* _NET_RADIX_H_ */
202