1 /*-
2  * Copyright (c) 2010 Rui Paulo <rpaulo@FreeBSD.org>
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
15  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
16  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
17  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
18  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
19  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
20  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
21  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
23  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24  *
25  * $FreeBSD$
26  */
27 
28 enum ieee80211_ratealgs {
29 	IEEE80211_RATECTL_AMRR		= 0,
30 	IEEE80211_RATECTL_RSSADAPT	= 1,
31 	IEEE80211_RATECTL_ONOE		= 2,
32 	IEEE80211_RATECTL_SAMPLE	= 3,
33 	IEEE80211_RATECTL_NONE		= 4,
34 	IEEE80211_RATECTL_MAX
35 };
36 
37 #define	IEEE80211_RATECTL_TX_SUCCESS	1
38 #define	IEEE80211_RATECTL_TX_FAILURE	0
39 
40 struct ieee80211_ratectl {
41 	const char *ir_name;
42 	int	(*ir_attach)(const struct ieee80211vap *);
43 	void	(*ir_detach)(const struct ieee80211vap *);
44 	void	(*ir_init)(struct ieee80211vap *);
45 	void	(*ir_deinit)(struct ieee80211vap *);
46 	void	(*ir_node_init)(struct ieee80211_node *);
47 	void	(*ir_node_deinit)(struct ieee80211_node *);
48 	int	(*ir_rate)(struct ieee80211_node *, void *, uint32_t);
49 	void	(*ir_tx_complete)(const struct ieee80211vap *,
50 	    			  const struct ieee80211_node *, int,
51 	    			  void *, void *);
52 	void	(*ir_tx_update)(const struct ieee80211vap *,
53 	    			const struct ieee80211_node *,
54 	    			void *, void *, void *);
55 	void	(*ir_setinterval)(const struct ieee80211vap *, int);
56 	void	(*ir_node_stats)(struct ieee80211_node *ni, struct sbuf *s);
57 };
58 
59 void	ieee80211_ratectl_register(int, const struct ieee80211_ratectl *);
60 void	ieee80211_ratectl_unregister(int);
61 void	ieee80211_ratectl_init(struct ieee80211vap *);
62 void	ieee80211_ratectl_set(struct ieee80211vap *, int);
63 
64 #ifdef MALLOC_DECLARE
65 MALLOC_DECLARE(M_80211_RATECTL);
66 #endif
67 
68 static __inline void
69 ieee80211_ratectl_deinit(struct ieee80211vap *vap)
70 {
71 	vap->iv_rate->ir_deinit(vap);
72 }
73 
74 static __inline void
75 ieee80211_ratectl_node_init(struct ieee80211_node *ni)
76 {
77 	const struct ieee80211vap *vap = ni->ni_vap;
78 
79 	vap->iv_rate->ir_node_init(ni);
80 }
81 
82 static __inline void
83 ieee80211_ratectl_node_deinit(struct ieee80211_node *ni)
84 {
85 	const struct ieee80211vap *vap = ni->ni_vap;
86 
87 	vap->iv_rate->ir_node_deinit(ni);
88 }
89 
90 static __inline int
91 ieee80211_ratectl_rate(struct ieee80211_node *ni, void *arg, uint32_t iarg)
92 {
93 	const struct ieee80211vap *vap = ni->ni_vap;
94 
95 	return vap->iv_rate->ir_rate(ni, arg, iarg);
96 }
97 
98 static __inline void
99 ieee80211_ratectl_tx_complete(const struct ieee80211vap *vap,
100     const struct ieee80211_node *ni, int status, void *arg1, void *arg2)
101 {
102 	vap->iv_rate->ir_tx_complete(vap, ni, status, arg1, arg2);
103 }
104 
105 static __inline void
106 ieee80211_ratectl_tx_update(const struct ieee80211vap *vap,
107     const struct ieee80211_node *ni, void *arg1, void *arg2, void *arg3)
108 {
109 	if (vap->iv_rate->ir_tx_update == NULL)
110 		return;
111 	vap->iv_rate->ir_tx_update(vap, ni, arg1, arg2, arg3);
112 }
113 
114 static __inline void
115 ieee80211_ratectl_setinterval(const struct ieee80211vap *vap, int msecs)
116 {
117 	if (vap->iv_rate->ir_setinterval == NULL)
118 		return;
119 	vap->iv_rate->ir_setinterval(vap, msecs);
120 }
121 
122 static __inline void
123 ieee80211_ratectl_node_stats(struct ieee80211_node *ni, struct sbuf *s)
124 {
125 	const struct ieee80211vap *vap = ni->ni_vap;
126 
127 	if (vap->iv_rate->ir_node_stats == NULL)
128 		return;
129 	vap->iv_rate->ir_node_stats(ni, s);
130 }
131