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 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 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 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 if (!ref) 58 return; 59 60 rcu_read_lock(); 61 62 chanctx_conf = rcu_dereference(sta->sdata->vif.chanctx_conf); 63 if (WARN_ON(!chanctx_conf)) { 64 rcu_read_unlock(); 65 return; 66 } 67 68 sband = local->hw.wiphy->bands[chanctx_conf->def.chan->band]; 69 rcu_read_unlock(); 70 71 ieee80211_sta_set_rx_nss(sta); 72 73 ref->ops->rate_init(ref->priv, sband, ista, priv_sta); 74 set_sta_flag(sta, WLAN_STA_RATE_CONTROL); 75 } 76 77 static inline void rate_control_rate_update(struct ieee80211_local *local, 78 struct ieee80211_supported_band *sband, 79 struct sta_info *sta, u32 changed) 80 { 81 struct rate_control_ref *ref = local->rate_ctrl; 82 struct ieee80211_sta *ista = &sta->sta; 83 void *priv_sta = sta->rate_ctrl_priv; 84 85 if (ref && ref->ops->rate_update) 86 ref->ops->rate_update(ref->priv, sband, ista, 87 priv_sta, changed); 88 drv_sta_rc_update(local, sta->sdata, &sta->sta, changed); 89 } 90 91 static inline void *rate_control_alloc_sta(struct rate_control_ref *ref, 92 struct ieee80211_sta *sta, 93 gfp_t gfp) 94 { 95 return ref->ops->alloc_sta(ref->priv, sta, gfp); 96 } 97 98 static inline void rate_control_free_sta(struct sta_info *sta) 99 { 100 struct rate_control_ref *ref = sta->rate_ctrl; 101 struct ieee80211_sta *ista = &sta->sta; 102 void *priv_sta = sta->rate_ctrl_priv; 103 104 ref->ops->free_sta(ref->priv, ista, priv_sta); 105 } 106 107 static inline void rate_control_add_sta_debugfs(struct sta_info *sta) 108 { 109 #ifdef CONFIG_MAC80211_DEBUGFS 110 struct rate_control_ref *ref = sta->rate_ctrl; 111 if (ref && sta->debugfs.dir && ref->ops->add_sta_debugfs) 112 ref->ops->add_sta_debugfs(ref->priv, sta->rate_ctrl_priv, 113 sta->debugfs.dir); 114 #endif 115 } 116 117 static inline void rate_control_remove_sta_debugfs(struct sta_info *sta) 118 { 119 #ifdef CONFIG_MAC80211_DEBUGFS 120 struct rate_control_ref *ref = sta->rate_ctrl; 121 if (ref && ref->ops->remove_sta_debugfs) 122 ref->ops->remove_sta_debugfs(ref->priv, sta->rate_ctrl_priv); 123 #endif 124 } 125 126 /* Get a reference to the rate control algorithm. If `name' is NULL, get the 127 * first available algorithm. */ 128 int ieee80211_init_rate_ctrl_alg(struct ieee80211_local *local, 129 const char *name); 130 void rate_control_deinitialize(struct ieee80211_local *local); 131 132 133 /* Rate control algorithms */ 134 #ifdef CONFIG_MAC80211_RC_PID 135 extern int rc80211_pid_init(void); 136 extern void rc80211_pid_exit(void); 137 #else 138 static inline int rc80211_pid_init(void) 139 { 140 return 0; 141 } 142 static inline void rc80211_pid_exit(void) 143 { 144 } 145 #endif 146 147 #ifdef CONFIG_MAC80211_RC_MINSTREL 148 extern int rc80211_minstrel_init(void); 149 extern void rc80211_minstrel_exit(void); 150 #else 151 static inline int rc80211_minstrel_init(void) 152 { 153 return 0; 154 } 155 static inline void rc80211_minstrel_exit(void) 156 { 157 } 158 #endif 159 160 #ifdef CONFIG_MAC80211_RC_MINSTREL_HT 161 extern int rc80211_minstrel_ht_init(void); 162 extern void rc80211_minstrel_ht_exit(void); 163 #else 164 static inline int rc80211_minstrel_ht_init(void) 165 { 166 return 0; 167 } 168 static inline void rc80211_minstrel_ht_exit(void) 169 { 170 } 171 #endif 172 173 174 #endif /* IEEE80211_RATE_H */ 175