sta_info.c 21.0 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 297 298
	for (i = 0; i < NUM_RX_DATA_QUEUES; i++)
		sta->last_seq_ctrl[i] = cpu_to_le16(USHORT_MAX);

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

304
#ifdef CONFIG_MAC80211_MESH
305
	sta->plink_state = PLINK_LISTEN;
306 307 308
	init_timer(&sta->plink_timer);
#endif

J
Johannes Berg 已提交
309 310 311 312 313 314 315 316
	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;
317
	int err = 0;
J
Johannes Berg 已提交
318

319 320 321 322 323
	/*
	 * 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.
	 */
324 325 326 327
	if (unlikely(!netif_running(sdata->dev))) {
		err = -ENETDOWN;
		goto out_free;
	}
328

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

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

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

353
		drv_sta_notify(local, &sdata->vif, STA_NOTIFY_ADD, &sta->sta);
354
	}
355

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

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

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

J
Johannes Berg 已提交
374 375 376 377
	if (ieee80211_vif_is_mesh(&sdata->vif))
		mesh_accept_plinks_update(sdata);

	return 0;
378 379 380 381
 out_free:
	BUG_ON(!err);
	__sta_info_free(local, sta);
	return err;
382 383
}

384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404
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)
{
405 406
	BUG_ON(!bss);

407
	__bss_tim_set(bss, sta->sta.aid);
408

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

void sta_info_set_tim_bit(struct sta_info *sta)
{
418
	unsigned long flags;
419

420 421
	BUG_ON(!sta->sdata->bss);

422 423 424
	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);
425 426 427 428 429
}

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

432
	__bss_tim_clear(bss, sta->sta.aid);
433

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

void sta_info_clear_tim_bit(struct sta_info *sta)
{
443
	unsigned long flags;
444

445 446
	BUG_ON(!sta->sdata->bss);

447 448 449
	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);
450 451
}

452
static void __sta_info_unlink(struct sta_info **sta)
453
{
454 455 456 457 458 459 460 461 462
	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;
	}
463

464 465 466 467 468
	if ((*sta)->key) {
		ieee80211_key_free((*sta)->key);
		WARN_ON((*sta)->key);
	}

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

471
	if (test_and_clear_sta_flags(*sta, WLAN_STA_PS)) {
472 473 474
		BUG_ON(!sdata->bss);

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

	local->num_sta--;

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

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

	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
498 499
	printk(KERN_DEBUG "%s: Removed STA %pM\n",
	       wiphy_name(local->hw.wiphy), (*sta)->sta.addr);
J
Johannes Berg 已提交
500 501
#endif /* CONFIG_MAC80211_VERBOSE_DEBUG */

502
	/*
J
Johannes Berg 已提交
503
	 * Finally, pull caller's reference if the STA is pinned by the
504 505 506 507 508
	 * 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
509 510 511 512 513 514 515 516
	 *  (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
517 518 519 520 521 522
	 *	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)
523
	 *  (4) We set the pin_status to DESTROY here when we
524
	 *	find such an item.
525
	 *  (5) sta_info_debugfs_add_work() will reset the pin_status
526 527 528 529 530 531 532 533 534
	 *	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;
	}
535 536
}

537 538 539 540 541 542 543 544 545
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);
}
546

547 548
static int sta_info_buffer_expired(struct sta_info *sta,
				   struct sk_buff *skb)
549
{
550
	struct ieee80211_tx_info *info;
551 552 553 554 555
	int timeout;

	if (!skb)
		return 0;

556
	info = IEEE80211_SKB_CB(skb);
557 558

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


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

	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);
581
		if (sta_info_buffer_expired(sta, skb))
582
			skb = __skb_dequeue(&sta->ps_tx_buf);
583
		else
584 585 586
			skb = NULL;
		spin_unlock_irqrestore(&sta->ps_tx_buf.lock, flags);

587
		if (!skb)
588
			break;
589

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

598 599
		if (skb_queue_empty(&sta->ps_tx_buf))
			sta_info_clear_tim_bit(sta);
600 601 602 603 604 605 606 607 608
	}
}


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

609 610
	rcu_read_lock();
	list_for_each_entry_rcu(sta, &local->sta_list, list)
611
		sta_info_cleanup_expire_buffered(local, sta);
612
	rcu_read_unlock();
613

614 615 616
	if (local->quiescing)
		return;

617 618
	local->sta_cleanup.expires =
		round_jiffies(jiffies + STA_INFO_CLEANUP_INTERVAL);
619 620 621
	add_timer(&local->sta_cleanup);
}

J
Jiri Benc 已提交
622
#ifdef CONFIG_MAC80211_DEBUGFS
J
Jiri Slaby 已提交
623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652
/*
 * 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;
}

653
static void sta_info_debugfs_add_work(struct work_struct *work)
J
Jiri Benc 已提交
654 655 656 657
{
	struct ieee80211_local *local =
		container_of(work, struct ieee80211_local, sta_debugfs_add);
	struct sta_info *sta, *tmp;
658
	unsigned long flags;
J
Jiri Benc 已提交
659

660 661
	/* We need to keep the RTNL across the whole pinned status. */
	rtnl_lock();
J
Jiri Benc 已提交
662 663
	while (1) {
		sta = NULL;
664 665

		spin_lock_irqsave(&local->sta_lock, flags);
J
Jiri Benc 已提交
666
		list_for_each_entry(tmp, &local->sta_list, list) {
J
Johannes Berg 已提交
667 668 669 670 671 672
			/*
			 * 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 已提交
673
				sta = tmp;
674
				__sta_info_pin(sta);
J
Jiri Benc 已提交
675 676 677
				break;
			}
		}
678
		spin_unlock_irqrestore(&local->sta_lock, flags);
J
Jiri Benc 已提交
679 680 681 682 683 684

		if (!sta)
			break;

		ieee80211_sta_debugfs_add(sta);
		rate_control_add_sta_debugfs(sta);
685 686

		sta = __sta_info_unpin(sta);
687
		sta_info_destroy(sta);
J
Jiri Benc 已提交
688
	}
689
	rtnl_unlock();
J
Jiri Benc 已提交
690 691 692
}
#endif

693 694
void sta_info_init(struct ieee80211_local *local)
{
695
	spin_lock_init(&local->sta_lock);
696 697
	INIT_LIST_HEAD(&local->sta_list);

698 699
	setup_timer(&local->sta_cleanup, sta_info_cleanup,
		    (unsigned long)local);
700 701
	local->sta_cleanup.expires =
		round_jiffies(jiffies + STA_INFO_CLEANUP_INTERVAL);
J
Jiri Benc 已提交
702 703

#ifdef CONFIG_MAC80211_DEBUGFS
704
	INIT_WORK(&local->sta_debugfs_add, sta_info_debugfs_add_work);
J
Jiri Benc 已提交
705
#endif
706 707 708 709 710 711 712 713 714 715 716
}

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);
717 718 719 720 721 722 723 724 725
#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 已提交
726

727
	sta_info_flush(local, NULL);
728 729 730 731
}

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

746
	might_sleep();
747

748 749 750 751
	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);
752
			if (sta) {
753
				list_add_tail(&sta->list, &tmp_list);
754 755
				ret++;
			}
756
		}
757
	}
758 759 760 761
	spin_unlock_irqrestore(&local->sta_lock, flags);

	list_for_each_entry_safe(sta, tmp, &tmp_list, list)
		sta_info_destroy(sta);
762 763

	return ret;
764
}
J
Johannes Berg 已提交
765

766 767 768 769 770 771 772 773 774 775 776 777
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
778 779
			printk(KERN_DEBUG "%s: expiring inactive STA %pM\n",
			       sdata->dev->name, sta->sta.addr);
780 781 782 783 784 785 786 787 788 789
#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);
}
790 791

struct ieee80211_sta *ieee80211_find_sta(struct ieee80211_hw *hw,
J
Johannes Berg 已提交
792
					 const u8 *addr)
793
{
794
	struct sta_info *sta = sta_info_get(hw_to_local(hw), addr);
795 796 797 798 799 800

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