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