[GitHub/mt8127/android_kernel_alcatel_ttab.git] / net / bridge / br_mdb.c

#include <linux/err.h>
#include <linux/igmp.h>
#include <linux/kernel.h>
#include <linux/netdevice.h>
#include <linux/rculist.h>
#include <linux/skbuff.h>
#include <linux/if_ether.h>
#include <net/ip.h>
#include <net/netlink.h>
#if IS_ENABLED(CONFIG_IPV6)
#include <net/ipv6.h>
#endif

#include "br_private.h"

static int br_rports_fill_info(struct sk_buff *skb, struct netlink_callback *cb,
			       struct net_device *dev)
{
	struct net_bridge *br = netdev_priv(dev);
	struct net_bridge_port *p;
	struct nlattr *nest;

	if (!br->multicast_router || hlist_empty(&br->router_list))
		return 0;

	nest = nla_nest_start(skb, MDBA_ROUTER);
	if (nest == NULL)
		return -EMSGSIZE;

	hlist_for_each_entry_rcu(p, &br->router_list, rlist) {
		if (p && nla_put_u32(skb, MDBA_ROUTER_PORT, p->dev->ifindex))
			goto fail;
	}

	nla_nest_end(skb, nest);
	return 0;
fail:
	nla_nest_cancel(skb, nest);
	return -EMSGSIZE;
}

static int br_mdb_fill_info(struct sk_buff *skb, struct netlink_callback *cb,
			    struct net_device *dev)
{
	struct net_bridge *br = netdev_priv(dev);
	struct net_bridge_mdb_htable *mdb;
	struct nlattr *nest, *nest2;
	int i, err = 0;
	int idx = 0, s_idx = cb->args[1];

	if (br->multicast_disabled)
		return 0;

	mdb = rcu_dereference(br->mdb);
	if (!mdb)
		return 0;

	nest = nla_nest_start(skb, MDBA_MDB);
	if (nest == NULL)
		return -EMSGSIZE;

	for (i = 0; i < mdb->max; i++) {
		struct net_bridge_mdb_entry *mp;
		struct net_bridge_port_group *p, **pp;
		struct net_bridge_port *port;

		hlist_for_each_entry_rcu(mp, &mdb->mhash[i], hlist[mdb->ver]) {
			if (idx < s_idx)
				goto skip;

			nest2 = nla_nest_start(skb, MDBA_MDB_ENTRY);
			if (nest2 == NULL) {
				err = -EMSGSIZE;
				goto out;
			}

			for (pp = &mp->ports;
			     (p = rcu_dereference(*pp)) != NULL;
			      pp = &p->next) {
				port = p->port;
				if (port) {
					struct br_mdb_entry e;
					memset(&e, 0, sizeof(e));
					e.ifindex = port->dev->ifindex;
					e.state = p->state;
					if (p->addr.proto == htons(ETH_P_IP))
						e.addr.u.ip4 = p->addr.u.ip4;
#if IS_ENABLED(CONFIG_IPV6)
					if (p->addr.proto == htons(ETH_P_IPV6))
						e.addr.u.ip6 = p->addr.u.ip6;
#endif
					e.addr.proto = p->addr.proto;
					if (nla_put(skb, MDBA_MDB_ENTRY_INFO, sizeof(e), &e)) {
						nla_nest_cancel(skb, nest2);
						err = -EMSGSIZE;
						goto out;
					}
				}
			}
			nla_nest_end(skb, nest2);
		skip:
			idx++;
		}
	}

out:
	cb->args[1] = idx;
	nla_nest_end(skb, nest);
	return err;
}

static int br_mdb_dump(struct sk_buff *skb, struct netlink_callback *cb)
{
	struct net_device *dev;
	struct net *net = sock_net(skb->sk);
	struct nlmsghdr *nlh = NULL;
	int idx = 0, s_idx;

	s_idx = cb->args[0];

	rcu_read_lock();

	/* In theory this could be wrapped to 0... */
	cb->seq = net->dev_base_seq + br_mdb_rehash_seq;

	for_each_netdev_rcu(net, dev) {
		if (dev->priv_flags & IFF_EBRIDGE) {
			struct br_port_msg *bpm;

			if (idx < s_idx)
				goto skip;

			nlh = nlmsg_put(skb, NETLINK_CB(cb->skb).portid,
					cb->nlh->nlmsg_seq, RTM_GETMDB,
					sizeof(*bpm), NLM_F_MULTI);
			if (nlh == NULL)
				break;

			bpm = nlmsg_data(nlh);
			memset(bpm, 0, sizeof(*bpm));
			bpm->ifindex = dev->ifindex;
			if (br_mdb_fill_info(skb, cb, dev) < 0)
				goto out;
			if (br_rports_fill_info(skb, cb, dev) < 0)
				goto out;

			cb->args[1] = 0;
			nlmsg_end(skb, nlh);
		skip:
			idx++;
		}
	}

out:
	if (nlh)
		nlmsg_end(skb, nlh);
	rcu_read_unlock();
	cb->args[0] = idx;
	return skb->len;
}

static int nlmsg_populate_mdb_fill(struct sk_buff *skb,
				   struct net_device *dev,
				   struct br_mdb_entry *entry, u32 pid,
				   u32 seq, int type, unsigned int flags)
{
	struct nlmsghdr *nlh;
	struct br_port_msg *bpm;
	struct nlattr *nest, *nest2;

	nlh = nlmsg_put(skb, pid, seq, type, sizeof(*bpm), NLM_F_MULTI);
	if (!nlh)
		return -EMSGSIZE;

	bpm = nlmsg_data(nlh);
	memset(bpm, 0, sizeof(*bpm));
	bpm->family  = AF_BRIDGE;
	bpm->ifindex = dev->ifindex;
	nest = nla_nest_start(skb, MDBA_MDB);
	if (nest == NULL)
		goto cancel;
	nest2 = nla_nest_start(skb, MDBA_MDB_ENTRY);
	if (nest2 == NULL)
		goto end;

	if (nla_put(skb, MDBA_MDB_ENTRY_INFO, sizeof(*entry), entry))
		goto end;

	nla_nest_end(skb, nest2);
	nla_nest_end(skb, nest);
	return nlmsg_end(skb, nlh);

end:
	nla_nest_end(skb, nest);
cancel:
	nlmsg_cancel(skb, nlh);
	return -EMSGSIZE;
}

static inline size_t rtnl_mdb_nlmsg_size(void)
{
	return NLMSG_ALIGN(sizeof(struct br_port_msg))
		+ nla_total_size(sizeof(struct br_mdb_entry));
}

static void __br_mdb_notify(struct net_device *dev, struct br_mdb_entry *entry,
			    int type)
{
	struct net *net = dev_net(dev);
	struct sk_buff *skb;
	int err = -ENOBUFS;

	skb = nlmsg_new(rtnl_mdb_nlmsg_size(), GFP_ATOMIC);
	if (!skb)
		goto errout;

	err = nlmsg_populate_mdb_fill(skb, dev, entry, 0, 0, type, NTF_SELF);
	if (err < 0) {
		kfree_skb(skb);
		goto errout;
	}

	rtnl_notify(skb, net, 0, RTNLGRP_MDB, NULL, GFP_ATOMIC);
	return;
errout:
	rtnl_set_sk_err(net, RTNLGRP_MDB, err);
}

void br_mdb_notify(struct net_device *dev, struct net_bridge_port *port,
		   struct br_ip *group, int type)
{
	struct br_mdb_entry entry;

	memset(&entry, 0, sizeof(entry));
	entry.ifindex = port->dev->ifindex;
	entry.addr.proto = group->proto;
	entry.addr.u.ip4 = group->u.ip4;
#if IS_ENABLED(CONFIG_IPV6)
	entry.addr.u.ip6 = group->u.ip6;
#endif
	__br_mdb_notify(dev, &entry, type);
}

static bool is_valid_mdb_entry(struct br_mdb_entry *entry)
{
	if (entry->ifindex == 0)
		return false;

	if (entry->addr.proto == htons(ETH_P_IP)) {
		if (!ipv4_is_multicast(entry->addr.u.ip4))
			return false;
		if (ipv4_is_local_multicast(entry->addr.u.ip4))
			return false;
#if IS_ENABLED(CONFIG_IPV6)
	} else if (entry->addr.proto == htons(ETH_P_IPV6)) {
		if (!ipv6_is_transient_multicast(&entry->addr.u.ip6))
			return false;
#endif
	} else
		return false;
	if (entry->state != MDB_PERMANENT && entry->state != MDB_TEMPORARY)
		return false;

	return true;
}

static int br_mdb_parse(struct sk_buff *skb, struct nlmsghdr *nlh,
			struct net_device **pdev, struct br_mdb_entry **pentry)
{
	struct net *net = sock_net(skb->sk);
	struct br_mdb_entry *entry;
	struct br_port_msg *bpm;
	struct nlattr *tb[MDBA_SET_ENTRY_MAX+1];
	struct net_device *dev;
	int err;

	err = nlmsg_parse(nlh, sizeof(*bpm), tb, MDBA_SET_ENTRY, NULL);
	if (err < 0)
		return err;

	bpm = nlmsg_data(nlh);
	if (bpm->ifindex == 0) {
		pr_info("PF_BRIDGE: br_mdb_parse() with invalid ifindex\n");
		return -EINVAL;
	}

	dev = __dev_get_by_index(net, bpm->ifindex);
	if (dev == NULL) {
		pr_info("PF_BRIDGE: br_mdb_parse() with unknown ifindex\n");
		return -ENODEV;
	}

	if (!(dev->priv_flags & IFF_EBRIDGE)) {
		pr_info("PF_BRIDGE: br_mdb_parse() with non-bridge\n");
		return -EOPNOTSUPP;
	}

	*pdev = dev;

	if (!tb[MDBA_SET_ENTRY] ||
	    nla_len(tb[MDBA_SET_ENTRY]) != sizeof(struct br_mdb_entry)) {
		pr_info("PF_BRIDGE: br_mdb_parse() with invalid attr\n");
		return -EINVAL;
	}

	entry = nla_data(tb[MDBA_SET_ENTRY]);
	if (!is_valid_mdb_entry(entry)) {
		pr_info("PF_BRIDGE: br_mdb_parse() with invalid entry\n");
		return -EINVAL;
	}

	*pentry = entry;
	return 0;
}

static int br_mdb_add_group(struct net_bridge *br, struct net_bridge_port *port,
			    struct br_ip *group, unsigned char state)
{
	struct net_bridge_mdb_entry *mp;
	struct net_bridge_port_group *p;
	struct net_bridge_port_group __rcu **pp;
	struct net_bridge_mdb_htable *mdb;
	int err;

	mdb = mlock_dereference(br->mdb, br);
	mp = br_mdb_ip_get(mdb, group);
	if (!mp) {
		mp = br_multicast_new_group(br, port, group);
		err = PTR_ERR(mp);
		if (IS_ERR(mp))
			return err;
	}

	for (pp = &mp->ports;
	     (p = mlock_dereference(*pp, br)) != NULL;
	     pp = &p->next) {
		if (p->port == port)
			return -EEXIST;
		if ((unsigned long)p->port < (unsigned long)port)
			break;
	}

	p = br_multicast_new_port_group(port, group, *pp, state);
	if (unlikely(!p))
		return -ENOMEM;
	rcu_assign_pointer(*pp, p);

	return 0;
}

static int __br_mdb_add(struct net *net, struct net_bridge *br,
			struct br_mdb_entry *entry)
{
	struct br_ip ip;
	struct net_device *dev;
	struct net_bridge_port *p;
	int ret;

	if (!netif_running(br->dev) || br->multicast_disabled)
		return -EINVAL;

	dev = __dev_get_by_index(net, entry->ifindex);
	if (!dev)
		return -ENODEV;

	p = br_port_get_rtnl(dev);
	if (!p || p->br != br || p->state == BR_STATE_DISABLED)
		return -EINVAL;

	memset(&ip, 0, sizeof(ip));
	ip.proto = entry->addr.proto;
	if (ip.proto == htons(ETH_P_IP))
		ip.u.ip4 = entry->addr.u.ip4;
#if IS_ENABLED(CONFIG_IPV6)
	else
		ip.u.ip6 = entry->addr.u.ip6;
#endif

	spin_lock_bh(&br->multicast_lock);
	ret = br_mdb_add_group(br, p, &ip, entry->state);
	spin_unlock_bh(&br->multicast_lock);
	return ret;
}

static int br_mdb_add(struct sk_buff *skb, struct nlmsghdr *nlh)
{
	struct net *net = sock_net(skb->sk);
	struct br_mdb_entry *entry;
	struct net_device *dev;
	struct net_bridge *br;
	int err;

	err = br_mdb_parse(skb, nlh, &dev, &entry);
	if (err < 0)
		return err;

	br = netdev_priv(dev);

	err = __br_mdb_add(net, br, entry);
	if (!err)
		__br_mdb_notify(dev, entry, RTM_NEWMDB);
	return err;
}

static int __br_mdb_del(struct net_bridge *br, struct br_mdb_entry *entry)
{
	struct net_bridge_mdb_htable *mdb;
	struct net_bridge_mdb_entry *mp;
	struct net_bridge_port_group *p;
	struct net_bridge_port_group __rcu **pp;
	struct br_ip ip;
	int err = -EINVAL;

	if (!netif_running(br->dev) || br->multicast_disabled)
		return -EINVAL;

	if (timer_pending(&br->multicast_querier_timer))
		return -EBUSY;

	memset(&ip, 0, sizeof(ip));
	ip.proto = entry->addr.proto;
	if (ip.proto == htons(ETH_P_IP))
		ip.u.ip4 = entry->addr.u.ip4;
#if IS_ENABLED(CONFIG_IPV6)
	else
		ip.u.ip6 = entry->addr.u.ip6;
#endif

	spin_lock_bh(&br->multicast_lock);
	mdb = mlock_dereference(br->mdb, br);

	mp = br_mdb_ip_get(mdb, &ip);
	if (!mp)
		goto unlock;

	for (pp = &mp->ports;
	     (p = mlock_dereference(*pp, br)) != NULL;
	     pp = &p->next) {
		if (!p->port || p->port->dev->ifindex != entry->ifindex)
			continue;

		if (p->port->state == BR_STATE_DISABLED)
			goto unlock;

		rcu_assign_pointer(*pp, p->next);
		hlist_del_init(&p->mglist);
		del_timer(&p->timer);
		call_rcu_bh(&p->rcu, br_multicast_free_pg);
		err = 0;

		if (!mp->ports && !mp->mglist &&
		    netif_running(br->dev))
			mod_timer(&mp->timer, jiffies);
		break;
	}

unlock:
	spin_unlock_bh(&br->multicast_lock);
	return err;
}

static int br_mdb_del(struct sk_buff *skb, struct nlmsghdr *nlh)
{
	struct net_device *dev;
	struct br_mdb_entry *entry;
	struct net_bridge *br;
	int err;

	err = br_mdb_parse(skb, nlh, &dev, &entry);
	if (err < 0)
		return err;

	br = netdev_priv(dev);

	err = __br_mdb_del(br, entry);
	if (!err)
		__br_mdb_notify(dev, entry, RTM_DELMDB);
	return err;
}

void br_mdb_init(void)
{
	rtnl_register(PF_BRIDGE, RTM_GETMDB, NULL, br_mdb_dump, NULL);
	rtnl_register(PF_BRIDGE, RTM_NEWMDB, br_mdb_add, NULL, NULL);
	rtnl_register(PF_BRIDGE, RTM_DELMDB, br_mdb_del, NULL, NULL);
}

void br_mdb_uninit(void)
{
	rtnl_unregister(PF_BRIDGE, RTM_GETMDB);
	rtnl_unregister(PF_BRIDGE, RTM_NEWMDB);
	rtnl_unregister(PF_BRIDGE, RTM_DELMDB);
}
Commit	Line	Data
ee07c6e7 CW	1	#include <linux/err.h>
	2	#include <linux/igmp.h>
	3	#include <linux/kernel.h>
	4	#include <linux/netdevice.h>
	5	#include <linux/rculist.h>
	6	#include <linux/skbuff.h>
cfd56754	7	#include <linux/if_ether.h>
ee07c6e7 CW	8	#include <net/ip.h>
	9	#include <net/netlink.h>
	10	#if IS_ENABLED(CONFIG_IPV6)
	11	#include <net/ipv6.h>
	12	#endif
	13
	14	#include "br_private.h"
	15
	16	static int br_rports_fill_info(struct sk_buff skb, struct netlink_callback cb,
	17	struct net_device *dev)
	18	{
	19	struct net_bridge *br = netdev_priv(dev);
	20	struct net_bridge_port *p;
ee07c6e7 CW	21	struct nlattr *nest;
	22
	23	if (!br->multicast_router \|\| hlist_empty(&br->router_list))
	24	return 0;
	25
	26	nest = nla_nest_start(skb, MDBA_ROUTER);
	27	if (nest == NULL)
	28	return -EMSGSIZE;
	29
b67bfe0d	30	hlist_for_each_entry_rcu(p, &br->router_list, rlist) {
ee07c6e7 CW	31	if (p && nla_put_u32(skb, MDBA_ROUTER_PORT, p->dev->ifindex))
	32	goto fail;
	33	}
	34
	35	nla_nest_end(skb, nest);
	36	return 0;
	37	fail:
	38	nla_nest_cancel(skb, nest);
	39	return -EMSGSIZE;
	40	}
	41
	42	static int br_mdb_fill_info(struct sk_buff skb, struct netlink_callback cb,
	43	struct net_device *dev)
	44	{
	45	struct net_bridge *br = netdev_priv(dev);
	46	struct net_bridge_mdb_htable *mdb;
	47	struct nlattr nest, nest2;
	48	int i, err = 0;
	49	int idx = 0, s_idx = cb->args[1];
	50
	51	if (br->multicast_disabled)
	52	return 0;
	53
	54	mdb = rcu_dereference(br->mdb);
	55	if (!mdb)
	56	return 0;
	57
	58	nest = nla_nest_start(skb, MDBA_MDB);
	59	if (nest == NULL)
	60	return -EMSGSIZE;
	61
	62	for (i = 0; i < mdb->max; i++) {
ee07c6e7 CW	63	struct net_bridge_mdb_entry *mp;
	64	struct net_bridge_port_group p, *pp;
	65	struct net_bridge_port *port;
	66
b67bfe0d	67	hlist_for_each_entry_rcu(mp, &mdb->mhash[i], hlist[mdb->ver]) {
ee07c6e7 CW	68	if (idx < s_idx)
	69	goto skip;
	70
	71	nest2 = nla_nest_start(skb, MDBA_MDB_ENTRY);
	72	if (nest2 == NULL) {
	73	err = -EMSGSIZE;
	74	goto out;
	75	}
	76
	77	for (pp = &mp->ports;
	78	(p = rcu_dereference(*pp)) != NULL;
	79	pp = &p->next) {
	80	port = p->port;
	81	if (port) {
	82	struct br_mdb_entry e;
c085c499	83	memset(&e, 0, sizeof(e));
ee07c6e7	84	e.ifindex = port->dev->ifindex;
ccb1c31a	85	e.state = p->state;
09d7cf7d VY	86	if (p->addr.proto == htons(ETH_P_IP))
09d7cf7d VY	87	e.addr.u.ip4 = p->addr.u.ip4;
ee07c6e7	88	#if IS_ENABLED(CONFIG_IPV6)
09d7cf7d VY	89	if (p->addr.proto == htons(ETH_P_IPV6))
09d7cf7d VY	90	e.addr.u.ip6 = p->addr.u.ip6;
ee07c6e7 CW	91	#endif
	92	e.addr.proto = p->addr.proto;
	93	if (nla_put(skb, MDBA_MDB_ENTRY_INFO, sizeof(e), &e)) {
	94	nla_nest_cancel(skb, nest2);
	95	err = -EMSGSIZE;
	96	goto out;
	97	}
	98	}
	99	}
	100	nla_nest_end(skb, nest2);
	101	skip:
	102	idx++;
	103	}
	104	}
	105
	106	out:
	107	cb->args[1] = idx;
	108	nla_nest_end(skb, nest);
	109	return err;
	110	}
	111
	112	static int br_mdb_dump(struct sk_buff skb, struct netlink_callback cb)
	113	{
	114	struct net_device *dev;
	115	struct net *net = sock_net(skb->sk);
	116	struct nlmsghdr *nlh = NULL;
	117	int idx = 0, s_idx;
	118
	119	s_idx = cb->args[0];
	120
	121	rcu_read_lock();
	122
2ce297fc CW	123	/* In theory this could be wrapped to 0... */
2ce297fc CW	124	cb->seq = net->dev_base_seq + br_mdb_rehash_seq;
ee07c6e7 CW	125
	126	for_each_netdev_rcu(net, dev) {
	127	if (dev->priv_flags & IFF_EBRIDGE) {
	128	struct br_port_msg *bpm;
	129
	130	if (idx < s_idx)
	131	goto skip;
	132
	133	nlh = nlmsg_put(skb, NETLINK_CB(cb->skb).portid,
	134	cb->nlh->nlmsg_seq, RTM_GETMDB,
	135	sizeof(*bpm), NLM_F_MULTI);
	136	if (nlh == NULL)
	137	break;
	138
	139	bpm = nlmsg_data(nlh);
c085c499	140	memset(bpm, 0, sizeof(*bpm));
ee07c6e7 CW	141	bpm->ifindex = dev->ifindex;
	142	if (br_mdb_fill_info(skb, cb, dev) < 0)
	143	goto out;
	144	if (br_rports_fill_info(skb, cb, dev) < 0)
	145	goto out;
	146
	147	cb->args[1] = 0;
	148	nlmsg_end(skb, nlh);
	149	skip:
	150	idx++;
	151	}
	152	}
	153
	154	out:
	155	if (nlh)
	156	nlmsg_end(skb, nlh);
	157	rcu_read_unlock();
	158	cb->args[0] = idx;
	159	return skb->len;
	160	}
	161
37a393bc CW	162	static int nlmsg_populate_mdb_fill(struct sk_buff *skb,
	163	struct net_device *dev,
	164	struct br_mdb_entry *entry, u32 pid,
	165	u32 seq, int type, unsigned int flags)
	166	{
	167	struct nlmsghdr *nlh;
	168	struct br_port_msg *bpm;
	169	struct nlattr nest, nest2;
	170
	171	nlh = nlmsg_put(skb, pid, seq, type, sizeof(*bpm), NLM_F_MULTI);
	172	if (!nlh)
	173	return -EMSGSIZE;
	174
	175	bpm = nlmsg_data(nlh);
c085c499	176	memset(bpm, 0, sizeof(*bpm));
37a393bc CW	177	bpm->family = AF_BRIDGE;
	178	bpm->ifindex = dev->ifindex;
	179	nest = nla_nest_start(skb, MDBA_MDB);
	180	if (nest == NULL)
	181	goto cancel;
	182	nest2 = nla_nest_start(skb, MDBA_MDB_ENTRY);
	183	if (nest2 == NULL)
	184	goto end;
	185
	186	if (nla_put(skb, MDBA_MDB_ENTRY_INFO, sizeof(*entry), entry))
	187	goto end;
	188
	189	nla_nest_end(skb, nest2);
	190	nla_nest_end(skb, nest);
	191	return nlmsg_end(skb, nlh);
	192
	193	end:
	194	nla_nest_end(skb, nest);
	195	cancel:
	196	nlmsg_cancel(skb, nlh);
	197	return -EMSGSIZE;
	198	}
	199
	200	static inline size_t rtnl_mdb_nlmsg_size(void)
	201	{
	202	return NLMSG_ALIGN(sizeof(struct br_port_msg))
	203	+ nla_total_size(sizeof(struct br_mdb_entry));
	204	}
	205
	206	static void __br_mdb_notify(struct net_device dev, struct br_mdb_entry entry,
	207	int type)
	208	{
	209	struct net *net = dev_net(dev);
	210	struct sk_buff *skb;
	211	int err = -ENOBUFS;
	212
	213	skb = nlmsg_new(rtnl_mdb_nlmsg_size(), GFP_ATOMIC);
	214	if (!skb)
	215	goto errout;
	216
	217	err = nlmsg_populate_mdb_fill(skb, dev, entry, 0, 0, type, NTF_SELF);
	218	if (err < 0) {
	219	kfree_skb(skb);
	220	goto errout;
	221	}
	222
	223	rtnl_notify(skb, net, 0, RTNLGRP_MDB, NULL, GFP_ATOMIC);
	224	return;
	225	errout:
	226	rtnl_set_sk_err(net, RTNLGRP_MDB, err);
	227	}
	228
	229	void br_mdb_notify(struct net_device dev, struct net_bridge_port port,
	230	struct br_ip *group, int type)
	231	{
	232	struct br_mdb_entry entry;
	233
c085c499	234	memset(&entry, 0, sizeof(entry));
37a393bc CW	235	entry.ifindex = port->dev->ifindex;
	236	entry.addr.proto = group->proto;
	237	entry.addr.u.ip4 = group->u.ip4;
	238	#if IS_ENABLED(CONFIG_IPV6)
	239	entry.addr.u.ip6 = group->u.ip6;
	240	#endif
	241	__br_mdb_notify(dev, &entry, type);
	242	}
	243
cfd56754 CW	244	static bool is_valid_mdb_entry(struct br_mdb_entry *entry)
	245	{
	246	if (entry->ifindex == 0)
	247	return false;
	248
	249	if (entry->addr.proto == htons(ETH_P_IP)) {
	250	if (!ipv4_is_multicast(entry->addr.u.ip4))
	251	return false;
	252	if (ipv4_is_local_multicast(entry->addr.u.ip4))
	253	return false;
	254	#if IS_ENABLED(CONFIG_IPV6)
	255	} else if (entry->addr.proto == htons(ETH_P_IPV6)) {
	256	if (!ipv6_is_transient_multicast(&entry->addr.u.ip6))
	257	return false;
	258	#endif
	259	} else
	260	return false;
ccb1c31a AW	261	if (entry->state != MDB_PERMANENT && entry->state != MDB_TEMPORARY)
ccb1c31a AW	262	return false;
cfd56754 CW	263
	264	return true;
	265	}
	266
	267	static int br_mdb_parse(struct sk_buff skb, struct nlmsghdr nlh,
	268	struct net_device pdev, struct br_mdb_entry pentry)
	269	{
	270	struct net *net = sock_net(skb->sk);
	271	struct br_mdb_entry *entry;
	272	struct br_port_msg *bpm;
	273	struct nlattr *tb[MDBA_SET_ENTRY_MAX+1];
	274	struct net_device *dev;
	275	int err;
	276
cfd56754 CW	277	err = nlmsg_parse(nlh, sizeof(*bpm), tb, MDBA_SET_ENTRY, NULL);
	278	if (err < 0)
	279	return err;
	280
	281	bpm = nlmsg_data(nlh);
	282	if (bpm->ifindex == 0) {
	283	pr_info("PF_BRIDGE: br_mdb_parse() with invalid ifindex\n");
	284	return -EINVAL;
	285	}
	286
	287	dev = __dev_get_by_index(net, bpm->ifindex);
	288	if (dev == NULL) {
	289	pr_info("PF_BRIDGE: br_mdb_parse() with unknown ifindex\n");
	290	return -ENODEV;
	291	}
	292
	293	if (!(dev->priv_flags & IFF_EBRIDGE)) {
	294	pr_info("PF_BRIDGE: br_mdb_parse() with non-bridge\n");
	295	return -EOPNOTSUPP;
	296	}
	297
	298	*pdev = dev;
	299
	300	if (!tb[MDBA_SET_ENTRY] \|\|
	301	nla_len(tb[MDBA_SET_ENTRY]) != sizeof(struct br_mdb_entry)) {
	302	pr_info("PF_BRIDGE: br_mdb_parse() with invalid attr\n");
	303	return -EINVAL;
	304	}
	305
	306	entry = nla_data(tb[MDBA_SET_ENTRY]);
	307	if (!is_valid_mdb_entry(entry)) {
	308	pr_info("PF_BRIDGE: br_mdb_parse() with invalid entry\n");
	309	return -EINVAL;
	310	}
	311
	312	*pentry = entry;
	313	return 0;
	314	}
	315
	316	static int br_mdb_add_group(struct net_bridge br, struct net_bridge_port port,
ccb1c31a	317	struct br_ip *group, unsigned char state)
cfd56754 CW	318	{
	319	struct net_bridge_mdb_entry *mp;
	320	struct net_bridge_port_group *p;
	321	struct net_bridge_port_group __rcu **pp;
	322	struct net_bridge_mdb_htable *mdb;
	323	int err;
	324
	325	mdb = mlock_dereference(br->mdb, br);
	326	mp = br_mdb_ip_get(mdb, group);
	327	if (!mp) {
	328	mp = br_multicast_new_group(br, port, group);
	329	err = PTR_ERR(mp);
	330	if (IS_ERR(mp))
	331	return err;
	332	}
	333
	334	for (pp = &mp->ports;
	335	(p = mlock_dereference(*pp, br)) != NULL;
	336	pp = &p->next) {
	337	if (p->port == port)
	338	return -EEXIST;
	339	if ((unsigned long)p->port < (unsigned long)port)
	340	break;
	341	}
	342
ccb1c31a	343	p = br_multicast_new_port_group(port, group, *pp, state);
cfd56754 CW	344	if (unlikely(!p))
	345	return -ENOMEM;
	346	rcu_assign_pointer(*pp, p);
	347
cfd56754 CW	348	return 0;
	349	}
	350
	351	static int __br_mdb_add(struct net net, struct net_bridge br,
	352	struct br_mdb_entry *entry)
	353	{
	354	struct br_ip ip;
	355	struct net_device *dev;
	356	struct net_bridge_port *p;
	357	int ret;
	358
	359	if (!netif_running(br->dev) \|\| br->multicast_disabled)
	360	return -EINVAL;
	361
	362	dev = __dev_get_by_index(net, entry->ifindex);
	363	if (!dev)
	364	return -ENODEV;
	365
	366	p = br_port_get_rtnl(dev);
	367	if (!p \|\| p->br != br \|\| p->state == BR_STATE_DISABLED)
	368	return -EINVAL;
	369
7865ece3	370	memset(&ip, 0, sizeof(ip));
cfd56754 CW	371	ip.proto = entry->addr.proto;
	372	if (ip.proto == htons(ETH_P_IP))
	373	ip.u.ip4 = entry->addr.u.ip4;
	374	#if IS_ENABLED(CONFIG_IPV6)
	375	else
	376	ip.u.ip6 = entry->addr.u.ip6;
	377	#endif
	378
	379	spin_lock_bh(&br->multicast_lock);
ccb1c31a	380	ret = br_mdb_add_group(br, p, &ip, entry->state);
cfd56754 CW	381	spin_unlock_bh(&br->multicast_lock);
	382	return ret;
	383	}
	384
661d2967	385	static int br_mdb_add(struct sk_buff skb, struct nlmsghdr nlh)
cfd56754 CW	386	{
	387	struct net *net = sock_net(skb->sk);
	388	struct br_mdb_entry *entry;
	389	struct net_device *dev;
	390	struct net_bridge *br;
	391	int err;
	392
	393	err = br_mdb_parse(skb, nlh, &dev, &entry);
	394	if (err < 0)
	395	return err;
	396
	397	br = netdev_priv(dev);
	398
	399	err = __br_mdb_add(net, br, entry);
	400	if (!err)
	401	__br_mdb_notify(dev, entry, RTM_NEWMDB);
	402	return err;
	403	}
	404
	405	static int __br_mdb_del(struct net_bridge br, struct br_mdb_entry entry)
	406	{
	407	struct net_bridge_mdb_htable *mdb;
	408	struct net_bridge_mdb_entry *mp;
	409	struct net_bridge_port_group *p;
	410	struct net_bridge_port_group __rcu **pp;
	411	struct br_ip ip;
	412	int err = -EINVAL;
	413
	414	if (!netif_running(br->dev) \|\| br->multicast_disabled)
	415	return -EINVAL;
	416
	417	if (timer_pending(&br->multicast_querier_timer))
	418	return -EBUSY;
	419
7865ece3	420	memset(&ip, 0, sizeof(ip));
cfd56754 CW	421	ip.proto = entry->addr.proto;
	422	if (ip.proto == htons(ETH_P_IP))
	423	ip.u.ip4 = entry->addr.u.ip4;
	424	#if IS_ENABLED(CONFIG_IPV6)
	425	else
	426	ip.u.ip6 = entry->addr.u.ip6;
	427	#endif
	428
	429	spin_lock_bh(&br->multicast_lock);
	430	mdb = mlock_dereference(br->mdb, br);
	431
	432	mp = br_mdb_ip_get(mdb, &ip);
	433	if (!mp)
	434	goto unlock;
	435
	436	for (pp = &mp->ports;
	437	(p = mlock_dereference(*pp, br)) != NULL;
	438	pp = &p->next) {
	439	if (!p->port \|\| p->port->dev->ifindex != entry->ifindex)
	440	continue;
	441
	442	if (p->port->state == BR_STATE_DISABLED)
	443	goto unlock;
	444
	445	rcu_assign_pointer(*pp, p->next);
	446	hlist_del_init(&p->mglist);
	447	del_timer(&p->timer);
	448	call_rcu_bh(&p->rcu, br_multicast_free_pg);
	449	err = 0;
	450
	451	if (!mp->ports && !mp->mglist &&
	452	netif_running(br->dev))
	453	mod_timer(&mp->timer, jiffies);
	454	break;
	455	}
	456
	457	unlock:
	458	spin_unlock_bh(&br->multicast_lock);
	459	return err;
	460	}
	461
661d2967	462	static int br_mdb_del(struct sk_buff skb, struct nlmsghdr nlh)
cfd56754 CW	463	{
	464	struct net_device *dev;
	465	struct br_mdb_entry *entry;
	466	struct net_bridge *br;
	467	int err;
	468
	469	err = br_mdb_parse(skb, nlh, &dev, &entry);
	470	if (err < 0)
	471	return err;
	472
	473	br = netdev_priv(dev);
	474
	475	err = __br_mdb_del(br, entry);
	476	if (!err)
	477	__br_mdb_notify(dev, entry, RTM_DELMDB);
	478	return err;
	479	}
	480
ee07c6e7 CW	481	void br_mdb_init(void)
	482	{
	483	rtnl_register(PF_BRIDGE, RTM_GETMDB, NULL, br_mdb_dump, NULL);
cfd56754 CW	484	rtnl_register(PF_BRIDGE, RTM_NEWMDB, br_mdb_add, NULL, NULL);
cfd56754 CW	485	rtnl_register(PF_BRIDGE, RTM_DELMDB, br_mdb_del, NULL, NULL);
ee07c6e7	486	}
63233159 VY	487
	488	void br_mdb_uninit(void)
	489	{
	490	rtnl_unregister(PF_BRIDGE, RTM_GETMDB);
	491	rtnl_unregister(PF_BRIDGE, RTM_NEWMDB);
	492	rtnl_unregister(PF_BRIDGE, RTM_DELMDB);
	493	}