fib_trie.c 59.2 KB
Newer Older
1 2 3 4 5 6 7 8 9
/*
 *   This program is free software; you can redistribute it and/or
 *   modify it under the terms of the GNU General Public License
 *   as published by the Free Software Foundation; either version
 *   2 of the License, or (at your option) any later version.
 *
 *   Robert Olsson <robert.olsson@its.uu.se> Uppsala Universitet
 *     & Swedish University of Agricultural Sciences.
 *
10
 *   Jens Laas <jens.laas@data.slu.se> Swedish University of
11
 *     Agricultural Sciences.
12
 *
13 14 15
 *   Hans Liss <hans.liss@its.uu.se>  Uppsala Universitet
 *
 * This work is based on the LPC-trie which is originally descibed in:
16
 *
17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43
 * An experimental study of compression methods for dynamic tries
 * Stefan Nilsson and Matti Tikkanen. Algorithmica, 33(1):19-33, 2002.
 * http://www.nada.kth.se/~snilsson/public/papers/dyntrie2/
 *
 *
 * IP-address lookup using LC-tries. Stefan Nilsson and Gunnar Karlsson
 * IEEE Journal on Selected Areas in Communications, 17(6):1083-1092, June 1999
 *
 * Version:	$Id: fib_trie.c,v 1.3 2005/06/08 14:20:01 robert Exp $
 *
 *
 * Code from fib_hash has been reused which includes the following header:
 *
 *
 * INET		An implementation of the TCP/IP protocol suite for the LINUX
 *		operating system.  INET is implemented using the  BSD Socket
 *		interface as the means of communication with the user level.
 *
 *		IPv4 FIB: lookup engine and maintenance routines.
 *
 *
 * Authors:	Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
 *
 *		This program is free software; you can redistribute it and/or
 *		modify it under the terms of the GNU General Public License
 *		as published by the Free Software Foundation; either version
 *		2 of the License, or (at your option) any later version.
R
Robert Olsson 已提交
44 45 46 47 48 49 50
 *
 * Substantial contributions to this work comes from:
 *
 *		David S. Miller, <davem@davemloft.net>
 *		Stephen Hemminger <shemminger@osdl.org>
 *		Paul E. McKenney <paulmck@us.ibm.com>
 *		Patrick McHardy <kaber@trash.net>
51 52
 */

R
Robert Olsson 已提交
53
#define VERSION "0.408"
54 55 56

#include <asm/uaccess.h>
#include <asm/system.h>
J
Jiri Slaby 已提交
57
#include <linux/bitops.h>
58 59 60 61 62 63 64 65 66
#include <linux/types.h>
#include <linux/kernel.h>
#include <linux/mm.h>
#include <linux/string.h>
#include <linux/socket.h>
#include <linux/sockios.h>
#include <linux/errno.h>
#include <linux/in.h>
#include <linux/inet.h>
S
Stephen Hemminger 已提交
67
#include <linux/inetdevice.h>
68 69 70
#include <linux/netdevice.h>
#include <linux/if_arp.h>
#include <linux/proc_fs.h>
R
Robert Olsson 已提交
71
#include <linux/rcupdate.h>
72 73 74 75
#include <linux/skbuff.h>
#include <linux/netlink.h>
#include <linux/init.h>
#include <linux/list.h>
76
#include <net/net_namespace.h>
77 78 79 80 81 82 83 84
#include <net/ip.h>
#include <net/protocol.h>
#include <net/route.h>
#include <net/tcp.h>
#include <net/sock.h>
#include <net/ip_fib.h>
#include "fib_lookup.h"

R
Robert Olsson 已提交
85
#define MAX_STAT_DEPTH 32
86 87 88 89 90 91 92 93

#define KEYLENGTH (8*sizeof(t_key))

typedef unsigned int t_key;

#define T_TNODE 0
#define T_LEAF  1
#define NODE_TYPE_MASK	0x1UL
R
Robert Olsson 已提交
94 95
#define NODE_TYPE(node) ((node)->parent & NODE_TYPE_MASK)

O
Olof Johansson 已提交
96 97
#define IS_TNODE(n) (!(n->parent & T_LEAF))
#define IS_LEAF(n) (n->parent & T_LEAF)
98 99

struct node {
O
Olof Johansson 已提交
100
	unsigned long parent;
101
	t_key key;
102 103 104
};

struct leaf {
O
Olof Johansson 已提交
105
	unsigned long parent;
106
	t_key key;
107
	struct hlist_head list;
R
Robert Olsson 已提交
108
	struct rcu_head rcu;
109 110 111 112
};

struct leaf_info {
	struct hlist_node hlist;
R
Robert Olsson 已提交
113
	struct rcu_head rcu;
114 115 116 117 118
	int plen;
	struct list_head falh;
};

struct tnode {
O
Olof Johansson 已提交
119
	unsigned long parent;
120
	t_key key;
121 122
	unsigned char pos;		/* 2log(KEYLENGTH) bits needed */
	unsigned char bits;		/* 2log(KEYLENGTH) bits needed */
123 124
	unsigned int full_children;	/* KEYLENGTH bits needed */
	unsigned int empty_children;	/* KEYLENGTH bits needed */
R
Robert Olsson 已提交
125
	struct rcu_head rcu;
O
Olof Johansson 已提交
126
	struct node *child[0];
127 128 129 130 131 132 133 134 135
};

#ifdef CONFIG_IP_FIB_TRIE_STATS
struct trie_use_stats {
	unsigned int gets;
	unsigned int backtrack;
	unsigned int semantic_match_passed;
	unsigned int semantic_match_miss;
	unsigned int null_node_hit;
136
	unsigned int resize_node_skipped;
137 138 139 140 141 142 143 144 145
};
#endif

struct trie_stat {
	unsigned int totdepth;
	unsigned int maxdepth;
	unsigned int tnodes;
	unsigned int leaves;
	unsigned int nullpointers;
R
Robert Olsson 已提交
146
	unsigned int nodesizes[MAX_STAT_DEPTH];
147
};
148 149

struct trie {
O
Olof Johansson 已提交
150
	struct node *trie;
151 152 153
#ifdef CONFIG_IP_FIB_TRIE_STATS
	struct trie_use_stats stats;
#endif
O
Olof Johansson 已提交
154
	int size;
155 156 157 158 159
};

static void put_child(struct trie *t, struct tnode *tn, int i, struct node *n);
static void tnode_put_child_reorg(struct tnode *tn, int i, struct node *n, int wasfull);
static struct node *resize(struct trie *t, struct tnode *tn);
160 161
static struct tnode *inflate(struct trie *t, struct tnode *tn);
static struct tnode *halve(struct trie *t, struct tnode *tn);
162 163
static void tnode_free(struct tnode *tn);

164
static struct kmem_cache *fn_alias_kmem __read_mostly;
165

S
Stephen Hemminger 已提交
166 167 168 169 170 171 172 173 174 175 176 177 178
static inline struct tnode *node_parent(struct node *node)
{
	struct tnode *ret;

	ret = (struct tnode *)(node->parent & ~NODE_TYPE_MASK);
	return rcu_dereference(ret);
}

static inline void node_set_parent(struct node *node, struct tnode *ptr)
{
	rcu_assign_pointer(node->parent,
			   (unsigned long)ptr | NODE_TYPE(node));
}
R
Robert Olsson 已提交
179 180 181

/* rcu_read_lock needs to be hold by caller from readside */

182
static inline struct node *tnode_get_child(struct tnode *tn, int i)
183
{
O
Olof Johansson 已提交
184
	BUG_ON(i >= 1 << tn->bits);
185

R
Robert Olsson 已提交
186
	return rcu_dereference(tn->child[i]);
187 188
}

S
Stephen Hemmigner 已提交
189
static inline int tnode_child_length(const struct tnode *tn)
190
{
O
Olof Johansson 已提交
191
	return 1 << tn->bits;
192 193
}

S
Stephen Hemminger 已提交
194 195 196 197 198
static inline t_key mask_pfx(t_key k, unsigned short l)
{
	return (l == 0) ? 0 : k >> (KEYLENGTH-l) << (KEYLENGTH-l);
}

199 200
static inline t_key tkey_extract_bits(t_key a, int offset, int bits)
{
O
Olof Johansson 已提交
201
	if (offset < KEYLENGTH)
202
		return ((t_key)(a << offset)) >> (KEYLENGTH - bits);
O
Olof Johansson 已提交
203
	else
204 205 206 207 208
		return 0;
}

static inline int tkey_equals(t_key a, t_key b)
{
209
	return a == b;
210 211 212 213
}

static inline int tkey_sub_equals(t_key a, int offset, int bits, t_key b)
{
214 215
	if (bits == 0 || offset >= KEYLENGTH)
		return 1;
O
Olof Johansson 已提交
216 217
	bits = bits > KEYLENGTH ? KEYLENGTH : bits;
	return ((a ^ b) << offset) >> (KEYLENGTH - bits) == 0;
218
}
219 220 221 222 223 224

static inline int tkey_mismatch(t_key a, int offset, t_key b)
{
	t_key diff = a ^ b;
	int i = offset;

225 226 227
	if (!diff)
		return 0;
	while ((diff << i) >> (KEYLENGTH-1) == 0)
228 229 230 231 232
		i++;
	return i;
}

/*
233 234
  To understand this stuff, an understanding of keys and all their bits is
  necessary. Every node in the trie has a key associated with it, but not
235 236 237 238
  all of the bits in that key are significant.

  Consider a node 'n' and its parent 'tp'.

239 240 241 242 243
  If n is a leaf, every bit in its key is significant. Its presence is
  necessitated by path compression, since during a tree traversal (when
  searching for a leaf - unless we are doing an insertion) we will completely
  ignore all skipped bits we encounter. Thus we need to verify, at the end of
  a potentially successful search, that we have indeed been walking the
244 245
  correct key path.

246 247 248 249 250
  Note that we can never "miss" the correct key in the tree if present by
  following the wrong path. Path compression ensures that segments of the key
  that are the same for all keys with a given prefix are skipped, but the
  skipped part *is* identical for each node in the subtrie below the skipped
  bit! trie_insert() in this implementation takes care of that - note the
251 252
  call to tkey_sub_equals() in trie_insert().

253
  if n is an internal node - a 'tnode' here, the various parts of its key
254 255
  have many different meanings.

256
  Example:
257 258 259
  _________________________________________________________________
  | i | i | i | i | i | i | i | N | N | N | S | S | S | S | S | C |
  -----------------------------------------------------------------
260
    0   1   2   3   4   5   6   7   8   9  10  11  12  13  14  15
261 262 263 264 265 266 267 268 269

  _________________________________________________________________
  | C | C | C | u | u | u | u | u | u | u | u | u | u | u | u | u |
  -----------------------------------------------------------------
   16  17  18  19  20  21  22  23  24  25  26  27  28  29  30  31

  tp->pos = 7
  tp->bits = 3
  n->pos = 15
O
Olof Johansson 已提交
270
  n->bits = 4
271

272 273
  First, let's just ignore the bits that come before the parent tp, that is
  the bits from 0 to (tp->pos-1). They are *known* but at this point we do
274 275 276
  not use them for anything.

  The bits from (tp->pos) to (tp->pos + tp->bits - 1) - "N", above - are the
277
  index into the parent's child array. That is, they will be used to find
278 279 280 281 282
  'n' among tp's children.

  The bits from (tp->pos + tp->bits) to (n->pos - 1) - "S" - are skipped bits
  for the node n.

283
  All the bits we have seen so far are significant to the node n. The rest
284 285
  of the bits are really not needed or indeed known in n->key.

286
  The bits from (n->pos) to (n->pos + n->bits - 1) - "C" - are the index into
287
  n's child array, and will of course be different for each child.
288

289

290 291 292 293 294
  The rest of the bits, from (n->pos + n->bits) onward, are completely unknown
  at this point.

*/

S
Stephen Hemminger 已提交
295
static inline void check_tnode(const struct tnode *tn)
296
{
S
Stephen Hemminger 已提交
297
	WARN_ON(tn && tn->pos+tn->bits > 32);
298 299
}

300 301 302 303
static const int halve_threshold = 25;
static const int inflate_threshold = 50;
static const int halve_threshold_root = 8;
static const int inflate_threshold_root = 15;
304

R
Robert Olsson 已提交
305 306

static void __alias_free_mem(struct rcu_head *head)
307
{
R
Robert Olsson 已提交
308 309
	struct fib_alias *fa = container_of(head, struct fib_alias, rcu);
	kmem_cache_free(fn_alias_kmem, fa);
310 311
}

R
Robert Olsson 已提交
312
static inline void alias_free_mem_rcu(struct fib_alias *fa)
313
{
R
Robert Olsson 已提交
314 315
	call_rcu(&fa->rcu, __alias_free_mem);
}
O
Olof Johansson 已提交
316

R
Robert Olsson 已提交
317 318 319 320
static void __leaf_free_rcu(struct rcu_head *head)
{
	kfree(container_of(head, struct leaf, rcu));
}
O
Olof Johansson 已提交
321

R
Robert Olsson 已提交
322
static void __leaf_info_free_rcu(struct rcu_head *head)
323
{
R
Robert Olsson 已提交
324
	kfree(container_of(head, struct leaf_info, rcu));
325 326
}

R
Robert Olsson 已提交
327
static inline void free_leaf_info(struct leaf_info *leaf)
328
{
R
Robert Olsson 已提交
329
	call_rcu(&leaf->rcu, __leaf_info_free_rcu);
330 331
}

332
static struct tnode *tnode_alloc(size_t size)
333
{
R
Robert Olsson 已提交
334 335 336
	struct page *pages;

	if (size <= PAGE_SIZE)
337
		return kzalloc(size, GFP_KERNEL);
R
Robert Olsson 已提交
338 339 340 341 342 343

	pages = alloc_pages(GFP_KERNEL|__GFP_ZERO, get_order(size));
	if (!pages)
		return NULL;

	return page_address(pages);
344 345
}

R
Robert Olsson 已提交
346
static void __tnode_free_rcu(struct rcu_head *head)
347
{
R
Robert Olsson 已提交
348
	struct tnode *tn = container_of(head, struct tnode, rcu);
349 350
	size_t size = sizeof(struct tnode) +
		      (sizeof(struct node *) << tn->bits);
351 352 353 354 355 356 357

	if (size <= PAGE_SIZE)
		kfree(tn);
	else
		free_pages((unsigned long)tn, get_order(size));
}

R
Robert Olsson 已提交
358 359
static inline void tnode_free(struct tnode *tn)
{
S
Stephen Hemminger 已提交
360
	if (IS_LEAF(tn)) {
R
Robert Olsson 已提交
361 362
		struct leaf *l = (struct leaf *) tn;
		call_rcu_bh(&l->rcu, __leaf_free_rcu);
S
Stephen Hemminger 已提交
363
	} else
R
Robert Olsson 已提交
364
		call_rcu(&tn->rcu, __tnode_free_rcu);
R
Robert Olsson 已提交
365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386
}

static struct leaf *leaf_new(void)
{
	struct leaf *l = kmalloc(sizeof(struct leaf),  GFP_KERNEL);
	if (l) {
		l->parent = T_LEAF;
		INIT_HLIST_HEAD(&l->list);
	}
	return l;
}

static struct leaf_info *leaf_info_new(int plen)
{
	struct leaf_info *li = kmalloc(sizeof(struct leaf_info),  GFP_KERNEL);
	if (li) {
		li->plen = plen;
		INIT_LIST_HEAD(&li->falh);
	}
	return li;
}

387 388
static struct tnode* tnode_new(t_key key, int pos, int bits)
{
389
	size_t sz = sizeof(struct tnode) + (sizeof(struct node *) << bits);
390
	struct tnode *tn = tnode_alloc(sz);
391

O
Olof Johansson 已提交
392
	if (tn) {
R
Robert Olsson 已提交
393
		tn->parent = T_TNODE;
394 395 396 397 398 399
		tn->pos = pos;
		tn->bits = bits;
		tn->key = key;
		tn->full_children = 0;
		tn->empty_children = 1<<bits;
	}
400

401 402
	pr_debug("AT %p s=%u %lu\n", tn, (unsigned int) sizeof(struct tnode),
		 (unsigned long) (sizeof(struct node) << bits));
403 404 405 406 407 408 409 410
	return tn;
}

/*
 * Check whether a tnode 'n' is "full", i.e. it is an internal node
 * and no bits are skipped. See discussion in dyntree paper p. 6
 */

S
Stephen Hemmigner 已提交
411
static inline int tnode_full(const struct tnode *tn, const struct node *n)
412
{
413
	if (n == NULL || IS_LEAF(n))
414 415 416 417 418
		return 0;

	return ((struct tnode *) n)->pos == tn->pos + tn->bits;
}

419
static inline void put_child(struct trie *t, struct tnode *tn, int i, struct node *n)
420 421 422 423
{
	tnode_put_child_reorg(tn, i, n, -1);
}

424
 /*
425 426 427 428
  * Add a child at position i overwriting the old value.
  * Update the value of full_children and empty_children.
  */

429
static void tnode_put_child_reorg(struct tnode *tn, int i, struct node *n, int wasfull)
430
{
R
Robert Olsson 已提交
431
	struct node *chi = tn->child[i];
432 433
	int isfull;

S
Stephen Hemminger 已提交
434 435
	BUG_ON(i >= 1<<tn->bits);

436 437 438 439 440 441

	/* update emptyChildren */
	if (n == NULL && chi != NULL)
		tn->empty_children++;
	else if (n != NULL && chi == NULL)
		tn->empty_children--;
442

443
	/* update fullChildren */
O
Olof Johansson 已提交
444
	if (wasfull == -1)
445 446 447
		wasfull = tnode_full(tn, chi);

	isfull = tnode_full(tn, n);
448
	if (wasfull && !isfull)
449
		tn->full_children--;
450
	else if (!wasfull && isfull)
451
		tn->full_children++;
O
Olof Johansson 已提交
452

453
	if (n)
S
Stephen Hemminger 已提交
454
		node_set_parent(n, tn);
455

R
Robert Olsson 已提交
456
	rcu_assign_pointer(tn->child[i], n);
457 458
}

459
static struct node *resize(struct trie *t, struct tnode *tn)
460 461
{
	int i;
462
	int err = 0;
463
	struct tnode *old_tn;
464 465
	int inflate_threshold_use;
	int halve_threshold_use;
R
Robert Olsson 已提交
466
	int max_resize;
467

468
	if (!tn)
469 470
		return NULL;

S
Stephen Hemminger 已提交
471 472
	pr_debug("In tnode_resize %p inflate_threshold=%d threshold=%d\n",
		 tn, inflate_threshold, halve_threshold);
473 474 475 476 477 478 479 480 481

	/* No children */
	if (tn->empty_children == tnode_child_length(tn)) {
		tnode_free(tn);
		return NULL;
	}
	/* One child */
	if (tn->empty_children == tnode_child_length(tn) - 1)
		for (i = 0; i < tnode_child_length(tn); i++) {
O
Olof Johansson 已提交
482
			struct node *n;
483

O
Olof Johansson 已提交
484
			n = tn->child[i];
R
Robert Olsson 已提交
485
			if (!n)
O
Olof Johansson 已提交
486 487 488
				continue;

			/* compress one level */
S
Stephen Hemminger 已提交
489
			node_set_parent(n, NULL);
O
Olof Johansson 已提交
490 491
			tnode_free(tn);
			return n;
492
		}
493
	/*
494 495 496 497 498
	 * Double as long as the resulting node has a number of
	 * nonempty nodes that are above the threshold.
	 */

	/*
499 500
	 * From "Implementing a dynamic compressed trie" by Stefan Nilsson of
	 * the Helsinki University of Technology and Matti Tikkanen of Nokia
501
	 * Telecommunications, page 6:
502
	 * "A node is doubled if the ratio of non-empty children to all
503 504
	 * children in the *doubled* node is at least 'high'."
	 *
505 506 507 508 509
	 * 'high' in this instance is the variable 'inflate_threshold'. It
	 * is expressed as a percentage, so we multiply it with
	 * tnode_child_length() and instead of multiplying by 2 (since the
	 * child array will be doubled by inflate()) and multiplying
	 * the left-hand side by 100 (to handle the percentage thing) we
510
	 * multiply the left-hand side by 50.
511 512 513 514
	 *
	 * The left-hand side may look a bit weird: tnode_child_length(tn)
	 * - tn->empty_children is of course the number of non-null children
	 * in the current node. tn->full_children is the number of "full"
515
	 * children, that is non-null tnodes with a skip value of 0.
516
	 * All of those will be doubled in the resulting inflated tnode, so
517
	 * we just count them one extra time here.
518
	 *
519
	 * A clearer way to write this would be:
520
	 *
521
	 * to_be_doubled = tn->full_children;
522
	 * not_to_be_doubled = tnode_child_length(tn) - tn->empty_children -
523 524 525 526
	 *     tn->full_children;
	 *
	 * new_child_length = tnode_child_length(tn) * 2;
	 *
527
	 * new_fill_factor = 100 * (not_to_be_doubled + 2*to_be_doubled) /
528 529
	 *      new_child_length;
	 * if (new_fill_factor >= inflate_threshold)
530 531 532
	 *
	 * ...and so on, tho it would mess up the while () loop.
	 *
533 534 535
	 * anyway,
	 * 100 * (not_to_be_doubled + 2*to_be_doubled) / new_child_length >=
	 *      inflate_threshold
536
	 *
537 538 539
	 * avoid a division:
	 * 100 * (not_to_be_doubled + 2*to_be_doubled) >=
	 *      inflate_threshold * new_child_length
540
	 *
541
	 * expand not_to_be_doubled and to_be_doubled, and shorten:
542
	 * 100 * (tnode_child_length(tn) - tn->empty_children +
O
Olof Johansson 已提交
543
	 *    tn->full_children) >= inflate_threshold * new_child_length
544
	 *
545
	 * expand new_child_length:
546
	 * 100 * (tnode_child_length(tn) - tn->empty_children +
O
Olof Johansson 已提交
547
	 *    tn->full_children) >=
548
	 *      inflate_threshold * tnode_child_length(tn) * 2
549
	 *
550
	 * shorten again:
551
	 * 50 * (tn->full_children + tnode_child_length(tn) -
O
Olof Johansson 已提交
552
	 *    tn->empty_children) >= inflate_threshold *
553
	 *    tnode_child_length(tn)
554
	 *
555 556 557
	 */

	check_tnode(tn);
558

559 560
	/* Keep root node larger  */

S
Stephen Hemminger 已提交
561
	if (!tn->parent)
562
		inflate_threshold_use = inflate_threshold_root;
563
	else
564 565
		inflate_threshold_use = inflate_threshold;

566
	err = 0;
R
Robert Olsson 已提交
567 568
	max_resize = 10;
	while ((tn->full_children > 0 &&  max_resize-- &&
569
	       50 * (tn->full_children + tnode_child_length(tn) - tn->empty_children) >=
570
				inflate_threshold_use * tnode_child_length(tn))) {
571

572 573 574 575
		old_tn = tn;
		tn = inflate(t, tn);
		if (IS_ERR(tn)) {
			tn = old_tn;
576 577 578 579 580
#ifdef CONFIG_IP_FIB_TRIE_STATS
			t->stats.resize_node_skipped++;
#endif
			break;
		}
581 582
	}

R
Robert Olsson 已提交
583 584 585 586 587 588 589 590 591
	if (max_resize < 0) {
		if (!tn->parent)
			printk(KERN_WARNING "Fix inflate_threshold_root. Now=%d size=%d bits\n",
			       inflate_threshold_root, tn->bits);
		else
			printk(KERN_WARNING "Fix inflate_threshold. Now=%d size=%d bits\n",
			       inflate_threshold, tn->bits);
	}

592 593 594 595 596 597
	check_tnode(tn);

	/*
	 * Halve as long as the number of empty children in this
	 * node is above threshold.
	 */
598

599 600 601

	/* Keep root node larger  */

S
Stephen Hemminger 已提交
602
	if (!tn->parent)
603
		halve_threshold_use = halve_threshold_root;
604
	else
605 606
		halve_threshold_use = halve_threshold;

607
	err = 0;
R
Robert Olsson 已提交
608 609
	max_resize = 10;
	while (tn->bits > 1 &&  max_resize-- &&
610
	       100 * (tnode_child_length(tn) - tn->empty_children) <
611
	       halve_threshold_use * tnode_child_length(tn)) {
612

613 614 615 616
		old_tn = tn;
		tn = halve(t, tn);
		if (IS_ERR(tn)) {
			tn = old_tn;
617 618 619 620 621 622
#ifdef CONFIG_IP_FIB_TRIE_STATS
			t->stats.resize_node_skipped++;
#endif
			break;
		}
	}
623

R
Robert Olsson 已提交
624 625 626 627 628 629 630 631
	if (max_resize < 0) {
		if (!tn->parent)
			printk(KERN_WARNING "Fix halve_threshold_root. Now=%d size=%d bits\n",
			       halve_threshold_root, tn->bits);
		else
			printk(KERN_WARNING "Fix halve_threshold. Now=%d size=%d bits\n",
			       halve_threshold, tn->bits);
	}
632

633 634 635
	/* Only one child remains */
	if (tn->empty_children == tnode_child_length(tn) - 1)
		for (i = 0; i < tnode_child_length(tn); i++) {
O
Olof Johansson 已提交
636
			struct node *n;
637

O
Olof Johansson 已提交
638
			n = tn->child[i];
R
Robert Olsson 已提交
639
			if (!n)
O
Olof Johansson 已提交
640 641 642 643
				continue;

			/* compress one level */

S
Stephen Hemminger 已提交
644
			node_set_parent(n, NULL);
O
Olof Johansson 已提交
645 646
			tnode_free(tn);
			return n;
647 648 649 650 651
		}

	return (struct node *) tn;
}

652
static struct tnode *inflate(struct trie *t, struct tnode *tn)
653 654 655 656 657
{
	struct tnode *oldtnode = tn;
	int olen = tnode_child_length(tn);
	int i;

S
Stephen Hemminger 已提交
658
	pr_debug("In inflate\n");
659 660 661

	tn = tnode_new(oldtnode->key, oldtnode->pos, oldtnode->bits + 1);

S
Stephen Hemminger 已提交
662
	if (!tn)
663
		return ERR_PTR(-ENOMEM);
664 665

	/*
666 667 668
	 * Preallocate and store tnodes before the actual work so we
	 * don't get into an inconsistent state if memory allocation
	 * fails. In case of failure we return the oldnode and  inflate
669 670
	 * of tnode is ignored.
	 */
O
Olof Johansson 已提交
671 672

	for (i = 0; i < olen; i++) {
673 674 675 676 677 678 679
		struct tnode *inode = (struct tnode *) tnode_get_child(oldtnode, i);

		if (inode &&
		    IS_TNODE(inode) &&
		    inode->pos == oldtnode->pos + oldtnode->bits &&
		    inode->bits > 1) {
			struct tnode *left, *right;
S
Stephen Hemminger 已提交
680
			t_key m = ~0U << (KEYLENGTH - 1) >> inode->pos;
681

682 683
			left = tnode_new(inode->key&(~m), inode->pos + 1,
					 inode->bits - 1);
684 685
			if (!left)
				goto nomem;
O
Olof Johansson 已提交
686

687 688 689
			right = tnode_new(inode->key|m, inode->pos + 1,
					  inode->bits - 1);

690
			if (!right) {
691 692
				tnode_free(left);
				goto nomem;
693
			}
694 695 696 697 698 699

			put_child(t, tn, 2*i, (struct node *) left);
			put_child(t, tn, 2*i+1, (struct node *) right);
		}
	}

O
Olof Johansson 已提交
700
	for (i = 0; i < olen; i++) {
701
		struct tnode *inode;
702
		struct node *node = tnode_get_child(oldtnode, i);
O
Olof Johansson 已提交
703 704
		struct tnode *left, *right;
		int size, j;
705

706 707 708 709 710 711
		/* An empty child */
		if (node == NULL)
			continue;

		/* A leaf or an internal node with skipped bits */

712
		if (IS_LEAF(node) || ((struct tnode *) node)->pos >
713
		   tn->pos + tn->bits - 1) {
714
			if (tkey_extract_bits(node->key, oldtnode->pos + oldtnode->bits,
715 716 717 718 719 720 721 722 723 724 725 726 727 728 729
					     1) == 0)
				put_child(t, tn, 2*i, node);
			else
				put_child(t, tn, 2*i+1, node);
			continue;
		}

		/* An internal node with two children */
		inode = (struct tnode *) node;

		if (inode->bits == 1) {
			put_child(t, tn, 2*i, inode->child[0]);
			put_child(t, tn, 2*i+1, inode->child[1]);

			tnode_free(inode);
O
Olof Johansson 已提交
730
			continue;
731 732
		}

O
Olof Johansson 已提交
733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750
		/* An internal node with more than two children */

		/* We will replace this node 'inode' with two new
		 * ones, 'left' and 'right', each with half of the
		 * original children. The two new nodes will have
		 * a position one bit further down the key and this
		 * means that the "significant" part of their keys
		 * (see the discussion near the top of this file)
		 * will differ by one bit, which will be "0" in
		 * left's key and "1" in right's key. Since we are
		 * moving the key position by one step, the bit that
		 * we are moving away from - the bit at position
		 * (inode->pos) - is the one that will differ between
		 * left and right. So... we synthesize that bit in the
		 * two  new keys.
		 * The mask 'm' below will be a single "one" bit at
		 * the position (inode->pos)
		 */
751

O
Olof Johansson 已提交
752 753 754
		/* Use the old key, but set the new significant
		 *   bit to zero.
		 */
755

O
Olof Johansson 已提交
756 757
		left = (struct tnode *) tnode_get_child(tn, 2*i);
		put_child(t, tn, 2*i, NULL);
758

O
Olof Johansson 已提交
759
		BUG_ON(!left);
760

O
Olof Johansson 已提交
761 762
		right = (struct tnode *) tnode_get_child(tn, 2*i+1);
		put_child(t, tn, 2*i+1, NULL);
763

O
Olof Johansson 已提交
764
		BUG_ON(!right);
765

O
Olof Johansson 已提交
766 767 768 769
		size = tnode_child_length(left);
		for (j = 0; j < size; j++) {
			put_child(t, left, j, inode->child[j]);
			put_child(t, right, j, inode->child[j + size]);
770
		}
O
Olof Johansson 已提交
771 772 773 774
		put_child(t, tn, 2*i, resize(t, left));
		put_child(t, tn, 2*i+1, resize(t, right));

		tnode_free(inode);
775 776 777
	}
	tnode_free(oldtnode);
	return tn;
778 779 780 781 782
nomem:
	{
		int size = tnode_child_length(tn);
		int j;

S
Stephen Hemminger 已提交
783
		for (j = 0; j < size; j++)
784 785 786 787
			if (tn->child[j])
				tnode_free((struct tnode *)tn->child[j]);

		tnode_free(tn);
S
Stephen Hemminger 已提交
788

789 790
		return ERR_PTR(-ENOMEM);
	}
791 792
}

793
static struct tnode *halve(struct trie *t, struct tnode *tn)
794 795 796 797 798 799
{
	struct tnode *oldtnode = tn;
	struct node *left, *right;
	int i;
	int olen = tnode_child_length(tn);

S
Stephen Hemminger 已提交
800
	pr_debug("In halve\n");
801 802

	tn = tnode_new(oldtnode->key, oldtnode->pos, oldtnode->bits - 1);
803

804 805
	if (!tn)
		return ERR_PTR(-ENOMEM);
806 807

	/*
808 809 810
	 * Preallocate and store tnodes before the actual work so we
	 * don't get into an inconsistent state if memory allocation
	 * fails. In case of failure we return the oldnode and halve
811 812 813
	 * of tnode is ignored.
	 */

O
Olof Johansson 已提交
814
	for (i = 0; i < olen; i += 2) {
815 816
		left = tnode_get_child(oldtnode, i);
		right = tnode_get_child(oldtnode, i+1);
817

818
		/* Two nonempty children */
S
Stephen Hemminger 已提交
819
		if (left && right) {
820
			struct tnode *newn;
S
Stephen Hemminger 已提交
821

822
			newn = tnode_new(left->key, tn->pos + tn->bits, 1);
S
Stephen Hemminger 已提交
823 824

			if (!newn)
825
				goto nomem;
S
Stephen Hemminger 已提交
826

827
			put_child(t, tn, i/2, (struct node *)newn);
828 829 830
		}

	}
831

O
Olof Johansson 已提交
832 833 834
	for (i = 0; i < olen; i += 2) {
		struct tnode *newBinNode;

835 836
		left = tnode_get_child(oldtnode, i);
		right = tnode_get_child(oldtnode, i+1);
837

838 839 840 841 842
		/* At least one of the children is empty */
		if (left == NULL) {
			if (right == NULL)    /* Both are empty */
				continue;
			put_child(t, tn, i/2, right);
O
Olof Johansson 已提交
843
			continue;
S
Stephen Hemminger 已提交
844
		}
O
Olof Johansson 已提交
845 846

		if (right == NULL) {
847
			put_child(t, tn, i/2, left);
O
Olof Johansson 已提交
848 849
			continue;
		}
850

851
		/* Two nonempty children */
O
Olof Johansson 已提交
852 853 854 855 856
		newBinNode = (struct tnode *) tnode_get_child(tn, i/2);
		put_child(t, tn, i/2, NULL);
		put_child(t, newBinNode, 0, left);
		put_child(t, newBinNode, 1, right);
		put_child(t, tn, i/2, resize(t, newBinNode));
857 858 859
	}
	tnode_free(oldtnode);
	return tn;
860 861 862 863 864
nomem:
	{
		int size = tnode_child_length(tn);
		int j;

S
Stephen Hemminger 已提交
865
		for (j = 0; j < size; j++)
866 867 868 869
			if (tn->child[j])
				tnode_free((struct tnode *)tn->child[j]);

		tnode_free(tn);
S
Stephen Hemminger 已提交
870

871 872
		return ERR_PTR(-ENOMEM);
	}
873 874
}

R
Robert Olsson 已提交
875
/* readside must use rcu_read_lock currently dump routines
R
Robert Olsson 已提交
876 877
 via get_fa_head and dump */

R
Robert Olsson 已提交
878
static struct leaf_info *find_leaf_info(struct leaf *l, int plen)
879
{
R
Robert Olsson 已提交
880
	struct hlist_head *head = &l->list;
881 882 883
	struct hlist_node *node;
	struct leaf_info *li;

R
Robert Olsson 已提交
884
	hlist_for_each_entry_rcu(li, node, head, hlist)
885
		if (li->plen == plen)
886
			return li;
O
Olof Johansson 已提交
887

888 889 890 891 892
	return NULL;
}

static inline struct list_head * get_fa_head(struct leaf *l, int plen)
{
R
Robert Olsson 已提交
893
	struct leaf_info *li = find_leaf_info(l, plen);
894

O
Olof Johansson 已提交
895 896
	if (!li)
		return NULL;
897

O
Olof Johansson 已提交
898
	return &li->falh;
899 900 901 902
}

static void insert_leaf_info(struct hlist_head *head, struct leaf_info *new)
{
903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919
	struct leaf_info *li = NULL, *last = NULL;
	struct hlist_node *node;

	if (hlist_empty(head)) {
		hlist_add_head_rcu(&new->hlist, head);
	} else {
		hlist_for_each_entry(li, node, head, hlist) {
			if (new->plen > li->plen)
				break;

			last = li;
		}
		if (last)
			hlist_add_after_rcu(&last->hlist, &new->hlist);
		else
			hlist_add_before_rcu(&new->hlist, &li->hlist);
	}
920 921
}

R
Robert Olsson 已提交
922 923
/* rcu_read_lock needs to be hold by caller from readside */

924 925 926 927 928 929 930 931
static struct leaf *
fib_find_node(struct trie *t, u32 key)
{
	int pos;
	struct tnode *tn;
	struct node *n;

	pos = 0;
R
Robert Olsson 已提交
932
	n = rcu_dereference(t->trie);
933 934 935

	while (n != NULL &&  NODE_TYPE(n) == T_TNODE) {
		tn = (struct tnode *) n;
O
Olof Johansson 已提交
936

937
		check_tnode(tn);
O
Olof Johansson 已提交
938

939
		if (tkey_sub_equals(tn->key, pos, tn->pos-pos, key)) {
O
Olof Johansson 已提交
940
			pos = tn->pos + tn->bits;
941
			n = tnode_get_child(tn, tkey_extract_bits(key, tn->pos, tn->bits));
O
Olof Johansson 已提交
942
		} else
943 944 945 946
			break;
	}
	/* Case we have found a leaf. Compare prefixes */

O
Olof Johansson 已提交
947 948 949
	if (n != NULL && IS_LEAF(n) && tkey_equals(key, n->key))
		return (struct leaf *)n;

950 951 952 953 954 955
	return NULL;
}

static struct node *trie_rebalance(struct trie *t, struct tnode *tn)
{
	int wasfull;
S
Stephen Hemminger 已提交
956 957
	t_key cindex, key = tn->key;
	struct tnode *tp;
958

S
Stephen Hemminger 已提交
959
	while (tn != NULL && (tp = node_parent((struct node *)tn)) != NULL) {
960 961 962 963
		cindex = tkey_extract_bits(key, tp->pos, tp->bits);
		wasfull = tnode_full(tp, tnode_get_child(tp, cindex));
		tn = (struct tnode *) resize (t, (struct tnode *)tn);
		tnode_put_child_reorg((struct tnode *)tp, cindex,(struct node*)tn, wasfull);
O
Olof Johansson 已提交
964

S
Stephen Hemminger 已提交
965 966
		tp = node_parent((struct node *) tn);
		if (!tp)
967
			break;
S
Stephen Hemminger 已提交
968
		tn = tp;
969
	}
S
Stephen Hemminger 已提交
970

971
	/* Handle last (top) tnode */
972
	if (IS_TNODE(tn))
973 974 975 976 977
		tn = (struct tnode*) resize(t, (struct tnode *)tn);

	return (struct node*) tn;
}

R
Robert Olsson 已提交
978 979
/* only used from updater-side */

980
static struct list_head *fib_insert_node(struct trie *t, u32 key, int plen)
981 982 983 984 985 986
{
	int pos, newpos;
	struct tnode *tp = NULL, *tn = NULL;
	struct node *n;
	struct leaf *l;
	int missbit;
987
	struct list_head *fa_head = NULL;
988 989 990 991
	struct leaf_info *li;
	t_key cindex;

	pos = 0;
992
	n = t->trie;
993

994 995
	/* If we point to NULL, stop. Either the tree is empty and we should
	 * just put a new leaf in if, or we have reached an empty child slot,
996
	 * and we should just put our new leaf in that.
997 998
	 * If we point to a T_TNODE, check if it matches our key. Note that
	 * a T_TNODE might be skipping any number of bits - its 'pos' need
999 1000
	 * not be the parent's 'pos'+'bits'!
	 *
1001
	 * If it does match the current key, get pos/bits from it, extract
1002 1003 1004 1005
	 * the index from our key, push the T_TNODE and walk the tree.
	 *
	 * If it doesn't, we have to replace it with a new T_TNODE.
	 *
1006 1007 1008
	 * If we point to a T_LEAF, it might or might not have the same key
	 * as we do. If it does, just change the value, update the T_LEAF's
	 * value, and return it.
1009 1010 1011 1012 1013
	 * If it doesn't, we need to replace it with a T_TNODE.
	 */

	while (n != NULL &&  NODE_TYPE(n) == T_TNODE) {
		tn = (struct tnode *) n;
O
Olof Johansson 已提交
1014

1015
		check_tnode(tn);
O
Olof Johansson 已提交
1016

1017
		if (tkey_sub_equals(tn->key, pos, tn->pos-pos, key)) {
1018
			tp = tn;
O
Olof Johansson 已提交
1019
			pos = tn->pos + tn->bits;
1020 1021
			n = tnode_get_child(tn, tkey_extract_bits(key, tn->pos, tn->bits));

S
Stephen Hemminger 已提交
1022
			BUG_ON(n && node_parent(n) != tn);
O
Olof Johansson 已提交
1023
		} else
1024 1025 1026 1027 1028 1029
			break;
	}

	/*
	 * n  ----> NULL, LEAF or TNODE
	 *
1030
	 * tp is n's (parent) ----> NULL or TNODE
1031 1032
	 */

O
Olof Johansson 已提交
1033
	BUG_ON(tp && IS_LEAF(tp));
1034 1035 1036

	/* Case 1: n is a leaf. Compare prefixes */

1037
	if (n != NULL && IS_LEAF(n) && tkey_equals(key, n->key)) {
1038
		l = (struct leaf *) n;
1039
		li = leaf_info_new(plen);
O
Olof Johansson 已提交
1040

1041 1042
		if (!li)
			return NULL;
1043 1044 1045 1046 1047 1048 1049 1050

		fa_head = &li->falh;
		insert_leaf_info(&l->list, li);
		goto done;
	}
	t->size++;
	l = leaf_new();

1051 1052
	if (!l)
		return NULL;
1053 1054 1055 1056

	l->key = key;
	li = leaf_info_new(plen);

1057
	if (!li) {
1058
		tnode_free((struct tnode *) l);
1059
		return NULL;
1060
	}
1061 1062 1063 1064 1065

	fa_head = &li->falh;
	insert_leaf_info(&l->list, li);

	if (t->trie && n == NULL) {
O
Olof Johansson 已提交
1066
		/* Case 2: n is NULL, and will just insert a new leaf */
1067

S
Stephen Hemminger 已提交
1068
		node_set_parent((struct node *)l, tp);
1069

O
Olof Johansson 已提交
1070 1071 1072 1073
		cindex = tkey_extract_bits(key, tp->pos, tp->bits);
		put_child(t, (struct tnode *)tp, cindex, (struct node *)l);
	} else {
		/* Case 3: n is a LEAF or a TNODE and the key doesn't match. */
1074 1075
		/*
		 *  Add a new tnode here
1076 1077 1078 1079
		 *  first tnode need some special handling
		 */

		if (tp)
O
Olof Johansson 已提交
1080
			pos = tp->pos+tp->bits;
1081
		else
O
Olof Johansson 已提交
1082 1083
			pos = 0;

1084
		if (n) {
1085 1086
			newpos = tkey_mismatch(key, pos, n->key);
			tn = tnode_new(n->key, newpos, 1);
O
Olof Johansson 已提交
1087
		} else {
1088
			newpos = 0;
1089
			tn = tnode_new(key, newpos, 1); /* First tnode */
1090 1091
		}

1092
		if (!tn) {
1093 1094
			free_leaf_info(li);
			tnode_free((struct tnode *) l);
1095
			return NULL;
O
Olof Johansson 已提交
1096 1097
		}

S
Stephen Hemminger 已提交
1098
		node_set_parent((struct node *)tn, tp);
1099

O
Olof Johansson 已提交
1100
		missbit = tkey_extract_bits(key, newpos, 1);
1101 1102 1103
		put_child(t, tn, missbit, (struct node *)l);
		put_child(t, tn, 1-missbit, n);

1104
		if (tp) {
1105 1106
			cindex = tkey_extract_bits(key, tp->pos, tp->bits);
			put_child(t, (struct tnode *)tp, cindex, (struct node *)tn);
O
Olof Johansson 已提交
1107
		} else {
R
Robert Olsson 已提交
1108
			rcu_assign_pointer(t->trie, (struct node *)tn); /* First tnode */
1109 1110 1111
			tp = tn;
		}
	}
O
Olof Johansson 已提交
1112 1113

	if (tp && tp->pos + tp->bits > 32)
S
Stephen Hemminger 已提交
1114
		printk(KERN_WARNING "fib_trie tp=%p pos=%d, bits=%d, key=%0x plen=%d\n",
1115
		       tp, tp->pos, tp->bits, key, plen);
O
Olof Johansson 已提交
1116

1117
	/* Rebalance the trie */
R
Robert Olsson 已提交
1118 1119

	rcu_assign_pointer(t->trie, trie_rebalance(t, tp));
1120
done:
1121 1122 1123
	return fa_head;
}

1124 1125 1126
/*
 * Caller must hold RTNL.
 */
1127
static int fn_trie_insert(struct fib_table *tb, struct fib_config *cfg)
1128 1129 1130
{
	struct trie *t = (struct trie *) tb->tb_data;
	struct fib_alias *fa, *new_fa;
1131
	struct list_head *fa_head = NULL;
1132
	struct fib_info *fi;
1133 1134
	int plen = cfg->fc_dst_len;
	u8 tos = cfg->fc_tos;
1135 1136 1137 1138 1139 1140 1141
	u32 key, mask;
	int err;
	struct leaf *l;

	if (plen > 32)
		return -EINVAL;

1142
	key = ntohl(cfg->fc_dst);
1143

1144
	pr_debug("Insert table=%u %08x/%d\n", tb->tb_id, key, plen);
1145

O
Olof Johansson 已提交
1146
	mask = ntohl(inet_make_mask(plen));
1147

1148
	if (key & ~mask)
1149 1150 1151 1152
		return -EINVAL;

	key = key & mask;

1153 1154 1155
	fi = fib_create_info(cfg);
	if (IS_ERR(fi)) {
		err = PTR_ERR(fi);
1156
		goto err;
1157
	}
1158 1159

	l = fib_find_node(t, key);
1160
	fa = NULL;
1161

1162
	if (l) {
1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177
		fa_head = get_fa_head(l, plen);
		fa = fib_find_alias(fa_head, tos, fi->fib_priority);
	}

	/* Now fa, if non-NULL, points to the first fib alias
	 * with the same keys [prefix,tos,priority], if such key already
	 * exists or to the node before which we will insert new one.
	 *
	 * If fa is NULL, we will need to allocate a new one and
	 * insert to the head of f.
	 *
	 * If f is NULL, no fib node matched the destination key
	 * and we need to allocate a new one of those as well.
	 */

O
Olof Johansson 已提交
1178
	if (fa && fa->fa_info->fib_priority == fi->fib_priority) {
1179 1180 1181
		struct fib_alias *fa_orig;

		err = -EEXIST;
1182
		if (cfg->fc_nlflags & NLM_F_EXCL)
1183 1184
			goto out;

1185
		if (cfg->fc_nlflags & NLM_F_REPLACE) {
1186 1187 1188
			struct fib_info *fi_drop;
			u8 state;

1189 1190 1191
			if (fi->fib_treeref > 1)
				goto out;

R
Robert Olsson 已提交
1192
			err = -ENOBUFS;
1193
			new_fa = kmem_cache_alloc(fn_alias_kmem, GFP_KERNEL);
R
Robert Olsson 已提交
1194 1195
			if (new_fa == NULL)
				goto out;
1196 1197

			fi_drop = fa->fa_info;
R
Robert Olsson 已提交
1198 1199
			new_fa->fa_tos = fa->fa_tos;
			new_fa->fa_info = fi;
1200 1201
			new_fa->fa_type = cfg->fc_type;
			new_fa->fa_scope = cfg->fc_scope;
1202
			state = fa->fa_state;
R
Robert Olsson 已提交
1203
			new_fa->fa_state &= ~FA_S_ACCESSED;
1204

R
Robert Olsson 已提交
1205 1206
			list_replace_rcu(&fa->fa_list, &new_fa->fa_list);
			alias_free_mem_rcu(fa);
1207 1208 1209

			fib_release_info(fi_drop);
			if (state & FA_S_ACCESSED)
O
Olof Johansson 已提交
1210
				rt_cache_flush(-1);
1211 1212
			rtmsg_fib(RTM_NEWROUTE, htonl(key), new_fa, plen,
				tb->tb_id, &cfg->fc_nlinfo, NLM_F_REPLACE);
1213

O
Olof Johansson 已提交
1214
			goto succeeded;
1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225
		}
		/* Error if we find a perfect match which
		 * uses the same scope, type, and nexthop
		 * information.
		 */
		fa_orig = fa;
		list_for_each_entry(fa, fa_orig->fa_list.prev, fa_list) {
			if (fa->fa_tos != tos)
				break;
			if (fa->fa_info->fib_priority != fi->fib_priority)
				break;
1226 1227
			if (fa->fa_type == cfg->fc_type &&
			    fa->fa_scope == cfg->fc_scope &&
1228 1229 1230 1231
			    fa->fa_info == fi) {
				goto out;
			}
		}
1232
		if (!(cfg->fc_nlflags & NLM_F_APPEND))
1233 1234 1235
			fa = fa_orig;
	}
	err = -ENOENT;
1236
	if (!(cfg->fc_nlflags & NLM_F_CREATE))
1237 1238 1239
		goto out;

	err = -ENOBUFS;
1240
	new_fa = kmem_cache_alloc(fn_alias_kmem, GFP_KERNEL);
1241 1242 1243 1244 1245
	if (new_fa == NULL)
		goto out;

	new_fa->fa_info = fi;
	new_fa->fa_tos = tos;
1246 1247
	new_fa->fa_type = cfg->fc_type;
	new_fa->fa_scope = cfg->fc_scope;
1248 1249 1250 1251 1252
	new_fa->fa_state = 0;
	/*
	 * Insert new entry to the list.
	 */

1253
	if (!fa_head) {
1254 1255 1256
		fa_head = fib_insert_node(t, key, plen);
		if (unlikely(!fa_head)) {
			err = -ENOMEM;
1257
			goto out_free_new_fa;
1258
		}
1259
	}
1260

R
Robert Olsson 已提交
1261 1262
	list_add_tail_rcu(&new_fa->fa_list,
			  (fa ? &fa->fa_list : fa_head));
1263 1264

	rt_cache_flush(-1);
1265
	rtmsg_fib(RTM_NEWROUTE, htonl(key), new_fa, plen, tb->tb_id,
1266
		  &cfg->fc_nlinfo, 0);
1267 1268
succeeded:
	return 0;
1269 1270 1271

out_free_new_fa:
	kmem_cache_free(fn_alias_kmem, new_fa);
1272 1273
out:
	fib_release_info(fi);
O
Olof Johansson 已提交
1274
err:
1275 1276 1277
	return err;
}

R
Robert Olsson 已提交
1278

R
Robert Olsson 已提交
1279
/* should be called with rcu_read_lock */
S
Stephen Hemminger 已提交
1280 1281
static inline int check_leaf(struct trie *t, struct leaf *l,
			     t_key key, int *plen, const struct flowi *flp,
1282
			     struct fib_result *res)
1283
{
1284
	int err, i;
1285
	__be32 mask;
1286 1287 1288
	struct leaf_info *li;
	struct hlist_head *hhead = &l->list;
	struct hlist_node *node;
1289

R
Robert Olsson 已提交
1290
	hlist_for_each_entry_rcu(li, node, hhead, hlist) {
1291
		i = li->plen;
1292 1293
		mask = inet_make_mask(i);
		if (l->key != (key & ntohl(mask)))
1294 1295
			continue;

1296
		if ((err = fib_semantic_match(&li->falh, flp, res, htonl(l->key), mask, i)) <= 0) {
1297 1298 1299 1300
			*plen = i;
#ifdef CONFIG_IP_FIB_TRIE_STATS
			t->stats.semantic_match_passed++;
#endif
1301
			return err;
1302 1303 1304 1305 1306
		}
#ifdef CONFIG_IP_FIB_TRIE_STATS
		t->stats.semantic_match_miss++;
#endif
	}
1307
	return 1;
1308 1309 1310 1311 1312 1313 1314 1315 1316 1317
}

static int
fn_trie_lookup(struct fib_table *tb, const struct flowi *flp, struct fib_result *res)
{
	struct trie *t = (struct trie *) tb->tb_data;
	int plen, ret = 0;
	struct node *n;
	struct tnode *pn;
	int pos, bits;
O
Olof Johansson 已提交
1318
	t_key key = ntohl(flp->fl4_dst);
1319 1320 1321
	int chopped_off;
	t_key cindex = 0;
	int current_prefix_length = KEYLENGTH;
O
Olof Johansson 已提交
1322 1323 1324 1325
	struct tnode *cn;
	t_key node_prefix, key_prefix, pref_mismatch;
	int mp;

R
Robert Olsson 已提交
1326
	rcu_read_lock();
O
Olof Johansson 已提交
1327

R
Robert Olsson 已提交
1328
	n = rcu_dereference(t->trie);
1329
	if (!n)
1330 1331 1332 1333 1334 1335 1336 1337
		goto failed;

#ifdef CONFIG_IP_FIB_TRIE_STATS
	t->stats.gets++;
#endif

	/* Just a leaf? */
	if (IS_LEAF(n)) {
1338
		if ((ret = check_leaf(t, (struct leaf *)n, key, &plen, flp, res)) <= 0)
1339 1340 1341 1342 1343
			goto found;
		goto failed;
	}
	pn = (struct tnode *) n;
	chopped_off = 0;
1344

O
Olof Johansson 已提交
1345
	while (pn) {
1346 1347 1348
		pos = pn->pos;
		bits = pn->bits;

1349
		if (!chopped_off)
S
Stephen Hemminger 已提交
1350 1351
			cindex = tkey_extract_bits(mask_pfx(key, current_prefix_length),
						   pos, bits);
1352 1353 1354 1355 1356 1357 1358 1359 1360 1361

		n = tnode_get_child(pn, cindex);

		if (n == NULL) {
#ifdef CONFIG_IP_FIB_TRIE_STATS
			t->stats.null_node_hit++;
#endif
			goto backtrace;
		}

O
Olof Johansson 已提交
1362 1363 1364 1365 1366 1367 1368
		if (IS_LEAF(n)) {
			if ((ret = check_leaf(t, (struct leaf *)n, key, &plen, flp, res)) <= 0)
				goto found;
			else
				goto backtrace;
		}

1369 1370
#define HL_OPTIMIZE
#ifdef HL_OPTIMIZE
O
Olof Johansson 已提交
1371
		cn = (struct tnode *)n;
1372

O
Olof Johansson 已提交
1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387
		/*
		 * It's a tnode, and we can do some extra checks here if we
		 * like, to avoid descending into a dead-end branch.
		 * This tnode is in the parent's child array at index
		 * key[p_pos..p_pos+p_bits] but potentially with some bits
		 * chopped off, so in reality the index may be just a
		 * subprefix, padded with zero at the end.
		 * We can also take a look at any skipped bits in this
		 * tnode - everything up to p_pos is supposed to be ok,
		 * and the non-chopped bits of the index (se previous
		 * paragraph) are also guaranteed ok, but the rest is
		 * considered unknown.
		 *
		 * The skipped bits are key[pos+bits..cn->pos].
		 */
1388

O
Olof Johansson 已提交
1389 1390 1391 1392 1393 1394 1395 1396 1397
		/* If current_prefix_length < pos+bits, we are already doing
		 * actual prefix  matching, which means everything from
		 * pos+(bits-chopped_off) onward must be zero along some
		 * branch of this subtree - otherwise there is *no* valid
		 * prefix present. Here we can only check the skipped
		 * bits. Remember, since we have already indexed into the
		 * parent's child array, we know that the bits we chopped of
		 * *are* zero.
		 */
1398

O
Olof Johansson 已提交
1399
		/* NOTA BENE: CHECKING ONLY SKIPPED BITS FOR THE NEW NODE HERE */
1400

O
Olof Johansson 已提交
1401 1402 1403 1404 1405 1406
		if (current_prefix_length < pos+bits) {
			if (tkey_extract_bits(cn->key, current_prefix_length,
						cn->pos - current_prefix_length) != 0 ||
			    !(cn->child[0]))
				goto backtrace;
		}
1407

O
Olof Johansson 已提交
1408 1409 1410 1411 1412 1413 1414 1415 1416
		/*
		 * If chopped_off=0, the index is fully validated and we
		 * only need to look at the skipped bits for this, the new,
		 * tnode. What we actually want to do is to find out if
		 * these skipped bits match our key perfectly, or if we will
		 * have to count on finding a matching prefix further down,
		 * because if we do, we would like to have some way of
		 * verifying the existence of such a prefix at this point.
		 */
1417

O
Olof Johansson 已提交
1418 1419 1420 1421 1422 1423 1424 1425
		/* The only thing we can do at this point is to verify that
		 * any such matching prefix can indeed be a prefix to our
		 * key, and if the bits in the node we are inspecting that
		 * do not match our key are not ZERO, this cannot be true.
		 * Thus, find out where there is a mismatch (before cn->pos)
		 * and verify that all the mismatching bits are zero in the
		 * new tnode's key.
		 */
1426

O
Olof Johansson 已提交
1427 1428 1429 1430 1431 1432 1433 1434 1435 1436
		/* Note: We aren't very concerned about the piece of the key
		 * that precede pn->pos+pn->bits, since these have already been
		 * checked. The bits after cn->pos aren't checked since these are
		 * by definition "unknown" at this point. Thus, what we want to
		 * see is if we are about to enter the "prefix matching" state,
		 * and in that case verify that the skipped bits that will prevail
		 * throughout this subtree are zero, as they have to be if we are
		 * to find a matching prefix.
		 */

S
Stephen Hemminger 已提交
1437 1438
		node_prefix = mask_pfx(cn->key, cn->pos);
		key_prefix = mask_pfx(key, cn->pos);
O
Olof Johansson 已提交
1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456
		pref_mismatch = key_prefix^node_prefix;
		mp = 0;

		/* In short: If skipped bits in this node do not match the search
		 * key, enter the "prefix matching" state.directly.
		 */
		if (pref_mismatch) {
			while (!(pref_mismatch & (1<<(KEYLENGTH-1)))) {
				mp++;
				pref_mismatch = pref_mismatch <<1;
			}
			key_prefix = tkey_extract_bits(cn->key, mp, cn->pos-mp);

			if (key_prefix != 0)
				goto backtrace;

			if (current_prefix_length >= cn->pos)
				current_prefix_length = mp;
1457
		}
O
Olof Johansson 已提交
1458 1459 1460 1461 1462
#endif
		pn = (struct tnode *)n; /* Descend */
		chopped_off = 0;
		continue;

1463 1464 1465 1466
backtrace:
		chopped_off++;

		/* As zero don't change the child key (cindex) */
O
Olof Johansson 已提交
1467
		while ((chopped_off <= pn->bits) && !(cindex & (1<<(chopped_off-1))))
1468 1469 1470 1471 1472
			chopped_off++;

		/* Decrease current_... with bits chopped off */
		if (current_prefix_length > pn->pos + pn->bits - chopped_off)
			current_prefix_length = pn->pos + pn->bits - chopped_off;
O
Olof Johansson 已提交
1473

1474
		/*
1475
		 * Either we do the actual chop off according or if we have
1476 1477 1478
		 * chopped off all bits in this tnode walk up to our parent.
		 */

O
Olof Johansson 已提交
1479
		if (chopped_off <= pn->bits) {
1480
			cindex &= ~(1 << (chopped_off-1));
O
Olof Johansson 已提交
1481
		} else {
S
Stephen Hemminger 已提交
1482 1483
			struct tnode *parent = node_parent((struct node *) pn);
			if (!parent)
1484
				goto failed;
O
Olof Johansson 已提交
1485

1486
			/* Get Child's index */
S
Stephen Hemminger 已提交
1487 1488
			cindex = tkey_extract_bits(pn->key, parent->pos, parent->bits);
			pn = parent;
1489 1490 1491 1492 1493 1494
			chopped_off = 0;

#ifdef CONFIG_IP_FIB_TRIE_STATS
			t->stats.backtrack++;
#endif
			goto backtrace;
1495
		}
1496 1497
	}
failed:
1498
	ret = 1;
1499
found:
R
Robert Olsson 已提交
1500
	rcu_read_unlock();
1501 1502 1503
	return ret;
}

R
Robert Olsson 已提交
1504
/* only called from updater side */
1505 1506 1507 1508 1509 1510 1511
static int trie_leaf_remove(struct trie *t, t_key key)
{
	t_key cindex;
	struct tnode *tp = NULL;
	struct node *n = t->trie;
	struct leaf *l;

S
Stephen Hemminger 已提交
1512
	pr_debug("entering trie_leaf_remove(%p)\n", n);
1513 1514

	/* Note that in the case skipped bits, those bits are *not* checked!
1515
	 * When we finish this, we will have NULL or a T_LEAF, and the
1516 1517 1518
	 * T_LEAF may or may not match our key.
	 */

O
Olof Johansson 已提交
1519
	while (n != NULL && IS_TNODE(n)) {
1520 1521 1522 1523
		struct tnode *tn = (struct tnode *) n;
		check_tnode(tn);
		n = tnode_get_child(tn ,tkey_extract_bits(key, tn->pos, tn->bits));

S
Stephen Hemminger 已提交
1524
		BUG_ON(n && node_parent(n) != tn);
O
Olof Johansson 已提交
1525
	}
1526 1527
	l = (struct leaf *) n;

1528
	if (!n || !tkey_equals(l->key, key))
1529
		return 0;
1530 1531 1532 1533

	/*
	 * Key found.
	 * Remove the leaf and rebalance the tree
1534 1535 1536 1537
	 */

	t->size--;

S
Stephen Hemminger 已提交
1538
	tp = node_parent(n);
1539 1540
	tnode_free((struct tnode *) n);

1541
	if (tp) {
1542 1543
		cindex = tkey_extract_bits(key, tp->pos, tp->bits);
		put_child(t, (struct tnode *)tp, cindex, NULL);
R
Robert Olsson 已提交
1544
		rcu_assign_pointer(t->trie, trie_rebalance(t, tp));
O
Olof Johansson 已提交
1545
	} else
R
Robert Olsson 已提交
1546
		rcu_assign_pointer(t->trie, NULL);
1547 1548 1549 1550

	return 1;
}

1551 1552 1553
/*
 * Caller must hold RTNL.
 */
1554
static int fn_trie_delete(struct fib_table *tb, struct fib_config *cfg)
1555 1556 1557
{
	struct trie *t = (struct trie *) tb->tb_data;
	u32 key, mask;
1558 1559
	int plen = cfg->fc_dst_len;
	u8 tos = cfg->fc_tos;
1560 1561 1562
	struct fib_alias *fa, *fa_to_delete;
	struct list_head *fa_head;
	struct leaf *l;
O
Olof Johansson 已提交
1563 1564
	struct leaf_info *li;

1565
	if (plen > 32)
1566 1567
		return -EINVAL;

1568
	key = ntohl(cfg->fc_dst);
O
Olof Johansson 已提交
1569
	mask = ntohl(inet_make_mask(plen));
1570

1571
	if (key & ~mask)
1572 1573 1574 1575 1576
		return -EINVAL;

	key = key & mask;
	l = fib_find_node(t, key);

1577
	if (!l)
1578 1579 1580 1581 1582 1583 1584 1585
		return -ESRCH;

	fa_head = get_fa_head(l, plen);
	fa = fib_find_alias(fa_head, tos, 0);

	if (!fa)
		return -ESRCH;

S
Stephen Hemminger 已提交
1586
	pr_debug("Deleting %08x/%d tos=%d t=%p\n", key, plen, tos, t);
1587 1588 1589

	fa_to_delete = NULL;
	fa_head = fa->fa_list.prev;
R
Robert Olsson 已提交
1590

1591 1592 1593 1594 1595 1596
	list_for_each_entry(fa, fa_head, fa_list) {
		struct fib_info *fi = fa->fa_info;

		if (fa->fa_tos != tos)
			break;

1597 1598 1599 1600 1601 1602
		if ((!cfg->fc_type || fa->fa_type == cfg->fc_type) &&
		    (cfg->fc_scope == RT_SCOPE_NOWHERE ||
		     fa->fa_scope == cfg->fc_scope) &&
		    (!cfg->fc_protocol ||
		     fi->fib_protocol == cfg->fc_protocol) &&
		    fib_nh_match(cfg, fi) == 0) {
1603 1604 1605 1606 1607
			fa_to_delete = fa;
			break;
		}
	}

O
Olof Johansson 已提交
1608 1609
	if (!fa_to_delete)
		return -ESRCH;
1610

O
Olof Johansson 已提交
1611
	fa = fa_to_delete;
1612
	rtmsg_fib(RTM_DELROUTE, htonl(key), fa, plen, tb->tb_id,
1613
		  &cfg->fc_nlinfo, 0);
O
Olof Johansson 已提交
1614 1615

	l = fib_find_node(t, key);
R
Robert Olsson 已提交
1616
	li = find_leaf_info(l, plen);
1617

R
Robert Olsson 已提交
1618
	list_del_rcu(&fa->fa_list);
1619

O
Olof Johansson 已提交
1620
	if (list_empty(fa_head)) {
R
Robert Olsson 已提交
1621
		hlist_del_rcu(&li->hlist);
O
Olof Johansson 已提交
1622
		free_leaf_info(li);
R
Robert Olsson 已提交
1623
	}
1624

O
Olof Johansson 已提交
1625 1626
	if (hlist_empty(&l->list))
		trie_leaf_remove(t, key);
1627

O
Olof Johansson 已提交
1628 1629
	if (fa->fa_state & FA_S_ACCESSED)
		rt_cache_flush(-1);
1630

R
Robert Olsson 已提交
1631 1632
	fib_release_info(fa->fa_info);
	alias_free_mem_rcu(fa);
O
Olof Johansson 已提交
1633
	return 0;
1634 1635 1636 1637 1638 1639 1640 1641 1642 1643
}

static int trie_flush_list(struct trie *t, struct list_head *head)
{
	struct fib_alias *fa, *fa_node;
	int found = 0;

	list_for_each_entry_safe(fa, fa_node, head, fa_list) {
		struct fib_info *fi = fa->fa_info;

R
Robert Olsson 已提交
1644 1645 1646 1647
		if (fi && (fi->fib_flags & RTNH_F_DEAD)) {
			list_del_rcu(&fa->fa_list);
			fib_release_info(fa->fa_info);
			alias_free_mem_rcu(fa);
1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664
			found++;
		}
	}
	return found;
}

static int trie_flush_leaf(struct trie *t, struct leaf *l)
{
	int found = 0;
	struct hlist_head *lih = &l->list;
	struct hlist_node *node, *tmp;
	struct leaf_info *li = NULL;

	hlist_for_each_entry_safe(li, node, tmp, lih, hlist) {
		found += trie_flush_list(t, &li->falh);

		if (list_empty(&li->falh)) {
R
Robert Olsson 已提交
1665
			hlist_del_rcu(&li->hlist);
1666 1667 1668 1669 1670 1671
			free_leaf_info(li);
		}
	}
	return found;
}

R
Robert Olsson 已提交
1672 1673
/* rcu_read_lock needs to be hold by caller from readside */

1674 1675 1676 1677 1678
static struct leaf *nextleaf(struct trie *t, struct leaf *thisleaf)
{
	struct node *c = (struct node *) thisleaf;
	struct tnode *p;
	int idx;
R
Robert Olsson 已提交
1679
	struct node *trie = rcu_dereference(t->trie);
1680

1681
	if (c == NULL) {
R
Robert Olsson 已提交
1682
		if (trie == NULL)
1683 1684
			return NULL;

R
Robert Olsson 已提交
1685 1686
		if (IS_LEAF(trie))          /* trie w. just a leaf */
			return (struct leaf *) trie;
1687

R
Robert Olsson 已提交
1688
		p = (struct tnode*) trie;  /* Start */
O
Olof Johansson 已提交
1689
	} else
S
Stephen Hemminger 已提交
1690
		p = node_parent(c);
1691

1692 1693 1694 1695
	while (p) {
		int pos, last;

		/*  Find the next child of the parent */
1696 1697 1698
		if (c)
			pos = 1 + tkey_extract_bits(c->key, p->pos, p->bits);
		else
1699 1700 1701
			pos = 0;

		last = 1 << p->bits;
O
Olof Johansson 已提交
1702
		for (idx = pos; idx < last ; idx++) {
R
Robert Olsson 已提交
1703 1704 1705
			c = rcu_dereference(p->child[idx]);

			if (!c)
O
Olof Johansson 已提交
1706 1707 1708
				continue;

			/* Decend if tnode */
R
Robert Olsson 已提交
1709 1710
			while (IS_TNODE(c)) {
				p = (struct tnode *) c;
1711
				idx = 0;
O
Olof Johansson 已提交
1712 1713 1714

				/* Rightmost non-NULL branch */
				if (p && IS_TNODE(p))
R
Robert Olsson 已提交
1715 1716
					while (!(c = rcu_dereference(p->child[idx]))
					       && idx < (1<<p->bits)) idx++;
O
Olof Johansson 已提交
1717 1718

				/* Done with this tnode? */
R
Robert Olsson 已提交
1719
				if (idx >= (1 << p->bits) || !c)
O
Olof Johansson 已提交
1720
					goto up;
1721
			}
R
Robert Olsson 已提交
1722
			return (struct leaf *) c;
1723 1724 1725
		}
up:
		/* No more children go up one step  */
O
Olof Johansson 已提交
1726
		c = (struct node *) p;
S
Stephen Hemminger 已提交
1727
		p = node_parent(c);
1728 1729 1730 1731
	}
	return NULL; /* Ready. Root of trie */
}

1732 1733 1734
/*
 * Caller must hold RTNL.
 */
1735 1736 1737 1738 1739 1740
static int fn_trie_flush(struct fib_table *tb)
{
	struct trie *t = (struct trie *) tb->tb_data;
	struct leaf *ll = NULL, *l = NULL;
	int found = 0, h;

O
Olof Johansson 已提交
1741
	for (h = 0; (l = nextleaf(t, l)) != NULL; h++) {
1742 1743 1744 1745 1746 1747 1748 1749 1750 1751
		found += trie_flush_leaf(t, l);

		if (ll && hlist_empty(&ll->list))
			trie_leaf_remove(t, ll->key);
		ll = l;
	}

	if (ll && hlist_empty(&ll->list))
		trie_leaf_remove(t, ll->key);

S
Stephen Hemminger 已提交
1752
	pr_debug("trie_flush found=%d\n", found);
1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770
	return found;
}

static void
fn_trie_select_default(struct fib_table *tb, const struct flowi *flp, struct fib_result *res)
{
	struct trie *t = (struct trie *) tb->tb_data;
	int order, last_idx;
	struct fib_info *fi = NULL;
	struct fib_info *last_resort;
	struct fib_alias *fa = NULL;
	struct list_head *fa_head;
	struct leaf *l;

	last_idx = -1;
	last_resort = NULL;
	order = -1;

R
Robert Olsson 已提交
1771
	rcu_read_lock();
1772

1773
	l = fib_find_node(t, 0);
1774
	if (!l)
1775 1776 1777
		goto out;

	fa_head = get_fa_head(l, 0);
1778
	if (!fa_head)
1779 1780
		goto out;

1781
	if (list_empty(fa_head))
1782 1783
		goto out;

R
Robert Olsson 已提交
1784
	list_for_each_entry_rcu(fa, fa_head, fa_list) {
1785
		struct fib_info *next_fi = fa->fa_info;
O
Olof Johansson 已提交
1786

1787 1788 1789
		if (fa->fa_scope != res->scope ||
		    fa->fa_type != RTN_UNICAST)
			continue;
O
Olof Johansson 已提交
1790

1791 1792 1793 1794 1795 1796
		if (next_fi->fib_priority > res->fi->fib_priority)
			break;
		if (!next_fi->fib_nh[0].nh_gw ||
		    next_fi->fib_nh[0].nh_scope != RT_SCOPE_LINK)
			continue;
		fa->fa_state |= FA_S_ACCESSED;
O
Olof Johansson 已提交
1797

1798 1799 1800 1801
		if (fi == NULL) {
			if (next_fi != res->fi)
				break;
		} else if (!fib_detect_death(fi, order, &last_resort,
1802
					     &last_idx, tb->tb_default)) {
1803
			fib_result_assign(res, fi);
1804
			tb->tb_default = order;
1805 1806 1807 1808 1809 1810
			goto out;
		}
		fi = next_fi;
		order++;
	}
	if (order <= 0 || fi == NULL) {
1811
		tb->tb_default = -1;
1812 1813 1814
		goto out;
	}

1815 1816
	if (!fib_detect_death(fi, order, &last_resort, &last_idx,
				tb->tb_default)) {
1817
		fib_result_assign(res, fi);
1818
		tb->tb_default = order;
1819 1820
		goto out;
	}
1821 1822
	if (last_idx >= 0)
		fib_result_assign(res, last_resort);
1823 1824
	tb->tb_default = last_idx;
out:
R
Robert Olsson 已提交
1825
	rcu_read_unlock();
1826 1827
}

1828
static int fn_trie_dump_fa(t_key key, int plen, struct list_head *fah, struct fib_table *tb,
1829 1830 1831 1832 1833
			   struct sk_buff *skb, struct netlink_callback *cb)
{
	int i, s_i;
	struct fib_alias *fa;

A
Al Viro 已提交
1834
	__be32 xkey = htonl(key);
1835

1836
	s_i = cb->args[4];
1837 1838
	i = 0;

R
Robert Olsson 已提交
1839 1840 1841
	/* rcu_read_lock is hold by caller */

	list_for_each_entry_rcu(fa, fah, fa_list) {
1842 1843 1844 1845
		if (i < s_i) {
			i++;
			continue;
		}
S
Stephen Hemminger 已提交
1846
		BUG_ON(!fa->fa_info);
1847 1848 1849 1850 1851 1852 1853

		if (fib_dump_info(skb, NETLINK_CB(cb->skb).pid,
				  cb->nlh->nlmsg_seq,
				  RTM_NEWROUTE,
				  tb->tb_id,
				  fa->fa_type,
				  fa->fa_scope,
1854
				  xkey,
1855 1856
				  plen,
				  fa->fa_tos,
1857
				  fa->fa_info, 0) < 0) {
1858
			cb->args[4] = i;
1859
			return -1;
O
Olof Johansson 已提交
1860
		}
1861 1862
		i++;
	}
1863
	cb->args[4] = i;
1864 1865 1866
	return skb->len;
}

1867
static int fn_trie_dump_plen(struct trie *t, int plen, struct fib_table *tb, struct sk_buff *skb,
1868 1869 1870 1871 1872 1873
			     struct netlink_callback *cb)
{
	int h, s_h;
	struct list_head *fa_head;
	struct leaf *l = NULL;

1874
	s_h = cb->args[3];
1875

O
Olof Johansson 已提交
1876
	for (h = 0; (l = nextleaf(t, l)) != NULL; h++) {
1877 1878 1879
		if (h < s_h)
			continue;
		if (h > s_h)
1880 1881
			memset(&cb->args[4], 0,
			       sizeof(cb->args) - 4*sizeof(cb->args[0]));
1882 1883

		fa_head = get_fa_head(l, plen);
O
Olof Johansson 已提交
1884

1885
		if (!fa_head)
1886 1887
			continue;

1888
		if (list_empty(fa_head))
1889 1890 1891
			continue;

		if (fn_trie_dump_fa(l->key, plen, fa_head, tb, skb, cb)<0) {
1892
			cb->args[3] = h;
1893 1894 1895
			return -1;
		}
	}
1896
	cb->args[3] = h;
1897 1898 1899 1900 1901 1902 1903 1904
	return skb->len;
}

static int fn_trie_dump(struct fib_table *tb, struct sk_buff *skb, struct netlink_callback *cb)
{
	int m, s_m;
	struct trie *t = (struct trie *) tb->tb_data;

1905
	s_m = cb->args[2];
1906

R
Robert Olsson 已提交
1907
	rcu_read_lock();
O
Olof Johansson 已提交
1908
	for (m = 0; m <= 32; m++) {
1909 1910 1911
		if (m < s_m)
			continue;
		if (m > s_m)
1912 1913
			memset(&cb->args[3], 0,
				sizeof(cb->args) - 3*sizeof(cb->args[0]));
1914 1915

		if (fn_trie_dump_plen(t, 32-m, tb, skb, cb)<0) {
1916
			cb->args[2] = m;
1917 1918 1919
			goto out;
		}
	}
R
Robert Olsson 已提交
1920
	rcu_read_unlock();
1921
	cb->args[2] = m;
1922
	return skb->len;
O
Olof Johansson 已提交
1923
out:
R
Robert Olsson 已提交
1924
	rcu_read_unlock();
1925 1926 1927 1928 1929
	return -1;
}

/* Fix more generic FIB names for init later */

1930
struct fib_table *fib_hash_init(u32 id)
1931 1932 1933 1934 1935 1936 1937 1938
{
	struct fib_table *tb;
	struct trie *t;

	if (fn_alias_kmem == NULL)
		fn_alias_kmem = kmem_cache_create("ip_fib_alias",
						  sizeof(struct fib_alias),
						  0, SLAB_HWCACHE_ALIGN,
1939
						  NULL);
1940 1941 1942 1943 1944 1945 1946

	tb = kmalloc(sizeof(struct fib_table) + sizeof(struct trie),
		     GFP_KERNEL);
	if (tb == NULL)
		return NULL;

	tb->tb_id = id;
1947
	tb->tb_default = -1;
1948 1949 1950 1951 1952 1953 1954 1955
	tb->tb_lookup = fn_trie_lookup;
	tb->tb_insert = fn_trie_insert;
	tb->tb_delete = fn_trie_delete;
	tb->tb_flush = fn_trie_flush;
	tb->tb_select_default = fn_trie_select_default;
	tb->tb_dump = fn_trie_dump;

	t = (struct trie *) tb->tb_data;
1956
	memset(t, 0, sizeof(*t));
1957 1958

	if (id == RT_TABLE_LOCAL)
S
Stephen Hemminger 已提交
1959
		printk(KERN_INFO "IPv4 FIB: Using LC-trie version %s\n", VERSION);
1960 1961 1962 1963

	return tb;
}

1964 1965 1966
#ifdef CONFIG_PROC_FS
/* Depth first Trie walk iterator */
struct fib_trie_iter {
1967
	struct seq_net_private p;
1968
	struct trie *trie_local, *trie_main;
1969 1970 1971 1972 1973
	struct tnode *tnode;
	struct trie *trie;
	unsigned index;
	unsigned depth;
};
1974

1975
static struct node *fib_trie_get_next(struct fib_trie_iter *iter)
1976
{
1977 1978 1979
	struct tnode *tn = iter->tnode;
	unsigned cindex = iter->index;
	struct tnode *p;
1980

1981 1982 1983 1984
	/* A single entry routing table */
	if (!tn)
		return NULL;

1985 1986 1987 1988 1989
	pr_debug("get_next iter={node=%p index=%d depth=%d}\n",
		 iter->tnode, iter->index, iter->depth);
rescan:
	while (cindex < (1<<tn->bits)) {
		struct node *n = tnode_get_child(tn, cindex);
1990

1991 1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002
		if (n) {
			if (IS_LEAF(n)) {
				iter->tnode = tn;
				iter->index = cindex + 1;
			} else {
				/* push down one level */
				iter->tnode = (struct tnode *) n;
				iter->index = 0;
				++iter->depth;
			}
			return n;
		}
2003

2004 2005
		++cindex;
	}
O
Olof Johansson 已提交
2006

2007
	/* Current node exhausted, pop back up */
S
Stephen Hemminger 已提交
2008
	p = node_parent((struct node *)tn);
2009 2010 2011 2012 2013
	if (p) {
		cindex = tkey_extract_bits(tn->key, p->pos, p->bits)+1;
		tn = p;
		--iter->depth;
		goto rescan;
2014
	}
2015 2016 2017

	/* got root? */
	return NULL;
2018 2019
}

2020 2021
static struct node *fib_trie_get_first(struct fib_trie_iter *iter,
				       struct trie *t)
2022
{
2023 2024
	struct node *n ;

S
Stephen Hemminger 已提交
2025
	if (!t)
2026 2027 2028 2029
		return NULL;

	n = rcu_dereference(t->trie);

S
Stephen Hemminger 已提交
2030
	if (!iter)
2031
		return NULL;
2032

2033 2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044
	if (n) {
		if (IS_TNODE(n)) {
			iter->tnode = (struct tnode *) n;
			iter->trie = t;
			iter->index = 0;
			iter->depth = 1;
		} else {
			iter->tnode = NULL;
			iter->trie  = t;
			iter->index = 0;
			iter->depth = 0;
		}
2045
		return n;
O
Olof Johansson 已提交
2046
	}
2047 2048
	return NULL;
}
O
Olof Johansson 已提交
2049

2050 2051 2052 2053
static void trie_collect_stats(struct trie *t, struct trie_stat *s)
{
	struct node *n;
	struct fib_trie_iter iter;
O
Olof Johansson 已提交
2054

2055
	memset(s, 0, sizeof(*s));
O
Olof Johansson 已提交
2056

2057 2058 2059 2060 2061 2062 2063 2064 2065 2066 2067 2068 2069
	rcu_read_lock();
	for (n = fib_trie_get_first(&iter, t); n;
	     n = fib_trie_get_next(&iter)) {
		if (IS_LEAF(n)) {
			s->leaves++;
			s->totdepth += iter.depth;
			if (iter.depth > s->maxdepth)
				s->maxdepth = iter.depth;
		} else {
			const struct tnode *tn = (const struct tnode *) n;
			int i;

			s->tnodes++;
S
Stephen Hemminger 已提交
2070
			if (tn->bits < MAX_STAT_DEPTH)
R
Robert Olsson 已提交
2071 2072
				s->nodesizes[tn->bits]++;

2073 2074 2075
			for (i = 0; i < (1<<tn->bits); i++)
				if (!tn->child[i])
					s->nullpointers++;
2076 2077
		}
	}
R
Robert Olsson 已提交
2078
	rcu_read_unlock();
2079 2080
}

2081 2082 2083 2084
/*
 *	This outputs /proc/net/fib_triestats
 */
static void trie_show_stats(struct seq_file *seq, struct trie_stat *stat)
2085
{
2086
	unsigned i, max, pointers, bytes, avdepth;
2087

2088 2089 2090 2091
	if (stat->leaves)
		avdepth = stat->totdepth*100 / stat->leaves;
	else
		avdepth = 0;
O
Olof Johansson 已提交
2092

2093
	seq_printf(seq, "\tAver depth:     %u.%02d\n", avdepth / 100, avdepth % 100 );
2094
	seq_printf(seq, "\tMax depth:      %u\n", stat->maxdepth);
O
Olof Johansson 已提交
2095

2096
	seq_printf(seq, "\tLeaves:         %u\n", stat->leaves);
O
Olof Johansson 已提交
2097

2098
	bytes = sizeof(struct leaf) * stat->leaves;
2099
	seq_printf(seq, "\tInternal nodes: %u\n\t", stat->tnodes);
2100
	bytes += sizeof(struct tnode) * stat->tnodes;
2101

R
Robert Olsson 已提交
2102 2103
	max = MAX_STAT_DEPTH;
	while (max > 0 && stat->nodesizes[max-1] == 0)
2104
		max--;
2105

2106 2107 2108
	pointers = 0;
	for (i = 1; i <= max; i++)
		if (stat->nodesizes[i] != 0) {
2109
			seq_printf(seq, "  %u: %u",  i, stat->nodesizes[i]);
2110 2111 2112
			pointers += (1<<i) * stat->nodesizes[i];
		}
	seq_putc(seq, '\n');
2113
	seq_printf(seq, "\tPointers: %u\n", pointers);
R
Robert Olsson 已提交
2114

2115
	bytes += sizeof(struct node *) * pointers;
2116 2117
	seq_printf(seq, "Null ptrs: %u\n", stat->nullpointers);
	seq_printf(seq, "Total size: %u  kB\n", (bytes + 1023) / 1024);
2118
}
R
Robert Olsson 已提交
2119

2120
#ifdef CONFIG_IP_FIB_TRIE_STATS
2121 2122 2123 2124 2125 2126 2127 2128 2129 2130
static void trie_show_usage(struct seq_file *seq,
			    const struct trie_use_stats *stats)
{
	seq_printf(seq, "\nCounters:\n---------\n");
	seq_printf(seq,"gets = %u\n", stats->gets);
	seq_printf(seq,"backtracks = %u\n", stats->backtrack);
	seq_printf(seq,"semantic match passed = %u\n", stats->semantic_match_passed);
	seq_printf(seq,"semantic match miss = %u\n", stats->semantic_match_miss);
	seq_printf(seq,"null node hit= %u\n", stats->null_node_hit);
	seq_printf(seq,"skipped node resize = %u\n\n", stats->resize_node_skipped);
2131
}
2132 2133
#endif /*  CONFIG_IP_FIB_TRIE_STATS */

2134

2135 2136
static int fib_triestat_seq_show(struct seq_file *seq, void *v)
{
2137
	struct net *net = (struct net *)seq->private;
2138
	struct trie *trie_local, *trie_main;
2139
	struct trie_stat *stat;
2140 2141 2142
	struct fib_table *tb;

	trie_local = NULL;
2143
	tb = fib_get_table(net, RT_TABLE_LOCAL);
2144 2145 2146 2147
	if (tb)
		trie_local = (struct trie *) tb->tb_data;

	trie_main = NULL;
2148
	tb = fib_get_table(net, RT_TABLE_MAIN);
2149 2150 2151
	if (tb)
		trie_main = (struct trie *) tb->tb_data;

O
Olof Johansson 已提交
2152

2153 2154 2155
	stat = kmalloc(sizeof(*stat), GFP_KERNEL);
	if (!stat)
		return -ENOMEM;
O
Olof Johansson 已提交
2156

2157 2158
	seq_printf(seq, "Basic info: size of leaf: %Zd bytes, size of tnode: %Zd bytes.\n",
		   sizeof(struct leaf), sizeof(struct tnode));
O
Olof Johansson 已提交
2159

2160 2161 2162 2163
	if (trie_local) {
		seq_printf(seq, "Local:\n");
		trie_collect_stats(trie_local, stat);
		trie_show_stats(seq, stat);
2164 2165 2166
#ifdef CONFIG_IP_FIB_TRIE_STATS
		trie_show_usage(seq, &trie_local->stats);
#endif
2167
	}
O
Olof Johansson 已提交
2168

2169 2170 2171 2172
	if (trie_main) {
		seq_printf(seq, "Main:\n");
		trie_collect_stats(trie_main, stat);
		trie_show_stats(seq, stat);
2173 2174 2175
#ifdef CONFIG_IP_FIB_TRIE_STATS
		trie_show_usage(seq, &trie_main->stats);
#endif
2176
	}
2177
	kfree(stat);
2178

2179
	return 0;
2180 2181
}

2182
static int fib_triestat_seq_open(struct inode *inode, struct file *file)
2183
{
2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202
	int err;
	struct net *net;

	net = get_proc_net(inode);
	if (net == NULL)
		return -ENXIO;
	err = single_open(file, fib_triestat_seq_show, net);
	if (err < 0) {
		put_net(net);
		return err;
	}
	return 0;
}

static int fib_triestat_seq_release(struct inode *ino, struct file *f)
{
	struct seq_file *seq = f->private_data;
	put_net(seq->private);
	return single_release(ino, f);
2203 2204
}

2205
static const struct file_operations fib_triestat_fops = {
2206 2207 2208 2209
	.owner	= THIS_MODULE,
	.open	= fib_triestat_seq_open,
	.read	= seq_read,
	.llseek	= seq_lseek,
2210
	.release = fib_triestat_seq_release,
2211 2212 2213 2214
};

static struct node *fib_trie_get_idx(struct fib_trie_iter *iter,
				      loff_t pos)
2215
{
2216 2217 2218
	loff_t idx = 0;
	struct node *n;

2219
	for (n = fib_trie_get_first(iter, iter->trie_local);
2220 2221 2222 2223 2224
	     n; ++idx, n = fib_trie_get_next(iter)) {
		if (pos == idx)
			return n;
	}

2225
	for (n = fib_trie_get_first(iter, iter->trie_main);
2226 2227 2228 2229
	     n; ++idx, n = fib_trie_get_next(iter)) {
		if (pos == idx)
			return n;
	}
2230 2231 2232
	return NULL;
}

2233
static void *fib_trie_seq_start(struct seq_file *seq, loff_t *pos)
2234
	__acquires(RCU)
2235
{
2236 2237 2238 2239
	struct fib_trie_iter *iter = seq->private;
	struct fib_table *tb;

	if (!iter->trie_local) {
2240
		tb = fib_get_table(iter->p.net, RT_TABLE_LOCAL);
2241 2242 2243 2244
		if (tb)
			iter->trie_local = (struct trie *) tb->tb_data;
	}
	if (!iter->trie_main) {
2245
		tb = fib_get_table(iter->p.net, RT_TABLE_MAIN);
2246 2247 2248
		if (tb)
			iter->trie_main = (struct trie *) tb->tb_data;
	}
2249 2250
	rcu_read_lock();
	if (*pos == 0)
O
Olof Johansson 已提交
2251
		return SEQ_START_TOKEN;
2252
	return fib_trie_get_idx(iter, *pos - 1);
2253 2254
}

2255
static void *fib_trie_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2256
{
2257 2258 2259
	struct fib_trie_iter *iter = seq->private;
	void *l = v;

2260
	++*pos;
O
Olof Johansson 已提交
2261
	if (v == SEQ_START_TOKEN)
2262
		return fib_trie_get_idx(iter, 0);
2263

2264 2265 2266 2267
	v = fib_trie_get_next(iter);
	BUG_ON(v == l);
	if (v)
		return v;
2268

2269
	/* continue scan in next trie */
2270 2271
	if (iter->trie == iter->trie_local)
		return fib_trie_get_first(iter, iter->trie_main);
2272

2273 2274
	return NULL;
}
2275

2276
static void fib_trie_seq_stop(struct seq_file *seq, void *v)
2277
	__releases(RCU)
2278
{
2279 2280
	rcu_read_unlock();
}
O
Olof Johansson 已提交
2281

2282 2283 2284 2285
static void seq_indent(struct seq_file *seq, int n)
{
	while (n-- > 0) seq_puts(seq, "   ");
}
2286

2287
static inline const char *rtn_scope(char *buf, size_t len, enum rt_scope_t s)
2288
{
S
Stephen Hemminger 已提交
2289
	switch (s) {
2290 2291 2292 2293 2294 2295
	case RT_SCOPE_UNIVERSE: return "universe";
	case RT_SCOPE_SITE:	return "site";
	case RT_SCOPE_LINK:	return "link";
	case RT_SCOPE_HOST:	return "host";
	case RT_SCOPE_NOWHERE:	return "nowhere";
	default:
2296
		snprintf(buf, len, "scope=%d", s);
2297 2298 2299
		return buf;
	}
}
2300

2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311 2312 2313 2314
static const char *rtn_type_names[__RTN_MAX] = {
	[RTN_UNSPEC] = "UNSPEC",
	[RTN_UNICAST] = "UNICAST",
	[RTN_LOCAL] = "LOCAL",
	[RTN_BROADCAST] = "BROADCAST",
	[RTN_ANYCAST] = "ANYCAST",
	[RTN_MULTICAST] = "MULTICAST",
	[RTN_BLACKHOLE] = "BLACKHOLE",
	[RTN_UNREACHABLE] = "UNREACHABLE",
	[RTN_PROHIBIT] = "PROHIBIT",
	[RTN_THROW] = "THROW",
	[RTN_NAT] = "NAT",
	[RTN_XRESOLVE] = "XRESOLVE",
};
2315

2316
static inline const char *rtn_type(char *buf, size_t len, unsigned t)
2317 2318 2319
{
	if (t < __RTN_MAX && rtn_type_names[t])
		return rtn_type_names[t];
2320
	snprintf(buf, len, "type %u", t);
2321
	return buf;
2322 2323
}

2324 2325
/* Pretty print the trie */
static int fib_trie_seq_show(struct seq_file *seq, void *v)
2326
{
2327 2328
	const struct fib_trie_iter *iter = seq->private;
	struct node *n = v;
2329

2330 2331
	if (v == SEQ_START_TOKEN)
		return 0;
2332

S
Stephen Hemminger 已提交
2333
	if (!node_parent(n)) {
2334
		if (iter->trie == iter->trie_local)
2335 2336 2337 2338 2339
			seq_puts(seq, "<local>:\n");
		else
			seq_puts(seq, "<main>:\n");
	}

2340 2341
	if (IS_TNODE(n)) {
		struct tnode *tn = (struct tnode *) n;
S
Stephen Hemminger 已提交
2342
		__be32 prf = htonl(mask_pfx(tn->key, tn->pos));
O
Olof Johansson 已提交
2343

2344 2345
		seq_indent(seq, iter->depth-1);
		seq_printf(seq, "  +-- %d.%d.%d.%d/%d %d %d %d\n",
2346
			   NIPQUAD(prf), tn->pos, tn->bits, tn->full_children,
2347
			   tn->empty_children);
2348

2349 2350 2351
	} else {
		struct leaf *l = (struct leaf *) n;
		int i;
A
Al Viro 已提交
2352
		__be32 val = htonl(l->key);
2353 2354 2355 2356

		seq_indent(seq, iter->depth);
		seq_printf(seq, "  |-- %d.%d.%d.%d\n", NIPQUAD(val));
		for (i = 32; i >= 0; i--) {
R
Robert Olsson 已提交
2357
			struct leaf_info *li = find_leaf_info(l, i);
2358

2359 2360
			if (li) {
				struct fib_alias *fa;
2361

2362
				list_for_each_entry_rcu(fa, &li->falh, fa_list) {
2363 2364
					char buf1[32], buf2[32];

2365 2366
					seq_indent(seq, iter->depth+1);
					seq_printf(seq, "  /%d %s %s", i,
2367 2368 2369 2370
						   rtn_scope(buf1, sizeof(buf1),
							     fa->fa_scope),
						   rtn_type(buf2, sizeof(buf2),
							     fa->fa_type));
2371 2372 2373 2374 2375 2376 2377
					if (fa->fa_tos)
						seq_printf(seq, "tos =%d\n",
							   fa->fa_tos);
					seq_putc(seq, '\n');
				}
			}
		}
2378
	}
2379

2380 2381 2382
	return 0;
}

2383
static const struct seq_operations fib_trie_seq_ops = {
2384 2385 2386 2387
	.start  = fib_trie_seq_start,
	.next   = fib_trie_seq_next,
	.stop   = fib_trie_seq_stop,
	.show   = fib_trie_seq_show,
2388 2389
};

2390
static int fib_trie_seq_open(struct inode *inode, struct file *file)
2391
{
2392 2393
	return seq_open_net(inode, file, &fib_trie_seq_ops,
			    sizeof(struct fib_trie_iter));
2394 2395
}

2396
static const struct file_operations fib_trie_fops = {
2397 2398 2399 2400
	.owner  = THIS_MODULE,
	.open   = fib_trie_seq_open,
	.read   = seq_read,
	.llseek = seq_lseek,
2401
	.release = seq_release_net,
2402 2403
};

A
Al Viro 已提交
2404
static unsigned fib_flag_trans(int type, __be32 mask, const struct fib_info *fi)
2405
{
2406 2407 2408 2409
	static unsigned type2flags[RTN_MAX + 1] = {
		[7] = RTF_REJECT, [8] = RTF_REJECT,
	};
	unsigned flags = type2flags[type];
2410

2411 2412
	if (fi && fi->fib_nh->nh_gw)
		flags |= RTF_GATEWAY;
A
Al Viro 已提交
2413
	if (mask == htonl(0xFFFFFFFF))
2414 2415 2416
		flags |= RTF_HOST;
	flags |= RTF_UP;
	return flags;
2417 2418
}

2419 2420 2421 2422 2423 2424 2425
/*
 *	This outputs /proc/net/route.
 *	The format of the file is not supposed to be changed
 * 	and needs to be same as fib_hash output to avoid breaking
 *	legacy utilities
 */
static int fib_route_seq_show(struct seq_file *seq, void *v)
2426
{
2427
	const struct fib_trie_iter *iter = seq->private;
2428 2429 2430
	struct leaf *l = v;
	int i;
	char bf[128];
2431

2432 2433 2434 2435 2436 2437
	if (v == SEQ_START_TOKEN) {
		seq_printf(seq, "%-127s\n", "Iface\tDestination\tGateway "
			   "\tFlags\tRefCnt\tUse\tMetric\tMask\t\tMTU"
			   "\tWindow\tIRTT");
		return 0;
	}
2438

2439
	if (iter->trie == iter->trie_local)
2440
		return 0;
2441 2442
	if (IS_TNODE(l))
		return 0;
2443

2444
	for (i=32; i>=0; i--) {
R
Robert Olsson 已提交
2445
		struct leaf_info *li = find_leaf_info(l, i);
2446
		struct fib_alias *fa;
A
Al Viro 已提交
2447
		__be32 mask, prefix;
O
Olof Johansson 已提交
2448

2449 2450
		if (!li)
			continue;
2451

2452 2453
		mask = inet_make_mask(li->plen);
		prefix = htonl(l->key);
2454

2455
		list_for_each_entry_rcu(fa, &li->falh, fa_list) {
2456
			const struct fib_info *fi = fa->fa_info;
2457
			unsigned flags = fib_flag_trans(fa->fa_type, mask, fi);
2458

2459 2460 2461
			if (fa->fa_type == RTN_BROADCAST
			    || fa->fa_type == RTN_MULTICAST)
				continue;
2462

2463 2464 2465 2466 2467 2468 2469 2470 2471 2472 2473 2474 2475 2476 2477 2478
			if (fi)
				snprintf(bf, sizeof(bf),
					 "%s\t%08X\t%08X\t%04X\t%d\t%u\t%d\t%08X\t%d\t%u\t%u",
					 fi->fib_dev ? fi->fib_dev->name : "*",
					 prefix,
					 fi->fib_nh->nh_gw, flags, 0, 0,
					 fi->fib_priority,
					 mask,
					 (fi->fib_advmss ? fi->fib_advmss + 40 : 0),
					 fi->fib_window,
					 fi->fib_rtt >> 3);
			else
				snprintf(bf, sizeof(bf),
					 "*\t%08X\t%08X\t%04X\t%d\t%u\t%d\t%08X\t%d\t%u\t%u",
					 prefix, 0, flags, 0, 0, 0,
					 mask, 0, 0, 0);
2479

2480 2481
			seq_printf(seq, "%-127s\n", bf);
		}
2482 2483 2484 2485 2486
	}

	return 0;
}

2487
static const struct seq_operations fib_route_seq_ops = {
2488 2489 2490 2491
	.start  = fib_trie_seq_start,
	.next   = fib_trie_seq_next,
	.stop   = fib_trie_seq_stop,
	.show   = fib_route_seq_show,
2492 2493
};

2494
static int fib_route_seq_open(struct inode *inode, struct file *file)
2495
{
2496 2497
	return seq_open_net(inode, file, &fib_route_seq_ops,
			    sizeof(struct fib_trie_iter));
2498 2499
}

2500
static const struct file_operations fib_route_fops = {
2501 2502 2503 2504
	.owner  = THIS_MODULE,
	.open   = fib_route_seq_open,
	.read   = seq_read,
	.llseek = seq_lseek,
2505
	.release = seq_release_net,
2506 2507
};

2508
int __net_init fib_proc_init(struct net *net)
2509
{
2510
	if (!proc_net_fops_create(net, "fib_trie", S_IRUGO, &fib_trie_fops))
2511 2512
		goto out1;

2513 2514
	if (!proc_net_fops_create(net, "fib_triestat", S_IRUGO,
				  &fib_triestat_fops))
2515 2516
		goto out2;

2517
	if (!proc_net_fops_create(net, "route", S_IRUGO, &fib_route_fops))
2518 2519
		goto out3;

2520
	return 0;
2521 2522

out3:
2523
	proc_net_remove(net, "fib_triestat");
2524
out2:
2525
	proc_net_remove(net, "fib_trie");
2526 2527
out1:
	return -ENOMEM;
2528 2529
}

2530
void __net_exit fib_proc_exit(struct net *net)
2531
{
2532 2533 2534
	proc_net_remove(net, "fib_trie");
	proc_net_remove(net, "fib_triestat");
	proc_net_remove(net, "route");
2535 2536 2537
}

#endif /* CONFIG_PROC_FS */