sta_info.c revision 75de9113bb9dc4939a7cd54e4bdfad555b35f5b1
1/* 2 * Copyright 2002-2005, Instant802 Networks, Inc. 3 * Copyright 2006-2007 Jiri Benc <jbenc@suse.cz> 4 * 5 * This program is free software; you can redistribute it and/or modify 6 * it under the terms of the GNU General Public License version 2 as 7 * published by the Free Software Foundation. 8 */ 9 10#include <linux/module.h> 11#include <linux/init.h> 12#include <linux/etherdevice.h> 13#include <linux/netdevice.h> 14#include <linux/types.h> 15#include <linux/slab.h> 16#include <linux/skbuff.h> 17#include <linux/if_arp.h> 18#include <linux/timer.h> 19#include <linux/rtnetlink.h> 20 21#include <net/mac80211.h> 22#include "ieee80211_i.h" 23#include "driver-ops.h" 24#include "rate.h" 25#include "sta_info.h" 26#include "debugfs_sta.h" 27#include "mesh.h" 28#include "wme.h" 29 30/** 31 * DOC: STA information lifetime rules 32 * 33 * STA info structures (&struct sta_info) are managed in a hash table 34 * for faster lookup and a list for iteration. They are managed using 35 * RCU, i.e. access to the list and hash table is protected by RCU. 36 * 37 * Upon allocating a STA info structure with sta_info_alloc(), the caller 38 * owns that structure. It must then insert it into the hash table using 39 * either sta_info_insert() or sta_info_insert_rcu(); only in the latter 40 * case (which acquires an rcu read section but must not be called from 41 * within one) will the pointer still be valid after the call. Note that 42 * the caller may not do much with the STA info before inserting it, in 43 * particular, it may not start any mesh peer link management or add 44 * encryption keys. 45 * 46 * When the insertion fails (sta_info_insert()) returns non-zero), the 47 * structure will have been freed by sta_info_insert()! 48 * 49 * Station entries are added by mac80211 when you establish a link with a 50 * peer. This means different things for the different type of interfaces 51 * we support. For a regular station this mean we add the AP sta when we 52 * receive an association response from the AP. For IBSS this occurs when 53 * get to know about a peer on the same IBSS. For WDS we add the sta for 54 * the peer immediately upon device open. When using AP mode we add stations 55 * for each respective station upon request from userspace through nl80211. 56 * 57 * In order to remove a STA info structure, various sta_info_destroy_*() 58 * calls are available. 59 * 60 * There is no concept of ownership on a STA entry, each structure is 61 * owned by the global hash table/list until it is removed. All users of 62 * the structure need to be RCU protected so that the structure won't be 63 * freed before they are done using it. 64 */ 65 66/* Caller must hold local->sta_mtx */ 67static int sta_info_hash_del(struct ieee80211_local *local, 68 struct sta_info *sta) 69{ 70 struct sta_info *s; 71 72 s = rcu_dereference_protected(local->sta_hash[STA_HASH(sta->sta.addr)], 73 lockdep_is_held(&local->sta_mtx)); 74 if (!s) 75 return -ENOENT; 76 if (s == sta) { 77 rcu_assign_pointer(local->sta_hash[STA_HASH(sta->sta.addr)], 78 s->hnext); 79 return 0; 80 } 81 82 while (rcu_access_pointer(s->hnext) && 83 rcu_access_pointer(s->hnext) != sta) 84 s = rcu_dereference_protected(s->hnext, 85 lockdep_is_held(&local->sta_mtx)); 86 if (rcu_access_pointer(s->hnext)) { 87 rcu_assign_pointer(s->hnext, sta->hnext); 88 return 0; 89 } 90 91 return -ENOENT; 92} 93 94static void cleanup_single_sta(struct sta_info *sta) 95{ 96 int ac, i; 97 struct tid_ampdu_tx *tid_tx; 98 struct ieee80211_sub_if_data *sdata = sta->sdata; 99 struct ieee80211_local *local = sdata->local; 100 struct ps_data *ps; 101 102 /* 103 * At this point, when being called as call_rcu callback, 104 * neither mac80211 nor the driver can reference this 105 * sta struct any more except by still existing timers 106 * associated with this station that we clean up below. 107 * 108 * Note though that this still uses the sdata and even 109 * calls the driver in AP and mesh mode, so interfaces 110 * of those types mush use call sta_info_flush_cleanup() 111 * (typically via sta_info_flush()) before deconfiguring 112 * the driver. 113 * 114 * In station mode, nothing happens here so it doesn't 115 * have to (and doesn't) do that, this is intentional to 116 * speed up roaming. 117 */ 118 119 if (test_sta_flag(sta, WLAN_STA_PS_STA)) { 120 if (sta->sdata->vif.type == NL80211_IFTYPE_AP || 121 sta->sdata->vif.type == NL80211_IFTYPE_AP_VLAN) 122 ps = &sdata->bss->ps; 123 else 124 return; 125 126 clear_sta_flag(sta, WLAN_STA_PS_STA); 127 128 atomic_dec(&ps->num_sta_ps); 129 sta_info_recalc_tim(sta); 130 } 131 132 for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) { 133 local->total_ps_buffered -= skb_queue_len(&sta->ps_tx_buf[ac]); 134 ieee80211_purge_tx_queue(&local->hw, &sta->ps_tx_buf[ac]); 135 ieee80211_purge_tx_queue(&local->hw, &sta->tx_filtered[ac]); 136 } 137 138#ifdef CONFIG_MAC80211_MESH 139 if (ieee80211_vif_is_mesh(&sdata->vif)) { 140 mesh_accept_plinks_update(sdata); 141 mesh_plink_deactivate(sta); 142 del_timer_sync(&sta->plink_timer); 143 } 144#endif 145 146 cancel_work_sync(&sta->drv_unblock_wk); 147 148 /* 149 * Destroy aggregation state here. It would be nice to wait for the 150 * driver to finish aggregation stop and then clean up, but for now 151 * drivers have to handle aggregation stop being requested, followed 152 * directly by station destruction. 153 */ 154 for (i = 0; i < IEEE80211_NUM_TIDS; i++) { 155 tid_tx = rcu_dereference_raw(sta->ampdu_mlme.tid_tx[i]); 156 if (!tid_tx) 157 continue; 158 ieee80211_purge_tx_queue(&local->hw, &tid_tx->pending); 159 kfree(tid_tx); 160 } 161 162 sta_info_free(local, sta); 163} 164 165void ieee80211_cleanup_sdata_stas(struct ieee80211_sub_if_data *sdata) 166{ 167 struct sta_info *sta; 168 169 spin_lock_bh(&sdata->cleanup_stations_lock); 170 while (!list_empty(&sdata->cleanup_stations)) { 171 sta = list_first_entry(&sdata->cleanup_stations, 172 struct sta_info, list); 173 list_del(&sta->list); 174 spin_unlock_bh(&sdata->cleanup_stations_lock); 175 176 cleanup_single_sta(sta); 177 178 spin_lock_bh(&sdata->cleanup_stations_lock); 179 } 180 181 spin_unlock_bh(&sdata->cleanup_stations_lock); 182} 183 184static void free_sta_rcu(struct rcu_head *h) 185{ 186 struct sta_info *sta = container_of(h, struct sta_info, rcu_head); 187 struct ieee80211_sub_if_data *sdata = sta->sdata; 188 189 spin_lock(&sdata->cleanup_stations_lock); 190 list_add_tail(&sta->list, &sdata->cleanup_stations); 191 spin_unlock(&sdata->cleanup_stations_lock); 192 193 ieee80211_queue_work(&sdata->local->hw, &sdata->cleanup_stations_wk); 194} 195 196/* protected by RCU */ 197struct sta_info *sta_info_get(struct ieee80211_sub_if_data *sdata, 198 const u8 *addr) 199{ 200 struct ieee80211_local *local = sdata->local; 201 struct sta_info *sta; 202 203 sta = rcu_dereference_check(local->sta_hash[STA_HASH(addr)], 204 lockdep_is_held(&local->sta_mtx)); 205 while (sta) { 206 if (sta->sdata == sdata && 207 ether_addr_equal(sta->sta.addr, addr)) 208 break; 209 sta = rcu_dereference_check(sta->hnext, 210 lockdep_is_held(&local->sta_mtx)); 211 } 212 return sta; 213} 214 215/* 216 * Get sta info either from the specified interface 217 * or from one of its vlans 218 */ 219struct sta_info *sta_info_get_bss(struct ieee80211_sub_if_data *sdata, 220 const u8 *addr) 221{ 222 struct ieee80211_local *local = sdata->local; 223 struct sta_info *sta; 224 225 sta = rcu_dereference_check(local->sta_hash[STA_HASH(addr)], 226 lockdep_is_held(&local->sta_mtx)); 227 while (sta) { 228 if ((sta->sdata == sdata || 229 (sta->sdata->bss && sta->sdata->bss == sdata->bss)) && 230 ether_addr_equal(sta->sta.addr, addr)) 231 break; 232 sta = rcu_dereference_check(sta->hnext, 233 lockdep_is_held(&local->sta_mtx)); 234 } 235 return sta; 236} 237 238struct sta_info *sta_info_get_by_idx(struct ieee80211_sub_if_data *sdata, 239 int idx) 240{ 241 struct ieee80211_local *local = sdata->local; 242 struct sta_info *sta; 243 int i = 0; 244 245 list_for_each_entry_rcu(sta, &local->sta_list, list) { 246 if (sdata != sta->sdata) 247 continue; 248 if (i < idx) { 249 ++i; 250 continue; 251 } 252 return sta; 253 } 254 255 return NULL; 256} 257 258/** 259 * sta_info_free - free STA 260 * 261 * @local: pointer to the global information 262 * @sta: STA info to free 263 * 264 * This function must undo everything done by sta_info_alloc() 265 * that may happen before sta_info_insert(). It may only be 266 * called when sta_info_insert() has not been attempted (and 267 * if that fails, the station is freed anyway.) 268 */ 269void sta_info_free(struct ieee80211_local *local, struct sta_info *sta) 270{ 271 if (sta->rate_ctrl) 272 rate_control_free_sta(sta); 273 274 sta_dbg(sta->sdata, "Destroyed STA %pM\n", sta->sta.addr); 275 276 kfree(sta); 277} 278 279/* Caller must hold local->sta_mtx */ 280static void sta_info_hash_add(struct ieee80211_local *local, 281 struct sta_info *sta) 282{ 283 lockdep_assert_held(&local->sta_mtx); 284 sta->hnext = local->sta_hash[STA_HASH(sta->sta.addr)]; 285 rcu_assign_pointer(local->sta_hash[STA_HASH(sta->sta.addr)], sta); 286} 287 288static void sta_unblock(struct work_struct *wk) 289{ 290 struct sta_info *sta; 291 292 sta = container_of(wk, struct sta_info, drv_unblock_wk); 293 294 if (sta->dead) 295 return; 296 297 if (!test_sta_flag(sta, WLAN_STA_PS_STA)) { 298 local_bh_disable(); 299 ieee80211_sta_ps_deliver_wakeup(sta); 300 local_bh_enable(); 301 } else if (test_and_clear_sta_flag(sta, WLAN_STA_PSPOLL)) { 302 clear_sta_flag(sta, WLAN_STA_PS_DRIVER); 303 304 local_bh_disable(); 305 ieee80211_sta_ps_deliver_poll_response(sta); 306 local_bh_enable(); 307 } else if (test_and_clear_sta_flag(sta, WLAN_STA_UAPSD)) { 308 clear_sta_flag(sta, WLAN_STA_PS_DRIVER); 309 310 local_bh_disable(); 311 ieee80211_sta_ps_deliver_uapsd(sta); 312 local_bh_enable(); 313 } else 314 clear_sta_flag(sta, WLAN_STA_PS_DRIVER); 315} 316 317static int sta_prepare_rate_control(struct ieee80211_local *local, 318 struct sta_info *sta, gfp_t gfp) 319{ 320 if (local->hw.flags & IEEE80211_HW_HAS_RATE_CONTROL) 321 return 0; 322 323 sta->rate_ctrl = local->rate_ctrl; 324 sta->rate_ctrl_priv = rate_control_alloc_sta(sta->rate_ctrl, 325 &sta->sta, gfp); 326 if (!sta->rate_ctrl_priv) 327 return -ENOMEM; 328 329 return 0; 330} 331 332struct sta_info *sta_info_alloc(struct ieee80211_sub_if_data *sdata, 333 const u8 *addr, gfp_t gfp) 334{ 335 struct ieee80211_local *local = sdata->local; 336 struct sta_info *sta; 337 struct timespec uptime; 338 int i; 339 340 sta = kzalloc(sizeof(*sta) + local->hw.sta_data_size, gfp); 341 if (!sta) 342 return NULL; 343 344 spin_lock_init(&sta->lock); 345 INIT_WORK(&sta->drv_unblock_wk, sta_unblock); 346 INIT_WORK(&sta->ampdu_mlme.work, ieee80211_ba_session_work); 347 mutex_init(&sta->ampdu_mlme.mtx); 348 349 memcpy(sta->sta.addr, addr, ETH_ALEN); 350 sta->local = local; 351 sta->sdata = sdata; 352 sta->last_rx = jiffies; 353 354 sta->sta_state = IEEE80211_STA_NONE; 355 356 do_posix_clock_monotonic_gettime(&uptime); 357 sta->last_connected = uptime.tv_sec; 358 ewma_init(&sta->avg_signal, 1024, 8); 359 360 if (sta_prepare_rate_control(local, sta, gfp)) { 361 kfree(sta); 362 return NULL; 363 } 364 365 for (i = 0; i < IEEE80211_NUM_TIDS; i++) { 366 /* 367 * timer_to_tid must be initialized with identity mapping 368 * to enable session_timer's data differentiation. See 369 * sta_rx_agg_session_timer_expired for usage. 370 */ 371 sta->timer_to_tid[i] = i; 372 } 373 for (i = 0; i < IEEE80211_NUM_ACS; i++) { 374 skb_queue_head_init(&sta->ps_tx_buf[i]); 375 skb_queue_head_init(&sta->tx_filtered[i]); 376 } 377 378 for (i = 0; i < IEEE80211_NUM_TIDS; i++) 379 sta->last_seq_ctrl[i] = cpu_to_le16(USHRT_MAX); 380 381 sta_dbg(sdata, "Allocated STA %pM\n", sta->sta.addr); 382 383#ifdef CONFIG_MAC80211_MESH 384 sta->plink_state = NL80211_PLINK_LISTEN; 385 init_timer(&sta->plink_timer); 386#endif 387 388 return sta; 389} 390 391static int sta_info_insert_check(struct sta_info *sta) 392{ 393 struct ieee80211_sub_if_data *sdata = sta->sdata; 394 395 /* 396 * Can't be a WARN_ON because it can be triggered through a race: 397 * something inserts a STA (on one CPU) without holding the RTNL 398 * and another CPU turns off the net device. 399 */ 400 if (unlikely(!ieee80211_sdata_running(sdata))) 401 return -ENETDOWN; 402 403 if (WARN_ON(ether_addr_equal(sta->sta.addr, sdata->vif.addr) || 404 is_multicast_ether_addr(sta->sta.addr))) 405 return -EINVAL; 406 407 return 0; 408} 409 410static int sta_info_insert_drv_state(struct ieee80211_local *local, 411 struct ieee80211_sub_if_data *sdata, 412 struct sta_info *sta) 413{ 414 enum ieee80211_sta_state state; 415 int err = 0; 416 417 for (state = IEEE80211_STA_NOTEXIST; state < sta->sta_state; state++) { 418 err = drv_sta_state(local, sdata, sta, state, state + 1); 419 if (err) 420 break; 421 } 422 423 if (!err) { 424 /* 425 * Drivers using legacy sta_add/sta_remove callbacks only 426 * get uploaded set to true after sta_add is called. 427 */ 428 if (!local->ops->sta_add) 429 sta->uploaded = true; 430 return 0; 431 } 432 433 if (sdata->vif.type == NL80211_IFTYPE_ADHOC) { 434 sdata_info(sdata, 435 "failed to move IBSS STA %pM to state %d (%d) - keeping it anyway\n", 436 sta->sta.addr, state + 1, err); 437 err = 0; 438 } 439 440 /* unwind on error */ 441 for (; state > IEEE80211_STA_NOTEXIST; state--) 442 WARN_ON(drv_sta_state(local, sdata, sta, state, state - 1)); 443 444 return err; 445} 446 447/* 448 * should be called with sta_mtx locked 449 * this function replaces the mutex lock 450 * with a RCU lock 451 */ 452static int sta_info_insert_finish(struct sta_info *sta) __acquires(RCU) 453{ 454 struct ieee80211_local *local = sta->local; 455 struct ieee80211_sub_if_data *sdata = sta->sdata; 456 struct station_info sinfo; 457 int err = 0; 458 459 lockdep_assert_held(&local->sta_mtx); 460 461 /* check if STA exists already */ 462 if (sta_info_get_bss(sdata, sta->sta.addr)) { 463 err = -EEXIST; 464 goto out_err; 465 } 466 467 /* notify driver */ 468 err = sta_info_insert_drv_state(local, sdata, sta); 469 if (err) 470 goto out_err; 471 472 local->num_sta++; 473 local->sta_generation++; 474 smp_mb(); 475 476 /* make the station visible */ 477 sta_info_hash_add(local, sta); 478 479 list_add_rcu(&sta->list, &local->sta_list); 480 481 set_sta_flag(sta, WLAN_STA_INSERTED); 482 483 ieee80211_sta_debugfs_add(sta); 484 rate_control_add_sta_debugfs(sta); 485 486 memset(&sinfo, 0, sizeof(sinfo)); 487 sinfo.filled = 0; 488 sinfo.generation = local->sta_generation; 489 cfg80211_new_sta(sdata->dev, sta->sta.addr, &sinfo, GFP_KERNEL); 490 491 sta_dbg(sdata, "Inserted STA %pM\n", sta->sta.addr); 492 493 /* move reference to rcu-protected */ 494 rcu_read_lock(); 495 mutex_unlock(&local->sta_mtx); 496 497 if (ieee80211_vif_is_mesh(&sdata->vif)) 498 mesh_accept_plinks_update(sdata); 499 500 return 0; 501 out_err: 502 mutex_unlock(&local->sta_mtx); 503 rcu_read_lock(); 504 return err; 505} 506 507int sta_info_insert_rcu(struct sta_info *sta) __acquires(RCU) 508{ 509 struct ieee80211_local *local = sta->local; 510 int err = 0; 511 512 might_sleep(); 513 514 err = sta_info_insert_check(sta); 515 if (err) { 516 rcu_read_lock(); 517 goto out_free; 518 } 519 520 mutex_lock(&local->sta_mtx); 521 522 err = sta_info_insert_finish(sta); 523 if (err) 524 goto out_free; 525 526 return 0; 527 out_free: 528 BUG_ON(!err); 529 sta_info_free(local, sta); 530 return err; 531} 532 533int sta_info_insert(struct sta_info *sta) 534{ 535 int err = sta_info_insert_rcu(sta); 536 537 rcu_read_unlock(); 538 539 return err; 540} 541 542static inline void __bss_tim_set(u8 *tim, u16 id) 543{ 544 /* 545 * This format has been mandated by the IEEE specifications, 546 * so this line may not be changed to use the __set_bit() format. 547 */ 548 tim[id / 8] |= (1 << (id % 8)); 549} 550 551static inline void __bss_tim_clear(u8 *tim, u16 id) 552{ 553 /* 554 * This format has been mandated by the IEEE specifications, 555 * so this line may not be changed to use the __clear_bit() format. 556 */ 557 tim[id / 8] &= ~(1 << (id % 8)); 558} 559 560static unsigned long ieee80211_tids_for_ac(int ac) 561{ 562 /* If we ever support TIDs > 7, this obviously needs to be adjusted */ 563 switch (ac) { 564 case IEEE80211_AC_VO: 565 return BIT(6) | BIT(7); 566 case IEEE80211_AC_VI: 567 return BIT(4) | BIT(5); 568 case IEEE80211_AC_BE: 569 return BIT(0) | BIT(3); 570 case IEEE80211_AC_BK: 571 return BIT(1) | BIT(2); 572 default: 573 WARN_ON(1); 574 return 0; 575 } 576} 577 578void sta_info_recalc_tim(struct sta_info *sta) 579{ 580 struct ieee80211_local *local = sta->local; 581 struct ps_data *ps; 582 unsigned long flags; 583 bool indicate_tim = false; 584 u8 ignore_for_tim = sta->sta.uapsd_queues; 585 int ac; 586 u16 id; 587 588 if (sta->sdata->vif.type == NL80211_IFTYPE_AP || 589 sta->sdata->vif.type == NL80211_IFTYPE_AP_VLAN) { 590 if (WARN_ON_ONCE(!sta->sdata->bss)) 591 return; 592 593 ps = &sta->sdata->bss->ps; 594 id = sta->sta.aid; 595 } else { 596 return; 597 } 598 599 /* No need to do anything if the driver does all */ 600 if (local->hw.flags & IEEE80211_HW_AP_LINK_PS) 601 return; 602 603 if (sta->dead) 604 goto done; 605 606 /* 607 * If all ACs are delivery-enabled then we should build 608 * the TIM bit for all ACs anyway; if only some are then 609 * we ignore those and build the TIM bit using only the 610 * non-enabled ones. 611 */ 612 if (ignore_for_tim == BIT(IEEE80211_NUM_ACS) - 1) 613 ignore_for_tim = 0; 614 615 for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) { 616 unsigned long tids; 617 618 if (ignore_for_tim & BIT(ac)) 619 continue; 620 621 indicate_tim |= !skb_queue_empty(&sta->tx_filtered[ac]) || 622 !skb_queue_empty(&sta->ps_tx_buf[ac]); 623 if (indicate_tim) 624 break; 625 626 tids = ieee80211_tids_for_ac(ac); 627 628 indicate_tim |= 629 sta->driver_buffered_tids & tids; 630 } 631 632 done: 633 spin_lock_irqsave(&local->tim_lock, flags); 634 635 if (indicate_tim) 636 __bss_tim_set(ps->tim, id); 637 else 638 __bss_tim_clear(ps->tim, id); 639 640 if (local->ops->set_tim) { 641 local->tim_in_locked_section = true; 642 drv_set_tim(local, &sta->sta, indicate_tim); 643 local->tim_in_locked_section = false; 644 } 645 646 spin_unlock_irqrestore(&local->tim_lock, flags); 647} 648 649static bool sta_info_buffer_expired(struct sta_info *sta, struct sk_buff *skb) 650{ 651 struct ieee80211_tx_info *info; 652 int timeout; 653 654 if (!skb) 655 return false; 656 657 info = IEEE80211_SKB_CB(skb); 658 659 /* Timeout: (2 * listen_interval * beacon_int * 1024 / 1000000) sec */ 660 timeout = (sta->listen_interval * 661 sta->sdata->vif.bss_conf.beacon_int * 662 32 / 15625) * HZ; 663 if (timeout < STA_TX_BUFFER_EXPIRE) 664 timeout = STA_TX_BUFFER_EXPIRE; 665 return time_after(jiffies, info->control.jiffies + timeout); 666} 667 668 669static bool sta_info_cleanup_expire_buffered_ac(struct ieee80211_local *local, 670 struct sta_info *sta, int ac) 671{ 672 unsigned long flags; 673 struct sk_buff *skb; 674 675 /* 676 * First check for frames that should expire on the filtered 677 * queue. Frames here were rejected by the driver and are on 678 * a separate queue to avoid reordering with normal PS-buffered 679 * frames. They also aren't accounted for right now in the 680 * total_ps_buffered counter. 681 */ 682 for (;;) { 683 spin_lock_irqsave(&sta->tx_filtered[ac].lock, flags); 684 skb = skb_peek(&sta->tx_filtered[ac]); 685 if (sta_info_buffer_expired(sta, skb)) 686 skb = __skb_dequeue(&sta->tx_filtered[ac]); 687 else 688 skb = NULL; 689 spin_unlock_irqrestore(&sta->tx_filtered[ac].lock, flags); 690 691 /* 692 * Frames are queued in order, so if this one 693 * hasn't expired yet we can stop testing. If 694 * we actually reached the end of the queue we 695 * also need to stop, of course. 696 */ 697 if (!skb) 698 break; 699 ieee80211_free_txskb(&local->hw, skb); 700 } 701 702 /* 703 * Now also check the normal PS-buffered queue, this will 704 * only find something if the filtered queue was emptied 705 * since the filtered frames are all before the normal PS 706 * buffered frames. 707 */ 708 for (;;) { 709 spin_lock_irqsave(&sta->ps_tx_buf[ac].lock, flags); 710 skb = skb_peek(&sta->ps_tx_buf[ac]); 711 if (sta_info_buffer_expired(sta, skb)) 712 skb = __skb_dequeue(&sta->ps_tx_buf[ac]); 713 else 714 skb = NULL; 715 spin_unlock_irqrestore(&sta->ps_tx_buf[ac].lock, flags); 716 717 /* 718 * frames are queued in order, so if this one 719 * hasn't expired yet (or we reached the end of 720 * the queue) we can stop testing 721 */ 722 if (!skb) 723 break; 724 725 local->total_ps_buffered--; 726 ps_dbg(sta->sdata, "Buffered frame expired (STA %pM)\n", 727 sta->sta.addr); 728 ieee80211_free_txskb(&local->hw, skb); 729 } 730 731 /* 732 * Finally, recalculate the TIM bit for this station -- it might 733 * now be clear because the station was too slow to retrieve its 734 * frames. 735 */ 736 sta_info_recalc_tim(sta); 737 738 /* 739 * Return whether there are any frames still buffered, this is 740 * used to check whether the cleanup timer still needs to run, 741 * if there are no frames we don't need to rearm the timer. 742 */ 743 return !(skb_queue_empty(&sta->ps_tx_buf[ac]) && 744 skb_queue_empty(&sta->tx_filtered[ac])); 745} 746 747static bool sta_info_cleanup_expire_buffered(struct ieee80211_local *local, 748 struct sta_info *sta) 749{ 750 bool have_buffered = false; 751 int ac; 752 753 /* This is only necessary for stations on BSS interfaces */ 754 if (!sta->sdata->bss) 755 return false; 756 757 for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) 758 have_buffered |= 759 sta_info_cleanup_expire_buffered_ac(local, sta, ac); 760 761 return have_buffered; 762} 763 764int __must_check __sta_info_destroy(struct sta_info *sta) 765{ 766 struct ieee80211_local *local; 767 struct ieee80211_sub_if_data *sdata; 768 int ret, i; 769 770 might_sleep(); 771 772 if (!sta) 773 return -ENOENT; 774 775 local = sta->local; 776 sdata = sta->sdata; 777 778 lockdep_assert_held(&local->sta_mtx); 779 780 /* 781 * Before removing the station from the driver and 782 * rate control, it might still start new aggregation 783 * sessions -- block that to make sure the tear-down 784 * will be sufficient. 785 */ 786 set_sta_flag(sta, WLAN_STA_BLOCK_BA); 787 ieee80211_sta_tear_down_BA_sessions(sta, false); 788 789 ret = sta_info_hash_del(local, sta); 790 if (ret) 791 return ret; 792 793 list_del_rcu(&sta->list); 794 795 mutex_lock(&local->key_mtx); 796 for (i = 0; i < NUM_DEFAULT_KEYS; i++) 797 __ieee80211_key_free(key_mtx_dereference(local, sta->gtk[i])); 798 if (sta->ptk) 799 __ieee80211_key_free(key_mtx_dereference(local, sta->ptk)); 800 mutex_unlock(&local->key_mtx); 801 802 sta->dead = true; 803 804 local->num_sta--; 805 local->sta_generation++; 806 807 if (sdata->vif.type == NL80211_IFTYPE_AP_VLAN) 808 RCU_INIT_POINTER(sdata->u.vlan.sta, NULL); 809 810 while (sta->sta_state > IEEE80211_STA_NONE) { 811 ret = sta_info_move_state(sta, sta->sta_state - 1); 812 if (ret) { 813 WARN_ON_ONCE(1); 814 break; 815 } 816 } 817 818 if (sta->uploaded) { 819 ret = drv_sta_state(local, sdata, sta, IEEE80211_STA_NONE, 820 IEEE80211_STA_NOTEXIST); 821 WARN_ON_ONCE(ret != 0); 822 } 823 824 sta_dbg(sdata, "Removed STA %pM\n", sta->sta.addr); 825 826 cfg80211_del_sta(sdata->dev, sta->sta.addr, GFP_KERNEL); 827 828 rate_control_remove_sta_debugfs(sta); 829 ieee80211_sta_debugfs_remove(sta); 830 831 call_rcu(&sta->rcu_head, free_sta_rcu); 832 833 return 0; 834} 835 836int sta_info_destroy_addr(struct ieee80211_sub_if_data *sdata, const u8 *addr) 837{ 838 struct sta_info *sta; 839 int ret; 840 841 mutex_lock(&sdata->local->sta_mtx); 842 sta = sta_info_get(sdata, addr); 843 ret = __sta_info_destroy(sta); 844 mutex_unlock(&sdata->local->sta_mtx); 845 846 return ret; 847} 848 849int sta_info_destroy_addr_bss(struct ieee80211_sub_if_data *sdata, 850 const u8 *addr) 851{ 852 struct sta_info *sta; 853 int ret; 854 855 mutex_lock(&sdata->local->sta_mtx); 856 sta = sta_info_get_bss(sdata, addr); 857 ret = __sta_info_destroy(sta); 858 mutex_unlock(&sdata->local->sta_mtx); 859 860 return ret; 861} 862 863static void sta_info_cleanup(unsigned long data) 864{ 865 struct ieee80211_local *local = (struct ieee80211_local *) data; 866 struct sta_info *sta; 867 bool timer_needed = false; 868 869 rcu_read_lock(); 870 list_for_each_entry_rcu(sta, &local->sta_list, list) 871 if (sta_info_cleanup_expire_buffered(local, sta)) 872 timer_needed = true; 873 rcu_read_unlock(); 874 875 if (local->quiescing) 876 return; 877 878 if (!timer_needed) 879 return; 880 881 mod_timer(&local->sta_cleanup, 882 round_jiffies(jiffies + STA_INFO_CLEANUP_INTERVAL)); 883} 884 885void sta_info_init(struct ieee80211_local *local) 886{ 887 spin_lock_init(&local->tim_lock); 888 mutex_init(&local->sta_mtx); 889 INIT_LIST_HEAD(&local->sta_list); 890 891 setup_timer(&local->sta_cleanup, sta_info_cleanup, 892 (unsigned long)local); 893} 894 895void sta_info_stop(struct ieee80211_local *local) 896{ 897 del_timer_sync(&local->sta_cleanup); 898} 899 900 901int sta_info_flush_defer(struct ieee80211_sub_if_data *sdata) 902{ 903 struct ieee80211_local *local = sdata->local; 904 struct sta_info *sta, *tmp; 905 int ret = 0; 906 907 might_sleep(); 908 909 mutex_lock(&local->sta_mtx); 910 list_for_each_entry_safe(sta, tmp, &local->sta_list, list) { 911 if (sdata == sta->sdata) { 912 WARN_ON(__sta_info_destroy(sta)); 913 ret++; 914 } 915 } 916 mutex_unlock(&local->sta_mtx); 917 918 return ret; 919} 920 921void sta_info_flush_cleanup(struct ieee80211_sub_if_data *sdata) 922{ 923 ieee80211_cleanup_sdata_stas(sdata); 924 cancel_work_sync(&sdata->cleanup_stations_wk); 925} 926 927void ieee80211_sta_expire(struct ieee80211_sub_if_data *sdata, 928 unsigned long exp_time) 929{ 930 struct ieee80211_local *local = sdata->local; 931 struct sta_info *sta, *tmp; 932 933 mutex_lock(&local->sta_mtx); 934 935 list_for_each_entry_safe(sta, tmp, &local->sta_list, list) { 936 if (sdata != sta->sdata) 937 continue; 938 939 if (time_after(jiffies, sta->last_rx + exp_time)) { 940 sta_dbg(sta->sdata, "expiring inactive STA %pM\n", 941 sta->sta.addr); 942 WARN_ON(__sta_info_destroy(sta)); 943 } 944 } 945 946 mutex_unlock(&local->sta_mtx); 947} 948 949struct ieee80211_sta *ieee80211_find_sta_by_ifaddr(struct ieee80211_hw *hw, 950 const u8 *addr, 951 const u8 *localaddr) 952{ 953 struct sta_info *sta, *nxt; 954 955 /* 956 * Just return a random station if localaddr is NULL 957 * ... first in list. 958 */ 959 for_each_sta_info(hw_to_local(hw), addr, sta, nxt) { 960 if (localaddr && 961 !ether_addr_equal(sta->sdata->vif.addr, localaddr)) 962 continue; 963 if (!sta->uploaded) 964 return NULL; 965 return &sta->sta; 966 } 967 968 return NULL; 969} 970EXPORT_SYMBOL_GPL(ieee80211_find_sta_by_ifaddr); 971 972struct ieee80211_sta *ieee80211_find_sta(struct ieee80211_vif *vif, 973 const u8 *addr) 974{ 975 struct sta_info *sta; 976 977 if (!vif) 978 return NULL; 979 980 sta = sta_info_get_bss(vif_to_sdata(vif), addr); 981 if (!sta) 982 return NULL; 983 984 if (!sta->uploaded) 985 return NULL; 986 987 return &sta->sta; 988} 989EXPORT_SYMBOL(ieee80211_find_sta); 990 991static void clear_sta_ps_flags(void *_sta) 992{ 993 struct sta_info *sta = _sta; 994 struct ieee80211_sub_if_data *sdata = sta->sdata; 995 struct ps_data *ps; 996 997 if (sdata->vif.type == NL80211_IFTYPE_AP || 998 sdata->vif.type == NL80211_IFTYPE_AP_VLAN) 999 ps = &sdata->bss->ps; 1000 else 1001 return; 1002 1003 clear_sta_flag(sta, WLAN_STA_PS_DRIVER); 1004 if (test_and_clear_sta_flag(sta, WLAN_STA_PS_STA)) 1005 atomic_dec(&ps->num_sta_ps); 1006} 1007 1008/* powersave support code */ 1009void ieee80211_sta_ps_deliver_wakeup(struct sta_info *sta) 1010{ 1011 struct ieee80211_sub_if_data *sdata = sta->sdata; 1012 struct ieee80211_local *local = sdata->local; 1013 struct sk_buff_head pending; 1014 int filtered = 0, buffered = 0, ac; 1015 unsigned long flags; 1016 1017 clear_sta_flag(sta, WLAN_STA_SP); 1018 1019 BUILD_BUG_ON(BITS_TO_LONGS(IEEE80211_NUM_TIDS) > 1); 1020 sta->driver_buffered_tids = 0; 1021 1022 if (!(local->hw.flags & IEEE80211_HW_AP_LINK_PS)) 1023 drv_sta_notify(local, sdata, STA_NOTIFY_AWAKE, &sta->sta); 1024 1025 skb_queue_head_init(&pending); 1026 1027 /* Send all buffered frames to the station */ 1028 for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) { 1029 int count = skb_queue_len(&pending), tmp; 1030 1031 spin_lock_irqsave(&sta->tx_filtered[ac].lock, flags); 1032 skb_queue_splice_tail_init(&sta->tx_filtered[ac], &pending); 1033 spin_unlock_irqrestore(&sta->tx_filtered[ac].lock, flags); 1034 tmp = skb_queue_len(&pending); 1035 filtered += tmp - count; 1036 count = tmp; 1037 1038 spin_lock_irqsave(&sta->ps_tx_buf[ac].lock, flags); 1039 skb_queue_splice_tail_init(&sta->ps_tx_buf[ac], &pending); 1040 spin_unlock_irqrestore(&sta->ps_tx_buf[ac].lock, flags); 1041 tmp = skb_queue_len(&pending); 1042 buffered += tmp - count; 1043 } 1044 1045 ieee80211_add_pending_skbs_fn(local, &pending, clear_sta_ps_flags, sta); 1046 1047 local->total_ps_buffered -= buffered; 1048 1049 sta_info_recalc_tim(sta); 1050 1051 ps_dbg(sdata, 1052 "STA %pM aid %d sending %d filtered/%d PS frames since STA not sleeping anymore\n", 1053 sta->sta.addr, sta->sta.aid, filtered, buffered); 1054} 1055 1056static void ieee80211_send_null_response(struct ieee80211_sub_if_data *sdata, 1057 struct sta_info *sta, int tid, 1058 enum ieee80211_frame_release_type reason) 1059{ 1060 struct ieee80211_local *local = sdata->local; 1061 struct ieee80211_qos_hdr *nullfunc; 1062 struct sk_buff *skb; 1063 int size = sizeof(*nullfunc); 1064 __le16 fc; 1065 bool qos = test_sta_flag(sta, WLAN_STA_WME); 1066 struct ieee80211_tx_info *info; 1067 struct ieee80211_chanctx_conf *chanctx_conf; 1068 1069 if (qos) { 1070 fc = cpu_to_le16(IEEE80211_FTYPE_DATA | 1071 IEEE80211_STYPE_QOS_NULLFUNC | 1072 IEEE80211_FCTL_FROMDS); 1073 } else { 1074 size -= 2; 1075 fc = cpu_to_le16(IEEE80211_FTYPE_DATA | 1076 IEEE80211_STYPE_NULLFUNC | 1077 IEEE80211_FCTL_FROMDS); 1078 } 1079 1080 skb = dev_alloc_skb(local->hw.extra_tx_headroom + size); 1081 if (!skb) 1082 return; 1083 1084 skb_reserve(skb, local->hw.extra_tx_headroom); 1085 1086 nullfunc = (void *) skb_put(skb, size); 1087 nullfunc->frame_control = fc; 1088 nullfunc->duration_id = 0; 1089 memcpy(nullfunc->addr1, sta->sta.addr, ETH_ALEN); 1090 memcpy(nullfunc->addr2, sdata->vif.addr, ETH_ALEN); 1091 memcpy(nullfunc->addr3, sdata->vif.addr, ETH_ALEN); 1092 1093 skb->priority = tid; 1094 skb_set_queue_mapping(skb, ieee802_1d_to_ac[tid]); 1095 if (qos) { 1096 nullfunc->qos_ctrl = cpu_to_le16(tid); 1097 1098 if (reason == IEEE80211_FRAME_RELEASE_UAPSD) 1099 nullfunc->qos_ctrl |= 1100 cpu_to_le16(IEEE80211_QOS_CTL_EOSP); 1101 } 1102 1103 info = IEEE80211_SKB_CB(skb); 1104 1105 /* 1106 * Tell TX path to send this frame even though the 1107 * STA may still remain is PS mode after this frame 1108 * exchange. Also set EOSP to indicate this packet 1109 * ends the poll/service period. 1110 */ 1111 info->flags |= IEEE80211_TX_CTL_NO_PS_BUFFER | 1112 IEEE80211_TX_STATUS_EOSP | 1113 IEEE80211_TX_CTL_REQ_TX_STATUS; 1114 1115 drv_allow_buffered_frames(local, sta, BIT(tid), 1, reason, false); 1116 1117 rcu_read_lock(); 1118 chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf); 1119 if (WARN_ON(!chanctx_conf)) { 1120 rcu_read_unlock(); 1121 kfree_skb(skb); 1122 return; 1123 } 1124 1125 ieee80211_xmit(sdata, skb, chanctx_conf->def.chan->band); 1126 rcu_read_unlock(); 1127} 1128 1129static void 1130ieee80211_sta_ps_deliver_response(struct sta_info *sta, 1131 int n_frames, u8 ignored_acs, 1132 enum ieee80211_frame_release_type reason) 1133{ 1134 struct ieee80211_sub_if_data *sdata = sta->sdata; 1135 struct ieee80211_local *local = sdata->local; 1136 bool found = false; 1137 bool more_data = false; 1138 int ac; 1139 unsigned long driver_release_tids = 0; 1140 struct sk_buff_head frames; 1141 1142 /* Service or PS-Poll period starts */ 1143 set_sta_flag(sta, WLAN_STA_SP); 1144 1145 __skb_queue_head_init(&frames); 1146 1147 /* 1148 * Get response frame(s) and more data bit for it. 1149 */ 1150 for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) { 1151 unsigned long tids; 1152 1153 if (ignored_acs & BIT(ac)) 1154 continue; 1155 1156 tids = ieee80211_tids_for_ac(ac); 1157 1158 if (!found) { 1159 driver_release_tids = sta->driver_buffered_tids & tids; 1160 if (driver_release_tids) { 1161 found = true; 1162 } else { 1163 struct sk_buff *skb; 1164 1165 while (n_frames > 0) { 1166 skb = skb_dequeue(&sta->tx_filtered[ac]); 1167 if (!skb) { 1168 skb = skb_dequeue( 1169 &sta->ps_tx_buf[ac]); 1170 if (skb) 1171 local->total_ps_buffered--; 1172 } 1173 if (!skb) 1174 break; 1175 n_frames--; 1176 found = true; 1177 __skb_queue_tail(&frames, skb); 1178 } 1179 } 1180 1181 /* 1182 * If the driver has data on more than one TID then 1183 * certainly there's more data if we release just a 1184 * single frame now (from a single TID). 1185 */ 1186 if (reason == IEEE80211_FRAME_RELEASE_PSPOLL && 1187 hweight16(driver_release_tids) > 1) { 1188 more_data = true; 1189 driver_release_tids = 1190 BIT(ffs(driver_release_tids) - 1); 1191 break; 1192 } 1193 } 1194 1195 if (!skb_queue_empty(&sta->tx_filtered[ac]) || 1196 !skb_queue_empty(&sta->ps_tx_buf[ac])) { 1197 more_data = true; 1198 break; 1199 } 1200 } 1201 1202 if (!found) { 1203 int tid; 1204 1205 /* 1206 * For PS-Poll, this can only happen due to a race condition 1207 * when we set the TIM bit and the station notices it, but 1208 * before it can poll for the frame we expire it. 1209 * 1210 * For uAPSD, this is said in the standard (11.2.1.5 h): 1211 * At each unscheduled SP for a non-AP STA, the AP shall 1212 * attempt to transmit at least one MSDU or MMPDU, but no 1213 * more than the value specified in the Max SP Length field 1214 * in the QoS Capability element from delivery-enabled ACs, 1215 * that are destined for the non-AP STA. 1216 * 1217 * Since we have no other MSDU/MMPDU, transmit a QoS null frame. 1218 */ 1219 1220 /* This will evaluate to 1, 3, 5 or 7. */ 1221 tid = 7 - ((ffs(~ignored_acs) - 1) << 1); 1222 1223 ieee80211_send_null_response(sdata, sta, tid, reason); 1224 return; 1225 } 1226 1227 if (!driver_release_tids) { 1228 struct sk_buff_head pending; 1229 struct sk_buff *skb; 1230 int num = 0; 1231 u16 tids = 0; 1232 1233 skb_queue_head_init(&pending); 1234 1235 while ((skb = __skb_dequeue(&frames))) { 1236 struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb); 1237 struct ieee80211_hdr *hdr = (void *) skb->data; 1238 u8 *qoshdr = NULL; 1239 1240 num++; 1241 1242 /* 1243 * Tell TX path to send this frame even though the 1244 * STA may still remain is PS mode after this frame 1245 * exchange. 1246 */ 1247 info->flags |= IEEE80211_TX_CTL_NO_PS_BUFFER; 1248 1249 /* 1250 * Use MoreData flag to indicate whether there are 1251 * more buffered frames for this STA 1252 */ 1253 if (more_data || !skb_queue_empty(&frames)) 1254 hdr->frame_control |= 1255 cpu_to_le16(IEEE80211_FCTL_MOREDATA); 1256 else 1257 hdr->frame_control &= 1258 cpu_to_le16(~IEEE80211_FCTL_MOREDATA); 1259 1260 if (ieee80211_is_data_qos(hdr->frame_control) || 1261 ieee80211_is_qos_nullfunc(hdr->frame_control)) 1262 qoshdr = ieee80211_get_qos_ctl(hdr); 1263 1264 /* end service period after last frame */ 1265 if (skb_queue_empty(&frames)) { 1266 if (reason == IEEE80211_FRAME_RELEASE_UAPSD && 1267 qoshdr) 1268 *qoshdr |= IEEE80211_QOS_CTL_EOSP; 1269 1270 info->flags |= IEEE80211_TX_STATUS_EOSP | 1271 IEEE80211_TX_CTL_REQ_TX_STATUS; 1272 } 1273 1274 if (qoshdr) 1275 tids |= BIT(*qoshdr & IEEE80211_QOS_CTL_TID_MASK); 1276 else 1277 tids |= BIT(0); 1278 1279 __skb_queue_tail(&pending, skb); 1280 } 1281 1282 drv_allow_buffered_frames(local, sta, tids, num, 1283 reason, more_data); 1284 1285 ieee80211_add_pending_skbs(local, &pending); 1286 1287 sta_info_recalc_tim(sta); 1288 } else { 1289 /* 1290 * We need to release a frame that is buffered somewhere in the 1291 * driver ... it'll have to handle that. 1292 * Note that, as per the comment above, it'll also have to see 1293 * if there is more than just one frame on the specific TID that 1294 * we're releasing from, and it needs to set the more-data bit 1295 * accordingly if we tell it that there's no more data. If we do 1296 * tell it there's more data, then of course the more-data bit 1297 * needs to be set anyway. 1298 */ 1299 drv_release_buffered_frames(local, sta, driver_release_tids, 1300 n_frames, reason, more_data); 1301 1302 /* 1303 * Note that we don't recalculate the TIM bit here as it would 1304 * most likely have no effect at all unless the driver told us 1305 * that the TID became empty before returning here from the 1306 * release function. 1307 * Either way, however, when the driver tells us that the TID 1308 * became empty we'll do the TIM recalculation. 1309 */ 1310 } 1311} 1312 1313void ieee80211_sta_ps_deliver_poll_response(struct sta_info *sta) 1314{ 1315 u8 ignore_for_response = sta->sta.uapsd_queues; 1316 1317 /* 1318 * If all ACs are delivery-enabled then we should reply 1319 * from any of them, if only some are enabled we reply 1320 * only from the non-enabled ones. 1321 */ 1322 if (ignore_for_response == BIT(IEEE80211_NUM_ACS) - 1) 1323 ignore_for_response = 0; 1324 1325 ieee80211_sta_ps_deliver_response(sta, 1, ignore_for_response, 1326 IEEE80211_FRAME_RELEASE_PSPOLL); 1327} 1328 1329void ieee80211_sta_ps_deliver_uapsd(struct sta_info *sta) 1330{ 1331 int n_frames = sta->sta.max_sp; 1332 u8 delivery_enabled = sta->sta.uapsd_queues; 1333 1334 /* 1335 * If we ever grow support for TSPEC this might happen if 1336 * the TSPEC update from hostapd comes in between a trigger 1337 * frame setting WLAN_STA_UAPSD in the RX path and this 1338 * actually getting called. 1339 */ 1340 if (!delivery_enabled) 1341 return; 1342 1343 switch (sta->sta.max_sp) { 1344 case 1: 1345 n_frames = 2; 1346 break; 1347 case 2: 1348 n_frames = 4; 1349 break; 1350 case 3: 1351 n_frames = 6; 1352 break; 1353 case 0: 1354 /* XXX: what is a good value? */ 1355 n_frames = 8; 1356 break; 1357 } 1358 1359 ieee80211_sta_ps_deliver_response(sta, n_frames, ~delivery_enabled, 1360 IEEE80211_FRAME_RELEASE_UAPSD); 1361} 1362 1363void ieee80211_sta_block_awake(struct ieee80211_hw *hw, 1364 struct ieee80211_sta *pubsta, bool block) 1365{ 1366 struct sta_info *sta = container_of(pubsta, struct sta_info, sta); 1367 1368 trace_api_sta_block_awake(sta->local, pubsta, block); 1369 1370 if (block) 1371 set_sta_flag(sta, WLAN_STA_PS_DRIVER); 1372 else if (test_sta_flag(sta, WLAN_STA_PS_DRIVER)) 1373 ieee80211_queue_work(hw, &sta->drv_unblock_wk); 1374} 1375EXPORT_SYMBOL(ieee80211_sta_block_awake); 1376 1377void ieee80211_sta_eosp_irqsafe(struct ieee80211_sta *pubsta) 1378{ 1379 struct sta_info *sta = container_of(pubsta, struct sta_info, sta); 1380 struct ieee80211_local *local = sta->local; 1381 struct sk_buff *skb; 1382 struct skb_eosp_msg_data *data; 1383 1384 trace_api_eosp(local, pubsta); 1385 1386 skb = alloc_skb(0, GFP_ATOMIC); 1387 if (!skb) { 1388 /* too bad ... but race is better than loss */ 1389 clear_sta_flag(sta, WLAN_STA_SP); 1390 return; 1391 } 1392 1393 data = (void *)skb->cb; 1394 memcpy(data->sta, pubsta->addr, ETH_ALEN); 1395 memcpy(data->iface, sta->sdata->vif.addr, ETH_ALEN); 1396 skb->pkt_type = IEEE80211_EOSP_MSG; 1397 skb_queue_tail(&local->skb_queue, skb); 1398 tasklet_schedule(&local->tasklet); 1399} 1400EXPORT_SYMBOL(ieee80211_sta_eosp_irqsafe); 1401 1402void ieee80211_sta_set_buffered(struct ieee80211_sta *pubsta, 1403 u8 tid, bool buffered) 1404{ 1405 struct sta_info *sta = container_of(pubsta, struct sta_info, sta); 1406 1407 if (WARN_ON(tid >= IEEE80211_NUM_TIDS)) 1408 return; 1409 1410 if (buffered) 1411 set_bit(tid, &sta->driver_buffered_tids); 1412 else 1413 clear_bit(tid, &sta->driver_buffered_tids); 1414 1415 sta_info_recalc_tim(sta); 1416} 1417EXPORT_SYMBOL(ieee80211_sta_set_buffered); 1418 1419int sta_info_move_state(struct sta_info *sta, 1420 enum ieee80211_sta_state new_state) 1421{ 1422 might_sleep(); 1423 1424 if (sta->sta_state == new_state) 1425 return 0; 1426 1427 /* check allowed transitions first */ 1428 1429 switch (new_state) { 1430 case IEEE80211_STA_NONE: 1431 if (sta->sta_state != IEEE80211_STA_AUTH) 1432 return -EINVAL; 1433 break; 1434 case IEEE80211_STA_AUTH: 1435 if (sta->sta_state != IEEE80211_STA_NONE && 1436 sta->sta_state != IEEE80211_STA_ASSOC) 1437 return -EINVAL; 1438 break; 1439 case IEEE80211_STA_ASSOC: 1440 if (sta->sta_state != IEEE80211_STA_AUTH && 1441 sta->sta_state != IEEE80211_STA_AUTHORIZED) 1442 return -EINVAL; 1443 break; 1444 case IEEE80211_STA_AUTHORIZED: 1445 if (sta->sta_state != IEEE80211_STA_ASSOC) 1446 return -EINVAL; 1447 break; 1448 default: 1449 WARN(1, "invalid state %d", new_state); 1450 return -EINVAL; 1451 } 1452 1453 sta_dbg(sta->sdata, "moving STA %pM to state %d\n", 1454 sta->sta.addr, new_state); 1455 1456 /* 1457 * notify the driver before the actual changes so it can 1458 * fail the transition 1459 */ 1460 if (test_sta_flag(sta, WLAN_STA_INSERTED)) { 1461 int err = drv_sta_state(sta->local, sta->sdata, sta, 1462 sta->sta_state, new_state); 1463 if (err) 1464 return err; 1465 } 1466 1467 /* reflect the change in all state variables */ 1468 1469 switch (new_state) { 1470 case IEEE80211_STA_NONE: 1471 if (sta->sta_state == IEEE80211_STA_AUTH) 1472 clear_bit(WLAN_STA_AUTH, &sta->_flags); 1473 break; 1474 case IEEE80211_STA_AUTH: 1475 if (sta->sta_state == IEEE80211_STA_NONE) 1476 set_bit(WLAN_STA_AUTH, &sta->_flags); 1477 else if (sta->sta_state == IEEE80211_STA_ASSOC) 1478 clear_bit(WLAN_STA_ASSOC, &sta->_flags); 1479 break; 1480 case IEEE80211_STA_ASSOC: 1481 if (sta->sta_state == IEEE80211_STA_AUTH) { 1482 set_bit(WLAN_STA_ASSOC, &sta->_flags); 1483 } else if (sta->sta_state == IEEE80211_STA_AUTHORIZED) { 1484 if (sta->sdata->vif.type == NL80211_IFTYPE_AP || 1485 (sta->sdata->vif.type == NL80211_IFTYPE_AP_VLAN && 1486 !sta->sdata->u.vlan.sta)) 1487 atomic_dec(&sta->sdata->bss->num_mcast_sta); 1488 clear_bit(WLAN_STA_AUTHORIZED, &sta->_flags); 1489 } 1490 break; 1491 case IEEE80211_STA_AUTHORIZED: 1492 if (sta->sta_state == IEEE80211_STA_ASSOC) { 1493 if (sta->sdata->vif.type == NL80211_IFTYPE_AP || 1494 (sta->sdata->vif.type == NL80211_IFTYPE_AP_VLAN && 1495 !sta->sdata->u.vlan.sta)) 1496 atomic_inc(&sta->sdata->bss->num_mcast_sta); 1497 set_bit(WLAN_STA_AUTHORIZED, &sta->_flags); 1498 } 1499 break; 1500 default: 1501 break; 1502 } 1503 1504 sta->sta_state = new_state; 1505 1506 return 0; 1507} 1508