1 /*
2  * Copyright 2002-2005, Devicescape Software, Inc.
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License version 2 as
6  * published by the Free Software Foundation.
7  */
8 
9 #ifndef STA_INFO_H
10 #define STA_INFO_H
11 
12 #include <linux/list.h>
13 #include <linux/types.h>
14 #include <linux/if_ether.h>
15 #include <linux/workqueue.h>
16 #include <linux/average.h>
17 #include <linux/etherdevice.h>
18 #include "key.h"
19 
20 /**
21  * enum ieee80211_sta_info_flags - Stations flags
22  *
23  * These flags are used with &struct sta_info's @flags member, but
24  * only indirectly with set_sta_flag() and friends.
25  *
26  * @WLAN_STA_AUTH: Station is authenticated.
27  * @WLAN_STA_ASSOC: Station is associated.
28  * @WLAN_STA_PS_STA: Station is in power-save mode
29  * @WLAN_STA_AUTHORIZED: Station is authorized to send/receive traffic.
30  *	This bit is always checked so needs to be enabled for all stations
31  *	when virtual port control is not in use.
32  * @WLAN_STA_SHORT_PREAMBLE: Station is capable of receiving short-preamble
33  *	frames.
34  * @WLAN_STA_WME: Station is a QoS-STA.
35  * @WLAN_STA_WDS: Station is one of our WDS peers.
36  * @WLAN_STA_CLEAR_PS_FILT: Clear PS filter in hardware (using the
37  *	IEEE80211_TX_CTL_CLEAR_PS_FILT control flag) when the next
38  *	frame to this station is transmitted.
39  * @WLAN_STA_MFP: Management frame protection is used with this STA.
40  * @WLAN_STA_BLOCK_BA: Used to deny ADDBA requests (both TX and RX)
41  *	during suspend/resume and station removal.
42  * @WLAN_STA_PS_DRIVER: driver requires keeping this station in
43  *	power-save mode logically to flush frames that might still
44  *	be in the queues
45  * @WLAN_STA_PSPOLL: Station sent PS-poll while driver was keeping
46  *	station in power-save mode, reply when the driver unblocks.
47  * @WLAN_STA_TDLS_PEER: Station is a TDLS peer.
48  * @WLAN_STA_TDLS_PEER_AUTH: This TDLS peer is authorized to send direct
49  *	packets. This means the link is enabled.
50  * @WLAN_STA_UAPSD: Station requested unscheduled SP while driver was
51  *	keeping station in power-save mode, reply when the driver
52  *	unblocks the station.
53  * @WLAN_STA_SP: Station is in a service period, so don't try to
54  *	reply to other uAPSD trigger frames or PS-Poll.
55  * @WLAN_STA_4ADDR_EVENT: 4-addr event was already sent for this frame.
56  * @WLAN_STA_INSERTED: This station is inserted into the hash table.
57  * @WLAN_STA_RATE_CONTROL: rate control was initialized for this station.
58  */
59 enum ieee80211_sta_info_flags {
60 	WLAN_STA_AUTH,
61 	WLAN_STA_ASSOC,
62 	WLAN_STA_PS_STA,
63 	WLAN_STA_AUTHORIZED,
64 	WLAN_STA_SHORT_PREAMBLE,
65 	WLAN_STA_WME,
66 	WLAN_STA_WDS,
67 	WLAN_STA_CLEAR_PS_FILT,
68 	WLAN_STA_MFP,
69 	WLAN_STA_BLOCK_BA,
70 	WLAN_STA_PS_DRIVER,
71 	WLAN_STA_PSPOLL,
72 	WLAN_STA_TDLS_PEER,
73 	WLAN_STA_TDLS_PEER_AUTH,
74 	WLAN_STA_UAPSD,
75 	WLAN_STA_SP,
76 	WLAN_STA_4ADDR_EVENT,
77 	WLAN_STA_INSERTED,
78 	WLAN_STA_RATE_CONTROL,
79 };
80 
81 #define STA_TID_NUM 16
82 #define ADDBA_RESP_INTERVAL HZ
83 #define HT_AGG_MAX_RETRIES		15
84 #define HT_AGG_BURST_RETRIES		3
85 #define HT_AGG_RETRIES_PERIOD		(15 * HZ)
86 
87 #define HT_AGG_STATE_DRV_READY		0
88 #define HT_AGG_STATE_RESPONSE_RECEIVED	1
89 #define HT_AGG_STATE_OPERATIONAL	2
90 #define HT_AGG_STATE_STOPPING		3
91 #define HT_AGG_STATE_WANT_START		4
92 #define HT_AGG_STATE_WANT_STOP		5
93 
94 /**
95  * struct tid_ampdu_tx - TID aggregation information (Tx).
96  *
97  * @rcu_head: rcu head for freeing structure
98  * @session_timer: check if we keep Tx-ing on the TID (by timeout value)
99  * @addba_resp_timer: timer for peer's response to addba request
100  * @pending: pending frames queue -- use sta's spinlock to protect
101  * @dialog_token: dialog token for aggregation session
102  * @timeout: session timeout value to be filled in ADDBA requests
103  * @state: session state (see above)
104  * @stop_initiator: initiator of a session stop
105  * @tx_stop: TX DelBA frame when stopping
106  * @buf_size: reorder buffer size at receiver
107  * @failed_bar_ssn: ssn of the last failed BAR tx attempt
108  * @bar_pending: BAR needs to be re-sent
109  *
110  * This structure's lifetime is managed by RCU, assignments to
111  * the array holding it must hold the aggregation mutex.
112  *
113  * The TX path can access it under RCU lock-free if, and
114  * only if, the state has the flag %HT_AGG_STATE_OPERATIONAL
115  * set. Otherwise, the TX path must also acquire the spinlock
116  * and re-check the state, see comments in the tx code
117  * touching it.
118  */
119 struct tid_ampdu_tx {
120 	struct rcu_head rcu_head;
121 	struct timer_list session_timer;
122 	struct timer_list addba_resp_timer;
123 	struct sk_buff_head pending;
124 	unsigned long state;
125 	u16 timeout;
126 	u8 dialog_token;
127 	u8 stop_initiator;
128 	bool tx_stop;
129 	u8 buf_size;
130 
131 	u16 failed_bar_ssn;
132 	bool bar_pending;
133 };
134 
135 /**
136  * struct tid_ampdu_rx - TID aggregation information (Rx).
137  *
138  * @reorder_buf: buffer to reorder incoming aggregated MPDUs
139  * @reorder_time: jiffies when skb was added
140  * @session_timer: check if peer keeps Tx-ing on the TID (by timeout value)
141  * @reorder_timer: releases expired frames from the reorder buffer.
142  * @head_seq_num: head sequence number in reordering buffer.
143  * @stored_mpdu_num: number of MPDUs in reordering buffer
144  * @ssn: Starting Sequence Number expected to be aggregated.
145  * @buf_size: buffer size for incoming A-MPDUs
146  * @timeout: reset timer value (in TUs).
147  * @dialog_token: dialog token for aggregation session
148  * @rcu_head: RCU head used for freeing this struct
149  * @reorder_lock: serializes access to reorder buffer, see below.
150  *
151  * This structure's lifetime is managed by RCU, assignments to
152  * the array holding it must hold the aggregation mutex.
153  *
154  * The @reorder_lock is used to protect the members of this
155  * struct, except for @timeout, @buf_size and @dialog_token,
156  * which are constant across the lifetime of the struct (the
157  * dialog token being used only for debugging).
158  */
159 struct tid_ampdu_rx {
160 	struct rcu_head rcu_head;
161 	spinlock_t reorder_lock;
162 	struct sk_buff **reorder_buf;
163 	unsigned long *reorder_time;
164 	struct timer_list session_timer;
165 	struct timer_list reorder_timer;
166 	u16 head_seq_num;
167 	u16 stored_mpdu_num;
168 	u16 ssn;
169 	u16 buf_size;
170 	u16 timeout;
171 	u8 dialog_token;
172 };
173 
174 /**
175  * struct sta_ampdu_mlme - STA aggregation information.
176  *
177  * @tid_rx: aggregation info for Rx per TID -- RCU protected
178  * @tid_tx: aggregation info for Tx per TID
179  * @tid_start_tx: sessions where start was requested
180  * @addba_req_num: number of times addBA request has been sent.
181  * @last_addba_req_time: timestamp of the last addBA request.
182  * @dialog_token_allocator: dialog token enumerator for each new session;
183  * @work: work struct for starting/stopping aggregation
184  * @tid_rx_timer_expired: bitmap indicating on which TIDs the
185  *	RX timer expired until the work for it runs
186  * @tid_rx_stop_requested:  bitmap indicating which BA sessions per TID the
187  *	driver requested to close until the work for it runs
188  * @mtx: mutex to protect all TX data (except non-NULL assignments
189  *	to tid_tx[idx], which are protected by the sta spinlock)
190  */
191 struct sta_ampdu_mlme {
192 	struct mutex mtx;
193 	/* rx */
194 	struct tid_ampdu_rx __rcu *tid_rx[STA_TID_NUM];
195 	unsigned long tid_rx_timer_expired[BITS_TO_LONGS(STA_TID_NUM)];
196 	unsigned long tid_rx_stop_requested[BITS_TO_LONGS(STA_TID_NUM)];
197 	/* tx */
198 	struct work_struct work;
199 	struct tid_ampdu_tx __rcu *tid_tx[STA_TID_NUM];
200 	struct tid_ampdu_tx *tid_start_tx[STA_TID_NUM];
201 	unsigned long last_addba_req_time[STA_TID_NUM];
202 	u8 addba_req_num[STA_TID_NUM];
203 	u8 dialog_token_allocator;
204 };
205 
206 
207 /**
208  * struct sta_info - STA information
209  *
210  * This structure collects information about a station that
211  * mac80211 is communicating with.
212  *
213  * @list: global linked list entry
214  * @hnext: hash table linked list pointer
215  * @local: pointer to the global information
216  * @sdata: virtual interface this station belongs to
217  * @ptk: peer key negotiated with this station, if any
218  * @gtk: group keys negotiated with this station, if any
219  * @rate_ctrl: rate control algorithm reference
220  * @rate_ctrl_priv: rate control private per-STA pointer
221  * @last_tx_rate: rate used for last transmit, to report to userspace as
222  *	"the" transmit rate
223  * @last_rx_rate_idx: rx status rate index of the last data packet
224  * @last_rx_rate_flag: rx status flag of the last data packet
225  * @lock: used for locking all fields that require locking, see comments
226  *	in the header file.
227  * @drv_unblock_wk: used for driver PS unblocking
228  * @listen_interval: listen interval of this station, when we're acting as AP
229  * @_flags: STA flags, see &enum ieee80211_sta_info_flags, do not use directly
230  * @ps_lock: used for powersave (when mac80211 is the AP) related locking
231  * @ps_tx_buf: buffers (per AC) of frames to transmit to this station
232  *	when it leaves power saving state or polls
233  * @tx_filtered: buffers (per AC) of frames we already tried to
234  *	transmit but were filtered by hardware due to STA having
235  *	entered power saving state, these are also delivered to
236  *	the station when it leaves powersave or polls for frames
237  * @driver_buffered_tids: bitmap of TIDs the driver has data buffered on
238  * @rx_packets: Number of MSDUs received from this STA
239  * @rx_bytes: Number of bytes received from this STA
240  * @wep_weak_iv_count: number of weak WEP IVs received from this station
241  * @last_rx: time (in jiffies) when last frame was received from this STA
242  * @last_connected: time (in seconds) when a station got connected
243  * @num_duplicates: number of duplicate frames received from this STA
244  * @rx_fragments: number of received MPDUs
245  * @rx_dropped: number of dropped MPDUs from this STA
246  * @last_signal: signal of last received frame from this STA
247  * @avg_signal: moving average of signal of received frames from this STA
248  * @last_seq_ctrl: last received seq/frag number from this STA (per RX queue)
249  * @tx_filtered_count: number of frames the hardware filtered for this STA
250  * @tx_retry_failed: number of frames that failed retry
251  * @tx_retry_count: total number of retries for frames to this STA
252  * @fail_avg: moving percentage of failed MSDUs
253  * @tx_packets: number of RX/TX MSDUs
254  * @tx_bytes: number of bytes transmitted to this STA
255  * @tx_fragments: number of transmitted MPDUs
256  * @tid_seq: per-TID sequence numbers for sending to this STA
257  * @ampdu_mlme: A-MPDU state machine state
258  * @timer_to_tid: identity mapping to ID timers
259  * @llid: Local link ID
260  * @plid: Peer link ID
261  * @reason: Cancel reason on PLINK_HOLDING state
262  * @plink_retries: Retries in establishment
263  * @ignore_plink_timer: ignore the peer-link timer (used internally)
264  * @plink_state: peer link state
265  * @plink_timeout: timeout of peer link
266  * @plink_timer: peer link watch timer
267  * @plink_timer_was_running: used by suspend/resume to restore timers
268  * @debugfs: debug filesystem info
269  * @dead: set to true when sta is unlinked
270  * @uploaded: set to true when sta is uploaded to the driver
271  * @lost_packets: number of consecutive lost packets
272  * @sta: station information we share with the driver
273  * @sta_state: duplicates information about station state (for debug)
274  * @beacon_loss_count: number of times beacon loss has triggered
275  */
276 struct sta_info {
277 	/* General information, mostly static */
278 	struct list_head list;
279 	struct sta_info __rcu *hnext;
280 	struct ieee80211_local *local;
281 	struct ieee80211_sub_if_data *sdata;
282 	struct ieee80211_key __rcu *gtk[NUM_DEFAULT_KEYS + NUM_DEFAULT_MGMT_KEYS];
283 	struct ieee80211_key __rcu *ptk;
284 	struct rate_control_ref *rate_ctrl;
285 	void *rate_ctrl_priv;
286 	spinlock_t lock;
287 
288 	struct work_struct drv_unblock_wk;
289 
290 	u16 listen_interval;
291 
292 	bool dead;
293 
294 	bool uploaded;
295 
296 	enum ieee80211_sta_state sta_state;
297 
298 	/* use the accessors defined below */
299 	unsigned long _flags;
300 
301 	/* STA powersave lock and frame queues */
302 	spinlock_t ps_lock;
303 	struct sk_buff_head ps_tx_buf[IEEE80211_NUM_ACS];
304 	struct sk_buff_head tx_filtered[IEEE80211_NUM_ACS];
305 	unsigned long driver_buffered_tids;
306 
307 	/* Updated from RX path only, no locking requirements */
308 	unsigned long rx_packets, rx_bytes;
309 	unsigned long wep_weak_iv_count;
310 	unsigned long last_rx;
311 	long last_connected;
312 	unsigned long num_duplicates;
313 	unsigned long rx_fragments;
314 	unsigned long rx_dropped;
315 	int last_signal;
316 	struct ewma avg_signal;
317 	/* Plus 1 for non-QoS frames */
318 	__le16 last_seq_ctrl[NUM_RX_DATA_QUEUES + 1];
319 
320 	/* Updated from TX status path only, no locking requirements */
321 	unsigned long tx_filtered_count;
322 	unsigned long tx_retry_failed, tx_retry_count;
323 	/* moving percentage of failed MSDUs */
324 	unsigned int fail_avg;
325 
326 	/* Updated from TX path only, no locking requirements */
327 	unsigned long tx_packets;
328 	unsigned long tx_bytes;
329 	unsigned long tx_fragments;
330 	struct ieee80211_tx_rate last_tx_rate;
331 	int last_rx_rate_idx;
332 	int last_rx_rate_flag;
333 	u16 tid_seq[IEEE80211_QOS_CTL_TID_MASK + 1];
334 
335 	/*
336 	 * Aggregation information, locked with lock.
337 	 */
338 	struct sta_ampdu_mlme ampdu_mlme;
339 	u8 timer_to_tid[STA_TID_NUM];
340 
341 #ifdef CONFIG_MAC80211_MESH
342 	/*
343 	 * Mesh peer link attributes
344 	 * TODO: move to a sub-structure that is referenced with pointer?
345 	 */
346 	__le16 llid;
347 	__le16 plid;
348 	__le16 reason;
349 	u8 plink_retries;
350 	bool ignore_plink_timer;
351 	bool plink_timer_was_running;
352 	enum nl80211_plink_state plink_state;
353 	u32 plink_timeout;
354 	struct timer_list plink_timer;
355 #endif
356 
357 #ifdef CONFIG_MAC80211_DEBUGFS
358 	struct sta_info_debugfsdentries {
359 		struct dentry *dir;
360 		bool add_has_run;
361 	} debugfs;
362 #endif
363 
364 	unsigned int lost_packets;
365 	unsigned int beacon_loss_count;
366 
367 	/* keep last! */
368 	struct ieee80211_sta sta;
369 };
370 
sta_plink_state(struct sta_info * sta)371 static inline enum nl80211_plink_state sta_plink_state(struct sta_info *sta)
372 {
373 #ifdef CONFIG_MAC80211_MESH
374 	return sta->plink_state;
375 #endif
376 	return NL80211_PLINK_LISTEN;
377 }
378 
set_sta_flag(struct sta_info * sta,enum ieee80211_sta_info_flags flag)379 static inline void set_sta_flag(struct sta_info *sta,
380 				enum ieee80211_sta_info_flags flag)
381 {
382 	WARN_ON(flag == WLAN_STA_AUTH ||
383 		flag == WLAN_STA_ASSOC ||
384 		flag == WLAN_STA_AUTHORIZED);
385 	set_bit(flag, &sta->_flags);
386 }
387 
clear_sta_flag(struct sta_info * sta,enum ieee80211_sta_info_flags flag)388 static inline void clear_sta_flag(struct sta_info *sta,
389 				  enum ieee80211_sta_info_flags flag)
390 {
391 	WARN_ON(flag == WLAN_STA_AUTH ||
392 		flag == WLAN_STA_ASSOC ||
393 		flag == WLAN_STA_AUTHORIZED);
394 	clear_bit(flag, &sta->_flags);
395 }
396 
test_sta_flag(struct sta_info * sta,enum ieee80211_sta_info_flags flag)397 static inline int test_sta_flag(struct sta_info *sta,
398 				enum ieee80211_sta_info_flags flag)
399 {
400 	return test_bit(flag, &sta->_flags);
401 }
402 
test_and_clear_sta_flag(struct sta_info * sta,enum ieee80211_sta_info_flags flag)403 static inline int test_and_clear_sta_flag(struct sta_info *sta,
404 					  enum ieee80211_sta_info_flags flag)
405 {
406 	WARN_ON(flag == WLAN_STA_AUTH ||
407 		flag == WLAN_STA_ASSOC ||
408 		flag == WLAN_STA_AUTHORIZED);
409 	return test_and_clear_bit(flag, &sta->_flags);
410 }
411 
test_and_set_sta_flag(struct sta_info * sta,enum ieee80211_sta_info_flags flag)412 static inline int test_and_set_sta_flag(struct sta_info *sta,
413 					enum ieee80211_sta_info_flags flag)
414 {
415 	WARN_ON(flag == WLAN_STA_AUTH ||
416 		flag == WLAN_STA_ASSOC ||
417 		flag == WLAN_STA_AUTHORIZED);
418 	return test_and_set_bit(flag, &sta->_flags);
419 }
420 
421 int sta_info_move_state(struct sta_info *sta,
422 			enum ieee80211_sta_state new_state);
423 
sta_info_pre_move_state(struct sta_info * sta,enum ieee80211_sta_state new_state)424 static inline void sta_info_pre_move_state(struct sta_info *sta,
425 					   enum ieee80211_sta_state new_state)
426 {
427 	int ret;
428 
429 	WARN_ON_ONCE(test_sta_flag(sta, WLAN_STA_INSERTED));
430 
431 	ret = sta_info_move_state(sta, new_state);
432 	WARN_ON_ONCE(ret);
433 }
434 
435 
436 void ieee80211_assign_tid_tx(struct sta_info *sta, int tid,
437 			     struct tid_ampdu_tx *tid_tx);
438 
439 static inline struct tid_ampdu_tx *
rcu_dereference_protected_tid_tx(struct sta_info * sta,int tid)440 rcu_dereference_protected_tid_tx(struct sta_info *sta, int tid)
441 {
442 	return rcu_dereference_protected(sta->ampdu_mlme.tid_tx[tid],
443 					 lockdep_is_held(&sta->lock) ||
444 					 lockdep_is_held(&sta->ampdu_mlme.mtx));
445 }
446 
447 #define STA_HASH_SIZE 256
448 #define STA_HASH(sta) (sta[5])
449 
450 
451 /* Maximum number of frames to buffer per power saving station per AC */
452 #define STA_MAX_TX_BUFFER	64
453 
454 /* Minimum buffered frame expiry time. If STA uses listen interval that is
455  * smaller than this value, the minimum value here is used instead. */
456 #define STA_TX_BUFFER_EXPIRE (10 * HZ)
457 
458 /* How often station data is cleaned up (e.g., expiration of buffered frames)
459  */
460 #define STA_INFO_CLEANUP_INTERVAL (10 * HZ)
461 
462 /*
463  * Get a STA info, must be under RCU read lock.
464  */
465 struct sta_info *sta_info_get(struct ieee80211_sub_if_data *sdata,
466 			      const u8 *addr);
467 
468 struct sta_info *sta_info_get_bss(struct ieee80211_sub_if_data *sdata,
469 				  const u8 *addr);
470 
471 static inline
for_each_sta_info_type_check(struct ieee80211_local * local,const u8 * addr,struct sta_info * sta,struct sta_info * nxt)472 void for_each_sta_info_type_check(struct ieee80211_local *local,
473 				  const u8 *addr,
474 				  struct sta_info *sta,
475 				  struct sta_info *nxt)
476 {
477 }
478 
479 #define for_each_sta_info(local, _addr, _sta, nxt)			\
480 	for (	/* initialise loop */					\
481 		_sta = rcu_dereference(local->sta_hash[STA_HASH(_addr)]),\
482 		nxt = _sta ? rcu_dereference(_sta->hnext) : NULL;	\
483 		/* typecheck */						\
484 		for_each_sta_info_type_check(local, (_addr), _sta, nxt),\
485 		/* continue condition */				\
486 		_sta;							\
487 		/* advance loop */					\
488 		_sta = nxt,						\
489 		nxt = _sta ? rcu_dereference(_sta->hnext) : NULL	\
490 	     )								\
491 	/* compare address and run code only if it matches */		\
492 	if (compare_ether_addr(_sta->sta.addr, (_addr)) == 0)
493 
494 /*
495  * Get STA info by index, BROKEN!
496  */
497 struct sta_info *sta_info_get_by_idx(struct ieee80211_sub_if_data *sdata,
498 				     int idx);
499 /*
500  * Create a new STA info, caller owns returned structure
501  * until sta_info_insert().
502  */
503 struct sta_info *sta_info_alloc(struct ieee80211_sub_if_data *sdata,
504 				const u8 *addr, gfp_t gfp);
505 
506 void sta_info_free(struct ieee80211_local *local, struct sta_info *sta);
507 
508 /*
509  * Insert STA info into hash table/list, returns zero or a
510  * -EEXIST if (if the same MAC address is already present).
511  *
512  * Calling the non-rcu version makes the caller relinquish,
513  * the _rcu version calls read_lock_rcu() and must be called
514  * without it held.
515  */
516 int sta_info_insert(struct sta_info *sta);
517 int sta_info_insert_rcu(struct sta_info *sta) __acquires(RCU);
518 
519 int __must_check __sta_info_destroy(struct sta_info *sta);
520 int sta_info_destroy_addr(struct ieee80211_sub_if_data *sdata,
521 			  const u8 *addr);
522 int sta_info_destroy_addr_bss(struct ieee80211_sub_if_data *sdata,
523 			      const u8 *addr);
524 
525 void sta_info_recalc_tim(struct sta_info *sta);
526 
527 void sta_info_init(struct ieee80211_local *local);
528 void sta_info_stop(struct ieee80211_local *local);
529 int sta_info_flush(struct ieee80211_local *local,
530 		   struct ieee80211_sub_if_data *sdata);
531 void sta_set_rate_info_tx(struct sta_info *sta,
532 			  const struct ieee80211_tx_rate *rate,
533 			  struct rate_info *rinfo);
534 void ieee80211_sta_expire(struct ieee80211_sub_if_data *sdata,
535 			  unsigned long exp_time);
536 
537 void ieee80211_sta_ps_deliver_wakeup(struct sta_info *sta);
538 void ieee80211_sta_ps_deliver_poll_response(struct sta_info *sta);
539 void ieee80211_sta_ps_deliver_uapsd(struct sta_info *sta);
540 
541 #endif /* STA_INFO_H */
542