1 /*
2 * Copyright 2002-2005, Instant802 Networks, Inc.
3 * Copyright 2005, 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 #ifndef IEEE80211_RATE_H
12 #define IEEE80211_RATE_H
13
14 #include <linux/netdevice.h>
15 #include <linux/skbuff.h>
16 #include <linux/types.h>
17 #include <net/mac80211.h>
18 #include "ieee80211_i.h"
19 #include "sta_info.h"
20 #include "driver-ops.h"
21
22 struct rate_control_ref {
23 struct ieee80211_local *local;
24 const struct rate_control_ops *ops;
25 void *priv;
26 };
27
28 void rate_control_get_rate(struct ieee80211_sub_if_data *sdata,
29 struct sta_info *sta,
30 struct ieee80211_tx_rate_control *txrc);
31
rate_control_tx_status(struct ieee80211_local * local,struct ieee80211_supported_band * sband,struct sta_info * sta,struct sk_buff * skb)32 static inline void rate_control_tx_status(struct ieee80211_local *local,
33 struct ieee80211_supported_band *sband,
34 struct sta_info *sta,
35 struct sk_buff *skb)
36 {
37 struct rate_control_ref *ref = local->rate_ctrl;
38 struct ieee80211_sta *ista = &sta->sta;
39 void *priv_sta = sta->rate_ctrl_priv;
40
41 if (!ref || !test_sta_flag(sta, WLAN_STA_RATE_CONTROL))
42 return;
43
44 ref->ops->tx_status(ref->priv, sband, ista, priv_sta, skb);
45 }
46
47
rate_control_rate_init(struct sta_info * sta)48 static inline void rate_control_rate_init(struct sta_info *sta)
49 {
50 struct ieee80211_local *local = sta->sdata->local;
51 struct rate_control_ref *ref = sta->rate_ctrl;
52 struct ieee80211_sta *ista = &sta->sta;
53 void *priv_sta = sta->rate_ctrl_priv;
54 struct ieee80211_supported_band *sband;
55 struct ieee80211_chanctx_conf *chanctx_conf;
56
57 ieee80211_sta_set_rx_nss(sta);
58
59 if (!ref)
60 return;
61
62 rcu_read_lock();
63
64 chanctx_conf = rcu_dereference(sta->sdata->vif.chanctx_conf);
65 if (WARN_ON(!chanctx_conf)) {
66 rcu_read_unlock();
67 return;
68 }
69
70 sband = local->hw.wiphy->bands[chanctx_conf->def.chan->band];
71
72 ref->ops->rate_init(ref->priv, sband, &chanctx_conf->def, ista,
73 priv_sta);
74 rcu_read_unlock();
75 set_sta_flag(sta, WLAN_STA_RATE_CONTROL);
76 }
77
rate_control_rate_update(struct ieee80211_local * local,struct ieee80211_supported_band * sband,struct sta_info * sta,u32 changed)78 static inline void rate_control_rate_update(struct ieee80211_local *local,
79 struct ieee80211_supported_band *sband,
80 struct sta_info *sta, u32 changed)
81 {
82 struct rate_control_ref *ref = local->rate_ctrl;
83 struct ieee80211_sta *ista = &sta->sta;
84 void *priv_sta = sta->rate_ctrl_priv;
85 struct ieee80211_chanctx_conf *chanctx_conf;
86
87 if (ref && ref->ops->rate_update) {
88 rcu_read_lock();
89
90 chanctx_conf = rcu_dereference(sta->sdata->vif.chanctx_conf);
91 if (WARN_ON(!chanctx_conf)) {
92 rcu_read_unlock();
93 return;
94 }
95
96 ref->ops->rate_update(ref->priv, sband, &chanctx_conf->def,
97 ista, priv_sta, changed);
98 rcu_read_unlock();
99 }
100 drv_sta_rc_update(local, sta->sdata, &sta->sta, changed);
101 }
102
rate_control_alloc_sta(struct rate_control_ref * ref,struct ieee80211_sta * sta,gfp_t gfp)103 static inline void *rate_control_alloc_sta(struct rate_control_ref *ref,
104 struct ieee80211_sta *sta,
105 gfp_t gfp)
106 {
107 return ref->ops->alloc_sta(ref->priv, sta, gfp);
108 }
109
rate_control_free_sta(struct sta_info * sta)110 static inline void rate_control_free_sta(struct sta_info *sta)
111 {
112 struct rate_control_ref *ref = sta->rate_ctrl;
113 struct ieee80211_sta *ista = &sta->sta;
114 void *priv_sta = sta->rate_ctrl_priv;
115
116 ref->ops->free_sta(ref->priv, ista, priv_sta);
117 }
118
rate_control_add_sta_debugfs(struct sta_info * sta)119 static inline void rate_control_add_sta_debugfs(struct sta_info *sta)
120 {
121 #ifdef CONFIG_MAC80211_DEBUGFS
122 struct rate_control_ref *ref = sta->rate_ctrl;
123 if (ref && sta->debugfs.dir && ref->ops->add_sta_debugfs)
124 ref->ops->add_sta_debugfs(ref->priv, sta->rate_ctrl_priv,
125 sta->debugfs.dir);
126 #endif
127 }
128
rate_control_remove_sta_debugfs(struct sta_info * sta)129 static inline void rate_control_remove_sta_debugfs(struct sta_info *sta)
130 {
131 #ifdef CONFIG_MAC80211_DEBUGFS
132 struct rate_control_ref *ref = sta->rate_ctrl;
133 if (ref && ref->ops->remove_sta_debugfs)
134 ref->ops->remove_sta_debugfs(ref->priv, sta->rate_ctrl_priv);
135 #endif
136 }
137
138 /* Get a reference to the rate control algorithm. If `name' is NULL, get the
139 * first available algorithm. */
140 int ieee80211_init_rate_ctrl_alg(struct ieee80211_local *local,
141 const char *name);
142 void rate_control_deinitialize(struct ieee80211_local *local);
143
144
145 /* Rate control algorithms */
146 #ifdef CONFIG_MAC80211_RC_MINSTREL
147 int rc80211_minstrel_init(void);
148 void rc80211_minstrel_exit(void);
149 #else
rc80211_minstrel_init(void)150 static inline int rc80211_minstrel_init(void)
151 {
152 return 0;
153 }
rc80211_minstrel_exit(void)154 static inline void rc80211_minstrel_exit(void)
155 {
156 }
157 #endif
158
159 #ifdef CONFIG_MAC80211_RC_MINSTREL_HT
160 int rc80211_minstrel_ht_init(void);
161 void rc80211_minstrel_ht_exit(void);
162 #else
rc80211_minstrel_ht_init(void)163 static inline int rc80211_minstrel_ht_init(void)
164 {
165 return 0;
166 }
rc80211_minstrel_ht_exit(void)167 static inline void rc80211_minstrel_ht_exit(void)
168 {
169 }
170 #endif
171
172
173 #endif /* IEEE80211_RATE_H */
174