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