| 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 */ |