blob: 4a8e370862bca453cd6e939162ded09de6bdf831 [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
Jarek Poplawskif6b085b2010-09-07 07:51:17 +0000189 return rcu_dereference_check(ret,
190 rcu_read_lock_held() ||
191 lockdep_rtnl_is_held());
Stephen Hemminger06801912007-08-10 15:22:13 -0700192}
193
Stephen Hemminger6440cc92008-03-22 17:59:58 -0700194/* Same as rcu_assign_pointer
195 * but that macro() assumes that value is a pointer.
196 */
Stephen Hemminger06801912007-08-10 15:22:13 -0700197static inline void node_set_parent(struct node *node, struct tnode *ptr)
198{
Stephen Hemminger6440cc92008-03-22 17:59:58 -0700199 smp_wmb();
200 node->parent = (unsigned long)ptr | NODE_TYPE(node);
Stephen Hemminger06801912007-08-10 15:22:13 -0700201}
Robert Olsson2373ce12005-08-25 13:01:29 -0700202
Eric Dumazetb59cfbf2008-01-18 03:31:36 -0800203static inline struct node *tnode_get_child(struct tnode *tn, unsigned int i)
Robert Olsson19baf832005-06-21 12:43:18 -0700204{
Eric Dumazetb59cfbf2008-01-18 03:31:36 -0800205 BUG_ON(i >= 1U << tn->bits);
Robert Olsson19baf832005-06-21 12:43:18 -0700206
Eric Dumazetb59cfbf2008-01-18 03:31:36 -0800207 return tn->child[i];
208}
209
210static inline struct node *tnode_get_child_rcu(struct tnode *tn, unsigned int i)
211{
212 struct node *ret = tnode_get_child(tn, i);
213
Eric Dumazet4eaa0e32010-04-14 16:13:29 -0700214 return rcu_dereference_check(ret,
215 rcu_read_lock_held() ||
216 lockdep_rtnl_is_held());
Robert Olsson19baf832005-06-21 12:43:18 -0700217}
218
Stephen Hemmignerbb435b82005-08-09 20:25:39 -0700219static inline int tnode_child_length(const struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700220{
Olof Johansson91b9a272005-08-09 20:24:39 -0700221 return 1 << tn->bits;
Robert Olsson19baf832005-06-21 12:43:18 -0700222}
223
Stephen Hemmingerab66b4a2007-08-10 15:22:58 -0700224static inline t_key mask_pfx(t_key k, unsigned short l)
225{
226 return (l == 0) ? 0 : k >> (KEYLENGTH-l) << (KEYLENGTH-l);
227}
228
Robert Olsson19baf832005-06-21 12:43:18 -0700229static inline t_key tkey_extract_bits(t_key a, int offset, int bits)
230{
Olof Johansson91b9a272005-08-09 20:24:39 -0700231 if (offset < KEYLENGTH)
Robert Olsson19baf832005-06-21 12:43:18 -0700232 return ((t_key)(a << offset)) >> (KEYLENGTH - bits);
Olof Johansson91b9a272005-08-09 20:24:39 -0700233 else
Robert Olsson19baf832005-06-21 12:43:18 -0700234 return 0;
235}
236
237static inline int tkey_equals(t_key a, t_key b)
238{
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700239 return a == b;
Robert Olsson19baf832005-06-21 12:43:18 -0700240}
241
242static inline int tkey_sub_equals(t_key a, int offset, int bits, t_key b)
243{
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700244 if (bits == 0 || offset >= KEYLENGTH)
245 return 1;
Olof Johansson91b9a272005-08-09 20:24:39 -0700246 bits = bits > KEYLENGTH ? KEYLENGTH : bits;
247 return ((a ^ b) << offset) >> (KEYLENGTH - bits) == 0;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700248}
Robert Olsson19baf832005-06-21 12:43:18 -0700249
250static inline int tkey_mismatch(t_key a, int offset, t_key b)
251{
252 t_key diff = a ^ b;
253 int i = offset;
254
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700255 if (!diff)
256 return 0;
257 while ((diff << i) >> (KEYLENGTH-1) == 0)
Robert Olsson19baf832005-06-21 12:43:18 -0700258 i++;
259 return i;
260}
261
Robert Olsson19baf832005-06-21 12:43:18 -0700262/*
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900263 To understand this stuff, an understanding of keys and all their bits is
264 necessary. Every node in the trie has a key associated with it, but not
Robert Olsson19baf832005-06-21 12:43:18 -0700265 all of the bits in that key are significant.
266
267 Consider a node 'n' and its parent 'tp'.
268
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900269 If n is a leaf, every bit in its key is significant. Its presence is
270 necessitated by path compression, since during a tree traversal (when
271 searching for a leaf - unless we are doing an insertion) we will completely
272 ignore all skipped bits we encounter. Thus we need to verify, at the end of
273 a potentially successful search, that we have indeed been walking the
Robert Olsson19baf832005-06-21 12:43:18 -0700274 correct key path.
275
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900276 Note that we can never "miss" the correct key in the tree if present by
277 following the wrong path. Path compression ensures that segments of the key
278 that are the same for all keys with a given prefix are skipped, but the
279 skipped part *is* identical for each node in the subtrie below the skipped
280 bit! trie_insert() in this implementation takes care of that - note the
Robert Olsson19baf832005-06-21 12:43:18 -0700281 call to tkey_sub_equals() in trie_insert().
282
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900283 if n is an internal node - a 'tnode' here, the various parts of its key
Robert Olsson19baf832005-06-21 12:43:18 -0700284 have many different meanings.
285
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900286 Example:
Robert Olsson19baf832005-06-21 12:43:18 -0700287 _________________________________________________________________
288 | i | i | i | i | i | i | i | N | N | N | S | S | S | S | S | C |
289 -----------------------------------------------------------------
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900290 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
Robert Olsson19baf832005-06-21 12:43:18 -0700291
292 _________________________________________________________________
293 | C | C | C | u | u | u | u | u | u | u | u | u | u | u | u | u |
294 -----------------------------------------------------------------
295 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
296
297 tp->pos = 7
298 tp->bits = 3
299 n->pos = 15
Olof Johansson91b9a272005-08-09 20:24:39 -0700300 n->bits = 4
Robert Olsson19baf832005-06-21 12:43:18 -0700301
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900302 First, let's just ignore the bits that come before the parent tp, that is
303 the bits from 0 to (tp->pos-1). They are *known* but at this point we do
Robert Olsson19baf832005-06-21 12:43:18 -0700304 not use them for anything.
305
306 The bits from (tp->pos) to (tp->pos + tp->bits - 1) - "N", above - are the
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900307 index into the parent's child array. That is, they will be used to find
Robert Olsson19baf832005-06-21 12:43:18 -0700308 'n' among tp's children.
309
310 The bits from (tp->pos + tp->bits) to (n->pos - 1) - "S" - are skipped bits
311 for the node n.
312
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900313 All the bits we have seen so far are significant to the node n. The rest
Robert Olsson19baf832005-06-21 12:43:18 -0700314 of the bits are really not needed or indeed known in n->key.
315
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900316 The bits from (n->pos) to (n->pos + n->bits - 1) - "C" - are the index into
Robert Olsson19baf832005-06-21 12:43:18 -0700317 n's child array, and will of course be different for each child.
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900318
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700319
Robert Olsson19baf832005-06-21 12:43:18 -0700320 The rest of the bits, from (n->pos + n->bits) onward, are completely unknown
321 at this point.
322
323*/
324
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700325static inline void check_tnode(const struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700326{
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700327 WARN_ON(tn && tn->pos+tn->bits > 32);
Robert Olsson19baf832005-06-21 12:43:18 -0700328}
329
Denis V. Lunevf5026fa2007-12-13 09:47:57 -0800330static const int halve_threshold = 25;
331static const int inflate_threshold = 50;
Jarek Poplawski345aa032009-07-07 19:39:16 -0700332static const int halve_threshold_root = 15;
Jens Låås80b71b82009-08-28 23:57:15 -0700333static const int inflate_threshold_root = 30;
Robert Olsson2373ce12005-08-25 13:01:29 -0700334
335static void __alias_free_mem(struct rcu_head *head)
336{
337 struct fib_alias *fa = container_of(head, struct fib_alias, rcu);
338 kmem_cache_free(fn_alias_kmem, fa);
339}
340
341static inline void alias_free_mem_rcu(struct fib_alias *fa)
342{
343 call_rcu(&fa->rcu, __alias_free_mem);
344}
345
346static void __leaf_free_rcu(struct rcu_head *head)
347{
Stephen Hemmingerbc3c8c12008-01-22 21:51:50 -0800348 struct leaf *l = container_of(head, struct leaf, rcu);
349 kmem_cache_free(trie_leaf_kmem, l);
Robert Olsson2373ce12005-08-25 13:01:29 -0700350}
351
Stephen Hemminger387a5482008-04-10 03:47:34 -0700352static inline void free_leaf(struct leaf *l)
353{
354 call_rcu_bh(&l->rcu, __leaf_free_rcu);
355}
356
Robert Olsson2373ce12005-08-25 13:01:29 -0700357static void __leaf_info_free_rcu(struct rcu_head *head)
358{
359 kfree(container_of(head, struct leaf_info, rcu));
360}
361
362static inline void free_leaf_info(struct leaf_info *leaf)
363{
364 call_rcu(&leaf->rcu, __leaf_info_free_rcu);
365}
366
Eric Dumazet8d965442008-01-13 22:31:44 -0800367static struct tnode *tnode_alloc(size_t size)
Robert Olsson2373ce12005-08-25 13:01:29 -0700368{
Robert Olsson2373ce12005-08-25 13:01:29 -0700369 if (size <= PAGE_SIZE)
Eric Dumazet8d965442008-01-13 22:31:44 -0800370 return kzalloc(size, GFP_KERNEL);
Stephen Hemminger15be75c2008-04-10 02:56:38 -0700371 else
372 return __vmalloc(size, GFP_KERNEL | __GFP_ZERO, PAGE_KERNEL);
373}
Robert Olsson2373ce12005-08-25 13:01:29 -0700374
Stephen Hemminger15be75c2008-04-10 02:56:38 -0700375static void __tnode_vfree(struct work_struct *arg)
376{
377 struct tnode *tn = container_of(arg, struct tnode, work);
378 vfree(tn);
Robert Olsson2373ce12005-08-25 13:01:29 -0700379}
380
381static void __tnode_free_rcu(struct rcu_head *head)
382{
383 struct tnode *tn = container_of(head, struct tnode, rcu);
Eric Dumazet8d965442008-01-13 22:31:44 -0800384 size_t size = sizeof(struct tnode) +
385 (sizeof(struct node *) << tn->bits);
Robert Olsson2373ce12005-08-25 13:01:29 -0700386
387 if (size <= PAGE_SIZE)
388 kfree(tn);
Stephen Hemminger15be75c2008-04-10 02:56:38 -0700389 else {
390 INIT_WORK(&tn->work, __tnode_vfree);
391 schedule_work(&tn->work);
392 }
Robert Olsson2373ce12005-08-25 13:01:29 -0700393}
394
395static inline void tnode_free(struct tnode *tn)
396{
Stephen Hemminger387a5482008-04-10 03:47:34 -0700397 if (IS_LEAF(tn))
398 free_leaf((struct leaf *) tn);
399 else
Robert Olsson550e29b2006-04-04 12:53:35 -0700400 call_rcu(&tn->rcu, __tnode_free_rcu);
Robert Olsson2373ce12005-08-25 13:01:29 -0700401}
402
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700403static void tnode_free_safe(struct tnode *tn)
404{
405 BUG_ON(IS_LEAF(tn));
Jarek Poplawski7b855762009-06-18 00:28:51 -0700406 tn->tnode_free = tnode_free_head;
407 tnode_free_head = tn;
Jarek Poplawskic3059472009-07-14 08:33:08 +0000408 tnode_free_size += sizeof(struct tnode) +
409 (sizeof(struct node *) << tn->bits);
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700410}
411
412static void tnode_free_flush(void)
413{
414 struct tnode *tn;
415
416 while ((tn = tnode_free_head)) {
417 tnode_free_head = tn->tnode_free;
418 tn->tnode_free = NULL;
419 tnode_free(tn);
420 }
Jarek Poplawskic3059472009-07-14 08:33:08 +0000421
422 if (tnode_free_size >= PAGE_SIZE * sync_pages) {
423 tnode_free_size = 0;
424 synchronize_rcu();
425 }
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700426}
427
Robert Olsson19baf832005-06-21 12:43:18 -0700428static struct leaf *leaf_new(void)
429{
Stephen Hemmingerbc3c8c12008-01-22 21:51:50 -0800430 struct leaf *l = kmem_cache_alloc(trie_leaf_kmem, GFP_KERNEL);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700431 if (l) {
Robert Olsson2373ce12005-08-25 13:01:29 -0700432 l->parent = T_LEAF;
Robert Olsson19baf832005-06-21 12:43:18 -0700433 INIT_HLIST_HEAD(&l->list);
434 }
435 return l;
436}
437
438static struct leaf_info *leaf_info_new(int plen)
439{
440 struct leaf_info *li = kmalloc(sizeof(struct leaf_info), GFP_KERNEL);
Robert Olsson2373ce12005-08-25 13:01:29 -0700441 if (li) {
442 li->plen = plen;
443 INIT_LIST_HEAD(&li->falh);
Patrick McHardyf0e36f82005-07-05 14:44:55 -0700444 }
Robert Olsson2373ce12005-08-25 13:01:29 -0700445 return li;
Patrick McHardyf0e36f82005-07-05 14:44:55 -0700446}
447
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800448static struct tnode *tnode_new(t_key key, int pos, int bits)
Robert Olsson19baf832005-06-21 12:43:18 -0700449{
Eric Dumazet8d965442008-01-13 22:31:44 -0800450 size_t sz = sizeof(struct tnode) + (sizeof(struct node *) << bits);
Patrick McHardyf0e36f82005-07-05 14:44:55 -0700451 struct tnode *tn = tnode_alloc(sz);
Robert Olsson19baf832005-06-21 12:43:18 -0700452
Olof Johansson91b9a272005-08-09 20:24:39 -0700453 if (tn) {
Robert Olsson2373ce12005-08-25 13:01:29 -0700454 tn->parent = T_TNODE;
Robert Olsson19baf832005-06-21 12:43:18 -0700455 tn->pos = pos;
456 tn->bits = bits;
457 tn->key = key;
458 tn->full_children = 0;
459 tn->empty_children = 1<<bits;
460 }
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700461
Eric Dumazet8d965442008-01-13 22:31:44 -0800462 pr_debug("AT %p s=%u %lu\n", tn, (unsigned int) sizeof(struct tnode),
463 (unsigned long) (sizeof(struct node) << bits));
Robert Olsson19baf832005-06-21 12:43:18 -0700464 return tn;
465}
466
Robert Olsson19baf832005-06-21 12:43:18 -0700467/*
468 * Check whether a tnode 'n' is "full", i.e. it is an internal node
469 * and no bits are skipped. See discussion in dyntree paper p. 6
470 */
471
Stephen Hemmignerbb435b82005-08-09 20:25:39 -0700472static inline int tnode_full(const struct tnode *tn, const struct node *n)
Robert Olsson19baf832005-06-21 12:43:18 -0700473{
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700474 if (n == NULL || IS_LEAF(n))
Robert Olsson19baf832005-06-21 12:43:18 -0700475 return 0;
476
477 return ((struct tnode *) n)->pos == tn->pos + tn->bits;
478}
479
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800480static inline void put_child(struct trie *t, struct tnode *tn, int i,
481 struct node *n)
Robert Olsson19baf832005-06-21 12:43:18 -0700482{
483 tnode_put_child_reorg(tn, i, n, -1);
484}
485
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700486 /*
Robert Olsson19baf832005-06-21 12:43:18 -0700487 * Add a child at position i overwriting the old value.
488 * Update the value of full_children and empty_children.
489 */
490
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800491static void tnode_put_child_reorg(struct tnode *tn, int i, struct node *n,
492 int wasfull)
Robert Olsson19baf832005-06-21 12:43:18 -0700493{
Robert Olsson2373ce12005-08-25 13:01:29 -0700494 struct node *chi = tn->child[i];
Robert Olsson19baf832005-06-21 12:43:18 -0700495 int isfull;
496
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700497 BUG_ON(i >= 1<<tn->bits);
498
Robert Olsson19baf832005-06-21 12:43:18 -0700499 /* update emptyChildren */
500 if (n == NULL && chi != NULL)
501 tn->empty_children++;
502 else if (n != NULL && chi == NULL)
503 tn->empty_children--;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700504
Robert Olsson19baf832005-06-21 12:43:18 -0700505 /* update fullChildren */
Olof Johansson91b9a272005-08-09 20:24:39 -0700506 if (wasfull == -1)
Robert Olsson19baf832005-06-21 12:43:18 -0700507 wasfull = tnode_full(tn, chi);
508
509 isfull = tnode_full(tn, n);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700510 if (wasfull && !isfull)
Robert Olsson19baf832005-06-21 12:43:18 -0700511 tn->full_children--;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700512 else if (!wasfull && isfull)
Robert Olsson19baf832005-06-21 12:43:18 -0700513 tn->full_children++;
Olof Johansson91b9a272005-08-09 20:24:39 -0700514
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700515 if (n)
Stephen Hemminger06801912007-08-10 15:22:13 -0700516 node_set_parent(n, tn);
Robert Olsson19baf832005-06-21 12:43:18 -0700517
Robert Olsson2373ce12005-08-25 13:01:29 -0700518 rcu_assign_pointer(tn->child[i], n);
Robert Olsson19baf832005-06-21 12:43:18 -0700519}
520
Jens Låås80b71b82009-08-28 23:57:15 -0700521#define MAX_WORK 10
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700522static struct node *resize(struct trie *t, struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700523{
524 int i;
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700525 struct tnode *old_tn;
Robert Olssone6308be2005-10-04 13:01:58 -0700526 int inflate_threshold_use;
527 int halve_threshold_use;
Jens Låås80b71b82009-08-28 23:57:15 -0700528 int max_work;
Robert Olsson19baf832005-06-21 12:43:18 -0700529
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900530 if (!tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700531 return NULL;
532
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700533 pr_debug("In tnode_resize %p inflate_threshold=%d threshold=%d\n",
534 tn, inflate_threshold, halve_threshold);
Robert Olsson19baf832005-06-21 12:43:18 -0700535
536 /* No children */
537 if (tn->empty_children == tnode_child_length(tn)) {
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700538 tnode_free_safe(tn);
Robert Olsson19baf832005-06-21 12:43:18 -0700539 return NULL;
540 }
541 /* One child */
542 if (tn->empty_children == tnode_child_length(tn) - 1)
Jens Låås80b71b82009-08-28 23:57:15 -0700543 goto one_child;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700544 /*
Robert Olsson19baf832005-06-21 12:43:18 -0700545 * Double as long as the resulting node has a number of
546 * nonempty nodes that are above the threshold.
547 */
548
549 /*
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700550 * From "Implementing a dynamic compressed trie" by Stefan Nilsson of
551 * the Helsinki University of Technology and Matti Tikkanen of Nokia
Robert Olsson19baf832005-06-21 12:43:18 -0700552 * Telecommunications, page 6:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700553 * "A node is doubled if the ratio of non-empty children to all
Robert Olsson19baf832005-06-21 12:43:18 -0700554 * children in the *doubled* node is at least 'high'."
555 *
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700556 * 'high' in this instance is the variable 'inflate_threshold'. It
557 * is expressed as a percentage, so we multiply it with
558 * tnode_child_length() and instead of multiplying by 2 (since the
559 * child array will be doubled by inflate()) and multiplying
560 * the left-hand side by 100 (to handle the percentage thing) we
Robert Olsson19baf832005-06-21 12:43:18 -0700561 * multiply the left-hand side by 50.
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700562 *
563 * The left-hand side may look a bit weird: tnode_child_length(tn)
564 * - tn->empty_children is of course the number of non-null children
565 * in the current node. tn->full_children is the number of "full"
Robert Olsson19baf832005-06-21 12:43:18 -0700566 * children, that is non-null tnodes with a skip value of 0.
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700567 * All of those will be doubled in the resulting inflated tnode, so
Robert Olsson19baf832005-06-21 12:43:18 -0700568 * we just count them one extra time here.
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700569 *
Robert Olsson19baf832005-06-21 12:43:18 -0700570 * A clearer way to write this would be:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700571 *
Robert Olsson19baf832005-06-21 12:43:18 -0700572 * to_be_doubled = tn->full_children;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700573 * not_to_be_doubled = tnode_child_length(tn) - tn->empty_children -
Robert Olsson19baf832005-06-21 12:43:18 -0700574 * tn->full_children;
575 *
576 * new_child_length = tnode_child_length(tn) * 2;
577 *
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700578 * new_fill_factor = 100 * (not_to_be_doubled + 2*to_be_doubled) /
Robert Olsson19baf832005-06-21 12:43:18 -0700579 * new_child_length;
580 * if (new_fill_factor >= inflate_threshold)
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700581 *
582 * ...and so on, tho it would mess up the while () loop.
583 *
Robert Olsson19baf832005-06-21 12:43:18 -0700584 * anyway,
585 * 100 * (not_to_be_doubled + 2*to_be_doubled) / new_child_length >=
586 * inflate_threshold
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700587 *
Robert Olsson19baf832005-06-21 12:43:18 -0700588 * avoid a division:
589 * 100 * (not_to_be_doubled + 2*to_be_doubled) >=
590 * inflate_threshold * new_child_length
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700591 *
Robert Olsson19baf832005-06-21 12:43:18 -0700592 * expand not_to_be_doubled and to_be_doubled, and shorten:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700593 * 100 * (tnode_child_length(tn) - tn->empty_children +
Olof Johansson91b9a272005-08-09 20:24:39 -0700594 * tn->full_children) >= inflate_threshold * new_child_length
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700595 *
Robert Olsson19baf832005-06-21 12:43:18 -0700596 * expand new_child_length:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700597 * 100 * (tnode_child_length(tn) - tn->empty_children +
Olof Johansson91b9a272005-08-09 20:24:39 -0700598 * tn->full_children) >=
Robert Olsson19baf832005-06-21 12:43:18 -0700599 * inflate_threshold * tnode_child_length(tn) * 2
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700600 *
Robert Olsson19baf832005-06-21 12:43:18 -0700601 * shorten again:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700602 * 50 * (tn->full_children + tnode_child_length(tn) -
Olof Johansson91b9a272005-08-09 20:24:39 -0700603 * tn->empty_children) >= inflate_threshold *
Robert Olsson19baf832005-06-21 12:43:18 -0700604 * tnode_child_length(tn)
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700605 *
Robert Olsson19baf832005-06-21 12:43:18 -0700606 */
607
608 check_tnode(tn);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700609
Robert Olssone6308be2005-10-04 13:01:58 -0700610 /* Keep root node larger */
611
Jens Låås80b71b82009-08-28 23:57:15 -0700612 if (!node_parent((struct node*) tn)) {
613 inflate_threshold_use = inflate_threshold_root;
614 halve_threshold_use = halve_threshold_root;
615 }
616 else {
Robert Olssone6308be2005-10-04 13:01:58 -0700617 inflate_threshold_use = inflate_threshold;
Jens Låås80b71b82009-08-28 23:57:15 -0700618 halve_threshold_use = halve_threshold;
619 }
Robert Olssone6308be2005-10-04 13:01:58 -0700620
Jens Låås80b71b82009-08-28 23:57:15 -0700621 max_work = MAX_WORK;
622 while ((tn->full_children > 0 && max_work-- &&
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800623 50 * (tn->full_children + tnode_child_length(tn)
624 - tn->empty_children)
625 >= inflate_threshold_use * tnode_child_length(tn))) {
Robert Olsson19baf832005-06-21 12:43:18 -0700626
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700627 old_tn = tn;
628 tn = inflate(t, tn);
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800629
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700630 if (IS_ERR(tn)) {
631 tn = old_tn;
Robert Olsson2f368952005-07-05 15:02:40 -0700632#ifdef CONFIG_IP_FIB_TRIE_STATS
633 t->stats.resize_node_skipped++;
634#endif
635 break;
636 }
Robert Olsson19baf832005-06-21 12:43:18 -0700637 }
638
639 check_tnode(tn);
640
Jens Låås80b71b82009-08-28 23:57:15 -0700641 /* Return if at least one inflate is run */
642 if( max_work != MAX_WORK)
643 return (struct node *) tn;
644
Robert Olsson19baf832005-06-21 12:43:18 -0700645 /*
646 * Halve as long as the number of empty children in this
647 * node is above threshold.
648 */
Robert Olsson2f368952005-07-05 15:02:40 -0700649
Jens Låås80b71b82009-08-28 23:57:15 -0700650 max_work = MAX_WORK;
651 while (tn->bits > 1 && max_work-- &&
Robert Olsson19baf832005-06-21 12:43:18 -0700652 100 * (tnode_child_length(tn) - tn->empty_children) <
Robert Olssone6308be2005-10-04 13:01:58 -0700653 halve_threshold_use * tnode_child_length(tn)) {
Robert Olsson19baf832005-06-21 12:43:18 -0700654
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700655 old_tn = tn;
656 tn = halve(t, tn);
657 if (IS_ERR(tn)) {
658 tn = old_tn;
Robert Olsson2f368952005-07-05 15:02:40 -0700659#ifdef CONFIG_IP_FIB_TRIE_STATS
660 t->stats.resize_node_skipped++;
661#endif
662 break;
663 }
664 }
665
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700666
Robert Olsson19baf832005-06-21 12:43:18 -0700667 /* Only one child remains */
Jens Låås80b71b82009-08-28 23:57:15 -0700668 if (tn->empty_children == tnode_child_length(tn) - 1) {
669one_child:
Robert Olsson19baf832005-06-21 12:43:18 -0700670 for (i = 0; i < tnode_child_length(tn); i++) {
Olof Johansson91b9a272005-08-09 20:24:39 -0700671 struct node *n;
672
Olof Johansson91b9a272005-08-09 20:24:39 -0700673 n = tn->child[i];
Robert Olsson2373ce12005-08-25 13:01:29 -0700674 if (!n)
Olof Johansson91b9a272005-08-09 20:24:39 -0700675 continue;
Olof Johansson91b9a272005-08-09 20:24:39 -0700676
677 /* compress one level */
678
Stephen Hemminger06801912007-08-10 15:22:13 -0700679 node_set_parent(n, NULL);
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700680 tnode_free_safe(tn);
Olof Johansson91b9a272005-08-09 20:24:39 -0700681 return n;
Robert Olsson19baf832005-06-21 12:43:18 -0700682 }
Jens Låås80b71b82009-08-28 23:57:15 -0700683 }
Robert Olsson19baf832005-06-21 12:43:18 -0700684 return (struct node *) tn;
685}
686
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700687static struct tnode *inflate(struct trie *t, struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700688{
Robert Olsson19baf832005-06-21 12:43:18 -0700689 struct tnode *oldtnode = tn;
690 int olen = tnode_child_length(tn);
691 int i;
692
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700693 pr_debug("In inflate\n");
Robert Olsson19baf832005-06-21 12:43:18 -0700694
695 tn = tnode_new(oldtnode->key, oldtnode->pos, oldtnode->bits + 1);
696
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700697 if (!tn)
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700698 return ERR_PTR(-ENOMEM);
Robert Olsson2f368952005-07-05 15:02:40 -0700699
700 /*
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700701 * Preallocate and store tnodes before the actual work so we
702 * don't get into an inconsistent state if memory allocation
703 * fails. In case of failure we return the oldnode and inflate
Robert Olsson2f368952005-07-05 15:02:40 -0700704 * of tnode is ignored.
705 */
Olof Johansson91b9a272005-08-09 20:24:39 -0700706
707 for (i = 0; i < olen; i++) {
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800708 struct tnode *inode;
Robert Olsson2f368952005-07-05 15:02:40 -0700709
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800710 inode = (struct tnode *) tnode_get_child(oldtnode, i);
Robert Olsson2f368952005-07-05 15:02:40 -0700711 if (inode &&
712 IS_TNODE(inode) &&
713 inode->pos == oldtnode->pos + oldtnode->bits &&
714 inode->bits > 1) {
715 struct tnode *left, *right;
Stephen Hemmingerab66b4a2007-08-10 15:22:58 -0700716 t_key m = ~0U << (KEYLENGTH - 1) >> inode->pos;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700717
Robert Olsson2f368952005-07-05 15:02:40 -0700718 left = tnode_new(inode->key&(~m), inode->pos + 1,
719 inode->bits - 1);
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700720 if (!left)
721 goto nomem;
Olof Johansson91b9a272005-08-09 20:24:39 -0700722
Robert Olsson2f368952005-07-05 15:02:40 -0700723 right = tnode_new(inode->key|m, inode->pos + 1,
724 inode->bits - 1);
725
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900726 if (!right) {
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700727 tnode_free(left);
728 goto nomem;
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900729 }
Robert Olsson2f368952005-07-05 15:02:40 -0700730
731 put_child(t, tn, 2*i, (struct node *) left);
732 put_child(t, tn, 2*i+1, (struct node *) right);
733 }
734 }
735
Olof Johansson91b9a272005-08-09 20:24:39 -0700736 for (i = 0; i < olen; i++) {
Stephen Hemmingerc95aaf92008-01-12 21:25:02 -0800737 struct tnode *inode;
Robert Olsson19baf832005-06-21 12:43:18 -0700738 struct node *node = tnode_get_child(oldtnode, i);
Olof Johansson91b9a272005-08-09 20:24:39 -0700739 struct tnode *left, *right;
740 int size, j;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700741
Robert Olsson19baf832005-06-21 12:43:18 -0700742 /* An empty child */
743 if (node == NULL)
744 continue;
745
746 /* A leaf or an internal node with skipped bits */
747
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700748 if (IS_LEAF(node) || ((struct tnode *) node)->pos >
Robert Olsson19baf832005-06-21 12:43:18 -0700749 tn->pos + tn->bits - 1) {
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800750 if (tkey_extract_bits(node->key,
751 oldtnode->pos + oldtnode->bits,
752 1) == 0)
Robert Olsson19baf832005-06-21 12:43:18 -0700753 put_child(t, tn, 2*i, node);
754 else
755 put_child(t, tn, 2*i+1, node);
756 continue;
757 }
758
759 /* An internal node with two children */
760 inode = (struct tnode *) node;
761
762 if (inode->bits == 1) {
763 put_child(t, tn, 2*i, inode->child[0]);
764 put_child(t, tn, 2*i+1, inode->child[1]);
765
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700766 tnode_free_safe(inode);
Olof Johansson91b9a272005-08-09 20:24:39 -0700767 continue;
Robert Olsson19baf832005-06-21 12:43:18 -0700768 }
769
Olof Johansson91b9a272005-08-09 20:24:39 -0700770 /* An internal node with more than two children */
Robert Olsson19baf832005-06-21 12:43:18 -0700771
Olof Johansson91b9a272005-08-09 20:24:39 -0700772 /* We will replace this node 'inode' with two new
773 * ones, 'left' and 'right', each with half of the
774 * original children. The two new nodes will have
775 * a position one bit further down the key and this
776 * means that the "significant" part of their keys
777 * (see the discussion near the top of this file)
778 * will differ by one bit, which will be "0" in
779 * left's key and "1" in right's key. Since we are
780 * moving the key position by one step, the bit that
781 * we are moving away from - the bit at position
782 * (inode->pos) - is the one that will differ between
783 * left and right. So... we synthesize that bit in the
784 * two new keys.
785 * The mask 'm' below will be a single "one" bit at
786 * the position (inode->pos)
787 */
Robert Olsson19baf832005-06-21 12:43:18 -0700788
Olof Johansson91b9a272005-08-09 20:24:39 -0700789 /* Use the old key, but set the new significant
790 * bit to zero.
791 */
Robert Olsson19baf832005-06-21 12:43:18 -0700792
Olof Johansson91b9a272005-08-09 20:24:39 -0700793 left = (struct tnode *) tnode_get_child(tn, 2*i);
794 put_child(t, tn, 2*i, NULL);
Robert Olsson19baf832005-06-21 12:43:18 -0700795
Olof Johansson91b9a272005-08-09 20:24:39 -0700796 BUG_ON(!left);
Robert Olsson2f368952005-07-05 15:02:40 -0700797
Olof Johansson91b9a272005-08-09 20:24:39 -0700798 right = (struct tnode *) tnode_get_child(tn, 2*i+1);
799 put_child(t, tn, 2*i+1, NULL);
Robert Olsson2f368952005-07-05 15:02:40 -0700800
Olof Johansson91b9a272005-08-09 20:24:39 -0700801 BUG_ON(!right);
Robert Olsson2f368952005-07-05 15:02:40 -0700802
Olof Johansson91b9a272005-08-09 20:24:39 -0700803 size = tnode_child_length(left);
804 for (j = 0; j < size; j++) {
805 put_child(t, left, j, inode->child[j]);
806 put_child(t, right, j, inode->child[j + size]);
Robert Olsson19baf832005-06-21 12:43:18 -0700807 }
Olof Johansson91b9a272005-08-09 20:24:39 -0700808 put_child(t, tn, 2*i, resize(t, left));
809 put_child(t, tn, 2*i+1, resize(t, right));
810
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700811 tnode_free_safe(inode);
Robert Olsson19baf832005-06-21 12:43:18 -0700812 }
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700813 tnode_free_safe(oldtnode);
Robert Olsson19baf832005-06-21 12:43:18 -0700814 return tn;
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700815nomem:
816 {
817 int size = tnode_child_length(tn);
818 int j;
819
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700820 for (j = 0; j < size; j++)
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700821 if (tn->child[j])
822 tnode_free((struct tnode *)tn->child[j]);
823
824 tnode_free(tn);
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700825
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700826 return ERR_PTR(-ENOMEM);
827 }
Robert Olsson19baf832005-06-21 12:43:18 -0700828}
829
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700830static struct tnode *halve(struct trie *t, struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700831{
832 struct tnode *oldtnode = tn;
833 struct node *left, *right;
834 int i;
835 int olen = tnode_child_length(tn);
836
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700837 pr_debug("In halve\n");
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700838
839 tn = tnode_new(oldtnode->key, oldtnode->pos, oldtnode->bits - 1);
Robert Olsson19baf832005-06-21 12:43:18 -0700840
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700841 if (!tn)
842 return ERR_PTR(-ENOMEM);
Robert Olsson2f368952005-07-05 15:02:40 -0700843
844 /*
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700845 * Preallocate and store tnodes before the actual work so we
846 * don't get into an inconsistent state if memory allocation
847 * fails. In case of failure we return the oldnode and halve
Robert Olsson2f368952005-07-05 15:02:40 -0700848 * of tnode is ignored.
849 */
850
Olof Johansson91b9a272005-08-09 20:24:39 -0700851 for (i = 0; i < olen; i += 2) {
Robert Olsson2f368952005-07-05 15:02:40 -0700852 left = tnode_get_child(oldtnode, i);
853 right = tnode_get_child(oldtnode, i+1);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700854
Robert Olsson2f368952005-07-05 15:02:40 -0700855 /* Two nonempty children */
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700856 if (left && right) {
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700857 struct tnode *newn;
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700858
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700859 newn = tnode_new(left->key, tn->pos + tn->bits, 1);
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700860
861 if (!newn)
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700862 goto nomem;
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700863
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700864 put_child(t, tn, i/2, (struct node *)newn);
Robert Olsson2f368952005-07-05 15:02:40 -0700865 }
Robert Olsson2f368952005-07-05 15:02:40 -0700866
Robert Olsson2f368952005-07-05 15:02:40 -0700867 }
Robert Olsson19baf832005-06-21 12:43:18 -0700868
Olof Johansson91b9a272005-08-09 20:24:39 -0700869 for (i = 0; i < olen; i += 2) {
870 struct tnode *newBinNode;
871
Robert Olsson19baf832005-06-21 12:43:18 -0700872 left = tnode_get_child(oldtnode, i);
873 right = tnode_get_child(oldtnode, i+1);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700874
Robert Olsson19baf832005-06-21 12:43:18 -0700875 /* At least one of the children is empty */
876 if (left == NULL) {
877 if (right == NULL) /* Both are empty */
878 continue;
879 put_child(t, tn, i/2, right);
Olof Johansson91b9a272005-08-09 20:24:39 -0700880 continue;
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700881 }
Olof Johansson91b9a272005-08-09 20:24:39 -0700882
883 if (right == NULL) {
Robert Olsson19baf832005-06-21 12:43:18 -0700884 put_child(t, tn, i/2, left);
Olof Johansson91b9a272005-08-09 20:24:39 -0700885 continue;
886 }
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700887
Robert Olsson19baf832005-06-21 12:43:18 -0700888 /* Two nonempty children */
Olof Johansson91b9a272005-08-09 20:24:39 -0700889 newBinNode = (struct tnode *) tnode_get_child(tn, i/2);
890 put_child(t, tn, i/2, NULL);
Olof Johansson91b9a272005-08-09 20:24:39 -0700891 put_child(t, newBinNode, 0, left);
892 put_child(t, newBinNode, 1, right);
893 put_child(t, tn, i/2, resize(t, newBinNode));
Robert Olsson19baf832005-06-21 12:43:18 -0700894 }
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -0700895 tnode_free_safe(oldtnode);
Robert Olsson19baf832005-06-21 12:43:18 -0700896 return tn;
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700897nomem:
898 {
899 int size = tnode_child_length(tn);
900 int j;
901
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700902 for (j = 0; j < size; j++)
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700903 if (tn->child[j])
904 tnode_free((struct tnode *)tn->child[j]);
905
906 tnode_free(tn);
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700907
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700908 return ERR_PTR(-ENOMEM);
909 }
Robert Olsson19baf832005-06-21 12:43:18 -0700910}
911
Robert Olsson772cb712005-09-19 15:31:18 -0700912/* readside must use rcu_read_lock currently dump routines
Robert Olsson2373ce12005-08-25 13:01:29 -0700913 via get_fa_head and dump */
914
Robert Olsson772cb712005-09-19 15:31:18 -0700915static struct leaf_info *find_leaf_info(struct leaf *l, int plen)
Robert Olsson19baf832005-06-21 12:43:18 -0700916{
Robert Olsson772cb712005-09-19 15:31:18 -0700917 struct hlist_head *head = &l->list;
Robert Olsson19baf832005-06-21 12:43:18 -0700918 struct hlist_node *node;
919 struct leaf_info *li;
920
Robert Olsson2373ce12005-08-25 13:01:29 -0700921 hlist_for_each_entry_rcu(li, node, head, hlist)
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700922 if (li->plen == plen)
Robert Olsson19baf832005-06-21 12:43:18 -0700923 return li;
Olof Johansson91b9a272005-08-09 20:24:39 -0700924
Robert Olsson19baf832005-06-21 12:43:18 -0700925 return NULL;
926}
927
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800928static inline struct list_head *get_fa_head(struct leaf *l, int plen)
Robert Olsson19baf832005-06-21 12:43:18 -0700929{
Robert Olsson772cb712005-09-19 15:31:18 -0700930 struct leaf_info *li = find_leaf_info(l, plen);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700931
Olof Johansson91b9a272005-08-09 20:24:39 -0700932 if (!li)
933 return NULL;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700934
Olof Johansson91b9a272005-08-09 20:24:39 -0700935 return &li->falh;
Robert Olsson19baf832005-06-21 12:43:18 -0700936}
937
938static void insert_leaf_info(struct hlist_head *head, struct leaf_info *new)
939{
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900940 struct leaf_info *li = NULL, *last = NULL;
941 struct hlist_node *node;
Robert Olsson19baf832005-06-21 12:43:18 -0700942
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900943 if (hlist_empty(head)) {
944 hlist_add_head_rcu(&new->hlist, head);
945 } else {
946 hlist_for_each_entry(li, node, head, hlist) {
947 if (new->plen > li->plen)
948 break;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700949
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900950 last = li;
951 }
952 if (last)
953 hlist_add_after_rcu(&last->hlist, &new->hlist);
954 else
955 hlist_add_before_rcu(&new->hlist, &li->hlist);
956 }
Robert Olsson19baf832005-06-21 12:43:18 -0700957}
958
Robert Olsson2373ce12005-08-25 13:01:29 -0700959/* rcu_read_lock needs to be hold by caller from readside */
960
Robert Olsson19baf832005-06-21 12:43:18 -0700961static struct leaf *
962fib_find_node(struct trie *t, u32 key)
963{
964 int pos;
965 struct tnode *tn;
966 struct node *n;
967
968 pos = 0;
Paul E. McKenney634a4b22010-03-21 18:01:05 -0700969 n = rcu_dereference_check(t->trie,
970 rcu_read_lock_held() ||
971 lockdep_rtnl_is_held());
Robert Olsson19baf832005-06-21 12:43:18 -0700972
973 while (n != NULL && NODE_TYPE(n) == T_TNODE) {
974 tn = (struct tnode *) n;
Olof Johansson91b9a272005-08-09 20:24:39 -0700975
Robert Olsson19baf832005-06-21 12:43:18 -0700976 check_tnode(tn);
Olof Johansson91b9a272005-08-09 20:24:39 -0700977
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700978 if (tkey_sub_equals(tn->key, pos, tn->pos-pos, key)) {
Olof Johansson91b9a272005-08-09 20:24:39 -0700979 pos = tn->pos + tn->bits;
Stephen Hemmingera07f5f52008-01-22 21:53:36 -0800980 n = tnode_get_child_rcu(tn,
981 tkey_extract_bits(key,
982 tn->pos,
983 tn->bits));
Olof Johansson91b9a272005-08-09 20:24:39 -0700984 } else
Robert Olsson19baf832005-06-21 12:43:18 -0700985 break;
986 }
987 /* Case we have found a leaf. Compare prefixes */
988
Olof Johansson91b9a272005-08-09 20:24:39 -0700989 if (n != NULL && IS_LEAF(n) && tkey_equals(key, n->key))
990 return (struct leaf *)n;
991
Robert Olsson19baf832005-06-21 12:43:18 -0700992 return NULL;
993}
994
Jarek Poplawski7b855762009-06-18 00:28:51 -0700995static void trie_rebalance(struct trie *t, struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700996{
Robert Olsson19baf832005-06-21 12:43:18 -0700997 int wasfull;
Robert Olsson3ed18d72009-05-21 15:20:59 -0700998 t_key cindex, key;
Stephen Hemminger06801912007-08-10 15:22:13 -0700999 struct tnode *tp;
Robert Olsson19baf832005-06-21 12:43:18 -07001000
Robert Olsson3ed18d72009-05-21 15:20:59 -07001001 key = tn->key;
1002
Stephen Hemminger06801912007-08-10 15:22:13 -07001003 while (tn != NULL && (tp = node_parent((struct node *)tn)) != NULL) {
Robert Olsson19baf832005-06-21 12:43:18 -07001004 cindex = tkey_extract_bits(key, tp->pos, tp->bits);
1005 wasfull = tnode_full(tp, tnode_get_child(tp, cindex));
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001006 tn = (struct tnode *) resize(t, (struct tnode *)tn);
1007
1008 tnode_put_child_reorg((struct tnode *)tp, cindex,
1009 (struct node *)tn, wasfull);
Olof Johansson91b9a272005-08-09 20:24:39 -07001010
Stephen Hemminger06801912007-08-10 15:22:13 -07001011 tp = node_parent((struct node *) tn);
Jarek Poplawski008440e2009-06-30 12:47:19 -07001012 if (!tp)
1013 rcu_assign_pointer(t->trie, (struct node *)tn);
1014
Jarek Poplawskie0f7cb82009-06-15 02:31:29 -07001015 tnode_free_flush();
Stephen Hemminger06801912007-08-10 15:22:13 -07001016 if (!tp)
Robert Olsson19baf832005-06-21 12:43:18 -07001017 break;
Stephen Hemminger06801912007-08-10 15:22:13 -07001018 tn = tp;
Robert Olsson19baf832005-06-21 12:43:18 -07001019 }
Stephen Hemminger06801912007-08-10 15:22:13 -07001020
Robert Olsson19baf832005-06-21 12:43:18 -07001021 /* Handle last (top) tnode */
Jarek Poplawski7b855762009-06-18 00:28:51 -07001022 if (IS_TNODE(tn))
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001023 tn = (struct tnode *)resize(t, (struct tnode *)tn);
Robert Olsson19baf832005-06-21 12:43:18 -07001024
Jarek Poplawski7b855762009-06-18 00:28:51 -07001025 rcu_assign_pointer(t->trie, (struct node *)tn);
1026 tnode_free_flush();
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{
Jarek Poplawskif6b085b2010-09-07 07:51:17 +00001758 struct tnode *n = (struct tnode *) rcu_dereference_check(t->trie,
1759 rcu_read_lock_held() ||
1760 lockdep_rtnl_is_held());
Stephen Hemminger82cfbb02008-01-22 21:55:32 -08001761
1762 if (!n)
1763 return NULL;
1764
1765 if (IS_LEAF(n)) /* trie is just a leaf */
1766 return (struct leaf *) n;
1767
1768 return leaf_walk_rcu(n, NULL);
1769}
1770
1771static struct leaf *trie_nextleaf(struct leaf *l)
1772{
1773 struct node *c = (struct node *) l;
Jarek Poplawskib902e572009-07-14 11:20:32 +00001774 struct tnode *p = node_parent_rcu(c);
Stephen Hemminger82cfbb02008-01-22 21:55:32 -08001775
1776 if (!p)
1777 return NULL; /* trie with just one leaf */
1778
1779 return leaf_walk_rcu(p, c);
Robert Olsson19baf832005-06-21 12:43:18 -07001780}
1781
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001782static struct leaf *trie_leafindex(struct trie *t, int index)
1783{
1784 struct leaf *l = trie_firstleaf(t);
1785
Stephen Hemmingerec28cf72008-02-11 21:12:49 -08001786 while (l && index-- > 0)
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001787 l = trie_nextleaf(l);
Stephen Hemmingerec28cf72008-02-11 21:12:49 -08001788
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001789 return l;
1790}
1791
1792
Robert Olssond562f1f2007-03-26 14:22:22 -07001793/*
1794 * Caller must hold RTNL.
1795 */
Stephen Hemminger16c6cf82009-09-20 10:35:36 +00001796int fib_table_flush(struct fib_table *tb)
Robert Olsson19baf832005-06-21 12:43:18 -07001797{
1798 struct trie *t = (struct trie *) tb->tb_data;
Stephen Hemminger9195bef2008-01-22 21:56:34 -08001799 struct leaf *l, *ll = NULL;
Stephen Hemminger82cfbb02008-01-22 21:55:32 -08001800 int found = 0;
Robert Olsson19baf832005-06-21 12:43:18 -07001801
Stephen Hemminger82cfbb02008-01-22 21:55:32 -08001802 for (l = trie_firstleaf(t); l; l = trie_nextleaf(l)) {
Stephen Hemmingeref3660c2008-04-10 03:46:12 -07001803 found += trie_flush_leaf(l);
Robert Olsson19baf832005-06-21 12:43:18 -07001804
1805 if (ll && hlist_empty(&ll->list))
Stephen Hemminger9195bef2008-01-22 21:56:34 -08001806 trie_leaf_remove(t, ll);
Robert Olsson19baf832005-06-21 12:43:18 -07001807 ll = l;
1808 }
1809
1810 if (ll && hlist_empty(&ll->list))
Stephen Hemminger9195bef2008-01-22 21:56:34 -08001811 trie_leaf_remove(t, ll);
Robert Olsson19baf832005-06-21 12:43:18 -07001812
Stephen Hemminger0c7770c2005-08-23 21:59:41 -07001813 pr_debug("trie_flush found=%d\n", found);
Robert Olsson19baf832005-06-21 12:43:18 -07001814 return found;
1815}
1816
Stephen Hemminger16c6cf82009-09-20 10:35:36 +00001817void fib_table_select_default(struct fib_table *tb,
1818 const struct flowi *flp,
1819 struct fib_result *res)
Robert Olsson19baf832005-06-21 12:43:18 -07001820{
1821 struct trie *t = (struct trie *) tb->tb_data;
1822 int order, last_idx;
1823 struct fib_info *fi = NULL;
1824 struct fib_info *last_resort;
1825 struct fib_alias *fa = NULL;
1826 struct list_head *fa_head;
1827 struct leaf *l;
1828
1829 last_idx = -1;
1830 last_resort = NULL;
1831 order = -1;
1832
Robert Olsson2373ce12005-08-25 13:01:29 -07001833 rcu_read_lock();
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001834
Robert Olsson19baf832005-06-21 12:43:18 -07001835 l = fib_find_node(t, 0);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001836 if (!l)
Robert Olsson19baf832005-06-21 12:43:18 -07001837 goto out;
1838
1839 fa_head = get_fa_head(l, 0);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001840 if (!fa_head)
Robert Olsson19baf832005-06-21 12:43:18 -07001841 goto out;
1842
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001843 if (list_empty(fa_head))
Robert Olsson19baf832005-06-21 12:43:18 -07001844 goto out;
1845
Robert Olsson2373ce12005-08-25 13:01:29 -07001846 list_for_each_entry_rcu(fa, fa_head, fa_list) {
Robert Olsson19baf832005-06-21 12:43:18 -07001847 struct fib_info *next_fi = fa->fa_info;
Olof Johansson91b9a272005-08-09 20:24:39 -07001848
Robert Olsson19baf832005-06-21 12:43:18 -07001849 if (fa->fa_scope != res->scope ||
1850 fa->fa_type != RTN_UNICAST)
1851 continue;
Olof Johansson91b9a272005-08-09 20:24:39 -07001852
Robert Olsson19baf832005-06-21 12:43:18 -07001853 if (next_fi->fib_priority > res->fi->fib_priority)
1854 break;
1855 if (!next_fi->fib_nh[0].nh_gw ||
1856 next_fi->fib_nh[0].nh_scope != RT_SCOPE_LINK)
1857 continue;
1858 fa->fa_state |= FA_S_ACCESSED;
Olof Johansson91b9a272005-08-09 20:24:39 -07001859
Robert Olsson19baf832005-06-21 12:43:18 -07001860 if (fi == NULL) {
1861 if (next_fi != res->fi)
1862 break;
1863 } else if (!fib_detect_death(fi, order, &last_resort,
Denis V. Lunev971b8932007-12-08 00:32:23 -08001864 &last_idx, tb->tb_default)) {
Denis V. Luneva2bbe682007-12-08 00:31:44 -08001865 fib_result_assign(res, fi);
Denis V. Lunev971b8932007-12-08 00:32:23 -08001866 tb->tb_default = order;
Robert Olsson19baf832005-06-21 12:43:18 -07001867 goto out;
1868 }
1869 fi = next_fi;
1870 order++;
1871 }
1872 if (order <= 0 || fi == NULL) {
Denis V. Lunev971b8932007-12-08 00:32:23 -08001873 tb->tb_default = -1;
Robert Olsson19baf832005-06-21 12:43:18 -07001874 goto out;
1875 }
1876
Denis V. Lunev971b8932007-12-08 00:32:23 -08001877 if (!fib_detect_death(fi, order, &last_resort, &last_idx,
1878 tb->tb_default)) {
Denis V. Luneva2bbe682007-12-08 00:31:44 -08001879 fib_result_assign(res, fi);
Denis V. Lunev971b8932007-12-08 00:32:23 -08001880 tb->tb_default = order;
Robert Olsson19baf832005-06-21 12:43:18 -07001881 goto out;
1882 }
Denis V. Luneva2bbe682007-12-08 00:31:44 -08001883 if (last_idx >= 0)
1884 fib_result_assign(res, last_resort);
Denis V. Lunev971b8932007-12-08 00:32:23 -08001885 tb->tb_default = last_idx;
1886out:
Robert Olsson2373ce12005-08-25 13:01:29 -07001887 rcu_read_unlock();
Robert Olsson19baf832005-06-21 12:43:18 -07001888}
1889
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08001890static int fn_trie_dump_fa(t_key key, int plen, struct list_head *fah,
1891 struct fib_table *tb,
Robert Olsson19baf832005-06-21 12:43:18 -07001892 struct sk_buff *skb, struct netlink_callback *cb)
1893{
1894 int i, s_i;
1895 struct fib_alias *fa;
Al Viro32ab5f82006-09-26 22:21:45 -07001896 __be32 xkey = htonl(key);
Robert Olsson19baf832005-06-21 12:43:18 -07001897
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001898 s_i = cb->args[5];
Robert Olsson19baf832005-06-21 12:43:18 -07001899 i = 0;
1900
Robert Olsson2373ce12005-08-25 13:01:29 -07001901 /* rcu_read_lock is hold by caller */
1902
1903 list_for_each_entry_rcu(fa, fah, fa_list) {
Robert Olsson19baf832005-06-21 12:43:18 -07001904 if (i < s_i) {
1905 i++;
1906 continue;
1907 }
Robert Olsson19baf832005-06-21 12:43:18 -07001908
1909 if (fib_dump_info(skb, NETLINK_CB(cb->skb).pid,
1910 cb->nlh->nlmsg_seq,
1911 RTM_NEWROUTE,
1912 tb->tb_id,
1913 fa->fa_type,
1914 fa->fa_scope,
Thomas Grafbe403ea2006-08-17 18:15:17 -07001915 xkey,
Robert Olsson19baf832005-06-21 12:43:18 -07001916 plen,
1917 fa->fa_tos,
Stephen Hemminger64347f72008-01-22 21:55:01 -08001918 fa->fa_info, NLM_F_MULTI) < 0) {
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001919 cb->args[5] = i;
Robert Olsson19baf832005-06-21 12:43:18 -07001920 return -1;
Olof Johansson91b9a272005-08-09 20:24:39 -07001921 }
Robert Olsson19baf832005-06-21 12:43:18 -07001922 i++;
1923 }
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001924 cb->args[5] = i;
Robert Olsson19baf832005-06-21 12:43:18 -07001925 return skb->len;
1926}
1927
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001928static int fn_trie_dump_leaf(struct leaf *l, struct fib_table *tb,
1929 struct sk_buff *skb, struct netlink_callback *cb)
Robert Olsson19baf832005-06-21 12:43:18 -07001930{
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001931 struct leaf_info *li;
1932 struct hlist_node *node;
1933 int i, s_i;
Robert Olsson19baf832005-06-21 12:43:18 -07001934
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001935 s_i = cb->args[4];
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001936 i = 0;
Robert Olsson19baf832005-06-21 12:43:18 -07001937
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001938 /* rcu_read_lock is hold by caller */
1939 hlist_for_each_entry_rcu(li, node, &l->list, hlist) {
1940 if (i < s_i) {
1941 i++;
Robert Olsson19baf832005-06-21 12:43:18 -07001942 continue;
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001943 }
Robert Olsson19baf832005-06-21 12:43:18 -07001944
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001945 if (i > s_i)
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001946 cb->args[5] = 0;
Olof Johansson91b9a272005-08-09 20:24:39 -07001947
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001948 if (list_empty(&li->falh))
Robert Olsson19baf832005-06-21 12:43:18 -07001949 continue;
1950
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001951 if (fn_trie_dump_fa(l->key, li->plen, &li->falh, tb, skb, cb) < 0) {
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001952 cb->args[4] = i;
Robert Olsson19baf832005-06-21 12:43:18 -07001953 return -1;
1954 }
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001955 i++;
Robert Olsson19baf832005-06-21 12:43:18 -07001956 }
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001957
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001958 cb->args[4] = i;
Robert Olsson19baf832005-06-21 12:43:18 -07001959 return skb->len;
1960}
1961
Stephen Hemminger16c6cf82009-09-20 10:35:36 +00001962int fib_table_dump(struct fib_table *tb, struct sk_buff *skb,
1963 struct netlink_callback *cb)
Robert Olsson19baf832005-06-21 12:43:18 -07001964{
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001965 struct leaf *l;
Robert Olsson19baf832005-06-21 12:43:18 -07001966 struct trie *t = (struct trie *) tb->tb_data;
Stephen Hemmingerd5ce8a02008-01-22 21:57:22 -08001967 t_key key = cb->args[2];
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001968 int count = cb->args[3];
Robert Olsson19baf832005-06-21 12:43:18 -07001969
Robert Olsson2373ce12005-08-25 13:01:29 -07001970 rcu_read_lock();
Stephen Hemmingerd5ce8a02008-01-22 21:57:22 -08001971 /* Dump starting at last key.
1972 * Note: 0.0.0.0/0 (ie default) is first key.
1973 */
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001974 if (count == 0)
Stephen Hemmingerd5ce8a02008-01-22 21:57:22 -08001975 l = trie_firstleaf(t);
1976 else {
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001977 /* Normally, continue from last key, but if that is missing
1978 * fallback to using slow rescan
1979 */
Stephen Hemmingerd5ce8a02008-01-22 21:57:22 -08001980 l = fib_find_node(t, key);
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001981 if (!l)
1982 l = trie_leafindex(t, count);
Stephen Hemmingerd5ce8a02008-01-22 21:57:22 -08001983 }
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001984
Stephen Hemmingerd5ce8a02008-01-22 21:57:22 -08001985 while (l) {
1986 cb->args[2] = l->key;
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001987 if (fn_trie_dump_leaf(l, tb, skb, cb) < 0) {
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001988 cb->args[3] = count;
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001989 rcu_read_unlock();
Stephen Hemmingera88ee222008-01-22 21:56:11 -08001990 return -1;
Robert Olsson19baf832005-06-21 12:43:18 -07001991 }
Stephen Hemmingerd5ce8a02008-01-22 21:57:22 -08001992
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001993 ++count;
Stephen Hemmingerd5ce8a02008-01-22 21:57:22 -08001994 l = trie_nextleaf(l);
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001995 memset(&cb->args[4], 0,
1996 sizeof(cb->args) - 4*sizeof(cb->args[0]));
Robert Olsson19baf832005-06-21 12:43:18 -07001997 }
Stephen Hemminger71d67e62008-01-31 16:45:47 -08001998 cb->args[3] = count;
Robert Olsson2373ce12005-08-25 13:01:29 -07001999 rcu_read_unlock();
Stephen Hemmingera88ee222008-01-22 21:56:11 -08002000
Robert Olsson19baf832005-06-21 12:43:18 -07002001 return skb->len;
Robert Olsson19baf832005-06-21 12:43:18 -07002002}
2003
Stephen Hemminger7f9b8052008-01-14 23:14:20 -08002004void __init fib_hash_init(void)
2005{
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08002006 fn_alias_kmem = kmem_cache_create("ip_fib_alias",
2007 sizeof(struct fib_alias),
Stephen Hemmingerbc3c8c12008-01-22 21:51:50 -08002008 0, SLAB_PANIC, NULL);
2009
2010 trie_leaf_kmem = kmem_cache_create("ip_fib_trie",
2011 max(sizeof(struct leaf),
2012 sizeof(struct leaf_info)),
2013 0, SLAB_PANIC, NULL);
Stephen Hemminger7f9b8052008-01-14 23:14:20 -08002014}
Robert Olsson19baf832005-06-21 12:43:18 -07002015
Stephen Hemminger7f9b8052008-01-14 23:14:20 -08002016
2017/* Fix more generic FIB names for init later */
2018struct fib_table *fib_hash_table(u32 id)
Robert Olsson19baf832005-06-21 12:43:18 -07002019{
2020 struct fib_table *tb;
2021 struct trie *t;
2022
Robert Olsson19baf832005-06-21 12:43:18 -07002023 tb = kmalloc(sizeof(struct fib_table) + sizeof(struct trie),
2024 GFP_KERNEL);
2025 if (tb == NULL)
2026 return NULL;
2027
2028 tb->tb_id = id;
Denis V. Lunev971b8932007-12-08 00:32:23 -08002029 tb->tb_default = -1;
Robert Olsson19baf832005-06-21 12:43:18 -07002030
2031 t = (struct trie *) tb->tb_data;
Stephen Hemmingerc28a1cf2008-01-12 20:49:13 -08002032 memset(t, 0, sizeof(*t));
Robert Olsson19baf832005-06-21 12:43:18 -07002033
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07002034 if (id == RT_TABLE_LOCAL)
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08002035 pr_info("IPv4 FIB: Using LC-trie version %s\n", VERSION);
Robert Olsson19baf832005-06-21 12:43:18 -07002036
2037 return tb;
2038}
2039
Robert Olsson19baf832005-06-21 12:43:18 -07002040#ifdef CONFIG_PROC_FS
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002041/* Depth first Trie walk iterator */
2042struct fib_trie_iter {
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002043 struct seq_net_private p;
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002044 struct fib_table *tb;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002045 struct tnode *tnode;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002046 unsigned index;
2047 unsigned depth;
2048};
Robert Olsson19baf832005-06-21 12:43:18 -07002049
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002050static struct node *fib_trie_get_next(struct fib_trie_iter *iter)
Robert Olsson19baf832005-06-21 12:43:18 -07002051{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002052 struct tnode *tn = iter->tnode;
2053 unsigned cindex = iter->index;
2054 struct tnode *p;
2055
Eric W. Biederman6640e692007-01-24 14:42:04 -08002056 /* A single entry routing table */
2057 if (!tn)
2058 return NULL;
2059
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002060 pr_debug("get_next iter={node=%p index=%d depth=%d}\n",
2061 iter->tnode, iter->index, iter->depth);
2062rescan:
2063 while (cindex < (1<<tn->bits)) {
Eric Dumazetb59cfbf2008-01-18 03:31:36 -08002064 struct node *n = tnode_get_child_rcu(tn, cindex);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002065
2066 if (n) {
2067 if (IS_LEAF(n)) {
2068 iter->tnode = tn;
2069 iter->index = cindex + 1;
2070 } else {
2071 /* push down one level */
2072 iter->tnode = (struct tnode *) n;
2073 iter->index = 0;
2074 ++iter->depth;
2075 }
2076 return n;
2077 }
2078
2079 ++cindex;
2080 }
2081
2082 /* Current node exhausted, pop back up */
Eric Dumazetb59cfbf2008-01-18 03:31:36 -08002083 p = node_parent_rcu((struct node *)tn);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002084 if (p) {
2085 cindex = tkey_extract_bits(tn->key, p->pos, p->bits)+1;
2086 tn = p;
2087 --iter->depth;
2088 goto rescan;
2089 }
2090
2091 /* got root? */
Robert Olsson19baf832005-06-21 12:43:18 -07002092 return NULL;
2093}
2094
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002095static struct node *fib_trie_get_first(struct fib_trie_iter *iter,
2096 struct trie *t)
Robert Olsson19baf832005-06-21 12:43:18 -07002097{
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002098 struct node *n;
Robert Olsson5ddf0eb2006-03-20 21:34:12 -08002099
Stephen Hemminger132adf52007-03-08 20:44:43 -08002100 if (!t)
Robert Olsson5ddf0eb2006-03-20 21:34:12 -08002101 return NULL;
2102
2103 n = rcu_dereference(t->trie);
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002104 if (!n)
Robert Olsson5ddf0eb2006-03-20 21:34:12 -08002105 return NULL;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002106
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002107 if (IS_TNODE(n)) {
2108 iter->tnode = (struct tnode *) n;
2109 iter->index = 0;
2110 iter->depth = 1;
2111 } else {
2112 iter->tnode = NULL;
2113 iter->index = 0;
2114 iter->depth = 0;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002115 }
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002116
2117 return n;
Robert Olsson19baf832005-06-21 12:43:18 -07002118}
2119
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002120static void trie_collect_stats(struct trie *t, struct trie_stat *s)
Robert Olsson19baf832005-06-21 12:43:18 -07002121{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002122 struct node *n;
2123 struct fib_trie_iter iter;
Robert Olsson19baf832005-06-21 12:43:18 -07002124
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002125 memset(s, 0, sizeof(*s));
Robert Olsson19baf832005-06-21 12:43:18 -07002126
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002127 rcu_read_lock();
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002128 for (n = fib_trie_get_first(&iter, t); n; n = fib_trie_get_next(&iter)) {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002129 if (IS_LEAF(n)) {
Stephen Hemminger93672292008-01-22 21:54:05 -08002130 struct leaf *l = (struct leaf *)n;
2131 struct leaf_info *li;
2132 struct hlist_node *tmp;
2133
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002134 s->leaves++;
2135 s->totdepth += iter.depth;
2136 if (iter.depth > s->maxdepth)
2137 s->maxdepth = iter.depth;
Stephen Hemminger93672292008-01-22 21:54:05 -08002138
2139 hlist_for_each_entry_rcu(li, tmp, &l->list, hlist)
2140 ++s->prefixes;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002141 } else {
2142 const struct tnode *tn = (const struct tnode *) n;
2143 int i;
Robert Olsson19baf832005-06-21 12:43:18 -07002144
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002145 s->tnodes++;
Stephen Hemminger132adf52007-03-08 20:44:43 -08002146 if (tn->bits < MAX_STAT_DEPTH)
Robert Olsson06ef9212006-03-20 21:35:01 -08002147 s->nodesizes[tn->bits]++;
2148
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002149 for (i = 0; i < (1<<tn->bits); i++)
2150 if (!tn->child[i])
2151 s->nullpointers++;
2152 }
2153 }
2154 rcu_read_unlock();
Robert Olsson19baf832005-06-21 12:43:18 -07002155}
2156
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07002157/*
Robert Olsson19baf832005-06-21 12:43:18 -07002158 * This outputs /proc/net/fib_triestats
Robert Olsson19baf832005-06-21 12:43:18 -07002159 */
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002160static void trie_show_stats(struct seq_file *seq, struct trie_stat *stat)
Robert Olsson19baf832005-06-21 12:43:18 -07002161{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002162 unsigned i, max, pointers, bytes, avdepth;
Robert Olsson19baf832005-06-21 12:43:18 -07002163
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002164 if (stat->leaves)
2165 avdepth = stat->totdepth*100 / stat->leaves;
2166 else
2167 avdepth = 0;
Robert Olsson19baf832005-06-21 12:43:18 -07002168
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08002169 seq_printf(seq, "\tAver depth: %u.%02d\n",
2170 avdepth / 100, avdepth % 100);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002171 seq_printf(seq, "\tMax depth: %u\n", stat->maxdepth);
Robert Olsson19baf832005-06-21 12:43:18 -07002172
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002173 seq_printf(seq, "\tLeaves: %u\n", stat->leaves);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002174 bytes = sizeof(struct leaf) * stat->leaves;
Stephen Hemminger93672292008-01-22 21:54:05 -08002175
2176 seq_printf(seq, "\tPrefixes: %u\n", stat->prefixes);
2177 bytes += sizeof(struct leaf_info) * stat->prefixes;
2178
Stephen Hemminger187b5182008-01-12 20:55:55 -08002179 seq_printf(seq, "\tInternal nodes: %u\n\t", stat->tnodes);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002180 bytes += sizeof(struct tnode) * stat->tnodes;
Robert Olsson19baf832005-06-21 12:43:18 -07002181
Robert Olsson06ef9212006-03-20 21:35:01 -08002182 max = MAX_STAT_DEPTH;
2183 while (max > 0 && stat->nodesizes[max-1] == 0)
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002184 max--;
Robert Olsson19baf832005-06-21 12:43:18 -07002185
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002186 pointers = 0;
2187 for (i = 1; i <= max; i++)
2188 if (stat->nodesizes[i] != 0) {
Stephen Hemminger187b5182008-01-12 20:55:55 -08002189 seq_printf(seq, " %u: %u", i, stat->nodesizes[i]);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002190 pointers += (1<<i) * stat->nodesizes[i];
2191 }
2192 seq_putc(seq, '\n');
Stephen Hemminger187b5182008-01-12 20:55:55 -08002193 seq_printf(seq, "\tPointers: %u\n", pointers);
Robert Olsson19baf832005-06-21 12:43:18 -07002194
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002195 bytes += sizeof(struct node *) * pointers;
Stephen Hemminger187b5182008-01-12 20:55:55 -08002196 seq_printf(seq, "Null ptrs: %u\n", stat->nullpointers);
2197 seq_printf(seq, "Total size: %u kB\n", (bytes + 1023) / 1024);
Stephen Hemminger66a2f7f2008-01-12 21:23:17 -08002198}
Robert Olsson19baf832005-06-21 12:43:18 -07002199
2200#ifdef CONFIG_IP_FIB_TRIE_STATS
Stephen Hemminger66a2f7f2008-01-12 21:23:17 -08002201static void trie_show_usage(struct seq_file *seq,
2202 const struct trie_use_stats *stats)
2203{
2204 seq_printf(seq, "\nCounters:\n---------\n");
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08002205 seq_printf(seq, "gets = %u\n", stats->gets);
2206 seq_printf(seq, "backtracks = %u\n", stats->backtrack);
2207 seq_printf(seq, "semantic match passed = %u\n",
2208 stats->semantic_match_passed);
2209 seq_printf(seq, "semantic match miss = %u\n",
2210 stats->semantic_match_miss);
2211 seq_printf(seq, "null node hit= %u\n", stats->null_node_hit);
2212 seq_printf(seq, "skipped node resize = %u\n\n",
2213 stats->resize_node_skipped);
Robert Olsson19baf832005-06-21 12:43:18 -07002214}
Stephen Hemminger66a2f7f2008-01-12 21:23:17 -08002215#endif /* CONFIG_IP_FIB_TRIE_STATS */
2216
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002217static void fib_table_print(struct seq_file *seq, struct fib_table *tb)
Stephen Hemmingerd717a9a2008-01-14 23:11:54 -08002218{
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002219 if (tb->tb_id == RT_TABLE_LOCAL)
2220 seq_puts(seq, "Local:\n");
2221 else if (tb->tb_id == RT_TABLE_MAIN)
2222 seq_puts(seq, "Main:\n");
2223 else
2224 seq_printf(seq, "Id %d:\n", tb->tb_id);
Stephen Hemmingerd717a9a2008-01-14 23:11:54 -08002225}
Robert Olsson19baf832005-06-21 12:43:18 -07002226
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002227
Robert Olsson19baf832005-06-21 12:43:18 -07002228static int fib_triestat_seq_show(struct seq_file *seq, void *v)
2229{
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002230 struct net *net = (struct net *)seq->private;
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002231 unsigned int h;
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002232
Stephen Hemmingerd717a9a2008-01-14 23:11:54 -08002233 seq_printf(seq,
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08002234 "Basic info: size of leaf:"
2235 " %Zd bytes, size of tnode: %Zd bytes.\n",
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002236 sizeof(struct leaf), sizeof(struct tnode));
Olof Johansson91b9a272005-08-09 20:24:39 -07002237
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002238 for (h = 0; h < FIB_TABLE_HASHSZ; h++) {
2239 struct hlist_head *head = &net->ipv4.fib_table_hash[h];
2240 struct hlist_node *node;
2241 struct fib_table *tb;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002242
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002243 hlist_for_each_entry_rcu(tb, node, head, tb_hlist) {
2244 struct trie *t = (struct trie *) tb->tb_data;
2245 struct trie_stat stat;
2246
2247 if (!t)
2248 continue;
2249
2250 fib_table_print(seq, tb);
2251
2252 trie_collect_stats(t, &stat);
2253 trie_show_stats(seq, &stat);
2254#ifdef CONFIG_IP_FIB_TRIE_STATS
2255 trie_show_usage(seq, &t->stats);
2256#endif
2257 }
2258 }
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002259
Robert Olsson19baf832005-06-21 12:43:18 -07002260 return 0;
2261}
2262
Robert Olsson19baf832005-06-21 12:43:18 -07002263static int fib_triestat_seq_open(struct inode *inode, struct file *file)
2264{
Pavel Emelyanovde05c552008-07-18 04:07:21 -07002265 return single_open_net(inode, file, fib_triestat_seq_show);
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002266}
2267
Arjan van de Ven9a321442007-02-12 00:55:35 -08002268static const struct file_operations fib_triestat_fops = {
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07002269 .owner = THIS_MODULE,
2270 .open = fib_triestat_seq_open,
2271 .read = seq_read,
2272 .llseek = seq_lseek,
Pavel Emelyanovb6fcbdb2008-07-18 04:07:44 -07002273 .release = single_release_net,
Robert Olsson19baf832005-06-21 12:43:18 -07002274};
2275
YOSHIFUJI Hideaki12188542008-03-26 02:36:06 +09002276static struct node *fib_trie_get_idx(struct seq_file *seq, loff_t pos)
Robert Olsson19baf832005-06-21 12:43:18 -07002277{
YOSHIFUJI Hideaki12188542008-03-26 02:36:06 +09002278 struct fib_trie_iter *iter = seq->private;
2279 struct net *net = seq_file_net(seq);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002280 loff_t idx = 0;
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002281 unsigned int h;
Robert Olsson19baf832005-06-21 12:43:18 -07002282
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002283 for (h = 0; h < FIB_TABLE_HASHSZ; h++) {
2284 struct hlist_head *head = &net->ipv4.fib_table_hash[h];
2285 struct hlist_node *node;
2286 struct fib_table *tb;
2287
2288 hlist_for_each_entry_rcu(tb, node, head, tb_hlist) {
2289 struct node *n;
2290
2291 for (n = fib_trie_get_first(iter,
2292 (struct trie *) tb->tb_data);
2293 n; n = fib_trie_get_next(iter))
2294 if (pos == idx++) {
2295 iter->tb = tb;
2296 return n;
2297 }
2298 }
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002299 }
Robert Olsson19baf832005-06-21 12:43:18 -07002300
Robert Olsson19baf832005-06-21 12:43:18 -07002301 return NULL;
2302}
2303
2304static void *fib_trie_seq_start(struct seq_file *seq, loff_t *pos)
Stephen Hemmingerc95aaf92008-01-12 21:25:02 -08002305 __acquires(RCU)
Robert Olsson19baf832005-06-21 12:43:18 -07002306{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002307 rcu_read_lock();
YOSHIFUJI Hideaki12188542008-03-26 02:36:06 +09002308 return fib_trie_get_idx(seq, *pos);
Robert Olsson19baf832005-06-21 12:43:18 -07002309}
2310
2311static void *fib_trie_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2312{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002313 struct fib_trie_iter *iter = seq->private;
YOSHIFUJI Hideaki12188542008-03-26 02:36:06 +09002314 struct net *net = seq_file_net(seq);
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002315 struct fib_table *tb = iter->tb;
2316 struct hlist_node *tb_node;
2317 unsigned int h;
2318 struct node *n;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002319
Robert Olsson19baf832005-06-21 12:43:18 -07002320 ++*pos;
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002321 /* next node in same table */
2322 n = fib_trie_get_next(iter);
2323 if (n)
2324 return n;
Olof Johansson91b9a272005-08-09 20:24:39 -07002325
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002326 /* walk rest of this hash chain */
2327 h = tb->tb_id & (FIB_TABLE_HASHSZ - 1);
2328 while ( (tb_node = rcu_dereference(tb->tb_hlist.next)) ) {
2329 tb = hlist_entry(tb_node, struct fib_table, tb_hlist);
2330 n = fib_trie_get_first(iter, (struct trie *) tb->tb_data);
2331 if (n)
2332 goto found;
2333 }
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002334
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002335 /* new hash chain */
2336 while (++h < FIB_TABLE_HASHSZ) {
2337 struct hlist_head *head = &net->ipv4.fib_table_hash[h];
2338 hlist_for_each_entry_rcu(tb, tb_node, head, tb_hlist) {
2339 n = fib_trie_get_first(iter, (struct trie *) tb->tb_data);
2340 if (n)
2341 goto found;
2342 }
2343 }
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002344 return NULL;
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002345
2346found:
2347 iter->tb = tb;
2348 return n;
Robert Olsson19baf832005-06-21 12:43:18 -07002349}
2350
2351static void fib_trie_seq_stop(struct seq_file *seq, void *v)
Stephen Hemmingerc95aaf92008-01-12 21:25:02 -08002352 __releases(RCU)
Robert Olsson19baf832005-06-21 12:43:18 -07002353{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002354 rcu_read_unlock();
Robert Olsson19baf832005-06-21 12:43:18 -07002355}
2356
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002357static void seq_indent(struct seq_file *seq, int n)
2358{
2359 while (n-- > 0) seq_puts(seq, " ");
2360}
Robert Olsson19baf832005-06-21 12:43:18 -07002361
Eric Dumazet28d36e32008-01-14 23:09:56 -08002362static inline const char *rtn_scope(char *buf, size_t len, enum rt_scope_t s)
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002363{
Stephen Hemminger132adf52007-03-08 20:44:43 -08002364 switch (s) {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002365 case RT_SCOPE_UNIVERSE: return "universe";
2366 case RT_SCOPE_SITE: return "site";
2367 case RT_SCOPE_LINK: return "link";
2368 case RT_SCOPE_HOST: return "host";
2369 case RT_SCOPE_NOWHERE: return "nowhere";
2370 default:
Eric Dumazet28d36e32008-01-14 23:09:56 -08002371 snprintf(buf, len, "scope=%d", s);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002372 return buf;
2373 }
2374}
2375
Jan Engelhardt36cbd3d2009-08-05 10:42:58 -07002376static const char *const rtn_type_names[__RTN_MAX] = {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002377 [RTN_UNSPEC] = "UNSPEC",
2378 [RTN_UNICAST] = "UNICAST",
2379 [RTN_LOCAL] = "LOCAL",
2380 [RTN_BROADCAST] = "BROADCAST",
2381 [RTN_ANYCAST] = "ANYCAST",
2382 [RTN_MULTICAST] = "MULTICAST",
2383 [RTN_BLACKHOLE] = "BLACKHOLE",
2384 [RTN_UNREACHABLE] = "UNREACHABLE",
2385 [RTN_PROHIBIT] = "PROHIBIT",
2386 [RTN_THROW] = "THROW",
2387 [RTN_NAT] = "NAT",
2388 [RTN_XRESOLVE] = "XRESOLVE",
2389};
2390
Eric Dumazet28d36e32008-01-14 23:09:56 -08002391static inline const char *rtn_type(char *buf, size_t len, unsigned t)
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002392{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002393 if (t < __RTN_MAX && rtn_type_names[t])
2394 return rtn_type_names[t];
Eric Dumazet28d36e32008-01-14 23:09:56 -08002395 snprintf(buf, len, "type %u", t);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002396 return buf;
2397}
2398
2399/* Pretty print the trie */
Robert Olsson19baf832005-06-21 12:43:18 -07002400static int fib_trie_seq_show(struct seq_file *seq, void *v)
2401{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002402 const struct fib_trie_iter *iter = seq->private;
2403 struct node *n = v;
Robert Olsson19baf832005-06-21 12:43:18 -07002404
Stephen Hemminger3d3b2d22008-03-23 22:43:56 -07002405 if (!node_parent_rcu(n))
2406 fib_table_print(seq, iter->tb);
Robert Olsson095b8502007-01-26 19:06:01 -08002407
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002408 if (IS_TNODE(n)) {
2409 struct tnode *tn = (struct tnode *) n;
Stephen Hemmingerab66b4a2007-08-10 15:22:58 -07002410 __be32 prf = htonl(mask_pfx(tn->key, tn->pos));
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002411
Robert Olsson1d25cd62005-09-19 15:29:52 -07002412 seq_indent(seq, iter->depth-1);
Harvey Harrison673d57e2008-10-31 00:53:57 -07002413 seq_printf(seq, " +-- %pI4/%d %d %d %d\n",
2414 &prf, tn->pos, tn->bits, tn->full_children,
Robert Olsson1d25cd62005-09-19 15:29:52 -07002415 tn->empty_children);
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +09002416
Olof Johansson91b9a272005-08-09 20:24:39 -07002417 } else {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002418 struct leaf *l = (struct leaf *) n;
Stephen Hemminger13280422008-01-22 21:54:37 -08002419 struct leaf_info *li;
2420 struct hlist_node *node;
Al Viro32ab5f82006-09-26 22:21:45 -07002421 __be32 val = htonl(l->key);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002422
2423 seq_indent(seq, iter->depth);
Harvey Harrison673d57e2008-10-31 00:53:57 -07002424 seq_printf(seq, " |-- %pI4\n", &val);
Eric Dumazet28d36e32008-01-14 23:09:56 -08002425
Stephen Hemminger13280422008-01-22 21:54:37 -08002426 hlist_for_each_entry_rcu(li, node, &l->list, hlist) {
2427 struct fib_alias *fa;
Eric Dumazet28d36e32008-01-14 23:09:56 -08002428
Stephen Hemminger13280422008-01-22 21:54:37 -08002429 list_for_each_entry_rcu(fa, &li->falh, fa_list) {
2430 char buf1[32], buf2[32];
Eric Dumazet28d36e32008-01-14 23:09:56 -08002431
Stephen Hemminger13280422008-01-22 21:54:37 -08002432 seq_indent(seq, iter->depth+1);
2433 seq_printf(seq, " /%d %s %s", li->plen,
2434 rtn_scope(buf1, sizeof(buf1),
2435 fa->fa_scope),
2436 rtn_type(buf2, sizeof(buf2),
2437 fa->fa_type));
2438 if (fa->fa_tos)
Denis V. Lunevb9c4d822008-02-05 02:58:45 -08002439 seq_printf(seq, " tos=%d", fa->fa_tos);
Stephen Hemminger13280422008-01-22 21:54:37 -08002440 seq_putc(seq, '\n');
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002441 }
2442 }
Robert Olsson19baf832005-06-21 12:43:18 -07002443 }
2444
2445 return 0;
2446}
2447
Stephen Hemmingerf6908082007-03-12 14:34:29 -07002448static const struct seq_operations fib_trie_seq_ops = {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002449 .start = fib_trie_seq_start,
2450 .next = fib_trie_seq_next,
2451 .stop = fib_trie_seq_stop,
2452 .show = fib_trie_seq_show,
Robert Olsson19baf832005-06-21 12:43:18 -07002453};
2454
2455static int fib_trie_seq_open(struct inode *inode, struct file *file)
2456{
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002457 return seq_open_net(inode, file, &fib_trie_seq_ops,
2458 sizeof(struct fib_trie_iter));
Robert Olsson19baf832005-06-21 12:43:18 -07002459}
2460
Arjan van de Ven9a321442007-02-12 00:55:35 -08002461static const struct file_operations fib_trie_fops = {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002462 .owner = THIS_MODULE,
2463 .open = fib_trie_seq_open,
2464 .read = seq_read,
2465 .llseek = seq_lseek,
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002466 .release = seq_release_net,
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002467};
2468
Stephen Hemminger8315f5d2008-02-11 21:14:39 -08002469struct fib_route_iter {
2470 struct seq_net_private p;
2471 struct trie *main_trie;
2472 loff_t pos;
2473 t_key key;
2474};
2475
2476static struct leaf *fib_route_get_idx(struct fib_route_iter *iter, loff_t pos)
2477{
2478 struct leaf *l = NULL;
2479 struct trie *t = iter->main_trie;
2480
2481 /* use cache location of last found key */
2482 if (iter->pos > 0 && pos >= iter->pos && (l = fib_find_node(t, iter->key)))
2483 pos -= iter->pos;
2484 else {
2485 iter->pos = 0;
2486 l = trie_firstleaf(t);
2487 }
2488
2489 while (l && pos-- > 0) {
2490 iter->pos++;
2491 l = trie_nextleaf(l);
2492 }
2493
2494 if (l)
2495 iter->key = pos; /* remember it */
2496 else
2497 iter->pos = 0; /* forget it */
2498
2499 return l;
2500}
2501
2502static void *fib_route_seq_start(struct seq_file *seq, loff_t *pos)
2503 __acquires(RCU)
2504{
2505 struct fib_route_iter *iter = seq->private;
2506 struct fib_table *tb;
2507
2508 rcu_read_lock();
YOSHIFUJI Hideaki12188542008-03-26 02:36:06 +09002509 tb = fib_get_table(seq_file_net(seq), RT_TABLE_MAIN);
Stephen Hemminger8315f5d2008-02-11 21:14:39 -08002510 if (!tb)
2511 return NULL;
2512
2513 iter->main_trie = (struct trie *) tb->tb_data;
2514 if (*pos == 0)
2515 return SEQ_START_TOKEN;
2516 else
2517 return fib_route_get_idx(iter, *pos - 1);
2518}
2519
2520static void *fib_route_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2521{
2522 struct fib_route_iter *iter = seq->private;
2523 struct leaf *l = v;
2524
2525 ++*pos;
2526 if (v == SEQ_START_TOKEN) {
2527 iter->pos = 0;
2528 l = trie_firstleaf(iter->main_trie);
2529 } else {
2530 iter->pos++;
2531 l = trie_nextleaf(l);
2532 }
2533
2534 if (l)
2535 iter->key = l->key;
2536 else
2537 iter->pos = 0;
2538 return l;
2539}
2540
2541static void fib_route_seq_stop(struct seq_file *seq, void *v)
2542 __releases(RCU)
2543{
2544 rcu_read_unlock();
2545}
2546
Al Viro32ab5f82006-09-26 22:21:45 -07002547static unsigned fib_flag_trans(int type, __be32 mask, const struct fib_info *fi)
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002548{
2549 static unsigned type2flags[RTN_MAX + 1] = {
2550 [7] = RTF_REJECT, [8] = RTF_REJECT,
2551 };
2552 unsigned flags = type2flags[type];
2553
2554 if (fi && fi->fib_nh->nh_gw)
2555 flags |= RTF_GATEWAY;
Al Viro32ab5f82006-09-26 22:21:45 -07002556 if (mask == htonl(0xFFFFFFFF))
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002557 flags |= RTF_HOST;
2558 flags |= RTF_UP;
2559 return flags;
2560}
2561
2562/*
2563 * This outputs /proc/net/route.
2564 * The format of the file is not supposed to be changed
2565 * and needs to be same as fib_hash output to avoid breaking
2566 * legacy utilities
2567 */
2568static int fib_route_seq_show(struct seq_file *seq, void *v)
2569{
2570 struct leaf *l = v;
Stephen Hemminger13280422008-01-22 21:54:37 -08002571 struct leaf_info *li;
2572 struct hlist_node *node;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002573
2574 if (v == SEQ_START_TOKEN) {
2575 seq_printf(seq, "%-127s\n", "Iface\tDestination\tGateway "
2576 "\tFlags\tRefCnt\tUse\tMetric\tMask\t\tMTU"
2577 "\tWindow\tIRTT");
2578 return 0;
2579 }
2580
Stephen Hemminger13280422008-01-22 21:54:37 -08002581 hlist_for_each_entry_rcu(li, node, &l->list, hlist) {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002582 struct fib_alias *fa;
Al Viro32ab5f82006-09-26 22:21:45 -07002583 __be32 mask, prefix;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002584
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002585 mask = inet_make_mask(li->plen);
2586 prefix = htonl(l->key);
2587
2588 list_for_each_entry_rcu(fa, &li->falh, fa_list) {
Herbert Xu1371e372005-10-15 09:42:39 +10002589 const struct fib_info *fi = fa->fa_info;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002590 unsigned flags = fib_flag_trans(fa->fa_type, mask, fi);
Pavel Emelyanov5e659e42008-04-24 01:02:16 -07002591 int len;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002592
2593 if (fa->fa_type == RTN_BROADCAST
2594 || fa->fa_type == RTN_MULTICAST)
2595 continue;
2596
2597 if (fi)
Pavel Emelyanov5e659e42008-04-24 01:02:16 -07002598 seq_printf(seq,
2599 "%s\t%08X\t%08X\t%04X\t%d\t%u\t"
2600 "%d\t%08X\t%d\t%u\t%u%n",
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002601 fi->fib_dev ? fi->fib_dev->name : "*",
2602 prefix,
2603 fi->fib_nh->nh_gw, flags, 0, 0,
2604 fi->fib_priority,
2605 mask,
Stephen Hemmingera07f5f52008-01-22 21:53:36 -08002606 (fi->fib_advmss ?
2607 fi->fib_advmss + 40 : 0),
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002608 fi->fib_window,
Pavel Emelyanov5e659e42008-04-24 01:02:16 -07002609 fi->fib_rtt >> 3, &len);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002610 else
Pavel Emelyanov5e659e42008-04-24 01:02:16 -07002611 seq_printf(seq,
2612 "*\t%08X\t%08X\t%04X\t%d\t%u\t"
2613 "%d\t%08X\t%d\t%u\t%u%n",
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002614 prefix, 0, flags, 0, 0, 0,
Pavel Emelyanov5e659e42008-04-24 01:02:16 -07002615 mask, 0, 0, 0, &len);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002616
Pavel Emelyanov5e659e42008-04-24 01:02:16 -07002617 seq_printf(seq, "%*s\n", 127 - len, "");
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002618 }
2619 }
2620
2621 return 0;
2622}
2623
Stephen Hemmingerf6908082007-03-12 14:34:29 -07002624static const struct seq_operations fib_route_seq_ops = {
Stephen Hemminger8315f5d2008-02-11 21:14:39 -08002625 .start = fib_route_seq_start,
2626 .next = fib_route_seq_next,
2627 .stop = fib_route_seq_stop,
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002628 .show = fib_route_seq_show,
2629};
2630
2631static int fib_route_seq_open(struct inode *inode, struct file *file)
2632{
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002633 return seq_open_net(inode, file, &fib_route_seq_ops,
Stephen Hemminger8315f5d2008-02-11 21:14:39 -08002634 sizeof(struct fib_route_iter));
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002635}
2636
Arjan van de Ven9a321442007-02-12 00:55:35 -08002637static const struct file_operations fib_route_fops = {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002638 .owner = THIS_MODULE,
2639 .open = fib_route_seq_open,
2640 .read = seq_read,
2641 .llseek = seq_lseek,
Denis V. Lunev1c340b22008-01-10 03:27:17 -08002642 .release = seq_release_net,
Robert Olsson19baf832005-06-21 12:43:18 -07002643};
2644
Denis V. Lunev61a02652008-01-10 03:21:09 -08002645int __net_init fib_proc_init(struct net *net)
Robert Olsson19baf832005-06-21 12:43:18 -07002646{
Denis V. Lunev61a02652008-01-10 03:21:09 -08002647 if (!proc_net_fops_create(net, "fib_trie", S_IRUGO, &fib_trie_fops))
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002648 goto out1;
2649
Denis V. Lunev61a02652008-01-10 03:21:09 -08002650 if (!proc_net_fops_create(net, "fib_triestat", S_IRUGO,
2651 &fib_triestat_fops))
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002652 goto out2;
2653
Denis V. Lunev61a02652008-01-10 03:21:09 -08002654 if (!proc_net_fops_create(net, "route", S_IRUGO, &fib_route_fops))
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002655 goto out3;
2656
Robert Olsson19baf832005-06-21 12:43:18 -07002657 return 0;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002658
2659out3:
Denis V. Lunev61a02652008-01-10 03:21:09 -08002660 proc_net_remove(net, "fib_triestat");
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002661out2:
Denis V. Lunev61a02652008-01-10 03:21:09 -08002662 proc_net_remove(net, "fib_trie");
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002663out1:
2664 return -ENOMEM;
Robert Olsson19baf832005-06-21 12:43:18 -07002665}
2666
Denis V. Lunev61a02652008-01-10 03:21:09 -08002667void __net_exit fib_proc_exit(struct net *net)
Robert Olsson19baf832005-06-21 12:43:18 -07002668{
Denis V. Lunev61a02652008-01-10 03:21:09 -08002669 proc_net_remove(net, "fib_trie");
2670 proc_net_remove(net, "fib_triestat");
2671 proc_net_remove(net, "route");
Robert Olsson19baf832005-06-21 12:43:18 -07002672}
2673
2674#endif /* CONFIG_PROC_FS */