mac80211: add fast-rx path
[cascardo/linux.git] / net / mac80211 / sta_info.h
1 /*
2  * Copyright 2002-2005, Devicescape Software, Inc.
3  * Copyright 2013-2014  Intel Mobile Communications GmbH
4  * Copyright(c) 2015-2016 Intel Deutschland GmbH
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License version 2 as
8  * published by the Free Software Foundation.
9  */
10
11 #ifndef STA_INFO_H
12 #define STA_INFO_H
13
14 #include <linux/list.h>
15 #include <linux/types.h>
16 #include <linux/if_ether.h>
17 #include <linux/workqueue.h>
18 #include <linux/average.h>
19 #include <linux/etherdevice.h>
20 #include <linux/rhashtable.h>
21 #include <linux/u64_stats_sync.h>
22 #include "key.h"
23
24 /**
25  * enum ieee80211_sta_info_flags - Stations flags
26  *
27  * These flags are used with &struct sta_info's @flags member, but
28  * only indirectly with set_sta_flag() and friends.
29  *
30  * @WLAN_STA_AUTH: Station is authenticated.
31  * @WLAN_STA_ASSOC: Station is associated.
32  * @WLAN_STA_PS_STA: Station is in power-save mode
33  * @WLAN_STA_AUTHORIZED: Station is authorized to send/receive traffic.
34  *      This bit is always checked so needs to be enabled for all stations
35  *      when virtual port control is not in use.
36  * @WLAN_STA_SHORT_PREAMBLE: Station is capable of receiving short-preamble
37  *      frames.
38  * @WLAN_STA_WDS: Station is one of our WDS peers.
39  * @WLAN_STA_CLEAR_PS_FILT: Clear PS filter in hardware (using the
40  *      IEEE80211_TX_CTL_CLEAR_PS_FILT control flag) when the next
41  *      frame to this station is transmitted.
42  * @WLAN_STA_MFP: Management frame protection is used with this STA.
43  * @WLAN_STA_BLOCK_BA: Used to deny ADDBA requests (both TX and RX)
44  *      during suspend/resume and station removal.
45  * @WLAN_STA_PS_DRIVER: driver requires keeping this station in
46  *      power-save mode logically to flush frames that might still
47  *      be in the queues
48  * @WLAN_STA_PSPOLL: Station sent PS-poll while driver was keeping
49  *      station in power-save mode, reply when the driver unblocks.
50  * @WLAN_STA_TDLS_PEER: Station is a TDLS peer.
51  * @WLAN_STA_TDLS_PEER_AUTH: This TDLS peer is authorized to send direct
52  *      packets. This means the link is enabled.
53  * @WLAN_STA_TDLS_INITIATOR: We are the initiator of the TDLS link with this
54  *      station.
55  * @WLAN_STA_TDLS_CHAN_SWITCH: This TDLS peer supports TDLS channel-switching
56  * @WLAN_STA_TDLS_OFF_CHANNEL: The local STA is currently off-channel with this
57  *      TDLS peer
58  * @WLAN_STA_TDLS_WIDER_BW: This TDLS peer supports working on a wider bw on
59  *      the BSS base channel.
60  * @WLAN_STA_UAPSD: Station requested unscheduled SP while driver was
61  *      keeping station in power-save mode, reply when the driver
62  *      unblocks the station.
63  * @WLAN_STA_SP: Station is in a service period, so don't try to
64  *      reply to other uAPSD trigger frames or PS-Poll.
65  * @WLAN_STA_4ADDR_EVENT: 4-addr event was already sent for this frame.
66  * @WLAN_STA_INSERTED: This station is inserted into the hash table.
67  * @WLAN_STA_RATE_CONTROL: rate control was initialized for this station.
68  * @WLAN_STA_TOFFSET_KNOWN: toffset calculated for this station is valid.
69  * @WLAN_STA_MPSP_OWNER: local STA is owner of a mesh Peer Service Period.
70  * @WLAN_STA_MPSP_RECIPIENT: local STA is recipient of a MPSP.
71  * @WLAN_STA_PS_DELIVER: station woke up, but we're still blocking TX
72  *      until pending frames are delivered
73  *
74  * @NUM_WLAN_STA_FLAGS: number of defined flags
75  */
76 enum ieee80211_sta_info_flags {
77         WLAN_STA_AUTH,
78         WLAN_STA_ASSOC,
79         WLAN_STA_PS_STA,
80         WLAN_STA_AUTHORIZED,
81         WLAN_STA_SHORT_PREAMBLE,
82         WLAN_STA_WDS,
83         WLAN_STA_CLEAR_PS_FILT,
84         WLAN_STA_MFP,
85         WLAN_STA_BLOCK_BA,
86         WLAN_STA_PS_DRIVER,
87         WLAN_STA_PSPOLL,
88         WLAN_STA_TDLS_PEER,
89         WLAN_STA_TDLS_PEER_AUTH,
90         WLAN_STA_TDLS_INITIATOR,
91         WLAN_STA_TDLS_CHAN_SWITCH,
92         WLAN_STA_TDLS_OFF_CHANNEL,
93         WLAN_STA_TDLS_WIDER_BW,
94         WLAN_STA_UAPSD,
95         WLAN_STA_SP,
96         WLAN_STA_4ADDR_EVENT,
97         WLAN_STA_INSERTED,
98         WLAN_STA_RATE_CONTROL,
99         WLAN_STA_TOFFSET_KNOWN,
100         WLAN_STA_MPSP_OWNER,
101         WLAN_STA_MPSP_RECIPIENT,
102         WLAN_STA_PS_DELIVER,
103
104         NUM_WLAN_STA_FLAGS,
105 };
106
107 #define ADDBA_RESP_INTERVAL HZ
108 #define HT_AGG_MAX_RETRIES              15
109 #define HT_AGG_BURST_RETRIES            3
110 #define HT_AGG_RETRIES_PERIOD           (15 * HZ)
111
112 #define HT_AGG_STATE_DRV_READY          0
113 #define HT_AGG_STATE_RESPONSE_RECEIVED  1
114 #define HT_AGG_STATE_OPERATIONAL        2
115 #define HT_AGG_STATE_STOPPING           3
116 #define HT_AGG_STATE_WANT_START         4
117 #define HT_AGG_STATE_WANT_STOP          5
118
119 enum ieee80211_agg_stop_reason {
120         AGG_STOP_DECLINED,
121         AGG_STOP_LOCAL_REQUEST,
122         AGG_STOP_PEER_REQUEST,
123         AGG_STOP_DESTROY_STA,
124 };
125
126 /**
127  * struct tid_ampdu_tx - TID aggregation information (Tx).
128  *
129  * @rcu_head: rcu head for freeing structure
130  * @session_timer: check if we keep Tx-ing on the TID (by timeout value)
131  * @addba_resp_timer: timer for peer's response to addba request
132  * @pending: pending frames queue -- use sta's spinlock to protect
133  * @dialog_token: dialog token for aggregation session
134  * @timeout: session timeout value to be filled in ADDBA requests
135  * @state: session state (see above)
136  * @last_tx: jiffies of last tx activity
137  * @stop_initiator: initiator of a session stop
138  * @tx_stop: TX DelBA frame when stopping
139  * @buf_size: reorder buffer size at receiver
140  * @failed_bar_ssn: ssn of the last failed BAR tx attempt
141  * @bar_pending: BAR needs to be re-sent
142  * @amsdu: support A-MSDU withing A-MDPU
143  *
144  * This structure's lifetime is managed by RCU, assignments to
145  * the array holding it must hold the aggregation mutex.
146  *
147  * The TX path can access it under RCU lock-free if, and
148  * only if, the state has the flag %HT_AGG_STATE_OPERATIONAL
149  * set. Otherwise, the TX path must also acquire the spinlock
150  * and re-check the state, see comments in the tx code
151  * touching it.
152  */
153 struct tid_ampdu_tx {
154         struct rcu_head rcu_head;
155         struct timer_list session_timer;
156         struct timer_list addba_resp_timer;
157         struct sk_buff_head pending;
158         unsigned long state;
159         unsigned long last_tx;
160         u16 timeout;
161         u8 dialog_token;
162         u8 stop_initiator;
163         bool tx_stop;
164         u8 buf_size;
165
166         u16 failed_bar_ssn;
167         bool bar_pending;
168         bool amsdu;
169 };
170
171 /**
172  * struct tid_ampdu_rx - TID aggregation information (Rx).
173  *
174  * @reorder_buf: buffer to reorder incoming aggregated MPDUs. An MPDU may be an
175  *      A-MSDU with individually reported subframes.
176  * @reorder_buf_filtered: bitmap indicating where there are filtered frames in
177  *      the reorder buffer that should be ignored when releasing frames
178  * @reorder_time: jiffies when skb was added
179  * @session_timer: check if peer keeps Tx-ing on the TID (by timeout value)
180  * @reorder_timer: releases expired frames from the reorder buffer.
181  * @last_rx: jiffies of last rx activity
182  * @head_seq_num: head sequence number in reordering buffer.
183  * @stored_mpdu_num: number of MPDUs in reordering buffer
184  * @ssn: Starting Sequence Number expected to be aggregated.
185  * @buf_size: buffer size for incoming A-MPDUs
186  * @timeout: reset timer value (in TUs).
187  * @dialog_token: dialog token for aggregation session
188  * @rcu_head: RCU head used for freeing this struct
189  * @reorder_lock: serializes access to reorder buffer, see below.
190  * @auto_seq: used for offloaded BA sessions to automatically pick head_seq_and
191  *      and ssn.
192  * @removed: this session is removed (but might have been found due to RCU)
193  *
194  * This structure's lifetime is managed by RCU, assignments to
195  * the array holding it must hold the aggregation mutex.
196  *
197  * The @reorder_lock is used to protect the members of this
198  * struct, except for @timeout, @buf_size and @dialog_token,
199  * which are constant across the lifetime of the struct (the
200  * dialog token being used only for debugging).
201  */
202 struct tid_ampdu_rx {
203         struct rcu_head rcu_head;
204         spinlock_t reorder_lock;
205         u64 reorder_buf_filtered;
206         struct sk_buff_head *reorder_buf;
207         unsigned long *reorder_time;
208         struct timer_list session_timer;
209         struct timer_list reorder_timer;
210         unsigned long last_rx;
211         u16 head_seq_num;
212         u16 stored_mpdu_num;
213         u16 ssn;
214         u16 buf_size;
215         u16 timeout;
216         u8 dialog_token;
217         bool auto_seq;
218         bool removed;
219 };
220
221 /**
222  * struct sta_ampdu_mlme - STA aggregation information.
223  *
224  * @mtx: mutex to protect all TX data (except non-NULL assignments
225  *      to tid_tx[idx], which are protected by the sta spinlock)
226  *      tid_start_tx is also protected by sta->lock.
227  * @tid_rx: aggregation info for Rx per TID -- RCU protected
228  * @tid_rx_timer_expired: bitmap indicating on which TIDs the
229  *      RX timer expired until the work for it runs
230  * @tid_rx_stop_requested:  bitmap indicating which BA sessions per TID the
231  *      driver requested to close until the work for it runs
232  * @agg_session_valid: bitmap indicating which TID has a rx BA session open on
233  * @work: work struct for starting/stopping aggregation
234  * @tid_tx: aggregation info for Tx per TID
235  * @tid_start_tx: sessions where start was requested
236  * @last_addba_req_time: timestamp of the last addBA request.
237  * @addba_req_num: number of times addBA request has been sent.
238  * @dialog_token_allocator: dialog token enumerator for each new session;
239  */
240 struct sta_ampdu_mlme {
241         struct mutex mtx;
242         /* rx */
243         struct tid_ampdu_rx __rcu *tid_rx[IEEE80211_NUM_TIDS];
244         unsigned long tid_rx_timer_expired[BITS_TO_LONGS(IEEE80211_NUM_TIDS)];
245         unsigned long tid_rx_stop_requested[BITS_TO_LONGS(IEEE80211_NUM_TIDS)];
246         unsigned long agg_session_valid[BITS_TO_LONGS(IEEE80211_NUM_TIDS)];
247         /* tx */
248         struct work_struct work;
249         struct tid_ampdu_tx __rcu *tid_tx[IEEE80211_NUM_TIDS];
250         struct tid_ampdu_tx *tid_start_tx[IEEE80211_NUM_TIDS];
251         unsigned long last_addba_req_time[IEEE80211_NUM_TIDS];
252         u8 addba_req_num[IEEE80211_NUM_TIDS];
253         u8 dialog_token_allocator;
254 };
255
256
257 /* Value to indicate no TID reservation */
258 #define IEEE80211_TID_UNRESERVED        0xff
259
260 #define IEEE80211_FAST_XMIT_MAX_IV      18
261
262 /**
263  * struct ieee80211_fast_tx - TX fastpath information
264  * @key: key to use for hw crypto
265  * @hdr: the 802.11 header to put with the frame
266  * @hdr_len: actual 802.11 header length
267  * @sa_offs: offset of the SA
268  * @da_offs: offset of the DA
269  * @pn_offs: offset where to put PN for crypto (or 0 if not needed)
270  * @band: band this will be transmitted on, for tx_info
271  * @rcu_head: RCU head to free this struct
272  *
273  * This struct is small enough so that the common case (maximum crypto
274  * header length of 8 like for CCMP/GCMP) fits into a single 64-byte
275  * cache line.
276  */
277 struct ieee80211_fast_tx {
278         struct ieee80211_key *key;
279         u8 hdr_len;
280         u8 sa_offs, da_offs, pn_offs;
281         u8 band;
282         u8 hdr[30 + 2 + IEEE80211_FAST_XMIT_MAX_IV +
283                sizeof(rfc1042_header)];
284
285         struct rcu_head rcu_head;
286 };
287
288 /**
289  * struct ieee80211_fast_rx - RX fastpath information
290  * @dev: netdevice for reporting the SKB
291  * @vif_type: (P2P-less) interface type of the original sdata (sdata->vif.type)
292  * @vif_addr: interface address
293  * @rfc1042_hdr: copy of the RFC 1042 SNAP header (to have in cache)
294  * @control_port_protocol: control port protocol copied from sdata
295  * @expected_ds_bits: from/to DS bits expected
296  * @icv_len: length of the MIC if present
297  * @key: bool indicating encryption is expected (key is set)
298  * @sta_notify: notify the MLME code (once)
299  * @internal_forward: forward froms internally on AP/VLAN type interfaces
300  * @da_offs: offset of the DA in the header (for header conversion)
301  * @sa_offs: offset of the SA in the header (for header conversion)
302  * @rcu_head: RCU head for freeing this structure
303  */
304 struct ieee80211_fast_rx {
305         struct net_device *dev;
306         enum nl80211_iftype vif_type;
307         u8 vif_addr[ETH_ALEN] __aligned(2);
308         u8 rfc1042_hdr[6] __aligned(2);
309         __be16 control_port_protocol;
310         __le16 expected_ds_bits;
311         u8 icv_len;
312         u8 key:1,
313            sta_notify:1,
314            internal_forward:1;
315         u8 da_offs, sa_offs;
316
317         struct rcu_head rcu_head;
318 };
319
320 /**
321  * struct mesh_sta - mesh STA information
322  * @plink_lock: serialize access to plink fields
323  * @llid: Local link ID
324  * @plid: Peer link ID
325  * @aid: local aid supplied by peer
326  * @reason: Cancel reason on PLINK_HOLDING state
327  * @plink_retries: Retries in establishment
328  * @plink_state: peer link state
329  * @plink_timeout: timeout of peer link
330  * @plink_timer: peer link watch timer
331  * @t_offset: timing offset relative to this host
332  * @t_offset_setpoint: reference timing offset of this sta to be used when
333  *      calculating clockdrift
334  * @local_pm: local link-specific power save mode
335  * @peer_pm: peer-specific power save mode towards local STA
336  * @nonpeer_pm: STA power save mode towards non-peer neighbors
337  * @processed_beacon: set to true after peer rates and capabilities are
338  *      processed
339  * @fail_avg: moving percentage of failed MSDUs
340  */
341 struct mesh_sta {
342         struct timer_list plink_timer;
343
344         s64 t_offset;
345         s64 t_offset_setpoint;
346
347         spinlock_t plink_lock;
348         u16 llid;
349         u16 plid;
350         u16 aid;
351         u16 reason;
352         u8 plink_retries;
353
354         bool processed_beacon;
355
356         enum nl80211_plink_state plink_state;
357         u32 plink_timeout;
358
359         /* mesh power save */
360         enum nl80211_mesh_power_mode local_pm;
361         enum nl80211_mesh_power_mode peer_pm;
362         enum nl80211_mesh_power_mode nonpeer_pm;
363
364         /* moving percentage of failed MSDUs */
365         unsigned int fail_avg;
366 };
367
368 DECLARE_EWMA(signal, 1024, 8)
369
370 /**
371  * struct sta_info - STA information
372  *
373  * This structure collects information about a station that
374  * mac80211 is communicating with.
375  *
376  * @list: global linked list entry
377  * @free_list: list entry for keeping track of stations to free
378  * @hash_node: hash node for rhashtable
379  * @addr: station's MAC address - duplicated from public part to
380  *      let the hash table work with just a single cacheline
381  * @local: pointer to the global information
382  * @sdata: virtual interface this station belongs to
383  * @ptk: peer keys negotiated with this station, if any
384  * @ptk_idx: last installed peer key index
385  * @gtk: group keys negotiated with this station, if any
386  * @rate_ctrl: rate control algorithm reference
387  * @rate_ctrl_lock: spinlock used to protect rate control data
388  *      (data inside the algorithm, so serializes calls there)
389  * @rate_ctrl_priv: rate control private per-STA pointer
390  * @lock: used for locking all fields that require locking, see comments
391  *      in the header file.
392  * @drv_deliver_wk: used for delivering frames after driver PS unblocking
393  * @listen_interval: listen interval of this station, when we're acting as AP
394  * @_flags: STA flags, see &enum ieee80211_sta_info_flags, do not use directly
395  * @ps_lock: used for powersave (when mac80211 is the AP) related locking
396  * @ps_tx_buf: buffers (per AC) of frames to transmit to this station
397  *      when it leaves power saving state or polls
398  * @tx_filtered: buffers (per AC) of frames we already tried to
399  *      transmit but were filtered by hardware due to STA having
400  *      entered power saving state, these are also delivered to
401  *      the station when it leaves powersave or polls for frames
402  * @driver_buffered_tids: bitmap of TIDs the driver has data buffered on
403  * @txq_buffered_tids: bitmap of TIDs that mac80211 has txq data buffered on
404  * @last_connected: time (in seconds) when a station got connected
405  * @last_seq_ctrl: last received seq/frag number from this STA (per TID
406  *      plus one for non-QoS frames)
407  * @tid_seq: per-TID sequence numbers for sending to this STA
408  * @ampdu_mlme: A-MPDU state machine state
409  * @timer_to_tid: identity mapping to ID timers
410  * @mesh: mesh STA information
411  * @debugfs_dir: debug filesystem directory dentry
412  * @dead: set to true when sta is unlinked
413  * @removed: set to true when sta is being removed from sta_list
414  * @uploaded: set to true when sta is uploaded to the driver
415  * @sta: station information we share with the driver
416  * @sta_state: duplicates information about station state (for debug)
417  * @beacon_loss_count: number of times beacon loss has triggered
418  * @rcu_head: RCU head used for freeing this station struct
419  * @cur_max_bandwidth: maximum bandwidth to use for TX to the station,
420  *      taken from HT/VHT capabilities or VHT operating mode notification
421  * @known_smps_mode: the smps_mode the client thinks we are in. Relevant for
422  *      AP only.
423  * @cipher_scheme: optional cipher scheme for this station
424  * @reserved_tid: reserved TID (if any, otherwise IEEE80211_TID_UNRESERVED)
425  * @fast_tx: TX fastpath information
426  * @fast_rx: RX fastpath information
427  * @tdls_chandef: a TDLS peer can have a wider chandef that is compatible to
428  *      the BSS one.
429  * @tx_stats: TX statistics
430  * @rx_stats: RX statistics
431  * @status_stats: TX status statistics
432  */
433 struct sta_info {
434         /* General information, mostly static */
435         struct list_head list, free_list;
436         struct rcu_head rcu_head;
437         struct rhash_head hash_node;
438         u8 addr[ETH_ALEN];
439         struct ieee80211_local *local;
440         struct ieee80211_sub_if_data *sdata;
441         struct ieee80211_key __rcu *gtk[NUM_DEFAULT_KEYS + NUM_DEFAULT_MGMT_KEYS];
442         struct ieee80211_key __rcu *ptk[NUM_DEFAULT_KEYS];
443         u8 ptk_idx;
444         struct rate_control_ref *rate_ctrl;
445         void *rate_ctrl_priv;
446         spinlock_t rate_ctrl_lock;
447         spinlock_t lock;
448
449         struct ieee80211_fast_tx __rcu *fast_tx;
450         struct ieee80211_fast_rx __rcu *fast_rx;
451
452 #ifdef CONFIG_MAC80211_MESH
453         struct mesh_sta *mesh;
454 #endif
455
456         struct work_struct drv_deliver_wk;
457
458         u16 listen_interval;
459
460         bool dead;
461         bool removed;
462
463         bool uploaded;
464
465         enum ieee80211_sta_state sta_state;
466
467         /* use the accessors defined below */
468         unsigned long _flags;
469
470         /* STA powersave lock and frame queues */
471         spinlock_t ps_lock;
472         struct sk_buff_head ps_tx_buf[IEEE80211_NUM_ACS];
473         struct sk_buff_head tx_filtered[IEEE80211_NUM_ACS];
474         unsigned long driver_buffered_tids;
475         unsigned long txq_buffered_tids;
476
477         long last_connected;
478
479         /* Updated from RX path only, no locking requirements */
480         struct {
481                 unsigned long packets;
482                 unsigned long last_rx;
483                 unsigned long num_duplicates;
484                 unsigned long fragments;
485                 unsigned long dropped;
486                 int last_signal;
487                 u8 chains;
488                 s8 chain_signal_last[IEEE80211_MAX_CHAINS];
489                 u16 last_rate;
490
491                 struct u64_stats_sync syncp;
492                 u64 bytes;
493                 u64 msdu[IEEE80211_NUM_TIDS + 1];
494         } rx_stats;
495         struct {
496                 struct ewma_signal signal;
497                 struct ewma_signal chain_signal[IEEE80211_MAX_CHAINS];
498         } rx_stats_avg;
499
500         /* Plus 1 for non-QoS frames */
501         __le16 last_seq_ctrl[IEEE80211_NUM_TIDS + 1];
502
503         /* Updated from TX status path only, no locking requirements */
504         struct {
505                 unsigned long filtered;
506                 unsigned long retry_failed, retry_count;
507                 unsigned int lost_packets;
508                 unsigned long last_tdls_pkt_time;
509                 u64 msdu_retries[IEEE80211_NUM_TIDS + 1];
510                 u64 msdu_failed[IEEE80211_NUM_TIDS + 1];
511                 unsigned long last_ack;
512         } status_stats;
513
514         /* Updated from TX path only, no locking requirements */
515         struct {
516                 u64 packets[IEEE80211_NUM_ACS];
517                 u64 bytes[IEEE80211_NUM_ACS];
518                 struct ieee80211_tx_rate last_rate;
519                 u64 msdu[IEEE80211_NUM_TIDS + 1];
520         } tx_stats;
521         u16 tid_seq[IEEE80211_QOS_CTL_TID_MASK + 1];
522
523         /*
524          * Aggregation information, locked with lock.
525          */
526         struct sta_ampdu_mlme ampdu_mlme;
527         u8 timer_to_tid[IEEE80211_NUM_TIDS];
528
529 #ifdef CONFIG_MAC80211_DEBUGFS
530         struct dentry *debugfs_dir;
531 #endif
532
533         enum ieee80211_sta_rx_bandwidth cur_max_bandwidth;
534
535         enum ieee80211_smps_mode known_smps_mode;
536         const struct ieee80211_cipher_scheme *cipher_scheme;
537
538         u8 reserved_tid;
539
540         struct cfg80211_chan_def tdls_chandef;
541
542         /* keep last! */
543         struct ieee80211_sta sta;
544 };
545
546 static inline enum nl80211_plink_state sta_plink_state(struct sta_info *sta)
547 {
548 #ifdef CONFIG_MAC80211_MESH
549         return sta->mesh->plink_state;
550 #endif
551         return NL80211_PLINK_LISTEN;
552 }
553
554 static inline void set_sta_flag(struct sta_info *sta,
555                                 enum ieee80211_sta_info_flags flag)
556 {
557         WARN_ON(flag == WLAN_STA_AUTH ||
558                 flag == WLAN_STA_ASSOC ||
559                 flag == WLAN_STA_AUTHORIZED);
560         set_bit(flag, &sta->_flags);
561 }
562
563 static inline void clear_sta_flag(struct sta_info *sta,
564                                   enum ieee80211_sta_info_flags flag)
565 {
566         WARN_ON(flag == WLAN_STA_AUTH ||
567                 flag == WLAN_STA_ASSOC ||
568                 flag == WLAN_STA_AUTHORIZED);
569         clear_bit(flag, &sta->_flags);
570 }
571
572 static inline int test_sta_flag(struct sta_info *sta,
573                                 enum ieee80211_sta_info_flags flag)
574 {
575         return test_bit(flag, &sta->_flags);
576 }
577
578 static inline int test_and_clear_sta_flag(struct sta_info *sta,
579                                           enum ieee80211_sta_info_flags flag)
580 {
581         WARN_ON(flag == WLAN_STA_AUTH ||
582                 flag == WLAN_STA_ASSOC ||
583                 flag == WLAN_STA_AUTHORIZED);
584         return test_and_clear_bit(flag, &sta->_flags);
585 }
586
587 static inline int test_and_set_sta_flag(struct sta_info *sta,
588                                         enum ieee80211_sta_info_flags flag)
589 {
590         WARN_ON(flag == WLAN_STA_AUTH ||
591                 flag == WLAN_STA_ASSOC ||
592                 flag == WLAN_STA_AUTHORIZED);
593         return test_and_set_bit(flag, &sta->_flags);
594 }
595
596 int sta_info_move_state(struct sta_info *sta,
597                         enum ieee80211_sta_state new_state);
598
599 static inline void sta_info_pre_move_state(struct sta_info *sta,
600                                            enum ieee80211_sta_state new_state)
601 {
602         int ret;
603
604         WARN_ON_ONCE(test_sta_flag(sta, WLAN_STA_INSERTED));
605
606         ret = sta_info_move_state(sta, new_state);
607         WARN_ON_ONCE(ret);
608 }
609
610
611 void ieee80211_assign_tid_tx(struct sta_info *sta, int tid,
612                              struct tid_ampdu_tx *tid_tx);
613
614 static inline struct tid_ampdu_tx *
615 rcu_dereference_protected_tid_tx(struct sta_info *sta, int tid)
616 {
617         return rcu_dereference_protected(sta->ampdu_mlme.tid_tx[tid],
618                                          lockdep_is_held(&sta->lock) ||
619                                          lockdep_is_held(&sta->ampdu_mlme.mtx));
620 }
621
622 /* Maximum number of frames to buffer per power saving station per AC */
623 #define STA_MAX_TX_BUFFER       64
624
625 /* Minimum buffered frame expiry time. If STA uses listen interval that is
626  * smaller than this value, the minimum value here is used instead. */
627 #define STA_TX_BUFFER_EXPIRE (10 * HZ)
628
629 /* How often station data is cleaned up (e.g., expiration of buffered frames)
630  */
631 #define STA_INFO_CLEANUP_INTERVAL (10 * HZ)
632
633 /*
634  * Get a STA info, must be under RCU read lock.
635  */
636 struct sta_info *sta_info_get(struct ieee80211_sub_if_data *sdata,
637                               const u8 *addr);
638
639 struct sta_info *sta_info_get_bss(struct ieee80211_sub_if_data *sdata,
640                                   const u8 *addr);
641
642 u32 sta_addr_hash(const void *key, u32 length, u32 seed);
643
644 #define _sta_bucket_idx(_tbl, _a)                                       \
645         rht_bucket_index(_tbl, sta_addr_hash(_a, ETH_ALEN, (_tbl)->hash_rnd))
646
647 #define for_each_sta_info(local, tbl, _addr, _sta, _tmp)                \
648         rht_for_each_entry_rcu(_sta, _tmp, tbl,                         \
649                                _sta_bucket_idx(tbl, _addr),             \
650                                hash_node)                               \
651         /* compare address and run code only if it matches */           \
652         if (ether_addr_equal(_sta->addr, (_addr)))
653
654 /*
655  * Get STA info by index, BROKEN!
656  */
657 struct sta_info *sta_info_get_by_idx(struct ieee80211_sub_if_data *sdata,
658                                      int idx);
659 /*
660  * Create a new STA info, caller owns returned structure
661  * until sta_info_insert().
662  */
663 struct sta_info *sta_info_alloc(struct ieee80211_sub_if_data *sdata,
664                                 const u8 *addr, gfp_t gfp);
665
666 void sta_info_free(struct ieee80211_local *local, struct sta_info *sta);
667
668 /*
669  * Insert STA info into hash table/list, returns zero or a
670  * -EEXIST if (if the same MAC address is already present).
671  *
672  * Calling the non-rcu version makes the caller relinquish,
673  * the _rcu version calls read_lock_rcu() and must be called
674  * without it held.
675  */
676 int sta_info_insert(struct sta_info *sta);
677 int sta_info_insert_rcu(struct sta_info *sta) __acquires(RCU);
678
679 int __must_check __sta_info_destroy(struct sta_info *sta);
680 int sta_info_destroy_addr(struct ieee80211_sub_if_data *sdata,
681                           const u8 *addr);
682 int sta_info_destroy_addr_bss(struct ieee80211_sub_if_data *sdata,
683                               const u8 *addr);
684
685 void sta_info_recalc_tim(struct sta_info *sta);
686
687 int sta_info_init(struct ieee80211_local *local);
688 void sta_info_stop(struct ieee80211_local *local);
689
690 /**
691  * sta_info_flush - flush matching STA entries from the STA table
692  *
693  * Returns the number of removed STA entries.
694  *
695  * @sdata: sdata to remove all stations from
696  * @vlans: if the given interface is an AP interface, also flush VLANs
697  */
698 int __sta_info_flush(struct ieee80211_sub_if_data *sdata, bool vlans);
699
700 static inline int sta_info_flush(struct ieee80211_sub_if_data *sdata)
701 {
702         return __sta_info_flush(sdata, false);
703 }
704
705 void sta_set_rate_info_tx(struct sta_info *sta,
706                           const struct ieee80211_tx_rate *rate,
707                           struct rate_info *rinfo);
708 void sta_set_sinfo(struct sta_info *sta, struct station_info *sinfo);
709
710 void ieee80211_sta_expire(struct ieee80211_sub_if_data *sdata,
711                           unsigned long exp_time);
712 u8 sta_info_tx_streams(struct sta_info *sta);
713
714 void ieee80211_sta_ps_deliver_wakeup(struct sta_info *sta);
715 void ieee80211_sta_ps_deliver_poll_response(struct sta_info *sta);
716 void ieee80211_sta_ps_deliver_uapsd(struct sta_info *sta);
717
718 unsigned long ieee80211_sta_last_active(struct sta_info *sta);
719
720 #define STA_STATS_RATE_INVALID          0
721 #define STA_STATS_RATE_VHT              0x8000
722 #define STA_STATS_RATE_HT               0x4000
723 #define STA_STATS_RATE_LEGACY           0x2000
724 #define STA_STATS_RATE_SGI              0x1000
725 #define STA_STATS_RATE_BW_SHIFT         9
726 #define STA_STATS_RATE_BW_MASK          (0x7 << STA_STATS_RATE_BW_SHIFT)
727
728 static inline u16 sta_stats_encode_rate(struct ieee80211_rx_status *s)
729 {
730         u16 r = s->rate_idx;
731
732         if (s->vht_flag & RX_VHT_FLAG_80MHZ)
733                 r |= RATE_INFO_BW_80 << STA_STATS_RATE_BW_SHIFT;
734         else if (s->vht_flag & RX_VHT_FLAG_160MHZ)
735                 r |= RATE_INFO_BW_160 << STA_STATS_RATE_BW_SHIFT;
736         else if (s->flag & RX_FLAG_40MHZ)
737                 r |= RATE_INFO_BW_40 << STA_STATS_RATE_BW_SHIFT;
738         else if (s->flag & RX_FLAG_10MHZ)
739                 r |= RATE_INFO_BW_10 << STA_STATS_RATE_BW_SHIFT;
740         else if (s->flag & RX_FLAG_5MHZ)
741                 r |= RATE_INFO_BW_5 << STA_STATS_RATE_BW_SHIFT;
742         else
743                 r |= RATE_INFO_BW_20 << STA_STATS_RATE_BW_SHIFT;
744
745         if (s->flag & RX_FLAG_SHORT_GI)
746                 r |= STA_STATS_RATE_SGI;
747
748         if (s->flag & RX_FLAG_VHT)
749                 r |= STA_STATS_RATE_VHT | (s->vht_nss << 4);
750         else if (s->flag & RX_FLAG_HT)
751                 r |= STA_STATS_RATE_HT;
752         else
753                 r |= STA_STATS_RATE_LEGACY | (s->band << 4);
754
755         return r;
756 }
757
758 #endif /* STA_INFO_H */