ip6_fib.c 53.6 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;
L
Linus Torvalds 已提交
46
	int (*func)(struct rt6_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 58 59 60 61 62
static struct rt6_info *fib6_find_prefix(struct net *net,
					 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 109 110 111 112 113
void fib6_update_sernum(struct rt6_info *rt)
{
	struct fib6_table *table = rt->rt6i_table;
	struct net *net = dev_net(rt->dst.dev);
	struct fib6_node *fn;

114
	spin_lock_bh(&table->tb6_lock);
115 116 117 118
	fn = rcu_dereference_protected(rt->rt6i_node,
			lockdep_is_held(&table->tb6_lock));
	if (fn)
		fn->fn_sernum = fib6_new_sernum(net);
119
	spin_unlock_bh(&table->tb6_lock);
120 121
}

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

/*
 *	test bit
 */
132 133 134 135 136
#if defined(__LITTLE_ENDIAN)
# define BITOP_BE32_SWIZZLE	(0x1F & ~7)
#else
# define BITOP_BE32_SWIZZLE	0
#endif
L
Linus Torvalds 已提交
137

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

W
Wei Wang 已提交
152
static struct fib6_node *node_alloc(struct net *net)
L
Linus Torvalds 已提交
153 154 155
{
	struct fib6_node *fn;

156
	fn = kmem_cache_zalloc(fib6_node_kmem, GFP_ATOMIC);
W
Wei Wang 已提交
157 158
	if (fn)
		net->ipv6.rt6_stats->fib_nodes++;
L
Linus Torvalds 已提交
159 160 161 162

	return fn;
}

W
Wei Wang 已提交
163
static void node_free_immediate(struct net *net, struct fib6_node *fn)
164 165
{
	kmem_cache_free(fib6_node_kmem, fn);
W
Wei Wang 已提交
166
	net->ipv6.rt6_stats->fib_nodes--;
167 168 169
}

static void node_free_rcu(struct rcu_head *head)
L
Linus Torvalds 已提交
170
{
171 172
	struct fib6_node *fn = container_of(head, struct fib6_node, rcu);

L
Linus Torvalds 已提交
173 174 175
	kmem_cache_free(fib6_node_kmem, fn);
}

W
Wei Wang 已提交
176
static void node_free(struct net *net, struct fib6_node *fn)
177 178
{
	call_rcu(&fn->rcu, node_free_rcu);
W
Wei Wang 已提交
179
	net->ipv6.rt6_stats->fib_nodes--;
180 181
}

182
void rt6_free_pcpu(struct rt6_info *non_pcpu_rt)
M
Martin KaFai Lau 已提交
183 184 185 186 187 188 189 190 191 192 193 194 195
{
	int cpu;

	if (!non_pcpu_rt->rt6i_pcpu)
		return;

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

		ppcpu_rt = per_cpu_ptr(non_pcpu_rt->rt6i_pcpu, cpu);
		pcpu_rt = *ppcpu_rt;
		if (pcpu_rt) {
W
Wei Wang 已提交
196
			dst_dev_put(&pcpu_rt->dst);
197
			dst_release(&pcpu_rt->dst);
M
Martin KaFai Lau 已提交
198 199 200 201
			*ppcpu_rt = NULL;
		}
	}
}
202
EXPORT_SYMBOL_GPL(rt6_free_pcpu);
L
Linus Torvalds 已提交
203

204 205 206 207 208 209
static void fib6_free_table(struct fib6_table *table)
{
	inetpeer_invalidate_tree(&table->tb6_peers);
	kfree(table);
}

210
static void fib6_link_table(struct net *net, struct fib6_table *tb)
211 212 213
{
	unsigned int h;

214 215 216 217
	/*
	 * Initialize table lock at a single place to give lockdep a key,
	 * tables aren't visible prior to being linked to the list.
	 */
218
	spin_lock_init(&tb->tb6_lock);
219
	h = tb->tb6_id & (FIB6_TABLE_HASHSZ - 1);
220 221 222 223 224

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

228
#ifdef CONFIG_IPV6_MULTIPLE_TABLES
229

230
static struct fib6_table *fib6_alloc_table(struct net *net, u32 id)
T
Thomas Graf 已提交
231 232 233 234
{
	struct fib6_table *table;

	table = kzalloc(sizeof(*table), GFP_ATOMIC);
235
	if (table) {
T
Thomas Graf 已提交
236
		table->tb6_id = id;
237 238
		rcu_assign_pointer(table->tb6_root.leaf,
				   net->ipv6.ip6_null_entry);
T
Thomas Graf 已提交
239
		table->tb6_root.fn_flags = RTN_ROOT | RTN_TL_ROOT | RTN_RTINFO;
240
		inet_peer_base_init(&table->tb6_peers);
T
Thomas Graf 已提交
241 242 243 244 245
	}

	return table;
}

246
struct fib6_table *fib6_new_table(struct net *net, u32 id)
T
Thomas Graf 已提交
247 248 249 250 251
{
	struct fib6_table *tb;

	if (id == 0)
		id = RT6_TABLE_MAIN;
252
	tb = fib6_get_table(net, id);
T
Thomas Graf 已提交
253 254 255
	if (tb)
		return tb;

256
	tb = fib6_alloc_table(net, id);
257
	if (tb)
258
		fib6_link_table(net, tb);
T
Thomas Graf 已提交
259 260 261

	return tb;
}
262
EXPORT_SYMBOL_GPL(fib6_new_table);
T
Thomas Graf 已提交
263

264
struct fib6_table *fib6_get_table(struct net *net, u32 id)
T
Thomas Graf 已提交
265 266
{
	struct fib6_table *tb;
267
	struct hlist_head *head;
T
Thomas Graf 已提交
268 269 270 271
	unsigned int h;

	if (id == 0)
		id = RT6_TABLE_MAIN;
272
	h = id & (FIB6_TABLE_HASHSZ - 1);
T
Thomas Graf 已提交
273
	rcu_read_lock();
274
	head = &net->ipv6.fib_table_hash[h];
275
	hlist_for_each_entry_rcu(tb, head, tb6_hlist) {
T
Thomas Graf 已提交
276 277 278 279 280 281 282 283 284
		if (tb->tb6_id == id) {
			rcu_read_unlock();
			return tb;
		}
	}
	rcu_read_unlock();

	return NULL;
}
285
EXPORT_SYMBOL_GPL(fib6_get_table);
T
Thomas Graf 已提交
286

287
static void __net_init fib6_tables_init(struct net *net)
T
Thomas Graf 已提交
288
{
289 290
	fib6_link_table(net, net->ipv6.fib6_main_tbl);
	fib6_link_table(net, net->ipv6.fib6_local_tbl);
T
Thomas Graf 已提交
291 292 293
}
#else

294
struct fib6_table *fib6_new_table(struct net *net, u32 id)
T
Thomas Graf 已提交
295
{
296
	return fib6_get_table(net, id);
T
Thomas Graf 已提交
297 298
}

299
struct fib6_table *fib6_get_table(struct net *net, u32 id)
T
Thomas Graf 已提交
300
{
301
	  return net->ipv6.fib6_main_tbl;
T
Thomas Graf 已提交
302 303
}

304
struct dst_entry *fib6_rule_lookup(struct net *net, struct flowi6 *fl6,
305
				   int flags, pol_lookup_t lookup)
T
Thomas Graf 已提交
306
{
307 308 309
	struct rt6_info *rt;

	rt = lookup(net, net->ipv6.fib6_main_tbl, fl6, flags);
310
	if (rt->dst.error == -EAGAIN) {
311 312 313 314 315 316
		ip6_rt_put(rt);
		rt = net->ipv6.ip6_null_entry;
		dst_hold(&rt->dst);
	}

	return &rt->dst;
T
Thomas Graf 已提交
317 318
}

319
static void __net_init fib6_tables_init(struct net *net)
T
Thomas Graf 已提交
320
{
321
	fib6_link_table(net, net->ipv6.fib6_main_tbl);
T
Thomas Graf 已提交
322 323 324 325
}

#endif

326 327 328 329 330 331 332 333 334
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;

335
		hlist_for_each_entry_rcu(tb, head, tb6_hlist)
336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353
			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,
				    struct rt6_info *rt)
{
	struct fib6_entry_notifier_info info = {
		.rt = rt,
	};

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

354 355
static int call_fib6_entry_notifiers(struct net *net,
				     enum fib_event_type event_type,
D
David Ahern 已提交
356 357
				     struct rt6_info *rt,
				     struct netlink_ext_ack *extack)
358 359
{
	struct fib6_entry_notifier_info info = {
D
David Ahern 已提交
360
		.info.extack = extack,
361 362 363
		.rt = rt,
	};

364
	rt->rt6i_table->fib_seq++;
365 366 367
	return call_fib6_notifiers(net, event_type, &info.info);
}

368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383
struct fib6_dump_arg {
	struct net *net;
	struct notifier_block *nb;
};

static void fib6_rt_dump(struct rt6_info *rt, struct fib6_dump_arg *arg)
{
	if (rt == arg->net->ipv6.ip6_null_entry)
		return;
	call_fib6_entry_notifier(arg->nb, arg->net, FIB_EVENT_ENTRY_ADD, rt);
}

static int fib6_node_dump(struct fib6_walker *w)
{
	struct rt6_info *rt;

384
	for_each_fib6_walker_rt(w)
385 386 387 388 389 390 391 392 393
		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;
394
	spin_lock_bh(&tb->tb6_lock);
395
	fib6_walk(net, w);
396
	spin_unlock_bh(&tb->tb6_lock);
397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427
}

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

428
static int fib6_dump_node(struct fib6_walker *w)
429 430 431 432
{
	int res;
	struct rt6_info *rt;

433
	for_each_fib6_walker_rt(w) {
434 435 436 437 438 439
		res = rt6_dump_route(rt, w->args);
		if (res < 0) {
			/* Frame is full, suspend walking */
			w->leaf = rt;
			return 1;
		}
440 441 442 443 444 445 446 447 448 449

		/* 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)
		 */
		if (rt->rt6i_nsiblings)
			rt = list_last_entry(&rt->rt6i_siblings,
					     struct rt6_info,
					     rt6i_siblings);
450 451 452 453 454 455 456
	}
	w->leaf = NULL;
	return 0;
}

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

	if (w) {
461 462
		if (cb->args[4]) {
			cb->args[4] = 0;
M
Michal Kubeček 已提交
463
			fib6_walker_unlink(net, w);
464
		}
465 466 467
		cb->args[2] = 0;
		kfree(w);
	}
468
	cb->done = (void *)cb->args[3];
469 470 471 472 473 474 475 476 477 478 479 480
	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 已提交
481
	struct net *net = sock_net(skb->sk);
482
	struct fib6_walker *w;
483 484 485 486 487 488
	int res;

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

	if (cb->args[4] == 0) {
489 490 491
		w->count = 0;
		w->skip = 0;

492
		spin_lock_bh(&table->tb6_lock);
M
Michal Kubeček 已提交
493
		res = fib6_walk(net, w);
494
		spin_unlock_bh(&table->tb6_lock);
495
		if (res > 0) {
496
			cb->args[4] = 1;
497 498
			cb->args[5] = w->root->fn_sernum;
		}
499
	} else {
500 501 502 503 504 505 506 507 508
		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;

509
		spin_lock_bh(&table->tb6_lock);
510
		res = fib6_walk_continue(w);
511
		spin_unlock_bh(&table->tb6_lock);
512
		if (res <= 0) {
M
Michal Kubeček 已提交
513
			fib6_walker_unlink(net, w);
514
			cb->args[4] = 0;
515 516
		}
	}
517

518 519 520
	return res;
}

521
static int inet6_dump_fib(struct sk_buff *skb, struct netlink_callback *cb)
522
{
523
	struct net *net = sock_net(skb->sk);
524 525 526
	unsigned int h, s_h;
	unsigned int e = 0, s_e;
	struct rt6_rtnl_dump_arg arg;
527
	struct fib6_walker *w;
528
	struct fib6_table *tb;
529
	struct hlist_head *head;
530 531 532 533 534 535
	int res = 0;

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

	w = (void *)cb->args[2];
536
	if (!w) {
537 538 539 540 541 542 543 544 545 546 547
		/* 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);
548
		if (!w)
549 550 551 552 553 554 555
			return -ENOMEM;
		w->func = fib6_dump_node;
		cb->args[2] = (long)w;
	}

	arg.skb = skb;
	arg.cb = cb;
556
	arg.net = net;
557 558
	w->args = &arg;

559
	rcu_read_lock();
560
	for (h = s_h; h < FIB6_TABLE_HASHSZ; h++, s_e = 0) {
561
		e = 0;
562
		head = &net->ipv6.fib_table_hash[h];
563
		hlist_for_each_entry_rcu(tb, head, tb6_hlist) {
564 565 566 567 568 569 570 571 572 573
			if (e < s_e)
				goto next;
			res = fib6_dump_table(tb, skb, cb);
			if (res != 0)
				goto out;
next:
			e++;
		}
	}
out:
574
	rcu_read_unlock();
575 576 577 578 579 580 581 582
	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 已提交
583 584 585 586 587 588 589 590 591

/*
 *	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 已提交
592 593
static struct fib6_node *fib6_add_1(struct net *net,
				    struct fib6_table *table,
594 595 596 597 598
				    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 已提交
599 600 601 602 603
{
	struct fib6_node *fn, *in, *ln;
	struct fib6_node *pn = NULL;
	struct rt6key *key;
	int	bit;
604
	__be32	dir = 0;
L
Linus Torvalds 已提交
605 606 607 608 609 610 611 612

	RT6_TRACE("fib6_add_1\n");

	/* insert node in tree */

	fn = root;

	do {
613 614 615
		struct rt6_info *leaf = rcu_dereference_protected(fn->leaf,
					    lockdep_is_held(&table->tb6_lock));
		key = (struct rt6key *)((u8 *)leaf + offset);
L
Linus Torvalds 已提交
616 617 618 619 620

		/*
		 *	Prefix match
		 */
		if (plen < fn->fn_bit ||
621
		    !ipv6_prefix_equal(&key->addr, addr, fn->fn_bit)) {
622 623
			if (!allow_create) {
				if (replace_required) {
624 625
					NL_SET_ERR_MSG(extack,
						       "Can not replace route - no match found");
626
					pr_warn("Can't replace route, no match found\n");
627 628
					return ERR_PTR(-ENOENT);
				}
629
				pr_warn("NLM_F_CREATE should be set when creating new route\n");
630
			}
L
Linus Torvalds 已提交
631
			goto insert_above;
632
		}
633

L
Linus Torvalds 已提交
634 635 636
		/*
		 *	Exact match ?
		 */
637

L
Linus Torvalds 已提交
638 639
		if (plen == fn->fn_bit) {
			/* clean up an intermediate node */
640
			if (!(fn->fn_flags & RTN_RTINFO)) {
641 642
				RCU_INIT_POINTER(fn->leaf, NULL);
				rt6_release(leaf);
L
Linus Torvalds 已提交
643
			}
644

L
Linus Torvalds 已提交
645 646 647 648 649 650
			return fn;
		}

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

L
Linus Torvalds 已提交
652 653 654
		/* Try to walk down on tree. */
		dir = addr_bit_set(addr, fn->fn_bit);
		pn = fn;
655 656 657 658 659
		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 已提交
660 661
	} while (fn);

662
	if (!allow_create) {
663 664 665 666 667 668 669 670 671
		/* 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
		 */
672
		if (replace_required) {
673 674
			NL_SET_ERR_MSG(extack,
				       "Can not replace route - no match found");
675
			pr_warn("Can't replace route, no match found\n");
676 677
			return ERR_PTR(-ENOENT);
		}
678
		pr_warn("NLM_F_CREATE should be set when creating new route\n");
679
	}
L
Linus Torvalds 已提交
680 681 682 683 684
	/*
	 *	We walked to the bottom of tree.
	 *	Create new leaf node without children.
	 */

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

687
	if (!ln)
688
		return ERR_PTR(-ENOMEM);
L
Linus Torvalds 已提交
689
	ln->fn_bit = plen;
690
	RCU_INIT_POINTER(ln->parent, pn);
L
Linus Torvalds 已提交
691 692

	if (dir)
693
		rcu_assign_pointer(pn->right, ln);
L
Linus Torvalds 已提交
694
	else
695
		rcu_assign_pointer(pn->left, ln);
L
Linus Torvalds 已提交
696 697 698 699 700 701

	return ln;


insert_above:
	/*
702
	 * split since we don't have a common prefix anymore or
L
Linus Torvalds 已提交
703 704 705 706 707 708
	 * 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
	 */

709 710
	pn = rcu_dereference_protected(fn->parent,
				       lockdep_is_held(&table->tb6_lock));
L
Linus Torvalds 已提交
711 712 713

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

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

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

720 721
	/*
	 *		(intermediate)[in]
L
Linus Torvalds 已提交
722 723 724 725
	 *	          /	   \
	 *	(new leaf node)[ln] (old node)[fn]
	 */
	if (plen > bit) {
W
Wei Wang 已提交
726 727
		in = node_alloc(net);
		ln = node_alloc(net);
728

729
		if (!in || !ln) {
L
Linus Torvalds 已提交
730
			if (in)
W
Wei Wang 已提交
731
				node_free_immediate(net, in);
L
Linus Torvalds 已提交
732
			if (ln)
W
Wei Wang 已提交
733
				node_free_immediate(net, ln);
734
			return ERR_PTR(-ENOMEM);
L
Linus Torvalds 已提交
735 736
		}

737 738
		/*
		 * new intermediate node.
L
Linus Torvalds 已提交
739 740 741 742 743 744 745 746
		 * 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;

747
		RCU_INIT_POINTER(in->parent, pn);
L
Linus Torvalds 已提交
748
		in->leaf = fn->leaf;
749 750
		atomic_inc(&rcu_dereference_protected(in->leaf,
				lockdep_is_held(&table->tb6_lock))->rt6i_ref);
L
Linus Torvalds 已提交
751 752 753

		/* update parent pointer */
		if (dir)
754
			rcu_assign_pointer(pn->right, in);
L
Linus Torvalds 已提交
755
		else
756
			rcu_assign_pointer(pn->left, in);
L
Linus Torvalds 已提交
757 758 759

		ln->fn_bit = plen;

760 761
		RCU_INIT_POINTER(ln->parent, in);
		rcu_assign_pointer(fn->parent, in);
L
Linus Torvalds 已提交
762 763

		if (addr_bit_set(addr, bit)) {
764 765
			rcu_assign_pointer(in->right, ln);
			rcu_assign_pointer(in->left, fn);
L
Linus Torvalds 已提交
766
		} else {
767 768
			rcu_assign_pointer(in->left, ln);
			rcu_assign_pointer(in->right, fn);
L
Linus Torvalds 已提交
769 770 771
		}
	} else { /* plen <= bit */

772
		/*
L
Linus Torvalds 已提交
773 774 775 776 777
		 *		(new leaf node)[ln]
		 *	          /	   \
		 *	     (old node)[fn] NULL
		 */

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

780
		if (!ln)
781
			return ERR_PTR(-ENOMEM);
L
Linus Torvalds 已提交
782 783 784

		ln->fn_bit = plen;

785
		RCU_INIT_POINTER(ln->parent, pn);
L
Linus Torvalds 已提交
786 787

		if (addr_bit_set(&key->addr, plen))
788
			RCU_INIT_POINTER(ln->right, fn);
L
Linus Torvalds 已提交
789
		else
790 791 792
			RCU_INIT_POINTER(ln->left, fn);

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

794 795 796 797
		if (dir)
			rcu_assign_pointer(pn->right, ln);
		else
			rcu_assign_pointer(pn->left, ln);
L
Linus Torvalds 已提交
798 799 800 801
	}
	return ln;
}

802
static bool rt6_qualify_for_ecmp(struct rt6_info *rt)
803 804 805 806 807
{
	return (rt->rt6i_flags & (RTF_GATEWAY|RTF_ADDRCONF|RTF_DYNAMIC)) ==
	       RTF_GATEWAY;
}

808
static void fib6_copy_metrics(u32 *mp, const struct mx6_config *mxc)
809
{
810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834
	int i;

	for (i = 0; i < RTAX_MAX; i++) {
		if (test_bit(i, mxc->mx_valid))
			mp[i] = mxc->mx[i];
	}
}

static int fib6_commit_metrics(struct dst_entry *dst, struct mx6_config *mxc)
{
	if (!mxc->mx)
		return 0;

	if (dst->flags & DST_HOST) {
		u32 *mp = dst_metrics_write_ptr(dst);

		if (unlikely(!mp))
			return -ENOMEM;

		fib6_copy_metrics(mp, mxc);
	} else {
		dst_init_metrics(dst, mxc->mx, false);

		/* We've stolen mx now. */
		mxc->mx = NULL;
835
	}
836

837 838 839
	return 0;
}

840 841 842
static void fib6_purge_rt(struct rt6_info *rt, struct fib6_node *fn,
			  struct net *net)
{
843 844
	struct fib6_table *table = rt->rt6i_table;

845 846 847 848 849 850 851 852
	if (atomic_read(&rt->rt6i_ref) != 1) {
		/* 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) {
853 854 855 856 857 858 859
			struct rt6_info *leaf = rcu_dereference_protected(fn->leaf,
					    lockdep_is_held(&table->tb6_lock));
			struct rt6_info *new_leaf;
			if (!(fn->fn_flags & RTN_RTINFO) && leaf == rt) {
				new_leaf = fib6_find_prefix(net, table, fn);
				atomic_inc(&new_leaf->rt6i_ref);
				rcu_assign_pointer(fn->leaf, new_leaf);
860 861
				rt6_release(rt);
			}
862 863
			fn = rcu_dereference_protected(fn->parent,
				    lockdep_is_held(&table->tb6_lock));
864 865 866 867
		}
	}
}

L
Linus Torvalds 已提交
868 869 870 871 872
/*
 *	Insert routing information in a node.
 */

static int fib6_add_rt2node(struct fib6_node *fn, struct rt6_info *rt,
D
David Ahern 已提交
873 874
			    struct nl_info *info, struct mx6_config *mxc,
			    struct netlink_ext_ack *extack)
L
Linus Torvalds 已提交
875
{
876 877
	struct rt6_info *leaf = rcu_dereference_protected(fn->leaf,
				    lockdep_is_held(&rt->rt6i_table->tb6_lock));
L
Linus Torvalds 已提交
878
	struct rt6_info *iter = NULL;
879 880
	struct rt6_info __rcu **ins;
	struct rt6_info __rcu **fallback_ins = NULL;
881 882 883 884
	int replace = (info->nlh &&
		       (info->nlh->nlmsg_flags & NLM_F_REPLACE));
	int add = (!info->nlh ||
		   (info->nlh->nlmsg_flags & NLM_F_CREATE));
885
	int found = 0;
886
	bool rt_can_ecmp = rt6_qualify_for_ecmp(rt);
887
	u16 nlflags = NLM_F_EXCL;
888
	int err;
L
Linus Torvalds 已提交
889

890 891 892
	if (info->nlh && (info->nlh->nlmsg_flags & NLM_F_APPEND))
		nlflags |= NLM_F_APPEND;

L
Linus Torvalds 已提交
893 894
	ins = &fn->leaf;

895 896 897
	for (iter = leaf; iter;
	     iter = rcu_dereference_protected(iter->dst.rt6_next,
				lockdep_is_held(&rt->rt6i_table->tb6_lock))) {
L
Linus Torvalds 已提交
898 899 900 901 902 903 904 905
		/*
		 *	Search for duplicates
		 */

		if (iter->rt6i_metric == rt->rt6i_metric) {
			/*
			 *	Same priority level
			 */
906 907
			if (info->nlh &&
			    (info->nlh->nlmsg_flags & NLM_F_EXCL))
908
				return -EEXIST;
909 910

			nlflags &= ~NLM_F_EXCL;
911
			if (replace) {
912 913 914 915 916 917 918
				if (rt_can_ecmp == rt6_qualify_for_ecmp(iter)) {
					found++;
					break;
				}
				if (rt_can_ecmp)
					fallback_ins = fallback_ins ?: ins;
				goto next_iter;
919
			}
L
Linus Torvalds 已提交
920

921
			if (rt6_duplicate_nexthop(iter, rt)) {
922 923
				if (rt->rt6i_nsiblings)
					rt->rt6i_nsiblings = 0;
924
				if (!(iter->rt6i_flags & RTF_EXPIRES))
L
Linus Torvalds 已提交
925
					return -EEXIST;
926 927 928 929
				if (!(rt->rt6i_flags & RTF_EXPIRES))
					rt6_clean_expires(iter);
				else
					rt6_set_expires(iter, rt->dst.expires);
930
				iter->rt6i_pmtu = rt->rt6i_pmtu;
L
Linus Torvalds 已提交
931 932
				return -EEXIST;
			}
933 934 935 936 937 938 939 940 941 942 943
			/* 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.
			 */
944 945
			if (rt_can_ecmp &&
			    rt6_qualify_for_ecmp(iter))
946
				rt->rt6i_nsiblings++;
L
Linus Torvalds 已提交
947 948 949 950 951
		}

		if (iter->rt6i_metric > rt->rt6i_metric)
			break;

952
next_iter:
953
		ins = &iter->dst.rt6_next;
L
Linus Torvalds 已提交
954 955
	}

956 957 958
	if (fallback_ins && !found) {
		/* No ECMP-able route found, replace first non-ECMP one */
		ins = fallback_ins;
959 960
		iter = rcu_dereference_protected(*ins,
				    lockdep_is_held(&rt->rt6i_table->tb6_lock));
961 962 963
		found++;
	}

964 965 966 967
	/* Reset round-robin state, if necessary */
	if (ins == &fn->leaf)
		fn->rr_ptr = NULL;

968 969 970 971 972 973
	/* Link this route to others same route. */
	if (rt->rt6i_nsiblings) {
		unsigned int rt6i_nsiblings;
		struct rt6_info *sibling, *temp_sibling;

		/* Find the first route that have the same metric */
974
		sibling = leaf;
975
		while (sibling) {
976 977
			if (sibling->rt6i_metric == rt->rt6i_metric &&
			    rt6_qualify_for_ecmp(sibling)) {
978 979 980 981
				list_add_tail(&rt->rt6i_siblings,
					      &sibling->rt6i_siblings);
				break;
			}
982 983
			sibling = rcu_dereference_protected(sibling->dst.rt6_next,
				    lockdep_is_held(&rt->rt6i_table->tb6_lock));
984 985 986 987 988 989 990 991 992 993 994 995 996 997 998
		}
		/* For each sibling in the list, increment the counter of
		 * siblings. BUG() if counters does not match, list of siblings
		 * is broken!
		 */
		rt6i_nsiblings = 0;
		list_for_each_entry_safe(sibling, temp_sibling,
					 &rt->rt6i_siblings, rt6i_siblings) {
			sibling->rt6i_nsiblings++;
			BUG_ON(sibling->rt6i_nsiblings != rt->rt6i_nsiblings);
			rt6i_nsiblings++;
		}
		BUG_ON(rt6i_nsiblings != rt->rt6i_nsiblings);
	}

L
Linus Torvalds 已提交
999 1000 1001
	/*
	 *	insert node
	 */
1002 1003
	if (!replace) {
		if (!add)
1004
			pr_warn("NLM_F_CREATE should be set when creating new route\n");
1005 1006

add:
1007
		nlflags |= NLM_F_CREATE;
1008 1009 1010 1011
		err = fib6_commit_metrics(&rt->dst, mxc);
		if (err)
			return err;

1012
		rcu_assign_pointer(rt->dst.rt6_next, iter);
1013
		atomic_inc(&rt->rt6i_ref);
1014 1015
		rcu_assign_pointer(rt->rt6i_node, fn);
		rcu_assign_pointer(*ins, rt);
1016
		call_fib6_entry_notifiers(info->nl_net, FIB_EVENT_ENTRY_ADD,
D
David Ahern 已提交
1017
					  rt, extack);
1018 1019
		if (!info->skip_notify)
			inet6_rt_notify(RTM_NEWROUTE, rt, info, nlflags);
1020 1021
		info->nl_net->ipv6.rt6_stats->fib_rt_entries++;

1022
		if (!(fn->fn_flags & RTN_RTINFO)) {
1023 1024 1025
			info->nl_net->ipv6.rt6_stats->fib_route_nodes++;
			fn->fn_flags |= RTN_RTINFO;
		}
L
Linus Torvalds 已提交
1026

1027
	} else {
1028 1029
		int nsiblings;

1030 1031 1032
		if (!found) {
			if (add)
				goto add;
1033
			pr_warn("NLM_F_REPLACE set, but no existing node found!\n");
1034 1035
			return -ENOENT;
		}
1036 1037 1038 1039 1040

		err = fib6_commit_metrics(&rt->dst, mxc);
		if (err)
			return err;

1041
		atomic_inc(&rt->rt6i_ref);
W
Wei Wang 已提交
1042
		rcu_assign_pointer(rt->rt6i_node, fn);
1043
		rt->dst.rt6_next = iter->dst.rt6_next;
1044
		rcu_assign_pointer(*ins, rt);
1045
		call_fib6_entry_notifiers(info->nl_net, FIB_EVENT_ENTRY_REPLACE,
D
David Ahern 已提交
1046
					  rt, extack);
1047 1048
		if (!info->skip_notify)
			inet6_rt_notify(RTM_NEWROUTE, rt, info, NLM_F_REPLACE);
1049
		if (!(fn->fn_flags & RTN_RTINFO)) {
1050 1051 1052
			info->nl_net->ipv6.rt6_stats->fib_route_nodes++;
			fn->fn_flags |= RTN_RTINFO;
		}
1053
		nsiblings = iter->rt6i_nsiblings;
1054
		iter->rt6i_node = NULL;
1055
		fib6_purge_rt(iter, fn, info->nl_net);
1056
		if (rcu_access_pointer(fn->rr_ptr) == iter)
1057
			fn->rr_ptr = NULL;
1058
		rt6_release(iter);
1059 1060 1061 1062

		if (nsiblings) {
			/* Replacing an ECMP route, remove all siblings */
			ins = &rt->dst.rt6_next;
1063 1064
			iter = rcu_dereference_protected(*ins,
				    lockdep_is_held(&rt->rt6i_table->tb6_lock));
1065
			while (iter) {
1066 1067
				if (iter->rt6i_metric > rt->rt6i_metric)
					break;
1068 1069
				if (rt6_qualify_for_ecmp(iter)) {
					*ins = iter->dst.rt6_next;
1070
					iter->rt6i_node = NULL;
1071
					fib6_purge_rt(iter, fn, info->nl_net);
1072
					if (rcu_access_pointer(fn->rr_ptr) == iter)
1073
						fn->rr_ptr = NULL;
1074 1075
					rt6_release(iter);
					nsiblings--;
W
Wei Wang 已提交
1076
					info->nl_net->ipv6.rt6_stats->fib_rt_entries--;
1077 1078 1079
				} else {
					ins = &iter->dst.rt6_next;
				}
1080 1081
				iter = rcu_dereference_protected(*ins,
					lockdep_is_held(&rt->rt6i_table->tb6_lock));
1082 1083 1084
			}
			WARN_ON(nsiblings != 0);
		}
L
Linus Torvalds 已提交
1085 1086 1087 1088 1089
	}

	return 0;
}

1090
static void fib6_start_gc(struct net *net, struct rt6_info *rt)
L
Linus Torvalds 已提交
1091
{
1092
	if (!timer_pending(&net->ipv6.ip6_fib_timer) &&
1093
	    (rt->rt6i_flags & (RTF_EXPIRES | RTF_CACHE)))
1094
		mod_timer(&net->ipv6.ip6_fib_timer,
S
Stephen Hemminger 已提交
1095
			  jiffies + net->ipv6.sysctl.ip6_rt_gc_interval);
L
Linus Torvalds 已提交
1096 1097
}

1098
void fib6_force_start_gc(struct net *net)
L
Linus Torvalds 已提交
1099
{
1100 1101
	if (!timer_pending(&net->ipv6.ip6_fib_timer))
		mod_timer(&net->ipv6.ip6_fib_timer,
S
Stephen Hemminger 已提交
1102
			  jiffies + net->ipv6.sysctl.ip6_rt_gc_interval);
L
Linus Torvalds 已提交
1103 1104
}

1105 1106 1107 1108 1109 1110 1111 1112 1113 1114
static void fib6_update_sernum_upto_root(struct rt6_info *rt,
					 int sernum)
{
	struct fib6_node *fn = rcu_dereference_protected(rt->rt6i_node,
				lockdep_is_held(&rt->rt6i_table->tb6_lock));

	/* paired with smp_rmb() in rt6_get_cookie_safe() */
	smp_wmb();
	while (fn) {
		fn->fn_sernum = sernum;
1115 1116
		fn = rcu_dereference_protected(fn->parent,
				lockdep_is_held(&rt->rt6i_table->tb6_lock));
1117 1118 1119
	}
}

L
Linus Torvalds 已提交
1120 1121 1122 1123
/*
 *	Add routing information to the routing tree.
 *	<destination addr>/<source addr>
 *	with source addr info in sub-trees
1124
 *	Need to own table->tb6_lock
L
Linus Torvalds 已提交
1125 1126
 */

1127
int fib6_add(struct fib6_node *root, struct rt6_info *rt,
1128 1129
	     struct nl_info *info, struct mx6_config *mxc,
	     struct netlink_ext_ack *extack)
L
Linus Torvalds 已提交
1130
{
1131
	struct fib6_table *table = rt->rt6i_table;
1132
	struct fib6_node *fn, *pn = NULL;
L
Linus Torvalds 已提交
1133
	int err = -ENOMEM;
1134 1135
	int allow_create = 1;
	int replace_required = 0;
1136
	int sernum = fib6_new_sernum(info->nl_net);
1137

W
Wei Wang 已提交
1138
	if (WARN_ON_ONCE(!atomic_read(&rt->dst.__refcnt)))
M
Martin KaFai Lau 已提交
1139
		return -EINVAL;
1140 1141
	if (WARN_ON_ONCE(rt->rt6i_flags & RTF_CACHE))
		return -EINVAL;
M
Martin KaFai Lau 已提交
1142

1143 1144
	if (info->nlh) {
		if (!(info->nlh->nlmsg_flags & NLM_F_CREATE))
1145
			allow_create = 0;
1146
		if (info->nlh->nlmsg_flags & NLM_F_REPLACE)
1147 1148 1149
			replace_required = 1;
	}
	if (!allow_create && !replace_required)
1150
		pr_warn("RTM_NEWROUTE with no NLM_F_CREATE or NLM_F_REPLACE\n");
L
Linus Torvalds 已提交
1151

W
Wei Wang 已提交
1152
	fn = fib6_add_1(info->nl_net, table, root,
1153
			&rt->rt6i_dst.addr, rt->rt6i_dst.plen,
1154
			offsetof(struct rt6_info, rt6i_dst), allow_create,
1155
			replace_required, extack);
1156 1157
	if (IS_ERR(fn)) {
		err = PTR_ERR(fn);
1158
		fn = NULL;
L
Linus Torvalds 已提交
1159
		goto out;
1160
	}
L
Linus Torvalds 已提交
1161

1162 1163
	pn = fn;

L
Linus Torvalds 已提交
1164 1165 1166 1167
#ifdef CONFIG_IPV6_SUBTREES
	if (rt->rt6i_src.plen) {
		struct fib6_node *sn;

1168
		if (!rcu_access_pointer(fn->subtree)) {
L
Linus Torvalds 已提交
1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181
			struct fib6_node *sfn;

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

			/* Create subtree root node */
W
Wei Wang 已提交
1182
			sfn = node_alloc(info->nl_net);
1183
			if (!sfn)
1184
				goto failure;
L
Linus Torvalds 已提交
1185

1186
			atomic_inc(&info->nl_net->ipv6.ip6_null_entry->rt6i_ref);
1187 1188
			rcu_assign_pointer(sfn->leaf,
					   info->nl_net->ipv6.ip6_null_entry);
L
Linus Torvalds 已提交
1189 1190 1191 1192
			sfn->fn_flags = RTN_ROOT;

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

W
Wei Wang 已提交
1193 1194
			sn = fib6_add_1(info->nl_net, table, sfn,
					&rt->rt6i_src.addr, rt->rt6i_src.plen,
1195
					offsetof(struct rt6_info, rt6i_src),
1196
					allow_create, replace_required, extack);
L
Linus Torvalds 已提交
1197

1198
			if (IS_ERR(sn)) {
L
Linus Torvalds 已提交
1199
				/* If it is failed, discard just allocated
1200
				   root, and then (in failure) stale node
L
Linus Torvalds 已提交
1201 1202
				   in main tree.
				 */
W
Wei Wang 已提交
1203
				node_free_immediate(info->nl_net, sfn);
1204
				err = PTR_ERR(sn);
1205
				goto failure;
L
Linus Torvalds 已提交
1206 1207 1208
			}

			/* Now link new subtree to main tree */
1209 1210
			rcu_assign_pointer(sfn->parent, fn);
			rcu_assign_pointer(fn->subtree, sfn);
L
Linus Torvalds 已提交
1211
		} else {
W
Wei Wang 已提交
1212 1213
			sn = fib6_add_1(info->nl_net, table, FIB6_SUBTREE(fn),
					&rt->rt6i_src.addr, rt->rt6i_src.plen,
1214
					offsetof(struct rt6_info, rt6i_src),
1215
					allow_create, replace_required, extack);
L
Linus Torvalds 已提交
1216

1217 1218
			if (IS_ERR(sn)) {
				err = PTR_ERR(sn);
1219
				goto failure;
1220
			}
L
Linus Torvalds 已提交
1221 1222
		}

1223
		if (!rcu_access_pointer(fn->leaf)) {
1224
			atomic_inc(&rt->rt6i_ref);
1225
			rcu_assign_pointer(fn->leaf, rt);
1226
		}
L
Linus Torvalds 已提交
1227 1228 1229 1230
		fn = sn;
	}
#endif

D
David Ahern 已提交
1231
	err = fib6_add_rt2node(fn, rt, info, mxc, extack);
1232 1233
	if (!err) {
		fib6_update_sernum_upto_root(rt, sernum);
1234
		fib6_start_gc(info->nl_net, rt);
1235
	}
L
Linus Torvalds 已提交
1236 1237

out:
1238 1239 1240 1241 1242 1243
	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.
		 */
1244 1245 1246 1247 1248
		struct rt6_info *pn_leaf = rcu_dereference_protected(pn->leaf,
					    lockdep_is_held(&table->tb6_lock));
		if (pn != fn && pn_leaf == rt) {
			pn_leaf = NULL;
			RCU_INIT_POINTER(pn->leaf, NULL);
1249 1250
			atomic_dec(&rt->rt6i_ref);
		}
1251 1252
		if (pn != fn && !pn_leaf && !(pn->fn_flags & RTN_RTINFO)) {
			pn_leaf = fib6_find_prefix(info->nl_net, table, pn);
1253
#if RT6_DEBUG >= 2
1254 1255 1256
			if (!pn_leaf) {
				WARN_ON(!pn_leaf);
				pn_leaf = info->nl_net->ipv6.ip6_null_entry;
1257 1258
			}
#endif
1259 1260
			atomic_inc(&pn_leaf->rt6i_ref);
			rcu_assign_pointer(pn->leaf, pn_leaf);
1261 1262
		}
#endif
1263
		goto failure;
1264
	}
L
Linus Torvalds 已提交
1265 1266
	return err;

1267 1268 1269 1270 1271 1272
failure:
	/* fn->leaf could be NULL if 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.
	 * In both cases, fib6_repair_tree() should be called to fix
	 * fn->leaf.
L
Linus Torvalds 已提交
1273 1274
	 */
	if (fn && !(fn->fn_flags & (RTN_RTINFO|RTN_ROOT)))
1275
		fib6_repair_tree(info->nl_net, table, fn);
1276 1277 1278
	/* Always release dst as dst->__refcnt is guaranteed
	 * to be taken before entering this function
	 */
1279
	dst_release_immediate(&rt->dst);
L
Linus Torvalds 已提交
1280 1281 1282 1283 1284 1285 1286 1287 1288
	return err;
}

/*
 *	Routing tree lookup
 *
 */

struct lookup_args {
1289
	int			offset;		/* key offset on rt6_info	*/
1290
	const struct in6_addr	*addr;		/* search key			*/
L
Linus Torvalds 已提交
1291 1292
};

1293 1294
static struct fib6_node *fib6_lookup_1(struct fib6_node *root,
				       struct lookup_args *args)
L
Linus Torvalds 已提交
1295 1296
{
	struct fib6_node *fn;
A
Al Viro 已提交
1297
	__be32 dir;
L
Linus Torvalds 已提交
1298

1299 1300 1301
	if (unlikely(args->offset == 0))
		return NULL;

L
Linus Torvalds 已提交
1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312
	/*
	 *	Descend on a tree
	 */

	fn = root;

	for (;;) {
		struct fib6_node *next;

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

1313 1314
		next = dir ? rcu_dereference(fn->right) :
			     rcu_dereference(fn->left);
L
Linus Torvalds 已提交
1315 1316 1317 1318 1319 1320 1321 1322

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

1323
	while (fn) {
1324 1325 1326 1327
		struct fib6_node *subtree = FIB6_SUBTREE(fn);

		if (subtree || fn->fn_flags & RTN_RTINFO) {
			struct rt6_info *leaf = rcu_dereference(fn->leaf);
L
Linus Torvalds 已提交
1328 1329
			struct rt6key *key;

W
Wei Wang 已提交
1330 1331 1332 1333
			if (!leaf)
				goto backtrack;

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

1335 1336
			if (ipv6_prefix_equal(&key->addr, args->addr, key->plen)) {
#ifdef CONFIG_IPV6_SUBTREES
1337
				if (subtree) {
1338
					struct fib6_node *sfn;
1339
					sfn = fib6_lookup_1(subtree, args + 1);
1340 1341 1342 1343
					if (!sfn)
						goto backtrack;
					fn = sfn;
				}
1344
#endif
1345
				if (fn->fn_flags & RTN_RTINFO)
1346 1347
					return fn;
			}
L
Linus Torvalds 已提交
1348
		}
1349
backtrack:
1350 1351 1352
		if (fn->fn_flags & RTN_ROOT)
			break;

1353
		fn = rcu_dereference(fn->parent);
L
Linus Torvalds 已提交
1354 1355 1356 1357 1358
	}

	return NULL;
}

1359 1360
/* called with rcu_read_lock() held
 */
1361 1362
struct fib6_node *fib6_lookup(struct fib6_node *root, const struct in6_addr *daddr,
			      const struct in6_addr *saddr)
L
Linus Torvalds 已提交
1363 1364
{
	struct fib6_node *fn;
1365 1366 1367 1368 1369
	struct lookup_args args[] = {
		{
			.offset = offsetof(struct rt6_info, rt6i_dst),
			.addr = daddr,
		},
L
Linus Torvalds 已提交
1370
#ifdef CONFIG_IPV6_SUBTREES
1371 1372 1373 1374
		{
			.offset = offsetof(struct rt6_info, rt6i_src),
			.addr = saddr,
		},
L
Linus Torvalds 已提交
1375
#endif
1376 1377 1378 1379
		{
			.offset = 0,	/* sentinel */
		}
	};
L
Linus Torvalds 已提交
1380

1381
	fn = fib6_lookup_1(root, daddr ? args : args + 1);
1382
	if (!fn || fn->fn_flags & RTN_TL_ROOT)
L
Linus Torvalds 已提交
1383 1384 1385 1386 1387 1388 1389 1390
		fn = root;

	return fn;
}

/*
 *	Get node with specified destination prefix (and source prefix,
 *	if subtrees are used)
1391 1392 1393 1394 1395 1396
 *	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 已提交
1397 1398 1399
 */


1400 1401
static struct fib6_node *fib6_locate_1(struct fib6_node *root,
				       const struct in6_addr *addr,
1402 1403
				       int plen, int offset,
				       bool exact_match)
L
Linus Torvalds 已提交
1404
{
1405
	struct fib6_node *fn, *prev = NULL;
L
Linus Torvalds 已提交
1406 1407

	for (fn = root; fn ; ) {
1408
		struct rt6_info *leaf = rcu_dereference(fn->leaf);
W
Wei Wang 已提交
1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419
		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 已提交
1420 1421 1422 1423 1424 1425

		/*
		 *	Prefix match
		 */
		if (plen < fn->fn_bit ||
		    !ipv6_prefix_equal(&key->addr, addr, fn->fn_bit))
1426
			goto out;
L
Linus Torvalds 已提交
1427 1428 1429 1430

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

1431 1432
		prev = fn;

W
Wei Wang 已提交
1433
next:
L
Linus Torvalds 已提交
1434 1435 1436 1437
		/*
		 *	We have more bits to go
		 */
		if (addr_bit_set(addr, fn->fn_bit))
1438
			fn = rcu_dereference(fn->right);
L
Linus Torvalds 已提交
1439
		else
1440
			fn = rcu_dereference(fn->left);
L
Linus Torvalds 已提交
1441
	}
1442 1443 1444 1445 1446
out:
	if (exact_match)
		return NULL;
	else
		return prev;
L
Linus Torvalds 已提交
1447 1448
}

1449 1450
struct fib6_node *fib6_locate(struct fib6_node *root,
			      const struct in6_addr *daddr, int dst_len,
1451 1452
			      const struct in6_addr *saddr, int src_len,
			      bool exact_match)
L
Linus Torvalds 已提交
1453 1454 1455 1456
{
	struct fib6_node *fn;

	fn = fib6_locate_1(root, daddr, dst_len,
1457 1458
			   offsetof(struct rt6_info, rt6i_dst),
			   exact_match);
L
Linus Torvalds 已提交
1459 1460 1461

#ifdef CONFIG_IPV6_SUBTREES
	if (src_len) {
1462
		WARN_ON(saddr == NULL);
1463 1464 1465 1466 1467
		if (fn) {
			struct fib6_node *subtree = FIB6_SUBTREE(fn);

			if (subtree) {
				fn = fib6_locate_1(subtree, saddr, src_len,
1468 1469
					   offsetof(struct rt6_info, rt6i_src),
					   exact_match);
1470 1471
			}
		}
L
Linus Torvalds 已提交
1472 1473 1474
	}
#endif

1475
	if (fn && fn->fn_flags & RTN_RTINFO)
L
Linus Torvalds 已提交
1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486
		return fn;

	return NULL;
}


/*
 *	Deletion
 *
 */

1487 1488 1489
static struct rt6_info *fib6_find_prefix(struct net *net,
					 struct fib6_table *table,
					 struct fib6_node *fn)
L
Linus Torvalds 已提交
1490
{
1491 1492
	struct fib6_node *child_left, *child_right;

1493
	if (fn->fn_flags & RTN_ROOT)
1494
		return net->ipv6.ip6_null_entry;
L
Linus Torvalds 已提交
1495

1496
	while (fn) {
1497 1498 1499 1500 1501 1502 1503 1504 1505 1506
		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 已提交
1507

1508
		fn = FIB6_SUBTREE(fn);
L
Linus Torvalds 已提交
1509 1510 1511 1512 1513 1514 1515
	}
	return NULL;
}

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

1519
static struct fib6_node *fib6_repair_tree(struct net *net,
1520 1521
					  struct fib6_table *table,
					  struct fib6_node *fn)
L
Linus Torvalds 已提交
1522 1523 1524
{
	int children;
	int nstate;
1525
	struct fib6_node *child;
1526
	struct fib6_walker *w;
L
Linus Torvalds 已提交
1527 1528 1529
	int iter = 0;

	for (;;) {
1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545
		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));
		struct rt6_info *fn_leaf = rcu_dereference_protected(fn->leaf,
					    lockdep_is_held(&table->tb6_lock));
		struct rt6_info *pn_leaf = rcu_dereference_protected(pn->leaf,
					    lockdep_is_held(&table->tb6_lock));
		struct rt6_info *new_fn_leaf;

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

1549 1550
		WARN_ON(fn->fn_flags & RTN_RTINFO);
		WARN_ON(fn->fn_flags & RTN_TL_ROOT);
1551
		WARN_ON(fn_leaf);
L
Linus Torvalds 已提交
1552 1553 1554

		children = 0;
		child = NULL;
1555 1556 1557 1558
		if (fn_r)
			child = fn_r, children |= 1;
		if (fn_l)
			child = fn_l, children |= 2;
L
Linus Torvalds 已提交
1559

1560
		if (children == 3 || FIB6_SUBTREE(fn)
L
Linus Torvalds 已提交
1561 1562
#ifdef CONFIG_IPV6_SUBTREES
		    /* Subtree root (i.e. fn) may have one child */
1563
		    || (children && fn->fn_flags & RTN_ROOT)
L
Linus Torvalds 已提交
1564 1565
#endif
		    ) {
1566
			new_fn_leaf = fib6_find_prefix(net, table, fn);
L
Linus Torvalds 已提交
1567
#if RT6_DEBUG >= 2
1568 1569 1570
			if (!new_fn_leaf) {
				WARN_ON(!new_fn_leaf);
				new_fn_leaf = net->ipv6.ip6_null_entry;
L
Linus Torvalds 已提交
1571 1572
			}
#endif
1573 1574 1575
			atomic_inc(&new_fn_leaf->rt6i_ref);
			rcu_assign_pointer(fn->leaf, new_fn_leaf);
			return pn;
L
Linus Torvalds 已提交
1576 1577 1578
		}

#ifdef CONFIG_IPV6_SUBTREES
1579
		if (FIB6_SUBTREE(pn) == fn) {
1580
			WARN_ON(!(fn->fn_flags & RTN_ROOT));
1581
			RCU_INIT_POINTER(pn->subtree, NULL);
L
Linus Torvalds 已提交
1582 1583
			nstate = FWS_L;
		} else {
1584
			WARN_ON(fn->fn_flags & RTN_ROOT);
L
Linus Torvalds 已提交
1585
#endif
1586 1587 1588 1589
			if (pn_r == fn)
				rcu_assign_pointer(pn->right, child);
			else if (pn_l == fn)
				rcu_assign_pointer(pn->left, child);
L
Linus Torvalds 已提交
1590
#if RT6_DEBUG >= 2
1591 1592
			else
				WARN_ON(1);
L
Linus Torvalds 已提交
1593 1594
#endif
			if (child)
1595
				rcu_assign_pointer(child->parent, pn);
L
Linus Torvalds 已提交
1596 1597 1598 1599 1600
			nstate = FWS_R;
#ifdef CONFIG_IPV6_SUBTREES
		}
#endif

M
Michal Kubeček 已提交
1601 1602
		read_lock(&net->ipv6.fib6_walker_lock);
		FOR_WALKERS(net, w) {
1603
			if (!child) {
1604
				if (w->node == fn) {
L
Linus Torvalds 已提交
1605 1606 1607 1608 1609 1610 1611 1612 1613
					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);
1614
						w->state = w->state >= FWS_R ? FWS_U : FWS_INIT;
L
Linus Torvalds 已提交
1615 1616
					} else {
						RT6_TRACE("W %p adjusted by delnode 2, s=%d\n", w, w->state);
1617
						w->state = w->state >= FWS_C ? FWS_U : FWS_INIT;
L
Linus Torvalds 已提交
1618 1619 1620 1621
					}
				}
			}
		}
M
Michal Kubeček 已提交
1622
		read_unlock(&net->ipv6.fib6_walker_lock);
L
Linus Torvalds 已提交
1623

W
Wei Wang 已提交
1624
		node_free(net, fn);
1625
		if (pn->fn_flags & RTN_RTINFO || FIB6_SUBTREE(pn))
L
Linus Torvalds 已提交
1626 1627
			return pn;

1628 1629
		RCU_INIT_POINTER(pn->leaf, NULL);
		rt6_release(pn_leaf);
L
Linus Torvalds 已提交
1630 1631 1632 1633
		fn = pn;
	}
}

1634 1635
static void fib6_del_route(struct fib6_table *table, struct fib6_node *fn,
			   struct rt6_info __rcu **rtp, struct nl_info *info)
L
Linus Torvalds 已提交
1636
{
1637
	struct fib6_walker *w;
1638 1639
	struct rt6_info *rt = rcu_dereference_protected(*rtp,
				    lockdep_is_held(&table->tb6_lock));
1640
	struct net *net = info->nl_net;
L
Linus Torvalds 已提交
1641 1642 1643

	RT6_TRACE("fib6_del_route\n");

1644 1645
	WARN_ON_ONCE(rt->rt6i_flags & RTF_CACHE);

L
Linus Torvalds 已提交
1646
	/* Unlink it */
1647
	*rtp = rt->dst.rt6_next;
L
Linus Torvalds 已提交
1648
	rt->rt6i_node = NULL;
1649 1650
	net->ipv6.rt6_stats->fib_rt_entries--;
	net->ipv6.rt6_stats->fib_discarded_routes++;
L
Linus Torvalds 已提交
1651

1652 1653 1654
	/* Flush all cached dst in exception table */
	rt6_flush_exceptions(rt);

1655
	/* Reset round-robin state, if necessary */
1656
	if (rcu_access_pointer(fn->rr_ptr) == rt)
1657 1658
		fn->rr_ptr = NULL;

1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669
	/* Remove this entry from other siblings */
	if (rt->rt6i_nsiblings) {
		struct rt6_info *sibling, *next_sibling;

		list_for_each_entry_safe(sibling, next_sibling,
					 &rt->rt6i_siblings, rt6i_siblings)
			sibling->rt6i_nsiblings--;
		rt->rt6i_nsiblings = 0;
		list_del_init(&rt->rt6i_siblings);
	}

L
Linus Torvalds 已提交
1670
	/* Adjust walkers */
M
Michal Kubeček 已提交
1671 1672
	read_lock(&net->ipv6.fib6_walker_lock);
	FOR_WALKERS(net, w) {
L
Linus Torvalds 已提交
1673 1674
		if (w->state == FWS_C && w->leaf == rt) {
			RT6_TRACE("walker %p adjusted by delroute\n", w);
1675 1676
			w->leaf = rcu_dereference_protected(rt->dst.rt6_next,
					    lockdep_is_held(&table->tb6_lock));
1677
			if (!w->leaf)
L
Linus Torvalds 已提交
1678 1679 1680
				w->state = FWS_U;
		}
	}
M
Michal Kubeček 已提交
1681
	read_unlock(&net->ipv6.fib6_walker_lock);
L
Linus Torvalds 已提交
1682 1683

	/* If it was last route, expunge its radix tree node */
1684
	if (!rcu_access_pointer(fn->leaf)) {
L
Linus Torvalds 已提交
1685
		fn->fn_flags &= ~RTN_RTINFO;
1686
		net->ipv6.rt6_stats->fib_route_nodes--;
1687
		fn = fib6_repair_tree(net, table, fn);
L
Linus Torvalds 已提交
1688 1689
	}

1690
	fib6_purge_rt(rt, fn, net);
L
Linus Torvalds 已提交
1691

D
David Ahern 已提交
1692
	call_fib6_entry_notifiers(net, FIB_EVENT_ENTRY_DEL, rt, NULL);
1693 1694
	if (!info->skip_notify)
		inet6_rt_notify(RTM_DELROUTE, rt, info, 0);
L
Linus Torvalds 已提交
1695 1696 1697
	rt6_release(rt);
}

1698
/* Need to own table->tb6_lock */
1699
int fib6_del(struct rt6_info *rt, struct nl_info *info)
L
Linus Torvalds 已提交
1700
{
W
Wei Wang 已提交
1701 1702
	struct fib6_node *fn = rcu_dereference_protected(rt->rt6i_node,
				    lockdep_is_held(&rt->rt6i_table->tb6_lock));
1703
	struct fib6_table *table = rt->rt6i_table;
1704
	struct net *net = info->nl_net;
1705 1706
	struct rt6_info __rcu **rtp;
	struct rt6_info __rcu **rtp_next;
L
Linus Torvalds 已提交
1707 1708

#if RT6_DEBUG >= 2
1709
	if (rt->dst.obsolete > 0) {
1710
		WARN_ON(fn);
L
Linus Torvalds 已提交
1711 1712 1713
		return -ENOENT;
	}
#endif
1714
	if (!fn || rt == net->ipv6.ip6_null_entry)
L
Linus Torvalds 已提交
1715 1716
		return -ENOENT;

1717
	WARN_ON(!(fn->fn_flags & RTN_RTINFO));
L
Linus Torvalds 已提交
1718

1719 1720 1721
	/* remove cached dst from exception table */
	if (rt->rt6i_flags & RTF_CACHE)
		return rt6_remove_exception_rt(rt);
L
Linus Torvalds 已提交
1722 1723 1724 1725 1726

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

1727 1728 1729 1730 1731
	for (rtp = &fn->leaf; *rtp; rtp = rtp_next) {
		struct rt6_info *cur = rcu_dereference_protected(*rtp,
					lockdep_is_held(&table->tb6_lock));
		if (rt == cur) {
			fib6_del_route(table, fn, rtp, info);
L
Linus Torvalds 已提交
1732 1733
			return 0;
		}
1734
		rtp_next = &cur->dst.rt6_next;
L
Linus Torvalds 已提交
1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745
	}
	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,
1746
 *	cleaning tree when a device goes down etc. etc.
L
Linus Torvalds 已提交
1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760
 *
 *	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.
1761 1762
 *
 *	This function is called with tb6_lock held.
L
Linus Torvalds 已提交
1763 1764
 */

1765
static int fib6_walk_continue(struct fib6_walker *w)
L
Linus Torvalds 已提交
1766
{
1767
	struct fib6_node *fn, *pn, *left, *right;
L
Linus Torvalds 已提交
1768

1769 1770 1771
	/* w->root should always be table->tb6_root */
	WARN_ON_ONCE(!(w->root->fn_flags & RTN_TL_ROOT));

L
Linus Torvalds 已提交
1772 1773
	for (;;) {
		fn = w->node;
1774
		if (!fn)
L
Linus Torvalds 已提交
1775 1776 1777 1778 1779
			return 0;

		switch (w->state) {
#ifdef CONFIG_IPV6_SUBTREES
		case FWS_S:
1780 1781
			if (FIB6_SUBTREE(fn)) {
				w->node = FIB6_SUBTREE(fn);
L
Linus Torvalds 已提交
1782 1783 1784
				continue;
			}
			w->state = FWS_L;
1785
#endif
1786
			/* fall through */
L
Linus Torvalds 已提交
1787
		case FWS_L:
1788 1789 1790
			left = rcu_dereference_protected(fn->left, 1);
			if (left) {
				w->node = left;
L
Linus Torvalds 已提交
1791 1792 1793 1794
				w->state = FWS_INIT;
				continue;
			}
			w->state = FWS_R;
1795
			/* fall through */
L
Linus Torvalds 已提交
1796
		case FWS_R:
1797 1798 1799
			right = rcu_dereference_protected(fn->right, 1);
			if (right) {
				w->node = right;
L
Linus Torvalds 已提交
1800 1801 1802 1803
				w->state = FWS_INIT;
				continue;
			}
			w->state = FWS_C;
1804
			w->leaf = rcu_dereference_protected(fn->leaf, 1);
1805
			/* fall through */
L
Linus Torvalds 已提交
1806
		case FWS_C:
1807
			if (w->leaf && fn->fn_flags & RTN_RTINFO) {
1808 1809
				int err;

E
Eric Dumazet 已提交
1810 1811
				if (w->skip) {
					w->skip--;
1812
					goto skip;
1813 1814 1815
				}

				err = w->func(w);
L
Linus Torvalds 已提交
1816 1817
				if (err)
					return err;
1818 1819

				w->count++;
L
Linus Torvalds 已提交
1820 1821
				continue;
			}
1822
skip:
L
Linus Torvalds 已提交
1823
			w->state = FWS_U;
1824
			/* fall through */
L
Linus Torvalds 已提交
1825 1826 1827
		case FWS_U:
			if (fn == w->root)
				return 0;
1828 1829 1830
			pn = rcu_dereference_protected(fn->parent, 1);
			left = rcu_dereference_protected(pn->left, 1);
			right = rcu_dereference_protected(pn->right, 1);
L
Linus Torvalds 已提交
1831 1832
			w->node = pn;
#ifdef CONFIG_IPV6_SUBTREES
1833
			if (FIB6_SUBTREE(pn) == fn) {
1834
				WARN_ON(!(fn->fn_flags & RTN_ROOT));
L
Linus Torvalds 已提交
1835 1836 1837 1838
				w->state = FWS_L;
				continue;
			}
#endif
1839
			if (left == fn) {
L
Linus Torvalds 已提交
1840 1841 1842
				w->state = FWS_R;
				continue;
			}
1843
			if (right == fn) {
L
Linus Torvalds 已提交
1844
				w->state = FWS_C;
1845
				w->leaf = rcu_dereference_protected(w->node->leaf, 1);
L
Linus Torvalds 已提交
1846 1847 1848
				continue;
			}
#if RT6_DEBUG >= 2
1849
			WARN_ON(1);
L
Linus Torvalds 已提交
1850 1851 1852 1853 1854
#endif
		}
	}
}

M
Michal Kubeček 已提交
1855
static int fib6_walk(struct net *net, struct fib6_walker *w)
L
Linus Torvalds 已提交
1856 1857 1858 1859 1860 1861
{
	int res;

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

M
Michal Kubeček 已提交
1862
	fib6_walker_link(net, w);
L
Linus Torvalds 已提交
1863 1864
	res = fib6_walk_continue(w);
	if (res <= 0)
M
Michal Kubeček 已提交
1865
		fib6_walker_unlink(net, w);
L
Linus Torvalds 已提交
1866 1867 1868
	return res;
}

1869
static int fib6_clean_node(struct fib6_walker *w)
L
Linus Torvalds 已提交
1870 1871 1872
{
	int res;
	struct rt6_info *rt;
1873
	struct fib6_cleaner *c = container_of(w, struct fib6_cleaner, w);
1874 1875 1876
	struct nl_info info = {
		.nl_net = c->net,
	};
L
Linus Torvalds 已提交
1877

1878 1879 1880 1881 1882 1883 1884 1885 1886 1887
	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;
	}

1888
	for_each_fib6_walker_rt(w) {
L
Linus Torvalds 已提交
1889 1890 1891
		res = c->func(rt, c->arg);
		if (res < 0) {
			w->leaf = rt;
1892
			res = fib6_del(rt, &info);
L
Linus Torvalds 已提交
1893 1894
			if (res) {
#if RT6_DEBUG >= 2
1895
				pr_debug("%s: del failed: rt=%p@%p err=%d\n",
W
Wei Wang 已提交
1896 1897 1898
					 __func__, rt,
					 rcu_access_pointer(rt->rt6i_node),
					 res);
L
Linus Torvalds 已提交
1899 1900 1901 1902 1903
#endif
				continue;
			}
			return 0;
		}
1904
		WARN_ON(res != 0);
L
Linus Torvalds 已提交
1905 1906 1907 1908 1909 1910 1911
	}
	w->leaf = rt;
	return 0;
}

/*
 *	Convenient frontend to tree walker.
1912
 *
L
Linus Torvalds 已提交
1913 1914 1915 1916 1917
 *	func is called on each route.
 *		It may return -1 -> delete this route.
 *		              0  -> continue walking
 */

1918
static void fib6_clean_tree(struct net *net, struct fib6_node *root,
A
Adrian Bunk 已提交
1919
			    int (*func)(struct rt6_info *, void *arg),
1920
			    int sernum, void *arg)
L
Linus Torvalds 已提交
1921
{
1922
	struct fib6_cleaner c;
L
Linus Torvalds 已提交
1923 1924 1925

	c.w.root = root;
	c.w.func = fib6_clean_node;
1926 1927
	c.w.count = 0;
	c.w.skip = 0;
L
Linus Torvalds 已提交
1928
	c.func = func;
1929
	c.sernum = sernum;
L
Linus Torvalds 已提交
1930
	c.arg = arg;
1931
	c.net = net;
L
Linus Torvalds 已提交
1932

M
Michal Kubeček 已提交
1933
	fib6_walk(net, &c.w);
L
Linus Torvalds 已提交
1934 1935
}

1936 1937 1938
static void __fib6_clean_all(struct net *net,
			     int (*func)(struct rt6_info *, void *),
			     int sernum, void *arg)
T
Thomas Graf 已提交
1939 1940
{
	struct fib6_table *table;
1941
	struct hlist_head *head;
1942
	unsigned int h;
T
Thomas Graf 已提交
1943

1944
	rcu_read_lock();
1945
	for (h = 0; h < FIB6_TABLE_HASHSZ; h++) {
1946
		head = &net->ipv6.fib_table_hash[h];
1947
		hlist_for_each_entry_rcu(table, head, tb6_hlist) {
1948
			spin_lock_bh(&table->tb6_lock);
1949
			fib6_clean_tree(net, &table->tb6_root,
1950
					func, sernum, arg);
1951
			spin_unlock_bh(&table->tb6_lock);
T
Thomas Graf 已提交
1952 1953
		}
	}
1954
	rcu_read_unlock();
T
Thomas Graf 已提交
1955 1956
}

1957 1958 1959 1960 1961 1962
void fib6_clean_all(struct net *net, int (*func)(struct rt6_info *, void *),
		    void *arg)
{
	__fib6_clean_all(net, func, FIB6_NO_SERNUM_CHANGE, arg);
}

H
Hannes Frederic Sowa 已提交
1963 1964
static void fib6_flush_trees(struct net *net)
{
1965
	int new_sernum = fib6_new_sernum(net);
H
Hannes Frederic Sowa 已提交
1966

1967
	__fib6_clean_all(net, NULL, new_sernum, NULL);
H
Hannes Frederic Sowa 已提交
1968 1969
}

L
Linus Torvalds 已提交
1970 1971 1972 1973 1974 1975
/*
 *	Garbage collection
 */

static int fib6_age(struct rt6_info *rt, void *arg)
{
1976
	struct fib6_gc_args *gc_args = arg;
L
Linus Torvalds 已提交
1977 1978 1979 1980 1981 1982 1983
	unsigned long now = jiffies;

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

1984 1985
	if (rt->rt6i_flags & RTF_EXPIRES && rt->dst.expires) {
		if (time_after(now, rt->dst.expires)) {
L
Linus Torvalds 已提交
1986 1987 1988
			RT6_TRACE("expiring %p\n", rt);
			return -1;
		}
1989
		gc_args->more++;
L
Linus Torvalds 已提交
1990 1991
	}

1992 1993 1994 1995 1996 1997
	/*	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 已提交
1998 1999 2000
	return 0;
}

2001
void fib6_run_gc(unsigned long expires, struct net *net, bool force)
L
Linus Torvalds 已提交
2002
{
2003
	struct fib6_gc_args gc_args;
2004 2005
	unsigned long now;

2006
	if (force) {
2007 2008
		spin_lock_bh(&net->ipv6.fib6_gc_lock);
	} else if (!spin_trylock_bh(&net->ipv6.fib6_gc_lock)) {
2009 2010
		mod_timer(&net->ipv6.ip6_fib_timer, jiffies + HZ);
		return;
L
Linus Torvalds 已提交
2011
	}
2012 2013
	gc_args.timeout = expires ? (int)expires :
			  net->ipv6.sysctl.ip6_rt_gc_interval;
2014
	gc_args.more = 0;
2015

2016
	fib6_clean_all(net, fib6_age, &gc_args);
2017 2018
	now = jiffies;
	net->ipv6.ip6_rt_last_gc = now;
L
Linus Torvalds 已提交
2019 2020

	if (gc_args.more)
S
Stephen Hemminger 已提交
2021
		mod_timer(&net->ipv6.ip6_fib_timer,
2022
			  round_jiffies(now
S
Stephen Hemminger 已提交
2023
					+ net->ipv6.sysctl.ip6_rt_gc_interval));
2024 2025
	else
		del_timer(&net->ipv6.ip6_fib_timer);
2026
	spin_unlock_bh(&net->ipv6.fib6_gc_lock);
L
Linus Torvalds 已提交
2027 2028
}

2029
static void fib6_gc_timer_cb(struct timer_list *t)
2030
{
2031 2032 2033
	struct net *arg = from_timer(arg, t, ipv6.ip6_fib_timer);

	fib6_run_gc(0, arg, true);
2034 2035
}

2036
static int __net_init fib6_net_init(struct net *net)
L
Linus Torvalds 已提交
2037
{
2038
	size_t size = sizeof(struct hlist_head) * FIB6_TABLE_HASHSZ;
2039 2040 2041 2042 2043
	int err;

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

2045
	spin_lock_init(&net->ipv6.fib6_gc_lock);
M
Michal Kubeček 已提交
2046 2047
	rwlock_init(&net->ipv6.fib6_walker_lock);
	INIT_LIST_HEAD(&net->ipv6.fib6_walkers);
2048
	timer_setup(&net->ipv6.ip6_fib_timer, fib6_gc_timer_cb, 0);
2049

2050 2051 2052 2053
	net->ipv6.rt6_stats = kzalloc(sizeof(*net->ipv6.rt6_stats), GFP_KERNEL);
	if (!net->ipv6.rt6_stats)
		goto out_timer;

2054 2055 2056 2057
	/* 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);
2058
	if (!net->ipv6.fib_table_hash)
2059
		goto out_rt6_stats;
2060

2061 2062 2063
	net->ipv6.fib6_main_tbl = kzalloc(sizeof(*net->ipv6.fib6_main_tbl),
					  GFP_KERNEL);
	if (!net->ipv6.fib6_main_tbl)
2064 2065
		goto out_fib_table_hash;

2066
	net->ipv6.fib6_main_tbl->tb6_id = RT6_TABLE_MAIN;
2067 2068
	rcu_assign_pointer(net->ipv6.fib6_main_tbl->tb6_root.leaf,
			   net->ipv6.ip6_null_entry);
2069 2070
	net->ipv6.fib6_main_tbl->tb6_root.fn_flags =
		RTN_ROOT | RTN_TL_ROOT | RTN_RTINFO;
2071
	inet_peer_base_init(&net->ipv6.fib6_main_tbl->tb6_peers);
2072 2073

#ifdef CONFIG_IPV6_MULTIPLE_TABLES
2074 2075 2076
	net->ipv6.fib6_local_tbl = kzalloc(sizeof(*net->ipv6.fib6_local_tbl),
					   GFP_KERNEL);
	if (!net->ipv6.fib6_local_tbl)
2077
		goto out_fib6_main_tbl;
2078
	net->ipv6.fib6_local_tbl->tb6_id = RT6_TABLE_LOCAL;
2079 2080
	rcu_assign_pointer(net->ipv6.fib6_local_tbl->tb6_root.leaf,
			   net->ipv6.ip6_null_entry);
2081 2082
	net->ipv6.fib6_local_tbl->tb6_root.fn_flags =
		RTN_ROOT | RTN_TL_ROOT | RTN_RTINFO;
2083
	inet_peer_base_init(&net->ipv6.fib6_local_tbl->tb6_peers);
2084
#endif
2085
	fib6_tables_init(net);
2086

2087
	return 0;
2088

2089 2090
#ifdef CONFIG_IPV6_MULTIPLE_TABLES
out_fib6_main_tbl:
2091
	kfree(net->ipv6.fib6_main_tbl);
2092 2093
#endif
out_fib_table_hash:
2094
	kfree(net->ipv6.fib_table_hash);
2095 2096
out_rt6_stats:
	kfree(net->ipv6.rt6_stats);
2097
out_timer:
2098
	fib6_notifier_exit(net);
2099
	return -ENOMEM;
2100
}
2101 2102 2103

static void fib6_net_exit(struct net *net)
{
2104 2105
	unsigned int i;

2106
	rt6_ifdown(net, NULL);
2107 2108
	del_timer_sync(&net->ipv6.ip6_fib_timer);

E
Eric Dumazet 已提交
2109
	for (i = 0; i < FIB6_TABLE_HASHSZ; i++) {
2110 2111 2112 2113 2114 2115 2116 2117 2118 2119
		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);
		}
	}

2120
	kfree(net->ipv6.fib_table_hash);
2121
	kfree(net->ipv6.rt6_stats);
2122
	fib6_notifier_exit(net);
2123 2124 2125 2126 2127 2128 2129 2130 2131 2132
}

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

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

2134 2135 2136 2137 2138 2139 2140 2141 2142
	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)
2143
		goto out_kmem_cache_create;
2144 2145

	ret = __rtnl_register(PF_INET6, RTM_GETROUTE, NULL, inet6_dump_fib,
2146
			      0);
2147 2148
	if (ret)
		goto out_unregister_subsys;
H
Hannes Frederic Sowa 已提交
2149 2150

	__fib6_flush_trees = fib6_flush_trees;
2151 2152 2153
out:
	return ret;

2154 2155
out_unregister_subsys:
	unregister_pernet_subsys(&fib6_net_ops);
2156 2157 2158
out_kmem_cache_create:
	kmem_cache_destroy(fib6_node_kmem);
	goto out;
L
Linus Torvalds 已提交
2159 2160 2161 2162
}

void fib6_gc_cleanup(void)
{
2163
	unregister_pernet_subsys(&fib6_net_ops);
L
Linus Torvalds 已提交
2164 2165
	kmem_cache_destroy(fib6_node_kmem);
}
2166 2167 2168 2169 2170

#ifdef CONFIG_PROC_FS

struct ipv6_route_iter {
	struct seq_net_private p;
2171
	struct fib6_walker w;
2172 2173
	loff_t skip;
	struct fib6_table *tbl;
2174
	int sernum;
2175 2176 2177 2178 2179 2180 2181 2182 2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201
};

static int ipv6_route_seq_show(struct seq_file *seq, void *v)
{
	struct rt6_info *rt = v;
	struct ipv6_route_iter *iter = seq->private;

	seq_printf(seq, "%pi6 %02x ", &rt->rt6i_dst.addr, rt->rt6i_dst.plen);

#ifdef CONFIG_IPV6_SUBTREES
	seq_printf(seq, "%pi6 %02x ", &rt->rt6i_src.addr, rt->rt6i_src.plen);
#else
	seq_puts(seq, "00000000000000000000000000000000 00 ");
#endif
	if (rt->rt6i_flags & RTF_GATEWAY)
		seq_printf(seq, "%pi6", &rt->rt6i_gateway);
	else
		seq_puts(seq, "00000000000000000000000000000000");

	seq_printf(seq, " %08x %08x %08x %08x %8s\n",
		   rt->rt6i_metric, atomic_read(&rt->dst.__refcnt),
		   rt->dst.__use, rt->rt6i_flags,
		   rt->dst.dev ? rt->dst.dev->name : "");
	iter->w.leaf = NULL;
	return 0;
}

2202
static int ipv6_route_yield(struct fib6_walker *w)
2203 2204 2205 2206 2207 2208 2209
{
	struct ipv6_route_iter *iter = w->args;

	if (!iter->skip)
		return 1;

	do {
2210 2211 2212
		iter->w.leaf = rcu_dereference_protected(
				iter->w.leaf->dst.rt6_next,
				lockdep_is_held(&iter->tbl->tb6_lock));
2213 2214 2215 2216 2217 2218 2219 2220
		iter->skip--;
		if (!iter->skip && iter->w.leaf)
			return 1;
	} while (iter->w.leaf);

	return 0;
}

M
Michal Kubeček 已提交
2221 2222
static void ipv6_route_seq_setup_walk(struct ipv6_route_iter *iter,
				      struct net *net)
2223 2224 2225 2226 2227 2228 2229
{
	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;
2230
	iter->sernum = iter->w.root->fn_sernum;
2231
	INIT_LIST_HEAD(&iter->w.lh);
M
Michal Kubeček 已提交
2232
	fib6_walker_link(net, &iter->w);
2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255
}

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

2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266
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;
	}
}

2267 2268 2269 2270 2271 2272 2273 2274 2275 2276
static void *ipv6_route_seq_next(struct seq_file *seq, void *v, loff_t *pos)
{
	int r;
	struct rt6_info *n;
	struct net *net = seq_file_net(seq);
	struct ipv6_route_iter *iter = seq->private;

	if (!v)
		goto iter_table;

2277
	n = rcu_dereference_bh(((struct rt6_info *)v)->dst.rt6_next);
2278 2279 2280 2281 2282 2283
	if (n) {
		++*pos;
		return n;
	}

iter_table:
2284
	ipv6_route_check_sernum(iter);
2285
	spin_lock_bh(&iter->tbl->tb6_lock);
2286
	r = fib6_walk_continue(&iter->w);
2287
	spin_unlock_bh(&iter->tbl->tb6_lock);
2288 2289 2290 2291 2292
	if (r > 0) {
		if (v)
			++*pos;
		return iter->w.leaf;
	} else if (r < 0) {
M
Michal Kubeček 已提交
2293
		fib6_walker_unlink(net, &iter->w);
2294 2295
		return NULL;
	}
M
Michal Kubeček 已提交
2296
	fib6_walker_unlink(net, &iter->w);
2297 2298 2299 2300 2301

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

M
Michal Kubeček 已提交
2302
	ipv6_route_seq_setup_walk(iter, net);
2303 2304 2305 2306 2307 2308 2309 2310 2311 2312 2313 2314 2315 2316
	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 已提交
2317
		ipv6_route_seq_setup_walk(iter, net);
2318 2319 2320 2321 2322 2323 2324 2325
		return ipv6_route_seq_next(seq, NULL, pos);
	} else {
		return NULL;
	}
}

static bool ipv6_route_iter_active(struct ipv6_route_iter *iter)
{
2326
	struct fib6_walker *w = &iter->w;
2327 2328 2329 2330 2331 2332
	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 已提交
2333
	struct net *net = seq_file_net(seq);
2334 2335 2336
	struct ipv6_route_iter *iter = seq->private;

	if (ipv6_route_iter_active(iter))
M
Michal Kubeček 已提交
2337
		fib6_walker_unlink(net, &iter->w);
2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355

	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 */