Merge branch 'fix/hda' into for-linus
[deliverable/linux.git] / net / ipv4 / fib_trie.c
1 /*
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 *
10 * Jens Laas <jens.laas@data.slu.se> Swedish University of
11 * Agricultural Sciences.
12 *
13 * Hans Liss <hans.liss@its.uu.se> Uppsala Universitet
14 *
15 * This work is based on the LPC-trie which is originally descibed in:
16 *
17 * 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 *
26 * Code from fib_hash has been reused which includes the following header:
27 *
28 *
29 * INET An implementation of the TCP/IP protocol suite for the LINUX
30 * operating system. INET is implemented using the BSD Socket
31 * interface as the means of communication with the user level.
32 *
33 * IPv4 FIB: lookup engine and maintenance routines.
34 *
35 *
36 * Authors: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
37 *
38 * This program is free software; you can redistribute it and/or
39 * modify it under the terms of the GNU General Public License
40 * as published by the Free Software Foundation; either version
41 * 2 of the License, or (at your option) any later version.
42 *
43 * Substantial contributions to this work comes from:
44 *
45 * David S. Miller, <davem@davemloft.net>
46 * Stephen Hemminger <shemminger@osdl.org>
47 * Paul E. McKenney <paulmck@us.ibm.com>
48 * Patrick McHardy <kaber@trash.net>
49 */
50
51 #define VERSION "0.409"
52
53 #include <asm/uaccess.h>
54 #include <asm/system.h>
55 #include <linux/bitops.h>
56 #include <linux/types.h>
57 #include <linux/kernel.h>
58 #include <linux/mm.h>
59 #include <linux/string.h>
60 #include <linux/socket.h>
61 #include <linux/sockios.h>
62 #include <linux/errno.h>
63 #include <linux/in.h>
64 #include <linux/inet.h>
65 #include <linux/inetdevice.h>
66 #include <linux/netdevice.h>
67 #include <linux/if_arp.h>
68 #include <linux/proc_fs.h>
69 #include <linux/rcupdate.h>
70 #include <linux/skbuff.h>
71 #include <linux/netlink.h>
72 #include <linux/init.h>
73 #include <linux/list.h>
74 #include <linux/slab.h>
75 #include <net/net_namespace.h>
76 #include <net/ip.h>
77 #include <net/protocol.h>
78 #include <net/route.h>
79 #include <net/tcp.h>
80 #include <net/sock.h>
81 #include <net/ip_fib.h>
82 #include "fib_lookup.h"
83
84 #define MAX_STAT_DEPTH 32
85
86 #define KEYLENGTH (8*sizeof(t_key))
87
88 typedef unsigned int t_key;
89
90 #define T_TNODE 0
91 #define T_LEAF 1
92 #define NODE_TYPE_MASK 0x1UL
93 #define NODE_TYPE(node) ((node)->parent & NODE_TYPE_MASK)
94
95 #define IS_TNODE(n) (!(n->parent & T_LEAF))
96 #define IS_LEAF(n) (n->parent & T_LEAF)
97
98 struct node {
99 unsigned long parent;
100 t_key key;
101 };
102
103 struct leaf {
104 unsigned long parent;
105 t_key key;
106 struct hlist_head list;
107 struct rcu_head rcu;
108 };
109
110 struct leaf_info {
111 struct hlist_node hlist;
112 struct rcu_head rcu;
113 int plen;
114 struct list_head falh;
115 };
116
117 struct tnode {
118 unsigned long parent;
119 t_key key;
120 unsigned char pos; /* 2log(KEYLENGTH) bits needed */
121 unsigned char bits; /* 2log(KEYLENGTH) bits needed */
122 unsigned int full_children; /* KEYLENGTH bits needed */
123 unsigned int empty_children; /* KEYLENGTH bits needed */
124 union {
125 struct rcu_head rcu;
126 struct work_struct work;
127 struct tnode *tnode_free;
128 };
129 struct node *child[0];
130 };
131
132 #ifdef CONFIG_IP_FIB_TRIE_STATS
133 struct trie_use_stats {
134 unsigned int gets;
135 unsigned int backtrack;
136 unsigned int semantic_match_passed;
137 unsigned int semantic_match_miss;
138 unsigned int null_node_hit;
139 unsigned int resize_node_skipped;
140 };
141 #endif
142
143 struct trie_stat {
144 unsigned int totdepth;
145 unsigned int maxdepth;
146 unsigned int tnodes;
147 unsigned int leaves;
148 unsigned int nullpointers;
149 unsigned int prefixes;
150 unsigned int nodesizes[MAX_STAT_DEPTH];
151 };
152
153 struct trie {
154 struct node *trie;
155 #ifdef CONFIG_IP_FIB_TRIE_STATS
156 struct trie_use_stats stats;
157 #endif
158 };
159
160 static void put_child(struct trie *t, struct tnode *tn, int i, struct node *n);
161 static void tnode_put_child_reorg(struct tnode *tn, int i, struct node *n,
162 int wasfull);
163 static struct node *resize(struct trie *t, struct tnode *tn);
164 static struct tnode *inflate(struct trie *t, struct tnode *tn);
165 static struct tnode *halve(struct trie *t, struct tnode *tn);
166 /* tnodes to free after resize(); protected by RTNL */
167 static struct tnode *tnode_free_head;
168 static size_t tnode_free_size;
169
170 /*
171 * synchronize_rcu after call_rcu for that many pages; it should be especially
172 * useful before resizing the root node with PREEMPT_NONE configs; the value was
173 * obtained experimentally, aiming to avoid visible slowdown.
174 */
175 static const int sync_pages = 128;
176
177 static struct kmem_cache *fn_alias_kmem __read_mostly;
178 static struct kmem_cache *trie_leaf_kmem __read_mostly;
179
180 static inline struct tnode *node_parent(struct node *node)
181 {
182 return (struct tnode *)(node->parent & ~NODE_TYPE_MASK);
183 }
184
185 static inline struct tnode *node_parent_rcu(struct node *node)
186 {
187 struct tnode *ret = node_parent(node);
188
189 return rcu_dereference(ret);
190 }
191
192 /* Same as rcu_assign_pointer
193 * but that macro() assumes that value is a pointer.
194 */
195 static inline void node_set_parent(struct node *node, struct tnode *ptr)
196 {
197 smp_wmb();
198 node->parent = (unsigned long)ptr | NODE_TYPE(node);
199 }
200
201 static inline struct node *tnode_get_child(struct tnode *tn, unsigned int i)
202 {
203 BUG_ON(i >= 1U << tn->bits);
204
205 return tn->child[i];
206 }
207
208 static inline struct node *tnode_get_child_rcu(struct tnode *tn, unsigned int i)
209 {
210 struct node *ret = tnode_get_child(tn, i);
211
212 return rcu_dereference_check(ret,
213 rcu_read_lock_held() ||
214 lockdep_rtnl_is_held());
215 }
216
217 static inline int tnode_child_length(const struct tnode *tn)
218 {
219 return 1 << tn->bits;
220 }
221
222 static inline t_key mask_pfx(t_key k, unsigned short l)
223 {
224 return (l == 0) ? 0 : k >> (KEYLENGTH-l) << (KEYLENGTH-l);
225 }
226
227 static inline t_key tkey_extract_bits(t_key a, int offset, int bits)
228 {
229 if (offset < KEYLENGTH)
230 return ((t_key)(a << offset)) >> (KEYLENGTH - bits);
231 else
232 return 0;
233 }
234
235 static inline int tkey_equals(t_key a, t_key b)
236 {
237 return a == b;
238 }
239
240 static inline int tkey_sub_equals(t_key a, int offset, int bits, t_key b)
241 {
242 if (bits == 0 || offset >= KEYLENGTH)
243 return 1;
244 bits = bits > KEYLENGTH ? KEYLENGTH : bits;
245 return ((a ^ b) << offset) >> (KEYLENGTH - bits) == 0;
246 }
247
248 static inline int tkey_mismatch(t_key a, int offset, t_key b)
249 {
250 t_key diff = a ^ b;
251 int i = offset;
252
253 if (!diff)
254 return 0;
255 while ((diff << i) >> (KEYLENGTH-1) == 0)
256 i++;
257 return i;
258 }
259
260 /*
261 To understand this stuff, an understanding of keys and all their bits is
262 necessary. Every node in the trie has a key associated with it, but not
263 all of the bits in that key are significant.
264
265 Consider a node 'n' and its parent 'tp'.
266
267 If n is a leaf, every bit in its key is significant. Its presence is
268 necessitated by path compression, since during a tree traversal (when
269 searching for a leaf - unless we are doing an insertion) we will completely
270 ignore all skipped bits we encounter. Thus we need to verify, at the end of
271 a potentially successful search, that we have indeed been walking the
272 correct key path.
273
274 Note that we can never "miss" the correct key in the tree if present by
275 following the wrong path. Path compression ensures that segments of the key
276 that are the same for all keys with a given prefix are skipped, but the
277 skipped part *is* identical for each node in the subtrie below the skipped
278 bit! trie_insert() in this implementation takes care of that - note the
279 call to tkey_sub_equals() in trie_insert().
280
281 if n is an internal node - a 'tnode' here, the various parts of its key
282 have many different meanings.
283
284 Example:
285 _________________________________________________________________
286 | i | i | i | i | i | i | i | N | N | N | S | S | S | S | S | C |
287 -----------------------------------------------------------------
288 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
289
290 _________________________________________________________________
291 | C | C | C | u | u | u | u | u | u | u | u | u | u | u | u | u |
292 -----------------------------------------------------------------
293 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
294
295 tp->pos = 7
296 tp->bits = 3
297 n->pos = 15
298 n->bits = 4
299
300 First, let's just ignore the bits that come before the parent tp, that is
301 the bits from 0 to (tp->pos-1). They are *known* but at this point we do
302 not use them for anything.
303
304 The bits from (tp->pos) to (tp->pos + tp->bits - 1) - "N", above - are the
305 index into the parent's child array. That is, they will be used to find
306 'n' among tp's children.
307
308 The bits from (tp->pos + tp->bits) to (n->pos - 1) - "S" - are skipped bits
309 for the node n.
310
311 All the bits we have seen so far are significant to the node n. The rest
312 of the bits are really not needed or indeed known in n->key.
313
314 The bits from (n->pos) to (n->pos + n->bits - 1) - "C" - are the index into
315 n's child array, and will of course be different for each child.
316
317
318 The rest of the bits, from (n->pos + n->bits) onward, are completely unknown
319 at this point.
320
321 */
322
323 static inline void check_tnode(const struct tnode *tn)
324 {
325 WARN_ON(tn && tn->pos+tn->bits > 32);
326 }
327
328 static const int halve_threshold = 25;
329 static const int inflate_threshold = 50;
330 static const int halve_threshold_root = 15;
331 static const int inflate_threshold_root = 30;
332
333 static void __alias_free_mem(struct rcu_head *head)
334 {
335 struct fib_alias *fa = container_of(head, struct fib_alias, rcu);
336 kmem_cache_free(fn_alias_kmem, fa);
337 }
338
339 static inline void alias_free_mem_rcu(struct fib_alias *fa)
340 {
341 call_rcu(&fa->rcu, __alias_free_mem);
342 }
343
344 static void __leaf_free_rcu(struct rcu_head *head)
345 {
346 struct leaf *l = container_of(head, struct leaf, rcu);
347 kmem_cache_free(trie_leaf_kmem, l);
348 }
349
350 static inline void free_leaf(struct leaf *l)
351 {
352 call_rcu_bh(&l->rcu, __leaf_free_rcu);
353 }
354
355 static void __leaf_info_free_rcu(struct rcu_head *head)
356 {
357 kfree(container_of(head, struct leaf_info, rcu));
358 }
359
360 static inline void free_leaf_info(struct leaf_info *leaf)
361 {
362 call_rcu(&leaf->rcu, __leaf_info_free_rcu);
363 }
364
365 static struct tnode *tnode_alloc(size_t size)
366 {
367 if (size <= PAGE_SIZE)
368 return kzalloc(size, GFP_KERNEL);
369 else
370 return __vmalloc(size, GFP_KERNEL | __GFP_ZERO, PAGE_KERNEL);
371 }
372
373 static void __tnode_vfree(struct work_struct *arg)
374 {
375 struct tnode *tn = container_of(arg, struct tnode, work);
376 vfree(tn);
377 }
378
379 static void __tnode_free_rcu(struct rcu_head *head)
380 {
381 struct tnode *tn = container_of(head, struct tnode, rcu);
382 size_t size = sizeof(struct tnode) +
383 (sizeof(struct node *) << tn->bits);
384
385 if (size <= PAGE_SIZE)
386 kfree(tn);
387 else {
388 INIT_WORK(&tn->work, __tnode_vfree);
389 schedule_work(&tn->work);
390 }
391 }
392
393 static inline void tnode_free(struct tnode *tn)
394 {
395 if (IS_LEAF(tn))
396 free_leaf((struct leaf *) tn);
397 else
398 call_rcu(&tn->rcu, __tnode_free_rcu);
399 }
400
401 static void tnode_free_safe(struct tnode *tn)
402 {
403 BUG_ON(IS_LEAF(tn));
404 tn->tnode_free = tnode_free_head;
405 tnode_free_head = tn;
406 tnode_free_size += sizeof(struct tnode) +
407 (sizeof(struct node *) << tn->bits);
408 }
409
410 static void tnode_free_flush(void)
411 {
412 struct tnode *tn;
413
414 while ((tn = tnode_free_head)) {
415 tnode_free_head = tn->tnode_free;
416 tn->tnode_free = NULL;
417 tnode_free(tn);
418 }
419
420 if (tnode_free_size >= PAGE_SIZE * sync_pages) {
421 tnode_free_size = 0;
422 synchronize_rcu();
423 }
424 }
425
426 static struct leaf *leaf_new(void)
427 {
428 struct leaf *l = kmem_cache_alloc(trie_leaf_kmem, GFP_KERNEL);
429 if (l) {
430 l->parent = T_LEAF;
431 INIT_HLIST_HEAD(&l->list);
432 }
433 return l;
434 }
435
436 static struct leaf_info *leaf_info_new(int plen)
437 {
438 struct leaf_info *li = kmalloc(sizeof(struct leaf_info), GFP_KERNEL);
439 if (li) {
440 li->plen = plen;
441 INIT_LIST_HEAD(&li->falh);
442 }
443 return li;
444 }
445
446 static struct tnode *tnode_new(t_key key, int pos, int bits)
447 {
448 size_t sz = sizeof(struct tnode) + (sizeof(struct node *) << bits);
449 struct tnode *tn = tnode_alloc(sz);
450
451 if (tn) {
452 tn->parent = T_TNODE;
453 tn->pos = pos;
454 tn->bits = bits;
455 tn->key = key;
456 tn->full_children = 0;
457 tn->empty_children = 1<<bits;
458 }
459
460 pr_debug("AT %p s=%u %lu\n", tn, (unsigned int) sizeof(struct tnode),
461 (unsigned long) (sizeof(struct node) << bits));
462 return tn;
463 }
464
465 /*
466 * Check whether a tnode 'n' is "full", i.e. it is an internal node
467 * and no bits are skipped. See discussion in dyntree paper p. 6
468 */
469
470 static inline int tnode_full(const struct tnode *tn, const struct node *n)
471 {
472 if (n == NULL || IS_LEAF(n))
473 return 0;
474
475 return ((struct tnode *) n)->pos == tn->pos + tn->bits;
476 }
477
478 static inline void put_child(struct trie *t, struct tnode *tn, int i,
479 struct node *n)
480 {
481 tnode_put_child_reorg(tn, i, n, -1);
482 }
483
484 /*
485 * Add a child at position i overwriting the old value.
486 * Update the value of full_children and empty_children.
487 */
488
489 static void tnode_put_child_reorg(struct tnode *tn, int i, struct node *n,
490 int wasfull)
491 {
492 struct node *chi = tn->child[i];
493 int isfull;
494
495 BUG_ON(i >= 1<<tn->bits);
496
497 /* update emptyChildren */
498 if (n == NULL && chi != NULL)
499 tn->empty_children++;
500 else if (n != NULL && chi == NULL)
501 tn->empty_children--;
502
503 /* update fullChildren */
504 if (wasfull == -1)
505 wasfull = tnode_full(tn, chi);
506
507 isfull = tnode_full(tn, n);
508 if (wasfull && !isfull)
509 tn->full_children--;
510 else if (!wasfull && isfull)
511 tn->full_children++;
512
513 if (n)
514 node_set_parent(n, tn);
515
516 rcu_assign_pointer(tn->child[i], n);
517 }
518
519 #define MAX_WORK 10
520 static struct node *resize(struct trie *t, struct tnode *tn)
521 {
522 int i;
523 struct tnode *old_tn;
524 int inflate_threshold_use;
525 int halve_threshold_use;
526 int max_work;
527
528 if (!tn)
529 return NULL;
530
531 pr_debug("In tnode_resize %p inflate_threshold=%d threshold=%d\n",
532 tn, inflate_threshold, halve_threshold);
533
534 /* No children */
535 if (tn->empty_children == tnode_child_length(tn)) {
536 tnode_free_safe(tn);
537 return NULL;
538 }
539 /* One child */
540 if (tn->empty_children == tnode_child_length(tn) - 1)
541 goto one_child;
542 /*
543 * Double as long as the resulting node has a number of
544 * nonempty nodes that are above the threshold.
545 */
546
547 /*
548 * From "Implementing a dynamic compressed trie" by Stefan Nilsson of
549 * the Helsinki University of Technology and Matti Tikkanen of Nokia
550 * Telecommunications, page 6:
551 * "A node is doubled if the ratio of non-empty children to all
552 * children in the *doubled* node is at least 'high'."
553 *
554 * 'high' in this instance is the variable 'inflate_threshold'. It
555 * is expressed as a percentage, so we multiply it with
556 * tnode_child_length() and instead of multiplying by 2 (since the
557 * child array will be doubled by inflate()) and multiplying
558 * the left-hand side by 100 (to handle the percentage thing) we
559 * multiply the left-hand side by 50.
560 *
561 * The left-hand side may look a bit weird: tnode_child_length(tn)
562 * - tn->empty_children is of course the number of non-null children
563 * in the current node. tn->full_children is the number of "full"
564 * children, that is non-null tnodes with a skip value of 0.
565 * All of those will be doubled in the resulting inflated tnode, so
566 * we just count them one extra time here.
567 *
568 * A clearer way to write this would be:
569 *
570 * to_be_doubled = tn->full_children;
571 * not_to_be_doubled = tnode_child_length(tn) - tn->empty_children -
572 * tn->full_children;
573 *
574 * new_child_length = tnode_child_length(tn) * 2;
575 *
576 * new_fill_factor = 100 * (not_to_be_doubled + 2*to_be_doubled) /
577 * new_child_length;
578 * if (new_fill_factor >= inflate_threshold)
579 *
580 * ...and so on, tho it would mess up the while () loop.
581 *
582 * anyway,
583 * 100 * (not_to_be_doubled + 2*to_be_doubled) / new_child_length >=
584 * inflate_threshold
585 *
586 * avoid a division:
587 * 100 * (not_to_be_doubled + 2*to_be_doubled) >=
588 * inflate_threshold * new_child_length
589 *
590 * expand not_to_be_doubled and to_be_doubled, and shorten:
591 * 100 * (tnode_child_length(tn) - tn->empty_children +
592 * tn->full_children) >= inflate_threshold * new_child_length
593 *
594 * expand new_child_length:
595 * 100 * (tnode_child_length(tn) - tn->empty_children +
596 * tn->full_children) >=
597 * inflate_threshold * tnode_child_length(tn) * 2
598 *
599 * shorten again:
600 * 50 * (tn->full_children + tnode_child_length(tn) -
601 * tn->empty_children) >= inflate_threshold *
602 * tnode_child_length(tn)
603 *
604 */
605
606 check_tnode(tn);
607
608 /* Keep root node larger */
609
610 if (!node_parent((struct node*) tn)) {
611 inflate_threshold_use = inflate_threshold_root;
612 halve_threshold_use = halve_threshold_root;
613 }
614 else {
615 inflate_threshold_use = inflate_threshold;
616 halve_threshold_use = halve_threshold;
617 }
618
619 max_work = MAX_WORK;
620 while ((tn->full_children > 0 && max_work-- &&
621 50 * (tn->full_children + tnode_child_length(tn)
622 - tn->empty_children)
623 >= inflate_threshold_use * tnode_child_length(tn))) {
624
625 old_tn = tn;
626 tn = inflate(t, tn);
627
628 if (IS_ERR(tn)) {
629 tn = old_tn;
630 #ifdef CONFIG_IP_FIB_TRIE_STATS
631 t->stats.resize_node_skipped++;
632 #endif
633 break;
634 }
635 }
636
637 check_tnode(tn);
638
639 /* Return if at least one inflate is run */
640 if( max_work != MAX_WORK)
641 return (struct node *) tn;
642
643 /*
644 * Halve as long as the number of empty children in this
645 * node is above threshold.
646 */
647
648 max_work = MAX_WORK;
649 while (tn->bits > 1 && max_work-- &&
650 100 * (tnode_child_length(tn) - tn->empty_children) <
651 halve_threshold_use * tnode_child_length(tn)) {
652
653 old_tn = tn;
654 tn = halve(t, tn);
655 if (IS_ERR(tn)) {
656 tn = old_tn;
657 #ifdef CONFIG_IP_FIB_TRIE_STATS
658 t->stats.resize_node_skipped++;
659 #endif
660 break;
661 }
662 }
663
664
665 /* Only one child remains */
666 if (tn->empty_children == tnode_child_length(tn) - 1) {
667 one_child:
668 for (i = 0; i < tnode_child_length(tn); i++) {
669 struct node *n;
670
671 n = tn->child[i];
672 if (!n)
673 continue;
674
675 /* compress one level */
676
677 node_set_parent(n, NULL);
678 tnode_free_safe(tn);
679 return n;
680 }
681 }
682 return (struct node *) tn;
683 }
684
685 static struct tnode *inflate(struct trie *t, struct tnode *tn)
686 {
687 struct tnode *oldtnode = tn;
688 int olen = tnode_child_length(tn);
689 int i;
690
691 pr_debug("In inflate\n");
692
693 tn = tnode_new(oldtnode->key, oldtnode->pos, oldtnode->bits + 1);
694
695 if (!tn)
696 return ERR_PTR(-ENOMEM);
697
698 /*
699 * Preallocate and store tnodes before the actual work so we
700 * don't get into an inconsistent state if memory allocation
701 * fails. In case of failure we return the oldnode and inflate
702 * of tnode is ignored.
703 */
704
705 for (i = 0; i < olen; i++) {
706 struct tnode *inode;
707
708 inode = (struct tnode *) tnode_get_child(oldtnode, i);
709 if (inode &&
710 IS_TNODE(inode) &&
711 inode->pos == oldtnode->pos + oldtnode->bits &&
712 inode->bits > 1) {
713 struct tnode *left, *right;
714 t_key m = ~0U << (KEYLENGTH - 1) >> inode->pos;
715
716 left = tnode_new(inode->key&(~m), inode->pos + 1,
717 inode->bits - 1);
718 if (!left)
719 goto nomem;
720
721 right = tnode_new(inode->key|m, inode->pos + 1,
722 inode->bits - 1);
723
724 if (!right) {
725 tnode_free(left);
726 goto nomem;
727 }
728
729 put_child(t, tn, 2*i, (struct node *) left);
730 put_child(t, tn, 2*i+1, (struct node *) right);
731 }
732 }
733
734 for (i = 0; i < olen; i++) {
735 struct tnode *inode;
736 struct node *node = tnode_get_child(oldtnode, i);
737 struct tnode *left, *right;
738 int size, j;
739
740 /* An empty child */
741 if (node == NULL)
742 continue;
743
744 /* A leaf or an internal node with skipped bits */
745
746 if (IS_LEAF(node) || ((struct tnode *) node)->pos >
747 tn->pos + tn->bits - 1) {
748 if (tkey_extract_bits(node->key,
749 oldtnode->pos + oldtnode->bits,
750 1) == 0)
751 put_child(t, tn, 2*i, node);
752 else
753 put_child(t, tn, 2*i+1, node);
754 continue;
755 }
756
757 /* An internal node with two children */
758 inode = (struct tnode *) node;
759
760 if (inode->bits == 1) {
761 put_child(t, tn, 2*i, inode->child[0]);
762 put_child(t, tn, 2*i+1, inode->child[1]);
763
764 tnode_free_safe(inode);
765 continue;
766 }
767
768 /* An internal node with more than two children */
769
770 /* We will replace this node 'inode' with two new
771 * ones, 'left' and 'right', each with half of the
772 * original children. The two new nodes will have
773 * a position one bit further down the key and this
774 * means that the "significant" part of their keys
775 * (see the discussion near the top of this file)
776 * will differ by one bit, which will be "0" in
777 * left's key and "1" in right's key. Since we are
778 * moving the key position by one step, the bit that
779 * we are moving away from - the bit at position
780 * (inode->pos) - is the one that will differ between
781 * left and right. So... we synthesize that bit in the
782 * two new keys.
783 * The mask 'm' below will be a single "one" bit at
784 * the position (inode->pos)
785 */
786
787 /* Use the old key, but set the new significant
788 * bit to zero.
789 */
790
791 left = (struct tnode *) tnode_get_child(tn, 2*i);
792 put_child(t, tn, 2*i, NULL);
793
794 BUG_ON(!left);
795
796 right = (struct tnode *) tnode_get_child(tn, 2*i+1);
797 put_child(t, tn, 2*i+1, NULL);
798
799 BUG_ON(!right);
800
801 size = tnode_child_length(left);
802 for (j = 0; j < size; j++) {
803 put_child(t, left, j, inode->child[j]);
804 put_child(t, right, j, inode->child[j + size]);
805 }
806 put_child(t, tn, 2*i, resize(t, left));
807 put_child(t, tn, 2*i+1, resize(t, right));
808
809 tnode_free_safe(inode);
810 }
811 tnode_free_safe(oldtnode);
812 return tn;
813 nomem:
814 {
815 int size = tnode_child_length(tn);
816 int j;
817
818 for (j = 0; j < size; j++)
819 if (tn->child[j])
820 tnode_free((struct tnode *)tn->child[j]);
821
822 tnode_free(tn);
823
824 return ERR_PTR(-ENOMEM);
825 }
826 }
827
828 static struct tnode *halve(struct trie *t, struct tnode *tn)
829 {
830 struct tnode *oldtnode = tn;
831 struct node *left, *right;
832 int i;
833 int olen = tnode_child_length(tn);
834
835 pr_debug("In halve\n");
836
837 tn = tnode_new(oldtnode->key, oldtnode->pos, oldtnode->bits - 1);
838
839 if (!tn)
840 return ERR_PTR(-ENOMEM);
841
842 /*
843 * Preallocate and store tnodes before the actual work so we
844 * don't get into an inconsistent state if memory allocation
845 * fails. In case of failure we return the oldnode and halve
846 * of tnode is ignored.
847 */
848
849 for (i = 0; i < olen; i += 2) {
850 left = tnode_get_child(oldtnode, i);
851 right = tnode_get_child(oldtnode, i+1);
852
853 /* Two nonempty children */
854 if (left && right) {
855 struct tnode *newn;
856
857 newn = tnode_new(left->key, tn->pos + tn->bits, 1);
858
859 if (!newn)
860 goto nomem;
861
862 put_child(t, tn, i/2, (struct node *)newn);
863 }
864
865 }
866
867 for (i = 0; i < olen; i += 2) {
868 struct tnode *newBinNode;
869
870 left = tnode_get_child(oldtnode, i);
871 right = tnode_get_child(oldtnode, i+1);
872
873 /* At least one of the children is empty */
874 if (left == NULL) {
875 if (right == NULL) /* Both are empty */
876 continue;
877 put_child(t, tn, i/2, right);
878 continue;
879 }
880
881 if (right == NULL) {
882 put_child(t, tn, i/2, left);
883 continue;
884 }
885
886 /* Two nonempty children */
887 newBinNode = (struct tnode *) tnode_get_child(tn, i/2);
888 put_child(t, tn, i/2, NULL);
889 put_child(t, newBinNode, 0, left);
890 put_child(t, newBinNode, 1, right);
891 put_child(t, tn, i/2, resize(t, newBinNode));
892 }
893 tnode_free_safe(oldtnode);
894 return tn;
895 nomem:
896 {
897 int size = tnode_child_length(tn);
898 int j;
899
900 for (j = 0; j < size; j++)
901 if (tn->child[j])
902 tnode_free((struct tnode *)tn->child[j]);
903
904 tnode_free(tn);
905
906 return ERR_PTR(-ENOMEM);
907 }
908 }
909
910 /* readside must use rcu_read_lock currently dump routines
911 via get_fa_head and dump */
912
913 static struct leaf_info *find_leaf_info(struct leaf *l, int plen)
914 {
915 struct hlist_head *head = &l->list;
916 struct hlist_node *node;
917 struct leaf_info *li;
918
919 hlist_for_each_entry_rcu(li, node, head, hlist)
920 if (li->plen == plen)
921 return li;
922
923 return NULL;
924 }
925
926 static inline struct list_head *get_fa_head(struct leaf *l, int plen)
927 {
928 struct leaf_info *li = find_leaf_info(l, plen);
929
930 if (!li)
931 return NULL;
932
933 return &li->falh;
934 }
935
936 static void insert_leaf_info(struct hlist_head *head, struct leaf_info *new)
937 {
938 struct leaf_info *li = NULL, *last = NULL;
939 struct hlist_node *node;
940
941 if (hlist_empty(head)) {
942 hlist_add_head_rcu(&new->hlist, head);
943 } else {
944 hlist_for_each_entry(li, node, head, hlist) {
945 if (new->plen > li->plen)
946 break;
947
948 last = li;
949 }
950 if (last)
951 hlist_add_after_rcu(&last->hlist, &new->hlist);
952 else
953 hlist_add_before_rcu(&new->hlist, &li->hlist);
954 }
955 }
956
957 /* rcu_read_lock needs to be hold by caller from readside */
958
959 static struct leaf *
960 fib_find_node(struct trie *t, u32 key)
961 {
962 int pos;
963 struct tnode *tn;
964 struct node *n;
965
966 pos = 0;
967 n = rcu_dereference_check(t->trie,
968 rcu_read_lock_held() ||
969 lockdep_rtnl_is_held());
970
971 while (n != NULL && NODE_TYPE(n) == T_TNODE) {
972 tn = (struct tnode *) n;
973
974 check_tnode(tn);
975
976 if (tkey_sub_equals(tn->key, pos, tn->pos-pos, key)) {
977 pos = tn->pos + tn->bits;
978 n = tnode_get_child_rcu(tn,
979 tkey_extract_bits(key,
980 tn->pos,
981 tn->bits));
982 } else
983 break;
984 }
985 /* Case we have found a leaf. Compare prefixes */
986
987 if (n != NULL && IS_LEAF(n) && tkey_equals(key, n->key))
988 return (struct leaf *)n;
989
990 return NULL;
991 }
992
993 static void trie_rebalance(struct trie *t, struct tnode *tn)
994 {
995 int wasfull;
996 t_key cindex, key;
997 struct tnode *tp;
998
999 key = tn->key;
1000
1001 while (tn != NULL && (tp = node_parent((struct node *)tn)) != NULL) {
1002 cindex = tkey_extract_bits(key, tp->pos, tp->bits);
1003 wasfull = tnode_full(tp, tnode_get_child(tp, cindex));
1004 tn = (struct tnode *) resize(t, (struct tnode *)tn);
1005
1006 tnode_put_child_reorg((struct tnode *)tp, cindex,
1007 (struct node *)tn, wasfull);
1008
1009 tp = node_parent((struct node *) tn);
1010 if (!tp)
1011 rcu_assign_pointer(t->trie, (struct node *)tn);
1012
1013 tnode_free_flush();
1014 if (!tp)
1015 break;
1016 tn = tp;
1017 }
1018
1019 /* Handle last (top) tnode */
1020 if (IS_TNODE(tn))
1021 tn = (struct tnode *)resize(t, (struct tnode *)tn);
1022
1023 rcu_assign_pointer(t->trie, (struct node *)tn);
1024 tnode_free_flush();
1025
1026 return;
1027 }
1028
1029 /* only used from updater-side */
1030
1031 static struct list_head *fib_insert_node(struct trie *t, u32 key, int plen)
1032 {
1033 int pos, newpos;
1034 struct tnode *tp = NULL, *tn = NULL;
1035 struct node *n;
1036 struct leaf *l;
1037 int missbit;
1038 struct list_head *fa_head = NULL;
1039 struct leaf_info *li;
1040 t_key cindex;
1041
1042 pos = 0;
1043 n = t->trie;
1044
1045 /* If we point to NULL, stop. Either the tree is empty and we should
1046 * just put a new leaf in if, or we have reached an empty child slot,
1047 * and we should just put our new leaf in that.
1048 * If we point to a T_TNODE, check if it matches our key. Note that
1049 * a T_TNODE might be skipping any number of bits - its 'pos' need
1050 * not be the parent's 'pos'+'bits'!
1051 *
1052 * If it does match the current key, get pos/bits from it, extract
1053 * the index from our key, push the T_TNODE and walk the tree.
1054 *
1055 * If it doesn't, we have to replace it with a new T_TNODE.
1056 *
1057 * If we point to a T_LEAF, it might or might not have the same key
1058 * as we do. If it does, just change the value, update the T_LEAF's
1059 * value, and return it.
1060 * If it doesn't, we need to replace it with a T_TNODE.
1061 */
1062
1063 while (n != NULL && NODE_TYPE(n) == T_TNODE) {
1064 tn = (struct tnode *) n;
1065
1066 check_tnode(tn);
1067
1068 if (tkey_sub_equals(tn->key, pos, tn->pos-pos, key)) {
1069 tp = tn;
1070 pos = tn->pos + tn->bits;
1071 n = tnode_get_child(tn,
1072 tkey_extract_bits(key,
1073 tn->pos,
1074 tn->bits));
1075
1076 BUG_ON(n && node_parent(n) != tn);
1077 } else
1078 break;
1079 }
1080
1081 /*
1082 * n ----> NULL, LEAF or TNODE
1083 *
1084 * tp is n's (parent) ----> NULL or TNODE
1085 */
1086
1087 BUG_ON(tp && IS_LEAF(tp));
1088
1089 /* Case 1: n is a leaf. Compare prefixes */
1090
1091 if (n != NULL && IS_LEAF(n) && tkey_equals(key, n->key)) {
1092 l = (struct leaf *) n;
1093 li = leaf_info_new(plen);
1094
1095 if (!li)
1096 return NULL;
1097
1098 fa_head = &li->falh;
1099 insert_leaf_info(&l->list, li);
1100 goto done;
1101 }
1102 l = leaf_new();
1103
1104 if (!l)
1105 return NULL;
1106
1107 l->key = key;
1108 li = leaf_info_new(plen);
1109
1110 if (!li) {
1111 free_leaf(l);
1112 return NULL;
1113 }
1114
1115 fa_head = &li->falh;
1116 insert_leaf_info(&l->list, li);
1117
1118 if (t->trie && n == NULL) {
1119 /* Case 2: n is NULL, and will just insert a new leaf */
1120
1121 node_set_parent((struct node *)l, tp);
1122
1123 cindex = tkey_extract_bits(key, tp->pos, tp->bits);
1124 put_child(t, (struct tnode *)tp, cindex, (struct node *)l);
1125 } else {
1126 /* Case 3: n is a LEAF or a TNODE and the key doesn't match. */
1127 /*
1128 * Add a new tnode here
1129 * first tnode need some special handling
1130 */
1131
1132 if (tp)
1133 pos = tp->pos+tp->bits;
1134 else
1135 pos = 0;
1136
1137 if (n) {
1138 newpos = tkey_mismatch(key, pos, n->key);
1139 tn = tnode_new(n->key, newpos, 1);
1140 } else {
1141 newpos = 0;
1142 tn = tnode_new(key, newpos, 1); /* First tnode */
1143 }
1144
1145 if (!tn) {
1146 free_leaf_info(li);
1147 free_leaf(l);
1148 return NULL;
1149 }
1150
1151 node_set_parent((struct node *)tn, tp);
1152
1153 missbit = tkey_extract_bits(key, newpos, 1);
1154 put_child(t, tn, missbit, (struct node *)l);
1155 put_child(t, tn, 1-missbit, n);
1156
1157 if (tp) {
1158 cindex = tkey_extract_bits(key, tp->pos, tp->bits);
1159 put_child(t, (struct tnode *)tp, cindex,
1160 (struct node *)tn);
1161 } else {
1162 rcu_assign_pointer(t->trie, (struct node *)tn);
1163 tp = tn;
1164 }
1165 }
1166
1167 if (tp && tp->pos + tp->bits > 32)
1168 pr_warning("fib_trie"
1169 " tp=%p pos=%d, bits=%d, key=%0x plen=%d\n",
1170 tp, tp->pos, tp->bits, key, plen);
1171
1172 /* Rebalance the trie */
1173
1174 trie_rebalance(t, tp);
1175 done:
1176 return fa_head;
1177 }
1178
1179 /*
1180 * Caller must hold RTNL.
1181 */
1182 int fib_table_insert(struct fib_table *tb, struct fib_config *cfg)
1183 {
1184 struct trie *t = (struct trie *) tb->tb_data;
1185 struct fib_alias *fa, *new_fa;
1186 struct list_head *fa_head = NULL;
1187 struct fib_info *fi;
1188 int plen = cfg->fc_dst_len;
1189 u8 tos = cfg->fc_tos;
1190 u32 key, mask;
1191 int err;
1192 struct leaf *l;
1193
1194 if (plen > 32)
1195 return -EINVAL;
1196
1197 key = ntohl(cfg->fc_dst);
1198
1199 pr_debug("Insert table=%u %08x/%d\n", tb->tb_id, key, plen);
1200
1201 mask = ntohl(inet_make_mask(plen));
1202
1203 if (key & ~mask)
1204 return -EINVAL;
1205
1206 key = key & mask;
1207
1208 fi = fib_create_info(cfg);
1209 if (IS_ERR(fi)) {
1210 err = PTR_ERR(fi);
1211 goto err;
1212 }
1213
1214 l = fib_find_node(t, key);
1215 fa = NULL;
1216
1217 if (l) {
1218 fa_head = get_fa_head(l, plen);
1219 fa = fib_find_alias(fa_head, tos, fi->fib_priority);
1220 }
1221
1222 /* Now fa, if non-NULL, points to the first fib alias
1223 * with the same keys [prefix,tos,priority], if such key already
1224 * exists or to the node before which we will insert new one.
1225 *
1226 * If fa is NULL, we will need to allocate a new one and
1227 * insert to the head of f.
1228 *
1229 * If f is NULL, no fib node matched the destination key
1230 * and we need to allocate a new one of those as well.
1231 */
1232
1233 if (fa && fa->fa_tos == tos &&
1234 fa->fa_info->fib_priority == fi->fib_priority) {
1235 struct fib_alias *fa_first, *fa_match;
1236
1237 err = -EEXIST;
1238 if (cfg->fc_nlflags & NLM_F_EXCL)
1239 goto out;
1240
1241 /* We have 2 goals:
1242 * 1. Find exact match for type, scope, fib_info to avoid
1243 * duplicate routes
1244 * 2. Find next 'fa' (or head), NLM_F_APPEND inserts before it
1245 */
1246 fa_match = NULL;
1247 fa_first = fa;
1248 fa = list_entry(fa->fa_list.prev, struct fib_alias, fa_list);
1249 list_for_each_entry_continue(fa, fa_head, fa_list) {
1250 if (fa->fa_tos != tos)
1251 break;
1252 if (fa->fa_info->fib_priority != fi->fib_priority)
1253 break;
1254 if (fa->fa_type == cfg->fc_type &&
1255 fa->fa_scope == cfg->fc_scope &&
1256 fa->fa_info == fi) {
1257 fa_match = fa;
1258 break;
1259 }
1260 }
1261
1262 if (cfg->fc_nlflags & NLM_F_REPLACE) {
1263 struct fib_info *fi_drop;
1264 u8 state;
1265
1266 fa = fa_first;
1267 if (fa_match) {
1268 if (fa == fa_match)
1269 err = 0;
1270 goto out;
1271 }
1272 err = -ENOBUFS;
1273 new_fa = kmem_cache_alloc(fn_alias_kmem, GFP_KERNEL);
1274 if (new_fa == NULL)
1275 goto out;
1276
1277 fi_drop = fa->fa_info;
1278 new_fa->fa_tos = fa->fa_tos;
1279 new_fa->fa_info = fi;
1280 new_fa->fa_type = cfg->fc_type;
1281 new_fa->fa_scope = cfg->fc_scope;
1282 state = fa->fa_state;
1283 new_fa->fa_state = state & ~FA_S_ACCESSED;
1284
1285 list_replace_rcu(&fa->fa_list, &new_fa->fa_list);
1286 alias_free_mem_rcu(fa);
1287
1288 fib_release_info(fi_drop);
1289 if (state & FA_S_ACCESSED)
1290 rt_cache_flush(cfg->fc_nlinfo.nl_net, -1);
1291 rtmsg_fib(RTM_NEWROUTE, htonl(key), new_fa, plen,
1292 tb->tb_id, &cfg->fc_nlinfo, NLM_F_REPLACE);
1293
1294 goto succeeded;
1295 }
1296 /* Error if we find a perfect match which
1297 * uses the same scope, type, and nexthop
1298 * information.
1299 */
1300 if (fa_match)
1301 goto out;
1302
1303 if (!(cfg->fc_nlflags & NLM_F_APPEND))
1304 fa = fa_first;
1305 }
1306 err = -ENOENT;
1307 if (!(cfg->fc_nlflags & NLM_F_CREATE))
1308 goto out;
1309
1310 err = -ENOBUFS;
1311 new_fa = kmem_cache_alloc(fn_alias_kmem, GFP_KERNEL);
1312 if (new_fa == NULL)
1313 goto out;
1314
1315 new_fa->fa_info = fi;
1316 new_fa->fa_tos = tos;
1317 new_fa->fa_type = cfg->fc_type;
1318 new_fa->fa_scope = cfg->fc_scope;
1319 new_fa->fa_state = 0;
1320 /*
1321 * Insert new entry to the list.
1322 */
1323
1324 if (!fa_head) {
1325 fa_head = fib_insert_node(t, key, plen);
1326 if (unlikely(!fa_head)) {
1327 err = -ENOMEM;
1328 goto out_free_new_fa;
1329 }
1330 }
1331
1332 list_add_tail_rcu(&new_fa->fa_list,
1333 (fa ? &fa->fa_list : fa_head));
1334
1335 rt_cache_flush(cfg->fc_nlinfo.nl_net, -1);
1336 rtmsg_fib(RTM_NEWROUTE, htonl(key), new_fa, plen, tb->tb_id,
1337 &cfg->fc_nlinfo, 0);
1338 succeeded:
1339 return 0;
1340
1341 out_free_new_fa:
1342 kmem_cache_free(fn_alias_kmem, new_fa);
1343 out:
1344 fib_release_info(fi);
1345 err:
1346 return err;
1347 }
1348
1349 /* should be called with rcu_read_lock */
1350 static int check_leaf(struct trie *t, struct leaf *l,
1351 t_key key, const struct flowi *flp,
1352 struct fib_result *res)
1353 {
1354 struct leaf_info *li;
1355 struct hlist_head *hhead = &l->list;
1356 struct hlist_node *node;
1357
1358 hlist_for_each_entry_rcu(li, node, hhead, hlist) {
1359 int err;
1360 int plen = li->plen;
1361 __be32 mask = inet_make_mask(plen);
1362
1363 if (l->key != (key & ntohl(mask)))
1364 continue;
1365
1366 err = fib_semantic_match(&li->falh, flp, res, plen);
1367
1368 #ifdef CONFIG_IP_FIB_TRIE_STATS
1369 if (err <= 0)
1370 t->stats.semantic_match_passed++;
1371 else
1372 t->stats.semantic_match_miss++;
1373 #endif
1374 if (err <= 0)
1375 return err;
1376 }
1377
1378 return 1;
1379 }
1380
1381 int fib_table_lookup(struct fib_table *tb, const struct flowi *flp,
1382 struct fib_result *res)
1383 {
1384 struct trie *t = (struct trie *) tb->tb_data;
1385 int ret;
1386 struct node *n;
1387 struct tnode *pn;
1388 int pos, bits;
1389 t_key key = ntohl(flp->fl4_dst);
1390 int chopped_off;
1391 t_key cindex = 0;
1392 int current_prefix_length = KEYLENGTH;
1393 struct tnode *cn;
1394 t_key node_prefix, key_prefix, pref_mismatch;
1395 int mp;
1396
1397 rcu_read_lock();
1398
1399 n = rcu_dereference(t->trie);
1400 if (!n)
1401 goto failed;
1402
1403 #ifdef CONFIG_IP_FIB_TRIE_STATS
1404 t->stats.gets++;
1405 #endif
1406
1407 /* Just a leaf? */
1408 if (IS_LEAF(n)) {
1409 ret = check_leaf(t, (struct leaf *)n, key, flp, res);
1410 goto found;
1411 }
1412
1413 pn = (struct tnode *) n;
1414 chopped_off = 0;
1415
1416 while (pn) {
1417 pos = pn->pos;
1418 bits = pn->bits;
1419
1420 if (!chopped_off)
1421 cindex = tkey_extract_bits(mask_pfx(key, current_prefix_length),
1422 pos, bits);
1423
1424 n = tnode_get_child_rcu(pn, cindex);
1425
1426 if (n == NULL) {
1427 #ifdef CONFIG_IP_FIB_TRIE_STATS
1428 t->stats.null_node_hit++;
1429 #endif
1430 goto backtrace;
1431 }
1432
1433 if (IS_LEAF(n)) {
1434 ret = check_leaf(t, (struct leaf *)n, key, flp, res);
1435 if (ret > 0)
1436 goto backtrace;
1437 goto found;
1438 }
1439
1440 cn = (struct tnode *)n;
1441
1442 /*
1443 * It's a tnode, and we can do some extra checks here if we
1444 * like, to avoid descending into a dead-end branch.
1445 * This tnode is in the parent's child array at index
1446 * key[p_pos..p_pos+p_bits] but potentially with some bits
1447 * chopped off, so in reality the index may be just a
1448 * subprefix, padded with zero at the end.
1449 * We can also take a look at any skipped bits in this
1450 * tnode - everything up to p_pos is supposed to be ok,
1451 * and the non-chopped bits of the index (se previous
1452 * paragraph) are also guaranteed ok, but the rest is
1453 * considered unknown.
1454 *
1455 * The skipped bits are key[pos+bits..cn->pos].
1456 */
1457
1458 /* If current_prefix_length < pos+bits, we are already doing
1459 * actual prefix matching, which means everything from
1460 * pos+(bits-chopped_off) onward must be zero along some
1461 * branch of this subtree - otherwise there is *no* valid
1462 * prefix present. Here we can only check the skipped
1463 * bits. Remember, since we have already indexed into the
1464 * parent's child array, we know that the bits we chopped of
1465 * *are* zero.
1466 */
1467
1468 /* NOTA BENE: Checking only skipped bits
1469 for the new node here */
1470
1471 if (current_prefix_length < pos+bits) {
1472 if (tkey_extract_bits(cn->key, current_prefix_length,
1473 cn->pos - current_prefix_length)
1474 || !(cn->child[0]))
1475 goto backtrace;
1476 }
1477
1478 /*
1479 * If chopped_off=0, the index is fully validated and we
1480 * only need to look at the skipped bits for this, the new,
1481 * tnode. What we actually want to do is to find out if
1482 * these skipped bits match our key perfectly, or if we will
1483 * have to count on finding a matching prefix further down,
1484 * because if we do, we would like to have some way of
1485 * verifying the existence of such a prefix at this point.
1486 */
1487
1488 /* The only thing we can do at this point is to verify that
1489 * any such matching prefix can indeed be a prefix to our
1490 * key, and if the bits in the node we are inspecting that
1491 * do not match our key are not ZERO, this cannot be true.
1492 * Thus, find out where there is a mismatch (before cn->pos)
1493 * and verify that all the mismatching bits are zero in the
1494 * new tnode's key.
1495 */
1496
1497 /*
1498 * Note: We aren't very concerned about the piece of
1499 * the key that precede pn->pos+pn->bits, since these
1500 * have already been checked. The bits after cn->pos
1501 * aren't checked since these are by definition
1502 * "unknown" at this point. Thus, what we want to see
1503 * is if we are about to enter the "prefix matching"
1504 * state, and in that case verify that the skipped
1505 * bits that will prevail throughout this subtree are
1506 * zero, as they have to be if we are to find a
1507 * matching prefix.
1508 */
1509
1510 node_prefix = mask_pfx(cn->key, cn->pos);
1511 key_prefix = mask_pfx(key, cn->pos);
1512 pref_mismatch = key_prefix^node_prefix;
1513 mp = 0;
1514
1515 /*
1516 * In short: If skipped bits in this node do not match
1517 * the search key, enter the "prefix matching"
1518 * state.directly.
1519 */
1520 if (pref_mismatch) {
1521 while (!(pref_mismatch & (1<<(KEYLENGTH-1)))) {
1522 mp++;
1523 pref_mismatch = pref_mismatch << 1;
1524 }
1525 key_prefix = tkey_extract_bits(cn->key, mp, cn->pos-mp);
1526
1527 if (key_prefix != 0)
1528 goto backtrace;
1529
1530 if (current_prefix_length >= cn->pos)
1531 current_prefix_length = mp;
1532 }
1533
1534 pn = (struct tnode *)n; /* Descend */
1535 chopped_off = 0;
1536 continue;
1537
1538 backtrace:
1539 chopped_off++;
1540
1541 /* As zero don't change the child key (cindex) */
1542 while ((chopped_off <= pn->bits)
1543 && !(cindex & (1<<(chopped_off-1))))
1544 chopped_off++;
1545
1546 /* Decrease current_... with bits chopped off */
1547 if (current_prefix_length > pn->pos + pn->bits - chopped_off)
1548 current_prefix_length = pn->pos + pn->bits
1549 - chopped_off;
1550
1551 /*
1552 * Either we do the actual chop off according or if we have
1553 * chopped off all bits in this tnode walk up to our parent.
1554 */
1555
1556 if (chopped_off <= pn->bits) {
1557 cindex &= ~(1 << (chopped_off-1));
1558 } else {
1559 struct tnode *parent = node_parent_rcu((struct node *) pn);
1560 if (!parent)
1561 goto failed;
1562
1563 /* Get Child's index */
1564 cindex = tkey_extract_bits(pn->key, parent->pos, parent->bits);
1565 pn = parent;
1566 chopped_off = 0;
1567
1568 #ifdef CONFIG_IP_FIB_TRIE_STATS
1569 t->stats.backtrack++;
1570 #endif
1571 goto backtrace;
1572 }
1573 }
1574 failed:
1575 ret = 1;
1576 found:
1577 rcu_read_unlock();
1578 return ret;
1579 }
1580
1581 /*
1582 * Remove the leaf and return parent.
1583 */
1584 static void trie_leaf_remove(struct trie *t, struct leaf *l)
1585 {
1586 struct tnode *tp = node_parent((struct node *) l);
1587
1588 pr_debug("entering trie_leaf_remove(%p)\n", l);
1589
1590 if (tp) {
1591 t_key cindex = tkey_extract_bits(l->key, tp->pos, tp->bits);
1592 put_child(t, (struct tnode *)tp, cindex, NULL);
1593 trie_rebalance(t, tp);
1594 } else
1595 rcu_assign_pointer(t->trie, NULL);
1596
1597 free_leaf(l);
1598 }
1599
1600 /*
1601 * Caller must hold RTNL.
1602 */
1603 int fib_table_delete(struct fib_table *tb, struct fib_config *cfg)
1604 {
1605 struct trie *t = (struct trie *) tb->tb_data;
1606 u32 key, mask;
1607 int plen = cfg->fc_dst_len;
1608 u8 tos = cfg->fc_tos;
1609 struct fib_alias *fa, *fa_to_delete;
1610 struct list_head *fa_head;
1611 struct leaf *l;
1612 struct leaf_info *li;
1613
1614 if (plen > 32)
1615 return -EINVAL;
1616
1617 key = ntohl(cfg->fc_dst);
1618 mask = ntohl(inet_make_mask(plen));
1619
1620 if (key & ~mask)
1621 return -EINVAL;
1622
1623 key = key & mask;
1624 l = fib_find_node(t, key);
1625
1626 if (!l)
1627 return -ESRCH;
1628
1629 fa_head = get_fa_head(l, plen);
1630 fa = fib_find_alias(fa_head, tos, 0);
1631
1632 if (!fa)
1633 return -ESRCH;
1634
1635 pr_debug("Deleting %08x/%d tos=%d t=%p\n", key, plen, tos, t);
1636
1637 fa_to_delete = NULL;
1638 fa = list_entry(fa->fa_list.prev, struct fib_alias, fa_list);
1639 list_for_each_entry_continue(fa, fa_head, fa_list) {
1640 struct fib_info *fi = fa->fa_info;
1641
1642 if (fa->fa_tos != tos)
1643 break;
1644
1645 if ((!cfg->fc_type || fa->fa_type == cfg->fc_type) &&
1646 (cfg->fc_scope == RT_SCOPE_NOWHERE ||
1647 fa->fa_scope == cfg->fc_scope) &&
1648 (!cfg->fc_protocol ||
1649 fi->fib_protocol == cfg->fc_protocol) &&
1650 fib_nh_match(cfg, fi) == 0) {
1651 fa_to_delete = fa;
1652 break;
1653 }
1654 }
1655
1656 if (!fa_to_delete)
1657 return -ESRCH;
1658
1659 fa = fa_to_delete;
1660 rtmsg_fib(RTM_DELROUTE, htonl(key), fa, plen, tb->tb_id,
1661 &cfg->fc_nlinfo, 0);
1662
1663 l = fib_find_node(t, key);
1664 li = find_leaf_info(l, plen);
1665
1666 list_del_rcu(&fa->fa_list);
1667
1668 if (list_empty(fa_head)) {
1669 hlist_del_rcu(&li->hlist);
1670 free_leaf_info(li);
1671 }
1672
1673 if (hlist_empty(&l->list))
1674 trie_leaf_remove(t, l);
1675
1676 if (fa->fa_state & FA_S_ACCESSED)
1677 rt_cache_flush(cfg->fc_nlinfo.nl_net, -1);
1678
1679 fib_release_info(fa->fa_info);
1680 alias_free_mem_rcu(fa);
1681 return 0;
1682 }
1683
1684 static int trie_flush_list(struct list_head *head)
1685 {
1686 struct fib_alias *fa, *fa_node;
1687 int found = 0;
1688
1689 list_for_each_entry_safe(fa, fa_node, head, fa_list) {
1690 struct fib_info *fi = fa->fa_info;
1691
1692 if (fi && (fi->fib_flags & RTNH_F_DEAD)) {
1693 list_del_rcu(&fa->fa_list);
1694 fib_release_info(fa->fa_info);
1695 alias_free_mem_rcu(fa);
1696 found++;
1697 }
1698 }
1699 return found;
1700 }
1701
1702 static int trie_flush_leaf(struct leaf *l)
1703 {
1704 int found = 0;
1705 struct hlist_head *lih = &l->list;
1706 struct hlist_node *node, *tmp;
1707 struct leaf_info *li = NULL;
1708
1709 hlist_for_each_entry_safe(li, node, tmp, lih, hlist) {
1710 found += trie_flush_list(&li->falh);
1711
1712 if (list_empty(&li->falh)) {
1713 hlist_del_rcu(&li->hlist);
1714 free_leaf_info(li);
1715 }
1716 }
1717 return found;
1718 }
1719
1720 /*
1721 * Scan for the next right leaf starting at node p->child[idx]
1722 * Since we have back pointer, no recursion necessary.
1723 */
1724 static struct leaf *leaf_walk_rcu(struct tnode *p, struct node *c)
1725 {
1726 do {
1727 t_key idx;
1728
1729 if (c)
1730 idx = tkey_extract_bits(c->key, p->pos, p->bits) + 1;
1731 else
1732 idx = 0;
1733
1734 while (idx < 1u << p->bits) {
1735 c = tnode_get_child_rcu(p, idx++);
1736 if (!c)
1737 continue;
1738
1739 if (IS_LEAF(c)) {
1740 prefetch(p->child[idx]);
1741 return (struct leaf *) c;
1742 }
1743
1744 /* Rescan start scanning in new node */
1745 p = (struct tnode *) c;
1746 idx = 0;
1747 }
1748
1749 /* Node empty, walk back up to parent */
1750 c = (struct node *) p;
1751 } while ( (p = node_parent_rcu(c)) != NULL);
1752
1753 return NULL; /* Root of trie */
1754 }
1755
1756 static struct leaf *trie_firstleaf(struct trie *t)
1757 {
1758 struct tnode *n = (struct tnode *) rcu_dereference(t->trie);
1759
1760 if (!n)
1761 return NULL;
1762
1763 if (IS_LEAF(n)) /* trie is just a leaf */
1764 return (struct leaf *) n;
1765
1766 return leaf_walk_rcu(n, NULL);
1767 }
1768
1769 static struct leaf *trie_nextleaf(struct leaf *l)
1770 {
1771 struct node *c = (struct node *) l;
1772 struct tnode *p = node_parent_rcu(c);
1773
1774 if (!p)
1775 return NULL; /* trie with just one leaf */
1776
1777 return leaf_walk_rcu(p, c);
1778 }
1779
1780 static struct leaf *trie_leafindex(struct trie *t, int index)
1781 {
1782 struct leaf *l = trie_firstleaf(t);
1783
1784 while (l && index-- > 0)
1785 l = trie_nextleaf(l);
1786
1787 return l;
1788 }
1789
1790
1791 /*
1792 * Caller must hold RTNL.
1793 */
1794 int fib_table_flush(struct fib_table *tb)
1795 {
1796 struct trie *t = (struct trie *) tb->tb_data;
1797 struct leaf *l, *ll = NULL;
1798 int found = 0;
1799
1800 for (l = trie_firstleaf(t); l; l = trie_nextleaf(l)) {
1801 found += trie_flush_leaf(l);
1802
1803 if (ll && hlist_empty(&ll->list))
1804 trie_leaf_remove(t, ll);
1805 ll = l;
1806 }
1807
1808 if (ll && hlist_empty(&ll->list))
1809 trie_leaf_remove(t, ll);
1810
1811 pr_debug("trie_flush found=%d\n", found);
1812 return found;
1813 }
1814
1815 void fib_table_select_default(struct fib_table *tb,
1816 const struct flowi *flp,
1817 struct fib_result *res)
1818 {
1819 struct trie *t = (struct trie *) tb->tb_data;
1820 int order, last_idx;
1821 struct fib_info *fi = NULL;
1822 struct fib_info *last_resort;
1823 struct fib_alias *fa = NULL;
1824 struct list_head *fa_head;
1825 struct leaf *l;
1826
1827 last_idx = -1;
1828 last_resort = NULL;
1829 order = -1;
1830
1831 rcu_read_lock();
1832
1833 l = fib_find_node(t, 0);
1834 if (!l)
1835 goto out;
1836
1837 fa_head = get_fa_head(l, 0);
1838 if (!fa_head)
1839 goto out;
1840
1841 if (list_empty(fa_head))
1842 goto out;
1843
1844 list_for_each_entry_rcu(fa, fa_head, fa_list) {
1845 struct fib_info *next_fi = fa->fa_info;
1846
1847 if (fa->fa_scope != res->scope ||
1848 fa->fa_type != RTN_UNICAST)
1849 continue;
1850
1851 if (next_fi->fib_priority > res->fi->fib_priority)
1852 break;
1853 if (!next_fi->fib_nh[0].nh_gw ||
1854 next_fi->fib_nh[0].nh_scope != RT_SCOPE_LINK)
1855 continue;
1856 fa->fa_state |= FA_S_ACCESSED;
1857
1858 if (fi == NULL) {
1859 if (next_fi != res->fi)
1860 break;
1861 } else if (!fib_detect_death(fi, order, &last_resort,
1862 &last_idx, tb->tb_default)) {
1863 fib_result_assign(res, fi);
1864 tb->tb_default = order;
1865 goto out;
1866 }
1867 fi = next_fi;
1868 order++;
1869 }
1870 if (order <= 0 || fi == NULL) {
1871 tb->tb_default = -1;
1872 goto out;
1873 }
1874
1875 if (!fib_detect_death(fi, order, &last_resort, &last_idx,
1876 tb->tb_default)) {
1877 fib_result_assign(res, fi);
1878 tb->tb_default = order;
1879 goto out;
1880 }
1881 if (last_idx >= 0)
1882 fib_result_assign(res, last_resort);
1883 tb->tb_default = last_idx;
1884 out:
1885 rcu_read_unlock();
1886 }
1887
1888 static int fn_trie_dump_fa(t_key key, int plen, struct list_head *fah,
1889 struct fib_table *tb,
1890 struct sk_buff *skb, struct netlink_callback *cb)
1891 {
1892 int i, s_i;
1893 struct fib_alias *fa;
1894 __be32 xkey = htonl(key);
1895
1896 s_i = cb->args[5];
1897 i = 0;
1898
1899 /* rcu_read_lock is hold by caller */
1900
1901 list_for_each_entry_rcu(fa, fah, fa_list) {
1902 if (i < s_i) {
1903 i++;
1904 continue;
1905 }
1906
1907 if (fib_dump_info(skb, NETLINK_CB(cb->skb).pid,
1908 cb->nlh->nlmsg_seq,
1909 RTM_NEWROUTE,
1910 tb->tb_id,
1911 fa->fa_type,
1912 fa->fa_scope,
1913 xkey,
1914 plen,
1915 fa->fa_tos,
1916 fa->fa_info, NLM_F_MULTI) < 0) {
1917 cb->args[5] = i;
1918 return -1;
1919 }
1920 i++;
1921 }
1922 cb->args[5] = i;
1923 return skb->len;
1924 }
1925
1926 static int fn_trie_dump_leaf(struct leaf *l, struct fib_table *tb,
1927 struct sk_buff *skb, struct netlink_callback *cb)
1928 {
1929 struct leaf_info *li;
1930 struct hlist_node *node;
1931 int i, s_i;
1932
1933 s_i = cb->args[4];
1934 i = 0;
1935
1936 /* rcu_read_lock is hold by caller */
1937 hlist_for_each_entry_rcu(li, node, &l->list, hlist) {
1938 if (i < s_i) {
1939 i++;
1940 continue;
1941 }
1942
1943 if (i > s_i)
1944 cb->args[5] = 0;
1945
1946 if (list_empty(&li->falh))
1947 continue;
1948
1949 if (fn_trie_dump_fa(l->key, li->plen, &li->falh, tb, skb, cb) < 0) {
1950 cb->args[4] = i;
1951 return -1;
1952 }
1953 i++;
1954 }
1955
1956 cb->args[4] = i;
1957 return skb->len;
1958 }
1959
1960 int fib_table_dump(struct fib_table *tb, struct sk_buff *skb,
1961 struct netlink_callback *cb)
1962 {
1963 struct leaf *l;
1964 struct trie *t = (struct trie *) tb->tb_data;
1965 t_key key = cb->args[2];
1966 int count = cb->args[3];
1967
1968 rcu_read_lock();
1969 /* Dump starting at last key.
1970 * Note: 0.0.0.0/0 (ie default) is first key.
1971 */
1972 if (count == 0)
1973 l = trie_firstleaf(t);
1974 else {
1975 /* Normally, continue from last key, but if that is missing
1976 * fallback to using slow rescan
1977 */
1978 l = fib_find_node(t, key);
1979 if (!l)
1980 l = trie_leafindex(t, count);
1981 }
1982
1983 while (l) {
1984 cb->args[2] = l->key;
1985 if (fn_trie_dump_leaf(l, tb, skb, cb) < 0) {
1986 cb->args[3] = count;
1987 rcu_read_unlock();
1988 return -1;
1989 }
1990
1991 ++count;
1992 l = trie_nextleaf(l);
1993 memset(&cb->args[4], 0,
1994 sizeof(cb->args) - 4*sizeof(cb->args[0]));
1995 }
1996 cb->args[3] = count;
1997 rcu_read_unlock();
1998
1999 return skb->len;
2000 }
2001
2002 void __init fib_hash_init(void)
2003 {
2004 fn_alias_kmem = kmem_cache_create("ip_fib_alias",
2005 sizeof(struct fib_alias),
2006 0, SLAB_PANIC, NULL);
2007
2008 trie_leaf_kmem = kmem_cache_create("ip_fib_trie",
2009 max(sizeof(struct leaf),
2010 sizeof(struct leaf_info)),
2011 0, SLAB_PANIC, NULL);
2012 }
2013
2014
2015 /* Fix more generic FIB names for init later */
2016 struct fib_table *fib_hash_table(u32 id)
2017 {
2018 struct fib_table *tb;
2019 struct trie *t;
2020
2021 tb = kmalloc(sizeof(struct fib_table) + sizeof(struct trie),
2022 GFP_KERNEL);
2023 if (tb == NULL)
2024 return NULL;
2025
2026 tb->tb_id = id;
2027 tb->tb_default = -1;
2028
2029 t = (struct trie *) tb->tb_data;
2030 memset(t, 0, sizeof(*t));
2031
2032 if (id == RT_TABLE_LOCAL)
2033 pr_info("IPv4 FIB: Using LC-trie version %s\n", VERSION);
2034
2035 return tb;
2036 }
2037
2038 #ifdef CONFIG_PROC_FS
2039 /* Depth first Trie walk iterator */
2040 struct fib_trie_iter {
2041 struct seq_net_private p;
2042 struct fib_table *tb;
2043 struct tnode *tnode;
2044 unsigned index;
2045 unsigned depth;
2046 };
2047
2048 static struct node *fib_trie_get_next(struct fib_trie_iter *iter)
2049 {
2050 struct tnode *tn = iter->tnode;
2051 unsigned cindex = iter->index;
2052 struct tnode *p;
2053
2054 /* A single entry routing table */
2055 if (!tn)
2056 return NULL;
2057
2058 pr_debug("get_next iter={node=%p index=%d depth=%d}\n",
2059 iter->tnode, iter->index, iter->depth);
2060 rescan:
2061 while (cindex < (1<<tn->bits)) {
2062 struct node *n = tnode_get_child_rcu(tn, cindex);
2063
2064 if (n) {
2065 if (IS_LEAF(n)) {
2066 iter->tnode = tn;
2067 iter->index = cindex + 1;
2068 } else {
2069 /* push down one level */
2070 iter->tnode = (struct tnode *) n;
2071 iter->index = 0;
2072 ++iter->depth;
2073 }
2074 return n;
2075 }
2076
2077 ++cindex;
2078 }
2079
2080 /* Current node exhausted, pop back up */
2081 p = node_parent_rcu((struct node *)tn);
2082 if (p) {
2083 cindex = tkey_extract_bits(tn->key, p->pos, p->bits)+1;
2084 tn = p;
2085 --iter->depth;
2086 goto rescan;
2087 }
2088
2089 /* got root? */
2090 return NULL;
2091 }
2092
2093 static struct node *fib_trie_get_first(struct fib_trie_iter *iter,
2094 struct trie *t)
2095 {
2096 struct node *n;
2097
2098 if (!t)
2099 return NULL;
2100
2101 n = rcu_dereference(t->trie);
2102 if (!n)
2103 return NULL;
2104
2105 if (IS_TNODE(n)) {
2106 iter->tnode = (struct tnode *) n;
2107 iter->index = 0;
2108 iter->depth = 1;
2109 } else {
2110 iter->tnode = NULL;
2111 iter->index = 0;
2112 iter->depth = 0;
2113 }
2114
2115 return n;
2116 }
2117
2118 static void trie_collect_stats(struct trie *t, struct trie_stat *s)
2119 {
2120 struct node *n;
2121 struct fib_trie_iter iter;
2122
2123 memset(s, 0, sizeof(*s));
2124
2125 rcu_read_lock();
2126 for (n = fib_trie_get_first(&iter, t); n; n = fib_trie_get_next(&iter)) {
2127 if (IS_LEAF(n)) {
2128 struct leaf *l = (struct leaf *)n;
2129 struct leaf_info *li;
2130 struct hlist_node *tmp;
2131
2132 s->leaves++;
2133 s->totdepth += iter.depth;
2134 if (iter.depth > s->maxdepth)
2135 s->maxdepth = iter.depth;
2136
2137 hlist_for_each_entry_rcu(li, tmp, &l->list, hlist)
2138 ++s->prefixes;
2139 } else {
2140 const struct tnode *tn = (const struct tnode *) n;
2141 int i;
2142
2143 s->tnodes++;
2144 if (tn->bits < MAX_STAT_DEPTH)
2145 s->nodesizes[tn->bits]++;
2146
2147 for (i = 0; i < (1<<tn->bits); i++)
2148 if (!tn->child[i])
2149 s->nullpointers++;
2150 }
2151 }
2152 rcu_read_unlock();
2153 }
2154
2155 /*
2156 * This outputs /proc/net/fib_triestats
2157 */
2158 static void trie_show_stats(struct seq_file *seq, struct trie_stat *stat)
2159 {
2160 unsigned i, max, pointers, bytes, avdepth;
2161
2162 if (stat->leaves)
2163 avdepth = stat->totdepth*100 / stat->leaves;
2164 else
2165 avdepth = 0;
2166
2167 seq_printf(seq, "\tAver depth: %u.%02d\n",
2168 avdepth / 100, avdepth % 100);
2169 seq_printf(seq, "\tMax depth: %u\n", stat->maxdepth);
2170
2171 seq_printf(seq, "\tLeaves: %u\n", stat->leaves);
2172 bytes = sizeof(struct leaf) * stat->leaves;
2173
2174 seq_printf(seq, "\tPrefixes: %u\n", stat->prefixes);
2175 bytes += sizeof(struct leaf_info) * stat->prefixes;
2176
2177 seq_printf(seq, "\tInternal nodes: %u\n\t", stat->tnodes);
2178 bytes += sizeof(struct tnode) * stat->tnodes;
2179
2180 max = MAX_STAT_DEPTH;
2181 while (max > 0 && stat->nodesizes[max-1] == 0)
2182 max--;
2183
2184 pointers = 0;
2185 for (i = 1; i <= max; i++)
2186 if (stat->nodesizes[i] != 0) {
2187 seq_printf(seq, " %u: %u", i, stat->nodesizes[i]);
2188 pointers += (1<<i) * stat->nodesizes[i];
2189 }
2190 seq_putc(seq, '\n');
2191 seq_printf(seq, "\tPointers: %u\n", pointers);
2192
2193 bytes += sizeof(struct node *) * pointers;
2194 seq_printf(seq, "Null ptrs: %u\n", stat->nullpointers);
2195 seq_printf(seq, "Total size: %u kB\n", (bytes + 1023) / 1024);
2196 }
2197
2198 #ifdef CONFIG_IP_FIB_TRIE_STATS
2199 static void trie_show_usage(struct seq_file *seq,
2200 const struct trie_use_stats *stats)
2201 {
2202 seq_printf(seq, "\nCounters:\n---------\n");
2203 seq_printf(seq, "gets = %u\n", stats->gets);
2204 seq_printf(seq, "backtracks = %u\n", stats->backtrack);
2205 seq_printf(seq, "semantic match passed = %u\n",
2206 stats->semantic_match_passed);
2207 seq_printf(seq, "semantic match miss = %u\n",
2208 stats->semantic_match_miss);
2209 seq_printf(seq, "null node hit= %u\n", stats->null_node_hit);
2210 seq_printf(seq, "skipped node resize = %u\n\n",
2211 stats->resize_node_skipped);
2212 }
2213 #endif /* CONFIG_IP_FIB_TRIE_STATS */
2214
2215 static void fib_table_print(struct seq_file *seq, struct fib_table *tb)
2216 {
2217 if (tb->tb_id == RT_TABLE_LOCAL)
2218 seq_puts(seq, "Local:\n");
2219 else if (tb->tb_id == RT_TABLE_MAIN)
2220 seq_puts(seq, "Main:\n");
2221 else
2222 seq_printf(seq, "Id %d:\n", tb->tb_id);
2223 }
2224
2225
2226 static int fib_triestat_seq_show(struct seq_file *seq, void *v)
2227 {
2228 struct net *net = (struct net *)seq->private;
2229 unsigned int h;
2230
2231 seq_printf(seq,
2232 "Basic info: size of leaf:"
2233 " %Zd bytes, size of tnode: %Zd bytes.\n",
2234 sizeof(struct leaf), sizeof(struct tnode));
2235
2236 for (h = 0; h < FIB_TABLE_HASHSZ; h++) {
2237 struct hlist_head *head = &net->ipv4.fib_table_hash[h];
2238 struct hlist_node *node;
2239 struct fib_table *tb;
2240
2241 hlist_for_each_entry_rcu(tb, node, head, tb_hlist) {
2242 struct trie *t = (struct trie *) tb->tb_data;
2243 struct trie_stat stat;
2244
2245 if (!t)
2246 continue;
2247
2248 fib_table_print(seq, tb);
2249
2250 trie_collect_stats(t, &stat);
2251 trie_show_stats(seq, &stat);
2252 #ifdef CONFIG_IP_FIB_TRIE_STATS
2253 trie_show_usage(seq, &t->stats);
2254 #endif
2255 }
2256 }
2257
2258 return 0;
2259 }
2260
2261 static int fib_triestat_seq_open(struct inode *inode, struct file *file)
2262 {
2263 return single_open_net(inode, file, fib_triestat_seq_show);
2264 }
2265
2266 static const struct file_operations fib_triestat_fops = {
2267 .owner = THIS_MODULE,
2268 .open = fib_triestat_seq_open,
2269 .read = seq_read,
2270 .llseek = seq_lseek,
2271 .release = single_release_net,
2272 };
2273
2274 static struct node *fib_trie_get_idx(struct seq_file *seq, loff_t pos)
2275 {
2276 struct fib_trie_iter *iter = seq->private;
2277 struct net *net = seq_file_net(seq);
2278 loff_t idx = 0;
2279 unsigned int h;
2280
2281 for (h = 0; h < FIB_TABLE_HASHSZ; h++) {
2282 struct hlist_head *head = &net->ipv4.fib_table_hash[h];
2283 struct hlist_node *node;
2284 struct fib_table *tb;
2285
2286 hlist_for_each_entry_rcu(tb, node, head, tb_hlist) {
2287 struct node *n;
2288
2289 for (n = fib_trie_get_first(iter,
2290 (struct trie *) tb->tb_data);
2291 n; n = fib_trie_get_next(iter))
2292 if (pos == idx++) {
2293 iter->tb = tb;
2294 return n;
2295 }
2296 }
2297 }
2298
2299 return NULL;
2300 }
2301
2302 static void *fib_trie_seq_start(struct seq_file *seq, loff_t *pos)
2303 __acquires(RCU)
2304 {
2305 rcu_read_lock();
2306 return fib_trie_get_idx(seq, *pos);
2307 }
2308
2309 static void *fib_trie_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2310 {
2311 struct fib_trie_iter *iter = seq->private;
2312 struct net *net = seq_file_net(seq);
2313 struct fib_table *tb = iter->tb;
2314 struct hlist_node *tb_node;
2315 unsigned int h;
2316 struct node *n;
2317
2318 ++*pos;
2319 /* next node in same table */
2320 n = fib_trie_get_next(iter);
2321 if (n)
2322 return n;
2323
2324 /* walk rest of this hash chain */
2325 h = tb->tb_id & (FIB_TABLE_HASHSZ - 1);
2326 while ( (tb_node = rcu_dereference(tb->tb_hlist.next)) ) {
2327 tb = hlist_entry(tb_node, struct fib_table, tb_hlist);
2328 n = fib_trie_get_first(iter, (struct trie *) tb->tb_data);
2329 if (n)
2330 goto found;
2331 }
2332
2333 /* new hash chain */
2334 while (++h < FIB_TABLE_HASHSZ) {
2335 struct hlist_head *head = &net->ipv4.fib_table_hash[h];
2336 hlist_for_each_entry_rcu(tb, tb_node, head, tb_hlist) {
2337 n = fib_trie_get_first(iter, (struct trie *) tb->tb_data);
2338 if (n)
2339 goto found;
2340 }
2341 }
2342 return NULL;
2343
2344 found:
2345 iter->tb = tb;
2346 return n;
2347 }
2348
2349 static void fib_trie_seq_stop(struct seq_file *seq, void *v)
2350 __releases(RCU)
2351 {
2352 rcu_read_unlock();
2353 }
2354
2355 static void seq_indent(struct seq_file *seq, int n)
2356 {
2357 while (n-- > 0) seq_puts(seq, " ");
2358 }
2359
2360 static inline const char *rtn_scope(char *buf, size_t len, enum rt_scope_t s)
2361 {
2362 switch (s) {
2363 case RT_SCOPE_UNIVERSE: return "universe";
2364 case RT_SCOPE_SITE: return "site";
2365 case RT_SCOPE_LINK: return "link";
2366 case RT_SCOPE_HOST: return "host";
2367 case RT_SCOPE_NOWHERE: return "nowhere";
2368 default:
2369 snprintf(buf, len, "scope=%d", s);
2370 return buf;
2371 }
2372 }
2373
2374 static const char *const rtn_type_names[__RTN_MAX] = {
2375 [RTN_UNSPEC] = "UNSPEC",
2376 [RTN_UNICAST] = "UNICAST",
2377 [RTN_LOCAL] = "LOCAL",
2378 [RTN_BROADCAST] = "BROADCAST",
2379 [RTN_ANYCAST] = "ANYCAST",
2380 [RTN_MULTICAST] = "MULTICAST",
2381 [RTN_BLACKHOLE] = "BLACKHOLE",
2382 [RTN_UNREACHABLE] = "UNREACHABLE",
2383 [RTN_PROHIBIT] = "PROHIBIT",
2384 [RTN_THROW] = "THROW",
2385 [RTN_NAT] = "NAT",
2386 [RTN_XRESOLVE] = "XRESOLVE",
2387 };
2388
2389 static inline const char *rtn_type(char *buf, size_t len, unsigned t)
2390 {
2391 if (t < __RTN_MAX && rtn_type_names[t])
2392 return rtn_type_names[t];
2393 snprintf(buf, len, "type %u", t);
2394 return buf;
2395 }
2396
2397 /* Pretty print the trie */
2398 static int fib_trie_seq_show(struct seq_file *seq, void *v)
2399 {
2400 const struct fib_trie_iter *iter = seq->private;
2401 struct node *n = v;
2402
2403 if (!node_parent_rcu(n))
2404 fib_table_print(seq, iter->tb);
2405
2406 if (IS_TNODE(n)) {
2407 struct tnode *tn = (struct tnode *) n;
2408 __be32 prf = htonl(mask_pfx(tn->key, tn->pos));
2409
2410 seq_indent(seq, iter->depth-1);
2411 seq_printf(seq, " +-- %pI4/%d %d %d %d\n",
2412 &prf, tn->pos, tn->bits, tn->full_children,
2413 tn->empty_children);
2414
2415 } else {
2416 struct leaf *l = (struct leaf *) n;
2417 struct leaf_info *li;
2418 struct hlist_node *node;
2419 __be32 val = htonl(l->key);
2420
2421 seq_indent(seq, iter->depth);
2422 seq_printf(seq, " |-- %pI4\n", &val);
2423
2424 hlist_for_each_entry_rcu(li, node, &l->list, hlist) {
2425 struct fib_alias *fa;
2426
2427 list_for_each_entry_rcu(fa, &li->falh, fa_list) {
2428 char buf1[32], buf2[32];
2429
2430 seq_indent(seq, iter->depth+1);
2431 seq_printf(seq, " /%d %s %s", li->plen,
2432 rtn_scope(buf1, sizeof(buf1),
2433 fa->fa_scope),
2434 rtn_type(buf2, sizeof(buf2),
2435 fa->fa_type));
2436 if (fa->fa_tos)
2437 seq_printf(seq, " tos=%d", fa->fa_tos);
2438 seq_putc(seq, '\n');
2439 }
2440 }
2441 }
2442
2443 return 0;
2444 }
2445
2446 static const struct seq_operations fib_trie_seq_ops = {
2447 .start = fib_trie_seq_start,
2448 .next = fib_trie_seq_next,
2449 .stop = fib_trie_seq_stop,
2450 .show = fib_trie_seq_show,
2451 };
2452
2453 static int fib_trie_seq_open(struct inode *inode, struct file *file)
2454 {
2455 return seq_open_net(inode, file, &fib_trie_seq_ops,
2456 sizeof(struct fib_trie_iter));
2457 }
2458
2459 static const struct file_operations fib_trie_fops = {
2460 .owner = THIS_MODULE,
2461 .open = fib_trie_seq_open,
2462 .read = seq_read,
2463 .llseek = seq_lseek,
2464 .release = seq_release_net,
2465 };
2466
2467 struct fib_route_iter {
2468 struct seq_net_private p;
2469 struct trie *main_trie;
2470 loff_t pos;
2471 t_key key;
2472 };
2473
2474 static struct leaf *fib_route_get_idx(struct fib_route_iter *iter, loff_t pos)
2475 {
2476 struct leaf *l = NULL;
2477 struct trie *t = iter->main_trie;
2478
2479 /* use cache location of last found key */
2480 if (iter->pos > 0 && pos >= iter->pos && (l = fib_find_node(t, iter->key)))
2481 pos -= iter->pos;
2482 else {
2483 iter->pos = 0;
2484 l = trie_firstleaf(t);
2485 }
2486
2487 while (l && pos-- > 0) {
2488 iter->pos++;
2489 l = trie_nextleaf(l);
2490 }
2491
2492 if (l)
2493 iter->key = pos; /* remember it */
2494 else
2495 iter->pos = 0; /* forget it */
2496
2497 return l;
2498 }
2499
2500 static void *fib_route_seq_start(struct seq_file *seq, loff_t *pos)
2501 __acquires(RCU)
2502 {
2503 struct fib_route_iter *iter = seq->private;
2504 struct fib_table *tb;
2505
2506 rcu_read_lock();
2507 tb = fib_get_table(seq_file_net(seq), RT_TABLE_MAIN);
2508 if (!tb)
2509 return NULL;
2510
2511 iter->main_trie = (struct trie *) tb->tb_data;
2512 if (*pos == 0)
2513 return SEQ_START_TOKEN;
2514 else
2515 return fib_route_get_idx(iter, *pos - 1);
2516 }
2517
2518 static void *fib_route_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2519 {
2520 struct fib_route_iter *iter = seq->private;
2521 struct leaf *l = v;
2522
2523 ++*pos;
2524 if (v == SEQ_START_TOKEN) {
2525 iter->pos = 0;
2526 l = trie_firstleaf(iter->main_trie);
2527 } else {
2528 iter->pos++;
2529 l = trie_nextleaf(l);
2530 }
2531
2532 if (l)
2533 iter->key = l->key;
2534 else
2535 iter->pos = 0;
2536 return l;
2537 }
2538
2539 static void fib_route_seq_stop(struct seq_file *seq, void *v)
2540 __releases(RCU)
2541 {
2542 rcu_read_unlock();
2543 }
2544
2545 static unsigned fib_flag_trans(int type, __be32 mask, const struct fib_info *fi)
2546 {
2547 static unsigned type2flags[RTN_MAX + 1] = {
2548 [7] = RTF_REJECT, [8] = RTF_REJECT,
2549 };
2550 unsigned flags = type2flags[type];
2551
2552 if (fi && fi->fib_nh->nh_gw)
2553 flags |= RTF_GATEWAY;
2554 if (mask == htonl(0xFFFFFFFF))
2555 flags |= RTF_HOST;
2556 flags |= RTF_UP;
2557 return flags;
2558 }
2559
2560 /*
2561 * This outputs /proc/net/route.
2562 * The format of the file is not supposed to be changed
2563 * and needs to be same as fib_hash output to avoid breaking
2564 * legacy utilities
2565 */
2566 static int fib_route_seq_show(struct seq_file *seq, void *v)
2567 {
2568 struct leaf *l = v;
2569 struct leaf_info *li;
2570 struct hlist_node *node;
2571
2572 if (v == SEQ_START_TOKEN) {
2573 seq_printf(seq, "%-127s\n", "Iface\tDestination\tGateway "
2574 "\tFlags\tRefCnt\tUse\tMetric\tMask\t\tMTU"
2575 "\tWindow\tIRTT");
2576 return 0;
2577 }
2578
2579 hlist_for_each_entry_rcu(li, node, &l->list, hlist) {
2580 struct fib_alias *fa;
2581 __be32 mask, prefix;
2582
2583 mask = inet_make_mask(li->plen);
2584 prefix = htonl(l->key);
2585
2586 list_for_each_entry_rcu(fa, &li->falh, fa_list) {
2587 const struct fib_info *fi = fa->fa_info;
2588 unsigned flags = fib_flag_trans(fa->fa_type, mask, fi);
2589 int len;
2590
2591 if (fa->fa_type == RTN_BROADCAST
2592 || fa->fa_type == RTN_MULTICAST)
2593 continue;
2594
2595 if (fi)
2596 seq_printf(seq,
2597 "%s\t%08X\t%08X\t%04X\t%d\t%u\t"
2598 "%d\t%08X\t%d\t%u\t%u%n",
2599 fi->fib_dev ? fi->fib_dev->name : "*",
2600 prefix,
2601 fi->fib_nh->nh_gw, flags, 0, 0,
2602 fi->fib_priority,
2603 mask,
2604 (fi->fib_advmss ?
2605 fi->fib_advmss + 40 : 0),
2606 fi->fib_window,
2607 fi->fib_rtt >> 3, &len);
2608 else
2609 seq_printf(seq,
2610 "*\t%08X\t%08X\t%04X\t%d\t%u\t"
2611 "%d\t%08X\t%d\t%u\t%u%n",
2612 prefix, 0, flags, 0, 0, 0,
2613 mask, 0, 0, 0, &len);
2614
2615 seq_printf(seq, "%*s\n", 127 - len, "");
2616 }
2617 }
2618
2619 return 0;
2620 }
2621
2622 static const struct seq_operations fib_route_seq_ops = {
2623 .start = fib_route_seq_start,
2624 .next = fib_route_seq_next,
2625 .stop = fib_route_seq_stop,
2626 .show = fib_route_seq_show,
2627 };
2628
2629 static int fib_route_seq_open(struct inode *inode, struct file *file)
2630 {
2631 return seq_open_net(inode, file, &fib_route_seq_ops,
2632 sizeof(struct fib_route_iter));
2633 }
2634
2635 static const struct file_operations fib_route_fops = {
2636 .owner = THIS_MODULE,
2637 .open = fib_route_seq_open,
2638 .read = seq_read,
2639 .llseek = seq_lseek,
2640 .release = seq_release_net,
2641 };
2642
2643 int __net_init fib_proc_init(struct net *net)
2644 {
2645 if (!proc_net_fops_create(net, "fib_trie", S_IRUGO, &fib_trie_fops))
2646 goto out1;
2647
2648 if (!proc_net_fops_create(net, "fib_triestat", S_IRUGO,
2649 &fib_triestat_fops))
2650 goto out2;
2651
2652 if (!proc_net_fops_create(net, "route", S_IRUGO, &fib_route_fops))
2653 goto out3;
2654
2655 return 0;
2656
2657 out3:
2658 proc_net_remove(net, "fib_triestat");
2659 out2:
2660 proc_net_remove(net, "fib_trie");
2661 out1:
2662 return -ENOMEM;
2663 }
2664
2665 void __net_exit fib_proc_exit(struct net *net)
2666 {
2667 proc_net_remove(net, "fib_trie");
2668 proc_net_remove(net, "fib_triestat");
2669 proc_net_remove(net, "route");
2670 }
2671
2672 #endif /* CONFIG_PROC_FS */
This page took 0.157681 seconds and 5 git commands to generate.