fib_trie.c 58.9 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 57 58 59 60 61 62 63 64 65 66

#include <asm/uaccess.h>
#include <asm/system.h>
#include <asm/bitops.h>
#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 76 77 78 79 80 81 82 83 84
#include <linux/skbuff.h>
#include <linux/netlink.h>
#include <linux/init.h>
#include <linux/list.h>
#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"

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

#define KEYLENGTH (8*sizeof(t_key))
#define MASK_PFX(k, l) (((l)==0)?0:(k >> (KEYLENGTH-l)) << (KEYLENGTH-l))
#define TKEY_GET_MASK(offset, bits) (((bits)==0)?0:((t_key)(-1) << (KEYLENGTH - bits) >> offset))

typedef unsigned int t_key;

#define T_TNODE 0
#define T_LEAF  1
#define NODE_TYPE_MASK	0x1UL
O
Olof Johansson 已提交
96
#define NODE_PARENT(node) \
R
Robert Olsson 已提交
97 98 99 100 101 102 103
	((struct tnode *)rcu_dereference(((node)->parent & ~NODE_TYPE_MASK)))

#define NODE_TYPE(node) ((node)->parent & NODE_TYPE_MASK)

#define NODE_SET_PARENT(node, ptr)		\
	rcu_assign_pointer((node)->parent,	\
			   ((unsigned long)(ptr)) | NODE_TYPE(node))
O
Olof Johansson 已提交
104 105 106

#define IS_TNODE(n) (!(n->parent & T_LEAF))
#define IS_LEAF(n) (n->parent & T_LEAF)
107 108

struct node {
O
Olof Johansson 已提交
109 110
	t_key key;
	unsigned long parent;
111 112 113
};

struct leaf {
O
Olof Johansson 已提交
114 115
	t_key key;
	unsigned long parent;
116
	struct hlist_head list;
R
Robert Olsson 已提交
117
	struct rcu_head rcu;
118 119 120 121
};

struct leaf_info {
	struct hlist_node hlist;
R
Robert Olsson 已提交
122
	struct rcu_head rcu;
123 124 125 126 127
	int plen;
	struct list_head falh;
};

struct tnode {
O
Olof Johansson 已提交
128 129 130 131 132 133
	t_key key;
	unsigned long parent;
	unsigned short pos:5;		/* 2log(KEYLENGTH) bits needed */
	unsigned short bits:5;		/* 2log(KEYLENGTH) bits needed */
	unsigned short full_children;	/* KEYLENGTH bits needed */
	unsigned short empty_children;	/* KEYLENGTH bits needed */
R
Robert Olsson 已提交
134
	struct rcu_head rcu;
O
Olof Johansson 已提交
135
	struct node *child[0];
136 137 138 139 140 141 142 143 144
};

#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;
145
	unsigned int resize_node_skipped;
146 147 148 149 150 151 152 153 154
};
#endif

struct trie_stat {
	unsigned int totdepth;
	unsigned int maxdepth;
	unsigned int tnodes;
	unsigned int leaves;
	unsigned int nullpointers;
R
Robert Olsson 已提交
155
	unsigned int nodesizes[MAX_STAT_DEPTH];
156
};
157 158

struct trie {
O
Olof Johansson 已提交
159
	struct node *trie;
160 161 162
#ifdef CONFIG_IP_FIB_TRIE_STATS
	struct trie_use_stats stats;
#endif
O
Olof Johansson 已提交
163
	int size;
164 165 166 167 168 169
	unsigned int revision;
};

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);
170 171
static struct tnode *inflate(struct trie *t, struct tnode *tn);
static struct tnode *halve(struct trie *t, struct tnode *tn);
172 173
static void tnode_free(struct tnode *tn);

174
static struct kmem_cache *fn_alias_kmem __read_mostly;
175 176
static struct trie *trie_local = NULL, *trie_main = NULL;

R
Robert Olsson 已提交
177 178 179

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

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

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

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

static inline t_key tkey_extract_bits(t_key a, int offset, int bits)
{
O
Olof Johansson 已提交
194
	if (offset < KEYLENGTH)
195
		return ((t_key)(a << offset)) >> (KEYLENGTH - bits);
O
Olof Johansson 已提交
196
	else
197 198 199 200 201
		return 0;
}

static inline int tkey_equals(t_key a, t_key b)
{
202
	return a == b;
203 204 205 206
}

static inline int tkey_sub_equals(t_key a, int offset, int bits, t_key b)
{
207 208
	if (bits == 0 || offset >= KEYLENGTH)
		return 1;
O
Olof Johansson 已提交
209 210
	bits = bits > KEYLENGTH ? KEYLENGTH : bits;
	return ((a ^ b) << offset) >> (KEYLENGTH - bits) == 0;
211
}
212 213 214 215 216 217

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

218 219 220
	if (!diff)
		return 0;
	while ((diff << i) >> (KEYLENGTH-1) == 0)
221 222 223 224 225
		i++;
	return i;
}

/*
226 227
  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
228 229 230 231
  all of the bits in that key are significant.

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

232 233 234 235 236
  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
237 238
  correct key path.

239 240 241 242 243
  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
244 245
  call to tkey_sub_equals() in trie_insert().

246
  if n is an internal node - a 'tnode' here, the various parts of its key
247 248
  have many different meanings.

249
  Example:
250 251 252
  _________________________________________________________________
  | i | i | i | i | i | i | i | N | N | N | S | S | S | S | S | C |
  -----------------------------------------------------------------
253
    0   1   2   3   4   5   6   7   8   9  10  11  12  13  14  15
254 255 256 257 258 259 260 261 262

  _________________________________________________________________
  | 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 已提交
263
  n->bits = 4
264

265 266
  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
267 268 269
  not use them for anything.

  The bits from (tp->pos) to (tp->pos + tp->bits - 1) - "N", above - are the
270
  index into the parent's child array. That is, they will be used to find
271 272 273 274 275
  'n' among tp's children.

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

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

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

282

283 284 285 286 287
  The rest of the bits, from (n->pos + n->bits) onward, are completely unknown
  at this point.

*/

S
Stephen Hemminger 已提交
288
static inline void check_tnode(const struct tnode *tn)
289
{
S
Stephen Hemminger 已提交
290
	WARN_ON(tn && tn->pos+tn->bits > 32);
291 292 293 294
}

static int halve_threshold = 25;
static int inflate_threshold = 50;
295 296
static int halve_threshold_root = 8;
static int inflate_threshold_root = 15;
297

R
Robert Olsson 已提交
298 299

static void __alias_free_mem(struct rcu_head *head)
300
{
R
Robert Olsson 已提交
301 302
	struct fib_alias *fa = container_of(head, struct fib_alias, rcu);
	kmem_cache_free(fn_alias_kmem, fa);
303 304
}

R
Robert Olsson 已提交
305
static inline void alias_free_mem_rcu(struct fib_alias *fa)
306
{
R
Robert Olsson 已提交
307 308
	call_rcu(&fa->rcu, __alias_free_mem);
}
O
Olof Johansson 已提交
309

R
Robert Olsson 已提交
310 311 312 313
static void __leaf_free_rcu(struct rcu_head *head)
{
	kfree(container_of(head, struct leaf, rcu));
}
O
Olof Johansson 已提交
314

R
Robert Olsson 已提交
315
static void __leaf_info_free_rcu(struct rcu_head *head)
316
{
R
Robert Olsson 已提交
317
	kfree(container_of(head, struct leaf_info, rcu));
318 319
}

R
Robert Olsson 已提交
320
static inline void free_leaf_info(struct leaf_info *leaf)
321
{
R
Robert Olsson 已提交
322
	call_rcu(&leaf->rcu, __leaf_info_free_rcu);
323 324
}

325 326
static struct tnode *tnode_alloc(unsigned int size)
{
R
Robert Olsson 已提交
327 328 329 330 331 332 333 334 335 336
	struct page *pages;

	if (size <= PAGE_SIZE)
		return kcalloc(size, 1, GFP_KERNEL);

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

	return page_address(pages);
337 338
}

R
Robert Olsson 已提交
339
static void __tnode_free_rcu(struct rcu_head *head)
340
{
R
Robert Olsson 已提交
341
	struct tnode *tn = container_of(head, struct tnode, rcu);
342
	unsigned int size = sizeof(struct tnode) +
R
Robert Olsson 已提交
343
		(1 << tn->bits) * sizeof(struct node *);
344 345 346 347 348 349 350

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

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

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;
}

380 381 382 383
static struct tnode* tnode_new(t_key key, int pos, int bits)
{
	int nchildren = 1<<bits;
	int sz = sizeof(struct tnode) + nchildren * sizeof(struct node *);
384
	struct tnode *tn = tnode_alloc(sz);
385

O
Olof Johansson 已提交
386
	if (tn) {
387
		memset(tn, 0, sz);
R
Robert Olsson 已提交
388
		tn->parent = T_TNODE;
389 390 391 392 393 394
		tn->pos = pos;
		tn->bits = bits;
		tn->key = key;
		tn->full_children = 0;
		tn->empty_children = 1<<bits;
	}
395

S
Stephen Hemminger 已提交
396 397
	pr_debug("AT %p s=%u %u\n", tn, (unsigned int) sizeof(struct tnode),
		 (unsigned int) (sizeof(struct node) * 1<<bits));
398 399 400 401 402 403 404 405
	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 已提交
406
static inline int tnode_full(const struct tnode *tn, const struct node *n)
407
{
408
	if (n == NULL || IS_LEAF(n))
409 410 411 412 413
		return 0;

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

414
static inline void put_child(struct trie *t, struct tnode *tn, int i, struct node *n)
415 416 417 418
{
	tnode_put_child_reorg(tn, i, n, -1);
}

419
 /*
420 421 422 423
  * Add a child at position i overwriting the old value.
  * Update the value of full_children and empty_children.
  */

424
static void tnode_put_child_reorg(struct tnode *tn, int i, struct node *n, int wasfull)
425
{
R
Robert Olsson 已提交
426
	struct node *chi = tn->child[i];
427 428
	int isfull;

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

431 432 433 434 435 436

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

438
	/* update fullChildren */
O
Olof Johansson 已提交
439
	if (wasfull == -1)
440 441 442
		wasfull = tnode_full(tn, chi);

	isfull = tnode_full(tn, n);
443
	if (wasfull && !isfull)
444
		tn->full_children--;
445
	else if (!wasfull && isfull)
446
		tn->full_children++;
O
Olof Johansson 已提交
447

448 449
	if (n)
		NODE_SET_PARENT(n, tn);
450

R
Robert Olsson 已提交
451
	rcu_assign_pointer(tn->child[i], n);
452 453
}

454
static struct node *resize(struct trie *t, struct tnode *tn)
455 456
{
	int i;
457
	int err = 0;
458
	struct tnode *old_tn;
459 460
	int inflate_threshold_use;
	int halve_threshold_use;
R
Robert Olsson 已提交
461
	int max_resize;
462

463
	if (!tn)
464 465
		return NULL;

S
Stephen Hemminger 已提交
466 467
	pr_debug("In tnode_resize %p inflate_threshold=%d threshold=%d\n",
		 tn, inflate_threshold, halve_threshold);
468 469 470 471 472 473 474 475 476

	/* 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 已提交
477
			struct node *n;
478

O
Olof Johansson 已提交
479
			n = tn->child[i];
R
Robert Olsson 已提交
480
			if (!n)
O
Olof Johansson 已提交
481 482 483
				continue;

			/* compress one level */
R
Robert Olsson 已提交
484
			NODE_SET_PARENT(n, NULL);
O
Olof Johansson 已提交
485 486
			tnode_free(tn);
			return n;
487
		}
488
	/*
489 490 491 492 493
	 * Double as long as the resulting node has a number of
	 * nonempty nodes that are above the threshold.
	 */

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

	check_tnode(tn);
553

554 555
	/* Keep root node larger  */

S
Stephen Hemminger 已提交
556
	if (!tn->parent)
557
		inflate_threshold_use = inflate_threshold_root;
558
	else
559 560
		inflate_threshold_use = inflate_threshold;

561
	err = 0;
R
Robert Olsson 已提交
562 563
	max_resize = 10;
	while ((tn->full_children > 0 &&  max_resize-- &&
564
	       50 * (tn->full_children + tnode_child_length(tn) - tn->empty_children) >=
565
				inflate_threshold_use * tnode_child_length(tn))) {
566

567 568 569 570
		old_tn = tn;
		tn = inflate(t, tn);
		if (IS_ERR(tn)) {
			tn = old_tn;
571 572 573 574 575
#ifdef CONFIG_IP_FIB_TRIE_STATS
			t->stats.resize_node_skipped++;
#endif
			break;
		}
576 577
	}

R
Robert Olsson 已提交
578 579 580 581 582 583 584 585 586
	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);
	}

587 588 589 590 591 592
	check_tnode(tn);

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

594 595 596

	/* Keep root node larger  */

S
Stephen Hemminger 已提交
597
	if (!tn->parent)
598
		halve_threshold_use = halve_threshold_root;
599
	else
600 601
		halve_threshold_use = halve_threshold;

602
	err = 0;
R
Robert Olsson 已提交
603 604
	max_resize = 10;
	while (tn->bits > 1 &&  max_resize-- &&
605
	       100 * (tnode_child_length(tn) - tn->empty_children) <
606
	       halve_threshold_use * tnode_child_length(tn)) {
607

608 609 610 611
		old_tn = tn;
		tn = halve(t, tn);
		if (IS_ERR(tn)) {
			tn = old_tn;
612 613 614 615 616 617
#ifdef CONFIG_IP_FIB_TRIE_STATS
			t->stats.resize_node_skipped++;
#endif
			break;
		}
	}
618

R
Robert Olsson 已提交
619 620 621 622 623 624 625 626
	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);
	}
627

628 629 630
	/* 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 已提交
631
			struct node *n;
632

O
Olof Johansson 已提交
633
			n = tn->child[i];
R
Robert Olsson 已提交
634
			if (!n)
O
Olof Johansson 已提交
635 636 637 638
				continue;

			/* compress one level */

R
Robert Olsson 已提交
639
			NODE_SET_PARENT(n, NULL);
O
Olof Johansson 已提交
640 641
			tnode_free(tn);
			return n;
642 643 644 645 646
		}

	return (struct node *) tn;
}

647
static struct tnode *inflate(struct trie *t, struct tnode *tn)
648 649 650 651 652 653
{
	struct tnode *inode;
	struct tnode *oldtnode = tn;
	int olen = tnode_child_length(tn);
	int i;

S
Stephen Hemminger 已提交
654
	pr_debug("In inflate\n");
655 656 657

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

S
Stephen Hemminger 已提交
658
	if (!tn)
659
		return ERR_PTR(-ENOMEM);
660 661

	/*
662 663 664
	 * 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
665 666
	 * of tnode is ignored.
	 */
O
Olof Johansson 已提交
667 668

	for (i = 0; i < olen; i++) {
669 670 671 672 673 674 675 676
		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;
			t_key m = TKEY_GET_MASK(inode->pos, 1);
677

678 679
			left = tnode_new(inode->key&(~m), inode->pos + 1,
					 inode->bits - 1);
680 681
			if (!left)
				goto nomem;
O
Olof Johansson 已提交
682

683 684 685
			right = tnode_new(inode->key|m, inode->pos + 1,
					  inode->bits - 1);

686
			if (!right) {
687 688
				tnode_free(left);
				goto nomem;
689
			}
690 691 692 693 694 695

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

O
Olof Johansson 已提交
696
	for (i = 0; i < olen; i++) {
697
		struct node *node = tnode_get_child(oldtnode, i);
O
Olof Johansson 已提交
698 699
		struct tnode *left, *right;
		int size, j;
700

701 702 703 704 705 706
		/* An empty child */
		if (node == NULL)
			continue;

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

707
		if (IS_LEAF(node) || ((struct tnode *) node)->pos >
708
		   tn->pos + tn->bits - 1) {
709
			if (tkey_extract_bits(node->key, oldtnode->pos + oldtnode->bits,
710 711 712 713 714 715 716 717 718 719 720 721 722 723 724
					     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 已提交
725
			continue;
726 727
		}

O
Olof Johansson 已提交
728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745
		/* 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)
		 */
746

O
Olof Johansson 已提交
747 748 749
		/* Use the old key, but set the new significant
		 *   bit to zero.
		 */
750

O
Olof Johansson 已提交
751 752
		left = (struct tnode *) tnode_get_child(tn, 2*i);
		put_child(t, tn, 2*i, NULL);
753

O
Olof Johansson 已提交
754
		BUG_ON(!left);
755

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

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

O
Olof Johansson 已提交
761 762 763 764
		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]);
765
		}
O
Olof Johansson 已提交
766 767 768 769
		put_child(t, tn, 2*i, resize(t, left));
		put_child(t, tn, 2*i+1, resize(t, right));

		tnode_free(inode);
770 771 772
	}
	tnode_free(oldtnode);
	return tn;
773 774 775 776 777
nomem:
	{
		int size = tnode_child_length(tn);
		int j;

S
Stephen Hemminger 已提交
778
		for (j = 0; j < size; j++)
779 780 781 782
			if (tn->child[j])
				tnode_free((struct tnode *)tn->child[j]);

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

784 785
		return ERR_PTR(-ENOMEM);
	}
786 787
}

788
static struct tnode *halve(struct trie *t, struct tnode *tn)
789 790 791 792 793 794
{
	struct tnode *oldtnode = tn;
	struct node *left, *right;
	int i;
	int olen = tnode_child_length(tn);

S
Stephen Hemminger 已提交
795
	pr_debug("In halve\n");
796 797

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

799 800
	if (!tn)
		return ERR_PTR(-ENOMEM);
801 802

	/*
803 804 805
	 * 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
806 807 808
	 * of tnode is ignored.
	 */

O
Olof Johansson 已提交
809
	for (i = 0; i < olen; i += 2) {
810 811
		left = tnode_get_child(oldtnode, i);
		right = tnode_get_child(oldtnode, i+1);
812

813
		/* Two nonempty children */
S
Stephen Hemminger 已提交
814
		if (left && right) {
815
			struct tnode *newn;
S
Stephen Hemminger 已提交
816

817
			newn = tnode_new(left->key, tn->pos + tn->bits, 1);
S
Stephen Hemminger 已提交
818 819

			if (!newn)
820
				goto nomem;
S
Stephen Hemminger 已提交
821

822
			put_child(t, tn, i/2, (struct node *)newn);
823 824 825
		}

	}
826

O
Olof Johansson 已提交
827 828 829
	for (i = 0; i < olen; i += 2) {
		struct tnode *newBinNode;

830 831
		left = tnode_get_child(oldtnode, i);
		right = tnode_get_child(oldtnode, i+1);
832

833 834 835 836 837
		/* 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 已提交
838
			continue;
S
Stephen Hemminger 已提交
839
		}
O
Olof Johansson 已提交
840 841

		if (right == NULL) {
842
			put_child(t, tn, i/2, left);
O
Olof Johansson 已提交
843 844
			continue;
		}
845

846
		/* Two nonempty children */
O
Olof Johansson 已提交
847 848 849 850 851
		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));
852 853 854
	}
	tnode_free(oldtnode);
	return tn;
855 856 857 858 859
nomem:
	{
		int size = tnode_child_length(tn);
		int j;

S
Stephen Hemminger 已提交
860
		for (j = 0; j < size; j++)
861 862 863 864
			if (tn->child[j])
				tnode_free((struct tnode *)tn->child[j]);

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

866 867
		return ERR_PTR(-ENOMEM);
	}
868 869
}

O
Olof Johansson 已提交
870
static void trie_init(struct trie *t)
871
{
O
Olof Johansson 已提交
872 873 874 875
	if (!t)
		return;

	t->size = 0;
R
Robert Olsson 已提交
876
	rcu_assign_pointer(t->trie, NULL);
O
Olof Johansson 已提交
877
	t->revision = 0;
878
#ifdef CONFIG_IP_FIB_TRIE_STATS
O
Olof Johansson 已提交
879
	memset(&t->stats, 0, sizeof(struct trie_use_stats));
880 881 882
#endif
}

R
Robert Olsson 已提交
883
/* readside must use rcu_read_lock currently dump routines
R
Robert Olsson 已提交
884 885
 via get_fa_head and dump */

R
Robert Olsson 已提交
886
static struct leaf_info *find_leaf_info(struct leaf *l, int plen)
887
{
R
Robert Olsson 已提交
888
	struct hlist_head *head = &l->list;
889 890 891
	struct hlist_node *node;
	struct leaf_info *li;

R
Robert Olsson 已提交
892
	hlist_for_each_entry_rcu(li, node, head, hlist)
893
		if (li->plen == plen)
894
			return li;
O
Olof Johansson 已提交
895

896 897 898 899 900
	return NULL;
}

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

O
Olof Johansson 已提交
903 904
	if (!li)
		return NULL;
905

O
Olof Johansson 已提交
906
	return &li->falh;
907 908 909 910
}

static void insert_leaf_info(struct hlist_head *head, struct leaf_info *new)
{
911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927
	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);
	}
928 929
}

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

932 933 934 935 936 937 938 939
static struct leaf *
fib_find_node(struct trie *t, u32 key)
{
	int pos;
	struct tnode *tn;
	struct node *n;

	pos = 0;
R
Robert Olsson 已提交
940
	n = rcu_dereference(t->trie);
941 942 943

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

945
		check_tnode(tn);
O
Olof Johansson 已提交
946

947
		if (tkey_sub_equals(tn->key, pos, tn->pos-pos, key)) {
O
Olof Johansson 已提交
948
			pos = tn->pos + tn->bits;
949
			n = tnode_get_child(tn, tkey_extract_bits(key, tn->pos, tn->bits));
O
Olof Johansson 已提交
950
		} else
951 952 953 954
			break;
	}
	/* Case we have found a leaf. Compare prefixes */

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

958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975
	return NULL;
}

static struct node *trie_rebalance(struct trie *t, struct tnode *tn)
{
	int wasfull;
	t_key cindex, key;
	struct tnode *tp = NULL;

	key = tn->key;

	while (tn != NULL && NODE_PARENT(tn) != NULL) {

		tp = NODE_PARENT(tn);
		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 已提交
976

977
		if (!NODE_PARENT(tn))
978 979 980 981 982
			break;

		tn = NODE_PARENT(tn);
	}
	/* Handle last (top) tnode */
983
	if (IS_TNODE(tn))
984 985 986 987 988
		tn = (struct tnode*) resize(t, (struct tnode *)tn);

	return (struct node*) tn;
}

R
Robert Olsson 已提交
989 990
/* only used from updater-side */

991 992
static  struct list_head *
fib_insert_node(struct trie *t, int *err, u32 key, int plen)
993 994 995 996 997 998
{
	int pos, newpos;
	struct tnode *tp = NULL, *tn = NULL;
	struct node *n;
	struct leaf *l;
	int missbit;
999
	struct list_head *fa_head = NULL;
1000 1001 1002 1003
	struct leaf_info *li;
	t_key cindex;

	pos = 0;
1004
	n = t->trie;
1005

1006 1007
	/* 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,
1008
	 * and we should just put our new leaf in that.
1009 1010
	 * 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
1011 1012
	 * not be the parent's 'pos'+'bits'!
	 *
1013
	 * If it does match the current key, get pos/bits from it, extract
1014 1015 1016 1017
	 * 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.
	 *
1018 1019 1020
	 * 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.
1021 1022 1023 1024 1025
	 * 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 已提交
1026

1027
		check_tnode(tn);
O
Olof Johansson 已提交
1028

1029
		if (tkey_sub_equals(tn->key, pos, tn->pos-pos, key)) {
1030
			tp = tn;
O
Olof Johansson 已提交
1031
			pos = tn->pos + tn->bits;
1032 1033
			n = tnode_get_child(tn, tkey_extract_bits(key, tn->pos, tn->bits));

S
Stephen Hemminger 已提交
1034
			BUG_ON(n && NODE_PARENT(n) != tn);
O
Olof Johansson 已提交
1035
		} else
1036 1037 1038 1039 1040 1041
			break;
	}

	/*
	 * n  ----> NULL, LEAF or TNODE
	 *
1042
	 * tp is n's (parent) ----> NULL or TNODE
1043 1044
	 */

O
Olof Johansson 已提交
1045
	BUG_ON(tp && IS_LEAF(tp));
1046 1047 1048

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

1049
	if (n != NULL && IS_LEAF(n) && tkey_equals(key, n->key)) {
O
Olof Johansson 已提交
1050 1051
		struct leaf *l = (struct leaf *) n;

1052
		li = leaf_info_new(plen);
O
Olof Johansson 已提交
1053

1054
		if (!li) {
1055 1056 1057
			*err = -ENOMEM;
			goto err;
		}
1058 1059 1060 1061 1062 1063 1064 1065

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

1066
	if (!l) {
1067 1068 1069
		*err = -ENOMEM;
		goto err;
	}
1070 1071 1072 1073

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

1074
	if (!li) {
1075 1076 1077 1078
		tnode_free((struct tnode *) l);
		*err = -ENOMEM;
		goto err;
	}
1079 1080 1081 1082 1083

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

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

		NODE_SET_PARENT(l, tp);

O
Olof Johansson 已提交
1088 1089 1090 1091
		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. */
1092 1093
		/*
		 *  Add a new tnode here
1094 1095 1096 1097
		 *  first tnode need some special handling
		 */

		if (tp)
O
Olof Johansson 已提交
1098
			pos = tp->pos+tp->bits;
1099
		else
O
Olof Johansson 已提交
1100 1101
			pos = 0;

1102
		if (n) {
1103 1104
			newpos = tkey_mismatch(key, pos, n->key);
			tn = tnode_new(n->key, newpos, 1);
O
Olof Johansson 已提交
1105
		} else {
1106
			newpos = 0;
1107
			tn = tnode_new(key, newpos, 1); /* First tnode */
1108 1109
		}

1110
		if (!tn) {
1111 1112 1113 1114
			free_leaf_info(li);
			tnode_free((struct tnode *) l);
			*err = -ENOMEM;
			goto err;
O
Olof Johansson 已提交
1115 1116
		}

1117 1118
		NODE_SET_PARENT(tn, tp);

O
Olof Johansson 已提交
1119
		missbit = tkey_extract_bits(key, newpos, 1);
1120 1121 1122
		put_child(t, tn, missbit, (struct node *)l);
		put_child(t, tn, 1-missbit, n);

1123
		if (tp) {
1124 1125
			cindex = tkey_extract_bits(key, tp->pos, tp->bits);
			put_child(t, (struct tnode *)tp, cindex, (struct node *)tn);
O
Olof Johansson 已提交
1126
		} else {
R
Robert Olsson 已提交
1127
			rcu_assign_pointer(t->trie, (struct node *)tn); /* First tnode */
1128 1129 1130
			tp = tn;
		}
	}
O
Olof Johansson 已提交
1131 1132

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

1136
	/* Rebalance the trie */
R
Robert Olsson 已提交
1137 1138

	rcu_assign_pointer(t->trie, trie_rebalance(t, tp));
1139 1140
done:
	t->revision++;
O
Olof Johansson 已提交
1141
err:
1142 1143 1144
	return fa_head;
}

1145 1146 1147
/*
 * Caller must hold RTNL.
 */
1148
static int fn_trie_insert(struct fib_table *tb, struct fib_config *cfg)
1149 1150 1151
{
	struct trie *t = (struct trie *) tb->tb_data;
	struct fib_alias *fa, *new_fa;
1152
	struct list_head *fa_head = NULL;
1153
	struct fib_info *fi;
1154 1155
	int plen = cfg->fc_dst_len;
	u8 tos = cfg->fc_tos;
1156 1157 1158 1159 1160 1161 1162
	u32 key, mask;
	int err;
	struct leaf *l;

	if (plen > 32)
		return -EINVAL;

1163
	key = ntohl(cfg->fc_dst);
1164

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

O
Olof Johansson 已提交
1167
	mask = ntohl(inet_make_mask(plen));
1168

1169
	if (key & ~mask)
1170 1171 1172 1173
		return -EINVAL;

	key = key & mask;

1174 1175 1176
	fi = fib_create_info(cfg);
	if (IS_ERR(fi)) {
		err = PTR_ERR(fi);
1177
		goto err;
1178
	}
1179 1180

	l = fib_find_node(t, key);
1181
	fa = NULL;
1182

1183
	if (l) {
1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198
		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 已提交
1199
	if (fa && fa->fa_info->fib_priority == fi->fib_priority) {
1200 1201 1202
		struct fib_alias *fa_orig;

		err = -EEXIST;
1203
		if (cfg->fc_nlflags & NLM_F_EXCL)
1204 1205
			goto out;

1206
		if (cfg->fc_nlflags & NLM_F_REPLACE) {
1207 1208 1209
			struct fib_info *fi_drop;
			u8 state;

R
Robert Olsson 已提交
1210
			err = -ENOBUFS;
1211
			new_fa = kmem_cache_alloc(fn_alias_kmem, GFP_KERNEL);
R
Robert Olsson 已提交
1212 1213
			if (new_fa == NULL)
				goto out;
1214 1215

			fi_drop = fa->fa_info;
R
Robert Olsson 已提交
1216 1217
			new_fa->fa_tos = fa->fa_tos;
			new_fa->fa_info = fi;
1218 1219
			new_fa->fa_type = cfg->fc_type;
			new_fa->fa_scope = cfg->fc_scope;
1220
			state = fa->fa_state;
R
Robert Olsson 已提交
1221
			new_fa->fa_state &= ~FA_S_ACCESSED;
1222

R
Robert Olsson 已提交
1223 1224
			list_replace_rcu(&fa->fa_list, &new_fa->fa_list);
			alias_free_mem_rcu(fa);
1225 1226 1227

			fib_release_info(fi_drop);
			if (state & FA_S_ACCESSED)
O
Olof Johansson 已提交
1228
				rt_cache_flush(-1);
1229

O
Olof Johansson 已提交
1230
			goto succeeded;
1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241
		}
		/* 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;
1242 1243
			if (fa->fa_type == cfg->fc_type &&
			    fa->fa_scope == cfg->fc_scope &&
1244 1245 1246 1247
			    fa->fa_info == fi) {
				goto out;
			}
		}
1248
		if (!(cfg->fc_nlflags & NLM_F_APPEND))
1249 1250 1251
			fa = fa_orig;
	}
	err = -ENOENT;
1252
	if (!(cfg->fc_nlflags & NLM_F_CREATE))
1253 1254 1255
		goto out;

	err = -ENOBUFS;
1256
	new_fa = kmem_cache_alloc(fn_alias_kmem, GFP_KERNEL);
1257 1258 1259 1260 1261
	if (new_fa == NULL)
		goto out;

	new_fa->fa_info = fi;
	new_fa->fa_tos = tos;
1262 1263
	new_fa->fa_type = cfg->fc_type;
	new_fa->fa_scope = cfg->fc_scope;
1264 1265 1266 1267 1268
	new_fa->fa_state = 0;
	/*
	 * Insert new entry to the list.
	 */

1269
	if (!fa_head) {
1270
		err = 0;
1271
		fa_head = fib_insert_node(t, &err, key, plen);
1272
		if (err)
1273 1274
			goto out_free_new_fa;
	}
1275

R
Robert Olsson 已提交
1276 1277
	list_add_tail_rcu(&new_fa->fa_list,
			  (fa ? &fa->fa_list : fa_head));
1278 1279

	rt_cache_flush(-1);
1280 1281
	rtmsg_fib(RTM_NEWROUTE, htonl(key), new_fa, plen, tb->tb_id,
		  &cfg->fc_nlinfo);
1282 1283
succeeded:
	return 0;
1284 1285 1286

out_free_new_fa:
	kmem_cache_free(fn_alias_kmem, new_fa);
1287 1288
out:
	fib_release_info(fi);
O
Olof Johansson 已提交
1289
err:
1290 1291 1292
	return err;
}

R
Robert Olsson 已提交
1293

R
Robert Olsson 已提交
1294
/* should be called with rcu_read_lock */
S
Stephen Hemminger 已提交
1295 1296
static inline int check_leaf(struct trie *t, struct leaf *l,
			     t_key key, int *plen, const struct flowi *flp,
1297
			     struct fib_result *res)
1298
{
1299
	int err, i;
1300
	__be32 mask;
1301 1302 1303
	struct leaf_info *li;
	struct hlist_head *hhead = &l->list;
	struct hlist_node *node;
1304

R
Robert Olsson 已提交
1305
	hlist_for_each_entry_rcu(li, node, hhead, hlist) {
1306
		i = li->plen;
1307 1308
		mask = inet_make_mask(i);
		if (l->key != (key & ntohl(mask)))
1309 1310
			continue;

1311
		if ((err = fib_semantic_match(&li->falh, flp, res, htonl(l->key), mask, i)) <= 0) {
1312 1313 1314 1315
			*plen = i;
#ifdef CONFIG_IP_FIB_TRIE_STATS
			t->stats.semantic_match_passed++;
#endif
1316
			return err;
1317 1318 1319 1320 1321
		}
#ifdef CONFIG_IP_FIB_TRIE_STATS
		t->stats.semantic_match_miss++;
#endif
	}
1322
	return 1;
1323 1324 1325 1326 1327 1328 1329 1330 1331 1332
}

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 已提交
1333
	t_key key = ntohl(flp->fl4_dst);
1334 1335 1336
	int chopped_off;
	t_key cindex = 0;
	int current_prefix_length = KEYLENGTH;
O
Olof Johansson 已提交
1337 1338 1339 1340
	struct tnode *cn;
	t_key node_prefix, key_prefix, pref_mismatch;
	int mp;

R
Robert Olsson 已提交
1341
	rcu_read_lock();
O
Olof Johansson 已提交
1342

R
Robert Olsson 已提交
1343
	n = rcu_dereference(t->trie);
1344
	if (!n)
1345 1346 1347 1348 1349 1350 1351 1352
		goto failed;

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

	/* Just a leaf? */
	if (IS_LEAF(n)) {
1353
		if ((ret = check_leaf(t, (struct leaf *)n, key, &plen, flp, res)) <= 0)
1354 1355 1356 1357 1358
			goto found;
		goto failed;
	}
	pn = (struct tnode *) n;
	chopped_off = 0;
1359

O
Olof Johansson 已提交
1360
	while (pn) {
1361 1362 1363
		pos = pn->pos;
		bits = pn->bits;

1364
		if (!chopped_off)
1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375
			cindex = tkey_extract_bits(MASK_PFX(key, current_prefix_length), pos, bits);

		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 已提交
1376 1377 1378 1379 1380 1381 1382
		if (IS_LEAF(n)) {
			if ((ret = check_leaf(t, (struct leaf *)n, key, &plen, flp, res)) <= 0)
				goto found;
			else
				goto backtrace;
		}

1383 1384
#define HL_OPTIMIZE
#ifdef HL_OPTIMIZE
O
Olof Johansson 已提交
1385
		cn = (struct tnode *)n;
1386

O
Olof Johansson 已提交
1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401
		/*
		 * 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].
		 */
1402

O
Olof Johansson 已提交
1403 1404 1405 1406 1407 1408 1409 1410 1411
		/* 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.
		 */
1412

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

O
Olof Johansson 已提交
1415 1416 1417 1418 1419 1420
		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;
		}
1421

O
Olof Johansson 已提交
1422 1423 1424 1425 1426 1427 1428 1429 1430
		/*
		 * 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.
		 */
1431

O
Olof Johansson 已提交
1432 1433 1434 1435 1436 1437 1438 1439
		/* 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.
		 */
1440

O
Olof Johansson 已提交
1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470
		/* 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.
		 */

		node_prefix = MASK_PFX(cn->key, cn->pos);
		key_prefix = MASK_PFX(key, cn->pos);
		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;
1471
		}
O
Olof Johansson 已提交
1472 1473 1474 1475 1476
#endif
		pn = (struct tnode *)n; /* Descend */
		chopped_off = 0;
		continue;

1477 1478 1479 1480
backtrace:
		chopped_off++;

		/* As zero don't change the child key (cindex) */
O
Olof Johansson 已提交
1481
		while ((chopped_off <= pn->bits) && !(cindex & (1<<(chopped_off-1))))
1482 1483 1484 1485 1486
			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 已提交
1487

1488
		/*
1489
		 * Either we do the actual chop off according or if we have
1490 1491 1492
		 * chopped off all bits in this tnode walk up to our parent.
		 */

O
Olof Johansson 已提交
1493
		if (chopped_off <= pn->bits) {
1494
			cindex &= ~(1 << (chopped_off-1));
O
Olof Johansson 已提交
1495
		} else {
1496
			if (NODE_PARENT(pn) == NULL)
1497
				goto failed;
O
Olof Johansson 已提交
1498

1499 1500 1501 1502 1503 1504 1505 1506 1507
			/* Get Child's index */
			cindex = tkey_extract_bits(pn->key, NODE_PARENT(pn)->pos, NODE_PARENT(pn)->bits);
			pn = NODE_PARENT(pn);
			chopped_off = 0;

#ifdef CONFIG_IP_FIB_TRIE_STATS
			t->stats.backtrack++;
#endif
			goto backtrace;
1508
		}
1509 1510
	}
failed:
1511
	ret = 1;
1512
found:
R
Robert Olsson 已提交
1513
	rcu_read_unlock();
1514 1515 1516
	return ret;
}

R
Robert Olsson 已提交
1517
/* only called from updater side */
1518 1519 1520 1521 1522 1523 1524
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 已提交
1525
	pr_debug("entering trie_leaf_remove(%p)\n", n);
1526 1527

	/* Note that in the case skipped bits, those bits are *not* checked!
1528
	 * When we finish this, we will have NULL or a T_LEAF, and the
1529 1530 1531
	 * T_LEAF may or may not match our key.
	 */

O
Olof Johansson 已提交
1532
	while (n != NULL && IS_TNODE(n)) {
1533 1534 1535 1536
		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 已提交
1537
		BUG_ON(n && NODE_PARENT(n) != tn);
O
Olof Johansson 已提交
1538
	}
1539 1540
	l = (struct leaf *) n;

1541
	if (!n || !tkey_equals(l->key, key))
1542
		return 0;
1543 1544 1545 1546

	/*
	 * Key found.
	 * Remove the leaf and rebalance the tree
1547 1548 1549 1550 1551 1552 1553 1554
	 */

	t->revision++;
	t->size--;

	tp = NODE_PARENT(n);
	tnode_free((struct tnode *) n);

1555
	if (tp) {
1556 1557
		cindex = tkey_extract_bits(key, tp->pos, tp->bits);
		put_child(t, (struct tnode *)tp, cindex, NULL);
R
Robert Olsson 已提交
1558
		rcu_assign_pointer(t->trie, trie_rebalance(t, tp));
O
Olof Johansson 已提交
1559
	} else
R
Robert Olsson 已提交
1560
		rcu_assign_pointer(t->trie, NULL);
1561 1562 1563 1564

	return 1;
}

1565 1566 1567
/*
 * Caller must hold RTNL.
 */
1568
static int fn_trie_delete(struct fib_table *tb, struct fib_config *cfg)
1569 1570 1571
{
	struct trie *t = (struct trie *) tb->tb_data;
	u32 key, mask;
1572 1573
	int plen = cfg->fc_dst_len;
	u8 tos = cfg->fc_tos;
1574 1575 1576
	struct fib_alias *fa, *fa_to_delete;
	struct list_head *fa_head;
	struct leaf *l;
O
Olof Johansson 已提交
1577 1578
	struct leaf_info *li;

1579
	if (plen > 32)
1580 1581
		return -EINVAL;

1582
	key = ntohl(cfg->fc_dst);
O
Olof Johansson 已提交
1583
	mask = ntohl(inet_make_mask(plen));
1584

1585
	if (key & ~mask)
1586 1587 1588 1589 1590
		return -EINVAL;

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

1591
	if (!l)
1592 1593 1594 1595 1596 1597 1598 1599
		return -ESRCH;

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

	if (!fa)
		return -ESRCH;

S
Stephen Hemminger 已提交
1600
	pr_debug("Deleting %08x/%d tos=%d t=%p\n", key, plen, tos, t);
1601 1602 1603

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

1605 1606 1607 1608 1609 1610
	list_for_each_entry(fa, fa_head, fa_list) {
		struct fib_info *fi = fa->fa_info;

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

1611 1612 1613 1614 1615 1616
		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) {
1617 1618 1619 1620 1621
			fa_to_delete = fa;
			break;
		}
	}

O
Olof Johansson 已提交
1622 1623
	if (!fa_to_delete)
		return -ESRCH;
1624

O
Olof Johansson 已提交
1625
	fa = fa_to_delete;
1626 1627
	rtmsg_fib(RTM_DELROUTE, htonl(key), fa, plen, tb->tb_id,
		  &cfg->fc_nlinfo);
O
Olof Johansson 已提交
1628 1629

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

R
Robert Olsson 已提交
1632
	list_del_rcu(&fa->fa_list);
1633

O
Olof Johansson 已提交
1634
	if (list_empty(fa_head)) {
R
Robert Olsson 已提交
1635
		hlist_del_rcu(&li->hlist);
O
Olof Johansson 已提交
1636
		free_leaf_info(li);
R
Robert Olsson 已提交
1637
	}
1638

O
Olof Johansson 已提交
1639 1640
	if (hlist_empty(&l->list))
		trie_leaf_remove(t, key);
1641

O
Olof Johansson 已提交
1642 1643
	if (fa->fa_state & FA_S_ACCESSED)
		rt_cache_flush(-1);
1644

R
Robert Olsson 已提交
1645 1646
	fib_release_info(fa->fa_info);
	alias_free_mem_rcu(fa);
O
Olof Johansson 已提交
1647
	return 0;
1648 1649 1650 1651 1652 1653 1654 1655 1656 1657
}

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 已提交
1658 1659 1660 1661
		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);
1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678
			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 已提交
1679
			hlist_del_rcu(&li->hlist);
1680 1681 1682 1683 1684 1685
			free_leaf_info(li);
		}
	}
	return found;
}

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

1688 1689 1690 1691 1692
static struct leaf *nextleaf(struct trie *t, struct leaf *thisleaf)
{
	struct node *c = (struct node *) thisleaf;
	struct tnode *p;
	int idx;
R
Robert Olsson 已提交
1693
	struct node *trie = rcu_dereference(t->trie);
1694

1695
	if (c == NULL) {
R
Robert Olsson 已提交
1696
		if (trie == NULL)
1697 1698
			return NULL;

R
Robert Olsson 已提交
1699 1700
		if (IS_LEAF(trie))          /* trie w. just a leaf */
			return (struct leaf *) trie;
1701

R
Robert Olsson 已提交
1702
		p = (struct tnode*) trie;  /* Start */
O
Olof Johansson 已提交
1703
	} else
1704
		p = (struct tnode *) NODE_PARENT(c);
1705

1706 1707 1708 1709
	while (p) {
		int pos, last;

		/*  Find the next child of the parent */
1710 1711 1712
		if (c)
			pos = 1 + tkey_extract_bits(c->key, p->pos, p->bits);
		else
1713 1714 1715
			pos = 0;

		last = 1 << p->bits;
O
Olof Johansson 已提交
1716
		for (idx = pos; idx < last ; idx++) {
R
Robert Olsson 已提交
1717 1718 1719
			c = rcu_dereference(p->child[idx]);

			if (!c)
O
Olof Johansson 已提交
1720 1721 1722
				continue;

			/* Decend if tnode */
R
Robert Olsson 已提交
1723 1724
			while (IS_TNODE(c)) {
				p = (struct tnode *) c;
1725
				idx = 0;
O
Olof Johansson 已提交
1726 1727 1728

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

				/* Done with this tnode? */
R
Robert Olsson 已提交
1733
				if (idx >= (1 << p->bits) || !c)
O
Olof Johansson 已提交
1734
					goto up;
1735
			}
R
Robert Olsson 已提交
1736
			return (struct leaf *) c;
1737 1738 1739
		}
up:
		/* No more children go up one step  */
O
Olof Johansson 已提交
1740
		c = (struct node *) p;
1741 1742 1743 1744 1745
		p = (struct tnode *) NODE_PARENT(p);
	}
	return NULL; /* Ready. Root of trie */
}

1746 1747 1748
/*
 * Caller must hold RTNL.
 */
1749 1750 1751 1752 1753 1754 1755 1756
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;

	t->revision++;

O
Olof Johansson 已提交
1757
	for (h = 0; (l = nextleaf(t, l)) != NULL; h++) {
1758 1759 1760 1761 1762 1763 1764 1765 1766 1767
		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 已提交
1768
	pr_debug("trie_flush found=%d\n", found);
1769 1770 1771
	return found;
}

O
Olof Johansson 已提交
1772
static int trie_last_dflt = -1;
1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788

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 已提交
1789
	rcu_read_lock();
1790

1791
	l = fib_find_node(t, 0);
1792
	if (!l)
1793 1794 1795
		goto out;

	fa_head = get_fa_head(l, 0);
1796
	if (!fa_head)
1797 1798
		goto out;

1799
	if (list_empty(fa_head))
1800 1801
		goto out;

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

1805 1806 1807
		if (fa->fa_scope != res->scope ||
		    fa->fa_type != RTN_UNICAST)
			continue;
O
Olof Johansson 已提交
1808

1809 1810 1811 1812 1813 1814
		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 已提交
1815

1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852
		if (fi == NULL) {
			if (next_fi != res->fi)
				break;
		} else if (!fib_detect_death(fi, order, &last_resort,
					     &last_idx, &trie_last_dflt)) {
			if (res->fi)
				fib_info_put(res->fi);
			res->fi = fi;
			atomic_inc(&fi->fib_clntref);
			trie_last_dflt = order;
			goto out;
		}
		fi = next_fi;
		order++;
	}
	if (order <= 0 || fi == NULL) {
		trie_last_dflt = -1;
		goto out;
	}

	if (!fib_detect_death(fi, order, &last_resort, &last_idx, &trie_last_dflt)) {
		if (res->fi)
			fib_info_put(res->fi);
		res->fi = fi;
		atomic_inc(&fi->fib_clntref);
		trie_last_dflt = order;
		goto out;
	}
	if (last_idx >= 0) {
		if (res->fi)
			fib_info_put(res->fi);
		res->fi = last_resort;
		if (last_resort)
			atomic_inc(&last_resort->fib_clntref);
	}
	trie_last_dflt = last_idx;
 out:;
R
Robert Olsson 已提交
1853
	rcu_read_unlock();
1854 1855
}

1856
static int fn_trie_dump_fa(t_key key, int plen, struct list_head *fah, struct fib_table *tb,
1857 1858 1859 1860 1861
			   struct sk_buff *skb, struct netlink_callback *cb)
{
	int i, s_i;
	struct fib_alias *fa;

A
Al Viro 已提交
1862
	__be32 xkey = htonl(key);
1863

1864
	s_i = cb->args[4];
1865 1866
	i = 0;

R
Robert Olsson 已提交
1867 1868 1869
	/* rcu_read_lock is hold by caller */

	list_for_each_entry_rcu(fa, fah, fa_list) {
1870 1871 1872 1873
		if (i < s_i) {
			i++;
			continue;
		}
S
Stephen Hemminger 已提交
1874
		BUG_ON(!fa->fa_info);
1875 1876 1877 1878 1879 1880 1881

		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,
1882
				  xkey,
1883 1884
				  plen,
				  fa->fa_tos,
1885
				  fa->fa_info, 0) < 0) {
1886
			cb->args[4] = i;
1887
			return -1;
O
Olof Johansson 已提交
1888
		}
1889 1890
		i++;
	}
1891
	cb->args[4] = i;
1892 1893 1894
	return skb->len;
}

1895
static int fn_trie_dump_plen(struct trie *t, int plen, struct fib_table *tb, struct sk_buff *skb,
1896 1897 1898 1899 1900 1901
			     struct netlink_callback *cb)
{
	int h, s_h;
	struct list_head *fa_head;
	struct leaf *l = NULL;

1902
	s_h = cb->args[3];
1903

O
Olof Johansson 已提交
1904
	for (h = 0; (l = nextleaf(t, l)) != NULL; h++) {
1905 1906 1907
		if (h < s_h)
			continue;
		if (h > s_h)
1908 1909
			memset(&cb->args[4], 0,
			       sizeof(cb->args) - 4*sizeof(cb->args[0]));
1910 1911

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

1913
		if (!fa_head)
1914 1915
			continue;

1916
		if (list_empty(fa_head))
1917 1918 1919
			continue;

		if (fn_trie_dump_fa(l->key, plen, fa_head, tb, skb, cb)<0) {
1920
			cb->args[3] = h;
1921 1922 1923
			return -1;
		}
	}
1924
	cb->args[3] = h;
1925 1926 1927 1928 1929 1930 1931 1932
	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;

1933
	s_m = cb->args[2];
1934

R
Robert Olsson 已提交
1935
	rcu_read_lock();
O
Olof Johansson 已提交
1936
	for (m = 0; m <= 32; m++) {
1937 1938 1939
		if (m < s_m)
			continue;
		if (m > s_m)
1940 1941
			memset(&cb->args[3], 0,
				sizeof(cb->args) - 3*sizeof(cb->args[0]));
1942 1943

		if (fn_trie_dump_plen(t, 32-m, tb, skb, cb)<0) {
1944
			cb->args[2] = m;
1945 1946 1947
			goto out;
		}
	}
R
Robert Olsson 已提交
1948
	rcu_read_unlock();
1949
	cb->args[2] = m;
1950
	return skb->len;
O
Olof Johansson 已提交
1951
out:
R
Robert Olsson 已提交
1952
	rcu_read_unlock();
1953 1954 1955 1956 1957 1958
	return -1;
}

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

#ifdef CONFIG_IP_MULTIPLE_TABLES
1959
struct fib_table * fib_hash_init(u32 id)
1960
#else
1961
struct fib_table * __init fib_hash_init(u32 id)
1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988 1989 1990
#endif
{
	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,
						  NULL, NULL);

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

	tb->tb_id = id;
	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;
	memset(tb->tb_data, 0, sizeof(struct trie));

	t = (struct trie *) tb->tb_data;

	trie_init(t);

1991
	if (id == RT_TABLE_LOCAL)
O
Olof Johansson 已提交
1992
		trie_local = t;
1993
	else if (id == RT_TABLE_MAIN)
O
Olof Johansson 已提交
1994
		trie_main = t;
1995 1996

	if (id == RT_TABLE_LOCAL)
S
Stephen Hemminger 已提交
1997
		printk(KERN_INFO "IPv4 FIB: Using LC-trie version %s\n", VERSION);
1998 1999 2000 2001

	return tb;
}

2002 2003 2004 2005 2006 2007 2008 2009
#ifdef CONFIG_PROC_FS
/* Depth first Trie walk iterator */
struct fib_trie_iter {
	struct tnode *tnode;
	struct trie *trie;
	unsigned index;
	unsigned depth;
};
2010

2011
static struct node *fib_trie_get_next(struct fib_trie_iter *iter)
2012
{
2013 2014 2015
	struct tnode *tn = iter->tnode;
	unsigned cindex = iter->index;
	struct tnode *p;
2016

2017 2018 2019 2020
	/* A single entry routing table */
	if (!tn)
		return NULL;

2021 2022 2023 2024 2025
	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);
2026

2027 2028 2029 2030 2031 2032 2033 2034 2035 2036 2037 2038
		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;
		}
2039

2040 2041
		++cindex;
	}
O
Olof Johansson 已提交
2042

2043 2044 2045 2046 2047 2048 2049
	/* Current node exhausted, pop back up */
	p = NODE_PARENT(tn);
	if (p) {
		cindex = tkey_extract_bits(tn->key, p->pos, p->bits)+1;
		tn = p;
		--iter->depth;
		goto rescan;
2050
	}
2051 2052 2053

	/* got root? */
	return NULL;
2054 2055
}

2056 2057
static struct node *fib_trie_get_first(struct fib_trie_iter *iter,
				       struct trie *t)
2058
{
2059 2060
	struct node *n ;

S
Stephen Hemminger 已提交
2061
	if (!t)
2062 2063 2064 2065
		return NULL;

	n = rcu_dereference(t->trie);

S
Stephen Hemminger 已提交
2066
	if (!iter)
2067
		return NULL;
2068

2069 2070 2071 2072 2073 2074 2075 2076 2077 2078 2079 2080
	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;
		}
2081
		return n;
O
Olof Johansson 已提交
2082
	}
2083 2084
	return NULL;
}
O
Olof Johansson 已提交
2085

2086 2087 2088 2089
static void trie_collect_stats(struct trie *t, struct trie_stat *s)
{
	struct node *n;
	struct fib_trie_iter iter;
O
Olof Johansson 已提交
2090

2091
	memset(s, 0, sizeof(*s));
O
Olof Johansson 已提交
2092

2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105
	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 已提交
2106
			if (tn->bits < MAX_STAT_DEPTH)
R
Robert Olsson 已提交
2107 2108
				s->nodesizes[tn->bits]++;

2109 2110 2111
			for (i = 0; i < (1<<tn->bits); i++)
				if (!tn->child[i])
					s->nullpointers++;
2112 2113
		}
	}
R
Robert Olsson 已提交
2114
	rcu_read_unlock();
2115 2116
}

2117 2118 2119 2120
/*
 *	This outputs /proc/net/fib_triestats
 */
static void trie_show_stats(struct seq_file *seq, struct trie_stat *stat)
2121
{
2122
	unsigned i, max, pointers, bytes, avdepth;
2123

2124 2125 2126 2127
	if (stat->leaves)
		avdepth = stat->totdepth*100 / stat->leaves;
	else
		avdepth = 0;
O
Olof Johansson 已提交
2128

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

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

2134 2135 2136
	bytes = sizeof(struct leaf) * stat->leaves;
	seq_printf(seq, "\tInternal nodes: %d\n\t", stat->tnodes);
	bytes += sizeof(struct tnode) * stat->tnodes;
2137

R
Robert Olsson 已提交
2138 2139
	max = MAX_STAT_DEPTH;
	while (max > 0 && stat->nodesizes[max-1] == 0)
2140
		max--;
2141

2142 2143 2144 2145 2146 2147 2148 2149
	pointers = 0;
	for (i = 1; i <= max; i++)
		if (stat->nodesizes[i] != 0) {
			seq_printf(seq, "  %d: %d",  i, stat->nodesizes[i]);
			pointers += (1<<i) * stat->nodesizes[i];
		}
	seq_putc(seq, '\n');
	seq_printf(seq, "\tPointers: %d\n", pointers);
R
Robert Olsson 已提交
2150

2151 2152 2153
	bytes += sizeof(struct node *) * pointers;
	seq_printf(seq, "Null ptrs: %d\n", stat->nullpointers);
	seq_printf(seq, "Total size: %d  kB\n", (bytes + 1023) / 1024);
R
Robert Olsson 已提交
2154

2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167
#ifdef CONFIG_IP_FIB_TRIE_STATS
	seq_printf(seq, "Counters:\n---------\n");
	seq_printf(seq,"gets = %d\n", t->stats.gets);
	seq_printf(seq,"backtracks = %d\n", t->stats.backtrack);
	seq_printf(seq,"semantic match passed = %d\n", t->stats.semantic_match_passed);
	seq_printf(seq,"semantic match miss = %d\n", t->stats.semantic_match_miss);
	seq_printf(seq,"null node hit= %d\n", t->stats.null_node_hit);
	seq_printf(seq,"skipped node resize = %d\n", t->stats.resize_node_skipped);
#ifdef CLEAR_STATS
	memset(&(t->stats), 0, sizeof(t->stats));
#endif
#endif /*  CONFIG_IP_FIB_TRIE_STATS */
}
2168

2169 2170 2171
static int fib_triestat_seq_show(struct seq_file *seq, void *v)
{
	struct trie_stat *stat;
O
Olof Johansson 已提交
2172

2173 2174 2175
	stat = kmalloc(sizeof(*stat), GFP_KERNEL);
	if (!stat)
		return -ENOMEM;
O
Olof Johansson 已提交
2176

2177 2178
	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 已提交
2179

2180 2181 2182 2183 2184
	if (trie_local) {
		seq_printf(seq, "Local:\n");
		trie_collect_stats(trie_local, stat);
		trie_show_stats(seq, stat);
	}
O
Olof Johansson 已提交
2185

2186 2187 2188 2189
	if (trie_main) {
		seq_printf(seq, "Main:\n");
		trie_collect_stats(trie_main, stat);
		trie_show_stats(seq, stat);
2190
	}
2191
	kfree(stat);
2192

2193
	return 0;
2194 2195
}

2196
static int fib_triestat_seq_open(struct inode *inode, struct file *file)
2197
{
2198
	return single_open(file, fib_triestat_seq_show, NULL);
2199 2200
}

2201
static const struct file_operations fib_triestat_fops = {
2202 2203 2204 2205 2206 2207 2208 2209 2210
	.owner	= THIS_MODULE,
	.open	= fib_triestat_seq_open,
	.read	= seq_read,
	.llseek	= seq_lseek,
	.release = single_release,
};

static struct node *fib_trie_get_idx(struct fib_trie_iter *iter,
				      loff_t pos)
2211
{
2212 2213 2214 2215 2216 2217 2218 2219 2220 2221 2222 2223 2224 2225
	loff_t idx = 0;
	struct node *n;

	for (n = fib_trie_get_first(iter, trie_local);
	     n; ++idx, n = fib_trie_get_next(iter)) {
		if (pos == idx)
			return n;
	}

	for (n = fib_trie_get_first(iter, trie_main);
	     n; ++idx, n = fib_trie_get_next(iter)) {
		if (pos == idx)
			return n;
	}
2226 2227 2228
	return NULL;
}

2229
static void *fib_trie_seq_start(struct seq_file *seq, loff_t *pos)
2230
{
2231 2232
	rcu_read_lock();
	if (*pos == 0)
O
Olof Johansson 已提交
2233
		return SEQ_START_TOKEN;
2234
	return fib_trie_get_idx(seq->private, *pos - 1);
2235 2236
}

2237
static void *fib_trie_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2238
{
2239 2240 2241
	struct fib_trie_iter *iter = seq->private;
	void *l = v;

2242
	++*pos;
O
Olof Johansson 已提交
2243
	if (v == SEQ_START_TOKEN)
2244
		return fib_trie_get_idx(iter, 0);
2245

2246 2247 2248 2249
	v = fib_trie_get_next(iter);
	BUG_ON(v == l);
	if (v)
		return v;
2250

2251 2252 2253
	/* continue scan in next trie */
	if (iter->trie == trie_local)
		return fib_trie_get_first(iter, trie_main);
2254

2255 2256
	return NULL;
}
2257

2258
static void fib_trie_seq_stop(struct seq_file *seq, void *v)
2259
{
2260 2261
	rcu_read_unlock();
}
O
Olof Johansson 已提交
2262

2263 2264 2265 2266
static void seq_indent(struct seq_file *seq, int n)
{
	while (n-- > 0) seq_puts(seq, "   ");
}
2267

2268 2269 2270
static inline const char *rtn_scope(enum rt_scope_t s)
{
	static char buf[32];
2271

S
Stephen Hemminger 已提交
2272
	switch (s) {
2273 2274 2275 2276 2277 2278 2279 2280 2281 2282
	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:
		snprintf(buf, sizeof(buf), "scope=%d", s);
		return buf;
	}
}
2283

2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297
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",
};
2298

2299 2300 2301
static inline const char *rtn_type(unsigned t)
{
	static char buf[32];
2302

2303 2304 2305 2306
	if (t < __RTN_MAX && rtn_type_names[t])
		return rtn_type_names[t];
	snprintf(buf, sizeof(buf), "type %d", t);
	return buf;
2307 2308
}

2309 2310
/* Pretty print the trie */
static int fib_trie_seq_show(struct seq_file *seq, void *v)
2311
{
2312 2313
	const struct fib_trie_iter *iter = seq->private;
	struct node *n = v;
2314

2315 2316
	if (v == SEQ_START_TOKEN)
		return 0;
2317

2318 2319 2320 2321 2322 2323 2324
	if (!NODE_PARENT(n)) {
		if (iter->trie == trie_local)
			seq_puts(seq, "<local>:\n");
		else
			seq_puts(seq, "<main>:\n");
	}

2325 2326
	if (IS_TNODE(n)) {
		struct tnode *tn = (struct tnode *) n;
A
Al Viro 已提交
2327
		__be32 prf = htonl(MASK_PFX(tn->key, tn->pos));
O
Olof Johansson 已提交
2328

2329 2330
		seq_indent(seq, iter->depth-1);
		seq_printf(seq, "  +-- %d.%d.%d.%d/%d %d %d %d\n",
2331
			   NIPQUAD(prf), tn->pos, tn->bits, tn->full_children,
2332
			   tn->empty_children);
2333

2334 2335 2336
	} else {
		struct leaf *l = (struct leaf *) n;
		int i;
A
Al Viro 已提交
2337
		__be32 val = htonl(l->key);
2338 2339 2340 2341

		seq_indent(seq, iter->depth);
		seq_printf(seq, "  |-- %d.%d.%d.%d\n", NIPQUAD(val));
		for (i = 32; i >= 0; i--) {
R
Robert Olsson 已提交
2342
			struct leaf_info *li = find_leaf_info(l, i);
2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355 2356
			if (li) {
				struct fib_alias *fa;
				list_for_each_entry_rcu(fa, &li->falh, fa_list) {
					seq_indent(seq, iter->depth+1);
					seq_printf(seq, "  /%d %s %s", i,
						   rtn_scope(fa->fa_scope),
						   rtn_type(fa->fa_type));
					if (fa->fa_tos)
						seq_printf(seq, "tos =%d\n",
							   fa->fa_tos);
					seq_putc(seq, '\n');
				}
			}
		}
2357
	}
2358

2359 2360 2361
	return 0;
}

2362
static const struct seq_operations fib_trie_seq_ops = {
2363 2364 2365 2366
	.start  = fib_trie_seq_start,
	.next   = fib_trie_seq_next,
	.stop   = fib_trie_seq_stop,
	.show   = fib_trie_seq_show,
2367 2368
};

2369
static int fib_trie_seq_open(struct inode *inode, struct file *file)
2370 2371 2372
{
	struct seq_file *seq;
	int rc = -ENOMEM;
2373
	struct fib_trie_iter *s = kmalloc(sizeof(*s), GFP_KERNEL);
2374

2375 2376 2377 2378
	if (!s)
		goto out;

	rc = seq_open(file, &fib_trie_seq_ops);
2379 2380 2381
	if (rc)
		goto out_kfree;

2382 2383 2384
	seq	     = file->private_data;
	seq->private = s;
	memset(s, 0, sizeof(*s));
2385 2386 2387
out:
	return rc;
out_kfree:
2388
	kfree(s);
2389 2390 2391
	goto out;
}

2392
static const struct file_operations fib_trie_fops = {
2393 2394 2395 2396
	.owner  = THIS_MODULE,
	.open   = fib_trie_seq_open,
	.read   = seq_read,
	.llseek = seq_lseek,
2397
	.release = seq_release_private,
2398 2399
};

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

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

2415 2416 2417 2418 2419 2420 2421
/*
 *	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)
2422
{
2423
	const struct fib_trie_iter *iter = seq->private;
2424 2425 2426
	struct leaf *l = v;
	int i;
	char bf[128];
2427

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

2435 2436
	if (iter->trie == trie_local)
		return 0;
2437 2438
	if (IS_TNODE(l))
		return 0;
2439

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

2445 2446
		if (!li)
			continue;
2447

2448 2449
		mask = inet_make_mask(li->plen);
		prefix = htonl(l->key);
2450

2451
		list_for_each_entry_rcu(fa, &li->falh, fa_list) {
2452
			const struct fib_info *fi = fa->fa_info;
2453
			unsigned flags = fib_flag_trans(fa->fa_type, mask, fi);
2454

2455 2456 2457
			if (fa->fa_type == RTN_BROADCAST
			    || fa->fa_type == RTN_MULTICAST)
				continue;
2458

2459 2460 2461 2462 2463 2464 2465 2466 2467 2468 2469 2470 2471 2472 2473 2474
			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);
2475

2476 2477
			seq_printf(seq, "%-127s\n", bf);
		}
2478 2479 2480 2481 2482
	}

	return 0;
}

2483
static const struct seq_operations fib_route_seq_ops = {
2484 2485 2486 2487
	.start  = fib_trie_seq_start,
	.next   = fib_trie_seq_next,
	.stop   = fib_trie_seq_stop,
	.show   = fib_route_seq_show,
2488 2489
};

2490
static int fib_route_seq_open(struct inode *inode, struct file *file)
2491 2492 2493
{
	struct seq_file *seq;
	int rc = -ENOMEM;
2494
	struct fib_trie_iter *s = kmalloc(sizeof(*s), GFP_KERNEL);
2495

2496 2497 2498 2499
	if (!s)
		goto out;

	rc = seq_open(file, &fib_route_seq_ops);
2500 2501 2502
	if (rc)
		goto out_kfree;

2503 2504 2505
	seq	     = file->private_data;
	seq->private = s;
	memset(s, 0, sizeof(*s));
2506 2507 2508
out:
	return rc;
out_kfree:
2509
	kfree(s);
2510 2511 2512
	goto out;
}

2513
static const struct file_operations fib_route_fops = {
2514 2515 2516 2517 2518
	.owner  = THIS_MODULE,
	.open   = fib_route_seq_open,
	.read   = seq_read,
	.llseek = seq_lseek,
	.release = seq_release_private,
2519 2520 2521 2522
};

int __init fib_proc_init(void)
{
2523 2524 2525 2526 2527 2528 2529 2530 2531
	if (!proc_net_fops_create("fib_trie", S_IRUGO, &fib_trie_fops))
		goto out1;

	if (!proc_net_fops_create("fib_triestat", S_IRUGO, &fib_triestat_fops))
		goto out2;

	if (!proc_net_fops_create("route", S_IRUGO, &fib_route_fops))
		goto out3;

2532
	return 0;
2533 2534 2535 2536 2537 2538 2539

out3:
	proc_net_remove("fib_triestat");
out2:
	proc_net_remove("fib_trie");
out1:
	return -ENOMEM;
2540 2541 2542 2543 2544
}

void __init fib_proc_exit(void)
{
	proc_net_remove("fib_trie");
2545 2546
	proc_net_remove("fib_triestat");
	proc_net_remove("route");
2547 2548 2549
}

#endif /* CONFIG_PROC_FS */