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