Merge branch '8021ad'
[firefly-linux-kernel-4.4.55.git] / net / 8021q / vlan_core.c
1 #include <linux/skbuff.h>
2 #include <linux/netdevice.h>
3 #include <linux/if_vlan.h>
4 #include <linux/netpoll.h>
5 #include <linux/export.h>
6 #include "vlan.h"
7
8 bool vlan_do_receive(struct sk_buff **skbp)
9 {
10         struct sk_buff *skb = *skbp;
11         __be16 vlan_proto = skb->vlan_proto;
12         u16 vlan_id = skb->vlan_tci & VLAN_VID_MASK;
13         struct net_device *vlan_dev;
14         struct vlan_pcpu_stats *rx_stats;
15
16         vlan_dev = vlan_find_dev(skb->dev, vlan_proto, vlan_id);
17         if (!vlan_dev)
18                 return false;
19
20         skb = *skbp = skb_share_check(skb, GFP_ATOMIC);
21         if (unlikely(!skb))
22                 return false;
23
24         skb->dev = vlan_dev;
25         if (skb->pkt_type == PACKET_OTHERHOST) {
26                 /* Our lower layer thinks this is not local, let's make sure.
27                  * This allows the VLAN to have a different MAC than the
28                  * underlying device, and still route correctly. */
29                 if (ether_addr_equal(eth_hdr(skb)->h_dest, vlan_dev->dev_addr))
30                         skb->pkt_type = PACKET_HOST;
31         }
32
33         if (!(vlan_dev_priv(vlan_dev)->flags & VLAN_FLAG_REORDER_HDR)) {
34                 unsigned int offset = skb->data - skb_mac_header(skb);
35
36                 /*
37                  * vlan_insert_tag expect skb->data pointing to mac header.
38                  * So change skb->data before calling it and change back to
39                  * original position later
40                  */
41                 skb_push(skb, offset);
42                 skb = *skbp = vlan_insert_tag(skb, skb->vlan_proto,
43                                               skb->vlan_tci);
44                 if (!skb)
45                         return false;
46                 skb_pull(skb, offset + VLAN_HLEN);
47                 skb_reset_mac_len(skb);
48         }
49
50         skb->priority = vlan_get_ingress_priority(vlan_dev, skb->vlan_tci);
51         skb->vlan_tci = 0;
52
53         rx_stats = this_cpu_ptr(vlan_dev_priv(vlan_dev)->vlan_pcpu_stats);
54
55         u64_stats_update_begin(&rx_stats->syncp);
56         rx_stats->rx_packets++;
57         rx_stats->rx_bytes += skb->len;
58         if (skb->pkt_type == PACKET_MULTICAST)
59                 rx_stats->rx_multicast++;
60         u64_stats_update_end(&rx_stats->syncp);
61
62         return true;
63 }
64
65 /* Must be invoked with rcu_read_lock. */
66 struct net_device *__vlan_find_dev_deep(struct net_device *dev,
67                                         __be16 vlan_proto, u16 vlan_id)
68 {
69         struct vlan_info *vlan_info = rcu_dereference(dev->vlan_info);
70
71         if (vlan_info) {
72                 return vlan_group_get_device(&vlan_info->grp,
73                                              vlan_proto, vlan_id);
74         } else {
75                 /*
76                  * Lower devices of master uppers (bonding, team) do not have
77                  * grp assigned to themselves. Grp is assigned to upper device
78                  * instead.
79                  */
80                 struct net_device *upper_dev;
81
82                 upper_dev = netdev_master_upper_dev_get_rcu(dev);
83                 if (upper_dev)
84                         return __vlan_find_dev_deep(upper_dev,
85                                                     vlan_proto, vlan_id);
86         }
87
88         return NULL;
89 }
90 EXPORT_SYMBOL(__vlan_find_dev_deep);
91
92 struct net_device *vlan_dev_real_dev(const struct net_device *dev)
93 {
94         return vlan_dev_priv(dev)->real_dev;
95 }
96 EXPORT_SYMBOL(vlan_dev_real_dev);
97
98 u16 vlan_dev_vlan_id(const struct net_device *dev)
99 {
100         return vlan_dev_priv(dev)->vlan_id;
101 }
102 EXPORT_SYMBOL(vlan_dev_vlan_id);
103
104 static struct sk_buff *vlan_reorder_header(struct sk_buff *skb)
105 {
106         if (skb_cow(skb, skb_headroom(skb)) < 0)
107                 return NULL;
108         memmove(skb->data - ETH_HLEN, skb->data - VLAN_ETH_HLEN, 2 * ETH_ALEN);
109         skb->mac_header += VLAN_HLEN;
110         return skb;
111 }
112
113 struct sk_buff *vlan_untag(struct sk_buff *skb)
114 {
115         struct vlan_hdr *vhdr;
116         u16 vlan_tci;
117
118         if (unlikely(vlan_tx_tag_present(skb))) {
119                 /* vlan_tci is already set-up so leave this for another time */
120                 return skb;
121         }
122
123         skb = skb_share_check(skb, GFP_ATOMIC);
124         if (unlikely(!skb))
125                 goto err_free;
126
127         if (unlikely(!pskb_may_pull(skb, VLAN_HLEN)))
128                 goto err_free;
129
130         vhdr = (struct vlan_hdr *) skb->data;
131         vlan_tci = ntohs(vhdr->h_vlan_TCI);
132         __vlan_hwaccel_put_tag(skb, skb->protocol, vlan_tci);
133
134         skb_pull_rcsum(skb, VLAN_HLEN);
135         vlan_set_encap_proto(skb, vhdr);
136
137         skb = vlan_reorder_header(skb);
138         if (unlikely(!skb))
139                 goto err_free;
140
141         skb_reset_network_header(skb);
142         skb_reset_transport_header(skb);
143         skb_reset_mac_len(skb);
144
145         return skb;
146
147 err_free:
148         kfree_skb(skb);
149         return NULL;
150 }
151 EXPORT_SYMBOL(vlan_untag);
152
153
154 /*
155  * vlan info and vid list
156  */
157
158 static void vlan_group_free(struct vlan_group *grp)
159 {
160         int i;
161
162         for (i = 0; i < VLAN_GROUP_ARRAY_SPLIT_PARTS; i++)
163                 kfree(grp->vlan_devices_arrays[i]);
164 }
165
166 static void vlan_info_free(struct vlan_info *vlan_info)
167 {
168         vlan_group_free(&vlan_info->grp);
169         kfree(vlan_info);
170 }
171
172 static void vlan_info_rcu_free(struct rcu_head *rcu)
173 {
174         vlan_info_free(container_of(rcu, struct vlan_info, rcu));
175 }
176
177 static struct vlan_info *vlan_info_alloc(struct net_device *dev)
178 {
179         struct vlan_info *vlan_info;
180
181         vlan_info = kzalloc(sizeof(struct vlan_info), GFP_KERNEL);
182         if (!vlan_info)
183                 return NULL;
184
185         vlan_info->real_dev = dev;
186         INIT_LIST_HEAD(&vlan_info->vid_list);
187         return vlan_info;
188 }
189
190 struct vlan_vid_info {
191         struct list_head list;
192         __be16 proto;
193         u16 vid;
194         int refcount;
195 };
196
197 static bool vlan_hw_filter_capable(const struct net_device *dev,
198                                      const struct vlan_vid_info *vid_info)
199 {
200         if (vid_info->proto == htons(ETH_P_8021Q) &&
201             dev->features & NETIF_F_HW_VLAN_CTAG_FILTER)
202                 return true;
203         if (vid_info->proto == htons(ETH_P_8021AD) &&
204             dev->features & NETIF_F_HW_VLAN_STAG_FILTER)
205                 return true;
206         return false;
207 }
208
209 static struct vlan_vid_info *vlan_vid_info_get(struct vlan_info *vlan_info,
210                                                __be16 proto, u16 vid)
211 {
212         struct vlan_vid_info *vid_info;
213
214         list_for_each_entry(vid_info, &vlan_info->vid_list, list) {
215                 if (vid_info->proto == proto && vid_info->vid == vid)
216                         return vid_info;
217         }
218         return NULL;
219 }
220
221 static struct vlan_vid_info *vlan_vid_info_alloc(__be16 proto, u16 vid)
222 {
223         struct vlan_vid_info *vid_info;
224
225         vid_info = kzalloc(sizeof(struct vlan_vid_info), GFP_KERNEL);
226         if (!vid_info)
227                 return NULL;
228         vid_info->proto = proto;
229         vid_info->vid = vid;
230
231         return vid_info;
232 }
233
234 static int __vlan_vid_add(struct vlan_info *vlan_info, __be16 proto, u16 vid,
235                           struct vlan_vid_info **pvid_info)
236 {
237         struct net_device *dev = vlan_info->real_dev;
238         const struct net_device_ops *ops = dev->netdev_ops;
239         struct vlan_vid_info *vid_info;
240         int err;
241
242         vid_info = vlan_vid_info_alloc(proto, vid);
243         if (!vid_info)
244                 return -ENOMEM;
245
246         if (vlan_hw_filter_capable(dev, vid_info)) {
247                 err =  ops->ndo_vlan_rx_add_vid(dev, proto, vid);
248                 if (err) {
249                         kfree(vid_info);
250                         return err;
251                 }
252         }
253         list_add(&vid_info->list, &vlan_info->vid_list);
254         vlan_info->nr_vids++;
255         *pvid_info = vid_info;
256         return 0;
257 }
258
259 int vlan_vid_add(struct net_device *dev, __be16 proto, u16 vid)
260 {
261         struct vlan_info *vlan_info;
262         struct vlan_vid_info *vid_info;
263         bool vlan_info_created = false;
264         int err;
265
266         ASSERT_RTNL();
267
268         vlan_info = rtnl_dereference(dev->vlan_info);
269         if (!vlan_info) {
270                 vlan_info = vlan_info_alloc(dev);
271                 if (!vlan_info)
272                         return -ENOMEM;
273                 vlan_info_created = true;
274         }
275         vid_info = vlan_vid_info_get(vlan_info, proto, vid);
276         if (!vid_info) {
277                 err = __vlan_vid_add(vlan_info, proto, vid, &vid_info);
278                 if (err)
279                         goto out_free_vlan_info;
280         }
281         vid_info->refcount++;
282
283         if (vlan_info_created)
284                 rcu_assign_pointer(dev->vlan_info, vlan_info);
285
286         return 0;
287
288 out_free_vlan_info:
289         if (vlan_info_created)
290                 kfree(vlan_info);
291         return err;
292 }
293 EXPORT_SYMBOL(vlan_vid_add);
294
295 static void __vlan_vid_del(struct vlan_info *vlan_info,
296                            struct vlan_vid_info *vid_info)
297 {
298         struct net_device *dev = vlan_info->real_dev;
299         const struct net_device_ops *ops = dev->netdev_ops;
300         __be16 proto = vid_info->proto;
301         u16 vid = vid_info->vid;
302         int err;
303
304         if (vlan_hw_filter_capable(dev, vid_info)) {
305                 err = ops->ndo_vlan_rx_kill_vid(dev, proto, vid);
306                 if (err) {
307                         pr_warn("failed to kill vid %04x/%d for device %s\n",
308                                 proto, vid, dev->name);
309                 }
310         }
311         list_del(&vid_info->list);
312         kfree(vid_info);
313         vlan_info->nr_vids--;
314 }
315
316 void vlan_vid_del(struct net_device *dev, __be16 proto, u16 vid)
317 {
318         struct vlan_info *vlan_info;
319         struct vlan_vid_info *vid_info;
320
321         ASSERT_RTNL();
322
323         vlan_info = rtnl_dereference(dev->vlan_info);
324         if (!vlan_info)
325                 return;
326
327         vid_info = vlan_vid_info_get(vlan_info, proto, vid);
328         if (!vid_info)
329                 return;
330         vid_info->refcount--;
331         if (vid_info->refcount == 0) {
332                 __vlan_vid_del(vlan_info, vid_info);
333                 if (vlan_info->nr_vids == 0) {
334                         RCU_INIT_POINTER(dev->vlan_info, NULL);
335                         call_rcu(&vlan_info->rcu, vlan_info_rcu_free);
336                 }
337         }
338 }
339 EXPORT_SYMBOL(vlan_vid_del);
340
341 int vlan_vids_add_by_dev(struct net_device *dev,
342                          const struct net_device *by_dev)
343 {
344         struct vlan_vid_info *vid_info;
345         struct vlan_info *vlan_info;
346         int err;
347
348         ASSERT_RTNL();
349
350         vlan_info = rtnl_dereference(by_dev->vlan_info);
351         if (!vlan_info)
352                 return 0;
353
354         list_for_each_entry(vid_info, &vlan_info->vid_list, list) {
355                 err = vlan_vid_add(dev, vid_info->proto, vid_info->vid);
356                 if (err)
357                         goto unwind;
358         }
359         return 0;
360
361 unwind:
362         list_for_each_entry_continue_reverse(vid_info,
363                                              &vlan_info->vid_list,
364                                              list) {
365                 vlan_vid_del(dev, vid_info->proto, vid_info->vid);
366         }
367
368         return err;
369 }
370 EXPORT_SYMBOL(vlan_vids_add_by_dev);
371
372 void vlan_vids_del_by_dev(struct net_device *dev,
373                           const struct net_device *by_dev)
374 {
375         struct vlan_vid_info *vid_info;
376         struct vlan_info *vlan_info;
377
378         ASSERT_RTNL();
379
380         vlan_info = rtnl_dereference(by_dev->vlan_info);
381         if (!vlan_info)
382                 return;
383
384         list_for_each_entry(vid_info, &vlan_info->vid_list, list)
385                 vlan_vid_del(dev, vid_info->proto, vid_info->vid);
386 }
387 EXPORT_SYMBOL(vlan_vids_del_by_dev);
388
389 bool vlan_uses_dev(const struct net_device *dev)
390 {
391         struct vlan_info *vlan_info;
392
393         ASSERT_RTNL();
394
395         vlan_info = rtnl_dereference(dev->vlan_info);
396         if (!vlan_info)
397                 return false;
398         return vlan_info->grp.nr_vlan_devs ? true : false;
399 }
400 EXPORT_SYMBOL(vlan_uses_dev);