1 // SPDX-License-Identifier: GPL-2.0 2 /* Copyright (C) B.A.T.M.A.N. contributors: 3 * 4 * Marek Lindner 5 */ 6 7 #include "gateway_common.h" 8 #include "main.h" 9 10 #include <linux/atomic.h> 11 #include <linux/byteorder/generic.h> 12 #include <linux/errno.h> 13 #include <linux/kstrtox.h> 14 #include <linux/limits.h> 15 #include <linux/math64.h> 16 #include <linux/netdevice.h> 17 #include <linux/stddef.h> 18 #include <linux/string.h> 19 #include <uapi/linux/batadv_packet.h> 20 #include <uapi/linux/batman_adv.h> 21 22 #include "gateway_client.h" 23 #include "log.h" 24 #include "tvlv.h" 25 26 /** 27 * batadv_parse_throughput() - parse supplied string buffer to extract 28 * throughput information 29 * @net_dev: the soft interface net device 30 * @buff: string buffer to parse 31 * @description: text shown when throughput string cannot be parsed 32 * @throughput: pointer holding the returned throughput information 33 * 34 * Return: false on parse error and true otherwise. 35 */ 36 bool batadv_parse_throughput(struct net_device *net_dev, char *buff, 37 const char *description, u32 *throughput) 38 { 39 enum batadv_bandwidth_units bw_unit_type = BATADV_BW_UNIT_KBIT; 40 u64 lthroughput; 41 char *tmp_ptr; 42 int ret; 43 44 if (strlen(buff) > 4) { 45 tmp_ptr = buff + strlen(buff) - 4; 46 47 if (strncasecmp(tmp_ptr, "mbit", 4) == 0) 48 bw_unit_type = BATADV_BW_UNIT_MBIT; 49 50 if (strncasecmp(tmp_ptr, "kbit", 4) == 0 || 51 bw_unit_type == BATADV_BW_UNIT_MBIT) 52 *tmp_ptr = '\0'; 53 } 54 55 ret = kstrtou64(buff, 10, <hroughput); 56 if (ret) { 57 batadv_err(net_dev, 58 "Invalid throughput speed for %s: %s\n", 59 description, buff); 60 return false; 61 } 62 63 switch (bw_unit_type) { 64 case BATADV_BW_UNIT_MBIT: 65 /* prevent overflow */ 66 if (U64_MAX / 10 < lthroughput) { 67 batadv_err(net_dev, 68 "Throughput speed for %s too large: %s\n", 69 description, buff); 70 return false; 71 } 72 73 lthroughput *= 10; 74 break; 75 case BATADV_BW_UNIT_KBIT: 76 default: 77 lthroughput = div_u64(lthroughput, 100); 78 break; 79 } 80 81 if (lthroughput > U32_MAX) { 82 batadv_err(net_dev, 83 "Throughput speed for %s too large: %s\n", 84 description, buff); 85 return false; 86 } 87 88 *throughput = lthroughput; 89 90 return true; 91 } 92 93 /** 94 * batadv_parse_gw_bandwidth() - parse supplied string buffer to extract 95 * download and upload bandwidth information 96 * @net_dev: the soft interface net device 97 * @buff: string buffer to parse 98 * @down: pointer holding the returned download bandwidth information 99 * @up: pointer holding the returned upload bandwidth information 100 * 101 * Return: false on parse error and true otherwise. 102 */ 103 static bool batadv_parse_gw_bandwidth(struct net_device *net_dev, char *buff, 104 u32 *down, u32 *up) 105 { 106 char *slash_ptr; 107 bool ret; 108 109 slash_ptr = strchr(buff, '/'); 110 if (slash_ptr) 111 *slash_ptr = 0; 112 113 ret = batadv_parse_throughput(net_dev, buff, "download gateway speed", 114 down); 115 if (!ret) 116 return false; 117 118 /* we also got some upload info */ 119 if (slash_ptr) { 120 ret = batadv_parse_throughput(net_dev, slash_ptr + 1, 121 "upload gateway speed", up); 122 if (!ret) 123 return false; 124 } 125 126 return true; 127 } 128 129 /** 130 * batadv_gw_tvlv_container_update() - update the gw tvlv container after 131 * gateway setting change 132 * @bat_priv: the bat priv with all the soft interface information 133 */ 134 void batadv_gw_tvlv_container_update(struct batadv_priv *bat_priv) 135 { 136 struct batadv_tvlv_gateway_data gw; 137 u32 down, up; 138 char gw_mode; 139 140 gw_mode = atomic_read(&bat_priv->gw.mode); 141 142 switch (gw_mode) { 143 case BATADV_GW_MODE_OFF: 144 case BATADV_GW_MODE_CLIENT: 145 batadv_tvlv_container_unregister(bat_priv, BATADV_TVLV_GW, 1); 146 break; 147 case BATADV_GW_MODE_SERVER: 148 down = atomic_read(&bat_priv->gw.bandwidth_down); 149 up = atomic_read(&bat_priv->gw.bandwidth_up); 150 gw.bandwidth_down = htonl(down); 151 gw.bandwidth_up = htonl(up); 152 batadv_tvlv_container_register(bat_priv, BATADV_TVLV_GW, 1, 153 &gw, sizeof(gw)); 154 break; 155 } 156 } 157 158 /** 159 * batadv_gw_bandwidth_set() - Parse and set download/upload gateway bandwidth 160 * from supplied string buffer 161 * @net_dev: netdev struct of the soft interface 162 * @buff: the buffer containing the user data 163 * @count: number of bytes in the buffer 164 * 165 * Return: 'count' on success or a negative error code in case of failure 166 */ 167 ssize_t batadv_gw_bandwidth_set(struct net_device *net_dev, char *buff, 168 size_t count) 169 { 170 struct batadv_priv *bat_priv = netdev_priv(net_dev); 171 u32 down_curr; 172 u32 up_curr; 173 u32 down_new = 0; 174 u32 up_new = 0; 175 bool ret; 176 177 down_curr = (unsigned int)atomic_read(&bat_priv->gw.bandwidth_down); 178 up_curr = (unsigned int)atomic_read(&bat_priv->gw.bandwidth_up); 179 180 ret = batadv_parse_gw_bandwidth(net_dev, buff, &down_new, &up_new); 181 if (!ret) 182 return -EINVAL; 183 184 if (!down_new) 185 down_new = 1; 186 187 if (!up_new) 188 up_new = down_new / 5; 189 190 if (!up_new) 191 up_new = 1; 192 193 if (down_curr == down_new && up_curr == up_new) 194 return count; 195 196 batadv_gw_reselect(bat_priv); 197 batadv_info(net_dev, 198 "Changing gateway bandwidth from: '%u.%u/%u.%u MBit' to: '%u.%u/%u.%u MBit'\n", 199 down_curr / 10, down_curr % 10, up_curr / 10, up_curr % 10, 200 down_new / 10, down_new % 10, up_new / 10, up_new % 10); 201 202 atomic_set(&bat_priv->gw.bandwidth_down, down_new); 203 atomic_set(&bat_priv->gw.bandwidth_up, up_new); 204 batadv_gw_tvlv_container_update(bat_priv); 205 206 return count; 207 } 208 209 /** 210 * batadv_gw_tvlv_ogm_handler_v1() - process incoming gateway tvlv container 211 * @bat_priv: the bat priv with all the soft interface information 212 * @orig: the orig_node of the ogm 213 * @flags: flags indicating the tvlv state (see batadv_tvlv_handler_flags) 214 * @tvlv_value: tvlv buffer containing the gateway data 215 * @tvlv_value_len: tvlv buffer length 216 */ 217 static void batadv_gw_tvlv_ogm_handler_v1(struct batadv_priv *bat_priv, 218 struct batadv_orig_node *orig, 219 u8 flags, 220 void *tvlv_value, u16 tvlv_value_len) 221 { 222 struct batadv_tvlv_gateway_data gateway, *gateway_ptr; 223 224 /* only fetch the tvlv value if the handler wasn't called via the 225 * CIFNOTFND flag and if there is data to fetch 226 */ 227 if (flags & BATADV_TVLV_HANDLER_OGM_CIFNOTFND || 228 tvlv_value_len < sizeof(gateway)) { 229 gateway.bandwidth_down = 0; 230 gateway.bandwidth_up = 0; 231 } else { 232 gateway_ptr = tvlv_value; 233 gateway.bandwidth_down = gateway_ptr->bandwidth_down; 234 gateway.bandwidth_up = gateway_ptr->bandwidth_up; 235 if (gateway.bandwidth_down == 0 || 236 gateway.bandwidth_up == 0) { 237 gateway.bandwidth_down = 0; 238 gateway.bandwidth_up = 0; 239 } 240 } 241 242 batadv_gw_node_update(bat_priv, orig, &gateway); 243 244 /* restart gateway selection */ 245 if (gateway.bandwidth_down != 0 && 246 atomic_read(&bat_priv->gw.mode) == BATADV_GW_MODE_CLIENT) 247 batadv_gw_check_election(bat_priv, orig); 248 } 249 250 /** 251 * batadv_gw_init() - initialise the gateway handling internals 252 * @bat_priv: the bat priv with all the soft interface information 253 */ 254 void batadv_gw_init(struct batadv_priv *bat_priv) 255 { 256 if (bat_priv->algo_ops->gw.init_sel_class) 257 bat_priv->algo_ops->gw.init_sel_class(bat_priv); 258 else 259 atomic_set(&bat_priv->gw.sel_class, 1); 260 261 batadv_tvlv_handler_register(bat_priv, batadv_gw_tvlv_ogm_handler_v1, 262 NULL, NULL, BATADV_TVLV_GW, 1, 263 BATADV_TVLV_HANDLER_OGM_CIFNOTFND); 264 } 265 266 /** 267 * batadv_gw_free() - free the gateway handling internals 268 * @bat_priv: the bat priv with all the soft interface information 269 */ 270 void batadv_gw_free(struct batadv_priv *bat_priv) 271 { 272 batadv_tvlv_container_unregister(bat_priv, BATADV_TVLV_GW, 1); 273 batadv_tvlv_handler_unregister(bat_priv, BATADV_TVLV_GW, 1); 274 } 275