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 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 struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb); 41 42 if (!ref || !test_sta_flag(sta, WLAN_STA_RATE_CONTROL)) 43 return; 44 45 spin_lock_bh(&sta->rate_ctrl_lock); 46 if (ref->ops->tx_status) 47 ref->ops->tx_status(ref->priv, sband, ista, priv_sta, skb); 48 else 49 ref->ops->tx_status_noskb(ref->priv, sband, ista, priv_sta, info); 50 spin_unlock_bh(&sta->rate_ctrl_lock); 51 } 52 53 static inline void 54 rate_control_tx_status_noskb(struct ieee80211_local *local, 55 struct ieee80211_supported_band *sband, 56 struct sta_info *sta, 57 struct ieee80211_tx_info *info) 58 { 59 struct rate_control_ref *ref = local->rate_ctrl; 60 struct ieee80211_sta *ista = &sta->sta; 61 void *priv_sta = sta->rate_ctrl_priv; 62 63 if (!ref || !test_sta_flag(sta, WLAN_STA_RATE_CONTROL)) 64 return; 65 66 if (WARN_ON_ONCE(!ref->ops->tx_status_noskb)) 67 return; 68 69 spin_lock_bh(&sta->rate_ctrl_lock); 70 ref->ops->tx_status_noskb(ref->priv, sband, ista, priv_sta, info); 71 spin_unlock_bh(&sta->rate_ctrl_lock); 72 } 73 74 static inline void rate_control_rate_init(struct sta_info *sta) 75 { 76 struct ieee80211_local *local = sta->sdata->local; 77 struct rate_control_ref *ref = sta->rate_ctrl; 78 struct ieee80211_sta *ista = &sta->sta; 79 void *priv_sta = sta->rate_ctrl_priv; 80 struct ieee80211_supported_band *sband; 81 struct ieee80211_chanctx_conf *chanctx_conf; 82 83 ieee80211_sta_set_rx_nss(sta); 84 85 if (!ref) 86 return; 87 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 sband = local->hw.wiphy->bands[chanctx_conf->def.chan->band]; 97 98 spin_lock_bh(&sta->rate_ctrl_lock); 99 ref->ops->rate_init(ref->priv, sband, &chanctx_conf->def, ista, 100 priv_sta); 101 spin_unlock_bh(&sta->rate_ctrl_lock); 102 rcu_read_unlock(); 103 set_sta_flag(sta, WLAN_STA_RATE_CONTROL); 104 } 105 106 static inline void rate_control_rate_update(struct ieee80211_local *local, 107 struct ieee80211_supported_band *sband, 108 struct sta_info *sta, u32 changed) 109 { 110 struct rate_control_ref *ref = local->rate_ctrl; 111 struct ieee80211_sta *ista = &sta->sta; 112 void *priv_sta = sta->rate_ctrl_priv; 113 struct ieee80211_chanctx_conf *chanctx_conf; 114 115 if (ref && ref->ops->rate_update) { 116 rcu_read_lock(); 117 118 chanctx_conf = rcu_dereference(sta->sdata->vif.chanctx_conf); 119 if (WARN_ON(!chanctx_conf)) { 120 rcu_read_unlock(); 121 return; 122 } 123 124 spin_lock_bh(&sta->rate_ctrl_lock); 125 ref->ops->rate_update(ref->priv, sband, &chanctx_conf->def, 126 ista, priv_sta, changed); 127 spin_unlock_bh(&sta->rate_ctrl_lock); 128 rcu_read_unlock(); 129 } 130 drv_sta_rc_update(local, sta->sdata, &sta->sta, changed); 131 } 132 133 static inline void *rate_control_alloc_sta(struct rate_control_ref *ref, 134 struct sta_info *sta, gfp_t gfp) 135 { 136 spin_lock_init(&sta->rate_ctrl_lock); 137 return ref->ops->alloc_sta(ref->priv, &sta->sta, gfp); 138 } 139 140 static inline void rate_control_free_sta(struct sta_info *sta) 141 { 142 struct rate_control_ref *ref = sta->rate_ctrl; 143 struct ieee80211_sta *ista = &sta->sta; 144 void *priv_sta = sta->rate_ctrl_priv; 145 146 ref->ops->free_sta(ref->priv, ista, priv_sta); 147 } 148 149 static inline void rate_control_add_sta_debugfs(struct sta_info *sta) 150 { 151 #ifdef CONFIG_MAC80211_DEBUGFS 152 struct rate_control_ref *ref = sta->rate_ctrl; 153 if (ref && sta->debugfs.dir && ref->ops->add_sta_debugfs) 154 ref->ops->add_sta_debugfs(ref->priv, sta->rate_ctrl_priv, 155 sta->debugfs.dir); 156 #endif 157 } 158 159 static inline void rate_control_remove_sta_debugfs(struct sta_info *sta) 160 { 161 #ifdef CONFIG_MAC80211_DEBUGFS 162 struct rate_control_ref *ref = sta->rate_ctrl; 163 if (ref && ref->ops->remove_sta_debugfs) 164 ref->ops->remove_sta_debugfs(ref->priv, sta->rate_ctrl_priv); 165 #endif 166 } 167 168 /* Get a reference to the rate control algorithm. If `name' is NULL, get the 169 * first available algorithm. */ 170 int ieee80211_init_rate_ctrl_alg(struct ieee80211_local *local, 171 const char *name); 172 void rate_control_deinitialize(struct ieee80211_local *local); 173 174 175 /* Rate control algorithms */ 176 #ifdef CONFIG_MAC80211_RC_MINSTREL 177 int rc80211_minstrel_init(void); 178 void rc80211_minstrel_exit(void); 179 #else 180 static inline int rc80211_minstrel_init(void) 181 { 182 return 0; 183 } 184 static inline void rc80211_minstrel_exit(void) 185 { 186 } 187 #endif 188 189 #ifdef CONFIG_MAC80211_RC_MINSTREL_HT 190 int rc80211_minstrel_ht_init(void); 191 void rc80211_minstrel_ht_exit(void); 192 #else 193 static inline int rc80211_minstrel_ht_init(void) 194 { 195 return 0; 196 } 197 static inline void rc80211_minstrel_ht_exit(void) 198 { 199 } 200 #endif 201 202 203 #endif /* IEEE80211_RATE_H */ 204