blob: c98f115fb0fde6d97136695075fa45c04976c4be [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 *
Robert Olsson19baf832005-06-21 12:43:18 -070025 *
26 * Code from fib_hash has been reused which includes the following header:
27 *
28 *
29 * INET An implementation of the TCP/IP protocol suite for the LINUX
30 * operating system. INET is implemented using the BSD Socket
31 * interface as the means of communication with the user level.
32 *
33 * IPv4 FIB: lookup engine and maintenance routines.
34 *
35 *
36 * Authors: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
37 *
38 * This program is free software; you can redistribute it and/or
39 * modify it under the terms of the GNU General Public License
40 * as published by the Free Software Foundation; either version
41 * 2 of the License, or (at your option) any later version.
Robert Olssonfd966252005-12-22 11:25:10 -080042 *
43 * Substantial contributions to this work comes from:
44 *
45 * David S. Miller, <davem@davemloft.net>
46 * Stephen Hemminger <shemminger@osdl.org>
47 * Paul E. McKenney <paulmck@us.ibm.com>
48 * Patrick McHardy <kaber@trash.net>
Robert Olsson19baf832005-06-21 12:43:18 -070049 */
50
Jens Låås80b71b82009-08-28 23:57:15 -070051#define VERSION "0.409"
Robert Olsson19baf832005-06-21 12:43:18 -070052
Robert Olsson19baf832005-06-21 12:43:18 -070053#include <asm/uaccess.h>
54#include <asm/system.h>
Jiri Slaby1977f032007-10-18 23:40:25 -070055#include <linux/bitops.h>
Robert Olsson19baf832005-06-21 12:43:18 -070056#include <linux/types.h>
57#include <linux/kernel.h>
Robert Olsson19baf832005-06-21 12:43:18 -070058#include <linux/mm.h>
59#include <linux/string.h>
60#include <linux/socket.h>
61#include <linux/sockios.h>
62#include <linux/errno.h>
63#include <linux/in.h>
64#include <linux/inet.h>
Stephen Hemmingercd8787a2006-01-03 14:38:34 -080065#include <linux/inetdevice.h>
Robert Olsson19baf832005-06-21 12:43:18 -070066#include <linux/netdevice.h>
67#include <linux/if_arp.h>
68#include <linux/proc_fs.h>
Robert Olsson2373ce12005-08-25 13:01:29 -070069#include <linux/rcupdate.h>
Robert Olsson19baf832005-06-21 12:43:18 -070070#include <linux/skbuff.h>
71#include <linux/netlink.h>
72#include <linux/init.h>
73#include <linux/list.h>
Tejun Heo5a0e3ad2010-03-24 17:04:11 +090074#include <linux/slab.h>
Eric W. Biederman457c4cb2007-09-12 12:01:34 +020075#include <net/net_namespace.h>
Robert Olsson19baf832005-06-21 12:43:18 -070076#include <net/ip.h>
77#include <net/protocol.h>
78#include <net/route.h>
79#include <net/tcp.h>
80#include <net/sock.h>
81#include <net/ip_fib.h>
82#include "fib_lookup.h"
83
Robert Olsson06ef9212006-03-20 21:35:01 -080084#define MAX_STAT_DEPTH 32
Robert Olsson19baf832005-06-21 12:43:18 -070085
Robert Olsson19baf832005-06-21 12:43:18 -070086#define KEYLENGTH (8*sizeof(t_key))
Robert Olsson19baf832005-06-21 12:43:18 -070087
Robert Olsson19baf832005-06-21 12:43:18 -070088typedef unsigned int t_key;
89
90#define T_TNODE 0
91#define T_LEAF 1
92#define NODE_TYPE_MASK 0x1UL
Robert Olsson2373ce12005-08-25 13:01:29 -070093#define NODE_TYPE(node) ((node)->parent & NODE_TYPE_MASK)
94
Olof Johansson91b9a272005-08-09 20:24:39 -070095#define IS_TNODE(n) (!(n->parent & T_LEAF))
96#define IS_LEAF(n) (n->parent & T_LEAF)
Robert Olsson19baf832005-06-21 12:43:18 -070097
98struct node {
Olof Johansson91b9a272005-08-09 20:24:39 -070099 unsigned long parent;
Eric Dumazet8d965442008-01-13 22:31:44 -0800100 t_key key;
Robert Olsson19baf832005-06-21 12:43:18 -0700101};
102
103struct leaf {
Olof Johansson91b9a272005-08-09 20:24:39 -0700104 unsigned long parent;
Eric Dumazet8d965442008-01-13 22:31:44 -0800105 t_key key;
Robert Olsson19baf832005-06-21 12:43:18 -0700106 struct hlist_head list;
Robert Olsson2373ce12005-08-25 13:01:29 -0700107 struct rcu_head rcu;
Robert Olsson19baf832005-06-21 12:43:18 -0700108};
109
110struct leaf_info {
111 struct hlist_node hlist;
Robert Olsson2373ce12005-08-25 13:01:29 -0700112 struct rcu_head rcu;
Robert Olsson19baf832005-06-21 12:43:18 -0700113 int plen;
114 struct list_head falh;
115};
116
117struct tnode {
Olof Johansson91b9a272005-08-09 20:24:39 -0700118 unsigned long parent;
Eric Dumazet8d965442008-01-13 22:31:44 -0800119 t_key key;
Eric Dumazet112d8cf2008-01-12 21:27:41 -0800120 unsigned char pos; /* 2log(KEYLENGTH) bits needed */
121 unsigned char bits; /* 2log(KEYLENGTH) bits needed */
Eric Dumazet8d965442008-01-13 22:31:44 -0800122 unsigned int full_children; /* KEYLENGTH bits needed */
123 unsigned int empty_children; /* KEYLENGTH bits needed */
Stephen Hemminger15be75c2008-04-10 02:56:38 -0700124 union {
125 struct rcu_head rcu;
126 struct work_struct work;
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700127 struct tnode *tnode_free;
Stephen Hemminger15be75c2008-04-10 02:56:38 -0700128 };
Olof Johansson91b9a272005-08-09 20:24:39 -0700129 struct node *child[0];
Robert Olsson19baf832005-06-21 12:43:18 -0700130};
131
132#ifdef CONFIG_IP_FIB_TRIE_STATS
133struct trie_use_stats {
134 unsigned int gets;
135 unsigned int backtrack;
136 unsigned int semantic_match_passed;
137 unsigned int semantic_match_miss;
138 unsigned int null_node_hit;
Robert Olsson2f368952005-07-05 15:02:40 -0700139 unsigned int resize_node_skipped;
Robert Olsson19baf832005-06-21 12:43:18 -0700140};
141#endif
142
143struct trie_stat {
144 unsigned int totdepth;
145 unsigned int maxdepth;
146 unsigned int tnodes;
147 unsigned int leaves;
148 unsigned int nullpointers;
Stephen Hemminger93672292008-01-22 21:54:05 -0800149 unsigned int prefixes;
Robert Olsson06ef9212006-03-20 21:35:01 -0800150 unsigned int nodesizes[MAX_STAT_DEPTH];
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700151};
Robert Olsson19baf832005-06-21 12:43:18 -0700152
153struct trie {
Olof Johansson91b9a272005-08-09 20:24:39 -0700154 struct node *trie;
Robert Olsson19baf832005-06-21 12:43:18 -0700155#ifdef CONFIG_IP_FIB_TRIE_STATS
156 struct trie_use_stats stats;
157#endif
Robert Olsson19baf832005-06-21 12:43:18 -0700158};
159
Robert Olsson19baf832005-06-21 12:43:18 -0700160static void put_child(struct trie *t, struct tnode *tn, int i, struct node *n);
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800161static void tnode_put_child_reorg(struct tnode *tn, int i, struct node *n,
162 int wasfull);
Robert Olsson19baf832005-06-21 12:43:18 -0700163static struct node *resize(struct trie *t, struct tnode *tn);
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700164static struct tnode *inflate(struct trie *t, struct tnode *tn);
165static struct tnode *halve(struct trie *t, struct tnode *tn);
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700166/* tnodes to free after resize(); protected by RTNL */
167static struct tnode *tnode_free_head;
Jarek Poplawskic3059472009-07-14 08:33:08 +0000168static size_t tnode_free_size;
169
170/*
171 * synchronize_rcu after call_rcu for that many pages; it should be especially
172 * useful before resizing the root node with PREEMPT_NONE configs; the value was
173 * obtained experimentally, aiming to avoid visible slowdown.
174 */
175static const int sync_pages = 128;
Robert Olsson19baf832005-06-21 12:43:18 -0700176
Christoph Lametere18b8902006-12-06 20:33:20 -0800177static struct kmem_cache *fn_alias_kmem __read_mostly;
Stephen Hemmingerbc3c8c12008-01-22 21:51:50 -0800178static struct kmem_cache *trie_leaf_kmem __read_mostly;
Robert Olsson19baf832005-06-21 12:43:18 -0700179
Stephen Hemminger06801912007-08-10 15:22:13 -0700180static inline struct tnode *node_parent(struct node *node)
181{
Eric Dumazetb59cfbf2008-01-18 03:31:36 -0800182 return (struct tnode *)(node->parent & ~NODE_TYPE_MASK);
183}
Stephen Hemminger06801912007-08-10 15:22:13 -0700184
Eric Dumazetb59cfbf2008-01-18 03:31:36 -0800185static inline struct tnode *node_parent_rcu(struct node *node)
186{
187 struct tnode *ret = node_parent(node);
188
Stephen Hemminger06801912007-08-10 15:22:13 -0700189 return rcu_dereference(ret);
190}
191
Stephen Hemminger6440cc92008-03-22 17:59:58 -0700192/* Same as rcu_assign_pointer
193 * but that macro() assumes that value is a pointer.
194 */
Stephen Hemminger06801912007-08-10 15:22:13 -0700195static inline void node_set_parent(struct node *node, struct tnode *ptr)
196{
Stephen Hemminger6440cc92008-03-22 17:59:58 -0700197 smp_wmb();
198 node->parent = (unsigned long)ptr | NODE_TYPE(node);
Stephen Hemminger06801912007-08-10 15:22:13 -0700199}
Robert Olsson2373ce12005-08-25 13:01:29 -0700200
Eric Dumazetb59cfbf2008-01-18 03:31:36 -0800201static inline struct node *tnode_get_child(struct tnode *tn, unsigned int i)
Robert Olsson19baf832005-06-21 12:43:18 -0700202{
Eric Dumazetb59cfbf2008-01-18 03:31:36 -0800203 BUG_ON(i >= 1U << tn->bits);
Robert Olsson19baf832005-06-21 12:43:18 -0700204
Eric Dumazetb59cfbf2008-01-18 03:31:36 -0800205 return tn->child[i];
206}
207
208static inline struct node *tnode_get_child_rcu(struct tnode *tn, unsigned int i)
209{
210 struct node *ret = tnode_get_child(tn, i);
211
Eric Dumazet4eaa0e32010-04-14 16:13:29 -0700212 return rcu_dereference_check(ret,
213 rcu_read_lock_held() ||
214 lockdep_rtnl_is_held());
Robert Olsson19baf832005-06-21 12:43:18 -0700215}
216
Stephen Hemmignerbb435b82005-08-09 20:25:39 -0700217static inline int tnode_child_length(const struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700218{
Olof Johansson91b9a272005-08-09 20:24:39 -0700219 return 1 << tn->bits;
Robert Olsson19baf832005-06-21 12:43:18 -0700220}
221
Stephen Hemmingerab66b4a2007-08-10 15:22:58 -0700222static inline t_key mask_pfx(t_key k, unsigned short l)
223{
224 return (l == 0) ? 0 : k >> (KEYLENGTH-l) << (KEYLENGTH-l);
225}
226
Robert Olsson19baf832005-06-21 12:43:18 -0700227static inline t_key tkey_extract_bits(t_key a, int offset, int bits)
228{
Olof Johansson91b9a272005-08-09 20:24:39 -0700229 if (offset < KEYLENGTH)
Robert Olsson19baf832005-06-21 12:43:18 -0700230 return ((t_key)(a << offset)) >> (KEYLENGTH - bits);
Olof Johansson91b9a272005-08-09 20:24:39 -0700231 else
Robert Olsson19baf832005-06-21 12:43:18 -0700232 return 0;
233}
234
235static inline int tkey_equals(t_key a, t_key b)
236{
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700237 return a == b;
Robert Olsson19baf832005-06-21 12:43:18 -0700238}
239
240static inline int tkey_sub_equals(t_key a, int offset, int bits, t_key b)
241{
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700242 if (bits == 0 || offset >= KEYLENGTH)
243 return 1;
Olof Johansson91b9a272005-08-09 20:24:39 -0700244 bits = bits > KEYLENGTH ? KEYLENGTH : bits;
245 return ((a ^ b) << offset) >> (KEYLENGTH - bits) == 0;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700246}
Robert Olsson19baf832005-06-21 12:43:18 -0700247
248static inline int tkey_mismatch(t_key a, int offset, t_key b)
249{
250 t_key diff = a ^ b;
251 int i = offset;
252
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700253 if (!diff)
254 return 0;
255 while ((diff << i) >> (KEYLENGTH-1) == 0)
Robert Olsson19baf832005-06-21 12:43:18 -0700256 i++;
257 return i;
258}
259
Robert Olsson19baf832005-06-21 12:43:18 -0700260/*
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900261 To understand this stuff, an understanding of keys and all their bits is
262 necessary. Every node in the trie has a key associated with it, but not
Robert Olsson19baf832005-06-21 12:43:18 -0700263 all of the bits in that key are significant.
264
265 Consider a node 'n' and its parent 'tp'.
266
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900267 If n is a leaf, every bit in its key is significant. Its presence is
268 necessitated by path compression, since during a tree traversal (when
269 searching for a leaf - unless we are doing an insertion) we will completely
270 ignore all skipped bits we encounter. Thus we need to verify, at the end of
271 a potentially successful search, that we have indeed been walking the
Robert Olsson19baf832005-06-21 12:43:18 -0700272 correct key path.
273
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900274 Note that we can never "miss" the correct key in the tree if present by
275 following the wrong path. Path compression ensures that segments of the key
276 that are the same for all keys with a given prefix are skipped, but the
277 skipped part *is* identical for each node in the subtrie below the skipped
278 bit! trie_insert() in this implementation takes care of that - note the
Robert Olsson19baf832005-06-21 12:43:18 -0700279 call to tkey_sub_equals() in trie_insert().
280
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900281 if n is an internal node - a 'tnode' here, the various parts of its key
Robert Olsson19baf832005-06-21 12:43:18 -0700282 have many different meanings.
283
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900284 Example:
Robert Olsson19baf832005-06-21 12:43:18 -0700285 _________________________________________________________________
286 | i | i | i | i | i | i | i | N | N | N | S | S | S | S | S | C |
287 -----------------------------------------------------------------
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900288 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
Robert Olsson19baf832005-06-21 12:43:18 -0700289
290 _________________________________________________________________
291 | C | C | C | u | u | u | u | u | u | u | u | u | u | u | u | u |
292 -----------------------------------------------------------------
293 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
294
295 tp->pos = 7
296 tp->bits = 3
297 n->pos = 15
Olof Johansson91b9a272005-08-09 20:24:39 -0700298 n->bits = 4
Robert Olsson19baf832005-06-21 12:43:18 -0700299
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900300 First, let's just ignore the bits that come before the parent tp, that is
301 the bits from 0 to (tp->pos-1). They are *known* but at this point we do
Robert Olsson19baf832005-06-21 12:43:18 -0700302 not use them for anything.
303
304 The bits from (tp->pos) to (tp->pos + tp->bits - 1) - "N", above - are the
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900305 index into the parent's child array. That is, they will be used to find
Robert Olsson19baf832005-06-21 12:43:18 -0700306 'n' among tp's children.
307
308 The bits from (tp->pos + tp->bits) to (n->pos - 1) - "S" - are skipped bits
309 for the node n.
310
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900311 All the bits we have seen so far are significant to the node n. The rest
Robert Olsson19baf832005-06-21 12:43:18 -0700312 of the bits are really not needed or indeed known in n->key.
313
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900314 The bits from (n->pos) to (n->pos + n->bits - 1) - "C" - are the index into
Robert Olsson19baf832005-06-21 12:43:18 -0700315 n's child array, and will of course be different for each child.
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900316
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700317
Robert Olsson19baf832005-06-21 12:43:18 -0700318 The rest of the bits, from (n->pos + n->bits) onward, are completely unknown
319 at this point.
320
321*/
322
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700323static inline void check_tnode(const struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700324{
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700325 WARN_ON(tn && tn->pos+tn->bits > 32);
Robert Olsson19baf832005-06-21 12:43:18 -0700326}
327
Denis V. Lunevf5026fa2007-12-13 09:47:57 -0800328static const int halve_threshold = 25;
329static const int inflate_threshold = 50;
Jarek Poplawski345aa032009-07-07 19:39:16 -0700330static const int halve_threshold_root = 15;
Jens Låås80b71b82009-08-28 23:57:15 -0700331static const int inflate_threshold_root = 30;
Robert Olsson2373ce12005-08-25 13:01:29 -0700332
333static void __alias_free_mem(struct rcu_head *head)
334{
335 struct fib_alias *fa = container_of(head, struct fib_alias, rcu);
336 kmem_cache_free(fn_alias_kmem, fa);
337}
338
339static inline void alias_free_mem_rcu(struct fib_alias *fa)
340{
341 call_rcu(&fa->rcu, __alias_free_mem);
342}
343
344static void __leaf_free_rcu(struct rcu_head *head)
345{
Stephen Hemmingerbc3c8c12008-01-22 21:51:50 -0800346 struct leaf *l = container_of(head, struct leaf, rcu);
347 kmem_cache_free(trie_leaf_kmem, l);
Robert Olsson2373ce12005-08-25 13:01:29 -0700348}
349
Stephen Hemminger387a5482008-04-10 03:47:34 -0700350static inline void free_leaf(struct leaf *l)
351{
352 call_rcu_bh(&l->rcu, __leaf_free_rcu);
353}
354
Robert Olsson2373ce12005-08-25 13:01:29 -0700355static void __leaf_info_free_rcu(struct rcu_head *head)
356{
357 kfree(container_of(head, struct leaf_info, rcu));
358}
359
360static inline void free_leaf_info(struct leaf_info *leaf)
361{
362 call_rcu(&leaf->rcu, __leaf_info_free_rcu);
363}
364
Eric Dumazet8d965442008-01-13 22:31:44 -0800365static struct tnode *tnode_alloc(size_t size)
Robert Olsson2373ce12005-08-25 13:01:29 -0700366{
Robert Olsson2373ce12005-08-25 13:01:29 -0700367 if (size <= PAGE_SIZE)
Eric Dumazet8d965442008-01-13 22:31:44 -0800368 return kzalloc(size, GFP_KERNEL);
Stephen Hemminger15be75c2008-04-10 02:56:38 -0700369 else
370 return __vmalloc(size, GFP_KERNEL | __GFP_ZERO, PAGE_KERNEL);
371}
Robert Olsson2373ce12005-08-25 13:01:29 -0700372
Stephen Hemminger15be75c2008-04-10 02:56:38 -0700373static void __tnode_vfree(struct work_struct *arg)
374{
375 struct tnode *tn = container_of(arg, struct tnode, work);
376 vfree(tn);
Robert Olsson2373ce12005-08-25 13:01:29 -0700377}
378
379static void __tnode_free_rcu(struct rcu_head *head)
380{
381 struct tnode *tn = container_of(head, struct tnode, rcu);
Eric Dumazet8d965442008-01-13 22:31:44 -0800382 size_t size = sizeof(struct tnode) +
383 (sizeof(struct node *) << tn->bits);
Robert Olsson2373ce12005-08-25 13:01:29 -0700384
385 if (size <= PAGE_SIZE)
386 kfree(tn);
Stephen Hemminger15be75c2008-04-10 02:56:38 -0700387 else {
388 INIT_WORK(&tn->work, __tnode_vfree);
389 schedule_work(&tn->work);
390 }
Robert Olsson2373ce12005-08-25 13:01:29 -0700391}
392
393static inline void tnode_free(struct tnode *tn)
394{
Stephen Hemminger387a5482008-04-10 03:47:34 -0700395 if (IS_LEAF(tn))
396 free_leaf((struct leaf *) tn);
397 else
Robert Olsson550e29b2006-04-04 12:53:35 -0700398 call_rcu(&tn->rcu, __tnode_free_rcu);
Robert Olsson2373ce12005-08-25 13:01:29 -0700399}
400
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700401static void tnode_free_safe(struct tnode *tn)
402{
403 BUG_ON(IS_LEAF(tn));
Jarek Poplawski7b855762009-06-18 00:28:51 -0700404 tn->tnode_free = tnode_free_head;
405 tnode_free_head = tn;
Jarek Poplawskic3059472009-07-14 08:33:08 +0000406 tnode_free_size += sizeof(struct tnode) +
407 (sizeof(struct node *) << tn->bits);
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700408}
409
410static void tnode_free_flush(void)
411{
412 struct tnode *tn;
413
414 while ((tn = tnode_free_head)) {
415 tnode_free_head = tn->tnode_free;
416 tn->tnode_free = NULL;
417 tnode_free(tn);
418 }
Jarek Poplawskic3059472009-07-14 08:33:08 +0000419
420 if (tnode_free_size >= PAGE_SIZE * sync_pages) {
421 tnode_free_size = 0;
422 synchronize_rcu();
423 }
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700424}
425
Robert Olsson19baf832005-06-21 12:43:18 -0700426static struct leaf *leaf_new(void)
427{
Stephen Hemmingerbc3c8c12008-01-22 21:51:50 -0800428 struct leaf *l = kmem_cache_alloc(trie_leaf_kmem, GFP_KERNEL);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700429 if (l) {
Robert Olsson2373ce12005-08-25 13:01:29 -0700430 l->parent = T_LEAF;
Robert Olsson19baf832005-06-21 12:43:18 -0700431 INIT_HLIST_HEAD(&l->list);
432 }
433 return l;
434}
435
436static struct leaf_info *leaf_info_new(int plen)
437{
438 struct leaf_info *li = kmalloc(sizeof(struct leaf_info), GFP_KERNEL);
Robert Olsson2373ce12005-08-25 13:01:29 -0700439 if (li) {
440 li->plen = plen;
441 INIT_LIST_HEAD(&li->falh);
Patrick McHardyf0e36f82005-07-05 14:44:55 -0700442 }
Robert Olsson2373ce12005-08-25 13:01:29 -0700443 return li;
Patrick McHardyf0e36f82005-07-05 14:44:55 -0700444}
445
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800446static struct tnode *tnode_new(t_key key, int pos, int bits)
Robert Olsson19baf832005-06-21 12:43:18 -0700447{
Eric Dumazet8d965442008-01-13 22:31:44 -0800448 size_t sz = sizeof(struct tnode) + (sizeof(struct node *) << bits);
Patrick McHardyf0e36f82005-07-05 14:44:55 -0700449 struct tnode *tn = tnode_alloc(sz);
Robert Olsson19baf832005-06-21 12:43:18 -0700450
Olof Johansson91b9a272005-08-09 20:24:39 -0700451 if (tn) {
Robert Olsson2373ce12005-08-25 13:01:29 -0700452 tn->parent = T_TNODE;
Robert Olsson19baf832005-06-21 12:43:18 -0700453 tn->pos = pos;
454 tn->bits = bits;
455 tn->key = key;
456 tn->full_children = 0;
457 tn->empty_children = 1<<bits;
458 }
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700459
Eric Dumazet8d965442008-01-13 22:31:44 -0800460 pr_debug("AT %p s=%u %lu\n", tn, (unsigned int) sizeof(struct tnode),
461 (unsigned long) (sizeof(struct node) << bits));
Robert Olsson19baf832005-06-21 12:43:18 -0700462 return tn;
463}
464
Robert Olsson19baf832005-06-21 12:43:18 -0700465/*
466 * Check whether a tnode 'n' is "full", i.e. it is an internal node
467 * and no bits are skipped. See discussion in dyntree paper p. 6
468 */
469
Stephen Hemmignerbb435b82005-08-09 20:25:39 -0700470static inline int tnode_full(const struct tnode *tn, const struct node *n)
Robert Olsson19baf832005-06-21 12:43:18 -0700471{
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700472 if (n == NULL || IS_LEAF(n))
Robert Olsson19baf832005-06-21 12:43:18 -0700473 return 0;
474
475 return ((struct tnode *) n)->pos == tn->pos + tn->bits;
476}
477
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800478static inline void put_child(struct trie *t, struct tnode *tn, int i,
479 struct node *n)
Robert Olsson19baf832005-06-21 12:43:18 -0700480{
481 tnode_put_child_reorg(tn, i, n, -1);
482}
483
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700484 /*
Robert Olsson19baf832005-06-21 12:43:18 -0700485 * Add a child at position i overwriting the old value.
486 * Update the value of full_children and empty_children.
487 */
488
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800489static void tnode_put_child_reorg(struct tnode *tn, int i, struct node *n,
490 int wasfull)
Robert Olsson19baf832005-06-21 12:43:18 -0700491{
Robert Olsson2373ce12005-08-25 13:01:29 -0700492 struct node *chi = tn->child[i];
Robert Olsson19baf832005-06-21 12:43:18 -0700493 int isfull;
494
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700495 BUG_ON(i >= 1<<tn->bits);
496
Robert Olsson19baf832005-06-21 12:43:18 -0700497 /* update emptyChildren */
498 if (n == NULL && chi != NULL)
499 tn->empty_children++;
500 else if (n != NULL && chi == NULL)
501 tn->empty_children--;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700502
Robert Olsson19baf832005-06-21 12:43:18 -0700503 /* update fullChildren */
Olof Johansson91b9a272005-08-09 20:24:39 -0700504 if (wasfull == -1)
Robert Olsson19baf832005-06-21 12:43:18 -0700505 wasfull = tnode_full(tn, chi);
506
507 isfull = tnode_full(tn, n);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700508 if (wasfull && !isfull)
Robert Olsson19baf832005-06-21 12:43:18 -0700509 tn->full_children--;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700510 else if (!wasfull && isfull)
Robert Olsson19baf832005-06-21 12:43:18 -0700511 tn->full_children++;
Olof Johansson91b9a272005-08-09 20:24:39 -0700512
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700513 if (n)
Stephen Hemminger06801912007-08-10 15:22:13 -0700514 node_set_parent(n, tn);
Robert Olsson19baf832005-06-21 12:43:18 -0700515
Robert Olsson2373ce12005-08-25 13:01:29 -0700516 rcu_assign_pointer(tn->child[i], n);
Robert Olsson19baf832005-06-21 12:43:18 -0700517}
518
Jens Låås80b71b82009-08-28 23:57:15 -0700519#define MAX_WORK 10
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700520static struct node *resize(struct trie *t, struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700521{
522 int i;
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700523 struct tnode *old_tn;
Robert Olssone6308be2005-10-04 13:01:58 -0700524 int inflate_threshold_use;
525 int halve_threshold_use;
Jens Låås80b71b82009-08-28 23:57:15 -0700526 int max_work;
Robert Olsson19baf832005-06-21 12:43:18 -0700527
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900528 if (!tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700529 return NULL;
530
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700531 pr_debug("In tnode_resize %p inflate_threshold=%d threshold=%d\n",
532 tn, inflate_threshold, halve_threshold);
Robert Olsson19baf832005-06-21 12:43:18 -0700533
534 /* No children */
535 if (tn->empty_children == tnode_child_length(tn)) {
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700536 tnode_free_safe(tn);
Robert Olsson19baf832005-06-21 12:43:18 -0700537 return NULL;
538 }
539 /* One child */
540 if (tn->empty_children == tnode_child_length(tn) - 1)
Jens Låås80b71b82009-08-28 23:57:15 -0700541 goto one_child;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700542 /*
Robert Olsson19baf832005-06-21 12:43:18 -0700543 * Double as long as the resulting node has a number of
544 * nonempty nodes that are above the threshold.
545 */
546
547 /*
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700548 * From "Implementing a dynamic compressed trie" by Stefan Nilsson of
549 * the Helsinki University of Technology and Matti Tikkanen of Nokia
Robert Olsson19baf832005-06-21 12:43:18 -0700550 * Telecommunications, page 6:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700551 * "A node is doubled if the ratio of non-empty children to all
Robert Olsson19baf832005-06-21 12:43:18 -0700552 * children in the *doubled* node is at least 'high'."
553 *
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700554 * 'high' in this instance is the variable 'inflate_threshold'. It
555 * is expressed as a percentage, so we multiply it with
556 * tnode_child_length() and instead of multiplying by 2 (since the
557 * child array will be doubled by inflate()) and multiplying
558 * the left-hand side by 100 (to handle the percentage thing) we
Robert Olsson19baf832005-06-21 12:43:18 -0700559 * multiply the left-hand side by 50.
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700560 *
561 * The left-hand side may look a bit weird: tnode_child_length(tn)
562 * - tn->empty_children is of course the number of non-null children
563 * in the current node. tn->full_children is the number of "full"
Robert Olsson19baf832005-06-21 12:43:18 -0700564 * children, that is non-null tnodes with a skip value of 0.
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700565 * All of those will be doubled in the resulting inflated tnode, so
Robert Olsson19baf832005-06-21 12:43:18 -0700566 * we just count them one extra time here.
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700567 *
Robert Olsson19baf832005-06-21 12:43:18 -0700568 * A clearer way to write this would be:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700569 *
Robert Olsson19baf832005-06-21 12:43:18 -0700570 * to_be_doubled = tn->full_children;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700571 * not_to_be_doubled = tnode_child_length(tn) - tn->empty_children -
Robert Olsson19baf832005-06-21 12:43:18 -0700572 * tn->full_children;
573 *
574 * new_child_length = tnode_child_length(tn) * 2;
575 *
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700576 * new_fill_factor = 100 * (not_to_be_doubled + 2*to_be_doubled) /
Robert Olsson19baf832005-06-21 12:43:18 -0700577 * new_child_length;
578 * if (new_fill_factor >= inflate_threshold)
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700579 *
580 * ...and so on, tho it would mess up the while () loop.
581 *
Robert Olsson19baf832005-06-21 12:43:18 -0700582 * anyway,
583 * 100 * (not_to_be_doubled + 2*to_be_doubled) / new_child_length >=
584 * inflate_threshold
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700585 *
Robert Olsson19baf832005-06-21 12:43:18 -0700586 * avoid a division:
587 * 100 * (not_to_be_doubled + 2*to_be_doubled) >=
588 * inflate_threshold * new_child_length
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700589 *
Robert Olsson19baf832005-06-21 12:43:18 -0700590 * expand not_to_be_doubled and to_be_doubled, and shorten:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700591 * 100 * (tnode_child_length(tn) - tn->empty_children +
Olof Johansson91b9a272005-08-09 20:24:39 -0700592 * tn->full_children) >= inflate_threshold * new_child_length
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700593 *
Robert Olsson19baf832005-06-21 12:43:18 -0700594 * expand new_child_length:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700595 * 100 * (tnode_child_length(tn) - tn->empty_children +
Olof Johansson91b9a272005-08-09 20:24:39 -0700596 * tn->full_children) >=
Robert Olsson19baf832005-06-21 12:43:18 -0700597 * inflate_threshold * tnode_child_length(tn) * 2
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700598 *
Robert Olsson19baf832005-06-21 12:43:18 -0700599 * shorten again:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700600 * 50 * (tn->full_children + tnode_child_length(tn) -
Olof Johansson91b9a272005-08-09 20:24:39 -0700601 * tn->empty_children) >= inflate_threshold *
Robert Olsson19baf832005-06-21 12:43:18 -0700602 * tnode_child_length(tn)
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700603 *
Robert Olsson19baf832005-06-21 12:43:18 -0700604 */
605
606 check_tnode(tn);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700607
Robert Olssone6308be2005-10-04 13:01:58 -0700608 /* Keep root node larger */
609
Jens Låås80b71b82009-08-28 23:57:15 -0700610 if (!node_parent((struct node*) tn)) {
611 inflate_threshold_use = inflate_threshold_root;
612 halve_threshold_use = halve_threshold_root;
613 }
614 else {
Robert Olssone6308be2005-10-04 13:01:58 -0700615 inflate_threshold_use = inflate_threshold;
Jens Låås80b71b82009-08-28 23:57:15 -0700616 halve_threshold_use = halve_threshold;
617 }
Robert Olssone6308be2005-10-04 13:01:58 -0700618
Jens Låås80b71b82009-08-28 23:57:15 -0700619 max_work = MAX_WORK;
620 while ((tn->full_children > 0 && max_work-- &&
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800621 50 * (tn->full_children + tnode_child_length(tn)
622 - tn->empty_children)
623 >= inflate_threshold_use * tnode_child_length(tn))) {
Robert Olsson19baf832005-06-21 12:43:18 -0700624
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700625 old_tn = tn;
626 tn = inflate(t, tn);
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800627
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700628 if (IS_ERR(tn)) {
629 tn = old_tn;
Robert Olsson2f368952005-07-05 15:02:40 -0700630#ifdef CONFIG_IP_FIB_TRIE_STATS
631 t->stats.resize_node_skipped++;
632#endif
633 break;
634 }
Robert Olsson19baf832005-06-21 12:43:18 -0700635 }
636
637 check_tnode(tn);
638
Jens Låås80b71b82009-08-28 23:57:15 -0700639 /* Return if at least one inflate is run */
640 if( max_work != MAX_WORK)
641 return (struct node *) tn;
642
Robert Olsson19baf832005-06-21 12:43:18 -0700643 /*
644 * Halve as long as the number of empty children in this
645 * node is above threshold.
646 */
Robert Olsson2f368952005-07-05 15:02:40 -0700647
Jens Låås80b71b82009-08-28 23:57:15 -0700648 max_work = MAX_WORK;
649 while (tn->bits > 1 && max_work-- &&
Robert Olsson19baf832005-06-21 12:43:18 -0700650 100 * (tnode_child_length(tn) - tn->empty_children) <
Robert Olssone6308be2005-10-04 13:01:58 -0700651 halve_threshold_use * tnode_child_length(tn)) {
Robert Olsson19baf832005-06-21 12:43:18 -0700652
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700653 old_tn = tn;
654 tn = halve(t, tn);
655 if (IS_ERR(tn)) {
656 tn = old_tn;
Robert Olsson2f368952005-07-05 15:02:40 -0700657#ifdef CONFIG_IP_FIB_TRIE_STATS
658 t->stats.resize_node_skipped++;
659#endif
660 break;
661 }
662 }
663
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700664
Robert Olsson19baf832005-06-21 12:43:18 -0700665 /* Only one child remains */
Jens Låås80b71b82009-08-28 23:57:15 -0700666 if (tn->empty_children == tnode_child_length(tn) - 1) {
667one_child:
Robert Olsson19baf832005-06-21 12:43:18 -0700668 for (i = 0; i < tnode_child_length(tn); i++) {
Olof Johansson91b9a272005-08-09 20:24:39 -0700669 struct node *n;
670
Olof Johansson91b9a272005-08-09 20:24:39 -0700671 n = tn->child[i];
Robert Olsson2373ce12005-08-25 13:01:29 -0700672 if (!n)
Olof Johansson91b9a272005-08-09 20:24:39 -0700673 continue;
Olof Johansson91b9a272005-08-09 20:24:39 -0700674
675 /* compress one level */
676
Stephen Hemminger06801912007-08-10 15:22:13 -0700677 node_set_parent(n, NULL);
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700678 tnode_free_safe(tn);
Olof Johansson91b9a272005-08-09 20:24:39 -0700679 return n;
Robert Olsson19baf832005-06-21 12:43:18 -0700680 }
Jens Låås80b71b82009-08-28 23:57:15 -0700681 }
Robert Olsson19baf832005-06-21 12:43:18 -0700682 return (struct node *) tn;
683}
684
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700685static struct tnode *inflate(struct trie *t, struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700686{
Robert Olsson19baf832005-06-21 12:43:18 -0700687 struct tnode *oldtnode = tn;
688 int olen = tnode_child_length(tn);
689 int i;
690
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700691 pr_debug("In inflate\n");
Robert Olsson19baf832005-06-21 12:43:18 -0700692
693 tn = tnode_new(oldtnode->key, oldtnode->pos, oldtnode->bits + 1);
694
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700695 if (!tn)
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700696 return ERR_PTR(-ENOMEM);
Robert Olsson2f368952005-07-05 15:02:40 -0700697
698 /*
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700699 * Preallocate and store tnodes before the actual work so we
700 * don't get into an inconsistent state if memory allocation
701 * fails. In case of failure we return the oldnode and inflate
Robert Olsson2f368952005-07-05 15:02:40 -0700702 * of tnode is ignored.
703 */
Olof Johansson91b9a272005-08-09 20:24:39 -0700704
705 for (i = 0; i < olen; i++) {
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800706 struct tnode *inode;
Robert Olsson2f368952005-07-05 15:02:40 -0700707
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800708 inode = (struct tnode *) tnode_get_child(oldtnode, i);
Robert Olsson2f368952005-07-05 15:02:40 -0700709 if (inode &&
710 IS_TNODE(inode) &&
711 inode->pos == oldtnode->pos + oldtnode->bits &&
712 inode->bits > 1) {
713 struct tnode *left, *right;
Stephen Hemmingerab66b4a2007-08-10 15:22:58 -0700714 t_key m = ~0U << (KEYLENGTH - 1) >> inode->pos;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700715
Robert Olsson2f368952005-07-05 15:02:40 -0700716 left = tnode_new(inode->key&(~m), inode->pos + 1,
717 inode->bits - 1);
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700718 if (!left)
719 goto nomem;
Olof Johansson91b9a272005-08-09 20:24:39 -0700720
Robert Olsson2f368952005-07-05 15:02:40 -0700721 right = tnode_new(inode->key|m, inode->pos + 1,
722 inode->bits - 1);
723
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900724 if (!right) {
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700725 tnode_free(left);
726 goto nomem;
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900727 }
Robert Olsson2f368952005-07-05 15:02:40 -0700728
729 put_child(t, tn, 2*i, (struct node *) left);
730 put_child(t, tn, 2*i+1, (struct node *) right);
731 }
732 }
733
Olof Johansson91b9a272005-08-09 20:24:39 -0700734 for (i = 0; i < olen; i++) {
Stephen Hemmingerc95aaf92008-01-12 21:25:02 -0800735 struct tnode *inode;
Robert Olsson19baf832005-06-21 12:43:18 -0700736 struct node *node = tnode_get_child(oldtnode, i);
Olof Johansson91b9a272005-08-09 20:24:39 -0700737 struct tnode *left, *right;
738 int size, j;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700739
Robert Olsson19baf832005-06-21 12:43:18 -0700740 /* An empty child */
741 if (node == NULL)
742 continue;
743
744 /* A leaf or an internal node with skipped bits */
745
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700746 if (IS_LEAF(node) || ((struct tnode *) node)->pos >
Robert Olsson19baf832005-06-21 12:43:18 -0700747 tn->pos + tn->bits - 1) {
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800748 if (tkey_extract_bits(node->key,
749 oldtnode->pos + oldtnode->bits,
750 1) == 0)
Robert Olsson19baf832005-06-21 12:43:18 -0700751 put_child(t, tn, 2*i, node);
752 else
753 put_child(t, tn, 2*i+1, node);
754 continue;
755 }
756
757 /* An internal node with two children */
758 inode = (struct tnode *) node;
759
760 if (inode->bits == 1) {
761 put_child(t, tn, 2*i, inode->child[0]);
762 put_child(t, tn, 2*i+1, inode->child[1]);
763
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700764 tnode_free_safe(inode);
Olof Johansson91b9a272005-08-09 20:24:39 -0700765 continue;
Robert Olsson19baf832005-06-21 12:43:18 -0700766 }
767
Olof Johansson91b9a272005-08-09 20:24:39 -0700768 /* An internal node with more than two children */
Robert Olsson19baf832005-06-21 12:43:18 -0700769
Olof Johansson91b9a272005-08-09 20:24:39 -0700770 /* We will replace this node 'inode' with two new
771 * ones, 'left' and 'right', each with half of the
772 * original children. The two new nodes will have
773 * a position one bit further down the key and this
774 * means that the "significant" part of their keys
775 * (see the discussion near the top of this file)
776 * will differ by one bit, which will be "0" in
777 * left's key and "1" in right's key. Since we are
778 * moving the key position by one step, the bit that
779 * we are moving away from - the bit at position
780 * (inode->pos) - is the one that will differ between
781 * left and right. So... we synthesize that bit in the
782 * two new keys.
783 * The mask 'm' below will be a single "one" bit at
784 * the position (inode->pos)
785 */
Robert Olsson19baf832005-06-21 12:43:18 -0700786
Olof Johansson91b9a272005-08-09 20:24:39 -0700787 /* Use the old key, but set the new significant
788 * bit to zero.
789 */
Robert Olsson19baf832005-06-21 12:43:18 -0700790
Olof Johansson91b9a272005-08-09 20:24:39 -0700791 left = (struct tnode *) tnode_get_child(tn, 2*i);
792 put_child(t, tn, 2*i, NULL);
Robert Olsson19baf832005-06-21 12:43:18 -0700793
Olof Johansson91b9a272005-08-09 20:24:39 -0700794 BUG_ON(!left);
Robert Olsson2f368952005-07-05 15:02:40 -0700795
Olof Johansson91b9a272005-08-09 20:24:39 -0700796 right = (struct tnode *) tnode_get_child(tn, 2*i+1);
797 put_child(t, tn, 2*i+1, NULL);
Robert Olsson2f368952005-07-05 15:02:40 -0700798
Olof Johansson91b9a272005-08-09 20:24:39 -0700799 BUG_ON(!right);
Robert Olsson2f368952005-07-05 15:02:40 -0700800
Olof Johansson91b9a272005-08-09 20:24:39 -0700801 size = tnode_child_length(left);
802 for (j = 0; j < size; j++) {
803 put_child(t, left, j, inode->child[j]);
804 put_child(t, right, j, inode->child[j + size]);
Robert Olsson19baf832005-06-21 12:43:18 -0700805 }
Olof Johansson91b9a272005-08-09 20:24:39 -0700806 put_child(t, tn, 2*i, resize(t, left));
807 put_child(t, tn, 2*i+1, resize(t, right));
808
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700809 tnode_free_safe(inode);
Robert Olsson19baf832005-06-21 12:43:18 -0700810 }
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700811 tnode_free_safe(oldtnode);
Robert Olsson19baf832005-06-21 12:43:18 -0700812 return tn;
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700813nomem:
814 {
815 int size = tnode_child_length(tn);
816 int j;
817
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700818 for (j = 0; j < size; j++)
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700819 if (tn->child[j])
820 tnode_free((struct tnode *)tn->child[j]);
821
822 tnode_free(tn);
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700823
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700824 return ERR_PTR(-ENOMEM);
825 }
Robert Olsson19baf832005-06-21 12:43:18 -0700826}
827
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700828static struct tnode *halve(struct trie *t, struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700829{
830 struct tnode *oldtnode = tn;
831 struct node *left, *right;
832 int i;
833 int olen = tnode_child_length(tn);
834
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700835 pr_debug("In halve\n");
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700836
837 tn = tnode_new(oldtnode->key, oldtnode->pos, oldtnode->bits - 1);
Robert Olsson19baf832005-06-21 12:43:18 -0700838
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700839 if (!tn)
840 return ERR_PTR(-ENOMEM);
Robert Olsson2f368952005-07-05 15:02:40 -0700841
842 /*
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700843 * Preallocate and store tnodes before the actual work so we
844 * don't get into an inconsistent state if memory allocation
845 * fails. In case of failure we return the oldnode and halve
Robert Olsson2f368952005-07-05 15:02:40 -0700846 * of tnode is ignored.
847 */
848
Olof Johansson91b9a272005-08-09 20:24:39 -0700849 for (i = 0; i < olen; i += 2) {
Robert Olsson2f368952005-07-05 15:02:40 -0700850 left = tnode_get_child(oldtnode, i);
851 right = tnode_get_child(oldtnode, i+1);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700852
Robert Olsson2f368952005-07-05 15:02:40 -0700853 /* Two nonempty children */
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700854 if (left && right) {
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700855 struct tnode *newn;
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700856
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700857 newn = tnode_new(left->key, tn->pos + tn->bits, 1);
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700858
859 if (!newn)
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700860 goto nomem;
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700861
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700862 put_child(t, tn, i/2, (struct node *)newn);
Robert Olsson2f368952005-07-05 15:02:40 -0700863 }
Robert Olsson2f368952005-07-05 15:02:40 -0700864
Robert Olsson2f368952005-07-05 15:02:40 -0700865 }
Robert Olsson19baf832005-06-21 12:43:18 -0700866
Olof Johansson91b9a272005-08-09 20:24:39 -0700867 for (i = 0; i < olen; i += 2) {
868 struct tnode *newBinNode;
869
Robert Olsson19baf832005-06-21 12:43:18 -0700870 left = tnode_get_child(oldtnode, i);
871 right = tnode_get_child(oldtnode, i+1);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700872
Robert Olsson19baf832005-06-21 12:43:18 -0700873 /* At least one of the children is empty */
874 if (left == NULL) {
875 if (right == NULL) /* Both are empty */
876 continue;
877 put_child(t, tn, i/2, right);
Olof Johansson91b9a272005-08-09 20:24:39 -0700878 continue;
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700879 }
Olof Johansson91b9a272005-08-09 20:24:39 -0700880
881 if (right == NULL) {
Robert Olsson19baf832005-06-21 12:43:18 -0700882 put_child(t, tn, i/2, left);
Olof Johansson91b9a272005-08-09 20:24:39 -0700883 continue;
884 }
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700885
Robert Olsson19baf832005-06-21 12:43:18 -0700886 /* Two nonempty children */
Olof Johansson91b9a272005-08-09 20:24:39 -0700887 newBinNode = (struct tnode *) tnode_get_child(tn, i/2);
888 put_child(t, tn, i/2, NULL);
Olof Johansson91b9a272005-08-09 20:24:39 -0700889 put_child(t, newBinNode, 0, left);
890 put_child(t, newBinNode, 1, right);
891 put_child(t, tn, i/2, resize(t, newBinNode));
Robert Olsson19baf832005-06-21 12:43:18 -0700892 }
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700893 tnode_free_safe(oldtnode);
Robert Olsson19baf832005-06-21 12:43:18 -0700894 return tn;
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700895nomem:
896 {
897 int size = tnode_child_length(tn);
898 int j;
899
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700900 for (j = 0; j < size; j++)
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700901 if (tn->child[j])
902 tnode_free((struct tnode *)tn->child[j]);
903
904 tnode_free(tn);
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700905
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700906 return ERR_PTR(-ENOMEM);
907 }
Robert Olsson19baf832005-06-21 12:43:18 -0700908}
909
Robert Olsson772cb712005-09-19 15:31:18 -0700910/* readside must use rcu_read_lock currently dump routines
Robert Olsson2373ce12005-08-25 13:01:29 -0700911 via get_fa_head and dump */
912
Robert Olsson772cb712005-09-19 15:31:18 -0700913static struct leaf_info *find_leaf_info(struct leaf *l, int plen)
Robert Olsson19baf832005-06-21 12:43:18 -0700914{
Robert Olsson772cb712005-09-19 15:31:18 -0700915 struct hlist_head *head = &l->list;
Robert Olsson19baf832005-06-21 12:43:18 -0700916 struct hlist_node *node;
917 struct leaf_info *li;
918
Robert Olsson2373ce12005-08-25 13:01:29 -0700919 hlist_for_each_entry_rcu(li, node, head, hlist)
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700920 if (li->plen == plen)
Robert Olsson19baf832005-06-21 12:43:18 -0700921 return li;
Olof Johansson91b9a272005-08-09 20:24:39 -0700922
Robert Olsson19baf832005-06-21 12:43:18 -0700923 return NULL;
924}
925
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800926static inline struct list_head *get_fa_head(struct leaf *l, int plen)
Robert Olsson19baf832005-06-21 12:43:18 -0700927{
Robert Olsson772cb712005-09-19 15:31:18 -0700928 struct leaf_info *li = find_leaf_info(l, plen);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700929
Olof Johansson91b9a272005-08-09 20:24:39 -0700930 if (!li)
931 return NULL;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700932
Olof Johansson91b9a272005-08-09 20:24:39 -0700933 return &li->falh;
Robert Olsson19baf832005-06-21 12:43:18 -0700934}
935
936static void insert_leaf_info(struct hlist_head *head, struct leaf_info *new)
937{
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900938 struct leaf_info *li = NULL, *last = NULL;
939 struct hlist_node *node;
Robert Olsson19baf832005-06-21 12:43:18 -0700940
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900941 if (hlist_empty(head)) {
942 hlist_add_head_rcu(&new->hlist, head);
943 } else {
944 hlist_for_each_entry(li, node, head, hlist) {
945 if (new->plen > li->plen)
946 break;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700947
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900948 last = li;
949 }
950 if (last)
951 hlist_add_after_rcu(&last->hlist, &new->hlist);
952 else
953 hlist_add_before_rcu(&new->hlist, &li->hlist);
954 }
Robert Olsson19baf832005-06-21 12:43:18 -0700955}
956
Robert Olsson2373ce12005-08-25 13:01:29 -0700957/* rcu_read_lock needs to be hold by caller from readside */
958
Robert Olsson19baf832005-06-21 12:43:18 -0700959static struct leaf *
960fib_find_node(struct trie *t, u32 key)
961{
962 int pos;
963 struct tnode *tn;
964 struct node *n;
965
966 pos = 0;
Paul E. McKenney634a4b22010-03-21 18:01:05 -0700967 n = rcu_dereference_check(t->trie,
968 rcu_read_lock_held() ||
969 lockdep_rtnl_is_held());
Robert Olsson19baf832005-06-21 12:43:18 -0700970
971 while (n != NULL && NODE_TYPE(n) == T_TNODE) {
972 tn = (struct tnode *) n;
Olof Johansson91b9a272005-08-09 20:24:39 -0700973
Robert Olsson19baf832005-06-21 12:43:18 -0700974 check_tnode(tn);
Olof Johansson91b9a272005-08-09 20:24:39 -0700975
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700976 if (tkey_sub_equals(tn->key, pos, tn->pos-pos, key)) {
Olof Johansson91b9a272005-08-09 20:24:39 -0700977 pos = tn->pos + tn->bits;
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800978 n = tnode_get_child_rcu(tn,
979 tkey_extract_bits(key,
980 tn->pos,
981 tn->bits));
Olof Johansson91b9a272005-08-09 20:24:39 -0700982 } else
Robert Olsson19baf832005-06-21 12:43:18 -0700983 break;
984 }
985 /* Case we have found a leaf. Compare prefixes */
986
Olof Johansson91b9a272005-08-09 20:24:39 -0700987 if (n != NULL && IS_LEAF(n) && tkey_equals(key, n->key))
988 return (struct leaf *)n;
989
Robert Olsson19baf832005-06-21 12:43:18 -0700990 return NULL;
991}
992
Jarek Poplawski7b855762009-06-18 00:28:51 -0700993static void trie_rebalance(struct trie *t, struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700994{
Robert Olsson19baf832005-06-21 12:43:18 -0700995 int wasfull;
Robert Olsson3ed18d72009-05-21 15:20:59 -0700996 t_key cindex, key;
Stephen Hemminger06801912007-08-10 15:22:13 -0700997 struct tnode *tp;
Robert Olsson19baf832005-06-21 12:43:18 -0700998
Robert Olsson3ed18d72009-05-21 15:20:59 -0700999 key = tn->key;
1000
Stephen Hemminger06801912007-08-10 15:22:13 -07001001 while (tn != NULL && (tp = node_parent((struct node *)tn)) != NULL) {
Robert Olsson19baf832005-06-21 12:43:18 -07001002 cindex = tkey_extract_bits(key, tp->pos, tp->bits);
1003 wasfull = tnode_full(tp, tnode_get_child(tp, cindex));
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001004 tn = (struct tnode *) resize(t, (struct tnode *)tn);
1005
1006 tnode_put_child_reorg((struct tnode *)tp, cindex,
1007 (struct node *)tn, wasfull);
Olof Johansson91b9a272005-08-09 20:24:39 -07001008
Stephen Hemminger06801912007-08-10 15:22:13 -07001009 tp = node_parent((struct node *) tn);
Jarek Poplawski008440e2009-06-30 12:47:19 -07001010 if (!tp)
1011 rcu_assign_pointer(t->trie, (struct node *)tn);
1012
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -07001013 tnode_free_flush();
Stephen Hemminger06801912007-08-10 15:22:13 -07001014 if (!tp)
Robert Olsson19baf832005-06-21 12:43:18 -07001015 break;
Stephen Hemminger06801912007-08-10 15:22:13 -07001016 tn = tp;
Robert Olsson19baf832005-06-21 12:43:18 -07001017 }
Stephen Hemminger06801912007-08-10 15:22:13 -07001018
Robert Olsson19baf832005-06-21 12:43:18 -07001019 /* Handle last (top) tnode */
Jarek Poplawski7b855762009-06-18 00:28:51 -07001020 if (IS_TNODE(tn))
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001021 tn = (struct tnode *)resize(t, (struct tnode *)tn);
Robert Olsson19baf832005-06-21 12:43:18 -07001022
Jarek Poplawski7b855762009-06-18 00:28:51 -07001023 rcu_assign_pointer(t->trie, (struct node *)tn);
1024 tnode_free_flush();
1025
1026 return;
Robert Olsson19baf832005-06-21 12:43:18 -07001027}
1028
Robert Olsson2373ce12005-08-25 13:01:29 -07001029/* only used from updater-side */
1030
Stephen Hemmingerfea86ad2008-01-12 20:57:07 -08001031static struct list_head *fib_insert_node(struct trie *t, u32 key, int plen)
Robert Olsson19baf832005-06-21 12:43:18 -07001032{
1033 int pos, newpos;
1034 struct tnode *tp = NULL, *tn = NULL;
1035 struct node *n;
1036 struct leaf *l;
1037 int missbit;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001038 struct list_head *fa_head = NULL;
Robert Olsson19baf832005-06-21 12:43:18 -07001039 struct leaf_info *li;
1040 t_key cindex;
1041
1042 pos = 0;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001043 n = t->trie;
Robert Olsson19baf832005-06-21 12:43:18 -07001044
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001045 /* If we point to NULL, stop. Either the tree is empty and we should
1046 * just put a new leaf in if, or we have reached an empty child slot,
Robert Olsson19baf832005-06-21 12:43:18 -07001047 * and we should just put our new leaf in that.
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001048 * If we point to a T_TNODE, check if it matches our key. Note that
1049 * a T_TNODE might be skipping any number of bits - its 'pos' need
Robert Olsson19baf832005-06-21 12:43:18 -07001050 * not be the parent's 'pos'+'bits'!
1051 *
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001052 * If it does match the current key, get pos/bits from it, extract
Robert Olsson19baf832005-06-21 12:43:18 -07001053 * the index from our key, push the T_TNODE and walk the tree.
1054 *
1055 * If it doesn't, we have to replace it with a new T_TNODE.
1056 *
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001057 * If we point to a T_LEAF, it might or might not have the same key
1058 * as we do. If it does, just change the value, update the T_LEAF's
1059 * value, and return it.
Robert Olsson19baf832005-06-21 12:43:18 -07001060 * If it doesn't, we need to replace it with a T_TNODE.
1061 */
1062
1063 while (n != NULL && NODE_TYPE(n) == T_TNODE) {
1064 tn = (struct tnode *) n;
Olof Johansson91b9a272005-08-09 20:24:39 -07001065
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001066 check_tnode(tn);
Olof Johansson91b9a272005-08-09 20:24:39 -07001067
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001068 if (tkey_sub_equals(tn->key, pos, tn->pos-pos, key)) {
Robert Olsson19baf832005-06-21 12:43:18 -07001069 tp = tn;
Olof Johansson91b9a272005-08-09 20:24:39 -07001070 pos = tn->pos + tn->bits;
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001071 n = tnode_get_child(tn,
1072 tkey_extract_bits(key,
1073 tn->pos,
1074 tn->bits));
Robert Olsson19baf832005-06-21 12:43:18 -07001075
Stephen Hemminger06801912007-08-10 15:22:13 -07001076 BUG_ON(n && node_parent(n) != tn);
Olof Johansson91b9a272005-08-09 20:24:39 -07001077 } else
Robert Olsson19baf832005-06-21 12:43:18 -07001078 break;
1079 }
1080
1081 /*
1082 * n ----> NULL, LEAF or TNODE
1083 *
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001084 * tp is n's (parent) ----> NULL or TNODE
Robert Olsson19baf832005-06-21 12:43:18 -07001085 */
1086
Olof Johansson91b9a272005-08-09 20:24:39 -07001087 BUG_ON(tp && IS_LEAF(tp));
Robert Olsson19baf832005-06-21 12:43:18 -07001088
1089 /* Case 1: n is a leaf. Compare prefixes */
1090
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001091 if (n != NULL && IS_LEAF(n) && tkey_equals(key, n->key)) {
Stephen Hemmingerc95aaf92008-01-12 21:25:02 -08001092 l = (struct leaf *) n;
Robert Olsson19baf832005-06-21 12:43:18 -07001093 li = leaf_info_new(plen);
Olof Johansson91b9a272005-08-09 20:24:39 -07001094
Stephen Hemmingerfea86ad2008-01-12 20:57:07 -08001095 if (!li)
1096 return NULL;
Robert Olsson19baf832005-06-21 12:43:18 -07001097
1098 fa_head = &li->falh;
1099 insert_leaf_info(&l->list, li);
1100 goto done;
1101 }
Robert Olsson19baf832005-06-21 12:43:18 -07001102 l = leaf_new();
1103
Stephen Hemmingerfea86ad2008-01-12 20:57:07 -08001104 if (!l)
1105 return NULL;
Robert Olsson19baf832005-06-21 12:43:18 -07001106
1107 l->key = key;
1108 li = leaf_info_new(plen);
1109
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001110 if (!li) {
Stephen Hemminger387a5482008-04-10 03:47:34 -07001111 free_leaf(l);
Stephen Hemmingerfea86ad2008-01-12 20:57:07 -08001112 return NULL;
Robert Olssonf835e472005-06-28 15:00:39 -07001113 }
Robert Olsson19baf832005-06-21 12:43:18 -07001114
1115 fa_head = &li->falh;
1116 insert_leaf_info(&l->list, li);
1117
Robert Olsson19baf832005-06-21 12:43:18 -07001118 if (t->trie && n == NULL) {
Olof Johansson91b9a272005-08-09 20:24:39 -07001119 /* Case 2: n is NULL, and will just insert a new leaf */
Robert Olsson19baf832005-06-21 12:43:18 -07001120
Stephen Hemminger06801912007-08-10 15:22:13 -07001121 node_set_parent((struct node *)l, tp);
Robert Olsson19baf832005-06-21 12:43:18 -07001122
Olof Johansson91b9a272005-08-09 20:24:39 -07001123 cindex = tkey_extract_bits(key, tp->pos, tp->bits);
1124 put_child(t, (struct tnode *)tp, cindex, (struct node *)l);
1125 } else {
1126 /* Case 3: n is a LEAF or a TNODE and the key doesn't match. */
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001127 /*
1128 * Add a new tnode here
Robert Olsson19baf832005-06-21 12:43:18 -07001129 * first tnode need some special handling
1130 */
1131
1132 if (tp)
Olof Johansson91b9a272005-08-09 20:24:39 -07001133 pos = tp->pos+tp->bits;
Robert Olsson19baf832005-06-21 12:43:18 -07001134 else
Olof Johansson91b9a272005-08-09 20:24:39 -07001135 pos = 0;
1136
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001137 if (n) {
Robert Olsson19baf832005-06-21 12:43:18 -07001138 newpos = tkey_mismatch(key, pos, n->key);
1139 tn = tnode_new(n->key, newpos, 1);
Olof Johansson91b9a272005-08-09 20:24:39 -07001140 } else {
Robert Olsson19baf832005-06-21 12:43:18 -07001141 newpos = 0;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001142 tn = tnode_new(key, newpos, 1); /* First tnode */
Robert Olsson19baf832005-06-21 12:43:18 -07001143 }
Robert Olsson19baf832005-06-21 12:43:18 -07001144
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001145 if (!tn) {
Robert Olssonf835e472005-06-28 15:00:39 -07001146 free_leaf_info(li);
Stephen Hemminger387a5482008-04-10 03:47:34 -07001147 free_leaf(l);
Stephen Hemmingerfea86ad2008-01-12 20:57:07 -08001148 return NULL;
Olof Johansson91b9a272005-08-09 20:24:39 -07001149 }
1150
Stephen Hemminger06801912007-08-10 15:22:13 -07001151 node_set_parent((struct node *)tn, tp);
Robert Olsson19baf832005-06-21 12:43:18 -07001152
Olof Johansson91b9a272005-08-09 20:24:39 -07001153 missbit = tkey_extract_bits(key, newpos, 1);
Robert Olsson19baf832005-06-21 12:43:18 -07001154 put_child(t, tn, missbit, (struct node *)l);
1155 put_child(t, tn, 1-missbit, n);
1156
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001157 if (tp) {
Robert Olsson19baf832005-06-21 12:43:18 -07001158 cindex = tkey_extract_bits(key, tp->pos, tp->bits);
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001159 put_child(t, (struct tnode *)tp, cindex,
1160 (struct node *)tn);
Olof Johansson91b9a272005-08-09 20:24:39 -07001161 } else {
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001162 rcu_assign_pointer(t->trie, (struct node *)tn);
Robert Olsson19baf832005-06-21 12:43:18 -07001163 tp = tn;
1164 }
1165 }
Olof Johansson91b9a272005-08-09 20:24:39 -07001166
1167 if (tp && tp->pos + tp->bits > 32)
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001168 pr_warning("fib_trie"
1169 " tp=%p pos=%d, bits=%d, key=%0x plen=%d\n",
1170 tp, tp->pos, tp->bits, key, plen);
Olof Johansson91b9a272005-08-09 20:24:39 -07001171
Robert Olsson19baf832005-06-21 12:43:18 -07001172 /* Rebalance the trie */
Robert Olsson2373ce12005-08-25 13:01:29 -07001173
Jarek Poplawski7b855762009-06-18 00:28:51 -07001174 trie_rebalance(t, tp);
Robert Olssonf835e472005-06-28 15:00:39 -07001175done:
Robert Olsson19baf832005-06-21 12:43:18 -07001176 return fa_head;
1177}
1178
Robert Olssond562f1f2007-03-26 14:22:22 -07001179/*
1180 * Caller must hold RTNL.
1181 */
Stephen Hemminger16c6cf82009-09-20 10:35:36 +00001182int fib_table_insert(struct fib_table *tb, struct fib_config *cfg)
Robert Olsson19baf832005-06-21 12:43:18 -07001183{
1184 struct trie *t = (struct trie *) tb->tb_data;
1185 struct fib_alias *fa, *new_fa;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001186 struct list_head *fa_head = NULL;
Robert Olsson19baf832005-06-21 12:43:18 -07001187 struct fib_info *fi;
Thomas Graf4e902c52006-08-17 18:14:52 -07001188 int plen = cfg->fc_dst_len;
1189 u8 tos = cfg->fc_tos;
Robert Olsson19baf832005-06-21 12:43:18 -07001190 u32 key, mask;
1191 int err;
1192 struct leaf *l;
1193
1194 if (plen > 32)
1195 return -EINVAL;
1196
Thomas Graf4e902c52006-08-17 18:14:52 -07001197 key = ntohl(cfg->fc_dst);
Robert Olsson19baf832005-06-21 12:43:18 -07001198
Patrick McHardy2dfe55b2006-08-10 23:08:33 -07001199 pr_debug("Insert table=%u %08x/%d\n", tb->tb_id, key, plen);
Robert Olsson19baf832005-06-21 12:43:18 -07001200
Olof Johansson91b9a272005-08-09 20:24:39 -07001201 mask = ntohl(inet_make_mask(plen));
Robert Olsson19baf832005-06-21 12:43:18 -07001202
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001203 if (key & ~mask)
Robert Olsson19baf832005-06-21 12:43:18 -07001204 return -EINVAL;
1205
1206 key = key & mask;
1207
Thomas Graf4e902c52006-08-17 18:14:52 -07001208 fi = fib_create_info(cfg);
1209 if (IS_ERR(fi)) {
1210 err = PTR_ERR(fi);
Robert Olsson19baf832005-06-21 12:43:18 -07001211 goto err;
Thomas Graf4e902c52006-08-17 18:14:52 -07001212 }
Robert Olsson19baf832005-06-21 12:43:18 -07001213
1214 l = fib_find_node(t, key);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001215 fa = NULL;
Robert Olsson19baf832005-06-21 12:43:18 -07001216
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001217 if (l) {
Robert Olsson19baf832005-06-21 12:43:18 -07001218 fa_head = get_fa_head(l, plen);
1219 fa = fib_find_alias(fa_head, tos, fi->fib_priority);
1220 }
1221
1222 /* Now fa, if non-NULL, points to the first fib alias
1223 * with the same keys [prefix,tos,priority], if such key already
1224 * exists or to the node before which we will insert new one.
1225 *
1226 * If fa is NULL, we will need to allocate a new one and
1227 * insert to the head of f.
1228 *
1229 * If f is NULL, no fib node matched the destination key
1230 * and we need to allocate a new one of those as well.
1231 */
1232
Julian Anastasov936f6f82008-01-28 21:18:06 -08001233 if (fa && fa->fa_tos == tos &&
1234 fa->fa_info->fib_priority == fi->fib_priority) {
1235 struct fib_alias *fa_first, *fa_match;
Robert Olsson19baf832005-06-21 12:43:18 -07001236
1237 err = -EEXIST;
Thomas Graf4e902c52006-08-17 18:14:52 -07001238 if (cfg->fc_nlflags & NLM_F_EXCL)
Robert Olsson19baf832005-06-21 12:43:18 -07001239 goto out;
1240
Julian Anastasov936f6f82008-01-28 21:18:06 -08001241 /* We have 2 goals:
1242 * 1. Find exact match for type, scope, fib_info to avoid
1243 * duplicate routes
1244 * 2. Find next 'fa' (or head), NLM_F_APPEND inserts before it
1245 */
1246 fa_match = NULL;
1247 fa_first = fa;
1248 fa = list_entry(fa->fa_list.prev, struct fib_alias, fa_list);
1249 list_for_each_entry_continue(fa, fa_head, fa_list) {
1250 if (fa->fa_tos != tos)
1251 break;
1252 if (fa->fa_info->fib_priority != fi->fib_priority)
1253 break;
1254 if (fa->fa_type == cfg->fc_type &&
1255 fa->fa_scope == cfg->fc_scope &&
1256 fa->fa_info == fi) {
1257 fa_match = fa;
1258 break;
1259 }
1260 }
1261
Thomas Graf4e902c52006-08-17 18:14:52 -07001262 if (cfg->fc_nlflags & NLM_F_REPLACE) {
Robert Olsson19baf832005-06-21 12:43:18 -07001263 struct fib_info *fi_drop;
1264 u8 state;
1265
Julian Anastasov936f6f82008-01-28 21:18:06 -08001266 fa = fa_first;
1267 if (fa_match) {
1268 if (fa == fa_match)
1269 err = 0;
Joonwoo Park67250332008-01-18 03:45:18 -08001270 goto out;
Julian Anastasov936f6f82008-01-28 21:18:06 -08001271 }
Robert Olsson2373ce12005-08-25 13:01:29 -07001272 err = -ENOBUFS;
Christoph Lametere94b1762006-12-06 20:33:17 -08001273 new_fa = kmem_cache_alloc(fn_alias_kmem, GFP_KERNEL);
Robert Olsson2373ce12005-08-25 13:01:29 -07001274 if (new_fa == NULL)
1275 goto out;
Robert Olsson19baf832005-06-21 12:43:18 -07001276
1277 fi_drop = fa->fa_info;
Robert Olsson2373ce12005-08-25 13:01:29 -07001278 new_fa->fa_tos = fa->fa_tos;
1279 new_fa->fa_info = fi;
Thomas Graf4e902c52006-08-17 18:14:52 -07001280 new_fa->fa_type = cfg->fc_type;
1281 new_fa->fa_scope = cfg->fc_scope;
Robert Olsson19baf832005-06-21 12:43:18 -07001282 state = fa->fa_state;
Julian Anastasov936f6f82008-01-28 21:18:06 -08001283 new_fa->fa_state = state & ~FA_S_ACCESSED;
Robert Olsson19baf832005-06-21 12:43:18 -07001284
Robert Olsson2373ce12005-08-25 13:01:29 -07001285 list_replace_rcu(&fa->fa_list, &new_fa->fa_list);
1286 alias_free_mem_rcu(fa);
Robert Olsson19baf832005-06-21 12:43:18 -07001287
1288 fib_release_info(fi_drop);
1289 if (state & FA_S_ACCESSED)
Denis V. Lunev76e6ebf2008-07-05 19:00:44 -07001290 rt_cache_flush(cfg->fc_nlinfo.nl_net, -1);
Milan Kocianb8f55832007-05-23 14:55:06 -07001291 rtmsg_fib(RTM_NEWROUTE, htonl(key), new_fa, plen,
1292 tb->tb_id, &cfg->fc_nlinfo, NLM_F_REPLACE);
Robert Olsson19baf832005-06-21 12:43:18 -07001293
Olof Johansson91b9a272005-08-09 20:24:39 -07001294 goto succeeded;
Robert Olsson19baf832005-06-21 12:43:18 -07001295 }
1296 /* Error if we find a perfect match which
1297 * uses the same scope, type, and nexthop
1298 * information.
1299 */
Julian Anastasov936f6f82008-01-28 21:18:06 -08001300 if (fa_match)
1301 goto out;
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001302
Thomas Graf4e902c52006-08-17 18:14:52 -07001303 if (!(cfg->fc_nlflags & NLM_F_APPEND))
Julian Anastasov936f6f82008-01-28 21:18:06 -08001304 fa = fa_first;
Robert Olsson19baf832005-06-21 12:43:18 -07001305 }
1306 err = -ENOENT;
Thomas Graf4e902c52006-08-17 18:14:52 -07001307 if (!(cfg->fc_nlflags & NLM_F_CREATE))
Robert Olsson19baf832005-06-21 12:43:18 -07001308 goto out;
1309
1310 err = -ENOBUFS;
Christoph Lametere94b1762006-12-06 20:33:17 -08001311 new_fa = kmem_cache_alloc(fn_alias_kmem, GFP_KERNEL);
Robert Olsson19baf832005-06-21 12:43:18 -07001312 if (new_fa == NULL)
1313 goto out;
1314
1315 new_fa->fa_info = fi;
1316 new_fa->fa_tos = tos;
Thomas Graf4e902c52006-08-17 18:14:52 -07001317 new_fa->fa_type = cfg->fc_type;
1318 new_fa->fa_scope = cfg->fc_scope;
Robert Olsson19baf832005-06-21 12:43:18 -07001319 new_fa->fa_state = 0;
Robert Olsson19baf832005-06-21 12:43:18 -07001320 /*
1321 * Insert new entry to the list.
1322 */
1323
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001324 if (!fa_head) {
Stephen Hemmingerfea86ad2008-01-12 20:57:07 -08001325 fa_head = fib_insert_node(t, key, plen);
1326 if (unlikely(!fa_head)) {
1327 err = -ENOMEM;
Robert Olssonf835e472005-06-28 15:00:39 -07001328 goto out_free_new_fa;
Stephen Hemmingerfea86ad2008-01-12 20:57:07 -08001329 }
Robert Olssonf835e472005-06-28 15:00:39 -07001330 }
Robert Olsson19baf832005-06-21 12:43:18 -07001331
Robert Olsson2373ce12005-08-25 13:01:29 -07001332 list_add_tail_rcu(&new_fa->fa_list,
1333 (fa ? &fa->fa_list : fa_head));
Robert Olsson19baf832005-06-21 12:43:18 -07001334
Denis V. Lunev76e6ebf2008-07-05 19:00:44 -07001335 rt_cache_flush(cfg->fc_nlinfo.nl_net, -1);
Thomas Graf4e902c52006-08-17 18:14:52 -07001336 rtmsg_fib(RTM_NEWROUTE, htonl(key), new_fa, plen, tb->tb_id,
Milan Kocianb8f55832007-05-23 14:55:06 -07001337 &cfg->fc_nlinfo, 0);
Robert Olsson19baf832005-06-21 12:43:18 -07001338succeeded:
1339 return 0;
Robert Olssonf835e472005-06-28 15:00:39 -07001340
1341out_free_new_fa:
1342 kmem_cache_free(fn_alias_kmem, new_fa);
Robert Olsson19baf832005-06-21 12:43:18 -07001343out:
1344 fib_release_info(fi);
Olof Johansson91b9a272005-08-09 20:24:39 -07001345err:
Robert Olsson19baf832005-06-21 12:43:18 -07001346 return err;
1347}
1348
Robert Olsson772cb712005-09-19 15:31:18 -07001349/* should be called with rcu_read_lock */
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001350static int check_leaf(struct trie *t, struct leaf *l,
1351 t_key key, const struct flowi *flp,
1352 struct fib_result *res)
Robert Olsson19baf832005-06-21 12:43:18 -07001353{
Robert Olsson19baf832005-06-21 12:43:18 -07001354 struct leaf_info *li;
1355 struct hlist_head *hhead = &l->list;
1356 struct hlist_node *node;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001357
Robert Olsson2373ce12005-08-25 13:01:29 -07001358 hlist_for_each_entry_rcu(li, node, hhead, hlist) {
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001359 int err;
1360 int plen = li->plen;
1361 __be32 mask = inet_make_mask(plen);
1362
Al Viro888454c2006-09-19 13:42:46 -07001363 if (l->key != (key & ntohl(mask)))
Robert Olsson19baf832005-06-21 12:43:18 -07001364 continue;
1365
Rami Rosene204a342009-05-18 01:19:12 +00001366 err = fib_semantic_match(&li->falh, flp, res, plen);
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001367
Robert Olsson19baf832005-06-21 12:43:18 -07001368#ifdef CONFIG_IP_FIB_TRIE_STATS
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001369 if (err <= 0)
Robert Olsson19baf832005-06-21 12:43:18 -07001370 t->stats.semantic_match_passed++;
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001371 else
1372 t->stats.semantic_match_miss++;
Robert Olsson19baf832005-06-21 12:43:18 -07001373#endif
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001374 if (err <= 0)
Ben Hutchings2e655572008-07-10 16:52:52 -07001375 return err;
Robert Olsson19baf832005-06-21 12:43:18 -07001376 }
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001377
Ben Hutchings2e655572008-07-10 16:52:52 -07001378 return 1;
Robert Olsson19baf832005-06-21 12:43:18 -07001379}
1380
Stephen Hemminger16c6cf82009-09-20 10:35:36 +00001381int fib_table_lookup(struct fib_table *tb, const struct flowi *flp,
1382 struct fib_result *res)
Robert Olsson19baf832005-06-21 12:43:18 -07001383{
1384 struct trie *t = (struct trie *) tb->tb_data;
Ben Hutchings2e655572008-07-10 16:52:52 -07001385 int ret;
Robert Olsson19baf832005-06-21 12:43:18 -07001386 struct node *n;
1387 struct tnode *pn;
1388 int pos, bits;
Olof Johansson91b9a272005-08-09 20:24:39 -07001389 t_key key = ntohl(flp->fl4_dst);
Robert Olsson19baf832005-06-21 12:43:18 -07001390 int chopped_off;
1391 t_key cindex = 0;
1392 int current_prefix_length = KEYLENGTH;
Olof Johansson91b9a272005-08-09 20:24:39 -07001393 struct tnode *cn;
1394 t_key node_prefix, key_prefix, pref_mismatch;
1395 int mp;
1396
Robert Olsson2373ce12005-08-25 13:01:29 -07001397 rcu_read_lock();
Robert Olsson19baf832005-06-21 12:43:18 -07001398
Robert Olsson2373ce12005-08-25 13:01:29 -07001399 n = rcu_dereference(t->trie);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001400 if (!n)
Robert Olsson19baf832005-06-21 12:43:18 -07001401 goto failed;
1402
1403#ifdef CONFIG_IP_FIB_TRIE_STATS
1404 t->stats.gets++;
1405#endif
1406
1407 /* Just a leaf? */
1408 if (IS_LEAF(n)) {
Ben Hutchings2e655572008-07-10 16:52:52 -07001409 ret = check_leaf(t, (struct leaf *)n, key, flp, res);
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001410 goto found;
Robert Olsson19baf832005-06-21 12:43:18 -07001411 }
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001412
Robert Olsson19baf832005-06-21 12:43:18 -07001413 pn = (struct tnode *) n;
1414 chopped_off = 0;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001415
Olof Johansson91b9a272005-08-09 20:24:39 -07001416 while (pn) {
Robert Olsson19baf832005-06-21 12:43:18 -07001417 pos = pn->pos;
1418 bits = pn->bits;
1419
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001420 if (!chopped_off)
Stephen Hemmingerab66b4a2007-08-10 15:22:58 -07001421 cindex = tkey_extract_bits(mask_pfx(key, current_prefix_length),
1422 pos, bits);
Robert Olsson19baf832005-06-21 12:43:18 -07001423
Jarek Poplawskib902e572009-07-14 11:20:32 +00001424 n = tnode_get_child_rcu(pn, cindex);
Robert Olsson19baf832005-06-21 12:43:18 -07001425
1426 if (n == NULL) {
1427#ifdef CONFIG_IP_FIB_TRIE_STATS
1428 t->stats.null_node_hit++;
1429#endif
1430 goto backtrace;
1431 }
1432
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001433 if (IS_LEAF(n)) {
Ben Hutchings2e655572008-07-10 16:52:52 -07001434 ret = check_leaf(t, (struct leaf *)n, key, flp, res);
1435 if (ret > 0)
Olof Johansson91b9a272005-08-09 20:24:39 -07001436 goto backtrace;
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001437 goto found;
Olof Johansson91b9a272005-08-09 20:24:39 -07001438 }
1439
Olof Johansson91b9a272005-08-09 20:24:39 -07001440 cn = (struct tnode *)n;
1441
1442 /*
1443 * It's a tnode, and we can do some extra checks here if we
1444 * like, to avoid descending into a dead-end branch.
1445 * This tnode is in the parent's child array at index
1446 * key[p_pos..p_pos+p_bits] but potentially with some bits
1447 * chopped off, so in reality the index may be just a
1448 * subprefix, padded with zero at the end.
1449 * We can also take a look at any skipped bits in this
1450 * tnode - everything up to p_pos is supposed to be ok,
1451 * and the non-chopped bits of the index (se previous
1452 * paragraph) are also guaranteed ok, but the rest is
1453 * considered unknown.
1454 *
1455 * The skipped bits are key[pos+bits..cn->pos].
1456 */
1457
1458 /* If current_prefix_length < pos+bits, we are already doing
1459 * actual prefix matching, which means everything from
1460 * pos+(bits-chopped_off) onward must be zero along some
1461 * branch of this subtree - otherwise there is *no* valid
1462 * prefix present. Here we can only check the skipped
1463 * bits. Remember, since we have already indexed into the
1464 * parent's child array, we know that the bits we chopped of
1465 * *are* zero.
1466 */
1467
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001468 /* NOTA BENE: Checking only skipped bits
1469 for the new node here */
Olof Johansson91b9a272005-08-09 20:24:39 -07001470
1471 if (current_prefix_length < pos+bits) {
1472 if (tkey_extract_bits(cn->key, current_prefix_length,
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001473 cn->pos - current_prefix_length)
1474 || !(cn->child[0]))
Olof Johansson91b9a272005-08-09 20:24:39 -07001475 goto backtrace;
1476 }
1477
1478 /*
1479 * If chopped_off=0, the index is fully validated and we
1480 * only need to look at the skipped bits for this, the new,
1481 * tnode. What we actually want to do is to find out if
1482 * these skipped bits match our key perfectly, or if we will
1483 * have to count on finding a matching prefix further down,
1484 * because if we do, we would like to have some way of
1485 * verifying the existence of such a prefix at this point.
1486 */
1487
1488 /* The only thing we can do at this point is to verify that
1489 * any such matching prefix can indeed be a prefix to our
1490 * key, and if the bits in the node we are inspecting that
1491 * do not match our key are not ZERO, this cannot be true.
1492 * Thus, find out where there is a mismatch (before cn->pos)
1493 * and verify that all the mismatching bits are zero in the
1494 * new tnode's key.
1495 */
1496
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001497 /*
1498 * Note: We aren't very concerned about the piece of
1499 * the key that precede pn->pos+pn->bits, since these
1500 * have already been checked. The bits after cn->pos
1501 * aren't checked since these are by definition
1502 * "unknown" at this point. Thus, what we want to see
1503 * is if we are about to enter the "prefix matching"
1504 * state, and in that case verify that the skipped
1505 * bits that will prevail throughout this subtree are
1506 * zero, as they have to be if we are to find a
1507 * matching prefix.
Olof Johansson91b9a272005-08-09 20:24:39 -07001508 */
1509
Stephen Hemmingerab66b4a2007-08-10 15:22:58 -07001510 node_prefix = mask_pfx(cn->key, cn->pos);
1511 key_prefix = mask_pfx(key, cn->pos);
Olof Johansson91b9a272005-08-09 20:24:39 -07001512 pref_mismatch = key_prefix^node_prefix;
1513 mp = 0;
1514
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001515 /*
1516 * In short: If skipped bits in this node do not match
1517 * the search key, enter the "prefix matching"
1518 * state.directly.
Olof Johansson91b9a272005-08-09 20:24:39 -07001519 */
1520 if (pref_mismatch) {
1521 while (!(pref_mismatch & (1<<(KEYLENGTH-1)))) {
1522 mp++;
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001523 pref_mismatch = pref_mismatch << 1;
Olof Johansson91b9a272005-08-09 20:24:39 -07001524 }
1525 key_prefix = tkey_extract_bits(cn->key, mp, cn->pos-mp);
1526
1527 if (key_prefix != 0)
1528 goto backtrace;
1529
1530 if (current_prefix_length >= cn->pos)
1531 current_prefix_length = mp;
1532 }
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001533
Olof Johansson91b9a272005-08-09 20:24:39 -07001534 pn = (struct tnode *)n; /* Descend */
1535 chopped_off = 0;
1536 continue;
1537
Robert Olsson19baf832005-06-21 12:43:18 -07001538backtrace:
1539 chopped_off++;
1540
1541 /* As zero don't change the child key (cindex) */
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001542 while ((chopped_off <= pn->bits)
1543 && !(cindex & (1<<(chopped_off-1))))
Robert Olsson19baf832005-06-21 12:43:18 -07001544 chopped_off++;
Robert Olsson19baf832005-06-21 12:43:18 -07001545
1546 /* Decrease current_... with bits chopped off */
1547 if (current_prefix_length > pn->pos + pn->bits - chopped_off)
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001548 current_prefix_length = pn->pos + pn->bits
1549 - chopped_off;
Olof Johansson91b9a272005-08-09 20:24:39 -07001550
Robert Olsson19baf832005-06-21 12:43:18 -07001551 /*
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001552 * Either we do the actual chop off according or if we have
Robert Olsson19baf832005-06-21 12:43:18 -07001553 * chopped off all bits in this tnode walk up to our parent.
1554 */
1555
Olof Johansson91b9a272005-08-09 20:24:39 -07001556 if (chopped_off <= pn->bits) {
Robert Olsson19baf832005-06-21 12:43:18 -07001557 cindex &= ~(1 << (chopped_off-1));
Olof Johansson91b9a272005-08-09 20:24:39 -07001558 } else {
Jarek Poplawskib902e572009-07-14 11:20:32 +00001559 struct tnode *parent = node_parent_rcu((struct node *) pn);
Stephen Hemminger06801912007-08-10 15:22:13 -07001560 if (!parent)
Robert Olsson19baf832005-06-21 12:43:18 -07001561 goto failed;
Olof Johansson91b9a272005-08-09 20:24:39 -07001562
Robert Olsson19baf832005-06-21 12:43:18 -07001563 /* Get Child's index */
Stephen Hemminger06801912007-08-10 15:22:13 -07001564 cindex = tkey_extract_bits(pn->key, parent->pos, parent->bits);
1565 pn = parent;
Robert Olsson19baf832005-06-21 12:43:18 -07001566 chopped_off = 0;
1567
1568#ifdef CONFIG_IP_FIB_TRIE_STATS
1569 t->stats.backtrack++;
1570#endif
1571 goto backtrace;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001572 }
Robert Olsson19baf832005-06-21 12:43:18 -07001573 }
1574failed:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001575 ret = 1;
Robert Olsson19baf832005-06-21 12:43:18 -07001576found:
Robert Olsson2373ce12005-08-25 13:01:29 -07001577 rcu_read_unlock();
Robert Olsson19baf832005-06-21 12:43:18 -07001578 return ret;
1579}
1580
Stephen Hemminger9195bef2008-01-22 21:56:34 -08001581/*
1582 * Remove the leaf and return parent.
1583 */
1584static void trie_leaf_remove(struct trie *t, struct leaf *l)
Robert Olsson19baf832005-06-21 12:43:18 -07001585{
Stephen Hemminger9195bef2008-01-22 21:56:34 -08001586 struct tnode *tp = node_parent((struct node *) l);
Robert Olsson19baf832005-06-21 12:43:18 -07001587
Stephen Hemminger9195bef2008-01-22 21:56:34 -08001588 pr_debug("entering trie_leaf_remove(%p)\n", l);
Robert Olsson19baf832005-06-21 12:43:18 -07001589
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001590 if (tp) {
Stephen Hemminger9195bef2008-01-22 21:56:34 -08001591 t_key cindex = tkey_extract_bits(l->key, tp->pos, tp->bits);
Robert Olsson19baf832005-06-21 12:43:18 -07001592 put_child(t, (struct tnode *)tp, cindex, NULL);
Jarek Poplawski7b855762009-06-18 00:28:51 -07001593 trie_rebalance(t, tp);
Olof Johansson91b9a272005-08-09 20:24:39 -07001594 } else
Robert Olsson2373ce12005-08-25 13:01:29 -07001595 rcu_assign_pointer(t->trie, NULL);
Robert Olsson19baf832005-06-21 12:43:18 -07001596
Stephen Hemminger387a5482008-04-10 03:47:34 -07001597 free_leaf(l);
Robert Olsson19baf832005-06-21 12:43:18 -07001598}
1599
Robert Olssond562f1f2007-03-26 14:22:22 -07001600/*
1601 * Caller must hold RTNL.
1602 */
Stephen Hemminger16c6cf82009-09-20 10:35:36 +00001603int fib_table_delete(struct fib_table *tb, struct fib_config *cfg)
Robert Olsson19baf832005-06-21 12:43:18 -07001604{
1605 struct trie *t = (struct trie *) tb->tb_data;
1606 u32 key, mask;
Thomas Graf4e902c52006-08-17 18:14:52 -07001607 int plen = cfg->fc_dst_len;
1608 u8 tos = cfg->fc_tos;
Robert Olsson19baf832005-06-21 12:43:18 -07001609 struct fib_alias *fa, *fa_to_delete;
1610 struct list_head *fa_head;
1611 struct leaf *l;
Olof Johansson91b9a272005-08-09 20:24:39 -07001612 struct leaf_info *li;
1613
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001614 if (plen > 32)
Robert Olsson19baf832005-06-21 12:43:18 -07001615 return -EINVAL;
1616
Thomas Graf4e902c52006-08-17 18:14:52 -07001617 key = ntohl(cfg->fc_dst);
Olof Johansson91b9a272005-08-09 20:24:39 -07001618 mask = ntohl(inet_make_mask(plen));
Robert Olsson19baf832005-06-21 12:43:18 -07001619
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001620 if (key & ~mask)
Robert Olsson19baf832005-06-21 12:43:18 -07001621 return -EINVAL;
1622
1623 key = key & mask;
1624 l = fib_find_node(t, key);
1625
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001626 if (!l)
Robert Olsson19baf832005-06-21 12:43:18 -07001627 return -ESRCH;
1628
1629 fa_head = get_fa_head(l, plen);
1630 fa = fib_find_alias(fa_head, tos, 0);
1631
1632 if (!fa)
1633 return -ESRCH;
1634
Stephen Hemminger0c7770c2005-08-23 21:59:41 -07001635 pr_debug("Deleting %08x/%d tos=%d t=%p\n", key, plen, tos, t);
Robert Olsson19baf832005-06-21 12:43:18 -07001636
1637 fa_to_delete = NULL;
Julian Anastasov936f6f82008-01-28 21:18:06 -08001638 fa = list_entry(fa->fa_list.prev, struct fib_alias, fa_list);
1639 list_for_each_entry_continue(fa, fa_head, fa_list) {
Robert Olsson19baf832005-06-21 12:43:18 -07001640 struct fib_info *fi = fa->fa_info;
1641
1642 if (fa->fa_tos != tos)
1643 break;
1644
Thomas Graf4e902c52006-08-17 18:14:52 -07001645 if ((!cfg->fc_type || fa->fa_type == cfg->fc_type) &&
1646 (cfg->fc_scope == RT_SCOPE_NOWHERE ||
1647 fa->fa_scope == cfg->fc_scope) &&
1648 (!cfg->fc_protocol ||
1649 fi->fib_protocol == cfg->fc_protocol) &&
1650 fib_nh_match(cfg, fi) == 0) {
Robert Olsson19baf832005-06-21 12:43:18 -07001651 fa_to_delete = fa;
1652 break;
1653 }
1654 }
1655
Olof Johansson91b9a272005-08-09 20:24:39 -07001656 if (!fa_to_delete)
1657 return -ESRCH;
Robert Olsson19baf832005-06-21 12:43:18 -07001658
Olof Johansson91b9a272005-08-09 20:24:39 -07001659 fa = fa_to_delete;
Thomas Graf4e902c52006-08-17 18:14:52 -07001660 rtmsg_fib(RTM_DELROUTE, htonl(key), fa, plen, tb->tb_id,
Milan Kocianb8f55832007-05-23 14:55:06 -07001661 &cfg->fc_nlinfo, 0);
Robert Olsson19baf832005-06-21 12:43:18 -07001662
Olof Johansson91b9a272005-08-09 20:24:39 -07001663 l = fib_find_node(t, key);
Robert Olsson772cb712005-09-19 15:31:18 -07001664 li = find_leaf_info(l, plen);
Robert Olsson19baf832005-06-21 12:43:18 -07001665
Robert Olsson2373ce12005-08-25 13:01:29 -07001666 list_del_rcu(&fa->fa_list);
Robert Olsson19baf832005-06-21 12:43:18 -07001667
Olof Johansson91b9a272005-08-09 20:24:39 -07001668 if (list_empty(fa_head)) {
Robert Olsson2373ce12005-08-25 13:01:29 -07001669 hlist_del_rcu(&li->hlist);
Olof Johansson91b9a272005-08-09 20:24:39 -07001670 free_leaf_info(li);
Robert Olsson2373ce12005-08-25 13:01:29 -07001671 }
Olof Johansson91b9a272005-08-09 20:24:39 -07001672
1673 if (hlist_empty(&l->list))
Stephen Hemminger9195bef2008-01-22 21:56:34 -08001674 trie_leaf_remove(t, l);
Olof Johansson91b9a272005-08-09 20:24:39 -07001675
1676 if (fa->fa_state & FA_S_ACCESSED)
Denis V. Lunev76e6ebf2008-07-05 19:00:44 -07001677 rt_cache_flush(cfg->fc_nlinfo.nl_net, -1);
Olof Johansson91b9a272005-08-09 20:24:39 -07001678
Robert Olsson2373ce12005-08-25 13:01:29 -07001679 fib_release_info(fa->fa_info);
1680 alias_free_mem_rcu(fa);
Olof Johansson91b9a272005-08-09 20:24:39 -07001681 return 0;
Robert Olsson19baf832005-06-21 12:43:18 -07001682}
1683
Stephen Hemmingeref3660c2008-04-10 03:46:12 -07001684static int trie_flush_list(struct list_head *head)
Robert Olsson19baf832005-06-21 12:43:18 -07001685{
1686 struct fib_alias *fa, *fa_node;
1687 int found = 0;
1688
1689 list_for_each_entry_safe(fa, fa_node, head, fa_list) {
1690 struct fib_info *fi = fa->fa_info;
Robert Olsson19baf832005-06-21 12:43:18 -07001691
Robert Olsson2373ce12005-08-25 13:01:29 -07001692 if (fi && (fi->fib_flags & RTNH_F_DEAD)) {
1693 list_del_rcu(&fa->fa_list);
1694 fib_release_info(fa->fa_info);
1695 alias_free_mem_rcu(fa);
Robert Olsson19baf832005-06-21 12:43:18 -07001696 found++;
1697 }
1698 }
1699 return found;
1700}
1701
Stephen Hemmingeref3660c2008-04-10 03:46:12 -07001702static int trie_flush_leaf(struct leaf *l)
Robert Olsson19baf832005-06-21 12:43:18 -07001703{
1704 int found = 0;
1705 struct hlist_head *lih = &l->list;
1706 struct hlist_node *node, *tmp;
1707 struct leaf_info *li = NULL;
1708
1709 hlist_for_each_entry_safe(li, node, tmp, lih, hlist) {
Stephen Hemmingeref3660c2008-04-10 03:46:12 -07001710 found += trie_flush_list(&li->falh);
Robert Olsson19baf832005-06-21 12:43:18 -07001711
1712 if (list_empty(&li->falh)) {
Robert Olsson2373ce12005-08-25 13:01:29 -07001713 hlist_del_rcu(&li->hlist);
Robert Olsson19baf832005-06-21 12:43:18 -07001714 free_leaf_info(li);
1715 }
1716 }
1717 return found;
1718}
1719
Stephen Hemminger82cfbb02008-01-22 21:55:32 -08001720/*
1721 * Scan for the next right leaf starting at node p->child[idx]
1722 * Since we have back pointer, no recursion necessary.
1723 */
1724static struct leaf *leaf_walk_rcu(struct tnode *p, struct node *c)
Robert Olsson19baf832005-06-21 12:43:18 -07001725{
Stephen Hemminger82cfbb02008-01-22 21:55:32 -08001726 do {
1727 t_key idx;
Robert Olsson19baf832005-06-21 12:43:18 -07001728
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001729 if (c)
Stephen Hemminger82cfbb02008-01-22 21:55:32 -08001730 idx = tkey_extract_bits(c->key, p->pos, p->bits) + 1;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001731 else
Stephen Hemminger82cfbb02008-01-22 21:55:32 -08001732 idx = 0;
Robert Olsson19baf832005-06-21 12:43:18 -07001733
Stephen Hemminger82cfbb02008-01-22 21:55:32 -08001734 while (idx < 1u << p->bits) {
1735 c = tnode_get_child_rcu(p, idx++);
Robert Olsson2373ce12005-08-25 13:01:29 -07001736 if (!c)
Olof Johansson91b9a272005-08-09 20:24:39 -07001737 continue;
Robert Olsson19baf832005-06-21 12:43:18 -07001738
Stephen Hemminger82cfbb02008-01-22 21:55:32 -08001739 if (IS_LEAF(c)) {
1740 prefetch(p->child[idx]);
1741 return (struct leaf *) c;
Robert Olsson19baf832005-06-21 12:43:18 -07001742 }
Stephen Hemminger82cfbb02008-01-22 21:55:32 -08001743
1744 /* Rescan start scanning in new node */
1745 p = (struct tnode *) c;
1746 idx = 0;
Robert Olsson19baf832005-06-21 12:43:18 -07001747 }
Stephen Hemminger82cfbb02008-01-22 21:55:32 -08001748
1749 /* Node empty, walk back up to parent */
Olof Johansson91b9a272005-08-09 20:24:39 -07001750 c = (struct node *) p;
Stephen Hemminger82cfbb02008-01-22 21:55:32 -08001751 } while ( (p = node_parent_rcu(c)) != NULL);
1752
1753 return NULL; /* Root of trie */
1754}
1755
Stephen Hemminger82cfbb02008-01-22 21:55:32 -08001756static struct leaf *trie_firstleaf(struct trie *t)
1757{
1758 struct tnode *n = (struct tnode *) rcu_dereference(t->trie);
1759
1760 if (!n)
1761 return NULL;
1762
1763 if (IS_LEAF(n)) /* trie is just a leaf */
1764 return (struct leaf *) n;
1765
1766 return leaf_walk_rcu(n, NULL);
1767}
1768
1769static struct leaf *trie_nextleaf(struct leaf *l)
1770{
1771 struct node *c = (struct node *) l;
Jarek Poplawskib902e572009-07-14 11:20:32 +00001772 struct tnode *p = node_parent_rcu(c);
Stephen Hemminger82cfbb02008-01-22 21:55:32 -08001773
1774 if (!p)
1775 return NULL; /* trie with just one leaf */
1776
1777 return leaf_walk_rcu(p, c);
Robert Olsson19baf832005-06-21 12:43:18 -07001778}
1779
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001780static struct leaf *trie_leafindex(struct trie *t, int index)
1781{
1782 struct leaf *l = trie_firstleaf(t);
1783
Stephen Hemmingerec28cf72008-02-11 21:12:49 -08001784 while (l && index-- > 0)
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001785 l = trie_nextleaf(l);
Stephen Hemmingerec28cf72008-02-11 21:12:49 -08001786
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001787 return l;
1788}
1789
1790
Robert Olssond562f1f2007-03-26 14:22:22 -07001791/*
1792 * Caller must hold RTNL.
1793 */
Stephen Hemminger16c6cf82009-09-20 10:35:36 +00001794int fib_table_flush(struct fib_table *tb)
Robert Olsson19baf832005-06-21 12:43:18 -07001795{
1796 struct trie *t = (struct trie *) tb->tb_data;
Stephen Hemminger9195bef2008-01-22 21:56:34 -08001797 struct leaf *l, *ll = NULL;
Stephen Hemminger82cfbb02008-01-22 21:55:32 -08001798 int found = 0;
Robert Olsson19baf832005-06-21 12:43:18 -07001799
Stephen Hemminger82cfbb02008-01-22 21:55:32 -08001800 for (l = trie_firstleaf(t); l; l = trie_nextleaf(l)) {
Stephen Hemmingeref3660c2008-04-10 03:46:12 -07001801 found += trie_flush_leaf(l);
Robert Olsson19baf832005-06-21 12:43:18 -07001802
1803 if (ll && hlist_empty(&ll->list))
Stephen Hemminger9195bef2008-01-22 21:56:34 -08001804 trie_leaf_remove(t, ll);
Robert Olsson19baf832005-06-21 12:43:18 -07001805 ll = l;
1806 }
1807
1808 if (ll && hlist_empty(&ll->list))
Stephen Hemminger9195bef2008-01-22 21:56:34 -08001809 trie_leaf_remove(t, ll);
Robert Olsson19baf832005-06-21 12:43:18 -07001810
Stephen Hemminger0c7770c2005-08-23 21:59:41 -07001811 pr_debug("trie_flush found=%d\n", found);
Robert Olsson19baf832005-06-21 12:43:18 -07001812 return found;
1813}
1814
Stephen Hemminger16c6cf82009-09-20 10:35:36 +00001815void fib_table_select_default(struct fib_table *tb,
1816 const struct flowi *flp,
1817 struct fib_result *res)
Robert Olsson19baf832005-06-21 12:43:18 -07001818{
1819 struct trie *t = (struct trie *) tb->tb_data;
1820 int order, last_idx;
1821 struct fib_info *fi = NULL;
1822 struct fib_info *last_resort;
1823 struct fib_alias *fa = NULL;
1824 struct list_head *fa_head;
1825 struct leaf *l;
1826
1827 last_idx = -1;
1828 last_resort = NULL;
1829 order = -1;
1830
Robert Olsson2373ce12005-08-25 13:01:29 -07001831 rcu_read_lock();
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001832
Robert Olsson19baf832005-06-21 12:43:18 -07001833 l = fib_find_node(t, 0);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001834 if (!l)
Robert Olsson19baf832005-06-21 12:43:18 -07001835 goto out;
1836
1837 fa_head = get_fa_head(l, 0);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001838 if (!fa_head)
Robert Olsson19baf832005-06-21 12:43:18 -07001839 goto out;
1840
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001841 if (list_empty(fa_head))
Robert Olsson19baf832005-06-21 12:43:18 -07001842 goto out;
1843
Robert Olsson2373ce12005-08-25 13:01:29 -07001844 list_for_each_entry_rcu(fa, fa_head, fa_list) {
Robert Olsson19baf832005-06-21 12:43:18 -07001845 struct fib_info *next_fi = fa->fa_info;
Olof Johansson91b9a272005-08-09 20:24:39 -07001846
Robert Olsson19baf832005-06-21 12:43:18 -07001847 if (fa->fa_scope != res->scope ||
1848 fa->fa_type != RTN_UNICAST)
1849 continue;
Olof Johansson91b9a272005-08-09 20:24:39 -07001850
Robert Olsson19baf832005-06-21 12:43:18 -07001851 if (next_fi->fib_priority > res->fi->fib_priority)
1852 break;
1853 if (!next_fi->fib_nh[0].nh_gw ||
1854 next_fi->fib_nh[0].nh_scope != RT_SCOPE_LINK)
1855 continue;
1856 fa->fa_state |= FA_S_ACCESSED;
Olof Johansson91b9a272005-08-09 20:24:39 -07001857
Robert Olsson19baf832005-06-21 12:43:18 -07001858 if (fi == NULL) {
1859 if (next_fi != res->fi)
1860 break;
1861 } else if (!fib_detect_death(fi, order, &last_resort,
Denis V. Lunev971b8932007-12-08 00:32:23 -08001862 &last_idx, tb->tb_default)) {
Denis V. Luneva2bbe682007-12-08 00:31:44 -08001863 fib_result_assign(res, fi);
Denis V. Lunev971b8932007-12-08 00:32:23 -08001864 tb->tb_default = order;
Robert Olsson19baf832005-06-21 12:43:18 -07001865 goto out;
1866 }
1867 fi = next_fi;
1868 order++;
1869 }
1870 if (order <= 0 || fi == NULL) {
Denis V. Lunev971b8932007-12-08 00:32:23 -08001871 tb->tb_default = -1;
Robert Olsson19baf832005-06-21 12:43:18 -07001872 goto out;
1873 }
1874
Denis V. Lunev971b8932007-12-08 00:32:23 -08001875 if (!fib_detect_death(fi, order, &last_resort, &last_idx,
1876 tb->tb_default)) {
Denis V. Luneva2bbe682007-12-08 00:31:44 -08001877 fib_result_assign(res, fi);
Denis V. Lunev971b8932007-12-08 00:32:23 -08001878 tb->tb_default = order;
Robert Olsson19baf832005-06-21 12:43:18 -07001879 goto out;
1880 }
Denis V. Luneva2bbe682007-12-08 00:31:44 -08001881 if (last_idx >= 0)
1882 fib_result_assign(res, last_resort);
Denis V. Lunev971b8932007-12-08 00:32:23 -08001883 tb->tb_default = last_idx;
1884out:
Robert Olsson2373ce12005-08-25 13:01:29 -07001885 rcu_read_unlock();
Robert Olsson19baf832005-06-21 12:43:18 -07001886}
1887
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001888static int fn_trie_dump_fa(t_key key, int plen, struct list_head *fah,
1889 struct fib_table *tb,
Robert Olsson19baf832005-06-21 12:43:18 -07001890 struct sk_buff *skb, struct netlink_callback *cb)
1891{
1892 int i, s_i;
1893 struct fib_alias *fa;
Al Viro32ab5f82006-09-26 22:21:45 -07001894 __be32 xkey = htonl(key);
Robert Olsson19baf832005-06-21 12:43:18 -07001895
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001896 s_i = cb->args[5];
Robert Olsson19baf832005-06-21 12:43:18 -07001897 i = 0;
1898
Robert Olsson2373ce12005-08-25 13:01:29 -07001899 /* rcu_read_lock is hold by caller */
1900
1901 list_for_each_entry_rcu(fa, fah, fa_list) {
Robert Olsson19baf832005-06-21 12:43:18 -07001902 if (i < s_i) {
1903 i++;
1904 continue;
1905 }
Robert Olsson19baf832005-06-21 12:43:18 -07001906
1907 if (fib_dump_info(skb, NETLINK_CB(cb->skb).pid,
1908 cb->nlh->nlmsg_seq,
1909 RTM_NEWROUTE,
1910 tb->tb_id,
1911 fa->fa_type,
1912 fa->fa_scope,
Thomas Grafbe403ea2006-08-17 18:15:17 -07001913 xkey,
Robert Olsson19baf832005-06-21 12:43:18 -07001914 plen,
1915 fa->fa_tos,
Stephen Hemminger64347f72008-01-22 21:55:01 -08001916 fa->fa_info, NLM_F_MULTI) < 0) {
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001917 cb->args[5] = i;
Robert Olsson19baf832005-06-21 12:43:18 -07001918 return -1;
Olof Johansson91b9a272005-08-09 20:24:39 -07001919 }
Robert Olsson19baf832005-06-21 12:43:18 -07001920 i++;
1921 }
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001922 cb->args[5] = i;
Robert Olsson19baf832005-06-21 12:43:18 -07001923 return skb->len;
1924}
1925
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001926static int fn_trie_dump_leaf(struct leaf *l, struct fib_table *tb,
1927 struct sk_buff *skb, struct netlink_callback *cb)
Robert Olsson19baf832005-06-21 12:43:18 -07001928{
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001929 struct leaf_info *li;
1930 struct hlist_node *node;
1931 int i, s_i;
Robert Olsson19baf832005-06-21 12:43:18 -07001932
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001933 s_i = cb->args[4];
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001934 i = 0;
Robert Olsson19baf832005-06-21 12:43:18 -07001935
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001936 /* rcu_read_lock is hold by caller */
1937 hlist_for_each_entry_rcu(li, node, &l->list, hlist) {
1938 if (i < s_i) {
1939 i++;
Robert Olsson19baf832005-06-21 12:43:18 -07001940 continue;
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001941 }
Robert Olsson19baf832005-06-21 12:43:18 -07001942
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001943 if (i > s_i)
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001944 cb->args[5] = 0;
Olof Johansson91b9a272005-08-09 20:24:39 -07001945
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001946 if (list_empty(&li->falh))
Robert Olsson19baf832005-06-21 12:43:18 -07001947 continue;
1948
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001949 if (fn_trie_dump_fa(l->key, li->plen, &li->falh, tb, skb, cb) < 0) {
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001950 cb->args[4] = i;
Robert Olsson19baf832005-06-21 12:43:18 -07001951 return -1;
1952 }
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001953 i++;
Robert Olsson19baf832005-06-21 12:43:18 -07001954 }
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001955
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001956 cb->args[4] = i;
Robert Olsson19baf832005-06-21 12:43:18 -07001957 return skb->len;
1958}
1959
Stephen Hemminger16c6cf82009-09-20 10:35:36 +00001960int fib_table_dump(struct fib_table *tb, struct sk_buff *skb,
1961 struct netlink_callback *cb)
Robert Olsson19baf832005-06-21 12:43:18 -07001962{
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001963 struct leaf *l;
Robert Olsson19baf832005-06-21 12:43:18 -07001964 struct trie *t = (struct trie *) tb->tb_data;
Stephen Hemmingerd5ce8a02008-01-22 21:57:22 -08001965 t_key key = cb->args[2];
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001966 int count = cb->args[3];
Robert Olsson19baf832005-06-21 12:43:18 -07001967
Robert Olsson2373ce12005-08-25 13:01:29 -07001968 rcu_read_lock();
Stephen Hemmingerd5ce8a02008-01-22 21:57:22 -08001969 /* Dump starting at last key.
1970 * Note: 0.0.0.0/0 (ie default) is first key.
1971 */
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001972 if (count == 0)
Stephen Hemmingerd5ce8a02008-01-22 21:57:22 -08001973 l = trie_firstleaf(t);
1974 else {
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001975 /* Normally, continue from last key, but if that is missing
1976 * fallback to using slow rescan
1977 */
Stephen Hemmingerd5ce8a02008-01-22 21:57:22 -08001978 l = fib_find_node(t, key);
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001979 if (!l)
1980 l = trie_leafindex(t, count);
Stephen Hemmingerd5ce8a02008-01-22 21:57:22 -08001981 }
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001982
Stephen Hemmingerd5ce8a02008-01-22 21:57:22 -08001983 while (l) {
1984 cb->args[2] = l->key;
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001985 if (fn_trie_dump_leaf(l, tb, skb, cb) < 0) {
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001986 cb->args[3] = count;
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001987 rcu_read_unlock();
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001988 return -1;
Robert Olsson19baf832005-06-21 12:43:18 -07001989 }
Stephen Hemmingerd5ce8a02008-01-22 21:57:22 -08001990
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001991 ++count;
Stephen Hemmingerd5ce8a02008-01-22 21:57:22 -08001992 l = trie_nextleaf(l);
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001993 memset(&cb->args[4], 0,
1994 sizeof(cb->args) - 4*sizeof(cb->args[0]));
Robert Olsson19baf832005-06-21 12:43:18 -07001995 }
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001996 cb->args[3] = count;
Robert Olsson2373ce12005-08-25 13:01:29 -07001997 rcu_read_unlock();
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001998
Robert Olsson19baf832005-06-21 12:43:18 -07001999 return skb->len;
Robert Olsson19baf832005-06-21 12:43:18 -07002000}
2001
Stephen Hemminger7f9b8052008-01-14 23:14:20 -08002002void __init fib_hash_init(void)
2003{
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08002004 fn_alias_kmem = kmem_cache_create("ip_fib_alias",
2005 sizeof(struct fib_alias),
Stephen Hemmingerbc3c8c12008-01-22 21:51:50 -08002006 0, SLAB_PANIC, NULL);
2007
2008 trie_leaf_kmem = kmem_cache_create("ip_fib_trie",
2009 max(sizeof(struct leaf),
2010 sizeof(struct leaf_info)),
2011 0, SLAB_PANIC, NULL);
Stephen Hemminger7f9b8052008-01-14 23:14:20 -08002012}
Robert Olsson19baf832005-06-21 12:43:18 -07002013
Stephen Hemminger7f9b8052008-01-14 23:14:20 -08002014
2015/* Fix more generic FIB names for init later */
2016struct fib_table *fib_hash_table(u32 id)
Robert Olsson19baf832005-06-21 12:43:18 -07002017{
2018 struct fib_table *tb;
2019 struct trie *t;
2020
Robert Olsson19baf832005-06-21 12:43:18 -07002021 tb = kmalloc(sizeof(struct fib_table) + sizeof(struct trie),
2022 GFP_KERNEL);
2023 if (tb == NULL)
2024 return NULL;
2025
2026 tb->tb_id = id;
Denis V. Lunev971b8932007-12-08 00:32:23 -08002027 tb->tb_default = -1;
Robert Olsson19baf832005-06-21 12:43:18 -07002028
2029 t = (struct trie *) tb->tb_data;
Stephen Hemmingerc28a1cf2008-01-12 20:49:13 -08002030 memset(t, 0, sizeof(*t));
Robert Olsson19baf832005-06-21 12:43:18 -07002031
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07002032 if (id == RT_TABLE_LOCAL)
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08002033 pr_info("IPv4 FIB: Using LC-trie version %s\n", VERSION);
Robert Olsson19baf832005-06-21 12:43:18 -07002034
2035 return tb;
2036}
2037
Robert Olsson19baf832005-06-21 12:43:18 -07002038#ifdef CONFIG_PROC_FS
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002039/* Depth first Trie walk iterator */
2040struct fib_trie_iter {
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002041 struct seq_net_private p;
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002042 struct fib_table *tb;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002043 struct tnode *tnode;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002044 unsigned index;
2045 unsigned depth;
2046};
Robert Olsson19baf832005-06-21 12:43:18 -07002047
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002048static struct node *fib_trie_get_next(struct fib_trie_iter *iter)
Robert Olsson19baf832005-06-21 12:43:18 -07002049{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002050 struct tnode *tn = iter->tnode;
2051 unsigned cindex = iter->index;
2052 struct tnode *p;
2053
Eric W. Biederman6640e692007-01-24 14:42:04 -08002054 /* A single entry routing table */
2055 if (!tn)
2056 return NULL;
2057
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002058 pr_debug("get_next iter={node=%p index=%d depth=%d}\n",
2059 iter->tnode, iter->index, iter->depth);
2060rescan:
2061 while (cindex < (1<<tn->bits)) {
Eric Dumazetb59cfbf2008-01-18 03:31:36 -08002062 struct node *n = tnode_get_child_rcu(tn, cindex);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002063
2064 if (n) {
2065 if (IS_LEAF(n)) {
2066 iter->tnode = tn;
2067 iter->index = cindex + 1;
2068 } else {
2069 /* push down one level */
2070 iter->tnode = (struct tnode *) n;
2071 iter->index = 0;
2072 ++iter->depth;
2073 }
2074 return n;
2075 }
2076
2077 ++cindex;
2078 }
2079
2080 /* Current node exhausted, pop back up */
Eric Dumazetb59cfbf2008-01-18 03:31:36 -08002081 p = node_parent_rcu((struct node *)tn);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002082 if (p) {
2083 cindex = tkey_extract_bits(tn->key, p->pos, p->bits)+1;
2084 tn = p;
2085 --iter->depth;
2086 goto rescan;
2087 }
2088
2089 /* got root? */
Robert Olsson19baf832005-06-21 12:43:18 -07002090 return NULL;
2091}
2092
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002093static struct node *fib_trie_get_first(struct fib_trie_iter *iter,
2094 struct trie *t)
Robert Olsson19baf832005-06-21 12:43:18 -07002095{
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002096 struct node *n;
Robert Olsson5ddf0eb2006-03-20 21:34:12 -08002097
Stephen Hemminger132adf52007-03-08 20:44:43 -08002098 if (!t)
Robert Olsson5ddf0eb2006-03-20 21:34:12 -08002099 return NULL;
2100
2101 n = rcu_dereference(t->trie);
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002102 if (!n)
Robert Olsson5ddf0eb2006-03-20 21:34:12 -08002103 return NULL;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002104
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002105 if (IS_TNODE(n)) {
2106 iter->tnode = (struct tnode *) n;
2107 iter->index = 0;
2108 iter->depth = 1;
2109 } else {
2110 iter->tnode = NULL;
2111 iter->index = 0;
2112 iter->depth = 0;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002113 }
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002114
2115 return n;
Robert Olsson19baf832005-06-21 12:43:18 -07002116}
2117
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002118static void trie_collect_stats(struct trie *t, struct trie_stat *s)
Robert Olsson19baf832005-06-21 12:43:18 -07002119{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002120 struct node *n;
2121 struct fib_trie_iter iter;
Robert Olsson19baf832005-06-21 12:43:18 -07002122
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002123 memset(s, 0, sizeof(*s));
Robert Olsson19baf832005-06-21 12:43:18 -07002124
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002125 rcu_read_lock();
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002126 for (n = fib_trie_get_first(&iter, t); n; n = fib_trie_get_next(&iter)) {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002127 if (IS_LEAF(n)) {
Stephen Hemminger93672292008-01-22 21:54:05 -08002128 struct leaf *l = (struct leaf *)n;
2129 struct leaf_info *li;
2130 struct hlist_node *tmp;
2131
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002132 s->leaves++;
2133 s->totdepth += iter.depth;
2134 if (iter.depth > s->maxdepth)
2135 s->maxdepth = iter.depth;
Stephen Hemminger93672292008-01-22 21:54:05 -08002136
2137 hlist_for_each_entry_rcu(li, tmp, &l->list, hlist)
2138 ++s->prefixes;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002139 } else {
2140 const struct tnode *tn = (const struct tnode *) n;
2141 int i;
Robert Olsson19baf832005-06-21 12:43:18 -07002142
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002143 s->tnodes++;
Stephen Hemminger132adf52007-03-08 20:44:43 -08002144 if (tn->bits < MAX_STAT_DEPTH)
Robert Olsson06ef9212006-03-20 21:35:01 -08002145 s->nodesizes[tn->bits]++;
2146
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002147 for (i = 0; i < (1<<tn->bits); i++)
2148 if (!tn->child[i])
2149 s->nullpointers++;
2150 }
2151 }
2152 rcu_read_unlock();
Robert Olsson19baf832005-06-21 12:43:18 -07002153}
2154
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07002155/*
Robert Olsson19baf832005-06-21 12:43:18 -07002156 * This outputs /proc/net/fib_triestats
Robert Olsson19baf832005-06-21 12:43:18 -07002157 */
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002158static void trie_show_stats(struct seq_file *seq, struct trie_stat *stat)
Robert Olsson19baf832005-06-21 12:43:18 -07002159{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002160 unsigned i, max, pointers, bytes, avdepth;
Robert Olsson19baf832005-06-21 12:43:18 -07002161
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002162 if (stat->leaves)
2163 avdepth = stat->totdepth*100 / stat->leaves;
2164 else
2165 avdepth = 0;
Robert Olsson19baf832005-06-21 12:43:18 -07002166
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08002167 seq_printf(seq, "\tAver depth: %u.%02d\n",
2168 avdepth / 100, avdepth % 100);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002169 seq_printf(seq, "\tMax depth: %u\n", stat->maxdepth);
Robert Olsson19baf832005-06-21 12:43:18 -07002170
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002171 seq_printf(seq, "\tLeaves: %u\n", stat->leaves);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002172 bytes = sizeof(struct leaf) * stat->leaves;
Stephen Hemminger93672292008-01-22 21:54:05 -08002173
2174 seq_printf(seq, "\tPrefixes: %u\n", stat->prefixes);
2175 bytes += sizeof(struct leaf_info) * stat->prefixes;
2176
Stephen Hemminger187b5182008-01-12 20:55:55 -08002177 seq_printf(seq, "\tInternal nodes: %u\n\t", stat->tnodes);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002178 bytes += sizeof(struct tnode) * stat->tnodes;
Robert Olsson19baf832005-06-21 12:43:18 -07002179
Robert Olsson06ef9212006-03-20 21:35:01 -08002180 max = MAX_STAT_DEPTH;
2181 while (max > 0 && stat->nodesizes[max-1] == 0)
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002182 max--;
Robert Olsson19baf832005-06-21 12:43:18 -07002183
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002184 pointers = 0;
2185 for (i = 1; i <= max; i++)
2186 if (stat->nodesizes[i] != 0) {
Stephen Hemminger187b5182008-01-12 20:55:55 -08002187 seq_printf(seq, " %u: %u", i, stat->nodesizes[i]);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002188 pointers += (1<<i) * stat->nodesizes[i];
2189 }
2190 seq_putc(seq, '\n');
Stephen Hemminger187b5182008-01-12 20:55:55 -08002191 seq_printf(seq, "\tPointers: %u\n", pointers);
Robert Olsson19baf832005-06-21 12:43:18 -07002192
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002193 bytes += sizeof(struct node *) * pointers;
Stephen Hemminger187b5182008-01-12 20:55:55 -08002194 seq_printf(seq, "Null ptrs: %u\n", stat->nullpointers);
2195 seq_printf(seq, "Total size: %u kB\n", (bytes + 1023) / 1024);
Stephen Hemminger66a2f7f2008-01-12 21:23:17 -08002196}
Robert Olsson19baf832005-06-21 12:43:18 -07002197
2198#ifdef CONFIG_IP_FIB_TRIE_STATS
Stephen Hemminger66a2f7f2008-01-12 21:23:17 -08002199static void trie_show_usage(struct seq_file *seq,
2200 const struct trie_use_stats *stats)
2201{
2202 seq_printf(seq, "\nCounters:\n---------\n");
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08002203 seq_printf(seq, "gets = %u\n", stats->gets);
2204 seq_printf(seq, "backtracks = %u\n", stats->backtrack);
2205 seq_printf(seq, "semantic match passed = %u\n",
2206 stats->semantic_match_passed);
2207 seq_printf(seq, "semantic match miss = %u\n",
2208 stats->semantic_match_miss);
2209 seq_printf(seq, "null node hit= %u\n", stats->null_node_hit);
2210 seq_printf(seq, "skipped node resize = %u\n\n",
2211 stats->resize_node_skipped);
Robert Olsson19baf832005-06-21 12:43:18 -07002212}
Stephen Hemminger66a2f7f2008-01-12 21:23:17 -08002213#endif /* CONFIG_IP_FIB_TRIE_STATS */
2214
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002215static void fib_table_print(struct seq_file *seq, struct fib_table *tb)
Stephen Hemmingerd717a9a2008-01-14 23:11:54 -08002216{
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002217 if (tb->tb_id == RT_TABLE_LOCAL)
2218 seq_puts(seq, "Local:\n");
2219 else if (tb->tb_id == RT_TABLE_MAIN)
2220 seq_puts(seq, "Main:\n");
2221 else
2222 seq_printf(seq, "Id %d:\n", tb->tb_id);
Stephen Hemmingerd717a9a2008-01-14 23:11:54 -08002223}
Robert Olsson19baf832005-06-21 12:43:18 -07002224
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002225
Robert Olsson19baf832005-06-21 12:43:18 -07002226static int fib_triestat_seq_show(struct seq_file *seq, void *v)
2227{
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002228 struct net *net = (struct net *)seq->private;
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002229 unsigned int h;
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002230
Stephen Hemmingerd717a9a2008-01-14 23:11:54 -08002231 seq_printf(seq,
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08002232 "Basic info: size of leaf:"
2233 " %Zd bytes, size of tnode: %Zd bytes.\n",
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002234 sizeof(struct leaf), sizeof(struct tnode));
Olof Johansson91b9a272005-08-09 20:24:39 -07002235
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002236 for (h = 0; h < FIB_TABLE_HASHSZ; h++) {
2237 struct hlist_head *head = &net->ipv4.fib_table_hash[h];
2238 struct hlist_node *node;
2239 struct fib_table *tb;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002240
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002241 hlist_for_each_entry_rcu(tb, node, head, tb_hlist) {
2242 struct trie *t = (struct trie *) tb->tb_data;
2243 struct trie_stat stat;
2244
2245 if (!t)
2246 continue;
2247
2248 fib_table_print(seq, tb);
2249
2250 trie_collect_stats(t, &stat);
2251 trie_show_stats(seq, &stat);
2252#ifdef CONFIG_IP_FIB_TRIE_STATS
2253 trie_show_usage(seq, &t->stats);
2254#endif
2255 }
2256 }
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002257
Robert Olsson19baf832005-06-21 12:43:18 -07002258 return 0;
2259}
2260
Robert Olsson19baf832005-06-21 12:43:18 -07002261static int fib_triestat_seq_open(struct inode *inode, struct file *file)
2262{
Pavel Emelyanovde05c552008-07-18 04:07:21 -07002263 return single_open_net(inode, file, fib_triestat_seq_show);
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002264}
2265
Arjan van de Ven9a321442007-02-12 00:55:35 -08002266static const struct file_operations fib_triestat_fops = {
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07002267 .owner = THIS_MODULE,
2268 .open = fib_triestat_seq_open,
2269 .read = seq_read,
2270 .llseek = seq_lseek,
Pavel Emelyanovb6fcbdb2008-07-18 04:07:44 -07002271 .release = single_release_net,
Robert Olsson19baf832005-06-21 12:43:18 -07002272};
2273
YOSHIFUJI Hideaki12188542008-03-26 02:36:06 +09002274static struct node *fib_trie_get_idx(struct seq_file *seq, loff_t pos)
Robert Olsson19baf832005-06-21 12:43:18 -07002275{
YOSHIFUJI Hideaki12188542008-03-26 02:36:06 +09002276 struct fib_trie_iter *iter = seq->private;
2277 struct net *net = seq_file_net(seq);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002278 loff_t idx = 0;
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002279 unsigned int h;
Robert Olsson19baf832005-06-21 12:43:18 -07002280
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002281 for (h = 0; h < FIB_TABLE_HASHSZ; h++) {
2282 struct hlist_head *head = &net->ipv4.fib_table_hash[h];
2283 struct hlist_node *node;
2284 struct fib_table *tb;
2285
2286 hlist_for_each_entry_rcu(tb, node, head, tb_hlist) {
2287 struct node *n;
2288
2289 for (n = fib_trie_get_first(iter,
2290 (struct trie *) tb->tb_data);
2291 n; n = fib_trie_get_next(iter))
2292 if (pos == idx++) {
2293 iter->tb = tb;
2294 return n;
2295 }
2296 }
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002297 }
Robert Olsson19baf832005-06-21 12:43:18 -07002298
Robert Olsson19baf832005-06-21 12:43:18 -07002299 return NULL;
2300}
2301
2302static void *fib_trie_seq_start(struct seq_file *seq, loff_t *pos)
Stephen Hemmingerc95aaf92008-01-12 21:25:02 -08002303 __acquires(RCU)
Robert Olsson19baf832005-06-21 12:43:18 -07002304{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002305 rcu_read_lock();
YOSHIFUJI Hideaki12188542008-03-26 02:36:06 +09002306 return fib_trie_get_idx(seq, *pos);
Robert Olsson19baf832005-06-21 12:43:18 -07002307}
2308
2309static void *fib_trie_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2310{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002311 struct fib_trie_iter *iter = seq->private;
YOSHIFUJI Hideaki12188542008-03-26 02:36:06 +09002312 struct net *net = seq_file_net(seq);
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002313 struct fib_table *tb = iter->tb;
2314 struct hlist_node *tb_node;
2315 unsigned int h;
2316 struct node *n;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002317
Robert Olsson19baf832005-06-21 12:43:18 -07002318 ++*pos;
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002319 /* next node in same table */
2320 n = fib_trie_get_next(iter);
2321 if (n)
2322 return n;
Olof Johansson91b9a272005-08-09 20:24:39 -07002323
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002324 /* walk rest of this hash chain */
2325 h = tb->tb_id & (FIB_TABLE_HASHSZ - 1);
2326 while ( (tb_node = rcu_dereference(tb->tb_hlist.next)) ) {
2327 tb = hlist_entry(tb_node, struct fib_table, tb_hlist);
2328 n = fib_trie_get_first(iter, (struct trie *) tb->tb_data);
2329 if (n)
2330 goto found;
2331 }
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002332
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002333 /* new hash chain */
2334 while (++h < FIB_TABLE_HASHSZ) {
2335 struct hlist_head *head = &net->ipv4.fib_table_hash[h];
2336 hlist_for_each_entry_rcu(tb, tb_node, head, tb_hlist) {
2337 n = fib_trie_get_first(iter, (struct trie *) tb->tb_data);
2338 if (n)
2339 goto found;
2340 }
2341 }
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002342 return NULL;
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002343
2344found:
2345 iter->tb = tb;
2346 return n;
Robert Olsson19baf832005-06-21 12:43:18 -07002347}
2348
2349static void fib_trie_seq_stop(struct seq_file *seq, void *v)
Stephen Hemmingerc95aaf92008-01-12 21:25:02 -08002350 __releases(RCU)
Robert Olsson19baf832005-06-21 12:43:18 -07002351{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002352 rcu_read_unlock();
Robert Olsson19baf832005-06-21 12:43:18 -07002353}
2354
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002355static void seq_indent(struct seq_file *seq, int n)
2356{
2357 while (n-- > 0) seq_puts(seq, " ");
2358}
Robert Olsson19baf832005-06-21 12:43:18 -07002359
Eric Dumazet28d36e32008-01-14 23:09:56 -08002360static inline const char *rtn_scope(char *buf, size_t len, enum rt_scope_t s)
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002361{
Stephen Hemminger132adf52007-03-08 20:44:43 -08002362 switch (s) {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002363 case RT_SCOPE_UNIVERSE: return "universe";
2364 case RT_SCOPE_SITE: return "site";
2365 case RT_SCOPE_LINK: return "link";
2366 case RT_SCOPE_HOST: return "host";
2367 case RT_SCOPE_NOWHERE: return "nowhere";
2368 default:
Eric Dumazet28d36e32008-01-14 23:09:56 -08002369 snprintf(buf, len, "scope=%d", s);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002370 return buf;
2371 }
2372}
2373
Jan Engelhardt36cbd3d2009-08-05 10:42:58 -07002374static const char *const rtn_type_names[__RTN_MAX] = {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002375 [RTN_UNSPEC] = "UNSPEC",
2376 [RTN_UNICAST] = "UNICAST",
2377 [RTN_LOCAL] = "LOCAL",
2378 [RTN_BROADCAST] = "BROADCAST",
2379 [RTN_ANYCAST] = "ANYCAST",
2380 [RTN_MULTICAST] = "MULTICAST",
2381 [RTN_BLACKHOLE] = "BLACKHOLE",
2382 [RTN_UNREACHABLE] = "UNREACHABLE",
2383 [RTN_PROHIBIT] = "PROHIBIT",
2384 [RTN_THROW] = "THROW",
2385 [RTN_NAT] = "NAT",
2386 [RTN_XRESOLVE] = "XRESOLVE",
2387};
2388
Eric Dumazet28d36e32008-01-14 23:09:56 -08002389static inline const char *rtn_type(char *buf, size_t len, unsigned t)
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002390{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002391 if (t < __RTN_MAX && rtn_type_names[t])
2392 return rtn_type_names[t];
Eric Dumazet28d36e32008-01-14 23:09:56 -08002393 snprintf(buf, len, "type %u", t);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002394 return buf;
2395}
2396
2397/* Pretty print the trie */
Robert Olsson19baf832005-06-21 12:43:18 -07002398static int fib_trie_seq_show(struct seq_file *seq, void *v)
2399{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002400 const struct fib_trie_iter *iter = seq->private;
2401 struct node *n = v;
Robert Olsson19baf832005-06-21 12:43:18 -07002402
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002403 if (!node_parent_rcu(n))
2404 fib_table_print(seq, iter->tb);
Robert Olsson095b8502007-01-26 19:06:01 -08002405
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002406 if (IS_TNODE(n)) {
2407 struct tnode *tn = (struct tnode *) n;
Stephen Hemmingerab66b4a2007-08-10 15:22:58 -07002408 __be32 prf = htonl(mask_pfx(tn->key, tn->pos));
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002409
Robert Olsson1d25cd62005-09-19 15:29:52 -07002410 seq_indent(seq, iter->depth-1);
Harvey Harrison673d57e2008-10-31 00:53:57 -07002411 seq_printf(seq, " +-- %pI4/%d %d %d %d\n",
2412 &prf, tn->pos, tn->bits, tn->full_children,
Robert Olsson1d25cd62005-09-19 15:29:52 -07002413 tn->empty_children);
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +09002414
Olof Johansson91b9a272005-08-09 20:24:39 -07002415 } else {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002416 struct leaf *l = (struct leaf *) n;
Stephen Hemminger13280422008-01-22 21:54:37 -08002417 struct leaf_info *li;
2418 struct hlist_node *node;
Al Viro32ab5f82006-09-26 22:21:45 -07002419 __be32 val = htonl(l->key);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002420
2421 seq_indent(seq, iter->depth);
Harvey Harrison673d57e2008-10-31 00:53:57 -07002422 seq_printf(seq, " |-- %pI4\n", &val);
Eric Dumazet28d36e32008-01-14 23:09:56 -08002423
Stephen Hemminger13280422008-01-22 21:54:37 -08002424 hlist_for_each_entry_rcu(li, node, &l->list, hlist) {
2425 struct fib_alias *fa;
Eric Dumazet28d36e32008-01-14 23:09:56 -08002426
Stephen Hemminger13280422008-01-22 21:54:37 -08002427 list_for_each_entry_rcu(fa, &li->falh, fa_list) {
2428 char buf1[32], buf2[32];
Eric Dumazet28d36e32008-01-14 23:09:56 -08002429
Stephen Hemminger13280422008-01-22 21:54:37 -08002430 seq_indent(seq, iter->depth+1);
2431 seq_printf(seq, " /%d %s %s", li->plen,
2432 rtn_scope(buf1, sizeof(buf1),
2433 fa->fa_scope),
2434 rtn_type(buf2, sizeof(buf2),
2435 fa->fa_type));
2436 if (fa->fa_tos)
Denis V. Lunevb9c4d822008-02-05 02:58:45 -08002437 seq_printf(seq, " tos=%d", fa->fa_tos);
Stephen Hemminger13280422008-01-22 21:54:37 -08002438 seq_putc(seq, '\n');
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002439 }
2440 }
Robert Olsson19baf832005-06-21 12:43:18 -07002441 }
2442
2443 return 0;
2444}
2445
Stephen Hemmingerf6908082007-03-12 14:34:29 -07002446static const struct seq_operations fib_trie_seq_ops = {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002447 .start = fib_trie_seq_start,
2448 .next = fib_trie_seq_next,
2449 .stop = fib_trie_seq_stop,
2450 .show = fib_trie_seq_show,
Robert Olsson19baf832005-06-21 12:43:18 -07002451};
2452
2453static int fib_trie_seq_open(struct inode *inode, struct file *file)
2454{
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002455 return seq_open_net(inode, file, &fib_trie_seq_ops,
2456 sizeof(struct fib_trie_iter));
Robert Olsson19baf832005-06-21 12:43:18 -07002457}
2458
Arjan van de Ven9a321442007-02-12 00:55:35 -08002459static const struct file_operations fib_trie_fops = {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002460 .owner = THIS_MODULE,
2461 .open = fib_trie_seq_open,
2462 .read = seq_read,
2463 .llseek = seq_lseek,
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002464 .release = seq_release_net,
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002465};
2466
Stephen Hemminger8315f5d2008-02-11 21:14:39 -08002467struct fib_route_iter {
2468 struct seq_net_private p;
2469 struct trie *main_trie;
2470 loff_t pos;
2471 t_key key;
2472};
2473
2474static struct leaf *fib_route_get_idx(struct fib_route_iter *iter, loff_t pos)
2475{
2476 struct leaf *l = NULL;
2477 struct trie *t = iter->main_trie;
2478
2479 /* use cache location of last found key */
2480 if (iter->pos > 0 && pos >= iter->pos && (l = fib_find_node(t, iter->key)))
2481 pos -= iter->pos;
2482 else {
2483 iter->pos = 0;
2484 l = trie_firstleaf(t);
2485 }
2486
2487 while (l && pos-- > 0) {
2488 iter->pos++;
2489 l = trie_nextleaf(l);
2490 }
2491
2492 if (l)
2493 iter->key = pos; /* remember it */
2494 else
2495 iter->pos = 0; /* forget it */
2496
2497 return l;
2498}
2499
2500static void *fib_route_seq_start(struct seq_file *seq, loff_t *pos)
2501 __acquires(RCU)
2502{
2503 struct fib_route_iter *iter = seq->private;
2504 struct fib_table *tb;
2505
2506 rcu_read_lock();
YOSHIFUJI Hideaki12188542008-03-26 02:36:06 +09002507 tb = fib_get_table(seq_file_net(seq), RT_TABLE_MAIN);
Stephen Hemminger8315f5d2008-02-11 21:14:39 -08002508 if (!tb)
2509 return NULL;
2510
2511 iter->main_trie = (struct trie *) tb->tb_data;
2512 if (*pos == 0)
2513 return SEQ_START_TOKEN;
2514 else
2515 return fib_route_get_idx(iter, *pos - 1);
2516}
2517
2518static void *fib_route_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2519{
2520 struct fib_route_iter *iter = seq->private;
2521 struct leaf *l = v;
2522
2523 ++*pos;
2524 if (v == SEQ_START_TOKEN) {
2525 iter->pos = 0;
2526 l = trie_firstleaf(iter->main_trie);
2527 } else {
2528 iter->pos++;
2529 l = trie_nextleaf(l);
2530 }
2531
2532 if (l)
2533 iter->key = l->key;
2534 else
2535 iter->pos = 0;
2536 return l;
2537}
2538
2539static void fib_route_seq_stop(struct seq_file *seq, void *v)
2540 __releases(RCU)
2541{
2542 rcu_read_unlock();
2543}
2544
Al Viro32ab5f82006-09-26 22:21:45 -07002545static unsigned fib_flag_trans(int type, __be32 mask, const struct fib_info *fi)
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002546{
2547 static unsigned type2flags[RTN_MAX + 1] = {
2548 [7] = RTF_REJECT, [8] = RTF_REJECT,
2549 };
2550 unsigned flags = type2flags[type];
2551
2552 if (fi && fi->fib_nh->nh_gw)
2553 flags |= RTF_GATEWAY;
Al Viro32ab5f82006-09-26 22:21:45 -07002554 if (mask == htonl(0xFFFFFFFF))
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002555 flags |= RTF_HOST;
2556 flags |= RTF_UP;
2557 return flags;
2558}
2559
2560/*
2561 * This outputs /proc/net/route.
2562 * The format of the file is not supposed to be changed
2563 * and needs to be same as fib_hash output to avoid breaking
2564 * legacy utilities
2565 */
2566static int fib_route_seq_show(struct seq_file *seq, void *v)
2567{
2568 struct leaf *l = v;
Stephen Hemminger13280422008-01-22 21:54:37 -08002569 struct leaf_info *li;
2570 struct hlist_node *node;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002571
2572 if (v == SEQ_START_TOKEN) {
2573 seq_printf(seq, "%-127s\n", "Iface\tDestination\tGateway "
2574 "\tFlags\tRefCnt\tUse\tMetric\tMask\t\tMTU"
2575 "\tWindow\tIRTT");
2576 return 0;
2577 }
2578
Stephen Hemminger13280422008-01-22 21:54:37 -08002579 hlist_for_each_entry_rcu(li, node, &l->list, hlist) {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002580 struct fib_alias *fa;
Al Viro32ab5f82006-09-26 22:21:45 -07002581 __be32 mask, prefix;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002582
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002583 mask = inet_make_mask(li->plen);
2584 prefix = htonl(l->key);
2585
2586 list_for_each_entry_rcu(fa, &li->falh, fa_list) {
Herbert Xu1371e372005-10-15 09:42:39 +10002587 const struct fib_info *fi = fa->fa_info;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002588 unsigned flags = fib_flag_trans(fa->fa_type, mask, fi);
Pavel Emelyanov5e659e42008-04-24 01:02:16 -07002589 int len;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002590
2591 if (fa->fa_type == RTN_BROADCAST
2592 || fa->fa_type == RTN_MULTICAST)
2593 continue;
2594
2595 if (fi)
Pavel Emelyanov5e659e42008-04-24 01:02:16 -07002596 seq_printf(seq,
2597 "%s\t%08X\t%08X\t%04X\t%d\t%u\t"
2598 "%d\t%08X\t%d\t%u\t%u%n",
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002599 fi->fib_dev ? fi->fib_dev->name : "*",
2600 prefix,
2601 fi->fib_nh->nh_gw, flags, 0, 0,
2602 fi->fib_priority,
2603 mask,
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08002604 (fi->fib_advmss ?
2605 fi->fib_advmss + 40 : 0),
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002606 fi->fib_window,
Pavel Emelyanov5e659e42008-04-24 01:02:16 -07002607 fi->fib_rtt >> 3, &len);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002608 else
Pavel Emelyanov5e659e42008-04-24 01:02:16 -07002609 seq_printf(seq,
2610 "*\t%08X\t%08X\t%04X\t%d\t%u\t"
2611 "%d\t%08X\t%d\t%u\t%u%n",
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002612 prefix, 0, flags, 0, 0, 0,
Pavel Emelyanov5e659e42008-04-24 01:02:16 -07002613 mask, 0, 0, 0, &len);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002614
Pavel Emelyanov5e659e42008-04-24 01:02:16 -07002615 seq_printf(seq, "%*s\n", 127 - len, "");
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002616 }
2617 }
2618
2619 return 0;
2620}
2621
Stephen Hemmingerf6908082007-03-12 14:34:29 -07002622static const struct seq_operations fib_route_seq_ops = {
Stephen Hemminger8315f5d2008-02-11 21:14:39 -08002623 .start = fib_route_seq_start,
2624 .next = fib_route_seq_next,
2625 .stop = fib_route_seq_stop,
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002626 .show = fib_route_seq_show,
2627};
2628
2629static int fib_route_seq_open(struct inode *inode, struct file *file)
2630{
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002631 return seq_open_net(inode, file, &fib_route_seq_ops,
Stephen Hemminger8315f5d2008-02-11 21:14:39 -08002632 sizeof(struct fib_route_iter));
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002633}
2634
Arjan van de Ven9a321442007-02-12 00:55:35 -08002635static const struct file_operations fib_route_fops = {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002636 .owner = THIS_MODULE,
2637 .open = fib_route_seq_open,
2638 .read = seq_read,
2639 .llseek = seq_lseek,
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002640 .release = seq_release_net,
Robert Olsson19baf832005-06-21 12:43:18 -07002641};
2642
Denis V. Lunev61a02652008-01-10 03:21:09 -08002643int __net_init fib_proc_init(struct net *net)
Robert Olsson19baf832005-06-21 12:43:18 -07002644{
Denis V. Lunev61a02652008-01-10 03:21:09 -08002645 if (!proc_net_fops_create(net, "fib_trie", S_IRUGO, &fib_trie_fops))
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002646 goto out1;
2647
Denis V. Lunev61a02652008-01-10 03:21:09 -08002648 if (!proc_net_fops_create(net, "fib_triestat", S_IRUGO,
2649 &fib_triestat_fops))
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002650 goto out2;
2651
Denis V. Lunev61a02652008-01-10 03:21:09 -08002652 if (!proc_net_fops_create(net, "route", S_IRUGO, &fib_route_fops))
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002653 goto out3;
2654
Robert Olsson19baf832005-06-21 12:43:18 -07002655 return 0;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002656
2657out3:
Denis V. Lunev61a02652008-01-10 03:21:09 -08002658 proc_net_remove(net, "fib_triestat");
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002659out2:
Denis V. Lunev61a02652008-01-10 03:21:09 -08002660 proc_net_remove(net, "fib_trie");
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002661out1:
2662 return -ENOMEM;
Robert Olsson19baf832005-06-21 12:43:18 -07002663}
2664
Denis V. Lunev61a02652008-01-10 03:21:09 -08002665void __net_exit fib_proc_exit(struct net *net)
Robert Olsson19baf832005-06-21 12:43:18 -07002666{
Denis V. Lunev61a02652008-01-10 03:21:09 -08002667 proc_net_remove(net, "fib_trie");
2668 proc_net_remove(net, "fib_triestat");
2669 proc_net_remove(net, "route");
Robert Olsson19baf832005-06-21 12:43:18 -07002670}
2671
2672#endif /* CONFIG_PROC_FS */