sta_info.c 20.9 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
/*
 * Copyright 2002-2005, Instant802 Networks, Inc.
 * Copyright 2006-2007	Jiri Benc <jbenc@suse.cz>
 *
 * 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/module.h>
#include <linux/init.h>
#include <linux/netdevice.h>
#include <linux/types.h>
#include <linux/slab.h>
#include <linux/skbuff.h>
#include <linux/if_arp.h>
17
#include <linux/timer.h>
18
#include <linux/rtnetlink.h>
19 20 21

#include <net/mac80211.h>
#include "ieee80211_i.h"
22
#include "driver-ops.h"
J
Johannes Berg 已提交
23
#include "rate.h"
24
#include "sta_info.h"
J
Jiri Benc 已提交
25
#include "debugfs_sta.h"
26
#include "mesh.h"
27

28 29 30 31 32 33 34
/**
 * DOC: STA information lifetime rules
 *
 * STA info structures (&struct sta_info) are managed in a hash table
 * for faster lookup and a list for iteration. They are managed using
 * RCU, i.e. access to the list and hash table is protected by RCU.
 *
35 36 37 38 39
 * Upon allocating a STA info structure with sta_info_alloc(), the caller owns
 * that structure. It must then either destroy it using sta_info_destroy()
 * (which is pretty useless) or insert it into the hash table using
 * sta_info_insert() which demotes the reference from ownership to a regular
 * RCU-protected reference; if the function is called without protection by an
40 41 42 43 44 45
 * RCU critical section the reference is instantly invalidated. Note that the
 * caller may not do much with the STA info before inserting it, in particular,
 * it may not start any mesh peer link management or add encryption keys.
 *
 * When the insertion fails (sta_info_insert()) returns non-zero), the
 * structure will have been freed by sta_info_insert()!
46 47 48 49
 *
 * Because there are debugfs entries for each station, and adding those
 * must be able to sleep, it is also possible to "pin" a station entry,
 * that means it can be removed from the hash table but not be freed.
50 51
 * See the comment in __sta_info_unlink() for more information, this is
 * an internal capability only.
52 53
 *
 * In order to remove a STA info structure, the caller needs to first
J
Johannes Berg 已提交
54
 * unlink it (sta_info_unlink()) from the list and hash tables and
55 56 57 58 59
 * then destroy it; sta_info_destroy() will wait for an RCU grace period
 * to elapse before actually freeing it. Due to the pinning and the
 * possibility of multiple callers trying to remove the same STA info at
 * the same time, sta_info_unlink() can clear the STA info pointer it is
 * passed to indicate that the STA info is owned by somebody else now.
60
 *
J
Johannes Berg 已提交
61
 * If sta_info_unlink() did not clear the pointer then the caller owns
62
 * the STA info structure now and is responsible of destroying it with
63
 * a call to sta_info_destroy().
64 65 66 67 68 69
 *
 * In all other cases, there is no concept of ownership on a STA entry,
 * each structure is owned by the global hash table/list until it is
 * removed. All users of the structure need to be RCU protected so that
 * the structure won't be freed before they are done using it.
 */
70 71

/* Caller must hold local->sta_lock */
72 73
static int sta_info_hash_del(struct ieee80211_local *local,
			     struct sta_info *sta)
74 75 76
{
	struct sta_info *s;

77
	s = local->sta_hash[STA_HASH(sta->sta.addr)];
78
	if (!s)
79 80
		return -ENOENT;
	if (s == sta) {
81
		rcu_assign_pointer(local->sta_hash[STA_HASH(sta->sta.addr)],
82
				   s->hnext);
83
		return 0;
84 85
	}

86
	while (s->hnext && s->hnext != sta)
87
		s = s->hnext;
88
	if (s->hnext) {
89
		rcu_assign_pointer(s->hnext, sta->hnext);
90 91
		return 0;
	}
92

93
	return -ENOENT;
94 95
}

96
/* protected by RCU */
97
struct sta_info *sta_info_get(struct ieee80211_local *local, const u8 *addr)
98 99 100
{
	struct sta_info *sta;

101
	sta = rcu_dereference(local->sta_hash[STA_HASH(addr)]);
102
	while (sta) {
103
		if (memcmp(sta->sta.addr, addr, ETH_ALEN) == 0)
104
			break;
105
		sta = rcu_dereference(sta->hnext);
106
	}
107 108 109
	return sta;
}

110 111 112 113 114 115
struct sta_info *sta_info_get_by_idx(struct ieee80211_local *local, int idx,
				     struct net_device *dev)
{
	struct sta_info *sta;
	int i = 0;

116
	list_for_each_entry_rcu(sta, &local->sta_list, list) {
117 118
		if (dev && dev != sta->sdata->dev)
			continue;
119 120 121 122
		if (i < idx) {
			++i;
			continue;
		}
123
		return sta;
124 125 126 127
	}

	return NULL;
}
128

129 130 131
/**
 * __sta_info_free - internal STA free helper
 *
R
Randy Dunlap 已提交
132
 * @local: pointer to the global information
133 134 135 136 137 138 139 140
 * @sta: STA info to free
 *
 * This function must undo everything done by sta_info_alloc()
 * that may happen before sta_info_insert().
 */
static void __sta_info_free(struct ieee80211_local *local,
			    struct sta_info *sta)
{
141
	rate_control_free_sta(sta);
142 143 144
	rate_control_put(sta->rate_ctrl);

#ifdef CONFIG_MAC80211_VERBOSE_DEBUG
145 146
	printk(KERN_DEBUG "%s: Destroyed STA %pM\n",
	       wiphy_name(local->hw.wiphy), sta->sta.addr);
147 148 149 150 151
#endif /* CONFIG_MAC80211_VERBOSE_DEBUG */

	kfree(sta);
}

152
void sta_info_destroy(struct sta_info *sta)
153
{
154
	struct ieee80211_local *local;
155
	struct sk_buff *skb;
156
	int i;
J
Johannes Berg 已提交
157

158 159
	might_sleep();

J
Johannes Berg 已提交
160 161
	if (!sta)
		return;
162

163
	local = sta->local;
164 165 166 167 168 169 170 171 172

	rate_control_remove_sta_debugfs(sta);
	ieee80211_sta_debugfs_remove(sta);

#ifdef CONFIG_MAC80211_MESH
	if (ieee80211_vif_is_mesh(&sta->sdata->vif))
		mesh_plink_deactivate(sta);
#endif

173 174 175 176 177 178 179 180 181 182
	/*
	 * We have only unlinked the key, and actually destroying it
	 * may mean it is removed from hardware which requires that
	 * the key->sta pointer is still valid, so flush the key todo
	 * list here.
	 *
	 * ieee80211_key_todo() will synchronize_rcu() so after this
	 * nothing can reference this sta struct any more.
	 */
	ieee80211_key_todo();
183 184 185 186 187 188

#ifdef CONFIG_MAC80211_MESH
	if (ieee80211_vif_is_mesh(&sta->sdata->vif))
		del_timer_sync(&sta->plink_timer);
#endif

189 190 191 192
	while ((skb = skb_dequeue(&sta->ps_tx_buf)) != NULL) {
		local->total_ps_buffered--;
		dev_kfree_skb_any(skb);
	}
193 194

	while ((skb = skb_dequeue(&sta->tx_filtered)) != NULL)
195
		dev_kfree_skb_any(skb);
196

197
	for (i = 0; i <  STA_TID_NUM; i++) {
198 199 200
		struct tid_ampdu_rx *tid_rx;
		struct tid_ampdu_tx *tid_tx;

201
		spin_lock_bh(&sta->lock);
202 203 204 205
		tid_rx = sta->ampdu_mlme.tid_rx[i];
		/* Make sure timer won't free the tid_rx struct, see below */
		if (tid_rx)
			tid_rx->shutdown = true;
206

207
		spin_unlock_bh(&sta->lock);
208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231

		/*
		 * Outside spinlock - shutdown is true now so that the timer
		 * won't free tid_rx, we have to do that now. Can't let the
		 * timer do it because we have to sync the timer outside the
		 * lock that it takes itself.
		 */
		if (tid_rx) {
			del_timer_sync(&tid_rx->session_timer);
			kfree(tid_rx);
		}

		/*
		 * No need to do such complications for TX agg sessions, the
		 * path leading to freeing the tid_tx struct goes via a call
		 * from the driver, and thus needs to look up the sta struct
		 * again, which cannot be found when we get here. Hence, we
		 * just need to delete the timer and free the aggregation
		 * info; we won't be telling the peer about it then but that
		 * doesn't matter if we're not talking to it again anyway.
		 */
		tid_tx = sta->ampdu_mlme.tid_tx[i];
		if (tid_tx) {
			del_timer_sync(&tid_tx->addba_resp_timer);
232 233 234 235 236
			/*
			 * STA removed while aggregation session being
			 * started? Bit odd, but purge frames anyway.
			 */
			skb_queue_purge(&tid_tx->pending);
237 238
			kfree(tid_tx);
		}
239
	}
240

241
	__sta_info_free(local, sta);
242 243 244
}


245 246 247
/* Caller must hold local->sta_lock */
static void sta_info_hash_add(struct ieee80211_local *local,
			      struct sta_info *sta)
248
{
249 250
	sta->hnext = local->sta_hash[STA_HASH(sta->sta.addr)];
	rcu_assign_pointer(local->sta_hash[STA_HASH(sta->sta.addr)], sta);
251 252
}

J
Johannes Berg 已提交
253 254
struct sta_info *sta_info_alloc(struct ieee80211_sub_if_data *sdata,
				u8 *addr, gfp_t gfp)
255
{
256
	struct ieee80211_local *local = sdata->local;
257
	struct sta_info *sta;
258
	int i;
259

260
	sta = kzalloc(sizeof(*sta) + local->hw.sta_data_size, gfp);
261
	if (!sta)
J
Johannes Berg 已提交
262
		return NULL;
263

264
	spin_lock_init(&sta->lock);
265
	spin_lock_init(&sta->flaglock);
266

267
	memcpy(sta->sta.addr, addr, ETH_ALEN);
268 269
	sta->local = local;
	sta->sdata = sdata;
270 271

	sta->rate_ctrl = rate_control_get(local->rate_ctrl);
272
	sta->rate_ctrl_priv = rate_control_alloc_sta(sta->rate_ctrl,
273
						     &sta->sta, gfp);
274 275 276
	if (!sta->rate_ctrl_priv) {
		rate_control_put(sta->rate_ctrl);
		kfree(sta);
J
Johannes Berg 已提交
277
		return NULL;
278 279
	}

280 281 282 283 284
	for (i = 0; i < STA_TID_NUM; i++) {
		/* timer_to_tid must be initialized with identity mapping to
		 * enable session_timer's data differentiation. refer to
		 * sta_rx_agg_session_timer_expired for useage */
		sta->timer_to_tid[i] = i;
285 286 287 288 289 290 291
		/* rx */
		sta->ampdu_mlme.tid_state_rx[i] = HT_AGG_STATE_IDLE;
		sta->ampdu_mlme.tid_rx[i] = NULL;
		/* tx */
		sta->ampdu_mlme.tid_state_tx[i] = HT_AGG_STATE_IDLE;
		sta->ampdu_mlme.tid_tx[i] = NULL;
		sta->ampdu_mlme.addba_req_num[i] = 0;
292
	}
293 294
	skb_queue_head_init(&sta->ps_tx_buf);
	skb_queue_head_init(&sta->tx_filtered);
J
Johannes Berg 已提交
295 296

#ifdef CONFIG_MAC80211_VERBOSE_DEBUG
297 298
	printk(KERN_DEBUG "%s: Allocated STA %pM\n",
	       wiphy_name(local->hw.wiphy), sta->sta.addr);
J
Johannes Berg 已提交
299 300
#endif /* CONFIG_MAC80211_VERBOSE_DEBUG */

301
#ifdef CONFIG_MAC80211_MESH
302
	sta->plink_state = PLINK_LISTEN;
303 304 305
	init_timer(&sta->plink_timer);
#endif

J
Johannes Berg 已提交
306 307 308 309 310 311 312 313
	return sta;
}

int sta_info_insert(struct sta_info *sta)
{
	struct ieee80211_local *local = sta->local;
	struct ieee80211_sub_if_data *sdata = sta->sdata;
	unsigned long flags;
314
	int err = 0;
J
Johannes Berg 已提交
315

316 317 318 319 320
	/*
	 * Can't be a WARN_ON because it can be triggered through a race:
	 * something inserts a STA (on one CPU) without holding the RTNL
	 * and another CPU turns off the net device.
	 */
321 322 323 324
	if (unlikely(!netif_running(sdata->dev))) {
		err = -ENETDOWN;
		goto out_free;
	}
325

326
	if (WARN_ON(compare_ether_addr(sta->sta.addr, sdata->dev->dev_addr) == 0 ||
J
Johannes Berg 已提交
327
		    is_multicast_ether_addr(sta->sta.addr))) {
328 329 330
		err = -EINVAL;
		goto out_free;
	}
331

332
	spin_lock_irqsave(&local->sta_lock, flags);
333
	/* check if STA exists already */
334
	if (sta_info_get(local, sta->sta.addr)) {
335
		spin_unlock_irqrestore(&local->sta_lock, flags);
336 337
		err = -EEXIST;
		goto out_free;
338
	}
339 340 341
	list_add(&sta->list, &local->sta_list);
	local->num_sta++;
	sta_info_hash_add(local, sta);
342

343 344
	/* notify driver */
	if (local->ops->sta_notify) {
345
		if (sdata->vif.type == NL80211_IFTYPE_AP_VLAN)
346 347 348
			sdata = container_of(sdata->bss,
					     struct ieee80211_sub_if_data,
					     u.ap);
349

350
		drv_sta_notify(local, &sdata->vif, STA_NOTIFY_ADD, &sta->sta);
351
	}
352

353
#ifdef CONFIG_MAC80211_VERBOSE_DEBUG
354 355
	printk(KERN_DEBUG "%s: Inserted STA %pM\n",
	       wiphy_name(local->hw.wiphy), sta->sta.addr);
356 357
#endif /* CONFIG_MAC80211_VERBOSE_DEBUG */

J
Johannes Berg 已提交
358 359
	spin_unlock_irqrestore(&local->sta_lock, flags);

J
Jiri Benc 已提交
360
#ifdef CONFIG_MAC80211_DEBUGFS
361 362
	/*
	 * Debugfs entry adding might sleep, so schedule process
363
	 * context task for adding entry for STAs that do not yet
364 365 366 367
	 * have one.
	 * NOTE: due to auto-freeing semantics this may only be done
	 *       if the insertion is successful!
	 */
368
	schedule_work(&local->sta_debugfs_add);
J
Jiri Benc 已提交
369 370
#endif

J
Johannes Berg 已提交
371 372 373 374
	if (ieee80211_vif_is_mesh(&sdata->vif))
		mesh_accept_plinks_update(sdata);

	return 0;
375 376 377 378
 out_free:
	BUG_ON(!err);
	__sta_info_free(local, sta);
	return err;
379 380
}

381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401
static inline void __bss_tim_set(struct ieee80211_if_ap *bss, u16 aid)
{
	/*
	 * This format has been mandated by the IEEE specifications,
	 * so this line may not be changed to use the __set_bit() format.
	 */
	bss->tim[aid / 8] |= (1 << (aid % 8));
}

static inline void __bss_tim_clear(struct ieee80211_if_ap *bss, u16 aid)
{
	/*
	 * This format has been mandated by the IEEE specifications,
	 * so this line may not be changed to use the __clear_bit() format.
	 */
	bss->tim[aid / 8] &= ~(1 << (aid % 8));
}

static void __sta_info_set_tim_bit(struct ieee80211_if_ap *bss,
				   struct sta_info *sta)
{
402 403
	BUG_ON(!bss);

404
	__bss_tim_set(bss, sta->sta.aid);
405

406 407
	if (sta->local->ops->set_tim) {
		sta->local->tim_in_locked_section = true;
408
		drv_set_tim(sta->local, &sta->sta, true);
409 410
		sta->local->tim_in_locked_section = false;
	}
411 412 413 414
}

void sta_info_set_tim_bit(struct sta_info *sta)
{
415
	unsigned long flags;
416

417 418
	BUG_ON(!sta->sdata->bss);

419 420 421
	spin_lock_irqsave(&sta->local->sta_lock, flags);
	__sta_info_set_tim_bit(sta->sdata->bss, sta);
	spin_unlock_irqrestore(&sta->local->sta_lock, flags);
422 423 424 425 426
}

static void __sta_info_clear_tim_bit(struct ieee80211_if_ap *bss,
				     struct sta_info *sta)
{
427 428
	BUG_ON(!bss);

429
	__bss_tim_clear(bss, sta->sta.aid);
430

431 432
	if (sta->local->ops->set_tim) {
		sta->local->tim_in_locked_section = true;
433
		drv_set_tim(sta->local, &sta->sta, false);
434 435
		sta->local->tim_in_locked_section = false;
	}
436 437 438 439
}

void sta_info_clear_tim_bit(struct sta_info *sta)
{
440
	unsigned long flags;
441

442 443
	BUG_ON(!sta->sdata->bss);

444 445 446
	spin_lock_irqsave(&sta->local->sta_lock, flags);
	__sta_info_clear_tim_bit(sta->sdata->bss, sta);
	spin_unlock_irqrestore(&sta->local->sta_lock, flags);
447 448
}

449
static void __sta_info_unlink(struct sta_info **sta)
450
{
451 452 453 454 455 456 457 458 459
	struct ieee80211_local *local = (*sta)->local;
	struct ieee80211_sub_if_data *sdata = (*sta)->sdata;
	/*
	 * pull caller's reference if we're already gone.
	 */
	if (sta_info_hash_del(local, *sta)) {
		*sta = NULL;
		return;
	}
460

461 462 463 464 465
	if ((*sta)->key) {
		ieee80211_key_free((*sta)->key);
		WARN_ON((*sta)->key);
	}

J
Johannes Berg 已提交
466 467
	list_del(&(*sta)->list);

468
	if (test_and_clear_sta_flags(*sta, WLAN_STA_PS)) {
469 470 471
		BUG_ON(!sdata->bss);

		atomic_dec(&sdata->bss->num_sta_ps);
J
Johannes Berg 已提交
472 473 474 475 476 477
		__sta_info_clear_tim_bit(sdata->bss, *sta);
	}

	local->num_sta--;

	if (local->ops->sta_notify) {
478
		if (sdata->vif.type == NL80211_IFTYPE_AP_VLAN)
479 480 481
			sdata = container_of(sdata->bss,
					     struct ieee80211_sub_if_data,
					     u.ap);
J
Johannes Berg 已提交
482

483 484
		drv_sta_notify(local, &sdata->vif, STA_NOTIFY_REMOVE,
			       &(*sta)->sta);
J
Johannes Berg 已提交
485 486 487 488 489 490 491 492 493 494
	}

	if (ieee80211_vif_is_mesh(&sdata->vif)) {
		mesh_accept_plinks_update(sdata);
#ifdef CONFIG_MAC80211_MESH
		del_timer(&(*sta)->plink_timer);
#endif
	}

#ifdef CONFIG_MAC80211_VERBOSE_DEBUG
495 496
	printk(KERN_DEBUG "%s: Removed STA %pM\n",
	       wiphy_name(local->hw.wiphy), (*sta)->sta.addr);
J
Johannes Berg 已提交
497 498
#endif /* CONFIG_MAC80211_VERBOSE_DEBUG */

499
	/*
J
Johannes Berg 已提交
500
	 * Finally, pull caller's reference if the STA is pinned by the
501 502 503 504 505
	 * task that is adding the debugfs entries. In that case, we
	 * leave the STA "to be freed".
	 *
	 * The rules are not trivial, but not too complex either:
	 *  (1) pin_status is only modified under the sta_lock
506 507 508 509 510 511 512 513
	 *  (2) STAs may only be pinned under the RTNL so that
	 *	sta_info_flush() is guaranteed to actually destroy
	 *	all STAs that are active for a given interface, this
	 *	is required for correctness because otherwise we
	 *	could notify a driver that an interface is going
	 *	away and only after that (!) notify it about a STA
	 *	on that interface going away.
	 *  (3) sta_info_debugfs_add_work() will set the status
514 515 516 517 518 519
	 *	to PINNED when it found an item that needs a new
	 *	debugfs directory created. In that case, that item
	 *	must not be freed although all *RCU* users are done
	 *	with it. Hence, we tell the caller of _unlink()
	 *	that the item is already gone (as can happen when
	 *	two tasks try to unlink/destroy at the same time)
520
	 *  (4) We set the pin_status to DESTROY here when we
521
	 *	find such an item.
522
	 *  (5) sta_info_debugfs_add_work() will reset the pin_status
523 524 525 526 527 528 529 530 531
	 *	from PINNED to NORMAL when it is done with the item,
	 *	but will check for DESTROY before resetting it in
	 *	which case it will free the item.
	 */
	if ((*sta)->pin_status == STA_INFO_PIN_STAT_PINNED) {
		(*sta)->pin_status = STA_INFO_PIN_STAT_DESTROY;
		*sta = NULL;
		return;
	}
532 533
}

534 535 536 537 538 539 540 541 542
void sta_info_unlink(struct sta_info **sta)
{
	struct ieee80211_local *local = (*sta)->local;
	unsigned long flags;

	spin_lock_irqsave(&local->sta_lock, flags);
	__sta_info_unlink(sta);
	spin_unlock_irqrestore(&local->sta_lock, flags);
}
543

544 545
static int sta_info_buffer_expired(struct sta_info *sta,
				   struct sk_buff *skb)
546
{
547
	struct ieee80211_tx_info *info;
548 549 550 551 552
	int timeout;

	if (!skb)
		return 0;

553
	info = IEEE80211_SKB_CB(skb);
554 555

	/* Timeout: (2 * listen_interval * beacon_int * 1024 / 1000000) sec */
556 557 558
	timeout = (sta->listen_interval *
		   sta->sdata->vif.bss_conf.beacon_int *
		   32 / 15625) * HZ;
559 560
	if (timeout < STA_TX_BUFFER_EXPIRE)
		timeout = STA_TX_BUFFER_EXPIRE;
561
	return time_after(jiffies, info->control.jiffies + timeout);
562 563 564 565 566 567 568 569
}


static void sta_info_cleanup_expire_buffered(struct ieee80211_local *local,
					     struct sta_info *sta)
{
	unsigned long flags;
	struct sk_buff *skb;
570
	struct ieee80211_sub_if_data *sdata;
571 572 573 574 575 576 577

	if (skb_queue_empty(&sta->ps_tx_buf))
		return;

	for (;;) {
		spin_lock_irqsave(&sta->ps_tx_buf.lock, flags);
		skb = skb_peek(&sta->ps_tx_buf);
578
		if (sta_info_buffer_expired(sta, skb))
579
			skb = __skb_dequeue(&sta->ps_tx_buf);
580
		else
581 582 583
			skb = NULL;
		spin_unlock_irqrestore(&sta->ps_tx_buf.lock, flags);

584
		if (!skb)
585
			break;
586

587
		sdata = sta->sdata;
588
		local->total_ps_buffered--;
589
#ifdef CONFIG_MAC80211_VERBOSE_PS_DEBUG
590 591
		printk(KERN_DEBUG "Buffered frame expired (STA %pM)\n",
		       sta->sta.addr);
592
#endif
593 594
		dev_kfree_skb(skb);

595 596
		if (skb_queue_empty(&sta->ps_tx_buf))
			sta_info_clear_tim_bit(sta);
597 598 599 600 601 602 603 604 605
	}
}


static void sta_info_cleanup(unsigned long data)
{
	struct ieee80211_local *local = (struct ieee80211_local *) data;
	struct sta_info *sta;

606 607
	rcu_read_lock();
	list_for_each_entry_rcu(sta, &local->sta_list, list)
608
		sta_info_cleanup_expire_buffered(local, sta);
609
	rcu_read_unlock();
610

611 612
	local->sta_cleanup.expires =
		round_jiffies(jiffies + STA_INFO_CLEANUP_INTERVAL);
613 614 615
	add_timer(&local->sta_cleanup);
}

J
Jiri Benc 已提交
616
#ifdef CONFIG_MAC80211_DEBUGFS
J
Jiri Slaby 已提交
617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646
/*
 * See comment in __sta_info_unlink,
 * caller must hold local->sta_lock.
 */
static void __sta_info_pin(struct sta_info *sta)
{
	WARN_ON(sta->pin_status != STA_INFO_PIN_STAT_NORMAL);
	sta->pin_status = STA_INFO_PIN_STAT_PINNED;
}

/*
 * See comment in __sta_info_unlink, returns sta if it
 * needs to be destroyed.
 */
static struct sta_info *__sta_info_unpin(struct sta_info *sta)
{
	struct sta_info *ret = NULL;
	unsigned long flags;

	spin_lock_irqsave(&sta->local->sta_lock, flags);
	WARN_ON(sta->pin_status != STA_INFO_PIN_STAT_DESTROY &&
		sta->pin_status != STA_INFO_PIN_STAT_PINNED);
	if (sta->pin_status == STA_INFO_PIN_STAT_DESTROY)
		ret = sta;
	sta->pin_status = STA_INFO_PIN_STAT_NORMAL;
	spin_unlock_irqrestore(&sta->local->sta_lock, flags);

	return ret;
}

647
static void sta_info_debugfs_add_work(struct work_struct *work)
J
Jiri Benc 已提交
648 649 650 651
{
	struct ieee80211_local *local =
		container_of(work, struct ieee80211_local, sta_debugfs_add);
	struct sta_info *sta, *tmp;
652
	unsigned long flags;
J
Jiri Benc 已提交
653

654 655
	/* We need to keep the RTNL across the whole pinned status. */
	rtnl_lock();
J
Jiri Benc 已提交
656 657
	while (1) {
		sta = NULL;
658 659

		spin_lock_irqsave(&local->sta_lock, flags);
J
Jiri Benc 已提交
660
		list_for_each_entry(tmp, &local->sta_list, list) {
J
Johannes Berg 已提交
661 662 663 664 665 666
			/*
			 * debugfs.add_has_run will be set by
			 * ieee80211_sta_debugfs_add regardless
			 * of what else it does.
			 */
			if (!tmp->debugfs.add_has_run) {
J
Jiri Benc 已提交
667
				sta = tmp;
668
				__sta_info_pin(sta);
J
Jiri Benc 已提交
669 670 671
				break;
			}
		}
672
		spin_unlock_irqrestore(&local->sta_lock, flags);
J
Jiri Benc 已提交
673 674 675 676 677 678

		if (!sta)
			break;

		ieee80211_sta_debugfs_add(sta);
		rate_control_add_sta_debugfs(sta);
679 680

		sta = __sta_info_unpin(sta);
681
		sta_info_destroy(sta);
J
Jiri Benc 已提交
682
	}
683
	rtnl_unlock();
J
Jiri Benc 已提交
684 685 686
}
#endif

687 688
void sta_info_init(struct ieee80211_local *local)
{
689
	spin_lock_init(&local->sta_lock);
690 691
	INIT_LIST_HEAD(&local->sta_list);

692 693
	setup_timer(&local->sta_cleanup, sta_info_cleanup,
		    (unsigned long)local);
694 695
	local->sta_cleanup.expires =
		round_jiffies(jiffies + STA_INFO_CLEANUP_INTERVAL);
J
Jiri Benc 已提交
696 697

#ifdef CONFIG_MAC80211_DEBUGFS
698
	INIT_WORK(&local->sta_debugfs_add, sta_info_debugfs_add_work);
J
Jiri Benc 已提交
699
#endif
700 701 702 703 704 705 706 707 708 709 710
}

int sta_info_start(struct ieee80211_local *local)
{
	add_timer(&local->sta_cleanup);
	return 0;
}

void sta_info_stop(struct ieee80211_local *local)
{
	del_timer(&local->sta_cleanup);
711 712 713 714 715 716 717 718 719
#ifdef CONFIG_MAC80211_DEBUGFS
	/*
	 * Make sure the debugfs adding work isn't pending after this
	 * because we're about to be destroyed. It doesn't matter
	 * whether it ran or not since we're going to flush all STAs
	 * anyway.
	 */
	cancel_work_sync(&local->sta_debugfs_add);
#endif
J
Johannes Berg 已提交
720

721
	sta_info_flush(local, NULL);
722 723 724 725
}

/**
 * sta_info_flush - flush matching STA entries from the STA table
726 727 728
 *
 * Returns the number of removed STA entries.
 *
729
 * @local: local interface data
730
 * @sdata: matching rule for the net device (sta->dev) or %NULL to match all STAs
731
 */
732
int sta_info_flush(struct ieee80211_local *local,
733
		   struct ieee80211_sub_if_data *sdata)
734 735
{
	struct sta_info *sta, *tmp;
736
	LIST_HEAD(tmp_list);
737
	int ret = 0;
738
	unsigned long flags;
739

740
	might_sleep();
741

742 743 744 745
	spin_lock_irqsave(&local->sta_lock, flags);
	list_for_each_entry_safe(sta, tmp, &local->sta_list, list) {
		if (!sdata || sdata == sta->sdata) {
			__sta_info_unlink(&sta);
746
			if (sta) {
747
				list_add_tail(&sta->list, &tmp_list);
748 749
				ret++;
			}
750
		}
751
	}
752 753 754 755
	spin_unlock_irqrestore(&local->sta_lock, flags);

	list_for_each_entry_safe(sta, tmp, &tmp_list, list)
		sta_info_destroy(sta);
756 757

	return ret;
758
}
J
Johannes Berg 已提交
759

760 761 762 763 764 765 766 767 768 769 770 771
void ieee80211_sta_expire(struct ieee80211_sub_if_data *sdata,
			  unsigned long exp_time)
{
	struct ieee80211_local *local = sdata->local;
	struct sta_info *sta, *tmp;
	LIST_HEAD(tmp_list);
	unsigned long flags;

	spin_lock_irqsave(&local->sta_lock, flags);
	list_for_each_entry_safe(sta, tmp, &local->sta_list, list)
		if (time_after(jiffies, sta->last_rx + exp_time)) {
#ifdef CONFIG_MAC80211_IBSS_DEBUG
772 773
			printk(KERN_DEBUG "%s: expiring inactive STA %pM\n",
			       sdata->dev->name, sta->sta.addr);
774 775 776 777 778 779 780 781 782 783
#endif
			__sta_info_unlink(&sta);
			if (sta)
				list_add(&sta->list, &tmp_list);
		}
	spin_unlock_irqrestore(&local->sta_lock, flags);

	list_for_each_entry_safe(sta, tmp, &tmp_list, list)
		sta_info_destroy(sta);
}
784 785

struct ieee80211_sta *ieee80211_find_sta(struct ieee80211_hw *hw,
J
Johannes Berg 已提交
786
					 const u8 *addr)
787
{
788
	struct sta_info *sta = sta_info_get(hw_to_local(hw), addr);
789 790 791 792 793 794

	if (!sta)
		return NULL;
	return &sta->sta;
}
EXPORT_SYMBOL(ieee80211_find_sta);