1 /******************************************************************************
2  *
3  * Copyright(c) 2005 - 2014 Intel Corporation. All rights reserved.
4  * Copyright(c) 2013 - 2015 Intel Mobile Communications GmbH
5  * Copyright(c) 2016 Intel Deutschland GmbH
6  *
7  * This program is free software; you can redistribute it and/or modify it
8  * under the terms of version 2 of the GNU General Public License as
9  * published by the Free Software Foundation.
10  *
11  * This program is distributed in the hope that it will be useful, but WITHOUT
12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
14  * more details.
15  *
16  * You should have received a copy of the GNU General Public License along with
17  * this program; if not, write to the Free Software Foundation, Inc.,
18  * 51 Franklin Street, Fifth Floor, Boston, MA 02110, USA
19  *
20  * The full GNU General Public License is included in this distribution in the
21  * file called LICENSE.
22  *
23  * Contact Information:
24  *  Intel Linux Wireless <linuxwifi@intel.com>
25  * Intel Corporation, 5200 N.E. Elam Young Parkway, Hillsboro, OR 97124-6497
26  *
27  *****************************************************************************/
28 #include <linux/kernel.h>
29 #include <linux/skbuff.h>
30 #include <linux/slab.h>
31 #include <net/mac80211.h>
32 
33 #include <linux/netdevice.h>
34 #include <linux/etherdevice.h>
35 #include <linux/delay.h>
36 
37 #include <linux/workqueue.h>
38 #include "rs.h"
39 #include "fw-api.h"
40 #include "sta.h"
41 #include "iwl-op-mode.h"
42 #include "mvm.h"
43 #include "debugfs.h"
44 
45 #define RS_NAME "iwl-mvm-rs"
46 
47 #define IWL_RATE_MAX_WINDOW		62	/* # tx in history window */
48 
49 /* Calculations of success ratio are done in fixed point where 12800 is 100%.
50  * Use this macro when dealing with thresholds consts set as a percentage
51  */
52 #define RS_PERCENT(x) (128 * x)
53 
54 static u8 rs_ht_to_legacy[] = {
55 	[IWL_RATE_MCS_0_INDEX] = IWL_RATE_6M_INDEX,
56 	[IWL_RATE_MCS_1_INDEX] = IWL_RATE_9M_INDEX,
57 	[IWL_RATE_MCS_2_INDEX] = IWL_RATE_12M_INDEX,
58 	[IWL_RATE_MCS_3_INDEX] = IWL_RATE_18M_INDEX,
59 	[IWL_RATE_MCS_4_INDEX] = IWL_RATE_24M_INDEX,
60 	[IWL_RATE_MCS_5_INDEX] = IWL_RATE_36M_INDEX,
61 	[IWL_RATE_MCS_6_INDEX] = IWL_RATE_48M_INDEX,
62 	[IWL_RATE_MCS_7_INDEX] = IWL_RATE_54M_INDEX,
63 	[IWL_RATE_MCS_8_INDEX] = IWL_RATE_54M_INDEX,
64 	[IWL_RATE_MCS_9_INDEX] = IWL_RATE_54M_INDEX,
65 };
66 
67 static const u8 ant_toggle_lookup[] = {
68 	[ANT_NONE] = ANT_NONE,
69 	[ANT_A] = ANT_B,
70 	[ANT_B] = ANT_C,
71 	[ANT_AB] = ANT_BC,
72 	[ANT_C] = ANT_A,
73 	[ANT_AC] = ANT_AB,
74 	[ANT_BC] = ANT_AC,
75 	[ANT_ABC] = ANT_ABC,
76 };
77 
78 #define IWL_DECLARE_RATE_INFO(r, s, rp, rn)			      \
79 	[IWL_RATE_##r##M_INDEX] = { IWL_RATE_##r##M_PLCP,	      \
80 				    IWL_RATE_HT_SISO_MCS_##s##_PLCP,  \
81 				    IWL_RATE_HT_MIMO2_MCS_##s##_PLCP, \
82 				    IWL_RATE_VHT_SISO_MCS_##s##_PLCP, \
83 				    IWL_RATE_VHT_MIMO2_MCS_##s##_PLCP,\
84 				    IWL_RATE_##rp##M_INDEX,	      \
85 				    IWL_RATE_##rn##M_INDEX }
86 
87 #define IWL_DECLARE_MCS_RATE(s)						  \
88 	[IWL_RATE_MCS_##s##_INDEX] = { IWL_RATE_INVM_PLCP,		  \
89 				       IWL_RATE_HT_SISO_MCS_##s##_PLCP,	  \
90 				       IWL_RATE_HT_MIMO2_MCS_##s##_PLCP,  \
91 				       IWL_RATE_VHT_SISO_MCS_##s##_PLCP,  \
92 				       IWL_RATE_VHT_MIMO2_MCS_##s##_PLCP, \
93 				       IWL_RATE_INVM_INDEX,	          \
94 				       IWL_RATE_INVM_INDEX }
95 
96 /*
97  * Parameter order:
98  *   rate, ht rate, prev rate, next rate
99  *
100  * If there isn't a valid next or previous rate then INV is used which
101  * maps to IWL_RATE_INVALID
102  *
103  */
104 static const struct iwl_rs_rate_info iwl_rates[IWL_RATE_COUNT] = {
105 	IWL_DECLARE_RATE_INFO(1, INV, INV, 2),   /*  1mbps */
106 	IWL_DECLARE_RATE_INFO(2, INV, 1, 5),     /*  2mbps */
107 	IWL_DECLARE_RATE_INFO(5, INV, 2, 11),    /*5.5mbps */
108 	IWL_DECLARE_RATE_INFO(11, INV, 9, 12),   /* 11mbps */
109 	IWL_DECLARE_RATE_INFO(6, 0, 5, 11),      /*  6mbps ; MCS 0 */
110 	IWL_DECLARE_RATE_INFO(9, INV, 6, 11),    /*  9mbps */
111 	IWL_DECLARE_RATE_INFO(12, 1, 11, 18),    /* 12mbps ; MCS 1 */
112 	IWL_DECLARE_RATE_INFO(18, 2, 12, 24),    /* 18mbps ; MCS 2 */
113 	IWL_DECLARE_RATE_INFO(24, 3, 18, 36),    /* 24mbps ; MCS 3 */
114 	IWL_DECLARE_RATE_INFO(36, 4, 24, 48),    /* 36mbps ; MCS 4 */
115 	IWL_DECLARE_RATE_INFO(48, 5, 36, 54),    /* 48mbps ; MCS 5 */
116 	IWL_DECLARE_RATE_INFO(54, 6, 48, INV),   /* 54mbps ; MCS 6 */
117 	IWL_DECLARE_MCS_RATE(7),                 /* MCS 7 */
118 	IWL_DECLARE_MCS_RATE(8),                 /* MCS 8 */
119 	IWL_DECLARE_MCS_RATE(9),                 /* MCS 9 */
120 };
121 
122 enum rs_action {
123 	RS_ACTION_STAY = 0,
124 	RS_ACTION_DOWNSCALE = -1,
125 	RS_ACTION_UPSCALE = 1,
126 };
127 
128 enum rs_column_mode {
129 	RS_INVALID = 0,
130 	RS_LEGACY,
131 	RS_SISO,
132 	RS_MIMO2,
133 };
134 
135 #define MAX_NEXT_COLUMNS 7
136 #define MAX_COLUMN_CHECKS 3
137 
138 struct rs_tx_column;
139 
140 typedef bool (*allow_column_func_t) (struct iwl_mvm *mvm,
141 				     struct ieee80211_sta *sta,
142 				     struct rs_rate *rate,
143 				     const struct rs_tx_column *next_col);
144 
145 struct rs_tx_column {
146 	enum rs_column_mode mode;
147 	u8 ant;
148 	bool sgi;
149 	enum rs_column next_columns[MAX_NEXT_COLUMNS];
150 	allow_column_func_t checks[MAX_COLUMN_CHECKS];
151 };
152 
153 static bool rs_ant_allow(struct iwl_mvm *mvm, struct ieee80211_sta *sta,
154 			 struct rs_rate *rate,
155 			 const struct rs_tx_column *next_col)
156 {
157 	return iwl_mvm_bt_coex_is_ant_avail(mvm, next_col->ant);
158 }
159 
160 static bool rs_mimo_allow(struct iwl_mvm *mvm, struct ieee80211_sta *sta,
161 			  struct rs_rate *rate,
162 			  const struct rs_tx_column *next_col)
163 {
164 	struct iwl_mvm_sta *mvmsta;
165 	struct iwl_mvm_vif *mvmvif;
166 
167 	if (!sta->ht_cap.ht_supported)
168 		return false;
169 
170 	if (sta->smps_mode == IEEE80211_SMPS_STATIC)
171 		return false;
172 
173 	if (num_of_ant(iwl_mvm_get_valid_tx_ant(mvm)) < 2)
174 		return false;
175 
176 	if (!iwl_mvm_bt_coex_is_mimo_allowed(mvm, sta))
177 		return false;
178 
179 	mvmsta = iwl_mvm_sta_from_mac80211(sta);
180 	mvmvif = iwl_mvm_vif_from_mac80211(mvmsta->vif);
181 
182 	if (mvm->nvm_data->sku_cap_mimo_disabled)
183 		return false;
184 
185 	return true;
186 }
187 
188 static bool rs_siso_allow(struct iwl_mvm *mvm, struct ieee80211_sta *sta,
189 			  struct rs_rate *rate,
190 			  const struct rs_tx_column *next_col)
191 {
192 	if (!sta->ht_cap.ht_supported)
193 		return false;
194 
195 	return true;
196 }
197 
198 static bool rs_sgi_allow(struct iwl_mvm *mvm, struct ieee80211_sta *sta,
199 			 struct rs_rate *rate,
200 			 const struct rs_tx_column *next_col)
201 {
202 	struct ieee80211_sta_ht_cap *ht_cap = &sta->ht_cap;
203 	struct ieee80211_sta_vht_cap *vht_cap = &sta->vht_cap;
204 
205 	if (is_ht20(rate) && (ht_cap->cap &
206 			     IEEE80211_HT_CAP_SGI_20))
207 		return true;
208 	if (is_ht40(rate) && (ht_cap->cap &
209 			     IEEE80211_HT_CAP_SGI_40))
210 		return true;
211 	if (is_ht80(rate) && (vht_cap->cap &
212 			     IEEE80211_VHT_CAP_SHORT_GI_80))
213 		return true;
214 	if (is_ht160(rate) && (vht_cap->cap &
215 			     IEEE80211_VHT_CAP_SHORT_GI_160))
216 		return true;
217 
218 	return false;
219 }
220 
221 static const struct rs_tx_column rs_tx_columns[] = {
222 	[RS_COLUMN_LEGACY_ANT_A] = {
223 		.mode = RS_LEGACY,
224 		.ant = ANT_A,
225 		.next_columns = {
226 			RS_COLUMN_LEGACY_ANT_B,
227 			RS_COLUMN_SISO_ANT_A,
228 			RS_COLUMN_MIMO2,
229 			RS_COLUMN_INVALID,
230 			RS_COLUMN_INVALID,
231 			RS_COLUMN_INVALID,
232 			RS_COLUMN_INVALID,
233 		},
234 		.checks = {
235 			rs_ant_allow,
236 		},
237 	},
238 	[RS_COLUMN_LEGACY_ANT_B] = {
239 		.mode = RS_LEGACY,
240 		.ant = ANT_B,
241 		.next_columns = {
242 			RS_COLUMN_LEGACY_ANT_A,
243 			RS_COLUMN_SISO_ANT_B,
244 			RS_COLUMN_MIMO2,
245 			RS_COLUMN_INVALID,
246 			RS_COLUMN_INVALID,
247 			RS_COLUMN_INVALID,
248 			RS_COLUMN_INVALID,
249 		},
250 		.checks = {
251 			rs_ant_allow,
252 		},
253 	},
254 	[RS_COLUMN_SISO_ANT_A] = {
255 		.mode = RS_SISO,
256 		.ant = ANT_A,
257 		.next_columns = {
258 			RS_COLUMN_SISO_ANT_B,
259 			RS_COLUMN_MIMO2,
260 			RS_COLUMN_SISO_ANT_A_SGI,
261 			RS_COLUMN_LEGACY_ANT_A,
262 			RS_COLUMN_LEGACY_ANT_B,
263 			RS_COLUMN_INVALID,
264 			RS_COLUMN_INVALID,
265 		},
266 		.checks = {
267 			rs_siso_allow,
268 			rs_ant_allow,
269 		},
270 	},
271 	[RS_COLUMN_SISO_ANT_B] = {
272 		.mode = RS_SISO,
273 		.ant = ANT_B,
274 		.next_columns = {
275 			RS_COLUMN_SISO_ANT_A,
276 			RS_COLUMN_MIMO2,
277 			RS_COLUMN_SISO_ANT_B_SGI,
278 			RS_COLUMN_LEGACY_ANT_A,
279 			RS_COLUMN_LEGACY_ANT_B,
280 			RS_COLUMN_INVALID,
281 			RS_COLUMN_INVALID,
282 		},
283 		.checks = {
284 			rs_siso_allow,
285 			rs_ant_allow,
286 		},
287 	},
288 	[RS_COLUMN_SISO_ANT_A_SGI] = {
289 		.mode = RS_SISO,
290 		.ant = ANT_A,
291 		.sgi = true,
292 		.next_columns = {
293 			RS_COLUMN_SISO_ANT_B_SGI,
294 			RS_COLUMN_MIMO2_SGI,
295 			RS_COLUMN_SISO_ANT_A,
296 			RS_COLUMN_LEGACY_ANT_A,
297 			RS_COLUMN_LEGACY_ANT_B,
298 			RS_COLUMN_INVALID,
299 			RS_COLUMN_INVALID,
300 		},
301 		.checks = {
302 			rs_siso_allow,
303 			rs_ant_allow,
304 			rs_sgi_allow,
305 		},
306 	},
307 	[RS_COLUMN_SISO_ANT_B_SGI] = {
308 		.mode = RS_SISO,
309 		.ant = ANT_B,
310 		.sgi = true,
311 		.next_columns = {
312 			RS_COLUMN_SISO_ANT_A_SGI,
313 			RS_COLUMN_MIMO2_SGI,
314 			RS_COLUMN_SISO_ANT_B,
315 			RS_COLUMN_LEGACY_ANT_A,
316 			RS_COLUMN_LEGACY_ANT_B,
317 			RS_COLUMN_INVALID,
318 			RS_COLUMN_INVALID,
319 		},
320 		.checks = {
321 			rs_siso_allow,
322 			rs_ant_allow,
323 			rs_sgi_allow,
324 		},
325 	},
326 	[RS_COLUMN_MIMO2] = {
327 		.mode = RS_MIMO2,
328 		.ant = ANT_AB,
329 		.next_columns = {
330 			RS_COLUMN_SISO_ANT_A,
331 			RS_COLUMN_MIMO2_SGI,
332 			RS_COLUMN_LEGACY_ANT_A,
333 			RS_COLUMN_LEGACY_ANT_B,
334 			RS_COLUMN_INVALID,
335 			RS_COLUMN_INVALID,
336 			RS_COLUMN_INVALID,
337 		},
338 		.checks = {
339 			rs_mimo_allow,
340 		},
341 	},
342 	[RS_COLUMN_MIMO2_SGI] = {
343 		.mode = RS_MIMO2,
344 		.ant = ANT_AB,
345 		.sgi = true,
346 		.next_columns = {
347 			RS_COLUMN_SISO_ANT_A_SGI,
348 			RS_COLUMN_MIMO2,
349 			RS_COLUMN_LEGACY_ANT_A,
350 			RS_COLUMN_LEGACY_ANT_B,
351 			RS_COLUMN_INVALID,
352 			RS_COLUMN_INVALID,
353 			RS_COLUMN_INVALID,
354 		},
355 		.checks = {
356 			rs_mimo_allow,
357 			rs_sgi_allow,
358 		},
359 	},
360 };
361 
362 static inline u8 rs_extract_rate(u32 rate_n_flags)
363 {
364 	/* also works for HT because bits 7:6 are zero there */
365 	return (u8)(rate_n_flags & RATE_LEGACY_RATE_MSK);
366 }
367 
368 static int iwl_hwrate_to_plcp_idx(u32 rate_n_flags)
369 {
370 	int idx = 0;
371 
372 	if (rate_n_flags & RATE_MCS_HT_MSK) {
373 		idx = rate_n_flags & RATE_HT_MCS_RATE_CODE_MSK;
374 		idx += IWL_RATE_MCS_0_INDEX;
375 
376 		/* skip 9M not supported in HT*/
377 		if (idx >= IWL_RATE_9M_INDEX)
378 			idx += 1;
379 		if ((idx >= IWL_FIRST_HT_RATE) && (idx <= IWL_LAST_HT_RATE))
380 			return idx;
381 	} else if (rate_n_flags & RATE_MCS_VHT_MSK) {
382 		idx = rate_n_flags & RATE_VHT_MCS_RATE_CODE_MSK;
383 		idx += IWL_RATE_MCS_0_INDEX;
384 
385 		/* skip 9M not supported in VHT*/
386 		if (idx >= IWL_RATE_9M_INDEX)
387 			idx++;
388 		if ((idx >= IWL_FIRST_VHT_RATE) && (idx <= IWL_LAST_VHT_RATE))
389 			return idx;
390 	} else {
391 		/* legacy rate format, search for match in table */
392 
393 		u8 legacy_rate = rs_extract_rate(rate_n_flags);
394 		for (idx = 0; idx < ARRAY_SIZE(iwl_rates); idx++)
395 			if (iwl_rates[idx].plcp == legacy_rate)
396 				return idx;
397 	}
398 
399 	return IWL_RATE_INVALID;
400 }
401 
402 static void rs_rate_scale_perform(struct iwl_mvm *mvm,
403 				  struct ieee80211_sta *sta,
404 				  struct iwl_lq_sta *lq_sta,
405 				  int tid, bool ndp);
406 static void rs_fill_lq_cmd(struct iwl_mvm *mvm,
407 			   struct ieee80211_sta *sta,
408 			   struct iwl_lq_sta *lq_sta,
409 			   const struct rs_rate *initial_rate);
410 static void rs_stay_in_table(struct iwl_lq_sta *lq_sta, bool force_search);
411 
412 /**
413  * The following tables contain the expected throughput metrics for all rates
414  *
415  *	1, 2, 5.5, 11, 6, 9, 12, 18, 24, 36, 48, 54, 60 MBits
416  *
417  * where invalid entries are zeros.
418  *
419  * CCK rates are only valid in legacy table and will only be used in G
420  * (2.4 GHz) band.
421  */
422 
423 static const u16 expected_tpt_legacy[IWL_RATE_COUNT] = {
424 	7, 13, 35, 58, 40, 57, 72, 98, 121, 154, 177, 186, 0, 0, 0
425 };
426 
427 /* Expected TpT tables. 4 indexes:
428  * 0 - NGI, 1 - SGI, 2 - AGG+NGI, 3 - AGG+SGI
429  */
430 static const u16 expected_tpt_siso_20MHz[4][IWL_RATE_COUNT] = {
431 	{0, 0, 0, 0, 42, 0,  76, 102, 124, 159, 183, 193, 202, 216, 0},
432 	{0, 0, 0, 0, 46, 0,  82, 110, 132, 168, 192, 202, 210, 225, 0},
433 	{0, 0, 0, 0, 49, 0,  97, 145, 192, 285, 375, 420, 464, 551, 0},
434 	{0, 0, 0, 0, 54, 0, 108, 160, 213, 315, 415, 465, 513, 608, 0},
435 };
436 
437 static const u16 expected_tpt_siso_40MHz[4][IWL_RATE_COUNT] = {
438 	{0, 0, 0, 0,  77, 0, 127, 160, 184, 220, 242, 250,  257,  269,  275},
439 	{0, 0, 0, 0,  83, 0, 135, 169, 193, 229, 250, 257,  264,  275,  280},
440 	{0, 0, 0, 0, 101, 0, 199, 295, 389, 570, 744, 828,  911, 1070, 1173},
441 	{0, 0, 0, 0, 112, 0, 220, 326, 429, 629, 819, 912, 1000, 1173, 1284},
442 };
443 
444 static const u16 expected_tpt_siso_80MHz[4][IWL_RATE_COUNT] = {
445 	{0, 0, 0, 0, 130, 0, 191, 223, 244,  273,  288,  294,  298,  305,  308},
446 	{0, 0, 0, 0, 138, 0, 200, 231, 251,  279,  293,  298,  302,  308,  312},
447 	{0, 0, 0, 0, 217, 0, 429, 634, 834, 1220, 1585, 1760, 1931, 2258, 2466},
448 	{0, 0, 0, 0, 241, 0, 475, 701, 921, 1343, 1741, 1931, 2117, 2468, 2691},
449 };
450 
451 static const u16 expected_tpt_siso_160MHz[4][IWL_RATE_COUNT] = {
452 	{0, 0, 0, 0, 191, 0, 244, 288,  298,  308,  313,  318,  323,  328,  330},
453 	{0, 0, 0, 0, 200, 0, 251, 293,  302,  312,  317,  322,  327,  332,  334},
454 	{0, 0, 0, 0, 439, 0, 875, 1307, 1736, 2584, 3419, 3831, 4240, 5049, 5581},
455 	{0, 0, 0, 0, 488, 0, 972, 1451, 1925, 2864, 3785, 4240, 4691, 5581, 6165},
456 };
457 
458 static const u16 expected_tpt_mimo2_20MHz[4][IWL_RATE_COUNT] = {
459 	{0, 0, 0, 0,  74, 0, 123, 155, 179, 213, 235, 243, 250,  261, 0},
460 	{0, 0, 0, 0,  81, 0, 131, 164, 187, 221, 242, 250, 256,  267, 0},
461 	{0, 0, 0, 0,  98, 0, 193, 286, 375, 550, 718, 799, 878, 1032, 0},
462 	{0, 0, 0, 0, 109, 0, 214, 316, 414, 607, 790, 879, 965, 1132, 0},
463 };
464 
465 static const u16 expected_tpt_mimo2_40MHz[4][IWL_RATE_COUNT] = {
466 	{0, 0, 0, 0, 123, 0, 182, 214, 235,  264,  279,  285,  289,  296,  300},
467 	{0, 0, 0, 0, 131, 0, 191, 222, 242,  270,  284,  289,  293,  300,  303},
468 	{0, 0, 0, 0, 200, 0, 390, 571, 741, 1067, 1365, 1505, 1640, 1894, 2053},
469 	{0, 0, 0, 0, 221, 0, 430, 630, 816, 1169, 1490, 1641, 1784, 2053, 2221},
470 };
471 
472 static const u16 expected_tpt_mimo2_80MHz[4][IWL_RATE_COUNT] = {
473 	{0, 0, 0, 0, 182, 0, 240,  264,  278,  299,  308,  311,  313,  317,  319},
474 	{0, 0, 0, 0, 190, 0, 247,  269,  282,  302,  310,  313,  315,  319,  320},
475 	{0, 0, 0, 0, 428, 0, 833, 1215, 1577, 2254, 2863, 3147, 3418, 3913, 4219},
476 	{0, 0, 0, 0, 474, 0, 920, 1338, 1732, 2464, 3116, 3418, 3705, 4225, 4545},
477 };
478 
479 static const u16 expected_tpt_mimo2_160MHz[4][IWL_RATE_COUNT] = {
480 	{0, 0, 0, 0, 240, 0, 278,  308,  313,  319,  322,  324,  328,  330,   334},
481 	{0, 0, 0, 0, 247, 0, 282,  310,  315,  320,  323,  325,  329,  332,   338},
482 	{0, 0, 0, 0, 875, 0, 1735, 2582, 3414, 5043, 6619, 7389, 8147, 9629,  10592},
483 	{0, 0, 0, 0, 971, 0, 1925, 2861, 3779, 5574, 7304, 8147, 8976, 10592, 11640},
484 };
485 
486 /* mbps, mcs */
487 static const struct iwl_rate_mcs_info iwl_rate_mcs[IWL_RATE_COUNT] = {
488 	{  "1", "BPSK DSSS"},
489 	{  "2", "QPSK DSSS"},
490 	{"5.5", "BPSK CCK"},
491 	{ "11", "QPSK CCK"},
492 	{  "6", "BPSK 1/2"},
493 	{  "9", "BPSK 1/2"},
494 	{ "12", "QPSK 1/2"},
495 	{ "18", "QPSK 3/4"},
496 	{ "24", "16QAM 1/2"},
497 	{ "36", "16QAM 3/4"},
498 	{ "48", "64QAM 2/3"},
499 	{ "54", "64QAM 3/4"},
500 	{ "60", "64QAM 5/6"},
501 };
502 
503 #define MCS_INDEX_PER_STREAM	(8)
504 
505 static const char *rs_pretty_ant(u8 ant)
506 {
507 	static const char * const ant_name[] = {
508 		[ANT_NONE] = "None",
509 		[ANT_A]    = "A",
510 		[ANT_B]    = "B",
511 		[ANT_AB]   = "AB",
512 		[ANT_C]    = "C",
513 		[ANT_AC]   = "AC",
514 		[ANT_BC]   = "BC",
515 		[ANT_ABC]  = "ABC",
516 	};
517 
518 	if (ant > ANT_ABC)
519 		return "UNKNOWN";
520 
521 	return ant_name[ant];
522 }
523 
524 static const char *rs_pretty_lq_type(enum iwl_table_type type)
525 {
526 	static const char * const lq_types[] = {
527 		[LQ_NONE] = "NONE",
528 		[LQ_LEGACY_A] = "LEGACY_A",
529 		[LQ_LEGACY_G] = "LEGACY_G",
530 		[LQ_HT_SISO] = "HT SISO",
531 		[LQ_HT_MIMO2] = "HT MIMO",
532 		[LQ_VHT_SISO] = "VHT SISO",
533 		[LQ_VHT_MIMO2] = "VHT MIMO",
534 	};
535 
536 	if (type < LQ_NONE || type >= LQ_MAX)
537 		return "UNKNOWN";
538 
539 	return lq_types[type];
540 }
541 
542 static char *rs_pretty_rate(const struct rs_rate *rate)
543 {
544 	static char buf[40];
545 	static const char * const legacy_rates[] = {
546 		[IWL_RATE_1M_INDEX] = "1M",
547 		[IWL_RATE_2M_INDEX] = "2M",
548 		[IWL_RATE_5M_INDEX] = "5.5M",
549 		[IWL_RATE_11M_INDEX] = "11M",
550 		[IWL_RATE_6M_INDEX] = "6M",
551 		[IWL_RATE_9M_INDEX] = "9M",
552 		[IWL_RATE_12M_INDEX] = "12M",
553 		[IWL_RATE_18M_INDEX] = "18M",
554 		[IWL_RATE_24M_INDEX] = "24M",
555 		[IWL_RATE_36M_INDEX] = "36M",
556 		[IWL_RATE_48M_INDEX] = "48M",
557 		[IWL_RATE_54M_INDEX] = "54M",
558 	};
559 	static const char *const ht_vht_rates[] = {
560 		[IWL_RATE_MCS_0_INDEX] = "MCS0",
561 		[IWL_RATE_MCS_1_INDEX] = "MCS1",
562 		[IWL_RATE_MCS_2_INDEX] = "MCS2",
563 		[IWL_RATE_MCS_3_INDEX] = "MCS3",
564 		[IWL_RATE_MCS_4_INDEX] = "MCS4",
565 		[IWL_RATE_MCS_5_INDEX] = "MCS5",
566 		[IWL_RATE_MCS_6_INDEX] = "MCS6",
567 		[IWL_RATE_MCS_7_INDEX] = "MCS7",
568 		[IWL_RATE_MCS_8_INDEX] = "MCS8",
569 		[IWL_RATE_MCS_9_INDEX] = "MCS9",
570 	};
571 	const char *rate_str;
572 
573 	if (is_type_legacy(rate->type) && (rate->index <= IWL_RATE_54M_INDEX))
574 		rate_str = legacy_rates[rate->index];
575 	else if ((is_type_ht(rate->type) || is_type_vht(rate->type)) &&
576 		 (rate->index >= IWL_RATE_MCS_0_INDEX) &&
577 		 (rate->index <= IWL_RATE_MCS_9_INDEX))
578 		rate_str = ht_vht_rates[rate->index];
579 	else
580 		rate_str = "BAD_RATE";
581 
582 	sprintf(buf, "(%s|%s|%s)", rs_pretty_lq_type(rate->type),
583 		rs_pretty_ant(rate->ant), rate_str);
584 	return buf;
585 }
586 
587 static inline void rs_dump_rate(struct iwl_mvm *mvm, const struct rs_rate *rate,
588 				const char *prefix)
589 {
590 	IWL_DEBUG_RATE(mvm,
591 		       "%s: %s BW: %d SGI: %d LDPC: %d STBC: %d\n",
592 		       prefix, rs_pretty_rate(rate), rate->bw,
593 		       rate->sgi, rate->ldpc, rate->stbc);
594 }
595 
596 static void rs_rate_scale_clear_window(struct iwl_rate_scale_data *window)
597 {
598 	window->data = 0;
599 	window->success_counter = 0;
600 	window->success_ratio = IWL_INVALID_VALUE;
601 	window->counter = 0;
602 	window->average_tpt = IWL_INVALID_VALUE;
603 }
604 
605 static void rs_rate_scale_clear_tbl_windows(struct iwl_mvm *mvm,
606 					    struct iwl_scale_tbl_info *tbl)
607 {
608 	int i;
609 
610 	IWL_DEBUG_RATE(mvm, "Clearing up window stats\n");
611 	for (i = 0; i < IWL_RATE_COUNT; i++)
612 		rs_rate_scale_clear_window(&tbl->win[i]);
613 
614 	for (i = 0; i < ARRAY_SIZE(tbl->tpc_win); i++)
615 		rs_rate_scale_clear_window(&tbl->tpc_win[i]);
616 }
617 
618 static inline u8 rs_is_valid_ant(u8 valid_antenna, u8 ant_type)
619 {
620 	return (ant_type & valid_antenna) == ant_type;
621 }
622 
623 static int rs_tl_turn_on_agg_for_tid(struct iwl_mvm *mvm,
624 				     struct iwl_lq_sta *lq_data, u8 tid,
625 				     struct ieee80211_sta *sta)
626 {
627 	int ret = -EAGAIN;
628 
629 	IWL_DEBUG_HT(mvm, "Starting Tx agg: STA: %pM tid: %d\n",
630 		     sta->addr, tid);
631 	ret = ieee80211_start_tx_ba_session(sta, tid, 5000);
632 	if (ret == -EAGAIN) {
633 		/*
634 		 * driver and mac80211 is out of sync
635 		 * this might be cause by reloading firmware
636 		 * stop the tx ba session here
637 		 */
638 		IWL_ERR(mvm, "Fail start Tx agg on tid: %d\n",
639 			tid);
640 		ieee80211_stop_tx_ba_session(sta, tid);
641 	}
642 	return ret;
643 }
644 
645 static void rs_tl_turn_on_agg(struct iwl_mvm *mvm, u8 tid,
646 			      struct iwl_lq_sta *lq_data,
647 			      struct ieee80211_sta *sta)
648 {
649 	if (tid < IWL_MAX_TID_COUNT)
650 		rs_tl_turn_on_agg_for_tid(mvm, lq_data, tid, sta);
651 	else
652 		IWL_ERR(mvm, "tid exceeds max TID count: %d/%d\n",
653 			tid, IWL_MAX_TID_COUNT);
654 }
655 
656 static inline int get_num_of_ant_from_rate(u32 rate_n_flags)
657 {
658 	return !!(rate_n_flags & RATE_MCS_ANT_A_MSK) +
659 	       !!(rate_n_flags & RATE_MCS_ANT_B_MSK) +
660 	       !!(rate_n_flags & RATE_MCS_ANT_C_MSK);
661 }
662 
663 /*
664  * Static function to get the expected throughput from an iwl_scale_tbl_info
665  * that wraps a NULL pointer check
666  */
667 static s32 get_expected_tpt(struct iwl_scale_tbl_info *tbl, int rs_index)
668 {
669 	if (tbl->expected_tpt)
670 		return tbl->expected_tpt[rs_index];
671 	return 0;
672 }
673 
674 /**
675  * rs_collect_tx_data - Update the success/failure sliding window
676  *
677  * We keep a sliding window of the last 62 packets transmitted
678  * at this rate.  window->data contains the bitmask of successful
679  * packets.
680  */
681 static int _rs_collect_tx_data(struct iwl_mvm *mvm,
682 			       struct iwl_scale_tbl_info *tbl,
683 			       int scale_index, int attempts, int successes,
684 			       struct iwl_rate_scale_data *window)
685 {
686 	static const u64 mask = (((u64)1) << (IWL_RATE_MAX_WINDOW - 1));
687 	s32 fail_count, tpt;
688 
689 	/* Get expected throughput */
690 	tpt = get_expected_tpt(tbl, scale_index);
691 
692 	/*
693 	 * Keep track of only the latest 62 tx frame attempts in this rate's
694 	 * history window; anything older isn't really relevant any more.
695 	 * If we have filled up the sliding window, drop the oldest attempt;
696 	 * if the oldest attempt (highest bit in bitmap) shows "success",
697 	 * subtract "1" from the success counter (this is the main reason
698 	 * we keep these bitmaps!).
699 	 */
700 	while (attempts > 0) {
701 		if (window->counter >= IWL_RATE_MAX_WINDOW) {
702 			/* remove earliest */
703 			window->counter = IWL_RATE_MAX_WINDOW - 1;
704 
705 			if (window->data & mask) {
706 				window->data &= ~mask;
707 				window->success_counter--;
708 			}
709 		}
710 
711 		/* Increment frames-attempted counter */
712 		window->counter++;
713 
714 		/* Shift bitmap by one frame to throw away oldest history */
715 		window->data <<= 1;
716 
717 		/* Mark the most recent #successes attempts as successful */
718 		if (successes > 0) {
719 			window->success_counter++;
720 			window->data |= 0x1;
721 			successes--;
722 		}
723 
724 		attempts--;
725 	}
726 
727 	/* Calculate current success ratio, avoid divide-by-0! */
728 	if (window->counter > 0)
729 		window->success_ratio = 128 * (100 * window->success_counter)
730 					/ window->counter;
731 	else
732 		window->success_ratio = IWL_INVALID_VALUE;
733 
734 	fail_count = window->counter - window->success_counter;
735 
736 	/* Calculate average throughput, if we have enough history. */
737 	if ((fail_count >= IWL_MVM_RS_RATE_MIN_FAILURE_TH) ||
738 	    (window->success_counter >= IWL_MVM_RS_RATE_MIN_SUCCESS_TH))
739 		window->average_tpt = (window->success_ratio * tpt + 64) / 128;
740 	else
741 		window->average_tpt = IWL_INVALID_VALUE;
742 
743 	return 0;
744 }
745 
746 static int rs_collect_tpc_data(struct iwl_mvm *mvm,
747 			       struct iwl_lq_sta *lq_sta,
748 			       struct iwl_scale_tbl_info *tbl,
749 			       int scale_index, int attempts, int successes,
750 			       u8 reduced_txp)
751 {
752 	struct iwl_rate_scale_data *window = NULL;
753 
754 	if (WARN_ON_ONCE(reduced_txp > TPC_MAX_REDUCTION))
755 		return -EINVAL;
756 
757 	window = &tbl->tpc_win[reduced_txp];
758 	return  _rs_collect_tx_data(mvm, tbl, scale_index, attempts, successes,
759 				    window);
760 }
761 
762 static int rs_collect_tlc_data(struct iwl_mvm *mvm,
763 			       struct iwl_lq_sta *lq_sta,
764 			       struct iwl_scale_tbl_info *tbl,
765 			       int scale_index, int attempts, int successes)
766 {
767 	struct iwl_rate_scale_data *window = NULL;
768 
769 	if (scale_index < 0 || scale_index >= IWL_RATE_COUNT)
770 		return -EINVAL;
771 
772 	if (tbl->column != RS_COLUMN_INVALID) {
773 		struct lq_sta_pers *pers = &lq_sta->pers;
774 
775 		pers->tx_stats[tbl->column][scale_index].total += attempts;
776 		pers->tx_stats[tbl->column][scale_index].success += successes;
777 	}
778 
779 	/* Select window for current tx bit rate */
780 	window = &(tbl->win[scale_index]);
781 	return _rs_collect_tx_data(mvm, tbl, scale_index, attempts, successes,
782 				   window);
783 }
784 
785 /* Convert rs_rate object into ucode rate bitmask */
786 static u32 ucode_rate_from_rs_rate(struct iwl_mvm *mvm,
787 				  struct rs_rate *rate)
788 {
789 	u32 ucode_rate = 0;
790 	int index = rate->index;
791 
792 	ucode_rate |= ((rate->ant << RATE_MCS_ANT_POS) &
793 			 RATE_MCS_ANT_ABC_MSK);
794 
795 	if (is_legacy(rate)) {
796 		ucode_rate |= iwl_rates[index].plcp;
797 		if (index >= IWL_FIRST_CCK_RATE && index <= IWL_LAST_CCK_RATE)
798 			ucode_rate |= RATE_MCS_CCK_MSK;
799 		return ucode_rate;
800 	}
801 
802 	if (is_ht(rate)) {
803 		if (index < IWL_FIRST_HT_RATE || index > IWL_LAST_HT_RATE) {
804 			IWL_ERR(mvm, "Invalid HT rate index %d\n", index);
805 			index = IWL_LAST_HT_RATE;
806 		}
807 		ucode_rate |= RATE_MCS_HT_MSK;
808 
809 		if (is_ht_siso(rate))
810 			ucode_rate |= iwl_rates[index].plcp_ht_siso;
811 		else if (is_ht_mimo2(rate))
812 			ucode_rate |= iwl_rates[index].plcp_ht_mimo2;
813 		else
814 			WARN_ON_ONCE(1);
815 	} else if (is_vht(rate)) {
816 		if (index < IWL_FIRST_VHT_RATE || index > IWL_LAST_VHT_RATE) {
817 			IWL_ERR(mvm, "Invalid VHT rate index %d\n", index);
818 			index = IWL_LAST_VHT_RATE;
819 		}
820 		ucode_rate |= RATE_MCS_VHT_MSK;
821 		if (is_vht_siso(rate))
822 			ucode_rate |= iwl_rates[index].plcp_vht_siso;
823 		else if (is_vht_mimo2(rate))
824 			ucode_rate |= iwl_rates[index].plcp_vht_mimo2;
825 		else
826 			WARN_ON_ONCE(1);
827 
828 	} else {
829 		IWL_ERR(mvm, "Invalid rate->type %d\n", rate->type);
830 	}
831 
832 	if (is_siso(rate) && rate->stbc) {
833 		/* To enable STBC we need to set both a flag and ANT_AB */
834 		ucode_rate |= RATE_MCS_ANT_AB_MSK;
835 		ucode_rate |= RATE_MCS_VHT_STBC_MSK;
836 	}
837 
838 	ucode_rate |= rate->bw;
839 	if (rate->sgi)
840 		ucode_rate |= RATE_MCS_SGI_MSK;
841 	if (rate->ldpc)
842 		ucode_rate |= RATE_MCS_LDPC_MSK;
843 
844 	return ucode_rate;
845 }
846 
847 /* Convert a ucode rate into an rs_rate object */
848 static int rs_rate_from_ucode_rate(const u32 ucode_rate,
849 				   enum nl80211_band band,
850 				   struct rs_rate *rate)
851 {
852 	u32 ant_msk = ucode_rate & RATE_MCS_ANT_ABC_MSK;
853 	u8 num_of_ant = get_num_of_ant_from_rate(ucode_rate);
854 	u8 nss;
855 
856 	memset(rate, 0, sizeof(*rate));
857 	rate->index = iwl_hwrate_to_plcp_idx(ucode_rate);
858 
859 	if (rate->index == IWL_RATE_INVALID)
860 		return -EINVAL;
861 
862 	rate->ant = (ant_msk >> RATE_MCS_ANT_POS);
863 
864 	/* Legacy */
865 	if (!(ucode_rate & RATE_MCS_HT_MSK) &&
866 	    !(ucode_rate & RATE_MCS_VHT_MSK)) {
867 		if (num_of_ant == 1) {
868 			if (band == NL80211_BAND_5GHZ)
869 				rate->type = LQ_LEGACY_A;
870 			else
871 				rate->type = LQ_LEGACY_G;
872 		}
873 
874 		return 0;
875 	}
876 
877 	/* HT or VHT */
878 	if (ucode_rate & RATE_MCS_SGI_MSK)
879 		rate->sgi = true;
880 	if (ucode_rate & RATE_MCS_LDPC_MSK)
881 		rate->ldpc = true;
882 	if (ucode_rate & RATE_MCS_VHT_STBC_MSK)
883 		rate->stbc = true;
884 	if (ucode_rate & RATE_MCS_BF_MSK)
885 		rate->bfer = true;
886 
887 	rate->bw = ucode_rate & RATE_MCS_CHAN_WIDTH_MSK;
888 
889 	if (ucode_rate & RATE_MCS_HT_MSK) {
890 		nss = ((ucode_rate & RATE_HT_MCS_NSS_MSK) >>
891 		       RATE_HT_MCS_NSS_POS) + 1;
892 
893 		if (nss == 1) {
894 			rate->type = LQ_HT_SISO;
895 			WARN_ONCE(!rate->stbc && !rate->bfer && num_of_ant != 1,
896 				  "stbc %d bfer %d",
897 				  rate->stbc, rate->bfer);
898 		} else if (nss == 2) {
899 			rate->type = LQ_HT_MIMO2;
900 			WARN_ON_ONCE(num_of_ant != 2);
901 		} else {
902 			WARN_ON_ONCE(1);
903 		}
904 	} else if (ucode_rate & RATE_MCS_VHT_MSK) {
905 		nss = ((ucode_rate & RATE_VHT_MCS_NSS_MSK) >>
906 		       RATE_VHT_MCS_NSS_POS) + 1;
907 
908 		if (nss == 1) {
909 			rate->type = LQ_VHT_SISO;
910 			WARN_ONCE(!rate->stbc && !rate->bfer && num_of_ant != 1,
911 				  "stbc %d bfer %d",
912 				  rate->stbc, rate->bfer);
913 		} else if (nss == 2) {
914 			rate->type = LQ_VHT_MIMO2;
915 			WARN_ON_ONCE(num_of_ant != 2);
916 		} else {
917 			WARN_ON_ONCE(1);
918 		}
919 	}
920 
921 	WARN_ON_ONCE(rate->bw == RATE_MCS_CHAN_WIDTH_80 &&
922 		     !is_vht(rate));
923 
924 	return 0;
925 }
926 
927 /* switch to another antenna/antennas and return 1 */
928 /* if no other valid antenna found, return 0 */
929 static int rs_toggle_antenna(u32 valid_ant, struct rs_rate *rate)
930 {
931 	u8 new_ant_type;
932 
933 	if (!rate->ant || rate->ant > ANT_ABC)
934 		return 0;
935 
936 	if (!rs_is_valid_ant(valid_ant, rate->ant))
937 		return 0;
938 
939 	new_ant_type = ant_toggle_lookup[rate->ant];
940 
941 	while ((new_ant_type != rate->ant) &&
942 	       !rs_is_valid_ant(valid_ant, new_ant_type))
943 		new_ant_type = ant_toggle_lookup[new_ant_type];
944 
945 	if (new_ant_type == rate->ant)
946 		return 0;
947 
948 	rate->ant = new_ant_type;
949 
950 	return 1;
951 }
952 
953 static u16 rs_get_supported_rates(struct iwl_lq_sta *lq_sta,
954 				  struct rs_rate *rate)
955 {
956 	if (is_legacy(rate))
957 		return lq_sta->active_legacy_rate;
958 	else if (is_siso(rate))
959 		return lq_sta->active_siso_rate;
960 	else if (is_mimo2(rate))
961 		return lq_sta->active_mimo2_rate;
962 
963 	WARN_ON_ONCE(1);
964 	return 0;
965 }
966 
967 static u16 rs_get_adjacent_rate(struct iwl_mvm *mvm, u8 index, u16 rate_mask,
968 				int rate_type)
969 {
970 	u8 high = IWL_RATE_INVALID;
971 	u8 low = IWL_RATE_INVALID;
972 
973 	/* 802.11A or ht walks to the next literal adjacent rate in
974 	 * the rate table */
975 	if (is_type_a_band(rate_type) || !is_type_legacy(rate_type)) {
976 		int i;
977 		u32 mask;
978 
979 		/* Find the previous rate that is in the rate mask */
980 		i = index - 1;
981 		for (mask = (1 << i); i >= 0; i--, mask >>= 1) {
982 			if (rate_mask & mask) {
983 				low = i;
984 				break;
985 			}
986 		}
987 
988 		/* Find the next rate that is in the rate mask */
989 		i = index + 1;
990 		for (mask = (1 << i); i < IWL_RATE_COUNT; i++, mask <<= 1) {
991 			if (rate_mask & mask) {
992 				high = i;
993 				break;
994 			}
995 		}
996 
997 		return (high << 8) | low;
998 	}
999 
1000 	low = index;
1001 	while (low != IWL_RATE_INVALID) {
1002 		low = iwl_rates[low].prev_rs;
1003 		if (low == IWL_RATE_INVALID)
1004 			break;
1005 		if (rate_mask & (1 << low))
1006 			break;
1007 	}
1008 
1009 	high = index;
1010 	while (high != IWL_RATE_INVALID) {
1011 		high = iwl_rates[high].next_rs;
1012 		if (high == IWL_RATE_INVALID)
1013 			break;
1014 		if (rate_mask & (1 << high))
1015 			break;
1016 	}
1017 
1018 	return (high << 8) | low;
1019 }
1020 
1021 static inline bool rs_rate_supported(struct iwl_lq_sta *lq_sta,
1022 				     struct rs_rate *rate)
1023 {
1024 	return BIT(rate->index) & rs_get_supported_rates(lq_sta, rate);
1025 }
1026 
1027 /* Get the next supported lower rate in the current column.
1028  * Return true if bottom rate in the current column was reached
1029  */
1030 static bool rs_get_lower_rate_in_column(struct iwl_lq_sta *lq_sta,
1031 					struct rs_rate *rate)
1032 {
1033 	u8 low;
1034 	u16 high_low;
1035 	u16 rate_mask;
1036 	struct iwl_mvm *mvm = lq_sta->pers.drv;
1037 
1038 	rate_mask = rs_get_supported_rates(lq_sta, rate);
1039 	high_low = rs_get_adjacent_rate(mvm, rate->index, rate_mask,
1040 					rate->type);
1041 	low = high_low & 0xff;
1042 
1043 	/* Bottom rate of column reached */
1044 	if (low == IWL_RATE_INVALID)
1045 		return true;
1046 
1047 	rate->index = low;
1048 	return false;
1049 }
1050 
1051 /* Get the next rate to use following a column downgrade */
1052 static void rs_get_lower_rate_down_column(struct iwl_lq_sta *lq_sta,
1053 					  struct rs_rate *rate)
1054 {
1055 	struct iwl_mvm *mvm = lq_sta->pers.drv;
1056 
1057 	if (is_legacy(rate)) {
1058 		/* No column to downgrade from Legacy */
1059 		return;
1060 	} else if (is_siso(rate)) {
1061 		/* Downgrade to Legacy if we were in SISO */
1062 		if (lq_sta->band == NL80211_BAND_5GHZ)
1063 			rate->type = LQ_LEGACY_A;
1064 		else
1065 			rate->type = LQ_LEGACY_G;
1066 
1067 		rate->bw = RATE_MCS_CHAN_WIDTH_20;
1068 
1069 		WARN_ON_ONCE(rate->index < IWL_RATE_MCS_0_INDEX ||
1070 			     rate->index > IWL_RATE_MCS_9_INDEX);
1071 
1072 		rate->index = rs_ht_to_legacy[rate->index];
1073 		rate->ldpc = false;
1074 	} else {
1075 		/* Downgrade to SISO with same MCS if in MIMO  */
1076 		rate->type = is_vht_mimo2(rate) ?
1077 			LQ_VHT_SISO : LQ_HT_SISO;
1078 	}
1079 
1080 	if (num_of_ant(rate->ant) > 1)
1081 		rate->ant = first_antenna(iwl_mvm_get_valid_tx_ant(mvm));
1082 
1083 	/* Relevant in both switching to SISO or Legacy */
1084 	rate->sgi = false;
1085 
1086 	if (!rs_rate_supported(lq_sta, rate))
1087 		rs_get_lower_rate_in_column(lq_sta, rate);
1088 }
1089 
1090 /* Check if both rates are identical
1091  * allow_ant_mismatch enables matching a SISO rate on ANT_A or ANT_B
1092  * with a rate indicating STBC/BFER and ANT_AB.
1093  */
1094 static inline bool rs_rate_equal(struct rs_rate *a,
1095 				 struct rs_rate *b,
1096 				 bool allow_ant_mismatch)
1097 
1098 {
1099 	bool ant_match = (a->ant == b->ant) && (a->stbc == b->stbc) &&
1100 		(a->bfer == b->bfer);
1101 
1102 	if (allow_ant_mismatch) {
1103 		if (a->stbc || a->bfer) {
1104 			WARN_ONCE(a->ant != ANT_AB, "stbc %d bfer %d ant %d",
1105 				  a->stbc, a->bfer, a->ant);
1106 			ant_match |= (b->ant == ANT_A || b->ant == ANT_B);
1107 		} else if (b->stbc || b->bfer) {
1108 			WARN_ONCE(b->ant != ANT_AB, "stbc %d bfer %d ant %d",
1109 				  b->stbc, b->bfer, b->ant);
1110 			ant_match |= (a->ant == ANT_A || a->ant == ANT_B);
1111 		}
1112 	}
1113 
1114 	return (a->type == b->type) && (a->bw == b->bw) && (a->sgi == b->sgi) &&
1115 		(a->ldpc == b->ldpc) && (a->index == b->index) && ant_match;
1116 }
1117 
1118 /* Check if both rates share the same column */
1119 static inline bool rs_rate_column_match(struct rs_rate *a,
1120 					struct rs_rate *b)
1121 {
1122 	bool ant_match;
1123 
1124 	if (a->stbc || a->bfer)
1125 		ant_match = (b->ant == ANT_A || b->ant == ANT_B);
1126 	else
1127 		ant_match = (a->ant == b->ant);
1128 
1129 	return (a->type == b->type) && (a->bw == b->bw) && (a->sgi == b->sgi)
1130 		&& ant_match;
1131 }
1132 
1133 static inline enum rs_column rs_get_column_from_rate(struct rs_rate *rate)
1134 {
1135 	if (is_legacy(rate)) {
1136 		if (rate->ant == ANT_A)
1137 			return RS_COLUMN_LEGACY_ANT_A;
1138 
1139 		if (rate->ant == ANT_B)
1140 			return RS_COLUMN_LEGACY_ANT_B;
1141 
1142 		goto err;
1143 	}
1144 
1145 	if (is_siso(rate)) {
1146 		if (rate->ant == ANT_A || rate->stbc || rate->bfer)
1147 			return rate->sgi ? RS_COLUMN_SISO_ANT_A_SGI :
1148 				RS_COLUMN_SISO_ANT_A;
1149 
1150 		if (rate->ant == ANT_B)
1151 			return rate->sgi ? RS_COLUMN_SISO_ANT_B_SGI :
1152 				RS_COLUMN_SISO_ANT_B;
1153 
1154 		goto err;
1155 	}
1156 
1157 	if (is_mimo(rate))
1158 		return rate->sgi ? RS_COLUMN_MIMO2_SGI : RS_COLUMN_MIMO2;
1159 
1160 err:
1161 	return RS_COLUMN_INVALID;
1162 }
1163 
1164 static u8 rs_get_tid(struct ieee80211_hdr *hdr)
1165 {
1166 	u8 tid = IWL_MAX_TID_COUNT;
1167 
1168 	if (ieee80211_is_data_qos(hdr->frame_control)) {
1169 		u8 *qc = ieee80211_get_qos_ctl(hdr);
1170 		tid = qc[0] & 0xf;
1171 	}
1172 
1173 	if (unlikely(tid > IWL_MAX_TID_COUNT))
1174 		tid = IWL_MAX_TID_COUNT;
1175 
1176 	return tid;
1177 }
1178 
1179 void iwl_mvm_rs_tx_status(struct iwl_mvm *mvm, struct ieee80211_sta *sta,
1180 			  int tid, struct ieee80211_tx_info *info, bool ndp)
1181 {
1182 	int legacy_success;
1183 	int retries;
1184 	int i;
1185 	struct iwl_lq_cmd *table;
1186 	u32 lq_hwrate;
1187 	struct rs_rate lq_rate, tx_resp_rate;
1188 	struct iwl_scale_tbl_info *curr_tbl, *other_tbl, *tmp_tbl;
1189 	u8 reduced_txp = (uintptr_t)info->status.status_driver_data[0];
1190 	u32 tx_resp_hwrate = (uintptr_t)info->status.status_driver_data[1];
1191 	struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
1192 	struct iwl_lq_sta *lq_sta = &mvmsta->lq_sta;
1193 	bool allow_ant_mismatch = fw_has_api(&mvm->fw->ucode_capa,
1194 					     IWL_UCODE_TLV_API_LQ_SS_PARAMS);
1195 
1196 	/* Treat uninitialized rate scaling data same as non-existing. */
1197 	if (!lq_sta) {
1198 		IWL_DEBUG_RATE(mvm, "Station rate scaling not created yet.\n");
1199 		return;
1200 	} else if (!lq_sta->pers.drv) {
1201 		IWL_DEBUG_RATE(mvm, "Rate scaling not initialized yet.\n");
1202 		return;
1203 	}
1204 
1205 	/* This packet was aggregated but doesn't carry status info */
1206 	if ((info->flags & IEEE80211_TX_CTL_AMPDU) &&
1207 	    !(info->flags & IEEE80211_TX_STAT_AMPDU))
1208 		return;
1209 
1210 	rs_rate_from_ucode_rate(tx_resp_hwrate, info->band, &tx_resp_rate);
1211 
1212 #ifdef CONFIG_MAC80211_DEBUGFS
1213 	/* Disable last tx check if we are debugging with fixed rate but
1214 	 * update tx stats */
1215 	if (lq_sta->pers.dbg_fixed_rate) {
1216 		int index = tx_resp_rate.index;
1217 		enum rs_column column;
1218 		int attempts, success;
1219 
1220 		column = rs_get_column_from_rate(&tx_resp_rate);
1221 		if (WARN_ONCE(column == RS_COLUMN_INVALID,
1222 			      "Can't map rate 0x%x to column",
1223 			      tx_resp_hwrate))
1224 			return;
1225 
1226 		if (info->flags & IEEE80211_TX_STAT_AMPDU) {
1227 			attempts = info->status.ampdu_len;
1228 			success = info->status.ampdu_ack_len;
1229 		} else {
1230 			attempts = info->status.rates[0].count;
1231 			success = !!(info->flags & IEEE80211_TX_STAT_ACK);
1232 		}
1233 
1234 		lq_sta->pers.tx_stats[column][index].total += attempts;
1235 		lq_sta->pers.tx_stats[column][index].success += success;
1236 
1237 		IWL_DEBUG_RATE(mvm, "Fixed rate 0x%x success %d attempts %d\n",
1238 			       tx_resp_hwrate, success, attempts);
1239 		return;
1240 	}
1241 #endif
1242 
1243 	if (time_after(jiffies,
1244 		       (unsigned long)(lq_sta->last_tx +
1245 				       (IWL_MVM_RS_IDLE_TIMEOUT * HZ)))) {
1246 		int t;
1247 
1248 		IWL_DEBUG_RATE(mvm, "Tx idle for too long. reinit rs\n");
1249 		for (t = 0; t < IWL_MAX_TID_COUNT; t++)
1250 			ieee80211_stop_tx_ba_session(sta, t);
1251 
1252 		iwl_mvm_rs_rate_init(mvm, sta, info->band, false);
1253 		return;
1254 	}
1255 	lq_sta->last_tx = jiffies;
1256 
1257 	/* Ignore this Tx frame response if its initial rate doesn't match
1258 	 * that of latest Link Quality command.  There may be stragglers
1259 	 * from a previous Link Quality command, but we're no longer interested
1260 	 * in those; they're either from the "active" mode while we're trying
1261 	 * to check "search" mode, or a prior "search" mode after we've moved
1262 	 * to a new "search" mode (which might become the new "active" mode).
1263 	 */
1264 	table = &lq_sta->lq;
1265 	lq_hwrate = le32_to_cpu(table->rs_table[0]);
1266 	rs_rate_from_ucode_rate(lq_hwrate, info->band, &lq_rate);
1267 
1268 	/* Here we actually compare this rate to the latest LQ command */
1269 	if (!rs_rate_equal(&tx_resp_rate, &lq_rate, allow_ant_mismatch)) {
1270 		IWL_DEBUG_RATE(mvm,
1271 			       "initial tx resp rate 0x%x does not match 0x%x\n",
1272 			       tx_resp_hwrate, lq_hwrate);
1273 
1274 		/*
1275 		 * Since rates mis-match, the last LQ command may have failed.
1276 		 * After IWL_MISSED_RATE_MAX mis-matches, resync the uCode with
1277 		 * ... driver.
1278 		 */
1279 		lq_sta->missed_rate_counter++;
1280 		if (lq_sta->missed_rate_counter > IWL_MVM_RS_MISSED_RATE_MAX) {
1281 			lq_sta->missed_rate_counter = 0;
1282 			IWL_DEBUG_RATE(mvm,
1283 				       "Too many rates mismatch. Send sync LQ. rs_state %d\n",
1284 				       lq_sta->rs_state);
1285 			iwl_mvm_send_lq_cmd(mvm, &lq_sta->lq, false);
1286 		}
1287 		/* Regardless, ignore this status info for outdated rate */
1288 		return;
1289 	} else
1290 		/* Rate did match, so reset the missed_rate_counter */
1291 		lq_sta->missed_rate_counter = 0;
1292 
1293 	if (!lq_sta->search_better_tbl) {
1294 		curr_tbl = &(lq_sta->lq_info[lq_sta->active_tbl]);
1295 		other_tbl = &(lq_sta->lq_info[1 - lq_sta->active_tbl]);
1296 	} else {
1297 		curr_tbl = &(lq_sta->lq_info[1 - lq_sta->active_tbl]);
1298 		other_tbl = &(lq_sta->lq_info[lq_sta->active_tbl]);
1299 	}
1300 
1301 	if (WARN_ON_ONCE(!rs_rate_column_match(&lq_rate, &curr_tbl->rate))) {
1302 		IWL_DEBUG_RATE(mvm,
1303 			       "Neither active nor search matches tx rate\n");
1304 		tmp_tbl = &(lq_sta->lq_info[lq_sta->active_tbl]);
1305 		rs_dump_rate(mvm, &tmp_tbl->rate, "ACTIVE");
1306 		tmp_tbl = &(lq_sta->lq_info[1 - lq_sta->active_tbl]);
1307 		rs_dump_rate(mvm, &tmp_tbl->rate, "SEARCH");
1308 		rs_dump_rate(mvm, &lq_rate, "ACTUAL");
1309 
1310 		/*
1311 		 * no matching table found, let's by-pass the data collection
1312 		 * and continue to perform rate scale to find the rate table
1313 		 */
1314 		rs_stay_in_table(lq_sta, true);
1315 		goto done;
1316 	}
1317 
1318 	/*
1319 	 * Updating the frame history depends on whether packets were
1320 	 * aggregated.
1321 	 *
1322 	 * For aggregation, all packets were transmitted at the same rate, the
1323 	 * first index into rate scale table.
1324 	 */
1325 	if (info->flags & IEEE80211_TX_STAT_AMPDU) {
1326 		rs_collect_tpc_data(mvm, lq_sta, curr_tbl, lq_rate.index,
1327 				    info->status.ampdu_len,
1328 				    info->status.ampdu_ack_len,
1329 				    reduced_txp);
1330 
1331 		/* ampdu_ack_len = 0 marks no BA was received. For TLC, treat
1332 		 * it as a single frame loss as we don't want the success ratio
1333 		 * to dip too quickly because a BA wasn't received.
1334 		 * For TPC, there's no need for this optimisation since we want
1335 		 * to recover very quickly from a bad power reduction and,
1336 		 * therefore we'd like the success ratio to get an immediate hit
1337 		 * when failing to get a BA, so we'd switch back to a lower or
1338 		 * zero power reduction. When FW transmits agg with a rate
1339 		 * different from the initial rate, it will not use reduced txp
1340 		 * and will send BA notification twice (one empty with reduced
1341 		 * txp equal to the value from LQ and one with reduced txp 0).
1342 		 * We need to update counters for each txp level accordingly.
1343 		 */
1344 		if (info->status.ampdu_ack_len == 0)
1345 			info->status.ampdu_len = 1;
1346 
1347 		rs_collect_tlc_data(mvm, lq_sta, curr_tbl, lq_rate.index,
1348 				    info->status.ampdu_len,
1349 				    info->status.ampdu_ack_len);
1350 
1351 		/* Update success/fail counts if not searching for new mode */
1352 		if (lq_sta->rs_state == RS_STATE_STAY_IN_COLUMN) {
1353 			lq_sta->total_success += info->status.ampdu_ack_len;
1354 			lq_sta->total_failed += (info->status.ampdu_len -
1355 					info->status.ampdu_ack_len);
1356 		}
1357 	} else {
1358 		/* For legacy, update frame history with for each Tx retry. */
1359 		retries = info->status.rates[0].count - 1;
1360 		/* HW doesn't send more than 15 retries */
1361 		retries = min(retries, 15);
1362 
1363 		/* The last transmission may have been successful */
1364 		legacy_success = !!(info->flags & IEEE80211_TX_STAT_ACK);
1365 		/* Collect data for each rate used during failed TX attempts */
1366 		for (i = 0; i <= retries; ++i) {
1367 			lq_hwrate = le32_to_cpu(table->rs_table[i]);
1368 			rs_rate_from_ucode_rate(lq_hwrate, info->band,
1369 						&lq_rate);
1370 			/*
1371 			 * Only collect stats if retried rate is in the same RS
1372 			 * table as active/search.
1373 			 */
1374 			if (rs_rate_column_match(&lq_rate, &curr_tbl->rate))
1375 				tmp_tbl = curr_tbl;
1376 			else if (rs_rate_column_match(&lq_rate,
1377 						      &other_tbl->rate))
1378 				tmp_tbl = other_tbl;
1379 			else
1380 				continue;
1381 
1382 			rs_collect_tpc_data(mvm, lq_sta, tmp_tbl,
1383 					    lq_rate.index, 1,
1384 					    i < retries ? 0 : legacy_success,
1385 					    reduced_txp);
1386 			rs_collect_tlc_data(mvm, lq_sta, tmp_tbl,
1387 					    lq_rate.index, 1,
1388 					    i < retries ? 0 : legacy_success);
1389 		}
1390 
1391 		/* Update success/fail counts if not searching for new mode */
1392 		if (lq_sta->rs_state == RS_STATE_STAY_IN_COLUMN) {
1393 			lq_sta->total_success += legacy_success;
1394 			lq_sta->total_failed += retries + (1 - legacy_success);
1395 		}
1396 	}
1397 	/* The last TX rate is cached in lq_sta; it's set in if/else above */
1398 	lq_sta->last_rate_n_flags = lq_hwrate;
1399 	IWL_DEBUG_RATE(mvm, "reduced txpower: %d\n", reduced_txp);
1400 done:
1401 	/* See if there's a better rate or modulation mode to try. */
1402 	if (sta->supp_rates[info->band])
1403 		rs_rate_scale_perform(mvm, sta, lq_sta, tid, ndp);
1404 }
1405 
1406 /*
1407  * mac80211 sends us Tx status
1408  */
1409 static void rs_mac80211_tx_status(void *mvm_r,
1410 				  struct ieee80211_supported_band *sband,
1411 				  struct ieee80211_sta *sta, void *priv_sta,
1412 				  struct sk_buff *skb)
1413 {
1414 	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
1415 	struct iwl_op_mode *op_mode = (struct iwl_op_mode *)mvm_r;
1416 	struct iwl_mvm *mvm = IWL_OP_MODE_GET_MVM(op_mode);
1417 	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1418 
1419 	if (!iwl_mvm_sta_from_mac80211(sta)->vif)
1420 		return;
1421 
1422 	if (!ieee80211_is_data(hdr->frame_control) ||
1423 	    info->flags & IEEE80211_TX_CTL_NO_ACK)
1424 		return;
1425 
1426 	iwl_mvm_rs_tx_status(mvm, sta, rs_get_tid(hdr), info,
1427 			     ieee80211_is_qos_nullfunc(hdr->frame_control));
1428 }
1429 
1430 /*
1431  * Begin a period of staying with a selected modulation mode.
1432  * Set "stay_in_tbl" flag to prevent any mode switches.
1433  * Set frame tx success limits according to legacy vs. high-throughput,
1434  * and reset overall (spanning all rates) tx success history statistics.
1435  * These control how long we stay using same modulation mode before
1436  * searching for a new mode.
1437  */
1438 static void rs_set_stay_in_table(struct iwl_mvm *mvm, u8 is_legacy,
1439 				 struct iwl_lq_sta *lq_sta)
1440 {
1441 	IWL_DEBUG_RATE(mvm, "Moving to RS_STATE_STAY_IN_COLUMN\n");
1442 	lq_sta->rs_state = RS_STATE_STAY_IN_COLUMN;
1443 	if (is_legacy) {
1444 		lq_sta->table_count_limit = IWL_MVM_RS_LEGACY_TABLE_COUNT;
1445 		lq_sta->max_failure_limit = IWL_MVM_RS_LEGACY_FAILURE_LIMIT;
1446 		lq_sta->max_success_limit = IWL_MVM_RS_LEGACY_SUCCESS_LIMIT;
1447 	} else {
1448 		lq_sta->table_count_limit = IWL_MVM_RS_NON_LEGACY_TABLE_COUNT;
1449 		lq_sta->max_failure_limit = IWL_MVM_RS_NON_LEGACY_FAILURE_LIMIT;
1450 		lq_sta->max_success_limit = IWL_MVM_RS_NON_LEGACY_SUCCESS_LIMIT;
1451 	}
1452 	lq_sta->table_count = 0;
1453 	lq_sta->total_failed = 0;
1454 	lq_sta->total_success = 0;
1455 	lq_sta->flush_timer = jiffies;
1456 	lq_sta->visited_columns = 0;
1457 }
1458 
1459 static inline int rs_get_max_rate_from_mask(unsigned long rate_mask)
1460 {
1461 	if (rate_mask)
1462 		return find_last_bit(&rate_mask, BITS_PER_LONG);
1463 	return IWL_RATE_INVALID;
1464 }
1465 
1466 static int rs_get_max_allowed_rate(struct iwl_lq_sta *lq_sta,
1467 				   const struct rs_tx_column *column)
1468 {
1469 	switch (column->mode) {
1470 	case RS_LEGACY:
1471 		return lq_sta->max_legacy_rate_idx;
1472 	case RS_SISO:
1473 		return lq_sta->max_siso_rate_idx;
1474 	case RS_MIMO2:
1475 		return lq_sta->max_mimo2_rate_idx;
1476 	default:
1477 		WARN_ON_ONCE(1);
1478 	}
1479 
1480 	return lq_sta->max_legacy_rate_idx;
1481 }
1482 
1483 static const u16 *rs_get_expected_tpt_table(struct iwl_lq_sta *lq_sta,
1484 					    const struct rs_tx_column *column,
1485 					    u32 bw)
1486 {
1487 	/* Used to choose among HT tables */
1488 	const u16 (*ht_tbl_pointer)[IWL_RATE_COUNT];
1489 
1490 	if (WARN_ON_ONCE(column->mode != RS_LEGACY &&
1491 			 column->mode != RS_SISO &&
1492 			 column->mode != RS_MIMO2))
1493 		return expected_tpt_legacy;
1494 
1495 	/* Legacy rates have only one table */
1496 	if (column->mode == RS_LEGACY)
1497 		return expected_tpt_legacy;
1498 
1499 	ht_tbl_pointer = expected_tpt_mimo2_20MHz;
1500 	/* Choose among many HT tables depending on number of streams
1501 	 * (SISO/MIMO2), channel width (20/40/80), SGI, and aggregation
1502 	 * status */
1503 	if (column->mode == RS_SISO) {
1504 		switch (bw) {
1505 		case RATE_MCS_CHAN_WIDTH_20:
1506 			ht_tbl_pointer = expected_tpt_siso_20MHz;
1507 			break;
1508 		case RATE_MCS_CHAN_WIDTH_40:
1509 			ht_tbl_pointer = expected_tpt_siso_40MHz;
1510 			break;
1511 		case RATE_MCS_CHAN_WIDTH_80:
1512 			ht_tbl_pointer = expected_tpt_siso_80MHz;
1513 			break;
1514 		case RATE_MCS_CHAN_WIDTH_160:
1515 			ht_tbl_pointer = expected_tpt_siso_160MHz;
1516 			break;
1517 		default:
1518 			WARN_ON_ONCE(1);
1519 		}
1520 	} else if (column->mode == RS_MIMO2) {
1521 		switch (bw) {
1522 		case RATE_MCS_CHAN_WIDTH_20:
1523 			ht_tbl_pointer = expected_tpt_mimo2_20MHz;
1524 			break;
1525 		case RATE_MCS_CHAN_WIDTH_40:
1526 			ht_tbl_pointer = expected_tpt_mimo2_40MHz;
1527 			break;
1528 		case RATE_MCS_CHAN_WIDTH_80:
1529 			ht_tbl_pointer = expected_tpt_mimo2_80MHz;
1530 			break;
1531 		case RATE_MCS_CHAN_WIDTH_160:
1532 			ht_tbl_pointer = expected_tpt_mimo2_160MHz;
1533 			break;
1534 		default:
1535 			WARN_ON_ONCE(1);
1536 		}
1537 	} else {
1538 		WARN_ON_ONCE(1);
1539 	}
1540 
1541 	if (!column->sgi && !lq_sta->is_agg)		/* Normal */
1542 		return ht_tbl_pointer[0];
1543 	else if (column->sgi && !lq_sta->is_agg)        /* SGI */
1544 		return ht_tbl_pointer[1];
1545 	else if (!column->sgi && lq_sta->is_agg)        /* AGG */
1546 		return ht_tbl_pointer[2];
1547 	else						/* AGG+SGI */
1548 		return ht_tbl_pointer[3];
1549 }
1550 
1551 static void rs_set_expected_tpt_table(struct iwl_lq_sta *lq_sta,
1552 				      struct iwl_scale_tbl_info *tbl)
1553 {
1554 	struct rs_rate *rate = &tbl->rate;
1555 	const struct rs_tx_column *column = &rs_tx_columns[tbl->column];
1556 
1557 	tbl->expected_tpt = rs_get_expected_tpt_table(lq_sta, column, rate->bw);
1558 }
1559 
1560 static s32 rs_get_best_rate(struct iwl_mvm *mvm,
1561 			    struct iwl_lq_sta *lq_sta,
1562 			    struct iwl_scale_tbl_info *tbl,	/* "search" */
1563 			    unsigned long rate_mask, s8 index)
1564 {
1565 	struct iwl_scale_tbl_info *active_tbl =
1566 	    &(lq_sta->lq_info[lq_sta->active_tbl]);
1567 	s32 success_ratio = active_tbl->win[index].success_ratio;
1568 	u16 expected_current_tpt = active_tbl->expected_tpt[index];
1569 	const u16 *tpt_tbl = tbl->expected_tpt;
1570 	u16 high_low;
1571 	u32 target_tpt;
1572 	int rate_idx;
1573 
1574 	if (success_ratio >= RS_PERCENT(IWL_MVM_RS_SR_NO_DECREASE)) {
1575 		target_tpt = 100 * expected_current_tpt;
1576 		IWL_DEBUG_RATE(mvm,
1577 			       "SR %d high. Find rate exceeding EXPECTED_CURRENT %d\n",
1578 			       success_ratio, target_tpt);
1579 	} else {
1580 		target_tpt = lq_sta->last_tpt;
1581 		IWL_DEBUG_RATE(mvm,
1582 			       "SR %d not that good. Find rate exceeding ACTUAL_TPT %d\n",
1583 			       success_ratio, target_tpt);
1584 	}
1585 
1586 	rate_idx = find_first_bit(&rate_mask, BITS_PER_LONG);
1587 
1588 	while (rate_idx != IWL_RATE_INVALID) {
1589 		if (target_tpt < (100 * tpt_tbl[rate_idx]))
1590 			break;
1591 
1592 		high_low = rs_get_adjacent_rate(mvm, rate_idx, rate_mask,
1593 						tbl->rate.type);
1594 
1595 		rate_idx = (high_low >> 8) & 0xff;
1596 	}
1597 
1598 	IWL_DEBUG_RATE(mvm, "Best rate found %d target_tp %d expected_new %d\n",
1599 		       rate_idx, target_tpt,
1600 		       rate_idx != IWL_RATE_INVALID ?
1601 		       100 * tpt_tbl[rate_idx] : IWL_INVALID_VALUE);
1602 
1603 	return rate_idx;
1604 }
1605 
1606 static u32 rs_bw_from_sta_bw(struct ieee80211_sta *sta)
1607 {
1608 	switch (sta->bandwidth) {
1609 	case IEEE80211_STA_RX_BW_160:
1610 		return RATE_MCS_CHAN_WIDTH_160;
1611 	case IEEE80211_STA_RX_BW_80:
1612 		return RATE_MCS_CHAN_WIDTH_80;
1613 	case IEEE80211_STA_RX_BW_40:
1614 		return RATE_MCS_CHAN_WIDTH_40;
1615 	case IEEE80211_STA_RX_BW_20:
1616 	default:
1617 		return RATE_MCS_CHAN_WIDTH_20;
1618 	}
1619 }
1620 
1621 /*
1622  * Check whether we should continue using same modulation mode, or
1623  * begin search for a new mode, based on:
1624  * 1) # tx successes or failures while using this mode
1625  * 2) # times calling this function
1626  * 3) elapsed time in this mode (not used, for now)
1627  */
1628 static void rs_stay_in_table(struct iwl_lq_sta *lq_sta, bool force_search)
1629 {
1630 	struct iwl_scale_tbl_info *tbl;
1631 	int active_tbl;
1632 	int flush_interval_passed = 0;
1633 	struct iwl_mvm *mvm;
1634 
1635 	mvm = lq_sta->pers.drv;
1636 	active_tbl = lq_sta->active_tbl;
1637 
1638 	tbl = &(lq_sta->lq_info[active_tbl]);
1639 
1640 	/* If we've been disallowing search, see if we should now allow it */
1641 	if (lq_sta->rs_state == RS_STATE_STAY_IN_COLUMN) {
1642 		/* Elapsed time using current modulation mode */
1643 		if (lq_sta->flush_timer)
1644 			flush_interval_passed =
1645 				time_after(jiffies,
1646 					   (unsigned long)(lq_sta->flush_timer +
1647 							   (IWL_MVM_RS_STAY_IN_COLUMN_TIMEOUT * HZ)));
1648 
1649 		/*
1650 		 * Check if we should allow search for new modulation mode.
1651 		 * If many frames have failed or succeeded, or we've used
1652 		 * this same modulation for a long time, allow search, and
1653 		 * reset history stats that keep track of whether we should
1654 		 * allow a new search.  Also (below) reset all bitmaps and
1655 		 * stats in active history.
1656 		 */
1657 		if (force_search ||
1658 		    (lq_sta->total_failed > lq_sta->max_failure_limit) ||
1659 		    (lq_sta->total_success > lq_sta->max_success_limit) ||
1660 		    ((!lq_sta->search_better_tbl) &&
1661 		     (lq_sta->flush_timer) && (flush_interval_passed))) {
1662 			IWL_DEBUG_RATE(mvm,
1663 				       "LQ: stay is expired %d %d %d\n",
1664 				     lq_sta->total_failed,
1665 				     lq_sta->total_success,
1666 				     flush_interval_passed);
1667 
1668 			/* Allow search for new mode */
1669 			lq_sta->rs_state = RS_STATE_SEARCH_CYCLE_STARTED;
1670 			IWL_DEBUG_RATE(mvm,
1671 				       "Moving to RS_STATE_SEARCH_CYCLE_STARTED\n");
1672 			lq_sta->total_failed = 0;
1673 			lq_sta->total_success = 0;
1674 			lq_sta->flush_timer = 0;
1675 			/* mark the current column as visited */
1676 			lq_sta->visited_columns = BIT(tbl->column);
1677 		/*
1678 		 * Else if we've used this modulation mode enough repetitions
1679 		 * (regardless of elapsed time or success/failure), reset
1680 		 * history bitmaps and rate-specific stats for all rates in
1681 		 * active table.
1682 		 */
1683 		} else {
1684 			lq_sta->table_count++;
1685 			if (lq_sta->table_count >=
1686 			    lq_sta->table_count_limit) {
1687 				lq_sta->table_count = 0;
1688 
1689 				IWL_DEBUG_RATE(mvm,
1690 					       "LQ: stay in table clear win\n");
1691 				rs_rate_scale_clear_tbl_windows(mvm, tbl);
1692 			}
1693 		}
1694 
1695 		/* If transitioning to allow "search", reset all history
1696 		 * bitmaps and stats in active table (this will become the new
1697 		 * "search" table). */
1698 		if (lq_sta->rs_state == RS_STATE_SEARCH_CYCLE_STARTED) {
1699 			rs_rate_scale_clear_tbl_windows(mvm, tbl);
1700 		}
1701 	}
1702 }
1703 
1704 static void rs_set_amsdu_len(struct iwl_mvm *mvm, struct ieee80211_sta *sta,
1705 			     struct iwl_scale_tbl_info *tbl,
1706 			     enum rs_action scale_action)
1707 {
1708 	struct iwl_mvm_sta *sta_priv = iwl_mvm_sta_from_mac80211(sta);
1709 
1710 	if ((!is_vht(&tbl->rate) && !is_ht(&tbl->rate)) ||
1711 	    tbl->rate.index < IWL_RATE_MCS_5_INDEX ||
1712 	    scale_action == RS_ACTION_DOWNSCALE)
1713 		sta_priv->tlc_amsdu = false;
1714 	else
1715 		sta_priv->tlc_amsdu = true;
1716 }
1717 
1718 /*
1719  * setup rate table in uCode
1720  */
1721 static void rs_update_rate_tbl(struct iwl_mvm *mvm,
1722 			       struct ieee80211_sta *sta,
1723 			       struct iwl_lq_sta *lq_sta,
1724 			       struct iwl_scale_tbl_info *tbl)
1725 {
1726 	rs_fill_lq_cmd(mvm, sta, lq_sta, &tbl->rate);
1727 	iwl_mvm_send_lq_cmd(mvm, &lq_sta->lq, false);
1728 }
1729 
1730 static bool rs_tweak_rate_tbl(struct iwl_mvm *mvm,
1731 			      struct ieee80211_sta *sta,
1732 			      struct iwl_lq_sta *lq_sta,
1733 			      struct iwl_scale_tbl_info *tbl,
1734 			      enum rs_action scale_action)
1735 {
1736 	if (sta->bandwidth != IEEE80211_STA_RX_BW_80)
1737 		return false;
1738 
1739 	if (!is_vht_siso(&tbl->rate))
1740 		return false;
1741 
1742 	if ((tbl->rate.bw == RATE_MCS_CHAN_WIDTH_80) &&
1743 	    (tbl->rate.index == IWL_RATE_MCS_0_INDEX) &&
1744 	    (scale_action == RS_ACTION_DOWNSCALE)) {
1745 		tbl->rate.bw = RATE_MCS_CHAN_WIDTH_20;
1746 		tbl->rate.index = IWL_RATE_MCS_4_INDEX;
1747 		IWL_DEBUG_RATE(mvm, "Switch 80Mhz SISO MCS0 -> 20Mhz MCS4\n");
1748 		goto tweaked;
1749 	}
1750 
1751 	/* Go back to 80Mhz MCS1 only if we've established that 20Mhz MCS5 is
1752 	 * sustainable, i.e. we're past the test window. We can't go back
1753 	 * if MCS5 is just tested as this will happen always after switching
1754 	 * to 20Mhz MCS4 because the rate stats are cleared.
1755 	 */
1756 	if ((tbl->rate.bw == RATE_MCS_CHAN_WIDTH_20) &&
1757 	    (((tbl->rate.index == IWL_RATE_MCS_5_INDEX) &&
1758 	     (scale_action == RS_ACTION_STAY)) ||
1759 	     ((tbl->rate.index > IWL_RATE_MCS_5_INDEX) &&
1760 	      (scale_action == RS_ACTION_UPSCALE)))) {
1761 		tbl->rate.bw = RATE_MCS_CHAN_WIDTH_80;
1762 		tbl->rate.index = IWL_RATE_MCS_1_INDEX;
1763 		IWL_DEBUG_RATE(mvm, "Switch 20Mhz SISO MCS5 -> 80Mhz MCS1\n");
1764 		goto tweaked;
1765 	}
1766 
1767 	return false;
1768 
1769 tweaked:
1770 	rs_set_expected_tpt_table(lq_sta, tbl);
1771 	rs_rate_scale_clear_tbl_windows(mvm, tbl);
1772 	return true;
1773 }
1774 
1775 static enum rs_column rs_get_next_column(struct iwl_mvm *mvm,
1776 					 struct iwl_lq_sta *lq_sta,
1777 					 struct ieee80211_sta *sta,
1778 					 struct iwl_scale_tbl_info *tbl)
1779 {
1780 	int i, j, max_rate;
1781 	enum rs_column next_col_id;
1782 	const struct rs_tx_column *curr_col = &rs_tx_columns[tbl->column];
1783 	const struct rs_tx_column *next_col;
1784 	allow_column_func_t allow_func;
1785 	u8 valid_ants = iwl_mvm_get_valid_tx_ant(mvm);
1786 	const u16 *expected_tpt_tbl;
1787 	u16 tpt, max_expected_tpt;
1788 
1789 	for (i = 0; i < MAX_NEXT_COLUMNS; i++) {
1790 		next_col_id = curr_col->next_columns[i];
1791 
1792 		if (next_col_id == RS_COLUMN_INVALID)
1793 			continue;
1794 
1795 		if (lq_sta->visited_columns & BIT(next_col_id)) {
1796 			IWL_DEBUG_RATE(mvm, "Skip already visited column %d\n",
1797 				       next_col_id);
1798 			continue;
1799 		}
1800 
1801 		next_col = &rs_tx_columns[next_col_id];
1802 
1803 		if (!rs_is_valid_ant(valid_ants, next_col->ant)) {
1804 			IWL_DEBUG_RATE(mvm,
1805 				       "Skip column %d as ANT config isn't supported by chip. valid_ants 0x%x column ant 0x%x\n",
1806 				       next_col_id, valid_ants, next_col->ant);
1807 			continue;
1808 		}
1809 
1810 		for (j = 0; j < MAX_COLUMN_CHECKS; j++) {
1811 			allow_func = next_col->checks[j];
1812 			if (allow_func && !allow_func(mvm, sta, &tbl->rate,
1813 						      next_col))
1814 				break;
1815 		}
1816 
1817 		if (j != MAX_COLUMN_CHECKS) {
1818 			IWL_DEBUG_RATE(mvm,
1819 				       "Skip column %d: not allowed (check %d failed)\n",
1820 				       next_col_id, j);
1821 
1822 			continue;
1823 		}
1824 
1825 		tpt = lq_sta->last_tpt / 100;
1826 		expected_tpt_tbl = rs_get_expected_tpt_table(lq_sta, next_col,
1827 						     rs_bw_from_sta_bw(sta));
1828 		if (WARN_ON_ONCE(!expected_tpt_tbl))
1829 			continue;
1830 
1831 		max_rate = rs_get_max_allowed_rate(lq_sta, next_col);
1832 		if (max_rate == IWL_RATE_INVALID) {
1833 			IWL_DEBUG_RATE(mvm,
1834 				       "Skip column %d: no rate is allowed in this column\n",
1835 				       next_col_id);
1836 			continue;
1837 		}
1838 
1839 		max_expected_tpt = expected_tpt_tbl[max_rate];
1840 		if (tpt >= max_expected_tpt) {
1841 			IWL_DEBUG_RATE(mvm,
1842 				       "Skip column %d: can't beat current TPT. Max expected %d current %d\n",
1843 				       next_col_id, max_expected_tpt, tpt);
1844 			continue;
1845 		}
1846 
1847 		IWL_DEBUG_RATE(mvm,
1848 			       "Found potential column %d. Max expected %d current %d\n",
1849 			       next_col_id, max_expected_tpt, tpt);
1850 		break;
1851 	}
1852 
1853 	if (i == MAX_NEXT_COLUMNS)
1854 		return RS_COLUMN_INVALID;
1855 
1856 	return next_col_id;
1857 }
1858 
1859 static int rs_switch_to_column(struct iwl_mvm *mvm,
1860 			       struct iwl_lq_sta *lq_sta,
1861 			       struct ieee80211_sta *sta,
1862 			       enum rs_column col_id)
1863 {
1864 	struct iwl_scale_tbl_info *tbl = &(lq_sta->lq_info[lq_sta->active_tbl]);
1865 	struct iwl_scale_tbl_info *search_tbl =
1866 				&(lq_sta->lq_info[(1 - lq_sta->active_tbl)]);
1867 	struct rs_rate *rate = &search_tbl->rate;
1868 	const struct rs_tx_column *column = &rs_tx_columns[col_id];
1869 	const struct rs_tx_column *curr_column = &rs_tx_columns[tbl->column];
1870 	u32 sz = (sizeof(struct iwl_scale_tbl_info) -
1871 		  (sizeof(struct iwl_rate_scale_data) * IWL_RATE_COUNT));
1872 	unsigned long rate_mask = 0;
1873 	u32 rate_idx = 0;
1874 
1875 	memcpy(search_tbl, tbl, sz);
1876 
1877 	rate->sgi = column->sgi;
1878 	rate->ant = column->ant;
1879 
1880 	if (column->mode == RS_LEGACY) {
1881 		if (lq_sta->band == NL80211_BAND_5GHZ)
1882 			rate->type = LQ_LEGACY_A;
1883 		else
1884 			rate->type = LQ_LEGACY_G;
1885 
1886 		rate->bw = RATE_MCS_CHAN_WIDTH_20;
1887 		rate->ldpc = false;
1888 		rate_mask = lq_sta->active_legacy_rate;
1889 	} else if (column->mode == RS_SISO) {
1890 		rate->type = lq_sta->is_vht ? LQ_VHT_SISO : LQ_HT_SISO;
1891 		rate_mask = lq_sta->active_siso_rate;
1892 	} else if (column->mode == RS_MIMO2) {
1893 		rate->type = lq_sta->is_vht ? LQ_VHT_MIMO2 : LQ_HT_MIMO2;
1894 		rate_mask = lq_sta->active_mimo2_rate;
1895 	} else {
1896 		WARN_ONCE(1, "Bad column mode");
1897 	}
1898 
1899 	if (column->mode != RS_LEGACY) {
1900 		rate->bw = rs_bw_from_sta_bw(sta);
1901 		rate->ldpc = lq_sta->ldpc;
1902 	}
1903 
1904 	search_tbl->column = col_id;
1905 	rs_set_expected_tpt_table(lq_sta, search_tbl);
1906 
1907 	lq_sta->visited_columns |= BIT(col_id);
1908 
1909 	/* Get the best matching rate if we're changing modes. e.g.
1910 	 * SISO->MIMO, LEGACY->SISO, MIMO->SISO
1911 	 */
1912 	if (curr_column->mode != column->mode) {
1913 		rate_idx = rs_get_best_rate(mvm, lq_sta, search_tbl,
1914 					    rate_mask, rate->index);
1915 
1916 		if ((rate_idx == IWL_RATE_INVALID) ||
1917 		    !(BIT(rate_idx) & rate_mask)) {
1918 			IWL_DEBUG_RATE(mvm,
1919 				       "can not switch with index %d"
1920 				       " rate mask %lx\n",
1921 				       rate_idx, rate_mask);
1922 
1923 			goto err;
1924 		}
1925 
1926 		rate->index = rate_idx;
1927 	}
1928 
1929 	IWL_DEBUG_RATE(mvm, "Switched to column %d: Index %d\n",
1930 		       col_id, rate->index);
1931 
1932 	return 0;
1933 
1934 err:
1935 	rate->type = LQ_NONE;
1936 	return -1;
1937 }
1938 
1939 static enum rs_action rs_get_rate_action(struct iwl_mvm *mvm,
1940 					 struct iwl_scale_tbl_info *tbl,
1941 					 s32 sr, int low, int high,
1942 					 int current_tpt,
1943 					 int low_tpt, int high_tpt)
1944 {
1945 	enum rs_action action = RS_ACTION_STAY;
1946 
1947 	if ((sr <= RS_PERCENT(IWL_MVM_RS_SR_FORCE_DECREASE)) ||
1948 	    (current_tpt == 0)) {
1949 		IWL_DEBUG_RATE(mvm,
1950 			       "Decrease rate because of low SR\n");
1951 		return RS_ACTION_DOWNSCALE;
1952 	}
1953 
1954 	if ((low_tpt == IWL_INVALID_VALUE) &&
1955 	    (high_tpt == IWL_INVALID_VALUE) &&
1956 	    (high != IWL_RATE_INVALID)) {
1957 		IWL_DEBUG_RATE(mvm,
1958 			       "No data about high/low rates. Increase rate\n");
1959 		return RS_ACTION_UPSCALE;
1960 	}
1961 
1962 	if ((high_tpt == IWL_INVALID_VALUE) &&
1963 	    (high != IWL_RATE_INVALID) &&
1964 	    (low_tpt != IWL_INVALID_VALUE) &&
1965 	    (low_tpt < current_tpt)) {
1966 		IWL_DEBUG_RATE(mvm,
1967 			       "No data about high rate and low rate is worse. Increase rate\n");
1968 		return RS_ACTION_UPSCALE;
1969 	}
1970 
1971 	if ((high_tpt != IWL_INVALID_VALUE) &&
1972 	    (high_tpt > current_tpt)) {
1973 		IWL_DEBUG_RATE(mvm,
1974 			       "Higher rate is better. Increate rate\n");
1975 		return RS_ACTION_UPSCALE;
1976 	}
1977 
1978 	if ((low_tpt != IWL_INVALID_VALUE) &&
1979 	    (high_tpt != IWL_INVALID_VALUE) &&
1980 	    (low_tpt < current_tpt) &&
1981 	    (high_tpt < current_tpt)) {
1982 		IWL_DEBUG_RATE(mvm,
1983 			       "Both high and low are worse. Maintain rate\n");
1984 		return RS_ACTION_STAY;
1985 	}
1986 
1987 	if ((low_tpt != IWL_INVALID_VALUE) &&
1988 	    (low_tpt > current_tpt)) {
1989 		IWL_DEBUG_RATE(mvm,
1990 			       "Lower rate is better\n");
1991 		action = RS_ACTION_DOWNSCALE;
1992 		goto out;
1993 	}
1994 
1995 	if ((low_tpt == IWL_INVALID_VALUE) &&
1996 	    (low != IWL_RATE_INVALID)) {
1997 		IWL_DEBUG_RATE(mvm,
1998 			       "No data about lower rate\n");
1999 		action = RS_ACTION_DOWNSCALE;
2000 		goto out;
2001 	}
2002 
2003 	IWL_DEBUG_RATE(mvm, "Maintain rate\n");
2004 
2005 out:
2006 	if ((action == RS_ACTION_DOWNSCALE) && (low != IWL_RATE_INVALID)) {
2007 		if (sr >= RS_PERCENT(IWL_MVM_RS_SR_NO_DECREASE)) {
2008 			IWL_DEBUG_RATE(mvm,
2009 				       "SR is above NO DECREASE. Avoid downscale\n");
2010 			action = RS_ACTION_STAY;
2011 		} else if (current_tpt > (100 * tbl->expected_tpt[low])) {
2012 			IWL_DEBUG_RATE(mvm,
2013 				       "Current TPT is higher than max expected in low rate. Avoid downscale\n");
2014 			action = RS_ACTION_STAY;
2015 		} else {
2016 			IWL_DEBUG_RATE(mvm, "Decrease rate\n");
2017 		}
2018 	}
2019 
2020 	return action;
2021 }
2022 
2023 static bool rs_stbc_allow(struct iwl_mvm *mvm, struct ieee80211_sta *sta,
2024 			  struct iwl_lq_sta *lq_sta)
2025 {
2026 	/* Our chip supports Tx STBC and the peer is an HT/VHT STA which
2027 	 * supports STBC of at least 1*SS
2028 	 */
2029 	if (!lq_sta->stbc_capable)
2030 		return false;
2031 
2032 	if (!iwl_mvm_bt_coex_is_mimo_allowed(mvm, sta))
2033 		return false;
2034 
2035 	return true;
2036 }
2037 
2038 static void rs_get_adjacent_txp(struct iwl_mvm *mvm, int index,
2039 				int *weaker, int *stronger)
2040 {
2041 	*weaker = index + IWL_MVM_RS_TPC_TX_POWER_STEP;
2042 	if (*weaker > TPC_MAX_REDUCTION)
2043 		*weaker = TPC_INVALID;
2044 
2045 	*stronger = index - IWL_MVM_RS_TPC_TX_POWER_STEP;
2046 	if (*stronger < 0)
2047 		*stronger = TPC_INVALID;
2048 }
2049 
2050 static bool rs_tpc_allowed(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
2051 			   struct rs_rate *rate, enum nl80211_band band)
2052 {
2053 	int index = rate->index;
2054 	bool cam = (iwlmvm_mod_params.power_scheme == IWL_POWER_SCHEME_CAM);
2055 	bool sta_ps_disabled = (vif->type == NL80211_IFTYPE_STATION &&
2056 				!vif->bss_conf.ps);
2057 
2058 	IWL_DEBUG_RATE(mvm, "cam: %d sta_ps_disabled %d\n",
2059 		       cam, sta_ps_disabled);
2060 	/*
2061 	 * allow tpc only if power management is enabled, or bt coex
2062 	 * activity grade allows it and we are on 2.4Ghz.
2063 	 */
2064 	if ((cam || sta_ps_disabled) &&
2065 	    !iwl_mvm_bt_coex_is_tpc_allowed(mvm, band))
2066 		return false;
2067 
2068 	IWL_DEBUG_RATE(mvm, "check rate, table type: %d\n", rate->type);
2069 	if (is_legacy(rate))
2070 		return index == IWL_RATE_54M_INDEX;
2071 	if (is_ht(rate))
2072 		return index == IWL_RATE_MCS_7_INDEX;
2073 	if (is_vht(rate))
2074 		return index == IWL_RATE_MCS_7_INDEX ||
2075 		       index == IWL_RATE_MCS_8_INDEX ||
2076 		       index == IWL_RATE_MCS_9_INDEX;
2077 
2078 	WARN_ON_ONCE(1);
2079 	return false;
2080 }
2081 
2082 enum tpc_action {
2083 	TPC_ACTION_STAY,
2084 	TPC_ACTION_DECREASE,
2085 	TPC_ACTION_INCREASE,
2086 	TPC_ACTION_NO_RESTIRCTION,
2087 };
2088 
2089 static enum tpc_action rs_get_tpc_action(struct iwl_mvm *mvm,
2090 					 s32 sr, int weak, int strong,
2091 					 int current_tpt,
2092 					 int weak_tpt, int strong_tpt)
2093 {
2094 	/* stay until we have valid tpt */
2095 	if (current_tpt == IWL_INVALID_VALUE) {
2096 		IWL_DEBUG_RATE(mvm, "no current tpt. stay.\n");
2097 		return TPC_ACTION_STAY;
2098 	}
2099 
2100 	/* Too many failures, increase txp */
2101 	if (sr <= RS_PERCENT(IWL_MVM_RS_TPC_SR_FORCE_INCREASE) ||
2102 	    current_tpt == 0) {
2103 		IWL_DEBUG_RATE(mvm, "increase txp because of weak SR\n");
2104 		return TPC_ACTION_NO_RESTIRCTION;
2105 	}
2106 
2107 	/* try decreasing first if applicable */
2108 	if (sr >= RS_PERCENT(IWL_MVM_RS_TPC_SR_NO_INCREASE) &&
2109 	    weak != TPC_INVALID) {
2110 		if (weak_tpt == IWL_INVALID_VALUE &&
2111 		    (strong_tpt == IWL_INVALID_VALUE ||
2112 		     current_tpt >= strong_tpt)) {
2113 			IWL_DEBUG_RATE(mvm,
2114 				       "no weak txp measurement. decrease txp\n");
2115 			return TPC_ACTION_DECREASE;
2116 		}
2117 
2118 		if (weak_tpt > current_tpt) {
2119 			IWL_DEBUG_RATE(mvm,
2120 				       "lower txp has better tpt. decrease txp\n");
2121 			return TPC_ACTION_DECREASE;
2122 		}
2123 	}
2124 
2125 	/* next, increase if needed */
2126 	if (sr < RS_PERCENT(IWL_MVM_RS_TPC_SR_NO_INCREASE) &&
2127 	    strong != TPC_INVALID) {
2128 		if (weak_tpt == IWL_INVALID_VALUE &&
2129 		    strong_tpt != IWL_INVALID_VALUE &&
2130 		    current_tpt < strong_tpt) {
2131 			IWL_DEBUG_RATE(mvm,
2132 				       "higher txp has better tpt. increase txp\n");
2133 			return TPC_ACTION_INCREASE;
2134 		}
2135 
2136 		if (weak_tpt < current_tpt &&
2137 		    (strong_tpt == IWL_INVALID_VALUE ||
2138 		     strong_tpt > current_tpt)) {
2139 			IWL_DEBUG_RATE(mvm,
2140 				       "lower txp has worse tpt. increase txp\n");
2141 			return TPC_ACTION_INCREASE;
2142 		}
2143 	}
2144 
2145 	IWL_DEBUG_RATE(mvm, "no need to increase or decrease txp - stay\n");
2146 	return TPC_ACTION_STAY;
2147 }
2148 
2149 static bool rs_tpc_perform(struct iwl_mvm *mvm,
2150 			   struct ieee80211_sta *sta,
2151 			   struct iwl_lq_sta *lq_sta,
2152 			   struct iwl_scale_tbl_info *tbl)
2153 {
2154 	struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
2155 	struct ieee80211_vif *vif = mvm_sta->vif;
2156 	struct ieee80211_chanctx_conf *chanctx_conf;
2157 	enum nl80211_band band;
2158 	struct iwl_rate_scale_data *window;
2159 	struct rs_rate *rate = &tbl->rate;
2160 	enum tpc_action action;
2161 	s32 sr;
2162 	u8 cur = lq_sta->lq.reduced_tpc;
2163 	int current_tpt;
2164 	int weak, strong;
2165 	int weak_tpt = IWL_INVALID_VALUE, strong_tpt = IWL_INVALID_VALUE;
2166 
2167 #ifdef CONFIG_MAC80211_DEBUGFS
2168 	if (lq_sta->pers.dbg_fixed_txp_reduction <= TPC_MAX_REDUCTION) {
2169 		IWL_DEBUG_RATE(mvm, "fixed tpc: %d\n",
2170 			       lq_sta->pers.dbg_fixed_txp_reduction);
2171 		lq_sta->lq.reduced_tpc = lq_sta->pers.dbg_fixed_txp_reduction;
2172 		return cur != lq_sta->pers.dbg_fixed_txp_reduction;
2173 	}
2174 #endif
2175 
2176 	rcu_read_lock();
2177 	chanctx_conf = rcu_dereference(vif->chanctx_conf);
2178 	if (WARN_ON(!chanctx_conf))
2179 		band = NUM_NL80211_BANDS;
2180 	else
2181 		band = chanctx_conf->def.chan->band;
2182 	rcu_read_unlock();
2183 
2184 	if (!rs_tpc_allowed(mvm, vif, rate, band)) {
2185 		IWL_DEBUG_RATE(mvm,
2186 			       "tpc is not allowed. remove txp restrictions\n");
2187 		lq_sta->lq.reduced_tpc = TPC_NO_REDUCTION;
2188 		return cur != TPC_NO_REDUCTION;
2189 	}
2190 
2191 	rs_get_adjacent_txp(mvm, cur, &weak, &strong);
2192 
2193 	/* Collect measured throughputs for current and adjacent rates */
2194 	window = tbl->tpc_win;
2195 	sr = window[cur].success_ratio;
2196 	current_tpt = window[cur].average_tpt;
2197 	if (weak != TPC_INVALID)
2198 		weak_tpt = window[weak].average_tpt;
2199 	if (strong != TPC_INVALID)
2200 		strong_tpt = window[strong].average_tpt;
2201 
2202 	IWL_DEBUG_RATE(mvm,
2203 		       "(TPC: %d): cur_tpt %d SR %d weak %d strong %d weak_tpt %d strong_tpt %d\n",
2204 		       cur, current_tpt, sr, weak, strong,
2205 		       weak_tpt, strong_tpt);
2206 
2207 	action = rs_get_tpc_action(mvm, sr, weak, strong,
2208 				   current_tpt, weak_tpt, strong_tpt);
2209 
2210 	/* override actions if we are on the edge */
2211 	if (weak == TPC_INVALID && action == TPC_ACTION_DECREASE) {
2212 		IWL_DEBUG_RATE(mvm, "already in lowest txp, stay\n");
2213 		action = TPC_ACTION_STAY;
2214 	} else if (strong == TPC_INVALID &&
2215 		   (action == TPC_ACTION_INCREASE ||
2216 		    action == TPC_ACTION_NO_RESTIRCTION)) {
2217 		IWL_DEBUG_RATE(mvm, "already in highest txp, stay\n");
2218 		action = TPC_ACTION_STAY;
2219 	}
2220 
2221 	switch (action) {
2222 	case TPC_ACTION_DECREASE:
2223 		lq_sta->lq.reduced_tpc = weak;
2224 		return true;
2225 	case TPC_ACTION_INCREASE:
2226 		lq_sta->lq.reduced_tpc = strong;
2227 		return true;
2228 	case TPC_ACTION_NO_RESTIRCTION:
2229 		lq_sta->lq.reduced_tpc = TPC_NO_REDUCTION;
2230 		return true;
2231 	case TPC_ACTION_STAY:
2232 		/* do nothing */
2233 		break;
2234 	}
2235 	return false;
2236 }
2237 
2238 /*
2239  * Do rate scaling and search for new modulation mode.
2240  */
2241 static void rs_rate_scale_perform(struct iwl_mvm *mvm,
2242 				  struct ieee80211_sta *sta,
2243 				  struct iwl_lq_sta *lq_sta,
2244 				  int tid, bool ndp)
2245 {
2246 	int low = IWL_RATE_INVALID;
2247 	int high = IWL_RATE_INVALID;
2248 	int index;
2249 	struct iwl_rate_scale_data *window = NULL;
2250 	int current_tpt = IWL_INVALID_VALUE;
2251 	int low_tpt = IWL_INVALID_VALUE;
2252 	int high_tpt = IWL_INVALID_VALUE;
2253 	u32 fail_count;
2254 	enum rs_action scale_action = RS_ACTION_STAY;
2255 	u16 rate_mask;
2256 	u8 update_lq = 0;
2257 	struct iwl_scale_tbl_info *tbl, *tbl1;
2258 	u8 active_tbl = 0;
2259 	u8 done_search = 0;
2260 	u16 high_low;
2261 	s32 sr;
2262 	u8 prev_agg = lq_sta->is_agg;
2263 	struct iwl_mvm_sta *sta_priv = iwl_mvm_sta_from_mac80211(sta);
2264 	struct iwl_mvm_tid_data *tid_data;
2265 	struct rs_rate *rate;
2266 
2267 	lq_sta->is_agg = !!sta_priv->agg_tids;
2268 
2269 	/*
2270 	 * Select rate-scale / modulation-mode table to work with in
2271 	 * the rest of this function:  "search" if searching for better
2272 	 * modulation mode, or "active" if doing rate scaling within a mode.
2273 	 */
2274 	if (!lq_sta->search_better_tbl)
2275 		active_tbl = lq_sta->active_tbl;
2276 	else
2277 		active_tbl = 1 - lq_sta->active_tbl;
2278 
2279 	tbl = &(lq_sta->lq_info[active_tbl]);
2280 	rate = &tbl->rate;
2281 
2282 	if (prev_agg != lq_sta->is_agg) {
2283 		IWL_DEBUG_RATE(mvm,
2284 			       "Aggregation changed: prev %d current %d. Update expected TPT table\n",
2285 			       prev_agg, lq_sta->is_agg);
2286 		rs_set_expected_tpt_table(lq_sta, tbl);
2287 		rs_rate_scale_clear_tbl_windows(mvm, tbl);
2288 	}
2289 
2290 	/* current tx rate */
2291 	index = rate->index;
2292 
2293 	/* rates available for this association, and for modulation mode */
2294 	rate_mask = rs_get_supported_rates(lq_sta, rate);
2295 
2296 	if (!(BIT(index) & rate_mask)) {
2297 		IWL_ERR(mvm, "Current Rate is not valid\n");
2298 		if (lq_sta->search_better_tbl) {
2299 			/* revert to active table if search table is not valid*/
2300 			rate->type = LQ_NONE;
2301 			lq_sta->search_better_tbl = 0;
2302 			tbl = &(lq_sta->lq_info[lq_sta->active_tbl]);
2303 			rs_update_rate_tbl(mvm, sta, lq_sta, tbl);
2304 		}
2305 		return;
2306 	}
2307 
2308 	/* Get expected throughput table and history window for current rate */
2309 	if (!tbl->expected_tpt) {
2310 		IWL_ERR(mvm, "tbl->expected_tpt is NULL\n");
2311 		return;
2312 	}
2313 
2314 	/* TODO: handle rate_idx_mask and rate_idx_mcs_mask */
2315 	window = &(tbl->win[index]);
2316 
2317 	/*
2318 	 * If there is not enough history to calculate actual average
2319 	 * throughput, keep analyzing results of more tx frames, without
2320 	 * changing rate or mode (bypass most of the rest of this function).
2321 	 * Set up new rate table in uCode only if old rate is not supported
2322 	 * in current association (use new rate found above).
2323 	 */
2324 	fail_count = window->counter - window->success_counter;
2325 	if ((fail_count < IWL_MVM_RS_RATE_MIN_FAILURE_TH) &&
2326 	    (window->success_counter < IWL_MVM_RS_RATE_MIN_SUCCESS_TH)) {
2327 		IWL_DEBUG_RATE(mvm,
2328 			       "%s: Test Window: succ %d total %d\n",
2329 			       rs_pretty_rate(rate),
2330 			       window->success_counter, window->counter);
2331 
2332 		/* Can't calculate this yet; not enough history */
2333 		window->average_tpt = IWL_INVALID_VALUE;
2334 
2335 		/* Should we stay with this modulation mode,
2336 		 * or search for a new one? */
2337 		rs_stay_in_table(lq_sta, false);
2338 
2339 		return;
2340 	}
2341 
2342 	/* If we are searching for better modulation mode, check success. */
2343 	if (lq_sta->search_better_tbl) {
2344 		/* If good success, continue using the "search" mode;
2345 		 * no need to send new link quality command, since we're
2346 		 * continuing to use the setup that we've been trying. */
2347 		if (window->average_tpt > lq_sta->last_tpt) {
2348 			IWL_DEBUG_RATE(mvm,
2349 				       "SWITCHING TO NEW TABLE SR: %d "
2350 				       "cur-tpt %d old-tpt %d\n",
2351 				       window->success_ratio,
2352 				       window->average_tpt,
2353 				       lq_sta->last_tpt);
2354 
2355 			/* Swap tables; "search" becomes "active" */
2356 			lq_sta->active_tbl = active_tbl;
2357 			current_tpt = window->average_tpt;
2358 		/* Else poor success; go back to mode in "active" table */
2359 		} else {
2360 			IWL_DEBUG_RATE(mvm,
2361 				       "GOING BACK TO THE OLD TABLE: SR %d "
2362 				       "cur-tpt %d old-tpt %d\n",
2363 				       window->success_ratio,
2364 				       window->average_tpt,
2365 				       lq_sta->last_tpt);
2366 
2367 			/* Nullify "search" table */
2368 			rate->type = LQ_NONE;
2369 
2370 			/* Revert to "active" table */
2371 			active_tbl = lq_sta->active_tbl;
2372 			tbl = &(lq_sta->lq_info[active_tbl]);
2373 
2374 			/* Revert to "active" rate and throughput info */
2375 			index = tbl->rate.index;
2376 			current_tpt = lq_sta->last_tpt;
2377 
2378 			/* Need to set up a new rate table in uCode */
2379 			update_lq = 1;
2380 		}
2381 
2382 		/* Either way, we've made a decision; modulation mode
2383 		 * search is done, allow rate adjustment next time. */
2384 		lq_sta->search_better_tbl = 0;
2385 		done_search = 1;	/* Don't switch modes below! */
2386 		goto lq_update;
2387 	}
2388 
2389 	/* (Else) not in search of better modulation mode, try for better
2390 	 * starting rate, while staying in this mode. */
2391 	high_low = rs_get_adjacent_rate(mvm, index, rate_mask, rate->type);
2392 	low = high_low & 0xff;
2393 	high = (high_low >> 8) & 0xff;
2394 
2395 	/* TODO: handle rate_idx_mask and rate_idx_mcs_mask */
2396 
2397 	sr = window->success_ratio;
2398 
2399 	/* Collect measured throughputs for current and adjacent rates */
2400 	current_tpt = window->average_tpt;
2401 	if (low != IWL_RATE_INVALID)
2402 		low_tpt = tbl->win[low].average_tpt;
2403 	if (high != IWL_RATE_INVALID)
2404 		high_tpt = tbl->win[high].average_tpt;
2405 
2406 	IWL_DEBUG_RATE(mvm,
2407 		       "%s: cur_tpt %d SR %d low %d high %d low_tpt %d high_tpt %d\n",
2408 		       rs_pretty_rate(rate), current_tpt, sr,
2409 		       low, high, low_tpt, high_tpt);
2410 
2411 	scale_action = rs_get_rate_action(mvm, tbl, sr, low, high,
2412 					  current_tpt, low_tpt, high_tpt);
2413 
2414 	/* Force a search in case BT doesn't like us being in MIMO */
2415 	if (is_mimo(rate) &&
2416 	    !iwl_mvm_bt_coex_is_mimo_allowed(mvm, sta)) {
2417 		IWL_DEBUG_RATE(mvm,
2418 			       "BT Coex forbids MIMO. Search for new config\n");
2419 		rs_stay_in_table(lq_sta, true);
2420 		goto lq_update;
2421 	}
2422 
2423 	switch (scale_action) {
2424 	case RS_ACTION_DOWNSCALE:
2425 		/* Decrease starting rate, update uCode's rate table */
2426 		if (low != IWL_RATE_INVALID) {
2427 			update_lq = 1;
2428 			index = low;
2429 		} else {
2430 			IWL_DEBUG_RATE(mvm,
2431 				       "At the bottom rate. Can't decrease\n");
2432 		}
2433 
2434 		break;
2435 	case RS_ACTION_UPSCALE:
2436 		/* Increase starting rate, update uCode's rate table */
2437 		if (high != IWL_RATE_INVALID) {
2438 			update_lq = 1;
2439 			index = high;
2440 		} else {
2441 			IWL_DEBUG_RATE(mvm,
2442 				       "At the top rate. Can't increase\n");
2443 		}
2444 
2445 		break;
2446 	case RS_ACTION_STAY:
2447 		/* No change */
2448 		if (lq_sta->rs_state == RS_STATE_STAY_IN_COLUMN)
2449 			update_lq = rs_tpc_perform(mvm, sta, lq_sta, tbl);
2450 		break;
2451 	default:
2452 		break;
2453 	}
2454 
2455 lq_update:
2456 	/* Replace uCode's rate table for the destination station. */
2457 	if (update_lq) {
2458 		tbl->rate.index = index;
2459 		if (IWL_MVM_RS_80_20_FAR_RANGE_TWEAK)
2460 			rs_tweak_rate_tbl(mvm, sta, lq_sta, tbl, scale_action);
2461 		rs_set_amsdu_len(mvm, sta, tbl, scale_action);
2462 		rs_update_rate_tbl(mvm, sta, lq_sta, tbl);
2463 	}
2464 
2465 	rs_stay_in_table(lq_sta, false);
2466 
2467 	/*
2468 	 * Search for new modulation mode if we're:
2469 	 * 1)  Not changing rates right now
2470 	 * 2)  Not just finishing up a search
2471 	 * 3)  Allowing a new search
2472 	 */
2473 	if (!update_lq && !done_search &&
2474 	    lq_sta->rs_state == RS_STATE_SEARCH_CYCLE_STARTED
2475 	    && window->counter) {
2476 		enum rs_column next_column;
2477 
2478 		/* Save current throughput to compare with "search" throughput*/
2479 		lq_sta->last_tpt = current_tpt;
2480 
2481 		IWL_DEBUG_RATE(mvm,
2482 			       "Start Search: update_lq %d done_search %d rs_state %d win->counter %d\n",
2483 			       update_lq, done_search, lq_sta->rs_state,
2484 			       window->counter);
2485 
2486 		next_column = rs_get_next_column(mvm, lq_sta, sta, tbl);
2487 		if (next_column != RS_COLUMN_INVALID) {
2488 			int ret = rs_switch_to_column(mvm, lq_sta, sta,
2489 						      next_column);
2490 			if (!ret)
2491 				lq_sta->search_better_tbl = 1;
2492 		} else {
2493 			IWL_DEBUG_RATE(mvm,
2494 				       "No more columns to explore in search cycle. Go to RS_STATE_SEARCH_CYCLE_ENDED\n");
2495 			lq_sta->rs_state = RS_STATE_SEARCH_CYCLE_ENDED;
2496 		}
2497 
2498 		/* If new "search" mode was selected, set up in uCode table */
2499 		if (lq_sta->search_better_tbl) {
2500 			/* Access the "search" table, clear its history. */
2501 			tbl = &(lq_sta->lq_info[(1 - lq_sta->active_tbl)]);
2502 			rs_rate_scale_clear_tbl_windows(mvm, tbl);
2503 
2504 			/* Use new "search" start rate */
2505 			index = tbl->rate.index;
2506 
2507 			rs_dump_rate(mvm, &tbl->rate,
2508 				     "Switch to SEARCH TABLE:");
2509 			rs_update_rate_tbl(mvm, sta, lq_sta, tbl);
2510 		} else {
2511 			done_search = 1;
2512 		}
2513 	}
2514 
2515 	if (done_search && lq_sta->rs_state == RS_STATE_SEARCH_CYCLE_ENDED) {
2516 		/* If the "active" (non-search) mode was legacy,
2517 		 * and we've tried switching antennas,
2518 		 * but we haven't been able to try HT modes (not available),
2519 		 * stay with best antenna legacy modulation for a while
2520 		 * before next round of mode comparisons. */
2521 		tbl1 = &(lq_sta->lq_info[lq_sta->active_tbl]);
2522 		if (is_legacy(&tbl1->rate)) {
2523 			IWL_DEBUG_RATE(mvm, "LQ: STAY in legacy table\n");
2524 
2525 			if (tid != IWL_MAX_TID_COUNT) {
2526 				tid_data = &sta_priv->tid_data[tid];
2527 				if (tid_data->state != IWL_AGG_OFF) {
2528 					IWL_DEBUG_RATE(mvm,
2529 						       "Stop aggregation on tid %d\n",
2530 						       tid);
2531 					ieee80211_stop_tx_ba_session(sta, tid);
2532 				}
2533 			}
2534 			rs_set_stay_in_table(mvm, 1, lq_sta);
2535 		} else {
2536 		/* If we're in an HT mode, and all 3 mode switch actions
2537 		 * have been tried and compared, stay in this best modulation
2538 		 * mode for a while before next round of mode comparisons. */
2539 			if ((lq_sta->last_tpt > IWL_AGG_TPT_THREHOLD) &&
2540 			    (lq_sta->tx_agg_tid_en & (1 << tid)) &&
2541 			    (tid != IWL_MAX_TID_COUNT)) {
2542 				tid_data = &sta_priv->tid_data[tid];
2543 				if (tid_data->state == IWL_AGG_OFF && !ndp) {
2544 					IWL_DEBUG_RATE(mvm,
2545 						       "try to aggregate tid %d\n",
2546 						       tid);
2547 					rs_tl_turn_on_agg(mvm, tid,
2548 							  lq_sta, sta);
2549 				}
2550 			}
2551 			rs_set_stay_in_table(mvm, 0, lq_sta);
2552 		}
2553 	}
2554 }
2555 
2556 struct rs_init_rate_info {
2557 	s8 rssi;
2558 	u8 rate_idx;
2559 };
2560 
2561 static const struct rs_init_rate_info rs_optimal_rates_24ghz_legacy[] = {
2562 	{ -60, IWL_RATE_54M_INDEX },
2563 	{ -64, IWL_RATE_48M_INDEX },
2564 	{ -68, IWL_RATE_36M_INDEX },
2565 	{ -80, IWL_RATE_24M_INDEX },
2566 	{ -84, IWL_RATE_18M_INDEX },
2567 	{ -85, IWL_RATE_12M_INDEX },
2568 	{ -86, IWL_RATE_11M_INDEX },
2569 	{ -88, IWL_RATE_5M_INDEX  },
2570 	{ -90, IWL_RATE_2M_INDEX  },
2571 	{ S8_MIN, IWL_RATE_1M_INDEX },
2572 };
2573 
2574 static const struct rs_init_rate_info rs_optimal_rates_5ghz_legacy[] = {
2575 	{ -60, IWL_RATE_54M_INDEX },
2576 	{ -64, IWL_RATE_48M_INDEX },
2577 	{ -72, IWL_RATE_36M_INDEX },
2578 	{ -80, IWL_RATE_24M_INDEX },
2579 	{ -84, IWL_RATE_18M_INDEX },
2580 	{ -85, IWL_RATE_12M_INDEX },
2581 	{ -87, IWL_RATE_9M_INDEX  },
2582 	{ S8_MIN, IWL_RATE_6M_INDEX },
2583 };
2584 
2585 static const struct rs_init_rate_info rs_optimal_rates_ht[] = {
2586 	{ -60, IWL_RATE_MCS_7_INDEX },
2587 	{ -64, IWL_RATE_MCS_6_INDEX },
2588 	{ -68, IWL_RATE_MCS_5_INDEX },
2589 	{ -72, IWL_RATE_MCS_4_INDEX },
2590 	{ -80, IWL_RATE_MCS_3_INDEX },
2591 	{ -84, IWL_RATE_MCS_2_INDEX },
2592 	{ -85, IWL_RATE_MCS_1_INDEX },
2593 	{ S8_MIN, IWL_RATE_MCS_0_INDEX},
2594 };
2595 
2596 /* MCS index 9 is not valid for 20MHz VHT channel width,
2597  * but is ok for 40, 80 and 160MHz channels.
2598  */
2599 static const struct rs_init_rate_info rs_optimal_rates_vht_20mhz[] = {
2600 	{ -60, IWL_RATE_MCS_8_INDEX },
2601 	{ -64, IWL_RATE_MCS_7_INDEX },
2602 	{ -68, IWL_RATE_MCS_6_INDEX },
2603 	{ -72, IWL_RATE_MCS_5_INDEX },
2604 	{ -80, IWL_RATE_MCS_4_INDEX },
2605 	{ -84, IWL_RATE_MCS_3_INDEX },
2606 	{ -85, IWL_RATE_MCS_2_INDEX },
2607 	{ -87, IWL_RATE_MCS_1_INDEX },
2608 	{ S8_MIN, IWL_RATE_MCS_0_INDEX},
2609 };
2610 
2611 static const struct rs_init_rate_info rs_optimal_rates_vht[] = {
2612 	{ -60, IWL_RATE_MCS_9_INDEX },
2613 	{ -64, IWL_RATE_MCS_8_INDEX },
2614 	{ -68, IWL_RATE_MCS_7_INDEX },
2615 	{ -72, IWL_RATE_MCS_6_INDEX },
2616 	{ -80, IWL_RATE_MCS_5_INDEX },
2617 	{ -84, IWL_RATE_MCS_4_INDEX },
2618 	{ -85, IWL_RATE_MCS_3_INDEX },
2619 	{ -87, IWL_RATE_MCS_2_INDEX },
2620 	{ -88, IWL_RATE_MCS_1_INDEX },
2621 	{ S8_MIN, IWL_RATE_MCS_0_INDEX },
2622 };
2623 
2624 #define IWL_RS_LOW_RSSI_THRESHOLD (-76) /* dBm */
2625 
2626 /* Init the optimal rate based on STA caps
2627  * This combined with rssi is used to report the last tx rate
2628  * to userspace when we haven't transmitted enough frames.
2629  */
2630 static void rs_init_optimal_rate(struct iwl_mvm *mvm,
2631 				 struct ieee80211_sta *sta,
2632 				 struct iwl_lq_sta *lq_sta)
2633 {
2634 	struct rs_rate *rate = &lq_sta->optimal_rate;
2635 
2636 	if (lq_sta->max_mimo2_rate_idx != IWL_RATE_INVALID)
2637 		rate->type = lq_sta->is_vht ? LQ_VHT_MIMO2 : LQ_HT_MIMO2;
2638 	else if (lq_sta->max_siso_rate_idx != IWL_RATE_INVALID)
2639 		rate->type = lq_sta->is_vht ? LQ_VHT_SISO : LQ_HT_SISO;
2640 	else if (lq_sta->band == NL80211_BAND_5GHZ)
2641 		rate->type = LQ_LEGACY_A;
2642 	else
2643 		rate->type = LQ_LEGACY_G;
2644 
2645 	rate->bw = rs_bw_from_sta_bw(sta);
2646 	rate->sgi = rs_sgi_allow(mvm, sta, rate, NULL);
2647 
2648 	/* ANT/LDPC/STBC aren't relevant for the rate reported to userspace */
2649 
2650 	if (is_mimo(rate)) {
2651 		lq_sta->optimal_rate_mask = lq_sta->active_mimo2_rate;
2652 	} else if (is_siso(rate)) {
2653 		lq_sta->optimal_rate_mask = lq_sta->active_siso_rate;
2654 	} else {
2655 		lq_sta->optimal_rate_mask = lq_sta->active_legacy_rate;
2656 
2657 		if (lq_sta->band == NL80211_BAND_5GHZ) {
2658 			lq_sta->optimal_rates = rs_optimal_rates_5ghz_legacy;
2659 			lq_sta->optimal_nentries =
2660 				ARRAY_SIZE(rs_optimal_rates_5ghz_legacy);
2661 		} else {
2662 			lq_sta->optimal_rates = rs_optimal_rates_24ghz_legacy;
2663 			lq_sta->optimal_nentries =
2664 				ARRAY_SIZE(rs_optimal_rates_24ghz_legacy);
2665 		}
2666 	}
2667 
2668 	if (is_vht(rate)) {
2669 		if (rate->bw == RATE_MCS_CHAN_WIDTH_20) {
2670 			lq_sta->optimal_rates = rs_optimal_rates_vht_20mhz;
2671 			lq_sta->optimal_nentries =
2672 				ARRAY_SIZE(rs_optimal_rates_vht_20mhz);
2673 		} else {
2674 			lq_sta->optimal_rates = rs_optimal_rates_vht;
2675 			lq_sta->optimal_nentries =
2676 				ARRAY_SIZE(rs_optimal_rates_vht);
2677 		}
2678 	} else if (is_ht(rate)) {
2679 		lq_sta->optimal_rates = rs_optimal_rates_ht;
2680 		lq_sta->optimal_nentries = ARRAY_SIZE(rs_optimal_rates_ht);
2681 	}
2682 }
2683 
2684 /* Compute the optimal rate index based on RSSI */
2685 static struct rs_rate *rs_get_optimal_rate(struct iwl_mvm *mvm,
2686 					   struct iwl_lq_sta *lq_sta)
2687 {
2688 	struct rs_rate *rate = &lq_sta->optimal_rate;
2689 	int i;
2690 
2691 	rate->index = find_first_bit(&lq_sta->optimal_rate_mask,
2692 				     BITS_PER_LONG);
2693 
2694 	for (i = 0; i < lq_sta->optimal_nentries; i++) {
2695 		int rate_idx = lq_sta->optimal_rates[i].rate_idx;
2696 
2697 		if ((lq_sta->pers.last_rssi >= lq_sta->optimal_rates[i].rssi) &&
2698 		    (BIT(rate_idx) & lq_sta->optimal_rate_mask)) {
2699 			rate->index = rate_idx;
2700 			break;
2701 		}
2702 	}
2703 
2704 	return rate;
2705 }
2706 
2707 /* Choose an initial legacy rate and antenna to use based on the RSSI
2708  * of last Rx
2709  */
2710 static void rs_get_initial_rate(struct iwl_mvm *mvm,
2711 				struct ieee80211_sta *sta,
2712 				struct iwl_lq_sta *lq_sta,
2713 				enum nl80211_band band,
2714 				struct rs_rate *rate)
2715 {
2716 	int i, nentries;
2717 	unsigned long active_rate;
2718 	s8 best_rssi = S8_MIN;
2719 	u8 best_ant = ANT_NONE;
2720 	u8 valid_tx_ant = iwl_mvm_get_valid_tx_ant(mvm);
2721 	const struct rs_init_rate_info *initial_rates;
2722 
2723 	for (i = 0; i < ARRAY_SIZE(lq_sta->pers.chain_signal); i++) {
2724 		if (!(lq_sta->pers.chains & BIT(i)))
2725 			continue;
2726 
2727 		if (lq_sta->pers.chain_signal[i] > best_rssi) {
2728 			best_rssi = lq_sta->pers.chain_signal[i];
2729 			best_ant = BIT(i);
2730 		}
2731 	}
2732 
2733 	IWL_DEBUG_RATE(mvm, "Best ANT: %s Best RSSI: %d\n",
2734 		       rs_pretty_ant(best_ant), best_rssi);
2735 
2736 	if (best_ant != ANT_A && best_ant != ANT_B)
2737 		rate->ant = first_antenna(valid_tx_ant);
2738 	else
2739 		rate->ant = best_ant;
2740 
2741 	rate->sgi = false;
2742 	rate->ldpc = false;
2743 	rate->bw = RATE_MCS_CHAN_WIDTH_20;
2744 
2745 	rate->index = find_first_bit(&lq_sta->active_legacy_rate,
2746 				     BITS_PER_LONG);
2747 
2748 	if (band == NL80211_BAND_5GHZ) {
2749 		rate->type = LQ_LEGACY_A;
2750 		initial_rates = rs_optimal_rates_5ghz_legacy;
2751 		nentries = ARRAY_SIZE(rs_optimal_rates_5ghz_legacy);
2752 	} else {
2753 		rate->type = LQ_LEGACY_G;
2754 		initial_rates = rs_optimal_rates_24ghz_legacy;
2755 		nentries = ARRAY_SIZE(rs_optimal_rates_24ghz_legacy);
2756 	}
2757 
2758 	if (!IWL_MVM_RS_RSSI_BASED_INIT_RATE)
2759 		goto out;
2760 
2761 	/* Start from a higher rate if the corresponding debug capability
2762 	 * is enabled. The rate is chosen according to AP capabilities.
2763 	 * In case of VHT/HT when the rssi is low fallback to the case of
2764 	 * legacy rates.
2765 	 */
2766 	if (sta->vht_cap.vht_supported &&
2767 	    best_rssi > IWL_RS_LOW_RSSI_THRESHOLD) {
2768 		switch (sta->bandwidth) {
2769 		case IEEE80211_STA_RX_BW_160:
2770 		case IEEE80211_STA_RX_BW_80:
2771 		case IEEE80211_STA_RX_BW_40:
2772 			initial_rates = rs_optimal_rates_vht;
2773 			nentries = ARRAY_SIZE(rs_optimal_rates_vht);
2774 			break;
2775 		case IEEE80211_STA_RX_BW_20:
2776 			initial_rates = rs_optimal_rates_vht_20mhz;
2777 			nentries = ARRAY_SIZE(rs_optimal_rates_vht_20mhz);
2778 			break;
2779 		default:
2780 			IWL_ERR(mvm, "Invalid BW %d\n", sta->bandwidth);
2781 			goto out;
2782 		}
2783 
2784 		active_rate = lq_sta->active_siso_rate;
2785 		rate->type = LQ_VHT_SISO;
2786 		rate->bw = rs_bw_from_sta_bw(sta);
2787 	} else if (sta->ht_cap.ht_supported &&
2788 		   best_rssi > IWL_RS_LOW_RSSI_THRESHOLD) {
2789 		initial_rates = rs_optimal_rates_ht;
2790 		nentries = ARRAY_SIZE(rs_optimal_rates_ht);
2791 		active_rate = lq_sta->active_siso_rate;
2792 		rate->type = LQ_HT_SISO;
2793 	} else {
2794 		active_rate = lq_sta->active_legacy_rate;
2795 	}
2796 
2797 	for (i = 0; i < nentries; i++) {
2798 		int rate_idx = initial_rates[i].rate_idx;
2799 
2800 		if ((best_rssi >= initial_rates[i].rssi) &&
2801 		    (BIT(rate_idx) & active_rate)) {
2802 			rate->index = rate_idx;
2803 			break;
2804 		}
2805 	}
2806 
2807 out:
2808 	rs_dump_rate(mvm, rate, "INITIAL");
2809 }
2810 
2811 /* Save info about RSSI of last Rx */
2812 void rs_update_last_rssi(struct iwl_mvm *mvm,
2813 			 struct iwl_lq_sta *lq_sta,
2814 			 struct ieee80211_rx_status *rx_status)
2815 {
2816 	int i;
2817 
2818 	lq_sta->pers.chains = rx_status->chains;
2819 	lq_sta->pers.chain_signal[0] = rx_status->chain_signal[0];
2820 	lq_sta->pers.chain_signal[1] = rx_status->chain_signal[1];
2821 	lq_sta->pers.chain_signal[2] = rx_status->chain_signal[2];
2822 	lq_sta->pers.last_rssi = S8_MIN;
2823 
2824 	for (i = 0; i < ARRAY_SIZE(lq_sta->pers.chain_signal); i++) {
2825 		if (!(lq_sta->pers.chains & BIT(i)))
2826 			continue;
2827 
2828 		if (lq_sta->pers.chain_signal[i] > lq_sta->pers.last_rssi)
2829 			lq_sta->pers.last_rssi = lq_sta->pers.chain_signal[i];
2830 	}
2831 }
2832 
2833 /**
2834  * rs_initialize_lq - Initialize a station's hardware rate table
2835  *
2836  * The uCode's station table contains a table of fallback rates
2837  * for automatic fallback during transmission.
2838  *
2839  * NOTE: This sets up a default set of values.  These will be replaced later
2840  *       if the driver's iwl-agn-rs rate scaling algorithm is used, instead of
2841  *       rc80211_simple.
2842  *
2843  * NOTE: Run REPLY_ADD_STA command to set up station table entry, before
2844  *       calling this function (which runs REPLY_TX_LINK_QUALITY_CMD,
2845  *       which requires station table entry to exist).
2846  */
2847 static void rs_initialize_lq(struct iwl_mvm *mvm,
2848 			     struct ieee80211_sta *sta,
2849 			     struct iwl_lq_sta *lq_sta,
2850 			     enum nl80211_band band,
2851 			     bool init)
2852 {
2853 	struct iwl_scale_tbl_info *tbl;
2854 	struct rs_rate *rate;
2855 	u8 active_tbl = 0;
2856 
2857 	if (!sta || !lq_sta)
2858 		return;
2859 
2860 	if (!lq_sta->search_better_tbl)
2861 		active_tbl = lq_sta->active_tbl;
2862 	else
2863 		active_tbl = 1 - lq_sta->active_tbl;
2864 
2865 	tbl = &(lq_sta->lq_info[active_tbl]);
2866 	rate = &tbl->rate;
2867 
2868 	rs_get_initial_rate(mvm, sta, lq_sta, band, rate);
2869 	rs_init_optimal_rate(mvm, sta, lq_sta);
2870 
2871 	WARN_ON_ONCE(rate->ant != ANT_A && rate->ant != ANT_B);
2872 	tbl->column = rs_get_column_from_rate(rate);
2873 
2874 	rs_set_expected_tpt_table(lq_sta, tbl);
2875 	rs_fill_lq_cmd(mvm, sta, lq_sta, rate);
2876 	/* TODO restore station should remember the lq cmd */
2877 	iwl_mvm_send_lq_cmd(mvm, &lq_sta->lq, init);
2878 }
2879 
2880 static void rs_get_rate(void *mvm_r, struct ieee80211_sta *sta, void *mvm_sta,
2881 			struct ieee80211_tx_rate_control *txrc)
2882 {
2883 	struct sk_buff *skb = txrc->skb;
2884 	struct iwl_op_mode *op_mode __maybe_unused =
2885 			(struct iwl_op_mode *)mvm_r;
2886 	struct iwl_mvm *mvm __maybe_unused = IWL_OP_MODE_GET_MVM(op_mode);
2887 	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
2888 	struct iwl_lq_sta *lq_sta = mvm_sta;
2889 	struct rs_rate *optimal_rate;
2890 	u32 last_ucode_rate;
2891 
2892 	if (sta && !iwl_mvm_sta_from_mac80211(sta)->vif) {
2893 		/* if vif isn't initialized mvm doesn't know about
2894 		 * this station, so don't do anything with the it
2895 		 */
2896 		sta = NULL;
2897 		mvm_sta = NULL;
2898 	}
2899 
2900 	/* TODO: handle rate_idx_mask and rate_idx_mcs_mask */
2901 
2902 	/* Treat uninitialized rate scaling data same as non-existing. */
2903 	if (lq_sta && !lq_sta->pers.drv) {
2904 		IWL_DEBUG_RATE(mvm, "Rate scaling not initialized yet.\n");
2905 		mvm_sta = NULL;
2906 	}
2907 
2908 	/* Send management frames and NO_ACK data using lowest rate. */
2909 	if (rate_control_send_low(sta, mvm_sta, txrc))
2910 		return;
2911 
2912 	iwl_mvm_hwrate_to_tx_rate(lq_sta->last_rate_n_flags,
2913 				  info->band, &info->control.rates[0]);
2914 	info->control.rates[0].count = 1;
2915 
2916 	/* Report the optimal rate based on rssi and STA caps if we haven't
2917 	 * converged yet (too little traffic) or exploring other modulations
2918 	 */
2919 	if (lq_sta->rs_state != RS_STATE_STAY_IN_COLUMN) {
2920 		optimal_rate = rs_get_optimal_rate(mvm, lq_sta);
2921 		last_ucode_rate = ucode_rate_from_rs_rate(mvm,
2922 							  optimal_rate);
2923 		iwl_mvm_hwrate_to_tx_rate(last_ucode_rate, info->band,
2924 					  &txrc->reported_rate);
2925 	}
2926 }
2927 
2928 static void *rs_alloc_sta(void *mvm_rate, struct ieee80211_sta *sta,
2929 			  gfp_t gfp)
2930 {
2931 	struct iwl_mvm_sta *sta_priv = iwl_mvm_sta_from_mac80211(sta);
2932 	struct iwl_op_mode *op_mode = (struct iwl_op_mode *)mvm_rate;
2933 	struct iwl_mvm *mvm  = IWL_OP_MODE_GET_MVM(op_mode);
2934 	struct iwl_lq_sta *lq_sta = &sta_priv->lq_sta;
2935 
2936 	IWL_DEBUG_RATE(mvm, "create station rate scale window\n");
2937 
2938 	lq_sta->pers.drv = mvm;
2939 #ifdef CONFIG_MAC80211_DEBUGFS
2940 	lq_sta->pers.dbg_fixed_rate = 0;
2941 	lq_sta->pers.dbg_fixed_txp_reduction = TPC_INVALID;
2942 	lq_sta->pers.ss_force = RS_SS_FORCE_NONE;
2943 #endif
2944 	lq_sta->pers.chains = 0;
2945 	memset(lq_sta->pers.chain_signal, 0, sizeof(lq_sta->pers.chain_signal));
2946 	lq_sta->pers.last_rssi = S8_MIN;
2947 
2948 	return &sta_priv->lq_sta;
2949 }
2950 
2951 static int rs_vht_highest_rx_mcs_index(struct ieee80211_sta_vht_cap *vht_cap,
2952 				       int nss)
2953 {
2954 	u16 rx_mcs = le16_to_cpu(vht_cap->vht_mcs.rx_mcs_map) &
2955 		(0x3 << (2 * (nss - 1)));
2956 	rx_mcs >>= (2 * (nss - 1));
2957 
2958 	if (rx_mcs == IEEE80211_VHT_MCS_SUPPORT_0_7)
2959 		return IWL_RATE_MCS_7_INDEX;
2960 	else if (rx_mcs == IEEE80211_VHT_MCS_SUPPORT_0_8)
2961 		return IWL_RATE_MCS_8_INDEX;
2962 	else if (rx_mcs == IEEE80211_VHT_MCS_SUPPORT_0_9)
2963 		return IWL_RATE_MCS_9_INDEX;
2964 
2965 	WARN_ON_ONCE(rx_mcs != IEEE80211_VHT_MCS_NOT_SUPPORTED);
2966 	return -1;
2967 }
2968 
2969 static void rs_vht_set_enabled_rates(struct ieee80211_sta *sta,
2970 				     struct ieee80211_sta_vht_cap *vht_cap,
2971 				     struct iwl_lq_sta *lq_sta)
2972 {
2973 	int i;
2974 	int highest_mcs = rs_vht_highest_rx_mcs_index(vht_cap, 1);
2975 
2976 	if (highest_mcs >= IWL_RATE_MCS_0_INDEX) {
2977 		for (i = IWL_RATE_MCS_0_INDEX; i <= highest_mcs; i++) {
2978 			if (i == IWL_RATE_9M_INDEX)
2979 				continue;
2980 
2981 			/* VHT MCS9 isn't valid for 20Mhz for NSS=1,2 */
2982 			if (i == IWL_RATE_MCS_9_INDEX &&
2983 			    sta->bandwidth == IEEE80211_STA_RX_BW_20)
2984 				continue;
2985 
2986 			lq_sta->active_siso_rate |= BIT(i);
2987 		}
2988 	}
2989 
2990 	if (sta->rx_nss < 2)
2991 		return;
2992 
2993 	highest_mcs = rs_vht_highest_rx_mcs_index(vht_cap, 2);
2994 	if (highest_mcs >= IWL_RATE_MCS_0_INDEX) {
2995 		for (i = IWL_RATE_MCS_0_INDEX; i <= highest_mcs; i++) {
2996 			if (i == IWL_RATE_9M_INDEX)
2997 				continue;
2998 
2999 			/* VHT MCS9 isn't valid for 20Mhz for NSS=1,2 */
3000 			if (i == IWL_RATE_MCS_9_INDEX &&
3001 			    sta->bandwidth == IEEE80211_STA_RX_BW_20)
3002 				continue;
3003 
3004 			lq_sta->active_mimo2_rate |= BIT(i);
3005 		}
3006 	}
3007 }
3008 
3009 static void rs_ht_init(struct iwl_mvm *mvm,
3010 		       struct ieee80211_sta *sta,
3011 		       struct iwl_lq_sta *lq_sta,
3012 		       struct ieee80211_sta_ht_cap *ht_cap)
3013 {
3014 	/* active_siso_rate mask includes 9 MBits (bit 5),
3015 	 * and CCK (bits 0-3), supp_rates[] does not;
3016 	 * shift to convert format, force 9 MBits off.
3017 	 */
3018 	lq_sta->active_siso_rate = ht_cap->mcs.rx_mask[0] << 1;
3019 	lq_sta->active_siso_rate |= ht_cap->mcs.rx_mask[0] & 0x1;
3020 	lq_sta->active_siso_rate &= ~((u16)0x2);
3021 	lq_sta->active_siso_rate <<= IWL_FIRST_OFDM_RATE;
3022 
3023 	lq_sta->active_mimo2_rate = ht_cap->mcs.rx_mask[1] << 1;
3024 	lq_sta->active_mimo2_rate |= ht_cap->mcs.rx_mask[1] & 0x1;
3025 	lq_sta->active_mimo2_rate &= ~((u16)0x2);
3026 	lq_sta->active_mimo2_rate <<= IWL_FIRST_OFDM_RATE;
3027 
3028 	if (mvm->cfg->ht_params->ldpc &&
3029 	    (ht_cap->cap & IEEE80211_HT_CAP_LDPC_CODING))
3030 		lq_sta->ldpc = true;
3031 
3032 	if (mvm->cfg->ht_params->stbc &&
3033 	    (num_of_ant(iwl_mvm_get_valid_tx_ant(mvm)) > 1) &&
3034 	    (ht_cap->cap & IEEE80211_HT_CAP_RX_STBC))
3035 		lq_sta->stbc_capable = true;
3036 
3037 	lq_sta->is_vht = false;
3038 }
3039 
3040 static void rs_vht_init(struct iwl_mvm *mvm,
3041 			struct ieee80211_sta *sta,
3042 			struct iwl_lq_sta *lq_sta,
3043 			struct ieee80211_sta_vht_cap *vht_cap)
3044 {
3045 	rs_vht_set_enabled_rates(sta, vht_cap, lq_sta);
3046 
3047 	if (mvm->cfg->ht_params->ldpc &&
3048 	    (vht_cap->cap & IEEE80211_VHT_CAP_RXLDPC))
3049 		lq_sta->ldpc = true;
3050 
3051 	if (mvm->cfg->ht_params->stbc &&
3052 	    (num_of_ant(iwl_mvm_get_valid_tx_ant(mvm)) > 1) &&
3053 	    (vht_cap->cap & IEEE80211_VHT_CAP_RXSTBC_MASK))
3054 		lq_sta->stbc_capable = true;
3055 
3056 	if (fw_has_capa(&mvm->fw->ucode_capa, IWL_UCODE_TLV_CAPA_BEAMFORMER) &&
3057 	    (num_of_ant(iwl_mvm_get_valid_tx_ant(mvm)) > 1) &&
3058 	    (vht_cap->cap & IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE))
3059 		lq_sta->bfer_capable = true;
3060 
3061 	lq_sta->is_vht = true;
3062 }
3063 
3064 #ifdef CONFIG_IWLWIFI_DEBUGFS
3065 static void iwl_mvm_reset_frame_stats(struct iwl_mvm *mvm)
3066 {
3067 	spin_lock_bh(&mvm->drv_stats_lock);
3068 	memset(&mvm->drv_rx_stats, 0, sizeof(mvm->drv_rx_stats));
3069 	spin_unlock_bh(&mvm->drv_stats_lock);
3070 }
3071 
3072 void iwl_mvm_update_frame_stats(struct iwl_mvm *mvm, u32 rate, bool agg)
3073 {
3074 	u8 nss = 0, mcs = 0;
3075 
3076 	spin_lock(&mvm->drv_stats_lock);
3077 
3078 	if (agg)
3079 		mvm->drv_rx_stats.agg_frames++;
3080 
3081 	mvm->drv_rx_stats.success_frames++;
3082 
3083 	switch (rate & RATE_MCS_CHAN_WIDTH_MSK) {
3084 	case RATE_MCS_CHAN_WIDTH_20:
3085 		mvm->drv_rx_stats.bw_20_frames++;
3086 		break;
3087 	case RATE_MCS_CHAN_WIDTH_40:
3088 		mvm->drv_rx_stats.bw_40_frames++;
3089 		break;
3090 	case RATE_MCS_CHAN_WIDTH_80:
3091 		mvm->drv_rx_stats.bw_80_frames++;
3092 		break;
3093 	case RATE_MCS_CHAN_WIDTH_160:
3094 		mvm->drv_rx_stats.bw_160_frames++;
3095 		break;
3096 	default:
3097 		WARN_ONCE(1, "bad BW. rate 0x%x", rate);
3098 	}
3099 
3100 	if (rate & RATE_MCS_HT_MSK) {
3101 		mvm->drv_rx_stats.ht_frames++;
3102 		mcs = rate & RATE_HT_MCS_RATE_CODE_MSK;
3103 		nss = ((rate & RATE_HT_MCS_NSS_MSK) >> RATE_HT_MCS_NSS_POS) + 1;
3104 	} else if (rate & RATE_MCS_VHT_MSK) {
3105 		mvm->drv_rx_stats.vht_frames++;
3106 		mcs = rate & RATE_VHT_MCS_RATE_CODE_MSK;
3107 		nss = ((rate & RATE_VHT_MCS_NSS_MSK) >>
3108 		       RATE_VHT_MCS_NSS_POS) + 1;
3109 	} else {
3110 		mvm->drv_rx_stats.legacy_frames++;
3111 	}
3112 
3113 	if (nss == 1)
3114 		mvm->drv_rx_stats.siso_frames++;
3115 	else if (nss == 2)
3116 		mvm->drv_rx_stats.mimo2_frames++;
3117 
3118 	if (rate & RATE_MCS_SGI_MSK)
3119 		mvm->drv_rx_stats.sgi_frames++;
3120 	else
3121 		mvm->drv_rx_stats.ngi_frames++;
3122 
3123 	mvm->drv_rx_stats.last_rates[mvm->drv_rx_stats.last_frame_idx] = rate;
3124 	mvm->drv_rx_stats.last_frame_idx =
3125 		(mvm->drv_rx_stats.last_frame_idx + 1) %
3126 			ARRAY_SIZE(mvm->drv_rx_stats.last_rates);
3127 
3128 	spin_unlock(&mvm->drv_stats_lock);
3129 }
3130 #endif
3131 
3132 /*
3133  * Called after adding a new station to initialize rate scaling
3134  */
3135 void iwl_mvm_rs_rate_init(struct iwl_mvm *mvm, struct ieee80211_sta *sta,
3136 			  enum nl80211_band band, bool init)
3137 {
3138 	int i, j;
3139 	struct ieee80211_hw *hw = mvm->hw;
3140 	struct ieee80211_sta_ht_cap *ht_cap = &sta->ht_cap;
3141 	struct ieee80211_sta_vht_cap *vht_cap = &sta->vht_cap;
3142 	struct iwl_mvm_sta *sta_priv = iwl_mvm_sta_from_mac80211(sta);
3143 	struct iwl_lq_sta *lq_sta = &sta_priv->lq_sta;
3144 	struct ieee80211_supported_band *sband;
3145 	unsigned long supp; /* must be unsigned long for for_each_set_bit */
3146 
3147 	/* clear all non-persistent lq data */
3148 	memset(lq_sta, 0, offsetof(typeof(*lq_sta), pers));
3149 
3150 	sband = hw->wiphy->bands[band];
3151 
3152 	lq_sta->lq.sta_id = sta_priv->sta_id;
3153 	sta_priv->tlc_amsdu = false;
3154 
3155 	for (j = 0; j < LQ_SIZE; j++)
3156 		rs_rate_scale_clear_tbl_windows(mvm, &lq_sta->lq_info[j]);
3157 
3158 	lq_sta->flush_timer = 0;
3159 	lq_sta->last_tx = jiffies;
3160 
3161 	IWL_DEBUG_RATE(mvm,
3162 		       "LQ: *** rate scale station global init for station %d ***\n",
3163 		       sta_priv->sta_id);
3164 	/* TODO: what is a good starting rate for STA? About middle? Maybe not
3165 	 * the lowest or the highest rate.. Could consider using RSSI from
3166 	 * previous packets? Need to have IEEE 802.1X auth succeed immediately
3167 	 * after assoc.. */
3168 
3169 	lq_sta->missed_rate_counter = IWL_MVM_RS_MISSED_RATE_MAX;
3170 	lq_sta->band = sband->band;
3171 	/*
3172 	 * active legacy rates as per supported rates bitmap
3173 	 */
3174 	supp = sta->supp_rates[sband->band];
3175 	lq_sta->active_legacy_rate = 0;
3176 	for_each_set_bit(i, &supp, BITS_PER_LONG)
3177 		lq_sta->active_legacy_rate |= BIT(sband->bitrates[i].hw_value);
3178 
3179 	/* TODO: should probably account for rx_highest for both HT/VHT */
3180 	if (!vht_cap || !vht_cap->vht_supported)
3181 		rs_ht_init(mvm, sta, lq_sta, ht_cap);
3182 	else
3183 		rs_vht_init(mvm, sta, lq_sta, vht_cap);
3184 
3185 	lq_sta->max_legacy_rate_idx =
3186 		rs_get_max_rate_from_mask(lq_sta->active_legacy_rate);
3187 	lq_sta->max_siso_rate_idx =
3188 		rs_get_max_rate_from_mask(lq_sta->active_siso_rate);
3189 	lq_sta->max_mimo2_rate_idx =
3190 		rs_get_max_rate_from_mask(lq_sta->active_mimo2_rate);
3191 
3192 	IWL_DEBUG_RATE(mvm,
3193 		       "LEGACY=%lX SISO=%lX MIMO2=%lX VHT=%d LDPC=%d STBC=%d BFER=%d\n",
3194 		       lq_sta->active_legacy_rate,
3195 		       lq_sta->active_siso_rate,
3196 		       lq_sta->active_mimo2_rate,
3197 		       lq_sta->is_vht, lq_sta->ldpc, lq_sta->stbc_capable,
3198 		       lq_sta->bfer_capable);
3199 	IWL_DEBUG_RATE(mvm, "MAX RATE: LEGACY=%d SISO=%d MIMO2=%d\n",
3200 		       lq_sta->max_legacy_rate_idx,
3201 		       lq_sta->max_siso_rate_idx,
3202 		       lq_sta->max_mimo2_rate_idx);
3203 
3204 	/* These values will be overridden later */
3205 	lq_sta->lq.single_stream_ant_msk =
3206 		first_antenna(iwl_mvm_get_valid_tx_ant(mvm));
3207 	lq_sta->lq.dual_stream_ant_msk = ANT_AB;
3208 
3209 	/* as default allow aggregation for all tids */
3210 	lq_sta->tx_agg_tid_en = IWL_AGG_ALL_TID;
3211 	lq_sta->is_agg = 0;
3212 #ifdef CONFIG_IWLWIFI_DEBUGFS
3213 	iwl_mvm_reset_frame_stats(mvm);
3214 #endif
3215 	rs_initialize_lq(mvm, sta, lq_sta, band, init);
3216 }
3217 
3218 static void rs_rate_update(void *mvm_r,
3219 			   struct ieee80211_supported_band *sband,
3220 			   struct cfg80211_chan_def *chandef,
3221 			   struct ieee80211_sta *sta, void *priv_sta,
3222 			   u32 changed)
3223 {
3224 	u8 tid;
3225 	struct iwl_op_mode *op_mode  =
3226 			(struct iwl_op_mode *)mvm_r;
3227 	struct iwl_mvm *mvm = IWL_OP_MODE_GET_MVM(op_mode);
3228 
3229 	if (!iwl_mvm_sta_from_mac80211(sta)->vif)
3230 		return;
3231 
3232 	/* Stop any ongoing aggregations as rs starts off assuming no agg */
3233 	for (tid = 0; tid < IWL_MAX_TID_COUNT; tid++)
3234 		ieee80211_stop_tx_ba_session(sta, tid);
3235 
3236 	iwl_mvm_rs_rate_init(mvm, sta, sband->band, false);
3237 }
3238 
3239 #ifdef CONFIG_MAC80211_DEBUGFS
3240 static void rs_build_rates_table_from_fixed(struct iwl_mvm *mvm,
3241 					    struct iwl_lq_cmd *lq_cmd,
3242 					    enum nl80211_band band,
3243 					    u32 ucode_rate)
3244 {
3245 	struct rs_rate rate;
3246 	int i;
3247 	int num_rates = ARRAY_SIZE(lq_cmd->rs_table);
3248 	__le32 ucode_rate_le32 = cpu_to_le32(ucode_rate);
3249 	u8 ant = (ucode_rate & RATE_MCS_ANT_ABC_MSK) >> RATE_MCS_ANT_POS;
3250 
3251 	for (i = 0; i < num_rates; i++)
3252 		lq_cmd->rs_table[i] = ucode_rate_le32;
3253 
3254 	rs_rate_from_ucode_rate(ucode_rate, band, &rate);
3255 
3256 	if (is_mimo(&rate))
3257 		lq_cmd->mimo_delim = num_rates - 1;
3258 	else
3259 		lq_cmd->mimo_delim = 0;
3260 
3261 	lq_cmd->reduced_tpc = 0;
3262 
3263 	if (num_of_ant(ant) == 1)
3264 		lq_cmd->single_stream_ant_msk = ant;
3265 
3266 	lq_cmd->agg_frame_cnt_limit = LINK_QUAL_AGG_FRAME_LIMIT_DEF;
3267 }
3268 #endif /* CONFIG_MAC80211_DEBUGFS */
3269 
3270 static void rs_fill_rates_for_column(struct iwl_mvm *mvm,
3271 				     struct iwl_lq_sta *lq_sta,
3272 				     struct rs_rate *rate,
3273 				     __le32 *rs_table, int *rs_table_index,
3274 				     int num_rates, int num_retries,
3275 				     u8 valid_tx_ant, bool toggle_ant)
3276 {
3277 	int i, j;
3278 	__le32 ucode_rate;
3279 	bool bottom_reached = false;
3280 	int prev_rate_idx = rate->index;
3281 	int end = LINK_QUAL_MAX_RETRY_NUM;
3282 	int index = *rs_table_index;
3283 
3284 	for (i = 0; i < num_rates && index < end; i++) {
3285 		for (j = 0; j < num_retries && index < end; j++, index++) {
3286 			ucode_rate = cpu_to_le32(ucode_rate_from_rs_rate(mvm,
3287 									 rate));
3288 			rs_table[index] = ucode_rate;
3289 			if (toggle_ant)
3290 				rs_toggle_antenna(valid_tx_ant, rate);
3291 		}
3292 
3293 		prev_rate_idx = rate->index;
3294 		bottom_reached = rs_get_lower_rate_in_column(lq_sta, rate);
3295 		if (bottom_reached && !is_legacy(rate))
3296 			break;
3297 	}
3298 
3299 	if (!bottom_reached && !is_legacy(rate))
3300 		rate->index = prev_rate_idx;
3301 
3302 	*rs_table_index = index;
3303 }
3304 
3305 /* Building the rate table is non trivial. When we're in MIMO2/VHT/80Mhz/SGI
3306  * column the rate table should look like this:
3307  *
3308  * rate[0] 0x400D019 VHT | ANT: AB BW: 80Mhz MCS: 9 NSS: 2 SGI
3309  * rate[1] 0x400D019 VHT | ANT: AB BW: 80Mhz MCS: 9 NSS: 2 SGI
3310  * rate[2] 0x400D018 VHT | ANT: AB BW: 80Mhz MCS: 8 NSS: 2 SGI
3311  * rate[3] 0x400D018 VHT | ANT: AB BW: 80Mhz MCS: 8 NSS: 2 SGI
3312  * rate[4] 0x400D017 VHT | ANT: AB BW: 80Mhz MCS: 7 NSS: 2 SGI
3313  * rate[5] 0x400D017 VHT | ANT: AB BW: 80Mhz MCS: 7 NSS: 2 SGI
3314  * rate[6] 0x4005007 VHT | ANT: A BW: 80Mhz MCS: 7 NSS: 1 NGI
3315  * rate[7] 0x4009006 VHT | ANT: B BW: 80Mhz MCS: 6 NSS: 1 NGI
3316  * rate[8] 0x4005005 VHT | ANT: A BW: 80Mhz MCS: 5 NSS: 1 NGI
3317  * rate[9] 0x800B Legacy | ANT: B Rate: 36 Mbps
3318  * rate[10] 0x4009 Legacy | ANT: A Rate: 24 Mbps
3319  * rate[11] 0x8007 Legacy | ANT: B Rate: 18 Mbps
3320  * rate[12] 0x4005 Legacy | ANT: A Rate: 12 Mbps
3321  * rate[13] 0x800F Legacy | ANT: B Rate: 9 Mbps
3322  * rate[14] 0x400D Legacy | ANT: A Rate: 6 Mbps
3323  * rate[15] 0x800D Legacy | ANT: B Rate: 6 Mbps
3324  */
3325 static void rs_build_rates_table(struct iwl_mvm *mvm,
3326 				 struct ieee80211_sta *sta,
3327 				 struct iwl_lq_sta *lq_sta,
3328 				 const struct rs_rate *initial_rate)
3329 {
3330 	struct rs_rate rate;
3331 	int num_rates, num_retries, index = 0;
3332 	u8 valid_tx_ant = 0;
3333 	struct iwl_lq_cmd *lq_cmd = &lq_sta->lq;
3334 	bool toggle_ant = false;
3335 
3336 	memcpy(&rate, initial_rate, sizeof(rate));
3337 
3338 	valid_tx_ant = iwl_mvm_get_valid_tx_ant(mvm);
3339 
3340 	/* TODO: remove old API when min FW API hits 14 */
3341 	if (!fw_has_api(&mvm->fw->ucode_capa, IWL_UCODE_TLV_API_LQ_SS_PARAMS) &&
3342 	    rs_stbc_allow(mvm, sta, lq_sta))
3343 		rate.stbc = true;
3344 
3345 	if (is_siso(&rate)) {
3346 		num_rates = IWL_MVM_RS_INITIAL_SISO_NUM_RATES;
3347 		num_retries = IWL_MVM_RS_HT_VHT_RETRIES_PER_RATE;
3348 	} else if (is_mimo(&rate)) {
3349 		num_rates = IWL_MVM_RS_INITIAL_MIMO_NUM_RATES;
3350 		num_retries = IWL_MVM_RS_HT_VHT_RETRIES_PER_RATE;
3351 	} else {
3352 		num_rates = IWL_MVM_RS_INITIAL_LEGACY_NUM_RATES;
3353 		num_retries = IWL_MVM_RS_INITIAL_LEGACY_RETRIES;
3354 		toggle_ant = true;
3355 	}
3356 
3357 	rs_fill_rates_for_column(mvm, lq_sta, &rate, lq_cmd->rs_table, &index,
3358 				 num_rates, num_retries, valid_tx_ant,
3359 				 toggle_ant);
3360 
3361 	rs_get_lower_rate_down_column(lq_sta, &rate);
3362 
3363 	if (is_siso(&rate)) {
3364 		num_rates = IWL_MVM_RS_SECONDARY_SISO_NUM_RATES;
3365 		num_retries = IWL_MVM_RS_SECONDARY_SISO_RETRIES;
3366 		lq_cmd->mimo_delim = index;
3367 	} else if (is_legacy(&rate)) {
3368 		num_rates = IWL_MVM_RS_SECONDARY_LEGACY_NUM_RATES;
3369 		num_retries = IWL_MVM_RS_SECONDARY_LEGACY_RETRIES;
3370 	} else {
3371 		WARN_ON_ONCE(1);
3372 	}
3373 
3374 	toggle_ant = true;
3375 
3376 	rs_fill_rates_for_column(mvm, lq_sta, &rate, lq_cmd->rs_table, &index,
3377 				 num_rates, num_retries, valid_tx_ant,
3378 				 toggle_ant);
3379 
3380 	rs_get_lower_rate_down_column(lq_sta, &rate);
3381 
3382 	num_rates = IWL_MVM_RS_SECONDARY_LEGACY_NUM_RATES;
3383 	num_retries = IWL_MVM_RS_SECONDARY_LEGACY_RETRIES;
3384 
3385 	rs_fill_rates_for_column(mvm, lq_sta, &rate, lq_cmd->rs_table, &index,
3386 				 num_rates, num_retries, valid_tx_ant,
3387 				 toggle_ant);
3388 
3389 }
3390 
3391 struct rs_bfer_active_iter_data {
3392 	struct ieee80211_sta *exclude_sta;
3393 	struct iwl_mvm_sta *bfer_mvmsta;
3394 };
3395 
3396 static void rs_bfer_active_iter(void *_data,
3397 				struct ieee80211_sta *sta)
3398 {
3399 	struct rs_bfer_active_iter_data *data = _data;
3400 	struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
3401 	struct iwl_lq_cmd *lq_cmd = &mvmsta->lq_sta.lq;
3402 	u32 ss_params = le32_to_cpu(lq_cmd->ss_params);
3403 
3404 	if (sta == data->exclude_sta)
3405 		return;
3406 
3407 	/* The current sta has BFER allowed */
3408 	if (ss_params & LQ_SS_BFER_ALLOWED) {
3409 		WARN_ON_ONCE(data->bfer_mvmsta != NULL);
3410 
3411 		data->bfer_mvmsta = mvmsta;
3412 	}
3413 }
3414 
3415 static int rs_bfer_priority(struct iwl_mvm_sta *sta)
3416 {
3417 	int prio = -1;
3418 	enum nl80211_iftype viftype = ieee80211_vif_type_p2p(sta->vif);
3419 
3420 	switch (viftype) {
3421 	case NL80211_IFTYPE_AP:
3422 	case NL80211_IFTYPE_P2P_GO:
3423 		prio = 3;
3424 		break;
3425 	case NL80211_IFTYPE_P2P_CLIENT:
3426 		prio = 2;
3427 		break;
3428 	case NL80211_IFTYPE_STATION:
3429 		prio = 1;
3430 		break;
3431 	default:
3432 		WARN_ONCE(true, "viftype %d sta_id %d", viftype, sta->sta_id);
3433 		prio = -1;
3434 	}
3435 
3436 	return prio;
3437 }
3438 
3439 /* Returns >0 if sta1 has a higher BFER priority compared to sta2 */
3440 static int rs_bfer_priority_cmp(struct iwl_mvm_sta *sta1,
3441 				struct iwl_mvm_sta *sta2)
3442 {
3443 	int prio1 = rs_bfer_priority(sta1);
3444 	int prio2 = rs_bfer_priority(sta2);
3445 
3446 	if (prio1 > prio2)
3447 		return 1;
3448 	if (prio1 < prio2)
3449 		return -1;
3450 	return 0;
3451 }
3452 
3453 static void rs_set_lq_ss_params(struct iwl_mvm *mvm,
3454 				struct ieee80211_sta *sta,
3455 				struct iwl_lq_sta *lq_sta,
3456 				const struct rs_rate *initial_rate)
3457 {
3458 	struct iwl_lq_cmd *lq_cmd = &lq_sta->lq;
3459 	struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
3460 	struct rs_bfer_active_iter_data data = {
3461 		.exclude_sta = sta,
3462 		.bfer_mvmsta = NULL,
3463 	};
3464 	struct iwl_mvm_sta *bfer_mvmsta = NULL;
3465 	u32 ss_params = LQ_SS_PARAMS_VALID;
3466 
3467 	if (!iwl_mvm_bt_coex_is_mimo_allowed(mvm, sta))
3468 		goto out;
3469 
3470 #ifdef CONFIG_MAC80211_DEBUGFS
3471 	/* Check if forcing the decision is configured.
3472 	 * Note that SISO is forced by not allowing STBC or BFER
3473 	 */
3474 	if (lq_sta->pers.ss_force == RS_SS_FORCE_STBC)
3475 		ss_params |= (LQ_SS_STBC_1SS_ALLOWED | LQ_SS_FORCE);
3476 	else if (lq_sta->pers.ss_force == RS_SS_FORCE_BFER)
3477 		ss_params |= (LQ_SS_BFER_ALLOWED | LQ_SS_FORCE);
3478 
3479 	if (lq_sta->pers.ss_force != RS_SS_FORCE_NONE) {
3480 		IWL_DEBUG_RATE(mvm, "Forcing single stream Tx decision %d\n",
3481 			       lq_sta->pers.ss_force);
3482 		goto out;
3483 	}
3484 #endif
3485 
3486 	if (lq_sta->stbc_capable)
3487 		ss_params |= LQ_SS_STBC_1SS_ALLOWED;
3488 
3489 	if (!lq_sta->bfer_capable)
3490 		goto out;
3491 
3492 	ieee80211_iterate_stations_atomic(mvm->hw,
3493 					  rs_bfer_active_iter,
3494 					  &data);
3495 	bfer_mvmsta = data.bfer_mvmsta;
3496 
3497 	/* This code is safe as it doesn't run concurrently for different
3498 	 * stations. This is guaranteed by the fact that calls to
3499 	 * ieee80211_tx_status wouldn't run concurrently for a single HW.
3500 	 */
3501 	if (!bfer_mvmsta) {
3502 		IWL_DEBUG_RATE(mvm, "No sta with BFER allowed found. Allow\n");
3503 
3504 		ss_params |= LQ_SS_BFER_ALLOWED;
3505 		goto out;
3506 	}
3507 
3508 	IWL_DEBUG_RATE(mvm, "Found existing sta %d with BFER activated\n",
3509 		       bfer_mvmsta->sta_id);
3510 
3511 	/* Disallow BFER on another STA if active and we're a higher priority */
3512 	if (rs_bfer_priority_cmp(mvmsta, bfer_mvmsta) > 0) {
3513 		struct iwl_lq_cmd *bfersta_lq_cmd = &bfer_mvmsta->lq_sta.lq;
3514 		u32 bfersta_ss_params = le32_to_cpu(bfersta_lq_cmd->ss_params);
3515 
3516 		bfersta_ss_params &= ~LQ_SS_BFER_ALLOWED;
3517 		bfersta_lq_cmd->ss_params = cpu_to_le32(bfersta_ss_params);
3518 		iwl_mvm_send_lq_cmd(mvm, bfersta_lq_cmd, false);
3519 
3520 		ss_params |= LQ_SS_BFER_ALLOWED;
3521 		IWL_DEBUG_RATE(mvm,
3522 			       "Lower priority BFER sta found (%d). Switch BFER\n",
3523 			       bfer_mvmsta->sta_id);
3524 	}
3525 out:
3526 	lq_cmd->ss_params = cpu_to_le32(ss_params);
3527 }
3528 
3529 static void rs_fill_lq_cmd(struct iwl_mvm *mvm,
3530 			   struct ieee80211_sta *sta,
3531 			   struct iwl_lq_sta *lq_sta,
3532 			   const struct rs_rate *initial_rate)
3533 {
3534 	struct iwl_lq_cmd *lq_cmd = &lq_sta->lq;
3535 	struct iwl_mvm_sta *mvmsta;
3536 	struct iwl_mvm_vif *mvmvif;
3537 
3538 	lq_cmd->agg_disable_start_th = IWL_MVM_RS_AGG_DISABLE_START;
3539 	lq_cmd->agg_time_limit =
3540 		cpu_to_le16(IWL_MVM_RS_AGG_TIME_LIMIT);
3541 
3542 #ifdef CONFIG_MAC80211_DEBUGFS
3543 	if (lq_sta->pers.dbg_fixed_rate) {
3544 		rs_build_rates_table_from_fixed(mvm, lq_cmd,
3545 						lq_sta->band,
3546 						lq_sta->pers.dbg_fixed_rate);
3547 		return;
3548 	}
3549 #endif
3550 	if (WARN_ON_ONCE(!sta || !initial_rate))
3551 		return;
3552 
3553 	rs_build_rates_table(mvm, sta, lq_sta, initial_rate);
3554 
3555 	if (fw_has_api(&mvm->fw->ucode_capa, IWL_UCODE_TLV_API_LQ_SS_PARAMS))
3556 		rs_set_lq_ss_params(mvm, sta, lq_sta, initial_rate);
3557 
3558 	mvmsta = iwl_mvm_sta_from_mac80211(sta);
3559 	mvmvif = iwl_mvm_vif_from_mac80211(mvmsta->vif);
3560 
3561 	if (num_of_ant(initial_rate->ant) == 1)
3562 		lq_cmd->single_stream_ant_msk = initial_rate->ant;
3563 
3564 	lq_cmd->agg_frame_cnt_limit = mvmsta->max_agg_bufsize;
3565 
3566 	/*
3567 	 * In case of low latency, tell the firmware to leave a frame in the
3568 	 * Tx Fifo so that it can start a transaction in the same TxOP. This
3569 	 * basically allows the firmware to send bursts.
3570 	 */
3571 	if (iwl_mvm_vif_low_latency(mvmvif))
3572 		lq_cmd->agg_frame_cnt_limit--;
3573 
3574 	if (mvmsta->vif->p2p)
3575 		lq_cmd->flags |= LQ_FLAG_USE_RTS_MSK;
3576 
3577 	lq_cmd->agg_time_limit =
3578 			cpu_to_le16(iwl_mvm_coex_agg_time_limit(mvm, sta));
3579 }
3580 
3581 static void *rs_alloc(struct ieee80211_hw *hw, struct dentry *debugfsdir)
3582 {
3583 	return hw->priv;
3584 }
3585 /* rate scale requires free function to be implemented */
3586 static void rs_free(void *mvm_rate)
3587 {
3588 	return;
3589 }
3590 
3591 static void rs_free_sta(void *mvm_r, struct ieee80211_sta *sta,
3592 			void *mvm_sta)
3593 {
3594 	struct iwl_op_mode *op_mode __maybe_unused = mvm_r;
3595 	struct iwl_mvm *mvm __maybe_unused = IWL_OP_MODE_GET_MVM(op_mode);
3596 
3597 	IWL_DEBUG_RATE(mvm, "enter\n");
3598 	IWL_DEBUG_RATE(mvm, "leave\n");
3599 }
3600 
3601 #ifdef CONFIG_MAC80211_DEBUGFS
3602 int rs_pretty_print_rate(char *buf, const u32 rate)
3603 {
3604 
3605 	char *type, *bw;
3606 	u8 mcs = 0, nss = 0;
3607 	u8 ant = (rate & RATE_MCS_ANT_ABC_MSK) >> RATE_MCS_ANT_POS;
3608 
3609 	if (!(rate & RATE_MCS_HT_MSK) &&
3610 	    !(rate & RATE_MCS_VHT_MSK)) {
3611 		int index = iwl_hwrate_to_plcp_idx(rate);
3612 
3613 		return sprintf(buf, "Legacy | ANT: %s Rate: %s Mbps\n",
3614 			       rs_pretty_ant(ant),
3615 			       index == IWL_RATE_INVALID ? "BAD" :
3616 			       iwl_rate_mcs[index].mbps);
3617 	}
3618 
3619 	if (rate & RATE_MCS_VHT_MSK) {
3620 		type = "VHT";
3621 		mcs = rate & RATE_VHT_MCS_RATE_CODE_MSK;
3622 		nss = ((rate & RATE_VHT_MCS_NSS_MSK)
3623 		       >> RATE_VHT_MCS_NSS_POS) + 1;
3624 	} else if (rate & RATE_MCS_HT_MSK) {
3625 		type = "HT";
3626 		mcs = rate & RATE_HT_MCS_INDEX_MSK;
3627 	} else {
3628 		type = "Unknown"; /* shouldn't happen */
3629 	}
3630 
3631 	switch (rate & RATE_MCS_CHAN_WIDTH_MSK) {
3632 	case RATE_MCS_CHAN_WIDTH_20:
3633 		bw = "20Mhz";
3634 		break;
3635 	case RATE_MCS_CHAN_WIDTH_40:
3636 		bw = "40Mhz";
3637 		break;
3638 	case RATE_MCS_CHAN_WIDTH_80:
3639 		bw = "80Mhz";
3640 		break;
3641 	case RATE_MCS_CHAN_WIDTH_160:
3642 		bw = "160Mhz";
3643 		break;
3644 	default:
3645 		bw = "BAD BW";
3646 	}
3647 
3648 	return sprintf(buf, "%s | ANT: %s BW: %s MCS: %d NSS: %d %s%s%s%s%s\n",
3649 		       type, rs_pretty_ant(ant), bw, mcs, nss,
3650 		       (rate & RATE_MCS_SGI_MSK) ? "SGI " : "NGI ",
3651 		       (rate & RATE_MCS_HT_STBC_MSK) ? "STBC " : "",
3652 		       (rate & RATE_MCS_LDPC_MSK) ? "LDPC " : "",
3653 		       (rate & RATE_MCS_BF_MSK) ? "BF " : "",
3654 		       (rate & RATE_MCS_ZLF_MSK) ? "ZLF " : "");
3655 }
3656 
3657 /**
3658  * Program the device to use fixed rate for frame transmit
3659  * This is for debugging/testing only
3660  * once the device start use fixed rate, we need to reload the module
3661  * to being back the normal operation.
3662  */
3663 static void rs_program_fix_rate(struct iwl_mvm *mvm,
3664 				struct iwl_lq_sta *lq_sta)
3665 {
3666 	lq_sta->active_legacy_rate = 0x0FFF;	/* 1 - 54 MBits, includes CCK */
3667 	lq_sta->active_siso_rate   = 0x1FD0;	/* 6 - 60 MBits, no 9, no CCK */
3668 	lq_sta->active_mimo2_rate  = 0x1FD0;	/* 6 - 60 MBits, no 9, no CCK */
3669 
3670 	IWL_DEBUG_RATE(mvm, "sta_id %d rate 0x%X\n",
3671 		       lq_sta->lq.sta_id, lq_sta->pers.dbg_fixed_rate);
3672 
3673 	if (lq_sta->pers.dbg_fixed_rate) {
3674 		rs_fill_lq_cmd(mvm, NULL, lq_sta, NULL);
3675 		iwl_mvm_send_lq_cmd(lq_sta->pers.drv, &lq_sta->lq, false);
3676 	}
3677 }
3678 
3679 static ssize_t rs_sta_dbgfs_scale_table_write(struct file *file,
3680 			const char __user *user_buf, size_t count, loff_t *ppos)
3681 {
3682 	struct iwl_lq_sta *lq_sta = file->private_data;
3683 	struct iwl_mvm *mvm;
3684 	char buf[64];
3685 	size_t buf_size;
3686 	u32 parsed_rate;
3687 
3688 	mvm = lq_sta->pers.drv;
3689 	memset(buf, 0, sizeof(buf));
3690 	buf_size = min(count, sizeof(buf) -  1);
3691 	if (copy_from_user(buf, user_buf, buf_size))
3692 		return -EFAULT;
3693 
3694 	if (sscanf(buf, "%x", &parsed_rate) == 1)
3695 		lq_sta->pers.dbg_fixed_rate = parsed_rate;
3696 	else
3697 		lq_sta->pers.dbg_fixed_rate = 0;
3698 
3699 	rs_program_fix_rate(mvm, lq_sta);
3700 
3701 	return count;
3702 }
3703 
3704 static ssize_t rs_sta_dbgfs_scale_table_read(struct file *file,
3705 			char __user *user_buf, size_t count, loff_t *ppos)
3706 {
3707 	char *buff;
3708 	int desc = 0;
3709 	int i = 0;
3710 	ssize_t ret;
3711 
3712 	struct iwl_lq_sta *lq_sta = file->private_data;
3713 	struct iwl_mvm_sta *mvmsta =
3714 		container_of(lq_sta, struct iwl_mvm_sta, lq_sta);
3715 	struct iwl_mvm *mvm;
3716 	struct iwl_scale_tbl_info *tbl = &(lq_sta->lq_info[lq_sta->active_tbl]);
3717 	struct rs_rate *rate = &tbl->rate;
3718 	u32 ss_params;
3719 
3720 	mvm = lq_sta->pers.drv;
3721 	buff = kmalloc(2048, GFP_KERNEL);
3722 	if (!buff)
3723 		return -ENOMEM;
3724 
3725 	desc += sprintf(buff+desc, "sta_id %d\n", lq_sta->lq.sta_id);
3726 	desc += sprintf(buff+desc, "failed=%d success=%d rate=0%lX\n",
3727 			lq_sta->total_failed, lq_sta->total_success,
3728 			lq_sta->active_legacy_rate);
3729 	desc += sprintf(buff+desc, "fixed rate 0x%X\n",
3730 			lq_sta->pers.dbg_fixed_rate);
3731 	desc += sprintf(buff+desc, "valid_tx_ant %s%s%s\n",
3732 	    (iwl_mvm_get_valid_tx_ant(mvm) & ANT_A) ? "ANT_A," : "",
3733 	    (iwl_mvm_get_valid_tx_ant(mvm) & ANT_B) ? "ANT_B," : "",
3734 	    (iwl_mvm_get_valid_tx_ant(mvm) & ANT_C) ? "ANT_C" : "");
3735 	desc += sprintf(buff+desc, "lq type %s\n",
3736 			(is_legacy(rate)) ? "legacy" :
3737 			is_vht(rate) ? "VHT" : "HT");
3738 	if (!is_legacy(rate)) {
3739 		desc += sprintf(buff + desc, " %s",
3740 		   (is_siso(rate)) ? "SISO" : "MIMO2");
3741 		desc += sprintf(buff + desc, " %s",
3742 				(is_ht20(rate)) ? "20MHz" :
3743 				(is_ht40(rate)) ? "40MHz" :
3744 				(is_ht80(rate)) ? "80MHz" :
3745 				(is_ht160(rate)) ? "160MHz" : "BAD BW");
3746 		desc += sprintf(buff + desc, " %s %s %s %s\n",
3747 				(rate->sgi) ? "SGI" : "NGI",
3748 				(rate->ldpc) ? "LDPC" : "BCC",
3749 				(lq_sta->is_agg) ? "AGG on" : "",
3750 				(mvmsta->tlc_amsdu) ? "AMSDU on" : "");
3751 	}
3752 	desc += sprintf(buff+desc, "last tx rate=0x%X\n",
3753 			lq_sta->last_rate_n_flags);
3754 	desc += sprintf(buff+desc,
3755 			"general: flags=0x%X mimo-d=%d s-ant=0x%x d-ant=0x%x\n",
3756 			lq_sta->lq.flags,
3757 			lq_sta->lq.mimo_delim,
3758 			lq_sta->lq.single_stream_ant_msk,
3759 			lq_sta->lq.dual_stream_ant_msk);
3760 
3761 	desc += sprintf(buff+desc,
3762 			"agg: time_limit=%d dist_start_th=%d frame_cnt_limit=%d\n",
3763 			le16_to_cpu(lq_sta->lq.agg_time_limit),
3764 			lq_sta->lq.agg_disable_start_th,
3765 			lq_sta->lq.agg_frame_cnt_limit);
3766 
3767 	desc += sprintf(buff+desc, "reduced tpc=%d\n", lq_sta->lq.reduced_tpc);
3768 	ss_params = le32_to_cpu(lq_sta->lq.ss_params);
3769 	desc += sprintf(buff+desc, "single stream params: %s%s%s%s\n",
3770 			(ss_params & LQ_SS_PARAMS_VALID) ?
3771 			"VALID" : "INVALID",
3772 			(ss_params & LQ_SS_BFER_ALLOWED) ?
3773 			", BFER" : "",
3774 			(ss_params & LQ_SS_STBC_1SS_ALLOWED) ?
3775 			", STBC" : "",
3776 			(ss_params & LQ_SS_FORCE) ?
3777 			", FORCE" : "");
3778 	desc += sprintf(buff+desc,
3779 			"Start idx [0]=0x%x [1]=0x%x [2]=0x%x [3]=0x%x\n",
3780 			lq_sta->lq.initial_rate_index[0],
3781 			lq_sta->lq.initial_rate_index[1],
3782 			lq_sta->lq.initial_rate_index[2],
3783 			lq_sta->lq.initial_rate_index[3]);
3784 
3785 	for (i = 0; i < LINK_QUAL_MAX_RETRY_NUM; i++) {
3786 		u32 r = le32_to_cpu(lq_sta->lq.rs_table[i]);
3787 
3788 		desc += sprintf(buff+desc, " rate[%d] 0x%X ", i, r);
3789 		desc += rs_pretty_print_rate(buff+desc, r);
3790 	}
3791 
3792 	ret = simple_read_from_buffer(user_buf, count, ppos, buff, desc);
3793 	kfree(buff);
3794 	return ret;
3795 }
3796 
3797 static const struct file_operations rs_sta_dbgfs_scale_table_ops = {
3798 	.write = rs_sta_dbgfs_scale_table_write,
3799 	.read = rs_sta_dbgfs_scale_table_read,
3800 	.open = simple_open,
3801 	.llseek = default_llseek,
3802 };
3803 static ssize_t rs_sta_dbgfs_stats_table_read(struct file *file,
3804 			char __user *user_buf, size_t count, loff_t *ppos)
3805 {
3806 	char *buff;
3807 	int desc = 0;
3808 	int i, j;
3809 	ssize_t ret;
3810 	struct iwl_scale_tbl_info *tbl;
3811 	struct rs_rate *rate;
3812 	struct iwl_lq_sta *lq_sta = file->private_data;
3813 
3814 	buff = kmalloc(1024, GFP_KERNEL);
3815 	if (!buff)
3816 		return -ENOMEM;
3817 
3818 	for (i = 0; i < LQ_SIZE; i++) {
3819 		tbl = &(lq_sta->lq_info[i]);
3820 		rate = &tbl->rate;
3821 		desc += sprintf(buff+desc,
3822 				"%s type=%d SGI=%d BW=%s DUP=0\n"
3823 				"index=%d\n",
3824 				lq_sta->active_tbl == i ? "*" : "x",
3825 				rate->type,
3826 				rate->sgi,
3827 				is_ht20(rate) ? "20MHz" :
3828 				is_ht40(rate) ? "40MHz" :
3829 				is_ht80(rate) ? "80MHz" :
3830 				is_ht160(rate) ? "160MHz" : "ERR",
3831 				rate->index);
3832 		for (j = 0; j < IWL_RATE_COUNT; j++) {
3833 			desc += sprintf(buff+desc,
3834 				"counter=%d success=%d %%=%d\n",
3835 				tbl->win[j].counter,
3836 				tbl->win[j].success_counter,
3837 				tbl->win[j].success_ratio);
3838 		}
3839 	}
3840 	ret = simple_read_from_buffer(user_buf, count, ppos, buff, desc);
3841 	kfree(buff);
3842 	return ret;
3843 }
3844 
3845 static const struct file_operations rs_sta_dbgfs_stats_table_ops = {
3846 	.read = rs_sta_dbgfs_stats_table_read,
3847 	.open = simple_open,
3848 	.llseek = default_llseek,
3849 };
3850 
3851 static ssize_t rs_sta_dbgfs_drv_tx_stats_read(struct file *file,
3852 					      char __user *user_buf,
3853 					      size_t count, loff_t *ppos)
3854 {
3855 	static const char * const column_name[] = {
3856 		[RS_COLUMN_LEGACY_ANT_A] = "LEGACY_ANT_A",
3857 		[RS_COLUMN_LEGACY_ANT_B] = "LEGACY_ANT_B",
3858 		[RS_COLUMN_SISO_ANT_A] = "SISO_ANT_A",
3859 		[RS_COLUMN_SISO_ANT_B] = "SISO_ANT_B",
3860 		[RS_COLUMN_SISO_ANT_A_SGI] = "SISO_ANT_A_SGI",
3861 		[RS_COLUMN_SISO_ANT_B_SGI] = "SISO_ANT_B_SGI",
3862 		[RS_COLUMN_MIMO2] = "MIMO2",
3863 		[RS_COLUMN_MIMO2_SGI] = "MIMO2_SGI",
3864 	};
3865 
3866 	static const char * const rate_name[] = {
3867 		[IWL_RATE_1M_INDEX] = "1M",
3868 		[IWL_RATE_2M_INDEX] = "2M",
3869 		[IWL_RATE_5M_INDEX] = "5.5M",
3870 		[IWL_RATE_11M_INDEX] = "11M",
3871 		[IWL_RATE_6M_INDEX] = "6M|MCS0",
3872 		[IWL_RATE_9M_INDEX] = "9M",
3873 		[IWL_RATE_12M_INDEX] = "12M|MCS1",
3874 		[IWL_RATE_18M_INDEX] = "18M|MCS2",
3875 		[IWL_RATE_24M_INDEX] = "24M|MCS3",
3876 		[IWL_RATE_36M_INDEX] = "36M|MCS4",
3877 		[IWL_RATE_48M_INDEX] = "48M|MCS5",
3878 		[IWL_RATE_54M_INDEX] = "54M|MCS6",
3879 		[IWL_RATE_MCS_7_INDEX] = "MCS7",
3880 		[IWL_RATE_MCS_8_INDEX] = "MCS8",
3881 		[IWL_RATE_MCS_9_INDEX] = "MCS9",
3882 	};
3883 
3884 	char *buff, *pos, *endpos;
3885 	int col, rate;
3886 	ssize_t ret;
3887 	struct iwl_lq_sta *lq_sta = file->private_data;
3888 	struct rs_rate_stats *stats;
3889 	static const size_t bufsz = 1024;
3890 
3891 	buff = kmalloc(bufsz, GFP_KERNEL);
3892 	if (!buff)
3893 		return -ENOMEM;
3894 
3895 	pos = buff;
3896 	endpos = pos + bufsz;
3897 
3898 	pos += scnprintf(pos, endpos - pos, "COLUMN,");
3899 	for (rate = 0; rate < IWL_RATE_COUNT; rate++)
3900 		pos += scnprintf(pos, endpos - pos, "%s,", rate_name[rate]);
3901 	pos += scnprintf(pos, endpos - pos, "\n");
3902 
3903 	for (col = 0; col < RS_COLUMN_COUNT; col++) {
3904 		pos += scnprintf(pos, endpos - pos,
3905 				 "%s,", column_name[col]);
3906 
3907 		for (rate = 0; rate < IWL_RATE_COUNT; rate++) {
3908 			stats = &(lq_sta->pers.tx_stats[col][rate]);
3909 			pos += scnprintf(pos, endpos - pos,
3910 					 "%llu/%llu,",
3911 					 stats->success,
3912 					 stats->total);
3913 		}
3914 		pos += scnprintf(pos, endpos - pos, "\n");
3915 	}
3916 
3917 	ret = simple_read_from_buffer(user_buf, count, ppos, buff, pos - buff);
3918 	kfree(buff);
3919 	return ret;
3920 }
3921 
3922 static ssize_t rs_sta_dbgfs_drv_tx_stats_write(struct file *file,
3923 					       const char __user *user_buf,
3924 					       size_t count, loff_t *ppos)
3925 {
3926 	struct iwl_lq_sta *lq_sta = file->private_data;
3927 	memset(lq_sta->pers.tx_stats, 0, sizeof(lq_sta->pers.tx_stats));
3928 
3929 	return count;
3930 }
3931 
3932 static const struct file_operations rs_sta_dbgfs_drv_tx_stats_ops = {
3933 	.read = rs_sta_dbgfs_drv_tx_stats_read,
3934 	.write = rs_sta_dbgfs_drv_tx_stats_write,
3935 	.open = simple_open,
3936 	.llseek = default_llseek,
3937 };
3938 
3939 static ssize_t iwl_dbgfs_ss_force_read(struct file *file,
3940 				       char __user *user_buf,
3941 				       size_t count, loff_t *ppos)
3942 {
3943 	struct iwl_lq_sta *lq_sta = file->private_data;
3944 	char buf[12];
3945 	int bufsz = sizeof(buf);
3946 	int pos = 0;
3947 	static const char * const ss_force_name[] = {
3948 		[RS_SS_FORCE_NONE] = "none",
3949 		[RS_SS_FORCE_STBC] = "stbc",
3950 		[RS_SS_FORCE_BFER] = "bfer",
3951 		[RS_SS_FORCE_SISO] = "siso",
3952 	};
3953 
3954 	pos += scnprintf(buf+pos, bufsz-pos, "%s\n",
3955 			 ss_force_name[lq_sta->pers.ss_force]);
3956 	return simple_read_from_buffer(user_buf, count, ppos, buf, pos);
3957 }
3958 
3959 static ssize_t iwl_dbgfs_ss_force_write(struct iwl_lq_sta *lq_sta, char *buf,
3960 					size_t count, loff_t *ppos)
3961 {
3962 	struct iwl_mvm *mvm = lq_sta->pers.drv;
3963 	int ret = 0;
3964 
3965 	if (!strncmp("none", buf, 4)) {
3966 		lq_sta->pers.ss_force = RS_SS_FORCE_NONE;
3967 	} else if (!strncmp("siso", buf, 4)) {
3968 		lq_sta->pers.ss_force = RS_SS_FORCE_SISO;
3969 	} else if (!strncmp("stbc", buf, 4)) {
3970 		if (lq_sta->stbc_capable) {
3971 			lq_sta->pers.ss_force = RS_SS_FORCE_STBC;
3972 		} else {
3973 			IWL_ERR(mvm,
3974 				"can't force STBC. peer doesn't support\n");
3975 			ret = -EINVAL;
3976 		}
3977 	} else if (!strncmp("bfer", buf, 4)) {
3978 		if (lq_sta->bfer_capable) {
3979 			lq_sta->pers.ss_force = RS_SS_FORCE_BFER;
3980 		} else {
3981 			IWL_ERR(mvm,
3982 				"can't force BFER. peer doesn't support\n");
3983 			ret = -EINVAL;
3984 		}
3985 	} else {
3986 		IWL_ERR(mvm, "valid values none|siso|stbc|bfer\n");
3987 		ret = -EINVAL;
3988 	}
3989 	return ret ?: count;
3990 }
3991 
3992 #define MVM_DEBUGFS_READ_WRITE_FILE_OPS(name, bufsz) \
3993 	_MVM_DEBUGFS_READ_WRITE_FILE_OPS(name, bufsz, struct iwl_lq_sta)
3994 #define MVM_DEBUGFS_ADD_FILE_RS(name, parent, mode) do {		\
3995 		if (!debugfs_create_file(#name, mode, parent, lq_sta,	\
3996 					 &iwl_dbgfs_##name##_ops))	\
3997 			goto err;					\
3998 	} while (0)
3999 
4000 MVM_DEBUGFS_READ_WRITE_FILE_OPS(ss_force, 32);
4001 
4002 static void rs_add_debugfs(void *mvm, void *priv_sta, struct dentry *dir)
4003 {
4004 	struct iwl_lq_sta *lq_sta = priv_sta;
4005 	struct iwl_mvm_sta *mvmsta;
4006 
4007 	mvmsta = container_of(lq_sta, struct iwl_mvm_sta, lq_sta);
4008 
4009 	if (!mvmsta->vif)
4010 		return;
4011 
4012 	debugfs_create_file("rate_scale_table", S_IRUSR | S_IWUSR, dir,
4013 			    lq_sta, &rs_sta_dbgfs_scale_table_ops);
4014 	debugfs_create_file("rate_stats_table", S_IRUSR, dir,
4015 			    lq_sta, &rs_sta_dbgfs_stats_table_ops);
4016 	debugfs_create_file("drv_tx_stats", S_IRUSR | S_IWUSR, dir,
4017 			    lq_sta, &rs_sta_dbgfs_drv_tx_stats_ops);
4018 	debugfs_create_u8("tx_agg_tid_enable", S_IRUSR | S_IWUSR, dir,
4019 			  &lq_sta->tx_agg_tid_en);
4020 	debugfs_create_u8("reduced_tpc", S_IRUSR | S_IWUSR, dir,
4021 			  &lq_sta->pers.dbg_fixed_txp_reduction);
4022 
4023 	MVM_DEBUGFS_ADD_FILE_RS(ss_force, dir, S_IRUSR | S_IWUSR);
4024 	return;
4025 err:
4026 	IWL_ERR((struct iwl_mvm *)mvm, "Can't create debugfs entity\n");
4027 }
4028 
4029 static void rs_remove_debugfs(void *mvm, void *mvm_sta)
4030 {
4031 }
4032 #endif
4033 
4034 /*
4035  * Initialization of rate scaling information is done by driver after
4036  * the station is added. Since mac80211 calls this function before a
4037  * station is added we ignore it.
4038  */
4039 static void rs_rate_init_stub(void *mvm_r,
4040 			      struct ieee80211_supported_band *sband,
4041 			      struct cfg80211_chan_def *chandef,
4042 			      struct ieee80211_sta *sta, void *mvm_sta)
4043 {
4044 }
4045 
4046 static const struct rate_control_ops rs_mvm_ops = {
4047 	.name = RS_NAME,
4048 	.tx_status = rs_mac80211_tx_status,
4049 	.get_rate = rs_get_rate,
4050 	.rate_init = rs_rate_init_stub,
4051 	.alloc = rs_alloc,
4052 	.free = rs_free,
4053 	.alloc_sta = rs_alloc_sta,
4054 	.free_sta = rs_free_sta,
4055 	.rate_update = rs_rate_update,
4056 #ifdef CONFIG_MAC80211_DEBUGFS
4057 	.add_sta_debugfs = rs_add_debugfs,
4058 	.remove_sta_debugfs = rs_remove_debugfs,
4059 #endif
4060 };
4061 
4062 int iwl_mvm_rate_control_register(void)
4063 {
4064 	return ieee80211_rate_control_register(&rs_mvm_ops);
4065 }
4066 
4067 void iwl_mvm_rate_control_unregister(void)
4068 {
4069 	ieee80211_rate_control_unregister(&rs_mvm_ops);
4070 }
4071 
4072 /**
4073  * iwl_mvm_tx_protection - Gets LQ command, change it to enable/disable
4074  * Tx protection, according to this request and previous requests,
4075  * and send the LQ command.
4076  * @mvmsta: The station
4077  * @enable: Enable Tx protection?
4078  */
4079 int iwl_mvm_tx_protection(struct iwl_mvm *mvm, struct iwl_mvm_sta *mvmsta,
4080 			  bool enable)
4081 {
4082 	struct iwl_lq_cmd *lq = &mvmsta->lq_sta.lq;
4083 
4084 	lockdep_assert_held(&mvm->mutex);
4085 
4086 	if (enable) {
4087 		if (mvmsta->tx_protection == 0)
4088 			lq->flags |= LQ_FLAG_USE_RTS_MSK;
4089 		mvmsta->tx_protection++;
4090 	} else {
4091 		mvmsta->tx_protection--;
4092 		if (mvmsta->tx_protection == 0)
4093 			lq->flags &= ~LQ_FLAG_USE_RTS_MSK;
4094 	}
4095 
4096 	return iwl_mvm_send_lq_cmd(mvm, lq, false);
4097 }
4098