]> git.proxmox.com Git - mirror_ubuntu-artful-kernel.git/blob - net/batman-adv/gateway_common.c
Merge git://git.infradead.org/intel-iommu
[mirror_ubuntu-artful-kernel.git] / net / batman-adv / gateway_common.c
1 /* Copyright (C) 2009-2015 B.A.T.M.A.N. contributors:
2 *
3 * Marek Lindner
4 *
5 * This program is free software; you can redistribute it and/or
6 * modify it under the terms of version 2 of the GNU General Public
7 * License as published by the Free Software Foundation.
8 *
9 * This program is distributed in the hope that it will be useful, but
10 * WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, see <http://www.gnu.org/licenses/>.
16 */
17
18 #include "gateway_common.h"
19 #include "main.h"
20
21 #include <linux/atomic.h>
22 #include <linux/errno.h>
23 #include <linux/byteorder/generic.h>
24 #include <linux/kernel.h>
25 #include <linux/math64.h>
26 #include <linux/netdevice.h>
27 #include <linux/stddef.h>
28 #include <linux/string.h>
29
30 #include "gateway_client.h"
31 #include "packet.h"
32
33 /**
34 * batadv_parse_gw_bandwidth - parse supplied string buffer to extract download
35 * and upload bandwidth information
36 * @net_dev: the soft interface net device
37 * @buff: string buffer to parse
38 * @down: pointer holding the returned download bandwidth information
39 * @up: pointer holding the returned upload bandwidth information
40 *
41 * Returns false on parse error and true otherwise.
42 */
43 static bool batadv_parse_gw_bandwidth(struct net_device *net_dev, char *buff,
44 u32 *down, u32 *up)
45 {
46 enum batadv_bandwidth_units bw_unit_type = BATADV_BW_UNIT_KBIT;
47 char *slash_ptr, *tmp_ptr;
48 u64 ldown, lup;
49 int ret;
50
51 slash_ptr = strchr(buff, '/');
52 if (slash_ptr)
53 *slash_ptr = 0;
54
55 if (strlen(buff) > 4) {
56 tmp_ptr = buff + strlen(buff) - 4;
57
58 if (strncasecmp(tmp_ptr, "mbit", 4) == 0)
59 bw_unit_type = BATADV_BW_UNIT_MBIT;
60
61 if ((strncasecmp(tmp_ptr, "kbit", 4) == 0) ||
62 (bw_unit_type == BATADV_BW_UNIT_MBIT))
63 *tmp_ptr = '\0';
64 }
65
66 ret = kstrtou64(buff, 10, &ldown);
67 if (ret) {
68 batadv_err(net_dev,
69 "Download speed of gateway mode invalid: %s\n",
70 buff);
71 return false;
72 }
73
74 switch (bw_unit_type) {
75 case BATADV_BW_UNIT_MBIT:
76 /* prevent overflow */
77 if (U64_MAX / 10 < ldown) {
78 batadv_err(net_dev,
79 "Download speed of gateway mode too large: %s\n",
80 buff);
81 return false;
82 }
83
84 ldown *= 10;
85 break;
86 case BATADV_BW_UNIT_KBIT:
87 default:
88 ldown = div_u64(ldown, 100);
89 break;
90 }
91
92 if (U32_MAX < ldown) {
93 batadv_err(net_dev,
94 "Download speed of gateway mode too large: %s\n",
95 buff);
96 return false;
97 }
98
99 *down = ldown;
100
101 /* we also got some upload info */
102 if (slash_ptr) {
103 bw_unit_type = BATADV_BW_UNIT_KBIT;
104
105 if (strlen(slash_ptr + 1) > 4) {
106 tmp_ptr = slash_ptr + 1 - 4 + strlen(slash_ptr + 1);
107
108 if (strncasecmp(tmp_ptr, "mbit", 4) == 0)
109 bw_unit_type = BATADV_BW_UNIT_MBIT;
110
111 if ((strncasecmp(tmp_ptr, "kbit", 4) == 0) ||
112 (bw_unit_type == BATADV_BW_UNIT_MBIT))
113 *tmp_ptr = '\0';
114 }
115
116 ret = kstrtou64(slash_ptr + 1, 10, &lup);
117 if (ret) {
118 batadv_err(net_dev,
119 "Upload speed of gateway mode invalid: %s\n",
120 slash_ptr + 1);
121 return false;
122 }
123
124 switch (bw_unit_type) {
125 case BATADV_BW_UNIT_MBIT:
126 /* prevent overflow */
127 if (U64_MAX / 10 < lup) {
128 batadv_err(net_dev,
129 "Upload speed of gateway mode too large: %s\n",
130 slash_ptr + 1);
131 return false;
132 }
133
134 lup *= 10;
135 break;
136 case BATADV_BW_UNIT_KBIT:
137 default:
138 lup = div_u64(lup, 100);
139 break;
140 }
141
142 if (U32_MAX < lup) {
143 batadv_err(net_dev,
144 "Upload speed of gateway mode too large: %s\n",
145 slash_ptr + 1);
146 return false;
147 }
148
149 *up = lup;
150 }
151
152 return true;
153 }
154
155 /**
156 * batadv_gw_tvlv_container_update - update the gw tvlv container after gateway
157 * setting change
158 * @bat_priv: the bat priv with all the soft interface information
159 */
160 void batadv_gw_tvlv_container_update(struct batadv_priv *bat_priv)
161 {
162 struct batadv_tvlv_gateway_data gw;
163 u32 down, up;
164 char gw_mode;
165
166 gw_mode = atomic_read(&bat_priv->gw_mode);
167
168 switch (gw_mode) {
169 case BATADV_GW_MODE_OFF:
170 case BATADV_GW_MODE_CLIENT:
171 batadv_tvlv_container_unregister(bat_priv, BATADV_TVLV_GW, 1);
172 break;
173 case BATADV_GW_MODE_SERVER:
174 down = atomic_read(&bat_priv->gw.bandwidth_down);
175 up = atomic_read(&bat_priv->gw.bandwidth_up);
176 gw.bandwidth_down = htonl(down);
177 gw.bandwidth_up = htonl(up);
178 batadv_tvlv_container_register(bat_priv, BATADV_TVLV_GW, 1,
179 &gw, sizeof(gw));
180 break;
181 }
182 }
183
184 ssize_t batadv_gw_bandwidth_set(struct net_device *net_dev, char *buff,
185 size_t count)
186 {
187 struct batadv_priv *bat_priv = netdev_priv(net_dev);
188 u32 down_curr;
189 u32 up_curr;
190 u32 down_new = 0;
191 u32 up_new = 0;
192 bool ret;
193
194 down_curr = (unsigned int)atomic_read(&bat_priv->gw.bandwidth_down);
195 up_curr = (unsigned int)atomic_read(&bat_priv->gw.bandwidth_up);
196
197 ret = batadv_parse_gw_bandwidth(net_dev, buff, &down_new, &up_new);
198 if (!ret)
199 return -EINVAL;
200
201 if (!down_new)
202 down_new = 1;
203
204 if (!up_new)
205 up_new = down_new / 5;
206
207 if (!up_new)
208 up_new = 1;
209
210 if ((down_curr == down_new) && (up_curr == up_new))
211 return count;
212
213 batadv_gw_reselect(bat_priv);
214 batadv_info(net_dev,
215 "Changing gateway bandwidth from: '%u.%u/%u.%u MBit' to: '%u.%u/%u.%u MBit'\n",
216 down_curr / 10, down_curr % 10, up_curr / 10, up_curr % 10,
217 down_new / 10, down_new % 10, up_new / 10, up_new % 10);
218
219 atomic_set(&bat_priv->gw.bandwidth_down, down_new);
220 atomic_set(&bat_priv->gw.bandwidth_up, up_new);
221 batadv_gw_tvlv_container_update(bat_priv);
222
223 return count;
224 }
225
226 /**
227 * batadv_gw_tvlv_ogm_handler_v1 - process incoming gateway tvlv container
228 * @bat_priv: the bat priv with all the soft interface information
229 * @orig: the orig_node of the ogm
230 * @flags: flags indicating the tvlv state (see batadv_tvlv_handler_flags)
231 * @tvlv_value: tvlv buffer containing the gateway data
232 * @tvlv_value_len: tvlv buffer length
233 */
234 static void batadv_gw_tvlv_ogm_handler_v1(struct batadv_priv *bat_priv,
235 struct batadv_orig_node *orig,
236 u8 flags,
237 void *tvlv_value, u16 tvlv_value_len)
238 {
239 struct batadv_tvlv_gateway_data gateway, *gateway_ptr;
240
241 /* only fetch the tvlv value if the handler wasn't called via the
242 * CIFNOTFND flag and if there is data to fetch
243 */
244 if ((flags & BATADV_TVLV_HANDLER_OGM_CIFNOTFND) ||
245 (tvlv_value_len < sizeof(gateway))) {
246 gateway.bandwidth_down = 0;
247 gateway.bandwidth_up = 0;
248 } else {
249 gateway_ptr = tvlv_value;
250 gateway.bandwidth_down = gateway_ptr->bandwidth_down;
251 gateway.bandwidth_up = gateway_ptr->bandwidth_up;
252 if ((gateway.bandwidth_down == 0) ||
253 (gateway.bandwidth_up == 0)) {
254 gateway.bandwidth_down = 0;
255 gateway.bandwidth_up = 0;
256 }
257 }
258
259 batadv_gw_node_update(bat_priv, orig, &gateway);
260
261 /* restart gateway selection if fast or late switching was enabled */
262 if ((gateway.bandwidth_down != 0) &&
263 (atomic_read(&bat_priv->gw_mode) == BATADV_GW_MODE_CLIENT) &&
264 (atomic_read(&bat_priv->gw_sel_class) > 2))
265 batadv_gw_check_election(bat_priv, orig);
266 }
267
268 /**
269 * batadv_gw_init - initialise the gateway handling internals
270 * @bat_priv: the bat priv with all the soft interface information
271 */
272 void batadv_gw_init(struct batadv_priv *bat_priv)
273 {
274 batadv_tvlv_handler_register(bat_priv, batadv_gw_tvlv_ogm_handler_v1,
275 NULL, BATADV_TVLV_GW, 1,
276 BATADV_TVLV_HANDLER_OGM_CIFNOTFND);
277 }
278
279 /**
280 * batadv_gw_free - free the gateway handling internals
281 * @bat_priv: the bat priv with all the soft interface information
282 */
283 void batadv_gw_free(struct batadv_priv *bat_priv)
284 {
285 batadv_tvlv_container_unregister(bat_priv, BATADV_TVLV_GW, 1);
286 batadv_tvlv_handler_unregister(bat_priv, BATADV_TVLV_GW, 1);
287 }