blob: 4bbfaeedf210d163596ab38f3ce4b5a1a483c287 [file] [log] [blame]
Robert Olsson19baf832005-06-21 12:43:18 -07001/*
2 * This program is free software; you can redistribute it and/or
3 * modify it under the terms of the GNU General Public License
4 * as published by the Free Software Foundation; either version
5 * 2 of the License, or (at your option) any later version.
6 *
7 * Robert Olsson <robert.olsson@its.uu.se> Uppsala Universitet
8 * & Swedish University of Agricultural Sciences.
9 *
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +090010 * Jens Laas <jens.laas@data.slu.se> Swedish University of
Robert Olsson19baf832005-06-21 12:43:18 -070011 * Agricultural Sciences.
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +090012 *
Robert Olsson19baf832005-06-21 12:43:18 -070013 * Hans Liss <hans.liss@its.uu.se> Uppsala Universitet
14 *
15 * This work is based on the LPC-trie which is originally descibed in:
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +090016 *
Robert Olsson19baf832005-06-21 12:43:18 -070017 * An experimental study of compression methods for dynamic tries
18 * Stefan Nilsson and Matti Tikkanen. Algorithmica, 33(1):19-33, 2002.
19 * http://www.nada.kth.se/~snilsson/public/papers/dyntrie2/
20 *
21 *
22 * IP-address lookup using LC-tries. Stefan Nilsson and Gunnar Karlsson
23 * IEEE Journal on Selected Areas in Communications, 17(6):1083-1092, June 1999
24 *
25 * Version: $Id: fib_trie.c,v 1.3 2005/06/08 14:20:01 robert Exp $
26 *
27 *
28 * Code from fib_hash has been reused which includes the following header:
29 *
30 *
31 * INET An implementation of the TCP/IP protocol suite for the LINUX
32 * operating system. INET is implemented using the BSD Socket
33 * interface as the means of communication with the user level.
34 *
35 * IPv4 FIB: lookup engine and maintenance routines.
36 *
37 *
38 * Authors: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
39 *
40 * This program is free software; you can redistribute it and/or
41 * modify it under the terms of the GNU General Public License
42 * as published by the Free Software Foundation; either version
43 * 2 of the License, or (at your option) any later version.
Robert Olssonfd966252005-12-22 11:25:10 -080044 *
45 * Substantial contributions to this work comes from:
46 *
47 * David S. Miller, <davem@davemloft.net>
48 * Stephen Hemminger <shemminger@osdl.org>
49 * Paul E. McKenney <paulmck@us.ibm.com>
50 * Patrick McHardy <kaber@trash.net>
Robert Olsson19baf832005-06-21 12:43:18 -070051 */
52
Robert Olsson05eee482007-03-19 16:27:37 -070053#define VERSION "0.408"
Robert Olsson19baf832005-06-21 12:43:18 -070054
Robert Olsson19baf832005-06-21 12:43:18 -070055#include <asm/uaccess.h>
56#include <asm/system.h>
Jiri Slaby1977f032007-10-18 23:40:25 -070057#include <linux/bitops.h>
Robert Olsson19baf832005-06-21 12:43:18 -070058#include <linux/types.h>
59#include <linux/kernel.h>
Robert Olsson19baf832005-06-21 12:43:18 -070060#include <linux/mm.h>
61#include <linux/string.h>
62#include <linux/socket.h>
63#include <linux/sockios.h>
64#include <linux/errno.h>
65#include <linux/in.h>
66#include <linux/inet.h>
Stephen Hemmingercd8787a2006-01-03 14:38:34 -080067#include <linux/inetdevice.h>
Robert Olsson19baf832005-06-21 12:43:18 -070068#include <linux/netdevice.h>
69#include <linux/if_arp.h>
70#include <linux/proc_fs.h>
Robert Olsson2373ce12005-08-25 13:01:29 -070071#include <linux/rcupdate.h>
Robert Olsson19baf832005-06-21 12:43:18 -070072#include <linux/skbuff.h>
73#include <linux/netlink.h>
74#include <linux/init.h>
75#include <linux/list.h>
Eric W. Biederman457c4cb2007-09-12 12:01:34 +020076#include <net/net_namespace.h>
Robert Olsson19baf832005-06-21 12:43:18 -070077#include <net/ip.h>
78#include <net/protocol.h>
79#include <net/route.h>
80#include <net/tcp.h>
81#include <net/sock.h>
82#include <net/ip_fib.h>
83#include "fib_lookup.h"
84
85#undef CONFIG_IP_FIB_TRIE_STATS
Robert Olsson06ef9212006-03-20 21:35:01 -080086#define MAX_STAT_DEPTH 32
Robert Olsson19baf832005-06-21 12:43:18 -070087
Robert Olsson19baf832005-06-21 12:43:18 -070088#define KEYLENGTH (8*sizeof(t_key))
Robert Olsson19baf832005-06-21 12:43:18 -070089
Robert Olsson19baf832005-06-21 12:43:18 -070090typedef unsigned int t_key;
91
92#define T_TNODE 0
93#define T_LEAF 1
94#define NODE_TYPE_MASK 0x1UL
Robert Olsson2373ce12005-08-25 13:01:29 -070095#define NODE_TYPE(node) ((node)->parent & NODE_TYPE_MASK)
96
Olof Johansson91b9a272005-08-09 20:24:39 -070097#define IS_TNODE(n) (!(n->parent & T_LEAF))
98#define IS_LEAF(n) (n->parent & T_LEAF)
Robert Olsson19baf832005-06-21 12:43:18 -070099
100struct node {
Olof Johansson91b9a272005-08-09 20:24:39 -0700101 t_key key;
102 unsigned long parent;
Robert Olsson19baf832005-06-21 12:43:18 -0700103};
104
105struct leaf {
Olof Johansson91b9a272005-08-09 20:24:39 -0700106 t_key key;
107 unsigned long parent;
Robert Olsson19baf832005-06-21 12:43:18 -0700108 struct hlist_head list;
Robert Olsson2373ce12005-08-25 13:01:29 -0700109 struct rcu_head rcu;
Robert Olsson19baf832005-06-21 12:43:18 -0700110};
111
112struct leaf_info {
113 struct hlist_node hlist;
Robert Olsson2373ce12005-08-25 13:01:29 -0700114 struct rcu_head rcu;
Robert Olsson19baf832005-06-21 12:43:18 -0700115 int plen;
116 struct list_head falh;
117};
118
119struct tnode {
Olof Johansson91b9a272005-08-09 20:24:39 -0700120 t_key key;
121 unsigned long parent;
122 unsigned short pos:5; /* 2log(KEYLENGTH) bits needed */
123 unsigned short bits:5; /* 2log(KEYLENGTH) bits needed */
124 unsigned short full_children; /* KEYLENGTH bits needed */
125 unsigned short empty_children; /* KEYLENGTH bits needed */
Robert Olsson2373ce12005-08-25 13:01:29 -0700126 struct rcu_head rcu;
Olof Johansson91b9a272005-08-09 20:24:39 -0700127 struct node *child[0];
Robert Olsson19baf832005-06-21 12:43:18 -0700128};
129
130#ifdef CONFIG_IP_FIB_TRIE_STATS
131struct trie_use_stats {
132 unsigned int gets;
133 unsigned int backtrack;
134 unsigned int semantic_match_passed;
135 unsigned int semantic_match_miss;
136 unsigned int null_node_hit;
Robert Olsson2f368952005-07-05 15:02:40 -0700137 unsigned int resize_node_skipped;
Robert Olsson19baf832005-06-21 12:43:18 -0700138};
139#endif
140
141struct trie_stat {
142 unsigned int totdepth;
143 unsigned int maxdepth;
144 unsigned int tnodes;
145 unsigned int leaves;
146 unsigned int nullpointers;
Robert Olsson06ef9212006-03-20 21:35:01 -0800147 unsigned int nodesizes[MAX_STAT_DEPTH];
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700148};
Robert Olsson19baf832005-06-21 12:43:18 -0700149
150struct trie {
Olof Johansson91b9a272005-08-09 20:24:39 -0700151 struct node *trie;
Robert Olsson19baf832005-06-21 12:43:18 -0700152#ifdef CONFIG_IP_FIB_TRIE_STATS
153 struct trie_use_stats stats;
154#endif
Olof Johansson91b9a272005-08-09 20:24:39 -0700155 int size;
Robert Olsson19baf832005-06-21 12:43:18 -0700156 unsigned int revision;
157};
158
Robert Olsson19baf832005-06-21 12:43:18 -0700159static void put_child(struct trie *t, struct tnode *tn, int i, struct node *n);
160static void tnode_put_child_reorg(struct tnode *tn, int i, struct node *n, int wasfull);
Robert Olsson19baf832005-06-21 12:43:18 -0700161static struct node *resize(struct trie *t, struct tnode *tn);
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700162static struct tnode *inflate(struct trie *t, struct tnode *tn);
163static struct tnode *halve(struct trie *t, struct tnode *tn);
Robert Olsson19baf832005-06-21 12:43:18 -0700164static void tnode_free(struct tnode *tn);
Robert Olsson19baf832005-06-21 12:43:18 -0700165
Christoph Lametere18b8902006-12-06 20:33:20 -0800166static struct kmem_cache *fn_alias_kmem __read_mostly;
Robert Olsson19baf832005-06-21 12:43:18 -0700167
Stephen Hemminger06801912007-08-10 15:22:13 -0700168static inline struct tnode *node_parent(struct node *node)
169{
170 struct tnode *ret;
171
172 ret = (struct tnode *)(node->parent & ~NODE_TYPE_MASK);
173 return rcu_dereference(ret);
174}
175
176static inline void node_set_parent(struct node *node, struct tnode *ptr)
177{
178 rcu_assign_pointer(node->parent,
179 (unsigned long)ptr | NODE_TYPE(node));
180}
Robert Olsson2373ce12005-08-25 13:01:29 -0700181
182/* rcu_read_lock needs to be hold by caller from readside */
183
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700184static inline struct node *tnode_get_child(struct tnode *tn, int i)
Robert Olsson19baf832005-06-21 12:43:18 -0700185{
Olof Johansson91b9a272005-08-09 20:24:39 -0700186 BUG_ON(i >= 1 << tn->bits);
Robert Olsson19baf832005-06-21 12:43:18 -0700187
Robert Olsson2373ce12005-08-25 13:01:29 -0700188 return rcu_dereference(tn->child[i]);
Robert Olsson19baf832005-06-21 12:43:18 -0700189}
190
Stephen Hemmignerbb435b82005-08-09 20:25:39 -0700191static inline int tnode_child_length(const struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700192{
Olof Johansson91b9a272005-08-09 20:24:39 -0700193 return 1 << tn->bits;
Robert Olsson19baf832005-06-21 12:43:18 -0700194}
195
Stephen Hemmingerab66b4a2007-08-10 15:22:58 -0700196static inline t_key mask_pfx(t_key k, unsigned short l)
197{
198 return (l == 0) ? 0 : k >> (KEYLENGTH-l) << (KEYLENGTH-l);
199}
200
Robert Olsson19baf832005-06-21 12:43:18 -0700201static inline t_key tkey_extract_bits(t_key a, int offset, int bits)
202{
Olof Johansson91b9a272005-08-09 20:24:39 -0700203 if (offset < KEYLENGTH)
Robert Olsson19baf832005-06-21 12:43:18 -0700204 return ((t_key)(a << offset)) >> (KEYLENGTH - bits);
Olof Johansson91b9a272005-08-09 20:24:39 -0700205 else
Robert Olsson19baf832005-06-21 12:43:18 -0700206 return 0;
207}
208
209static inline int tkey_equals(t_key a, t_key b)
210{
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700211 return a == b;
Robert Olsson19baf832005-06-21 12:43:18 -0700212}
213
214static inline int tkey_sub_equals(t_key a, int offset, int bits, t_key b)
215{
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700216 if (bits == 0 || offset >= KEYLENGTH)
217 return 1;
Olof Johansson91b9a272005-08-09 20:24:39 -0700218 bits = bits > KEYLENGTH ? KEYLENGTH : bits;
219 return ((a ^ b) << offset) >> (KEYLENGTH - bits) == 0;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700220}
Robert Olsson19baf832005-06-21 12:43:18 -0700221
222static inline int tkey_mismatch(t_key a, int offset, t_key b)
223{
224 t_key diff = a ^ b;
225 int i = offset;
226
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700227 if (!diff)
228 return 0;
229 while ((diff << i) >> (KEYLENGTH-1) == 0)
Robert Olsson19baf832005-06-21 12:43:18 -0700230 i++;
231 return i;
232}
233
Robert Olsson19baf832005-06-21 12:43:18 -0700234/*
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900235 To understand this stuff, an understanding of keys and all their bits is
236 necessary. Every node in the trie has a key associated with it, but not
Robert Olsson19baf832005-06-21 12:43:18 -0700237 all of the bits in that key are significant.
238
239 Consider a node 'n' and its parent 'tp'.
240
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900241 If n is a leaf, every bit in its key is significant. Its presence is
242 necessitated by path compression, since during a tree traversal (when
243 searching for a leaf - unless we are doing an insertion) we will completely
244 ignore all skipped bits we encounter. Thus we need to verify, at the end of
245 a potentially successful search, that we have indeed been walking the
Robert Olsson19baf832005-06-21 12:43:18 -0700246 correct key path.
247
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900248 Note that we can never "miss" the correct key in the tree if present by
249 following the wrong path. Path compression ensures that segments of the key
250 that are the same for all keys with a given prefix are skipped, but the
251 skipped part *is* identical for each node in the subtrie below the skipped
252 bit! trie_insert() in this implementation takes care of that - note the
Robert Olsson19baf832005-06-21 12:43:18 -0700253 call to tkey_sub_equals() in trie_insert().
254
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900255 if n is an internal node - a 'tnode' here, the various parts of its key
Robert Olsson19baf832005-06-21 12:43:18 -0700256 have many different meanings.
257
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900258 Example:
Robert Olsson19baf832005-06-21 12:43:18 -0700259 _________________________________________________________________
260 | i | i | i | i | i | i | i | N | N | N | S | S | S | S | S | C |
261 -----------------------------------------------------------------
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900262 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
Robert Olsson19baf832005-06-21 12:43:18 -0700263
264 _________________________________________________________________
265 | C | C | C | u | u | u | u | u | u | u | u | u | u | u | u | u |
266 -----------------------------------------------------------------
267 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
268
269 tp->pos = 7
270 tp->bits = 3
271 n->pos = 15
Olof Johansson91b9a272005-08-09 20:24:39 -0700272 n->bits = 4
Robert Olsson19baf832005-06-21 12:43:18 -0700273
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900274 First, let's just ignore the bits that come before the parent tp, that is
275 the bits from 0 to (tp->pos-1). They are *known* but at this point we do
Robert Olsson19baf832005-06-21 12:43:18 -0700276 not use them for anything.
277
278 The bits from (tp->pos) to (tp->pos + tp->bits - 1) - "N", above - are the
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900279 index into the parent's child array. That is, they will be used to find
Robert Olsson19baf832005-06-21 12:43:18 -0700280 'n' among tp's children.
281
282 The bits from (tp->pos + tp->bits) to (n->pos - 1) - "S" - are skipped bits
283 for the node n.
284
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900285 All the bits we have seen so far are significant to the node n. The rest
Robert Olsson19baf832005-06-21 12:43:18 -0700286 of the bits are really not needed or indeed known in n->key.
287
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900288 The bits from (n->pos) to (n->pos + n->bits - 1) - "C" - are the index into
Robert Olsson19baf832005-06-21 12:43:18 -0700289 n's child array, and will of course be different for each child.
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900290
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700291
Robert Olsson19baf832005-06-21 12:43:18 -0700292 The rest of the bits, from (n->pos + n->bits) onward, are completely unknown
293 at this point.
294
295*/
296
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700297static inline void check_tnode(const struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700298{
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700299 WARN_ON(tn && tn->pos+tn->bits > 32);
Robert Olsson19baf832005-06-21 12:43:18 -0700300}
301
Denis V. Lunevf5026fa2007-12-13 09:47:57 -0800302static const int halve_threshold = 25;
303static const int inflate_threshold = 50;
304static const int halve_threshold_root = 8;
305static const int inflate_threshold_root = 15;
Robert Olsson19baf832005-06-21 12:43:18 -0700306
Robert Olsson2373ce12005-08-25 13:01:29 -0700307
308static void __alias_free_mem(struct rcu_head *head)
309{
310 struct fib_alias *fa = container_of(head, struct fib_alias, rcu);
311 kmem_cache_free(fn_alias_kmem, fa);
312}
313
314static inline void alias_free_mem_rcu(struct fib_alias *fa)
315{
316 call_rcu(&fa->rcu, __alias_free_mem);
317}
318
319static void __leaf_free_rcu(struct rcu_head *head)
320{
321 kfree(container_of(head, struct leaf, rcu));
322}
323
Robert Olsson2373ce12005-08-25 13:01:29 -0700324static void __leaf_info_free_rcu(struct rcu_head *head)
325{
326 kfree(container_of(head, struct leaf_info, rcu));
327}
328
329static inline void free_leaf_info(struct leaf_info *leaf)
330{
331 call_rcu(&leaf->rcu, __leaf_info_free_rcu);
332}
333
334static struct tnode *tnode_alloc(unsigned int size)
335{
336 struct page *pages;
337
338 if (size <= PAGE_SIZE)
339 return kcalloc(size, 1, GFP_KERNEL);
340
341 pages = alloc_pages(GFP_KERNEL|__GFP_ZERO, get_order(size));
342 if (!pages)
343 return NULL;
344
345 return page_address(pages);
346}
347
348static void __tnode_free_rcu(struct rcu_head *head)
349{
350 struct tnode *tn = container_of(head, struct tnode, rcu);
351 unsigned int size = sizeof(struct tnode) +
352 (1 << tn->bits) * sizeof(struct node *);
353
354 if (size <= PAGE_SIZE)
355 kfree(tn);
356 else
357 free_pages((unsigned long)tn, get_order(size));
358}
359
360static inline void tnode_free(struct tnode *tn)
361{
Stephen Hemminger132adf52007-03-08 20:44:43 -0800362 if (IS_LEAF(tn)) {
Robert Olsson550e29b2006-04-04 12:53:35 -0700363 struct leaf *l = (struct leaf *) tn;
364 call_rcu_bh(&l->rcu, __leaf_free_rcu);
Stephen Hemminger132adf52007-03-08 20:44:43 -0800365 } else
Robert Olsson550e29b2006-04-04 12:53:35 -0700366 call_rcu(&tn->rcu, __tnode_free_rcu);
Robert Olsson2373ce12005-08-25 13:01:29 -0700367}
368
Robert Olsson19baf832005-06-21 12:43:18 -0700369static struct leaf *leaf_new(void)
370{
371 struct leaf *l = kmalloc(sizeof(struct leaf), GFP_KERNEL);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700372 if (l) {
Robert Olsson2373ce12005-08-25 13:01:29 -0700373 l->parent = T_LEAF;
Robert Olsson19baf832005-06-21 12:43:18 -0700374 INIT_HLIST_HEAD(&l->list);
375 }
376 return l;
377}
378
379static struct leaf_info *leaf_info_new(int plen)
380{
381 struct leaf_info *li = kmalloc(sizeof(struct leaf_info), GFP_KERNEL);
Robert Olsson2373ce12005-08-25 13:01:29 -0700382 if (li) {
383 li->plen = plen;
384 INIT_LIST_HEAD(&li->falh);
Patrick McHardyf0e36f82005-07-05 14:44:55 -0700385 }
Robert Olsson2373ce12005-08-25 13:01:29 -0700386 return li;
Patrick McHardyf0e36f82005-07-05 14:44:55 -0700387}
388
Robert Olsson19baf832005-06-21 12:43:18 -0700389static struct tnode* tnode_new(t_key key, int pos, int bits)
390{
391 int nchildren = 1<<bits;
392 int sz = sizeof(struct tnode) + nchildren * sizeof(struct node *);
Patrick McHardyf0e36f82005-07-05 14:44:55 -0700393 struct tnode *tn = tnode_alloc(sz);
Robert Olsson19baf832005-06-21 12:43:18 -0700394
Olof Johansson91b9a272005-08-09 20:24:39 -0700395 if (tn) {
Robert Olsson19baf832005-06-21 12:43:18 -0700396 memset(tn, 0, sz);
Robert Olsson2373ce12005-08-25 13:01:29 -0700397 tn->parent = T_TNODE;
Robert Olsson19baf832005-06-21 12:43:18 -0700398 tn->pos = pos;
399 tn->bits = bits;
400 tn->key = key;
401 tn->full_children = 0;
402 tn->empty_children = 1<<bits;
403 }
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700404
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700405 pr_debug("AT %p s=%u %u\n", tn, (unsigned int) sizeof(struct tnode),
406 (unsigned int) (sizeof(struct node) * 1<<bits));
Robert Olsson19baf832005-06-21 12:43:18 -0700407 return tn;
408}
409
Robert Olsson19baf832005-06-21 12:43:18 -0700410/*
411 * Check whether a tnode 'n' is "full", i.e. it is an internal node
412 * and no bits are skipped. See discussion in dyntree paper p. 6
413 */
414
Stephen Hemmignerbb435b82005-08-09 20:25:39 -0700415static inline int tnode_full(const struct tnode *tn, const struct node *n)
Robert Olsson19baf832005-06-21 12:43:18 -0700416{
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700417 if (n == NULL || IS_LEAF(n))
Robert Olsson19baf832005-06-21 12:43:18 -0700418 return 0;
419
420 return ((struct tnode *) n)->pos == tn->pos + tn->bits;
421}
422
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700423static inline void put_child(struct trie *t, struct tnode *tn, int i, struct node *n)
Robert Olsson19baf832005-06-21 12:43:18 -0700424{
425 tnode_put_child_reorg(tn, i, n, -1);
426}
427
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700428 /*
Robert Olsson19baf832005-06-21 12:43:18 -0700429 * Add a child at position i overwriting the old value.
430 * Update the value of full_children and empty_children.
431 */
432
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700433static void tnode_put_child_reorg(struct tnode *tn, int i, struct node *n, int wasfull)
Robert Olsson19baf832005-06-21 12:43:18 -0700434{
Robert Olsson2373ce12005-08-25 13:01:29 -0700435 struct node *chi = tn->child[i];
Robert Olsson19baf832005-06-21 12:43:18 -0700436 int isfull;
437
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700438 BUG_ON(i >= 1<<tn->bits);
439
Robert Olsson19baf832005-06-21 12:43:18 -0700440
441 /* update emptyChildren */
442 if (n == NULL && chi != NULL)
443 tn->empty_children++;
444 else if (n != NULL && chi == NULL)
445 tn->empty_children--;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700446
Robert Olsson19baf832005-06-21 12:43:18 -0700447 /* update fullChildren */
Olof Johansson91b9a272005-08-09 20:24:39 -0700448 if (wasfull == -1)
Robert Olsson19baf832005-06-21 12:43:18 -0700449 wasfull = tnode_full(tn, chi);
450
451 isfull = tnode_full(tn, n);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700452 if (wasfull && !isfull)
Robert Olsson19baf832005-06-21 12:43:18 -0700453 tn->full_children--;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700454 else if (!wasfull && isfull)
Robert Olsson19baf832005-06-21 12:43:18 -0700455 tn->full_children++;
Olof Johansson91b9a272005-08-09 20:24:39 -0700456
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700457 if (n)
Stephen Hemminger06801912007-08-10 15:22:13 -0700458 node_set_parent(n, tn);
Robert Olsson19baf832005-06-21 12:43:18 -0700459
Robert Olsson2373ce12005-08-25 13:01:29 -0700460 rcu_assign_pointer(tn->child[i], n);
Robert Olsson19baf832005-06-21 12:43:18 -0700461}
462
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700463static struct node *resize(struct trie *t, struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700464{
465 int i;
Robert Olsson2f368952005-07-05 15:02:40 -0700466 int err = 0;
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700467 struct tnode *old_tn;
Robert Olssone6308be2005-10-04 13:01:58 -0700468 int inflate_threshold_use;
469 int halve_threshold_use;
Robert Olsson05eee482007-03-19 16:27:37 -0700470 int max_resize;
Robert Olsson19baf832005-06-21 12:43:18 -0700471
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900472 if (!tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700473 return NULL;
474
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700475 pr_debug("In tnode_resize %p inflate_threshold=%d threshold=%d\n",
476 tn, inflate_threshold, halve_threshold);
Robert Olsson19baf832005-06-21 12:43:18 -0700477
478 /* No children */
479 if (tn->empty_children == tnode_child_length(tn)) {
480 tnode_free(tn);
481 return NULL;
482 }
483 /* One child */
484 if (tn->empty_children == tnode_child_length(tn) - 1)
485 for (i = 0; i < tnode_child_length(tn); i++) {
Olof Johansson91b9a272005-08-09 20:24:39 -0700486 struct node *n;
Robert Olsson19baf832005-06-21 12:43:18 -0700487
Olof Johansson91b9a272005-08-09 20:24:39 -0700488 n = tn->child[i];
Robert Olsson2373ce12005-08-25 13:01:29 -0700489 if (!n)
Olof Johansson91b9a272005-08-09 20:24:39 -0700490 continue;
Olof Johansson91b9a272005-08-09 20:24:39 -0700491
492 /* compress one level */
Stephen Hemminger06801912007-08-10 15:22:13 -0700493 node_set_parent(n, NULL);
Olof Johansson91b9a272005-08-09 20:24:39 -0700494 tnode_free(tn);
495 return n;
Robert Olsson19baf832005-06-21 12:43:18 -0700496 }
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700497 /*
Robert Olsson19baf832005-06-21 12:43:18 -0700498 * Double as long as the resulting node has a number of
499 * nonempty nodes that are above the threshold.
500 */
501
502 /*
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700503 * From "Implementing a dynamic compressed trie" by Stefan Nilsson of
504 * the Helsinki University of Technology and Matti Tikkanen of Nokia
Robert Olsson19baf832005-06-21 12:43:18 -0700505 * Telecommunications, page 6:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700506 * "A node is doubled if the ratio of non-empty children to all
Robert Olsson19baf832005-06-21 12:43:18 -0700507 * children in the *doubled* node is at least 'high'."
508 *
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700509 * 'high' in this instance is the variable 'inflate_threshold'. It
510 * is expressed as a percentage, so we multiply it with
511 * tnode_child_length() and instead of multiplying by 2 (since the
512 * child array will be doubled by inflate()) and multiplying
513 * the left-hand side by 100 (to handle the percentage thing) we
Robert Olsson19baf832005-06-21 12:43:18 -0700514 * multiply the left-hand side by 50.
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700515 *
516 * The left-hand side may look a bit weird: tnode_child_length(tn)
517 * - tn->empty_children is of course the number of non-null children
518 * in the current node. tn->full_children is the number of "full"
Robert Olsson19baf832005-06-21 12:43:18 -0700519 * children, that is non-null tnodes with a skip value of 0.
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700520 * All of those will be doubled in the resulting inflated tnode, so
Robert Olsson19baf832005-06-21 12:43:18 -0700521 * we just count them one extra time here.
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700522 *
Robert Olsson19baf832005-06-21 12:43:18 -0700523 * A clearer way to write this would be:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700524 *
Robert Olsson19baf832005-06-21 12:43:18 -0700525 * to_be_doubled = tn->full_children;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700526 * not_to_be_doubled = tnode_child_length(tn) - tn->empty_children -
Robert Olsson19baf832005-06-21 12:43:18 -0700527 * tn->full_children;
528 *
529 * new_child_length = tnode_child_length(tn) * 2;
530 *
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700531 * new_fill_factor = 100 * (not_to_be_doubled + 2*to_be_doubled) /
Robert Olsson19baf832005-06-21 12:43:18 -0700532 * new_child_length;
533 * if (new_fill_factor >= inflate_threshold)
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700534 *
535 * ...and so on, tho it would mess up the while () loop.
536 *
Robert Olsson19baf832005-06-21 12:43:18 -0700537 * anyway,
538 * 100 * (not_to_be_doubled + 2*to_be_doubled) / new_child_length >=
539 * inflate_threshold
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700540 *
Robert Olsson19baf832005-06-21 12:43:18 -0700541 * avoid a division:
542 * 100 * (not_to_be_doubled + 2*to_be_doubled) >=
543 * inflate_threshold * new_child_length
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700544 *
Robert Olsson19baf832005-06-21 12:43:18 -0700545 * expand not_to_be_doubled and to_be_doubled, and shorten:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700546 * 100 * (tnode_child_length(tn) - tn->empty_children +
Olof Johansson91b9a272005-08-09 20:24:39 -0700547 * tn->full_children) >= inflate_threshold * new_child_length
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700548 *
Robert Olsson19baf832005-06-21 12:43:18 -0700549 * expand new_child_length:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700550 * 100 * (tnode_child_length(tn) - tn->empty_children +
Olof Johansson91b9a272005-08-09 20:24:39 -0700551 * tn->full_children) >=
Robert Olsson19baf832005-06-21 12:43:18 -0700552 * inflate_threshold * tnode_child_length(tn) * 2
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700553 *
Robert Olsson19baf832005-06-21 12:43:18 -0700554 * shorten again:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700555 * 50 * (tn->full_children + tnode_child_length(tn) -
Olof Johansson91b9a272005-08-09 20:24:39 -0700556 * tn->empty_children) >= inflate_threshold *
Robert Olsson19baf832005-06-21 12:43:18 -0700557 * tnode_child_length(tn)
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700558 *
Robert Olsson19baf832005-06-21 12:43:18 -0700559 */
560
561 check_tnode(tn);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700562
Robert Olssone6308be2005-10-04 13:01:58 -0700563 /* Keep root node larger */
564
Stephen Hemminger132adf52007-03-08 20:44:43 -0800565 if (!tn->parent)
Robert Olssone6308be2005-10-04 13:01:58 -0700566 inflate_threshold_use = inflate_threshold_root;
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900567 else
Robert Olssone6308be2005-10-04 13:01:58 -0700568 inflate_threshold_use = inflate_threshold;
569
Robert Olsson2f368952005-07-05 15:02:40 -0700570 err = 0;
Robert Olsson05eee482007-03-19 16:27:37 -0700571 max_resize = 10;
572 while ((tn->full_children > 0 && max_resize-- &&
Robert Olsson19baf832005-06-21 12:43:18 -0700573 50 * (tn->full_children + tnode_child_length(tn) - tn->empty_children) >=
Robert Olssone6308be2005-10-04 13:01:58 -0700574 inflate_threshold_use * tnode_child_length(tn))) {
Robert Olsson19baf832005-06-21 12:43:18 -0700575
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700576 old_tn = tn;
577 tn = inflate(t, tn);
578 if (IS_ERR(tn)) {
579 tn = old_tn;
Robert Olsson2f368952005-07-05 15:02:40 -0700580#ifdef CONFIG_IP_FIB_TRIE_STATS
581 t->stats.resize_node_skipped++;
582#endif
583 break;
584 }
Robert Olsson19baf832005-06-21 12:43:18 -0700585 }
586
Robert Olsson05eee482007-03-19 16:27:37 -0700587 if (max_resize < 0) {
588 if (!tn->parent)
589 printk(KERN_WARNING "Fix inflate_threshold_root. Now=%d size=%d bits\n",
590 inflate_threshold_root, tn->bits);
591 else
592 printk(KERN_WARNING "Fix inflate_threshold. Now=%d size=%d bits\n",
593 inflate_threshold, tn->bits);
594 }
595
Robert Olsson19baf832005-06-21 12:43:18 -0700596 check_tnode(tn);
597
598 /*
599 * Halve as long as the number of empty children in this
600 * node is above threshold.
601 */
Robert Olsson2f368952005-07-05 15:02:40 -0700602
Robert Olssone6308be2005-10-04 13:01:58 -0700603
604 /* Keep root node larger */
605
Stephen Hemminger132adf52007-03-08 20:44:43 -0800606 if (!tn->parent)
Robert Olssone6308be2005-10-04 13:01:58 -0700607 halve_threshold_use = halve_threshold_root;
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900608 else
Robert Olssone6308be2005-10-04 13:01:58 -0700609 halve_threshold_use = halve_threshold;
610
Robert Olsson2f368952005-07-05 15:02:40 -0700611 err = 0;
Robert Olsson05eee482007-03-19 16:27:37 -0700612 max_resize = 10;
613 while (tn->bits > 1 && max_resize-- &&
Robert Olsson19baf832005-06-21 12:43:18 -0700614 100 * (tnode_child_length(tn) - tn->empty_children) <
Robert Olssone6308be2005-10-04 13:01:58 -0700615 halve_threshold_use * tnode_child_length(tn)) {
Robert Olsson19baf832005-06-21 12:43:18 -0700616
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700617 old_tn = tn;
618 tn = halve(t, tn);
619 if (IS_ERR(tn)) {
620 tn = old_tn;
Robert Olsson2f368952005-07-05 15:02:40 -0700621#ifdef CONFIG_IP_FIB_TRIE_STATS
622 t->stats.resize_node_skipped++;
623#endif
624 break;
625 }
626 }
627
Robert Olsson05eee482007-03-19 16:27:37 -0700628 if (max_resize < 0) {
629 if (!tn->parent)
630 printk(KERN_WARNING "Fix halve_threshold_root. Now=%d size=%d bits\n",
631 halve_threshold_root, tn->bits);
632 else
633 printk(KERN_WARNING "Fix halve_threshold. Now=%d size=%d bits\n",
634 halve_threshold, tn->bits);
635 }
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700636
Robert Olsson19baf832005-06-21 12:43:18 -0700637 /* Only one child remains */
Robert Olsson19baf832005-06-21 12:43:18 -0700638 if (tn->empty_children == tnode_child_length(tn) - 1)
639 for (i = 0; i < tnode_child_length(tn); i++) {
Olof Johansson91b9a272005-08-09 20:24:39 -0700640 struct node *n;
641
Olof Johansson91b9a272005-08-09 20:24:39 -0700642 n = tn->child[i];
Robert Olsson2373ce12005-08-25 13:01:29 -0700643 if (!n)
Olof Johansson91b9a272005-08-09 20:24:39 -0700644 continue;
Olof Johansson91b9a272005-08-09 20:24:39 -0700645
646 /* compress one level */
647
Stephen Hemminger06801912007-08-10 15:22:13 -0700648 node_set_parent(n, NULL);
Olof Johansson91b9a272005-08-09 20:24:39 -0700649 tnode_free(tn);
650 return n;
Robert Olsson19baf832005-06-21 12:43:18 -0700651 }
652
653 return (struct node *) tn;
654}
655
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700656static struct tnode *inflate(struct trie *t, struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700657{
658 struct tnode *inode;
659 struct tnode *oldtnode = tn;
660 int olen = tnode_child_length(tn);
661 int i;
662
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700663 pr_debug("In inflate\n");
Robert Olsson19baf832005-06-21 12:43:18 -0700664
665 tn = tnode_new(oldtnode->key, oldtnode->pos, oldtnode->bits + 1);
666
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700667 if (!tn)
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700668 return ERR_PTR(-ENOMEM);
Robert Olsson2f368952005-07-05 15:02:40 -0700669
670 /*
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700671 * Preallocate and store tnodes before the actual work so we
672 * don't get into an inconsistent state if memory allocation
673 * fails. In case of failure we return the oldnode and inflate
Robert Olsson2f368952005-07-05 15:02:40 -0700674 * of tnode is ignored.
675 */
Olof Johansson91b9a272005-08-09 20:24:39 -0700676
677 for (i = 0; i < olen; i++) {
Robert Olsson2f368952005-07-05 15:02:40 -0700678 struct tnode *inode = (struct tnode *) tnode_get_child(oldtnode, i);
679
680 if (inode &&
681 IS_TNODE(inode) &&
682 inode->pos == oldtnode->pos + oldtnode->bits &&
683 inode->bits > 1) {
684 struct tnode *left, *right;
Stephen Hemmingerab66b4a2007-08-10 15:22:58 -0700685 t_key m = ~0U << (KEYLENGTH - 1) >> inode->pos;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700686
Robert Olsson2f368952005-07-05 15:02:40 -0700687 left = tnode_new(inode->key&(~m), inode->pos + 1,
688 inode->bits - 1);
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700689 if (!left)
690 goto nomem;
Olof Johansson91b9a272005-08-09 20:24:39 -0700691
Robert Olsson2f368952005-07-05 15:02:40 -0700692 right = tnode_new(inode->key|m, inode->pos + 1,
693 inode->bits - 1);
694
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900695 if (!right) {
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700696 tnode_free(left);
697 goto nomem;
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900698 }
Robert Olsson2f368952005-07-05 15:02:40 -0700699
700 put_child(t, tn, 2*i, (struct node *) left);
701 put_child(t, tn, 2*i+1, (struct node *) right);
702 }
703 }
704
Olof Johansson91b9a272005-08-09 20:24:39 -0700705 for (i = 0; i < olen; i++) {
Robert Olsson19baf832005-06-21 12:43:18 -0700706 struct node *node = tnode_get_child(oldtnode, i);
Olof Johansson91b9a272005-08-09 20:24:39 -0700707 struct tnode *left, *right;
708 int size, j;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700709
Robert Olsson19baf832005-06-21 12:43:18 -0700710 /* An empty child */
711 if (node == NULL)
712 continue;
713
714 /* A leaf or an internal node with skipped bits */
715
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700716 if (IS_LEAF(node) || ((struct tnode *) node)->pos >
Robert Olsson19baf832005-06-21 12:43:18 -0700717 tn->pos + tn->bits - 1) {
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700718 if (tkey_extract_bits(node->key, oldtnode->pos + oldtnode->bits,
Robert Olsson19baf832005-06-21 12:43:18 -0700719 1) == 0)
720 put_child(t, tn, 2*i, node);
721 else
722 put_child(t, tn, 2*i+1, node);
723 continue;
724 }
725
726 /* An internal node with two children */
727 inode = (struct tnode *) node;
728
729 if (inode->bits == 1) {
730 put_child(t, tn, 2*i, inode->child[0]);
731 put_child(t, tn, 2*i+1, inode->child[1]);
732
733 tnode_free(inode);
Olof Johansson91b9a272005-08-09 20:24:39 -0700734 continue;
Robert Olsson19baf832005-06-21 12:43:18 -0700735 }
736
Olof Johansson91b9a272005-08-09 20:24:39 -0700737 /* An internal node with more than two children */
Robert Olsson19baf832005-06-21 12:43:18 -0700738
Olof Johansson91b9a272005-08-09 20:24:39 -0700739 /* We will replace this node 'inode' with two new
740 * ones, 'left' and 'right', each with half of the
741 * original children. The two new nodes will have
742 * a position one bit further down the key and this
743 * means that the "significant" part of their keys
744 * (see the discussion near the top of this file)
745 * will differ by one bit, which will be "0" in
746 * left's key and "1" in right's key. Since we are
747 * moving the key position by one step, the bit that
748 * we are moving away from - the bit at position
749 * (inode->pos) - is the one that will differ between
750 * left and right. So... we synthesize that bit in the
751 * two new keys.
752 * The mask 'm' below will be a single "one" bit at
753 * the position (inode->pos)
754 */
Robert Olsson19baf832005-06-21 12:43:18 -0700755
Olof Johansson91b9a272005-08-09 20:24:39 -0700756 /* Use the old key, but set the new significant
757 * bit to zero.
758 */
Robert Olsson19baf832005-06-21 12:43:18 -0700759
Olof Johansson91b9a272005-08-09 20:24:39 -0700760 left = (struct tnode *) tnode_get_child(tn, 2*i);
761 put_child(t, tn, 2*i, NULL);
Robert Olsson19baf832005-06-21 12:43:18 -0700762
Olof Johansson91b9a272005-08-09 20:24:39 -0700763 BUG_ON(!left);
Robert Olsson2f368952005-07-05 15:02:40 -0700764
Olof Johansson91b9a272005-08-09 20:24:39 -0700765 right = (struct tnode *) tnode_get_child(tn, 2*i+1);
766 put_child(t, tn, 2*i+1, NULL);
Robert Olsson2f368952005-07-05 15:02:40 -0700767
Olof Johansson91b9a272005-08-09 20:24:39 -0700768 BUG_ON(!right);
Robert Olsson2f368952005-07-05 15:02:40 -0700769
Olof Johansson91b9a272005-08-09 20:24:39 -0700770 size = tnode_child_length(left);
771 for (j = 0; j < size; j++) {
772 put_child(t, left, j, inode->child[j]);
773 put_child(t, right, j, inode->child[j + size]);
Robert Olsson19baf832005-06-21 12:43:18 -0700774 }
Olof Johansson91b9a272005-08-09 20:24:39 -0700775 put_child(t, tn, 2*i, resize(t, left));
776 put_child(t, tn, 2*i+1, resize(t, right));
777
778 tnode_free(inode);
Robert Olsson19baf832005-06-21 12:43:18 -0700779 }
780 tnode_free(oldtnode);
781 return tn;
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700782nomem:
783 {
784 int size = tnode_child_length(tn);
785 int j;
786
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700787 for (j = 0; j < size; j++)
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700788 if (tn->child[j])
789 tnode_free((struct tnode *)tn->child[j]);
790
791 tnode_free(tn);
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700792
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700793 return ERR_PTR(-ENOMEM);
794 }
Robert Olsson19baf832005-06-21 12:43:18 -0700795}
796
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700797static struct tnode *halve(struct trie *t, struct tnode *tn)
Robert Olsson19baf832005-06-21 12:43:18 -0700798{
799 struct tnode *oldtnode = tn;
800 struct node *left, *right;
801 int i;
802 int olen = tnode_child_length(tn);
803
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700804 pr_debug("In halve\n");
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700805
806 tn = tnode_new(oldtnode->key, oldtnode->pos, oldtnode->bits - 1);
Robert Olsson19baf832005-06-21 12:43:18 -0700807
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700808 if (!tn)
809 return ERR_PTR(-ENOMEM);
Robert Olsson2f368952005-07-05 15:02:40 -0700810
811 /*
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700812 * Preallocate and store tnodes before the actual work so we
813 * don't get into an inconsistent state if memory allocation
814 * fails. In case of failure we return the oldnode and halve
Robert Olsson2f368952005-07-05 15:02:40 -0700815 * of tnode is ignored.
816 */
817
Olof Johansson91b9a272005-08-09 20:24:39 -0700818 for (i = 0; i < olen; i += 2) {
Robert Olsson2f368952005-07-05 15:02:40 -0700819 left = tnode_get_child(oldtnode, i);
820 right = tnode_get_child(oldtnode, i+1);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700821
Robert Olsson2f368952005-07-05 15:02:40 -0700822 /* Two nonempty children */
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700823 if (left && right) {
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700824 struct tnode *newn;
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700825
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700826 newn = tnode_new(left->key, tn->pos + tn->bits, 1);
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700827
828 if (!newn)
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700829 goto nomem;
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700830
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700831 put_child(t, tn, i/2, (struct node *)newn);
Robert Olsson2f368952005-07-05 15:02:40 -0700832 }
Robert Olsson2f368952005-07-05 15:02:40 -0700833
Robert Olsson2f368952005-07-05 15:02:40 -0700834 }
Robert Olsson19baf832005-06-21 12:43:18 -0700835
Olof Johansson91b9a272005-08-09 20:24:39 -0700836 for (i = 0; i < olen; i += 2) {
837 struct tnode *newBinNode;
838
Robert Olsson19baf832005-06-21 12:43:18 -0700839 left = tnode_get_child(oldtnode, i);
840 right = tnode_get_child(oldtnode, i+1);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700841
Robert Olsson19baf832005-06-21 12:43:18 -0700842 /* At least one of the children is empty */
843 if (left == NULL) {
844 if (right == NULL) /* Both are empty */
845 continue;
846 put_child(t, tn, i/2, right);
Olof Johansson91b9a272005-08-09 20:24:39 -0700847 continue;
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700848 }
Olof Johansson91b9a272005-08-09 20:24:39 -0700849
850 if (right == NULL) {
Robert Olsson19baf832005-06-21 12:43:18 -0700851 put_child(t, tn, i/2, left);
Olof Johansson91b9a272005-08-09 20:24:39 -0700852 continue;
853 }
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700854
Robert Olsson19baf832005-06-21 12:43:18 -0700855 /* Two nonempty children */
Olof Johansson91b9a272005-08-09 20:24:39 -0700856 newBinNode = (struct tnode *) tnode_get_child(tn, i/2);
857 put_child(t, tn, i/2, NULL);
Olof Johansson91b9a272005-08-09 20:24:39 -0700858 put_child(t, newBinNode, 0, left);
859 put_child(t, newBinNode, 1, right);
860 put_child(t, tn, i/2, resize(t, newBinNode));
Robert Olsson19baf832005-06-21 12:43:18 -0700861 }
862 tnode_free(oldtnode);
863 return tn;
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700864nomem:
865 {
866 int size = tnode_child_length(tn);
867 int j;
868
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700869 for (j = 0; j < size; j++)
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700870 if (tn->child[j])
871 tnode_free((struct tnode *)tn->child[j]);
872
873 tnode_free(tn);
Stephen Hemminger0c7770c2005-08-23 21:59:41 -0700874
Robert Olsson2f80b3c2005-08-09 20:25:06 -0700875 return ERR_PTR(-ENOMEM);
876 }
Robert Olsson19baf832005-06-21 12:43:18 -0700877}
878
Olof Johansson91b9a272005-08-09 20:24:39 -0700879static void trie_init(struct trie *t)
Robert Olsson19baf832005-06-21 12:43:18 -0700880{
Olof Johansson91b9a272005-08-09 20:24:39 -0700881 if (!t)
882 return;
883
884 t->size = 0;
Robert Olsson2373ce12005-08-25 13:01:29 -0700885 rcu_assign_pointer(t->trie, NULL);
Olof Johansson91b9a272005-08-09 20:24:39 -0700886 t->revision = 0;
Robert Olsson19baf832005-06-21 12:43:18 -0700887#ifdef CONFIG_IP_FIB_TRIE_STATS
Olof Johansson91b9a272005-08-09 20:24:39 -0700888 memset(&t->stats, 0, sizeof(struct trie_use_stats));
Robert Olsson19baf832005-06-21 12:43:18 -0700889#endif
Robert Olsson19baf832005-06-21 12:43:18 -0700890}
891
Robert Olsson772cb712005-09-19 15:31:18 -0700892/* readside must use rcu_read_lock currently dump routines
Robert Olsson2373ce12005-08-25 13:01:29 -0700893 via get_fa_head and dump */
894
Robert Olsson772cb712005-09-19 15:31:18 -0700895static struct leaf_info *find_leaf_info(struct leaf *l, int plen)
Robert Olsson19baf832005-06-21 12:43:18 -0700896{
Robert Olsson772cb712005-09-19 15:31:18 -0700897 struct hlist_head *head = &l->list;
Robert Olsson19baf832005-06-21 12:43:18 -0700898 struct hlist_node *node;
899 struct leaf_info *li;
900
Robert Olsson2373ce12005-08-25 13:01:29 -0700901 hlist_for_each_entry_rcu(li, node, head, hlist)
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700902 if (li->plen == plen)
Robert Olsson19baf832005-06-21 12:43:18 -0700903 return li;
Olof Johansson91b9a272005-08-09 20:24:39 -0700904
Robert Olsson19baf832005-06-21 12:43:18 -0700905 return NULL;
906}
907
908static inline struct list_head * get_fa_head(struct leaf *l, int plen)
909{
Robert Olsson772cb712005-09-19 15:31:18 -0700910 struct leaf_info *li = find_leaf_info(l, plen);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700911
Olof Johansson91b9a272005-08-09 20:24:39 -0700912 if (!li)
913 return NULL;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700914
Olof Johansson91b9a272005-08-09 20:24:39 -0700915 return &li->falh;
Robert Olsson19baf832005-06-21 12:43:18 -0700916}
917
918static void insert_leaf_info(struct hlist_head *head, struct leaf_info *new)
919{
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900920 struct leaf_info *li = NULL, *last = NULL;
921 struct hlist_node *node;
Robert Olsson19baf832005-06-21 12:43:18 -0700922
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900923 if (hlist_empty(head)) {
924 hlist_add_head_rcu(&new->hlist, head);
925 } else {
926 hlist_for_each_entry(li, node, head, hlist) {
927 if (new->plen > li->plen)
928 break;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700929
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +0900930 last = li;
931 }
932 if (last)
933 hlist_add_after_rcu(&last->hlist, &new->hlist);
934 else
935 hlist_add_before_rcu(&new->hlist, &li->hlist);
936 }
Robert Olsson19baf832005-06-21 12:43:18 -0700937}
938
Robert Olsson2373ce12005-08-25 13:01:29 -0700939/* rcu_read_lock needs to be hold by caller from readside */
940
Robert Olsson19baf832005-06-21 12:43:18 -0700941static struct leaf *
942fib_find_node(struct trie *t, u32 key)
943{
944 int pos;
945 struct tnode *tn;
946 struct node *n;
947
948 pos = 0;
Robert Olsson2373ce12005-08-25 13:01:29 -0700949 n = rcu_dereference(t->trie);
Robert Olsson19baf832005-06-21 12:43:18 -0700950
951 while (n != NULL && NODE_TYPE(n) == T_TNODE) {
952 tn = (struct tnode *) n;
Olof Johansson91b9a272005-08-09 20:24:39 -0700953
Robert Olsson19baf832005-06-21 12:43:18 -0700954 check_tnode(tn);
Olof Johansson91b9a272005-08-09 20:24:39 -0700955
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700956 if (tkey_sub_equals(tn->key, pos, tn->pos-pos, key)) {
Olof Johansson91b9a272005-08-09 20:24:39 -0700957 pos = tn->pos + tn->bits;
Robert Olsson19baf832005-06-21 12:43:18 -0700958 n = tnode_get_child(tn, tkey_extract_bits(key, tn->pos, tn->bits));
Olof Johansson91b9a272005-08-09 20:24:39 -0700959 } else
Robert Olsson19baf832005-06-21 12:43:18 -0700960 break;
961 }
962 /* Case we have found a leaf. Compare prefixes */
963
Olof Johansson91b9a272005-08-09 20:24:39 -0700964 if (n != NULL && IS_LEAF(n) && tkey_equals(key, n->key))
965 return (struct leaf *)n;
966
Robert Olsson19baf832005-06-21 12:43:18 -0700967 return NULL;
968}
969
970static struct node *trie_rebalance(struct trie *t, struct tnode *tn)
971{
Robert Olsson19baf832005-06-21 12:43:18 -0700972 int wasfull;
Stephen Hemminger06801912007-08-10 15:22:13 -0700973 t_key cindex, key = tn->key;
974 struct tnode *tp;
Robert Olsson19baf832005-06-21 12:43:18 -0700975
Stephen Hemminger06801912007-08-10 15:22:13 -0700976 while (tn != NULL && (tp = node_parent((struct node *)tn)) != NULL) {
Robert Olsson19baf832005-06-21 12:43:18 -0700977 cindex = tkey_extract_bits(key, tp->pos, tp->bits);
978 wasfull = tnode_full(tp, tnode_get_child(tp, cindex));
979 tn = (struct tnode *) resize (t, (struct tnode *)tn);
980 tnode_put_child_reorg((struct tnode *)tp, cindex,(struct node*)tn, wasfull);
Olof Johansson91b9a272005-08-09 20:24:39 -0700981
Stephen Hemminger06801912007-08-10 15:22:13 -0700982 tp = node_parent((struct node *) tn);
983 if (!tp)
Robert Olsson19baf832005-06-21 12:43:18 -0700984 break;
Stephen Hemminger06801912007-08-10 15:22:13 -0700985 tn = tp;
Robert Olsson19baf832005-06-21 12:43:18 -0700986 }
Stephen Hemminger06801912007-08-10 15:22:13 -0700987
Robert Olsson19baf832005-06-21 12:43:18 -0700988 /* Handle last (top) tnode */
Stephen Hemmingerc877efb2005-07-19 14:01:51 -0700989 if (IS_TNODE(tn))
Robert Olsson19baf832005-06-21 12:43:18 -0700990 tn = (struct tnode*) resize(t, (struct tnode *)tn);
991
992 return (struct node*) tn;
993}
994
Robert Olsson2373ce12005-08-25 13:01:29 -0700995/* only used from updater-side */
996
Robert Olssonf835e472005-06-28 15:00:39 -0700997static struct list_head *
998fib_insert_node(struct trie *t, int *err, u32 key, int plen)
Robert Olsson19baf832005-06-21 12:43:18 -0700999{
1000 int pos, newpos;
1001 struct tnode *tp = NULL, *tn = NULL;
1002 struct node *n;
1003 struct leaf *l;
1004 int missbit;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001005 struct list_head *fa_head = NULL;
Robert Olsson19baf832005-06-21 12:43:18 -07001006 struct leaf_info *li;
1007 t_key cindex;
1008
1009 pos = 0;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001010 n = t->trie;
Robert Olsson19baf832005-06-21 12:43:18 -07001011
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001012 /* If we point to NULL, stop. Either the tree is empty and we should
1013 * just put a new leaf in if, or we have reached an empty child slot,
Robert Olsson19baf832005-06-21 12:43:18 -07001014 * and we should just put our new leaf in that.
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001015 * If we point to a T_TNODE, check if it matches our key. Note that
1016 * a T_TNODE might be skipping any number of bits - its 'pos' need
Robert Olsson19baf832005-06-21 12:43:18 -07001017 * not be the parent's 'pos'+'bits'!
1018 *
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001019 * If it does match the current key, get pos/bits from it, extract
Robert Olsson19baf832005-06-21 12:43:18 -07001020 * the index from our key, push the T_TNODE and walk the tree.
1021 *
1022 * If it doesn't, we have to replace it with a new T_TNODE.
1023 *
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001024 * If we point to a T_LEAF, it might or might not have the same key
1025 * as we do. If it does, just change the value, update the T_LEAF's
1026 * value, and return it.
Robert Olsson19baf832005-06-21 12:43:18 -07001027 * If it doesn't, we need to replace it with a T_TNODE.
1028 */
1029
1030 while (n != NULL && NODE_TYPE(n) == T_TNODE) {
1031 tn = (struct tnode *) n;
Olof Johansson91b9a272005-08-09 20:24:39 -07001032
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001033 check_tnode(tn);
Olof Johansson91b9a272005-08-09 20:24:39 -07001034
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001035 if (tkey_sub_equals(tn->key, pos, tn->pos-pos, key)) {
Robert Olsson19baf832005-06-21 12:43:18 -07001036 tp = tn;
Olof Johansson91b9a272005-08-09 20:24:39 -07001037 pos = tn->pos + tn->bits;
Robert Olsson19baf832005-06-21 12:43:18 -07001038 n = tnode_get_child(tn, tkey_extract_bits(key, tn->pos, tn->bits));
1039
Stephen Hemminger06801912007-08-10 15:22:13 -07001040 BUG_ON(n && node_parent(n) != tn);
Olof Johansson91b9a272005-08-09 20:24:39 -07001041 } else
Robert Olsson19baf832005-06-21 12:43:18 -07001042 break;
1043 }
1044
1045 /*
1046 * n ----> NULL, LEAF or TNODE
1047 *
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001048 * tp is n's (parent) ----> NULL or TNODE
Robert Olsson19baf832005-06-21 12:43:18 -07001049 */
1050
Olof Johansson91b9a272005-08-09 20:24:39 -07001051 BUG_ON(tp && IS_LEAF(tp));
Robert Olsson19baf832005-06-21 12:43:18 -07001052
1053 /* Case 1: n is a leaf. Compare prefixes */
1054
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001055 if (n != NULL && IS_LEAF(n) && tkey_equals(key, n->key)) {
Olof Johansson91b9a272005-08-09 20:24:39 -07001056 struct leaf *l = (struct leaf *) n;
1057
Robert Olsson19baf832005-06-21 12:43:18 -07001058 li = leaf_info_new(plen);
Olof Johansson91b9a272005-08-09 20:24:39 -07001059
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001060 if (!li) {
Robert Olssonf835e472005-06-28 15:00:39 -07001061 *err = -ENOMEM;
1062 goto err;
1063 }
Robert Olsson19baf832005-06-21 12:43:18 -07001064
1065 fa_head = &li->falh;
1066 insert_leaf_info(&l->list, li);
1067 goto done;
1068 }
1069 t->size++;
1070 l = leaf_new();
1071
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001072 if (!l) {
Robert Olssonf835e472005-06-28 15:00:39 -07001073 *err = -ENOMEM;
1074 goto err;
1075 }
Robert Olsson19baf832005-06-21 12:43:18 -07001076
1077 l->key = key;
1078 li = leaf_info_new(plen);
1079
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001080 if (!li) {
Robert Olssonf835e472005-06-28 15:00:39 -07001081 tnode_free((struct tnode *) l);
1082 *err = -ENOMEM;
1083 goto err;
1084 }
Robert Olsson19baf832005-06-21 12:43:18 -07001085
1086 fa_head = &li->falh;
1087 insert_leaf_info(&l->list, li);
1088
Robert Olsson19baf832005-06-21 12:43:18 -07001089 if (t->trie && n == NULL) {
Olof Johansson91b9a272005-08-09 20:24:39 -07001090 /* Case 2: n is NULL, and will just insert a new leaf */
Robert Olsson19baf832005-06-21 12:43:18 -07001091
Stephen Hemminger06801912007-08-10 15:22:13 -07001092 node_set_parent((struct node *)l, tp);
Robert Olsson19baf832005-06-21 12:43:18 -07001093
Olof Johansson91b9a272005-08-09 20:24:39 -07001094 cindex = tkey_extract_bits(key, tp->pos, tp->bits);
1095 put_child(t, (struct tnode *)tp, cindex, (struct node *)l);
1096 } else {
1097 /* Case 3: n is a LEAF or a TNODE and the key doesn't match. */
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001098 /*
1099 * Add a new tnode here
Robert Olsson19baf832005-06-21 12:43:18 -07001100 * first tnode need some special handling
1101 */
1102
1103 if (tp)
Olof Johansson91b9a272005-08-09 20:24:39 -07001104 pos = tp->pos+tp->bits;
Robert Olsson19baf832005-06-21 12:43:18 -07001105 else
Olof Johansson91b9a272005-08-09 20:24:39 -07001106 pos = 0;
1107
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001108 if (n) {
Robert Olsson19baf832005-06-21 12:43:18 -07001109 newpos = tkey_mismatch(key, pos, n->key);
1110 tn = tnode_new(n->key, newpos, 1);
Olof Johansson91b9a272005-08-09 20:24:39 -07001111 } else {
Robert Olsson19baf832005-06-21 12:43:18 -07001112 newpos = 0;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001113 tn = tnode_new(key, newpos, 1); /* First tnode */
Robert Olsson19baf832005-06-21 12:43:18 -07001114 }
Robert Olsson19baf832005-06-21 12:43:18 -07001115
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001116 if (!tn) {
Robert Olssonf835e472005-06-28 15:00:39 -07001117 free_leaf_info(li);
1118 tnode_free((struct tnode *) l);
1119 *err = -ENOMEM;
1120 goto err;
Olof Johansson91b9a272005-08-09 20:24:39 -07001121 }
1122
Stephen Hemminger06801912007-08-10 15:22:13 -07001123 node_set_parent((struct node *)tn, tp);
Robert Olsson19baf832005-06-21 12:43:18 -07001124
Olof Johansson91b9a272005-08-09 20:24:39 -07001125 missbit = tkey_extract_bits(key, newpos, 1);
Robert Olsson19baf832005-06-21 12:43:18 -07001126 put_child(t, tn, missbit, (struct node *)l);
1127 put_child(t, tn, 1-missbit, n);
1128
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001129 if (tp) {
Robert Olsson19baf832005-06-21 12:43:18 -07001130 cindex = tkey_extract_bits(key, tp->pos, tp->bits);
1131 put_child(t, (struct tnode *)tp, cindex, (struct node *)tn);
Olof Johansson91b9a272005-08-09 20:24:39 -07001132 } else {
Robert Olsson2373ce12005-08-25 13:01:29 -07001133 rcu_assign_pointer(t->trie, (struct node *)tn); /* First tnode */
Robert Olsson19baf832005-06-21 12:43:18 -07001134 tp = tn;
1135 }
1136 }
Olof Johansson91b9a272005-08-09 20:24:39 -07001137
1138 if (tp && tp->pos + tp->bits > 32)
Stephen Hemminger78c66712005-09-21 00:15:39 -07001139 printk(KERN_WARNING "fib_trie tp=%p pos=%d, bits=%d, key=%0x plen=%d\n",
Robert Olsson19baf832005-06-21 12:43:18 -07001140 tp, tp->pos, tp->bits, key, plen);
Olof Johansson91b9a272005-08-09 20:24:39 -07001141
Robert Olsson19baf832005-06-21 12:43:18 -07001142 /* Rebalance the trie */
Robert Olsson2373ce12005-08-25 13:01:29 -07001143
1144 rcu_assign_pointer(t->trie, trie_rebalance(t, tp));
Robert Olssonf835e472005-06-28 15:00:39 -07001145done:
1146 t->revision++;
Olof Johansson91b9a272005-08-09 20:24:39 -07001147err:
Robert Olsson19baf832005-06-21 12:43:18 -07001148 return fa_head;
1149}
1150
Robert Olssond562f1f2007-03-26 14:22:22 -07001151/*
1152 * Caller must hold RTNL.
1153 */
Thomas Graf4e902c52006-08-17 18:14:52 -07001154static int fn_trie_insert(struct fib_table *tb, struct fib_config *cfg)
Robert Olsson19baf832005-06-21 12:43:18 -07001155{
1156 struct trie *t = (struct trie *) tb->tb_data;
1157 struct fib_alias *fa, *new_fa;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001158 struct list_head *fa_head = NULL;
Robert Olsson19baf832005-06-21 12:43:18 -07001159 struct fib_info *fi;
Thomas Graf4e902c52006-08-17 18:14:52 -07001160 int plen = cfg->fc_dst_len;
1161 u8 tos = cfg->fc_tos;
Robert Olsson19baf832005-06-21 12:43:18 -07001162 u32 key, mask;
1163 int err;
1164 struct leaf *l;
1165
1166 if (plen > 32)
1167 return -EINVAL;
1168
Thomas Graf4e902c52006-08-17 18:14:52 -07001169 key = ntohl(cfg->fc_dst);
Robert Olsson19baf832005-06-21 12:43:18 -07001170
Patrick McHardy2dfe55b2006-08-10 23:08:33 -07001171 pr_debug("Insert table=%u %08x/%d\n", tb->tb_id, key, plen);
Robert Olsson19baf832005-06-21 12:43:18 -07001172
Olof Johansson91b9a272005-08-09 20:24:39 -07001173 mask = ntohl(inet_make_mask(plen));
Robert Olsson19baf832005-06-21 12:43:18 -07001174
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001175 if (key & ~mask)
Robert Olsson19baf832005-06-21 12:43:18 -07001176 return -EINVAL;
1177
1178 key = key & mask;
1179
Thomas Graf4e902c52006-08-17 18:14:52 -07001180 fi = fib_create_info(cfg);
1181 if (IS_ERR(fi)) {
1182 err = PTR_ERR(fi);
Robert Olsson19baf832005-06-21 12:43:18 -07001183 goto err;
Thomas Graf4e902c52006-08-17 18:14:52 -07001184 }
Robert Olsson19baf832005-06-21 12:43:18 -07001185
1186 l = fib_find_node(t, key);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001187 fa = NULL;
Robert Olsson19baf832005-06-21 12:43:18 -07001188
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001189 if (l) {
Robert Olsson19baf832005-06-21 12:43:18 -07001190 fa_head = get_fa_head(l, plen);
1191 fa = fib_find_alias(fa_head, tos, fi->fib_priority);
1192 }
1193
1194 /* Now fa, if non-NULL, points to the first fib alias
1195 * with the same keys [prefix,tos,priority], if such key already
1196 * exists or to the node before which we will insert new one.
1197 *
1198 * If fa is NULL, we will need to allocate a new one and
1199 * insert to the head of f.
1200 *
1201 * If f is NULL, no fib node matched the destination key
1202 * and we need to allocate a new one of those as well.
1203 */
1204
Olof Johansson91b9a272005-08-09 20:24:39 -07001205 if (fa && fa->fa_info->fib_priority == fi->fib_priority) {
Robert Olsson19baf832005-06-21 12:43:18 -07001206 struct fib_alias *fa_orig;
1207
1208 err = -EEXIST;
Thomas Graf4e902c52006-08-17 18:14:52 -07001209 if (cfg->fc_nlflags & NLM_F_EXCL)
Robert Olsson19baf832005-06-21 12:43:18 -07001210 goto out;
1211
Thomas Graf4e902c52006-08-17 18:14:52 -07001212 if (cfg->fc_nlflags & NLM_F_REPLACE) {
Robert Olsson19baf832005-06-21 12:43:18 -07001213 struct fib_info *fi_drop;
1214 u8 state;
1215
Joonwoo Park67250332008-01-18 03:45:18 -08001216 if (fi->fib_treeref > 1)
1217 goto out;
1218
Robert Olsson2373ce12005-08-25 13:01:29 -07001219 err = -ENOBUFS;
Christoph Lametere94b1762006-12-06 20:33:17 -08001220 new_fa = kmem_cache_alloc(fn_alias_kmem, GFP_KERNEL);
Robert Olsson2373ce12005-08-25 13:01:29 -07001221 if (new_fa == NULL)
1222 goto out;
Robert Olsson19baf832005-06-21 12:43:18 -07001223
1224 fi_drop = fa->fa_info;
Robert Olsson2373ce12005-08-25 13:01:29 -07001225 new_fa->fa_tos = fa->fa_tos;
1226 new_fa->fa_info = fi;
Thomas Graf4e902c52006-08-17 18:14:52 -07001227 new_fa->fa_type = cfg->fc_type;
1228 new_fa->fa_scope = cfg->fc_scope;
Robert Olsson19baf832005-06-21 12:43:18 -07001229 state = fa->fa_state;
Robert Olsson2373ce12005-08-25 13:01:29 -07001230 new_fa->fa_state &= ~FA_S_ACCESSED;
Robert Olsson19baf832005-06-21 12:43:18 -07001231
Robert Olsson2373ce12005-08-25 13:01:29 -07001232 list_replace_rcu(&fa->fa_list, &new_fa->fa_list);
1233 alias_free_mem_rcu(fa);
Robert Olsson19baf832005-06-21 12:43:18 -07001234
1235 fib_release_info(fi_drop);
1236 if (state & FA_S_ACCESSED)
Olof Johansson91b9a272005-08-09 20:24:39 -07001237 rt_cache_flush(-1);
Milan Kocianb8f55832007-05-23 14:55:06 -07001238 rtmsg_fib(RTM_NEWROUTE, htonl(key), new_fa, plen,
1239 tb->tb_id, &cfg->fc_nlinfo, NLM_F_REPLACE);
Robert Olsson19baf832005-06-21 12:43:18 -07001240
Olof Johansson91b9a272005-08-09 20:24:39 -07001241 goto succeeded;
Robert Olsson19baf832005-06-21 12:43:18 -07001242 }
1243 /* Error if we find a perfect match which
1244 * uses the same scope, type, and nexthop
1245 * information.
1246 */
1247 fa_orig = fa;
1248 list_for_each_entry(fa, fa_orig->fa_list.prev, fa_list) {
1249 if (fa->fa_tos != tos)
1250 break;
1251 if (fa->fa_info->fib_priority != fi->fib_priority)
1252 break;
Thomas Graf4e902c52006-08-17 18:14:52 -07001253 if (fa->fa_type == cfg->fc_type &&
1254 fa->fa_scope == cfg->fc_scope &&
Robert Olsson19baf832005-06-21 12:43:18 -07001255 fa->fa_info == fi) {
1256 goto out;
1257 }
1258 }
Thomas Graf4e902c52006-08-17 18:14:52 -07001259 if (!(cfg->fc_nlflags & NLM_F_APPEND))
Robert Olsson19baf832005-06-21 12:43:18 -07001260 fa = fa_orig;
1261 }
1262 err = -ENOENT;
Thomas Graf4e902c52006-08-17 18:14:52 -07001263 if (!(cfg->fc_nlflags & NLM_F_CREATE))
Robert Olsson19baf832005-06-21 12:43:18 -07001264 goto out;
1265
1266 err = -ENOBUFS;
Christoph Lametere94b1762006-12-06 20:33:17 -08001267 new_fa = kmem_cache_alloc(fn_alias_kmem, GFP_KERNEL);
Robert Olsson19baf832005-06-21 12:43:18 -07001268 if (new_fa == NULL)
1269 goto out;
1270
1271 new_fa->fa_info = fi;
1272 new_fa->fa_tos = tos;
Thomas Graf4e902c52006-08-17 18:14:52 -07001273 new_fa->fa_type = cfg->fc_type;
1274 new_fa->fa_scope = cfg->fc_scope;
Robert Olsson19baf832005-06-21 12:43:18 -07001275 new_fa->fa_state = 0;
Robert Olsson19baf832005-06-21 12:43:18 -07001276 /*
1277 * Insert new entry to the list.
1278 */
1279
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001280 if (!fa_head) {
Robert Olssonf835e472005-06-28 15:00:39 -07001281 err = 0;
Herbert Xub47b2ec2006-07-12 13:29:56 -07001282 fa_head = fib_insert_node(t, &err, key, plen);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001283 if (err)
Robert Olssonf835e472005-06-28 15:00:39 -07001284 goto out_free_new_fa;
1285 }
Robert Olsson19baf832005-06-21 12:43:18 -07001286
Robert Olsson2373ce12005-08-25 13:01:29 -07001287 list_add_tail_rcu(&new_fa->fa_list,
1288 (fa ? &fa->fa_list : fa_head));
Robert Olsson19baf832005-06-21 12:43:18 -07001289
1290 rt_cache_flush(-1);
Thomas Graf4e902c52006-08-17 18:14:52 -07001291 rtmsg_fib(RTM_NEWROUTE, htonl(key), new_fa, plen, tb->tb_id,
Milan Kocianb8f55832007-05-23 14:55:06 -07001292 &cfg->fc_nlinfo, 0);
Robert Olsson19baf832005-06-21 12:43:18 -07001293succeeded:
1294 return 0;
Robert Olssonf835e472005-06-28 15:00:39 -07001295
1296out_free_new_fa:
1297 kmem_cache_free(fn_alias_kmem, new_fa);
Robert Olsson19baf832005-06-21 12:43:18 -07001298out:
1299 fib_release_info(fi);
Olof Johansson91b9a272005-08-09 20:24:39 -07001300err:
Robert Olsson19baf832005-06-21 12:43:18 -07001301 return err;
1302}
1303
Robert Olsson2373ce12005-08-25 13:01:29 -07001304
Robert Olsson772cb712005-09-19 15:31:18 -07001305/* should be called with rcu_read_lock */
Stephen Hemminger0c7770c2005-08-23 21:59:41 -07001306static inline int check_leaf(struct trie *t, struct leaf *l,
1307 t_key key, int *plen, const struct flowi *flp,
Patrick McHardy06c74272005-08-23 22:06:09 -07001308 struct fib_result *res)
Robert Olsson19baf832005-06-21 12:43:18 -07001309{
Patrick McHardy06c74272005-08-23 22:06:09 -07001310 int err, i;
Al Viro888454c2006-09-19 13:42:46 -07001311 __be32 mask;
Robert Olsson19baf832005-06-21 12:43:18 -07001312 struct leaf_info *li;
1313 struct hlist_head *hhead = &l->list;
1314 struct hlist_node *node;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001315
Robert Olsson2373ce12005-08-25 13:01:29 -07001316 hlist_for_each_entry_rcu(li, node, hhead, hlist) {
Robert Olsson19baf832005-06-21 12:43:18 -07001317 i = li->plen;
Al Viro888454c2006-09-19 13:42:46 -07001318 mask = inet_make_mask(i);
1319 if (l->key != (key & ntohl(mask)))
Robert Olsson19baf832005-06-21 12:43:18 -07001320 continue;
1321
Al Viro888454c2006-09-19 13:42:46 -07001322 if ((err = fib_semantic_match(&li->falh, flp, res, htonl(l->key), mask, i)) <= 0) {
Robert Olsson19baf832005-06-21 12:43:18 -07001323 *plen = i;
1324#ifdef CONFIG_IP_FIB_TRIE_STATS
1325 t->stats.semantic_match_passed++;
1326#endif
Patrick McHardy06c74272005-08-23 22:06:09 -07001327 return err;
Robert Olsson19baf832005-06-21 12:43:18 -07001328 }
1329#ifdef CONFIG_IP_FIB_TRIE_STATS
1330 t->stats.semantic_match_miss++;
1331#endif
1332 }
Patrick McHardy06c74272005-08-23 22:06:09 -07001333 return 1;
Robert Olsson19baf832005-06-21 12:43:18 -07001334}
1335
1336static int
1337fn_trie_lookup(struct fib_table *tb, const struct flowi *flp, struct fib_result *res)
1338{
1339 struct trie *t = (struct trie *) tb->tb_data;
1340 int plen, ret = 0;
1341 struct node *n;
1342 struct tnode *pn;
1343 int pos, bits;
Olof Johansson91b9a272005-08-09 20:24:39 -07001344 t_key key = ntohl(flp->fl4_dst);
Robert Olsson19baf832005-06-21 12:43:18 -07001345 int chopped_off;
1346 t_key cindex = 0;
1347 int current_prefix_length = KEYLENGTH;
Olof Johansson91b9a272005-08-09 20:24:39 -07001348 struct tnode *cn;
1349 t_key node_prefix, key_prefix, pref_mismatch;
1350 int mp;
1351
Robert Olsson2373ce12005-08-25 13:01:29 -07001352 rcu_read_lock();
Robert Olsson19baf832005-06-21 12:43:18 -07001353
Robert Olsson2373ce12005-08-25 13:01:29 -07001354 n = rcu_dereference(t->trie);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001355 if (!n)
Robert Olsson19baf832005-06-21 12:43:18 -07001356 goto failed;
1357
1358#ifdef CONFIG_IP_FIB_TRIE_STATS
1359 t->stats.gets++;
1360#endif
1361
1362 /* Just a leaf? */
1363 if (IS_LEAF(n)) {
Patrick McHardy06c74272005-08-23 22:06:09 -07001364 if ((ret = check_leaf(t, (struct leaf *)n, key, &plen, flp, res)) <= 0)
Robert Olsson19baf832005-06-21 12:43:18 -07001365 goto found;
1366 goto failed;
1367 }
1368 pn = (struct tnode *) n;
1369 chopped_off = 0;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001370
Olof Johansson91b9a272005-08-09 20:24:39 -07001371 while (pn) {
Robert Olsson19baf832005-06-21 12:43:18 -07001372 pos = pn->pos;
1373 bits = pn->bits;
1374
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001375 if (!chopped_off)
Stephen Hemmingerab66b4a2007-08-10 15:22:58 -07001376 cindex = tkey_extract_bits(mask_pfx(key, current_prefix_length),
1377 pos, bits);
Robert Olsson19baf832005-06-21 12:43:18 -07001378
1379 n = tnode_get_child(pn, cindex);
1380
1381 if (n == NULL) {
1382#ifdef CONFIG_IP_FIB_TRIE_STATS
1383 t->stats.null_node_hit++;
1384#endif
1385 goto backtrace;
1386 }
1387
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001388 if (IS_LEAF(n)) {
Patrick McHardy06c74272005-08-23 22:06:09 -07001389 if ((ret = check_leaf(t, (struct leaf *)n, key, &plen, flp, res)) <= 0)
Robert Olsson19baf832005-06-21 12:43:18 -07001390 goto found;
Olof Johansson91b9a272005-08-09 20:24:39 -07001391 else
1392 goto backtrace;
1393 }
1394
1395#define HL_OPTIMIZE
1396#ifdef HL_OPTIMIZE
1397 cn = (struct tnode *)n;
1398
1399 /*
1400 * It's a tnode, and we can do some extra checks here if we
1401 * like, to avoid descending into a dead-end branch.
1402 * This tnode is in the parent's child array at index
1403 * key[p_pos..p_pos+p_bits] but potentially with some bits
1404 * chopped off, so in reality the index may be just a
1405 * subprefix, padded with zero at the end.
1406 * We can also take a look at any skipped bits in this
1407 * tnode - everything up to p_pos is supposed to be ok,
1408 * and the non-chopped bits of the index (se previous
1409 * paragraph) are also guaranteed ok, but the rest is
1410 * considered unknown.
1411 *
1412 * The skipped bits are key[pos+bits..cn->pos].
1413 */
1414
1415 /* If current_prefix_length < pos+bits, we are already doing
1416 * actual prefix matching, which means everything from
1417 * pos+(bits-chopped_off) onward must be zero along some
1418 * branch of this subtree - otherwise there is *no* valid
1419 * prefix present. Here we can only check the skipped
1420 * bits. Remember, since we have already indexed into the
1421 * parent's child array, we know that the bits we chopped of
1422 * *are* zero.
1423 */
1424
1425 /* NOTA BENE: CHECKING ONLY SKIPPED BITS FOR THE NEW NODE HERE */
1426
1427 if (current_prefix_length < pos+bits) {
1428 if (tkey_extract_bits(cn->key, current_prefix_length,
1429 cn->pos - current_prefix_length) != 0 ||
1430 !(cn->child[0]))
1431 goto backtrace;
1432 }
1433
1434 /*
1435 * If chopped_off=0, the index is fully validated and we
1436 * only need to look at the skipped bits for this, the new,
1437 * tnode. What we actually want to do is to find out if
1438 * these skipped bits match our key perfectly, or if we will
1439 * have to count on finding a matching prefix further down,
1440 * because if we do, we would like to have some way of
1441 * verifying the existence of such a prefix at this point.
1442 */
1443
1444 /* The only thing we can do at this point is to verify that
1445 * any such matching prefix can indeed be a prefix to our
1446 * key, and if the bits in the node we are inspecting that
1447 * do not match our key are not ZERO, this cannot be true.
1448 * Thus, find out where there is a mismatch (before cn->pos)
1449 * and verify that all the mismatching bits are zero in the
1450 * new tnode's key.
1451 */
1452
1453 /* Note: We aren't very concerned about the piece of the key
1454 * that precede pn->pos+pn->bits, since these have already been
1455 * checked. The bits after cn->pos aren't checked since these are
1456 * by definition "unknown" at this point. Thus, what we want to
1457 * see is if we are about to enter the "prefix matching" state,
1458 * and in that case verify that the skipped bits that will prevail
1459 * throughout this subtree are zero, as they have to be if we are
1460 * to find a matching prefix.
1461 */
1462
Stephen Hemmingerab66b4a2007-08-10 15:22:58 -07001463 node_prefix = mask_pfx(cn->key, cn->pos);
1464 key_prefix = mask_pfx(key, cn->pos);
Olof Johansson91b9a272005-08-09 20:24:39 -07001465 pref_mismatch = key_prefix^node_prefix;
1466 mp = 0;
1467
1468 /* In short: If skipped bits in this node do not match the search
1469 * key, enter the "prefix matching" state.directly.
1470 */
1471 if (pref_mismatch) {
1472 while (!(pref_mismatch & (1<<(KEYLENGTH-1)))) {
1473 mp++;
1474 pref_mismatch = pref_mismatch <<1;
1475 }
1476 key_prefix = tkey_extract_bits(cn->key, mp, cn->pos-mp);
1477
1478 if (key_prefix != 0)
1479 goto backtrace;
1480
1481 if (current_prefix_length >= cn->pos)
1482 current_prefix_length = mp;
1483 }
1484#endif
1485 pn = (struct tnode *)n; /* Descend */
1486 chopped_off = 0;
1487 continue;
1488
Robert Olsson19baf832005-06-21 12:43:18 -07001489backtrace:
1490 chopped_off++;
1491
1492 /* As zero don't change the child key (cindex) */
Olof Johansson91b9a272005-08-09 20:24:39 -07001493 while ((chopped_off <= pn->bits) && !(cindex & (1<<(chopped_off-1))))
Robert Olsson19baf832005-06-21 12:43:18 -07001494 chopped_off++;
Robert Olsson19baf832005-06-21 12:43:18 -07001495
1496 /* Decrease current_... with bits chopped off */
1497 if (current_prefix_length > pn->pos + pn->bits - chopped_off)
1498 current_prefix_length = pn->pos + pn->bits - chopped_off;
Olof Johansson91b9a272005-08-09 20:24:39 -07001499
Robert Olsson19baf832005-06-21 12:43:18 -07001500 /*
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001501 * Either we do the actual chop off according or if we have
Robert Olsson19baf832005-06-21 12:43:18 -07001502 * chopped off all bits in this tnode walk up to our parent.
1503 */
1504
Olof Johansson91b9a272005-08-09 20:24:39 -07001505 if (chopped_off <= pn->bits) {
Robert Olsson19baf832005-06-21 12:43:18 -07001506 cindex &= ~(1 << (chopped_off-1));
Olof Johansson91b9a272005-08-09 20:24:39 -07001507 } else {
Stephen Hemminger06801912007-08-10 15:22:13 -07001508 struct tnode *parent = node_parent((struct node *) pn);
1509 if (!parent)
Robert Olsson19baf832005-06-21 12:43:18 -07001510 goto failed;
Olof Johansson91b9a272005-08-09 20:24:39 -07001511
Robert Olsson19baf832005-06-21 12:43:18 -07001512 /* Get Child's index */
Stephen Hemminger06801912007-08-10 15:22:13 -07001513 cindex = tkey_extract_bits(pn->key, parent->pos, parent->bits);
1514 pn = parent;
Robert Olsson19baf832005-06-21 12:43:18 -07001515 chopped_off = 0;
1516
1517#ifdef CONFIG_IP_FIB_TRIE_STATS
1518 t->stats.backtrack++;
1519#endif
1520 goto backtrace;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001521 }
Robert Olsson19baf832005-06-21 12:43:18 -07001522 }
1523failed:
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001524 ret = 1;
Robert Olsson19baf832005-06-21 12:43:18 -07001525found:
Robert Olsson2373ce12005-08-25 13:01:29 -07001526 rcu_read_unlock();
Robert Olsson19baf832005-06-21 12:43:18 -07001527 return ret;
1528}
1529
Robert Olsson2373ce12005-08-25 13:01:29 -07001530/* only called from updater side */
Robert Olsson19baf832005-06-21 12:43:18 -07001531static int trie_leaf_remove(struct trie *t, t_key key)
1532{
1533 t_key cindex;
1534 struct tnode *tp = NULL;
1535 struct node *n = t->trie;
1536 struct leaf *l;
1537
Stephen Hemminger0c7770c2005-08-23 21:59:41 -07001538 pr_debug("entering trie_leaf_remove(%p)\n", n);
Robert Olsson19baf832005-06-21 12:43:18 -07001539
1540 /* Note that in the case skipped bits, those bits are *not* checked!
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001541 * When we finish this, we will have NULL or a T_LEAF, and the
Robert Olsson19baf832005-06-21 12:43:18 -07001542 * T_LEAF may or may not match our key.
1543 */
1544
Olof Johansson91b9a272005-08-09 20:24:39 -07001545 while (n != NULL && IS_TNODE(n)) {
Robert Olsson19baf832005-06-21 12:43:18 -07001546 struct tnode *tn = (struct tnode *) n;
1547 check_tnode(tn);
1548 n = tnode_get_child(tn ,tkey_extract_bits(key, tn->pos, tn->bits));
1549
Stephen Hemminger06801912007-08-10 15:22:13 -07001550 BUG_ON(n && node_parent(n) != tn);
Olof Johansson91b9a272005-08-09 20:24:39 -07001551 }
Robert Olsson19baf832005-06-21 12:43:18 -07001552 l = (struct leaf *) n;
1553
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001554 if (!n || !tkey_equals(l->key, key))
Robert Olsson19baf832005-06-21 12:43:18 -07001555 return 0;
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001556
1557 /*
1558 * Key found.
1559 * Remove the leaf and rebalance the tree
Robert Olsson19baf832005-06-21 12:43:18 -07001560 */
1561
1562 t->revision++;
1563 t->size--;
1564
Stephen Hemminger06801912007-08-10 15:22:13 -07001565 tp = node_parent(n);
Robert Olsson19baf832005-06-21 12:43:18 -07001566 tnode_free((struct tnode *) n);
1567
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001568 if (tp) {
Robert Olsson19baf832005-06-21 12:43:18 -07001569 cindex = tkey_extract_bits(key, tp->pos, tp->bits);
1570 put_child(t, (struct tnode *)tp, cindex, NULL);
Robert Olsson2373ce12005-08-25 13:01:29 -07001571 rcu_assign_pointer(t->trie, trie_rebalance(t, tp));
Olof Johansson91b9a272005-08-09 20:24:39 -07001572 } else
Robert Olsson2373ce12005-08-25 13:01:29 -07001573 rcu_assign_pointer(t->trie, NULL);
Robert Olsson19baf832005-06-21 12:43:18 -07001574
1575 return 1;
1576}
1577
Robert Olssond562f1f2007-03-26 14:22:22 -07001578/*
1579 * Caller must hold RTNL.
1580 */
Thomas Graf4e902c52006-08-17 18:14:52 -07001581static int fn_trie_delete(struct fib_table *tb, struct fib_config *cfg)
Robert Olsson19baf832005-06-21 12:43:18 -07001582{
1583 struct trie *t = (struct trie *) tb->tb_data;
1584 u32 key, mask;
Thomas Graf4e902c52006-08-17 18:14:52 -07001585 int plen = cfg->fc_dst_len;
1586 u8 tos = cfg->fc_tos;
Robert Olsson19baf832005-06-21 12:43:18 -07001587 struct fib_alias *fa, *fa_to_delete;
1588 struct list_head *fa_head;
1589 struct leaf *l;
Olof Johansson91b9a272005-08-09 20:24:39 -07001590 struct leaf_info *li;
1591
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001592 if (plen > 32)
Robert Olsson19baf832005-06-21 12:43:18 -07001593 return -EINVAL;
1594
Thomas Graf4e902c52006-08-17 18:14:52 -07001595 key = ntohl(cfg->fc_dst);
Olof Johansson91b9a272005-08-09 20:24:39 -07001596 mask = ntohl(inet_make_mask(plen));
Robert Olsson19baf832005-06-21 12:43:18 -07001597
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001598 if (key & ~mask)
Robert Olsson19baf832005-06-21 12:43:18 -07001599 return -EINVAL;
1600
1601 key = key & mask;
1602 l = fib_find_node(t, key);
1603
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001604 if (!l)
Robert Olsson19baf832005-06-21 12:43:18 -07001605 return -ESRCH;
1606
1607 fa_head = get_fa_head(l, plen);
1608 fa = fib_find_alias(fa_head, tos, 0);
1609
1610 if (!fa)
1611 return -ESRCH;
1612
Stephen Hemminger0c7770c2005-08-23 21:59:41 -07001613 pr_debug("Deleting %08x/%d tos=%d t=%p\n", key, plen, tos, t);
Robert Olsson19baf832005-06-21 12:43:18 -07001614
1615 fa_to_delete = NULL;
1616 fa_head = fa->fa_list.prev;
Robert Olsson2373ce12005-08-25 13:01:29 -07001617
Robert Olsson19baf832005-06-21 12:43:18 -07001618 list_for_each_entry(fa, fa_head, fa_list) {
1619 struct fib_info *fi = fa->fa_info;
1620
1621 if (fa->fa_tos != tos)
1622 break;
1623
Thomas Graf4e902c52006-08-17 18:14:52 -07001624 if ((!cfg->fc_type || fa->fa_type == cfg->fc_type) &&
1625 (cfg->fc_scope == RT_SCOPE_NOWHERE ||
1626 fa->fa_scope == cfg->fc_scope) &&
1627 (!cfg->fc_protocol ||
1628 fi->fib_protocol == cfg->fc_protocol) &&
1629 fib_nh_match(cfg, fi) == 0) {
Robert Olsson19baf832005-06-21 12:43:18 -07001630 fa_to_delete = fa;
1631 break;
1632 }
1633 }
1634
Olof Johansson91b9a272005-08-09 20:24:39 -07001635 if (!fa_to_delete)
1636 return -ESRCH;
Robert Olsson19baf832005-06-21 12:43:18 -07001637
Olof Johansson91b9a272005-08-09 20:24:39 -07001638 fa = fa_to_delete;
Thomas Graf4e902c52006-08-17 18:14:52 -07001639 rtmsg_fib(RTM_DELROUTE, htonl(key), fa, plen, tb->tb_id,
Milan Kocianb8f55832007-05-23 14:55:06 -07001640 &cfg->fc_nlinfo, 0);
Robert Olsson19baf832005-06-21 12:43:18 -07001641
Olof Johansson91b9a272005-08-09 20:24:39 -07001642 l = fib_find_node(t, key);
Robert Olsson772cb712005-09-19 15:31:18 -07001643 li = find_leaf_info(l, plen);
Robert Olsson19baf832005-06-21 12:43:18 -07001644
Robert Olsson2373ce12005-08-25 13:01:29 -07001645 list_del_rcu(&fa->fa_list);
Robert Olsson19baf832005-06-21 12:43:18 -07001646
Olof Johansson91b9a272005-08-09 20:24:39 -07001647 if (list_empty(fa_head)) {
Robert Olsson2373ce12005-08-25 13:01:29 -07001648 hlist_del_rcu(&li->hlist);
Olof Johansson91b9a272005-08-09 20:24:39 -07001649 free_leaf_info(li);
Robert Olsson2373ce12005-08-25 13:01:29 -07001650 }
Olof Johansson91b9a272005-08-09 20:24:39 -07001651
1652 if (hlist_empty(&l->list))
1653 trie_leaf_remove(t, key);
1654
1655 if (fa->fa_state & FA_S_ACCESSED)
1656 rt_cache_flush(-1);
1657
Robert Olsson2373ce12005-08-25 13:01:29 -07001658 fib_release_info(fa->fa_info);
1659 alias_free_mem_rcu(fa);
Olof Johansson91b9a272005-08-09 20:24:39 -07001660 return 0;
Robert Olsson19baf832005-06-21 12:43:18 -07001661}
1662
1663static int trie_flush_list(struct trie *t, struct list_head *head)
1664{
1665 struct fib_alias *fa, *fa_node;
1666 int found = 0;
1667
1668 list_for_each_entry_safe(fa, fa_node, head, fa_list) {
1669 struct fib_info *fi = fa->fa_info;
Robert Olsson19baf832005-06-21 12:43:18 -07001670
Robert Olsson2373ce12005-08-25 13:01:29 -07001671 if (fi && (fi->fib_flags & RTNH_F_DEAD)) {
1672 list_del_rcu(&fa->fa_list);
1673 fib_release_info(fa->fa_info);
1674 alias_free_mem_rcu(fa);
Robert Olsson19baf832005-06-21 12:43:18 -07001675 found++;
1676 }
1677 }
1678 return found;
1679}
1680
1681static int trie_flush_leaf(struct trie *t, struct leaf *l)
1682{
1683 int found = 0;
1684 struct hlist_head *lih = &l->list;
1685 struct hlist_node *node, *tmp;
1686 struct leaf_info *li = NULL;
1687
1688 hlist_for_each_entry_safe(li, node, tmp, lih, hlist) {
Robert Olsson19baf832005-06-21 12:43:18 -07001689 found += trie_flush_list(t, &li->falh);
1690
1691 if (list_empty(&li->falh)) {
Robert Olsson2373ce12005-08-25 13:01:29 -07001692 hlist_del_rcu(&li->hlist);
Robert Olsson19baf832005-06-21 12:43:18 -07001693 free_leaf_info(li);
1694 }
1695 }
1696 return found;
1697}
1698
Robert Olsson2373ce12005-08-25 13:01:29 -07001699/* rcu_read_lock needs to be hold by caller from readside */
1700
Robert Olsson19baf832005-06-21 12:43:18 -07001701static struct leaf *nextleaf(struct trie *t, struct leaf *thisleaf)
1702{
1703 struct node *c = (struct node *) thisleaf;
1704 struct tnode *p;
1705 int idx;
Robert Olsson2373ce12005-08-25 13:01:29 -07001706 struct node *trie = rcu_dereference(t->trie);
Robert Olsson19baf832005-06-21 12:43:18 -07001707
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001708 if (c == NULL) {
Robert Olsson2373ce12005-08-25 13:01:29 -07001709 if (trie == NULL)
Robert Olsson19baf832005-06-21 12:43:18 -07001710 return NULL;
1711
Robert Olsson2373ce12005-08-25 13:01:29 -07001712 if (IS_LEAF(trie)) /* trie w. just a leaf */
1713 return (struct leaf *) trie;
Robert Olsson19baf832005-06-21 12:43:18 -07001714
Robert Olsson2373ce12005-08-25 13:01:29 -07001715 p = (struct tnode*) trie; /* Start */
Olof Johansson91b9a272005-08-09 20:24:39 -07001716 } else
Stephen Hemminger06801912007-08-10 15:22:13 -07001717 p = node_parent(c);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001718
Robert Olsson19baf832005-06-21 12:43:18 -07001719 while (p) {
1720 int pos, last;
1721
1722 /* Find the next child of the parent */
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001723 if (c)
1724 pos = 1 + tkey_extract_bits(c->key, p->pos, p->bits);
1725 else
Robert Olsson19baf832005-06-21 12:43:18 -07001726 pos = 0;
1727
1728 last = 1 << p->bits;
Olof Johansson91b9a272005-08-09 20:24:39 -07001729 for (idx = pos; idx < last ; idx++) {
Robert Olsson2373ce12005-08-25 13:01:29 -07001730 c = rcu_dereference(p->child[idx]);
1731
1732 if (!c)
Olof Johansson91b9a272005-08-09 20:24:39 -07001733 continue;
Robert Olsson19baf832005-06-21 12:43:18 -07001734
Olof Johansson91b9a272005-08-09 20:24:39 -07001735 /* Decend if tnode */
Robert Olsson2373ce12005-08-25 13:01:29 -07001736 while (IS_TNODE(c)) {
1737 p = (struct tnode *) c;
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +09001738 idx = 0;
Robert Olsson19baf832005-06-21 12:43:18 -07001739
Olof Johansson91b9a272005-08-09 20:24:39 -07001740 /* Rightmost non-NULL branch */
1741 if (p && IS_TNODE(p))
Robert Olsson2373ce12005-08-25 13:01:29 -07001742 while (!(c = rcu_dereference(p->child[idx]))
1743 && idx < (1<<p->bits)) idx++;
Robert Olsson19baf832005-06-21 12:43:18 -07001744
Olof Johansson91b9a272005-08-09 20:24:39 -07001745 /* Done with this tnode? */
Robert Olsson2373ce12005-08-25 13:01:29 -07001746 if (idx >= (1 << p->bits) || !c)
Olof Johansson91b9a272005-08-09 20:24:39 -07001747 goto up;
Robert Olsson19baf832005-06-21 12:43:18 -07001748 }
Robert Olsson2373ce12005-08-25 13:01:29 -07001749 return (struct leaf *) c;
Robert Olsson19baf832005-06-21 12:43:18 -07001750 }
1751up:
1752 /* No more children go up one step */
Olof Johansson91b9a272005-08-09 20:24:39 -07001753 c = (struct node *) p;
Stephen Hemminger06801912007-08-10 15:22:13 -07001754 p = node_parent(c);
Robert Olsson19baf832005-06-21 12:43:18 -07001755 }
1756 return NULL; /* Ready. Root of trie */
1757}
1758
Robert Olssond562f1f2007-03-26 14:22:22 -07001759/*
1760 * Caller must hold RTNL.
1761 */
Robert Olsson19baf832005-06-21 12:43:18 -07001762static int fn_trie_flush(struct fib_table *tb)
1763{
1764 struct trie *t = (struct trie *) tb->tb_data;
1765 struct leaf *ll = NULL, *l = NULL;
1766 int found = 0, h;
1767
1768 t->revision++;
1769
Olof Johansson91b9a272005-08-09 20:24:39 -07001770 for (h = 0; (l = nextleaf(t, l)) != NULL; h++) {
Robert Olsson19baf832005-06-21 12:43:18 -07001771 found += trie_flush_leaf(t, l);
1772
1773 if (ll && hlist_empty(&ll->list))
1774 trie_leaf_remove(t, ll->key);
1775 ll = l;
1776 }
1777
1778 if (ll && hlist_empty(&ll->list))
1779 trie_leaf_remove(t, ll->key);
1780
Stephen Hemminger0c7770c2005-08-23 21:59:41 -07001781 pr_debug("trie_flush found=%d\n", found);
Robert Olsson19baf832005-06-21 12:43:18 -07001782 return found;
1783}
1784
Robert Olsson19baf832005-06-21 12:43:18 -07001785static void
1786fn_trie_select_default(struct fib_table *tb, const struct flowi *flp, struct fib_result *res)
1787{
1788 struct trie *t = (struct trie *) tb->tb_data;
1789 int order, last_idx;
1790 struct fib_info *fi = NULL;
1791 struct fib_info *last_resort;
1792 struct fib_alias *fa = NULL;
1793 struct list_head *fa_head;
1794 struct leaf *l;
1795
1796 last_idx = -1;
1797 last_resort = NULL;
1798 order = -1;
1799
Robert Olsson2373ce12005-08-25 13:01:29 -07001800 rcu_read_lock();
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001801
Robert Olsson19baf832005-06-21 12:43:18 -07001802 l = fib_find_node(t, 0);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001803 if (!l)
Robert Olsson19baf832005-06-21 12:43:18 -07001804 goto out;
1805
1806 fa_head = get_fa_head(l, 0);
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001807 if (!fa_head)
Robert Olsson19baf832005-06-21 12:43:18 -07001808 goto out;
1809
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001810 if (list_empty(fa_head))
Robert Olsson19baf832005-06-21 12:43:18 -07001811 goto out;
1812
Robert Olsson2373ce12005-08-25 13:01:29 -07001813 list_for_each_entry_rcu(fa, fa_head, fa_list) {
Robert Olsson19baf832005-06-21 12:43:18 -07001814 struct fib_info *next_fi = fa->fa_info;
Olof Johansson91b9a272005-08-09 20:24:39 -07001815
Robert Olsson19baf832005-06-21 12:43:18 -07001816 if (fa->fa_scope != res->scope ||
1817 fa->fa_type != RTN_UNICAST)
1818 continue;
Olof Johansson91b9a272005-08-09 20:24:39 -07001819
Robert Olsson19baf832005-06-21 12:43:18 -07001820 if (next_fi->fib_priority > res->fi->fib_priority)
1821 break;
1822 if (!next_fi->fib_nh[0].nh_gw ||
1823 next_fi->fib_nh[0].nh_scope != RT_SCOPE_LINK)
1824 continue;
1825 fa->fa_state |= FA_S_ACCESSED;
Olof Johansson91b9a272005-08-09 20:24:39 -07001826
Robert Olsson19baf832005-06-21 12:43:18 -07001827 if (fi == NULL) {
1828 if (next_fi != res->fi)
1829 break;
1830 } else if (!fib_detect_death(fi, order, &last_resort,
Denis V. Lunev971b8932007-12-08 00:32:23 -08001831 &last_idx, tb->tb_default)) {
Denis V. Luneva2bbe682007-12-08 00:31:44 -08001832 fib_result_assign(res, fi);
Denis V. Lunev971b8932007-12-08 00:32:23 -08001833 tb->tb_default = order;
Robert Olsson19baf832005-06-21 12:43:18 -07001834 goto out;
1835 }
1836 fi = next_fi;
1837 order++;
1838 }
1839 if (order <= 0 || fi == NULL) {
Denis V. Lunev971b8932007-12-08 00:32:23 -08001840 tb->tb_default = -1;
Robert Olsson19baf832005-06-21 12:43:18 -07001841 goto out;
1842 }
1843
Denis V. Lunev971b8932007-12-08 00:32:23 -08001844 if (!fib_detect_death(fi, order, &last_resort, &last_idx,
1845 tb->tb_default)) {
Denis V. Luneva2bbe682007-12-08 00:31:44 -08001846 fib_result_assign(res, fi);
Denis V. Lunev971b8932007-12-08 00:32:23 -08001847 tb->tb_default = order;
Robert Olsson19baf832005-06-21 12:43:18 -07001848 goto out;
1849 }
Denis V. Luneva2bbe682007-12-08 00:31:44 -08001850 if (last_idx >= 0)
1851 fib_result_assign(res, last_resort);
Denis V. Lunev971b8932007-12-08 00:32:23 -08001852 tb->tb_default = last_idx;
1853out:
Robert Olsson2373ce12005-08-25 13:01:29 -07001854 rcu_read_unlock();
Robert Olsson19baf832005-06-21 12:43:18 -07001855}
1856
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001857static int fn_trie_dump_fa(t_key key, int plen, struct list_head *fah, struct fib_table *tb,
Robert Olsson19baf832005-06-21 12:43:18 -07001858 struct sk_buff *skb, struct netlink_callback *cb)
1859{
1860 int i, s_i;
1861 struct fib_alias *fa;
1862
Al Viro32ab5f82006-09-26 22:21:45 -07001863 __be32 xkey = htonl(key);
Robert Olsson19baf832005-06-21 12:43:18 -07001864
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001865 s_i = cb->args[4];
Robert Olsson19baf832005-06-21 12:43:18 -07001866 i = 0;
1867
Robert Olsson2373ce12005-08-25 13:01:29 -07001868 /* rcu_read_lock is hold by caller */
1869
1870 list_for_each_entry_rcu(fa, fah, fa_list) {
Robert Olsson19baf832005-06-21 12:43:18 -07001871 if (i < s_i) {
1872 i++;
1873 continue;
1874 }
Stephen Hemminger78c66712005-09-21 00:15:39 -07001875 BUG_ON(!fa->fa_info);
Robert Olsson19baf832005-06-21 12:43:18 -07001876
1877 if (fib_dump_info(skb, NETLINK_CB(cb->skb).pid,
1878 cb->nlh->nlmsg_seq,
1879 RTM_NEWROUTE,
1880 tb->tb_id,
1881 fa->fa_type,
1882 fa->fa_scope,
Thomas Grafbe403ea2006-08-17 18:15:17 -07001883 xkey,
Robert Olsson19baf832005-06-21 12:43:18 -07001884 plen,
1885 fa->fa_tos,
David S. Miller90f66912005-06-21 14:43:28 -07001886 fa->fa_info, 0) < 0) {
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001887 cb->args[4] = i;
Robert Olsson19baf832005-06-21 12:43:18 -07001888 return -1;
Olof Johansson91b9a272005-08-09 20:24:39 -07001889 }
Robert Olsson19baf832005-06-21 12:43:18 -07001890 i++;
1891 }
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001892 cb->args[4] = i;
Robert Olsson19baf832005-06-21 12:43:18 -07001893 return skb->len;
1894}
1895
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001896static int fn_trie_dump_plen(struct trie *t, int plen, struct fib_table *tb, struct sk_buff *skb,
Robert Olsson19baf832005-06-21 12:43:18 -07001897 struct netlink_callback *cb)
1898{
1899 int h, s_h;
1900 struct list_head *fa_head;
1901 struct leaf *l = NULL;
Robert Olsson19baf832005-06-21 12:43:18 -07001902
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001903 s_h = cb->args[3];
Robert Olsson19baf832005-06-21 12:43:18 -07001904
Olof Johansson91b9a272005-08-09 20:24:39 -07001905 for (h = 0; (l = nextleaf(t, l)) != NULL; h++) {
Robert Olsson19baf832005-06-21 12:43:18 -07001906 if (h < s_h)
1907 continue;
1908 if (h > s_h)
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001909 memset(&cb->args[4], 0,
1910 sizeof(cb->args) - 4*sizeof(cb->args[0]));
Robert Olsson19baf832005-06-21 12:43:18 -07001911
1912 fa_head = get_fa_head(l, plen);
Olof Johansson91b9a272005-08-09 20:24:39 -07001913
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001914 if (!fa_head)
Robert Olsson19baf832005-06-21 12:43:18 -07001915 continue;
1916
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001917 if (list_empty(fa_head))
Robert Olsson19baf832005-06-21 12:43:18 -07001918 continue;
1919
1920 if (fn_trie_dump_fa(l->key, plen, fa_head, tb, skb, cb)<0) {
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001921 cb->args[3] = h;
Robert Olsson19baf832005-06-21 12:43:18 -07001922 return -1;
1923 }
1924 }
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001925 cb->args[3] = h;
Robert Olsson19baf832005-06-21 12:43:18 -07001926 return skb->len;
1927}
1928
1929static int fn_trie_dump(struct fib_table *tb, struct sk_buff *skb, struct netlink_callback *cb)
1930{
1931 int m, s_m;
1932 struct trie *t = (struct trie *) tb->tb_data;
1933
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001934 s_m = cb->args[2];
Robert Olsson19baf832005-06-21 12:43:18 -07001935
Robert Olsson2373ce12005-08-25 13:01:29 -07001936 rcu_read_lock();
Olof Johansson91b9a272005-08-09 20:24:39 -07001937 for (m = 0; m <= 32; m++) {
Robert Olsson19baf832005-06-21 12:43:18 -07001938 if (m < s_m)
1939 continue;
1940 if (m > s_m)
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001941 memset(&cb->args[3], 0,
1942 sizeof(cb->args) - 3*sizeof(cb->args[0]));
Robert Olsson19baf832005-06-21 12:43:18 -07001943
1944 if (fn_trie_dump_plen(t, 32-m, tb, skb, cb)<0) {
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001945 cb->args[2] = m;
Robert Olsson19baf832005-06-21 12:43:18 -07001946 goto out;
1947 }
1948 }
Robert Olsson2373ce12005-08-25 13:01:29 -07001949 rcu_read_unlock();
Patrick McHardy1af5a8c2006-08-10 23:10:46 -07001950 cb->args[2] = m;
Robert Olsson19baf832005-06-21 12:43:18 -07001951 return skb->len;
Olof Johansson91b9a272005-08-09 20:24:39 -07001952out:
Robert Olsson2373ce12005-08-25 13:01:29 -07001953 rcu_read_unlock();
Robert Olsson19baf832005-06-21 12:43:18 -07001954 return -1;
1955}
1956
1957/* Fix more generic FIB names for init later */
1958
1959#ifdef CONFIG_IP_MULTIPLE_TABLES
Patrick McHardy2dfe55b2006-08-10 23:08:33 -07001960struct fib_table * fib_hash_init(u32 id)
Robert Olsson19baf832005-06-21 12:43:18 -07001961#else
Patrick McHardy2dfe55b2006-08-10 23:08:33 -07001962struct fib_table * __init fib_hash_init(u32 id)
Robert Olsson19baf832005-06-21 12:43:18 -07001963#endif
1964{
1965 struct fib_table *tb;
1966 struct trie *t;
1967
1968 if (fn_alias_kmem == NULL)
1969 fn_alias_kmem = kmem_cache_create("ip_fib_alias",
1970 sizeof(struct fib_alias),
1971 0, SLAB_HWCACHE_ALIGN,
Paul Mundt20c2df82007-07-20 10:11:58 +09001972 NULL);
Robert Olsson19baf832005-06-21 12:43:18 -07001973
1974 tb = kmalloc(sizeof(struct fib_table) + sizeof(struct trie),
1975 GFP_KERNEL);
1976 if (tb == NULL)
1977 return NULL;
1978
1979 tb->tb_id = id;
Denis V. Lunev971b8932007-12-08 00:32:23 -08001980 tb->tb_default = -1;
Robert Olsson19baf832005-06-21 12:43:18 -07001981 tb->tb_lookup = fn_trie_lookup;
1982 tb->tb_insert = fn_trie_insert;
1983 tb->tb_delete = fn_trie_delete;
1984 tb->tb_flush = fn_trie_flush;
1985 tb->tb_select_default = fn_trie_select_default;
1986 tb->tb_dump = fn_trie_dump;
1987 memset(tb->tb_data, 0, sizeof(struct trie));
1988
1989 t = (struct trie *) tb->tb_data;
1990
1991 trie_init(t);
1992
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07001993 if (id == RT_TABLE_LOCAL)
Stephen Hemminger78c66712005-09-21 00:15:39 -07001994 printk(KERN_INFO "IPv4 FIB: Using LC-trie version %s\n", VERSION);
Robert Olsson19baf832005-06-21 12:43:18 -07001995
1996 return tb;
1997}
1998
Robert Olsson19baf832005-06-21 12:43:18 -07001999#ifdef CONFIG_PROC_FS
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002000/* Depth first Trie walk iterator */
2001struct fib_trie_iter {
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002002 struct trie *trie_local, *trie_main;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002003 struct tnode *tnode;
2004 struct trie *trie;
2005 unsigned index;
2006 unsigned depth;
2007};
Robert Olsson19baf832005-06-21 12:43:18 -07002008
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002009static struct node *fib_trie_get_next(struct fib_trie_iter *iter)
Robert Olsson19baf832005-06-21 12:43:18 -07002010{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002011 struct tnode *tn = iter->tnode;
2012 unsigned cindex = iter->index;
2013 struct tnode *p;
2014
Eric W. Biederman6640e692007-01-24 14:42:04 -08002015 /* A single entry routing table */
2016 if (!tn)
2017 return NULL;
2018
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002019 pr_debug("get_next iter={node=%p index=%d depth=%d}\n",
2020 iter->tnode, iter->index, iter->depth);
2021rescan:
2022 while (cindex < (1<<tn->bits)) {
2023 struct node *n = tnode_get_child(tn, cindex);
2024
2025 if (n) {
2026 if (IS_LEAF(n)) {
2027 iter->tnode = tn;
2028 iter->index = cindex + 1;
2029 } else {
2030 /* push down one level */
2031 iter->tnode = (struct tnode *) n;
2032 iter->index = 0;
2033 ++iter->depth;
2034 }
2035 return n;
2036 }
2037
2038 ++cindex;
2039 }
2040
2041 /* Current node exhausted, pop back up */
Stephen Hemminger06801912007-08-10 15:22:13 -07002042 p = node_parent((struct node *)tn);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002043 if (p) {
2044 cindex = tkey_extract_bits(tn->key, p->pos, p->bits)+1;
2045 tn = p;
2046 --iter->depth;
2047 goto rescan;
2048 }
2049
2050 /* got root? */
Robert Olsson19baf832005-06-21 12:43:18 -07002051 return NULL;
2052}
2053
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002054static struct node *fib_trie_get_first(struct fib_trie_iter *iter,
2055 struct trie *t)
Robert Olsson19baf832005-06-21 12:43:18 -07002056{
Robert Olsson5ddf0eb2006-03-20 21:34:12 -08002057 struct node *n ;
2058
Stephen Hemminger132adf52007-03-08 20:44:43 -08002059 if (!t)
Robert Olsson5ddf0eb2006-03-20 21:34:12 -08002060 return NULL;
2061
2062 n = rcu_dereference(t->trie);
2063
Stephen Hemminger132adf52007-03-08 20:44:43 -08002064 if (!iter)
Robert Olsson5ddf0eb2006-03-20 21:34:12 -08002065 return NULL;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002066
Eric W. Biederman6640e692007-01-24 14:42:04 -08002067 if (n) {
2068 if (IS_TNODE(n)) {
2069 iter->tnode = (struct tnode *) n;
2070 iter->trie = t;
2071 iter->index = 0;
2072 iter->depth = 1;
2073 } else {
2074 iter->tnode = NULL;
2075 iter->trie = t;
2076 iter->index = 0;
2077 iter->depth = 0;
2078 }
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002079 return n;
2080 }
Robert Olsson19baf832005-06-21 12:43:18 -07002081 return NULL;
2082}
2083
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002084static void trie_collect_stats(struct trie *t, struct trie_stat *s)
Robert Olsson19baf832005-06-21 12:43:18 -07002085{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002086 struct node *n;
2087 struct fib_trie_iter iter;
Robert Olsson19baf832005-06-21 12:43:18 -07002088
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002089 memset(s, 0, sizeof(*s));
Robert Olsson19baf832005-06-21 12:43:18 -07002090
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002091 rcu_read_lock();
2092 for (n = fib_trie_get_first(&iter, t); n;
2093 n = fib_trie_get_next(&iter)) {
2094 if (IS_LEAF(n)) {
2095 s->leaves++;
2096 s->totdepth += iter.depth;
2097 if (iter.depth > s->maxdepth)
2098 s->maxdepth = iter.depth;
2099 } else {
2100 const struct tnode *tn = (const struct tnode *) n;
2101 int i;
Robert Olsson19baf832005-06-21 12:43:18 -07002102
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002103 s->tnodes++;
Stephen Hemminger132adf52007-03-08 20:44:43 -08002104 if (tn->bits < MAX_STAT_DEPTH)
Robert Olsson06ef9212006-03-20 21:35:01 -08002105 s->nodesizes[tn->bits]++;
2106
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002107 for (i = 0; i < (1<<tn->bits); i++)
2108 if (!tn->child[i])
2109 s->nullpointers++;
2110 }
2111 }
2112 rcu_read_unlock();
Robert Olsson19baf832005-06-21 12:43:18 -07002113}
2114
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07002115/*
Robert Olsson19baf832005-06-21 12:43:18 -07002116 * This outputs /proc/net/fib_triestats
Robert Olsson19baf832005-06-21 12:43:18 -07002117 */
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002118static void trie_show_stats(struct seq_file *seq, struct trie_stat *stat)
Robert Olsson19baf832005-06-21 12:43:18 -07002119{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002120 unsigned i, max, pointers, bytes, avdepth;
Robert Olsson19baf832005-06-21 12:43:18 -07002121
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002122 if (stat->leaves)
2123 avdepth = stat->totdepth*100 / stat->leaves;
2124 else
2125 avdepth = 0;
Robert Olsson19baf832005-06-21 12:43:18 -07002126
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002127 seq_printf(seq, "\tAver depth: %d.%02d\n", avdepth / 100, avdepth % 100 );
2128 seq_printf(seq, "\tMax depth: %u\n", stat->maxdepth);
Robert Olsson19baf832005-06-21 12:43:18 -07002129
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002130 seq_printf(seq, "\tLeaves: %u\n", stat->leaves);
Olof Johansson91b9a272005-08-09 20:24:39 -07002131
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002132 bytes = sizeof(struct leaf) * stat->leaves;
2133 seq_printf(seq, "\tInternal nodes: %d\n\t", stat->tnodes);
2134 bytes += sizeof(struct tnode) * stat->tnodes;
Robert Olsson19baf832005-06-21 12:43:18 -07002135
Robert Olsson06ef9212006-03-20 21:35:01 -08002136 max = MAX_STAT_DEPTH;
2137 while (max > 0 && stat->nodesizes[max-1] == 0)
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002138 max--;
Robert Olsson19baf832005-06-21 12:43:18 -07002139
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002140 pointers = 0;
2141 for (i = 1; i <= max; i++)
2142 if (stat->nodesizes[i] != 0) {
2143 seq_printf(seq, " %d: %d", i, stat->nodesizes[i]);
2144 pointers += (1<<i) * stat->nodesizes[i];
2145 }
2146 seq_putc(seq, '\n');
2147 seq_printf(seq, "\tPointers: %d\n", pointers);
Robert Olsson19baf832005-06-21 12:43:18 -07002148
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002149 bytes += sizeof(struct node *) * pointers;
2150 seq_printf(seq, "Null ptrs: %d\n", stat->nullpointers);
2151 seq_printf(seq, "Total size: %d kB\n", (bytes + 1023) / 1024);
Robert Olsson19baf832005-06-21 12:43:18 -07002152
2153#ifdef CONFIG_IP_FIB_TRIE_STATS
2154 seq_printf(seq, "Counters:\n---------\n");
2155 seq_printf(seq,"gets = %d\n", t->stats.gets);
2156 seq_printf(seq,"backtracks = %d\n", t->stats.backtrack);
2157 seq_printf(seq,"semantic match passed = %d\n", t->stats.semantic_match_passed);
2158 seq_printf(seq,"semantic match miss = %d\n", t->stats.semantic_match_miss);
2159 seq_printf(seq,"null node hit= %d\n", t->stats.null_node_hit);
Robert Olsson2f368952005-07-05 15:02:40 -07002160 seq_printf(seq,"skipped node resize = %d\n", t->stats.resize_node_skipped);
Robert Olsson19baf832005-06-21 12:43:18 -07002161#ifdef CLEAR_STATS
2162 memset(&(t->stats), 0, sizeof(t->stats));
2163#endif
2164#endif /* CONFIG_IP_FIB_TRIE_STATS */
2165}
2166
2167static int fib_triestat_seq_show(struct seq_file *seq, void *v)
2168{
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002169 struct trie *trie_local, *trie_main;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002170 struct trie_stat *stat;
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002171 struct fib_table *tb;
2172
2173 trie_local = NULL;
2174 tb = fib_get_table(RT_TABLE_LOCAL);
2175 if (tb)
2176 trie_local = (struct trie *) tb->tb_data;
2177
2178 trie_main = NULL;
2179 tb = fib_get_table(RT_TABLE_MAIN);
2180 if (tb)
2181 trie_main = (struct trie *) tb->tb_data;
2182
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07002183
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002184 stat = kmalloc(sizeof(*stat), GFP_KERNEL);
2185 if (!stat)
2186 return -ENOMEM;
Robert Olsson19baf832005-06-21 12:43:18 -07002187
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002188 seq_printf(seq, "Basic info: size of leaf: %Zd bytes, size of tnode: %Zd bytes.\n",
2189 sizeof(struct leaf), sizeof(struct tnode));
Olof Johansson91b9a272005-08-09 20:24:39 -07002190
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002191 if (trie_local) {
2192 seq_printf(seq, "Local:\n");
2193 trie_collect_stats(trie_local, stat);
2194 trie_show_stats(seq, stat);
Robert Olsson19baf832005-06-21 12:43:18 -07002195 }
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002196
2197 if (trie_main) {
2198 seq_printf(seq, "Main:\n");
2199 trie_collect_stats(trie_main, stat);
2200 trie_show_stats(seq, stat);
2201 }
2202 kfree(stat);
2203
Robert Olsson19baf832005-06-21 12:43:18 -07002204 return 0;
2205}
2206
Robert Olsson19baf832005-06-21 12:43:18 -07002207static int fib_triestat_seq_open(struct inode *inode, struct file *file)
2208{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002209 return single_open(file, fib_triestat_seq_show, NULL);
Robert Olsson19baf832005-06-21 12:43:18 -07002210}
2211
Arjan van de Ven9a321442007-02-12 00:55:35 -08002212static const struct file_operations fib_triestat_fops = {
Stephen Hemmingerc877efb2005-07-19 14:01:51 -07002213 .owner = THIS_MODULE,
2214 .open = fib_triestat_seq_open,
2215 .read = seq_read,
2216 .llseek = seq_lseek,
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002217 .release = single_release,
Robert Olsson19baf832005-06-21 12:43:18 -07002218};
2219
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002220static struct node *fib_trie_get_idx(struct fib_trie_iter *iter,
2221 loff_t pos)
Robert Olsson19baf832005-06-21 12:43:18 -07002222{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002223 loff_t idx = 0;
2224 struct node *n;
Robert Olsson19baf832005-06-21 12:43:18 -07002225
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002226 for (n = fib_trie_get_first(iter, iter->trie_local);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002227 n; ++idx, n = fib_trie_get_next(iter)) {
2228 if (pos == idx)
2229 return n;
2230 }
Robert Olsson19baf832005-06-21 12:43:18 -07002231
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002232 for (n = fib_trie_get_first(iter, iter->trie_main);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002233 n; ++idx, n = fib_trie_get_next(iter)) {
2234 if (pos == idx)
2235 return n;
2236 }
Robert Olsson19baf832005-06-21 12:43:18 -07002237 return NULL;
2238}
2239
2240static void *fib_trie_seq_start(struct seq_file *seq, loff_t *pos)
2241{
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002242 struct fib_trie_iter *iter = seq->private;
2243 struct fib_table *tb;
2244
2245 if (!iter->trie_local) {
2246 tb = fib_get_table(RT_TABLE_LOCAL);
2247 if (tb)
2248 iter->trie_local = (struct trie *) tb->tb_data;
2249 }
2250 if (!iter->trie_main) {
2251 tb = fib_get_table(RT_TABLE_MAIN);
2252 if (tb)
2253 iter->trie_main = (struct trie *) tb->tb_data;
2254 }
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002255 rcu_read_lock();
2256 if (*pos == 0)
Olof Johansson91b9a272005-08-09 20:24:39 -07002257 return SEQ_START_TOKEN;
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002258 return fib_trie_get_idx(iter, *pos - 1);
Robert Olsson19baf832005-06-21 12:43:18 -07002259}
2260
2261static void *fib_trie_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2262{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002263 struct fib_trie_iter *iter = seq->private;
2264 void *l = v;
2265
Robert Olsson19baf832005-06-21 12:43:18 -07002266 ++*pos;
Olof Johansson91b9a272005-08-09 20:24:39 -07002267 if (v == SEQ_START_TOKEN)
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002268 return fib_trie_get_idx(iter, 0);
Olof Johansson91b9a272005-08-09 20:24:39 -07002269
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002270 v = fib_trie_get_next(iter);
2271 BUG_ON(v == l);
2272 if (v)
2273 return v;
2274
2275 /* continue scan in next trie */
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002276 if (iter->trie == iter->trie_local)
2277 return fib_trie_get_first(iter, iter->trie_main);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002278
2279 return NULL;
Robert Olsson19baf832005-06-21 12:43:18 -07002280}
2281
2282static void fib_trie_seq_stop(struct seq_file *seq, void *v)
2283{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002284 rcu_read_unlock();
Robert Olsson19baf832005-06-21 12:43:18 -07002285}
2286
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002287static void seq_indent(struct seq_file *seq, int n)
2288{
2289 while (n-- > 0) seq_puts(seq, " ");
2290}
Robert Olsson19baf832005-06-21 12:43:18 -07002291
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002292static inline const char *rtn_scope(enum rt_scope_t s)
2293{
2294 static char buf[32];
2295
Stephen Hemminger132adf52007-03-08 20:44:43 -08002296 switch (s) {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002297 case RT_SCOPE_UNIVERSE: return "universe";
2298 case RT_SCOPE_SITE: return "site";
2299 case RT_SCOPE_LINK: return "link";
2300 case RT_SCOPE_HOST: return "host";
2301 case RT_SCOPE_NOWHERE: return "nowhere";
2302 default:
2303 snprintf(buf, sizeof(buf), "scope=%d", s);
2304 return buf;
2305 }
2306}
2307
2308static const char *rtn_type_names[__RTN_MAX] = {
2309 [RTN_UNSPEC] = "UNSPEC",
2310 [RTN_UNICAST] = "UNICAST",
2311 [RTN_LOCAL] = "LOCAL",
2312 [RTN_BROADCAST] = "BROADCAST",
2313 [RTN_ANYCAST] = "ANYCAST",
2314 [RTN_MULTICAST] = "MULTICAST",
2315 [RTN_BLACKHOLE] = "BLACKHOLE",
2316 [RTN_UNREACHABLE] = "UNREACHABLE",
2317 [RTN_PROHIBIT] = "PROHIBIT",
2318 [RTN_THROW] = "THROW",
2319 [RTN_NAT] = "NAT",
2320 [RTN_XRESOLVE] = "XRESOLVE",
2321};
2322
2323static inline const char *rtn_type(unsigned t)
2324{
2325 static char buf[32];
2326
2327 if (t < __RTN_MAX && rtn_type_names[t])
2328 return rtn_type_names[t];
2329 snprintf(buf, sizeof(buf), "type %d", t);
2330 return buf;
2331}
2332
2333/* Pretty print the trie */
Robert Olsson19baf832005-06-21 12:43:18 -07002334static int fib_trie_seq_show(struct seq_file *seq, void *v)
2335{
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002336 const struct fib_trie_iter *iter = seq->private;
2337 struct node *n = v;
Robert Olsson19baf832005-06-21 12:43:18 -07002338
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002339 if (v == SEQ_START_TOKEN)
2340 return 0;
Robert Olsson19baf832005-06-21 12:43:18 -07002341
Stephen Hemminger06801912007-08-10 15:22:13 -07002342 if (!node_parent(n)) {
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002343 if (iter->trie == iter->trie_local)
Robert Olsson095b8502007-01-26 19:06:01 -08002344 seq_puts(seq, "<local>:\n");
2345 else
2346 seq_puts(seq, "<main>:\n");
2347 }
2348
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002349 if (IS_TNODE(n)) {
2350 struct tnode *tn = (struct tnode *) n;
Stephen Hemmingerab66b4a2007-08-10 15:22:58 -07002351 __be32 prf = htonl(mask_pfx(tn->key, tn->pos));
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002352
Robert Olsson1d25cd62005-09-19 15:29:52 -07002353 seq_indent(seq, iter->depth-1);
2354 seq_printf(seq, " +-- %d.%d.%d.%d/%d %d %d %d\n",
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +09002355 NIPQUAD(prf), tn->pos, tn->bits, tn->full_children,
Robert Olsson1d25cd62005-09-19 15:29:52 -07002356 tn->empty_children);
YOSHIFUJI Hideakie905a9e2007-02-09 23:24:47 +09002357
Olof Johansson91b9a272005-08-09 20:24:39 -07002358 } else {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002359 struct leaf *l = (struct leaf *) n;
2360 int i;
Al Viro32ab5f82006-09-26 22:21:45 -07002361 __be32 val = htonl(l->key);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002362
2363 seq_indent(seq, iter->depth);
2364 seq_printf(seq, " |-- %d.%d.%d.%d\n", NIPQUAD(val));
2365 for (i = 32; i >= 0; i--) {
Robert Olsson772cb712005-09-19 15:31:18 -07002366 struct leaf_info *li = find_leaf_info(l, i);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002367 if (li) {
2368 struct fib_alias *fa;
2369 list_for_each_entry_rcu(fa, &li->falh, fa_list) {
2370 seq_indent(seq, iter->depth+1);
2371 seq_printf(seq, " /%d %s %s", i,
2372 rtn_scope(fa->fa_scope),
2373 rtn_type(fa->fa_type));
2374 if (fa->fa_tos)
2375 seq_printf(seq, "tos =%d\n",
2376 fa->fa_tos);
2377 seq_putc(seq, '\n');
2378 }
2379 }
2380 }
Robert Olsson19baf832005-06-21 12:43:18 -07002381 }
2382
2383 return 0;
2384}
2385
Stephen Hemmingerf6908082007-03-12 14:34:29 -07002386static const struct seq_operations fib_trie_seq_ops = {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002387 .start = fib_trie_seq_start,
2388 .next = fib_trie_seq_next,
2389 .stop = fib_trie_seq_stop,
2390 .show = fib_trie_seq_show,
Robert Olsson19baf832005-06-21 12:43:18 -07002391};
2392
2393static int fib_trie_seq_open(struct inode *inode, struct file *file)
2394{
Pavel Emelyanovcf7732e2007-10-10 02:29:29 -07002395 return seq_open_private(file, &fib_trie_seq_ops,
2396 sizeof(struct fib_trie_iter));
Robert Olsson19baf832005-06-21 12:43:18 -07002397}
2398
Arjan van de Ven9a321442007-02-12 00:55:35 -08002399static const struct file_operations fib_trie_fops = {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002400 .owner = THIS_MODULE,
2401 .open = fib_trie_seq_open,
2402 .read = seq_read,
2403 .llseek = seq_lseek,
2404 .release = seq_release_private,
2405};
2406
Al Viro32ab5f82006-09-26 22:21:45 -07002407static unsigned fib_flag_trans(int type, __be32 mask, const struct fib_info *fi)
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002408{
2409 static unsigned type2flags[RTN_MAX + 1] = {
2410 [7] = RTF_REJECT, [8] = RTF_REJECT,
2411 };
2412 unsigned flags = type2flags[type];
2413
2414 if (fi && fi->fib_nh->nh_gw)
2415 flags |= RTF_GATEWAY;
Al Viro32ab5f82006-09-26 22:21:45 -07002416 if (mask == htonl(0xFFFFFFFF))
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002417 flags |= RTF_HOST;
2418 flags |= RTF_UP;
2419 return flags;
2420}
2421
2422/*
2423 * This outputs /proc/net/route.
2424 * The format of the file is not supposed to be changed
2425 * and needs to be same as fib_hash output to avoid breaking
2426 * legacy utilities
2427 */
2428static int fib_route_seq_show(struct seq_file *seq, void *v)
2429{
Patrick McHardyc9e53cb2005-11-20 21:09:00 -08002430 const struct fib_trie_iter *iter = seq->private;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002431 struct leaf *l = v;
2432 int i;
2433 char bf[128];
2434
2435 if (v == SEQ_START_TOKEN) {
2436 seq_printf(seq, "%-127s\n", "Iface\tDestination\tGateway "
2437 "\tFlags\tRefCnt\tUse\tMetric\tMask\t\tMTU"
2438 "\tWindow\tIRTT");
2439 return 0;
2440 }
2441
Eric W. Biederman877a9bf2007-12-07 00:47:47 -08002442 if (iter->trie == iter->trie_local)
Patrick McHardyc9e53cb2005-11-20 21:09:00 -08002443 return 0;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002444 if (IS_TNODE(l))
2445 return 0;
2446
2447 for (i=32; i>=0; i--) {
Robert Olsson772cb712005-09-19 15:31:18 -07002448 struct leaf_info *li = find_leaf_info(l, i);
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002449 struct fib_alias *fa;
Al Viro32ab5f82006-09-26 22:21:45 -07002450 __be32 mask, prefix;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002451
2452 if (!li)
2453 continue;
2454
2455 mask = inet_make_mask(li->plen);
2456 prefix = htonl(l->key);
2457
2458 list_for_each_entry_rcu(fa, &li->falh, fa_list) {
Herbert Xu1371e372005-10-15 09:42:39 +10002459 const struct fib_info *fi = fa->fa_info;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002460 unsigned flags = fib_flag_trans(fa->fa_type, mask, fi);
2461
2462 if (fa->fa_type == RTN_BROADCAST
2463 || fa->fa_type == RTN_MULTICAST)
2464 continue;
2465
2466 if (fi)
2467 snprintf(bf, sizeof(bf),
2468 "%s\t%08X\t%08X\t%04X\t%d\t%u\t%d\t%08X\t%d\t%u\t%u",
2469 fi->fib_dev ? fi->fib_dev->name : "*",
2470 prefix,
2471 fi->fib_nh->nh_gw, flags, 0, 0,
2472 fi->fib_priority,
2473 mask,
2474 (fi->fib_advmss ? fi->fib_advmss + 40 : 0),
2475 fi->fib_window,
2476 fi->fib_rtt >> 3);
2477 else
2478 snprintf(bf, sizeof(bf),
2479 "*\t%08X\t%08X\t%04X\t%d\t%u\t%d\t%08X\t%d\t%u\t%u",
2480 prefix, 0, flags, 0, 0, 0,
2481 mask, 0, 0, 0);
2482
2483 seq_printf(seq, "%-127s\n", bf);
2484 }
2485 }
2486
2487 return 0;
2488}
2489
Stephen Hemmingerf6908082007-03-12 14:34:29 -07002490static const struct seq_operations fib_route_seq_ops = {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002491 .start = fib_trie_seq_start,
2492 .next = fib_trie_seq_next,
2493 .stop = fib_trie_seq_stop,
2494 .show = fib_route_seq_show,
2495};
2496
2497static int fib_route_seq_open(struct inode *inode, struct file *file)
2498{
Pavel Emelyanovcf7732e2007-10-10 02:29:29 -07002499 return seq_open_private(file, &fib_route_seq_ops,
2500 sizeof(struct fib_trie_iter));
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002501}
2502
Arjan van de Ven9a321442007-02-12 00:55:35 -08002503static const struct file_operations fib_route_fops = {
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002504 .owner = THIS_MODULE,
2505 .open = fib_route_seq_open,
2506 .read = seq_read,
2507 .llseek = seq_lseek,
2508 .release = seq_release_private,
Robert Olsson19baf832005-06-21 12:43:18 -07002509};
2510
Denis V. Lunev61a02652008-01-10 03:21:09 -08002511int __net_init fib_proc_init(struct net *net)
Robert Olsson19baf832005-06-21 12:43:18 -07002512{
Denis V. Lunev61a02652008-01-10 03:21:09 -08002513 if (!proc_net_fops_create(net, "fib_trie", S_IRUGO, &fib_trie_fops))
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002514 goto out1;
2515
Denis V. Lunev61a02652008-01-10 03:21:09 -08002516 if (!proc_net_fops_create(net, "fib_triestat", S_IRUGO,
2517 &fib_triestat_fops))
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002518 goto out2;
2519
Denis V. Lunev61a02652008-01-10 03:21:09 -08002520 if (!proc_net_fops_create(net, "route", S_IRUGO, &fib_route_fops))
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002521 goto out3;
2522
Robert Olsson19baf832005-06-21 12:43:18 -07002523 return 0;
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002524
2525out3:
Denis V. Lunev61a02652008-01-10 03:21:09 -08002526 proc_net_remove(net, "fib_triestat");
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002527out2:
Denis V. Lunev61a02652008-01-10 03:21:09 -08002528 proc_net_remove(net, "fib_trie");
Stephen Hemmingercb7b5932005-09-09 13:35:42 -07002529out1:
2530 return -ENOMEM;
Robert Olsson19baf832005-06-21 12:43:18 -07002531}
2532
Denis V. Lunev61a02652008-01-10 03:21:09 -08002533void __net_exit fib_proc_exit(struct net *net)
Robert Olsson19baf832005-06-21 12:43:18 -07002534{
Denis V. Lunev61a02652008-01-10 03:21:09 -08002535 proc_net_remove(net, "fib_trie");
2536 proc_net_remove(net, "fib_triestat");
2537 proc_net_remove(net, "route");
Robert Olsson19baf832005-06-21 12:43:18 -07002538}
2539
2540#endif /* CONFIG_PROC_FS */