ip6_fib.c 55.2 KB
Newer Older
L
Linus Torvalds 已提交
1
/*
2
 *	Linux INET6 implementation
L
Linus Torvalds 已提交
3 4 5
 *	Forwarding Information Database
 *
 *	Authors:
6
 *	Pedro Roque		<roque@di.fc.ul.pt>
L
Linus Torvalds 已提交
7 8 9 10 11
 *
 *	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.
12 13 14 15 16 17
 *
 *	Changes:
 *	Yuji SEKIYA @USAGI:	Support default route on router node;
 *				remove ip6_null_entry from the top of
 *				routing table.
 *	Ville Nuorvala:		Fixed routing subtrees.
L
Linus Torvalds 已提交
18
 */
19 20 21

#define pr_fmt(fmt) "IPv6: " fmt

L
Linus Torvalds 已提交
22 23 24 25 26 27 28
#include <linux/errno.h>
#include <linux/types.h>
#include <linux/net.h>
#include <linux/route.h>
#include <linux/netdevice.h>
#include <linux/in6.h>
#include <linux/init.h>
T
Thomas Graf 已提交
29
#include <linux/list.h>
30
#include <linux/slab.h>
L
Linus Torvalds 已提交
31 32 33 34

#include <net/ipv6.h>
#include <net/ndisc.h>
#include <net/addrconf.h>
35
#include <net/lwtunnel.h>
36
#include <net/fib_notifier.h>
L
Linus Torvalds 已提交
37 38 39 40

#include <net/ip6_fib.h>
#include <net/ip6_route.h>

41
static struct kmem_cache *fib6_node_kmem __read_mostly;
L
Linus Torvalds 已提交
42

43 44
struct fib6_cleaner {
	struct fib6_walker w;
45
	struct net *net;
46
	int (*func)(struct fib6_info *, void *arg);
47
	int sernum;
L
Linus Torvalds 已提交
48 49 50 51 52 53 54 55 56
	void *arg;
};

#ifdef CONFIG_IPV6_SUBTREES
#define FWS_INIT FWS_S
#else
#define FWS_INIT FWS_L
#endif

57
static struct fib6_info *fib6_find_prefix(struct net *net,
58 59 60 61 62
					 struct fib6_table *table,
					 struct fib6_node *fn);
static struct fib6_node *fib6_repair_tree(struct net *net,
					  struct fib6_table *table,
					  struct fib6_node *fn);
M
Michal Kubeček 已提交
63
static int fib6_walk(struct net *net, struct fib6_walker *w);
64
static int fib6_walk_continue(struct fib6_walker *w);
L
Linus Torvalds 已提交
65 66 67 68 69 70 71 72

/*
 *	A routing update causes an increase of the serial number on the
 *	affected subtree. This allows for cached routes to be asynchronously
 *	tested when modifications are made to the destination cache as a
 *	result of redirects, path MTU changes, etc.
 */

73
static void fib6_gc_timer_cb(struct timer_list *t);
74

M
Michal Kubeček 已提交
75 76
#define FOR_WALKERS(net, w) \
	list_for_each_entry(w, &(net)->ipv6.fib6_walkers, lh)
L
Linus Torvalds 已提交
77

M
Michal Kubeček 已提交
78
static void fib6_walker_link(struct net *net, struct fib6_walker *w)
A
Adrian Bunk 已提交
79
{
M
Michal Kubeček 已提交
80 81 82
	write_lock_bh(&net->ipv6.fib6_walker_lock);
	list_add(&w->lh, &net->ipv6.fib6_walkers);
	write_unlock_bh(&net->ipv6.fib6_walker_lock);
A
Adrian Bunk 已提交
83 84
}

M
Michal Kubeček 已提交
85
static void fib6_walker_unlink(struct net *net, struct fib6_walker *w)
A
Adrian Bunk 已提交
86
{
M
Michal Kubeček 已提交
87
	write_lock_bh(&net->ipv6.fib6_walker_lock);
88
	list_del(&w->lh);
M
Michal Kubeček 已提交
89
	write_unlock_bh(&net->ipv6.fib6_walker_lock);
A
Adrian Bunk 已提交
90
}
91

92
static int fib6_new_sernum(struct net *net)
L
Linus Torvalds 已提交
93
{
94 95 96
	int new, old;

	do {
97
		old = atomic_read(&net->ipv6.fib6_sernum);
98
		new = old < INT_MAX ? old + 1 : 1;
99 100
	} while (atomic_cmpxchg(&net->ipv6.fib6_sernum,
				old, new) != old);
101
	return new;
L
Linus Torvalds 已提交
102 103
}

104 105 106 107
enum {
	FIB6_NO_SERNUM_CHANGE = 0,
};

108
void fib6_update_sernum(struct net *net, struct fib6_info *f6i)
109 110 111
{
	struct fib6_node *fn;

112 113
	fn = rcu_dereference_protected(f6i->fib6_node,
			lockdep_is_held(&f6i->fib6_table->tb6_lock));
114 115 116 117
	if (fn)
		fn->fn_sernum = fib6_new_sernum(net);
}

L
Linus Torvalds 已提交
118 119 120
/*
 *	Auxiliary address test functions for the radix tree.
 *
121
 *	These assume a 32bit processor (although it will work on
L
Linus Torvalds 已提交
122 123 124 125 126 127
 *	64bit processors)
 */

/*
 *	test bit
 */
128 129 130 131 132
#if defined(__LITTLE_ENDIAN)
# define BITOP_BE32_SWIZZLE	(0x1F & ~7)
#else
# define BITOP_BE32_SWIZZLE	0
#endif
L
Linus Torvalds 已提交
133

134
static __be32 addr_bit_set(const void *token, int fn_bit)
L
Linus Torvalds 已提交
135
{
136
	const __be32 *addr = token;
137 138
	/*
	 * Here,
139
	 *	1 << ((~fn_bit ^ BITOP_BE32_SWIZZLE) & 0x1f)
140 141 142 143
	 * is optimized version of
	 *	htonl(1 << ((~fn_bit)&0x1F))
	 * See include/asm-generic/bitops/le.h.
	 */
144 145
	return (__force __be32)(1 << ((~fn_bit ^ BITOP_BE32_SWIZZLE) & 0x1f)) &
	       addr[fn_bit >> 5];
L
Linus Torvalds 已提交
146 147
}

148
struct fib6_info *fib6_info_alloc(gfp_t gfp_flags)
149
{
150
	struct fib6_info *f6i;
151 152 153 154 155 156 157 158 159 160 161

	f6i = kzalloc(sizeof(*f6i), gfp_flags);
	if (!f6i)
		return NULL;

	f6i->rt6i_pcpu = alloc_percpu_gfp(struct rt6_info *, gfp_flags);
	if (!f6i->rt6i_pcpu) {
		kfree(f6i);
		return NULL;
	}

162
	INIT_LIST_HEAD(&f6i->fib6_siblings);
163 164
	f6i->fib6_metrics = (struct dst_metrics *)&dst_default_metrics;

165
	atomic_inc(&f6i->fib6_ref);
166 167 168 169

	return f6i;
}

170
void fib6_info_destroy(struct fib6_info *f6i)
171 172
{
	struct rt6_exception_bucket *bucket;
173
	struct dst_metrics *m;
174

175
	WARN_ON(f6i->fib6_node);
176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202

	bucket = rcu_dereference_protected(f6i->rt6i_exception_bucket, 1);
	if (bucket) {
		f6i->rt6i_exception_bucket = NULL;
		kfree(bucket);
	}

	if (f6i->rt6i_pcpu) {
		int cpu;

		for_each_possible_cpu(cpu) {
			struct rt6_info **ppcpu_rt;
			struct rt6_info *pcpu_rt;

			ppcpu_rt = per_cpu_ptr(f6i->rt6i_pcpu, cpu);
			pcpu_rt = *ppcpu_rt;
			if (pcpu_rt) {
				dst_dev_put(&pcpu_rt->dst);
				dst_release(&pcpu_rt->dst);
				*ppcpu_rt = NULL;
			}
		}
	}

	if (f6i->fib6_nh.nh_dev)
		dev_put(f6i->fib6_nh.nh_dev);

203 204 205 206
	m = f6i->fib6_metrics;
	if (m != &dst_default_metrics && refcount_dec_and_test(&m->refcnt))
		kfree(m);

207 208 209 210
	kfree(f6i);
}
EXPORT_SYMBOL_GPL(fib6_info_destroy);

W
Wei Wang 已提交
211
static struct fib6_node *node_alloc(struct net *net)
L
Linus Torvalds 已提交
212 213 214
{
	struct fib6_node *fn;

215
	fn = kmem_cache_zalloc(fib6_node_kmem, GFP_ATOMIC);
W
Wei Wang 已提交
216 217
	if (fn)
		net->ipv6.rt6_stats->fib_nodes++;
L
Linus Torvalds 已提交
218 219 220 221

	return fn;
}

W
Wei Wang 已提交
222
static void node_free_immediate(struct net *net, struct fib6_node *fn)
223 224
{
	kmem_cache_free(fib6_node_kmem, fn);
W
Wei Wang 已提交
225
	net->ipv6.rt6_stats->fib_nodes--;
226 227 228
}

static void node_free_rcu(struct rcu_head *head)
L
Linus Torvalds 已提交
229
{
230 231
	struct fib6_node *fn = container_of(head, struct fib6_node, rcu);

L
Linus Torvalds 已提交
232 233 234
	kmem_cache_free(fib6_node_kmem, fn);
}

W
Wei Wang 已提交
235
static void node_free(struct net *net, struct fib6_node *fn)
236 237
{
	call_rcu(&fn->rcu, node_free_rcu);
W
Wei Wang 已提交
238
	net->ipv6.rt6_stats->fib_nodes--;
239 240
}

241 242 243 244 245 246
static void fib6_free_table(struct fib6_table *table)
{
	inetpeer_invalidate_tree(&table->tb6_peers);
	kfree(table);
}

247
static void fib6_link_table(struct net *net, struct fib6_table *tb)
248 249 250
{
	unsigned int h;

251 252 253 254
	/*
	 * Initialize table lock at a single place to give lockdep a key,
	 * tables aren't visible prior to being linked to the list.
	 */
255
	spin_lock_init(&tb->tb6_lock);
256
	h = tb->tb6_id & (FIB6_TABLE_HASHSZ - 1);
257 258 259 260 261

	/*
	 * No protection necessary, this is the only list mutatation
	 * operation, tables never disappear once they exist.
	 */
262
	hlist_add_head_rcu(&tb->tb6_hlist, &net->ipv6.fib_table_hash[h]);
263
}
T
Thomas Graf 已提交
264

265
#ifdef CONFIG_IPV6_MULTIPLE_TABLES
266

267
static struct fib6_table *fib6_alloc_table(struct net *net, u32 id)
T
Thomas Graf 已提交
268 269 270 271
{
	struct fib6_table *table;

	table = kzalloc(sizeof(*table), GFP_ATOMIC);
272
	if (table) {
T
Thomas Graf 已提交
273
		table->tb6_id = id;
274
		rcu_assign_pointer(table->tb6_root.leaf,
D
David Ahern 已提交
275
				   net->ipv6.fib6_null_entry);
T
Thomas Graf 已提交
276
		table->tb6_root.fn_flags = RTN_ROOT | RTN_TL_ROOT | RTN_RTINFO;
277
		inet_peer_base_init(&table->tb6_peers);
T
Thomas Graf 已提交
278 279 280 281 282
	}

	return table;
}

283
struct fib6_table *fib6_new_table(struct net *net, u32 id)
T
Thomas Graf 已提交
284 285 286 287 288
{
	struct fib6_table *tb;

	if (id == 0)
		id = RT6_TABLE_MAIN;
289
	tb = fib6_get_table(net, id);
T
Thomas Graf 已提交
290 291 292
	if (tb)
		return tb;

293
	tb = fib6_alloc_table(net, id);
294
	if (tb)
295
		fib6_link_table(net, tb);
T
Thomas Graf 已提交
296 297 298

	return tb;
}
299
EXPORT_SYMBOL_GPL(fib6_new_table);
T
Thomas Graf 已提交
300

301
struct fib6_table *fib6_get_table(struct net *net, u32 id)
T
Thomas Graf 已提交
302 303
{
	struct fib6_table *tb;
304
	struct hlist_head *head;
T
Thomas Graf 已提交
305 306 307 308
	unsigned int h;

	if (id == 0)
		id = RT6_TABLE_MAIN;
309
	h = id & (FIB6_TABLE_HASHSZ - 1);
T
Thomas Graf 已提交
310
	rcu_read_lock();
311
	head = &net->ipv6.fib_table_hash[h];
312
	hlist_for_each_entry_rcu(tb, head, tb6_hlist) {
T
Thomas Graf 已提交
313 314 315 316 317 318 319 320 321
		if (tb->tb6_id == id) {
			rcu_read_unlock();
			return tb;
		}
	}
	rcu_read_unlock();

	return NULL;
}
322
EXPORT_SYMBOL_GPL(fib6_get_table);
T
Thomas Graf 已提交
323

324
static void __net_init fib6_tables_init(struct net *net)
T
Thomas Graf 已提交
325
{
326 327
	fib6_link_table(net, net->ipv6.fib6_main_tbl);
	fib6_link_table(net, net->ipv6.fib6_local_tbl);
T
Thomas Graf 已提交
328 329 330
}
#else

331
struct fib6_table *fib6_new_table(struct net *net, u32 id)
T
Thomas Graf 已提交
332
{
333
	return fib6_get_table(net, id);
T
Thomas Graf 已提交
334 335
}

336
struct fib6_table *fib6_get_table(struct net *net, u32 id)
T
Thomas Graf 已提交
337
{
338
	  return net->ipv6.fib6_main_tbl;
T
Thomas Graf 已提交
339 340
}

341
struct dst_entry *fib6_rule_lookup(struct net *net, struct flowi6 *fl6,
D
David Ahern 已提交
342
				   const struct sk_buff *skb,
343
				   int flags, pol_lookup_t lookup)
T
Thomas Graf 已提交
344
{
345 346
	struct rt6_info *rt;

D
David Ahern 已提交
347
	rt = lookup(net, net->ipv6.fib6_main_tbl, fl6, skb, flags);
348
	if (rt->dst.error == -EAGAIN) {
349 350 351 352 353 354
		ip6_rt_put(rt);
		rt = net->ipv6.ip6_null_entry;
		dst_hold(&rt->dst);
	}

	return &rt->dst;
T
Thomas Graf 已提交
355 356
}

357
static void __net_init fib6_tables_init(struct net *net)
T
Thomas Graf 已提交
358
{
359
	fib6_link_table(net, net->ipv6.fib6_main_tbl);
T
Thomas Graf 已提交
360 361 362 363
}

#endif

364 365 366 367 368 369 370 371 372
unsigned int fib6_tables_seq_read(struct net *net)
{
	unsigned int h, fib_seq = 0;

	rcu_read_lock();
	for (h = 0; h < FIB6_TABLE_HASHSZ; h++) {
		struct hlist_head *head = &net->ipv6.fib_table_hash[h];
		struct fib6_table *tb;

373
		hlist_for_each_entry_rcu(tb, head, tb6_hlist)
374 375 376 377 378 379 380 381 382
			fib_seq += tb->fib_seq;
	}
	rcu_read_unlock();

	return fib_seq;
}

static int call_fib6_entry_notifier(struct notifier_block *nb, struct net *net,
				    enum fib_event_type event_type,
383
				    struct fib6_info *rt)
384 385 386 387 388 389 390 391
{
	struct fib6_entry_notifier_info info = {
		.rt = rt,
	};

	return call_fib6_notifier(nb, net, event_type, &info.info);
}

392 393
static int call_fib6_entry_notifiers(struct net *net,
				     enum fib_event_type event_type,
394
				     struct fib6_info *rt,
D
David Ahern 已提交
395
				     struct netlink_ext_ack *extack)
396 397
{
	struct fib6_entry_notifier_info info = {
D
David Ahern 已提交
398
		.info.extack = extack,
399 400 401
		.rt = rt,
	};

402
	rt->fib6_table->fib_seq++;
403 404 405
	return call_fib6_notifiers(net, event_type, &info.info);
}

406 407 408 409 410
struct fib6_dump_arg {
	struct net *net;
	struct notifier_block *nb;
};

411
static void fib6_rt_dump(struct fib6_info *rt, struct fib6_dump_arg *arg)
412
{
D
David Ahern 已提交
413
	if (rt == arg->net->ipv6.fib6_null_entry)
414 415 416 417 418 419
		return;
	call_fib6_entry_notifier(arg->nb, arg->net, FIB_EVENT_ENTRY_ADD, rt);
}

static int fib6_node_dump(struct fib6_walker *w)
{
420
	struct fib6_info *rt;
421

422
	for_each_fib6_walker_rt(w)
423 424 425 426 427 428 429 430 431
		fib6_rt_dump(rt, w->args);
	w->leaf = NULL;
	return 0;
}

static void fib6_table_dump(struct net *net, struct fib6_table *tb,
			    struct fib6_walker *w)
{
	w->root = &tb->tb6_root;
432
	spin_lock_bh(&tb->tb6_lock);
433
	fib6_walk(net, w);
434
	spin_unlock_bh(&tb->tb6_lock);
435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465
}

/* Called with rcu_read_lock() */
int fib6_tables_dump(struct net *net, struct notifier_block *nb)
{
	struct fib6_dump_arg arg;
	struct fib6_walker *w;
	unsigned int h;

	w = kzalloc(sizeof(*w), GFP_ATOMIC);
	if (!w)
		return -ENOMEM;

	w->func = fib6_node_dump;
	arg.net = net;
	arg.nb = nb;
	w->args = &arg;

	for (h = 0; h < FIB6_TABLE_HASHSZ; h++) {
		struct hlist_head *head = &net->ipv6.fib_table_hash[h];
		struct fib6_table *tb;

		hlist_for_each_entry_rcu(tb, head, tb6_hlist)
			fib6_table_dump(net, tb, w);
	}

	kfree(w);

	return 0;
}

466
static int fib6_dump_node(struct fib6_walker *w)
467 468
{
	int res;
469
	struct fib6_info *rt;
470

471
	for_each_fib6_walker_rt(w) {
472 473 474 475 476 477
		res = rt6_dump_route(rt, w->args);
		if (res < 0) {
			/* Frame is full, suspend walking */
			w->leaf = rt;
			return 1;
		}
478 479 480 481 482 483

		/* Multipath routes are dumped in one route with the
		 * RTA_MULTIPATH attribute. Jump 'rt' to point to the
		 * last sibling of this route (no need to dump the
		 * sibling routes again)
		 */
484 485
		if (rt->fib6_nsiblings)
			rt = list_last_entry(&rt->fib6_siblings,
486
					     struct fib6_info,
487
					     fib6_siblings);
488 489 490 491 492 493 494
	}
	w->leaf = NULL;
	return 0;
}

static void fib6_dump_end(struct netlink_callback *cb)
{
M
Michal Kubeček 已提交
495
	struct net *net = sock_net(cb->skb->sk);
496
	struct fib6_walker *w = (void *)cb->args[2];
497 498

	if (w) {
499 500
		if (cb->args[4]) {
			cb->args[4] = 0;
M
Michal Kubeček 已提交
501
			fib6_walker_unlink(net, w);
502
		}
503 504 505
		cb->args[2] = 0;
		kfree(w);
	}
506
	cb->done = (void *)cb->args[3];
507 508 509 510 511 512 513 514 515 516 517 518
	cb->args[1] = 3;
}

static int fib6_dump_done(struct netlink_callback *cb)
{
	fib6_dump_end(cb);
	return cb->done ? cb->done(cb) : 0;
}

static int fib6_dump_table(struct fib6_table *table, struct sk_buff *skb,
			   struct netlink_callback *cb)
{
M
Michal Kubeček 已提交
519
	struct net *net = sock_net(skb->sk);
520
	struct fib6_walker *w;
521 522 523 524 525 526
	int res;

	w = (void *)cb->args[2];
	w->root = &table->tb6_root;

	if (cb->args[4] == 0) {
527 528 529
		w->count = 0;
		w->skip = 0;

530
		spin_lock_bh(&table->tb6_lock);
M
Michal Kubeček 已提交
531
		res = fib6_walk(net, w);
532
		spin_unlock_bh(&table->tb6_lock);
533
		if (res > 0) {
534
			cb->args[4] = 1;
535 536
			cb->args[5] = w->root->fn_sernum;
		}
537
	} else {
538 539 540 541 542 543 544 545 546
		if (cb->args[5] != w->root->fn_sernum) {
			/* Begin at the root if the tree changed */
			cb->args[5] = w->root->fn_sernum;
			w->state = FWS_INIT;
			w->node = w->root;
			w->skip = w->count;
		} else
			w->skip = 0;

547
		spin_lock_bh(&table->tb6_lock);
548
		res = fib6_walk_continue(w);
549
		spin_unlock_bh(&table->tb6_lock);
550
		if (res <= 0) {
M
Michal Kubeček 已提交
551
			fib6_walker_unlink(net, w);
552
			cb->args[4] = 0;
553 554
		}
	}
555

556 557 558
	return res;
}

559
static int inet6_dump_fib(struct sk_buff *skb, struct netlink_callback *cb)
560
{
561
	struct net *net = sock_net(skb->sk);
562 563 564
	unsigned int h, s_h;
	unsigned int e = 0, s_e;
	struct rt6_rtnl_dump_arg arg;
565
	struct fib6_walker *w;
566
	struct fib6_table *tb;
567
	struct hlist_head *head;
568 569 570 571 572 573
	int res = 0;

	s_h = cb->args[0];
	s_e = cb->args[1];

	w = (void *)cb->args[2];
574
	if (!w) {
575 576 577 578 579 580 581 582 583 584 585
		/* New dump:
		 *
		 * 1. hook callback destructor.
		 */
		cb->args[3] = (long)cb->done;
		cb->done = fib6_dump_done;

		/*
		 * 2. allocate and initialize walker.
		 */
		w = kzalloc(sizeof(*w), GFP_ATOMIC);
586
		if (!w)
587 588 589 590 591 592 593
			return -ENOMEM;
		w->func = fib6_dump_node;
		cb->args[2] = (long)w;
	}

	arg.skb = skb;
	arg.cb = cb;
594
	arg.net = net;
595 596
	w->args = &arg;

597
	rcu_read_lock();
598
	for (h = s_h; h < FIB6_TABLE_HASHSZ; h++, s_e = 0) {
599
		e = 0;
600
		head = &net->ipv6.fib_table_hash[h];
601
		hlist_for_each_entry_rcu(tb, head, tb6_hlist) {
602 603 604 605 606 607 608 609 610 611
			if (e < s_e)
				goto next;
			res = fib6_dump_table(tb, skb, cb);
			if (res != 0)
				goto out;
next:
			e++;
		}
	}
out:
612
	rcu_read_unlock();
613 614 615 616 617 618 619 620
	cb->args[1] = e;
	cb->args[0] = h;

	res = res < 0 ? res : skb->len;
	if (res <= 0)
		fib6_dump_end(cb);
	return res;
}
L
Linus Torvalds 已提交
621

622
void fib6_metric_set(struct fib6_info *f6i, int metric, u32 val)
623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639
{
	if (!f6i)
		return;

	if (f6i->fib6_metrics == &dst_default_metrics) {
		struct dst_metrics *p = kzalloc(sizeof(*p), GFP_ATOMIC);

		if (!p)
			return;

		refcount_set(&p->refcnt, 1);
		f6i->fib6_metrics = p;
	}

	f6i->fib6_metrics->metrics[metric - 1] = val;
}

L
Linus Torvalds 已提交
640 641 642 643 644 645 646 647
/*
 *	Routing Table
 *
 *	return the appropriate node for a routing tree "add" operation
 *	by either creating and inserting or by returning an existing
 *	node.
 */

W
Wei Wang 已提交
648 649
static struct fib6_node *fib6_add_1(struct net *net,
				    struct fib6_table *table,
650 651 652 653 654
				    struct fib6_node *root,
				    struct in6_addr *addr, int plen,
				    int offset, int allow_create,
				    int replace_required,
				    struct netlink_ext_ack *extack)
L
Linus Torvalds 已提交
655 656 657 658 659
{
	struct fib6_node *fn, *in, *ln;
	struct fib6_node *pn = NULL;
	struct rt6key *key;
	int	bit;
660
	__be32	dir = 0;
L
Linus Torvalds 已提交
661 662 663 664 665 666 667 668

	RT6_TRACE("fib6_add_1\n");

	/* insert node in tree */

	fn = root;

	do {
669
		struct fib6_info *leaf = rcu_dereference_protected(fn->leaf,
670 671
					    lockdep_is_held(&table->tb6_lock));
		key = (struct rt6key *)((u8 *)leaf + offset);
L
Linus Torvalds 已提交
672 673 674 675 676

		/*
		 *	Prefix match
		 */
		if (plen < fn->fn_bit ||
677
		    !ipv6_prefix_equal(&key->addr, addr, fn->fn_bit)) {
678 679
			if (!allow_create) {
				if (replace_required) {
680 681
					NL_SET_ERR_MSG(extack,
						       "Can not replace route - no match found");
682
					pr_warn("Can't replace route, no match found\n");
683 684
					return ERR_PTR(-ENOENT);
				}
685
				pr_warn("NLM_F_CREATE should be set when creating new route\n");
686
			}
L
Linus Torvalds 已提交
687
			goto insert_above;
688
		}
689

L
Linus Torvalds 已提交
690 691 692
		/*
		 *	Exact match ?
		 */
693

L
Linus Torvalds 已提交
694 695
		if (plen == fn->fn_bit) {
			/* clean up an intermediate node */
696
			if (!(fn->fn_flags & RTN_RTINFO)) {
697
				RCU_INIT_POINTER(fn->leaf, NULL);
698
				fib6_info_release(leaf);
699 700 701
			/* remove null_entry in the root node */
			} else if (fn->fn_flags & RTN_TL_ROOT &&
				   rcu_access_pointer(fn->leaf) ==
D
David Ahern 已提交
702
				   net->ipv6.fib6_null_entry) {
703
				RCU_INIT_POINTER(fn->leaf, NULL);
L
Linus Torvalds 已提交
704
			}
705

L
Linus Torvalds 已提交
706 707 708 709 710 711
			return fn;
		}

		/*
		 *	We have more bits to go
		 */
712

L
Linus Torvalds 已提交
713 714 715
		/* Try to walk down on tree. */
		dir = addr_bit_set(addr, fn->fn_bit);
		pn = fn;
716 717 718 719 720
		fn = dir ?
		     rcu_dereference_protected(fn->right,
					lockdep_is_held(&table->tb6_lock)) :
		     rcu_dereference_protected(fn->left,
					lockdep_is_held(&table->tb6_lock));
L
Linus Torvalds 已提交
721 722
	} while (fn);

723
	if (!allow_create) {
724 725 726 727 728 729 730 731 732
		/* We should not create new node because
		 * NLM_F_REPLACE was specified without NLM_F_CREATE
		 * I assume it is safe to require NLM_F_CREATE when
		 * REPLACE flag is used! Later we may want to remove the
		 * check for replace_required, because according
		 * to netlink specification, NLM_F_CREATE
		 * MUST be specified if new route is created.
		 * That would keep IPv6 consistent with IPv4
		 */
733
		if (replace_required) {
734 735
			NL_SET_ERR_MSG(extack,
				       "Can not replace route - no match found");
736
			pr_warn("Can't replace route, no match found\n");
737 738
			return ERR_PTR(-ENOENT);
		}
739
		pr_warn("NLM_F_CREATE should be set when creating new route\n");
740
	}
L
Linus Torvalds 已提交
741 742 743 744 745
	/*
	 *	We walked to the bottom of tree.
	 *	Create new leaf node without children.
	 */

W
Wei Wang 已提交
746
	ln = node_alloc(net);
L
Linus Torvalds 已提交
747

748
	if (!ln)
749
		return ERR_PTR(-ENOMEM);
L
Linus Torvalds 已提交
750
	ln->fn_bit = plen;
751
	RCU_INIT_POINTER(ln->parent, pn);
L
Linus Torvalds 已提交
752 753

	if (dir)
754
		rcu_assign_pointer(pn->right, ln);
L
Linus Torvalds 已提交
755
	else
756
		rcu_assign_pointer(pn->left, ln);
L
Linus Torvalds 已提交
757 758 759 760 761 762

	return ln;


insert_above:
	/*
763
	 * split since we don't have a common prefix anymore or
L
Linus Torvalds 已提交
764 765 766 767 768 769
	 * we have a less significant route.
	 * we've to insert an intermediate node on the list
	 * this new node will point to the one we need to create
	 * and the current
	 */

770 771
	pn = rcu_dereference_protected(fn->parent,
				       lockdep_is_held(&table->tb6_lock));
L
Linus Torvalds 已提交
772 773 774

	/* find 1st bit in difference between the 2 addrs.

775
	   See comment in __ipv6_addr_diff: bit may be an invalid value,
L
Linus Torvalds 已提交
776 777
	   but if it is >= plen, the value is ignored in any case.
	 */
778

779
	bit = __ipv6_addr_diff(addr, &key->addr, sizeof(*addr));
L
Linus Torvalds 已提交
780

781 782
	/*
	 *		(intermediate)[in]
L
Linus Torvalds 已提交
783 784 785 786
	 *	          /	   \
	 *	(new leaf node)[ln] (old node)[fn]
	 */
	if (plen > bit) {
W
Wei Wang 已提交
787 788
		in = node_alloc(net);
		ln = node_alloc(net);
789

790
		if (!in || !ln) {
L
Linus Torvalds 已提交
791
			if (in)
W
Wei Wang 已提交
792
				node_free_immediate(net, in);
L
Linus Torvalds 已提交
793
			if (ln)
W
Wei Wang 已提交
794
				node_free_immediate(net, ln);
795
			return ERR_PTR(-ENOMEM);
L
Linus Torvalds 已提交
796 797
		}

798 799
		/*
		 * new intermediate node.
L
Linus Torvalds 已提交
800 801 802 803 804 805 806 807
		 * RTN_RTINFO will
		 * be off since that an address that chooses one of
		 * the branches would not match less specific routes
		 * in the other branch
		 */

		in->fn_bit = bit;

808
		RCU_INIT_POINTER(in->parent, pn);
L
Linus Torvalds 已提交
809
		in->leaf = fn->leaf;
810
		atomic_inc(&rcu_dereference_protected(in->leaf,
811
				lockdep_is_held(&table->tb6_lock))->fib6_ref);
L
Linus Torvalds 已提交
812 813 814

		/* update parent pointer */
		if (dir)
815
			rcu_assign_pointer(pn->right, in);
L
Linus Torvalds 已提交
816
		else
817
			rcu_assign_pointer(pn->left, in);
L
Linus Torvalds 已提交
818 819 820

		ln->fn_bit = plen;

821 822
		RCU_INIT_POINTER(ln->parent, in);
		rcu_assign_pointer(fn->parent, in);
L
Linus Torvalds 已提交
823 824

		if (addr_bit_set(addr, bit)) {
825 826
			rcu_assign_pointer(in->right, ln);
			rcu_assign_pointer(in->left, fn);
L
Linus Torvalds 已提交
827
		} else {
828 829
			rcu_assign_pointer(in->left, ln);
			rcu_assign_pointer(in->right, fn);
L
Linus Torvalds 已提交
830 831 832
		}
	} else { /* plen <= bit */

833
		/*
L
Linus Torvalds 已提交
834 835 836 837 838
		 *		(new leaf node)[ln]
		 *	          /	   \
		 *	     (old node)[fn] NULL
		 */

W
Wei Wang 已提交
839
		ln = node_alloc(net);
L
Linus Torvalds 已提交
840

841
		if (!ln)
842
			return ERR_PTR(-ENOMEM);
L
Linus Torvalds 已提交
843 844 845

		ln->fn_bit = plen;

846
		RCU_INIT_POINTER(ln->parent, pn);
L
Linus Torvalds 已提交
847 848

		if (addr_bit_set(&key->addr, plen))
849
			RCU_INIT_POINTER(ln->right, fn);
L
Linus Torvalds 已提交
850
		else
851 852 853
			RCU_INIT_POINTER(ln->left, fn);

		rcu_assign_pointer(fn->parent, ln);
L
Linus Torvalds 已提交
854

855 856 857 858
		if (dir)
			rcu_assign_pointer(pn->right, ln);
		else
			rcu_assign_pointer(pn->left, ln);
L
Linus Torvalds 已提交
859 860 861 862
	}
	return ln;
}

863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883
static void fib6_drop_pcpu_from(struct fib6_info *f6i,
				const struct fib6_table *table)
{
	int cpu;

	/* release the reference to this fib entry from
	 * all of its cached pcpu routes
	 */
	for_each_possible_cpu(cpu) {
		struct rt6_info **ppcpu_rt;
		struct rt6_info *pcpu_rt;

		ppcpu_rt = per_cpu_ptr(f6i->rt6i_pcpu, cpu);
		pcpu_rt = *ppcpu_rt;
		if (pcpu_rt) {
			fib6_info_release(pcpu_rt->from);
			pcpu_rt->from = NULL;
		}
	}
}

884
static void fib6_purge_rt(struct fib6_info *rt, struct fib6_node *fn,
885 886
			  struct net *net)
{
887
	struct fib6_table *table = rt->fib6_table;
888

889
	if (atomic_read(&rt->fib6_ref) != 1) {
890 891 892 893 894 895 896
		/* This route is used as dummy address holder in some split
		 * nodes. It is not leaked, but it still holds other resources,
		 * which must be released in time. So, scan ascendant nodes
		 * and replace dummy references to this route with references
		 * to still alive ones.
		 */
		while (fn) {
897
			struct fib6_info *leaf = rcu_dereference_protected(fn->leaf,
898
					    lockdep_is_held(&table->tb6_lock));
899
			struct fib6_info *new_leaf;
900 901
			if (!(fn->fn_flags & RTN_RTINFO) && leaf == rt) {
				new_leaf = fib6_find_prefix(net, table, fn);
902
				atomic_inc(&new_leaf->fib6_ref);
903

904
				rcu_assign_pointer(fn->leaf, new_leaf);
905
				fib6_info_release(rt);
906
			}
907 908
			fn = rcu_dereference_protected(fn->parent,
				    lockdep_is_held(&table->tb6_lock));
909
		}
910

911 912
		if (rt->rt6i_pcpu)
			fib6_drop_pcpu_from(rt, table);
913 914 915
	}
}

L
Linus Torvalds 已提交
916 917 918 919
/*
 *	Insert routing information in a node.
 */

920
static int fib6_add_rt2node(struct fib6_node *fn, struct fib6_info *rt,
921
			    struct nl_info *info,
D
David Ahern 已提交
922
			    struct netlink_ext_ack *extack)
L
Linus Torvalds 已提交
923
{
924
	struct fib6_info *leaf = rcu_dereference_protected(fn->leaf,
925
				    lockdep_is_held(&rt->fib6_table->tb6_lock));
926 927 928
	struct fib6_info *iter = NULL;
	struct fib6_info __rcu **ins;
	struct fib6_info __rcu **fallback_ins = NULL;
929 930 931 932
	int replace = (info->nlh &&
		       (info->nlh->nlmsg_flags & NLM_F_REPLACE));
	int add = (!info->nlh ||
		   (info->nlh->nlmsg_flags & NLM_F_CREATE));
933
	int found = 0;
934
	bool rt_can_ecmp = rt6_qualify_for_ecmp(rt);
935
	u16 nlflags = NLM_F_EXCL;
936
	int err;
L
Linus Torvalds 已提交
937

938 939 940
	if (info->nlh && (info->nlh->nlmsg_flags & NLM_F_APPEND))
		nlflags |= NLM_F_APPEND;

L
Linus Torvalds 已提交
941 942
	ins = &fn->leaf;

943
	for (iter = leaf; iter;
944
	     iter = rcu_dereference_protected(iter->rt6_next,
945
				lockdep_is_held(&rt->fib6_table->tb6_lock))) {
L
Linus Torvalds 已提交
946 947 948 949
		/*
		 *	Search for duplicates
		 */

950
		if (iter->fib6_metric == rt->fib6_metric) {
L
Linus Torvalds 已提交
951 952 953
			/*
			 *	Same priority level
			 */
954 955
			if (info->nlh &&
			    (info->nlh->nlmsg_flags & NLM_F_EXCL))
956
				return -EEXIST;
957 958

			nlflags &= ~NLM_F_EXCL;
959
			if (replace) {
960 961 962 963 964 965 966
				if (rt_can_ecmp == rt6_qualify_for_ecmp(iter)) {
					found++;
					break;
				}
				if (rt_can_ecmp)
					fallback_ins = fallback_ins ?: ins;
				goto next_iter;
967
			}
L
Linus Torvalds 已提交
968

969
			if (rt6_duplicate_nexthop(iter, rt)) {
970 971 972
				if (rt->fib6_nsiblings)
					rt->fib6_nsiblings = 0;
				if (!(iter->fib6_flags & RTF_EXPIRES))
L
Linus Torvalds 已提交
973
					return -EEXIST;
974
				if (!(rt->fib6_flags & RTF_EXPIRES))
975
					fib6_clean_expires(iter);
976
				else
977
					fib6_set_expires(iter, rt->expires);
978
				fib6_metric_set(iter, RTAX_MTU, rt->fib6_pmtu);
L
Linus Torvalds 已提交
979 980
				return -EEXIST;
			}
981 982 983 984 985 986 987 988 989 990 991
			/* If we have the same destination and the same metric,
			 * but not the same gateway, then the route we try to
			 * add is sibling to this route, increment our counter
			 * of siblings, and later we will add our route to the
			 * list.
			 * Only static routes (which don't have flag
			 * RTF_EXPIRES) are used for ECMPv6.
			 *
			 * To avoid long list, we only had siblings if the
			 * route have a gateway.
			 */
992 993
			if (rt_can_ecmp &&
			    rt6_qualify_for_ecmp(iter))
994
				rt->fib6_nsiblings++;
L
Linus Torvalds 已提交
995 996
		}

997
		if (iter->fib6_metric > rt->fib6_metric)
L
Linus Torvalds 已提交
998 999
			break;

1000
next_iter:
1001
		ins = &iter->rt6_next;
L
Linus Torvalds 已提交
1002 1003
	}

1004 1005 1006
	if (fallback_ins && !found) {
		/* No ECMP-able route found, replace first non-ECMP one */
		ins = fallback_ins;
1007
		iter = rcu_dereference_protected(*ins,
1008
				    lockdep_is_held(&rt->fib6_table->tb6_lock));
1009 1010 1011
		found++;
	}

1012 1013 1014 1015
	/* Reset round-robin state, if necessary */
	if (ins == &fn->leaf)
		fn->rr_ptr = NULL;

1016
	/* Link this route to others same route. */
1017 1018
	if (rt->fib6_nsiblings) {
		unsigned int fib6_nsiblings;
1019
		struct fib6_info *sibling, *temp_sibling;
1020 1021

		/* Find the first route that have the same metric */
1022
		sibling = leaf;
1023
		while (sibling) {
1024
			if (sibling->fib6_metric == rt->fib6_metric &&
1025
			    rt6_qualify_for_ecmp(sibling)) {
1026 1027
				list_add_tail(&rt->fib6_siblings,
					      &sibling->fib6_siblings);
1028 1029
				break;
			}
1030
			sibling = rcu_dereference_protected(sibling->rt6_next,
1031
				    lockdep_is_held(&rt->fib6_table->tb6_lock));
1032 1033 1034 1035 1036
		}
		/* For each sibling in the list, increment the counter of
		 * siblings. BUG() if counters does not match, list of siblings
		 * is broken!
		 */
1037
		fib6_nsiblings = 0;
1038
		list_for_each_entry_safe(sibling, temp_sibling,
1039 1040 1041 1042
					 &rt->fib6_siblings, fib6_siblings) {
			sibling->fib6_nsiblings++;
			BUG_ON(sibling->fib6_nsiblings != rt->fib6_nsiblings);
			fib6_nsiblings++;
1043
		}
1044
		BUG_ON(fib6_nsiblings != rt->fib6_nsiblings);
1045
		rt6_multipath_rebalance(temp_sibling);
1046 1047
	}

L
Linus Torvalds 已提交
1048 1049 1050
	/*
	 *	insert node
	 */
1051 1052
	if (!replace) {
		if (!add)
1053
			pr_warn("NLM_F_CREATE should be set when creating new route\n");
1054 1055

add:
1056
		nlflags |= NLM_F_CREATE;
1057

1058 1059 1060 1061 1062 1063
		err = call_fib6_entry_notifiers(info->nl_net,
						FIB_EVENT_ENTRY_ADD,
						rt, extack);
		if (err)
			return err;

1064
		rcu_assign_pointer(rt->rt6_next, iter);
1065 1066
		atomic_inc(&rt->fib6_ref);
		rcu_assign_pointer(rt->fib6_node, fn);
1067
		rcu_assign_pointer(*ins, rt);
1068 1069
		if (!info->skip_notify)
			inet6_rt_notify(RTM_NEWROUTE, rt, info, nlflags);
1070 1071
		info->nl_net->ipv6.rt6_stats->fib_rt_entries++;

1072
		if (!(fn->fn_flags & RTN_RTINFO)) {
1073 1074 1075
			info->nl_net->ipv6.rt6_stats->fib_route_nodes++;
			fn->fn_flags |= RTN_RTINFO;
		}
L
Linus Torvalds 已提交
1076

1077
	} else {
1078 1079
		int nsiblings;

1080 1081 1082
		if (!found) {
			if (add)
				goto add;
1083
			pr_warn("NLM_F_REPLACE set, but no existing node found!\n");
1084 1085
			return -ENOENT;
		}
1086

1087 1088 1089 1090 1091 1092
		err = call_fib6_entry_notifiers(info->nl_net,
						FIB_EVENT_ENTRY_REPLACE,
						rt, extack);
		if (err)
			return err;

1093 1094
		atomic_inc(&rt->fib6_ref);
		rcu_assign_pointer(rt->fib6_node, fn);
1095
		rt->rt6_next = iter->rt6_next;
1096
		rcu_assign_pointer(*ins, rt);
1097 1098
		if (!info->skip_notify)
			inet6_rt_notify(RTM_NEWROUTE, rt, info, NLM_F_REPLACE);
1099
		if (!(fn->fn_flags & RTN_RTINFO)) {
1100 1101 1102
			info->nl_net->ipv6.rt6_stats->fib_route_nodes++;
			fn->fn_flags |= RTN_RTINFO;
		}
1103 1104
		nsiblings = iter->fib6_nsiblings;
		iter->fib6_node = NULL;
1105
		fib6_purge_rt(iter, fn, info->nl_net);
1106
		if (rcu_access_pointer(fn->rr_ptr) == iter)
1107
			fn->rr_ptr = NULL;
1108
		fib6_info_release(iter);
1109 1110 1111

		if (nsiblings) {
			/* Replacing an ECMP route, remove all siblings */
1112
			ins = &rt->rt6_next;
1113
			iter = rcu_dereference_protected(*ins,
1114
				    lockdep_is_held(&rt->fib6_table->tb6_lock));
1115
			while (iter) {
1116
				if (iter->fib6_metric > rt->fib6_metric)
1117
					break;
1118
				if (rt6_qualify_for_ecmp(iter)) {
1119
					*ins = iter->rt6_next;
1120
					iter->fib6_node = NULL;
1121
					fib6_purge_rt(iter, fn, info->nl_net);
1122
					if (rcu_access_pointer(fn->rr_ptr) == iter)
1123
						fn->rr_ptr = NULL;
1124
					fib6_info_release(iter);
1125
					nsiblings--;
W
Wei Wang 已提交
1126
					info->nl_net->ipv6.rt6_stats->fib_rt_entries--;
1127
				} else {
1128
					ins = &iter->rt6_next;
1129
				}
1130
				iter = rcu_dereference_protected(*ins,
1131
					lockdep_is_held(&rt->fib6_table->tb6_lock));
1132 1133 1134
			}
			WARN_ON(nsiblings != 0);
		}
L
Linus Torvalds 已提交
1135 1136 1137 1138 1139
	}

	return 0;
}

1140
static void fib6_start_gc(struct net *net, struct fib6_info *rt)
L
Linus Torvalds 已提交
1141
{
1142
	if (!timer_pending(&net->ipv6.ip6_fib_timer) &&
1143
	    (rt->fib6_flags & RTF_EXPIRES))
1144
		mod_timer(&net->ipv6.ip6_fib_timer,
S
Stephen Hemminger 已提交
1145
			  jiffies + net->ipv6.sysctl.ip6_rt_gc_interval);
L
Linus Torvalds 已提交
1146 1147
}

1148
void fib6_force_start_gc(struct net *net)
L
Linus Torvalds 已提交
1149
{
1150 1151
	if (!timer_pending(&net->ipv6.ip6_fib_timer))
		mod_timer(&net->ipv6.ip6_fib_timer,
S
Stephen Hemminger 已提交
1152
			  jiffies + net->ipv6.sysctl.ip6_rt_gc_interval);
L
Linus Torvalds 已提交
1153 1154
}

1155
static void __fib6_update_sernum_upto_root(struct fib6_info *rt,
1156
					   int sernum)
1157
{
1158 1159
	struct fib6_node *fn = rcu_dereference_protected(rt->fib6_node,
				lockdep_is_held(&rt->fib6_table->tb6_lock));
1160 1161 1162 1163 1164

	/* paired with smp_rmb() in rt6_get_cookie_safe() */
	smp_wmb();
	while (fn) {
		fn->fn_sernum = sernum;
1165
		fn = rcu_dereference_protected(fn->parent,
1166
				lockdep_is_held(&rt->fib6_table->tb6_lock));
1167 1168 1169
	}
}

1170
void fib6_update_sernum_upto_root(struct net *net, struct fib6_info *rt)
1171 1172 1173 1174
{
	__fib6_update_sernum_upto_root(rt, fib6_new_sernum(net));
}

L
Linus Torvalds 已提交
1175 1176 1177 1178
/*
 *	Add routing information to the routing tree.
 *	<destination addr>/<source addr>
 *	with source addr info in sub-trees
1179
 *	Need to own table->tb6_lock
L
Linus Torvalds 已提交
1180 1181
 */

1182
int fib6_add(struct fib6_node *root, struct fib6_info *rt,
1183
	     struct nl_info *info, struct netlink_ext_ack *extack)
L
Linus Torvalds 已提交
1184
{
1185
	struct fib6_table *table = rt->fib6_table;
1186
	struct fib6_node *fn, *pn = NULL;
L
Linus Torvalds 已提交
1187
	int err = -ENOMEM;
1188 1189
	int allow_create = 1;
	int replace_required = 0;
1190
	int sernum = fib6_new_sernum(info->nl_net);
1191 1192 1193

	if (info->nlh) {
		if (!(info->nlh->nlmsg_flags & NLM_F_CREATE))
1194
			allow_create = 0;
1195
		if (info->nlh->nlmsg_flags & NLM_F_REPLACE)
1196 1197 1198
			replace_required = 1;
	}
	if (!allow_create && !replace_required)
1199
		pr_warn("RTM_NEWROUTE with no NLM_F_CREATE or NLM_F_REPLACE\n");
L
Linus Torvalds 已提交
1200

W
Wei Wang 已提交
1201
	fn = fib6_add_1(info->nl_net, table, root,
1202 1203
			&rt->fib6_dst.addr, rt->fib6_dst.plen,
			offsetof(struct fib6_info, fib6_dst), allow_create,
1204
			replace_required, extack);
1205 1206
	if (IS_ERR(fn)) {
		err = PTR_ERR(fn);
1207
		fn = NULL;
L
Linus Torvalds 已提交
1208
		goto out;
1209
	}
L
Linus Torvalds 已提交
1210

1211 1212
	pn = fn;

L
Linus Torvalds 已提交
1213
#ifdef CONFIG_IPV6_SUBTREES
1214
	if (rt->fib6_src.plen) {
L
Linus Torvalds 已提交
1215 1216
		struct fib6_node *sn;

1217
		if (!rcu_access_pointer(fn->subtree)) {
L
Linus Torvalds 已提交
1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230
			struct fib6_node *sfn;

			/*
			 * Create subtree.
			 *
			 *		fn[main tree]
			 *		|
			 *		sfn[subtree root]
			 *		   \
			 *		    sn[new leaf node]
			 */

			/* Create subtree root node */
W
Wei Wang 已提交
1231
			sfn = node_alloc(info->nl_net);
1232
			if (!sfn)
1233
				goto failure;
L
Linus Torvalds 已提交
1234

1235
			atomic_inc(&info->nl_net->ipv6.fib6_null_entry->fib6_ref);
1236
			rcu_assign_pointer(sfn->leaf,
D
David Ahern 已提交
1237
					   info->nl_net->ipv6.fib6_null_entry);
L
Linus Torvalds 已提交
1238 1239 1240 1241
			sfn->fn_flags = RTN_ROOT;

			/* Now add the first leaf node to new subtree */

W
Wei Wang 已提交
1242
			sn = fib6_add_1(info->nl_net, table, sfn,
1243 1244
					&rt->fib6_src.addr, rt->fib6_src.plen,
					offsetof(struct fib6_info, fib6_src),
1245
					allow_create, replace_required, extack);
L
Linus Torvalds 已提交
1246

1247
			if (IS_ERR(sn)) {
L
Linus Torvalds 已提交
1248
				/* If it is failed, discard just allocated
1249
				   root, and then (in failure) stale node
L
Linus Torvalds 已提交
1250 1251
				   in main tree.
				 */
W
Wei Wang 已提交
1252
				node_free_immediate(info->nl_net, sfn);
1253
				err = PTR_ERR(sn);
1254
				goto failure;
L
Linus Torvalds 已提交
1255 1256 1257
			}

			/* Now link new subtree to main tree */
1258 1259
			rcu_assign_pointer(sfn->parent, fn);
			rcu_assign_pointer(fn->subtree, sfn);
L
Linus Torvalds 已提交
1260
		} else {
W
Wei Wang 已提交
1261
			sn = fib6_add_1(info->nl_net, table, FIB6_SUBTREE(fn),
1262 1263
					&rt->fib6_src.addr, rt->fib6_src.plen,
					offsetof(struct fib6_info, fib6_src),
1264
					allow_create, replace_required, extack);
L
Linus Torvalds 已提交
1265

1266 1267
			if (IS_ERR(sn)) {
				err = PTR_ERR(sn);
1268
				goto failure;
1269
			}
L
Linus Torvalds 已提交
1270 1271
		}

1272
		if (!rcu_access_pointer(fn->leaf)) {
1273 1274 1275
			if (fn->fn_flags & RTN_TL_ROOT) {
				/* put back null_entry for root node */
				rcu_assign_pointer(fn->leaf,
D
David Ahern 已提交
1276
					    info->nl_net->ipv6.fib6_null_entry);
1277
			} else {
1278
				atomic_inc(&rt->fib6_ref);
1279 1280
				rcu_assign_pointer(fn->leaf, rt);
			}
1281
		}
L
Linus Torvalds 已提交
1282 1283 1284 1285
		fn = sn;
	}
#endif

1286
	err = fib6_add_rt2node(fn, rt, info, extack);
1287
	if (!err) {
1288
		__fib6_update_sernum_upto_root(rt, sernum);
1289
		fib6_start_gc(info->nl_net, rt);
1290
	}
L
Linus Torvalds 已提交
1291 1292

out:
1293 1294 1295 1296 1297 1298
	if (err) {
#ifdef CONFIG_IPV6_SUBTREES
		/*
		 * If fib6_add_1 has cleared the old leaf pointer in the
		 * super-tree leaf node we have to find a new one for it.
		 */
1299
		if (pn != fn) {
1300
			struct fib6_info *pn_leaf =
1301 1302 1303 1304 1305
				rcu_dereference_protected(pn->leaf,
				    lockdep_is_held(&table->tb6_lock));
			if (pn_leaf == rt) {
				pn_leaf = NULL;
				RCU_INIT_POINTER(pn->leaf, NULL);
1306
				fib6_info_release(rt);
1307
			}
1308 1309 1310 1311 1312 1313 1314
			if (!pn_leaf && !(pn->fn_flags & RTN_RTINFO)) {
				pn_leaf = fib6_find_prefix(info->nl_net, table,
							   pn);
#if RT6_DEBUG >= 2
				if (!pn_leaf) {
					WARN_ON(!pn_leaf);
					pn_leaf =
D
David Ahern 已提交
1315
					    info->nl_net->ipv6.fib6_null_entry;
1316
				}
1317
#endif
1318
				fib6_info_hold(pn_leaf);
1319 1320
				rcu_assign_pointer(pn->leaf, pn_leaf);
			}
1321 1322
		}
#endif
1323
		goto failure;
1324
	}
L
Linus Torvalds 已提交
1325 1326
	return err;

1327
failure:
1328 1329 1330 1331 1332 1333
	/* fn->leaf could be NULL and fib6_repair_tree() needs to be called if:
	 * 1. fn is an intermediate node and we failed to add the new
	 * route to it in both subtree creation failure and fib6_add_rt2node()
	 * failure case.
	 * 2. fn is the root node in the table and we fail to add the first
	 * default route to it.
L
Linus Torvalds 已提交
1334
	 */
1335 1336 1337 1338
	if (fn &&
	    (!(fn->fn_flags & (RTN_RTINFO|RTN_ROOT)) ||
	     (fn->fn_flags & RTN_TL_ROOT &&
	      !rcu_access_pointer(fn->leaf))))
1339
		fib6_repair_tree(info->nl_net, table, fn);
L
Linus Torvalds 已提交
1340 1341 1342 1343 1344 1345 1346 1347 1348
	return err;
}

/*
 *	Routing tree lookup
 *
 */

struct lookup_args {
1349
	int			offset;		/* key offset on fib6_info */
1350
	const struct in6_addr	*addr;		/* search key			*/
L
Linus Torvalds 已提交
1351 1352
};

1353 1354
static struct fib6_node *fib6_lookup_1(struct fib6_node *root,
				       struct lookup_args *args)
L
Linus Torvalds 已提交
1355 1356
{
	struct fib6_node *fn;
A
Al Viro 已提交
1357
	__be32 dir;
L
Linus Torvalds 已提交
1358

1359 1360 1361
	if (unlikely(args->offset == 0))
		return NULL;

L
Linus Torvalds 已提交
1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372
	/*
	 *	Descend on a tree
	 */

	fn = root;

	for (;;) {
		struct fib6_node *next;

		dir = addr_bit_set(args->addr, fn->fn_bit);

1373 1374
		next = dir ? rcu_dereference(fn->right) :
			     rcu_dereference(fn->left);
L
Linus Torvalds 已提交
1375 1376 1377 1378 1379 1380 1381 1382

		if (next) {
			fn = next;
			continue;
		}
		break;
	}

1383
	while (fn) {
1384 1385 1386
		struct fib6_node *subtree = FIB6_SUBTREE(fn);

		if (subtree || fn->fn_flags & RTN_RTINFO) {
1387
			struct fib6_info *leaf = rcu_dereference(fn->leaf);
L
Linus Torvalds 已提交
1388 1389
			struct rt6key *key;

W
Wei Wang 已提交
1390 1391 1392 1393
			if (!leaf)
				goto backtrack;

			key = (struct rt6key *) ((u8 *)leaf + args->offset);
L
Linus Torvalds 已提交
1394

1395 1396
			if (ipv6_prefix_equal(&key->addr, args->addr, key->plen)) {
#ifdef CONFIG_IPV6_SUBTREES
1397
				if (subtree) {
1398
					struct fib6_node *sfn;
1399
					sfn = fib6_lookup_1(subtree, args + 1);
1400 1401 1402 1403
					if (!sfn)
						goto backtrack;
					fn = sfn;
				}
1404
#endif
1405
				if (fn->fn_flags & RTN_RTINFO)
1406 1407
					return fn;
			}
L
Linus Torvalds 已提交
1408
		}
1409
backtrack:
1410 1411 1412
		if (fn->fn_flags & RTN_ROOT)
			break;

1413
		fn = rcu_dereference(fn->parent);
L
Linus Torvalds 已提交
1414 1415 1416 1417 1418
	}

	return NULL;
}

1419 1420
/* called with rcu_read_lock() held
 */
1421 1422
struct fib6_node *fib6_lookup(struct fib6_node *root, const struct in6_addr *daddr,
			      const struct in6_addr *saddr)
L
Linus Torvalds 已提交
1423 1424
{
	struct fib6_node *fn;
1425 1426
	struct lookup_args args[] = {
		{
1427
			.offset = offsetof(struct fib6_info, fib6_dst),
1428 1429
			.addr = daddr,
		},
L
Linus Torvalds 已提交
1430
#ifdef CONFIG_IPV6_SUBTREES
1431
		{
1432
			.offset = offsetof(struct fib6_info, fib6_src),
1433 1434
			.addr = saddr,
		},
L
Linus Torvalds 已提交
1435
#endif
1436 1437 1438 1439
		{
			.offset = 0,	/* sentinel */
		}
	};
L
Linus Torvalds 已提交
1440

1441
	fn = fib6_lookup_1(root, daddr ? args : args + 1);
1442
	if (!fn || fn->fn_flags & RTN_TL_ROOT)
L
Linus Torvalds 已提交
1443 1444 1445 1446 1447 1448 1449 1450
		fn = root;

	return fn;
}

/*
 *	Get node with specified destination prefix (and source prefix,
 *	if subtrees are used)
1451 1452 1453 1454 1455 1456
 *	exact_match == true means we try to find fn with exact match of
 *	the passed in prefix addr
 *	exact_match == false means we try to find fn with longest prefix
 *	match of the passed in prefix addr. This is useful for finding fn
 *	for cached route as it will be stored in the exception table under
 *	the node with longest prefix length.
L
Linus Torvalds 已提交
1457 1458 1459
 */


1460 1461
static struct fib6_node *fib6_locate_1(struct fib6_node *root,
				       const struct in6_addr *addr,
1462 1463
				       int plen, int offset,
				       bool exact_match)
L
Linus Torvalds 已提交
1464
{
1465
	struct fib6_node *fn, *prev = NULL;
L
Linus Torvalds 已提交
1466 1467

	for (fn = root; fn ; ) {
1468
		struct fib6_info *leaf = rcu_dereference(fn->leaf);
W
Wei Wang 已提交
1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479
		struct rt6key *key;

		/* This node is being deleted */
		if (!leaf) {
			if (plen <= fn->fn_bit)
				goto out;
			else
				goto next;
		}

		key = (struct rt6key *)((u8 *)leaf + offset);
L
Linus Torvalds 已提交
1480 1481 1482 1483 1484 1485

		/*
		 *	Prefix match
		 */
		if (plen < fn->fn_bit ||
		    !ipv6_prefix_equal(&key->addr, addr, fn->fn_bit))
1486
			goto out;
L
Linus Torvalds 已提交
1487 1488 1489 1490

		if (plen == fn->fn_bit)
			return fn;

1491 1492
		prev = fn;

W
Wei Wang 已提交
1493
next:
L
Linus Torvalds 已提交
1494 1495 1496 1497
		/*
		 *	We have more bits to go
		 */
		if (addr_bit_set(addr, fn->fn_bit))
1498
			fn = rcu_dereference(fn->right);
L
Linus Torvalds 已提交
1499
		else
1500
			fn = rcu_dereference(fn->left);
L
Linus Torvalds 已提交
1501
	}
1502 1503 1504 1505 1506
out:
	if (exact_match)
		return NULL;
	else
		return prev;
L
Linus Torvalds 已提交
1507 1508
}

1509 1510
struct fib6_node *fib6_locate(struct fib6_node *root,
			      const struct in6_addr *daddr, int dst_len,
1511 1512
			      const struct in6_addr *saddr, int src_len,
			      bool exact_match)
L
Linus Torvalds 已提交
1513 1514 1515 1516
{
	struct fib6_node *fn;

	fn = fib6_locate_1(root, daddr, dst_len,
1517
			   offsetof(struct fib6_info, fib6_dst),
1518
			   exact_match);
L
Linus Torvalds 已提交
1519 1520 1521

#ifdef CONFIG_IPV6_SUBTREES
	if (src_len) {
1522
		WARN_ON(saddr == NULL);
1523 1524 1525 1526 1527
		if (fn) {
			struct fib6_node *subtree = FIB6_SUBTREE(fn);

			if (subtree) {
				fn = fib6_locate_1(subtree, saddr, src_len,
1528
					   offsetof(struct fib6_info, fib6_src),
1529
					   exact_match);
1530 1531
			}
		}
L
Linus Torvalds 已提交
1532 1533 1534
	}
#endif

1535
	if (fn && fn->fn_flags & RTN_RTINFO)
L
Linus Torvalds 已提交
1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546
		return fn;

	return NULL;
}


/*
 *	Deletion
 *
 */

1547
static struct fib6_info *fib6_find_prefix(struct net *net,
1548 1549
					 struct fib6_table *table,
					 struct fib6_node *fn)
L
Linus Torvalds 已提交
1550
{
1551 1552
	struct fib6_node *child_left, *child_right;

1553
	if (fn->fn_flags & RTN_ROOT)
D
David Ahern 已提交
1554
		return net->ipv6.fib6_null_entry;
L
Linus Torvalds 已提交
1555

1556
	while (fn) {
1557 1558 1559 1560 1561 1562 1563 1564 1565 1566
		child_left = rcu_dereference_protected(fn->left,
				    lockdep_is_held(&table->tb6_lock));
		child_right = rcu_dereference_protected(fn->right,
				    lockdep_is_held(&table->tb6_lock));
		if (child_left)
			return rcu_dereference_protected(child_left->leaf,
					lockdep_is_held(&table->tb6_lock));
		if (child_right)
			return rcu_dereference_protected(child_right->leaf,
					lockdep_is_held(&table->tb6_lock));
L
Linus Torvalds 已提交
1567

1568
		fn = FIB6_SUBTREE(fn);
L
Linus Torvalds 已提交
1569 1570 1571 1572 1573 1574 1575
	}
	return NULL;
}

/*
 *	Called to trim the tree of intermediate nodes when possible. "fn"
 *	is the node we want to try and remove.
1576
 *	Need to own table->tb6_lock
L
Linus Torvalds 已提交
1577 1578
 */

1579
static struct fib6_node *fib6_repair_tree(struct net *net,
1580 1581
					  struct fib6_table *table,
					  struct fib6_node *fn)
L
Linus Torvalds 已提交
1582 1583 1584
{
	int children;
	int nstate;
1585
	struct fib6_node *child;
1586
	struct fib6_walker *w;
L
Linus Torvalds 已提交
1587 1588
	int iter = 0;

1589 1590
	/* Set fn->leaf to null_entry for root node. */
	if (fn->fn_flags & RTN_TL_ROOT) {
D
David Ahern 已提交
1591
		rcu_assign_pointer(fn->leaf, net->ipv6.fib6_null_entry);
1592 1593 1594
		return fn;
	}

L
Linus Torvalds 已提交
1595
	for (;;) {
1596 1597 1598 1599 1600 1601 1602 1603 1604 1605
		struct fib6_node *fn_r = rcu_dereference_protected(fn->right,
					    lockdep_is_held(&table->tb6_lock));
		struct fib6_node *fn_l = rcu_dereference_protected(fn->left,
					    lockdep_is_held(&table->tb6_lock));
		struct fib6_node *pn = rcu_dereference_protected(fn->parent,
					    lockdep_is_held(&table->tb6_lock));
		struct fib6_node *pn_r = rcu_dereference_protected(pn->right,
					    lockdep_is_held(&table->tb6_lock));
		struct fib6_node *pn_l = rcu_dereference_protected(pn->left,
					    lockdep_is_held(&table->tb6_lock));
1606
		struct fib6_info *fn_leaf = rcu_dereference_protected(fn->leaf,
1607
					    lockdep_is_held(&table->tb6_lock));
1608
		struct fib6_info *pn_leaf = rcu_dereference_protected(pn->leaf,
1609
					    lockdep_is_held(&table->tb6_lock));
1610
		struct fib6_info *new_fn_leaf;
1611

L
Linus Torvalds 已提交
1612 1613 1614
		RT6_TRACE("fixing tree: plen=%d iter=%d\n", fn->fn_bit, iter);
		iter++;

1615 1616
		WARN_ON(fn->fn_flags & RTN_RTINFO);
		WARN_ON(fn->fn_flags & RTN_TL_ROOT);
1617
		WARN_ON(fn_leaf);
L
Linus Torvalds 已提交
1618 1619 1620

		children = 0;
		child = NULL;
1621 1622 1623 1624
		if (fn_r)
			child = fn_r, children |= 1;
		if (fn_l)
			child = fn_l, children |= 2;
L
Linus Torvalds 已提交
1625

1626
		if (children == 3 || FIB6_SUBTREE(fn)
L
Linus Torvalds 已提交
1627 1628
#ifdef CONFIG_IPV6_SUBTREES
		    /* Subtree root (i.e. fn) may have one child */
1629
		    || (children && fn->fn_flags & RTN_ROOT)
L
Linus Torvalds 已提交
1630 1631
#endif
		    ) {
1632
			new_fn_leaf = fib6_find_prefix(net, table, fn);
L
Linus Torvalds 已提交
1633
#if RT6_DEBUG >= 2
1634 1635
			if (!new_fn_leaf) {
				WARN_ON(!new_fn_leaf);
D
David Ahern 已提交
1636
				new_fn_leaf = net->ipv6.fib6_null_entry;
L
Linus Torvalds 已提交
1637 1638
			}
#endif
1639
			fib6_info_hold(new_fn_leaf);
1640 1641
			rcu_assign_pointer(fn->leaf, new_fn_leaf);
			return pn;
L
Linus Torvalds 已提交
1642 1643 1644
		}

#ifdef CONFIG_IPV6_SUBTREES
1645
		if (FIB6_SUBTREE(pn) == fn) {
1646
			WARN_ON(!(fn->fn_flags & RTN_ROOT));
1647
			RCU_INIT_POINTER(pn->subtree, NULL);
L
Linus Torvalds 已提交
1648 1649
			nstate = FWS_L;
		} else {
1650
			WARN_ON(fn->fn_flags & RTN_ROOT);
L
Linus Torvalds 已提交
1651
#endif
1652 1653 1654 1655
			if (pn_r == fn)
				rcu_assign_pointer(pn->right, child);
			else if (pn_l == fn)
				rcu_assign_pointer(pn->left, child);
L
Linus Torvalds 已提交
1656
#if RT6_DEBUG >= 2
1657 1658
			else
				WARN_ON(1);
L
Linus Torvalds 已提交
1659 1660
#endif
			if (child)
1661
				rcu_assign_pointer(child->parent, pn);
L
Linus Torvalds 已提交
1662 1663 1664 1665 1666
			nstate = FWS_R;
#ifdef CONFIG_IPV6_SUBTREES
		}
#endif

M
Michal Kubeček 已提交
1667 1668
		read_lock(&net->ipv6.fib6_walker_lock);
		FOR_WALKERS(net, w) {
1669
			if (!child) {
1670
				if (w->node == fn) {
L
Linus Torvalds 已提交
1671 1672 1673 1674 1675 1676 1677 1678 1679
					RT6_TRACE("W %p adjusted by delnode 1, s=%d/%d\n", w, w->state, nstate);
					w->node = pn;
					w->state = nstate;
				}
			} else {
				if (w->node == fn) {
					w->node = child;
					if (children&2) {
						RT6_TRACE("W %p adjusted by delnode 2, s=%d\n", w, w->state);
1680
						w->state = w->state >= FWS_R ? FWS_U : FWS_INIT;
L
Linus Torvalds 已提交
1681 1682
					} else {
						RT6_TRACE("W %p adjusted by delnode 2, s=%d\n", w, w->state);
1683
						w->state = w->state >= FWS_C ? FWS_U : FWS_INIT;
L
Linus Torvalds 已提交
1684 1685 1686 1687
					}
				}
			}
		}
M
Michal Kubeček 已提交
1688
		read_unlock(&net->ipv6.fib6_walker_lock);
L
Linus Torvalds 已提交
1689

W
Wei Wang 已提交
1690
		node_free(net, fn);
1691
		if (pn->fn_flags & RTN_RTINFO || FIB6_SUBTREE(pn))
L
Linus Torvalds 已提交
1692 1693
			return pn;

1694
		RCU_INIT_POINTER(pn->leaf, NULL);
1695
		fib6_info_release(pn_leaf);
L
Linus Torvalds 已提交
1696 1697 1698 1699
		fn = pn;
	}
}

1700
static void fib6_del_route(struct fib6_table *table, struct fib6_node *fn,
1701
			   struct fib6_info __rcu **rtp, struct nl_info *info)
L
Linus Torvalds 已提交
1702
{
1703
	struct fib6_walker *w;
1704
	struct fib6_info *rt = rcu_dereference_protected(*rtp,
1705
				    lockdep_is_held(&table->tb6_lock));
1706
	struct net *net = info->nl_net;
L
Linus Torvalds 已提交
1707 1708 1709 1710

	RT6_TRACE("fib6_del_route\n");

	/* Unlink it */
1711
	*rtp = rt->rt6_next;
1712
	rt->fib6_node = NULL;
1713 1714
	net->ipv6.rt6_stats->fib_rt_entries--;
	net->ipv6.rt6_stats->fib_discarded_routes++;
L
Linus Torvalds 已提交
1715

1716 1717 1718
	/* Flush all cached dst in exception table */
	rt6_flush_exceptions(rt);

1719
	/* Reset round-robin state, if necessary */
1720
	if (rcu_access_pointer(fn->rr_ptr) == rt)
1721 1722
		fn->rr_ptr = NULL;

1723
	/* Remove this entry from other siblings */
1724
	if (rt->fib6_nsiblings) {
1725
		struct fib6_info *sibling, *next_sibling;
1726 1727

		list_for_each_entry_safe(sibling, next_sibling,
1728 1729 1730 1731
					 &rt->fib6_siblings, fib6_siblings)
			sibling->fib6_nsiblings--;
		rt->fib6_nsiblings = 0;
		list_del_init(&rt->fib6_siblings);
1732
		rt6_multipath_rebalance(next_sibling);
1733 1734
	}

L
Linus Torvalds 已提交
1735
	/* Adjust walkers */
M
Michal Kubeček 已提交
1736 1737
	read_lock(&net->ipv6.fib6_walker_lock);
	FOR_WALKERS(net, w) {
L
Linus Torvalds 已提交
1738 1739
		if (w->state == FWS_C && w->leaf == rt) {
			RT6_TRACE("walker %p adjusted by delroute\n", w);
1740
			w->leaf = rcu_dereference_protected(rt->rt6_next,
1741
					    lockdep_is_held(&table->tb6_lock));
1742
			if (!w->leaf)
L
Linus Torvalds 已提交
1743 1744 1745
				w->state = FWS_U;
		}
	}
M
Michal Kubeček 已提交
1746
	read_unlock(&net->ipv6.fib6_walker_lock);
L
Linus Torvalds 已提交
1747

1748 1749 1750 1751
	/* If it was last route, call fib6_repair_tree() to:
	 * 1. For root node, put back null_entry as how the table was created.
	 * 2. For other nodes, expunge its radix tree node.
	 */
1752
	if (!rcu_access_pointer(fn->leaf)) {
1753 1754 1755 1756
		if (!(fn->fn_flags & RTN_TL_ROOT)) {
			fn->fn_flags &= ~RTN_RTINFO;
			net->ipv6.rt6_stats->fib_route_nodes--;
		}
1757
		fn = fib6_repair_tree(net, table, fn);
L
Linus Torvalds 已提交
1758 1759
	}

1760
	fib6_purge_rt(rt, fn, net);
L
Linus Torvalds 已提交
1761

D
David Ahern 已提交
1762
	call_fib6_entry_notifiers(net, FIB_EVENT_ENTRY_DEL, rt, NULL);
1763 1764
	if (!info->skip_notify)
		inet6_rt_notify(RTM_DELROUTE, rt, info, 0);
1765
	fib6_info_release(rt);
L
Linus Torvalds 已提交
1766 1767
}

1768
/* Need to own table->tb6_lock */
1769
int fib6_del(struct fib6_info *rt, struct nl_info *info)
L
Linus Torvalds 已提交
1770
{
1771 1772 1773
	struct fib6_node *fn = rcu_dereference_protected(rt->fib6_node,
				    lockdep_is_held(&rt->fib6_table->tb6_lock));
	struct fib6_table *table = rt->fib6_table;
1774
	struct net *net = info->nl_net;
1775 1776
	struct fib6_info __rcu **rtp;
	struct fib6_info __rcu **rtp_next;
L
Linus Torvalds 已提交
1777

D
David Ahern 已提交
1778
	if (!fn || rt == net->ipv6.fib6_null_entry)
L
Linus Torvalds 已提交
1779 1780
		return -ENOENT;

1781
	WARN_ON(!(fn->fn_flags & RTN_RTINFO));
L
Linus Torvalds 已提交
1782 1783 1784 1785 1786

	/*
	 *	Walk the leaf entries looking for ourself
	 */

1787
	for (rtp = &fn->leaf; *rtp; rtp = rtp_next) {
1788
		struct fib6_info *cur = rcu_dereference_protected(*rtp,
1789 1790 1791
					lockdep_is_held(&table->tb6_lock));
		if (rt == cur) {
			fib6_del_route(table, fn, rtp, info);
L
Linus Torvalds 已提交
1792 1793
			return 0;
		}
1794
		rtp_next = &cur->rt6_next;
L
Linus Torvalds 已提交
1795 1796 1797 1798 1799 1800 1801 1802 1803 1804 1805
	}
	return -ENOENT;
}

/*
 *	Tree traversal function.
 *
 *	Certainly, it is not interrupt safe.
 *	However, it is internally reenterable wrt itself and fib6_add/fib6_del.
 *	It means, that we can modify tree during walking
 *	and use this function for garbage collection, clone pruning,
1806
 *	cleaning tree when a device goes down etc. etc.
L
Linus Torvalds 已提交
1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820
 *
 *	It guarantees that every node will be traversed,
 *	and that it will be traversed only once.
 *
 *	Callback function w->func may return:
 *	0 -> continue walking.
 *	positive value -> walking is suspended (used by tree dumps,
 *	and probably by gc, if it will be split to several slices)
 *	negative value -> terminate walking.
 *
 *	The function itself returns:
 *	0   -> walk is complete.
 *	>0  -> walk is incomplete (i.e. suspended)
 *	<0  -> walk is terminated by an error.
1821 1822
 *
 *	This function is called with tb6_lock held.
L
Linus Torvalds 已提交
1823 1824
 */

1825
static int fib6_walk_continue(struct fib6_walker *w)
L
Linus Torvalds 已提交
1826
{
1827
	struct fib6_node *fn, *pn, *left, *right;
L
Linus Torvalds 已提交
1828

1829 1830 1831
	/* w->root should always be table->tb6_root */
	WARN_ON_ONCE(!(w->root->fn_flags & RTN_TL_ROOT));

L
Linus Torvalds 已提交
1832 1833
	for (;;) {
		fn = w->node;
1834
		if (!fn)
L
Linus Torvalds 已提交
1835 1836 1837 1838 1839
			return 0;

		switch (w->state) {
#ifdef CONFIG_IPV6_SUBTREES
		case FWS_S:
1840 1841
			if (FIB6_SUBTREE(fn)) {
				w->node = FIB6_SUBTREE(fn);
L
Linus Torvalds 已提交
1842 1843 1844
				continue;
			}
			w->state = FWS_L;
1845
#endif
1846
			/* fall through */
L
Linus Torvalds 已提交
1847
		case FWS_L:
1848 1849 1850
			left = rcu_dereference_protected(fn->left, 1);
			if (left) {
				w->node = left;
L
Linus Torvalds 已提交
1851 1852 1853 1854
				w->state = FWS_INIT;
				continue;
			}
			w->state = FWS_R;
1855
			/* fall through */
L
Linus Torvalds 已提交
1856
		case FWS_R:
1857 1858 1859
			right = rcu_dereference_protected(fn->right, 1);
			if (right) {
				w->node = right;
L
Linus Torvalds 已提交
1860 1861 1862 1863
				w->state = FWS_INIT;
				continue;
			}
			w->state = FWS_C;
1864
			w->leaf = rcu_dereference_protected(fn->leaf, 1);
1865
			/* fall through */
L
Linus Torvalds 已提交
1866
		case FWS_C:
1867
			if (w->leaf && fn->fn_flags & RTN_RTINFO) {
1868 1869
				int err;

E
Eric Dumazet 已提交
1870 1871
				if (w->skip) {
					w->skip--;
1872
					goto skip;
1873 1874 1875
				}

				err = w->func(w);
L
Linus Torvalds 已提交
1876 1877
				if (err)
					return err;
1878 1879

				w->count++;
L
Linus Torvalds 已提交
1880 1881
				continue;
			}
1882
skip:
L
Linus Torvalds 已提交
1883
			w->state = FWS_U;
1884
			/* fall through */
L
Linus Torvalds 已提交
1885 1886 1887
		case FWS_U:
			if (fn == w->root)
				return 0;
1888 1889 1890
			pn = rcu_dereference_protected(fn->parent, 1);
			left = rcu_dereference_protected(pn->left, 1);
			right = rcu_dereference_protected(pn->right, 1);
L
Linus Torvalds 已提交
1891 1892
			w->node = pn;
#ifdef CONFIG_IPV6_SUBTREES
1893
			if (FIB6_SUBTREE(pn) == fn) {
1894
				WARN_ON(!(fn->fn_flags & RTN_ROOT));
L
Linus Torvalds 已提交
1895 1896 1897 1898
				w->state = FWS_L;
				continue;
			}
#endif
1899
			if (left == fn) {
L
Linus Torvalds 已提交
1900 1901 1902
				w->state = FWS_R;
				continue;
			}
1903
			if (right == fn) {
L
Linus Torvalds 已提交
1904
				w->state = FWS_C;
1905
				w->leaf = rcu_dereference_protected(w->node->leaf, 1);
L
Linus Torvalds 已提交
1906 1907 1908
				continue;
			}
#if RT6_DEBUG >= 2
1909
			WARN_ON(1);
L
Linus Torvalds 已提交
1910 1911 1912 1913 1914
#endif
		}
	}
}

M
Michal Kubeček 已提交
1915
static int fib6_walk(struct net *net, struct fib6_walker *w)
L
Linus Torvalds 已提交
1916 1917 1918 1919 1920 1921
{
	int res;

	w->state = FWS_INIT;
	w->node = w->root;

M
Michal Kubeček 已提交
1922
	fib6_walker_link(net, w);
L
Linus Torvalds 已提交
1923 1924
	res = fib6_walk_continue(w);
	if (res <= 0)
M
Michal Kubeček 已提交
1925
		fib6_walker_unlink(net, w);
L
Linus Torvalds 已提交
1926 1927 1928
	return res;
}

1929
static int fib6_clean_node(struct fib6_walker *w)
L
Linus Torvalds 已提交
1930 1931
{
	int res;
1932
	struct fib6_info *rt;
1933
	struct fib6_cleaner *c = container_of(w, struct fib6_cleaner, w);
1934 1935 1936
	struct nl_info info = {
		.nl_net = c->net,
	};
L
Linus Torvalds 已提交
1937

1938 1939 1940 1941 1942 1943 1944 1945 1946 1947
	if (c->sernum != FIB6_NO_SERNUM_CHANGE &&
	    w->node->fn_sernum != c->sernum)
		w->node->fn_sernum = c->sernum;

	if (!c->func) {
		WARN_ON_ONCE(c->sernum == FIB6_NO_SERNUM_CHANGE);
		w->leaf = NULL;
		return 0;
	}

1948
	for_each_fib6_walker_rt(w) {
L
Linus Torvalds 已提交
1949
		res = c->func(rt, c->arg);
1950
		if (res == -1) {
L
Linus Torvalds 已提交
1951
			w->leaf = rt;
1952
			res = fib6_del(rt, &info);
L
Linus Torvalds 已提交
1953 1954
			if (res) {
#if RT6_DEBUG >= 2
1955
				pr_debug("%s: del failed: rt=%p@%p err=%d\n",
W
Wei Wang 已提交
1956
					 __func__, rt,
1957
					 rcu_access_pointer(rt->fib6_node),
W
Wei Wang 已提交
1958
					 res);
L
Linus Torvalds 已提交
1959 1960 1961 1962
#endif
				continue;
			}
			return 0;
1963
		} else if (res == -2) {
1964
			if (WARN_ON(!rt->fib6_nsiblings))
1965
				continue;
1966 1967
			rt = list_last_entry(&rt->fib6_siblings,
					     struct fib6_info, fib6_siblings);
1968
			continue;
L
Linus Torvalds 已提交
1969
		}
1970
		WARN_ON(res != 0);
L
Linus Torvalds 已提交
1971 1972 1973 1974 1975 1976 1977
	}
	w->leaf = rt;
	return 0;
}

/*
 *	Convenient frontend to tree walker.
1978
 *
L
Linus Torvalds 已提交
1979
 *	func is called on each route.
1980 1981
 *		It may return -2 -> skip multipath route.
 *			      -1 -> delete this route.
L
Linus Torvalds 已提交
1982 1983 1984
 *		              0  -> continue walking
 */

1985
static void fib6_clean_tree(struct net *net, struct fib6_node *root,
1986
			    int (*func)(struct fib6_info *, void *arg),
1987
			    int sernum, void *arg)
L
Linus Torvalds 已提交
1988
{
1989
	struct fib6_cleaner c;
L
Linus Torvalds 已提交
1990 1991 1992

	c.w.root = root;
	c.w.func = fib6_clean_node;
1993 1994
	c.w.count = 0;
	c.w.skip = 0;
L
Linus Torvalds 已提交
1995
	c.func = func;
1996
	c.sernum = sernum;
L
Linus Torvalds 已提交
1997
	c.arg = arg;
1998
	c.net = net;
L
Linus Torvalds 已提交
1999

M
Michal Kubeček 已提交
2000
	fib6_walk(net, &c.w);
L
Linus Torvalds 已提交
2001 2002
}

2003
static void __fib6_clean_all(struct net *net,
2004
			     int (*func)(struct fib6_info *, void *),
2005
			     int sernum, void *arg)
T
Thomas Graf 已提交
2006 2007
{
	struct fib6_table *table;
2008
	struct hlist_head *head;
2009
	unsigned int h;
T
Thomas Graf 已提交
2010

2011
	rcu_read_lock();
2012
	for (h = 0; h < FIB6_TABLE_HASHSZ; h++) {
2013
		head = &net->ipv6.fib_table_hash[h];
2014
		hlist_for_each_entry_rcu(table, head, tb6_hlist) {
2015
			spin_lock_bh(&table->tb6_lock);
2016
			fib6_clean_tree(net, &table->tb6_root,
2017
					func, sernum, arg);
2018
			spin_unlock_bh(&table->tb6_lock);
T
Thomas Graf 已提交
2019 2020
		}
	}
2021
	rcu_read_unlock();
T
Thomas Graf 已提交
2022 2023
}

2024
void fib6_clean_all(struct net *net, int (*func)(struct fib6_info *, void *),
2025 2026 2027 2028 2029
		    void *arg)
{
	__fib6_clean_all(net, func, FIB6_NO_SERNUM_CHANGE, arg);
}

H
Hannes Frederic Sowa 已提交
2030 2031
static void fib6_flush_trees(struct net *net)
{
2032
	int new_sernum = fib6_new_sernum(net);
H
Hannes Frederic Sowa 已提交
2033

2034
	__fib6_clean_all(net, NULL, new_sernum, NULL);
H
Hannes Frederic Sowa 已提交
2035 2036
}

L
Linus Torvalds 已提交
2037 2038 2039 2040
/*
 *	Garbage collection
 */

2041
static int fib6_age(struct fib6_info *rt, void *arg)
L
Linus Torvalds 已提交
2042
{
2043
	struct fib6_gc_args *gc_args = arg;
L
Linus Torvalds 已提交
2044 2045 2046 2047 2048 2049 2050
	unsigned long now = jiffies;

	/*
	 *	check addrconf expiration here.
	 *	Routes are expired even if they are in use.
	 */

2051
	if (rt->fib6_flags & RTF_EXPIRES && rt->expires) {
2052
		if (time_after(now, rt->expires)) {
L
Linus Torvalds 已提交
2053 2054 2055
			RT6_TRACE("expiring %p\n", rt);
			return -1;
		}
2056
		gc_args->more++;
L
Linus Torvalds 已提交
2057 2058
	}

2059 2060 2061 2062 2063 2064
	/*	Also age clones in the exception table.
	 *	Note, that clones are aged out
	 *	only if they are not in use now.
	 */
	rt6_age_exceptions(rt, gc_args, now);

L
Linus Torvalds 已提交
2065 2066 2067
	return 0;
}

2068
void fib6_run_gc(unsigned long expires, struct net *net, bool force)
L
Linus Torvalds 已提交
2069
{
2070
	struct fib6_gc_args gc_args;
2071 2072
	unsigned long now;

2073
	if (force) {
2074 2075
		spin_lock_bh(&net->ipv6.fib6_gc_lock);
	} else if (!spin_trylock_bh(&net->ipv6.fib6_gc_lock)) {
2076 2077
		mod_timer(&net->ipv6.ip6_fib_timer, jiffies + HZ);
		return;
L
Linus Torvalds 已提交
2078
	}
2079 2080
	gc_args.timeout = expires ? (int)expires :
			  net->ipv6.sysctl.ip6_rt_gc_interval;
2081
	gc_args.more = 0;
2082

2083
	fib6_clean_all(net, fib6_age, &gc_args);
2084 2085
	now = jiffies;
	net->ipv6.ip6_rt_last_gc = now;
L
Linus Torvalds 已提交
2086 2087

	if (gc_args.more)
S
Stephen Hemminger 已提交
2088
		mod_timer(&net->ipv6.ip6_fib_timer,
2089
			  round_jiffies(now
S
Stephen Hemminger 已提交
2090
					+ net->ipv6.sysctl.ip6_rt_gc_interval));
2091 2092
	else
		del_timer(&net->ipv6.ip6_fib_timer);
2093
	spin_unlock_bh(&net->ipv6.fib6_gc_lock);
L
Linus Torvalds 已提交
2094 2095
}

2096
static void fib6_gc_timer_cb(struct timer_list *t)
2097
{
2098 2099 2100
	struct net *arg = from_timer(arg, t, ipv6.ip6_fib_timer);

	fib6_run_gc(0, arg, true);
2101 2102
}

2103
static int __net_init fib6_net_init(struct net *net)
L
Linus Torvalds 已提交
2104
{
2105
	size_t size = sizeof(struct hlist_head) * FIB6_TABLE_HASHSZ;
2106 2107 2108 2109 2110
	int err;

	err = fib6_notifier_init(net);
	if (err)
		return err;
2111

2112
	spin_lock_init(&net->ipv6.fib6_gc_lock);
M
Michal Kubeček 已提交
2113 2114
	rwlock_init(&net->ipv6.fib6_walker_lock);
	INIT_LIST_HEAD(&net->ipv6.fib6_walkers);
2115
	timer_setup(&net->ipv6.ip6_fib_timer, fib6_gc_timer_cb, 0);
2116

2117 2118 2119 2120
	net->ipv6.rt6_stats = kzalloc(sizeof(*net->ipv6.rt6_stats), GFP_KERNEL);
	if (!net->ipv6.rt6_stats)
		goto out_timer;

2121 2122 2123 2124
	/* Avoid false sharing : Use at least a full cache line */
	size = max_t(size_t, size, L1_CACHE_BYTES);

	net->ipv6.fib_table_hash = kzalloc(size, GFP_KERNEL);
2125
	if (!net->ipv6.fib_table_hash)
2126
		goto out_rt6_stats;
2127

2128 2129 2130
	net->ipv6.fib6_main_tbl = kzalloc(sizeof(*net->ipv6.fib6_main_tbl),
					  GFP_KERNEL);
	if (!net->ipv6.fib6_main_tbl)
2131 2132
		goto out_fib_table_hash;

2133
	net->ipv6.fib6_main_tbl->tb6_id = RT6_TABLE_MAIN;
2134
	rcu_assign_pointer(net->ipv6.fib6_main_tbl->tb6_root.leaf,
D
David Ahern 已提交
2135
			   net->ipv6.fib6_null_entry);
2136 2137
	net->ipv6.fib6_main_tbl->tb6_root.fn_flags =
		RTN_ROOT | RTN_TL_ROOT | RTN_RTINFO;
2138
	inet_peer_base_init(&net->ipv6.fib6_main_tbl->tb6_peers);
2139 2140

#ifdef CONFIG_IPV6_MULTIPLE_TABLES
2141 2142 2143
	net->ipv6.fib6_local_tbl = kzalloc(sizeof(*net->ipv6.fib6_local_tbl),
					   GFP_KERNEL);
	if (!net->ipv6.fib6_local_tbl)
2144
		goto out_fib6_main_tbl;
2145
	net->ipv6.fib6_local_tbl->tb6_id = RT6_TABLE_LOCAL;
2146
	rcu_assign_pointer(net->ipv6.fib6_local_tbl->tb6_root.leaf,
D
David Ahern 已提交
2147
			   net->ipv6.fib6_null_entry);
2148 2149
	net->ipv6.fib6_local_tbl->tb6_root.fn_flags =
		RTN_ROOT | RTN_TL_ROOT | RTN_RTINFO;
2150
	inet_peer_base_init(&net->ipv6.fib6_local_tbl->tb6_peers);
2151
#endif
2152
	fib6_tables_init(net);
2153

2154
	return 0;
2155

2156 2157
#ifdef CONFIG_IPV6_MULTIPLE_TABLES
out_fib6_main_tbl:
2158
	kfree(net->ipv6.fib6_main_tbl);
2159 2160
#endif
out_fib_table_hash:
2161
	kfree(net->ipv6.fib_table_hash);
2162 2163
out_rt6_stats:
	kfree(net->ipv6.rt6_stats);
2164
out_timer:
2165
	fib6_notifier_exit(net);
2166
	return -ENOMEM;
2167
}
2168 2169 2170

static void fib6_net_exit(struct net *net)
{
2171 2172
	unsigned int i;

2173 2174
	del_timer_sync(&net->ipv6.ip6_fib_timer);

E
Eric Dumazet 已提交
2175
	for (i = 0; i < FIB6_TABLE_HASHSZ; i++) {
2176 2177 2178 2179 2180 2181 2182 2183 2184 2185
		struct hlist_head *head = &net->ipv6.fib_table_hash[i];
		struct hlist_node *tmp;
		struct fib6_table *tb;

		hlist_for_each_entry_safe(tb, tmp, head, tb6_hlist) {
			hlist_del(&tb->tb6_hlist);
			fib6_free_table(tb);
		}
	}

2186
	kfree(net->ipv6.fib_table_hash);
2187
	kfree(net->ipv6.rt6_stats);
2188
	fib6_notifier_exit(net);
2189 2190 2191 2192 2193 2194 2195 2196 2197 2198
}

static struct pernet_operations fib6_net_ops = {
	.init = fib6_net_init,
	.exit = fib6_net_exit,
};

int __init fib6_init(void)
{
	int ret = -ENOMEM;
2199

2200 2201 2202 2203 2204 2205 2206 2207 2208
	fib6_node_kmem = kmem_cache_create("fib6_nodes",
					   sizeof(struct fib6_node),
					   0, SLAB_HWCACHE_ALIGN,
					   NULL);
	if (!fib6_node_kmem)
		goto out;

	ret = register_pernet_subsys(&fib6_net_ops);
	if (ret)
2209
		goto out_kmem_cache_create;
2210

2211 2212
	ret = rtnl_register_module(THIS_MODULE, PF_INET6, RTM_GETROUTE, NULL,
				   inet6_dump_fib, 0);
2213 2214
	if (ret)
		goto out_unregister_subsys;
H
Hannes Frederic Sowa 已提交
2215 2216

	__fib6_flush_trees = fib6_flush_trees;
2217 2218 2219
out:
	return ret;

2220 2221
out_unregister_subsys:
	unregister_pernet_subsys(&fib6_net_ops);
2222 2223 2224
out_kmem_cache_create:
	kmem_cache_destroy(fib6_node_kmem);
	goto out;
L
Linus Torvalds 已提交
2225 2226 2227 2228
}

void fib6_gc_cleanup(void)
{
2229
	unregister_pernet_subsys(&fib6_net_ops);
L
Linus Torvalds 已提交
2230 2231
	kmem_cache_destroy(fib6_node_kmem);
}
2232 2233 2234 2235 2236

#ifdef CONFIG_PROC_FS

struct ipv6_route_iter {
	struct seq_net_private p;
2237
	struct fib6_walker w;
2238 2239
	loff_t skip;
	struct fib6_table *tbl;
2240
	int sernum;
2241 2242 2243 2244
};

static int ipv6_route_seq_show(struct seq_file *seq, void *v)
{
2245
	struct fib6_info *rt = v;
2246
	struct ipv6_route_iter *iter = seq->private;
2247
	const struct net_device *dev;
2248

2249
	seq_printf(seq, "%pi6 %02x ", &rt->fib6_dst.addr, rt->fib6_dst.plen);
2250 2251

#ifdef CONFIG_IPV6_SUBTREES
2252
	seq_printf(seq, "%pi6 %02x ", &rt->fib6_src.addr, rt->fib6_src.plen);
2253 2254 2255
#else
	seq_puts(seq, "00000000000000000000000000000000 00 ");
#endif
2256
	if (rt->fib6_flags & RTF_GATEWAY)
2257
		seq_printf(seq, "%pi6", &rt->fib6_nh.nh_gw);
2258 2259 2260
	else
		seq_puts(seq, "00000000000000000000000000000000");

2261
	dev = rt->fib6_nh.nh_dev;
2262
	seq_printf(seq, " %08x %08x %08x %08x %8s\n",
2263 2264
		   rt->fib6_metric, atomic_read(&rt->fib6_ref), 0,
		   rt->fib6_flags, dev ? dev->name : "");
2265 2266 2267 2268
	iter->w.leaf = NULL;
	return 0;
}

2269
static int ipv6_route_yield(struct fib6_walker *w)
2270 2271 2272 2273 2274 2275 2276
{
	struct ipv6_route_iter *iter = w->args;

	if (!iter->skip)
		return 1;

	do {
2277
		iter->w.leaf = rcu_dereference_protected(
2278
				iter->w.leaf->rt6_next,
2279
				lockdep_is_held(&iter->tbl->tb6_lock));
2280 2281 2282 2283 2284 2285 2286 2287
		iter->skip--;
		if (!iter->skip && iter->w.leaf)
			return 1;
	} while (iter->w.leaf);

	return 0;
}

M
Michal Kubeček 已提交
2288 2289
static void ipv6_route_seq_setup_walk(struct ipv6_route_iter *iter,
				      struct net *net)
2290 2291 2292 2293 2294 2295 2296
{
	memset(&iter->w, 0, sizeof(iter->w));
	iter->w.func = ipv6_route_yield;
	iter->w.root = &iter->tbl->tb6_root;
	iter->w.state = FWS_INIT;
	iter->w.node = iter->w.root;
	iter->w.args = iter;
2297
	iter->sernum = iter->w.root->fn_sernum;
2298
	INIT_LIST_HEAD(&iter->w.lh);
M
Michal Kubeček 已提交
2299
	fib6_walker_link(net, &iter->w);
2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311 2312 2313 2314 2315 2316 2317 2318 2319 2320 2321 2322
}

static struct fib6_table *ipv6_route_seq_next_table(struct fib6_table *tbl,
						    struct net *net)
{
	unsigned int h;
	struct hlist_node *node;

	if (tbl) {
		h = (tbl->tb6_id & (FIB6_TABLE_HASHSZ - 1)) + 1;
		node = rcu_dereference_bh(hlist_next_rcu(&tbl->tb6_hlist));
	} else {
		h = 0;
		node = NULL;
	}

	while (!node && h < FIB6_TABLE_HASHSZ) {
		node = rcu_dereference_bh(
			hlist_first_rcu(&net->ipv6.fib_table_hash[h++]));
	}
	return hlist_entry_safe(node, struct fib6_table, tb6_hlist);
}

2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333
static void ipv6_route_check_sernum(struct ipv6_route_iter *iter)
{
	if (iter->sernum != iter->w.root->fn_sernum) {
		iter->sernum = iter->w.root->fn_sernum;
		iter->w.state = FWS_INIT;
		iter->w.node = iter->w.root;
		WARN_ON(iter->w.skip);
		iter->w.skip = iter->w.count;
	}
}

2334 2335 2336
static void *ipv6_route_seq_next(struct seq_file *seq, void *v, loff_t *pos)
{
	int r;
2337
	struct fib6_info *n;
2338 2339 2340 2341 2342 2343
	struct net *net = seq_file_net(seq);
	struct ipv6_route_iter *iter = seq->private;

	if (!v)
		goto iter_table;

2344
	n = rcu_dereference_bh(((struct fib6_info *)v)->rt6_next);
2345 2346 2347 2348 2349 2350
	if (n) {
		++*pos;
		return n;
	}

iter_table:
2351
	ipv6_route_check_sernum(iter);
2352
	spin_lock_bh(&iter->tbl->tb6_lock);
2353
	r = fib6_walk_continue(&iter->w);
2354
	spin_unlock_bh(&iter->tbl->tb6_lock);
2355 2356 2357 2358 2359
	if (r > 0) {
		if (v)
			++*pos;
		return iter->w.leaf;
	} else if (r < 0) {
M
Michal Kubeček 已提交
2360
		fib6_walker_unlink(net, &iter->w);
2361 2362
		return NULL;
	}
M
Michal Kubeček 已提交
2363
	fib6_walker_unlink(net, &iter->w);
2364 2365 2366 2367 2368

	iter->tbl = ipv6_route_seq_next_table(iter->tbl, net);
	if (!iter->tbl)
		return NULL;

M
Michal Kubeček 已提交
2369
	ipv6_route_seq_setup_walk(iter, net);
2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383
	goto iter_table;
}

static void *ipv6_route_seq_start(struct seq_file *seq, loff_t *pos)
	__acquires(RCU_BH)
{
	struct net *net = seq_file_net(seq);
	struct ipv6_route_iter *iter = seq->private;

	rcu_read_lock_bh();
	iter->tbl = ipv6_route_seq_next_table(NULL, net);
	iter->skip = *pos;

	if (iter->tbl) {
M
Michal Kubeček 已提交
2384
		ipv6_route_seq_setup_walk(iter, net);
2385 2386 2387 2388 2389 2390 2391 2392
		return ipv6_route_seq_next(seq, NULL, pos);
	} else {
		return NULL;
	}
}

static bool ipv6_route_iter_active(struct ipv6_route_iter *iter)
{
2393
	struct fib6_walker *w = &iter->w;
2394 2395 2396 2397 2398 2399
	return w->node && !(w->state == FWS_U && w->node == w->root);
}

static void ipv6_route_seq_stop(struct seq_file *seq, void *v)
	__releases(RCU_BH)
{
M
Michal Kubeček 已提交
2400
	struct net *net = seq_file_net(seq);
2401 2402 2403
	struct ipv6_route_iter *iter = seq->private;

	if (ipv6_route_iter_active(iter))
M
Michal Kubeček 已提交
2404
		fib6_walker_unlink(net, &iter->w);
2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422

	rcu_read_unlock_bh();
}

static const struct seq_operations ipv6_route_seq_ops = {
	.start	= ipv6_route_seq_start,
	.next	= ipv6_route_seq_next,
	.stop	= ipv6_route_seq_stop,
	.show	= ipv6_route_seq_show
};

int ipv6_route_open(struct inode *inode, struct file *file)
{
	return seq_open_net(inode, file, &ipv6_route_seq_ops,
			    sizeof(struct ipv6_route_iter));
}

#endif /* CONFIG_PROC_FS */