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 void rate_control_rate_init(struct sta_info *sta); 75 void rate_control_rate_update(struct ieee80211_local *local, 76 struct ieee80211_supported_band *sband, 77 struct sta_info *sta, u32 changed); 78 79 static inline void *rate_control_alloc_sta(struct rate_control_ref *ref, 80 struct sta_info *sta, gfp_t gfp) 81 { 82 spin_lock_init(&sta->rate_ctrl_lock); 83 return ref->ops->alloc_sta(ref->priv, &sta->sta, gfp); 84 } 85 86 static inline void rate_control_free_sta(struct sta_info *sta) 87 { 88 struct rate_control_ref *ref = sta->rate_ctrl; 89 struct ieee80211_sta *ista = &sta->sta; 90 void *priv_sta = sta->rate_ctrl_priv; 91 92 ref->ops->free_sta(ref->priv, ista, priv_sta); 93 } 94 95 static inline void rate_control_add_sta_debugfs(struct sta_info *sta) 96 { 97 #ifdef CONFIG_MAC80211_DEBUGFS 98 struct rate_control_ref *ref = sta->rate_ctrl; 99 if (ref && sta->debugfs.dir && ref->ops->add_sta_debugfs) 100 ref->ops->add_sta_debugfs(ref->priv, sta->rate_ctrl_priv, 101 sta->debugfs.dir); 102 #endif 103 } 104 105 static inline void rate_control_remove_sta_debugfs(struct sta_info *sta) 106 { 107 #ifdef CONFIG_MAC80211_DEBUGFS 108 struct rate_control_ref *ref = sta->rate_ctrl; 109 if (ref && ref->ops->remove_sta_debugfs) 110 ref->ops->remove_sta_debugfs(ref->priv, sta->rate_ctrl_priv); 111 #endif 112 } 113 114 /* Get a reference to the rate control algorithm. If `name' is NULL, get the 115 * first available algorithm. */ 116 int ieee80211_init_rate_ctrl_alg(struct ieee80211_local *local, 117 const char *name); 118 void rate_control_deinitialize(struct ieee80211_local *local); 119 120 121 /* Rate control algorithms */ 122 #ifdef CONFIG_MAC80211_RC_MINSTREL 123 int rc80211_minstrel_init(void); 124 void rc80211_minstrel_exit(void); 125 #else 126 static inline int rc80211_minstrel_init(void) 127 { 128 return 0; 129 } 130 static inline void rc80211_minstrel_exit(void) 131 { 132 } 133 #endif 134 135 #ifdef CONFIG_MAC80211_RC_MINSTREL_HT 136 int rc80211_minstrel_ht_init(void); 137 void rc80211_minstrel_ht_exit(void); 138 #else 139 static inline int rc80211_minstrel_ht_init(void) 140 { 141 return 0; 142 } 143 static inline void rc80211_minstrel_ht_exit(void) 144 { 145 } 146 #endif 147 148 149 #endif /* IEEE80211_RATE_H */ 150