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 22struct rate_control_ref { 23 struct ieee80211_local *local; 24 const struct rate_control_ops *ops; 25 void *priv; 26}; 27 28void rate_control_get_rate(struct ieee80211_sub_if_data *sdata, 29 struct sta_info *sta, 30 struct ieee80211_tx_rate_control *txrc); 31 32static 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 48static 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 78static 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 103static 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 110static 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 119static 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 129static 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. */ 140int ieee80211_init_rate_ctrl_alg(struct ieee80211_local *local, 141 const char *name); 142void rate_control_deinitialize(struct ieee80211_local *local); 143 144 145/* Rate control algorithms */ 146#ifdef CONFIG_MAC80211_RC_MINSTREL 147int rc80211_minstrel_init(void); 148void rc80211_minstrel_exit(void); 149#else 150static inline int rc80211_minstrel_init(void) 151{ 152 return 0; 153} 154static inline void rc80211_minstrel_exit(void) 155{ 156} 157#endif 158 159#ifdef CONFIG_MAC80211_RC_MINSTREL_HT 160int rc80211_minstrel_ht_init(void); 161void rc80211_minstrel_ht_exit(void); 162#else 163static inline int rc80211_minstrel_ht_init(void) 164{ 165 return 0; 166} 167static inline void rc80211_minstrel_ht_exit(void) 168{ 169} 170#endif 171 172 173#endif /* IEEE80211_RATE_H */ 174