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
rate_control_tx_status(struct ieee80211_local * local,struct ieee80211_supported_band * sband,struct sta_info * sta,struct sk_buff * skb)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
rate_control_tx_status_noskb(struct ieee80211_local * local,struct ieee80211_supported_band * sband,struct sta_info * sta,struct ieee80211_tx_info * info)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
rate_control_alloc_sta(struct rate_control_ref * ref,struct sta_info * sta,gfp_t gfp)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
rate_control_free_sta(struct sta_info * sta)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
rate_control_add_sta_debugfs(struct sta_info * sta)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
rate_control_remove_sta_debugfs(struct sta_info * sta)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
rc80211_minstrel_init(void)126 static inline int rc80211_minstrel_init(void)
127 {
128 return 0;
129 }
rc80211_minstrel_exit(void)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
rc80211_minstrel_ht_init(void)139 static inline int rc80211_minstrel_ht_init(void)
140 {
141 return 0;
142 }
rc80211_minstrel_ht_exit(void)143 static inline void rc80211_minstrel_ht_exit(void)
144 {
145 }
146 #endif
147
148
149 #endif /* IEEE80211_RATE_H */
150