blob: c19d68551eb1472c5ad8b5e5dbb021083c179b70 [file] [log] [blame]
Robert Olsson19baf832005-06-21 12:43:18 -07001/*
2 * This program is free software; you can redistribute it and/or
3 * modify it under the terms of the GNU General Public License
4 * as published by the Free Software Foundation; either version
5 * 2 of the License, or (at your option) any later version.
6 *
7 * Robert Olsson <robert.olsson@its.uu.se> Uppsala Universitet
8 * & Swedish University of Agricultural Sciences.
9 *
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +090010 * Jens Laas <jens.laas@data.slu.se> Swedish University of
Robert Olsson19baf832005-06-21 12:43:18 -070011 * Agricultural Sciences.
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +090012 *
Robert Olsson19baf832005-06-21 12:43:18 -070013 * Hans Liss <hans.liss@its.uu.se> Uppsala Universitet
14 *
15 * This work is based on the LPC-trie which is originally descibed in:
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +090016 *
Robert Olsson19baf832005-06-21 12:43:18 -070017 * An experimental study of compression methods for dynamic tries
18 * Stefan Nilsson and Matti Tikkanen. Algorithmica, 33(1):19-33, 2002.
19 * http://www.nada.kth.se/~snilsson/public/papers/dyntrie2/
20 *
21 *
22 * IP-address lookup using LC-tries. Stefan Nilsson and Gunnar Karlsson
23 * IEEE Journal on Selected Areas in Communications, 17(6):1083-1092, June 1999
24 *
25 * Version: $Id: fib_trie.c,v 1.3 2005/06/08 14:20:01 robert Exp $
26 *
27 *
28 * Code from fib_hash has been reused which includes the following header:
29 *
30 *
31 * INET An implementation of the TCP/IP protocol suite for the LINUX
32 * operating system. INET is implemented using the BSD Socket
33 * interface as the means of communication with the user level.
34 *
35 * IPv4 FIB: lookup engine and maintenance routines.
36 *
37 *
38 * Authors: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
39 *
40 * This program is free software; you can redistribute it and/or
41 * modify it under the terms of the GNU General Public License
42 * as published by the Free Software Foundation; either version
43 * 2 of the License, or (at your option) any later version.
Robert Olssonfd966252005-12-22 11:25:10 -080044 *
45 * Substantial contributions to this work comes from:
46 *
47 * David S. Miller, <davem@davemloft.net>
48 * Stephen Hemminger <shemminger@osdl.org>
49 * Paul E. McKenney <paulmck@us.ibm.com>
50 * Patrick McHardy <kaber@trash.net>
Robert Olsson19baf832005-06-21 12:43:18 -070051 */
52
Robert Olsson05eee482007-03-19 16:27:37 -070053#define VERSION "0.408"
Robert Olsson19baf832005-06-21 12:43:18 -070054
Robert Olsson19baf832005-06-21 12:43:18 -070055#include <asm/uaccess.h>
56#include <asm/system.h>
Jiri Slaby1977f032007-10-18 23:40:25 -070057#include <linux/bitops.h>
Robert Olsson19baf832005-06-21 12:43:18 -070058#include <linux/types.h>
59#include <linux/kernel.h>
Robert Olsson19baf832005-06-21 12:43:18 -070060#include <linux/mm.h>
61#include <linux/string.h>
62#include <linux/socket.h>
63#include <linux/sockios.h>
64#include <linux/errno.h>
65#include <linux/in.h>
66#include <linux/inet.h>
Stephen Hemmingercd8787a2006-01-03 14:38:34 -080067#include <linux/inetdevice.h>
Robert Olsson19baf832005-06-21 12:43:18 -070068#include <linux/netdevice.h>
69#include <linux/if_arp.h>
70#include <linux/proc_fs.h>
Robert Olsson2373ce12005-08-25 13:01:29 -070071#include <linux/rcupdate.h>
Robert Olsson19baf832005-06-21 12:43:18 -070072#include <linux/skbuff.h>
73#include <linux/netlink.h>
74#include <linux/init.h>
75#include <linux/list.h>
Eric W. Biederman457c4cb2007-09-12 12:01:34 +020076#include <net/net_namespace.h>
Robert Olsson19baf832005-06-21 12:43:18 -070077#include <net/ip.h>
78#include <net/protocol.h>
79#include <net/route.h>
80#include <net/tcp.h>
81#include <net/sock.h>
82#include <net/ip_fib.h>
83#include "fib_lookup.h"
84
Robert Olsson06ef9212006-03-20 21:35:01 -080085#define MAX_STAT_DEPTH 32
Robert Olsson19baf832005-06-21 12:43:18 -070086
Robert Olsson19baf832005-06-21 12:43:18 -070087#define KEYLENGTH (8*sizeof(t_key))
Robert Olsson19baf832005-06-21 12:43:18 -070088
Robert Olsson19baf832005-06-21 12:43:18 -070089typedef unsigned int t_key;
90
91#define T_TNODE 0
92#define T_LEAF 1
93#define NODE_TYPE_MASK 0x1UL
Robert Olsson2373ce12005-08-25 13:01:29 -070094#define NODE_TYPE(node) ((node)->parent & NODE_TYPE_MASK)
95
Olof Johansson91b9a272005-08-09 20:24:39 -070096#define IS_TNODE(n) (!(n->parent & T_LEAF))
97#define IS_LEAF(n) (n->parent & T_LEAF)
Robert Olsson19baf832005-06-21 12:43:18 -070098
99struct node {
Olof Johansson91b9a272005-08-09 20:24:39 -0700100 unsigned long parent;
Eric Dumazet8d965442008-01-13 22:31:44 -0800101 t_key key;
Robert Olsson19baf832005-06-21 12:43:18 -0700102};
103
104struct leaf {
Olof Johansson91b9a272005-08-09 20:24:39 -0700105 unsigned long parent;
Eric Dumazet8d965442008-01-13 22:31:44 -0800106 t_key key;
Robert Olsson19baf832005-06-21 12:43:18 -0700107 struct hlist_head list;
Robert Olsson2373ce12005-08-25 13:01:29 -0700108 struct rcu_head rcu;
Robert Olsson19baf832005-06-21 12:43:18 -0700109};
110
111struct leaf_info {
112 struct hlist_node hlist;
Robert Olsson2373ce12005-08-25 13:01:29 -0700113 struct rcu_head rcu;
Robert Olsson19baf832005-06-21 12:43:18 -0700114 int plen;
115 struct list_head falh;
116};
117
118struct tnode {
Olof Johansson91b9a272005-08-09 20:24:39 -0700119 unsigned long parent;
Eric Dumazet8d965442008-01-13 22:31:44 -0800120 t_key key;
Eric Dumazet112d8cf2008-01-12 21:27:41 -0800121 unsigned char pos; /* 2log(KEYLENGTH) bits needed */
122 unsigned char bits; /* 2log(KEYLENGTH) bits needed */
Eric Dumazet8d965442008-01-13 22:31:44 -0800123 unsigned int full_children; /* KEYLENGTH bits needed */
124 unsigned int empty_children; /* KEYLENGTH bits needed */
Robert Olsson2373ce12005-08-25 13:01:29 -0700125 struct rcu_head rcu;
Olof Johansson91b9a272005-08-09 20:24:39 -0700126 struct node *child[0];
Robert Olsson19baf832005-06-21 12:43:18 -0700127};
128
129#ifdef CONFIG_IP_FIB_TRIE_STATS
130struct trie_use_stats {
131 unsigned int gets;
132 unsigned int backtrack;
133 unsigned int semantic_match_passed;
134 unsigned int semantic_match_miss;
135 unsigned int null_node_hit;
Robert Olsson2f368952005-07-05 15:02:40 -0700136 unsigned int resize_node_skipped;
Robert Olsson19baf832005-06-21 12:43:18 -0700137};
138#endif
139
140struct trie_stat {
141 unsigned int totdepth;
142 unsigned int maxdepth;
143 unsigned int tnodes;
144 unsigned int leaves;
145 unsigned int nullpointers;
Stephen Hemminger93672292008-01-22 21:54:05 -0800146 unsigned int prefixes;
Robert Olsson06ef9212006-03-20 21:35:01 -0800147 unsigned int nodesizes[MAX_STAT_DEPTH];
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700148};
Robert Olsson19baf832005-06-21 12:43:18 -0700149
150struct trie {
Olof Johansson91b9a272005-08-09 20:24:39 -0700151 struct node *trie;
Robert Olsson19baf832005-06-21 12:43:18 -0700152#ifdef CONFIG_IP_FIB_TRIE_STATS
153 struct trie_use_stats stats;
154#endif
Robert Olsson19baf832005-06-21 12:43:18 -0700155};
156
Robert Olsson19baf832005-06-21 12:43:18 -0700157static void put_child(struct trie *t, struct tnode *tn, int i, struct node *n);
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800158static void tnode_put_child_reorg(struct tnode *tn, int i, struct node *n,
159 int wasfull);
Robert Olsson19baf832005-06-21 12:43:18 -0700160static struct node *resize(struct trie *t, struct tnode *tn);
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700161static struct tnode *inflate(struct trie *t, struct tnode *tn);
162static struct tnode *halve(struct trie *t, struct tnode *tn);
Robert Olsson19baf832005-06-21 12:43:18 -0700163static void tnode_free(struct tnode *tn);
Robert Olsson19baf832005-06-21 12:43:18 -0700164
Christoph Lametere18b8902006-12-06 20:33:20 -0800165static struct kmem_cache *fn_alias_kmem __read_mostly;
Stephen Hemmingerbc3c8c12008-01-22 21:51:50 -0800166static struct kmem_cache *trie_leaf_kmem __read_mostly;
Robert Olsson19baf832005-06-21 12:43:18 -0700167
Stephen Hemminger06801912007-08-10 15:22:13 -0700168static inline struct tnode *node_parent(struct node *node)
169{
Eric Dumazetb59cfbf2008-01-18 03:31:36 -0800170 return (struct tnode *)(node->parent & ~NODE_TYPE_MASK);
171}
Stephen Hemminger06801912007-08-10 15:22:13 -0700172
Eric Dumazetb59cfbf2008-01-18 03:31:36 -0800173static inline struct tnode *node_parent_rcu(struct node *node)
174{
175 struct tnode *ret = node_parent(node);
176
Stephen Hemminger06801912007-08-10 15:22:13 -0700177 return rcu_dereference(ret);
178}
179
180static inline void node_set_parent(struct node *node, struct tnode *ptr)
181{
182 rcu_assign_pointer(node->parent,
183 (unsigned long)ptr | NODE_TYPE(node));
184}
Robert Olsson2373ce12005-08-25 13:01:29 -0700185
Eric Dumazetb59cfbf2008-01-18 03:31:36 -0800186static inline struct node *tnode_get_child(struct tnode *tn, unsigned int i)
Robert Olsson19baf832005-06-21 12:43:18 -0700187{
Eric Dumazetb59cfbf2008-01-18 03:31:36 -0800188 BUG_ON(i >= 1U << tn->bits);
Robert Olsson19baf832005-06-21 12:43:18 -0700189
Eric Dumazetb59cfbf2008-01-18 03:31:36 -0800190 return tn->child[i];
191}
192
193static inline struct node *tnode_get_child_rcu(struct tnode *tn, unsigned int i)
194{
195 struct node *ret = tnode_get_child(tn, i);
196
197 return rcu_dereference(ret);
Robert Olsson19baf832005-06-21 12:43:18 -0700198}
199
Stephen Hemmignerbb435b82005-08-09 20:25:39 -0700200static inline int tnode_child_length(const struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700201{
Olof Johansson91b9a272005-08-09 20:24:39 -0700202 return 1 << tn->bits;
Robert Olsson19baf832005-06-21 12:43:18 -0700203}
204
Stephen Hemmingerab66b4a2007-08-10 15:22:58 -0700205static inline t_key mask_pfx(t_key k, unsigned short l)
206{
207 return (l == 0) ? 0 : k >> (KEYLENGTH-l) << (KEYLENGTH-l);
208}
209
Robert Olsson19baf832005-06-21 12:43:18 -0700210static inline t_key tkey_extract_bits(t_key a, int offset, int bits)
211{
Olof Johansson91b9a272005-08-09 20:24:39 -0700212 if (offset < KEYLENGTH)
Robert Olsson19baf832005-06-21 12:43:18 -0700213 return ((t_key)(a << offset)) >> (KEYLENGTH - bits);
Olof Johansson91b9a272005-08-09 20:24:39 -0700214 else
Robert Olsson19baf832005-06-21 12:43:18 -0700215 return 0;
216}
217
218static inline int tkey_equals(t_key a, t_key b)
219{
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700220 return a == b;
Robert Olsson19baf832005-06-21 12:43:18 -0700221}
222
223static inline int tkey_sub_equals(t_key a, int offset, int bits, t_key b)
224{
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700225 if (bits == 0 || offset >= KEYLENGTH)
226 return 1;
Olof Johansson91b9a272005-08-09 20:24:39 -0700227 bits = bits > KEYLENGTH ? KEYLENGTH : bits;
228 return ((a ^ b) << offset) >> (KEYLENGTH - bits) == 0;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700229}
Robert Olsson19baf832005-06-21 12:43:18 -0700230
231static inline int tkey_mismatch(t_key a, int offset, t_key b)
232{
233 t_key diff = a ^ b;
234 int i = offset;
235
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700236 if (!diff)
237 return 0;
238 while ((diff << i) >> (KEYLENGTH-1) == 0)
Robert Olsson19baf832005-06-21 12:43:18 -0700239 i++;
240 return i;
241}
242
Robert Olsson19baf832005-06-21 12:43:18 -0700243/*
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900244 To understand this stuff, an understanding of keys and all their bits is
245 necessary. Every node in the trie has a key associated with it, but not
Robert Olsson19baf832005-06-21 12:43:18 -0700246 all of the bits in that key are significant.
247
248 Consider a node 'n' and its parent 'tp'.
249
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900250 If n is a leaf, every bit in its key is significant. Its presence is
251 necessitated by path compression, since during a tree traversal (when
252 searching for a leaf - unless we are doing an insertion) we will completely
253 ignore all skipped bits we encounter. Thus we need to verify, at the end of
254 a potentially successful search, that we have indeed been walking the
Robert Olsson19baf832005-06-21 12:43:18 -0700255 correct key path.
256
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900257 Note that we can never "miss" the correct key in the tree if present by
258 following the wrong path. Path compression ensures that segments of the key
259 that are the same for all keys with a given prefix are skipped, but the
260 skipped part *is* identical for each node in the subtrie below the skipped
261 bit! trie_insert() in this implementation takes care of that - note the
Robert Olsson19baf832005-06-21 12:43:18 -0700262 call to tkey_sub_equals() in trie_insert().
263
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900264 if n is an internal node - a 'tnode' here, the various parts of its key
Robert Olsson19baf832005-06-21 12:43:18 -0700265 have many different meanings.
266
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900267 Example:
Robert Olsson19baf832005-06-21 12:43:18 -0700268 _________________________________________________________________
269 | i | i | i | i | i | i | i | N | N | N | S | S | S | S | S | C |
270 -----------------------------------------------------------------
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900271 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
Robert Olsson19baf832005-06-21 12:43:18 -0700272
273 _________________________________________________________________
274 | C | C | C | u | u | u | u | u | u | u | u | u | u | u | u | u |
275 -----------------------------------------------------------------
276 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
277
278 tp->pos = 7
279 tp->bits = 3
280 n->pos = 15
Olof Johansson91b9a272005-08-09 20:24:39 -0700281 n->bits = 4
Robert Olsson19baf832005-06-21 12:43:18 -0700282
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900283 First, let's just ignore the bits that come before the parent tp, that is
284 the bits from 0 to (tp->pos-1). They are *known* but at this point we do
Robert Olsson19baf832005-06-21 12:43:18 -0700285 not use them for anything.
286
287 The bits from (tp->pos) to (tp->pos + tp->bits - 1) - "N", above - are the
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900288 index into the parent's child array. That is, they will be used to find
Robert Olsson19baf832005-06-21 12:43:18 -0700289 'n' among tp's children.
290
291 The bits from (tp->pos + tp->bits) to (n->pos - 1) - "S" - are skipped bits
292 for the node n.
293
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900294 All the bits we have seen so far are significant to the node n. The rest
Robert Olsson19baf832005-06-21 12:43:18 -0700295 of the bits are really not needed or indeed known in n->key.
296
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900297 The bits from (n->pos) to (n->pos + n->bits - 1) - "C" - are the index into
Robert Olsson19baf832005-06-21 12:43:18 -0700298 n's child array, and will of course be different for each child.
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900299
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700300
Robert Olsson19baf832005-06-21 12:43:18 -0700301 The rest of the bits, from (n->pos + n->bits) onward, are completely unknown
302 at this point.
303
304*/
305
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700306static inline void check_tnode(const struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700307{
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700308 WARN_ON(tn && tn->pos+tn->bits > 32);
Robert Olsson19baf832005-06-21 12:43:18 -0700309}
310
Denis V. Lunevf5026fa2007-12-13 09:47:57 -0800311static const int halve_threshold = 25;
312static const int inflate_threshold = 50;
313static const int halve_threshold_root = 8;
314static const int inflate_threshold_root = 15;
Robert Olsson19baf832005-06-21 12:43:18 -0700315
Robert Olsson2373ce12005-08-25 13:01:29 -0700316
317static void __alias_free_mem(struct rcu_head *head)
318{
319 struct fib_alias *fa = container_of(head, struct fib_alias, rcu);
320 kmem_cache_free(fn_alias_kmem, fa);
321}
322
323static inline void alias_free_mem_rcu(struct fib_alias *fa)
324{
325 call_rcu(&fa->rcu, __alias_free_mem);
326}
327
328static void __leaf_free_rcu(struct rcu_head *head)
329{
Stephen Hemmingerbc3c8c12008-01-22 21:51:50 -0800330 struct leaf *l = container_of(head, struct leaf, rcu);
331 kmem_cache_free(trie_leaf_kmem, l);
Robert Olsson2373ce12005-08-25 13:01:29 -0700332}
333
Robert Olsson2373ce12005-08-25 13:01:29 -0700334static void __leaf_info_free_rcu(struct rcu_head *head)
335{
336 kfree(container_of(head, struct leaf_info, rcu));
337}
338
339static inline void free_leaf_info(struct leaf_info *leaf)
340{
341 call_rcu(&leaf->rcu, __leaf_info_free_rcu);
342}
343
Eric Dumazet8d965442008-01-13 22:31:44 -0800344static struct tnode *tnode_alloc(size_t size)
Robert Olsson2373ce12005-08-25 13:01:29 -0700345{
346 struct page *pages;
347
348 if (size <= PAGE_SIZE)
Eric Dumazet8d965442008-01-13 22:31:44 -0800349 return kzalloc(size, GFP_KERNEL);
Robert Olsson2373ce12005-08-25 13:01:29 -0700350
351 pages = alloc_pages(GFP_KERNEL|__GFP_ZERO, get_order(size));
352 if (!pages)
353 return NULL;
354
355 return page_address(pages);
356}
357
358static void __tnode_free_rcu(struct rcu_head *head)
359{
360 struct tnode *tn = container_of(head, struct tnode, rcu);
Eric Dumazet8d965442008-01-13 22:31:44 -0800361 size_t size = sizeof(struct tnode) +
362 (sizeof(struct node *) << tn->bits);
Robert Olsson2373ce12005-08-25 13:01:29 -0700363
364 if (size <= PAGE_SIZE)
365 kfree(tn);
366 else
367 free_pages((unsigned long)tn, get_order(size));
368}
369
370static inline void tnode_free(struct tnode *tn)
371{
Stephen Hemminger132adf52007-03-08 20:44:43 -0800372 if (IS_LEAF(tn)) {
Robert Olsson550e29b2006-04-04 12:53:35 -0700373 struct leaf *l = (struct leaf *) tn;
374 call_rcu_bh(&l->rcu, __leaf_free_rcu);
Stephen Hemminger132adf52007-03-08 20:44:43 -0800375 } else
Robert Olsson550e29b2006-04-04 12:53:35 -0700376 call_rcu(&tn->rcu, __tnode_free_rcu);
Robert Olsson2373ce12005-08-25 13:01:29 -0700377}
378
Robert Olsson19baf832005-06-21 12:43:18 -0700379static struct leaf *leaf_new(void)
380{
Stephen Hemmingerbc3c8c12008-01-22 21:51:50 -0800381 struct leaf *l = kmem_cache_alloc(trie_leaf_kmem, GFP_KERNEL);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700382 if (l) {
Robert Olsson2373ce12005-08-25 13:01:29 -0700383 l->parent = T_LEAF;
Robert Olsson19baf832005-06-21 12:43:18 -0700384 INIT_HLIST_HEAD(&l->list);
385 }
386 return l;
387}
388
389static struct leaf_info *leaf_info_new(int plen)
390{
391 struct leaf_info *li = kmalloc(sizeof(struct leaf_info), GFP_KERNEL);
Robert Olsson2373ce12005-08-25 13:01:29 -0700392 if (li) {
393 li->plen = plen;
394 INIT_LIST_HEAD(&li->falh);
Patrick McHardyf0e36f82005-07-05 14:44:55 -0700395 }
Robert Olsson2373ce12005-08-25 13:01:29 -0700396 return li;
Patrick McHardyf0e36f82005-07-05 14:44:55 -0700397}
398
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800399static struct tnode *tnode_new(t_key key, int pos, int bits)
Robert Olsson19baf832005-06-21 12:43:18 -0700400{
Eric Dumazet8d965442008-01-13 22:31:44 -0800401 size_t sz = sizeof(struct tnode) + (sizeof(struct node *) << bits);
Patrick McHardyf0e36f82005-07-05 14:44:55 -0700402 struct tnode *tn = tnode_alloc(sz);
Robert Olsson19baf832005-06-21 12:43:18 -0700403
Olof Johansson91b9a272005-08-09 20:24:39 -0700404 if (tn) {
Robert Olsson2373ce12005-08-25 13:01:29 -0700405 tn->parent = T_TNODE;
Robert Olsson19baf832005-06-21 12:43:18 -0700406 tn->pos = pos;
407 tn->bits = bits;
408 tn->key = key;
409 tn->full_children = 0;
410 tn->empty_children = 1<<bits;
411 }
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700412
Eric Dumazet8d965442008-01-13 22:31:44 -0800413 pr_debug("AT %p s=%u %lu\n", tn, (unsigned int) sizeof(struct tnode),
414 (unsigned long) (sizeof(struct node) << bits));
Robert Olsson19baf832005-06-21 12:43:18 -0700415 return tn;
416}
417
Robert Olsson19baf832005-06-21 12:43:18 -0700418/*
419 * Check whether a tnode 'n' is "full", i.e. it is an internal node
420 * and no bits are skipped. See discussion in dyntree paper p. 6
421 */
422
Stephen Hemmignerbb435b82005-08-09 20:25:39 -0700423static inline int tnode_full(const struct tnode *tn, const struct node *n)
Robert Olsson19baf832005-06-21 12:43:18 -0700424{
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700425 if (n == NULL || IS_LEAF(n))
Robert Olsson19baf832005-06-21 12:43:18 -0700426 return 0;
427
428 return ((struct tnode *) n)->pos == tn->pos + tn->bits;
429}
430
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800431static inline void put_child(struct trie *t, struct tnode *tn, int i,
432 struct node *n)
Robert Olsson19baf832005-06-21 12:43:18 -0700433{
434 tnode_put_child_reorg(tn, i, n, -1);
435}
436
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700437 /*
Robert Olsson19baf832005-06-21 12:43:18 -0700438 * Add a child at position i overwriting the old value.
439 * Update the value of full_children and empty_children.
440 */
441
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800442static void tnode_put_child_reorg(struct tnode *tn, int i, struct node *n,
443 int wasfull)
Robert Olsson19baf832005-06-21 12:43:18 -0700444{
Robert Olsson2373ce12005-08-25 13:01:29 -0700445 struct node *chi = tn->child[i];
Robert Olsson19baf832005-06-21 12:43:18 -0700446 int isfull;
447
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700448 BUG_ON(i >= 1<<tn->bits);
449
Robert Olsson19baf832005-06-21 12:43:18 -0700450
451 /* update emptyChildren */
452 if (n == NULL && chi != NULL)
453 tn->empty_children++;
454 else if (n != NULL && chi == NULL)
455 tn->empty_children--;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700456
Robert Olsson19baf832005-06-21 12:43:18 -0700457 /* update fullChildren */
Olof Johansson91b9a272005-08-09 20:24:39 -0700458 if (wasfull == -1)
Robert Olsson19baf832005-06-21 12:43:18 -0700459 wasfull = tnode_full(tn, chi);
460
461 isfull = tnode_full(tn, n);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700462 if (wasfull && !isfull)
Robert Olsson19baf832005-06-21 12:43:18 -0700463 tn->full_children--;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700464 else if (!wasfull && isfull)
Robert Olsson19baf832005-06-21 12:43:18 -0700465 tn->full_children++;
Olof Johansson91b9a272005-08-09 20:24:39 -0700466
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700467 if (n)
Stephen Hemminger06801912007-08-10 15:22:13 -0700468 node_set_parent(n, tn);
Robert Olsson19baf832005-06-21 12:43:18 -0700469
Robert Olsson2373ce12005-08-25 13:01:29 -0700470 rcu_assign_pointer(tn->child[i], n);
Robert Olsson19baf832005-06-21 12:43:18 -0700471}
472
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700473static struct node *resize(struct trie *t, struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700474{
475 int i;
Robert Olsson2f368952005-07-05 15:02:40 -0700476 int err = 0;
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700477 struct tnode *old_tn;
Robert Olssone6308be2005-10-04 13:01:58 -0700478 int inflate_threshold_use;
479 int halve_threshold_use;
Robert Olsson05eee482007-03-19 16:27:37 -0700480 int max_resize;
Robert Olsson19baf832005-06-21 12:43:18 -0700481
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900482 if (!tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700483 return NULL;
484
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700485 pr_debug("In tnode_resize %p inflate_threshold=%d threshold=%d\n",
486 tn, inflate_threshold, halve_threshold);
Robert Olsson19baf832005-06-21 12:43:18 -0700487
488 /* No children */
489 if (tn->empty_children == tnode_child_length(tn)) {
490 tnode_free(tn);
491 return NULL;
492 }
493 /* One child */
494 if (tn->empty_children == tnode_child_length(tn) - 1)
495 for (i = 0; i < tnode_child_length(tn); i++) {
Olof Johansson91b9a272005-08-09 20:24:39 -0700496 struct node *n;
Robert Olsson19baf832005-06-21 12:43:18 -0700497
Olof Johansson91b9a272005-08-09 20:24:39 -0700498 n = tn->child[i];
Robert Olsson2373ce12005-08-25 13:01:29 -0700499 if (!n)
Olof Johansson91b9a272005-08-09 20:24:39 -0700500 continue;
Olof Johansson91b9a272005-08-09 20:24:39 -0700501
502 /* compress one level */
Stephen Hemminger06801912007-08-10 15:22:13 -0700503 node_set_parent(n, NULL);
Olof Johansson91b9a272005-08-09 20:24:39 -0700504 tnode_free(tn);
505 return n;
Robert Olsson19baf832005-06-21 12:43:18 -0700506 }
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700507 /*
Robert Olsson19baf832005-06-21 12:43:18 -0700508 * Double as long as the resulting node has a number of
509 * nonempty nodes that are above the threshold.
510 */
511
512 /*
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700513 * From "Implementing a dynamic compressed trie" by Stefan Nilsson of
514 * the Helsinki University of Technology and Matti Tikkanen of Nokia
Robert Olsson19baf832005-06-21 12:43:18 -0700515 * Telecommunications, page 6:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700516 * "A node is doubled if the ratio of non-empty children to all
Robert Olsson19baf832005-06-21 12:43:18 -0700517 * children in the *doubled* node is at least 'high'."
518 *
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700519 * 'high' in this instance is the variable 'inflate_threshold'. It
520 * is expressed as a percentage, so we multiply it with
521 * tnode_child_length() and instead of multiplying by 2 (since the
522 * child array will be doubled by inflate()) and multiplying
523 * the left-hand side by 100 (to handle the percentage thing) we
Robert Olsson19baf832005-06-21 12:43:18 -0700524 * multiply the left-hand side by 50.
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700525 *
526 * The left-hand side may look a bit weird: tnode_child_length(tn)
527 * - tn->empty_children is of course the number of non-null children
528 * in the current node. tn->full_children is the number of "full"
Robert Olsson19baf832005-06-21 12:43:18 -0700529 * children, that is non-null tnodes with a skip value of 0.
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700530 * All of those will be doubled in the resulting inflated tnode, so
Robert Olsson19baf832005-06-21 12:43:18 -0700531 * we just count them one extra time here.
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700532 *
Robert Olsson19baf832005-06-21 12:43:18 -0700533 * A clearer way to write this would be:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700534 *
Robert Olsson19baf832005-06-21 12:43:18 -0700535 * to_be_doubled = tn->full_children;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700536 * not_to_be_doubled = tnode_child_length(tn) - tn->empty_children -
Robert Olsson19baf832005-06-21 12:43:18 -0700537 * tn->full_children;
538 *
539 * new_child_length = tnode_child_length(tn) * 2;
540 *
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700541 * new_fill_factor = 100 * (not_to_be_doubled + 2*to_be_doubled) /
Robert Olsson19baf832005-06-21 12:43:18 -0700542 * new_child_length;
543 * if (new_fill_factor >= inflate_threshold)
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700544 *
545 * ...and so on, tho it would mess up the while () loop.
546 *
Robert Olsson19baf832005-06-21 12:43:18 -0700547 * anyway,
548 * 100 * (not_to_be_doubled + 2*to_be_doubled) / new_child_length >=
549 * inflate_threshold
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700550 *
Robert Olsson19baf832005-06-21 12:43:18 -0700551 * avoid a division:
552 * 100 * (not_to_be_doubled + 2*to_be_doubled) >=
553 * inflate_threshold * new_child_length
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700554 *
Robert Olsson19baf832005-06-21 12:43:18 -0700555 * expand not_to_be_doubled and to_be_doubled, and shorten:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700556 * 100 * (tnode_child_length(tn) - tn->empty_children +
Olof Johansson91b9a272005-08-09 20:24:39 -0700557 * tn->full_children) >= inflate_threshold * new_child_length
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700558 *
Robert Olsson19baf832005-06-21 12:43:18 -0700559 * expand new_child_length:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700560 * 100 * (tnode_child_length(tn) - tn->empty_children +
Olof Johansson91b9a272005-08-09 20:24:39 -0700561 * tn->full_children) >=
Robert Olsson19baf832005-06-21 12:43:18 -0700562 * inflate_threshold * tnode_child_length(tn) * 2
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700563 *
Robert Olsson19baf832005-06-21 12:43:18 -0700564 * shorten again:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700565 * 50 * (tn->full_children + tnode_child_length(tn) -
Olof Johansson91b9a272005-08-09 20:24:39 -0700566 * tn->empty_children) >= inflate_threshold *
Robert Olsson19baf832005-06-21 12:43:18 -0700567 * tnode_child_length(tn)
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700568 *
Robert Olsson19baf832005-06-21 12:43:18 -0700569 */
570
571 check_tnode(tn);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700572
Robert Olssone6308be2005-10-04 13:01:58 -0700573 /* Keep root node larger */
574
Stephen Hemminger132adf52007-03-08 20:44:43 -0800575 if (!tn->parent)
Robert Olssone6308be2005-10-04 13:01:58 -0700576 inflate_threshold_use = inflate_threshold_root;
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900577 else
Robert Olssone6308be2005-10-04 13:01:58 -0700578 inflate_threshold_use = inflate_threshold;
579
Robert Olsson2f368952005-07-05 15:02:40 -0700580 err = 0;
Robert Olsson05eee482007-03-19 16:27:37 -0700581 max_resize = 10;
582 while ((tn->full_children > 0 && max_resize-- &&
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800583 50 * (tn->full_children + tnode_child_length(tn)
584 - tn->empty_children)
585 >= inflate_threshold_use * tnode_child_length(tn))) {
Robert Olsson19baf832005-06-21 12:43:18 -0700586
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700587 old_tn = tn;
588 tn = inflate(t, tn);
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800589
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700590 if (IS_ERR(tn)) {
591 tn = old_tn;
Robert Olsson2f368952005-07-05 15:02:40 -0700592#ifdef CONFIG_IP_FIB_TRIE_STATS
593 t->stats.resize_node_skipped++;
594#endif
595 break;
596 }
Robert Olsson19baf832005-06-21 12:43:18 -0700597 }
598
Robert Olsson05eee482007-03-19 16:27:37 -0700599 if (max_resize < 0) {
600 if (!tn->parent)
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800601 pr_warning("Fix inflate_threshold_root."
602 " Now=%d size=%d bits\n",
603 inflate_threshold_root, tn->bits);
Robert Olsson05eee482007-03-19 16:27:37 -0700604 else
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800605 pr_warning("Fix inflate_threshold."
606 " Now=%d size=%d bits\n",
607 inflate_threshold, tn->bits);
Robert Olsson05eee482007-03-19 16:27:37 -0700608 }
609
Robert Olsson19baf832005-06-21 12:43:18 -0700610 check_tnode(tn);
611
612 /*
613 * Halve as long as the number of empty children in this
614 * node is above threshold.
615 */
Robert Olsson2f368952005-07-05 15:02:40 -0700616
Robert Olssone6308be2005-10-04 13:01:58 -0700617
618 /* Keep root node larger */
619
Stephen Hemminger132adf52007-03-08 20:44:43 -0800620 if (!tn->parent)
Robert Olssone6308be2005-10-04 13:01:58 -0700621 halve_threshold_use = halve_threshold_root;
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900622 else
Robert Olssone6308be2005-10-04 13:01:58 -0700623 halve_threshold_use = halve_threshold;
624
Robert Olsson2f368952005-07-05 15:02:40 -0700625 err = 0;
Robert Olsson05eee482007-03-19 16:27:37 -0700626 max_resize = 10;
627 while (tn->bits > 1 && max_resize-- &&
Robert Olsson19baf832005-06-21 12:43:18 -0700628 100 * (tnode_child_length(tn) - tn->empty_children) <
Robert Olssone6308be2005-10-04 13:01:58 -0700629 halve_threshold_use * tnode_child_length(tn)) {
Robert Olsson19baf832005-06-21 12:43:18 -0700630
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700631 old_tn = tn;
632 tn = halve(t, tn);
633 if (IS_ERR(tn)) {
634 tn = old_tn;
Robert Olsson2f368952005-07-05 15:02:40 -0700635#ifdef CONFIG_IP_FIB_TRIE_STATS
636 t->stats.resize_node_skipped++;
637#endif
638 break;
639 }
640 }
641
Robert Olsson05eee482007-03-19 16:27:37 -0700642 if (max_resize < 0) {
643 if (!tn->parent)
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800644 pr_warning("Fix halve_threshold_root."
645 " Now=%d size=%d bits\n",
646 halve_threshold_root, tn->bits);
Robert Olsson05eee482007-03-19 16:27:37 -0700647 else
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800648 pr_warning("Fix halve_threshold."
649 " Now=%d size=%d bits\n",
650 halve_threshold, tn->bits);
Robert Olsson05eee482007-03-19 16:27:37 -0700651 }
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700652
Robert Olsson19baf832005-06-21 12:43:18 -0700653 /* Only one child remains */
Robert Olsson19baf832005-06-21 12:43:18 -0700654 if (tn->empty_children == tnode_child_length(tn) - 1)
655 for (i = 0; i < tnode_child_length(tn); i++) {
Olof Johansson91b9a272005-08-09 20:24:39 -0700656 struct node *n;
657
Olof Johansson91b9a272005-08-09 20:24:39 -0700658 n = tn->child[i];
Robert Olsson2373ce12005-08-25 13:01:29 -0700659 if (!n)
Olof Johansson91b9a272005-08-09 20:24:39 -0700660 continue;
Olof Johansson91b9a272005-08-09 20:24:39 -0700661
662 /* compress one level */
663
Stephen Hemminger06801912007-08-10 15:22:13 -0700664 node_set_parent(n, NULL);
Olof Johansson91b9a272005-08-09 20:24:39 -0700665 tnode_free(tn);
666 return n;
Robert Olsson19baf832005-06-21 12:43:18 -0700667 }
668
669 return (struct node *) tn;
670}
671
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700672static struct tnode *inflate(struct trie *t, struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700673{
Robert Olsson19baf832005-06-21 12:43:18 -0700674 struct tnode *oldtnode = tn;
675 int olen = tnode_child_length(tn);
676 int i;
677
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700678 pr_debug("In inflate\n");
Robert Olsson19baf832005-06-21 12:43:18 -0700679
680 tn = tnode_new(oldtnode->key, oldtnode->pos, oldtnode->bits + 1);
681
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700682 if (!tn)
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700683 return ERR_PTR(-ENOMEM);
Robert Olsson2f368952005-07-05 15:02:40 -0700684
685 /*
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700686 * Preallocate and store tnodes before the actual work so we
687 * don't get into an inconsistent state if memory allocation
688 * fails. In case of failure we return the oldnode and inflate
Robert Olsson2f368952005-07-05 15:02:40 -0700689 * of tnode is ignored.
690 */
Olof Johansson91b9a272005-08-09 20:24:39 -0700691
692 for (i = 0; i < olen; i++) {
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800693 struct tnode *inode;
Robert Olsson2f368952005-07-05 15:02:40 -0700694
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800695 inode = (struct tnode *) tnode_get_child(oldtnode, i);
Robert Olsson2f368952005-07-05 15:02:40 -0700696 if (inode &&
697 IS_TNODE(inode) &&
698 inode->pos == oldtnode->pos + oldtnode->bits &&
699 inode->bits > 1) {
700 struct tnode *left, *right;
Stephen Hemmingerab66b4a2007-08-10 15:22:58 -0700701 t_key m = ~0U << (KEYLENGTH - 1) >> inode->pos;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700702
Robert Olsson2f368952005-07-05 15:02:40 -0700703 left = tnode_new(inode->key&(~m), inode->pos + 1,
704 inode->bits - 1);
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700705 if (!left)
706 goto nomem;
Olof Johansson91b9a272005-08-09 20:24:39 -0700707
Robert Olsson2f368952005-07-05 15:02:40 -0700708 right = tnode_new(inode->key|m, inode->pos + 1,
709 inode->bits - 1);
710
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900711 if (!right) {
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700712 tnode_free(left);
713 goto nomem;
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900714 }
Robert Olsson2f368952005-07-05 15:02:40 -0700715
716 put_child(t, tn, 2*i, (struct node *) left);
717 put_child(t, tn, 2*i+1, (struct node *) right);
718 }
719 }
720
Olof Johansson91b9a272005-08-09 20:24:39 -0700721 for (i = 0; i < olen; i++) {
Stephen Hemmingerc95aaf92008-01-12 21:25:02 -0800722 struct tnode *inode;
Robert Olsson19baf832005-06-21 12:43:18 -0700723 struct node *node = tnode_get_child(oldtnode, i);
Olof Johansson91b9a272005-08-09 20:24:39 -0700724 struct tnode *left, *right;
725 int size, j;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700726
Robert Olsson19baf832005-06-21 12:43:18 -0700727 /* An empty child */
728 if (node == NULL)
729 continue;
730
731 /* A leaf or an internal node with skipped bits */
732
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700733 if (IS_LEAF(node) || ((struct tnode *) node)->pos >
Robert Olsson19baf832005-06-21 12:43:18 -0700734 tn->pos + tn->bits - 1) {
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800735 if (tkey_extract_bits(node->key,
736 oldtnode->pos + oldtnode->bits,
737 1) == 0)
Robert Olsson19baf832005-06-21 12:43:18 -0700738 put_child(t, tn, 2*i, node);
739 else
740 put_child(t, tn, 2*i+1, node);
741 continue;
742 }
743
744 /* An internal node with two children */
745 inode = (struct tnode *) node;
746
747 if (inode->bits == 1) {
748 put_child(t, tn, 2*i, inode->child[0]);
749 put_child(t, tn, 2*i+1, inode->child[1]);
750
751 tnode_free(inode);
Olof Johansson91b9a272005-08-09 20:24:39 -0700752 continue;
Robert Olsson19baf832005-06-21 12:43:18 -0700753 }
754
Olof Johansson91b9a272005-08-09 20:24:39 -0700755 /* An internal node with more than two children */
Robert Olsson19baf832005-06-21 12:43:18 -0700756
Olof Johansson91b9a272005-08-09 20:24:39 -0700757 /* We will replace this node 'inode' with two new
758 * ones, 'left' and 'right', each with half of the
759 * original children. The two new nodes will have
760 * a position one bit further down the key and this
761 * means that the "significant" part of their keys
762 * (see the discussion near the top of this file)
763 * will differ by one bit, which will be "0" in
764 * left's key and "1" in right's key. Since we are
765 * moving the key position by one step, the bit that
766 * we are moving away from - the bit at position
767 * (inode->pos) - is the one that will differ between
768 * left and right. So... we synthesize that bit in the
769 * two new keys.
770 * The mask 'm' below will be a single "one" bit at
771 * the position (inode->pos)
772 */
Robert Olsson19baf832005-06-21 12:43:18 -0700773
Olof Johansson91b9a272005-08-09 20:24:39 -0700774 /* Use the old key, but set the new significant
775 * bit to zero.
776 */
Robert Olsson19baf832005-06-21 12:43:18 -0700777
Olof Johansson91b9a272005-08-09 20:24:39 -0700778 left = (struct tnode *) tnode_get_child(tn, 2*i);
779 put_child(t, tn, 2*i, NULL);
Robert Olsson19baf832005-06-21 12:43:18 -0700780
Olof Johansson91b9a272005-08-09 20:24:39 -0700781 BUG_ON(!left);
Robert Olsson2f368952005-07-05 15:02:40 -0700782
Olof Johansson91b9a272005-08-09 20:24:39 -0700783 right = (struct tnode *) tnode_get_child(tn, 2*i+1);
784 put_child(t, tn, 2*i+1, NULL);
Robert Olsson2f368952005-07-05 15:02:40 -0700785
Olof Johansson91b9a272005-08-09 20:24:39 -0700786 BUG_ON(!right);
Robert Olsson2f368952005-07-05 15:02:40 -0700787
Olof Johansson91b9a272005-08-09 20:24:39 -0700788 size = tnode_child_length(left);
789 for (j = 0; j < size; j++) {
790 put_child(t, left, j, inode->child[j]);
791 put_child(t, right, j, inode->child[j + size]);
Robert Olsson19baf832005-06-21 12:43:18 -0700792 }
Olof Johansson91b9a272005-08-09 20:24:39 -0700793 put_child(t, tn, 2*i, resize(t, left));
794 put_child(t, tn, 2*i+1, resize(t, right));
795
796 tnode_free(inode);
Robert Olsson19baf832005-06-21 12:43:18 -0700797 }
798 tnode_free(oldtnode);
799 return tn;
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700800nomem:
801 {
802 int size = tnode_child_length(tn);
803 int j;
804
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700805 for (j = 0; j < size; j++)
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700806 if (tn->child[j])
807 tnode_free((struct tnode *)tn->child[j]);
808
809 tnode_free(tn);
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700810
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700811 return ERR_PTR(-ENOMEM);
812 }
Robert Olsson19baf832005-06-21 12:43:18 -0700813}
814
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700815static struct tnode *halve(struct trie *t, struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700816{
817 struct tnode *oldtnode = tn;
818 struct node *left, *right;
819 int i;
820 int olen = tnode_child_length(tn);
821
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700822 pr_debug("In halve\n");
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700823
824 tn = tnode_new(oldtnode->key, oldtnode->pos, oldtnode->bits - 1);
Robert Olsson19baf832005-06-21 12:43:18 -0700825
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700826 if (!tn)
827 return ERR_PTR(-ENOMEM);
Robert Olsson2f368952005-07-05 15:02:40 -0700828
829 /*
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700830 * Preallocate and store tnodes before the actual work so we
831 * don't get into an inconsistent state if memory allocation
832 * fails. In case of failure we return the oldnode and halve
Robert Olsson2f368952005-07-05 15:02:40 -0700833 * of tnode is ignored.
834 */
835
Olof Johansson91b9a272005-08-09 20:24:39 -0700836 for (i = 0; i < olen; i += 2) {
Robert Olsson2f368952005-07-05 15:02:40 -0700837 left = tnode_get_child(oldtnode, i);
838 right = tnode_get_child(oldtnode, i+1);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700839
Robert Olsson2f368952005-07-05 15:02:40 -0700840 /* Two nonempty children */
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700841 if (left && right) {
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700842 struct tnode *newn;
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700843
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700844 newn = tnode_new(left->key, tn->pos + tn->bits, 1);
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700845
846 if (!newn)
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700847 goto nomem;
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700848
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700849 put_child(t, tn, i/2, (struct node *)newn);
Robert Olsson2f368952005-07-05 15:02:40 -0700850 }
Robert Olsson2f368952005-07-05 15:02:40 -0700851
Robert Olsson2f368952005-07-05 15:02:40 -0700852 }
Robert Olsson19baf832005-06-21 12:43:18 -0700853
Olof Johansson91b9a272005-08-09 20:24:39 -0700854 for (i = 0; i < olen; i += 2) {
855 struct tnode *newBinNode;
856
Robert Olsson19baf832005-06-21 12:43:18 -0700857 left = tnode_get_child(oldtnode, i);
858 right = tnode_get_child(oldtnode, i+1);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700859
Robert Olsson19baf832005-06-21 12:43:18 -0700860 /* At least one of the children is empty */
861 if (left == NULL) {
862 if (right == NULL) /* Both are empty */
863 continue;
864 put_child(t, tn, i/2, right);
Olof Johansson91b9a272005-08-09 20:24:39 -0700865 continue;
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700866 }
Olof Johansson91b9a272005-08-09 20:24:39 -0700867
868 if (right == NULL) {
Robert Olsson19baf832005-06-21 12:43:18 -0700869 put_child(t, tn, i/2, left);
Olof Johansson91b9a272005-08-09 20:24:39 -0700870 continue;
871 }
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700872
Robert Olsson19baf832005-06-21 12:43:18 -0700873 /* Two nonempty children */
Olof Johansson91b9a272005-08-09 20:24:39 -0700874 newBinNode = (struct tnode *) tnode_get_child(tn, i/2);
875 put_child(t, tn, i/2, NULL);
Olof Johansson91b9a272005-08-09 20:24:39 -0700876 put_child(t, newBinNode, 0, left);
877 put_child(t, newBinNode, 1, right);
878 put_child(t, tn, i/2, resize(t, newBinNode));
Robert Olsson19baf832005-06-21 12:43:18 -0700879 }
880 tnode_free(oldtnode);
881 return tn;
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700882nomem:
883 {
884 int size = tnode_child_length(tn);
885 int j;
886
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700887 for (j = 0; j < size; j++)
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700888 if (tn->child[j])
889 tnode_free((struct tnode *)tn->child[j]);
890
891 tnode_free(tn);
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700892
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700893 return ERR_PTR(-ENOMEM);
894 }
Robert Olsson19baf832005-06-21 12:43:18 -0700895}
896
Robert Olsson772cb712005-09-19 15:31:18 -0700897/* readside must use rcu_read_lock currently dump routines
Robert Olsson2373ce12005-08-25 13:01:29 -0700898 via get_fa_head and dump */
899
Robert Olsson772cb712005-09-19 15:31:18 -0700900static struct leaf_info *find_leaf_info(struct leaf *l, int plen)
Robert Olsson19baf832005-06-21 12:43:18 -0700901{
Robert Olsson772cb712005-09-19 15:31:18 -0700902 struct hlist_head *head = &l->list;
Robert Olsson19baf832005-06-21 12:43:18 -0700903 struct hlist_node *node;
904 struct leaf_info *li;
905
Robert Olsson2373ce12005-08-25 13:01:29 -0700906 hlist_for_each_entry_rcu(li, node, head, hlist)
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700907 if (li->plen == plen)
Robert Olsson19baf832005-06-21 12:43:18 -0700908 return li;
Olof Johansson91b9a272005-08-09 20:24:39 -0700909
Robert Olsson19baf832005-06-21 12:43:18 -0700910 return NULL;
911}
912
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800913static inline struct list_head *get_fa_head(struct leaf *l, int plen)
Robert Olsson19baf832005-06-21 12:43:18 -0700914{
Robert Olsson772cb712005-09-19 15:31:18 -0700915 struct leaf_info *li = find_leaf_info(l, plen);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700916
Olof Johansson91b9a272005-08-09 20:24:39 -0700917 if (!li)
918 return NULL;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700919
Olof Johansson91b9a272005-08-09 20:24:39 -0700920 return &li->falh;
Robert Olsson19baf832005-06-21 12:43:18 -0700921}
922
923static void insert_leaf_info(struct hlist_head *head, struct leaf_info *new)
924{
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900925 struct leaf_info *li = NULL, *last = NULL;
926 struct hlist_node *node;
Robert Olsson19baf832005-06-21 12:43:18 -0700927
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900928 if (hlist_empty(head)) {
929 hlist_add_head_rcu(&new->hlist, head);
930 } else {
931 hlist_for_each_entry(li, node, head, hlist) {
932 if (new->plen > li->plen)
933 break;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700934
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900935 last = li;
936 }
937 if (last)
938 hlist_add_after_rcu(&last->hlist, &new->hlist);
939 else
940 hlist_add_before_rcu(&new->hlist, &li->hlist);
941 }
Robert Olsson19baf832005-06-21 12:43:18 -0700942}
943
Robert Olsson2373ce12005-08-25 13:01:29 -0700944/* rcu_read_lock needs to be hold by caller from readside */
945
Robert Olsson19baf832005-06-21 12:43:18 -0700946static struct leaf *
947fib_find_node(struct trie *t, u32 key)
948{
949 int pos;
950 struct tnode *tn;
951 struct node *n;
952
953 pos = 0;
Robert Olsson2373ce12005-08-25 13:01:29 -0700954 n = rcu_dereference(t->trie);
Robert Olsson19baf832005-06-21 12:43:18 -0700955
956 while (n != NULL && NODE_TYPE(n) == T_TNODE) {
957 tn = (struct tnode *) n;
Olof Johansson91b9a272005-08-09 20:24:39 -0700958
Robert Olsson19baf832005-06-21 12:43:18 -0700959 check_tnode(tn);
Olof Johansson91b9a272005-08-09 20:24:39 -0700960
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700961 if (tkey_sub_equals(tn->key, pos, tn->pos-pos, key)) {
Olof Johansson91b9a272005-08-09 20:24:39 -0700962 pos = tn->pos + tn->bits;
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800963 n = tnode_get_child_rcu(tn,
964 tkey_extract_bits(key,
965 tn->pos,
966 tn->bits));
Olof Johansson91b9a272005-08-09 20:24:39 -0700967 } else
Robert Olsson19baf832005-06-21 12:43:18 -0700968 break;
969 }
970 /* Case we have found a leaf. Compare prefixes */
971
Olof Johansson91b9a272005-08-09 20:24:39 -0700972 if (n != NULL && IS_LEAF(n) && tkey_equals(key, n->key))
973 return (struct leaf *)n;
974
Robert Olsson19baf832005-06-21 12:43:18 -0700975 return NULL;
976}
977
978static struct node *trie_rebalance(struct trie *t, struct tnode *tn)
979{
Robert Olsson19baf832005-06-21 12:43:18 -0700980 int wasfull;
Stephen Hemminger06801912007-08-10 15:22:13 -0700981 t_key cindex, key = tn->key;
982 struct tnode *tp;
Robert Olsson19baf832005-06-21 12:43:18 -0700983
Stephen Hemminger06801912007-08-10 15:22:13 -0700984 while (tn != NULL && (tp = node_parent((struct node *)tn)) != NULL) {
Robert Olsson19baf832005-06-21 12:43:18 -0700985 cindex = tkey_extract_bits(key, tp->pos, tp->bits);
986 wasfull = tnode_full(tp, tnode_get_child(tp, cindex));
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800987 tn = (struct tnode *) resize(t, (struct tnode *)tn);
988
989 tnode_put_child_reorg((struct tnode *)tp, cindex,
990 (struct node *)tn, wasfull);
Olof Johansson91b9a272005-08-09 20:24:39 -0700991
Stephen Hemminger06801912007-08-10 15:22:13 -0700992 tp = node_parent((struct node *) tn);
993 if (!tp)
Robert Olsson19baf832005-06-21 12:43:18 -0700994 break;
Stephen Hemminger06801912007-08-10 15:22:13 -0700995 tn = tp;
Robert Olsson19baf832005-06-21 12:43:18 -0700996 }
Stephen Hemminger06801912007-08-10 15:22:13 -0700997
Robert Olsson19baf832005-06-21 12:43:18 -0700998 /* Handle last (top) tnode */
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700999 if (IS_TNODE(tn))
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001000 tn = (struct tnode *)resize(t, (struct tnode *)tn);
Robert Olsson19baf832005-06-21 12:43:18 -07001001
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001002 return (struct node *)tn;
Robert Olsson19baf832005-06-21 12:43:18 -07001003}
1004
Robert Olsson2373ce12005-08-25 13:01:29 -07001005/* only used from updater-side */
1006
Stephen Hemmingerfea86ad2008-01-12 20:57:07 -08001007static struct list_head *fib_insert_node(struct trie *t, u32 key, int plen)
Robert Olsson19baf832005-06-21 12:43:18 -07001008{
1009 int pos, newpos;
1010 struct tnode *tp = NULL, *tn = NULL;
1011 struct node *n;
1012 struct leaf *l;
1013 int missbit;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001014 struct list_head *fa_head = NULL;
Robert Olsson19baf832005-06-21 12:43:18 -07001015 struct leaf_info *li;
1016 t_key cindex;
1017
1018 pos = 0;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001019 n = t->trie;
Robert Olsson19baf832005-06-21 12:43:18 -07001020
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001021 /* If we point to NULL, stop. Either the tree is empty and we should
1022 * just put a new leaf in if, or we have reached an empty child slot,
Robert Olsson19baf832005-06-21 12:43:18 -07001023 * and we should just put our new leaf in that.
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001024 * If we point to a T_TNODE, check if it matches our key. Note that
1025 * a T_TNODE might be skipping any number of bits - its 'pos' need
Robert Olsson19baf832005-06-21 12:43:18 -07001026 * not be the parent's 'pos'+'bits'!
1027 *
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001028 * If it does match the current key, get pos/bits from it, extract
Robert Olsson19baf832005-06-21 12:43:18 -07001029 * the index from our key, push the T_TNODE and walk the tree.
1030 *
1031 * If it doesn't, we have to replace it with a new T_TNODE.
1032 *
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001033 * If we point to a T_LEAF, it might or might not have the same key
1034 * as we do. If it does, just change the value, update the T_LEAF's
1035 * value, and return it.
Robert Olsson19baf832005-06-21 12:43:18 -07001036 * If it doesn't, we need to replace it with a T_TNODE.
1037 */
1038
1039 while (n != NULL && NODE_TYPE(n) == T_TNODE) {
1040 tn = (struct tnode *) n;
Olof Johansson91b9a272005-08-09 20:24:39 -07001041
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001042 check_tnode(tn);
Olof Johansson91b9a272005-08-09 20:24:39 -07001043
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001044 if (tkey_sub_equals(tn->key, pos, tn->pos-pos, key)) {
Robert Olsson19baf832005-06-21 12:43:18 -07001045 tp = tn;
Olof Johansson91b9a272005-08-09 20:24:39 -07001046 pos = tn->pos + tn->bits;
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001047 n = tnode_get_child(tn,
1048 tkey_extract_bits(key,
1049 tn->pos,
1050 tn->bits));
Robert Olsson19baf832005-06-21 12:43:18 -07001051
Stephen Hemminger06801912007-08-10 15:22:13 -07001052 BUG_ON(n && node_parent(n) != tn);
Olof Johansson91b9a272005-08-09 20:24:39 -07001053 } else
Robert Olsson19baf832005-06-21 12:43:18 -07001054 break;
1055 }
1056
1057 /*
1058 * n ----> NULL, LEAF or TNODE
1059 *
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001060 * tp is n's (parent) ----> NULL or TNODE
Robert Olsson19baf832005-06-21 12:43:18 -07001061 */
1062
Olof Johansson91b9a272005-08-09 20:24:39 -07001063 BUG_ON(tp && IS_LEAF(tp));
Robert Olsson19baf832005-06-21 12:43:18 -07001064
1065 /* Case 1: n is a leaf. Compare prefixes */
1066
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001067 if (n != NULL && IS_LEAF(n) && tkey_equals(key, n->key)) {
Stephen Hemmingerc95aaf92008-01-12 21:25:02 -08001068 l = (struct leaf *) n;
Robert Olsson19baf832005-06-21 12:43:18 -07001069 li = leaf_info_new(plen);
Olof Johansson91b9a272005-08-09 20:24:39 -07001070
Stephen Hemmingerfea86ad2008-01-12 20:57:07 -08001071 if (!li)
1072 return NULL;
Robert Olsson19baf832005-06-21 12:43:18 -07001073
1074 fa_head = &li->falh;
1075 insert_leaf_info(&l->list, li);
1076 goto done;
1077 }
Robert Olsson19baf832005-06-21 12:43:18 -07001078 l = leaf_new();
1079
Stephen Hemmingerfea86ad2008-01-12 20:57:07 -08001080 if (!l)
1081 return NULL;
Robert Olsson19baf832005-06-21 12:43:18 -07001082
1083 l->key = key;
1084 li = leaf_info_new(plen);
1085
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001086 if (!li) {
Robert Olssonf835e472005-06-28 15:00:39 -07001087 tnode_free((struct tnode *) l);
Stephen Hemmingerfea86ad2008-01-12 20:57:07 -08001088 return NULL;
Robert Olssonf835e472005-06-28 15:00:39 -07001089 }
Robert Olsson19baf832005-06-21 12:43:18 -07001090
1091 fa_head = &li->falh;
1092 insert_leaf_info(&l->list, li);
1093
Robert Olsson19baf832005-06-21 12:43:18 -07001094 if (t->trie && n == NULL) {
Olof Johansson91b9a272005-08-09 20:24:39 -07001095 /* Case 2: n is NULL, and will just insert a new leaf */
Robert Olsson19baf832005-06-21 12:43:18 -07001096
Stephen Hemminger06801912007-08-10 15:22:13 -07001097 node_set_parent((struct node *)l, tp);
Robert Olsson19baf832005-06-21 12:43:18 -07001098
Olof Johansson91b9a272005-08-09 20:24:39 -07001099 cindex = tkey_extract_bits(key, tp->pos, tp->bits);
1100 put_child(t, (struct tnode *)tp, cindex, (struct node *)l);
1101 } else {
1102 /* Case 3: n is a LEAF or a TNODE and the key doesn't match. */
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001103 /*
1104 * Add a new tnode here
Robert Olsson19baf832005-06-21 12:43:18 -07001105 * first tnode need some special handling
1106 */
1107
1108 if (tp)
Olof Johansson91b9a272005-08-09 20:24:39 -07001109 pos = tp->pos+tp->bits;
Robert Olsson19baf832005-06-21 12:43:18 -07001110 else
Olof Johansson91b9a272005-08-09 20:24:39 -07001111 pos = 0;
1112
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001113 if (n) {
Robert Olsson19baf832005-06-21 12:43:18 -07001114 newpos = tkey_mismatch(key, pos, n->key);
1115 tn = tnode_new(n->key, newpos, 1);
Olof Johansson91b9a272005-08-09 20:24:39 -07001116 } else {
Robert Olsson19baf832005-06-21 12:43:18 -07001117 newpos = 0;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001118 tn = tnode_new(key, newpos, 1); /* First tnode */
Robert Olsson19baf832005-06-21 12:43:18 -07001119 }
Robert Olsson19baf832005-06-21 12:43:18 -07001120
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001121 if (!tn) {
Robert Olssonf835e472005-06-28 15:00:39 -07001122 free_leaf_info(li);
1123 tnode_free((struct tnode *) l);
Stephen Hemmingerfea86ad2008-01-12 20:57:07 -08001124 return NULL;
Olof Johansson91b9a272005-08-09 20:24:39 -07001125 }
1126
Stephen Hemminger06801912007-08-10 15:22:13 -07001127 node_set_parent((struct node *)tn, tp);
Robert Olsson19baf832005-06-21 12:43:18 -07001128
Olof Johansson91b9a272005-08-09 20:24:39 -07001129 missbit = tkey_extract_bits(key, newpos, 1);
Robert Olsson19baf832005-06-21 12:43:18 -07001130 put_child(t, tn, missbit, (struct node *)l);
1131 put_child(t, tn, 1-missbit, n);
1132
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001133 if (tp) {
Robert Olsson19baf832005-06-21 12:43:18 -07001134 cindex = tkey_extract_bits(key, tp->pos, tp->bits);
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001135 put_child(t, (struct tnode *)tp, cindex,
1136 (struct node *)tn);
Olof Johansson91b9a272005-08-09 20:24:39 -07001137 } else {
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001138 rcu_assign_pointer(t->trie, (struct node *)tn);
Robert Olsson19baf832005-06-21 12:43:18 -07001139 tp = tn;
1140 }
1141 }
Olof Johansson91b9a272005-08-09 20:24:39 -07001142
1143 if (tp && tp->pos + tp->bits > 32)
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001144 pr_warning("fib_trie"
1145 " tp=%p pos=%d, bits=%d, key=%0x plen=%d\n",
1146 tp, tp->pos, tp->bits, key, plen);
Olof Johansson91b9a272005-08-09 20:24:39 -07001147
Robert Olsson19baf832005-06-21 12:43:18 -07001148 /* Rebalance the trie */
Robert Olsson2373ce12005-08-25 13:01:29 -07001149
1150 rcu_assign_pointer(t->trie, trie_rebalance(t, tp));
Robert Olssonf835e472005-06-28 15:00:39 -07001151done:
Robert Olsson19baf832005-06-21 12:43:18 -07001152 return fa_head;
1153}
1154
Robert Olssond562f1f2007-03-26 14:22:22 -07001155/*
1156 * Caller must hold RTNL.
1157 */
Thomas Graf4e902c52006-08-17 18:14:52 -07001158static int fn_trie_insert(struct fib_table *tb, struct fib_config *cfg)
Robert Olsson19baf832005-06-21 12:43:18 -07001159{
1160 struct trie *t = (struct trie *) tb->tb_data;
1161 struct fib_alias *fa, *new_fa;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001162 struct list_head *fa_head = NULL;
Robert Olsson19baf832005-06-21 12:43:18 -07001163 struct fib_info *fi;
Thomas Graf4e902c52006-08-17 18:14:52 -07001164 int plen = cfg->fc_dst_len;
1165 u8 tos = cfg->fc_tos;
Robert Olsson19baf832005-06-21 12:43:18 -07001166 u32 key, mask;
1167 int err;
1168 struct leaf *l;
1169
1170 if (plen > 32)
1171 return -EINVAL;
1172
Thomas Graf4e902c52006-08-17 18:14:52 -07001173 key = ntohl(cfg->fc_dst);
Robert Olsson19baf832005-06-21 12:43:18 -07001174
Patrick McHardy2dfe55b2006-08-10 23:08:33 -07001175 pr_debug("Insert table=%u %08x/%d\n", tb->tb_id, key, plen);
Robert Olsson19baf832005-06-21 12:43:18 -07001176
Olof Johansson91b9a272005-08-09 20:24:39 -07001177 mask = ntohl(inet_make_mask(plen));
Robert Olsson19baf832005-06-21 12:43:18 -07001178
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001179 if (key & ~mask)
Robert Olsson19baf832005-06-21 12:43:18 -07001180 return -EINVAL;
1181
1182 key = key & mask;
1183
Thomas Graf4e902c52006-08-17 18:14:52 -07001184 fi = fib_create_info(cfg);
1185 if (IS_ERR(fi)) {
1186 err = PTR_ERR(fi);
Robert Olsson19baf832005-06-21 12:43:18 -07001187 goto err;
Thomas Graf4e902c52006-08-17 18:14:52 -07001188 }
Robert Olsson19baf832005-06-21 12:43:18 -07001189
1190 l = fib_find_node(t, key);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001191 fa = NULL;
Robert Olsson19baf832005-06-21 12:43:18 -07001192
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001193 if (l) {
Robert Olsson19baf832005-06-21 12:43:18 -07001194 fa_head = get_fa_head(l, plen);
1195 fa = fib_find_alias(fa_head, tos, fi->fib_priority);
1196 }
1197
1198 /* Now fa, if non-NULL, points to the first fib alias
1199 * with the same keys [prefix,tos,priority], if such key already
1200 * exists or to the node before which we will insert new one.
1201 *
1202 * If fa is NULL, we will need to allocate a new one and
1203 * insert to the head of f.
1204 *
1205 * If f is NULL, no fib node matched the destination key
1206 * and we need to allocate a new one of those as well.
1207 */
1208
Olof Johansson91b9a272005-08-09 20:24:39 -07001209 if (fa && fa->fa_info->fib_priority == fi->fib_priority) {
Robert Olsson19baf832005-06-21 12:43:18 -07001210 struct fib_alias *fa_orig;
1211
1212 err = -EEXIST;
Thomas Graf4e902c52006-08-17 18:14:52 -07001213 if (cfg->fc_nlflags & NLM_F_EXCL)
Robert Olsson19baf832005-06-21 12:43:18 -07001214 goto out;
1215
Thomas Graf4e902c52006-08-17 18:14:52 -07001216 if (cfg->fc_nlflags & NLM_F_REPLACE) {
Robert Olsson19baf832005-06-21 12:43:18 -07001217 struct fib_info *fi_drop;
1218 u8 state;
1219
Joonwoo Park67250332008-01-18 03:45:18 -08001220 if (fi->fib_treeref > 1)
1221 goto out;
1222
Robert Olsson2373ce12005-08-25 13:01:29 -07001223 err = -ENOBUFS;
Christoph Lametere94b1762006-12-06 20:33:17 -08001224 new_fa = kmem_cache_alloc(fn_alias_kmem, GFP_KERNEL);
Robert Olsson2373ce12005-08-25 13:01:29 -07001225 if (new_fa == NULL)
1226 goto out;
Robert Olsson19baf832005-06-21 12:43:18 -07001227
1228 fi_drop = fa->fa_info;
Robert Olsson2373ce12005-08-25 13:01:29 -07001229 new_fa->fa_tos = fa->fa_tos;
1230 new_fa->fa_info = fi;
Thomas Graf4e902c52006-08-17 18:14:52 -07001231 new_fa->fa_type = cfg->fc_type;
1232 new_fa->fa_scope = cfg->fc_scope;
Robert Olsson19baf832005-06-21 12:43:18 -07001233 state = fa->fa_state;
Robert Olsson2373ce12005-08-25 13:01:29 -07001234 new_fa->fa_state &= ~FA_S_ACCESSED;
Robert Olsson19baf832005-06-21 12:43:18 -07001235
Robert Olsson2373ce12005-08-25 13:01:29 -07001236 list_replace_rcu(&fa->fa_list, &new_fa->fa_list);
1237 alias_free_mem_rcu(fa);
Robert Olsson19baf832005-06-21 12:43:18 -07001238
1239 fib_release_info(fi_drop);
1240 if (state & FA_S_ACCESSED)
Olof Johansson91b9a272005-08-09 20:24:39 -07001241 rt_cache_flush(-1);
Milan Kocianb8f55832007-05-23 14:55:06 -07001242 rtmsg_fib(RTM_NEWROUTE, htonl(key), new_fa, plen,
1243 tb->tb_id, &cfg->fc_nlinfo, NLM_F_REPLACE);
Robert Olsson19baf832005-06-21 12:43:18 -07001244
Olof Johansson91b9a272005-08-09 20:24:39 -07001245 goto succeeded;
Robert Olsson19baf832005-06-21 12:43:18 -07001246 }
1247 /* Error if we find a perfect match which
1248 * uses the same scope, type, and nexthop
1249 * information.
1250 */
1251 fa_orig = fa;
1252 list_for_each_entry(fa, fa_orig->fa_list.prev, fa_list) {
1253 if (fa->fa_tos != tos)
1254 break;
1255 if (fa->fa_info->fib_priority != fi->fib_priority)
1256 break;
Thomas Graf4e902c52006-08-17 18:14:52 -07001257 if (fa->fa_type == cfg->fc_type &&
1258 fa->fa_scope == cfg->fc_scope &&
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001259 fa->fa_info == fi)
Robert Olsson19baf832005-06-21 12:43:18 -07001260 goto out;
Robert Olsson19baf832005-06-21 12:43:18 -07001261 }
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001262
Thomas Graf4e902c52006-08-17 18:14:52 -07001263 if (!(cfg->fc_nlflags & NLM_F_APPEND))
Robert Olsson19baf832005-06-21 12:43:18 -07001264 fa = fa_orig;
1265 }
1266 err = -ENOENT;
Thomas Graf4e902c52006-08-17 18:14:52 -07001267 if (!(cfg->fc_nlflags & NLM_F_CREATE))
Robert Olsson19baf832005-06-21 12:43:18 -07001268 goto out;
1269
1270 err = -ENOBUFS;
Christoph Lametere94b1762006-12-06 20:33:17 -08001271 new_fa = kmem_cache_alloc(fn_alias_kmem, GFP_KERNEL);
Robert Olsson19baf832005-06-21 12:43:18 -07001272 if (new_fa == NULL)
1273 goto out;
1274
1275 new_fa->fa_info = fi;
1276 new_fa->fa_tos = tos;
Thomas Graf4e902c52006-08-17 18:14:52 -07001277 new_fa->fa_type = cfg->fc_type;
1278 new_fa->fa_scope = cfg->fc_scope;
Robert Olsson19baf832005-06-21 12:43:18 -07001279 new_fa->fa_state = 0;
Robert Olsson19baf832005-06-21 12:43:18 -07001280 /*
1281 * Insert new entry to the list.
1282 */
1283
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001284 if (!fa_head) {
Stephen Hemmingerfea86ad2008-01-12 20:57:07 -08001285 fa_head = fib_insert_node(t, key, plen);
1286 if (unlikely(!fa_head)) {
1287 err = -ENOMEM;
Robert Olssonf835e472005-06-28 15:00:39 -07001288 goto out_free_new_fa;
Stephen Hemmingerfea86ad2008-01-12 20:57:07 -08001289 }
Robert Olssonf835e472005-06-28 15:00:39 -07001290 }
Robert Olsson19baf832005-06-21 12:43:18 -07001291
Robert Olsson2373ce12005-08-25 13:01:29 -07001292 list_add_tail_rcu(&new_fa->fa_list,
1293 (fa ? &fa->fa_list : fa_head));
Robert Olsson19baf832005-06-21 12:43:18 -07001294
1295 rt_cache_flush(-1);
Thomas Graf4e902c52006-08-17 18:14:52 -07001296 rtmsg_fib(RTM_NEWROUTE, htonl(key), new_fa, plen, tb->tb_id,
Milan Kocianb8f55832007-05-23 14:55:06 -07001297 &cfg->fc_nlinfo, 0);
Robert Olsson19baf832005-06-21 12:43:18 -07001298succeeded:
1299 return 0;
Robert Olssonf835e472005-06-28 15:00:39 -07001300
1301out_free_new_fa:
1302 kmem_cache_free(fn_alias_kmem, new_fa);
Robert Olsson19baf832005-06-21 12:43:18 -07001303out:
1304 fib_release_info(fi);
Olof Johansson91b9a272005-08-09 20:24:39 -07001305err:
Robert Olsson19baf832005-06-21 12:43:18 -07001306 return err;
1307}
1308
Robert Olsson2373ce12005-08-25 13:01:29 -07001309
Robert Olsson772cb712005-09-19 15:31:18 -07001310/* should be called with rcu_read_lock */
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001311static int check_leaf(struct trie *t, struct leaf *l,
1312 t_key key, const struct flowi *flp,
1313 struct fib_result *res)
Robert Olsson19baf832005-06-21 12:43:18 -07001314{
Robert Olsson19baf832005-06-21 12:43:18 -07001315 struct leaf_info *li;
1316 struct hlist_head *hhead = &l->list;
1317 struct hlist_node *node;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001318
Robert Olsson2373ce12005-08-25 13:01:29 -07001319 hlist_for_each_entry_rcu(li, node, hhead, hlist) {
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001320 int err;
1321 int plen = li->plen;
1322 __be32 mask = inet_make_mask(plen);
1323
Al Viro888454c2006-09-19 13:42:46 -07001324 if (l->key != (key & ntohl(mask)))
Robert Olsson19baf832005-06-21 12:43:18 -07001325 continue;
1326
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001327 err = fib_semantic_match(&li->falh, flp, res,
1328 htonl(l->key), mask, plen);
1329
Robert Olsson19baf832005-06-21 12:43:18 -07001330#ifdef CONFIG_IP_FIB_TRIE_STATS
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001331 if (err <= 0)
Robert Olsson19baf832005-06-21 12:43:18 -07001332 t->stats.semantic_match_passed++;
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001333 else
1334 t->stats.semantic_match_miss++;
Robert Olsson19baf832005-06-21 12:43:18 -07001335#endif
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001336 if (err <= 0)
1337 return plen;
Robert Olsson19baf832005-06-21 12:43:18 -07001338 }
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001339
1340 return -1;
Robert Olsson19baf832005-06-21 12:43:18 -07001341}
1342
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001343static int fn_trie_lookup(struct fib_table *tb, const struct flowi *flp,
1344 struct fib_result *res)
Robert Olsson19baf832005-06-21 12:43:18 -07001345{
1346 struct trie *t = (struct trie *) tb->tb_data;
1347 int plen, ret = 0;
1348 struct node *n;
1349 struct tnode *pn;
1350 int pos, bits;
Olof Johansson91b9a272005-08-09 20:24:39 -07001351 t_key key = ntohl(flp->fl4_dst);
Robert Olsson19baf832005-06-21 12:43:18 -07001352 int chopped_off;
1353 t_key cindex = 0;
1354 int current_prefix_length = KEYLENGTH;
Olof Johansson91b9a272005-08-09 20:24:39 -07001355 struct tnode *cn;
1356 t_key node_prefix, key_prefix, pref_mismatch;
1357 int mp;
1358
Robert Olsson2373ce12005-08-25 13:01:29 -07001359 rcu_read_lock();
Robert Olsson19baf832005-06-21 12:43:18 -07001360
Robert Olsson2373ce12005-08-25 13:01:29 -07001361 n = rcu_dereference(t->trie);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001362 if (!n)
Robert Olsson19baf832005-06-21 12:43:18 -07001363 goto failed;
1364
1365#ifdef CONFIG_IP_FIB_TRIE_STATS
1366 t->stats.gets++;
1367#endif
1368
1369 /* Just a leaf? */
1370 if (IS_LEAF(n)) {
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001371 plen = check_leaf(t, (struct leaf *)n, key, flp, res);
1372 if (plen < 0)
1373 goto failed;
1374 ret = 0;
1375 goto found;
Robert Olsson19baf832005-06-21 12:43:18 -07001376 }
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001377
Robert Olsson19baf832005-06-21 12:43:18 -07001378 pn = (struct tnode *) n;
1379 chopped_off = 0;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001380
Olof Johansson91b9a272005-08-09 20:24:39 -07001381 while (pn) {
Robert Olsson19baf832005-06-21 12:43:18 -07001382 pos = pn->pos;
1383 bits = pn->bits;
1384
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001385 if (!chopped_off)
Stephen Hemmingerab66b4a2007-08-10 15:22:58 -07001386 cindex = tkey_extract_bits(mask_pfx(key, current_prefix_length),
1387 pos, bits);
Robert Olsson19baf832005-06-21 12:43:18 -07001388
1389 n = tnode_get_child(pn, cindex);
1390
1391 if (n == NULL) {
1392#ifdef CONFIG_IP_FIB_TRIE_STATS
1393 t->stats.null_node_hit++;
1394#endif
1395 goto backtrace;
1396 }
1397
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001398 if (IS_LEAF(n)) {
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001399 plen = check_leaf(t, (struct leaf *)n, key, flp, res);
1400 if (plen < 0)
Olof Johansson91b9a272005-08-09 20:24:39 -07001401 goto backtrace;
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001402
1403 ret = 0;
1404 goto found;
Olof Johansson91b9a272005-08-09 20:24:39 -07001405 }
1406
Olof Johansson91b9a272005-08-09 20:24:39 -07001407 cn = (struct tnode *)n;
1408
1409 /*
1410 * It's a tnode, and we can do some extra checks here if we
1411 * like, to avoid descending into a dead-end branch.
1412 * This tnode is in the parent's child array at index
1413 * key[p_pos..p_pos+p_bits] but potentially with some bits
1414 * chopped off, so in reality the index may be just a
1415 * subprefix, padded with zero at the end.
1416 * We can also take a look at any skipped bits in this
1417 * tnode - everything up to p_pos is supposed to be ok,
1418 * and the non-chopped bits of the index (se previous
1419 * paragraph) are also guaranteed ok, but the rest is
1420 * considered unknown.
1421 *
1422 * The skipped bits are key[pos+bits..cn->pos].
1423 */
1424
1425 /* If current_prefix_length < pos+bits, we are already doing
1426 * actual prefix matching, which means everything from
1427 * pos+(bits-chopped_off) onward must be zero along some
1428 * branch of this subtree - otherwise there is *no* valid
1429 * prefix present. Here we can only check the skipped
1430 * bits. Remember, since we have already indexed into the
1431 * parent's child array, we know that the bits we chopped of
1432 * *are* zero.
1433 */
1434
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001435 /* NOTA BENE: Checking only skipped bits
1436 for the new node here */
Olof Johansson91b9a272005-08-09 20:24:39 -07001437
1438 if (current_prefix_length < pos+bits) {
1439 if (tkey_extract_bits(cn->key, current_prefix_length,
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001440 cn->pos - current_prefix_length)
1441 || !(cn->child[0]))
Olof Johansson91b9a272005-08-09 20:24:39 -07001442 goto backtrace;
1443 }
1444
1445 /*
1446 * If chopped_off=0, the index is fully validated and we
1447 * only need to look at the skipped bits for this, the new,
1448 * tnode. What we actually want to do is to find out if
1449 * these skipped bits match our key perfectly, or if we will
1450 * have to count on finding a matching prefix further down,
1451 * because if we do, we would like to have some way of
1452 * verifying the existence of such a prefix at this point.
1453 */
1454
1455 /* The only thing we can do at this point is to verify that
1456 * any such matching prefix can indeed be a prefix to our
1457 * key, and if the bits in the node we are inspecting that
1458 * do not match our key are not ZERO, this cannot be true.
1459 * Thus, find out where there is a mismatch (before cn->pos)
1460 * and verify that all the mismatching bits are zero in the
1461 * new tnode's key.
1462 */
1463
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001464 /*
1465 * Note: We aren't very concerned about the piece of
1466 * the key that precede pn->pos+pn->bits, since these
1467 * have already been checked. The bits after cn->pos
1468 * aren't checked since these are by definition
1469 * "unknown" at this point. Thus, what we want to see
1470 * is if we are about to enter the "prefix matching"
1471 * state, and in that case verify that the skipped
1472 * bits that will prevail throughout this subtree are
1473 * zero, as they have to be if we are to find a
1474 * matching prefix.
Olof Johansson91b9a272005-08-09 20:24:39 -07001475 */
1476
Stephen Hemmingerab66b4a2007-08-10 15:22:58 -07001477 node_prefix = mask_pfx(cn->key, cn->pos);
1478 key_prefix = mask_pfx(key, cn->pos);
Olof Johansson91b9a272005-08-09 20:24:39 -07001479 pref_mismatch = key_prefix^node_prefix;
1480 mp = 0;
1481
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001482 /*
1483 * In short: If skipped bits in this node do not match
1484 * the search key, enter the "prefix matching"
1485 * state.directly.
Olof Johansson91b9a272005-08-09 20:24:39 -07001486 */
1487 if (pref_mismatch) {
1488 while (!(pref_mismatch & (1<<(KEYLENGTH-1)))) {
1489 mp++;
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001490 pref_mismatch = pref_mismatch << 1;
Olof Johansson91b9a272005-08-09 20:24:39 -07001491 }
1492 key_prefix = tkey_extract_bits(cn->key, mp, cn->pos-mp);
1493
1494 if (key_prefix != 0)
1495 goto backtrace;
1496
1497 if (current_prefix_length >= cn->pos)
1498 current_prefix_length = mp;
1499 }
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001500
Olof Johansson91b9a272005-08-09 20:24:39 -07001501 pn = (struct tnode *)n; /* Descend */
1502 chopped_off = 0;
1503 continue;
1504
Robert Olsson19baf832005-06-21 12:43:18 -07001505backtrace:
1506 chopped_off++;
1507
1508 /* As zero don't change the child key (cindex) */
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001509 while ((chopped_off <= pn->bits)
1510 && !(cindex & (1<<(chopped_off-1))))
Robert Olsson19baf832005-06-21 12:43:18 -07001511 chopped_off++;
Robert Olsson19baf832005-06-21 12:43:18 -07001512
1513 /* Decrease current_... with bits chopped off */
1514 if (current_prefix_length > pn->pos + pn->bits - chopped_off)
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001515 current_prefix_length = pn->pos + pn->bits
1516 - chopped_off;
Olof Johansson91b9a272005-08-09 20:24:39 -07001517
Robert Olsson19baf832005-06-21 12:43:18 -07001518 /*
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001519 * Either we do the actual chop off according or if we have
Robert Olsson19baf832005-06-21 12:43:18 -07001520 * chopped off all bits in this tnode walk up to our parent.
1521 */
1522
Olof Johansson91b9a272005-08-09 20:24:39 -07001523 if (chopped_off <= pn->bits) {
Robert Olsson19baf832005-06-21 12:43:18 -07001524 cindex &= ~(1 << (chopped_off-1));
Olof Johansson91b9a272005-08-09 20:24:39 -07001525 } else {
Stephen Hemminger06801912007-08-10 15:22:13 -07001526 struct tnode *parent = node_parent((struct node *) pn);
1527 if (!parent)
Robert Olsson19baf832005-06-21 12:43:18 -07001528 goto failed;
Olof Johansson91b9a272005-08-09 20:24:39 -07001529
Robert Olsson19baf832005-06-21 12:43:18 -07001530 /* Get Child's index */
Stephen Hemminger06801912007-08-10 15:22:13 -07001531 cindex = tkey_extract_bits(pn->key, parent->pos, parent->bits);
1532 pn = parent;
Robert Olsson19baf832005-06-21 12:43:18 -07001533 chopped_off = 0;
1534
1535#ifdef CONFIG_IP_FIB_TRIE_STATS
1536 t->stats.backtrack++;
1537#endif
1538 goto backtrace;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001539 }
Robert Olsson19baf832005-06-21 12:43:18 -07001540 }
1541failed:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001542 ret = 1;
Robert Olsson19baf832005-06-21 12:43:18 -07001543found:
Robert Olsson2373ce12005-08-25 13:01:29 -07001544 rcu_read_unlock();
Robert Olsson19baf832005-06-21 12:43:18 -07001545 return ret;
1546}
1547
Robert Olsson2373ce12005-08-25 13:01:29 -07001548/* only called from updater side */
Robert Olsson19baf832005-06-21 12:43:18 -07001549static int trie_leaf_remove(struct trie *t, t_key key)
1550{
1551 t_key cindex;
1552 struct tnode *tp = NULL;
1553 struct node *n = t->trie;
1554 struct leaf *l;
1555
Stephen Hemminger0c7770c2005-08-23 21:59:41 -07001556 pr_debug("entering trie_leaf_remove(%p)\n", n);
Robert Olsson19baf832005-06-21 12:43:18 -07001557
1558 /* Note that in the case skipped bits, those bits are *not* checked!
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001559 * When we finish this, we will have NULL or a T_LEAF, and the
Robert Olsson19baf832005-06-21 12:43:18 -07001560 * T_LEAF may or may not match our key.
1561 */
1562
Olof Johansson91b9a272005-08-09 20:24:39 -07001563 while (n != NULL && IS_TNODE(n)) {
Robert Olsson19baf832005-06-21 12:43:18 -07001564 struct tnode *tn = (struct tnode *) n;
1565 check_tnode(tn);
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001566 n = tnode_get_child(tn, tkey_extract_bits(key,
1567 tn->pos, tn->bits));
Robert Olsson19baf832005-06-21 12:43:18 -07001568
Stephen Hemminger06801912007-08-10 15:22:13 -07001569 BUG_ON(n && node_parent(n) != tn);
Olof Johansson91b9a272005-08-09 20:24:39 -07001570 }
Robert Olsson19baf832005-06-21 12:43:18 -07001571 l = (struct leaf *) n;
1572
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001573 if (!n || !tkey_equals(l->key, key))
Robert Olsson19baf832005-06-21 12:43:18 -07001574 return 0;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001575
1576 /*
1577 * Key found.
1578 * Remove the leaf and rebalance the tree
Robert Olsson19baf832005-06-21 12:43:18 -07001579 */
Stephen Hemminger06801912007-08-10 15:22:13 -07001580 tp = node_parent(n);
Robert Olsson19baf832005-06-21 12:43:18 -07001581 tnode_free((struct tnode *) n);
1582
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001583 if (tp) {
Robert Olsson19baf832005-06-21 12:43:18 -07001584 cindex = tkey_extract_bits(key, tp->pos, tp->bits);
1585 put_child(t, (struct tnode *)tp, cindex, NULL);
Robert Olsson2373ce12005-08-25 13:01:29 -07001586 rcu_assign_pointer(t->trie, trie_rebalance(t, tp));
Olof Johansson91b9a272005-08-09 20:24:39 -07001587 } else
Robert Olsson2373ce12005-08-25 13:01:29 -07001588 rcu_assign_pointer(t->trie, NULL);
Robert Olsson19baf832005-06-21 12:43:18 -07001589
1590 return 1;
1591}
1592
Robert Olssond562f1f2007-03-26 14:22:22 -07001593/*
1594 * Caller must hold RTNL.
1595 */
Thomas Graf4e902c52006-08-17 18:14:52 -07001596static int fn_trie_delete(struct fib_table *tb, struct fib_config *cfg)
Robert Olsson19baf832005-06-21 12:43:18 -07001597{
1598 struct trie *t = (struct trie *) tb->tb_data;
1599 u32 key, mask;
Thomas Graf4e902c52006-08-17 18:14:52 -07001600 int plen = cfg->fc_dst_len;
1601 u8 tos = cfg->fc_tos;
Robert Olsson19baf832005-06-21 12:43:18 -07001602 struct fib_alias *fa, *fa_to_delete;
1603 struct list_head *fa_head;
1604 struct leaf *l;
Olof Johansson91b9a272005-08-09 20:24:39 -07001605 struct leaf_info *li;
1606
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001607 if (plen > 32)
Robert Olsson19baf832005-06-21 12:43:18 -07001608 return -EINVAL;
1609
Thomas Graf4e902c52006-08-17 18:14:52 -07001610 key = ntohl(cfg->fc_dst);
Olof Johansson91b9a272005-08-09 20:24:39 -07001611 mask = ntohl(inet_make_mask(plen));
Robert Olsson19baf832005-06-21 12:43:18 -07001612
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001613 if (key & ~mask)
Robert Olsson19baf832005-06-21 12:43:18 -07001614 return -EINVAL;
1615
1616 key = key & mask;
1617 l = fib_find_node(t, key);
1618
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001619 if (!l)
Robert Olsson19baf832005-06-21 12:43:18 -07001620 return -ESRCH;
1621
1622 fa_head = get_fa_head(l, plen);
1623 fa = fib_find_alias(fa_head, tos, 0);
1624
1625 if (!fa)
1626 return -ESRCH;
1627
Stephen Hemminger0c7770c2005-08-23 21:59:41 -07001628 pr_debug("Deleting %08x/%d tos=%d t=%p\n", key, plen, tos, t);
Robert Olsson19baf832005-06-21 12:43:18 -07001629
1630 fa_to_delete = NULL;
1631 fa_head = fa->fa_list.prev;
Robert Olsson2373ce12005-08-25 13:01:29 -07001632
Robert Olsson19baf832005-06-21 12:43:18 -07001633 list_for_each_entry(fa, fa_head, fa_list) {
1634 struct fib_info *fi = fa->fa_info;
1635
1636 if (fa->fa_tos != tos)
1637 break;
1638
Thomas Graf4e902c52006-08-17 18:14:52 -07001639 if ((!cfg->fc_type || fa->fa_type == cfg->fc_type) &&
1640 (cfg->fc_scope == RT_SCOPE_NOWHERE ||
1641 fa->fa_scope == cfg->fc_scope) &&
1642 (!cfg->fc_protocol ||
1643 fi->fib_protocol == cfg->fc_protocol) &&
1644 fib_nh_match(cfg, fi) == 0) {
Robert Olsson19baf832005-06-21 12:43:18 -07001645 fa_to_delete = fa;
1646 break;
1647 }
1648 }
1649
Olof Johansson91b9a272005-08-09 20:24:39 -07001650 if (!fa_to_delete)
1651 return -ESRCH;
Robert Olsson19baf832005-06-21 12:43:18 -07001652
Olof Johansson91b9a272005-08-09 20:24:39 -07001653 fa = fa_to_delete;
Thomas Graf4e902c52006-08-17 18:14:52 -07001654 rtmsg_fib(RTM_DELROUTE, htonl(key), fa, plen, tb->tb_id,
Milan Kocianb8f55832007-05-23 14:55:06 -07001655 &cfg->fc_nlinfo, 0);
Robert Olsson19baf832005-06-21 12:43:18 -07001656
Olof Johansson91b9a272005-08-09 20:24:39 -07001657 l = fib_find_node(t, key);
Robert Olsson772cb712005-09-19 15:31:18 -07001658 li = find_leaf_info(l, plen);
Robert Olsson19baf832005-06-21 12:43:18 -07001659
Robert Olsson2373ce12005-08-25 13:01:29 -07001660 list_del_rcu(&fa->fa_list);
Robert Olsson19baf832005-06-21 12:43:18 -07001661
Olof Johansson91b9a272005-08-09 20:24:39 -07001662 if (list_empty(fa_head)) {
Robert Olsson2373ce12005-08-25 13:01:29 -07001663 hlist_del_rcu(&li->hlist);
Olof Johansson91b9a272005-08-09 20:24:39 -07001664 free_leaf_info(li);
Robert Olsson2373ce12005-08-25 13:01:29 -07001665 }
Olof Johansson91b9a272005-08-09 20:24:39 -07001666
1667 if (hlist_empty(&l->list))
1668 trie_leaf_remove(t, key);
1669
1670 if (fa->fa_state & FA_S_ACCESSED)
1671 rt_cache_flush(-1);
1672
Robert Olsson2373ce12005-08-25 13:01:29 -07001673 fib_release_info(fa->fa_info);
1674 alias_free_mem_rcu(fa);
Olof Johansson91b9a272005-08-09 20:24:39 -07001675 return 0;
Robert Olsson19baf832005-06-21 12:43:18 -07001676}
1677
1678static int trie_flush_list(struct trie *t, struct list_head *head)
1679{
1680 struct fib_alias *fa, *fa_node;
1681 int found = 0;
1682
1683 list_for_each_entry_safe(fa, fa_node, head, fa_list) {
1684 struct fib_info *fi = fa->fa_info;
Robert Olsson19baf832005-06-21 12:43:18 -07001685
Robert Olsson2373ce12005-08-25 13:01:29 -07001686 if (fi && (fi->fib_flags & RTNH_F_DEAD)) {
1687 list_del_rcu(&fa->fa_list);
1688 fib_release_info(fa->fa_info);
1689 alias_free_mem_rcu(fa);
Robert Olsson19baf832005-06-21 12:43:18 -07001690 found++;
1691 }
1692 }
1693 return found;
1694}
1695
1696static int trie_flush_leaf(struct trie *t, struct leaf *l)
1697{
1698 int found = 0;
1699 struct hlist_head *lih = &l->list;
1700 struct hlist_node *node, *tmp;
1701 struct leaf_info *li = NULL;
1702
1703 hlist_for_each_entry_safe(li, node, tmp, lih, hlist) {
Robert Olsson19baf832005-06-21 12:43:18 -07001704 found += trie_flush_list(t, &li->falh);
1705
1706 if (list_empty(&li->falh)) {
Robert Olsson2373ce12005-08-25 13:01:29 -07001707 hlist_del_rcu(&li->hlist);
Robert Olsson19baf832005-06-21 12:43:18 -07001708 free_leaf_info(li);
1709 }
1710 }
1711 return found;
1712}
1713
Robert Olsson2373ce12005-08-25 13:01:29 -07001714/* rcu_read_lock needs to be hold by caller from readside */
1715
Robert Olsson19baf832005-06-21 12:43:18 -07001716static struct leaf *nextleaf(struct trie *t, struct leaf *thisleaf)
1717{
1718 struct node *c = (struct node *) thisleaf;
1719 struct tnode *p;
1720 int idx;
Robert Olsson2373ce12005-08-25 13:01:29 -07001721 struct node *trie = rcu_dereference(t->trie);
Robert Olsson19baf832005-06-21 12:43:18 -07001722
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001723 if (c == NULL) {
Robert Olsson2373ce12005-08-25 13:01:29 -07001724 if (trie == NULL)
Robert Olsson19baf832005-06-21 12:43:18 -07001725 return NULL;
1726
Robert Olsson2373ce12005-08-25 13:01:29 -07001727 if (IS_LEAF(trie)) /* trie w. just a leaf */
1728 return (struct leaf *) trie;
Robert Olsson19baf832005-06-21 12:43:18 -07001729
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001730 p = (struct tnode *)trie; /* Start */
Olof Johansson91b9a272005-08-09 20:24:39 -07001731 } else
Eric Dumazetb59cfbf2008-01-18 03:31:36 -08001732 p = node_parent_rcu(c);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001733
Robert Olsson19baf832005-06-21 12:43:18 -07001734 while (p) {
1735 int pos, last;
1736
1737 /* Find the next child of the parent */
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001738 if (c)
1739 pos = 1 + tkey_extract_bits(c->key, p->pos, p->bits);
1740 else
Robert Olsson19baf832005-06-21 12:43:18 -07001741 pos = 0;
1742
1743 last = 1 << p->bits;
Olof Johansson91b9a272005-08-09 20:24:39 -07001744 for (idx = pos; idx < last ; idx++) {
Robert Olsson2373ce12005-08-25 13:01:29 -07001745 c = rcu_dereference(p->child[idx]);
1746
1747 if (!c)
Olof Johansson91b9a272005-08-09 20:24:39 -07001748 continue;
Robert Olsson19baf832005-06-21 12:43:18 -07001749
Olof Johansson91b9a272005-08-09 20:24:39 -07001750 /* Decend if tnode */
Robert Olsson2373ce12005-08-25 13:01:29 -07001751 while (IS_TNODE(c)) {
1752 p = (struct tnode *) c;
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +09001753 idx = 0;
Robert Olsson19baf832005-06-21 12:43:18 -07001754
Olof Johansson91b9a272005-08-09 20:24:39 -07001755 /* Rightmost non-NULL branch */
1756 if (p && IS_TNODE(p))
Robert Olsson2373ce12005-08-25 13:01:29 -07001757 while (!(c = rcu_dereference(p->child[idx]))
1758 && idx < (1<<p->bits)) idx++;
Robert Olsson19baf832005-06-21 12:43:18 -07001759
Olof Johansson91b9a272005-08-09 20:24:39 -07001760 /* Done with this tnode? */
Robert Olsson2373ce12005-08-25 13:01:29 -07001761 if (idx >= (1 << p->bits) || !c)
Olof Johansson91b9a272005-08-09 20:24:39 -07001762 goto up;
Robert Olsson19baf832005-06-21 12:43:18 -07001763 }
Robert Olsson2373ce12005-08-25 13:01:29 -07001764 return (struct leaf *) c;
Robert Olsson19baf832005-06-21 12:43:18 -07001765 }
1766up:
1767 /* No more children go up one step */
Olof Johansson91b9a272005-08-09 20:24:39 -07001768 c = (struct node *) p;
Eric Dumazetb59cfbf2008-01-18 03:31:36 -08001769 p = node_parent_rcu(c);
Robert Olsson19baf832005-06-21 12:43:18 -07001770 }
1771 return NULL; /* Ready. Root of trie */
1772}
1773
Robert Olssond562f1f2007-03-26 14:22:22 -07001774/*
1775 * Caller must hold RTNL.
1776 */
Robert Olsson19baf832005-06-21 12:43:18 -07001777static int fn_trie_flush(struct fib_table *tb)
1778{
1779 struct trie *t = (struct trie *) tb->tb_data;
1780 struct leaf *ll = NULL, *l = NULL;
1781 int found = 0, h;
1782
Olof Johansson91b9a272005-08-09 20:24:39 -07001783 for (h = 0; (l = nextleaf(t, l)) != NULL; h++) {
Robert Olsson19baf832005-06-21 12:43:18 -07001784 found += trie_flush_leaf(t, l);
1785
1786 if (ll && hlist_empty(&ll->list))
1787 trie_leaf_remove(t, ll->key);
1788 ll = l;
1789 }
1790
1791 if (ll && hlist_empty(&ll->list))
1792 trie_leaf_remove(t, ll->key);
1793
Stephen Hemminger0c7770c2005-08-23 21:59:41 -07001794 pr_debug("trie_flush found=%d\n", found);
Robert Olsson19baf832005-06-21 12:43:18 -07001795 return found;
1796}
1797
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001798static void fn_trie_select_default(struct fib_table *tb,
1799 const struct flowi *flp,
1800 struct fib_result *res)
Robert Olsson19baf832005-06-21 12:43:18 -07001801{
1802 struct trie *t = (struct trie *) tb->tb_data;
1803 int order, last_idx;
1804 struct fib_info *fi = NULL;
1805 struct fib_info *last_resort;
1806 struct fib_alias *fa = NULL;
1807 struct list_head *fa_head;
1808 struct leaf *l;
1809
1810 last_idx = -1;
1811 last_resort = NULL;
1812 order = -1;
1813
Robert Olsson2373ce12005-08-25 13:01:29 -07001814 rcu_read_lock();
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001815
Robert Olsson19baf832005-06-21 12:43:18 -07001816 l = fib_find_node(t, 0);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001817 if (!l)
Robert Olsson19baf832005-06-21 12:43:18 -07001818 goto out;
1819
1820 fa_head = get_fa_head(l, 0);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001821 if (!fa_head)
Robert Olsson19baf832005-06-21 12:43:18 -07001822 goto out;
1823
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001824 if (list_empty(fa_head))
Robert Olsson19baf832005-06-21 12:43:18 -07001825 goto out;
1826
Robert Olsson2373ce12005-08-25 13:01:29 -07001827 list_for_each_entry_rcu(fa, fa_head, fa_list) {
Robert Olsson19baf832005-06-21 12:43:18 -07001828 struct fib_info *next_fi = fa->fa_info;
Olof Johansson91b9a272005-08-09 20:24:39 -07001829
Robert Olsson19baf832005-06-21 12:43:18 -07001830 if (fa->fa_scope != res->scope ||
1831 fa->fa_type != RTN_UNICAST)
1832 continue;
Olof Johansson91b9a272005-08-09 20:24:39 -07001833
Robert Olsson19baf832005-06-21 12:43:18 -07001834 if (next_fi->fib_priority > res->fi->fib_priority)
1835 break;
1836 if (!next_fi->fib_nh[0].nh_gw ||
1837 next_fi->fib_nh[0].nh_scope != RT_SCOPE_LINK)
1838 continue;
1839 fa->fa_state |= FA_S_ACCESSED;
Olof Johansson91b9a272005-08-09 20:24:39 -07001840
Robert Olsson19baf832005-06-21 12:43:18 -07001841 if (fi == NULL) {
1842 if (next_fi != res->fi)
1843 break;
1844 } else if (!fib_detect_death(fi, order, &last_resort,
Denis V. Lunev971b8932007-12-08 00:32:23 -08001845 &last_idx, tb->tb_default)) {
Denis V. Luneva2bbe682007-12-08 00:31:44 -08001846 fib_result_assign(res, fi);
Denis V. Lunev971b8932007-12-08 00:32:23 -08001847 tb->tb_default = order;
Robert Olsson19baf832005-06-21 12:43:18 -07001848 goto out;
1849 }
1850 fi = next_fi;
1851 order++;
1852 }
1853 if (order <= 0 || fi == NULL) {
Denis V. Lunev971b8932007-12-08 00:32:23 -08001854 tb->tb_default = -1;
Robert Olsson19baf832005-06-21 12:43:18 -07001855 goto out;
1856 }
1857
Denis V. Lunev971b8932007-12-08 00:32:23 -08001858 if (!fib_detect_death(fi, order, &last_resort, &last_idx,
1859 tb->tb_default)) {
Denis V. Luneva2bbe682007-12-08 00:31:44 -08001860 fib_result_assign(res, fi);
Denis V. Lunev971b8932007-12-08 00:32:23 -08001861 tb->tb_default = order;
Robert Olsson19baf832005-06-21 12:43:18 -07001862 goto out;
1863 }
Denis V. Luneva2bbe682007-12-08 00:31:44 -08001864 if (last_idx >= 0)
1865 fib_result_assign(res, last_resort);
Denis V. Lunev971b8932007-12-08 00:32:23 -08001866 tb->tb_default = last_idx;
1867out:
Robert Olsson2373ce12005-08-25 13:01:29 -07001868 rcu_read_unlock();
Robert Olsson19baf832005-06-21 12:43:18 -07001869}
1870
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001871static int fn_trie_dump_fa(t_key key, int plen, struct list_head *fah,
1872 struct fib_table *tb,
Robert Olsson19baf832005-06-21 12:43:18 -07001873 struct sk_buff *skb, struct netlink_callback *cb)
1874{
1875 int i, s_i;
1876 struct fib_alias *fa;
1877
Al Viro32ab5f82006-09-26 22:21:45 -07001878 __be32 xkey = htonl(key);
Robert Olsson19baf832005-06-21 12:43:18 -07001879
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001880 s_i = cb->args[4];
Robert Olsson19baf832005-06-21 12:43:18 -07001881 i = 0;
1882
Robert Olsson2373ce12005-08-25 13:01:29 -07001883 /* rcu_read_lock is hold by caller */
1884
1885 list_for_each_entry_rcu(fa, fah, fa_list) {
Robert Olsson19baf832005-06-21 12:43:18 -07001886 if (i < s_i) {
1887 i++;
1888 continue;
1889 }
Stephen Hemminger78c66712005-09-21 00:15:39 -07001890 BUG_ON(!fa->fa_info);
Robert Olsson19baf832005-06-21 12:43:18 -07001891
1892 if (fib_dump_info(skb, NETLINK_CB(cb->skb).pid,
1893 cb->nlh->nlmsg_seq,
1894 RTM_NEWROUTE,
1895 tb->tb_id,
1896 fa->fa_type,
1897 fa->fa_scope,
Thomas Grafbe403ea2006-08-17 18:15:17 -07001898 xkey,
Robert Olsson19baf832005-06-21 12:43:18 -07001899 plen,
1900 fa->fa_tos,
David S. Miller90f66912005-06-21 14:43:28 -07001901 fa->fa_info, 0) < 0) {
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001902 cb->args[4] = i;
Robert Olsson19baf832005-06-21 12:43:18 -07001903 return -1;
Olof Johansson91b9a272005-08-09 20:24:39 -07001904 }
Robert Olsson19baf832005-06-21 12:43:18 -07001905 i++;
1906 }
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001907 cb->args[4] = i;
Robert Olsson19baf832005-06-21 12:43:18 -07001908 return skb->len;
1909}
1910
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001911static int fn_trie_dump_plen(struct trie *t, int plen, struct fib_table *tb,
1912 struct sk_buff *skb, struct netlink_callback *cb)
Robert Olsson19baf832005-06-21 12:43:18 -07001913{
1914 int h, s_h;
1915 struct list_head *fa_head;
1916 struct leaf *l = NULL;
Robert Olsson19baf832005-06-21 12:43:18 -07001917
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001918 s_h = cb->args[3];
Robert Olsson19baf832005-06-21 12:43:18 -07001919
Olof Johansson91b9a272005-08-09 20:24:39 -07001920 for (h = 0; (l = nextleaf(t, l)) != NULL; h++) {
Robert Olsson19baf832005-06-21 12:43:18 -07001921 if (h < s_h)
1922 continue;
1923 if (h > s_h)
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001924 memset(&cb->args[4], 0,
1925 sizeof(cb->args) - 4*sizeof(cb->args[0]));
Robert Olsson19baf832005-06-21 12:43:18 -07001926
1927 fa_head = get_fa_head(l, plen);
Olof Johansson91b9a272005-08-09 20:24:39 -07001928
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001929 if (!fa_head)
Robert Olsson19baf832005-06-21 12:43:18 -07001930 continue;
1931
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001932 if (list_empty(fa_head))
Robert Olsson19baf832005-06-21 12:43:18 -07001933 continue;
1934
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001935 if (fn_trie_dump_fa(l->key, plen, fa_head, tb, skb, cb) < 0) {
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001936 cb->args[3] = h;
Robert Olsson19baf832005-06-21 12:43:18 -07001937 return -1;
1938 }
1939 }
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001940 cb->args[3] = h;
Robert Olsson19baf832005-06-21 12:43:18 -07001941 return skb->len;
1942}
1943
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001944static int fn_trie_dump(struct fib_table *tb, struct sk_buff *skb,
1945 struct netlink_callback *cb)
Robert Olsson19baf832005-06-21 12:43:18 -07001946{
1947 int m, s_m;
1948 struct trie *t = (struct trie *) tb->tb_data;
1949
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001950 s_m = cb->args[2];
Robert Olsson19baf832005-06-21 12:43:18 -07001951
Robert Olsson2373ce12005-08-25 13:01:29 -07001952 rcu_read_lock();
Olof Johansson91b9a272005-08-09 20:24:39 -07001953 for (m = 0; m <= 32; m++) {
Robert Olsson19baf832005-06-21 12:43:18 -07001954 if (m < s_m)
1955 continue;
1956 if (m > s_m)
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001957 memset(&cb->args[3], 0,
1958 sizeof(cb->args) - 3*sizeof(cb->args[0]));
Robert Olsson19baf832005-06-21 12:43:18 -07001959
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001960 if (fn_trie_dump_plen(t, 32-m, tb, skb, cb) < 0) {
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001961 cb->args[2] = m;
Robert Olsson19baf832005-06-21 12:43:18 -07001962 goto out;
1963 }
1964 }
Robert Olsson2373ce12005-08-25 13:01:29 -07001965 rcu_read_unlock();
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001966 cb->args[2] = m;
Robert Olsson19baf832005-06-21 12:43:18 -07001967 return skb->len;
Olof Johansson91b9a272005-08-09 20:24:39 -07001968out:
Robert Olsson2373ce12005-08-25 13:01:29 -07001969 rcu_read_unlock();
Robert Olsson19baf832005-06-21 12:43:18 -07001970 return -1;
1971}
1972
Stephen Hemminger7f9b8052008-01-14 23:14:20 -08001973void __init fib_hash_init(void)
1974{
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001975 fn_alias_kmem = kmem_cache_create("ip_fib_alias",
1976 sizeof(struct fib_alias),
Stephen Hemmingerbc3c8c12008-01-22 21:51:50 -08001977 0, SLAB_PANIC, NULL);
1978
1979 trie_leaf_kmem = kmem_cache_create("ip_fib_trie",
1980 max(sizeof(struct leaf),
1981 sizeof(struct leaf_info)),
1982 0, SLAB_PANIC, NULL);
Stephen Hemminger7f9b8052008-01-14 23:14:20 -08001983}
Robert Olsson19baf832005-06-21 12:43:18 -07001984
Stephen Hemminger7f9b8052008-01-14 23:14:20 -08001985
1986/* Fix more generic FIB names for init later */
1987struct fib_table *fib_hash_table(u32 id)
Robert Olsson19baf832005-06-21 12:43:18 -07001988{
1989 struct fib_table *tb;
1990 struct trie *t;
1991
Robert Olsson19baf832005-06-21 12:43:18 -07001992 tb = kmalloc(sizeof(struct fib_table) + sizeof(struct trie),
1993 GFP_KERNEL);
1994 if (tb == NULL)
1995 return NULL;
1996
1997 tb->tb_id = id;
Denis V. Lunev971b8932007-12-08 00:32:23 -08001998 tb->tb_default = -1;
Robert Olsson19baf832005-06-21 12:43:18 -07001999 tb->tb_lookup = fn_trie_lookup;
2000 tb->tb_insert = fn_trie_insert;
2001 tb->tb_delete = fn_trie_delete;
2002 tb->tb_flush = fn_trie_flush;
2003 tb->tb_select_default = fn_trie_select_default;
2004 tb->tb_dump = fn_trie_dump;
Robert Olsson19baf832005-06-21 12:43:18 -07002005
2006 t = (struct trie *) tb->tb_data;
Stephen Hemmingerc28a1cf2008-01-12 20:49:13 -08002007 memset(t, 0, sizeof(*t));
Robert Olsson19baf832005-06-21 12:43:18 -07002008
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07002009 if (id == RT_TABLE_LOCAL)
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08002010 pr_info("IPv4 FIB: Using LC-trie version %s\n", VERSION);
Robert Olsson19baf832005-06-21 12:43:18 -07002011
2012 return tb;
2013}
2014
Robert Olsson19baf832005-06-21 12:43:18 -07002015#ifdef CONFIG_PROC_FS
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002016/* Depth first Trie walk iterator */
2017struct fib_trie_iter {
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002018 struct seq_net_private p;
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002019 struct trie *trie_local, *trie_main;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002020 struct tnode *tnode;
2021 struct trie *trie;
2022 unsigned index;
2023 unsigned depth;
2024};
Robert Olsson19baf832005-06-21 12:43:18 -07002025
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002026static struct node *fib_trie_get_next(struct fib_trie_iter *iter)
Robert Olsson19baf832005-06-21 12:43:18 -07002027{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002028 struct tnode *tn = iter->tnode;
2029 unsigned cindex = iter->index;
2030 struct tnode *p;
2031
Eric W. Biederman6640e692007-01-24 14:42:04 -08002032 /* A single entry routing table */
2033 if (!tn)
2034 return NULL;
2035
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002036 pr_debug("get_next iter={node=%p index=%d depth=%d}\n",
2037 iter->tnode, iter->index, iter->depth);
2038rescan:
2039 while (cindex < (1<<tn->bits)) {
Eric Dumazetb59cfbf2008-01-18 03:31:36 -08002040 struct node *n = tnode_get_child_rcu(tn, cindex);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002041
2042 if (n) {
2043 if (IS_LEAF(n)) {
2044 iter->tnode = tn;
2045 iter->index = cindex + 1;
2046 } else {
2047 /* push down one level */
2048 iter->tnode = (struct tnode *) n;
2049 iter->index = 0;
2050 ++iter->depth;
2051 }
2052 return n;
2053 }
2054
2055 ++cindex;
2056 }
2057
2058 /* Current node exhausted, pop back up */
Eric Dumazetb59cfbf2008-01-18 03:31:36 -08002059 p = node_parent_rcu((struct node *)tn);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002060 if (p) {
2061 cindex = tkey_extract_bits(tn->key, p->pos, p->bits)+1;
2062 tn = p;
2063 --iter->depth;
2064 goto rescan;
2065 }
2066
2067 /* got root? */
Robert Olsson19baf832005-06-21 12:43:18 -07002068 return NULL;
2069}
2070
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002071static struct node *fib_trie_get_first(struct fib_trie_iter *iter,
2072 struct trie *t)
Robert Olsson19baf832005-06-21 12:43:18 -07002073{
Robert Olsson5ddf0eb2006-03-20 21:34:12 -08002074 struct node *n ;
2075
Stephen Hemminger132adf52007-03-08 20:44:43 -08002076 if (!t)
Robert Olsson5ddf0eb2006-03-20 21:34:12 -08002077 return NULL;
2078
2079 n = rcu_dereference(t->trie);
2080
Stephen Hemminger132adf52007-03-08 20:44:43 -08002081 if (!iter)
Robert Olsson5ddf0eb2006-03-20 21:34:12 -08002082 return NULL;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002083
Eric W. Biederman6640e692007-01-24 14:42:04 -08002084 if (n) {
2085 if (IS_TNODE(n)) {
2086 iter->tnode = (struct tnode *) n;
2087 iter->trie = t;
2088 iter->index = 0;
2089 iter->depth = 1;
2090 } else {
2091 iter->tnode = NULL;
2092 iter->trie = t;
2093 iter->index = 0;
2094 iter->depth = 0;
2095 }
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002096 return n;
2097 }
Robert Olsson19baf832005-06-21 12:43:18 -07002098 return NULL;
2099}
2100
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002101static void trie_collect_stats(struct trie *t, struct trie_stat *s)
Robert Olsson19baf832005-06-21 12:43:18 -07002102{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002103 struct node *n;
2104 struct fib_trie_iter iter;
Robert Olsson19baf832005-06-21 12:43:18 -07002105
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002106 memset(s, 0, sizeof(*s));
Robert Olsson19baf832005-06-21 12:43:18 -07002107
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002108 rcu_read_lock();
2109 for (n = fib_trie_get_first(&iter, t); n;
2110 n = fib_trie_get_next(&iter)) {
2111 if (IS_LEAF(n)) {
Stephen Hemminger93672292008-01-22 21:54:05 -08002112 struct leaf *l = (struct leaf *)n;
2113 struct leaf_info *li;
2114 struct hlist_node *tmp;
2115
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002116 s->leaves++;
2117 s->totdepth += iter.depth;
2118 if (iter.depth > s->maxdepth)
2119 s->maxdepth = iter.depth;
Stephen Hemminger93672292008-01-22 21:54:05 -08002120
2121 hlist_for_each_entry_rcu(li, tmp, &l->list, hlist)
2122 ++s->prefixes;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002123 } else {
2124 const struct tnode *tn = (const struct tnode *) n;
2125 int i;
Robert Olsson19baf832005-06-21 12:43:18 -07002126
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002127 s->tnodes++;
Stephen Hemminger132adf52007-03-08 20:44:43 -08002128 if (tn->bits < MAX_STAT_DEPTH)
Robert Olsson06ef9212006-03-20 21:35:01 -08002129 s->nodesizes[tn->bits]++;
2130
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002131 for (i = 0; i < (1<<tn->bits); i++)
2132 if (!tn->child[i])
2133 s->nullpointers++;
2134 }
2135 }
2136 rcu_read_unlock();
Robert Olsson19baf832005-06-21 12:43:18 -07002137}
2138
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07002139/*
Robert Olsson19baf832005-06-21 12:43:18 -07002140 * This outputs /proc/net/fib_triestats
Robert Olsson19baf832005-06-21 12:43:18 -07002141 */
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002142static void trie_show_stats(struct seq_file *seq, struct trie_stat *stat)
Robert Olsson19baf832005-06-21 12:43:18 -07002143{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002144 unsigned i, max, pointers, bytes, avdepth;
Robert Olsson19baf832005-06-21 12:43:18 -07002145
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002146 if (stat->leaves)
2147 avdepth = stat->totdepth*100 / stat->leaves;
2148 else
2149 avdepth = 0;
Robert Olsson19baf832005-06-21 12:43:18 -07002150
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08002151 seq_printf(seq, "\tAver depth: %u.%02d\n",
2152 avdepth / 100, avdepth % 100);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002153 seq_printf(seq, "\tMax depth: %u\n", stat->maxdepth);
Robert Olsson19baf832005-06-21 12:43:18 -07002154
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002155 seq_printf(seq, "\tLeaves: %u\n", stat->leaves);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002156 bytes = sizeof(struct leaf) * stat->leaves;
Stephen Hemminger93672292008-01-22 21:54:05 -08002157
2158 seq_printf(seq, "\tPrefixes: %u\n", stat->prefixes);
2159 bytes += sizeof(struct leaf_info) * stat->prefixes;
2160
Stephen Hemminger187b5182008-01-12 20:55:55 -08002161 seq_printf(seq, "\tInternal nodes: %u\n\t", stat->tnodes);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002162 bytes += sizeof(struct tnode) * stat->tnodes;
Robert Olsson19baf832005-06-21 12:43:18 -07002163
Robert Olsson06ef9212006-03-20 21:35:01 -08002164 max = MAX_STAT_DEPTH;
2165 while (max > 0 && stat->nodesizes[max-1] == 0)
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002166 max--;
Robert Olsson19baf832005-06-21 12:43:18 -07002167
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002168 pointers = 0;
2169 for (i = 1; i <= max; i++)
2170 if (stat->nodesizes[i] != 0) {
Stephen Hemminger187b5182008-01-12 20:55:55 -08002171 seq_printf(seq, " %u: %u", i, stat->nodesizes[i]);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002172 pointers += (1<<i) * stat->nodesizes[i];
2173 }
2174 seq_putc(seq, '\n');
Stephen Hemminger187b5182008-01-12 20:55:55 -08002175 seq_printf(seq, "\tPointers: %u\n", pointers);
Robert Olsson19baf832005-06-21 12:43:18 -07002176
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002177 bytes += sizeof(struct node *) * pointers;
Stephen Hemminger187b5182008-01-12 20:55:55 -08002178 seq_printf(seq, "Null ptrs: %u\n", stat->nullpointers);
2179 seq_printf(seq, "Total size: %u kB\n", (bytes + 1023) / 1024);
Stephen Hemminger66a2f7f2008-01-12 21:23:17 -08002180}
Robert Olsson19baf832005-06-21 12:43:18 -07002181
2182#ifdef CONFIG_IP_FIB_TRIE_STATS
Stephen Hemminger66a2f7f2008-01-12 21:23:17 -08002183static void trie_show_usage(struct seq_file *seq,
2184 const struct trie_use_stats *stats)
2185{
2186 seq_printf(seq, "\nCounters:\n---------\n");
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08002187 seq_printf(seq, "gets = %u\n", stats->gets);
2188 seq_printf(seq, "backtracks = %u\n", stats->backtrack);
2189 seq_printf(seq, "semantic match passed = %u\n",
2190 stats->semantic_match_passed);
2191 seq_printf(seq, "semantic match miss = %u\n",
2192 stats->semantic_match_miss);
2193 seq_printf(seq, "null node hit= %u\n", stats->null_node_hit);
2194 seq_printf(seq, "skipped node resize = %u\n\n",
2195 stats->resize_node_skipped);
Robert Olsson19baf832005-06-21 12:43:18 -07002196}
Stephen Hemminger66a2f7f2008-01-12 21:23:17 -08002197#endif /* CONFIG_IP_FIB_TRIE_STATS */
2198
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08002199static void fib_trie_show(struct seq_file *seq, const char *name,
2200 struct trie *trie)
Stephen Hemmingerd717a9a2008-01-14 23:11:54 -08002201{
2202 struct trie_stat stat;
2203
Stephen Hemmingerd717a9a2008-01-14 23:11:54 -08002204 trie_collect_stats(trie, &stat);
Stephen Hemminger93672292008-01-22 21:54:05 -08002205 seq_printf(seq, "%s:\n", name);
Stephen Hemmingerd717a9a2008-01-14 23:11:54 -08002206 trie_show_stats(seq, &stat);
2207#ifdef CONFIG_IP_FIB_TRIE_STATS
2208 trie_show_usage(seq, &trie->stats);
2209#endif
2210}
Robert Olsson19baf832005-06-21 12:43:18 -07002211
2212static int fib_triestat_seq_show(struct seq_file *seq, void *v)
2213{
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002214 struct net *net = (struct net *)seq->private;
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002215 struct fib_table *tb;
2216
Stephen Hemmingerd717a9a2008-01-14 23:11:54 -08002217 seq_printf(seq,
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08002218 "Basic info: size of leaf:"
2219 " %Zd bytes, size of tnode: %Zd bytes.\n",
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002220 sizeof(struct leaf), sizeof(struct tnode));
Olof Johansson91b9a272005-08-09 20:24:39 -07002221
Stephen Hemmingerd717a9a2008-01-14 23:11:54 -08002222 tb = fib_get_table(net, RT_TABLE_LOCAL);
2223 if (tb)
2224 fib_trie_show(seq, "Local", (struct trie *) tb->tb_data);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002225
Stephen Hemmingerd717a9a2008-01-14 23:11:54 -08002226 tb = fib_get_table(net, RT_TABLE_MAIN);
2227 if (tb)
2228 fib_trie_show(seq, "Main", (struct trie *) tb->tb_data);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002229
Robert Olsson19baf832005-06-21 12:43:18 -07002230 return 0;
2231}
2232
Robert Olsson19baf832005-06-21 12:43:18 -07002233static int fib_triestat_seq_open(struct inode *inode, struct file *file)
2234{
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002235 int err;
2236 struct net *net;
2237
2238 net = get_proc_net(inode);
2239 if (net == NULL)
2240 return -ENXIO;
2241 err = single_open(file, fib_triestat_seq_show, net);
2242 if (err < 0) {
2243 put_net(net);
2244 return err;
2245 }
2246 return 0;
2247}
2248
2249static int fib_triestat_seq_release(struct inode *ino, struct file *f)
2250{
2251 struct seq_file *seq = f->private_data;
2252 put_net(seq->private);
2253 return single_release(ino, f);
Robert Olsson19baf832005-06-21 12:43:18 -07002254}
2255
Arjan van de Ven9a321442007-02-12 00:55:35 -08002256static const struct file_operations fib_triestat_fops = {
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07002257 .owner = THIS_MODULE,
2258 .open = fib_triestat_seq_open,
2259 .read = seq_read,
2260 .llseek = seq_lseek,
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002261 .release = fib_triestat_seq_release,
Robert Olsson19baf832005-06-21 12:43:18 -07002262};
2263
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002264static struct node *fib_trie_get_idx(struct fib_trie_iter *iter,
2265 loff_t pos)
Robert Olsson19baf832005-06-21 12:43:18 -07002266{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002267 loff_t idx = 0;
2268 struct node *n;
Robert Olsson19baf832005-06-21 12:43:18 -07002269
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002270 for (n = fib_trie_get_first(iter, iter->trie_local);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002271 n; ++idx, n = fib_trie_get_next(iter)) {
2272 if (pos == idx)
2273 return n;
2274 }
Robert Olsson19baf832005-06-21 12:43:18 -07002275
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002276 for (n = fib_trie_get_first(iter, iter->trie_main);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002277 n; ++idx, n = fib_trie_get_next(iter)) {
2278 if (pos == idx)
2279 return n;
2280 }
Robert Olsson19baf832005-06-21 12:43:18 -07002281 return NULL;
2282}
2283
2284static void *fib_trie_seq_start(struct seq_file *seq, loff_t *pos)
Stephen Hemmingerc95aaf92008-01-12 21:25:02 -08002285 __acquires(RCU)
Robert Olsson19baf832005-06-21 12:43:18 -07002286{
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002287 struct fib_trie_iter *iter = seq->private;
2288 struct fib_table *tb;
2289
2290 if (!iter->trie_local) {
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002291 tb = fib_get_table(iter->p.net, RT_TABLE_LOCAL);
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002292 if (tb)
2293 iter->trie_local = (struct trie *) tb->tb_data;
2294 }
2295 if (!iter->trie_main) {
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002296 tb = fib_get_table(iter->p.net, RT_TABLE_MAIN);
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002297 if (tb)
2298 iter->trie_main = (struct trie *) tb->tb_data;
2299 }
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002300 rcu_read_lock();
2301 if (*pos == 0)
Olof Johansson91b9a272005-08-09 20:24:39 -07002302 return SEQ_START_TOKEN;
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002303 return fib_trie_get_idx(iter, *pos - 1);
Robert Olsson19baf832005-06-21 12:43:18 -07002304}
2305
2306static void *fib_trie_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2307{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002308 struct fib_trie_iter *iter = seq->private;
2309 void *l = v;
2310
Robert Olsson19baf832005-06-21 12:43:18 -07002311 ++*pos;
Olof Johansson91b9a272005-08-09 20:24:39 -07002312 if (v == SEQ_START_TOKEN)
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002313 return fib_trie_get_idx(iter, 0);
Olof Johansson91b9a272005-08-09 20:24:39 -07002314
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002315 v = fib_trie_get_next(iter);
2316 BUG_ON(v == l);
2317 if (v)
2318 return v;
2319
2320 /* continue scan in next trie */
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002321 if (iter->trie == iter->trie_local)
2322 return fib_trie_get_first(iter, iter->trie_main);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002323
2324 return NULL;
Robert Olsson19baf832005-06-21 12:43:18 -07002325}
2326
2327static void fib_trie_seq_stop(struct seq_file *seq, void *v)
Stephen Hemmingerc95aaf92008-01-12 21:25:02 -08002328 __releases(RCU)
Robert Olsson19baf832005-06-21 12:43:18 -07002329{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002330 rcu_read_unlock();
Robert Olsson19baf832005-06-21 12:43:18 -07002331}
2332
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002333static void seq_indent(struct seq_file *seq, int n)
2334{
2335 while (n-- > 0) seq_puts(seq, " ");
2336}
Robert Olsson19baf832005-06-21 12:43:18 -07002337
Eric Dumazet28d36e32008-01-14 23:09:56 -08002338static inline const char *rtn_scope(char *buf, size_t len, enum rt_scope_t s)
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002339{
Stephen Hemminger132adf52007-03-08 20:44:43 -08002340 switch (s) {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002341 case RT_SCOPE_UNIVERSE: return "universe";
2342 case RT_SCOPE_SITE: return "site";
2343 case RT_SCOPE_LINK: return "link";
2344 case RT_SCOPE_HOST: return "host";
2345 case RT_SCOPE_NOWHERE: return "nowhere";
2346 default:
Eric Dumazet28d36e32008-01-14 23:09:56 -08002347 snprintf(buf, len, "scope=%d", s);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002348 return buf;
2349 }
2350}
2351
2352static const char *rtn_type_names[__RTN_MAX] = {
2353 [RTN_UNSPEC] = "UNSPEC",
2354 [RTN_UNICAST] = "UNICAST",
2355 [RTN_LOCAL] = "LOCAL",
2356 [RTN_BROADCAST] = "BROADCAST",
2357 [RTN_ANYCAST] = "ANYCAST",
2358 [RTN_MULTICAST] = "MULTICAST",
2359 [RTN_BLACKHOLE] = "BLACKHOLE",
2360 [RTN_UNREACHABLE] = "UNREACHABLE",
2361 [RTN_PROHIBIT] = "PROHIBIT",
2362 [RTN_THROW] = "THROW",
2363 [RTN_NAT] = "NAT",
2364 [RTN_XRESOLVE] = "XRESOLVE",
2365};
2366
Eric Dumazet28d36e32008-01-14 23:09:56 -08002367static inline const char *rtn_type(char *buf, size_t len, unsigned t)
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002368{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002369 if (t < __RTN_MAX && rtn_type_names[t])
2370 return rtn_type_names[t];
Eric Dumazet28d36e32008-01-14 23:09:56 -08002371 snprintf(buf, len, "type %u", t);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002372 return buf;
2373}
2374
2375/* Pretty print the trie */
Robert Olsson19baf832005-06-21 12:43:18 -07002376static int fib_trie_seq_show(struct seq_file *seq, void *v)
2377{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002378 const struct fib_trie_iter *iter = seq->private;
2379 struct node *n = v;
Robert Olsson19baf832005-06-21 12:43:18 -07002380
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002381 if (v == SEQ_START_TOKEN)
2382 return 0;
Robert Olsson19baf832005-06-21 12:43:18 -07002383
Eric Dumazetb59cfbf2008-01-18 03:31:36 -08002384 if (!node_parent_rcu(n)) {
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002385 if (iter->trie == iter->trie_local)
Robert Olsson095b8502007-01-26 19:06:01 -08002386 seq_puts(seq, "<local>:\n");
2387 else
2388 seq_puts(seq, "<main>:\n");
2389 }
2390
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002391 if (IS_TNODE(n)) {
2392 struct tnode *tn = (struct tnode *) n;
Stephen Hemmingerab66b4a2007-08-10 15:22:58 -07002393 __be32 prf = htonl(mask_pfx(tn->key, tn->pos));
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002394
Robert Olsson1d25cd62005-09-19 15:29:52 -07002395 seq_indent(seq, iter->depth-1);
2396 seq_printf(seq, " +-- %d.%d.%d.%d/%d %d %d %d\n",
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +09002397 NIPQUAD(prf), tn->pos, tn->bits, tn->full_children,
Robert Olsson1d25cd62005-09-19 15:29:52 -07002398 tn->empty_children);
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +09002399
Olof Johansson91b9a272005-08-09 20:24:39 -07002400 } else {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002401 struct leaf *l = (struct leaf *) n;
Stephen Hemminger13280422008-01-22 21:54:37 -08002402 struct leaf_info *li;
2403 struct hlist_node *node;
2404
Al Viro32ab5f82006-09-26 22:21:45 -07002405 __be32 val = htonl(l->key);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002406
2407 seq_indent(seq, iter->depth);
2408 seq_printf(seq, " |-- %d.%d.%d.%d\n", NIPQUAD(val));
Eric Dumazet28d36e32008-01-14 23:09:56 -08002409
Stephen Hemminger13280422008-01-22 21:54:37 -08002410 hlist_for_each_entry_rcu(li, node, &l->list, hlist) {
2411 struct fib_alias *fa;
Eric Dumazet28d36e32008-01-14 23:09:56 -08002412
Stephen Hemminger13280422008-01-22 21:54:37 -08002413 list_for_each_entry_rcu(fa, &li->falh, fa_list) {
2414 char buf1[32], buf2[32];
Eric Dumazet28d36e32008-01-14 23:09:56 -08002415
Stephen Hemminger13280422008-01-22 21:54:37 -08002416 seq_indent(seq, iter->depth+1);
2417 seq_printf(seq, " /%d %s %s", li->plen,
2418 rtn_scope(buf1, sizeof(buf1),
2419 fa->fa_scope),
2420 rtn_type(buf2, sizeof(buf2),
2421 fa->fa_type));
2422 if (fa->fa_tos)
2423 seq_printf(seq, "tos =%d\n",
2424 fa->fa_tos);
2425 seq_putc(seq, '\n');
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002426 }
2427 }
Robert Olsson19baf832005-06-21 12:43:18 -07002428 }
2429
2430 return 0;
2431}
2432
Stephen Hemmingerf6908082007-03-12 14:34:29 -07002433static const struct seq_operations fib_trie_seq_ops = {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002434 .start = fib_trie_seq_start,
2435 .next = fib_trie_seq_next,
2436 .stop = fib_trie_seq_stop,
2437 .show = fib_trie_seq_show,
Robert Olsson19baf832005-06-21 12:43:18 -07002438};
2439
2440static int fib_trie_seq_open(struct inode *inode, struct file *file)
2441{
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002442 return seq_open_net(inode, file, &fib_trie_seq_ops,
2443 sizeof(struct fib_trie_iter));
Robert Olsson19baf832005-06-21 12:43:18 -07002444}
2445
Arjan van de Ven9a321442007-02-12 00:55:35 -08002446static const struct file_operations fib_trie_fops = {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002447 .owner = THIS_MODULE,
2448 .open = fib_trie_seq_open,
2449 .read = seq_read,
2450 .llseek = seq_lseek,
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002451 .release = seq_release_net,
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002452};
2453
Al Viro32ab5f82006-09-26 22:21:45 -07002454static unsigned fib_flag_trans(int type, __be32 mask, const struct fib_info *fi)
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002455{
2456 static unsigned type2flags[RTN_MAX + 1] = {
2457 [7] = RTF_REJECT, [8] = RTF_REJECT,
2458 };
2459 unsigned flags = type2flags[type];
2460
2461 if (fi && fi->fib_nh->nh_gw)
2462 flags |= RTF_GATEWAY;
Al Viro32ab5f82006-09-26 22:21:45 -07002463 if (mask == htonl(0xFFFFFFFF))
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002464 flags |= RTF_HOST;
2465 flags |= RTF_UP;
2466 return flags;
2467}
2468
2469/*
2470 * This outputs /proc/net/route.
2471 * The format of the file is not supposed to be changed
2472 * and needs to be same as fib_hash output to avoid breaking
2473 * legacy utilities
2474 */
2475static int fib_route_seq_show(struct seq_file *seq, void *v)
2476{
Patrick McHardyc9e53cb2005-11-20 21:09:00 -08002477 const struct fib_trie_iter *iter = seq->private;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002478 struct leaf *l = v;
Stephen Hemminger13280422008-01-22 21:54:37 -08002479 struct leaf_info *li;
2480 struct hlist_node *node;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002481
2482 if (v == SEQ_START_TOKEN) {
2483 seq_printf(seq, "%-127s\n", "Iface\tDestination\tGateway "
2484 "\tFlags\tRefCnt\tUse\tMetric\tMask\t\tMTU"
2485 "\tWindow\tIRTT");
2486 return 0;
2487 }
2488
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002489 if (iter->trie == iter->trie_local)
Patrick McHardyc9e53cb2005-11-20 21:09:00 -08002490 return 0;
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08002491
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002492 if (IS_TNODE(l))
2493 return 0;
2494
Stephen Hemminger13280422008-01-22 21:54:37 -08002495 hlist_for_each_entry_rcu(li, node, &l->list, hlist) {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002496 struct fib_alias *fa;
Al Viro32ab5f82006-09-26 22:21:45 -07002497 __be32 mask, prefix;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002498
2499 if (!li)
2500 continue;
2501
2502 mask = inet_make_mask(li->plen);
2503 prefix = htonl(l->key);
2504
2505 list_for_each_entry_rcu(fa, &li->falh, fa_list) {
Herbert Xu1371e372005-10-15 09:42:39 +10002506 const struct fib_info *fi = fa->fa_info;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002507 unsigned flags = fib_flag_trans(fa->fa_type, mask, fi);
Stephen Hemminger13280422008-01-22 21:54:37 -08002508 char bf[128];
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002509
2510 if (fa->fa_type == RTN_BROADCAST
2511 || fa->fa_type == RTN_MULTICAST)
2512 continue;
2513
2514 if (fi)
2515 snprintf(bf, sizeof(bf),
2516 "%s\t%08X\t%08X\t%04X\t%d\t%u\t%d\t%08X\t%d\t%u\t%u",
2517 fi->fib_dev ? fi->fib_dev->name : "*",
2518 prefix,
2519 fi->fib_nh->nh_gw, flags, 0, 0,
2520 fi->fib_priority,
2521 mask,
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08002522 (fi->fib_advmss ?
2523 fi->fib_advmss + 40 : 0),
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002524 fi->fib_window,
2525 fi->fib_rtt >> 3);
2526 else
2527 snprintf(bf, sizeof(bf),
2528 "*\t%08X\t%08X\t%04X\t%d\t%u\t%d\t%08X\t%d\t%u\t%u",
2529 prefix, 0, flags, 0, 0, 0,
2530 mask, 0, 0, 0);
2531
2532 seq_printf(seq, "%-127s\n", bf);
2533 }
2534 }
2535
2536 return 0;
2537}
2538
Stephen Hemmingerf6908082007-03-12 14:34:29 -07002539static const struct seq_operations fib_route_seq_ops = {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002540 .start = fib_trie_seq_start,
2541 .next = fib_trie_seq_next,
2542 .stop = fib_trie_seq_stop,
2543 .show = fib_route_seq_show,
2544};
2545
2546static int fib_route_seq_open(struct inode *inode, struct file *file)
2547{
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002548 return seq_open_net(inode, file, &fib_route_seq_ops,
2549 sizeof(struct fib_trie_iter));
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002550}
2551
Arjan van de Ven9a321442007-02-12 00:55:35 -08002552static const struct file_operations fib_route_fops = {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002553 .owner = THIS_MODULE,
2554 .open = fib_route_seq_open,
2555 .read = seq_read,
2556 .llseek = seq_lseek,
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002557 .release = seq_release_net,
Robert Olsson19baf832005-06-21 12:43:18 -07002558};
2559
Denis V. Lunev61a02652008-01-10 03:21:09 -08002560int __net_init fib_proc_init(struct net *net)
Robert Olsson19baf832005-06-21 12:43:18 -07002561{
Denis V. Lunev61a02652008-01-10 03:21:09 -08002562 if (!proc_net_fops_create(net, "fib_trie", S_IRUGO, &fib_trie_fops))
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002563 goto out1;
2564
Denis V. Lunev61a02652008-01-10 03:21:09 -08002565 if (!proc_net_fops_create(net, "fib_triestat", S_IRUGO,
2566 &fib_triestat_fops))
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002567 goto out2;
2568
Denis V. Lunev61a02652008-01-10 03:21:09 -08002569 if (!proc_net_fops_create(net, "route", S_IRUGO, &fib_route_fops))
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002570 goto out3;
2571
Robert Olsson19baf832005-06-21 12:43:18 -07002572 return 0;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002573
2574out3:
Denis V. Lunev61a02652008-01-10 03:21:09 -08002575 proc_net_remove(net, "fib_triestat");
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002576out2:
Denis V. Lunev61a02652008-01-10 03:21:09 -08002577 proc_net_remove(net, "fib_trie");
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002578out1:
2579 return -ENOMEM;
Robert Olsson19baf832005-06-21 12:43:18 -07002580}
2581
Denis V. Lunev61a02652008-01-10 03:21:09 -08002582void __net_exit fib_proc_exit(struct net *net)
Robert Olsson19baf832005-06-21 12:43:18 -07002583{
Denis V. Lunev61a02652008-01-10 03:21:09 -08002584 proc_net_remove(net, "fib_trie");
2585 proc_net_remove(net, "fib_triestat");
2586 proc_net_remove(net, "route");
Robert Olsson19baf832005-06-21 12:43:18 -07002587}
2588
2589#endif /* CONFIG_PROC_FS */