mesh_pathtbl.c 28.6 KB
Newer Older
1
/*
R
Rui Paulo 已提交
2
 * Copyright (c) 2008, 2009 open80211s Ltd.
3 4 5 6 7 8 9 10 11 12
 * Author:     Luis Carlos Cobo <luisca@cozybit.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 */

#include <linux/etherdevice.h>
#include <linux/list.h>
#include <linux/random.h>
13
#include <linux/slab.h>
14 15 16
#include <linux/spinlock.h>
#include <linux/string.h>
#include <net/mac80211.h>
17
#include "wme.h"
18 19 20 21 22 23 24 25 26
#include "ieee80211_i.h"
#include "mesh.h"

/* There will be initially 2^INIT_PATHS_SIZE_ORDER buckets */
#define INIT_PATHS_SIZE_ORDER	2

/* Keep the mean chain length below this constant */
#define MEAN_CHAIN_LEN		2

J
Johannes Berg 已提交
27 28 29 30 31 32
static inline bool mpath_expired(struct mesh_path *mpath)
{
	return (mpath->flags & MESH_PATH_ACTIVE) &&
	       time_after(jiffies, mpath->exp_time) &&
	       !(mpath->flags & MESH_PATH_FIXED);
}
33 34 35 36 37 38 39 40 41 42

struct mpath_node {
	struct hlist_node list;
	struct rcu_head rcu;
	/* This indirection allows two different tables to point to the same
	 * mesh_path structure, useful when resizing
	 */
	struct mesh_path *mpath;
};

43 44
static struct mesh_table __rcu *mesh_paths;
static struct mesh_table __rcu *mpp_paths; /* Store paths for MPP&MAP */
45

46
int mesh_paths_generation;
47 48

/* This lock will have the grow table function as writer and add / delete nodes
49 50 51 52
 * as readers. RCU provides sufficient protection only when reading the table
 * (i.e. doing lookups).  Adding or adding or removing nodes requires we take
 * the read lock or we risk operating on an old table.  The write lock is only
 * needed when modifying the number of buckets a table.
53 54 55 56
 */
static DEFINE_RWLOCK(pathtbl_resize_lock);


57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74
static inline struct mesh_table *resize_dereference_mesh_paths(void)
{
	return rcu_dereference_protected(mesh_paths,
		lockdep_is_held(&pathtbl_resize_lock));
}

static inline struct mesh_table *resize_dereference_mpp_paths(void)
{
	return rcu_dereference_protected(mpp_paths,
		lockdep_is_held(&pathtbl_resize_lock));
}

/*
 * CAREFUL -- "tbl" must not be an expression,
 * in particular not an rcu_dereference(), since
 * it's used twice. So it is illegal to do
 *	for_each_mesh_entry(rcu_dereference(...), ...)
 */
75
#define for_each_mesh_entry(tbl, node, i) \
76
	for (i = 0; i <= tbl->hash_mask; i++) \
77
		hlist_for_each_entry_rcu(node, &tbl->hash_buckets[i], list)
78 79


80 81 82 83 84
static struct mesh_table *mesh_table_alloc(int size_order)
{
	int i;
	struct mesh_table *newtbl;

85
	newtbl = kmalloc(sizeof(struct mesh_table), GFP_ATOMIC);
86 87 88 89
	if (!newtbl)
		return NULL;

	newtbl->hash_buckets = kzalloc(sizeof(struct hlist_head) *
90
			(1 << size_order), GFP_ATOMIC);
91 92 93 94 95 96 97

	if (!newtbl->hash_buckets) {
		kfree(newtbl);
		return NULL;
	}

	newtbl->hashwlock = kmalloc(sizeof(spinlock_t) *
98
			(1 << size_order), GFP_ATOMIC);
99 100 101 102 103 104 105 106 107 108 109 110 111
	if (!newtbl->hashwlock) {
		kfree(newtbl->hash_buckets);
		kfree(newtbl);
		return NULL;
	}

	newtbl->size_order = size_order;
	newtbl->hash_mask = (1 << size_order) - 1;
	atomic_set(&newtbl->entries,  0);
	get_random_bytes(&newtbl->hash_rnd,
			sizeof(newtbl->hash_rnd));
	for (i = 0; i <= newtbl->hash_mask; i++)
		spin_lock_init(&newtbl->hashwlock[i]);
112
	spin_lock_init(&newtbl->gates_lock);
113 114 115 116

	return newtbl;
}

117 118 119 120 121 122 123
static void __mesh_table_free(struct mesh_table *tbl)
{
	kfree(tbl->hash_buckets);
	kfree(tbl->hashwlock);
	kfree(tbl);
}

124
static void mesh_table_free(struct mesh_table *tbl, bool free_leafs)
125 126 127
{
	struct hlist_head *mesh_hash;
	struct hlist_node *p, *q;
128
	struct mpath_node *gate;
129 130 131 132
	int i;

	mesh_hash = tbl->hash_buckets;
	for (i = 0; i <= tbl->hash_mask; i++) {
133
		spin_lock_bh(&tbl->hashwlock[i]);
134 135 136 137
		hlist_for_each_safe(p, q, &mesh_hash[i]) {
			tbl->free_node(p, free_leafs);
			atomic_dec(&tbl->entries);
		}
138
		spin_unlock_bh(&tbl->hashwlock[i]);
139
	}
140 141
	if (free_leafs) {
		spin_lock_bh(&tbl->gates_lock);
142
		hlist_for_each_entry_safe(gate, q,
143 144 145 146 147 148 149 150
					 tbl->known_gates, list) {
			hlist_del(&gate->list);
			kfree(gate);
		}
		kfree(tbl->known_gates);
		spin_unlock_bh(&tbl->gates_lock);
	}

151 152 153
	__mesh_table_free(tbl);
}

154
static int mesh_table_grow(struct mesh_table *oldtbl,
155
			   struct mesh_table *newtbl)
156 157 158 159 160
{
	struct hlist_head *oldhash;
	struct hlist_node *p, *q;
	int i;

161 162 163
	if (atomic_read(&oldtbl->entries)
			< oldtbl->mean_chain_len * (oldtbl->hash_mask + 1))
		return -EAGAIN;
164

165 166 167
	newtbl->free_node = oldtbl->free_node;
	newtbl->mean_chain_len = oldtbl->mean_chain_len;
	newtbl->copy_node = oldtbl->copy_node;
168
	newtbl->known_gates = oldtbl->known_gates;
169
	atomic_set(&newtbl->entries, atomic_read(&oldtbl->entries));
170

171 172
	oldhash = oldtbl->hash_buckets;
	for (i = 0; i <= oldtbl->hash_mask; i++)
173
		hlist_for_each(p, &oldhash[i])
174
			if (oldtbl->copy_node(p, newtbl) < 0)
175 176
				goto errcopy;

177
	return 0;
178 179 180 181

errcopy:
	for (i = 0; i <= newtbl->hash_mask; i++) {
		hlist_for_each_safe(p, q, &newtbl->hash_buckets[i])
182
			oldtbl->free_node(p, 0);
183
	}
184
	return -ENOMEM;
185 186
}

J
Johannes Berg 已提交
187
static u32 mesh_table_hash(const u8 *addr, struct ieee80211_sub_if_data *sdata,
188 189 190
			   struct mesh_table *tbl)
{
	/* Use last four bytes of hw addr and interface index as hash index */
J
Johannes Berg 已提交
191 192
	return jhash_2words(*(u32 *)(addr+2), sdata->dev->ifindex,
			    tbl->hash_rnd) & tbl->hash_mask;
193
}
194

195 196 197 198 199 200 201 202 203 204 205 206

/**
 *
 * mesh_path_assign_nexthop - update mesh path next hop
 *
 * @mpath: mesh path to update
 * @sta: next hop to assign
 *
 * Locking: mpath->state_lock must be held when calling this function
 */
void mesh_path_assign_nexthop(struct mesh_path *mpath, struct sta_info *sta)
{
207 208 209 210
	struct sk_buff *skb;
	struct ieee80211_hdr *hdr;
	unsigned long flags;

211
	rcu_assign_pointer(mpath->next_hop, sta);
212 213

	spin_lock_irqsave(&mpath->frame_queue.lock, flags);
214
	skb_queue_walk(&mpath->frame_queue, skb) {
215 216
		hdr = (struct ieee80211_hdr *) skb->data;
		memcpy(hdr->addr1, sta->sta.addr, ETH_ALEN);
217
		memcpy(hdr->addr2, mpath->sdata->vif.addr, ETH_ALEN);
M
Marco Porsch 已提交
218
		ieee80211_mps_set_frame_flags(sta->sdata, sta, hdr);
219 220 221
	}

	spin_unlock_irqrestore(&mpath->frame_queue.lock, flags);
222 223
}

224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259
static void prepare_for_gate(struct sk_buff *skb, char *dst_addr,
			     struct mesh_path *gate_mpath)
{
	struct ieee80211_hdr *hdr;
	struct ieee80211s_hdr *mshdr;
	int mesh_hdrlen, hdrlen;
	char *next_hop;

	hdr = (struct ieee80211_hdr *) skb->data;
	hdrlen = ieee80211_hdrlen(hdr->frame_control);
	mshdr = (struct ieee80211s_hdr *) (skb->data + hdrlen);

	if (!(mshdr->flags & MESH_FLAGS_AE)) {
		/* size of the fixed part of the mesh header */
		mesh_hdrlen = 6;

		/* make room for the two extended addresses */
		skb_push(skb, 2 * ETH_ALEN);
		memmove(skb->data, hdr, hdrlen + mesh_hdrlen);

		hdr = (struct ieee80211_hdr *) skb->data;

		/* we preserve the previous mesh header and only add
		 * the new addreses */
		mshdr = (struct ieee80211s_hdr *) (skb->data + hdrlen);
		mshdr->flags = MESH_FLAGS_AE_A5_A6;
		memcpy(mshdr->eaddr1, hdr->addr3, ETH_ALEN);
		memcpy(mshdr->eaddr2, hdr->addr4, ETH_ALEN);
	}

	/* update next hop */
	hdr = (struct ieee80211_hdr *) skb->data;
	rcu_read_lock();
	next_hop = rcu_dereference(gate_mpath->next_hop)->sta.addr;
	memcpy(hdr->addr1, next_hop, ETH_ALEN);
	rcu_read_unlock();
260
	memcpy(hdr->addr2, gate_mpath->sdata->vif.addr, ETH_ALEN);
261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285
	memcpy(hdr->addr3, dst_addr, ETH_ALEN);
}

/**
 *
 * mesh_path_move_to_queue - Move or copy frames from one mpath queue to another
 *
 * This function is used to transfer or copy frames from an unresolved mpath to
 * a gate mpath.  The function also adds the Address Extension field and
 * updates the next hop.
 *
 * If a frame already has an Address Extension field, only the next hop and
 * destination addresses are updated.
 *
 * The gate mpath must be an active mpath with a valid mpath->next_hop.
 *
 * @mpath: An active mpath the frames will be sent to (i.e. the gate)
 * @from_mpath: The failed mpath
 * @copy: When true, copy all the frames to the new mpath queue.  When false,
 * move them.
 */
static void mesh_path_move_to_queue(struct mesh_path *gate_mpath,
				    struct mesh_path *from_mpath,
				    bool copy)
{
286 287
	struct sk_buff *skb, *fskb, *tmp;
	struct sk_buff_head failq;
288 289 290 291 292 293 294 295 296 297 298
	unsigned long flags;

	BUG_ON(gate_mpath == from_mpath);
	BUG_ON(!gate_mpath->next_hop);

	__skb_queue_head_init(&failq);

	spin_lock_irqsave(&from_mpath->frame_queue.lock, flags);
	skb_queue_splice_init(&from_mpath->frame_queue, &failq);
	spin_unlock_irqrestore(&from_mpath->frame_queue.lock, flags);

299 300 301 302 303
	skb_queue_walk_safe(&failq, fskb, tmp) {
		if (skb_queue_len(&gate_mpath->frame_queue) >=
				  MESH_FRAME_QUEUE_LEN) {
			mpath_dbg(gate_mpath->sdata, "mpath queue full!\n");
			break;
304
		}
305

306 307 308 309
		skb = skb_copy(fskb, GFP_ATOMIC);
		if (WARN_ON(!skb))
			break;

310
		prepare_for_gate(skb, gate_mpath->dst, gate_mpath);
311 312 313 314 315 316 317
		skb_queue_tail(&gate_mpath->frame_queue, skb);

		if (copy)
			continue;

		__skb_unlink(fskb, &failq);
		kfree_skb(fskb);
318 319
	}

J
Johannes Berg 已提交
320 321
	mpath_dbg(gate_mpath->sdata, "Mpath queue for gate %pM has %d frames\n",
		  gate_mpath->dst, skb_queue_len(&gate_mpath->frame_queue));
322 323 324 325 326 327 328 329 330

	if (!copy)
		return;

	spin_lock_irqsave(&from_mpath->frame_queue.lock, flags);
	skb_queue_splice(&failq, &from_mpath->frame_queue);
	spin_unlock_irqrestore(&from_mpath->frame_queue.lock, flags);
}

331

J
Johannes Berg 已提交
332 333
static struct mesh_path *mpath_lookup(struct mesh_table *tbl, const u8 *dst,
				      struct ieee80211_sub_if_data *sdata)
334 335 336 337 338
{
	struct mesh_path *mpath;
	struct hlist_head *bucket;
	struct mpath_node *node;

339
	bucket = &tbl->hash_buckets[mesh_table_hash(dst, sdata, tbl)];
340
	hlist_for_each_entry_rcu(node, bucket, list) {
341
		mpath = node->mpath;
342
		if (mpath->sdata == sdata &&
343
		    ether_addr_equal(dst, mpath->dst)) {
J
Johannes Berg 已提交
344
			if (mpath_expired(mpath)) {
345
				spin_lock_bh(&mpath->state_lock);
346
				mpath->flags &= ~MESH_PATH_ACTIVE;
347 348 349 350 351 352 353 354
				spin_unlock_bh(&mpath->state_lock);
			}
			return mpath;
		}
	}
	return NULL;
}

355 356 357
/**
 * mesh_path_lookup - look up a path in the mesh path table
 * @sdata: local subif
J
Johannes Berg 已提交
358
 * @dst: hardware address (ETH_ALEN length) of destination
359 360 361 362 363
 *
 * Returns: pointer to the mesh path structure, or NULL if not found
 *
 * Locking: must be called within a read rcu section.
 */
J
Johannes Berg 已提交
364 365
struct mesh_path *
mesh_path_lookup(struct ieee80211_sub_if_data *sdata, const u8 *dst)
366
{
367
	return mpath_lookup(rcu_dereference(mesh_paths), dst, sdata);
368
}
369

J
Johannes Berg 已提交
370 371
struct mesh_path *
mpp_path_lookup(struct ieee80211_sub_if_data *sdata, const u8 *dst)
372
{
373
	return mpath_lookup(rcu_dereference(mpp_paths), dst, sdata);
374 375 376
}


377 378 379
/**
 * mesh_path_lookup_by_idx - look up a path in the mesh path table by its index
 * @idx: index
380
 * @sdata: local subif, or NULL for all entries
381 382 383 384 385
 *
 * Returns: pointer to the mesh path structure, or NULL if not found.
 *
 * Locking: must be called within a read rcu section.
 */
J
Johannes Berg 已提交
386 387
struct mesh_path *
mesh_path_lookup_by_idx(struct ieee80211_sub_if_data *sdata, int idx)
388
{
389
	struct mesh_table *tbl = rcu_dereference(mesh_paths);
390 391 392 393
	struct mpath_node *node;
	int i;
	int j = 0;

394
	for_each_mesh_entry(tbl, node, i) {
395
		if (sdata && node->mpath->sdata != sdata)
396
			continue;
397
		if (j++ == idx) {
J
Johannes Berg 已提交
398
			if (mpath_expired(node->mpath)) {
399
				spin_lock_bh(&node->mpath->state_lock);
400
				node->mpath->flags &= ~MESH_PATH_ACTIVE;
401 402 403 404
				spin_unlock_bh(&node->mpath->state_lock);
			}
			return node->mpath;
		}
405
	}
406 407 408 409

	return NULL;
}

410
/**
J
Johannes Berg 已提交
411 412
 * mesh_path_add_gate - add the given mpath to a mesh gate to our path table
 * @mpath: gate path to add to table
413
 */
J
Johannes Berg 已提交
414
int mesh_path_add_gate(struct mesh_path *mpath)
415
{
J
Johannes Berg 已提交
416
	struct mesh_table *tbl;
417 418 419 420
	struct mpath_node *gate, *new_gate;
	int err;

	rcu_read_lock();
J
Johannes Berg 已提交
421
	tbl = rcu_dereference(mesh_paths);
422

423
	hlist_for_each_entry_rcu(gate, tbl->known_gates, list)
424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440
		if (gate->mpath == mpath) {
			err = -EEXIST;
			goto err_rcu;
		}

	new_gate = kzalloc(sizeof(struct mpath_node), GFP_ATOMIC);
	if (!new_gate) {
		err = -ENOMEM;
		goto err_rcu;
	}

	mpath->is_gate = true;
	mpath->sdata->u.mesh.num_gates++;
	new_gate->mpath = mpath;
	spin_lock_bh(&tbl->gates_lock);
	hlist_add_head_rcu(&new_gate->list, tbl->known_gates);
	spin_unlock_bh(&tbl->gates_lock);
J
Johannes Berg 已提交
441 442 443
	mpath_dbg(mpath->sdata,
		  "Mesh path: Recorded new gate: %pM. %d known gates\n",
		  mpath->dst, mpath->sdata->u.mesh.num_gates);
J
Johannes Berg 已提交
444
	err = 0;
445 446 447 448 449 450 451 452 453 454 455 456
err_rcu:
	rcu_read_unlock();
	return err;
}

/**
 * mesh_gate_del - remove a mesh gate from the list of known gates
 * @tbl: table which holds our list of known gates
 * @mpath: gate mpath
 *
 * Locking: must be called inside rcu_read_lock() section
 */
J
Johannes Berg 已提交
457
static void mesh_gate_del(struct mesh_table *tbl, struct mesh_path *mpath)
458 459
{
	struct mpath_node *gate;
460
	struct hlist_node *q;
461

462
	hlist_for_each_entry_safe(gate, q, tbl->known_gates, list) {
J
Johannes Berg 已提交
463 464 465 466 467 468 469 470 471 472 473 474 475
		if (gate->mpath != mpath)
			continue;
		spin_lock_bh(&tbl->gates_lock);
		hlist_del_rcu(&gate->list);
		kfree_rcu(gate, rcu);
		spin_unlock_bh(&tbl->gates_lock);
		mpath->sdata->u.mesh.num_gates--;
		mpath->is_gate = false;
		mpath_dbg(mpath->sdata,
			  "Mesh path: Deleted gate: %pM. %d known gates\n",
			  mpath->dst, mpath->sdata->u.mesh.num_gates);
		break;
	}
476 477 478 479 480 481 482 483 484 485 486
}

/**
 * mesh_gate_num - number of gates known to this interface
 * @sdata: subif data
 */
int mesh_gate_num(struct ieee80211_sub_if_data *sdata)
{
	return sdata->u.mesh.num_gates;
}

487 488
/**
 * mesh_path_add - allocate and add a new path to the mesh path table
J
Johannes Berg 已提交
489
 * @dst: destination address of the path (ETH_ALEN length)
490
 * @sdata: local subif
491
 *
492
 * Returns: 0 on success
493 494 495
 *
 * State: the initial state of the new path is set to 0
 */
496 497
struct mesh_path *mesh_path_add(struct ieee80211_sub_if_data *sdata,
				const u8 *dst)
498
{
499 500
	struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
	struct ieee80211_local *local = sdata->local;
501
	struct mesh_table *tbl;
502 503 504 505
	struct mesh_path *mpath, *new_mpath;
	struct mpath_node *node, *new_node;
	struct hlist_head *bucket;
	int grow = 0;
506
	int err;
507 508
	u32 hash_idx;

509
	if (ether_addr_equal(dst, sdata->vif.addr))
510
		/* never add ourselves as neighbours */
511
		return ERR_PTR(-ENOTSUPP);
512 513

	if (is_multicast_ether_addr(dst))
514
		return ERR_PTR(-ENOTSUPP);
515

516
	if (atomic_add_unless(&sdata->u.mesh.mpaths, 1, MESH_MAX_MPATHS) == 0)
517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532
		return ERR_PTR(-ENOSPC);

	read_lock_bh(&pathtbl_resize_lock);
	tbl = resize_dereference_mesh_paths();

	hash_idx = mesh_table_hash(dst, sdata, tbl);
	bucket = &tbl->hash_buckets[hash_idx];

	spin_lock(&tbl->hashwlock[hash_idx]);

	hlist_for_each_entry(node, bucket, list) {
		mpath = node->mpath;
		if (mpath->sdata == sdata &&
		    ether_addr_equal(dst, mpath->dst))
			goto found;
	}
533

534
	err = -ENOMEM;
535
	new_mpath = kzalloc(sizeof(struct mesh_path), GFP_ATOMIC);
536 537 538
	if (!new_mpath)
		goto err_path_alloc;

539
	new_node = kmalloc(sizeof(struct mpath_node), GFP_ATOMIC);
540 541
	if (!new_node)
		goto err_node_alloc;
542

543
	memcpy(new_mpath->dst, dst, ETH_ALEN);
J
Johannes Berg 已提交
544
	eth_broadcast_addr(new_mpath->rann_snd_addr);
545
	new_mpath->is_root = false;
546
	new_mpath->sdata = sdata;
547 548 549 550 551 552 553 554 555 556
	new_mpath->flags = 0;
	skb_queue_head_init(&new_mpath->frame_queue);
	new_node->mpath = new_mpath;
	new_mpath->timer.data = (unsigned long) new_mpath;
	new_mpath->timer.function = mesh_path_timer;
	new_mpath->exp_time = jiffies;
	spin_lock_init(&new_mpath->state_lock);
	init_timer(&new_mpath->timer);

	hlist_add_head_rcu(&new_node->list, bucket);
557 558
	if (atomic_inc_return(&tbl->entries) >=
	    tbl->mean_chain_len * (tbl->hash_mask + 1))
559 560
		grow = 1;

561 562
	mesh_paths_generation++;

563
	if (grow) {
564
		set_bit(MESH_WORK_GROW_MPATH_TABLE,  &ifmsh->wrkq_flags);
J
Johannes Berg 已提交
565
		ieee80211_queue_work(&local->hw, &sdata->work);
566
	}
567 568
	mpath = new_mpath;
found:
569
	spin_unlock(&tbl->hashwlock[hash_idx]);
570
	read_unlock_bh(&pathtbl_resize_lock);
571 572
	return mpath;

573 574 575
err_node_alloc:
	kfree(new_mpath);
err_path_alloc:
576
	atomic_dec(&sdata->u.mesh.mpaths);
577 578 579
	spin_unlock(&tbl->hashwlock[hash_idx]);
	read_unlock_bh(&pathtbl_resize_lock);
	return ERR_PTR(err);
580 581
}

582 583 584 585 586 587 588
static void mesh_table_free_rcu(struct rcu_head *rcu)
{
	struct mesh_table *tbl = container_of(rcu, struct mesh_table, rcu_head);

	mesh_table_free(tbl, false);
}

589 590 591 592
void mesh_mpath_table_grow(void)
{
	struct mesh_table *oldtbl, *newtbl;

593
	write_lock_bh(&pathtbl_resize_lock);
594 595
	oldtbl = resize_dereference_mesh_paths();
	newtbl = mesh_table_alloc(oldtbl->size_order + 1);
596 597
	if (!newtbl)
		goto out;
598
	if (mesh_table_grow(oldtbl, newtbl) < 0) {
599
		__mesh_table_free(newtbl);
600
		goto out;
601 602 603
	}
	rcu_assign_pointer(mesh_paths, newtbl);

604 605 606 607
	call_rcu(&oldtbl->rcu_head, mesh_table_free_rcu);

 out:
	write_unlock_bh(&pathtbl_resize_lock);
608 609 610 611 612 613
}

void mesh_mpp_table_grow(void)
{
	struct mesh_table *oldtbl, *newtbl;

614
	write_lock_bh(&pathtbl_resize_lock);
615 616
	oldtbl = resize_dereference_mpp_paths();
	newtbl = mesh_table_alloc(oldtbl->size_order + 1);
617 618
	if (!newtbl)
		goto out;
619
	if (mesh_table_grow(oldtbl, newtbl) < 0) {
620
		__mesh_table_free(newtbl);
621
		goto out;
622 623
	}
	rcu_assign_pointer(mpp_paths, newtbl);
624
	call_rcu(&oldtbl->rcu_head, mesh_table_free_rcu);
625

626 627
 out:
	write_unlock_bh(&pathtbl_resize_lock);
628
}
629

J
Johannes Berg 已提交
630 631
int mpp_path_add(struct ieee80211_sub_if_data *sdata,
		 const u8 *dst, const u8 *mpp)
632
{
633 634
	struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
	struct ieee80211_local *local = sdata->local;
635
	struct mesh_table *tbl;
636 637 638 639 640 641 642
	struct mesh_path *mpath, *new_mpath;
	struct mpath_node *node, *new_node;
	struct hlist_head *bucket;
	int grow = 0;
	int err = 0;
	u32 hash_idx;

643
	if (ether_addr_equal(dst, sdata->vif.addr))
644 645 646 647 648 649 650
		/* never add ourselves as neighbours */
		return -ENOTSUPP;

	if (is_multicast_ether_addr(dst))
		return -ENOTSUPP;

	err = -ENOMEM;
651
	new_mpath = kzalloc(sizeof(struct mesh_path), GFP_ATOMIC);
652 653 654
	if (!new_mpath)
		goto err_path_alloc;

655
	new_node = kmalloc(sizeof(struct mpath_node), GFP_ATOMIC);
656 657 658
	if (!new_node)
		goto err_node_alloc;

659
	read_lock_bh(&pathtbl_resize_lock);
660 661 662 663 664 665
	memcpy(new_mpath->dst, dst, ETH_ALEN);
	memcpy(new_mpath->mpp, mpp, ETH_ALEN);
	new_mpath->sdata = sdata;
	new_mpath->flags = 0;
	skb_queue_head_init(&new_mpath->frame_queue);
	new_node->mpath = new_mpath;
T
Thomas Pedersen 已提交
666
	init_timer(&new_mpath->timer);
667 668 669
	new_mpath->exp_time = jiffies;
	spin_lock_init(&new_mpath->state_lock);

670
	tbl = resize_dereference_mpp_paths();
671

672 673 674
	hash_idx = mesh_table_hash(dst, sdata, tbl);
	bucket = &tbl->hash_buckets[hash_idx];

675
	spin_lock(&tbl->hashwlock[hash_idx]);
676 677

	err = -EEXIST;
678
	hlist_for_each_entry(node, bucket, list) {
679
		mpath = node->mpath;
680
		if (mpath->sdata == sdata &&
681
		    ether_addr_equal(dst, mpath->dst))
682 683 684 685
			goto err_exists;
	}

	hlist_add_head_rcu(&new_node->list, bucket);
686 687
	if (atomic_inc_return(&tbl->entries) >=
	    tbl->mean_chain_len * (tbl->hash_mask + 1))
688 689
		grow = 1;

690
	spin_unlock(&tbl->hashwlock[hash_idx]);
691
	read_unlock_bh(&pathtbl_resize_lock);
692
	if (grow) {
693
		set_bit(MESH_WORK_GROW_MPP_TABLE,  &ifmsh->wrkq_flags);
J
Johannes Berg 已提交
694
		ieee80211_queue_work(&local->hw, &sdata->work);
695 696 697 698
	}
	return 0;

err_exists:
699
	spin_unlock(&tbl->hashwlock[hash_idx]);
700
	read_unlock_bh(&pathtbl_resize_lock);
701 702 703 704 705 706 707 708
	kfree(new_node);
err_node_alloc:
	kfree(new_mpath);
err_path_alloc:
	return err;
}


709 710 711 712 713 714 715 716 717 718
/**
 * mesh_plink_broken - deactivates paths and sends perr when a link breaks
 *
 * @sta: broken peer link
 *
 * This function must be called from the rate control algorithm if enough
 * delivery errors suggest that a peer link is no longer usable.
 */
void mesh_plink_broken(struct sta_info *sta)
{
719
	struct mesh_table *tbl;
720
	static const u8 bcast[ETH_ALEN] = {0xff, 0xff, 0xff, 0xff, 0xff, 0xff};
721 722
	struct mesh_path *mpath;
	struct mpath_node *node;
723
	struct ieee80211_sub_if_data *sdata = sta->sdata;
724
	int i;
725
	__le16 reason = cpu_to_le16(WLAN_REASON_MESH_PATH_DEST_UNREACHABLE);
726 727

	rcu_read_lock();
728
	tbl = rcu_dereference(mesh_paths);
729
	for_each_mesh_entry(tbl, node, i) {
730
		mpath = node->mpath;
731
		if (rcu_dereference(mpath->next_hop) == sta &&
732 733
		    mpath->flags & MESH_PATH_ACTIVE &&
		    !(mpath->flags & MESH_PATH_FIXED)) {
734
			spin_lock_bh(&mpath->state_lock);
735
			mpath->flags &= ~MESH_PATH_ACTIVE;
736
			++mpath->sn;
737
			spin_unlock_bh(&mpath->state_lock);
J
Johannes Berg 已提交
738 739 740 741
			mesh_path_error_tx(sdata,
					   sdata->u.mesh.mshcfg.element_ttl,
					   mpath->dst, cpu_to_le32(mpath->sn),
					   reason, bcast);
742
		}
743 744 745 746
	}
	rcu_read_unlock();
}

747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772
static void mesh_path_node_reclaim(struct rcu_head *rp)
{
	struct mpath_node *node = container_of(rp, struct mpath_node, rcu);
	struct ieee80211_sub_if_data *sdata = node->mpath->sdata;

	del_timer_sync(&node->mpath->timer);
	atomic_dec(&sdata->u.mesh.mpaths);
	kfree(node->mpath);
	kfree(node);
}

/* needs to be called with the corresponding hashwlock taken */
static void __mesh_path_del(struct mesh_table *tbl, struct mpath_node *node)
{
	struct mesh_path *mpath;
	mpath = node->mpath;
	spin_lock(&mpath->state_lock);
	mpath->flags |= MESH_PATH_RESOLVING;
	if (mpath->is_gate)
		mesh_gate_del(tbl, mpath);
	hlist_del_rcu(&node->list);
	call_rcu(&node->rcu, mesh_path_node_reclaim);
	spin_unlock(&mpath->state_lock);
	atomic_dec(&tbl->entries);
}

773 774 775
/**
 * mesh_path_flush_by_nexthop - Deletes mesh paths if their next hop matches
 *
776
 * @sta: mesh peer to match
777
 *
778 779 780
 * RCU notes: this function is called when a mesh plink transitions from
 * PLINK_ESTAB to any other state, since PLINK_ESTAB state is the only one that
 * allows path creation. This will happen before the sta can be freed (because
781 782
 * sta_info_destroy() calls this) so any reader in a rcu read block will be
 * protected against the plink disappearing.
783 784 785
 */
void mesh_path_flush_by_nexthop(struct sta_info *sta)
{
786
	struct mesh_table *tbl;
787 788 789 790
	struct mesh_path *mpath;
	struct mpath_node *node;
	int i;

791
	rcu_read_lock();
792 793
	read_lock_bh(&pathtbl_resize_lock);
	tbl = resize_dereference_mesh_paths();
794
	for_each_mesh_entry(tbl, node, i) {
795
		mpath = node->mpath;
796
		if (rcu_dereference(mpath->next_hop) == sta) {
797
			spin_lock(&tbl->hashwlock[i]);
798
			__mesh_path_del(tbl, node);
799
			spin_unlock(&tbl->hashwlock[i]);
800
		}
801
	}
802
	read_unlock_bh(&pathtbl_resize_lock);
803
	rcu_read_unlock();
804 805
}

806 807
static void table_flush_by_iface(struct mesh_table *tbl,
				 struct ieee80211_sub_if_data *sdata)
808 809 810 811 812
{
	struct mesh_path *mpath;
	struct mpath_node *node;
	int i;

813
	WARN_ON(!rcu_read_lock_held());
814
	for_each_mesh_entry(tbl, node, i) {
815
		mpath = node->mpath;
816 817
		if (mpath->sdata != sdata)
			continue;
818
		spin_lock_bh(&tbl->hashwlock[i]);
819
		__mesh_path_del(tbl, node);
820
		spin_unlock_bh(&tbl->hashwlock[i]);
821 822 823
	}
}

824 825 826 827 828
/**
 * mesh_path_flush_by_iface - Deletes all mesh paths associated with a given iface
 *
 * This function deletes both mesh paths as well as mesh portal paths.
 *
829
 * @sdata: interface data to match
830 831 832
 *
 */
void mesh_path_flush_by_iface(struct ieee80211_sub_if_data *sdata)
833
{
834
	struct mesh_table *tbl;
835

836
	rcu_read_lock();
837 838
	read_lock_bh(&pathtbl_resize_lock);
	tbl = resize_dereference_mesh_paths();
839
	table_flush_by_iface(tbl, sdata);
840
	tbl = resize_dereference_mpp_paths();
841
	table_flush_by_iface(tbl, sdata);
842
	read_unlock_bh(&pathtbl_resize_lock);
843
	rcu_read_unlock();
844 845 846 847 848 849
}

/**
 * mesh_path_del - delete a mesh path from the table
 *
 * @addr: dst address (ETH_ALEN length)
850
 * @sdata: local subif
851
 *
852
 * Returns: 0 if successful
853
 */
J
Johannes Berg 已提交
854
int mesh_path_del(struct ieee80211_sub_if_data *sdata, const u8 *addr)
855
{
856
	struct mesh_table *tbl;
857 858 859 860 861 862
	struct mesh_path *mpath;
	struct mpath_node *node;
	struct hlist_head *bucket;
	int hash_idx;
	int err = 0;

863
	read_lock_bh(&pathtbl_resize_lock);
864 865 866
	tbl = resize_dereference_mesh_paths();
	hash_idx = mesh_table_hash(addr, sdata, tbl);
	bucket = &tbl->hash_buckets[hash_idx];
867

868
	spin_lock(&tbl->hashwlock[hash_idx]);
869
	hlist_for_each_entry(node, bucket, list) {
870
		mpath = node->mpath;
871
		if (mpath->sdata == sdata &&
872
		    ether_addr_equal(addr, mpath->dst)) {
873
			__mesh_path_del(tbl, node);
874 875 876 877 878 879
			goto enddel;
		}
	}

	err = -ENXIO;
enddel:
880
	mesh_paths_generation++;
881
	spin_unlock(&tbl->hashwlock[hash_idx]);
882
	read_unlock_bh(&pathtbl_resize_lock);
883 884 885 886 887 888 889 890 891 892 893 894 895
	return err;
}

/**
 * mesh_path_tx_pending - sends pending frames in a mesh path queue
 *
 * @mpath: mesh path to activate
 *
 * Locking: the state_lock of the mpath structure must NOT be held when calling
 * this function.
 */
void mesh_path_tx_pending(struct mesh_path *mpath)
{
896 897 898
	if (mpath->flags & MESH_PATH_ACTIVE)
		ieee80211_add_pending_skbs(mpath->sdata->local,
				&mpath->frame_queue);
899 900
}

901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927
/**
 * mesh_path_send_to_gates - sends pending frames to all known mesh gates
 *
 * @mpath: mesh path whose queue will be emptied
 *
 * If there is only one gate, the frames are transferred from the failed mpath
 * queue to that gate's queue.  If there are more than one gates, the frames
 * are copied from each gate to the next.  After frames are copied, the
 * mpath queues are emptied onto the transmission queue.
 */
int mesh_path_send_to_gates(struct mesh_path *mpath)
{
	struct ieee80211_sub_if_data *sdata = mpath->sdata;
	struct mesh_table *tbl;
	struct mesh_path *from_mpath = mpath;
	struct mpath_node *gate = NULL;
	bool copy = false;
	struct hlist_head *known_gates;

	rcu_read_lock();
	tbl = rcu_dereference(mesh_paths);
	known_gates = tbl->known_gates;
	rcu_read_unlock();

	if (!known_gates)
		return -EHOSTUNREACH;

928
	hlist_for_each_entry_rcu(gate, known_gates, list) {
929 930 931 932
		if (gate->mpath->sdata != sdata)
			continue;

		if (gate->mpath->flags & MESH_PATH_ACTIVE) {
J
Johannes Berg 已提交
933
			mpath_dbg(sdata, "Forwarding to %pM\n", gate->mpath->dst);
934 935 936 937
			mesh_path_move_to_queue(gate->mpath, from_mpath, copy);
			from_mpath = gate->mpath;
			copy = true;
		} else {
J
Johannes Berg 已提交
938 939 940
			mpath_dbg(sdata,
				  "Not forwarding %p (flags %#x)\n",
				  gate->mpath, gate->mpath->flags);
941 942 943
		}
	}

944
	hlist_for_each_entry_rcu(gate, known_gates, list)
945
		if (gate->mpath->sdata == sdata) {
J
Johannes Berg 已提交
946
			mpath_dbg(sdata, "Sending to %pM\n", gate->mpath->dst);
947 948 949 950 951 952
			mesh_path_tx_pending(gate->mpath);
		}

	return (from_mpath == mpath) ? -EHOSTUNREACH : 0;
}

953 954 955 956
/**
 * mesh_path_discard_frame - discard a frame whose path could not be resolved
 *
 * @skb: frame to discard
957
 * @sdata: network subif the frame was to be sent through
958 959 960
 *
 * Locking: the function must me called within a rcu_read_lock region
 */
J
Johannes Berg 已提交
961 962
void mesh_path_discard_frame(struct ieee80211_sub_if_data *sdata,
			     struct sk_buff *skb)
963 964
{
	kfree_skb(skb);
965
	sdata->u.mesh.mshstats.dropped_frames_no_route++;
966 967 968 969 970 971 972
}

/**
 * mesh_path_flush_pending - free the pending queue of a mesh path
 *
 * @mpath: mesh path whose queue has to be freed
 *
L
Lucas De Marchi 已提交
973
 * Locking: the function must me called within a rcu_read_lock region
974 975 976 977 978
 */
void mesh_path_flush_pending(struct mesh_path *mpath)
{
	struct sk_buff *skb;

979
	while ((skb = skb_dequeue(&mpath->frame_queue)) != NULL)
J
Johannes Berg 已提交
980
		mesh_path_discard_frame(mpath->sdata, skb);
981 982 983 984 985 986 987 988 989 990 991 992 993 994
}

/**
 * mesh_path_fix_nexthop - force a specific next hop for a mesh path
 *
 * @mpath: the mesh path to modify
 * @next_hop: the next hop to force
 *
 * Locking: this function must be called holding mpath->state_lock
 */
void mesh_path_fix_nexthop(struct mesh_path *mpath, struct sta_info *next_hop)
{
	spin_lock_bh(&mpath->state_lock);
	mesh_path_assign_nexthop(mpath, next_hop);
995
	mpath->sn = 0xffff;
996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010
	mpath->metric = 0;
	mpath->hop_count = 0;
	mpath->exp_time = 0;
	mpath->flags |= MESH_PATH_FIXED;
	mesh_path_activate(mpath);
	spin_unlock_bh(&mpath->state_lock);
	mesh_path_tx_pending(mpath);
}

static void mesh_path_node_free(struct hlist_node *p, bool free_leafs)
{
	struct mesh_path *mpath;
	struct mpath_node *node = hlist_entry(p, struct mpath_node, list);
	mpath = node->mpath;
	hlist_del_rcu(p);
1011 1012
	if (free_leafs) {
		del_timer_sync(&mpath->timer);
1013
		kfree(mpath);
1014
	}
1015 1016 1017
	kfree(node);
}

1018
static int mesh_path_node_copy(struct hlist_node *p, struct mesh_table *newtbl)
1019 1020 1021 1022 1023
{
	struct mesh_path *mpath;
	struct mpath_node *node, *new_node;
	u32 hash_idx;

1024
	new_node = kmalloc(sizeof(struct mpath_node), GFP_ATOMIC);
1025 1026 1027
	if (new_node == NULL)
		return -ENOMEM;

1028 1029 1030
	node = hlist_entry(p, struct mpath_node, list);
	mpath = node->mpath;
	new_node->mpath = mpath;
1031
	hash_idx = mesh_table_hash(mpath->dst, mpath->sdata, newtbl);
1032 1033
	hlist_add_head(&new_node->list,
			&newtbl->hash_buckets[hash_idx]);
1034
	return 0;
1035 1036 1037 1038
}

int mesh_pathtbl_init(void)
{
1039
	struct mesh_table *tbl_path, *tbl_mpp;
1040
	int ret;
1041 1042 1043

	tbl_path = mesh_table_alloc(INIT_PATHS_SIZE_ORDER);
	if (!tbl_path)
1044
		return -ENOMEM;
1045 1046 1047
	tbl_path->free_node = &mesh_path_node_free;
	tbl_path->copy_node = &mesh_path_node_copy;
	tbl_path->mean_chain_len = MEAN_CHAIN_LEN;
1048
	tbl_path->known_gates = kzalloc(sizeof(struct hlist_head), GFP_ATOMIC);
1049 1050 1051 1052
	if (!tbl_path->known_gates) {
		ret = -ENOMEM;
		goto free_path;
	}
1053 1054
	INIT_HLIST_HEAD(tbl_path->known_gates);

1055

1056 1057
	tbl_mpp = mesh_table_alloc(INIT_PATHS_SIZE_ORDER);
	if (!tbl_mpp) {
1058 1059
		ret = -ENOMEM;
		goto free_path;
1060
	}
1061 1062 1063
	tbl_mpp->free_node = &mesh_path_node_free;
	tbl_mpp->copy_node = &mesh_path_node_copy;
	tbl_mpp->mean_chain_len = MEAN_CHAIN_LEN;
1064
	tbl_mpp->known_gates = kzalloc(sizeof(struct hlist_head), GFP_ATOMIC);
1065 1066 1067 1068
	if (!tbl_mpp->known_gates) {
		ret = -ENOMEM;
		goto free_mpp;
	}
1069
	INIT_HLIST_HEAD(tbl_mpp->known_gates);
1070 1071 1072 1073

	/* Need no locking since this is during init */
	RCU_INIT_POINTER(mesh_paths, tbl_path);
	RCU_INIT_POINTER(mpp_paths, tbl_mpp);
1074

1075
	return 0;
1076 1077 1078 1079 1080 1081

free_mpp:
	mesh_table_free(tbl_mpp, true);
free_path:
	mesh_table_free(tbl_path, true);
	return ret;
1082 1083
}

1084
void mesh_path_expire(struct ieee80211_sub_if_data *sdata)
1085
{
1086
	struct mesh_table *tbl;
1087 1088 1089 1090
	struct mesh_path *mpath;
	struct mpath_node *node;
	int i;

1091 1092
	rcu_read_lock();
	tbl = rcu_dereference(mesh_paths);
1093
	for_each_mesh_entry(tbl, node, i) {
1094
		if (node->mpath->sdata != sdata)
1095 1096 1097 1098
			continue;
		mpath = node->mpath;
		if ((!(mpath->flags & MESH_PATH_RESOLVING)) &&
		    (!(mpath->flags & MESH_PATH_FIXED)) &&
1099
		     time_after(jiffies, mpath->exp_time + MESH_PATH_EXPIRE))
J
Johannes Berg 已提交
1100
			mesh_path_del(mpath->sdata, mpath->dst);
1101
	}
1102
	rcu_read_unlock();
1103 1104 1105 1106
}

void mesh_pathtbl_unregister(void)
{
1107
	/* no need for locking during exit path */
1108 1109
	mesh_table_free(rcu_dereference_protected(mesh_paths, 1), true);
	mesh_table_free(rcu_dereference_protected(mpp_paths, 1), true);
1110
}