OSDN Git Service

net: bridge: add support for per-port vlan stats
[tomoyo/tomoyo-test1.git] / net / bridge / br_vlan.c
1 #include <linux/kernel.h>
2 #include <linux/netdevice.h>
3 #include <linux/rtnetlink.h>
4 #include <linux/slab.h>
5 #include <net/switchdev.h>
6
7 #include "br_private.h"
8 #include "br_private_tunnel.h"
9
10 static inline int br_vlan_cmp(struct rhashtable_compare_arg *arg,
11                               const void *ptr)
12 {
13         const struct net_bridge_vlan *vle = ptr;
14         u16 vid = *(u16 *)arg->key;
15
16         return vle->vid != vid;
17 }
18
19 static const struct rhashtable_params br_vlan_rht_params = {
20         .head_offset = offsetof(struct net_bridge_vlan, vnode),
21         .key_offset = offsetof(struct net_bridge_vlan, vid),
22         .key_len = sizeof(u16),
23         .nelem_hint = 3,
24         .locks_mul = 1,
25         .max_size = VLAN_N_VID,
26         .obj_cmpfn = br_vlan_cmp,
27         .automatic_shrinking = true,
28 };
29
30 static struct net_bridge_vlan *br_vlan_lookup(struct rhashtable *tbl, u16 vid)
31 {
32         return rhashtable_lookup_fast(tbl, &vid, br_vlan_rht_params);
33 }
34
35 static bool __vlan_add_pvid(struct net_bridge_vlan_group *vg, u16 vid)
36 {
37         if (vg->pvid == vid)
38                 return false;
39
40         smp_wmb();
41         vg->pvid = vid;
42
43         return true;
44 }
45
46 static bool __vlan_delete_pvid(struct net_bridge_vlan_group *vg, u16 vid)
47 {
48         if (vg->pvid != vid)
49                 return false;
50
51         smp_wmb();
52         vg->pvid = 0;
53
54         return true;
55 }
56
57 /* return true if anything changed, false otherwise */
58 static bool __vlan_add_flags(struct net_bridge_vlan *v, u16 flags)
59 {
60         struct net_bridge_vlan_group *vg;
61         u16 old_flags = v->flags;
62         bool ret;
63
64         if (br_vlan_is_master(v))
65                 vg = br_vlan_group(v->br);
66         else
67                 vg = nbp_vlan_group(v->port);
68
69         if (flags & BRIDGE_VLAN_INFO_PVID)
70                 ret = __vlan_add_pvid(vg, v->vid);
71         else
72                 ret = __vlan_delete_pvid(vg, v->vid);
73
74         if (flags & BRIDGE_VLAN_INFO_UNTAGGED)
75                 v->flags |= BRIDGE_VLAN_INFO_UNTAGGED;
76         else
77                 v->flags &= ~BRIDGE_VLAN_INFO_UNTAGGED;
78
79         return ret || !!(old_flags ^ v->flags);
80 }
81
82 static int __vlan_vid_add(struct net_device *dev, struct net_bridge *br,
83                           u16 vid, u16 flags)
84 {
85         int err;
86
87         /* Try switchdev op first. In case it is not supported, fallback to
88          * 8021q add.
89          */
90         err = br_switchdev_port_vlan_add(dev, vid, flags);
91         if (err == -EOPNOTSUPP)
92                 return vlan_vid_add(dev, br->vlan_proto, vid);
93         return err;
94 }
95
96 static void __vlan_add_list(struct net_bridge_vlan *v)
97 {
98         struct net_bridge_vlan_group *vg;
99         struct list_head *headp, *hpos;
100         struct net_bridge_vlan *vent;
101
102         if (br_vlan_is_master(v))
103                 vg = br_vlan_group(v->br);
104         else
105                 vg = nbp_vlan_group(v->port);
106
107         headp = &vg->vlan_list;
108         list_for_each_prev(hpos, headp) {
109                 vent = list_entry(hpos, struct net_bridge_vlan, vlist);
110                 if (v->vid < vent->vid)
111                         continue;
112                 else
113                         break;
114         }
115         list_add_rcu(&v->vlist, hpos);
116 }
117
118 static void __vlan_del_list(struct net_bridge_vlan *v)
119 {
120         list_del_rcu(&v->vlist);
121 }
122
123 static int __vlan_vid_del(struct net_device *dev, struct net_bridge *br,
124                           u16 vid)
125 {
126         int err;
127
128         /* Try switchdev op first. In case it is not supported, fallback to
129          * 8021q del.
130          */
131         err = br_switchdev_port_vlan_del(dev, vid);
132         if (err == -EOPNOTSUPP) {
133                 vlan_vid_del(dev, br->vlan_proto, vid);
134                 return 0;
135         }
136         return err;
137 }
138
139 /* Returns a master vlan, if it didn't exist it gets created. In all cases a
140  * a reference is taken to the master vlan before returning.
141  */
142 static struct net_bridge_vlan *br_vlan_get_master(struct net_bridge *br, u16 vid)
143 {
144         struct net_bridge_vlan_group *vg;
145         struct net_bridge_vlan *masterv;
146
147         vg = br_vlan_group(br);
148         masterv = br_vlan_find(vg, vid);
149         if (!masterv) {
150                 bool changed;
151
152                 /* missing global ctx, create it now */
153                 if (br_vlan_add(br, vid, 0, &changed))
154                         return NULL;
155                 masterv = br_vlan_find(vg, vid);
156                 if (WARN_ON(!masterv))
157                         return NULL;
158                 refcount_set(&masterv->refcnt, 1);
159                 return masterv;
160         }
161         refcount_inc(&masterv->refcnt);
162
163         return masterv;
164 }
165
166 static void br_master_vlan_rcu_free(struct rcu_head *rcu)
167 {
168         struct net_bridge_vlan *v;
169
170         v = container_of(rcu, struct net_bridge_vlan, rcu);
171         WARN_ON(!br_vlan_is_master(v));
172         free_percpu(v->stats);
173         v->stats = NULL;
174         kfree(v);
175 }
176
177 static void br_vlan_put_master(struct net_bridge_vlan *masterv)
178 {
179         struct net_bridge_vlan_group *vg;
180
181         if (!br_vlan_is_master(masterv))
182                 return;
183
184         vg = br_vlan_group(masterv->br);
185         if (refcount_dec_and_test(&masterv->refcnt)) {
186                 rhashtable_remove_fast(&vg->vlan_hash,
187                                        &masterv->vnode, br_vlan_rht_params);
188                 __vlan_del_list(masterv);
189                 call_rcu(&masterv->rcu, br_master_vlan_rcu_free);
190         }
191 }
192
193 static void nbp_vlan_rcu_free(struct rcu_head *rcu)
194 {
195         struct net_bridge_vlan *v;
196
197         v = container_of(rcu, struct net_bridge_vlan, rcu);
198         WARN_ON(br_vlan_is_master(v));
199         /* if we had per-port stats configured then free them here */
200         if (v->brvlan->stats != v->stats)
201                 free_percpu(v->stats);
202         v->stats = NULL;
203         kfree(v);
204 }
205
206 /* This is the shared VLAN add function which works for both ports and bridge
207  * devices. There are four possible calls to this function in terms of the
208  * vlan entry type:
209  * 1. vlan is being added on a port (no master flags, global entry exists)
210  * 2. vlan is being added on a bridge (both master and brentry flags)
211  * 3. vlan is being added on a port, but a global entry didn't exist which
212  *    is being created right now (master flag set, brentry flag unset), the
213  *    global entry is used for global per-vlan features, but not for filtering
214  * 4. same as 3 but with both master and brentry flags set so the entry
215  *    will be used for filtering in both the port and the bridge
216  */
217 static int __vlan_add(struct net_bridge_vlan *v, u16 flags)
218 {
219         struct net_bridge_vlan *masterv = NULL;
220         struct net_bridge_port *p = NULL;
221         struct net_bridge_vlan_group *vg;
222         struct net_device *dev;
223         struct net_bridge *br;
224         int err;
225
226         if (br_vlan_is_master(v)) {
227                 br = v->br;
228                 dev = br->dev;
229                 vg = br_vlan_group(br);
230         } else {
231                 p = v->port;
232                 br = p->br;
233                 dev = p->dev;
234                 vg = nbp_vlan_group(p);
235         }
236
237         if (p) {
238                 /* Add VLAN to the device filter if it is supported.
239                  * This ensures tagged traffic enters the bridge when
240                  * promiscuous mode is disabled by br_manage_promisc().
241                  */
242                 err = __vlan_vid_add(dev, br, v->vid, flags);
243                 if (err)
244                         goto out;
245
246                 /* need to work on the master vlan too */
247                 if (flags & BRIDGE_VLAN_INFO_MASTER) {
248                         bool changed;
249
250                         err = br_vlan_add(br, v->vid,
251                                           flags | BRIDGE_VLAN_INFO_BRENTRY,
252                                           &changed);
253                         if (err)
254                                 goto out_filt;
255                 }
256
257                 masterv = br_vlan_get_master(br, v->vid);
258                 if (!masterv)
259                         goto out_filt;
260                 v->brvlan = masterv;
261                 if (br_opt_get(br, BROPT_VLAN_STATS_PER_PORT)) {
262                         v->stats = netdev_alloc_pcpu_stats(struct br_vlan_stats);
263                         if (!v->stats) {
264                                 err = -ENOMEM;
265                                 goto out_filt;
266                         }
267                 } else {
268                         v->stats = masterv->stats;
269                 }
270         } else {
271                 err = br_switchdev_port_vlan_add(dev, v->vid, flags);
272                 if (err && err != -EOPNOTSUPP)
273                         goto out;
274         }
275
276         /* Add the dev mac and count the vlan only if it's usable */
277         if (br_vlan_should_use(v)) {
278                 err = br_fdb_insert(br, p, dev->dev_addr, v->vid);
279                 if (err) {
280                         br_err(br, "failed insert local address into bridge forwarding table\n");
281                         goto out_filt;
282                 }
283                 vg->num_vlans++;
284         }
285
286         err = rhashtable_lookup_insert_fast(&vg->vlan_hash, &v->vnode,
287                                             br_vlan_rht_params);
288         if (err)
289                 goto out_fdb_insert;
290
291         __vlan_add_list(v);
292         __vlan_add_flags(v, flags);
293 out:
294         return err;
295
296 out_fdb_insert:
297         if (br_vlan_should_use(v)) {
298                 br_fdb_find_delete_local(br, p, dev->dev_addr, v->vid);
299                 vg->num_vlans--;
300         }
301
302 out_filt:
303         if (p) {
304                 __vlan_vid_del(dev, br, v->vid);
305                 if (masterv) {
306                         br_vlan_put_master(masterv);
307                         v->brvlan = NULL;
308                 }
309         } else {
310                 br_switchdev_port_vlan_del(dev, v->vid);
311         }
312
313         goto out;
314 }
315
316 static int __vlan_del(struct net_bridge_vlan *v)
317 {
318         struct net_bridge_vlan *masterv = v;
319         struct net_bridge_vlan_group *vg;
320         struct net_bridge_port *p = NULL;
321         int err = 0;
322
323         if (br_vlan_is_master(v)) {
324                 vg = br_vlan_group(v->br);
325         } else {
326                 p = v->port;
327                 vg = nbp_vlan_group(v->port);
328                 masterv = v->brvlan;
329         }
330
331         __vlan_delete_pvid(vg, v->vid);
332         if (p) {
333                 err = __vlan_vid_del(p->dev, p->br, v->vid);
334                 if (err)
335                         goto out;
336         } else {
337                 err = br_switchdev_port_vlan_del(v->br->dev, v->vid);
338                 if (err && err != -EOPNOTSUPP)
339                         goto out;
340                 err = 0;
341         }
342
343         if (br_vlan_should_use(v)) {
344                 v->flags &= ~BRIDGE_VLAN_INFO_BRENTRY;
345                 vg->num_vlans--;
346         }
347
348         if (masterv != v) {
349                 vlan_tunnel_info_del(vg, v);
350                 rhashtable_remove_fast(&vg->vlan_hash, &v->vnode,
351                                        br_vlan_rht_params);
352                 __vlan_del_list(v);
353                 call_rcu(&v->rcu, nbp_vlan_rcu_free);
354         }
355
356         br_vlan_put_master(masterv);
357 out:
358         return err;
359 }
360
361 static void __vlan_group_free(struct net_bridge_vlan_group *vg)
362 {
363         WARN_ON(!list_empty(&vg->vlan_list));
364         rhashtable_destroy(&vg->vlan_hash);
365         vlan_tunnel_deinit(vg);
366         kfree(vg);
367 }
368
369 static void __vlan_flush(struct net_bridge_vlan_group *vg)
370 {
371         struct net_bridge_vlan *vlan, *tmp;
372
373         __vlan_delete_pvid(vg, vg->pvid);
374         list_for_each_entry_safe(vlan, tmp, &vg->vlan_list, vlist)
375                 __vlan_del(vlan);
376 }
377
378 struct sk_buff *br_handle_vlan(struct net_bridge *br,
379                                const struct net_bridge_port *p,
380                                struct net_bridge_vlan_group *vg,
381                                struct sk_buff *skb)
382 {
383         struct br_vlan_stats *stats;
384         struct net_bridge_vlan *v;
385         u16 vid;
386
387         /* If this packet was not filtered at input, let it pass */
388         if (!BR_INPUT_SKB_CB(skb)->vlan_filtered)
389                 goto out;
390
391         /* At this point, we know that the frame was filtered and contains
392          * a valid vlan id.  If the vlan id has untagged flag set,
393          * send untagged; otherwise, send tagged.
394          */
395         br_vlan_get_tag(skb, &vid);
396         v = br_vlan_find(vg, vid);
397         /* Vlan entry must be configured at this point.  The
398          * only exception is the bridge is set in promisc mode and the
399          * packet is destined for the bridge device.  In this case
400          * pass the packet as is.
401          */
402         if (!v || !br_vlan_should_use(v)) {
403                 if ((br->dev->flags & IFF_PROMISC) && skb->dev == br->dev) {
404                         goto out;
405                 } else {
406                         kfree_skb(skb);
407                         return NULL;
408                 }
409         }
410         if (br_opt_get(br, BROPT_VLAN_STATS_ENABLED)) {
411                 stats = this_cpu_ptr(v->stats);
412                 u64_stats_update_begin(&stats->syncp);
413                 stats->tx_bytes += skb->len;
414                 stats->tx_packets++;
415                 u64_stats_update_end(&stats->syncp);
416         }
417
418         if (v->flags & BRIDGE_VLAN_INFO_UNTAGGED)
419                 skb->vlan_tci = 0;
420
421         if (p && (p->flags & BR_VLAN_TUNNEL) &&
422             br_handle_egress_vlan_tunnel(skb, v)) {
423                 kfree_skb(skb);
424                 return NULL;
425         }
426 out:
427         return skb;
428 }
429
430 /* Called under RCU */
431 static bool __allowed_ingress(const struct net_bridge *br,
432                               struct net_bridge_vlan_group *vg,
433                               struct sk_buff *skb, u16 *vid)
434 {
435         struct br_vlan_stats *stats;
436         struct net_bridge_vlan *v;
437         bool tagged;
438
439         BR_INPUT_SKB_CB(skb)->vlan_filtered = true;
440         /* If vlan tx offload is disabled on bridge device and frame was
441          * sent from vlan device on the bridge device, it does not have
442          * HW accelerated vlan tag.
443          */
444         if (unlikely(!skb_vlan_tag_present(skb) &&
445                      skb->protocol == br->vlan_proto)) {
446                 skb = skb_vlan_untag(skb);
447                 if (unlikely(!skb))
448                         return false;
449         }
450
451         if (!br_vlan_get_tag(skb, vid)) {
452                 /* Tagged frame */
453                 if (skb->vlan_proto != br->vlan_proto) {
454                         /* Protocol-mismatch, empty out vlan_tci for new tag */
455                         skb_push(skb, ETH_HLEN);
456                         skb = vlan_insert_tag_set_proto(skb, skb->vlan_proto,
457                                                         skb_vlan_tag_get(skb));
458                         if (unlikely(!skb))
459                                 return false;
460
461                         skb_pull(skb, ETH_HLEN);
462                         skb_reset_mac_len(skb);
463                         *vid = 0;
464                         tagged = false;
465                 } else {
466                         tagged = true;
467                 }
468         } else {
469                 /* Untagged frame */
470                 tagged = false;
471         }
472
473         if (!*vid) {
474                 u16 pvid = br_get_pvid(vg);
475
476                 /* Frame had a tag with VID 0 or did not have a tag.
477                  * See if pvid is set on this port.  That tells us which
478                  * vlan untagged or priority-tagged traffic belongs to.
479                  */
480                 if (!pvid)
481                         goto drop;
482
483                 /* PVID is set on this port.  Any untagged or priority-tagged
484                  * ingress frame is considered to belong to this vlan.
485                  */
486                 *vid = pvid;
487                 if (likely(!tagged))
488                         /* Untagged Frame. */
489                         __vlan_hwaccel_put_tag(skb, br->vlan_proto, pvid);
490                 else
491                         /* Priority-tagged Frame.
492                          * At this point, We know that skb->vlan_tci had
493                          * VLAN_TAG_PRESENT bit and its VID field was 0x000.
494                          * We update only VID field and preserve PCP field.
495                          */
496                         skb->vlan_tci |= pvid;
497
498                 /* if stats are disabled we can avoid the lookup */
499                 if (!br_opt_get(br, BROPT_VLAN_STATS_ENABLED))
500                         return true;
501         }
502         v = br_vlan_find(vg, *vid);
503         if (!v || !br_vlan_should_use(v))
504                 goto drop;
505
506         if (br_opt_get(br, BROPT_VLAN_STATS_ENABLED)) {
507                 stats = this_cpu_ptr(v->stats);
508                 u64_stats_update_begin(&stats->syncp);
509                 stats->rx_bytes += skb->len;
510                 stats->rx_packets++;
511                 u64_stats_update_end(&stats->syncp);
512         }
513
514         return true;
515
516 drop:
517         kfree_skb(skb);
518         return false;
519 }
520
521 bool br_allowed_ingress(const struct net_bridge *br,
522                         struct net_bridge_vlan_group *vg, struct sk_buff *skb,
523                         u16 *vid)
524 {
525         /* If VLAN filtering is disabled on the bridge, all packets are
526          * permitted.
527          */
528         if (!br_opt_get(br, BROPT_VLAN_ENABLED)) {
529                 BR_INPUT_SKB_CB(skb)->vlan_filtered = false;
530                 return true;
531         }
532
533         return __allowed_ingress(br, vg, skb, vid);
534 }
535
536 /* Called under RCU. */
537 bool br_allowed_egress(struct net_bridge_vlan_group *vg,
538                        const struct sk_buff *skb)
539 {
540         const struct net_bridge_vlan *v;
541         u16 vid;
542
543         /* If this packet was not filtered at input, let it pass */
544         if (!BR_INPUT_SKB_CB(skb)->vlan_filtered)
545                 return true;
546
547         br_vlan_get_tag(skb, &vid);
548         v = br_vlan_find(vg, vid);
549         if (v && br_vlan_should_use(v))
550                 return true;
551
552         return false;
553 }
554
555 /* Called under RCU */
556 bool br_should_learn(struct net_bridge_port *p, struct sk_buff *skb, u16 *vid)
557 {
558         struct net_bridge_vlan_group *vg;
559         struct net_bridge *br = p->br;
560
561         /* If filtering was disabled at input, let it pass. */
562         if (!br_opt_get(br, BROPT_VLAN_ENABLED))
563                 return true;
564
565         vg = nbp_vlan_group_rcu(p);
566         if (!vg || !vg->num_vlans)
567                 return false;
568
569         if (!br_vlan_get_tag(skb, vid) && skb->vlan_proto != br->vlan_proto)
570                 *vid = 0;
571
572         if (!*vid) {
573                 *vid = br_get_pvid(vg);
574                 if (!*vid)
575                         return false;
576
577                 return true;
578         }
579
580         if (br_vlan_find(vg, *vid))
581                 return true;
582
583         return false;
584 }
585
586 static int br_vlan_add_existing(struct net_bridge *br,
587                                 struct net_bridge_vlan_group *vg,
588                                 struct net_bridge_vlan *vlan,
589                                 u16 flags, bool *changed)
590 {
591         int err;
592
593         err = br_switchdev_port_vlan_add(br->dev, vlan->vid, flags);
594         if (err && err != -EOPNOTSUPP)
595                 return err;
596
597         if (!br_vlan_is_brentry(vlan)) {
598                 /* Trying to change flags of non-existent bridge vlan */
599                 if (!(flags & BRIDGE_VLAN_INFO_BRENTRY)) {
600                         err = -EINVAL;
601                         goto err_flags;
602                 }
603                 /* It was only kept for port vlans, now make it real */
604                 err = br_fdb_insert(br, NULL, br->dev->dev_addr,
605                                     vlan->vid);
606                 if (err) {
607                         br_err(br, "failed to insert local address into bridge forwarding table\n");
608                         goto err_fdb_insert;
609                 }
610
611                 refcount_inc(&vlan->refcnt);
612                 vlan->flags |= BRIDGE_VLAN_INFO_BRENTRY;
613                 vg->num_vlans++;
614                 *changed = true;
615         }
616
617         if (__vlan_add_flags(vlan, flags))
618                 *changed = true;
619
620         return 0;
621
622 err_fdb_insert:
623 err_flags:
624         br_switchdev_port_vlan_del(br->dev, vlan->vid);
625         return err;
626 }
627
628 /* Must be protected by RTNL.
629  * Must be called with vid in range from 1 to 4094 inclusive.
630  * changed must be true only if the vlan was created or updated
631  */
632 int br_vlan_add(struct net_bridge *br, u16 vid, u16 flags, bool *changed)
633 {
634         struct net_bridge_vlan_group *vg;
635         struct net_bridge_vlan *vlan;
636         int ret;
637
638         ASSERT_RTNL();
639
640         *changed = false;
641         vg = br_vlan_group(br);
642         vlan = br_vlan_find(vg, vid);
643         if (vlan)
644                 return br_vlan_add_existing(br, vg, vlan, flags, changed);
645
646         vlan = kzalloc(sizeof(*vlan), GFP_KERNEL);
647         if (!vlan)
648                 return -ENOMEM;
649
650         vlan->stats = netdev_alloc_pcpu_stats(struct br_vlan_stats);
651         if (!vlan->stats) {
652                 kfree(vlan);
653                 return -ENOMEM;
654         }
655         vlan->vid = vid;
656         vlan->flags = flags | BRIDGE_VLAN_INFO_MASTER;
657         vlan->flags &= ~BRIDGE_VLAN_INFO_PVID;
658         vlan->br = br;
659         if (flags & BRIDGE_VLAN_INFO_BRENTRY)
660                 refcount_set(&vlan->refcnt, 1);
661         ret = __vlan_add(vlan, flags);
662         if (ret) {
663                 free_percpu(vlan->stats);
664                 kfree(vlan);
665         } else {
666                 *changed = true;
667         }
668
669         return ret;
670 }
671
672 /* Must be protected by RTNL.
673  * Must be called with vid in range from 1 to 4094 inclusive.
674  */
675 int br_vlan_delete(struct net_bridge *br, u16 vid)
676 {
677         struct net_bridge_vlan_group *vg;
678         struct net_bridge_vlan *v;
679
680         ASSERT_RTNL();
681
682         vg = br_vlan_group(br);
683         v = br_vlan_find(vg, vid);
684         if (!v || !br_vlan_is_brentry(v))
685                 return -ENOENT;
686
687         br_fdb_find_delete_local(br, NULL, br->dev->dev_addr, vid);
688         br_fdb_delete_by_port(br, NULL, vid, 0);
689
690         vlan_tunnel_info_del(vg, v);
691
692         return __vlan_del(v);
693 }
694
695 void br_vlan_flush(struct net_bridge *br)
696 {
697         struct net_bridge_vlan_group *vg;
698
699         ASSERT_RTNL();
700
701         vg = br_vlan_group(br);
702         __vlan_flush(vg);
703         RCU_INIT_POINTER(br->vlgrp, NULL);
704         synchronize_rcu();
705         __vlan_group_free(vg);
706 }
707
708 struct net_bridge_vlan *br_vlan_find(struct net_bridge_vlan_group *vg, u16 vid)
709 {
710         if (!vg)
711                 return NULL;
712
713         return br_vlan_lookup(&vg->vlan_hash, vid);
714 }
715
716 /* Must be protected by RTNL. */
717 static void recalculate_group_addr(struct net_bridge *br)
718 {
719         if (br_opt_get(br, BROPT_GROUP_ADDR_SET))
720                 return;
721
722         spin_lock_bh(&br->lock);
723         if (!br_opt_get(br, BROPT_VLAN_ENABLED) ||
724             br->vlan_proto == htons(ETH_P_8021Q)) {
725                 /* Bridge Group Address */
726                 br->group_addr[5] = 0x00;
727         } else { /* vlan_enabled && ETH_P_8021AD */
728                 /* Provider Bridge Group Address */
729                 br->group_addr[5] = 0x08;
730         }
731         spin_unlock_bh(&br->lock);
732 }
733
734 /* Must be protected by RTNL. */
735 void br_recalculate_fwd_mask(struct net_bridge *br)
736 {
737         if (!br_opt_get(br, BROPT_VLAN_ENABLED) ||
738             br->vlan_proto == htons(ETH_P_8021Q))
739                 br->group_fwd_mask_required = BR_GROUPFWD_DEFAULT;
740         else /* vlan_enabled && ETH_P_8021AD */
741                 br->group_fwd_mask_required = BR_GROUPFWD_8021AD &
742                                               ~(1u << br->group_addr[5]);
743 }
744
745 int __br_vlan_filter_toggle(struct net_bridge *br, unsigned long val)
746 {
747         struct switchdev_attr attr = {
748                 .orig_dev = br->dev,
749                 .id = SWITCHDEV_ATTR_ID_BRIDGE_VLAN_FILTERING,
750                 .flags = SWITCHDEV_F_SKIP_EOPNOTSUPP,
751                 .u.vlan_filtering = val,
752         };
753         int err;
754
755         if (br_opt_get(br, BROPT_VLAN_ENABLED) == !!val)
756                 return 0;
757
758         err = switchdev_port_attr_set(br->dev, &attr);
759         if (err && err != -EOPNOTSUPP)
760                 return err;
761
762         br_opt_toggle(br, BROPT_VLAN_ENABLED, !!val);
763         br_manage_promisc(br);
764         recalculate_group_addr(br);
765         br_recalculate_fwd_mask(br);
766
767         return 0;
768 }
769
770 int br_vlan_filter_toggle(struct net_bridge *br, unsigned long val)
771 {
772         return __br_vlan_filter_toggle(br, val);
773 }
774
775 bool br_vlan_enabled(const struct net_device *dev)
776 {
777         struct net_bridge *br = netdev_priv(dev);
778
779         return br_opt_get(br, BROPT_VLAN_ENABLED);
780 }
781 EXPORT_SYMBOL_GPL(br_vlan_enabled);
782
783 int __br_vlan_set_proto(struct net_bridge *br, __be16 proto)
784 {
785         int err = 0;
786         struct net_bridge_port *p;
787         struct net_bridge_vlan *vlan;
788         struct net_bridge_vlan_group *vg;
789         __be16 oldproto;
790
791         if (br->vlan_proto == proto)
792                 return 0;
793
794         /* Add VLANs for the new proto to the device filter. */
795         list_for_each_entry(p, &br->port_list, list) {
796                 vg = nbp_vlan_group(p);
797                 list_for_each_entry(vlan, &vg->vlan_list, vlist) {
798                         err = vlan_vid_add(p->dev, proto, vlan->vid);
799                         if (err)
800                                 goto err_filt;
801                 }
802         }
803
804         oldproto = br->vlan_proto;
805         br->vlan_proto = proto;
806
807         recalculate_group_addr(br);
808         br_recalculate_fwd_mask(br);
809
810         /* Delete VLANs for the old proto from the device filter. */
811         list_for_each_entry(p, &br->port_list, list) {
812                 vg = nbp_vlan_group(p);
813                 list_for_each_entry(vlan, &vg->vlan_list, vlist)
814                         vlan_vid_del(p->dev, oldproto, vlan->vid);
815         }
816
817         return 0;
818
819 err_filt:
820         list_for_each_entry_continue_reverse(vlan, &vg->vlan_list, vlist)
821                 vlan_vid_del(p->dev, proto, vlan->vid);
822
823         list_for_each_entry_continue_reverse(p, &br->port_list, list) {
824                 vg = nbp_vlan_group(p);
825                 list_for_each_entry(vlan, &vg->vlan_list, vlist)
826                         vlan_vid_del(p->dev, proto, vlan->vid);
827         }
828
829         return err;
830 }
831
832 int br_vlan_set_proto(struct net_bridge *br, unsigned long val)
833 {
834         if (val != ETH_P_8021Q && val != ETH_P_8021AD)
835                 return -EPROTONOSUPPORT;
836
837         return __br_vlan_set_proto(br, htons(val));
838 }
839
840 int br_vlan_set_stats(struct net_bridge *br, unsigned long val)
841 {
842         switch (val) {
843         case 0:
844         case 1:
845                 br_opt_toggle(br, BROPT_VLAN_STATS_ENABLED, !!val);
846                 break;
847         default:
848                 return -EINVAL;
849         }
850
851         return 0;
852 }
853
854 int br_vlan_set_stats_per_port(struct net_bridge *br, unsigned long val)
855 {
856         struct net_bridge_port *p;
857
858         /* allow to change the option if there are no port vlans configured */
859         list_for_each_entry(p, &br->port_list, list) {
860                 struct net_bridge_vlan_group *vg = nbp_vlan_group(p);
861
862                 if (vg->num_vlans)
863                         return -EBUSY;
864         }
865
866         switch (val) {
867         case 0:
868         case 1:
869                 br_opt_toggle(br, BROPT_VLAN_STATS_PER_PORT, !!val);
870                 break;
871         default:
872                 return -EINVAL;
873         }
874
875         return 0;
876 }
877
878 static bool vlan_default_pvid(struct net_bridge_vlan_group *vg, u16 vid)
879 {
880         struct net_bridge_vlan *v;
881
882         if (vid != vg->pvid)
883                 return false;
884
885         v = br_vlan_lookup(&vg->vlan_hash, vid);
886         if (v && br_vlan_should_use(v) &&
887             (v->flags & BRIDGE_VLAN_INFO_UNTAGGED))
888                 return true;
889
890         return false;
891 }
892
893 static void br_vlan_disable_default_pvid(struct net_bridge *br)
894 {
895         struct net_bridge_port *p;
896         u16 pvid = br->default_pvid;
897
898         /* Disable default_pvid on all ports where it is still
899          * configured.
900          */
901         if (vlan_default_pvid(br_vlan_group(br), pvid))
902                 br_vlan_delete(br, pvid);
903
904         list_for_each_entry(p, &br->port_list, list) {
905                 if (vlan_default_pvid(nbp_vlan_group(p), pvid))
906                         nbp_vlan_delete(p, pvid);
907         }
908
909         br->default_pvid = 0;
910 }
911
912 int __br_vlan_set_default_pvid(struct net_bridge *br, u16 pvid)
913 {
914         const struct net_bridge_vlan *pvent;
915         struct net_bridge_vlan_group *vg;
916         struct net_bridge_port *p;
917         unsigned long *changed;
918         bool vlchange;
919         u16 old_pvid;
920         int err = 0;
921
922         if (!pvid) {
923                 br_vlan_disable_default_pvid(br);
924                 return 0;
925         }
926
927         changed = bitmap_zalloc(BR_MAX_PORTS, GFP_KERNEL);
928         if (!changed)
929                 return -ENOMEM;
930
931         old_pvid = br->default_pvid;
932
933         /* Update default_pvid config only if we do not conflict with
934          * user configuration.
935          */
936         vg = br_vlan_group(br);
937         pvent = br_vlan_find(vg, pvid);
938         if ((!old_pvid || vlan_default_pvid(vg, old_pvid)) &&
939             (!pvent || !br_vlan_should_use(pvent))) {
940                 err = br_vlan_add(br, pvid,
941                                   BRIDGE_VLAN_INFO_PVID |
942                                   BRIDGE_VLAN_INFO_UNTAGGED |
943                                   BRIDGE_VLAN_INFO_BRENTRY,
944                                   &vlchange);
945                 if (err)
946                         goto out;
947                 br_vlan_delete(br, old_pvid);
948                 set_bit(0, changed);
949         }
950
951         list_for_each_entry(p, &br->port_list, list) {
952                 /* Update default_pvid config only if we do not conflict with
953                  * user configuration.
954                  */
955                 vg = nbp_vlan_group(p);
956                 if ((old_pvid &&
957                      !vlan_default_pvid(vg, old_pvid)) ||
958                     br_vlan_find(vg, pvid))
959                         continue;
960
961                 err = nbp_vlan_add(p, pvid,
962                                    BRIDGE_VLAN_INFO_PVID |
963                                    BRIDGE_VLAN_INFO_UNTAGGED,
964                                    &vlchange);
965                 if (err)
966                         goto err_port;
967                 nbp_vlan_delete(p, old_pvid);
968                 set_bit(p->port_no, changed);
969         }
970
971         br->default_pvid = pvid;
972
973 out:
974         bitmap_free(changed);
975         return err;
976
977 err_port:
978         list_for_each_entry_continue_reverse(p, &br->port_list, list) {
979                 if (!test_bit(p->port_no, changed))
980                         continue;
981
982                 if (old_pvid)
983                         nbp_vlan_add(p, old_pvid,
984                                      BRIDGE_VLAN_INFO_PVID |
985                                      BRIDGE_VLAN_INFO_UNTAGGED,
986                                      &vlchange);
987                 nbp_vlan_delete(p, pvid);
988         }
989
990         if (test_bit(0, changed)) {
991                 if (old_pvid)
992                         br_vlan_add(br, old_pvid,
993                                     BRIDGE_VLAN_INFO_PVID |
994                                     BRIDGE_VLAN_INFO_UNTAGGED |
995                                     BRIDGE_VLAN_INFO_BRENTRY,
996                                     &vlchange);
997                 br_vlan_delete(br, pvid);
998         }
999         goto out;
1000 }
1001
1002 int br_vlan_set_default_pvid(struct net_bridge *br, unsigned long val)
1003 {
1004         u16 pvid = val;
1005         int err = 0;
1006
1007         if (val >= VLAN_VID_MASK)
1008                 return -EINVAL;
1009
1010         if (pvid == br->default_pvid)
1011                 goto out;
1012
1013         /* Only allow default pvid change when filtering is disabled */
1014         if (br_opt_get(br, BROPT_VLAN_ENABLED)) {
1015                 pr_info_once("Please disable vlan filtering to change default_pvid\n");
1016                 err = -EPERM;
1017                 goto out;
1018         }
1019         err = __br_vlan_set_default_pvid(br, pvid);
1020 out:
1021         return err;
1022 }
1023
1024 int br_vlan_init(struct net_bridge *br)
1025 {
1026         struct net_bridge_vlan_group *vg;
1027         int ret = -ENOMEM;
1028         bool changed;
1029
1030         vg = kzalloc(sizeof(*vg), GFP_KERNEL);
1031         if (!vg)
1032                 goto out;
1033         ret = rhashtable_init(&vg->vlan_hash, &br_vlan_rht_params);
1034         if (ret)
1035                 goto err_rhtbl;
1036         ret = vlan_tunnel_init(vg);
1037         if (ret)
1038                 goto err_tunnel_init;
1039         INIT_LIST_HEAD(&vg->vlan_list);
1040         br->vlan_proto = htons(ETH_P_8021Q);
1041         br->default_pvid = 1;
1042         rcu_assign_pointer(br->vlgrp, vg);
1043         ret = br_vlan_add(br, 1,
1044                           BRIDGE_VLAN_INFO_PVID | BRIDGE_VLAN_INFO_UNTAGGED |
1045                           BRIDGE_VLAN_INFO_BRENTRY, &changed);
1046         if (ret)
1047                 goto err_vlan_add;
1048
1049 out:
1050         return ret;
1051
1052 err_vlan_add:
1053         vlan_tunnel_deinit(vg);
1054 err_tunnel_init:
1055         rhashtable_destroy(&vg->vlan_hash);
1056 err_rhtbl:
1057         kfree(vg);
1058
1059         goto out;
1060 }
1061
1062 int nbp_vlan_init(struct net_bridge_port *p)
1063 {
1064         struct switchdev_attr attr = {
1065                 .orig_dev = p->br->dev,
1066                 .id = SWITCHDEV_ATTR_ID_BRIDGE_VLAN_FILTERING,
1067                 .flags = SWITCHDEV_F_SKIP_EOPNOTSUPP,
1068                 .u.vlan_filtering = br_opt_get(p->br, BROPT_VLAN_ENABLED),
1069         };
1070         struct net_bridge_vlan_group *vg;
1071         int ret = -ENOMEM;
1072
1073         vg = kzalloc(sizeof(struct net_bridge_vlan_group), GFP_KERNEL);
1074         if (!vg)
1075                 goto out;
1076
1077         ret = switchdev_port_attr_set(p->dev, &attr);
1078         if (ret && ret != -EOPNOTSUPP)
1079                 goto err_vlan_enabled;
1080
1081         ret = rhashtable_init(&vg->vlan_hash, &br_vlan_rht_params);
1082         if (ret)
1083                 goto err_rhtbl;
1084         ret = vlan_tunnel_init(vg);
1085         if (ret)
1086                 goto err_tunnel_init;
1087         INIT_LIST_HEAD(&vg->vlan_list);
1088         rcu_assign_pointer(p->vlgrp, vg);
1089         if (p->br->default_pvid) {
1090                 bool changed;
1091
1092                 ret = nbp_vlan_add(p, p->br->default_pvid,
1093                                    BRIDGE_VLAN_INFO_PVID |
1094                                    BRIDGE_VLAN_INFO_UNTAGGED,
1095                                    &changed);
1096                 if (ret)
1097                         goto err_vlan_add;
1098         }
1099 out:
1100         return ret;
1101
1102 err_vlan_add:
1103         RCU_INIT_POINTER(p->vlgrp, NULL);
1104         synchronize_rcu();
1105         vlan_tunnel_deinit(vg);
1106 err_tunnel_init:
1107         rhashtable_destroy(&vg->vlan_hash);
1108 err_rhtbl:
1109 err_vlan_enabled:
1110         kfree(vg);
1111
1112         goto out;
1113 }
1114
1115 /* Must be protected by RTNL.
1116  * Must be called with vid in range from 1 to 4094 inclusive.
1117  * changed must be true only if the vlan was created or updated
1118  */
1119 int nbp_vlan_add(struct net_bridge_port *port, u16 vid, u16 flags,
1120                  bool *changed)
1121 {
1122         struct net_bridge_vlan *vlan;
1123         int ret;
1124
1125         ASSERT_RTNL();
1126
1127         *changed = false;
1128         vlan = br_vlan_find(nbp_vlan_group(port), vid);
1129         if (vlan) {
1130                 /* Pass the flags to the hardware bridge */
1131                 ret = br_switchdev_port_vlan_add(port->dev, vid, flags);
1132                 if (ret && ret != -EOPNOTSUPP)
1133                         return ret;
1134                 *changed = __vlan_add_flags(vlan, flags);
1135
1136                 return 0;
1137         }
1138
1139         vlan = kzalloc(sizeof(*vlan), GFP_KERNEL);
1140         if (!vlan)
1141                 return -ENOMEM;
1142
1143         vlan->vid = vid;
1144         vlan->port = port;
1145         ret = __vlan_add(vlan, flags);
1146         if (ret)
1147                 kfree(vlan);
1148         else
1149                 *changed = true;
1150
1151         return ret;
1152 }
1153
1154 /* Must be protected by RTNL.
1155  * Must be called with vid in range from 1 to 4094 inclusive.
1156  */
1157 int nbp_vlan_delete(struct net_bridge_port *port, u16 vid)
1158 {
1159         struct net_bridge_vlan *v;
1160
1161         ASSERT_RTNL();
1162
1163         v = br_vlan_find(nbp_vlan_group(port), vid);
1164         if (!v)
1165                 return -ENOENT;
1166         br_fdb_find_delete_local(port->br, port, port->dev->dev_addr, vid);
1167         br_fdb_delete_by_port(port->br, port, vid, 0);
1168
1169         return __vlan_del(v);
1170 }
1171
1172 void nbp_vlan_flush(struct net_bridge_port *port)
1173 {
1174         struct net_bridge_vlan_group *vg;
1175
1176         ASSERT_RTNL();
1177
1178         vg = nbp_vlan_group(port);
1179         __vlan_flush(vg);
1180         RCU_INIT_POINTER(port->vlgrp, NULL);
1181         synchronize_rcu();
1182         __vlan_group_free(vg);
1183 }
1184
1185 void br_vlan_get_stats(const struct net_bridge_vlan *v,
1186                        struct br_vlan_stats *stats)
1187 {
1188         int i;
1189
1190         memset(stats, 0, sizeof(*stats));
1191         for_each_possible_cpu(i) {
1192                 u64 rxpackets, rxbytes, txpackets, txbytes;
1193                 struct br_vlan_stats *cpu_stats;
1194                 unsigned int start;
1195
1196                 cpu_stats = per_cpu_ptr(v->stats, i);
1197                 do {
1198                         start = u64_stats_fetch_begin_irq(&cpu_stats->syncp);
1199                         rxpackets = cpu_stats->rx_packets;
1200                         rxbytes = cpu_stats->rx_bytes;
1201                         txbytes = cpu_stats->tx_bytes;
1202                         txpackets = cpu_stats->tx_packets;
1203                 } while (u64_stats_fetch_retry_irq(&cpu_stats->syncp, start));
1204
1205                 stats->rx_packets += rxpackets;
1206                 stats->rx_bytes += rxbytes;
1207                 stats->tx_bytes += txbytes;
1208                 stats->tx_packets += txpackets;
1209         }
1210 }
1211
1212 int br_vlan_get_pvid(const struct net_device *dev, u16 *p_pvid)
1213 {
1214         struct net_bridge_vlan_group *vg;
1215
1216         ASSERT_RTNL();
1217         if (netif_is_bridge_master(dev))
1218                 vg = br_vlan_group(netdev_priv(dev));
1219         else
1220                 return -EINVAL;
1221
1222         *p_pvid = br_get_pvid(vg);
1223         return 0;
1224 }
1225 EXPORT_SYMBOL_GPL(br_vlan_get_pvid);
1226
1227 int br_vlan_get_info(const struct net_device *dev, u16 vid,
1228                      struct bridge_vlan_info *p_vinfo)
1229 {
1230         struct net_bridge_vlan_group *vg;
1231         struct net_bridge_vlan *v;
1232         struct net_bridge_port *p;
1233
1234         ASSERT_RTNL();
1235         p = br_port_get_check_rtnl(dev);
1236         if (p)
1237                 vg = nbp_vlan_group(p);
1238         else if (netif_is_bridge_master(dev))
1239                 vg = br_vlan_group(netdev_priv(dev));
1240         else
1241                 return -EINVAL;
1242
1243         v = br_vlan_find(vg, vid);
1244         if (!v)
1245                 return -ENOENT;
1246
1247         p_vinfo->vid = vid;
1248         p_vinfo->flags = v->flags;
1249         return 0;
1250 }
1251 EXPORT_SYMBOL_GPL(br_vlan_get_info);