xref: /openbmc/linux/net/bridge/br_mdb.c (revision e961f8c6)
1 // SPDX-License-Identifier: GPL-2.0
2 #include <linux/err.h>
3 #include <linux/igmp.h>
4 #include <linux/kernel.h>
5 #include <linux/netdevice.h>
6 #include <linux/rculist.h>
7 #include <linux/skbuff.h>
8 #include <linux/if_ether.h>
9 #include <net/ip.h>
10 #include <net/netlink.h>
11 #include <net/switchdev.h>
12 #if IS_ENABLED(CONFIG_IPV6)
13 #include <net/ipv6.h>
14 #include <net/addrconf.h>
15 #endif
16 
17 #include "br_private.h"
18 
19 static bool
20 br_ip4_rports_get_timer(struct net_bridge_mcast_port *pmctx,
21 			unsigned long *timer)
22 {
23 	*timer = br_timer_value(&pmctx->ip4_mc_router_timer);
24 	return !hlist_unhashed(&pmctx->ip4_rlist);
25 }
26 
27 static bool
28 br_ip6_rports_get_timer(struct net_bridge_mcast_port *pmctx,
29 			unsigned long *timer)
30 {
31 #if IS_ENABLED(CONFIG_IPV6)
32 	*timer = br_timer_value(&pmctx->ip6_mc_router_timer);
33 	return !hlist_unhashed(&pmctx->ip6_rlist);
34 #else
35 	*timer = 0;
36 	return false;
37 #endif
38 }
39 
40 static size_t __br_rports_one_size(void)
41 {
42 	return nla_total_size(sizeof(u32)) + /* MDBA_ROUTER_PORT */
43 	       nla_total_size(sizeof(u32)) + /* MDBA_ROUTER_PATTR_TIMER */
44 	       nla_total_size(sizeof(u8)) +  /* MDBA_ROUTER_PATTR_TYPE */
45 	       nla_total_size(sizeof(u32)) + /* MDBA_ROUTER_PATTR_INET_TIMER */
46 	       nla_total_size(sizeof(u32)) + /* MDBA_ROUTER_PATTR_INET6_TIMER */
47 	       nla_total_size(sizeof(u32));  /* MDBA_ROUTER_PATTR_VID */
48 }
49 
50 size_t br_rports_size(const struct net_bridge_mcast *brmctx)
51 {
52 	struct net_bridge_mcast_port *pmctx;
53 	size_t size = nla_total_size(0); /* MDBA_ROUTER */
54 
55 	rcu_read_lock();
56 	hlist_for_each_entry_rcu(pmctx, &brmctx->ip4_mc_router_list,
57 				 ip4_rlist)
58 		size += __br_rports_one_size();
59 
60 #if IS_ENABLED(CONFIG_IPV6)
61 	hlist_for_each_entry_rcu(pmctx, &brmctx->ip6_mc_router_list,
62 				 ip6_rlist)
63 		size += __br_rports_one_size();
64 #endif
65 	rcu_read_unlock();
66 
67 	return size;
68 }
69 
70 int br_rports_fill_info(struct sk_buff *skb,
71 			const struct net_bridge_mcast *brmctx)
72 {
73 	u16 vid = brmctx->vlan ? brmctx->vlan->vid : 0;
74 	bool have_ip4_mc_rtr, have_ip6_mc_rtr;
75 	unsigned long ip4_timer, ip6_timer;
76 	struct nlattr *nest, *port_nest;
77 	struct net_bridge_port *p;
78 
79 	if (!brmctx->multicast_router || !br_rports_have_mc_router(brmctx))
80 		return 0;
81 
82 	nest = nla_nest_start_noflag(skb, MDBA_ROUTER);
83 	if (nest == NULL)
84 		return -EMSGSIZE;
85 
86 	list_for_each_entry_rcu(p, &brmctx->br->port_list, list) {
87 		struct net_bridge_mcast_port *pmctx;
88 
89 		if (vid) {
90 			struct net_bridge_vlan *v;
91 
92 			v = br_vlan_find(nbp_vlan_group(p), vid);
93 			if (!v)
94 				continue;
95 			pmctx = &v->port_mcast_ctx;
96 		} else {
97 			pmctx = &p->multicast_ctx;
98 		}
99 
100 		have_ip4_mc_rtr = br_ip4_rports_get_timer(pmctx, &ip4_timer);
101 		have_ip6_mc_rtr = br_ip6_rports_get_timer(pmctx, &ip6_timer);
102 
103 		if (!have_ip4_mc_rtr && !have_ip6_mc_rtr)
104 			continue;
105 
106 		port_nest = nla_nest_start_noflag(skb, MDBA_ROUTER_PORT);
107 		if (!port_nest)
108 			goto fail;
109 
110 		if (nla_put_nohdr(skb, sizeof(u32), &p->dev->ifindex) ||
111 		    nla_put_u32(skb, MDBA_ROUTER_PATTR_TIMER,
112 				max(ip4_timer, ip6_timer)) ||
113 		    nla_put_u8(skb, MDBA_ROUTER_PATTR_TYPE,
114 			       p->multicast_ctx.multicast_router) ||
115 		    (have_ip4_mc_rtr &&
116 		     nla_put_u32(skb, MDBA_ROUTER_PATTR_INET_TIMER,
117 				 ip4_timer)) ||
118 		    (have_ip6_mc_rtr &&
119 		     nla_put_u32(skb, MDBA_ROUTER_PATTR_INET6_TIMER,
120 				 ip6_timer)) ||
121 		    (vid && nla_put_u16(skb, MDBA_ROUTER_PATTR_VID, vid))) {
122 			nla_nest_cancel(skb, port_nest);
123 			goto fail;
124 		}
125 		nla_nest_end(skb, port_nest);
126 	}
127 
128 	nla_nest_end(skb, nest);
129 	return 0;
130 fail:
131 	nla_nest_cancel(skb, nest);
132 	return -EMSGSIZE;
133 }
134 
135 static void __mdb_entry_fill_flags(struct br_mdb_entry *e, unsigned char flags)
136 {
137 	e->state = flags & MDB_PG_FLAGS_PERMANENT;
138 	e->flags = 0;
139 	if (flags & MDB_PG_FLAGS_OFFLOAD)
140 		e->flags |= MDB_FLAGS_OFFLOAD;
141 	if (flags & MDB_PG_FLAGS_FAST_LEAVE)
142 		e->flags |= MDB_FLAGS_FAST_LEAVE;
143 	if (flags & MDB_PG_FLAGS_STAR_EXCL)
144 		e->flags |= MDB_FLAGS_STAR_EXCL;
145 	if (flags & MDB_PG_FLAGS_BLOCKED)
146 		e->flags |= MDB_FLAGS_BLOCKED;
147 }
148 
149 static void __mdb_entry_to_br_ip(struct br_mdb_entry *entry, struct br_ip *ip,
150 				 struct nlattr **mdb_attrs)
151 {
152 	memset(ip, 0, sizeof(struct br_ip));
153 	ip->vid = entry->vid;
154 	ip->proto = entry->addr.proto;
155 	switch (ip->proto) {
156 	case htons(ETH_P_IP):
157 		ip->dst.ip4 = entry->addr.u.ip4;
158 		if (mdb_attrs && mdb_attrs[MDBE_ATTR_SOURCE])
159 			ip->src.ip4 = nla_get_in_addr(mdb_attrs[MDBE_ATTR_SOURCE]);
160 		break;
161 #if IS_ENABLED(CONFIG_IPV6)
162 	case htons(ETH_P_IPV6):
163 		ip->dst.ip6 = entry->addr.u.ip6;
164 		if (mdb_attrs && mdb_attrs[MDBE_ATTR_SOURCE])
165 			ip->src.ip6 = nla_get_in6_addr(mdb_attrs[MDBE_ATTR_SOURCE]);
166 		break;
167 #endif
168 	default:
169 		ether_addr_copy(ip->dst.mac_addr, entry->addr.u.mac_addr);
170 	}
171 
172 }
173 
174 static int __mdb_fill_srcs(struct sk_buff *skb,
175 			   struct net_bridge_port_group *p)
176 {
177 	struct net_bridge_group_src *ent;
178 	struct nlattr *nest, *nest_ent;
179 
180 	if (hlist_empty(&p->src_list))
181 		return 0;
182 
183 	nest = nla_nest_start(skb, MDBA_MDB_EATTR_SRC_LIST);
184 	if (!nest)
185 		return -EMSGSIZE;
186 
187 	hlist_for_each_entry_rcu(ent, &p->src_list, node,
188 				 lockdep_is_held(&p->key.port->br->multicast_lock)) {
189 		nest_ent = nla_nest_start(skb, MDBA_MDB_SRCLIST_ENTRY);
190 		if (!nest_ent)
191 			goto out_cancel_err;
192 		switch (ent->addr.proto) {
193 		case htons(ETH_P_IP):
194 			if (nla_put_in_addr(skb, MDBA_MDB_SRCATTR_ADDRESS,
195 					    ent->addr.src.ip4)) {
196 				nla_nest_cancel(skb, nest_ent);
197 				goto out_cancel_err;
198 			}
199 			break;
200 #if IS_ENABLED(CONFIG_IPV6)
201 		case htons(ETH_P_IPV6):
202 			if (nla_put_in6_addr(skb, MDBA_MDB_SRCATTR_ADDRESS,
203 					     &ent->addr.src.ip6)) {
204 				nla_nest_cancel(skb, nest_ent);
205 				goto out_cancel_err;
206 			}
207 			break;
208 #endif
209 		default:
210 			nla_nest_cancel(skb, nest_ent);
211 			continue;
212 		}
213 		if (nla_put_u32(skb, MDBA_MDB_SRCATTR_TIMER,
214 				br_timer_value(&ent->timer))) {
215 			nla_nest_cancel(skb, nest_ent);
216 			goto out_cancel_err;
217 		}
218 		nla_nest_end(skb, nest_ent);
219 	}
220 
221 	nla_nest_end(skb, nest);
222 
223 	return 0;
224 
225 out_cancel_err:
226 	nla_nest_cancel(skb, nest);
227 	return -EMSGSIZE;
228 }
229 
230 static int __mdb_fill_info(struct sk_buff *skb,
231 			   struct net_bridge_mdb_entry *mp,
232 			   struct net_bridge_port_group *p)
233 {
234 	bool dump_srcs_mode = false;
235 	struct timer_list *mtimer;
236 	struct nlattr *nest_ent;
237 	struct br_mdb_entry e;
238 	u8 flags = 0;
239 	int ifindex;
240 
241 	memset(&e, 0, sizeof(e));
242 	if (p) {
243 		ifindex = p->key.port->dev->ifindex;
244 		mtimer = &p->timer;
245 		flags = p->flags;
246 	} else {
247 		ifindex = mp->br->dev->ifindex;
248 		mtimer = &mp->timer;
249 	}
250 
251 	__mdb_entry_fill_flags(&e, flags);
252 	e.ifindex = ifindex;
253 	e.vid = mp->addr.vid;
254 	if (mp->addr.proto == htons(ETH_P_IP)) {
255 		e.addr.u.ip4 = mp->addr.dst.ip4;
256 #if IS_ENABLED(CONFIG_IPV6)
257 	} else if (mp->addr.proto == htons(ETH_P_IPV6)) {
258 		e.addr.u.ip6 = mp->addr.dst.ip6;
259 #endif
260 	} else {
261 		ether_addr_copy(e.addr.u.mac_addr, mp->addr.dst.mac_addr);
262 		e.state = MDB_PERMANENT;
263 	}
264 	e.addr.proto = mp->addr.proto;
265 	nest_ent = nla_nest_start_noflag(skb,
266 					 MDBA_MDB_ENTRY_INFO);
267 	if (!nest_ent)
268 		return -EMSGSIZE;
269 
270 	if (nla_put_nohdr(skb, sizeof(e), &e) ||
271 	    nla_put_u32(skb,
272 			MDBA_MDB_EATTR_TIMER,
273 			br_timer_value(mtimer)))
274 		goto nest_err;
275 
276 	switch (mp->addr.proto) {
277 	case htons(ETH_P_IP):
278 		dump_srcs_mode = !!(mp->br->multicast_ctx.multicast_igmp_version == 3);
279 		if (mp->addr.src.ip4) {
280 			if (nla_put_in_addr(skb, MDBA_MDB_EATTR_SOURCE,
281 					    mp->addr.src.ip4))
282 				goto nest_err;
283 			break;
284 		}
285 		break;
286 #if IS_ENABLED(CONFIG_IPV6)
287 	case htons(ETH_P_IPV6):
288 		dump_srcs_mode = !!(mp->br->multicast_ctx.multicast_mld_version == 2);
289 		if (!ipv6_addr_any(&mp->addr.src.ip6)) {
290 			if (nla_put_in6_addr(skb, MDBA_MDB_EATTR_SOURCE,
291 					     &mp->addr.src.ip6))
292 				goto nest_err;
293 			break;
294 		}
295 		break;
296 #endif
297 	default:
298 		ether_addr_copy(e.addr.u.mac_addr, mp->addr.dst.mac_addr);
299 	}
300 	if (p) {
301 		if (nla_put_u8(skb, MDBA_MDB_EATTR_RTPROT, p->rt_protocol))
302 			goto nest_err;
303 		if (dump_srcs_mode &&
304 		    (__mdb_fill_srcs(skb, p) ||
305 		     nla_put_u8(skb, MDBA_MDB_EATTR_GROUP_MODE,
306 				p->filter_mode)))
307 			goto nest_err;
308 	}
309 	nla_nest_end(skb, nest_ent);
310 
311 	return 0;
312 
313 nest_err:
314 	nla_nest_cancel(skb, nest_ent);
315 	return -EMSGSIZE;
316 }
317 
318 static int br_mdb_fill_info(struct sk_buff *skb, struct netlink_callback *cb,
319 			    struct net_device *dev)
320 {
321 	int idx = 0, s_idx = cb->args[1], err = 0, pidx = 0, s_pidx = cb->args[2];
322 	struct net_bridge *br = netdev_priv(dev);
323 	struct net_bridge_mdb_entry *mp;
324 	struct nlattr *nest, *nest2;
325 
326 	if (!br_opt_get(br, BROPT_MULTICAST_ENABLED))
327 		return 0;
328 
329 	nest = nla_nest_start_noflag(skb, MDBA_MDB);
330 	if (nest == NULL)
331 		return -EMSGSIZE;
332 
333 	hlist_for_each_entry_rcu(mp, &br->mdb_list, mdb_node) {
334 		struct net_bridge_port_group *p;
335 		struct net_bridge_port_group __rcu **pp;
336 
337 		if (idx < s_idx)
338 			goto skip;
339 
340 		nest2 = nla_nest_start_noflag(skb, MDBA_MDB_ENTRY);
341 		if (!nest2) {
342 			err = -EMSGSIZE;
343 			break;
344 		}
345 
346 		if (!s_pidx && mp->host_joined) {
347 			err = __mdb_fill_info(skb, mp, NULL);
348 			if (err) {
349 				nla_nest_cancel(skb, nest2);
350 				break;
351 			}
352 		}
353 
354 		for (pp = &mp->ports; (p = rcu_dereference(*pp)) != NULL;
355 		      pp = &p->next) {
356 			if (!p->key.port)
357 				continue;
358 			if (pidx < s_pidx)
359 				goto skip_pg;
360 
361 			err = __mdb_fill_info(skb, mp, p);
362 			if (err) {
363 				nla_nest_end(skb, nest2);
364 				goto out;
365 			}
366 skip_pg:
367 			pidx++;
368 		}
369 		pidx = 0;
370 		s_pidx = 0;
371 		nla_nest_end(skb, nest2);
372 skip:
373 		idx++;
374 	}
375 
376 out:
377 	cb->args[1] = idx;
378 	cb->args[2] = pidx;
379 	nla_nest_end(skb, nest);
380 	return err;
381 }
382 
383 int br_mdb_dump(struct net_device *dev, struct sk_buff *skb,
384 		struct netlink_callback *cb)
385 {
386 	struct net_bridge *br = netdev_priv(dev);
387 	struct br_port_msg *bpm;
388 	struct nlmsghdr *nlh;
389 	int err;
390 
391 	nlh = nlmsg_put(skb, NETLINK_CB(cb->skb).portid,
392 			cb->nlh->nlmsg_seq, RTM_GETMDB, sizeof(*bpm),
393 			NLM_F_MULTI);
394 	if (!nlh)
395 		return -EMSGSIZE;
396 
397 	bpm = nlmsg_data(nlh);
398 	memset(bpm, 0, sizeof(*bpm));
399 	bpm->ifindex = dev->ifindex;
400 
401 	rcu_read_lock();
402 
403 	err = br_mdb_fill_info(skb, cb, dev);
404 	if (err)
405 		goto out;
406 	err = br_rports_fill_info(skb, &br->multicast_ctx);
407 	if (err)
408 		goto out;
409 
410 out:
411 	rcu_read_unlock();
412 	nlmsg_end(skb, nlh);
413 	return err;
414 }
415 
416 static int nlmsg_populate_mdb_fill(struct sk_buff *skb,
417 				   struct net_device *dev,
418 				   struct net_bridge_mdb_entry *mp,
419 				   struct net_bridge_port_group *pg,
420 				   int type)
421 {
422 	struct nlmsghdr *nlh;
423 	struct br_port_msg *bpm;
424 	struct nlattr *nest, *nest2;
425 
426 	nlh = nlmsg_put(skb, 0, 0, type, sizeof(*bpm), 0);
427 	if (!nlh)
428 		return -EMSGSIZE;
429 
430 	bpm = nlmsg_data(nlh);
431 	memset(bpm, 0, sizeof(*bpm));
432 	bpm->family  = AF_BRIDGE;
433 	bpm->ifindex = dev->ifindex;
434 	nest = nla_nest_start_noflag(skb, MDBA_MDB);
435 	if (nest == NULL)
436 		goto cancel;
437 	nest2 = nla_nest_start_noflag(skb, MDBA_MDB_ENTRY);
438 	if (nest2 == NULL)
439 		goto end;
440 
441 	if (__mdb_fill_info(skb, mp, pg))
442 		goto end;
443 
444 	nla_nest_end(skb, nest2);
445 	nla_nest_end(skb, nest);
446 	nlmsg_end(skb, nlh);
447 	return 0;
448 
449 end:
450 	nla_nest_end(skb, nest);
451 cancel:
452 	nlmsg_cancel(skb, nlh);
453 	return -EMSGSIZE;
454 }
455 
456 static size_t rtnl_mdb_nlmsg_size(struct net_bridge_port_group *pg)
457 {
458 	size_t nlmsg_size = NLMSG_ALIGN(sizeof(struct br_port_msg)) +
459 			    nla_total_size(sizeof(struct br_mdb_entry)) +
460 			    nla_total_size(sizeof(u32));
461 	struct net_bridge_group_src *ent;
462 	size_t addr_size = 0;
463 
464 	if (!pg)
465 		goto out;
466 
467 	/* MDBA_MDB_EATTR_RTPROT */
468 	nlmsg_size += nla_total_size(sizeof(u8));
469 
470 	switch (pg->key.addr.proto) {
471 	case htons(ETH_P_IP):
472 		/* MDBA_MDB_EATTR_SOURCE */
473 		if (pg->key.addr.src.ip4)
474 			nlmsg_size += nla_total_size(sizeof(__be32));
475 		if (pg->key.port->br->multicast_ctx.multicast_igmp_version == 2)
476 			goto out;
477 		addr_size = sizeof(__be32);
478 		break;
479 #if IS_ENABLED(CONFIG_IPV6)
480 	case htons(ETH_P_IPV6):
481 		/* MDBA_MDB_EATTR_SOURCE */
482 		if (!ipv6_addr_any(&pg->key.addr.src.ip6))
483 			nlmsg_size += nla_total_size(sizeof(struct in6_addr));
484 		if (pg->key.port->br->multicast_ctx.multicast_mld_version == 1)
485 			goto out;
486 		addr_size = sizeof(struct in6_addr);
487 		break;
488 #endif
489 	}
490 
491 	/* MDBA_MDB_EATTR_GROUP_MODE */
492 	nlmsg_size += nla_total_size(sizeof(u8));
493 
494 	/* MDBA_MDB_EATTR_SRC_LIST nested attr */
495 	if (!hlist_empty(&pg->src_list))
496 		nlmsg_size += nla_total_size(0);
497 
498 	hlist_for_each_entry(ent, &pg->src_list, node) {
499 		/* MDBA_MDB_SRCLIST_ENTRY nested attr +
500 		 * MDBA_MDB_SRCATTR_ADDRESS + MDBA_MDB_SRCATTR_TIMER
501 		 */
502 		nlmsg_size += nla_total_size(0) +
503 			      nla_total_size(addr_size) +
504 			      nla_total_size(sizeof(u32));
505 	}
506 out:
507 	return nlmsg_size;
508 }
509 
510 void br_mdb_notify(struct net_device *dev,
511 		   struct net_bridge_mdb_entry *mp,
512 		   struct net_bridge_port_group *pg,
513 		   int type)
514 {
515 	struct net *net = dev_net(dev);
516 	struct sk_buff *skb;
517 	int err = -ENOBUFS;
518 
519 	br_switchdev_mdb_notify(dev, mp, pg, type);
520 
521 	skb = nlmsg_new(rtnl_mdb_nlmsg_size(pg), GFP_ATOMIC);
522 	if (!skb)
523 		goto errout;
524 
525 	err = nlmsg_populate_mdb_fill(skb, dev, mp, pg, type);
526 	if (err < 0) {
527 		kfree_skb(skb);
528 		goto errout;
529 	}
530 
531 	rtnl_notify(skb, net, 0, RTNLGRP_MDB, NULL, GFP_ATOMIC);
532 	return;
533 errout:
534 	rtnl_set_sk_err(net, RTNLGRP_MDB, err);
535 }
536 
537 static int nlmsg_populate_rtr_fill(struct sk_buff *skb,
538 				   struct net_device *dev,
539 				   int ifindex, u16 vid, u32 pid,
540 				   u32 seq, int type, unsigned int flags)
541 {
542 	struct nlattr *nest, *port_nest;
543 	struct br_port_msg *bpm;
544 	struct nlmsghdr *nlh;
545 
546 	nlh = nlmsg_put(skb, pid, seq, type, sizeof(*bpm), 0);
547 	if (!nlh)
548 		return -EMSGSIZE;
549 
550 	bpm = nlmsg_data(nlh);
551 	memset(bpm, 0, sizeof(*bpm));
552 	bpm->family = AF_BRIDGE;
553 	bpm->ifindex = dev->ifindex;
554 	nest = nla_nest_start_noflag(skb, MDBA_ROUTER);
555 	if (!nest)
556 		goto cancel;
557 
558 	port_nest = nla_nest_start_noflag(skb, MDBA_ROUTER_PORT);
559 	if (!port_nest)
560 		goto end;
561 	if (nla_put_nohdr(skb, sizeof(u32), &ifindex)) {
562 		nla_nest_cancel(skb, port_nest);
563 		goto end;
564 	}
565 	if (vid && nla_put_u16(skb, MDBA_ROUTER_PATTR_VID, vid)) {
566 		nla_nest_cancel(skb, port_nest);
567 		goto end;
568 	}
569 	nla_nest_end(skb, port_nest);
570 
571 	nla_nest_end(skb, nest);
572 	nlmsg_end(skb, nlh);
573 	return 0;
574 
575 end:
576 	nla_nest_end(skb, nest);
577 cancel:
578 	nlmsg_cancel(skb, nlh);
579 	return -EMSGSIZE;
580 }
581 
582 static inline size_t rtnl_rtr_nlmsg_size(void)
583 {
584 	return NLMSG_ALIGN(sizeof(struct br_port_msg))
585 		+ nla_total_size(sizeof(__u32))
586 		+ nla_total_size(sizeof(u16));
587 }
588 
589 void br_rtr_notify(struct net_device *dev, struct net_bridge_mcast_port *pmctx,
590 		   int type)
591 {
592 	struct net *net = dev_net(dev);
593 	struct sk_buff *skb;
594 	int err = -ENOBUFS;
595 	int ifindex;
596 	u16 vid;
597 
598 	ifindex = pmctx ? pmctx->port->dev->ifindex : 0;
599 	vid = pmctx && br_multicast_port_ctx_is_vlan(pmctx) ? pmctx->vlan->vid :
600 							      0;
601 	skb = nlmsg_new(rtnl_rtr_nlmsg_size(), GFP_ATOMIC);
602 	if (!skb)
603 		goto errout;
604 
605 	err = nlmsg_populate_rtr_fill(skb, dev, ifindex, vid, 0, 0, type,
606 				      NTF_SELF);
607 	if (err < 0) {
608 		kfree_skb(skb);
609 		goto errout;
610 	}
611 
612 	rtnl_notify(skb, net, 0, RTNLGRP_MDB, NULL, GFP_ATOMIC);
613 	return;
614 
615 errout:
616 	rtnl_set_sk_err(net, RTNLGRP_MDB, err);
617 }
618 
619 static const struct nla_policy
620 br_mdbe_src_list_entry_pol[MDBE_SRCATTR_MAX + 1] = {
621 	[MDBE_SRCATTR_ADDRESS] = NLA_POLICY_RANGE(NLA_BINARY,
622 						  sizeof(struct in_addr),
623 						  sizeof(struct in6_addr)),
624 };
625 
626 static const struct nla_policy
627 br_mdbe_src_list_pol[MDBE_SRC_LIST_MAX + 1] = {
628 	[MDBE_SRC_LIST_ENTRY] = NLA_POLICY_NESTED(br_mdbe_src_list_entry_pol),
629 };
630 
631 static const struct nla_policy br_mdbe_attrs_pol[MDBE_ATTR_MAX + 1] = {
632 	[MDBE_ATTR_SOURCE] = NLA_POLICY_RANGE(NLA_BINARY,
633 					      sizeof(struct in_addr),
634 					      sizeof(struct in6_addr)),
635 	[MDBE_ATTR_GROUP_MODE] = NLA_POLICY_RANGE(NLA_U8, MCAST_EXCLUDE,
636 						  MCAST_INCLUDE),
637 	[MDBE_ATTR_SRC_LIST] = NLA_POLICY_NESTED(br_mdbe_src_list_pol),
638 	[MDBE_ATTR_RTPROT] = NLA_POLICY_MIN(NLA_U8, RTPROT_STATIC),
639 };
640 
641 static bool is_valid_mdb_source(struct nlattr *attr, __be16 proto,
642 				struct netlink_ext_ack *extack)
643 {
644 	switch (proto) {
645 	case htons(ETH_P_IP):
646 		if (nla_len(attr) != sizeof(struct in_addr)) {
647 			NL_SET_ERR_MSG_MOD(extack, "IPv4 invalid source address length");
648 			return false;
649 		}
650 		if (ipv4_is_multicast(nla_get_in_addr(attr))) {
651 			NL_SET_ERR_MSG_MOD(extack, "IPv4 multicast source address is not allowed");
652 			return false;
653 		}
654 		break;
655 #if IS_ENABLED(CONFIG_IPV6)
656 	case htons(ETH_P_IPV6): {
657 		struct in6_addr src;
658 
659 		if (nla_len(attr) != sizeof(struct in6_addr)) {
660 			NL_SET_ERR_MSG_MOD(extack, "IPv6 invalid source address length");
661 			return false;
662 		}
663 		src = nla_get_in6_addr(attr);
664 		if (ipv6_addr_is_multicast(&src)) {
665 			NL_SET_ERR_MSG_MOD(extack, "IPv6 multicast source address is not allowed");
666 			return false;
667 		}
668 		break;
669 	}
670 #endif
671 	default:
672 		NL_SET_ERR_MSG_MOD(extack, "Invalid protocol used with source address");
673 		return false;
674 	}
675 
676 	return true;
677 }
678 
679 static struct net_bridge_mcast *
680 __br_mdb_choose_context(struct net_bridge *br,
681 			const struct br_mdb_entry *entry,
682 			struct netlink_ext_ack *extack)
683 {
684 	struct net_bridge_mcast *brmctx = NULL;
685 	struct net_bridge_vlan *v;
686 
687 	if (!br_opt_get(br, BROPT_MCAST_VLAN_SNOOPING_ENABLED)) {
688 		brmctx = &br->multicast_ctx;
689 		goto out;
690 	}
691 
692 	if (!entry->vid) {
693 		NL_SET_ERR_MSG_MOD(extack, "Cannot add an entry without a vlan when vlan snooping is enabled");
694 		goto out;
695 	}
696 
697 	v = br_vlan_find(br_vlan_group(br), entry->vid);
698 	if (!v) {
699 		NL_SET_ERR_MSG_MOD(extack, "Vlan is not configured");
700 		goto out;
701 	}
702 	if (br_multicast_ctx_vlan_global_disabled(&v->br_mcast_ctx)) {
703 		NL_SET_ERR_MSG_MOD(extack, "Vlan's multicast processing is disabled");
704 		goto out;
705 	}
706 	brmctx = &v->br_mcast_ctx;
707 out:
708 	return brmctx;
709 }
710 
711 static int br_mdb_replace_group_sg(const struct br_mdb_config *cfg,
712 				   struct net_bridge_mdb_entry *mp,
713 				   struct net_bridge_port_group *pg,
714 				   struct net_bridge_mcast *brmctx,
715 				   unsigned char flags)
716 {
717 	unsigned long now = jiffies;
718 
719 	pg->flags = flags;
720 	pg->rt_protocol = cfg->rt_protocol;
721 	if (!(flags & MDB_PG_FLAGS_PERMANENT) && !cfg->src_entry)
722 		mod_timer(&pg->timer,
723 			  now + brmctx->multicast_membership_interval);
724 	else
725 		del_timer(&pg->timer);
726 
727 	br_mdb_notify(cfg->br->dev, mp, pg, RTM_NEWMDB);
728 
729 	return 0;
730 }
731 
732 static int br_mdb_add_group_sg(const struct br_mdb_config *cfg,
733 			       struct net_bridge_mdb_entry *mp,
734 			       struct net_bridge_mcast *brmctx,
735 			       unsigned char flags,
736 			       struct netlink_ext_ack *extack)
737 {
738 	struct net_bridge_port_group __rcu **pp;
739 	struct net_bridge_port_group *p;
740 	unsigned long now = jiffies;
741 
742 	for (pp = &mp->ports;
743 	     (p = mlock_dereference(*pp, cfg->br)) != NULL;
744 	     pp = &p->next) {
745 		if (p->key.port == cfg->p) {
746 			if (!(cfg->nlflags & NLM_F_REPLACE)) {
747 				NL_SET_ERR_MSG_MOD(extack, "(S, G) group is already joined by port");
748 				return -EEXIST;
749 			}
750 			return br_mdb_replace_group_sg(cfg, mp, p, brmctx,
751 						       flags);
752 		}
753 		if ((unsigned long)p->key.port < (unsigned long)cfg->p)
754 			break;
755 	}
756 
757 	p = br_multicast_new_port_group(cfg->p, &cfg->group, *pp, flags, NULL,
758 					MCAST_INCLUDE, cfg->rt_protocol, extack);
759 	if (unlikely(!p))
760 		return -ENOMEM;
761 
762 	rcu_assign_pointer(*pp, p);
763 	if (!(flags & MDB_PG_FLAGS_PERMANENT) && !cfg->src_entry)
764 		mod_timer(&p->timer,
765 			  now + brmctx->multicast_membership_interval);
766 	br_mdb_notify(cfg->br->dev, mp, p, RTM_NEWMDB);
767 
768 	/* All of (*, G) EXCLUDE ports need to be added to the new (S, G) for
769 	 * proper replication.
770 	 */
771 	if (br_multicast_should_handle_mode(brmctx, cfg->group.proto)) {
772 		struct net_bridge_mdb_entry *star_mp;
773 		struct br_ip star_group;
774 
775 		star_group = p->key.addr;
776 		memset(&star_group.src, 0, sizeof(star_group.src));
777 		star_mp = br_mdb_ip_get(cfg->br, &star_group);
778 		if (star_mp)
779 			br_multicast_sg_add_exclude_ports(star_mp, p);
780 	}
781 
782 	return 0;
783 }
784 
785 static int br_mdb_add_group_src_fwd(const struct br_mdb_config *cfg,
786 				    struct br_ip *src_ip,
787 				    struct net_bridge_mcast *brmctx,
788 				    struct netlink_ext_ack *extack)
789 {
790 	struct net_bridge_mdb_entry *sgmp;
791 	struct br_mdb_config sg_cfg;
792 	struct br_ip sg_ip;
793 	u8 flags = 0;
794 
795 	sg_ip = cfg->group;
796 	sg_ip.src = src_ip->src;
797 	sgmp = br_multicast_new_group(cfg->br, &sg_ip);
798 	if (IS_ERR(sgmp)) {
799 		NL_SET_ERR_MSG_MOD(extack, "Failed to add (S, G) MDB entry");
800 		return PTR_ERR(sgmp);
801 	}
802 
803 	if (cfg->entry->state == MDB_PERMANENT)
804 		flags |= MDB_PG_FLAGS_PERMANENT;
805 	if (cfg->filter_mode == MCAST_EXCLUDE)
806 		flags |= MDB_PG_FLAGS_BLOCKED;
807 
808 	memset(&sg_cfg, 0, sizeof(sg_cfg));
809 	sg_cfg.br = cfg->br;
810 	sg_cfg.p = cfg->p;
811 	sg_cfg.entry = cfg->entry;
812 	sg_cfg.group = sg_ip;
813 	sg_cfg.src_entry = true;
814 	sg_cfg.filter_mode = MCAST_INCLUDE;
815 	sg_cfg.rt_protocol = cfg->rt_protocol;
816 	sg_cfg.nlflags = cfg->nlflags;
817 	return br_mdb_add_group_sg(&sg_cfg, sgmp, brmctx, flags, extack);
818 }
819 
820 static int br_mdb_add_group_src(const struct br_mdb_config *cfg,
821 				struct net_bridge_port_group *pg,
822 				struct net_bridge_mcast *brmctx,
823 				struct br_mdb_src_entry *src,
824 				struct netlink_ext_ack *extack)
825 {
826 	struct net_bridge_group_src *ent;
827 	unsigned long now = jiffies;
828 	int err;
829 
830 	ent = br_multicast_find_group_src(pg, &src->addr);
831 	if (!ent) {
832 		ent = br_multicast_new_group_src(pg, &src->addr);
833 		if (!ent) {
834 			NL_SET_ERR_MSG_MOD(extack, "Failed to add new source entry");
835 			return -ENOSPC;
836 		}
837 	} else if (!(cfg->nlflags & NLM_F_REPLACE)) {
838 		NL_SET_ERR_MSG_MOD(extack, "Source entry already exists");
839 		return -EEXIST;
840 	}
841 
842 	if (cfg->filter_mode == MCAST_INCLUDE &&
843 	    cfg->entry->state == MDB_TEMPORARY)
844 		mod_timer(&ent->timer, now + br_multicast_gmi(brmctx));
845 	else
846 		del_timer(&ent->timer);
847 
848 	/* Install a (S, G) forwarding entry for the source. */
849 	err = br_mdb_add_group_src_fwd(cfg, &src->addr, brmctx, extack);
850 	if (err)
851 		goto err_del_sg;
852 
853 	ent->flags = BR_SGRP_F_INSTALLED | BR_SGRP_F_USER_ADDED;
854 
855 	return 0;
856 
857 err_del_sg:
858 	__br_multicast_del_group_src(ent);
859 	return err;
860 }
861 
862 static void br_mdb_del_group_src(struct net_bridge_port_group *pg,
863 				 struct br_mdb_src_entry *src)
864 {
865 	struct net_bridge_group_src *ent;
866 
867 	ent = br_multicast_find_group_src(pg, &src->addr);
868 	if (WARN_ON_ONCE(!ent))
869 		return;
870 	br_multicast_del_group_src(ent, false);
871 }
872 
873 static int br_mdb_add_group_srcs(const struct br_mdb_config *cfg,
874 				 struct net_bridge_port_group *pg,
875 				 struct net_bridge_mcast *brmctx,
876 				 struct netlink_ext_ack *extack)
877 {
878 	int i, err;
879 
880 	for (i = 0; i < cfg->num_src_entries; i++) {
881 		err = br_mdb_add_group_src(cfg, pg, brmctx,
882 					   &cfg->src_entries[i], extack);
883 		if (err)
884 			goto err_del_group_srcs;
885 	}
886 
887 	return 0;
888 
889 err_del_group_srcs:
890 	for (i--; i >= 0; i--)
891 		br_mdb_del_group_src(pg, &cfg->src_entries[i]);
892 	return err;
893 }
894 
895 static int br_mdb_replace_group_srcs(const struct br_mdb_config *cfg,
896 				     struct net_bridge_port_group *pg,
897 				     struct net_bridge_mcast *brmctx,
898 				     struct netlink_ext_ack *extack)
899 {
900 	struct net_bridge_group_src *ent;
901 	struct hlist_node *tmp;
902 	int err;
903 
904 	hlist_for_each_entry(ent, &pg->src_list, node)
905 		ent->flags |= BR_SGRP_F_DELETE;
906 
907 	err = br_mdb_add_group_srcs(cfg, pg, brmctx, extack);
908 	if (err)
909 		goto err_clear_delete;
910 
911 	hlist_for_each_entry_safe(ent, tmp, &pg->src_list, node) {
912 		if (ent->flags & BR_SGRP_F_DELETE)
913 			br_multicast_del_group_src(ent, false);
914 	}
915 
916 	return 0;
917 
918 err_clear_delete:
919 	hlist_for_each_entry(ent, &pg->src_list, node)
920 		ent->flags &= ~BR_SGRP_F_DELETE;
921 	return err;
922 }
923 
924 static int br_mdb_replace_group_star_g(const struct br_mdb_config *cfg,
925 				       struct net_bridge_mdb_entry *mp,
926 				       struct net_bridge_port_group *pg,
927 				       struct net_bridge_mcast *brmctx,
928 				       unsigned char flags,
929 				       struct netlink_ext_ack *extack)
930 {
931 	unsigned long now = jiffies;
932 	int err;
933 
934 	err = br_mdb_replace_group_srcs(cfg, pg, brmctx, extack);
935 	if (err)
936 		return err;
937 
938 	pg->flags = flags;
939 	pg->filter_mode = cfg->filter_mode;
940 	pg->rt_protocol = cfg->rt_protocol;
941 	if (!(flags & MDB_PG_FLAGS_PERMANENT) &&
942 	    cfg->filter_mode == MCAST_EXCLUDE)
943 		mod_timer(&pg->timer,
944 			  now + brmctx->multicast_membership_interval);
945 	else
946 		del_timer(&pg->timer);
947 
948 	br_mdb_notify(cfg->br->dev, mp, pg, RTM_NEWMDB);
949 
950 	if (br_multicast_should_handle_mode(brmctx, cfg->group.proto))
951 		br_multicast_star_g_handle_mode(pg, cfg->filter_mode);
952 
953 	return 0;
954 }
955 
956 static int br_mdb_add_group_star_g(const struct br_mdb_config *cfg,
957 				   struct net_bridge_mdb_entry *mp,
958 				   struct net_bridge_mcast *brmctx,
959 				   unsigned char flags,
960 				   struct netlink_ext_ack *extack)
961 {
962 	struct net_bridge_port_group __rcu **pp;
963 	struct net_bridge_port_group *p;
964 	unsigned long now = jiffies;
965 	int err;
966 
967 	for (pp = &mp->ports;
968 	     (p = mlock_dereference(*pp, cfg->br)) != NULL;
969 	     pp = &p->next) {
970 		if (p->key.port == cfg->p) {
971 			if (!(cfg->nlflags & NLM_F_REPLACE)) {
972 				NL_SET_ERR_MSG_MOD(extack, "(*, G) group is already joined by port");
973 				return -EEXIST;
974 			}
975 			return br_mdb_replace_group_star_g(cfg, mp, p, brmctx,
976 							   flags, extack);
977 		}
978 		if ((unsigned long)p->key.port < (unsigned long)cfg->p)
979 			break;
980 	}
981 
982 	p = br_multicast_new_port_group(cfg->p, &cfg->group, *pp, flags, NULL,
983 					cfg->filter_mode, cfg->rt_protocol,
984 					extack);
985 	if (unlikely(!p))
986 		return -ENOMEM;
987 
988 	err = br_mdb_add_group_srcs(cfg, p, brmctx, extack);
989 	if (err)
990 		goto err_del_port_group;
991 
992 	rcu_assign_pointer(*pp, p);
993 	if (!(flags & MDB_PG_FLAGS_PERMANENT) &&
994 	    cfg->filter_mode == MCAST_EXCLUDE)
995 		mod_timer(&p->timer,
996 			  now + brmctx->multicast_membership_interval);
997 	br_mdb_notify(cfg->br->dev, mp, p, RTM_NEWMDB);
998 	/* If we are adding a new EXCLUDE port group (*, G), it needs to be
999 	 * also added to all (S, G) entries for proper replication.
1000 	 */
1001 	if (br_multicast_should_handle_mode(brmctx, cfg->group.proto) &&
1002 	    cfg->filter_mode == MCAST_EXCLUDE)
1003 		br_multicast_star_g_handle_mode(p, MCAST_EXCLUDE);
1004 
1005 	return 0;
1006 
1007 err_del_port_group:
1008 	br_multicast_del_port_group(p);
1009 	return err;
1010 }
1011 
1012 static int br_mdb_add_group(const struct br_mdb_config *cfg,
1013 			    struct netlink_ext_ack *extack)
1014 {
1015 	struct br_mdb_entry *entry = cfg->entry;
1016 	struct net_bridge_port *port = cfg->p;
1017 	struct net_bridge_mdb_entry *mp;
1018 	struct net_bridge *br = cfg->br;
1019 	struct net_bridge_mcast *brmctx;
1020 	struct br_ip group = cfg->group;
1021 	unsigned char flags = 0;
1022 
1023 	brmctx = __br_mdb_choose_context(br, entry, extack);
1024 	if (!brmctx)
1025 		return -EINVAL;
1026 
1027 	mp = br_multicast_new_group(br, &group);
1028 	if (IS_ERR(mp))
1029 		return PTR_ERR(mp);
1030 
1031 	/* host join */
1032 	if (!port) {
1033 		if (mp->host_joined) {
1034 			NL_SET_ERR_MSG_MOD(extack, "Group is already joined by host");
1035 			return -EEXIST;
1036 		}
1037 
1038 		br_multicast_host_join(brmctx, mp, false);
1039 		br_mdb_notify(br->dev, mp, NULL, RTM_NEWMDB);
1040 
1041 		return 0;
1042 	}
1043 
1044 	if (entry->state == MDB_PERMANENT)
1045 		flags |= MDB_PG_FLAGS_PERMANENT;
1046 
1047 	if (br_multicast_is_star_g(&group))
1048 		return br_mdb_add_group_star_g(cfg, mp, brmctx, flags, extack);
1049 	else
1050 		return br_mdb_add_group_sg(cfg, mp, brmctx, flags, extack);
1051 }
1052 
1053 static int __br_mdb_add(const struct br_mdb_config *cfg,
1054 			struct netlink_ext_ack *extack)
1055 {
1056 	int ret;
1057 
1058 	spin_lock_bh(&cfg->br->multicast_lock);
1059 	ret = br_mdb_add_group(cfg, extack);
1060 	spin_unlock_bh(&cfg->br->multicast_lock);
1061 
1062 	return ret;
1063 }
1064 
1065 static int br_mdb_config_src_entry_init(struct nlattr *src_entry,
1066 					struct br_mdb_src_entry *src,
1067 					__be16 proto,
1068 					struct netlink_ext_ack *extack)
1069 {
1070 	struct nlattr *tb[MDBE_SRCATTR_MAX + 1];
1071 	int err;
1072 
1073 	err = nla_parse_nested(tb, MDBE_SRCATTR_MAX, src_entry,
1074 			       br_mdbe_src_list_entry_pol, extack);
1075 	if (err)
1076 		return err;
1077 
1078 	if (NL_REQ_ATTR_CHECK(extack, src_entry, tb, MDBE_SRCATTR_ADDRESS))
1079 		return -EINVAL;
1080 
1081 	if (!is_valid_mdb_source(tb[MDBE_SRCATTR_ADDRESS], proto, extack))
1082 		return -EINVAL;
1083 
1084 	src->addr.proto = proto;
1085 	nla_memcpy(&src->addr.src, tb[MDBE_SRCATTR_ADDRESS],
1086 		   nla_len(tb[MDBE_SRCATTR_ADDRESS]));
1087 
1088 	return 0;
1089 }
1090 
1091 static int br_mdb_config_src_list_init(struct nlattr *src_list,
1092 				       struct br_mdb_config *cfg,
1093 				       struct netlink_ext_ack *extack)
1094 {
1095 	struct nlattr *src_entry;
1096 	int rem, err;
1097 	int i = 0;
1098 
1099 	nla_for_each_nested(src_entry, src_list, rem)
1100 		cfg->num_src_entries++;
1101 
1102 	if (cfg->num_src_entries >= PG_SRC_ENT_LIMIT) {
1103 		NL_SET_ERR_MSG_FMT_MOD(extack, "Exceeded maximum number of source entries (%u)",
1104 				       PG_SRC_ENT_LIMIT - 1);
1105 		return -EINVAL;
1106 	}
1107 
1108 	cfg->src_entries = kcalloc(cfg->num_src_entries,
1109 				   sizeof(struct br_mdb_src_entry), GFP_KERNEL);
1110 	if (!cfg->src_entries)
1111 		return -ENOMEM;
1112 
1113 	nla_for_each_nested(src_entry, src_list, rem) {
1114 		err = br_mdb_config_src_entry_init(src_entry,
1115 						   &cfg->src_entries[i],
1116 						   cfg->entry->addr.proto,
1117 						   extack);
1118 		if (err)
1119 			goto err_src_entry_init;
1120 		i++;
1121 	}
1122 
1123 	return 0;
1124 
1125 err_src_entry_init:
1126 	kfree(cfg->src_entries);
1127 	return err;
1128 }
1129 
1130 static void br_mdb_config_src_list_fini(struct br_mdb_config *cfg)
1131 {
1132 	kfree(cfg->src_entries);
1133 }
1134 
1135 static int br_mdb_config_attrs_init(struct nlattr *set_attrs,
1136 				    struct br_mdb_config *cfg,
1137 				    struct netlink_ext_ack *extack)
1138 {
1139 	struct nlattr *mdb_attrs[MDBE_ATTR_MAX + 1];
1140 	int err;
1141 
1142 	err = nla_parse_nested(mdb_attrs, MDBE_ATTR_MAX, set_attrs,
1143 			       br_mdbe_attrs_pol, extack);
1144 	if (err)
1145 		return err;
1146 
1147 	if (mdb_attrs[MDBE_ATTR_SOURCE] &&
1148 	    !is_valid_mdb_source(mdb_attrs[MDBE_ATTR_SOURCE],
1149 				 cfg->entry->addr.proto, extack))
1150 		return -EINVAL;
1151 
1152 	__mdb_entry_to_br_ip(cfg->entry, &cfg->group, mdb_attrs);
1153 
1154 	if (mdb_attrs[MDBE_ATTR_GROUP_MODE]) {
1155 		if (!cfg->p) {
1156 			NL_SET_ERR_MSG_MOD(extack, "Filter mode cannot be set for host groups");
1157 			return -EINVAL;
1158 		}
1159 		if (!br_multicast_is_star_g(&cfg->group)) {
1160 			NL_SET_ERR_MSG_MOD(extack, "Filter mode can only be set for (*, G) entries");
1161 			return -EINVAL;
1162 		}
1163 		cfg->filter_mode = nla_get_u8(mdb_attrs[MDBE_ATTR_GROUP_MODE]);
1164 	} else {
1165 		cfg->filter_mode = MCAST_EXCLUDE;
1166 	}
1167 
1168 	if (mdb_attrs[MDBE_ATTR_SRC_LIST]) {
1169 		if (!cfg->p) {
1170 			NL_SET_ERR_MSG_MOD(extack, "Source list cannot be set for host groups");
1171 			return -EINVAL;
1172 		}
1173 		if (!br_multicast_is_star_g(&cfg->group)) {
1174 			NL_SET_ERR_MSG_MOD(extack, "Source list can only be set for (*, G) entries");
1175 			return -EINVAL;
1176 		}
1177 		if (!mdb_attrs[MDBE_ATTR_GROUP_MODE]) {
1178 			NL_SET_ERR_MSG_MOD(extack, "Source list cannot be set without filter mode");
1179 			return -EINVAL;
1180 		}
1181 		err = br_mdb_config_src_list_init(mdb_attrs[MDBE_ATTR_SRC_LIST],
1182 						  cfg, extack);
1183 		if (err)
1184 			return err;
1185 	}
1186 
1187 	if (!cfg->num_src_entries && cfg->filter_mode == MCAST_INCLUDE) {
1188 		NL_SET_ERR_MSG_MOD(extack, "Cannot add (*, G) INCLUDE with an empty source list");
1189 		return -EINVAL;
1190 	}
1191 
1192 	if (mdb_attrs[MDBE_ATTR_RTPROT]) {
1193 		if (!cfg->p) {
1194 			NL_SET_ERR_MSG_MOD(extack, "Protocol cannot be set for host groups");
1195 			return -EINVAL;
1196 		}
1197 		cfg->rt_protocol = nla_get_u8(mdb_attrs[MDBE_ATTR_RTPROT]);
1198 	}
1199 
1200 	return 0;
1201 }
1202 
1203 static int br_mdb_config_init(struct br_mdb_config *cfg, struct net_device *dev,
1204 			      struct nlattr *tb[], u16 nlmsg_flags,
1205 			      struct netlink_ext_ack *extack)
1206 {
1207 	struct net *net = dev_net(dev);
1208 
1209 	memset(cfg, 0, sizeof(*cfg));
1210 	cfg->filter_mode = MCAST_EXCLUDE;
1211 	cfg->rt_protocol = RTPROT_STATIC;
1212 	cfg->nlflags = nlmsg_flags;
1213 
1214 	cfg->br = netdev_priv(dev);
1215 
1216 	if (!netif_running(cfg->br->dev)) {
1217 		NL_SET_ERR_MSG_MOD(extack, "Bridge device is not running");
1218 		return -EINVAL;
1219 	}
1220 
1221 	if (!br_opt_get(cfg->br, BROPT_MULTICAST_ENABLED)) {
1222 		NL_SET_ERR_MSG_MOD(extack, "Bridge's multicast processing is disabled");
1223 		return -EINVAL;
1224 	}
1225 
1226 	cfg->entry = nla_data(tb[MDBA_SET_ENTRY]);
1227 
1228 	if (cfg->entry->ifindex != cfg->br->dev->ifindex) {
1229 		struct net_device *pdev;
1230 
1231 		pdev = __dev_get_by_index(net, cfg->entry->ifindex);
1232 		if (!pdev) {
1233 			NL_SET_ERR_MSG_MOD(extack, "Port net device doesn't exist");
1234 			return -ENODEV;
1235 		}
1236 
1237 		cfg->p = br_port_get_rtnl(pdev);
1238 		if (!cfg->p) {
1239 			NL_SET_ERR_MSG_MOD(extack, "Net device is not a bridge port");
1240 			return -EINVAL;
1241 		}
1242 
1243 		if (cfg->p->br != cfg->br) {
1244 			NL_SET_ERR_MSG_MOD(extack, "Port belongs to a different bridge device");
1245 			return -EINVAL;
1246 		}
1247 	}
1248 
1249 	if (cfg->entry->addr.proto == htons(ETH_P_IP) &&
1250 	    ipv4_is_zeronet(cfg->entry->addr.u.ip4)) {
1251 		NL_SET_ERR_MSG_MOD(extack, "IPv4 entry group address 0.0.0.0 is not allowed");
1252 		return -EINVAL;
1253 	}
1254 
1255 	if (tb[MDBA_SET_ENTRY_ATTRS])
1256 		return br_mdb_config_attrs_init(tb[MDBA_SET_ENTRY_ATTRS], cfg,
1257 						extack);
1258 	else
1259 		__mdb_entry_to_br_ip(cfg->entry, &cfg->group, NULL);
1260 
1261 	return 0;
1262 }
1263 
1264 static void br_mdb_config_fini(struct br_mdb_config *cfg)
1265 {
1266 	br_mdb_config_src_list_fini(cfg);
1267 }
1268 
1269 int br_mdb_add(struct net_device *dev, struct nlattr *tb[], u16 nlmsg_flags,
1270 	       struct netlink_ext_ack *extack)
1271 {
1272 	struct net_bridge_vlan_group *vg;
1273 	struct net_bridge_vlan *v;
1274 	struct br_mdb_config cfg;
1275 	int err;
1276 
1277 	err = br_mdb_config_init(&cfg, dev, tb, nlmsg_flags, extack);
1278 	if (err)
1279 		return err;
1280 
1281 	err = -EINVAL;
1282 	/* host join errors which can happen before creating the group */
1283 	if (!cfg.p && !br_group_is_l2(&cfg.group)) {
1284 		/* don't allow any flags for host-joined IP groups */
1285 		if (cfg.entry->state) {
1286 			NL_SET_ERR_MSG_MOD(extack, "Flags are not allowed for host groups");
1287 			goto out;
1288 		}
1289 		if (!br_multicast_is_star_g(&cfg.group)) {
1290 			NL_SET_ERR_MSG_MOD(extack, "Groups with sources cannot be manually host joined");
1291 			goto out;
1292 		}
1293 	}
1294 
1295 	if (br_group_is_l2(&cfg.group) && cfg.entry->state != MDB_PERMANENT) {
1296 		NL_SET_ERR_MSG_MOD(extack, "Only permanent L2 entries allowed");
1297 		goto out;
1298 	}
1299 
1300 	if (cfg.p) {
1301 		if (cfg.p->state == BR_STATE_DISABLED && cfg.entry->state != MDB_PERMANENT) {
1302 			NL_SET_ERR_MSG_MOD(extack, "Port is in disabled state and entry is not permanent");
1303 			goto out;
1304 		}
1305 		vg = nbp_vlan_group(cfg.p);
1306 	} else {
1307 		vg = br_vlan_group(cfg.br);
1308 	}
1309 
1310 	/* If vlan filtering is enabled and VLAN is not specified
1311 	 * install mdb entry on all vlans configured on the port.
1312 	 */
1313 	if (br_vlan_enabled(cfg.br->dev) && vg && cfg.entry->vid == 0) {
1314 		list_for_each_entry(v, &vg->vlan_list, vlist) {
1315 			cfg.entry->vid = v->vid;
1316 			cfg.group.vid = v->vid;
1317 			err = __br_mdb_add(&cfg, extack);
1318 			if (err)
1319 				break;
1320 		}
1321 	} else {
1322 		err = __br_mdb_add(&cfg, extack);
1323 	}
1324 
1325 out:
1326 	br_mdb_config_fini(&cfg);
1327 	return err;
1328 }
1329 
1330 static int __br_mdb_del(const struct br_mdb_config *cfg)
1331 {
1332 	struct br_mdb_entry *entry = cfg->entry;
1333 	struct net_bridge *br = cfg->br;
1334 	struct net_bridge_mdb_entry *mp;
1335 	struct net_bridge_port_group *p;
1336 	struct net_bridge_port_group __rcu **pp;
1337 	struct br_ip ip = cfg->group;
1338 	int err = -EINVAL;
1339 
1340 	spin_lock_bh(&br->multicast_lock);
1341 	mp = br_mdb_ip_get(br, &ip);
1342 	if (!mp)
1343 		goto unlock;
1344 
1345 	/* host leave */
1346 	if (entry->ifindex == mp->br->dev->ifindex && mp->host_joined) {
1347 		br_multicast_host_leave(mp, false);
1348 		err = 0;
1349 		br_mdb_notify(br->dev, mp, NULL, RTM_DELMDB);
1350 		if (!mp->ports && netif_running(br->dev))
1351 			mod_timer(&mp->timer, jiffies);
1352 		goto unlock;
1353 	}
1354 
1355 	for (pp = &mp->ports;
1356 	     (p = mlock_dereference(*pp, br)) != NULL;
1357 	     pp = &p->next) {
1358 		if (!p->key.port || p->key.port->dev->ifindex != entry->ifindex)
1359 			continue;
1360 
1361 		br_multicast_del_pg(mp, p, pp);
1362 		err = 0;
1363 		break;
1364 	}
1365 
1366 unlock:
1367 	spin_unlock_bh(&br->multicast_lock);
1368 	return err;
1369 }
1370 
1371 int br_mdb_del(struct net_device *dev, struct nlattr *tb[],
1372 	       struct netlink_ext_ack *extack)
1373 {
1374 	struct net_bridge_vlan_group *vg;
1375 	struct net_bridge_vlan *v;
1376 	struct br_mdb_config cfg;
1377 	int err;
1378 
1379 	err = br_mdb_config_init(&cfg, dev, tb, 0, extack);
1380 	if (err)
1381 		return err;
1382 
1383 	if (cfg.p)
1384 		vg = nbp_vlan_group(cfg.p);
1385 	else
1386 		vg = br_vlan_group(cfg.br);
1387 
1388 	/* If vlan filtering is enabled and VLAN is not specified
1389 	 * delete mdb entry on all vlans configured on the port.
1390 	 */
1391 	if (br_vlan_enabled(cfg.br->dev) && vg && cfg.entry->vid == 0) {
1392 		list_for_each_entry(v, &vg->vlan_list, vlist) {
1393 			cfg.entry->vid = v->vid;
1394 			cfg.group.vid = v->vid;
1395 			err = __br_mdb_del(&cfg);
1396 		}
1397 	} else {
1398 		err = __br_mdb_del(&cfg);
1399 	}
1400 
1401 	br_mdb_config_fini(&cfg);
1402 	return err;
1403 }
1404