drm/vc4: crtc: Add null pointer error handling
[platform/kernel/linux-rpi.git] / drivers / net / geneve.c
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * GENEVE: Generic Network Virtualization Encapsulation
4  *
5  * Copyright (c) 2015 Red Hat, Inc.
6  */
7
8 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
9
10 #include <linux/kernel.h>
11 #include <linux/module.h>
12 #include <linux/etherdevice.h>
13 #include <linux/hash.h>
14 #include <net/ipv6_stubs.h>
15 #include <net/dst_metadata.h>
16 #include <net/gro_cells.h>
17 #include <net/rtnetlink.h>
18 #include <net/geneve.h>
19 #include <net/protocol.h>
20
21 #define GENEVE_NETDEV_VER       "0.6"
22
23 #define GENEVE_N_VID            (1u << 24)
24 #define GENEVE_VID_MASK         (GENEVE_N_VID - 1)
25
26 #define VNI_HASH_BITS           10
27 #define VNI_HASH_SIZE           (1<<VNI_HASH_BITS)
28
29 static bool log_ecn_error = true;
30 module_param(log_ecn_error, bool, 0644);
31 MODULE_PARM_DESC(log_ecn_error, "Log packets received with corrupted ECN");
32
33 #define GENEVE_VER 0
34 #define GENEVE_BASE_HLEN (sizeof(struct udphdr) + sizeof(struct genevehdr))
35 #define GENEVE_IPV4_HLEN (ETH_HLEN + sizeof(struct iphdr) + GENEVE_BASE_HLEN)
36 #define GENEVE_IPV6_HLEN (ETH_HLEN + sizeof(struct ipv6hdr) + GENEVE_BASE_HLEN)
37
38 /* per-network namespace private data for this module */
39 struct geneve_net {
40         struct list_head        geneve_list;
41         struct list_head        sock_list;
42 };
43
44 static unsigned int geneve_net_id;
45
46 struct geneve_dev_node {
47         struct hlist_node hlist;
48         struct geneve_dev *geneve;
49 };
50
51 /* Pseudo network device */
52 struct geneve_dev {
53         struct geneve_dev_node hlist4;  /* vni hash table for IPv4 socket */
54 #if IS_ENABLED(CONFIG_IPV6)
55         struct geneve_dev_node hlist6;  /* vni hash table for IPv6 socket */
56 #endif
57         struct net         *net;        /* netns for packet i/o */
58         struct net_device  *dev;        /* netdev for geneve tunnel */
59         struct ip_tunnel_info info;
60         struct geneve_sock __rcu *sock4;        /* IPv4 socket used for geneve tunnel */
61 #if IS_ENABLED(CONFIG_IPV6)
62         struct geneve_sock __rcu *sock6;        /* IPv6 socket used for geneve tunnel */
63 #endif
64         struct list_head   next;        /* geneve's per namespace list */
65         struct gro_cells   gro_cells;
66         bool               collect_md;
67         bool               use_udp6_rx_checksums;
68         bool               ttl_inherit;
69         enum ifla_geneve_df df;
70 };
71
72 struct geneve_sock {
73         bool                    collect_md;
74         struct list_head        list;
75         struct socket           *sock;
76         struct rcu_head         rcu;
77         int                     refcnt;
78         struct hlist_head       vni_list[VNI_HASH_SIZE];
79 };
80
81 static inline __u32 geneve_net_vni_hash(u8 vni[3])
82 {
83         __u32 vnid;
84
85         vnid = (vni[0] << 16) | (vni[1] << 8) | vni[2];
86         return hash_32(vnid, VNI_HASH_BITS);
87 }
88
89 static __be64 vni_to_tunnel_id(const __u8 *vni)
90 {
91 #ifdef __BIG_ENDIAN
92         return (vni[0] << 16) | (vni[1] << 8) | vni[2];
93 #else
94         return (__force __be64)(((__force u64)vni[0] << 40) |
95                                 ((__force u64)vni[1] << 48) |
96                                 ((__force u64)vni[2] << 56));
97 #endif
98 }
99
100 /* Convert 64 bit tunnel ID to 24 bit VNI. */
101 static void tunnel_id_to_vni(__be64 tun_id, __u8 *vni)
102 {
103 #ifdef __BIG_ENDIAN
104         vni[0] = (__force __u8)(tun_id >> 16);
105         vni[1] = (__force __u8)(tun_id >> 8);
106         vni[2] = (__force __u8)tun_id;
107 #else
108         vni[0] = (__force __u8)((__force u64)tun_id >> 40);
109         vni[1] = (__force __u8)((__force u64)tun_id >> 48);
110         vni[2] = (__force __u8)((__force u64)tun_id >> 56);
111 #endif
112 }
113
114 static bool eq_tun_id_and_vni(u8 *tun_id, u8 *vni)
115 {
116         return !memcmp(vni, &tun_id[5], 3);
117 }
118
119 static sa_family_t geneve_get_sk_family(struct geneve_sock *gs)
120 {
121         return gs->sock->sk->sk_family;
122 }
123
124 static struct geneve_dev *geneve_lookup(struct geneve_sock *gs,
125                                         __be32 addr, u8 vni[])
126 {
127         struct hlist_head *vni_list_head;
128         struct geneve_dev_node *node;
129         __u32 hash;
130
131         /* Find the device for this VNI */
132         hash = geneve_net_vni_hash(vni);
133         vni_list_head = &gs->vni_list[hash];
134         hlist_for_each_entry_rcu(node, vni_list_head, hlist) {
135                 if (eq_tun_id_and_vni((u8 *)&node->geneve->info.key.tun_id, vni) &&
136                     addr == node->geneve->info.key.u.ipv4.dst)
137                         return node->geneve;
138         }
139         return NULL;
140 }
141
142 #if IS_ENABLED(CONFIG_IPV6)
143 static struct geneve_dev *geneve6_lookup(struct geneve_sock *gs,
144                                          struct in6_addr addr6, u8 vni[])
145 {
146         struct hlist_head *vni_list_head;
147         struct geneve_dev_node *node;
148         __u32 hash;
149
150         /* Find the device for this VNI */
151         hash = geneve_net_vni_hash(vni);
152         vni_list_head = &gs->vni_list[hash];
153         hlist_for_each_entry_rcu(node, vni_list_head, hlist) {
154                 if (eq_tun_id_and_vni((u8 *)&node->geneve->info.key.tun_id, vni) &&
155                     ipv6_addr_equal(&addr6, &node->geneve->info.key.u.ipv6.dst))
156                         return node->geneve;
157         }
158         return NULL;
159 }
160 #endif
161
162 static inline struct genevehdr *geneve_hdr(const struct sk_buff *skb)
163 {
164         return (struct genevehdr *)(udp_hdr(skb) + 1);
165 }
166
167 static struct geneve_dev *geneve_lookup_skb(struct geneve_sock *gs,
168                                             struct sk_buff *skb)
169 {
170         static u8 zero_vni[3];
171         u8 *vni;
172
173         if (geneve_get_sk_family(gs) == AF_INET) {
174                 struct iphdr *iph;
175                 __be32 addr;
176
177                 iph = ip_hdr(skb); /* outer IP header... */
178
179                 if (gs->collect_md) {
180                         vni = zero_vni;
181                         addr = 0;
182                 } else {
183                         vni = geneve_hdr(skb)->vni;
184                         addr = iph->saddr;
185                 }
186
187                 return geneve_lookup(gs, addr, vni);
188 #if IS_ENABLED(CONFIG_IPV6)
189         } else if (geneve_get_sk_family(gs) == AF_INET6) {
190                 static struct in6_addr zero_addr6;
191                 struct ipv6hdr *ip6h;
192                 struct in6_addr addr6;
193
194                 ip6h = ipv6_hdr(skb); /* outer IPv6 header... */
195
196                 if (gs->collect_md) {
197                         vni = zero_vni;
198                         addr6 = zero_addr6;
199                 } else {
200                         vni = geneve_hdr(skb)->vni;
201                         addr6 = ip6h->saddr;
202                 }
203
204                 return geneve6_lookup(gs, addr6, vni);
205 #endif
206         }
207         return NULL;
208 }
209
210 /* geneve receive/decap routine */
211 static void geneve_rx(struct geneve_dev *geneve, struct geneve_sock *gs,
212                       struct sk_buff *skb)
213 {
214         struct genevehdr *gnvh = geneve_hdr(skb);
215         struct metadata_dst *tun_dst = NULL;
216         struct pcpu_sw_netstats *stats;
217         unsigned int len;
218         int err = 0;
219         void *oiph;
220
221         if (ip_tunnel_collect_metadata() || gs->collect_md) {
222                 __be16 flags;
223
224                 flags = TUNNEL_KEY | TUNNEL_GENEVE_OPT |
225                         (gnvh->oam ? TUNNEL_OAM : 0) |
226                         (gnvh->critical ? TUNNEL_CRIT_OPT : 0);
227
228                 tun_dst = udp_tun_rx_dst(skb, geneve_get_sk_family(gs), flags,
229                                          vni_to_tunnel_id(gnvh->vni),
230                                          gnvh->opt_len * 4);
231                 if (!tun_dst) {
232                         geneve->dev->stats.rx_dropped++;
233                         goto drop;
234                 }
235                 /* Update tunnel dst according to Geneve options. */
236                 ip_tunnel_info_opts_set(&tun_dst->u.tun_info,
237                                         gnvh->options, gnvh->opt_len * 4,
238                                         TUNNEL_GENEVE_OPT);
239         } else {
240                 /* Drop packets w/ critical options,
241                  * since we don't support any...
242                  */
243                 if (gnvh->critical) {
244                         geneve->dev->stats.rx_frame_errors++;
245                         geneve->dev->stats.rx_errors++;
246                         goto drop;
247                 }
248         }
249
250         skb_reset_mac_header(skb);
251         skb->protocol = eth_type_trans(skb, geneve->dev);
252         skb_postpull_rcsum(skb, eth_hdr(skb), ETH_HLEN);
253
254         if (tun_dst)
255                 skb_dst_set(skb, &tun_dst->dst);
256
257         /* Ignore packet loops (and multicast echo) */
258         if (ether_addr_equal(eth_hdr(skb)->h_source, geneve->dev->dev_addr)) {
259                 geneve->dev->stats.rx_errors++;
260                 goto drop;
261         }
262
263         oiph = skb_network_header(skb);
264         skb_reset_network_header(skb);
265
266         if (geneve_get_sk_family(gs) == AF_INET)
267                 err = IP_ECN_decapsulate(oiph, skb);
268 #if IS_ENABLED(CONFIG_IPV6)
269         else
270                 err = IP6_ECN_decapsulate(oiph, skb);
271 #endif
272
273         if (unlikely(err)) {
274                 if (log_ecn_error) {
275                         if (geneve_get_sk_family(gs) == AF_INET)
276                                 net_info_ratelimited("non-ECT from %pI4 "
277                                                      "with TOS=%#x\n",
278                                                      &((struct iphdr *)oiph)->saddr,
279                                                      ((struct iphdr *)oiph)->tos);
280 #if IS_ENABLED(CONFIG_IPV6)
281                         else
282                                 net_info_ratelimited("non-ECT from %pI6\n",
283                                                      &((struct ipv6hdr *)oiph)->saddr);
284 #endif
285                 }
286                 if (err > 1) {
287                         ++geneve->dev->stats.rx_frame_errors;
288                         ++geneve->dev->stats.rx_errors;
289                         goto drop;
290                 }
291         }
292
293         len = skb->len;
294         err = gro_cells_receive(&geneve->gro_cells, skb);
295         if (likely(err == NET_RX_SUCCESS)) {
296                 stats = this_cpu_ptr(geneve->dev->tstats);
297                 u64_stats_update_begin(&stats->syncp);
298                 stats->rx_packets++;
299                 stats->rx_bytes += len;
300                 u64_stats_update_end(&stats->syncp);
301         }
302         return;
303 drop:
304         /* Consume bad packet */
305         kfree_skb(skb);
306 }
307
308 /* Setup stats when device is created */
309 static int geneve_init(struct net_device *dev)
310 {
311         struct geneve_dev *geneve = netdev_priv(dev);
312         int err;
313
314         dev->tstats = netdev_alloc_pcpu_stats(struct pcpu_sw_netstats);
315         if (!dev->tstats)
316                 return -ENOMEM;
317
318         err = gro_cells_init(&geneve->gro_cells, dev);
319         if (err) {
320                 free_percpu(dev->tstats);
321                 return err;
322         }
323
324         err = dst_cache_init(&geneve->info.dst_cache, GFP_KERNEL);
325         if (err) {
326                 free_percpu(dev->tstats);
327                 gro_cells_destroy(&geneve->gro_cells);
328                 return err;
329         }
330         return 0;
331 }
332
333 static void geneve_uninit(struct net_device *dev)
334 {
335         struct geneve_dev *geneve = netdev_priv(dev);
336
337         dst_cache_destroy(&geneve->info.dst_cache);
338         gro_cells_destroy(&geneve->gro_cells);
339         free_percpu(dev->tstats);
340 }
341
342 /* Callback from net/ipv4/udp.c to receive packets */
343 static int geneve_udp_encap_recv(struct sock *sk, struct sk_buff *skb)
344 {
345         struct genevehdr *geneveh;
346         struct geneve_dev *geneve;
347         struct geneve_sock *gs;
348         int opts_len;
349
350         /* Need UDP and Geneve header to be present */
351         if (unlikely(!pskb_may_pull(skb, GENEVE_BASE_HLEN)))
352                 goto drop;
353
354         /* Return packets with reserved bits set */
355         geneveh = geneve_hdr(skb);
356         if (unlikely(geneveh->ver != GENEVE_VER))
357                 goto drop;
358
359         if (unlikely(geneveh->proto_type != htons(ETH_P_TEB)))
360                 goto drop;
361
362         gs = rcu_dereference_sk_user_data(sk);
363         if (!gs)
364                 goto drop;
365
366         geneve = geneve_lookup_skb(gs, skb);
367         if (!geneve)
368                 goto drop;
369
370         opts_len = geneveh->opt_len * 4;
371         if (iptunnel_pull_header(skb, GENEVE_BASE_HLEN + opts_len,
372                                  htons(ETH_P_TEB),
373                                  !net_eq(geneve->net, dev_net(geneve->dev)))) {
374                 geneve->dev->stats.rx_dropped++;
375                 goto drop;
376         }
377
378         geneve_rx(geneve, gs, skb);
379         return 0;
380
381 drop:
382         /* Consume bad packet */
383         kfree_skb(skb);
384         return 0;
385 }
386
387 /* Callback from net/ipv{4,6}/udp.c to check that we have a tunnel for errors */
388 static int geneve_udp_encap_err_lookup(struct sock *sk, struct sk_buff *skb)
389 {
390         struct genevehdr *geneveh;
391         struct geneve_sock *gs;
392         u8 zero_vni[3] = { 0 };
393         u8 *vni = zero_vni;
394
395         if (!pskb_may_pull(skb, skb_transport_offset(skb) + GENEVE_BASE_HLEN))
396                 return -EINVAL;
397
398         geneveh = geneve_hdr(skb);
399         if (geneveh->ver != GENEVE_VER)
400                 return -EINVAL;
401
402         if (geneveh->proto_type != htons(ETH_P_TEB))
403                 return -EINVAL;
404
405         gs = rcu_dereference_sk_user_data(sk);
406         if (!gs)
407                 return -ENOENT;
408
409         if (geneve_get_sk_family(gs) == AF_INET) {
410                 struct iphdr *iph = ip_hdr(skb);
411                 __be32 addr4 = 0;
412
413                 if (!gs->collect_md) {
414                         vni = geneve_hdr(skb)->vni;
415                         addr4 = iph->daddr;
416                 }
417
418                 return geneve_lookup(gs, addr4, vni) ? 0 : -ENOENT;
419         }
420
421 #if IS_ENABLED(CONFIG_IPV6)
422         if (geneve_get_sk_family(gs) == AF_INET6) {
423                 struct ipv6hdr *ip6h = ipv6_hdr(skb);
424                 struct in6_addr addr6;
425
426                 memset(&addr6, 0, sizeof(struct in6_addr));
427
428                 if (!gs->collect_md) {
429                         vni = geneve_hdr(skb)->vni;
430                         addr6 = ip6h->daddr;
431                 }
432
433                 return geneve6_lookup(gs, addr6, vni) ? 0 : -ENOENT;
434         }
435 #endif
436
437         return -EPFNOSUPPORT;
438 }
439
440 static struct socket *geneve_create_sock(struct net *net, bool ipv6,
441                                          __be16 port, bool ipv6_rx_csum)
442 {
443         struct socket *sock;
444         struct udp_port_cfg udp_conf;
445         int err;
446
447         memset(&udp_conf, 0, sizeof(udp_conf));
448
449         if (ipv6) {
450                 udp_conf.family = AF_INET6;
451                 udp_conf.ipv6_v6only = 1;
452                 udp_conf.use_udp6_rx_checksums = ipv6_rx_csum;
453         } else {
454                 udp_conf.family = AF_INET;
455                 udp_conf.local_ip.s_addr = htonl(INADDR_ANY);
456         }
457
458         udp_conf.local_udp_port = port;
459
460         /* Open UDP socket */
461         err = udp_sock_create(net, &udp_conf, &sock);
462         if (err < 0)
463                 return ERR_PTR(err);
464
465         return sock;
466 }
467
468 static int geneve_hlen(struct genevehdr *gh)
469 {
470         return sizeof(*gh) + gh->opt_len * 4;
471 }
472
473 static struct sk_buff *geneve_gro_receive(struct sock *sk,
474                                           struct list_head *head,
475                                           struct sk_buff *skb)
476 {
477         struct sk_buff *pp = NULL;
478         struct sk_buff *p;
479         struct genevehdr *gh, *gh2;
480         unsigned int hlen, gh_len, off_gnv;
481         const struct packet_offload *ptype;
482         __be16 type;
483         int flush = 1;
484
485         off_gnv = skb_gro_offset(skb);
486         hlen = off_gnv + sizeof(*gh);
487         gh = skb_gro_header_fast(skb, off_gnv);
488         if (skb_gro_header_hard(skb, hlen)) {
489                 gh = skb_gro_header_slow(skb, hlen, off_gnv);
490                 if (unlikely(!gh))
491                         goto out;
492         }
493
494         if (gh->ver != GENEVE_VER || gh->oam)
495                 goto out;
496         gh_len = geneve_hlen(gh);
497
498         hlen = off_gnv + gh_len;
499         if (skb_gro_header_hard(skb, hlen)) {
500                 gh = skb_gro_header_slow(skb, hlen, off_gnv);
501                 if (unlikely(!gh))
502                         goto out;
503         }
504
505         list_for_each_entry(p, head, list) {
506                 if (!NAPI_GRO_CB(p)->same_flow)
507                         continue;
508
509                 gh2 = (struct genevehdr *)(p->data + off_gnv);
510                 if (gh->opt_len != gh2->opt_len ||
511                     memcmp(gh, gh2, gh_len)) {
512                         NAPI_GRO_CB(p)->same_flow = 0;
513                         continue;
514                 }
515         }
516
517         type = gh->proto_type;
518
519         rcu_read_lock();
520         ptype = gro_find_receive_by_type(type);
521         if (!ptype)
522                 goto out_unlock;
523
524         skb_gro_pull(skb, gh_len);
525         skb_gro_postpull_rcsum(skb, gh, gh_len);
526         pp = call_gro_receive(ptype->callbacks.gro_receive, head, skb);
527         flush = 0;
528
529 out_unlock:
530         rcu_read_unlock();
531 out:
532         skb_gro_flush_final(skb, pp, flush);
533
534         return pp;
535 }
536
537 static int geneve_gro_complete(struct sock *sk, struct sk_buff *skb,
538                                int nhoff)
539 {
540         struct genevehdr *gh;
541         struct packet_offload *ptype;
542         __be16 type;
543         int gh_len;
544         int err = -ENOSYS;
545
546         gh = (struct genevehdr *)(skb->data + nhoff);
547         gh_len = geneve_hlen(gh);
548         type = gh->proto_type;
549
550         rcu_read_lock();
551         ptype = gro_find_complete_by_type(type);
552         if (ptype)
553                 err = ptype->callbacks.gro_complete(skb, nhoff + gh_len);
554
555         rcu_read_unlock();
556
557         skb_set_inner_mac_header(skb, nhoff + gh_len);
558
559         return err;
560 }
561
562 /* Create new listen socket if needed */
563 static struct geneve_sock *geneve_socket_create(struct net *net, __be16 port,
564                                                 bool ipv6, bool ipv6_rx_csum)
565 {
566         struct geneve_net *gn = net_generic(net, geneve_net_id);
567         struct geneve_sock *gs;
568         struct socket *sock;
569         struct udp_tunnel_sock_cfg tunnel_cfg;
570         int h;
571
572         gs = kzalloc(sizeof(*gs), GFP_KERNEL);
573         if (!gs)
574                 return ERR_PTR(-ENOMEM);
575
576         sock = geneve_create_sock(net, ipv6, port, ipv6_rx_csum);
577         if (IS_ERR(sock)) {
578                 kfree(gs);
579                 return ERR_CAST(sock);
580         }
581
582         gs->sock = sock;
583         gs->refcnt = 1;
584         for (h = 0; h < VNI_HASH_SIZE; ++h)
585                 INIT_HLIST_HEAD(&gs->vni_list[h]);
586
587         /* Initialize the geneve udp offloads structure */
588         udp_tunnel_notify_add_rx_port(gs->sock, UDP_TUNNEL_TYPE_GENEVE);
589
590         /* Mark socket as an encapsulation socket */
591         memset(&tunnel_cfg, 0, sizeof(tunnel_cfg));
592         tunnel_cfg.sk_user_data = gs;
593         tunnel_cfg.encap_type = 1;
594         tunnel_cfg.gro_receive = geneve_gro_receive;
595         tunnel_cfg.gro_complete = geneve_gro_complete;
596         tunnel_cfg.encap_rcv = geneve_udp_encap_recv;
597         tunnel_cfg.encap_err_lookup = geneve_udp_encap_err_lookup;
598         tunnel_cfg.encap_destroy = NULL;
599         setup_udp_tunnel_sock(net, sock, &tunnel_cfg);
600         list_add(&gs->list, &gn->sock_list);
601         return gs;
602 }
603
604 static void __geneve_sock_release(struct geneve_sock *gs)
605 {
606         if (!gs || --gs->refcnt)
607                 return;
608
609         list_del(&gs->list);
610         udp_tunnel_notify_del_rx_port(gs->sock, UDP_TUNNEL_TYPE_GENEVE);
611         udp_tunnel_sock_release(gs->sock);
612         kfree_rcu(gs, rcu);
613 }
614
615 static void geneve_sock_release(struct geneve_dev *geneve)
616 {
617         struct geneve_sock *gs4 = rtnl_dereference(geneve->sock4);
618 #if IS_ENABLED(CONFIG_IPV6)
619         struct geneve_sock *gs6 = rtnl_dereference(geneve->sock6);
620
621         rcu_assign_pointer(geneve->sock6, NULL);
622 #endif
623
624         rcu_assign_pointer(geneve->sock4, NULL);
625         synchronize_net();
626
627         __geneve_sock_release(gs4);
628 #if IS_ENABLED(CONFIG_IPV6)
629         __geneve_sock_release(gs6);
630 #endif
631 }
632
633 static struct geneve_sock *geneve_find_sock(struct geneve_net *gn,
634                                             sa_family_t family,
635                                             __be16 dst_port)
636 {
637         struct geneve_sock *gs;
638
639         list_for_each_entry(gs, &gn->sock_list, list) {
640                 if (inet_sk(gs->sock->sk)->inet_sport == dst_port &&
641                     geneve_get_sk_family(gs) == family) {
642                         return gs;
643                 }
644         }
645         return NULL;
646 }
647
648 static int geneve_sock_add(struct geneve_dev *geneve, bool ipv6)
649 {
650         struct net *net = geneve->net;
651         struct geneve_net *gn = net_generic(net, geneve_net_id);
652         struct geneve_dev_node *node;
653         struct geneve_sock *gs;
654         __u8 vni[3];
655         __u32 hash;
656
657         gs = geneve_find_sock(gn, ipv6 ? AF_INET6 : AF_INET, geneve->info.key.tp_dst);
658         if (gs) {
659                 gs->refcnt++;
660                 goto out;
661         }
662
663         gs = geneve_socket_create(net, geneve->info.key.tp_dst, ipv6,
664                                   geneve->use_udp6_rx_checksums);
665         if (IS_ERR(gs))
666                 return PTR_ERR(gs);
667
668 out:
669         gs->collect_md = geneve->collect_md;
670 #if IS_ENABLED(CONFIG_IPV6)
671         if (ipv6) {
672                 rcu_assign_pointer(geneve->sock6, gs);
673                 node = &geneve->hlist6;
674         } else
675 #endif
676         {
677                 rcu_assign_pointer(geneve->sock4, gs);
678                 node = &geneve->hlist4;
679         }
680         node->geneve = geneve;
681
682         tunnel_id_to_vni(geneve->info.key.tun_id, vni);
683         hash = geneve_net_vni_hash(vni);
684         hlist_add_head_rcu(&node->hlist, &gs->vni_list[hash]);
685         return 0;
686 }
687
688 static int geneve_open(struct net_device *dev)
689 {
690         struct geneve_dev *geneve = netdev_priv(dev);
691         bool metadata = geneve->collect_md;
692         bool ipv4, ipv6;
693         int ret = 0;
694
695         ipv6 = geneve->info.mode & IP_TUNNEL_INFO_IPV6 || metadata;
696         ipv4 = !ipv6 || metadata;
697 #if IS_ENABLED(CONFIG_IPV6)
698         if (ipv6) {
699                 ret = geneve_sock_add(geneve, true);
700                 if (ret < 0 && ret != -EAFNOSUPPORT)
701                         ipv4 = false;
702         }
703 #endif
704         if (ipv4)
705                 ret = geneve_sock_add(geneve, false);
706         if (ret < 0)
707                 geneve_sock_release(geneve);
708
709         return ret;
710 }
711
712 static int geneve_stop(struct net_device *dev)
713 {
714         struct geneve_dev *geneve = netdev_priv(dev);
715
716         hlist_del_init_rcu(&geneve->hlist4.hlist);
717 #if IS_ENABLED(CONFIG_IPV6)
718         hlist_del_init_rcu(&geneve->hlist6.hlist);
719 #endif
720         geneve_sock_release(geneve);
721         return 0;
722 }
723
724 static void geneve_build_header(struct genevehdr *geneveh,
725                                 const struct ip_tunnel_info *info)
726 {
727         geneveh->ver = GENEVE_VER;
728         geneveh->opt_len = info->options_len / 4;
729         geneveh->oam = !!(info->key.tun_flags & TUNNEL_OAM);
730         geneveh->critical = !!(info->key.tun_flags & TUNNEL_CRIT_OPT);
731         geneveh->rsvd1 = 0;
732         tunnel_id_to_vni(info->key.tun_id, geneveh->vni);
733         geneveh->proto_type = htons(ETH_P_TEB);
734         geneveh->rsvd2 = 0;
735
736         if (info->key.tun_flags & TUNNEL_GENEVE_OPT)
737                 ip_tunnel_info_opts_get(geneveh->options, info);
738 }
739
740 static int geneve_build_skb(struct dst_entry *dst, struct sk_buff *skb,
741                             const struct ip_tunnel_info *info,
742                             bool xnet, int ip_hdr_len)
743 {
744         bool udp_sum = !!(info->key.tun_flags & TUNNEL_CSUM);
745         struct genevehdr *gnvh;
746         int min_headroom;
747         int err;
748
749         skb_reset_mac_header(skb);
750         skb_scrub_packet(skb, xnet);
751
752         min_headroom = LL_RESERVED_SPACE(dst->dev) + dst->header_len +
753                        GENEVE_BASE_HLEN + info->options_len + ip_hdr_len;
754         err = skb_cow_head(skb, min_headroom);
755         if (unlikely(err))
756                 goto free_dst;
757
758         err = udp_tunnel_handle_offloads(skb, udp_sum);
759         if (err)
760                 goto free_dst;
761
762         gnvh = __skb_push(skb, sizeof(*gnvh) + info->options_len);
763         geneve_build_header(gnvh, info);
764         skb_set_inner_protocol(skb, htons(ETH_P_TEB));
765         return 0;
766
767 free_dst:
768         dst_release(dst);
769         return err;
770 }
771
772 static struct rtable *geneve_get_v4_rt(struct sk_buff *skb,
773                                        struct net_device *dev,
774                                        struct geneve_sock *gs4,
775                                        struct flowi4 *fl4,
776                                        const struct ip_tunnel_info *info)
777 {
778         bool use_cache = ip_tunnel_dst_cache_usable(skb, info);
779         struct geneve_dev *geneve = netdev_priv(dev);
780         struct dst_cache *dst_cache;
781         struct rtable *rt = NULL;
782         __u8 tos;
783
784         if (!gs4)
785                 return ERR_PTR(-EIO);
786
787         memset(fl4, 0, sizeof(*fl4));
788         fl4->flowi4_mark = skb->mark;
789         fl4->flowi4_proto = IPPROTO_UDP;
790         fl4->daddr = info->key.u.ipv4.dst;
791         fl4->saddr = info->key.u.ipv4.src;
792
793         tos = info->key.tos;
794         if ((tos == 1) && !geneve->collect_md) {
795                 tos = ip_tunnel_get_dsfield(ip_hdr(skb), skb);
796                 use_cache = false;
797         }
798         fl4->flowi4_tos = RT_TOS(tos);
799
800         dst_cache = (struct dst_cache *)&info->dst_cache;
801         if (use_cache) {
802                 rt = dst_cache_get_ip4(dst_cache, &fl4->saddr);
803                 if (rt)
804                         return rt;
805         }
806         rt = ip_route_output_key(geneve->net, fl4);
807         if (IS_ERR(rt)) {
808                 netdev_dbg(dev, "no route to %pI4\n", &fl4->daddr);
809                 return ERR_PTR(-ENETUNREACH);
810         }
811         if (rt->dst.dev == dev) { /* is this necessary? */
812                 netdev_dbg(dev, "circular route to %pI4\n", &fl4->daddr);
813                 ip_rt_put(rt);
814                 return ERR_PTR(-ELOOP);
815         }
816         if (use_cache)
817                 dst_cache_set_ip4(dst_cache, &rt->dst, fl4->saddr);
818         return rt;
819 }
820
821 #if IS_ENABLED(CONFIG_IPV6)
822 static struct dst_entry *geneve_get_v6_dst(struct sk_buff *skb,
823                                            struct net_device *dev,
824                                            struct geneve_sock *gs6,
825                                            struct flowi6 *fl6,
826                                            const struct ip_tunnel_info *info)
827 {
828         bool use_cache = ip_tunnel_dst_cache_usable(skb, info);
829         struct geneve_dev *geneve = netdev_priv(dev);
830         struct dst_entry *dst = NULL;
831         struct dst_cache *dst_cache;
832         __u8 prio;
833
834         if (!gs6)
835                 return ERR_PTR(-EIO);
836
837         memset(fl6, 0, sizeof(*fl6));
838         fl6->flowi6_mark = skb->mark;
839         fl6->flowi6_proto = IPPROTO_UDP;
840         fl6->daddr = info->key.u.ipv6.dst;
841         fl6->saddr = info->key.u.ipv6.src;
842         prio = info->key.tos;
843         if ((prio == 1) && !geneve->collect_md) {
844                 prio = ip_tunnel_get_dsfield(ip_hdr(skb), skb);
845                 use_cache = false;
846         }
847
848         fl6->flowlabel = ip6_make_flowinfo(RT_TOS(prio),
849                                            info->key.label);
850         dst_cache = (struct dst_cache *)&info->dst_cache;
851         if (use_cache) {
852                 dst = dst_cache_get_ip6(dst_cache, &fl6->saddr);
853                 if (dst)
854                         return dst;
855         }
856         dst = ipv6_stub->ipv6_dst_lookup_flow(geneve->net, gs6->sock->sk, fl6,
857                                               NULL);
858         if (IS_ERR(dst)) {
859                 netdev_dbg(dev, "no route to %pI6\n", &fl6->daddr);
860                 return ERR_PTR(-ENETUNREACH);
861         }
862         if (dst->dev == dev) { /* is this necessary? */
863                 netdev_dbg(dev, "circular route to %pI6\n", &fl6->daddr);
864                 dst_release(dst);
865                 return ERR_PTR(-ELOOP);
866         }
867
868         if (use_cache)
869                 dst_cache_set_ip6(dst_cache, dst, &fl6->saddr);
870         return dst;
871 }
872 #endif
873
874 static int geneve_xmit_skb(struct sk_buff *skb, struct net_device *dev,
875                            struct geneve_dev *geneve,
876                            const struct ip_tunnel_info *info)
877 {
878         bool xnet = !net_eq(geneve->net, dev_net(geneve->dev));
879         struct geneve_sock *gs4 = rcu_dereference(geneve->sock4);
880         const struct ip_tunnel_key *key = &info->key;
881         struct rtable *rt;
882         struct flowi4 fl4;
883         __u8 tos, ttl;
884         __be16 df = 0;
885         __be16 sport;
886         int err;
887
888         rt = geneve_get_v4_rt(skb, dev, gs4, &fl4, info);
889         if (IS_ERR(rt))
890                 return PTR_ERR(rt);
891
892         skb_tunnel_check_pmtu(skb, &rt->dst,
893                               GENEVE_IPV4_HLEN + info->options_len);
894
895         sport = udp_flow_src_port(geneve->net, skb, 1, USHRT_MAX, true);
896         if (geneve->collect_md) {
897                 tos = ip_tunnel_ecn_encap(key->tos, ip_hdr(skb), skb);
898                 ttl = key->ttl;
899
900                 df = key->tun_flags & TUNNEL_DONT_FRAGMENT ? htons(IP_DF) : 0;
901         } else {
902                 tos = ip_tunnel_ecn_encap(fl4.flowi4_tos, ip_hdr(skb), skb);
903                 if (geneve->ttl_inherit)
904                         ttl = ip_tunnel_get_ttl(ip_hdr(skb), skb);
905                 else
906                         ttl = key->ttl;
907                 ttl = ttl ? : ip4_dst_hoplimit(&rt->dst);
908
909                 if (geneve->df == GENEVE_DF_SET) {
910                         df = htons(IP_DF);
911                 } else if (geneve->df == GENEVE_DF_INHERIT) {
912                         struct ethhdr *eth = eth_hdr(skb);
913
914                         if (ntohs(eth->h_proto) == ETH_P_IPV6) {
915                                 df = htons(IP_DF);
916                         } else if (ntohs(eth->h_proto) == ETH_P_IP) {
917                                 struct iphdr *iph = ip_hdr(skb);
918
919                                 if (iph->frag_off & htons(IP_DF))
920                                         df = htons(IP_DF);
921                         }
922                 }
923         }
924
925         err = geneve_build_skb(&rt->dst, skb, info, xnet, sizeof(struct iphdr));
926         if (unlikely(err))
927                 return err;
928
929         udp_tunnel_xmit_skb(rt, gs4->sock->sk, skb, fl4.saddr, fl4.daddr,
930                             tos, ttl, df, sport, geneve->info.key.tp_dst,
931                             !net_eq(geneve->net, dev_net(geneve->dev)),
932                             !(info->key.tun_flags & TUNNEL_CSUM));
933         return 0;
934 }
935
936 #if IS_ENABLED(CONFIG_IPV6)
937 static int geneve6_xmit_skb(struct sk_buff *skb, struct net_device *dev,
938                             struct geneve_dev *geneve,
939                             const struct ip_tunnel_info *info)
940 {
941         bool xnet = !net_eq(geneve->net, dev_net(geneve->dev));
942         struct geneve_sock *gs6 = rcu_dereference(geneve->sock6);
943         const struct ip_tunnel_key *key = &info->key;
944         struct dst_entry *dst = NULL;
945         struct flowi6 fl6;
946         __u8 prio, ttl;
947         __be16 sport;
948         int err;
949
950         dst = geneve_get_v6_dst(skb, dev, gs6, &fl6, info);
951         if (IS_ERR(dst))
952                 return PTR_ERR(dst);
953
954         skb_tunnel_check_pmtu(skb, dst, GENEVE_IPV6_HLEN + info->options_len);
955
956         sport = udp_flow_src_port(geneve->net, skb, 1, USHRT_MAX, true);
957         if (geneve->collect_md) {
958                 prio = ip_tunnel_ecn_encap(key->tos, ip_hdr(skb), skb);
959                 ttl = key->ttl;
960         } else {
961                 prio = ip_tunnel_ecn_encap(ip6_tclass(fl6.flowlabel),
962                                            ip_hdr(skb), skb);
963                 if (geneve->ttl_inherit)
964                         ttl = ip_tunnel_get_ttl(ip_hdr(skb), skb);
965                 else
966                         ttl = key->ttl;
967                 ttl = ttl ? : ip6_dst_hoplimit(dst);
968         }
969         err = geneve_build_skb(dst, skb, info, xnet, sizeof(struct ipv6hdr));
970         if (unlikely(err))
971                 return err;
972
973         udp_tunnel6_xmit_skb(dst, gs6->sock->sk, skb, dev,
974                              &fl6.saddr, &fl6.daddr, prio, ttl,
975                              info->key.label, sport, geneve->info.key.tp_dst,
976                              !(info->key.tun_flags & TUNNEL_CSUM));
977         return 0;
978 }
979 #endif
980
981 static netdev_tx_t geneve_xmit(struct sk_buff *skb, struct net_device *dev)
982 {
983         struct geneve_dev *geneve = netdev_priv(dev);
984         struct ip_tunnel_info *info = NULL;
985         int err;
986
987         if (geneve->collect_md) {
988                 info = skb_tunnel_info(skb);
989                 if (unlikely(!info || !(info->mode & IP_TUNNEL_INFO_TX))) {
990                         netdev_dbg(dev, "no tunnel metadata\n");
991                         dev_kfree_skb(skb);
992                         dev->stats.tx_dropped++;
993                         return NETDEV_TX_OK;
994                 }
995         } else {
996                 info = &geneve->info;
997         }
998
999         rcu_read_lock();
1000 #if IS_ENABLED(CONFIG_IPV6)
1001         if (info->mode & IP_TUNNEL_INFO_IPV6)
1002                 err = geneve6_xmit_skb(skb, dev, geneve, info);
1003         else
1004 #endif
1005                 err = geneve_xmit_skb(skb, dev, geneve, info);
1006         rcu_read_unlock();
1007
1008         if (likely(!err))
1009                 return NETDEV_TX_OK;
1010
1011         dev_kfree_skb(skb);
1012
1013         if (err == -ELOOP)
1014                 dev->stats.collisions++;
1015         else if (err == -ENETUNREACH)
1016                 dev->stats.tx_carrier_errors++;
1017
1018         dev->stats.tx_errors++;
1019         return NETDEV_TX_OK;
1020 }
1021
1022 static int geneve_change_mtu(struct net_device *dev, int new_mtu)
1023 {
1024         if (new_mtu > dev->max_mtu)
1025                 new_mtu = dev->max_mtu;
1026         else if (new_mtu < dev->min_mtu)
1027                 new_mtu = dev->min_mtu;
1028
1029         dev->mtu = new_mtu;
1030         return 0;
1031 }
1032
1033 static int geneve_fill_metadata_dst(struct net_device *dev, struct sk_buff *skb)
1034 {
1035         struct ip_tunnel_info *info = skb_tunnel_info(skb);
1036         struct geneve_dev *geneve = netdev_priv(dev);
1037
1038         if (ip_tunnel_info_af(info) == AF_INET) {
1039                 struct rtable *rt;
1040                 struct flowi4 fl4;
1041                 struct geneve_sock *gs4 = rcu_dereference(geneve->sock4);
1042
1043                 rt = geneve_get_v4_rt(skb, dev, gs4, &fl4, info);
1044                 if (IS_ERR(rt))
1045                         return PTR_ERR(rt);
1046
1047                 ip_rt_put(rt);
1048                 info->key.u.ipv4.src = fl4.saddr;
1049 #if IS_ENABLED(CONFIG_IPV6)
1050         } else if (ip_tunnel_info_af(info) == AF_INET6) {
1051                 struct dst_entry *dst;
1052                 struct flowi6 fl6;
1053                 struct geneve_sock *gs6 = rcu_dereference(geneve->sock6);
1054
1055                 dst = geneve_get_v6_dst(skb, dev, gs6, &fl6, info);
1056                 if (IS_ERR(dst))
1057                         return PTR_ERR(dst);
1058
1059                 dst_release(dst);
1060                 info->key.u.ipv6.src = fl6.saddr;
1061 #endif
1062         } else {
1063                 return -EINVAL;
1064         }
1065
1066         info->key.tp_src = udp_flow_src_port(geneve->net, skb,
1067                                              1, USHRT_MAX, true);
1068         info->key.tp_dst = geneve->info.key.tp_dst;
1069         return 0;
1070 }
1071
1072 static const struct net_device_ops geneve_netdev_ops = {
1073         .ndo_init               = geneve_init,
1074         .ndo_uninit             = geneve_uninit,
1075         .ndo_open               = geneve_open,
1076         .ndo_stop               = geneve_stop,
1077         .ndo_start_xmit         = geneve_xmit,
1078         .ndo_get_stats64        = ip_tunnel_get_stats64,
1079         .ndo_change_mtu         = geneve_change_mtu,
1080         .ndo_validate_addr      = eth_validate_addr,
1081         .ndo_set_mac_address    = eth_mac_addr,
1082         .ndo_fill_metadata_dst  = geneve_fill_metadata_dst,
1083 };
1084
1085 static void geneve_get_drvinfo(struct net_device *dev,
1086                                struct ethtool_drvinfo *drvinfo)
1087 {
1088         strlcpy(drvinfo->version, GENEVE_NETDEV_VER, sizeof(drvinfo->version));
1089         strlcpy(drvinfo->driver, "geneve", sizeof(drvinfo->driver));
1090 }
1091
1092 static const struct ethtool_ops geneve_ethtool_ops = {
1093         .get_drvinfo    = geneve_get_drvinfo,
1094         .get_link       = ethtool_op_get_link,
1095 };
1096
1097 /* Info for udev, that this is a virtual tunnel endpoint */
1098 static struct device_type geneve_type = {
1099         .name = "geneve",
1100 };
1101
1102 /* Calls the ndo_udp_tunnel_add of the caller in order to
1103  * supply the listening GENEVE udp ports. Callers are expected
1104  * to implement the ndo_udp_tunnel_add.
1105  */
1106 static void geneve_offload_rx_ports(struct net_device *dev, bool push)
1107 {
1108         struct net *net = dev_net(dev);
1109         struct geneve_net *gn = net_generic(net, geneve_net_id);
1110         struct geneve_sock *gs;
1111
1112         rcu_read_lock();
1113         list_for_each_entry_rcu(gs, &gn->sock_list, list) {
1114                 if (push) {
1115                         udp_tunnel_push_rx_port(dev, gs->sock,
1116                                                 UDP_TUNNEL_TYPE_GENEVE);
1117                 } else {
1118                         udp_tunnel_drop_rx_port(dev, gs->sock,
1119                                                 UDP_TUNNEL_TYPE_GENEVE);
1120                 }
1121         }
1122         rcu_read_unlock();
1123 }
1124
1125 /* Initialize the device structure. */
1126 static void geneve_setup(struct net_device *dev)
1127 {
1128         ether_setup(dev);
1129
1130         dev->netdev_ops = &geneve_netdev_ops;
1131         dev->ethtool_ops = &geneve_ethtool_ops;
1132         dev->needs_free_netdev = true;
1133
1134         SET_NETDEV_DEVTYPE(dev, &geneve_type);
1135
1136         dev->features    |= NETIF_F_LLTX;
1137         dev->features    |= NETIF_F_SG | NETIF_F_HW_CSUM;
1138         dev->features    |= NETIF_F_RXCSUM;
1139         dev->features    |= NETIF_F_GSO_SOFTWARE;
1140
1141         dev->hw_features |= NETIF_F_SG | NETIF_F_HW_CSUM | NETIF_F_RXCSUM;
1142         dev->hw_features |= NETIF_F_GSO_SOFTWARE;
1143
1144         /* MTU range: 68 - (something less than 65535) */
1145         dev->min_mtu = ETH_MIN_MTU;
1146         /* The max_mtu calculation does not take account of GENEVE
1147          * options, to avoid excluding potentially valid
1148          * configurations. This will be further reduced by IPvX hdr size.
1149          */
1150         dev->max_mtu = IP_MAX_MTU - GENEVE_BASE_HLEN - dev->hard_header_len;
1151
1152         netif_keep_dst(dev);
1153         dev->priv_flags &= ~IFF_TX_SKB_SHARING;
1154         dev->priv_flags |= IFF_LIVE_ADDR_CHANGE | IFF_NO_QUEUE;
1155         eth_hw_addr_random(dev);
1156 }
1157
1158 static const struct nla_policy geneve_policy[IFLA_GENEVE_MAX + 1] = {
1159         [IFLA_GENEVE_ID]                = { .type = NLA_U32 },
1160         [IFLA_GENEVE_REMOTE]            = { .len = FIELD_SIZEOF(struct iphdr, daddr) },
1161         [IFLA_GENEVE_REMOTE6]           = { .len = sizeof(struct in6_addr) },
1162         [IFLA_GENEVE_TTL]               = { .type = NLA_U8 },
1163         [IFLA_GENEVE_TOS]               = { .type = NLA_U8 },
1164         [IFLA_GENEVE_LABEL]             = { .type = NLA_U32 },
1165         [IFLA_GENEVE_PORT]              = { .type = NLA_U16 },
1166         [IFLA_GENEVE_COLLECT_METADATA]  = { .type = NLA_FLAG },
1167         [IFLA_GENEVE_UDP_CSUM]          = { .type = NLA_U8 },
1168         [IFLA_GENEVE_UDP_ZERO_CSUM6_TX] = { .type = NLA_U8 },
1169         [IFLA_GENEVE_UDP_ZERO_CSUM6_RX] = { .type = NLA_U8 },
1170         [IFLA_GENEVE_TTL_INHERIT]       = { .type = NLA_U8 },
1171         [IFLA_GENEVE_DF]                = { .type = NLA_U8 },
1172 };
1173
1174 static int geneve_validate(struct nlattr *tb[], struct nlattr *data[],
1175                            struct netlink_ext_ack *extack)
1176 {
1177         if (tb[IFLA_ADDRESS]) {
1178                 if (nla_len(tb[IFLA_ADDRESS]) != ETH_ALEN) {
1179                         NL_SET_ERR_MSG_ATTR(extack, tb[IFLA_ADDRESS],
1180                                             "Provided link layer address is not Ethernet");
1181                         return -EINVAL;
1182                 }
1183
1184                 if (!is_valid_ether_addr(nla_data(tb[IFLA_ADDRESS]))) {
1185                         NL_SET_ERR_MSG_ATTR(extack, tb[IFLA_ADDRESS],
1186                                             "Provided Ethernet address is not unicast");
1187                         return -EADDRNOTAVAIL;
1188                 }
1189         }
1190
1191         if (!data) {
1192                 NL_SET_ERR_MSG(extack,
1193                                "Not enough attributes provided to perform the operation");
1194                 return -EINVAL;
1195         }
1196
1197         if (data[IFLA_GENEVE_ID]) {
1198                 __u32 vni =  nla_get_u32(data[IFLA_GENEVE_ID]);
1199
1200                 if (vni >= GENEVE_N_VID) {
1201                         NL_SET_ERR_MSG_ATTR(extack, data[IFLA_GENEVE_ID],
1202                                             "Geneve ID must be lower than 16777216");
1203                         return -ERANGE;
1204                 }
1205         }
1206
1207         if (data[IFLA_GENEVE_DF]) {
1208                 enum ifla_geneve_df df = nla_get_u8(data[IFLA_GENEVE_DF]);
1209
1210                 if (df < 0 || df > GENEVE_DF_MAX) {
1211                         NL_SET_ERR_MSG_ATTR(extack, data[IFLA_GENEVE_DF],
1212                                             "Invalid DF attribute");
1213                         return -EINVAL;
1214                 }
1215         }
1216
1217         return 0;
1218 }
1219
1220 static struct geneve_dev *geneve_find_dev(struct geneve_net *gn,
1221                                           const struct ip_tunnel_info *info,
1222                                           bool *tun_on_same_port,
1223                                           bool *tun_collect_md)
1224 {
1225         struct geneve_dev *geneve, *t = NULL;
1226
1227         *tun_on_same_port = false;
1228         *tun_collect_md = false;
1229         list_for_each_entry(geneve, &gn->geneve_list, next) {
1230                 if (info->key.tp_dst == geneve->info.key.tp_dst) {
1231                         *tun_collect_md = geneve->collect_md;
1232                         *tun_on_same_port = true;
1233                 }
1234                 if (info->key.tun_id == geneve->info.key.tun_id &&
1235                     info->key.tp_dst == geneve->info.key.tp_dst &&
1236                     !memcmp(&info->key.u, &geneve->info.key.u, sizeof(info->key.u)))
1237                         t = geneve;
1238         }
1239         return t;
1240 }
1241
1242 static bool is_tnl_info_zero(const struct ip_tunnel_info *info)
1243 {
1244         return !(info->key.tun_id || info->key.tun_flags || info->key.tos ||
1245                  info->key.ttl || info->key.label || info->key.tp_src ||
1246                  memchr_inv(&info->key.u, 0, sizeof(info->key.u)));
1247 }
1248
1249 static bool geneve_dst_addr_equal(struct ip_tunnel_info *a,
1250                                   struct ip_tunnel_info *b)
1251 {
1252         if (ip_tunnel_info_af(a) == AF_INET)
1253                 return a->key.u.ipv4.dst == b->key.u.ipv4.dst;
1254         else
1255                 return ipv6_addr_equal(&a->key.u.ipv6.dst, &b->key.u.ipv6.dst);
1256 }
1257
1258 static int geneve_configure(struct net *net, struct net_device *dev,
1259                             struct netlink_ext_ack *extack,
1260                             const struct ip_tunnel_info *info,
1261                             bool metadata, bool ipv6_rx_csum,
1262                             bool ttl_inherit, enum ifla_geneve_df df)
1263 {
1264         struct geneve_net *gn = net_generic(net, geneve_net_id);
1265         struct geneve_dev *t, *geneve = netdev_priv(dev);
1266         bool tun_collect_md, tun_on_same_port;
1267         int err, encap_len;
1268
1269         if (metadata && !is_tnl_info_zero(info)) {
1270                 NL_SET_ERR_MSG(extack,
1271                                "Device is externally controlled, so attributes (VNI, Port, and so on) must not be specified");
1272                 return -EINVAL;
1273         }
1274
1275         geneve->net = net;
1276         geneve->dev = dev;
1277
1278         t = geneve_find_dev(gn, info, &tun_on_same_port, &tun_collect_md);
1279         if (t)
1280                 return -EBUSY;
1281
1282         /* make enough headroom for basic scenario */
1283         encap_len = GENEVE_BASE_HLEN + ETH_HLEN;
1284         if (!metadata && ip_tunnel_info_af(info) == AF_INET) {
1285                 encap_len += sizeof(struct iphdr);
1286                 dev->max_mtu -= sizeof(struct iphdr);
1287         } else {
1288                 encap_len += sizeof(struct ipv6hdr);
1289                 dev->max_mtu -= sizeof(struct ipv6hdr);
1290         }
1291         dev->needed_headroom = encap_len + ETH_HLEN;
1292
1293         if (metadata) {
1294                 if (tun_on_same_port) {
1295                         NL_SET_ERR_MSG(extack,
1296                                        "There can be only one externally controlled device on a destination port");
1297                         return -EPERM;
1298                 }
1299         } else {
1300                 if (tun_collect_md) {
1301                         NL_SET_ERR_MSG(extack,
1302                                        "There already exists an externally controlled device on this destination port");
1303                         return -EPERM;
1304                 }
1305         }
1306
1307         dst_cache_reset(&geneve->info.dst_cache);
1308         geneve->info = *info;
1309         geneve->collect_md = metadata;
1310         geneve->use_udp6_rx_checksums = ipv6_rx_csum;
1311         geneve->ttl_inherit = ttl_inherit;
1312         geneve->df = df;
1313
1314         err = register_netdevice(dev);
1315         if (err)
1316                 return err;
1317
1318         list_add(&geneve->next, &gn->geneve_list);
1319         return 0;
1320 }
1321
1322 static void init_tnl_info(struct ip_tunnel_info *info, __u16 dst_port)
1323 {
1324         memset(info, 0, sizeof(*info));
1325         info->key.tp_dst = htons(dst_port);
1326 }
1327
1328 static int geneve_nl2info(struct nlattr *tb[], struct nlattr *data[],
1329                           struct netlink_ext_ack *extack,
1330                           struct ip_tunnel_info *info, bool *metadata,
1331                           bool *use_udp6_rx_checksums, bool *ttl_inherit,
1332                           enum ifla_geneve_df *df, bool changelink)
1333 {
1334         int attrtype;
1335
1336         if (data[IFLA_GENEVE_REMOTE] && data[IFLA_GENEVE_REMOTE6]) {
1337                 NL_SET_ERR_MSG(extack,
1338                                "Cannot specify both IPv4 and IPv6 Remote addresses");
1339                 return -EINVAL;
1340         }
1341
1342         if (data[IFLA_GENEVE_REMOTE]) {
1343                 if (changelink && (ip_tunnel_info_af(info) == AF_INET6)) {
1344                         attrtype = IFLA_GENEVE_REMOTE;
1345                         goto change_notsup;
1346                 }
1347
1348                 info->key.u.ipv4.dst =
1349                         nla_get_in_addr(data[IFLA_GENEVE_REMOTE]);
1350
1351                 if (ipv4_is_multicast(info->key.u.ipv4.dst)) {
1352                         NL_SET_ERR_MSG_ATTR(extack, data[IFLA_GENEVE_REMOTE],
1353                                             "Remote IPv4 address cannot be Multicast");
1354                         return -EINVAL;
1355                 }
1356         }
1357
1358         if (data[IFLA_GENEVE_REMOTE6]) {
1359 #if IS_ENABLED(CONFIG_IPV6)
1360                 if (changelink && (ip_tunnel_info_af(info) == AF_INET)) {
1361                         attrtype = IFLA_GENEVE_REMOTE6;
1362                         goto change_notsup;
1363                 }
1364
1365                 info->mode = IP_TUNNEL_INFO_IPV6;
1366                 info->key.u.ipv6.dst =
1367                         nla_get_in6_addr(data[IFLA_GENEVE_REMOTE6]);
1368
1369                 if (ipv6_addr_type(&info->key.u.ipv6.dst) &
1370                     IPV6_ADDR_LINKLOCAL) {
1371                         NL_SET_ERR_MSG_ATTR(extack, data[IFLA_GENEVE_REMOTE6],
1372                                             "Remote IPv6 address cannot be link-local");
1373                         return -EINVAL;
1374                 }
1375                 if (ipv6_addr_is_multicast(&info->key.u.ipv6.dst)) {
1376                         NL_SET_ERR_MSG_ATTR(extack, data[IFLA_GENEVE_REMOTE6],
1377                                             "Remote IPv6 address cannot be Multicast");
1378                         return -EINVAL;
1379                 }
1380                 info->key.tun_flags |= TUNNEL_CSUM;
1381                 *use_udp6_rx_checksums = true;
1382 #else
1383                 NL_SET_ERR_MSG_ATTR(extack, data[IFLA_GENEVE_REMOTE6],
1384                                     "IPv6 support not enabled in the kernel");
1385                 return -EPFNOSUPPORT;
1386 #endif
1387         }
1388
1389         if (data[IFLA_GENEVE_ID]) {
1390                 __u32 vni;
1391                 __u8 tvni[3];
1392                 __be64 tunid;
1393
1394                 vni = nla_get_u32(data[IFLA_GENEVE_ID]);
1395                 tvni[0] = (vni & 0x00ff0000) >> 16;
1396                 tvni[1] = (vni & 0x0000ff00) >> 8;
1397                 tvni[2] =  vni & 0x000000ff;
1398
1399                 tunid = vni_to_tunnel_id(tvni);
1400                 if (changelink && (tunid != info->key.tun_id)) {
1401                         attrtype = IFLA_GENEVE_ID;
1402                         goto change_notsup;
1403                 }
1404                 info->key.tun_id = tunid;
1405         }
1406
1407         if (data[IFLA_GENEVE_TTL_INHERIT]) {
1408                 if (nla_get_u8(data[IFLA_GENEVE_TTL_INHERIT]))
1409                         *ttl_inherit = true;
1410                 else
1411                         *ttl_inherit = false;
1412         } else if (data[IFLA_GENEVE_TTL]) {
1413                 info->key.ttl = nla_get_u8(data[IFLA_GENEVE_TTL]);
1414                 *ttl_inherit = false;
1415         }
1416
1417         if (data[IFLA_GENEVE_TOS])
1418                 info->key.tos = nla_get_u8(data[IFLA_GENEVE_TOS]);
1419
1420         if (data[IFLA_GENEVE_DF])
1421                 *df = nla_get_u8(data[IFLA_GENEVE_DF]);
1422
1423         if (data[IFLA_GENEVE_LABEL]) {
1424                 info->key.label = nla_get_be32(data[IFLA_GENEVE_LABEL]) &
1425                                   IPV6_FLOWLABEL_MASK;
1426                 if (info->key.label && (!(info->mode & IP_TUNNEL_INFO_IPV6))) {
1427                         NL_SET_ERR_MSG_ATTR(extack, data[IFLA_GENEVE_LABEL],
1428                                             "Label attribute only applies for IPv6 Geneve devices");
1429                         return -EINVAL;
1430                 }
1431         }
1432
1433         if (data[IFLA_GENEVE_PORT]) {
1434                 if (changelink) {
1435                         attrtype = IFLA_GENEVE_PORT;
1436                         goto change_notsup;
1437                 }
1438                 info->key.tp_dst = nla_get_be16(data[IFLA_GENEVE_PORT]);
1439         }
1440
1441         if (data[IFLA_GENEVE_COLLECT_METADATA]) {
1442                 if (changelink) {
1443                         attrtype = IFLA_GENEVE_COLLECT_METADATA;
1444                         goto change_notsup;
1445                 }
1446                 *metadata = true;
1447         }
1448
1449         if (data[IFLA_GENEVE_UDP_CSUM]) {
1450                 if (changelink) {
1451                         attrtype = IFLA_GENEVE_UDP_CSUM;
1452                         goto change_notsup;
1453                 }
1454                 if (nla_get_u8(data[IFLA_GENEVE_UDP_CSUM]))
1455                         info->key.tun_flags |= TUNNEL_CSUM;
1456         }
1457
1458         if (data[IFLA_GENEVE_UDP_ZERO_CSUM6_TX]) {
1459 #if IS_ENABLED(CONFIG_IPV6)
1460                 if (changelink) {
1461                         attrtype = IFLA_GENEVE_UDP_ZERO_CSUM6_TX;
1462                         goto change_notsup;
1463                 }
1464                 if (nla_get_u8(data[IFLA_GENEVE_UDP_ZERO_CSUM6_TX]))
1465                         info->key.tun_flags &= ~TUNNEL_CSUM;
1466 #else
1467                 NL_SET_ERR_MSG_ATTR(extack, data[IFLA_GENEVE_UDP_ZERO_CSUM6_TX],
1468                                     "IPv6 support not enabled in the kernel");
1469                 return -EPFNOSUPPORT;
1470 #endif
1471         }
1472
1473         if (data[IFLA_GENEVE_UDP_ZERO_CSUM6_RX]) {
1474 #if IS_ENABLED(CONFIG_IPV6)
1475                 if (changelink) {
1476                         attrtype = IFLA_GENEVE_UDP_ZERO_CSUM6_RX;
1477                         goto change_notsup;
1478                 }
1479                 if (nla_get_u8(data[IFLA_GENEVE_UDP_ZERO_CSUM6_RX]))
1480                         *use_udp6_rx_checksums = false;
1481 #else
1482                 NL_SET_ERR_MSG_ATTR(extack, data[IFLA_GENEVE_UDP_ZERO_CSUM6_RX],
1483                                     "IPv6 support not enabled in the kernel");
1484                 return -EPFNOSUPPORT;
1485 #endif
1486         }
1487
1488         return 0;
1489 change_notsup:
1490         NL_SET_ERR_MSG_ATTR(extack, data[attrtype],
1491                             "Changing VNI, Port, endpoint IP address family, external, and UDP checksum attributes are not supported");
1492         return -EOPNOTSUPP;
1493 }
1494
1495 static void geneve_link_config(struct net_device *dev,
1496                                struct ip_tunnel_info *info, struct nlattr *tb[])
1497 {
1498         struct geneve_dev *geneve = netdev_priv(dev);
1499         int ldev_mtu = 0;
1500
1501         if (tb[IFLA_MTU]) {
1502                 geneve_change_mtu(dev, nla_get_u32(tb[IFLA_MTU]));
1503                 return;
1504         }
1505
1506         switch (ip_tunnel_info_af(info)) {
1507         case AF_INET: {
1508                 struct flowi4 fl4 = { .daddr = info->key.u.ipv4.dst };
1509                 struct rtable *rt = ip_route_output_key(geneve->net, &fl4);
1510
1511                 if (!IS_ERR(rt) && rt->dst.dev) {
1512                         ldev_mtu = rt->dst.dev->mtu - GENEVE_IPV4_HLEN;
1513                         ip_rt_put(rt);
1514                 }
1515                 break;
1516         }
1517 #if IS_ENABLED(CONFIG_IPV6)
1518         case AF_INET6: {
1519                 struct rt6_info *rt;
1520
1521                 if (!__in6_dev_get(dev))
1522                         break;
1523
1524                 rt = rt6_lookup(geneve->net, &info->key.u.ipv6.dst, NULL, 0,
1525                                 NULL, 0);
1526
1527                 if (rt && rt->dst.dev)
1528                         ldev_mtu = rt->dst.dev->mtu - GENEVE_IPV6_HLEN;
1529                 ip6_rt_put(rt);
1530                 break;
1531         }
1532 #endif
1533         }
1534
1535         if (ldev_mtu <= 0)
1536                 return;
1537
1538         geneve_change_mtu(dev, ldev_mtu - info->options_len);
1539 }
1540
1541 static int geneve_newlink(struct net *net, struct net_device *dev,
1542                           struct nlattr *tb[], struct nlattr *data[],
1543                           struct netlink_ext_ack *extack)
1544 {
1545         enum ifla_geneve_df df = GENEVE_DF_UNSET;
1546         bool use_udp6_rx_checksums = false;
1547         struct ip_tunnel_info info;
1548         bool ttl_inherit = false;
1549         bool metadata = false;
1550         int err;
1551
1552         init_tnl_info(&info, GENEVE_UDP_PORT);
1553         err = geneve_nl2info(tb, data, extack, &info, &metadata,
1554                              &use_udp6_rx_checksums, &ttl_inherit, &df, false);
1555         if (err)
1556                 return err;
1557
1558         err = geneve_configure(net, dev, extack, &info, metadata,
1559                                use_udp6_rx_checksums, ttl_inherit, df);
1560         if (err)
1561                 return err;
1562
1563         geneve_link_config(dev, &info, tb);
1564
1565         return 0;
1566 }
1567
1568 /* Quiesces the geneve device data path for both TX and RX.
1569  *
1570  * On transmit geneve checks for non-NULL geneve_sock before it proceeds.
1571  * So, if we set that socket to NULL under RCU and wait for synchronize_net()
1572  * to complete for the existing set of in-flight packets to be transmitted,
1573  * then we would have quiesced the transmit data path. All the future packets
1574  * will get dropped until we unquiesce the data path.
1575  *
1576  * On receive geneve dereference the geneve_sock stashed in the socket. So,
1577  * if we set that to NULL under RCU and wait for synchronize_net() to
1578  * complete, then we would have quiesced the receive data path.
1579  */
1580 static void geneve_quiesce(struct geneve_dev *geneve, struct geneve_sock **gs4,
1581                            struct geneve_sock **gs6)
1582 {
1583         *gs4 = rtnl_dereference(geneve->sock4);
1584         rcu_assign_pointer(geneve->sock4, NULL);
1585         if (*gs4)
1586                 rcu_assign_sk_user_data((*gs4)->sock->sk, NULL);
1587 #if IS_ENABLED(CONFIG_IPV6)
1588         *gs6 = rtnl_dereference(geneve->sock6);
1589         rcu_assign_pointer(geneve->sock6, NULL);
1590         if (*gs6)
1591                 rcu_assign_sk_user_data((*gs6)->sock->sk, NULL);
1592 #else
1593         *gs6 = NULL;
1594 #endif
1595         synchronize_net();
1596 }
1597
1598 /* Resumes the geneve device data path for both TX and RX. */
1599 static void geneve_unquiesce(struct geneve_dev *geneve, struct geneve_sock *gs4,
1600                              struct geneve_sock __maybe_unused *gs6)
1601 {
1602         rcu_assign_pointer(geneve->sock4, gs4);
1603         if (gs4)
1604                 rcu_assign_sk_user_data(gs4->sock->sk, gs4);
1605 #if IS_ENABLED(CONFIG_IPV6)
1606         rcu_assign_pointer(geneve->sock6, gs6);
1607         if (gs6)
1608                 rcu_assign_sk_user_data(gs6->sock->sk, gs6);
1609 #endif
1610         synchronize_net();
1611 }
1612
1613 static int geneve_changelink(struct net_device *dev, struct nlattr *tb[],
1614                              struct nlattr *data[],
1615                              struct netlink_ext_ack *extack)
1616 {
1617         struct geneve_dev *geneve = netdev_priv(dev);
1618         struct geneve_sock *gs4, *gs6;
1619         struct ip_tunnel_info info;
1620         bool metadata;
1621         bool use_udp6_rx_checksums;
1622         enum ifla_geneve_df df;
1623         bool ttl_inherit;
1624         int err;
1625
1626         /* If the geneve device is configured for metadata (or externally
1627          * controlled, for example, OVS), then nothing can be changed.
1628          */
1629         if (geneve->collect_md)
1630                 return -EOPNOTSUPP;
1631
1632         /* Start with the existing info. */
1633         memcpy(&info, &geneve->info, sizeof(info));
1634         metadata = geneve->collect_md;
1635         use_udp6_rx_checksums = geneve->use_udp6_rx_checksums;
1636         ttl_inherit = geneve->ttl_inherit;
1637         err = geneve_nl2info(tb, data, extack, &info, &metadata,
1638                              &use_udp6_rx_checksums, &ttl_inherit, &df, true);
1639         if (err)
1640                 return err;
1641
1642         if (!geneve_dst_addr_equal(&geneve->info, &info)) {
1643                 dst_cache_reset(&info.dst_cache);
1644                 geneve_link_config(dev, &info, tb);
1645         }
1646
1647         geneve_quiesce(geneve, &gs4, &gs6);
1648         geneve->info = info;
1649         geneve->collect_md = metadata;
1650         geneve->use_udp6_rx_checksums = use_udp6_rx_checksums;
1651         geneve->ttl_inherit = ttl_inherit;
1652         geneve->df = df;
1653         geneve_unquiesce(geneve, gs4, gs6);
1654
1655         return 0;
1656 }
1657
1658 static void geneve_dellink(struct net_device *dev, struct list_head *head)
1659 {
1660         struct geneve_dev *geneve = netdev_priv(dev);
1661
1662         list_del(&geneve->next);
1663         unregister_netdevice_queue(dev, head);
1664 }
1665
1666 static size_t geneve_get_size(const struct net_device *dev)
1667 {
1668         return nla_total_size(sizeof(__u32)) +  /* IFLA_GENEVE_ID */
1669                 nla_total_size(sizeof(struct in6_addr)) + /* IFLA_GENEVE_REMOTE{6} */
1670                 nla_total_size(sizeof(__u8)) +  /* IFLA_GENEVE_TTL */
1671                 nla_total_size(sizeof(__u8)) +  /* IFLA_GENEVE_TOS */
1672                 nla_total_size(sizeof(__u8)) +  /* IFLA_GENEVE_DF */
1673                 nla_total_size(sizeof(__be32)) +  /* IFLA_GENEVE_LABEL */
1674                 nla_total_size(sizeof(__be16)) +  /* IFLA_GENEVE_PORT */
1675                 nla_total_size(0) +      /* IFLA_GENEVE_COLLECT_METADATA */
1676                 nla_total_size(sizeof(__u8)) + /* IFLA_GENEVE_UDP_CSUM */
1677                 nla_total_size(sizeof(__u8)) + /* IFLA_GENEVE_UDP_ZERO_CSUM6_TX */
1678                 nla_total_size(sizeof(__u8)) + /* IFLA_GENEVE_UDP_ZERO_CSUM6_RX */
1679                 nla_total_size(sizeof(__u8)) + /* IFLA_GENEVE_TTL_INHERIT */
1680                 0;
1681 }
1682
1683 static int geneve_fill_info(struct sk_buff *skb, const struct net_device *dev)
1684 {
1685         struct geneve_dev *geneve = netdev_priv(dev);
1686         struct ip_tunnel_info *info = &geneve->info;
1687         bool ttl_inherit = geneve->ttl_inherit;
1688         bool metadata = geneve->collect_md;
1689         __u8 tmp_vni[3];
1690         __u32 vni;
1691
1692         tunnel_id_to_vni(info->key.tun_id, tmp_vni);
1693         vni = (tmp_vni[0] << 16) | (tmp_vni[1] << 8) | tmp_vni[2];
1694         if (nla_put_u32(skb, IFLA_GENEVE_ID, vni))
1695                 goto nla_put_failure;
1696
1697         if (!metadata && ip_tunnel_info_af(info) == AF_INET) {
1698                 if (nla_put_in_addr(skb, IFLA_GENEVE_REMOTE,
1699                                     info->key.u.ipv4.dst))
1700                         goto nla_put_failure;
1701                 if (nla_put_u8(skb, IFLA_GENEVE_UDP_CSUM,
1702                                !!(info->key.tun_flags & TUNNEL_CSUM)))
1703                         goto nla_put_failure;
1704
1705 #if IS_ENABLED(CONFIG_IPV6)
1706         } else if (!metadata) {
1707                 if (nla_put_in6_addr(skb, IFLA_GENEVE_REMOTE6,
1708                                      &info->key.u.ipv6.dst))
1709                         goto nla_put_failure;
1710                 if (nla_put_u8(skb, IFLA_GENEVE_UDP_ZERO_CSUM6_TX,
1711                                !(info->key.tun_flags & TUNNEL_CSUM)))
1712                         goto nla_put_failure;
1713 #endif
1714         }
1715
1716         if (nla_put_u8(skb, IFLA_GENEVE_TTL, info->key.ttl) ||
1717             nla_put_u8(skb, IFLA_GENEVE_TOS, info->key.tos) ||
1718             nla_put_be32(skb, IFLA_GENEVE_LABEL, info->key.label))
1719                 goto nla_put_failure;
1720
1721         if (nla_put_u8(skb, IFLA_GENEVE_DF, geneve->df))
1722                 goto nla_put_failure;
1723
1724         if (nla_put_be16(skb, IFLA_GENEVE_PORT, info->key.tp_dst))
1725                 goto nla_put_failure;
1726
1727         if (metadata && nla_put_flag(skb, IFLA_GENEVE_COLLECT_METADATA))
1728                 goto nla_put_failure;
1729
1730 #if IS_ENABLED(CONFIG_IPV6)
1731         if (nla_put_u8(skb, IFLA_GENEVE_UDP_ZERO_CSUM6_RX,
1732                        !geneve->use_udp6_rx_checksums))
1733                 goto nla_put_failure;
1734 #endif
1735
1736         if (nla_put_u8(skb, IFLA_GENEVE_TTL_INHERIT, ttl_inherit))
1737                 goto nla_put_failure;
1738
1739         return 0;
1740
1741 nla_put_failure:
1742         return -EMSGSIZE;
1743 }
1744
1745 static struct rtnl_link_ops geneve_link_ops __read_mostly = {
1746         .kind           = "geneve",
1747         .maxtype        = IFLA_GENEVE_MAX,
1748         .policy         = geneve_policy,
1749         .priv_size      = sizeof(struct geneve_dev),
1750         .setup          = geneve_setup,
1751         .validate       = geneve_validate,
1752         .newlink        = geneve_newlink,
1753         .changelink     = geneve_changelink,
1754         .dellink        = geneve_dellink,
1755         .get_size       = geneve_get_size,
1756         .fill_info      = geneve_fill_info,
1757 };
1758
1759 struct net_device *geneve_dev_create_fb(struct net *net, const char *name,
1760                                         u8 name_assign_type, u16 dst_port)
1761 {
1762         struct nlattr *tb[IFLA_MAX + 1];
1763         struct ip_tunnel_info info;
1764         struct net_device *dev;
1765         LIST_HEAD(list_kill);
1766         int err;
1767
1768         memset(tb, 0, sizeof(tb));
1769         dev = rtnl_create_link(net, name, name_assign_type,
1770                                &geneve_link_ops, tb, NULL);
1771         if (IS_ERR(dev))
1772                 return dev;
1773
1774         init_tnl_info(&info, dst_port);
1775         err = geneve_configure(net, dev, NULL, &info,
1776                                true, true, false, GENEVE_DF_UNSET);
1777         if (err) {
1778                 free_netdev(dev);
1779                 return ERR_PTR(err);
1780         }
1781
1782         /* openvswitch users expect packet sizes to be unrestricted,
1783          * so set the largest MTU we can.
1784          */
1785         err = geneve_change_mtu(dev, IP_MAX_MTU);
1786         if (err)
1787                 goto err;
1788
1789         err = rtnl_configure_link(dev, NULL);
1790         if (err < 0)
1791                 goto err;
1792
1793         return dev;
1794 err:
1795         geneve_dellink(dev, &list_kill);
1796         unregister_netdevice_many(&list_kill);
1797         return ERR_PTR(err);
1798 }
1799 EXPORT_SYMBOL_GPL(geneve_dev_create_fb);
1800
1801 static int geneve_netdevice_event(struct notifier_block *unused,
1802                                   unsigned long event, void *ptr)
1803 {
1804         struct net_device *dev = netdev_notifier_info_to_dev(ptr);
1805
1806         if (event == NETDEV_UDP_TUNNEL_PUSH_INFO ||
1807             event == NETDEV_UDP_TUNNEL_DROP_INFO) {
1808                 geneve_offload_rx_ports(dev, event == NETDEV_UDP_TUNNEL_PUSH_INFO);
1809         } else if (event == NETDEV_UNREGISTER) {
1810                 geneve_offload_rx_ports(dev, false);
1811         } else if (event == NETDEV_REGISTER) {
1812                 geneve_offload_rx_ports(dev, true);
1813         }
1814
1815         return NOTIFY_DONE;
1816 }
1817
1818 static struct notifier_block geneve_notifier_block __read_mostly = {
1819         .notifier_call = geneve_netdevice_event,
1820 };
1821
1822 static __net_init int geneve_init_net(struct net *net)
1823 {
1824         struct geneve_net *gn = net_generic(net, geneve_net_id);
1825
1826         INIT_LIST_HEAD(&gn->geneve_list);
1827         INIT_LIST_HEAD(&gn->sock_list);
1828         return 0;
1829 }
1830
1831 static void geneve_destroy_tunnels(struct net *net, struct list_head *head)
1832 {
1833         struct geneve_net *gn = net_generic(net, geneve_net_id);
1834         struct geneve_dev *geneve, *next;
1835         struct net_device *dev, *aux;
1836
1837         /* gather any geneve devices that were moved into this ns */
1838         for_each_netdev_safe(net, dev, aux)
1839                 if (dev->rtnl_link_ops == &geneve_link_ops)
1840                         unregister_netdevice_queue(dev, head);
1841
1842         /* now gather any other geneve devices that were created in this ns */
1843         list_for_each_entry_safe(geneve, next, &gn->geneve_list, next) {
1844                 /* If geneve->dev is in the same netns, it was already added
1845                  * to the list by the previous loop.
1846                  */
1847                 if (!net_eq(dev_net(geneve->dev), net))
1848                         unregister_netdevice_queue(geneve->dev, head);
1849         }
1850 }
1851
1852 static void __net_exit geneve_exit_batch_net(struct list_head *net_list)
1853 {
1854         struct net *net;
1855         LIST_HEAD(list);
1856
1857         rtnl_lock();
1858         list_for_each_entry(net, net_list, exit_list)
1859                 geneve_destroy_tunnels(net, &list);
1860
1861         /* unregister the devices gathered above */
1862         unregister_netdevice_many(&list);
1863         rtnl_unlock();
1864
1865         list_for_each_entry(net, net_list, exit_list) {
1866                 const struct geneve_net *gn = net_generic(net, geneve_net_id);
1867
1868                 WARN_ON_ONCE(!list_empty(&gn->sock_list));
1869         }
1870 }
1871
1872 static struct pernet_operations geneve_net_ops = {
1873         .init = geneve_init_net,
1874         .exit_batch = geneve_exit_batch_net,
1875         .id   = &geneve_net_id,
1876         .size = sizeof(struct geneve_net),
1877 };
1878
1879 static int __init geneve_init_module(void)
1880 {
1881         int rc;
1882
1883         rc = register_pernet_subsys(&geneve_net_ops);
1884         if (rc)
1885                 goto out1;
1886
1887         rc = register_netdevice_notifier(&geneve_notifier_block);
1888         if (rc)
1889                 goto out2;
1890
1891         rc = rtnl_link_register(&geneve_link_ops);
1892         if (rc)
1893                 goto out3;
1894
1895         return 0;
1896 out3:
1897         unregister_netdevice_notifier(&geneve_notifier_block);
1898 out2:
1899         unregister_pernet_subsys(&geneve_net_ops);
1900 out1:
1901         return rc;
1902 }
1903 late_initcall(geneve_init_module);
1904
1905 static void __exit geneve_cleanup_module(void)
1906 {
1907         rtnl_link_unregister(&geneve_link_ops);
1908         unregister_netdevice_notifier(&geneve_notifier_block);
1909         unregister_pernet_subsys(&geneve_net_ops);
1910 }
1911 module_exit(geneve_cleanup_module);
1912
1913 MODULE_LICENSE("GPL");
1914 MODULE_VERSION(GENEVE_NETDEV_VER);
1915 MODULE_AUTHOR("John W. Linville <linville@tuxdriver.com>");
1916 MODULE_DESCRIPTION("Interface driver for GENEVE encapsulated traffic");
1917 MODULE_ALIAS_RTNL_LINK("geneve");