OSDN Git Service

ath9k/ath9k_htc: remove redundand checks for dual-stream tx support
[uclinux-h8/linux.git] / drivers / net / wireless / ath / ath9k / rc.c
1 /*
2  * Copyright (c) 2004 Video54 Technologies, Inc.
3  * Copyright (c) 2004-2009 Atheros Communications, Inc.
4  *
5  * Permission to use, copy, modify, and/or distribute this software for any
6  * purpose with or without fee is hereby granted, provided that the above
7  * copyright notice and this permission notice appear in all copies.
8  *
9  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16  */
17
18 #include <linux/slab.h>
19
20 #include "ath9k.h"
21
22 static const struct ath_rate_table ar5416_11na_ratetable = {
23         43,
24         8, /* MCS start */
25         {
26                 { VALID, VALID, WLAN_RC_PHY_OFDM, 6000, /* 6 Mb */
27                         5400, 0, 12, 0, 0, 0, 0, 0 },
28                 { VALID, VALID, WLAN_RC_PHY_OFDM, 9000, /* 9 Mb */
29                         7800,  1, 18, 0, 1, 1, 1, 1 },
30                 { VALID, VALID, WLAN_RC_PHY_OFDM, 12000, /* 12 Mb */
31                         10000, 2, 24, 2, 2, 2, 2, 2 },
32                 { VALID, VALID, WLAN_RC_PHY_OFDM, 18000, /* 18 Mb */
33                         13900, 3, 36, 2, 3, 3, 3, 3 },
34                 { VALID, VALID, WLAN_RC_PHY_OFDM, 24000, /* 24 Mb */
35                         17300, 4, 48, 4, 4, 4, 4, 4 },
36                 { VALID, VALID, WLAN_RC_PHY_OFDM, 36000, /* 36 Mb */
37                         23000, 5, 72, 4, 5, 5, 5, 5 },
38                 { VALID, VALID, WLAN_RC_PHY_OFDM, 48000, /* 48 Mb */
39                         27400, 6, 96, 4, 6, 6, 6, 6 },
40                 { VALID, VALID, WLAN_RC_PHY_OFDM, 54000, /* 54 Mb */
41                         29300, 7, 108, 4, 7, 7, 7, 7 },
42                 { VALID_2040, VALID_2040, WLAN_RC_PHY_HT_20_SS, 6500, /* 6.5 Mb */
43                         6400, 0, 0, 0, 8, 25, 8, 25 },
44                 { VALID_20, VALID_20, WLAN_RC_PHY_HT_20_SS, 13000, /* 13 Mb */
45                         12700, 1, 1, 2, 9, 26, 9, 26 },
46                 { VALID_20, VALID_20, WLAN_RC_PHY_HT_20_SS, 19500, /* 19.5 Mb */
47                         18800, 2, 2, 2, 10, 27, 10, 27 },
48                 { VALID_20, VALID_20, WLAN_RC_PHY_HT_20_SS, 26000, /* 26 Mb */
49                         25000, 3, 3, 4, 11, 28, 11, 28 },
50                 { VALID_20, VALID_20, WLAN_RC_PHY_HT_20_SS, 39000, /* 39 Mb */
51                         36700, 4, 4, 4, 12, 29, 12, 29 },
52                 { INVALID, VALID_20, WLAN_RC_PHY_HT_20_SS, 52000, /* 52 Mb */
53                         48100, 5, 5, 4, 13, 30, 13, 30 },
54                 { INVALID, VALID_20, WLAN_RC_PHY_HT_20_SS, 58500, /* 58.5 Mb */
55                         53500, 6, 6, 4, 14, 31, 14, 31 },
56                 { INVALID, VALID_20, WLAN_RC_PHY_HT_20_SS, 65000, /* 65 Mb */
57                         59000, 7, 7, 4, 15, 32, 15, 33 },
58                 { INVALID, INVALID, WLAN_RC_PHY_HT_20_DS, 13000, /* 13 Mb */
59                         12700, 8, 8, 3, 16, 34, 16, 34 },
60                 { INVALID, INVALID, WLAN_RC_PHY_HT_20_DS, 26000, /* 26 Mb */
61                         24800, 9, 9, 2, 17, 35, 17, 35 },
62                 { INVALID, INVALID, WLAN_RC_PHY_HT_20_DS, 39000, /* 39 Mb */
63                         36600, 10, 10, 2, 18, 36, 18, 36 },
64                 { VALID_20, INVALID, WLAN_RC_PHY_HT_20_DS, 52000, /* 52 Mb */
65                         48100, 11, 11, 4, 19, 37, 19, 37 },
66                 { VALID_20, INVALID, WLAN_RC_PHY_HT_20_DS, 78000, /* 78 Mb */
67                         69500, 12, 12, 4, 20, 38, 20, 38 },
68                 { VALID_20, INVALID, WLAN_RC_PHY_HT_20_DS, 104000, /* 104 Mb */
69                         89500, 13, 13, 4, 21, 39, 21, 39 },
70                 { VALID_20, INVALID, WLAN_RC_PHY_HT_20_DS, 117000, /* 117 Mb */
71                         98900, 14, 14, 4, 22, 40, 22, 40 },
72                 { VALID_20, INVALID, WLAN_RC_PHY_HT_20_DS, 130000, /* 130 Mb */
73                         108300, 15, 15, 4, 23, 41, 24, 42 },
74                 { VALID_20, INVALID, WLAN_RC_PHY_HT_20_DS_HGI, 144400, /* 144.4 Mb */
75                         12000, 15, 15, 4, 23, 41, 24, 42 },
76                 { VALID_40, VALID_40, WLAN_RC_PHY_HT_40_SS, 13500, /* 13.5 Mb */
77                         13200, 0, 0, 0, 8, 25, 25, 25 },
78                 { VALID_40, VALID_40, WLAN_RC_PHY_HT_40_SS, 27500, /* 27.0 Mb */
79                         25900, 1, 1, 2, 9, 26, 26, 26 },
80                 { VALID_40, VALID_40, WLAN_RC_PHY_HT_40_SS, 40500, /* 40.5 Mb */
81                         38600, 2, 2, 2, 10, 27, 27, 27 },
82                 { VALID_40, VALID_40, WLAN_RC_PHY_HT_40_SS, 54000, /* 54 Mb */
83                         49800, 3, 3, 4, 11, 28, 28, 28 },
84                 { VALID_40, VALID_40, WLAN_RC_PHY_HT_40_SS, 81500, /* 81 Mb */
85                         72200, 4, 4, 4, 12, 29, 29, 29 },
86                 { INVALID, VALID_40, WLAN_RC_PHY_HT_40_SS, 108000, /* 108 Mb */
87                         92900, 5, 5, 4, 13, 30, 30, 30 },
88                 { INVALID, VALID_40, WLAN_RC_PHY_HT_40_SS, 121500, /* 121.5 Mb */
89                         102700, 6, 6, 4, 14, 31, 31, 31 },
90                 { INVALID, VALID_40, WLAN_RC_PHY_HT_40_SS, 135000, /* 135 Mb */
91                         112000, 7, 7, 4, 15, 32, 33, 33 },
92                 { INVALID, VALID_40, WLAN_RC_PHY_HT_40_SS_HGI, 150000, /* 150 Mb */
93                         122000, 7, 7, 4, 15, 32, 33, 33 },
94                 { INVALID, INVALID, WLAN_RC_PHY_HT_40_DS, 27000, /* 27 Mb */
95                         25800, 8, 8, 0, 16, 34, 34, 34 },
96                 { INVALID, INVALID, WLAN_RC_PHY_HT_40_DS, 54000, /* 54 Mb */
97                         49800, 9, 9, 2, 17, 35, 35, 35 },
98                 { INVALID, INVALID, WLAN_RC_PHY_HT_40_DS, 81000, /* 81 Mb */
99                         71900, 10, 10, 2, 18, 36, 36, 36 },
100                 { VALID_40, INVALID, WLAN_RC_PHY_HT_40_DS, 108000, /* 108 Mb */
101                         92500, 11, 11, 4, 19, 37, 37, 37 },
102                 { VALID_40, INVALID, WLAN_RC_PHY_HT_40_DS, 162000, /* 162 Mb */
103                         130300, 12, 12, 4, 20, 38, 38, 38 },
104                 { VALID_40, INVALID, WLAN_RC_PHY_HT_40_DS, 216000, /* 216 Mb */
105                         162800, 13, 13, 4, 21, 39, 39, 39 },
106                 { VALID_40, INVALID, WLAN_RC_PHY_HT_40_DS, 243000, /* 243 Mb */
107                         178200, 14, 14, 4, 22, 40, 40, 40 },
108                 { VALID_40, INVALID, WLAN_RC_PHY_HT_40_DS, 270000, /* 270 Mb */
109                         192100, 15, 15, 4, 23, 41, 42, 42 },
110                 { VALID_40, INVALID, WLAN_RC_PHY_HT_40_DS_HGI, 300000, /* 300 Mb */
111                         207000, 15, 15, 4, 23, 41, 42, 42 },
112         },
113         50,  /* probe interval */
114         WLAN_RC_HT_FLAG,  /* Phy rates allowed initially */
115 };
116
117 /* 4ms frame limit not used for NG mode.  The values filled
118  * for HT are the 64K max aggregate limit */
119
120 static const struct ath_rate_table ar5416_11ng_ratetable = {
121         47,
122         12, /* MCS start */
123         {
124                 { VALID_ALL, VALID_ALL, WLAN_RC_PHY_CCK, 1000, /* 1 Mb */
125                         900, 0, 2, 0, 0, 0, 0, 0 },
126                 { VALID_ALL, VALID_ALL, WLAN_RC_PHY_CCK, 2000, /* 2 Mb */
127                         1900, 1, 4, 1, 1, 1, 1, 1 },
128                 { VALID_ALL, VALID_ALL, WLAN_RC_PHY_CCK, 5500, /* 5.5 Mb */
129                         4900, 2, 11, 2, 2, 2, 2, 2 },
130                 { VALID_ALL, VALID_ALL, WLAN_RC_PHY_CCK, 11000, /* 11 Mb */
131                         8100, 3, 22, 3, 3, 3, 3, 3 },
132                 { INVALID, INVALID, WLAN_RC_PHY_OFDM, 6000, /* 6 Mb */
133                         5400, 4, 12, 4, 4, 4, 4, 4 },
134                 { INVALID, INVALID, WLAN_RC_PHY_OFDM, 9000, /* 9 Mb */
135                         7800, 5, 18, 4, 5, 5, 5, 5 },
136                 { VALID, VALID, WLAN_RC_PHY_OFDM, 12000, /* 12 Mb */
137                         10100, 6, 24, 6, 6, 6, 6, 6 },
138                 { VALID, VALID, WLAN_RC_PHY_OFDM, 18000, /* 18 Mb */
139                         14100, 7, 36, 6, 7, 7, 7, 7 },
140                 { VALID, VALID, WLAN_RC_PHY_OFDM, 24000, /* 24 Mb */
141                         17700, 8, 48, 8, 8, 8, 8, 8 },
142                 { VALID, VALID, WLAN_RC_PHY_OFDM, 36000, /* 36 Mb */
143                         23700, 9, 72, 8, 9, 9, 9, 9 },
144                 { VALID, VALID, WLAN_RC_PHY_OFDM, 48000, /* 48 Mb */
145                         27400, 10, 96, 8, 10, 10, 10, 10 },
146                 { VALID, VALID, WLAN_RC_PHY_OFDM, 54000, /* 54 Mb */
147                         30900, 11, 108, 8, 11, 11, 11, 11 },
148                 { INVALID, INVALID, WLAN_RC_PHY_HT_20_SS, 6500, /* 6.5 Mb */
149                         6400, 0, 0, 4, 12, 29, 12, 29 },
150                 { VALID_20, VALID_20, WLAN_RC_PHY_HT_20_SS, 13000, /* 13 Mb */
151                         12700, 1, 1, 6, 13, 30, 13, 30 },
152                 { VALID_20, VALID_20, WLAN_RC_PHY_HT_20_SS, 19500, /* 19.5 Mb */
153                         18800, 2, 2, 6, 14, 31, 14, 31 },
154                 { VALID_20, VALID_20, WLAN_RC_PHY_HT_20_SS, 26000, /* 26 Mb */
155                         25000, 3, 3, 8, 15, 32, 15, 32 },
156                 { VALID_20, VALID_20, WLAN_RC_PHY_HT_20_SS, 39000, /* 39 Mb */
157                         36700, 4, 4, 8, 16, 33, 16, 33 },
158                 { INVALID, VALID_20, WLAN_RC_PHY_HT_20_SS, 52000, /* 52 Mb */
159                         48100, 5, 5, 8, 17, 34, 17, 34 },
160                 { INVALID,  VALID_20, WLAN_RC_PHY_HT_20_SS, 58500, /* 58.5 Mb */
161                         53500, 6, 6, 8, 18, 35, 18, 35 },
162                 { INVALID, VALID_20, WLAN_RC_PHY_HT_20_SS, 65000, /* 65 Mb */
163                         59000, 7, 7, 8, 19, 36, 19, 37 },
164                 { INVALID, INVALID, WLAN_RC_PHY_HT_20_DS, 13000, /* 13 Mb */
165                         12700, 8, 8, 4, 20, 38, 20, 38 },
166                 { INVALID, INVALID, WLAN_RC_PHY_HT_20_DS, 26000, /* 26 Mb */
167                         24800, 9, 9, 6, 21, 39, 21, 39 },
168                 { INVALID, INVALID, WLAN_RC_PHY_HT_20_DS, 39000, /* 39 Mb */
169                         36600, 10, 10, 6, 22, 40, 22, 40 },
170                 { VALID_20, INVALID, WLAN_RC_PHY_HT_20_DS, 52000, /* 52 Mb */
171                         48100, 11, 11, 8, 23, 41, 23, 41 },
172                 { VALID_20, INVALID, WLAN_RC_PHY_HT_20_DS, 78000, /* 78 Mb */
173                         69500, 12, 12, 8, 24, 42, 24, 42 },
174                 { VALID_20, INVALID, WLAN_RC_PHY_HT_20_DS, 104000, /* 104 Mb */
175                         89500, 13, 13, 8, 25, 43, 25, 43 },
176                 { VALID_20, INVALID, WLAN_RC_PHY_HT_20_DS, 117000, /* 117 Mb */
177                         98900, 14, 14, 8, 26, 44, 26, 44 },
178                 { VALID_20, INVALID, WLAN_RC_PHY_HT_20_DS, 130000, /* 130 Mb */
179                         108300, 15, 15, 8, 27, 45, 28, 46 },
180                 { VALID_20, INVALID, WLAN_RC_PHY_HT_20_DS_HGI, 144400, /* 130 Mb */
181                         120000, 15, 15, 8, 27, 45, 28, 46 },
182                 { VALID_40, VALID_40, WLAN_RC_PHY_HT_40_SS, 13500, /* 13.5 Mb */
183                         13200, 0, 0, 8, 12, 29, 29, 29 },
184                 { VALID_40, VALID_40, WLAN_RC_PHY_HT_40_SS, 27500, /* 27.0 Mb */
185                         25900, 1, 1, 8, 13, 30, 30, 30 },
186                 { VALID_40, VALID_40, WLAN_RC_PHY_HT_40_SS, 40500, /* 40.5 Mb */
187                         38600, 2, 2, 8, 14, 31, 31, 31 },
188                 { VALID_40, VALID_40, WLAN_RC_PHY_HT_40_SS, 54000, /* 54 Mb */
189                         49800, 3, 3, 8,  15, 32, 32, 32 },
190                 { VALID_40, VALID_40, WLAN_RC_PHY_HT_40_SS, 81500, /* 81 Mb */
191                         72200, 4, 4, 8, 16, 33, 33, 33 },
192                 { INVALID, VALID_40, WLAN_RC_PHY_HT_40_SS, 108000, /* 108 Mb */
193                         92900, 5, 5, 8, 17, 34, 34, 34 },
194                 { INVALID,  VALID_40, WLAN_RC_PHY_HT_40_SS, 121500, /* 121.5 Mb */
195                         102700, 6, 6, 8, 18, 35, 35, 35 },
196                 { INVALID, VALID_40, WLAN_RC_PHY_HT_40_SS, 135000, /* 135 Mb */
197                         112000, 7, 7, 8, 19, 36, 37, 37 },
198                 { INVALID, VALID_40, WLAN_RC_PHY_HT_40_SS_HGI, 150000, /* 150 Mb */
199                         122000, 7, 7, 8, 19, 36, 37, 37 },
200                 { INVALID, INVALID, WLAN_RC_PHY_HT_40_DS, 27000, /* 27 Mb */
201                         25800, 8, 8, 8, 20, 38, 38, 38 },
202                 { INVALID, INVALID, WLAN_RC_PHY_HT_40_DS, 54000, /* 54 Mb */
203                         49800, 9, 9, 8, 21, 39, 39, 39 },
204                 { INVALID, INVALID, WLAN_RC_PHY_HT_40_DS, 81000, /* 81 Mb */
205                         71900, 10, 10, 8, 22, 40, 40, 40 },
206                 { VALID_40, INVALID, WLAN_RC_PHY_HT_40_DS, 108000, /* 108 Mb */
207                         92500, 11, 11, 8, 23, 41, 41, 41 },
208                 { VALID_40, INVALID, WLAN_RC_PHY_HT_40_DS, 162000, /* 162 Mb */
209                         130300, 12, 12, 8, 24, 42, 42, 42 },
210                 { VALID_40, INVALID, WLAN_RC_PHY_HT_40_DS, 216000, /* 216 Mb */
211                         162800, 13, 13, 8, 25, 43, 43, 43 },
212                 { VALID_40, INVALID, WLAN_RC_PHY_HT_40_DS, 243000, /* 243 Mb */
213                         178200, 14, 14, 8, 26, 44, 44, 44 },
214                 { VALID_40, INVALID, WLAN_RC_PHY_HT_40_DS, 270000, /* 270 Mb */
215                         192100, 15, 15, 8, 27, 45, 46, 46 },
216                 { VALID_40, INVALID, WLAN_RC_PHY_HT_40_DS_HGI, 300000, /* 300 Mb */
217                         207000, 15, 15, 8, 27, 45, 46, 46 },
218         },
219         50,  /* probe interval */
220         WLAN_RC_HT_FLAG,  /* Phy rates allowed initially */
221 };
222
223 static const struct ath_rate_table ar5416_11a_ratetable = {
224         8,
225         0,
226         {
227                 { VALID, VALID, WLAN_RC_PHY_OFDM, 6000, /* 6 Mb */
228                         5400, 0, 12, 0, 0, 0 },
229                 { VALID, VALID, WLAN_RC_PHY_OFDM, 9000, /* 9 Mb */
230                         7800,  1, 18, 0, 1, 0 },
231                 { VALID, VALID, WLAN_RC_PHY_OFDM, 12000, /* 12 Mb */
232                         10000, 2, 24, 2, 2, 0 },
233                 { VALID, VALID, WLAN_RC_PHY_OFDM, 18000, /* 18 Mb */
234                         13900, 3, 36, 2, 3, 0 },
235                 { VALID, VALID, WLAN_RC_PHY_OFDM, 24000, /* 24 Mb */
236                         17300, 4, 48, 4, 4, 0 },
237                 { VALID, VALID, WLAN_RC_PHY_OFDM, 36000, /* 36 Mb */
238                         23000, 5, 72, 4, 5, 0 },
239                 { VALID, VALID, WLAN_RC_PHY_OFDM, 48000, /* 48 Mb */
240                         27400, 6, 96, 4, 6, 0 },
241                 { VALID, VALID, WLAN_RC_PHY_OFDM, 54000, /* 54 Mb */
242                         29300, 7, 108, 4, 7, 0 },
243         },
244         50,  /* probe interval */
245         0,   /* Phy rates allowed initially */
246 };
247
248 static const struct ath_rate_table ar5416_11g_ratetable = {
249         12,
250         0,
251         {
252                 { VALID, VALID, WLAN_RC_PHY_CCK, 1000, /* 1 Mb */
253                         900, 0, 2, 0, 0, 0 },
254                 { VALID, VALID, WLAN_RC_PHY_CCK, 2000, /* 2 Mb */
255                         1900, 1, 4, 1, 1, 0 },
256                 { VALID, VALID, WLAN_RC_PHY_CCK, 5500, /* 5.5 Mb */
257                         4900, 2, 11, 2, 2, 0 },
258                 { VALID, VALID, WLAN_RC_PHY_CCK, 11000, /* 11 Mb */
259                         8100, 3, 22, 3, 3, 0 },
260                 { INVALID, INVALID, WLAN_RC_PHY_OFDM, 6000, /* 6 Mb */
261                         5400, 4, 12, 4, 4, 0 },
262                 { INVALID, INVALID, WLAN_RC_PHY_OFDM, 9000, /* 9 Mb */
263                         7800, 5, 18, 4, 5, 0 },
264                 { VALID, VALID, WLAN_RC_PHY_OFDM, 12000, /* 12 Mb */
265                         10000, 6, 24, 6, 6, 0 },
266                 { VALID, VALID, WLAN_RC_PHY_OFDM, 18000, /* 18 Mb */
267                         13900, 7, 36, 6, 7, 0 },
268                 { VALID, VALID, WLAN_RC_PHY_OFDM, 24000, /* 24 Mb */
269                         17300, 8, 48, 8, 8, 0 },
270                 { VALID, VALID, WLAN_RC_PHY_OFDM, 36000, /* 36 Mb */
271                         23000, 9, 72, 8, 9, 0 },
272                 { VALID, VALID, WLAN_RC_PHY_OFDM, 48000, /* 48 Mb */
273                         27400, 10, 96, 8, 10, 0 },
274                 { VALID, VALID, WLAN_RC_PHY_OFDM, 54000, /* 54 Mb */
275                         29300, 11, 108, 8, 11, 0 },
276         },
277         50,  /* probe interval */
278         0,   /* Phy rates allowed initially */
279 };
280
281 static const struct ath_rate_table *hw_rate_table[ATH9K_MODE_MAX] = {
282         [ATH9K_MODE_11A] = &ar5416_11a_ratetable,
283         [ATH9K_MODE_11G] = &ar5416_11g_ratetable,
284         [ATH9K_MODE_11NA_HT20] = &ar5416_11na_ratetable,
285         [ATH9K_MODE_11NG_HT20] = &ar5416_11ng_ratetable,
286         [ATH9K_MODE_11NA_HT40PLUS] = &ar5416_11na_ratetable,
287         [ATH9K_MODE_11NA_HT40MINUS] = &ar5416_11na_ratetable,
288         [ATH9K_MODE_11NG_HT40PLUS] = &ar5416_11ng_ratetable,
289         [ATH9K_MODE_11NG_HT40MINUS] = &ar5416_11ng_ratetable,
290 };
291
292 static int ath_rc_get_rateindex(const struct ath_rate_table *rate_table,
293                                 struct ieee80211_tx_rate *rate);
294
295 static inline int8_t median(int8_t a, int8_t b, int8_t c)
296 {
297         if (a >= b) {
298                 if (b >= c)
299                         return b;
300                 else if (a > c)
301                         return c;
302                 else
303                         return a;
304         } else {
305                 if (a >= c)
306                         return a;
307                 else if (b >= c)
308                         return c;
309                 else
310                         return b;
311         }
312 }
313
314 static void ath_rc_sort_validrates(const struct ath_rate_table *rate_table,
315                                    struct ath_rate_priv *ath_rc_priv)
316 {
317         u8 i, j, idx, idx_next;
318
319         for (i = ath_rc_priv->max_valid_rate - 1; i > 0; i--) {
320                 for (j = 0; j <= i-1; j++) {
321                         idx = ath_rc_priv->valid_rate_index[j];
322                         idx_next = ath_rc_priv->valid_rate_index[j+1];
323
324                         if (rate_table->info[idx].ratekbps >
325                                 rate_table->info[idx_next].ratekbps) {
326                                 ath_rc_priv->valid_rate_index[j] = idx_next;
327                                 ath_rc_priv->valid_rate_index[j+1] = idx;
328                         }
329                 }
330         }
331 }
332
333 static void ath_rc_init_valid_txmask(struct ath_rate_priv *ath_rc_priv)
334 {
335         u8 i;
336
337         for (i = 0; i < ath_rc_priv->rate_table_size; i++)
338                 ath_rc_priv->valid_rate_index[i] = 0;
339 }
340
341 static inline void ath_rc_set_valid_txmask(struct ath_rate_priv *ath_rc_priv,
342                                            u8 index, int valid_tx_rate)
343 {
344         BUG_ON(index > ath_rc_priv->rate_table_size);
345         ath_rc_priv->valid_rate_index[index] = valid_tx_rate ? 1 : 0;
346 }
347
348 static inline
349 int ath_rc_get_nextvalid_txrate(const struct ath_rate_table *rate_table,
350                                 struct ath_rate_priv *ath_rc_priv,
351                                 u8 cur_valid_txrate,
352                                 u8 *next_idx)
353 {
354         u8 i;
355
356         for (i = 0; i < ath_rc_priv->max_valid_rate - 1; i++) {
357                 if (ath_rc_priv->valid_rate_index[i] == cur_valid_txrate) {
358                         *next_idx = ath_rc_priv->valid_rate_index[i+1];
359                         return 1;
360                 }
361         }
362
363         /* No more valid rates */
364         *next_idx = 0;
365
366         return 0;
367 }
368
369 /* Return true only for single stream */
370
371 static int ath_rc_valid_phyrate(u32 phy, u32 capflag, int ignore_cw)
372 {
373         if (WLAN_RC_PHY_HT(phy) && !(capflag & WLAN_RC_HT_FLAG))
374                 return 0;
375         if (WLAN_RC_PHY_DS(phy) && !(capflag & WLAN_RC_DS_FLAG))
376                 return 0;
377         if (WLAN_RC_PHY_SGI(phy) && !(capflag & WLAN_RC_SGI_FLAG))
378                 return 0;
379         if (!ignore_cw && WLAN_RC_PHY_HT(phy))
380                 if (WLAN_RC_PHY_40(phy) && !(capflag & WLAN_RC_40_FLAG))
381                         return 0;
382         return 1;
383 }
384
385 static inline int
386 ath_rc_get_lower_rix(const struct ath_rate_table *rate_table,
387                      struct ath_rate_priv *ath_rc_priv,
388                      u8 cur_valid_txrate, u8 *next_idx)
389 {
390         int8_t i;
391
392         for (i = 1; i < ath_rc_priv->max_valid_rate ; i++) {
393                 if (ath_rc_priv->valid_rate_index[i] == cur_valid_txrate) {
394                         *next_idx = ath_rc_priv->valid_rate_index[i-1];
395                         return 1;
396                 }
397         }
398
399         return 0;
400 }
401
402 static u8 ath_rc_init_validrates(struct ath_rate_priv *ath_rc_priv,
403                                  const struct ath_rate_table *rate_table,
404                                  u32 capflag)
405 {
406         u8 i, hi = 0;
407         u32 valid;
408
409         for (i = 0; i < rate_table->rate_cnt; i++) {
410                 valid = (!(ath_rc_priv->ht_cap & WLAN_RC_DS_FLAG) ?
411                          rate_table->info[i].valid_single_stream :
412                          rate_table->info[i].valid);
413                 if (valid == 1) {
414                         u32 phy = rate_table->info[i].phy;
415                         u8 valid_rate_count = 0;
416
417                         if (!ath_rc_valid_phyrate(phy, capflag, 0))
418                                 continue;
419
420                         valid_rate_count = ath_rc_priv->valid_phy_ratecnt[phy];
421
422                         ath_rc_priv->valid_phy_rateidx[phy][valid_rate_count] = i;
423                         ath_rc_priv->valid_phy_ratecnt[phy] += 1;
424                         ath_rc_set_valid_txmask(ath_rc_priv, i, 1);
425                         hi = A_MAX(hi, i);
426                 }
427         }
428
429         return hi;
430 }
431
432 static u8 ath_rc_setvalid_rates(struct ath_rate_priv *ath_rc_priv,
433                                 const struct ath_rate_table *rate_table,
434                                 struct ath_rateset *rateset,
435                                 u32 capflag)
436 {
437         u8 i, j, hi = 0;
438
439         /* Use intersection of working rates and valid rates */
440         for (i = 0; i < rateset->rs_nrates; i++) {
441                 for (j = 0; j < rate_table->rate_cnt; j++) {
442                         u32 phy = rate_table->info[j].phy;
443                         u32 valid = (!(ath_rc_priv->ht_cap & WLAN_RC_DS_FLAG) ?
444                                      rate_table->info[j].valid_single_stream :
445                                      rate_table->info[j].valid);
446                         u8 rate = rateset->rs_rates[i];
447                         u8 dot11rate = rate_table->info[j].dot11rate;
448
449                         /* We allow a rate only if its valid and the
450                          * capflag matches one of the validity
451                          * (VALID/VALID_20/VALID_40) flags */
452
453                         if ((rate == dot11rate) &&
454                             ((valid & WLAN_RC_CAP_MODE(capflag)) ==
455                              WLAN_RC_CAP_MODE(capflag)) &&
456                             !WLAN_RC_PHY_HT(phy)) {
457                                 u8 valid_rate_count = 0;
458
459                                 if (!ath_rc_valid_phyrate(phy, capflag, 0))
460                                         continue;
461
462                                 valid_rate_count =
463                                         ath_rc_priv->valid_phy_ratecnt[phy];
464
465                                 ath_rc_priv->valid_phy_rateidx[phy]
466                                         [valid_rate_count] = j;
467                                 ath_rc_priv->valid_phy_ratecnt[phy] += 1;
468                                 ath_rc_set_valid_txmask(ath_rc_priv, j, 1);
469                                 hi = A_MAX(hi, j);
470                         }
471                 }
472         }
473
474         return hi;
475 }
476
477 static u8 ath_rc_setvalid_htrates(struct ath_rate_priv *ath_rc_priv,
478                                   const struct ath_rate_table *rate_table,
479                                   u8 *mcs_set, u32 capflag)
480 {
481         struct ath_rateset *rateset = (struct ath_rateset *)mcs_set;
482
483         u8 i, j, hi = 0;
484
485         /* Use intersection of working rates and valid rates */
486         for (i = 0; i < rateset->rs_nrates; i++) {
487                 for (j = 0; j < rate_table->rate_cnt; j++) {
488                         u32 phy = rate_table->info[j].phy;
489                         u32 valid = (!(ath_rc_priv->ht_cap & WLAN_RC_DS_FLAG) ?
490                                      rate_table->info[j].valid_single_stream :
491                                      rate_table->info[j].valid);
492                         u8 rate = rateset->rs_rates[i];
493                         u8 dot11rate = rate_table->info[j].dot11rate;
494
495                         if ((rate != dot11rate) || !WLAN_RC_PHY_HT(phy) ||
496                             !WLAN_RC_PHY_HT_VALID(valid, capflag))
497                                 continue;
498
499                         if (!ath_rc_valid_phyrate(phy, capflag, 0))
500                                 continue;
501
502                         ath_rc_priv->valid_phy_rateidx[phy]
503                                 [ath_rc_priv->valid_phy_ratecnt[phy]] = j;
504                         ath_rc_priv->valid_phy_ratecnt[phy] += 1;
505                         ath_rc_set_valid_txmask(ath_rc_priv, j, 1);
506                         hi = A_MAX(hi, j);
507                 }
508         }
509
510         return hi;
511 }
512
513 /* Finds the highest rate index we can use */
514 static u8 ath_rc_get_highest_rix(struct ath_softc *sc,
515                                  struct ath_rate_priv *ath_rc_priv,
516                                  const struct ath_rate_table *rate_table,
517                                  int *is_probing)
518 {
519         u32 best_thruput, this_thruput, now_msec;
520         u8 rate, next_rate, best_rate, maxindex, minindex;
521         int8_t index = 0;
522
523         now_msec = jiffies_to_msecs(jiffies);
524         *is_probing = 0;
525         best_thruput = 0;
526         maxindex = ath_rc_priv->max_valid_rate-1;
527         minindex = 0;
528         best_rate = minindex;
529
530         /*
531          * Try the higher rate first. It will reduce memory moving time
532          * if we have very good channel characteristics.
533          */
534         for (index = maxindex; index >= minindex ; index--) {
535                 u8 per_thres;
536
537                 rate = ath_rc_priv->valid_rate_index[index];
538                 if (rate > ath_rc_priv->rate_max_phy)
539                         continue;
540
541                 /*
542                  * For TCP the average collision rate is around 11%,
543                  * so we ignore PERs less than this.  This is to
544                  * prevent the rate we are currently using (whose
545                  * PER might be in the 10-15 range because of TCP
546                  * collisions) looking worse than the next lower
547                  * rate whose PER has decayed close to 0.  If we
548                  * used to next lower rate, its PER would grow to
549                  * 10-15 and we would be worse off then staying
550                  * at the current rate.
551                  */
552                 per_thres = ath_rc_priv->per[rate];
553                 if (per_thres < 12)
554                         per_thres = 12;
555
556                 this_thruput = rate_table->info[rate].user_ratekbps *
557                         (100 - per_thres);
558
559                 if (best_thruput <= this_thruput) {
560                         best_thruput = this_thruput;
561                         best_rate    = rate;
562                 }
563         }
564
565         rate = best_rate;
566
567         /*
568          * Must check the actual rate (ratekbps) to account for
569          * non-monoticity of 11g's rate table
570          */
571
572         if (rate >= ath_rc_priv->rate_max_phy) {
573                 rate = ath_rc_priv->rate_max_phy;
574
575                 /* Probe the next allowed phy state */
576                 if (ath_rc_get_nextvalid_txrate(rate_table,
577                                         ath_rc_priv, rate, &next_rate) &&
578                     (now_msec - ath_rc_priv->probe_time >
579                      rate_table->probe_interval) &&
580                     (ath_rc_priv->hw_maxretry_pktcnt >= 1)) {
581                         rate = next_rate;
582                         ath_rc_priv->probe_rate = rate;
583                         ath_rc_priv->probe_time = now_msec;
584                         ath_rc_priv->hw_maxretry_pktcnt = 0;
585                         *is_probing = 1;
586                 }
587         }
588
589         if (rate > (ath_rc_priv->rate_table_size - 1))
590                 rate = ath_rc_priv->rate_table_size - 1;
591
592         if (rate_table->info[rate].valid &&
593             (ath_rc_priv->ht_cap & WLAN_RC_DS_FLAG))
594                 return rate;
595
596         if (rate_table->info[rate].valid_single_stream &&
597             !(ath_rc_priv->ht_cap & WLAN_RC_DS_FLAG))
598                 return rate;
599
600         /* This should not happen */
601         WARN_ON(1);
602
603         rate = ath_rc_priv->valid_rate_index[0];
604
605         return rate;
606 }
607
608 static void ath_rc_rate_set_series(const struct ath_rate_table *rate_table,
609                                    struct ieee80211_tx_rate *rate,
610                                    struct ieee80211_tx_rate_control *txrc,
611                                    u8 tries, u8 rix, int rtsctsenable)
612 {
613         rate->count = tries;
614         rate->idx = rate_table->info[rix].ratecode;
615
616         if (txrc->short_preamble)
617                 rate->flags |= IEEE80211_TX_RC_USE_SHORT_PREAMBLE;
618         if (txrc->rts || rtsctsenable)
619                 rate->flags |= IEEE80211_TX_RC_USE_RTS_CTS;
620
621         if (WLAN_RC_PHY_HT(rate_table->info[rix].phy)) {
622                 rate->flags |= IEEE80211_TX_RC_MCS;
623                 if (WLAN_RC_PHY_40(rate_table->info[rix].phy))
624                         rate->flags |= IEEE80211_TX_RC_40_MHZ_WIDTH;
625                 if (WLAN_RC_PHY_SGI(rate_table->info[rix].phy))
626                         rate->flags |= IEEE80211_TX_RC_SHORT_GI;
627         }
628 }
629
630 static void ath_rc_rate_set_rtscts(struct ath_softc *sc,
631                                    const struct ath_rate_table *rate_table,
632                                    struct ieee80211_tx_info *tx_info)
633 {
634         struct ieee80211_tx_rate *rates = tx_info->control.rates;
635         int i = 0, rix = 0, cix, enable_g_protection = 0;
636
637         /* get the cix for the lowest valid rix */
638         for (i = 3; i >= 0; i--) {
639                 if (rates[i].count && (rates[i].idx >= 0)) {
640                         rix = ath_rc_get_rateindex(rate_table, &rates[i]);
641                         break;
642                 }
643         }
644         cix = rate_table->info[rix].ctrl_rate;
645
646         /* All protection frames are transmited at 2Mb/s for 802.11g,
647          * otherwise we transmit them at 1Mb/s */
648         if (sc->hw->conf.channel->band == IEEE80211_BAND_2GHZ &&
649             !conf_is_ht(&sc->hw->conf))
650                 enable_g_protection = 1;
651
652         /*
653          * If 802.11g protection is enabled, determine whether to use RTS/CTS or
654          * just CTS.  Note that this is only done for OFDM/HT unicast frames.
655          */
656         if ((sc->sc_flags & SC_OP_PROTECT_ENABLE) &&
657             (rate_table->info[rix].phy == WLAN_RC_PHY_OFDM ||
658              WLAN_RC_PHY_HT(rate_table->info[rix].phy))) {
659                 rates[0].flags |= IEEE80211_TX_RC_USE_CTS_PROTECT;
660                 cix = rate_table->info[enable_g_protection].ctrl_rate;
661         }
662
663         tx_info->control.rts_cts_rate_idx = cix;
664 }
665
666 static void ath_get_rate(void *priv, struct ieee80211_sta *sta, void *priv_sta,
667                          struct ieee80211_tx_rate_control *txrc)
668 {
669         struct ath_softc *sc = priv;
670         struct ath_rate_priv *ath_rc_priv = priv_sta;
671         const struct ath_rate_table *rate_table;
672         struct sk_buff *skb = txrc->skb;
673         struct ieee80211_tx_info *tx_info = IEEE80211_SKB_CB(skb);
674         struct ieee80211_tx_rate *rates = tx_info->control.rates;
675         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
676         __le16 fc = hdr->frame_control;
677         u8 try_per_rate, i = 0, rix;
678         int is_probe = 0;
679
680         if (rate_control_send_low(sta, priv_sta, txrc))
681                 return;
682
683         /*
684          * For Multi Rate Retry we use a different number of
685          * retry attempt counts. This ends up looking like this:
686          *
687          * MRR[0] = 4
688          * MRR[1] = 4
689          * MRR[2] = 4
690          * MRR[3] = 8
691          *
692          */
693         try_per_rate = 4;
694
695         rate_table = sc->cur_rate_table;
696         rix = ath_rc_get_highest_rix(sc, ath_rc_priv, rate_table, &is_probe);
697
698         /*
699          * If we're in HT mode and both us and our peer supports LDPC.
700          * We don't need to check our own device's capabilities as our own
701          * ht capabilities would have already been intersected with our peer's.
702          */
703         if (conf_is_ht(&sc->hw->conf) &&
704             (sta->ht_cap.cap & IEEE80211_HT_CAP_LDPC_CODING))
705                 tx_info->flags |= IEEE80211_TX_CTL_LDPC;
706
707         if (conf_is_ht(&sc->hw->conf) &&
708             (sta->ht_cap.cap & IEEE80211_HT_CAP_TX_STBC))
709                 tx_info->flags |= (1 << IEEE80211_TX_CTL_STBC_SHIFT);
710
711         if (is_probe) {
712                 /* set one try for probe rates. For the
713                  * probes don't enable rts */
714                 ath_rc_rate_set_series(rate_table, &rates[i++], txrc,
715                                        1, rix, 0);
716
717                 /* Get the next tried/allowed rate. No RTS for the next series
718                  * after the probe rate
719                  */
720                 ath_rc_get_lower_rix(rate_table, ath_rc_priv, rix, &rix);
721                 ath_rc_rate_set_series(rate_table, &rates[i++], txrc,
722                                        try_per_rate, rix, 0);
723
724                 tx_info->flags |= IEEE80211_TX_CTL_RATE_CTRL_PROBE;
725         } else {
726                 /* Set the choosen rate. No RTS for first series entry. */
727                 ath_rc_rate_set_series(rate_table, &rates[i++], txrc,
728                                        try_per_rate, rix, 0);
729         }
730
731         /* Fill in the other rates for multirate retry */
732         for ( ; i < 4; i++) {
733                 /* Use twice the number of tries for the last MRR segment. */
734                 if (i + 1 == 4)
735                         try_per_rate = 8;
736
737                 ath_rc_get_lower_rix(rate_table, ath_rc_priv, rix, &rix);
738                 /* All other rates in the series have RTS enabled */
739                 ath_rc_rate_set_series(rate_table, &rates[i], txrc,
740                                        try_per_rate, rix, 1);
741         }
742
743         /*
744          * NB:Change rate series to enable aggregation when operating
745          * at lower MCS rates. When first rate in series is MCS2
746          * in HT40 @ 2.4GHz, series should look like:
747          *
748          * {MCS2, MCS1, MCS0, MCS0}.
749          *
750          * When first rate in series is MCS3 in HT20 @ 2.4GHz, series should
751          * look like:
752          *
753          * {MCS3, MCS2, MCS1, MCS1}
754          *
755          * So, set fourth rate in series to be same as third one for
756          * above conditions.
757          */
758         if ((sc->hw->conf.channel->band == IEEE80211_BAND_2GHZ) &&
759             (conf_is_ht(&sc->hw->conf))) {
760                 u8 dot11rate = rate_table->info[rix].dot11rate;
761                 u8 phy = rate_table->info[rix].phy;
762                 if (i == 4 &&
763                     ((dot11rate == 2 && phy == WLAN_RC_PHY_HT_40_SS) ||
764                      (dot11rate == 3 && phy == WLAN_RC_PHY_HT_20_SS))) {
765                         rates[3].idx = rates[2].idx;
766                         rates[3].flags = rates[2].flags;
767                 }
768         }
769
770         /*
771          * Force hardware to use computed duration for next
772          * fragment by disabling multi-rate retry, which
773          * updates duration based on the multi-rate duration table.
774          *
775          * FIXME: Fix duration
776          */
777         if (ieee80211_has_morefrags(fc) ||
778             (le16_to_cpu(hdr->seq_ctrl) & IEEE80211_SCTL_FRAG)) {
779                 rates[1].count = rates[2].count = rates[3].count = 0;
780                 rates[1].idx = rates[2].idx = rates[3].idx = 0;
781                 rates[0].count = ATH_TXMAXTRY;
782         }
783
784         /* Setup RTS/CTS */
785         ath_rc_rate_set_rtscts(sc, rate_table, tx_info);
786 }
787
788 static bool ath_rc_update_per(struct ath_softc *sc,
789                               const struct ath_rate_table *rate_table,
790                               struct ath_rate_priv *ath_rc_priv,
791                                   struct ieee80211_tx_info *tx_info,
792                               int tx_rate, int xretries, int retries,
793                               u32 now_msec)
794 {
795         bool state_change = false;
796         int count, n_bad_frames;
797         u8 last_per;
798         static u32 nretry_to_per_lookup[10] = {
799                 100 * 0 / 1,
800                 100 * 1 / 4,
801                 100 * 1 / 2,
802                 100 * 3 / 4,
803                 100 * 4 / 5,
804                 100 * 5 / 6,
805                 100 * 6 / 7,
806                 100 * 7 / 8,
807                 100 * 8 / 9,
808                 100 * 9 / 10
809         };
810
811         last_per = ath_rc_priv->per[tx_rate];
812         n_bad_frames = tx_info->status.ampdu_len - tx_info->status.ampdu_ack_len;
813
814         if (xretries) {
815                 if (xretries == 1) {
816                         ath_rc_priv->per[tx_rate] += 30;
817                         if (ath_rc_priv->per[tx_rate] > 100)
818                                 ath_rc_priv->per[tx_rate] = 100;
819                 } else {
820                         /* xretries == 2 */
821                         count = ARRAY_SIZE(nretry_to_per_lookup);
822                         if (retries >= count)
823                                 retries = count - 1;
824
825                         /* new_PER = 7/8*old_PER + 1/8*(currentPER) */
826                         ath_rc_priv->per[tx_rate] =
827                                 (u8)(last_per - (last_per >> 3) + (100 >> 3));
828                 }
829
830                 /* xretries == 1 or 2 */
831
832                 if (ath_rc_priv->probe_rate == tx_rate)
833                         ath_rc_priv->probe_rate = 0;
834
835         } else { /* xretries == 0 */
836                 count = ARRAY_SIZE(nretry_to_per_lookup);
837                 if (retries >= count)
838                         retries = count - 1;
839
840                 if (n_bad_frames) {
841                         /* new_PER = 7/8*old_PER + 1/8*(currentPER)
842                          * Assuming that n_frames is not 0.  The current PER
843                          * from the retries is 100 * retries / (retries+1),
844                          * since the first retries attempts failed, and the
845                          * next one worked.  For the one that worked,
846                          * n_bad_frames subframes out of n_frames wored,
847                          * so the PER for that part is
848                          * 100 * n_bad_frames / n_frames, and it contributes
849                          * 100 * n_bad_frames / (n_frames * (retries+1)) to
850                          * the above PER.  The expression below is a
851                          * simplified version of the sum of these two terms.
852                          */
853                         if (tx_info->status.ampdu_len > 0) {
854                                 int n_frames, n_bad_tries;
855                                 u8 cur_per, new_per;
856
857                                 n_bad_tries = retries * tx_info->status.ampdu_len +
858                                         n_bad_frames;
859                                 n_frames = tx_info->status.ampdu_len * (retries + 1);
860                                 cur_per = (100 * n_bad_tries / n_frames) >> 3;
861                                 new_per = (u8)(last_per - (last_per >> 3) + cur_per);
862                                 ath_rc_priv->per[tx_rate] = new_per;
863                         }
864                 } else {
865                         ath_rc_priv->per[tx_rate] =
866                                 (u8)(last_per - (last_per >> 3) +
867                                      (nretry_to_per_lookup[retries] >> 3));
868                 }
869
870
871                 /*
872                  * If we got at most one retry then increase the max rate if
873                  * this was a probe.  Otherwise, ignore the probe.
874                  */
875                 if (ath_rc_priv->probe_rate && ath_rc_priv->probe_rate == tx_rate) {
876                         if (retries > 0 || 2 * n_bad_frames > tx_info->status.ampdu_len) {
877                                 /*
878                                  * Since we probed with just a single attempt,
879                                  * any retries means the probe failed.  Also,
880                                  * if the attempt worked, but more than half
881                                  * the subframes were bad then also consider
882                                  * the probe a failure.
883                                  */
884                                 ath_rc_priv->probe_rate = 0;
885                         } else {
886                                 u8 probe_rate = 0;
887
888                                 ath_rc_priv->rate_max_phy =
889                                         ath_rc_priv->probe_rate;
890                                 probe_rate = ath_rc_priv->probe_rate;
891
892                                 if (ath_rc_priv->per[probe_rate] > 30)
893                                         ath_rc_priv->per[probe_rate] = 20;
894
895                                 ath_rc_priv->probe_rate = 0;
896
897                                 /*
898                                  * Since this probe succeeded, we allow the next
899                                  * probe twice as soon.  This allows the maxRate
900                                  * to move up faster if the probes are
901                                  * successful.
902                                  */
903                                 ath_rc_priv->probe_time =
904                                         now_msec - rate_table->probe_interval / 2;
905                         }
906                 }
907
908                 if (retries > 0) {
909                         /*
910                          * Don't update anything.  We don't know if
911                          * this was because of collisions or poor signal.
912                          */
913                         ath_rc_priv->hw_maxretry_pktcnt = 0;
914                 } else {
915                         /*
916                          * It worked with no retries. First ignore bogus (small)
917                          * rssi_ack values.
918                          */
919                         if (tx_rate == ath_rc_priv->rate_max_phy &&
920                             ath_rc_priv->hw_maxretry_pktcnt < 255) {
921                                 ath_rc_priv->hw_maxretry_pktcnt++;
922                         }
923
924                 }
925         }
926
927         return state_change;
928 }
929
930 /* Update PER, RSSI and whatever else that the code thinks it is doing.
931    If you can make sense of all this, you really need to go out more. */
932
933 static void ath_rc_update_ht(struct ath_softc *sc,
934                              struct ath_rate_priv *ath_rc_priv,
935                              struct ieee80211_tx_info *tx_info,
936                              int tx_rate, int xretries, int retries)
937 {
938         u32 now_msec = jiffies_to_msecs(jiffies);
939         int rate;
940         u8 last_per;
941         bool state_change = false;
942         const struct ath_rate_table *rate_table = sc->cur_rate_table;
943         int size = ath_rc_priv->rate_table_size;
944
945         if ((tx_rate < 0) || (tx_rate > rate_table->rate_cnt))
946                 return;
947
948         last_per = ath_rc_priv->per[tx_rate];
949
950         /* Update PER first */
951         state_change = ath_rc_update_per(sc, rate_table, ath_rc_priv,
952                                          tx_info, tx_rate, xretries,
953                                          retries, now_msec);
954
955         /*
956          * If this rate looks bad (high PER) then stop using it for
957          * a while (except if we are probing).
958          */
959         if (ath_rc_priv->per[tx_rate] >= 55 && tx_rate > 0 &&
960             rate_table->info[tx_rate].ratekbps <=
961             rate_table->info[ath_rc_priv->rate_max_phy].ratekbps) {
962                 ath_rc_get_lower_rix(rate_table, ath_rc_priv,
963                                      (u8)tx_rate, &ath_rc_priv->rate_max_phy);
964
965                 /* Don't probe for a little while. */
966                 ath_rc_priv->probe_time = now_msec;
967         }
968
969         /* Make sure the rates below this have lower PER */
970         /* Monotonicity is kept only for rates below the current rate. */
971         if (ath_rc_priv->per[tx_rate] < last_per) {
972                 for (rate = tx_rate - 1; rate >= 0; rate--) {
973
974                         if (ath_rc_priv->per[rate] >
975                             ath_rc_priv->per[rate+1]) {
976                                 ath_rc_priv->per[rate] =
977                                         ath_rc_priv->per[rate+1];
978                         }
979                 }
980         }
981
982         /* Maintain monotonicity for rates above the current rate */
983         for (rate = tx_rate; rate < size - 1; rate++) {
984                 if (ath_rc_priv->per[rate+1] <
985                     ath_rc_priv->per[rate])
986                         ath_rc_priv->per[rate+1] =
987                                 ath_rc_priv->per[rate];
988         }
989
990         /* Every so often, we reduce the thresholds
991          * and PER (different for CCK and OFDM). */
992         if (now_msec - ath_rc_priv->per_down_time >=
993             rate_table->probe_interval) {
994                 for (rate = 0; rate < size; rate++) {
995                         ath_rc_priv->per[rate] =
996                                 7 * ath_rc_priv->per[rate] / 8;
997                 }
998
999                 ath_rc_priv->per_down_time = now_msec;
1000         }
1001
1002         ath_debug_stat_retries(sc, tx_rate, xretries, retries,
1003                                ath_rc_priv->per[tx_rate]);
1004
1005 }
1006
1007 static int ath_rc_get_rateindex(const struct ath_rate_table *rate_table,
1008                                 struct ieee80211_tx_rate *rate)
1009 {
1010         int rix;
1011
1012         if (!(rate->flags & IEEE80211_TX_RC_MCS))
1013                 return rate->idx;
1014
1015         rix = rate->idx + rate_table->mcs_start;
1016         if ((rate->flags & IEEE80211_TX_RC_40_MHZ_WIDTH) &&
1017             (rate->flags & IEEE80211_TX_RC_SHORT_GI))
1018                 rix = rate_table->info[rix].ht_index;
1019         else if (rate->flags & IEEE80211_TX_RC_SHORT_GI)
1020                 rix = rate_table->info[rix].sgi_index;
1021         else if (rate->flags & IEEE80211_TX_RC_40_MHZ_WIDTH)
1022                 rix = rate_table->info[rix].cw40index;
1023         else
1024                 rix = rate_table->info[rix].base_index;
1025
1026         return rix;
1027 }
1028
1029 static void ath_rc_tx_status(struct ath_softc *sc,
1030                              struct ath_rate_priv *ath_rc_priv,
1031                              struct ieee80211_tx_info *tx_info,
1032                              int final_ts_idx, int xretries, int long_retry)
1033 {
1034         const struct ath_rate_table *rate_table;
1035         struct ieee80211_tx_rate *rates = tx_info->status.rates;
1036         u8 flags;
1037         u32 i = 0, rix;
1038
1039         rate_table = sc->cur_rate_table;
1040
1041         /*
1042          * If the first rate is not the final index, there
1043          * are intermediate rate failures to be processed.
1044          */
1045         if (final_ts_idx != 0) {
1046                 /* Process intermediate rates that failed.*/
1047                 for (i = 0; i < final_ts_idx ; i++) {
1048                         if (rates[i].count != 0 && (rates[i].idx >= 0)) {
1049                                 flags = rates[i].flags;
1050
1051                                 /* If HT40 and we have switched mode from
1052                                  * 40 to 20 => don't update */
1053
1054                                 if ((flags & IEEE80211_TX_RC_40_MHZ_WIDTH) &&
1055                                     !(ath_rc_priv->ht_cap & WLAN_RC_40_FLAG))
1056                                         return;
1057
1058                                 rix = ath_rc_get_rateindex(rate_table, &rates[i]);
1059                                 ath_rc_update_ht(sc, ath_rc_priv, tx_info,
1060                                                 rix, xretries ? 1 : 2,
1061                                                 rates[i].count);
1062                         }
1063                 }
1064         } else {
1065                 /*
1066                  * Handle the special case of MIMO PS burst, where the second
1067                  * aggregate is sent out with only one rate and one try.
1068                  * Treating it as an excessive retry penalizes the rate
1069                  * inordinately.
1070                  */
1071                 if (rates[0].count == 1 && xretries == 1)
1072                         xretries = 2;
1073         }
1074
1075         flags = rates[i].flags;
1076
1077         /* If HT40 and we have switched mode from 40 to 20 => don't update */
1078         if ((flags & IEEE80211_TX_RC_40_MHZ_WIDTH) &&
1079             !(ath_rc_priv->ht_cap & WLAN_RC_40_FLAG))
1080                 return;
1081
1082         rix = ath_rc_get_rateindex(rate_table, &rates[i]);
1083         ath_rc_update_ht(sc, ath_rc_priv, tx_info, rix, xretries, long_retry);
1084 }
1085
1086 static const
1087 struct ath_rate_table *ath_choose_rate_table(struct ath_softc *sc,
1088                                              enum ieee80211_band band,
1089                                              bool is_ht,
1090                                              bool is_cw_40)
1091 {
1092         int mode = 0;
1093         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1094
1095         switch(band) {
1096         case IEEE80211_BAND_2GHZ:
1097                 mode = ATH9K_MODE_11G;
1098                 if (is_ht)
1099                         mode = ATH9K_MODE_11NG_HT20;
1100                 if (is_cw_40)
1101                         mode = ATH9K_MODE_11NG_HT40PLUS;
1102                 break;
1103         case IEEE80211_BAND_5GHZ:
1104                 mode = ATH9K_MODE_11A;
1105                 if (is_ht)
1106                         mode = ATH9K_MODE_11NA_HT20;
1107                 if (is_cw_40)
1108                         mode = ATH9K_MODE_11NA_HT40PLUS;
1109                 break;
1110         default:
1111                 ath_print(common, ATH_DBG_CONFIG, "Invalid band\n");
1112                 return NULL;
1113         }
1114
1115         BUG_ON(mode >= ATH9K_MODE_MAX);
1116
1117         ath_print(common, ATH_DBG_CONFIG,
1118                   "Choosing rate table for mode: %d\n", mode);
1119
1120         sc->cur_rate_mode = mode;
1121         return hw_rate_table[mode];
1122 }
1123
1124 static void ath_rc_init(struct ath_softc *sc,
1125                         struct ath_rate_priv *ath_rc_priv,
1126                         struct ieee80211_supported_band *sband,
1127                         struct ieee80211_sta *sta,
1128                         const struct ath_rate_table *rate_table)
1129 {
1130         struct ath_rateset *rateset = &ath_rc_priv->neg_rates;
1131         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1132         u8 *ht_mcs = (u8 *)&ath_rc_priv->neg_ht_rates;
1133         u8 i, j, k, hi = 0, hthi = 0;
1134
1135         /* Initial rate table size. Will change depending
1136          * on the working rate set */
1137         ath_rc_priv->rate_table_size = RATE_TABLE_SIZE;
1138
1139         /* Initialize thresholds according to the global rate table */
1140         for (i = 0 ; i < ath_rc_priv->rate_table_size; i++) {
1141                 ath_rc_priv->per[i] = 0;
1142         }
1143
1144         /* Determine the valid rates */
1145         ath_rc_init_valid_txmask(ath_rc_priv);
1146
1147         for (i = 0; i < WLAN_RC_PHY_MAX; i++) {
1148                 for (j = 0; j < MAX_TX_RATE_PHY; j++)
1149                         ath_rc_priv->valid_phy_rateidx[i][j] = 0;
1150                 ath_rc_priv->valid_phy_ratecnt[i] = 0;
1151         }
1152
1153         if (!rateset->rs_nrates) {
1154                 /* No working rate, just initialize valid rates */
1155                 hi = ath_rc_init_validrates(ath_rc_priv, rate_table,
1156                                             ath_rc_priv->ht_cap);
1157         } else {
1158                 /* Use intersection of working rates and valid rates */
1159                 hi = ath_rc_setvalid_rates(ath_rc_priv, rate_table,
1160                                            rateset, ath_rc_priv->ht_cap);
1161                 if (ath_rc_priv->ht_cap & WLAN_RC_HT_FLAG) {
1162                         hthi = ath_rc_setvalid_htrates(ath_rc_priv,
1163                                                        rate_table,
1164                                                        ht_mcs,
1165                                                        ath_rc_priv->ht_cap);
1166                 }
1167                 hi = A_MAX(hi, hthi);
1168         }
1169
1170         ath_rc_priv->rate_table_size = hi + 1;
1171         ath_rc_priv->rate_max_phy = 0;
1172         BUG_ON(ath_rc_priv->rate_table_size > RATE_TABLE_SIZE);
1173
1174         for (i = 0, k = 0; i < WLAN_RC_PHY_MAX; i++) {
1175                 for (j = 0; j < ath_rc_priv->valid_phy_ratecnt[i]; j++) {
1176                         ath_rc_priv->valid_rate_index[k++] =
1177                                 ath_rc_priv->valid_phy_rateidx[i][j];
1178                 }
1179
1180                 if (!ath_rc_valid_phyrate(i, rate_table->initial_ratemax, 1)
1181                     || !ath_rc_priv->valid_phy_ratecnt[i])
1182                         continue;
1183
1184                 ath_rc_priv->rate_max_phy = ath_rc_priv->valid_phy_rateidx[i][j-1];
1185         }
1186         BUG_ON(ath_rc_priv->rate_table_size > RATE_TABLE_SIZE);
1187         BUG_ON(k > RATE_TABLE_SIZE);
1188
1189         ath_rc_priv->max_valid_rate = k;
1190         ath_rc_sort_validrates(rate_table, ath_rc_priv);
1191         ath_rc_priv->rate_max_phy = ath_rc_priv->valid_rate_index[k-4];
1192         sc->cur_rate_table = rate_table;
1193
1194         ath_print(common, ATH_DBG_CONFIG,
1195                   "RC Initialized with capabilities: 0x%x\n",
1196                   ath_rc_priv->ht_cap);
1197 }
1198
1199 static u8 ath_rc_build_ht_caps(struct ath_softc *sc, struct ieee80211_sta *sta,
1200                                bool is_cw40, bool is_sgi)
1201 {
1202         u8 caps = 0;
1203
1204         if (sta->ht_cap.ht_supported) {
1205                 caps = WLAN_RC_HT_FLAG;
1206                 if (sta->ht_cap.mcs.rx_mask[1])
1207                         caps |= WLAN_RC_DS_FLAG;
1208                 if (is_cw40)
1209                         caps |= WLAN_RC_40_FLAG;
1210                 if (is_sgi)
1211                         caps |= WLAN_RC_SGI_FLAG;
1212
1213         }
1214
1215         return caps;
1216 }
1217
1218 /***********************************/
1219 /* mac80211 Rate Control callbacks */
1220 /***********************************/
1221
1222 static void ath_tx_status(void *priv, struct ieee80211_supported_band *sband,
1223                           struct ieee80211_sta *sta, void *priv_sta,
1224                           struct sk_buff *skb)
1225 {
1226         struct ath_softc *sc = priv;
1227         struct ath_rate_priv *ath_rc_priv = priv_sta;
1228         struct ieee80211_tx_info *tx_info = IEEE80211_SKB_CB(skb);
1229         struct ieee80211_hdr *hdr;
1230         int final_ts_idx = 0, tx_status = 0, is_underrun = 0;
1231         int long_retry = 0;
1232         __le16 fc;
1233         int i;
1234
1235         hdr = (struct ieee80211_hdr *)skb->data;
1236         fc = hdr->frame_control;
1237         for (i = 0; i < IEEE80211_TX_MAX_RATES; i++) {
1238                 struct ieee80211_tx_rate *rate = &tx_info->status.rates[i];
1239                 if (!rate->count)
1240                         break;
1241
1242                 final_ts_idx = i;
1243                 long_retry = rate->count - 1;
1244         }
1245
1246         if (!priv_sta || !ieee80211_is_data(fc))
1247                 return;
1248
1249         /* This packet was aggregated but doesn't carry status info */
1250         if ((tx_info->flags & IEEE80211_TX_CTL_AMPDU) &&
1251             !(tx_info->flags & IEEE80211_TX_STAT_AMPDU))
1252                 return;
1253
1254         if (tx_info->flags & IEEE80211_TX_STAT_TX_FILTERED)
1255                 return;
1256
1257         /*
1258          * If an underrun error is seen assume it as an excessive retry only
1259          * if max frame trigger level has been reached (2 KB for singel stream,
1260          * and 4 KB for dual stream). Adjust the long retry as if the frame was
1261          * tried hw->max_rate_tries times to affect how ratectrl updates PER for
1262          * the failed rate. In case of congestion on the bus penalizing these
1263          * type of underruns should help hardware actually transmit new frames
1264          * successfully by eventually preferring slower rates. This itself
1265          * should also alleviate congestion on the bus.
1266          */
1267         if ((tx_info->pad[0] & ATH_TX_INFO_UNDERRUN) &&
1268             (sc->sc_ah->tx_trig_level >= ath_rc_priv->tx_triglevel_max)) {
1269                 tx_status = 1;
1270                 is_underrun = 1;
1271         }
1272
1273         if (tx_info->pad[0] & ATH_TX_INFO_XRETRY)
1274                 tx_status = 1;
1275
1276         ath_rc_tx_status(sc, ath_rc_priv, tx_info, final_ts_idx, tx_status,
1277                          (is_underrun) ? sc->hw->max_rate_tries : long_retry);
1278
1279         /* Check if aggregation has to be enabled for this tid */
1280         if (conf_is_ht(&sc->hw->conf) &&
1281             !(skb->protocol == cpu_to_be16(ETH_P_PAE))) {
1282                 if (ieee80211_is_data_qos(fc)) {
1283                         u8 *qc, tid;
1284                         struct ath_node *an;
1285
1286                         qc = ieee80211_get_qos_ctl(hdr);
1287                         tid = qc[0] & 0xf;
1288                         an = (struct ath_node *)sta->drv_priv;
1289
1290                         if(ath_tx_aggr_check(sc, an, tid))
1291                                 ieee80211_start_tx_ba_session(sta, tid);
1292                 }
1293         }
1294
1295         ath_debug_stat_rc(sc, ath_rc_get_rateindex(sc->cur_rate_table,
1296                 &tx_info->status.rates[final_ts_idx]));
1297 }
1298
1299 static void ath_rate_init(void *priv, struct ieee80211_supported_band *sband,
1300                           struct ieee80211_sta *sta, void *priv_sta)
1301 {
1302         struct ath_softc *sc = priv;
1303         struct ath_rate_priv *ath_rc_priv = priv_sta;
1304         const struct ath_rate_table *rate_table;
1305         bool is_cw40, is_sgi = false;
1306         int i, j = 0;
1307
1308         for (i = 0; i < sband->n_bitrates; i++) {
1309                 if (sta->supp_rates[sband->band] & BIT(i)) {
1310                         ath_rc_priv->neg_rates.rs_rates[j]
1311                                 = (sband->bitrates[i].bitrate * 2) / 10;
1312                         j++;
1313                 }
1314         }
1315         ath_rc_priv->neg_rates.rs_nrates = j;
1316
1317         if (sta->ht_cap.ht_supported) {
1318                 for (i = 0, j = 0; i < 77; i++) {
1319                         if (sta->ht_cap.mcs.rx_mask[i/8] & (1<<(i%8)))
1320                                 ath_rc_priv->neg_ht_rates.rs_rates[j++] = i;
1321                         if (j == ATH_RATE_MAX)
1322                                 break;
1323                 }
1324                 ath_rc_priv->neg_ht_rates.rs_nrates = j;
1325         }
1326
1327         is_cw40 = sta->ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40;
1328
1329         if (is_cw40)
1330                 is_sgi = sta->ht_cap.cap & IEEE80211_HT_CAP_SGI_40;
1331         else if (sc->sc_ah->caps.hw_caps & ATH9K_HW_CAP_SGI_20)
1332                 is_sgi = sta->ht_cap.cap & IEEE80211_HT_CAP_SGI_20;
1333
1334         /* Choose rate table first */
1335
1336         if ((sc->sc_ah->opmode == NL80211_IFTYPE_STATION) ||
1337             (sc->sc_ah->opmode == NL80211_IFTYPE_MESH_POINT) ||
1338             (sc->sc_ah->opmode == NL80211_IFTYPE_ADHOC)) {
1339                 rate_table = ath_choose_rate_table(sc, sband->band,
1340                                       sta->ht_cap.ht_supported, is_cw40);
1341         } else {
1342                 rate_table = hw_rate_table[sc->cur_rate_mode];
1343         }
1344
1345         ath_rc_priv->ht_cap = ath_rc_build_ht_caps(sc, sta, is_cw40, is_sgi);
1346         ath_rc_init(sc, priv_sta, sband, sta, rate_table);
1347 }
1348
1349 static void ath_rate_update(void *priv, struct ieee80211_supported_band *sband,
1350                             struct ieee80211_sta *sta, void *priv_sta,
1351                             u32 changed, enum nl80211_channel_type oper_chan_type)
1352 {
1353         struct ath_softc *sc = priv;
1354         struct ath_rate_priv *ath_rc_priv = priv_sta;
1355         const struct ath_rate_table *rate_table = NULL;
1356         bool oper_cw40 = false, oper_sgi;
1357         bool local_cw40 = (ath_rc_priv->ht_cap & WLAN_RC_40_FLAG) ?
1358                 true : false;
1359         bool local_sgi = (ath_rc_priv->ht_cap & WLAN_RC_SGI_FLAG) ?
1360                 true : false;
1361
1362         /* FIXME: Handle AP mode later when we support CWM */
1363
1364         if (changed & IEEE80211_RC_HT_CHANGED) {
1365                 if (sc->sc_ah->opmode != NL80211_IFTYPE_STATION)
1366                         return;
1367
1368                 if (oper_chan_type == NL80211_CHAN_HT40MINUS ||
1369                     oper_chan_type == NL80211_CHAN_HT40PLUS)
1370                         oper_cw40 = true;
1371
1372                 if (oper_cw40)
1373                         oper_sgi = (sta->ht_cap.cap & IEEE80211_HT_CAP_SGI_40) ?
1374                                    true : false;
1375                 else if (sc->sc_ah->caps.hw_caps & ATH9K_HW_CAP_SGI_20)
1376                         oper_sgi = (sta->ht_cap.cap & IEEE80211_HT_CAP_SGI_20) ?
1377                                    true : false;
1378                 else
1379                         oper_sgi = false;
1380
1381                 if ((local_cw40 != oper_cw40) || (local_sgi != oper_sgi)) {
1382                         rate_table = ath_choose_rate_table(sc, sband->band,
1383                                                    sta->ht_cap.ht_supported,
1384                                                    oper_cw40);
1385                         ath_rc_priv->ht_cap = ath_rc_build_ht_caps(sc, sta,
1386                                                    oper_cw40, oper_sgi);
1387                         ath_rc_init(sc, priv_sta, sband, sta, rate_table);
1388
1389                         ath_print(ath9k_hw_common(sc->sc_ah), ATH_DBG_CONFIG,
1390                                   "Operating HT Bandwidth changed to: %d\n",
1391                                   sc->hw->conf.channel_type);
1392                         sc->cur_rate_table = hw_rate_table[sc->cur_rate_mode];
1393                 }
1394         }
1395 }
1396
1397 static void *ath_rate_alloc(struct ieee80211_hw *hw, struct dentry *debugfsdir)
1398 {
1399         struct ath_wiphy *aphy = hw->priv;
1400         return aphy->sc;
1401 }
1402
1403 static void ath_rate_free(void *priv)
1404 {
1405         return;
1406 }
1407
1408 static void *ath_rate_alloc_sta(void *priv, struct ieee80211_sta *sta, gfp_t gfp)
1409 {
1410         struct ath_softc *sc = priv;
1411         struct ath_rate_priv *rate_priv;
1412
1413         rate_priv = kzalloc(sizeof(struct ath_rate_priv), gfp);
1414         if (!rate_priv) {
1415                 ath_print(ath9k_hw_common(sc->sc_ah), ATH_DBG_FATAL,
1416                           "Unable to allocate private rc structure\n");
1417                 return NULL;
1418         }
1419
1420         rate_priv->tx_triglevel_max = sc->sc_ah->caps.tx_triglevel_max;
1421
1422         return rate_priv;
1423 }
1424
1425 static void ath_rate_free_sta(void *priv, struct ieee80211_sta *sta,
1426                               void *priv_sta)
1427 {
1428         struct ath_rate_priv *rate_priv = priv_sta;
1429         kfree(rate_priv);
1430 }
1431
1432 static struct rate_control_ops ath_rate_ops = {
1433         .module = NULL,
1434         .name = "ath9k_rate_control",
1435         .tx_status = ath_tx_status,
1436         .get_rate = ath_get_rate,
1437         .rate_init = ath_rate_init,
1438         .rate_update = ath_rate_update,
1439         .alloc = ath_rate_alloc,
1440         .free = ath_rate_free,
1441         .alloc_sta = ath_rate_alloc_sta,
1442         .free_sta = ath_rate_free_sta,
1443 };
1444
1445 int ath_rate_control_register(void)
1446 {
1447         return ieee80211_rate_control_register(&ath_rate_ops);
1448 }
1449
1450 void ath_rate_control_unregister(void)
1451 {
1452         ieee80211_rate_control_unregister(&ath_rate_ops);
1453 }