blob: a9b46d8ea22ff696623cec4cad55949fc5afd2a1 [file] [log] [blame]
Jiri Bencf0706e82007-05-05 11:45:53 -07001/*
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>
Felix Fietkau888d04d2012-03-01 15:22:09 +010012#include <linux/etherdevice.h>
Jiri Bencf0706e82007-05-05 11:45:53 -070013#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>
Johannes Berg0d174402007-12-17 15:07:43 +010018#include <linux/timer.h>
Johannes Bergd0709a62008-02-25 16:27:46 +010019#include <linux/rtnetlink.h>
Jiri Bencf0706e82007-05-05 11:45:53 -070020
21#include <net/mac80211.h>
22#include "ieee80211_i.h"
Johannes Berg24487982009-04-23 18:52:52 +020023#include "driver-ops.h"
Johannes Berg2c8dccc2008-04-08 15:14:40 -040024#include "rate.h"
Jiri Bencf0706e82007-05-05 11:45:53 -070025#include "sta_info.h"
Jiri Bence9f207f2007-05-05 11:46:38 -070026#include "debugfs_sta.h"
Luis Carlos Coboee385852008-02-23 15:17:11 +010027#include "mesh.h"
Johannes Bergce662b442011-09-29 16:04:34 +020028#include "wme.h"
Jiri Bencf0706e82007-05-05 11:45:53 -070029
Johannes Bergd0709a62008-02-25 16:27:46 +010030/**
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 *
Johannes Berg34e89502010-02-03 13:59:58 +010037 * 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.
Johannes Berg93e5deb2008-04-01 15:21:00 +020045 *
46 * When the insertion fails (sta_info_insert()) returns non-zero), the
47 * structure will have been freed by sta_info_insert()!
Johannes Bergd0709a62008-02-25 16:27:46 +010048 *
Johannes Berg34e89502010-02-03 13:59:58 +010049 * Station entries are added by mac80211 when you establish a link with a
Luis R. Rodriguez7e189a12009-06-02 18:38:14 -040050 * 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
Lucas De Marchi25985ed2011-03-30 22:57:33 -030052 * receive an association response from the AP. For IBSS this occurs when
Johannes Berg34e89502010-02-03 13:59:58 +010053 * get to know about a peer on the same IBSS. For WDS we add the sta for
Lucas De Marchi25985ed2011-03-30 22:57:33 -030054 * the peer immediately upon device open. When using AP mode we add stations
Johannes Berg34e89502010-02-03 13:59:58 +010055 * for each respective station upon request from userspace through nl80211.
Luis R. Rodriguez7e189a12009-06-02 18:38:14 -040056 *
Johannes Berg34e89502010-02-03 13:59:58 +010057 * In order to remove a STA info structure, various sta_info_destroy_*()
58 * calls are available.
Johannes Bergd0709a62008-02-25 16:27:46 +010059 *
Johannes Berg34e89502010-02-03 13:59:58 +010060 * 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.
Johannes Bergd0709a62008-02-25 16:27:46 +010064 */
Jiri Bencf0706e82007-05-05 11:45:53 -070065
Johannes Berg4d339602011-12-15 11:24:20 +010066/* Caller must hold local->sta_mtx */
Michael Wube8755e2007-07-27 15:43:23 +020067static int sta_info_hash_del(struct ieee80211_local *local,
68 struct sta_info *sta)
Jiri Bencf0706e82007-05-05 11:45:53 -070069{
70 struct sta_info *s;
71
Johannes Berg40b275b2011-05-13 14:15:49 +020072 s = rcu_dereference_protected(local->sta_hash[STA_HASH(sta->sta.addr)],
Johannes Berg4d339602011-12-15 11:24:20 +010073 lockdep_is_held(&local->sta_mtx));
Jiri Bencf0706e82007-05-05 11:45:53 -070074 if (!s)
Michael Wube8755e2007-07-27 15:43:23 +020075 return -ENOENT;
76 if (s == sta) {
Eric Dumazetcf778b02012-01-12 04:41:32 +000077 rcu_assign_pointer(local->sta_hash[STA_HASH(sta->sta.addr)],
Johannes Bergd0709a62008-02-25 16:27:46 +010078 s->hnext);
Michael Wube8755e2007-07-27 15:43:23 +020079 return 0;
Jiri Bencf0706e82007-05-05 11:45:53 -070080 }
81
Johannes Berg40b275b2011-05-13 14:15:49 +020082 while (rcu_access_pointer(s->hnext) &&
83 rcu_access_pointer(s->hnext) != sta)
84 s = rcu_dereference_protected(s->hnext,
Johannes Berg4d339602011-12-15 11:24:20 +010085 lockdep_is_held(&local->sta_mtx));
Johannes Berg40b275b2011-05-13 14:15:49 +020086 if (rcu_access_pointer(s->hnext)) {
Eric Dumazetcf778b02012-01-12 04:41:32 +000087 rcu_assign_pointer(s->hnext, sta->hnext);
Michael Wube8755e2007-07-27 15:43:23 +020088 return 0;
89 }
Jiri Bencf0706e82007-05-05 11:45:53 -070090
Michael Wube8755e2007-07-27 15:43:23 +020091 return -ENOENT;
Jiri Bencf0706e82007-05-05 11:45:53 -070092}
93
Johannes Berg5108ca82014-02-17 20:49:03 +010094static void __cleanup_single_sta(struct sta_info *sta)
Eliad Pellerb22cfcf2012-09-09 14:43:51 +030095{
Eliad Pellerb22cfcf2012-09-09 14:43:51 +030096 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;
Marco Porschd012a602012-10-10 12:39:50 -0700100 struct ps_data *ps;
Eliad Pellerb22cfcf2012-09-09 14:43:51 +0300101
Johannes Berge3685e02014-02-20 11:19:58 +0100102 if (test_sta_flag(sta, WLAN_STA_PS_STA) ||
103 test_sta_flag(sta, WLAN_STA_PS_DRIVER)) {
Marco Porschd012a602012-10-10 12:39:50 -0700104 if (sta->sdata->vif.type == NL80211_IFTYPE_AP ||
105 sta->sdata->vif.type == NL80211_IFTYPE_AP_VLAN)
106 ps = &sdata->bss->ps;
Marco Porsch3f52b7e2013-01-30 18:14:08 +0100107 else if (ieee80211_vif_is_mesh(&sdata->vif))
108 ps = &sdata->u.mesh.ps;
Marco Porschd012a602012-10-10 12:39:50 -0700109 else
110 return;
Eliad Pellerb22cfcf2012-09-09 14:43:51 +0300111
112 clear_sta_flag(sta, WLAN_STA_PS_STA);
Johannes Berge3685e02014-02-20 11:19:58 +0100113 clear_sta_flag(sta, WLAN_STA_PS_DRIVER);
Eliad Pellerb22cfcf2012-09-09 14:43:51 +0300114
Marco Porschd012a602012-10-10 12:39:50 -0700115 atomic_dec(&ps->num_sta_ps);
Eliad Pellerb22cfcf2012-09-09 14:43:51 +0300116 sta_info_recalc_tim(sta);
117 }
118
119 for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) {
120 local->total_ps_buffered -= skb_queue_len(&sta->ps_tx_buf[ac]);
Felix Fietkau1f98ab72012-11-10 03:44:14 +0100121 ieee80211_purge_tx_queue(&local->hw, &sta->ps_tx_buf[ac]);
122 ieee80211_purge_tx_queue(&local->hw, &sta->tx_filtered[ac]);
Eliad Pellerb22cfcf2012-09-09 14:43:51 +0300123 }
124
Thomas Pedersen45b50282013-02-06 10:17:21 -0800125 if (ieee80211_vif_is_mesh(&sdata->vif))
126 mesh_sta_cleanup(sta);
Eliad Pellerb22cfcf2012-09-09 14:43:51 +0300127
128 cancel_work_sync(&sta->drv_unblock_wk);
129
130 /*
131 * Destroy aggregation state here. It would be nice to wait for the
132 * driver to finish aggregation stop and then clean up, but for now
133 * drivers have to handle aggregation stop being requested, followed
134 * directly by station destruction.
135 */
Johannes Berg5a306f52012-11-14 23:22:21 +0100136 for (i = 0; i < IEEE80211_NUM_TIDS; i++) {
Johannes Berg661eb382013-06-12 22:47:56 +0200137 kfree(sta->ampdu_mlme.tid_start_tx[i]);
Eliad Pellerb22cfcf2012-09-09 14:43:51 +0300138 tid_tx = rcu_dereference_raw(sta->ampdu_mlme.tid_tx[i]);
139 if (!tid_tx)
140 continue;
Felix Fietkau1f98ab72012-11-10 03:44:14 +0100141 ieee80211_purge_tx_queue(&local->hw, &tid_tx->pending);
Eliad Pellerb22cfcf2012-09-09 14:43:51 +0300142 kfree(tid_tx);
143 }
Johannes Berg5108ca82014-02-17 20:49:03 +0100144}
Eliad Pellerb22cfcf2012-09-09 14:43:51 +0300145
Johannes Berg5108ca82014-02-17 20:49:03 +0100146static void cleanup_single_sta(struct sta_info *sta)
147{
148 struct ieee80211_sub_if_data *sdata = sta->sdata;
149 struct ieee80211_local *local = sdata->local;
150
151 __cleanup_single_sta(sta);
Eliad Pellerb22cfcf2012-09-09 14:43:51 +0300152 sta_info_free(local, sta);
153}
154
Johannes Bergd0709a62008-02-25 16:27:46 +0100155/* protected by RCU */
Johannes Bergabe60632009-11-25 17:46:18 +0100156struct sta_info *sta_info_get(struct ieee80211_sub_if_data *sdata,
157 const u8 *addr)
Johannes Berg43ba7e92008-02-21 14:09:30 +0100158{
Johannes Bergabe60632009-11-25 17:46:18 +0100159 struct ieee80211_local *local = sdata->local;
Johannes Berg43ba7e92008-02-21 14:09:30 +0100160 struct sta_info *sta;
161
Johannes Berg03791852010-04-06 11:18:42 +0200162 sta = rcu_dereference_check(local->sta_hash[STA_HASH(addr)],
Johannes Berg03791852010-04-06 11:18:42 +0200163 lockdep_is_held(&local->sta_mtx));
Johannes Berg43ba7e92008-02-21 14:09:30 +0100164 while (sta) {
Johannes Bergabe60632009-11-25 17:46:18 +0100165 if (sta->sdata == sdata &&
Joe Perchesb203ca32012-05-08 18:56:52 +0000166 ether_addr_equal(sta->sta.addr, addr))
Johannes Berg43ba7e92008-02-21 14:09:30 +0100167 break;
Johannes Berg03791852010-04-06 11:18:42 +0200168 sta = rcu_dereference_check(sta->hnext,
Johannes Berg03791852010-04-06 11:18:42 +0200169 lockdep_is_held(&local->sta_mtx));
Johannes Berg43ba7e92008-02-21 14:09:30 +0100170 }
171 return sta;
172}
173
Felix Fietkau0e5ded52010-01-08 18:10:58 +0100174/*
175 * Get sta info either from the specified interface
176 * or from one of its vlans
177 */
178struct sta_info *sta_info_get_bss(struct ieee80211_sub_if_data *sdata,
179 const u8 *addr)
180{
181 struct ieee80211_local *local = sdata->local;
182 struct sta_info *sta;
183
Johannes Berg03791852010-04-06 11:18:42 +0200184 sta = rcu_dereference_check(local->sta_hash[STA_HASH(addr)],
Johannes Berg03791852010-04-06 11:18:42 +0200185 lockdep_is_held(&local->sta_mtx));
Felix Fietkau0e5ded52010-01-08 18:10:58 +0100186 while (sta) {
187 if ((sta->sdata == sdata ||
Johannes Berga2c1e3d2010-09-14 21:34:14 +0200188 (sta->sdata->bss && sta->sdata->bss == sdata->bss)) &&
Joe Perchesb203ca32012-05-08 18:56:52 +0000189 ether_addr_equal(sta->sta.addr, addr))
Felix Fietkau0e5ded52010-01-08 18:10:58 +0100190 break;
Johannes Berg03791852010-04-06 11:18:42 +0200191 sta = rcu_dereference_check(sta->hnext,
Johannes Berg03791852010-04-06 11:18:42 +0200192 lockdep_is_held(&local->sta_mtx));
Felix Fietkau0e5ded52010-01-08 18:10:58 +0100193 }
194 return sta;
195}
196
Johannes Berg3b53fde82009-11-16 12:00:37 +0100197struct sta_info *sta_info_get_by_idx(struct ieee80211_sub_if_data *sdata,
198 int idx)
Luis Carlos Coboee385852008-02-23 15:17:11 +0100199{
Johannes Berg3b53fde82009-11-16 12:00:37 +0100200 struct ieee80211_local *local = sdata->local;
Luis Carlos Coboee385852008-02-23 15:17:11 +0100201 struct sta_info *sta;
202 int i = 0;
203
Johannes Bergd0709a62008-02-25 16:27:46 +0100204 list_for_each_entry_rcu(sta, &local->sta_list, list) {
Johannes Berg3b53fde82009-11-16 12:00:37 +0100205 if (sdata != sta->sdata)
Luis Carlos Cobo2a8ca292008-02-29 17:51:25 -0800206 continue;
Luis Carlos Coboee385852008-02-23 15:17:11 +0100207 if (i < idx) {
208 ++i;
209 continue;
Luis Carlos Coboee385852008-02-23 15:17:11 +0100210 }
Luis Carlos Cobo2a8ca292008-02-29 17:51:25 -0800211 return sta;
Luis Carlos Coboee385852008-02-23 15:17:11 +0100212 }
Luis Carlos Coboee385852008-02-23 15:17:11 +0100213
214 return NULL;
215}
Jiri Bencf0706e82007-05-05 11:45:53 -0700216
Johannes Berg93e5deb2008-04-01 15:21:00 +0200217/**
Johannes Bergd9a7ddb2011-12-14 12:35:30 +0100218 * sta_info_free - free STA
Johannes Berg93e5deb2008-04-01 15:21:00 +0200219 *
Randy Dunlap6ef307b2008-07-03 13:52:18 -0700220 * @local: pointer to the global information
Johannes Berg93e5deb2008-04-01 15:21:00 +0200221 * @sta: STA info to free
222 *
223 * This function must undo everything done by sta_info_alloc()
Johannes Bergd9a7ddb2011-12-14 12:35:30 +0100224 * that may happen before sta_info_insert(). It may only be
225 * called when sta_info_insert() has not been attempted (and
226 * if that fails, the station is freed anyway.)
Johannes Berg93e5deb2008-04-01 15:21:00 +0200227 */
Johannes Bergd9a7ddb2011-12-14 12:35:30 +0100228void sta_info_free(struct ieee80211_local *local, struct sta_info *sta)
Johannes Berg93e5deb2008-04-01 15:21:00 +0200229{
Matti Gottliebad38bfc2013-11-18 19:06:45 +0200230 int i;
231
Johannes Berg889cbb92012-01-17 10:33:29 +0100232 if (sta->rate_ctrl)
Johannes Bergaf65cd962009-11-17 18:18:36 +0100233 rate_control_free_sta(sta);
Johannes Berg93e5deb2008-04-01 15:21:00 +0200234
Matti Gottliebad38bfc2013-11-18 19:06:45 +0200235 if (sta->tx_lat) {
236 for (i = 0; i < IEEE80211_NUM_TIDS; i++)
237 kfree(sta->tx_lat[i].bins);
238 kfree(sta->tx_lat);
239 }
240
Johannes Bergbdcbd8e2012-06-22 11:29:50 +0200241 sta_dbg(sta->sdata, "Destroyed STA %pM\n", sta->sta.addr);
Johannes Berg93e5deb2008-04-01 15:21:00 +0200242
Felix Fietkau53d04522014-05-27 22:33:57 +0200243 kfree(rcu_dereference_raw(sta->sta.rates));
Johannes Berg93e5deb2008-04-01 15:21:00 +0200244 kfree(sta);
245}
246
Johannes Berg4d339602011-12-15 11:24:20 +0100247/* Caller must hold local->sta_mtx */
Johannes Bergd0709a62008-02-25 16:27:46 +0100248static void sta_info_hash_add(struct ieee80211_local *local,
249 struct sta_info *sta)
Jiri Bencf0706e82007-05-05 11:45:53 -0700250{
Johannes Berg4d339602011-12-15 11:24:20 +0100251 lockdep_assert_held(&local->sta_mtx);
Johannes Berg17741cd2008-09-11 00:02:02 +0200252 sta->hnext = local->sta_hash[STA_HASH(sta->sta.addr)];
Eric Dumazetcf778b02012-01-12 04:41:32 +0000253 rcu_assign_pointer(local->sta_hash[STA_HASH(sta->sta.addr)], sta);
Jiri Bencf0706e82007-05-05 11:45:53 -0700254}
Jiri Bencf0706e82007-05-05 11:45:53 -0700255
Johannes Bergaf818582009-11-06 11:35:50 +0100256static void sta_unblock(struct work_struct *wk)
257{
258 struct sta_info *sta;
259
260 sta = container_of(wk, struct sta_info, drv_unblock_wk);
261
262 if (sta->dead)
263 return;
264
Helmut Schaa54420472012-01-17 09:22:49 +0100265 if (!test_sta_flag(sta, WLAN_STA_PS_STA)) {
266 local_bh_disable();
Johannes Bergaf818582009-11-06 11:35:50 +0100267 ieee80211_sta_ps_deliver_wakeup(sta);
Helmut Schaa54420472012-01-17 09:22:49 +0100268 local_bh_enable();
269 } else if (test_and_clear_sta_flag(sta, WLAN_STA_PSPOLL)) {
Johannes Bergc2c98fd2011-09-29 16:04:36 +0200270 clear_sta_flag(sta, WLAN_STA_PS_DRIVER);
Johannes Bergce662b442011-09-29 16:04:34 +0200271
272 local_bh_disable();
Johannes Bergaf818582009-11-06 11:35:50 +0100273 ieee80211_sta_ps_deliver_poll_response(sta);
Johannes Bergce662b442011-09-29 16:04:34 +0200274 local_bh_enable();
Johannes Bergc2c98fd2011-09-29 16:04:36 +0200275 } else if (test_and_clear_sta_flag(sta, WLAN_STA_UAPSD)) {
276 clear_sta_flag(sta, WLAN_STA_PS_DRIVER);
Johannes Bergce662b442011-09-29 16:04:34 +0200277
278 local_bh_disable();
Johannes Berg47086fc2011-09-29 16:04:33 +0200279 ieee80211_sta_ps_deliver_uapsd(sta);
Johannes Bergce662b442011-09-29 16:04:34 +0200280 local_bh_enable();
Johannes Berg50a94322010-11-16 11:50:28 -0800281 } else
Johannes Bergc2c98fd2011-09-29 16:04:36 +0200282 clear_sta_flag(sta, WLAN_STA_PS_DRIVER);
Johannes Bergaf818582009-11-06 11:35:50 +0100283}
284
Johannes Bergaf65cd962009-11-17 18:18:36 +0100285static int sta_prepare_rate_control(struct ieee80211_local *local,
286 struct sta_info *sta, gfp_t gfp)
287{
288 if (local->hw.flags & IEEE80211_HW_HAS_RATE_CONTROL)
289 return 0;
290
Johannes Berg889cbb92012-01-17 10:33:29 +0100291 sta->rate_ctrl = local->rate_ctrl;
Johannes Bergaf65cd962009-11-17 18:18:36 +0100292 sta->rate_ctrl_priv = rate_control_alloc_sta(sta->rate_ctrl,
293 &sta->sta, gfp);
Johannes Berg889cbb92012-01-17 10:33:29 +0100294 if (!sta->rate_ctrl_priv)
Johannes Bergaf65cd962009-11-17 18:18:36 +0100295 return -ENOMEM;
Johannes Bergaf65cd962009-11-17 18:18:36 +0100296
297 return 0;
298}
299
Johannes Berg73651ee2008-02-25 16:27:47 +0100300struct sta_info *sta_info_alloc(struct ieee80211_sub_if_data *sdata,
Johannes Berg56544162011-12-14 13:28:46 +0100301 const u8 *addr, gfp_t gfp)
Jiri Bencf0706e82007-05-05 11:45:53 -0700302{
Johannes Bergd0709a62008-02-25 16:27:46 +0100303 struct ieee80211_local *local = sdata->local;
Jiri Bencf0706e82007-05-05 11:45:53 -0700304 struct sta_info *sta;
Mohammed Shafi Shajakhanebe27c92011-04-08 21:24:24 +0530305 struct timespec uptime;
Matti Gottliebad38bfc2013-11-18 19:06:45 +0200306 struct ieee80211_tx_latency_bin_ranges *tx_latency;
Ron Rindjunsky16c5f152007-12-25 17:00:34 +0200307 int i;
Jiri Bencf0706e82007-05-05 11:45:53 -0700308
Johannes Berg17741cd2008-09-11 00:02:02 +0200309 sta = kzalloc(sizeof(*sta) + local->hw.sta_data_size, gfp);
Jiri Bencf0706e82007-05-05 11:45:53 -0700310 if (!sta)
Johannes Berg73651ee2008-02-25 16:27:47 +0100311 return NULL;
Jiri Bencf0706e82007-05-05 11:45:53 -0700312
Johannes Bergef04a292014-01-06 15:56:59 +0100313 rcu_read_lock();
314 tx_latency = rcu_dereference(local->tx_latency);
315 /* init stations Tx latency statistics && TID bins */
316 if (tx_latency) {
317 sta->tx_lat = kzalloc(IEEE80211_NUM_TIDS *
318 sizeof(struct ieee80211_tx_latency_stat),
319 GFP_ATOMIC);
320 if (!sta->tx_lat) {
321 rcu_read_unlock();
322 goto free;
323 }
324
325 if (tx_latency->n_ranges) {
326 for (i = 0; i < IEEE80211_NUM_TIDS; i++) {
327 /* size of bins is size of the ranges +1 */
328 sta->tx_lat[i].bin_count =
329 tx_latency->n_ranges + 1;
330 sta->tx_lat[i].bins =
331 kcalloc(sta->tx_lat[i].bin_count,
332 sizeof(u32), GFP_ATOMIC);
333 if (!sta->tx_lat[i].bins) {
334 rcu_read_unlock();
335 goto free;
336 }
337 }
338 }
339 }
340 rcu_read_unlock();
341
Johannes Berg07346f812008-05-03 01:02:02 +0200342 spin_lock_init(&sta->lock);
Emmanuel Grumbach1d147bf2014-02-20 09:22:11 +0200343 spin_lock_init(&sta->ps_lock);
Johannes Bergaf818582009-11-06 11:35:50 +0100344 INIT_WORK(&sta->drv_unblock_wk, sta_unblock);
Johannes Berg67c282c2010-06-10 10:21:43 +0200345 INIT_WORK(&sta->ampdu_mlme.work, ieee80211_ba_session_work);
Johannes Berga93e3642010-06-10 10:21:46 +0200346 mutex_init(&sta->ampdu_mlme.mtx);
Thomas Pedersen87f59c72013-03-01 22:02:52 -0800347#ifdef CONFIG_MAC80211_MESH
348 if (ieee80211_vif_is_mesh(&sdata->vif) &&
349 !sdata->u.mesh.user_mpm)
350 init_timer(&sta->plink_timer);
Thomas Pedersen6c7c4cb2013-06-20 23:50:59 -0700351 sta->nonpeer_pm = NL80211_MESH_POWER_ACTIVE;
Thomas Pedersen87f59c72013-03-01 22:02:52 -0800352#endif
Johannes Berg07346f812008-05-03 01:02:02 +0200353
Johannes Berg17741cd2008-09-11 00:02:02 +0200354 memcpy(sta->sta.addr, addr, ETH_ALEN);
Johannes Bergd0709a62008-02-25 16:27:46 +0100355 sta->local = local;
356 sta->sdata = sdata;
Felix Fietkau8bc8aec2011-03-21 20:01:00 +0100357 sta->last_rx = jiffies;
Jiri Bencf0706e82007-05-05 11:45:53 -0700358
Johannes Berg71ec3752012-01-20 13:55:20 +0100359 sta->sta_state = IEEE80211_STA_NONE;
360
Mohammed Shafi Shajakhanebe27c92011-04-08 21:24:24 +0530361 do_posix_clock_monotonic_gettime(&uptime);
362 sta->last_connected = uptime.tv_sec;
Bruno Randolf541a45a2010-12-02 19:12:43 +0900363 ewma_init(&sta->avg_signal, 1024, 8);
Felix Fietkauef0621e2013-04-22 16:29:31 +0200364 for (i = 0; i < ARRAY_SIZE(sta->chain_signal_avg); i++)
365 ewma_init(&sta->chain_signal_avg[i], 1024, 8);
Bruno Randolf541a45a2010-12-02 19:12:43 +0900366
Johannes Bergef04a292014-01-06 15:56:59 +0100367 if (sta_prepare_rate_control(local, sta, gfp))
368 goto free;
Jiri Bencf0706e82007-05-05 11:45:53 -0700369
Johannes Berg5a306f52012-11-14 23:22:21 +0100370 for (i = 0; i < IEEE80211_NUM_TIDS; i++) {
Johannes Berga622ab72010-06-10 10:21:39 +0200371 /*
372 * timer_to_tid must be initialized with identity mapping
373 * to enable session_timer's data differentiation. See
374 * sta_rx_agg_session_timer_expired for usage.
375 */
Ron Rindjunsky16c5f152007-12-25 17:00:34 +0200376 sta->timer_to_tid[i] = i;
Ron Rindjunsky16c5f152007-12-25 17:00:34 +0200377 }
Johannes Berg948d8872011-09-29 16:04:29 +0200378 for (i = 0; i < IEEE80211_NUM_ACS; i++) {
379 skb_queue_head_init(&sta->ps_tx_buf[i]);
380 skb_queue_head_init(&sta->tx_filtered[i]);
381 }
Johannes Berg73651ee2008-02-25 16:27:47 +0100382
Johannes Berg5a306f52012-11-14 23:22:21 +0100383 for (i = 0; i < IEEE80211_NUM_TIDS; i++)
Alexey Dobriyan4be929b2010-05-24 14:33:03 -0700384 sta->last_seq_ctrl[i] = cpu_to_le16(USHRT_MAX);
Senthil Balasubramaniancccaec92009-05-14 18:42:08 +0530385
Johannes Bergaf0ed692013-02-12 14:21:00 +0100386 sta->sta.smps_mode = IEEE80211_SMPS_OFF;
Emmanuel Grumbach687da132013-10-01 16:45:43 +0300387 if (sdata->vif.type == NL80211_IFTYPE_AP ||
388 sdata->vif.type == NL80211_IFTYPE_AP_VLAN) {
389 struct ieee80211_supported_band *sband =
390 local->hw.wiphy->bands[ieee80211_get_sdata_band(sdata)];
391 u8 smps = (sband->ht_cap.cap & IEEE80211_HT_CAP_SM_PS) >>
392 IEEE80211_HT_CAP_SM_PS_SHIFT;
393 /*
394 * Assume that hostapd advertises our caps in the beacon and
395 * this is the known_smps_mode for a station that just assciated
396 */
397 switch (smps) {
398 case WLAN_HT_SMPS_CONTROL_DISABLED:
399 sta->known_smps_mode = IEEE80211_SMPS_OFF;
400 break;
401 case WLAN_HT_SMPS_CONTROL_STATIC:
402 sta->known_smps_mode = IEEE80211_SMPS_STATIC;
403 break;
404 case WLAN_HT_SMPS_CONTROL_DYNAMIC:
405 sta->known_smps_mode = IEEE80211_SMPS_DYNAMIC;
406 break;
407 default:
408 WARN_ON(1);
409 }
410 }
Johannes Bergaf0ed692013-02-12 14:21:00 +0100411
Johannes Bergbdcbd8e2012-06-22 11:29:50 +0200412 sta_dbg(sdata, "Allocated STA %pM\n", sta->sta.addr);
Johannes Berg73651ee2008-02-25 16:27:47 +0100413 return sta;
Johannes Bergef04a292014-01-06 15:56:59 +0100414
415free:
416 if (sta->tx_lat) {
417 for (i = 0; i < IEEE80211_NUM_TIDS; i++)
418 kfree(sta->tx_lat[i].bins);
419 kfree(sta->tx_lat);
420 }
421 kfree(sta);
422 return NULL;
Johannes Berg73651ee2008-02-25 16:27:47 +0100423}
424
Guy Eilam8c71df72011-08-17 15:18:14 +0300425static int sta_info_insert_check(struct sta_info *sta)
Johannes Berg34e89502010-02-03 13:59:58 +0100426{
Johannes Berg34e89502010-02-03 13:59:58 +0100427 struct ieee80211_sub_if_data *sdata = sta->sdata;
Johannes Berg34e89502010-02-03 13:59:58 +0100428
Johannes Berg03e44972008-02-27 09:56:40 +0100429 /*
430 * Can't be a WARN_ON because it can be triggered through a race:
431 * something inserts a STA (on one CPU) without holding the RTNL
432 * and another CPU turns off the net device.
433 */
Guy Eilam8c71df72011-08-17 15:18:14 +0300434 if (unlikely(!ieee80211_sdata_running(sdata)))
435 return -ENETDOWN;
Johannes Berg03e44972008-02-27 09:56:40 +0100436
Joe Perchesb203ca32012-05-08 18:56:52 +0000437 if (WARN_ON(ether_addr_equal(sta->sta.addr, sdata->vif.addr) ||
Guy Eilam8c71df72011-08-17 15:18:14 +0300438 is_multicast_ether_addr(sta->sta.addr)))
439 return -EINVAL;
440
441 return 0;
442}
443
Johannes Bergf09603a2012-01-20 13:55:21 +0100444static int sta_info_insert_drv_state(struct ieee80211_local *local,
445 struct ieee80211_sub_if_data *sdata,
446 struct sta_info *sta)
447{
448 enum ieee80211_sta_state state;
449 int err = 0;
450
451 for (state = IEEE80211_STA_NOTEXIST; state < sta->sta_state; state++) {
452 err = drv_sta_state(local, sdata, sta, state, state + 1);
453 if (err)
454 break;
455 }
456
457 if (!err) {
Johannes Berga4ec45a2012-01-20 13:55:22 +0100458 /*
459 * Drivers using legacy sta_add/sta_remove callbacks only
460 * get uploaded set to true after sta_add is called.
461 */
462 if (!local->ops->sta_add)
463 sta->uploaded = true;
Johannes Bergf09603a2012-01-20 13:55:21 +0100464 return 0;
465 }
466
467 if (sdata->vif.type == NL80211_IFTYPE_ADHOC) {
Johannes Bergbdcbd8e2012-06-22 11:29:50 +0200468 sdata_info(sdata,
469 "failed to move IBSS STA %pM to state %d (%d) - keeping it anyway\n",
470 sta->sta.addr, state + 1, err);
Johannes Bergf09603a2012-01-20 13:55:21 +0100471 err = 0;
472 }
473
474 /* unwind on error */
475 for (; state > IEEE80211_STA_NOTEXIST; state--)
476 WARN_ON(drv_sta_state(local, sdata, sta, state, state - 1));
477
478 return err;
479}
480
Guy Eilam8c71df72011-08-17 15:18:14 +0300481/*
482 * should be called with sta_mtx locked
483 * this function replaces the mutex lock
484 * with a RCU lock
485 */
Johannes Berg4d339602011-12-15 11:24:20 +0100486static int sta_info_insert_finish(struct sta_info *sta) __acquires(RCU)
Guy Eilam8c71df72011-08-17 15:18:14 +0300487{
488 struct ieee80211_local *local = sta->local;
489 struct ieee80211_sub_if_data *sdata = sta->sdata;
Johannes Berg7852e362012-01-20 13:55:24 +0100490 struct station_info sinfo;
Guy Eilam8c71df72011-08-17 15:18:14 +0300491 int err = 0;
492
493 lockdep_assert_held(&local->sta_mtx);
Johannes Berg34e89502010-02-03 13:59:58 +0100494
Johannes Berg7852e362012-01-20 13:55:24 +0100495 /* check if STA exists already */
496 if (sta_info_get_bss(sdata, sta->sta.addr)) {
497 err = -EEXIST;
498 goto out_err;
Johannes Berg43ba7e92008-02-21 14:09:30 +0100499 }
Johannes Berg32bfd352007-12-19 01:31:26 +0100500
Johannes Berg7852e362012-01-20 13:55:24 +0100501 local->num_sta++;
502 local->sta_generation++;
503 smp_mb();
Johannes Berg4d339602011-12-15 11:24:20 +0100504
Johannes Berg5108ca82014-02-17 20:49:03 +0100505 /* simplify things and don't accept BA sessions yet */
506 set_sta_flag(sta, WLAN_STA_BLOCK_BA);
507
Johannes Berg7852e362012-01-20 13:55:24 +0100508 /* make the station visible */
509 sta_info_hash_add(local, sta);
Johannes Berg4d339602011-12-15 11:24:20 +0100510
Arik Nemtsov794454c2012-06-03 23:32:32 +0300511 list_add_rcu(&sta->list, &local->sta_list);
Johannes Berg83d5cc02012-01-12 09:31:10 +0100512
Johannes Berg5108ca82014-02-17 20:49:03 +0100513 /* notify driver */
514 err = sta_info_insert_drv_state(local, sdata, sta);
515 if (err)
516 goto out_remove;
517
Johannes Berg7852e362012-01-20 13:55:24 +0100518 set_sta_flag(sta, WLAN_STA_INSERTED);
Johannes Berg5108ca82014-02-17 20:49:03 +0100519 /* accept BA sessions now */
520 clear_sta_flag(sta, WLAN_STA_BLOCK_BA);
Johannes Berg4d339602011-12-15 11:24:20 +0100521
Eliad Peller21f659b2013-11-11 20:14:01 +0200522 ieee80211_recalc_min_chandef(sdata);
Johannes Berg7852e362012-01-20 13:55:24 +0100523 ieee80211_sta_debugfs_add(sta);
524 rate_control_add_sta_debugfs(sta);
Johannes Berg4d339602011-12-15 11:24:20 +0100525
Johannes Berg7852e362012-01-20 13:55:24 +0100526 memset(&sinfo, 0, sizeof(sinfo));
527 sinfo.filled = 0;
528 sinfo.generation = local->sta_generation;
529 cfg80211_new_sta(sdata->dev, sta->sta.addr, &sinfo, GFP_KERNEL);
Johannes Bergd0709a62008-02-25 16:27:46 +0100530
Johannes Bergbdcbd8e2012-06-22 11:29:50 +0200531 sta_dbg(sdata, "Inserted STA %pM\n", sta->sta.addr);
Jiri Bencf0706e82007-05-05 11:45:53 -0700532
Johannes Berg34e89502010-02-03 13:59:58 +0100533 /* move reference to rcu-protected */
534 rcu_read_lock();
535 mutex_unlock(&local->sta_mtx);
Jiri Bence9f207f2007-05-05 11:46:38 -0700536
Johannes Berg73651ee2008-02-25 16:27:47 +0100537 if (ieee80211_vif_is_mesh(&sdata->vif))
538 mesh_accept_plinks_update(sdata);
539
540 return 0;
Johannes Berg5108ca82014-02-17 20:49:03 +0100541 out_remove:
542 sta_info_hash_del(local, sta);
543 list_del_rcu(&sta->list);
544 local->num_sta--;
545 synchronize_net();
546 __cleanup_single_sta(sta);
Johannes Berg4d339602011-12-15 11:24:20 +0100547 out_err:
548 mutex_unlock(&local->sta_mtx);
549 rcu_read_lock();
550 return err;
Guy Eilam8c71df72011-08-17 15:18:14 +0300551}
552
553int sta_info_insert_rcu(struct sta_info *sta) __acquires(RCU)
554{
555 struct ieee80211_local *local = sta->local;
Zhao, Gang308f7fc2014-04-21 12:53:00 +0800556 int err;
Guy Eilam8c71df72011-08-17 15:18:14 +0300557
Johannes Berg4d339602011-12-15 11:24:20 +0100558 might_sleep();
559
Guy Eilam8c71df72011-08-17 15:18:14 +0300560 err = sta_info_insert_check(sta);
561 if (err) {
Jiri Bencf0706e82007-05-05 11:45:53 -0700562 rcu_read_lock();
563 goto out_free;
564 }
565
Johannes Berg004c8722008-02-20 11:21:35 +0100566 mutex_lock(&local->sta_mtx);
567
Johannes Berg4d339602011-12-15 11:24:20 +0100568 err = sta_info_insert_finish(sta);
Guy Eilam8c71df72011-08-17 15:18:14 +0300569 if (err)
Johannes Berg004c8722008-02-20 11:21:35 +0100570 goto out_free;
Johannes Bergd0709a62008-02-25 16:27:46 +0100571
Johannes Berg004c8722008-02-20 11:21:35 +0100572 return 0;
Johannes Berg93e5deb2008-04-01 15:21:00 +0200573 out_free:
Johannes Bergd9a7ddb2011-12-14 12:35:30 +0100574 sta_info_free(local, sta);
Johannes Berg93e5deb2008-04-01 15:21:00 +0200575 return err;
Jiri Bencf0706e82007-05-05 11:45:53 -0700576}
577
Johannes Berg34e89502010-02-03 13:59:58 +0100578int sta_info_insert(struct sta_info *sta)
579{
580 int err = sta_info_insert_rcu(sta);
581
582 rcu_read_unlock();
583
584 return err;
585}
586
Marco Porschd012a602012-10-10 12:39:50 -0700587static inline void __bss_tim_set(u8 *tim, u16 id)
Johannes Berg004c8722008-02-20 11:21:35 +0100588{
589 /*
590 * This format has been mandated by the IEEE specifications,
591 * so this line may not be changed to use the __set_bit() format.
592 */
Marco Porschd012a602012-10-10 12:39:50 -0700593 tim[id / 8] |= (1 << (id % 8));
Johannes Berg004c8722008-02-20 11:21:35 +0100594}
595
Marco Porschd012a602012-10-10 12:39:50 -0700596static inline void __bss_tim_clear(u8 *tim, u16 id)
Johannes Berg004c8722008-02-20 11:21:35 +0100597{
598 /*
599 * This format has been mandated by the IEEE specifications,
600 * so this line may not be changed to use the __clear_bit() format.
601 */
Marco Porschd012a602012-10-10 12:39:50 -0700602 tim[id / 8] &= ~(1 << (id % 8));
Johannes Berg004c8722008-02-20 11:21:35 +0100603}
604
Ilan Peer3d5839b2013-03-05 15:27:20 +0200605static inline bool __bss_tim_get(u8 *tim, u16 id)
606{
607 /*
608 * This format has been mandated by the IEEE specifications,
609 * so this line may not be changed to use the test_bit() format.
610 */
611 return tim[id / 8] & (1 << (id % 8));
612}
613
Johannes Berg948d8872011-09-29 16:04:29 +0200614static unsigned long ieee80211_tids_for_ac(int ac)
Johannes Berg004c8722008-02-20 11:21:35 +0100615{
Johannes Berg948d8872011-09-29 16:04:29 +0200616 /* If we ever support TIDs > 7, this obviously needs to be adjusted */
617 switch (ac) {
618 case IEEE80211_AC_VO:
619 return BIT(6) | BIT(7);
620 case IEEE80211_AC_VI:
621 return BIT(4) | BIT(5);
622 case IEEE80211_AC_BE:
623 return BIT(0) | BIT(3);
624 case IEEE80211_AC_BK:
625 return BIT(1) | BIT(2);
626 default:
627 WARN_ON(1);
628 return 0;
Johannes Bergd0709a62008-02-25 16:27:46 +0100629 }
Johannes Berg004c8722008-02-20 11:21:35 +0100630}
631
Johannes Bergc868cb352011-09-29 16:04:27 +0200632void sta_info_recalc_tim(struct sta_info *sta)
Johannes Berg004c8722008-02-20 11:21:35 +0100633{
Johannes Bergc868cb352011-09-29 16:04:27 +0200634 struct ieee80211_local *local = sta->local;
Marco Porschd012a602012-10-10 12:39:50 -0700635 struct ps_data *ps;
Johannes Berg948d8872011-09-29 16:04:29 +0200636 bool indicate_tim = false;
637 u8 ignore_for_tim = sta->sta.uapsd_queues;
638 int ac;
Marco Porschd012a602012-10-10 12:39:50 -0700639 u16 id;
Johannes Berg004c8722008-02-20 11:21:35 +0100640
Marco Porschd012a602012-10-10 12:39:50 -0700641 if (sta->sdata->vif.type == NL80211_IFTYPE_AP ||
642 sta->sdata->vif.type == NL80211_IFTYPE_AP_VLAN) {
643 if (WARN_ON_ONCE(!sta->sdata->bss))
644 return;
645
646 ps = &sta->sdata->bss->ps;
647 id = sta->sta.aid;
Marco Porsch3f52b7e2013-01-30 18:14:08 +0100648#ifdef CONFIG_MAC80211_MESH
649 } else if (ieee80211_vif_is_mesh(&sta->sdata->vif)) {
650 ps = &sta->sdata->u.mesh.ps;
Thomas Pedersen204d1302013-11-05 11:17:05 -0800651 /* TIM map only for 1 <= PLID <= IEEE80211_MAX_AID */
Chun-Yeow Yeoh6f101ef2013-11-13 15:43:03 +0800652 id = sta->plid % (IEEE80211_MAX_AID + 1);
Marco Porsch3f52b7e2013-01-30 18:14:08 +0100653#endif
Marco Porschd012a602012-10-10 12:39:50 -0700654 } else {
Johannes Bergc868cb352011-09-29 16:04:27 +0200655 return;
Marco Porschd012a602012-10-10 12:39:50 -0700656 }
Johannes Berg3e122be2008-07-09 14:40:34 +0200657
Johannes Bergc868cb352011-09-29 16:04:27 +0200658 /* No need to do anything if the driver does all */
659 if (local->hw.flags & IEEE80211_HW_AP_LINK_PS)
660 return;
Johannes Berg004c8722008-02-20 11:21:35 +0100661
Johannes Bergc868cb352011-09-29 16:04:27 +0200662 if (sta->dead)
663 goto done;
Johannes Berg3e122be2008-07-09 14:40:34 +0200664
Johannes Berg948d8872011-09-29 16:04:29 +0200665 /*
666 * If all ACs are delivery-enabled then we should build
667 * the TIM bit for all ACs anyway; if only some are then
668 * we ignore those and build the TIM bit using only the
669 * non-enabled ones.
670 */
671 if (ignore_for_tim == BIT(IEEE80211_NUM_ACS) - 1)
672 ignore_for_tim = 0;
Johannes Berg3e122be2008-07-09 14:40:34 +0200673
Johannes Berg948d8872011-09-29 16:04:29 +0200674 for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) {
675 unsigned long tids;
676
677 if (ignore_for_tim & BIT(ac))
678 continue;
679
680 indicate_tim |= !skb_queue_empty(&sta->tx_filtered[ac]) ||
681 !skb_queue_empty(&sta->ps_tx_buf[ac]);
682 if (indicate_tim)
683 break;
684
685 tids = ieee80211_tids_for_ac(ac);
686
687 indicate_tim |=
688 sta->driver_buffered_tids & tids;
Johannes Bergd0709a62008-02-25 16:27:46 +0100689 }
Johannes Berg004c8722008-02-20 11:21:35 +0100690
Johannes Bergc868cb352011-09-29 16:04:27 +0200691 done:
Johannes Berg65f704a2013-02-13 17:39:53 +0100692 spin_lock_bh(&local->tim_lock);
Johannes Berg004c8722008-02-20 11:21:35 +0100693
Ilan Peer3d5839b2013-03-05 15:27:20 +0200694 if (indicate_tim == __bss_tim_get(ps->tim, id))
695 goto out_unlock;
696
Johannes Berg948d8872011-09-29 16:04:29 +0200697 if (indicate_tim)
Marco Porschd012a602012-10-10 12:39:50 -0700698 __bss_tim_set(ps->tim, id);
Johannes Bergc868cb352011-09-29 16:04:27 +0200699 else
Marco Porschd012a602012-10-10 12:39:50 -0700700 __bss_tim_clear(ps->tim, id);
Johannes Berg3e122be2008-07-09 14:40:34 +0200701
Johannes Bergc868cb352011-09-29 16:04:27 +0200702 if (local->ops->set_tim) {
703 local->tim_in_locked_section = true;
Johannes Berg948d8872011-09-29 16:04:29 +0200704 drv_set_tim(local, &sta->sta, indicate_tim);
Johannes Bergc868cb352011-09-29 16:04:27 +0200705 local->tim_in_locked_section = false;
Johannes Berg004c8722008-02-20 11:21:35 +0100706 }
Johannes Berg004c8722008-02-20 11:21:35 +0100707
Ilan Peer3d5839b2013-03-05 15:27:20 +0200708out_unlock:
Johannes Berg65f704a2013-02-13 17:39:53 +0100709 spin_unlock_bh(&local->tim_lock);
Johannes Berg004c8722008-02-20 11:21:35 +0100710}
711
Johannes Bergcd0b8d82011-09-06 14:13:06 +0200712static bool sta_info_buffer_expired(struct sta_info *sta, struct sk_buff *skb)
Jiri Bencf0706e82007-05-05 11:45:53 -0700713{
Johannes Berge039fa42008-05-15 12:55:29 +0200714 struct ieee80211_tx_info *info;
Jiri Bencf0706e82007-05-05 11:45:53 -0700715 int timeout;
716
717 if (!skb)
Johannes Bergcd0b8d82011-09-06 14:13:06 +0200718 return false;
Jiri Bencf0706e82007-05-05 11:45:53 -0700719
Johannes Berge039fa42008-05-15 12:55:29 +0200720 info = IEEE80211_SKB_CB(skb);
Jiri Bencf0706e82007-05-05 11:45:53 -0700721
722 /* Timeout: (2 * listen_interval * beacon_int * 1024 / 1000000) sec */
Johannes Berg57c4d7b2009-04-23 16:10:04 +0200723 timeout = (sta->listen_interval *
724 sta->sdata->vif.bss_conf.beacon_int *
725 32 / 15625) * HZ;
Jiri Bencf0706e82007-05-05 11:45:53 -0700726 if (timeout < STA_TX_BUFFER_EXPIRE)
727 timeout = STA_TX_BUFFER_EXPIRE;
Johannes Berge039fa42008-05-15 12:55:29 +0200728 return time_after(jiffies, info->control.jiffies + timeout);
Jiri Bencf0706e82007-05-05 11:45:53 -0700729}
730
731
Johannes Berg948d8872011-09-29 16:04:29 +0200732static bool sta_info_cleanup_expire_buffered_ac(struct ieee80211_local *local,
733 struct sta_info *sta, int ac)
Jiri Bencf0706e82007-05-05 11:45:53 -0700734{
735 unsigned long flags;
736 struct sk_buff *skb;
737
Johannes Berg60750392011-09-29 16:04:28 +0200738 /*
739 * First check for frames that should expire on the filtered
740 * queue. Frames here were rejected by the driver and are on
741 * a separate queue to avoid reordering with normal PS-buffered
742 * frames. They also aren't accounted for right now in the
743 * total_ps_buffered counter.
744 */
Jiri Bencf0706e82007-05-05 11:45:53 -0700745 for (;;) {
Johannes Berg948d8872011-09-29 16:04:29 +0200746 spin_lock_irqsave(&sta->tx_filtered[ac].lock, flags);
747 skb = skb_peek(&sta->tx_filtered[ac]);
Johannes Berg57c4d7b2009-04-23 16:10:04 +0200748 if (sta_info_buffer_expired(sta, skb))
Johannes Berg948d8872011-09-29 16:04:29 +0200749 skb = __skb_dequeue(&sta->tx_filtered[ac]);
Johannes Berg836341a2008-02-20 02:07:21 +0100750 else
Jiri Bencf0706e82007-05-05 11:45:53 -0700751 skb = NULL;
Johannes Berg948d8872011-09-29 16:04:29 +0200752 spin_unlock_irqrestore(&sta->tx_filtered[ac].lock, flags);
Jiri Bencf0706e82007-05-05 11:45:53 -0700753
Johannes Berg60750392011-09-29 16:04:28 +0200754 /*
755 * Frames are queued in order, so if this one
756 * hasn't expired yet we can stop testing. If
757 * we actually reached the end of the queue we
758 * also need to stop, of course.
759 */
760 if (!skb)
761 break;
Felix Fietkaud4fa14c2012-10-10 22:40:23 +0200762 ieee80211_free_txskb(&local->hw, skb);
Johannes Berg60750392011-09-29 16:04:28 +0200763 }
764
765 /*
766 * Now also check the normal PS-buffered queue, this will
767 * only find something if the filtered queue was emptied
768 * since the filtered frames are all before the normal PS
769 * buffered frames.
770 */
Jiri Bencf0706e82007-05-05 11:45:53 -0700771 for (;;) {
Johannes Berg948d8872011-09-29 16:04:29 +0200772 spin_lock_irqsave(&sta->ps_tx_buf[ac].lock, flags);
773 skb = skb_peek(&sta->ps_tx_buf[ac]);
Jiri Bencf0706e82007-05-05 11:45:53 -0700774 if (sta_info_buffer_expired(sta, skb))
Johannes Berg948d8872011-09-29 16:04:29 +0200775 skb = __skb_dequeue(&sta->ps_tx_buf[ac]);
Jiri Bencf0706e82007-05-05 11:45:53 -0700776 else
777 skb = NULL;
Johannes Berg948d8872011-09-29 16:04:29 +0200778 spin_unlock_irqrestore(&sta->ps_tx_buf[ac].lock, flags);
Jiri Bencf0706e82007-05-05 11:45:53 -0700779
Johannes Berg60750392011-09-29 16:04:28 +0200780 /*
781 * frames are queued in order, so if this one
782 * hasn't expired yet (or we reached the end of
783 * the queue) we can stop testing
784 */
Johannes Berg836341a2008-02-20 02:07:21 +0100785 if (!skb)
Jiri Bencf0706e82007-05-05 11:45:53 -0700786 break;
Johannes Berg836341a2008-02-20 02:07:21 +0100787
Johannes Berg836341a2008-02-20 02:07:21 +0100788 local->total_ps_buffered--;
Johannes Bergbdcbd8e2012-06-22 11:29:50 +0200789 ps_dbg(sta->sdata, "Buffered frame expired (STA %pM)\n",
790 sta->sta.addr);
Felix Fietkaud4fa14c2012-10-10 22:40:23 +0200791 ieee80211_free_txskb(&local->hw, skb);
Jiri Bencf0706e82007-05-05 11:45:53 -0700792 }
Juuso Oikarinen3393a602010-04-19 10:12:52 +0300793
Johannes Berg60750392011-09-29 16:04:28 +0200794 /*
795 * Finally, recalculate the TIM bit for this station -- it might
796 * now be clear because the station was too slow to retrieve its
797 * frames.
798 */
799 sta_info_recalc_tim(sta);
800
801 /*
802 * Return whether there are any frames still buffered, this is
803 * used to check whether the cleanup timer still needs to run,
804 * if there are no frames we don't need to rearm the timer.
805 */
Johannes Berg948d8872011-09-29 16:04:29 +0200806 return !(skb_queue_empty(&sta->ps_tx_buf[ac]) &&
807 skb_queue_empty(&sta->tx_filtered[ac]));
808}
809
810static bool sta_info_cleanup_expire_buffered(struct ieee80211_local *local,
811 struct sta_info *sta)
812{
813 bool have_buffered = false;
814 int ac;
815
Marco Porsch3f52b7e2013-01-30 18:14:08 +0100816 /* This is only necessary for stations on BSS/MBSS interfaces */
817 if (!sta->sdata->bss &&
818 !ieee80211_vif_is_mesh(&sta->sdata->vif))
Johannes Berg948d8872011-09-29 16:04:29 +0200819 return false;
820
821 for (ac = 0; ac < IEEE80211_NUM_ACS; ac++)
822 have_buffered |=
823 sta_info_cleanup_expire_buffered_ac(local, sta, ac);
824
825 return have_buffered;
Jiri Bencf0706e82007-05-05 11:45:53 -0700826}
827
Johannes Bergd7782072013-12-04 23:12:31 +0100828static int __must_check __sta_info_destroy_part1(struct sta_info *sta)
Johannes Berg34e89502010-02-03 13:59:58 +0100829{
830 struct ieee80211_local *local;
831 struct ieee80211_sub_if_data *sdata;
Johannes Berg6d10e462013-03-06 23:09:11 +0100832 int ret;
Johannes Berg34e89502010-02-03 13:59:58 +0100833
834 might_sleep();
835
836 if (!sta)
837 return -ENOENT;
838
839 local = sta->local;
840 sdata = sta->sdata;
841
Johannes Berg83d5cc02012-01-12 09:31:10 +0100842 lockdep_assert_held(&local->sta_mtx);
843
Johannes Berg098a6072010-04-06 11:18:47 +0200844 /*
845 * Before removing the station from the driver and
846 * rate control, it might still start new aggregation
847 * sessions -- block that to make sure the tear-down
848 * will be sufficient.
849 */
Johannes Bergc2c98fd2011-09-29 16:04:36 +0200850 set_sta_flag(sta, WLAN_STA_BLOCK_BA);
Johannes Bergc82c4a82012-07-18 13:31:31 +0200851 ieee80211_sta_tear_down_BA_sessions(sta, AGG_STOP_DESTROY_STA);
Johannes Berg098a6072010-04-06 11:18:47 +0200852
Johannes Berg34e89502010-02-03 13:59:58 +0100853 ret = sta_info_hash_del(local, sta);
Johannes Bergb01711b2013-12-04 20:25:27 +0100854 if (WARN_ON(ret))
Johannes Berg34e89502010-02-03 13:59:58 +0100855 return ret;
856
Arik Nemtsov794454c2012-06-03 23:32:32 +0300857 list_del_rcu(&sta->list);
Johannes Berg4d339602011-12-15 11:24:20 +0100858
Johannes Berg6a9d1b92013-12-04 22:39:17 +0100859 drv_sta_pre_rcu_remove(local, sta->sdata, sta);
860
Johannes Berga710c812013-12-04 20:11:06 +0100861 if (sdata->vif.type == NL80211_IFTYPE_AP_VLAN &&
862 rcu_access_pointer(sdata->u.vlan.sta) == sta)
863 RCU_INIT_POINTER(sdata->u.vlan.sta, NULL);
864
Johannes Bergd7782072013-12-04 23:12:31 +0100865 return 0;
866}
867
868static void __sta_info_destroy_part2(struct sta_info *sta)
869{
870 struct ieee80211_local *local = sta->local;
871 struct ieee80211_sub_if_data *sdata = sta->sdata;
872 int ret;
873
874 /*
875 * NOTE: This assumes at least synchronize_net() was done
876 * after _part1 and before _part2!
877 */
878
879 might_sleep();
880 lockdep_assert_held(&local->sta_mtx);
881
Johannes Bergc8782072013-12-04 23:05:45 +0100882 /* now keys can no longer be reached */
Johannes Berg6d10e462013-03-06 23:09:11 +0100883 ieee80211_free_sta_keys(local, sta);
Johannes Berg34e89502010-02-03 13:59:58 +0100884
885 sta->dead = true;
886
Johannes Berg34e89502010-02-03 13:59:58 +0100887 local->num_sta--;
888 local->sta_generation++;
889
Johannes Berg83d5cc02012-01-12 09:31:10 +0100890 while (sta->sta_state > IEEE80211_STA_NONE) {
Johannes Bergf09603a2012-01-20 13:55:21 +0100891 ret = sta_info_move_state(sta, sta->sta_state - 1);
892 if (ret) {
Johannes Berg83d5cc02012-01-12 09:31:10 +0100893 WARN_ON_ONCE(1);
894 break;
895 }
896 }
Johannes Bergd9a7ddb2011-12-14 12:35:30 +0100897
Johannes Bergf09603a2012-01-20 13:55:21 +0100898 if (sta->uploaded) {
Johannes Bergf09603a2012-01-20 13:55:21 +0100899 ret = drv_sta_state(local, sdata, sta, IEEE80211_STA_NONE,
900 IEEE80211_STA_NOTEXIST);
901 WARN_ON_ONCE(ret != 0);
902 }
Johannes Berg34e89502010-02-03 13:59:58 +0100903
Johannes Bergbdcbd8e2012-06-22 11:29:50 +0200904 sta_dbg(sdata, "Removed STA %pM\n", sta->sta.addr);
905
Jouni Malinenec15e682011-03-23 15:29:52 +0200906 cfg80211_del_sta(sdata->dev, sta->sta.addr, GFP_KERNEL);
907
Johannes Berg34e89502010-02-03 13:59:58 +0100908 rate_control_remove_sta_debugfs(sta);
909 ieee80211_sta_debugfs_remove(sta);
Eliad Peller21f659b2013-11-11 20:14:01 +0200910 ieee80211_recalc_min_chandef(sdata);
Johannes Berg34e89502010-02-03 13:59:58 +0100911
Johannes Bergd34ba212013-12-04 22:46:11 +0100912 cleanup_single_sta(sta);
Johannes Bergd7782072013-12-04 23:12:31 +0100913}
914
915int __must_check __sta_info_destroy(struct sta_info *sta)
916{
917 int err = __sta_info_destroy_part1(sta);
918
919 if (err)
920 return err;
921
922 synchronize_net();
923
924 __sta_info_destroy_part2(sta);
Johannes Berg34e89502010-02-03 13:59:58 +0100925
926 return 0;
927}
928
929int sta_info_destroy_addr(struct ieee80211_sub_if_data *sdata, const u8 *addr)
930{
931 struct sta_info *sta;
932 int ret;
933
934 mutex_lock(&sdata->local->sta_mtx);
Johannes Berg7852e362012-01-20 13:55:24 +0100935 sta = sta_info_get(sdata, addr);
Johannes Berg34e89502010-02-03 13:59:58 +0100936 ret = __sta_info_destroy(sta);
937 mutex_unlock(&sdata->local->sta_mtx);
938
939 return ret;
940}
941
942int sta_info_destroy_addr_bss(struct ieee80211_sub_if_data *sdata,
943 const u8 *addr)
944{
945 struct sta_info *sta;
946 int ret;
947
948 mutex_lock(&sdata->local->sta_mtx);
Johannes Berg7852e362012-01-20 13:55:24 +0100949 sta = sta_info_get_bss(sdata, addr);
Johannes Berg34e89502010-02-03 13:59:58 +0100950 ret = __sta_info_destroy(sta);
951 mutex_unlock(&sdata->local->sta_mtx);
952
953 return ret;
954}
Jiri Bencf0706e82007-05-05 11:45:53 -0700955
956static void sta_info_cleanup(unsigned long data)
957{
958 struct ieee80211_local *local = (struct ieee80211_local *) data;
959 struct sta_info *sta;
Juuso Oikarinen3393a602010-04-19 10:12:52 +0300960 bool timer_needed = false;
Jiri Bencf0706e82007-05-05 11:45:53 -0700961
Johannes Bergd0709a62008-02-25 16:27:46 +0100962 rcu_read_lock();
963 list_for_each_entry_rcu(sta, &local->sta_list, list)
Juuso Oikarinen3393a602010-04-19 10:12:52 +0300964 if (sta_info_cleanup_expire_buffered(local, sta))
965 timer_needed = true;
Johannes Bergd0709a62008-02-25 16:27:46 +0100966 rcu_read_unlock();
Jiri Bencf0706e82007-05-05 11:45:53 -0700967
Johannes Berg5bb644a2009-05-17 11:40:42 +0200968 if (local->quiescing)
969 return;
970
Juuso Oikarinen3393a602010-04-19 10:12:52 +0300971 if (!timer_needed)
972 return;
973
Johannes Berg26d59532011-04-01 13:52:48 +0200974 mod_timer(&local->sta_cleanup,
975 round_jiffies(jiffies + STA_INFO_CLEANUP_INTERVAL));
Jiri Bencf0706e82007-05-05 11:45:53 -0700976}
977
978void sta_info_init(struct ieee80211_local *local)
979{
Johannes Berg4d339602011-12-15 11:24:20 +0100980 spin_lock_init(&local->tim_lock);
Johannes Berg34e89502010-02-03 13:59:58 +0100981 mutex_init(&local->sta_mtx);
Jiri Bencf0706e82007-05-05 11:45:53 -0700982 INIT_LIST_HEAD(&local->sta_list);
Jiri Bencf0706e82007-05-05 11:45:53 -0700983
Pavel Emelyanovb24b8a22008-01-23 21:20:07 -0800984 setup_timer(&local->sta_cleanup, sta_info_cleanup,
985 (unsigned long)local);
Jiri Bencf0706e82007-05-05 11:45:53 -0700986}
987
988void sta_info_stop(struct ieee80211_local *local)
989{
Johannes Berga56f9922012-12-13 23:08:52 +0100990 del_timer_sync(&local->sta_cleanup);
Jiri Bencf0706e82007-05-05 11:45:53 -0700991}
992
Johannes Berg051007d2012-12-13 23:49:02 +0100993
Johannes Berge7162512013-12-04 23:18:37 +0100994int __sta_info_flush(struct ieee80211_sub_if_data *sdata, bool vlans)
Jiri Bencf0706e82007-05-05 11:45:53 -0700995{
Johannes Bergb998e8b2012-12-13 23:07:46 +0100996 struct ieee80211_local *local = sdata->local;
Jiri Bencf0706e82007-05-05 11:45:53 -0700997 struct sta_info *sta, *tmp;
Johannes Bergd7782072013-12-04 23:12:31 +0100998 LIST_HEAD(free_list);
Johannes Berg44213b52008-02-25 16:27:49 +0100999 int ret = 0;
Jiri Bencf0706e82007-05-05 11:45:53 -07001000
Johannes Bergd0709a62008-02-25 16:27:46 +01001001 might_sleep();
1002
Johannes Berge7162512013-12-04 23:18:37 +01001003 WARN_ON(vlans && sdata->vif.type != NL80211_IFTYPE_AP);
1004 WARN_ON(vlans && !sdata->bss);
1005
Johannes Berg34e89502010-02-03 13:59:58 +01001006 mutex_lock(&local->sta_mtx);
Johannes Berg34e89502010-02-03 13:59:58 +01001007 list_for_each_entry_safe(sta, tmp, &local->sta_list, list) {
Johannes Berge7162512013-12-04 23:18:37 +01001008 if (sdata == sta->sdata ||
1009 (vlans && sdata->bss == sta->sdata->bss)) {
Johannes Bergd7782072013-12-04 23:12:31 +01001010 if (!WARN_ON(__sta_info_destroy_part1(sta)))
1011 list_add(&sta->free_list, &free_list);
Johannes Berg34316832012-02-25 21:40:46 +01001012 ret++;
1013 }
Johannes Berg34e89502010-02-03 13:59:58 +01001014 }
Johannes Bergd7782072013-12-04 23:12:31 +01001015
1016 if (!list_empty(&free_list)) {
1017 synchronize_net();
1018 list_for_each_entry_safe(sta, tmp, &free_list, free_list)
1019 __sta_info_destroy_part2(sta);
1020 }
Johannes Berg34e89502010-02-03 13:59:58 +01001021 mutex_unlock(&local->sta_mtx);
Johannes Berg44213b52008-02-25 16:27:49 +01001022
Johannes Berg051007d2012-12-13 23:49:02 +01001023 return ret;
1024}
1025
Johannes Berg24723d12008-09-11 00:01:46 +02001026void ieee80211_sta_expire(struct ieee80211_sub_if_data *sdata,
1027 unsigned long exp_time)
1028{
1029 struct ieee80211_local *local = sdata->local;
1030 struct sta_info *sta, *tmp;
Johannes Berg24723d12008-09-11 00:01:46 +02001031
Johannes Berg34e89502010-02-03 13:59:58 +01001032 mutex_lock(&local->sta_mtx);
Mohammed Shafi Shajakhane46a2cf2011-12-26 10:43:29 +05301033
1034 list_for_each_entry_safe(sta, tmp, &local->sta_list, list) {
Marek Lindnerec2b7742011-12-20 23:16:52 +08001035 if (sdata != sta->sdata)
1036 continue;
1037
Johannes Berg24723d12008-09-11 00:01:46 +02001038 if (time_after(jiffies, sta->last_rx + exp_time)) {
Mohammed Shafi Shajakhaneea57d42012-10-08 21:33:47 +05301039 sta_dbg(sta->sdata, "expiring inactive STA %pM\n",
1040 sta->sta.addr);
Marco Porsch3f52b7e2013-01-30 18:14:08 +01001041
1042 if (ieee80211_vif_is_mesh(&sdata->vif) &&
1043 test_sta_flag(sta, WLAN_STA_PS_STA))
1044 atomic_dec(&sdata->u.mesh.ps.num_sta_ps);
1045
Johannes Berg34e89502010-02-03 13:59:58 +01001046 WARN_ON(__sta_info_destroy(sta));
Johannes Berg24723d12008-09-11 00:01:46 +02001047 }
Mohammed Shafi Shajakhane46a2cf2011-12-26 10:43:29 +05301048 }
1049
Johannes Berg34e89502010-02-03 13:59:58 +01001050 mutex_unlock(&local->sta_mtx);
Johannes Berg24723d12008-09-11 00:01:46 +02001051}
Johannes Berg17741cd2008-09-11 00:02:02 +02001052
Ben Greear686b9cb2010-09-23 09:44:36 -07001053struct ieee80211_sta *ieee80211_find_sta_by_ifaddr(struct ieee80211_hw *hw,
1054 const u8 *addr,
1055 const u8 *localaddr)
Johannes Berg17741cd2008-09-11 00:02:02 +02001056{
Johannes Bergabe60632009-11-25 17:46:18 +01001057 struct sta_info *sta, *nxt;
Johannes Berg17741cd2008-09-11 00:02:02 +02001058
Ben Greear686b9cb2010-09-23 09:44:36 -07001059 /*
1060 * Just return a random station if localaddr is NULL
1061 * ... first in list.
1062 */
Johannes Bergf7c65592010-04-30 13:48:36 +02001063 for_each_sta_info(hw_to_local(hw), addr, sta, nxt) {
Ben Greear686b9cb2010-09-23 09:44:36 -07001064 if (localaddr &&
Joe Perchesb203ca32012-05-08 18:56:52 +00001065 !ether_addr_equal(sta->sdata->vif.addr, localaddr))
Ben Greear686b9cb2010-09-23 09:44:36 -07001066 continue;
Johannes Bergf7c65592010-04-30 13:48:36 +02001067 if (!sta->uploaded)
1068 return NULL;
Johannes Bergabe60632009-11-25 17:46:18 +01001069 return &sta->sta;
Johannes Bergf7c65592010-04-30 13:48:36 +02001070 }
1071
Johannes Bergabe60632009-11-25 17:46:18 +01001072 return NULL;
Johannes Berg17741cd2008-09-11 00:02:02 +02001073}
Ben Greear686b9cb2010-09-23 09:44:36 -07001074EXPORT_SYMBOL_GPL(ieee80211_find_sta_by_ifaddr);
Johannes Berg5ed176e2009-11-04 14:42:28 +01001075
1076struct ieee80211_sta *ieee80211_find_sta(struct ieee80211_vif *vif,
1077 const u8 *addr)
1078{
Johannes Bergf7c65592010-04-30 13:48:36 +02001079 struct sta_info *sta;
Johannes Berg5ed176e2009-11-04 14:42:28 +01001080
1081 if (!vif)
1082 return NULL;
1083
Johannes Bergf7c65592010-04-30 13:48:36 +02001084 sta = sta_info_get_bss(vif_to_sdata(vif), addr);
1085 if (!sta)
1086 return NULL;
Johannes Berg5ed176e2009-11-04 14:42:28 +01001087
Johannes Bergf7c65592010-04-30 13:48:36 +02001088 if (!sta->uploaded)
1089 return NULL;
1090
1091 return &sta->sta;
Johannes Berg5ed176e2009-11-04 14:42:28 +01001092}
Johannes Berg17741cd2008-09-11 00:02:02 +02001093EXPORT_SYMBOL(ieee80211_find_sta);
Johannes Bergaf818582009-11-06 11:35:50 +01001094
Johannes Berge3685e02014-02-20 11:19:58 +01001095/* powersave support code */
1096void ieee80211_sta_ps_deliver_wakeup(struct sta_info *sta)
Johannes Berg50a94322010-11-16 11:50:28 -08001097{
Helmut Schaa608383b2012-01-30 15:18:00 +01001098 struct ieee80211_sub_if_data *sdata = sta->sdata;
Johannes Berge3685e02014-02-20 11:19:58 +01001099 struct ieee80211_local *local = sdata->local;
1100 struct sk_buff_head pending;
1101 int filtered = 0, buffered = 0, ac;
1102 unsigned long flags;
Marco Porschd012a602012-10-10 12:39:50 -07001103 struct ps_data *ps;
1104
1105 if (sdata->vif.type == NL80211_IFTYPE_AP ||
1106 sdata->vif.type == NL80211_IFTYPE_AP_VLAN)
1107 ps = &sdata->bss->ps;
Marco Porsch3f52b7e2013-01-30 18:14:08 +01001108 else if (ieee80211_vif_is_mesh(&sdata->vif))
1109 ps = &sdata->u.mesh.ps;
Marco Porschd012a602012-10-10 12:39:50 -07001110 else
1111 return;
Johannes Berg50a94322010-11-16 11:50:28 -08001112
Johannes Bergc2c98fd2011-09-29 16:04:36 +02001113 clear_sta_flag(sta, WLAN_STA_SP);
Johannes Berg47086fc2011-09-29 16:04:33 +02001114
Johannes Berg5a306f52012-11-14 23:22:21 +01001115 BUILD_BUG_ON(BITS_TO_LONGS(IEEE80211_NUM_TIDS) > 1);
Johannes Berg948d8872011-09-29 16:04:29 +02001116 sta->driver_buffered_tids = 0;
1117
Arik Nemtsovd057e5a2011-01-31 22:29:13 +02001118 if (!(local->hw.flags & IEEE80211_HW_AP_LINK_PS))
1119 drv_sta_notify(local, sdata, STA_NOTIFY_AWAKE, &sta->sta);
Johannes Bergaf818582009-11-06 11:35:50 +01001120
Johannes Berg948d8872011-09-29 16:04:29 +02001121 skb_queue_head_init(&pending);
Johannes Bergaf818582009-11-06 11:35:50 +01001122
Emmanuel Grumbach1d147bf2014-02-20 09:22:11 +02001123 /* sync with ieee80211_tx_h_unicast_ps_buf */
1124 spin_lock(&sta->ps_lock);
Johannes Bergaf818582009-11-06 11:35:50 +01001125 /* Send all buffered frames to the station */
Johannes Berg948d8872011-09-29 16:04:29 +02001126 for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) {
1127 int count = skb_queue_len(&pending), tmp;
1128
Arik Nemtsov987c2852012-11-05 10:27:52 +02001129 spin_lock_irqsave(&sta->tx_filtered[ac].lock, flags);
Johannes Berg948d8872011-09-29 16:04:29 +02001130 skb_queue_splice_tail_init(&sta->tx_filtered[ac], &pending);
Arik Nemtsov987c2852012-11-05 10:27:52 +02001131 spin_unlock_irqrestore(&sta->tx_filtered[ac].lock, flags);
Johannes Berg948d8872011-09-29 16:04:29 +02001132 tmp = skb_queue_len(&pending);
1133 filtered += tmp - count;
1134 count = tmp;
1135
Arik Nemtsov987c2852012-11-05 10:27:52 +02001136 spin_lock_irqsave(&sta->ps_tx_buf[ac].lock, flags);
Johannes Berg948d8872011-09-29 16:04:29 +02001137 skb_queue_splice_tail_init(&sta->ps_tx_buf[ac], &pending);
Arik Nemtsov987c2852012-11-05 10:27:52 +02001138 spin_unlock_irqrestore(&sta->ps_tx_buf[ac].lock, flags);
Johannes Berg948d8872011-09-29 16:04:29 +02001139 tmp = skb_queue_len(&pending);
1140 buffered += tmp - count;
1141 }
1142
Johannes Berge3685e02014-02-20 11:19:58 +01001143 ieee80211_add_pending_skbs(local, &pending);
1144 clear_sta_flag(sta, WLAN_STA_PS_DRIVER);
1145 clear_sta_flag(sta, WLAN_STA_PS_STA);
Emmanuel Grumbach1d147bf2014-02-20 09:22:11 +02001146 spin_unlock(&sta->ps_lock);
Johannes Berg948d8872011-09-29 16:04:29 +02001147
Johannes Berge3685e02014-02-20 11:19:58 +01001148 atomic_dec(&ps->num_sta_ps);
1149
Emmanuel Grumbach687da132013-10-01 16:45:43 +03001150 /* This station just woke up and isn't aware of our SMPS state */
Chun-Yeow Yeoh062f1d62014-04-22 18:19:25 +08001151 if (!ieee80211_vif_is_mesh(&sdata->vif) &&
1152 !ieee80211_smps_is_restrictive(sta->known_smps_mode,
Emmanuel Grumbach687da132013-10-01 16:45:43 +03001153 sdata->smps_mode) &&
1154 sta->known_smps_mode != sdata->bss->req_smps &&
1155 sta_info_tx_streams(sta) != 1) {
1156 ht_dbg(sdata,
1157 "%pM just woke up and MIMO capable - update SMPS\n",
1158 sta->sta.addr);
1159 ieee80211_send_smps_action(sdata, sdata->bss->req_smps,
1160 sta->sta.addr,
1161 sdata->vif.bss_conf.bssid);
1162 }
1163
Johannes Bergaf818582009-11-06 11:35:50 +01001164 local->total_ps_buffered -= buffered;
1165
Johannes Bergc868cb352011-09-29 16:04:27 +02001166 sta_info_recalc_tim(sta);
1167
Johannes Bergbdcbd8e2012-06-22 11:29:50 +02001168 ps_dbg(sdata,
1169 "STA %pM aid %d sending %d filtered/%d PS frames since STA not sleeping anymore\n",
1170 sta->sta.addr, sta->sta.aid, filtered, buffered);
Johannes Bergaf818582009-11-06 11:35:50 +01001171}
1172
Johannes Bergce662b442011-09-29 16:04:34 +02001173static void ieee80211_send_null_response(struct ieee80211_sub_if_data *sdata,
1174 struct sta_info *sta, int tid,
Johannes Bergb77cf4f2014-01-09 00:00:38 +01001175 enum ieee80211_frame_release_type reason,
1176 bool call_driver)
Johannes Bergce662b442011-09-29 16:04:34 +02001177{
1178 struct ieee80211_local *local = sdata->local;
1179 struct ieee80211_qos_hdr *nullfunc;
1180 struct sk_buff *skb;
1181 int size = sizeof(*nullfunc);
1182 __le16 fc;
Johannes Bergc2c98fd2011-09-29 16:04:36 +02001183 bool qos = test_sta_flag(sta, WLAN_STA_WME);
Johannes Bergce662b442011-09-29 16:04:34 +02001184 struct ieee80211_tx_info *info;
Johannes Berg55de9082012-07-26 17:24:39 +02001185 struct ieee80211_chanctx_conf *chanctx_conf;
Johannes Bergce662b442011-09-29 16:04:34 +02001186
1187 if (qos) {
1188 fc = cpu_to_le16(IEEE80211_FTYPE_DATA |
1189 IEEE80211_STYPE_QOS_NULLFUNC |
1190 IEEE80211_FCTL_FROMDS);
1191 } else {
1192 size -= 2;
1193 fc = cpu_to_le16(IEEE80211_FTYPE_DATA |
1194 IEEE80211_STYPE_NULLFUNC |
1195 IEEE80211_FCTL_FROMDS);
1196 }
1197
1198 skb = dev_alloc_skb(local->hw.extra_tx_headroom + size);
1199 if (!skb)
1200 return;
1201
1202 skb_reserve(skb, local->hw.extra_tx_headroom);
1203
1204 nullfunc = (void *) skb_put(skb, size);
1205 nullfunc->frame_control = fc;
1206 nullfunc->duration_id = 0;
1207 memcpy(nullfunc->addr1, sta->sta.addr, ETH_ALEN);
1208 memcpy(nullfunc->addr2, sdata->vif.addr, ETH_ALEN);
1209 memcpy(nullfunc->addr3, sdata->vif.addr, ETH_ALEN);
Johannes Berg864a6042014-03-04 13:46:53 +01001210 nullfunc->seq_ctrl = 0;
Johannes Bergce662b442011-09-29 16:04:34 +02001211
Johannes Berg59b66252011-10-13 13:19:19 +02001212 skb->priority = tid;
1213 skb_set_queue_mapping(skb, ieee802_1d_to_ac[tid]);
Johannes Bergce662b442011-09-29 16:04:34 +02001214 if (qos) {
Johannes Bergce662b442011-09-29 16:04:34 +02001215 nullfunc->qos_ctrl = cpu_to_le16(tid);
1216
Johannes Berg40b96402011-09-29 16:04:38 +02001217 if (reason == IEEE80211_FRAME_RELEASE_UAPSD)
Johannes Bergce662b442011-09-29 16:04:34 +02001218 nullfunc->qos_ctrl |=
1219 cpu_to_le16(IEEE80211_QOS_CTL_EOSP);
1220 }
1221
1222 info = IEEE80211_SKB_CB(skb);
1223
1224 /*
1225 * Tell TX path to send this frame even though the
1226 * STA may still remain is PS mode after this frame
Johannes Bergdeeaee12011-09-29 16:04:35 +02001227 * exchange. Also set EOSP to indicate this packet
1228 * ends the poll/service period.
Johannes Bergce662b442011-09-29 16:04:34 +02001229 */
Johannes Berg02f2f1a2012-02-27 12:18:30 +01001230 info->flags |= IEEE80211_TX_CTL_NO_PS_BUFFER |
Felix Fietkaud6d23de22013-06-04 12:15:42 +02001231 IEEE80211_TX_CTL_PS_RESPONSE |
Johannes Bergdeeaee12011-09-29 16:04:35 +02001232 IEEE80211_TX_STATUS_EOSP |
1233 IEEE80211_TX_CTL_REQ_TX_STATUS;
Johannes Bergce662b442011-09-29 16:04:34 +02001234
Johannes Bergb77cf4f2014-01-09 00:00:38 +01001235 if (call_driver)
1236 drv_allow_buffered_frames(local, sta, BIT(tid), 1,
1237 reason, false);
Johannes Berg40b96402011-09-29 16:04:38 +02001238
Johannes Berg89afe612013-02-13 15:39:57 +01001239 skb->dev = sdata->dev;
1240
Johannes Berg55de9082012-07-26 17:24:39 +02001241 rcu_read_lock();
1242 chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
1243 if (WARN_ON(!chanctx_conf)) {
1244 rcu_read_unlock();
1245 kfree_skb(skb);
1246 return;
1247 }
1248
Johannes Berg4bf88532012-11-09 11:39:59 +01001249 ieee80211_xmit(sdata, skb, chanctx_conf->def.chan->band);
Johannes Berg55de9082012-07-26 17:24:39 +02001250 rcu_read_unlock();
Johannes Bergce662b442011-09-29 16:04:34 +02001251}
1252
Johannes Berg0a1cb802014-01-09 11:05:31 +01001253static int find_highest_prio_tid(unsigned long tids)
1254{
1255 /* lower 3 TIDs aren't ordered perfectly */
1256 if (tids & 0xF8)
1257 return fls(tids) - 1;
1258 /* TID 0 is BE just like TID 3 */
1259 if (tids & BIT(0))
1260 return 0;
1261 return fls(tids) - 1;
1262}
1263
Johannes Berg47086fc2011-09-29 16:04:33 +02001264static void
1265ieee80211_sta_ps_deliver_response(struct sta_info *sta,
1266 int n_frames, u8 ignored_acs,
1267 enum ieee80211_frame_release_type reason)
Johannes Bergaf818582009-11-06 11:35:50 +01001268{
1269 struct ieee80211_sub_if_data *sdata = sta->sdata;
1270 struct ieee80211_local *local = sdata->local;
Johannes Berg948d8872011-09-29 16:04:29 +02001271 bool more_data = false;
1272 int ac;
Johannes Berg4049e092011-09-29 16:04:32 +02001273 unsigned long driver_release_tids = 0;
Johannes Berg47086fc2011-09-29 16:04:33 +02001274 struct sk_buff_head frames;
1275
Johannes Bergdeeaee12011-09-29 16:04:35 +02001276 /* Service or PS-Poll period starts */
Johannes Bergc2c98fd2011-09-29 16:04:36 +02001277 set_sta_flag(sta, WLAN_STA_SP);
Johannes Bergdeeaee12011-09-29 16:04:35 +02001278
Johannes Berg47086fc2011-09-29 16:04:33 +02001279 __skb_queue_head_init(&frames);
Johannes Bergaf818582009-11-06 11:35:50 +01001280
Johannes Bergf9f760b2014-01-08 17:45:07 +01001281 /* Get response frame(s) and more data bit for the last one. */
Johannes Berg948d8872011-09-29 16:04:29 +02001282 for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) {
Johannes Berg4049e092011-09-29 16:04:32 +02001283 unsigned long tids;
1284
Johannes Berg47086fc2011-09-29 16:04:33 +02001285 if (ignored_acs & BIT(ac))
Johannes Berg948d8872011-09-29 16:04:29 +02001286 continue;
1287
Johannes Berg4049e092011-09-29 16:04:32 +02001288 tids = ieee80211_tids_for_ac(ac);
1289
Johannes Bergf9f760b2014-01-08 17:45:07 +01001290 /* if we already have frames from software, then we can't also
1291 * release from hardware queues
1292 */
1293 if (skb_queue_empty(&frames))
1294 driver_release_tids |= sta->driver_buffered_tids & tids;
Johannes Berg47086fc2011-09-29 16:04:33 +02001295
Johannes Bergf9f760b2014-01-08 17:45:07 +01001296 if (driver_release_tids) {
1297 /* If the driver has data on more than one TID then
Johannes Berg4049e092011-09-29 16:04:32 +02001298 * certainly there's more data if we release just a
Johannes Bergf9f760b2014-01-08 17:45:07 +01001299 * single frame now (from a single TID). This will
1300 * only happen for PS-Poll.
Johannes Berg4049e092011-09-29 16:04:32 +02001301 */
Johannes Berg47086fc2011-09-29 16:04:33 +02001302 if (reason == IEEE80211_FRAME_RELEASE_PSPOLL &&
1303 hweight16(driver_release_tids) > 1) {
Johannes Berg4049e092011-09-29 16:04:32 +02001304 more_data = true;
1305 driver_release_tids =
Johannes Berg0a1cb802014-01-09 11:05:31 +01001306 BIT(find_highest_prio_tid(
1307 driver_release_tids));
Johannes Berg4049e092011-09-29 16:04:32 +02001308 break;
Johannes Berg948d8872011-09-29 16:04:29 +02001309 }
Johannes Bergf9f760b2014-01-08 17:45:07 +01001310 } else {
1311 struct sk_buff *skb;
1312
1313 while (n_frames > 0) {
1314 skb = skb_dequeue(&sta->tx_filtered[ac]);
1315 if (!skb) {
1316 skb = skb_dequeue(
1317 &sta->ps_tx_buf[ac]);
1318 if (skb)
1319 local->total_ps_buffered--;
1320 }
1321 if (!skb)
1322 break;
1323 n_frames--;
1324 __skb_queue_tail(&frames, skb);
1325 }
Johannes Berg948d8872011-09-29 16:04:29 +02001326 }
1327
Johannes Bergf9f760b2014-01-08 17:45:07 +01001328 /* If we have more frames buffered on this AC, then set the
1329 * more-data bit and abort the loop since we can't send more
1330 * data from other ACs before the buffered frames from this.
1331 */
Johannes Berg948d8872011-09-29 16:04:29 +02001332 if (!skb_queue_empty(&sta->tx_filtered[ac]) ||
1333 !skb_queue_empty(&sta->ps_tx_buf[ac])) {
1334 more_data = true;
1335 break;
1336 }
Johannes Bergaf818582009-11-06 11:35:50 +01001337 }
Johannes Bergaf818582009-11-06 11:35:50 +01001338
Johannes Bergf9f760b2014-01-08 17:45:07 +01001339 if (skb_queue_empty(&frames) && !driver_release_tids) {
Johannes Bergce662b442011-09-29 16:04:34 +02001340 int tid;
Johannes Berg4049e092011-09-29 16:04:32 +02001341
Johannes Bergce662b442011-09-29 16:04:34 +02001342 /*
1343 * For PS-Poll, this can only happen due to a race condition
1344 * when we set the TIM bit and the station notices it, but
1345 * before it can poll for the frame we expire it.
1346 *
1347 * For uAPSD, this is said in the standard (11.2.1.5 h):
1348 * At each unscheduled SP for a non-AP STA, the AP shall
1349 * attempt to transmit at least one MSDU or MMPDU, but no
1350 * more than the value specified in the Max SP Length field
1351 * in the QoS Capability element from delivery-enabled ACs,
1352 * that are destined for the non-AP STA.
1353 *
1354 * Since we have no other MSDU/MMPDU, transmit a QoS null frame.
1355 */
1356
1357 /* This will evaluate to 1, 3, 5 or 7. */
1358 tid = 7 - ((ffs(~ignored_acs) - 1) << 1);
1359
Johannes Bergb77cf4f2014-01-09 00:00:38 +01001360 ieee80211_send_null_response(sdata, sta, tid, reason, true);
Johannes Bergf9f760b2014-01-08 17:45:07 +01001361 } else if (!driver_release_tids) {
Johannes Berg47086fc2011-09-29 16:04:33 +02001362 struct sk_buff_head pending;
1363 struct sk_buff *skb;
Johannes Berg40b96402011-09-29 16:04:38 +02001364 int num = 0;
1365 u16 tids = 0;
Johannes Bergb77cf4f2014-01-09 00:00:38 +01001366 bool need_null = false;
Johannes Bergaf818582009-11-06 11:35:50 +01001367
Johannes Berg47086fc2011-09-29 16:04:33 +02001368 skb_queue_head_init(&pending);
Johannes Bergaf818582009-11-06 11:35:50 +01001369
Johannes Berg47086fc2011-09-29 16:04:33 +02001370 while ((skb = __skb_dequeue(&frames))) {
1371 struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1372 struct ieee80211_hdr *hdr = (void *) skb->data;
Johannes Berg40b96402011-09-29 16:04:38 +02001373 u8 *qoshdr = NULL;
1374
1375 num++;
Johannes Bergaf818582009-11-06 11:35:50 +01001376
Johannes Berg47086fc2011-09-29 16:04:33 +02001377 /*
1378 * Tell TX path to send this frame even though the
1379 * STA may still remain is PS mode after this frame
1380 * exchange.
1381 */
Felix Fietkaud6d23de22013-06-04 12:15:42 +02001382 info->flags |= IEEE80211_TX_CTL_NO_PS_BUFFER |
1383 IEEE80211_TX_CTL_PS_RESPONSE;
Johannes Bergaf818582009-11-06 11:35:50 +01001384
Johannes Berg47086fc2011-09-29 16:04:33 +02001385 /*
1386 * Use MoreData flag to indicate whether there are
1387 * more buffered frames for this STA
1388 */
Janusz.Dziedzic@tieto.com24b9c372011-11-07 09:47:47 +02001389 if (more_data || !skb_queue_empty(&frames))
Johannes Berg47086fc2011-09-29 16:04:33 +02001390 hdr->frame_control |=
1391 cpu_to_le16(IEEE80211_FCTL_MOREDATA);
Janusz.Dziedzic@tieto.com24b9c372011-11-07 09:47:47 +02001392 else
1393 hdr->frame_control &=
1394 cpu_to_le16(~IEEE80211_FCTL_MOREDATA);
Johannes Berg47086fc2011-09-29 16:04:33 +02001395
Johannes Berg40b96402011-09-29 16:04:38 +02001396 if (ieee80211_is_data_qos(hdr->frame_control) ||
1397 ieee80211_is_qos_nullfunc(hdr->frame_control))
1398 qoshdr = ieee80211_get_qos_ctl(hdr);
1399
Johannes Bergb77cf4f2014-01-09 00:00:38 +01001400 tids |= BIT(skb->priority);
1401
1402 __skb_queue_tail(&pending, skb);
1403
1404 /* end service period after last frame or add one */
1405 if (!skb_queue_empty(&frames))
1406 continue;
1407
1408 if (reason != IEEE80211_FRAME_RELEASE_UAPSD) {
1409 /* for PS-Poll, there's only one frame */
1410 info->flags |= IEEE80211_TX_STATUS_EOSP |
1411 IEEE80211_TX_CTL_REQ_TX_STATUS;
1412 break;
1413 }
1414
1415 /* For uAPSD, things are a bit more complicated. If the
1416 * last frame has a QoS header (i.e. is a QoS-data or
1417 * QoS-nulldata frame) then just set the EOSP bit there
1418 * and be done.
1419 * If the frame doesn't have a QoS header (which means
1420 * it should be a bufferable MMPDU) then we can't set
1421 * the EOSP bit in the QoS header; add a QoS-nulldata
1422 * frame to the list to send it after the MMPDU.
1423 *
1424 * Note that this code is only in the mac80211-release
1425 * code path, we assume that the driver will not buffer
1426 * anything but QoS-data frames, or if it does, will
1427 * create the QoS-nulldata frame by itself if needed.
1428 *
1429 * Cf. 802.11-2012 10.2.1.10 (c).
1430 */
1431 if (qoshdr) {
1432 *qoshdr |= IEEE80211_QOS_CTL_EOSP;
Johannes Berg47086fc2011-09-29 16:04:33 +02001433
Marco Porsch52a3f202012-03-16 15:30:26 +01001434 info->flags |= IEEE80211_TX_STATUS_EOSP |
1435 IEEE80211_TX_CTL_REQ_TX_STATUS;
Johannes Bergb77cf4f2014-01-09 00:00:38 +01001436 } else {
1437 /* The standard isn't completely clear on this
1438 * as it says the more-data bit should be set
1439 * if there are more BUs. The QoS-Null frame
1440 * we're about to send isn't buffered yet, we
1441 * only create it below, but let's pretend it
1442 * was buffered just in case some clients only
1443 * expect more-data=0 when eosp=1.
1444 */
1445 hdr->frame_control |=
1446 cpu_to_le16(IEEE80211_FCTL_MOREDATA);
1447 need_null = true;
1448 num++;
Marco Porsch52a3f202012-03-16 15:30:26 +01001449 }
Johannes Bergb77cf4f2014-01-09 00:00:38 +01001450 break;
Johannes Berg47086fc2011-09-29 16:04:33 +02001451 }
1452
Johannes Berg40b96402011-09-29 16:04:38 +02001453 drv_allow_buffered_frames(local, sta, tids, num,
1454 reason, more_data);
1455
Johannes Berg47086fc2011-09-29 16:04:33 +02001456 ieee80211_add_pending_skbs(local, &pending);
Johannes Bergaf818582009-11-06 11:35:50 +01001457
Johannes Bergb77cf4f2014-01-09 00:00:38 +01001458 if (need_null)
1459 ieee80211_send_null_response(
1460 sdata, sta, find_highest_prio_tid(tids),
1461 reason, false);
1462
Johannes Bergc868cb352011-09-29 16:04:27 +02001463 sta_info_recalc_tim(sta);
Johannes Bergaf818582009-11-06 11:35:50 +01001464 } else {
1465 /*
Johannes Berg4049e092011-09-29 16:04:32 +02001466 * We need to release a frame that is buffered somewhere in the
1467 * driver ... it'll have to handle that.
Johannes Bergf9f760b2014-01-08 17:45:07 +01001468 * Note that the driver also has to check the number of frames
1469 * on the TIDs we're releasing from - if there are more than
1470 * n_frames it has to set the more-data bit (if we didn't ask
1471 * it to set it anyway due to other buffered frames); if there
1472 * are fewer than n_frames it has to make sure to adjust that
1473 * to allow the service period to end properly.
Johannes Bergaf818582009-11-06 11:35:50 +01001474 */
Johannes Berg4049e092011-09-29 16:04:32 +02001475 drv_release_buffered_frames(local, sta, driver_release_tids,
Johannes Berg47086fc2011-09-29 16:04:33 +02001476 n_frames, reason, more_data);
Johannes Berg4049e092011-09-29 16:04:32 +02001477
1478 /*
1479 * Note that we don't recalculate the TIM bit here as it would
1480 * most likely have no effect at all unless the driver told us
Johannes Bergf9f760b2014-01-08 17:45:07 +01001481 * that the TID(s) became empty before returning here from the
Johannes Berg4049e092011-09-29 16:04:32 +02001482 * release function.
Johannes Bergf9f760b2014-01-08 17:45:07 +01001483 * Either way, however, when the driver tells us that the TID(s)
Johannes Berg4049e092011-09-29 16:04:32 +02001484 * became empty we'll do the TIM recalculation.
1485 */
Johannes Bergaf818582009-11-06 11:35:50 +01001486 }
1487}
1488
Johannes Bergaf818582009-11-06 11:35:50 +01001489void ieee80211_sta_ps_deliver_poll_response(struct sta_info *sta)
1490{
Johannes Berg47086fc2011-09-29 16:04:33 +02001491 u8 ignore_for_response = sta->sta.uapsd_queues;
Johannes Bergaf818582009-11-06 11:35:50 +01001492
Johannes Berg47086fc2011-09-29 16:04:33 +02001493 /*
1494 * If all ACs are delivery-enabled then we should reply
1495 * from any of them, if only some are enabled we reply
1496 * only from the non-enabled ones.
1497 */
1498 if (ignore_for_response == BIT(IEEE80211_NUM_ACS) - 1)
1499 ignore_for_response = 0;
1500
1501 ieee80211_sta_ps_deliver_response(sta, 1, ignore_for_response,
1502 IEEE80211_FRAME_RELEASE_PSPOLL);
1503}
1504
1505void ieee80211_sta_ps_deliver_uapsd(struct sta_info *sta)
1506{
1507 int n_frames = sta->sta.max_sp;
1508 u8 delivery_enabled = sta->sta.uapsd_queues;
1509
1510 /*
1511 * If we ever grow support for TSPEC this might happen if
1512 * the TSPEC update from hostapd comes in between a trigger
1513 * frame setting WLAN_STA_UAPSD in the RX path and this
1514 * actually getting called.
1515 */
1516 if (!delivery_enabled)
1517 return;
1518
Johannes Berg47086fc2011-09-29 16:04:33 +02001519 switch (sta->sta.max_sp) {
1520 case 1:
1521 n_frames = 2;
1522 break;
1523 case 2:
1524 n_frames = 4;
1525 break;
1526 case 3:
1527 n_frames = 6;
1528 break;
1529 case 0:
1530 /* XXX: what is a good value? */
1531 n_frames = 8;
1532 break;
Johannes Bergaf818582009-11-06 11:35:50 +01001533 }
Johannes Bergaf818582009-11-06 11:35:50 +01001534
Johannes Berg47086fc2011-09-29 16:04:33 +02001535 ieee80211_sta_ps_deliver_response(sta, n_frames, ~delivery_enabled,
1536 IEEE80211_FRAME_RELEASE_UAPSD);
Johannes Bergaf818582009-11-06 11:35:50 +01001537}
1538
1539void ieee80211_sta_block_awake(struct ieee80211_hw *hw,
1540 struct ieee80211_sta *pubsta, bool block)
1541{
1542 struct sta_info *sta = container_of(pubsta, struct sta_info, sta);
1543
Johannes Bergb5878a22010-04-07 16:48:40 +02001544 trace_api_sta_block_awake(sta->local, pubsta, block);
1545
Johannes Bergaf818582009-11-06 11:35:50 +01001546 if (block)
Johannes Bergc2c98fd2011-09-29 16:04:36 +02001547 set_sta_flag(sta, WLAN_STA_PS_DRIVER);
1548 else if (test_sta_flag(sta, WLAN_STA_PS_DRIVER))
Johannes Bergaf818582009-11-06 11:35:50 +01001549 ieee80211_queue_work(hw, &sta->drv_unblock_wk);
1550}
1551EXPORT_SYMBOL(ieee80211_sta_block_awake);
Felix Fietkaudcf55fb2011-04-17 17:45:00 +02001552
Johannes Berge9437892013-02-15 21:38:08 +01001553void ieee80211_sta_eosp(struct ieee80211_sta *pubsta)
Johannes Berg37fbd902011-09-29 16:04:39 +02001554{
1555 struct sta_info *sta = container_of(pubsta, struct sta_info, sta);
1556 struct ieee80211_local *local = sta->local;
Johannes Berg37fbd902011-09-29 16:04:39 +02001557
1558 trace_api_eosp(local, pubsta);
1559
Johannes Berge9437892013-02-15 21:38:08 +01001560 clear_sta_flag(sta, WLAN_STA_SP);
Johannes Berg37fbd902011-09-29 16:04:39 +02001561}
Johannes Berge9437892013-02-15 21:38:08 +01001562EXPORT_SYMBOL(ieee80211_sta_eosp);
Johannes Berg37fbd902011-09-29 16:04:39 +02001563
Johannes Berg042ec452011-09-29 16:04:26 +02001564void ieee80211_sta_set_buffered(struct ieee80211_sta *pubsta,
1565 u8 tid, bool buffered)
Felix Fietkaudcf55fb2011-04-17 17:45:00 +02001566{
1567 struct sta_info *sta = container_of(pubsta, struct sta_info, sta);
1568
Johannes Berg5a306f52012-11-14 23:22:21 +01001569 if (WARN_ON(tid >= IEEE80211_NUM_TIDS))
Johannes Berg042ec452011-09-29 16:04:26 +02001570 return;
1571
Johannes Berg1b000782013-12-19 10:47:48 +01001572 trace_api_sta_set_buffered(sta->local, pubsta, tid, buffered);
1573
Johannes Berg948d8872011-09-29 16:04:29 +02001574 if (buffered)
1575 set_bit(tid, &sta->driver_buffered_tids);
1576 else
1577 clear_bit(tid, &sta->driver_buffered_tids);
1578
Johannes Bergc868cb352011-09-29 16:04:27 +02001579 sta_info_recalc_tim(sta);
Felix Fietkaudcf55fb2011-04-17 17:45:00 +02001580}
Johannes Berg042ec452011-09-29 16:04:26 +02001581EXPORT_SYMBOL(ieee80211_sta_set_buffered);
Johannes Bergd9a7ddb2011-12-14 12:35:30 +01001582
Johannes Berg83d5cc02012-01-12 09:31:10 +01001583int sta_info_move_state(struct sta_info *sta,
1584 enum ieee80211_sta_state new_state)
Johannes Bergd9a7ddb2011-12-14 12:35:30 +01001585{
Johannes Berg8bf11d82011-12-15 11:17:37 +01001586 might_sleep();
Johannes Bergd9a7ddb2011-12-14 12:35:30 +01001587
1588 if (sta->sta_state == new_state)
1589 return 0;
1590
Johannes Bergf09603a2012-01-20 13:55:21 +01001591 /* check allowed transitions first */
1592
Johannes Bergd9a7ddb2011-12-14 12:35:30 +01001593 switch (new_state) {
1594 case IEEE80211_STA_NONE:
Johannes Bergf09603a2012-01-20 13:55:21 +01001595 if (sta->sta_state != IEEE80211_STA_AUTH)
Johannes Bergd9a7ddb2011-12-14 12:35:30 +01001596 return -EINVAL;
1597 break;
1598 case IEEE80211_STA_AUTH:
Johannes Bergf09603a2012-01-20 13:55:21 +01001599 if (sta->sta_state != IEEE80211_STA_NONE &&
1600 sta->sta_state != IEEE80211_STA_ASSOC)
Johannes Bergd9a7ddb2011-12-14 12:35:30 +01001601 return -EINVAL;
1602 break;
1603 case IEEE80211_STA_ASSOC:
Johannes Bergf09603a2012-01-20 13:55:21 +01001604 if (sta->sta_state != IEEE80211_STA_AUTH &&
1605 sta->sta_state != IEEE80211_STA_AUTHORIZED)
Johannes Bergd9a7ddb2011-12-14 12:35:30 +01001606 return -EINVAL;
1607 break;
1608 case IEEE80211_STA_AUTHORIZED:
Johannes Bergf09603a2012-01-20 13:55:21 +01001609 if (sta->sta_state != IEEE80211_STA_ASSOC)
Johannes Bergd9a7ddb2011-12-14 12:35:30 +01001610 return -EINVAL;
1611 break;
1612 default:
1613 WARN(1, "invalid state %d", new_state);
1614 return -EINVAL;
1615 }
1616
Johannes Bergbdcbd8e2012-06-22 11:29:50 +02001617 sta_dbg(sta->sdata, "moving STA %pM to state %d\n",
1618 sta->sta.addr, new_state);
Johannes Bergf09603a2012-01-20 13:55:21 +01001619
1620 /*
1621 * notify the driver before the actual changes so it can
1622 * fail the transition
1623 */
1624 if (test_sta_flag(sta, WLAN_STA_INSERTED)) {
1625 int err = drv_sta_state(sta->local, sta->sdata, sta,
1626 sta->sta_state, new_state);
1627 if (err)
1628 return err;
1629 }
1630
1631 /* reflect the change in all state variables */
1632
1633 switch (new_state) {
1634 case IEEE80211_STA_NONE:
1635 if (sta->sta_state == IEEE80211_STA_AUTH)
1636 clear_bit(WLAN_STA_AUTH, &sta->_flags);
1637 break;
1638 case IEEE80211_STA_AUTH:
1639 if (sta->sta_state == IEEE80211_STA_NONE)
1640 set_bit(WLAN_STA_AUTH, &sta->_flags);
1641 else if (sta->sta_state == IEEE80211_STA_ASSOC)
1642 clear_bit(WLAN_STA_ASSOC, &sta->_flags);
1643 break;
1644 case IEEE80211_STA_ASSOC:
1645 if (sta->sta_state == IEEE80211_STA_AUTH) {
1646 set_bit(WLAN_STA_ASSOC, &sta->_flags);
1647 } else if (sta->sta_state == IEEE80211_STA_AUTHORIZED) {
Felix Fietkau7e3ed022012-04-23 19:49:03 +02001648 if (sta->sdata->vif.type == NL80211_IFTYPE_AP ||
1649 (sta->sdata->vif.type == NL80211_IFTYPE_AP_VLAN &&
1650 !sta->sdata->u.vlan.sta))
1651 atomic_dec(&sta->sdata->bss->num_mcast_sta);
Johannes Bergf09603a2012-01-20 13:55:21 +01001652 clear_bit(WLAN_STA_AUTHORIZED, &sta->_flags);
1653 }
1654 break;
1655 case IEEE80211_STA_AUTHORIZED:
1656 if (sta->sta_state == IEEE80211_STA_ASSOC) {
Felix Fietkau7e3ed022012-04-23 19:49:03 +02001657 if (sta->sdata->vif.type == NL80211_IFTYPE_AP ||
1658 (sta->sdata->vif.type == NL80211_IFTYPE_AP_VLAN &&
1659 !sta->sdata->u.vlan.sta))
1660 atomic_inc(&sta->sdata->bss->num_mcast_sta);
Johannes Bergf09603a2012-01-20 13:55:21 +01001661 set_bit(WLAN_STA_AUTHORIZED, &sta->_flags);
1662 }
1663 break;
1664 default:
1665 break;
1666 }
1667
Johannes Bergd9a7ddb2011-12-14 12:35:30 +01001668 sta->sta_state = new_state;
1669
1670 return 0;
1671}
Emmanuel Grumbach687da132013-10-01 16:45:43 +03001672
1673u8 sta_info_tx_streams(struct sta_info *sta)
1674{
1675 struct ieee80211_sta_ht_cap *ht_cap = &sta->sta.ht_cap;
1676 u8 rx_streams;
1677
1678 if (!sta->sta.ht_cap.ht_supported)
1679 return 1;
1680
1681 if (sta->sta.vht_cap.vht_supported) {
1682 int i;
1683 u16 tx_mcs_map =
1684 le16_to_cpu(sta->sta.vht_cap.vht_mcs.tx_mcs_map);
1685
1686 for (i = 7; i >= 0; i--)
1687 if ((tx_mcs_map & (0x3 << (i * 2))) !=
1688 IEEE80211_VHT_MCS_NOT_SUPPORTED)
1689 return i + 1;
1690 }
1691
1692 if (ht_cap->mcs.rx_mask[3])
1693 rx_streams = 4;
1694 else if (ht_cap->mcs.rx_mask[2])
1695 rx_streams = 3;
1696 else if (ht_cap->mcs.rx_mask[1])
1697 rx_streams = 2;
1698 else
1699 rx_streams = 1;
1700
1701 if (!(ht_cap->mcs.tx_params & IEEE80211_HT_MCS_TX_RX_DIFF))
1702 return rx_streams;
1703
1704 return ((ht_cap->mcs.tx_params & IEEE80211_HT_MCS_TX_MAX_STREAMS_MASK)
1705 >> IEEE80211_HT_MCS_TX_MAX_STREAMS_SHIFT) + 1;
1706}