[linux-2.6-block.git] / drivers / net / team / team_mode_random.c

/*
 * drivers/net/team/team_mode_random.c - Random mode for team
 * Copyright (c) 2013 Jiri Pirko <jiri@resnulli.us>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 */

#include <linux/kernel.h>
#include <linux/types.h>
#include <linux/module.h>
#include <linux/init.h>
#include <linux/skbuff.h>
#include <linux/if_team.h>

static bool rnd_transmit(struct team *team, struct sk_buff *skb)
{
	struct team_port *port;
	int port_index;

	port_index = prandom_u32_max(team->en_port_count);
	port = team_get_port_by_index_rcu(team, port_index);
	if (unlikely(!port))
		goto drop;
	port = team_get_first_port_txable_rcu(team, port);
	if (unlikely(!port))
		goto drop;
	if (team_dev_queue_xmit(team, port, skb))
		return false;
	return true;

drop:
	dev_kfree_skb_any(skb);
	return false;
}

static const struct team_mode_ops rnd_mode_ops = {
	.transmit		= rnd_transmit,
	.port_enter		= team_modeop_port_enter,
	.port_change_dev_addr	= team_modeop_port_change_dev_addr,
};

static const struct team_mode rnd_mode = {
	.kind		= "random",
	.owner		= THIS_MODULE,
	.ops		= &rnd_mode_ops,
	.lag_tx_type	= NETDEV_LAG_TX_TYPE_RANDOM,
};

static int __init rnd_init_module(void)
{
	return team_mode_register(&rnd_mode);
}

static void __exit rnd_cleanup_module(void)
{
	team_mode_unregister(&rnd_mode);
}

module_init(rnd_init_module);
module_exit(rnd_cleanup_module);

MODULE_LICENSE("GPL v2");
MODULE_AUTHOR("Jiri Pirko <jiri@resnulli.us>");
MODULE_DESCRIPTION("Random mode for team");
MODULE_ALIAS_TEAM_MODE("random");
Commit	Line	Data
753f9939 JP	1	/*
	2	* drivers/net/team/team_mode_random.c - Random mode for team
	3	* Copyright (c) 2013 Jiri Pirko <jiri@resnulli.us>
	4	*
	5	* This program is free software; you can redistribute it and/or modify
	6	* it under the terms of the GNU General Public License as published by
	7	* the Free Software Foundation; either version 2 of the License, or
	8	* (at your option) any later version.
	9	*/
	10
	11	#include <linux/kernel.h>
	12	#include <linux/types.h>
	13	#include <linux/module.h>
	14	#include <linux/init.h>
	15	#include <linux/skbuff.h>
753f9939 JP	16	#include <linux/if_team.h>
753f9939 JP	17
753f9939 JP	18	static bool rnd_transmit(struct team team, struct sk_buff skb)
	19	{
	20	struct team_port *port;
	21	int port_index;
	22
f337db64	23	port_index = prandom_u32_max(team->en_port_count);
753f9939	24	port = team_get_port_by_index_rcu(team, port_index);
76c455de JP	25	if (unlikely(!port))
76c455de JP	26	goto drop;
753f9939 JP	27	port = team_get_first_port_txable_rcu(team, port);
	28	if (unlikely(!port))
	29	goto drop;
	30	if (team_dev_queue_xmit(team, port, skb))
	31	return false;
	32	return true;
	33
	34	drop:
	35	dev_kfree_skb_any(skb);
	36	return false;
	37	}
	38
	39	static const struct team_mode_ops rnd_mode_ops = {
	40	.transmit = rnd_transmit,
	41	.port_enter = team_modeop_port_enter,
	42	.port_change_dev_addr = team_modeop_port_change_dev_addr,
	43	};
	44
	45	static const struct team_mode rnd_mode = {
	46	.kind = "random",
	47	.owner = THIS_MODULE,
	48	.ops = &rnd_mode_ops,
8fd72856	49	.lag_tx_type = NETDEV_LAG_TX_TYPE_RANDOM,
753f9939 JP	50	};
	51
	52	static int __init rnd_init_module(void)
	53	{
	54	return team_mode_register(&rnd_mode);
	55	}
	56
	57	static void __exit rnd_cleanup_module(void)
	58	{
	59	team_mode_unregister(&rnd_mode);
	60	}
	61
	62	module_init(rnd_init_module);
	63	module_exit(rnd_cleanup_module);
	64
	65	MODULE_LICENSE("GPL v2");
	66	MODULE_AUTHOR("Jiri Pirko <jiri@resnulli.us>");
	67	MODULE_DESCRIPTION("Random mode for team");
3a5f8997	68	MODULE_ALIAS_TEAM_MODE("random");