sta_info.c 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791
  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. #include <linux/module.h>
  10. #include <linux/init.h>
  11. #include <linux/netdevice.h>
  12. #include <linux/types.h>
  13. #include <linux/slab.h>
  14. #include <linux/skbuff.h>
  15. #include <linux/if_arp.h>
  16. #include <linux/timer.h>
  17. #include <linux/rtnetlink.h>
  18. #include <net/mac80211.h>
  19. #include "ieee80211_i.h"
  20. #include "rate.h"
  21. #include "sta_info.h"
  22. #include "debugfs_sta.h"
  23. #include "mesh.h"
  24. /**
  25. * DOC: STA information lifetime rules
  26. *
  27. * STA info structures (&struct sta_info) are managed in a hash table
  28. * for faster lookup and a list for iteration. They are managed using
  29. * RCU, i.e. access to the list and hash table is protected by RCU.
  30. *
  31. * Upon allocating a STA info structure with sta_info_alloc(), the caller owns
  32. * that structure. It must then either destroy it using sta_info_destroy()
  33. * (which is pretty useless) or insert it into the hash table using
  34. * sta_info_insert() which demotes the reference from ownership to a regular
  35. * RCU-protected reference; if the function is called without protection by an
  36. * RCU critical section the reference is instantly invalidated. Note that the
  37. * caller may not do much with the STA info before inserting it, in particular,
  38. * it may not start any mesh peer link management or add encryption keys.
  39. *
  40. * When the insertion fails (sta_info_insert()) returns non-zero), the
  41. * structure will have been freed by sta_info_insert()!
  42. *
  43. * Because there are debugfs entries for each station, and adding those
  44. * must be able to sleep, it is also possible to "pin" a station entry,
  45. * that means it can be removed from the hash table but not be freed.
  46. * See the comment in __sta_info_unlink() for more information, this is
  47. * an internal capability only.
  48. *
  49. * In order to remove a STA info structure, the caller needs to first
  50. * unlink it (sta_info_unlink()) from the list and hash tables and
  51. * then destroy it; sta_info_destroy() will wait for an RCU grace period
  52. * to elapse before actually freeing it. Due to the pinning and the
  53. * possibility of multiple callers trying to remove the same STA info at
  54. * the same time, sta_info_unlink() can clear the STA info pointer it is
  55. * passed to indicate that the STA info is owned by somebody else now.
  56. *
  57. * If sta_info_unlink() did not clear the pointer then the caller owns
  58. * the STA info structure now and is responsible of destroying it with
  59. * a call to sta_info_destroy().
  60. *
  61. * In all other cases, there is no concept of ownership on a STA entry,
  62. * each structure is owned by the global hash table/list until it is
  63. * removed. All users of the structure need to be RCU protected so that
  64. * the structure won't be freed before they are done using it.
  65. */
  66. /* Caller must hold local->sta_lock */
  67. static int sta_info_hash_del(struct ieee80211_local *local,
  68. struct sta_info *sta)
  69. {
  70. struct sta_info *s;
  71. s = local->sta_hash[STA_HASH(sta->addr)];
  72. if (!s)
  73. return -ENOENT;
  74. if (s == sta) {
  75. rcu_assign_pointer(local->sta_hash[STA_HASH(sta->addr)],
  76. s->hnext);
  77. return 0;
  78. }
  79. while (s->hnext && s->hnext != sta)
  80. s = s->hnext;
  81. if (s->hnext) {
  82. rcu_assign_pointer(s->hnext, sta->hnext);
  83. return 0;
  84. }
  85. return -ENOENT;
  86. }
  87. /* protected by RCU */
  88. static struct sta_info *__sta_info_find(struct ieee80211_local *local,
  89. u8 *addr)
  90. {
  91. struct sta_info *sta;
  92. sta = rcu_dereference(local->sta_hash[STA_HASH(addr)]);
  93. while (sta) {
  94. if (compare_ether_addr(sta->addr, addr) == 0)
  95. break;
  96. sta = rcu_dereference(sta->hnext);
  97. }
  98. return sta;
  99. }
  100. struct sta_info *sta_info_get(struct ieee80211_local *local, u8 *addr)
  101. {
  102. return __sta_info_find(local, addr);
  103. }
  104. EXPORT_SYMBOL(sta_info_get);
  105. struct sta_info *sta_info_get_by_idx(struct ieee80211_local *local, int idx,
  106. struct net_device *dev)
  107. {
  108. struct sta_info *sta;
  109. int i = 0;
  110. list_for_each_entry_rcu(sta, &local->sta_list, list) {
  111. if (dev && dev != sta->sdata->dev)
  112. continue;
  113. if (i < idx) {
  114. ++i;
  115. continue;
  116. }
  117. return sta;
  118. }
  119. return NULL;
  120. }
  121. /**
  122. * __sta_info_free - internal STA free helper
  123. *
  124. * @sta: STA info to free
  125. *
  126. * This function must undo everything done by sta_info_alloc()
  127. * that may happen before sta_info_insert().
  128. */
  129. static void __sta_info_free(struct ieee80211_local *local,
  130. struct sta_info *sta)
  131. {
  132. DECLARE_MAC_BUF(mbuf);
  133. rate_control_free_sta(sta->rate_ctrl, sta->rate_ctrl_priv);
  134. rate_control_put(sta->rate_ctrl);
  135. #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
  136. printk(KERN_DEBUG "%s: Destroyed STA %s\n",
  137. wiphy_name(local->hw.wiphy), print_mac(mbuf, sta->addr));
  138. #endif /* CONFIG_MAC80211_VERBOSE_DEBUG */
  139. kfree(sta);
  140. }
  141. void sta_info_destroy(struct sta_info *sta)
  142. {
  143. struct ieee80211_local *local;
  144. struct sk_buff *skb;
  145. int i;
  146. might_sleep();
  147. if (!sta)
  148. return;
  149. local = sta->local;
  150. rate_control_remove_sta_debugfs(sta);
  151. ieee80211_sta_debugfs_remove(sta);
  152. #ifdef CONFIG_MAC80211_MESH
  153. if (ieee80211_vif_is_mesh(&sta->sdata->vif))
  154. mesh_plink_deactivate(sta);
  155. #endif
  156. /*
  157. * We have only unlinked the key, and actually destroying it
  158. * may mean it is removed from hardware which requires that
  159. * the key->sta pointer is still valid, so flush the key todo
  160. * list here.
  161. *
  162. * ieee80211_key_todo() will synchronize_rcu() so after this
  163. * nothing can reference this sta struct any more.
  164. */
  165. ieee80211_key_todo();
  166. #ifdef CONFIG_MAC80211_MESH
  167. if (ieee80211_vif_is_mesh(&sta->sdata->vif))
  168. del_timer_sync(&sta->plink_timer);
  169. #endif
  170. while ((skb = skb_dequeue(&sta->ps_tx_buf)) != NULL) {
  171. local->total_ps_buffered--;
  172. dev_kfree_skb_any(skb);
  173. }
  174. while ((skb = skb_dequeue(&sta->tx_filtered)) != NULL)
  175. dev_kfree_skb_any(skb);
  176. for (i = 0; i < STA_TID_NUM; i++) {
  177. spin_lock_bh(&sta->ampdu_mlme.ampdu_rx);
  178. if (sta->ampdu_mlme.tid_rx[i])
  179. del_timer_sync(&sta->ampdu_mlme.tid_rx[i]->session_timer);
  180. spin_unlock_bh(&sta->ampdu_mlme.ampdu_rx);
  181. spin_lock_bh(&sta->ampdu_mlme.ampdu_tx);
  182. if (sta->ampdu_mlme.tid_tx[i])
  183. del_timer_sync(&sta->ampdu_mlme.tid_tx[i]->addba_resp_timer);
  184. spin_unlock_bh(&sta->ampdu_mlme.ampdu_tx);
  185. }
  186. __sta_info_free(local, sta);
  187. }
  188. /* Caller must hold local->sta_lock */
  189. static void sta_info_hash_add(struct ieee80211_local *local,
  190. struct sta_info *sta)
  191. {
  192. sta->hnext = local->sta_hash[STA_HASH(sta->addr)];
  193. rcu_assign_pointer(local->sta_hash[STA_HASH(sta->addr)], sta);
  194. }
  195. struct sta_info *sta_info_alloc(struct ieee80211_sub_if_data *sdata,
  196. u8 *addr, gfp_t gfp)
  197. {
  198. struct ieee80211_local *local = sdata->local;
  199. struct sta_info *sta;
  200. int i;
  201. DECLARE_MAC_BUF(mbuf);
  202. sta = kzalloc(sizeof(*sta), gfp);
  203. if (!sta)
  204. return NULL;
  205. memcpy(sta->addr, addr, ETH_ALEN);
  206. sta->local = local;
  207. sta->sdata = sdata;
  208. sta->rate_ctrl = rate_control_get(local->rate_ctrl);
  209. sta->rate_ctrl_priv = rate_control_alloc_sta(sta->rate_ctrl,
  210. gfp);
  211. if (!sta->rate_ctrl_priv) {
  212. rate_control_put(sta->rate_ctrl);
  213. kfree(sta);
  214. return NULL;
  215. }
  216. spin_lock_init(&sta->ampdu_mlme.ampdu_rx);
  217. spin_lock_init(&sta->ampdu_mlme.ampdu_tx);
  218. for (i = 0; i < STA_TID_NUM; i++) {
  219. /* timer_to_tid must be initialized with identity mapping to
  220. * enable session_timer's data differentiation. refer to
  221. * sta_rx_agg_session_timer_expired for useage */
  222. sta->timer_to_tid[i] = i;
  223. /* tid to tx queue: initialize according to HW (0 is valid) */
  224. sta->tid_to_tx_q[i] = local->hw.queues;
  225. /* rx */
  226. sta->ampdu_mlme.tid_state_rx[i] = HT_AGG_STATE_IDLE;
  227. sta->ampdu_mlme.tid_rx[i] = NULL;
  228. /* tx */
  229. sta->ampdu_mlme.tid_state_tx[i] = HT_AGG_STATE_IDLE;
  230. sta->ampdu_mlme.tid_tx[i] = NULL;
  231. sta->ampdu_mlme.addba_req_num[i] = 0;
  232. }
  233. skb_queue_head_init(&sta->ps_tx_buf);
  234. skb_queue_head_init(&sta->tx_filtered);
  235. #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
  236. printk(KERN_DEBUG "%s: Allocated STA %s\n",
  237. wiphy_name(local->hw.wiphy), print_mac(mbuf, sta->addr));
  238. #endif /* CONFIG_MAC80211_VERBOSE_DEBUG */
  239. #ifdef CONFIG_MAC80211_MESH
  240. sta->plink_state = PLINK_LISTEN;
  241. spin_lock_init(&sta->plink_lock);
  242. init_timer(&sta->plink_timer);
  243. #endif
  244. return sta;
  245. }
  246. int sta_info_insert(struct sta_info *sta)
  247. {
  248. struct ieee80211_local *local = sta->local;
  249. struct ieee80211_sub_if_data *sdata = sta->sdata;
  250. unsigned long flags;
  251. int err = 0;
  252. DECLARE_MAC_BUF(mac);
  253. /*
  254. * Can't be a WARN_ON because it can be triggered through a race:
  255. * something inserts a STA (on one CPU) without holding the RTNL
  256. * and another CPU turns off the net device.
  257. */
  258. if (unlikely(!netif_running(sdata->dev))) {
  259. err = -ENETDOWN;
  260. goto out_free;
  261. }
  262. if (WARN_ON(compare_ether_addr(sta->addr, sdata->dev->dev_addr) == 0 ||
  263. is_multicast_ether_addr(sta->addr))) {
  264. err = -EINVAL;
  265. goto out_free;
  266. }
  267. spin_lock_irqsave(&local->sta_lock, flags);
  268. /* check if STA exists already */
  269. if (__sta_info_find(local, sta->addr)) {
  270. spin_unlock_irqrestore(&local->sta_lock, flags);
  271. err = -EEXIST;
  272. goto out_free;
  273. }
  274. list_add(&sta->list, &local->sta_list);
  275. local->num_sta++;
  276. sta_info_hash_add(local, sta);
  277. /* notify driver */
  278. if (local->ops->sta_notify) {
  279. if (sdata->vif.type == IEEE80211_IF_TYPE_VLAN)
  280. sdata = sdata->u.vlan.ap;
  281. local->ops->sta_notify(local_to_hw(local), &sdata->vif,
  282. STA_NOTIFY_ADD, sta->addr);
  283. }
  284. #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
  285. printk(KERN_DEBUG "%s: Inserted STA %s\n",
  286. wiphy_name(local->hw.wiphy), print_mac(mac, sta->addr));
  287. #endif /* CONFIG_MAC80211_VERBOSE_DEBUG */
  288. spin_unlock_irqrestore(&local->sta_lock, flags);
  289. #ifdef CONFIG_MAC80211_DEBUGFS
  290. /*
  291. * Debugfs entry adding might sleep, so schedule process
  292. * context task for adding entry for STAs that do not yet
  293. * have one.
  294. * NOTE: due to auto-freeing semantics this may only be done
  295. * if the insertion is successful!
  296. */
  297. schedule_work(&local->sta_debugfs_add);
  298. #endif
  299. if (ieee80211_vif_is_mesh(&sdata->vif))
  300. mesh_accept_plinks_update(sdata);
  301. return 0;
  302. out_free:
  303. BUG_ON(!err);
  304. __sta_info_free(local, sta);
  305. return err;
  306. }
  307. static inline void __bss_tim_set(struct ieee80211_if_ap *bss, u16 aid)
  308. {
  309. /*
  310. * This format has been mandated by the IEEE specifications,
  311. * so this line may not be changed to use the __set_bit() format.
  312. */
  313. bss->tim[aid / 8] |= (1 << (aid % 8));
  314. }
  315. static inline void __bss_tim_clear(struct ieee80211_if_ap *bss, u16 aid)
  316. {
  317. /*
  318. * This format has been mandated by the IEEE specifications,
  319. * so this line may not be changed to use the __clear_bit() format.
  320. */
  321. bss->tim[aid / 8] &= ~(1 << (aid % 8));
  322. }
  323. static void __sta_info_set_tim_bit(struct ieee80211_if_ap *bss,
  324. struct sta_info *sta)
  325. {
  326. if (bss)
  327. __bss_tim_set(bss, sta->aid);
  328. if (sta->local->ops->set_tim) {
  329. sta->local->tim_in_locked_section = true;
  330. sta->local->ops->set_tim(local_to_hw(sta->local), sta->aid, 1);
  331. sta->local->tim_in_locked_section = false;
  332. }
  333. }
  334. void sta_info_set_tim_bit(struct sta_info *sta)
  335. {
  336. unsigned long flags;
  337. spin_lock_irqsave(&sta->local->sta_lock, flags);
  338. __sta_info_set_tim_bit(sta->sdata->bss, sta);
  339. spin_unlock_irqrestore(&sta->local->sta_lock, flags);
  340. }
  341. static void __sta_info_clear_tim_bit(struct ieee80211_if_ap *bss,
  342. struct sta_info *sta)
  343. {
  344. if (bss)
  345. __bss_tim_clear(bss, sta->aid);
  346. if (sta->local->ops->set_tim) {
  347. sta->local->tim_in_locked_section = true;
  348. sta->local->ops->set_tim(local_to_hw(sta->local), sta->aid, 0);
  349. sta->local->tim_in_locked_section = false;
  350. }
  351. }
  352. void sta_info_clear_tim_bit(struct sta_info *sta)
  353. {
  354. unsigned long flags;
  355. spin_lock_irqsave(&sta->local->sta_lock, flags);
  356. __sta_info_clear_tim_bit(sta->sdata->bss, sta);
  357. spin_unlock_irqrestore(&sta->local->sta_lock, flags);
  358. }
  359. void __sta_info_unlink(struct sta_info **sta)
  360. {
  361. struct ieee80211_local *local = (*sta)->local;
  362. struct ieee80211_sub_if_data *sdata = (*sta)->sdata;
  363. #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
  364. DECLARE_MAC_BUF(mbuf);
  365. #endif
  366. /*
  367. * pull caller's reference if we're already gone.
  368. */
  369. if (sta_info_hash_del(local, *sta)) {
  370. *sta = NULL;
  371. return;
  372. }
  373. if ((*sta)->key) {
  374. ieee80211_key_free((*sta)->key);
  375. WARN_ON((*sta)->key);
  376. }
  377. list_del(&(*sta)->list);
  378. if ((*sta)->flags & WLAN_STA_PS) {
  379. (*sta)->flags &= ~WLAN_STA_PS;
  380. if (sdata->bss)
  381. atomic_dec(&sdata->bss->num_sta_ps);
  382. __sta_info_clear_tim_bit(sdata->bss, *sta);
  383. }
  384. local->num_sta--;
  385. if (local->ops->sta_notify) {
  386. if (sdata->vif.type == IEEE80211_IF_TYPE_VLAN)
  387. sdata = sdata->u.vlan.ap;
  388. local->ops->sta_notify(local_to_hw(local), &sdata->vif,
  389. STA_NOTIFY_REMOVE, (*sta)->addr);
  390. }
  391. if (ieee80211_vif_is_mesh(&sdata->vif)) {
  392. mesh_accept_plinks_update(sdata);
  393. #ifdef CONFIG_MAC80211_MESH
  394. del_timer(&(*sta)->plink_timer);
  395. #endif
  396. }
  397. #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
  398. printk(KERN_DEBUG "%s: Removed STA %s\n",
  399. wiphy_name(local->hw.wiphy), print_mac(mbuf, (*sta)->addr));
  400. #endif /* CONFIG_MAC80211_VERBOSE_DEBUG */
  401. /*
  402. * Finally, pull caller's reference if the STA is pinned by the
  403. * task that is adding the debugfs entries. In that case, we
  404. * leave the STA "to be freed".
  405. *
  406. * The rules are not trivial, but not too complex either:
  407. * (1) pin_status is only modified under the sta_lock
  408. * (2) STAs may only be pinned under the RTNL so that
  409. * sta_info_flush() is guaranteed to actually destroy
  410. * all STAs that are active for a given interface, this
  411. * is required for correctness because otherwise we
  412. * could notify a driver that an interface is going
  413. * away and only after that (!) notify it about a STA
  414. * on that interface going away.
  415. * (3) sta_info_debugfs_add_work() will set the status
  416. * to PINNED when it found an item that needs a new
  417. * debugfs directory created. In that case, that item
  418. * must not be freed although all *RCU* users are done
  419. * with it. Hence, we tell the caller of _unlink()
  420. * that the item is already gone (as can happen when
  421. * two tasks try to unlink/destroy at the same time)
  422. * (4) We set the pin_status to DESTROY here when we
  423. * find such an item.
  424. * (5) sta_info_debugfs_add_work() will reset the pin_status
  425. * from PINNED to NORMAL when it is done with the item,
  426. * but will check for DESTROY before resetting it in
  427. * which case it will free the item.
  428. */
  429. if ((*sta)->pin_status == STA_INFO_PIN_STAT_PINNED) {
  430. (*sta)->pin_status = STA_INFO_PIN_STAT_DESTROY;
  431. *sta = NULL;
  432. return;
  433. }
  434. }
  435. void sta_info_unlink(struct sta_info **sta)
  436. {
  437. struct ieee80211_local *local = (*sta)->local;
  438. unsigned long flags;
  439. spin_lock_irqsave(&local->sta_lock, flags);
  440. __sta_info_unlink(sta);
  441. spin_unlock_irqrestore(&local->sta_lock, flags);
  442. }
  443. static inline int sta_info_buffer_expired(struct ieee80211_local *local,
  444. struct sta_info *sta,
  445. struct sk_buff *skb)
  446. {
  447. struct ieee80211_tx_packet_data *pkt_data;
  448. int timeout;
  449. if (!skb)
  450. return 0;
  451. pkt_data = (struct ieee80211_tx_packet_data *) skb->cb;
  452. /* Timeout: (2 * listen_interval * beacon_int * 1024 / 1000000) sec */
  453. timeout = (sta->listen_interval * local->hw.conf.beacon_int * 32 /
  454. 15625) * HZ;
  455. if (timeout < STA_TX_BUFFER_EXPIRE)
  456. timeout = STA_TX_BUFFER_EXPIRE;
  457. return time_after(jiffies, pkt_data->jiffies + timeout);
  458. }
  459. static void sta_info_cleanup_expire_buffered(struct ieee80211_local *local,
  460. struct sta_info *sta)
  461. {
  462. unsigned long flags;
  463. struct sk_buff *skb;
  464. struct ieee80211_sub_if_data *sdata;
  465. DECLARE_MAC_BUF(mac);
  466. if (skb_queue_empty(&sta->ps_tx_buf))
  467. return;
  468. for (;;) {
  469. spin_lock_irqsave(&sta->ps_tx_buf.lock, flags);
  470. skb = skb_peek(&sta->ps_tx_buf);
  471. if (sta_info_buffer_expired(local, sta, skb))
  472. skb = __skb_dequeue(&sta->ps_tx_buf);
  473. else
  474. skb = NULL;
  475. spin_unlock_irqrestore(&sta->ps_tx_buf.lock, flags);
  476. if (!skb)
  477. break;
  478. sdata = sta->sdata;
  479. local->total_ps_buffered--;
  480. printk(KERN_DEBUG "Buffered frame expired (STA "
  481. "%s)\n", print_mac(mac, sta->addr));
  482. dev_kfree_skb(skb);
  483. if (skb_queue_empty(&sta->ps_tx_buf))
  484. sta_info_clear_tim_bit(sta);
  485. }
  486. }
  487. static void sta_info_cleanup(unsigned long data)
  488. {
  489. struct ieee80211_local *local = (struct ieee80211_local *) data;
  490. struct sta_info *sta;
  491. rcu_read_lock();
  492. list_for_each_entry_rcu(sta, &local->sta_list, list)
  493. sta_info_cleanup_expire_buffered(local, sta);
  494. rcu_read_unlock();
  495. local->sta_cleanup.expires =
  496. round_jiffies(jiffies + STA_INFO_CLEANUP_INTERVAL);
  497. add_timer(&local->sta_cleanup);
  498. }
  499. #ifdef CONFIG_MAC80211_DEBUGFS
  500. /*
  501. * See comment in __sta_info_unlink,
  502. * caller must hold local->sta_lock.
  503. */
  504. static void __sta_info_pin(struct sta_info *sta)
  505. {
  506. WARN_ON(sta->pin_status != STA_INFO_PIN_STAT_NORMAL);
  507. sta->pin_status = STA_INFO_PIN_STAT_PINNED;
  508. }
  509. /*
  510. * See comment in __sta_info_unlink, returns sta if it
  511. * needs to be destroyed.
  512. */
  513. static struct sta_info *__sta_info_unpin(struct sta_info *sta)
  514. {
  515. struct sta_info *ret = NULL;
  516. unsigned long flags;
  517. spin_lock_irqsave(&sta->local->sta_lock, flags);
  518. WARN_ON(sta->pin_status != STA_INFO_PIN_STAT_DESTROY &&
  519. sta->pin_status != STA_INFO_PIN_STAT_PINNED);
  520. if (sta->pin_status == STA_INFO_PIN_STAT_DESTROY)
  521. ret = sta;
  522. sta->pin_status = STA_INFO_PIN_STAT_NORMAL;
  523. spin_unlock_irqrestore(&sta->local->sta_lock, flags);
  524. return ret;
  525. }
  526. static void sta_info_debugfs_add_work(struct work_struct *work)
  527. {
  528. struct ieee80211_local *local =
  529. container_of(work, struct ieee80211_local, sta_debugfs_add);
  530. struct sta_info *sta, *tmp;
  531. unsigned long flags;
  532. /* We need to keep the RTNL across the whole pinned status. */
  533. rtnl_lock();
  534. while (1) {
  535. sta = NULL;
  536. spin_lock_irqsave(&local->sta_lock, flags);
  537. list_for_each_entry(tmp, &local->sta_list, list) {
  538. if (!tmp->debugfs.dir) {
  539. sta = tmp;
  540. __sta_info_pin(sta);
  541. break;
  542. }
  543. }
  544. spin_unlock_irqrestore(&local->sta_lock, flags);
  545. if (!sta)
  546. break;
  547. ieee80211_sta_debugfs_add(sta);
  548. rate_control_add_sta_debugfs(sta);
  549. sta = __sta_info_unpin(sta);
  550. sta_info_destroy(sta);
  551. }
  552. rtnl_unlock();
  553. }
  554. #endif
  555. static void __ieee80211_run_pending_flush(struct ieee80211_local *local)
  556. {
  557. struct sta_info *sta;
  558. unsigned long flags;
  559. ASSERT_RTNL();
  560. spin_lock_irqsave(&local->sta_lock, flags);
  561. while (!list_empty(&local->sta_flush_list)) {
  562. sta = list_first_entry(&local->sta_flush_list,
  563. struct sta_info, list);
  564. list_del(&sta->list);
  565. spin_unlock_irqrestore(&local->sta_lock, flags);
  566. sta_info_destroy(sta);
  567. spin_lock_irqsave(&local->sta_lock, flags);
  568. }
  569. spin_unlock_irqrestore(&local->sta_lock, flags);
  570. }
  571. static void ieee80211_sta_flush_work(struct work_struct *work)
  572. {
  573. struct ieee80211_local *local =
  574. container_of(work, struct ieee80211_local, sta_flush_work);
  575. rtnl_lock();
  576. __ieee80211_run_pending_flush(local);
  577. rtnl_unlock();
  578. }
  579. void sta_info_init(struct ieee80211_local *local)
  580. {
  581. spin_lock_init(&local->sta_lock);
  582. INIT_LIST_HEAD(&local->sta_list);
  583. INIT_LIST_HEAD(&local->sta_flush_list);
  584. INIT_WORK(&local->sta_flush_work, ieee80211_sta_flush_work);
  585. setup_timer(&local->sta_cleanup, sta_info_cleanup,
  586. (unsigned long)local);
  587. local->sta_cleanup.expires =
  588. round_jiffies(jiffies + STA_INFO_CLEANUP_INTERVAL);
  589. #ifdef CONFIG_MAC80211_DEBUGFS
  590. INIT_WORK(&local->sta_debugfs_add, sta_info_debugfs_add_work);
  591. #endif
  592. }
  593. int sta_info_start(struct ieee80211_local *local)
  594. {
  595. add_timer(&local->sta_cleanup);
  596. return 0;
  597. }
  598. void sta_info_stop(struct ieee80211_local *local)
  599. {
  600. del_timer(&local->sta_cleanup);
  601. cancel_work_sync(&local->sta_flush_work);
  602. #ifdef CONFIG_MAC80211_DEBUGFS
  603. /*
  604. * Make sure the debugfs adding work isn't pending after this
  605. * because we're about to be destroyed. It doesn't matter
  606. * whether it ran or not since we're going to flush all STAs
  607. * anyway.
  608. */
  609. cancel_work_sync(&local->sta_debugfs_add);
  610. #endif
  611. rtnl_lock();
  612. sta_info_flush(local, NULL);
  613. __ieee80211_run_pending_flush(local);
  614. rtnl_unlock();
  615. }
  616. /**
  617. * sta_info_flush - flush matching STA entries from the STA table
  618. *
  619. * Returns the number of removed STA entries.
  620. *
  621. * @local: local interface data
  622. * @sdata: matching rule for the net device (sta->dev) or %NULL to match all STAs
  623. */
  624. int sta_info_flush(struct ieee80211_local *local,
  625. struct ieee80211_sub_if_data *sdata)
  626. {
  627. struct sta_info *sta, *tmp;
  628. LIST_HEAD(tmp_list);
  629. int ret = 0;
  630. unsigned long flags;
  631. might_sleep();
  632. ASSERT_RTNL();
  633. spin_lock_irqsave(&local->sta_lock, flags);
  634. list_for_each_entry_safe(sta, tmp, &local->sta_list, list) {
  635. if (!sdata || sdata == sta->sdata) {
  636. __sta_info_unlink(&sta);
  637. if (sta) {
  638. list_add_tail(&sta->list, &tmp_list);
  639. ret++;
  640. }
  641. }
  642. }
  643. spin_unlock_irqrestore(&local->sta_lock, flags);
  644. list_for_each_entry_safe(sta, tmp, &tmp_list, list)
  645. sta_info_destroy(sta);
  646. return ret;
  647. }
  648. /**
  649. * sta_info_flush_delayed - flush matching STA entries from the STA table
  650. *
  651. * This function unlinks all stations for a given interface and queues
  652. * them for freeing. Note that the workqueue function scheduled here has
  653. * to run before any new keys can be added to the system to avoid set_key()
  654. * callback ordering issues.
  655. *
  656. * @sdata: the interface
  657. */
  658. void sta_info_flush_delayed(struct ieee80211_sub_if_data *sdata)
  659. {
  660. struct ieee80211_local *local = sdata->local;
  661. struct sta_info *sta, *tmp;
  662. unsigned long flags;
  663. bool work = false;
  664. spin_lock_irqsave(&local->sta_lock, flags);
  665. list_for_each_entry_safe(sta, tmp, &local->sta_list, list) {
  666. if (sdata == sta->sdata) {
  667. __sta_info_unlink(&sta);
  668. if (sta) {
  669. list_add_tail(&sta->list,
  670. &local->sta_flush_list);
  671. work = true;
  672. }
  673. }
  674. }
  675. if (work)
  676. schedule_work(&local->sta_flush_work);
  677. spin_unlock_irqrestore(&local->sta_lock, flags);
  678. }