Commit | Line | Data |
---|---|---|
f0706e82 JB |
1 | /* |
2 | * Copyright 2002-2005, Instant802 Networks, Inc. | |
3 | * Copyright 2005-2006, Devicescape Software, Inc. | |
4 | * Copyright (c) 2006 Jiri Benc <jbenc@suse.cz> | |
5 | * | |
6 | * This program is free software; you can redistribute it and/or modify | |
7 | * it under the terms of the GNU General Public License version 2 as | |
8 | * published by the Free Software Foundation. | |
9 | */ | |
10 | ||
11 | #include <linux/kernel.h> | |
ff688089 | 12 | #include <linux/rtnetlink.h> |
3a9a231d | 13 | #include <linux/module.h> |
cc01f9b5 | 14 | #include <linux/slab.h> |
2c8dccc7 | 15 | #include "rate.h" |
f0706e82 | 16 | #include "ieee80211_i.h" |
4b7679a5 | 17 | #include "debugfs.h" |
f0706e82 JB |
18 | |
19 | struct rate_control_alg { | |
20 | struct list_head list; | |
631ad703 | 21 | const struct rate_control_ops *ops; |
f0706e82 JB |
22 | }; |
23 | ||
24 | static LIST_HEAD(rate_ctrl_algs); | |
25 | static DEFINE_MUTEX(rate_ctrl_mutex); | |
26 | ||
c21b39ac SB |
27 | static char *ieee80211_default_rc_algo = CONFIG_MAC80211_RC_DEFAULT; |
28 | module_param(ieee80211_default_rc_algo, charp, 0644); | |
29 | MODULE_PARM_DESC(ieee80211_default_rc_algo, | |
30 | "Default rate control algorithm for mac80211 to use"); | |
31 | ||
631ad703 | 32 | int ieee80211_rate_control_register(const struct rate_control_ops *ops) |
f0706e82 JB |
33 | { |
34 | struct rate_control_alg *alg; | |
35 | ||
ac71c691 JB |
36 | if (!ops->name) |
37 | return -EINVAL; | |
38 | ||
999acd9c JB |
39 | mutex_lock(&rate_ctrl_mutex); |
40 | list_for_each_entry(alg, &rate_ctrl_algs, list) { | |
41 | if (!strcmp(alg->ops->name, ops->name)) { | |
42 | /* don't register an algorithm twice */ | |
43 | WARN_ON(1); | |
b808ab16 | 44 | mutex_unlock(&rate_ctrl_mutex); |
999acd9c JB |
45 | return -EALREADY; |
46 | } | |
47 | } | |
48 | ||
dd00cc48 | 49 | alg = kzalloc(sizeof(*alg), GFP_KERNEL); |
f0706e82 | 50 | if (alg == NULL) { |
999acd9c | 51 | mutex_unlock(&rate_ctrl_mutex); |
f0706e82 JB |
52 | return -ENOMEM; |
53 | } | |
f0706e82 JB |
54 | alg->ops = ops; |
55 | ||
f0706e82 JB |
56 | list_add_tail(&alg->list, &rate_ctrl_algs); |
57 | mutex_unlock(&rate_ctrl_mutex); | |
58 | ||
59 | return 0; | |
60 | } | |
61 | EXPORT_SYMBOL(ieee80211_rate_control_register); | |
62 | ||
631ad703 | 63 | void ieee80211_rate_control_unregister(const struct rate_control_ops *ops) |
f0706e82 JB |
64 | { |
65 | struct rate_control_alg *alg; | |
66 | ||
67 | mutex_lock(&rate_ctrl_mutex); | |
68 | list_for_each_entry(alg, &rate_ctrl_algs, list) { | |
69 | if (alg->ops == ops) { | |
70 | list_del(&alg->list); | |
20880e89 | 71 | kfree(alg); |
f0706e82 JB |
72 | break; |
73 | } | |
74 | } | |
75 | mutex_unlock(&rate_ctrl_mutex); | |
f0706e82 JB |
76 | } |
77 | EXPORT_SYMBOL(ieee80211_rate_control_unregister); | |
78 | ||
631ad703 | 79 | static const struct rate_control_ops * |
f0706e82 JB |
80 | ieee80211_try_rate_control_ops_get(const char *name) |
81 | { | |
82 | struct rate_control_alg *alg; | |
631ad703 | 83 | const struct rate_control_ops *ops = NULL; |
f0706e82 | 84 | |
ac71c691 JB |
85 | if (!name) |
86 | return NULL; | |
87 | ||
f0706e82 JB |
88 | mutex_lock(&rate_ctrl_mutex); |
89 | list_for_each_entry(alg, &rate_ctrl_algs, list) { | |
cc01f9b5 JB |
90 | if (!strcmp(alg->ops->name, name)) { |
91 | ops = alg->ops; | |
92 | break; | |
93 | } | |
f0706e82 JB |
94 | } |
95 | mutex_unlock(&rate_ctrl_mutex); | |
96 | return ops; | |
97 | } | |
98 | ||
c21b39ac | 99 | /* Get the rate control algorithm. */ |
631ad703 | 100 | static const struct rate_control_ops * |
f0706e82 JB |
101 | ieee80211_rate_control_ops_get(const char *name) |
102 | { | |
631ad703 | 103 | const struct rate_control_ops *ops; |
c21b39ac | 104 | const char *alg_name; |
f0706e82 | 105 | |
d6d1b650 | 106 | kparam_block_sysfs_write(ieee80211_default_rc_algo); |
ac71c691 | 107 | if (!name) |
c21b39ac SB |
108 | alg_name = ieee80211_default_rc_algo; |
109 | else | |
110 | alg_name = name; | |
ac71c691 | 111 | |
c21b39ac | 112 | ops = ieee80211_try_rate_control_ops_get(alg_name); |
c21b39ac SB |
113 | if (!ops && name) |
114 | /* try default if specific alg requested but not found */ | |
115 | ops = ieee80211_try_rate_control_ops_get(ieee80211_default_rc_algo); | |
116 | ||
4b475898 JB |
117 | /* try built-in one if specific alg requested but not found */ |
118 | if (!ops && strlen(CONFIG_MAC80211_RC_DEFAULT)) | |
119 | ops = ieee80211_try_rate_control_ops_get(CONFIG_MAC80211_RC_DEFAULT); | |
d6d1b650 | 120 | kparam_unblock_sysfs_write(ieee80211_default_rc_algo); |
4b475898 | 121 | |
f0706e82 JB |
122 | return ops; |
123 | } | |
124 | ||
4b7679a5 JB |
125 | #ifdef CONFIG_MAC80211_DEBUGFS |
126 | static ssize_t rcname_read(struct file *file, char __user *userbuf, | |
127 | size_t count, loff_t *ppos) | |
128 | { | |
129 | struct rate_control_ref *ref = file->private_data; | |
130 | int len = strlen(ref->ops->name); | |
131 | ||
132 | return simple_read_from_buffer(userbuf, count, ppos, | |
133 | ref->ops->name, len); | |
134 | } | |
135 | ||
136 | static const struct file_operations rcname_ops = { | |
137 | .read = rcname_read, | |
234e3405 | 138 | .open = simple_open, |
6038f373 | 139 | .llseek = default_llseek, |
4b7679a5 JB |
140 | }; |
141 | #endif | |
142 | ||
209c671d | 143 | static struct rate_control_ref *rate_control_alloc(const char *name, |
f0706e82 JB |
144 | struct ieee80211_local *local) |
145 | { | |
4b7679a5 | 146 | struct dentry *debugfsdir = NULL; |
f0706e82 JB |
147 | struct rate_control_ref *ref; |
148 | ||
149 | ref = kmalloc(sizeof(struct rate_control_ref), GFP_KERNEL); | |
150 | if (!ref) | |
cc01f9b5 | 151 | return NULL; |
4b7679a5 | 152 | ref->local = local; |
f0706e82 JB |
153 | ref->ops = ieee80211_rate_control_ops_get(name); |
154 | if (!ref->ops) | |
cc01f9b5 | 155 | goto free; |
4b7679a5 JB |
156 | |
157 | #ifdef CONFIG_MAC80211_DEBUGFS | |
158 | debugfsdir = debugfs_create_dir("rc", local->hw.wiphy->debugfsdir); | |
159 | local->debugfs.rcdir = debugfsdir; | |
7bcfaf2f | 160 | debugfs_create_file("name", 0400, debugfsdir, ref, &rcname_ops); |
4b7679a5 JB |
161 | #endif |
162 | ||
163 | ref->priv = ref->ops->alloc(&local->hw, debugfsdir); | |
f0706e82 | 164 | if (!ref->priv) |
cc01f9b5 | 165 | goto free; |
f0706e82 JB |
166 | return ref; |
167 | ||
cc01f9b5 | 168 | free: |
f0706e82 | 169 | kfree(ref); |
f0706e82 JB |
170 | return NULL; |
171 | } | |
172 | ||
889cbb91 | 173 | static void rate_control_free(struct rate_control_ref *ctrl_ref) |
f0706e82 | 174 | { |
f0706e82 | 175 | ctrl_ref->ops->free(ctrl_ref->priv); |
4b7679a5 JB |
176 | |
177 | #ifdef CONFIG_MAC80211_DEBUGFS | |
7bcfaf2f | 178 | debugfs_remove_recursive(ctrl_ref->local->debugfs.rcdir); |
4b7679a5 JB |
179 | ctrl_ref->local->debugfs.rcdir = NULL; |
180 | #endif | |
181 | ||
f0706e82 JB |
182 | kfree(ctrl_ref); |
183 | } | |
184 | ||
b6f35301 | 185 | static bool rc_no_data_or_no_ack_use_min(struct ieee80211_tx_rate_control *txrc) |
4c6d4f5c LR |
186 | { |
187 | struct sk_buff *skb = txrc->skb; | |
188 | struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data; | |
189 | struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb); | |
190 | __le16 fc; | |
191 | ||
192 | fc = hdr->frame_control; | |
193 | ||
b6f35301 RM |
194 | return (info->flags & (IEEE80211_TX_CTL_NO_ACK | |
195 | IEEE80211_TX_CTL_USE_MINRATE)) || | |
196 | !ieee80211_is_data(fc); | |
4c6d4f5c LR |
197 | } |
198 | ||
06703079 | 199 | static void rc_send_low_basicrate(s8 *idx, u32 basic_rates, |
8f0729b1 | 200 | struct ieee80211_supported_band *sband) |
e00cfce0 JM |
201 | { |
202 | u8 i; | |
203 | ||
204 | if (basic_rates == 0) | |
205 | return; /* assume basic rates unknown and accept rate */ | |
206 | if (*idx < 0) | |
207 | return; | |
208 | if (basic_rates & (1 << *idx)) | |
209 | return; /* selected rate is a basic rate */ | |
210 | ||
8f0729b1 | 211 | for (i = *idx + 1; i <= sband->n_bitrates; i++) { |
e00cfce0 JM |
212 | if (basic_rates & (1 << i)) { |
213 | *idx = i; | |
214 | return; | |
215 | } | |
216 | } | |
217 | ||
218 | /* could not find a basic rate; use original selection */ | |
219 | } | |
220 | ||
2103dec1 SW |
221 | static void __rate_control_send_low(struct ieee80211_hw *hw, |
222 | struct ieee80211_supported_band *sband, | |
223 | struct ieee80211_sta *sta, | |
1d2d350b AO |
224 | struct ieee80211_tx_info *info, |
225 | u32 rate_mask) | |
aad14ceb RM |
226 | { |
227 | int i; | |
2103dec1 SW |
228 | u32 rate_flags = |
229 | ieee80211_chandef_rate_flags(&hw->conf.chandef); | |
aad14ceb | 230 | |
2103dec1 SW |
231 | if ((sband->band == IEEE80211_BAND_2GHZ) && |
232 | (info->flags & IEEE80211_TX_CTL_NO_CCK_RATE)) | |
233 | rate_flags |= IEEE80211_RATE_ERP_G; | |
234 | ||
235 | info->control.rates[0].idx = 0; | |
aad14ceb | 236 | for (i = 0; i < sband->n_bitrates; i++) { |
1d2d350b AO |
237 | if (!(rate_mask & BIT(i))) |
238 | continue; | |
239 | ||
1431fcb7 AO |
240 | if ((rate_flags & sband->bitrates[i].flags) != rate_flags) |
241 | continue; | |
242 | ||
2103dec1 | 243 | if (!rate_supported(sta, sband->band, i)) |
aad14ceb RM |
244 | continue; |
245 | ||
2103dec1 SW |
246 | info->control.rates[0].idx = i; |
247 | break; | |
aad14ceb | 248 | } |
2103dec1 | 249 | WARN_ON_ONCE(i == sband->n_bitrates); |
0d528d85 FF |
250 | |
251 | info->control.rates[0].count = | |
252 | (info->flags & IEEE80211_TX_CTL_NO_ACK) ? | |
253 | 1 : hw->max_rate_tries; | |
254 | ||
255 | info->control.skip_table = 1; | |
256 | } | |
257 | ||
aad14ceb | 258 | |
06703079 | 259 | bool rate_control_send_low(struct ieee80211_sta *pubsta, |
4c6d4f5c LR |
260 | void *priv_sta, |
261 | struct ieee80211_tx_rate_control *txrc) | |
262 | { | |
263 | struct ieee80211_tx_info *info = IEEE80211_SKB_CB(txrc->skb); | |
dd5b4cc7 | 264 | struct ieee80211_supported_band *sband = txrc->sband; |
06703079 | 265 | struct sta_info *sta; |
dd5b4cc7 | 266 | int mcast_rate; |
06703079 | 267 | bool use_basicrate = false; |
4c6d4f5c | 268 | |
06703079 | 269 | if (!pubsta || !priv_sta || rc_no_data_or_no_ack_use_min(txrc)) { |
1d2d350b AO |
270 | __rate_control_send_low(txrc->hw, sband, pubsta, info, |
271 | txrc->rate_idx_mask); | |
0d528d85 | 272 | |
06703079 | 273 | if (!pubsta && txrc->bss) { |
dd5b4cc7 FF |
274 | mcast_rate = txrc->bss_conf->mcast_rate[sband->band]; |
275 | if (mcast_rate > 0) { | |
276 | info->control.rates[0].idx = mcast_rate - 1; | |
277 | return true; | |
278 | } | |
06703079 CYY |
279 | use_basicrate = true; |
280 | } else if (pubsta) { | |
281 | sta = container_of(pubsta, struct sta_info, sta); | |
282 | if (ieee80211_vif_is_mesh(&sta->sdata->vif)) | |
283 | use_basicrate = true; | |
284 | } | |
dd5b4cc7 | 285 | |
06703079 CYY |
286 | if (use_basicrate) |
287 | rc_send_low_basicrate(&info->control.rates[0].idx, | |
e00cfce0 | 288 | txrc->bss_conf->basic_rates, |
dd5b4cc7 | 289 | sband); |
06703079 | 290 | |
4c6d4f5c LR |
291 | return true; |
292 | } | |
293 | return false; | |
294 | } | |
295 | EXPORT_SYMBOL(rate_control_send_low); | |
296 | ||
19468413 SW |
297 | static bool rate_idx_match_legacy_mask(struct ieee80211_tx_rate *rate, |
298 | int n_bitrates, u32 mask) | |
37eb0b16 JM |
299 | { |
300 | int j; | |
301 | ||
302 | /* See whether the selected rate or anything below it is allowed. */ | |
303 | for (j = rate->idx; j >= 0; j--) { | |
304 | if (mask & (1 << j)) { | |
305 | /* Okay, found a suitable rate. Use it. */ | |
306 | rate->idx = j; | |
19468413 | 307 | return true; |
37eb0b16 JM |
308 | } |
309 | } | |
310 | ||
311 | /* Try to find a higher rate that would be allowed */ | |
312 | for (j = rate->idx + 1; j < n_bitrates; j++) { | |
313 | if (mask & (1 << j)) { | |
314 | /* Okay, found a suitable rate. Use it. */ | |
315 | rate->idx = j; | |
19468413 SW |
316 | return true; |
317 | } | |
318 | } | |
319 | return false; | |
320 | } | |
321 | ||
322 | static bool rate_idx_match_mcs_mask(struct ieee80211_tx_rate *rate, | |
323 | u8 mcs_mask[IEEE80211_HT_MCS_MASK_LEN]) | |
324 | { | |
325 | int i, j; | |
326 | int ridx, rbit; | |
327 | ||
328 | ridx = rate->idx / 8; | |
329 | rbit = rate->idx % 8; | |
330 | ||
331 | /* sanity check */ | |
910570b5 | 332 | if (ridx < 0 || ridx >= IEEE80211_HT_MCS_MASK_LEN) |
19468413 SW |
333 | return false; |
334 | ||
335 | /* See whether the selected rate or anything below it is allowed. */ | |
336 | for (i = ridx; i >= 0; i--) { | |
337 | for (j = rbit; j >= 0; j--) | |
338 | if (mcs_mask[i] & BIT(j)) { | |
339 | rate->idx = i * 8 + j; | |
340 | return true; | |
341 | } | |
342 | rbit = 7; | |
343 | } | |
344 | ||
345 | /* Try to find a higher rate that would be allowed */ | |
346 | ridx = (rate->idx + 1) / 8; | |
347 | rbit = (rate->idx + 1) % 8; | |
348 | ||
349 | for (i = ridx; i < IEEE80211_HT_MCS_MASK_LEN; i++) { | |
350 | for (j = rbit; j < 8; j++) | |
351 | if (mcs_mask[i] & BIT(j)) { | |
352 | rate->idx = i * 8 + j; | |
353 | return true; | |
354 | } | |
355 | rbit = 0; | |
356 | } | |
357 | return false; | |
358 | } | |
359 | ||
360 | ||
361 | ||
362 | static void rate_idx_match_mask(struct ieee80211_tx_rate *rate, | |
0d528d85 FF |
363 | struct ieee80211_supported_band *sband, |
364 | enum nl80211_chan_width chan_width, | |
19468413 SW |
365 | u32 mask, |
366 | u8 mcs_mask[IEEE80211_HT_MCS_MASK_LEN]) | |
367 | { | |
368 | struct ieee80211_tx_rate alt_rate; | |
369 | ||
370 | /* handle HT rates */ | |
371 | if (rate->flags & IEEE80211_TX_RC_MCS) { | |
372 | if (rate_idx_match_mcs_mask(rate, mcs_mask)) | |
373 | return; | |
374 | ||
375 | /* also try the legacy rates. */ | |
376 | alt_rate.idx = 0; | |
377 | /* keep protection flags */ | |
378 | alt_rate.flags = rate->flags & | |
379 | (IEEE80211_TX_RC_USE_RTS_CTS | | |
380 | IEEE80211_TX_RC_USE_CTS_PROTECT | | |
381 | IEEE80211_TX_RC_USE_SHORT_PREAMBLE); | |
382 | alt_rate.count = rate->count; | |
383 | if (rate_idx_match_legacy_mask(&alt_rate, | |
0d528d85 | 384 | sband->n_bitrates, mask)) { |
19468413 SW |
385 | *rate = alt_rate; |
386 | return; | |
387 | } | |
628c010f | 388 | } else if (!(rate->flags & IEEE80211_TX_RC_VHT_MCS)) { |
19468413 | 389 | /* handle legacy rates */ |
0d528d85 | 390 | if (rate_idx_match_legacy_mask(rate, sband->n_bitrates, mask)) |
19468413 SW |
391 | return; |
392 | ||
393 | /* if HT BSS, and we handle a data frame, also try HT rates */ | |
0418a445 SW |
394 | switch (chan_width) { |
395 | case NL80211_CHAN_WIDTH_20_NOHT: | |
396 | case NL80211_CHAN_WIDTH_5: | |
397 | case NL80211_CHAN_WIDTH_10: | |
19468413 | 398 | return; |
0418a445 SW |
399 | default: |
400 | break; | |
401 | } | |
19468413 SW |
402 | |
403 | alt_rate.idx = 0; | |
404 | /* keep protection flags */ | |
405 | alt_rate.flags = rate->flags & | |
406 | (IEEE80211_TX_RC_USE_RTS_CTS | | |
407 | IEEE80211_TX_RC_USE_CTS_PROTECT | | |
408 | IEEE80211_TX_RC_USE_SHORT_PREAMBLE); | |
409 | alt_rate.count = rate->count; | |
410 | ||
411 | alt_rate.flags |= IEEE80211_TX_RC_MCS; | |
412 | ||
0d528d85 | 413 | if (chan_width == NL80211_CHAN_WIDTH_40) |
19468413 SW |
414 | alt_rate.flags |= IEEE80211_TX_RC_40_MHZ_WIDTH; |
415 | ||
416 | if (rate_idx_match_mcs_mask(&alt_rate, mcs_mask)) { | |
417 | *rate = alt_rate; | |
37eb0b16 JM |
418 | return; |
419 | } | |
420 | } | |
421 | ||
422 | /* | |
423 | * Uh.. No suitable rate exists. This should not really happen with | |
424 | * sane TX rate mask configurations. However, should someone manage to | |
425 | * configure supported rates and TX rate mask in incompatible way, | |
426 | * allow the frame to be transmitted with whatever the rate control | |
427 | * selected. | |
428 | */ | |
429 | } | |
430 | ||
0d528d85 FF |
431 | static void rate_fixup_ratelist(struct ieee80211_vif *vif, |
432 | struct ieee80211_supported_band *sband, | |
433 | struct ieee80211_tx_info *info, | |
434 | struct ieee80211_tx_rate *rates, | |
435 | int max_rates) | |
436 | { | |
437 | struct ieee80211_rate *rate; | |
438 | bool inval = false; | |
439 | int i; | |
440 | ||
441 | /* | |
442 | * Set up the RTS/CTS rate as the fastest basic rate | |
443 | * that is not faster than the data rate unless there | |
444 | * is no basic rate slower than the data rate, in which | |
445 | * case we pick the slowest basic rate | |
446 | * | |
447 | * XXX: Should this check all retry rates? | |
448 | */ | |
336004e2 FF |
449 | if (!(rates[0].flags & |
450 | (IEEE80211_TX_RC_MCS | IEEE80211_TX_RC_VHT_MCS))) { | |
0d528d85 | 451 | u32 basic_rates = vif->bss_conf.basic_rates; |
c7abf25a | 452 | s8 baserate = basic_rates ? ffs(basic_rates) - 1 : 0; |
0d528d85 FF |
453 | |
454 | rate = &sband->bitrates[rates[0].idx]; | |
455 | ||
456 | for (i = 0; i < sband->n_bitrates; i++) { | |
457 | /* must be a basic rate */ | |
458 | if (!(basic_rates & BIT(i))) | |
459 | continue; | |
460 | /* must not be faster than the data rate */ | |
461 | if (sband->bitrates[i].bitrate > rate->bitrate) | |
462 | continue; | |
463 | /* maximum */ | |
464 | if (sband->bitrates[baserate].bitrate < | |
465 | sband->bitrates[i].bitrate) | |
466 | baserate = i; | |
467 | } | |
468 | ||
469 | info->control.rts_cts_rate_idx = baserate; | |
470 | } | |
471 | ||
472 | for (i = 0; i < max_rates; i++) { | |
473 | /* | |
474 | * make sure there's no valid rate following | |
475 | * an invalid one, just in case drivers don't | |
476 | * take the API seriously to stop at -1. | |
477 | */ | |
478 | if (inval) { | |
479 | rates[i].idx = -1; | |
480 | continue; | |
481 | } | |
482 | if (rates[i].idx < 0) { | |
483 | inval = true; | |
484 | continue; | |
485 | } | |
486 | ||
487 | /* | |
488 | * For now assume MCS is already set up correctly, this | |
489 | * needs to be fixed. | |
490 | */ | |
491 | if (rates[i].flags & IEEE80211_TX_RC_MCS) { | |
492 | WARN_ON(rates[i].idx > 76); | |
493 | ||
494 | if (!(rates[i].flags & IEEE80211_TX_RC_USE_RTS_CTS) && | |
495 | info->control.use_cts_prot) | |
496 | rates[i].flags |= | |
497 | IEEE80211_TX_RC_USE_CTS_PROTECT; | |
498 | continue; | |
499 | } | |
500 | ||
501 | if (rates[i].flags & IEEE80211_TX_RC_VHT_MCS) { | |
502 | WARN_ON(ieee80211_rate_get_vht_mcs(&rates[i]) > 9); | |
503 | continue; | |
504 | } | |
505 | ||
506 | /* set up RTS protection if desired */ | |
507 | if (info->control.use_rts) { | |
508 | rates[i].flags |= IEEE80211_TX_RC_USE_RTS_CTS; | |
509 | info->control.use_cts_prot = false; | |
510 | } | |
511 | ||
512 | /* RC is busted */ | |
513 | if (WARN_ON_ONCE(rates[i].idx >= sband->n_bitrates)) { | |
514 | rates[i].idx = -1; | |
515 | continue; | |
516 | } | |
517 | ||
518 | rate = &sband->bitrates[rates[i].idx]; | |
519 | ||
520 | /* set up short preamble */ | |
521 | if (info->control.short_preamble && | |
522 | rate->flags & IEEE80211_RATE_SHORT_PREAMBLE) | |
523 | rates[i].flags |= IEEE80211_TX_RC_USE_SHORT_PREAMBLE; | |
524 | ||
525 | /* set up G protection */ | |
526 | if (!(rates[i].flags & IEEE80211_TX_RC_USE_RTS_CTS) && | |
527 | info->control.use_cts_prot && | |
528 | rate->flags & IEEE80211_RATE_ERP_G) | |
529 | rates[i].flags |= IEEE80211_TX_RC_USE_CTS_PROTECT; | |
530 | } | |
531 | } | |
532 | ||
533 | ||
534 | static void rate_control_fill_sta_table(struct ieee80211_sta *sta, | |
535 | struct ieee80211_tx_info *info, | |
536 | struct ieee80211_tx_rate *rates, | |
537 | int max_rates) | |
538 | { | |
539 | struct ieee80211_sta_rates *ratetbl = NULL; | |
540 | int i; | |
541 | ||
542 | if (sta && !info->control.skip_table) | |
543 | ratetbl = rcu_dereference(sta->rates); | |
544 | ||
545 | /* Fill remaining rate slots with data from the sta rate table. */ | |
546 | max_rates = min_t(int, max_rates, IEEE80211_TX_RATE_TABLE_SIZE); | |
547 | for (i = 0; i < max_rates; i++) { | |
548 | if (i < ARRAY_SIZE(info->control.rates) && | |
549 | info->control.rates[i].idx >= 0 && | |
550 | info->control.rates[i].count) { | |
551 | if (rates != info->control.rates) | |
552 | rates[i] = info->control.rates[i]; | |
553 | } else if (ratetbl) { | |
554 | rates[i].idx = ratetbl->rate[i].idx; | |
555 | rates[i].flags = ratetbl->rate[i].flags; | |
556 | if (info->control.use_rts) | |
557 | rates[i].count = ratetbl->rate[i].count_rts; | |
558 | else if (info->control.use_cts_prot) | |
559 | rates[i].count = ratetbl->rate[i].count_cts; | |
560 | else | |
561 | rates[i].count = ratetbl->rate[i].count; | |
562 | } else { | |
563 | rates[i].idx = -1; | |
564 | rates[i].count = 0; | |
565 | } | |
566 | ||
567 | if (rates[i].idx < 0 || !rates[i].count) | |
568 | break; | |
569 | } | |
570 | } | |
571 | ||
572 | static void rate_control_apply_mask(struct ieee80211_sub_if_data *sdata, | |
573 | struct ieee80211_sta *sta, | |
574 | struct ieee80211_supported_band *sband, | |
575 | struct ieee80211_tx_info *info, | |
576 | struct ieee80211_tx_rate *rates, | |
577 | int max_rates) | |
578 | { | |
579 | enum nl80211_chan_width chan_width; | |
580 | u8 mcs_mask[IEEE80211_HT_MCS_MASK_LEN]; | |
581 | bool has_mcs_mask; | |
582 | u32 mask; | |
2103dec1 | 583 | u32 rate_flags; |
0d528d85 FF |
584 | int i; |
585 | ||
586 | /* | |
587 | * Try to enforce the rateidx mask the user wanted. skip this if the | |
588 | * default mask (allow all rates) is used to save some processing for | |
589 | * the common case. | |
590 | */ | |
591 | mask = sdata->rc_rateidx_mask[info->band]; | |
592 | has_mcs_mask = sdata->rc_has_mcs_mask[info->band]; | |
2103dec1 SW |
593 | rate_flags = |
594 | ieee80211_chandef_rate_flags(&sdata->vif.bss_conf.chandef); | |
595 | for (i = 0; i < sband->n_bitrates; i++) | |
596 | if ((rate_flags & sband->bitrates[i].flags) != rate_flags) | |
597 | mask &= ~BIT(i); | |
598 | ||
0d528d85 FF |
599 | if (mask == (1 << sband->n_bitrates) - 1 && !has_mcs_mask) |
600 | return; | |
601 | ||
602 | if (has_mcs_mask) | |
603 | memcpy(mcs_mask, sdata->rc_rateidx_mcs_mask[info->band], | |
604 | sizeof(mcs_mask)); | |
605 | else | |
606 | memset(mcs_mask, 0xff, sizeof(mcs_mask)); | |
607 | ||
608 | if (sta) { | |
609 | /* Filter out rates that the STA does not support */ | |
610 | mask &= sta->supp_rates[info->band]; | |
611 | for (i = 0; i < sizeof(mcs_mask); i++) | |
612 | mcs_mask[i] &= sta->ht_cap.mcs.rx_mask[i]; | |
613 | } | |
614 | ||
615 | /* | |
616 | * Make sure the rate index selected for each TX rate is | |
617 | * included in the configured mask and change the rate indexes | |
618 | * if needed. | |
619 | */ | |
620 | chan_width = sdata->vif.bss_conf.chandef.width; | |
621 | for (i = 0; i < max_rates; i++) { | |
622 | /* Skip invalid rates */ | |
623 | if (rates[i].idx < 0) | |
624 | break; | |
625 | ||
795d855d | 626 | rate_idx_match_mask(&rates[i], sband, chan_width, mask, |
0d528d85 FF |
627 | mcs_mask); |
628 | } | |
629 | } | |
630 | ||
631 | void ieee80211_get_tx_rates(struct ieee80211_vif *vif, | |
632 | struct ieee80211_sta *sta, | |
633 | struct sk_buff *skb, | |
634 | struct ieee80211_tx_rate *dest, | |
635 | int max_rates) | |
636 | { | |
637 | struct ieee80211_sub_if_data *sdata; | |
638 | struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data; | |
639 | struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb); | |
640 | struct ieee80211_supported_band *sband; | |
641 | ||
642 | rate_control_fill_sta_table(sta, info, dest, max_rates); | |
643 | ||
644 | if (!vif) | |
645 | return; | |
646 | ||
647 | sdata = vif_to_sdata(vif); | |
648 | sband = sdata->local->hw.wiphy->bands[info->band]; | |
649 | ||
650 | if (ieee80211_is_data(hdr->frame_control)) | |
651 | rate_control_apply_mask(sdata, sta, sband, info, dest, max_rates); | |
652 | ||
653 | if (dest[0].idx < 0) | |
1d2d350b AO |
654 | __rate_control_send_low(&sdata->local->hw, sband, sta, info, |
655 | sdata->rc_rateidx_mask[info->band]); | |
0d528d85 FF |
656 | |
657 | if (sta) | |
658 | rate_fixup_ratelist(vif, sband, info, dest, max_rates); | |
659 | } | |
660 | EXPORT_SYMBOL(ieee80211_get_tx_rates); | |
661 | ||
4b7679a5 | 662 | void rate_control_get_rate(struct ieee80211_sub_if_data *sdata, |
e6a9854b JB |
663 | struct sta_info *sta, |
664 | struct ieee80211_tx_rate_control *txrc) | |
1abbe498 | 665 | { |
4b7679a5 JB |
666 | struct rate_control_ref *ref = sdata->local->rate_ctrl; |
667 | void *priv_sta = NULL; | |
668 | struct ieee80211_sta *ista = NULL; | |
e6a9854b | 669 | struct ieee80211_tx_info *info = IEEE80211_SKB_CB(txrc->skb); |
1abbe498 | 670 | int i; |
1abbe498 | 671 | |
e1936e94 | 672 | if (sta && test_sta_flag(sta, WLAN_STA_RATE_CONTROL)) { |
4b7679a5 JB |
673 | ista = &sta->sta; |
674 | priv_sta = sta->rate_ctrl_priv; | |
675 | } | |
676 | ||
e6a9854b JB |
677 | for (i = 0; i < IEEE80211_TX_MAX_RATES; i++) { |
678 | info->control.rates[i].idx = -1; | |
679 | info->control.rates[i].flags = 0; | |
8617b093 | 680 | info->control.rates[i].count = 0; |
e6a9854b JB |
681 | } |
682 | ||
5affcd6b JO |
683 | if (sdata->local->hw.flags & IEEE80211_HW_HAS_RATE_CONTROL) |
684 | return; | |
685 | ||
37eb0b16 | 686 | ref->ops->get_rate(ref->priv, ista, priv_sta, txrc); |
e6a9854b | 687 | |
0d528d85 FF |
688 | if (sdata->local->hw.flags & IEEE80211_HW_SUPPORTS_RC_TABLE) |
689 | return; | |
690 | ||
691 | ieee80211_get_tx_rates(&sdata->vif, ista, txrc->skb, | |
692 | info->control.rates, | |
693 | ARRAY_SIZE(info->control.rates)); | |
694 | } | |
e6a9854b | 695 | |
0d528d85 FF |
696 | int rate_control_set_rates(struct ieee80211_hw *hw, |
697 | struct ieee80211_sta *pubsta, | |
698 | struct ieee80211_sta_rates *rates) | |
699 | { | |
f815e2b3 | 700 | struct sta_info *sta = container_of(pubsta, struct sta_info, sta); |
f6b3d85f | 701 | struct ieee80211_sta_rates *old; |
0d528d85 | 702 | |
f6b3d85f FF |
703 | /* |
704 | * mac80211 guarantees that this function will not be called | |
705 | * concurrently, so the following RCU access is safe, even without | |
706 | * extra locking. This can not be checked easily, so we just set | |
707 | * the condition to true. | |
708 | */ | |
709 | old = rcu_dereference_protected(pubsta->rates, true); | |
0d528d85 FF |
710 | rcu_assign_pointer(pubsta->rates, rates); |
711 | if (old) | |
712 | kfree_rcu(old, rcu_head); | |
713 | ||
f815e2b3 JB |
714 | drv_sta_rate_tbl_update(hw_to_local(hw), sta->sdata, pubsta); |
715 | ||
0d528d85 | 716 | return 0; |
1abbe498 | 717 | } |
0d528d85 | 718 | EXPORT_SYMBOL(rate_control_set_rates); |
1abbe498 | 719 | |
ff688089 JB |
720 | int ieee80211_init_rate_ctrl_alg(struct ieee80211_local *local, |
721 | const char *name) | |
722 | { | |
889cbb91 | 723 | struct rate_control_ref *ref; |
ff688089 JB |
724 | |
725 | ASSERT_RTNL(); | |
af65cd96 | 726 | |
3b8d81e0 | 727 | if (local->open_count) |
ff688089 JB |
728 | return -EBUSY; |
729 | ||
af65cd96 JB |
730 | if (local->hw.flags & IEEE80211_HW_HAS_RATE_CONTROL) { |
731 | if (WARN_ON(!local->ops->set_rts_threshold)) | |
732 | return -EINVAL; | |
733 | return 0; | |
734 | } | |
735 | ||
ff688089 JB |
736 | ref = rate_control_alloc(name, local); |
737 | if (!ref) { | |
0fb9a9ec JP |
738 | wiphy_warn(local->hw.wiphy, |
739 | "Failed to select rate control algorithm\n"); | |
ff688089 JB |
740 | return -ENOENT; |
741 | } | |
742 | ||
889cbb91 | 743 | WARN_ON(local->rate_ctrl); |
ff688089 | 744 | local->rate_ctrl = ref; |
ff688089 | 745 | |
0fb9a9ec JP |
746 | wiphy_debug(local->hw.wiphy, "Selected rate control algorithm '%s'\n", |
747 | ref->ops->name); | |
ff688089 | 748 | |
ff688089 JB |
749 | return 0; |
750 | } | |
751 | ||
752 | void rate_control_deinitialize(struct ieee80211_local *local) | |
753 | { | |
754 | struct rate_control_ref *ref; | |
755 | ||
756 | ref = local->rate_ctrl; | |
af65cd96 JB |
757 | |
758 | if (!ref) | |
759 | return; | |
760 | ||
ff688089 | 761 | local->rate_ctrl = NULL; |
889cbb91 | 762 | rate_control_free(ref); |
ff688089 | 763 | } |
c21b39ac | 764 |