[linux-2.6-block.git] / net / batman-adv / gateway_common.c

/* Copyright (C) 2009-2015 B.A.T.M.A.N. contributors:
 *
 * Marek Lindner
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of version 2 of the GNU General Public
 * License as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, see <http://www.gnu.org/licenses/>.
 */

#include "gateway_common.h"
#include "main.h"

#include <linux/atomic.h>
#include <linux/byteorder/generic.h>
#include <linux/kernel.h>
#include <linux/netdevice.h>
#include <linux/stddef.h>
#include <linux/string.h>

#include "gateway_client.h"
#include "packet.h"

/**
 * batadv_parse_gw_bandwidth - parse supplied string buffer to extract download
 *  and upload bandwidth information
 * @net_dev: the soft interface net device
 * @buff: string buffer to parse
 * @down: pointer holding the returned download bandwidth information
 * @up: pointer holding the returned upload bandwidth information
 *
 * Returns false on parse error and true otherwise.
 */
static bool batadv_parse_gw_bandwidth(struct net_device *net_dev, char *buff,
				      uint32_t *down, uint32_t *up)
{
	enum batadv_bandwidth_units bw_unit_type = BATADV_BW_UNIT_KBIT;
	char *slash_ptr, *tmp_ptr;
	long ldown, lup;
	int ret;

	slash_ptr = strchr(buff, '/');
	if (slash_ptr)
		*slash_ptr = 0;

	if (strlen(buff) > 4) {
		tmp_ptr = buff + strlen(buff) - 4;

		if (strncasecmp(tmp_ptr, "mbit", 4) == 0)
			bw_unit_type = BATADV_BW_UNIT_MBIT;

		if ((strncasecmp(tmp_ptr, "kbit", 4) == 0) ||
		    (bw_unit_type == BATADV_BW_UNIT_MBIT))
			*tmp_ptr = '\0';
	}

	ret = kstrtol(buff, 10, &ldown);
	if (ret) {
		batadv_err(net_dev,
			   "Download speed of gateway mode invalid: %s\n",
			   buff);
		return false;
	}

	switch (bw_unit_type) {
	case BATADV_BW_UNIT_MBIT:
		*down = ldown * 10;
		break;
	case BATADV_BW_UNIT_KBIT:
	default:
		*down = ldown / 100;
		break;
	}

	/* we also got some upload info */
	if (slash_ptr) {
		bw_unit_type = BATADV_BW_UNIT_KBIT;

		if (strlen(slash_ptr + 1) > 4) {
			tmp_ptr = slash_ptr + 1 - 4 + strlen(slash_ptr + 1);

			if (strncasecmp(tmp_ptr, "mbit", 4) == 0)
				bw_unit_type = BATADV_BW_UNIT_MBIT;

			if ((strncasecmp(tmp_ptr, "kbit", 4) == 0) ||
			    (bw_unit_type == BATADV_BW_UNIT_MBIT))
				*tmp_ptr = '\0';
		}

		ret = kstrtol(slash_ptr + 1, 10, &lup);
		if (ret) {
			batadv_err(net_dev,
				   "Upload speed of gateway mode invalid: %s\n",
				   slash_ptr + 1);
			return false;
		}

		switch (bw_unit_type) {
		case BATADV_BW_UNIT_MBIT:
			*up = lup * 10;
			break;
		case BATADV_BW_UNIT_KBIT:
		default:
			*up = lup / 100;
			break;
		}
	}

	return true;
}

/**
 * batadv_gw_tvlv_container_update - update the gw tvlv container after gateway
 *  setting change
 * @bat_priv: the bat priv with all the soft interface information
 */
void batadv_gw_tvlv_container_update(struct batadv_priv *bat_priv)
{
	struct batadv_tvlv_gateway_data gw;
	uint32_t down, up;
	char gw_mode;

	gw_mode = atomic_read(&bat_priv->gw_mode);

	switch (gw_mode) {
	case BATADV_GW_MODE_OFF:
	case BATADV_GW_MODE_CLIENT:
		batadv_tvlv_container_unregister(bat_priv, BATADV_TVLV_GW, 1);
		break;
	case BATADV_GW_MODE_SERVER:
		down = atomic_read(&bat_priv->gw.bandwidth_down);
		up = atomic_read(&bat_priv->gw.bandwidth_up);
		gw.bandwidth_down = htonl(down);
		gw.bandwidth_up = htonl(up);
		batadv_tvlv_container_register(bat_priv, BATADV_TVLV_GW, 1,
					       &gw, sizeof(gw));
		break;
	}
}

ssize_t batadv_gw_bandwidth_set(struct net_device *net_dev, char *buff,
				size_t count)
{
	struct batadv_priv *bat_priv = netdev_priv(net_dev);
	uint32_t down_curr, up_curr, down_new = 0, up_new = 0;
	bool ret;

	down_curr = (unsigned int)atomic_read(&bat_priv->gw.bandwidth_down);
	up_curr = (unsigned int)atomic_read(&bat_priv->gw.bandwidth_up);

	ret = batadv_parse_gw_bandwidth(net_dev, buff, &down_new, &up_new);
	if (!ret)
		goto end;

	if (!down_new)
		down_new = 1;

	if (!up_new)
		up_new = down_new / 5;

	if (!up_new)
		up_new = 1;

	if ((down_curr == down_new) && (up_curr == up_new))
		return count;

	batadv_gw_reselect(bat_priv);
	batadv_info(net_dev,
		    "Changing gateway bandwidth from: '%u.%u/%u.%u MBit' to: '%u.%u/%u.%u MBit'\n",
		    down_curr / 10, down_curr % 10, up_curr / 10, up_curr % 10,
		    down_new / 10, down_new % 10, up_new / 10, up_new % 10);

	atomic_set(&bat_priv->gw.bandwidth_down, down_new);
	atomic_set(&bat_priv->gw.bandwidth_up, up_new);
	batadv_gw_tvlv_container_update(bat_priv);

end:
	return count;
}

/**
 * batadv_gw_tvlv_ogm_handler_v1 - process incoming gateway tvlv container
 * @bat_priv: the bat priv with all the soft interface information
 * @orig: the orig_node of the ogm
 * @flags: flags indicating the tvlv state (see batadv_tvlv_handler_flags)
 * @tvlv_value: tvlv buffer containing the gateway data
 * @tvlv_value_len: tvlv buffer length
 */
static void batadv_gw_tvlv_ogm_handler_v1(struct batadv_priv *bat_priv,
					  struct batadv_orig_node *orig,
					  uint8_t flags,
					  void *tvlv_value,
					  uint16_t tvlv_value_len)
{
	struct batadv_tvlv_gateway_data gateway, *gateway_ptr;

	/* only fetch the tvlv value if the handler wasn't called via the
	 * CIFNOTFND flag and if there is data to fetch
	 */
	if ((flags & BATADV_TVLV_HANDLER_OGM_CIFNOTFND) ||
	    (tvlv_value_len < sizeof(gateway))) {
		gateway.bandwidth_down = 0;
		gateway.bandwidth_up = 0;
	} else {
		gateway_ptr = tvlv_value;
		gateway.bandwidth_down = gateway_ptr->bandwidth_down;
		gateway.bandwidth_up = gateway_ptr->bandwidth_up;
		if ((gateway.bandwidth_down == 0) ||
		    (gateway.bandwidth_up == 0)) {
			gateway.bandwidth_down = 0;
			gateway.bandwidth_up = 0;
		}
	}

	batadv_gw_node_update(bat_priv, orig, &gateway);

	/* restart gateway selection if fast or late switching was enabled */
	if ((gateway.bandwidth_down != 0) &&
	    (atomic_read(&bat_priv->gw_mode) == BATADV_GW_MODE_CLIENT) &&
	    (atomic_read(&bat_priv->gw_sel_class) > 2))
		batadv_gw_check_election(bat_priv, orig);
}

/**
 * batadv_gw_init - initialise the gateway handling internals
 * @bat_priv: the bat priv with all the soft interface information
 */
void batadv_gw_init(struct batadv_priv *bat_priv)
{
	batadv_tvlv_handler_register(bat_priv, batadv_gw_tvlv_ogm_handler_v1,
				     NULL, BATADV_TVLV_GW, 1,
				     BATADV_TVLV_HANDLER_OGM_CIFNOTFND);
}

/**
 * batadv_gw_free - free the gateway handling internals
 * @bat_priv: the bat priv with all the soft interface information
 */
void batadv_gw_free(struct batadv_priv *bat_priv)
{
	batadv_tvlv_container_unregister(bat_priv, BATADV_TVLV_GW, 1);
	batadv_tvlv_handler_unregister(bat_priv, BATADV_TVLV_GW, 1);
}
Commit	Line	Data
9f6446c7	1	/* Copyright (C) 2009-2015 B.A.T.M.A.N. contributors:
c6c8fea2 SE	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
ebf38fb7	15	* along with this program; if not, see <http://www.gnu.org/licenses/>.
c6c8fea2 SE	16	*/
c6c8fea2 SE	17
c6c8fea2	18	#include "gateway_common.h"
1e2c2a4f SE	19	#include "main.h"
	20
	21	#include <linux/atomic.h>
	22	#include <linux/byteorder/generic.h>
	23	#include <linux/kernel.h>
	24	#include <linux/netdevice.h>
	25	#include <linux/stddef.h>
	26	#include <linux/string.h>
	27
c6c8fea2	28	#include "gateway_client.h"
1e2c2a4f	29	#include "packet.h"
c6c8fea2	30
414254e3 ML	31	/**
	32	* batadv_parse_gw_bandwidth - parse supplied string buffer to extract download
	33	* and upload bandwidth information
	34	* @net_dev: the soft interface net device
	35	* @buff: string buffer to parse
	36	* @down: pointer holding the returned download bandwidth information
	37	* @up: pointer holding the returned upload bandwidth information
	38	*
	39	* Returns false on parse error and true otherwise.
	40	*/
8e714a5d	41	static bool batadv_parse_gw_bandwidth(struct net_device net_dev, char buff,
414254e3	42	uint32_t down, uint32_t up)
c6c8fea2	43	{
414254e3	44	enum batadv_bandwidth_units bw_unit_type = BATADV_BW_UNIT_KBIT;
c6c8fea2	45	char slash_ptr, tmp_ptr;
37a4065e	46	long ldown, lup;
414254e3	47	int ret;
c6c8fea2 SE	48
	49	slash_ptr = strchr(buff, '/');
	50	if (slash_ptr)
	51	*slash_ptr = 0;
	52
	53	if (strlen(buff) > 4) {
	54	tmp_ptr = buff + strlen(buff) - 4;
	55
b7a8d756	56	if (strncasecmp(tmp_ptr, "mbit", 4) == 0)
414254e3	57	bw_unit_type = BATADV_BW_UNIT_MBIT;
c6c8fea2	58
b7a8d756	59	if ((strncasecmp(tmp_ptr, "kbit", 4) == 0) \|\|
414254e3	60	(bw_unit_type == BATADV_BW_UNIT_MBIT))
c6c8fea2 SE	61	*tmp_ptr = '\0';
	62	}
	63
25a92b13	64	ret = kstrtol(buff, 10, &ldown);
c6c8fea2	65	if (ret) {
3e34819e SE	66	batadv_err(net_dev,
	67	"Download speed of gateway mode invalid: %s\n",
	68	buff);
c6c8fea2 SE	69	return false;
	70	}
	71
414254e3 ML	72	switch (bw_unit_type) {
	73	case BATADV_BW_UNIT_MBIT:
	74	down = ldown 10;
	75	break;
	76	case BATADV_BW_UNIT_KBIT:
	77	default:
	78	*down = ldown / 100;
	79	break;
	80	}
c6c8fea2 SE	81
	82	/* we also got some upload info */
	83	if (slash_ptr) {
414254e3	84	bw_unit_type = BATADV_BW_UNIT_KBIT;
c6c8fea2 SE	85
	86	if (strlen(slash_ptr + 1) > 4) {
	87	tmp_ptr = slash_ptr + 1 - 4 + strlen(slash_ptr + 1);
	88
b7a8d756	89	if (strncasecmp(tmp_ptr, "mbit", 4) == 0)
414254e3	90	bw_unit_type = BATADV_BW_UNIT_MBIT;
c6c8fea2	91
b7a8d756	92	if ((strncasecmp(tmp_ptr, "kbit", 4) == 0) \|\|
414254e3	93	(bw_unit_type == BATADV_BW_UNIT_MBIT))
c6c8fea2 SE	94	*tmp_ptr = '\0';
	95	}
	96
25a92b13	97	ret = kstrtol(slash_ptr + 1, 10, &lup);
c6c8fea2	98	if (ret) {
3e34819e SE	99	batadv_err(net_dev,
	100	"Upload speed of gateway mode invalid: %s\n",
	101	slash_ptr + 1);
c6c8fea2 SE	102	return false;
	103	}
	104
414254e3 ML	105	switch (bw_unit_type) {
	106	case BATADV_BW_UNIT_MBIT:
	107	up = lup 10;
	108	break;
	109	case BATADV_BW_UNIT_KBIT:
	110	default:
	111	*up = lup / 100;
	112	break;
	113	}
c6c8fea2 SE	114	}
	115
	116	return true;
	117	}
	118
414254e3 ML	119	/**
	120	* batadv_gw_tvlv_container_update - update the gw tvlv container after gateway
	121	* setting change
	122	* @bat_priv: the bat priv with all the soft interface information
	123	*/
	124	void batadv_gw_tvlv_container_update(struct batadv_priv *bat_priv)
	125	{
	126	struct batadv_tvlv_gateway_data gw;
	127	uint32_t down, up;
	128	char gw_mode;
	129
	130	gw_mode = atomic_read(&bat_priv->gw_mode);
	131
	132	switch (gw_mode) {
	133	case BATADV_GW_MODE_OFF:
	134	case BATADV_GW_MODE_CLIENT:
	135	batadv_tvlv_container_unregister(bat_priv, BATADV_TVLV_GW, 1);
	136	break;
	137	case BATADV_GW_MODE_SERVER:
	138	down = atomic_read(&bat_priv->gw.bandwidth_down);
	139	up = atomic_read(&bat_priv->gw.bandwidth_up);
	140	gw.bandwidth_down = htonl(down);
	141	gw.bandwidth_up = htonl(up);
	142	batadv_tvlv_container_register(bat_priv, BATADV_TVLV_GW, 1,
	143	&gw, sizeof(gw));
	144	break;
	145	}
	146	}
	147
84d5e5e0 SE	148	ssize_t batadv_gw_bandwidth_set(struct net_device net_dev, char buff,
84d5e5e0 SE	149	size_t count)
c6c8fea2	150	{
56303d34	151	struct batadv_priv *bat_priv = netdev_priv(net_dev);
414254e3	152	uint32_t down_curr, up_curr, down_new = 0, up_new = 0;
c6c8fea2 SE	153	bool ret;
c6c8fea2 SE	154
414254e3 ML	155	down_curr = (unsigned int)atomic_read(&bat_priv->gw.bandwidth_down);
	156	up_curr = (unsigned int)atomic_read(&bat_priv->gw.bandwidth_up);
	157
	158	ret = batadv_parse_gw_bandwidth(net_dev, buff, &down_new, &up_new);
c6c8fea2 SE	159	if (!ret)
	160	goto end;
	161
414254e3 ML	162	if (!down_new)
414254e3 ML	163	down_new = 1;
c6c8fea2	164
414254e3 ML	165	if (!up_new)
414254e3 ML	166	up_new = down_new / 5;
c6c8fea2	167
414254e3 ML	168	if (!up_new)
414254e3 ML	169	up_new = 1;
c6c8fea2	170
414254e3	171	if ((down_curr == down_new) && (up_curr == up_new))
dafe94b2 ML	172	return count;
dafe94b2 ML	173
4e820e72	174	batadv_gw_reselect(bat_priv);
3e34819e	175	batadv_info(net_dev,
414254e3 ML	176	"Changing gateway bandwidth from: '%u.%u/%u.%u MBit' to: '%u.%u/%u.%u MBit'\n",
	177	down_curr / 10, down_curr % 10, up_curr / 10, up_curr % 10,
	178	down_new / 10, down_new % 10, up_new / 10, up_new % 10);
c6c8fea2	179
414254e3 ML	180	atomic_set(&bat_priv->gw.bandwidth_down, down_new);
	181	atomic_set(&bat_priv->gw.bandwidth_up, up_new);
	182	batadv_gw_tvlv_container_update(bat_priv);
c6c8fea2 SE	183
	184	end:
	185	return count;
	186	}
414254e3 ML	187
	188	/**
	189	* batadv_gw_tvlv_ogm_handler_v1 - process incoming gateway tvlv container
	190	* @bat_priv: the bat priv with all the soft interface information
	191	* @orig: the orig_node of the ogm
	192	* @flags: flags indicating the tvlv state (see batadv_tvlv_handler_flags)
	193	* @tvlv_value: tvlv buffer containing the gateway data
	194	* @tvlv_value_len: tvlv buffer length
	195	*/
	196	static void batadv_gw_tvlv_ogm_handler_v1(struct batadv_priv *bat_priv,
	197	struct batadv_orig_node *orig,
	198	uint8_t flags,
	199	void *tvlv_value,
	200	uint16_t tvlv_value_len)
	201	{
	202	struct batadv_tvlv_gateway_data gateway, *gateway_ptr;
	203
	204	/* only fetch the tvlv value if the handler wasn't called via the
	205	* CIFNOTFND flag and if there is data to fetch
	206	*/
	207	if ((flags & BATADV_TVLV_HANDLER_OGM_CIFNOTFND) \|\|
	208	(tvlv_value_len < sizeof(gateway))) {
	209	gateway.bandwidth_down = 0;
	210	gateway.bandwidth_up = 0;
	211	} else {
	212	gateway_ptr = tvlv_value;
	213	gateway.bandwidth_down = gateway_ptr->bandwidth_down;
	214	gateway.bandwidth_up = gateway_ptr->bandwidth_up;
	215	if ((gateway.bandwidth_down == 0) \|\|
	216	(gateway.bandwidth_up == 0)) {
	217	gateway.bandwidth_down = 0;
	218	gateway.bandwidth_up = 0;
	219	}
	220	}
	221
	222	batadv_gw_node_update(bat_priv, orig, &gateway);
	223
	224	/* restart gateway selection if fast or late switching was enabled */
	225	if ((gateway.bandwidth_down != 0) &&
	226	(atomic_read(&bat_priv->gw_mode) == BATADV_GW_MODE_CLIENT) &&
	227	(atomic_read(&bat_priv->gw_sel_class) > 2))
	228	batadv_gw_check_election(bat_priv, orig);
	229	}
	230
	231	/**
	232	* batadv_gw_init - initialise the gateway handling internals
	233	* @bat_priv: the bat priv with all the soft interface information
	234	*/
	235	void batadv_gw_init(struct batadv_priv *bat_priv)
	236	{
	237	batadv_tvlv_handler_register(bat_priv, batadv_gw_tvlv_ogm_handler_v1,
	238	NULL, BATADV_TVLV_GW, 1,
	239	BATADV_TVLV_HANDLER_OGM_CIFNOTFND);
	240	}
	241
	242	/**
	243	* batadv_gw_free - free the gateway handling internals
	244	* @bat_priv: the bat priv with all the soft interface information
	245	*/
	246	void batadv_gw_free(struct batadv_priv *bat_priv)
	247	{
	248	batadv_tvlv_container_unregister(bat_priv, BATADV_TVLV_GW, 1);
	249	batadv_tvlv_handler_unregister(bat_priv, BATADV_TVLV_GW, 1);
	250	}