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 <linux/kref.h>
18 #include <net/mac80211.h>
19 #include "ieee80211_i.h"
20 #include "sta_info.h"
21 
22 struct rate_control_ref {
23 	struct ieee80211_local *local;
24 	struct rate_control_ops *ops;
25 	void *priv;
26 	struct kref kref;
27 };
28 
29 void rate_control_get_rate(struct ieee80211_sub_if_data *sdata,
30 			   struct sta_info *sta,
31 			   struct ieee80211_tx_rate_control *txrc);
32 struct rate_control_ref *rate_control_get(struct rate_control_ref *ref);
33 void rate_control_put(struct rate_control_ref *ref);
34 
rate_control_tx_status(struct ieee80211_local * local,struct ieee80211_supported_band * sband,struct sta_info * sta,struct sk_buff * skb)35 static inline void rate_control_tx_status(struct ieee80211_local *local,
36 					  struct ieee80211_supported_band *sband,
37 					  struct sta_info *sta,
38 					  struct sk_buff *skb)
39 {
40 	struct rate_control_ref *ref = local->rate_ctrl;
41 	struct ieee80211_sta *ista = &sta->sta;
42 	void *priv_sta = sta->rate_ctrl_priv;
43 
44 	if (!ref || !test_sta_flag(sta, WLAN_STA_RATE_CONTROL))
45 		return;
46 
47 	ref->ops->tx_status(ref->priv, sband, ista, priv_sta, skb);
48 }
49 
50 
rate_control_rate_init(struct sta_info * sta)51 static inline void rate_control_rate_init(struct sta_info *sta)
52 {
53 	struct ieee80211_local *local = sta->sdata->local;
54 	struct rate_control_ref *ref = sta->rate_ctrl;
55 	struct ieee80211_sta *ista = &sta->sta;
56 	void *priv_sta = sta->rate_ctrl_priv;
57 	struct ieee80211_supported_band *sband;
58 
59 	if (!ref)
60 		return;
61 
62 	sband = local->hw.wiphy->bands[local->hw.conf.channel->band];
63 
64 	ref->ops->rate_init(ref->priv, sband, ista, priv_sta);
65 	set_sta_flag(sta, WLAN_STA_RATE_CONTROL);
66 }
67 
rate_control_rate_update(struct ieee80211_local * local,struct ieee80211_supported_band * sband,struct sta_info * sta,u32 changed,enum nl80211_channel_type oper_chan_type)68 static inline void rate_control_rate_update(struct ieee80211_local *local,
69 				    struct ieee80211_supported_band *sband,
70 				    struct sta_info *sta, u32 changed,
71 				    enum nl80211_channel_type oper_chan_type)
72 {
73 	struct rate_control_ref *ref = local->rate_ctrl;
74 	struct ieee80211_sta *ista = &sta->sta;
75 	void *priv_sta = sta->rate_ctrl_priv;
76 
77 	if (ref && ref->ops->rate_update)
78 		ref->ops->rate_update(ref->priv, sband, ista,
79 				      priv_sta, changed, oper_chan_type);
80 }
81 
rate_control_alloc_sta(struct rate_control_ref * ref,struct ieee80211_sta * sta,gfp_t gfp)82 static inline void *rate_control_alloc_sta(struct rate_control_ref *ref,
83 					   struct ieee80211_sta *sta,
84 					   gfp_t gfp)
85 {
86 	return ref->ops->alloc_sta(ref->priv, sta, gfp);
87 }
88 
rate_control_free_sta(struct sta_info * sta)89 static inline void rate_control_free_sta(struct sta_info *sta)
90 {
91 	struct rate_control_ref *ref = sta->rate_ctrl;
92 	struct ieee80211_sta *ista = &sta->sta;
93 	void *priv_sta = sta->rate_ctrl_priv;
94 
95 	ref->ops->free_sta(ref->priv, ista, priv_sta);
96 }
97 
rate_control_add_sta_debugfs(struct sta_info * sta)98 static inline void rate_control_add_sta_debugfs(struct sta_info *sta)
99 {
100 #ifdef CONFIG_MAC80211_DEBUGFS
101 	struct rate_control_ref *ref = sta->rate_ctrl;
102 	if (ref && sta->debugfs.dir && ref->ops->add_sta_debugfs)
103 		ref->ops->add_sta_debugfs(ref->priv, sta->rate_ctrl_priv,
104 					  sta->debugfs.dir);
105 #endif
106 }
107 
rate_control_remove_sta_debugfs(struct sta_info * sta)108 static inline void rate_control_remove_sta_debugfs(struct sta_info *sta)
109 {
110 #ifdef CONFIG_MAC80211_DEBUGFS
111 	struct rate_control_ref *ref = sta->rate_ctrl;
112 	if (ref && ref->ops->remove_sta_debugfs)
113 		ref->ops->remove_sta_debugfs(ref->priv, sta->rate_ctrl_priv);
114 #endif
115 }
116 
117 /* Get a reference to the rate control algorithm. If `name' is NULL, get the
118  * first available algorithm. */
119 int ieee80211_init_rate_ctrl_alg(struct ieee80211_local *local,
120 				 const char *name);
121 void rate_control_deinitialize(struct ieee80211_local *local);
122 
123 
124 /* Rate control algorithms */
125 #ifdef CONFIG_MAC80211_RC_PID
126 extern int rc80211_pid_init(void);
127 extern void rc80211_pid_exit(void);
128 #else
rc80211_pid_init(void)129 static inline int rc80211_pid_init(void)
130 {
131 	return 0;
132 }
rc80211_pid_exit(void)133 static inline void rc80211_pid_exit(void)
134 {
135 }
136 #endif
137 
138 #ifdef CONFIG_MAC80211_RC_MINSTREL
139 extern int rc80211_minstrel_init(void);
140 extern void rc80211_minstrel_exit(void);
141 #else
rc80211_minstrel_init(void)142 static inline int rc80211_minstrel_init(void)
143 {
144 	return 0;
145 }
rc80211_minstrel_exit(void)146 static inline void rc80211_minstrel_exit(void)
147 {
148 }
149 #endif
150 
151 #ifdef CONFIG_MAC80211_RC_MINSTREL_HT
152 extern int rc80211_minstrel_ht_init(void);
153 extern void rc80211_minstrel_ht_exit(void);
154 #else
rc80211_minstrel_ht_init(void)155 static inline int rc80211_minstrel_ht_init(void)
156 {
157 	return 0;
158 }
rc80211_minstrel_ht_exit(void)159 static inline void rc80211_minstrel_ht_exit(void)
160 {
161 }
162 #endif
163 
164 
165 #endif /* IEEE80211_RATE_H */
166