[deliverable/linux.git] / drivers / net / team / team_mode_activebackup.c

/*
 * net/drivers/team/team_mode_activebackup.c - Active-backup mode for team
 * Copyright (c) 2011 Jiri Pirko <jpirko@redhat.com>
 *
 * 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/errno.h>
#include <linux/netdevice.h>
#include <net/rtnetlink.h>
#include <linux/if_team.h>

struct ab_priv {
	struct team_port __rcu *active_port;
};

static struct ab_priv *ab_priv(struct team *team)
{
	return (struct ab_priv *) &team->mode_priv;
}

static rx_handler_result_t ab_receive(struct team *team, struct team_port *port,
				      struct sk_buff *skb) {
	struct team_port *active_port;

	active_port = rcu_dereference(ab_priv(team)->active_port);
	if (active_port != port)
		return RX_HANDLER_EXACT;
	return RX_HANDLER_ANOTHER;
}

static bool ab_transmit(struct team *team, struct sk_buff *skb)
{
	struct team_port *active_port;

	active_port = rcu_dereference(ab_priv(team)->active_port);
	if (unlikely(!active_port))
		goto drop;
	skb->dev = active_port->dev;
	if (dev_queue_xmit(skb))
		return false;
	return true;

drop:
	dev_kfree_skb_any(skb);
	return false;
}

static void ab_port_leave(struct team *team, struct team_port *port)
{
	if (ab_priv(team)->active_port == port)
		rcu_assign_pointer(ab_priv(team)->active_port, NULL);
}

static int ab_active_port_get(struct team *team, void *arg)
{
	u32 *ifindex = arg;

	*ifindex = 0;
	if (ab_priv(team)->active_port)
		*ifindex = ab_priv(team)->active_port->dev->ifindex;
	return 0;
}

static int ab_active_port_set(struct team *team, void *arg)
{
	u32 *ifindex = arg;
	struct team_port *port;

	list_for_each_entry_rcu(port, &team->port_list, list) {
		if (port->dev->ifindex == *ifindex) {
			rcu_assign_pointer(ab_priv(team)->active_port, port);
			return 0;
		}
	}
	return -ENOENT;
}

static struct team_option ab_options[] = {
	{
		.name = "activeport",
		.type = TEAM_OPTION_TYPE_U32,
		.getter = ab_active_port_get,
		.setter = ab_active_port_set,
	},
};

int ab_init(struct team *team)
{
	team_options_register(team, ab_options, ARRAY_SIZE(ab_options));
	return 0;
}

void ab_exit(struct team *team)
{
	team_options_unregister(team, ab_options, ARRAY_SIZE(ab_options));
}

static const struct team_mode_ops ab_mode_ops = {
	.init			= ab_init,
	.exit			= ab_exit,
	.receive		= ab_receive,
	.transmit		= ab_transmit,
	.port_leave		= ab_port_leave,
};

static struct team_mode ab_mode = {
	.kind		= "activebackup",
	.owner		= THIS_MODULE,
	.priv_size	= sizeof(struct ab_priv),
	.ops		= &ab_mode_ops,
};

static int __init ab_init_module(void)
{
	return team_mode_register(&ab_mode);
}

static void __exit ab_cleanup_module(void)
{
	team_mode_unregister(&ab_mode);
}

module_init(ab_init_module);
module_exit(ab_cleanup_module);

MODULE_LICENSE("GPL v2");
MODULE_AUTHOR("Jiri Pirko <jpirko@redhat.com>");
MODULE_DESCRIPTION("Active-backup mode for team");
MODULE_ALIAS("team-mode-activebackup");
Commit	Line	Data
3d249d4c JP	1	/*
	2	* net/drivers/team/team_mode_activebackup.c - Active-backup mode for team
	3	* Copyright (c) 2011 Jiri Pirko <jpirko@redhat.com>
	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/errno.h>
	16	#include <linux/netdevice.h>
	17	#include <net/rtnetlink.h>
	18	#include <linux/if_team.h>
	19
	20	struct ab_priv {
	21	struct team_port __rcu *active_port;
	22	};
	23
	24	static struct ab_priv ab_priv(struct team team)
	25	{
	26	return (struct ab_priv *) &team->mode_priv;
	27	}
	28
	29	static rx_handler_result_t ab_receive(struct team team, struct team_port port,
	30	struct sk_buff *skb) {
	31	struct team_port *active_port;
	32
	33	active_port = rcu_dereference(ab_priv(team)->active_port);
	34	if (active_port != port)
	35	return RX_HANDLER_EXACT;
	36	return RX_HANDLER_ANOTHER;
	37	}
	38
	39	static bool ab_transmit(struct team team, struct sk_buff skb)
	40	{
	41	struct team_port *active_port;
	42
	43	active_port = rcu_dereference(ab_priv(team)->active_port);
	44	if (unlikely(!active_port))
	45	goto drop;
	46	skb->dev = active_port->dev;
	47	if (dev_queue_xmit(skb))
	48	return false;
	49	return true;
	50
	51	drop:
	52	dev_kfree_skb_any(skb);
	53	return false;
	54	}
	55
	56	static void ab_port_leave(struct team team, struct team_port port)
	57	{
	58	if (ab_priv(team)->active_port == port)
	59	rcu_assign_pointer(ab_priv(team)->active_port, NULL);
	60	}
	61
	62	static int ab_active_port_get(struct team team, void arg)
	63	{
	64	u32 *ifindex = arg;
65
66	*ifindex = 0;
67	if (ab_priv(team)->active_port)
68	*ifindex = ab_priv(team)->active_port->dev->ifindex;
69	return 0;
70	}
71
72	static int ab_active_port_set(struct team team, void arg)
73	{
74	u32 *ifindex = arg;
75	struct team_port *port;
76
77	list_for_each_entry_rcu(port, &team->port_list, list) {
78	if (port->dev->ifindex == *ifindex) {
79	rcu_assign_pointer(ab_priv(team)->active_port, port);
80	return 0;
81	}
82	}
83	return -ENOENT;
84	}
85
86	static struct team_option ab_options[] = {
87	{
88	.name = "activeport",
89	.type = TEAM_OPTION_TYPE_U32,
90	.getter = ab_active_port_get,
91	.setter = ab_active_port_set,
92	},
93	};
94
95	int ab_init(struct team *team)
96	{
97	team_options_register(team, ab_options, ARRAY_SIZE(ab_options));
98	return 0;
99	}
100
101	void ab_exit(struct team *team)
102	{
103	team_options_unregister(team, ab_options, ARRAY_SIZE(ab_options));
104	}
105
106	static const struct team_mode_ops ab_mode_ops = {
107	.init = ab_init,
108	.exit = ab_exit,
109	.receive = ab_receive,
110	.transmit = ab_transmit,
111	.port_leave = ab_port_leave,
112	};
113
114	static struct team_mode ab_mode = {
115	.kind = "activebackup",
116	.owner = THIS_MODULE,
117	.priv_size = sizeof(struct ab_priv),
118	.ops = &ab_mode_ops,
119	};
120
121	static int __init ab_init_module(void)
122	{
123	return team_mode_register(&ab_mode);
124	}
125
126	static void __exit ab_cleanup_module(void)
127	{
128	team_mode_unregister(&ab_mode);
129	}
130
131	module_init(ab_init_module);
132	module_exit(ab_cleanup_module);
133
134	MODULE_LICENSE("GPL v2");
135	MODULE_AUTHOR("Jiri Pirko <jpirko@redhat.com>");
136	MODULE_DESCRIPTION("Active-backup mode for team");
137	MODULE_ALIAS("team-mode-activebackup");