[linux-2.6-block.git] / net / mac80211 / rate.h

/*
 * Copyright 2002-2005, Instant802 Networks, Inc.
 * Copyright 2005, Devicescape Software, Inc.
 * Copyright (c) 2006 Jiri Benc <jbenc@suse.cz>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 */

#ifndef IEEE80211_RATE_H
#define IEEE80211_RATE_H

#include <linux/netdevice.h>
#include <linux/skbuff.h>
#include <linux/types.h>
#include <net/mac80211.h>
#include "ieee80211_i.h"
#include "sta_info.h"
#include "driver-ops.h"

struct rate_control_ref {
	struct ieee80211_local *local;
	const struct rate_control_ops *ops;
	void *priv;
};

void rate_control_get_rate(struct ieee80211_sub_if_data *sdata,
			   struct sta_info *sta,
			   struct ieee80211_tx_rate_control *txrc);

static inline void rate_control_tx_status(struct ieee80211_local *local,
					  struct ieee80211_supported_band *sband,
					  struct sta_info *sta,
					  struct sk_buff *skb)
{
	struct rate_control_ref *ref = local->rate_ctrl;
	struct ieee80211_sta *ista = &sta->sta;
	void *priv_sta = sta->rate_ctrl_priv;
	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);

	if (!ref || !test_sta_flag(sta, WLAN_STA_RATE_CONTROL))
		return;

	spin_lock_bh(&sta->rate_ctrl_lock);
	if (ref->ops->tx_status)
		ref->ops->tx_status(ref->priv, sband, ista, priv_sta, skb);
	else
		ref->ops->tx_status_noskb(ref->priv, sband, ista, priv_sta, info);
	spin_unlock_bh(&sta->rate_ctrl_lock);
}

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)
{
	struct rate_control_ref *ref = local->rate_ctrl;
	struct ieee80211_sta *ista = &sta->sta;
	void *priv_sta = sta->rate_ctrl_priv;

	if (!ref || !test_sta_flag(sta, WLAN_STA_RATE_CONTROL))
		return;

	if (WARN_ON_ONCE(!ref->ops->tx_status_noskb))
		return;

	spin_lock_bh(&sta->rate_ctrl_lock);
	ref->ops->tx_status_noskb(ref->priv, sband, ista, priv_sta, info);
	spin_unlock_bh(&sta->rate_ctrl_lock);
}

static inline void rate_control_rate_init(struct sta_info *sta)
{
	struct ieee80211_local *local = sta->sdata->local;
	struct rate_control_ref *ref = sta->rate_ctrl;
	struct ieee80211_sta *ista = &sta->sta;
	void *priv_sta = sta->rate_ctrl_priv;
	struct ieee80211_supported_band *sband;
	struct ieee80211_chanctx_conf *chanctx_conf;

	ieee80211_sta_set_rx_nss(sta);

	if (!ref)
		return;

	rcu_read_lock();

	chanctx_conf = rcu_dereference(sta->sdata->vif.chanctx_conf);
	if (WARN_ON(!chanctx_conf)) {
		rcu_read_unlock();
		return;
	}

	sband = local->hw.wiphy->bands[chanctx_conf->def.chan->band];

	spin_lock_bh(&sta->rate_ctrl_lock);
	ref->ops->rate_init(ref->priv, sband, &chanctx_conf->def, ista,
			    priv_sta);
	spin_unlock_bh(&sta->rate_ctrl_lock);
	rcu_read_unlock();
	set_sta_flag(sta, WLAN_STA_RATE_CONTROL);
}

static inline void rate_control_rate_update(struct ieee80211_local *local,
				    struct ieee80211_supported_band *sband,
				    struct sta_info *sta, u32 changed)
{
	struct rate_control_ref *ref = local->rate_ctrl;
	struct ieee80211_sta *ista = &sta->sta;
	void *priv_sta = sta->rate_ctrl_priv;
	struct ieee80211_chanctx_conf *chanctx_conf;

	if (ref && ref->ops->rate_update) {
		rcu_read_lock();

		chanctx_conf = rcu_dereference(sta->sdata->vif.chanctx_conf);
		if (WARN_ON(!chanctx_conf)) {
			rcu_read_unlock();
			return;
		}

		spin_lock_bh(&sta->rate_ctrl_lock);
		ref->ops->rate_update(ref->priv, sband, &chanctx_conf->def,
				      ista, priv_sta, changed);
		spin_unlock_bh(&sta->rate_ctrl_lock);
		rcu_read_unlock();
	}
	drv_sta_rc_update(local, sta->sdata, &sta->sta, changed);
}

static inline void *rate_control_alloc_sta(struct rate_control_ref *ref,
					   struct sta_info *sta, gfp_t gfp)
{
	spin_lock_init(&sta->rate_ctrl_lock);
	return ref->ops->alloc_sta(ref->priv, &sta->sta, gfp);
}

static inline void rate_control_free_sta(struct sta_info *sta)
{
	struct rate_control_ref *ref = sta->rate_ctrl;
	struct ieee80211_sta *ista = &sta->sta;
	void *priv_sta = sta->rate_ctrl_priv;

	ref->ops->free_sta(ref->priv, ista, priv_sta);
}

static inline void rate_control_add_sta_debugfs(struct sta_info *sta)
{
#ifdef CONFIG_MAC80211_DEBUGFS
	struct rate_control_ref *ref = sta->rate_ctrl;
	if (ref && sta->debugfs.dir && ref->ops->add_sta_debugfs)
		ref->ops->add_sta_debugfs(ref->priv, sta->rate_ctrl_priv,
					  sta->debugfs.dir);
#endif
}

static inline void rate_control_remove_sta_debugfs(struct sta_info *sta)
{
#ifdef CONFIG_MAC80211_DEBUGFS
	struct rate_control_ref *ref = sta->rate_ctrl;
	if (ref && ref->ops->remove_sta_debugfs)
		ref->ops->remove_sta_debugfs(ref->priv, sta->rate_ctrl_priv);
#endif
}

/* Get a reference to the rate control algorithm. If `name' is NULL, get the
 * first available algorithm. */
int ieee80211_init_rate_ctrl_alg(struct ieee80211_local *local,
				 const char *name);
void rate_control_deinitialize(struct ieee80211_local *local);


/* Rate control algorithms */
#ifdef CONFIG_MAC80211_RC_MINSTREL
int rc80211_minstrel_init(void);
void rc80211_minstrel_exit(void);
#else
static inline int rc80211_minstrel_init(void)
{
	return 0;
}
static inline void rc80211_minstrel_exit(void)
{
}
#endif

#ifdef CONFIG_MAC80211_RC_MINSTREL_HT
int rc80211_minstrel_ht_init(void);
void rc80211_minstrel_ht_exit(void);
#else
static inline int rc80211_minstrel_ht_init(void)
{
	return 0;
}
static inline void rc80211_minstrel_ht_exit(void)
{
}
#endif


#endif /* IEEE80211_RATE_H */
Commit	Line	Data
f0706e82 JB	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"
8f727ef3	20	#include "driver-ops.h"
f0706e82	21
f0706e82	22	struct rate_control_ref {
4b7679a5	23	struct ieee80211_local *local;
631ad703	24	const struct rate_control_ops *ops;
f0706e82	25	void *priv;
f0706e82 JB	26	};
f0706e82 JB	27
4b7679a5	28	void rate_control_get_rate(struct ieee80211_sub_if_data *sdata,
e6a9854b JB	29	struct sta_info *sta,
e6a9854b JB	30	struct ieee80211_tx_rate_control *txrc);
f0706e82	31
4b7679a5 JB	32	static inline void rate_control_tx_status(struct ieee80211_local *local,
	33	struct ieee80211_supported_band *sband,
	34	struct sta_info *sta,
e039fa4a	35	struct sk_buff *skb)
f0706e82 JB	36	{
f0706e82 JB	37	struct rate_control_ref *ref = local->rate_ctrl;
4b7679a5 JB	38	struct ieee80211_sta *ista = &sta->sta;
4b7679a5 JB	39	void *priv_sta = sta->rate_ctrl_priv;
f684565e	40	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
3c384053	41
2cfc6fc5	42	if (!ref \|\| !test_sta_flag(sta, WLAN_STA_RATE_CONTROL))
3c384053 V	43	return;
3c384053 V	44
35c347ac	45	spin_lock_bh(&sta->rate_ctrl_lock);
f684565e FF	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);
35c347ac	50	spin_unlock_bh(&sta->rate_ctrl_lock);
f0706e82 JB	51	}
f0706e82 JB	52
f027c2ac FF	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
35c347ac	69	spin_lock_bh(&sta->rate_ctrl_lock);
f027c2ac	70	ref->ops->tx_status_noskb(ref->priv, sband, ista, priv_sta, info);
35c347ac	71	spin_unlock_bh(&sta->rate_ctrl_lock);
f027c2ac	72	}
f0706e82	73
4b7679a5	74	static inline void rate_control_rate_init(struct sta_info *sta)
f0706e82	75	{
4b7679a5	76	struct ieee80211_local *local = sta->sdata->local;
f0706e82	77	struct rate_control_ref *ref = sta->rate_ctrl;
4b7679a5 JB	78	struct ieee80211_sta *ista = &sta->sta;
	79	void *priv_sta = sta->rate_ctrl_priv;
	80	struct ieee80211_supported_band *sband;
55de908a	81	struct ieee80211_chanctx_conf *chanctx_conf;
4b7679a5	82
ddcc347b MK	83	ieee80211_sta_set_rx_nss(sta);
ddcc347b MK	84
af65cd96 JB	85	if (!ref)
	86	return;
	87
55de908a JB	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
4bf88530	96	sband = local->hw.wiphy->bands[chanctx_conf->def.chan->band];
4b7679a5	97
35c347ac	98	spin_lock_bh(&sta->rate_ctrl_lock);
3de805cf SW	99	ref->ops->rate_init(ref->priv, sband, &chanctx_conf->def, ista,
3de805cf SW	100	priv_sta);
35c347ac	101	spin_unlock_bh(&sta->rate_ctrl_lock);
3de805cf	102	rcu_read_unlock();
e1936e94	103	set_sta_flag(sta, WLAN_STA_RATE_CONTROL);
f0706e82 JB	104	}
f0706e82 JB	105
81cb7623 S	106	static inline void rate_control_rate_update(struct ieee80211_local *local,
81cb7623 S	107	struct ieee80211_supported_band *sband,
64f68e5d	108	struct sta_info *sta, u32 changed)
81cb7623 S	109	{
	110	struct rate_control_ref *ref = local->rate_ctrl;
	111	struct ieee80211_sta *ista = &sta->sta;
	112	void *priv_sta = sta->rate_ctrl_priv;
3de805cf SW	113	struct ieee80211_chanctx_conf *chanctx_conf;
	114
	115	if (ref && ref->ops->rate_update) {
	116	rcu_read_lock();
81cb7623	117
3de805cf SW	118	chanctx_conf = rcu_dereference(sta->sdata->vif.chanctx_conf);
	119	if (WARN_ON(!chanctx_conf)) {
	120	rcu_read_unlock();
	121	return;
	122	}
	123
35c347ac	124	spin_lock_bh(&sta->rate_ctrl_lock);
3de805cf SW	125	ref->ops->rate_update(ref->priv, sband, &chanctx_conf->def,
3de805cf SW	126	ista, priv_sta, changed);
35c347ac	127	spin_unlock_bh(&sta->rate_ctrl_lock);
3de805cf SW	128	rcu_read_unlock();
3de805cf SW	129	}
8f727ef3	130	drv_sta_rc_update(local, sta->sdata, &sta->sta, changed);
81cb7623	131	}
f0706e82	132
f0706e82	133	static inline void rate_control_alloc_sta(struct rate_control_ref ref,
35c347ac	134	struct sta_info *sta, gfp_t gfp)
f0706e82	135	{
35c347ac JB	136	spin_lock_init(&sta->rate_ctrl_lock);
35c347ac JB	137	return ref->ops->alloc_sta(ref->priv, &sta->sta, gfp);
f0706e82 JB	138	}
f0706e82 JB	139
4b7679a5	140	static inline void rate_control_free_sta(struct sta_info *sta)
f0706e82	141	{
4b7679a5 JB	142	struct rate_control_ref *ref = sta->rate_ctrl;
	143	struct ieee80211_sta *ista = &sta->sta;
	144	void *priv_sta = sta->rate_ctrl_priv;
	145
	146	ref->ops->free_sta(ref->priv, ista, priv_sta);
f0706e82 JB	147	}
f0706e82 JB	148
e9f207f0 JB	149	static inline void rate_control_add_sta_debugfs(struct sta_info *sta)
	150	{
	151	#ifdef CONFIG_MAC80211_DEBUGFS
	152	struct rate_control_ref *ref = sta->rate_ctrl;
af65cd96	153	if (ref && sta->debugfs.dir && ref->ops->add_sta_debugfs)
e9f207f0 JB	154	ref->ops->add_sta_debugfs(ref->priv, sta->rate_ctrl_priv,
	155	sta->debugfs.dir);
	156	#endif
	157	}
	158
	159	static inline void rate_control_remove_sta_debugfs(struct sta_info *sta)
	160	{
	161	#ifdef CONFIG_MAC80211_DEBUGFS
	162	struct rate_control_ref *ref = sta->rate_ctrl;
af65cd96	163	if (ref && ref->ops->remove_sta_debugfs)
e9f207f0 JB	164	ref->ops->remove_sta_debugfs(ref->priv, sta->rate_ctrl_priv);
	165	#endif
	166	}
	167
209c671d AS	168	/* Get a reference to the rate control algorithm. If `name' is NULL, get the
209c671d AS	169	* first available algorithm. */
ff688089 JB	170	int ieee80211_init_rate_ctrl_alg(struct ieee80211_local *local,
	171	const char *name);
	172	void rate_control_deinitialize(struct ieee80211_local *local);
	173
4b475898 JB	174
4b475898 JB	175	/* Rate control algorithms */
cccf129f	176	#ifdef CONFIG_MAC80211_RC_MINSTREL
c1b1203d JP	177	int rc80211_minstrel_init(void);
c1b1203d JP	178	void rc80211_minstrel_exit(void);
cccf129f FF	179	#else
	180	static inline int rc80211_minstrel_init(void)
	181	{
	182	return 0;
	183	}
	184	static inline void rc80211_minstrel_exit(void)
	185	{
	186	}
	187	#endif
	188
ec8aa669	189	#ifdef CONFIG_MAC80211_RC_MINSTREL_HT
c1b1203d JP	190	int rc80211_minstrel_ht_init(void);
c1b1203d JP	191	void rc80211_minstrel_ht_exit(void);
ec8aa669 FF	192	#else
	193	static inline int rc80211_minstrel_ht_init(void)
	194	{
	195	return 0;
	196	}
	197	static inline void rc80211_minstrel_ht_exit(void)
	198	{
	199	}
	200	#endif
	201
cccf129f	202
f0706e82	203	#endif /* IEEE80211_RATE_H */